1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * vmx_close_while_nested
4  *
5  * Copyright (C) 2019, Red Hat, Inc.
6  *
7  * Verify that nothing bad happens if a KVM user exits with open
8  * file descriptors while executing a nested guest.
9  */
10 
11 #include "test_util.h"
12 #include "kvm_util.h"
13 #include "processor.h"
14 #include "vmx.h"
15 
16 #include <string.h>
17 #include <sys/ioctl.h>
18 
19 #include "kselftest.h"
20 
21 #define VCPU_ID		5
22 
23 enum {
24 	PORT_L0_EXIT = 0x2000,
25 };
26 
27 /* The virtual machine object. */
28 static struct kvm_vm *vm;
29 
l2_guest_code(void)30 static void l2_guest_code(void)
31 {
32 	/* Exit to L0 */
33         asm volatile("inb %%dx, %%al"
34                      : : [port] "d" (PORT_L0_EXIT) : "rax");
35 }
36 
l1_guest_code(struct vmx_pages * vmx_pages)37 static void l1_guest_code(struct vmx_pages *vmx_pages)
38 {
39 #define L2_GUEST_STACK_SIZE 64
40 	unsigned long l2_guest_stack[L2_GUEST_STACK_SIZE];
41 
42 	GUEST_ASSERT(prepare_for_vmx_operation(vmx_pages));
43 	GUEST_ASSERT(load_vmcs(vmx_pages));
44 
45 	/* Prepare the VMCS for L2 execution. */
46 	prepare_vmcs(vmx_pages, l2_guest_code,
47 		     &l2_guest_stack[L2_GUEST_STACK_SIZE]);
48 
49 	GUEST_ASSERT(!vmlaunch());
50 	GUEST_ASSERT(0);
51 }
52 
main(int argc,char * argv[])53 int main(int argc, char *argv[])
54 {
55 	vm_vaddr_t vmx_pages_gva;
56 
57 	nested_vmx_check_supported();
58 
59 	vm = vm_create_default(VCPU_ID, 0, (void *) l1_guest_code);
60 	vcpu_set_cpuid(vm, VCPU_ID, kvm_get_supported_cpuid());
61 
62 	/* Allocate VMX pages and shared descriptors (vmx_pages). */
63 	vcpu_alloc_vmx(vm, &vmx_pages_gva);
64 	vcpu_args_set(vm, VCPU_ID, 1, vmx_pages_gva);
65 
66 	for (;;) {
67 		volatile struct kvm_run *run = vcpu_state(vm, VCPU_ID);
68 		struct ucall uc;
69 
70 		vcpu_run(vm, VCPU_ID);
71 		TEST_ASSERT(run->exit_reason == KVM_EXIT_IO,
72 			    "Got exit_reason other than KVM_EXIT_IO: %u (%s)\n",
73 			    run->exit_reason,
74 			    exit_reason_str(run->exit_reason));
75 
76 		if (run->io.port == PORT_L0_EXIT)
77 			break;
78 
79 		switch (get_ucall(vm, VCPU_ID, &uc)) {
80 		case UCALL_ABORT:
81 			TEST_ASSERT(false, "%s", (const char *)uc.args[0]);
82 			/* NOT REACHED */
83 		default:
84 			TEST_ASSERT(false, "Unknown ucall 0x%x.", uc.cmd);
85 		}
86 	}
87 }
88