1
2
3
4
5
6
7
8#ifndef SELFTEST_KVM_PROCESSOR_H
9#define SELFTEST_KVM_PROCESSOR_H
10
11#include <assert.h>
12#include <stdint.h>
13
14#include <asm/msr-index.h>
15
16#define X86_EFLAGS_FIXED (1u << 1)
17
18#define X86_CR4_VME (1ul << 0)
19#define X86_CR4_PVI (1ul << 1)
20#define X86_CR4_TSD (1ul << 2)
21#define X86_CR4_DE (1ul << 3)
22#define X86_CR4_PSE (1ul << 4)
23#define X86_CR4_PAE (1ul << 5)
24#define X86_CR4_MCE (1ul << 6)
25#define X86_CR4_PGE (1ul << 7)
26#define X86_CR4_PCE (1ul << 8)
27#define X86_CR4_OSFXSR (1ul << 9)
28#define X86_CR4_OSXMMEXCPT (1ul << 10)
29#define X86_CR4_UMIP (1ul << 11)
30#define X86_CR4_LA57 (1ul << 12)
31#define X86_CR4_VMXE (1ul << 13)
32#define X86_CR4_SMXE (1ul << 14)
33#define X86_CR4_FSGSBASE (1ul << 16)
34#define X86_CR4_PCIDE (1ul << 17)
35#define X86_CR4_OSXSAVE (1ul << 18)
36#define X86_CR4_SMEP (1ul << 20)
37#define X86_CR4_SMAP (1ul << 21)
38#define X86_CR4_PKE (1ul << 22)
39
40
41#define CPUID_VMX (1ul << 5)
42#define CPUID_SMX (1ul << 6)
43#define CPUID_PCID (1ul << 17)
44#define CPUID_XSAVE (1ul << 26)
45
46
47#define CPUID_FSGSBASE (1ul << 0)
48#define CPUID_SMEP (1ul << 7)
49#define CPUID_SMAP (1ul << 20)
50
51
52#define CPUID_UMIP (1ul << 2)
53#define CPUID_PKU (1ul << 3)
54#define CPUID_LA57 (1ul << 16)
55
56#define UNEXPECTED_VECTOR_PORT 0xfff0u
57
58
59struct gpr64_regs {
60 u64 rax;
61 u64 rcx;
62 u64 rdx;
63 u64 rbx;
64 u64 rsp;
65 u64 rbp;
66 u64 rsi;
67 u64 rdi;
68 u64 r8;
69 u64 r9;
70 u64 r10;
71 u64 r11;
72 u64 r12;
73 u64 r13;
74 u64 r14;
75 u64 r15;
76};
77
78struct desc64 {
79 uint16_t limit0;
80 uint16_t base0;
81 unsigned base1:8, type:4, s:1, dpl:2, p:1;
82 unsigned limit1:4, avl:1, l:1, db:1, g:1, base2:8;
83 uint32_t base3;
84 uint32_t zero1;
85} __attribute__((packed));
86
87struct desc_ptr {
88 uint16_t size;
89 uint64_t address;
90} __attribute__((packed));
91
92static inline uint64_t get_desc64_base(const struct desc64 *desc)
93{
94 return ((uint64_t)desc->base3 << 32) |
95 (desc->base0 | ((desc->base1) << 16) | ((desc->base2) << 24));
96}
97
98static inline uint64_t rdtsc(void)
99{
100 uint32_t eax, edx;
101 uint64_t tsc_val;
102
103
104
105
106
107
108 __asm__ __volatile__("lfence; rdtsc; lfence" : "=a"(eax), "=d"(edx));
109 tsc_val = ((uint64_t)edx) << 32 | eax;
110 return tsc_val;
111}
112
113static inline uint64_t rdtscp(uint32_t *aux)
114{
115 uint32_t eax, edx;
116
117 __asm__ __volatile__("rdtscp" : "=a"(eax), "=d"(edx), "=c"(*aux));
118 return ((uint64_t)edx) << 32 | eax;
119}
120
121static inline uint64_t rdmsr(uint32_t msr)
122{
123 uint32_t a, d;
124
125 __asm__ __volatile__("rdmsr" : "=a"(a), "=d"(d) : "c"(msr) : "memory");
126
127 return a | ((uint64_t) d << 32);
128}
129
130static inline void wrmsr(uint32_t msr, uint64_t value)
131{
132 uint32_t a = value;
133 uint32_t d = value >> 32;
134
135 __asm__ __volatile__("wrmsr" :: "a"(a), "d"(d), "c"(msr) : "memory");
136}
137
138
139static inline uint16_t inw(uint16_t port)
140{
141 uint16_t tmp;
142
143 __asm__ __volatile__("in %%dx, %%ax"
144 : "=a" (tmp)
145 : "d" (port));
146
147 return tmp;
148}
149
150static inline uint16_t get_es(void)
151{
152 uint16_t es;
153
154 __asm__ __volatile__("mov %%es, %[es]"
155 : [es]"=rm"(es));
156 return es;
157}
158
159static inline uint16_t get_cs(void)
160{
161 uint16_t cs;
162
163 __asm__ __volatile__("mov %%cs, %[cs]"
164 : [cs]"=rm"(cs));
165 return cs;
166}
167
168static inline uint16_t get_ss(void)
169{
170 uint16_t ss;
171
172 __asm__ __volatile__("mov %%ss, %[ss]"
173 : [ss]"=rm"(ss));
174 return ss;
175}
176
177static inline uint16_t get_ds(void)
178{
179 uint16_t ds;
180
181 __asm__ __volatile__("mov %%ds, %[ds]"
182 : [ds]"=rm"(ds));
183 return ds;
184}
185
186static inline uint16_t get_fs(void)
187{
188 uint16_t fs;
189
190 __asm__ __volatile__("mov %%fs, %[fs]"
191 : [fs]"=rm"(fs));
192 return fs;
193}
194
195static inline uint16_t get_gs(void)
196{
197 uint16_t gs;
198
199 __asm__ __volatile__("mov %%gs, %[gs]"
200 : [gs]"=rm"(gs));
201 return gs;
202}
203
204static inline uint16_t get_tr(void)
205{
206 uint16_t tr;
207
208 __asm__ __volatile__("str %[tr]"
209 : [tr]"=rm"(tr));
210 return tr;
211}
212
213static inline uint64_t get_cr0(void)
214{
215 uint64_t cr0;
216
217 __asm__ __volatile__("mov %%cr0, %[cr0]"
218 : [cr0]"=r"(cr0));
219 return cr0;
220}
221
222static inline uint64_t get_cr3(void)
223{
224 uint64_t cr3;
225
226 __asm__ __volatile__("mov %%cr3, %[cr3]"
227 : [cr3]"=r"(cr3));
228 return cr3;
229}
230
231static inline uint64_t get_cr4(void)
232{
233 uint64_t cr4;
234
235 __asm__ __volatile__("mov %%cr4, %[cr4]"
236 : [cr4]"=r"(cr4));
237 return cr4;
238}
239
240static inline void set_cr4(uint64_t val)
241{
242 __asm__ __volatile__("mov %0, %%cr4" : : "r" (val) : "memory");
243}
244
245static inline struct desc_ptr get_gdt(void)
246{
247 struct desc_ptr gdt;
248 __asm__ __volatile__("sgdt %[gdt]"
249 : [gdt]"=m"(gdt));
250 return gdt;
251}
252
253static inline struct desc_ptr get_idt(void)
254{
255 struct desc_ptr idt;
256 __asm__ __volatile__("sidt %[idt]"
257 : [idt]"=m"(idt));
258 return idt;
259}
260
261static inline void outl(uint16_t port, uint32_t value)
262{
263 __asm__ __volatile__("outl %%eax, %%dx" : : "d"(port), "a"(value));
264}
265
266#define SET_XMM(__var, __xmm) \
267 asm volatile("movq %0, %%"#__xmm : : "r"(__var) : #__xmm)
268
269static inline void set_xmm(int n, unsigned long val)
270{
271 switch (n) {
272 case 0:
273 SET_XMM(val, xmm0);
274 break;
275 case 1:
276 SET_XMM(val, xmm1);
277 break;
278 case 2:
279 SET_XMM(val, xmm2);
280 break;
281 case 3:
282 SET_XMM(val, xmm3);
283 break;
284 case 4:
285 SET_XMM(val, xmm4);
286 break;
287 case 5:
288 SET_XMM(val, xmm5);
289 break;
290 case 6:
291 SET_XMM(val, xmm6);
292 break;
293 case 7:
294 SET_XMM(val, xmm7);
295 break;
296 }
297}
298
299typedef unsigned long v1di __attribute__ ((vector_size (8)));
300static inline unsigned long get_xmm(int n)
301{
302 assert(n >= 0 && n <= 7);
303
304 register v1di xmm0 __asm__("%xmm0");
305 register v1di xmm1 __asm__("%xmm1");
306 register v1di xmm2 __asm__("%xmm2");
307 register v1di xmm3 __asm__("%xmm3");
308 register v1di xmm4 __asm__("%xmm4");
309 register v1di xmm5 __asm__("%xmm5");
310 register v1di xmm6 __asm__("%xmm6");
311 register v1di xmm7 __asm__("%xmm7");
312 switch (n) {
313 case 0:
314 return (unsigned long)xmm0;
315 case 1:
316 return (unsigned long)xmm1;
317 case 2:
318 return (unsigned long)xmm2;
319 case 3:
320 return (unsigned long)xmm3;
321 case 4:
322 return (unsigned long)xmm4;
323 case 5:
324 return (unsigned long)xmm5;
325 case 6:
326 return (unsigned long)xmm6;
327 case 7:
328 return (unsigned long)xmm7;
329 }
330 return 0;
331}
332
333bool is_intel_cpu(void);
334
335struct kvm_x86_state;
336struct kvm_x86_state *vcpu_save_state(struct kvm_vm *vm, uint32_t vcpuid);
337void vcpu_load_state(struct kvm_vm *vm, uint32_t vcpuid,
338 struct kvm_x86_state *state);
339
340struct kvm_msr_list *kvm_get_msr_index_list(void);
341
342struct kvm_cpuid2 *kvm_get_supported_cpuid(void);
343void vcpu_set_cpuid(struct kvm_vm *vm, uint32_t vcpuid,
344 struct kvm_cpuid2 *cpuid);
345
346struct kvm_cpuid_entry2 *
347kvm_get_supported_cpuid_index(uint32_t function, uint32_t index);
348
349static inline struct kvm_cpuid_entry2 *
350kvm_get_supported_cpuid_entry(uint32_t function)
351{
352 return kvm_get_supported_cpuid_index(function, 0);
353}
354
355uint64_t vcpu_get_msr(struct kvm_vm *vm, uint32_t vcpuid, uint64_t msr_index);
356int _vcpu_set_msr(struct kvm_vm *vm, uint32_t vcpuid, uint64_t msr_index,
357 uint64_t msr_value);
358void vcpu_set_msr(struct kvm_vm *vm, uint32_t vcpuid, uint64_t msr_index,
359 uint64_t msr_value);
360
361uint32_t kvm_get_cpuid_max_basic(void);
362uint32_t kvm_get_cpuid_max_extended(void);
363void kvm_get_cpu_address_width(unsigned int *pa_bits, unsigned int *va_bits);
364
365struct ex_regs {
366 uint64_t rax, rcx, rdx, rbx;
367 uint64_t rbp, rsi, rdi;
368 uint64_t r8, r9, r10, r11;
369 uint64_t r12, r13, r14, r15;
370 uint64_t vector;
371 uint64_t error_code;
372 uint64_t rip;
373 uint64_t cs;
374 uint64_t rflags;
375};
376
377void vm_init_descriptor_tables(struct kvm_vm *vm);
378void vcpu_init_descriptor_tables(struct kvm_vm *vm, uint32_t vcpuid);
379void vm_handle_exception(struct kvm_vm *vm, int vector,
380 void (*handler)(struct ex_regs *));
381
382
383
384
385
386
387
388
389bool set_cpuid(struct kvm_cpuid2 *cpuid, struct kvm_cpuid_entry2 *ent);
390
391uint64_t kvm_hypercall(uint64_t nr, uint64_t a0, uint64_t a1, uint64_t a2,
392 uint64_t a3);
393
394
395
396
397#define X86_CR0_PE (1UL<<0)
398#define X86_CR0_MP (1UL<<1)
399#define X86_CR0_EM (1UL<<2)
400#define X86_CR0_TS (1UL<<3)
401#define X86_CR0_ET (1UL<<4)
402#define X86_CR0_NE (1UL<<5)
403#define X86_CR0_WP (1UL<<16)
404#define X86_CR0_AM (1UL<<18)
405#define X86_CR0_NW (1UL<<29)
406#define X86_CR0_CD (1UL<<30)
407#define X86_CR0_PG (1UL<<31)
408
409#define APIC_BASE_MSR 0x800
410#define X2APIC_ENABLE (1UL << 10)
411#define APIC_ICR 0x300
412#define APIC_DEST_SELF 0x40000
413#define APIC_DEST_ALLINC 0x80000
414#define APIC_DEST_ALLBUT 0xC0000
415#define APIC_ICR_RR_MASK 0x30000
416#define APIC_ICR_RR_INVALID 0x00000
417#define APIC_ICR_RR_INPROG 0x10000
418#define APIC_ICR_RR_VALID 0x20000
419#define APIC_INT_LEVELTRIG 0x08000
420#define APIC_INT_ASSERT 0x04000
421#define APIC_ICR_BUSY 0x01000
422#define APIC_DEST_LOGICAL 0x00800
423#define APIC_DEST_PHYSICAL 0x00000
424#define APIC_DM_FIXED 0x00000
425#define APIC_DM_FIXED_MASK 0x00700
426#define APIC_DM_LOWEST 0x00100
427#define APIC_DM_SMI 0x00200
428#define APIC_DM_REMRD 0x00300
429#define APIC_DM_NMI 0x00400
430#define APIC_DM_INIT 0x00500
431#define APIC_DM_STARTUP 0x00600
432#define APIC_DM_EXTINT 0x00700
433#define APIC_VECTOR_MASK 0x000FF
434#define APIC_ICR2 0x310
435
436
437#define VMX_EPT_VPID_CAP_AD_BITS (1ULL << 21)
438
439#endif
440