qemuhotplugtest.c 29.8 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
/*
 * Copyright (C) 2011-2013 Red Hat, Inc.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library.  If not, see
 * <http://www.gnu.org/licenses/>.
 *
 */

#include <config.h>

22
#include "qemu/qemu_alias.h"
23 24 25 26 27
#include "qemu/qemu_conf.h"
#include "qemu/qemu_hotplug.h"
#include "qemumonitortestutils.h"
#include "testutils.h"
#include "testutilsqemu.h"
28
#include "testutilsqemuschema.h"
29
#include "virhostdev.h"
30 31 32 33 34 35 36 37 38
#include "virerror.h"
#include "virstring.h"
#include "virthread.h"
#include "virfile.h"

#define VIR_FROM_THIS VIR_FROM_NONE

static virQEMUDriver driver;

39 40 41 42 43 44
enum {
    ATTACH,
    DETACH,
    UPDATE
};

45 46
#define QEMU_HOTPLUG_TEST_DOMAIN_ID 7

47 48 49 50 51
struct qemuHotplugTestData {
    const char *domain_filename;
    const char *device_filename;
    bool fail;
    const char *const *mon;
52 53 54
    int action;
    bool keep;
    virDomainObjPtr vm;
55
    bool deviceDeletedEvent;
56 57 58 59 60
};

static int
qemuHotplugCreateObjects(virDomainXMLOptionPtr xmlopt,
                         virDomainObjPtr *vm,
61
                         const char *domxml)
62
{
63
    qemuDomainObjPrivatePtr priv = NULL;
64
    const unsigned int parseFlags = 0;
65 66

    if (!(*vm = virDomainObjNew(xmlopt)))
67
        return -1;
68

69 70 71
    priv = (*vm)->privateData;

    if (!(priv->qemuCaps = virQEMUCapsNew()))
72
        return -1;
73

74
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_VIRTIO_SCSI);
75
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_DEVICE_USB_STORAGE);
76
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_CCW);
77 78
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_DEVICE_IVSHMEM_PLAIN);
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_DEVICE_IVSHMEM_DOORBELL);
79
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_SCSI_DISK_WWN);
80 81
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_DEVICE_VFIO_PCI);
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_DEVICE_SPAPR_PCI_HOST_BRIDGE);
82 83 84
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_DEVICE_QXL);
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_DEVICE_VGA);
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_DEVICE_CIRRUS_VGA);
85 86
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_PIIX_DISABLE_S3);
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_PIIX_DISABLE_S4);
87
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_VNC);
88 89
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_SPICE);
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_SPICE_FILE_XFER_DISABLE);
90 91
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_PR_MANAGER_HELPER);
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_SCSI_BLOCK);
92

93
    if (qemuTestCapsCacheInsert(driver.qemuCapsCache, priv->qemuCaps) < 0)
94
        return -1;
P
Pavel Fedin 已提交
95 96 97

    if (!((*vm)->def = virDomainDefParseString(domxml,
                                               driver.xmlopt,
98
                                               NULL,
99
                                               parseFlags)))
100
        return -1;
P
Pavel Fedin 已提交
101

102 103
    if (qemuDomainAssignAddresses((*vm)->def, priv->qemuCaps,
                                  &driver, *vm, true) < 0) {
104
        return -1;
105
    }
106 107

    if (qemuAssignDeviceAliases((*vm)->def, priv->qemuCaps) < 0)
108
        return -1;
109

110 111
    (*vm)->def->id = QEMU_HOTPLUG_TEST_DOMAIN_ID;

112
    if (qemuDomainSetPrivatePaths(&driver, *vm) < 0)
113
        return -1;
114

115
    return 0;
116 117
}

118 119 120 121 122 123 124
static int
testQemuHotplugAttach(virDomainObjPtr vm,
                      virDomainDeviceDefPtr dev)
{
    int ret = -1;

    switch (dev->type) {
125 126 127 128
    case VIR_DOMAIN_DEVICE_DISK:
        /* conn in only used for storage pool and secrets lookup so as long
         * as we don't use any of them, passing NULL should be safe
         */
129
        ret = qemuDomainAttachDeviceDiskLive(&driver, vm, dev);
130
        break;
131
    case VIR_DOMAIN_DEVICE_CHR:
132
        ret = qemuDomainAttachChrDevice(&driver, vm, dev->data.chr);
133
        break;
134 135 136
    case VIR_DOMAIN_DEVICE_SHMEM:
        ret = qemuDomainAttachShmemDevice(&driver, vm, dev->data.shmem);
        break;
M
Michal Privoznik 已提交
137 138 139
    case VIR_DOMAIN_DEVICE_WATCHDOG:
        ret = qemuDomainAttachWatchdog(&driver, vm, dev->data.watchdog);
        break;
140 141 142
    case VIR_DOMAIN_DEVICE_HOSTDEV:
        ret = qemuDomainAttachHostDevice(&driver, vm, dev->data.hostdev);
        break;
143
    default:
144
        VIR_TEST_VERBOSE("device type '%s' cannot be attached",
145
                virDomainDeviceTypeToString(dev->type));
146 147 148 149 150 151 152 153
        break;
    }

    return ret;
}

static int
testQemuHotplugDetach(virDomainObjPtr vm,
154 155
                      virDomainDeviceDefPtr dev,
                      bool async)
156 157 158 159
{
    int ret = -1;

    switch (dev->type) {
160
    case VIR_DOMAIN_DEVICE_DISK:
161
    case VIR_DOMAIN_DEVICE_CHR:
162
    case VIR_DOMAIN_DEVICE_SHMEM:
M
Michal Privoznik 已提交
163
    case VIR_DOMAIN_DEVICE_WATCHDOG:
164
    case VIR_DOMAIN_DEVICE_HOSTDEV:
165
        ret = qemuDomainDetachDeviceLive(vm, dev, &driver, async);
M
Michal Privoznik 已提交
166
        break;
167
    default:
168
        VIR_TEST_VERBOSE("device type '%s' cannot be detached",
169
                virDomainDeviceTypeToString(dev->type));
170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191
        break;
    }

    return ret;
}

static int
testQemuHotplugUpdate(virDomainObjPtr vm,
                      virDomainDeviceDefPtr dev)
{
    int ret = -1;

    /* XXX Ideally, we would call qemuDomainUpdateDeviceLive here.  But that
     * would require us to provide virConnectPtr and virDomainPtr (they're used
     * in case of updating a disk device. So for now, we will proceed with
     * breaking the function into pieces. If we ever learn how to fake those
     * required object, we can replace this code then. */
    switch (dev->type) {
    case VIR_DOMAIN_DEVICE_GRAPHICS:
        ret = qemuDomainChangeGraphics(&driver, vm, dev->data.graphics);
        break;
    default:
192
        VIR_TEST_VERBOSE("device type '%s' cannot be updated",
193
                virDomainDeviceTypeToString(dev->type));
194 195 196 197 198 199
        break;
    }

    return ret;
}

200 201 202
static int
testQemuHotplugCheckResult(virDomainObjPtr vm,
                           const char *expected,
203
                           const char *expectedFile,
204 205 206 207 208
                           bool fail)
{
    char *actual;
    int ret;

209
    actual = virDomainDefFormat(vm->def, driver.xmlopt,
210
                                VIR_DOMAIN_DEF_FORMAT_SECURE);
211 212
    if (!actual)
        return -1;
213
    vm->def->id = QEMU_HOTPLUG_TEST_DOMAIN_ID;
214 215

    if (STREQ(expected, actual)) {
216
        if (fail)
217
            VIR_TEST_VERBOSE("domain XML should not match the expected result");
218 219 220
        ret = 0;
    } else {
        if (!fail)
221 222 223
            virTestDifferenceFull(stderr,
                                  expected, expectedFile,
                                  actual, NULL);
224 225 226 227 228 229 230
        ret = -1;
    }

    VIR_FREE(actual);
    return ret;
}

231 232 233 234 235 236 237
static int
testQemuHotplug(const void *data)
{
    int ret = -1;
    struct qemuHotplugTestData *test = (struct qemuHotplugTestData *) data;
    char *domain_filename = NULL;
    char *device_filename = NULL;
238 239
    char *result_filename = NULL;
    char *domain_xml = NULL;
240
    char *device_xml = NULL;
241
    char *result_xml = NULL;
242 243
    const char *const *tmp;
    bool fail = test->fail;
244
    bool keep = test->keep;
245
    unsigned int device_parse_flags = 0;
246 247 248 249 250 251
    virDomainObjPtr vm = NULL;
    virDomainDeviceDefPtr dev = NULL;
    virCapsPtr caps = NULL;
    qemuMonitorTestPtr test_mon = NULL;
    qemuDomainObjPrivatePtr priv = NULL;

252 253 254 255 256 257 258
    domain_filename = g_strdup_printf("%s/qemuhotplugtestdomains/qemuhotplug-%s.xml",
                                      abs_srcdir, test->domain_filename);
    device_filename = g_strdup_printf("%s/qemuhotplugtestdevices/qemuhotplug-%s.xml",
                                      abs_srcdir, test->device_filename);
    result_filename = g_strdup_printf("%s/qemuhotplugtestdomains/qemuhotplug-%s+%s.xml",
                                      abs_srcdir, test->domain_filename,
                                      test->device_filename);
259

260 261
    if (virTestLoadFile(domain_filename, &domain_xml) < 0 ||
        virTestLoadFile(device_filename, &device_xml) < 0)
262 263
        goto cleanup;

264
    if (test->action == ATTACH &&
265
        virTestLoadFile(result_filename, &result_xml) < 0)
266 267 268 269 270
        goto cleanup;

    if (!(caps = virQEMUDriverGetCapabilities(&driver, false)))
        goto cleanup;

271 272
    if (test->vm) {
        vm = test->vm;
273
        if (!vm->def) {
274
            VIR_TEST_VERBOSE("test skipped due to failure of dependent test");
275 276
            goto cleanup;
        }
277
    } else {
278
        if (qemuHotplugCreateObjects(driver.xmlopt, &vm, domain_xml) < 0)
279 280
            goto cleanup;
    }
281

282
    if (test->action == ATTACH)
283
        device_parse_flags = VIR_DOMAIN_DEF_PARSE_INACTIVE;
284

285
    if (!(dev = virDomainDeviceDefParse(device_xml, vm->def,
286
                                        driver.xmlopt, NULL,
287
                                        device_parse_flags)))
288 289 290 291
        goto cleanup;

    /* Now is the best time to feed the spoofed monitor with predefined
     * replies. */
292
    if (!(test_mon = qemuMonitorTestNew(driver.xmlopt, vm, &driver,
293
                                        NULL, NULL)))
294 295 296 297 298 299 300 301 302 303 304 305 306 307
        goto cleanup;

    tmp = test->mon;
    while (tmp && *tmp) {
        const char *command_name;
        const char *response;

        if (!(command_name = *tmp++) ||
            !(response = *tmp++))
            break;
        if (qemuMonitorTestAddItem(test_mon, command_name, response) < 0)
            goto cleanup;
    }

308
    priv = vm->privateData;
309 310 311 312 313 314 315
    priv->mon = qemuMonitorTestGetMonitor(test_mon);

    /* XXX We need to unlock the monitor here, as
     * qemuDomainObjEnterMonitorInternal (called from qemuDomainChangeGraphics)
     * tries to lock it again */
    virObjectUnlock(priv->mon);

316 317 318
    switch (test->action) {
    case ATTACH:
        ret = testQemuHotplugAttach(vm, dev);
319 320 321 322 323
        if (ret == 0) {
            /* vm->def stolen dev->data.* so we just need to free the dev
             * envelope */
            VIR_FREE(dev);
        }
324
        if (ret == 0 || fail)
325 326
            ret = testQemuHotplugCheckResult(vm, result_xml,
                                             result_filename, fail);
327
        break;
328 329

    case DETACH:
330
        ret = testQemuHotplugDetach(vm, dev, false);
331
        if (ret == 0 || fail)
332 333
            ret = testQemuHotplugCheckResult(vm, domain_xml,
                                             domain_filename, fail);
334
        break;
335 336 337

    case UPDATE:
        ret = testQemuHotplugUpdate(vm, dev);
338 339
    }

340 341
    virObjectLock(priv->mon);

342
 cleanup:
343 344
    VIR_FREE(domain_filename);
    VIR_FREE(device_filename);
345 346
    VIR_FREE(result_filename);
    VIR_FREE(domain_xml);
347
    VIR_FREE(device_xml);
348
    VIR_FREE(result_xml);
349 350 351
    /* don't dispose test monitor with VM */
    if (priv)
        priv->mon = NULL;
352 353 354 355 356 357
    if (keep) {
        test->vm = vm;
    } else {
        virObjectUnref(vm);
        test->vm = NULL;
    }
358 359 360 361 362 363
    virDomainDeviceDefFree(dev);
    virObjectUnref(caps);
    qemuMonitorTestFree(test_mon);
    return ((ret < 0 && fail) || (!ret && !fail)) ? 0 : -1;
}

364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381

struct testQemuHotplugCpuData {
    char *file_xml_dom;
    char *file_xml_res_live;
    char *file_xml_res_conf;
    char *file_json_monitor;

    char *xml_dom;

    virDomainObjPtr vm;
    qemuMonitorTestPtr mon;
    bool modern;
};


static void
testQemuHotplugCpuDataFree(struct testQemuHotplugCpuData *data)
{
382
    qemuDomainObjPrivatePtr priv;
383
    qemuMonitorPtr mon;
384

385 386 387 388 389 390 391 392 393 394
    if (!data)
        return;

    VIR_FREE(data->file_xml_dom);
    VIR_FREE(data->file_xml_res_live);
    VIR_FREE(data->file_xml_res_conf);
    VIR_FREE(data->file_json_monitor);

    VIR_FREE(data->xml_dom);

395 396 397 398 399 400 401
    if (data->vm) {
        priv = data->vm->privateData;
        priv->mon = NULL;

        virObjectUnref(data->vm);
    }

402 403
    mon = qemuMonitorTestGetMonitor(data->mon);
    virObjectLock(mon);
404
    qemuMonitorTestFree(data->mon);
405
    VIR_FREE(data);
406 407 408 409 410
}


static struct testQemuHotplugCpuData *
testQemuHotplugCpuPrepare(const char *test,
411 412
                          bool modern,
                          virHashTablePtr qmpschema)
413 414 415 416 417 418
{
    qemuDomainObjPrivatePtr priv = NULL;
    virCapsPtr caps = NULL;
    char *prefix = NULL;
    struct testQemuHotplugCpuData *data = NULL;

419
    prefix = g_strdup_printf("%s/qemuhotplugtestcpus/%s", abs_srcdir, test);
420 421 422 423 424 425

    if (VIR_ALLOC(data) < 0)
        goto error;

    data->modern = modern;

426 427 428 429
    data->file_xml_dom = g_strdup_printf("%s-domain.xml", prefix);
    data->file_xml_res_live = g_strdup_printf("%s-result-live.xml", prefix);
    data->file_xml_res_conf = g_strdup_printf("%s-result-conf.xml", prefix);
    data->file_json_monitor = g_strdup_printf("%s-monitor.json", prefix);
430 431 432 433

    if (virTestLoadFile(data->file_xml_dom, &data->xml_dom) < 0)
        goto error;

434
    if (qemuHotplugCreateObjects(driver.xmlopt, &data->vm, data->xml_dom) < 0)
435 436 437 438 439 440 441
        goto error;

    if (!(caps = virQEMUDriverGetCapabilities(&driver, false)))
        goto error;

    /* create vm->newDef */
    data->vm->persistent = true;
442
    if (virDomainObjSetDefTransient(driver.xmlopt, data->vm, NULL) < 0)
443 444 445 446
        goto error;

    priv = data->vm->privateData;

447 448
    virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_QUERY_CPUS_FAST);

449 450 451 452
    if (data->modern)
        virQEMUCapsSet(priv->qemuCaps, QEMU_CAPS_QUERY_HOTPLUGGABLE_CPUS);

    if (!(data->mon = qemuMonitorTestNewFromFileFull(data->file_json_monitor,
453
                                                     &driver, data->vm, qmpschema)))
454 455
        goto error;

456 457
    qemuMonitorTestAllowUnusedCommands(data->mon);

458 459 460 461 462 463
    priv->mon = qemuMonitorTestGetMonitor(data->mon);
    virObjectUnlock(priv->mon);

    if (qemuDomainRefreshVcpuInfo(&driver, data->vm, 0, false) < 0)
        goto error;

464 465
    VIR_FREE(prefix);

466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483
    return data;

 error:
    virObjectUnref(caps);
    testQemuHotplugCpuDataFree(data);
    VIR_FREE(prefix);
    return NULL;
}


static int
testQemuHotplugCpuFinalize(struct testQemuHotplugCpuData *data)
{
    int ret = -1;
    char *activeXML = NULL;
    char *configXML = NULL;

    if (data->file_xml_res_live) {
484
        if (!(activeXML = virDomainDefFormat(data->vm->def, driver.xmlopt,
485 486 487 488 489 490 491 492
                                             VIR_DOMAIN_DEF_FORMAT_SECURE)))
            goto cleanup;

        if (virTestCompareToFile(activeXML, data->file_xml_res_live) < 0)
            goto cleanup;
    }

    if (data->file_xml_res_conf) {
493
        if (!(configXML = virDomainDefFormat(data->vm->newDef, driver.xmlopt,
494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513
                                             VIR_DOMAIN_DEF_FORMAT_SECURE |
                                             VIR_DOMAIN_DEF_FORMAT_INACTIVE)))
            goto cleanup;

        if (virTestCompareToFile(configXML, data->file_xml_res_conf) < 0)
            goto cleanup;
    }

    ret = 0;

 cleanup:
     VIR_FREE(activeXML);
     VIR_FREE(configXML);
     return ret;
}


struct testQemuHotplugCpuParams {
    const char *test;
    int newcpus;
514 515
    const char *cpumap;
    bool state;
516 517
    bool modern;
    bool fail;
518
    virHashTablePtr schema;
519 520 521 522 523 524 525 526 527 528 529
};


static int
testQemuHotplugCpuGroup(const void *opaque)
{
    const struct testQemuHotplugCpuParams *params = opaque;
    struct testQemuHotplugCpuData *data = NULL;
    int ret = -1;
    int rc;

530 531
    if (!(data = testQemuHotplugCpuPrepare(params->test, params->modern,
                                           params->schema)))
532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557
        return -1;

    rc = qemuDomainSetVcpusInternal(&driver, data->vm, data->vm->def,
                                    data->vm->newDef, params->newcpus,
                                    params->modern);

    if (params->fail) {
        if (rc == 0)
            fprintf(stderr, "cpu test '%s' should have failed\n", params->test);
        else
            ret = 0;

        goto cleanup;
    } else {
        if (rc < 0)
            goto cleanup;
    }

    ret = testQemuHotplugCpuFinalize(data);

 cleanup:
    testQemuHotplugCpuDataFree(data);
    return ret;
}


558 559 560 561 562 563 564 565 566
static int
testQemuHotplugCpuIndividual(const void *opaque)
{
    const struct testQemuHotplugCpuParams *params = opaque;
    struct testQemuHotplugCpuData *data = NULL;
    virBitmapPtr map = NULL;
    int ret = -1;
    int rc;

567 568
    if (!(data = testQemuHotplugCpuPrepare(params->test, params->modern,
                                           params->schema)))
569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596
        return -1;

    if (virBitmapParse(params->cpumap, &map, 128) < 0)
        goto cleanup;

    rc = qemuDomainSetVcpuInternal(&driver, data->vm, data->vm->def,
                                   data->vm->newDef, map, params->state);

    if (params->fail) {
        if (rc == 0)
            fprintf(stderr, "cpu test '%s' should have failed\n", params->test);
        else
            ret = 0;

        goto cleanup;
    } else {
        if (rc < 0)
            goto cleanup;
    }

    ret = testQemuHotplugCpuFinalize(data);

 cleanup:
    virBitmapFree(map);
    testQemuHotplugCpuDataFree(data);
    return ret;
}

597
#define FAKEROOTDIRTEMPLATE abs_builddir "/fakerootdir-XXXXXX"
598 599


600 601 602
static int
mymain(void)
{
J
Ján Tomko 已提交
603
    g_autoptr(virHashTable) qmpschema = NULL;
604
    int ret = 0;
605
    struct qemuHotplugTestData data = {0};
606
    struct testQemuHotplugCpuParams cpudata;
J
John Ferlan 已提交
607
    g_autofree char *fakerootdir = NULL;
608
    g_autoptr(virQEMUDriverConfig) cfg = NULL;
609

610
    fakerootdir = g_strdup(FAKEROOTDIRTEMPLATE);
611

J
Ján Tomko 已提交
612
    if (!g_mkdtemp(fakerootdir)) {
613 614 615 616
        fprintf(stderr, "Cannot create fakerootdir");
        abort();
    }

617
    g_setenv("LIBVIRT_FAKE_ROOT_DIR", fakerootdir, TRUE);
618

619
    if (qemuTestDriverInit(&driver) < 0)
620 621
        return EXIT_FAILURE;

622 623
    cfg = virQEMUDriverGetConfig(&driver);

624 625 626 627
    virEventRegisterDefaultImpl();

    VIR_FREE(driver.config->spiceListen);
    VIR_FREE(driver.config->vncListen);
628
    /* some dummy values from 'config file' */
629
    driver.config->spicePassword = g_strdup("123456");
630

631
    if (!(driver.domainEventState = virObjectEventStateNew()))
632 633
        return EXIT_FAILURE;

634
    if (!(qmpschema = testQEMUSchemaLoad("x86_64"))) {
635 636 637 638 639 640
        VIR_TEST_VERBOSE("failed to load qapi schema\n");
        return EXIT_FAILURE;
    }

    cpudata.schema = qmpschema;

641 642 643 644 645 646
    driver.lockManager = virLockManagerPluginNew("nop", "qemu",
                                                 driver.config->configBaseDir,
                                                 0);
    if (!driver.lockManager)
        return EXIT_FAILURE;

647
    driver.hostdevMgr = virHostdevManagerGetDefault();
648 649 650 651 652
    if (driver.hostdevMgr == NULL) {
        VIR_TEST_VERBOSE("Could not initialize HostdevManager - %s\n",
                         virGetLastErrorMessage());
        return EXIT_FAILURE;
    }
653

654

655
#define DO_TEST(file, ACTION, dev, fial, kep, ...) \
656 657 658 659 660 661 662 663 664 665 666
    do { \
        const char *my_mon[] = { __VA_ARGS__, NULL}; \
        const char *name = file " " #ACTION " " dev; \
        data.action = ACTION; \
        data.domain_filename = file; \
        data.device_filename = dev; \
        data.fail = fial; \
        data.mon = my_mon; \
        data.keep = kep; \
        if (virTestRun(name, testQemuHotplug, &data) < 0) \
            ret = -1; \
667
    } while (0)
668

669
#define DO_TEST_ATTACH(file, dev, fial, kep, ...) \
670
    DO_TEST(file, ATTACH, dev, fial, kep, __VA_ARGS__)
671

672
#define DO_TEST_DETACH(file, dev, fial, kep, ...) \
673
    DO_TEST(file, DETACH, dev, fial, kep, __VA_ARGS__)
674

675
#define DO_TEST_UPDATE(file, dev, fial, kep, ...) \
676
    DO_TEST(file, UPDATE, dev, fial, kep, __VA_ARGS__)
677

678 679

#define QMP_OK      "{\"return\": {}}"
680
#define HMP(msg)    "{\"return\": \"" msg "\"}"
681

682
#define QMP_DEVICE_DELETED(dev) \
683 684 685 686 687 688 689 690 691 692
    "{" \
    "    \"timestamp\": {" \
    "        \"seconds\": 1374137171," \
    "        \"microseconds\": 2659" \
    "    }," \
    "    \"event\": \"DEVICE_DELETED\"," \
    "    \"data\": {" \
    "        \"device\": \"" dev "\"," \
    "        \"path\": \"/machine/peripheral/" dev "\"" \
    "    }" \
693 694
    "}\r\n"

695
    cfg->spiceTLS = true;
696 697
    DO_TEST_UPDATE("graphics-spice", "graphics-spice-nochange", false, false, NULL);
    DO_TEST_UPDATE("graphics-spice-timeout", "graphics-spice-timeout-nochange", false, false,
698
                   "set_password", QMP_OK, "expire_password", QMP_OK);
699
    DO_TEST_UPDATE("graphics-spice-timeout", "graphics-spice-timeout-password", false, false,
700
                   "set_password", QMP_OK, "expire_password", QMP_OK);
701
    DO_TEST_UPDATE("graphics-spice", "graphics-spice-listen", true, false, NULL);
702
    DO_TEST_UPDATE("graphics-spice-listen-network", "graphics-spice-listen-network-password", false, false,
703
                   "set_password", QMP_OK, "expire_password", QMP_OK);
704
    cfg->spiceTLS = false;
705 706 707
    /* Strange huh? Currently, only graphics can be updated :-P */
    DO_TEST_UPDATE("disk-cdrom", "disk-cdrom-nochange", true, false, NULL);

708
    DO_TEST_ATTACH("console-compat-2-live", "console-virtio", false, true,
709
                   "chardev-add", "{\"return\": {\"pty\": \"/dev/pts/26\"}}",
710
                   "device_add", QMP_OK);
711

712
    DO_TEST_DETACH("console-compat-2-live", "console-virtio", false, false,
713
                   "device_del", QMP_DEVICE_DELETED("console1") QMP_OK,
714
                   "chardev-remove", QMP_OK);
715

716 717 718
    DO_TEST_ATTACH("base-live", "disk-virtio", false, true,
                   "human-monitor-command", HMP("OK\\r\\n"),
                   "device_add", QMP_OK);
719
    DO_TEST_DETACH("base-live", "disk-virtio", true, true,
720
                   "device_del", QMP_OK);
721
    DO_TEST_DETACH("base-live", "disk-virtio", false, false,
722 723 724
                   "device_del", QMP_DEVICE_DELETED("virtio-disk4") QMP_OK,
                   "human-monitor-command", HMP(""));

725 726 727
    DO_TEST_ATTACH("base-live", "disk-usb", false, true,
                   "human-monitor-command", HMP("OK\\r\\n"),
                   "device_add", QMP_OK);
728
    DO_TEST_DETACH("base-live", "disk-usb", true, true,
729
                   "device_del", QMP_OK);
730
    DO_TEST_DETACH("base-live", "disk-usb", false, false,
731 732 733
                   "device_del", QMP_DEVICE_DELETED("usb-disk16") QMP_OK,
                   "human-monitor-command", HMP(""));

734 735 736
    DO_TEST_ATTACH("base-live", "disk-scsi", false, true,
                   "human-monitor-command", HMP("OK\\r\\n"),
                   "device_add", QMP_OK);
737
    DO_TEST_DETACH("base-live", "disk-scsi", true, true,
738
                   "device_del", QMP_OK);
739
    DO_TEST_DETACH("base-live", "disk-scsi", false, false,
740 741 742
                   "device_del", QMP_DEVICE_DELETED("scsi0-0-0-5") QMP_OK,
                   "human-monitor-command", HMP(""));

743 744 745 746 747 748 749 750 751
    DO_TEST_ATTACH("base-without-scsi-controller-live", "disk-scsi-2", false, true,
                   /* Four controllers added */
                   "device_add", QMP_OK,
                   "device_add", QMP_OK,
                   "device_add", QMP_OK,
                   "device_add", QMP_OK,
                   "human-monitor-command", HMP("OK\\r\\n"),
                   /* Disk added */
                   "device_add", QMP_OK);
752
    DO_TEST_DETACH("base-with-scsi-controller-live", "disk-scsi-2", true, true,
753
                   "device_del", QMP_OK);
754
    DO_TEST_DETACH("base-with-scsi-controller-live", "disk-scsi-2", false, false,
755
                   "device_del", QMP_DEVICE_DELETED("scsi3-0-5-6") QMP_OK,
756 757
                   "human-monitor-command", HMP(""));

758 759 760 761 762
    DO_TEST_ATTACH("base-live", "disk-scsi-multipath", false, true,
                   "object-add", QMP_OK,
                   "human-monitor-command", HMP("OK\\r\\n"),
                   "device_add", QMP_OK);
    DO_TEST_DETACH("base-live", "disk-scsi-multipath", true, true,
763
                   "device_del", QMP_OK);
764 765
    DO_TEST_DETACH("base-live", "disk-scsi-multipath", false, false,
                   "device_del", QMP_DEVICE_DELETED("scsi0-0-0-0") QMP_OK,
766 767
                   "human-monitor-command", HMP(""),
                   "object-del", QMP_OK);
768

769
    DO_TEST_ATTACH("base-live", "qemu-agent", false, true,
770 771
                   "chardev-add", QMP_OK,
                   "device_add", QMP_OK);
772
    DO_TEST_DETACH("base-live", "qemu-agent-detach", false, false,
773
                   "device_del", QMP_DEVICE_DELETED("channel0") QMP_OK,
774 775
                   "chardev-remove", QMP_OK);

776 777 778 779
    DO_TEST_ATTACH("base-ccw-live", "ccw-virtio", false, true,
                   "human-monitor-command", HMP("OK\\r\\n"),
                   "device_add", QMP_OK);
    DO_TEST_DETACH("base-ccw-live", "ccw-virtio", false, false,
780
                   "device_del", QMP_DEVICE_DELETED("virtio-disk4") QMP_OK,
781 782 783 784 785 786 787
                   "human-monitor-command", HMP(""));

    DO_TEST_ATTACH("base-ccw-live-with-ccw-virtio", "ccw-virtio-2", false, true,
                   "human-monitor-command", HMP("OK\\r\\n"),
                   "device_add", QMP_OK);

    DO_TEST_DETACH("base-ccw-live-with-ccw-virtio", "ccw-virtio-2", false, false,
788
                   "device_del", QMP_DEVICE_DELETED("virtio-disk0") QMP_OK,
789 790 791 792 793 794 795
                   "human-monitor-command", HMP(""));

    DO_TEST_ATTACH("base-ccw-live-with-ccw-virtio", "ccw-virtio-2-explicit", false, true,
                   "human-monitor-command", HMP("OK\\r\\n"),
                   "device_add", QMP_OK);

    DO_TEST_DETACH("base-ccw-live-with-ccw-virtio", "ccw-virtio-2-explicit", false, false,
796
                   "device_del", QMP_DEVICE_DELETED("virtio-disk0") QMP_OK,
797 798 799 800 801 802 803 804
                   "human-monitor-command", HMP(""));

    /* Attach a second device, then detach the first one. Then attach the first one again. */
    DO_TEST_ATTACH("base-ccw-live-with-ccw-virtio", "ccw-virtio-2-explicit", false, true,
                   "human-monitor-command", HMP("OK\\r\\n"),
                   "device_add", QMP_OK);

    DO_TEST_DETACH("base-ccw-live-with-2-ccw-virtio", "ccw-virtio-1-explicit", false, true,
805
                   "device_del", QMP_DEVICE_DELETED("virtio-disk4") QMP_OK,
806 807 808 809 810 811
                   "human-monitor-command", HMP(""));

    DO_TEST_ATTACH("base-ccw-live-with-2-ccw-virtio", "ccw-virtio-1-reverse", false, false,
                   "human-monitor-command", HMP("OK\\r\\n"),
                   "device_add", QMP_OK);

812 813 814 815 816 817 818
    DO_TEST_ATTACH("base-live", "ivshmem-plain", false, true,
                   "object-add", QMP_OK,
                   "device_add", QMP_OK);
    DO_TEST_ATTACH("base-live", "ivshmem-doorbell", false, true,
                   "chardev-add", QMP_OK,
                   "device_add", QMP_OK);
    DO_TEST_DETACH("base-live+ivshmem-plain", "ivshmem-doorbell-detach", false, true,
819
                   "device_del", QMP_DEVICE_DELETED("shmem1") QMP_OK,
820 821
                   "chardev-remove", QMP_OK);
    DO_TEST_DETACH("base-live", "ivshmem-plain-detach", false, false,
822
                   "device_del", QMP_DEVICE_DELETED("shmem0") QMP_OK,
823
                   "object-del", QMP_OK);
824
    DO_TEST_ATTACH("base-live+disk-scsi-wwn",
M
Michal Privoznik 已提交
825
                   "disk-scsi-duplicate-wwn", false, false,
826 827
                   "human-monitor-command", HMP("OK\\r\\n"),
                   "device_add", QMP_OK);
828

829 830 831 832 833 834 835 836 837
    DO_TEST_ATTACH("base-live", "hostdev-pci", false, true,
                   "device_add", QMP_OK);
    DO_TEST_DETACH("base-live", "hostdev-pci", false, false,
                   "device_del", QMP_DEVICE_DELETED("hostdev0") QMP_OK);
    DO_TEST_ATTACH("pseries-base-live", "hostdev-pci", false, true,
                   "device_add", QMP_OK);
    DO_TEST_DETACH("pseries-base-live", "hostdev-pci", false, false,
                   "device_del", QMP_DEVICE_DELETED("hostdev0") QMP_OK);

M
Michal Privoznik 已提交
838
    DO_TEST_ATTACH("base-live", "watchdog", false, true,
M
Michal Privoznik 已提交
839 840
                   "watchdog-set-action", QMP_OK,
                   "device_add", QMP_OK);
M
Michal Privoznik 已提交
841
    DO_TEST_DETACH("base-live", "watchdog-full", false, false,
842
                   "device_del", QMP_DEVICE_DELETED("watchdog0") QMP_OK);
M
Michal Privoznik 已提交
843

844 845 846 847
    DO_TEST_ATTACH("base-live", "watchdog-user-alias", false, true,
                   "watchdog-set-action", QMP_OK,
                   "device_add", QMP_OK);
    DO_TEST_DETACH("base-live", "watchdog-user-alias-full", false, false,
848
                   "device_del", QMP_DEVICE_DELETED("ua-UserWatchdog") QMP_OK);
849

850 851 852 853 854 855
    DO_TEST_ATTACH("base-live", "guestfwd", false, true,
                   "chardev-add", QMP_OK,
                   "netdev_add", QMP_OK);
    DO_TEST_DETACH("base-live", "guestfwd", false, false,
                   "netdev_del", QMP_OK);

856 857 858 859 860 861 862 863 864
#define DO_TEST_CPU_GROUP(prefix, vcpus, modernhp, expectfail) \
    do { \
        cpudata.test = prefix; \
        cpudata.newcpus = vcpus; \
        cpudata.modern = modernhp; \
        cpudata.fail = expectfail; \
        if (virTestRun("hotplug vcpus group " prefix, \
                       testQemuHotplugCpuGroup, &cpudata) < 0) \
            ret = -1; \
865 866 867
    } while (0)

    DO_TEST_CPU_GROUP("x86-modern-bulk", 7, true, false);
868
    DO_TEST_CPU_GROUP("x86-old-bulk", 7, false, false);
869 870 871 872
    DO_TEST_CPU_GROUP("ppc64-modern-bulk", 24, true, false);
    DO_TEST_CPU_GROUP("ppc64-modern-bulk", 15, true, true);
    DO_TEST_CPU_GROUP("ppc64-modern-bulk", 23, true, true);
    DO_TEST_CPU_GROUP("ppc64-modern-bulk", 25, true, true);
873

874 875 876 877 878 879 880 881 882 883
#define DO_TEST_CPU_INDIVIDUAL(prefix, mapstr, statefl, modernhp, expectfail) \
    do { \
        cpudata.test = prefix; \
        cpudata.cpumap = mapstr; \
        cpudata.state = statefl; \
        cpudata.modern = modernhp; \
        cpudata.fail = expectfail; \
        if (virTestRun("hotplug vcpus group " prefix, \
                       testQemuHotplugCpuIndividual, &cpudata) < 0) \
            ret = -1; \
884 885 886 887 888 889 890 891 892 893
    } while (0)

    DO_TEST_CPU_INDIVIDUAL("x86-modern-individual-add", "7", true, true, false);
    DO_TEST_CPU_INDIVIDUAL("x86-modern-individual-add", "6,7", true, true, true);
    DO_TEST_CPU_INDIVIDUAL("x86-modern-individual-add", "7", false, true, true);

    DO_TEST_CPU_INDIVIDUAL("ppc64-modern-individual", "16-23", true, true, false);
    DO_TEST_CPU_INDIVIDUAL("ppc64-modern-individual", "16-22", true, true, true);
    DO_TEST_CPU_INDIVIDUAL("ppc64-modern-individual", "17", true, true, true);

894 895 896
    if (getenv("LIBVIRT_SKIP_CLEANUP") == NULL)
        virFileDeleteTree(fakerootdir);

897
    qemuTestDriverFree(&driver);
898
    virObjectUnref(data.vm);
899 900 901
    return (ret == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
}

902 903
VIR_TEST_MAIN_PRELOAD(mymain,
                      VIR_TEST_MOCK("virpci"),
M
Michal Privoznik 已提交
904
                      VIR_TEST_MOCK("domaincaps"),
905 906
                      VIR_TEST_MOCK("virprocess"),
                      VIR_TEST_MOCK("qemuhotplug"));