LXR qemu/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 "libqos/libqtest.h"
  15#include "qapi/qmp/qdict.h"
  16#include "qapi/qmp/qstring.h"
  17
  18static void device_del(QTestState *qtest, const char *id)
  19{
  20    QDict *resp;
  21
  22    resp = qtest_qmp(qtest,
  23                     "{'execute': 'device_del', 'arguments': { 'id': %s } }", id);
  24
  25    g_assert(qdict_haskey(resp, "return"));
  26    qobject_unref(resp);
  27}
  28
  29static void system_reset(QTestState *qtest)
  30{
  31    QDict *resp;
  32
  33    resp = qtest_qmp(qtest, "{'execute': 'system_reset'}");
  34    g_assert(qdict_haskey(resp, "return"));
  35    qobject_unref(resp);
  36}
  37
  38static void wait_device_deleted_event(QTestState *qtest, const char *id)
  39{
  40    QDict *resp, *data;
  41    QString *qstr;
  42
  43    /*
  44     * Other devices might get removed along with the removed device. Skip
  45     * these. The device of interest will be the last one.
  46     */
  47    for (;;) {
  48        resp = qtest_qmp_eventwait_ref(qtest, "DEVICE_DELETED");
  49        data = qdict_get_qdict(resp, "data");
  50        if (!data || !qdict_get(data, "device")) {
  51            qobject_unref(resp);
  52            continue;
  53        }
  54        qstr = qobject_to(QString, qdict_get(data, "device"));
  55        g_assert(qstr);
  56        if (!strcmp(qstring_get_str(qstr), id)) {
  57            qobject_unref(resp);
  58            break;
  59        }
  60        qobject_unref(resp);
  61    }
  62}
  63
  64static void test_pci_unplug_request(void)
  65{
  66    QTestState *qtest = qtest_initf("-device virtio-mouse-pci,id=dev0");
  67
  68    /*
  69     * Request device removal. As the guest is not running, the request won't
  70     * be processed. However during system reset, the removal will be
  71     * handled, removing the device.
  72     */
  73    device_del(qtest, "dev0");
  74    system_reset(qtest);
  75    wait_device_deleted_event(qtest, "dev0");
  76
  77    qtest_quit(qtest);
  78}
  79
  80static void test_ccw_unplug(void)
  81{
  82    QTestState *qtest = qtest_initf("-device virtio-balloon-ccw,id=dev0");
  83
  84    device_del(qtest, "dev0");
  85    wait_device_deleted_event(qtest, "dev0");
  86
  87    qtest_quit(qtest);
  88}
  89
  90static void test_spapr_cpu_unplug_request(void)
  91{
  92    QTestState *qtest;
  93
  94    qtest = qtest_initf("-cpu power9_v2.0 -smp 1,maxcpus=2 "
  95                        "-device power9_v2.0-spapr-cpu-core,core-id=1,id=dev0");
  96
  97    /* similar to test_pci_unplug_request */
  98    device_del(qtest, "dev0");
  99    system_reset(qtest);
 100    wait_device_deleted_event(qtest, "dev0");
 101
 102    qtest_quit(qtest);
 103}
 104
 105static void test_spapr_memory_unplug_request(void)
 106{
 107    QTestState *qtest;
 108
 109    qtest = qtest_initf("-m 256M,slots=1,maxmem=768M "
 110                        "-object memory-backend-ram,id=mem0,size=512M "
 111                        "-device pc-dimm,id=dev0,memdev=mem0");
 112
 113    /* similar to test_pci_unplug_request */
 114    device_del(qtest, "dev0");
 115    system_reset(qtest);
 116    wait_device_deleted_event(qtest, "dev0");
 117
 118    qtest_quit(qtest);
 119}
 120
 121static void test_spapr_phb_unplug_request(void)
 122{
 123    QTestState *qtest;
 124
 125    qtest = qtest_initf("-device spapr-pci-host-bridge,index=1,id=dev0");
 126
 127    /* similar to test_pci_unplug_request */
 128    device_del(qtest, "dev0");
 129    system_reset(qtest);
 130    wait_device_deleted_event(qtest, "dev0");
 131
 132    qtest_quit(qtest);
 133}
 134
 135int main(int argc, char **argv)
 136{
 137    const char *arch = qtest_get_arch();
 138
 139    g_test_init(&argc, &argv, NULL);
 140
 141    /*
 142     * We need a system that will process unplug requests during system resets
 143     * and does not do PCI surprise removal. This holds for x86 ACPI,
 144     * s390x and spapr.
 145     */
 146    qtest_add_func("/device-plug/pci-unplug-request",
 147                   test_pci_unplug_request);
 148
 149    if (!strcmp(arch, "s390x")) {
 150        qtest_add_func("/device-plug/ccw-unplug",
 151                       test_ccw_unplug);
 152    }
 153
 154    if (!strcmp(arch, "ppc64")) {
 155        qtest_add_func("/device-plug/spapr-cpu-unplug-request",
 156                       test_spapr_cpu_unplug_request);
 157        qtest_add_func("/device-plug/spapr-memory-unplug-request",
 158                       test_spapr_memory_unplug_request);
 159        qtest_add_func("/device-plug/spapr-phb-unplug-request",
 160                       test_spapr_phb_unplug_request);
 161    }
 162
 163    return g_test_run();
 164}
 165