uml_driver.c 51.9 KB
Newer Older
1 2 3
/*
 * uml_driver.c: core driver methods for managing UML guests
 *
4
 * Copyright (C) 2006, 2007, 2008, 2009 Red Hat, Inc.
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
 * Copyright (C) 2006-2008 Daniel P. Berrange
 *
 * 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
 *
 * Author: Daniel P. Berrange <berrange@redhat.com>
 */

#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>
#include <sys/ioctl.h>
#include <sys/inotify.h>
47
#include <sys/un.h>
48 49 50 51 52 53 54 55 56 57 58 59 60

#include "uml_driver.h"
#include "uml_conf.h"
#include "event.h"
#include "buf.h"
#include "util.h"
#include "nodeinfo.h"
#include "stats_linux.h"
#include "capabilities.h"
#include "memory.h"
#include "uuid.h"
#include "domain_conf.h"
#include "datatypes.h"
61
#include "logging.h"
62

63 64
#define VIR_FROM_THIS VIR_FROM_UML

65 66 67
/* For storing short-lived temporary files. */
#define TEMPDIR LOCAL_STATE_DIR "/cache/libvirt"

68 69 70 71 72 73 74 75
typedef struct _umlDomainObjPrivate umlDomainObjPrivate;
typedef umlDomainObjPrivate *umlDomainObjPrivatePtr;
struct _umlDomainObjPrivate {
    int monitor;
    int monitorWatch;
};


76 77
static int umlShutdown(void);

78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
static void *umlDomainObjPrivateAlloc(void)
{
    umlDomainObjPrivatePtr priv;

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

    priv->monitor = -1;
    priv->monitorWatch = -1;

    return priv;
}

static void umlDomainObjPrivateFree(void *data)
{
    umlDomainObjPrivatePtr priv = data;

    VIR_FREE(priv);
}


99 100
static void umlDriverLock(struct uml_driver *driver)
{
101
    virMutexLock(&driver->lock);
102 103 104
}
static void umlDriverUnlock(struct uml_driver *driver)
{
105
    virMutexUnlock(&driver->lock);
106 107
}

108 109 110 111

static int umlOpenMonitor(virConnectPtr conn,
                          struct uml_driver *driver,
                          virDomainObjPtr vm);
112
static int umlReadPidFile(struct uml_driver *driver,
113 114 115 116 117 118 119 120 121 122 123
                          virDomainObjPtr vm);

static int umlSetCloseExec(int fd) {
    int flags;
    if ((flags = fcntl(fd, F_GETFD)) < 0)
        goto error;
    flags |= FD_CLOEXEC;
    if ((fcntl(fd, F_SETFD, flags)) < 0)
        goto error;
    return 0;
 error:
D
Daniel Veillard 已提交
124
    VIR_ERROR0(_("Failed to set close-on-exec file descriptor flag"));
125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144
    return -1;
}

static int umlStartVMDaemon(virConnectPtr conn,
                            struct uml_driver *driver,
                            virDomainObjPtr vm);

static void umlShutdownVMDaemon(virConnectPtr conn,
                                struct uml_driver *driver,
                                virDomainObjPtr vm);


static int umlMonitorCommand (virConnectPtr conn,
                              const struct uml_driver *driver,
                              const virDomainObjPtr vm,
                              const char *cmd,
                              char **reply);

static struct uml_driver *uml_driver = NULL;

145 146 147 148 149 150 151 152 153 154 155 156 157
struct umlAutostartData {
    struct uml_driver *driver;
    virConnectPtr conn;
};

static void
umlAutostartDomain(void *payload, const char *name ATTRIBUTE_UNUSED, void *opaque)
{
    virDomainObjPtr vm = payload;
    const struct umlAutostartData *data = opaque;

    virDomainObjLock(vm);
    if (vm->autostart &&
D
Daniel P. Berrange 已提交
158
        !virDomainObjIsActive(vm)) {
159 160 161 162 163 164 165 166 167
        virResetLastError();
        if (umlStartVMDaemon(data->conn, data->driver, vm) < 0) {
            virErrorPtr err = virGetLastError();
            VIR_ERROR(_("Failed to autostart VM '%s': %s"),
                      vm->def->name, err->message);
        }
    }
    virDomainObjUnlock(vm);
}
168 169 170

static void
umlAutostartConfigs(struct uml_driver *driver) {
171 172 173 174 175
    /* XXX: Figure out a better way todo this. The domain
     * startup code needs a connection handle in order
     * to lookup the bridge associated with a virtual
     * network
     */
176 177 178
    virConnectPtr conn = virConnectOpen(driver->privileged ?
                                        "uml:///system" :
                                        "uml:///session");
179
    /* Ignoring NULL conn which is mostly harmless here */
180

181 182 183
    struct umlAutostartData data = { driver, conn };

    virHashForEach(driver->domains.objs, umlAutostartDomain, &data);
184

185 186
    if (conn)
        virConnectClose(conn);
187 188 189 190
}


static int
191
umlIdentifyOneChrPTY(struct uml_driver *driver,
192 193 194 195 196 197 198
                     virDomainObjPtr dom,
                     virDomainChrDefPtr def,
                     const char *dev)
{
    char *cmd;
    char *res = NULL;
    int retries = 0;
199
    if (virAsprintf(&cmd, "config %s%d", dev, def->target.port) < 0) {
200
        virReportOOMError();
201 202 203
        return -1;
    }
requery:
204 205
    if (umlMonitorCommand(NULL, driver, dom, cmd, &res) < 0)
        return -1;
206

207
    if (res && STRPREFIX(res, "pts:")) {
208 209
        VIR_FREE(def->data.file.path);
        if ((def->data.file.path = strdup(res + 4)) == NULL) {
210
            virReportOOMError();
211 212 213 214
            VIR_FREE(res);
            VIR_FREE(cmd);
            return -1;
        }
215
    } else if (!res || STRPREFIX(res, "pts")) {
216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231
        /* It can take a while to startup, so retry for
           upto 5 seconds */
        /* XXX should do this in a better non-blocking
           way somehow ...perhaps register a timer */
        if (retries++ < 50) {
            usleep(1000*10);
            goto requery;
        }
    }

    VIR_FREE(cmd);
    VIR_FREE(res);
    return 0;
}

static int
232
umlIdentifyChrPTY(struct uml_driver *driver,
233 234 235 236 237 238
                  virDomainObjPtr dom)
{
    int i;

    if (dom->def->console &&
        dom->def->console->type == VIR_DOMAIN_CHR_TYPE_PTY)
239
        if (umlIdentifyOneChrPTY(driver, dom,
240 241 242 243 244
                                 dom->def->console, "con") < 0)
            return -1;

    for (i = 0 ; i < dom->def->nserials; i++)
        if (dom->def->serials[i]->type == VIR_DOMAIN_CHR_TYPE_PTY &&
245
            umlIdentifyOneChrPTY(driver, dom,
246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264
                                 dom->def->serials[i], "ssl") < 0)
            return -1;

    return 0;
}

static void
umlInotifyEvent(int watch,
                int fd,
                int events ATTRIBUTE_UNUSED,
                void *data)
{
    char buf[1024];
    struct inotify_event *e;
    int got;
    char *tmp, *name;
    struct uml_driver *driver = data;
    virDomainObjPtr dom;

265
    umlDriverLock(driver);
266
    if (watch != driver->inotifyWatch)
267
        goto cleanup;
268 269 270 271 272 273

reread:
    got = read(fd, buf, sizeof(buf));
    if (got == -1) {
        if (errno == EINTR)
            goto reread;
274
        goto cleanup;
275 276 277 278 279
    }

    tmp = buf;
    while (got) {
        if (got < sizeof(struct inotify_event))
280
            goto cleanup; /* bad */
281 282 283 284 285 286

        e = (struct inotify_event *)tmp;
        tmp += sizeof(struct inotify_event);
        got -= sizeof(struct inotify_event);

        if (got < e->len)
287
            goto cleanup;
288 289 290 291 292 293 294 295 296 297 298 299 300

        tmp += e->len;
        got -= e->len;

        name = (char *)&(e->name);

        dom = virDomainFindByName(&driver->domains, name);

        if (!dom) {
            continue;
        }

        if (e->mask & IN_DELETE) {
301
            VIR_DEBUG("Got inotify domain shutdown '%s'", name);
D
Daniel P. Berrange 已提交
302
            if (!virDomainObjIsActive(dom)) {
303
                virDomainObjUnlock(dom);
304 305 306
                continue;
            }

307
            umlShutdownVMDaemon(NULL, driver, dom);
308
        } else if (e->mask & (IN_CREATE | IN_MODIFY)) {
309
            VIR_DEBUG("Got inotify domain startup '%s'", name);
D
Daniel P. Berrange 已提交
310
            if (virDomainObjIsActive(dom)) {
311
                virDomainObjUnlock(dom);
312 313 314
                continue;
            }

315
            if (umlReadPidFile(driver, dom) < 0) {
316
                virDomainObjUnlock(dom);
317 318 319 320 321 322
                continue;
            }

            dom->def->id = driver->nextvmid++;
            dom->state = VIR_DOMAIN_RUNNING;

323 324
            if (umlOpenMonitor(NULL, driver, dom) < 0) {
                VIR_WARN0("Could not open monitor for new domain");
325
                umlShutdownVMDaemon(NULL, driver, dom);
326
            } else if (umlIdentifyChrPTY(driver, dom) < 0) {
327
                VIR_WARN0("Could not identify charater devices for new domain");
328
                umlShutdownVMDaemon(NULL, driver, dom);
329
            }
330
        }
331
        virDomainObjUnlock(dom);
332
    }
333 334 335

cleanup:
    umlDriverUnlock(driver);
336 337 338 339 340 341 342 343
}

/**
 * umlStartup:
 *
 * Initialization function for the Uml daemon
 */
static int
344
umlStartup(int privileged) {
345 346 347
    uid_t uid = geteuid();
    char *base = NULL;
    char driverConf[PATH_MAX];
348
    char *userdir = NULL;
349 350 351 352

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

353 354
    uml_driver->privileged = privileged;

355 356 357 358
    if (virMutexInit(&uml_driver->lock) < 0) {
        VIR_FREE(uml_driver);
        return -1;
    }
359 360
    umlDriverLock(uml_driver);

361 362
    /* Don't have a dom0 so start from 1 */
    uml_driver->nextvmid = 1;
363
    uml_driver->inotifyWatch = -1;
364

365 366 367
    if (virDomainObjListInit(&uml_driver->domains) < 0)
        goto error;

368
    userdir = virGetUserDirectory(uid);
369
    if (!userdir)
370
        goto error;
371

372
    if (privileged) {
373 374
        if (virAsprintf(&uml_driver->logDir,
                        "%s/log/libvirt/uml", LOCAL_STATE_DIR) == -1)
375 376 377 378 379
            goto out_of_memory;

        if ((base = strdup (SYSCONF_DIR "/libvirt")) == NULL)
            goto out_of_memory;
    } else {
380

381
        if (virAsprintf(&uml_driver->logDir,
382
                        "%s/.libvirt/uml/log", userdir) == -1)
383 384
            goto out_of_memory;

385
        if (virAsprintf(&base, "%s/.libvirt", userdir) == -1)
386 387 388
            goto out_of_memory;
    }

389
    if (virAsprintf(&uml_driver->monitorDir,
390
                    "%s/.uml", userdir) == -1)
391 392 393 394 395 396 397 398 399
        goto out_of_memory;

    /* Configuration paths are either ~/.libvirt/uml/... (session) or
     * /etc/libvirt/uml/... (system).
     */
    if (snprintf (driverConf, sizeof(driverConf), "%s/uml.conf", base) == -1)
        goto out_of_memory;
    driverConf[sizeof(driverConf)-1] = '\0';

400
    if (virAsprintf(&uml_driver->configDir, "%s/uml", base) == -1)
401 402
        goto out_of_memory;

403
    if (virAsprintf(&uml_driver->autostartDir, "%s/uml/autostart", base) == -1)
404 405 406 407 408 409 410
        goto out_of_memory;

    VIR_FREE(base);

    if ((uml_driver->caps = umlCapsInit()) == NULL)
        goto out_of_memory;

411 412
    uml_driver->caps->privateDataAllocFunc = umlDomainObjPrivateAlloc;
    uml_driver->caps->privateDataFreeFunc = umlDomainObjPrivateFree;
413 414

    if ((uml_driver->inotifyFD = inotify_init()) < 0) {
D
Daniel Veillard 已提交
415
        VIR_ERROR0(_("cannot initialize inotify"));
416
        goto error;
417 418
    }

L
Laine Stump 已提交
419
    if (virFileMakePath(uml_driver->monitorDir) != 0) {
420
        char ebuf[1024];
D
Daniel Veillard 已提交
421
        VIR_ERROR(_("Failed to create monitor directory %s: %s"),
422
               uml_driver->monitorDir, virStrerror(errno, ebuf, sizeof ebuf));
423
        goto error;
424 425
    }

426
    VIR_INFO("Adding inotify watch on %s", uml_driver->monitorDir);
427 428 429
    if (inotify_add_watch(uml_driver->inotifyFD,
                          uml_driver->monitorDir,
                          IN_CREATE | IN_MODIFY | IN_DELETE) < 0) {
430
        goto error;
431 432
    }

433 434
    if ((uml_driver->inotifyWatch =
         virEventAddHandle(uml_driver->inotifyFD, POLLIN,
435 436
                           umlInotifyEvent, uml_driver, NULL)) < 0)
        goto error;
437

438
    if (virDomainLoadAllConfigs(uml_driver->caps,
439 440 441
                                &uml_driver->domains,
                                uml_driver->configDir,
                                uml_driver->autostartDir,
442
                                0, NULL, NULL) < 0)
443 444
        goto error;

445 446
    umlAutostartConfigs(uml_driver);

447
    umlDriverUnlock(uml_driver);
448 449
    VIR_FREE(userdir);

450 451
    return 0;

452
out_of_memory:
D
Daniel Veillard 已提交
453
    VIR_ERROR0(_("umlStartup: out of memory"));
454 455

error:
456
    VIR_FREE(userdir);
457
    VIR_FREE(base);
458 459
    umlDriverUnlock(uml_driver);
    umlShutdown();
460 461 462 463 464 465 466 467 468 469 470 471 472 473
    return -1;
}

/**
 * umlReload:
 *
 * Function to restart the Uml daemon, it will recheck the configuration
 * files and update its state and the networking
 */
static int
umlReload(void) {
    if (!uml_driver)
        return 0;

474
    umlDriverLock(uml_driver);
475
    virDomainLoadAllConfigs(uml_driver->caps,
476 477 478
                            &uml_driver->domains,
                            uml_driver->configDir,
                            uml_driver->autostartDir,
479
                            0, NULL, NULL);
480 481

    umlAutostartConfigs(uml_driver);
482
    umlDriverUnlock(uml_driver);
483 484 485 486 487 488 489 490 491 492 493 494 495 496

    return 0;
}

/**
 * umlActive:
 *
 * Checks if the Uml daemon is active, i.e. has an active domain or
 * an active network
 *
 * Returns 1 if active, 0 otherwise
 */
static int
umlActive(void) {
497
    int active = 0;
498 499 500 501

    if (!uml_driver)
        return 0;

502
    umlDriverLock(uml_driver);
503
    active = virDomainObjListNumOfDomains(&uml_driver->domains, 1);
504
    umlDriverUnlock(uml_driver);
505

506
    return active;
507 508
}

509 510 511 512 513 514 515
static void
umlShutdownOneVM(void *payload, const char *name ATTRIBUTE_UNUSED, void *opaque)
{
    virDomainObjPtr dom = payload;
    struct uml_driver *driver = opaque;

    virDomainObjLock(dom);
D
Daniel P. Berrange 已提交
516
    if (virDomainObjIsActive(dom))
517 518 519 520
        umlShutdownVMDaemon(NULL, driver, dom);
    virDomainObjUnlock(dom);
}

521 522 523 524 525 526 527 528 529 530
/**
 * umlShutdown:
 *
 * Shutdown the Uml daemon, it will stop all active domains and networks
 */
static int
umlShutdown(void) {
    if (!uml_driver)
        return -1;

531
    umlDriverLock(uml_driver);
532 533
    if (uml_driver->inotifyWatch != -1)
        virEventRemoveHandle(uml_driver->inotifyWatch);
534 535 536
    close(uml_driver->inotifyFD);
    virCapabilitiesFree(uml_driver->caps);

537 538 539
    /* shutdown active VMs
     * XXX allow them to stay around & reconnect */
    virHashForEach(uml_driver->domains.objs, umlShutdownOneVM, uml_driver);
540

541
    virDomainObjListDeinit(&uml_driver->domains);
542 543 544 545 546 547 548 549 550

    VIR_FREE(uml_driver->logDir);
    VIR_FREE(uml_driver->configDir);
    VIR_FREE(uml_driver->autostartDir);
    VIR_FREE(uml_driver->monitorDir);

    if (uml_driver->brctl)
        brShutdown(uml_driver->brctl);

551
    umlDriverUnlock(uml_driver);
552
    virMutexDestroy(&uml_driver->lock);
553 554 555 556 557 558
    VIR_FREE(uml_driver);

    return 0;
}


559
static int umlReadPidFile(struct uml_driver *driver,
560 561 562 563 564 565 566 567
                          virDomainObjPtr vm)
{
    int rc = -1;
    FILE *file;
    char *pidfile = NULL;
    int retries = 0;

    vm->pid = -1;
568 569
    if (virAsprintf(&pidfile, "%s/%s/pid",
                    driver->monitorDir, vm->def->name) < 0) {
570
        virReportOOMError();
571 572 573 574 575 576 577 578 579 580 581 582 583 584 585
        return -1;
    }

reopen:
    if (!(file = fopen(pidfile, "r"))) {
        if (errno == ENOENT &&
            retries++ < 50) {
            usleep(1000 * 100);
            goto reopen;
        }
        goto cleanup;
    }

    if (fscanf(file, "%d", &vm->pid) != 1) {
        errno = EINVAL;
586
        fclose(file);
587 588 589 590 591 592 593 594 595 596
        goto cleanup;
    }

    if (fclose(file) < 0)
        goto cleanup;

    rc = 0;

 cleanup:
    if (rc != 0)
597
        virReportSystemError(errno,
598 599
                             _("failed to read pid: %s"),
                             pidfile);
600 601 602 603 604 605 606 607 608
    VIR_FREE(pidfile);
    return rc;
}

static int umlMonitorAddress(virConnectPtr conn,
                             const struct uml_driver *driver,
                             virDomainObjPtr vm,
                             struct sockaddr_un *addr) {
    char *sockname;
C
Chris Lalancette 已提交
609
    int retval = 0;
610

611 612
    if (virAsprintf(&sockname, "%s/%s/mconsole",
                    driver->monitorDir, vm->def->name) < 0) {
613
        virReportOOMError();
614 615 616 617 618
        return -1;
    }

    memset(addr, 0, sizeof *addr);
    addr->sun_family = AF_UNIX;
C
Chris Lalancette 已提交
619 620 621 622 623
    if (virStrcpyStatic(addr->sun_path, sockname) == NULL) {
        umlReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                       _("Unix path %s too long for destination"), sockname);
        retval = -1;
    }
624
    VIR_FREE(sockname);
C
Chris Lalancette 已提交
625
    return retval;
626 627 628 629 630 631 632 633
}

static int umlOpenMonitor(virConnectPtr conn,
                          struct uml_driver *driver,
                          virDomainObjPtr vm) {
    struct sockaddr_un addr;
    struct stat sb;
    int retries = 0;
634
    umlDomainObjPrivatePtr priv = vm->privateData;
635 636 637 638

    if (umlMonitorAddress(conn, driver, vm, &addr) < 0)
        return -1;

639
    VIR_DEBUG("Dest address for monitor is '%s'", addr.sun_path);
640 641 642
restat:
    if (stat(addr.sun_path, &sb) < 0) {
        if (errno == ENOENT &&
643
            retries++ < 50) {
644 645 646 647 648 649
            usleep(1000 * 100);
            goto restat;
        }
        return -1;
    }

650
    if ((priv->monitor = socket(PF_UNIX, SOCK_DGRAM, 0)) < 0) {
651
        virReportSystemError(errno,
652
                             "%s", _("cannot open socket"));
653 654 655 656
        return -1;
    }

    memset(addr.sun_path, 0, sizeof addr.sun_path);
657 658
    sprintf(addr.sun_path + 1, "libvirt-uml-%u", vm->pid);
    VIR_DEBUG("Reply address for monitor is '%s'", addr.sun_path+1);
659
    if (bind(priv->monitor, (struct sockaddr *)&addr, sizeof addr) < 0) {
660
        virReportSystemError(errno,
661
                             "%s", _("cannot bind socket"));
662 663
        close(priv->monitor);
        priv->monitor = -1;
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 691 692 693 694 695 696 697 698 699 700 701
        return -1;
    }

    return 0;
}


#define MONITOR_MAGIC 0xcafebabe
#define MONITOR_BUFLEN 512
#define MONITOR_VERSION 2

struct monitor_request {
    uint32_t magic;
    uint32_t version;
    uint32_t length;
    char data[MONITOR_BUFLEN];
};

struct monitor_response {
    uint32_t error;
    uint32_t extra;
    uint32_t length;
    char data[MONITOR_BUFLEN];
};


static int umlMonitorCommand(virConnectPtr conn,
                             const struct uml_driver *driver,
                             const virDomainObjPtr vm,
                             const char *cmd,
                             char **reply)
{
    struct monitor_request req;
    struct monitor_response res;
    char *retdata = NULL;
    int retlen = 0, ret = 0;
    struct sockaddr_un addr;
    unsigned int addrlen;
702
    umlDomainObjPrivatePtr priv = vm->privateData;
703

704 705
    VIR_DEBUG("Run command '%s'", cmd);

706 707 708 709 710 711 712 713 714 715
    *reply = NULL;

    if (umlMonitorAddress(conn, driver, vm, &addr) < 0)
        return -1;

    memset(&req, 0, sizeof(req));
    req.magic = MONITOR_MAGIC;
    req.version = MONITOR_VERSION;
    req.length = strlen(cmd);
    if (req.length > (MONITOR_BUFLEN-1)) {
716
        virReportSystemError(EINVAL,
717 718
                             _("cannot send too long command %s (%d bytes)"),
                             cmd, req.length);
719 720
        return -1;
    }
C
Chris Lalancette 已提交
721 722 723 724 725
    if (virStrcpyStatic(req.data, cmd) == NULL) {
        umlReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                       _("Command %s too long for destination"), cmd);
        return -1;
    }
726

727
    if (sendto(priv->monitor, &req, sizeof req, 0,
728
               (struct sockaddr *)&addr, sizeof addr) != (sizeof req)) {
729
        virReportSystemError(errno,
730 731
                             _("cannot send command %s"),
                             cmd);
732 733 734 735 736
        return -1;
    }

    do {
        addrlen = sizeof(addr);
737
        if (recvfrom(priv->monitor, &res, sizeof res, 0,
738
                     (struct sockaddr *)&addr, &addrlen) < 0) {
739
            virReportSystemError(errno,
740 741
                                 _("cannot read reply %s"),
                                 cmd);
742 743 744 745
            goto error;
        }

        if (VIR_REALLOC_N(retdata, retlen + res.length) < 0) {
746
            virReportOOMError();
747 748 749 750 751 752 753 754 755 756 757
            goto error;
        }
        memcpy(retdata + retlen, res.data, res.length);
        retlen += res.length - 1;
        retdata[retlen] = '\0';

        if (res.error)
            ret = -1;

    } while (res.extra);

758 759
    VIR_DEBUG("Command reply is '%s'", NULLSTR(retdata));

760 761 762 763 764 765 766 767 768 769
    *reply = retdata;

    return ret;

error:
    VIR_FREE(retdata);
    return -1;
}


770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798
static int umlCleanupTapDevices(virConnectPtr conn ATTRIBUTE_UNUSED,
                                virDomainObjPtr vm) {
    int i;
    int err;
    int ret = 0;
    brControl *brctl = NULL;
    VIR_ERROR0("Cleanup tap");
    if (brInit(&brctl) < 0)
        return -1;

    for (i = 0 ; i < vm->def->nnets ; i++) {
        virDomainNetDefPtr def = vm->def->nets[i];

        if (def->type != VIR_DOMAIN_NET_TYPE_BRIDGE &&
            def->type != VIR_DOMAIN_NET_TYPE_NETWORK)
            continue;

        VIR_ERROR("Cleanup '%s'", def->ifname);
        err = brDeleteTap(brctl, def->ifname);
        if (err) {
            VIR_ERROR("Cleanup failed %d", err);
            ret = -1;
        }
    }
    VIR_ERROR0("Cleanup tap done");
    brShutdown(brctl);
    return ret;
}

799 800 801 802 803
static int umlStartVMDaemon(virConnectPtr conn,
                            struct uml_driver *driver,
                            virDomainObjPtr vm) {
    const char **argv = NULL, **tmp;
    const char **progenv = NULL;
804 805
    int i, ret;
    pid_t pid;
806 807 808 809
    char *logfile;
    int logfd = -1;
    struct stat sb;
    fd_set keepfd;
810
    char ebuf[1024];
811
    umlDomainObjPrivatePtr priv = vm->privateData;
812 813 814

    FD_ZERO(&keepfd);

D
Daniel P. Berrange 已提交
815
    if (virDomainObjIsActive(vm)) {
816 817 818 819 820 821 822 823 824 825 826 827 828 829 830
        umlReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("VM is already active"));
        return -1;
    }

    if (!vm->def->os.kernel) {
        umlReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                       "%s", _("no kernel specified"));
        return -1;
    }
    /* Make sure the binary we are about to try exec'ing exists.
     * Technically we could catch the exec() failure, but that's
     * in a sub-process so its hard to feed back a useful error
     */
    if (stat(vm->def->os.kernel, &sb) < 0) {
831
        virReportSystemError(errno,
832 833
                             _("Cannot find UML kernel %s"),
                             vm->def->os.kernel);
834 835 836
        return -1;
    }

L
Laine Stump 已提交
837
    if (virFileMakePath(driver->logDir) != 0) {
838
        virReportSystemError(errno,
839 840
                             _("cannot create log directory %s"),
                             driver->logDir);
841 842 843
        return -1;
    }

844 845
    if (virAsprintf(&logfile, "%s/%s.log",
                    driver->logDir, vm->def->name) < 0) {
846
        virReportOOMError();
847 848 849 850 851
        return -1;
    }

    if ((logfd = open(logfile, O_CREAT | O_TRUNC | O_WRONLY,
                      S_IRUSR | S_IWUSR)) < 0) {
852
        virReportSystemError(errno,
853 854
                             _("failed to create logfile %s"),
                             logfile);
855 856 857 858 859 860
        VIR_FREE(logfile);
        return -1;
    }
    VIR_FREE(logfile);

    if (umlSetCloseExec(logfd) < 0) {
861
        virReportSystemError(errno,
862
                             "%s", _("Unable to set VM logfile close-on-exec flag"));
863 864 865 866 867
        close(logfd);
        return -1;
    }

    if (umlBuildCommandLine(conn, driver, vm,
868
                            &argv, &progenv) < 0) {
869
        close(logfd);
870
        umlCleanupTapDevices(conn, vm);
871 872 873 874 875 876
        return -1;
    }

    tmp = progenv;
    while (*tmp) {
        if (safewrite(logfd, *tmp, strlen(*tmp)) < 0)
D
Daniel Veillard 已提交
877
            VIR_WARN(_("Unable to write envv to logfile: %s"),
878
                   virStrerror(errno, ebuf, sizeof ebuf));
879
        if (safewrite(logfd, " ", 1) < 0)
D
Daniel Veillard 已提交
880
            VIR_WARN(_("Unable to write envv to logfile: %s"),
881
                   virStrerror(errno, ebuf, sizeof ebuf));
882 883 884 885 886
        tmp++;
    }
    tmp = argv;
    while (*tmp) {
        if (safewrite(logfd, *tmp, strlen(*tmp)) < 0)
D
Daniel Veillard 已提交
887
            VIR_WARN(_("Unable to write argv to logfile: %s"),
888
                   virStrerror(errno, ebuf, sizeof ebuf));
889
        if (safewrite(logfd, " ", 1) < 0)
D
Daniel Veillard 已提交
890
            VIR_WARN(_("Unable to write argv to logfile: %s"),
891
                   virStrerror(errno, ebuf, sizeof ebuf));
892 893 894
        tmp++;
    }
    if (safewrite(logfd, "\n", 1) < 0)
D
Daniel Veillard 已提交
895
        VIR_WARN(_("Unable to write argv to logfile: %s"),
896
                 virStrerror(errno, ebuf, sizeof ebuf));
897

898
    priv->monitor = -1;
899

900
    ret = virExecDaemonize(argv, progenv, &keepfd, &pid,
901
                           -1, &logfd, &logfd,
902 903
                           VIR_EXEC_CLEAR_CAPS,
                           NULL, NULL, NULL);
904 905 906 907 908 909 910 911 912 913
    close(logfd);

    for (i = 0 ; argv[i] ; i++)
        VIR_FREE(argv[i]);
    VIR_FREE(argv);

    for (i = 0 ; progenv[i] ; i++)
        VIR_FREE(progenv[i]);
    VIR_FREE(progenv);

914 915
    if (ret < 0)
        umlCleanupTapDevices(conn, vm);
916 917 918

    /* NB we don't mark it running here - we do that async
       with inotify */
919 920 921
    /* XXX what if someone else tries to start it again
       before we get the inotification ? Sounds like
       trouble.... */
922 923 924 925 926 927 928 929 930

    return ret;
}

static void umlShutdownVMDaemon(virConnectPtr conn ATTRIBUTE_UNUSED,
                                struct uml_driver *driver ATTRIBUTE_UNUSED,
                                virDomainObjPtr vm)
{
    int ret;
931 932
    umlDomainObjPrivatePtr priv = vm->privateData;

D
Daniel P. Berrange 已提交
933
    if (!virDomainObjIsActive(vm))
934 935
        return;

936
    virKillProcess(vm->pid, SIGTERM);
937

938 939 940
    if (priv->monitor != -1)
        close(priv->monitor);
    priv->monitor = -1;
941 942

    if ((ret = waitpid(vm->pid, NULL, 0)) != vm->pid) {
D
Daniel Veillard 已提交
943
        VIR_WARN(_("Got unexpected pid %d != %d"),
944 945 946 947 948 949 950
               ret, vm->pid);
    }

    vm->pid = -1;
    vm->def->id = -1;
    vm->state = VIR_DOMAIN_SHUTOFF;

951 952
    umlCleanupTapDevices(conn, vm);

953 954 955 956 957 958 959 960 961 962 963 964
    if (vm->newDef) {
        virDomainDefFree(vm->def);
        vm->def = vm->newDef;
        vm->def->id = -1;
        vm->newDef = NULL;
    }
}


static virDrvOpenStatus umlOpen(virConnectPtr conn,
                                virConnectAuthPtr auth ATTRIBUTE_UNUSED,
                                int flags ATTRIBUTE_UNUSED) {
965 966 967
    if (conn->uri == NULL) {
        if (uml_driver == NULL)
            return VIR_DRV_OPEN_DECLINED;
968

969
        conn->uri = xmlParseURI(uml_driver->privileged ?
970 971 972
                                "uml:///system" :
                                "uml:///session");
        if (!conn->uri) {
973
            virReportOOMError();
974 975 976 977 978 979 980 981 982 983
            return VIR_DRV_OPEN_ERROR;
        }
    } else {
        if (conn->uri->scheme == NULL ||
            STRNEQ (conn->uri->scheme, "uml"))
            return VIR_DRV_OPEN_DECLINED;

        /* Allow remote driver to deal with URIs with hostname server */
        if (conn->uri->server != NULL)
            return VIR_DRV_OPEN_DECLINED;
984 985


986
        /* Check path and tell them correct path if they made a mistake */
987
        if (uml_driver->privileged) {
988
            if (STRNEQ (conn->uri->path, "/system") &&
989 990 991 992 993 994 995 996 997 998 999 1000 1001
                STRNEQ (conn->uri->path, "/session")) {
                umlReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                               _("unexpected UML URI path '%s', try uml:///system"),
                               conn->uri->path);
                return VIR_DRV_OPEN_ERROR;
            }
        } else {
            if (STRNEQ (conn->uri->path, "/session")) {
                umlReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                               _("unexpected UML URI path '%s', try uml:///session"),
                               conn->uri->path);
                return VIR_DRV_OPEN_ERROR;
            }
1002
        }
1003 1004 1005 1006 1007

        /* URI was good, but driver isn't active */
        if (uml_driver == NULL) {
            umlReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR, "%s",
                           _("uml state driver is not active"));
1008 1009 1010 1011 1012 1013 1014 1015 1016 1017
            return VIR_DRV_OPEN_ERROR;
        }
    }

    conn->privateData = uml_driver;

    return VIR_DRV_OPEN_SUCCESS;
}

static int umlClose(virConnectPtr conn) {
1018
    /*struct uml_driver *driver = conn->privateData;*/
1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029

    conn->privateData = NULL;

    return 0;
}

static const char *umlGetType(virConnectPtr conn ATTRIBUTE_UNUSED) {
    return "UML";
}


1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043
static int umlIsSecure(virConnectPtr conn ATTRIBUTE_UNUSED)
{
    /* Trivially secure, since always inside the daemon */
    return 1;
}


static int umlIsEncrypted(virConnectPtr conn ATTRIBUTE_UNUSED)
{
    /* Not encrypted, but remote driver takes care of that */
    return 0;
}


1044 1045 1046 1047
static char *umlGetCapabilities(virConnectPtr conn) {
    struct uml_driver *driver = (struct uml_driver *)conn->privateData;
    char *xml;

1048
    umlDriverLock(driver);
1049
    if ((xml = virCapabilitiesFormatXML(driver->caps)) == NULL)
1050
        virReportOOMError();
1051
    umlDriverUnlock(driver);
1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075

    return xml;
}



static int umlGetProcessInfo(unsigned long long *cpuTime, int pid) {
    char proc[PATH_MAX];
    FILE *pidinfo;
    unsigned long long usertime, systime;

    if (snprintf(proc, sizeof(proc), "/proc/%d/stat", pid) >= (int)sizeof(proc)) {
        return -1;
    }

    if (!(pidinfo = fopen(proc, "r"))) {
        /*printf("cannot read pid info");*/
        /* VM probably shut down, so fake 0 */
        *cpuTime = 0;
        return 0;
    }

    if (fscanf(pidinfo, "%*d %*s %*c %*d %*d %*d %*d %*d %*u %*u %*u %*u %*u %llu %llu", &usertime, &systime) != 2) {
        umlDebug("not enough arg");
1076
        fclose(pidinfo);
1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097
        return -1;
    }

    /* We got jiffies
     * We want nanoseconds
     * _SC_CLK_TCK is jiffies per second
     * So calulate thus....
     */
    *cpuTime = 1000ull * 1000ull * 1000ull * (usertime + systime) / (unsigned long long)sysconf(_SC_CLK_TCK);

    umlDebug("Got %llu %llu %llu", usertime, systime, *cpuTime);

    fclose(pidinfo);

    return 0;
}


static virDomainPtr umlDomainLookupByID(virConnectPtr conn,
                                          int id) {
    struct uml_driver *driver = (struct uml_driver *)conn->privateData;
1098 1099
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
1100

1101
    umlDriverLock(driver);
1102
    vm = virDomainFindByID(&driver->domains, id);
1103 1104
    umlDriverUnlock(driver);

1105 1106
    if (!vm) {
        umlReportError(conn, NULL, NULL, VIR_ERR_NO_DOMAIN, NULL);
1107
        goto cleanup;
1108 1109 1110 1111
    }

    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
    if (dom) dom->id = vm->def->id;
1112 1113

cleanup:
1114 1115
    if (vm)
        virDomainObjUnlock(vm);
1116 1117
    return dom;
}
1118

1119 1120 1121
static virDomainPtr umlDomainLookupByUUID(virConnectPtr conn,
                                            const unsigned char *uuid) {
    struct uml_driver *driver = (struct uml_driver *)conn->privateData;
1122 1123
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
1124

1125
    umlDriverLock(driver);
1126
    vm = virDomainFindByUUID(&driver->domains, uuid);
1127 1128
    umlDriverUnlock(driver);

1129 1130
    if (!vm) {
        umlReportError(conn, NULL, NULL, VIR_ERR_NO_DOMAIN, NULL);
1131
        goto cleanup;
1132 1133 1134 1135
    }

    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
    if (dom) dom->id = vm->def->id;
1136 1137

cleanup:
1138 1139
    if (vm)
        virDomainObjUnlock(vm);
1140 1141
    return dom;
}
1142

1143 1144 1145
static virDomainPtr umlDomainLookupByName(virConnectPtr conn,
                                            const char *name) {
    struct uml_driver *driver = (struct uml_driver *)conn->privateData;
1146 1147
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
1148

1149
    umlDriverLock(driver);
1150
    vm = virDomainFindByName(&driver->domains, name);
1151 1152
    umlDriverUnlock(driver);

1153 1154
    if (!vm) {
        umlReportError(conn, NULL, NULL, VIR_ERR_NO_DOMAIN, NULL);
1155
        goto cleanup;
1156 1157 1158 1159
    }

    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
    if (dom) dom->id = vm->def->id;
1160 1161

cleanup:
1162 1163
    if (vm)
        virDomainObjUnlock(vm);
1164 1165 1166
    return dom;
}

1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211

static int umlDomainIsActive(virDomainPtr dom)
{
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr obj;
    int ret = -1;

    umlDriverLock(driver);
    obj = virDomainFindByUUID(&driver->domains, dom->uuid);
    umlDriverUnlock(driver);
    if (!obj) {
        umlReportError(dom->conn, NULL, NULL, VIR_ERR_NO_DOMAIN, NULL);
        goto cleanup;
    }
    ret = virDomainObjIsActive(obj);

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


static int umlDomainIsPersistent(virDomainPtr dom)
{
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr obj;
    int ret = -1;

    umlDriverLock(driver);
    obj = virDomainFindByUUID(&driver->domains, dom->uuid);
    umlDriverUnlock(driver);
    if (!obj) {
        umlReportError(dom->conn, NULL, NULL, VIR_ERR_NO_DOMAIN, NULL);
        goto cleanup;
    }
    ret = obj->persistent;

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


1212
static int umlGetVersion(virConnectPtr conn, unsigned long *version) {
1213
    struct uml_driver *driver = conn->privateData;
1214 1215
    struct utsname ut;
    int major, minor, micro;
1216
    int ret = -1;
1217 1218 1219

    uname(&ut);

1220
    umlDriverLock(driver);
1221 1222 1223 1224
    if (sscanf(ut.release, "%u.%u.%u",
               &major, &minor, &micro) != 3) {
        umlReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                       _("cannot parse version %s"), ut.release);
1225
        goto cleanup;
1226 1227
    }

1228 1229 1230 1231 1232 1233
    *version = driver->umlVersion;
    ret = 0;

cleanup:
    umlDriverUnlock(driver);
    return ret;
1234 1235 1236
}

static int umlListDomains(virConnectPtr conn, int *ids, int nids) {
1237
    struct uml_driver *driver = conn->privateData;
1238
    int n;
1239

1240
    umlDriverLock(driver);
1241
    n = virDomainObjListGetActiveIDs(&driver->domains, ids, nids);
1242
    umlDriverUnlock(driver);
1243

1244
    return n;
1245 1246
}
static int umlNumDomains(virConnectPtr conn) {
1247
    struct uml_driver *driver = conn->privateData;
1248
    int n;
1249

1250
    umlDriverLock(driver);
1251
    n = virDomainObjListNumOfDomains(&driver->domains, 1);
1252
    umlDriverUnlock(driver);
1253 1254 1255 1256 1257

    return n;
}
static virDomainPtr umlDomainCreate(virConnectPtr conn, const char *xml,
                                      unsigned int flags ATTRIBUTE_UNUSED) {
1258
    struct uml_driver *driver = conn->privateData;
1259
    virDomainDefPtr def;
1260
    virDomainObjPtr vm = NULL;
1261
    virDomainPtr dom = NULL;
1262

1263
    umlDriverLock(driver);
1264
    if (!(def = virDomainDefParseString(driver->caps, xml,
1265
                                        VIR_DOMAIN_XML_INACTIVE)))
1266
        goto cleanup;
1267

1268
    if (virDomainObjIsDuplicate(&driver->domains, def, 1) < 0)
1269
        goto cleanup;
1270

1271
    if (!(vm = virDomainAssignDef(driver->caps,
1272
                                  &driver->domains,
1273 1274 1275
                                  def)))
        goto cleanup;
    def = NULL;
1276 1277 1278 1279

    if (umlStartVMDaemon(conn, driver, vm) < 0) {
        virDomainRemoveInactive(&driver->domains,
                                vm);
1280 1281
        vm = NULL;
        goto cleanup;
1282 1283 1284 1285
    }

    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
    if (dom) dom->id = vm->def->id;
1286 1287 1288

cleanup:
    virDomainDefFree(def);
1289 1290 1291
    if (vm)
        virDomainObjUnlock(vm);
    umlDriverUnlock(driver);
1292 1293 1294 1295 1296
    return dom;
}


static int umlDomainShutdown(virDomainPtr dom) {
1297 1298
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
1299
    char *info = NULL;
1300
    int ret = -1;
1301

1302
    umlDriverLock(driver);
1303
    vm = virDomainFindByID(&driver->domains, dom->id);
1304
    umlDriverUnlock(driver);
1305 1306 1307
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         _("no domain with matching id %d"), dom->id);
1308
        goto cleanup;
1309 1310 1311 1312 1313 1314
    }

#if 0
    if (umlMonitorCommand(driver, vm, "system_powerdown", &info) < 0) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("shutdown operation failed"));
1315
        goto cleanup;
1316
    }
1317
    ret = 0;
1318 1319
#endif

1320 1321
cleanup:
    VIR_FREE(info);
1322 1323
    if (vm)
        virDomainObjUnlock(vm);
1324
    return ret;
1325 1326 1327 1328
}


static int umlDomainDestroy(virDomainPtr dom) {
1329 1330 1331
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1332

1333
    umlDriverLock(driver);
1334
    vm = virDomainFindByID(&driver->domains, dom->id);
1335 1336 1337
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         _("no domain with matching id %d"), dom->id);
1338
        goto cleanup;
1339 1340 1341
    }

    umlShutdownVMDaemon(dom->conn, driver, vm);
1342
    if (!vm->persistent) {
1343 1344
        virDomainRemoveInactive(&driver->domains,
                                vm);
1345 1346 1347
        vm = NULL;
    }
    ret = 0;
1348

1349
cleanup:
1350 1351 1352
    if (vm)
        virDomainObjUnlock(vm);
    umlDriverUnlock(driver);
1353
    return ret;
1354 1355 1356 1357
}


static char *umlDomainGetOSType(virDomainPtr dom) {
1358 1359 1360
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *type = NULL;
1361

1362
    umlDriverLock(driver);
1363
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1364
    umlDriverUnlock(driver);
1365 1366 1367
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
1368
        goto cleanup;
1369 1370
    }

1371
    if (!(type = strdup(vm->def->os.type)))
1372
        virReportOOMError();
1373 1374

cleanup:
1375 1376
    if (vm)
        virDomainObjUnlock(vm);
1377 1378 1379 1380 1381
    return type;
}

/* Returns max memory in kb, 0 if error */
static unsigned long umlDomainGetMaxMemory(virDomainPtr dom) {
1382 1383 1384
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    unsigned long ret = 0;
1385

1386
    umlDriverLock(driver);
1387
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1388 1389
    umlDriverUnlock(driver);

1390 1391 1392 1393 1394
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
1395 1396
                       _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
1397
    }
1398
    ret = vm->def->maxmem;
1399

1400
cleanup:
1401 1402
    if (vm)
        virDomainObjUnlock(vm);
1403
    return ret;
1404 1405 1406
}

static int umlDomainSetMaxMemory(virDomainPtr dom, unsigned long newmax) {
1407 1408 1409
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1410

1411
    umlDriverLock(driver);
1412
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1413 1414
    umlDriverUnlock(driver);

1415 1416 1417 1418 1419 1420
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
1421
        goto cleanup;
1422 1423 1424 1425 1426
    }

    if (newmax < vm->def->memory) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                         "%s", _("cannot set max memory lower than current memory"));
1427
        goto cleanup;
1428 1429 1430
    }

    vm->def->maxmem = newmax;
1431 1432 1433
    ret = 0;

cleanup:
1434 1435
    if (vm)
        virDomainObjUnlock(vm);
1436
    return ret;
1437 1438 1439
}

static int umlDomainSetMemory(virDomainPtr dom, unsigned long newmem) {
1440 1441 1442
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1443

1444
    umlDriverLock(driver);
1445
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1446 1447
    umlDriverUnlock(driver);

1448 1449 1450 1451 1452 1453
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
1454
        goto cleanup;
1455 1456
    }

D
Daniel P. Berrange 已提交
1457
    if (virDomainObjIsActive(vm)) {
1458 1459
        umlReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("cannot set memory of an active domain"));
1460
        goto cleanup;
1461 1462 1463 1464 1465
    }

    if (newmem > vm->def->maxmem) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                         "%s", _("cannot set memory higher than max memory"));
1466
        goto cleanup;
1467 1468 1469
    }

    vm->def->memory = newmem;
1470 1471 1472
    ret = 0;

cleanup:
1473 1474
    if (vm)
        virDomainObjUnlock(vm);
1475
    return ret;
1476 1477 1478 1479
}

static int umlDomainGetInfo(virDomainPtr dom,
                              virDomainInfoPtr info) {
1480 1481 1482 1483
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;

1484
    umlDriverLock(driver);
1485
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1486 1487
    umlDriverUnlock(driver);

1488 1489
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
1490 1491
                       "%s", _("no domain with matching uuid"));
        goto cleanup;
1492 1493 1494 1495
    }

    info->state = vm->state;

D
Daniel P. Berrange 已提交
1496
    if (!virDomainObjIsActive(vm)) {
1497 1498 1499
        info->cpuTime = 0;
    } else {
        if (umlGetProcessInfo(&(info->cpuTime), vm->pid) < 0) {
1500 1501 1502
            umlReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                           "%s", _("cannot read cputime for domain"));
            goto cleanup;
1503 1504 1505 1506 1507 1508
        }
    }

    info->maxMem = vm->def->maxmem;
    info->memory = vm->def->memory;
    info->nrVirtCpu = vm->def->vcpus;
1509 1510 1511
    ret = 0;

cleanup:
1512 1513
    if (vm)
        virDomainObjUnlock(vm);
1514
    return ret;
1515 1516 1517 1518 1519
}


static char *umlDomainDumpXML(virDomainPtr dom,
                                int flags ATTRIBUTE_UNUSED) {
1520 1521 1522 1523
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *ret = NULL;

1524
    umlDriverLock(driver);
1525
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1526 1527
    umlDriverUnlock(driver);

1528 1529 1530
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
1531
        goto cleanup;
1532 1533
    }

1534
    ret = virDomainDefFormat((flags & VIR_DOMAIN_XML_INACTIVE) && vm->newDef ?
1535 1536 1537 1538
                             vm->newDef : vm->def,
                             flags);

cleanup:
1539 1540
    if (vm)
        virDomainObjUnlock(vm);
1541
    return ret;
1542 1543 1544 1545 1546
}


static int umlListDefinedDomains(virConnectPtr conn,
                            char **const names, int nnames) {
1547
    struct uml_driver *driver = conn->privateData;
1548
    int n;
1549

1550
    umlDriverLock(driver);
1551
    n = virDomainObjListGetInactiveNames(&driver->domains, names, nnames);
1552
    umlDriverUnlock(driver);
1553

1554
    return n;
1555 1556 1557
}

static int umlNumDefinedDomains(virConnectPtr conn) {
1558
    struct uml_driver *driver = conn->privateData;
1559
    int n;
1560

1561
    umlDriverLock(driver);
1562
    n = virDomainObjListNumOfDomains(&driver->domains, 0);
1563
    umlDriverUnlock(driver);
1564 1565 1566 1567 1568 1569

    return n;
}


static int umlDomainStart(virDomainPtr dom) {
1570 1571 1572
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1573

1574
    umlDriverLock(driver);
1575
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1576

1577 1578 1579
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
1580
        goto cleanup;
1581 1582
    }

1583 1584 1585
    ret = umlStartVMDaemon(dom->conn, driver, vm);

cleanup:
1586 1587 1588
    if (vm)
        virDomainObjUnlock(vm);
    umlDriverUnlock(driver);
1589
    return ret;
1590 1591 1592 1593
}


static virDomainPtr umlDomainDefine(virConnectPtr conn, const char *xml) {
1594
    struct uml_driver *driver = conn->privateData;
1595
    virDomainDefPtr def;
1596
    virDomainObjPtr vm = NULL;
1597
    virDomainPtr dom = NULL;
1598

1599
    umlDriverLock(driver);
1600
    if (!(def = virDomainDefParseString(driver->caps, xml,
1601
                                        VIR_DOMAIN_XML_INACTIVE)))
1602
        goto cleanup;
1603

1604 1605 1606
    if (virDomainObjIsDuplicate(&driver->domains, def, 0) < 0)
        goto cleanup;

1607
    if (!(vm = virDomainAssignDef(driver->caps,
1608
                                  &driver->domains,
1609 1610 1611
                                  def)))
        goto cleanup;
    def = NULL;
1612 1613
    vm->persistent = 1;

1614
    if (virDomainSaveConfig(driver->configDir,
1615 1616 1617
                            vm->newDef ? vm->newDef : vm->def) < 0) {
        virDomainRemoveInactive(&driver->domains,
                                vm);
1618
        vm = NULL;
1619
        goto cleanup;
1620 1621 1622 1623
    }

    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
    if (dom) dom->id = vm->def->id;
1624 1625 1626

cleanup:
    virDomainDefFree(def);
1627 1628 1629
    if (vm)
        virDomainObjUnlock(vm);
    umlDriverUnlock(driver);
1630 1631 1632 1633
    return dom;
}

static int umlDomainUndefine(virDomainPtr dom) {
1634 1635 1636
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1637

1638
    umlDriverLock(driver);
1639
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1640 1641 1642
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
1643
        goto cleanup;
1644 1645
    }

D
Daniel P. Berrange 已提交
1646
    if (virDomainObjIsActive(vm)) {
1647 1648
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot delete active domain"));
1649
        goto cleanup;
1650 1651 1652 1653 1654
    }

    if (!vm->persistent) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot undefine transient domain"));
1655
        goto cleanup;
1656 1657
    }

1658
    if (virDomainDeleteConfig(driver->configDir, driver->autostartDir, vm) < 0)
1659
        goto cleanup;
1660 1661 1662

    virDomainRemoveInactive(&driver->domains,
                            vm);
1663
    vm = NULL;
1664
    ret = 0;
1665

1666
cleanup:
1667 1668 1669
    if (vm)
        virDomainObjUnlock(vm);
    umlDriverUnlock(driver);
1670
    return ret;
1671 1672 1673 1674 1675 1676
}



static int umlDomainGetAutostart(virDomainPtr dom,
                            int *autostart) {
1677 1678 1679
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1680

1681
    umlDriverLock(driver);
1682
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1683

1684 1685 1686
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
1687
        goto cleanup;
1688 1689 1690
    }

    *autostart = vm->autostart;
1691
    ret = 0;
1692

1693
cleanup:
1694 1695
    if (vm)
        virDomainObjUnlock(vm);
1696
    umlDriverUnlock(driver);
1697
    return ret;
1698 1699 1700 1701
}

static int umlDomainSetAutostart(virDomainPtr dom,
                                   int autostart) {
1702
    struct uml_driver *driver = dom->conn->privateData;
1703
    virDomainObjPtr vm;
1704 1705 1706
    char *configFile = NULL, *autostartLink = NULL;
    int ret = -1;

1707 1708 1709
    umlDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);

1710 1711 1712
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
1713
        goto cleanup;
1714 1715 1716 1717 1718
    }

    if (!vm->persistent) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot set autostart for transient domain"));
1719
        goto cleanup;
1720 1721 1722 1723
    }

    autostart = (autostart != 0);

1724
    if (vm->autostart != autostart) {
1725
        if ((configFile = virDomainConfigFile(driver->configDir, vm->def->name)) == NULL)
1726
            goto cleanup;
1727
        if ((autostartLink = virDomainConfigFile(driver->autostartDir, vm->def->name)) == NULL)
1728
            goto cleanup;
1729

1730 1731
        if (autostart) {
            int err;
1732

1733
            if ((err = virFileMakePath(driver->autostartDir))) {
1734
                virReportSystemError(err,
1735 1736
                                     _("cannot create autostart directory %s"),
                                     driver->autostartDir);
1737 1738
                goto cleanup;
            }
1739

1740
            if (symlink(configFile, autostartLink) < 0) {
1741
                virReportSystemError(errno,
1742 1743
                                     _("Failed to create symlink '%s to '%s'"),
                                     autostartLink, configFile);
1744 1745 1746 1747
                goto cleanup;
            }
        } else {
            if (unlink(autostartLink) < 0 && errno != ENOENT && errno != ENOTDIR) {
1748
                virReportSystemError(errno,
1749 1750
                                     _("Failed to delete symlink '%s'"),
                                     autostartLink);
1751 1752
                goto cleanup;
            }
1753 1754
        }

1755
        vm->autostart = autostart;
1756 1757 1758 1759 1760 1761
    }
    ret = 0;

cleanup:
    VIR_FREE(configFile);
    VIR_FREE(autostartLink);
1762 1763
    if (vm)
        virDomainObjUnlock(vm);
1764
    umlDriverUnlock(driver);
1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775
    return ret;
}


static int
umlDomainBlockPeek (virDomainPtr dom,
                      const char *path,
                      unsigned long long offset, size_t size,
                      void *buffer,
                      unsigned int flags ATTRIBUTE_UNUSED)
{
1776 1777 1778
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int fd = -1, ret = -1, i;
1779

1780
    umlDriverLock(driver);
1781
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1782 1783
    umlDriverUnlock(driver);

1784 1785
    if (!vm) {
        umlReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
J
Jim Meyering 已提交
1786
                        "%s", _("no domain with matching uuid"));
1787
        goto cleanup;
1788 1789 1790 1791
    }

    if (!path || path[0] == '\0') {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
J
Jim Meyering 已提交
1792
                       "%s", _("NULL or empty path"));
1793
        goto cleanup;
1794 1795 1796 1797 1798
    }

    /* Check the path belongs to this domain. */
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (vm->def->disks[i]->src != NULL &&
1799 1800 1801 1802
            STREQ (vm->def->disks[i]->src, path)) {
            ret = 0;
            break;
        }
1803 1804
    }

1805 1806 1807 1808 1809
    if (ret == 0) {
        ret = -1;
        /* The path is correct, now try to open it and get its size. */
        fd = open (path, O_RDONLY);
        if (fd == -1) {
1810
            virReportSystemError(errno,
1811
                                 _("cannot open %s"), path);
1812 1813
            goto cleanup;
        }
1814

1815 1816 1817 1818 1819 1820
        /* Seek and read. */
        /* NB. Because we configure with AC_SYS_LARGEFILE, off_t should
         * be 64 bits on all platforms.
         */
        if (lseek (fd, offset, SEEK_SET) == (off_t) -1 ||
            saferead (fd, buffer, size) == (ssize_t) -1) {
1821
            virReportSystemError(errno,
1822
                                 _("cannot read %s"), path);
1823 1824 1825 1826 1827 1828
            goto cleanup;
        }

        ret = 0;
    } else {
        umlReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
J
Jim Meyering 已提交
1829
                        "%s", _("invalid path"));
1830 1831
    }

1832
cleanup:
1833
    if (fd >= 0) close (fd);
1834 1835
    if (vm)
        virDomainObjUnlock(vm);
1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848
    return ret;
}



static virDriver umlDriver = {
    VIR_DRV_UML,
    "UML",
    umlOpen, /* open */
    umlClose, /* close */
    NULL, /* supports_feature */
    umlGetType, /* type */
    umlGetVersion, /* version */
1849
    NULL, /* libvirtVersion (impl. in libvirt.c) */
1850
    virGetHostname, /* getHostname */
1851
    NULL, /* getMaxVcpus */
1852
    nodeGetInfo, /* nodeGetInfo */
1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876
    umlGetCapabilities, /* getCapabilities */
    umlListDomains, /* listDomains */
    umlNumDomains, /* numOfDomains */
    umlDomainCreate, /* domainCreateXML */
    umlDomainLookupByID, /* domainLookupByID */
    umlDomainLookupByUUID, /* domainLookupByUUID */
    umlDomainLookupByName, /* domainLookupByName */
    NULL, /* domainSuspend */
    NULL, /* domainResume */
    umlDomainShutdown, /* domainShutdown */
    NULL, /* domainReboot */
    umlDomainDestroy, /* domainDestroy */
    umlDomainGetOSType, /* domainGetOSType */
    umlDomainGetMaxMemory, /* domainGetMaxMemory */
    umlDomainSetMaxMemory, /* domainSetMaxMemory */
    umlDomainSetMemory, /* domainSetMemory */
    umlDomainGetInfo, /* domainGetInfo */
    NULL, /* domainSave */
    NULL, /* domainRestore */
    NULL, /* domainCoreDump */
    NULL, /* domainSetVcpus */
    NULL, /* domainPinVcpu */
    NULL, /* domainGetVcpus */
    NULL, /* domainGetMaxVcpus */
1877 1878
    NULL, /* domainGetSecurityLabel */
    NULL, /* nodeGetSecurityModel */
1879
    umlDomainDumpXML, /* domainDumpXML */
1880 1881
    NULL, /* domainXMLFromNative */
    NULL, /* domainXMLToNative */
1882 1883
    umlListDefinedDomains, /* listDefinedDomains */
    umlNumDefinedDomains, /* numOfDefinedDomains */
1884 1885 1886 1887
    umlDomainStart, /* domainCreate */
    umlDomainDefine, /* domainDefineXML */
    umlDomainUndefine, /* domainUndefine */
    NULL, /* domainAttachDevice */
1888
    NULL, /* domainAttachDeviceFlags */
1889
    NULL, /* domainDetachDevice */
1890
    NULL, /* domainDetachDeviceFlags */
1891 1892 1893 1894 1895 1896 1897 1898 1899 1900
    umlDomainGetAutostart, /* domainGetAutostart */
    umlDomainSetAutostart, /* domainSetAutostart */
    NULL, /* domainGetSchedulerType */
    NULL, /* domainGetSchedulerParameters */
    NULL, /* domainSetSchedulerParameters */
    NULL, /* domainMigratePrepare */
    NULL, /* domainMigratePerform */
    NULL, /* domainMigrateFinish */
    NULL, /* domainBlockStats */
    NULL, /* domainInterfaceStats */
1901
    NULL, /* domainMemoryStats */
1902 1903
    umlDomainBlockPeek, /* domainBlockPeek */
    NULL, /* domainMemoryPeek */
1904 1905
    nodeGetCellsFreeMemory, /* nodeGetCellsFreeMemory */
    nodeGetFreeMemory,  /* getFreeMemory */
1906
    NULL, /* domainEventRegister */
1907
    NULL, /* domainEventDeregister */
1908 1909
    NULL, /* domainMigratePrepare2 */
    NULL, /* domainMigrateFinish2 */
1910
    NULL, /* nodeDeviceDettach */
1911 1912
    NULL, /* nodeDeviceReAttach */
    NULL, /* nodeDeviceReset */
C
Chris Lalancette 已提交
1913
    NULL, /* domainMigratePrepareTunnel */
1914 1915 1916 1917
    umlIsEncrypted,
    umlIsSecure,
    umlDomainIsActive,
    umlDomainIsPersistent,
J
Jiri Denemark 已提交
1918
    NULL, /* cpuCompare */
1919
    NULL, /* cpuBaseline */
1920 1921 1922 1923
};


static virStateDriver umlStateDriver = {
1924
    .name = "UML",
1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935
    .initialize = umlStartup,
    .cleanup = umlShutdown,
    .reload = umlReload,
    .active = umlActive,
};

int umlRegister(void) {
    virRegisterDriver(&umlDriver);
    virRegisterStateDriver(&umlStateDriver);
    return 0;
}