xref: /qemu/tests/qtest/device-plug-test.c (revision 60f782b6) 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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