test_driver.c 150.3 KB
Newer Older
1 2 3
/*
 * test.c: A "mock" hypervisor for use by application unit tests
 *
4
 * Copyright (C) 2006-2010 Red Hat, Inc.
5
 * Copyright (C) 2006 Daniel P. Berrange
6
 *
7 8 9 10 11 12 13 14 15 16 17 18 19
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA
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

34 35

#include "virterror_internal.h"
36
#include "datatypes.h"
37
#include "test_driver.h"
38
#include "buf.h"
39
#include "util.h"
40
#include "uuid.h"
41
#include "capabilities.h"
42
#include "memory.h"
43
#include "network_conf.h"
L
Laine Stump 已提交
44
#include "interface_conf.h"
45
#include "domain_conf.h"
46 47
#include "domain_event.h"
#include "event.h"
C
Cole Robinson 已提交
48
#include "storage_conf.h"
49
#include "node_device_conf.h"
50
#include "xml.h"
51
#include "threads.h"
52
#include "logging.h"
53

54 55
#define VIR_FROM_THIS VIR_FROM_TEST

56 57 58 59 60 61 62 63 64
/* Driver specific info to carry with a domain */
struct _testDomainObjPrivate {
    virVcpuInfoPtr vcpu_infos;

    unsigned char *cpumaps;
};
typedef struct _testDomainObjPrivate testDomainObjPrivate;
typedef struct _testDomainObjPrivate *testDomainObjPrivatePtr;

65 66 67 68 69 70 71 72 73 74 75
#define MAX_CPUS 128

struct _testCell {
    unsigned long mem;
    int numCpus;
    int cpus[MAX_CPUS];
};
typedef struct _testCell testCell;
typedef struct _testCell *testCellPtr;

#define MAX_CELLS 128
76

77
struct _testConn {
78
    virMutex lock;
79

80 81
    char path[PATH_MAX];
    int nextDomID;
82
    virCapsPtr caps;
83
    virNodeInfo nodeInfo;
84
    virDomainObjList domains;
85
    virNetworkObjList networks;
L
Laine Stump 已提交
86
    virInterfaceObjList ifaces;
C
Cole Robinson 已提交
87
    virStoragePoolObjList pools;
88
    virNodeDeviceObjList devs;
89 90
    int numCells;
    testCell cells[MAX_CELLS];
91 92 93 94 95 96 97


    /* An array of callbacks */
    virDomainEventCallbackListPtr domainEventCallbacks;
    virDomainEventQueuePtr domainEventQueue;
    int domainEventTimer;
    int domainEventDispatching;
98 99 100
};
typedef struct _testConn testConn;
typedef struct _testConn *testConnPtr;
101

102
#define TEST_MODEL "i686"
103
#define TEST_MODEL_WORDSIZE 32
104
#define TEST_EMULATOR "/usr/bin/test-hv"
105

106
static const virNodeInfo defaultNodeInfo = {
107
    TEST_MODEL,
108 109 110 111 112 113 114
    1024*1024*3, /* 3 GB */
    16,
    1400,
    2,
    2,
    2,
    2,
115 116
};

117

118 119 120
#define testError(code, ...)                                      \
        virReportErrorHelper(NULL, VIR_FROM_TEST, code, __FILE__, \
                             __FUNCTION__, __LINE__, __VA_ARGS__)
121

122 123 124 125 126 127
static int testClose(virConnectPtr conn);
static void testDomainEventFlush(int timer, void *opaque);
static void testDomainEventQueue(testConnPtr driver,
                                 virDomainEventPtr event);


128 129
static void testDriverLock(testConnPtr driver)
{
130
    virMutexLock(&driver->lock);
131 132 133 134
}

static void testDriverUnlock(testConnPtr driver)
{
135
    virMutexUnlock(&driver->lock);
136 137
}

138 139 140 141 142 143 144 145 146 147 148 149 150 151
static void *testDomainObjPrivateAlloc(void)
{
    testDomainObjPrivatePtr priv;

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

    return priv;
}

static void testDomainObjPrivateFree(void *data)
{
    testDomainObjPrivatePtr priv = data;

D
Daniel P. Berrange 已提交
152
    VIR_FREE(priv->vcpu_infos);
153 154 155 156 157
    VIR_FREE(priv->cpumaps);
    VIR_FREE(priv);
}


158 159
static virCapsPtr
testBuildCapabilities(virConnectPtr conn) {
160
    testConnPtr privconn = conn->privateData;
161 162 163 164
    virCapsPtr caps;
    virCapsGuestPtr guest;
    const char *const guest_types[] = { "hvm", "xen" };
    int i;
165

166 167
    if ((caps = virCapabilitiesNew(TEST_MODEL, 0, 0)) == NULL)
        goto no_memory;
168

169 170 171 172
    if (virCapabilitiesAddHostFeature(caps, "pae") < 0)
        goto no_memory;
    if (virCapabilitiesAddHostFeature(caps ,"nonpae") < 0)
        goto no_memory;
173

174 175 176 177
    for (i = 0; i < privconn->numCells; i++) {
        if (virCapabilitiesAddHostNUMACell(caps, i, privconn->cells[i].numCpus,
                                           privconn->cells[i].cpus) < 0)
            goto no_memory;
178 179
    }

180 181 182 183 184 185 186 187 188 189
    for (i = 0; i < ARRAY_CARDINALITY(guest_types) ; i++) {
        if ((guest = virCapabilitiesAddGuest(caps,
                                             guest_types[i],
                                             TEST_MODEL,
                                             TEST_MODEL_WORDSIZE,
                                             TEST_EMULATOR,
                                             NULL,
                                             0,
                                             NULL)) == NULL)
            goto no_memory;
190

191 192 193 194 195 196 197
        if (virCapabilitiesAddGuestDomain(guest,
                                          "test",
                                          NULL,
                                          NULL,
                                          0,
                                          NULL) == NULL)
            goto no_memory;
198

199 200 201 202
        if (virCapabilitiesAddGuestFeature(guest, "pae", 1, 1) == NULL)
            goto no_memory;
        if (virCapabilitiesAddGuestFeature(guest ,"nonpae", 1, 1) == NULL)
            goto no_memory;
203 204
    }

205 206 207
    caps->privateDataAllocFunc = testDomainObjPrivateAlloc;
    caps->privateDataFreeFunc = testDomainObjPrivateFree;

208 209 210 211 212 213 214 215
    caps->host.secModel.model = strdup("testSecurity");
    if (!caps->host.secModel.model)
        goto no_memory;

    caps->host.secModel.doi = strdup("");
    if (!caps->host.secModel.doi)
        goto no_memory;

216
    return caps;
217

218
no_memory:
219
    virReportOOMError();
220 221
    virCapabilitiesFree(caps);
    return NULL;
222 223
}

224

225 226 227 228 229 230 231 232 233 234
static const char *defaultDomainXML =
"<domain type='test'>"
"  <name>test</name>"
"  <memory>8388608</memory>"
"  <currentMemory>2097152</currentMemory>"
"  <vcpu>2</vcpu>"
"  <os>"
"    <type>hvm</type>"
"  </os>"
"</domain>";
235 236


237 238 239 240 241 242 243 244 245 246 247
static const char *defaultNetworkXML =
"<network>"
"  <name>default</name>"
"  <bridge name='virbr0' />"
"  <forward/>"
"  <ip address='192.168.122.1' netmask='255.255.255.0'>"
"    <dhcp>"
"      <range start='192.168.122.2' end='192.168.122.254' />"
"    </dhcp>"
"  </ip>"
"</network>";
248

L
Laine Stump 已提交
249 250 251 252 253 254 255 256 257 258 259
static const char *defaultInterfaceXML =
"<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>"
"</interface>";

C
Cole Robinson 已提交
260 261 262 263 264 265 266 267
static const char *defaultPoolXML =
"<pool type='dir'>"
"  <name>default-pool</name>"
"  <target>"
"    <path>/default-pool</path>"
"  </target>"
"</pool>";

268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290
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";

291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308
static const char *defaultNodeXML =
"<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>";

309
static const unsigned long long defaultPoolCap = (100 * 1024 * 1024 * 1024ull);
C
Cole Robinson 已提交
310 311
static const unsigned long long defaultPoolAlloc = 0;

312
static int testStoragePoolObjSetDefaults(virStoragePoolObjPtr pool);
313
static int testNodeGetInfo(virConnectPtr conn, virNodeInfoPtr info);
314

315
static char *
316
testDomainGenerateIfname(virDomainDefPtr domdef) {
317 318 319 320 321 322 323 324
    int maxif = 1024;
    int ifctr, i;

    for (ifctr = 0; ifctr < maxif; ++ifctr) {
        char *ifname;
        int found = 0;

        if (virAsprintf(&ifname, "testnet%d", ifctr) < 0) {
325
            virReportOOMError();
326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341
            return NULL;
        }

        /* Generate network interface names */
        for (i = 0 ; i < domdef->nnets ; i++) {
            if (domdef->nets[i]->ifname &&
                STREQ (domdef->nets[i]->ifname, ifname)) {
                found = 1;
                break;
            }
        }

        if (!found)
            return ifname;
    }

342
    testError(VIR_ERR_INTERNAL_ERROR,
343 344 345 346
              _("Exceeded max iface limit %d"), maxif);
    return NULL;
}

347
static int
348
testDomainGenerateIfnames(virDomainDefPtr domdef)
349 350 351 352 353 354 355 356
{
    int i = 0;

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

357
        ifname = testDomainGenerateIfname(domdef);
358
        if (!ifname)
359
            return -1;
360 361 362 363

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

364
    return 0;
365 366
}

367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430
/* Helper to update info for a single VCPU */
static int
testDomainUpdateVCPU(virConnectPtr conn ATTRIBUTE_UNUSED,
                     virDomainObjPtr dom,
                     int vcpu,
                     int maplen,
                     int maxcpu)
{
    testDomainObjPrivatePtr privdata = dom->privateData;
    virVcpuInfoPtr info = &privdata->vcpu_infos[vcpu];
    unsigned char *cpumap = VIR_GET_CPUMAP(privdata->cpumaps, maplen, vcpu);
    int j;

    memset(info, 0, sizeof(virVcpuInfo));
    memset(cpumap, 0, maplen);

    info->number    = vcpu;
    info->state     = VIR_VCPU_RUNNING;
    info->cpuTime   = 5000000;
    info->cpu       = 0;

    if (dom->def->cpumask) {
        for (j = 0; j < maxcpu && j < VIR_DOMAIN_CPUMASK_LEN; ++j) {
            if (dom->def->cpumask[j]) {
                VIR_USE_CPU(cpumap, j);
                info->cpu = j;
            }
        }
    } else {
        for (j = 0; j < maxcpu; ++j) {
            if ((j % 3) == 0) {
                /* Mark of every third CPU as usable */
                VIR_USE_CPU(cpumap, j);
                info->cpu = j;
            }
        }
    }

    return 0;
}

/*
 * Update domain VCPU amount and info
 *
 * @conn: virConnectPtr
 * @dom : domain needing updates
 * @nvcpus: New amount of vcpus for the domain
 * @clear_all: If true, rebuild info for ALL vcpus, not just newly added vcpus
 */
static int
testDomainUpdateVCPUs(virConnectPtr conn,
                      virDomainObjPtr dom,
                      int nvcpus,
                      unsigned int clear_all)
{
    testConnPtr privconn = conn->privateData;
    testDomainObjPrivatePtr privdata = dom->privateData;
    int i, ret = -1;
    int cpumaplen, maxcpu;

    maxcpu  = VIR_NODEINFO_MAXCPUS(privconn->nodeInfo);
    cpumaplen = VIR_CPU_MAPLEN(maxcpu);

    if (VIR_REALLOC_N(privdata->vcpu_infos, nvcpus) < 0) {
431
        virReportOOMError();
432 433 434 435
        goto cleanup;
    }

    if (VIR_REALLOC_N(privdata->cpumaps, nvcpus * cpumaplen) < 0) {
436
        virReportOOMError();
437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458
        goto cleanup;
    }

    /* Set running VCPU and cpumap state */
    if (clear_all) {
        for (i = 0; i < nvcpus; ++i)
            if (testDomainUpdateVCPU(conn, dom, i, cpumaplen, maxcpu) < 0)
                goto cleanup;

    } else if (nvcpus > dom->def->vcpus) {
        /* VCPU amount has grown, populate info for the new vcpus */
        for (i = dom->def->vcpus; i < nvcpus; ++i)
            if (testDomainUpdateVCPU(conn, dom, i, cpumaplen, maxcpu) < 0)
                goto cleanup;
    }

    ret = 0;
cleanup:
    return ret;
}

/* Set up domain runtime state */
459 460 461 462 463
static int
testDomainStartState(virConnectPtr conn,
                     virDomainObjPtr dom)
{
    testConnPtr privconn = conn->privateData;
464
    int ret = -1;
465

466 467 468 469
    if (testDomainUpdateVCPUs(conn, dom, dom->def->vcpus, 1) < 0)
        goto cleanup;

    /* Set typical run state */
470 471 472
    dom->state = VIR_DOMAIN_RUNNING;
    dom->def->id = privconn->nextDomID++;

473 474 475
    ret = 0;
cleanup:
    return ret;
476
}
477

478 479 480 481
static void
testDomainShutdownState(virDomainPtr domain,
                        virDomainObjPtr privdom)
{
482 483 484 485 486 487
    if (privdom->newDef) {
        virDomainDefFree(privdom->def);
        privdom->def = privdom->newDef;
        privdom->newDef = NULL;
    }

488 489 490 491 492
    privdom->state = VIR_DOMAIN_SHUTOFF;
    privdom->def->id = -1;
    domain->id = -1;
}

493
static int testOpenDefault(virConnectPtr conn) {
494 495
    int u;
    struct timeval tv;
496
    testConnPtr privconn;
497 498 499 500
    virDomainDefPtr domdef = NULL;
    virDomainObjPtr domobj = NULL;
    virNetworkDefPtr netdef = NULL;
    virNetworkObjPtr netobj = NULL;
L
Laine Stump 已提交
501 502
    virInterfaceDefPtr interfacedef = NULL;
    virInterfaceObjPtr interfaceobj = NULL;
C
Cole Robinson 已提交
503 504
    virStoragePoolDefPtr pooldef = NULL;
    virStoragePoolObjPtr poolobj = NULL;
505 506
    virNodeDeviceDefPtr nodedef = NULL;
    virNodeDeviceObjPtr nodeobj = NULL;
507

508
    if (VIR_ALLOC(privconn) < 0) {
509
        virReportOOMError();
510 511
        return VIR_DRV_OPEN_ERROR;
    }
512
    if (virMutexInit(&privconn->lock) < 0) {
513
        testError(VIR_ERR_INTERNAL_ERROR,
514 515 516 517 518
                  "%s", _("cannot initialize mutex"));
        VIR_FREE(privconn);
        return VIR_DRV_OPEN_ERROR;
    }

519
    testDriverLock(privconn);
520
    conn->privateData = privconn;
521 522

    if (gettimeofday(&tv, NULL) < 0) {
523
        virReportSystemError(errno,
524
                             "%s", _("getting time of day"));
525
        goto error;
526 527
    }

528 529 530
    if (virDomainObjListInit(&privconn->domains) < 0)
        goto error;

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

533 534 535 536 537 538 539 540 541 542
    // Numa setup
    privconn->numCells = 2;
    for (u = 0; u < 2; ++u) {
        privconn->cells[u].numCpus = 8;
        privconn->cells[u].mem = (u + 1) * 2048 * 1024;
    }
    for (u = 0 ; u < 16 ; u++) {
        privconn->cells[u % 2].cpus[(u / 2)] = u;
    }

543 544 545 546 547
    if (!(privconn->caps = testBuildCapabilities(conn)))
        goto error;

    privconn->nextDomID = 1;

548
    if (!(domdef = virDomainDefParseString(privconn->caps,
549 550
                                           defaultDomainXML,
                                           VIR_DOMAIN_XML_INACTIVE)))
551
        goto error;
552
    if (testDomainGenerateIfnames(domdef) < 0)
553
        goto error;
554
    if (!(domobj = virDomainAssignDef(privconn->caps,
555
                                      &privconn->domains, domdef, false)))
556 557
        goto error;
    domdef = NULL;
558 559 560 561 562 563

    if (testDomainStartState(conn, domobj) < 0) {
        virDomainObjUnlock(domobj);
        goto error;
    }

564
    domobj->persistent = 1;
565
    virDomainObjUnlock(domobj);
566

567
    if (!(netdef = virNetworkDefParseString(defaultNetworkXML)))
568
        goto error;
569
    if (!(netobj = virNetworkAssignDef(&privconn->networks, netdef))) {
570 571 572 573 574
        virNetworkDefFree(netdef);
        goto error;
    }
    netobj->active = 1;
    netobj->persistent = 1;
575
    virNetworkObjUnlock(netobj);
576

577
    if (!(interfacedef = virInterfaceDefParseString(defaultInterfaceXML)))
L
Laine Stump 已提交
578
        goto error;
579
    if (!(interfaceobj = virInterfaceAssignDef(&privconn->ifaces, interfacedef))) {
L
Laine Stump 已提交
580 581 582 583 584 585
        virInterfaceDefFree(interfacedef);
        goto error;
    }
    interfaceobj->active = 1;
    virInterfaceObjUnlock(interfaceobj);

586
    if (!(pooldef = virStoragePoolDefParseString(defaultPoolXML)))
C
Cole Robinson 已提交
587 588
        goto error;

589
    if (!(poolobj = virStoragePoolObjAssignDef(&privconn->pools,
C
Cole Robinson 已提交
590 591 592 593
                                               pooldef))) {
        virStoragePoolDefFree(pooldef);
        goto error;
    }
594

595
    if (testStoragePoolObjSetDefaults(poolobj) == -1) {
596
        virStoragePoolObjUnlock(poolobj);
C
Cole Robinson 已提交
597
        goto error;
598
    }
C
Cole Robinson 已提交
599
    poolobj->active = 1;
600
    virStoragePoolObjUnlock(poolobj);
C
Cole Robinson 已提交
601

602
    /* Init default node device */
603
    if (!(nodedef = virNodeDeviceDefParseString(defaultNodeXML, 0)))
604
        goto error;
605
    if (!(nodeobj = virNodeDeviceAssignDef(&privconn->devs,
606 607 608 609 610 611
                                           nodedef))) {
        virNodeDeviceDefFree(nodedef);
        goto error;
    }
    virNodeDeviceObjUnlock(nodeobj);

612
    testDriverUnlock(privconn);
613

614 615 616
    return VIR_DRV_OPEN_SUCCESS;

error:
617
    virDomainObjListDeinit(&privconn->domains);
618
    virNetworkObjListFree(&privconn->networks);
L
Laine Stump 已提交
619
    virInterfaceObjListFree(&privconn->ifaces);
C
Cole Robinson 已提交
620
    virStoragePoolObjListFree(&privconn->pools);
621
    virNodeDeviceObjListFree(&privconn->devs);
622
    virCapabilitiesFree(privconn->caps);
623
    testDriverUnlock(privconn);
624
    conn->privateData = NULL;
625
    VIR_FREE(privconn);
626
    virDomainDefFree(domdef);
627
    return VIR_DRV_OPEN_ERROR;
628 629 630 631
}


static char *testBuildFilename(const char *relativeTo,
632 633 634 635 636 637 638 639
                               const char *filename) {
    char *offset;
    int baseLen;
    if (!filename || filename[0] == '\0')
        return (NULL);
    if (filename[0] == '/')
        return strdup(filename);

640
    offset = strrchr(relativeTo, '/');
641
    if ((baseLen = (offset-relativeTo+1))) {
642
        char *absFile;
C
Chris Lalancette 已提交
643 644
        int totalLen = baseLen + strlen(filename) + 1;
        if (VIR_ALLOC_N(absFile, totalLen) < 0)
645
            return NULL;
C
Chris Lalancette 已提交
646 647 648 649
        if (virStrncpy(absFile, relativeTo, baseLen, totalLen) == NULL) {
            VIR_FREE(absFile);
            return NULL;
        }
650 651 652 653 654
        strcat(absFile, filename);
        return absFile;
    } else {
        return strdup(filename);
    }
655 656
}

657
static int testOpenVolumesForPool(xmlDocPtr xml,
658 659 660 661 662 663 664
                                  xmlXPathContextPtr ctxt,
                                  const char *file,
                                  virStoragePoolObjPtr pool,
                                  int poolidx) {
    char *vol_xpath;
    int i, ret, func_ret = -1;
    xmlNodePtr *vols = NULL;
665
    virStorageVolDefPtr def = NULL;
666 667 668

    /* Find storage volumes */
    if (virAsprintf(&vol_xpath, "/node/pool[%d]/volume", poolidx) < 0) {
669
        virReportOOMError();
670 671 672
        goto error;
    }

673
    ret = virXPathNodeSet(vol_xpath, ctxt, &vols);
674 675
    VIR_FREE(vol_xpath);
    if (ret < 0) {
676
        testError(VIR_ERR_XML_ERROR,
677 678 679 680 681 682 683 684 685 686
                  _("node vol list for pool '%s'"), pool->def->name);
        goto error;
    }

    for (i = 0 ; i < ret ; i++) {
        char *relFile = virXMLPropString(vols[i], "file");
        if (relFile != NULL) {
            char *absFile = testBuildFilename(file, relFile);
            VIR_FREE(relFile);
            if (!absFile) {
687
                testError(VIR_ERR_INTERNAL_ERROR, "%s",
688 689 690 691
                          _("resolving volume filename"));
                goto error;
            }

692
            def = virStorageVolDefParseFile(pool->def, absFile);
693 694 695 696
            VIR_FREE(absFile);
            if (!def)
                goto error;
        } else {
697
            if ((def = virStorageVolDefParseNode(pool->def, xml,
698 699 700 701 702 703 704
                                                 vols[i])) == NULL) {
                goto error;
            }
        }

        if (VIR_REALLOC_N(pool->volumes.objs,
                          pool->volumes.count+1) < 0) {
705
            virReportOOMError();
706 707 708
            goto error;
        }

709 710 711 712 713 714 715
        if (def->target.path == NULL) {
            if (virAsprintf(&def->target.path, "%s/%s",
                            pool->def->target.path,
                            def->name) == -1) {
                virReportOOMError();
                goto error;
            }
716 717 718
        }

        if (def->key == NULL) {
719 720 721 722 723
            def->key = strdup(def->target.path);
            if (def->key == NULL) {
                virReportOOMError();
                goto error;
            }
724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740
        }

        pool->def->allocation += def->allocation;
        pool->def->available = (pool->def->capacity -
                                pool->def->allocation);

        pool->volumes.objs[pool->volumes.count++] = def;
        def = NULL;
    }

    func_ret = 0;
error:
    virStorageVolDefFree(def);
    VIR_FREE(vols);
    return func_ret;
}

741
static int testOpenFromFile(virConnectPtr conn,
742
                            const char *file) {
743
    int fd = -1, i, ret;
744 745
    long l;
    char *str;
746
    xmlDocPtr xml = NULL;
747
    xmlNodePtr root = NULL;
748 749
    xmlNodePtr *domains = NULL, *networks = NULL, *ifaces = NULL,
               *pools = NULL, *devs = NULL;
750 751
    xmlXPathContextPtr ctxt = NULL;
    virNodeInfoPtr nodeInfo;
752
    virNetworkObjPtr net;
L
Laine Stump 已提交
753
    virInterfaceObjPtr iface;
754
    virDomainObjPtr dom;
755 756
    testConnPtr privconn;
    if (VIR_ALLOC(privconn) < 0) {
757
        virReportOOMError();
758 759
        return VIR_DRV_OPEN_ERROR;
    }
760
    if (virMutexInit(&privconn->lock) < 0) {
761
        testError(VIR_ERR_INTERNAL_ERROR,
762 763 764 765 766
                  "%s", _("cannot initialize mutex"));
        VIR_FREE(privconn);
        return VIR_DRV_OPEN_ERROR;
    }

767
    testDriverLock(privconn);
768
    conn->privateData = privconn;
769

770 771 772
    if (virDomainObjListInit(&privconn->domains) < 0)
        goto error;

773 774
    if (!(privconn->caps = testBuildCapabilities(conn)))
        goto error;
775 776

    if ((fd = open(file, O_RDONLY)) < 0) {
777
        virReportSystemError(errno,
778 779
                             _("loading host definition file '%s'"),
                             file);
780
        goto error;
781 782
    }

783 784 785
    if (!(xml = xmlReadFd(fd, file, NULL,
                          XML_PARSE_NOENT | XML_PARSE_NONET |
                          XML_PARSE_NOERROR | XML_PARSE_NOWARNING))) {
786
        testError(VIR_ERR_INTERNAL_ERROR,
787
                  _("Invalid XML in file '%s'"), file);
788
        goto error;
789
    }
790 791
    close(fd);
    fd = -1;
792

793 794
    root = xmlDocGetRootElement(xml);
    if ((root == NULL) || (!xmlStrEqual(root->name, BAD_CAST "node"))) {
795
        testError(VIR_ERR_XML_ERROR, "%s",
796
                  _("Root element is not 'node'"));
797
        goto error;
798 799
    }

800 801
    ctxt = xmlXPathNewContext(xml);
    if (ctxt == NULL) {
802
        testError(VIR_ERR_INTERNAL_ERROR, "%s",
C
Cole Robinson 已提交
803
                  _("creating xpath context"));
804
        goto error;
805
    }
806

807
    privconn->nextDomID = 1;
808
    privconn->numCells = 0;
C
Chris Lalancette 已提交
809
    if (virStrcpyStatic(privconn->path, file) == NULL) {
810
        testError(VIR_ERR_INTERNAL_ERROR,
C
Chris Lalancette 已提交
811 812 813
                  _("Path %s too big for destination"), file);
        goto error;
    }
814 815 816
    memmove(&privconn->nodeInfo, &defaultNodeInfo, sizeof(defaultNodeInfo));

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

825
    ret = virXPathLong("string(/node/cpu/sockets[1])", ctxt, &l);
826 827 828
    if (ret == 0) {
        nodeInfo->sockets = l;
    } else if (ret == -2) {
829
        testError(VIR_ERR_XML_ERROR, "%s", _("node cpu sockets"));
830
        goto error;
831
    }
832

833
    ret = virXPathLong("string(/node/cpu/cores[1])", ctxt, &l);
834 835 836
    if (ret == 0) {
        nodeInfo->cores = l;
    } else if (ret == -2) {
837
        testError(VIR_ERR_XML_ERROR, "%s", _("node cpu cores"));
838
        goto error;
839 840
    }

841
    ret = virXPathLong("string(/node/cpu/threads[1])", ctxt, &l);
842 843 844
    if (ret == 0) {
        nodeInfo->threads = l;
    } else if (ret == -2) {
845
        testError(VIR_ERR_XML_ERROR, "%s", _("node cpu threads"));
846
        goto error;
847
    }
848

849
    nodeInfo->cpus = nodeInfo->cores * nodeInfo->threads * nodeInfo->sockets * nodeInfo->nodes;
850
    ret = virXPathLong("string(/node/cpu/active[1])", ctxt, &l);
851 852
    if (ret == 0) {
        if (l < nodeInfo->cpus) {
853 854
            nodeInfo->cpus = l;
        }
855
    } else if (ret == -2) {
856
        testError(VIR_ERR_XML_ERROR, "%s", _("node active cpu"));
857
        goto error;
858
    }
859
    ret = virXPathLong("string(/node/cpu/mhz[1])", ctxt, &l);
860 861 862
    if (ret == 0) {
        nodeInfo->mhz = l;
    } else if (ret == -2) {
863
        testError(VIR_ERR_XML_ERROR, "%s", _("node cpu mhz"));
864
        goto error;
865 866
    }

867
    str = virXPathString("string(/node/cpu/model[1])", ctxt);
868
    if (str != NULL) {
C
Chris Lalancette 已提交
869
        if (virStrcpyStatic(nodeInfo->model, str) == NULL) {
870
            testError(VIR_ERR_INTERNAL_ERROR,
C
Chris Lalancette 已提交
871 872 873 874
                      _("Model %s too big for destination"), str);
            VIR_FREE(str);
            goto error;
        }
875
        VIR_FREE(str);
876 877
    }

878
    ret = virXPathLong("string(/node/memory[1])", ctxt, &l);
879 880 881
    if (ret == 0) {
        nodeInfo->memory = l;
    } else if (ret == -2) {
882
        testError(VIR_ERR_XML_ERROR, "%s", _("node memory"));
883
        goto error;
884
    }
885

886
    ret = virXPathNodeSet("/node/domain", ctxt, &domains);
887
    if (ret < 0) {
888
        testError(VIR_ERR_XML_ERROR, "%s", _("node domain list"));
889
        goto error;
890
    }
891

892
    for (i = 0 ; i < ret ; i++) {
893 894 895 896 897 898
        virDomainDefPtr def;
        char *relFile = virXMLPropString(domains[i], "file");
        if (relFile != NULL) {
            char *absFile = testBuildFilename(file, relFile);
            VIR_FREE(relFile);
            if (!absFile) {
899
                testError(VIR_ERR_INTERNAL_ERROR, "%s", _("resolving domain filename"));
900 901
                goto error;
            }
902
            def = virDomainDefParseFile(privconn->caps, absFile,
903
                                        VIR_DOMAIN_XML_INACTIVE);
904
            VIR_FREE(absFile);
905 906 907
            if (!def)
                goto error;
        } else {
908 909
            if ((def = virDomainDefParseNode(privconn->caps, xml, domains[i],
                                             VIR_DOMAIN_XML_INACTIVE)) == NULL)
910 911 912
                goto error;
        }

913
        if (testDomainGenerateIfnames(def) < 0 ||
914
            !(dom = virDomainAssignDef(privconn->caps,
915
                                       &privconn->domains, def, false))) {
916
            virDomainDefFree(def);
917 918
            goto error;
        }
919

920 921 922 923 924
        if (testDomainStartState(conn, dom) < 0) {
            virDomainObjUnlock(dom);
            goto error;
        }

925
        dom->persistent = 1;
926
        virDomainObjUnlock(dom);
927
    }
928
    VIR_FREE(domains);
929

930
    ret = virXPathNodeSet("/node/network", ctxt, &networks);
931
    if (ret < 0) {
932
        testError(VIR_ERR_XML_ERROR, "%s", _("node network list"));
933 934 935 936 937 938 939 940
        goto error;
    }
    for (i = 0 ; i < ret ; i++) {
        virNetworkDefPtr def;
        char *relFile = virXMLPropString(networks[i], "file");
        if (relFile != NULL) {
            char *absFile = testBuildFilename(file, relFile);
            VIR_FREE(relFile);
941
            if (!absFile) {
942
                testError(VIR_ERR_INTERNAL_ERROR, "%s", _("resolving network filename"));
943 944
                goto error;
            }
945

946
            def = virNetworkDefParseFile(absFile);
947
            VIR_FREE(absFile);
948 949 950
            if (!def)
                goto error;
        } else {
951
            if ((def = virNetworkDefParseNode(xml, networks[i])) == NULL)
952
                goto error;
953
        }
954
        if (!(net = virNetworkAssignDef(&privconn->networks,
955 956 957
                                        def))) {
            virNetworkDefFree(def);
            goto error;
958
        }
959
        net->persistent = 1;
960
        net->active = 1;
961
        virNetworkObjUnlock(net);
962
    }
963
    VIR_FREE(networks);
964

L
Laine Stump 已提交
965
    /* Parse interface definitions */
966
    ret = virXPathNodeSet("/node/interface", ctxt, &ifaces);
L
Laine Stump 已提交
967
    if (ret < 0) {
968
        testError(VIR_ERR_XML_ERROR, "%s", _("node interface list"));
L
Laine Stump 已提交
969 970 971 972 973 974 975 976 977
        goto error;
    }
    for (i = 0 ; i < ret ; i++) {
        virInterfaceDefPtr def;
        char *relFile = virXMLPropString(ifaces[i], "file");
        if (relFile != NULL) {
            char *absFile = testBuildFilename(file, relFile);
            VIR_FREE(relFile);
            if (!absFile) {
978
                testError(VIR_ERR_INTERNAL_ERROR, "%s", _("resolving interface filename"));
L
Laine Stump 已提交
979 980 981
                goto error;
            }

982
            def = virInterfaceDefParseFile(absFile);
L
Laine Stump 已提交
983 984 985 986
            VIR_FREE(absFile);
            if (!def)
                goto error;
        } else {
987
            if ((def = virInterfaceDefParseNode(xml, ifaces[i])) == NULL)
L
Laine Stump 已提交
988 989
                goto error;
        }
990

991
        if (!(iface = virInterfaceAssignDef(&privconn->ifaces, def))) {
L
Laine Stump 已提交
992 993 994
            virInterfaceDefFree(def);
            goto error;
        }
995 996

        iface->active = 1;
L
Laine Stump 已提交
997 998 999 1000
        virInterfaceObjUnlock(iface);
    }
    VIR_FREE(ifaces);

C
Cole Robinson 已提交
1001
    /* Parse Storage Pool list */
1002
    ret = virXPathNodeSet("/node/pool", ctxt, &pools);
C
Cole Robinson 已提交
1003
    if (ret < 0) {
1004
        testError(VIR_ERR_XML_ERROR, "%s", _("node pool list"));
C
Cole Robinson 已提交
1005 1006 1007 1008 1009 1010 1011 1012 1013 1014
        goto error;
    }
    for (i = 0 ; i < ret ; i++) {
        virStoragePoolDefPtr def;
        virStoragePoolObjPtr pool;
        char *relFile = virXMLPropString(pools[i], "file");
        if (relFile != NULL) {
            char *absFile = testBuildFilename(file, relFile);
            VIR_FREE(relFile);
            if (!absFile) {
1015
                testError(VIR_ERR_INTERNAL_ERROR, "%s",
C
Cole Robinson 已提交
1016 1017 1018 1019
                          _("resolving pool filename"));
                goto error;
            }

1020
            def = virStoragePoolDefParseFile(absFile);
C
Cole Robinson 已提交
1021 1022 1023 1024
            VIR_FREE(absFile);
            if (!def)
                goto error;
        } else {
1025
            if ((def = virStoragePoolDefParseNode(xml,
1026
                                                  pools[i])) == NULL) {
C
Cole Robinson 已提交
1027 1028 1029 1030
                goto error;
            }
        }

1031
        if (!(pool = virStoragePoolObjAssignDef(&privconn->pools,
C
Cole Robinson 已提交
1032 1033 1034 1035 1036
                                                def))) {
            virStoragePoolDefFree(def);
            goto error;
        }

1037
        if (testStoragePoolObjSetDefaults(pool) == -1) {
1038
            virStoragePoolObjUnlock(pool);
C
Cole Robinson 已提交
1039
            goto error;
1040
        }
C
Cole Robinson 已提交
1041
        pool->active = 1;
1042 1043

        /* Find storage volumes */
1044
        if (testOpenVolumesForPool(xml, ctxt, file, pool, i+1) < 0) {
1045 1046 1047 1048
            virStoragePoolObjUnlock(pool);
            goto error;
        }

1049
        virStoragePoolObjUnlock(pool);
C
Cole Robinson 已提交
1050
    }
1051
    VIR_FREE(pools);
C
Cole Robinson 已提交
1052

1053
    ret = virXPathNodeSet("/node/device", ctxt, &devs);
1054
    if (ret < 0) {
1055
        testError(VIR_ERR_XML_ERROR, "%s", _("node device list"));
1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067
        goto error;
    }
    for (i = 0 ; i < ret ; i++) {
        virNodeDeviceDefPtr def;
        virNodeDeviceObjPtr dev;
        char *relFile = virXMLPropString(devs[i], "file");

        if (relFile != NULL) {
            char *absFile = testBuildFilename(file, relFile);
            VIR_FREE(relFile);

            if (!absFile) {
1068
                testError(VIR_ERR_INTERNAL_ERROR, "%s",
1069 1070 1071 1072
                          _("resolving device filename"));
                goto error;
            }

1073
            def = virNodeDeviceDefParseFile(absFile, 0);
1074 1075 1076 1077
            VIR_FREE(absFile);
            if (!def)
                goto error;
        } else {
1078
            if ((def = virNodeDeviceDefParseNode(xml, devs[i], 0)) == NULL)
1079 1080
                goto error;
        }
1081
        if (!(dev = virNodeDeviceAssignDef(&privconn->devs, def))) {
1082 1083 1084 1085 1086 1087 1088 1089
            virNodeDeviceDefFree(def);
            goto error;
        }
        virNodeDeviceObjUnlock(dev);
    }
    VIR_FREE(devs);


J
Jim Meyering 已提交
1090
    xmlXPathFreeContext(ctxt);
1091
    xmlFreeDoc(xml);
1092
    testDriverUnlock(privconn);
1093

1094
    return (0);
1095 1096

 error:
J
Jim Meyering 已提交
1097
    xmlXPathFreeContext(ctxt);
1098
    xmlFreeDoc(xml);
1099 1100
    VIR_FREE(domains);
    VIR_FREE(networks);
L
Laine Stump 已提交
1101
    VIR_FREE(ifaces);
C
Cole Robinson 已提交
1102
    VIR_FREE(pools);
1103 1104
    if (fd != -1)
        close(fd);
1105
    virDomainObjListDeinit(&privconn->domains);
1106
    virNetworkObjListFree(&privconn->networks);
L
Laine Stump 已提交
1107
    virInterfaceObjListFree(&privconn->ifaces);
C
Cole Robinson 已提交
1108
    virStoragePoolObjListFree(&privconn->pools);
1109
    testDriverUnlock(privconn);
1110
    VIR_FREE(privconn);
1111
    conn->privateData = NULL;
1112
    return VIR_DRV_OPEN_ERROR;
1113 1114
}

1115

1116
static virDrvOpenStatus testOpen(virConnectPtr conn,
1117
                    virConnectAuthPtr auth ATTRIBUTE_UNUSED,
1118
                    int flags ATTRIBUTE_UNUSED)
1119
{
1120
    int ret;
1121

1122
    if (!conn->uri)
1123
        return VIR_DRV_OPEN_DECLINED;
1124

1125
    if (!conn->uri->scheme || STRNEQ(conn->uri->scheme, "test"))
1126
        return VIR_DRV_OPEN_DECLINED;
1127

1128
    /* Remote driver should handle these. */
1129
    if (conn->uri->server)
1130 1131
        return VIR_DRV_OPEN_DECLINED;

1132
    /* From this point on, the connection is for us. */
1133 1134 1135
    if (!conn->uri->path
        || conn->uri->path[0] == '\0'
        || (conn->uri->path[0] == '/' && conn->uri->path[1] == '\0')) {
1136 1137
        testError(VIR_ERR_INVALID_ARG,
                  "%s", _("testOpen: supply a path or use test:///default"));
1138 1139
        return VIR_DRV_OPEN_ERROR;
    }
1140

1141
    if (STREQ(conn->uri->path, "/default"))
1142 1143
        ret = testOpenDefault(conn);
    else
1144
        ret = testOpenFromFile(conn,
1145
                               conn->uri->path);
1146

1147 1148
    if (ret == VIR_DRV_OPEN_SUCCESS) {
        testConnPtr privconn = conn->privateData;
1149
        testDriverLock(privconn);
1150 1151 1152
        /* Init callback list */
        if (VIR_ALLOC(privconn->domainEventCallbacks) < 0 ||
            !(privconn->domainEventQueue = virDomainEventQueueNew())) {
1153
            virReportOOMError();
1154
            testDriverUnlock(privconn);
1155 1156 1157 1158 1159 1160 1161 1162
            testClose(conn);
            return VIR_DRV_OPEN_ERROR;
        }

        if ((privconn->domainEventTimer =
             virEventAddTimeout(-1, testDomainEventFlush, privconn, NULL)) < 0)
            DEBUG0("virEventAddTimeout failed: No addTimeoutImpl defined. "
                   "continuing without events.");
1163
        testDriverUnlock(privconn);
1164 1165
    }

1166
    return (ret);
1167 1168
}

1169
static int testClose(virConnectPtr conn)
1170
{
1171
    testConnPtr privconn = conn->privateData;
1172
    testDriverLock(privconn);
1173
    virCapabilitiesFree(privconn->caps);
1174
    virDomainObjListDeinit(&privconn->domains);
D
Daniel P. Berrange 已提交
1175
    virNodeDeviceObjListFree(&privconn->devs);
1176
    virNetworkObjListFree(&privconn->networks);
L
Laine Stump 已提交
1177
    virInterfaceObjListFree(&privconn->ifaces);
C
Cole Robinson 已提交
1178
    virStoragePoolObjListFree(&privconn->pools);
1179 1180 1181 1182 1183 1184 1185

    virDomainEventCallbackListFree(privconn->domainEventCallbacks);
    virDomainEventQueueFree(privconn->domainEventQueue);

    if (privconn->domainEventTimer != -1)
        virEventRemoveTimeout(privconn->domainEventTimer);

1186
    testDriverUnlock(privconn);
1187
    virMutexDestroy(&privconn->lock);
1188

1189
    VIR_FREE (privconn);
1190
    conn->privateData = NULL;
1191
    return 0;
1192 1193
}

1194 1195
static int testGetVersion(virConnectPtr conn ATTRIBUTE_UNUSED,
                          unsigned long *hvVer)
1196
{
1197 1198
    *hvVer = 2;
    return (0);
1199 1200
}

1201 1202 1203 1204 1205 1206 1207 1208 1209 1210
static int testIsSecure(virConnectPtr conn ATTRIBUTE_UNUSED)
{
    return 1;
}

static int testIsEncrypted(virConnectPtr conn ATTRIBUTE_UNUSED)
{
    return 0;
}

1211 1212 1213 1214 1215 1216 1217 1218
static int testGetMaxVCPUs(virConnectPtr conn ATTRIBUTE_UNUSED,
                           const char *type ATTRIBUTE_UNUSED)
{
    return 32;
}

static int testNodeGetInfo(virConnectPtr conn,
                           virNodeInfoPtr info)
1219
{
1220
    testConnPtr privconn = conn->privateData;
1221
    testDriverLock(privconn);
1222
    memcpy(info, &privconn->nodeInfo, sizeof(virNodeInfo));
1223
    testDriverUnlock(privconn);
1224
    return (0);
1225 1226
}

1227
static char *testGetCapabilities (virConnectPtr conn)
1228
{
1229
    testConnPtr privconn = conn->privateData;
1230
    char *xml;
1231
    testDriverLock(privconn);
1232
    if ((xml = virCapabilitiesFormatXML(privconn->caps)) == NULL)
1233
        virReportOOMError();
1234
    testDriverUnlock(privconn);
1235
    return xml;
1236 1237
}

1238
static int testNumOfDomains(virConnectPtr conn)
1239
{
1240
    testConnPtr privconn = conn->privateData;
1241
    int count;
1242

1243
    testDriverLock(privconn);
1244
    count = virDomainObjListNumOfDomains(&privconn->domains, 1);
1245
    testDriverUnlock(privconn);
1246

1247
    return count;
1248 1249
}

1250 1251 1252 1253 1254 1255 1256 1257 1258 1259
static int testDomainIsActive(virDomainPtr dom)
{
    testConnPtr privconn = dom->conn->privateData;
    virDomainObjPtr obj;
    int ret = -1;

    testDriverLock(privconn);
    obj = virDomainFindByUUID(&privconn->domains, dom->uuid);
    testDriverUnlock(privconn);
    if (!obj) {
1260
        testError(VIR_ERR_NO_DOMAIN, NULL);
1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280
        goto cleanup;
    }
    ret = virDomainObjIsActive(obj);

cleanup:
    if (obj)
        virDomainObjUnlock(obj);
    return ret;
}

static int testDomainIsPersistent(virDomainPtr dom)
{
    testConnPtr privconn = dom->conn->privateData;
    virDomainObjPtr obj;
    int ret = -1;

    testDriverLock(privconn);
    obj = virDomainFindByUUID(&privconn->domains, dom->uuid);
    testDriverUnlock(privconn);
    if (!obj) {
1281
        testError(VIR_ERR_NO_DOMAIN, NULL);
1282 1283 1284 1285 1286 1287 1288 1289 1290 1291
        goto cleanup;
    }
    ret = obj->persistent;

cleanup:
    if (obj)
        virDomainObjUnlock(obj);
    return ret;
}

1292
static virDomainPtr
1293
testDomainCreateXML(virConnectPtr conn, const char *xml,
1294
                      unsigned int flags)
1295
{
1296
    testConnPtr privconn = conn->privateData;
1297
    virDomainPtr ret = NULL;
1298
    virDomainDefPtr def;
1299
    virDomainObjPtr dom = NULL;
1300
    virDomainEventPtr event = NULL;
1301

1302 1303
    virCheckFlags(0, NULL);

1304
    testDriverLock(privconn);
1305
    if ((def = virDomainDefParseString(privconn->caps, xml,
1306
                                       VIR_DOMAIN_XML_INACTIVE)) == NULL)
1307
        goto cleanup;
1308

1309 1310 1311
    if (virDomainObjIsDuplicate(&privconn->domains, def, 1) < 0)
        goto cleanup;

1312
    if (testDomainGenerateIfnames(def) < 0)
1313
        goto cleanup;
1314
    if (!(dom = virDomainAssignDef(privconn->caps,
1315
                                   &privconn->domains, def, false)))
1316 1317
        goto cleanup;
    def = NULL;
1318 1319 1320

    if (testDomainStartState(conn, dom) < 0)
        goto cleanup;
1321

1322 1323 1324 1325
    event = virDomainEventNewFromObj(dom,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_BOOTED);

1326
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid);
1327
    if (ret)
1328
        ret->id = dom->def->id;
1329 1330

cleanup:
1331 1332
    if (dom)
        virDomainObjUnlock(dom);
1333 1334
    if (event)
        testDomainEventQueue(privconn, event);
1335
    virDomainDefFree(def);
1336
    testDriverUnlock(privconn);
1337
    return ret;
1338 1339 1340
}


1341 1342
static virDomainPtr testLookupDomainByID(virConnectPtr conn,
                                         int id)
1343
{
1344
    testConnPtr privconn = conn->privateData;
1345 1346
    virDomainPtr ret = NULL;
    virDomainObjPtr dom;
1347

1348 1349 1350 1351 1352
    testDriverLock(privconn);
    dom = virDomainFindByID(&privconn->domains, id);
    testDriverUnlock(privconn);

    if (dom == NULL) {
1353
        testError(VIR_ERR_NO_DOMAIN, NULL);
1354
        goto cleanup;
1355 1356
    }

1357
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid);
1358 1359 1360 1361
    if (ret)
        ret->id = dom->def->id;

cleanup:
1362 1363
    if (dom)
        virDomainObjUnlock(dom);
1364
    return ret;
1365 1366
}

1367 1368
static virDomainPtr testLookupDomainByUUID(virConnectPtr conn,
                                           const unsigned char *uuid)
1369
{
1370
    testConnPtr privconn = conn->privateData;
1371 1372
    virDomainPtr ret = NULL;
    virDomainObjPtr dom ;
1373

1374 1375 1376 1377 1378
    testDriverLock(privconn);
    dom = virDomainFindByUUID(&privconn->domains, uuid);
    testDriverUnlock(privconn);

    if (dom == NULL) {
1379
        testError(VIR_ERR_NO_DOMAIN, NULL);
1380
        goto cleanup;
1381
    }
1382

1383
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid);
1384 1385 1386 1387
    if (ret)
        ret->id = dom->def->id;

cleanup:
1388 1389
    if (dom)
        virDomainObjUnlock(dom);
1390
    return ret;
1391 1392
}

1393 1394
static virDomainPtr testLookupDomainByName(virConnectPtr conn,
                                           const char *name)
1395
{
1396
    testConnPtr privconn = conn->privateData;
1397 1398
    virDomainPtr ret = NULL;
    virDomainObjPtr dom;
1399

1400 1401 1402 1403 1404
    testDriverLock(privconn);
    dom = virDomainFindByName(&privconn->domains, name);
    testDriverUnlock(privconn);

    if (dom == NULL) {
1405
        testError(VIR_ERR_NO_DOMAIN, NULL);
1406
        goto cleanup;
1407
    }
1408

1409
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid);
1410 1411 1412 1413
    if (ret)
        ret->id = dom->def->id;

cleanup:
1414 1415
    if (dom)
        virDomainObjUnlock(dom);
1416
    return ret;
1417 1418
}

1419 1420 1421
static int testListDomains (virConnectPtr conn,
                            int *ids,
                            int maxids)
1422
{
1423
    testConnPtr privconn = conn->privateData;
1424
    int n;
1425

1426
    testDriverLock(privconn);
1427
    n = virDomainObjListGetActiveIDs(&privconn->domains, ids, maxids);
1428
    testDriverUnlock(privconn);
1429

1430
    return n;
1431 1432
}

1433
static int testDestroyDomain (virDomainPtr domain)
1434
{
1435 1436
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1437
    virDomainEventPtr event = NULL;
1438
    int ret = -1;
1439

1440
    testDriverLock(privconn);
1441 1442 1443 1444
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
1445
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
1446
        goto cleanup;
1447
    }
1448

1449
    testDomainShutdownState(domain, privdom);
1450 1451 1452
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_DESTROYED);
1453

1454 1455 1456
    if (!privdom->persistent) {
        virDomainRemoveInactive(&privconn->domains,
                                privdom);
1457
        privdom = NULL;
1458
    }
1459 1460 1461

    ret = 0;
cleanup:
1462 1463
    if (privdom)
        virDomainObjUnlock(privdom);
1464 1465
    if (event)
        testDomainEventQueue(privconn, event);
1466
    testDriverUnlock(privconn);
1467
    return ret;
1468 1469
}

1470
static int testResumeDomain (virDomainPtr domain)
1471
{
1472 1473
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1474
    virDomainEventPtr event = NULL;
1475
    int ret = -1;
1476

1477
    testDriverLock(privconn);
1478 1479
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1480
    testDriverUnlock(privconn);
1481 1482

    if (privdom == NULL) {
1483
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
1484
        goto cleanup;
1485
    }
1486

1487
    if (privdom->state != VIR_DOMAIN_PAUSED) {
1488
        testError(VIR_ERR_INTERNAL_ERROR, _("domain '%s' not paused"),
1489
                  domain->name);
1490
        goto cleanup;
1491
    }
1492

1493
    privdom->state = VIR_DOMAIN_RUNNING;
1494 1495 1496
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_RESUMED,
                                     VIR_DOMAIN_EVENT_RESUMED_UNPAUSED);
1497 1498 1499
    ret = 0;

cleanup:
1500 1501
    if (privdom)
        virDomainObjUnlock(privdom);
1502 1503 1504 1505 1506
    if (event) {
        testDriverLock(privconn);
        testDomainEventQueue(privconn, event);
        testDriverUnlock(privconn);
    }
1507
    return ret;
1508 1509
}

1510
static int testPauseDomain (virDomainPtr domain)
1511
{
1512 1513
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1514
    virDomainEventPtr event = NULL;
1515
    int ret = -1;
1516

1517
    testDriverLock(privconn);
1518 1519
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1520
    testDriverUnlock(privconn);
1521 1522

    if (privdom == NULL) {
1523
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
1524
        goto cleanup;
1525
    }
1526

1527 1528
    if (privdom->state == VIR_DOMAIN_SHUTOFF ||
        privdom->state == VIR_DOMAIN_PAUSED) {
1529
        testError(VIR_ERR_INTERNAL_ERROR, _("domain '%s' not running"),
1530
                  domain->name);
1531
        goto cleanup;
1532
    }
1533

1534
    privdom->state = VIR_DOMAIN_PAUSED;
1535 1536 1537
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_SUSPENDED,
                                     VIR_DOMAIN_EVENT_SUSPENDED_PAUSED);
1538 1539 1540
    ret = 0;

cleanup:
1541 1542
    if (privdom)
        virDomainObjUnlock(privdom);
1543 1544 1545 1546 1547 1548

    if (event) {
        testDriverLock(privconn);
        testDomainEventQueue(privconn, event);
        testDriverUnlock(privconn);
    }
1549
    return ret;
1550 1551
}

1552
static int testShutdownDomain (virDomainPtr domain)
1553
{
1554 1555
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1556
    virDomainEventPtr event = NULL;
1557
    int ret = -1;
1558

1559
    testDriverLock(privconn);
1560 1561 1562 1563
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
1564
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
1565
        goto cleanup;
1566
    }
1567

1568
    if (privdom->state == VIR_DOMAIN_SHUTOFF) {
1569
        testError(VIR_ERR_INTERNAL_ERROR,
1570
                  _("domain '%s' not running"), domain->name);
1571
        goto cleanup;
1572
    }
1573

1574
    testDomainShutdownState(domain, privdom);
1575 1576 1577
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_SHUTDOWN);
1578

1579 1580 1581 1582 1583
    if (!privdom->persistent) {
        virDomainRemoveInactive(&privconn->domains,
                                privdom);
        privdom = NULL;
    }
1584

1585
    ret = 0;
1586
cleanup:
1587 1588
    if (privdom)
        virDomainObjUnlock(privdom);
1589 1590
    if (event)
        testDomainEventQueue(privconn, event);
1591
    testDriverUnlock(privconn);
1592
    return ret;
1593 1594 1595
}

/* Similar behaviour as shutdown */
1596 1597
static int testRebootDomain (virDomainPtr domain,
                             unsigned int action ATTRIBUTE_UNUSED)
1598
{
1599 1600
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1601
    virDomainEventPtr event = NULL;
1602
    int ret = -1;
1603

1604
    testDriverLock(privconn);
1605 1606 1607 1608
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
1609
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
1610
        goto cleanup;
1611
    }
1612

1613 1614 1615 1616
    privdom->state = VIR_DOMAIN_SHUTDOWN;
    switch (privdom->def->onReboot) {
    case VIR_DOMAIN_LIFECYCLE_DESTROY:
        privdom->state = VIR_DOMAIN_SHUTOFF;
1617 1618
        break;

1619 1620
    case VIR_DOMAIN_LIFECYCLE_RESTART:
        privdom->state = VIR_DOMAIN_RUNNING;
1621 1622
        break;

1623 1624
    case VIR_DOMAIN_LIFECYCLE_PRESERVE:
        privdom->state = VIR_DOMAIN_SHUTOFF;
1625 1626
        break;

1627 1628
    case VIR_DOMAIN_LIFECYCLE_RESTART_RENAME:
        privdom->state = VIR_DOMAIN_RUNNING;
1629
        break;
1630

1631
    default:
1632
        privdom->state = VIR_DOMAIN_SHUTOFF;
1633 1634
        break;
    }
1635

1636
    if (privdom->state == VIR_DOMAIN_SHUTOFF) {
1637
        testDomainShutdownState(domain, privdom);
1638 1639 1640
        event = virDomainEventNewFromObj(privdom,
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_SHUTDOWN);
1641

1642 1643 1644 1645 1646
        if (!privdom->persistent) {
            virDomainRemoveInactive(&privconn->domains,
                                    privdom);
            privdom = NULL;
        }
1647 1648
    }

1649 1650
    ret = 0;
cleanup:
1651 1652
    if (privdom)
        virDomainObjUnlock(privdom);
1653 1654
    if (event)
        testDomainEventQueue(privconn, event);
1655
    testDriverUnlock(privconn);
1656
    return ret;
1657 1658
}

1659 1660
static int testGetDomainInfo (virDomainPtr domain,
                              virDomainInfoPtr info)
1661
{
1662
    testConnPtr privconn = domain->conn->privateData;
1663
    struct timeval tv;
1664
    virDomainObjPtr privdom;
1665
    int ret = -1;
1666

1667
    testDriverLock(privconn);
1668 1669
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1670
    testDriverUnlock(privconn);
1671 1672

    if (privdom == NULL) {
1673
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
1674
        goto cleanup;
1675
    }
1676 1677

    if (gettimeofday(&tv, NULL) < 0) {
1678
        testError(VIR_ERR_INTERNAL_ERROR,
J
Jim Meyering 已提交
1679
                  "%s", _("getting time of day"));
1680
        goto cleanup;
1681 1682
    }

1683 1684 1685 1686 1687
    info->state = privdom->state;
    info->memory = privdom->def->memory;
    info->maxMem = privdom->def->maxmem;
    info->nrVirtCpu = privdom->def->vcpus;
    info->cpuTime = ((tv.tv_sec * 1000ll * 1000ll  * 1000ll) + (tv.tv_usec * 1000ll));
1688 1689 1690
    ret = 0;

cleanup:
1691 1692
    if (privdom)
        virDomainObjUnlock(privdom);
1693
    return ret;
1694 1695
}

1696 1697 1698 1699 1700
#define TEST_SAVE_MAGIC "TestGuestMagic"

static int testDomainSave(virDomainPtr domain,
                          const char *path)
{
1701
    testConnPtr privconn = domain->conn->privateData;
1702 1703 1704
    char *xml = NULL;
    int fd = -1;
    int len;
1705
    virDomainObjPtr privdom;
1706
    virDomainEventPtr event = NULL;
1707
    int ret = -1;
1708

1709
    testDriverLock(privconn);
1710 1711 1712 1713
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
1714
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
1715
        goto cleanup;
1716
    }
1717

1718
    xml = virDomainDefFormat(privdom->def,
C
Cole Robinson 已提交
1719 1720
                             VIR_DOMAIN_XML_SECURE);

1721
    if (xml == NULL) {
1722
        virReportSystemError(errno,
1723 1724
                             _("saving domain '%s' failed to allocate space for metadata"),
                             domain->name);
1725
        goto cleanup;
1726
    }
1727 1728

    if ((fd = open(path, O_CREAT|O_TRUNC|O_WRONLY, S_IRUSR|S_IWUSR)) < 0) {
1729
        virReportSystemError(errno,
1730 1731
                             _("saving domain '%s' to '%s': open failed"),
                             domain->name, path);
1732
        goto cleanup;
1733
    }
1734
    len = strlen(xml);
1735
    if (safewrite(fd, TEST_SAVE_MAGIC, sizeof(TEST_SAVE_MAGIC)) < 0) {
1736
        virReportSystemError(errno,
1737 1738
                             _("saving domain '%s' to '%s': write failed"),
                             domain->name, path);
1739
        goto cleanup;
1740
    }
1741
    if (safewrite(fd, (char*)&len, sizeof(len)) < 0) {
1742
        virReportSystemError(errno,
1743 1744
                             _("saving domain '%s' to '%s': write failed"),
                             domain->name, path);
1745
        goto cleanup;
1746
    }
1747
    if (safewrite(fd, xml, len) < 0) {
1748
        virReportSystemError(errno,
1749 1750
                             _("saving domain '%s' to '%s': write failed"),
                             domain->name, path);
1751
        goto cleanup;
1752
    }
1753

1754
    if (close(fd) < 0) {
1755
        virReportSystemError(errno,
1756 1757
                             _("saving domain '%s' to '%s': write failed"),
                             domain->name, path);
1758
        goto cleanup;
1759
    }
1760 1761
    fd = -1;

1762
    testDomainShutdownState(domain, privdom);
1763 1764 1765
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_SAVED);
1766

1767 1768 1769
    if (!privdom->persistent) {
        virDomainRemoveInactive(&privconn->domains,
                                privdom);
1770
        privdom = NULL;
1771
    }
1772

1773
    ret = 0;
1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784
cleanup:
    VIR_FREE(xml);

    /* Don't report failure in close or unlink, because
     * in either case we're already in a failure scenario
     * and have reported a earlier error */
    if (ret != 0) {
        if (fd != -1)
            close(fd);
        unlink(path);
    }
1785 1786
    if (privdom)
        virDomainObjUnlock(privdom);
1787 1788
    if (event)
        testDomainEventQueue(privconn, event);
1789
    testDriverUnlock(privconn);
1790
    return ret;
1791 1792
}

1793 1794
static int testDomainRestore(virConnectPtr conn,
                             const char *path)
1795
{
1796
    testConnPtr privconn = conn->privateData;
1797
    char *xml = NULL;
1798
    char magic[15];
1799 1800 1801
    int fd = -1;
    int len;
    virDomainDefPtr def = NULL;
1802
    virDomainObjPtr dom = NULL;
1803
    virDomainEventPtr event = NULL;
1804
    int ret = -1;
1805 1806

    if ((fd = open(path, O_RDONLY)) < 0) {
1807
        virReportSystemError(errno,
1808 1809
                             _("cannot read domain image '%s'"),
                             path);
1810
        goto cleanup;
1811
    }
1812
    if (saferead(fd, magic, sizeof(magic)) != sizeof(magic)) {
1813
        virReportSystemError(errno,
1814 1815
                             _("incomplete save header in '%s'"),
                             path);
1816
        goto cleanup;
1817
    }
1818
    if (memcmp(magic, TEST_SAVE_MAGIC, sizeof(magic))) {
1819
        testError(VIR_ERR_INTERNAL_ERROR,
J
Jim Meyering 已提交
1820
                  "%s", _("mismatched header magic"));
1821
        goto cleanup;
1822
    }
1823
    if (saferead(fd, (char*)&len, sizeof(len)) != sizeof(len)) {
1824
        virReportSystemError(errno,
1825 1826
                             _("failed to read metadata length in '%s'"),
                             path);
1827
        goto cleanup;
1828 1829
    }
    if (len < 1 || len > 8192) {
1830
        testError(VIR_ERR_INTERNAL_ERROR,
J
Jim Meyering 已提交
1831
                  "%s", _("length of metadata out of range"));
1832
        goto cleanup;
1833
    }
1834
    if (VIR_ALLOC_N(xml, len+1) < 0) {
1835
        virReportOOMError();
1836
        goto cleanup;
1837
    }
1838
    if (saferead(fd, xml, len) != len) {
1839
        virReportSystemError(errno,
1840
                             _("incomplete metdata in '%s'"), path);
1841
        goto cleanup;
1842 1843
    }
    xml[len] = '\0';
1844

1845
    testDriverLock(privconn);
1846
    def = virDomainDefParseString(privconn->caps, xml,
1847
                                  VIR_DOMAIN_XML_INACTIVE);
1848
    if (!def)
1849
        goto cleanup;
1850

1851 1852 1853
    if (virDomainObjIsDuplicate(&privconn->domains, def, 1) < 0)
        goto cleanup;

1854
    if (testDomainGenerateIfnames(def) < 0)
1855
        goto cleanup;
1856
    if (!(dom = virDomainAssignDef(privconn->caps,
1857
                                   &privconn->domains, def, true)))
1858 1859
        goto cleanup;
    def = NULL;
1860

1861 1862 1863
    if (testDomainStartState(conn, dom) < 0)
        goto cleanup;

1864 1865 1866
    event = virDomainEventNewFromObj(dom,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_RESTORED);
1867
    ret = 0;
1868 1869 1870 1871 1872 1873

cleanup:
    virDomainDefFree(def);
    VIR_FREE(xml);
    if (fd != -1)
        close(fd);
1874 1875
    if (dom)
        virDomainObjUnlock(dom);
1876 1877
    if (event)
        testDomainEventQueue(privconn, event);
1878
    testDriverUnlock(privconn);
1879
    return ret;
1880 1881
}

1882 1883 1884
static int testDomainCoreDump(virDomainPtr domain,
                              const char *to,
                              int flags ATTRIBUTE_UNUSED)
1885
{
1886
    testConnPtr privconn = domain->conn->privateData;
1887
    int fd = -1;
1888
    virDomainObjPtr privdom;
1889
    virDomainEventPtr event = NULL;
1890
    int ret = -1;
1891

1892
    testDriverLock(privconn);
1893 1894 1895 1896
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
1897
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
1898
        goto cleanup;
1899
    }
1900 1901

    if ((fd = open(to, O_CREAT|O_TRUNC|O_WRONLY, S_IRUSR|S_IWUSR)) < 0) {
1902
        virReportSystemError(errno,
1903 1904
                             _("domain '%s' coredump: failed to open %s"),
                             domain->name, to);
1905
        goto cleanup;
1906
    }
1907
    if (safewrite(fd, TEST_SAVE_MAGIC, sizeof(TEST_SAVE_MAGIC)) < 0) {
1908
        virReportSystemError(errno,
1909 1910
                             _("domain '%s' coredump: failed to write header to %s"),
                             domain->name, to);
1911
        goto cleanup;
1912
    }
1913
    if (close(fd) < 0) {
1914
        virReportSystemError(errno,
1915 1916
                             _("domain '%s' coredump: write failed: %s"),
                             domain->name, to);
1917
        goto cleanup;
1918
    }
1919

1920 1921 1922 1923 1924 1925 1926 1927 1928 1929
    if (flags & VIR_DUMP_CRASH) {
        testDomainShutdownState(domain, privdom);
        event = virDomainEventNewFromObj(privdom,
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_CRASHED);
        if (!privdom->persistent) {
            virDomainRemoveInactive(&privconn->domains,
                                    privdom);
            privdom = NULL;
        }
1930
    }
1931

1932
    ret = 0;
1933 1934 1935
cleanup:
    if (fd != -1)
        close(fd);
1936 1937
    if (privdom)
        virDomainObjUnlock(privdom);
1938 1939
    if (event)
        testDomainEventQueue(privconn, event);
1940
    testDriverUnlock(privconn);
1941
    return ret;
1942 1943
}

1944
static char *testGetOSType(virDomainPtr dom ATTRIBUTE_UNUSED) {
1945 1946
    char *ret = strdup("linux");
    if (!ret)
1947
        virReportOOMError();
1948
    return ret;
1949 1950 1951
}

static unsigned long testGetMaxMemory(virDomainPtr domain) {
1952 1953
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1954
    unsigned long ret = 0;
1955

1956
    testDriverLock(privconn);
1957 1958
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1959
    testDriverUnlock(privconn);
1960 1961

    if (privdom == NULL) {
1962
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
1963
        goto cleanup;
1964
    }
1965

1966 1967 1968
    ret = privdom->def->maxmem;

cleanup:
1969 1970
    if (privdom)
        virDomainObjUnlock(privdom);
1971
    return ret;
1972 1973 1974 1975 1976
}

static int testSetMaxMemory(virDomainPtr domain,
                            unsigned long memory)
{
1977 1978
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1979
    int ret = -1;
1980

1981
    testDriverLock(privconn);
1982 1983
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1984
    testDriverUnlock(privconn);
1985 1986

    if (privdom == NULL) {
1987
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
1988
        goto cleanup;
1989
    }
1990 1991

    /* XXX validate not over host memory wrt to other domains */
1992
    privdom->def->maxmem = memory;
1993 1994 1995
    ret = 0;

cleanup:
1996 1997
    if (privdom)
        virDomainObjUnlock(privdom);
1998
    return ret;
1999 2000
}

2001 2002 2003
static int testSetMemory(virDomainPtr domain,
                         unsigned long memory)
{
2004 2005
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
2006
    int ret = -1;
2007

2008
    testDriverLock(privconn);
2009 2010
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
2011
    testDriverUnlock(privconn);
2012 2013

    if (privdom == NULL) {
2014
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2015
        goto cleanup;
2016
    }
2017

2018
    if (memory > privdom->def->maxmem) {
2019
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2020
        goto cleanup;
2021
    }
2022

2023
    privdom->def->memory = memory;
2024 2025 2026
    ret = 0;

cleanup:
2027 2028
    if (privdom)
        virDomainObjUnlock(privdom);
2029
    return ret;
2030 2031
}

C
Cole Robinson 已提交
2032 2033 2034 2035 2036
static int testDomainGetMaxVcpus(virDomainPtr domain)
{
    return testGetMaxVCPUs(domain->conn, "test");
}

2037 2038
static int testSetVcpus(virDomainPtr domain,
                        unsigned int nrCpus) {
2039
    testConnPtr privconn = domain->conn->privateData;
2040
    virDomainObjPtr privdom = NULL;
C
Cole Robinson 已提交
2041 2042 2043 2044 2045 2046
    int ret = -1, maxvcpus;

    /* Do this first before locking */
    maxvcpus = testDomainGetMaxVcpus(domain);
    if (maxvcpus < 0)
        goto cleanup;
2047

2048 2049 2050 2051 2052 2053
    testDriverLock(privconn);
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
    testDriverUnlock(privconn);

    if (privdom == NULL) {
2054
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2055
        goto cleanup;
2056
    }
2057

C
Cole Robinson 已提交
2058
    if (!virDomainObjIsActive(privdom)) {
2059
        testError(VIR_ERR_OPERATION_INVALID,
C
Cole Robinson 已提交
2060 2061 2062 2063
                  "%s", _("cannot hotplug vcpus for an inactive domain"));
        goto cleanup;
    }

2064
    /* We allow more cpus in guest than host */
C
Cole Robinson 已提交
2065
    if (nrCpus > maxvcpus) {
2066
        testError(VIR_ERR_INVALID_ARG,
C
Cole Robinson 已提交
2067 2068
                  "requested cpu amount exceeds maximum (%d > %d)",
                  nrCpus, maxvcpus);
2069
        goto cleanup;
2070
    }
2071

2072 2073 2074 2075
    /* Update VCPU state for the running domain */
    if (testDomainUpdateVCPUs(domain->conn, privdom, nrCpus, 0) < 0)
        goto cleanup;

2076
    privdom->def->vcpus = nrCpus;
2077 2078 2079
    ret = 0;

cleanup:
2080 2081
    if (privdom)
        virDomainObjUnlock(privdom);
2082
    return ret;
2083 2084
}

C
Cole Robinson 已提交
2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103
static int testDomainGetVcpus(virDomainPtr domain,
                              virVcpuInfoPtr info,
                              int maxinfo,
                              unsigned char *cpumaps,
                              int maplen)
{
    testConnPtr privconn = domain->conn->privateData;
    testDomainObjPrivatePtr privdomdata;
    virDomainObjPtr privdom;
    int i, v, maxcpu, hostcpus;
    int ret = -1;
    struct timeval tv;
    unsigned long long statbase;

    testDriverLock(privconn);
    privdom = virDomainFindByName(&privconn->domains, domain->name);
    testDriverUnlock(privconn);

    if (privdom == NULL) {
2104
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
C
Cole Robinson 已提交
2105 2106 2107 2108
        goto cleanup;
    }

    if (!virDomainObjIsActive(privdom)) {
2109
        testError(VIR_ERR_OPERATION_INVALID,
C
Cole Robinson 已提交
2110 2111 2112 2113 2114 2115 2116
                  "%s",_("cannot list vcpus for an inactive domain"));
        goto cleanup;
    }

    privdomdata = privdom->privateData;

    if (gettimeofday(&tv, NULL) < 0) {
2117
        virReportSystemError(errno,
C
Cole Robinson 已提交
2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172
                             "%s", _("getting time of day"));
        goto cleanup;
    }

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


    hostcpus = VIR_NODEINFO_MAXCPUS(privconn->nodeInfo);
    maxcpu = maplen * 8;
    if (maxcpu > hostcpus)
        maxcpu = hostcpus;

    /* Clamp to actual number of vcpus */
    if (maxinfo > privdom->def->vcpus)
        maxinfo = privdom->def->vcpus;

    /* Populate virVcpuInfo structures */
    if (info != NULL) {
        memset(info, 0, sizeof(*info) * maxinfo);

        for (i = 0 ; i < maxinfo ; i++) {
            virVcpuInfo privinfo = privdomdata->vcpu_infos[i];

            info[i].number = privinfo.number;
            info[i].state = privinfo.state;
            info[i].cpu = privinfo.cpu;

            /* Fake an increasing cpu time value */
            info[i].cpuTime = statbase / 10;
        }
    }

    /* Populate cpumaps */
    if (cpumaps != NULL) {
        int privmaplen = VIR_CPU_MAPLEN(hostcpus);
        memset(cpumaps, 0, maplen * maxinfo);

        for (v = 0 ; v < maxinfo ; v++) {
            unsigned char *cpumap = VIR_GET_CPUMAP(cpumaps, maplen, v);

            for (i = 0 ; i < maxcpu ; i++) {
                if (VIR_CPU_USABLE(privdomdata->cpumaps, privmaplen, v, i)) {
                    VIR_USE_CPU(cpumap, i);
                }
            }
        }
    }

    ret = maxinfo;
cleanup:
    if (privdom)
        virDomainObjUnlock(privdom);
    return ret;
}

C
Cole Robinson 已提交
2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189
static int testDomainPinVcpu(virDomainPtr domain,
                             unsigned int vcpu,
                             unsigned char *cpumap,
                             int maplen)
{
    testConnPtr privconn = domain->conn->privateData;
    testDomainObjPrivatePtr privdomdata;
    virDomainObjPtr privdom;
    unsigned char *privcpumap;
    int i, maxcpu, hostcpus, privmaplen;
    int ret = -1;

    testDriverLock(privconn);
    privdom = virDomainFindByName(&privconn->domains, domain->name);
    testDriverUnlock(privconn);

    if (privdom == NULL) {
2190
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
C
Cole Robinson 已提交
2191 2192 2193 2194
        goto cleanup;
    }

    if (!virDomainObjIsActive(privdom)) {
2195
        testError(VIR_ERR_OPERATION_INVALID,
C
Cole Robinson 已提交
2196 2197 2198 2199 2200
                  "%s",_("cannot pin vcpus on an inactive domain"));
        goto cleanup;
    }

    if (vcpu > privdom->def->vcpus) {
2201
        testError(VIR_ERR_INVALID_ARG, "%s",
C
Cole Robinson 已提交
2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229
                  _("requested vcpu is higher than allocated vcpus"));
        goto cleanup;
    }

    privdomdata = privdom->privateData;
    hostcpus = VIR_NODEINFO_MAXCPUS(privconn->nodeInfo);
    privmaplen = VIR_CPU_MAPLEN(hostcpus);

    maxcpu = maplen * 8;
    if (maxcpu > hostcpus)
        maxcpu = hostcpus;

    privcpumap = VIR_GET_CPUMAP(privdomdata->cpumaps, privmaplen, vcpu);
    memset(privcpumap, 0, privmaplen);

    for (i = 0 ; i < maxcpu ; i++) {
        if (VIR_CPU_USABLE(cpumap, maplen, 0, i)) {
            VIR_USE_CPU(privcpumap, i);
        }
    }

    ret = 0;
cleanup:
    if (privdom)
        virDomainObjUnlock(privdom);
    return ret;
}

2230
static char *testDomainDumpXML(virDomainPtr domain, int flags)
2231
{
2232
    testConnPtr privconn = domain->conn->privateData;
2233
    virDomainDefPtr def;
2234
    virDomainObjPtr privdom;
2235 2236
    char *ret = NULL;

2237 2238 2239 2240 2241 2242
    testDriverLock(privconn);
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
    testDriverUnlock(privconn);

    if (privdom == NULL) {
2243
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2244
        goto cleanup;
2245
    }
2246

2247 2248
    def = (flags & VIR_DOMAIN_XML_INACTIVE) &&
        privdom->newDef ? privdom->newDef : privdom->def;
2249

2250
    ret = virDomainDefFormat(def,
2251 2252 2253
                             flags);

cleanup:
2254 2255
    if (privdom)
        virDomainObjUnlock(privdom);
2256
    return ret;
2257
}
2258

2259
static int testNumOfDefinedDomains(virConnectPtr conn) {
2260
    testConnPtr privconn = conn->privateData;
2261
    int count;
2262

2263
    testDriverLock(privconn);
2264
    count = virDomainObjListNumOfDomains(&privconn->domains, 0);
2265
    testDriverUnlock(privconn);
2266

2267
    return count;
2268 2269
}

2270 2271 2272
static int testListDefinedDomains(virConnectPtr conn,
                                  char **const names,
                                  int maxnames) {
2273

2274
    testConnPtr privconn = conn->privateData;
2275
    int n;
2276

2277
    testDriverLock(privconn);
2278
    memset(names, 0, sizeof(*names)*maxnames);
2279
    n = virDomainObjListGetInactiveNames(&privconn->domains, names, maxnames);
2280
    testDriverUnlock(privconn);
2281

2282
    return n;
2283 2284
}

2285
static virDomainPtr testDomainDefineXML(virConnectPtr conn,
2286
                                        const char *xml) {
2287
    testConnPtr privconn = conn->privateData;
2288
    virDomainPtr ret = NULL;
2289
    virDomainDefPtr def;
2290
    virDomainObjPtr dom = NULL;
2291
    virDomainEventPtr event = NULL;
2292
    int dupVM;
2293

2294
    testDriverLock(privconn);
2295
    if ((def = virDomainDefParseString(privconn->caps, xml,
2296
                                       VIR_DOMAIN_XML_INACTIVE)) == NULL)
2297
        goto cleanup;
2298

2299 2300 2301
    if ((dupVM = virDomainObjIsDuplicate(&privconn->domains, def, 0)) < 0)
        goto cleanup;

2302
    if (testDomainGenerateIfnames(def) < 0)
2303
        goto cleanup;
2304
    if (!(dom = virDomainAssignDef(privconn->caps,
2305
                                   &privconn->domains, def, false)))
2306
        goto cleanup;
2307
    def = NULL;
2308
    dom->persistent = 1;
2309

2310 2311
    event = virDomainEventNewFromObj(dom,
                                     VIR_DOMAIN_EVENT_DEFINED,
2312 2313 2314
                                     !dupVM ?
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED :
                                     VIR_DOMAIN_EVENT_DEFINED_UPDATED);
2315

2316
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid);
2317
    if (ret)
2318
        ret->id = dom->def->id;
2319 2320 2321

cleanup:
    virDomainDefFree(def);
2322 2323
    if (dom)
        virDomainObjUnlock(dom);
2324 2325
    if (event)
        testDomainEventQueue(privconn, event);
2326
    testDriverUnlock(privconn);
2327
    return ret;
2328 2329
}

2330 2331 2332
static int testNodeGetCellsFreeMemory(virConnectPtr conn,
                                      unsigned long long *freemems,
                                      int startCell, int maxCells) {
2333
    testConnPtr privconn = conn->privateData;
2334
    int i, j;
2335
    int ret = -1;
2336

2337
    testDriverLock(privconn);
2338
    if (startCell > privconn->numCells) {
2339
        testError(VIR_ERR_INVALID_ARG,
J
Jim Meyering 已提交
2340
                  "%s", _("Range exceeds available cells"));
2341
        goto cleanup;
2342 2343 2344 2345 2346 2347 2348
    }

    for (i = startCell, j = 0;
         (i < privconn->numCells && j < maxCells) ;
         ++i, ++j) {
        freemems[j] = privconn->cells[i].mem;
    }
2349
    ret = j;
2350

2351
cleanup:
2352
    testDriverUnlock(privconn);
2353
    return ret;
2354 2355 2356
}


2357
static int testDomainCreateWithFlags(virDomainPtr domain, unsigned int flags) {
2358 2359
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
2360
    virDomainEventPtr event = NULL;
2361
    int ret = -1;
2362

2363 2364
    virCheckFlags(0, -1);

2365
    testDriverLock(privconn);
2366 2367 2368 2369
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
2370
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2371
        goto cleanup;
2372
    }
2373

2374
    if (privdom->state != VIR_DOMAIN_SHUTOFF) {
2375
        testError(VIR_ERR_INTERNAL_ERROR,
2376
                  _("Domain '%s' is already running"), domain->name);
2377
        goto cleanup;
2378 2379
    }

2380 2381 2382 2383
    if (testDomainStartState(domain->conn, privdom) < 0)
        goto cleanup;
    domain->id = privdom->def->id;

2384 2385 2386
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_BOOTED);
2387
    ret = 0;
2388

2389
cleanup:
2390 2391
    if (privdom)
        virDomainObjUnlock(privdom);
2392 2393
    if (event)
        testDomainEventQueue(privconn, event);
2394
    testDriverUnlock(privconn);
2395
    return ret;
2396 2397
}

2398 2399 2400 2401
static int testDomainCreate(virDomainPtr domain) {
    return testDomainCreateWithFlags(domain, 0);
}

2402
static int testDomainUndefine(virDomainPtr domain) {
2403 2404
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
2405
    virDomainEventPtr event = NULL;
2406
    int ret = -1;
2407

2408
    testDriverLock(privconn);
2409 2410 2411 2412
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
2413
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2414
        goto cleanup;
2415
    }
2416

2417
    if (privdom->state != VIR_DOMAIN_SHUTOFF) {
2418
        testError(VIR_ERR_INTERNAL_ERROR,
2419
                  _("Domain '%s' is still running"), domain->name);
2420
        goto cleanup;
2421 2422
    }

2423
    privdom->state = VIR_DOMAIN_SHUTOFF;
2424 2425 2426
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_UNDEFINED,
                                     VIR_DOMAIN_EVENT_UNDEFINED_REMOVED);
2427 2428
    virDomainRemoveInactive(&privconn->domains,
                            privdom);
2429
    privdom = NULL;
2430
    ret = 0;
2431

2432
cleanup:
2433 2434
    if (privdom)
        virDomainObjUnlock(privdom);
2435 2436
    if (event)
        testDomainEventQueue(privconn, event);
2437
    testDriverUnlock(privconn);
2438
    return ret;
2439 2440
}

2441 2442 2443
static int testDomainGetAutostart(virDomainPtr domain,
                                  int *autostart)
{
2444 2445
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
2446
    int ret = -1;
2447

2448
    testDriverLock(privconn);
2449 2450
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
2451
    testDriverUnlock(privconn);
2452 2453

    if (privdom == NULL) {
2454
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2455
        goto cleanup;
2456 2457
    }

2458
    *autostart = privdom->autostart;
2459 2460 2461
    ret = 0;

cleanup:
2462 2463
    if (privdom)
        virDomainObjUnlock(privdom);
2464
    return ret;
2465 2466 2467 2468 2469 2470
}


static int testDomainSetAutostart(virDomainPtr domain,
                                  int autostart)
{
2471 2472
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
2473
    int ret = -1;
2474

2475
    testDriverLock(privconn);
2476 2477
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
2478
    testDriverUnlock(privconn);
2479 2480

    if (privdom == NULL) {
2481
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2482
        goto cleanup;
2483 2484
    }

2485
    privdom->autostart = autostart ? 1 : 0;
2486 2487 2488
    ret = 0;

cleanup:
2489 2490
    if (privdom)
        virDomainObjUnlock(privdom);
2491
    return ret;
2492
}
2493

2494
static char *testDomainGetSchedulerType(virDomainPtr domain ATTRIBUTE_UNUSED,
2495 2496
                                        int *nparams)
{
2497 2498
    char *type = NULL;

2499 2500
    *nparams = 1;
    type = strdup("fair");
2501
    if (!type)
2502
        virReportOOMError();
2503

2504 2505 2506 2507 2508 2509 2510
    return type;
}

static int testDomainGetSchedulerParams(virDomainPtr domain,
                                        virSchedParameterPtr params,
                                        int *nparams)
{
2511 2512
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
2513
    int ret = -1;
2514

2515
    testDriverLock(privconn);
2516 2517
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
2518
    testDriverUnlock(privconn);
2519 2520

    if (privdom == NULL) {
2521
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2522
        goto cleanup;
2523 2524
    }

2525
    if (*nparams != 1) {
2526
        testError(VIR_ERR_INVALID_ARG, "nparams");
2527
        goto cleanup;
2528
    }
2529 2530
    strcpy(params[0].field, "weight");
    params[0].type = VIR_DOMAIN_SCHED_FIELD_UINT;
2531 2532 2533
    /* XXX */
    /*params[0].value.ui = privdom->weight;*/
    params[0].value.ui = 50;
2534 2535 2536
    ret = 0;

cleanup:
2537 2538
    if (privdom)
        virDomainObjUnlock(privdom);
2539
    return ret;
2540
}
2541 2542


2543 2544 2545 2546
static int testDomainSetSchedulerParams(virDomainPtr domain,
                                        virSchedParameterPtr params,
                                        int nparams)
{
2547 2548
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
2549
    int ret = -1;
2550

2551
    testDriverLock(privconn);
2552 2553
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
2554
    testDriverUnlock(privconn);
2555 2556

    if (privdom == NULL) {
2557
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2558
        goto cleanup;
2559 2560
    }

2561
    if (nparams != 1) {
2562
        testError(VIR_ERR_INVALID_ARG, "nparams");
2563
        goto cleanup;
2564
    }
2565
    if (STRNEQ(params[0].field, "weight")) {
2566
        testError(VIR_ERR_INVALID_ARG, "field");
2567
        goto cleanup;
2568 2569
    }
    if (params[0].type != VIR_DOMAIN_SCHED_FIELD_UINT) {
2570
        testError(VIR_ERR_INVALID_ARG, "type");
2571
        goto cleanup;
2572
    }
2573 2574
    /* XXX */
    /*privdom->weight = params[0].value.ui;*/
2575 2576 2577
    ret = 0;

cleanup:
2578 2579
    if (privdom)
        virDomainObjUnlock(privdom);
2580
    return ret;
2581 2582
}

2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596 2597 2598
static int testDomainBlockStats(virDomainPtr domain,
                                const char *path,
                                struct _virDomainBlockStats *stats)
{
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
    struct timeval tv;
    unsigned long long statbase;
    int i, found = 0, ret = -1;

    testDriverLock(privconn);
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
    testDriverUnlock(privconn);

    if (privdom == NULL) {
2599
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2600 2601 2602 2603 2604 2605 2606 2607 2608 2609 2610
        goto error;
    }

    for (i = 0 ; i < privdom->def->ndisks ; i++) {
        if (STREQ(path, privdom->def->disks[i]->dst)) {
            found = 1;
            break;
        }
    }

    if (!found) {
2611
        testError(VIR_ERR_INVALID_ARG,
2612 2613 2614 2615 2616
                  _("invalid path: %s"), path);
        goto error;
    }

    if (gettimeofday(&tv, NULL) < 0) {
2617
        virReportSystemError(errno,
2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652
                             "%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;
error:
    if (privdom)
        virDomainObjUnlock(privdom);
    return ret;
}

static int testDomainInterfaceStats(virDomainPtr domain,
                                    const char *path,
                                    struct _virDomainInterfaceStats *stats)
{
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
    struct timeval tv;
    unsigned long long statbase;
    int i, found = 0, ret = -1;

    testDriverLock(privconn);
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
    testDriverUnlock(privconn);

    if (privdom == NULL) {
2653
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665
        goto error;
    }

    for (i = 0 ; i < privdom->def->nnets ; i++) {
        if (privdom->def->nets[i]->ifname &&
            STREQ (privdom->def->nets[i]->ifname, path)) {
            found = 1;
            break;
        }
    }

    if (!found) {
2666
        testError(VIR_ERR_INVALID_ARG,
2667 2668 2669 2670 2671
                  _("invalid path, '%s' is not a known interface"), path);
        goto error;
    }

    if (gettimeofday(&tv, NULL) < 0) {
2672
        virReportSystemError(errno,
2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690 2691 2692 2693 2694
                             "%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;
error:
    if (privdom)
        virDomainObjUnlock(privdom);
    return ret;
}

2695
static virDrvOpenStatus testOpenNetwork(virConnectPtr conn,
2696
                                        virConnectAuthPtr auth ATTRIBUTE_UNUSED,
2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713
                                        int flags ATTRIBUTE_UNUSED) {
    if (STRNEQ(conn->driver->name, "Test"))
        return VIR_DRV_OPEN_DECLINED;

    conn->networkPrivateData = conn->privateData;
    return VIR_DRV_OPEN_SUCCESS;
}

static int testCloseNetwork(virConnectPtr conn) {
    conn->networkPrivateData = NULL;
    return 0;
}


static virNetworkPtr testLookupNetworkByUUID(virConnectPtr conn,
                                           const unsigned char *uuid)
{
2714 2715
    testConnPtr privconn = conn->privateData;
    virNetworkObjPtr net;
2716
    virNetworkPtr ret = NULL;
2717

2718 2719 2720 2721 2722
    testDriverLock(privconn);
    net = virNetworkFindByUUID(&privconn->networks, uuid);
    testDriverUnlock(privconn);

    if (net == NULL) {
2723
        testError(VIR_ERR_NO_NETWORK, NULL);
2724
        goto cleanup;
2725 2726
    }

2727 2728 2729
    ret = virGetNetwork(conn, net->def->name, net->def->uuid);

cleanup:
2730 2731
    if (net)
        virNetworkObjUnlock(net);
2732
    return ret;
2733
}
2734

2735
static virNetworkPtr testLookupNetworkByName(virConnectPtr conn,
2736
                                             const char *name)
2737
{
2738
    testConnPtr privconn = conn->privateData;
2739 2740
    virNetworkObjPtr net;
    virNetworkPtr ret = NULL;
2741

2742 2743 2744 2745 2746
    testDriverLock(privconn);
    net = virNetworkFindByName(&privconn->networks, name);
    testDriverUnlock(privconn);

    if (net == NULL) {
2747
        testError(VIR_ERR_NO_NETWORK, NULL);
2748
        goto cleanup;
2749 2750
    }

2751 2752 2753
    ret = virGetNetwork(conn, net->def->name, net->def->uuid);

cleanup:
2754 2755
    if (net)
        virNetworkObjUnlock(net);
2756
    return ret;
2757 2758 2759 2760
}


static int testNumNetworks(virConnectPtr conn) {
2761
    testConnPtr privconn = conn->privateData;
2762
    int numActive = 0, i;
2763

2764 2765 2766
    testDriverLock(privconn);
    for (i = 0 ; i < privconn->networks.count ; i++) {
        virNetworkObjLock(privconn->networks.objs[i]);
D
Daniel P. Berrange 已提交
2767
        if (virNetworkObjIsActive(privconn->networks.objs[i]))
2768
            numActive++;
2769 2770 2771
        virNetworkObjUnlock(privconn->networks.objs[i]);
    }
    testDriverUnlock(privconn);
2772

2773
    return numActive;
2774 2775 2776
}

static int testListNetworks(virConnectPtr conn, char **const names, int nnames) {
2777
    testConnPtr privconn = conn->privateData;
2778
    int n = 0, i;
2779

2780
    testDriverLock(privconn);
2781
    memset(names, 0, sizeof(*names)*nnames);
2782 2783
    for (i = 0 ; i < privconn->networks.count && n < nnames ; i++) {
        virNetworkObjLock(privconn->networks.objs[i]);
D
Daniel P. Berrange 已提交
2784
        if (virNetworkObjIsActive(privconn->networks.objs[i]) &&
2785 2786
            !(names[n++] = strdup(privconn->networks.objs[i]->def->name))) {
            virNetworkObjUnlock(privconn->networks.objs[i]);
2787
            goto no_memory;
2788 2789 2790 2791
        }
        virNetworkObjUnlock(privconn->networks.objs[i]);
    }
    testDriverUnlock(privconn);
2792

2793 2794 2795
    return n;

no_memory:
2796
    virReportOOMError();
2797 2798
    for (n = 0 ; n < nnames ; n++)
        VIR_FREE(names[n]);
2799
    testDriverUnlock(privconn);
2800
    return -1;
2801 2802 2803
}

static int testNumDefinedNetworks(virConnectPtr conn) {
2804
    testConnPtr privconn = conn->privateData;
2805
    int numInactive = 0, i;
2806

2807 2808 2809
    testDriverLock(privconn);
    for (i = 0 ; i < privconn->networks.count ; i++) {
        virNetworkObjLock(privconn->networks.objs[i]);
D
Daniel P. Berrange 已提交
2810
        if (!virNetworkObjIsActive(privconn->networks.objs[i]))
2811
            numInactive++;
2812 2813 2814
        virNetworkObjUnlock(privconn->networks.objs[i]);
    }
    testDriverUnlock(privconn);
2815

2816
    return numInactive;
2817 2818 2819
}

static int testListDefinedNetworks(virConnectPtr conn, char **const names, int nnames) {
2820
    testConnPtr privconn = conn->privateData;
2821
    int n = 0, i;
2822

2823
    testDriverLock(privconn);
2824
    memset(names, 0, sizeof(*names)*nnames);
2825 2826
    for (i = 0 ; i < privconn->networks.count && n < nnames ; i++) {
        virNetworkObjLock(privconn->networks.objs[i]);
D
Daniel P. Berrange 已提交
2827
        if (!virNetworkObjIsActive(privconn->networks.objs[i]) &&
2828 2829
            !(names[n++] = strdup(privconn->networks.objs[i]->def->name))) {
            virNetworkObjUnlock(privconn->networks.objs[i]);
2830
            goto no_memory;
2831 2832 2833 2834
        }
        virNetworkObjUnlock(privconn->networks.objs[i]);
    }
    testDriverUnlock(privconn);
2835

2836 2837 2838
    return n;

no_memory:
2839
    virReportOOMError();
2840 2841
    for (n = 0 ; n < nnames ; n++)
        VIR_FREE(names[n]);
2842
    testDriverUnlock(privconn);
2843
    return -1;
2844 2845
}

2846 2847 2848 2849 2850 2851 2852 2853 2854 2855 2856

static int testNetworkIsActive(virNetworkPtr net)
{
    testConnPtr privconn = net->conn->privateData;
    virNetworkObjPtr obj;
    int ret = -1;

    testDriverLock(privconn);
    obj = virNetworkFindByUUID(&privconn->networks, net->uuid);
    testDriverUnlock(privconn);
    if (!obj) {
2857
        testError(VIR_ERR_NO_NETWORK, NULL);
2858 2859 2860 2861 2862 2863 2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877
        goto cleanup;
    }
    ret = virNetworkObjIsActive(obj);

cleanup:
    if (obj)
        virNetworkObjUnlock(obj);
    return ret;
}

static int testNetworkIsPersistent(virNetworkPtr net)
{
    testConnPtr privconn = net->conn->privateData;
    virNetworkObjPtr obj;
    int ret = -1;

    testDriverLock(privconn);
    obj = virNetworkFindByUUID(&privconn->networks, net->uuid);
    testDriverUnlock(privconn);
    if (!obj) {
2878
        testError(VIR_ERR_NO_NETWORK, NULL);
2879 2880 2881 2882 2883 2884 2885 2886 2887 2888 2889
        goto cleanup;
    }
    ret = obj->persistent;

cleanup:
    if (obj)
        virNetworkObjUnlock(obj);
    return ret;
}


2890
static virNetworkPtr testNetworkCreate(virConnectPtr conn, const char *xml) {
2891
    testConnPtr privconn = conn->privateData;
2892
    virNetworkDefPtr def;
2893
    virNetworkObjPtr net = NULL;
2894
    virNetworkPtr ret = NULL;
2895

2896
    testDriverLock(privconn);
2897
    if ((def = virNetworkDefParseString(xml)) == NULL)
2898
        goto cleanup;
2899

2900
    if ((net = virNetworkAssignDef(&privconn->networks, def)) == NULL)
2901 2902
        goto cleanup;
    def = NULL;
2903
    net->active = 1;
2904

2905
    ret = virGetNetwork(conn, net->def->name, net->def->uuid);
2906

2907 2908
cleanup:
    virNetworkDefFree(def);
2909 2910 2911
    if (net)
        virNetworkObjUnlock(net);
    testDriverUnlock(privconn);
2912
    return ret;
2913 2914 2915
}

static virNetworkPtr testNetworkDefine(virConnectPtr conn, const char *xml) {
2916
    testConnPtr privconn = conn->privateData;
2917
    virNetworkDefPtr def;
2918
    virNetworkObjPtr net = NULL;
2919
    virNetworkPtr ret = NULL;
2920

2921
    testDriverLock(privconn);
2922
    if ((def = virNetworkDefParseString(xml)) == NULL)
2923
        goto cleanup;
2924

2925
    if ((net = virNetworkAssignDef(&privconn->networks, def)) == NULL)
2926 2927
        goto cleanup;
    def = NULL;
2928
    net->persistent = 1;
2929

2930
    ret = virGetNetwork(conn, net->def->name, net->def->uuid);
2931 2932 2933

cleanup:
    virNetworkDefFree(def);
2934 2935 2936
    if (net)
        virNetworkObjUnlock(net);
    testDriverUnlock(privconn);
2937
    return ret;
2938 2939 2940
}

static int testNetworkUndefine(virNetworkPtr network) {
2941 2942
    testConnPtr privconn = network->conn->privateData;
    virNetworkObjPtr privnet;
2943
    int ret = -1;
2944

2945
    testDriverLock(privconn);
2946 2947 2948 2949
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);

    if (privnet == NULL) {
2950
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2951
        goto cleanup;
2952
    }
2953

D
Daniel P. Berrange 已提交
2954
    if (virNetworkObjIsActive(privnet)) {
2955
        testError(VIR_ERR_INTERNAL_ERROR,
2956
                  _("Network '%s' is still running"), network->name);
2957
        goto cleanup;
2958 2959
    }

2960 2961
    virNetworkRemoveInactive(&privconn->networks,
                             privnet);
2962
    privnet = NULL;
2963
    ret = 0;
2964

2965
cleanup:
2966 2967 2968
    if (privnet)
        virNetworkObjUnlock(privnet);
    testDriverUnlock(privconn);
2969
    return ret;
2970 2971 2972
}

static int testNetworkStart(virNetworkPtr network) {
2973 2974
    testConnPtr privconn = network->conn->privateData;
    virNetworkObjPtr privnet;
2975
    int ret = -1;
2976

2977
    testDriverLock(privconn);
2978 2979
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);
2980
    testDriverUnlock(privconn);
2981 2982

    if (privnet == NULL) {
2983
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
2984
        goto cleanup;
2985
    }
2986

D
Daniel P. Berrange 已提交
2987
    if (virNetworkObjIsActive(privnet)) {
2988
        testError(VIR_ERR_INTERNAL_ERROR,
2989
                  _("Network '%s' is already running"), network->name);
2990
        goto cleanup;
2991 2992
    }

2993
    privnet->active = 1;
2994
    ret = 0;
2995

2996
cleanup:
2997 2998
    if (privnet)
        virNetworkObjUnlock(privnet);
2999
    return ret;
3000 3001 3002
}

static int testNetworkDestroy(virNetworkPtr network) {
3003 3004
    testConnPtr privconn = network->conn->privateData;
    virNetworkObjPtr privnet;
3005
    int ret = -1;
3006

3007
    testDriverLock(privconn);
3008 3009 3010 3011
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);

    if (privnet == NULL) {
3012
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
3013
        goto cleanup;
3014
    }
3015

3016 3017 3018 3019
    privnet->active = 0;
    if (!privnet->persistent) {
        virNetworkRemoveInactive(&privconn->networks,
                                 privnet);
3020
        privnet = NULL;
3021
    }
3022 3023 3024
    ret = 0;

cleanup:
3025 3026 3027
    if (privnet)
        virNetworkObjUnlock(privnet);
    testDriverUnlock(privconn);
3028
    return ret;
3029 3030 3031
}

static char *testNetworkDumpXML(virNetworkPtr network, int flags ATTRIBUTE_UNUSED) {
3032 3033
    testConnPtr privconn = network->conn->privateData;
    virNetworkObjPtr privnet;
3034
    char *ret = NULL;
3035

3036
    testDriverLock(privconn);
3037 3038
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);
3039
    testDriverUnlock(privconn);
3040 3041

    if (privnet == NULL) {
3042
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
3043
        goto cleanup;
3044
    }
3045

3046
    ret = virNetworkDefFormat(privnet->def);
3047 3048

cleanup:
3049 3050
    if (privnet)
        virNetworkObjUnlock(privnet);
3051
    return ret;
3052 3053 3054
}

static char *testNetworkGetBridgeName(virNetworkPtr network) {
3055
    testConnPtr privconn = network->conn->privateData;
3056
    char *bridge = NULL;
3057 3058
    virNetworkObjPtr privnet;

3059
    testDriverLock(privconn);
3060 3061
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);
3062
    testDriverUnlock(privconn);
3063 3064

    if (privnet == NULL) {
3065
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
3066
        goto cleanup;
3067 3068
    }

3069
    if (!(privnet->def->bridge)) {
3070
        testError(VIR_ERR_INTERNAL_ERROR,
3071 3072 3073 3074 3075 3076
                  _("network '%s' does not have a bridge name."),
                  privnet->def->name);
        goto cleanup;
    }

    if (!(bridge = strdup(privnet->def->bridge))) {
3077
        virReportOOMError();
3078
        goto cleanup;
3079
    }
3080 3081

cleanup:
3082 3083
    if (privnet)
        virNetworkObjUnlock(privnet);
3084 3085 3086 3087 3088
    return bridge;
}

static int testNetworkGetAutostart(virNetworkPtr network,
                                   int *autostart) {
3089 3090
    testConnPtr privconn = network->conn->privateData;
    virNetworkObjPtr privnet;
3091
    int ret = -1;
3092

3093
    testDriverLock(privconn);
3094 3095
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);
3096
    testDriverUnlock(privconn);
3097 3098

    if (privnet == NULL) {
3099
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
3100
        goto cleanup;
3101 3102
    }

3103
    *autostart = privnet->autostart;
3104 3105 3106
    ret = 0;

cleanup:
3107 3108
    if (privnet)
        virNetworkObjUnlock(privnet);
3109
    return ret;
3110 3111 3112 3113
}

static int testNetworkSetAutostart(virNetworkPtr network,
                                   int autostart) {
3114 3115
    testConnPtr privconn = network->conn->privateData;
    virNetworkObjPtr privnet;
3116
    int ret = -1;
3117

3118
    testDriverLock(privconn);
3119 3120
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);
3121
    testDriverUnlock(privconn);
3122 3123

    if (privnet == NULL) {
3124
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
3125
        goto cleanup;
3126 3127
    }

3128
    privnet->autostart = autostart ? 1 : 0;
3129 3130 3131
    ret = 0;

cleanup:
3132 3133
    if (privnet)
        virNetworkObjUnlock(privnet);
3134
    return ret;
3135
}
3136

C
Cole Robinson 已提交
3137

L
Laine Stump 已提交
3138 3139 3140 3141 3142 3143 3144 3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165 3166 3167
/*
 * Physical host interface routines
 */

static virDrvOpenStatus testOpenInterface(virConnectPtr conn,
                                          virConnectAuthPtr auth ATTRIBUTE_UNUSED,
                                          int flags ATTRIBUTE_UNUSED)
{
    if (STRNEQ(conn->driver->name, "Test"))
        return VIR_DRV_OPEN_DECLINED;

    conn->interfacePrivateData = conn->privateData;
    return VIR_DRV_OPEN_SUCCESS;
}

static int testCloseInterface(virConnectPtr conn)
{
    conn->interfacePrivateData = NULL;
    return 0;
}


static int testNumOfInterfaces(virConnectPtr conn)
{
    testConnPtr privconn = conn->privateData;
    int i, count = 0;

    testDriverLock(privconn);
    for (i = 0 ; (i < privconn->ifaces.count); i++) {
        virInterfaceObjLock(privconn->ifaces.objs[i]);
D
Daniel P. Berrange 已提交
3168
        if (virInterfaceObjIsActive(privconn->ifaces.objs[i])) {
L
Laine Stump 已提交
3169 3170 3171 3172 3173 3174 3175 3176 3177 3178 3179 3180 3181 3182 3183 3184 3185
            count++;
        }
        virInterfaceObjUnlock(privconn->ifaces.objs[i]);
    }
    testDriverUnlock(privconn);
    return count;
}

static int testListInterfaces(virConnectPtr conn, char **const names, int nnames)
{
    testConnPtr privconn = conn->privateData;
    int n = 0, i;

    testDriverLock(privconn);
    memset(names, 0, sizeof(*names)*nnames);
    for (i = 0 ; (i < privconn->ifaces.count) && (n < nnames); i++) {
        virInterfaceObjLock(privconn->ifaces.objs[i]);
D
Daniel P. Berrange 已提交
3186
        if (virInterfaceObjIsActive(privconn->ifaces.objs[i])) {
L
Laine Stump 已提交
3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198
            if (!(names[n++] = strdup(privconn->ifaces.objs[i]->def->name))) {
                virInterfaceObjUnlock(privconn->ifaces.objs[i]);
                goto no_memory;
            }
        }
        virInterfaceObjUnlock(privconn->ifaces.objs[i]);
    }
    testDriverUnlock(privconn);

    return n;

no_memory:
3199
    virReportOOMError();
L
Laine Stump 已提交
3200 3201 3202 3203 3204 3205 3206 3207 3208 3209 3210 3211 3212 3213
    for (n = 0 ; n < nnames ; n++)
        VIR_FREE(names[n]);
    testDriverUnlock(privconn);
    return -1;
}

static int testNumOfDefinedInterfaces(virConnectPtr conn)
{
    testConnPtr privconn = conn->privateData;
    int i, count = 0;

    testDriverLock(privconn);
    for (i = 0 ; i < privconn->ifaces.count; i++) {
        virInterfaceObjLock(privconn->ifaces.objs[i]);
D
Daniel P. Berrange 已提交
3214
        if (!virInterfaceObjIsActive(privconn->ifaces.objs[i])) {
L
Laine Stump 已提交
3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226 3227 3228 3229 3230 3231
            count++;
        }
        virInterfaceObjUnlock(privconn->ifaces.objs[i]);
    }
    testDriverUnlock(privconn);
    return count;
}

static int testListDefinedInterfaces(virConnectPtr conn, char **const names, int nnames)
{
    testConnPtr privconn = conn->privateData;
    int n = 0, i;

    testDriverLock(privconn);
    memset(names, 0, sizeof(*names)*nnames);
    for (i = 0 ; (i < privconn->ifaces.count) && (n < nnames); i++) {
        virInterfaceObjLock(privconn->ifaces.objs[i]);
D
Daniel P. Berrange 已提交
3232
        if (!virInterfaceObjIsActive(privconn->ifaces.objs[i])) {
L
Laine Stump 已提交
3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244
            if (!(names[n++] = strdup(privconn->ifaces.objs[i]->def->name))) {
                virInterfaceObjUnlock(privconn->ifaces.objs[i]);
                goto no_memory;
            }
        }
        virInterfaceObjUnlock(privconn->ifaces.objs[i]);
    }
    testDriverUnlock(privconn);

    return n;

no_memory:
3245
    virReportOOMError();
L
Laine Stump 已提交
3246 3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262 3263
    for (n = 0 ; n < nnames ; n++)
        VIR_FREE(names[n]);
    testDriverUnlock(privconn);
    return -1;
}

static virInterfacePtr testLookupInterfaceByName(virConnectPtr conn,
                                                 const char *name)
{
    testConnPtr privconn = conn->privateData;
    virInterfaceObjPtr iface;
    virInterfacePtr ret = NULL;

    testDriverLock(privconn);
    iface = virInterfaceFindByName(&privconn->ifaces, name);
    testDriverUnlock(privconn);

    if (iface == NULL) {
3264
        testError(VIR_ERR_NO_INTERFACE, NULL);
L
Laine Stump 已提交
3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283 3284 3285 3286 3287 3288
        goto cleanup;
    }

    ret = virGetInterface(conn, iface->def->name, iface->def->mac);

cleanup:
    if (iface)
        virInterfaceObjUnlock(iface);
    return ret;
}

static virInterfacePtr testLookupInterfaceByMACString(virConnectPtr conn,
                                                      const char *mac)
{
    testConnPtr privconn = conn->privateData;
    virInterfaceObjPtr iface;
    int ifacect;
    virInterfacePtr ret = NULL;

    testDriverLock(privconn);
    ifacect = virInterfaceFindByMACString(&privconn->ifaces, mac, &iface, 1);
    testDriverUnlock(privconn);

    if (ifacect == 0) {
3289
        testError(VIR_ERR_NO_INTERFACE, NULL);
L
Laine Stump 已提交
3290 3291 3292 3293
        goto cleanup;
    }

    if (ifacect > 1) {
3294
        testError(VIR_ERR_MULTIPLE_INTERFACES, NULL);
L
Laine Stump 已提交
3295 3296 3297 3298 3299 3300 3301 3302 3303 3304 3305
        goto cleanup;
    }

    ret = virGetInterface(conn, iface->def->name, iface->def->mac);

cleanup:
    if (iface)
        virInterfaceObjUnlock(iface);
    return ret;
}

3306 3307 3308 3309 3310 3311 3312 3313 3314 3315
static int testInterfaceIsActive(virInterfacePtr iface)
{
    testConnPtr privconn = iface->conn->privateData;
    virInterfaceObjPtr obj;
    int ret = -1;

    testDriverLock(privconn);
    obj = virInterfaceFindByName(&privconn->ifaces, iface->name);
    testDriverUnlock(privconn);
    if (!obj) {
3316
        testError(VIR_ERR_NO_INTERFACE, NULL);
3317 3318 3319 3320 3321 3322 3323 3324 3325 3326 3327
        goto cleanup;
    }
    ret = virInterfaceObjIsActive(obj);

cleanup:
    if (obj)
        virInterfaceObjUnlock(obj);
    return ret;
}


L
Laine Stump 已提交
3328 3329 3330 3331 3332 3333 3334 3335 3336 3337 3338 3339 3340
static char *testInterfaceGetXMLDesc(virInterfacePtr iface,
                                     unsigned int flags ATTRIBUTE_UNUSED)
{
    testConnPtr privconn = iface->conn->privateData;
    virInterfaceObjPtr privinterface;
    char *ret = NULL;

    testDriverLock(privconn);
    privinterface = virInterfaceFindByName(&privconn->ifaces,
                                           iface->name);
    testDriverUnlock(privconn);

    if (privinterface == NULL) {
3341
        testError(VIR_ERR_NO_INTERFACE, __FUNCTION__);
L
Laine Stump 已提交
3342 3343 3344
        goto cleanup;
    }

3345
    ret = virInterfaceDefFormat(privinterface->def);
L
Laine Stump 已提交
3346 3347 3348 3349 3350 3351 3352 3353 3354 3355 3356 3357 3358 3359 3360 3361 3362

cleanup:
    if (privinterface)
        virInterfaceObjUnlock(privinterface);
    return ret;
}


static virInterfacePtr testInterfaceDefineXML(virConnectPtr conn, const char *xmlStr,
                                              unsigned int flags ATTRIBUTE_UNUSED)
{
    testConnPtr privconn = conn->privateData;
    virInterfaceDefPtr def;
    virInterfaceObjPtr iface = NULL;
    virInterfacePtr ret = NULL;

    testDriverLock(privconn);
3363
    if ((def = virInterfaceDefParseString(xmlStr)) == NULL)
L
Laine Stump 已提交
3364 3365
        goto cleanup;

3366
    if ((iface = virInterfaceAssignDef(&privconn->ifaces, def)) == NULL)
L
Laine Stump 已提交
3367 3368 3369 3370 3371 3372 3373 3374 3375 3376 3377 3378 3379 3380 3381 3382 3383 3384 3385 3386 3387 3388 3389 3390
        goto cleanup;
    def = NULL;

    ret = virGetInterface(conn, iface->def->name, iface->def->mac);

cleanup:
    virInterfaceDefFree(def);
    if (iface)
        virInterfaceObjUnlock(iface);
    testDriverUnlock(privconn);
    return ret;
}

static int testInterfaceUndefine(virInterfacePtr iface)
{
    testConnPtr privconn = iface->conn->privateData;
    virInterfaceObjPtr privinterface;
    int ret = -1;

    testDriverLock(privconn);
    privinterface = virInterfaceFindByName(&privconn->ifaces,
                                           iface->name);

    if (privinterface == NULL) {
3391
        testError(VIR_ERR_NO_INTERFACE, NULL);
L
Laine Stump 已提交
3392 3393 3394 3395 3396 3397 3398 3399 3400 3401 3402 3403 3404 3405 3406 3407 3408 3409 3410 3411 3412 3413 3414 3415
        goto cleanup;
    }

    virInterfaceRemove(&privconn->ifaces,
                       privinterface);
    ret = 0;

cleanup:
    testDriverUnlock(privconn);
    return ret;
}

static int testInterfaceCreate(virInterfacePtr iface,
                               unsigned int flags ATTRIBUTE_UNUSED)
{
    testConnPtr privconn = iface->conn->privateData;
    virInterfaceObjPtr privinterface;
    int ret = -1;

    testDriverLock(privconn);
    privinterface = virInterfaceFindByName(&privconn->ifaces,
                                           iface->name);

    if (privinterface == NULL) {
3416
        testError(VIR_ERR_NO_INTERFACE, NULL);
L
Laine Stump 已提交
3417 3418 3419 3420
        goto cleanup;
    }

    if (privinterface->active != 0) {
3421
        testError(VIR_ERR_OPERATION_INVALID, NULL);
L
Laine Stump 已提交
3422 3423 3424 3425 3426 3427 3428 3429 3430 3431 3432 3433 3434 3435 3436 3437 3438 3439 3440 3441 3442 3443 3444 3445 3446
        goto cleanup;
    }

    privinterface->active = 1;
    ret = 0;

cleanup:
    if (privinterface)
        virInterfaceObjUnlock(privinterface);
    testDriverUnlock(privconn);
    return ret;
}

static int testInterfaceDestroy(virInterfacePtr iface,
                                unsigned int flags ATTRIBUTE_UNUSED)
{
    testConnPtr privconn = iface->conn->privateData;
    virInterfaceObjPtr privinterface;
    int ret = -1;

    testDriverLock(privconn);
    privinterface = virInterfaceFindByName(&privconn->ifaces,
                                           iface->name);

    if (privinterface == NULL) {
3447
        testError(VIR_ERR_NO_INTERFACE, NULL);
L
Laine Stump 已提交
3448 3449 3450 3451
        goto cleanup;
    }

    if (privinterface->active == 0) {
3452
        testError(VIR_ERR_OPERATION_INVALID, NULL);
L
Laine Stump 已提交
3453 3454 3455 3456 3457 3458 3459 3460 3461 3462 3463 3464 3465 3466 3467
        goto cleanup;
    }

    privinterface->active = 0;
    ret = 0;

cleanup:
    if (privinterface)
        virInterfaceObjUnlock(privinterface);
    testDriverUnlock(privconn);
    return ret;
}



C
Cole Robinson 已提交
3468 3469 3470 3471
/*
 * Storage Driver routines
 */

3472

3473
static int testStoragePoolObjSetDefaults(virStoragePoolObjPtr pool) {
C
Cole Robinson 已提交
3474 3475 3476 3477 3478 3479 3480

    pool->def->capacity = defaultPoolCap;
    pool->def->allocation = defaultPoolAlloc;
    pool->def->available = defaultPoolCap - defaultPoolAlloc;

    pool->configFile = strdup("\0");
    if (!pool->configFile) {
3481
        virReportOOMError();
C
Cole Robinson 已提交
3482 3483 3484 3485 3486 3487
        return -1;
    }

    return 0;
}

3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502
static virDrvOpenStatus testStorageOpen(virConnectPtr conn,
                                        virConnectAuthPtr auth ATTRIBUTE_UNUSED,
                                        int flags ATTRIBUTE_UNUSED) {
    if (STRNEQ(conn->driver->name, "Test"))
        return VIR_DRV_OPEN_DECLINED;

    conn->storagePrivateData = conn->privateData;
    return VIR_DRV_OPEN_SUCCESS;
}

static int testStorageClose(virConnectPtr conn) {
    conn->storagePrivateData = NULL;
    return 0;
}

3503

C
Cole Robinson 已提交
3504 3505 3506
static virStoragePoolPtr
testStoragePoolLookupByUUID(virConnectPtr conn,
                            const unsigned char *uuid) {
3507
    testConnPtr privconn = conn->privateData;
3508 3509
    virStoragePoolObjPtr pool;
    virStoragePoolPtr ret = NULL;
C
Cole Robinson 已提交
3510

3511
    testDriverLock(privconn);
3512
    pool = virStoragePoolObjFindByUUID(&privconn->pools, uuid);
3513
    testDriverUnlock(privconn);
3514 3515

    if (pool == NULL) {
3516
        testError(VIR_ERR_NO_STORAGE_POOL, NULL);
3517
        goto cleanup;
C
Cole Robinson 已提交
3518 3519
    }

3520 3521 3522
    ret = virGetStoragePool(conn, pool->def->name, pool->def->uuid);

cleanup:
3523 3524
    if (pool)
        virStoragePoolObjUnlock(pool);
3525
    return ret;
C
Cole Robinson 已提交
3526 3527 3528 3529 3530
}

static virStoragePoolPtr
testStoragePoolLookupByName(virConnectPtr conn,
                            const char *name) {
3531
    testConnPtr privconn = conn->privateData;
3532 3533
    virStoragePoolObjPtr pool;
    virStoragePoolPtr ret = NULL;
C
Cole Robinson 已提交
3534

3535
    testDriverLock(privconn);
3536
    pool = virStoragePoolObjFindByName(&privconn->pools, name);
3537
    testDriverUnlock(privconn);
3538 3539

    if (pool == NULL) {
3540
        testError(VIR_ERR_NO_STORAGE_POOL, NULL);
3541
        goto cleanup;
C
Cole Robinson 已提交
3542 3543
    }

3544 3545 3546
    ret = virGetStoragePool(conn, pool->def->name, pool->def->uuid);

cleanup:
3547 3548
    if (pool)
        virStoragePoolObjUnlock(pool);
3549
    return ret;
C
Cole Robinson 已提交
3550 3551 3552 3553 3554 3555 3556 3557 3558
}

static virStoragePoolPtr
testStoragePoolLookupByVolume(virStorageVolPtr vol) {
    return testStoragePoolLookupByName(vol->conn, vol->pool);
}

static int
testStorageNumPools(virConnectPtr conn) {
3559
    testConnPtr privconn = conn->privateData;
C
Cole Robinson 已提交
3560 3561
    int numActive = 0, i;

3562
    testDriverLock(privconn);
C
Cole Robinson 已提交
3563 3564 3565
    for (i = 0 ; i < privconn->pools.count ; i++)
        if (virStoragePoolObjIsActive(privconn->pools.objs[i]))
            numActive++;
3566
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
3567 3568 3569 3570 3571 3572 3573 3574

    return numActive;
}

static int
testStorageListPools(virConnectPtr conn,
                     char **const names,
                     int nnames) {
3575
    testConnPtr privconn = conn->privateData;
C
Cole Robinson 已提交
3576 3577
    int n = 0, i;

3578
    testDriverLock(privconn);
C
Cole Robinson 已提交
3579
    memset(names, 0, sizeof(*names)*nnames);
3580 3581
    for (i = 0 ; i < privconn->pools.count && n < nnames ; i++) {
        virStoragePoolObjLock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
3582
        if (virStoragePoolObjIsActive(privconn->pools.objs[i]) &&
3583 3584
            !(names[n++] = strdup(privconn->pools.objs[i]->def->name))) {
            virStoragePoolObjUnlock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
3585
            goto no_memory;
3586 3587 3588 3589
        }
        virStoragePoolObjUnlock(privconn->pools.objs[i]);
    }
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
3590 3591 3592 3593

    return n;

no_memory:
3594
    virReportOOMError();
C
Cole Robinson 已提交
3595 3596
    for (n = 0 ; n < nnames ; n++)
        VIR_FREE(names[n]);
3597
    testDriverUnlock(privconn);
3598
    return -1;
C
Cole Robinson 已提交
3599 3600 3601 3602
}

static int
testStorageNumDefinedPools(virConnectPtr conn) {
3603
    testConnPtr privconn = conn->privateData;
C
Cole Robinson 已提交
3604 3605
    int numInactive = 0, i;

3606 3607 3608
    testDriverLock(privconn);
    for (i = 0 ; i < privconn->pools.count ; i++) {
        virStoragePoolObjLock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
3609 3610
        if (!virStoragePoolObjIsActive(privconn->pools.objs[i]))
            numInactive++;
3611 3612 3613
        virStoragePoolObjUnlock(privconn->pools.objs[i]);
    }
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
3614 3615 3616 3617 3618 3619 3620 3621

    return numInactive;
}

static int
testStorageListDefinedPools(virConnectPtr conn,
                            char **const names,
                            int nnames) {
3622
    testConnPtr privconn = conn->privateData;
C
Cole Robinson 已提交
3623 3624
    int n = 0, i;

3625
    testDriverLock(privconn);
C
Cole Robinson 已提交
3626
    memset(names, 0, sizeof(*names)*nnames);
3627 3628
    for (i = 0 ; i < privconn->pools.count && n < nnames ; i++) {
        virStoragePoolObjLock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
3629
        if (!virStoragePoolObjIsActive(privconn->pools.objs[i]) &&
3630 3631
            !(names[n++] = strdup(privconn->pools.objs[i]->def->name))) {
            virStoragePoolObjUnlock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
3632
            goto no_memory;
3633 3634 3635 3636
        }
        virStoragePoolObjUnlock(privconn->pools.objs[i]);
    }
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
3637 3638 3639 3640

    return n;

no_memory:
3641
    virReportOOMError();
C
Cole Robinson 已提交
3642 3643
    for (n = 0 ; n < nnames ; n++)
        VIR_FREE(names[n]);
3644
    testDriverUnlock(privconn);
3645
    return -1;
C
Cole Robinson 已提交
3646 3647 3648
}


3649 3650 3651 3652 3653 3654 3655 3656 3657 3658
static int testStoragePoolIsActive(virStoragePoolPtr pool)
{
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr obj;
    int ret = -1;

    testDriverLock(privconn);
    obj = virStoragePoolObjFindByUUID(&privconn->pools, pool->uuid);
    testDriverUnlock(privconn);
    if (!obj) {
3659
        testError(VIR_ERR_NO_STORAGE_POOL, NULL);
3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679
        goto cleanup;
    }
    ret = virStoragePoolObjIsActive(obj);

cleanup:
    if (obj)
        virStoragePoolObjUnlock(obj);
    return ret;
}

static int testStoragePoolIsPersistent(virStoragePoolPtr pool)
{
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr obj;
    int ret = -1;

    testDriverLock(privconn);
    obj = virStoragePoolObjFindByUUID(&privconn->pools, pool->uuid);
    testDriverUnlock(privconn);
    if (!obj) {
3680
        testError(VIR_ERR_NO_STORAGE_POOL, NULL);
3681 3682 3683 3684 3685 3686 3687 3688 3689 3690 3691 3692
        goto cleanup;
    }
    ret = obj->configFile ? 1 : 0;

cleanup:
    if (obj)
        virStoragePoolObjUnlock(obj);
    return ret;
}



C
Cole Robinson 已提交
3693
static int
3694
testStoragePoolStart(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3695
                     unsigned int flags ATTRIBUTE_UNUSED) {
3696 3697
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3698
    int ret = -1;
3699

3700
    testDriverLock(privconn);
3701 3702
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3703
    testDriverUnlock(privconn);
3704 3705

    if (privpool == NULL) {
3706
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
3707
        goto cleanup;
3708 3709
    }

3710
    if (virStoragePoolObjIsActive(privpool)) {
3711
        testError(VIR_ERR_INTERNAL_ERROR,
3712 3713 3714
                  _("storage pool '%s' is already active"), pool->name);
        goto cleanup;
    }
C
Cole Robinson 已提交
3715 3716

    privpool->active = 1;
3717
    ret = 0;
C
Cole Robinson 已提交
3718

3719
cleanup:
3720 3721
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3722
    return ret;
C
Cole Robinson 已提交
3723 3724 3725
}

static char *
3726
testStorageFindPoolSources(virConnectPtr conn ATTRIBUTE_UNUSED,
3727 3728
                           const char *type,
                           const char *srcSpec,
C
Cole Robinson 已提交
3729 3730
                           unsigned int flags ATTRIBUTE_UNUSED)
{
3731 3732 3733 3734 3735 3736
    virStoragePoolSourcePtr source = NULL;
    int pool_type;
    char *ret = NULL;

    pool_type = virStoragePoolTypeFromString(type);
    if (!pool_type) {
3737
        testError(VIR_ERR_INTERNAL_ERROR,
3738 3739 3740 3741 3742
                  _("unknown storage pool type %s"), type);
        goto cleanup;
    }

    if (srcSpec) {
3743
        source = virStoragePoolDefParseSourceString(srcSpec, pool_type);
3744 3745 3746 3747 3748 3749 3750 3751 3752
        if (!source)
            goto cleanup;
    }

    switch (pool_type) {

    case VIR_STORAGE_POOL_LOGICAL:
        ret = strdup(defaultPoolSourcesLogicalXML);
        if (!ret)
3753
            virReportOOMError();
3754 3755 3756 3757
        break;

    case VIR_STORAGE_POOL_NETFS:
        if (!source || !source->host.name) {
3758
            testError(VIR_ERR_INVALID_ARG,
3759 3760 3761 3762 3763 3764
                      "%s", "hostname must be specified for netfs sources");
            goto cleanup;
        }

        if (virAsprintf(&ret, defaultPoolSourcesNetFSXML,
                        source->host.name) < 0)
3765
            virReportOOMError();
3766 3767 3768
        break;

    default:
3769
        testError(VIR_ERR_NO_SUPPORT,
3770 3771 3772 3773 3774
                  _("pool type '%s' does not support source discovery"), type);
    }

cleanup:
    virStoragePoolSourceFree(source);
3775
    VIR_FREE(source);
3776
    return ret;
C
Cole Robinson 已提交
3777 3778 3779 3780 3781 3782 3783
}


static virStoragePoolPtr
testStoragePoolCreate(virConnectPtr conn,
                      const char *xml,
                      unsigned int flags ATTRIBUTE_UNUSED) {
3784
    testConnPtr privconn = conn->privateData;
C
Cole Robinson 已提交
3785
    virStoragePoolDefPtr def;
3786
    virStoragePoolObjPtr pool = NULL;
3787
    virStoragePoolPtr ret = NULL;
C
Cole Robinson 已提交
3788

3789
    testDriverLock(privconn);
3790
    if (!(def = virStoragePoolDefParseString(xml)))
3791
        goto cleanup;
C
Cole Robinson 已提交
3792

3793 3794 3795 3796
    pool = virStoragePoolObjFindByUUID(&privconn->pools, def->uuid);
    if (!pool)
        pool = virStoragePoolObjFindByName(&privconn->pools, def->name);
    if (pool) {
3797
        testError(VIR_ERR_INTERNAL_ERROR,
C
Cole Robinson 已提交
3798
                  "%s", _("storage pool already exists"));
3799
        goto cleanup;
C
Cole Robinson 已提交
3800 3801
    }

3802
    if (!(pool = virStoragePoolObjAssignDef(&privconn->pools, def)))
3803
        goto cleanup;
3804
    def = NULL;
C
Cole Robinson 已提交
3805

3806
    if (testStoragePoolObjSetDefaults(pool) == -1) {
C
Cole Robinson 已提交
3807
        virStoragePoolObjRemove(&privconn->pools, pool);
3808 3809
        pool = NULL;
        goto cleanup;
C
Cole Robinson 已提交
3810 3811 3812
    }
    pool->active = 1;

3813 3814 3815 3816
    ret = virGetStoragePool(conn, pool->def->name, pool->def->uuid);

cleanup:
    virStoragePoolDefFree(def);
3817 3818 3819
    if (pool)
        virStoragePoolObjUnlock(pool);
    testDriverUnlock(privconn);
3820
    return ret;
C
Cole Robinson 已提交
3821 3822 3823 3824 3825 3826
}

static virStoragePoolPtr
testStoragePoolDefine(virConnectPtr conn,
                      const char *xml,
                      unsigned int flags ATTRIBUTE_UNUSED) {
3827
    testConnPtr privconn = conn->privateData;
C
Cole Robinson 已提交
3828
    virStoragePoolDefPtr def;
3829
    virStoragePoolObjPtr pool = NULL;
3830
    virStoragePoolPtr ret = NULL;
C
Cole Robinson 已提交
3831

3832
    testDriverLock(privconn);
3833
    if (!(def = virStoragePoolDefParseString(xml)))
3834
        goto cleanup;
C
Cole Robinson 已提交
3835 3836 3837 3838 3839

    def->capacity = defaultPoolCap;
    def->allocation = defaultPoolAlloc;
    def->available = defaultPoolCap - defaultPoolAlloc;

3840
    if (!(pool = virStoragePoolObjAssignDef(&privconn->pools, def)))
3841 3842
        goto cleanup;
    def = NULL;
C
Cole Robinson 已提交
3843

3844
    if (testStoragePoolObjSetDefaults(pool) == -1) {
C
Cole Robinson 已提交
3845
        virStoragePoolObjRemove(&privconn->pools, pool);
3846 3847
        pool = NULL;
        goto cleanup;
C
Cole Robinson 已提交
3848 3849
    }

3850 3851 3852 3853
    ret = virGetStoragePool(conn, pool->def->name, pool->def->uuid);

cleanup:
    virStoragePoolDefFree(def);
3854 3855 3856
    if (pool)
        virStoragePoolObjUnlock(pool);
    testDriverUnlock(privconn);
3857
    return ret;
C
Cole Robinson 已提交
3858 3859 3860
}

static int
3861 3862 3863
testStoragePoolUndefine(virStoragePoolPtr pool) {
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3864
    int ret = -1;
3865

3866
    testDriverLock(privconn);
3867 3868 3869 3870
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);

    if (privpool == NULL) {
3871
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
3872
        goto cleanup;
3873 3874
    }

3875
    if (virStoragePoolObjIsActive(privpool)) {
3876
        testError(VIR_ERR_INTERNAL_ERROR,
3877 3878 3879
                  _("storage pool '%s' is already active"), pool->name);
        goto cleanup;
    }
C
Cole Robinson 已提交
3880 3881

    virStoragePoolObjRemove(&privconn->pools, privpool);
3882
    ret = 0;
C
Cole Robinson 已提交
3883

3884
cleanup:
3885 3886 3887
    if (privpool)
        virStoragePoolObjUnlock(privpool);
    testDriverUnlock(privconn);
3888
    return ret;
C
Cole Robinson 已提交
3889 3890 3891
}

static int
3892
testStoragePoolBuild(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3893
                     unsigned int flags ATTRIBUTE_UNUSED) {
3894 3895
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3896
    int ret = -1;
3897

3898
    testDriverLock(privconn);
3899 3900
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3901
    testDriverUnlock(privconn);
3902 3903

    if (privpool == NULL) {
3904
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
3905
        goto cleanup;
3906 3907
    }

3908
    if (virStoragePoolObjIsActive(privpool)) {
3909
        testError(VIR_ERR_INTERNAL_ERROR,
3910 3911 3912
                  _("storage pool '%s' is already active"), pool->name);
        goto cleanup;
    }
3913
    ret = 0;
C
Cole Robinson 已提交
3914

3915
cleanup:
3916 3917
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3918
    return ret;
C
Cole Robinson 已提交
3919 3920 3921 3922
}


static int
3923 3924 3925
testStoragePoolDestroy(virStoragePoolPtr pool) {
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3926
    int ret = -1;
3927

3928
    testDriverLock(privconn);
3929 3930 3931 3932
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);

    if (privpool == NULL) {
3933
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
3934
        goto cleanup;
3935 3936 3937
    }

    if (!virStoragePoolObjIsActive(privpool)) {
3938
        testError(VIR_ERR_INTERNAL_ERROR,
3939
                  _("storage pool '%s' is not active"), pool->name);
3940
        goto cleanup;
3941
    }
C
Cole Robinson 已提交
3942 3943 3944

    privpool->active = 0;

3945
    if (privpool->configFile == NULL) {
C
Cole Robinson 已提交
3946
        virStoragePoolObjRemove(&privconn->pools, privpool);
3947 3948
        privpool = NULL;
    }
3949
    ret = 0;
C
Cole Robinson 已提交
3950

3951
cleanup:
3952 3953 3954
    if (privpool)
        virStoragePoolObjUnlock(privpool);
    testDriverUnlock(privconn);
3955
    return ret;
C
Cole Robinson 已提交
3956 3957 3958 3959
}


static int
3960
testStoragePoolDelete(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3961
                      unsigned int flags ATTRIBUTE_UNUSED) {
3962 3963
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3964
    int ret = -1;
3965

3966
    testDriverLock(privconn);
3967 3968
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3969
    testDriverUnlock(privconn);
3970 3971

    if (privpool == NULL) {
3972
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
3973 3974 3975 3976
        goto cleanup;
    }

    if (virStoragePoolObjIsActive(privpool)) {
3977
        testError(VIR_ERR_INTERNAL_ERROR,
3978 3979
                  _("storage pool '%s' is already active"), pool->name);
        goto cleanup;
3980 3981
    }

3982
    ret = 0;
C
Cole Robinson 已提交
3983

3984
cleanup:
3985 3986
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3987
    return ret;
C
Cole Robinson 已提交
3988 3989 3990 3991
}


static int
3992
testStoragePoolRefresh(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3993
                       unsigned int flags ATTRIBUTE_UNUSED) {
3994 3995
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3996
    int ret = -1;
3997

3998
    testDriverLock(privconn);
3999 4000
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
4001
    testDriverUnlock(privconn);
4002 4003

    if (privpool == NULL) {
4004
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4005
        goto cleanup;
4006 4007 4008
    }

    if (!virStoragePoolObjIsActive(privpool)) {
4009
        testError(VIR_ERR_INTERNAL_ERROR,
4010
                  _("storage pool '%s' is not active"), pool->name);
4011
        goto cleanup;
4012
    }
4013
    ret = 0;
C
Cole Robinson 已提交
4014

4015
cleanup:
4016 4017
    if (privpool)
        virStoragePoolObjUnlock(privpool);
4018
    return ret;
C
Cole Robinson 已提交
4019 4020 4021 4022
}


static int
4023
testStoragePoolGetInfo(virStoragePoolPtr pool,
C
Cole Robinson 已提交
4024
                       virStoragePoolInfoPtr info) {
4025 4026
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
4027
    int ret = -1;
4028

4029
    testDriverLock(privconn);
4030 4031
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
4032
    testDriverUnlock(privconn);
4033 4034

    if (privpool == NULL) {
4035
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4036
        goto cleanup;
4037
    }
C
Cole Robinson 已提交
4038 4039 4040 4041 4042 4043 4044 4045 4046

    memset(info, 0, sizeof(virStoragePoolInfo));
    if (privpool->active)
        info->state = VIR_STORAGE_POOL_RUNNING;
    else
        info->state = VIR_STORAGE_POOL_INACTIVE;
    info->capacity = privpool->def->capacity;
    info->allocation = privpool->def->allocation;
    info->available = privpool->def->available;
4047
    ret = 0;
C
Cole Robinson 已提交
4048

4049
cleanup:
4050 4051
    if (privpool)
        virStoragePoolObjUnlock(privpool);
4052
    return ret;
C
Cole Robinson 已提交
4053 4054 4055
}

static char *
4056
testStoragePoolDumpXML(virStoragePoolPtr pool,
C
Cole Robinson 已提交
4057
                       unsigned int flags ATTRIBUTE_UNUSED) {
4058 4059
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
4060
    char *ret = NULL;
4061

4062
    testDriverLock(privconn);
4063 4064
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
4065
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
4066

4067
    if (privpool == NULL) {
4068
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4069
        goto cleanup;
4070 4071
    }

4072
    ret = virStoragePoolDefFormat(privpool->def);
4073 4074

cleanup:
4075 4076
    if (privpool)
        virStoragePoolObjUnlock(privpool);
4077
    return ret;
C
Cole Robinson 已提交
4078 4079 4080
}

static int
4081
testStoragePoolGetAutostart(virStoragePoolPtr pool,
C
Cole Robinson 已提交
4082
                            int *autostart) {
4083 4084
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
4085
    int ret = -1;
4086

4087
    testDriverLock(privconn);
4088 4089
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
4090
    testDriverUnlock(privconn);
4091 4092

    if (privpool == NULL) {
4093
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4094
        goto cleanup;
4095
    }
C
Cole Robinson 已提交
4096 4097 4098 4099 4100 4101

    if (!privpool->configFile) {
        *autostart = 0;
    } else {
        *autostart = privpool->autostart;
    }
4102
    ret = 0;
C
Cole Robinson 已提交
4103

4104
cleanup:
4105 4106
    if (privpool)
        virStoragePoolObjUnlock(privpool);
4107
    return ret;
C
Cole Robinson 已提交
4108 4109 4110
}

static int
4111
testStoragePoolSetAutostart(virStoragePoolPtr pool,
C
Cole Robinson 已提交
4112
                            int autostart) {
4113 4114
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
4115
    int ret = -1;
4116

4117
    testDriverLock(privconn);
4118 4119
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
4120
    testDriverUnlock(privconn);
4121 4122

    if (privpool == NULL) {
4123
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4124
        goto cleanup;
4125
    }
C
Cole Robinson 已提交
4126 4127

    if (!privpool->configFile) {
4128
        testError(VIR_ERR_INVALID_ARG,
C
Cole Robinson 已提交
4129
                  "%s", _("pool has no config file"));
4130
        goto cleanup;
C
Cole Robinson 已提交
4131 4132 4133 4134
    }

    autostart = (autostart != 0);
    privpool->autostart = autostart;
4135 4136 4137
    ret = 0;

cleanup:
4138 4139
    if (privpool)
        virStoragePoolObjUnlock(privpool);
4140
    return ret;
C
Cole Robinson 已提交
4141 4142 4143 4144
}


static int
4145 4146 4147
testStoragePoolNumVolumes(virStoragePoolPtr pool) {
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
4148
    int ret = -1;
4149

4150
    testDriverLock(privconn);
4151 4152
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
4153
    testDriverUnlock(privconn);
4154 4155

    if (privpool == NULL) {
4156
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4157
        goto cleanup;
4158 4159 4160
    }

    if (!virStoragePoolObjIsActive(privpool)) {
4161
        testError(VIR_ERR_INTERNAL_ERROR,
4162
                  _("storage pool '%s' is not active"), pool->name);
4163
        goto cleanup;
4164
    }
C
Cole Robinson 已提交
4165

4166 4167 4168
    ret = privpool->volumes.count;

cleanup:
4169 4170
    if (privpool)
        virStoragePoolObjUnlock(privpool);
4171
    return ret;
C
Cole Robinson 已提交
4172 4173 4174
}

static int
4175
testStoragePoolListVolumes(virStoragePoolPtr pool,
C
Cole Robinson 已提交
4176 4177
                           char **const names,
                           int maxnames) {
4178 4179
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
C
Cole Robinson 已提交
4180 4181
    int i = 0, n = 0;

4182
    memset(names, 0, maxnames * sizeof(*names));
4183 4184

    testDriverLock(privconn);
4185 4186
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
4187
    testDriverUnlock(privconn);
4188 4189

    if (privpool == NULL) {
4190
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4191
        goto cleanup;
4192 4193 4194 4195
    }


    if (!virStoragePoolObjIsActive(privpool)) {
4196
        testError(VIR_ERR_INTERNAL_ERROR,
4197
                  _("storage pool '%s' is not active"), pool->name);
4198
        goto cleanup;
4199 4200
    }

C
Cole Robinson 已提交
4201 4202
    for (i = 0 ; i < privpool->volumes.count && n < maxnames ; i++) {
        if ((names[n++] = strdup(privpool->volumes.objs[i]->name)) == NULL) {
4203
            virReportOOMError();
C
Cole Robinson 已提交
4204 4205 4206 4207
            goto cleanup;
        }
    }

4208
    virStoragePoolObjUnlock(privpool);
C
Cole Robinson 已提交
4209 4210 4211 4212 4213 4214
    return n;

 cleanup:
    for (n = 0 ; n < maxnames ; n++)
        VIR_FREE(names[i]);

4215
    memset(names, 0, maxnames * sizeof(*names));
4216 4217
    if (privpool)
        virStoragePoolObjUnlock(privpool);
C
Cole Robinson 已提交
4218 4219 4220 4221 4222
    return -1;
}


static virStorageVolPtr
4223
testStorageVolumeLookupByName(virStoragePoolPtr pool,
C
Cole Robinson 已提交
4224
                              const char *name ATTRIBUTE_UNUSED) {
4225 4226 4227
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
    virStorageVolDefPtr privvol;
4228
    virStorageVolPtr ret = NULL;
4229

4230
    testDriverLock(privconn);
4231 4232
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
4233
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
4234

4235
    if (privpool == NULL) {
4236
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4237
        goto cleanup;
4238 4239 4240 4241
    }


    if (!virStoragePoolObjIsActive(privpool)) {
4242
        testError(VIR_ERR_INTERNAL_ERROR,
4243
                  _("storage pool '%s' is not active"), pool->name);
4244
        goto cleanup;
4245 4246 4247 4248 4249
    }

    privvol = virStorageVolDefFindByName(privpool, name);

    if (!privvol) {
4250
        testError(VIR_ERR_INVALID_STORAGE_VOL,
C
Cole Robinson 已提交
4251
                  _("no storage vol with matching name '%s'"), name);
4252
        goto cleanup;
C
Cole Robinson 已提交
4253 4254
    }

4255 4256 4257 4258
    ret = virGetStorageVol(pool->conn, privpool->def->name,
                           privvol->name, privvol->key);

cleanup:
4259 4260
    if (privpool)
        virStoragePoolObjUnlock(privpool);
4261
    return ret;
C
Cole Robinson 已提交
4262 4263 4264 4265 4266 4267
}


static virStorageVolPtr
testStorageVolumeLookupByKey(virConnectPtr conn,
                             const char *key) {
4268
    testConnPtr privconn = conn->privateData;
C
Cole Robinson 已提交
4269
    unsigned int i;
4270
    virStorageVolPtr ret = NULL;
C
Cole Robinson 已提交
4271

4272
    testDriverLock(privconn);
C
Cole Robinson 已提交
4273
    for (i = 0 ; i < privconn->pools.count ; i++) {
4274
        virStoragePoolObjLock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
4275
        if (virStoragePoolObjIsActive(privconn->pools.objs[i])) {
4276
            virStorageVolDefPtr privvol =
C
Cole Robinson 已提交
4277 4278
                virStorageVolDefFindByKey(privconn->pools.objs[i], key);

4279 4280 4281 4282 4283
            if (privvol) {
                ret = virGetStorageVol(conn,
                                       privconn->pools.objs[i]->def->name,
                                       privvol->name,
                                       privvol->key);
4284
                virStoragePoolObjUnlock(privconn->pools.objs[i]);
4285 4286
                break;
            }
C
Cole Robinson 已提交
4287
        }
4288
        virStoragePoolObjUnlock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
4289
    }
4290
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
4291

4292
    if (!ret)
4293
        testError(VIR_ERR_INVALID_STORAGE_VOL,
4294 4295 4296
                  _("no storage vol with matching key '%s'"), key);

    return ret;
C
Cole Robinson 已提交
4297 4298 4299 4300 4301
}

static virStorageVolPtr
testStorageVolumeLookupByPath(virConnectPtr conn,
                              const char *path) {
4302
    testConnPtr privconn = conn->privateData;
C
Cole Robinson 已提交
4303
    unsigned int i;
4304
    virStorageVolPtr ret = NULL;
C
Cole Robinson 已提交
4305

4306
    testDriverLock(privconn);
C
Cole Robinson 已提交
4307
    for (i = 0 ; i < privconn->pools.count ; i++) {
4308
        virStoragePoolObjLock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
4309
        if (virStoragePoolObjIsActive(privconn->pools.objs[i])) {
4310
            virStorageVolDefPtr privvol =
C
Cole Robinson 已提交
4311 4312
                virStorageVolDefFindByPath(privconn->pools.objs[i], path);

4313 4314 4315 4316 4317
            if (privvol) {
                ret = virGetStorageVol(conn,
                                       privconn->pools.objs[i]->def->name,
                                       privvol->name,
                                       privvol->key);
4318
                virStoragePoolObjUnlock(privconn->pools.objs[i]);
4319 4320
                break;
            }
C
Cole Robinson 已提交
4321
        }
4322
        virStoragePoolObjUnlock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
4323
    }
4324
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
4325

4326
    if (!ret)
4327
        testError(VIR_ERR_INVALID_STORAGE_VOL,
4328 4329 4330
                  _("no storage vol with matching path '%s'"), path);

    return ret;
C
Cole Robinson 已提交
4331 4332 4333
}

static virStorageVolPtr
4334
testStorageVolumeCreateXML(virStoragePoolPtr pool,
C
Cole Robinson 已提交
4335 4336
                           const char *xmldesc,
                           unsigned int flags ATTRIBUTE_UNUSED) {
4337 4338
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
4339 4340
    virStorageVolDefPtr privvol = NULL;
    virStorageVolPtr ret = NULL;
4341

4342
    testDriverLock(privconn);
4343 4344
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
4345
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
4346

4347
    if (privpool == NULL) {
4348
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4349
        goto cleanup;
4350 4351 4352
    }

    if (!virStoragePoolObjIsActive(privpool)) {
4353
        testError(VIR_ERR_INTERNAL_ERROR,
4354
                  _("storage pool '%s' is not active"), pool->name);
4355
        goto cleanup;
4356
    }
C
Cole Robinson 已提交
4357

4358
    privvol = virStorageVolDefParseString(privpool->def, xmldesc);
4359
    if (privvol == NULL)
4360
        goto cleanup;
4361 4362

    if (virStorageVolDefFindByName(privpool, privvol->name)) {
4363
        testError(VIR_ERR_INVALID_STORAGE_VOL,
C
Cole Robinson 已提交
4364
                  "%s", _("storage vol already exists"));
4365
        goto cleanup;
C
Cole Robinson 已提交
4366 4367 4368
    }

    /* Make sure enough space */
4369
    if ((privpool->def->allocation + privvol->allocation) >
C
Cole Robinson 已提交
4370
         privpool->def->capacity) {
4371
        testError(VIR_ERR_INTERNAL_ERROR,
C
Cole Robinson 已提交
4372
                  _("Not enough free space in pool for volume '%s'"),
4373
                  privvol->name);
4374
        goto cleanup;
C
Cole Robinson 已提交
4375 4376 4377 4378
    }

    if (VIR_REALLOC_N(privpool->volumes.objs,
                      privpool->volumes.count+1) < 0) {
4379
        virReportOOMError();
4380
        goto cleanup;
C
Cole Robinson 已提交
4381 4382
    }

4383 4384 4385
    if (virAsprintf(&privvol->target.path, "%s/%s",
                    privpool->def->target.path,
                    privvol->name) == -1) {
4386
        virReportOOMError();
4387
        goto cleanup;
C
Cole Robinson 已提交
4388 4389
    }

4390 4391
    privvol->key = strdup(privvol->target.path);
    if (privvol->key == NULL) {
4392
        virReportOOMError();
4393
        goto cleanup;
C
Cole Robinson 已提交
4394 4395
    }

4396
    privpool->def->allocation += privvol->allocation;
C
Cole Robinson 已提交
4397 4398 4399
    privpool->def->available = (privpool->def->capacity -
                                privpool->def->allocation);

4400
    privpool->volumes.objs[privpool->volumes.count++] = privvol;
C
Cole Robinson 已提交
4401

4402 4403
    ret = virGetStorageVol(pool->conn, privpool->def->name,
                           privvol->name, privvol->key);
4404
    privvol = NULL;
4405 4406 4407

cleanup:
    virStorageVolDefFree(privvol);
4408 4409
    if (privpool)
        virStoragePoolObjUnlock(privpool);
4410
    return ret;
C
Cole Robinson 已提交
4411 4412
}

4413 4414 4415 4416 4417 4418 4419 4420 4421 4422 4423 4424 4425 4426 4427 4428
static virStorageVolPtr
testStorageVolumeCreateXMLFrom(virStoragePoolPtr pool,
                               const char *xmldesc,
                               virStorageVolPtr clonevol,
                               unsigned int flags ATTRIBUTE_UNUSED) {
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
    virStorageVolDefPtr privvol = NULL, origvol = NULL;
    virStorageVolPtr ret = NULL;

    testDriverLock(privconn);
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
    testDriverUnlock(privconn);

    if (privpool == NULL) {
4429
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4430 4431 4432 4433
        goto cleanup;
    }

    if (!virStoragePoolObjIsActive(privpool)) {
4434
        testError(VIR_ERR_INTERNAL_ERROR,
4435 4436 4437 4438
                  _("storage pool '%s' is not active"), pool->name);
        goto cleanup;
    }

4439
    privvol = virStorageVolDefParseString(privpool->def, xmldesc);
4440 4441 4442 4443
    if (privvol == NULL)
        goto cleanup;

    if (virStorageVolDefFindByName(privpool, privvol->name)) {
4444
        testError(VIR_ERR_INVALID_STORAGE_VOL,
4445 4446 4447 4448 4449 4450
                  "%s", _("storage vol already exists"));
        goto cleanup;
    }

    origvol = virStorageVolDefFindByName(privpool, clonevol->name);
    if (!origvol) {
4451
        testError(VIR_ERR_INVALID_STORAGE_VOL,
4452 4453 4454 4455 4456 4457 4458 4459
                  _("no storage vol with matching name '%s'"),
                  clonevol->name);
        goto cleanup;
    }

    /* Make sure enough space */
    if ((privpool->def->allocation + privvol->allocation) >
         privpool->def->capacity) {
4460
        testError(VIR_ERR_INTERNAL_ERROR,
4461 4462 4463 4464 4465 4466 4467 4468 4469
                  _("Not enough free space in pool for volume '%s'"),
                  privvol->name);
        goto cleanup;
    }
    privpool->def->available = (privpool->def->capacity -
                                privpool->def->allocation);

    if (VIR_REALLOC_N(privpool->volumes.objs,
                      privpool->volumes.count+1) < 0) {
4470
        virReportOOMError();
4471 4472 4473
        goto cleanup;
    }

4474 4475 4476
    if (virAsprintf(&privvol->target.path, "%s/%s",
                    privpool->def->target.path,
                    privvol->name) == -1) {
4477
        virReportOOMError();
4478 4479 4480 4481 4482
        goto cleanup;
    }

    privvol->key = strdup(privvol->target.path);
    if (privvol->key == NULL) {
4483
        virReportOOMError();
4484 4485 4486 4487 4488 4489 4490 4491 4492 4493 4494 4495 4496 4497 4498 4499 4500 4501 4502 4503
        goto cleanup;
    }

    privpool->def->allocation += privvol->allocation;
    privpool->def->available = (privpool->def->capacity -
                                privpool->def->allocation);

    privpool->volumes.objs[privpool->volumes.count++] = privvol;

    ret = virGetStorageVol(pool->conn, privpool->def->name,
                           privvol->name, privvol->key);
    privvol = NULL;

cleanup:
    virStorageVolDefFree(privvol);
    if (privpool)
        virStoragePoolObjUnlock(privpool);
    return ret;
}

C
Cole Robinson 已提交
4504
static int
4505
testStorageVolumeDelete(virStorageVolPtr vol,
C
Cole Robinson 已提交
4506
                        unsigned int flags ATTRIBUTE_UNUSED) {
4507 4508 4509
    testConnPtr privconn = vol->conn->privateData;
    virStoragePoolObjPtr privpool;
    virStorageVolDefPtr privvol;
C
Cole Robinson 已提交
4510
    int i;
4511
    int ret = -1;
C
Cole Robinson 已提交
4512

4513
    testDriverLock(privconn);
4514 4515
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           vol->pool);
4516
    testDriverUnlock(privconn);
4517 4518

    if (privpool == NULL) {
4519
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4520
        goto cleanup;
4521 4522 4523 4524 4525 4526
    }


    privvol = virStorageVolDefFindByName(privpool, vol->name);

    if (privvol == NULL) {
4527
        testError(VIR_ERR_INVALID_STORAGE_VOL,
4528 4529
                  _("no storage vol with matching name '%s'"),
                  vol->name);
4530
        goto cleanup;
4531 4532 4533
    }

    if (!virStoragePoolObjIsActive(privpool)) {
4534
        testError(VIR_ERR_INTERNAL_ERROR,
4535
                  _("storage pool '%s' is not active"), vol->pool);
4536
        goto cleanup;
4537 4538 4539
    }


C
Cole Robinson 已提交
4540 4541 4542 4543 4544 4545 4546 4547 4548 4549 4550 4551 4552 4553 4554 4555 4556 4557 4558 4559 4560 4561 4562
    privpool->def->allocation -= privvol->allocation;
    privpool->def->available = (privpool->def->capacity -
                                privpool->def->allocation);

    for (i = 0 ; i < privpool->volumes.count ; i++) {
        if (privpool->volumes.objs[i] == privvol) {
            virStorageVolDefFree(privvol);

            if (i < (privpool->volumes.count - 1))
                memmove(privpool->volumes.objs + i,
                        privpool->volumes.objs + i + 1,
                        sizeof(*(privpool->volumes.objs)) *
                                (privpool->volumes.count - (i + 1)));

            if (VIR_REALLOC_N(privpool->volumes.objs,
                              privpool->volumes.count - 1) < 0) {
                ; /* Failure to reduce memory allocation isn't fatal */
            }
            privpool->volumes.count--;

            break;
        }
    }
4563
    ret = 0;
C
Cole Robinson 已提交
4564

4565
cleanup:
4566 4567
    if (privpool)
        virStoragePoolObjUnlock(privpool);
4568
    return ret;
C
Cole Robinson 已提交
4569 4570 4571 4572 4573 4574 4575 4576 4577 4578 4579 4580 4581 4582 4583 4584
}


static int testStorageVolumeTypeForPool(int pooltype) {

    switch(pooltype) {
        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;
    }
}

static int
4585
testStorageVolumeGetInfo(virStorageVolPtr vol,
C
Cole Robinson 已提交
4586
                         virStorageVolInfoPtr info) {
4587 4588 4589
    testConnPtr privconn = vol->conn->privateData;
    virStoragePoolObjPtr privpool;
    virStorageVolDefPtr privvol;
4590
    int ret = -1;
4591

4592
    testDriverLock(privconn);
4593 4594
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           vol->pool);
4595
    testDriverUnlock(privconn);
4596 4597

    if (privpool == NULL) {
4598
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4599
        goto cleanup;
4600 4601 4602 4603 4604
    }

    privvol = virStorageVolDefFindByName(privpool, vol->name);

    if (privvol == NULL) {
4605
        testError(VIR_ERR_INVALID_STORAGE_VOL,
4606 4607
                  _("no storage vol with matching name '%s'"),
                  vol->name);
4608
        goto cleanup;
4609 4610 4611
    }

    if (!virStoragePoolObjIsActive(privpool)) {
4612
        testError(VIR_ERR_INTERNAL_ERROR,
4613
                  _("storage pool '%s' is not active"), vol->pool);
4614
        goto cleanup;
4615
    }
C
Cole Robinson 已提交
4616 4617 4618 4619 4620

    memset(info, 0, sizeof(*info));
    info->type = testStorageVolumeTypeForPool(privpool->def->type);
    info->capacity = privvol->capacity;
    info->allocation = privvol->allocation;
4621
    ret = 0;
C
Cole Robinson 已提交
4622

4623
cleanup:
4624 4625
    if (privpool)
        virStoragePoolObjUnlock(privpool);
4626
    return ret;
C
Cole Robinson 已提交
4627 4628 4629
}

static char *
4630
testStorageVolumeGetXMLDesc(virStorageVolPtr vol,
C
Cole Robinson 已提交
4631
                            unsigned int flags ATTRIBUTE_UNUSED) {
4632 4633 4634
    testConnPtr privconn = vol->conn->privateData;
    virStoragePoolObjPtr privpool;
    virStorageVolDefPtr privvol;
4635
    char *ret = NULL;
4636

4637
    testDriverLock(privconn);
4638 4639
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           vol->pool);
4640
    testDriverUnlock(privconn);
4641 4642

    if (privpool == NULL) {
4643
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4644
        goto cleanup;
4645 4646 4647 4648 4649
    }

    privvol = virStorageVolDefFindByName(privpool, vol->name);

    if (privvol == NULL) {
4650
        testError(VIR_ERR_INVALID_STORAGE_VOL,
4651 4652
                  _("no storage vol with matching name '%s'"),
                  vol->name);
4653
        goto cleanup;
4654
    }
C
Cole Robinson 已提交
4655

4656
    if (!virStoragePoolObjIsActive(privpool)) {
4657
        testError(VIR_ERR_INTERNAL_ERROR,
4658
                  _("storage pool '%s' is not active"), vol->pool);
4659
        goto cleanup;
4660 4661
    }

4662
    ret = virStorageVolDefFormat(privpool->def, privvol);
4663 4664

cleanup:
4665 4666
    if (privpool)
        virStoragePoolObjUnlock(privpool);
4667
    return ret;
C
Cole Robinson 已提交
4668 4669 4670
}

static char *
4671 4672 4673 4674
testStorageVolumeGetPath(virStorageVolPtr vol) {
    testConnPtr privconn = vol->conn->privateData;
    virStoragePoolObjPtr privpool;
    virStorageVolDefPtr privvol;
4675
    char *ret = NULL;
4676

4677
    testDriverLock(privconn);
4678 4679
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           vol->pool);
4680
    testDriverUnlock(privconn);
4681 4682

    if (privpool == NULL) {
4683
        testError(VIR_ERR_INVALID_ARG, __FUNCTION__);
4684
        goto cleanup;
4685 4686 4687 4688 4689
    }

    privvol = virStorageVolDefFindByName(privpool, vol->name);

    if (privvol == NULL) {
4690
        testError(VIR_ERR_INVALID_STORAGE_VOL,
4691 4692
                  _("no storage vol with matching name '%s'"),
                  vol->name);
4693
        goto cleanup;
4694 4695 4696
    }

    if (!virStoragePoolObjIsActive(privpool)) {
4697
        testError(VIR_ERR_INTERNAL_ERROR,
4698
                  _("storage pool '%s' is not active"), vol->pool);
4699
        goto cleanup;
4700 4701
    }

C
Cole Robinson 已提交
4702
    ret = strdup(privvol->target.path);
4703
    if (ret == NULL)
4704
        virReportOOMError();
4705 4706

cleanup:
4707 4708
    if (privpool)
        virStoragePoolObjUnlock(privpool);
C
Cole Robinson 已提交
4709 4710 4711
    return ret;
}

4712

4713
/* Node device implementations */
4714 4715 4716 4717 4718 4719 4720 4721 4722 4723 4724 4725 4726 4727 4728
static virDrvOpenStatus testDevMonOpen(virConnectPtr conn,
                                       virConnectAuthPtr auth ATTRIBUTE_UNUSED,
                                       int flags ATTRIBUTE_UNUSED) {
    if (STRNEQ(conn->driver->name, "Test"))
        return VIR_DRV_OPEN_DECLINED;

    conn->devMonPrivateData = conn->privateData;
    return VIR_DRV_OPEN_SUCCESS;
}

static int testDevMonClose(virConnectPtr conn) {
    conn->devMonPrivateData = NULL;
    return 0;
}

4729 4730 4731 4732 4733 4734 4735 4736 4737 4738 4739 4740 4741 4742 4743 4744 4745 4746 4747 4748 4749 4750 4751 4752 4753 4754 4755 4756 4757 4758 4759 4760 4761 4762 4763 4764 4765 4766 4767 4768 4769 4770 4771 4772 4773 4774 4775 4776 4777 4778 4779 4780 4781 4782 4783 4784 4785 4786 4787 4788 4789 4790 4791 4792 4793 4794
static int
testNodeNumOfDevices(virConnectPtr conn,
                     const char *cap,
                     unsigned int flags ATTRIBUTE_UNUSED)
{
    testConnPtr driver = conn->privateData;
    int ndevs = 0;
    unsigned int i;

    testDriverLock(driver);
    for (i = 0; i < driver->devs.count; i++)
        if ((cap == NULL) ||
            virNodeDeviceHasCap(driver->devs.objs[i], cap))
            ++ndevs;
    testDriverUnlock(driver);

    return ndevs;
}

static int
testNodeListDevices(virConnectPtr conn,
                    const char *cap,
                    char **const names,
                    int maxnames,
                    unsigned int flags ATTRIBUTE_UNUSED)
{
    testConnPtr driver = conn->privateData;
    int ndevs = 0;
    unsigned int i;

    testDriverLock(driver);
    for (i = 0; i < driver->devs.count && ndevs < maxnames; i++) {
        virNodeDeviceObjLock(driver->devs.objs[i]);
        if (cap == NULL ||
            virNodeDeviceHasCap(driver->devs.objs[i], cap)) {
            if ((names[ndevs++] = strdup(driver->devs.objs[i]->def->name)) == NULL) {
                virNodeDeviceObjUnlock(driver->devs.objs[i]);
                goto failure;
            }
        }
        virNodeDeviceObjUnlock(driver->devs.objs[i]);
    }
    testDriverUnlock(driver);

    return ndevs;

 failure:
    testDriverUnlock(driver);
    --ndevs;
    while (--ndevs >= 0)
        VIR_FREE(names[ndevs]);
    return -1;
}

static virNodeDevicePtr
testNodeDeviceLookupByName(virConnectPtr conn, const char *name)
{
    testConnPtr driver = conn->privateData;
    virNodeDeviceObjPtr obj;
    virNodeDevicePtr ret = NULL;

    testDriverLock(driver);
    obj = virNodeDeviceFindByName(&driver->devs, name);
    testDriverUnlock(driver);

    if (!obj) {
4795
        virNodeDeviceReportError(VIR_ERR_NO_NODE_DEVICE, NULL);
4796 4797 4798 4799 4800 4801 4802 4803 4804 4805 4806 4807 4808 4809 4810 4811 4812 4813 4814 4815 4816 4817 4818 4819
        goto cleanup;
    }

    ret = virGetNodeDevice(conn, name);

cleanup:
    if (obj)
        virNodeDeviceObjUnlock(obj);
    return ret;
}

static char *
testNodeDeviceDumpXML(virNodeDevicePtr dev,
                      unsigned int flags ATTRIBUTE_UNUSED)
{
    testConnPtr driver = dev->conn->privateData;
    virNodeDeviceObjPtr obj;
    char *ret = NULL;

    testDriverLock(driver);
    obj = virNodeDeviceFindByName(&driver->devs, dev->name);
    testDriverUnlock(driver);

    if (!obj) {
4820 4821
        virNodeDeviceReportError(VIR_ERR_NO_NODE_DEVICE,
                                 _("no node device with matching name '%s'"),
4822 4823 4824 4825
                                 dev->name);
        goto cleanup;
    }

4826
    ret = virNodeDeviceDefFormat(obj->def);
4827 4828 4829 4830 4831 4832 4833 4834 4835 4836 4837 4838 4839 4840 4841 4842 4843 4844 4845

cleanup:
    if (obj)
        virNodeDeviceObjUnlock(obj);
    return ret;
}

static char *
testNodeDeviceGetParent(virNodeDevicePtr dev)
{
    testConnPtr driver = dev->conn->privateData;
    virNodeDeviceObjPtr obj;
    char *ret = NULL;

    testDriverLock(driver);
    obj = virNodeDeviceFindByName(&driver->devs, dev->name);
    testDriverUnlock(driver);

    if (!obj) {
4846
        virNodeDeviceReportError(VIR_ERR_NO_NODE_DEVICE,
4847 4848 4849 4850 4851 4852 4853 4854
                                _("no node device with matching name '%s'"),
                                 dev->name);
        goto cleanup;
    }

    if (obj->def->parent) {
        ret = strdup(obj->def->parent);
        if (!ret)
4855
            virReportOOMError();
4856
    } else {
4857
        virNodeDeviceReportError(VIR_ERR_INTERNAL_ERROR,
4858 4859 4860 4861 4862 4863 4864 4865 4866
                                 "%s", _("no parent for this device"));
    }

cleanup:
    if (obj)
        virNodeDeviceObjUnlock(obj);
    return ret;
}

4867

4868 4869 4870 4871 4872 4873 4874 4875 4876 4877 4878 4879 4880 4881
static int
testNodeDeviceNumOfCaps(virNodeDevicePtr dev)
{
    testConnPtr driver = dev->conn->privateData;
    virNodeDeviceObjPtr obj;
    virNodeDevCapsDefPtr caps;
    int ncaps = 0;
    int ret = -1;

    testDriverLock(driver);
    obj = virNodeDeviceFindByName(&driver->devs, dev->name);
    testDriverUnlock(driver);

    if (!obj) {
4882 4883
        virNodeDeviceReportError(VIR_ERR_NO_NODE_DEVICE,
                                 _("no node device with matching name '%s'"),
4884 4885 4886 4887 4888 4889 4890 4891 4892 4893 4894 4895 4896 4897 4898 4899 4900 4901 4902 4903 4904 4905 4906 4907 4908 4909 4910 4911 4912
                                 dev->name);
        goto cleanup;
    }

    for (caps = obj->def->caps; caps; caps = caps->next)
        ++ncaps;
    ret = ncaps;

cleanup:
    if (obj)
        virNodeDeviceObjUnlock(obj);
    return ret;
}


static int
testNodeDeviceListCaps(virNodeDevicePtr dev, char **const names, int maxnames)
{
    testConnPtr driver = dev->conn->privateData;
    virNodeDeviceObjPtr obj;
    virNodeDevCapsDefPtr caps;
    int ncaps = 0;
    int ret = -1;

    testDriverLock(driver);
    obj = virNodeDeviceFindByName(&driver->devs, dev->name);
    testDriverUnlock(driver);

    if (!obj) {
4913
        virNodeDeviceReportError(VIR_ERR_NO_NODE_DEVICE,
4914 4915 4916 4917 4918 4919 4920 4921 4922 4923 4924 4925 4926 4927 4928 4929 4930 4931 4932 4933 4934 4935 4936
                                _("no node device with matching name '%s'"),
                                 dev->name);
        goto cleanup;
    }

    for (caps = obj->def->caps; caps && ncaps < maxnames; caps = caps->next) {
        names[ncaps] = strdup(virNodeDevCapTypeToString(caps->type));
        if (names[ncaps++] == NULL)
            goto cleanup;
    }
    ret = ncaps;

cleanup:
    if (obj)
        virNodeDeviceObjUnlock(obj);
    if (ret == -1) {
        --ncaps;
        while (--ncaps >= 0)
            VIR_FREE(names[ncaps]);
    }
    return ret;
}

4937 4938 4939 4940 4941 4942 4943 4944 4945 4946 4947 4948 4949 4950 4951
static virNodeDevicePtr
testNodeDeviceCreateXML(virConnectPtr conn,
                        const char *xmlDesc,
                        unsigned int flags ATTRIBUTE_UNUSED)
{
    testConnPtr driver = conn->privateData;
    virNodeDeviceDefPtr def = NULL;
    virNodeDeviceObjPtr obj = NULL;
    char *wwnn = NULL, *wwpn = NULL;
    int parent_host = -1;
    virNodeDevicePtr dev = NULL;
    virNodeDevCapsDefPtr caps;

    testDriverLock(driver);

4952
    def = virNodeDeviceDefParseString(xmlDesc, CREATE_DEVICE);
4953 4954 4955 4956 4957
    if (def == NULL) {
        goto cleanup;
    }

    /* We run these next two simply for validation */
4958
    if (virNodeDeviceGetWWNs(def, &wwnn, &wwpn) == -1) {
4959 4960 4961
        goto cleanup;
    }

4962
    if (virNodeDeviceGetParentHost(&driver->devs,
4963 4964 4965 4966 4967 4968 4969 4970 4971 4972
                                   def->name,
                                   def->parent,
                                   &parent_host) == -1) {
        goto cleanup;
    }

    /* 'name' is supposed to be filled in by the node device backend, which
     * we don't have. Use WWPN instead. */
    VIR_FREE(def->name);
    if (!(def->name = strdup(wwpn))) {
4973
        virReportOOMError();
4974 4975 4976 4977 4978 4979 4980 4981 4982 4983 4984 4985 4986 4987 4988
        goto cleanup;
    }

    /* Fill in a random 'host' value, since this would also come from
     * the backend */
    caps = def->caps;
    while (caps) {
        if (caps->type != VIR_NODE_DEV_CAP_SCSI_HOST)
            continue;

        caps->data.scsi_host.host = virRandom(1024);
        caps = caps->next;
    }


4989
    if (!(obj = virNodeDeviceAssignDef(&driver->devs, def))) {
4990 4991 4992 4993 4994 4995 4996 4997
        goto cleanup;
    }
    virNodeDeviceObjUnlock(obj);

    dev = virGetNodeDevice(conn, def->name);
    def = NULL;
cleanup:
    testDriverUnlock(driver);
4998
    virNodeDeviceDefFree(def);
4999 5000 5001 5002 5003 5004 5005 5006 5007 5008 5009 5010 5011 5012 5013 5014 5015 5016 5017
    VIR_FREE(wwnn);
    VIR_FREE(wwpn);
    return dev;
}

static int
testNodeDeviceDestroy(virNodeDevicePtr dev)
{
    int ret = 0;
    testConnPtr driver = dev->conn->privateData;
    virNodeDeviceObjPtr obj = NULL;
    char *parent_name = NULL, *wwnn = NULL, *wwpn = NULL;
    int parent_host = -1;

    testDriverLock(driver);
    obj = virNodeDeviceFindByName(&driver->devs, dev->name);
    testDriverUnlock(driver);

    if (!obj) {
5018
        virNodeDeviceReportError(VIR_ERR_NO_NODE_DEVICE, NULL);
5019 5020 5021
        goto out;
    }

5022
    if (virNodeDeviceGetWWNs(obj->def, &wwnn, &wwpn) == -1) {
5023 5024 5025 5026 5027
        goto out;
    }

    parent_name = strdup(obj->def->parent);
    if (parent_name == NULL) {
5028
        virReportOOMError();
5029 5030 5031 5032 5033 5034 5035 5036 5037 5038
        goto out;
    }

    /* virNodeDeviceGetParentHost will cause the device object's lock to be
     * taken, so we have to dup the parent's name and drop the lock
     * before calling it.  We don't need the reference to the object
     * any more once we have the parent's name.  */
    virNodeDeviceObjUnlock(obj);

    /* We do this just for basic validation */
5039
    if (virNodeDeviceGetParentHost(&driver->devs,
5040 5041 5042 5043 5044 5045 5046 5047 5048 5049 5050 5051 5052 5053 5054 5055 5056 5057 5058
                                   dev->name,
                                   parent_name,
                                   &parent_host) == -1) {
        obj = NULL;
        goto out;
    }

    virNodeDeviceObjLock(obj);
    virNodeDeviceObjRemove(&driver->devs, obj);

out:
    if (obj)
        virNodeDeviceObjUnlock(obj);
    VIR_FREE(parent_name);
    VIR_FREE(wwnn);
    VIR_FREE(wwpn);
    return ret;
}

5059 5060

/* Domain event implementations */
5061
static int
5062 5063 5064 5065
testDomainEventRegister(virConnectPtr conn,
                        virConnectDomainEventCallback callback,
                        void *opaque,
                        virFreeCallback freecb)
5066 5067 5068 5069 5070 5071 5072 5073 5074 5075 5076 5077
{
    testConnPtr driver = conn->privateData;
    int ret;

    testDriverLock(driver);
    ret = virDomainEventCallbackListAdd(conn, driver->domainEventCallbacks,
                                        callback, opaque, freecb);
    testDriverUnlock(driver);

    return ret;
}

5078

5079
static int
5080 5081
testDomainEventDeregister(virConnectPtr conn,
                          virConnectDomainEventCallback callback)
5082 5083 5084 5085 5086 5087 5088 5089 5090 5091 5092 5093 5094 5095 5096 5097
{
    testConnPtr driver = conn->privateData;
    int ret;

    testDriverLock(driver);
    if (driver->domainEventDispatching)
        ret = virDomainEventCallbackListMarkDelete(conn, driver->domainEventCallbacks,
                                                   callback);
    else
        ret = virDomainEventCallbackListRemove(conn, driver->domainEventCallbacks,
                                               callback);
    testDriverUnlock(driver);

    return ret;
}

5098 5099 5100 5101 5102 5103 5104 5105 5106 5107 5108 5109 5110 5111 5112 5113 5114 5115 5116 5117 5118 5119 5120 5121 5122 5123 5124 5125 5126 5127 5128 5129 5130 5131 5132 5133 5134 5135 5136 5137 5138

static int
testDomainEventRegisterAny(virConnectPtr conn,
                           virDomainPtr dom,
                           int eventID,
                           virConnectDomainEventGenericCallback callback,
                           void *opaque,
                           virFreeCallback freecb)
{
    testConnPtr driver = conn->privateData;
    int ret;

    testDriverLock(driver);
    ret = virDomainEventCallbackListAddID(conn, driver->domainEventCallbacks,
                                          dom, eventID,
                                          callback, opaque, freecb);
    testDriverUnlock(driver);

    return ret;
}

static int
testDomainEventDeregisterAny(virConnectPtr conn,
                             int callbackID)
{
    testConnPtr driver = conn->privateData;
    int ret;

    testDriverLock(driver);
    if (driver->domainEventDispatching)
        ret = virDomainEventCallbackListMarkDeleteID(conn, driver->domainEventCallbacks,
                                                     callbackID);
    else
        ret = virDomainEventCallbackListRemoveID(conn, driver->domainEventCallbacks,
                                                 callbackID);
    testDriverUnlock(driver);

    return ret;
}


5139 5140
static void testDomainEventDispatchFunc(virConnectPtr conn,
                                        virDomainEventPtr event,
5141
                                        virConnectDomainEventGenericCallback cb,
5142 5143 5144 5145 5146 5147 5148 5149 5150 5151 5152 5153 5154 5155 5156 5157 5158 5159 5160 5161 5162 5163 5164 5165 5166 5167 5168 5169 5170 5171 5172 5173 5174 5175 5176 5177 5178 5179 5180 5181 5182 5183 5184 5185 5186 5187 5188 5189 5190 5191 5192 5193 5194 5195 5196 5197 5198
                                        void *cbopaque,
                                        void *opaque)
{
    testConnPtr driver = opaque;

    /* Drop the lock whle dispatching, for sake of re-entrancy */
    testDriverUnlock(driver);
    virDomainEventDispatchDefaultFunc(conn, event, cb, cbopaque, NULL);
    testDriverLock(driver);
}

static void testDomainEventFlush(int timer ATTRIBUTE_UNUSED, void *opaque)
{
    testConnPtr driver = opaque;
    virDomainEventQueue tempQueue;

    testDriverLock(driver);

    driver->domainEventDispatching = 1;

    /* Copy the queue, so we're reentrant safe */
    tempQueue.count = driver->domainEventQueue->count;
    tempQueue.events = driver->domainEventQueue->events;
    driver->domainEventQueue->count = 0;
    driver->domainEventQueue->events = NULL;

    virEventUpdateTimeout(driver->domainEventTimer, -1);
    virDomainEventQueueDispatch(&tempQueue,
                                driver->domainEventCallbacks,
                                testDomainEventDispatchFunc,
                                driver);

    /* Purge any deleted callbacks */
    virDomainEventCallbackListPurgeMarked(driver->domainEventCallbacks);

    driver->domainEventDispatching = 0;
    testDriverUnlock(driver);
}


/* driver must be locked before calling */
static void testDomainEventQueue(testConnPtr driver,
                                 virDomainEventPtr event)
{
    if (driver->domainEventTimer < 0) {
        virDomainEventFree(event);
        return;
    }

    if (virDomainEventQueuePush(driver->domainEventQueue,
                                event) < 0)
        virDomainEventFree(event);

    if (driver->domainEventQueue->count == 1)
        virEventUpdateTimeout(driver->domainEventTimer, 0);
}

5199 5200 5201 5202 5203 5204
static virDrvOpenStatus testSecretOpen(virConnectPtr conn,
                                       virConnectAuthPtr auth ATTRIBUTE_UNUSED,
                                       int flags ATTRIBUTE_UNUSED) {
    if (STRNEQ(conn->driver->name, "Test"))
        return VIR_DRV_OPEN_DECLINED;

5205
    conn->secretPrivateData = conn->privateData;
5206 5207 5208 5209 5210 5211 5212
    return VIR_DRV_OPEN_SUCCESS;
}

static int testSecretClose(virConnectPtr conn) {
    conn->secretPrivateData = NULL;
    return 0;
}
5213

5214 5215 5216 5217 5218 5219 5220

static virDrvOpenStatus testNWFilterOpen(virConnectPtr conn,
                                         virConnectAuthPtr auth ATTRIBUTE_UNUSED,
                                         int flags ATTRIBUTE_UNUSED) {
    if (STRNEQ(conn->driver->name, "Test"))
        return VIR_DRV_OPEN_DECLINED;

5221
    conn->nwfilterPrivateData = conn->privateData;
5222 5223 5224 5225 5226 5227 5228 5229
    return VIR_DRV_OPEN_SUCCESS;
}

static int testNWFilterClose(virConnectPtr conn) {
    conn->nwfilterPrivateData = NULL;
    return 0;
}

5230 5231 5232 5233 5234
static virDriver testDriver = {
    VIR_DRV_TEST,
    "Test",
    testOpen, /* open */
    testClose, /* close */
5235
    NULL, /* supports_feature */
5236 5237
    NULL, /* type */
    testGetVersion, /* version */
5238
    NULL, /* libvirtVersion (impl. in libvirt.c) */
5239
    virGetHostname, /* getHostname */
5240 5241 5242 5243 5244
    testGetMaxVCPUs, /* getMaxVcpus */
    testNodeGetInfo, /* nodeGetInfo */
    testGetCapabilities, /* getCapabilities */
    testListDomains, /* listDomains */
    testNumOfDomains, /* numOfDomains */
5245
    testDomainCreateXML, /* domainCreateXML */
5246 5247 5248 5249 5250 5251 5252 5253 5254 5255 5256 5257 5258 5259 5260 5261 5262
    testLookupDomainByID, /* domainLookupByID */
    testLookupDomainByUUID, /* domainLookupByUUID */
    testLookupDomainByName, /* domainLookupByName */
    testPauseDomain, /* domainSuspend */
    testResumeDomain, /* domainResume */
    testShutdownDomain, /* domainShutdown */
    testRebootDomain, /* domainReboot */
    testDestroyDomain, /* domainDestroy */
    testGetOSType, /* domainGetOSType */
    testGetMaxMemory, /* domainGetMaxMemory */
    testSetMaxMemory, /* domainSetMaxMemory */
    testSetMemory, /* domainSetMemory */
    testGetDomainInfo, /* domainGetInfo */
    testDomainSave, /* domainSave */
    testDomainRestore, /* domainRestore */
    testDomainCoreDump, /* domainCoreDump */
    testSetVcpus, /* domainSetVcpus */
C
Cole Robinson 已提交
5263
    testDomainPinVcpu, /* domainPinVcpu */
C
Cole Robinson 已提交
5264
    testDomainGetVcpus, /* domainGetVcpus */
C
Cole Robinson 已提交
5265
    testDomainGetMaxVcpus, /* domainGetMaxVcpus */
5266 5267
    NULL, /* domainGetSecurityLabel */
    NULL, /* nodeGetSecurityModel */
5268
    testDomainDumpXML, /* domainDumpXML */
5269 5270
    NULL, /* domainXMLFromNative */
    NULL, /* domainXMLToNative */
5271 5272 5273
    testListDefinedDomains, /* listDefinedDomains */
    testNumOfDefinedDomains, /* numOfDefinedDomains */
    testDomainCreate, /* domainCreate */
5274
    testDomainCreateWithFlags, /* domainCreateWithFlags */
5275 5276 5277
    testDomainDefineXML, /* domainDefineXML */
    testDomainUndefine, /* domainUndefine */
    NULL, /* domainAttachDevice */
5278
    NULL, /* domainAttachDeviceFlags */
5279
    NULL, /* domainDetachDevice */
5280
    NULL, /* domainDetachDeviceFlags */
5281
    NULL, /* domainUpdateDeviceFlags */
5282 5283 5284 5285 5286
    testDomainGetAutostart, /* domainGetAutostart */
    testDomainSetAutostart, /* domainSetAutostart */
    testDomainGetSchedulerType, /* domainGetSchedulerType */
    testDomainGetSchedulerParams, /* domainGetSchedulerParameters */
    testDomainSetSchedulerParams, /* domainSetSchedulerParameters */
5287 5288 5289
    NULL, /* domainMigratePrepare */
    NULL, /* domainMigratePerform */
    NULL, /* domainMigrateFinish */
5290 5291
    testDomainBlockStats, /* domainBlockStats */
    testDomainInterfaceStats, /* domainInterfaceStats */
5292
    NULL, /* domainMemoryStats */
R
Richard W.M. Jones 已提交
5293
    NULL, /* domainBlockPeek */
R
Richard W.M. Jones 已提交
5294
    NULL, /* domainMemoryPeek */
5295
    NULL, /* domainGetBlockInfo */
5296
    testNodeGetCellsFreeMemory, /* nodeGetCellsFreeMemory */
5297
    NULL, /* getFreeMemory */
5298 5299
    testDomainEventRegister, /* domainEventRegister */
    testDomainEventDeregister, /* domainEventDeregister */
D
Daniel Veillard 已提交
5300 5301
    NULL, /* domainMigratePrepare2 */
    NULL, /* domainMigrateFinish2 */
5302
    NULL, /* nodeDeviceDettach */
5303 5304
    NULL, /* nodeDeviceReAttach */
    NULL, /* nodeDeviceReset */
C
Chris Lalancette 已提交
5305
    NULL, /* domainMigratePrepareTunnel */
5306 5307 5308 5309
    testIsEncrypted, /* isEncrypted */
    testIsSecure, /* isEncrypted */
    testDomainIsActive, /* domainIsActive */
    testDomainIsPersistent, /* domainIsPersistent */
J
Jiri Denemark 已提交
5310
    NULL, /* cpuCompare */
5311
    NULL, /* cpuBaseline */
5312
    NULL, /* domainGetJobInfo */
5313
    NULL, /* domainAbortJob */
5314
    NULL, /* domainMigrateSetMaxDowntime */
5315 5316
    testDomainEventRegisterAny, /* domainEventRegisterAny */
    testDomainEventDeregisterAny, /* domainEventDeregisterAny */
5317 5318 5319
    NULL, /* domainManagedSave */
    NULL, /* domainHasManagedSaveImage */
    NULL, /* domainManagedSaveRemove */
C
Chris Lalancette 已提交
5320 5321 5322 5323 5324 5325 5326 5327 5328
    NULL, /* domainSnapshotCreateXML */
    NULL, /* domainSnapshotDumpXML */
    NULL, /* domainSnapshotNum */
    NULL, /* domainSnapshotListNames */
    NULL, /* domainSnapshotLookupByName */
    NULL, /* domainHasCurrentSnapshot */
    NULL, /* domainSnapshotCurrent */
    NULL, /* domainRevertToSnapshot */
    NULL, /* domainSnapshotDelete */
C
Chris Lalancette 已提交
5329
    NULL, /* qemuDomainMonitorCommand */
5330 5331
    NULL, /* domainSetMemoryParameters */
    NULL, /* domainGetMemoryParameters */
5332 5333 5334 5335 5336 5337 5338 5339 5340 5341 5342 5343 5344 5345 5346 5347 5348 5349 5350 5351 5352
};

static virNetworkDriver testNetworkDriver = {
    "Test",
    testOpenNetwork, /* open */
    testCloseNetwork, /* close */
    testNumNetworks, /* numOfNetworks */
    testListNetworks, /* listNetworks */
    testNumDefinedNetworks, /* numOfDefinedNetworks */
    testListDefinedNetworks, /* listDefinedNetworks */
    testLookupNetworkByUUID, /* networkLookupByUUID */
    testLookupNetworkByName, /* networkLookupByName */
    testNetworkCreate, /* networkCreateXML */
    testNetworkDefine, /* networkDefineXML */
    testNetworkUndefine, /* networkUndefine */
    testNetworkStart, /* networkCreate */
    testNetworkDestroy, /* networkDestroy */
    testNetworkDumpXML, /* networkDumpXML */
    testNetworkGetBridgeName, /* networkGetBridgeName */
    testNetworkGetAutostart, /* networkGetAutostart */
    testNetworkSetAutostart, /* networkSetAutostart */
5353 5354
    testNetworkIsActive, /* networkIsActive */
    testNetworkIsPersistent, /* networkIsPersistent */
5355 5356
};

L
Laine Stump 已提交
5357 5358 5359 5360 5361 5362 5363 5364 5365 5366 5367 5368 5369 5370 5371
static virInterfaceDriver testInterfaceDriver = {
    "Test",                     /* name */
    testOpenInterface,          /* open */
    testCloseInterface,         /* close */
    testNumOfInterfaces,        /* numOfInterfaces */
    testListInterfaces,         /* listInterfaces */
    testNumOfDefinedInterfaces, /* numOfDefinedInterfaces */
    testListDefinedInterfaces,  /* listDefinedInterfaces */
    testLookupInterfaceByName,  /* interfaceLookupByName */
    testLookupInterfaceByMACString, /* interfaceLookupByMACString */
    testInterfaceGetXMLDesc,    /* interfaceGetXMLDesc */
    testInterfaceDefineXML,     /* interfaceDefineXML */
    testInterfaceUndefine,      /* interfaceUndefine */
    testInterfaceCreate,        /* interfaceCreate */
    testInterfaceDestroy,       /* interfaceDestroy */
5372
    testInterfaceIsActive,      /* interfaceIsActive */
L
Laine Stump 已提交
5373 5374 5375
};


5376 5377 5378 5379
static virStorageDriver testStorageDriver = {
    .name = "Test",
    .open = testStorageOpen,
    .close = testStorageClose,
C
Cole Robinson 已提交
5380 5381 5382 5383 5384 5385 5386 5387 5388 5389 5390 5391 5392 5393 5394 5395 5396 5397 5398 5399 5400 5401 5402 5403 5404 5405 5406 5407

    .numOfPools = testStorageNumPools,
    .listPools = testStorageListPools,
    .numOfDefinedPools = testStorageNumDefinedPools,
    .listDefinedPools = testStorageListDefinedPools,
    .findPoolSources = testStorageFindPoolSources,
    .poolLookupByName = testStoragePoolLookupByName,
    .poolLookupByUUID = testStoragePoolLookupByUUID,
    .poolLookupByVolume = testStoragePoolLookupByVolume,
    .poolCreateXML = testStoragePoolCreate,
    .poolDefineXML = testStoragePoolDefine,
    .poolBuild = testStoragePoolBuild,
    .poolUndefine = testStoragePoolUndefine,
    .poolCreate = testStoragePoolStart,
    .poolDestroy = testStoragePoolDestroy,
    .poolDelete = testStoragePoolDelete,
    .poolRefresh = testStoragePoolRefresh,
    .poolGetInfo = testStoragePoolGetInfo,
    .poolGetXMLDesc = testStoragePoolDumpXML,
    .poolGetAutostart = testStoragePoolGetAutostart,
    .poolSetAutostart = testStoragePoolSetAutostart,
    .poolNumOfVolumes = testStoragePoolNumVolumes,
    .poolListVolumes = testStoragePoolListVolumes,

    .volLookupByName = testStorageVolumeLookupByName,
    .volLookupByKey = testStorageVolumeLookupByKey,
    .volLookupByPath = testStorageVolumeLookupByPath,
    .volCreateXML = testStorageVolumeCreateXML,
5408
    .volCreateXMLFrom = testStorageVolumeCreateXMLFrom,
C
Cole Robinson 已提交
5409 5410 5411 5412
    .volDelete = testStorageVolumeDelete,
    .volGetInfo = testStorageVolumeGetInfo,
    .volGetXMLDesc = testStorageVolumeGetXMLDesc,
    .volGetPath = testStorageVolumeGetPath,
5413 5414
    .poolIsActive = testStoragePoolIsActive,
    .poolIsPersistent = testStoragePoolIsPersistent,
5415 5416
};

5417 5418 5419 5420
static virDeviceMonitor testDevMonitor = {
    .name = "Test",
    .open = testDevMonOpen,
    .close = testDevMonClose,
5421 5422 5423 5424 5425 5426 5427 5428

    .numOfDevices = testNodeNumOfDevices,
    .listDevices = testNodeListDevices,
    .deviceLookupByName = testNodeDeviceLookupByName,
    .deviceDumpXML = testNodeDeviceDumpXML,
    .deviceGetParent = testNodeDeviceGetParent,
    .deviceNumOfCaps = testNodeDeviceNumOfCaps,
    .deviceListCaps = testNodeDeviceListCaps,
5429 5430
    .deviceCreateXML = testNodeDeviceCreateXML,
    .deviceDestroy = testNodeDeviceDestroy,
5431 5432
};

5433 5434 5435 5436 5437
static virSecretDriver testSecretDriver = {
    .name = "Test",
    .open = testSecretOpen,
    .close = testSecretClose,
};
5438 5439


5440 5441 5442 5443 5444 5445
static virNWFilterDriver testNWFilterDriver = {
    .name = "Test",
    .open = testNWFilterOpen,
    .close = testNWFilterClose,
};

5446 5447 5448 5449 5450 5451 5452 5453 5454 5455 5456 5457
/**
 * testRegister:
 *
 * Registers the test driver
 */
int
testRegister(void)
{
    if (virRegisterDriver(&testDriver) < 0)
        return -1;
    if (virRegisterNetworkDriver(&testNetworkDriver) < 0)
        return -1;
L
Laine Stump 已提交
5458 5459
    if (virRegisterInterfaceDriver(&testInterfaceDriver) < 0)
        return -1;
5460 5461
    if (virRegisterStorageDriver(&testStorageDriver) < 0)
        return -1;
5462 5463
    if (virRegisterDeviceMonitor(&testDevMonitor) < 0)
        return -1;
5464 5465
    if (virRegisterSecretDriver(&testSecretDriver) < 0)
        return -1;
5466 5467
    if (virRegisterNWFilterDriver(&testNWFilterDriver) < 0)
        return -1;
5468

5469 5470
    return 0;
}