// SPDX-License-Identifier: GPL-2.0-only
/*
 * vcpu_width_config - Test KVM_ARM_VCPU_INIT() with KVM_ARM_VCPU_EL1_32BIT.
 *
 * Copyright (c) 2022 Google LLC.
 *
 * This is a test that ensures that non-mixed-width vCPUs (all 64bit vCPUs
 * or all 32bit vcPUs) can be configured and mixed-width vCPUs cannot be
 * configured.
 */

#include "kvm_util.h"
#include "processor.h"
#include "test_util.h"


/*
 * Add a vCPU, run KVM_ARM_VCPU_INIT with @init0, and then
 * add another vCPU, and run KVM_ARM_VCPU_INIT with @init1.
 */
static int add_init_2vcpus(struct kvm_vcpu_init *init0,
      struct kvm_vcpu_init *init1)
{
 struct kvm_vcpu *vcpu0, *vcpu1;
 struct kvm_vm *vm;
 int ret;

 vm = vm_create_barebones();

 vcpu0 = __vm_vcpu_add(vm, 0);
 ret = __vcpu_ioctl(vcpu0, KVM_ARM_VCPU_INIT, init0);
 if (ret)
  goto free_exit;

 vcpu1 = __vm_vcpu_add(vm, 1);
 ret = __vcpu_ioctl(vcpu1, KVM_ARM_VCPU_INIT, init1);

free_exit:
 kvm_vm_free(vm);
 return ret;
}

/*
 * Add two vCPUs, then run KVM_ARM_VCPU_INIT for one vCPU with @init0,
 * and run KVM_ARM_VCPU_INIT for another vCPU with @init1.
 */
static int add_2vcpus_init_2vcpus(struct kvm_vcpu_init *init0,
      struct kvm_vcpu_init *init1)
{
 struct kvm_vcpu *vcpu0, *vcpu1;
 struct kvm_vm *vm;
 int ret;

 vm = vm_create_barebones();

 vcpu0 = __vm_vcpu_add(vm, 0);
 vcpu1 = __vm_vcpu_add(vm, 1);

 ret = __vcpu_ioctl(vcpu0, KVM_ARM_VCPU_INIT, init0);
 if (ret)
  goto free_exit;

 ret = __vcpu_ioctl(vcpu1, KVM_ARM_VCPU_INIT, init1);

free_exit:
 kvm_vm_free(vm);
 return ret;
}

/*
 * Tests that two 64bit vCPUs can be configured, two 32bit vCPUs can be
 * configured, and two mixed-width vCPUs cannot be configured.
 * Each of those three cases, configure vCPUs in two different orders.
 * The one is running KVM_CREATE_VCPU for 2 vCPUs, and then running
 * KVM_ARM_VCPU_INIT for them.
 * The other is running KVM_CREATE_VCPU and KVM_ARM_VCPU_INIT for a vCPU,
 * and then run those commands for another vCPU.
 */
int main(void)
{
 struct kvm_vcpu_init init0, init1;
 struct kvm_vm *vm;
 int ret;

 TEST_REQUIRE(kvm_has_cap(KVM_CAP_ARM_EL1_32BIT));

 /* Get the preferred target type and copy that to init1 for later use */
 vm = vm_create_barebones();
 vm_ioctl(vm, KVM_ARM_PREFERRED_TARGET, &init0);
 kvm_vm_free(vm);
 init1 = init0;

 /* Test with 64bit vCPUs */
 ret = add_init_2vcpus(&init0, &init0);
 TEST_ASSERT(ret == 0,
      "Configuring 64bit EL1 vCPUs failed unexpectedly");
 ret = add_2vcpus_init_2vcpus(&init0, &init0);
 TEST_ASSERT(ret == 0,
      "Configuring 64bit EL1 vCPUs failed unexpectedly");

 /* Test with 32bit vCPUs */
 init0.features[0] = (1 << KVM_ARM_VCPU_EL1_32BIT);
 ret = add_init_2vcpus(&init0, &init0);
 TEST_ASSERT(ret == 0,
      "Configuring 32bit EL1 vCPUs failed unexpectedly");
 ret = add_2vcpus_init_2vcpus(&init0, &init0);
 TEST_ASSERT(ret == 0,
      "Configuring 32bit EL1 vCPUs failed unexpectedly");

 /* Test with mixed-width vCPUs  */
 init0.features[0] = 0;
 init1.features[0] = (1 << KVM_ARM_VCPU_EL1_32BIT);
 ret = add_init_2vcpus(&init0, &init1);
 TEST_ASSERT(ret != 0,
      "Configuring mixed-width vCPUs worked unexpectedly");
 ret = add_2vcpus_init_2vcpus(&init0, &init1);
 TEST_ASSERT(ret != 0,
      "Configuring mixed-width vCPUs worked unexpectedly");

 return 0;
}