1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * Test that KVM_SET_BOOT_CPU_ID works as intended
4 *
5 * Copyright (C) 2020, Red Hat, Inc.
6 */
7 #include <fcntl.h>
8 #include <stdio.h>
9 #include <stdlib.h>
10 #include <string.h>
11 #include <sys/ioctl.h>
12
13 #include "test_util.h"
14 #include "kvm_util.h"
15 #include "processor.h"
16 #include "apic.h"
17
guest_bsp_vcpu(void * arg)18 static void guest_bsp_vcpu(void *arg)
19 {
20 GUEST_SYNC(1);
21
22 GUEST_ASSERT_NE(get_bsp_flag(), 0);
23
24 GUEST_DONE();
25 }
26
guest_not_bsp_vcpu(void * arg)27 static void guest_not_bsp_vcpu(void *arg)
28 {
29 GUEST_SYNC(1);
30
31 GUEST_ASSERT_EQ(get_bsp_flag(), 0);
32
33 GUEST_DONE();
34 }
35
test_set_invalid_bsp(struct kvm_vm * vm)36 static void test_set_invalid_bsp(struct kvm_vm *vm)
37 {
38 unsigned long max_vcpu_id = vm_check_cap(vm, KVM_CAP_MAX_VCPU_ID);
39 int r;
40
41 if (max_vcpu_id) {
42 r = __vm_ioctl(vm, KVM_SET_BOOT_CPU_ID, (void *)(max_vcpu_id + 1));
43 TEST_ASSERT(r == -1 && errno == EINVAL, "BSP with ID > MAX should fail");
44 }
45
46 r = __vm_ioctl(vm, KVM_SET_BOOT_CPU_ID, (void *)(1L << 32));
47 TEST_ASSERT(r == -1 && errno == EINVAL, "BSP with ID[63:32]!=0 should fail");
48 }
49
test_set_bsp_busy(struct kvm_vcpu * vcpu,const char * msg)50 static void test_set_bsp_busy(struct kvm_vcpu *vcpu, const char *msg)
51 {
52 int r = __vm_ioctl(vcpu->vm, KVM_SET_BOOT_CPU_ID,
53 (void *)(unsigned long)vcpu->id);
54
55 TEST_ASSERT(r == -1 && errno == EBUSY, "KVM_SET_BOOT_CPU_ID set %s", msg);
56 }
57
run_vcpu(struct kvm_vcpu * vcpu)58 static void run_vcpu(struct kvm_vcpu *vcpu)
59 {
60 struct ucall uc;
61 int stage;
62
63 for (stage = 0; stage < 2; stage++) {
64
65 vcpu_run(vcpu);
66
67 switch (get_ucall(vcpu, &uc)) {
68 case UCALL_SYNC:
69 TEST_ASSERT(!strcmp((const char *)uc.args[0], "hello") &&
70 uc.args[1] == stage + 1,
71 "Stage %d: Unexpected register values vmexit, got %lx",
72 stage + 1, (ulong)uc.args[1]);
73 test_set_bsp_busy(vcpu, "while running vm");
74 break;
75 case UCALL_DONE:
76 TEST_ASSERT(stage == 1,
77 "Expected GUEST_DONE in stage 2, got stage %d",
78 stage);
79 break;
80 case UCALL_ABORT:
81 REPORT_GUEST_ASSERT(uc);
82 default:
83 TEST_ASSERT(false, "Unexpected exit: %s",
84 exit_reason_str(vcpu->run->exit_reason));
85 }
86 }
87 }
88
create_vm(uint32_t nr_vcpus,uint32_t bsp_vcpu_id,struct kvm_vcpu * vcpus[])89 static struct kvm_vm *create_vm(uint32_t nr_vcpus, uint32_t bsp_vcpu_id,
90 struct kvm_vcpu *vcpus[])
91 {
92 struct kvm_vm *vm;
93 uint32_t i;
94
95 vm = vm_create(nr_vcpus);
96
97 test_set_invalid_bsp(vm);
98
99 vm_ioctl(vm, KVM_SET_BOOT_CPU_ID, (void *)(unsigned long)bsp_vcpu_id);
100
101 for (i = 0; i < nr_vcpus; i++)
102 vcpus[i] = vm_vcpu_add(vm, i, i == bsp_vcpu_id ? guest_bsp_vcpu :
103 guest_not_bsp_vcpu);
104 return vm;
105 }
106
run_vm_bsp(uint32_t bsp_vcpu_id)107 static void run_vm_bsp(uint32_t bsp_vcpu_id)
108 {
109 struct kvm_vcpu *vcpus[2];
110 struct kvm_vm *vm;
111
112 vm = create_vm(ARRAY_SIZE(vcpus), bsp_vcpu_id, vcpus);
113
114 run_vcpu(vcpus[0]);
115 run_vcpu(vcpus[1]);
116
117 kvm_vm_free(vm);
118 }
119
check_set_bsp_busy(void)120 static void check_set_bsp_busy(void)
121 {
122 struct kvm_vcpu *vcpus[2];
123 struct kvm_vm *vm;
124
125 vm = create_vm(ARRAY_SIZE(vcpus), 0, vcpus);
126
127 test_set_bsp_busy(vcpus[1], "after adding vcpu");
128
129 run_vcpu(vcpus[0]);
130 run_vcpu(vcpus[1]);
131
132 test_set_bsp_busy(vcpus[1], "to a terminated vcpu");
133
134 kvm_vm_free(vm);
135 }
136
main(int argc,char * argv[])137 int main(int argc, char *argv[])
138 {
139 TEST_REQUIRE(kvm_has_cap(KVM_CAP_SET_BOOT_CPU_ID));
140
141 run_vm_bsp(0);
142 run_vm_bsp(1);
143 run_vm_bsp(0);
144
145 check_set_bsp_busy();
146 }
147