KVM: MMU: Use slab caches for shadow pages and their headers
[safe/jmp/linux-2.6] / drivers / kvm / kvm.h
1 #ifndef __KVM_H
2 #define __KVM_H
3
4 /*
5  * This work is licensed under the terms of the GNU GPL, version 2.  See
6  * the COPYING file in the top-level directory.
7  */
8
9 #include <linux/types.h>
10 #include <linux/list.h>
11 #include <linux/mutex.h>
12 #include <linux/spinlock.h>
13 #include <linux/signal.h>
14 #include <linux/sched.h>
15 #include <linux/mm.h>
16 #include <asm/signal.h>
17
18 #include "vmx.h"
19 #include <linux/kvm.h>
20 #include <linux/kvm_para.h>
21
22 #define CR0_PE_MASK (1ULL << 0)
23 #define CR0_MP_MASK (1ULL << 1)
24 #define CR0_TS_MASK (1ULL << 3)
25 #define CR0_NE_MASK (1ULL << 5)
26 #define CR0_WP_MASK (1ULL << 16)
27 #define CR0_NW_MASK (1ULL << 29)
28 #define CR0_CD_MASK (1ULL << 30)
29 #define CR0_PG_MASK (1ULL << 31)
30
31 #define CR3_WPT_MASK (1ULL << 3)
32 #define CR3_PCD_MASK (1ULL << 4)
33
34 #define CR3_RESEVED_BITS 0x07ULL
35 #define CR3_L_MODE_RESEVED_BITS (~((1ULL << 40) - 1) | 0x0fe7ULL)
36 #define CR3_FLAGS_MASK ((1ULL << 5) - 1)
37
38 #define CR4_VME_MASK (1ULL << 0)
39 #define CR4_PSE_MASK (1ULL << 4)
40 #define CR4_PAE_MASK (1ULL << 5)
41 #define CR4_PGE_MASK (1ULL << 7)
42 #define CR4_VMXE_MASK (1ULL << 13)
43
44 #define KVM_GUEST_CR0_MASK \
45         (CR0_PG_MASK | CR0_PE_MASK | CR0_WP_MASK | CR0_NE_MASK \
46          | CR0_NW_MASK | CR0_CD_MASK)
47 #define KVM_VM_CR0_ALWAYS_ON \
48         (CR0_PG_MASK | CR0_PE_MASK | CR0_WP_MASK | CR0_NE_MASK | CR0_TS_MASK \
49          | CR0_MP_MASK)
50 #define KVM_GUEST_CR4_MASK \
51         (CR4_PSE_MASK | CR4_PAE_MASK | CR4_PGE_MASK | CR4_VMXE_MASK | CR4_VME_MASK)
52 #define KVM_PMODE_VM_CR4_ALWAYS_ON (CR4_VMXE_MASK | CR4_PAE_MASK)
53 #define KVM_RMODE_VM_CR4_ALWAYS_ON (CR4_VMXE_MASK | CR4_PAE_MASK | CR4_VME_MASK)
54
55 #define INVALID_PAGE (~(hpa_t)0)
56 #define UNMAPPED_GVA (~(gpa_t)0)
57
58 #define KVM_MAX_VCPUS 1
59 #define KVM_ALIAS_SLOTS 4
60 #define KVM_MEMORY_SLOTS 4
61 #define KVM_NUM_MMU_PAGES 1024
62 #define KVM_MIN_FREE_MMU_PAGES 5
63 #define KVM_REFILL_PAGES 25
64 #define KVM_MAX_CPUID_ENTRIES 40
65
66 #define FX_IMAGE_SIZE 512
67 #define FX_IMAGE_ALIGN 16
68 #define FX_BUF_SIZE (2 * FX_IMAGE_SIZE + FX_IMAGE_ALIGN)
69
70 #define DE_VECTOR 0
71 #define NM_VECTOR 7
72 #define DF_VECTOR 8
73 #define TS_VECTOR 10
74 #define NP_VECTOR 11
75 #define SS_VECTOR 12
76 #define GP_VECTOR 13
77 #define PF_VECTOR 14
78
79 #define SELECTOR_TI_MASK (1 << 2)
80 #define SELECTOR_RPL_MASK 0x03
81
82 #define IOPL_SHIFT 12
83
84 #define KVM_PIO_PAGE_OFFSET 1
85
86 /*
87  * Address types:
88  *
89  *  gva - guest virtual address
90  *  gpa - guest physical address
91  *  gfn - guest frame number
92  *  hva - host virtual address
93  *  hpa - host physical address
94  *  hfn - host frame number
95  */
96
97 typedef unsigned long  gva_t;
98 typedef u64            gpa_t;
99 typedef unsigned long  gfn_t;
100
101 typedef unsigned long  hva_t;
102 typedef u64            hpa_t;
103 typedef unsigned long  hfn_t;
104
105 #define NR_PTE_CHAIN_ENTRIES 5
106
107 struct kvm_pte_chain {
108         u64 *parent_ptes[NR_PTE_CHAIN_ENTRIES];
109         struct hlist_node link;
110 };
111
112 /*
113  * kvm_mmu_page_role, below, is defined as:
114  *
115  *   bits 0:3 - total guest paging levels (2-4, or zero for real mode)
116  *   bits 4:7 - page table level for this shadow (1-4)
117  *   bits 8:9 - page table quadrant for 2-level guests
118  *   bit   16 - "metaphysical" - gfn is not a real page (huge page/real mode)
119  *   bits 17:18 - "access" - the user and writable bits of a huge page pde
120  */
121 union kvm_mmu_page_role {
122         unsigned word;
123         struct {
124                 unsigned glevels : 4;
125                 unsigned level : 4;
126                 unsigned quadrant : 2;
127                 unsigned pad_for_nice_hex_output : 6;
128                 unsigned metaphysical : 1;
129                 unsigned hugepage_access : 2;
130         };
131 };
132
133 struct kvm_mmu_page {
134         struct list_head link;
135         struct hlist_node hash_link;
136
137         /*
138          * The following two entries are used to key the shadow page in the
139          * hash table.
140          */
141         gfn_t gfn;
142         union kvm_mmu_page_role role;
143
144         u64 *spt;
145         unsigned long slot_bitmap; /* One bit set per slot which has memory
146                                     * in this shadow page.
147                                     */
148         int multimapped;         /* More than one parent_pte? */
149         int root_count;          /* Currently serving as active root */
150         union {
151                 u64 *parent_pte;               /* !multimapped */
152                 struct hlist_head parent_ptes; /* multimapped, kvm_pte_chain */
153         };
154 };
155
156 struct vmcs {
157         u32 revision_id;
158         u32 abort;
159         char data[0];
160 };
161
162 #define vmx_msr_entry kvm_msr_entry
163
164 struct kvm_vcpu;
165
166 /*
167  * x86 supports 3 paging modes (4-level 64-bit, 3-level 64-bit, and 2-level
168  * 32-bit).  The kvm_mmu structure abstracts the details of the current mmu
169  * mode.
170  */
171 struct kvm_mmu {
172         void (*new_cr3)(struct kvm_vcpu *vcpu);
173         int (*page_fault)(struct kvm_vcpu *vcpu, gva_t gva, u32 err);
174         void (*free)(struct kvm_vcpu *vcpu);
175         gpa_t (*gva_to_gpa)(struct kvm_vcpu *vcpu, gva_t gva);
176         hpa_t root_hpa;
177         int root_level;
178         int shadow_root_level;
179
180         u64 *pae_root;
181 };
182
183 #define KVM_NR_MEM_OBJS 20
184
185 struct kvm_mmu_memory_cache {
186         int nobjs;
187         void *objects[KVM_NR_MEM_OBJS];
188 };
189
190 /*
191  * We don't want allocation failures within the mmu code, so we preallocate
192  * enough memory for a single page fault in a cache.
193  */
194 struct kvm_guest_debug {
195         int enabled;
196         unsigned long bp[4];
197         int singlestep;
198 };
199
200 enum {
201         VCPU_REGS_RAX = 0,
202         VCPU_REGS_RCX = 1,
203         VCPU_REGS_RDX = 2,
204         VCPU_REGS_RBX = 3,
205         VCPU_REGS_RSP = 4,
206         VCPU_REGS_RBP = 5,
207         VCPU_REGS_RSI = 6,
208         VCPU_REGS_RDI = 7,
209 #ifdef CONFIG_X86_64
210         VCPU_REGS_R8 = 8,
211         VCPU_REGS_R9 = 9,
212         VCPU_REGS_R10 = 10,
213         VCPU_REGS_R11 = 11,
214         VCPU_REGS_R12 = 12,
215         VCPU_REGS_R13 = 13,
216         VCPU_REGS_R14 = 14,
217         VCPU_REGS_R15 = 15,
218 #endif
219         NR_VCPU_REGS
220 };
221
222 enum {
223         VCPU_SREG_CS,
224         VCPU_SREG_DS,
225         VCPU_SREG_ES,
226         VCPU_SREG_FS,
227         VCPU_SREG_GS,
228         VCPU_SREG_SS,
229         VCPU_SREG_TR,
230         VCPU_SREG_LDTR,
231 };
232
233 struct kvm_pio_request {
234         unsigned long count;
235         int cur_count;
236         struct page *guest_pages[2];
237         unsigned guest_page_offset;
238         int in;
239         int size;
240         int string;
241         int down;
242         int rep;
243 };
244
245 struct kvm_stat {
246         u32 pf_fixed;
247         u32 pf_guest;
248         u32 tlb_flush;
249         u32 invlpg;
250
251         u32 exits;
252         u32 io_exits;
253         u32 mmio_exits;
254         u32 signal_exits;
255         u32 irq_window_exits;
256         u32 halt_exits;
257         u32 request_irq_exits;
258         u32 irq_exits;
259         u32 light_exits;
260         u32 efer_reload;
261 };
262
263 struct kvm_vcpu {
264         struct kvm *kvm;
265         union {
266                 struct vmcs *vmcs;
267                 struct vcpu_svm *svm;
268         };
269         struct mutex mutex;
270         int   cpu;
271         int   launched;
272         u64 host_tsc;
273         struct kvm_run *run;
274         int interrupt_window_open;
275         unsigned long irq_summary; /* bit vector: 1 per word in irq_pending */
276 #define NR_IRQ_WORDS KVM_IRQ_BITMAP_SIZE(unsigned long)
277         unsigned long irq_pending[NR_IRQ_WORDS];
278         unsigned long regs[NR_VCPU_REGS]; /* for rsp: vcpu_load_rsp_rip() */
279         unsigned long rip;      /* needs vcpu_load_rsp_rip() */
280
281         unsigned long cr0;
282         unsigned long cr2;
283         unsigned long cr3;
284         gpa_t para_state_gpa;
285         struct page *para_state_page;
286         gpa_t hypercall_gpa;
287         unsigned long cr4;
288         unsigned long cr8;
289         u64 pdptrs[4]; /* pae */
290         u64 shadow_efer;
291         u64 apic_base;
292         u64 ia32_misc_enable_msr;
293         int nmsrs;
294         int save_nmsrs;
295         int msr_offset_efer;
296 #ifdef CONFIG_X86_64
297         int msr_offset_kernel_gs_base;
298 #endif
299         struct vmx_msr_entry *guest_msrs;
300         struct vmx_msr_entry *host_msrs;
301
302         struct kvm_mmu mmu;
303
304         struct kvm_mmu_memory_cache mmu_pte_chain_cache;
305         struct kvm_mmu_memory_cache mmu_rmap_desc_cache;
306         struct kvm_mmu_memory_cache mmu_page_cache;
307         struct kvm_mmu_memory_cache mmu_page_header_cache;
308
309         gfn_t last_pt_write_gfn;
310         int   last_pt_write_count;
311
312         struct kvm_guest_debug guest_debug;
313
314         char fx_buf[FX_BUF_SIZE];
315         char *host_fx_image;
316         char *guest_fx_image;
317         int fpu_active;
318         int guest_fpu_loaded;
319         struct vmx_host_state {
320                 int loaded;
321                 u16 fs_sel, gs_sel, ldt_sel;
322                 int fs_gs_ldt_reload_needed;
323         } vmx_host_state;
324
325         int mmio_needed;
326         int mmio_read_completed;
327         int mmio_is_write;
328         int mmio_size;
329         unsigned char mmio_data[8];
330         gpa_t mmio_phys_addr;
331         gva_t mmio_fault_cr2;
332         struct kvm_pio_request pio;
333         void *pio_data;
334
335         int sigset_active;
336         sigset_t sigset;
337
338         struct kvm_stat stat;
339
340         struct {
341                 int active;
342                 u8 save_iopl;
343                 struct kvm_save_segment {
344                         u16 selector;
345                         unsigned long base;
346                         u32 limit;
347                         u32 ar;
348                 } tr, es, ds, fs, gs;
349         } rmode;
350
351         int cpuid_nent;
352         struct kvm_cpuid_entry cpuid_entries[KVM_MAX_CPUID_ENTRIES];
353 };
354
355 struct kvm_mem_alias {
356         gfn_t base_gfn;
357         unsigned long npages;
358         gfn_t target_gfn;
359 };
360
361 struct kvm_memory_slot {
362         gfn_t base_gfn;
363         unsigned long npages;
364         unsigned long flags;
365         struct page **phys_mem;
366         unsigned long *dirty_bitmap;
367 };
368
369 struct kvm {
370         spinlock_t lock; /* protects everything except vcpus */
371         int naliases;
372         struct kvm_mem_alias aliases[KVM_ALIAS_SLOTS];
373         int nmemslots;
374         struct kvm_memory_slot memslots[KVM_MEMORY_SLOTS];
375         /*
376          * Hash table of struct kvm_mmu_page.
377          */
378         struct list_head active_mmu_pages;
379         int n_free_mmu_pages;
380         struct hlist_head mmu_page_hash[KVM_NUM_MMU_PAGES];
381         struct kvm_vcpu vcpus[KVM_MAX_VCPUS];
382         int memory_config_version;
383         int busy;
384         unsigned long rmap_overflow;
385         struct list_head vm_list;
386         struct file *filp;
387 };
388
389 struct descriptor_table {
390         u16 limit;
391         unsigned long base;
392 } __attribute__((packed));
393
394 struct kvm_arch_ops {
395         int (*cpu_has_kvm_support)(void);          /* __init */
396         int (*disabled_by_bios)(void);             /* __init */
397         void (*hardware_enable)(void *dummy);      /* __init */
398         void (*hardware_disable)(void *dummy);
399         int (*hardware_setup)(void);               /* __init */
400         void (*hardware_unsetup)(void);            /* __exit */
401
402         int (*vcpu_create)(struct kvm_vcpu *vcpu);
403         void (*vcpu_free)(struct kvm_vcpu *vcpu);
404
405         void (*vcpu_load)(struct kvm_vcpu *vcpu);
406         void (*vcpu_put)(struct kvm_vcpu *vcpu);
407         void (*vcpu_decache)(struct kvm_vcpu *vcpu);
408
409         int (*set_guest_debug)(struct kvm_vcpu *vcpu,
410                                struct kvm_debug_guest *dbg);
411         int (*get_msr)(struct kvm_vcpu *vcpu, u32 msr_index, u64 *pdata);
412         int (*set_msr)(struct kvm_vcpu *vcpu, u32 msr_index, u64 data);
413         u64 (*get_segment_base)(struct kvm_vcpu *vcpu, int seg);
414         void (*get_segment)(struct kvm_vcpu *vcpu,
415                             struct kvm_segment *var, int seg);
416         void (*set_segment)(struct kvm_vcpu *vcpu,
417                             struct kvm_segment *var, int seg);
418         void (*get_cs_db_l_bits)(struct kvm_vcpu *vcpu, int *db, int *l);
419         void (*decache_cr4_guest_bits)(struct kvm_vcpu *vcpu);
420         void (*set_cr0)(struct kvm_vcpu *vcpu, unsigned long cr0);
421         void (*set_cr3)(struct kvm_vcpu *vcpu, unsigned long cr3);
422         void (*set_cr4)(struct kvm_vcpu *vcpu, unsigned long cr4);
423         void (*set_efer)(struct kvm_vcpu *vcpu, u64 efer);
424         void (*get_idt)(struct kvm_vcpu *vcpu, struct descriptor_table *dt);
425         void (*set_idt)(struct kvm_vcpu *vcpu, struct descriptor_table *dt);
426         void (*get_gdt)(struct kvm_vcpu *vcpu, struct descriptor_table *dt);
427         void (*set_gdt)(struct kvm_vcpu *vcpu, struct descriptor_table *dt);
428         unsigned long (*get_dr)(struct kvm_vcpu *vcpu, int dr);
429         void (*set_dr)(struct kvm_vcpu *vcpu, int dr, unsigned long value,
430                        int *exception);
431         void (*cache_regs)(struct kvm_vcpu *vcpu);
432         void (*decache_regs)(struct kvm_vcpu *vcpu);
433         unsigned long (*get_rflags)(struct kvm_vcpu *vcpu);
434         void (*set_rflags)(struct kvm_vcpu *vcpu, unsigned long rflags);
435
436         void (*invlpg)(struct kvm_vcpu *vcpu, gva_t addr);
437         void (*tlb_flush)(struct kvm_vcpu *vcpu);
438         void (*inject_page_fault)(struct kvm_vcpu *vcpu,
439                                   unsigned long addr, u32 err_code);
440
441         void (*inject_gp)(struct kvm_vcpu *vcpu, unsigned err_code);
442
443         int (*run)(struct kvm_vcpu *vcpu, struct kvm_run *run);
444         int (*vcpu_setup)(struct kvm_vcpu *vcpu);
445         void (*skip_emulated_instruction)(struct kvm_vcpu *vcpu);
446         void (*patch_hypercall)(struct kvm_vcpu *vcpu,
447                                 unsigned char *hypercall_addr);
448 };
449
450 extern struct kvm_arch_ops *kvm_arch_ops;
451
452 #define kvm_printf(kvm, fmt ...) printk(KERN_DEBUG fmt)
453 #define vcpu_printf(vcpu, fmt...) kvm_printf(vcpu->kvm, fmt)
454
455 int kvm_init_arch(struct kvm_arch_ops *ops, struct module *module);
456 void kvm_exit_arch(void);
457
458 int kvm_mmu_module_init(void);
459 void kvm_mmu_module_exit(void);
460
461 void kvm_mmu_destroy(struct kvm_vcpu *vcpu);
462 int kvm_mmu_create(struct kvm_vcpu *vcpu);
463 int kvm_mmu_setup(struct kvm_vcpu *vcpu);
464
465 int kvm_mmu_reset_context(struct kvm_vcpu *vcpu);
466 void kvm_mmu_slot_remove_write_access(struct kvm_vcpu *vcpu, int slot);
467 void kvm_mmu_zap_all(struct kvm_vcpu *vcpu);
468
469 hpa_t gpa_to_hpa(struct kvm_vcpu *vcpu, gpa_t gpa);
470 #define HPA_MSB ((sizeof(hpa_t) * 8) - 1)
471 #define HPA_ERR_MASK ((hpa_t)1 << HPA_MSB)
472 static inline int is_error_hpa(hpa_t hpa) { return hpa >> HPA_MSB; }
473 hpa_t gva_to_hpa(struct kvm_vcpu *vcpu, gva_t gva);
474 struct page *gva_to_page(struct kvm_vcpu *vcpu, gva_t gva);
475
476 void kvm_emulator_want_group7_invlpg(void);
477
478 extern hpa_t bad_page_address;
479
480 struct page *gfn_to_page(struct kvm *kvm, gfn_t gfn);
481 struct kvm_memory_slot *gfn_to_memslot(struct kvm *kvm, gfn_t gfn);
482 void mark_page_dirty(struct kvm *kvm, gfn_t gfn);
483
484 enum emulation_result {
485         EMULATE_DONE,       /* no further processing */
486         EMULATE_DO_MMIO,      /* kvm_run filled with mmio request */
487         EMULATE_FAIL,         /* can't emulate this instruction */
488 };
489
490 int emulate_instruction(struct kvm_vcpu *vcpu, struct kvm_run *run,
491                         unsigned long cr2, u16 error_code);
492 void realmode_lgdt(struct kvm_vcpu *vcpu, u16 size, unsigned long address);
493 void realmode_lidt(struct kvm_vcpu *vcpu, u16 size, unsigned long address);
494 void realmode_lmsw(struct kvm_vcpu *vcpu, unsigned long msw,
495                    unsigned long *rflags);
496
497 unsigned long realmode_get_cr(struct kvm_vcpu *vcpu, int cr);
498 void realmode_set_cr(struct kvm_vcpu *vcpu, int cr, unsigned long value,
499                      unsigned long *rflags);
500
501 struct x86_emulate_ctxt;
502
503 int kvm_setup_pio(struct kvm_vcpu *vcpu, struct kvm_run *run, int in,
504                   int size, unsigned long count, int string, int down,
505                   gva_t address, int rep, unsigned port);
506 void kvm_emulate_cpuid(struct kvm_vcpu *vcpu);
507 int emulate_invlpg(struct kvm_vcpu *vcpu, gva_t address);
508 int emulate_clts(struct kvm_vcpu *vcpu);
509 int emulator_get_dr(struct x86_emulate_ctxt* ctxt, int dr,
510                     unsigned long *dest);
511 int emulator_set_dr(struct x86_emulate_ctxt *ctxt, int dr,
512                     unsigned long value);
513
514 void set_cr0(struct kvm_vcpu *vcpu, unsigned long cr0);
515 void set_cr3(struct kvm_vcpu *vcpu, unsigned long cr0);
516 void set_cr4(struct kvm_vcpu *vcpu, unsigned long cr0);
517 void set_cr8(struct kvm_vcpu *vcpu, unsigned long cr0);
518 void lmsw(struct kvm_vcpu *vcpu, unsigned long msw);
519
520 int kvm_get_msr_common(struct kvm_vcpu *vcpu, u32 msr, u64 *pdata);
521 int kvm_set_msr_common(struct kvm_vcpu *vcpu, u32 msr, u64 data);
522
523 void fx_init(struct kvm_vcpu *vcpu);
524
525 void load_msrs(struct vmx_msr_entry *e, int n);
526 void save_msrs(struct vmx_msr_entry *e, int n);
527 void kvm_resched(struct kvm_vcpu *vcpu);
528 void kvm_load_guest_fpu(struct kvm_vcpu *vcpu);
529 void kvm_put_guest_fpu(struct kvm_vcpu *vcpu);
530
531 int kvm_read_guest(struct kvm_vcpu *vcpu,
532                gva_t addr,
533                unsigned long size,
534                void *dest);
535
536 int kvm_write_guest(struct kvm_vcpu *vcpu,
537                 gva_t addr,
538                 unsigned long size,
539                 void *data);
540
541 unsigned long segment_base(u16 selector);
542
543 void kvm_mmu_pte_write(struct kvm_vcpu *vcpu, gpa_t gpa,
544                        const u8 *old, const u8 *new, int bytes);
545 int kvm_mmu_unprotect_page_virt(struct kvm_vcpu *vcpu, gva_t gva);
546 void kvm_mmu_free_some_pages(struct kvm_vcpu *vcpu);
547
548 int kvm_hypercall(struct kvm_vcpu *vcpu, struct kvm_run *run);
549
550 static inline int kvm_mmu_page_fault(struct kvm_vcpu *vcpu, gva_t gva,
551                                      u32 error_code)
552 {
553         if (unlikely(vcpu->kvm->n_free_mmu_pages < KVM_MIN_FREE_MMU_PAGES))
554                 kvm_mmu_free_some_pages(vcpu);
555         return vcpu->mmu.page_fault(vcpu, gva, error_code);
556 }
557
558 static inline int is_long_mode(struct kvm_vcpu *vcpu)
559 {
560 #ifdef CONFIG_X86_64
561         return vcpu->shadow_efer & EFER_LME;
562 #else
563         return 0;
564 #endif
565 }
566
567 static inline int is_pae(struct kvm_vcpu *vcpu)
568 {
569         return vcpu->cr4 & CR4_PAE_MASK;
570 }
571
572 static inline int is_pse(struct kvm_vcpu *vcpu)
573 {
574         return vcpu->cr4 & CR4_PSE_MASK;
575 }
576
577 static inline int is_paging(struct kvm_vcpu *vcpu)
578 {
579         return vcpu->cr0 & CR0_PG_MASK;
580 }
581
582 static inline int memslot_id(struct kvm *kvm, struct kvm_memory_slot *slot)
583 {
584         return slot - kvm->memslots;
585 }
586
587 static inline struct kvm_mmu_page *page_header(hpa_t shadow_page)
588 {
589         struct page *page = pfn_to_page(shadow_page >> PAGE_SHIFT);
590
591         return (struct kvm_mmu_page *)page_private(page);
592 }
593
594 static inline u16 read_fs(void)
595 {
596         u16 seg;
597         asm ("mov %%fs, %0" : "=g"(seg));
598         return seg;
599 }
600
601 static inline u16 read_gs(void)
602 {
603         u16 seg;
604         asm ("mov %%gs, %0" : "=g"(seg));
605         return seg;
606 }
607
608 static inline u16 read_ldt(void)
609 {
610         u16 ldt;
611         asm ("sldt %0" : "=g"(ldt));
612         return ldt;
613 }
614
615 static inline void load_fs(u16 sel)
616 {
617         asm ("mov %0, %%fs" : : "rm"(sel));
618 }
619
620 static inline void load_gs(u16 sel)
621 {
622         asm ("mov %0, %%gs" : : "rm"(sel));
623 }
624
625 #ifndef load_ldt
626 static inline void load_ldt(u16 sel)
627 {
628         asm ("lldt %0" : : "rm"(sel));
629 }
630 #endif
631
632 static inline void get_idt(struct descriptor_table *table)
633 {
634         asm ("sidt %0" : "=m"(*table));
635 }
636
637 static inline void get_gdt(struct descriptor_table *table)
638 {
639         asm ("sgdt %0" : "=m"(*table));
640 }
641
642 static inline unsigned long read_tr_base(void)
643 {
644         u16 tr;
645         asm ("str %0" : "=g"(tr));
646         return segment_base(tr);
647 }
648
649 #ifdef CONFIG_X86_64
650 static inline unsigned long read_msr(unsigned long msr)
651 {
652         u64 value;
653
654         rdmsrl(msr, value);
655         return value;
656 }
657 #endif
658
659 static inline void fx_save(void *image)
660 {
661         asm ("fxsave (%0)":: "r" (image));
662 }
663
664 static inline void fx_restore(void *image)
665 {
666         asm ("fxrstor (%0)":: "r" (image));
667 }
668
669 static inline void fpu_init(void)
670 {
671         asm ("finit");
672 }
673
674 static inline u32 get_rdx_init_val(void)
675 {
676         return 0x600; /* P6 family */
677 }
678
679 #define ASM_VMX_VMCLEAR_RAX       ".byte 0x66, 0x0f, 0xc7, 0x30"
680 #define ASM_VMX_VMLAUNCH          ".byte 0x0f, 0x01, 0xc2"
681 #define ASM_VMX_VMRESUME          ".byte 0x0f, 0x01, 0xc3"
682 #define ASM_VMX_VMPTRLD_RAX       ".byte 0x0f, 0xc7, 0x30"
683 #define ASM_VMX_VMREAD_RDX_RAX    ".byte 0x0f, 0x78, 0xd0"
684 #define ASM_VMX_VMWRITE_RAX_RDX   ".byte 0x0f, 0x79, 0xd0"
685 #define ASM_VMX_VMWRITE_RSP_RDX   ".byte 0x0f, 0x79, 0xd4"
686 #define ASM_VMX_VMXOFF            ".byte 0x0f, 0x01, 0xc4"
687 #define ASM_VMX_VMXON_RAX         ".byte 0xf3, 0x0f, 0xc7, 0x30"
688
689 #define MSR_IA32_TIME_STAMP_COUNTER             0x010
690
691 #define TSS_IOPB_BASE_OFFSET 0x66
692 #define TSS_BASE_SIZE 0x68
693 #define TSS_IOPB_SIZE (65536 / 8)
694 #define TSS_REDIRECTION_SIZE (256 / 8)
695 #define RMODE_TSS_SIZE (TSS_BASE_SIZE + TSS_REDIRECTION_SIZE + TSS_IOPB_SIZE + 1)
696
697 #endif