Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
kernel
os
linux
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
18static 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
27static 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
36static 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
50static 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
58static 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
89static struct kvm_vm *create_vm(u32 nr_vcpus, u32 bsp_vcpu_id,
90 struct kvm_vcpu *vcpus[])
91{
92 struct kvm_vm *vm;
93 u32 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
107static void run_vm_bsp(u32 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
120static 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
137int 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}