1 #ifndef __IDT_TEST__ 2 #define __IDT_TEST__ 3 4 #include <setjmp.h> 5 6 void setup_idt(void); 7 void setup_alt_stack(void); 8 9 struct ex_regs { 10 unsigned long rax, rcx, rdx, rbx; 11 unsigned long dummy, rbp, rsi, rdi; 12 #ifdef __x86_64__ 13 unsigned long r8, r9, r10, r11; 14 unsigned long r12, r13, r14, r15; 15 #endif 16 unsigned long vector; 17 unsigned long error_code; 18 unsigned long rip; 19 unsigned long cs; 20 unsigned long rflags; 21 }; 22 23 typedef void (*handler)(struct ex_regs *regs); 24 25 typedef struct { 26 u16 prev; 27 u16 res1; 28 u32 esp0; 29 u16 ss0; 30 u16 res2; 31 u32 esp1; 32 u16 ss1; 33 u16 res3; 34 u32 esp2; 35 u16 ss2; 36 u16 res4; 37 u32 cr3; 38 u32 eip; 39 u32 eflags; 40 u32 eax, ecx, edx, ebx, esp, ebp, esi, edi; 41 u16 es; 42 u16 res5; 43 u16 cs; 44 u16 res6; 45 u16 ss; 46 u16 res7; 47 u16 ds; 48 u16 res8; 49 u16 fs; 50 u16 res9; 51 u16 gs; 52 u16 res10; 53 u16 ldt; 54 u16 res11; 55 u16 t:1; 56 u16 res12:15; 57 u16 iomap_base; 58 } tss32_t; 59 60 typedef struct __attribute__((packed)) { 61 u32 res1; 62 u64 rsp0; 63 u64 rsp1; 64 u64 rsp2; 65 u64 res2; 66 u64 ist1; 67 u64 ist2; 68 u64 ist3; 69 u64 ist4; 70 u64 ist5; 71 u64 ist6; 72 u64 ist7; 73 u64 res3; 74 u16 res4; 75 u16 iomap_base; 76 } tss64_t; 77 78 #ifdef __x86_64 79 #define ASM_TRY(catch) \ 80 "movl $0, %%gs:4 \n\t" \ 81 ".pushsection .data.ex \n\t" \ 82 ".quad 1111f, " catch "\n\t" \ 83 ".popsection \n\t" \ 84 "1111:" 85 #else 86 #define ASM_TRY(catch) \ 87 "movl $0, %%gs:4 \n\t" \ 88 ".pushsection .data.ex \n\t" \ 89 ".long 1111f, " catch "\n\t" \ 90 ".popsection \n\t" \ 91 "1111:" 92 #endif 93 94 #define DB_VECTOR 1 95 #define BP_VECTOR 3 96 #define UD_VECTOR 6 97 #define GP_VECTOR 13 98 99 /* 100 * selector 32-bit 64-bit 101 * 0x00 NULL descriptor NULL descriptor 102 * 0x08 ring-0 code segment (32-bit) ring-0 code segment (64-bit) 103 * 0x10 ring-0 data segment (32-bit) ring-0 data segment (32/64-bit) 104 * 0x18 ring-0 code segment (P=0) ring-0 code segment (64-bit, P=0) 105 * 0x20 intr_alt_stack TSS ring-0 code segment (32-bit) 106 * 0x28 ring-0 code segment (16-bit) same 107 * 0x30 ring-0 data segment (16-bit) same 108 * 0x38 (0x3b) ring-3 code segment (32-bit) same 109 * 0x40 (0x43) ring-3 data segment (32-bit) ring-3 data segment (32/64-bit) 110 * 0x48 (0x4b) **unused** ring-3 code segment (64-bit) 111 * 0x50--0x78 free to use for test cases same 112 * 0x80 primary TSS (CPU 0) same 113 * 114 * Note that the same segment can be used for 32-bit and 64-bit data segments 115 * (the L bit is only defined for code segments) 116 * 117 * Selectors 0x08-0x10 and 0x3b-0x4b are set up for use with the SYSCALL 118 * and SYSRET instructions. 119 */ 120 121 #define KERNEL_CS 0x08 122 #define KERNEL_DS 0x10 123 #define NP_SEL 0x18 124 #ifdef __x86_64__ 125 #define KERNEL_CS32 0x20 126 #else 127 #define TSS_INTR 0x20 128 #endif 129 #define KERNEL_CS16 0x28 130 #define KERNEL_DS16 0x30 131 #define USER_CS32 0x3b 132 #define USER_DS 0x43 133 #ifdef __x86_64__ 134 #define USER_CS64 0x4b 135 #endif 136 137 /* Synonyms */ 138 #define KERNEL_DS32 KERNEL_DS 139 #define USER_DS32 USER_DS 140 141 #ifdef __x86_64__ 142 #define KERNEL_CS64 KERNEL_CS 143 #define USER_CS USER_CS64 144 #define KERNEL_DS64 KERNEL_DS 145 #define USER_DS64 USER_DS 146 #else 147 #define KERNEL_CS32 KERNEL_CS 148 #define USER_CS USER_CS32 149 #endif 150 151 #define FIRST_SPARE_SEL 0x50 152 #define TSS_MAIN 0x80 153 154 typedef struct { 155 unsigned short offset0; 156 unsigned short selector; 157 unsigned short ist : 3; 158 unsigned short : 5; 159 unsigned short type : 4; 160 unsigned short : 1; 161 unsigned short dpl : 2; 162 unsigned short p : 1; 163 unsigned short offset1; 164 #ifdef __x86_64__ 165 unsigned offset2; 166 unsigned reserved; 167 #endif 168 } idt_entry_t; 169 170 typedef struct { 171 u16 limit_low; 172 u16 base_low; 173 u8 base_middle; 174 u8 access; 175 u8 granularity; 176 u8 base_high; 177 } gdt_entry_t; 178 179 struct segment_desc64 { 180 uint16_t limit1; 181 uint16_t base1; 182 uint8_t base2; 183 union { 184 uint16_t type_limit_flags; /* Type and limit flags */ 185 struct { 186 uint16_t type:4; 187 uint16_t s:1; 188 uint16_t dpl:2; 189 uint16_t p:1; 190 uint16_t limit:4; 191 uint16_t avl:1; 192 uint16_t l:1; 193 uint16_t db:1; 194 uint16_t g:1; 195 } __attribute__((__packed__)); 196 } __attribute__((__packed__)); 197 uint8_t base3; 198 uint32_t base4; 199 uint32_t zero; 200 } __attribute__((__packed__)); 201 202 #define DESC_BUSY ((uint64_t) 1 << 41) 203 204 extern idt_entry_t boot_idt[256]; 205 206 #ifndef __x86_64__ 207 extern gdt_entry_t gdt32[]; 208 extern tss32_t tss; 209 extern tss32_t tss_intr; 210 void set_gdt_task_gate(u16 tss_sel, u16 sel); 211 void set_idt_task_gate(int vec, u16 sel); 212 void set_intr_task_gate(int vec, void *fn); 213 void setup_tss32(void); 214 #else 215 extern tss64_t tss; 216 #endif 217 218 unsigned exception_vector(void); 219 int write_cr4_checking(unsigned long val); 220 unsigned exception_error_code(void); 221 bool exception_rflags_rf(void); 222 void set_idt_entry(int vec, void *addr, int dpl); 223 void set_idt_sel(int vec, u16 sel); 224 void set_idt_dpl(int vec, u16 dpl); 225 void set_gdt_entry(int sel, u32 base, u32 limit, u8 access, u8 gran); 226 void set_intr_alt_stack(int e, void *fn); 227 void print_current_tss_info(void); 228 handler handle_exception(u8 v, handler fn); 229 void unhandled_exception(struct ex_regs *regs, bool cpu); 230 231 bool test_for_exception(unsigned int ex, void (*trigger_func)(void *data), 232 void *data); 233 void __set_exception_jmpbuf(jmp_buf *addr); 234 #define set_exception_jmpbuf(jmpbuf) \ 235 (setjmp(jmpbuf) ? : (__set_exception_jmpbuf(&(jmpbuf)), 0)) 236 237 static inline void *get_idt_addr(idt_entry_t *entry) 238 { 239 uintptr_t addr = entry->offset0 | ((u32)entry->offset1 << 16); 240 #ifdef __x86_64__ 241 addr |= (u64)entry->offset2 << 32; 242 #endif 243 return (void *)addr; 244 } 245 246 #endif 247