1 // SPDX-License-Identifier: GPL-2.0-only
2 #include "test_util.h"
3 #include "kvm_util.h"
4 #include "processor.h"
5 #include "vmx.h"
6 
7 #include <string.h>
8 #include <sys/ioctl.h>
9 
10 #include "kselftest.h"
11 
12 #define ARBITRARY_IO_PORT	0x2000
13 
14 /* The virtual machine object. */
15 static struct kvm_vm *vm;
16 
17 static void l2_guest_code(void)
18 {
19 	asm volatile("inb %%dx, %%al"
20 		     : : [port] "d" (ARBITRARY_IO_PORT) : "rax");
21 }
22 
23 void l1_guest_code(struct vmx_pages *vmx)
24 {
25 #define L2_GUEST_STACK_SIZE 64
26 	unsigned long l2_guest_stack[L2_GUEST_STACK_SIZE];
27 
28 	GUEST_ASSERT(vmx->vmcs_gpa);
29 	GUEST_ASSERT(prepare_for_vmx_operation(vmx));
30 	GUEST_ASSERT(load_vmcs(vmx));
31 
32 	prepare_vmcs(vmx, l2_guest_code,
33 		     &l2_guest_stack[L2_GUEST_STACK_SIZE]);
34 
35 	GUEST_ASSERT(!vmlaunch());
36 	/* L2 should triple fault after a triple fault event injected. */
37 	GUEST_ASSERT(vmreadz(VM_EXIT_REASON) == EXIT_REASON_TRIPLE_FAULT);
38 	GUEST_DONE();
39 }
40 
41 int main(void)
42 {
43 	struct kvm_vcpu *vcpu;
44 	struct kvm_run *run;
45 	struct kvm_vcpu_events events;
46 	vm_vaddr_t vmx_pages_gva;
47 	struct ucall uc;
48 
49 	TEST_REQUIRE(kvm_cpu_has(X86_FEATURE_VMX));
50 
51 	TEST_REQUIRE(kvm_has_cap(KVM_CAP_X86_TRIPLE_FAULT_EVENT));
52 
53 	vm = vm_create_with_one_vcpu(&vcpu, l1_guest_code);
54 	vm_enable_cap(vm, KVM_CAP_X86_TRIPLE_FAULT_EVENT, 1);
55 
56 	run = vcpu->run;
57 	vcpu_alloc_vmx(vm, &vmx_pages_gva);
58 	vcpu_args_set(vcpu, 1, vmx_pages_gva);
59 	vcpu_run(vcpu);
60 
61 	TEST_ASSERT(run->exit_reason == KVM_EXIT_IO,
62 		    "Expected KVM_EXIT_IO, got: %u (%s)\n",
63 		    run->exit_reason, exit_reason_str(run->exit_reason));
64 	TEST_ASSERT(run->io.port == ARBITRARY_IO_PORT,
65 		    "Expected IN from port %d from L2, got port %d",
66 		    ARBITRARY_IO_PORT, run->io.port);
67 	vcpu_events_get(vcpu, &events);
68 	events.flags |= KVM_VCPUEVENT_VALID_TRIPLE_FAULT;
69 	events.triple_fault.pending = true;
70 	vcpu_events_set(vcpu, &events);
71 	run->immediate_exit = true;
72 	vcpu_run_complete_io(vcpu);
73 
74 	vcpu_events_get(vcpu, &events);
75 	TEST_ASSERT(events.flags & KVM_VCPUEVENT_VALID_TRIPLE_FAULT,
76 		    "Triple fault event invalid");
77 	TEST_ASSERT(events.triple_fault.pending,
78 		    "No triple fault pending");
79 	vcpu_run(vcpu);
80 
81 	switch (get_ucall(vcpu, &uc)) {
82 	case UCALL_DONE:
83 		break;
84 	case UCALL_ABORT:
85 		REPORT_GUEST_ASSERT(uc);
86 	default:
87 		TEST_FAIL("Unexpected ucall: %lu", uc.cmd);
88 	}
89 
90 }
91