122 lines · c
1// SPDX-License-Identifier: GPL-2.0-only2/*3 * vcpu_width_config - Test KVM_ARM_VCPU_INIT() with KVM_ARM_VCPU_EL1_32BIT.4 *5 * Copyright (c) 2022 Google LLC.6 *7 * This is a test that ensures that non-mixed-width vCPUs (all 64bit vCPUs8 * or all 32bit vcPUs) can be configured and mixed-width vCPUs cannot be9 * configured.10 */11 12#include "kvm_util.h"13#include "processor.h"14#include "test_util.h"15 16 17/*18 * Add a vCPU, run KVM_ARM_VCPU_INIT with @init0, and then19 * add another vCPU, and run KVM_ARM_VCPU_INIT with @init1.20 */21static int add_init_2vcpus(struct kvm_vcpu_init *init0,22 struct kvm_vcpu_init *init1)23{24 struct kvm_vcpu *vcpu0, *vcpu1;25 struct kvm_vm *vm;26 int ret;27 28 vm = vm_create_barebones();29 30 vcpu0 = __vm_vcpu_add(vm, 0);31 ret = __vcpu_ioctl(vcpu0, KVM_ARM_VCPU_INIT, init0);32 if (ret)33 goto free_exit;34 35 vcpu1 = __vm_vcpu_add(vm, 1);36 ret = __vcpu_ioctl(vcpu1, KVM_ARM_VCPU_INIT, init1);37 38free_exit:39 kvm_vm_free(vm);40 return ret;41}42 43/*44 * Add two vCPUs, then run KVM_ARM_VCPU_INIT for one vCPU with @init0,45 * and run KVM_ARM_VCPU_INIT for another vCPU with @init1.46 */47static int add_2vcpus_init_2vcpus(struct kvm_vcpu_init *init0,48 struct kvm_vcpu_init *init1)49{50 struct kvm_vcpu *vcpu0, *vcpu1;51 struct kvm_vm *vm;52 int ret;53 54 vm = vm_create_barebones();55 56 vcpu0 = __vm_vcpu_add(vm, 0);57 vcpu1 = __vm_vcpu_add(vm, 1);58 59 ret = __vcpu_ioctl(vcpu0, KVM_ARM_VCPU_INIT, init0);60 if (ret)61 goto free_exit;62 63 ret = __vcpu_ioctl(vcpu1, KVM_ARM_VCPU_INIT, init1);64 65free_exit:66 kvm_vm_free(vm);67 return ret;68}69 70/*71 * Tests that two 64bit vCPUs can be configured, two 32bit vCPUs can be72 * configured, and two mixed-width vCPUs cannot be configured.73 * Each of those three cases, configure vCPUs in two different orders.74 * The one is running KVM_CREATE_VCPU for 2 vCPUs, and then running75 * KVM_ARM_VCPU_INIT for them.76 * The other is running KVM_CREATE_VCPU and KVM_ARM_VCPU_INIT for a vCPU,77 * and then run those commands for another vCPU.78 */79int main(void)80{81 struct kvm_vcpu_init init0, init1;82 struct kvm_vm *vm;83 int ret;84 85 TEST_REQUIRE(kvm_has_cap(KVM_CAP_ARM_EL1_32BIT));86 87 /* Get the preferred target type and copy that to init1 for later use */88 vm = vm_create_barebones();89 vm_ioctl(vm, KVM_ARM_PREFERRED_TARGET, &init0);90 kvm_vm_free(vm);91 init1 = init0;92 93 /* Test with 64bit vCPUs */94 ret = add_init_2vcpus(&init0, &init0);95 TEST_ASSERT(ret == 0,96 "Configuring 64bit EL1 vCPUs failed unexpectedly");97 ret = add_2vcpus_init_2vcpus(&init0, &init0);98 TEST_ASSERT(ret == 0,99 "Configuring 64bit EL1 vCPUs failed unexpectedly");100 101 /* Test with 32bit vCPUs */102 init0.features[0] = (1 << KVM_ARM_VCPU_EL1_32BIT);103 ret = add_init_2vcpus(&init0, &init0);104 TEST_ASSERT(ret == 0,105 "Configuring 32bit EL1 vCPUs failed unexpectedly");106 ret = add_2vcpus_init_2vcpus(&init0, &init0);107 TEST_ASSERT(ret == 0,108 "Configuring 32bit EL1 vCPUs failed unexpectedly");109 110 /* Test with mixed-width vCPUs */111 init0.features[0] = 0;112 init1.features[0] = (1 << KVM_ARM_VCPU_EL1_32BIT);113 ret = add_init_2vcpus(&init0, &init1);114 TEST_ASSERT(ret != 0,115 "Configuring mixed-width vCPUs worked unexpectedly");116 ret = add_2vcpus_init_2vcpus(&init0, &init1);117 TEST_ASSERT(ret != 0,118 "Configuring mixed-width vCPUs worked unexpectedly");119 120 return 0;121}122