qemu_driver.c 164.0 KB
Newer Older
D
Daniel P. Berrange 已提交
1 2 3
/*
 * driver.c: core driver methods for managing qemu guests
 *
4
 * Copyright (C) 2006, 2007, 2008, 2009 Red Hat, Inc.
D
Daniel P. Berrange 已提交
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
 * Copyright (C) 2006 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>
 */

24
#include <config.h>
25

D
Daniel P. Berrange 已提交
26 27 28 29 30 31 32 33 34 35 36
#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>
37
#include <sys/utsname.h>
38 39 40 41
#include <sys/stat.h>
#include <fcntl.h>
#include <signal.h>
#include <paths.h>
42 43
#include <pwd.h>
#include <stdio.h>
44
#include <sys/wait.h>
45
#include <sys/ioctl.h>
D
Daniel P. Berrange 已提交
46

47
#if HAVE_NUMACTL
48
#define NUMA_VERSION1_COMPATIBILITY 1
49 50 51
#include <numa.h>
#endif

52 53 54 55
#if HAVE_SCHED_H
#include <sched.h>
#endif

56
#include "virterror_internal.h"
57
#include "logging.h"
58
#include "datatypes.h"
59 60
#include "qemu_driver.h"
#include "qemu_conf.h"
J
Jim Meyering 已提交
61
#include "c-ctype.h"
62
#include "event.h"
63
#include "buf.h"
64
#include "util.h"
65
#include "nodeinfo.h"
66
#include "stats_linux.h"
67
#include "capabilities.h"
68
#include "memory.h"
69
#include "uuid.h"
70
#include "domain_conf.h"
71 72
#include "node_device_conf.h"
#include "pci.h"
73 74
#include "security.h"

75

76 77
#define VIR_FROM_THIS VIR_FROM_QEMU

R
Richard W.M. Jones 已提交
78 79 80
/* For storing short-lived temporary files. */
#define TEMPDIR LOCAL_STATE_DIR "/cache/libvirt"

81 82 83 84
#define QEMU_CMD_PROMPT "\n(qemu) "
#define QEMU_PASSWD_PROMPT "Password: "


85 86
static int qemudShutdown(void);

87 88
static void qemuDriverLock(struct qemud_driver *driver)
{
89
    virMutexLock(&driver->lock);
90 91 92
}
static void qemuDriverUnlock(struct qemud_driver *driver)
{
93
    virMutexUnlock(&driver->lock);
94 95
}

96 97 98
static void qemuDomainEventFlush(int timer, void *opaque);
static void qemuDomainEventQueue(struct qemud_driver *driver,
                                 virDomainEventPtr event);
99

100 101
static void qemudDispatchVMEvent(int watch,
                                 int fd,
102 103 104
                                 int events,
                                 void *opaque);

105 106
static int qemudStartVMDaemon(virConnectPtr conn,
                              struct qemud_driver *driver,
107
                              virDomainObjPtr vm,
108 109
                              const char *migrateFrom,
                              int stdin_fd);
110

111 112
static void qemudShutdownVMDaemon(virConnectPtr conn,
                                  struct qemud_driver *driver,
113
                                  virDomainObjPtr vm);
114

115
static int qemudDomainGetMaxVcpus(virDomainPtr dom);
116

117 118 119 120 121 122 123 124
static int qemudMonitorCommand(const virDomainObjPtr vm,
                               const char *cmd,
                               char **reply);
static int qemudMonitorCommandExtra(const virDomainObjPtr vm,
                                    const char *cmd,
                                    const char *extra,
                                    const char *extraPrompt,
                                    char **reply);
125 126 127
static int qemudDomainSetMemoryBalloon(virConnectPtr conn,
                                       virDomainObjPtr vm,
                                       unsigned long newmem);
128

J
Jim Meyering 已提交
129
static struct qemud_driver *qemu_driver = NULL;
130 131


132 133 134 135 136 137
static int
qemudLogFD(virConnectPtr conn, const char* logDir, const char* name)
{
    char logfile[PATH_MAX];
    mode_t logmode;
    uid_t uid = geteuid();
G
Guido Günther 已提交
138
    int ret, fd = -1;
139

G
Guido Günther 已提交
140 141
    if ((ret = snprintf(logfile, sizeof(logfile), "%s/%s.log", logDir, name))
        < 0 || ret >= sizeof(logfile)) {
142
        virReportOOMError(conn);
143 144 145 146 147 148 149 150 151
        return -1;
    }

    logmode = O_CREAT | O_WRONLY;
    if (uid != 0)
        logmode |= O_TRUNC;
    else
        logmode |= O_APPEND;
    if ((fd = open(logfile, logmode, S_IRUSR | S_IWUSR)) < 0) {
152 153 154
        virReportSystemError(conn, errno,
                             _("failed to create logfile %s"),
                             logfile);
155 156
        return -1;
    }
157
    if (virSetCloseExec(fd) < 0) {
158 159
        virReportSystemError(conn, errno, "%s",
                             _("Unable to set VM logfile close-on-exec flag"));
160 161 162 163 164 165 166
        close(fd);
        return -1;
    }
    return fd;
}


167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183
static int
qemudLogReadFD(virConnectPtr conn, const char* logDir, const char* name, off_t pos)
{
    char logfile[PATH_MAX];
    mode_t logmode = O_RDONLY;
    int ret, fd = -1;

    if ((ret = snprintf(logfile, sizeof(logfile), "%s/%s.log", logDir, name))
        < 0 || ret >= sizeof(logfile)) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("failed to build logfile name %s/%s.log"),
                         logDir, name);
        return -1;
    }


    if ((fd = open(logfile, logmode)) < 0) {
184 185 186
        virReportSystemError(conn, errno,
                             _("failed to create logfile %s"),
                             logfile);
187 188
        return -1;
    }
189
    if (virSetCloseExec(fd) < 0) {
190 191
        virReportSystemError(conn, errno, "%s",
                             _("Unable to set VM logfile close-on-exec flag"));
192 193 194 195
        close(fd);
        return -1;
    }
    if (lseek(fd, pos, SEEK_SET) < 0) {
196 197 198
        virReportSystemError(conn, errno,
                             _("Unable to seek to %lld in %s"),
                             (long long) pos, logfile);
199 200 201 202 203 204
        close(fd);
    }
    return fd;
}


205 206 207
static void
qemudAutostartConfigs(struct qemud_driver *driver) {
    unsigned int i;
208 209 210 211 212 213 214 215 216
    /* 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
     */
    virConnectPtr conn = virConnectOpen(getuid() ?
                                        "qemu:///session" :
                                        "qemu:///system");
    /* Ignoring NULL conn which is mostly harmless here */
217

218
    qemuDriverLock(driver);
219
    for (i = 0 ; i < driver->domains.count ; i++) {
220
        virDomainObjPtr vm = driver->domains.objs[i];
221
        virDomainObjLock(vm);
222 223
        if (vm->autostart &&
            !virDomainIsActive(vm)) {
224
            int ret = qemudStartVMDaemon(conn, driver, vm, NULL, -1);
225 226
            if (ret < 0) {
                virErrorPtr err = virGetLastError();
227 228
                VIR_ERROR(_("Failed to autostart VM '%s': %s\n"),
                          vm->def->name,
229
                          err ? err->message : "");
230
            } else {
231 232 233 234 235 236
                virDomainEventPtr event =
                    virDomainEventNewFromObj(vm,
                                             VIR_DOMAIN_EVENT_STARTED,
                                             VIR_DOMAIN_EVENT_STARTED_BOOTED);
                if (event)
                    qemuDomainEventQueue(driver, event);
237
            }
238
        }
239
        virDomainObjUnlock(vm);
240
    }
241
    qemuDriverUnlock(driver);
242

243 244
    if (conn)
        virConnectClose(conn);
245 246
}

247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263

/**
 * qemudRemoveDomainStatus
 *
 * remove all state files of a domain from statedir
 *
 * Returns 0 on success
 */
static int
qemudRemoveDomainStatus(virConnectPtr conn,
                        struct qemud_driver *driver,
                        virDomainObjPtr vm)
{
    int rc = -1;
    char *file = NULL;

    if (virAsprintf(&file, "%s/%s.xml", driver->stateDir, vm->def->name) < 0) {
264
        virReportOOMError(conn);
265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283
        goto cleanup;
    }

    if (unlink(file) < 0 && errno != ENOENT && errno != ENOTDIR) {
        qemudReportError(conn, vm, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("Failed to unlink status file %s"), file);
        goto cleanup;
    }

    if(virFileDeletePid(driver->stateDir, vm->def->name))
        goto cleanup;

    rc = 0;
cleanup:
    VIR_FREE(file);
    return rc;
}


284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314
static int qemudOpenMonitor(virConnectPtr conn,
                            struct qemud_driver* driver,
                            virDomainObjPtr vm,
                            const char *monitor,
                            int reconnect);

/**
 * qemudReconnectVMs
 *
 * Reconnect running vms to the daemon process
 */
static int
qemudReconnectVMs(struct qemud_driver *driver)
{
    int i;

    for (i = 0 ; i < driver->domains.count ; i++) {
        virDomainObjPtr vm = driver->domains.objs[i];
        qemudDomainStatusPtr status = NULL;
        char *config = NULL;
        int rc;

        virDomainObjLock(vm);
        if ((rc = virFileReadPid(driver->stateDir, vm->def->name, &vm->pid)) == 0)
            DEBUG("Found pid %d for '%s'", vm->pid, vm->def->name);
        else
            goto next;

        if ((config = virDomainConfigFile(NULL,
                                          driver->stateDir,
                                          vm->def->name)) == NULL) {
315 316
            VIR_ERROR(_("Failed to read domain status for %s\n"),
                      vm->def->name);
317 318 319 320 321 322 323 324
            goto next_error;
        }

        status = qemudDomainStatusParseFile(NULL, driver->caps, config, 0);
        if (status) {
            vm->newDef = vm->def;
            vm->def = status->def;
        } else {
325 326
            VIR_ERROR(_("Failed to parse domain status for %s\n"),
                      vm->def->name);
327 328 329 330
            goto next_error;
        }

        if ((rc = qemudOpenMonitor(NULL, driver, vm, status->monitorpath, 1)) != 0) {
331 332
            VIR_ERROR(_("Failed to reconnect monitor for %s: %d\n"),
                      vm->def->name, rc);
333
            goto next_error;
334
        }
335

336
        if ((vm->logfile = qemudLogFD(NULL, driver->logDir, vm->def->name)) < 0)
337
            goto next_error;
338 339 340 341 342 343 344 345 346 347 348

        if (vm->def->id >= driver->nextvmid)
            driver->nextvmid = vm->def->id + 1;

        vm->state = status->state;
        goto next;

next_error:
        /* we failed to reconnect the vm so remove it's traces */
        vm->def->id = -1;
        qemudRemoveDomainStatus(NULL, driver, vm);
349 350 351 352 353 354
        /* Restore orignal def, if we'd loaded a live one */
        if (vm->newDef) {
            virDomainDefFree(vm->def);
            vm->def = vm->newDef;
            vm->newDef = NULL;
        }
355 356
next:
        virDomainObjUnlock(vm);
357 358
        if (status)
            VIR_FREE(status->monitorpath);
359 360 361 362 363 364
        VIR_FREE(status);
        VIR_FREE(config);
    }
    return 0;
}

365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415
static int
qemudSecurityInit(struct qemud_driver *qemud_drv)
{
    int ret;
    const char *doi, *model;
    virCapsPtr caps;
    virSecurityDriverPtr security_drv;

    ret = virSecurityDriverStartup(&security_drv,
                                   qemud_drv->securityDriverName);
    if (ret == -1) {
        VIR_ERROR0(_("Failed to start security driver"));
        return -1;
    }
    /* No security driver wanted to be enabled: just return */
    if (ret == -2) {
        VIR_INFO0(_("No security driver available"));
        return 0;
    }

    qemud_drv->securityDriver = security_drv;
    doi = virSecurityDriverGetDOI(security_drv);
    model = virSecurityDriverGetModel(security_drv);

    VIR_DEBUG("Initialized security driver \"%s\" with "
              "DOI \"%s\"", model, doi);

    /*
     * Add security policy host caps now that the security driver is
     * initialized.
     */
    caps = qemud_drv->caps;

    caps->host.secModel.model = strdup(model);
    if (!caps->host.secModel.model) {
        char ebuf[1024];
        VIR_ERROR(_("Failed to copy secModel model: %s"),
                  virStrerror(errno, ebuf, sizeof ebuf));
        return -1;
    }

    caps->host.secModel.doi = strdup(doi);
    if (!caps->host.secModel.doi) {
        char ebuf[1024];
        VIR_ERROR(_("Failed to copy secModel DOI: %s"),
                  virStrerror(errno, ebuf, sizeof ebuf));
        return -1;
    }

    return 0;
}
416

417 418 419 420 421 422 423
/**
 * qemudStartup:
 *
 * Initialization function for the QEmu daemon
 */
static int
qemudStartup(void) {
424 425
    uid_t uid = geteuid();
    char *base = NULL;
D
Daniel P. Berrange 已提交
426
    char driverConf[PATH_MAX];
427

428
    if (VIR_ALLOC(qemu_driver) < 0)
429 430
        return -1;

431
    if (virMutexInit(&qemu_driver->lock) < 0) {
432
        VIR_ERROR("%s", _("cannot initialize mutex"));
433 434 435
        VIR_FREE(qemu_driver);
        return -1;
    }
436 437
    qemuDriverLock(qemu_driver);

438 439 440
    /* Don't have a dom0 so start from 1 */
    qemu_driver->nextvmid = 1;

441 442
    /* Init callback list */
    if(VIR_ALLOC(qemu_driver->domainEventCallbacks) < 0)
443
        goto out_of_memory;
444 445 446 447 448 449
    if (!(qemu_driver->domainEventQueue = virDomainEventQueueNew()))
        goto out_of_memory;

    if ((qemu_driver->domainEventTimer =
         virEventAddTimeout(-1, qemuDomainEventFlush, qemu_driver, NULL)) < 0)
        goto error;
450

451
    if (!uid) {
452 453
        if (virAsprintf(&qemu_driver->logDir,
                        "%s/log/libvirt/qemu", LOCAL_STATE_DIR) == -1)
454
            goto out_of_memory;
455

D
Daniel P. Berrange 已提交
456
        if ((base = strdup (SYSCONF_DIR "/libvirt")) == NULL)
457
            goto out_of_memory;
458 459 460 461

        if (virAsprintf(&qemu_driver->stateDir,
                      "%s/run/libvirt/qemu/", LOCAL_STATE_DIR) == -1)
            goto out_of_memory;
462
    } else {
463 464
        char *userdir = virGetUserDirectory(NULL, uid);
        if (!userdir)
465
            goto error;
466

467
        if (virAsprintf(&qemu_driver->logDir,
468 469
                        "%s/.libvirt/qemu/log", userdir) == -1) {
            VIR_FREE(userdir);
470
            goto out_of_memory;
471
        }
472

473 474
        if (virAsprintf(&base, "%s/.libvirt", userdir) == -1) {
            VIR_FREE(userdir);
475
            goto out_of_memory;
476 477
        }
        VIR_FREE(userdir);
478 479 480 481 482 483

        if (virAsprintf(&qemu_driver->stateDir, "%s/qemu/run", base) == -1)
            goto out_of_memory;
    }

    if (virFileMakePath(qemu_driver->stateDir) < 0) {
484
        char ebuf[1024];
485 486
        VIR_ERROR(_("Failed to create state dir '%s': %s\n"),
                  qemu_driver->stateDir, virStrerror(errno, ebuf, sizeof ebuf));
487
        goto error;
488 489 490 491 492
    }

    /* Configuration paths are either ~/.libvirt/qemu/... (session) or
     * /etc/libvirt/qemu/... (system).
     */
D
Daniel P. Berrange 已提交
493
    if (snprintf (driverConf, sizeof(driverConf), "%s/qemu.conf", base) == -1)
494
        goto out_of_memory;
D
Daniel P. Berrange 已提交
495
    driverConf[sizeof(driverConf)-1] = '\0';
496

497
    if (virAsprintf(&qemu_driver->configDir, "%s/qemu", base) == -1)
498 499
        goto out_of_memory;

500
    if (virAsprintf(&qemu_driver->autostartDir, "%s/qemu/autostart", base) == -1)
501 502
        goto out_of_memory;

503
    VIR_FREE(base);
504 505 506

    if ((qemu_driver->caps = qemudCapsInit()) == NULL)
        goto out_of_memory;
D
Daniel P. Berrange 已提交
507

508
    if (qemudLoadDriverConfig(qemu_driver, driverConf) < 0) {
509 510 511
        goto error;
    }

512
    if (qemudSecurityInit(qemu_driver) < 0) {
513
        goto error;
D
Daniel P. Berrange 已提交
514 515
    }

516 517 518 519
    if (virDomainLoadAllConfigs(NULL,
                                qemu_driver->caps,
                                &qemu_driver->domains,
                                qemu_driver->configDir,
520
                                qemu_driver->autostartDir,
521 522
                                NULL, NULL) < 0)
        goto error;
523
    qemudReconnectVMs(qemu_driver);
524 525
    qemuDriverUnlock(qemu_driver);

526 527
    qemudAutostartConfigs(qemu_driver);

528

529 530
    return 0;

531
out_of_memory:
532
    virReportOOMError(NULL);
533 534 535
error:
    if (qemu_driver)
        qemuDriverUnlock(qemu_driver);
536
    VIR_FREE(base);
537
    qemudShutdown();
538 539 540
    return -1;
}

541 542 543 544
static void qemudNotifyLoadDomain(virDomainObjPtr vm, int newVM, void *opaque)
{
    struct qemud_driver *driver = opaque;

545 546 547 548 549 550 551 552
    if (newVM) {
        virDomainEventPtr event =
            virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_DEFINED,
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED);
        if (event)
            qemuDomainEventQueue(driver, event);
    }
553 554
}

555 556 557 558 559 560 561 562
/**
 * qemudReload:
 *
 * Function to restart the QEmu daemon, it will recheck the configuration
 * files and update its state and the networking
 */
static int
qemudReload(void) {
563 564 565
    if (!qemu_driver)
        return 0;

566
    qemuDriverLock(qemu_driver);
567 568 569 570
    virDomainLoadAllConfigs(NULL,
                            qemu_driver->caps,
                            &qemu_driver->domains,
                            qemu_driver->configDir,
571 572
                            qemu_driver->autostartDir,
                            qemudNotifyLoadDomain, qemu_driver);
573
    qemuDriverUnlock(qemu_driver);
574

575
    qemudAutostartConfigs(qemu_driver);
576 577

    return 0;
578 579
}

580 581 582 583 584 585 586 587 588 589
/**
 * qemudActive:
 *
 * Checks if the QEmu daemon is active, i.e. has an active domain or
 * an active network
 *
 * Returns 1 if active, 0 otherwise
 */
static int
qemudActive(void) {
590
    unsigned int i;
591
    int active = 0;
592

593 594 595
    if (!qemu_driver)
        return 0;

596 597 598 599 600 601 602 603
    qemuDriverLock(qemu_driver);
    for (i = 0 ; i < qemu_driver->domains.count ; i++) {
        virDomainObjPtr vm = qemu_driver->domains.objs[i];
        virDomainObjLock(vm);
        if (virDomainIsActive(vm))
            active = 1;
        virDomainObjUnlock(vm);
    }
604

605 606
    qemuDriverUnlock(qemu_driver);
    return active;
607 608
}

609 610 611 612 613 614 615
/**
 * qemudShutdown:
 *
 * Shutdown the QEmu daemon, it will stop all active domains and networks
 */
static int
qemudShutdown(void) {
616

617
    if (!qemu_driver)
618
        return -1;
619

620
    qemuDriverLock(qemu_driver);
621 622
    virCapabilitiesFree(qemu_driver->caps);

623
    virDomainObjListFree(&qemu_driver->domains);
624

625
    VIR_FREE(qemu_driver->securityDriverName);
626
    VIR_FREE(qemu_driver->logDir);
627 628
    VIR_FREE(qemu_driver->configDir);
    VIR_FREE(qemu_driver->autostartDir);
629
    VIR_FREE(qemu_driver->stateDir);
630
    VIR_FREE(qemu_driver->vncTLSx509certdir);
J
Jim Meyering 已提交
631
    VIR_FREE(qemu_driver->vncListen);
632
    VIR_FREE(qemu_driver->vncPassword);
633
    VIR_FREE(qemu_driver->vncSASLdir);
D
Daniel P. Berrange 已提交
634

635 636
    /* Free domain callback list */
    virDomainEventCallbackListFree(qemu_driver->domainEventCallbacks);
637 638 639 640
    virDomainEventQueueFree(qemu_driver->domainEventQueue);

    if (qemu_driver->domainEventTimer != -1)
        virEventRemoveTimeout(qemu_driver->domainEventTimer);
641

642 643 644
    if (qemu_driver->brctl)
        brShutdown(qemu_driver->brctl);

645
    qemuDriverUnlock(qemu_driver);
646
    virMutexDestroy(&qemu_driver->lock);
647
    VIR_FREE(qemu_driver);
648 649

    return 0;
650 651 652
}

/* Return -1 for error, 1 to continue reading and 0 for success */
653
typedef int qemudHandlerMonitorOutput(virConnectPtr conn,
654
                                      virDomainObjPtr vm,
655 656 657
                                      const char *output,
                                      int fd);

658 659 660
/*
 * Returns -1 for error, 0 on end-of-file, 1 for success
 */
661
static int
662
qemudReadMonitorOutput(virConnectPtr conn,
663
                       virDomainObjPtr vm,
664 665
                       int fd,
                       char *buf,
G
Guido Günther 已提交
666
                       size_t buflen,
667
                       qemudHandlerMonitorOutput func,
668 669
                       const char *what,
                       int timeout)
670
{
G
Guido Günther 已提交
671
    size_t got = 0;
672
    buf[0] = '\0';
673
    timeout *= 1000; /* poll wants milli seconds */
674

675
    /* Consume & discard the initial greeting */
676
    while (got < (buflen-1)) {
G
Guido Günther 已提交
677
        ssize_t ret;
678 679

        ret = read(fd, buf+got, buflen-got-1);
680

681 682 683 684 685 686
        if (ret < 0) {
            struct pollfd pfd = { .fd = fd, .events = POLLIN };
            if (errno == EINTR)
                continue;

            if (errno != EAGAIN) {
687 688 689
                virReportSystemError(conn, errno,
                                     _("Failure while reading %s startup output"),
                                     what);
690 691 692
                return -1;
            }

693
            ret = poll(&pfd, 1, timeout);
694
            if (ret == 0) {
695
                qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
696
                                 _("Timed out while reading %s startup output"), what);
697 698 699
                return -1;
            } else if (ret == -1) {
                if (errno != EINTR) {
700 701 702
                    virReportSystemError(conn, errno,
                                         _("Failure while reading %s startup output"),
                                         what);
703 704 705 706 707 708 709 710
                    return -1;
                }
            } else {
                /* Make sure we continue loop & read any further data
                   available before dealing with EOF */
                if (pfd.revents & (POLLIN | POLLHUP))
                    continue;

711
                qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
712
                                 _("Failure while reading %s startup output"), what);
713 714
                return -1;
            }
715 716
        } else if (ret == 0) {
            return 0;
717 718 719
        } else {
            got += ret;
            buf[got] = '\0';
720 721 722 723 724 725
            ret = func(conn, vm, buf, fd);
            if (ret == -1)
                return -1;
            if (ret == 1)
                continue;
            return 1;
726 727 728
        }
    }

729
    qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
730
                     _("Out of space while reading %s startup output"), what);
731 732 733 734
    return -1;

}

735 736 737 738 739 740 741 742 743

/*
 * Returns -1 for error, 0 on success
 */
static int
qemudReadLogOutput(virConnectPtr conn,
                   virDomainObjPtr vm,
                   int fd,
                   char *buf,
G
Guido Günther 已提交
744
                   size_t buflen,
745 746 747 748 749
                   qemudHandlerMonitorOutput func,
                   const char *what,
                   int timeout)
{
    int retries = timeout*10;
750
    int got = 0;
751 752 753
    buf[0] = '\0';

    while (retries) {
G
Guido Günther 已提交
754
        ssize_t ret;
755
        int isdead = 0;
G
Guido Günther 已提交
756

757 758
        if (kill(vm->pid, 0) == -1 && errno == ESRCH)
            isdead = 1;
759

760 761
        ret = saferead(fd, buf+got, buflen-got-1);
        if (ret < 0) {
762 763 764 765 766 767
            virReportSystemError(conn, errno,
                                 _("Failure while reading %s log output"),
                                 what);
            return -1;
        }

768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783
        got += ret;
        buf[got] = '\0';
        if (got == buflen-1) {
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                             _("Out of space while reading %s log output"),
                             what);
            return -1;
        }

        if (isdead) {
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                             _("Process exited while reading %s log output"),
                             what);
            return -1;
        }

784 785 786 787 788 789 790 791 792 793 794 795 796
        ret = func(conn, vm, buf, fd);
        if (ret <= 0)
            return ret;

        usleep(100*1000);
        retries--;
    }
    if (retries == 0)
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("Timed out while reading %s log output"), what);
    return -1;
}

797
static int
798
qemudCheckMonitorPrompt(virConnectPtr conn ATTRIBUTE_UNUSED,
799
                        virDomainObjPtr vm,
800 801 802 803 804 805 806 807 808 809 810
                        const char *output,
                        int fd)
{
    if (strstr(output, "(qemu) ") == NULL)
        return 1; /* keep reading */

    vm->monitor = fd;

    return 0;
}

811
static int qemudOpenMonitor(virConnectPtr conn,
812
                            struct qemud_driver* driver,
813
                            virDomainObjPtr vm,
814 815
                            const char *monitor,
                            int reconnect) {
816 817 818 819
    int monfd;
    char buf[1024];
    int ret = -1;

D
Daniel Veillard 已提交
820
    if ((monfd = open(monitor, O_RDWR)) < 0) {
821
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
822
                         _("Unable to open monitor path %s"), monitor);
823 824
        return -1;
    }
825
    if (virSetCloseExec(monfd) < 0) {
826
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
827
                         "%s", _("Unable to set monitor close-on-exec flag"));
828 829
        goto error;
    }
830
    if (virSetNonBlock(monfd) < 0) {
831
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
832
                         "%s", _("Unable to put monitor into non-blocking mode"));
833 834 835
        goto error;
    }

836
    if (!reconnect) {
837 838 839 840
        if (qemudReadMonitorOutput(conn,
                                   vm, monfd,
                                   buf, sizeof(buf),
                                   qemudCheckMonitorPrompt,
841
                                   "monitor", 10) <= 0)
842 843 844
            ret = -1;
        else
            ret = 0;
845 846 847 848 849 850 851
    } else {
        vm->monitor = monfd;
        ret = 0;
    }

    if (ret != 0)
         goto error;
852

853
    if (!(vm->monitorpath = strdup(monitor))) {
854
        virReportOOMError(conn);
855 856 857
        goto error;
    }

858 859 860
    if ((vm->monitorWatch = virEventAddHandle(vm->monitor, 0,
                                              qemudDispatchVMEvent,
                                              driver, NULL)) < 0)
861 862 863
        goto error;


864 865 866 867
    /* Keep monitor open upon success */
    if (ret == 0)
        return ret;

868 869 870 871 872
 error:
    close(monfd);
    return ret;
}

873
/* Returns -1 for error, 0 success, 1 continue reading */
874 875
static int qemudExtractMonitorPath(virConnectPtr conn,
                                   const char *haystack,
876
                                   size_t *offset,
877
                                   char **path) {
878
    static const char needle[] = "char device redirected to";
879
    char *tmp, *dev;
880

881
    VIR_FREE(*path);
882
    /* First look for our magic string */
883 884 885 886 887
    if (!(tmp = strstr(haystack + *offset, needle))) {
        return 1;
    }
    tmp += sizeof(needle);
    dev = tmp;
888

889 890 891 892 893
    /*
     * And look for first whitespace character and nul terminate
     * to mark end of the pty path
     */
    while (*tmp) {
894
        if (c_isspace(*tmp)) {
895
            if (VIR_ALLOC_N(*path, (tmp-dev)+1) < 0) {
896
                virReportOOMError(conn);
897 898 899 900
                return -1;
            }
            strncpy(*path, dev, (tmp-dev));
            (*path)[(tmp-dev)] = '\0';
901
            /* ... now further update offset till we get EOL */
902
            *offset = tmp - haystack;
903 904
            return 0;
        }
905
        tmp++;
906 907 908 909 910
    }

    /*
     * We found a path, but didn't find any whitespace,
     * so it must be still incomplete - we should at
911 912
     * least see a \n - indicate that we want to carry
     * on trying again
913
     */
914
    return 1;
915 916 917
}

static int
918
qemudFindCharDevicePTYs(virConnectPtr conn,
919
                        virDomainObjPtr vm,
920 921
                        const char *output,
                        int fd ATTRIBUTE_UNUSED)
922
{
923
    struct qemud_driver* driver = conn->privateData;
924
    char *monitor = NULL;
925
    size_t offset = 0;
926
    int ret, i;
927 928 929 930 931

    /* The order in which QEMU prints out the PTY paths is
       the order in which it procsses its monitor, serial
       and parallel device args. This code must match that
       ordering.... */
932

933
    /* So first comes the monitor device */
934 935
    if ((ret = qemudExtractMonitorPath(conn, output, &offset, &monitor)) != 0)
        goto cleanup;
936

937
    /* then the serial devices */
938 939
    for (i = 0 ; i < vm->def->nserials ; i++) {
        virDomainChrDefPtr chr = vm->def->serials[i];
940 941 942 943
        if (chr->type == VIR_DOMAIN_CHR_TYPE_PTY) {
            if ((ret = qemudExtractMonitorPath(conn, output, &offset,
                                               &chr->data.file.path)) != 0)
                goto cleanup;
944 945 946 947
        }
    }

    /* and finally the parallel devices */
948 949
    for (i = 0 ; i < vm->def->nparallels ; i++) {
        virDomainChrDefPtr chr = vm->def->parallels[i];
950 951 952 953
        if (chr->type == VIR_DOMAIN_CHR_TYPE_PTY) {
            if ((ret = qemudExtractMonitorPath(conn, output, &offset,
                                               &chr->data.file.path)) != 0)
                goto cleanup;
954 955 956 957
        }
    }

    /* Got them all, so now open the monitor console */
958
    ret = qemudOpenMonitor(conn, driver, vm, monitor, 0);
959 960 961 962

cleanup:
    VIR_FREE(monitor);
    return ret;
963 964
}

965
static int qemudWaitForMonitor(virConnectPtr conn,
966 967 968
                               struct qemud_driver* driver,
                               virDomainObjPtr vm, off_t pos)
{
969
    char buf[4096]; /* Plenty of space to get startup greeting */
970 971 972 973 974
    int logfd;
    int ret;

    if ((logfd = qemudLogReadFD(conn, driver->logDir, vm->def->name, pos))
        < 0)
975
        return -1;
976

977 978 979
    ret = qemudReadLogOutput(conn, vm, logfd, buf, sizeof(buf),
                             qemudFindCharDevicePTYs,
                             "console", 3);
980
    if (close(logfd) < 0) {
981
        char ebuf[4096];
982
        VIR_WARN(_("Unable to close logfile: %s\n"),
983 984
                 virStrerror(errno, ebuf, sizeof ebuf));
    }
985

986
    if (ret == 0) /* success */
987 988 989 990 991 992
        return 0;

    /* Unexpected end of file - inform user of QEMU log data */
    qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                     _("unable to start guest: %s"), buf);
    return -1;
993 994
}

995 996
static int
qemudDetectVcpuPIDs(virConnectPtr conn,
997
                    virDomainObjPtr vm) {
998 999 1000 1001 1002 1003
    char *qemucpus = NULL;
    char *line;
    int lastVcpu = -1;

    /* Only KVM has seperate threads for CPUs,
       others just use main QEMU process for CPU */
1004
    if (vm->def->virtType != VIR_DOMAIN_VIRT_KVM)
1005 1006 1007 1008 1009
        vm->nvcpupids = 1;
    else
        vm->nvcpupids = vm->def->vcpus;

    if (VIR_ALLOC_N(vm->vcpupids, vm->nvcpupids) < 0) {
1010
        virReportOOMError(conn);
1011 1012 1013
        return -1;
    }

1014
    if (vm->def->virtType != VIR_DOMAIN_VIRT_KVM) {
1015 1016 1017 1018
        vm->vcpupids[0] = vm->pid;
        return 0;
    }

1019
    if (qemudMonitorCommand(vm, "info cpus", &qemucpus) < 0) {
1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot run monitor command to fetch CPU thread info"));
        VIR_FREE(vm->vcpupids);
        vm->nvcpupids = 0;
        return -1;
    }

    /*
     * This is the gross format we're about to parse :-{
     *
     * (qemu) info cpus
     * * CPU #0: pc=0x00000000000f0c4a thread_id=30019
     *   CPU #1: pc=0x00000000fffffff0 thread_id=30020
     *   CPU #2: pc=0x00000000fffffff0 thread_id=30021
     *
     */
    line = qemucpus;
    do {
        char *offset = strchr(line, '#');
        char *end = NULL;
        int vcpu = 0, tid = 0;

        /* See if we're all done */
        if (offset == NULL)
            break;

        /* Extract VCPU number */
        if (virStrToLong_i(offset + 1, &end, 10, &vcpu) < 0)
            goto error;
        if (end == NULL || *end != ':')
            goto error;

        /* Extract host Thread ID */
        if ((offset = strstr(line, "thread_id=")) == NULL)
            goto error;
        if (virStrToLong_i(offset + strlen("thread_id="), &end, 10, &tid) < 0)
            goto error;
        if (end == NULL || !c_isspace(*end))
            goto error;

        /* Validate the VCPU is in expected range & order */
        if (vcpu > vm->nvcpupids ||
            vcpu != (lastVcpu + 1))
            goto error;

        lastVcpu = vcpu;
        vm->vcpupids[vcpu] = tid;

        /* Skip to next data line */
        line = strchr(offset, '\r');
        if (line == NULL)
            line = strchr(offset, '\n');
    } while (line != NULL);

    /* Validate we got data for all VCPUs we expected */
    if (lastVcpu != (vm->def->vcpus - 1))
        goto error;

1078
    VIR_FREE(qemucpus);
1079 1080 1081 1082
    return 0;

error:
    VIR_FREE(vm->vcpupids);
1083 1084
    vm->nvcpupids = 0;
    VIR_FREE(qemucpus);
1085 1086 1087 1088 1089 1090 1091 1092

    /* Explicitly return success, not error. Older KVM does
       not have vCPU -> Thread mapping info and we don't
       want to break its use. This merely disables ability
       to pin vCPUS with libvirt */
    return 0;
}

1093 1094
static int
qemudInitCpus(virConnectPtr conn,
D
Daniel Veillard 已提交
1095 1096
              virDomainObjPtr vm,
              const char *migrateFrom) {
1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111
    char *info = NULL;
#if HAVE_SCHED_GETAFFINITY
    cpu_set_t mask;
    int i, maxcpu = QEMUD_CPUMASK_LEN;
    virNodeInfo nodeinfo;

    if (virNodeInfoPopulate(conn, &nodeinfo) < 0)
        return -1;

    /* setaffinity fails if you set bits for CPUs which
     * aren't present, so we have to limit ourselves */
    if (maxcpu > nodeinfo.cpus)
        maxcpu = nodeinfo.cpus;

    CPU_ZERO(&mask);
D
Daniel P. Berrange 已提交
1112 1113 1114 1115 1116 1117
    if (vm->def->cpumask) {
        for (i = 0 ; i < maxcpu ; i++)
            if (vm->def->cpumask[i])
                CPU_SET(i, &mask);
    } else {
        for (i = 0 ; i < maxcpu ; i++)
1118
            CPU_SET(i, &mask);
D
Daniel P. Berrange 已提交
1119
    }
1120 1121 1122 1123

    for (i = 0 ; i < vm->nvcpupids ; i++) {
        if (sched_setaffinity(vm->vcpupids[i],
                              sizeof(mask), &mask) < 0) {
1124 1125
            virReportSystemError(conn, errno, "%s",
                                 _("failed to set CPU affinity"));
1126 1127 1128 1129 1130
            return -1;
        }
    }
#endif /* HAVE_SCHED_GETAFFINITY */

D
Daniel Veillard 已提交
1131 1132
    if (migrateFrom == NULL) {
        /* Allow the CPUS to start executing */
1133
        if (qemudMonitorCommand(vm, "cont", &info) < 0) {
D
Daniel Veillard 已提交
1134 1135 1136 1137 1138
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                             "%s", _("resume operation failed"));
            return -1;
        }
        VIR_FREE(info);
1139 1140 1141 1142 1143 1144
    }

    return 0;
}


1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156
static int
qemudInitPasswords(virConnectPtr conn,
                   struct qemud_driver *driver,
                   virDomainObjPtr vm) {
    char *info = NULL;

    /*
     * NB: Might have more passwords to set in the future. eg a qcow
     * disk decryption password, but there's no monitor command
     * for that yet...
     */

1157 1158 1159
    if ((vm->def->ngraphics == 1) &&
        vm->def->graphics[0]->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
        (vm->def->graphics[0]->data.vnc.passwd || driver->vncPassword)) {
1160 1161

        if (qemudMonitorCommandExtra(vm, "change vnc password",
1162 1163
                                     vm->def->graphics[0]->data.vnc.passwd ?
                                     vm->def->graphics[0]->data.vnc.passwd :
1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177
                                     driver->vncPassword,
                                     QEMU_PASSWD_PROMPT,
                                     &info) < 0) {
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                             "%s", _("setting VNC password failed"));
            return -1;
        }
        VIR_FREE(info);
    }

    return 0;
}


1178
static int qemudNextFreeVNCPort(struct qemud_driver *driver ATTRIBUTE_UNUSED) {
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
    int i;

    for (i = 5900 ; i < 6000 ; i++) {
        int fd;
        int reuse = 1;
        struct sockaddr_in addr;
        addr.sin_family = AF_INET;
        addr.sin_port = htons(i);
        addr.sin_addr.s_addr = htonl(INADDR_ANY);
        fd = socket(PF_INET, SOCK_STREAM, 0);
        if (fd < 0)
            return -1;

        if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (void*)&reuse, sizeof(reuse)) < 0) {
            close(fd);
            break;
        }

        if (bind(fd, (struct sockaddr*)&addr, sizeof(addr)) == 0) {
            /* Not in use, lets grab it */
            close(fd);
            return i;
        }
        close(fd);

        if (errno == EADDRINUSE) {
            /* In use, try next */
            continue;
        }
        /* Some other bad failure, get out.. */
        break;
    }
    return -1;
}

1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231
static int qemuPrepareHostDevices(virConnectPtr conn,
                                  virDomainDefPtr def) {
    int i;

    /* We have to use 2 loops here. *All* devices must
     * be detached before we reset any of them, because
     * in some cases you have to reset the whole PCI,
     * which impacts all devices on it
     */

    for (i = 0 ; i < def->nhostdevs ; i++) {
        virDomainHostdevDefPtr hostdev = def->hostdevs[i];

        if (hostdev->mode != VIR_DOMAIN_HOSTDEV_MODE_SUBSYS)
            continue;
        if (hostdev->source.subsys.type != VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_PCI)
            continue;

1232
        if (hostdev->managed) {
1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286
            pciDevice *dev = pciGetDevice(conn,
                                          hostdev->source.subsys.u.pci.domain,
                                          hostdev->source.subsys.u.pci.bus,
                                          hostdev->source.subsys.u.pci.slot,
                                          hostdev->source.subsys.u.pci.function);
            if (!dev)
                goto error;

            if (pciDettachDevice(conn, dev) < 0) {
                pciFreeDevice(conn, dev);
                goto error;
            }

            pciFreeDevice(conn, dev);
        } /* else {
             XXX validate that non-managed device isn't in use, eg
             by checking that device is either un-bound, or bound
             to pci-stub.ko
        } */
    }

    /* Now that all the PCI hostdevs have be dettached, we can safely
     * reset them */
    for (i = 0 ; i < def->nhostdevs ; i++) {
        virDomainHostdevDefPtr hostdev = def->hostdevs[i];
        pciDevice *dev;

        if (hostdev->mode != VIR_DOMAIN_HOSTDEV_MODE_SUBSYS)
            continue;
        if (hostdev->source.subsys.type != VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_PCI)
            continue;

        dev = pciGetDevice(conn,
                           hostdev->source.subsys.u.pci.domain,
                           hostdev->source.subsys.u.pci.bus,
                           hostdev->source.subsys.u.pci.slot,
                           hostdev->source.subsys.u.pci.function);
        if (!dev)
            goto error;

        if (pciResetDevice(conn, dev) < 0) {
            pciFreeDevice(conn, dev);
            goto error;
        }

        pciFreeDevice(conn, dev);
    }

    return 0;

error:
    return -1;
}

1287 1288 1289 1290 1291 1292 1293 1294 1295
static int qemudDomainSetSecurityLabel(virConnectPtr conn, struct qemud_driver *driver, virDomainObjPtr vm)
{
    if (vm->def->seclabel.label != NULL)
        if (driver->securityDriver && driver->securityDriver->domainSetSecurityLabel)
            return driver->securityDriver->domainSetSecurityLabel(conn, driver->securityDriver,
                                                                 vm);
    return 0;
}

1296 1297 1298
static virDomainPtr qemudDomainLookupByName(virConnectPtr conn,
                                            const char *name);

1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315
struct gemudHookData {
        virConnectPtr conn;
        virDomainObjPtr vm;
        struct qemud_driver *driver;
};

static int qemudSecurityHook(void *data) {
        struct gemudHookData *h = (struct gemudHookData *) data;

        if (qemudDomainSetSecurityLabel(h->conn, h->driver, h->vm) < 0) {
                qemudReportError(h->conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                                 _("Failed to set security label"));
                return -1;
        }
        return 0;
}

1316 1317
static int qemudStartVMDaemon(virConnectPtr conn,
                              struct qemud_driver *driver,
1318
                              virDomainObjPtr vm,
1319 1320
                              const char *migrateFrom,
                              int stdin_fd) {
1321
    const char **argv = NULL, **tmp;
1322
    const char **progenv = NULL;
1323
    int i, ret;
1324
    struct stat sb;
1325 1326
    int *tapfds = NULL;
    int ntapfds = 0;
1327
    unsigned int qemuCmdFlags;
1328
    fd_set keepfd;
1329
    const char *emulator;
G
Guido Günther 已提交
1330
    pid_t child;
1331
    int pos = -1;
1332
    char ebuf[1024];
1333
    char *pidfile = NULL;
1334

1335 1336 1337 1338 1339
    struct gemudHookData hookData;
    hookData.conn = conn;
    hookData.vm = vm;
    hookData.driver = driver;

1340
    FD_ZERO(&keepfd);
1341

1342
    if (virDomainIsActive(vm)) {
1343
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_INVALID,
1344
                         "%s", _("VM is already active"));
1345 1346 1347
        return -1;
    }

1348 1349 1350 1351 1352 1353 1354 1355
    /* If you are using a SecurityDriver with dynamic labelling,
       then generate a security label for isolation */
    if (vm->def->seclabel.type == VIR_DOMAIN_SECLABEL_DYNAMIC &&
        driver->securityDriver &&
        driver->securityDriver->domainGenSecurityLabel &&
        driver->securityDriver->domainGenSecurityLabel(conn, vm) < 0)
        return -1;

1356 1357 1358
    if ((vm->def->ngraphics == 1) &&
        vm->def->graphics[0]->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
        vm->def->graphics[0]->data.vnc.autoport) {
1359
        int port = qemudNextFreeVNCPort(driver);
1360
        if (port < 0) {
1361
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1362
                             "%s", _("Unable to find an unused VNC port"));
1363
            goto cleanup;
1364
        }
1365
        vm->def->graphics[0]->data.vnc.port = port;
1366
    }
1367

1368
    if (virFileMakePath(driver->logDir) < 0) {
1369 1370 1371
        virReportSystemError(conn, errno,
                             _("cannot create log directory %s"),
                             driver->logDir);
1372
        goto cleanup;
1373 1374
    }

1375
    if((vm->logfile = qemudLogFD(conn, driver->logDir, vm->def->name)) < 0)
1376
        goto cleanup;
1377

1378 1379 1380 1381
    emulator = vm->def->emulator;
    if (!emulator)
        emulator = virDomainDefDefaultEmulator(conn, vm->def, driver->caps);
    if (!emulator)
1382
        goto cleanup;
1383

1384 1385 1386 1387
    /* 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
     */
1388
    if (stat(emulator, &sb) < 0) {
1389 1390 1391
        virReportSystemError(conn, errno,
                             _("Cannot find QEMU binary %s"),
                             emulator);
1392
        goto cleanup;
1393 1394
    }

1395
    if (qemudExtractVersionInfo(emulator,
1396
                                NULL,
1397
                                &qemuCmdFlags) < 0)
1398
        goto cleanup;
1399

1400
    if (qemuPrepareHostDevices(conn, vm->def) < 0)
1401
        goto cleanup;
1402

D
Daniel P. Berrange 已提交
1403 1404 1405 1406 1407 1408 1409
    if ((ret = virFileDeletePid(driver->stateDir, vm->def->name)) != 0) {
        virReportSystemError(conn, ret,
                             _("Cannot remove stale PID file for %s"),
                             vm->def->name);
        goto cleanup;
    }

1410 1411 1412 1413 1414
    if (!(pidfile = virFilePid(driver->stateDir, vm->def->name))) {
        virReportSystemError(conn, errno,
                             "%s", _("Failed to build pidfile path."));
        goto cleanup;
    }
D
Daniel P. Berrange 已提交
1415

1416
    vm->def->id = driver->nextvmid++;
1417
    if (qemudBuildCommandLine(conn, driver, vm->def,
1418
                              qemuCmdFlags, &argv, &progenv,
1419 1420
                              &tapfds, &ntapfds, migrateFrom) < 0)
        goto cleanup;
1421

1422 1423 1424
    tmp = progenv;
    while (*tmp) {
        if (safewrite(vm->logfile, *tmp, strlen(*tmp)) < 0)
1425
            VIR_WARN(_("Unable to write envv to logfile: %s\n"),
1426
                     virStrerror(errno, ebuf, sizeof ebuf));
1427
        if (safewrite(vm->logfile, " ", 1) < 0)
1428
            VIR_WARN(_("Unable to write envv to logfile: %s\n"),
1429
                     virStrerror(errno, ebuf, sizeof ebuf));
1430 1431
        tmp++;
    }
1432 1433
    tmp = argv;
    while (*tmp) {
1434
        if (safewrite(vm->logfile, *tmp, strlen(*tmp)) < 0)
1435
            VIR_WARN(_("Unable to write argv to logfile: %s\n"),
1436
                     virStrerror(errno, ebuf, sizeof ebuf));
1437
        if (safewrite(vm->logfile, " ", 1) < 0)
1438
            VIR_WARN(_("Unable to write argv to logfile: %s\n"),
1439
                     virStrerror(errno, ebuf, sizeof ebuf));
1440 1441
        tmp++;
    }
1442
    if (safewrite(vm->logfile, "\n", 1) < 0)
1443
        VIR_WARN(_("Unable to write argv to logfile: %s\n"),
1444
                 virStrerror(errno, ebuf, sizeof ebuf));
1445

1446
    if ((pos = lseek(vm->logfile, 0, SEEK_END)) < 0)
1447
        VIR_WARN(_("Unable to seek to end of logfile: %s\n"),
1448
                 virStrerror(errno, ebuf, sizeof ebuf));
1449

1450 1451 1452
    for (i = 0 ; i < ntapfds ; i++)
        FD_SET(tapfds[i], &keepfd);

1453 1454 1455
    ret = virExecDaemonize(conn, argv, progenv, &keepfd, &child,
                           stdin_fd, &vm->logfile, &vm->logfile,
                           VIR_EXEC_NONBLOCK,
1456 1457 1458
                           qemudSecurityHook, &hookData,
                           pidfile);
    VIR_FREE(pidfile);
G
Guido Günther 已提交
1459 1460 1461

    /* wait for qemu process to to show up */
    if (ret == 0) {
1462
        if (virFileReadPid(driver->stateDir, vm->def->name, &vm->pid)) {
1463
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1464
                             _("Domain %s didn't show up\n"), vm->def->name);
1465
            ret = -1;
G
Guido Günther 已提交
1466
        }
1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477
    } else if (ret == -2) {
        /* The virExec process that launches the daemon failed. Pending on
         * when it failed (we can't determine for sure), there may be
         * extra info in the domain log (if the hook failed for example).
         *
         * Pretend like things succeeded, and let 'WaitForMonitor' report
         * the log contents for us.
         */
        vm->pid = child;
        ret = 0;
    }
1478 1479

    vm->state = migrateFrom ? VIR_DOMAIN_PAUSED : VIR_DOMAIN_RUNNING;
1480

1481
    for (i = 0 ; argv[i] ; i++)
1482 1483
        VIR_FREE(argv[i]);
    VIR_FREE(argv);
1484

1485 1486 1487 1488
    for (i = 0 ; progenv[i] ; i++)
        VIR_FREE(progenv[i]);
    VIR_FREE(progenv);

1489 1490 1491
    if (tapfds) {
        for (i = 0 ; i < ntapfds ; i++) {
            close(tapfds[i]);
1492
        }
1493
        VIR_FREE(tapfds);
1494 1495
    }

1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507
    if (ret == -1)
        goto cleanup;

    if ((qemudWaitForMonitor(conn, driver, vm, pos) < 0) ||
        (qemudDetectVcpuPIDs(conn, vm) < 0) ||
        (qemudInitCpus(conn, vm, migrateFrom) < 0) ||
        (qemudInitPasswords(conn, driver, vm) < 0) ||
        (qemudDomainSetMemoryBalloon(conn, vm, vm->def->memory) < 0) ||
        (qemudSaveDomainStatus(conn, qemu_driver, vm) < 0)) {
        qemudShutdownVMDaemon(conn, driver, vm);
        ret = -1;
        /* No need for 'goto cleanup' now since qemudShutdownVMDaemon does enough */
1508 1509
    }

1510
    return ret;
1511 1512 1513 1514 1515 1516 1517

cleanup:
    if (vm->def->seclabel.type == VIR_DOMAIN_SECLABEL_DYNAMIC) {
        VIR_FREE(vm->def->seclabel.model);
        VIR_FREE(vm->def->seclabel.label);
        VIR_FREE(vm->def->seclabel.imagelabel);
    }
1518 1519 1520 1521
    if ((vm->def->ngraphics == 1) &&
        vm->def->graphics[0]->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
        vm->def->graphics[0]->data.vnc.autoport)
        vm->def->graphics[0]->data.vnc.port = -1;
1522 1523 1524 1525 1526 1527
    if (vm->logfile != -1) {
        close(vm->logfile);
        vm->logfile = -1;
    }
    vm->def->id = -1;
    return -1;
1528 1529 1530
}


1531
static void qemudShutdownVMDaemon(virConnectPtr conn ATTRIBUTE_UNUSED,
1532 1533
                                  struct qemud_driver *driver,
                                  virDomainObjPtr vm) {
D
Daniel P. Berrange 已提交
1534 1535
    int ret;

1536
    if (!virDomainIsActive(vm))
1537
        return;
1538

1539
    VIR_DEBUG(_("Shutting down VM '%s'\n"), vm->def->name);
1540

G
Guido Günther 已提交
1541 1542
    if (virKillProcess(vm->pid, 0) == 0 &&
        virKillProcess(vm->pid, SIGTERM) < 0)
1543 1544 1545
        virReportSystemError(conn, errno,
                             _("Failed to send SIGTERM to %s (%d)"),
                             vm->def->name, vm->pid);
1546

1547 1548 1549
    if (vm->monitorWatch != -1) {
        virEventRemoveHandle(vm->monitorWatch);
        vm->monitorWatch = -1;
1550
    }
1551

1552 1553
    if (close(vm->logfile) < 0) {
        char ebuf[1024];
1554
        VIR_WARN(_("Unable to close logfile: %s\n"),
1555 1556
                 virStrerror(errno, ebuf, sizeof ebuf));
    }
1557 1558 1559 1560 1561
    if (vm->monitor != -1)
        close(vm->monitor);
    vm->logfile = -1;
    vm->monitor = -1;

G
Guido Günther 已提交
1562 1563
    /* shut it off for sure */
    virKillProcess(vm->pid, SIGKILL);
1564

1565 1566 1567 1568
    /* Reset Security Labels */
    if (driver->securityDriver)
        driver->securityDriver->domainRestoreSecurityLabel(conn, vm);

1569 1570 1571 1572 1573 1574 1575
    /* Clear out dynamically assigned labels */
    if (vm->def->seclabel.type == VIR_DOMAIN_SECLABEL_DYNAMIC) {
        VIR_FREE(vm->def->seclabel.model);
        VIR_FREE(vm->def->seclabel.label);
        VIR_FREE(vm->def->seclabel.imagelabel);
    }

1576
    if (qemudRemoveDomainStatus(conn, driver, vm) < 0) {
1577
        VIR_WARN(_("Failed to remove domain status for %s"),
1578 1579
                 vm->def->name);
    }
D
Daniel P. Berrange 已提交
1580 1581 1582 1583 1584 1585
    if ((ret = virFileDeletePid(driver->stateDir, vm->def->name)) != 0) {
        char ebuf[1024];
        VIR_WARN(_("Failed to remove PID file for %s: %s"),
                 vm->def->name, virStrerror(errno, ebuf, sizeof ebuf));
    }

1586
    vm->pid = -1;
1587
    vm->def->id = -1;
1588
    vm->state = VIR_DOMAIN_SHUTOFF;
1589
    VIR_FREE(vm->vcpupids);
1590
    vm->nvcpupids = 0;
1591 1592

    if (vm->newDef) {
1593
        virDomainDefFree(vm->def);
1594
        vm->def = vm->newDef;
1595
        vm->def->id = -1;
1596 1597 1598 1599 1600
        vm->newDef = NULL;
    }
}


1601
static void
1602
qemudDispatchVMEvent(int watch, int fd, int events, void *opaque) {
1603
    struct qemud_driver *driver = opaque;
1604
    virDomainObjPtr vm = NULL;
1605
    virDomainEventPtr event = NULL;
1606
    unsigned int i;
1607
    int quit = 0, failed = 0;
1608

1609
    qemuDriverLock(driver);
1610
    for (i = 0 ; i < driver->domains.count ; i++) {
1611 1612 1613
        virDomainObjPtr tmpvm = driver->domains.objs[i];
        virDomainObjLock(tmpvm);
        if (virDomainIsActive(tmpvm) &&
1614
            tmpvm->monitorWatch == watch) {
1615
            vm = tmpvm;
1616
            break;
1617
        }
1618
        virDomainObjUnlock(tmpvm);
1619 1620 1621
    }

    if (!vm)
1622
        goto cleanup;
1623

1624
    if (vm->monitor != fd) {
1625 1626
        failed = 1;
    } else {
1627
        if (events & (VIR_EVENT_HANDLE_HANGUP | VIR_EVENT_HANDLE_ERROR))
1628
            quit = 1;
1629
        else {
1630 1631
            VIR_ERROR(_("unhandled fd event %d for %s"),
                      events, vm->def->name);
1632
            failed = 1;
1633
        }
1634 1635
    }

1636
    if (failed || quit) {
1637 1638 1639 1640 1641
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         quit ?
                                         VIR_DOMAIN_EVENT_STOPPED_SHUTDOWN :
                                         VIR_DOMAIN_EVENT_STOPPED_FAILED);
1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652
        qemudShutdownVMDaemon(NULL, driver, vm);
        if (!vm->persistent) {
            virDomainRemoveInactive(&driver->domains,
                                    vm);
            vm = NULL;
        }
    }

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
1653 1654
    if (event)
        qemuDomainEventQueue(driver, event);
1655
    qemuDriverUnlock(driver);
1656 1657
}

1658
static int
1659 1660 1661 1662 1663
qemudMonitorCommandExtra(const virDomainObjPtr vm,
                         const char *cmd,
                         const char *extra,
                         const char *extraPrompt,
                         char **reply) {
D
Daniel P. Berrange 已提交
1664 1665
    int size = 0;
    char *buf = NULL;
1666
    size_t cmdlen = strlen(cmd);
1667
    size_t extralen = extra ? strlen(extra) : 0;
D
Daniel P. Berrange 已提交
1668

1669 1670 1671
    if (safewrite(vm->monitor, cmd, cmdlen) != cmdlen)
        return -1;
    if (safewrite(vm->monitor, "\r", 1) != 1)
D
Daniel P. Berrange 已提交
1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683
        return -1;

    *reply = NULL;

    for (;;) {
        struct pollfd fd = { vm->monitor, POLLIN | POLLERR | POLLHUP, 0 };
        char *tmp;

        /* Read all the data QEMU has sent thus far */
        for (;;) {
            char data[1024];
            int got = read(vm->monitor, data, sizeof(data));
D
Daniel P. Berrange 已提交
1684

1685 1686
            if (got == 0)
                goto error;
D
Daniel P. Berrange 已提交
1687 1688 1689 1690 1691
            if (got < 0) {
                if (errno == EINTR)
                    continue;
                if (errno == EAGAIN)
                    break;
1692
                goto error;
1693
            }
1694
            if (VIR_REALLOC_N(buf, size+got+1) < 0)
1695 1696
                goto error;

D
Daniel P. Berrange 已提交
1697 1698 1699 1700
            memmove(buf+size, data, got);
            buf[size+got] = '\0';
            size += got;
        }
1701

D
Daniel P. Berrange 已提交
1702
        /* Look for QEMU prompt to indicate completion */
1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727
        if (buf) {
            if (extra) {
                if (strstr(buf, extraPrompt) != NULL) {
                    if (safewrite(vm->monitor, extra, extralen) != extralen)
                        return -1;
                    if (safewrite(vm->monitor, "\r", 1) != 1)
                        return -1;
                    extra = NULL;
                }
            } else if ((tmp = strstr(buf, QEMU_CMD_PROMPT)) != NULL) {
                char *commptr = NULL, *nlptr = NULL;
                /* Preserve the newline */
                tmp[1] = '\0';

                /* The monitor doesn't dump clean output after we have written to
                 * it. Every character we write dumps a bunch of useless stuff,
                 * so the result looks like "cXcoXcomXcommXcommaXcommanXcommand"
                 * Try to throw away everything before the first full command
                 * occurence, and inbetween the command and the newline starting
                 * the response
                 */
                if ((commptr = strstr(buf, cmd)))
                    memmove(buf, commptr, strlen(commptr)+1);
                if ((nlptr = strchr(buf, '\n')))
                    memmove(buf+strlen(cmd), nlptr, strlen(nlptr)+1);
C
Cole Robinson 已提交
1728

1729 1730
                break;
            }
D
Daniel P. Berrange 已提交
1731 1732 1733 1734 1735 1736
        }
    pollagain:
        /* Need to wait for more data */
        if (poll(&fd, 1, -1) < 0) {
            if (errno == EINTR)
                goto pollagain;
1737
            goto error;
D
Daniel P. Berrange 已提交
1738 1739 1740 1741
        }
    }
    *reply = buf;
    return 0;
1742 1743

 error:
1744
    VIR_FREE(buf);
1745
    return -1;
D
Daniel P. Berrange 已提交
1746 1747
}

1748 1749 1750 1751 1752 1753 1754 1755
static int
qemudMonitorCommand(const virDomainObjPtr vm,
                    const char *cmd,
                    char **reply) {
    return qemudMonitorCommandExtra(vm, cmd, NULL, NULL, reply);
}


1756 1757 1758 1759 1760 1761
/**
 * qemudProbe:
 *
 * Probe for the availability of the qemu driver, assume the
 * presence of QEmu emulation if the binaries are installed
 */
1762
static int qemudProbe(void)
1763 1764 1765
{
    if ((virFileExists("/usr/bin/qemu")) ||
        (virFileExists("/usr/bin/qemu-kvm")) ||
G
Guido Günther 已提交
1766
        (virFileExists("/usr/bin/kvm")) ||
1767 1768 1769 1770
        (virFileExists("/usr/bin/xenner")))
        return 1;

    return 0;
1771
}
1772

1773
static virDrvOpenStatus qemudOpen(virConnectPtr conn,
1774
                                  virConnectAuthPtr auth ATTRIBUTE_UNUSED,
1775
                                  int flags ATTRIBUTE_UNUSED) {
1776 1777 1778
    uid_t uid = getuid();

    if (qemu_driver == NULL)
1779
        goto decline;
1780

1781 1782 1783 1784 1785 1786
    if (!qemudProbe())
        goto decline;

    if (conn->uri == NULL) {
        conn->uri = xmlParseURI(uid ? "qemu:///session" : "qemu:///system");
        if (!conn->uri) {
1787
            virReportOOMError(conn);
1788 1789 1790 1791
            return VIR_DRV_OPEN_ERROR;
        }
    } else if (conn->uri->scheme == NULL ||
               conn->uri->path == NULL)
1792 1793
        goto decline;

1794
    if (STRNEQ (conn->uri->scheme, "qemu"))
1795 1796
        goto decline;

1797
    if (uid != 0) {
1798
        if (STRNEQ (conn->uri->path, "/session"))
1799
            goto decline;
1800
    } else { /* root */
1801 1802
        if (STRNEQ (conn->uri->path, "/system") &&
            STRNEQ (conn->uri->path, "/session"))
1803
            goto decline;
1804 1805 1806 1807 1808
    }

    conn->privateData = qemu_driver;

    return VIR_DRV_OPEN_SUCCESS;
1809 1810

 decline:
1811
    return VIR_DRV_OPEN_DECLINED;
1812 1813 1814
}

static int qemudClose(virConnectPtr conn) {
1815
    struct qemud_driver *driver = conn->privateData;
1816 1817

    /* Get rid of callbacks registered for this conn */
1818
    qemuDriverLock(driver);
1819
    virDomainEventCallbackListRemoveConn(conn, driver->domainEventCallbacks);
1820
    qemuDriverUnlock(driver);
1821 1822 1823 1824 1825 1826

    conn->privateData = NULL;

    return 0;
}

D
Daniel Veillard 已提交
1827 1828 1829 1830 1831 1832 1833 1834 1835 1836
/* Which features are supported by this driver? */
static int
qemudSupportsFeature (virConnectPtr conn ATTRIBUTE_UNUSED, int feature)
{
    switch (feature) {
    case VIR_DRV_FEATURE_MIGRATION_V2: return 1;
    default: return 0;
    }
}

1837
static const char *qemudGetType(virConnectPtr conn ATTRIBUTE_UNUSED) {
1838
    return "QEMU";
1839 1840
}

1841 1842 1843 1844 1845

static int kvmGetMaxVCPUs(void) {
    int maxvcpus = 1;

    int r, fd;
1846

1847 1848
    fd = open(KVM_DEVICE, O_RDONLY);
    if (fd < 0) {
1849 1850
        virReportSystemError(NULL, errno, _("Unable to open %s"), KVM_DEVICE);
        return -1;
1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861
    }

    r = ioctl(fd, KVM_CHECK_EXTENSION, KVM_CAP_NR_VCPUS);
    if (r > 0)
        maxvcpus = r;

    close(fd);
    return maxvcpus;
}


1862
static int qemudGetMaxVCPUs(virConnectPtr conn, const char *type) {
1863 1864 1865
    if (!type)
        return 16;

1866
    if (STRCASEEQ(type, "qemu"))
1867 1868
        return 16;

1869
    if (STRCASEEQ(type, "kvm"))
1870
        return kvmGetMaxVCPUs();
1871

1872
    if (STRCASEEQ(type, "kqemu"))
1873
        return 1;
1874 1875 1876

    qemudReportError(conn, NULL, NULL, VIR_ERR_INVALID_ARG,
                     _("unknown type '%s'"), type);
1877 1878 1879
    return -1;
}

1880 1881 1882
static int qemudGetNodeInfo(virConnectPtr conn,
                            virNodeInfoPtr nodeinfo) {
    return virNodeInfoPopulate(conn, nodeinfo);
1883 1884 1885
}


1886
static char *qemudGetCapabilities(virConnectPtr conn) {
1887
    struct qemud_driver *driver = conn->privateData;
1888
    char *xml = NULL;
1889

1890
    qemuDriverLock(driver);
1891 1892 1893
    virCapabilitiesFree(qemu_driver->caps);
    if ((qemu_driver->caps = qemudCapsInit()) == NULL ||
        (xml = virCapabilitiesFormatXML(driver->caps)) == NULL)
1894
        virReportOOMError(conn);
1895
    qemuDriverUnlock(driver);
1896

1897
    return xml;
1898 1899 1900
}


1901 1902 1903 1904 1905 1906 1907 1908
#if HAVE_NUMACTL
static int
qemudNodeGetCellsFreeMemory(virConnectPtr conn,
                            unsigned long long *freeMems,
                            int startCell,
                            int maxCells)
{
    int n, lastCell, numCells;
1909
    int ret = -1;
D
Daniel P. Berrange 已提交
1910
    int maxCell;
1911 1912 1913 1914

    if (numa_available() < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("NUMA not supported on this host"));
1915
        goto cleanup;
1916
    }
D
Daniel P. Berrange 已提交
1917 1918 1919 1920 1921 1922 1923
    maxCell = numa_max_node();
    if (startCell > maxCell) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("start cell %d out of range (0-%d)"),
                         startCell, maxCell);
        goto cleanup;
    }
1924
    lastCell = startCell + maxCells - 1;
D
Daniel P. Berrange 已提交
1925 1926
    if (lastCell > maxCell)
        lastCell = maxCell;
1927 1928 1929 1930 1931 1932

    for (numCells = 0, n = startCell ; n <= lastCell ; n++) {
        long long mem;
        if (numa_node_size64(n, &mem) < 0) {
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                             "%s", _("Failed to query NUMA free memory"));
1933
            goto cleanup;
1934 1935 1936
        }
        freeMems[numCells++] = mem;
    }
1937 1938 1939 1940
    ret = numCells;

cleanup:
    return ret;
1941 1942 1943 1944 1945
}

static unsigned long long
qemudNodeGetFreeMemory (virConnectPtr conn)
{
D
Daniel P. Berrange 已提交
1946
    unsigned long long freeMem = 0;
1947
    int n;
1948

1949 1950 1951
    if (numa_available() < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("NUMA not supported on this host"));
1952
        goto cleanup;
1953 1954 1955 1956 1957 1958 1959
    }

    for (n = 0 ; n <= numa_max_node() ; n++) {
        long long mem;
        if (numa_node_size64(n, &mem) < 0) {
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                             "%s", _("Failed to query NUMA free memory"));
1960
            goto cleanup;
1961 1962 1963 1964
        }
        freeMem += mem;
    }

1965
cleanup:
1966 1967 1968 1969
    return freeMem;
}

#endif
1970

D
Daniel P. Berrange 已提交
1971 1972 1973
static int qemudGetProcessInfo(unsigned long long *cpuTime, int pid) {
    char proc[PATH_MAX];
    FILE *pidinfo;
1974
    unsigned long long usertime, systime;
D
Daniel P. Berrange 已提交
1975 1976 1977 1978 1979 1980

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

    if (!(pidinfo = fopen(proc, "r"))) {
1981
        /*printf("cannot read pid info");*/
D
Daniel P. Berrange 已提交
1982 1983 1984 1985 1986
        /* VM probably shut down, so fake 0 */
        *cpuTime = 0;
        return 0;
    }

1987
    if (fscanf(pidinfo, "%*d %*s %*c %*d %*d %*d %*d %*d %*u %*u %*u %*u %*u %llu %llu", &usertime, &systime) != 2) {
1988
        qemudDebug("not enough arg");
D
Daniel P. Berrange 已提交
1989 1990 1991 1992 1993 1994 1995 1996
        return -1;
    }

    /* We got jiffies
     * We want nanoseconds
     * _SC_CLK_TCK is jiffies per second
     * So calulate thus....
     */
1997
    *cpuTime = 1000ull * 1000ull * 1000ull * (usertime + systime) / (unsigned long long)sysconf(_SC_CLK_TCK);
D
Daniel P. Berrange 已提交
1998

1999
    qemudDebug("Got %llu %llu %llu", usertime, systime, *cpuTime);
D
Daniel P. Berrange 已提交
2000 2001 2002 2003 2004 2005 2006

    fclose(pidinfo);

    return 0;
}


2007
static virDomainPtr qemudDomainLookupByID(virConnectPtr conn,
2008
                                          int id) {
2009 2010 2011 2012
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;

2013
    qemuDriverLock(driver);
2014
    vm  = virDomainFindByID(&driver->domains, id);
2015
    qemuDriverUnlock(driver);
2016 2017

    if (!vm) {
2018 2019
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching id %d"), id);
2020
        goto cleanup;
2021 2022
    }

2023
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
2024
    if (dom) dom->id = vm->def->id;
2025 2026

cleanup:
2027 2028
    if (vm)
        virDomainObjUnlock(vm);
2029 2030
    return dom;
}
2031

2032
static virDomainPtr qemudDomainLookupByUUID(virConnectPtr conn,
2033
                                            const unsigned char *uuid) {
2034 2035 2036
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
2037

2038
    qemuDriverLock(driver);
2039
    vm = virDomainFindByUUID(&driver->domains, uuid);
2040 2041
    qemuDriverUnlock(driver);

2042
    if (!vm) {
2043 2044 2045 2046
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(uuid, uuidstr);
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuid);
2047
        goto cleanup;
2048 2049
    }

2050
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
2051
    if (dom) dom->id = vm->def->id;
2052 2053

cleanup:
2054 2055
    if (vm)
        virDomainObjUnlock(vm);
2056 2057
    return dom;
}
2058

2059
static virDomainPtr qemudDomainLookupByName(virConnectPtr conn,
2060
                                            const char *name) {
2061 2062 2063
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
2064

2065
    qemuDriverLock(driver);
2066
    vm = virDomainFindByName(&driver->domains, name);
2067 2068
    qemuDriverUnlock(driver);

2069
    if (!vm) {
2070 2071
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching name '%s'"), name);
2072
        goto cleanup;
2073 2074
    }

2075
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
2076
    if (dom) dom->id = vm->def->id;
2077 2078

cleanup:
2079 2080
    if (vm)
        virDomainObjUnlock(vm);
2081 2082 2083
    return dom;
}

2084
static int qemudGetVersion(virConnectPtr conn, unsigned long *version) {
2085 2086 2087
    struct qemud_driver *driver = conn->privateData;
    int ret = -1;

2088
    qemuDriverLock(driver);
2089
    if (qemudExtractVersion(conn, driver) < 0)
2090
        goto cleanup;
2091

2092
    *version = qemu_driver->qemuVersion;
2093 2094 2095
    ret = 0;

cleanup:
2096
    qemuDriverUnlock(driver);
2097
    return ret;
D
Daniel P. Berrange 已提交
2098 2099
}

2100 2101 2102
static char *
qemudGetHostname (virConnectPtr conn)
{
2103
    char *result;
2104

2105 2106
    result = virGetHostname();
    if (result == NULL) {
2107 2108
        virReportSystemError (conn, errno,
                              "%s", _("failed to determine host name"));
2109 2110 2111
        return NULL;
    }
    /* Caller frees this string. */
2112
    return result;
2113 2114
}

2115
static int qemudListDomains(virConnectPtr conn, int *ids, int nids) {
2116
    struct qemud_driver *driver = conn->privateData;
2117 2118
    int got = 0, i;

2119 2120 2121
    qemuDriverLock(driver);
    for (i = 0 ; i < driver->domains.count && got < nids ; i++) {
        virDomainObjLock(driver->domains.objs[i]);
2122 2123
        if (virDomainIsActive(driver->domains.objs[i]))
            ids[got++] = driver->domains.objs[i]->def->id;
2124 2125 2126
        virDomainObjUnlock(driver->domains.objs[i]);
    }
    qemuDriverUnlock(driver);
2127

D
Daniel P. Berrange 已提交
2128 2129
    return got;
}
2130

2131
static int qemudNumDomains(virConnectPtr conn) {
2132
    struct qemud_driver *driver = conn->privateData;
2133 2134
    int n = 0, i;

2135 2136 2137
    qemuDriverLock(driver);
    for (i = 0 ; i < driver->domains.count ; i++) {
        virDomainObjLock(driver->domains.objs[i]);
2138
        if (virDomainIsActive(driver->domains.objs[i]))
2139
            n++;
2140 2141 2142
        virDomainObjUnlock(driver->domains.objs[i]);
    }
    qemuDriverUnlock(driver);
2143

2144
    return n;
D
Daniel P. Berrange 已提交
2145
}
2146

2147
static virDomainPtr qemudDomainCreate(virConnectPtr conn, const char *xml,
2148
                                      unsigned int flags ATTRIBUTE_UNUSED) {
2149
    struct qemud_driver *driver = conn->privateData;
2150
    virDomainDefPtr def;
2151
    virDomainObjPtr vm = NULL;
2152
    virDomainPtr dom = NULL;
2153
    virDomainEventPtr event = NULL;
D
Daniel P. Berrange 已提交
2154

2155
    qemuDriverLock(driver);
2156 2157
    if (!(def = virDomainDefParseString(conn, driver->caps, xml,
                                        VIR_DOMAIN_XML_INACTIVE)))
2158
        goto cleanup;
2159

2160 2161 2162
    if (virSecurityDriverVerify(conn, def) < 0)
        goto cleanup;

2163
    /* See if a VM with matching UUID already exists */
2164
    vm = virDomainFindByUUID(&driver->domains, def->uuid);
2165
    if (vm) {
2166 2167 2168 2169 2170 2171 2172 2173 2174
        /* UUID matches, but if names don't match, refuse it */
        if (STRNEQ(vm->def->name, def->name)) {
            char uuidstr[VIR_UUID_STRING_BUFLEN];
            virUUIDFormat(vm->def->uuid, uuidstr);
            qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                             _("domain '%s' is already defined with uuid %s"),
                             vm->def->name, uuidstr);
            goto cleanup;
        }
2175

2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193
        /* UUID & name match, but if VM is already active, refuse it */
        if (virDomainIsActive(vm)) {
            qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                             _("domain is already active as '%s'"), vm->def->name);
            goto cleanup;
        }
        virDomainObjUnlock(vm);
    } else {
        /* UUID does not match, but if a name matches, refuse it */
        vm = virDomainFindByName(&driver->domains, def->name);
        if (vm) {
            char uuidstr[VIR_UUID_STRING_BUFLEN];
            virUUIDFormat(vm->def->uuid, uuidstr);
            qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                             _("domain '%s' is already defined with uuid %s"),
                             def->name, uuidstr);
            goto cleanup;
        }
2194
    }
2195

2196 2197
    if (!(vm = virDomainAssignDef(conn,
                                  &driver->domains,
2198 2199 2200 2201
                                  def)))
        goto cleanup;

    def = NULL;
D
Daniel P. Berrange 已提交
2202

2203
    if (qemudStartVMDaemon(conn, driver, vm, NULL, -1) < 0) {
2204 2205
        virDomainRemoveInactive(&driver->domains,
                                vm);
2206
        vm = NULL;
2207
        goto cleanup;
D
Daniel P. Berrange 已提交
2208
    }
2209 2210 2211 2212

    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_BOOTED);
D
Daniel P. Berrange 已提交
2213

2214
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
2215
    if (dom) dom->id = vm->def->id;
2216 2217 2218

cleanup:
    virDomainDefFree(def);
2219 2220
    if (vm)
        virDomainObjUnlock(vm);
2221 2222
    if (event)
        qemuDomainEventQueue(driver, event);
2223
    qemuDriverUnlock(driver);
2224
    return dom;
D
Daniel P. Berrange 已提交
2225 2226 2227
}


2228
static int qemudDomainSuspend(virDomainPtr dom) {
2229
    struct qemud_driver *driver = dom->conn->privateData;
D
Daniel P. Berrange 已提交
2230
    char *info;
2231 2232
    virDomainObjPtr vm;
    int ret = -1;
2233
    virDomainEventPtr event = NULL;
2234

2235
    qemuDriverLock(driver);
2236
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2237

D
Daniel P. Berrange 已提交
2238
    if (!vm) {
2239 2240 2241 2242
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
2243
        goto cleanup;
D
Daniel P. Berrange 已提交
2244
    }
2245
    if (!virDomainIsActive(vm)) {
2246
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID,
2247
                         "%s", _("domain is not running"));
2248
        goto cleanup;
D
Daniel P. Berrange 已提交
2249
    }
2250
    if (vm->state != VIR_DOMAIN_PAUSED) {
2251
        if (qemudMonitorCommand(vm, "stop", &info) < 0) {
2252 2253 2254 2255 2256 2257
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                             "%s", _("suspend operation failed"));
            goto cleanup;
        }
        vm->state = VIR_DOMAIN_PAUSED;
        qemudDebug("Reply %s", info);
2258 2259 2260
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_SUSPENDED,
                                         VIR_DOMAIN_EVENT_SUSPENDED_PAUSED);
2261
        VIR_FREE(info);
D
Daniel P. Berrange 已提交
2262
    }
2263 2264
    if (qemudSaveDomainStatus(dom->conn, driver, vm) < 0)
        goto cleanup;
2265 2266 2267
    ret = 0;

cleanup:
2268 2269
    if (vm)
        virDomainObjUnlock(vm);
2270

2271
    if (event)
2272
        qemuDomainEventQueue(driver, event);
2273
    qemuDriverUnlock(driver);
2274
    return ret;
D
Daniel P. Berrange 已提交
2275 2276 2277
}


2278
static int qemudDomainResume(virDomainPtr dom) {
2279
    struct qemud_driver *driver = dom->conn->privateData;
D
Daniel P. Berrange 已提交
2280
    char *info;
2281 2282
    virDomainObjPtr vm;
    int ret = -1;
2283
    virDomainEventPtr event = NULL;
2284

2285
    qemuDriverLock(driver);
2286
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2287

D
Daniel P. Berrange 已提交
2288
    if (!vm) {
2289 2290 2291 2292
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
2293
        goto cleanup;
D
Daniel P. Berrange 已提交
2294
    }
2295
    if (!virDomainIsActive(vm)) {
2296
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID,
2297
                         "%s", _("domain is not running"));
2298
        goto cleanup;
D
Daniel P. Berrange 已提交
2299
    }
2300
    if (vm->state == VIR_DOMAIN_PAUSED) {
2301
        if (qemudMonitorCommand(vm, "cont", &info) < 0) {
2302 2303 2304 2305 2306 2307
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                             "%s", _("resume operation failed"));
            goto cleanup;
        }
        vm->state = VIR_DOMAIN_RUNNING;
        qemudDebug("Reply %s", info);
2308 2309 2310
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_RESUMED,
                                         VIR_DOMAIN_EVENT_RESUMED_UNPAUSED);
2311
        VIR_FREE(info);
D
Daniel P. Berrange 已提交
2312
    }
2313 2314
    if (qemudSaveDomainStatus(dom->conn, driver, vm) < 0)
        goto cleanup;
2315 2316 2317
    ret = 0;

cleanup:
2318 2319
    if (vm)
        virDomainObjUnlock(vm);
2320
    if (event)
2321
        qemuDomainEventQueue(driver, event);
2322
    qemuDriverUnlock(driver);
2323
    return ret;
D
Daniel P. Berrange 已提交
2324 2325 2326
}


2327
static int qemudDomainShutdown(virDomainPtr dom) {
2328 2329
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2330
    char* info;
2331
    int ret = -1;
2332

2333
    qemuDriverLock(driver);
2334
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2335 2336
    qemuDriverUnlock(driver);

2337
    if (!vm) {
2338 2339 2340 2341
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
2342
        goto cleanup;
2343 2344
    }

2345
    if (qemudMonitorCommand(vm, "system_powerdown", &info) < 0) {
2346
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2347
                         "%s", _("shutdown operation failed"));
2348
        goto cleanup;
2349
    }
D
Daniel Veillard 已提交
2350
    VIR_FREE(info);
2351 2352 2353
    ret = 0;

cleanup:
2354 2355
    if (vm)
        virDomainObjUnlock(vm);
2356
    return ret;
2357 2358 2359
}


2360
static int qemudDomainDestroy(virDomainPtr dom) {
2361 2362 2363
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
2364
    virDomainEventPtr event = NULL;
2365

2366
    qemuDriverLock(driver);
2367
    vm  = virDomainFindByUUID(&driver->domains, dom->uuid);
D
Daniel P. Berrange 已提交
2368
    if (!vm) {
2369 2370 2371 2372
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
2373
        goto cleanup;
D
Daniel P. Berrange 已提交
2374
    }
2375 2376 2377 2378 2379
    if (!virDomainIsActive(vm)) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("domain is not running"));
        goto cleanup;
    }
2380

2381
    qemudShutdownVMDaemon(dom->conn, driver, vm);
2382 2383 2384
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_DESTROYED);
2385
    if (!vm->persistent) {
2386 2387
        virDomainRemoveInactive(&driver->domains,
                                vm);
2388 2389
        vm = NULL;
    }
2390 2391 2392
    ret = 0;

cleanup:
2393 2394
    if (vm)
        virDomainObjUnlock(vm);
2395 2396
    if (event)
        qemuDomainEventQueue(driver, event);
2397
    qemuDriverUnlock(driver);
2398
    return ret;
D
Daniel P. Berrange 已提交
2399 2400 2401
}


2402
static char *qemudDomainGetOSType(virDomainPtr dom) {
2403 2404 2405
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *type = NULL;
2406

2407
    qemuDriverLock(driver);
2408
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2409
    qemuDriverUnlock(driver);
2410
    if (!vm) {
2411 2412 2413 2414
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
2415
        goto cleanup;
2416 2417
    }

2418
    if (!(type = strdup(vm->def->os.type)))
2419
        virReportOOMError(dom->conn);
2420 2421

cleanup:
2422 2423
    if (vm)
        virDomainObjUnlock(vm);
2424 2425 2426
    return type;
}

2427 2428
/* Returns max memory in kb, 0 if error */
static unsigned long qemudDomainGetMaxMemory(virDomainPtr dom) {
2429 2430 2431
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    unsigned long ret = 0;
2432

2433
    qemuDriverLock(driver);
2434
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2435 2436
    qemuDriverUnlock(driver);

2437
    if (!vm) {
2438 2439
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2440
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
2441
                         _("no domain with matching uuid '%s'"), uuidstr);
2442
        goto cleanup;
2443 2444
    }

2445 2446 2447
    ret = vm->def->maxmem;

cleanup:
2448 2449
    if (vm)
        virDomainObjUnlock(vm);
2450
    return ret;
2451 2452 2453
}

static int qemudDomainSetMaxMemory(virDomainPtr dom, unsigned long newmax) {
2454 2455 2456
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
2457

2458
    qemuDriverLock(driver);
2459
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2460 2461
    qemuDriverUnlock(driver);

2462
    if (!vm) {
2463 2464
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2465
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
2466
                         _("no domain with matching uuid '%s'"), uuidstr);
2467
        goto cleanup;
2468 2469 2470 2471
    }

    if (newmax < vm->def->memory) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
2472
                         "%s", _("cannot set max memory lower than current memory"));
2473
        goto cleanup;;
2474 2475 2476
    }

    vm->def->maxmem = newmax;
2477 2478 2479
    ret = 0;

cleanup:
2480 2481
    if (vm)
        virDomainObjUnlock(vm);
2482
    return ret;
2483 2484
}

2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499

/* The reply from QEMU contains 'ballon: actual=421' where value is in MB */
#define BALLOON_PREFIX "balloon: actual="

/*
 * Returns: 0 if balloon not supported, +1 if balloon query worked
 * or -1 on failure
 */
static int qemudDomainGetMemoryBalloon(virConnectPtr conn,
                                       virDomainObjPtr vm,
                                       unsigned long *currmem) {
    char *reply = NULL;
    int ret = -1;
    char *offset;

2500 2501 2502
    if (!virDomainIsActive(vm))
        return 0;

2503 2504 2505 2506 2507 2508
    if (qemudMonitorCommand(vm, "info balloon", &reply) < 0) {
        qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("could not query memory balloon allocation"));
        goto cleanup;
    }

2509
    DEBUG ("%s: balloon reply: '%s'", vm->def->name, reply);
2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562
    if ((offset = strstr(reply, BALLOON_PREFIX)) != NULL) {
        unsigned int memMB;
        char *end;
        offset += strlen(BALLOON_PREFIX);
        if (virStrToLong_ui(offset, &end, 10, &memMB) < 0) {
            qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                             "%s", _("could not parse memory balloon allocation"));
            goto cleanup;
        }
        *currmem = memMB * 1024;
        ret = 1;
    } else {
        /* We don't raise an error here, since its to be expected that
         * many QEMU's don't support ballooning
         */
        ret = 0;
    }

cleanup:
    VIR_FREE(reply);
    return ret;
}

/*
 * Returns: 0 if balloon not supported, +1 if balloon query worked
 * or -1 on failure
 */
static int qemudDomainSetMemoryBalloon(virConnectPtr conn,
                                       virDomainObjPtr vm,
                                       unsigned long newmem) {
    char *cmd;
    char *reply = NULL;
    int ret = -1;

    /*
     * 'newmem' is in KB, QEMU monitor works in MB, and we all wish
     * we just worked in bytes with unsigned long long everywhere.
     */
    if (virAsprintf(&cmd, "balloon %lu", (newmem / 1024)) < 0) {
        virReportOOMError(conn);
        goto cleanup;
    }

    if (qemudMonitorCommand(vm, cmd, &reply) < 0) {
        qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("could not balloon memory allocation"));
        VIR_FREE(cmd);
        goto cleanup;
    }
    VIR_FREE(cmd);

    /* If the command failed qemu prints: 'unknown command'
     * No message is printed on success it seems */
2563
    DEBUG ("%s: balloon reply: %s",vm->def->name,  reply);
2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576
    if (strstr(reply, "\nunknown command:")) {
        /* Don't set error - it is expected memory balloon fails on many qemu */
        ret = 0;
    } else {
        ret = 1;
    }

cleanup:
    VIR_FREE(reply);
    return ret;
}


2577
static int qemudDomainSetMemory(virDomainPtr dom, unsigned long newmem) {
2578 2579 2580
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
2581

2582
    qemuDriverLock(driver);
2583
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2584
    qemuDriverUnlock(driver);
2585
    if (!vm) {
2586 2587
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2588
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
2589
                         _("no domain with matching uuid '%s'"), uuidstr);
2590
        goto cleanup;
2591 2592 2593 2594
    }

    if (newmem > vm->def->maxmem) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
2595
                         "%s", _("cannot set memory higher than max memory"));
2596
        goto cleanup;
2597 2598
    }

2599 2600 2601 2602 2603 2604 2605 2606 2607
    if (virDomainIsActive(vm)) {
        ret = qemudDomainSetMemoryBalloon(dom->conn, vm, newmem);
        if (ret == 0)
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                             "%s", _("cannot set memory of an active domain"));
    } else {
        vm->def->memory = newmem;
        ret = 0;
    }
2608 2609

cleanup:
2610 2611
    if (vm)
        virDomainObjUnlock(vm);
2612
    return ret;
2613 2614
}

2615
static int qemudDomainGetInfo(virDomainPtr dom,
2616
                              virDomainInfoPtr info) {
2617 2618 2619
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
2620 2621
    int err;
    unsigned long balloon;
2622

2623
    qemuDriverLock(driver);
2624
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2625
    qemuDriverUnlock(driver);
D
Daniel P. Berrange 已提交
2626
    if (!vm) {
2627 2628 2629 2630
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
2631
        goto cleanup;
D
Daniel P. Berrange 已提交
2632 2633
    }

2634
    info->state = vm->state;
D
Daniel P. Berrange 已提交
2635

2636
    if (!virDomainIsActive(vm)) {
2637
        info->cpuTime = 0;
D
Daniel P. Berrange 已提交
2638
    } else {
2639
        if (qemudGetProcessInfo(&(info->cpuTime), vm->pid) < 0) {
2640
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED, ("cannot read cputime for domain"));
2641
            goto cleanup;
D
Daniel P. Berrange 已提交
2642 2643 2644
        }
    }

2645 2646 2647 2648
    err = qemudDomainGetMemoryBalloon(dom->conn, vm, &balloon);
    if (err < 0)
        goto cleanup;

2649
    info->maxMem = vm->def->maxmem;
2650 2651 2652 2653 2654
    if (err == 0)
        /* Balloon not supported, so maxmem is always the allocation */
        info->memory = vm->def->maxmem;
    else
        info->memory = balloon;
2655
    info->nrVirtCpu = vm->def->vcpus;
2656 2657 2658
    ret = 0;

cleanup:
2659 2660
    if (vm)
        virDomainObjUnlock(vm);
2661
    return ret;
D
Daniel P. Berrange 已提交
2662 2663 2664
}


D
Daniel P. Berrange 已提交
2665
static char *qemudEscape(const char *in, int shell)
2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686
{
    int len = 0;
    int i, j;
    char *out;

    /* To pass through the QEMU monitor, we need to use escape
       sequences: \r, \n, \", \\

       To pass through both QEMU + the shell, we need to escape
       the single character ' as the five characters '\\''
    */

    for (i = 0; in[i] != '\0'; i++) {
        switch(in[i]) {
        case '\r':
        case '\n':
        case '"':
        case '\\':
            len += 2;
            break;
        case '\'':
D
Daniel P. Berrange 已提交
2687 2688 2689 2690
            if (shell)
                len += 5;
            else
                len += 1;
2691 2692 2693 2694 2695 2696 2697
            break;
        default:
            len += 1;
            break;
        }
    }

2698
    if (VIR_ALLOC_N(out, len + 1) < 0)
2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716
        return NULL;

    for (i = j = 0; in[i] != '\0'; i++) {
        switch(in[i]) {
        case '\r':
            out[j++] = '\\';
            out[j++] = 'r';
            break;
        case '\n':
            out[j++] = '\\';
            out[j++] = 'n';
            break;
        case '"':
        case '\\':
            out[j++] = '\\';
            out[j++] = in[i];
            break;
        case '\'':
D
Daniel P. Berrange 已提交
2717 2718 2719 2720 2721 2722 2723 2724 2725
            if (shell) {
                out[j++] = '\'';
                out[j++] = '\\';
                out[j++] = '\\';
                out[j++] = '\'';
                out[j++] = '\'';
            } else {
                out[j++] = in[i];
            }
2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736
            break;
        default:
            out[j++] = in[i];
            break;
        }
    }
    out[j] = '\0';

    return out;
}

2737 2738 2739 2740 2741
static char *qemudEscapeMonitorArg(const char *in)
{
    return qemudEscape(in, 0);
}

D
Daniel P. Berrange 已提交
2742 2743 2744 2745
static char *qemudEscapeShellArg(const char *in)
{
    return qemudEscape(in, 1);
}
2746

2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757
#define QEMUD_SAVE_MAGIC "LibvirtQemudSave"
#define QEMUD_SAVE_VERSION 1

struct qemud_save_header {
    char magic[sizeof(QEMUD_SAVE_MAGIC)-1];
    int version;
    int xml_len;
    int was_running;
    int unused[16];
};

2758
static int qemudDomainSave(virDomainPtr dom,
2759
                           const char *path) {
2760 2761 2762 2763 2764 2765 2766
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *command = NULL;
    char *info = NULL;
    int fd = -1;
    char *safe_path = NULL;
    char *xml = NULL;
2767
    struct qemud_save_header header;
2768
    int ret = -1;
2769
    virDomainEventPtr event = NULL;
2770 2771 2772 2773 2774

    memset(&header, 0, sizeof(header));
    memcpy(header.magic, QEMUD_SAVE_MAGIC, sizeof(header.magic));
    header.version = QEMUD_SAVE_VERSION;

2775
    qemuDriverLock(driver);
2776
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2777

D
Daniel P. Berrange 已提交
2778
    if (!vm) {
2779 2780 2781 2782
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
2783
        goto cleanup;
D
Daniel P. Berrange 已提交
2784
    }
2785

2786
    if (!virDomainIsActive(vm)) {
2787
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID,
2788
                         "%s", _("domain is not running"));
2789
        goto cleanup;
D
Daniel P. Berrange 已提交
2790
    }
2791 2792 2793 2794

    /* Pause */
    if (vm->state == VIR_DOMAIN_RUNNING) {
        header.was_running = 1;
2795
        if (qemudMonitorCommand(vm, "stop", &info) < 0) {
2796
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2797
                             "%s", _("suspend operation failed"));
2798
            goto cleanup;
2799
        }
2800 2801 2802
        vm->state = VIR_DOMAIN_PAUSED;
        qemudDebug("Reply %s", info);
        VIR_FREE(info);
2803 2804 2805
    }

    /* Get XML for the domain */
2806
    xml = virDomainDefFormat(dom->conn, vm->def, VIR_DOMAIN_XML_SECURE);
2807 2808
    if (!xml) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2809
                         "%s", _("failed to get domain xml"));
2810
        goto cleanup;
2811 2812 2813 2814 2815 2816
    }
    header.xml_len = strlen(xml) + 1;

    /* Write header to file, followed by XML */
    if ((fd = open(path, O_CREAT|O_TRUNC|O_WRONLY, S_IRUSR|S_IWUSR)) < 0) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2817
                         _("failed to create '%s'"), path);
2818
        goto cleanup;
2819 2820 2821 2822
    }

    if (safewrite(fd, &header, sizeof(header)) != sizeof(header)) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2823
                         "%s", _("failed to write save header"));
2824
        goto cleanup;
2825 2826 2827 2828
    }

    if (safewrite(fd, xml, header.xml_len) != header.xml_len) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2829
                         "%s", _("failed to write xml"));
2830
        goto cleanup;
2831 2832
    }

2833
    if (close(fd) < 0) {
2834 2835 2836
        virReportSystemError(dom->conn, errno,
                             _("unable to save file %s"),
                             path);
2837 2838 2839
        goto cleanup;
    }
    fd = -1;
2840 2841 2842 2843

    /* Migrate to file */
    safe_path = qemudEscapeShellArg(path);
    if (!safe_path) {
2844
        virReportOOMError(dom->conn);
2845
        goto cleanup;
2846
    }
2847
    if (virAsprintf(&command, "migrate \"exec:"
2848
                  "dd of='%s' oflag=append conv=notrunc 2>/dev/null"
2849
                  "\"", safe_path) == -1) {
2850
        virReportOOMError(dom->conn);
2851 2852
        command = NULL;
        goto cleanup;
2853 2854
    }

2855
    if (qemudMonitorCommand(vm, command, &info) < 0) {
2856
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2857
                         "%s", _("migrate operation failed"));
2858
        goto cleanup;
2859 2860
    }

2861
    DEBUG ("%s: migrate reply: %s", vm->def->name, info);
2862 2863 2864 2865 2866 2867 2868

    /* If the command isn't supported then qemu prints:
     * unknown command: migrate" */
    if (strstr(info, "unknown command:")) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                          "%s",
                          _("'migrate' not supported by this qemu"));
2869
        goto cleanup;
2870 2871
    }

2872 2873
    /* Shut it down */
    qemudShutdownVMDaemon(dom->conn, driver, vm);
2874 2875 2876
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_SAVED);
2877
    if (!vm->persistent) {
2878 2879
        virDomainRemoveInactive(&driver->domains,
                                vm);
2880 2881
        vm = NULL;
    }
2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892
    ret = 0;

cleanup:
    if (fd != -1)
        close(fd);
    VIR_FREE(xml);
    VIR_FREE(safe_path);
    VIR_FREE(command);
    VIR_FREE(info);
    if (ret != 0)
        unlink(path);
2893 2894
    if (vm)
        virDomainObjUnlock(vm);
2895 2896
    if (event)
        qemuDomainEventQueue(driver, event);
2897
    qemuDriverUnlock(driver);
2898
    return ret;
D
Daniel P. Berrange 已提交
2899 2900 2901
}


2902
static int qemudDomainSetVcpus(virDomainPtr dom, unsigned int nvcpus) {
2903 2904
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2905
    int max;
2906
    int ret = -1;
2907
    const char *type;
2908

2909
    qemuDriverLock(driver);
2910
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2911 2912
    qemuDriverUnlock(driver);

2913
    if (!vm) {
2914 2915
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2916
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
2917
                         _("no domain with matching uuid '%s'"), uuidstr);
2918
        goto cleanup;
2919 2920
    }

2921
    if (virDomainIsActive(vm)) {
2922
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID, "%s",
2923
                         _("cannot change vcpu count of an active domain"));
2924
        goto cleanup;
2925 2926
    }

2927 2928 2929 2930 2931 2932 2933 2934
    if (!(type = virDomainVirtTypeToString(vm->def->virtType))) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("unknown virt type in domain definition '%d'"),
                         vm->def->virtType);
        goto cleanup;
    }

    if ((max = qemudGetMaxVCPUs(dom->conn, type)) < 0) {
2935 2936
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR, "%s",
                         _("could not determine max vcpus for the domain"));
2937
        goto cleanup;
2938 2939 2940 2941 2942 2943
    }

    if (nvcpus > max) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                         _("requested vcpus is greater than max allowable"
                           " vcpus for the domain: %d > %d"), nvcpus, max);
2944
        goto cleanup;
2945 2946 2947
    }

    vm->def->vcpus = nvcpus;
2948 2949 2950
    ret = 0;

cleanup:
2951 2952
    if (vm)
        virDomainObjUnlock(vm);
2953
    return ret;
2954 2955
}

2956 2957 2958 2959 2960 2961 2962

#if HAVE_SCHED_GETAFFINITY
static int
qemudDomainPinVcpu(virDomainPtr dom,
                   unsigned int vcpu,
                   unsigned char *cpumap,
                   int maplen) {
2963 2964
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2965 2966 2967
    cpu_set_t mask;
    int i, maxcpu;
    virNodeInfo nodeinfo;
2968
    int ret = -1;
2969

2970
    qemuDriverLock(driver);
2971
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2972 2973
    qemuDriverUnlock(driver);

2974 2975 2976 2977 2978 2979 2980 2981
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

2982
    if (!virDomainIsActive(vm)) {
2983
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID,
2984
                         "%s",_("cannot pin vcpus on an inactive domain"));
2985
        goto cleanup;
2986 2987 2988 2989 2990 2991
    }

    if (vcpu > (vm->nvcpupids-1)) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                         _("vcpu number out of range %d > %d"),
                         vcpu, vm->nvcpupids);
2992
        goto cleanup;
2993 2994 2995
    }

    if (virNodeInfoPopulate(dom->conn, &nodeinfo) < 0)
2996
        goto cleanup;
2997 2998 2999 3000 3001 3002 3003 3004 3005 3006 3007 3008 3009

    maxcpu = maplen * 8;
    if (maxcpu > nodeinfo.cpus)
        maxcpu = nodeinfo.cpus;

    CPU_ZERO(&mask);
    for (i = 0 ; i < maxcpu ; i++) {
        if ((cpumap[i/8] >> (i % 8)) & 1)
            CPU_SET(i, &mask);
    }

    if (vm->vcpupids != NULL) {
        if (sched_setaffinity(vm->vcpupids[vcpu], sizeof(mask), &mask) < 0) {
3010 3011
            virReportSystemError(dom->conn, errno, "%s",
                                 _("cannot set affinity"));
3012
            goto cleanup;
3013 3014 3015 3016
        }
    } else {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("cpu affinity is not supported"));
3017
        goto cleanup;
3018
    }
3019
    ret = 0;
3020

3021
cleanup:
3022 3023
    if (vm)
        virDomainObjUnlock(vm);
3024
    return ret;
3025 3026 3027 3028 3029 3030 3031 3032
}

static int
qemudDomainGetVcpus(virDomainPtr dom,
                    virVcpuInfoPtr info,
                    int maxinfo,
                    unsigned char *cpumaps,
                    int maplen) {
3033 3034
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
3035 3036
    virNodeInfo nodeinfo;
    int i, v, maxcpu;
3037
    int ret = -1;
3038

3039
    qemuDriverLock(driver);
3040
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
3041 3042
    qemuDriverUnlock(driver);

3043 3044 3045 3046 3047 3048 3049 3050
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

3051
    if (!virDomainIsActive(vm)) {
3052
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID,
3053
                         "%s",_("cannot pin vcpus on an inactive domain"));
3054
        goto cleanup;
3055 3056 3057
    }

    if (virNodeInfoPopulate(dom->conn, &nodeinfo) < 0)
3058
        goto cleanup;
3059 3060 3061 3062 3063 3064 3065 3066 3067

    maxcpu = maplen * 8;
    if (maxcpu > nodeinfo.cpus)
        maxcpu = nodeinfo.cpus;

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

3068 3069 3070 3071 3072 3073 3074 3075
    if (maxinfo >= 1) {
        if (info != NULL) {
            memset(info, 0, sizeof(*info) * maxinfo);
            for (i = 0 ; i < maxinfo ; i++) {
                info[i].number = i;
                info[i].state = VIR_VCPU_RUNNING;
                /* XXX cpu time, current pCPU mapping */
            }
3076 3077
        }

3078 3079 3080 3081 3082 3083 3084 3085 3086
        if (cpumaps != NULL) {
            memset(cpumaps, 0, maplen * maxinfo);
            if (vm->vcpupids != NULL) {
                for (v = 0 ; v < maxinfo ; v++) {
                    cpu_set_t mask;
                    unsigned char *cpumap = VIR_GET_CPUMAP(cpumaps, maplen, v);
                    CPU_ZERO(&mask);

                    if (sched_getaffinity(vm->vcpupids[v], sizeof(mask), &mask) < 0) {
3087 3088
                        virReportSystemError(dom->conn, errno, "%s",
                                             _("cannot get affinity"));
3089 3090 3091 3092 3093 3094
                        goto cleanup;
                    }

                    for (i = 0 ; i < maxcpu ; i++)
                        if (CPU_ISSET(i, &mask))
                            VIR_USE_CPU(cpumap, i);
3095
                }
3096 3097 3098 3099
            } else {
                qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                                 "%s", _("cpu affinity is not available"));
                goto cleanup;
3100 3101 3102
            }
        }
    }
3103
    ret = maxinfo;
3104

3105
cleanup:
3106 3107
    if (vm)
        virDomainObjUnlock(vm);
3108
    return ret;
3109 3110 3111 3112
}
#endif /* HAVE_SCHED_GETAFFINITY */


3113
static int qemudDomainGetMaxVcpus(virDomainPtr dom) {
3114 3115
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
3116
    const char *type;
3117
    int ret = -1;
3118

3119
    qemuDriverLock(driver);
3120
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
3121 3122
    qemuDriverUnlock(driver);

3123
    if (!vm) {
3124 3125
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
3126
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
3127
                         _("no domain with matching uuid '%s'"), uuidstr);
3128
        goto cleanup;
3129 3130
    }

3131
    if (!(type = virDomainVirtTypeToString(vm->def->virtType))) {
3132 3133 3134
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("unknown virt type in domain definition '%d'"),
                         vm->def->virtType);
3135
        goto cleanup;
3136 3137
    }

3138
    ret = qemudGetMaxVCPUs(dom->conn, type);
3139

3140
cleanup:
3141 3142
    if (vm)
        virDomainObjUnlock(vm);
3143 3144 3145
    return ret;
}

3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158
static int qemudDomainGetSecurityLabel(virDomainPtr dom, virSecurityLabelPtr seclabel)
{
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
    virDomainObjPtr vm;
    const char *type;
    int ret = -1;

    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);

    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
3159
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173 3174 3175 3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199
                         _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    if (!(type = virDomainVirtTypeToString(vm->def->virtType))) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("unknown virt type in domain definition '%d'"),
                         vm->def->virtType);
        goto cleanup;
    }

    /*
     * Theoretically, the pid can be replaced during this operation and
     * return the label of a different process.  If atomicity is needed,
     * further validation will be required.
     *
     * Comment from Dan Berrange:
     *
     *   Well the PID as stored in the virDomainObjPtr can't be changed
     *   because you've got a locked object.  The OS level PID could have
     *   exited, though and in extreme circumstances have cycled through all
     *   PIDs back to ours. We could sanity check that our PID still exists
     *   after reading the label, by checking that our FD connecting to the
     *   QEMU monitor hasn't seen SIGHUP/ERR on poll().
     */
    if (virDomainIsActive(vm)) {
        if (driver->securityDriver && driver->securityDriver->domainGetSecurityLabel) {
            if (driver->securityDriver->domainGetSecurityLabel(dom->conn, vm, seclabel) == -1) {
                qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                                 _("Failed to get security label"));
                goto cleanup;
            }
        }
    }

    ret = 0;

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
3200
    qemuDriverUnlock(driver);
3201 3202 3203
    return ret;
}

3204 3205
static int qemudNodeGetSecurityModel(virConnectPtr conn,
                                     virSecurityModelPtr secmodel)
3206 3207 3208
{
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
    char *p;
3209
    int ret = 0;
3210

3211 3212 3213 3214 3215
    qemuDriverLock(driver);
    if (!driver->securityDriver) {
        ret = -2;
        goto cleanup;
    }
3216

3217 3218 3219 3220 3221
    p = driver->caps->host.secModel.model;
    if (strlen(p) >= VIR_SECURITY_MODEL_BUFLEN-1) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("security model string exceeds max %d bytes"),
                         VIR_SECURITY_MODEL_BUFLEN-1);
3222 3223
        ret = -1;
        goto cleanup;
3224 3225 3226 3227 3228 3229 3230 3231
    }
    strcpy(secmodel->model, p);

    p = driver->caps->host.secModel.doi;
    if (strlen(p) >= VIR_SECURITY_DOI_BUFLEN-1) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("security DOI string exceeds max %d bytes"),
                         VIR_SECURITY_DOI_BUFLEN-1);
3232 3233
        ret = -1;
        goto cleanup;
3234 3235
    }
    strcpy(secmodel->doi, p);
3236 3237 3238 3239

cleanup:
    qemuDriverUnlock(driver);
    return ret;
3240 3241 3242
}

/* TODO: check seclabel restore */
3243
static int qemudDomainRestore(virConnectPtr conn,
3244 3245 3246
                              const char *path) {
    struct qemud_driver *driver = conn->privateData;
    virDomainDefPtr def = NULL;
3247
    virDomainObjPtr vm = NULL;
3248 3249 3250
    int fd = -1;
    int ret = -1;
    char *xml = NULL;
3251
    struct qemud_save_header header;
3252
    virDomainEventPtr event = NULL;
3253

3254
    qemuDriverLock(driver);
3255 3256 3257
    /* Verify the header and read the XML */
    if ((fd = open(path, O_RDONLY)) < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
3258
                         "%s", _("cannot read domain image"));
3259
        goto cleanup;
3260 3261 3262 3263
    }

    if (saferead(fd, &header, sizeof(header)) != sizeof(header)) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
3264
                         "%s", _("failed to read qemu header"));
3265
        goto cleanup;
3266 3267 3268 3269
    }

    if (memcmp(header.magic, QEMUD_SAVE_MAGIC, sizeof(header.magic)) != 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
3270
                         "%s", _("image magic is incorrect"));
3271
        goto cleanup;
3272 3273 3274 3275
    }

    if (header.version > QEMUD_SAVE_VERSION) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
3276
                         _("image version is not supported (%d > %d)"),
3277
                         header.version, QEMUD_SAVE_VERSION);
3278
        goto cleanup;
3279 3280
    }

3281
    if (VIR_ALLOC_N(xml, header.xml_len) < 0) {
3282
        virReportOOMError(conn);
3283
        goto cleanup;
3284 3285 3286 3287
    }

    if (saferead(fd, xml, header.xml_len) != header.xml_len) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
3288
                         "%s", _("failed to read XML"));
3289
        goto cleanup;
3290 3291 3292
    }

    /* Create a domain from this XML */
3293 3294
    if (!(def = virDomainDefParseString(conn, driver->caps, xml,
                                        VIR_DOMAIN_XML_INACTIVE))) {
3295
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
3296
                         "%s", _("failed to parse XML"));
3297
        goto cleanup;
3298 3299
    }

3300
    /* See if a VM with matching UUID already exists */
3301
    vm = virDomainFindByUUID(&driver->domains, def->uuid);
3302
    if (vm) {
3303 3304 3305 3306 3307 3308 3309 3310 3311 3312 3313
        /* UUID matches, but if names don't match, refuse it */
        if (STRNEQ(vm->def->name, def->name)) {
            char uuidstr[VIR_UUID_STRING_BUFLEN];
            virUUIDFormat(vm->def->uuid, uuidstr);
            qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                             _("domain '%s' is already defined with uuid %s"),
                             vm->def->name, uuidstr);
            goto cleanup;
        }

        /* UUID & name match, but if VM is already active, refuse it */
3314
        if (virDomainIsActive(vm)) {
3315
            qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_INVALID,
3316 3317
                             _("domain is already active as '%s'"), vm->def->name);
            goto cleanup;
3318 3319 3320 3321 3322 3323 3324 3325 3326 3327 3328 3329
        }
        virDomainObjUnlock(vm);
    } else {
        /* UUID does not match, but if a name matches, refuse it */
        vm = virDomainFindByName(&driver->domains, def->name);
        if (vm) {
            char uuidstr[VIR_UUID_STRING_BUFLEN];
            virUUIDFormat(vm->def->uuid, uuidstr);
            qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                             _("domain '%s' is already defined with uuid %s"),
                             def->name, uuidstr);
            goto cleanup;
3330
        }
3331 3332
    }

3333 3334 3335
    if (!(vm = virDomainAssignDef(conn,
                                  &driver->domains,
                                  def))) {
3336
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
3337
                         "%s", _("failed to assign new VM"));
3338
        goto cleanup;
3339
    }
3340
    def = NULL;
3341 3342

    /* Set the migration source and start it up. */
3343
    ret = qemudStartVMDaemon(conn, driver, vm, "stdio", fd);
3344
    close(fd);
3345
    fd = -1;
3346
    if (ret < 0) {
3347
        if (!vm->persistent) {
3348 3349
            virDomainRemoveInactive(&driver->domains,
                                    vm);
3350 3351
            vm = NULL;
        }
3352
        goto cleanup;
3353 3354
    }

3355 3356 3357
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_RESTORED);
3358

3359 3360 3361
    /* If it was running before, resume it now. */
    if (header.was_running) {
        char *info;
3362
        if (qemudMonitorCommand(vm, "cont", &info) < 0) {
3363
            qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
3364
                             "%s", _("failed to resume domain"));
3365
            goto cleanup;
3366
        }
3367
        VIR_FREE(info);
3368 3369
        vm->state = VIR_DOMAIN_RUNNING;
    }
3370
    ret = 0;
3371

3372 3373 3374 3375 3376
cleanup:
    virDomainDefFree(def);
    VIR_FREE(xml);
    if (fd != -1)
        close(fd);
3377 3378
    if (vm)
        virDomainObjUnlock(vm);
3379 3380
    if (event)
        qemuDomainEventQueue(driver, event);
3381
    qemuDriverUnlock(driver);
3382
    return ret;
D
Daniel P. Berrange 已提交
3383 3384 3385
}


3386
static char *qemudDomainDumpXML(virDomainPtr dom,
3387
                                int flags) {
3388 3389 3390
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *ret = NULL;
3391 3392
    unsigned long balloon;
    int err;
3393

3394
    qemuDriverLock(driver);
3395
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
3396 3397
    qemuDriverUnlock(driver);

D
Daniel P. Berrange 已提交
3398
    if (!vm) {
3399 3400 3401 3402
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
3403
        goto cleanup;
D
Daniel P. Berrange 已提交
3404 3405
    }

3406 3407 3408 3409 3410 3411 3412
    /* Refresh current memory based on balloon info */
    err = qemudDomainGetMemoryBalloon(dom->conn, vm, &balloon);
    if (err < 0)
        goto cleanup;
    if (err > 0)
        vm->def->memory = balloon;

3413 3414 3415 3416 3417 3418
    ret = virDomainDefFormat(dom->conn,
                             (flags & VIR_DOMAIN_XML_INACTIVE) && vm->newDef ?
                             vm->newDef : vm->def,
                             flags);

cleanup:
3419 3420
    if (vm)
        virDomainObjUnlock(vm);
3421
    return ret;
D
Daniel P. Berrange 已提交
3422 3423 3424
}


3425 3426 3427 3428 3429 3430 3431 3432 3433 3434 3435 3436 3437 3438 3439 3440 3441 3442 3443 3444 3445 3446 3447 3448
static char *qemuDomainXMLFromNative(virConnectPtr conn,
                                     const char *format,
                                     const char *config,
                                     unsigned int flags ATTRIBUTE_UNUSED) {
    virDomainDefPtr def = NULL;
    char *xml = NULL;

    if (STRNEQ(format, QEMU_CONFIG_FORMAT_ARGV)) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INVALID_ARG,
                         _("unsupported config type %s"), format);
        goto cleanup;
    }

    def = qemuParseCommandLineString(conn, config);
    if (!def)
        goto cleanup;

    xml = virDomainDefFormat(conn, def, VIR_DOMAIN_XML_INACTIVE);

cleanup:
    virDomainDefFree(def);
    return xml;
}

3449 3450 3451 3452 3453 3454 3455 3456 3457 3458 3459 3460 3461 3462 3463 3464
static char *qemuDomainXMLToNative(virConnectPtr conn,
                                   const char *format,
                                   const char *xmlData,
                                   unsigned int flags ATTRIBUTE_UNUSED) {
    struct qemud_driver *driver = conn->privateData;
    virDomainDefPtr def = NULL;
    const char *emulator;
    unsigned int qemuCmdFlags;
    struct stat sb;
    const char **retargv = NULL;
    const char **retenv = NULL;
    const char **tmp;
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    char *ret = NULL;
    int i;

3465 3466
    qemuDriverLock(driver);

3467 3468 3469 3470 3471 3472 3473 3474 3475 3476 3477 3478 3479 3480 3481 3482 3483 3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502 3503 3504 3505 3506 3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530 3531 3532 3533 3534 3535 3536 3537 3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555 3556 3557 3558 3559 3560 3561 3562 3563 3564
    if (STRNEQ(format, QEMU_CONFIG_FORMAT_ARGV)) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INVALID_ARG,
                         _("unsupported config type %s"), format);
        goto cleanup;
    }

    def = virDomainDefParseString(conn, driver->caps, xmlData, 0);
    if (!def)
        goto cleanup;

    /* Since we're just exporting args, we can't do bridge/network
     * setups, since libvirt will normally create TAP devices
     * directly. We convert those configs into generic 'ethernet'
     * config and assume the user has suitable 'ifup-qemu' scripts
     */
    for (i = 0 ; i < def->nnets ; i++) {
        virDomainNetDefPtr net = def->nets[i];
        if (net->type == VIR_DOMAIN_NET_TYPE_NETWORK) {
            VIR_FREE(net->data.network.name);

            memset(net, 0, sizeof *net);

            net->type = VIR_DOMAIN_NET_TYPE_ETHERNET;
            net->data.ethernet.dev = NULL;
            net->data.ethernet.script = NULL;
            net->data.ethernet.ipaddr = NULL;
        } else if (net->type == VIR_DOMAIN_NET_TYPE_BRIDGE) {
            char *brname = net->data.bridge.brname;
            char *script = net->data.bridge.script;
            char *ipaddr = net->data.bridge.ipaddr;

            memset(net, 0, sizeof *net);

            net->type = VIR_DOMAIN_NET_TYPE_ETHERNET;
            net->data.ethernet.dev = brname;
            net->data.ethernet.script = script;
            net->data.ethernet.ipaddr = ipaddr;
        }
    }
    for (i = 0 ; i < def->ngraphics ; i++) {
        if (def->graphics[i]->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
            def->graphics[i]->data.vnc.autoport)
            def->graphics[i]->data.vnc.port = 5900;
    }
    emulator = def->emulator;
    if (!emulator)
        emulator = virDomainDefDefaultEmulator(conn, def, driver->caps);
    if (!emulator)
        goto cleanup;

    /* 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(emulator, &sb) < 0) {
        virReportSystemError(conn, errno,
                             _("Cannot find QEMU binary %s"),
                             emulator);
        goto cleanup;
    }

    if (qemudExtractVersionInfo(emulator,
                                NULL,
                                &qemuCmdFlags) < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("Cannot determine QEMU argv syntax %s"),
                         emulator);
        goto cleanup;
    }


    if (qemudBuildCommandLine(conn, driver, def,
                              qemuCmdFlags,
                              &retargv, &retenv,
                              NULL, NULL, /* Don't want it to create TAP devices */
                              NULL) < 0) {
        goto cleanup;
    }

    tmp = retenv;
    while (*tmp) {
        virBufferAdd(&buf, *tmp, strlen(*tmp));
        virBufferAddLit(&buf, " ");
        tmp++;
    }
    tmp = retargv;
    while (*tmp) {
        virBufferAdd(&buf, *tmp, strlen(*tmp));
        virBufferAddLit(&buf, " ");
        tmp++;
    }

    if (virBufferError(&buf))
        goto cleanup;

    ret = virBufferContentAndReset(&buf);

cleanup:
3565
    qemuDriverUnlock(driver);
3566 3567 3568 3569 3570 3571 3572 3573 3574 3575 3576 3577 3578
    for (tmp = retargv ; tmp && *tmp ; tmp++)
        VIR_FREE(*tmp);
    VIR_FREE(retargv);

    for (tmp = retenv ; tmp && *tmp ; tmp++)
        VIR_FREE(*tmp);
    VIR_FREE(retenv);

    virDomainDefFree(def);
    return ret;
}


3579
static int qemudListDefinedDomains(virConnectPtr conn,
3580
                            char **const names, int nnames) {
3581
    struct qemud_driver *driver = conn->privateData;
3582
    int got = 0, i;
3583

3584
    qemuDriverLock(driver);
3585
    for (i = 0 ; i < driver->domains.count && got < nnames ; i++) {
3586
        virDomainObjLock(driver->domains.objs[i]);
3587 3588
        if (!virDomainIsActive(driver->domains.objs[i])) {
            if (!(names[got++] = strdup(driver->domains.objs[i]->def->name))) {
3589
                virReportOOMError(conn);
3590
                virDomainObjUnlock(driver->domains.objs[i]);
3591 3592
                goto cleanup;
            }
3593
        }
3594
        virDomainObjUnlock(driver->domains.objs[i]);
D
Daniel P. Berrange 已提交
3595
    }
3596

3597
    qemuDriverUnlock(driver);
D
Daniel P. Berrange 已提交
3598
    return got;
3599 3600 3601

 cleanup:
    for (i = 0 ; i < got ; i++)
3602
        VIR_FREE(names[i]);
3603
    qemuDriverUnlock(driver);
3604
    return -1;
D
Daniel P. Berrange 已提交
3605 3606
}

3607
static int qemudNumDefinedDomains(virConnectPtr conn) {
3608
    struct qemud_driver *driver = conn->privateData;
3609 3610
    int n = 0, i;

3611
    qemuDriverLock(driver);
3612 3613
    for (i = 0 ; i < driver->domains.count ; i++)
        if (!virDomainIsActive(driver->domains.objs[i]))
3614
            n++;
3615
    qemuDriverUnlock(driver);
3616

3617
    return n;
D
Daniel P. Berrange 已提交
3618 3619 3620
}


3621
static int qemudDomainStart(virDomainPtr dom) {
3622 3623 3624
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
3625
    virDomainEventPtr event = NULL;
3626

3627
    qemuDriverLock(driver);
3628
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
3629

3630
    if (!vm) {
3631 3632 3633 3634
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
3635
        goto cleanup;
3636 3637
    }

3638
    ret = qemudStartVMDaemon(dom->conn, driver, vm, NULL, -1);
3639
    if (ret != -1)
3640 3641 3642
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_STARTED,
                                         VIR_DOMAIN_EVENT_STARTED_BOOTED);
3643 3644

cleanup:
3645 3646
    if (vm)
        virDomainObjUnlock(vm);
3647
    if (event)
3648
        qemuDomainEventQueue(driver, event);
3649
    qemuDriverUnlock(driver);
3650
    return ret;
D
Daniel P. Berrange 已提交
3651 3652 3653
}


3654
static virDomainPtr qemudDomainDefine(virConnectPtr conn, const char *xml) {
3655
    struct qemud_driver *driver = conn->privateData;
3656
    virDomainDefPtr def;
3657
    virDomainObjPtr vm = NULL;
3658
    virDomainPtr dom = NULL;
3659
    virDomainEventPtr event = NULL;
3660
    int newVM = 1;
3661

3662
    qemuDriverLock(driver);
3663 3664
    if (!(def = virDomainDefParseString(conn, driver->caps, xml,
                                        VIR_DOMAIN_XML_INACTIVE)))
3665
        goto cleanup;
3666

3667 3668 3669
    if (virSecurityDriverVerify(conn, def) < 0)
        goto cleanup;

3670 3671
    /* See if a VM with matching UUID already exists */
    vm = virDomainFindByUUID(&driver->domains, def->uuid);
3672
    if (vm) {
3673 3674 3675 3676 3677 3678 3679 3680 3681 3682 3683
        /* UUID matches, but if names don't match, refuse it */
        if (STRNEQ(vm->def->name, def->name)) {
            char uuidstr[VIR_UUID_STRING_BUFLEN];
            virUUIDFormat(vm->def->uuid, uuidstr);
            qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                             _("domain '%s' is already defined with uuid %s"),
                             vm->def->name, uuidstr);
            goto cleanup;
        }

        /* UUID & name match */
3684
        virDomainObjUnlock(vm);
3685
        newVM = 0;
3686 3687 3688 3689 3690 3691 3692 3693 3694 3695 3696
    } else {
        /* UUID does not match, but if a name matches, refuse it */
        vm = virDomainFindByName(&driver->domains, def->name);
        if (vm) {
            char uuidstr[VIR_UUID_STRING_BUFLEN];
            virUUIDFormat(vm->def->uuid, uuidstr);
            qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                             _("domain '%s' is already defined with uuid %s"),
                             def->name, uuidstr);
            goto cleanup;
        }
3697
    }
3698

3699 3700 3701
    if (!(vm = virDomainAssignDef(conn,
                                  &driver->domains,
                                  def))) {
3702
        goto cleanup;
3703
    }
3704
    def = NULL;
3705
    vm->persistent = 1;
3706

3707 3708
    if (virDomainSaveConfig(conn,
                            driver->configDir,
3709
                            vm->newDef ? vm->newDef : vm->def) < 0) {
3710 3711
        virDomainRemoveInactive(&driver->domains,
                                vm);
3712
        vm = NULL;
3713
        goto cleanup;
3714 3715
    }

3716 3717 3718 3719 3720
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_DEFINED,
                                     newVM ?
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED :
                                     VIR_DOMAIN_EVENT_DEFINED_UPDATED);
3721

3722
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
3723
    if (dom) dom->id = vm->def->id;
3724 3725

cleanup:
3726
    virDomainDefFree(def);
3727 3728
    if (vm)
        virDomainObjUnlock(vm);
3729 3730
    if (event)
        qemuDomainEventQueue(driver, event);
3731
    qemuDriverUnlock(driver);
3732
    return dom;
D
Daniel P. Berrange 已提交
3733 3734
}

3735
static int qemudDomainUndefine(virDomainPtr dom) {
3736 3737
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
3738
    virDomainEventPtr event = NULL;
3739
    int ret = -1;
D
Daniel P. Berrange 已提交
3740

3741
    qemuDriverLock(driver);
3742
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
3743

D
Daniel P. Berrange 已提交
3744
    if (!vm) {
3745 3746 3747 3748
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
3749
        goto cleanup;
D
Daniel P. Berrange 已提交
3750 3751
    }

3752
    if (virDomainIsActive(vm)) {
3753
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID,
3754
                         "%s", _("cannot delete active domain"));
3755
        goto cleanup;
D
Daniel P. Berrange 已提交
3756 3757
    }

3758 3759 3760
    if (!vm->persistent) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot undefine transient domain"));
3761
        goto cleanup;
3762 3763 3764
    }

    if (virDomainDeleteConfig(dom->conn, driver->configDir, driver->autostartDir, vm) < 0)
3765
        goto cleanup;
D
Daniel P. Berrange 已提交
3766

3767 3768 3769
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_UNDEFINED,
                                     VIR_DOMAIN_EVENT_UNDEFINED_REMOVED);
3770

3771 3772
    virDomainRemoveInactive(&driver->domains,
                            vm);
3773
    vm = NULL;
3774
    ret = 0;
D
Daniel P. Berrange 已提交
3775

3776
cleanup:
3777 3778
    if (vm)
        virDomainObjUnlock(vm);
3779 3780
    if (event)
        qemuDomainEventQueue(driver, event);
3781
    qemuDriverUnlock(driver);
3782
    return ret;
D
Daniel P. Berrange 已提交
3783 3784
}

3785
/* Return the disks name for use in monitor commands */
3786
static char *qemudDiskDeviceName(const virConnectPtr conn,
3787
                                 const virDomainDiskDefPtr disk) {
3788 3789 3790 3791 3792 3793

    int busid, devid;
    int ret;
    char *devname;

    if (virDiskNameToBusDeviceIndex(disk, &busid, &devid) < 0) {
3794
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
3795 3796 3797 3798 3799 3800 3801
                         _("cannot convert disk '%s' to bus/device index"),
                         disk->dst);
        return NULL;
    }

    switch (disk->bus) {
        case VIR_DOMAIN_DISK_BUS_IDE:
3802
            if (disk->device== VIR_DOMAIN_DISK_DEVICE_DISK)
3803
                ret = virAsprintf(&devname, "ide%d-hd%d", busid, devid);
3804
            else
3805
                ret = virAsprintf(&devname, "ide%d-cd%d", busid, devid);
3806 3807
            break;
        case VIR_DOMAIN_DISK_BUS_SCSI:
3808
            if (disk->device == VIR_DOMAIN_DISK_DEVICE_DISK)
3809
                ret = virAsprintf(&devname, "scsi%d-hd%d", busid, devid);
3810
            else
3811
                ret = virAsprintf(&devname, "scsi%d-cd%d", busid, devid);
3812 3813
            break;
        case VIR_DOMAIN_DISK_BUS_FDC:
3814
            ret = virAsprintf(&devname, "floppy%d", devid);
3815 3816
            break;
        case VIR_DOMAIN_DISK_BUS_VIRTIO:
3817
            ret = virAsprintf(&devname, "virtio%d", devid);
3818 3819
            break;
        default:
3820
            qemudReportError(conn, NULL, NULL, VIR_ERR_NO_SUPPORT,
3821 3822 3823 3824 3825 3826
                             _("Unsupported disk name mapping for bus '%s'"),
                             virDomainDiskBusTypeToString(disk->bus));
            return NULL;
    }

    if (ret == -1) {
3827
        virReportOOMError(conn);
3828 3829 3830 3831 3832 3833
        return NULL;
    }

    return devname;
}

3834 3835
static int qemudDomainChangeEjectableMedia(virConnectPtr conn,
                                           virDomainObjPtr vm,
3836 3837
                                           virDomainDeviceDefPtr dev)
{
3838
    virDomainDiskDefPtr origdisk = NULL, newdisk;
3839
    char *cmd, *reply, *safe_path;
3840
    char *devname = NULL;
3841
    unsigned int qemuCmdFlags;
3842
    int i;
3843

3844
    origdisk = NULL;
3845
    newdisk = dev->data.disk;
3846 3847 3848 3849
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (vm->def->disks[i]->bus == newdisk->bus &&
            STREQ(vm->def->disks[i]->dst, newdisk->dst)) {
            origdisk = vm->def->disks[i];
3850
            break;
3851
        }
3852 3853 3854
    }

    if (!origdisk) {
3855
        qemudReportError(conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
3856 3857 3858 3859 3860 3861 3862 3863
                         _("No device with bus '%s' and target '%s'"),
                         virDomainDiskBusTypeToString(newdisk->bus),
                         newdisk->dst);
        return -1;
    }

    if (qemudExtractVersionInfo(vm->def->emulator,
                                NULL,
3864
                                &qemuCmdFlags) < 0)
3865 3866 3867
        return -1;

    if (qemuCmdFlags & QEMUD_CMD_FLAG_DRIVE) {
3868
        if (!(devname = qemudDiskDeviceName(conn, newdisk)))
3869 3870 3871 3872 3873 3874 3875 3876 3877
            return -1;
    } else {
        /* Back compat for no -drive option */
        if (newdisk->device == VIR_DOMAIN_DISK_DEVICE_FLOPPY)
            devname = strdup(newdisk->dst);
        else if (newdisk->device == VIR_DOMAIN_DISK_DEVICE_CDROM &&
                 STREQ(newdisk->dst, "hdc"))
            devname = strdup("cdrom");
        else {
3878
            qemudReportError(conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
3879 3880 3881 3882 3883 3884 3885 3886
                             _("Emulator version does not support removable "
                               "media for device '%s' and target '%s'"),
                               virDomainDiskDeviceTypeToString(newdisk->device),
                               newdisk->dst);
            return -1;
        }

        if (!devname) {
3887
            virReportOOMError(conn);
3888 3889 3890
            return -1;
        }
    }
3891

3892
    if (newdisk->src) {
3893 3894
        safe_path = qemudEscapeMonitorArg(newdisk->src);
        if (!safe_path) {
3895
            virReportOOMError(conn);
3896
            VIR_FREE(devname);
3897 3898
            return -1;
        }
3899
        if (virAsprintf(&cmd, "change %s \"%s\"", devname, safe_path) == -1) {
3900
            virReportOOMError(conn);
3901
            VIR_FREE(safe_path);
3902
            VIR_FREE(devname);
3903 3904
            return -1;
        }
3905
        VIR_FREE(safe_path);
3906

3907
    } else if (virAsprintf(&cmd, "eject %s", devname) == -1) {
3908
        virReportOOMError(conn);
3909
        VIR_FREE(devname);
3910 3911
        return -1;
    }
3912
    VIR_FREE(devname);
3913

3914
    if (qemudMonitorCommand(vm, cmd, &reply) < 0) {
3915
        qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
C
Cole Robinson 已提交
3916
                         "%s", _("could not change cdrom media"));
3917
        VIR_FREE(cmd);
3918 3919
        return -1;
    }
3920 3921 3922 3923

    /* If the command failed qemu prints:
     * device not found, device is locked ...
     * No message is printed on success it seems */
3924
    DEBUG ("%s: ejectable media change reply: %s", vm->def->name, reply);
3925
    if (strstr(reply, "\ndevice ")) {
3926
        qemudReportError (conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
C
Cole Robinson 已提交
3927
                          _("changing cdrom media failed: %s"), reply);
3928 3929 3930 3931
        VIR_FREE(reply);
        VIR_FREE(cmd);
        return -1;
    }
3932 3933
    VIR_FREE(reply);
    VIR_FREE(cmd);
3934

3935 3936
    VIR_FREE(origdisk->src);
    origdisk->src = newdisk->src;
3937
    newdisk->src = NULL;
3938
    origdisk->type = newdisk->type;
3939 3940 3941
    return 0;
}

3942 3943 3944
static int qemudDomainAttachPciDiskDevice(virConnectPtr conn,
                                          virDomainObjPtr vm,
                                          virDomainDeviceDefPtr dev)
3945 3946
{
    int ret, i;
3947
    char *cmd, *reply, *s;
3948 3949 3950 3951 3952
    char *safe_path;
    const char* type = virDomainDiskBusTypeToString(dev->data.disk->bus);

    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (STREQ(vm->def->disks[i]->dst, dev->data.disk->dst)) {
3953
            qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3954 3955 3956 3957 3958 3959
                           _("target %s already exists"), dev->data.disk->dst);
            return -1;
        }
    }

    if (VIR_REALLOC_N(vm->def->disks, vm->def->ndisks+1) < 0) {
3960
        virReportOOMError(conn);
3961 3962 3963 3964 3965
        return -1;
    }

    safe_path = qemudEscapeMonitorArg(dev->data.disk->src);
    if (!safe_path) {
3966
        virReportOOMError(conn);
3967 3968 3969
        return -1;
    }

3970 3971
    ret = virAsprintf(&cmd, "pci_add 0 storage file=%s,if=%s",
                      safe_path, type);
3972 3973
    VIR_FREE(safe_path);
    if (ret == -1) {
3974
        virReportOOMError(conn);
3975 3976 3977
        return ret;
    }

3978
    if (qemudMonitorCommand(vm, cmd, &reply) < 0) {
3979
        qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3980 3981 3982 3983 3984
                         _("cannot attach %s disk"), type);
        VIR_FREE(cmd);
        return -1;
    }

3985
    DEBUG ("%s: pci_add reply: %s", vm->def->name, reply);
3986 3987
    /* If the command succeeds qemu prints:
     * OK bus 0... */
3988 3989 3990 3991 3992 3993
#define PCI_ATTACH_OK_MSG "OK bus 0, slot "
    if ((s=strstr(reply, PCI_ATTACH_OK_MSG))) {
        char* dummy = s;
        s += strlen(PCI_ATTACH_OK_MSG);

        if (virStrToLong_i ((const char*)s, &dummy, 10, &dev->data.disk->slotnum) == -1)
3994
            VIR_WARN("%s", _("Unable to parse slot number\n"));
3995
    } else {
3996
        qemudReportError (conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3997 3998 3999 4000 4001 4002 4003 4004 4005 4006 4007 4008 4009 4010
                          _("adding %s disk failed"), type);
        VIR_FREE(reply);
        VIR_FREE(cmd);
        return -1;
    }

    vm->def->disks[vm->def->ndisks++] = dev->data.disk;
    qsort(vm->def->disks, vm->def->ndisks, sizeof(*vm->def->disks),
          virDomainDiskQSort);

    VIR_FREE(reply);
    VIR_FREE(cmd);
    return 0;
}
4011

4012 4013 4014
static int qemudDomainAttachUsbMassstorageDevice(virConnectPtr conn,
                                                 virDomainObjPtr vm,
                                                 virDomainDeviceDefPtr dev)
4015
{
4016 4017
    int ret, i;
    char *safe_path;
4018 4019
    char *cmd, *reply;

4020 4021
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (STREQ(vm->def->disks[i]->dst, dev->data.disk->dst)) {
4022
            qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
4023 4024 4025 4026 4027
                           _("target %s already exists"), dev->data.disk->dst);
            return -1;
        }
    }

4028
    if (VIR_REALLOC_N(vm->def->disks, vm->def->ndisks+1) < 0) {
4029
        virReportOOMError(conn);
4030
        return -1;
4031 4032
    }

4033 4034
    safe_path = qemudEscapeMonitorArg(dev->data.disk->src);
    if (!safe_path) {
4035
        virReportOOMError(conn);
4036 4037 4038
        return -1;
    }

4039
    ret = virAsprintf(&cmd, "usb_add disk:%s", safe_path);
4040
    VIR_FREE(safe_path);
4041
    if (ret == -1) {
4042
        virReportOOMError(conn);
4043 4044 4045
        return ret;
    }

4046
    if (qemudMonitorCommand(vm, cmd, &reply) < 0) {
4047
        qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
4048
                         "%s", _("cannot attach usb disk"));
4049 4050 4051 4052
        VIR_FREE(cmd);
        return -1;
    }

4053
    DEBUG ("%s: attach_usb reply: %s",vm->def->name,  reply);
4054 4055 4056
    /* If the command failed qemu prints:
     * Could not add ... */
    if (strstr(reply, "Could not add ")) {
4057
        qemudReportError (conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
4058
                          "%s",
4059
                          _("adding usb disk failed"));
4060 4061 4062 4063
        VIR_FREE(reply);
        VIR_FREE(cmd);
        return -1;
    }
4064

4065 4066 4067
    vm->def->disks[vm->def->ndisks++] = dev->data.disk;
    qsort(vm->def->disks, vm->def->ndisks, sizeof(*vm->def->disks),
          virDomainDiskQSort);
4068

4069 4070 4071 4072 4073
    VIR_FREE(reply);
    VIR_FREE(cmd);
    return 0;
}

4074 4075 4076
static int qemudDomainAttachHostDevice(virConnectPtr conn,
                                       virDomainObjPtr vm,
                                       virDomainDeviceDefPtr dev)
4077 4078 4079 4080
{
    int ret;
    char *cmd, *reply;

4081
    if (VIR_REALLOC_N(vm->def->hostdevs, vm->def->nhostdevs+1) < 0) {
4082
        virReportOOMError(conn);
4083 4084
        return -1;
    }
4085

4086
    if (dev->data.hostdev->source.subsys.u.usb.vendor) {
4087 4088 4089
        ret = virAsprintf(&cmd, "usb_add host:%.4x:%.4x",
                          dev->data.hostdev->source.subsys.u.usb.vendor,
                          dev->data.hostdev->source.subsys.u.usb.product);
4090
    } else {
4091 4092 4093
        ret = virAsprintf(&cmd, "usb_add host:%.3d.%.3d",
                          dev->data.hostdev->source.subsys.u.usb.bus,
                          dev->data.hostdev->source.subsys.u.usb.device);
4094 4095
    }
    if (ret == -1) {
4096
        virReportOOMError(conn);
4097 4098 4099
        return -1;
    }

4100
    if (qemudMonitorCommand(vm, cmd, &reply) < 0) {
4101
        qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
4102 4103 4104 4105 4106
                         "%s", _("cannot attach usb device"));
        VIR_FREE(cmd);
        return -1;
    }

4107
    DEBUG ("%s: attach_usb reply: %s", vm->def->name, reply);
4108 4109 4110
    /* If the command failed qemu prints:
     * Could not add ... */
    if (strstr(reply, "Could not add ")) {
4111
        qemudReportError (conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
4112 4113 4114 4115 4116 4117
                          "%s",
                          _("adding usb device failed"));
        VIR_FREE(reply);
        VIR_FREE(cmd);
        return -1;
    }
4118

4119
    vm->def->hostdevs[vm->def->nhostdevs++] = dev->data.hostdev;
4120

4121 4122
    VIR_FREE(reply);
    VIR_FREE(cmd);
4123 4124 4125
    return 0;
}

4126 4127
static int qemudDomainAttachDevice(virDomainPtr dom,
                                   const char *xml) {
4128 4129 4130 4131
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    virDomainDeviceDefPtr dev = NULL;
    int ret = -1;
4132

4133
    qemuDriverLock(driver);
4134
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4135
    if (!vm) {
4136 4137 4138 4139
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
4140
        goto cleanup;
4141 4142 4143
    }

    if (!virDomainIsActive(vm)) {
4144
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID,
4145
                         "%s", _("cannot attach device on inactive domain"));
4146
        goto cleanup;
4147 4148
    }

4149 4150
    dev = virDomainDeviceDefParse(dom->conn, driver->caps, vm->def, xml,
                                  VIR_DOMAIN_XML_INACTIVE);
4151 4152 4153
    if (dev == NULL)
        goto cleanup;

4154

4155 4156
    if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
        switch (dev->data.disk->device) {
4157 4158
        case VIR_DOMAIN_DISK_DEVICE_CDROM:
        case VIR_DOMAIN_DISK_DEVICE_FLOPPY:
4159
            ret = qemudDomainChangeEjectableMedia(dom->conn, vm, dev);
4160
            break;
4161

4162 4163
        case VIR_DOMAIN_DISK_DEVICE_DISK:
            if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_USB) {
4164
                ret = qemudDomainAttachUsbMassstorageDevice(dom->conn, vm, dev);
4165 4166
            } else if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_SCSI ||
                       dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_VIRTIO) {
4167
                ret = qemudDomainAttachPciDiskDevice(dom->conn, vm, dev);
4168 4169 4170 4171 4172
            } else {
                qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                                 _("disk bus '%s' cannot be hotplugged."),
                                 virDomainDiskBusTypeToString(dev->data.disk->bus));
                goto cleanup;
4173
            }
4174
            if (driver->securityDriver)
4175
                driver->securityDriver->domainSetSecurityImageLabel(dom->conn, vm, dev->data.disk);
4176
            break;
4177

4178 4179
        default:
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
4180 4181
                             _("disk device type '%s' cannot be hotplugged"),
                             virDomainDiskDeviceTypeToString(dev->data.disk->device));
4182
            goto cleanup;
4183
        }
4184
    } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV &&
4185 4186
               dev->data.hostdev->mode == VIR_DOMAIN_HOSTDEV_MODE_SUBSYS &&
               dev->data.hostdev->source.subsys.type == VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB) {
4187
        ret = qemudDomainAttachHostDevice(dom->conn, vm, dev);
4188
    } else {
4189
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
4190 4191
                         _("device type '%s' cannot be attached"),
                         virDomainDeviceTypeToString(dev->type));
4192
        goto cleanup;
4193 4194
    }

4195 4196 4197
    if (!ret && qemudSaveDomainStatus(dom->conn, driver, vm) < 0)
        ret = -1;

4198
cleanup:
G
Guido Günther 已提交
4199 4200
    if (ret < 0)
        virDomainDeviceDefFree(dev);
4201 4202
    if (vm)
        virDomainObjUnlock(vm);
4203
    qemuDriverUnlock(driver);
4204 4205 4206
    return ret;
}

4207 4208
static int qemudDomainDetachPciDiskDevice(virConnectPtr conn,
                                          virDomainObjPtr vm, virDomainDeviceDefPtr dev)
4209 4210
{
    int i, ret = -1;
4211 4212
    char *cmd = NULL;
    char *reply = NULL;
4213 4214 4215 4216 4217 4218 4219 4220 4221 4222
    virDomainDiskDefPtr detach = NULL;

    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (STREQ(vm->def->disks[i]->dst, dev->data.disk->dst)) {
            detach = vm->def->disks[i];
            break;
        }
    }

    if (!detach) {
4223
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
4224
                         _("disk %s not found"), dev->data.disk->dst);
4225
        goto cleanup;
4226 4227 4228
    }

    if (detach->slotnum < 1) {
4229
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
4230
                         _("disk %s cannot be detached - invalid slot number %d"),
4231
                           detach->dst, detach->slotnum);
4232
        goto cleanup;
4233 4234
    }

4235
    if (virAsprintf(&cmd, "pci_del 0 %d", detach->slotnum) < 0) {
4236
        virReportOOMError(conn);
4237
        goto cleanup;
4238 4239
    }

4240
    if (qemudMonitorCommand(vm, cmd, &reply) < 0) {
4241
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
4242
                          _("failed to execute detach disk %s command"), detach->dst);
4243
        goto cleanup;
4244 4245
    }

4246
    DEBUG ("%s: pci_del reply: %s",vm->def->name,  reply);
4247 4248 4249
    /* If the command fails due to a wrong slot qemu prints: invalid slot,
     * nothing is printed on success */
    if (strstr(reply, "invalid slot")) {
4250
        qemudReportError (conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
4251
                          _("failed to detach disk %s: invalid slot %d"),
4252 4253
                          detach->dst, detach->slotnum);
        goto cleanup;
4254 4255 4256 4257 4258
    }

    if (vm->def->ndisks > 1) {
        vm->def->disks[i] = vm->def->disks[--vm->def->ndisks];
        if (VIR_REALLOC_N(vm->def->disks, vm->def->ndisks) < 0) {
4259
            virReportOOMError(conn);
4260
            goto cleanup;
4261 4262 4263 4264 4265 4266 4267 4268
        }
        qsort(vm->def->disks, vm->def->ndisks, sizeof(*vm->def->disks),
              virDomainDiskQSort);
    } else {
        VIR_FREE(vm->def->disks[0]);
        vm->def->ndisks = 0;
    }
    ret = 0;
4269 4270

cleanup:
4271 4272 4273 4274 4275 4276 4277
    VIR_FREE(reply);
    VIR_FREE(cmd);
    return ret;
}

static int qemudDomainDetachDevice(virDomainPtr dom,
                                   const char *xml) {
4278 4279 4280 4281
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    virDomainDeviceDefPtr dev = NULL;
    int ret = -1;
4282

4283
    qemuDriverLock(driver);
4284
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4285
    if (!vm) {
4286 4287 4288 4289
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
4290
        goto cleanup;
4291 4292 4293
    }

    if (!virDomainIsActive(vm)) {
4294
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID,
G
Guido Günther 已提交
4295
                         "%s", _("cannot detach device on inactive domain"));
4296
        goto cleanup;
4297 4298
    }

4299 4300
    dev = virDomainDeviceDefParse(dom->conn, driver->caps, vm->def, xml,
                                  VIR_DOMAIN_XML_INACTIVE);
4301 4302 4303
    if (dev == NULL)
        goto cleanup;

4304 4305 4306 4307

    if (dev->type == VIR_DOMAIN_DEVICE_DISK &&
        dev->data.disk->device == VIR_DOMAIN_DISK_DEVICE_DISK &&
        (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_SCSI ||
4308
         dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_VIRTIO)) {
4309
        ret = qemudDomainDetachPciDiskDevice(dom->conn, vm, dev);
4310
        if (driver->securityDriver)
4311
            driver->securityDriver->domainRestoreSecurityImageLabel(dom->conn, dev->data.disk);
4312
    }
4313
    else
4314 4315 4316
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("only SCSI or virtio disk device can be detached dynamically"));

4317 4318 4319
    if (!ret && qemudSaveDomainStatus(dom->conn, driver, vm) < 0)
        ret = -1;

4320 4321
cleanup:
    virDomainDeviceDefFree(dev);
4322 4323
    if (vm)
        virDomainObjUnlock(vm);
4324
    qemuDriverUnlock(driver);
4325 4326 4327
    return ret;
}

4328
static int qemudDomainGetAutostart(virDomainPtr dom,
4329
                                   int *autostart) {
4330 4331 4332
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
4333

4334
    qemuDriverLock(driver);
4335
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4336 4337
    qemuDriverUnlock(driver);

4338
    if (!vm) {
4339 4340 4341 4342
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
4343
        goto cleanup;
4344 4345 4346
    }

    *autostart = vm->autostart;
4347
    ret = 0;
4348

4349
cleanup:
4350 4351
    if (vm)
        virDomainObjUnlock(vm);
4352
    return ret;
4353 4354
}

4355
static int qemudDomainSetAutostart(virDomainPtr dom,
4356
                                   int autostart) {
4357 4358
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
4359 4360
    char *configFile = NULL, *autostartLink = NULL;
    int ret = -1;
4361

4362
    qemuDriverLock(driver);
4363
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4364

4365
    if (!vm) {
4366 4367 4368 4369
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
4370
        goto cleanup;
4371 4372
    }

4373 4374 4375
    if (!vm->persistent) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot set autostart for transient domain"));
4376
        goto cleanup;
4377 4378
    }

4379 4380
    autostart = (autostart != 0);

4381 4382 4383 4384 4385
    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;
4386

4387 4388
        if (autostart) {
            int err;
4389

4390
            if ((err = virFileMakePath(driver->autostartDir))) {
4391 4392 4393
                virReportSystemError(dom->conn, err,
                                     _("cannot create autostart directory %s"),
                                     driver->autostartDir);
4394 4395
                goto cleanup;
            }
4396

4397
            if (symlink(configFile, autostartLink) < 0) {
4398 4399 4400
                virReportSystemError(dom->conn, errno,
                                     _("Failed to create symlink '%s to '%s'"),
                                     autostartLink, configFile);
4401 4402 4403 4404
                goto cleanup;
            }
        } else {
            if (unlink(autostartLink) < 0 && errno != ENOENT && errno != ENOTDIR) {
4405 4406 4407
                virReportSystemError(dom->conn, errno,
                                     _("Failed to delete symlink '%s'"),
                                     autostartLink);
4408 4409
                goto cleanup;
            }
4410 4411
        }

4412
        vm->autostart = autostart;
4413
    }
4414
    ret = 0;
4415

4416 4417 4418
cleanup:
    VIR_FREE(configFile);
    VIR_FREE(autostartLink);
4419 4420
    if (vm)
        virDomainObjUnlock(vm);
4421
    qemuDriverUnlock(driver);
4422
    return ret;
4423 4424
}

4425 4426 4427 4428 4429 4430 4431 4432 4433
/* This uses the 'info blockstats' monitor command which was
 * integrated into both qemu & kvm in late 2007.  If the command is
 * not supported we detect this and return the appropriate error.
 */
static int
qemudDomainBlockStats (virDomainPtr dom,
                       const char *path,
                       struct _virDomainBlockStats *stats)
{
4434
    struct qemud_driver *driver = dom->conn->privateData;
4435
    char *dummy, *info = NULL;
4436
    const char *p, *eol;
4437
    const char *qemu_dev_name = NULL;
4438
    size_t len;
4439
    int i, ret = -1;
4440
    virDomainObjPtr vm;
4441
    virDomainDiskDefPtr disk = NULL;
4442

4443
    qemuDriverLock(driver);
4444
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4445
    qemuDriverUnlock(driver);
4446
    if (!vm) {
4447 4448 4449 4450
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
4451
        goto cleanup;
4452
    }
4453
    if (!virDomainIsActive (vm)) {
4454
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID,
4455
                          "%s", _("domain is not running"));
4456
        goto cleanup;
4457 4458
    }

4459 4460 4461 4462 4463 4464 4465 4466
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (STREQ(path, vm->def->disks[i]->dst)) {
            disk = vm->def->disks[i];
            break;
        }
    }

    if (!disk) {
4467 4468
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                          _("invalid path: %s"), path);
4469
        goto cleanup;
4470 4471
    }

4472
    qemu_dev_name = qemudDiskDeviceName(dom->conn, disk);
4473
    if (!qemu_dev_name)
4474
        goto cleanup;
4475 4476
    len = strlen (qemu_dev_name);

4477
    if (qemudMonitorCommand (vm, "info blockstats", &info) < 0) {
4478 4479
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("'info blockstats' command failed"));
4480
        goto cleanup;
4481
    }
4482
    DEBUG ("%s: info blockstats reply: %s", vm->def->name, info);
4483 4484 4485 4486 4487 4488

    /* If the command isn't supported then qemu prints the supported
     * info commands, so the output starts "info ".  Since this is
     * unlikely to be the name of a block device, we can use this
     * to detect if qemu supports the command.
     */
C
Cole Robinson 已提交
4489
    if (strstr(info, "\ninfo ")) {
4490 4491 4492
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                          "%s",
                          _("'info blockstats' not supported by this qemu"));
4493
        goto cleanup;
4494 4495 4496 4497 4498 4499 4500 4501 4502 4503 4504 4505 4506 4507 4508 4509 4510 4511 4512 4513 4514 4515 4516 4517 4518 4519
    }

    stats->rd_req = -1;
    stats->rd_bytes = -1;
    stats->wr_req = -1;
    stats->wr_bytes = -1;
    stats->errs = -1;

    /* The output format for both qemu & KVM is:
     *   blockdevice: rd_bytes=% wr_bytes=% rd_operations=% wr_operations=%
     *   (repeated for each block device)
     * where '%' is a 64 bit number.
     */
    p = info;

    while (*p) {
        if (STREQLEN (p, qemu_dev_name, len)
            && p[len] == ':' && p[len+1] == ' ') {

            eol = strchr (p, '\n');
            if (!eol)
                eol = p + strlen (p);

            p += len+2;         /* Skip to first label. */

            while (*p) {
4520
                if (STRPREFIX (p, "rd_bytes=")) {
4521 4522
                    p += 9;
                    if (virStrToLong_ll (p, &dummy, 10, &stats->rd_bytes) == -1)
4523 4524
                        DEBUG ("%s: error reading rd_bytes: %s",
                               vm->def->name, p);
4525
                } else if (STRPREFIX (p, "wr_bytes=")) {
4526 4527
                    p += 9;
                    if (virStrToLong_ll (p, &dummy, 10, &stats->wr_bytes) == -1)
4528 4529
                        DEBUG ("%s: error reading wr_bytes: %s",
                               vm->def->name, p);
4530
                } else if (STRPREFIX (p, "rd_operations=")) {
4531 4532
                    p += 14;
                    if (virStrToLong_ll (p, &dummy, 10, &stats->rd_req) == -1)
4533 4534
                        DEBUG ("%s: error reading rd_req: %s",
                               vm->def->name, p);
4535
                } else if (STRPREFIX (p, "wr_operations=")) {
4536 4537
                    p += 14;
                    if (virStrToLong_ll (p, &dummy, 10, &stats->wr_req) == -1)
4538 4539
                        DEBUG ("%s: error reading wr_req: %s",
                               vm->def->name, p);
4540
                } else
4541
                    DEBUG ("%s: unknown block stat near %s", vm->def->name, p);
4542 4543 4544 4545 4546 4547

                /* Skip to next label. */
                p = strchr (p, ' ');
                if (!p || p >= eol) break;
                p++;
            }
4548
            ret = 0;
4549
            goto cleanup;
4550 4551 4552 4553 4554 4555 4556 4557 4558 4559 4560
        }

        /* Skip to next line. */
        p = strchr (p, '\n');
        if (!p) break;
        p++;
    }

    /* If we reach here then the device was not found. */
    qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                      _("device not found: %s (%s)"), path, qemu_dev_name);
4561
 cleanup:
4562 4563
    VIR_FREE(qemu_dev_name);
    VIR_FREE(info);
4564 4565
    if (vm)
        virDomainObjUnlock(vm);
4566
    return ret;
4567 4568
}

4569
#ifdef __linux__
4570 4571 4572 4573 4574
static int
qemudDomainInterfaceStats (virDomainPtr dom,
                           const char *path,
                           struct _virDomainInterfaceStats *stats)
{
4575 4576
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
4577
    int i;
4578
    int ret = -1;
4579

4580
    qemuDriverLock(driver);
4581
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4582 4583
    qemuDriverUnlock(driver);

4584
    if (!vm) {
4585 4586 4587 4588
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
4589
        goto cleanup;
4590 4591
    }

4592
    if (!virDomainIsActive(vm)) {
4593
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID,
4594
                         "%s", _("domain is not running"));
4595
        goto cleanup;
4596 4597 4598 4599
    }

    if (!path || path[0] == '\0') {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
4600
                         "%s", _("NULL or empty path"));
4601
        goto cleanup;
4602 4603 4604
    }

    /* Check the path is one of the domain's network interfaces. */
4605 4606
    for (i = 0 ; i < vm->def->nnets ; i++) {
        if (vm->def->nets[i]->ifname &&
4607 4608 4609 4610
            STREQ (vm->def->nets[i]->ifname, path)) {
            ret = 0;
            break;
        }
4611 4612
    }

4613 4614 4615 4616 4617
    if (ret == 0)
        ret = linuxDomainInterfaceStats (dom->conn, path, stats);
    else
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                          _("invalid path, '%s' is not a known interface"), path);
4618

4619
cleanup:
4620 4621
    if (vm)
        virDomainObjUnlock(vm);
4622 4623
    return ret;
}
4624
#else
4625 4626 4627 4628
static int
qemudDomainInterfaceStats (virDomainPtr dom,
                           const char *path ATTRIBUTE_UNUSED,
                           struct _virDomainInterfaceStats *stats ATTRIBUTE_UNUSED)
4629 4630 4631 4632
    qemudReportError (dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                      "%s", __FUNCTION__);
    return -1;
}
4633
#endif
4634

4635 4636 4637 4638 4639 4640 4641
static int
qemudDomainBlockPeek (virDomainPtr dom,
                      const char *path,
                      unsigned long long offset, size_t size,
                      void *buffer,
                      unsigned int flags ATTRIBUTE_UNUSED)
{
4642 4643 4644
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int fd = -1, ret = -1, i;
4645

4646
    qemuDriverLock(driver);
4647
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4648 4649
    qemuDriverUnlock(driver);

4650
    if (!vm) {
4651 4652 4653 4654
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
4655
        goto cleanup;
4656 4657 4658 4659
    }

    if (!path || path[0] == '\0') {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
J
Jim Meyering 已提交
4660
                         "%s", _("NULL or empty path"));
4661
        goto cleanup;
4662 4663 4664
    }

    /* Check the path belongs to this domain. */
4665 4666
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (vm->def->disks[i]->src != NULL &&
4667 4668 4669 4670
            STREQ (vm->def->disks[i]->src, path)) {
            ret = 0;
            break;
        }
4671 4672
    }

4673 4674 4675 4676 4677
    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) {
4678 4679
            virReportSystemError (dom->conn, errno,
                                  _("%s: failed to open"), path);
4680 4681
            goto cleanup;
        }
4682

4683 4684 4685 4686 4687 4688
        /* 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) {
4689 4690
            virReportSystemError (dom->conn, errno,
                                  _("%s: failed to seek or read"), path);
4691 4692 4693 4694 4695 4696 4697
            goto cleanup;
        }

        ret = 0;
    } else {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                          "%s", _("invalid path"));
4698 4699
    }

4700 4701 4702
cleanup:
    if (fd >= 0)
        close (fd);
4703 4704
    if (vm)
        virDomainObjUnlock(vm);
4705 4706 4707
    return ret;
}

R
Richard W.M. Jones 已提交
4708 4709 4710 4711 4712 4713
static int
qemudDomainMemoryPeek (virDomainPtr dom,
                       unsigned long long offset, size_t size,
                       void *buffer,
                       unsigned int flags)
{
4714 4715 4716
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char cmd[256], *info = NULL;
R
Richard W.M. Jones 已提交
4717 4718 4719
    char tmp[] = TEMPDIR "/qemu.mem.XXXXXX";
    int fd = -1, ret = -1;

4720
    qemuDriverLock(driver);
4721
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4722
    qemuDriverUnlock(driver);
R
Richard W.M. Jones 已提交
4723 4724

    if (!vm) {
4725 4726 4727 4728
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
4729 4730 4731 4732 4733 4734 4735
        goto cleanup;
    }

    if (flags != VIR_MEMORY_VIRTUAL) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                          "%s", _("QEMU driver only supports virtual memory addrs"));
        goto cleanup;
R
Richard W.M. Jones 已提交
4736 4737
    }

4738
    if (!virDomainIsActive(vm)) {
4739
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID,
R
Richard W.M. Jones 已提交
4740
                         "%s", _("domain is not running"));
4741
        goto cleanup;
R
Richard W.M. Jones 已提交
4742 4743 4744 4745
    }

    /* Create a temporary filename. */
    if ((fd = mkstemp (tmp)) == -1) {
4746 4747
        virReportSystemError (dom->conn, errno,
                              _("mkstemp(\"%s\") failed"), tmp);
4748
        goto cleanup;
R
Richard W.M. Jones 已提交
4749 4750 4751 4752
    }

    /* Issue the memsave command. */
    snprintf (cmd, sizeof cmd, "memsave %llu %zi \"%s\"", offset, size, tmp);
4753
    if (qemudMonitorCommand (vm, cmd, &info) < 0) {
R
Richard W.M. Jones 已提交
4754
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
4755
                          "%s", _("'memsave' command failed"));
4756
        goto cleanup;
R
Richard W.M. Jones 已提交
4757 4758
    }

4759
    DEBUG ("%s: memsave reply: %s", vm->def->name, info);
R
Richard W.M. Jones 已提交
4760 4761 4762

    /* Read the memory file into buffer. */
    if (saferead (fd, buffer, size) == (ssize_t) -1) {
4763 4764 4765
        virReportSystemError (dom->conn, errno,
                              _("failed to read temporary file "
                                "created with template %s"), tmp);
4766
        goto cleanup;
R
Richard W.M. Jones 已提交
4767 4768 4769
    }

    ret = 0;
4770 4771 4772

cleanup:
    VIR_FREE(info);
R
Richard W.M. Jones 已提交
4773 4774
    if (fd >= 0) close (fd);
    unlink (tmp);
4775 4776
    if (vm)
        virDomainObjUnlock(vm);
R
Richard W.M. Jones 已提交
4777 4778 4779
    return ret;
}

4780

4781 4782
static int
qemudDomainEventRegister (virConnectPtr conn,
4783
                          virConnectDomainEventCallback callback,
4784 4785
                          void *opaque,
                          virFreeCallback freecb)
4786
{
4787 4788 4789
    struct qemud_driver *driver = conn->privateData;
    int ret;

4790
    qemuDriverLock(driver);
4791 4792
    ret = virDomainEventCallbackListAdd(conn, driver->domainEventCallbacks,
                                        callback, opaque, freecb);
4793
    qemuDriverUnlock(driver);
4794

4795
    return ret;
4796 4797 4798 4799
}

static int
qemudDomainEventDeregister (virConnectPtr conn,
4800
                            virConnectDomainEventCallback callback)
4801
{
4802 4803 4804
    struct qemud_driver *driver = conn->privateData;
    int ret;

4805
    qemuDriverLock(driver);
4806 4807 4808 4809 4810 4811
    if (driver->domainEventDispatching)
        ret = virDomainEventCallbackListMarkDelete(conn, driver->domainEventCallbacks,
                                                   callback);
    else
        ret = virDomainEventCallbackListRemove(conn, driver->domainEventCallbacks,
                                               callback);
4812
    qemuDriverUnlock(driver);
4813

4814
    return ret;
4815 4816
}

4817 4818 4819 4820 4821
static void qemuDomainEventDispatchFunc(virConnectPtr conn,
                                        virDomainEventPtr event,
                                        virConnectDomainEventCallback cb,
                                        void *cbopaque,
                                        void *opaque)
4822
{
4823
    struct qemud_driver *driver = opaque;
4824

4825 4826 4827 4828 4829 4830 4831 4832 4833 4834 4835 4836 4837 4838 4839 4840 4841 4842 4843 4844 4845 4846 4847 4848 4849 4850 4851 4852 4853 4854 4855 4856 4857 4858 4859 4860 4861 4862 4863 4864 4865 4866 4867 4868
    /* Drop the lock whle dispatching, for sake of re-entrancy */
    qemuDriverUnlock(driver);
    virDomainEventDispatchDefaultFunc(conn, event, cb, cbopaque, NULL);
    qemuDriverLock(driver);
}

static void qemuDomainEventFlush(int timer ATTRIBUTE_UNUSED, void *opaque)
{
    struct qemud_driver *driver = opaque;
    virDomainEventQueue tempQueue;

    qemuDriverLock(driver);

    driver->domainEventDispatching = 1;

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

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

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

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


/* driver must be locked before calling */
static void qemuDomainEventQueue(struct qemud_driver *driver,
                                 virDomainEventPtr event)
{
    if (virDomainEventQueuePush(driver->domainEventQueue,
                                event) < 0)
        virDomainEventFree(event);
    if (qemu_driver->domainEventQueue->count == 1)
        virEventUpdateTimeout(driver->domainEventTimer, 0);
4869 4870
}

D
Daniel Veillard 已提交
4871 4872 4873 4874 4875 4876 4877 4878 4879 4880 4881 4882 4883 4884 4885 4886 4887 4888
/* Migration support. */

/* Prepare is the first step, and it runs on the destination host.
 *
 * This starts an empty VM listening on a TCP port.
 */
static int
qemudDomainMigratePrepare2 (virConnectPtr dconn,
                            char **cookie ATTRIBUTE_UNUSED,
                            int *cookielen ATTRIBUTE_UNUSED,
                            const char *uri_in,
                            char **uri_out,
                            unsigned long flags ATTRIBUTE_UNUSED,
                            const char *dname,
                            unsigned long resource ATTRIBUTE_UNUSED,
                            const char *dom_xml)
{
    static int port = 0;
4889 4890
    struct qemud_driver *driver = dconn->privateData;
    virDomainDefPtr def = NULL;
D
Daniel Veillard 已提交
4891 4892 4893 4894 4895
    virDomainObjPtr vm = NULL;
    int this_port;
    char hostname [HOST_NAME_MAX+1];
    char migrateFrom [64];
    const char *p;
4896
    virDomainEventPtr event = NULL;
4897 4898 4899
    int ret = -1;;

    *uri_out = NULL;
D
Daniel Veillard 已提交
4900

4901
    qemuDriverLock(driver);
D
Daniel Veillard 已提交
4902 4903 4904
    if (!dom_xml) {
        qemudReportError (dconn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                          "%s", _("no domain XML passed"));
4905
        goto cleanup;
D
Daniel Veillard 已提交
4906 4907 4908 4909 4910 4911 4912 4913 4914 4915 4916 4917 4918 4919 4920 4921 4922 4923
    }

    /* The URI passed in may be NULL or a string "tcp://somehostname:port".
     *
     * If the URI passed in is NULL then we allocate a port number
     * from our pool of port numbers and return a URI of
     * "tcp://ourhostname:port".
     *
     * If the URI passed in is not NULL then we try to parse out the
     * port number and use that (note that the hostname is assumed
     * to be a correct hostname which refers to the target machine).
     */
    if (uri_in == NULL) {
        this_port = QEMUD_MIGRATION_FIRST_PORT + port++;
        if (port == QEMUD_MIGRATION_NUM_PORTS) port = 0;

        /* Get hostname */
        if (gethostname (hostname, HOST_NAME_MAX+1) == -1) {
4924 4925
            virReportSystemError (dconn, errno,
                                  "%s", _("failed to determine host name"));
4926
            goto cleanup;
D
Daniel Veillard 已提交
4927 4928 4929
        }

        /* Caller frees */
4930
        if (virAsprintf(uri_out, "tcp:%s:%d", hostname, this_port) < 0) {
4931
            virReportOOMError (dconn);
4932
            goto cleanup;
D
Daniel Veillard 已提交
4933 4934 4935 4936 4937 4938
        }
    } else {
        /* Check the URI starts with "tcp:".  We will escape the
         * URI when passing it to the qemu monitor, so bad
         * characters in hostname part don't matter.
         */
4939
        if (!STRPREFIX (uri_in, "tcp:")) {
D
Daniel Veillard 已提交
4940 4941
            qemudReportError (dconn, NULL, NULL, VIR_ERR_INVALID_ARG,
                  "%s", _("only tcp URIs are supported for KVM migrations"));
4942
            goto cleanup;
D
Daniel Veillard 已提交
4943 4944 4945 4946 4947 4948 4949 4950 4951
        }

        /* Get the port number. */
        p = strrchr (uri_in, ':');
        p++; /* definitely has a ':' in it, see above */
        this_port = virParseNumber (&p);
        if (this_port == -1 || p-uri_in != strlen (uri_in)) {
            qemudReportError (dconn, NULL, NULL, VIR_ERR_INVALID_ARG,
                              "%s", _("URI did not have ':port' at the end"));
4952
            goto cleanup;
D
Daniel Veillard 已提交
4953 4954 4955 4956
        }
    }

    /* Parse the domain XML. */
4957 4958
    if (!(def = virDomainDefParseString(dconn, driver->caps, dom_xml,
                                        VIR_DOMAIN_XML_INACTIVE))) {
D
Daniel Veillard 已提交
4959 4960
        qemudReportError (dconn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("failed to parse XML"));
4961
        goto cleanup;
D
Daniel Veillard 已提交
4962 4963 4964 4965 4966 4967 4968 4969 4970 4971 4972 4973 4974 4975 4976 4977
    }

    /* Target domain name, maybe renamed. */
    dname = dname ? dname : def->name;

#if 1
    /* Ensure the name and UUID don't already exist in an active VM */
    vm = virDomainFindByUUID(&driver->domains, def->uuid);
#else
    /* For TESTING ONLY you can change #if 1 -> #if 0 above and use
     * this code which lets you do localhost migrations.  You must still
     * supply a fresh 'dname' but this code assigns a random UUID.
     */
    if (virUUIDGenerate (def->uuid) == -1) {
        qemudReportError (dconn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
            _("could not generate random UUID"));
4978
        goto cleanup;
D
Daniel Veillard 已提交
4979 4980 4981 4982 4983 4984 4985 4986 4987
    }
#endif

    if (!vm) vm = virDomainFindByName(&driver->domains, dname);
    if (vm) {
        if (virDomainIsActive(vm)) {
            qemudReportError (dconn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                              _("domain with the same name or UUID already exists as '%s'"),
                              vm->def->name);
4988
            goto cleanup;
D
Daniel Veillard 已提交
4989
        }
4990
        virDomainObjUnlock(vm);
D
Daniel Veillard 已提交
4991 4992 4993 4994 4995 4996 4997
    }

    if (!(vm = virDomainAssignDef(dconn,
                                  &driver->domains,
                                  def))) {
        qemudReportError (dconn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("failed to assign new VM"));
4998
        goto cleanup;
D
Daniel Veillard 已提交
4999
    }
5000
    def = NULL;
D
Daniel Veillard 已提交
5001 5002 5003 5004 5005 5006 5007 5008

    /* Domain starts inactive, even if the domain XML had an id field. */
    vm->def->id = -1;

    /* Start the QEMU daemon, with the same command-line arguments plus
     * -incoming tcp:0.0.0.0:port
     */
    snprintf (migrateFrom, sizeof (migrateFrom), "tcp:0.0.0.0:%d", this_port);
5009
    if (qemudStartVMDaemon (dconn, driver, vm, migrateFrom, -1) < 0) {
5010
        if (!vm->persistent) {
D
Daniel Veillard 已提交
5011
            virDomainRemoveInactive(&driver->domains, vm);
5012 5013
            vm = NULL;
        }
5014
        goto cleanup;
D
Daniel Veillard 已提交
5015
    }
5016 5017 5018 5019

    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_MIGRATED);
5020
    ret = 0;
D
Daniel Veillard 已提交
5021

5022 5023 5024 5025 5026
cleanup:
    virDomainDefFree(def);
    if (ret != 0) {
        VIR_FREE(*uri_out);
    }
5027 5028
    if (vm)
        virDomainObjUnlock(vm);
5029 5030
    if (event)
        qemuDomainEventQueue(driver, event);
5031
    qemuDriverUnlock(driver);
5032
    return ret;
D
Daniel Veillard 已提交
5033 5034 5035 5036 5037 5038 5039 5040 5041 5042 5043 5044
}

/* Perform is the second step, and it runs on the source host. */
static int
qemudDomainMigratePerform (virDomainPtr dom,
                           const char *cookie ATTRIBUTE_UNUSED,
                           int cookielen ATTRIBUTE_UNUSED,
                           const char *uri,
                           unsigned long flags ATTRIBUTE_UNUSED,
                           const char *dname ATTRIBUTE_UNUSED,
                           unsigned long resource)
{
5045 5046
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
5047
    virDomainEventPtr event = NULL;
D
Daniel Veillard 已提交
5048 5049
    char *safe_uri;
    char cmd[HOST_NAME_MAX+50];
5050 5051
    char *info = NULL;
    int ret = -1;
5052
    int paused = 0;
D
Daniel Veillard 已提交
5053

5054
    qemuDriverLock(driver);
5055
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
D
Daniel Veillard 已提交
5056
    if (!vm) {
5057 5058 5059 5060
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_DOMAIN,
                         _("no domain with matching uuid '%s'"), uuidstr);
5061
        goto cleanup;
D
Daniel Veillard 已提交
5062 5063 5064
    }

    if (!virDomainIsActive(vm)) {
5065
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_INVALID,
D
Daniel Veillard 已提交
5066
                          "%s", _("domain is not running"));
5067
        goto cleanup;
D
Daniel Veillard 已提交
5068 5069
    }

5070 5071
    if (!(flags & VIR_MIGRATE_LIVE)) {
        /* Pause domain for non-live migration */
5072 5073 5074 5075 5076
        if (qemudMonitorCommand (vm, "stop", &info) < 0) {
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                             "%s", _("off-line migration specified, but suspend operation failed"));
            goto cleanup;
        }
5077
        DEBUG ("%s: stop reply: %s", vm->def->name, info);
5078
        VIR_FREE(info);
5079
        paused = 1;
5080

5081 5082 5083 5084 5085 5086
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_SUSPENDED,
                                         VIR_DOMAIN_EVENT_SUSPENDED_MIGRATED);
        if (event)
            qemuDomainEventQueue(driver, event);
        event = NULL;
5087 5088
    }

D
Daniel Veillard 已提交
5089 5090 5091
    if (resource > 0) {
        /* Issue migrate_set_speed command.  Don't worry if it fails. */
        snprintf (cmd, sizeof cmd, "migrate_set_speed %lum", resource);
5092
        qemudMonitorCommand (vm, cmd, &info);
D
Daniel Veillard 已提交
5093

5094
        DEBUG ("%s: migrate_set_speed reply: %s", vm->def->name, info);
D
Daniel Veillard 已提交
5095 5096 5097 5098 5099 5100
        VIR_FREE (info);
    }

    /* Issue the migrate command. */
    safe_uri = qemudEscapeMonitorArg (uri);
    if (!safe_uri) {
5101
        virReportOOMError (dom->conn);
5102
        goto cleanup;
D
Daniel Veillard 已提交
5103 5104 5105 5106
    }
    snprintf (cmd, sizeof cmd, "migrate \"%s\"", safe_uri);
    VIR_FREE (safe_uri);

5107
    if (qemudMonitorCommand (vm, cmd, &info) < 0) {
D
Daniel Veillard 已提交
5108 5109
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("migrate operation failed"));
5110
        goto cleanup;
D
Daniel Veillard 已提交
5111 5112
    }

5113
    DEBUG ("%s: migrate reply: %s", vm->def->name, info);
D
Daniel Veillard 已提交
5114 5115 5116 5117 5118

    /* Now check for "fail" in the output string */
    if (strstr(info, "fail") != NULL) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                          _("migrate failed: %s"), info);
5119
        goto cleanup;
D
Daniel Veillard 已提交
5120 5121 5122 5123
    }

    /* Clean up the source domain. */
    qemudShutdownVMDaemon (dom->conn, driver, vm);
5124
    paused = 0;
5125 5126 5127 5128

    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_MIGRATED);
5129
    if (!vm->persistent) {
D
Daniel Veillard 已提交
5130
        virDomainRemoveInactive(&driver->domains, vm);
5131 5132
        vm = NULL;
    }
5133
    ret = 0;
D
Daniel Veillard 已提交
5134

5135
cleanup:
5136 5137 5138
    /* Note that we have to free info *first*, since we are re-using the
     * variable below (and otherwise might cause a memory leak)
     */
5139
    VIR_FREE(info);
5140 5141 5142 5143 5144 5145 5146 5147

    if (paused) {
        /* we got here through some sort of failure; start the domain again */
        if (qemudMonitorCommand (vm, "cont", &info) < 0) {
            /* Hm, we already know we are in error here.  We don't want to
             * overwrite the previous error, though, so we just throw something
             * to the logs and hope for the best
             */
5148 5149
            VIR_ERROR(_("Failed to resume guest %s after failure\n"),
                      vm->def->name);
5150 5151
        }
        else {
5152
            DEBUG ("%s: cont reply: %s", vm->def->name, info);
5153 5154 5155 5156 5157 5158 5159 5160
            VIR_FREE(info);
        }

        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_RESUMED,
                                         VIR_DOMAIN_EVENT_RESUMED_MIGRATED);
    }

5161 5162
    if (vm)
        virDomainObjUnlock(vm);
5163 5164
    if (event)
        qemuDomainEventQueue(driver, event);
5165
    qemuDriverUnlock(driver);
5166
    return ret;
D
Daniel Veillard 已提交
5167 5168 5169 5170 5171 5172 5173 5174 5175 5176 5177 5178
}

/* Finish is the third and final step, and it runs on the destination host. */
static virDomainPtr
qemudDomainMigrateFinish2 (virConnectPtr dconn,
                           const char *dname,
                           const char *cookie ATTRIBUTE_UNUSED,
                           int cookielen ATTRIBUTE_UNUSED,
                           const char *uri ATTRIBUTE_UNUSED,
                           unsigned long flags ATTRIBUTE_UNUSED,
                           int retcode)
{
5179 5180 5181
    struct qemud_driver *driver = dconn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
5182
    virDomainEventPtr event = NULL;
D
Daniel Veillard 已提交
5183 5184
    char *info = NULL;

5185
    qemuDriverLock(driver);
5186
    vm = virDomainFindByName(&driver->domains, dname);
D
Daniel Veillard 已提交
5187
    if (!vm) {
5188 5189
        qemudReportError (dconn, NULL, NULL, VIR_ERR_NO_DOMAIN,
                          _("no domain with matching name '%s'"), dname);
5190
        goto cleanup;
D
Daniel Veillard 已提交
5191 5192 5193 5194 5195 5196 5197 5198 5199
    }

    /* Did the migration go as planned?  If yes, return the domain
     * object, but if no, clean up the empty qemu process.
     */
    if (retcode == 0) {
        dom = virGetDomain (dconn, vm->def->name, vm->def->uuid);
        VIR_FREE(info);
        vm->state = VIR_DOMAIN_RUNNING;
5200 5201 5202
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_RESUMED,
                                         VIR_DOMAIN_EVENT_RESUMED_MIGRATED);
D
Daniel Veillard 已提交
5203 5204
    } else {
        qemudShutdownVMDaemon (dconn, driver, vm);
5205 5206 5207
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_FAILED);
5208
        if (!vm->persistent) {
D
Daniel Veillard 已提交
5209
            virDomainRemoveInactive(&driver->domains, vm);
5210 5211
            vm = NULL;
        }
D
Daniel Veillard 已提交
5212
    }
5213 5214

cleanup:
5215 5216
    if (vm)
        virDomainObjUnlock(vm);
5217 5218
    if (event)
        qemuDomainEventQueue(driver, event);
5219
    qemuDriverUnlock(driver);
5220
    return dom;
D
Daniel Veillard 已提交
5221 5222
}

5223 5224 5225 5226 5227 5228 5229 5230 5231 5232 5233 5234 5235 5236 5237 5238 5239 5240 5241 5242 5243 5244 5245 5246 5247 5248 5249 5250 5251 5252 5253 5254 5255 5256 5257 5258 5259 5260 5261 5262 5263 5264 5265 5266 5267 5268 5269 5270 5271 5272 5273 5274 5275 5276 5277 5278 5279 5280 5281 5282 5283 5284 5285 5286 5287 5288 5289 5290 5291 5292 5293 5294 5295 5296 5297 5298 5299 5300 5301 5302 5303 5304 5305 5306 5307 5308 5309 5310 5311 5312 5313 5314 5315 5316 5317 5318 5319 5320 5321 5322 5323 5324 5325 5326 5327 5328 5329 5330 5331 5332 5333 5334 5335 5336 5337
static int
qemudNodeDeviceGetPciInfo (virNodeDevicePtr dev,
                           unsigned *domain,
                           unsigned *bus,
                           unsigned *slot,
                           unsigned *function)
{
    virNodeDeviceDefPtr def = NULL;
    virNodeDevCapsDefPtr cap;
    char *xml = NULL;
    int ret = -1;

    xml = virNodeDeviceGetXMLDesc(dev, 0);
    if (!xml)
        goto out;

    def = virNodeDeviceDefParseString(dev->conn, xml);
    if (!def)
        goto out;

    cap = def->caps;
    while (cap) {
        if (cap->type == VIR_NODE_DEV_CAP_PCI_DEV) {
            *domain   = cap->data.pci_dev.domain;
            *bus      = cap->data.pci_dev.bus;
            *slot     = cap->data.pci_dev.slot;
            *function = cap->data.pci_dev.function;
            break;
        }

        cap = cap->next;
    }

    if (!cap) {
        qemudReportError(dev->conn, NULL, NULL, VIR_ERR_INVALID_ARG,
                         _("device %s is not a PCI device"), dev->name);
        goto out;
    }

    ret = 0;
out:
    virNodeDeviceDefFree(def);
    VIR_FREE(xml);
    return ret;
}

static int
qemudNodeDeviceDettach (virNodeDevicePtr dev)
{
    pciDevice *pci;
    unsigned domain, bus, slot, function;
    int ret = -1;

    if (qemudNodeDeviceGetPciInfo(dev, &domain, &bus, &slot, &function) < 0)
        return -1;

    pci = pciGetDevice(dev->conn, domain, bus, slot, function);
    if (!pci)
        return -1;

    if (pciDettachDevice(dev->conn, pci) < 0)
        goto out;

    ret = 0;
out:
    pciFreeDevice(dev->conn, pci);
    return ret;
}

static int
qemudNodeDeviceReAttach (virNodeDevicePtr dev)
{
    pciDevice *pci;
    unsigned domain, bus, slot, function;
    int ret = -1;

    if (qemudNodeDeviceGetPciInfo(dev, &domain, &bus, &slot, &function) < 0)
        return -1;

    pci = pciGetDevice(dev->conn, domain, bus, slot, function);
    if (!pci)
        return -1;

    if (pciReAttachDevice(dev->conn, pci) < 0)
        goto out;

    ret = 0;
out:
    pciFreeDevice(dev->conn, pci);
    return ret;
}

static int
qemudNodeDeviceReset (virNodeDevicePtr dev)
{
    pciDevice *pci;
    unsigned domain, bus, slot, function;
    int ret = -1;

    if (qemudNodeDeviceGetPciInfo(dev, &domain, &bus, &slot, &function) < 0)
        return -1;

    pci = pciGetDevice(dev->conn, domain, bus, slot, function);
    if (!pci)
        return -1;

    if (pciResetDevice(dev->conn, pci) < 0)
        goto out;

    ret = 0;
out:
    pciFreeDevice(dev->conn, pci);
    return ret;
}

5338 5339 5340 5341 5342
static virDriver qemuDriver = {
    VIR_DRV_QEMU,
    "QEMU",
    qemudOpen, /* open */
    qemudClose, /* close */
D
Daniel Veillard 已提交
5343
    qemudSupportsFeature, /* supports_feature */
5344 5345
    qemudGetType, /* type */
    qemudGetVersion, /* version */
5346
    qemudGetHostname, /* getHostname */
5347 5348 5349 5350 5351
    qemudGetMaxVCPUs, /* getMaxVcpus */
    qemudGetNodeInfo, /* nodeGetInfo */
    qemudGetCapabilities, /* getCapabilities */
    qemudListDomains, /* listDomains */
    qemudNumDomains, /* numOfDomains */
5352
    qemudDomainCreate, /* domainCreateXML */
5353 5354 5355 5356 5357
    qemudDomainLookupByID, /* domainLookupByID */
    qemudDomainLookupByUUID, /* domainLookupByUUID */
    qemudDomainLookupByName, /* domainLookupByName */
    qemudDomainSuspend, /* domainSuspend */
    qemudDomainResume, /* domainResume */
5358
    qemudDomainShutdown, /* domainShutdown */
5359 5360 5361
    NULL, /* domainReboot */
    qemudDomainDestroy, /* domainDestroy */
    qemudDomainGetOSType, /* domainGetOSType */
5362 5363 5364
    qemudDomainGetMaxMemory, /* domainGetMaxMemory */
    qemudDomainSetMaxMemory, /* domainSetMaxMemory */
    qemudDomainSetMemory, /* domainSetMemory */
5365 5366 5367 5368
    qemudDomainGetInfo, /* domainGetInfo */
    qemudDomainSave, /* domainSave */
    qemudDomainRestore, /* domainRestore */
    NULL, /* domainCoreDump */
5369
    qemudDomainSetVcpus, /* domainSetVcpus */
5370 5371 5372 5373
#if HAVE_SCHED_GETAFFINITY
    qemudDomainPinVcpu, /* domainPinVcpu */
    qemudDomainGetVcpus, /* domainGetVcpus */
#else
5374 5375
    NULL, /* domainPinVcpu */
    NULL, /* domainGetVcpus */
5376
#endif
5377
    qemudDomainGetMaxVcpus, /* domainGetMaxVcpus */
5378 5379
    qemudDomainGetSecurityLabel, /* domainGetSecurityLabel */
    qemudNodeGetSecurityModel, /* nodeGetSecurityModel */
5380
    qemudDomainDumpXML, /* domainDumpXML */
5381
    qemuDomainXMLFromNative, /* domainXmlFromNative */
5382
    qemuDomainXMLToNative, /* domainXMLToNative */
5383 5384
    qemudListDefinedDomains, /* listDefinedDomains */
    qemudNumDefinedDomains, /* numOfDefinedDomains */
5385 5386 5387
    qemudDomainStart, /* domainCreate */
    qemudDomainDefine, /* domainDefineXML */
    qemudDomainUndefine, /* domainUndefine */
5388
    qemudDomainAttachDevice, /* domainAttachDevice */
5389
    qemudDomainDetachDevice, /* domainDetachDevice */
5390 5391 5392 5393 5394
    qemudDomainGetAutostart, /* domainGetAutostart */
    qemudDomainSetAutostart, /* domainSetAutostart */
    NULL, /* domainGetSchedulerType */
    NULL, /* domainGetSchedulerParameters */
    NULL, /* domainSetSchedulerParameters */
D
Daniel Veillard 已提交
5395 5396
    NULL, /* domainMigratePrepare (v1) */
    qemudDomainMigratePerform, /* domainMigratePerform */
5397
    NULL, /* domainMigrateFinish */
5398
    qemudDomainBlockStats, /* domainBlockStats */
5399
    qemudDomainInterfaceStats, /* domainInterfaceStats */
5400
    qemudDomainBlockPeek, /* domainBlockPeek */
R
Richard W.M. Jones 已提交
5401
    qemudDomainMemoryPeek, /* domainMemoryPeek */
5402 5403 5404 5405
#if HAVE_NUMACTL
    qemudNodeGetCellsFreeMemory, /* nodeGetCellsFreeMemory */
    qemudNodeGetFreeMemory,  /* getFreeMemory */
#else
5406
    NULL, /* nodeGetCellsFreeMemory */
5407
    NULL, /* getFreeMemory */
5408
#endif
5409 5410
    qemudDomainEventRegister, /* domainEventRegister */
    qemudDomainEventDeregister, /* domainEventDeregister */
D
Daniel Veillard 已提交
5411 5412
    qemudDomainMigratePrepare2, /* domainMigratePrepare2 */
    qemudDomainMigrateFinish2, /* domainMigrateFinish2 */
5413 5414 5415
    qemudNodeDeviceDettach, /* nodeDeviceDettach */
    qemudNodeDeviceReAttach, /* nodeDeviceReAttach */
    qemudNodeDeviceReset, /* nodeDeviceReset */
5416 5417 5418
};


5419
static virStateDriver qemuStateDriver = {
5420 5421 5422 5423
    .initialize = qemudStartup,
    .cleanup = qemudShutdown,
    .reload = qemudReload,
    .active = qemudActive,
5424
};
5425

5426
int qemuRegister(void) {
5427 5428 5429 5430
    virRegisterDriver(&qemuDriver);
    virRegisterStateDriver(&qemuStateDriver);
    return 0;
}