test_driver.c 194.8 KB
Newer Older
1
/*
2
 * test_driver.c: A "mock" hypervisor for use by application unit tests
3
 *
4
 * Copyright (C) 2006-2015 Red Hat, Inc.
5
 * Copyright (C) 2006 Daniel P. Berrange
6
 *
7 8 9 10 11 12 13 14 15 16 17
 * 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
18
 * License along with this library.  If not, see
O
Osier Yang 已提交
19
 * <http://www.gnu.org/licenses/>.
20 21 22 23
 *
 * Daniel Berrange <berrange@redhat.com>
 */

24
#include <config.h>
25

26 27 28
#include <stdio.h>
#include <string.h>
#include <sys/time.h>
29 30
#include <fcntl.h>
#include <unistd.h>
31
#include <sys/stat.h>
C
Cole Robinson 已提交
32
#include <libxml/xmlsave.h>
33
#include <libxml/xpathInternals.h>
34

35

36
#include "virerror.h"
37
#include "datatypes.h"
38
#include "test_driver.h"
39
#include "virbuffer.h"
40
#include "viruuid.h"
41
#include "capabilities.h"
42
#include "configmake.h"
43
#include "viralloc.h"
J
John Ferlan 已提交
44
#include "virnetworkobj.h"
L
Laine Stump 已提交
45
#include "interface_conf.h"
46
#include "domain_conf.h"
47
#include "domain_event.h"
48
#include "network_event.h"
49
#include "snapshot_conf.h"
50
#include "virfdstream.h"
C
Cole Robinson 已提交
51
#include "storage_conf.h"
J
John Ferlan 已提交
52
#include "virstorageobj.h"
53
#include "storage_event.h"
54
#include "node_device_conf.h"
J
John Ferlan 已提交
55
#include "virnodedeviceobj.h"
56
#include "node_device_event.h"
57
#include "virxml.h"
58
#include "virthread.h"
59
#include "virlog.h"
E
Eric Blake 已提交
60
#include "virfile.h"
61
#include "virtypedparam.h"
62
#include "virrandom.h"
63
#include "virstring.h"
64
#include "cpu/cpu.h"
65
#include "virauth.h"
66
#include "viratomic.h"
67
#include "virdomainobjlist.h"
J
John Ferlan 已提交
68
#include "virinterfaceobj.h"
69
#include "virhostcpu.h"
70

71 72
#define VIR_FROM_THIS VIR_FROM_TEST

73 74
VIR_LOG_INIT("test.test_driver");

75

76 77 78 79
#define MAX_CPUS 128

struct _testCell {
    unsigned long mem;
80
    unsigned long freeMem;
81
    int numCpus;
82
    virCapsHostNUMACellCPU cpus[MAX_CPUS];
83 84 85 86 87
};
typedef struct _testCell testCell;
typedef struct _testCell *testCellPtr;

#define MAX_CELLS 128
88

89 90 91 92 93 94 95
struct _testAuth {
    char *username;
    char *password;
};
typedef struct _testAuth testAuth;
typedef struct _testAuth *testAuthPtr;

96
struct _testDriver {
97
    virMutex lock;
98

99
    virNodeInfo nodeInfo;
100
    virInterfaceObjListPtr ifaces;
101
    bool transaction_running;
102
    virInterfaceObjListPtr backupIfaces;
C
Cole Robinson 已提交
103
    virStoragePoolObjList pools;
104
    virNodeDeviceObjListPtr devs;
105 106
    int numCells;
    testCell cells[MAX_CELLS];
107 108
    size_t numAuths;
    testAuthPtr auths;
109

110 111 112
    /* virAtomic access only */
    volatile int nextDomID;

113 114 115 116 117 118 119 120 121 122
    /* immutable pointer, immutable object after being initialized with
     * testBuildCapabilities */
    virCapsPtr caps;

    /* immutable pointer, immutable object */
    virDomainXMLOptionPtr xmlopt;

    /* immutable pointer, self-locking APIs */
    virDomainObjListPtr domains;
    virNetworkObjListPtr networks;
123
    virObjectEventStatePtr eventState;
124
};
125 126
typedef struct _testDriver testDriver;
typedef testDriver *testDriverPtr;
127

128
static testDriverPtr defaultConn;
129
static int defaultConnections;
130
static virMutex defaultLock = VIR_MUTEX_INITIALIZER;
131

132
#define TEST_MODEL "i686"
133
#define TEST_EMULATOR "/usr/bin/test-hv"
134

135
static const virNodeInfo defaultNodeInfo = {
136
    TEST_MODEL,
137 138 139 140 141 142 143
    1024*1024*3, /* 3 GB */
    16,
    1400,
    2,
    2,
    2,
    2,
144 145
};

146 147 148 149 150 151 152 153 154
static void
testDriverFree(testDriverPtr driver)
{
    if (!driver)
        return;

    virObjectUnref(driver->caps);
    virObjectUnref(driver->xmlopt);
    virObjectUnref(driver->domains);
155
    virNodeDeviceObjListFree(driver->devs);
156
    virObjectUnref(driver->networks);
157
    virInterfaceObjListFree(driver->ifaces);
158
    virStoragePoolObjListFree(&driver->pools);
159
    virObjectUnref(driver->eventState);
160 161 162 163 164
    virMutexUnlock(&driver->lock);
    virMutexDestroy(&driver->lock);

    VIR_FREE(driver);
}
165

166

167
static void testDriverLock(testDriverPtr driver)
168
{
169
    virMutexLock(&driver->lock);
170 171
}

172
static void testDriverUnlock(testDriverPtr driver)
173
{
174
    virMutexUnlock(&driver->lock);
175 176
}

177 178 179 180 181 182 183 184 185
static void testObjectEventQueue(testDriverPtr driver,
                                 virObjectEventPtr event)
{
    if (!event)
        return;

    virObjectEventStateQueue(driver->eventState, event);
}

186 187 188 189 190 191
#define TEST_NAMESPACE_HREF "http://libvirt.org/schemas/domain/test/1.0"

typedef struct _testDomainNamespaceDef testDomainNamespaceDef;
typedef testDomainNamespaceDef *testDomainNamespaceDefPtr;
struct _testDomainNamespaceDef {
    int runstate;
192
    bool transient;
C
Cole Robinson 已提交
193
    bool hasManagedSave;
194 195 196

    unsigned int num_snap_nodes;
    xmlNodePtr *snap_nodes;
197 198 199 200 201 202
};

static void
testDomainDefNamespaceFree(void *data)
{
    testDomainNamespaceDefPtr nsdata = data;
203 204 205 206 207 208 209 210 211
    size_t i;

    if (!nsdata)
        return;

    for (i = 0; i < nsdata->num_snap_nodes; i++)
        xmlFreeNode(nsdata->snap_nodes[i]);

    VIR_FREE(nsdata->snap_nodes);
212 213 214 215 216 217 218 219 220 221
    VIR_FREE(nsdata);
}

static int
testDomainDefNamespaceParse(xmlDocPtr xml ATTRIBUTE_UNUSED,
                            xmlNodePtr root ATTRIBUTE_UNUSED,
                            xmlXPathContextPtr ctxt,
                            void **data)
{
    testDomainNamespaceDefPtr nsdata = NULL;
222 223 224
    xmlNodePtr *nodes = NULL;
    int tmp, n;
    size_t i;
225 226 227 228 229 230 231 232 233 234 235 236 237
    unsigned int tmpuint;

    if (xmlXPathRegisterNs(ctxt, BAD_CAST "test",
                           BAD_CAST TEST_NAMESPACE_HREF) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to register xml namespace '%s'"),
                       TEST_NAMESPACE_HREF);
        return -1;
    }

    if (VIR_ALLOC(nsdata) < 0)
        return -1;

238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256
    n = virXPathNodeSet("./test:domainsnapshot", ctxt, &nodes);
    if (n < 0)
        goto error;

    if (n && VIR_ALLOC_N(nsdata->snap_nodes, n) < 0)
        goto error;

    for (i = 0; i < n; i++) {
        xmlNodePtr newnode = xmlCopyNode(nodes[i], 1);
        if (!newnode) {
            virReportOOMError();
            goto error;
        }

        nsdata->snap_nodes[nsdata->num_snap_nodes] = newnode;
        nsdata->num_snap_nodes++;
    }
    VIR_FREE(nodes);

257 258 259 260 261 262 263
    tmp = virXPathBoolean("boolean(./test:transient)", ctxt);
    if (tmp == -1) {
        virReportError(VIR_ERR_XML_ERROR, "%s", _("invalid transient"));
        goto error;
    }
    nsdata->transient = tmp;

C
Cole Robinson 已提交
264 265 266 267 268 269 270
    tmp = virXPathBoolean("boolean(./test:hasmanagedsave)", ctxt);
    if (tmp == -1) {
        virReportError(VIR_ERR_XML_ERROR, "%s", _("invalid hasmanagedsave"));
        goto error;
    }
    nsdata->hasManagedSave = tmp;

271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286
    tmp = virXPathUInt("string(./test:runstate)", ctxt, &tmpuint);
    if (tmp == 0) {
        if (tmpuint >= VIR_DOMAIN_LAST) {
            virReportError(VIR_ERR_XML_ERROR,
                           _("runstate '%d' out of range'"), tmpuint);
            goto error;
        }
        nsdata->runstate = tmpuint;
    } else if (tmp == -1) {
        nsdata->runstate = VIR_DOMAIN_RUNNING;
    } else if (tmp == -2) {
        virReportError(VIR_ERR_XML_ERROR, "%s",
                       _("invalid runstate"));
        goto error;
    }

287 288 289 290 291
    if (nsdata->transient && nsdata->runstate == VIR_DOMAIN_SHUTOFF) {
        virReportError(VIR_ERR_XML_ERROR, "%s",
            _("transient domain cannot have runstate 'shutoff'"));
        goto error;
    }
C
Cole Robinson 已提交
292 293 294 295 296
    if (nsdata->hasManagedSave && nsdata->runstate != VIR_DOMAIN_SHUTOFF) {
        virReportError(VIR_ERR_XML_ERROR, "%s",
            _("domain with managedsave data can only have runstate 'shutoff'"));
        goto error;
    }
297

298 299 300
    *data = nsdata;
    return 0;

301
 error:
302
    VIR_FREE(nodes);
303 304 305
    testDomainDefNamespaceFree(nsdata);
    return -1;
}
306

307
static virCapsPtr
308 309
testBuildCapabilities(virConnectPtr conn)
{
310
    testDriverPtr privconn = conn->privateData;
311 312
    virCapsPtr caps;
    virCapsGuestPtr guest;
313 314
    int guest_types[] = { VIR_DOMAIN_OSTYPE_HVM,
                          VIR_DOMAIN_OSTYPE_XEN };
315
    size_t i, j;
316

317
    if ((caps = virCapabilitiesNew(VIR_ARCH_I686, false, false)) == NULL)
318
        goto error;
319

320
    if (virCapabilitiesAddHostFeature(caps, "pae") < 0)
321
        goto error;
322
    if (virCapabilitiesAddHostFeature(caps, "nonpae") < 0)
323
        goto error;
324

325 326 327 328 329 330
    if (VIR_ALLOC_N(caps->host.pagesSize, 2) < 0)
        goto error;

    caps->host.pagesSize[caps->host.nPagesSize++] = 4;
    caps->host.pagesSize[caps->host.nPagesSize++] = 2048;

331
    for (i = 0; i < privconn->numCells; i++) {
332
        virCapsHostNUMACellCPUPtr cpu_cells;
333 334
        virCapsHostNUMACellPageInfoPtr pages;
        size_t nPages;
335

336 337 338 339 340 341 342
        if (VIR_ALLOC_N(cpu_cells, privconn->cells[i].numCpus) < 0 ||
            VIR_ALLOC_N(pages, caps->host.nPagesSize) < 0) {
                VIR_FREE(cpu_cells);
                goto error;
            }

        nPages = caps->host.nPagesSize;
343 344 345 346

        memcpy(cpu_cells, privconn->cells[i].cpus,
               sizeof(*cpu_cells) * privconn->cells[i].numCpus);

347 348 349 350
        for (j = 0; j < nPages; j++)
            pages[j].size = caps->host.pagesSize[j];

        pages[0].avail = privconn->cells[i].mem / pages[0].size;
351

352
        if (virCapabilitiesAddHostNUMACell(caps, i, privconn->cells[i].mem,
353
                                           privconn->cells[i].numCpus,
354
                                           cpu_cells, 0, NULL, nPages, pages) < 0)
355
            goto error;
356 357
    }

358
    for (i = 0; i < ARRAY_CARDINALITY(guest_types); i++) {
359 360
        if ((guest = virCapabilitiesAddGuest(caps,
                                             guest_types[i],
361
                                             VIR_ARCH_I686,
362 363 364 365
                                             TEST_EMULATOR,
                                             NULL,
                                             0,
                                             NULL)) == NULL)
366
            goto error;
367

368
        if (virCapabilitiesAddGuestDomain(guest,
369
                                          VIR_DOMAIN_VIRT_TEST,
370 371 372 373
                                          NULL,
                                          NULL,
                                          0,
                                          NULL) == NULL)
374
            goto error;
375

376
        if (virCapabilitiesAddGuestFeature(guest, "pae", true, true) == NULL)
377
            goto error;
378
        if (virCapabilitiesAddGuestFeature(guest, "nonpae", true, true) == NULL)
379
            goto error;
380 381
    }

382 383
    caps->host.nsecModels = 1;
    if (VIR_ALLOC_N(caps->host.secModels, caps->host.nsecModels) < 0)
384
        goto error;
385 386
    if (VIR_STRDUP(caps->host.secModels[0].model, "testSecurity") < 0)
        goto error;
387

388 389
    if (VIR_STRDUP(caps->host.secModels[0].doi, "") < 0)
        goto error;
390

391
    return caps;
392

393
 error:
394
    virObjectUnref(caps);
395
    return NULL;
396 397
}

398

399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416
static testDriverPtr
testDriverNew(void)
{
    virDomainXMLNamespace ns = {
        .parse = testDomainDefNamespaceParse,
        .free = testDomainDefNamespaceFree,
    };
    testDriverPtr ret;

    if (VIR_ALLOC(ret) < 0)
        return NULL;

    if (virMutexInit(&ret->lock) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("cannot initialize mutex"));
        goto error;
    }

417
    if (!(ret->xmlopt = virDomainXMLOptionNew(NULL, NULL, &ns, NULL, NULL)) ||
418
        !(ret->eventState = virObjectEventStateNew()) ||
419
        !(ret->ifaces = virInterfaceObjListNew()) ||
420
        !(ret->domains = virDomainObjListNew()) ||
421 422
        !(ret->networks = virNetworkObjListNew()) ||
        !(ret->devs = virNodeDeviceObjListNew()))
423 424
        goto error;

425
    virAtomicIntSet(&ret->nextDomID, 1);
426 427 428 429 430 431 432 433 434

    return ret;

 error:
    testDriverFree(ret);
    return NULL;
}


435 436
static const char *defaultConnXML =
"<node>"
437 438
"<domain type='test'>"
"  <name>test</name>"
439
"  <uuid>6695eb01-f6a4-8304-79aa-97f2502e193f</uuid>"
440 441 442 443 444 445
"  <memory>8388608</memory>"
"  <currentMemory>2097152</currentMemory>"
"  <vcpu>2</vcpu>"
"  <os>"
"    <type>hvm</type>"
"  </os>"
446 447
"</domain>"
""
448 449
"<network>"
"  <name>default</name>"
450
"  <uuid>dd8fe884-6c02-601e-7551-cca97df1c5df</uuid>"
451
"  <bridge name='virbr0'/>"
452 453 454
"  <forward/>"
"  <ip address='192.168.122.1' netmask='255.255.255.0'>"
"    <dhcp>"
455
"      <range start='192.168.122.2' end='192.168.122.254'/>"
456 457
"    </dhcp>"
"  </ip>"
458 459
"</network>"
""
L
Laine Stump 已提交
460 461 462 463 464 465 466 467
"<interface type=\"ethernet\" name=\"eth1\">"
"  <start mode=\"onboot\"/>"
"  <mac address=\"aa:bb:cc:dd:ee:ff\"/>"
"  <mtu size=\"1492\"/>"
"  <protocol family=\"ipv4\">"
"    <ip address=\"192.168.0.5\" prefix=\"24\"/>"
"    <route gateway=\"192.168.0.1\"/>"
"  </protocol>"
468 469
"</interface>"
""
C
Cole Robinson 已提交
470 471
"<pool type='dir'>"
"  <name>default-pool</name>"
472
"  <uuid>dfe224cb-28fb-8dd0-c4b2-64eb3f0f4566</uuid>"
C
Cole Robinson 已提交
473 474 475
"  <target>"
"    <path>/default-pool</path>"
"  </target>"
476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493
"</pool>"
""
"<device>"
"  <name>computer</name>"
"  <capability type='system'>"
"    <hardware>"
"      <vendor>Libvirt</vendor>"
"      <version>Test driver</version>"
"      <serial>123456</serial>"
"      <uuid>11111111-2222-3333-4444-555555555555</uuid>"
"    </hardware>"
"    <firmware>"
"      <vendor>Libvirt</vendor>"
"      <version>Test Driver</version>"
"      <release_date>01/22/2007</release_date>"
"    </firmware>"
"  </capability>"
"</device>"
494
"<device>"
495
"  <name>scsi_host1</name>"
496 497 498
"  <parent>computer</parent>"
"  <capability type='scsi_host'>"
"    <host>1</host>"
499
"    <unique_id>0</unique_id>"
500 501 502
"    <capability type='fc_host'>"
"      <wwnn>2000000012341234</wwnn>"
"      <wwpn>1000000012341234</wwpn>"
503 504 505 506
"      <fabric_wwn>2000000043214321</fabric_wwn>"
"    </capability>"
"    <capability type='vport_ops'>"
"      <max_vports>127</max_vports>"
507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523
"      <vports>1</vports>"
"    </capability>"
"  </capability>"
"</device>"
"<device>"
"  <name>scsi_host2</name>"
"  <parent>computer</parent>"
"  <capability type='scsi_host'>"
"    <host>2</host>"
"    <unique_id>1</unique_id>"
"    <capability type='fc_host'>"
"      <wwnn>2000000056785678</wwnn>"
"      <wwpn>1000000056785678</wwpn>"
"      <fabric_wwn>2000000087658765</fabric_wwn>"
"    </capability>"
"    <capability type='vport_ops'>"
"      <max_vports>127</max_vports>"
524
"      <vports>0</vports>"
525 526 527
"    </capability>"
"  </capability>"
"</device>"
528 529 530 531 532 533 534 535 536 537 538 539 540
"<device>"
"  <name>scsi_host11</name>"
"  <parent>scsi_host1</parent>"
"  <capability type='scsi_host'>"
"    <host>11</host>"
"    <unique_id>10</unique_id>"
"    <capability type='fc_host'>"
"      <wwnn>2000000034563456</wwnn>"
"      <wwpn>1000000034563456</wwpn>"
"      <fabric_wwn>2000000043214321</fabric_wwn>"
"    </capability>"
"  </capability>"
 "</device>"
541 542
"</node>";

C
Cole Robinson 已提交
543

544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566
static const char *defaultPoolSourcesLogicalXML =
"<sources>\n"
"  <source>\n"
"    <device path='/dev/sda20'/>\n"
"    <name>testvg1</name>\n"
"    <format type='lvm2'/>\n"
"  </source>\n"
"  <source>\n"
"    <device path='/dev/sda21'/>\n"
"    <name>testvg2</name>\n"
"    <format type='lvm2'/>\n"
"  </source>\n"
"</sources>\n";

static const char *defaultPoolSourcesNetFSXML =
"<sources>\n"
"  <source>\n"
"    <host name='%s'/>\n"
"    <dir path='/testshare'/>\n"
"    <format type='nfs'/>\n"
"  </source>\n"
"</sources>\n";

567
static const unsigned long long defaultPoolCap = (100 * 1024 * 1024 * 1024ull);
568
static const unsigned long long defaultPoolAlloc;
C
Cole Robinson 已提交
569

570
static int testStoragePoolObjSetDefaults(virStoragePoolObjPtr obj);
571
static int testNodeGetInfo(virConnectPtr conn, virNodeInfoPtr info);
572

573 574 575 576
static virDomainObjPtr
testDomObjFromDomain(virDomainPtr domain)
{
    virDomainObjPtr vm;
577
    testDriverPtr driver = domain->conn->privateData;
578 579
    char uuidstr[VIR_UUID_STRING_BUFLEN];

580
    vm = virDomainObjListFindByUUIDRef(driver->domains, domain->uuid);
581 582 583 584 585 586 587 588 589 590
    if (!vm) {
        virUUIDFormat(domain->uuid, uuidstr);
        virReportError(VIR_ERR_NO_DOMAIN,
                       _("no domain with matching uuid '%s' (%s)"),
                       uuidstr, domain->name);
    }

    return vm;
}

591
static char *
592 593
testDomainGenerateIfname(virDomainDefPtr domdef)
{
594
    int maxif = 1024;
595
    int ifctr;
596 597

    for (ifctr = 0; ifctr < maxif; ++ifctr) {
M
Michal Privoznik 已提交
598
        virDomainNetDefPtr net = NULL;
599 600
        char *ifname;

601
        if (virAsprintf(&ifname, "testnet%d", ifctr) < 0)
602 603 604
            return NULL;

        /* Generate network interface names */
M
Michal Privoznik 已提交
605
        if (!(net = virDomainNetFindByName(domdef, ifname)))
606
            return ifname;
607
        VIR_FREE(ifname);
608 609
    }

610 611
    virReportError(VIR_ERR_INTERNAL_ERROR,
                   _("Exceeded max iface limit %d"), maxif);
612 613 614
    return NULL;
}

615
static int
616
testDomainGenerateIfnames(virDomainDefPtr domdef)
617
{
618
    size_t i = 0;
619 620 621 622 623 624

    for (i = 0; i < domdef->nnets; i++) {
        char *ifname;
        if (domdef->nets[i]->ifname)
            continue;

625
        ifname = testDomainGenerateIfname(domdef);
626
        if (!ifname)
627
            return -1;
628 629 630 631

        domdef->nets[i]->ifname = ifname;
    }

632
    return 0;
633 634
}

635

636 637
static void
testDomainShutdownState(virDomainPtr domain,
J
Jiri Denemark 已提交
638 639
                        virDomainObjPtr privdom,
                        virDomainShutoffReason reason)
640
{
641
    virDomainObjRemoveTransientDef(privdom);
J
Jiri Denemark 已提交
642
    virDomainObjSetState(privdom, VIR_DOMAIN_SHUTOFF, reason);
643

644 645 646 647
    if (domain)
        domain->id = -1;
}

648
/* Set up domain runtime state */
649
static int
650
testDomainStartState(testDriverPtr privconn,
J
Jiri Denemark 已提交
651 652
                     virDomainObjPtr dom,
                     virDomainRunningReason reason)
653
{
654
    int ret = -1;
655

J
Jiri Denemark 已提交
656
    virDomainObjSetState(dom, VIR_DOMAIN_RUNNING, reason);
657
    dom->def->id = virAtomicIntAdd(&privconn->nextDomID, 1);
658

659
    if (virDomainObjSetDefTransient(privconn->caps,
660
                                    privconn->xmlopt,
661
                                    dom) < 0) {
662 663 664
        goto cleanup;
    }

C
Cole Robinson 已提交
665
    dom->hasManagedSave = false;
666
    ret = 0;
667
 cleanup:
668
    if (ret < 0)
J
Jiri Denemark 已提交
669
        testDomainShutdownState(NULL, dom, VIR_DOMAIN_SHUTOFF_FAILED);
670
    return ret;
671
}
672

673

674
static char *testBuildFilename(const char *relativeTo,
675 676
                               const char *filename)
{
677 678
    char *offset;
    int baseLen;
679 680
    char *ret;

681
    if (!filename || filename[0] == '\0')
682
        return NULL;
683 684 685 686
    if (filename[0] == '/') {
        ignore_value(VIR_STRDUP(ret, filename));
        return ret;
    }
687

688
    offset = strrchr(relativeTo, '/');
689
    if ((baseLen = (offset-relativeTo+1))) {
690
        char *absFile;
C
Chris Lalancette 已提交
691 692
        int totalLen = baseLen + strlen(filename) + 1;
        if (VIR_ALLOC_N(absFile, totalLen) < 0)
693
            return NULL;
C
Chris Lalancette 已提交
694 695 696 697
        if (virStrncpy(absFile, relativeTo, baseLen, totalLen) == NULL) {
            VIR_FREE(absFile);
            return NULL;
        }
698 699 700
        strcat(absFile, filename);
        return absFile;
    } else {
701 702
        ignore_value(VIR_STRDUP(ret, filename));
        return ret;
703
    }
704 705
}

C
Cole Robinson 已提交
706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736
static xmlNodePtr
testParseXMLDocFromFile(xmlNodePtr node, const char *file, const char *type)
{
    xmlNodePtr ret = NULL;
    xmlDocPtr doc = NULL;
    char *absFile = NULL;
    char *relFile = virXMLPropString(node, "file");

    if (relFile != NULL) {
        absFile = testBuildFilename(file, relFile);
        VIR_FREE(relFile);
        if (!absFile) {
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("resolving %s filename"), type);
            return NULL;
        }

        if (!(doc = virXMLParse(absFile, NULL, type)))
            goto error;

        ret = xmlCopyNode(xmlDocGetRootElement(doc), 1);
        if (!ret) {
            virReportOOMError();
            goto error;
        }
        xmlReplaceNode(node, ret);
        xmlFreeNode(node);
    } else {
        ret = node;
    }

737
 error:
C
Cole Robinson 已提交
738 739 740 741 742
    xmlFreeDoc(doc);
    VIR_FREE(absFile);
    return ret;
}

743 744 745
static int
testParseNodeInfo(virNodeInfoPtr nodeInfo, xmlXPathContextPtr ctxt)
{
746
    char *str;
747 748
    long l;
    int ret;
749

750
    ret = virXPathLong("string(/node/cpu/nodes[1])", ctxt, &l);
751 752 753
    if (ret == 0) {
        nodeInfo->nodes = l;
    } else if (ret == -2) {
754 755
        virReportError(VIR_ERR_XML_ERROR, "%s",
                       _("invalid node cpu nodes value"));
756
        goto error;
757
    }
758

759
    ret = virXPathLong("string(/node/cpu/sockets[1])", ctxt, &l);
760 761 762
    if (ret == 0) {
        nodeInfo->sockets = l;
    } else if (ret == -2) {
763 764
        virReportError(VIR_ERR_XML_ERROR, "%s",
                       _("invalid node cpu sockets value"));
765
        goto error;
766
    }
767

768
    ret = virXPathLong("string(/node/cpu/cores[1])", ctxt, &l);
769 770 771
    if (ret == 0) {
        nodeInfo->cores = l;
    } else if (ret == -2) {
772 773
        virReportError(VIR_ERR_XML_ERROR, "%s",
                       _("invalid node cpu cores value"));
774
        goto error;
775 776
    }

777
    ret = virXPathLong("string(/node/cpu/threads[1])", ctxt, &l);
778 779 780
    if (ret == 0) {
        nodeInfo->threads = l;
    } else if (ret == -2) {
781 782
        virReportError(VIR_ERR_XML_ERROR, "%s",
                       _("invalid node cpu threads value"));
783
        goto error;
784
    }
785

786 787
    nodeInfo->cpus = (nodeInfo->cores * nodeInfo->threads *
                      nodeInfo->sockets * nodeInfo->nodes);
788
    ret = virXPathLong("string(/node/cpu/active[1])", ctxt, &l);
789
    if (ret == 0) {
790
        if (l < nodeInfo->cpus)
791
            nodeInfo->cpus = l;
792
    } else if (ret == -2) {
793 794
        virReportError(VIR_ERR_XML_ERROR, "%s",
                       _("invalid node cpu active value"));
795
        goto error;
796
    }
797
    ret = virXPathLong("string(/node/cpu/mhz[1])", ctxt, &l);
798 799 800
    if (ret == 0) {
        nodeInfo->mhz = l;
    } else if (ret == -2) {
801 802
        virReportError(VIR_ERR_XML_ERROR, "%s",
                       _("invalid node cpu mhz value"));
803
        goto error;
804 805
    }

806
    str = virXPathString("string(/node/cpu/model[1])", ctxt);
807
    if (str != NULL) {
C
Chris Lalancette 已提交
808
        if (virStrcpyStatic(nodeInfo->model, str) == NULL) {
809 810
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Model %s too big for destination"), str);
C
Chris Lalancette 已提交
811 812 813
            VIR_FREE(str);
            goto error;
        }
814
        VIR_FREE(str);
815 816
    }

817
    ret = virXPathLong("string(/node/memory[1])", ctxt, &l);
818 819 820
    if (ret == 0) {
        nodeInfo->memory = l;
    } else if (ret == -2) {
821 822
        virReportError(VIR_ERR_XML_ERROR, "%s",
                       _("invalid node memory value"));
823
        goto error;
824
    }
825

826
    return 0;
827
 error:
828 829 830
    return -1;
}

831
static int
832
testParseDomainSnapshots(testDriverPtr privconn,
833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882
                         virDomainObjPtr domobj,
                         const char *file,
                         xmlXPathContextPtr ctxt)
{
    size_t i;
    int ret = -1;
    testDomainNamespaceDefPtr nsdata = domobj->def->namespaceData;
    xmlNodePtr *nodes = nsdata->snap_nodes;

    for (i = 0; i < nsdata->num_snap_nodes; i++) {
        virDomainSnapshotObjPtr snap;
        virDomainSnapshotDefPtr def;
        xmlNodePtr node = testParseXMLDocFromFile(nodes[i], file,
                                                  "domainsnapshot");
        if (!node)
            goto error;

        def = virDomainSnapshotDefParseNode(ctxt->doc, node,
                                            privconn->caps,
                                            privconn->xmlopt,
                                            VIR_DOMAIN_SNAPSHOT_PARSE_DISKS |
                                            VIR_DOMAIN_SNAPSHOT_PARSE_INTERNAL |
                                            VIR_DOMAIN_SNAPSHOT_PARSE_REDEFINE);
        if (!def)
            goto error;

        if (!(snap = virDomainSnapshotAssignDef(domobj->snapshots, def))) {
            virDomainSnapshotDefFree(def);
            goto error;
        }

        if (def->current) {
            if (domobj->current_snapshot) {
                virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                               _("more than one snapshot claims to be active"));
                goto error;
            }

            domobj->current_snapshot = snap;
        }
    }

    if (virDomainSnapshotUpdateRelations(domobj->snapshots) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Snapshots have inconsistent relations for "
                         "domain %s"), domobj->def->name);
        goto error;
    }

    ret = 0;
883
 error:
884 885 886
    return ret;
}

887
static int
888
testParseDomains(testDriverPtr privconn,
C
Cole Robinson 已提交
889 890
                 const char *file,
                 xmlXPathContextPtr ctxt)
891 892 893 894 895 896 897
{
    int num, ret = -1;
    size_t i;
    xmlNodePtr *nodes = NULL;
    virDomainObjPtr obj;

    num = virXPathNodeSet("/node/domain", ctxt, &nodes);
898
    if (num < 0)
899 900
        goto error;

901
    for (i = 0; i < num; i++) {
902
        virDomainDefPtr def;
903
        testDomainNamespaceDefPtr nsdata;
C
Cole Robinson 已提交
904 905 906 907 908
        xmlNodePtr node = testParseXMLDocFromFile(nodes[i], file, "domain");
        if (!node)
            goto error;

        def = virDomainDefParseNode(ctxt->doc, node,
909
                                    privconn->caps, privconn->xmlopt, NULL,
910
                                    VIR_DOMAIN_DEF_PARSE_INACTIVE);
C
Cole Robinson 已提交
911 912
        if (!def)
            goto error;
913

914
        if (testDomainGenerateIfnames(def) < 0 ||
915
            !(obj = virDomainObjListAdd(privconn->domains,
916
                                        def,
917
                                        privconn->xmlopt,
918
                                        0, NULL))) {
919
            virDomainDefFree(def);
920 921
            goto error;
        }
922

923 924 925 926 927
        if (testParseDomainSnapshots(privconn, obj, file, ctxt) < 0) {
            virObjectUnlock(obj);
            goto error;
        }

928
        nsdata = def->namespaceData;
929
        obj->persistent = !nsdata->transient;
C
Cole Robinson 已提交
930
        obj->hasManagedSave = nsdata->hasManagedSave;
931 932 933 934 935 936 937 938 939

        if (nsdata->runstate != VIR_DOMAIN_SHUTOFF) {
            if (testDomainStartState(privconn, obj,
                                     VIR_DOMAIN_RUNNING_BOOTED) < 0) {
                virObjectUnlock(obj);
                goto error;
            }
        } else {
            testDomainShutdownState(NULL, obj, 0);
940
        }
941
        virDomainObjSetState(obj, nsdata->runstate, 0);
942

943
        virObjectUnlock(obj);
944
    }
945

946
    ret = 0;
947
 error:
948 949 950 951
    VIR_FREE(nodes);
    return ret;
}

952

953
static int
954
testParseNetworks(testDriverPtr privconn,
C
Cole Robinson 已提交
955 956
                  const char *file,
                  xmlXPathContextPtr ctxt)
957 958 959 960 961 962 963
{
    int num, ret = -1;
    size_t i;
    xmlNodePtr *nodes = NULL;
    virNetworkObjPtr obj;

    num = virXPathNodeSet("/node/network", ctxt, &nodes);
964
    if (num < 0)
965
        goto error;
966 967

    for (i = 0; i < num; i++) {
968
        virNetworkDefPtr def;
C
Cole Robinson 已提交
969 970 971
        xmlNodePtr node = testParseXMLDocFromFile(nodes[i], file, "network");
        if (!node)
            goto error;
972

C
Cole Robinson 已提交
973 974 975
        def = virNetworkDefParseNode(ctxt->doc, node);
        if (!def)
            goto error;
976

977
        if (!(obj = virNetworkObjAssignDef(privconn->networks, def, 0))) {
978 979
            virNetworkDefFree(def);
            goto error;
980
        }
981

982
        virNetworkObjSetActive(obj, true);
983
        virNetworkObjEndAPI(&obj);
984
    }
985

986
    ret = 0;
987
 error:
988 989 990 991
    VIR_FREE(nodes);
    return ret;
}

992

993
static int
994
testParseInterfaces(testDriverPtr privconn,
C
Cole Robinson 已提交
995 996
                    const char *file,
                    xmlXPathContextPtr ctxt)
997 998 999 1000 1001 1002 1003
{
    int num, ret = -1;
    size_t i;
    xmlNodePtr *nodes = NULL;
    virInterfaceObjPtr obj;

    num = virXPathNodeSet("/node/interface", ctxt, &nodes);
1004
    if (num < 0)
L
Laine Stump 已提交
1005
        goto error;
1006 1007

    for (i = 0; i < num; i++) {
L
Laine Stump 已提交
1008
        virInterfaceDefPtr def;
C
Cole Robinson 已提交
1009 1010 1011 1012
        xmlNodePtr node = testParseXMLDocFromFile(nodes[i], file,
                                                   "interface");
        if (!node)
            goto error;
L
Laine Stump 已提交
1013

C
Cole Robinson 已提交
1014 1015 1016
        def = virInterfaceDefParseNode(ctxt->doc, node);
        if (!def)
            goto error;
1017

1018
        if (!(obj = virInterfaceObjListAssignDef(privconn->ifaces, def))) {
L
Laine Stump 已提交
1019 1020 1021
            virInterfaceDefFree(def);
            goto error;
        }
1022

1023
        virInterfaceObjSetActive(obj, true);
1024
        virInterfaceObjEndAPI(&obj);
1025 1026 1027
    }

    ret = 0;
1028
 error:
1029 1030 1031 1032
    VIR_FREE(nodes);
    return ret;
}

1033

1034
static int
C
Cole Robinson 已提交
1035
testOpenVolumesForPool(const char *file,
1036
                       xmlXPathContextPtr ctxt,
1037 1038
                       virStoragePoolObjPtr obj,
                       int objidx)
1039
{
1040
    virStoragePoolDefPtr def = virStoragePoolObjGetDef(obj);
1041 1042 1043 1044
    char *vol_xpath;
    size_t i;
    int num, ret = -1;
    xmlNodePtr *nodes = NULL;
1045
    virStorageVolDefPtr volDef = NULL;
1046 1047

    /* Find storage volumes */
1048
    if (virAsprintf(&vol_xpath, "/node/pool[%d]/volume", objidx) < 0)
1049 1050 1051 1052
        goto error;

    num = virXPathNodeSet(vol_xpath, ctxt, &nodes);
    VIR_FREE(vol_xpath);
1053
    if (num < 0)
1054 1055 1056
        goto error;

    for (i = 0; i < num; i++) {
C
Cole Robinson 已提交
1057 1058 1059 1060
        xmlNodePtr node = testParseXMLDocFromFile(nodes[i], file,
                                                   "volume");
        if (!node)
            goto error;
1061

1062
        if (!(volDef = virStorageVolDefParseNode(def, ctxt->doc, node, 0)))
C
Cole Robinson 已提交
1063
            goto error;
1064

1065 1066
        if (!volDef->target.path) {
            if (virAsprintf(&volDef->target.path, "%s/%s",
1067
                            def->target.path, volDef->name) < 0)
1068 1069 1070
                goto error;
        }

1071
        if (!volDef->key && VIR_STRDUP(volDef->key, volDef->target.path) < 0)
1072
            goto error;
1073

1074
        if (virStoragePoolObjAddVol(obj, volDef) < 0)
1075
            goto error;
1076

1077 1078
        def->allocation += volDef->target.allocation;
        def->available = (def->capacity - def->allocation);
1079
        volDef = NULL;
L
Laine Stump 已提交
1080 1081
    }

1082
    ret = 0;
1083
 error:
1084
    virStorageVolDefFree(volDef);
1085 1086 1087 1088
    VIR_FREE(nodes);
    return ret;
}

1089

1090
static int
1091
testParseStorage(testDriverPtr privconn,
C
Cole Robinson 已提交
1092 1093
                 const char *file,
                 xmlXPathContextPtr ctxt)
1094 1095 1096 1097 1098 1099 1100
{
    int num, ret = -1;
    size_t i;
    xmlNodePtr *nodes = NULL;
    virStoragePoolObjPtr obj;

    num = virXPathNodeSet("/node/pool", ctxt, &nodes);
1101
    if (num < 0)
C
Cole Robinson 已提交
1102
        goto error;
1103 1104

    for (i = 0; i < num; i++) {
C
Cole Robinson 已提交
1105
        virStoragePoolDefPtr def;
C
Cole Robinson 已提交
1106 1107 1108 1109
        xmlNodePtr node = testParseXMLDocFromFile(nodes[i], file,
                                                   "pool");
        if (!node)
            goto error;
C
Cole Robinson 已提交
1110

C
Cole Robinson 已提交
1111 1112 1113
        def = virStoragePoolDefParseNode(ctxt->doc, node);
        if (!def)
            goto error;
C
Cole Robinson 已提交
1114

1115
        if (!(obj = virStoragePoolObjAssignDef(&privconn->pools,
C
Cole Robinson 已提交
1116 1117 1118 1119 1120
                                                def))) {
            virStoragePoolDefFree(def);
            goto error;
        }

1121 1122
        if (testStoragePoolObjSetDefaults(obj) == -1) {
            virStoragePoolObjUnlock(obj);
C
Cole Robinson 已提交
1123
            goto error;
1124
        }
1125
        virStoragePoolObjSetActive(obj, true);
1126 1127

        /* Find storage volumes */
C
Cole Robinson 已提交
1128
        if (testOpenVolumesForPool(file, ctxt, obj, i+1) < 0) {
1129
            virStoragePoolObjUnlock(obj);
1130 1131 1132
            goto error;
        }

1133
        virStoragePoolObjUnlock(obj);
C
Cole Robinson 已提交
1134 1135
    }

1136
    ret = 0;
1137
 error:
1138 1139 1140 1141
    VIR_FREE(nodes);
    return ret;
}

1142

1143
static int
1144
testParseNodedevs(testDriverPtr privconn,
C
Cole Robinson 已提交
1145 1146
                  const char *file,
                  xmlXPathContextPtr ctxt)
1147 1148 1149 1150 1151 1152 1153
{
    int num, ret = -1;
    size_t i;
    xmlNodePtr *nodes = NULL;
    virNodeDeviceObjPtr obj;

    num = virXPathNodeSet("/node/device", ctxt, &nodes);
1154
    if (num < 0)
1155
        goto error;
1156 1157

    for (i = 0; i < num; i++) {
1158
        virNodeDeviceDefPtr def;
C
Cole Robinson 已提交
1159 1160 1161 1162
        xmlNodePtr node = testParseXMLDocFromFile(nodes[i], file,
                                                  "nodedev");
        if (!node)
            goto error;
1163

C
Cole Robinson 已提交
1164 1165 1166
        def = virNodeDeviceDefParseNode(ctxt->doc, node, 0, NULL);
        if (!def)
            goto error;
1167

1168
        if (!(obj = virNodeDeviceObjListAssignDef(privconn->devs, def))) {
1169 1170 1171
            virNodeDeviceDefFree(def);
            goto error;
        }
1172

1173
        virNodeDeviceObjEndAPI(&obj);
1174 1175 1176
    }

    ret = 0;
1177
 error:
1178 1179 1180 1181
    VIR_FREE(nodes);
    return ret;
}

1182
static int
1183
testParseAuthUsers(testDriverPtr privconn,
1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216
                   xmlXPathContextPtr ctxt)
{
    int num, ret = -1;
    size_t i;
    xmlNodePtr *nodes = NULL;

    num = virXPathNodeSet("/node/auth/user", ctxt, &nodes);
    if (num < 0)
        goto error;

    privconn->numAuths = num;
    if (num && VIR_ALLOC_N(privconn->auths, num) < 0)
        goto error;

    for (i = 0; i < num; i++) {
        char *username, *password;

        ctxt->node = nodes[i];
        username = virXPathString("string(.)", ctxt);
        if (!username || STREQ(username, "")) {
            virReportError(VIR_ERR_XML_ERROR, "%s",
                           _("missing username in /node/auth/user field"));
            VIR_FREE(username);
            goto error;
        }
        /* This field is optional. */
        password = virXMLPropString(nodes[i], "password");

        privconn->auths[i].username = username;
        privconn->auths[i].password = password;
    }

    ret = 0;
1217
 error:
1218 1219 1220
    VIR_FREE(nodes);
    return ret;
}
1221

C
Cole Robinson 已提交
1222 1223 1224 1225 1226
static int
testOpenParse(testDriverPtr privconn,
              const char *file,
              xmlXPathContextPtr ctxt)
{
1227
    if (!virXMLNodeNameEqual(ctxt->node, "node")) {
C
Cole Robinson 已提交
1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252
        virReportError(VIR_ERR_XML_ERROR, "%s",
                       _("Root element is not 'node'"));
        goto error;
    }

    if (testParseNodeInfo(&privconn->nodeInfo, ctxt) < 0)
        goto error;
    if (testParseDomains(privconn, file, ctxt) < 0)
        goto error;
    if (testParseNetworks(privconn, file, ctxt) < 0)
        goto error;
    if (testParseInterfaces(privconn, file, ctxt) < 0)
        goto error;
    if (testParseStorage(privconn, file, ctxt) < 0)
        goto error;
    if (testParseNodedevs(privconn, file, ctxt) < 0)
        goto error;
    if (testParseAuthUsers(privconn, ctxt) < 0)
        goto error;

    return 0;
 error:
    return -1;
}

1253 1254
/* No shared state between simultaneous test connections initialized
 * from a file.  */
1255 1256 1257 1258 1259
static int
testOpenFromFile(virConnectPtr conn, const char *file)
{
    xmlDocPtr doc = NULL;
    xmlXPathContextPtr ctxt = NULL;
1260
    testDriverPtr privconn;
1261

1262
    if (!(privconn = testDriverNew()))
1263
        return VIR_DRV_OPEN_ERROR;
1264

1265 1266 1267 1268 1269 1270
    testDriverLock(privconn);
    conn->privateData = privconn;

    if (!(privconn->caps = testBuildCapabilities(conn)))
        goto error;

1271
    if (!(doc = virXMLParseFileCtxt(file, &ctxt)))
1272 1273 1274 1275 1276
        goto error;

    privconn->numCells = 0;
    memmove(&privconn->nodeInfo, &defaultNodeInfo, sizeof(defaultNodeInfo));

C
Cole Robinson 已提交
1277
    if (testOpenParse(privconn, file, ctxt) < 0)
1278
        goto error;
1279

J
Jim Meyering 已提交
1280
    xmlXPathFreeContext(ctxt);
1281
    xmlFreeDoc(doc);
1282
    testDriverUnlock(privconn);
1283

1284
    return 0;
1285 1286

 error:
J
Jim Meyering 已提交
1287
    xmlXPathFreeContext(ctxt);
1288
    xmlFreeDoc(doc);
1289
    testDriverFree(privconn);
1290
    conn->privateData = NULL;
1291
    return VIR_DRV_OPEN_ERROR;
1292 1293
}

1294 1295 1296 1297 1298 1299 1300
/* Simultaneous test:///default connections should share the same
 * common state (among other things, this allows testing event
 * detection in one connection for an action caused in another).  */
static int
testOpenDefault(virConnectPtr conn)
{
    testDriverPtr privconn = NULL;
1301 1302
    xmlDocPtr doc = NULL;
    xmlXPathContextPtr ctxt = NULL;
1303
    size_t i;
1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320

    virMutexLock(&defaultLock);
    if (defaultConnections++) {
        conn->privateData = defaultConn;
        virMutexUnlock(&defaultLock);
        return VIR_DRV_OPEN_SUCCESS;
    }

    if (!(privconn = testDriverNew()))
        goto error;

    conn->privateData = privconn;

    memmove(&privconn->nodeInfo, &defaultNodeInfo, sizeof(defaultNodeInfo));

    /* Numa setup */
    privconn->numCells = 2;
1321 1322 1323 1324
    for (i = 0; i < privconn->numCells; i++) {
        privconn->cells[i].numCpus = 8;
        privconn->cells[i].mem = (i + 1) * 2048 * 1024;
        privconn->cells[i].freeMem = (i + 1) * 1024 * 1024;
1325
    }
1326
    for (i = 0; i < 16; i++) {
1327 1328 1329
        virBitmapPtr siblings = virBitmapNew(16);
        if (!siblings)
            goto error;
1330 1331 1332 1333 1334
        ignore_value(virBitmapSetBit(siblings, i));
        privconn->cells[i / 8].cpus[(i % 8)].id = i;
        privconn->cells[i / 8].cpus[(i % 8)].socket_id = i / 8;
        privconn->cells[i / 8].cpus[(i % 8)].core_id = i % 8;
        privconn->cells[i / 8].cpus[(i % 8)].siblings = siblings;
1335 1336 1337 1338 1339
    }

    if (!(privconn->caps = testBuildCapabilities(conn)))
        goto error;

1340 1341
    if (!(doc = virXMLParseStringCtxt(defaultConnXML,
                                      _("(test driver)"), &ctxt)))
1342 1343
        goto error;

1344
    if (testOpenParse(privconn, NULL, ctxt) < 0)
1345 1346 1347 1348
        goto error;

    defaultConn = privconn;

1349 1350
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(doc);
1351 1352 1353 1354 1355 1356
    virMutexUnlock(&defaultLock);

    return VIR_DRV_OPEN_SUCCESS;

 error:
    testDriverFree(privconn);
1357 1358
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(doc);
1359 1360 1361 1362 1363 1364
    conn->privateData = NULL;
    defaultConnections--;
    virMutexUnlock(&defaultLock);
    return VIR_DRV_OPEN_ERROR;
}

1365 1366 1367 1368
static int
testConnectAuthenticate(virConnectPtr conn,
                        virConnectAuthPtr auth)
{
1369
    testDriverPtr privconn = conn->privateData;
1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393
    int ret = -1;
    ssize_t i;
    char *username = NULL, *password = NULL;

    if (privconn->numAuths == 0)
        return 0;

    /* Authentication is required because the test XML contains a
     * non-empty <auth/> section.  First we must ask for a username.
     */
    username = virAuthGetUsername(conn, auth, "test", NULL, "localhost"/*?*/);
    if (!username) {
        virReportError(VIR_ERR_AUTH_FAILED, "%s",
                       _("authentication failed when asking for username"));
        goto cleanup;
    }

    /* Does the username exist? */
    for (i = 0; i < privconn->numAuths; ++i) {
        if (STREQ(privconn->auths[i].username, username))
            goto found_user;
    }
    i = -1;

1394
 found_user:
1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413
    /* Even if we didn't find the user, we still ask for a password. */
    if (i == -1 || privconn->auths[i].password != NULL) {
        password = virAuthGetPassword(conn, auth, "test",
                                      username, "localhost");
        if (password == NULL) {
            virReportError(VIR_ERR_AUTH_FAILED, "%s",
                           _("authentication failed when asking for password"));
            goto cleanup;
        }
    }

    if (i == -1 ||
        (password && STRNEQ(privconn->auths[i].password, password))) {
        virReportError(VIR_ERR_AUTH_FAILED, "%s",
                       _("authentication failed, see test XML for the correct username/password"));
        goto cleanup;
    }

    ret = 0;
1414
 cleanup:
1415 1416 1417 1418
    VIR_FREE(username);
    VIR_FREE(password);
    return ret;
}
1419

1420
static virDrvOpenStatus testConnectOpen(virConnectPtr conn,
1421
                                        virConnectAuthPtr auth,
1422
                                        virConfPtr conf ATTRIBUTE_UNUSED,
1423
                                        unsigned int flags)
1424
{
1425
    int ret;
1426

E
Eric Blake 已提交
1427 1428
    virCheckFlags(VIR_CONNECT_RO, VIR_DRV_OPEN_ERROR);

1429
    if (!conn->uri)
1430
        return VIR_DRV_OPEN_DECLINED;
1431

1432
    if (!conn->uri->scheme || STRNEQ(conn->uri->scheme, "test"))
1433
        return VIR_DRV_OPEN_DECLINED;
1434

1435
    /* Remote driver should handle these. */
1436
    if (conn->uri->server)
1437 1438
        return VIR_DRV_OPEN_DECLINED;

1439
    /* From this point on, the connection is for us. */
1440 1441 1442
    if (!conn->uri->path
        || conn->uri->path[0] == '\0'
        || (conn->uri->path[0] == '/' && conn->uri->path[1] == '\0')) {
1443 1444
        virReportError(VIR_ERR_INVALID_ARG,
                       "%s", _("testOpen: supply a path or use test:///default"));
1445 1446
        return VIR_DRV_OPEN_ERROR;
    }
1447

1448
    if (STREQ(conn->uri->path, "/default"))
1449 1450
        ret = testOpenDefault(conn);
    else
1451
        ret = testOpenFromFile(conn,
1452
                               conn->uri->path);
1453

1454 1455 1456
    if (ret != VIR_DRV_OPEN_SUCCESS)
        return ret;

1457 1458 1459 1460
    /* Fake authentication. */
    if (testConnectAuthenticate(conn, auth) < 0)
        return VIR_DRV_OPEN_ERROR;

1461
    return VIR_DRV_OPEN_SUCCESS;
1462 1463
}

1464
static int testConnectClose(virConnectPtr conn)
1465
{
1466
    testDriverPtr privconn = conn->privateData;
1467
    bool dflt = false;
1468

1469 1470
    if (privconn == defaultConn) {
        dflt = true;
1471 1472 1473 1474 1475 1476 1477
        virMutexLock(&defaultLock);
        if (--defaultConnections) {
            virMutexUnlock(&defaultLock);
            return 0;
        }
    }

1478
    testDriverLock(privconn);
1479
    testDriverFree(privconn);
1480 1481 1482

    if (dflt) {
        defaultConn = NULL;
1483
        virMutexUnlock(&defaultLock);
1484 1485
    }

1486
    conn->privateData = NULL;
1487
    return 0;
1488 1489
}

1490 1491
static int testConnectGetVersion(virConnectPtr conn ATTRIBUTE_UNUSED,
                                 unsigned long *hvVer)
1492
{
1493
    *hvVer = 2;
1494
    return 0;
1495 1496
}

1497 1498 1499 1500 1501 1502
static char *testConnectGetHostname(virConnectPtr conn ATTRIBUTE_UNUSED)
{
    return virGetHostname();
}


1503
static int testConnectIsSecure(virConnectPtr conn ATTRIBUTE_UNUSED)
1504 1505 1506 1507
{
    return 1;
}

1508
static int testConnectIsEncrypted(virConnectPtr conn ATTRIBUTE_UNUSED)
1509 1510 1511 1512
{
    return 0;
}

1513
static int testConnectIsAlive(virConnectPtr conn ATTRIBUTE_UNUSED)
1514 1515 1516 1517
{
    return 1;
}

1518 1519
static int testConnectGetMaxVcpus(virConnectPtr conn ATTRIBUTE_UNUSED,
                                  const char *type ATTRIBUTE_UNUSED)
1520 1521 1522 1523
{
    return 32;
}

1524 1525 1526 1527 1528 1529
static char *
testConnectBaselineCPU(virConnectPtr conn ATTRIBUTE_UNUSED,
                       const char **xmlCPUs,
                       unsigned int ncpus,
                       unsigned int flags)
{
J
Jiri Denemark 已提交
1530 1531 1532
    virCPUDefPtr *cpus = NULL;
    virCPUDefPtr cpu = NULL;
    char *cpustr = NULL;
1533 1534 1535

    virCheckFlags(VIR_CONNECT_BASELINE_CPU_EXPAND_FEATURES, NULL);

J
Jiri Denemark 已提交
1536 1537 1538 1539 1540 1541 1542 1543 1544 1545
    if (!(cpus = virCPUDefListParse(xmlCPUs, ncpus, VIR_CPU_TYPE_HOST)))
        goto cleanup;

    if (!(cpu = cpuBaseline(cpus, ncpus, NULL, 0, false)))
        goto cleanup;

    if ((flags & VIR_CONNECT_BASELINE_CPU_EXPAND_FEATURES) &&
        virCPUExpandFeatures(cpus[0]->arch, cpu) < 0)
        goto cleanup;

1546
    cpustr = virCPUDefFormat(cpu, NULL);
J
Jiri Denemark 已提交
1547 1548 1549 1550

 cleanup:
    virCPUDefListFree(cpus);
    virCPUDefFree(cpu);
1551

J
Jiri Denemark 已提交
1552
    return cpustr;
1553 1554
}

1555 1556
static int testNodeGetInfo(virConnectPtr conn,
                           virNodeInfoPtr info)
1557
{
1558
    testDriverPtr privconn = conn->privateData;
1559
    testDriverLock(privconn);
1560
    memcpy(info, &privconn->nodeInfo, sizeof(virNodeInfo));
1561
    testDriverUnlock(privconn);
1562
    return 0;
1563 1564
}

1565
static char *testConnectGetCapabilities(virConnectPtr conn)
1566
{
1567
    testDriverPtr privconn = conn->privateData;
1568
    char *xml;
1569
    testDriverLock(privconn);
1570
    xml = virCapabilitiesFormatXML(privconn->caps);
1571
    testDriverUnlock(privconn);
1572
    return xml;
1573 1574
}

1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594
static char *
testConnectGetSysinfo(virConnectPtr conn ATTRIBUTE_UNUSED,
                      unsigned int flags)
{
    char *ret;
    const char *sysinfo = "<sysinfo type='smbios'>\n"
           "  <bios>\n"
           "    <entry name='vendor'>LENOVO</entry>\n"
           "    <entry name='version'>G4ETA1WW (2.61 )</entry>\n"
           "    <entry name='date'>05/07/2014</entry>\n"
           "    <entry name='release'>2.61</entry>\n"
           "  </bios>\n"
           "</sysinfo>\n";

    virCheckFlags(0, NULL);

    ignore_value(VIR_STRDUP(ret, sysinfo));
    return ret;
}

1595 1596 1597 1598 1599 1600
static const char *
testConnectGetType(virConnectPtr conn ATTRIBUTE_UNUSED)
{
    return "TEST";
}

1601
static int testConnectNumOfDomains(virConnectPtr conn)
1602
{
1603
    testDriverPtr privconn = conn->privateData;
1604
    int count;
1605

1606
    testDriverLock(privconn);
1607
    count = virDomainObjListNumOfDomains(privconn->domains, true, NULL, NULL);
1608
    testDriverUnlock(privconn);
1609

1610
    return count;
1611 1612
}

1613 1614 1615
static int testDomainIsActive(virDomainPtr dom)
{
    virDomainObjPtr obj;
1616
    int ret;
1617

1618 1619
    if (!(obj = testDomObjFromDomain(dom)))
        return -1;
1620

1621 1622
    ret = virDomainObjIsActive(obj);
    virDomainObjEndAPI(&obj);
1623 1624 1625 1626 1627 1628
    return ret;
}

static int testDomainIsPersistent(virDomainPtr dom)
{
    virDomainObjPtr obj;
1629 1630 1631 1632
    int ret;

    if (!(obj = testDomObjFromDomain(dom)))
        return -1;
1633 1634 1635

    ret = obj->persistent;

1636
    virDomainObjEndAPI(&obj);
1637 1638 1639
    return ret;
}

1640 1641 1642 1643 1644
static int testDomainIsUpdated(virDomainPtr dom ATTRIBUTE_UNUSED)
{
    return 0;
}

1645
static virDomainPtr
1646
testDomainCreateXML(virConnectPtr conn, const char *xml,
1647
                      unsigned int flags)
1648
{
1649
    testDriverPtr privconn = conn->privateData;
1650
    virDomainPtr ret = NULL;
1651
    virDomainDefPtr def;
1652
    virDomainObjPtr dom = NULL;
1653
    virObjectEventPtr event = NULL;
1654
    unsigned int parse_flags = VIR_DOMAIN_DEF_PARSE_INACTIVE;
1655

1656 1657 1658
    virCheckFlags(VIR_DOMAIN_START_VALIDATE, NULL);

    if (flags & VIR_DOMAIN_START_VALIDATE)
1659
        parse_flags |= VIR_DOMAIN_DEF_PARSE_VALIDATE_SCHEMA;
1660

1661
    testDriverLock(privconn);
1662
    if ((def = virDomainDefParseString(xml, privconn->caps, privconn->xmlopt,
1663
                                       NULL, parse_flags)) == NULL)
1664
        goto cleanup;
1665

1666
    if (testDomainGenerateIfnames(def) < 0)
1667
        goto cleanup;
1668
    if (!(dom = virDomainObjListAdd(privconn->domains,
1669
                                    def,
1670
                                    privconn->xmlopt,
1671
                                    VIR_DOMAIN_OBJ_LIST_ADD_LIVE |
1672 1673
                                    VIR_DOMAIN_OBJ_LIST_ADD_CHECK_LIVE,
                                    NULL)))
1674 1675
        goto cleanup;
    def = NULL;
1676

1677 1678 1679 1680 1681
    if (testDomainStartState(privconn, dom, VIR_DOMAIN_RUNNING_BOOTED) < 0) {
        if (!dom->persistent) {
            virDomainObjListRemove(privconn->domains, dom);
            dom = NULL;
        }
1682
        goto cleanup;
1683
    }
1684

1685
    event = virDomainEventLifecycleNewFromObj(dom,
1686 1687 1688
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_BOOTED);

1689
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid, dom->def->id);
1690

1691
 cleanup:
1692
    if (dom)
1693
        virObjectUnlock(dom);
1694
    testObjectEventQueue(privconn, event);
1695
    virDomainDefFree(def);
1696
    testDriverUnlock(privconn);
1697
    return ret;
1698 1699 1700
}


1701
static virDomainPtr testDomainLookupByID(virConnectPtr conn,
1702
                                         int id)
1703
{
1704
    testDriverPtr privconn = conn->privateData;
1705 1706
    virDomainPtr ret = NULL;
    virDomainObjPtr dom;
1707

1708
    if (!(dom = virDomainObjListFindByID(privconn->domains, id))) {
1709
        virReportError(VIR_ERR_NO_DOMAIN, NULL);
1710
        goto cleanup;
1711 1712
    }

1713
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid, dom->def->id);
1714

1715
 cleanup:
1716
    if (dom)
1717
        virObjectUnlock(dom);
1718
    return ret;
1719 1720
}

1721
static virDomainPtr testDomainLookupByUUID(virConnectPtr conn,
1722
                                           const unsigned char *uuid)
1723
{
1724
    testDriverPtr privconn = conn->privateData;
1725
    virDomainPtr ret = NULL;
1726
    virDomainObjPtr dom;
1727

1728
    if (!(dom = virDomainObjListFindByUUID(privconn->domains, uuid))) {
1729
        virReportError(VIR_ERR_NO_DOMAIN, NULL);
1730
        goto cleanup;
1731
    }
1732

1733
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid, dom->def->id);
1734

1735
 cleanup:
1736
    if (dom)
1737
        virObjectUnlock(dom);
1738
    return ret;
1739 1740
}

1741
static virDomainPtr testDomainLookupByName(virConnectPtr conn,
1742
                                           const char *name)
1743
{
1744
    testDriverPtr privconn = conn->privateData;
1745 1746
    virDomainPtr ret = NULL;
    virDomainObjPtr dom;
1747

1748
    if (!(dom = virDomainObjListFindByName(privconn->domains, name))) {
1749
        virReportError(VIR_ERR_NO_DOMAIN, NULL);
1750
        goto cleanup;
1751
    }
1752

1753
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid, dom->def->id);
1754

1755
 cleanup:
1756
    virDomainObjEndAPI(&dom);
1757
    return ret;
1758 1759
}

1760 1761 1762
static int testConnectListDomains(virConnectPtr conn,
                                  int *ids,
                                  int maxids)
1763
{
1764
    testDriverPtr privconn = conn->privateData;
1765

1766 1767
    return virDomainObjListGetActiveIDs(privconn->domains, ids, maxids,
                                        NULL, NULL);
1768 1769
}

1770
static int testDomainDestroy(virDomainPtr domain)
1771
{
1772
    testDriverPtr privconn = domain->conn->privateData;
1773
    virDomainObjPtr privdom;
1774
    virObjectEventPtr event = NULL;
1775
    int ret = -1;
1776

1777
    if (!(privdom = testDomObjFromDomain(domain)))
1778
        goto cleanup;
1779

1780 1781 1782 1783 1784 1785
    if (!virDomainObjIsActive(privdom)) {
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s", _("domain is not running"));
        goto cleanup;
    }

J
Jiri Denemark 已提交
1786
    testDomainShutdownState(domain, privdom, VIR_DOMAIN_SHUTOFF_DESTROYED);
1787
    event = virDomainEventLifecycleNewFromObj(privdom,
1788 1789
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_DESTROYED);
1790

1791 1792
    if (!privdom->persistent)
        virDomainObjListRemove(privconn->domains, privdom);
1793 1794

    ret = 0;
1795
 cleanup:
1796
    virDomainObjEndAPI(&privdom);
1797
    testObjectEventQueue(privconn, event);
1798
    return ret;
1799 1800
}

1801
static int testDomainResume(virDomainPtr domain)
1802
{
1803
    testDriverPtr privconn = domain->conn->privateData;
1804
    virDomainObjPtr privdom;
1805
    virObjectEventPtr event = NULL;
1806
    int ret = -1;
1807

1808 1809
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
1810

J
Jiri Denemark 已提交
1811
    if (virDomainObjGetState(privdom, NULL) != VIR_DOMAIN_PAUSED) {
1812 1813
        virReportError(VIR_ERR_INTERNAL_ERROR, _("domain '%s' not paused"),
                       domain->name);
1814
        goto cleanup;
1815
    }
1816

J
Jiri Denemark 已提交
1817 1818
    virDomainObjSetState(privdom, VIR_DOMAIN_RUNNING,
                         VIR_DOMAIN_RUNNING_UNPAUSED);
1819
    event = virDomainEventLifecycleNewFromObj(privdom,
1820 1821
                                     VIR_DOMAIN_EVENT_RESUMED,
                                     VIR_DOMAIN_EVENT_RESUMED_UNPAUSED);
1822 1823
    ret = 0;

1824
 cleanup:
1825
    virDomainObjEndAPI(&privdom);
1826
    testObjectEventQueue(privconn, event);
1827
    return ret;
1828 1829
}

1830
static int testDomainSuspend(virDomainPtr domain)
1831
{
1832
    testDriverPtr privconn = domain->conn->privateData;
1833
    virDomainObjPtr privdom;
1834
    virObjectEventPtr event = NULL;
1835
    int ret = -1;
J
Jiri Denemark 已提交
1836
    int state;
1837

1838 1839
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
1840

J
Jiri Denemark 已提交
1841 1842
    state = virDomainObjGetState(privdom, NULL);
    if (state == VIR_DOMAIN_SHUTOFF || state == VIR_DOMAIN_PAUSED) {
1843 1844
        virReportError(VIR_ERR_INTERNAL_ERROR, _("domain '%s' not running"),
                       domain->name);
1845
        goto cleanup;
1846
    }
1847

J
Jiri Denemark 已提交
1848
    virDomainObjSetState(privdom, VIR_DOMAIN_PAUSED, VIR_DOMAIN_PAUSED_USER);
1849
    event = virDomainEventLifecycleNewFromObj(privdom,
1850 1851
                                     VIR_DOMAIN_EVENT_SUSPENDED,
                                     VIR_DOMAIN_EVENT_SUSPENDED_PAUSED);
1852 1853
    ret = 0;

1854
 cleanup:
1855
    virDomainObjEndAPI(&privdom);
1856
    testObjectEventQueue(privconn, event);
1857
    return ret;
1858 1859
}

1860
static int testDomainShutdownFlags(virDomainPtr domain,
1861
                                   unsigned int flags)
1862
{
1863
    testDriverPtr privconn = domain->conn->privateData;
1864
    virDomainObjPtr privdom;
1865
    virObjectEventPtr event = NULL;
1866
    int ret = -1;
1867

1868 1869
    virCheckFlags(0, -1);

1870

1871
    if (!(privdom = testDomObjFromDomain(domain)))
1872
        goto cleanup;
1873

J
Jiri Denemark 已提交
1874
    if (virDomainObjGetState(privdom, NULL) == VIR_DOMAIN_SHUTOFF) {
1875 1876
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("domain '%s' not running"), domain->name);
1877
        goto cleanup;
1878
    }
1879

J
Jiri Denemark 已提交
1880
    testDomainShutdownState(domain, privdom, VIR_DOMAIN_SHUTOFF_SHUTDOWN);
1881
    event = virDomainEventLifecycleNewFromObj(privdom,
1882 1883
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_SHUTDOWN);
1884

1885 1886
    if (!privdom->persistent)
        virDomainObjListRemove(privconn->domains, privdom);
1887

1888
    ret = 0;
1889
 cleanup:
1890
    virDomainObjEndAPI(&privdom);
1891
    testObjectEventQueue(privconn, event);
1892
    return ret;
1893 1894
}

1895
static int testDomainShutdown(virDomainPtr domain)
1896
{
1897
    return testDomainShutdownFlags(domain, 0);
1898 1899
}

1900
/* Similar behaviour as shutdown */
1901
static int testDomainReboot(virDomainPtr domain,
1902
                            unsigned int action ATTRIBUTE_UNUSED)
1903
{
1904
    testDriverPtr privconn = domain->conn->privateData;
1905
    virDomainObjPtr privdom;
1906
    virObjectEventPtr event = NULL;
1907
    int ret = -1;
1908 1909


1910
    if (!(privdom = testDomObjFromDomain(domain)))
1911
        goto cleanup;
1912

1913 1914 1915 1916 1917 1918
    if (!virDomainObjIsActive(privdom)) {
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s", _("domain is not running"));
        goto cleanup;
    }

J
Jiri Denemark 已提交
1919 1920 1921
    virDomainObjSetState(privdom, VIR_DOMAIN_SHUTDOWN,
                         VIR_DOMAIN_SHUTDOWN_USER);

1922 1923
    switch (privdom->def->onReboot) {
    case VIR_DOMAIN_LIFECYCLE_DESTROY:
J
Jiri Denemark 已提交
1924 1925
        virDomainObjSetState(privdom, VIR_DOMAIN_SHUTOFF,
                             VIR_DOMAIN_SHUTOFF_SHUTDOWN);
1926 1927
        break;

1928
    case VIR_DOMAIN_LIFECYCLE_RESTART:
J
Jiri Denemark 已提交
1929 1930
        virDomainObjSetState(privdom, VIR_DOMAIN_RUNNING,
                             VIR_DOMAIN_RUNNING_BOOTED);
1931 1932
        break;

1933
    case VIR_DOMAIN_LIFECYCLE_PRESERVE:
J
Jiri Denemark 已提交
1934 1935
        virDomainObjSetState(privdom, VIR_DOMAIN_SHUTOFF,
                             VIR_DOMAIN_SHUTOFF_SHUTDOWN);
1936 1937
        break;

1938
    case VIR_DOMAIN_LIFECYCLE_RESTART_RENAME:
J
Jiri Denemark 已提交
1939 1940
        virDomainObjSetState(privdom, VIR_DOMAIN_RUNNING,
                             VIR_DOMAIN_RUNNING_BOOTED);
1941
        break;
1942

1943
    default:
J
Jiri Denemark 已提交
1944 1945
        virDomainObjSetState(privdom, VIR_DOMAIN_SHUTOFF,
                             VIR_DOMAIN_SHUTOFF_SHUTDOWN);
1946 1947
        break;
    }
1948

J
Jiri Denemark 已提交
1949 1950
    if (virDomainObjGetState(privdom, NULL) == VIR_DOMAIN_SHUTOFF) {
        testDomainShutdownState(domain, privdom, VIR_DOMAIN_SHUTOFF_SHUTDOWN);
1951
        event = virDomainEventLifecycleNewFromObj(privdom,
1952 1953
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_SHUTDOWN);
1954

1955 1956
        if (!privdom->persistent)
            virDomainObjListRemove(privconn->domains, privdom);
1957 1958
    }

1959
    ret = 0;
1960
 cleanup:
1961
    virDomainObjEndAPI(&privdom);
1962
    testObjectEventQueue(privconn, event);
1963
    return ret;
1964 1965
}

1966
static int testDomainGetInfo(virDomainPtr domain,
1967
                             virDomainInfoPtr info)
1968
{
1969
    struct timeval tv;
1970
    virDomainObjPtr privdom;
1971
    int ret = -1;
1972

1973 1974
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
1975 1976

    if (gettimeofday(&tv, NULL) < 0) {
1977 1978
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("getting time of day"));
1979
        goto cleanup;
1980 1981
    }

J
Jiri Denemark 已提交
1982
    info->state = virDomainObjGetState(privdom, NULL);
1983
    info->memory = privdom->def->mem.cur_balloon;
1984
    info->maxMem = virDomainDefGetMemoryTotal(privdom->def);
1985
    info->nrVirtCpu = virDomainDefGetVcpus(privdom->def);
1986
    info->cpuTime = ((tv.tv_sec * 1000ll * 1000ll  * 1000ll) + (tv.tv_usec * 1000ll));
1987 1988
    ret = 0;

1989
 cleanup:
1990
    virDomainObjEndAPI(&privdom);
1991
    return ret;
1992 1993
}

1994 1995 1996 1997 1998 1999 2000 2001 2002 2003
static int
testDomainGetState(virDomainPtr domain,
                   int *state,
                   int *reason,
                   unsigned int flags)
{
    virDomainObjPtr privdom;

    virCheckFlags(0, -1);

2004 2005
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
2006

J
Jiri Denemark 已提交
2007
    *state = virDomainObjGetState(privdom, reason);
2008

2009
    virDomainObjEndAPI(&privdom);
2010 2011

    return 0;
2012 2013
}

2014 2015
#define TEST_SAVE_MAGIC "TestGuestMagic"

2016 2017 2018
static int
testDomainSaveFlags(virDomainPtr domain, const char *path,
                    const char *dxml, unsigned int flags)
2019
{
2020
    testDriverPtr privconn = domain->conn->privateData;
2021 2022 2023
    char *xml = NULL;
    int fd = -1;
    int len;
2024
    virDomainObjPtr privdom;
2025
    virObjectEventPtr event = NULL;
2026
    int ret = -1;
2027

2028 2029
    virCheckFlags(0, -1);
    if (dxml) {
2030 2031
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("xml modification unsupported"));
2032 2033 2034
        return -1;
    }

2035

2036
    if (!(privdom = testDomObjFromDomain(domain)))
2037
        goto cleanup;
2038

2039 2040 2041 2042 2043 2044
    if (!virDomainObjIsActive(privdom)) {
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s", _("domain is not running"));
        goto cleanup;
    }

2045
    xml = virDomainDefFormat(privdom->def, privconn->caps,
2046
                             VIR_DOMAIN_DEF_FORMAT_SECURE);
C
Cole Robinson 已提交
2047

2048
    if (xml == NULL) {
2049
        virReportSystemError(errno,
2050 2051
                             _("saving domain '%s' failed to allocate space for metadata"),
                             domain->name);
2052
        goto cleanup;
2053
    }
2054 2055

    if ((fd = open(path, O_CREAT|O_TRUNC|O_WRONLY, S_IRUSR|S_IWUSR)) < 0) {
2056
        virReportSystemError(errno,
2057 2058
                             _("saving domain '%s' to '%s': open failed"),
                             domain->name, path);
2059
        goto cleanup;
2060
    }
2061
    len = strlen(xml);
2062
    if (safewrite(fd, TEST_SAVE_MAGIC, sizeof(TEST_SAVE_MAGIC)) < 0) {
2063
        virReportSystemError(errno,
2064 2065
                             _("saving domain '%s' to '%s': write failed"),
                             domain->name, path);
2066
        goto cleanup;
2067
    }
2068
    if (safewrite(fd, (char*)&len, sizeof(len)) < 0) {
2069
        virReportSystemError(errno,
2070 2071
                             _("saving domain '%s' to '%s': write failed"),
                             domain->name, path);
2072
        goto cleanup;
2073
    }
2074
    if (safewrite(fd, xml, len) < 0) {
2075
        virReportSystemError(errno,
2076 2077
                             _("saving domain '%s' to '%s': write failed"),
                             domain->name, path);
2078
        goto cleanup;
2079
    }
2080

2081
    if (VIR_CLOSE(fd) < 0) {
2082
        virReportSystemError(errno,
2083 2084
                             _("saving domain '%s' to '%s': write failed"),
                             domain->name, path);
2085
        goto cleanup;
2086
    }
2087 2088
    fd = -1;

J
Jiri Denemark 已提交
2089
    testDomainShutdownState(domain, privdom, VIR_DOMAIN_SHUTOFF_SAVED);
2090
    event = virDomainEventLifecycleNewFromObj(privdom,
2091 2092
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_SAVED);
2093

2094 2095
    if (!privdom->persistent)
        virDomainObjListRemove(privconn->domains, privdom);
2096

2097
    ret = 0;
2098
 cleanup:
2099 2100 2101 2102
    VIR_FREE(xml);

    /* Don't report failure in close or unlink, because
     * in either case we're already in a failure scenario
Y
Yuri Chornoivan 已提交
2103
     * and have reported an earlier error */
2104
    if (ret != 0) {
2105
        VIR_FORCE_CLOSE(fd);
2106 2107
        unlink(path);
    }
2108
    virDomainObjEndAPI(&privdom);
2109
    testObjectEventQueue(privconn, event);
2110
    return ret;
2111 2112
}

2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124
static int
testDomainSave(virDomainPtr domain,
               const char *path)
{
    return testDomainSaveFlags(domain, path, NULL, 0);
}

static int
testDomainRestoreFlags(virConnectPtr conn,
                       const char *path,
                       const char *dxml,
                       unsigned int flags)
2125
{
2126
    testDriverPtr privconn = conn->privateData;
2127
    char *xml = NULL;
2128
    char magic[15];
2129 2130 2131
    int fd = -1;
    int len;
    virDomainDefPtr def = NULL;
2132
    virDomainObjPtr dom = NULL;
2133
    virObjectEventPtr event = NULL;
2134
    int ret = -1;
2135

2136 2137
    virCheckFlags(0, -1);
    if (dxml) {
2138 2139
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("xml modification unsupported"));
2140 2141 2142
        return -1;
    }

2143
    if ((fd = open(path, O_RDONLY)) < 0) {
2144
        virReportSystemError(errno,
2145 2146
                             _("cannot read domain image '%s'"),
                             path);
2147
        goto cleanup;
2148
    }
2149
    if (saferead(fd, magic, sizeof(magic)) != sizeof(magic)) {
2150
        virReportSystemError(errno,
2151 2152
                             _("incomplete save header in '%s'"),
                             path);
2153
        goto cleanup;
2154
    }
2155
    if (memcmp(magic, TEST_SAVE_MAGIC, sizeof(magic))) {
2156 2157
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("mismatched header magic"));
2158
        goto cleanup;
2159
    }
2160
    if (saferead(fd, (char*)&len, sizeof(len)) != sizeof(len)) {
2161
        virReportSystemError(errno,
2162 2163
                             _("failed to read metadata length in '%s'"),
                             path);
2164
        goto cleanup;
2165 2166
    }
    if (len < 1 || len > 8192) {
2167 2168
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("length of metadata out of range"));
2169
        goto cleanup;
2170
    }
2171
    if (VIR_ALLOC_N(xml, len+1) < 0)
2172
        goto cleanup;
2173
    if (saferead(fd, xml, len) != len) {
2174
        virReportSystemError(errno,
2175
                             _("incomplete metadata in '%s'"), path);
2176
        goto cleanup;
2177 2178
    }
    xml[len] = '\0';
2179

2180
    def = virDomainDefParseString(xml, privconn->caps, privconn->xmlopt,
2181
                                  NULL, VIR_DOMAIN_DEF_PARSE_INACTIVE);
2182
    if (!def)
2183
        goto cleanup;
2184

2185
    if (testDomainGenerateIfnames(def) < 0)
2186
        goto cleanup;
2187
    if (!(dom = virDomainObjListAdd(privconn->domains,
2188
                                    def,
2189
                                    privconn->xmlopt,
2190 2191 2192
                                    VIR_DOMAIN_OBJ_LIST_ADD_LIVE |
                                    VIR_DOMAIN_OBJ_LIST_ADD_CHECK_LIVE,
                                    NULL)))
2193 2194
        goto cleanup;
    def = NULL;
2195

2196 2197 2198 2199 2200
    if (testDomainStartState(privconn, dom, VIR_DOMAIN_RUNNING_RESTORED) < 0) {
        if (!dom->persistent) {
            virDomainObjListRemove(privconn->domains, dom);
            dom = NULL;
        }
2201
        goto cleanup;
2202
    }
2203

2204
    event = virDomainEventLifecycleNewFromObj(dom,
2205 2206
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_RESTORED);
2207
    ret = 0;
2208

2209
 cleanup:
2210 2211
    virDomainDefFree(def);
    VIR_FREE(xml);
2212
    VIR_FORCE_CLOSE(fd);
2213
    if (dom)
2214
        virObjectUnlock(dom);
2215
    testObjectEventQueue(privconn, event);
2216
    return ret;
2217 2218
}

2219 2220 2221 2222 2223 2224 2225
static int
testDomainRestore(virConnectPtr conn,
                  const char *path)
{
    return testDomainRestoreFlags(conn, path, NULL, 0);
}

2226 2227 2228 2229
static int testDomainCoreDumpWithFormat(virDomainPtr domain,
                                        const char *to,
                                        unsigned int dumpformat,
                                        unsigned int flags)
2230
{
2231
    testDriverPtr privconn = domain->conn->privateData;
2232
    int fd = -1;
2233
    virDomainObjPtr privdom;
2234
    virObjectEventPtr event = NULL;
2235
    int ret = -1;
2236

E
Eric Blake 已提交
2237 2238
    virCheckFlags(VIR_DUMP_CRASH, -1);

2239

2240
    if (!(privdom = testDomObjFromDomain(domain)))
2241
        goto cleanup;
2242

2243 2244 2245 2246 2247 2248
    if (!virDomainObjIsActive(privdom)) {
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s", _("domain is not running"));
        goto cleanup;
    }

2249
    if ((fd = open(to, O_CREAT|O_TRUNC|O_WRONLY, S_IRUSR|S_IWUSR)) < 0) {
2250
        virReportSystemError(errno,
2251 2252
                             _("domain '%s' coredump: failed to open %s"),
                             domain->name, to);
2253
        goto cleanup;
2254
    }
2255
    if (safewrite(fd, TEST_SAVE_MAGIC, sizeof(TEST_SAVE_MAGIC)) < 0) {
2256
        virReportSystemError(errno,
2257 2258
                             _("domain '%s' coredump: failed to write header to %s"),
                             domain->name, to);
2259
        goto cleanup;
2260
    }
2261
    if (VIR_CLOSE(fd) < 0) {
2262
        virReportSystemError(errno,
2263 2264
                             _("domain '%s' coredump: write failed: %s"),
                             domain->name, to);
2265
        goto cleanup;
2266
    }
2267

2268 2269 2270 2271 2272 2273 2274
    /* we don't support non-raw formats in test driver */
    if (dumpformat != VIR_DOMAIN_CORE_DUMP_FORMAT_RAW) {
        virReportError(VIR_ERR_OPERATION_UNSUPPORTED, "%s",
                       _("kdump-compressed format is not supported here"));
        goto cleanup;
    }

2275
    if (flags & VIR_DUMP_CRASH) {
J
Jiri Denemark 已提交
2276
        testDomainShutdownState(domain, privdom, VIR_DOMAIN_SHUTOFF_CRASHED);
2277
        event = virDomainEventLifecycleNewFromObj(privdom,
2278 2279
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_CRASHED);
2280 2281
        if (!privdom->persistent)
            virDomainObjListRemove(privconn->domains, privdom);
2282
    }
2283

2284
    ret = 0;
2285
 cleanup:
2286
    VIR_FORCE_CLOSE(fd);
2287
    virDomainObjEndAPI(&privdom);
2288
    testObjectEventQueue(privconn, event);
2289
    return ret;
2290 2291
}

2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305

static int
testDomainCoreDump(virDomainPtr domain,
                   const char *to,
                   unsigned int flags)
{
    return testDomainCoreDumpWithFormat(domain, to,
                                        VIR_DOMAIN_CORE_DUMP_FORMAT_RAW, flags);
}


static char *
testDomainGetOSType(virDomainPtr dom ATTRIBUTE_UNUSED)
{
2306 2307 2308
    char *ret;

    ignore_value(VIR_STRDUP(ret, "linux"));
2309
    return ret;
2310 2311
}

2312 2313 2314

static unsigned long long
testDomainGetMaxMemory(virDomainPtr domain)
2315
{
2316
    virDomainObjPtr privdom;
2317
    unsigned long long ret = 0;
2318

2319 2320
    if (!(privdom = testDomObjFromDomain(domain)))
        return 0;
2321

2322
    ret = virDomainDefGetMemoryTotal(privdom->def);
2323

2324
    virDomainObjEndAPI(&privdom);
2325
    return ret;
2326 2327
}

2328 2329
static int testDomainSetMaxMemory(virDomainPtr domain,
                                  unsigned long memory)
2330
{
2331 2332
    virDomainObjPtr privdom;

2333 2334
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
2335 2336

    /* XXX validate not over host memory wrt to other domains */
2337
    virDomainDefSetMemoryTotal(privdom->def, memory);
2338

2339
    virDomainObjEndAPI(&privdom);
2340
    return 0;
2341 2342
}

2343 2344
static int testDomainSetMemory(virDomainPtr domain,
                               unsigned long memory)
2345
{
2346
    virDomainObjPtr privdom;
2347
    int ret = -1;
2348

2349 2350
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
2351

2352
    if (memory > virDomainDefGetMemoryTotal(privdom->def)) {
2353
        virReportError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2354
        goto cleanup;
2355
    }
2356

2357
    privdom->def->mem.cur_balloon = memory;
2358 2359
    ret = 0;

2360
 cleanup:
2361
    virDomainObjEndAPI(&privdom);
2362
    return ret;
2363 2364
}

2365 2366
static int
testDomainGetVcpusFlags(virDomainPtr domain, unsigned int flags)
C
Cole Robinson 已提交
2367
{
2368 2369 2370 2371
    virDomainObjPtr vm;
    virDomainDefPtr def;
    int ret = -1;

2372 2373
    virCheckFlags(VIR_DOMAIN_AFFECT_LIVE |
                  VIR_DOMAIN_AFFECT_CONFIG |
2374 2375
                  VIR_DOMAIN_VCPU_MAXIMUM, -1);

2376 2377
    if (!(vm = testDomObjFromDomain(domain)))
        return -1;
2378

2379
    if (!(def = virDomainObjGetOneDef(vm, flags)))
2380
        goto cleanup;
2381

2382 2383 2384
    if (flags & VIR_DOMAIN_VCPU_MAXIMUM)
        ret = virDomainDefGetVcpusMax(def);
    else
2385
        ret = virDomainDefGetVcpus(def);
2386

2387
 cleanup:
2388
    virDomainObjEndAPI(&vm);
2389
    return ret;
C
Cole Robinson 已提交
2390 2391
}

2392 2393 2394
static int
testDomainGetMaxVcpus(virDomainPtr domain)
{
2395
    return testDomainGetVcpusFlags(domain, (VIR_DOMAIN_AFFECT_LIVE |
2396 2397 2398 2399 2400 2401 2402
                                            VIR_DOMAIN_VCPU_MAXIMUM));
}

static int
testDomainSetVcpusFlags(virDomainPtr domain, unsigned int nrCpus,
                        unsigned int flags)
{
2403
    testDriverPtr driver = domain->conn->privateData;
2404
    virDomainObjPtr privdom = NULL;
2405
    virDomainDefPtr def;
2406
    virDomainDefPtr persistentDef;
C
Cole Robinson 已提交
2407 2408
    int ret = -1, maxvcpus;

2409 2410
    virCheckFlags(VIR_DOMAIN_AFFECT_LIVE |
                  VIR_DOMAIN_AFFECT_CONFIG |
2411 2412
                  VIR_DOMAIN_VCPU_MAXIMUM, -1);

2413
    if ((maxvcpus = testConnectGetMaxVcpus(domain->conn, NULL)) < 0)
2414
        return -1;
2415 2416

    if (nrCpus > maxvcpus) {
2417
        virReportError(VIR_ERR_INVALID_ARG,
2418 2419
                       _("requested cpu amount exceeds maximum supported amount "
                         "(%d > %d)"), nrCpus, maxvcpus);
2420 2421
        return -1;
    }
2422

2423 2424
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
2425

2426
    if (virDomainObjGetDefs(privdom, flags, &def, &persistentDef) < 0)
C
Cole Robinson 已提交
2427 2428
        goto cleanup;

2429
    if (def && virDomainDefGetVcpusMax(def) < nrCpus) {
2430 2431
        virReportError(VIR_ERR_INVALID_ARG,
                       _("requested cpu amount exceeds maximum (%d > %d)"),
2432
                       nrCpus, virDomainDefGetVcpusMax(def));
2433
        goto cleanup;
2434
    }
2435

2436 2437
    if (persistentDef &&
        !(flags & VIR_DOMAIN_VCPU_MAXIMUM) &&
2438
        virDomainDefGetVcpusMax(persistentDef) < nrCpus) {
2439 2440
        virReportError(VIR_ERR_INVALID_ARG,
                       _("requested cpu amount exceeds maximum (%d > %d)"),
2441
                       nrCpus, virDomainDefGetVcpusMax(persistentDef));
2442
        goto cleanup;
2443
    }
2444

2445 2446 2447
    if (def &&
        virDomainDefSetVcpus(def, nrCpus) < 0)
        goto cleanup;
2448

2449 2450
    if (persistentDef) {
        if (flags & VIR_DOMAIN_VCPU_MAXIMUM) {
2451 2452
            if (virDomainDefSetVcpusMax(persistentDef, nrCpus,
                                        driver->xmlopt) < 0)
2453
                goto cleanup;
2454
        } else {
2455 2456
            if (virDomainDefSetVcpus(persistentDef, nrCpus) < 0)
                goto cleanup;
2457
        }
2458
    }
2459

2460 2461
    ret = 0;

2462
 cleanup:
2463
    virDomainObjEndAPI(&privdom);
2464
    return ret;
2465 2466
}

2467
static int
2468
testDomainSetVcpus(virDomainPtr domain, unsigned int nrCpus)
2469
{
2470
    return testDomainSetVcpusFlags(domain, nrCpus, VIR_DOMAIN_AFFECT_LIVE);
2471 2472
}

C
Cole Robinson 已提交
2473 2474 2475 2476 2477 2478
static int testDomainGetVcpus(virDomainPtr domain,
                              virVcpuInfoPtr info,
                              int maxinfo,
                              unsigned char *cpumaps,
                              int maplen)
{
2479
    testDriverPtr privconn = domain->conn->privateData;
C
Cole Robinson 已提交
2480
    virDomainObjPtr privdom;
2481
    virDomainDefPtr def;
2482
    size_t i;
2483
    int hostcpus;
C
Cole Robinson 已提交
2484 2485 2486
    int ret = -1;
    struct timeval tv;
    unsigned long long statbase;
2487
    virBitmapPtr allcpumap = NULL;
C
Cole Robinson 已提交
2488

2489 2490
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
C
Cole Robinson 已提交
2491 2492

    if (!virDomainObjIsActive(privdom)) {
2493
        virReportError(VIR_ERR_OPERATION_INVALID,
2494
                       "%s", _("cannot list vcpus for an inactive domain"));
C
Cole Robinson 已提交
2495 2496 2497
        goto cleanup;
    }

2498
    def = privdom->def;
C
Cole Robinson 已提交
2499 2500

    if (gettimeofday(&tv, NULL) < 0) {
2501
        virReportSystemError(errno,
C
Cole Robinson 已提交
2502 2503 2504 2505 2506 2507 2508
                             "%s", _("getting time of day"));
        goto cleanup;
    }

    statbase = (tv.tv_sec * 1000UL * 1000UL) + tv.tv_usec;

    hostcpus = VIR_NODEINFO_MAXCPUS(privconn->nodeInfo);
2509 2510 2511 2512 2513
    if (!(allcpumap = virBitmapNew(hostcpus)))
        goto cleanup;

    virBitmapSetAll(allcpumap);

C
Cole Robinson 已提交
2514
    /* Clamp to actual number of vcpus */
2515 2516
    if (maxinfo > virDomainDefGetVcpus(privdom->def))
        maxinfo = virDomainDefGetVcpus(privdom->def);
C
Cole Robinson 已提交
2517

2518 2519
    memset(info, 0, sizeof(*info) * maxinfo);
    memset(cpumaps, 0, maxinfo * maplen);
C
Cole Robinson 已提交
2520

2521
    for (i = 0; i < maxinfo; i++) {
2522
        virDomainVcpuDefPtr vcpu = virDomainDefGetVcpu(def, i);
2523
        virBitmapPtr bitmap = NULL;
C
Cole Robinson 已提交
2524

2525 2526
        if (!vcpu->online)
            continue;
C
Cole Robinson 已提交
2527

2528 2529
        if (vcpu->cpumask)
            bitmap = vcpu->cpumask;
2530 2531 2532 2533
        else if (def->cpumask)
            bitmap = def->cpumask;
        else
            bitmap = allcpumap;
C
Cole Robinson 已提交
2534

2535 2536
        if (cpumaps)
            virBitmapToDataBuf(bitmap, VIR_GET_CPUMAP(cpumaps, maplen, i), maplen);
C
Cole Robinson 已提交
2537

2538 2539 2540
        info[i].number = i;
        info[i].state = VIR_VCPU_RUNNING;
        info[i].cpu = virBitmapLastSetBit(bitmap);
C
Cole Robinson 已提交
2541

2542 2543
        /* Fake an increasing cpu time value */
        info[i].cpuTime = statbase / 10;
C
Cole Robinson 已提交
2544 2545 2546
    }

    ret = maxinfo;
2547
 cleanup:
2548
    virBitmapFree(allcpumap);
2549
    virDomainObjEndAPI(&privdom);
C
Cole Robinson 已提交
2550 2551 2552
    return ret;
}

C
Cole Robinson 已提交
2553 2554 2555 2556 2557
static int testDomainPinVcpu(virDomainPtr domain,
                             unsigned int vcpu,
                             unsigned char *cpumap,
                             int maplen)
{
2558
    virDomainVcpuDefPtr vcpuinfo;
C
Cole Robinson 已提交
2559
    virDomainObjPtr privdom;
2560
    virDomainDefPtr def;
C
Cole Robinson 已提交
2561 2562
    int ret = -1;

2563 2564
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
C
Cole Robinson 已提交
2565

2566 2567
    def = privdom->def;

C
Cole Robinson 已提交
2568
    if (!virDomainObjIsActive(privdom)) {
2569
        virReportError(VIR_ERR_OPERATION_INVALID,
2570
                       "%s", _("cannot pin vcpus on an inactive domain"));
C
Cole Robinson 已提交
2571 2572 2573
        goto cleanup;
    }

2574 2575
    if (!(vcpuinfo = virDomainDefGetVcpu(def, vcpu)) ||
        !vcpuinfo->online) {
2576 2577 2578
        virReportError(VIR_ERR_INVALID_ARG,
                       _("requested vcpu '%d' is not present in the domain"),
                       vcpu);
C
Cole Robinson 已提交
2579 2580 2581
        goto cleanup;
    }

2582 2583 2584
    virBitmapFree(vcpuinfo->cpumask);

    if (!(vcpuinfo->cpumask = virBitmapNewData(cpumap, maplen)))
2585
        goto cleanup;
C
Cole Robinson 已提交
2586 2587

    ret = 0;
2588

2589
 cleanup:
2590
    virDomainObjEndAPI(&privdom);
C
Cole Robinson 已提交
2591 2592 2593
    return ret;
}

2594 2595 2596 2597 2598 2599 2600
static int
testDomainGetVcpuPinInfo(virDomainPtr dom,
                        int ncpumaps,
                        unsigned char *cpumaps,
                        int maplen,
                        unsigned int flags)
{
2601
    testDriverPtr driver = dom->conn->privateData;
2602 2603
    virDomainObjPtr privdom;
    virDomainDefPtr def;
2604
    int ret = -1;
2605 2606 2607 2608 2609 2610 2611

    if (!(privdom = testDomObjFromDomain(dom)))
        return -1;

    if (!(def = virDomainObjGetOneDef(privdom, flags)))
        goto cleanup;

2612 2613 2614
    ret = virDomainDefGetVcpuPinInfoHelper(def, maplen, ncpumaps, cpumaps,
                                           VIR_NODEINFO_MAXCPUS(driver->nodeInfo),
                                           NULL);
2615 2616 2617 2618 2619 2620

 cleanup:
    virDomainObjEndAPI(&privdom);
    return ret;
}

2621
static char *testDomainGetXMLDesc(virDomainPtr domain, unsigned int flags)
2622
{
2623
    testDriverPtr privconn = domain->conn->privateData;
2624
    virDomainDefPtr def;
2625
    virDomainObjPtr privdom;
2626 2627
    char *ret = NULL;

2628 2629
    /* Flags checked by virDomainDefFormat */

2630 2631
    if (!(privdom = testDomObjFromDomain(domain)))
        return NULL;
2632

2633 2634
    def = (flags & VIR_DOMAIN_XML_INACTIVE) &&
        privdom->newDef ? privdom->newDef : privdom->def;
2635

2636 2637
    ret = virDomainDefFormat(def, privconn->caps,
                             virDomainDefFormatConvertXMLFlags(flags));
2638

2639
    virDomainObjEndAPI(&privdom);
2640
    return ret;
2641
}
2642

2643 2644
static int testConnectNumOfDefinedDomains(virConnectPtr conn)
{
2645
    testDriverPtr privconn = conn->privateData;
2646

2647
    return virDomainObjListNumOfDomains(privconn->domains, false, NULL, NULL);
2648 2649
}

2650 2651
static int testConnectListDefinedDomains(virConnectPtr conn,
                                         char **const names,
2652 2653
                                         int maxnames)
{
2654

2655
    testDriverPtr privconn = conn->privateData;
2656 2657

    memset(names, 0, sizeof(*names)*maxnames);
2658 2659
    return virDomainObjListGetInactiveNames(privconn->domains, names, maxnames,
                                            NULL, NULL);
2660 2661
}

2662 2663 2664
static virDomainPtr testDomainDefineXMLFlags(virConnectPtr conn,
                                             const char *xml,
                                             unsigned int flags)
2665
{
2666
    testDriverPtr privconn = conn->privateData;
2667
    virDomainPtr ret = NULL;
2668
    virDomainDefPtr def;
2669
    virDomainObjPtr dom = NULL;
2670
    virObjectEventPtr event = NULL;
2671
    virDomainDefPtr oldDef = NULL;
2672 2673 2674
    unsigned int parse_flags = VIR_DOMAIN_DEF_PARSE_INACTIVE;

    virCheckFlags(VIR_DOMAIN_DEFINE_VALIDATE, NULL);
2675

2676
    if (flags & VIR_DOMAIN_DEFINE_VALIDATE)
2677
        parse_flags |= VIR_DOMAIN_DEF_PARSE_VALIDATE_SCHEMA;
2678

2679
    if ((def = virDomainDefParseString(xml, privconn->caps, privconn->xmlopt,
2680
                                       NULL, parse_flags)) == NULL)
2681
        goto cleanup;
2682

2683 2684 2685
    if (virXMLCheckIllegalChars("name", def->name, "\n") < 0)
        goto cleanup;

2686
    if (testDomainGenerateIfnames(def) < 0)
2687
        goto cleanup;
2688
    if (!(dom = virDomainObjListAdd(privconn->domains,
2689
                                    def,
2690
                                    privconn->xmlopt,
2691 2692
                                    0,
                                    &oldDef)))
2693
        goto cleanup;
2694
    def = NULL;
2695
    dom->persistent = 1;
2696

2697
    event = virDomainEventLifecycleNewFromObj(dom,
2698
                                     VIR_DOMAIN_EVENT_DEFINED,
2699
                                     !oldDef ?
2700 2701
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED :
                                     VIR_DOMAIN_EVENT_DEFINED_UPDATED);
2702

2703
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid, dom->def->id);
2704

2705
 cleanup:
2706
    virDomainDefFree(def);
2707
    virDomainDefFree(oldDef);
2708
    if (dom)
2709
        virObjectUnlock(dom);
2710
    testObjectEventQueue(privconn, event);
2711
    return ret;
2712 2713
}

2714 2715 2716 2717 2718 2719
static virDomainPtr
testDomainDefineXML(virConnectPtr conn, const char *xml)
{
    return testDomainDefineXMLFlags(conn, xml, 0);
}

2720 2721 2722 2723 2724 2725
static char *testDomainGetMetadata(virDomainPtr dom,
                                   int type,
                                   const char *uri,
                                   unsigned int flags)
{
    virDomainObjPtr privdom;
2726
    char *ret;
2727 2728 2729 2730

    virCheckFlags(VIR_DOMAIN_AFFECT_LIVE |
                  VIR_DOMAIN_AFFECT_CONFIG, NULL);

2731 2732
    if (!(privdom = testDomObjFromDomain(dom)))
        return NULL;
2733

2734
    ret = virDomainObjGetMetadata(privdom, type, uri, flags);
2735

2736
    virDomainObjEndAPI(&privdom);
2737 2738 2739 2740 2741 2742 2743 2744 2745 2746
    return ret;
}

static int testDomainSetMetadata(virDomainPtr dom,
                                 int type,
                                 const char *metadata,
                                 const char *key,
                                 const char *uri,
                                 unsigned int flags)
{
2747
    testDriverPtr privconn = dom->conn->privateData;
2748
    virDomainObjPtr privdom;
2749
    int ret;
2750 2751 2752 2753

    virCheckFlags(VIR_DOMAIN_AFFECT_LIVE |
                  VIR_DOMAIN_AFFECT_CONFIG, -1);

2754 2755
    if (!(privdom = testDomObjFromDomain(dom)))
        return -1;
2756 2757 2758

    ret = virDomainObjSetMetadata(privdom, type, metadata, key, uri,
                                  privconn->caps, privconn->xmlopt,
2759
                                  NULL, NULL, flags);
2760

2761 2762 2763 2764 2765 2766
    if (ret == 0) {
        virObjectEventPtr ev = NULL;
        ev = virDomainEventMetadataChangeNewFromObj(privdom, type, uri);
        testObjectEventQueue(privconn, ev);
    }

2767
    virDomainObjEndAPI(&privdom);
2768 2769 2770 2771
    return ret;
}


2772 2773
static int testNodeGetCellsFreeMemory(virConnectPtr conn,
                                      unsigned long long *freemems,
2774 2775
                                      int startCell, int maxCells)
{
2776
    testDriverPtr privconn = conn->privateData;
2777 2778
    int cell;
    size_t i;
2779
    int ret = -1;
2780

2781
    testDriverLock(privconn);
2782
    if (startCell >= privconn->numCells) {
2783 2784
        virReportError(VIR_ERR_INVALID_ARG,
                       "%s", _("Range exceeds available cells"));
2785
        goto cleanup;
2786 2787
    }

2788 2789 2790 2791
    for (cell = startCell, i = 0;
         (cell < privconn->numCells && i < maxCells);
         ++cell, ++i) {
        freemems[i] = privconn->cells[cell].mem;
2792
    }
2793
    ret = i;
2794

2795
 cleanup:
2796
    testDriverUnlock(privconn);
2797
    return ret;
2798 2799
}

2800 2801 2802 2803 2804 2805 2806 2807 2808 2809 2810 2811 2812 2813 2814 2815 2816 2817 2818 2819 2820 2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834 2835 2836 2837 2838 2839 2840 2841 2842 2843 2844 2845
#define TEST_NB_CPU_STATS 4

static int
testNodeGetCPUStats(virConnectPtr conn ATTRIBUTE_UNUSED,
                    int cpuNum ATTRIBUTE_UNUSED,
                    virNodeCPUStatsPtr params,
                    int *nparams,
                    unsigned int flags)
{
    size_t i = 0;

    virCheckFlags(0, -1);

    if (params == NULL) {
        *nparams = TEST_NB_CPU_STATS;
        return 0;
    }

    for (i = 0; i < *nparams && i < 4; i++) {
        switch (i) {
        case 0:
            if (virHostCPUStatsAssign(&params[i],
                                      VIR_NODE_CPU_STATS_USER, 9797400000) < 0)
                return -1;
            break;
        case 1:
            if (virHostCPUStatsAssign(&params[i],
                                      VIR_NODE_CPU_STATS_KERNEL, 34678723400000) < 0)
                return -1;
            break;
        case 2:
            if (virHostCPUStatsAssign(&params[i],
                                      VIR_NODE_CPU_STATS_IDLE, 87264900000) < 0)
                return -1;
            break;
        case 3:
            if (virHostCPUStatsAssign(&params[i],
                                      VIR_NODE_CPU_STATS_IOWAIT, 763600000) < 0)
                return -1;
            break;
        }
    }

    *nparams = i;
    return 0;
}
2846

2847 2848 2849 2850 2851 2852 2853 2854 2855 2856 2857 2858 2859 2860 2861 2862
static unsigned long long
testNodeGetFreeMemory(virConnectPtr conn)
{
    testDriverPtr privconn = conn->privateData;
    unsigned int freeMem = 0;
    size_t i;

    testDriverLock(privconn);

    for (i = 0; i < privconn->numCells; i++)
        freeMem += privconn->cells[i].freeMem;

    testDriverUnlock(privconn);
    return freeMem;
}

2863 2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883 2884 2885 2886
static int
testNodeGetFreePages(virConnectPtr conn ATTRIBUTE_UNUSED,
                     unsigned int npages,
                     unsigned int *pages ATTRIBUTE_UNUSED,
                     int startCell ATTRIBUTE_UNUSED,
                     unsigned int cellCount,
                     unsigned long long *counts,
                     unsigned int flags)
{
    size_t i = 0, j = 0;
    int x = 6;

    virCheckFlags(0, -1);

    for (i = 0; i < cellCount; i++) {
        for (j = 0; j < npages; j++) {
            x = x * 2 + 7;
            counts[(i * npages) +  j] = x;
        }
    }

    return 0;
}

2887 2888
static int testDomainCreateWithFlags(virDomainPtr domain, unsigned int flags)
{
2889
    testDriverPtr privconn = domain->conn->privateData;
2890
    virDomainObjPtr privdom;
2891
    virObjectEventPtr event = NULL;
2892
    int ret = -1;
2893

2894 2895
    virCheckFlags(0, -1);

2896
    testDriverLock(privconn);
2897

2898
    if (!(privdom = testDomObjFromDomain(domain)))
2899
        goto cleanup;
2900

J
Jiri Denemark 已提交
2901
    if (virDomainObjGetState(privdom, NULL) != VIR_DOMAIN_SHUTOFF) {
2902 2903
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Domain '%s' is already running"), domain->name);
2904
        goto cleanup;
2905 2906
    }

2907
    if (testDomainStartState(privconn, privdom,
J
Jiri Denemark 已提交
2908
                             VIR_DOMAIN_RUNNING_BOOTED) < 0)
2909 2910 2911
        goto cleanup;
    domain->id = privdom->def->id;

2912
    event = virDomainEventLifecycleNewFromObj(privdom,
2913 2914
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_BOOTED);
2915
    ret = 0;
2916

2917
 cleanup:
2918
    virDomainObjEndAPI(&privdom);
2919
    testObjectEventQueue(privconn, event);
2920
    testDriverUnlock(privconn);
2921
    return ret;
2922 2923
}

2924 2925
static int testDomainCreate(virDomainPtr domain)
{
2926 2927 2928
    return testDomainCreateWithFlags(domain, 0);
}

2929 2930 2931
static int testDomainUndefineFlags(virDomainPtr domain,
                                   unsigned int flags)
{
2932
    testDriverPtr privconn = domain->conn->privateData;
2933
    virDomainObjPtr privdom;
2934
    virObjectEventPtr event = NULL;
2935
    int nsnapshots;
2936
    int ret = -1;
2937

2938 2939
    virCheckFlags(VIR_DOMAIN_UNDEFINE_MANAGED_SAVE |
                  VIR_DOMAIN_UNDEFINE_SNAPSHOTS_METADATA, -1);
2940

2941

2942
    if (!(privdom = testDomObjFromDomain(domain)))
2943
        goto cleanup;
2944

C
Cole Robinson 已提交
2945 2946 2947 2948 2949 2950 2951 2952
    if (privdom->hasManagedSave &&
        !(flags & VIR_DOMAIN_UNDEFINE_MANAGED_SAVE)) {
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                       _("Refusing to undefine while domain managed "
                         "save image exists"));
        goto cleanup;
    }

2953 2954 2955 2956 2957 2958 2959 2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970
    /* Requiring an inactive VM is part of the documented API for
     * UNDEFINE_SNAPSHOTS_METADATA
     */
    if (!virDomainObjIsActive(privdom) &&
        (nsnapshots = virDomainSnapshotObjListNum(privdom->snapshots,
                                                  NULL, 0))) {
        if (!(flags & VIR_DOMAIN_UNDEFINE_SNAPSHOTS_METADATA)) {
            virReportError(VIR_ERR_OPERATION_INVALID,
                           _("cannot delete inactive domain with %d "
                             "snapshots"),
                           nsnapshots);
            goto cleanup;
        }

        /* There isn't actually anything to do, we are just emulating qemu
         * behavior here. */
    }

2971
    event = virDomainEventLifecycleNewFromObj(privdom,
2972 2973
                                     VIR_DOMAIN_EVENT_UNDEFINED,
                                     VIR_DOMAIN_EVENT_UNDEFINED_REMOVED);
C
Cole Robinson 已提交
2974 2975
    privdom->hasManagedSave = false;

2976
    if (virDomainObjIsActive(privdom))
2977
        privdom->persistent = 0;
2978 2979
    else
        virDomainObjListRemove(privconn->domains, privdom);
2980

2981
    ret = 0;
2982

2983
 cleanup:
2984
    virDomainObjEndAPI(&privdom);
2985
    testObjectEventQueue(privconn, event);
2986
    return ret;
2987 2988
}

2989 2990 2991 2992 2993
static int testDomainUndefine(virDomainPtr domain)
{
    return testDomainUndefineFlags(domain, 0);
}

2994 2995 2996
static int testDomainGetAutostart(virDomainPtr domain,
                                  int *autostart)
{
2997 2998
    virDomainObjPtr privdom;

2999 3000
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
3001

3002
    *autostart = privdom->autostart;
3003

3004
    virDomainObjEndAPI(&privdom);
3005
    return 0;
3006 3007 3008 3009 3010 3011
}


static int testDomainSetAutostart(virDomainPtr domain,
                                  int autostart)
{
3012 3013
    virDomainObjPtr privdom;

3014 3015
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
3016

3017
    privdom->autostart = autostart ? 1 : 0;
3018

3019
    virDomainObjEndAPI(&privdom);
3020
    return 0;
3021
}
3022

3023
static char *testDomainGetSchedulerType(virDomainPtr domain ATTRIBUTE_UNUSED,
3024 3025
                                        int *nparams)
{
3026 3027
    char *type = NULL;

3028 3029 3030
    if (nparams)
        *nparams = 1;

3031
    ignore_value(VIR_STRDUP(type, "fair"));
3032

3033 3034 3035
    return type;
}

3036
static int
3037 3038 3039 3040
testDomainGetSchedulerParametersFlags(virDomainPtr domain,
                                      virTypedParameterPtr params,
                                      int *nparams,
                                      unsigned int flags)
3041
{
3042
    virDomainObjPtr privdom;
3043
    int ret = -1;
3044

3045 3046
    virCheckFlags(0, -1);

3047 3048
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
3049

3050 3051
    if (virTypedParameterAssign(params, VIR_DOMAIN_SCHEDULER_WEIGHT,
                                VIR_TYPED_PARAM_UINT, 50) < 0)
3052
        goto cleanup;
3053 3054
    /* XXX */
    /*params[0].value.ui = privdom->weight;*/
3055 3056

    *nparams = 1;
3057 3058
    ret = 0;

3059
 cleanup:
3060
    virDomainObjEndAPI(&privdom);
3061
    return ret;
3062
}
3063

3064
static int
3065 3066 3067
testDomainGetSchedulerParameters(virDomainPtr domain,
                                 virTypedParameterPtr params,
                                 int *nparams)
3068
{
3069
    return testDomainGetSchedulerParametersFlags(domain, params, nparams, 0);
3070
}
3071

3072
static int
3073 3074 3075 3076
testDomainSetSchedulerParametersFlags(virDomainPtr domain,
                                      virTypedParameterPtr params,
                                      int nparams,
                                      unsigned int flags)
3077
{
3078
    virDomainObjPtr privdom;
3079 3080
    int ret = -1;
    size_t i;
3081

3082
    virCheckFlags(0, -1);
3083 3084 3085 3086
    if (virTypedParamsValidate(params, nparams,
                               VIR_DOMAIN_SCHEDULER_WEIGHT,
                               VIR_TYPED_PARAM_UINT,
                               NULL) < 0)
3087
        return -1;
3088

3089 3090
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
3091

3092
    for (i = 0; i < nparams; i++) {
3093 3094 3095
        if (STREQ(params[i].field, VIR_DOMAIN_SCHEDULER_WEIGHT)) {
            /* XXX */
            /*privdom->weight = params[i].value.ui;*/
3096
        }
3097
    }
3098

3099 3100
    ret = 0;

3101
    virDomainObjEndAPI(&privdom);
3102
    return ret;
3103 3104
}

3105
static int
3106 3107 3108
testDomainSetSchedulerParameters(virDomainPtr domain,
                                 virTypedParameterPtr params,
                                 int nparams)
3109
{
3110
    return testDomainSetSchedulerParametersFlags(domain, params, nparams, 0);
3111 3112
}

3113 3114
static int testDomainBlockStats(virDomainPtr domain,
                                const char *path,
3115
                                virDomainBlockStatsPtr stats)
3116 3117 3118 3119
{
    virDomainObjPtr privdom;
    struct timeval tv;
    unsigned long long statbase;
3120
    int ret = -1;
3121

3122 3123 3124 3125 3126 3127
    if (!*path) {
        virReportError(VIR_ERR_OPERATION_UNSUPPORTED, "%s",
                       _("summary statistics are not supported yet"));
        return ret;
    }

3128 3129
    if (!(privdom = testDomObjFromDomain(domain)))
        return ret;
3130

3131 3132 3133 3134 3135 3136
    if (!virDomainObjIsActive(privdom)) {
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s", _("domain is not running"));
        goto error;
    }

3137
    if (virDomainDiskIndexByName(privdom->def, path, false) < 0) {
3138 3139
        virReportError(VIR_ERR_INVALID_ARG,
                       _("invalid path: %s"), path);
3140 3141 3142 3143
        goto error;
    }

    if (gettimeofday(&tv, NULL) < 0) {
3144
        virReportSystemError(errno,
3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157
                             "%s", _("getting time of day"));
        goto error;
    }

    /* No significance to these numbers, just enough to mix it up*/
    statbase = (tv.tv_sec * 1000UL * 1000UL) + tv.tv_usec;
    stats->rd_req = statbase / 10;
    stats->rd_bytes = statbase / 20;
    stats->wr_req = statbase / 30;
    stats->wr_bytes = statbase / 40;
    stats->errs = tv.tv_sec / 2;

    ret = 0;
3158
 error:
3159
    virDomainObjEndAPI(&privdom);
3160 3161 3162 3163 3164
    return ret;
}

static int testDomainInterfaceStats(virDomainPtr domain,
                                    const char *path,
3165
                                    virDomainInterfaceStatsPtr stats)
3166 3167 3168 3169
{
    virDomainObjPtr privdom;
    struct timeval tv;
    unsigned long long statbase;
M
Michal Privoznik 已提交
3170 3171 3172
    virDomainNetDefPtr net = NULL;
    int ret = -1;

3173

3174 3175
    if (!(privdom = testDomObjFromDomain(domain)))
        return -1;
3176

3177 3178 3179 3180 3181 3182
    if (!virDomainObjIsActive(privdom)) {
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s", _("domain is not running"));
        goto error;
    }

3183
    if (!(net = virDomainNetFindByName(privdom->def, path)))
3184 3185 3186
        goto error;

    if (gettimeofday(&tv, NULL) < 0) {
3187
        virReportSystemError(errno,
3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202 3203
                             "%s", _("getting time of day"));
        goto error;
    }

    /* No significance to these numbers, just enough to mix it up*/
    statbase = (tv.tv_sec * 1000UL * 1000UL) + tv.tv_usec;
    stats->rx_bytes = statbase / 10;
    stats->rx_packets = statbase / 100;
    stats->rx_errs = tv.tv_sec / 1;
    stats->rx_drop = tv.tv_sec / 2;
    stats->tx_bytes = statbase / 20;
    stats->tx_packets = statbase / 110;
    stats->tx_errs = tv.tv_sec / 3;
    stats->tx_drop = tv.tv_sec / 4;

    ret = 0;
3204
 error:
3205
    virDomainObjEndAPI(&privdom);
3206 3207 3208
    return ret;
}

3209

3210 3211 3212 3213
static virNetworkObjPtr
testNetworkObjFindByUUID(testDriverPtr privconn,
                         const unsigned char *uuid)
{
3214
    virNetworkObjPtr obj;
3215 3216
    char uuidstr[VIR_UUID_STRING_BUFLEN];

3217
    if (!(obj = virNetworkObjFindByUUID(privconn->networks, uuid))) {
3218 3219 3220 3221 3222 3223
        virUUIDFormat(uuid, uuidstr);
        virReportError(VIR_ERR_NO_NETWORK,
                       _("no network with matching uuid '%s'"),
                       uuidstr);
    }

3224
    return obj;
3225 3226 3227
}


3228 3229 3230
static virNetworkPtr
testNetworkLookupByUUID(virConnectPtr conn,
                        const unsigned char *uuid)
3231
{
3232
    testDriverPtr privconn = conn->privateData;
3233
    virNetworkObjPtr obj;
3234
    virNetworkDefPtr def;
3235
    virNetworkPtr net = NULL;
3236

3237
    if (!(obj = testNetworkObjFindByUUID(privconn, uuid)))
3238
        goto cleanup;
3239
    def = virNetworkObjGetDef(obj);
3240

3241
    net = virGetNetwork(conn, def->name, def->uuid);
3242

3243
 cleanup:
3244 3245
    virNetworkObjEndAPI(&obj);
    return net;
3246
}
3247

3248 3249 3250 3251 3252

static virNetworkObjPtr
testNetworkObjFindByName(testDriverPtr privconn,
                         const char *name)
{
3253
    virNetworkObjPtr obj;
3254

3255
    if (!(obj = virNetworkObjFindByName(privconn->networks, name)))
3256 3257 3258 3259
        virReportError(VIR_ERR_NO_NETWORK,
                       _("no network with matching name '%s'"),
                       name);

3260
    return obj;
3261 3262 3263
}


3264 3265 3266
static virNetworkPtr
testNetworkLookupByName(virConnectPtr conn,
                        const char *name)
3267
{
3268
    testDriverPtr privconn = conn->privateData;
3269
    virNetworkObjPtr obj;
3270
    virNetworkDefPtr def;
3271
    virNetworkPtr net = NULL;
3272

3273
    if (!(obj = testNetworkObjFindByName(privconn, name)))
3274
        goto cleanup;
3275
    def = virNetworkObjGetDef(obj);
3276

3277
    net = virGetNetwork(conn, def->name, def->uuid);
3278

3279
 cleanup:
3280 3281
    virNetworkObjEndAPI(&obj);
    return net;
3282 3283 3284
}


3285 3286
static int
testConnectNumOfNetworks(virConnectPtr conn)
3287
{
3288
    testDriverPtr privconn = conn->privateData;
3289
    int numActive;
3290

3291 3292
    numActive = virNetworkObjListNumOfNetworks(privconn->networks,
                                               true, NULL, conn);
3293
    return numActive;
3294 3295
}

3296 3297 3298 3299

static int
testConnectListNetworks(virConnectPtr conn,
                        char **const names,
3300
                        int maxnames)
3301
{
3302
    testDriverPtr privconn = conn->privateData;
3303
    int n;
3304

3305
    n = virNetworkObjListGetNames(privconn->networks,
3306
                                  true, names, maxnames, NULL, conn);
3307
    return n;
3308 3309
}

3310 3311 3312

static int
testConnectNumOfDefinedNetworks(virConnectPtr conn)
3313
{
3314
    testDriverPtr privconn = conn->privateData;
3315
    int numInactive;
3316

3317 3318
    numInactive = virNetworkObjListNumOfNetworks(privconn->networks,
                                                 false, NULL, conn);
3319
    return numInactive;
3320 3321
}

3322 3323 3324 3325

static int
testConnectListDefinedNetworks(virConnectPtr conn,
                               char **const names,
3326
                               int maxnames)
3327
{
3328
    testDriverPtr privconn = conn->privateData;
3329
    int n;
3330

3331
    n = virNetworkObjListGetNames(privconn->networks,
3332
                                  false, names, maxnames, NULL, conn);
3333
    return n;
3334 3335
}

3336

3337
static int
3338
testConnectListAllNetworks(virConnectPtr conn,
3339 3340 3341
                           virNetworkPtr **nets,
                           unsigned int flags)
{
3342
    testDriverPtr privconn = conn->privateData;
3343 3344 3345

    virCheckFlags(VIR_CONNECT_LIST_NETWORKS_FILTERS_ALL, -1);

3346
    return virNetworkObjListExport(conn, privconn->networks, nets, NULL, flags);
3347
}
3348

3349 3350 3351

static int
testNetworkIsActive(virNetworkPtr net)
3352
{
3353
    testDriverPtr privconn = net->conn->privateData;
3354 3355 3356
    virNetworkObjPtr obj;
    int ret = -1;

3357
    if (!(obj = testNetworkObjFindByUUID(privconn, net->uuid)))
3358
        goto cleanup;
3359

3360 3361
    ret = virNetworkObjIsActive(obj);

3362
 cleanup:
3363
    virNetworkObjEndAPI(&obj);
3364 3365 3366
    return ret;
}

3367 3368 3369

static int
testNetworkIsPersistent(virNetworkPtr net)
3370
{
3371
    testDriverPtr privconn = net->conn->privateData;
3372 3373 3374
    virNetworkObjPtr obj;
    int ret = -1;

3375
    if (!(obj = testNetworkObjFindByUUID(privconn, net->uuid)))
3376
        goto cleanup;
3377

3378
    ret = virNetworkObjIsPersistent(obj);
3379

3380
 cleanup:
3381
    virNetworkObjEndAPI(&obj);
3382 3383 3384 3385
    return ret;
}


3386 3387
static virNetworkPtr
testNetworkCreateXML(virConnectPtr conn, const char *xml)
3388
{
3389
    testDriverPtr privconn = conn->privateData;
3390
    virNetworkDefPtr newDef;
3391
    virNetworkObjPtr obj = NULL;
3392
    virNetworkDefPtr def;
3393
    virNetworkPtr net = NULL;
3394
    virObjectEventPtr event = NULL;
3395

3396
    if ((newDef = virNetworkDefParseString(xml)) == NULL)
3397
        goto cleanup;
3398

3399
    if (!(obj = virNetworkObjAssignDef(privconn->networks, newDef,
3400 3401
                                       VIR_NETWORK_OBJ_LIST_ADD_LIVE |
                                       VIR_NETWORK_OBJ_LIST_ADD_CHECK_LIVE)))
3402
        goto cleanup;
3403 3404
    newDef = NULL;
    def = virNetworkObjGetDef(obj);
3405
    virNetworkObjSetActive(obj, true);
3406

3407
    event = virNetworkEventLifecycleNew(def->name, def->uuid,
3408 3409
                                        VIR_NETWORK_EVENT_STARTED,
                                        0);
3410

3411
    net = virGetNetwork(conn, def->name, def->uuid);
3412

3413
 cleanup:
3414
    virNetworkDefFree(newDef);
3415
    testObjectEventQueue(privconn, event);
3416 3417
    virNetworkObjEndAPI(&obj);
    return net;
3418 3419
}

3420 3421 3422 3423

static virNetworkPtr
testNetworkDefineXML(virConnectPtr conn,
                     const char *xml)
3424
{
3425
    testDriverPtr privconn = conn->privateData;
3426
    virNetworkDefPtr newDef;
3427
    virNetworkObjPtr obj = NULL;
3428
    virNetworkDefPtr def;
3429
    virNetworkPtr net = NULL;
3430
    virObjectEventPtr event = NULL;
3431

3432
    if ((newDef = virNetworkDefParseString(xml)) == NULL)
3433
        goto cleanup;
3434

3435
    if (!(obj = virNetworkObjAssignDef(privconn->networks, newDef, 0)))
3436
        goto cleanup;
3437 3438
    newDef = NULL;
    def = virNetworkObjGetDef(obj);
3439

3440
    event = virNetworkEventLifecycleNew(def->name, def->uuid,
3441 3442
                                        VIR_NETWORK_EVENT_DEFINED,
                                        0);
3443

3444
    net = virGetNetwork(conn, def->name, def->uuid);
3445

3446
 cleanup:
3447
    virNetworkDefFree(newDef);
3448
    testObjectEventQueue(privconn, event);
3449 3450
    virNetworkObjEndAPI(&obj);
    return net;
3451 3452
}

3453 3454

static int
3455
testNetworkUndefine(virNetworkPtr net)
3456
{
3457 3458
    testDriverPtr privconn = net->conn->privateData;
    virNetworkObjPtr obj;
3459
    int ret = -1;
3460
    virObjectEventPtr event = NULL;
3461

3462
    if (!(obj = testNetworkObjFindByName(privconn, net->name)))
3463
        goto cleanup;
3464

3465
    if (virNetworkObjIsActive(obj)) {
3466
        virReportError(VIR_ERR_OPERATION_INVALID,
3467
                       _("Network '%s' is still running"), net->name);
3468
        goto cleanup;
3469 3470
    }

3471
    event = virNetworkEventLifecycleNew(net->name, net->uuid,
3472 3473
                                        VIR_NETWORK_EVENT_UNDEFINED,
                                        0);
3474

3475
    virNetworkObjRemoveInactive(privconn->networks, obj);
3476
    ret = 0;
3477

3478
 cleanup:
3479
    testObjectEventQueue(privconn, event);
3480
    virNetworkObjEndAPI(&obj);
3481
    return ret;
3482 3483
}

3484

3485 3486 3487 3488 3489 3490 3491 3492
static int
testNetworkUpdate(virNetworkPtr net,
                  unsigned int command,
                  unsigned int section,
                  int parentIndex,
                  const char *xml,
                  unsigned int flags)
{
3493
    testDriverPtr privconn = net->conn->privateData;
3494
    virNetworkObjPtr obj = NULL;
3495 3496 3497 3498 3499 3500
    int isActive, ret = -1;

    virCheckFlags(VIR_NETWORK_UPDATE_AFFECT_LIVE |
                  VIR_NETWORK_UPDATE_AFFECT_CONFIG,
                  -1);

3501
    if (!(obj = testNetworkObjFindByUUID(privconn, net->uuid)))
3502 3503 3504 3505 3506
        goto cleanup;

    /* VIR_NETWORK_UPDATE_AFFECT_CURRENT means "change LIVE if network
     * is active, else change CONFIG
    */
3507
    isActive = virNetworkObjIsActive(obj);
3508 3509 3510 3511 3512 3513 3514 3515 3516 3517
    if ((flags & (VIR_NETWORK_UPDATE_AFFECT_LIVE
                   | VIR_NETWORK_UPDATE_AFFECT_CONFIG)) ==
        VIR_NETWORK_UPDATE_AFFECT_CURRENT) {
        if (isActive)
            flags |= VIR_NETWORK_UPDATE_AFFECT_LIVE;
        else
            flags |= VIR_NETWORK_UPDATE_AFFECT_CONFIG;
    }

    /* update the network config in memory/on disk */
3518
    if (virNetworkObjUpdate(obj, command, section, parentIndex, xml, flags) < 0)
3519 3520 3521
       goto cleanup;

    ret = 0;
3522
 cleanup:
3523
    virNetworkObjEndAPI(&obj);
3524 3525 3526
    return ret;
}

3527 3528

static int
3529
testNetworkCreate(virNetworkPtr net)
3530
{
3531 3532
    testDriverPtr privconn = net->conn->privateData;
    virNetworkObjPtr obj;
3533
    virNetworkDefPtr def;
3534
    int ret = -1;
3535
    virObjectEventPtr event = NULL;
3536

3537
    if (!(obj = testNetworkObjFindByName(privconn, net->name)))
3538
        goto cleanup;
3539
    def = virNetworkObjGetDef(obj);
3540

3541
    if (virNetworkObjIsActive(obj)) {
3542
        virReportError(VIR_ERR_OPERATION_INVALID,
3543
                       _("Network '%s' is already running"), net->name);
3544
        goto cleanup;
3545 3546
    }

3547
    virNetworkObjSetActive(obj, true);
3548
    event = virNetworkEventLifecycleNew(def->name, def->uuid,
3549 3550
                                        VIR_NETWORK_EVENT_STARTED,
                                        0);
3551
    ret = 0;
3552

3553
 cleanup:
3554
    testObjectEventQueue(privconn, event);
3555
    virNetworkObjEndAPI(&obj);
3556
    return ret;
3557 3558
}

3559 3560

static int
3561
testNetworkDestroy(virNetworkPtr net)
3562
{
3563 3564
    testDriverPtr privconn = net->conn->privateData;
    virNetworkObjPtr obj;
3565
    virNetworkDefPtr def;
3566
    int ret = -1;
3567
    virObjectEventPtr event = NULL;
3568

3569
    if (!(obj = testNetworkObjFindByName(privconn, net->name)))
3570
        goto cleanup;
3571
    def = virNetworkObjGetDef(obj);
3572

3573
    virNetworkObjSetActive(obj, false);
3574
    event = virNetworkEventLifecycleNew(def->name, def->uuid,
3575 3576
                                        VIR_NETWORK_EVENT_STOPPED,
                                        0);
3577
    if (!virNetworkObjIsPersistent(obj))
3578
        virNetworkObjRemoveInactive(privconn->networks, obj);
3579

3580 3581
    ret = 0;

3582
 cleanup:
3583
    testObjectEventQueue(privconn, event);
3584
    virNetworkObjEndAPI(&obj);
3585
    return ret;
3586 3587
}

3588 3589

static char *
3590
testNetworkGetXMLDesc(virNetworkPtr net,
3591
                      unsigned int flags)
3592
{
3593 3594
    testDriverPtr privconn = net->conn->privateData;
    virNetworkObjPtr obj;
3595
    char *ret = NULL;
3596

E
Eric Blake 已提交
3597 3598
    virCheckFlags(0, NULL);

3599
    if (!(obj = testNetworkObjFindByName(privconn, net->name)))
3600
        goto cleanup;
3601

3602
    ret = virNetworkDefFormat(virNetworkObjGetDef(obj), flags);
3603

3604
 cleanup:
3605
    virNetworkObjEndAPI(&obj);
3606
    return ret;
3607 3608
}

3609 3610

static char *
3611
testNetworkGetBridgeName(virNetworkPtr net)
3612
{
3613
    testDriverPtr privconn = net->conn->privateData;
3614
    char *bridge = NULL;
3615
    virNetworkObjPtr obj;
3616
    virNetworkDefPtr def;
3617

3618
    if (!(obj = testNetworkObjFindByName(privconn, net->name)))
3619
        goto cleanup;
3620
    def = virNetworkObjGetDef(obj);
3621

3622
    if (!(def->bridge)) {
3623 3624
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("network '%s' does not have a bridge name."),
3625
                       def->name);
3626 3627 3628
        goto cleanup;
    }

3629
    ignore_value(VIR_STRDUP(bridge, def->bridge));
3630

3631
 cleanup:
3632
    virNetworkObjEndAPI(&obj);
3633 3634 3635
    return bridge;
}

3636 3637

static int
3638
testNetworkGetAutostart(virNetworkPtr net,
3639
                        int *autostart)
3640
{
3641 3642
    testDriverPtr privconn = net->conn->privateData;
    virNetworkObjPtr obj;
3643
    int ret = -1;
3644

3645
    if (!(obj = testNetworkObjFindByName(privconn, net->name)))
3646
        goto cleanup;
3647

3648
    *autostart = virNetworkObjIsAutostart(obj) ? 1 : 0;
3649 3650
    ret = 0;

3651
 cleanup:
3652
    virNetworkObjEndAPI(&obj);
3653
    return ret;
3654 3655
}

3656 3657

static int
3658
testNetworkSetAutostart(virNetworkPtr net,
3659
                        int autostart)
3660
{
3661 3662
    testDriverPtr privconn = net->conn->privateData;
    virNetworkObjPtr obj;
3663
    bool new_autostart = (autostart != 0);
3664
    int ret = -1;
3665

3666
    if (!(obj = testNetworkObjFindByName(privconn, net->name)))
3667
        goto cleanup;
3668

3669 3670
    virNetworkObjSetAutostart(obj, new_autostart);

3671 3672
    ret = 0;

3673
 cleanup:
3674
    virNetworkObjEndAPI(&obj);
3675
    return ret;
3676
}
3677

C
Cole Robinson 已提交
3678

L
Laine Stump 已提交
3679 3680 3681 3682 3683
/*
 * Physical host interface routines
 */


3684 3685 3686 3687
static virInterfaceObjPtr
testInterfaceObjFindByName(testDriverPtr privconn,
                           const char *name)
{
3688
    virInterfaceObjPtr obj;
3689 3690

    testDriverLock(privconn);
3691
    obj = virInterfaceObjListFindByName(privconn->ifaces, name);
3692 3693
    testDriverUnlock(privconn);

3694
    if (!obj)
3695 3696 3697 3698
        virReportError(VIR_ERR_NO_INTERFACE,
                       _("no interface with matching name '%s'"),
                       name);

3699
    return obj;
3700 3701 3702
}


3703 3704
static int
testConnectNumOfInterfaces(virConnectPtr conn)
L
Laine Stump 已提交
3705
{
3706
    testDriverPtr privconn = conn->privateData;
3707
    int ninterfaces;
L
Laine Stump 已提交
3708 3709

    testDriverLock(privconn);
3710
    ninterfaces = virInterfaceObjListNumOfInterfaces(privconn->ifaces, true);
L
Laine Stump 已提交
3711
    testDriverUnlock(privconn);
3712
    return ninterfaces;
L
Laine Stump 已提交
3713 3714
}

3715 3716 3717 3718 3719

static int
testConnectListInterfaces(virConnectPtr conn,
                          char **const names,
                          int maxnames)
L
Laine Stump 已提交
3720
{
3721
    testDriverPtr privconn = conn->privateData;
3722
    int nnames;
L
Laine Stump 已提交
3723 3724

    testDriverLock(privconn);
3725 3726
    nnames = virInterfaceObjListGetNames(privconn->ifaces, true,
                                         names, maxnames);
L
Laine Stump 已提交
3727 3728
    testDriverUnlock(privconn);

3729
    return nnames;
L
Laine Stump 已提交
3730 3731
}

3732 3733 3734

static int
testConnectNumOfDefinedInterfaces(virConnectPtr conn)
L
Laine Stump 已提交
3735
{
3736
    testDriverPtr privconn = conn->privateData;
3737
    int ninterfaces;
L
Laine Stump 已提交
3738 3739

    testDriverLock(privconn);
3740
    ninterfaces = virInterfaceObjListNumOfInterfaces(privconn->ifaces, false);
L
Laine Stump 已提交
3741
    testDriverUnlock(privconn);
3742
    return ninterfaces;
L
Laine Stump 已提交
3743 3744
}

3745 3746 3747 3748 3749

static int
testConnectListDefinedInterfaces(virConnectPtr conn,
                                 char **const names,
                                 int maxnames)
L
Laine Stump 已提交
3750
{
3751
    testDriverPtr privconn = conn->privateData;
3752
    int nnames;
L
Laine Stump 已提交
3753 3754

    testDriverLock(privconn);
3755 3756
    nnames = virInterfaceObjListGetNames(privconn->ifaces, false,
                                         names, maxnames);
L
Laine Stump 已提交
3757 3758
    testDriverUnlock(privconn);

3759
    return nnames;
L
Laine Stump 已提交
3760 3761
}

3762 3763 3764 3765

static virInterfacePtr
testInterfaceLookupByName(virConnectPtr conn,
                          const char *name)
L
Laine Stump 已提交
3766
{
3767
    testDriverPtr privconn = conn->privateData;
3768
    virInterfaceObjPtr obj;
3769
    virInterfaceDefPtr def;
L
Laine Stump 已提交
3770 3771
    virInterfacePtr ret = NULL;

3772
    if (!(obj = testInterfaceObjFindByName(privconn, name)))
3773
        return NULL;
3774
    def = virInterfaceObjGetDef(obj);
L
Laine Stump 已提交
3775

3776
    ret = virGetInterface(conn, def->name, def->mac);
L
Laine Stump 已提交
3777

3778
    virInterfaceObjEndAPI(&obj);
L
Laine Stump 已提交
3779 3780 3781
    return ret;
}

3782 3783 3784 3785

static virInterfacePtr
testInterfaceLookupByMACString(virConnectPtr conn,
                               const char *mac)
L
Laine Stump 已提交
3786
{
3787
    testDriverPtr privconn = conn->privateData;
L
Laine Stump 已提交
3788
    int ifacect;
3789
    char *ifacenames[] = { NULL, NULL };
L
Laine Stump 已提交
3790 3791 3792
    virInterfacePtr ret = NULL;

    testDriverLock(privconn);
3793 3794
    ifacect = virInterfaceObjListFindByMACString(privconn->ifaces, mac,
                                                 ifacenames, 2);
L
Laine Stump 已提交
3795 3796 3797
    testDriverUnlock(privconn);

    if (ifacect == 0) {
3798 3799
        virReportError(VIR_ERR_NO_INTERFACE,
                       _("no interface with matching mac '%s'"), mac);
L
Laine Stump 已提交
3800 3801 3802 3803
        goto cleanup;
    }

    if (ifacect > 1) {
3804
        virReportError(VIR_ERR_MULTIPLE_INTERFACES, NULL);
L
Laine Stump 已提交
3805 3806 3807
        goto cleanup;
    }

3808
    ret = virGetInterface(conn, ifacenames[0], mac);
L
Laine Stump 已提交
3809

3810
 cleanup:
3811 3812
    VIR_FREE(ifacenames[0]);
    VIR_FREE(ifacenames[1]);
L
Laine Stump 已提交
3813 3814 3815
    return ret;
}

3816 3817 3818

static int
testInterfaceIsActive(virInterfacePtr iface)
3819
{
3820
    testDriverPtr privconn = iface->conn->privateData;
3821 3822 3823
    virInterfaceObjPtr obj;
    int ret = -1;

3824
    if (!(obj = testInterfaceObjFindByName(privconn, iface->name)))
3825
        return -1;
3826

3827 3828
    ret = virInterfaceObjIsActive(obj);

3829
    virInterfaceObjEndAPI(&obj);
3830 3831 3832
    return ret;
}

3833 3834 3835 3836

static int
testInterfaceChangeBegin(virConnectPtr conn,
                         unsigned int flags)
3837
{
3838
    testDriverPtr privconn = conn->privateData;
3839 3840
    int ret = -1;

E
Eric Blake 已提交
3841 3842
    virCheckFlags(0, -1);

3843 3844
    testDriverLock(privconn);
    if (privconn->transaction_running) {
3845
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
3846
                       _("there is another transaction running."));
3847 3848 3849 3850 3851
        goto cleanup;
    }

    privconn->transaction_running = true;

3852
    if (!(privconn->backupIfaces = virInterfaceObjListClone(privconn->ifaces)))
3853 3854 3855
        goto cleanup;

    ret = 0;
3856
 cleanup:
3857 3858 3859 3860
    testDriverUnlock(privconn);
    return ret;
}

3861 3862 3863 3864

static int
testInterfaceChangeCommit(virConnectPtr conn,
                          unsigned int flags)
3865
{
3866
    testDriverPtr privconn = conn->privateData;
3867 3868
    int ret = -1;

E
Eric Blake 已提交
3869 3870
    virCheckFlags(0, -1);

3871 3872 3873
    testDriverLock(privconn);

    if (!privconn->transaction_running) {
3874
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
3875 3876
                       _("no transaction running, "
                         "nothing to be committed."));
3877 3878 3879
        goto cleanup;
    }

3880
    virInterfaceObjListFree(privconn->backupIfaces);
3881 3882 3883 3884
    privconn->transaction_running = false;

    ret = 0;

3885
 cleanup:
3886 3887 3888 3889 3890
    testDriverUnlock(privconn);

    return ret;
}

3891 3892 3893 3894

static int
testInterfaceChangeRollback(virConnectPtr conn,
                            unsigned int flags)
3895
{
3896
    testDriverPtr privconn = conn->privateData;
3897 3898
    int ret = -1;

E
Eric Blake 已提交
3899 3900
    virCheckFlags(0, -1);

3901 3902 3903
    testDriverLock(privconn);

    if (!privconn->transaction_running) {
3904
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
3905 3906
                       _("no transaction running, "
                         "nothing to rollback."));
3907 3908 3909
        goto cleanup;
    }

3910 3911 3912
    virInterfaceObjListFree(privconn->ifaces);
    privconn->ifaces = privconn->backupIfaces;
    privconn->backupIfaces = NULL;
3913 3914 3915 3916 3917

    privconn->transaction_running = false;

    ret = 0;

3918
 cleanup:
3919 3920 3921
    testDriverUnlock(privconn);
    return ret;
}
3922

3923 3924 3925 3926

static char *
testInterfaceGetXMLDesc(virInterfacePtr iface,
                        unsigned int flags)
L
Laine Stump 已提交
3927
{
3928
    testDriverPtr privconn = iface->conn->privateData;
3929
    virInterfaceObjPtr obj;
3930
    virInterfaceDefPtr def;
L
Laine Stump 已提交
3931 3932
    char *ret = NULL;

E
Eric Blake 已提交
3933 3934
    virCheckFlags(0, NULL);

3935
    if (!(obj = testInterfaceObjFindByName(privconn, iface->name)))
3936
        return NULL;
3937
    def = virInterfaceObjGetDef(obj);
L
Laine Stump 已提交
3938

3939
    ret = virInterfaceDefFormat(def);
L
Laine Stump 已提交
3940

3941
    virInterfaceObjEndAPI(&obj);
L
Laine Stump 已提交
3942 3943 3944 3945
    return ret;
}


3946 3947 3948 3949
static virInterfacePtr
testInterfaceDefineXML(virConnectPtr conn,
                       const char *xmlStr,
                       unsigned int flags)
L
Laine Stump 已提交
3950
{
3951
    testDriverPtr privconn = conn->privateData;
L
Laine Stump 已提交
3952
    virInterfaceDefPtr def;
3953
    virInterfaceObjPtr obj = NULL;
3954
    virInterfaceDefPtr objdef;
L
Laine Stump 已提交
3955 3956
    virInterfacePtr ret = NULL;

E
Eric Blake 已提交
3957 3958
    virCheckFlags(0, NULL);

L
Laine Stump 已提交
3959
    testDriverLock(privconn);
3960
    if ((def = virInterfaceDefParseString(xmlStr)) == NULL)
L
Laine Stump 已提交
3961 3962
        goto cleanup;

3963
    if ((obj = virInterfaceObjListAssignDef(privconn->ifaces, def)) == NULL)
L
Laine Stump 已提交
3964 3965
        goto cleanup;
    def = NULL;
3966
    objdef = virInterfaceObjGetDef(obj);
L
Laine Stump 已提交
3967

3968
    ret = virGetInterface(conn, objdef->name, objdef->mac);
L
Laine Stump 已提交
3969

3970
 cleanup:
L
Laine Stump 已提交
3971
    virInterfaceDefFree(def);
3972
    virInterfaceObjEndAPI(&obj);
L
Laine Stump 已提交
3973 3974 3975 3976
    testDriverUnlock(privconn);
    return ret;
}

3977 3978 3979

static int
testInterfaceUndefine(virInterfacePtr iface)
L
Laine Stump 已提交
3980
{
3981
    testDriverPtr privconn = iface->conn->privateData;
3982
    virInterfaceObjPtr obj;
L
Laine Stump 已提交
3983

3984
    if (!(obj = testInterfaceObjFindByName(privconn, iface->name)))
3985
        return -1;
L
Laine Stump 已提交
3986

3987
    virInterfaceObjListRemove(privconn->ifaces, obj);
3988
    virObjectUnref(obj);
L
Laine Stump 已提交
3989

3990
    return 0;
L
Laine Stump 已提交
3991 3992
}

3993 3994 3995 3996

static int
testInterfaceCreate(virInterfacePtr iface,
                    unsigned int flags)
L
Laine Stump 已提交
3997
{
3998
    testDriverPtr privconn = iface->conn->privateData;
3999
    virInterfaceObjPtr obj;
L
Laine Stump 已提交
4000 4001
    int ret = -1;

E
Eric Blake 已提交
4002 4003
    virCheckFlags(0, -1);

4004
    if (!(obj = testInterfaceObjFindByName(privconn, iface->name)))
4005
        return -1;
L
Laine Stump 已提交
4006

4007
    if (virInterfaceObjIsActive(obj)) {
4008
        virReportError(VIR_ERR_OPERATION_INVALID, NULL);
L
Laine Stump 已提交
4009 4010 4011
        goto cleanup;
    }

4012
    virInterfaceObjSetActive(obj, true);
L
Laine Stump 已提交
4013 4014
    ret = 0;

4015
 cleanup:
4016
    virInterfaceObjEndAPI(&obj);
L
Laine Stump 已提交
4017 4018 4019
    return ret;
}

4020 4021 4022 4023

static int
testInterfaceDestroy(virInterfacePtr iface,
                     unsigned int flags)
L
Laine Stump 已提交
4024
{
4025
    testDriverPtr privconn = iface->conn->privateData;
4026
    virInterfaceObjPtr obj;
L
Laine Stump 已提交
4027 4028
    int ret = -1;

E
Eric Blake 已提交
4029 4030
    virCheckFlags(0, -1);

4031
    if (!(obj = testInterfaceObjFindByName(privconn, iface->name)))
4032
        return -1;
L
Laine Stump 已提交
4033

4034
    if (!virInterfaceObjIsActive(obj)) {
4035
        virReportError(VIR_ERR_OPERATION_INVALID, NULL);
L
Laine Stump 已提交
4036 4037 4038
        goto cleanup;
    }

4039
    virInterfaceObjSetActive(obj, false);
L
Laine Stump 已提交
4040 4041
    ret = 0;

4042
 cleanup:
4043
    virInterfaceObjEndAPI(&obj);
L
Laine Stump 已提交
4044 4045 4046 4047 4048
    return ret;
}



C
Cole Robinson 已提交
4049 4050 4051 4052
/*
 * Storage Driver routines
 */

4053
static int
4054
testStoragePoolObjSetDefaults(virStoragePoolObjPtr obj)
4055
{
4056
    char *configFile;
4057
    virStoragePoolDefPtr def = virStoragePoolObjGetDef(obj);
C
Cole Robinson 已提交
4058

4059 4060 4061
    def->capacity = defaultPoolCap;
    def->allocation = defaultPoolAlloc;
    def->available = defaultPoolCap - defaultPoolAlloc;
C
Cole Robinson 已提交
4062

4063 4064 4065 4066 4067
    if (VIR_STRDUP(configFile, "") < 0)
        return -1;

    virStoragePoolObjSetConfigFile(obj, configFile);
    return 0;
C
Cole Robinson 已提交
4068 4069
}

4070

4071 4072 4073 4074
static virStoragePoolObjPtr
testStoragePoolObjFindByName(testDriverPtr privconn,
                             const char *name)
{
4075
    virStoragePoolObjPtr obj;
4076 4077

    testDriverLock(privconn);
4078
    obj = virStoragePoolObjFindByName(&privconn->pools, name);
4079 4080
    testDriverUnlock(privconn);

4081
    if (!obj)
4082 4083 4084 4085
        virReportError(VIR_ERR_NO_STORAGE_POOL,
                       _("no storage pool with matching name '%s'"),
                       name);

4086
    return obj;
4087 4088 4089
}


4090 4091 4092 4093 4094 4095 4096 4097 4098 4099 4100 4101 4102 4103 4104 4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119 4120 4121 4122 4123 4124 4125 4126 4127 4128 4129
static virStoragePoolObjPtr
testStoragePoolObjFindActiveByName(testDriverPtr privconn,
                                   const char *name)
{
    virStoragePoolObjPtr obj;

    if (!(obj = testStoragePoolObjFindByName(privconn, name)))
        return NULL;

    if (!virStoragePoolObjIsActive(obj)) {
        virReportError(VIR_ERR_OPERATION_INVALID,
                       _("storage pool '%s' is not active"), name);
        virStoragePoolObjUnlock(obj);
        return NULL;
    }

    return obj;
}


static virStoragePoolObjPtr
testStoragePoolObjFindInactiveByName(testDriverPtr privconn,
                                     const char *name)
{
    virStoragePoolObjPtr obj;

    if (!(obj = testStoragePoolObjFindByName(privconn, name)))
        return NULL;

    if (virStoragePoolObjIsActive(obj)) {
        virReportError(VIR_ERR_OPERATION_INVALID,
                       _("storage pool '%s' is active"), name);
        virStoragePoolObjUnlock(obj);
        return NULL;
    }

    return obj;
}


4130 4131 4132 4133
static virStoragePoolObjPtr
testStoragePoolObjFindByUUID(testDriverPtr privconn,
                             const unsigned char *uuid)
{
4134
    virStoragePoolObjPtr obj;
4135 4136 4137
    char uuidstr[VIR_UUID_STRING_BUFLEN];

    testDriverLock(privconn);
4138
    obj = virStoragePoolObjFindByUUID(&privconn->pools, uuid);
4139 4140
    testDriverUnlock(privconn);

4141
    if (!obj) {
4142 4143 4144 4145 4146 4147
        virUUIDFormat(uuid, uuidstr);
        virReportError(VIR_ERR_NO_STORAGE_POOL,
                       _("no storage pool with matching uuid '%s'"),
                       uuidstr);
    }

4148
    return obj;
4149 4150 4151
}


C
Cole Robinson 已提交
4152 4153
static virStoragePoolPtr
testStoragePoolLookupByUUID(virConnectPtr conn,
4154 4155
                            const unsigned char *uuid)
{
4156
    testDriverPtr privconn = conn->privateData;
4157
    virStoragePoolObjPtr obj;
4158
    virStoragePoolDefPtr def;
4159
    virStoragePoolPtr pool = NULL;
C
Cole Robinson 已提交
4160

4161
    if (!(obj = testStoragePoolObjFindByUUID(privconn, uuid)))
4162
        return NULL;
4163
    def = virStoragePoolObjGetDef(obj);
C
Cole Robinson 已提交
4164

4165
    pool = virGetStoragePool(conn, def->name, def->uuid, NULL, NULL);
4166

4167
    virStoragePoolObjUnlock(obj);
4168
    return pool;
C
Cole Robinson 已提交
4169 4170
}

4171

C
Cole Robinson 已提交
4172 4173
static virStoragePoolPtr
testStoragePoolLookupByName(virConnectPtr conn,
4174 4175
                            const char *name)
{
4176
    testDriverPtr privconn = conn->privateData;
4177
    virStoragePoolObjPtr obj;
4178
    virStoragePoolDefPtr def;
4179
    virStoragePoolPtr pool = NULL;
C
Cole Robinson 已提交
4180

4181
    if (!(obj = testStoragePoolObjFindByName(privconn, name)))
4182
        return NULL;
4183
    def = virStoragePoolObjGetDef(obj);
C
Cole Robinson 已提交
4184

4185
    pool = virGetStoragePool(conn, def->name, def->uuid, NULL, NULL);
4186

4187
    virStoragePoolObjUnlock(obj);
4188
    return pool;
C
Cole Robinson 已提交
4189 4190
}

4191

C
Cole Robinson 已提交
4192
static virStoragePoolPtr
4193 4194
testStoragePoolLookupByVolume(virStorageVolPtr vol)
{
C
Cole Robinson 已提交
4195 4196 4197
    return testStoragePoolLookupByName(vol->conn, vol->pool);
}

4198

C
Cole Robinson 已提交
4199
static int
4200 4201
testConnectNumOfStoragePools(virConnectPtr conn)
{
4202
    testDriverPtr privconn = conn->privateData;
4203
    int numActive = 0;
C
Cole Robinson 已提交
4204

4205
    testDriverLock(privconn);
4206 4207
    numActive = virStoragePoolObjNumOfStoragePools(&privconn->pools, conn,
                                                   true, NULL);
4208
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
4209 4210 4211 4212

    return numActive;
}

4213

C
Cole Robinson 已提交
4214
static int
4215 4216
testConnectListStoragePools(virConnectPtr conn,
                            char **const names,
4217
                            int maxnames)
4218
{
4219
    testDriverPtr privconn = conn->privateData;
4220
    int n = 0;
C
Cole Robinson 已提交
4221

4222
    testDriverLock(privconn);
4223 4224
    n = virStoragePoolObjGetNames(&privconn->pools, conn, true, NULL,
                                  names, maxnames);
4225
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
4226 4227 4228 4229

    return n;
}

4230

C
Cole Robinson 已提交
4231
static int
4232 4233
testConnectNumOfDefinedStoragePools(virConnectPtr conn)
{
4234
    testDriverPtr privconn = conn->privateData;
4235
    int numInactive = 0;
C
Cole Robinson 已提交
4236

4237
    testDriverLock(privconn);
4238 4239
    numInactive = virStoragePoolObjNumOfStoragePools(&privconn->pools, conn,
                                                     false, NULL);
4240
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
4241 4242 4243 4244

    return numInactive;
}

4245

C
Cole Robinson 已提交
4246
static int
4247 4248
testConnectListDefinedStoragePools(virConnectPtr conn,
                                   char **const names,
4249
                                   int maxnames)
4250
{
4251
    testDriverPtr privconn = conn->privateData;
4252
    int n = 0;
C
Cole Robinson 已提交
4253

4254
    testDriverLock(privconn);
4255 4256
    n = virStoragePoolObjGetNames(&privconn->pools, conn, false, NULL,
                                  names, maxnames);
4257
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
4258 4259 4260 4261

    return n;
}

4262

4263
static int
4264 4265 4266
testConnectListAllStoragePools(virConnectPtr conn,
                               virStoragePoolPtr **pools,
                               unsigned int flags)
4267
{
4268
    testDriverPtr privconn = conn->privateData;
4269 4270 4271 4272 4273
    int ret = -1;

    virCheckFlags(VIR_CONNECT_LIST_STORAGE_POOLS_FILTERS_ALL, -1);

    testDriverLock(privconn);
J
John Ferlan 已提交
4274
    ret = virStoragePoolObjListExport(conn, &privconn->pools, pools,
4275
                                      NULL, flags);
4276 4277 4278 4279
    testDriverUnlock(privconn);

    return ret;
}
C
Cole Robinson 已提交
4280

4281 4282 4283

static int
testStoragePoolIsActive(virStoragePoolPtr pool)
4284
{
4285
    testDriverPtr privconn = pool->conn->privateData;
4286 4287 4288
    virStoragePoolObjPtr obj;
    int ret = -1;

4289
    if (!(obj = testStoragePoolObjFindByUUID(privconn, pool->uuid)))
4290
        goto cleanup;
4291

4292 4293
    ret = virStoragePoolObjIsActive(obj);

4294
 cleanup:
4295 4296 4297 4298 4299
    if (obj)
        virStoragePoolObjUnlock(obj);
    return ret;
}

4300 4301 4302

static int
testStoragePoolIsPersistent(virStoragePoolPtr pool)
4303
{
4304
    testDriverPtr privconn = pool->conn->privateData;
4305 4306 4307
    virStoragePoolObjPtr obj;
    int ret = -1;

4308
    if (!(obj = testStoragePoolObjFindByUUID(privconn, pool->uuid)))
4309
        return -1;
4310

4311
    ret = virStoragePoolObjGetConfigFile(obj) ? 1 : 0;
4312

4313
    virStoragePoolObjUnlock(obj);
4314 4315 4316 4317
    return ret;
}


C
Cole Robinson 已提交
4318
static int
4319 4320
testStoragePoolCreate(virStoragePoolPtr pool,
                      unsigned int flags)
E
Eric Blake 已提交
4321
{
4322
    testDriverPtr privconn = pool->conn->privateData;
4323
    virStoragePoolObjPtr obj;
4324
    virObjectEventPtr event = NULL;
4325

E
Eric Blake 已提交
4326 4327
    virCheckFlags(0, -1);

4328 4329
    if (!(obj = testStoragePoolObjFindInactiveByName(privconn, pool->name)))
        return -1;
C
Cole Robinson 已提交
4330

4331
    virStoragePoolObjSetActive(obj, true);
4332 4333 4334 4335

    event = virStoragePoolEventLifecycleNew(pool->name, pool->uuid,
                                            VIR_STORAGE_POOL_EVENT_STARTED,
                                            0);
C
Cole Robinson 已提交
4336

4337
    testObjectEventQueue(privconn, event);
4338 4339
    virStoragePoolObjUnlock(obj);
    return 0;
C
Cole Robinson 已提交
4340 4341
}

4342

C
Cole Robinson 已提交
4343
static char *
4344 4345 4346 4347
testConnectFindStoragePoolSources(virConnectPtr conn ATTRIBUTE_UNUSED,
                                  const char *type,
                                  const char *srcSpec,
                                  unsigned int flags)
C
Cole Robinson 已提交
4348
{
4349 4350 4351 4352
    virStoragePoolSourcePtr source = NULL;
    int pool_type;
    char *ret = NULL;

E
Eric Blake 已提交
4353 4354
    virCheckFlags(0, NULL);

4355 4356
    pool_type = virStoragePoolTypeFromString(type);
    if (!pool_type) {
4357 4358
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("unknown storage pool type %s"), type);
4359 4360 4361 4362
        goto cleanup;
    }

    if (srcSpec) {
4363
        source = virStoragePoolDefParseSourceString(srcSpec, pool_type);
4364 4365 4366 4367 4368 4369 4370
        if (!source)
            goto cleanup;
    }

    switch (pool_type) {

    case VIR_STORAGE_POOL_LOGICAL:
4371
        ignore_value(VIR_STRDUP(ret, defaultPoolSourcesLogicalXML));
4372 4373 4374
        break;

    case VIR_STORAGE_POOL_NETFS:
4375
        if (!source || !source->hosts[0].name) {
4376 4377
            virReportError(VIR_ERR_INVALID_ARG,
                           "%s", _("hostname must be specified for netfs sources"));
4378 4379 4380
            goto cleanup;
        }

4381 4382
        ignore_value(virAsprintf(&ret, defaultPoolSourcesNetFSXML,
                                 source->hosts[0].name));
4383 4384 4385
        break;

    default:
4386 4387
        virReportError(VIR_ERR_NO_SUPPORT,
                       _("pool type '%s' does not support source discovery"), type);
4388 4389
    }

4390
 cleanup:
4391 4392
    virStoragePoolSourceFree(source);
    return ret;
C
Cole Robinson 已提交
4393 4394 4395
}


4396 4397 4398 4399 4400 4401 4402 4403 4404 4405 4406 4407 4408 4409 4410 4411 4412 4413 4414 4415 4416 4417
static virNodeDeviceObjPtr
testNodeDeviceMockCreateVport(testDriverPtr driver,
                              const char *wwnn,
                              const char *wwpn);
static int
testCreateVport(testDriverPtr driver,
                const char *wwnn,
                const char *wwpn)
{
    virNodeDeviceObjPtr obj = NULL;
    /* The storage_backend_scsi createVport() will use the input adapter
     * fields parent name, parent_wwnn/parent_wwpn, or parent_fabric_wwn
     * in order to determine whether the provided parent can be used to
     * create a vHBA or will find "an available vport capable" to create
     * a vHBA. In order to do this, it uses the virVHBA* API's which traverse
     * the sysfs looking at various fields (rather than going via nodedev).
     *
     * Since the test environ doesn't have the sysfs for the storage pool
     * test, at least for now use the node device test infrastructure to
     * create the vHBA. In the long run the result is the same. */
    if (!(obj = testNodeDeviceMockCreateVport(driver, wwnn, wwpn)))
        return -1;
4418
    virNodeDeviceObjEndAPI(&obj);
4419 4420 4421 4422 4423

    return 0;
}


C
Cole Robinson 已提交
4424
static virStoragePoolPtr
4425 4426 4427
testStoragePoolCreateXML(virConnectPtr conn,
                         const char *xml,
                         unsigned int flags)
E
Eric Blake 已提交
4428
{
4429
    testDriverPtr privconn = conn->privateData;
4430
    virStoragePoolDefPtr newDef;
4431
    virStoragePoolObjPtr obj = NULL;
4432
    virStoragePoolDefPtr def;
4433
    virStoragePoolPtr pool = NULL;
4434
    virObjectEventPtr event = NULL;
C
Cole Robinson 已提交
4435

E
Eric Blake 已提交
4436 4437
    virCheckFlags(0, NULL);

4438
    testDriverLock(privconn);
4439
    if (!(newDef = virStoragePoolDefParseString(xml)))
4440
        goto cleanup;
C
Cole Robinson 已提交
4441

4442
    obj = virStoragePoolObjFindByUUID(&privconn->pools, newDef->uuid);
4443
    if (!obj)
4444
        obj = virStoragePoolObjFindByName(&privconn->pools, newDef->name);
4445
    if (obj) {
4446 4447
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("storage pool already exists"));
4448
        goto cleanup;
C
Cole Robinson 已提交
4449 4450
    }

4451
    if (!(obj = virStoragePoolObjAssignDef(&privconn->pools, newDef)))
4452
        goto cleanup;
4453 4454
    newDef = NULL;
    def = virStoragePoolObjGetDef(obj);
C
Cole Robinson 已提交
4455

4456
    if (def->source.adapter.type == VIR_STORAGE_ADAPTER_TYPE_FC_HOST) {
4457 4458 4459 4460 4461
        /* In the real code, we'd call virVHBAManageVport followed by
         * find_new_device, but we cannot do that here since we're not
         * mocking udev. The mock routine will copy an existing vHBA and
         * rename a few fields to mock that. */
        if (testCreateVport(privconn,
4462 4463
                            def->source.adapter.data.fchost.wwnn,
                            def->source.adapter.data.fchost.wwpn) < 0) {
4464 4465
            virStoragePoolObjRemove(&privconn->pools, obj);
            obj = NULL;
4466 4467 4468 4469
            goto cleanup;
        }
    }

4470 4471 4472
    if (testStoragePoolObjSetDefaults(obj) == -1) {
        virStoragePoolObjRemove(&privconn->pools, obj);
        obj = NULL;
4473
        goto cleanup;
C
Cole Robinson 已提交
4474
    }
4475 4476 4477 4478

    /* *SetDefaults fills this in for the persistent pools, but this
     * would be a transient pool so remove it; otherwise, the Destroy
     * code will not Remove the pool */
4479
    virStoragePoolObjSetConfigFile(obj, NULL);
4480

4481
    virStoragePoolObjSetActive(obj, true);
C
Cole Robinson 已提交
4482

4483
    event = virStoragePoolEventLifecycleNew(def->name, def->uuid,
4484 4485 4486
                                            VIR_STORAGE_POOL_EVENT_STARTED,
                                            0);

4487
    pool = virGetStoragePool(conn, def->name, def->uuid, NULL, NULL);
4488

4489
 cleanup:
4490
    virStoragePoolDefFree(newDef);
4491
    testObjectEventQueue(privconn, event);
4492 4493
    if (obj)
        virStoragePoolObjUnlock(obj);
4494
    testDriverUnlock(privconn);
4495
    return pool;
C
Cole Robinson 已提交
4496 4497
}

4498

C
Cole Robinson 已提交
4499
static virStoragePoolPtr
4500 4501 4502
testStoragePoolDefineXML(virConnectPtr conn,
                         const char *xml,
                         unsigned int flags)
E
Eric Blake 已提交
4503
{
4504
    testDriverPtr privconn = conn->privateData;
4505
    virStoragePoolDefPtr newDef;
4506
    virStoragePoolObjPtr obj = NULL;
4507
    virStoragePoolDefPtr def;
4508
    virStoragePoolPtr pool = NULL;
4509
    virObjectEventPtr event = NULL;
C
Cole Robinson 已提交
4510

E
Eric Blake 已提交
4511 4512
    virCheckFlags(0, NULL);

4513
    testDriverLock(privconn);
4514
    if (!(newDef = virStoragePoolDefParseString(xml)))
4515
        goto cleanup;
C
Cole Robinson 已提交
4516

4517 4518 4519
    newDef->capacity = defaultPoolCap;
    newDef->allocation = defaultPoolAlloc;
    newDef->available = defaultPoolCap - defaultPoolAlloc;
C
Cole Robinson 已提交
4520

4521
    if (!(obj = virStoragePoolObjAssignDef(&privconn->pools, newDef)))
4522
        goto cleanup;
4523 4524
    newDef = NULL;
    def = virStoragePoolObjGetDef(obj);
C
Cole Robinson 已提交
4525

4526
    event = virStoragePoolEventLifecycleNew(def->name, def->uuid,
4527 4528 4529
                                            VIR_STORAGE_POOL_EVENT_DEFINED,
                                            0);

4530 4531 4532
    if (testStoragePoolObjSetDefaults(obj) == -1) {
        virStoragePoolObjRemove(&privconn->pools, obj);
        obj = NULL;
4533
        goto cleanup;
C
Cole Robinson 已提交
4534 4535
    }

4536
    pool = virGetStoragePool(conn, def->name, def->uuid, NULL, NULL);
4537

4538
 cleanup:
4539
    virStoragePoolDefFree(newDef);
4540
    testObjectEventQueue(privconn, event);
4541 4542
    if (obj)
        virStoragePoolObjUnlock(obj);
4543
    testDriverUnlock(privconn);
4544
    return pool;
C
Cole Robinson 已提交
4545 4546
}

4547

C
Cole Robinson 已提交
4548
static int
4549 4550
testStoragePoolUndefine(virStoragePoolPtr pool)
{
4551
    testDriverPtr privconn = pool->conn->privateData;
4552
    virStoragePoolObjPtr obj;
4553
    virObjectEventPtr event = NULL;
4554

4555 4556
    if (!(obj = testStoragePoolObjFindInactiveByName(privconn, pool->name)))
        return -1;
C
Cole Robinson 已提交
4557

4558 4559 4560 4561
    event = virStoragePoolEventLifecycleNew(pool->name, pool->uuid,
                                            VIR_STORAGE_POOL_EVENT_UNDEFINED,
                                            0);

4562
    virStoragePoolObjRemove(&privconn->pools, obj);
C
Cole Robinson 已提交
4563

4564
    testObjectEventQueue(privconn, event);
4565
    return 0;
C
Cole Robinson 已提交
4566 4567
}

4568

C
Cole Robinson 已提交
4569
static int
4570
testStoragePoolBuild(virStoragePoolPtr pool,
E
Eric Blake 已提交
4571 4572
                     unsigned int flags)
{
4573
    testDriverPtr privconn = pool->conn->privateData;
4574
    virStoragePoolObjPtr obj;
4575
    virObjectEventPtr event = NULL;
4576

E
Eric Blake 已提交
4577 4578
    virCheckFlags(0, -1);

4579 4580
    if (!(obj = testStoragePoolObjFindInactiveByName(privconn, pool->name)))
        return -1;
C
Cole Robinson 已提交
4581

4582 4583 4584 4585
    event = virStoragePoolEventLifecycleNew(pool->name, pool->uuid,
                                            VIR_STORAGE_POOL_EVENT_CREATED,
                                            0);

4586
    virStoragePoolObjUnlock(obj);
4587 4588

    testObjectEventQueue(privconn, event);
4589
    return 0;
C
Cole Robinson 已提交
4590 4591 4592
}


4593 4594 4595 4596 4597 4598 4599 4600 4601 4602 4603 4604 4605 4606 4607
static int
testDestroyVport(testDriverPtr privconn,
                 const char *wwnn ATTRIBUTE_UNUSED,
                 const char *wwpn ATTRIBUTE_UNUSED)
{
    virNodeDeviceObjPtr obj = NULL;
    virObjectEventPtr event = NULL;

    /* NB: Cannot use virVHBAGetHostByWWN (yet) like the storage_backend_scsi
     * deleteVport() helper since that traverses the file system looking for
     * the wwnn/wwpn. So our choice short term is to cheat and use the name
     * (scsi_host12) we know was created.
     *
     * Reaching across the boundaries of space and time into the
     * Node Device in order to remove */
4608 4609
    if (!(obj = virNodeDeviceObjListFindByName(privconn->devs,
                                               "scsi_host12"))) {
4610 4611
        virReportError(VIR_ERR_NO_NODE_DEVICE, "%s",
                       _("no node device with matching name 'scsi_host12'"));
4612
        return -1;
4613 4614 4615 4616 4617 4618
    }

    event = virNodeDeviceEventLifecycleNew("scsi_host12",
                                           VIR_NODE_DEVICE_EVENT_DELETED,
                                           0);

4619
    virNodeDeviceObjListRemove(privconn->devs, obj);
4620
    virObjectUnref(obj);
4621 4622

    testObjectEventQueue(privconn, event);
4623
    return 0;
4624 4625 4626
}


C
Cole Robinson 已提交
4627
static int
4628 4629
testStoragePoolDestroy(virStoragePoolPtr pool)
{
4630
    testDriverPtr privconn = pool->conn->privateData;
4631
    virStoragePoolObjPtr obj;
4632
    virStoragePoolDefPtr def;
4633
    int ret = -1;
4634
    virObjectEventPtr event = NULL;
4635

4636
    if (!(obj = testStoragePoolObjFindActiveByName(privconn, pool->name)))
4637
        return -1;
4638
    def = virStoragePoolObjGetDef(obj);
4639

4640
    virStoragePoolObjSetActive(obj, false);
4641

4642
    if (def->source.adapter.type == VIR_STORAGE_ADAPTER_TYPE_FC_HOST) {
4643
        if (testDestroyVport(privconn,
4644 4645
                             def->source.adapter.data.fchost.wwnn,
                             def->source.adapter.data.fchost.wwpn) < 0)
4646 4647 4648
            goto cleanup;
    }

4649 4650
    event = virStoragePoolEventLifecycleNew(def->name,
                                            def->uuid,
4651 4652
                                            VIR_STORAGE_POOL_EVENT_STOPPED,
                                            0);
C
Cole Robinson 已提交
4653

4654
    if (!(virStoragePoolObjGetConfigFile(obj))) {
4655 4656
        virStoragePoolObjRemove(&privconn->pools, obj);
        obj = NULL;
4657
    }
4658
    ret = 0;
C
Cole Robinson 已提交
4659

4660
 cleanup:
4661
    testObjectEventQueue(privconn, event);
4662 4663
    if (obj)
        virStoragePoolObjUnlock(obj);
4664
    return ret;
C
Cole Robinson 已提交
4665 4666 4667 4668
}


static int
4669
testStoragePoolDelete(virStoragePoolPtr pool,
E
Eric Blake 已提交
4670 4671
                      unsigned int flags)
{
4672
    testDriverPtr privconn = pool->conn->privateData;
4673
    virStoragePoolObjPtr obj;
4674
    virObjectEventPtr event = NULL;
4675

E
Eric Blake 已提交
4676 4677
    virCheckFlags(0, -1);

4678 4679
    if (!(obj = testStoragePoolObjFindInactiveByName(privconn, pool->name)))
        return -1;
C
Cole Robinson 已提交
4680

4681 4682 4683 4684 4685 4686
    event = virStoragePoolEventLifecycleNew(pool->name, pool->uuid,
                                            VIR_STORAGE_POOL_EVENT_DELETED,
                                            0);

    testObjectEventQueue(privconn, event);

4687 4688
    virStoragePoolObjUnlock(obj);
    return 0;
C
Cole Robinson 已提交
4689 4690 4691 4692
}


static int
4693
testStoragePoolRefresh(virStoragePoolPtr pool,
E
Eric Blake 已提交
4694 4695
                       unsigned int flags)
{
4696
    testDriverPtr privconn = pool->conn->privateData;
4697
    virStoragePoolObjPtr obj;
4698
    virObjectEventPtr event = NULL;
4699

E
Eric Blake 已提交
4700 4701
    virCheckFlags(0, -1);

4702 4703
    if (!(obj = testStoragePoolObjFindActiveByName(privconn, pool->name)))
        return -1;
4704

4705
    event = virStoragePoolEventRefreshNew(pool->name, pool->uuid);
C
Cole Robinson 已提交
4706

4707
    testObjectEventQueue(privconn, event);
4708 4709
    virStoragePoolObjUnlock(obj);
    return 0;
C
Cole Robinson 已提交
4710 4711 4712 4713
}


static int
4714
testStoragePoolGetInfo(virStoragePoolPtr pool,
4715 4716
                       virStoragePoolInfoPtr info)
{
4717
    testDriverPtr privconn = pool->conn->privateData;
4718
    virStoragePoolObjPtr obj;
4719
    virStoragePoolDefPtr def;
4720

4721
    if (!(obj = testStoragePoolObjFindByName(privconn, pool->name)))
4722
        return -1;
4723
    def = virStoragePoolObjGetDef(obj);
C
Cole Robinson 已提交
4724 4725

    memset(info, 0, sizeof(virStoragePoolInfo));
4726
    if (virStoragePoolObjIsActive(obj))
C
Cole Robinson 已提交
4727 4728 4729
        info->state = VIR_STORAGE_POOL_RUNNING;
    else
        info->state = VIR_STORAGE_POOL_INACTIVE;
4730 4731 4732
    info->capacity = def->capacity;
    info->allocation = def->allocation;
    info->available = def->available;
C
Cole Robinson 已提交
4733

4734 4735
    virStoragePoolObjUnlock(obj);
    return 0;
C
Cole Robinson 已提交
4736 4737
}

4738

C
Cole Robinson 已提交
4739
static char *
4740
testStoragePoolGetXMLDesc(virStoragePoolPtr pool,
E
Eric Blake 已提交
4741 4742
                          unsigned int flags)
{
4743
    testDriverPtr privconn = pool->conn->privateData;
4744
    virStoragePoolObjPtr obj;
4745
    char *ret = NULL;
4746

E
Eric Blake 已提交
4747 4748
    virCheckFlags(0, NULL);

4749
    if (!(obj = testStoragePoolObjFindByName(privconn, pool->name)))
4750
        return NULL;
4751

4752
    ret = virStoragePoolDefFormat(virStoragePoolObjGetDef(obj));
4753

4754
    virStoragePoolObjUnlock(obj);
4755
    return ret;
C
Cole Robinson 已提交
4756 4757
}

4758

C
Cole Robinson 已提交
4759
static int
4760
testStoragePoolGetAutostart(virStoragePoolPtr pool,
4761 4762
                            int *autostart)
{
4763
    testDriverPtr privconn = pool->conn->privateData;
4764
    virStoragePoolObjPtr obj;
4765

4766
    if (!(obj = testStoragePoolObjFindByName(privconn, pool->name)))
4767
        return -1;
C
Cole Robinson 已提交
4768

4769
    if (!virStoragePoolObjGetConfigFile(obj))
C
Cole Robinson 已提交
4770
        *autostart = 0;
4771
    else
4772
        *autostart = virStoragePoolObjIsAutostart(obj) ? 1 : 0;
C
Cole Robinson 已提交
4773

4774 4775
    virStoragePoolObjUnlock(obj);
    return 0;
C
Cole Robinson 已提交
4776 4777
}

4778

C
Cole Robinson 已提交
4779
static int
4780
testStoragePoolSetAutostart(virStoragePoolPtr pool,
4781 4782
                            int autostart)
{
4783
    testDriverPtr privconn = pool->conn->privateData;
4784
    virStoragePoolObjPtr obj;
4785
    bool new_autostart = (autostart != 0);
4786
    int ret = -1;
4787

4788
    if (!(obj = testStoragePoolObjFindByName(privconn, pool->name)))
4789
        return -1;
C
Cole Robinson 已提交
4790

4791
    if (!virStoragePoolObjGetConfigFile(obj)) {
4792 4793
        virReportError(VIR_ERR_INVALID_ARG,
                       "%s", _("pool has no config file"));
4794
        goto cleanup;
C
Cole Robinson 已提交
4795 4796
    }

4797
    virStoragePoolObjSetAutostart(obj, new_autostart);
4798 4799
    ret = 0;

4800
 cleanup:
4801
    virStoragePoolObjUnlock(obj);
4802
    return ret;
C
Cole Robinson 已提交
4803 4804 4805 4806
}


static int
4807 4808
testStoragePoolNumOfVolumes(virStoragePoolPtr pool)
{
4809
    testDriverPtr privconn = pool->conn->privateData;
4810
    virStoragePoolObjPtr obj;
4811
    int ret = -1;
4812

4813 4814
    if (!(obj = testStoragePoolObjFindActiveByName(privconn, pool->name)))
        return -1;
C
Cole Robinson 已提交
4815

4816
    ret = virStoragePoolObjNumOfVolumes(obj, pool->conn, NULL);
4817

4818
    virStoragePoolObjUnlock(obj);
4819
    return ret;
C
Cole Robinson 已提交
4820 4821
}

4822

C
Cole Robinson 已提交
4823
static int
4824
testStoragePoolListVolumes(virStoragePoolPtr pool,
C
Cole Robinson 已提交
4825
                           char **const names,
4826 4827
                           int maxnames)
{
4828
    testDriverPtr privconn = pool->conn->privateData;
4829
    virStoragePoolObjPtr obj;
4830
    int n = -1;
4831

4832
    if (!(obj = testStoragePoolObjFindActiveByName(privconn, pool->name)))
4833
        return -1;
4834

4835
    n = virStoragePoolObjVolumeGetNames(obj, pool->conn, NULL, names, maxnames);
C
Cole Robinson 已提交
4836

4837
    virStoragePoolObjUnlock(obj);
C
Cole Robinson 已提交
4838 4839 4840
    return n;
}

4841

4842
static int
4843
testStoragePoolListAllVolumes(virStoragePoolPtr pool,
4844
                              virStorageVolPtr **vols,
4845 4846
                              unsigned int flags)
{
4847 4848
    testDriverPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr obj;
4849 4850 4851 4852
    int ret = -1;

    virCheckFlags(0, -1);

4853
    if (!(obj = testStoragePoolObjFindByUUID(privconn, pool->uuid)))
4854
        return -1;
4855

4856
    if (!virStoragePoolObjIsActive(obj)) {
4857 4858 4859 4860 4861
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                       _("storage pool is not active"));
        goto cleanup;
    }

4862
    ret = virStoragePoolObjVolumeListExport(pool->conn, obj, vols, NULL);
4863 4864

 cleanup:
4865
    virStoragePoolObjUnlock(obj);
4866 4867 4868

    return ret;
}
C
Cole Robinson 已提交
4869

4870

4871 4872 4873 4874 4875 4876 4877 4878 4879 4880 4881 4882 4883 4884 4885
static virStorageVolDefPtr
testStorageVolDefFindByName(virStoragePoolObjPtr obj,
                            const char *name)
{
    virStorageVolDefPtr privvol;

    if (!(privvol = virStorageVolDefFindByName(obj, name))) {
        virReportError(VIR_ERR_NO_STORAGE_VOL,
                       _("no storage vol with matching name '%s'"), name);
    }

    return privvol;
}


C
Cole Robinson 已提交
4886
static virStorageVolPtr
4887
testStorageVolLookupByName(virStoragePoolPtr pool,
4888
                           const char *name)
4889
{
4890
    testDriverPtr privconn = pool->conn->privateData;
4891
    virStoragePoolObjPtr obj;
4892
    virStoragePoolDefPtr def;
4893
    virStorageVolDefPtr privvol;
4894
    virStorageVolPtr ret = NULL;
4895

4896 4897
    if (!(obj = testStoragePoolObjFindActiveByName(privconn, pool->name)))
        return NULL;
4898
    def = virStoragePoolObjGetDef(obj);
4899

4900
    if (!(privvol = testStorageVolDefFindByName(obj, name)))
4901
        goto cleanup;
C
Cole Robinson 已提交
4902

4903
    ret = virGetStorageVol(pool->conn, def->name,
4904 4905
                           privvol->name, privvol->key,
                           NULL, NULL);
4906

4907
 cleanup:
4908
    virStoragePoolObjUnlock(obj);
4909
    return ret;
C
Cole Robinson 已提交
4910 4911 4912 4913
}


static virStorageVolPtr
4914
testStorageVolLookupByKey(virConnectPtr conn,
4915 4916
                          const char *key)
{
4917
    testDriverPtr privconn = conn->privateData;
4918
    virStoragePoolObjPtr obj;
4919
    virStoragePoolDefPtr def;
4920
    size_t i;
4921
    virStorageVolPtr ret = NULL;
C
Cole Robinson 已提交
4922

4923
    testDriverLock(privconn);
4924
    for (i = 0; i < privconn->pools.count; i++) {
4925 4926
        obj = privconn->pools.objs[i];
        virStoragePoolObjLock(obj);
4927
        def = virStoragePoolObjGetDef(obj);
4928 4929
        if (virStoragePoolObjIsActive(obj)) {
            virStorageVolDefPtr privvol = virStorageVolDefFindByKey(obj, key);
C
Cole Robinson 已提交
4930

4931
            if (privvol) {
4932 4933
                ret = virGetStorageVol(conn, def->name,
                                       privvol->name, privvol->key,
4934
                                       NULL, NULL);
4935
                virStoragePoolObjUnlock(obj);
4936 4937
                break;
            }
C
Cole Robinson 已提交
4938
        }
4939
        virStoragePoolObjUnlock(obj);
C
Cole Robinson 已提交
4940
    }
4941
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
4942

4943
    if (!ret)
4944 4945
        virReportError(VIR_ERR_NO_STORAGE_VOL,
                       _("no storage vol with matching key '%s'"), key);
4946 4947

    return ret;
C
Cole Robinson 已提交
4948 4949
}

4950

C
Cole Robinson 已提交
4951
static virStorageVolPtr
4952
testStorageVolLookupByPath(virConnectPtr conn,
4953 4954
                           const char *path)
{
4955
    testDriverPtr privconn = conn->privateData;
4956
    virStoragePoolObjPtr obj;
4957
    virStoragePoolDefPtr def;
4958
    size_t i;
4959
    virStorageVolPtr ret = NULL;
C
Cole Robinson 已提交
4960

4961
    testDriverLock(privconn);
4962
    for (i = 0; i < privconn->pools.count; i++) {
4963 4964
        obj = privconn->pools.objs[i];
        virStoragePoolObjLock(obj);
4965
        def = virStoragePoolObjGetDef(obj);
4966 4967
        if (virStoragePoolObjIsActive(obj)) {
            virStorageVolDefPtr privvol = virStorageVolDefFindByPath(obj, path);
C
Cole Robinson 已提交
4968

4969
            if (privvol) {
4970 4971
                ret = virGetStorageVol(conn, def->name,
                                       privvol->name, privvol->key,
4972
                                       NULL, NULL);
4973
                virStoragePoolObjUnlock(obj);
4974 4975
                break;
            }
C
Cole Robinson 已提交
4976
        }
4977
        virStoragePoolObjUnlock(obj);
C
Cole Robinson 已提交
4978
    }
4979
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
4980

4981
    if (!ret)
4982 4983
        virReportError(VIR_ERR_NO_STORAGE_VOL,
                       _("no storage vol with matching path '%s'"), path);
4984 4985

    return ret;
C
Cole Robinson 已提交
4986 4987
}

4988

C
Cole Robinson 已提交
4989
static virStorageVolPtr
4990 4991 4992
testStorageVolCreateXML(virStoragePoolPtr pool,
                        const char *xmldesc,
                        unsigned int flags)
E
Eric Blake 已提交
4993
{
4994
    testDriverPtr privconn = pool->conn->privateData;
4995
    virStoragePoolObjPtr obj;
4996
    virStoragePoolDefPtr def;
4997 4998
    virStorageVolDefPtr privvol = NULL;
    virStorageVolPtr ret = NULL;
4999

E
Eric Blake 已提交
5000 5001
    virCheckFlags(0, NULL);

5002 5003
    if (!(obj = testStoragePoolObjFindActiveByName(privconn, pool->name)))
        return NULL;
5004
    def = virStoragePoolObjGetDef(obj);
C
Cole Robinson 已提交
5005

5006
    privvol = virStorageVolDefParseString(def, xmldesc, 0);
5007
    if (privvol == NULL)
5008
        goto cleanup;
5009

5010
    if (virStorageVolDefFindByName(obj, privvol->name)) {
5011 5012
        virReportError(VIR_ERR_OPERATION_FAILED,
                       "%s", _("storage vol already exists"));
5013
        goto cleanup;
C
Cole Robinson 已提交
5014 5015 5016
    }

    /* Make sure enough space */
5017 5018
    if ((def->allocation + privvol->target.allocation) >
         def->capacity) {
5019 5020 5021
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Not enough free space in pool for volume '%s'"),
                       privvol->name);
5022
        goto cleanup;
C
Cole Robinson 已提交
5023 5024
    }

5025
    if (virAsprintf(&privvol->target.path, "%s/%s",
5026
                    def->target.path, privvol->name) < 0)
5027
        goto cleanup;
C
Cole Robinson 已提交
5028

5029
    if (VIR_STRDUP(privvol->key, privvol->target.path) < 0 ||
5030
        virStoragePoolObjAddVol(obj, privvol) < 0)
5031
        goto cleanup;
C
Cole Robinson 已提交
5032

5033 5034
    def->allocation += privvol->target.allocation;
    def->available = (def->capacity - def->allocation);
C
Cole Robinson 已提交
5035

5036
    ret = virGetStorageVol(pool->conn, def->name,
5037 5038
                           privvol->name, privvol->key,
                           NULL, NULL);
5039
    privvol = NULL;
5040

5041
 cleanup:
5042
    virStorageVolDefFree(privvol);
5043
    virStoragePoolObjUnlock(obj);
5044
    return ret;
C
Cole Robinson 已提交
5045 5046
}

5047

5048
static virStorageVolPtr
5049 5050 5051 5052
testStorageVolCreateXMLFrom(virStoragePoolPtr pool,
                            const char *xmldesc,
                            virStorageVolPtr clonevol,
                            unsigned int flags)
E
Eric Blake 已提交
5053
{
5054
    testDriverPtr privconn = pool->conn->privateData;
5055
    virStoragePoolObjPtr obj;
5056
    virStoragePoolDefPtr def;
5057 5058 5059
    virStorageVolDefPtr privvol = NULL, origvol = NULL;
    virStorageVolPtr ret = NULL;

E
Eric Blake 已提交
5060 5061
    virCheckFlags(0, NULL);

5062 5063
    if (!(obj = testStoragePoolObjFindActiveByName(privconn, pool->name)))
        return NULL;
5064
    def = virStoragePoolObjGetDef(obj);
5065

5066
    privvol = virStorageVolDefParseString(def, xmldesc, 0);
5067 5068 5069
    if (privvol == NULL)
        goto cleanup;

5070
    if (virStorageVolDefFindByName(obj, privvol->name)) {
5071 5072
        virReportError(VIR_ERR_OPERATION_FAILED,
                       "%s", _("storage vol already exists"));
5073 5074 5075
        goto cleanup;
    }

5076
    origvol = virStorageVolDefFindByName(obj, clonevol->name);
5077
    if (!origvol) {
5078 5079 5080
        virReportError(VIR_ERR_NO_STORAGE_VOL,
                       _("no storage vol with matching name '%s'"),
                       clonevol->name);
5081 5082 5083 5084
        goto cleanup;
    }

    /* Make sure enough space */
5085
    if ((def->allocation + privvol->target.allocation) > def->capacity) {
5086 5087 5088
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Not enough free space in pool for volume '%s'"),
                       privvol->name);
5089 5090
        goto cleanup;
    }
5091
    def->available = (def->capacity - def->allocation);
5092

5093
    if (virAsprintf(&privvol->target.path, "%s/%s",
5094
                    def->target.path, privvol->name) < 0)
5095 5096
        goto cleanup;

5097
    if (VIR_STRDUP(privvol->key, privvol->target.path) < 0 ||
5098
        virStoragePoolObjAddVol(obj, privvol) < 0)
5099 5100
        goto cleanup;

5101 5102
    def->allocation += privvol->target.allocation;
    def->available = (def->capacity - def->allocation);
5103

5104
    ret = virGetStorageVol(pool->conn, def->name,
5105 5106
                           privvol->name, privvol->key,
                           NULL, NULL);
5107 5108
    privvol = NULL;

5109
 cleanup:
5110
    virStorageVolDefFree(privvol);
5111
    virStoragePoolObjUnlock(obj);
5112 5113 5114
    return ret;
}

5115

C
Cole Robinson 已提交
5116
static int
5117 5118
testStorageVolDelete(virStorageVolPtr vol,
                     unsigned int flags)
E
Eric Blake 已提交
5119
{
5120
    testDriverPtr privconn = vol->conn->privateData;
5121
    virStoragePoolObjPtr obj;
5122
    virStoragePoolDefPtr def;
5123
    virStorageVolDefPtr privvol;
5124
    int ret = -1;
C
Cole Robinson 已提交
5125

E
Eric Blake 已提交
5126 5127
    virCheckFlags(0, -1);

5128 5129
    if (!(obj = testStoragePoolObjFindActiveByName(privconn, vol->pool)))
        return -1;
5130
    def = virStoragePoolObjGetDef(obj);
5131

5132
    if (!(privvol = testStorageVolDefFindByName(obj, vol->name)))
5133
        goto cleanup;
5134

5135 5136
    def->allocation -= privvol->target.allocation;
    def->available = (def->capacity - def->allocation);
C
Cole Robinson 已提交
5137

5138
    virStoragePoolObjRemoveVol(obj, privvol);
C
Cole Robinson 已提交
5139

5140
    ret = 0;
C
Cole Robinson 已提交
5141

5142
 cleanup:
5143
    virStoragePoolObjUnlock(obj);
5144
    return ret;
C
Cole Robinson 已提交
5145 5146 5147
}


5148 5149
static int
testStorageVolumeTypeForPool(int pooltype)
5150
{
C
Cole Robinson 已提交
5151

5152
    switch (pooltype) {
C
Cole Robinson 已提交
5153 5154 5155 5156 5157 5158 5159 5160 5161
        case VIR_STORAGE_POOL_DIR:
        case VIR_STORAGE_POOL_FS:
        case VIR_STORAGE_POOL_NETFS:
            return VIR_STORAGE_VOL_FILE;
        default:
            return VIR_STORAGE_VOL_BLOCK;
    }
}

5162

C
Cole Robinson 已提交
5163
static int
5164
testStorageVolGetInfo(virStorageVolPtr vol,
5165 5166
                      virStorageVolInfoPtr info)
{
5167
    testDriverPtr privconn = vol->conn->privateData;
5168
    virStoragePoolObjPtr obj;
5169
    virStoragePoolDefPtr def;
5170
    virStorageVolDefPtr privvol;
5171
    int ret = -1;
5172

5173 5174
    if (!(obj = testStoragePoolObjFindActiveByName(privconn, vol->pool)))
        return -1;
5175
    def = virStoragePoolObjGetDef(obj);
5176

5177
    if (!(privvol = testStorageVolDefFindByName(obj, vol->name)))
5178
        goto cleanup;
5179

C
Cole Robinson 已提交
5180
    memset(info, 0, sizeof(*info));
5181
    info->type = testStorageVolumeTypeForPool(def->type);
5182 5183
    info->capacity = privvol->target.capacity;
    info->allocation = privvol->target.allocation;
5184
    ret = 0;
C
Cole Robinson 已提交
5185

5186
 cleanup:
5187
    virStoragePoolObjUnlock(obj);
5188
    return ret;
C
Cole Robinson 已提交
5189 5190
}

5191

C
Cole Robinson 已提交
5192
static char *
5193 5194
testStorageVolGetXMLDesc(virStorageVolPtr vol,
                         unsigned int flags)
E
Eric Blake 已提交
5195
{
5196
    testDriverPtr privconn = vol->conn->privateData;
5197
    virStoragePoolObjPtr obj;
5198
    virStorageVolDefPtr privvol;
5199
    char *ret = NULL;
5200

E
Eric Blake 已提交
5201 5202
    virCheckFlags(0, NULL);

5203 5204
    if (!(obj = testStoragePoolObjFindActiveByName(privconn, vol->pool)))
        return NULL;
5205

5206
    if (!(privvol = testStorageVolDefFindByName(obj, vol->name)))
5207
        goto cleanup;
C
Cole Robinson 已提交
5208

5209
    ret = virStorageVolDefFormat(virStoragePoolObjGetDef(obj), privvol);
5210

5211
 cleanup:
5212
    virStoragePoolObjUnlock(obj);
5213
    return ret;
C
Cole Robinson 已提交
5214 5215
}

5216

C
Cole Robinson 已提交
5217
static char *
5218 5219
testStorageVolGetPath(virStorageVolPtr vol)
{
5220
    testDriverPtr privconn = vol->conn->privateData;
5221
    virStoragePoolObjPtr obj;
5222
    virStorageVolDefPtr privvol;
5223
    char *ret = NULL;
5224

5225 5226
    if (!(obj = testStoragePoolObjFindActiveByName(privconn, vol->pool)))
        return NULL;
5227

5228
    if (!(privvol = testStorageVolDefFindByName(obj, vol->name)))
5229
        goto cleanup;
5230

5231
    ignore_value(VIR_STRDUP(ret, privvol->target.path));
5232

5233
 cleanup:
5234
    virStoragePoolObjUnlock(obj);
C
Cole Robinson 已提交
5235 5236 5237
    return ret;
}

5238

5239
/* Node device implementations */
5240

5241 5242 5243 5244 5245 5246
static virNodeDeviceObjPtr
testNodeDeviceObjFindByName(testDriverPtr driver,
                            const char *name)
{
    virNodeDeviceObjPtr obj;

5247
    if (!(obj = virNodeDeviceObjListFindByName(driver->devs, name)))
5248 5249 5250 5251 5252 5253 5254 5255
        virReportError(VIR_ERR_NO_NODE_DEVICE,
                       _("no node device with matching name '%s'"),
                       name);

    return obj;
}


5256 5257 5258
static int
testNodeNumOfDevices(virConnectPtr conn,
                     const char *cap,
E
Eric Blake 已提交
5259
                     unsigned int flags)
5260
{
5261
    testDriverPtr driver = conn->privateData;
5262

E
Eric Blake 已提交
5263 5264
    virCheckFlags(0, -1);

5265
    return virNodeDeviceObjListNumOfDevices(driver->devs, conn, cap, NULL);
5266 5267
}

5268

5269 5270 5271 5272 5273
static int
testNodeListDevices(virConnectPtr conn,
                    const char *cap,
                    char **const names,
                    int maxnames,
E
Eric Blake 已提交
5274
                    unsigned int flags)
5275
{
5276
    testDriverPtr driver = conn->privateData;
5277

E
Eric Blake 已提交
5278 5279
    virCheckFlags(0, -1);

5280 5281
    return virNodeDeviceObjListGetNames(driver->devs, conn, NULL,
                                        cap, names, maxnames);
5282 5283
}

5284

5285 5286 5287
static virNodeDevicePtr
testNodeDeviceLookupByName(virConnectPtr conn, const char *name)
{
5288
    testDriverPtr driver = conn->privateData;
5289
    virNodeDeviceObjPtr obj;
5290
    virNodeDeviceDefPtr def;
5291 5292
    virNodeDevicePtr ret = NULL;

5293
    if (!(obj = testNodeDeviceObjFindByName(driver, name)))
5294
        return NULL;
5295
    def = virNodeDeviceObjGetDef(obj);
5296

5297
    if ((ret = virGetNodeDevice(conn, name))) {
5298
        if (VIR_STRDUP(ret->parent, def->parent) < 0) {
5299
            virObjectUnref(ret);
5300 5301
            ret = NULL;
        }
5302
    }
5303

5304
    virNodeDeviceObjEndAPI(&obj);
5305 5306 5307 5308
    return ret;
}

static char *
5309
testNodeDeviceGetXMLDesc(virNodeDevicePtr dev,
E
Eric Blake 已提交
5310
                         unsigned int flags)
5311
{
5312
    testDriverPtr driver = dev->conn->privateData;
5313 5314 5315
    virNodeDeviceObjPtr obj;
    char *ret = NULL;

E
Eric Blake 已提交
5316 5317
    virCheckFlags(0, NULL);

5318
    if (!(obj = testNodeDeviceObjFindByName(driver, dev->name)))
5319
        return NULL;
5320

5321
    ret = virNodeDeviceDefFormat(virNodeDeviceObjGetDef(obj));
5322

5323
    virNodeDeviceObjEndAPI(&obj);
5324 5325 5326 5327 5328 5329
    return ret;
}

static char *
testNodeDeviceGetParent(virNodeDevicePtr dev)
{
5330
    testDriverPtr driver = dev->conn->privateData;
5331
    virNodeDeviceObjPtr obj;
5332
    virNodeDeviceDefPtr def;
5333 5334
    char *ret = NULL;

5335
    if (!(obj = testNodeDeviceObjFindByName(driver, dev->name)))
5336
        return NULL;
5337
    def = virNodeDeviceObjGetDef(obj);
5338

5339 5340
    if (def->parent) {
        ignore_value(VIR_STRDUP(ret, def->parent));
5341
    } else {
5342 5343
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("no parent for this device"));
5344 5345
    }

5346
    virNodeDeviceObjEndAPI(&obj);
5347 5348 5349
    return ret;
}

5350

5351 5352 5353
static int
testNodeDeviceNumOfCaps(virNodeDevicePtr dev)
{
5354
    testDriverPtr driver = dev->conn->privateData;
5355
    virNodeDeviceObjPtr obj;
5356
    virNodeDeviceDefPtr def;
5357 5358 5359
    virNodeDevCapsDefPtr caps;
    int ncaps = 0;

5360
    if (!(obj = testNodeDeviceObjFindByName(driver, dev->name)))
5361
        return -1;
5362
    def = virNodeDeviceObjGetDef(obj);
5363

5364
    for (caps = def->caps; caps; caps = caps->next)
5365 5366
        ++ncaps;

5367
    virNodeDeviceObjEndAPI(&obj);
5368
    return ncaps;
5369 5370 5371 5372 5373 5374
}


static int
testNodeDeviceListCaps(virNodeDevicePtr dev, char **const names, int maxnames)
{
5375
    testDriverPtr driver = dev->conn->privateData;
5376
    virNodeDeviceObjPtr obj;
5377
    virNodeDeviceDefPtr def;
5378 5379 5380
    virNodeDevCapsDefPtr caps;
    int ncaps = 0;

5381
    if (!(obj = testNodeDeviceObjFindByName(driver, dev->name)))
5382
        return -1;
5383
    def = virNodeDeviceObjGetDef(obj);
5384

5385
    for (caps = def->caps; caps && ncaps < maxnames; caps = caps->next) {
5386 5387 5388 5389
        if (VIR_STRDUP(names[ncaps],
                       virNodeDevCapTypeToString(caps->data.type)) < 0)
            goto error;
        ncaps++;
5390 5391
    }

5392
    virNodeDeviceObjEndAPI(&obj);
5393 5394 5395 5396 5397
    return ncaps;

 error:
    while (--ncaps >= 0)
        VIR_FREE(names[ncaps]);
5398
    virNodeDeviceObjEndAPI(&obj);
5399
    return -1;
5400 5401
}

5402

5403 5404
static virNodeDeviceObjPtr
testNodeDeviceMockCreateVport(testDriverPtr driver,
5405
                              const char *wwnn,
5406
                              const char *wwpn)
5407
{
5408 5409
    char *xml = NULL;
    virNodeDeviceDefPtr def = NULL;
5410
    virNodeDevCapsDefPtr caps;
5411
    virNodeDeviceObjPtr obj = NULL, objcopy = NULL;
5412
    virNodeDeviceDefPtr objdef;
5413
    virObjectEventPtr event = NULL;
5414

5415 5416 5417 5418 5419 5420 5421 5422 5423
    /* In the real code, we'd call virVHBAManageVport which would take the
     * wwnn/wwpn from the input XML in order to call the "vport_create"
     * function for the parent. That in turn would set off a sequence of
     * events resulting in the creation of a vHBA scsi_hostN in the
     * node device objects list using the "next" host number with the
     * wwnn/wwpn from the input XML. The following will mock this by
     * using the scsi_host11 definition, changing the name and the
     * scsi_host capability fields before calling virNodeDeviceAssignDef
     * to add the def to the node device objects list. */
5424 5425
    if (!(objcopy = virNodeDeviceObjListFindByName(driver->devs,
                                                   "scsi_host11")))
5426 5427
        goto cleanup;

5428
    xml = virNodeDeviceDefFormat(virNodeDeviceObjGetDef(objcopy));
5429
    virNodeDeviceObjEndAPI(&objcopy);
5430 5431 5432 5433
    if (!xml)
        goto cleanup;

    if (!(def = virNodeDeviceDefParseString(xml, EXISTING_DEVICE, NULL)))
5434 5435
        goto cleanup;

5436
    VIR_FREE(def->name);
5437
    if (VIR_STRDUP(def->name, "scsi_host12") < 0)
5438 5439
        goto cleanup;

5440 5441 5442
    /* Find the 'scsi_host' cap and alter the host # and unique_id and
     * then for the 'fc_host' capability modify the wwnn/wwpn to be that
     * of the input XML. */
5443 5444
    caps = def->caps;
    while (caps) {
5445
        if (caps->data.type != VIR_NODE_DEV_CAP_SCSI_HOST)
5446 5447
            continue;

5448 5449 5450 5451 5452 5453 5454 5455 5456 5457 5458 5459 5460 5461
        /* For the "fc_host" cap - change the wwnn/wwpn to match the input */
        if (caps->data.scsi_host.flags & VIR_NODE_DEV_CAP_FLAG_HBA_FC_HOST) {
            VIR_FREE(caps->data.scsi_host.wwnn);
            VIR_FREE(caps->data.scsi_host.wwpn);
            if (VIR_STRDUP(caps->data.scsi_host.wwnn, wwnn) < 0 ||
                VIR_STRDUP(caps->data.scsi_host.wwpn, wwpn) < 0)
                goto cleanup;
        } else {
            /* For the "scsi_host" cap, increment our host and unique_id to
             * give the appearance that something new was created - then add
             * that to the node device driver */
            caps->data.scsi_host.host++;
            caps->data.scsi_host.unique_id++;
        }
5462 5463 5464
        caps = caps->next;
    }

5465
    if (!(obj = virNodeDeviceObjListAssignDef(driver->devs, def)))
5466
        goto cleanup;
5467
    def = NULL;
5468
    objdef = virNodeDeviceObjGetDef(obj);
5469

5470
    event = virNodeDeviceEventLifecycleNew(objdef->name,
5471 5472
                                           VIR_NODE_DEVICE_EVENT_CREATED,
                                           0);
5473 5474 5475
    testObjectEventQueue(driver, event);

 cleanup:
5476
    VIR_FREE(xml);
5477 5478
    virNodeDeviceDefFree(def);
    return obj;
5479 5480 5481 5482 5483 5484 5485 5486 5487 5488 5489
}


static virNodeDevicePtr
testNodeDeviceCreateXML(virConnectPtr conn,
                        const char *xmlDesc,
                        unsigned int flags)
{
    testDriverPtr driver = conn->privateData;
    virNodeDeviceDefPtr def = NULL;
    char *wwnn = NULL, *wwpn = NULL;
5490 5491
    virNodeDevicePtr dev = NULL, ret = NULL;
    virNodeDeviceObjPtr obj = NULL;
5492
    virNodeDeviceDefPtr objdef;
5493 5494 5495 5496 5497 5498

    virCheckFlags(0, NULL);

    if (!(def = virNodeDeviceDefParseString(xmlDesc, CREATE_DEVICE, NULL)))
        goto cleanup;

5499 5500 5501
    /* We run this simply for validation - it essentially validates that
     * the input XML either has a wwnn/wwpn or virNodeDevCapSCSIHostParseXML
     * generated a wwnn/wwpn */
5502 5503 5504
    if (virNodeDeviceGetWWNs(def, &wwnn, &wwpn) < 0)
        goto cleanup;

5505 5506 5507
    /* Unlike the "real" code we don't need the parent_host in order to
     * call virVHBAManageVport, but still let's make sure the code finds
     * something valid and no one messed up the mock environment. */
5508
    if (virNodeDeviceObjListGetParentHost(driver->devs, def) < 0)
5509 5510 5511 5512
        goto cleanup;

    /* In the real code, we'd call virVHBAManageVport followed by
     * find_new_device, but we cannot do that here since we're not
5513 5514 5515
     * mocking udev. The mock routine will copy an existing vHBA and
     * rename a few fields to mock that. So in order to allow that to
     * work properly, we need to drop our lock */
5516 5517
    if (!(obj = testNodeDeviceMockCreateVport(driver, wwnn, wwpn)))
        goto cleanup;
5518
    objdef = virNodeDeviceObjGetDef(obj);
5519

5520
    if (!(dev = virGetNodeDevice(conn, objdef->name)))
5521 5522 5523 5524 5525 5526 5527 5528
        goto cleanup;

    VIR_FREE(dev->parent);
    if (VIR_STRDUP(dev->parent, def->parent) < 0)
        goto cleanup;

    ret = dev;
    dev = NULL;
5529

5530
 cleanup:
5531
    virNodeDeviceObjEndAPI(&obj);
5532
    virNodeDeviceDefFree(def);
5533
    virObjectUnref(dev);
5534 5535
    VIR_FREE(wwnn);
    VIR_FREE(wwpn);
5536
    return ret;
5537 5538 5539 5540 5541 5542
}

static int
testNodeDeviceDestroy(virNodeDevicePtr dev)
{
    int ret = 0;
5543
    testDriverPtr driver = dev->conn->privateData;
5544
    virNodeDeviceObjPtr obj = NULL;
5545
    virNodeDeviceObjPtr parentobj = NULL;
5546
    virNodeDeviceDefPtr def;
5547
    char *wwnn = NULL, *wwpn = NULL;
5548
    virObjectEventPtr event = NULL;
5549

5550
    if (!(obj = testNodeDeviceObjFindByName(driver, dev->name)))
5551
        return -1;
5552
    def = virNodeDeviceObjGetDef(obj);
5553

5554
    if (virNodeDeviceGetWWNs(def, &wwnn, &wwpn) == -1)
5555
        goto cleanup;
5556

5557 5558 5559 5560
    /* Unlike the real code we cannot run into the udevAddOneDevice race
     * which would replace obj->def, so no need to save off the parent,
     * but do need to drop the @obj lock so that the FindByName code doesn't
     * deadlock on ourselves */
5561
    virObjectUnlock(obj);
5562

5563 5564 5565 5566 5567 5568
    /* We do this just for basic validation and throw away the parentobj
     * since there's no vport_delete to be run */
    if (!(parentobj = virNodeDeviceObjListFindByName(driver->devs,
                                                     def->parent))) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("cannot find parent '%s' definition"), def->parent);
5569
        virObjectLock(obj);
5570
        goto cleanup;
5571
    }
5572
    virNodeDeviceObjEndAPI(&parentobj);
5573

5574 5575 5576 5577
    event = virNodeDeviceEventLifecycleNew(dev->name,
                                           VIR_NODE_DEVICE_EVENT_DELETED,
                                           0);

5578
    virObjectLock(obj);
5579
    virNodeDeviceObjListRemove(driver->devs, obj);
5580
    virObjectUnref(obj);
5581
    obj = NULL;
5582

5583
 cleanup:
5584
    virNodeDeviceObjEndAPI(&obj);
5585
    testObjectEventQueue(driver, event);
5586 5587 5588 5589 5590
    VIR_FREE(wwnn);
    VIR_FREE(wwpn);
    return ret;
}

5591 5592

/* Domain event implementations */
5593
static int
5594 5595 5596 5597
testConnectDomainEventRegister(virConnectPtr conn,
                               virConnectDomainEventCallback callback,
                               void *opaque,
                               virFreeCallback freecb)
5598
{
5599
    testDriverPtr driver = conn->privateData;
5600
    int ret = 0;
5601

5602
    if (virDomainEventStateRegister(conn, driver->eventState,
5603 5604
                                    callback, opaque, freecb) < 0)
        ret = -1;
5605 5606 5607 5608

    return ret;
}

5609

5610
static int
5611 5612
testConnectDomainEventDeregister(virConnectPtr conn,
                                 virConnectDomainEventCallback callback)
5613
{
5614
    testDriverPtr driver = conn->privateData;
5615
    int ret = 0;
5616

5617
    if (virDomainEventStateDeregister(conn, driver->eventState,
5618 5619
                                      callback) < 0)
        ret = -1;
5620 5621 5622 5623

    return ret;
}

5624 5625

static int
5626 5627 5628 5629 5630 5631
testConnectDomainEventRegisterAny(virConnectPtr conn,
                                  virDomainPtr dom,
                                  int eventID,
                                  virConnectDomainEventGenericCallback callback,
                                  void *opaque,
                                  virFreeCallback freecb)
5632
{
5633
    testDriverPtr driver = conn->privateData;
5634 5635
    int ret;

5636
    if (virDomainEventStateRegisterID(conn, driver->eventState,
5637 5638
                                      dom, eventID,
                                      callback, opaque, freecb, &ret) < 0)
5639
        ret = -1;
5640 5641 5642 5643 5644

    return ret;
}

static int
5645 5646
testConnectDomainEventDeregisterAny(virConnectPtr conn,
                                    int callbackID)
5647
{
5648
    testDriverPtr driver = conn->privateData;
5649
    int ret = 0;
5650

5651
    if (virObjectEventStateDeregisterID(conn, driver->eventState,
5652
                                        callbackID, true) < 0)
5653
        ret = -1;
5654 5655 5656 5657 5658

    return ret;
}


5659 5660 5661 5662 5663 5664 5665 5666
static int
testConnectNetworkEventRegisterAny(virConnectPtr conn,
                                   virNetworkPtr net,
                                   int eventID,
                                   virConnectNetworkEventGenericCallback callback,
                                   void *opaque,
                                   virFreeCallback freecb)
{
5667
    testDriverPtr driver = conn->privateData;
5668 5669
    int ret;

5670
    if (virNetworkEventStateRegisterID(conn, driver->eventState,
5671
                                       net, eventID, callback,
5672 5673 5674 5675 5676 5677 5678 5679 5680 5681
                                       opaque, freecb, &ret) < 0)
        ret = -1;

    return ret;
}

static int
testConnectNetworkEventDeregisterAny(virConnectPtr conn,
                                     int callbackID)
{
5682
    testDriverPtr driver = conn->privateData;
5683
    int ret = 0;
5684

5685
    if (virObjectEventStateDeregisterID(conn, driver->eventState,
5686
                                        callbackID, true) < 0)
5687
        ret = -1;
5688 5689 5690 5691

    return ret;
}

5692 5693 5694 5695 5696 5697 5698 5699 5700 5701 5702 5703 5704 5705 5706 5707 5708 5709 5710 5711 5712 5713 5714 5715 5716 5717 5718
static int
testConnectStoragePoolEventRegisterAny(virConnectPtr conn,
                                       virStoragePoolPtr pool,
                                       int eventID,
                                       virConnectStoragePoolEventGenericCallback callback,
                                       void *opaque,
                                       virFreeCallback freecb)
{
    testDriverPtr driver = conn->privateData;
    int ret;

    if (virStoragePoolEventStateRegisterID(conn, driver->eventState,
                                           pool, eventID, callback,
                                           opaque, freecb, &ret) < 0)
        ret = -1;

    return ret;
}

static int
testConnectStoragePoolEventDeregisterAny(virConnectPtr conn,
                                         int callbackID)
{
    testDriverPtr driver = conn->privateData;
    int ret = 0;

    if (virObjectEventStateDeregisterID(conn, driver->eventState,
5719
                                        callbackID, true) < 0)
5720 5721 5722 5723 5724
        ret = -1;

    return ret;
}

5725 5726 5727 5728 5729 5730 5731 5732 5733 5734 5735 5736 5737 5738 5739 5740 5741 5742 5743 5744 5745 5746 5747 5748 5749 5750 5751
static int
testConnectNodeDeviceEventRegisterAny(virConnectPtr conn,
                                      virNodeDevicePtr dev,
                                      int eventID,
                                      virConnectNodeDeviceEventGenericCallback callback,
                                      void *opaque,
                                      virFreeCallback freecb)
{
    testDriverPtr driver = conn->privateData;
    int ret;

    if (virNodeDeviceEventStateRegisterID(conn, driver->eventState,
                                          dev, eventID, callback,
                                          opaque, freecb, &ret) < 0)
        ret = -1;

    return ret;
}

static int
testConnectNodeDeviceEventDeregisterAny(virConnectPtr conn,
                                        int callbackID)
{
    testDriverPtr driver = conn->privateData;
    int ret = 0;

    if (virObjectEventStateDeregisterID(conn, driver->eventState,
5752
                                        callbackID, true) < 0)
5753 5754 5755 5756 5757
        ret = -1;

    return ret;
}

5758 5759 5760
static int testConnectListAllDomains(virConnectPtr conn,
                                     virDomainPtr **domains,
                                     unsigned int flags)
5761
{
5762
    testDriverPtr privconn = conn->privateData;
5763

O
Osier Yang 已提交
5764
    virCheckFlags(VIR_CONNECT_LIST_DOMAINS_FILTERS_ALL, -1);
5765

5766 5767
    return virDomainObjListExport(privconn->domains, conn, domains,
                                  NULL, flags);
5768 5769
}

5770
static int
P
Peter Krempa 已提交
5771
testNodeGetCPUMap(virConnectPtr conn ATTRIBUTE_UNUSED,
5772 5773 5774 5775 5776 5777 5778
                  unsigned char **cpumap,
                  unsigned int *online,
                  unsigned int flags)
{
    virCheckFlags(0, -1);

    if (cpumap) {
5779
        if (VIR_ALLOC_N(*cpumap, 1) < 0)
P
Peter Krempa 已提交
5780
            return -1;
5781 5782 5783 5784 5785 5786
        *cpumap[0] = 0x15;
    }

    if (online)
        *online = 3;

P
Peter Krempa 已提交
5787
    return  8;
5788 5789
}

5790 5791 5792 5793 5794 5795 5796 5797 5798 5799
static char *
testDomainScreenshot(virDomainPtr dom ATTRIBUTE_UNUSED,
                     virStreamPtr st,
                     unsigned int screen ATTRIBUTE_UNUSED,
                     unsigned int flags)
{
    char *ret = NULL;

    virCheckFlags(0, NULL);

5800
    if (VIR_STRDUP(ret, "image/png") < 0)
5801 5802
        return NULL;

D
Daniel P. Berrange 已提交
5803
    if (virFDStreamOpenFile(st, PKGDATADIR "/test-screenshot.png", 0, 0, O_RDONLY) < 0)
5804 5805 5806 5807 5808
        VIR_FREE(ret);

    return ret;
}

5809 5810
static int
testConnectGetCPUModelNames(virConnectPtr conn ATTRIBUTE_UNUSED,
J
Jiri Denemark 已提交
5811
                            const char *archName,
5812 5813 5814
                            char ***models,
                            unsigned int flags)
{
J
Jiri Denemark 已提交
5815 5816
    virArch arch;

5817
    virCheckFlags(0, -1);
J
Jiri Denemark 已提交
5818 5819 5820 5821 5822 5823 5824 5825

    if (!(arch = virArchFromString(archName))) {
        virReportError(VIR_ERR_INVALID_ARG,
                       _("cannot find architecture %s"),
                       archName);
        return -1;
    }

J
Jiri Denemark 已提交
5826
    return virCPUGetModels(arch, models);
5827
}
5828

C
Cole Robinson 已提交
5829 5830 5831
static int
testDomainManagedSave(virDomainPtr dom, unsigned int flags)
{
5832
    testDriverPtr privconn = dom->conn->privateData;
C
Cole Robinson 已提交
5833
    virDomainObjPtr vm = NULL;
5834
    virObjectEventPtr event = NULL;
C
Cole Robinson 已提交
5835 5836 5837 5838 5839 5840
    int ret = -1;

    virCheckFlags(VIR_DOMAIN_SAVE_BYPASS_CACHE |
                  VIR_DOMAIN_SAVE_RUNNING |
                  VIR_DOMAIN_SAVE_PAUSED, -1);

5841 5842
    if (!(vm = testDomObjFromDomain(dom)))
        return -1;
C
Cole Robinson 已提交
5843 5844 5845 5846 5847 5848 5849 5850 5851 5852 5853 5854 5855 5856

    if (!virDomainObjIsActive(vm)) {
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s", _("domain is not running"));
        goto cleanup;
    }

    if (!vm->persistent) {
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                       _("cannot do managed save for transient domain"));
        goto cleanup;
    }

    testDomainShutdownState(dom, vm, VIR_DOMAIN_SHUTOFF_SAVED);
5857
    event = virDomainEventLifecycleNewFromObj(vm,
C
Cole Robinson 已提交
5858 5859 5860 5861 5862
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_SAVED);
    vm->hasManagedSave = true;

    ret = 0;
5863
 cleanup:
5864
    virDomainObjEndAPI(&vm);
5865
    testObjectEventQueue(privconn, event);
C
Cole Robinson 已提交
5866 5867 5868 5869 5870 5871 5872 5873 5874

    return ret;
}


static int
testDomainHasManagedSaveImage(virDomainPtr dom, unsigned int flags)
{
    virDomainObjPtr vm;
5875
    int ret;
C
Cole Robinson 已提交
5876 5877 5878

    virCheckFlags(0, -1);

5879 5880
    if (!(vm = testDomObjFromDomain(dom)))
        return -1;
C
Cole Robinson 已提交
5881 5882

    ret = vm->hasManagedSave;
5883

5884
    virDomainObjEndAPI(&vm);
C
Cole Robinson 已提交
5885 5886 5887 5888 5889 5890 5891 5892 5893 5894
    return ret;
}

static int
testDomainManagedSaveRemove(virDomainPtr dom, unsigned int flags)
{
    virDomainObjPtr vm;

    virCheckFlags(0, -1);

5895 5896
    if (!(vm = testDomObjFromDomain(dom)))
        return -1;
C
Cole Robinson 已提交
5897 5898

    vm->hasManagedSave = false;
5899

5900
    virDomainObjEndAPI(&vm);
5901
    return 0;
C
Cole Robinson 已提交
5902 5903 5904
}


5905 5906 5907 5908 5909 5910 5911 5912 5913 5914 5915 5916 5917 5918 5919 5920 5921 5922 5923 5924 5925 5926 5927 5928 5929 5930 5931 5932 5933 5934 5935 5936 5937 5938
/*
 * Snapshot APIs
 */

static virDomainSnapshotObjPtr
testSnapObjFromName(virDomainObjPtr vm,
                    const char *name)
{
    virDomainSnapshotObjPtr snap = NULL;
    snap = virDomainSnapshotFindByName(vm->snapshots, name);
    if (!snap)
        virReportError(VIR_ERR_NO_DOMAIN_SNAPSHOT,
                       _("no domain snapshot with matching name '%s'"),
                       name);
    return snap;
}

static virDomainSnapshotObjPtr
testSnapObjFromSnapshot(virDomainObjPtr vm,
                        virDomainSnapshotPtr snapshot)
{
    return testSnapObjFromName(vm, snapshot->name);
}

static virDomainObjPtr
testDomObjFromSnapshot(virDomainSnapshotPtr snapshot)
{
    return testDomObjFromDomain(snapshot->domain);
}

static int
testDomainSnapshotNum(virDomainPtr domain, unsigned int flags)
{
    virDomainObjPtr vm = NULL;
5939
    int n;
5940 5941 5942 5943 5944

    virCheckFlags(VIR_DOMAIN_SNAPSHOT_LIST_ROOTS |
                  VIR_DOMAIN_SNAPSHOT_FILTERS_ALL, -1);

    if (!(vm = testDomObjFromDomain(domain)))
5945
        return -1;
5946 5947 5948

    n = virDomainSnapshotObjListNum(vm->snapshots, NULL, flags);

5949
    virDomainObjEndAPI(&vm);
5950 5951 5952 5953 5954 5955 5956 5957 5958 5959
    return n;
}

static int
testDomainSnapshotListNames(virDomainPtr domain,
                            char **names,
                            int nameslen,
                            unsigned int flags)
{
    virDomainObjPtr vm = NULL;
5960
    int n;
5961 5962 5963 5964 5965

    virCheckFlags(VIR_DOMAIN_SNAPSHOT_LIST_ROOTS |
                  VIR_DOMAIN_SNAPSHOT_FILTERS_ALL, -1);

    if (!(vm = testDomObjFromDomain(domain)))
5966
        return -1;
5967 5968 5969 5970

    n = virDomainSnapshotObjListGetNames(vm->snapshots, NULL, names, nameslen,
                                         flags);

5971
    virDomainObjEndAPI(&vm);
5972 5973 5974 5975 5976 5977 5978 5979 5980
    return n;
}

static int
testDomainListAllSnapshots(virDomainPtr domain,
                           virDomainSnapshotPtr **snaps,
                           unsigned int flags)
{
    virDomainObjPtr vm = NULL;
5981
    int n;
5982 5983 5984 5985 5986

    virCheckFlags(VIR_DOMAIN_SNAPSHOT_LIST_ROOTS |
                  VIR_DOMAIN_SNAPSHOT_FILTERS_ALL, -1);

    if (!(vm = testDomObjFromDomain(domain)))
5987
        return -1;
5988 5989 5990

    n = virDomainListSnapshots(vm->snapshots, NULL, domain, snaps, flags);

5991
    virDomainObjEndAPI(&vm);
5992 5993 5994 5995 5996 5997 5998 5999 6000 6001 6002 6003 6004 6005 6006 6007 6008
    return n;
}

static int
testDomainSnapshotListChildrenNames(virDomainSnapshotPtr snapshot,
                                    char **names,
                                    int nameslen,
                                    unsigned int flags)
{
    virDomainObjPtr vm = NULL;
    virDomainSnapshotObjPtr snap = NULL;
    int n = -1;

    virCheckFlags(VIR_DOMAIN_SNAPSHOT_LIST_DESCENDANTS |
                  VIR_DOMAIN_SNAPSHOT_FILTERS_ALL, -1);

    if (!(vm = testDomObjFromSnapshot(snapshot)))
6009
        return -1;
6010 6011 6012 6013 6014 6015 6016

    if (!(snap = testSnapObjFromSnapshot(vm, snapshot)))
        goto cleanup;

    n = virDomainSnapshotObjListGetNames(vm->snapshots, snap, names, nameslen,
                                         flags);

6017
 cleanup:
6018
    virDomainObjEndAPI(&vm);
6019 6020 6021 6022 6023 6024 6025 6026 6027 6028 6029 6030 6031 6032 6033
    return n;
}

static int
testDomainSnapshotNumChildren(virDomainSnapshotPtr snapshot,
                              unsigned int flags)
{
    virDomainObjPtr vm = NULL;
    virDomainSnapshotObjPtr snap = NULL;
    int n = -1;

    virCheckFlags(VIR_DOMAIN_SNAPSHOT_LIST_DESCENDANTS |
                  VIR_DOMAIN_SNAPSHOT_FILTERS_ALL, -1);

    if (!(vm = testDomObjFromSnapshot(snapshot)))
6034
        return -1;
6035 6036 6037 6038 6039 6040

    if (!(snap = testSnapObjFromSnapshot(vm, snapshot)))
        goto cleanup;

    n = virDomainSnapshotObjListNum(vm->snapshots, snap, flags);

6041
 cleanup:
6042
    virDomainObjEndAPI(&vm);
6043 6044 6045 6046 6047 6048 6049 6050 6051 6052 6053 6054 6055 6056 6057 6058
    return n;
}

static int
testDomainSnapshotListAllChildren(virDomainSnapshotPtr snapshot,
                                  virDomainSnapshotPtr **snaps,
                                  unsigned int flags)
{
    virDomainObjPtr vm = NULL;
    virDomainSnapshotObjPtr snap = NULL;
    int n = -1;

    virCheckFlags(VIR_DOMAIN_SNAPSHOT_LIST_DESCENDANTS |
                  VIR_DOMAIN_SNAPSHOT_FILTERS_ALL, -1);

    if (!(vm = testDomObjFromSnapshot(snapshot)))
6059
        return -1;
6060 6061 6062 6063 6064 6065 6066

    if (!(snap = testSnapObjFromSnapshot(vm, snapshot)))
        goto cleanup;

    n = virDomainListSnapshots(vm->snapshots, snap, snapshot->domain, snaps,
                               flags);

6067
 cleanup:
6068
    virDomainObjEndAPI(&vm);
6069 6070 6071 6072 6073 6074 6075 6076 6077 6078 6079 6080 6081 6082 6083
    return n;
}

static virDomainSnapshotPtr
testDomainSnapshotLookupByName(virDomainPtr domain,
                               const char *name,
                               unsigned int flags)
{
    virDomainObjPtr vm;
    virDomainSnapshotObjPtr snap = NULL;
    virDomainSnapshotPtr snapshot = NULL;

    virCheckFlags(0, NULL);

    if (!(vm = testDomObjFromDomain(domain)))
6084
        return NULL;
6085 6086 6087 6088 6089 6090

    if (!(snap = testSnapObjFromName(vm, name)))
        goto cleanup;

    snapshot = virGetDomainSnapshot(domain, snap->def->name);

6091
 cleanup:
6092
    virDomainObjEndAPI(&vm);
6093 6094 6095 6096 6097 6098 6099 6100
    return snapshot;
}

static int
testDomainHasCurrentSnapshot(virDomainPtr domain,
                             unsigned int flags)
{
    virDomainObjPtr vm;
6101
    int ret;
6102 6103 6104 6105

    virCheckFlags(0, -1);

    if (!(vm = testDomObjFromDomain(domain)))
6106
        return -1;
6107 6108 6109

    ret = (vm->current_snapshot != NULL);

6110
    virDomainObjEndAPI(&vm);
6111 6112 6113 6114 6115 6116 6117 6118 6119 6120 6121 6122 6123 6124
    return ret;
}

static virDomainSnapshotPtr
testDomainSnapshotGetParent(virDomainSnapshotPtr snapshot,
                            unsigned int flags)
{
    virDomainObjPtr vm;
    virDomainSnapshotObjPtr snap = NULL;
    virDomainSnapshotPtr parent = NULL;

    virCheckFlags(0, NULL);

    if (!(vm = testDomObjFromSnapshot(snapshot)))
6125
        return NULL;
6126 6127 6128 6129 6130 6131 6132 6133 6134 6135 6136 6137 6138

    if (!(snap = testSnapObjFromSnapshot(vm, snapshot)))
        goto cleanup;

    if (!snap->def->parent) {
        virReportError(VIR_ERR_NO_DOMAIN_SNAPSHOT,
                       _("snapshot '%s' does not have a parent"),
                       snap->def->name);
        goto cleanup;
    }

    parent = virGetDomainSnapshot(snapshot->domain, snap->def->parent);

6139
 cleanup:
6140
    virDomainObjEndAPI(&vm);
6141 6142 6143 6144 6145 6146 6147 6148 6149 6150 6151 6152 6153
    return parent;
}

static virDomainSnapshotPtr
testDomainSnapshotCurrent(virDomainPtr domain,
                          unsigned int flags)
{
    virDomainObjPtr vm;
    virDomainSnapshotPtr snapshot = NULL;

    virCheckFlags(0, NULL);

    if (!(vm = testDomObjFromDomain(domain)))
6154
        return NULL;
6155 6156 6157 6158 6159 6160 6161 6162 6163

    if (!vm->current_snapshot) {
        virReportError(VIR_ERR_NO_DOMAIN_SNAPSHOT, "%s",
                       _("the domain does not have a current snapshot"));
        goto cleanup;
    }

    snapshot = virGetDomainSnapshot(domain, vm->current_snapshot->def->name);

6164
 cleanup:
6165
    virDomainObjEndAPI(&vm);
6166 6167 6168 6169 6170 6171 6172 6173 6174 6175 6176
    return snapshot;
}

static char *
testDomainSnapshotGetXMLDesc(virDomainSnapshotPtr snapshot,
                             unsigned int flags)
{
    virDomainObjPtr vm = NULL;
    char *xml = NULL;
    virDomainSnapshotObjPtr snap = NULL;
    char uuidstr[VIR_UUID_STRING_BUFLEN];
6177
    testDriverPtr privconn = snapshot->domain->conn->privateData;
6178 6179 6180 6181

    virCheckFlags(VIR_DOMAIN_XML_SECURE, NULL);

    if (!(vm = testDomObjFromSnapshot(snapshot)))
6182
        return NULL;
6183 6184 6185 6186 6187 6188

    if (!(snap = testSnapObjFromSnapshot(vm, snapshot)))
        goto cleanup;

    virUUIDFormat(snapshot->domain->uuid, uuidstr);

6189
    xml = virDomainSnapshotDefFormat(uuidstr, snap->def, privconn->caps,
6190
                                     privconn->xmlopt,
6191 6192
                                     virDomainDefFormatConvertXMLFlags(flags),
                                     0);
6193

6194
 cleanup:
6195
    virDomainObjEndAPI(&vm);
6196 6197 6198 6199 6200 6201 6202 6203
    return xml;
}

static int
testDomainSnapshotIsCurrent(virDomainSnapshotPtr snapshot,
                            unsigned int flags)
{
    virDomainObjPtr vm = NULL;
6204
    int ret;
6205 6206 6207 6208

    virCheckFlags(0, -1);

    if (!(vm = testDomObjFromSnapshot(snapshot)))
6209
        return -1;
6210 6211 6212 6213

    ret = (vm->current_snapshot &&
           STREQ(snapshot->name, vm->current_snapshot->def->name));

6214
    virDomainObjEndAPI(&vm);
6215 6216 6217 6218 6219 6220 6221 6222 6223 6224 6225 6226 6227 6228
    return ret;
}


static int
testDomainSnapshotHasMetadata(virDomainSnapshotPtr snapshot,
                              unsigned int flags)
{
    virDomainObjPtr vm = NULL;
    int ret = -1;

    virCheckFlags(0, -1);

    if (!(vm = testDomObjFromSnapshot(snapshot)))
6229
        return -1;
6230

C
Cole Robinson 已提交
6231
    if (!testSnapObjFromSnapshot(vm, snapshot))
6232 6233 6234 6235
        goto cleanup;

    ret = 1;

6236
 cleanup:
6237
    virDomainObjEndAPI(&vm);
6238 6239 6240
    return ret;
}

6241 6242 6243 6244 6245 6246
static int
testDomainSnapshotAlignDisks(virDomainObjPtr vm,
                             virDomainSnapshotDefPtr def,
                             unsigned int flags)
{
    int align_location = VIR_DOMAIN_SNAPSHOT_LOCATION_INTERNAL;
E
Eric Blake 已提交
6247
    bool align_match = true;
6248 6249 6250 6251 6252 6253 6254 6255 6256 6257 6258 6259 6260 6261 6262 6263 6264 6265 6266 6267 6268 6269 6270 6271 6272 6273 6274 6275

    if (flags & VIR_DOMAIN_SNAPSHOT_CREATE_DISK_ONLY) {
        align_location = VIR_DOMAIN_SNAPSHOT_LOCATION_EXTERNAL;
        align_match = false;
        if (virDomainObjIsActive(vm))
            def->state = VIR_DOMAIN_DISK_SNAPSHOT;
        else
            def->state = VIR_DOMAIN_SHUTOFF;
        def->memory = VIR_DOMAIN_SNAPSHOT_LOCATION_NONE;
    } else if (def->memory == VIR_DOMAIN_SNAPSHOT_LOCATION_EXTERNAL) {
        def->state = virDomainObjGetState(vm, NULL);
        align_location = VIR_DOMAIN_SNAPSHOT_LOCATION_EXTERNAL;
        align_match = false;
    } else {
        def->state = virDomainObjGetState(vm, NULL);
        def->memory = def->state == VIR_DOMAIN_SHUTOFF ?
                      VIR_DOMAIN_SNAPSHOT_LOCATION_NONE :
                      VIR_DOMAIN_SNAPSHOT_LOCATION_INTERNAL;
    }

    return virDomainSnapshotAlignDisks(def, align_location, align_match);
}

static virDomainSnapshotPtr
testDomainSnapshotCreateXML(virDomainPtr domain,
                            const char *xmlDesc,
                            unsigned int flags)
{
6276
    testDriverPtr privconn = domain->conn->privateData;
6277 6278 6279 6280
    virDomainObjPtr vm = NULL;
    virDomainSnapshotDefPtr def = NULL;
    virDomainSnapshotObjPtr snap = NULL;
    virDomainSnapshotPtr snapshot = NULL;
6281
    virObjectEventPtr event = NULL;
6282
    char *xml = NULL;
6283 6284
    bool update_current = true;
    bool redefine = flags & VIR_DOMAIN_SNAPSHOT_CREATE_REDEFINE;
6285 6286 6287 6288 6289 6290 6291 6292
    unsigned int parse_flags = VIR_DOMAIN_SNAPSHOT_PARSE_DISKS;

    /*
     * DISK_ONLY: Not implemented yet
     * REUSE_EXT: Not implemented yet
     *
     * NO_METADATA: Explicitly not implemented
     *
6293
     * REDEFINE + CURRENT: Implemented
6294 6295 6296 6297 6298 6299
     * HALT: Implemented
     * QUIESCE: Nothing to do
     * ATOMIC: Nothing to do
     * LIVE: Nothing to do
     */
    virCheckFlags(
6300 6301
        VIR_DOMAIN_SNAPSHOT_CREATE_REDEFINE |
        VIR_DOMAIN_SNAPSHOT_CREATE_CURRENT |
6302 6303 6304 6305 6306
        VIR_DOMAIN_SNAPSHOT_CREATE_HALT |
        VIR_DOMAIN_SNAPSHOT_CREATE_QUIESCE |
        VIR_DOMAIN_SNAPSHOT_CREATE_ATOMIC |
        VIR_DOMAIN_SNAPSHOT_CREATE_LIVE, NULL);

6307 6308 6309 6310 6311
    if ((redefine && !(flags & VIR_DOMAIN_SNAPSHOT_CREATE_CURRENT)))
        update_current = false;
    if (redefine)
        parse_flags |= VIR_DOMAIN_SNAPSHOT_PARSE_REDEFINE;

6312 6313 6314 6315 6316 6317 6318 6319 6320 6321 6322 6323 6324 6325 6326
    if (!(vm = testDomObjFromDomain(domain)))
        goto cleanup;

    if (!vm->persistent && (flags & VIR_DOMAIN_SNAPSHOT_CREATE_HALT)) {
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                       _("cannot halt after transient domain snapshot"));
        goto cleanup;
    }

    if (!(def = virDomainSnapshotDefParseString(xmlDesc,
                                                privconn->caps,
                                                privconn->xmlopt,
                                                parse_flags)))
        goto cleanup;

6327
    if (redefine) {
C
Cole Robinson 已提交
6328
        if (virDomainSnapshotRedefinePrep(domain, vm, &def, &snap,
6329
                                          privconn->xmlopt,
C
Cole Robinson 已提交
6330
                                          &update_current, flags) < 0)
6331 6332 6333 6334 6335
            goto cleanup;
    } else {
        if (!(def->dom = virDomainDefCopy(vm->def,
                                          privconn->caps,
                                          privconn->xmlopt,
6336
                                          NULL,
6337 6338
                                          true)))
            goto cleanup;
6339

6340
        if (testDomainSnapshotAlignDisks(vm, def, flags) < 0)
6341 6342 6343
            goto cleanup;
    }

6344 6345 6346 6347
    if (!snap) {
        if (!(snap = virDomainSnapshotAssignDef(vm->snapshots, def)))
            goto cleanup;
        def = NULL;
6348 6349
    }

6350 6351 6352 6353 6354 6355 6356 6357 6358 6359
    if (!redefine) {
        if (vm->current_snapshot &&
            (VIR_STRDUP(snap->def->parent,
                        vm->current_snapshot->def->name) < 0))
            goto cleanup;

        if ((flags & VIR_DOMAIN_SNAPSHOT_CREATE_HALT) &&
            virDomainObjIsActive(vm)) {
            testDomainShutdownState(domain, vm,
                                    VIR_DOMAIN_SHUTOFF_FROM_SNAPSHOT);
6360
            event = virDomainEventLifecycleNewFromObj(vm, VIR_DOMAIN_EVENT_STOPPED,
6361 6362 6363
                                    VIR_DOMAIN_EVENT_STOPPED_FROM_SNAPSHOT);
        }
    }
6364 6365

    snapshot = virGetDomainSnapshot(domain, snap->def->name);
6366
 cleanup:
6367 6368 6369 6370
    VIR_FREE(xml);
    if (vm) {
        if (snapshot) {
            virDomainSnapshotObjPtr other;
6371 6372
            if (update_current)
                vm->current_snapshot = snap;
6373 6374 6375 6376 6377 6378 6379
            other = virDomainSnapshotFindByName(vm->snapshots,
                                                snap->def->parent);
            snap->parent = other;
            other->nchildren++;
            snap->sibling = other->first_child;
            other->first_child = snap;
        }
6380
        virDomainObjEndAPI(&vm);
6381
    }
6382
    testObjectEventQueue(privconn, event);
6383 6384 6385 6386 6387 6388 6389 6390 6391 6392 6393 6394
    virDomainSnapshotDefFree(def);
    return snapshot;
}


typedef struct _testSnapRemoveData testSnapRemoveData;
typedef testSnapRemoveData *testSnapRemoveDataPtr;
struct _testSnapRemoveData {
    virDomainObjPtr vm;
    bool current;
};

6395
static int
6396
testDomainSnapshotDiscardAll(void *payload,
6397 6398
                             const void *name ATTRIBUTE_UNUSED,
                             void *data)
6399 6400 6401 6402 6403 6404 6405
{
    virDomainSnapshotObjPtr snap = payload;
    testSnapRemoveDataPtr curr = data;

    if (snap->def->current)
        curr->current = true;
    virDomainSnapshotObjListRemove(curr->vm->snapshots, snap);
6406
    return 0;
6407 6408 6409 6410 6411 6412 6413 6414 6415 6416 6417
}

typedef struct _testSnapReparentData testSnapReparentData;
typedef testSnapReparentData *testSnapReparentDataPtr;
struct _testSnapReparentData {
    virDomainSnapshotObjPtr parent;
    virDomainObjPtr vm;
    int err;
    virDomainSnapshotObjPtr last;
};

6418
static int
6419 6420 6421 6422 6423 6424 6425
testDomainSnapshotReparentChildren(void *payload,
                                   const void *name ATTRIBUTE_UNUSED,
                                   void *data)
{
    virDomainSnapshotObjPtr snap = payload;
    testSnapReparentDataPtr rep = data;

6426
    if (rep->err < 0)
6427
        return 0;
6428 6429 6430 6431 6432 6433 6434

    VIR_FREE(snap->def->parent);
    snap->parent = rep->parent;

    if (rep->parent->def &&
        VIR_STRDUP(snap->def->parent, rep->parent->def->name) < 0) {
        rep->err = -1;
6435
        return 0;
6436 6437 6438 6439
    }

    if (!snap->sibling)
        rep->last = snap;
6440
    return 0;
6441 6442 6443 6444 6445 6446 6447 6448 6449 6450 6451 6452 6453 6454 6455 6456 6457 6458 6459 6460 6461 6462 6463 6464 6465 6466 6467 6468 6469
}

static int
testDomainSnapshotDelete(virDomainSnapshotPtr snapshot,
                         unsigned int flags)
{
    virDomainObjPtr vm = NULL;
    virDomainSnapshotObjPtr snap = NULL;
    virDomainSnapshotObjPtr parentsnap = NULL;
    int ret = -1;

    virCheckFlags(VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN |
                  VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN_ONLY, -1);

    if (!(vm = testDomObjFromSnapshot(snapshot)))
        return -1;

    if (!(snap = testSnapObjFromSnapshot(vm, snapshot)))
        goto cleanup;

    if (flags & (VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN |
                 VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN_ONLY)) {
        testSnapRemoveData rem;
        rem.vm = vm;
        rem.current = false;
        virDomainSnapshotForEachDescendant(snap,
                                           testDomainSnapshotDiscardAll,
                                           &rem);
        if (rem.current) {
6470
            if (flags & VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN_ONLY)
6471 6472 6473 6474 6475 6476 6477 6478 6479 6480 6481 6482 6483 6484 6485 6486 6487 6488 6489 6490 6491 6492 6493 6494 6495 6496 6497 6498 6499 6500 6501 6502 6503 6504 6505 6506 6507 6508 6509 6510 6511 6512 6513
                snap->def->current = true;
            vm->current_snapshot = snap;
        }
    } else if (snap->nchildren) {
        testSnapReparentData rep;
        rep.parent = snap->parent;
        rep.vm = vm;
        rep.err = 0;
        rep.last = NULL;
        virDomainSnapshotForEachChild(snap,
                                      testDomainSnapshotReparentChildren,
                                      &rep);
        if (rep.err < 0)
            goto cleanup;

        /* Can't modify siblings during ForEachChild, so do it now.  */
        snap->parent->nchildren += snap->nchildren;
        rep.last->sibling = snap->parent->first_child;
        snap->parent->first_child = snap->first_child;
    }

    if (flags & VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN_ONLY) {
        snap->nchildren = 0;
        snap->first_child = NULL;
    } else {
        virDomainSnapshotDropParent(snap);
        if (snap == vm->current_snapshot) {
            if (snap->def->parent) {
                parentsnap = virDomainSnapshotFindByName(vm->snapshots,
                                                         snap->def->parent);
                if (!parentsnap) {
                    VIR_WARN("missing parent snapshot matching name '%s'",
                             snap->def->parent);
                } else {
                    parentsnap->def->current = true;
                }
            }
            vm->current_snapshot = parentsnap;
        }
        virDomainSnapshotObjListRemove(vm->snapshots, snap);
    }

    ret = 0;
6514
 cleanup:
6515
    virDomainObjEndAPI(&vm);
6516 6517 6518 6519 6520 6521 6522
    return ret;
}

static int
testDomainRevertToSnapshot(virDomainSnapshotPtr snapshot,
                           unsigned int flags)
{
6523
    testDriverPtr privconn = snapshot->domain->conn->privateData;
6524 6525
    virDomainObjPtr vm = NULL;
    virDomainSnapshotObjPtr snap = NULL;
6526 6527
    virObjectEventPtr event = NULL;
    virObjectEventPtr event2 = NULL;
6528 6529 6530 6531 6532 6533 6534 6535 6536 6537 6538 6539 6540 6541 6542 6543 6544 6545 6546 6547 6548 6549 6550 6551 6552 6553 6554 6555 6556 6557 6558 6559 6560 6561 6562 6563 6564 6565 6566 6567 6568 6569 6570 6571 6572 6573 6574 6575 6576 6577 6578 6579 6580 6581 6582 6583 6584 6585 6586 6587 6588 6589 6590
    virDomainDefPtr config = NULL;
    int ret = -1;

    virCheckFlags(VIR_DOMAIN_SNAPSHOT_REVERT_RUNNING |
                  VIR_DOMAIN_SNAPSHOT_REVERT_PAUSED |
                  VIR_DOMAIN_SNAPSHOT_REVERT_FORCE, -1);

    /* We have the following transitions, which create the following events:
     * 1. inactive -> inactive: none
     * 2. inactive -> running:  EVENT_STARTED
     * 3. inactive -> paused:   EVENT_STARTED, EVENT_PAUSED
     * 4. running  -> inactive: EVENT_STOPPED
     * 5. running  -> running:  none
     * 6. running  -> paused:   EVENT_PAUSED
     * 7. paused   -> inactive: EVENT_STOPPED
     * 8. paused   -> running:  EVENT_RESUMED
     * 9. paused   -> paused:   none
     * Also, several transitions occur even if we fail partway through,
     * and use of FORCE can cause multiple transitions.
     */

    if (!(vm = testDomObjFromSnapshot(snapshot)))
        return -1;

    if (!(snap = testSnapObjFromSnapshot(vm, snapshot)))
        goto cleanup;

    if (!vm->persistent &&
        snap->def->state != VIR_DOMAIN_RUNNING &&
        snap->def->state != VIR_DOMAIN_PAUSED &&
        (flags & (VIR_DOMAIN_SNAPSHOT_REVERT_RUNNING |
                  VIR_DOMAIN_SNAPSHOT_REVERT_PAUSED)) == 0) {
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                       _("transient domain needs to request run or pause "
                         "to revert to inactive snapshot"));
        goto cleanup;
    }

    if (!(flags & VIR_DOMAIN_SNAPSHOT_REVERT_FORCE)) {
        if (!snap->def->dom) {
            virReportError(VIR_ERR_SNAPSHOT_REVERT_RISKY,
                           _("snapshot '%s' lacks domain '%s' rollback info"),
                           snap->def->name, vm->def->name);
            goto cleanup;
        }
        if (virDomainObjIsActive(vm) &&
            !(snap->def->state == VIR_DOMAIN_RUNNING
              || snap->def->state == VIR_DOMAIN_PAUSED) &&
            (flags & (VIR_DOMAIN_SNAPSHOT_REVERT_RUNNING |
                      VIR_DOMAIN_SNAPSHOT_REVERT_PAUSED))) {
            virReportError(VIR_ERR_SNAPSHOT_REVERT_RISKY, "%s",
                           _("must respawn guest to start inactive snapshot"));
            goto cleanup;
        }
    }


    if (vm->current_snapshot) {
        vm->current_snapshot->def->current = false;
        vm->current_snapshot = NULL;
    }

    snap->def->current = true;
6591 6592
    config = virDomainDefCopy(snap->def->dom, privconn->caps,
                              privconn->xmlopt, NULL, true);
6593 6594 6595 6596 6597 6598 6599 6600 6601 6602 6603 6604
    if (!config)
        goto cleanup;

    if (snap->def->state == VIR_DOMAIN_RUNNING ||
        snap->def->state == VIR_DOMAIN_PAUSED) {
        /* Transitions 2, 3, 5, 6, 8, 9 */
        bool was_running = false;
        bool was_stopped = false;

        if (virDomainObjIsActive(vm)) {
            /* Transitions 5, 6, 8, 9 */
            /* Check for ABI compatibility.  */
6605 6606
            if (!virDomainDefCheckABIStability(vm->def, config,
                                               privconn->xmlopt)) {
6607 6608 6609 6610 6611 6612 6613 6614 6615 6616 6617 6618 6619
                virErrorPtr err = virGetLastError();

                if (!(flags & VIR_DOMAIN_SNAPSHOT_REVERT_FORCE)) {
                    /* Re-spawn error using correct category. */
                    if (err->code == VIR_ERR_CONFIG_UNSUPPORTED)
                        virReportError(VIR_ERR_SNAPSHOT_REVERT_RISKY, "%s",
                                       err->str2);
                    goto cleanup;
                }

                virResetError(err);
                testDomainShutdownState(snapshot->domain, vm,
                                        VIR_DOMAIN_SHUTOFF_FROM_SNAPSHOT);
6620
                event = virDomainEventLifecycleNewFromObj(vm,
6621 6622
                            VIR_DOMAIN_EVENT_STOPPED,
                            VIR_DOMAIN_EVENT_STOPPED_FROM_SNAPSHOT);
6623
                testObjectEventQueue(privconn, event);
6624 6625 6626 6627 6628 6629 6630 6631 6632 6633 6634
                goto load;
            }

            if (virDomainObjGetState(vm, NULL) == VIR_DOMAIN_RUNNING) {
                /* Transitions 5, 6 */
                was_running = true;
                virDomainObjSetState(vm, VIR_DOMAIN_PAUSED,
                                     VIR_DOMAIN_PAUSED_FROM_SNAPSHOT);
                /* Create an event now in case the restore fails, so
                 * that user will be alerted that they are now paused.
                 * If restore later succeeds, we might replace this. */
6635
                event = virDomainEventLifecycleNewFromObj(vm,
6636 6637 6638 6639 6640 6641 6642 6643 6644 6645 6646 6647 6648
                                VIR_DOMAIN_EVENT_SUSPENDED,
                                VIR_DOMAIN_EVENT_SUSPENDED_FROM_SNAPSHOT);
            }
            virDomainObjAssignDef(vm, config, false, NULL);

        } else {
            /* Transitions 2, 3 */
        load:
            was_stopped = true;
            virDomainObjAssignDef(vm, config, false, NULL);
            if (testDomainStartState(privconn, vm,
                                VIR_DOMAIN_RUNNING_FROM_SNAPSHOT) < 0)
                goto cleanup;
6649
            event = virDomainEventLifecycleNewFromObj(vm,
6650 6651 6652 6653 6654 6655 6656 6657 6658 6659 6660 6661 6662
                                VIR_DOMAIN_EVENT_STARTED,
                                VIR_DOMAIN_EVENT_STARTED_FROM_SNAPSHOT);
        }

        /* Touch up domain state.  */
        if (!(flags & VIR_DOMAIN_SNAPSHOT_REVERT_RUNNING) &&
            (snap->def->state == VIR_DOMAIN_PAUSED ||
             (flags & VIR_DOMAIN_SNAPSHOT_REVERT_PAUSED))) {
            /* Transitions 3, 6, 9 */
            virDomainObjSetState(vm, VIR_DOMAIN_PAUSED,
                                 VIR_DOMAIN_PAUSED_FROM_SNAPSHOT);
            if (was_stopped) {
                /* Transition 3, use event as-is and add event2 */
6663
                event2 = virDomainEventLifecycleNewFromObj(vm,
6664 6665 6666 6667 6668
                                VIR_DOMAIN_EVENT_SUSPENDED,
                                VIR_DOMAIN_EVENT_SUSPENDED_FROM_SNAPSHOT);
            } /* else transition 6 and 9 use event as-is */
        } else {
            /* Transitions 2, 5, 8 */
C
Cédric Bosdonnat 已提交
6669
            virObjectUnref(event);
6670 6671 6672 6673
            event = NULL;

            if (was_stopped) {
                /* Transition 2 */
6674
                event = virDomainEventLifecycleNewFromObj(vm,
6675 6676 6677 6678
                                VIR_DOMAIN_EVENT_STARTED,
                                VIR_DOMAIN_EVENT_STARTED_FROM_SNAPSHOT);
            } else if (was_running) {
                /* Transition 8 */
6679
                event = virDomainEventLifecycleNewFromObj(vm,
6680 6681 6682 6683 6684 6685 6686 6687 6688 6689 6690 6691
                                VIR_DOMAIN_EVENT_RESUMED,
                                VIR_DOMAIN_EVENT_RESUMED);
            }
        }
    } else {
        /* Transitions 1, 4, 7 */
        virDomainObjAssignDef(vm, config, false, NULL);

        if (virDomainObjIsActive(vm)) {
            /* Transitions 4, 7 */
            testDomainShutdownState(snapshot->domain, vm,
                                    VIR_DOMAIN_SHUTOFF_FROM_SNAPSHOT);
6692
            event = virDomainEventLifecycleNewFromObj(vm,
6693 6694 6695 6696 6697 6698 6699 6700 6701
                                    VIR_DOMAIN_EVENT_STOPPED,
                                    VIR_DOMAIN_EVENT_STOPPED_FROM_SNAPSHOT);
        }

        if (flags & (VIR_DOMAIN_SNAPSHOT_REVERT_RUNNING |
                     VIR_DOMAIN_SNAPSHOT_REVERT_PAUSED)) {
            /* Flush first event, now do transition 2 or 3 */
            bool paused = (flags & VIR_DOMAIN_SNAPSHOT_REVERT_PAUSED) != 0;

6702
            testObjectEventQueue(privconn, event);
6703
            event = virDomainEventLifecycleNewFromObj(vm,
6704 6705 6706
                            VIR_DOMAIN_EVENT_STARTED,
                            VIR_DOMAIN_EVENT_STARTED_FROM_SNAPSHOT);
            if (paused) {
6707
                event2 = virDomainEventLifecycleNewFromObj(vm,
6708 6709 6710 6711 6712 6713 6714 6715
                                VIR_DOMAIN_EVENT_SUSPENDED,
                                VIR_DOMAIN_EVENT_SUSPENDED_FROM_SNAPSHOT);
            }
        }
    }

    vm->current_snapshot = snap;
    ret = 0;
6716
 cleanup:
6717
    if (event) {
6718
        testObjectEventQueue(privconn, event);
6719
        testObjectEventQueue(privconn, event2);
C
Cole Robinson 已提交
6720
    } else {
C
Cédric Bosdonnat 已提交
6721
        virObjectUnref(event2);
6722
    }
6723
    virDomainObjEndAPI(&vm);
6724 6725 6726 6727 6728

    return ret;
}


6729

6730
static virHypervisorDriver testHypervisorDriver = {
6731
    .name = "Test",
6732 6733 6734
    .connectOpen = testConnectOpen, /* 0.1.1 */
    .connectClose = testConnectClose, /* 0.1.1 */
    .connectGetVersion = testConnectGetVersion, /* 0.1.1 */
6735
    .connectGetHostname = testConnectGetHostname, /* 0.6.3 */
6736
    .connectGetMaxVcpus = testConnectGetMaxVcpus, /* 0.3.2 */
6737
    .nodeGetInfo = testNodeGetInfo, /* 0.1.1 */
6738
    .nodeGetCPUStats = testNodeGetCPUStats, /* 2.3.0 */
6739
    .nodeGetFreeMemory = testNodeGetFreeMemory, /* 2.3.0 */
6740
    .nodeGetFreePages = testNodeGetFreePages, /* 2.3.0 */
6741
    .connectGetCapabilities = testConnectGetCapabilities, /* 0.2.1 */
6742
    .connectGetSysinfo = testConnectGetSysinfo, /* 2.3.0 */
6743
    .connectGetType = testConnectGetType, /* 2.3.0 */
6744 6745 6746
    .connectListDomains = testConnectListDomains, /* 0.1.1 */
    .connectNumOfDomains = testConnectNumOfDomains, /* 0.1.1 */
    .connectListAllDomains = testConnectListAllDomains, /* 0.9.13 */
6747
    .domainCreateXML = testDomainCreateXML, /* 0.1.4 */
6748 6749 6750 6751 6752 6753 6754 6755 6756 6757 6758 6759 6760 6761
    .domainLookupByID = testDomainLookupByID, /* 0.1.1 */
    .domainLookupByUUID = testDomainLookupByUUID, /* 0.1.1 */
    .domainLookupByName = testDomainLookupByName, /* 0.1.1 */
    .domainSuspend = testDomainSuspend, /* 0.1.1 */
    .domainResume = testDomainResume, /* 0.1.1 */
    .domainShutdown = testDomainShutdown, /* 0.1.1 */
    .domainShutdownFlags = testDomainShutdownFlags, /* 0.9.10 */
    .domainReboot = testDomainReboot, /* 0.1.1 */
    .domainDestroy = testDomainDestroy, /* 0.1.1 */
    .domainGetOSType = testDomainGetOSType, /* 0.1.9 */
    .domainGetMaxMemory = testDomainGetMaxMemory, /* 0.1.4 */
    .domainSetMaxMemory = testDomainSetMaxMemory, /* 0.1.1 */
    .domainSetMemory = testDomainSetMemory, /* 0.1.4 */
    .domainGetInfo = testDomainGetInfo, /* 0.1.1 */
6762 6763
    .domainGetState = testDomainGetState, /* 0.9.2 */
    .domainSave = testDomainSave, /* 0.3.2 */
6764
    .domainSaveFlags = testDomainSaveFlags, /* 0.9.4 */
6765
    .domainRestore = testDomainRestore, /* 0.3.2 */
6766
    .domainRestoreFlags = testDomainRestoreFlags, /* 0.9.4 */
6767
    .domainCoreDump = testDomainCoreDump, /* 0.3.2 */
6768
    .domainCoreDumpWithFormat = testDomainCoreDumpWithFormat, /* 1.2.3 */
6769
    .domainSetVcpus = testDomainSetVcpus, /* 0.1.4 */
6770 6771 6772 6773
    .domainSetVcpusFlags = testDomainSetVcpusFlags, /* 0.8.5 */
    .domainGetVcpusFlags = testDomainGetVcpusFlags, /* 0.8.5 */
    .domainPinVcpu = testDomainPinVcpu, /* 0.7.3 */
    .domainGetVcpus = testDomainGetVcpus, /* 0.7.3 */
6774
    .domainGetVcpuPinInfo = testDomainGetVcpuPinInfo, /* 1.2.18 */
6775 6776
    .domainGetMaxVcpus = testDomainGetMaxVcpus, /* 0.7.3 */
    .domainGetXMLDesc = testDomainGetXMLDesc, /* 0.1.4 */
6777 6778
    .connectListDefinedDomains = testConnectListDefinedDomains, /* 0.1.11 */
    .connectNumOfDefinedDomains = testConnectNumOfDefinedDomains, /* 0.1.11 */
6779 6780 6781
    .domainCreate = testDomainCreate, /* 0.1.11 */
    .domainCreateWithFlags = testDomainCreateWithFlags, /* 0.8.2 */
    .domainDefineXML = testDomainDefineXML, /* 0.1.11 */
6782
    .domainDefineXMLFlags = testDomainDefineXMLFlags, /* 1.2.12 */
6783
    .domainUndefine = testDomainUndefine, /* 0.1.11 */
6784
    .domainUndefineFlags = testDomainUndefineFlags, /* 0.9.4 */
6785 6786 6787
    .domainGetAutostart = testDomainGetAutostart, /* 0.3.2 */
    .domainSetAutostart = testDomainSetAutostart, /* 0.3.2 */
    .domainGetSchedulerType = testDomainGetSchedulerType, /* 0.3.2 */
6788 6789 6790 6791
    .domainGetSchedulerParameters = testDomainGetSchedulerParameters, /* 0.3.2 */
    .domainGetSchedulerParametersFlags = testDomainGetSchedulerParametersFlags, /* 0.9.2 */
    .domainSetSchedulerParameters = testDomainSetSchedulerParameters, /* 0.3.2 */
    .domainSetSchedulerParametersFlags = testDomainSetSchedulerParametersFlags, /* 0.9.2 */
6792 6793 6794
    .domainBlockStats = testDomainBlockStats, /* 0.7.0 */
    .domainInterfaceStats = testDomainInterfaceStats, /* 0.7.0 */
    .nodeGetCellsFreeMemory = testNodeGetCellsFreeMemory, /* 0.4.2 */
6795 6796 6797 6798
    .connectDomainEventRegister = testConnectDomainEventRegister, /* 0.6.0 */
    .connectDomainEventDeregister = testConnectDomainEventDeregister, /* 0.6.0 */
    .connectIsEncrypted = testConnectIsEncrypted, /* 0.7.3 */
    .connectIsSecure = testConnectIsSecure, /* 0.7.3 */
6799 6800 6801
    .domainIsActive = testDomainIsActive, /* 0.7.3 */
    .domainIsPersistent = testDomainIsPersistent, /* 0.7.3 */
    .domainIsUpdated = testDomainIsUpdated, /* 0.8.6 */
6802 6803 6804
    .connectDomainEventRegisterAny = testConnectDomainEventRegisterAny, /* 0.8.0 */
    .connectDomainEventDeregisterAny = testConnectDomainEventDeregisterAny, /* 0.8.0 */
    .connectIsAlive = testConnectIsAlive, /* 0.9.8 */
6805
    .nodeGetCPUMap = testNodeGetCPUMap, /* 1.0.0 */
6806
    .domainScreenshot = testDomainScreenshot, /* 1.0.5 */
6807 6808
    .domainGetMetadata = testDomainGetMetadata, /* 1.1.3 */
    .domainSetMetadata = testDomainSetMetadata, /* 1.1.3 */
6809
    .connectGetCPUModelNames = testConnectGetCPUModelNames, /* 1.1.3 */
6810 6811 6812 6813 6814 6815 6816 6817 6818 6819 6820 6821 6822 6823 6824 6825 6826
    .domainManagedSave = testDomainManagedSave, /* 1.1.4 */
    .domainHasManagedSaveImage = testDomainHasManagedSaveImage, /* 1.1.4 */
    .domainManagedSaveRemove = testDomainManagedSaveRemove, /* 1.1.4 */

    .domainSnapshotNum = testDomainSnapshotNum, /* 1.1.4 */
    .domainSnapshotListNames = testDomainSnapshotListNames, /* 1.1.4 */
    .domainListAllSnapshots = testDomainListAllSnapshots, /* 1.1.4 */
    .domainSnapshotGetXMLDesc = testDomainSnapshotGetXMLDesc, /* 1.1.4 */
    .domainSnapshotNumChildren = testDomainSnapshotNumChildren, /* 1.1.4 */
    .domainSnapshotListChildrenNames = testDomainSnapshotListChildrenNames, /* 1.1.4 */
    .domainSnapshotListAllChildren = testDomainSnapshotListAllChildren, /* 1.1.4 */
    .domainSnapshotLookupByName = testDomainSnapshotLookupByName, /* 1.1.4 */
    .domainHasCurrentSnapshot = testDomainHasCurrentSnapshot, /* 1.1.4 */
    .domainSnapshotGetParent = testDomainSnapshotGetParent, /* 1.1.4 */
    .domainSnapshotCurrent = testDomainSnapshotCurrent, /* 1.1.4 */
    .domainSnapshotIsCurrent = testDomainSnapshotIsCurrent, /* 1.1.4 */
    .domainSnapshotHasMetadata = testDomainSnapshotHasMetadata, /* 1.1.4 */
6827 6828 6829
    .domainSnapshotCreateXML = testDomainSnapshotCreateXML, /* 1.1.4 */
    .domainRevertToSnapshot = testDomainRevertToSnapshot, /* 1.1.4 */
    .domainSnapshotDelete = testDomainSnapshotDelete, /* 1.1.4 */
6830

E
Eric Blake 已提交
6831
    .connectBaselineCPU = testConnectBaselineCPU, /* 1.2.0 */
6832 6833 6834
};

static virNetworkDriver testNetworkDriver = {
6835 6836 6837 6838 6839
    .connectNumOfNetworks = testConnectNumOfNetworks, /* 0.3.2 */
    .connectListNetworks = testConnectListNetworks, /* 0.3.2 */
    .connectNumOfDefinedNetworks = testConnectNumOfDefinedNetworks, /* 0.3.2 */
    .connectListDefinedNetworks = testConnectListDefinedNetworks, /* 0.3.2 */
    .connectListAllNetworks = testConnectListAllNetworks, /* 0.10.2 */
6840 6841
    .connectNetworkEventRegisterAny = testConnectNetworkEventRegisterAny, /* 1.2.1 */
    .connectNetworkEventDeregisterAny = testConnectNetworkEventDeregisterAny, /* 1.2.1 */
6842 6843 6844 6845
    .networkLookupByUUID = testNetworkLookupByUUID, /* 0.3.2 */
    .networkLookupByName = testNetworkLookupByName, /* 0.3.2 */
    .networkCreateXML = testNetworkCreateXML, /* 0.3.2 */
    .networkDefineXML = testNetworkDefineXML, /* 0.3.2 */
6846
    .networkUndefine = testNetworkUndefine, /* 0.3.2 */
6847
    .networkUpdate = testNetworkUpdate, /* 0.10.2 */
6848
    .networkCreate = testNetworkCreate, /* 0.3.2 */
6849 6850 6851 6852 6853 6854 6855
    .networkDestroy = testNetworkDestroy, /* 0.3.2 */
    .networkGetXMLDesc = testNetworkGetXMLDesc, /* 0.3.2 */
    .networkGetBridgeName = testNetworkGetBridgeName, /* 0.3.2 */
    .networkGetAutostart = testNetworkGetAutostart, /* 0.3.2 */
    .networkSetAutostart = testNetworkSetAutostart, /* 0.3.2 */
    .networkIsActive = testNetworkIsActive, /* 0.7.3 */
    .networkIsPersistent = testNetworkIsPersistent, /* 0.7.3 */
6856 6857
};

L
Laine Stump 已提交
6858
static virInterfaceDriver testInterfaceDriver = {
6859 6860 6861 6862 6863 6864
    .connectNumOfInterfaces = testConnectNumOfInterfaces, /* 0.7.0 */
    .connectListInterfaces = testConnectListInterfaces, /* 0.7.0 */
    .connectNumOfDefinedInterfaces = testConnectNumOfDefinedInterfaces, /* 0.7.0 */
    .connectListDefinedInterfaces = testConnectListDefinedInterfaces, /* 0.7.0 */
    .interfaceLookupByName = testInterfaceLookupByName, /* 0.7.0 */
    .interfaceLookupByMACString = testInterfaceLookupByMACString, /* 0.7.0 */
6865 6866 6867 6868 6869 6870
    .interfaceGetXMLDesc = testInterfaceGetXMLDesc, /* 0.7.0 */
    .interfaceDefineXML = testInterfaceDefineXML, /* 0.7.0 */
    .interfaceUndefine = testInterfaceUndefine, /* 0.7.0 */
    .interfaceCreate = testInterfaceCreate, /* 0.7.0 */
    .interfaceDestroy = testInterfaceDestroy, /* 0.7.0 */
    .interfaceIsActive = testInterfaceIsActive, /* 0.7.3 */
6871 6872 6873
    .interfaceChangeBegin = testInterfaceChangeBegin,   /* 0.9.2 */
    .interfaceChangeCommit = testInterfaceChangeCommit,  /* 0.9.2 */
    .interfaceChangeRollback = testInterfaceChangeRollback, /* 0.9.2 */
L
Laine Stump 已提交
6874 6875 6876
};


6877
static virStorageDriver testStorageDriver = {
6878 6879 6880 6881 6882 6883
    .connectNumOfStoragePools = testConnectNumOfStoragePools, /* 0.5.0 */
    .connectListStoragePools = testConnectListStoragePools, /* 0.5.0 */
    .connectNumOfDefinedStoragePools = testConnectNumOfDefinedStoragePools, /* 0.5.0 */
    .connectListDefinedStoragePools = testConnectListDefinedStoragePools, /* 0.5.0 */
    .connectListAllStoragePools = testConnectListAllStoragePools, /* 0.10.2 */
    .connectFindStoragePoolSources = testConnectFindStoragePoolSources, /* 0.5.0 */
6884 6885
    .connectStoragePoolEventRegisterAny = testConnectStoragePoolEventRegisterAny, /* 2.0.0 */
    .connectStoragePoolEventDeregisterAny = testConnectStoragePoolEventDeregisterAny, /* 2.0.0 */
6886 6887 6888
    .storagePoolLookupByName = testStoragePoolLookupByName, /* 0.5.0 */
    .storagePoolLookupByUUID = testStoragePoolLookupByUUID, /* 0.5.0 */
    .storagePoolLookupByVolume = testStoragePoolLookupByVolume, /* 0.5.0 */
6889 6890
    .storagePoolCreateXML = testStoragePoolCreateXML, /* 0.5.0 */
    .storagePoolDefineXML = testStoragePoolDefineXML, /* 0.5.0 */
6891 6892
    .storagePoolBuild = testStoragePoolBuild, /* 0.5.0 */
    .storagePoolUndefine = testStoragePoolUndefine, /* 0.5.0 */
6893
    .storagePoolCreate = testStoragePoolCreate, /* 0.5.0 */
6894 6895 6896 6897 6898 6899 6900
    .storagePoolDestroy = testStoragePoolDestroy, /* 0.5.0 */
    .storagePoolDelete = testStoragePoolDelete, /* 0.5.0 */
    .storagePoolRefresh = testStoragePoolRefresh, /* 0.5.0 */
    .storagePoolGetInfo = testStoragePoolGetInfo, /* 0.5.0 */
    .storagePoolGetXMLDesc = testStoragePoolGetXMLDesc, /* 0.5.0 */
    .storagePoolGetAutostart = testStoragePoolGetAutostart, /* 0.5.0 */
    .storagePoolSetAutostart = testStoragePoolSetAutostart, /* 0.5.0 */
6901
    .storagePoolNumOfVolumes = testStoragePoolNumOfVolumes, /* 0.5.0 */
6902 6903 6904
    .storagePoolListVolumes = testStoragePoolListVolumes, /* 0.5.0 */
    .storagePoolListAllVolumes = testStoragePoolListAllVolumes, /* 0.10.2 */

6905 6906 6907 6908 6909 6910 6911 6912 6913
    .storageVolLookupByName = testStorageVolLookupByName, /* 0.5.0 */
    .storageVolLookupByKey = testStorageVolLookupByKey, /* 0.5.0 */
    .storageVolLookupByPath = testStorageVolLookupByPath, /* 0.5.0 */
    .storageVolCreateXML = testStorageVolCreateXML, /* 0.5.0 */
    .storageVolCreateXMLFrom = testStorageVolCreateXMLFrom, /* 0.6.4 */
    .storageVolDelete = testStorageVolDelete, /* 0.5.0 */
    .storageVolGetInfo = testStorageVolGetInfo, /* 0.5.0 */
    .storageVolGetXMLDesc = testStorageVolGetXMLDesc, /* 0.5.0 */
    .storageVolGetPath = testStorageVolGetPath, /* 0.5.0 */
6914 6915
    .storagePoolIsActive = testStoragePoolIsActive, /* 0.7.3 */
    .storagePoolIsPersistent = testStoragePoolIsPersistent, /* 0.7.3 */
6916 6917
};

6918
static virNodeDeviceDriver testNodeDeviceDriver = {
6919 6920
    .connectNodeDeviceEventRegisterAny = testConnectNodeDeviceEventRegisterAny, /* 2.2.0 */
    .connectNodeDeviceEventDeregisterAny = testConnectNodeDeviceEventDeregisterAny, /* 2.2.0 */
6921 6922 6923 6924 6925 6926 6927 6928 6929
    .nodeNumOfDevices = testNodeNumOfDevices, /* 0.7.2 */
    .nodeListDevices = testNodeListDevices, /* 0.7.2 */
    .nodeDeviceLookupByName = testNodeDeviceLookupByName, /* 0.7.2 */
    .nodeDeviceGetXMLDesc = testNodeDeviceGetXMLDesc, /* 0.7.2 */
    .nodeDeviceGetParent = testNodeDeviceGetParent, /* 0.7.2 */
    .nodeDeviceNumOfCaps = testNodeDeviceNumOfCaps, /* 0.7.2 */
    .nodeDeviceListCaps = testNodeDeviceListCaps, /* 0.7.2 */
    .nodeDeviceCreateXML = testNodeDeviceCreateXML, /* 0.7.3 */
    .nodeDeviceDestroy = testNodeDeviceDestroy, /* 0.7.3 */
6930 6931
};

6932 6933 6934 6935 6936 6937 6938 6939
static virConnectDriver testConnectDriver = {
    .hypervisorDriver = &testHypervisorDriver,
    .interfaceDriver = &testInterfaceDriver,
    .networkDriver = &testNetworkDriver,
    .nodeDeviceDriver = &testNodeDeviceDriver,
    .nwfilterDriver = NULL,
    .secretDriver = NULL,
    .storageDriver = &testStorageDriver,
6940 6941
};

6942 6943 6944 6945 6946 6947 6948 6949
/**
 * testRegister:
 *
 * Registers the test driver
 */
int
testRegister(void)
{
6950 6951
    return virRegisterConnectDriver(&testConnectDriver,
                                    false);
6952
}