1 files changed, 16 insertions, 0 deletions

diff --git a/tools/testing/selftests/kvm/lib/kvm_util.c b/tools/testing/selftests/kvm/lib/kvm_util.c
index b52cfdefecbf..efa0aad8b3c6 100644
--- a/tools/testing/selftests/kvm/lib/kvm_util.c
+++ b/tools/testing/selftests/kvm/lib/kvm_util.c
@@ -1121,6 +1121,22 @@ int _vcpu_run(struct kvm_vm *vm, uint32_t vcpuid)
         return rc;
 }
 
+void vcpu_run_complete_io(struct kvm_vm *vm, uint32_t vcpuid)
+{
+        struct vcpu *vcpu = vcpu_find(vm, vcpuid);
+        int ret;
+
+        TEST_ASSERT(vcpu != NULL, "vcpu not found, vcpuid: %u", vcpuid);
+
+        vcpu->state->immediate_exit = 1;
+        ret = ioctl(vcpu->fd, KVM_RUN, NULL);
+        vcpu->state->immediate_exit = 0;
+
+        TEST_ASSERT(ret == -1 && errno == EINTR,
+                    "KVM_RUN IOCTL didn't exit immediately, rc: %i, errno: %i",
+                    ret, errno);
+}
+
 /*
  * VM VCPU Set MP State
  *