void (*set_cr)(int cr, ulong val, struct kvm_vcpu *vcpu);
        int (*cpl)(struct kvm_vcpu *vcpu);
        void (*set_rflags)(struct kvm_vcpu *vcpu, unsigned long rflags);
+       int (*get_dr)(int dr, unsigned long *dest, struct kvm_vcpu *vcpu);
+       int (*set_dr)(int dr, unsigned long value, struct kvm_vcpu *vcpu);
 };
 
 /* Type, address-of, and value of an instruction's operand. */
 
 int kvm_emulate_halt(struct kvm_vcpu *vcpu);
 int emulate_invlpg(struct kvm_vcpu *vcpu, gva_t address);
 int emulate_clts(struct kvm_vcpu *vcpu);
-int emulator_get_dr(struct x86_emulate_ctxt *ctxt, int dr,
-                   unsigned long *dest);
-int emulator_set_dr(struct x86_emulate_ctxt *ctxt, int dr,
-                   unsigned long value);
 
 void kvm_get_segment(struct kvm_vcpu *vcpu, struct kvm_segment *var, int seg);
 int kvm_load_segment_descriptor(struct kvm_vcpu *vcpu, u16 selector, int seg);
 
                        kvm_queue_exception(ctxt->vcpu, UD_VECTOR);
                        goto done;
                }
-               emulator_get_dr(ctxt, c->modrm_reg, &c->regs[c->modrm_rm]);
+               ops->get_dr(c->modrm_reg, &c->regs[c->modrm_rm], ctxt->vcpu);
                c->dst.type = OP_NONE;  /* no writeback */
                break;
        case 0x22: /* mov reg, cr */
                        kvm_queue_exception(ctxt->vcpu, UD_VECTOR);
                        goto done;
                }
-               emulator_set_dr(ctxt, c->modrm_reg, c->regs[c->modrm_rm]);
+
+               ops->set_dr(c->modrm_reg,c->regs[c->modrm_rm] &
+                           ((ctxt->mode == X86EMUL_MODE_PROT64) ? ~0ULL : ~0U),
+                       ctxt->vcpu);
                c->dst.type = OP_NONE;  /* no writeback */
                break;
        case 0x30:
 
        return X86EMUL_CONTINUE;
 }
 
-int emulator_get_dr(struct x86_emulate_ctxt *ctxt, int dr, unsigned long *dest)
+int emulator_get_dr(int dr, unsigned long *dest, struct kvm_vcpu *vcpu)
 {
-       return kvm_get_dr(ctxt->vcpu, dr, dest);
+       return kvm_get_dr(vcpu, dr, dest);
 }
 
-int emulator_set_dr(struct x86_emulate_ctxt *ctxt, int dr, unsigned long value)
+int emulator_set_dr(int dr, unsigned long value, struct kvm_vcpu *vcpu)
 {
-       unsigned long mask = (ctxt->mode == X86EMUL_MODE_PROT64) ? ~0ULL : ~0U;
-
-       return kvm_set_dr(ctxt->vcpu, dr, value & mask);
+       return kvm_set_dr(vcpu, dr, value);
 }
 
 void kvm_report_emulation_failure(struct kvm_vcpu *vcpu, const char *context)
        .set_cr              = emulator_set_cr,
        .cpl                 = emulator_get_cpl,
        .set_rflags          = emulator_set_rflags,
+       .get_dr              = emulator_get_dr,
+       .set_dr              = emulator_set_dr,
 };
 
 static void cache_all_regs(struct kvm_vcpu *vcpu)