tests/qtest/device-plug-test.c

   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 }