openvz_driver.c 22.9 KB
Newer Older
1 2 3 4 5
/*
 * openvz_driver.c: core driver methods for managing OpenVZ VEs
 *
 * Copyright (C) 2006, 2007 Binary Karma
 * Copyright (C) 2006 Shuveb Hussain
6
 * Copyright (C) 2007 Anoop Joe Cyriac
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
 *
 * 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
 *
22
 * Authors:
23 24 25
 * Shuveb Hussain <shuveb@binarykarma.com>
 * Anoop Joe Cyriac <anoop@binarykarma.com>
 *
26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
 */

#ifdef WITH_OPENVZ

#include <config.h>

#include <sys/types.h>
#include <sys/poll.h>
#include <dirent.h>
#include <limits.h>
#include <string.h>
#include <stdio.h>
#include <strings.h>
#include <stdarg.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <sys/utsname.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <signal.h>
#include <paths.h>
#include <pwd.h>
#include <stdio.h>
#include <sys/wait.h>

52
#include "internal.h"
53
#include "openvz_driver.h"
54 55
#include "event.h"
#include "buf.h"
56
#include "util.h"
57
#include "openvz_conf.h"
58
#include "nodeinfo.h"
59
#include "memory.h"
60

61 62 63
#define OPENVZ_MAX_ARG 28
#define CMDBUF_LEN 1488
#define CMDOP_LEN 288
64 65 66

static virDomainPtr openvzDomainLookupByID(virConnectPtr conn, int id);
static char *openvzGetOSType(virDomainPtr dom);
67
static int openvzGetNodeInfo(virConnectPtr conn, virNodeInfoPtr nodeinfo);
68 69 70 71 72 73
static virDomainPtr openvzDomainLookupByUUID(virConnectPtr conn, const unsigned char *uuid);
static virDomainPtr openvzDomainLookupByName(virConnectPtr conn, const char *name);
static int openvzDomainGetInfo(virDomainPtr dom, virDomainInfoPtr info);
static int openvzDomainShutdown(virDomainPtr dom);
static int openvzDomainReboot(virDomainPtr dom, unsigned int flags);
static int openvzDomainCreate(virDomainPtr dom);
74 75 76 77
static virDrvOpenStatus openvzOpen(virConnectPtr conn,
                                 xmlURIPtr uri,
                                 virConnectAuthPtr auth ATTRIBUTE_UNUSED,
                                 int flags ATTRIBUTE_UNUSED);
78 79 80 81 82 83 84 85 86 87
static int openvzClose(virConnectPtr conn);
static const char *openvzGetType(virConnectPtr conn ATTRIBUTE_UNUSED);
static int openvzListDomains(virConnectPtr conn, int *ids, int nids);
static int openvzNumDomains(virConnectPtr conn);
static int openvzListDefinedDomains(virConnectPtr conn, char **const names, int nnames);
static int openvzNumDefinedDomains(virConnectPtr conn);
static int openvzStartup(void);
static int openvzShutdown(void);
static int openvzReload(void);
static int openvzActive(void);
88 89

static virDomainPtr openvzDomainDefineXML(virConnectPtr conn, const char *xml);
90
static virDomainPtr openvzDomainCreateLinux(virConnectPtr conn, const char *xml,
91 92 93 94 95 96
        unsigned int flags ATTRIBUTE_UNUSED);

static int openvzDomainUndefine(virDomainPtr dom);
static int convCmdbufExec(char cmdbuf[], char *cmdExec[]);
static void cmdExecFree(char *cmdExec[]);

97 98
struct openvz_driver ovz_driver;

99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
static int convCmdbufExec(char cmdbuf[], char *cmdExec[])
{
    int i=0, limit = OPENVZ_MAX_ARG - 1;
    char cmdWord[CMDOP_LEN];
    while(*cmdbuf)
    {
        if(i >= limit)
        {
            cmdExec[i] = NULL;
            return -1;
        }
        sscanf(cmdbuf, "%s", cmdWord);
        cmdbuf += strlen(cmdWord);
        while(*cmdbuf == ' ') cmdbuf++;
        cmdExec[i++] = strdup(cmdWord);
    }
    cmdExec[i] = NULL;
    return i;
}

static void cmdExecFree(char *cmdExec[])
{
    int i=-1;
    while(cmdExec[++i])
    {
124
        VIR_FREE(cmdExec[i]);
125 126 127
    }
}

128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196
/* generate arguments to create OpenVZ container
   return -1 - error
           0 - OK
*/
static int openvzDomainDefineCmd(virConnectPtr conn,
                                 char *args[],
                                 int maxarg,
                                 struct openvz_vm_def *vmdef)
{
    int narg;

    for (narg = 0; narg < maxarg; narg++)
        args[narg] = NULL;

    if (vmdef == NULL){
        openvzError(conn, VIR_ERR_INTERNAL_ERROR,
                   _("Container is not defined"));
        return -1;
    }

#define ADD_ARG(thisarg)                                                \
    do {                                                                \
        if (narg >= maxarg)                                             \
                 goto no_memory;                                        \
        args[narg++] = thisarg;                                         \
    } while (0)

#define ADD_ARG_LIT(thisarg)                                            \
    do {                                                                \
        if (narg >= maxarg)                                             \
                 goto no_memory;                                        \
        if ((args[narg++] = strdup(thisarg)) == NULL)                   \
            goto no_memory;                                             \
    } while (0)

    narg = 0;
    ADD_ARG_LIT(VZCTL);
    ADD_ARG_LIT("--quiet");
    ADD_ARG_LIT("create");
    ADD_ARG_LIT(vmdef->name);

    if ((vmdef->fs.tmpl && *(vmdef->fs.tmpl))) {
        ADD_ARG_LIT("--ostemplate");
        ADD_ARG_LIT(vmdef->fs.tmpl);
    }
    if ((vmdef->profile && *(vmdef->profile))) {
        ADD_ARG_LIT("--config");
        ADD_ARG_LIT(vmdef->profile);
    }
    if ((vmdef->net.ips->ip && *(vmdef->net.ips->ip))) {
        ADD_ARG_LIT("--ipadd");
        ADD_ARG_LIT(vmdef->net.ips->ip);
    }
    if ((vmdef->net.hostname && *(vmdef->net.hostname))) {
        ADD_ARG_LIT("--hostname");
        ADD_ARG_LIT(vmdef->net.hostname);
    }

    ADD_ARG(NULL);
    return 0;
 no_memory:
    openvzError(conn, VIR_ERR_INTERNAL_ERROR,
                _("Could not put argument to %s"), VZCTL);
    return -1;
#undef ADD_ARG
#undef ADD_ARG_LIT
}


197 198 199
static virDomainPtr openvzDomainLookupByID(virConnectPtr conn,
                                   int id) {
    struct openvz_driver *driver = (struct openvz_driver *)conn->privateData;
200
    struct openvz_vm *vm;
201 202
    virDomainPtr dom;

203 204
    vm = openvzFindVMByID(driver, id);

205
    if (!vm) {
206
        openvzError(conn, VIR_ERR_NO_DOMAIN, NULL);
207 208 209 210 211
        return NULL;
    }

    dom = virGetDomain(conn, vm->vmdef->name, vm->vmdef->uuid);
    if (!dom) {
D
Daniel Veillard 已提交
212
        openvzError(conn, VIR_ERR_NO_MEMORY, _("virDomainPtr"));
213 214 215 216 217 218 219
        return NULL;
    }

    dom->id = vm->vpsid;
    return dom;
}

220
static char *openvzGetOSType(virDomainPtr dom ATTRIBUTE_UNUSED)
221 222
{
    /* OpenVZ runs on Linux and runs only Linux */
223
    return strdup("linux");
224 225 226 227 228 229 230 231 232 233
}


static virDomainPtr openvzDomainLookupByUUID(virConnectPtr conn,
                                     const unsigned char *uuid) {
    struct  openvz_driver *driver = (struct openvz_driver *)conn->privateData;
    struct openvz_vm *vm = openvzFindVMByUUID(driver, uuid);
    virDomainPtr dom;

    if (!vm) {
234
        openvzError(conn, VIR_ERR_NO_DOMAIN, NULL);
235 236 237 238 239
        return NULL;
    }

    dom = virGetDomain(conn, vm->vmdef->name, vm->vmdef->uuid);
    if (!dom) {
D
Daniel Veillard 已提交
240
        openvzError(conn, VIR_ERR_NO_MEMORY, _("virDomainPtr"));
241 242 243 244 245 246 247 248 249 250 251 252 253 254
        return NULL;
    }

    dom->id = vm->vpsid;
    return dom;
}

static virDomainPtr openvzDomainLookupByName(virConnectPtr conn,
                                     const char *name) {
    struct openvz_driver *driver = (struct openvz_driver *)conn->privateData;
    struct openvz_vm *vm = openvzFindVMByName(driver, name);
    virDomainPtr dom;

    if (!vm) {
255
        openvzError(conn, VIR_ERR_NO_DOMAIN, NULL);
256 257 258 259 260
        return NULL;
    }

    dom = virGetDomain(conn, vm->vmdef->name, vm->vmdef->uuid);
    if (!dom) {
D
Daniel Veillard 已提交
261
        openvzError(conn, VIR_ERR_NO_MEMORY, _("virDomainPtr"));
262 263 264 265 266 267 268 269 270 271 272
        return NULL;
    }

    dom->id = vm->vpsid;
    return dom;
}

static int openvzDomainGetInfo(virDomainPtr dom,
                       virDomainInfoPtr info) {
    struct openvz_driver *driver = (struct openvz_driver *)dom->conn->privateData;
    struct openvz_vm *vm = openvzFindVMByUUID(driver, dom->uuid);
273

274
    if (!vm) {
D
Daniel Veillard 已提交
275
        openvzError(dom->conn, VIR_ERR_INVALID_DOMAIN,
276
              _("no domain with matching uuid"));
277 278 279 280 281 282
        return -1;
    }

    info->state = vm->status;

    /* TODO These need to be calculated differently for OpenVZ */
283
    //info->cpuTime =
284 285 286 287 288 289 290 291 292
    //info->maxMem = vm->def->maxmem;
    //info->memory = vm->def->memory;
    //info->nrVirtCpu = vm->def->vcpus;
    return 0;
}

static int openvzDomainShutdown(virDomainPtr dom) {
    struct openvz_driver *driver = (struct openvz_driver *)dom->conn->privateData;
    struct openvz_vm *vm = openvzFindVMByID(driver, dom->id);
293
    const char *prog[] = {VZCTL, "--quiet", "stop", vm->vmdef->name, NULL};
294 295

    if (!vm) {
D
Daniel Veillard 已提交
296
        openvzError(dom->conn, VIR_ERR_INVALID_DOMAIN,
297
              _("no domain with matching id"));
298 299
        return -1;
    }
300

301
    if (vm->status != VIR_DOMAIN_RUNNING) {
D
Daniel Veillard 已提交
302
        openvzError(dom->conn, VIR_ERR_OPERATION_DENIED,
303
              _("domain is not in running state"));
304 305
        return -1;
    }
306

307
    if (virRun(dom->conn, (char **)prog, NULL) < 0) {
D
Daniel Veillard 已提交
308
        openvzError(dom->conn, VIR_ERR_INTERNAL_ERROR,
309
              _("Could not exec %s"), VZCTL);
310 311
        return -1;
    }
312

313 314 315 316
    vm->vpsid = -1;
    vm->status = VIR_DOMAIN_SHUTOFF;
    ovz_driver.num_inactive ++;
    ovz_driver.num_active --;
317

318
    return 0;
319 320
}

321 322
static int openvzDomainReboot(virDomainPtr dom,
                              unsigned int flags ATTRIBUTE_UNUSED) {
323 324
    struct openvz_driver *driver = (struct openvz_driver *)dom->conn->privateData;
    struct openvz_vm *vm = openvzFindVMByID(driver, dom->id);
325
    const char *prog[] = {VZCTL, "--quiet", "restart", vm->vmdef->name, NULL};
326 327

    if (!vm) {
D
Daniel Veillard 已提交
328
        openvzError(dom->conn, VIR_ERR_INVALID_DOMAIN,
329
              _("no domain with matching id"));
330 331
        return -1;
    }
332

333
    if (vm->status != VIR_DOMAIN_RUNNING) {
D
Daniel Veillard 已提交
334
        openvzError(dom->conn, VIR_ERR_OPERATION_DENIED,
335
              _("domain is not in running state"));
336 337
        return -1;
    }
338

339
    if (virRun(dom->conn, (char **)prog, NULL) < 0) {
D
Daniel Veillard 已提交
340
        openvzError(dom->conn, VIR_ERR_INTERNAL_ERROR,
341
               _("Could not exec %s"), VZCTL);
342 343
        return -1;
    }
344

345
    return 0;
346 347
}

348 349 350 351 352 353
static virDomainPtr
openvzDomainDefineXML(virConnectPtr conn, const char *xml)
{
    struct openvz_driver *driver = (struct openvz_driver *) conn->privateData;
    struct openvz_vm_def *vmdef = NULL;
    struct openvz_vm *vm = NULL;
354 355 356
    virDomainPtr dom = NULL;
    char *prog[OPENVZ_MAX_ARG];
    prog[0] = NULL;
357

358 359
    if ((vmdef = openvzParseVMDef(conn, xml, NULL)) == NULL)
        return NULL;
360 361 362

    vm = openvzFindVMByID(driver, strtoI(vmdef->name));
    if (vm) {
363 364
        openvzLog(OPENVZ_ERR, _("Already an OPENVZ VM active with the id '%s'"),
                  vmdef->name);
365
        return NULL;
366 367 368
    }
    if (!(vm = openvzAssignVMDef(conn, driver, vmdef))) {
        openvzFreeVMDef(vmdef);
369
        openvzLog(OPENVZ_ERR, "%s", _("Error creating OPENVZ VM"));
370 371
    }

372 373 374 375
    if (openvzDomainDefineCmd(conn, prog, OPENVZ_MAX_ARG, vmdef) < 0) {
        openvzError(conn, VIR_ERR_INTERNAL_ERROR,
                _("Error creating command for container"));
        goto exit;
376 377
    }

378
    if (virRun(conn, (char **)prog, NULL) < 0) {
D
Daniel Veillard 已提交
379
        openvzError(conn, VIR_ERR_INTERNAL_ERROR,
380
               _("Could not exec %s"), VZCTL);
381
        goto exit;
382
    }
383

384 385 386
    dom = virGetDomain(conn, vm->vmdef->name, vm->vmdef->uuid);
    if (dom)
        dom->id = vm->vpsid;
387 388
 exit:
    cmdExecFree(prog);
389 390 391 392 393 394 395 396 397
    return dom;
}

static virDomainPtr
openvzDomainCreateLinux(virConnectPtr conn, const char *xml,
                        unsigned int flags ATTRIBUTE_UNUSED)
{
    struct openvz_vm_def *vmdef = NULL;
    struct openvz_vm *vm = NULL;
398
    virDomainPtr dom = NULL;
399
    struct openvz_driver *driver = (struct openvz_driver *) conn->privateData;
400 401 402
    const char *progstart[] = {VZCTL, "--quiet", "start", NULL, NULL};
    char *progcreate[OPENVZ_MAX_ARG];
    progcreate[0] = NULL;
403 404 405 406 407 408 409

    if (!(vmdef = openvzParseVMDef(conn, xml, NULL)))
        return NULL;

    vm = openvzFindVMByID(driver, strtoI(vmdef->name));
    if (vm) {
        openvzFreeVMDef(vmdef);
410 411
        openvzLog(OPENVZ_ERR,
                  _("Already an OPENVZ VM defined with the id '%d'"),
412 413 414 415
                strtoI(vmdef->name));
        return NULL;
    }
    if (!(vm = openvzAssignVMDef(conn, driver, vmdef))) {
416
        openvzLog(OPENVZ_ERR, "%s", _("Error creating OPENVZ VM"));
417 418 419
        return NULL;
    }

420 421 422 423
    if (openvzDomainDefineCmd(conn, progcreate, OPENVZ_MAX_ARG, vmdef) < 0) {
        openvzError(conn, VIR_ERR_INTERNAL_ERROR,
                _("Error creating command for container"));
        goto exit;
424 425
    }

426
    if (virRun(conn, (char **)progcreate, NULL) < 0) {
D
Daniel Veillard 已提交
427
        openvzError(conn, VIR_ERR_INTERNAL_ERROR,
428
               _("Could not exec %s"), VZCTL);
429
        goto exit;
430
    }
431

432
    progstart[3] = vmdef->name;
433

434
    if (virRun(conn, (char **)progstart, NULL) < 0) {
D
Daniel Veillard 已提交
435
        openvzError(conn, VIR_ERR_INTERNAL_ERROR,
436
               _("Could not exec %s"), VZCTL);
437
        goto exit;
438
    }
439

440 441 442 443 444 445 446 447
    sscanf(vmdef->name, "%d", &vm->vpsid);
    vm->status = VIR_DOMAIN_RUNNING;
    ovz_driver.num_inactive--;
    ovz_driver.num_active++;

    dom = virGetDomain(conn, vm->vmdef->name, vm->vmdef->uuid);
    if (dom)
        dom->id = vm->vpsid;
448 449
 exit:
    cmdExecFree(progcreate);
450 451 452 453 454 455
    return dom;
}

static int
openvzDomainCreate(virDomainPtr dom)
{
456
    struct openvz_driver *driver = (struct openvz_driver *)dom->conn->privateData;
457
    struct openvz_vm *vm = openvzFindVMByName(driver, dom->name);
458
    const char *prog[] = {VZCTL, "--quiet", "start", vm->vmdef->name, NULL };
459 460

    if (!vm) {
D
Daniel Veillard 已提交
461
        openvzError(dom->conn, VIR_ERR_INVALID_DOMAIN,
462
              _("no domain with matching id"));
463 464
        return -1;
    }
465

466
    if (vm->status != VIR_DOMAIN_SHUTOFF) {
D
Daniel Veillard 已提交
467
        openvzError(dom->conn, VIR_ERR_OPERATION_DENIED,
468
              _("domain is not in shutoff state"));
469 470 471
        return -1;
    }

472
    if (virRun(dom->conn, (char **)prog, NULL) < 0) {
D
Daniel Veillard 已提交
473
        openvzError(dom->conn, VIR_ERR_INTERNAL_ERROR,
474
               _("Could not exec %s"), VZCTL);
475 476
        return -1;
    }
477

478
    sscanf(vm->vmdef->name, "%d", &vm->vpsid);
479 480 481
    vm->status = VIR_DOMAIN_RUNNING;
    ovz_driver.num_inactive --;
    ovz_driver.num_active ++;
482

483
    return 0;
484 485 486 487 488 489 490 491
}

static int
openvzDomainUndefine(virDomainPtr dom)
{
    virConnectPtr conn= dom->conn;
    struct openvz_driver *driver = (struct openvz_driver *) conn->privateData;
    struct openvz_vm *vm = openvzFindVMByUUID(driver, dom->uuid);
492
    const char *prog[] = { VZCTL, "--quiet", "destroy", vm->vmdef->name, NULL };
493 494

    if (!vm) {
D
Daniel Veillard 已提交
495
        openvzError(conn, VIR_ERR_INVALID_DOMAIN, _("no domain with matching uuid"));
496 497
        return -1;
    }
498

499
    if (openvzIsActiveVM(vm)) {
D
Daniel Veillard 已提交
500
        openvzError(conn, VIR_ERR_INTERNAL_ERROR, _("cannot delete active domain"));
501 502
        return -1;
    }
503

504
    if (virRun(conn, (char **)prog, NULL) < 0) {
D
Daniel Veillard 已提交
505
        openvzError(conn, VIR_ERR_INTERNAL_ERROR,
506
               _("Could not exec %s"), VZCTL);
507 508
        return -1;
    }
509

510
    openvzRemoveInactiveVM(driver, vm);
511
    return 0;
512 513
}

514 515 516 517 518 519
static int
openvzDomainSetAutostart(virDomainPtr dom, int autostart)
{
    virConnectPtr conn= dom->conn;
    struct openvz_driver *driver = (struct openvz_driver *) conn->privateData;
    struct openvz_vm *vm = openvzFindVMByUUID(driver, dom->uuid);
520
    const char *prog[] = { VZCTL, "--quiet", "set", vm->vmdef->name,
D
Daniel Veillard 已提交
521
                           "--onboot", autostart ? "yes" : "no",
522 523 524
                           "--save", NULL };

    if (!vm) {
D
Daniel Veillard 已提交
525
        openvzError(conn, VIR_ERR_INVALID_DOMAIN, _("no domain with matching uuid"));
526 527 528 529
        return -1;
    }

    if (virRun(conn, (char **)prog, NULL) < 0) {
D
Daniel Veillard 已提交
530
        openvzError(conn, VIR_ERR_INTERNAL_ERROR, _("Could not exec %s"), VZCTL);
531 532 533 534 535 536 537 538 539 540 541 542 543 544 545
        return -1;
    }

    return 0;
}

static int
openvzDomainGetAutostart(virDomainPtr dom, int *autostart)
{
    virConnectPtr conn= dom->conn;
    struct openvz_driver *driver = (struct openvz_driver *) conn->privateData;
    struct openvz_vm *vm = openvzFindVMByUUID(driver, dom->uuid);
    char value[1024];

    if (!vm) {
D
Daniel Veillard 已提交
546
        openvzError(conn, VIR_ERR_INVALID_DOMAIN, _("no domain with matching uuid"));
547 548 549 550
        return -1;
    }

    if (openvzReadConfigParam(vm->vpsid , "ONBOOT", value, sizeof(value)) < 0) {
D
Daniel Veillard 已提交
551
        openvzError(conn, VIR_ERR_INTERNAL_ERROR, _("Cound not read container config"));
552 553 554 555 556
        return -1;
    }

    *autostart = 0;
    if (STREQ(value,"yes"))
D
Daniel Veillard 已提交
557
        *autostart = 1;
558 559 560 561

    return 0;
}

562 563 564 565
static const char *openvzProbe(void)
{
#ifdef __linux__
    if ((getuid() == 0) && (virFileExists("/proc/vz")))
D
Daniel Veillard 已提交
566
        return("openvz:///system");
567 568 569 570
#endif
    return(NULL);
}

571
static virDrvOpenStatus openvzOpen(virConnectPtr conn,
572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588
                                 xmlURIPtr uri,
                                 virConnectAuthPtr auth ATTRIBUTE_UNUSED,
                                 int flags ATTRIBUTE_UNUSED)
{
   struct openvz_vm *vms;

    /*Just check if the user is root. Nothing really to open for OpenVZ */
   if (getuid()) { // OpenVZ tools can only be used by r00t
           return VIR_DRV_OPEN_DECLINED;
   } else {
       if (uri == NULL || uri->scheme == NULL || uri->path == NULL)
                   return VIR_DRV_OPEN_DECLINED;
       if (STRNEQ (uri->scheme, "openvz"))
                   return VIR_DRV_OPEN_DECLINED;
       if (STRNEQ (uri->path, "/system"))
                   return VIR_DRV_OPEN_DECLINED;
   }
589
    /* See if we are running an OpenVZ enabled kernel */
590 591 592 593
   if(access("/proc/vz/veinfo", F_OK) == -1 ||
               access("/proc/user_beancounters", F_OK) == -1) {
       return VIR_DRV_OPEN_DECLINED;
   }
594

595
   conn->privateData = &ovz_driver;
596

597 598 599
   virStateInitialize();
   vms = openvzGetVPSInfo(conn);
   ovz_driver.vms = vms;
600

601 602
   return VIR_DRV_OPEN_SUCCESS;
};
603 604

static int openvzClose(virConnectPtr conn) {
605

606 607 608 609 610 611 612 613 614 615 616 617 618
    struct openvz_driver *driver = (struct openvz_driver *)conn->privateData;
    struct openvz_vm *vm = driver->vms;

    while(vm) {
        openvzFreeVMDef(vm->vmdef);
        vm = vm->next;
    }
    vm = driver->vms;
    while (vm) {
        struct openvz_vm *prev = vm;
        vm = vm->next;
        free(prev);
    }
619

620 621 622 623 624 625 626 627 628
    conn->privateData = NULL;

    return 0;
}

static const char *openvzGetType(virConnectPtr conn ATTRIBUTE_UNUSED) {
    return strdup("OpenVZ");
}

629 630 631 632 633
static int openvzGetNodeInfo(virConnectPtr conn,
                             virNodeInfoPtr nodeinfo) {
    return virNodeInfoPopulate(conn, nodeinfo);
}

634 635
static int openvzListDomains(virConnectPtr conn, int *ids, int nids) {
    int got = 0;
636 637 638 639
    int veid, pid, outfd, errfd;
    int ret;
    char buf[32];
    const char *cmd[] = {VZLIST, "-ovpsid", "-H" , NULL};
640

641
    ret = virExec(conn, (char **)cmd, &pid, -1, &outfd, &errfd);
642
    if(ret == -1) {
D
Daniel Veillard 已提交
643
        openvzError(conn, VIR_ERR_INTERNAL_ERROR,
644
               _("Could not exec %s"), VZLIST);
D
Daniel P. Berrange 已提交
645
        return -1;
646 647
    }

648 649 650 651
    while(got < nids){
        ret = openvz_readline(outfd, buf, 32);
        if(!ret) break;
        sscanf(buf, "%d", &veid);
652 653 654
        ids[got] = veid;
        got ++;
    }
655
    waitpid(pid, NULL, 0);
656 657 658 659

    return got;
}

660
static int openvzNumDomains(virConnectPtr conn ATTRIBUTE_UNUSED) {
661 662 663 664 665 666
    return ovz_driver.num_active;
}

static int openvzListDefinedDomains(virConnectPtr conn,
                            char **const names, int nnames) {
    int got = 0;
667
    int veid, pid, outfd, errfd, ret;
668
    char vpsname[OPENVZ_NAME_MAX];
669
    char buf[32];
670
    const char *cmd[] = {VZLIST, "-ovpsid", "-H", "-S", NULL};
671 672

    /* the -S options lists only stopped domains */
673
    ret = virExec(conn, (char **)cmd, &pid, -1, &outfd, &errfd);
674
    if(ret == -1) {
D
Daniel Veillard 已提交
675
        openvzError(conn, VIR_ERR_INTERNAL_ERROR,
676
               _("Could not exec %s"), VZLIST);
D
Daniel P. Berrange 已提交
677
        return -1;
678 679
    }

680 681 682 683
    while(got < nnames){
        ret = openvz_readline(outfd, buf, 32);
        if(!ret) break;
        sscanf(buf, "%d\n", &veid);
684 685 686 687
        sprintf(vpsname, "%d", veid);
        names[got] = strdup(vpsname);
        got ++;
    }
688
    waitpid(pid, NULL, 0);
689 690 691
    return got;
}

692
static int openvzNumDefinedDomains(virConnectPtr conn ATTRIBUTE_UNUSED) {
693
    return ovz_driver.num_inactive;
694 695 696 697
}

static int openvzStartup(void) {
    openvzAssignUUIDs();
698

699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720
    return 0;
}

static int openvzShutdown(void) {

    return 0;
}

static int openvzReload(void) {

    return 0;
}

static int openvzActive(void) {

    return 1;
}

static virDriver openvzDriver = {
    VIR_DRV_OPENVZ,
    "OPENVZ",
    LIBVIR_VERSION_NUMBER,
721
    openvzProbe, /* probe */
722 723
    openvzOpen, /* open */
    openvzClose, /* close */
724
    NULL, /* supports_feature */
725 726 727 728 729
    openvzGetType, /* type */
    NULL, /* version */
    NULL, /* hostname */
    NULL, /* uri */
    NULL, /* getMaxVcpus */
730
    openvzGetNodeInfo, /* nodeGetInfo */
731 732 733
    NULL, /* getCapabilities */
    openvzListDomains, /* listDomains */
    openvzNumDomains, /* numOfDomains */
734
    openvzDomainCreateLinux, /* domainCreateLinux */
735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758
    openvzDomainLookupByID, /* domainLookupByID */
    openvzDomainLookupByUUID, /* domainLookupByUUID */
    openvzDomainLookupByName, /* domainLookupByName */
    NULL, /* domainSuspend */
    NULL, /* domainResume */
    openvzDomainShutdown, /* domainShutdown */
    openvzDomainReboot, /* domainReboot */
    openvzDomainShutdown, /* domainDestroy */
    openvzGetOSType, /* domainGetOSType */
    NULL, /* domainGetMaxMemory */
    NULL, /* domainSetMaxMemory */
    NULL, /* domainSetMemory */
    openvzDomainGetInfo, /* domainGetInfo */
    NULL, /* domainSave */
    NULL, /* domainRestore */
    NULL, /* domainCoreDump */
    NULL, /* domainSetVcpus */
    NULL, /* domainPinVcpu */
    NULL, /* domainGetVcpus */
    NULL, /* domainGetMaxVcpus */
    NULL, /* domainDumpXML */
    openvzListDefinedDomains, /* listDomains */
    openvzNumDefinedDomains, /* numOfDomains */
    openvzDomainCreate, /* domainCreate */
759 760
    openvzDomainDefineXML, /* domainDefineXML */
    openvzDomainUndefine, /* domainUndefine */
761 762
    NULL, /* domainAttachDevice */
    NULL, /* domainDetachDevice */
763 764
    openvzDomainGetAutostart, /* domainGetAutostart */
    openvzDomainSetAutostart, /* domainSetAutostart */
765 766 767
    NULL, /* domainGetSchedulerType */
    NULL, /* domainGetSchedulerParameters */
    NULL, /* domainSetSchedulerParameters */
768 769 770 771 772
    NULL, /* domainMigratePrepare */
    NULL, /* domainMigratePerform */
    NULL, /* domainMigrateFinish */
    NULL, /* domainBlockStats */
    NULL, /* domainInterfaceStats */
D
Daniel P. Berrange 已提交
773 774
    NULL, /* domainBlockPeek */
    NULL, /* domainMemoryPeek */
775
    NULL, /* nodeGetCellsFreeMemory */
776
    NULL, /* nodeGetFreeMemory */
777 778 779 780 781 782 783
};

static virStateDriver openvzStateDriver = {
    openvzStartup,
    openvzShutdown,
    openvzReload,
    openvzActive,
D
Daniel P. Berrange 已提交
784
    NULL, /* sigHandler */
785 786 787 788 789 790 791 792 793
};

int openvzRegister(void) {
    virRegisterDriver(&openvzDriver);
    virRegisterStateDriver(&openvzStateDriver);
    return 0;
}

#endif /* WITH_OPENVZ */