HVM-706 need static instrumentation in vcpu_load()/vcpu_put()

1 files changed, 4 insertions, 0 deletions

diff --git a/kvm.c b/kvm.c
index a7a9998..2383b1d 100644
--- a/kvm.c
+++ b/kvm.c
@@ -484,6 +484,8 @@ vcpu_load(struct kvm_vcpu *vcpu)
 	installctx(curthread, vcpu, kvm_ctx_save, kvm_ctx_restore, NULL,
 	    NULL, NULL, NULL);
 	kvm_arch_vcpu_load(vcpu, cpu);
+	kvm_ringbuf_record(&vcpu->kvcpu_ringbuf,
+	    KVM_RINGBUF_TAG_VCPULOAD, vcpu->cpu);
 	kpreempt_enable();
 }
 
@@ -502,6 +504,8 @@ vcpu_put(struct kvm_vcpu *vcpu)
 	kvm_fire_urn(vcpu);
 	removectx(curthread, vcpu, kvm_ctx_save, kvm_ctx_restore, NULL,
 	    NULL, NULL, NULL);
+	kvm_ringbuf_record(&vcpu->kvcpu_ringbuf,
+	    KVM_RINGBUF_TAG_VCPUPUT, vcpu->cpu);
 	kpreempt_enable();
 	mutex_exit(&vcpu->mutex);
 }