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 #define _GNU_SOURCE /* for program_invocation_name */
8 #include <fcntl.h>
9 #include <stdio.h>
10 #include <stdlib.h>
11 #include <string.h>
12 #include <sys/ioctl.h>
13
14 #include "test_util.h"
15 #include "kvm_util.h"
16 #include "processor.h"
17 #include "apic.h"
18
19 #define N_VCPU 2
20 #define VCPU_ID0 0
21 #define VCPU_ID1 1
22
guest_bsp_vcpu(void * arg)23 static void guest_bsp_vcpu(void *arg)
24 {
25 GUEST_SYNC(1);
26
27 GUEST_ASSERT(get_bsp_flag() != 0);
28
29 GUEST_DONE();
30 }
31
guest_not_bsp_vcpu(void * arg)32 static void guest_not_bsp_vcpu(void *arg)
33 {
34 GUEST_SYNC(1);
35
36 GUEST_ASSERT(get_bsp_flag() == 0);
37
38 GUEST_DONE();
39 }
40
test_set_boot_busy(struct kvm_vm * vm)41 static void test_set_boot_busy(struct kvm_vm *vm)
42 {
43 int res;
44
45 res = _vm_ioctl(vm, KVM_SET_BOOT_CPU_ID, (void *) VCPU_ID0);
46 TEST_ASSERT(res == -1 && errno == EBUSY,
47 "KVM_SET_BOOT_CPU_ID set while running vm");
48 }
49
run_vcpu(struct kvm_vm * vm,uint32_t vcpuid)50 static void run_vcpu(struct kvm_vm *vm, uint32_t vcpuid)
51 {
52 struct ucall uc;
53 int stage;
54
55 for (stage = 0; stage < 2; stage++) {
56
57 vcpu_run(vm, vcpuid);
58
59 switch (get_ucall(vm, vcpuid, &uc)) {
60 case UCALL_SYNC:
61 TEST_ASSERT(!strcmp((const char *)uc.args[0], "hello") &&
62 uc.args[1] == stage + 1,
63 "Stage %d: Unexpected register values vmexit, got %lx",
64 stage + 1, (ulong)uc.args[1]);
65 test_set_boot_busy(vm);
66 break;
67 case UCALL_DONE:
68 TEST_ASSERT(stage == 1,
69 "Expected GUEST_DONE in stage 2, got stage %d",
70 stage);
71 break;
72 case UCALL_ABORT:
73 TEST_ASSERT(false, "%s at %s:%ld\n\tvalues: %#lx, %#lx",
74 (const char *)uc.args[0], __FILE__,
75 uc.args[1], uc.args[2], uc.args[3]);
76 default:
77 TEST_ASSERT(false, "Unexpected exit: %s",
78 exit_reason_str(vcpu_state(vm, vcpuid)->exit_reason));
79 }
80 }
81 }
82
create_vm(void)83 static struct kvm_vm *create_vm(void)
84 {
85 struct kvm_vm *vm;
86 uint64_t vcpu_pages = (DEFAULT_STACK_PGS) * 2;
87 uint64_t extra_pg_pages = vcpu_pages / PTES_PER_MIN_PAGE * N_VCPU;
88 uint64_t pages = DEFAULT_GUEST_PHY_PAGES + vcpu_pages + extra_pg_pages;
89
90 pages = vm_adjust_num_guest_pages(VM_MODE_DEFAULT, pages);
91 vm = vm_create(VM_MODE_DEFAULT, pages, O_RDWR);
92
93 kvm_vm_elf_load(vm, program_invocation_name);
94 vm_create_irqchip(vm);
95
96 return vm;
97 }
98
add_x86_vcpu(struct kvm_vm * vm,uint32_t vcpuid,bool bsp_code)99 static void add_x86_vcpu(struct kvm_vm *vm, uint32_t vcpuid, bool bsp_code)
100 {
101 if (bsp_code)
102 vm_vcpu_add_default(vm, vcpuid, guest_bsp_vcpu);
103 else
104 vm_vcpu_add_default(vm, vcpuid, guest_not_bsp_vcpu);
105 }
106
run_vm_bsp(uint32_t bsp_vcpu)107 static void run_vm_bsp(uint32_t bsp_vcpu)
108 {
109 struct kvm_vm *vm;
110 bool is_bsp_vcpu1 = bsp_vcpu == VCPU_ID1;
111
112 vm = create_vm();
113
114 if (is_bsp_vcpu1)
115 vm_ioctl(vm, KVM_SET_BOOT_CPU_ID, (void *) VCPU_ID1);
116
117 add_x86_vcpu(vm, VCPU_ID0, !is_bsp_vcpu1);
118 add_x86_vcpu(vm, VCPU_ID1, is_bsp_vcpu1);
119
120 run_vcpu(vm, VCPU_ID0);
121 run_vcpu(vm, VCPU_ID1);
122
123 kvm_vm_free(vm);
124 }
125
check_set_bsp_busy(void)126 static void check_set_bsp_busy(void)
127 {
128 struct kvm_vm *vm;
129 int res;
130
131 vm = create_vm();
132
133 add_x86_vcpu(vm, VCPU_ID0, true);
134 add_x86_vcpu(vm, VCPU_ID1, false);
135
136 res = _vm_ioctl(vm, KVM_SET_BOOT_CPU_ID, (void *) VCPU_ID1);
137 TEST_ASSERT(res == -1 && errno == EBUSY, "KVM_SET_BOOT_CPU_ID set after adding vcpu");
138
139 run_vcpu(vm, VCPU_ID0);
140 run_vcpu(vm, VCPU_ID1);
141
142 res = _vm_ioctl(vm, KVM_SET_BOOT_CPU_ID, (void *) VCPU_ID1);
143 TEST_ASSERT(res == -1 && errno == EBUSY, "KVM_SET_BOOT_CPU_ID set to a terminated vcpu");
144
145 kvm_vm_free(vm);
146 }
147
main(int argc,char * argv[])148 int main(int argc, char *argv[])
149 {
150 if (!kvm_check_cap(KVM_CAP_SET_BOOT_CPU_ID)) {
151 print_skip("set_boot_cpu_id not available");
152 return 0;
153 }
154
155 run_vm_bsp(VCPU_ID0);
156 run_vm_bsp(VCPU_ID1);
157 run_vm_bsp(VCPU_ID0);
158
159 check_set_bsp_busy();
160 }
161