2 files changed, 19 insertions, 21 deletions

diff --git a/arch/x86/kvm/emulate.c b/arch/x86/kvm/emulate.c
index 0ff7d4bd1bb..57e0e291b38 100644
--- a/arch/x86/kvm/emulate.c
+++ b/arch/x86/kvm/emulate.c
@@ -3494,6 +3494,7 @@ x86_emulate_insn(struct x86_emulate_ctxt *ctxt)
         int rc = X86EMUL_CONTINUE;
         int saved_dst_type = c->dst.type;
         int irq; /* Used for int 3, int, and into */
+        struct desc_ptr desc_ptr;
 
         ctxt->decode.mem_read.pos = 0;
 
@@ -4005,9 +4006,6 @@ twobyte_insn:
         switch (c->b) {
         case 0x01: /* lgdt, lidt, lmsw */
                 switch (c->modrm_reg) {
-                        u16 size;
-                        unsigned long address;
-
                 case 0: /* vmcall */
                         if (c->modrm_mod != 3 || c->modrm_rm != 1)
                                 goto cannot_emulate;
@@ -4023,10 +4021,11 @@ twobyte_insn:
                         break;
                 case 2: /* lgdt */
                         rc = read_descriptor(ctxt, ops, c->src.addr.mem,
-                                             &size, &address, c->op_bytes);
+                                             &desc_ptr.size, &desc_ptr.address,
+                                             c->op_bytes);
                         if (rc != X86EMUL_CONTINUE)
                                 goto done;
-                        realmode_lgdt(ctxt->vcpu, size, address);
+                        ctxt->ops->set_gdt(ctxt, &desc_ptr);
                         /* Disable writeback. */
                         c->dst.type = OP_NONE;
                         break;
@@ -4041,11 +4040,12 @@ twobyte_insn:
                                 }
                         } else {
                                 rc = read_descriptor(ctxt, ops, c->src.addr.mem,
-                                                     &size, &address,
+                                                     &desc_ptr.size,
+                                                     &desc_ptr.address,
                                                      c->op_bytes);
                                 if (rc != X86EMUL_CONTINUE)
                                         goto done;
-                                realmode_lidt(ctxt->vcpu, size, address);
+                                ctxt->ops->set_idt(ctxt, &desc_ptr);
                         }
                         /* Disable writeback. */
                         c->dst.type = OP_NONE;
diff --git a/arch/x86/kvm/x86.c b/arch/x86/kvm/x86.c
index 4f7248ea6ca..7cd3a3b491d 100644
--- a/arch/x86/kvm/x86.c
+++ b/arch/x86/kvm/x86.c
@@ -4249,6 +4249,16 @@ static void emulator_get_idt(struct x86_emulate_ctxt *ctxt, struct desc_ptr *dt)
         kvm_x86_ops->get_idt(emul_to_vcpu(ctxt), dt);
 }
 
+static void emulator_set_gdt(struct x86_emulate_ctxt *ctxt, struct desc_ptr *dt)
+{
+        kvm_x86_ops->set_gdt(emul_to_vcpu(ctxt), dt);
+}
+
+static void emulator_set_idt(struct x86_emulate_ctxt *ctxt, struct desc_ptr *dt)
+{
+        kvm_x86_ops->set_idt(emul_to_vcpu(ctxt), dt);
+}
+
 static unsigned long emulator_get_cached_segment_base(
         struct x86_emulate_ctxt *ctxt, int seg)
 {
@@ -4388,6 +4398,8 @@ static struct x86_emulate_ops emulate_ops = {
         .get_cached_segment_base = emulator_get_cached_segment_base,
         .get_gdt             = emulator_get_gdt,
         .get_idt             = emulator_get_idt,
+        .set_gdt             = emulator_set_gdt,
+        .set_idt             = emulator_set_idt,
         .get_cr              = emulator_get_cr,
         .set_cr              = emulator_set_cr,
         .cpl                 = emulator_get_cpl,
@@ -5049,20 +5061,6 @@ int kvm_fix_hypercall(struct kvm_vcpu *vcpu)
                                        rip, instruction, 3, NULL);
 }
 
-void realmode_lgdt(struct kvm_vcpu *vcpu, u16 limit, unsigned long base)
-{
-        struct desc_ptr dt = { limit, base };
-
-        kvm_x86_ops->set_gdt(vcpu, &dt);
-}
-
-void realmode_lidt(struct kvm_vcpu *vcpu, u16 limit, unsigned long base)
-{
-        struct desc_ptr dt = { limit, base };
-
-        kvm_x86_ops->set_idt(vcpu, &dt);
-}
-
 static int move_to_next_stateful_cpuid_entry(struct kvm_vcpu *vcpu, int i)
 {
         struct kvm_cpuid_entry2 *e = &vcpu->arch.cpuid_entries[i];