Setting a vCPU's MP state to KVM_MP_STATE_STOPPED has the effect of
powering off the vCPU. Rather than using the vCPU init feature flag, use
the KVM_SET_MP_STATE ioctl to power off the target vCPU.

Signed-off-by: Oliver Upton <oup...@google.com>
---
 tools/testing/selftests/kvm/aarch64/psci_test.c | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/tools/testing/selftests/kvm/aarch64/psci_test.c 
b/tools/testing/selftests/kvm/aarch64/psci_test.c
index cebea7356e5a..8d043e12b137 100644
--- a/tools/testing/selftests/kvm/aarch64/psci_test.c
+++ b/tools/testing/selftests/kvm/aarch64/psci_test.c
@@ -60,6 +60,15 @@ static void guest_main(uint64_t target_cpu)
        GUEST_DONE();
 }
 
+static void vcpu_power_off(struct kvm_vm *vm, uint32_t vcpuid)
+{
+       struct kvm_mp_state mp_state = {
+               .mp_state = KVM_MP_STATE_STOPPED,
+       };
+
+       vcpu_set_mp_state(vm, vcpuid, &mp_state);
+}
+
 int main(void)
 {
        uint64_t target_mpidr, obs_pc, obs_x0;
@@ -75,12 +84,12 @@ int main(void)
        init.features[0] |= (1 << KVM_ARM_VCPU_PSCI_0_2);
 
        aarch64_vcpu_add_default(vm, VCPU_ID_SOURCE, &init, guest_main);
+       aarch64_vcpu_add_default(vm, VCPU_ID_TARGET, &init, guest_main);
 
        /*
         * make sure the target is already off when executing the test.
         */
-       init.features[0] |= (1 << KVM_ARM_VCPU_POWER_OFF);
-       aarch64_vcpu_add_default(vm, VCPU_ID_TARGET, &init, guest_main);
+       vcpu_power_off(vm, VCPU_ID_TARGET);
 
        get_reg(vm, VCPU_ID_TARGET, ARM64_SYS_REG(MPIDR_EL1), &target_mpidr);
        vcpu_args_set(vm, VCPU_ID_SOURCE, 1, target_mpidr & MPIDR_HWID_BITMASK);
-- 
2.33.0.685.g46640cef36-goog

_______________________________________________
kvmarm mailing list
kvmarm@lists.cs.columbia.edu
https://lists.cs.columbia.edu/mailman/listinfo/kvmarm

Reply via email to