test.c 118.6 KB
Newer Older
1 2 3
/*
 * test.c: A "mock" hypervisor for use by application unit tests
 *
4
 * Copyright (C) 2006-2009 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.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 "xml.h"
50
#include "threads.h"
51
#include "logging.h"
52

53 54
#define VIR_FROM_THIS VIR_FROM_TEST

55 56 57 58 59 60 61 62 63 64 65
#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
66

67
struct _testConn {
68
    virMutex lock;
69

70 71
    char path[PATH_MAX];
    int nextDomID;
72
    virCapsPtr caps;
73
    virNodeInfo nodeInfo;
74
    virDomainObjList domains;
75
    virNetworkObjList networks;
L
Laine Stump 已提交
76
    virInterfaceObjList ifaces;
C
Cole Robinson 已提交
77
    virStoragePoolObjList pools;
78 79
    int numCells;
    testCell cells[MAX_CELLS];
80 81 82 83 84 85 86


    /* An array of callbacks */
    virDomainEventCallbackListPtr domainEventCallbacks;
    virDomainEventQueuePtr domainEventQueue;
    int domainEventTimer;
    int domainEventDispatching;
87 88 89
};
typedef struct _testConn testConn;
typedef struct _testConn *testConnPtr;
90

91
#define TEST_MODEL "i686"
92
#define TEST_MODEL_WORDSIZE 32
93
#define TEST_EMULATOR "/usr/bin/test-hv"
94

95
static const virNodeInfo defaultNodeInfo = {
96
    TEST_MODEL,
97 98 99 100 101 102 103
    1024*1024*3, /* 3 GB */
    16,
    1400,
    2,
    2,
    2,
    2,
104 105
};

106

107
#define testError(conn, code, fmt...)                               \
108
        virReportErrorHelper(conn, VIR_FROM_TEST, code, __FILE__, \
109
                               __FUNCTION__, __LINE__, fmt)
110

111 112 113 114 115 116
static int testClose(virConnectPtr conn);
static void testDomainEventFlush(int timer, void *opaque);
static void testDomainEventQueue(testConnPtr driver,
                                 virDomainEventPtr event);


117 118
static void testDriverLock(testConnPtr driver)
{
119
    virMutexLock(&driver->lock);
120 121 122 123
}

static void testDriverUnlock(testConnPtr driver)
{
124
    virMutexUnlock(&driver->lock);
125 126
}

127 128
static virCapsPtr
testBuildCapabilities(virConnectPtr conn) {
129
    testConnPtr privconn = conn->privateData;
130 131 132 133
    virCapsPtr caps;
    virCapsGuestPtr guest;
    const char *const guest_types[] = { "hvm", "xen" };
    int i;
134

135 136
    if ((caps = virCapabilitiesNew(TEST_MODEL, 0, 0)) == NULL)
        goto no_memory;
137

138 139 140 141
    if (virCapabilitiesAddHostFeature(caps, "pae") < 0)
        goto no_memory;
    if (virCapabilitiesAddHostFeature(caps ,"nonpae") < 0)
        goto no_memory;
142

143 144 145 146
    for (i = 0; i < privconn->numCells; i++) {
        if (virCapabilitiesAddHostNUMACell(caps, i, privconn->cells[i].numCpus,
                                           privconn->cells[i].cpus) < 0)
            goto no_memory;
147 148
    }

149 150 151 152 153 154 155 156 157 158
    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;
159

160 161 162 163 164 165 166
        if (virCapabilitiesAddGuestDomain(guest,
                                          "test",
                                          NULL,
                                          NULL,
                                          0,
                                          NULL) == NULL)
            goto no_memory;
167

168 169 170 171
        if (virCapabilitiesAddGuestFeature(guest, "pae", 1, 1) == NULL)
            goto no_memory;
        if (virCapabilitiesAddGuestFeature(guest ,"nonpae", 1, 1) == NULL)
            goto no_memory;
172 173
    }

174
    return caps;
175

176
no_memory:
177
    virReportOOMError(conn);
178 179
    virCapabilitiesFree(caps);
    return NULL;
180 181
}

182

183 184 185 186 187 188 189 190 191 192
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>";
193 194


195 196 197 198 199 200 201 202 203 204 205
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>";
206

L
Laine Stump 已提交
207 208 209 210 211 212 213 214 215 216 217
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 已提交
218 219 220 221 222 223 224 225
static const char *defaultPoolXML =
"<pool type='dir'>"
"  <name>default-pool</name>"
"  <target>"
"    <path>/default-pool</path>"
"  </target>"
"</pool>";

226
static const unsigned long long defaultPoolCap = (100 * 1024 * 1024 * 1024ull);
C
Cole Robinson 已提交
227 228
static const unsigned long long defaultPoolAlloc = 0;

229
static int testStoragePoolObjSetDefaults(virConnectPtr conn, virStoragePoolObjPtr pool);
230

231
static int testOpenDefault(virConnectPtr conn) {
232 233
    int u;
    struct timeval tv;
234
    testConnPtr privconn;
235 236 237 238
    virDomainDefPtr domdef = NULL;
    virDomainObjPtr domobj = NULL;
    virNetworkDefPtr netdef = NULL;
    virNetworkObjPtr netobj = NULL;
L
Laine Stump 已提交
239 240
    virInterfaceDefPtr interfacedef = NULL;
    virInterfaceObjPtr interfaceobj = NULL;
C
Cole Robinson 已提交
241 242
    virStoragePoolDefPtr pooldef = NULL;
    virStoragePoolObjPtr poolobj = NULL;
243

244
    if (VIR_ALLOC(privconn) < 0) {
245
        virReportOOMError(conn);
246 247
        return VIR_DRV_OPEN_ERROR;
    }
248 249 250 251 252 253 254
    if (virMutexInit(&privconn->lock) < 0) {
        testError(conn, VIR_ERR_INTERNAL_ERROR,
                  "%s", _("cannot initialize mutex"));
        VIR_FREE(privconn);
        return VIR_DRV_OPEN_ERROR;
    }

255
    testDriverLock(privconn);
256
    conn->privateData = privconn;
257 258

    if (gettimeofday(&tv, NULL) < 0) {
259 260
        virReportSystemError(conn, errno,
                             "%s", _("getting time of day"));
261
        goto error;
262 263
    }

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

266 267 268 269 270 271 272 273 274 275
    // 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;
    }

276 277 278 279 280
    if (!(privconn->caps = testBuildCapabilities(conn)))
        goto error;

    privconn->nextDomID = 1;

281 282 283
    if (!(domdef = virDomainDefParseString(conn, privconn->caps,
                                           defaultDomainXML,
                                           VIR_DOMAIN_XML_INACTIVE)))
284 285 286 287 288
        goto error;
    if (!(domobj = virDomainAssignDef(conn, &privconn->domains, domdef))) {
        virDomainDefFree(domdef);
        goto error;
    }
289
    domobj->def->id = privconn->nextDomID++;
290 291
    domobj->state = VIR_DOMAIN_RUNNING;
    domobj->persistent = 1;
292
    virDomainObjUnlock(domobj);
293 294 295 296 297 298 299 300 301

    if (!(netdef = virNetworkDefParseString(conn, defaultNetworkXML)))
        goto error;
    if (!(netobj = virNetworkAssignDef(conn, &privconn->networks, netdef))) {
        virNetworkDefFree(netdef);
        goto error;
    }
    netobj->active = 1;
    netobj->persistent = 1;
302
    virNetworkObjUnlock(netobj);
303

L
Laine Stump 已提交
304 305 306 307 308 309 310 311 312
    if (!(interfacedef = virInterfaceDefParseString(conn, defaultInterfaceXML)))
        goto error;
    if (!(interfaceobj = virInterfaceAssignDef(conn, &privconn->ifaces, interfacedef))) {
        virInterfaceDefFree(interfacedef);
        goto error;
    }
    interfaceobj->active = 1;
    virInterfaceObjUnlock(interfaceobj);

313
    if (!(pooldef = virStoragePoolDefParseString(conn, defaultPoolXML)))
C
Cole Robinson 已提交
314 315 316 317 318 319 320
        goto error;

    if (!(poolobj = virStoragePoolObjAssignDef(conn, &privconn->pools,
                                               pooldef))) {
        virStoragePoolDefFree(pooldef);
        goto error;
    }
321

322
    if (testStoragePoolObjSetDefaults(conn, poolobj) == -1) {
323
        virStoragePoolObjUnlock(poolobj);
C
Cole Robinson 已提交
324
        goto error;
325
    }
C
Cole Robinson 已提交
326
    poolobj->active = 1;
327
    virStoragePoolObjUnlock(poolobj);
C
Cole Robinson 已提交
328

329
    testDriverUnlock(privconn);
330

331 332 333
    return VIR_DRV_OPEN_SUCCESS;

error:
334
    virDomainObjListFree(&privconn->domains);
335
    virNetworkObjListFree(&privconn->networks);
L
Laine Stump 已提交
336
    virInterfaceObjListFree(&privconn->ifaces);
C
Cole Robinson 已提交
337
    virStoragePoolObjListFree(&privconn->pools);
338
    virCapabilitiesFree(privconn->caps);
339
    testDriverUnlock(privconn);
340
    conn->privateData = NULL;
341 342
    VIR_FREE(privconn);
    return VIR_DRV_OPEN_ERROR;
343 344 345 346
}


static char *testBuildFilename(const char *relativeTo,
347 348 349 350 351 352 353 354
                               const char *filename) {
    char *offset;
    int baseLen;
    if (!filename || filename[0] == '\0')
        return (NULL);
    if (filename[0] == '/')
        return strdup(filename);

355
    offset = strrchr(relativeTo, '/');
356
    if ((baseLen = (offset-relativeTo+1))) {
357 358 359
        char *absFile;
        if (VIR_ALLOC_N(absFile, baseLen + strlen(filename) + 1) < 0)
            return NULL;
360 361 362 363 364 365 366
        strncpy(absFile, relativeTo, baseLen);
        absFile[baseLen] = '\0';
        strcat(absFile, filename);
        return absFile;
    } else {
        return strdup(filename);
    }
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 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449
static int testOpenVolumesForPool(virConnectPtr conn,
                                  xmlDocPtr xml,
                                  xmlXPathContextPtr ctxt,
                                  const char *file,
                                  virStoragePoolObjPtr pool,
                                  int poolidx) {
    char *vol_xpath;
    int i, ret, func_ret = -1;
    xmlNodePtr *vols = NULL;
    virStorageVolDefPtr def;

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

    ret = virXPathNodeSet(conn, vol_xpath, ctxt, &vols);
    VIR_FREE(vol_xpath);
    if (ret < 0) {
        testError(NULL, VIR_ERR_XML_ERROR,
                  _("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) {
                testError(NULL, VIR_ERR_INTERNAL_ERROR, "%s",
                          _("resolving volume filename"));
                goto error;
            }

            def = virStorageVolDefParseFile(conn, pool->def, absFile);
            VIR_FREE(absFile);
            if (!def)
                goto error;
        } else {
            if ((def = virStorageVolDefParseNode(conn, pool->def, xml,
                                                 vols[i])) == NULL) {
                goto error;
            }
        }

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

        if (virAsprintf(&def->target.path, "%s/%s",
                        pool->def->target.path,
                        def->name) == -1) {
            virReportOOMError(conn);
            goto error;
        }

        def->key = strdup(def->target.path);
        if (def->key == NULL) {
            virReportOOMError(conn);
            goto error;
        }

        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;
}

450
static int testOpenFromFile(virConnectPtr conn,
451
                            const char *file) {
452
    int fd = -1, i, ret;
453 454
    long l;
    char *str;
455
    xmlDocPtr xml = NULL;
456
    xmlNodePtr root = NULL;
L
Laine Stump 已提交
457
    xmlNodePtr *domains = NULL, *networks = NULL, *ifaces = NULL, *pools = NULL;
458 459
    xmlXPathContextPtr ctxt = NULL;
    virNodeInfoPtr nodeInfo;
460
    virNetworkObjPtr net;
L
Laine Stump 已提交
461
    virInterfaceObjPtr iface;
462
    virDomainObjPtr dom;
463 464
    testConnPtr privconn;
    if (VIR_ALLOC(privconn) < 0) {
465
        virReportOOMError(conn);
466 467
        return VIR_DRV_OPEN_ERROR;
    }
468 469 470 471 472 473 474
    if (virMutexInit(&privconn->lock) < 0) {
        testError(conn, VIR_ERR_INTERNAL_ERROR,
                  "%s", _("cannot initialize mutex"));
        VIR_FREE(privconn);
        return VIR_DRV_OPEN_ERROR;
    }

475
    testDriverLock(privconn);
476
    conn->privateData = privconn;
477 478 479

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

    if ((fd = open(file, O_RDONLY)) < 0) {
482 483 484
        virReportSystemError(NULL, errno,
                             _("loading host definition file '%s'"),
                             file);
485
        goto error;
486 487
    }

488 489 490
    if (!(xml = xmlReadFd(fd, file, NULL,
                          XML_PARSE_NOENT | XML_PARSE_NONET |
                          XML_PARSE_NOERROR | XML_PARSE_NOWARNING))) {
491 492
        testError(NULL, VIR_ERR_INTERNAL_ERROR,
                  _("Invalid XML in file '%s'"), file);
493
        goto error;
494
    }
495 496
    close(fd);
    fd = -1;
497

498 499
    root = xmlDocGetRootElement(xml);
    if ((root == NULL) || (!xmlStrEqual(root->name, BAD_CAST "node"))) {
500 501
        testError(NULL, VIR_ERR_XML_ERROR, "%s",
                  _("Root element is not 'node'"));
502
        goto error;
503 504
    }

505 506
    ctxt = xmlXPathNewContext(xml);
    if (ctxt == NULL) {
C
Cole Robinson 已提交
507 508
        testError(NULL, VIR_ERR_INTERNAL_ERROR, "%s",
                  _("creating xpath context"));
509
        goto error;
510
    }
511

512
    privconn->nextDomID = 1;
513
    privconn->numCells = 0;
514 515 516 517 518
    strncpy(privconn->path, file, PATH_MAX-1);
    privconn->path[PATH_MAX-1] = '\0';
    memmove(&privconn->nodeInfo, &defaultNodeInfo, sizeof(defaultNodeInfo));

    nodeInfo = &privconn->nodeInfo;
519
    ret = virXPathLong(conn, "string(/node/cpu/nodes[1])", ctxt, &l);
520 521 522
    if (ret == 0) {
        nodeInfo->nodes = l;
    } else if (ret == -2) {
J
Jim Meyering 已提交
523
        testError(NULL, VIR_ERR_XML_ERROR, "%s", _("node cpu numa nodes"));
524
        goto error;
525
    }
526

527
    ret = virXPathLong(conn, "string(/node/cpu/sockets[1])", ctxt, &l);
528 529 530
    if (ret == 0) {
        nodeInfo->sockets = l;
    } else if (ret == -2) {
J
Jim Meyering 已提交
531
        testError(NULL, VIR_ERR_XML_ERROR, "%s", _("node cpu sockets"));
532
        goto error;
533
    }
534

535
    ret = virXPathLong(conn, "string(/node/cpu/cores[1])", ctxt, &l);
536 537 538
    if (ret == 0) {
        nodeInfo->cores = l;
    } else if (ret == -2) {
J
Jim Meyering 已提交
539
        testError(NULL, VIR_ERR_XML_ERROR, "%s", _("node cpu cores"));
540
        goto error;
541 542
    }

543
    ret = virXPathLong(conn, "string(/node/cpu/threads[1])", ctxt, &l);
544 545 546
    if (ret == 0) {
        nodeInfo->threads = l;
    } else if (ret == -2) {
J
Jim Meyering 已提交
547
        testError(NULL, VIR_ERR_XML_ERROR, "%s", _("node cpu threads"));
548
        goto error;
549
    }
550

551
    nodeInfo->cpus = nodeInfo->cores * nodeInfo->threads * nodeInfo->sockets * nodeInfo->nodes;
552
    ret = virXPathLong(conn, "string(/node/cpu/active[1])", ctxt, &l);
553 554
    if (ret == 0) {
        if (l < nodeInfo->cpus) {
555 556
            nodeInfo->cpus = l;
        }
557
    } else if (ret == -2) {
J
Jim Meyering 已提交
558
        testError(NULL, VIR_ERR_XML_ERROR, "%s", _("node active cpu"));
559
        goto error;
560
    }
561
    ret = virXPathLong(conn, "string(/node/cpu/mhz[1])", ctxt, &l);
562 563 564
    if (ret == 0) {
        nodeInfo->mhz = l;
    } else if (ret == -2) {
J
Jim Meyering 已提交
565
        testError(NULL, VIR_ERR_XML_ERROR, "%s", _("node cpu mhz"));
566
        goto error;
567 568
    }

569
    str = virXPathString(conn, "string(/node/cpu/model[1])", ctxt);
570 571
    if (str != NULL) {
        strncpy(nodeInfo->model, str, sizeof(nodeInfo->model)-1);
572
        nodeInfo->model[sizeof(nodeInfo->model)-1] = '\0';
573
        VIR_FREE(str);
574 575
    }

576
    ret = virXPathLong(conn, "string(/node/memory[1])", ctxt, &l);
577 578 579
    if (ret == 0) {
        nodeInfo->memory = l;
    } else if (ret == -2) {
J
Jim Meyering 已提交
580
        testError(NULL, VIR_ERR_XML_ERROR, "%s", _("node memory"));
581
        goto error;
582
    }
583

584
    ret = virXPathNodeSet(conn, "/node/domain", ctxt, &domains);
585
    if (ret < 0) {
J
Jim Meyering 已提交
586
        testError(NULL, VIR_ERR_XML_ERROR, "%s", _("node domain list"));
587
        goto error;
588
    }
589

590
    for (i = 0 ; i < ret ; i++) {
591 592 593 594 595 596
        virDomainDefPtr def;
        char *relFile = virXMLPropString(domains[i], "file");
        if (relFile != NULL) {
            char *absFile = testBuildFilename(file, relFile);
            VIR_FREE(relFile);
            if (!absFile) {
J
Jim Meyering 已提交
597
                testError(NULL, VIR_ERR_INTERNAL_ERROR, "%s", _("resolving domain filename"));
598 599
                goto error;
            }
600 601
            def = virDomainDefParseFile(conn, privconn->caps, absFile,
                                        VIR_DOMAIN_XML_INACTIVE);
602
            VIR_FREE(absFile);
603 604 605
            if (!def)
                goto error;
        } else {
606 607
            if ((def = virDomainDefParseNode(conn, privconn->caps, xml, domains[i],
                                   VIR_DOMAIN_XML_INACTIVE)) == NULL)
608 609 610 611 612
                goto error;
        }

        if (!(dom = virDomainAssignDef(conn, &privconn->domains, def))) {
            virDomainDefFree(def);
613 614
            goto error;
        }
615 616 617 618

        dom->state = VIR_DOMAIN_RUNNING;
        dom->def->id = privconn->nextDomID++;
        dom->persistent = 1;
619
        virDomainObjUnlock(dom);
620
    }
621
    VIR_FREE(domains);
622

623
    ret = virXPathNodeSet(conn, "/node/network", ctxt, &networks);
624
    if (ret < 0) {
J
Jim Meyering 已提交
625
        testError(NULL, VIR_ERR_XML_ERROR, "%s", _("node network list"));
626 627 628 629 630 631 632 633
        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);
634
            if (!absFile) {
J
Jim Meyering 已提交
635
                testError(NULL, VIR_ERR_INTERNAL_ERROR, "%s", _("resolving network filename"));
636 637
                goto error;
            }
638 639

            def = virNetworkDefParseFile(conn, absFile);
640
            VIR_FREE(absFile);
641 642 643 644 645
            if (!def)
                goto error;
        } else {
            if ((def = virNetworkDefParseNode(conn, xml, networks[i])) == NULL)
                goto error;
646
        }
647 648 649 650
        if (!(net = virNetworkAssignDef(conn, &privconn->networks,
                                        def))) {
            virNetworkDefFree(def);
            goto error;
651
        }
652
        net->persistent = 1;
653
        net->active = 1;
654
        virNetworkObjUnlock(net);
655
    }
656
    VIR_FREE(networks);
657

L
Laine Stump 已提交
658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690
    /* Parse interface definitions */
    ret = virXPathNodeSet(conn, "/node/interface", ctxt, &ifaces);
    if (ret < 0) {
        testError(NULL, VIR_ERR_XML_ERROR, "%s", _("node interface list"));
        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) {
                testError(NULL, VIR_ERR_INTERNAL_ERROR, "%s", _("resolving interface filename"));
                goto error;
            }

            def = virInterfaceDefParseFile(conn, absFile);
            VIR_FREE(absFile);
            if (!def)
                goto error;
        } else {
        if ((def = virInterfaceDefParseNode(conn, xml, ifaces[i])) == NULL)
                goto error;
        }
        if (!(iface = virInterfaceAssignDef(conn, &privconn->ifaces, def))) {
            virInterfaceDefFree(def);
            goto error;
        }
        virInterfaceObjUnlock(iface);
    }
    VIR_FREE(ifaces);

C
Cole Robinson 已提交
691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709
    /* Parse Storage Pool list */
    ret = virXPathNodeSet(conn, "/node/pool", ctxt, &pools);
    if (ret < 0) {
        testError(NULL, VIR_ERR_XML_ERROR, "%s", _("node pool list"));
        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) {
                testError(NULL, VIR_ERR_INTERNAL_ERROR, "%s",
                          _("resolving pool filename"));
                goto error;
            }

710
            def = virStoragePoolDefParseFile(conn, absFile);
C
Cole Robinson 已提交
711 712 713 714
            VIR_FREE(absFile);
            if (!def)
                goto error;
        } else {
715 716
            if ((def = virStoragePoolDefParseNode(conn, xml,
                                                  pools[i])) == NULL) {
C
Cole Robinson 已提交
717 718 719 720 721 722 723 724 725 726
                goto error;
            }
        }

        if (!(pool = virStoragePoolObjAssignDef(conn, &privconn->pools,
                                                def))) {
            virStoragePoolDefFree(def);
            goto error;
        }

727
        if (testStoragePoolObjSetDefaults(conn, pool) == -1) {
728
            virStoragePoolObjUnlock(pool);
C
Cole Robinson 已提交
729
            goto error;
730
        }
C
Cole Robinson 已提交
731
        pool->active = 1;
732 733 734 735 736 737 738

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

739
        virStoragePoolObjUnlock(pool);
C
Cole Robinson 已提交
740
    }
741
    VIR_FREE(pools);
C
Cole Robinson 已提交
742

J
Jim Meyering 已提交
743
    xmlXPathFreeContext(ctxt);
744
    xmlFreeDoc(xml);
745
    testDriverUnlock(privconn);
746

747
    return (0);
748 749

 error:
J
Jim Meyering 已提交
750
    xmlXPathFreeContext(ctxt);
751
    xmlFreeDoc(xml);
752 753
    VIR_FREE(domains);
    VIR_FREE(networks);
L
Laine Stump 已提交
754
    VIR_FREE(ifaces);
C
Cole Robinson 已提交
755
    VIR_FREE(pools);
756 757
    if (fd != -1)
        close(fd);
758
    virDomainObjListFree(&privconn->domains);
759
    virNetworkObjListFree(&privconn->networks);
L
Laine Stump 已提交
760
    virInterfaceObjListFree(&privconn->ifaces);
C
Cole Robinson 已提交
761
    virStoragePoolObjListFree(&privconn->pools);
762
    testDriverUnlock(privconn);
763
    VIR_FREE(privconn);
764
    conn->privateData = NULL;
765
    return VIR_DRV_OPEN_ERROR;
766 767
}

768

769
static virDrvOpenStatus testOpen(virConnectPtr conn,
770
                    virConnectAuthPtr auth ATTRIBUTE_UNUSED,
771
                    int flags ATTRIBUTE_UNUSED)
772
{
773
    int ret;
774

775
    if (!conn->uri)
776
        return VIR_DRV_OPEN_DECLINED;
777

778
    if (!conn->uri->scheme || STRNEQ(conn->uri->scheme, "test"))
779
        return VIR_DRV_OPEN_DECLINED;
780

781
    /* Remote driver should handle these. */
782
    if (conn->uri->server)
783 784
        return VIR_DRV_OPEN_DECLINED;

785
    /* From this point on, the connection is for us. */
786 787 788
    if (!conn->uri->path
        || conn->uri->path[0] == '\0'
        || (conn->uri->path[0] == '/' && conn->uri->path[1] == '\0')) {
789
        testError (NULL, VIR_ERR_INVALID_ARG,
J
Jim Meyering 已提交
790
                   "%s", _("testOpen: supply a path or use test:///default"));
791 792
        return VIR_DRV_OPEN_ERROR;
    }
793

794
    if (STREQ(conn->uri->path, "/default"))
795 796
        ret = testOpenDefault(conn);
    else
797
        ret = testOpenFromFile(conn,
798
                               conn->uri->path);
799

800 801
    if (ret == VIR_DRV_OPEN_SUCCESS) {
        testConnPtr privconn = conn->privateData;
802
        testDriverLock(privconn);
803 804 805 806
        /* Init callback list */
        if (VIR_ALLOC(privconn->domainEventCallbacks) < 0 ||
            !(privconn->domainEventQueue = virDomainEventQueueNew())) {
            virReportOOMError(NULL);
807
            testDriverUnlock(privconn);
808 809 810 811 812 813 814 815
            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.");
816
        testDriverUnlock(privconn);
817 818
    }

819
    return (ret);
820 821
}

822
static int testClose(virConnectPtr conn)
823
{
824
    testConnPtr privconn = conn->privateData;
825
    testDriverLock(privconn);
826
    virCapabilitiesFree(privconn->caps);
827
    virDomainObjListFree(&privconn->domains);
828
    virNetworkObjListFree(&privconn->networks);
L
Laine Stump 已提交
829
    virInterfaceObjListFree(&privconn->ifaces);
C
Cole Robinson 已提交
830
    virStoragePoolObjListFree(&privconn->pools);
831 832 833 834 835 836 837

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

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

838
    testDriverUnlock(privconn);
839
    virMutexDestroy(&privconn->lock);
840

841
    VIR_FREE (privconn);
842
    conn->privateData = NULL;
843
    return 0;
844 845
}

846 847
static int testGetVersion(virConnectPtr conn ATTRIBUTE_UNUSED,
                          unsigned long *hvVer)
848
{
849 850
    *hvVer = 2;
    return (0);
851 852
}

853
static char *testGetHostname (virConnectPtr conn)
854
{
855
    char *result;
856

857 858
    result = virGetHostname();
    if (result == NULL) {
859 860
        virReportSystemError(conn, errno,
                             "%s", _("cannot lookup hostname"));
861 862
        return NULL;
    }
863 864
    /* Caller frees this string. */
    return result;
865 866
}

867 868 869 870 871 872 873 874
static int testGetMaxVCPUs(virConnectPtr conn ATTRIBUTE_UNUSED,
                           const char *type ATTRIBUTE_UNUSED)
{
    return 32;
}

static int testNodeGetInfo(virConnectPtr conn,
                           virNodeInfoPtr info)
875
{
876
    testConnPtr privconn = conn->privateData;
877
    testDriverLock(privconn);
878
    memcpy(info, &privconn->nodeInfo, sizeof(virNodeInfo));
879
    testDriverUnlock(privconn);
880
    return (0);
881 882
}

883
static char *testGetCapabilities (virConnectPtr conn)
884
{
885
    testConnPtr privconn = conn->privateData;
886
    char *xml;
887
    testDriverLock(privconn);
888
    if ((xml = virCapabilitiesFormatXML(privconn->caps)) == NULL)
889
        virReportOOMError(conn);
890
    testDriverUnlock(privconn);
891
    return xml;
892 893
}

894
static int testNumOfDomains(virConnectPtr conn)
895
{
896
    testConnPtr privconn = conn->privateData;
897
    unsigned int numActive = 0, i;
898

899
    testDriverLock(privconn);
900 901
    for (i = 0 ; i < privconn->domains.count ; i++)
        if (virDomainIsActive(privconn->domains.objs[i]))
902
            numActive++;
903
    testDriverUnlock(privconn);
904

905
    return numActive;
906 907
}

908
static virDomainPtr
909
testDomainCreateXML(virConnectPtr conn, const char *xml,
910
                      unsigned int flags ATTRIBUTE_UNUSED)
911
{
912
    testConnPtr privconn = conn->privateData;
913
    virDomainPtr ret = NULL;
914
    virDomainDefPtr def;
915
    virDomainObjPtr dom = NULL;
916
    virDomainEventPtr event = NULL;
917

918
    testDriverLock(privconn);
919 920
    if ((def = virDomainDefParseString(conn, privconn->caps, xml,
                                       VIR_DOMAIN_XML_INACTIVE)) == NULL)
921
        goto cleanup;
922

923 924 925
    if ((dom = virDomainAssignDef(conn, &privconn->domains,
                                  def)) == NULL) {
        virDomainDefFree(def);
926
        goto cleanup;
927
    }
928 929
    dom->state = VIR_DOMAIN_RUNNING;
    dom->def->id = privconn->nextDomID++;
930

931 932 933 934
    event = virDomainEventNewFromObj(dom,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_BOOTED);

935
    ret = virGetDomain(conn, def->name, def->uuid);
936 937 938 939
    if (ret)
        ret->id = def->id;

cleanup:
940 941
    if (dom)
        virDomainObjUnlock(dom);
942 943
    if (event)
        testDomainEventQueue(privconn, event);
944
    testDriverUnlock(privconn);
945
    return ret;
946 947 948
}


949 950
static virDomainPtr testLookupDomainByID(virConnectPtr conn,
                                         int id)
951
{
952
    testConnPtr privconn = conn->privateData;
953 954
    virDomainPtr ret = NULL;
    virDomainObjPtr dom;
955

956 957 958 959 960
    testDriverLock(privconn);
    dom = virDomainFindByID(&privconn->domains, id);
    testDriverUnlock(privconn);

    if (dom == NULL) {
961
        testError (conn, VIR_ERR_NO_DOMAIN, NULL);
962
        goto cleanup;
963 964
    }

965
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid);
966 967 968 969
    if (ret)
        ret->id = dom->def->id;

cleanup:
970 971
    if (dom)
        virDomainObjUnlock(dom);
972
    return ret;
973 974
}

975 976
static virDomainPtr testLookupDomainByUUID(virConnectPtr conn,
                                           const unsigned char *uuid)
977
{
978
    testConnPtr privconn = conn->privateData;
979 980
    virDomainPtr ret = NULL;
    virDomainObjPtr dom ;
981

982 983 984 985 986
    testDriverLock(privconn);
    dom = virDomainFindByUUID(&privconn->domains, uuid);
    testDriverUnlock(privconn);

    if (dom == NULL) {
987
        testError (conn, VIR_ERR_NO_DOMAIN, NULL);
988
        goto cleanup;
989
    }
990

991
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid);
992 993 994 995
    if (ret)
        ret->id = dom->def->id;

cleanup:
996 997
    if (dom)
        virDomainObjUnlock(dom);
998
    return ret;
999 1000
}

1001 1002
static virDomainPtr testLookupDomainByName(virConnectPtr conn,
                                           const char *name)
1003
{
1004
    testConnPtr privconn = conn->privateData;
1005 1006
    virDomainPtr ret = NULL;
    virDomainObjPtr dom;
1007

1008 1009 1010 1011 1012
    testDriverLock(privconn);
    dom = virDomainFindByName(&privconn->domains, name);
    testDriverUnlock(privconn);

    if (dom == NULL) {
1013
        testError (conn, VIR_ERR_NO_DOMAIN, NULL);
1014
        goto cleanup;
1015
    }
1016

1017
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid);
1018 1019 1020 1021
    if (ret)
        ret->id = dom->def->id;

cleanup:
1022 1023
    if (dom)
        virDomainObjUnlock(dom);
1024
    return ret;
1025 1026
}

1027 1028 1029
static int testListDomains (virConnectPtr conn,
                            int *ids,
                            int maxids)
1030
{
1031
    testConnPtr privconn = conn->privateData;
1032
    unsigned int n = 0, i;
1033

1034 1035 1036
    testDriverLock(privconn);
    for (i = 0 ; i < privconn->domains.count && n < maxids ; i++) {
        virDomainObjLock(privconn->domains.objs[i]);
1037 1038
        if (virDomainIsActive(privconn->domains.objs[i]))
            ids[n++] = privconn->domains.objs[i]->def->id;
1039 1040 1041
        virDomainObjUnlock(privconn->domains.objs[i]);
    }
    testDriverUnlock(privconn);
1042

1043
    return n;
1044 1045
}

1046
static int testDestroyDomain (virDomainPtr domain)
1047
{
1048 1049
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1050
    virDomainEventPtr event = NULL;
1051
    int ret = -1;
1052

1053
    testDriverLock(privconn);
1054 1055 1056 1057 1058
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1059
        goto cleanup;
1060
    }
1061

1062
    privdom->state = VIR_DOMAIN_SHUTOFF;
1063 1064
    privdom->def->id = -1;
    domain->id = -1;
1065 1066 1067
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_DESTROYED);
1068 1069 1070
    if (!privdom->persistent) {
        virDomainRemoveInactive(&privconn->domains,
                                privdom);
1071
        privdom = NULL;
1072
    }
1073

1074

1075 1076
    ret = 0;
cleanup:
1077 1078
    if (privdom)
        virDomainObjUnlock(privdom);
1079 1080
    if (event)
        testDomainEventQueue(privconn, event);
1081
    testDriverUnlock(privconn);
1082
    return ret;
1083 1084
}

1085
static int testResumeDomain (virDomainPtr domain)
1086
{
1087 1088
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1089
    virDomainEventPtr event = NULL;
1090
    int ret = -1;
1091

1092
    testDriverLock(privconn);
1093 1094
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1095
    testDriverUnlock(privconn);
1096 1097 1098

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1099
        goto cleanup;
1100
    }
1101

1102
    if (privdom->state != VIR_DOMAIN_PAUSED) {
1103 1104 1105
        testError(domain->conn,
                  VIR_ERR_INTERNAL_ERROR, _("domain '%s' not paused"),
                  domain->name);
1106
        goto cleanup;
1107
    }
1108

1109
    privdom->state = VIR_DOMAIN_RUNNING;
1110 1111 1112
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_RESUMED,
                                     VIR_DOMAIN_EVENT_RESUMED_UNPAUSED);
1113 1114 1115
    ret = 0;

cleanup:
1116 1117
    if (privdom)
        virDomainObjUnlock(privdom);
1118 1119 1120 1121 1122
    if (event) {
        testDriverLock(privconn);
        testDomainEventQueue(privconn, event);
        testDriverUnlock(privconn);
    }
1123
    return ret;
1124 1125
}

1126
static int testPauseDomain (virDomainPtr domain)
1127
{
1128 1129
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1130
    virDomainEventPtr event = NULL;
1131
    int ret = -1;
1132

1133
    testDriverLock(privconn);
1134 1135
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1136
    testDriverUnlock(privconn);
1137 1138 1139

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1140
        goto cleanup;
1141
    }
1142

1143 1144
    if (privdom->state == VIR_DOMAIN_SHUTOFF ||
        privdom->state == VIR_DOMAIN_PAUSED) {
1145 1146 1147
        testError(domain->conn,
                  VIR_ERR_INTERNAL_ERROR, _("domain '%s' not running"),
                  domain->name);
1148
        goto cleanup;
1149
    }
1150

1151
    privdom->state = VIR_DOMAIN_PAUSED;
1152 1153 1154
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_SUSPENDED,
                                     VIR_DOMAIN_EVENT_SUSPENDED_PAUSED);
1155 1156 1157
    ret = 0;

cleanup:
1158 1159
    if (privdom)
        virDomainObjUnlock(privdom);
1160 1161 1162 1163 1164 1165

    if (event) {
        testDriverLock(privconn);
        testDomainEventQueue(privconn, event);
        testDriverUnlock(privconn);
    }
1166
    return ret;
1167 1168
}

1169
static int testShutdownDomain (virDomainPtr domain)
1170
{
1171 1172
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1173
    virDomainEventPtr event = NULL;
1174
    int ret = -1;
1175

1176
    testDriverLock(privconn);
1177 1178 1179 1180 1181
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1182
        goto cleanup;
1183
    }
1184

1185
    if (privdom->state == VIR_DOMAIN_SHUTOFF) {
1186 1187
        testError(domain->conn, VIR_ERR_INTERNAL_ERROR,
                  _("domain '%s' not running"), domain->name);
1188
        goto cleanup;
1189
    }
1190

1191
    privdom->state = VIR_DOMAIN_SHUTOFF;
1192
    domain->id = -1;
1193
    privdom->def->id = -1;
1194 1195 1196
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_SHUTDOWN);
1197 1198 1199 1200 1201
    if (!privdom->persistent) {
        virDomainRemoveInactive(&privconn->domains,
                                privdom);
        privdom = NULL;
    }
1202
    ret = 0;
1203

1204
cleanup:
1205 1206
    if (privdom)
        virDomainObjUnlock(privdom);
1207 1208
    if (event)
        testDomainEventQueue(privconn, event);
1209
    testDriverUnlock(privconn);
1210
    return ret;
1211 1212 1213
}

/* Similar behaviour as shutdown */
1214 1215
static int testRebootDomain (virDomainPtr domain,
                             unsigned int action ATTRIBUTE_UNUSED)
1216
{
1217 1218
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1219
    virDomainEventPtr event = NULL;
1220
    int ret = -1;
1221

1222
    testDriverLock(privconn);
1223 1224 1225 1226 1227
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1228
        goto cleanup;
1229
    }
1230

1231 1232 1233 1234
    privdom->state = VIR_DOMAIN_SHUTDOWN;
    switch (privdom->def->onReboot) {
    case VIR_DOMAIN_LIFECYCLE_DESTROY:
        privdom->state = VIR_DOMAIN_SHUTOFF;
1235
        domain->id = -1;
1236
        privdom->def->id = -1;
1237 1238
        break;

1239 1240
    case VIR_DOMAIN_LIFECYCLE_RESTART:
        privdom->state = VIR_DOMAIN_RUNNING;
1241 1242
        break;

1243 1244
    case VIR_DOMAIN_LIFECYCLE_PRESERVE:
        privdom->state = VIR_DOMAIN_SHUTOFF;
1245
        domain->id = -1;
1246
        privdom->def->id = -1;
1247 1248
        break;

1249 1250
    case VIR_DOMAIN_LIFECYCLE_RESTART_RENAME:
        privdom->state = VIR_DOMAIN_RUNNING;
1251
        break;
1252

1253
    default:
1254
        privdom->state = VIR_DOMAIN_SHUTOFF;
1255
        domain->id = -1;
1256
        privdom->def->id = -1;
1257 1258
        break;
    }
1259

1260 1261 1262 1263 1264 1265 1266 1267 1268
    if (privdom->state == VIR_DOMAIN_SHUTOFF) {
        event = virDomainEventNewFromObj(privdom,
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_SHUTDOWN);
        if (!privdom->persistent) {
            virDomainRemoveInactive(&privconn->domains,
                                    privdom);
            privdom = NULL;
        }
1269 1270
    }

1271 1272 1273
    ret = 0;

cleanup:
1274 1275
    if (privdom)
        virDomainObjUnlock(privdom);
1276 1277
    if (event)
        testDomainEventQueue(privconn, event);
1278
    testDriverUnlock(privconn);
1279
    return ret;
1280 1281
}

1282 1283
static int testGetDomainInfo (virDomainPtr domain,
                              virDomainInfoPtr info)
1284
{
1285
    testConnPtr privconn = domain->conn->privateData;
1286
    struct timeval tv;
1287
    virDomainObjPtr privdom;
1288
    int ret = -1;
1289

1290
    testDriverLock(privconn);
1291 1292
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1293
    testDriverUnlock(privconn);
1294 1295 1296

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1297
        goto cleanup;
1298
    }
1299 1300

    if (gettimeofday(&tv, NULL) < 0) {
1301
        testError(domain->conn, VIR_ERR_INTERNAL_ERROR,
J
Jim Meyering 已提交
1302
                  "%s", _("getting time of day"));
1303
        goto cleanup;
1304 1305
    }

1306 1307 1308 1309 1310
    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));
1311 1312 1313
    ret = 0;

cleanup:
1314 1315
    if (privdom)
        virDomainObjUnlock(privdom);
1316
    return ret;
1317 1318
}

1319 1320 1321 1322 1323
#define TEST_SAVE_MAGIC "TestGuestMagic"

static int testDomainSave(virDomainPtr domain,
                          const char *path)
{
1324
    testConnPtr privconn = domain->conn->privateData;
1325 1326 1327
    char *xml = NULL;
    int fd = -1;
    int len;
1328
    virDomainObjPtr privdom;
1329
    virDomainEventPtr event = NULL;
1330
    int ret = -1;
1331

1332
    testDriverLock(privconn);
1333 1334 1335 1336 1337
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1338
        goto cleanup;
1339
    }
1340

C
Cole Robinson 已提交
1341 1342 1343 1344
    xml = virDomainDefFormat(domain->conn,
                             privdom->def,
                             VIR_DOMAIN_XML_SECURE);

1345
    if (xml == NULL) {
1346 1347 1348
        virReportSystemError(domain->conn, errno,
                             _("saving domain '%s' failed to allocate space for metadata"),
                             domain->name);
1349
        goto cleanup;
1350
    }
1351 1352

    if ((fd = open(path, O_CREAT|O_TRUNC|O_WRONLY, S_IRUSR|S_IWUSR)) < 0) {
1353 1354 1355
        virReportSystemError(domain->conn, errno,
                             _("saving domain '%s' to '%s': open failed"),
                             domain->name, path);
1356
        goto cleanup;
1357
    }
1358
    len = strlen(xml);
1359
    if (safewrite(fd, TEST_SAVE_MAGIC, sizeof(TEST_SAVE_MAGIC)) < 0) {
1360 1361 1362
        virReportSystemError(domain->conn, errno,
                             _("saving domain '%s' to '%s': write failed"),
                             domain->name, path);
1363
        goto cleanup;
1364
    }
1365
    if (safewrite(fd, (char*)&len, sizeof(len)) < 0) {
1366 1367 1368
        virReportSystemError(domain->conn, errno,
                             _("saving domain '%s' to '%s': write failed"),
                             domain->name, path);
1369
        goto cleanup;
1370
    }
1371
    if (safewrite(fd, xml, len) < 0) {
1372 1373 1374
        virReportSystemError(domain->conn, errno,
                             _("saving domain '%s' to '%s': write failed"),
                             domain->name, path);
1375
        goto cleanup;
1376
    }
1377

1378
    if (close(fd) < 0) {
1379 1380 1381
        virReportSystemError(domain->conn, errno,
                             _("saving domain '%s' to '%s': write failed"),
                             domain->name, path);
1382
        goto cleanup;
1383
    }
1384 1385
    fd = -1;

1386
    privdom->state = VIR_DOMAIN_SHUTOFF;
1387 1388 1389
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_SAVED);
1390 1391 1392
    if (!privdom->persistent) {
        virDomainRemoveInactive(&privconn->domains,
                                privdom);
1393
        privdom = NULL;
1394
    }
1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407
    ret = 0;

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);
    }
1408 1409
    if (privdom)
        virDomainObjUnlock(privdom);
1410 1411
    if (event)
        testDomainEventQueue(privconn, event);
1412
    testDriverUnlock(privconn);
1413
    return ret;
1414 1415
}

1416 1417
static int testDomainRestore(virConnectPtr conn,
                             const char *path)
1418
{
1419
    testConnPtr privconn = conn->privateData;
1420
    char *xml = NULL;
1421
    char magic[15];
1422 1423 1424
    int fd = -1;
    int len;
    virDomainDefPtr def = NULL;
1425
    virDomainObjPtr dom = NULL;
1426
    virDomainEventPtr event = NULL;
1427
    int ret = -1;
1428 1429

    if ((fd = open(path, O_RDONLY)) < 0) {
1430 1431 1432
        virReportSystemError(conn, errno,
                             _("cannot read domain image '%s'"),
                             path);
1433
        goto cleanup;
1434
    }
1435 1436 1437 1438
    if (saferead(fd, magic, sizeof(magic)) != sizeof(magic)) {
        virReportSystemError(conn, errno,
                             _("incomplete save header in '%s'"),
                             path);
1439
        goto cleanup;
1440
    }
1441
    if (memcmp(magic, TEST_SAVE_MAGIC, sizeof(magic))) {
1442
        testError(conn, VIR_ERR_INTERNAL_ERROR,
J
Jim Meyering 已提交
1443
                  "%s", _("mismatched header magic"));
1444
        goto cleanup;
1445
    }
1446 1447 1448 1449
    if (saferead(fd, (char*)&len, sizeof(len)) != sizeof(len)) {
        virReportSystemError(conn, errno,
                             _("failed to read metadata length in '%s'"),
                             path);
1450
        goto cleanup;
1451 1452
    }
    if (len < 1 || len > 8192) {
1453
        testError(conn, VIR_ERR_INTERNAL_ERROR,
J
Jim Meyering 已提交
1454
                  "%s", _("length of metadata out of range"));
1455
        goto cleanup;
1456
    }
1457
    if (VIR_ALLOC_N(xml, len+1) < 0) {
1458
        virReportOOMError(conn);
1459
        goto cleanup;
1460
    }
1461 1462 1463
    if (saferead(fd, xml, len) != len) {
        virReportSystemError(conn, errno,
                             _("incomplete metdata in '%s'"), path);
1464
        goto cleanup;
1465 1466
    }
    xml[len] = '\0';
1467

1468
    testDriverLock(privconn);
1469 1470
    def = virDomainDefParseString(conn, privconn->caps, xml,
                                  VIR_DOMAIN_XML_INACTIVE);
1471
    if (!def)
1472
        goto cleanup;
1473 1474

    if ((dom = virDomainAssignDef(conn, &privconn->domains,
1475 1476 1477
                                  def)) == NULL)
        goto cleanup;

1478 1479
    dom->state = VIR_DOMAIN_RUNNING;
    dom->def->id = privconn->nextDomID++;
1480
    def = NULL;
1481 1482 1483
    event = virDomainEventNewFromObj(dom,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_RESTORED);
1484
    ret = 0;
1485 1486 1487 1488 1489 1490

cleanup:
    virDomainDefFree(def);
    VIR_FREE(xml);
    if (fd != -1)
        close(fd);
1491 1492
    if (dom)
        virDomainObjUnlock(dom);
1493 1494
    if (event)
        testDomainEventQueue(privconn, event);
1495
    testDriverUnlock(privconn);
1496
    return ret;
1497 1498
}

1499 1500 1501
static int testDomainCoreDump(virDomainPtr domain,
                              const char *to,
                              int flags ATTRIBUTE_UNUSED)
1502
{
1503
    testConnPtr privconn = domain->conn->privateData;
1504
    int fd = -1;
1505
    virDomainObjPtr privdom;
1506
    virDomainEventPtr event = NULL;
1507
    int ret = -1;
1508

1509
    testDriverLock(privconn);
1510 1511 1512 1513 1514
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1515
        goto cleanup;
1516
    }
1517 1518

    if ((fd = open(to, O_CREAT|O_TRUNC|O_WRONLY, S_IRUSR|S_IWUSR)) < 0) {
1519 1520 1521
        virReportSystemError(domain->conn, errno,
                             _("domain '%s' coredump: failed to open %s"),
                             domain->name, to);
1522
        goto cleanup;
1523
    }
1524
    if (safewrite(fd, TEST_SAVE_MAGIC, sizeof(TEST_SAVE_MAGIC)) < 0) {
1525 1526 1527
        virReportSystemError(domain->conn, errno,
                             _("domain '%s' coredump: failed to write header to %s"),
                             domain->name, to);
1528
        goto cleanup;
1529
    }
1530
    if (close(fd) < 0) {
1531 1532 1533
        virReportSystemError(domain->conn, errno,
                             _("domain '%s' coredump: write failed: %s"),
                             domain->name, to);
1534
        goto cleanup;
1535
    }
1536
    privdom->state = VIR_DOMAIN_SHUTOFF;
1537 1538 1539
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_CRASHED);
1540 1541 1542
    if (!privdom->persistent) {
        virDomainRemoveInactive(&privconn->domains,
                                privdom);
1543
        privdom = NULL;
1544
    }
1545 1546 1547 1548 1549
    ret = 0;

cleanup:
    if (fd != -1)
        close(fd);
1550 1551
    if (privdom)
        virDomainObjUnlock(privdom);
1552 1553
    if (event)
        testDomainEventQueue(privconn, event);
1554
    testDriverUnlock(privconn);
1555
    return ret;
1556 1557
}

1558 1559 1560
static char *testGetOSType(virDomainPtr dom) {
    char *ret = strdup("linux");
    if (!ret)
1561
        virReportOOMError(dom->conn);
1562
    return ret;
1563 1564 1565
}

static unsigned long testGetMaxMemory(virDomainPtr domain) {
1566 1567
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1568
    unsigned long ret = 0;
1569

1570
    testDriverLock(privconn);
1571 1572
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1573
    testDriverUnlock(privconn);
1574 1575 1576

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1577
        goto cleanup;
1578
    }
1579

1580 1581 1582
    ret = privdom->def->maxmem;

cleanup:
1583 1584
    if (privdom)
        virDomainObjUnlock(privdom);
1585
    return ret;
1586 1587 1588 1589 1590
}

static int testSetMaxMemory(virDomainPtr domain,
                            unsigned long memory)
{
1591 1592
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1593
    int ret = -1;
1594

1595
    testDriverLock(privconn);
1596 1597
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1598
    testDriverUnlock(privconn);
1599 1600 1601

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1602
        goto cleanup;
1603
    }
1604 1605

    /* XXX validate not over host memory wrt to other domains */
1606
    privdom->def->maxmem = memory;
1607 1608 1609
    ret = 0;

cleanup:
1610 1611
    if (privdom)
        virDomainObjUnlock(privdom);
1612
    return ret;
1613 1614
}

1615 1616 1617
static int testSetMemory(virDomainPtr domain,
                         unsigned long memory)
{
1618 1619
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1620
    int ret = -1;
1621

1622
    testDriverLock(privconn);
1623 1624
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1625
    testDriverUnlock(privconn);
1626 1627 1628

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1629
        goto cleanup;
1630
    }
1631

1632
    if (memory > privdom->def->maxmem) {
1633
        testError(domain->conn,
1634
                  VIR_ERR_INVALID_ARG, __FUNCTION__);
1635
        goto cleanup;
1636
    }
1637

1638
    privdom->def->memory = memory;
1639 1640 1641
    ret = 0;

cleanup:
1642 1643
    if (privdom)
        virDomainObjUnlock(privdom);
1644
    return ret;
1645 1646 1647 1648
}

static int testSetVcpus(virDomainPtr domain,
                        unsigned int nrCpus) {
1649 1650
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1651 1652
    int ret = -1;

1653 1654 1655 1656 1657 1658
    testDriverLock(privconn);
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
    testDriverUnlock(privconn);

    if (privdom == NULL) {
1659
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1660
        goto cleanup;
1661
    }
1662

1663 1664
    /* We allow more cpus in guest than host */
    if (nrCpus > 32) {
1665
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1666
        goto cleanup;
1667
    }
1668

1669
    privdom->def->vcpus = nrCpus;
1670 1671 1672
    ret = 0;

cleanup:
1673 1674
    if (privdom)
        virDomainObjUnlock(privdom);
1675
    return ret;
1676 1677
}

1678
static char *testDomainDumpXML(virDomainPtr domain, int flags)
1679
{
1680
    testConnPtr privconn = domain->conn->privateData;
1681
    virDomainDefPtr def;
1682
    virDomainObjPtr privdom;
1683 1684
    char *ret = NULL;

1685 1686 1687 1688 1689 1690
    testDriverLock(privconn);
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
    testDriverUnlock(privconn);

    if (privdom == NULL) {
1691
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1692
        goto cleanup;
1693
    }
1694

1695 1696
    def = (flags & VIR_DOMAIN_XML_INACTIVE) &&
        privdom->newDef ? privdom->newDef : privdom->def;
1697

1698 1699 1700 1701 1702
    ret = virDomainDefFormat(domain->conn,
                             def,
                             flags);

cleanup:
1703 1704
    if (privdom)
        virDomainObjUnlock(privdom);
1705
    return ret;
1706
}
1707

1708
static int testNumOfDefinedDomains(virConnectPtr conn) {
1709
    testConnPtr privconn = conn->privateData;
1710
    unsigned int numInactive = 0, i;
1711

1712 1713 1714
    testDriverLock(privconn);
    for (i = 0 ; i < privconn->domains.count ; i++) {
        virDomainObjLock(privconn->domains.objs[i]);
1715
        if (!virDomainIsActive(privconn->domains.objs[i]))
1716
            numInactive++;
1717 1718 1719
        virDomainObjUnlock(privconn->domains.objs[i]);
    }
    testDriverUnlock(privconn);
1720

1721
    return numInactive;
1722 1723
}

1724 1725 1726
static int testListDefinedDomains(virConnectPtr conn,
                                  char **const names,
                                  int maxnames) {
1727
    testConnPtr privconn = conn->privateData;
1728
    unsigned int n = 0, i;
1729

1730
    testDriverLock(privconn);
1731
    memset(names, 0, sizeof(*names)*maxnames);
1732 1733
    for (i = 0 ; i < privconn->domains.count && n < maxnames ; i++) {
        virDomainObjLock(privconn->domains.objs[i]);
1734
        if (!virDomainIsActive(privconn->domains.objs[i]) &&
1735 1736
            !(names[n++] = strdup(privconn->domains.objs[i]->def->name))) {
            virDomainObjUnlock(privconn->domains.objs[i]);
1737
            goto no_memory;
1738 1739 1740 1741
        }
        virDomainObjUnlock(privconn->domains.objs[i]);
    }
    testDriverUnlock(privconn);
1742

1743 1744 1745
    return n;

no_memory:
1746
    virReportOOMError(conn);
1747 1748
    for (n = 0 ; n < maxnames ; n++)
        VIR_FREE(names[n]);
1749
    testDriverUnlock(privconn);
1750
    return -1;
1751 1752
}

1753
static virDomainPtr testDomainDefineXML(virConnectPtr conn,
1754
                                        const char *xml) {
1755
    testConnPtr privconn = conn->privateData;
1756
    virDomainPtr ret = NULL;
1757
    virDomainDefPtr def;
1758
    virDomainObjPtr dom = NULL;
1759
    virDomainEventPtr event = NULL;
1760

1761
    testDriverLock(privconn);
1762 1763
    if ((def = virDomainDefParseString(conn, privconn->caps, xml,
                                       VIR_DOMAIN_XML_INACTIVE)) == NULL)
1764
        goto cleanup;
1765

1766 1767
    if ((dom = virDomainAssignDef(conn, &privconn->domains,
                                  def)) == NULL) {
1768
        goto cleanup;
1769
    }
1770
    def = NULL;
1771
    dom->persistent = 1;
1772

1773 1774 1775
    event = virDomainEventNewFromObj(dom,
                                     VIR_DOMAIN_EVENT_DEFINED,
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED);
1776

1777
    ret = virGetDomain(conn, dom->def->name, dom->def->uuid);
1778
    if (ret)
1779
        ret->id = dom->def->id;
1780 1781 1782

cleanup:
    virDomainDefFree(def);
1783 1784
    if (dom)
        virDomainObjUnlock(dom);
1785 1786
    if (event)
        testDomainEventQueue(privconn, event);
1787
    testDriverUnlock(privconn);
1788
    return ret;
1789 1790
}

1791 1792 1793
static int testNodeGetCellsFreeMemory(virConnectPtr conn,
                                      unsigned long long *freemems,
                                      int startCell, int maxCells) {
1794
    testConnPtr privconn = conn->privateData;
1795
    int i, j;
1796
    int ret = -1;
1797

1798
    testDriverLock(privconn);
1799
    if (startCell > privconn->numCells) {
1800
        testError(conn, VIR_ERR_INVALID_ARG,
J
Jim Meyering 已提交
1801
                  "%s", _("Range exceeds available cells"));
1802
        goto cleanup;
1803 1804 1805 1806 1807 1808 1809
    }

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

1812
cleanup:
1813
    testDriverUnlock(privconn);
1814
    return ret;
1815 1816 1817
}


1818
static int testDomainCreate(virDomainPtr domain) {
1819 1820
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1821
    virDomainEventPtr event = NULL;
1822
    int ret = -1;
1823

1824
    testDriverLock(privconn);
1825 1826 1827 1828 1829
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1830
        goto cleanup;
1831
    }
1832

1833
    if (privdom->state != VIR_DOMAIN_SHUTOFF) {
1834 1835
        testError(domain->conn, VIR_ERR_INTERNAL_ERROR,
                  _("Domain '%s' is already running"), domain->name);
1836
        goto cleanup;
1837 1838
    }

1839 1840
    domain->id = privdom->def->id = privconn->nextDomID++;
    privdom->state = VIR_DOMAIN_RUNNING;
1841 1842 1843
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_BOOTED);
1844
    ret = 0;
1845

1846
cleanup:
1847 1848
    if (privdom)
        virDomainObjUnlock(privdom);
1849 1850
    if (event)
        testDomainEventQueue(privconn, event);
1851
    testDriverUnlock(privconn);
1852
    return ret;
1853 1854 1855
}

static int testDomainUndefine(virDomainPtr domain) {
1856 1857
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1858
    virDomainEventPtr event = NULL;
1859
    int ret = -1;
1860

1861
    testDriverLock(privconn);
1862 1863 1864 1865 1866
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1867
        goto cleanup;
1868
    }
1869

1870
    if (privdom->state != VIR_DOMAIN_SHUTOFF) {
1871 1872
        testError(domain->conn, VIR_ERR_INTERNAL_ERROR,
                  _("Domain '%s' is still running"), domain->name);
1873
        goto cleanup;
1874 1875
    }

1876
    privdom->state = VIR_DOMAIN_SHUTOFF;
1877 1878 1879
    event = virDomainEventNewFromObj(privdom,
                                     VIR_DOMAIN_EVENT_UNDEFINED,
                                     VIR_DOMAIN_EVENT_UNDEFINED_REMOVED);
1880 1881
    virDomainRemoveInactive(&privconn->domains,
                            privdom);
1882
    privdom = NULL;
1883
    ret = 0;
1884

1885
cleanup:
1886 1887
    if (privdom)
        virDomainObjUnlock(privdom);
1888 1889
    if (event)
        testDomainEventQueue(privconn, event);
1890
    testDriverUnlock(privconn);
1891
    return ret;
1892 1893
}

1894 1895 1896
static int testDomainGetAutostart(virDomainPtr domain,
                                  int *autostart)
{
1897 1898
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1899
    int ret = -1;
1900

1901
    testDriverLock(privconn);
1902 1903
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1904
    testDriverUnlock(privconn);
1905 1906 1907

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1908
        goto cleanup;
1909 1910
    }

1911
    *autostart = privdom->autostart;
1912 1913 1914
    ret = 0;

cleanup:
1915 1916
    if (privdom)
        virDomainObjUnlock(privdom);
1917
    return ret;
1918 1919 1920 1921 1922 1923
}


static int testDomainSetAutostart(virDomainPtr domain,
                                  int autostart)
{
1924 1925
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1926
    int ret = -1;
1927

1928
    testDriverLock(privconn);
1929 1930
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1931
    testDriverUnlock(privconn);
1932 1933 1934

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1935
        goto cleanup;
1936 1937
    }

1938
    privdom->autostart = autostart ? 1 : 0;
1939 1940 1941
    ret = 0;

cleanup:
1942 1943
    if (privdom)
        virDomainObjUnlock(privdom);
1944
    return ret;
1945
}
1946

1947 1948 1949
static char *testDomainGetSchedulerType(virDomainPtr domain,
                                        int *nparams)
{
1950 1951
    char *type = NULL;

1952 1953
    *nparams = 1;
    type = strdup("fair");
1954
    if (!type)
1955
        virReportOOMError(domain->conn);
1956

1957 1958 1959 1960 1961 1962 1963
    return type;
}

static int testDomainGetSchedulerParams(virDomainPtr domain,
                                        virSchedParameterPtr params,
                                        int *nparams)
{
1964 1965
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
1966
    int ret = -1;
1967

1968
    testDriverLock(privconn);
1969 1970
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
1971
    testDriverUnlock(privconn);
1972 1973 1974

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
1975
        goto cleanup;
1976 1977
    }

1978
    if (*nparams != 1) {
1979
        testError(domain->conn, VIR_ERR_INVALID_ARG, "nparams");
1980
        goto cleanup;
1981
    }
1982 1983
    strcpy(params[0].field, "weight");
    params[0].type = VIR_DOMAIN_SCHED_FIELD_UINT;
1984 1985 1986
    /* XXX */
    /*params[0].value.ui = privdom->weight;*/
    params[0].value.ui = 50;
1987 1988 1989
    ret = 0;

cleanup:
1990 1991
    if (privdom)
        virDomainObjUnlock(privdom);
1992
    return ret;
1993
}
1994 1995


1996 1997 1998 1999
static int testDomainSetSchedulerParams(virDomainPtr domain,
                                        virSchedParameterPtr params,
                                        int nparams)
{
2000 2001
    testConnPtr privconn = domain->conn->privateData;
    virDomainObjPtr privdom;
2002
    int ret = -1;
2003

2004
    testDriverLock(privconn);
2005 2006
    privdom = virDomainFindByName(&privconn->domains,
                                  domain->name);
2007
    testDriverUnlock(privconn);
2008 2009 2010

    if (privdom == NULL) {
        testError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
2011
        goto cleanup;
2012 2013
    }

2014
    if (nparams != 1) {
2015
        testError(domain->conn, VIR_ERR_INVALID_ARG, "nparams");
2016
        goto cleanup;
2017
    }
2018
    if (STRNEQ(params[0].field, "weight")) {
2019
        testError(domain->conn, VIR_ERR_INVALID_ARG, "field");
2020
        goto cleanup;
2021 2022
    }
    if (params[0].type != VIR_DOMAIN_SCHED_FIELD_UINT) {
2023
        testError(domain->conn, VIR_ERR_INVALID_ARG, "type");
2024
        goto cleanup;
2025
    }
2026 2027
    /* XXX */
    /*privdom->weight = params[0].value.ui;*/
2028 2029 2030
    ret = 0;

cleanup:
2031 2032
    if (privdom)
        virDomainObjUnlock(privdom);
2033
    return ret;
2034 2035 2036
}

static virDrvOpenStatus testOpenNetwork(virConnectPtr conn,
2037
                                        virConnectAuthPtr auth ATTRIBUTE_UNUSED,
2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054
                                        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)
{
2055 2056
    testConnPtr privconn = conn->privateData;
    virNetworkObjPtr net;
2057
    virNetworkPtr ret = NULL;
2058

2059 2060 2061 2062 2063
    testDriverLock(privconn);
    net = virNetworkFindByUUID(&privconn->networks, uuid);
    testDriverUnlock(privconn);

    if (net == NULL) {
2064
        testError (conn, VIR_ERR_NO_NETWORK, NULL);
2065
        goto cleanup;
2066 2067
    }

2068 2069 2070
    ret = virGetNetwork(conn, net->def->name, net->def->uuid);

cleanup:
2071 2072
    if (net)
        virNetworkObjUnlock(net);
2073
    return ret;
2074
}
2075

2076
static virNetworkPtr testLookupNetworkByName(virConnectPtr conn,
2077
                                             const char *name)
2078
{
2079
    testConnPtr privconn = conn->privateData;
2080 2081
    virNetworkObjPtr net;
    virNetworkPtr ret = NULL;
2082

2083 2084 2085 2086 2087
    testDriverLock(privconn);
    net = virNetworkFindByName(&privconn->networks, name);
    testDriverUnlock(privconn);

    if (net == NULL) {
2088
        testError (conn, VIR_ERR_NO_NETWORK, NULL);
2089
        goto cleanup;
2090 2091
    }

2092 2093 2094
    ret = virGetNetwork(conn, net->def->name, net->def->uuid);

cleanup:
2095 2096
    if (net)
        virNetworkObjUnlock(net);
2097
    return ret;
2098 2099 2100 2101
}


static int testNumNetworks(virConnectPtr conn) {
2102
    testConnPtr privconn = conn->privateData;
2103
    int numActive = 0, i;
2104

2105 2106 2107
    testDriverLock(privconn);
    for (i = 0 ; i < privconn->networks.count ; i++) {
        virNetworkObjLock(privconn->networks.objs[i]);
2108
        if (virNetworkIsActive(privconn->networks.objs[i]))
2109
            numActive++;
2110 2111 2112
        virNetworkObjUnlock(privconn->networks.objs[i]);
    }
    testDriverUnlock(privconn);
2113

2114
    return numActive;
2115 2116 2117
}

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

2121
    testDriverLock(privconn);
2122
    memset(names, 0, sizeof(*names)*nnames);
2123 2124
    for (i = 0 ; i < privconn->networks.count && n < nnames ; i++) {
        virNetworkObjLock(privconn->networks.objs[i]);
2125
        if (virNetworkIsActive(privconn->networks.objs[i]) &&
2126 2127
            !(names[n++] = strdup(privconn->networks.objs[i]->def->name))) {
            virNetworkObjUnlock(privconn->networks.objs[i]);
2128
            goto no_memory;
2129 2130 2131 2132
        }
        virNetworkObjUnlock(privconn->networks.objs[i]);
    }
    testDriverUnlock(privconn);
2133

2134 2135 2136
    return n;

no_memory:
2137
    virReportOOMError(conn);
2138 2139
    for (n = 0 ; n < nnames ; n++)
        VIR_FREE(names[n]);
2140
    testDriverUnlock(privconn);
2141
    return -1;
2142 2143 2144
}

static int testNumDefinedNetworks(virConnectPtr conn) {
2145
    testConnPtr privconn = conn->privateData;
2146
    int numInactive = 0, i;
2147

2148 2149 2150
    testDriverLock(privconn);
    for (i = 0 ; i < privconn->networks.count ; i++) {
        virNetworkObjLock(privconn->networks.objs[i]);
2151
        if (!virNetworkIsActive(privconn->networks.objs[i]))
2152
            numInactive++;
2153 2154 2155
        virNetworkObjUnlock(privconn->networks.objs[i]);
    }
    testDriverUnlock(privconn);
2156

2157
    return numInactive;
2158 2159 2160
}

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

2164
    testDriverLock(privconn);
2165
    memset(names, 0, sizeof(*names)*nnames);
2166 2167
    for (i = 0 ; i < privconn->networks.count && n < nnames ; i++) {
        virNetworkObjLock(privconn->networks.objs[i]);
2168
        if (!virNetworkIsActive(privconn->networks.objs[i]) &&
2169 2170
            !(names[n++] = strdup(privconn->networks.objs[i]->def->name))) {
            virNetworkObjUnlock(privconn->networks.objs[i]);
2171
            goto no_memory;
2172 2173 2174 2175
        }
        virNetworkObjUnlock(privconn->networks.objs[i]);
    }
    testDriverUnlock(privconn);
2176

2177 2178 2179
    return n;

no_memory:
2180
    virReportOOMError(conn);
2181 2182
    for (n = 0 ; n < nnames ; n++)
        VIR_FREE(names[n]);
2183
    testDriverUnlock(privconn);
2184
    return -1;
2185 2186 2187
}

static virNetworkPtr testNetworkCreate(virConnectPtr conn, const char *xml) {
2188
    testConnPtr privconn = conn->privateData;
2189
    virNetworkDefPtr def;
2190
    virNetworkObjPtr net = NULL;
2191
    virNetworkPtr ret = NULL;
2192

2193
    testDriverLock(privconn);
2194
    if ((def = virNetworkDefParseString(conn, xml)) == NULL)
2195
        goto cleanup;
2196

2197
    if ((net = virNetworkAssignDef(conn, &privconn->networks, def)) == NULL)
2198 2199
        goto cleanup;
    def = NULL;
2200
    net->active = 1;
2201

2202
    ret = virGetNetwork(conn, net->def->name, net->def->uuid);
2203

2204 2205
cleanup:
    virNetworkDefFree(def);
2206 2207 2208
    if (net)
        virNetworkObjUnlock(net);
    testDriverUnlock(privconn);
2209
    return ret;
2210 2211 2212
}

static virNetworkPtr testNetworkDefine(virConnectPtr conn, const char *xml) {
2213
    testConnPtr privconn = conn->privateData;
2214
    virNetworkDefPtr def;
2215
    virNetworkObjPtr net = NULL;
2216
    virNetworkPtr ret = NULL;
2217

2218
    testDriverLock(privconn);
2219
    if ((def = virNetworkDefParseString(conn, xml)) == NULL)
2220
        goto cleanup;
2221

2222
    if ((net = virNetworkAssignDef(conn, &privconn->networks, def)) == NULL)
2223 2224
        goto cleanup;
    def = NULL;
2225
    net->persistent = 1;
2226

2227
    ret = virGetNetwork(conn, net->def->name, net->def->uuid);
2228 2229 2230

cleanup:
    virNetworkDefFree(def);
2231 2232 2233
    if (net)
        virNetworkObjUnlock(net);
    testDriverUnlock(privconn);
2234
    return ret;
2235 2236 2237
}

static int testNetworkUndefine(virNetworkPtr network) {
2238 2239
    testConnPtr privconn = network->conn->privateData;
    virNetworkObjPtr privnet;
2240
    int ret = -1;
2241

2242
    testDriverLock(privconn);
2243 2244 2245 2246 2247
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);

    if (privnet == NULL) {
        testError(network->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
2248
        goto cleanup;
2249
    }
2250

2251
    if (virNetworkIsActive(privnet)) {
2252 2253
        testError(network->conn, VIR_ERR_INTERNAL_ERROR,
                  _("Network '%s' is still running"), network->name);
2254
        goto cleanup;
2255 2256
    }

2257 2258
    virNetworkRemoveInactive(&privconn->networks,
                             privnet);
2259
    privnet = NULL;
2260
    ret = 0;
2261

2262
cleanup:
2263 2264 2265
    if (privnet)
        virNetworkObjUnlock(privnet);
    testDriverUnlock(privconn);
2266
    return ret;
2267 2268 2269
}

static int testNetworkStart(virNetworkPtr network) {
2270 2271
    testConnPtr privconn = network->conn->privateData;
    virNetworkObjPtr privnet;
2272
    int ret = -1;
2273

2274
    testDriverLock(privconn);
2275 2276
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);
2277
    testDriverUnlock(privconn);
2278 2279 2280

    if (privnet == NULL) {
        testError(network->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
2281
        goto cleanup;
2282
    }
2283

2284
    if (virNetworkIsActive(privnet)) {
2285 2286
        testError(network->conn, VIR_ERR_INTERNAL_ERROR,
                  _("Network '%s' is already running"), network->name);
2287
        goto cleanup;
2288 2289
    }

2290
    privnet->active = 1;
2291
    ret = 0;
2292

2293
cleanup:
2294 2295
    if (privnet)
        virNetworkObjUnlock(privnet);
2296
    return ret;
2297 2298 2299
}

static int testNetworkDestroy(virNetworkPtr network) {
2300 2301
    testConnPtr privconn = network->conn->privateData;
    virNetworkObjPtr privnet;
2302
    int ret = -1;
2303

2304
    testDriverLock(privconn);
2305 2306 2307 2308 2309
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);

    if (privnet == NULL) {
        testError(network->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
2310
        goto cleanup;
2311
    }
2312

2313 2314 2315 2316
    privnet->active = 0;
    if (!privnet->persistent) {
        virNetworkRemoveInactive(&privconn->networks,
                                 privnet);
2317
        privnet = NULL;
2318
    }
2319 2320 2321
    ret = 0;

cleanup:
2322 2323 2324
    if (privnet)
        virNetworkObjUnlock(privnet);
    testDriverUnlock(privconn);
2325
    return ret;
2326 2327 2328
}

static char *testNetworkDumpXML(virNetworkPtr network, int flags ATTRIBUTE_UNUSED) {
2329 2330
    testConnPtr privconn = network->conn->privateData;
    virNetworkObjPtr privnet;
2331
    char *ret = NULL;
2332

2333
    testDriverLock(privconn);
2334 2335
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);
2336
    testDriverUnlock(privconn);
2337 2338 2339

    if (privnet == NULL) {
        testError(network->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
2340
        goto cleanup;
2341
    }
2342

2343 2344 2345
    ret = virNetworkDefFormat(network->conn, privnet->def);

cleanup:
2346 2347
    if (privnet)
        virNetworkObjUnlock(privnet);
2348
    return ret;
2349 2350 2351
}

static char *testNetworkGetBridgeName(virNetworkPtr network) {
2352
    testConnPtr privconn = network->conn->privateData;
2353
    char *bridge = NULL;
2354 2355
    virNetworkObjPtr privnet;

2356
    testDriverLock(privconn);
2357 2358
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);
2359
    testDriverUnlock(privconn);
2360 2361 2362

    if (privnet == NULL) {
        testError(network->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
2363
        goto cleanup;
2364 2365
    }

2366 2367
    if (privnet->def->bridge &&
        !(bridge = strdup(privnet->def->bridge))) {
2368
        virReportOOMError(network->conn);
2369
        goto cleanup;
2370
    }
2371 2372

cleanup:
2373 2374
    if (privnet)
        virNetworkObjUnlock(privnet);
2375 2376 2377 2378 2379
    return bridge;
}

static int testNetworkGetAutostart(virNetworkPtr network,
                                   int *autostart) {
2380 2381
    testConnPtr privconn = network->conn->privateData;
    virNetworkObjPtr privnet;
2382
    int ret = -1;
2383

2384
    testDriverLock(privconn);
2385 2386
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);
2387
    testDriverUnlock(privconn);
2388 2389 2390

    if (privnet == NULL) {
        testError(network->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
2391
        goto cleanup;
2392 2393
    }

2394
    *autostart = privnet->autostart;
2395 2396 2397
    ret = 0;

cleanup:
2398 2399
    if (privnet)
        virNetworkObjUnlock(privnet);
2400
    return ret;
2401 2402 2403 2404
}

static int testNetworkSetAutostart(virNetworkPtr network,
                                   int autostart) {
2405 2406
    testConnPtr privconn = network->conn->privateData;
    virNetworkObjPtr privnet;
2407
    int ret = -1;
2408

2409
    testDriverLock(privconn);
2410 2411
    privnet = virNetworkFindByName(&privconn->networks,
                                   network->name);
2412
    testDriverUnlock(privconn);
2413 2414 2415

    if (privnet == NULL) {
        testError(network->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
2416
        goto cleanup;
2417 2418
    }

2419
    privnet->autostart = autostart ? 1 : 0;
2420 2421 2422
    ret = 0;

cleanup:
2423 2424
    if (privnet)
        virNetworkObjUnlock(privnet);
2425
    return ret;
2426
}
2427

C
Cole Robinson 已提交
2428

L
Laine Stump 已提交
2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596 2597 2598 2599 2600 2601 2602 2603 2604 2605 2606 2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 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 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736
/*
 * 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]);
        if (virInterfaceIsActive(privconn->ifaces.objs[i])) {
            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]);
        if (virInterfaceIsActive(privconn->ifaces.objs[i])) {
            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:
    virReportOOMError(conn);
    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]);
        if (!virInterfaceIsActive(privconn->ifaces.objs[i])) {
            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]);
        if (!virInterfaceIsActive(privconn->ifaces.objs[i])) {
            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:
    virReportOOMError(conn);
    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) {
        testError (conn, VIR_ERR_NO_INTERFACE, NULL);
        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) {
        testError (conn, VIR_ERR_NO_INTERFACE, NULL);
        goto cleanup;
    }

    if (ifacect > 1) {
        testError (conn, VIR_ERR_MULTIPLE_INTERFACES, NULL);
        goto cleanup;
    }

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

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

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) {
        testError(iface->conn, VIR_ERR_NO_INTERFACE, __FUNCTION__);
        goto cleanup;
    }

    ret = virInterfaceDefFormat(iface->conn, privinterface->def);

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);
    if ((def = virInterfaceDefParseString(conn, xmlStr)) == NULL)
        goto cleanup;

    if ((iface = virInterfaceAssignDef(conn, &privconn->ifaces, def)) == NULL)
        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) {
        testError (iface->conn, VIR_ERR_NO_INTERFACE, NULL);
        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) {
        testError (iface->conn, VIR_ERR_NO_INTERFACE, NULL);
        goto cleanup;
    }

    if (privinterface->active != 0) {
        testError (iface->conn, VIR_ERR_OPERATION_INVALID, NULL);
        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) {
        testError (iface->conn, VIR_ERR_NO_INTERFACE, NULL);
        goto cleanup;
    }

    if (privinterface->active == 0) {
        testError (iface->conn, VIR_ERR_OPERATION_INVALID, NULL);
        goto cleanup;
    }

    privinterface->active = 0;
    ret = 0;

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



C
Cole Robinson 已提交
2737 2738 2739 2740
/*
 * Storage Driver routines
 */

2741 2742
static int testStoragePoolObjSetDefaults(virConnectPtr conn,
                                         virStoragePoolObjPtr pool) {
C
Cole Robinson 已提交
2743 2744 2745 2746 2747 2748 2749

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

    pool->configFile = strdup("\0");
    if (!pool->configFile) {
2750
        virReportOOMError(conn);
C
Cole Robinson 已提交
2751 2752 2753 2754 2755 2756
        return -1;
    }

    return 0;
}

2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771
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;
}

C
Cole Robinson 已提交
2772 2773 2774
static virStoragePoolPtr
testStoragePoolLookupByUUID(virConnectPtr conn,
                            const unsigned char *uuid) {
2775
    testConnPtr privconn = conn->privateData;
2776 2777
    virStoragePoolObjPtr pool;
    virStoragePoolPtr ret = NULL;
C
Cole Robinson 已提交
2778

2779
    testDriverLock(privconn);
2780
    pool = virStoragePoolObjFindByUUID(&privconn->pools, uuid);
2781
    testDriverUnlock(privconn);
2782 2783

    if (pool == NULL) {
C
Cole Robinson 已提交
2784
        testError (conn, VIR_ERR_NO_STORAGE_POOL, NULL);
2785
        goto cleanup;
C
Cole Robinson 已提交
2786 2787
    }

2788 2789 2790
    ret = virGetStoragePool(conn, pool->def->name, pool->def->uuid);

cleanup:
2791 2792
    if (pool)
        virStoragePoolObjUnlock(pool);
2793
    return ret;
C
Cole Robinson 已提交
2794 2795 2796 2797 2798
}

static virStoragePoolPtr
testStoragePoolLookupByName(virConnectPtr conn,
                            const char *name) {
2799
    testConnPtr privconn = conn->privateData;
2800 2801
    virStoragePoolObjPtr pool;
    virStoragePoolPtr ret = NULL;
C
Cole Robinson 已提交
2802

2803
    testDriverLock(privconn);
2804
    pool = virStoragePoolObjFindByName(&privconn->pools, name);
2805
    testDriverUnlock(privconn);
2806 2807

    if (pool == NULL) {
C
Cole Robinson 已提交
2808
        testError (conn, VIR_ERR_NO_STORAGE_POOL, NULL);
2809
        goto cleanup;
C
Cole Robinson 已提交
2810 2811
    }

2812 2813 2814
    ret = virGetStoragePool(conn, pool->def->name, pool->def->uuid);

cleanup:
2815 2816
    if (pool)
        virStoragePoolObjUnlock(pool);
2817
    return ret;
C
Cole Robinson 已提交
2818 2819 2820 2821 2822 2823 2824 2825 2826
}

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

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

2830
    testDriverLock(privconn);
C
Cole Robinson 已提交
2831 2832 2833
    for (i = 0 ; i < privconn->pools.count ; i++)
        if (virStoragePoolObjIsActive(privconn->pools.objs[i]))
            numActive++;
2834
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
2835 2836 2837 2838 2839 2840 2841 2842

    return numActive;
}

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

2846
    testDriverLock(privconn);
C
Cole Robinson 已提交
2847
    memset(names, 0, sizeof(*names)*nnames);
2848 2849
    for (i = 0 ; i < privconn->pools.count && n < nnames ; i++) {
        virStoragePoolObjLock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
2850
        if (virStoragePoolObjIsActive(privconn->pools.objs[i]) &&
2851 2852
            !(names[n++] = strdup(privconn->pools.objs[i]->def->name))) {
            virStoragePoolObjUnlock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
2853
            goto no_memory;
2854 2855 2856 2857
        }
        virStoragePoolObjUnlock(privconn->pools.objs[i]);
    }
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
2858 2859 2860 2861

    return n;

no_memory:
2862
    virReportOOMError(conn);
C
Cole Robinson 已提交
2863 2864
    for (n = 0 ; n < nnames ; n++)
        VIR_FREE(names[n]);
2865
    testDriverUnlock(privconn);
2866
    return -1;
C
Cole Robinson 已提交
2867 2868 2869 2870
}

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

2874 2875 2876
    testDriverLock(privconn);
    for (i = 0 ; i < privconn->pools.count ; i++) {
        virStoragePoolObjLock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
2877 2878
        if (!virStoragePoolObjIsActive(privconn->pools.objs[i]))
            numInactive++;
2879 2880 2881
        virStoragePoolObjUnlock(privconn->pools.objs[i]);
    }
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
2882 2883 2884 2885 2886 2887 2888 2889

    return numInactive;
}

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

2893
    testDriverLock(privconn);
C
Cole Robinson 已提交
2894
    memset(names, 0, sizeof(*names)*nnames);
2895 2896
    for (i = 0 ; i < privconn->pools.count && n < nnames ; i++) {
        virStoragePoolObjLock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
2897
        if (!virStoragePoolObjIsActive(privconn->pools.objs[i]) &&
2898 2899
            !(names[n++] = strdup(privconn->pools.objs[i]->def->name))) {
            virStoragePoolObjUnlock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
2900
            goto no_memory;
2901 2902 2903 2904
        }
        virStoragePoolObjUnlock(privconn->pools.objs[i]);
    }
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
2905 2906 2907 2908

    return n;

no_memory:
2909
    virReportOOMError(conn);
C
Cole Robinson 已提交
2910 2911
    for (n = 0 ; n < nnames ; n++)
        VIR_FREE(names[n]);
2912
    testDriverUnlock(privconn);
2913
    return -1;
C
Cole Robinson 已提交
2914 2915 2916 2917
}


static int
2918
testStoragePoolStart(virStoragePoolPtr pool,
C
Cole Robinson 已提交
2919
                     unsigned int flags ATTRIBUTE_UNUSED) {
2920 2921
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
2922
    int ret = -1;
2923

2924
    testDriverLock(privconn);
2925 2926
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
2927
    testDriverUnlock(privconn);
2928 2929 2930

    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
2931
        goto cleanup;
2932 2933
    }

2934 2935 2936 2937 2938
    if (virStoragePoolObjIsActive(privpool)) {
        testError(pool->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is already active"), pool->name);
        goto cleanup;
    }
C
Cole Robinson 已提交
2939 2940

    privpool->active = 1;
2941
    ret = 0;
C
Cole Robinson 已提交
2942

2943
cleanup:
2944 2945
    if (privpool)
        virStoragePoolObjUnlock(privpool);
2946
    return ret;
C
Cole Robinson 已提交
2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958 2959 2960 2961 2962
}

static char *
testStorageFindPoolSources(virConnectPtr conn ATTRIBUTE_UNUSED,
                           const char *type ATTRIBUTE_UNUSED,
                           const char *srcSpec ATTRIBUTE_UNUSED,
                           unsigned int flags ATTRIBUTE_UNUSED)
{
    return NULL;
}


static virStoragePoolPtr
testStoragePoolCreate(virConnectPtr conn,
                      const char *xml,
                      unsigned int flags ATTRIBUTE_UNUSED) {
2963
    testConnPtr privconn = conn->privateData;
C
Cole Robinson 已提交
2964
    virStoragePoolDefPtr def;
2965
    virStoragePoolObjPtr pool = NULL;
2966
    virStoragePoolPtr ret = NULL;
C
Cole Robinson 已提交
2967

2968
    testDriverLock(privconn);
2969
    if (!(def = virStoragePoolDefParseString(conn, xml)))
2970
        goto cleanup;
C
Cole Robinson 已提交
2971

2972 2973 2974 2975
    pool = virStoragePoolObjFindByUUID(&privconn->pools, def->uuid);
    if (!pool)
        pool = virStoragePoolObjFindByName(&privconn->pools, def->name);
    if (pool) {
C
Cole Robinson 已提交
2976 2977
        testError(conn, VIR_ERR_INTERNAL_ERROR,
                  "%s", _("storage pool already exists"));
2978
        goto cleanup;
C
Cole Robinson 已提交
2979 2980
    }

2981
    if (!(pool = virStoragePoolObjAssignDef(conn, &privconn->pools, def)))
2982
        goto cleanup;
2983
    def = NULL;
C
Cole Robinson 已提交
2984

2985
    if (testStoragePoolObjSetDefaults(conn, pool) == -1) {
C
Cole Robinson 已提交
2986
        virStoragePoolObjRemove(&privconn->pools, pool);
2987 2988
        pool = NULL;
        goto cleanup;
C
Cole Robinson 已提交
2989 2990 2991
    }
    pool->active = 1;

2992 2993 2994 2995
    ret = virGetStoragePool(conn, pool->def->name, pool->def->uuid);

cleanup:
    virStoragePoolDefFree(def);
2996 2997 2998
    if (pool)
        virStoragePoolObjUnlock(pool);
    testDriverUnlock(privconn);
2999
    return ret;
C
Cole Robinson 已提交
3000 3001 3002 3003 3004 3005
}

static virStoragePoolPtr
testStoragePoolDefine(virConnectPtr conn,
                      const char *xml,
                      unsigned int flags ATTRIBUTE_UNUSED) {
3006
    testConnPtr privconn = conn->privateData;
C
Cole Robinson 已提交
3007
    virStoragePoolDefPtr def;
3008
    virStoragePoolObjPtr pool = NULL;
3009
    virStoragePoolPtr ret = NULL;
C
Cole Robinson 已提交
3010

3011
    testDriverLock(privconn);
3012
    if (!(def = virStoragePoolDefParseString(conn, xml)))
3013
        goto cleanup;
C
Cole Robinson 已提交
3014 3015 3016 3017 3018

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

3019
    if (!(pool = virStoragePoolObjAssignDef(conn, &privconn->pools, def)))
3020 3021
        goto cleanup;
    def = NULL;
C
Cole Robinson 已提交
3022

3023
    if (testStoragePoolObjSetDefaults(conn, pool) == -1) {
C
Cole Robinson 已提交
3024
        virStoragePoolObjRemove(&privconn->pools, pool);
3025 3026
        pool = NULL;
        goto cleanup;
C
Cole Robinson 已提交
3027 3028
    }

3029 3030 3031 3032
    ret = virGetStoragePool(conn, pool->def->name, pool->def->uuid);

cleanup:
    virStoragePoolDefFree(def);
3033 3034 3035
    if (pool)
        virStoragePoolObjUnlock(pool);
    testDriverUnlock(privconn);
3036
    return ret;
C
Cole Robinson 已提交
3037 3038 3039
}

static int
3040 3041 3042
testStoragePoolUndefine(virStoragePoolPtr pool) {
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3043
    int ret = -1;
3044

3045
    testDriverLock(privconn);
3046 3047 3048 3049 3050
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);

    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3051
        goto cleanup;
3052 3053
    }

3054 3055 3056 3057 3058
    if (virStoragePoolObjIsActive(privpool)) {
        testError(pool->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is already active"), pool->name);
        goto cleanup;
    }
C
Cole Robinson 已提交
3059 3060

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

3063
cleanup:
3064 3065 3066
    if (privpool)
        virStoragePoolObjUnlock(privpool);
    testDriverUnlock(privconn);
3067
    return ret;
C
Cole Robinson 已提交
3068 3069 3070
}

static int
3071
testStoragePoolBuild(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3072
                     unsigned int flags ATTRIBUTE_UNUSED) {
3073 3074
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3075
    int ret = -1;
3076

3077
    testDriverLock(privconn);
3078 3079
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3080
    testDriverUnlock(privconn);
3081 3082 3083

    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3084
        goto cleanup;
3085 3086
    }

3087 3088 3089 3090 3091
    if (virStoragePoolObjIsActive(privpool)) {
        testError(pool->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is already active"), pool->name);
        goto cleanup;
    }
3092
    ret = 0;
C
Cole Robinson 已提交
3093

3094
cleanup:
3095 3096
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3097
    return ret;
C
Cole Robinson 已提交
3098 3099 3100 3101
}


static int
3102 3103 3104
testStoragePoolDestroy(virStoragePoolPtr pool) {
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3105
    int ret = -1;
3106

3107
    testDriverLock(privconn);
3108 3109 3110 3111 3112
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);

    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3113
        goto cleanup;
3114 3115 3116 3117 3118
    }

    if (!virStoragePoolObjIsActive(privpool)) {
        testError(pool->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is not active"), pool->name);
3119
        goto cleanup;
3120
    }
C
Cole Robinson 已提交
3121 3122 3123

    privpool->active = 0;

3124
    if (privpool->configFile == NULL) {
C
Cole Robinson 已提交
3125
        virStoragePoolObjRemove(&privconn->pools, privpool);
3126 3127
        privpool = NULL;
    }
3128
    ret = 0;
C
Cole Robinson 已提交
3129

3130
cleanup:
3131 3132 3133
    if (privpool)
        virStoragePoolObjUnlock(privpool);
    testDriverUnlock(privconn);
3134
    return ret;
C
Cole Robinson 已提交
3135 3136 3137 3138
}


static int
3139
testStoragePoolDelete(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3140
                      unsigned int flags ATTRIBUTE_UNUSED) {
3141 3142
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3143
    int ret = -1;
3144

3145
    testDriverLock(privconn);
3146 3147
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3148
    testDriverUnlock(privconn);
3149 3150 3151

    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3152 3153 3154 3155 3156 3157 3158
        goto cleanup;
    }

    if (virStoragePoolObjIsActive(privpool)) {
        testError(pool->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is already active"), pool->name);
        goto cleanup;
3159 3160
    }

3161
    ret = 0;
C
Cole Robinson 已提交
3162

3163
cleanup:
3164 3165
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3166
    return ret;
C
Cole Robinson 已提交
3167 3168 3169 3170
}


static int
3171
testStoragePoolRefresh(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3172
                       unsigned int flags ATTRIBUTE_UNUSED) {
3173 3174
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3175
    int ret = -1;
3176

3177
    testDriverLock(privconn);
3178 3179
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3180
    testDriverUnlock(privconn);
3181 3182 3183

    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3184
        goto cleanup;
3185 3186 3187 3188 3189
    }

    if (!virStoragePoolObjIsActive(privpool)) {
        testError(pool->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is not active"), pool->name);
3190
        goto cleanup;
3191
    }
3192
    ret = 0;
C
Cole Robinson 已提交
3193

3194
cleanup:
3195 3196
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3197
    return ret;
C
Cole Robinson 已提交
3198 3199 3200 3201
}


static int
3202
testStoragePoolGetInfo(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3203
                       virStoragePoolInfoPtr info) {
3204 3205
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3206
    int ret = -1;
3207

3208
    testDriverLock(privconn);
3209 3210
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3211
    testDriverUnlock(privconn);
3212 3213 3214

    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3215
        goto cleanup;
3216
    }
C
Cole Robinson 已提交
3217 3218 3219 3220 3221 3222 3223 3224 3225

    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;
3226
    ret = 0;
C
Cole Robinson 已提交
3227

3228
cleanup:
3229 3230
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3231
    return ret;
C
Cole Robinson 已提交
3232 3233 3234
}

static char *
3235
testStoragePoolDumpXML(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3236
                       unsigned int flags ATTRIBUTE_UNUSED) {
3237 3238
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3239
    char *ret = NULL;
3240

3241
    testDriverLock(privconn);
3242 3243
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3244
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
3245

3246 3247
    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3248
        goto cleanup;
3249 3250
    }

3251 3252 3253
    ret = virStoragePoolDefFormat(pool->conn, privpool->def);

cleanup:
3254 3255
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3256
    return ret;
C
Cole Robinson 已提交
3257 3258 3259
}

static int
3260
testStoragePoolGetAutostart(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3261
                            int *autostart) {
3262 3263
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3264
    int ret = -1;
3265

3266
    testDriverLock(privconn);
3267 3268
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3269
    testDriverUnlock(privconn);
3270 3271 3272

    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3273
        goto cleanup;
3274
    }
C
Cole Robinson 已提交
3275 3276 3277 3278 3279 3280

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

3283
cleanup:
3284 3285
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3286
    return ret;
C
Cole Robinson 已提交
3287 3288 3289
}

static int
3290
testStoragePoolSetAutostart(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3291
                            int autostart) {
3292 3293
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3294
    int ret = -1;
3295

3296
    testDriverLock(privconn);
3297 3298
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3299
    testDriverUnlock(privconn);
3300 3301 3302

    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3303
        goto cleanup;
3304
    }
C
Cole Robinson 已提交
3305 3306

    if (!privpool->configFile) {
3307
        testError(pool->conn, VIR_ERR_INVALID_ARG,
C
Cole Robinson 已提交
3308
                  "%s", _("pool has no config file"));
3309
        goto cleanup;
C
Cole Robinson 已提交
3310 3311 3312 3313
    }

    autostart = (autostart != 0);
    privpool->autostart = autostart;
3314 3315 3316
    ret = 0;

cleanup:
3317 3318
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3319
    return ret;
C
Cole Robinson 已提交
3320 3321 3322 3323
}


static int
3324 3325 3326
testStoragePoolNumVolumes(virStoragePoolPtr pool) {
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3327
    int ret = -1;
3328

3329
    testDriverLock(privconn);
3330 3331
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3332
    testDriverUnlock(privconn);
3333 3334 3335

    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3336
        goto cleanup;
3337 3338 3339 3340 3341
    }

    if (!virStoragePoolObjIsActive(privpool)) {
        testError(pool->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is not active"), pool->name);
3342
        goto cleanup;
3343
    }
C
Cole Robinson 已提交
3344

3345 3346 3347
    ret = privpool->volumes.count;

cleanup:
3348 3349
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3350
    return ret;
C
Cole Robinson 已提交
3351 3352 3353
}

static int
3354
testStoragePoolListVolumes(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3355 3356
                           char **const names,
                           int maxnames) {
3357 3358
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
C
Cole Robinson 已提交
3359 3360
    int i = 0, n = 0;

3361
    memset(names, 0, maxnames * sizeof(*names));
3362 3363

    testDriverLock(privconn);
3364 3365
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3366
    testDriverUnlock(privconn);
3367 3368 3369

    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3370
        goto cleanup;
3371 3372 3373 3374 3375 3376
    }


    if (!virStoragePoolObjIsActive(privpool)) {
        testError(pool->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is not active"), pool->name);
3377
        goto cleanup;
3378 3379
    }

C
Cole Robinson 已提交
3380 3381
    for (i = 0 ; i < privpool->volumes.count && n < maxnames ; i++) {
        if ((names[n++] = strdup(privpool->volumes.objs[i]->name)) == NULL) {
3382
            virReportOOMError(pool->conn);
C
Cole Robinson 已提交
3383 3384 3385 3386
            goto cleanup;
        }
    }

3387
    virStoragePoolObjUnlock(privpool);
C
Cole Robinson 已提交
3388 3389 3390 3391 3392 3393
    return n;

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

3394
    memset(names, 0, maxnames * sizeof(*names));
3395 3396
    if (privpool)
        virStoragePoolObjUnlock(privpool);
C
Cole Robinson 已提交
3397 3398 3399 3400 3401
    return -1;
}


static virStorageVolPtr
3402
testStorageVolumeLookupByName(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3403
                              const char *name ATTRIBUTE_UNUSED) {
3404 3405 3406
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
    virStorageVolDefPtr privvol;
3407
    virStorageVolPtr ret = NULL;
3408

3409
    testDriverLock(privconn);
3410 3411
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3412
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
3413

3414 3415
    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3416
        goto cleanup;
3417 3418 3419 3420 3421 3422
    }


    if (!virStoragePoolObjIsActive(privpool)) {
        testError(pool->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is not active"), pool->name);
3423
        goto cleanup;
3424 3425 3426 3427 3428 3429
    }

    privvol = virStorageVolDefFindByName(privpool, name);

    if (!privvol) {
        testError(pool->conn, VIR_ERR_INVALID_STORAGE_VOL,
C
Cole Robinson 已提交
3430
                  _("no storage vol with matching name '%s'"), name);
3431
        goto cleanup;
C
Cole Robinson 已提交
3432 3433
    }

3434 3435 3436 3437
    ret = virGetStorageVol(pool->conn, privpool->def->name,
                           privvol->name, privvol->key);

cleanup:
3438 3439
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3440
    return ret;
C
Cole Robinson 已提交
3441 3442 3443 3444 3445 3446
}


static virStorageVolPtr
testStorageVolumeLookupByKey(virConnectPtr conn,
                             const char *key) {
3447
    testConnPtr privconn = conn->privateData;
C
Cole Robinson 已提交
3448
    unsigned int i;
3449
    virStorageVolPtr ret = NULL;
C
Cole Robinson 已提交
3450

3451
    testDriverLock(privconn);
C
Cole Robinson 已提交
3452
    for (i = 0 ; i < privconn->pools.count ; i++) {
3453
        virStoragePoolObjLock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
3454
        if (virStoragePoolObjIsActive(privconn->pools.objs[i])) {
3455
            virStorageVolDefPtr privvol =
C
Cole Robinson 已提交
3456 3457
                virStorageVolDefFindByKey(privconn->pools.objs[i], key);

3458 3459 3460 3461 3462
            if (privvol) {
                ret = virGetStorageVol(conn,
                                       privconn->pools.objs[i]->def->name,
                                       privvol->name,
                                       privvol->key);
3463
                virStoragePoolObjUnlock(privconn->pools.objs[i]);
3464 3465
                break;
            }
C
Cole Robinson 已提交
3466
        }
3467
        virStoragePoolObjUnlock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
3468
    }
3469
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
3470

3471 3472 3473 3474 3475
    if (!ret)
        testError(conn, VIR_ERR_INVALID_STORAGE_VOL,
                  _("no storage vol with matching key '%s'"), key);

    return ret;
C
Cole Robinson 已提交
3476 3477 3478 3479 3480
}

static virStorageVolPtr
testStorageVolumeLookupByPath(virConnectPtr conn,
                              const char *path) {
3481
    testConnPtr privconn = conn->privateData;
C
Cole Robinson 已提交
3482
    unsigned int i;
3483
    virStorageVolPtr ret = NULL;
C
Cole Robinson 已提交
3484

3485
    testDriverLock(privconn);
C
Cole Robinson 已提交
3486
    for (i = 0 ; i < privconn->pools.count ; i++) {
3487
        virStoragePoolObjLock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
3488
        if (virStoragePoolObjIsActive(privconn->pools.objs[i])) {
3489
            virStorageVolDefPtr privvol =
C
Cole Robinson 已提交
3490 3491
                virStorageVolDefFindByPath(privconn->pools.objs[i], path);

3492 3493 3494 3495 3496
            if (privvol) {
                ret = virGetStorageVol(conn,
                                       privconn->pools.objs[i]->def->name,
                                       privvol->name,
                                       privvol->key);
3497
                virStoragePoolObjUnlock(privconn->pools.objs[i]);
3498 3499
                break;
            }
C
Cole Robinson 已提交
3500
        }
3501
        virStoragePoolObjUnlock(privconn->pools.objs[i]);
C
Cole Robinson 已提交
3502
    }
3503
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
3504

3505 3506 3507 3508 3509
    if (!ret)
        testError(conn, VIR_ERR_INVALID_STORAGE_VOL,
                  _("no storage vol with matching path '%s'"), path);

    return ret;
C
Cole Robinson 已提交
3510 3511 3512
}

static virStorageVolPtr
3513
testStorageVolumeCreateXML(virStoragePoolPtr pool,
C
Cole Robinson 已提交
3514 3515
                           const char *xmldesc,
                           unsigned int flags ATTRIBUTE_UNUSED) {
3516 3517
    testConnPtr privconn = pool->conn->privateData;
    virStoragePoolObjPtr privpool;
3518 3519
    virStorageVolDefPtr privvol = NULL;
    virStorageVolPtr ret = NULL;
3520

3521
    testDriverLock(privconn);
3522 3523
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           pool->name);
3524
    testDriverUnlock(privconn);
C
Cole Robinson 已提交
3525

3526 3527
    if (privpool == NULL) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3528
        goto cleanup;
3529 3530 3531 3532 3533
    }

    if (!virStoragePoolObjIsActive(privpool)) {
        testError(pool->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is not active"), pool->name);
3534
        goto cleanup;
3535
    }
C
Cole Robinson 已提交
3536

3537
    privvol = virStorageVolDefParseString(pool->conn, privpool->def, xmldesc);
3538
    if (privvol == NULL)
3539
        goto cleanup;
3540 3541

    if (virStorageVolDefFindByName(privpool, privvol->name)) {
3542
        testError(pool->conn, VIR_ERR_INVALID_STORAGE_VOL,
C
Cole Robinson 已提交
3543
                  "%s", _("storage vol already exists"));
3544
        goto cleanup;
C
Cole Robinson 已提交
3545 3546 3547
    }

    /* Make sure enough space */
3548
    if ((privpool->def->allocation + privvol->allocation) >
C
Cole Robinson 已提交
3549
         privpool->def->capacity) {
3550
        testError(pool->conn, VIR_ERR_INTERNAL_ERROR,
C
Cole Robinson 已提交
3551
                  _("Not enough free space in pool for volume '%s'"),
3552
                  privvol->name);
3553
        goto cleanup;
C
Cole Robinson 已提交
3554 3555 3556 3557
    }

    if (VIR_REALLOC_N(privpool->volumes.objs,
                      privpool->volumes.count+1) < 0) {
3558
        virReportOOMError(pool->conn);
3559
        goto cleanup;
C
Cole Robinson 已提交
3560 3561
    }

3562 3563 3564
    if (virAsprintf(&privvol->target.path, "%s/%s",
                    privpool->def->target.path,
                    privvol->name) == -1) {
3565
        virReportOOMError(pool->conn);
3566
        goto cleanup;
C
Cole Robinson 已提交
3567 3568
    }

3569 3570
    privvol->key = strdup(privvol->target.path);
    if (privvol->key == NULL) {
3571
        virReportOOMError(pool->conn);
3572
        goto cleanup;
C
Cole Robinson 已提交
3573 3574
    }

3575
    privpool->def->allocation += privvol->allocation;
C
Cole Robinson 已提交
3576 3577 3578
    privpool->def->available = (privpool->def->capacity -
                                privpool->def->allocation);

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

3581 3582
    ret = virGetStorageVol(pool->conn, privpool->def->name,
                           privvol->name, privvol->key);
3583
    privvol = NULL;
3584 3585 3586

cleanup:
    virStorageVolDefFree(privvol);
3587 3588
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3589
    return ret;
C
Cole Robinson 已提交
3590 3591
}

3592 3593 3594 3595 3596 3597 3598 3599 3600 3601 3602 3603 3604 3605 3606 3607 3608 3609 3610 3611 3612 3613 3614 3615 3616 3617
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) {
        testError(pool->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
        goto cleanup;
    }

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

3618
    privvol = virStorageVolDefParseString(pool->conn, privpool->def, xmldesc);
3619 3620 3621 3622 3623 3624 3625 3626 3627 3628 3629 3630 3631 3632 3633 3634 3635 3636 3637 3638 3639 3640 3641 3642 3643 3644 3645 3646 3647 3648 3649 3650 3651 3652
    if (privvol == NULL)
        goto cleanup;

    if (virStorageVolDefFindByName(privpool, privvol->name)) {
        testError(pool->conn, VIR_ERR_INVALID_STORAGE_VOL,
                  "%s", _("storage vol already exists"));
        goto cleanup;
    }

    origvol = virStorageVolDefFindByName(privpool, clonevol->name);
    if (!origvol) {
        testError(pool->conn, VIR_ERR_INVALID_STORAGE_VOL,
                  _("no storage vol with matching name '%s'"),
                  clonevol->name);
        goto cleanup;
    }

    /* Make sure enough space */
    if ((privpool->def->allocation + privvol->allocation) >
         privpool->def->capacity) {
        testError(pool->conn, VIR_ERR_INTERNAL_ERROR,
                  _("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) {
        virReportOOMError(pool->conn);
        goto cleanup;
    }

3653 3654 3655
    if (virAsprintf(&privvol->target.path, "%s/%s",
                    privpool->def->target.path,
                    privvol->name) == -1) {
3656 3657 3658 3659 3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682
        virReportOOMError(pool->conn);
        goto cleanup;
    }

    privvol->key = strdup(privvol->target.path);
    if (privvol->key == NULL) {
        virReportOOMError(pool->conn);
        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 已提交
3683
static int
3684
testStorageVolumeDelete(virStorageVolPtr vol,
C
Cole Robinson 已提交
3685
                        unsigned int flags ATTRIBUTE_UNUSED) {
3686 3687 3688
    testConnPtr privconn = vol->conn->privateData;
    virStoragePoolObjPtr privpool;
    virStorageVolDefPtr privvol;
C
Cole Robinson 已提交
3689
    int i;
3690
    int ret = -1;
C
Cole Robinson 已提交
3691

3692
    testDriverLock(privconn);
3693 3694
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           vol->pool);
3695
    testDriverUnlock(privconn);
3696 3697 3698

    if (privpool == NULL) {
        testError(vol->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3699
        goto cleanup;
3700 3701 3702 3703 3704 3705 3706 3707 3708
    }


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

    if (privvol == NULL) {
        testError(vol->conn, VIR_ERR_INVALID_STORAGE_VOL,
                  _("no storage vol with matching name '%s'"),
                  vol->name);
3709
        goto cleanup;
3710 3711 3712 3713 3714
    }

    if (!virStoragePoolObjIsActive(privpool)) {
        testError(vol->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is not active"), vol->pool);
3715
        goto cleanup;
3716 3717 3718
    }


C
Cole Robinson 已提交
3719 3720 3721 3722 3723 3724 3725 3726 3727 3728 3729 3730 3731 3732 3733 3734 3735 3736 3737 3738 3739 3740 3741
    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;
        }
    }
3742
    ret = 0;
C
Cole Robinson 已提交
3743

3744
cleanup:
3745 3746
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3747
    return ret;
C
Cole Robinson 已提交
3748 3749 3750 3751 3752 3753 3754 3755 3756 3757 3758 3759 3760 3761 3762 3763
}


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
3764
testStorageVolumeGetInfo(virStorageVolPtr vol,
C
Cole Robinson 已提交
3765
                         virStorageVolInfoPtr info) {
3766 3767 3768
    testConnPtr privconn = vol->conn->privateData;
    virStoragePoolObjPtr privpool;
    virStorageVolDefPtr privvol;
3769
    int ret = -1;
3770

3771
    testDriverLock(privconn);
3772 3773
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           vol->pool);
3774
    testDriverUnlock(privconn);
3775 3776 3777

    if (privpool == NULL) {
        testError(vol->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3778
        goto cleanup;
3779 3780 3781 3782 3783 3784 3785 3786
    }

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

    if (privvol == NULL) {
        testError(vol->conn, VIR_ERR_INVALID_STORAGE_VOL,
                  _("no storage vol with matching name '%s'"),
                  vol->name);
3787
        goto cleanup;
3788 3789 3790 3791 3792
    }

    if (!virStoragePoolObjIsActive(privpool)) {
        testError(vol->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is not active"), vol->pool);
3793
        goto cleanup;
3794
    }
C
Cole Robinson 已提交
3795 3796 3797 3798 3799

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

3802
cleanup:
3803 3804
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3805
    return ret;
C
Cole Robinson 已提交
3806 3807 3808
}

static char *
3809
testStorageVolumeGetXMLDesc(virStorageVolPtr vol,
C
Cole Robinson 已提交
3810
                            unsigned int flags ATTRIBUTE_UNUSED) {
3811 3812 3813
    testConnPtr privconn = vol->conn->privateData;
    virStoragePoolObjPtr privpool;
    virStorageVolDefPtr privvol;
3814
    char *ret = NULL;
3815

3816
    testDriverLock(privconn);
3817 3818
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           vol->pool);
3819
    testDriverUnlock(privconn);
3820 3821 3822

    if (privpool == NULL) {
        testError(vol->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3823
        goto cleanup;
3824 3825 3826 3827 3828 3829 3830 3831
    }

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

    if (privvol == NULL) {
        testError(vol->conn, VIR_ERR_INVALID_STORAGE_VOL,
                  _("no storage vol with matching name '%s'"),
                  vol->name);
3832
        goto cleanup;
3833
    }
C
Cole Robinson 已提交
3834

3835 3836 3837
    if (!virStoragePoolObjIsActive(privpool)) {
        testError(vol->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is not active"), vol->pool);
3838
        goto cleanup;
3839 3840
    }

3841 3842 3843
    ret = virStorageVolDefFormat(vol->conn, privpool->def, privvol);

cleanup:
3844 3845
    if (privpool)
        virStoragePoolObjUnlock(privpool);
3846
    return ret;
C
Cole Robinson 已提交
3847 3848 3849
}

static char *
3850 3851 3852 3853
testStorageVolumeGetPath(virStorageVolPtr vol) {
    testConnPtr privconn = vol->conn->privateData;
    virStoragePoolObjPtr privpool;
    virStorageVolDefPtr privvol;
3854
    char *ret = NULL;
3855

3856
    testDriverLock(privconn);
3857 3858
    privpool = virStoragePoolObjFindByName(&privconn->pools,
                                           vol->pool);
3859
    testDriverUnlock(privconn);
3860 3861 3862

    if (privpool == NULL) {
        testError(vol->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
3863
        goto cleanup;
3864 3865 3866 3867 3868 3869 3870 3871
    }

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

    if (privvol == NULL) {
        testError(vol->conn, VIR_ERR_INVALID_STORAGE_VOL,
                  _("no storage vol with matching name '%s'"),
                  vol->name);
3872
        goto cleanup;
3873 3874 3875 3876 3877
    }

    if (!virStoragePoolObjIsActive(privpool)) {
        testError(vol->conn, VIR_ERR_INTERNAL_ERROR,
                  _("storage pool '%s' is not active"), vol->pool);
3878
        goto cleanup;
3879 3880
    }

C
Cole Robinson 已提交
3881
    ret = strdup(privvol->target.path);
3882
    if (ret == NULL)
3883
        virReportOOMError(vol->conn);
3884 3885

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

3891

3892 3893 3894 3895 3896 3897 3898 3899 3900 3901 3902 3903 3904 3905 3906 3907
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;
}


3908 3909 3910 3911 3912 3913 3914 3915 3916 3917 3918 3919 3920 3921 3922 3923 3924 3925 3926 3927 3928 3929 3930 3931 3932 3933 3934 3935 3936 3937 3938 3939 3940 3941 3942 3943 3944 3945 3946 3947 3948 3949 3950 3951 3952 3953 3954 3955 3956 3957 3958 3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978 3979 3980 3981 3982 3983 3984 3985 3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997 3998 3999 4000 4001 4002 4003 4004
static int
testDomainEventRegister (virConnectPtr conn,
                         virConnectDomainEventCallback callback,
                         void *opaque,
                         virFreeCallback freecb)
{
    testConnPtr driver = conn->privateData;
    int ret;

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

    return ret;
}

static int
testDomainEventDeregister (virConnectPtr conn,
                           virConnectDomainEventCallback callback)
{
    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;
}

static void testDomainEventDispatchFunc(virConnectPtr conn,
                                        virDomainEventPtr event,
                                        virConnectDomainEventCallback cb,
                                        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);
}


4005 4006 4007 4008 4009
static virDriver testDriver = {
    VIR_DRV_TEST,
    "Test",
    testOpen, /* open */
    testClose, /* close */
4010
    NULL, /* supports_feature */
4011 4012
    NULL, /* type */
    testGetVersion, /* version */
4013
    testGetHostname, /* getHostname */
4014 4015 4016 4017 4018
    testGetMaxVCPUs, /* getMaxVcpus */
    testNodeGetInfo, /* nodeGetInfo */
    testGetCapabilities, /* getCapabilities */
    testListDomains, /* listDomains */
    testNumOfDomains, /* numOfDomains */
4019
    testDomainCreateXML, /* domainCreateXML */
4020 4021 4022 4023 4024 4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039
    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 */
    NULL, /* domainPinVcpu */
    NULL, /* domainGetVcpus */
    NULL, /* domainGetMaxVcpus */
4040 4041
    NULL, /* domainGetSecurityLabel */
    NULL, /* nodeGetSecurityModel */
4042
    testDomainDumpXML, /* domainDumpXML */
4043 4044
    NULL, /* domainXmlFromNative */
    NULL, /* domainXmlToNative */
4045 4046 4047 4048 4049 4050 4051 4052 4053 4054 4055 4056
    testListDefinedDomains, /* listDefinedDomains */
    testNumOfDefinedDomains, /* numOfDefinedDomains */
    testDomainCreate, /* domainCreate */
    testDomainDefineXML, /* domainDefineXML */
    testDomainUndefine, /* domainUndefine */
    NULL, /* domainAttachDevice */
    NULL, /* domainDetachDevice */
    testDomainGetAutostart, /* domainGetAutostart */
    testDomainSetAutostart, /* domainSetAutostart */
    testDomainGetSchedulerType, /* domainGetSchedulerType */
    testDomainGetSchedulerParams, /* domainGetSchedulerParameters */
    testDomainSetSchedulerParams, /* domainSetSchedulerParameters */
4057 4058 4059
    NULL, /* domainMigratePrepare */
    NULL, /* domainMigratePerform */
    NULL, /* domainMigrateFinish */
4060 4061
    NULL, /* domainBlockStats */
    NULL, /* domainInterfaceStats */
R
Richard W.M. Jones 已提交
4062
    NULL, /* domainBlockPeek */
R
Richard W.M. Jones 已提交
4063
    NULL, /* domainMemoryPeek */
4064
    testNodeGetCellsFreeMemory, /* nodeGetCellsFreeMemory */
4065
    NULL, /* getFreeMemory */
4066 4067
    testDomainEventRegister, /* domainEventRegister */
    testDomainEventDeregister, /* domainEventDeregister */
D
Daniel Veillard 已提交
4068 4069
    NULL, /* domainMigratePrepare2 */
    NULL, /* domainMigrateFinish2 */
4070
    NULL, /* nodeDeviceDettach */
4071 4072
    NULL, /* nodeDeviceReAttach */
    NULL, /* nodeDeviceReset */
4073 4074 4075 4076 4077 4078 4079 4080 4081 4082 4083 4084 4085 4086 4087 4088 4089 4090 4091 4092 4093 4094 4095
};

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 */
};

L
Laine Stump 已提交
4096 4097 4098 4099 4100 4101 4102 4103 4104 4105 4106 4107 4108 4109 4110 4111 4112 4113
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 */
};


4114 4115 4116 4117
static virStorageDriver testStorageDriver = {
    .name = "Test",
    .open = testStorageOpen,
    .close = testStorageClose,
C
Cole Robinson 已提交
4118 4119 4120 4121 4122 4123 4124 4125 4126 4127 4128 4129 4130 4131 4132 4133 4134 4135 4136 4137 4138 4139 4140 4141 4142 4143 4144 4145

    .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,
4146
    .volCreateXMLFrom = testStorageVolumeCreateXMLFrom,
C
Cole Robinson 已提交
4147 4148 4149 4150
    .volDelete = testStorageVolumeDelete,
    .volGetInfo = testStorageVolumeGetInfo,
    .volGetXMLDesc = testStorageVolumeGetXMLDesc,
    .volGetPath = testStorageVolumeGetPath,
4151 4152
};

4153 4154 4155 4156 4157 4158 4159 4160
static virDeviceMonitor testDevMonitor = {
    .name = "Test",
    .open = testDevMonOpen,
    .close = testDevMonClose,
};



4161 4162 4163 4164 4165 4166 4167 4168 4169 4170 4171 4172
/**
 * testRegister:
 *
 * Registers the test driver
 */
int
testRegister(void)
{
    if (virRegisterDriver(&testDriver) < 0)
        return -1;
    if (virRegisterNetworkDriver(&testNetworkDriver) < 0)
        return -1;
L
Laine Stump 已提交
4173 4174
    if (virRegisterInterfaceDriver(&testInterfaceDriver) < 0)
        return -1;
4175 4176
    if (virRegisterStorageDriver(&testStorageDriver) < 0)
        return -1;
4177 4178 4179
    if (virRegisterDeviceMonitor(&testDevMonitor) < 0)
        return -1;

4180 4181
    return 0;
}