1 /*
2 * QEMU device plug/unplug handling
3 *
4 * Copyright (C) 2019 Red Hat Inc.
5 *
6 * Authors:
7 * David Hildenbrand <david@redhat.com>
8 *
9 * This work is licensed under the terms of the GNU GPL, version 2 or later.
10 * See the COPYING file in the top-level directory.
11 */
12
13 #include "qemu/osdep.h"
14 #include "libqtest.h"
15 #include "qapi/qmp/qdict.h"
16 #include "qapi/qmp/qstring.h"
17
system_reset(QTestState * qtest)18 static void system_reset(QTestState *qtest)
19 {
20 QDict *resp;
21
22 resp = qtest_qmp(qtest, "{'execute': 'system_reset'}");
23 g_assert(qdict_haskey(resp, "return"));
24 qobject_unref(resp);
25 }
26
wait_device_deleted_event(QTestState * qtest,const char * id)27 static void wait_device_deleted_event(QTestState *qtest, const char *id)
28 {
29 QDict *resp, *data;
30 QString *qstr;
31
32 /*
33 * Other devices might get removed along with the removed device. Skip
34 * these. The device of interest will be the last one.
35 */
36 for (;;) {
37 resp = qtest_qmp_eventwait_ref(qtest, "DEVICE_DELETED");
38 data = qdict_get_qdict(resp, "data");
39 if (!data || !qdict_get(data, "device")) {
40 qobject_unref(resp);
41 continue;
42 }
43 qstr = qobject_to(QString, qdict_get(data, "device"));
44 g_assert(qstr);
45 if (!strcmp(qstring_get_str(qstr), id)) {
46 qobject_unref(resp);
47 break;
48 }
49 qobject_unref(resp);
50 }
51 }
52
process_device_remove(QTestState * qtest,const char * id)53 static void process_device_remove(QTestState *qtest, const char *id)
54 {
55 /*
56 * Request device removal. As the guest is not running, the request won't
57 * be processed. However during system reset, the removal will be
58 * handled, removing the device.
59 */
60 qtest_qmp_device_del_send(qtest, id);
61 system_reset(qtest);
62 wait_device_deleted_event(qtest, id);
63 }
64
test_pci_unplug_request(void)65 static void test_pci_unplug_request(void)
66 {
67 QTestState *qtest;
68 const char *arch = qtest_get_arch();
69 const char *machine_addition = "";
70
71 if (!qtest_has_device("virtio-mouse-pci")) {
72 g_test_skip("Device virtio-mouse-pci not available");
73 return;
74 }
75
76 if (strcmp(arch, "i386") == 0 || strcmp(arch, "x86_64") == 0) {
77 machine_addition = "-machine pc";
78 }
79
80 qtest = qtest_initf("%s -device virtio-mouse-pci,id=dev0",
81 machine_addition);
82
83 process_device_remove(qtest, "dev0");
84
85 qtest_quit(qtest);
86 }
87
test_q35_pci_unplug_request(void)88 static void test_q35_pci_unplug_request(void)
89 {
90 QTestState *qtest;
91
92 if (!qtest_has_device("virtio-mouse-pci")) {
93 g_test_skip("Device virtio-mouse-pci not available");
94 return;
95 }
96
97 qtest = qtest_initf("-machine q35 "
98 "-device pcie-root-port,id=p1 "
99 "-device pcie-pci-bridge,bus=p1,id=b1 "
100 "-device virtio-mouse-pci,bus=b1,id=dev0");
101
102 process_device_remove(qtest, "dev0");
103
104 qtest_quit(qtest);
105 }
106
test_pci_unplug_json_request(void)107 static void test_pci_unplug_json_request(void)
108 {
109 QTestState *qtest;
110 const char *arch = qtest_get_arch();
111 const char *machine_addition = "";
112
113 if (!qtest_has_device("virtio-mouse-pci")) {
114 g_test_skip("Device virtio-mouse-pci not available");
115 return;
116 }
117
118 if (strcmp(arch, "i386") == 0 || strcmp(arch, "x86_64") == 0) {
119 machine_addition = "-machine pc";
120 }
121
122 qtest = qtest_initf(
123 "%s -device \"{'driver': 'virtio-mouse-pci', 'id': 'dev0'}\"",
124 machine_addition);
125
126 process_device_remove(qtest, "dev0");
127
128 qtest_quit(qtest);
129 }
130
test_q35_pci_unplug_json_request(void)131 static void test_q35_pci_unplug_json_request(void)
132 {
133 QTestState *qtest;
134 const char *port = "-device \"{'driver': 'pcie-root-port', "
135 "'id': 'p1'}\"";
136
137 const char *bridge = "-device \"{'driver': 'pcie-pci-bridge', "
138 "'id': 'b1', "
139 "'bus': 'p1'}\"";
140
141 const char *device = "-device \"{'driver': 'virtio-mouse-pci', "
142 "'bus': 'b1', "
143 "'id': 'dev0'}\"";
144
145 if (!qtest_has_device("virtio-mouse-pci")) {
146 g_test_skip("Device virtio-mouse-pci not available");
147 return;
148 }
149
150 qtest = qtest_initf("-machine q35 %s %s %s", port, bridge, device);
151
152 process_device_remove(qtest, "dev0");
153
154 qtest_quit(qtest);
155 }
156
test_ccw_unplug(void)157 static void test_ccw_unplug(void)
158 {
159 QTestState *qtest;
160
161 if (!qtest_has_device("virtio-balloon-ccw")) {
162 g_test_skip("Device virtio-balloon-ccw not available");
163 return;
164 }
165
166 qtest = qtest_initf("-device virtio-balloon-ccw,id=dev0");
167
168 qtest_qmp_device_del_send(qtest, "dev0");
169 wait_device_deleted_event(qtest, "dev0");
170
171 qtest_quit(qtest);
172 }
173
test_spapr_cpu_unplug_request(void)174 static void test_spapr_cpu_unplug_request(void)
175 {
176 QTestState *qtest;
177
178 qtest = qtest_initf("-cpu power9_v2.2 -smp 1,maxcpus=2 "
179 "-device power9_v2.2-spapr-cpu-core,core-id=1,id=dev0");
180
181 /* similar to test_pci_unplug_request */
182 process_device_remove(qtest, "dev0");
183
184 qtest_quit(qtest);
185 }
186
test_spapr_memory_unplug_request(void)187 static void test_spapr_memory_unplug_request(void)
188 {
189 QTestState *qtest;
190
191 qtest = qtest_initf("-m 256M,slots=1,maxmem=768M "
192 "-object memory-backend-ram,id=mem0,size=512M "
193 "-device pc-dimm,id=dev0,memdev=mem0");
194
195 /* similar to test_pci_unplug_request */
196 process_device_remove(qtest, "dev0");
197
198 qtest_quit(qtest);
199 }
200
test_spapr_phb_unplug_request(void)201 static void test_spapr_phb_unplug_request(void)
202 {
203 QTestState *qtest;
204
205 qtest = qtest_initf("-device spapr-pci-host-bridge,index=1,id=dev0");
206
207 /* similar to test_pci_unplug_request */
208 process_device_remove(qtest, "dev0");
209
210 qtest_quit(qtest);
211 }
212
main(int argc,char ** argv)213 int main(int argc, char **argv)
214 {
215 const char *arch = qtest_get_arch();
216
217 g_test_init(&argc, &argv, NULL);
218
219 /*
220 * We need a system that will process unplug requests during system resets
221 * and does not do PCI surprise removal. This holds for x86 ACPI,
222 * s390x and spapr.
223 */
224 qtest_add_func("/device-plug/pci-unplug-request",
225 test_pci_unplug_request);
226 qtest_add_func("/device-plug/pci-unplug-json-request",
227 test_pci_unplug_json_request);
228
229 if (!strcmp(arch, "s390x")) {
230 qtest_add_func("/device-plug/ccw-unplug",
231 test_ccw_unplug);
232 }
233
234 if (!strcmp(arch, "ppc64")) {
235 qtest_add_func("/device-plug/spapr-cpu-unplug-request",
236 test_spapr_cpu_unplug_request);
237 qtest_add_func("/device-plug/spapr-memory-unplug-request",
238 test_spapr_memory_unplug_request);
239 qtest_add_func("/device-plug/spapr-phb-unplug-request",
240 test_spapr_phb_unplug_request);
241 }
242
243 if (!strcmp(arch, "x86_64") && qtest_has_machine("q35")) {
244 qtest_add_func("/device-plug/q35-pci-unplug-request",
245 test_q35_pci_unplug_request);
246 qtest_add_func("/device-plug/q35-pci-unplug-json-request",
247 test_q35_pci_unplug_json_request);
248 }
249
250 return g_test_run();
251 }
252