183b49c54SPatrick Mooney /*
283b49c54SPatrick Mooney  * This file and its contents are supplied under the terms of the
383b49c54SPatrick Mooney  * Common Development and Distribution License ("CDDL"), version 1.0.
483b49c54SPatrick Mooney  * You may only use this file in accordance with the terms of version
583b49c54SPatrick Mooney  * 1.0 of the CDDL.
683b49c54SPatrick Mooney  *
783b49c54SPatrick Mooney  * A full copy of the text of the CDDL should have accompanied this
883b49c54SPatrick Mooney  * source.  A copy of the CDDL is also available via the Internet at
983b49c54SPatrick Mooney  * http://www.illumos.org/license/CDDL.
1083b49c54SPatrick Mooney  */
1183b49c54SPatrick Mooney 
1283b49c54SPatrick Mooney /*
1383b49c54SPatrick Mooney  * Copyright 2022 Oxide Computer Company
1483b49c54SPatrick Mooney  */
1583b49c54SPatrick Mooney 
1683b49c54SPatrick Mooney #include <stdio.h>
1783b49c54SPatrick Mooney #include <unistd.h>
1883b49c54SPatrick Mooney #include <stdlib.h>
1983b49c54SPatrick Mooney #include <strings.h>
2083b49c54SPatrick Mooney #include <libgen.h>
2183b49c54SPatrick Mooney #include <assert.h>
22*32640292SAndy Fiddaman #include <errno.h>
2383b49c54SPatrick Mooney 
2483b49c54SPatrick Mooney #include <sys/types.h>
2583b49c54SPatrick Mooney #include <sys/sysmacros.h>
2683b49c54SPatrick Mooney #include <sys/debug.h>
2783b49c54SPatrick Mooney #include <sys/vmm.h>
2883b49c54SPatrick Mooney #include <sys/vmm_dev.h>
2983b49c54SPatrick Mooney #include <vmmapi.h>
3083b49c54SPatrick Mooney 
3183b49c54SPatrick Mooney #include "in_guest.h"
3283b49c54SPatrick Mooney 
3383b49c54SPatrick Mooney int
main(int argc,char * argv[])3483b49c54SPatrick Mooney main(int argc, char *argv[])
3583b49c54SPatrick Mooney {
3683b49c54SPatrick Mooney 	const char *test_suite_name = basename(argv[0]);
3783b49c54SPatrick Mooney 	struct vmctx *ctx = NULL;
38*32640292SAndy Fiddaman 	struct vcpu *vcpu;
3983b49c54SPatrick Mooney 	int err;
4083b49c54SPatrick Mooney 
4183b49c54SPatrick Mooney 	ctx = test_initialize(test_suite_name);
4283b49c54SPatrick Mooney 
43*32640292SAndy Fiddaman 	if ((vcpu = vm_vcpu_open(ctx, 0)) == NULL) {
44*32640292SAndy Fiddaman 		test_fail_errno(errno, "Could not open vcpu0");
45*32640292SAndy Fiddaman 	}
46*32640292SAndy Fiddaman 
47*32640292SAndy Fiddaman 	err = test_setup_vcpu(vcpu, MEM_LOC_PAYLOAD, MEM_LOC_STACK);
4883b49c54SPatrick Mooney 	if (err != 0) {
4983b49c54SPatrick Mooney 		test_fail_errno(err, "Could not initialize vcpu0");
5083b49c54SPatrick Mooney 	}
5183b49c54SPatrick Mooney 
5283b49c54SPatrick Mooney 	struct vm_entry ventry = { 0 };
5383b49c54SPatrick Mooney 	struct vm_exit vexit = { 0 };
5483b49c54SPatrick Mooney 	const enum vm_suspend_how expected_how = VM_SUSPEND_TRIPLEFAULT;
5583b49c54SPatrick Mooney 
5683b49c54SPatrick Mooney 	do {
5783b49c54SPatrick Mooney 		const enum vm_exit_kind kind =
58*32640292SAndy Fiddaman 		    test_run_vcpu(vcpu, &ventry, &vexit);
5983b49c54SPatrick Mooney 		switch (kind) {
6083b49c54SPatrick Mooney 		case VEK_REENTR:
6183b49c54SPatrick Mooney 			break;
6283b49c54SPatrick Mooney 		case VEK_UNHANDLED:
6383b49c54SPatrick Mooney 			/* We expect to immediately triple-fault */
6483b49c54SPatrick Mooney 			if (vexit.exitcode != VM_EXITCODE_SUSPENDED) {
6583b49c54SPatrick Mooney 				test_fail_vmexit(&vexit);
6683b49c54SPatrick Mooney 			}
6783b49c54SPatrick Mooney 			if (vexit.u.suspended.how != expected_how) {
6883b49c54SPatrick Mooney 				test_fail_msg("suspend_how %d != %d\n",
6983b49c54SPatrick Mooney 				    vexit.u.suspended.how, expected_how);
7083b49c54SPatrick Mooney 			}
7183b49c54SPatrick Mooney 			test_pass();
7283b49c54SPatrick Mooney 			break;
7383b49c54SPatrick Mooney 
7483b49c54SPatrick Mooney 		case VEK_TEST_PASS:
7583b49c54SPatrick Mooney 		case VEK_TEST_FAIL:
7683b49c54SPatrick Mooney 		default:
7783b49c54SPatrick Mooney 			test_fail_vmexit(&vexit);
7883b49c54SPatrick Mooney 			break;
7983b49c54SPatrick Mooney 		}
8083b49c54SPatrick Mooney 	} while (true);
8183b49c54SPatrick Mooney }
82