uml_driver.c 52.1 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 439 440 441 442

    if (virDomainLoadAllConfigs(NULL,
                                uml_driver->caps,
                                &uml_driver->domains,
                                uml_driver->configDir,
                                uml_driver->autostartDir,
443
                                0, NULL, NULL) < 0)
444 445
        goto error;

446 447
    umlAutostartConfigs(uml_driver);

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

451 452
    return 0;

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

error:
457
    VIR_FREE(userdir);
458
    VIR_FREE(base);
459 460
    umlDriverUnlock(uml_driver);
    umlShutdown();
461 462 463 464 465 466 467 468 469 470 471 472 473 474
    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;

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

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

    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) {
499
    int active = 0;
500 501 502 503

    if (!uml_driver)
        return 0;

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

508
    return active;
509 510
}

511 512 513 514 515 516 517
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 已提交
518
    if (virDomainObjIsActive(dom))
519 520 521 522
        umlShutdownVMDaemon(NULL, driver, dom);
    virDomainObjUnlock(dom);
}

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

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

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

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

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

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

    return 0;
}


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

    vm->pid = -1;
570 571
    if (virAsprintf(&pidfile, "%s/%s/pid",
                    driver->monitorDir, vm->def->name) < 0) {
572
        virReportOOMError();
573 574 575 576 577 578 579 580 581 582 583 584 585 586 587
        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;
588
        fclose(file);
589 590 591 592 593 594 595 596 597 598
        goto cleanup;
    }

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

    rc = 0;

 cleanup:
    if (rc != 0)
599
        virReportSystemError(errno,
600 601
                             _("failed to read pid: %s"),
                             pidfile);
602 603 604 605 606 607 608 609 610
    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 已提交
611
    int retval = 0;
612

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

    memset(addr, 0, sizeof *addr);
    addr->sun_family = AF_UNIX;
C
Chris Lalancette 已提交
621 622 623 624 625
    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;
    }
626
    VIR_FREE(sockname);
C
Chris Lalancette 已提交
627
    return retval;
628 629 630 631 632 633 634 635
}

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

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

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

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

    memset(addr.sun_path, 0, sizeof addr.sun_path);
659 660
    sprintf(addr.sun_path + 1, "libvirt-uml-%u", vm->pid);
    VIR_DEBUG("Reply address for monitor is '%s'", addr.sun_path+1);
661
    if (bind(priv->monitor, (struct sockaddr *)&addr, sizeof addr) < 0) {
662
        virReportSystemError(errno,
663
                             "%s", _("cannot bind socket"));
664 665
        close(priv->monitor);
        priv->monitor = -1;
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 702 703
        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;
704
    umlDomainObjPrivatePtr priv = vm->privateData;
705

706 707
    VIR_DEBUG("Run command '%s'", cmd);

708 709 710 711 712 713 714 715 716 717
    *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)) {
718
        virReportSystemError(EINVAL,
719 720
                             _("cannot send too long command %s (%d bytes)"),
                             cmd, req.length);
721 722
        return -1;
    }
C
Chris Lalancette 已提交
723 724 725 726 727
    if (virStrcpyStatic(req.data, cmd) == NULL) {
        umlReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                       _("Command %s too long for destination"), cmd);
        return -1;
    }
728

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

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

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

        if (res.error)
            ret = -1;

    } while (res.extra);

760 761
    VIR_DEBUG("Command reply is '%s'", NULLSTR(retdata));

762 763 764 765 766 767 768 769 770 771
    *reply = retdata;

    return ret;

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


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

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

    FD_ZERO(&keepfd);

D
Daniel P. Berrange 已提交
817
    if (virDomainObjIsActive(vm)) {
818 819 820 821 822 823 824 825 826 827 828 829 830 831 832
        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) {
833
        virReportSystemError(errno,
834 835
                             _("Cannot find UML kernel %s"),
                             vm->def->os.kernel);
836 837 838
        return -1;
    }

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

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

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

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

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

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

900
    priv->monitor = -1;
901

902
    ret = virExecDaemonize(argv, progenv, &keepfd, &pid,
903
                           -1, &logfd, &logfd,
904 905
                           VIR_EXEC_CLEAR_CAPS,
                           NULL, NULL, NULL);
906 907 908 909 910 911 912 913 914 915
    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);

916 917
    if (ret < 0)
        umlCleanupTapDevices(conn, vm);
918 919 920

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

    return ret;
}

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

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

938
    virKillProcess(vm->pid, SIGTERM);
939

940 941 942
    if (priv->monitor != -1)
        close(priv->monitor);
    priv->monitor = -1;
943 944

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

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

953 954
    umlCleanupTapDevices(conn, vm);

955 956 957 958 959 960 961 962 963 964 965 966
    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) {
967 968 969
    if (conn->uri == NULL) {
        if (uml_driver == NULL)
            return VIR_DRV_OPEN_DECLINED;
970

971
        conn->uri = xmlParseURI(uml_driver->privileged ?
972 973 974
                                "uml:///system" :
                                "uml:///session");
        if (!conn->uri) {
975
            virReportOOMError();
976 977 978 979 980 981 982 983 984 985
            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;
986 987


988
        /* Check path and tell them correct path if they made a mistake */
989
        if (uml_driver->privileged) {
990
            if (STRNEQ (conn->uri->path, "/system") &&
991 992 993 994 995 996 997 998 999 1000 1001 1002 1003
                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;
            }
1004
        }
1005 1006 1007 1008 1009

        /* 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"));
1010 1011 1012 1013 1014 1015 1016 1017 1018 1019
            return VIR_DRV_OPEN_ERROR;
        }
    }

    conn->privateData = uml_driver;

    return VIR_DRV_OPEN_SUCCESS;
}

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

    conn->privateData = NULL;

    return 0;
}

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


1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045
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;
}


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

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

    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");
1078
        fclose(pidinfo);
1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099
        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;
1100 1101
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
1102

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

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

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

cleanup:
1116 1117
    if (vm)
        virDomainObjUnlock(vm);
1118 1119
    return dom;
}
1120

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

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

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

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

cleanup:
1140 1141
    if (vm)
        virDomainObjUnlock(vm);
1142 1143
    return dom;
}
1144

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

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

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

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

cleanup:
1164 1165
    if (vm)
        virDomainObjUnlock(vm);
1166 1167 1168
    return dom;
}

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 1212 1213

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


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

    uname(&ut);

1222
    umlDriverLock(driver);
1223 1224 1225 1226
    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);
1227
        goto cleanup;
1228 1229
    }

1230 1231 1232 1233 1234 1235
    *version = driver->umlVersion;
    ret = 0;

cleanup:
    umlDriverUnlock(driver);
    return ret;
1236 1237 1238
}

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

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

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

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

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

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

1270
    if (virDomainObjIsDuplicate(&driver->domains, def, 1) < 0)
1271
        goto cleanup;
1272 1273

    if (!(vm = virDomainAssignDef(conn,
1274
                                  driver->caps,
1275
                                  &driver->domains,
1276 1277 1278
                                  def)))
        goto cleanup;
    def = NULL;
1279 1280 1281 1282

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

    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
    if (dom) dom->id = vm->def->id;
1289 1290 1291

cleanup:
    virDomainDefFree(def);
1292 1293 1294
    if (vm)
        virDomainObjUnlock(vm);
    umlDriverUnlock(driver);
1295 1296 1297 1298 1299
    return dom;
}


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

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

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

1323 1324
cleanup:
    VIR_FREE(info);
1325 1326
    if (vm)
        virDomainObjUnlock(vm);
1327
    return ret;
1328 1329 1330 1331
}


static int umlDomainDestroy(virDomainPtr dom) {
1332 1333 1334
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1335

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

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

1352
cleanup:
1353 1354 1355
    if (vm)
        virDomainObjUnlock(vm);
    umlDriverUnlock(driver);
1356
    return ret;
1357 1358 1359 1360
}


static char *umlDomainGetOSType(virDomainPtr dom) {
1361 1362 1363
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *type = NULL;
1364

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

1374
    if (!(type = strdup(vm->def->os.type)))
1375
        virReportOOMError();
1376 1377

cleanup:
1378 1379
    if (vm)
        virDomainObjUnlock(vm);
1380 1381 1382 1383 1384
    return type;
}

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

1389
    umlDriverLock(driver);
1390
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1391 1392
    umlDriverUnlock(driver);

1393 1394 1395 1396 1397
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];

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

1403
cleanup:
1404 1405
    if (vm)
        virDomainObjUnlock(vm);
1406
    return ret;
1407 1408 1409
}

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

1414
    umlDriverLock(driver);
1415
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1416 1417
    umlDriverUnlock(driver);

1418 1419 1420 1421 1422 1423
    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);
1424
        goto cleanup;
1425 1426 1427 1428 1429
    }

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

    vm->def->maxmem = newmax;
1434 1435 1436
    ret = 0;

cleanup:
1437 1438
    if (vm)
        virDomainObjUnlock(vm);
1439
    return ret;
1440 1441 1442
}

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

1447
    umlDriverLock(driver);
1448
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1449 1450
    umlDriverUnlock(driver);

1451 1452 1453 1454 1455 1456
    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);
1457
        goto cleanup;
1458 1459
    }

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

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

    vm->def->memory = newmem;
1473 1474 1475
    ret = 0;

cleanup:
1476 1477
    if (vm)
        virDomainObjUnlock(vm);
1478
    return ret;
1479 1480 1481 1482
}

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

1487
    umlDriverLock(driver);
1488
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1489 1490
    umlDriverUnlock(driver);

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

    info->state = vm->state;

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

    info->maxMem = vm->def->maxmem;
    info->memory = vm->def->memory;
    info->nrVirtCpu = vm->def->vcpus;
1512 1513 1514
    ret = 0;

cleanup:
1515 1516
    if (vm)
        virDomainObjUnlock(vm);
1517
    return ret;
1518 1519 1520 1521 1522
}


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

1527
    umlDriverLock(driver);
1528
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1529 1530
    umlDriverUnlock(driver);

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

1537 1538 1539 1540 1541 1542
    ret = virDomainDefFormat(dom->conn,
                             (flags & VIR_DOMAIN_XML_INACTIVE) && vm->newDef ?
                             vm->newDef : vm->def,
                             flags);

cleanup:
1543 1544
    if (vm)
        virDomainObjUnlock(vm);
1545
    return ret;
1546 1547 1548 1549 1550
}


static int umlListDefinedDomains(virConnectPtr conn,
                            char **const names, int nnames) {
1551
    struct uml_driver *driver = conn->privateData;
1552
    int n;
1553

1554
    umlDriverLock(driver);
1555
    n = virDomainObjListGetInactiveNames(&driver->domains, names, nnames);
1556
    umlDriverUnlock(driver);
1557

1558
    return n;
1559 1560 1561
}

static int umlNumDefinedDomains(virConnectPtr conn) {
1562
    struct uml_driver *driver = conn->privateData;
1563
    int n;
1564

1565
    umlDriverLock(driver);
1566
    n = virDomainObjListNumOfDomains(&driver->domains, 0);
1567
    umlDriverUnlock(driver);
1568 1569 1570 1571 1572 1573

    return n;
}


static int umlDomainStart(virDomainPtr dom) {
1574 1575 1576
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1577

1578
    umlDriverLock(driver);
1579
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1580

1581 1582 1583
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
1584
        goto cleanup;
1585 1586
    }

1587 1588 1589
    ret = umlStartVMDaemon(dom->conn, driver, vm);

cleanup:
1590 1591 1592
    if (vm)
        virDomainObjUnlock(vm);
    umlDriverUnlock(driver);
1593
    return ret;
1594 1595 1596 1597
}


static virDomainPtr umlDomainDefine(virConnectPtr conn, const char *xml) {
1598
    struct uml_driver *driver = conn->privateData;
1599
    virDomainDefPtr def;
1600
    virDomainObjPtr vm = NULL;
1601
    virDomainPtr dom = NULL;
1602

1603
    umlDriverLock(driver);
1604 1605
    if (!(def = virDomainDefParseString(conn, driver->caps, xml,
                                        VIR_DOMAIN_XML_INACTIVE)))
1606
        goto cleanup;
1607

1608 1609 1610
    if (virDomainObjIsDuplicate(&driver->domains, def, 0) < 0)
        goto cleanup;

1611
    if (!(vm = virDomainAssignDef(conn,
1612
                                  driver->caps,
1613
                                  &driver->domains,
1614 1615 1616
                                  def)))
        goto cleanup;
    def = NULL;
1617 1618 1619 1620 1621 1622 1623
    vm->persistent = 1;

    if (virDomainSaveConfig(conn,
                            driver->configDir,
                            vm->newDef ? vm->newDef : vm->def) < 0) {
        virDomainRemoveInactive(&driver->domains,
                                vm);
1624
        vm = NULL;
1625
        goto cleanup;
1626 1627 1628 1629
    }

    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
    if (dom) dom->id = vm->def->id;
1630 1631 1632

cleanup:
    virDomainDefFree(def);
1633 1634 1635
    if (vm)
        virDomainObjUnlock(vm);
    umlDriverUnlock(driver);
1636 1637 1638 1639
    return dom;
}

static int umlDomainUndefine(virDomainPtr dom) {
1640 1641 1642
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1643

1644
    umlDriverLock(driver);
1645
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1646 1647 1648
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
1649
        goto cleanup;
1650 1651
    }

D
Daniel P. Berrange 已提交
1652
    if (virDomainObjIsActive(vm)) {
1653 1654
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot delete active domain"));
1655
        goto cleanup;
1656 1657 1658 1659 1660
    }

    if (!vm->persistent) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot undefine transient domain"));
1661
        goto cleanup;
1662 1663 1664
    }

    if (virDomainDeleteConfig(dom->conn, driver->configDir, driver->autostartDir, vm) < 0)
1665
        goto cleanup;
1666 1667 1668

    virDomainRemoveInactive(&driver->domains,
                            vm);
1669
    vm = NULL;
1670
    ret = 0;
1671

1672
cleanup:
1673 1674 1675
    if (vm)
        virDomainObjUnlock(vm);
    umlDriverUnlock(driver);
1676
    return ret;
1677 1678 1679 1680 1681 1682
}



static int umlDomainGetAutostart(virDomainPtr dom,
                            int *autostart) {
1683 1684 1685
    struct uml_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1686

1687
    umlDriverLock(driver);
1688
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1689

1690 1691 1692
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
1693
        goto cleanup;
1694 1695 1696
    }

    *autostart = vm->autostart;
1697
    ret = 0;
1698

1699
cleanup:
1700 1701
    if (vm)
        virDomainObjUnlock(vm);
1702
    umlDriverUnlock(driver);
1703
    return ret;
1704 1705 1706 1707
}

static int umlDomainSetAutostart(virDomainPtr dom,
                                   int autostart) {
1708
    struct uml_driver *driver = dom->conn->privateData;
1709
    virDomainObjPtr vm;
1710 1711 1712
    char *configFile = NULL, *autostartLink = NULL;
    int ret = -1;

1713 1714 1715
    umlDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);

1716 1717 1718
    if (!vm) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
1719
        goto cleanup;
1720 1721 1722 1723 1724
    }

    if (!vm->persistent) {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot set autostart for transient domain"));
1725
        goto cleanup;
1726 1727 1728 1729
    }

    autostart = (autostart != 0);

1730 1731 1732 1733 1734
    if (vm->autostart != autostart) {
        if ((configFile = virDomainConfigFile(dom->conn, driver->configDir, vm->def->name)) == NULL)
            goto cleanup;
        if ((autostartLink = virDomainConfigFile(dom->conn, driver->autostartDir, vm->def->name)) == NULL)
            goto cleanup;
1735

1736 1737
        if (autostart) {
            int err;
1738

1739
            if ((err = virFileMakePath(driver->autostartDir))) {
1740
                virReportSystemError(err,
1741 1742
                                     _("cannot create autostart directory %s"),
                                     driver->autostartDir);
1743 1744
                goto cleanup;
            }
1745

1746
            if (symlink(configFile, autostartLink) < 0) {
1747
                virReportSystemError(errno,
1748 1749
                                     _("Failed to create symlink '%s to '%s'"),
                                     autostartLink, configFile);
1750 1751 1752 1753
                goto cleanup;
            }
        } else {
            if (unlink(autostartLink) < 0 && errno != ENOENT && errno != ENOTDIR) {
1754
                virReportSystemError(errno,
1755 1756
                                     _("Failed to delete symlink '%s'"),
                                     autostartLink);
1757 1758
                goto cleanup;
            }
1759 1760
        }

1761
        vm->autostart = autostart;
1762 1763 1764 1765 1766 1767
    }
    ret = 0;

cleanup:
    VIR_FREE(configFile);
    VIR_FREE(autostartLink);
1768 1769
    if (vm)
        virDomainObjUnlock(vm);
1770
    umlDriverUnlock(driver);
1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781
    return ret;
}


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

1786
    umlDriverLock(driver);
1787
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1788 1789
    umlDriverUnlock(driver);

1790 1791
    if (!vm) {
        umlReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
J
Jim Meyering 已提交
1792
                        "%s", _("no domain with matching uuid"));
1793
        goto cleanup;
1794 1795 1796 1797
    }

    if (!path || path[0] == '\0') {
        umlReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
J
Jim Meyering 已提交
1798
                       "%s", _("NULL or empty path"));
1799
        goto cleanup;
1800 1801 1802 1803 1804
    }

    /* Check the path belongs to this domain. */
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (vm->def->disks[i]->src != NULL &&
1805 1806 1807 1808
            STREQ (vm->def->disks[i]->src, path)) {
            ret = 0;
            break;
        }
1809 1810
    }

1811 1812 1813 1814 1815
    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) {
1816
            virReportSystemError(errno,
1817
                                 _("cannot open %s"), path);
1818 1819
            goto cleanup;
        }
1820

1821 1822 1823 1824 1825 1826
        /* 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) {
1827
            virReportSystemError(errno,
1828
                                 _("cannot read %s"), path);
1829 1830 1831 1832 1833 1834
            goto cleanup;
        }

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

1838
cleanup:
1839
    if (fd >= 0) close (fd);
1840 1841
    if (vm)
        virDomainObjUnlock(vm);
1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854
    return ret;
}



static virDriver umlDriver = {
    VIR_DRV_UML,
    "UML",
    umlOpen, /* open */
    umlClose, /* close */
    NULL, /* supports_feature */
    umlGetType, /* type */
    umlGetVersion, /* version */
1855
    NULL, /* libvirtVersion (impl. in libvirt.c) */
1856
    virGetHostname, /* getHostname */
1857
    NULL, /* getMaxVcpus */
1858
    nodeGetInfo, /* nodeGetInfo */
1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882
    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 */
1883 1884
    NULL, /* domainGetSecurityLabel */
    NULL, /* nodeGetSecurityModel */
1885
    umlDomainDumpXML, /* domainDumpXML */
1886 1887
    NULL, /* domainXMLFromNative */
    NULL, /* domainXMLToNative */
1888 1889
    umlListDefinedDomains, /* listDefinedDomains */
    umlNumDefinedDomains, /* numOfDefinedDomains */
1890 1891 1892 1893
    umlDomainStart, /* domainCreate */
    umlDomainDefine, /* domainDefineXML */
    umlDomainUndefine, /* domainUndefine */
    NULL, /* domainAttachDevice */
1894
    NULL, /* domainAttachDeviceFlags */
1895
    NULL, /* domainDetachDevice */
1896
    NULL, /* domainDetachDeviceFlags */
1897 1898 1899 1900 1901 1902 1903 1904 1905 1906
    umlDomainGetAutostart, /* domainGetAutostart */
    umlDomainSetAutostart, /* domainSetAutostart */
    NULL, /* domainGetSchedulerType */
    NULL, /* domainGetSchedulerParameters */
    NULL, /* domainSetSchedulerParameters */
    NULL, /* domainMigratePrepare */
    NULL, /* domainMigratePerform */
    NULL, /* domainMigrateFinish */
    NULL, /* domainBlockStats */
    NULL, /* domainInterfaceStats */
1907
    NULL, /* domainMemoryStats */
1908 1909
    umlDomainBlockPeek, /* domainBlockPeek */
    NULL, /* domainMemoryPeek */
1910 1911
    nodeGetCellsFreeMemory, /* nodeGetCellsFreeMemory */
    nodeGetFreeMemory,  /* getFreeMemory */
1912
    NULL, /* domainEventRegister */
1913
    NULL, /* domainEventDeregister */
1914 1915
    NULL, /* domainMigratePrepare2 */
    NULL, /* domainMigrateFinish2 */
1916
    NULL, /* nodeDeviceDettach */
1917 1918
    NULL, /* nodeDeviceReAttach */
    NULL, /* nodeDeviceReset */
C
Chris Lalancette 已提交
1919
    NULL, /* domainMigratePrepareTunnel */
1920 1921 1922 1923
    umlIsEncrypted,
    umlIsSecure,
    umlDomainIsActive,
    umlDomainIsPersistent,
J
Jiri Denemark 已提交
1924
    NULL, /* cpuCompare */
1925 1926 1927 1928
};


static virStateDriver umlStateDriver = {
1929
    .name = "UML",
1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940
    .initialize = umlStartup,
    .cleanup = umlShutdown,
    .reload = umlReload,
    .active = umlActive,
};

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