160325261SVipin Sharma // SPDX-License-Identifier: GPL-2.0-only
260325261SVipin Sharma /*
360325261SVipin Sharma * Test Hyper-V extended hypercall, HV_EXT_CALL_QUERY_CAPABILITIES (0x8001),
460325261SVipin Sharma * exit to userspace and receive result in guest.
560325261SVipin Sharma *
660325261SVipin Sharma * Negative tests are present in hyperv_features.c
760325261SVipin Sharma *
860325261SVipin Sharma * Copyright 2022 Google LLC
960325261SVipin Sharma * Author: Vipin Sharma <vipinsh@google.com>
1060325261SVipin Sharma */
1160325261SVipin Sharma #include "kvm_util.h"
1260325261SVipin Sharma #include "processor.h"
1360325261SVipin Sharma #include "hyperv.h"
1460325261SVipin Sharma
1560325261SVipin Sharma /* Any value is fine */
1660325261SVipin Sharma #define EXT_CAPABILITIES 0xbull
1760325261SVipin Sharma
guest_code(vm_paddr_t in_pg_gpa,vm_paddr_t out_pg_gpa,vm_vaddr_t out_pg_gva)1860325261SVipin Sharma static void guest_code(vm_paddr_t in_pg_gpa, vm_paddr_t out_pg_gpa,
1960325261SVipin Sharma vm_vaddr_t out_pg_gva)
2060325261SVipin Sharma {
2160325261SVipin Sharma uint64_t *output_gva;
2260325261SVipin Sharma
2360325261SVipin Sharma wrmsr(HV_X64_MSR_GUEST_OS_ID, HYPERV_LINUX_OS_ID);
2460325261SVipin Sharma wrmsr(HV_X64_MSR_HYPERCALL, in_pg_gpa);
2560325261SVipin Sharma
2660325261SVipin Sharma output_gva = (uint64_t *)out_pg_gva;
2760325261SVipin Sharma
2860325261SVipin Sharma hyperv_hypercall(HV_EXT_CALL_QUERY_CAPABILITIES, in_pg_gpa, out_pg_gpa);
2960325261SVipin Sharma
3060325261SVipin Sharma /* TLFS states output will be a uint64_t value */
3160325261SVipin Sharma GUEST_ASSERT_EQ(*output_gva, EXT_CAPABILITIES);
3260325261SVipin Sharma
3360325261SVipin Sharma GUEST_DONE();
3460325261SVipin Sharma }
3560325261SVipin Sharma
main(void)3660325261SVipin Sharma int main(void)
3760325261SVipin Sharma {
3860325261SVipin Sharma vm_vaddr_t hcall_out_page;
3960325261SVipin Sharma vm_vaddr_t hcall_in_page;
4060325261SVipin Sharma struct kvm_vcpu *vcpu;
4160325261SVipin Sharma struct kvm_run *run;
4260325261SVipin Sharma struct kvm_vm *vm;
4360325261SVipin Sharma uint64_t *outval;
4460325261SVipin Sharma struct ucall uc;
4560325261SVipin Sharma
46*6dac1195SVitaly Kuznetsov TEST_REQUIRE(kvm_has_cap(KVM_CAP_HYPERV_CPUID));
47*6dac1195SVitaly Kuznetsov
4860325261SVipin Sharma /* Verify if extended hypercalls are supported */
4960325261SVipin Sharma if (!kvm_cpuid_has(kvm_get_supported_hv_cpuid(),
5060325261SVipin Sharma HV_ENABLE_EXTENDED_HYPERCALLS)) {
5160325261SVipin Sharma print_skip("Extended calls not supported by the kernel");
5260325261SVipin Sharma exit(KSFT_SKIP);
5360325261SVipin Sharma }
5460325261SVipin Sharma
5560325261SVipin Sharma vm = vm_create_with_one_vcpu(&vcpu, guest_code);
5660325261SVipin Sharma run = vcpu->run;
5760325261SVipin Sharma vcpu_set_hv_cpuid(vcpu);
5860325261SVipin Sharma
5960325261SVipin Sharma /* Hypercall input */
6060325261SVipin Sharma hcall_in_page = vm_vaddr_alloc_pages(vm, 1);
6160325261SVipin Sharma memset(addr_gva2hva(vm, hcall_in_page), 0x0, vm->page_size);
6260325261SVipin Sharma
6360325261SVipin Sharma /* Hypercall output */
6460325261SVipin Sharma hcall_out_page = vm_vaddr_alloc_pages(vm, 1);
6560325261SVipin Sharma memset(addr_gva2hva(vm, hcall_out_page), 0x0, vm->page_size);
6660325261SVipin Sharma
6760325261SVipin Sharma vcpu_args_set(vcpu, 3, addr_gva2gpa(vm, hcall_in_page),
6860325261SVipin Sharma addr_gva2gpa(vm, hcall_out_page), hcall_out_page);
6960325261SVipin Sharma
7060325261SVipin Sharma vcpu_run(vcpu);
7160325261SVipin Sharma
7260325261SVipin Sharma TEST_ASSERT(run->exit_reason == KVM_EXIT_HYPERV,
7360325261SVipin Sharma "Unexpected exit reason: %u (%s)",
7460325261SVipin Sharma run->exit_reason, exit_reason_str(run->exit_reason));
7560325261SVipin Sharma
7660325261SVipin Sharma outval = addr_gpa2hva(vm, run->hyperv.u.hcall.params[1]);
7760325261SVipin Sharma *outval = EXT_CAPABILITIES;
7860325261SVipin Sharma run->hyperv.u.hcall.result = HV_STATUS_SUCCESS;
7960325261SVipin Sharma
8060325261SVipin Sharma vcpu_run(vcpu);
8160325261SVipin Sharma
8260325261SVipin Sharma TEST_ASSERT(run->exit_reason == KVM_EXIT_IO,
8360325261SVipin Sharma "Unexpected exit reason: %u (%s)",
8460325261SVipin Sharma run->exit_reason, exit_reason_str(run->exit_reason));
8560325261SVipin Sharma
8660325261SVipin Sharma switch (get_ucall(vcpu, &uc)) {
8760325261SVipin Sharma case UCALL_ABORT:
8882cb0ed6SSean Christopherson REPORT_GUEST_ASSERT(uc);
8960325261SVipin Sharma break;
9060325261SVipin Sharma case UCALL_DONE:
9160325261SVipin Sharma break;
9260325261SVipin Sharma default:
9360325261SVipin Sharma TEST_FAIL("Unhandled ucall: %ld", uc.cmd);
9460325261SVipin Sharma }
9560325261SVipin Sharma
9660325261SVipin Sharma kvm_vm_free(vm);
9760325261SVipin Sharma return 0;
9860325261SVipin Sharma }
99