qemu_driver.c 134.1 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 73
#define VIR_FROM_THIS VIR_FROM_QEMU

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

77 78 79 80
#define QEMU_CMD_PROMPT "\n(qemu) "
#define QEMU_PASSWD_PROMPT "Password: "


81 82
static int qemudShutdown(void);

83 84
#define qemudLog(level, msg...) fprintf(stderr, msg)

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

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

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

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

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

113
static int qemudDomainGetMaxVcpus(virDomainPtr dom);
114

115 116 117 118 119 120 121 122
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);
123

J
Jim Meyering 已提交
124
static struct qemud_driver *qemu_driver = NULL;
125 126


127 128 129 130 131 132
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 已提交
133
    int ret, fd = -1;
134

G
Guido Günther 已提交
135 136
    if ((ret = snprintf(logfile, sizeof(logfile), "%s/%s.log", logDir, name))
        < 0 || ret >= sizeof(logfile)) {
137
        virReportOOMError(conn);
138 139 140 141 142 143 144 145 146
        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) {
147 148 149
        virReportSystemError(conn, errno,
                             _("failed to create logfile %s"),
                             logfile);
150 151
        return -1;
    }
152
    if (virSetCloseExec(fd) < 0) {
153 154
        virReportSystemError(conn, errno, "%s",
                             _("Unable to set VM logfile close-on-exec flag"));
155 156 157 158 159 160 161
        close(fd);
        return -1;
    }
    return fd;
}


162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178
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) {
179 180 181
        virReportSystemError(conn, errno,
                             _("failed to create logfile %s"),
                             logfile);
182 183
        return -1;
    }
184
    if (virSetCloseExec(fd) < 0) {
185 186
        virReportSystemError(conn, errno, "%s",
                             _("Unable to set VM logfile close-on-exec flag"));
187 188 189 190
        close(fd);
        return -1;
    }
    if (lseek(fd, pos, SEEK_SET) < 0) {
191 192 193
        virReportSystemError(conn, errno,
                             _("Unable to seek to %lld in %s"),
                             (long long) pos, logfile);
194 195 196 197 198 199
        close(fd);
    }
    return fd;
}


200 201 202
static void
qemudAutostartConfigs(struct qemud_driver *driver) {
    unsigned int i;
203 204 205 206 207 208 209 210 211
    /* 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 */
212 213

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

    virConnectClose(conn);
237 238
}

239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255

/**
 * 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) {
256
        virReportOOMError(conn);
257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275
        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;
}


276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306
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) {
307
            qemudLog(QEMUD_ERROR, _("Failed to read domain status for %s\n"),
308 309 310 311 312 313 314 315 316
                     vm->def->name);
            goto next_error;
        }

        status = qemudDomainStatusParseFile(NULL, driver->caps, config, 0);
        if (status) {
            vm->newDef = vm->def;
            vm->def = status->def;
        } else {
317
            qemudLog(QEMUD_ERROR, _("Failed to parse domain status for %s\n"),
318 319 320 321 322
                     vm->def->name);
            goto next_error;
        }

        if ((rc = qemudOpenMonitor(NULL, driver, vm, status->monitorpath, 1)) != 0) {
323
            qemudLog(QEMUD_ERROR, _("Failed to reconnect monitor for %s: %d\n"),
324 325
                     vm->def->name, rc);
            goto next_error;
326
        }
327

328
        if ((vm->logfile = qemudLogFD(NULL, driver->logDir, vm->def->name)) < 0)
329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345
            return -1;

        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);
        virDomainDefFree(vm->def);
        vm->def = vm->newDef;
        vm->newDef = NULL;
next:
        virDomainObjUnlock(vm);
346 347
        if (status)
            VIR_FREE(status->monitorpath);
348 349 350 351 352 353 354
        VIR_FREE(status);
        VIR_FREE(config);
    }
    return 0;
}


355 356 357 358 359 360 361
/**
 * qemudStartup:
 *
 * Initialization function for the QEmu daemon
 */
static int
qemudStartup(void) {
362 363
    uid_t uid = geteuid();
    char *base = NULL;
D
Daniel P. Berrange 已提交
364
    char driverConf[PATH_MAX];
365

366
    if (VIR_ALLOC(qemu_driver) < 0)
367 368
        return -1;

369 370 371 372 373
    if (virMutexInit(&qemu_driver->lock) < 0) {
        qemudLog(QEMUD_ERROR, "%s", _("cannot initialize mutex"));
        VIR_FREE(qemu_driver);
        return -1;
    }
374 375
    qemuDriverLock(qemu_driver);

376 377 378
    /* Don't have a dom0 so start from 1 */
    qemu_driver->nextvmid = 1;

379 380
    /* Init callback list */
    if(VIR_ALLOC(qemu_driver->domainEventCallbacks) < 0)
381
        goto out_of_memory;
382 383 384 385 386 387
    if (!(qemu_driver->domainEventQueue = virDomainEventQueueNew()))
        goto out_of_memory;

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

389
    if (!uid) {
390 391
        if (virAsprintf(&qemu_driver->logDir,
                        "%s/log/libvirt/qemu", LOCAL_STATE_DIR) == -1)
392
            goto out_of_memory;
393

D
Daniel P. Berrange 已提交
394
        if ((base = strdup (SYSCONF_DIR "/libvirt")) == NULL)
395
            goto out_of_memory;
396 397 398 399

        if (virAsprintf(&qemu_driver->stateDir,
                      "%s/run/libvirt/qemu/", LOCAL_STATE_DIR) == -1)
            goto out_of_memory;
400
    } else {
401 402
        char *userdir = virGetUserDirectory(NULL, uid);
        if (!userdir)
403
            goto error;
404

405
        if (virAsprintf(&qemu_driver->logDir,
406 407
                        "%s/.libvirt/qemu/log", userdir) == -1) {
            VIR_FREE(userdir);
408
            goto out_of_memory;
409
        }
410

411 412
        if (virAsprintf(&base, "%s/.libvirt", userdir) == -1) {
            VIR_FREE(userdir);
413
            goto out_of_memory;
414 415
        }
        VIR_FREE(userdir);
416 417 418 419 420 421

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

    if (virFileMakePath(qemu_driver->stateDir) < 0) {
422
        char ebuf[1024];
423
        qemudLog(QEMUD_ERROR, _("Failed to create state dir '%s': %s\n"),
424
                 qemu_driver->stateDir, virStrerror(errno, ebuf, sizeof ebuf));
425
        goto error;
426 427 428 429 430
    }

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

435
    if (virAsprintf(&qemu_driver->configDir, "%s/qemu", base) == -1)
436 437
        goto out_of_memory;

438
    if (virAsprintf(&qemu_driver->autostartDir, "%s/qemu/autostart", base) == -1)
439 440
        goto out_of_memory;

441
    VIR_FREE(base);
442 443 444

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

    if (qemudLoadDriverConfig(qemu_driver, driverConf) < 0) {
447
        goto error;
D
Daniel P. Berrange 已提交
448 449
    }

450 451 452 453
    if (virDomainLoadAllConfigs(NULL,
                                qemu_driver->caps,
                                &qemu_driver->domains,
                                qemu_driver->configDir,
454
                                qemu_driver->autostartDir,
455 456
                                NULL, NULL) < 0)
        goto error;
457
    qemudReconnectVMs(qemu_driver);
458 459
    qemudAutostartConfigs(qemu_driver);

460 461
    qemuDriverUnlock(qemu_driver);

462 463
    return 0;

464
out_of_memory:
465
    qemudLog (QEMUD_ERROR,
466
              "%s", _("qemudStartup: out of memory\n"));
467 468 469
error:
    if (qemu_driver)
        qemuDriverUnlock(qemu_driver);
470
    VIR_FREE(base);
471
    qemudShutdown();
472 473 474
    return -1;
}

475 476 477 478
static void qemudNotifyLoadDomain(virDomainObjPtr vm, int newVM, void *opaque)
{
    struct qemud_driver *driver = opaque;

479 480 481 482 483 484 485 486
    if (newVM) {
        virDomainEventPtr event =
            virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_DEFINED,
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED);
        if (event)
            qemuDomainEventQueue(driver, event);
    }
487 488
}

489 490 491 492 493 494 495 496
/**
 * qemudReload:
 *
 * Function to restart the QEmu daemon, it will recheck the configuration
 * files and update its state and the networking
 */
static int
qemudReload(void) {
497 498 499
    if (!qemu_driver)
        return 0;

500
    qemuDriverLock(qemu_driver);
501 502 503 504
    virDomainLoadAllConfigs(NULL,
                            qemu_driver->caps,
                            &qemu_driver->domains,
                            qemu_driver->configDir,
505 506
                            qemu_driver->autostartDir,
                            qemudNotifyLoadDomain, qemu_driver);
507

508
    qemudAutostartConfigs(qemu_driver);
509
    qemuDriverUnlock(qemu_driver);
510 511

    return 0;
512 513
}

514 515 516 517 518 519 520 521 522 523
/**
 * 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) {
524
    unsigned int i;
525
    int active = 0;
526

527 528 529
    if (!qemu_driver)
        return 0;

530 531 532 533 534 535 536 537
    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);
    }
538

539 540
    qemuDriverUnlock(qemu_driver);
    return active;
541 542
}

543 544 545 546 547 548 549
/**
 * qemudShutdown:
 *
 * Shutdown the QEmu daemon, it will stop all active domains and networks
 */
static int
qemudShutdown(void) {
550

551
    if (!qemu_driver)
552
        return -1;
553

554
    qemuDriverLock(qemu_driver);
555 556
    virCapabilitiesFree(qemu_driver->caps);

557
    virDomainObjListFree(&qemu_driver->domains);
558

559
    VIR_FREE(qemu_driver->logDir);
560 561
    VIR_FREE(qemu_driver->configDir);
    VIR_FREE(qemu_driver->autostartDir);
562
    VIR_FREE(qemu_driver->stateDir);
563
    VIR_FREE(qemu_driver->vncTLSx509certdir);
J
Jim Meyering 已提交
564
    VIR_FREE(qemu_driver->vncListen);
565
    VIR_FREE(qemu_driver->vncPassword);
D
Daniel P. Berrange 已提交
566

567 568
    /* Free domain callback list */
    virDomainEventCallbackListFree(qemu_driver->domainEventCallbacks);
569 570 571 572
    virDomainEventQueueFree(qemu_driver->domainEventQueue);

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

574 575 576
    if (qemu_driver->brctl)
        brShutdown(qemu_driver->brctl);

577
    qemuDriverUnlock(qemu_driver);
578
    virMutexDestroy(&qemu_driver->lock);
579
    VIR_FREE(qemu_driver);
580 581

    return 0;
582 583 584
}

/* Return -1 for error, 1 to continue reading and 0 for success */
585
typedef int qemudHandlerMonitorOutput(virConnectPtr conn,
586
                                      virDomainObjPtr vm,
587 588 589
                                      const char *output,
                                      int fd);

590 591 592
/*
 * Returns -1 for error, 0 on end-of-file, 1 for success
 */
593
static int
594
qemudReadMonitorOutput(virConnectPtr conn,
595
                       virDomainObjPtr vm,
596 597 598 599
                       int fd,
                       char *buf,
                       int buflen,
                       qemudHandlerMonitorOutput func,
600 601
                       const char *what,
                       int timeout)
602 603 604 605
{
    int got = 0;
    buf[0] = '\0';

606
    /* Consume & discard the initial greeting */
607 608 609 610
    while (got < (buflen-1)) {
        int ret;

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

612 613 614 615 616 617
        if (ret < 0) {
            struct pollfd pfd = { .fd = fd, .events = POLLIN };
            if (errno == EINTR)
                continue;

            if (errno != EAGAIN) {
618 619 620
                virReportSystemError(conn, errno,
                                     _("Failure while reading %s startup output"),
                                     what);
621 622 623
                return -1;
            }

624
            ret = poll(&pfd, 1, timeout);
625
            if (ret == 0) {
626
                qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
627
                                 _("Timed out while reading %s startup output"), what);
628 629 630
                return -1;
            } else if (ret == -1) {
                if (errno != EINTR) {
631 632 633
                    virReportSystemError(conn, errno,
                                         _("Failure while reading %s startup output"),
                                         what);
634 635 636 637 638 639 640 641
                    return -1;
                }
            } else {
                /* Make sure we continue loop & read any further data
                   available before dealing with EOF */
                if (pfd.revents & (POLLIN | POLLHUP))
                    continue;

642
                qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
643
                                 _("Failure while reading %s startup output"), what);
644 645
                return -1;
            }
646 647
        } else if (ret == 0) {
            return 0;
648 649 650
        } else {
            got += ret;
            buf[got] = '\0';
651 652 653 654 655 656
            ret = func(conn, vm, buf, fd);
            if (ret == -1)
                return -1;
            if (ret == 1)
                continue;
            return 1;
657 658 659
        }
    }

660
    qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
661
                     _("Out of space while reading %s startup output"), what);
662 663 664 665 666
    return -1;

}

static int
667
qemudCheckMonitorPrompt(virConnectPtr conn ATTRIBUTE_UNUSED,
668
                        virDomainObjPtr vm,
669 670 671 672 673 674 675 676 677 678 679
                        const char *output,
                        int fd)
{
    if (strstr(output, "(qemu) ") == NULL)
        return 1; /* keep reading */

    vm->monitor = fd;

    return 0;
}

680
static int qemudOpenMonitor(virConnectPtr conn,
681
                            struct qemud_driver* driver,
682
                            virDomainObjPtr vm,
683 684
                            const char *monitor,
                            int reconnect) {
685 686 687 688
    int monfd;
    char buf[1024];
    int ret = -1;

D
Daniel Veillard 已提交
689
    if ((monfd = open(monitor, O_RDWR)) < 0) {
690
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
691
                         _("Unable to open monitor path %s"), monitor);
692 693
        return -1;
    }
694
    if (virSetCloseExec(monfd) < 0) {
695
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
696
                         "%s", _("Unable to set monitor close-on-exec flag"));
697 698
        goto error;
    }
699
    if (virSetNonBlock(monfd) < 0) {
700
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
701
                         "%s", _("Unable to put monitor into non-blocking mode"));
702 703 704
        goto error;
    }

705
    if (!reconnect) {
706 707 708 709 710 711 712 713
        if (qemudReadMonitorOutput(conn,
                                   vm, monfd,
                                   buf, sizeof(buf),
                                   qemudCheckMonitorPrompt,
                                   "monitor", 10000) <= 0)
            ret = -1;
        else
            ret = 0;
714 715 716 717 718 719 720
    } else {
        vm->monitor = monfd;
        ret = 0;
    }

    if (ret != 0)
         goto error;
721

722
    if (!(vm->monitorpath = strdup(monitor))) {
723
        virReportOOMError(conn);
724 725 726
        goto error;
    }

727 728 729 730 731 732
    if ((vm->monitor_watch = virEventAddHandle(vm->monitor, 0,
                                               qemudDispatchVMEvent,
                                               driver, NULL)) < 0)
        goto error;


733 734 735 736
    /* Keep monitor open upon success */
    if (ret == 0)
        return ret;

737 738 739 740 741
 error:
    close(monfd);
    return ret;
}

742 743
static int qemudExtractMonitorPath(virConnectPtr conn,
                                   const char *haystack,
744
                                   size_t *offset,
745
                                   char **path) {
746
    static const char needle[] = "char device redirected to";
747
    char *tmp, *dev;
748

749
    VIR_FREE(*path);
750
    /* First look for our magic string */
751 752 753 754 755
    if (!(tmp = strstr(haystack + *offset, needle))) {
        return 1;
    }
    tmp += sizeof(needle);
    dev = tmp;
756

757 758 759 760 761
    /*
     * And look for first whitespace character and nul terminate
     * to mark end of the pty path
     */
    while (*tmp) {
762
        if (c_isspace(*tmp)) {
763
            if (VIR_ALLOC_N(*path, (tmp-dev)+1) < 0) {
764
                virReportOOMError(conn);
765 766 767 768
                return -1;
            }
            strncpy(*path, dev, (tmp-dev));
            (*path)[(tmp-dev)] = '\0';
769
            /* ... now further update offset till we get EOL */
770
            *offset = tmp - haystack;
771 772
            return 0;
        }
773
        tmp++;
774 775 776 777 778
    }

    /*
     * We found a path, but didn't find any whitespace,
     * so it must be still incomplete - we should at
779 780
     * least see a \n - indicate that we want to carry
     * on trying again
781
     */
782
    return 1;
783 784 785
}

static int
786
qemudFindCharDevicePTYs(virConnectPtr conn,
787
                        virDomainObjPtr vm,
788 789
                        const char *output,
                        int fd ATTRIBUTE_UNUSED)
790
{
791
    struct qemud_driver* driver = conn->privateData;
792
    char *monitor = NULL;
793
    size_t offset = 0;
794
    int ret, i;
795 796 797 798 799

    /* 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.... */
800

801
    /* So first comes the monitor device */
802 803
    if ((ret = qemudExtractMonitorPath(conn, output, &offset, &monitor)) != 0)
        goto cleanup;
804

805
    /* then the serial devices */
806 807
    for (i = 0 ; i < vm->def->nserials ; i++) {
        virDomainChrDefPtr chr = vm->def->serials[i];
808 809 810 811
        if (chr->type == VIR_DOMAIN_CHR_TYPE_PTY) {
            if ((ret = qemudExtractMonitorPath(conn, output, &offset,
                                               &chr->data.file.path)) != 0)
                goto cleanup;
812 813 814 815
        }
    }

    /* and finally the parallel devices */
816 817
    for (i = 0 ; i < vm->def->nparallels ; i++) {
        virDomainChrDefPtr chr = vm->def->parallels[i];
818 819 820 821
        if (chr->type == VIR_DOMAIN_CHR_TYPE_PTY) {
            if ((ret = qemudExtractMonitorPath(conn, output, &offset,
                                               &chr->data.file.path)) != 0)
                goto cleanup;
822 823 824 825
        }
    }

    /* Got them all, so now open the monitor console */
826
    ret = qemudOpenMonitor(conn, driver, vm, monitor, 0);
827 828 829 830

cleanup:
    VIR_FREE(monitor);
    return ret;
831 832
}

833
static int qemudWaitForMonitor(virConnectPtr conn,
834 835 836
                               struct qemud_driver* driver,
                               virDomainObjPtr vm, off_t pos)
{
837
    char buf[1024]; /* Plenty of space to get startup greeting */
838 839 840 841 842
    int logfd;
    int ret;

    if ((logfd = qemudLogReadFD(conn, driver->logDir, vm->def->name, pos))
        < 0)
843
        return -1;
844 845 846 847

    ret = qemudReadMonitorOutput(conn, vm, logfd, buf, sizeof(buf),
                                 qemudFindCharDevicePTYs,
                                 "console", 3000);
848 849
    if (close(logfd) < 0) {
        char ebuf[1024];
850
        qemudLog(QEMUD_WARN, _("Unable to close logfile: %s\n"),
851 852
                 virStrerror(errno, ebuf, sizeof ebuf));
    }
853 854 855 856 857 858 859 860 861 862 863

    if (ret == 1) /* Success */
        return 0;

    if (ret == -1)
        return -1;

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

866 867
static int
qemudDetectVcpuPIDs(virConnectPtr conn,
868
                    virDomainObjPtr vm) {
869 870 871 872 873 874
    char *qemucpus = NULL;
    char *line;
    int lastVcpu = -1;

    /* Only KVM has seperate threads for CPUs,
       others just use main QEMU process for CPU */
875
    if (vm->def->virtType != VIR_DOMAIN_VIRT_KVM)
876 877 878 879 880
        vm->nvcpupids = 1;
    else
        vm->nvcpupids = vm->def->vcpus;

    if (VIR_ALLOC_N(vm->vcpupids, vm->nvcpupids) < 0) {
881
        virReportOOMError(conn);
882 883 884
        return -1;
    }

885
    if (vm->def->virtType != VIR_DOMAIN_VIRT_KVM) {
886 887 888 889
        vm->vcpupids[0] = vm->pid;
        return 0;
    }

890
    if (qemudMonitorCommand(vm, "info cpus", &qemucpus) < 0) {
891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948
        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;

949
    VIR_FREE(qemucpus);
950 951 952 953
    return 0;

error:
    VIR_FREE(vm->vcpupids);
954 955
    vm->nvcpupids = 0;
    VIR_FREE(qemucpus);
956 957 958 959 960 961 962 963

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

964 965
static int
qemudInitCpus(virConnectPtr conn,
D
Daniel Veillard 已提交
966 967
              virDomainObjPtr vm,
              const char *migrateFrom) {
968 969 970 971 972 973 974 975 976 977 978 979 980 981 982
    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 已提交
983 984 985 986 987 988
    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++)
989
            CPU_SET(i, &mask);
D
Daniel P. Berrange 已提交
990
    }
991 992 993 994

    for (i = 0 ; i < vm->nvcpupids ; i++) {
        if (sched_setaffinity(vm->vcpupids[i],
                              sizeof(mask), &mask) < 0) {
995 996
            virReportSystemError(conn, errno, "%s",
                                 _("failed to set CPU affinity"));
997 998 999 1000 1001
            return -1;
        }
    }
#endif /* HAVE_SCHED_GETAFFINITY */

D
Daniel Veillard 已提交
1002 1003
    if (migrateFrom == NULL) {
        /* Allow the CPUS to start executing */
1004
        if (qemudMonitorCommand(vm, "cont", &info) < 0) {
D
Daniel Veillard 已提交
1005 1006 1007 1008 1009
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                             "%s", _("resume operation failed"));
            return -1;
        }
        VIR_FREE(info);
1010 1011 1012 1013 1014 1015
    }

    return 0;
}


1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029
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...
     */

    if (vm->def->graphics &&
        vm->def->graphics->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
1030
        (vm->def->graphics->data.vnc.passwd || driver->vncPassword)) {
1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048

        if (qemudMonitorCommandExtra(vm, "change vnc password",
                                     vm->def->graphics->data.vnc.passwd ?
                                     vm->def->graphics->data.vnc.passwd :
                                     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;
}


1049
static int qemudNextFreeVNCPort(struct qemud_driver *driver ATTRIBUTE_UNUSED) {
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 1078 1079 1080 1081 1082 1083 1084
    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;
}

1085 1086 1087
static virDomainPtr qemudDomainLookupByName(virConnectPtr conn,
                                            const char *name);

1088 1089
static int qemudStartVMDaemon(virConnectPtr conn,
                              struct qemud_driver *driver,
1090
                              virDomainObjPtr vm,
1091 1092
                              const char *migrateFrom,
                              int stdin_fd) {
1093
    const char **argv = NULL, **tmp;
1094
    const char **progenv = NULL;
1095
    int i, ret;
1096
    struct stat sb;
1097 1098
    int *tapfds = NULL;
    int ntapfds = 0;
1099
    unsigned int qemuCmdFlags;
1100
    fd_set keepfd;
1101
    const char *emulator;
G
Guido Günther 已提交
1102
    pid_t child;
1103
    int pos = -1;
1104
    char ebuf[1024];
1105 1106

    FD_ZERO(&keepfd);
1107

1108
    if (virDomainIsActive(vm)) {
1109
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1110
                         "%s", _("VM is already active"));
1111 1112 1113
        return -1;
    }

1114 1115 1116
    if (vm->def->graphics &&
        vm->def->graphics->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
        vm->def->graphics->data.vnc.autoport) {
1117
        int port = qemudNextFreeVNCPort(driver);
1118
        if (port < 0) {
1119
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1120
                             "%s", _("Unable to find an unused VNC port"));
1121 1122
            return -1;
        }
1123 1124
        vm->def->graphics->data.vnc.port = port;
    }
1125

1126
    if (virFileMakePath(driver->logDir) < 0) {
1127 1128 1129
        virReportSystemError(conn, errno,
                             _("cannot create log directory %s"),
                             driver->logDir);
1130 1131 1132
        return -1;
    }

1133
    if((vm->logfile = qemudLogFD(conn, driver->logDir, vm->def->name)) < 0)
1134 1135
        return -1;

1136 1137 1138 1139 1140 1141
    emulator = vm->def->emulator;
    if (!emulator)
        emulator = virDomainDefDefaultEmulator(conn, vm->def, driver->caps);
    if (!emulator)
        return -1;

1142 1143 1144 1145
    /* 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
     */
1146
    if (stat(emulator, &sb) < 0) {
1147 1148 1149
        virReportSystemError(conn, errno,
                             _("Cannot find QEMU binary %s"),
                             emulator);
1150 1151 1152
        return -1;
    }

1153
    if (qemudExtractVersionInfo(emulator,
1154 1155 1156 1157
                                NULL,
                                &qemuCmdFlags) < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("Cannot determine QEMU argv syntax %s"),
1158
                         emulator);
1159 1160
        return -1;
    }
1161

1162
    vm->def->id = driver->nextvmid++;
1163
    if (qemudBuildCommandLine(conn, driver, vm,
1164
                              qemuCmdFlags, &argv, &progenv,
1165
                              &tapfds, &ntapfds, migrateFrom) < 0) {
1166
        close(vm->logfile);
1167
        vm->def->id = -1;
1168 1169 1170 1171
        vm->logfile = -1;
        return -1;
    }

1172 1173 1174
    tmp = progenv;
    while (*tmp) {
        if (safewrite(vm->logfile, *tmp, strlen(*tmp)) < 0)
1175
            qemudLog(QEMUD_WARN, _("Unable to write envv to logfile: %s\n"),
1176
                     virStrerror(errno, ebuf, sizeof ebuf));
1177
        if (safewrite(vm->logfile, " ", 1) < 0)
1178
            qemudLog(QEMUD_WARN, _("Unable to write envv to logfile: %s\n"),
1179
                     virStrerror(errno, ebuf, sizeof ebuf));
1180 1181
        tmp++;
    }
1182 1183
    tmp = argv;
    while (*tmp) {
1184
        if (safewrite(vm->logfile, *tmp, strlen(*tmp)) < 0)
1185
            qemudLog(QEMUD_WARN, _("Unable to write argv to logfile: %s\n"),
1186
                     virStrerror(errno, ebuf, sizeof ebuf));
1187
        if (safewrite(vm->logfile, " ", 1) < 0)
1188
            qemudLog(QEMUD_WARN, _("Unable to write argv to logfile: %s\n"),
1189
                     virStrerror(errno, ebuf, sizeof ebuf));
1190 1191
        tmp++;
    }
1192
    if (safewrite(vm->logfile, "\n", 1) < 0)
1193
        qemudLog(QEMUD_WARN, _("Unable to write argv to logfile: %s\n"),
1194
                 virStrerror(errno, ebuf, sizeof ebuf));
1195

1196
    if ((pos = lseek(vm->logfile, 0, SEEK_END)) < 0)
1197
        qemudLog(QEMUD_WARN, _("Unable to seek to end of logfile: %s\n"),
1198
                 virStrerror(errno, ebuf, sizeof ebuf));
1199

1200 1201 1202
    for (i = 0 ; i < ntapfds ; i++)
        FD_SET(tapfds[i], &keepfd);

G
Guido Günther 已提交
1203
    ret = virExec(conn, argv, progenv, &keepfd, &child,
1204
                  stdin_fd, &vm->logfile, &vm->logfile,
G
Guido Günther 已提交
1205 1206 1207 1208 1209
                  VIR_EXEC_NONBLOCK | VIR_EXEC_DAEMON);

    /* wait for qemu process to to show up */
    if (ret == 0) {
        int retries = 100;
1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225
        int childstat;

        while (waitpid(child, &childstat, 0) == -1 &&
               errno == EINTR);

        if (childstat == 0) {
            while (retries) {
                if ((ret = virFileReadPid(driver->stateDir, vm->def->name, &vm->pid)) == 0)
                    break;
                usleep(100*1000);
                retries--;
            }
            if (ret)
                qemudLog(QEMUD_WARN, _("Domain %s didn't show up\n"), vm->def->name);
        } else {
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
J
Jim Meyering 已提交
1226
                             "%s", _("Unable to daemonize QEMU process"));
1227
            ret = -1;
G
Guido Günther 已提交
1228 1229 1230 1231
        }
    }

    if (ret == 0) {
1232
        vm->state = migrateFrom ? VIR_DOMAIN_PAUSED : VIR_DOMAIN_RUNNING;
G
Guido Günther 已提交
1233
    } else
1234
        vm->def->id = -1;
1235

1236
    for (i = 0 ; argv[i] ; i++)
1237 1238
        VIR_FREE(argv[i]);
    VIR_FREE(argv);
1239

1240 1241 1242 1243
    for (i = 0 ; progenv[i] ; i++)
        VIR_FREE(progenv[i]);
    VIR_FREE(progenv);

1244 1245 1246
    if (tapfds) {
        for (i = 0 ; i < ntapfds ; i++) {
            close(tapfds[i]);
1247
        }
1248
        VIR_FREE(tapfds);
1249 1250
    }

1251
    if (ret == 0) {
1252
        if ((qemudWaitForMonitor(conn, driver, vm, pos) < 0) ||
1253
            (qemudDetectVcpuPIDs(conn, vm) < 0) ||
1254 1255
            (qemudInitCpus(conn, vm, migrateFrom) < 0) ||
            (qemudInitPasswords(conn, driver, vm) < 0)) {
1256 1257 1258
            qemudShutdownVMDaemon(conn, driver, vm);
            return -1;
        }
1259
    }
1260
    qemudSaveDomainStatus(conn, qemu_driver, vm);
1261

1262
    return ret;
1263 1264 1265
}


1266
static void qemudShutdownVMDaemon(virConnectPtr conn ATTRIBUTE_UNUSED,
1267 1268
                                  struct qemud_driver *driver,
                                  virDomainObjPtr vm) {
1269
    if (!virDomainIsActive(vm))
1270
        return;
1271

1272
    qemudLog(QEMUD_DEBUG, _("Shutting down VM '%s'\n"), vm->def->name);
1273

G
Guido Günther 已提交
1274 1275
    if (virKillProcess(vm->pid, 0) == 0 &&
        virKillProcess(vm->pid, SIGTERM) < 0)
1276 1277 1278
        virReportSystemError(conn, errno,
                             _("Failed to send SIGTERM to %s (%d)"),
                             vm->def->name, vm->pid);
1279

1280 1281 1282 1283
    if (vm->monitor_watch != -1) {
        virEventRemoveHandle(vm->monitor_watch);
        vm->monitor_watch = -1;
    }
1284

1285 1286
    if (close(vm->logfile) < 0) {
        char ebuf[1024];
1287
        qemudLog(QEMUD_WARN, _("Unable to close logfile: %s\n"),
1288 1289
                 virStrerror(errno, ebuf, sizeof ebuf));
    }
1290 1291 1292 1293 1294
    if (vm->monitor != -1)
        close(vm->monitor);
    vm->logfile = -1;
    vm->monitor = -1;

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

G
Guido Günther 已提交
1298
    qemudRemoveDomainStatus(conn, driver, vm);
1299
    vm->pid = -1;
1300
    vm->def->id = -1;
1301
    vm->state = VIR_DOMAIN_SHUTOFF;
1302
    VIR_FREE(vm->vcpupids);
1303
    vm->nvcpupids = 0;
1304 1305

    if (vm->newDef) {
1306
        virDomainDefFree(vm->def);
1307
        vm->def = vm->newDef;
1308
        vm->def->id = -1;
1309 1310 1311 1312 1313
        vm->newDef = NULL;
    }
}


1314
static void
1315
qemudDispatchVMEvent(int watch, int fd, int events, void *opaque) {
1316
    struct qemud_driver *driver = opaque;
1317
    virDomainObjPtr vm = NULL;
1318
    virDomainEventPtr event = NULL;
1319
    unsigned int i;
1320
    int quit = 0, failed = 0;
1321

1322
    qemuDriverLock(driver);
1323
    for (i = 0 ; i < driver->domains.count ; i++) {
1324 1325 1326
        virDomainObjPtr tmpvm = driver->domains.objs[i];
        virDomainObjLock(tmpvm);
        if (virDomainIsActive(tmpvm) &&
1327
            tmpvm->monitor_watch == watch) {
1328
            vm = tmpvm;
1329
            break;
1330
        }
1331
        virDomainObjUnlock(tmpvm);
1332 1333 1334
    }

    if (!vm)
1335
        goto cleanup;
1336

1337
    if (vm->monitor != fd) {
1338 1339
        failed = 1;
    } else {
1340
        if (events & (VIR_EVENT_HANDLE_HANGUP | VIR_EVENT_HANDLE_ERROR))
1341
            quit = 1;
1342 1343 1344 1345
        else {
            qemudLog(QEMUD_ERROR, _("unhandled fd event %d for %s"),
                                    events, vm->def->name);
            failed = 1;
1346
        }
1347 1348
    }

1349
    if (failed || quit) {
1350 1351 1352 1353 1354
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         quit ?
                                         VIR_DOMAIN_EVENT_STOPPED_SHUTDOWN :
                                         VIR_DOMAIN_EVENT_STOPPED_FAILED);
1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365
        qemudShutdownVMDaemon(NULL, driver, vm);
        if (!vm->persistent) {
            virDomainRemoveInactive(&driver->domains,
                                    vm);
            vm = NULL;
        }
    }

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
1366 1367
    if (event)
        qemuDomainEventQueue(driver, event);
1368
    qemuDriverUnlock(driver);
1369 1370
}

1371
static int
1372 1373 1374 1375 1376
qemudMonitorCommandExtra(const virDomainObjPtr vm,
                         const char *cmd,
                         const char *extra,
                         const char *extraPrompt,
                         char **reply) {
D
Daniel P. Berrange 已提交
1377 1378
    int size = 0;
    char *buf = NULL;
1379
    size_t cmdlen = strlen(cmd);
1380
    size_t extralen = extra ? strlen(extra) : 0;
D
Daniel P. Berrange 已提交
1381

1382 1383 1384
    if (safewrite(vm->monitor, cmd, cmdlen) != cmdlen)
        return -1;
    if (safewrite(vm->monitor, "\r", 1) != 1)
D
Daniel P. Berrange 已提交
1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396
        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 已提交
1397

1398 1399
            if (got == 0)
                goto error;
D
Daniel P. Berrange 已提交
1400 1401 1402 1403 1404
            if (got < 0) {
                if (errno == EINTR)
                    continue;
                if (errno == EAGAIN)
                    break;
1405
                goto error;
1406
            }
1407
            if (VIR_REALLOC_N(buf, size+got+1) < 0)
1408 1409
                goto error;

D
Daniel P. Berrange 已提交
1410 1411 1412 1413
            memmove(buf+size, data, got);
            buf[size+got] = '\0';
            size += got;
        }
1414

D
Daniel P. Berrange 已提交
1415
        /* Look for QEMU prompt to indicate completion */
1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440
        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 已提交
1441

1442 1443
                break;
            }
D
Daniel P. Berrange 已提交
1444 1445 1446 1447 1448 1449
        }
    pollagain:
        /* Need to wait for more data */
        if (poll(&fd, 1, -1) < 0) {
            if (errno == EINTR)
                goto pollagain;
1450
            goto error;
D
Daniel P. Berrange 已提交
1451 1452 1453
        }
    }

1454
    /* Log, but ignore failures to write logfile for VM */
1455 1456
    if (safewrite(vm->logfile, buf, strlen(buf)) < 0) {
        char ebuf[1024];
1457
        qemudLog(QEMUD_WARN, _("Unable to log VM console data: %s\n"),
1458 1459
                 virStrerror(errno, ebuf, sizeof ebuf));
    }
1460

D
Daniel P. Berrange 已提交
1461 1462
    *reply = buf;
    return 0;
1463 1464 1465 1466

 error:
    if (buf) {
        /* Log, but ignore failures to write logfile for VM */
1467 1468
        if (safewrite(vm->logfile, buf, strlen(buf)) < 0) {
            char ebuf[1024];
1469
            qemudLog(QEMUD_WARN, _("Unable to log VM console data: %s\n"),
1470 1471
                     virStrerror(errno, ebuf, sizeof ebuf));
        }
1472
        VIR_FREE(buf);
1473 1474
    }
    return -1;
D
Daniel P. Berrange 已提交
1475 1476
}

1477 1478 1479 1480 1481 1482 1483 1484
static int
qemudMonitorCommand(const virDomainObjPtr vm,
                    const char *cmd,
                    char **reply) {
    return qemudMonitorCommandExtra(vm, cmd, NULL, NULL, reply);
}


1485 1486 1487 1488 1489 1490
/**
 * qemudProbe:
 *
 * Probe for the availability of the qemu driver, assume the
 * presence of QEmu emulation if the binaries are installed
 */
1491
static int qemudProbe(void)
1492 1493 1494
{
    if ((virFileExists("/usr/bin/qemu")) ||
        (virFileExists("/usr/bin/qemu-kvm")) ||
G
Guido Günther 已提交
1495
        (virFileExists("/usr/bin/kvm")) ||
1496 1497 1498 1499
        (virFileExists("/usr/bin/xenner")))
        return 1;

    return 0;
1500
}
1501

1502
static virDrvOpenStatus qemudOpen(virConnectPtr conn,
1503
                                  virConnectAuthPtr auth ATTRIBUTE_UNUSED,
1504
                                  int flags ATTRIBUTE_UNUSED) {
1505 1506 1507
    uid_t uid = getuid();

    if (qemu_driver == NULL)
1508
        goto decline;
1509

1510 1511 1512 1513 1514 1515
    if (!qemudProbe())
        goto decline;

    if (conn->uri == NULL) {
        conn->uri = xmlParseURI(uid ? "qemu:///session" : "qemu:///system");
        if (!conn->uri) {
1516
            virReportOOMError(conn);
1517 1518 1519 1520
            return VIR_DRV_OPEN_ERROR;
        }
    } else if (conn->uri->scheme == NULL ||
               conn->uri->path == NULL)
1521 1522
        goto decline;

1523
    if (STRNEQ (conn->uri->scheme, "qemu"))
1524 1525
        goto decline;

1526
    if (uid != 0) {
1527
        if (STRNEQ (conn->uri->path, "/session"))
1528
            goto decline;
1529
    } else { /* root */
1530 1531
        if (STRNEQ (conn->uri->path, "/system") &&
            STRNEQ (conn->uri->path, "/session"))
1532
            goto decline;
1533 1534 1535 1536 1537
    }

    conn->privateData = qemu_driver;

    return VIR_DRV_OPEN_SUCCESS;
1538 1539

 decline:
1540
    return VIR_DRV_OPEN_DECLINED;
1541 1542 1543
}

static int qemudClose(virConnectPtr conn) {
1544
    struct qemud_driver *driver = conn->privateData;
1545 1546 1547

    /* Get rid of callbacks registered for this conn */
    virDomainEventCallbackListRemoveConn(conn, driver->domainEventCallbacks);
1548 1549 1550 1551 1552 1553

    conn->privateData = NULL;

    return 0;
}

D
Daniel Veillard 已提交
1554 1555 1556 1557 1558 1559 1560 1561 1562 1563
/* 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;
    }
}

1564
static const char *qemudGetType(virConnectPtr conn ATTRIBUTE_UNUSED) {
1565
    return "QEMU";
1566 1567
}

1568 1569 1570 1571 1572

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

    int r, fd;
1573

1574 1575
    fd = open(KVM_DEVICE, O_RDONLY);
    if (fd < 0) {
1576 1577
        virReportSystemError(NULL, errno, _("Unable to open %s"), KVM_DEVICE);
        return -1;
1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588
    }

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

    close(fd);
    return maxvcpus;
}


1589
static int qemudGetMaxVCPUs(virConnectPtr conn, const char *type) {
1590 1591 1592
    if (!type)
        return 16;

1593
    if (STRCASEEQ(type, "qemu"))
1594 1595
        return 16;

1596
    if (STRCASEEQ(type, "kvm"))
1597
        return kvmGetMaxVCPUs();
1598

1599
    if (STRCASEEQ(type, "kqemu"))
1600
        return 1;
1601 1602 1603

    qemudReportError(conn, NULL, NULL, VIR_ERR_INVALID_ARG,
                     _("unknown type '%s'"), type);
1604 1605 1606
    return -1;
}

1607 1608 1609
static int qemudGetNodeInfo(virConnectPtr conn,
                            virNodeInfoPtr nodeinfo) {
    return virNodeInfoPopulate(conn, nodeinfo);
1610 1611 1612
}


1613
static char *qemudGetCapabilities(virConnectPtr conn) {
1614
    struct qemud_driver *driver = conn->privateData;
1615
    char *xml;
1616

1617
    qemuDriverLock(driver);
1618
    if ((xml = virCapabilitiesFormatXML(driver->caps)) == NULL)
1619
        virReportOOMError(conn);
1620
    qemuDriverUnlock(driver);
1621

1622
    return xml;
1623 1624 1625
}


1626 1627 1628 1629 1630 1631 1632 1633
#if HAVE_NUMACTL
static int
qemudNodeGetCellsFreeMemory(virConnectPtr conn,
                            unsigned long long *freeMems,
                            int startCell,
                            int maxCells)
{
    int n, lastCell, numCells;
1634
    int ret = -1;
1635 1636 1637 1638

    if (numa_available() < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("NUMA not supported on this host"));
1639
        goto cleanup;
1640 1641 1642 1643 1644 1645 1646 1647 1648 1649
    }
    lastCell = startCell + maxCells - 1;
    if (lastCell > numa_max_node())
        lastCell = numa_max_node();

    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"));
1650
            goto cleanup;
1651 1652 1653
        }
        freeMems[numCells++] = mem;
    }
1654 1655 1656 1657
    ret = numCells;

cleanup:
    return ret;
1658 1659 1660 1661 1662
}

static unsigned long long
qemudNodeGetFreeMemory (virConnectPtr conn)
{
1663
    unsigned long long freeMem = -1;
1664
    int n;
1665

1666 1667 1668
    if (numa_available() < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("NUMA not supported on this host"));
1669
        goto cleanup;
1670 1671 1672 1673 1674 1675 1676
    }

    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"));
1677
            goto cleanup;
1678 1679 1680 1681
        }
        freeMem += mem;
    }

1682
cleanup:
1683 1684 1685 1686
    return freeMem;
}

#endif
1687

D
Daniel P. Berrange 已提交
1688 1689 1690
static int qemudGetProcessInfo(unsigned long long *cpuTime, int pid) {
    char proc[PATH_MAX];
    FILE *pidinfo;
1691
    unsigned long long usertime, systime;
D
Daniel P. Berrange 已提交
1692 1693 1694 1695 1696 1697

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

    if (!(pidinfo = fopen(proc, "r"))) {
1698
        /*printf("cannot read pid info");*/
D
Daniel P. Berrange 已提交
1699 1700 1701 1702 1703
        /* VM probably shut down, so fake 0 */
        *cpuTime = 0;
        return 0;
    }

1704
    if (fscanf(pidinfo, "%*d %*s %*c %*d %*d %*d %*d %*d %*u %*u %*u %*u %*u %llu %llu", &usertime, &systime) != 2) {
1705
        qemudDebug("not enough arg");
D
Daniel P. Berrange 已提交
1706 1707 1708 1709 1710 1711 1712 1713
        return -1;
    }

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

1716
    qemudDebug("Got %llu %llu %llu", usertime, systime, *cpuTime);
D
Daniel P. Berrange 已提交
1717 1718 1719 1720 1721 1722 1723

    fclose(pidinfo);

    return 0;
}


1724
static virDomainPtr qemudDomainLookupByID(virConnectPtr conn,
1725
                                          int id) {
1726 1727 1728 1729
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;

1730
    qemuDriverLock(driver);
1731
    vm  = virDomainFindByID(&driver->domains, id);
1732
    qemuDriverUnlock(driver);
1733 1734

    if (!vm) {
1735
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_DOMAIN, NULL);
1736
        goto cleanup;
1737 1738
    }

1739
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1740
    if (dom) dom->id = vm->def->id;
1741 1742

cleanup:
1743 1744
    if (vm)
        virDomainObjUnlock(vm);
1745 1746
    return dom;
}
1747

1748
static virDomainPtr qemudDomainLookupByUUID(virConnectPtr conn,
1749
                                            const unsigned char *uuid) {
1750 1751 1752
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
1753

1754
    qemuDriverLock(driver);
1755
    vm = virDomainFindByUUID(&driver->domains, uuid);
1756 1757
    qemuDriverUnlock(driver);

1758
    if (!vm) {
1759
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_DOMAIN, NULL);
1760
        goto cleanup;
1761 1762
    }

1763
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1764
    if (dom) dom->id = vm->def->id;
1765 1766

cleanup:
1767 1768
    if (vm)
        virDomainObjUnlock(vm);
1769 1770
    return dom;
}
1771

1772
static virDomainPtr qemudDomainLookupByName(virConnectPtr conn,
1773
                                            const char *name) {
1774 1775 1776
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
1777

1778
    qemuDriverLock(driver);
1779
    vm = virDomainFindByName(&driver->domains, name);
1780 1781
    qemuDriverUnlock(driver);

1782
    if (!vm) {
1783
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_DOMAIN, NULL);
1784
        goto cleanup;
1785 1786
    }

1787
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1788
    if (dom) dom->id = vm->def->id;
1789 1790

cleanup:
1791 1792
    if (vm)
        virDomainObjUnlock(vm);
1793 1794 1795
    return dom;
}

1796
static int qemudGetVersion(virConnectPtr conn, unsigned long *version) {
1797 1798 1799
    struct qemud_driver *driver = conn->privateData;
    int ret = -1;

1800
    qemuDriverLock(driver);
1801
    if (qemudExtractVersion(conn, driver) < 0)
1802
        goto cleanup;
1803

1804
    *version = qemu_driver->qemuVersion;
1805 1806 1807
    ret = 0;

cleanup:
1808
    qemuDriverUnlock(driver);
1809
    return ret;
D
Daniel P. Berrange 已提交
1810 1811
}

1812 1813 1814
static char *
qemudGetHostname (virConnectPtr conn)
{
1815
    char *result;
1816

1817 1818
    result = virGetHostname();
    if (result == NULL) {
1819 1820
        virReportSystemError (conn, errno,
                              "%s", _("failed to determine host name"));
1821 1822 1823
        return NULL;
    }
    /* Caller frees this string. */
1824
    return result;
1825 1826
}

1827
static int qemudListDomains(virConnectPtr conn, int *ids, int nids) {
1828
    struct qemud_driver *driver = conn->privateData;
1829 1830
    int got = 0, i;

1831 1832 1833
    qemuDriverLock(driver);
    for (i = 0 ; i < driver->domains.count && got < nids ; i++) {
        virDomainObjLock(driver->domains.objs[i]);
1834 1835
        if (virDomainIsActive(driver->domains.objs[i]))
            ids[got++] = driver->domains.objs[i]->def->id;
1836 1837 1838
        virDomainObjUnlock(driver->domains.objs[i]);
    }
    qemuDriverUnlock(driver);
1839

D
Daniel P. Berrange 已提交
1840 1841
    return got;
}
1842

1843
static int qemudNumDomains(virConnectPtr conn) {
1844
    struct qemud_driver *driver = conn->privateData;
1845 1846
    int n = 0, i;

1847 1848 1849
    qemuDriverLock(driver);
    for (i = 0 ; i < driver->domains.count ; i++) {
        virDomainObjLock(driver->domains.objs[i]);
1850
        if (virDomainIsActive(driver->domains.objs[i]))
1851
            n++;
1852 1853 1854
        virDomainObjUnlock(driver->domains.objs[i]);
    }
    qemuDriverUnlock(driver);
1855

1856
    return n;
D
Daniel P. Berrange 已提交
1857
}
1858

1859
static virDomainPtr qemudDomainCreate(virConnectPtr conn, const char *xml,
1860
                                      unsigned int flags ATTRIBUTE_UNUSED) {
1861
    struct qemud_driver *driver = conn->privateData;
1862
    virDomainDefPtr def;
1863
    virDomainObjPtr vm = NULL;
1864
    virDomainPtr dom = NULL;
1865
    virDomainEventPtr event = NULL;
D
Daniel P. Berrange 已提交
1866

1867
    qemuDriverLock(driver);
1868 1869
    if (!(def = virDomainDefParseString(conn, driver->caps, xml,
                                        VIR_DOMAIN_XML_INACTIVE)))
1870
        goto cleanup;
1871

1872
    vm = virDomainFindByName(&driver->domains, def->name);
1873
    if (vm) {
1874
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
1875
                         _("domain '%s' is already defined"),
1876
                         def->name);
1877
        goto cleanup;
1878
    }
1879
    vm = virDomainFindByUUID(&driver->domains, def->uuid);
1880 1881 1882 1883 1884
    if (vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(def->uuid, uuidstr);
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
1885
                         _("domain with uuid '%s' is already defined"),
1886
                         uuidstr);
1887
        goto cleanup;
1888
    }
1889

1890 1891
    if (!(vm = virDomainAssignDef(conn,
                                  &driver->domains,
1892 1893 1894 1895
                                  def)))
        goto cleanup;

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

1897
    if (qemudStartVMDaemon(conn, driver, vm, NULL, -1) < 0) {
1898 1899
        virDomainRemoveInactive(&driver->domains,
                                vm);
1900
        vm = NULL;
1901
        goto cleanup;
D
Daniel P. Berrange 已提交
1902
    }
1903 1904 1905 1906

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

1908
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1909
    if (dom) dom->id = vm->def->id;
1910 1911 1912

cleanup:
    virDomainDefFree(def);
1913 1914
    if (vm)
        virDomainObjUnlock(vm);
1915 1916
    if (event)
        qemuDomainEventQueue(driver, event);
1917
    qemuDriverUnlock(driver);
1918
    return dom;
D
Daniel P. Berrange 已提交
1919 1920 1921
}


1922
static int qemudDomainSuspend(virDomainPtr dom) {
1923
    struct qemud_driver *driver = dom->conn->privateData;
D
Daniel P. Berrange 已提交
1924
    char *info;
1925 1926
    virDomainObjPtr vm;
    int ret = -1;
1927
    virDomainEventPtr event = NULL;
1928

1929
    qemuDriverLock(driver);
1930
    vm = virDomainFindByID(&driver->domains, dom->id);
1931 1932
    qemuDriverUnlock(driver);

D
Daniel P. Berrange 已提交
1933
    if (!vm) {
1934
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN, _("no domain with matching id %d"), dom->id);
1935
        goto cleanup;
D
Daniel P. Berrange 已提交
1936
    }
1937
    if (!virDomainIsActive(vm)) {
1938 1939
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("domain is not running"));
1940
        goto cleanup;
D
Daniel P. Berrange 已提交
1941
    }
1942
    if (vm->state != VIR_DOMAIN_PAUSED) {
1943
        if (qemudMonitorCommand(vm, "stop", &info) < 0) {
1944 1945 1946 1947 1948 1949
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                             "%s", _("suspend operation failed"));
            goto cleanup;
        }
        vm->state = VIR_DOMAIN_PAUSED;
        qemudDebug("Reply %s", info);
1950 1951 1952
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_SUSPENDED,
                                         VIR_DOMAIN_EVENT_SUSPENDED_PAUSED);
1953
        VIR_FREE(info);
D
Daniel P. Berrange 已提交
1954
    }
1955
    qemudSaveDomainStatus(dom->conn, driver, vm);
1956 1957 1958
    ret = 0;

cleanup:
1959 1960
    if (vm)
        virDomainObjUnlock(vm);
1961 1962 1963 1964 1965 1966

    if (event) {
        qemuDriverLock(driver);
        qemuDomainEventQueue(driver, event);
        qemuDriverUnlock(driver);
    }
1967
    return ret;
D
Daniel P. Berrange 已提交
1968 1969 1970
}


1971
static int qemudDomainResume(virDomainPtr dom) {
1972
    struct qemud_driver *driver = dom->conn->privateData;
D
Daniel P. Berrange 已提交
1973
    char *info;
1974 1975
    virDomainObjPtr vm;
    int ret = -1;
1976
    virDomainEventPtr event = NULL;
1977

1978
    qemuDriverLock(driver);
1979
    vm = virDomainFindByID(&driver->domains, dom->id);
1980 1981
    qemuDriverUnlock(driver);

D
Daniel P. Berrange 已提交
1982
    if (!vm) {
1983 1984
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         _("no domain with matching id %d"), dom->id);
1985
        goto cleanup;
D
Daniel P. Berrange 已提交
1986
    }
1987
    if (!virDomainIsActive(vm)) {
1988 1989
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("domain is not running"));
1990
        goto cleanup;
D
Daniel P. Berrange 已提交
1991
    }
1992
    if (vm->state == VIR_DOMAIN_PAUSED) {
1993
        if (qemudMonitorCommand(vm, "cont", &info) < 0) {
1994 1995 1996 1997 1998 1999
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                             "%s", _("resume operation failed"));
            goto cleanup;
        }
        vm->state = VIR_DOMAIN_RUNNING;
        qemudDebug("Reply %s", info);
2000 2001 2002
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_RESUMED,
                                         VIR_DOMAIN_EVENT_RESUMED_UNPAUSED);
2003
        VIR_FREE(info);
D
Daniel P. Berrange 已提交
2004
    }
2005
    qemudSaveDomainStatus(dom->conn, driver, vm);
2006 2007 2008
    ret = 0;

cleanup:
2009 2010
    if (vm)
        virDomainObjUnlock(vm);
2011 2012 2013 2014 2015
    if (event) {
        qemuDriverLock(driver);
        qemuDomainEventQueue(driver, event);
        qemuDriverUnlock(driver);
    }
2016
    return ret;
D
Daniel P. Berrange 已提交
2017 2018 2019
}


2020
static int qemudDomainShutdown(virDomainPtr dom) {
2021 2022
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2023
    char* info;
2024
    int ret = -1;
2025

2026
    qemuDriverLock(driver);
2027
    vm = virDomainFindByID(&driver->domains, dom->id);
2028 2029
    qemuDriverUnlock(driver);

2030 2031
    if (!vm) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2032
                         _("no domain with matching id %d"), dom->id);
2033
        goto cleanup;
2034 2035
    }

2036
    if (qemudMonitorCommand(vm, "system_powerdown", &info) < 0) {
2037
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2038
                         "%s", _("shutdown operation failed"));
2039
        goto cleanup;
2040
    }
D
Daniel Veillard 已提交
2041
    VIR_FREE(info);
2042 2043 2044
    ret = 0;

cleanup:
2045 2046
    if (vm)
        virDomainObjUnlock(vm);
2047
    return ret;
2048 2049 2050
}


2051
static int qemudDomainDestroy(virDomainPtr dom) {
2052 2053 2054
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
2055
    virDomainEventPtr event = NULL;
2056

2057
    qemuDriverLock(driver);
2058
    vm  = virDomainFindByID(&driver->domains, dom->id);
D
Daniel P. Berrange 已提交
2059
    if (!vm) {
2060
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2061
                         _("no domain with matching id %d"), dom->id);
2062
        goto cleanup;
D
Daniel P. Berrange 已提交
2063
    }
2064

2065
    qemudShutdownVMDaemon(dom->conn, driver, vm);
2066 2067 2068
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_DESTROYED);
2069
    if (!vm->persistent) {
2070 2071
        virDomainRemoveInactive(&driver->domains,
                                vm);
2072 2073
        vm = NULL;
    }
2074 2075 2076
    ret = 0;

cleanup:
2077 2078
    if (vm)
        virDomainObjUnlock(vm);
2079 2080
    if (event)
        qemuDomainEventQueue(driver, event);
2081
    qemuDriverUnlock(driver);
2082
    return ret;
D
Daniel P. Berrange 已提交
2083 2084 2085
}


2086
static char *qemudDomainGetOSType(virDomainPtr dom) {
2087 2088 2089
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *type = NULL;
2090

2091
    qemuDriverLock(driver);
2092
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2093
    qemuDriverUnlock(driver);
2094 2095
    if (!vm) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2096
                         "%s", _("no domain with matching uuid"));
2097
        goto cleanup;
2098 2099
    }

2100
    if (!(type = strdup(vm->def->os.type)))
2101
        virReportOOMError(dom->conn);
2102 2103

cleanup:
2104 2105
    if (vm)
        virDomainObjUnlock(vm);
2106 2107 2108
    return type;
}

2109 2110
/* Returns max memory in kb, 0 if error */
static unsigned long qemudDomainGetMaxMemory(virDomainPtr dom) {
2111 2112 2113
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    unsigned long ret = 0;
2114

2115
    qemuDriverLock(driver);
2116
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2117 2118
    qemuDriverUnlock(driver);

2119
    if (!vm) {
2120 2121 2122
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
2123
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2124
                         _("no domain with matching uuid '%s'"), uuidstr);
2125
        goto cleanup;
2126 2127
    }

2128 2129 2130
    ret = vm->def->maxmem;

cleanup:
2131 2132
    if (vm)
        virDomainObjUnlock(vm);
2133
    return ret;
2134 2135 2136
}

static int qemudDomainSetMaxMemory(virDomainPtr dom, unsigned long newmax) {
2137 2138 2139
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
2140

2141
    qemuDriverLock(driver);
2142
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2143 2144
    qemuDriverUnlock(driver);

2145
    if (!vm) {
2146 2147 2148
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
2149
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2150
                         _("no domain with matching uuid '%s'"), uuidstr);
2151
        goto cleanup;
2152 2153 2154 2155
    }

    if (newmax < vm->def->memory) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
2156
                         "%s", _("cannot set max memory lower than current memory"));
2157
        goto cleanup;;
2158 2159 2160
    }

    vm->def->maxmem = newmax;
2161 2162 2163
    ret = 0;

cleanup:
2164 2165
    if (vm)
        virDomainObjUnlock(vm);
2166
    return ret;
2167 2168 2169
}

static int qemudDomainSetMemory(virDomainPtr dom, unsigned long newmem) {
2170 2171 2172
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
2173

2174
    qemuDriverLock(driver);
2175
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2176
    qemuDriverUnlock(driver);
2177
    if (!vm) {
2178 2179 2180
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
2181
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2182
                         _("no domain with matching uuid '%s'"), uuidstr);
2183
        goto cleanup;
2184 2185
    }

2186
    if (virDomainIsActive(vm)) {
2187
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
2188
                         "%s", _("cannot set memory of an active domain"));
2189
        goto cleanup;
2190 2191 2192 2193
    }

    if (newmem > vm->def->maxmem) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
2194
                         "%s", _("cannot set memory higher than max memory"));
2195
        goto cleanup;
2196 2197 2198
    }

    vm->def->memory = newmem;
2199 2200 2201
    ret = 0;

cleanup:
2202 2203
    if (vm)
        virDomainObjUnlock(vm);
2204
    return ret;
2205 2206
}

2207
static int qemudDomainGetInfo(virDomainPtr dom,
2208
                              virDomainInfoPtr info) {
2209 2210 2211 2212
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;

2213
    qemuDriverLock(driver);
2214
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2215
    qemuDriverUnlock(driver);
D
Daniel P. Berrange 已提交
2216
    if (!vm) {
2217 2218
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
2219
        goto cleanup;
D
Daniel P. Berrange 已提交
2220 2221
    }

2222
    info->state = vm->state;
D
Daniel P. Berrange 已提交
2223

2224
    if (!virDomainIsActive(vm)) {
2225
        info->cpuTime = 0;
D
Daniel P. Berrange 已提交
2226
    } else {
2227
        if (qemudGetProcessInfo(&(info->cpuTime), vm->pid) < 0) {
2228
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED, ("cannot read cputime for domain"));
2229
            goto cleanup;
D
Daniel P. Berrange 已提交
2230 2231 2232
        }
    }

2233 2234 2235
    info->maxMem = vm->def->maxmem;
    info->memory = vm->def->memory;
    info->nrVirtCpu = vm->def->vcpus;
2236 2237 2238
    ret = 0;

cleanup:
2239 2240
    if (vm)
        virDomainObjUnlock(vm);
2241
    return ret;
D
Daniel P. Berrange 已提交
2242 2243 2244
}


D
Daniel P. Berrange 已提交
2245
static char *qemudEscape(const char *in, int shell)
2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266
{
    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 已提交
2267 2268 2269 2270
            if (shell)
                len += 5;
            else
                len += 1;
2271 2272 2273 2274 2275 2276 2277
            break;
        default:
            len += 1;
            break;
        }
    }

2278
    if (VIR_ALLOC_N(out, len + 1) < 0)
2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296
        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 已提交
2297 2298 2299 2300 2301 2302 2303 2304 2305
            if (shell) {
                out[j++] = '\'';
                out[j++] = '\\';
                out[j++] = '\\';
                out[j++] = '\'';
                out[j++] = '\'';
            } else {
                out[j++] = in[i];
            }
2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316
            break;
        default:
            out[j++] = in[i];
            break;
        }
    }
    out[j] = '\0';

    return out;
}

2317 2318 2319 2320 2321
static char *qemudEscapeMonitorArg(const char *in)
{
    return qemudEscape(in, 0);
}

D
Daniel P. Berrange 已提交
2322 2323 2324 2325
static char *qemudEscapeShellArg(const char *in)
{
    return qemudEscape(in, 1);
}
2326

2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337
#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];
};

2338
static int qemudDomainSave(virDomainPtr dom,
2339
                           const char *path) {
2340 2341 2342 2343 2344 2345 2346
    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;
2347
    struct qemud_save_header header;
2348
    int ret = -1;
2349
    virDomainEventPtr event = NULL;
2350 2351 2352 2353 2354

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

2355
    qemuDriverLock(driver);
2356 2357
    vm = virDomainFindByID(&driver->domains, dom->id);

D
Daniel P. Berrange 已提交
2358
    if (!vm) {
2359
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2360
                         _("no domain with matching id %d"), dom->id);
2361
        goto cleanup;
D
Daniel P. Berrange 已提交
2362
    }
2363

2364
    if (!virDomainIsActive(vm)) {
2365
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2366
                         "%s", _("domain is not running"));
2367
        goto cleanup;
D
Daniel P. Berrange 已提交
2368
    }
2369 2370 2371 2372

    /* Pause */
    if (vm->state == VIR_DOMAIN_RUNNING) {
        header.was_running = 1;
2373
        if (qemudMonitorCommand(vm, "stop", &info) < 0) {
2374
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2375
                             "%s", _("suspend operation failed"));
2376
            goto cleanup;
2377
        }
2378 2379 2380
        vm->state = VIR_DOMAIN_PAUSED;
        qemudDebug("Reply %s", info);
        VIR_FREE(info);
2381 2382 2383
    }

    /* Get XML for the domain */
2384
    xml = virDomainDefFormat(dom->conn, vm->def, VIR_DOMAIN_XML_SECURE);
2385 2386
    if (!xml) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2387
                         "%s", _("failed to get domain xml"));
2388
        goto cleanup;
2389 2390 2391 2392 2393 2394
    }
    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,
2395
                         _("failed to create '%s'"), path);
2396
        goto cleanup;
2397 2398 2399 2400
    }

    if (safewrite(fd, &header, sizeof(header)) != sizeof(header)) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2401
                         "%s", _("failed to write save header"));
2402
        goto cleanup;
2403 2404 2405 2406
    }

    if (safewrite(fd, xml, header.xml_len) != header.xml_len) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2407
                         "%s", _("failed to write xml"));
2408
        goto cleanup;
2409 2410
    }

2411
    if (close(fd) < 0) {
2412 2413 2414
        virReportSystemError(dom->conn, errno,
                             _("unable to save file %s"),
                             path);
2415 2416 2417
        goto cleanup;
    }
    fd = -1;
2418 2419 2420 2421 2422

    /* Migrate to file */
    safe_path = qemudEscapeShellArg(path);
    if (!safe_path) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2423
                         "%s", _("out of memory"));
2424
        goto cleanup;
2425
    }
2426
    if (virAsprintf(&command, "migrate \"exec:"
2427
                  "dd of='%s' oflag=append conv=notrunc 2>/dev/null"
2428
                  "\"", safe_path) == -1) {
2429
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2430
                         "%s", _("out of memory"));
2431 2432
        command = NULL;
        goto cleanup;
2433 2434
    }

2435
    if (qemudMonitorCommand(vm, command, &info) < 0) {
2436
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2437
                         "%s", _("migrate operation failed"));
2438
        goto cleanup;
2439 2440
    }

2441 2442 2443 2444 2445 2446 2447 2448
    DEBUG ("migrate reply: %s", info);

    /* 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"));
2449
        goto cleanup;
2450 2451
    }

2452 2453
    /* Shut it down */
    qemudShutdownVMDaemon(dom->conn, driver, vm);
2454 2455 2456
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_SAVED);
2457
    if (!vm->persistent) {
2458 2459
        virDomainRemoveInactive(&driver->domains,
                                vm);
2460 2461
        vm = NULL;
    }
2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472
    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);
2473 2474
    if (vm)
        virDomainObjUnlock(vm);
2475 2476
    if (event)
        qemuDomainEventQueue(driver, event);
2477
    qemuDriverUnlock(driver);
2478
    return ret;
D
Daniel P. Berrange 已提交
2479 2480 2481
}


2482
static int qemudDomainSetVcpus(virDomainPtr dom, unsigned int nvcpus) {
2483 2484
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2485
    int max;
2486
    int ret = -1;
2487

2488
    qemuDriverLock(driver);
2489
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2490 2491
    qemuDriverUnlock(driver);

2492
    if (!vm) {
2493 2494 2495
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
2496
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2497
                         _("no domain with matching uuid '%s'"), uuidstr);
2498
        goto cleanup;
2499 2500
    }

2501
    if (virDomainIsActive(vm)) {
2502
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT, "%s",
2503
                         _("cannot change vcpu count of an active domain"));
2504
        goto cleanup;
2505 2506 2507 2508 2509
    }

    if ((max = qemudDomainGetMaxVcpus(dom)) < 0) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR, "%s",
                         _("could not determine max vcpus for the domain"));
2510
        goto cleanup;
2511 2512 2513 2514 2515 2516
    }

    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);
2517
        goto cleanup;
2518 2519 2520
    }

    vm->def->vcpus = nvcpus;
2521 2522 2523
    ret = 0;

cleanup:
2524 2525
    if (vm)
        virDomainObjUnlock(vm);
2526
    return ret;
2527 2528
}

2529 2530 2531 2532 2533 2534 2535

#if HAVE_SCHED_GETAFFINITY
static int
qemudDomainPinVcpu(virDomainPtr dom,
                   unsigned int vcpu,
                   unsigned char *cpumap,
                   int maplen) {
2536 2537
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2538 2539 2540
    cpu_set_t mask;
    int i, maxcpu;
    virNodeInfo nodeinfo;
2541
    int ret = -1;
2542

2543
    qemuDriverLock(driver);
2544
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2545 2546
    qemuDriverUnlock(driver);

2547
    if (!virDomainIsActive(vm)) {
2548 2549
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                         "%s",_("cannot pin vcpus on an inactive domain"));
2550
        goto cleanup;
2551 2552 2553 2554 2555 2556
    }

    if (vcpu > (vm->nvcpupids-1)) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                         _("vcpu number out of range %d > %d"),
                         vcpu, vm->nvcpupids);
2557
        goto cleanup;
2558 2559 2560
    }

    if (virNodeInfoPopulate(dom->conn, &nodeinfo) < 0)
2561
        goto cleanup;
2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574

    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) {
2575 2576
            virReportSystemError(dom->conn, errno, "%s",
                                 _("cannot set affinity"));
2577
            goto cleanup;
2578 2579 2580 2581
        }
    } else {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("cpu affinity is not supported"));
2582
        goto cleanup;
2583
    }
2584
    ret = 0;
2585

2586
cleanup:
2587 2588
    if (vm)
        virDomainObjUnlock(vm);
2589
    return ret;
2590 2591 2592 2593 2594 2595 2596 2597
}

static int
qemudDomainGetVcpus(virDomainPtr dom,
                    virVcpuInfoPtr info,
                    int maxinfo,
                    unsigned char *cpumaps,
                    int maplen) {
2598 2599
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2600 2601
    virNodeInfo nodeinfo;
    int i, v, maxcpu;
2602
    int ret = -1;
2603

2604
    qemuDriverLock(driver);
2605
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2606 2607
    qemuDriverUnlock(driver);

2608
    if (!virDomainIsActive(vm)) {
2609 2610
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                         "%s",_("cannot pin vcpus on an inactive domain"));
2611
        goto cleanup;
2612 2613 2614
    }

    if (virNodeInfoPopulate(dom->conn, &nodeinfo) < 0)
2615
        goto cleanup;
2616 2617 2618 2619 2620 2621 2622 2623 2624

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

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

2625 2626 2627 2628 2629 2630 2631 2632
    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 */
            }
2633 2634
        }

2635 2636 2637 2638 2639 2640 2641 2642 2643
        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) {
2644 2645
                        virReportSystemError(dom->conn, errno, "%s",
                                             _("cannot get affinity"));
2646 2647 2648 2649 2650 2651
                        goto cleanup;
                    }

                    for (i = 0 ; i < maxcpu ; i++)
                        if (CPU_ISSET(i, &mask))
                            VIR_USE_CPU(cpumap, i);
2652
                }
2653 2654 2655 2656
            } else {
                qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                                 "%s", _("cpu affinity is not available"));
                goto cleanup;
2657 2658 2659
            }
        }
    }
2660
    ret = maxinfo;
2661

2662
cleanup:
2663 2664
    if (vm)
        virDomainObjUnlock(vm);
2665
    return ret;
2666 2667 2668 2669
}
#endif /* HAVE_SCHED_GETAFFINITY */


2670
static int qemudDomainGetMaxVcpus(virDomainPtr dom) {
2671 2672
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2673
    const char *type;
2674
    int ret = -1;
2675

2676
    qemuDriverLock(driver);
2677
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2678 2679
    qemuDriverUnlock(driver);

2680
    if (!vm) {
2681 2682 2683
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
2684
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2685
                         _("no domain with matching uuid '%s'"), uuidstr);
2686
        goto cleanup;
2687 2688
    }

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

2696
    ret = qemudGetMaxVCPUs(dom->conn, type);
2697

2698
cleanup:
2699 2700
    if (vm)
        virDomainObjUnlock(vm);
2701 2702 2703 2704
    return ret;
}


2705
static int qemudDomainRestore(virConnectPtr conn,
2706 2707 2708
                              const char *path) {
    struct qemud_driver *driver = conn->privateData;
    virDomainDefPtr def = NULL;
2709
    virDomainObjPtr vm = NULL;
2710 2711 2712
    int fd = -1;
    int ret = -1;
    char *xml = NULL;
2713
    struct qemud_save_header header;
2714
    virDomainEventPtr event = NULL;
2715

2716
    qemuDriverLock(driver);
2717 2718 2719
    /* Verify the header and read the XML */
    if ((fd = open(path, O_RDONLY)) < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2720
                         "%s", _("cannot read domain image"));
2721
        goto cleanup;
2722 2723 2724 2725
    }

    if (saferead(fd, &header, sizeof(header)) != sizeof(header)) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2726
                         "%s", _("failed to read qemu header"));
2727
        goto cleanup;
2728 2729 2730 2731
    }

    if (memcmp(header.magic, QEMUD_SAVE_MAGIC, sizeof(header.magic)) != 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2732
                         "%s", _("image magic is incorrect"));
2733
        goto cleanup;
2734 2735 2736 2737
    }

    if (header.version > QEMUD_SAVE_VERSION) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2738
                         _("image version is not supported (%d > %d)"),
2739
                         header.version, QEMUD_SAVE_VERSION);
2740
        goto cleanup;
2741 2742
    }

2743
    if (VIR_ALLOC_N(xml, header.xml_len) < 0) {
2744
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2745
                         "%s", _("out of memory"));
2746
        goto cleanup;
2747 2748 2749 2750
    }

    if (saferead(fd, xml, header.xml_len) != header.xml_len) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2751
                         "%s", _("failed to read XML"));
2752
        goto cleanup;
2753 2754 2755
    }

    /* Create a domain from this XML */
2756 2757
    if (!(def = virDomainDefParseString(conn, driver->caps, xml,
                                        VIR_DOMAIN_XML_INACTIVE))) {
2758
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2759
                         "%s", _("failed to parse XML"));
2760
        goto cleanup;
2761 2762 2763
    }

    /* Ensure the name and UUID don't already exist in an active VM */
2764
    vm = virDomainFindByUUID(&driver->domains, def->uuid);
2765
    if (!vm)
2766
        vm = virDomainFindByName(&driver->domains, def->name);
2767 2768 2769 2770 2771 2772 2773 2774
    if (vm) {
        if (virDomainIsActive(vm)) {
            qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                             _("domain is already active as '%s'"), vm->def->name);
            goto cleanup;
        } else {
            virDomainObjUnlock(vm);
        }
2775 2776
    }

2777 2778 2779
    if (!(vm = virDomainAssignDef(conn,
                                  &driver->domains,
                                  def))) {
2780
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2781
                         "%s", _("failed to assign new VM"));
2782
        goto cleanup;
2783
    }
2784
    def = NULL;
2785 2786

    /* Set the migration source and start it up. */
2787
    ret = qemudStartVMDaemon(conn, driver, vm, "stdio", fd);
2788
    close(fd);
2789
    fd = -1;
2790
    if (ret < 0) {
2791
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2792
                         "%s", _("failed to start VM"));
2793
        if (!vm->persistent) {
2794 2795
            virDomainRemoveInactive(&driver->domains,
                                    vm);
2796 2797
            vm = NULL;
        }
2798
        goto cleanup;
2799 2800
    }

2801 2802 2803
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_RESTORED);
2804

2805 2806 2807
    /* If it was running before, resume it now. */
    if (header.was_running) {
        char *info;
2808
        if (qemudMonitorCommand(vm, "cont", &info) < 0) {
2809
            qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2810
                             "%s", _("failed to resume domain"));
2811
            goto cleanup;
2812
        }
2813
        VIR_FREE(info);
2814 2815
        vm->state = VIR_DOMAIN_RUNNING;
    }
2816
    ret = 0;
2817

2818 2819 2820 2821 2822
cleanup:
    virDomainDefFree(def);
    VIR_FREE(xml);
    if (fd != -1)
        close(fd);
2823 2824
    if (vm)
        virDomainObjUnlock(vm);
2825 2826
    if (event)
        qemuDomainEventQueue(driver, event);
2827
    qemuDriverUnlock(driver);
2828
    return ret;
D
Daniel P. Berrange 已提交
2829 2830 2831
}


2832
static char *qemudDomainDumpXML(virDomainPtr dom,
2833
                                int flags ATTRIBUTE_UNUSED) {
2834 2835 2836 2837
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *ret = NULL;

2838
    qemuDriverLock(driver);
2839
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2840 2841
    qemuDriverUnlock(driver);

D
Daniel P. Berrange 已提交
2842
    if (!vm) {
2843 2844
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
2845
        goto cleanup;
D
Daniel P. Berrange 已提交
2846 2847
    }

2848 2849 2850 2851 2852 2853
    ret = virDomainDefFormat(dom->conn,
                             (flags & VIR_DOMAIN_XML_INACTIVE) && vm->newDef ?
                             vm->newDef : vm->def,
                             flags);

cleanup:
2854 2855
    if (vm)
        virDomainObjUnlock(vm);
2856
    return ret;
D
Daniel P. Berrange 已提交
2857 2858 2859
}


2860
static int qemudListDefinedDomains(virConnectPtr conn,
2861
                            char **const names, int nnames) {
2862
    struct qemud_driver *driver = conn->privateData;
2863
    int got = 0, i;
2864

2865
    qemuDriverLock(driver);
2866
    for (i = 0 ; i < driver->domains.count && got < nnames ; i++) {
2867
        virDomainObjLock(driver->domains.objs[i]);
2868 2869
        if (!virDomainIsActive(driver->domains.objs[i])) {
            if (!(names[got++] = strdup(driver->domains.objs[i]->def->name))) {
2870
                virReportOOMError(conn);
2871
                virDomainObjUnlock(driver->domains.objs[i]);
2872 2873
                goto cleanup;
            }
2874
        }
2875
        virDomainObjUnlock(driver->domains.objs[i]);
D
Daniel P. Berrange 已提交
2876
    }
2877

2878
    qemuDriverUnlock(driver);
D
Daniel P. Berrange 已提交
2879
    return got;
2880 2881 2882

 cleanup:
    for (i = 0 ; i < got ; i++)
2883
        VIR_FREE(names[i]);
2884
    qemuDriverUnlock(driver);
2885
    return -1;
D
Daniel P. Berrange 已提交
2886 2887
}

2888
static int qemudNumDefinedDomains(virConnectPtr conn) {
2889
    struct qemud_driver *driver = conn->privateData;
2890 2891
    int n = 0, i;

2892
    qemuDriverLock(driver);
2893 2894
    for (i = 0 ; i < driver->domains.count ; i++)
        if (!virDomainIsActive(driver->domains.objs[i]))
2895
            n++;
2896
    qemuDriverUnlock(driver);
2897

2898
    return n;
D
Daniel P. Berrange 已提交
2899 2900 2901
}


2902
static int qemudDomainStart(virDomainPtr dom) {
2903 2904 2905
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
2906
    virDomainEventPtr event = NULL;
2907

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

2912
    if (!vm) {
2913
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2914
                         "%s", _("no domain with matching uuid"));
2915
        goto cleanup;
2916 2917
    }

2918
    ret = qemudStartVMDaemon(dom->conn, driver, vm, NULL, -1);
2919
    if (ret != -1)
2920 2921 2922
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_STARTED,
                                         VIR_DOMAIN_EVENT_STARTED_BOOTED);
2923 2924

cleanup:
2925 2926
    if (vm)
        virDomainObjUnlock(vm);
2927 2928 2929 2930 2931
    if (event) {
        qemuDriverLock(driver);
        qemuDomainEventQueue(driver, event);
        qemuDriverUnlock(driver);
    }
2932
    return ret;
D
Daniel P. Berrange 已提交
2933 2934 2935
}


2936
static virDomainPtr qemudDomainDefine(virConnectPtr conn, const char *xml) {
2937
    struct qemud_driver *driver = conn->privateData;
2938
    virDomainDefPtr def;
2939
    virDomainObjPtr vm = NULL;
2940
    virDomainPtr dom = NULL;
2941
    virDomainEventPtr event = NULL;
2942
    int newVM = 1;
2943

2944
    qemuDriverLock(driver);
2945 2946
    if (!(def = virDomainDefParseString(conn, driver->caps, xml,
                                        VIR_DOMAIN_XML_INACTIVE)))
2947
        goto cleanup;
2948

2949
    vm = virDomainFindByName(&driver->domains, def->name);
2950 2951
    if (vm) {
        virDomainObjUnlock(vm);
2952
        newVM = 0;
2953
    }
2954

2955 2956 2957 2958
    if (!(vm = virDomainAssignDef(conn,
                                  &driver->domains,
                                  def))) {
        virDomainDefFree(def);
2959
        goto cleanup;
2960
    }
2961
    vm->persistent = 1;
2962

2963 2964
    if (virDomainSaveConfig(conn,
                            driver->configDir,
2965
                            vm->newDef ? vm->newDef : vm->def) < 0) {
2966 2967
        virDomainRemoveInactive(&driver->domains,
                                vm);
2968
        vm = NULL;
2969
        goto cleanup;
2970 2971
    }

2972 2973 2974 2975 2976
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_DEFINED,
                                     newVM ?
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED :
                                     VIR_DOMAIN_EVENT_DEFINED_UPDATED);
2977

2978
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
2979
    if (dom) dom->id = vm->def->id;
2980 2981

cleanup:
2982 2983
    if (vm)
        virDomainObjUnlock(vm);
2984 2985
    if (event)
        qemuDomainEventQueue(driver, event);
2986
    qemuDriverUnlock(driver);
2987
    return dom;
D
Daniel P. Berrange 已提交
2988 2989
}

2990
static int qemudDomainUndefine(virDomainPtr dom) {
2991 2992
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2993
    virDomainEventPtr event = NULL;
2994
    int ret = -1;
D
Daniel P. Berrange 已提交
2995

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

D
Daniel P. Berrange 已提交
2999
    if (!vm) {
3000 3001
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
3002
        goto cleanup;
D
Daniel P. Berrange 已提交
3003 3004
    }

3005
    if (virDomainIsActive(vm)) {
3006 3007
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot delete active domain"));
3008
        goto cleanup;
D
Daniel P. Berrange 已提交
3009 3010
    }

3011 3012 3013
    if (!vm->persistent) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot undefine transient domain"));
3014
        goto cleanup;
3015 3016 3017
    }

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

3020 3021 3022
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_UNDEFINED,
                                     VIR_DOMAIN_EVENT_UNDEFINED_REMOVED);
3023

3024 3025
    virDomainRemoveInactive(&driver->domains,
                            vm);
3026
    vm = NULL;
3027
    ret = 0;
D
Daniel P. Berrange 已提交
3028

3029
cleanup:
3030 3031
    if (vm)
        virDomainObjUnlock(vm);
3032 3033
    if (event)
        qemuDomainEventQueue(driver, event);
3034
    qemuDriverUnlock(driver);
3035
    return ret;
D
Daniel P. Berrange 已提交
3036 3037
}

3038
/* Return the disks name for use in monitor commands */
3039
static char *qemudDiskDeviceName(const virConnectPtr conn,
3040
                                 const virDomainDiskDefPtr disk) {
3041 3042 3043 3044 3045 3046

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

    if (virDiskNameToBusDeviceIndex(disk, &busid, &devid) < 0) {
3047
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
3048 3049 3050 3051 3052 3053 3054
                         _("cannot convert disk '%s' to bus/device index"),
                         disk->dst);
        return NULL;
    }

    switch (disk->bus) {
        case VIR_DOMAIN_DISK_BUS_IDE:
3055
            if (disk->device== VIR_DOMAIN_DISK_DEVICE_DISK)
3056
                ret = virAsprintf(&devname, "ide%d-hd%d", busid, devid);
3057
            else
3058
                ret = virAsprintf(&devname, "ide%d-cd%d", busid, devid);
3059 3060
            break;
        case VIR_DOMAIN_DISK_BUS_SCSI:
3061
            if (disk->device == VIR_DOMAIN_DISK_DEVICE_DISK)
3062
                ret = virAsprintf(&devname, "scsi%d-hd%d", busid, devid);
3063
            else
3064
                ret = virAsprintf(&devname, "scsi%d-cd%d", busid, devid);
3065 3066
            break;
        case VIR_DOMAIN_DISK_BUS_FDC:
3067
            ret = virAsprintf(&devname, "floppy%d", devid);
3068 3069
            break;
        case VIR_DOMAIN_DISK_BUS_VIRTIO:
3070
            ret = virAsprintf(&devname, "virtio%d", devid);
3071 3072
            break;
        default:
3073
            qemudReportError(conn, NULL, NULL, VIR_ERR_NO_SUPPORT,
3074 3075 3076 3077 3078 3079
                             _("Unsupported disk name mapping for bus '%s'"),
                             virDomainDiskBusTypeToString(disk->bus));
            return NULL;
    }

    if (ret == -1) {
3080
        virReportOOMError(conn);
3081 3082 3083 3084 3085 3086
        return NULL;
    }

    return devname;
}

3087 3088
static int qemudDomainChangeEjectableMedia(virConnectPtr conn,
                                           virDomainObjPtr vm,
3089 3090
                                           virDomainDeviceDefPtr dev)
{
3091
    virDomainDiskDefPtr origdisk = NULL, newdisk;
3092
    char *cmd, *reply, *safe_path;
3093
    char *devname = NULL;
3094
    unsigned int qemuCmdFlags;
3095
    int i;
3096

3097
    origdisk = NULL;
3098
    newdisk = dev->data.disk;
3099 3100 3101 3102
    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];
3103
            break;
3104
        }
3105 3106 3107
    }

    if (!origdisk) {
3108
        qemudReportError(conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
3109 3110 3111 3112 3113 3114 3115 3116 3117
                         _("No device with bus '%s' and target '%s'"),
                         virDomainDiskBusTypeToString(newdisk->bus),
                         newdisk->dst);
        return -1;
    }

    if (qemudExtractVersionInfo(vm->def->emulator,
                                NULL,
                                &qemuCmdFlags) < 0) {
3118
        qemudReportError(conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
3119 3120 3121 3122 3123 3124
                         _("Cannot determine QEMU argv syntax %s"),
                         vm->def->emulator);
        return -1;
    }

    if (qemuCmdFlags & QEMUD_CMD_FLAG_DRIVE) {
3125
        if (!(devname = qemudDiskDeviceName(conn, newdisk)))
3126 3127 3128 3129 3130 3131 3132 3133 3134
            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 {
3135
            qemudReportError(conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
3136 3137 3138 3139 3140 3141 3142 3143
                             _("Emulator version does not support removable "
                               "media for device '%s' and target '%s'"),
                               virDomainDiskDeviceTypeToString(newdisk->device),
                               newdisk->dst);
            return -1;
        }

        if (!devname) {
3144
            virReportOOMError(conn);
3145 3146 3147
            return -1;
        }
    }
3148

3149
    if (newdisk->src) {
3150 3151
        safe_path = qemudEscapeMonitorArg(newdisk->src);
        if (!safe_path) {
3152
            virReportOOMError(conn);
3153
            VIR_FREE(devname);
3154 3155
            return -1;
        }
3156
        if (virAsprintf(&cmd, "change %s \"%s\"", devname, safe_path) == -1) {
3157
            virReportOOMError(conn);
3158
            VIR_FREE(safe_path);
3159
            VIR_FREE(devname);
3160 3161
            return -1;
        }
3162
        VIR_FREE(safe_path);
3163

3164
    } else if (virAsprintf(&cmd, "eject %s", devname) == -1) {
3165
        virReportOOMError(conn);
3166
        VIR_FREE(devname);
3167 3168
        return -1;
    }
3169
    VIR_FREE(devname);
3170

3171
    if (qemudMonitorCommand(vm, cmd, &reply) < 0) {
3172
        qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
C
Cole Robinson 已提交
3173
                         "%s", _("could not change cdrom media"));
3174
        VIR_FREE(cmd);
3175 3176
        return -1;
    }
3177 3178 3179 3180

    /* If the command failed qemu prints:
     * device not found, device is locked ...
     * No message is printed on success it seems */
3181
    DEBUG ("ejectable media change reply: %s", reply);
3182
    if (strstr(reply, "\ndevice ")) {
3183
        qemudReportError (conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
C
Cole Robinson 已提交
3184
                          _("changing cdrom media failed: %s"), reply);
3185 3186 3187 3188
        VIR_FREE(reply);
        VIR_FREE(cmd);
        return -1;
    }
3189 3190
    VIR_FREE(reply);
    VIR_FREE(cmd);
3191

3192 3193
    VIR_FREE(origdisk->src);
    origdisk->src = newdisk->src;
3194
    newdisk->src = NULL;
3195
    origdisk->type = newdisk->type;
3196 3197 3198
    return 0;
}

3199 3200 3201
static int qemudDomainAttachPciDiskDevice(virConnectPtr conn,
                                          virDomainObjPtr vm,
                                          virDomainDeviceDefPtr dev)
3202 3203
{
    int ret, i;
3204
    char *cmd, *reply, *s;
3205 3206 3207 3208 3209
    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)) {
3210
            qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3211 3212 3213 3214 3215 3216
                           _("target %s already exists"), dev->data.disk->dst);
            return -1;
        }
    }

    if (VIR_REALLOC_N(vm->def->disks, vm->def->ndisks+1) < 0) {
3217
        virReportOOMError(conn);
3218 3219 3220 3221 3222
        return -1;
    }

    safe_path = qemudEscapeMonitorArg(dev->data.disk->src);
    if (!safe_path) {
3223
        qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3224 3225 3226 3227
                         "%s", _("out of memory"));
        return -1;
    }

3228 3229
    ret = virAsprintf(&cmd, "pci_add 0 storage file=%s,if=%s",
                      safe_path, type);
3230 3231
    VIR_FREE(safe_path);
    if (ret == -1) {
3232
        virReportOOMError(conn);
3233 3234 3235
        return ret;
    }

3236
    if (qemudMonitorCommand(vm, cmd, &reply) < 0) {
3237
        qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3238 3239 3240 3241 3242 3243 3244 3245
                         _("cannot attach %s disk"), type);
        VIR_FREE(cmd);
        return -1;
    }

    DEBUG ("pci_add reply: %s", reply);
    /* If the command succeeds qemu prints:
     * OK bus 0... */
3246 3247 3248 3249 3250 3251
#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)
J
Jim Meyering 已提交
3252
            qemudLog(QEMUD_WARN, "%s", _("Unable to parse slot number\n"));
3253
    } else {
3254
        qemudReportError (conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268
                          _("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;
}
3269

3270 3271 3272
static int qemudDomainAttachUsbMassstorageDevice(virConnectPtr conn,
                                                 virDomainObjPtr vm,
                                                 virDomainDeviceDefPtr dev)
3273
{
3274 3275
    int ret, i;
    char *safe_path;
3276 3277
    char *cmd, *reply;

3278 3279
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (STREQ(vm->def->disks[i]->dst, dev->data.disk->dst)) {
3280
            qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3281 3282 3283 3284 3285
                           _("target %s already exists"), dev->data.disk->dst);
            return -1;
        }
    }

3286
    if (VIR_REALLOC_N(vm->def->disks, vm->def->ndisks+1) < 0) {
3287
        virReportOOMError(conn);
3288
        return -1;
3289 3290
    }

3291 3292
    safe_path = qemudEscapeMonitorArg(dev->data.disk->src);
    if (!safe_path) {
3293
        qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3294 3295 3296 3297
                         "%s", _("out of memory"));
        return -1;
    }

3298
    ret = virAsprintf(&cmd, "usb_add disk:%s", safe_path);
3299
    VIR_FREE(safe_path);
3300
    if (ret == -1) {
3301
        virReportOOMError(conn);
3302 3303 3304
        return ret;
    }

3305
    if (qemudMonitorCommand(vm, cmd, &reply) < 0) {
3306
        qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3307
                         "%s", _("cannot attach usb disk"));
3308 3309 3310 3311 3312 3313 3314 3315
        VIR_FREE(cmd);
        return -1;
    }

    DEBUG ("attach_usb reply: %s", reply);
    /* If the command failed qemu prints:
     * Could not add ... */
    if (strstr(reply, "Could not add ")) {
3316
        qemudReportError (conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3317
                          "%s",
3318
                          _("adding usb disk failed"));
3319 3320 3321 3322
        VIR_FREE(reply);
        VIR_FREE(cmd);
        return -1;
    }
3323

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

3328 3329 3330 3331 3332
    VIR_FREE(reply);
    VIR_FREE(cmd);
    return 0;
}

3333 3334 3335
static int qemudDomainAttachHostDevice(virConnectPtr conn,
                                       virDomainObjPtr vm,
                                       virDomainDeviceDefPtr dev)
3336 3337 3338 3339
{
    int ret;
    char *cmd, *reply;

3340
    if (VIR_REALLOC_N(vm->def->hostdevs, vm->def->nhostdevs+1) < 0) {
3341
        virReportOOMError(conn);
3342 3343
        return -1;
    }
3344

3345
    if (dev->data.hostdev->source.subsys.u.usb.vendor) {
3346 3347 3348
        ret = virAsprintf(&cmd, "usb_add host:%.4x:%.4x",
                          dev->data.hostdev->source.subsys.u.usb.vendor,
                          dev->data.hostdev->source.subsys.u.usb.product);
3349
    } else {
3350 3351 3352
        ret = virAsprintf(&cmd, "usb_add host:%.3d.%.3d",
                          dev->data.hostdev->source.subsys.u.usb.bus,
                          dev->data.hostdev->source.subsys.u.usb.device);
3353 3354
    }
    if (ret == -1) {
3355
        virReportOOMError(conn);
3356 3357 3358
        return -1;
    }

3359
    if (qemudMonitorCommand(vm, cmd, &reply) < 0) {
3360
        qemudReportError(conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3361 3362 3363 3364 3365 3366 3367 3368 3369
                         "%s", _("cannot attach usb device"));
        VIR_FREE(cmd);
        return -1;
    }

    DEBUG ("attach_usb reply: %s", reply);
    /* If the command failed qemu prints:
     * Could not add ... */
    if (strstr(reply, "Could not add ")) {
3370
        qemudReportError (conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3371 3372 3373 3374 3375 3376
                          "%s",
                          _("adding usb device failed"));
        VIR_FREE(reply);
        VIR_FREE(cmd);
        return -1;
    }
3377

3378
    vm->def->hostdevs[vm->def->nhostdevs++] = dev->data.hostdev;
3379

3380 3381
    VIR_FREE(reply);
    VIR_FREE(cmd);
3382 3383 3384
    return 0;
}

3385 3386
static int qemudDomainAttachDevice(virDomainPtr dom,
                                   const char *xml) {
3387 3388 3389 3390
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    virDomainDeviceDefPtr dev = NULL;
    int ret = -1;
3391

3392
    qemuDriverLock(driver);
3393
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
3394
    if (!vm) {
3395
        qemuDriverUnlock(driver);
3396 3397
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
3398
        goto cleanup;
3399 3400 3401
    }

    if (!virDomainIsActive(vm)) {
3402
        qemuDriverUnlock(driver);
3403 3404
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot attach device on inactive domain"));
3405
        goto cleanup;
3406 3407
    }

3408 3409
    dev = virDomainDeviceDefParse(dom->conn, driver->caps, vm->def, xml,
                                  VIR_DOMAIN_XML_INACTIVE);
3410
    qemuDriverUnlock(driver);
3411 3412 3413
    if (dev == NULL)
        goto cleanup;

3414

3415 3416
    if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
        switch (dev->data.disk->device) {
3417 3418
        case VIR_DOMAIN_DISK_DEVICE_CDROM:
        case VIR_DOMAIN_DISK_DEVICE_FLOPPY:
3419
            ret = qemudDomainChangeEjectableMedia(dom->conn, vm, dev);
3420 3421 3422
            break;
        case VIR_DOMAIN_DISK_DEVICE_DISK:
            if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_USB) {
3423
                ret = qemudDomainAttachUsbMassstorageDevice(dom->conn, vm, dev);
3424 3425
            } else if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_SCSI ||
                       dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_VIRTIO) {
3426
                ret = qemudDomainAttachPciDiskDevice(dom->conn, vm, dev);
3427 3428 3429 3430 3431 3432
            }
            break;
        default:
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                             "%s", _("this disk device type cannot be attached"));
            goto cleanup;
3433
        }
3434
    } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV &&
3435 3436
               dev->data.hostdev->mode == VIR_DOMAIN_HOSTDEV_MODE_SUBSYS &&
               dev->data.hostdev->source.subsys.type == VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB) {
3437
        ret = qemudDomainAttachHostDevice(dom->conn, vm, dev);
3438
    } else {
3439
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
3440
                         "%s", _("this device type cannot be attached"));
3441
        goto cleanup;
3442 3443
    }

3444
    qemudSaveDomainStatus(dom->conn, driver, vm);
3445
cleanup:
G
Guido Günther 已提交
3446 3447
    if (ret < 0)
        virDomainDeviceDefFree(dev);
3448 3449
    if (vm)
        virDomainObjUnlock(vm);
3450 3451 3452
    return ret;
}

3453 3454
static int qemudDomainDetachPciDiskDevice(virConnectPtr conn,
                                          virDomainObjPtr vm, virDomainDeviceDefPtr dev)
3455 3456
{
    int i, ret = -1;
3457 3458
    char *cmd = NULL;
    char *reply = NULL;
3459 3460 3461 3462 3463 3464 3465 3466 3467 3468
    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) {
3469
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
3470
                         _("disk %s not found"), dev->data.disk->dst);
3471
        goto cleanup;
3472 3473 3474
    }

    if (detach->slotnum < 1) {
3475
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
3476
                         _("disk %s cannot be detached - invalid slot number %d"),
3477
                           detach->dst, detach->slotnum);
3478
        goto cleanup;
3479 3480
    }

3481
    if (virAsprintf(&cmd, "pci_del 0 %d", detach->slotnum) < 0) {
3482
        virReportOOMError(conn);
3483
        goto cleanup;
3484 3485
    }

3486
    if (qemudMonitorCommand(vm, cmd, &reply) < 0) {
3487
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
3488
                          _("failed to execute detach disk %s command"), detach->dst);
3489
        goto cleanup;
3490 3491 3492 3493 3494 3495
    }

    DEBUG ("pci_del reply: %s", reply);
    /* If the command fails due to a wrong slot qemu prints: invalid slot,
     * nothing is printed on success */
    if (strstr(reply, "invalid slot")) {
3496
        qemudReportError (conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
3497
                          _("failed to detach disk %s: invalid slot %d"),
3498 3499
                          detach->dst, detach->slotnum);
        goto cleanup;
3500 3501 3502 3503 3504
    }

    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) {
3505
            virReportOOMError(conn);
3506
            goto cleanup;
3507 3508 3509 3510 3511 3512 3513 3514
        }
        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;
3515 3516

cleanup:
3517 3518 3519 3520 3521 3522 3523
    VIR_FREE(reply);
    VIR_FREE(cmd);
    return ret;
}

static int qemudDomainDetachDevice(virDomainPtr dom,
                                   const char *xml) {
3524 3525 3526 3527
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    virDomainDeviceDefPtr dev = NULL;
    int ret = -1;
3528

3529
    qemuDriverLock(driver);
3530
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
3531
    if (!vm) {
3532
        qemuDriverUnlock(driver);
3533 3534
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
3535
        goto cleanup;
3536 3537 3538
    }

    if (!virDomainIsActive(vm)) {
3539
        qemuDriverUnlock(driver);
3540
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
G
Guido Günther 已提交
3541
                         "%s", _("cannot detach device on inactive domain"));
3542
        goto cleanup;
3543 3544
    }

3545 3546
    dev = virDomainDeviceDefParse(dom->conn, driver->caps, vm->def, xml,
                                  VIR_DOMAIN_XML_INACTIVE);
3547
    qemuDriverUnlock(driver);
3548 3549 3550
    if (dev == NULL)
        goto cleanup;

3551 3552 3553 3554 3555

    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 ||
         dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_VIRTIO))
3556
        ret = qemudDomainDetachPciDiskDevice(dom->conn, vm, dev);
3557
    else
3558 3559 3560
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("only SCSI or virtio disk device can be detached dynamically"));

3561
    qemudSaveDomainStatus(dom->conn, driver, vm);
3562 3563
cleanup:
    virDomainDeviceDefFree(dev);
3564 3565
    if (vm)
        virDomainObjUnlock(vm);
3566 3567 3568
    return ret;
}

3569
static int qemudDomainGetAutostart(virDomainPtr dom,
3570
                                   int *autostart) {
3571 3572 3573
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
3574

3575
    qemuDriverLock(driver);
3576
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
3577 3578
    qemuDriverUnlock(driver);

3579
    if (!vm) {
3580 3581
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
3582
        goto cleanup;
3583 3584 3585
    }

    *autostart = vm->autostart;
3586
    ret = 0;
3587

3588
cleanup:
3589 3590
    if (vm)
        virDomainObjUnlock(vm);
3591
    return ret;
3592 3593
}

3594
static int qemudDomainSetAutostart(virDomainPtr dom,
3595
                                   int autostart) {
3596 3597
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
3598 3599
    char *configFile = NULL, *autostartLink = NULL;
    int ret = -1;
3600

3601
    qemuDriverLock(driver);
3602
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
3603 3604
    qemuDriverUnlock(driver);

3605
    if (!vm) {
3606 3607
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
3608
        goto cleanup;
3609 3610
    }

3611 3612 3613
    if (!vm->persistent) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot set autostart for transient domain"));
3614
        goto cleanup;
3615 3616
    }

3617 3618
    autostart = (autostart != 0);

3619 3620 3621 3622 3623
    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;
3624

3625 3626
        if (autostart) {
            int err;
3627

3628
            if ((err = virFileMakePath(driver->autostartDir))) {
3629 3630 3631
                virReportSystemError(dom->conn, err,
                                     _("cannot create autostart directory %s"),
                                     driver->autostartDir);
3632 3633
                goto cleanup;
            }
3634

3635
            if (symlink(configFile, autostartLink) < 0) {
3636 3637 3638
                virReportSystemError(dom->conn, errno,
                                     _("Failed to create symlink '%s to '%s'"),
                                     autostartLink, configFile);
3639 3640 3641 3642
                goto cleanup;
            }
        } else {
            if (unlink(autostartLink) < 0 && errno != ENOENT && errno != ENOTDIR) {
3643 3644 3645
                virReportSystemError(dom->conn, errno,
                                     _("Failed to delete symlink '%s'"),
                                     autostartLink);
3646 3647
                goto cleanup;
            }
3648 3649
        }

3650
        vm->autostart = autostart;
3651
    }
3652
    ret = 0;
3653

3654 3655 3656
cleanup:
    VIR_FREE(configFile);
    VIR_FREE(autostartLink);
3657 3658
    if (vm)
        virDomainObjUnlock(vm);
3659
    return ret;
3660 3661
}

3662 3663 3664 3665 3666 3667 3668 3669 3670
/* 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)
{
3671
    struct qemud_driver *driver = dom->conn->privateData;
3672
    char *dummy, *info = NULL;
3673
    const char *p, *eol;
3674
    const char *qemu_dev_name = NULL;
3675
    size_t len;
3676
    int i, ret = -1;
3677
    virDomainObjPtr vm;
3678
    virDomainDiskDefPtr disk = NULL;
3679

3680
    qemuDriverLock(driver);
3681
    vm = virDomainFindByID(&driver->domains, dom->id);
3682
    qemuDriverUnlock(driver);
3683 3684 3685
    if (!vm) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                          _("no domain with matching id %d"), dom->id);
3686
        goto cleanup;
3687
    }
3688
    if (!virDomainIsActive (vm)) {
3689 3690
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("domain is not running"));
3691
        goto cleanup;
3692 3693
    }

3694 3695 3696 3697 3698 3699 3700 3701
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (STREQ(path, vm->def->disks[i]->dst)) {
            disk = vm->def->disks[i];
            break;
        }
    }

    if (!disk) {
3702 3703
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                          _("invalid path: %s"), path);
3704
        goto cleanup;
3705 3706
    }

3707
    qemu_dev_name = qemudDiskDeviceName(dom->conn, disk);
3708
    if (!qemu_dev_name)
3709
        goto cleanup;
3710 3711
    len = strlen (qemu_dev_name);

3712
    if (qemudMonitorCommand (vm, "info blockstats", &info) < 0) {
3713 3714
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("'info blockstats' command failed"));
3715
        goto cleanup;
3716 3717 3718 3719 3720 3721 3722 3723
    }
    DEBUG ("info blockstats reply: %s", info);

    /* 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 已提交
3724
    if (strstr(info, "\ninfo ")) {
3725 3726 3727
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                          "%s",
                          _("'info blockstats' not supported by this qemu"));
3728
        goto cleanup;
3729 3730 3731 3732 3733 3734 3735 3736 3737 3738 3739 3740 3741 3742 3743 3744 3745 3746 3747 3748 3749 3750 3751 3752 3753 3754
    }

    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) {
3755
                if (STRPREFIX (p, "rd_bytes=")) {
3756 3757 3758
                    p += 9;
                    if (virStrToLong_ll (p, &dummy, 10, &stats->rd_bytes) == -1)
                        DEBUG ("error reading rd_bytes: %s", p);
3759
                } else if (STRPREFIX (p, "wr_bytes=")) {
3760 3761 3762
                    p += 9;
                    if (virStrToLong_ll (p, &dummy, 10, &stats->wr_bytes) == -1)
                        DEBUG ("error reading wr_bytes: %s", p);
3763
                } else if (STRPREFIX (p, "rd_operations=")) {
3764 3765 3766
                    p += 14;
                    if (virStrToLong_ll (p, &dummy, 10, &stats->rd_req) == -1)
                        DEBUG ("error reading rd_req: %s", p);
3767
                } else if (STRPREFIX (p, "wr_operations=")) {
3768 3769 3770 3771 3772 3773 3774 3775 3776 3777 3778
                    p += 14;
                    if (virStrToLong_ll (p, &dummy, 10, &stats->wr_req) == -1)
                        DEBUG ("error reading wr_req: %s", p);
                } else
                    DEBUG ("unknown block stat near %s", p);

                /* Skip to next label. */
                p = strchr (p, ' ');
                if (!p || p >= eol) break;
                p++;
            }
3779
            ret = 0;
3780
            goto cleanup;
3781 3782 3783 3784 3785 3786 3787 3788 3789 3790 3791
        }

        /* 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);
3792
 cleanup:
3793 3794
    VIR_FREE(qemu_dev_name);
    VIR_FREE(info);
3795 3796
    if (vm)
        virDomainObjUnlock(vm);
3797
    return ret;
3798 3799
}

3800
#ifdef __linux__
3801 3802 3803 3804 3805
static int
qemudDomainInterfaceStats (virDomainPtr dom,
                           const char *path,
                           struct _virDomainInterfaceStats *stats)
{
3806 3807
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
3808
    int i;
3809
    int ret = -1;
3810

3811
    qemuDriverLock(driver);
3812
    vm = virDomainFindByID(&driver->domains, dom->id);
3813 3814
    qemuDriverUnlock(driver);

3815 3816
    if (!vm) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
3817
                          _("no domain with matching id %d"), dom->id);
3818
        goto cleanup;
3819 3820
    }

3821
    if (!virDomainIsActive(vm)) {
3822
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3823
                         "%s", _("domain is not running"));
3824
        goto cleanup;
3825 3826 3827 3828
    }

    if (!path || path[0] == '\0') {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
3829
                         "%s", _("NULL or empty path"));
3830
        goto cleanup;
3831 3832 3833
    }

    /* Check the path is one of the domain's network interfaces. */
3834 3835
    for (i = 0 ; i < vm->def->nnets ; i++) {
        if (vm->def->nets[i]->ifname &&
3836 3837 3838 3839
            STREQ (vm->def->nets[i]->ifname, path)) {
            ret = 0;
            break;
        }
3840 3841
    }

3842 3843 3844 3845 3846
    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);
3847

3848
cleanup:
3849 3850
    if (vm)
        virDomainObjUnlock(vm);
3851 3852
    return ret;
}
3853
#else
3854 3855 3856 3857
static int
qemudDomainInterfaceStats (virDomainPtr dom,
                           const char *path ATTRIBUTE_UNUSED,
                           struct _virDomainInterfaceStats *stats ATTRIBUTE_UNUSED)
3858 3859 3860 3861
    qemudReportError (dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                      "%s", __FUNCTION__);
    return -1;
}
3862
#endif
3863

3864 3865 3866 3867 3868 3869 3870
static int
qemudDomainBlockPeek (virDomainPtr dom,
                      const char *path,
                      unsigned long long offset, size_t size,
                      void *buffer,
                      unsigned int flags ATTRIBUTE_UNUSED)
{
3871 3872 3873
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int fd = -1, ret = -1, i;
3874

3875
    qemuDriverLock(driver);
3876
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
3877 3878
    qemuDriverUnlock(driver);

3879 3880
    if (!vm) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
J
Jim Meyering 已提交
3881
                          "%s", _("no domain with matching uuid"));
3882
        goto cleanup;
3883 3884 3885 3886
    }

    if (!path || path[0] == '\0') {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
J
Jim Meyering 已提交
3887
                         "%s", _("NULL or empty path"));
3888
        goto cleanup;
3889 3890 3891
    }

    /* Check the path belongs to this domain. */
3892 3893
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (vm->def->disks[i]->src != NULL &&
3894 3895 3896 3897
            STREQ (vm->def->disks[i]->src, path)) {
            ret = 0;
            break;
        }
3898 3899
    }

3900 3901 3902 3903 3904
    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) {
3905 3906
            virReportSystemError (dom->conn, errno,
                                  _("%s: failed to open"), path);
3907 3908
            goto cleanup;
        }
3909

3910 3911 3912 3913 3914 3915
        /* 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) {
3916 3917
            virReportSystemError (dom->conn, errno,
                                  _("%s: failed to seek or read"), path);
3918 3919 3920 3921 3922 3923 3924
            goto cleanup;
        }

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

3927 3928 3929
cleanup:
    if (fd >= 0)
        close (fd);
3930 3931
    if (vm)
        virDomainObjUnlock(vm);
3932 3933 3934
    return ret;
}

R
Richard W.M. Jones 已提交
3935 3936 3937 3938 3939 3940
static int
qemudDomainMemoryPeek (virDomainPtr dom,
                       unsigned long long offset, size_t size,
                       void *buffer,
                       unsigned int flags)
{
3941 3942 3943
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char cmd[256], *info = NULL;
R
Richard W.M. Jones 已提交
3944 3945 3946
    char tmp[] = TEMPDIR "/qemu.mem.XXXXXX";
    int fd = -1, ret = -1;

3947
    qemuDriverLock(driver);
3948
    vm = virDomainFindByID(&driver->domains, dom->id);
3949
    qemuDriverUnlock(driver);
R
Richard W.M. Jones 已提交
3950 3951 3952 3953

    if (!vm) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                          _("no domain with matching id %d"), dom->id);
3954 3955 3956 3957 3958 3959 3960
        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 已提交
3961 3962
    }

3963
    if (!virDomainIsActive(vm)) {
R
Richard W.M. Jones 已提交
3964 3965
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("domain is not running"));
3966
        goto cleanup;
R
Richard W.M. Jones 已提交
3967 3968 3969 3970
    }

    /* Create a temporary filename. */
    if ((fd = mkstemp (tmp)) == -1) {
3971 3972
        virReportSystemError (dom->conn, errno,
                              _("mkstemp(\"%s\") failed"), tmp);
3973
        goto cleanup;
R
Richard W.M. Jones 已提交
3974 3975 3976 3977
    }

    /* Issue the memsave command. */
    snprintf (cmd, sizeof cmd, "memsave %llu %zi \"%s\"", offset, size, tmp);
3978
    if (qemudMonitorCommand (vm, cmd, &info) < 0) {
R
Richard W.M. Jones 已提交
3979
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3980
                          "%s", _("'memsave' command failed"));
3981
        goto cleanup;
R
Richard W.M. Jones 已提交
3982 3983 3984 3985 3986 3987
    }

    DEBUG ("memsave reply: %s", info);

    /* Read the memory file into buffer. */
    if (saferead (fd, buffer, size) == (ssize_t) -1) {
3988 3989 3990
        virReportSystemError (dom->conn, errno,
                              _("failed to read temporary file "
                                "created with template %s"), tmp);
3991
        goto cleanup;
R
Richard W.M. Jones 已提交
3992 3993 3994
    }

    ret = 0;
3995 3996 3997

cleanup:
    VIR_FREE(info);
R
Richard W.M. Jones 已提交
3998 3999
    if (fd >= 0) close (fd);
    unlink (tmp);
4000 4001
    if (vm)
        virDomainObjUnlock(vm);
R
Richard W.M. Jones 已提交
4002 4003 4004
    return ret;
}

4005

4006 4007
static int
qemudDomainEventRegister (virConnectPtr conn,
4008
                          virConnectDomainEventCallback callback,
4009 4010
                          void *opaque,
                          virFreeCallback freecb)
4011
{
4012 4013 4014
    struct qemud_driver *driver = conn->privateData;
    int ret;

4015
    qemuDriverLock(driver);
4016 4017
    ret = virDomainEventCallbackListAdd(conn, driver->domainEventCallbacks,
                                        callback, opaque, freecb);
4018
    qemuDriverUnlock(driver);
4019

4020
    return ret;
4021 4022 4023 4024
}

static int
qemudDomainEventDeregister (virConnectPtr conn,
4025
                            virConnectDomainEventCallback callback)
4026
{
4027 4028 4029
    struct qemud_driver *driver = conn->privateData;
    int ret;

4030
    qemuDriverLock(driver);
4031 4032 4033 4034 4035 4036
    if (driver->domainEventDispatching)
        ret = virDomainEventCallbackListMarkDelete(conn, driver->domainEventCallbacks,
                                                   callback);
    else
        ret = virDomainEventCallbackListRemove(conn, driver->domainEventCallbacks,
                                               callback);
4037
    qemuDriverUnlock(driver);
4038

4039
    return ret;
4040 4041
}

4042 4043 4044 4045 4046
static void qemuDomainEventDispatchFunc(virConnectPtr conn,
                                        virDomainEventPtr event,
                                        virConnectDomainEventCallback cb,
                                        void *cbopaque,
                                        void *opaque)
4047
{
4048
    struct qemud_driver *driver = opaque;
4049

4050 4051 4052 4053 4054 4055 4056 4057 4058 4059 4060 4061 4062 4063 4064 4065 4066 4067 4068 4069 4070 4071 4072 4073 4074 4075 4076 4077 4078 4079 4080 4081 4082 4083 4084 4085 4086 4087 4088 4089 4090 4091 4092 4093
    /* 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);
4094 4095
}

D
Daniel Veillard 已提交
4096 4097 4098 4099 4100 4101 4102 4103 4104 4105 4106 4107 4108 4109 4110 4111 4112 4113
/* 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;
4114 4115
    struct qemud_driver *driver = dconn->privateData;
    virDomainDefPtr def = NULL;
D
Daniel Veillard 已提交
4116 4117 4118 4119 4120
    virDomainObjPtr vm = NULL;
    int this_port;
    char hostname [HOST_NAME_MAX+1];
    char migrateFrom [64];
    const char *p;
4121
    virDomainEventPtr event = NULL;
4122 4123 4124
    int ret = -1;;

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

4126
    qemuDriverLock(driver);
D
Daniel Veillard 已提交
4127 4128 4129
    if (!dom_xml) {
        qemudReportError (dconn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                          "%s", _("no domain XML passed"));
4130
        goto cleanup;
D
Daniel Veillard 已提交
4131 4132 4133 4134 4135 4136 4137 4138 4139 4140 4141 4142 4143 4144 4145 4146 4147 4148
    }

    /* 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) {
4149 4150
            virReportSystemError (dconn, errno,
                                  "%s", _("failed to determine host name"));
4151
            goto cleanup;
D
Daniel Veillard 已提交
4152 4153 4154
        }

        /* Caller frees */
4155
        if (virAsprintf(uri_out, "tcp:%s:%d", hostname, this_port) < 0) {
4156
            virReportOOMError (dconn);
4157
            goto cleanup;
D
Daniel Veillard 已提交
4158 4159 4160 4161 4162 4163 4164 4165 4166
        }
    } 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.
         */
        if (!STREQLEN (uri_in, "tcp:", 6)) {
            qemudReportError (dconn, NULL, NULL, VIR_ERR_INVALID_ARG,
                  "%s", _("only tcp URIs are supported for KVM migrations"));
4167
            goto cleanup;
D
Daniel Veillard 已提交
4168 4169 4170 4171 4172 4173 4174 4175 4176
        }

        /* 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"));
4177
            goto cleanup;
D
Daniel Veillard 已提交
4178 4179 4180 4181
        }
    }

    /* Parse the domain XML. */
4182 4183
    if (!(def = virDomainDefParseString(dconn, driver->caps, dom_xml,
                                        VIR_DOMAIN_XML_INACTIVE))) {
D
Daniel Veillard 已提交
4184 4185
        qemudReportError (dconn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("failed to parse XML"));
4186
        goto cleanup;
D
Daniel Veillard 已提交
4187 4188 4189 4190 4191 4192 4193 4194 4195 4196 4197 4198 4199 4200 4201 4202
    }

    /* 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"));
4203
        goto cleanup;
D
Daniel Veillard 已提交
4204 4205 4206 4207 4208 4209 4210 4211 4212
    }
#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);
4213
            goto cleanup;
D
Daniel Veillard 已提交
4214 4215 4216 4217 4218 4219 4220 4221
        }
    }

    if (!(vm = virDomainAssignDef(dconn,
                                  &driver->domains,
                                  def))) {
        qemudReportError (dconn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("failed to assign new VM"));
4222
        goto cleanup;
D
Daniel Veillard 已提交
4223
    }
4224
    def = NULL;
D
Daniel Veillard 已提交
4225 4226 4227 4228 4229 4230 4231 4232

    /* 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);
4233
    if (qemudStartVMDaemon (dconn, driver, vm, migrateFrom, -1) < 0) {
D
Daniel Veillard 已提交
4234 4235
        qemudReportError (dconn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("failed to start listening VM"));
4236
        if (!vm->persistent) {
D
Daniel Veillard 已提交
4237
            virDomainRemoveInactive(&driver->domains, vm);
4238 4239
            vm = NULL;
        }
4240
        goto cleanup;
D
Daniel Veillard 已提交
4241
    }
4242 4243 4244 4245

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

4248 4249 4250 4251 4252
cleanup:
    virDomainDefFree(def);
    if (ret != 0) {
        VIR_FREE(*uri_out);
    }
4253 4254
    if (vm)
        virDomainObjUnlock(vm);
4255 4256
    if (event)
        qemuDomainEventQueue(driver, event);
4257
    qemuDriverUnlock(driver);
4258
    return ret;
D
Daniel Veillard 已提交
4259 4260 4261 4262 4263 4264 4265 4266 4267 4268 4269 4270
}

/* 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)
{
4271 4272
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
4273
    virDomainEventPtr event = NULL;
D
Daniel Veillard 已提交
4274 4275
    char *safe_uri;
    char cmd[HOST_NAME_MAX+50];
4276 4277
    char *info = NULL;
    int ret = -1;
D
Daniel Veillard 已提交
4278

4279
    qemuDriverLock(driver);
4280
    vm = virDomainFindByID(&driver->domains, dom->id);
D
Daniel Veillard 已提交
4281 4282 4283
    if (!vm) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                          _("no domain with matching id %d"), dom->id);
4284
        goto cleanup;
D
Daniel Veillard 已提交
4285 4286 4287 4288 4289
    }

    if (!virDomainIsActive(vm)) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("domain is not running"));
4290
        goto cleanup;
D
Daniel Veillard 已提交
4291 4292
    }

4293 4294 4295
    if (!(flags & VIR_MIGRATE_LIVE)) {
        /* Pause domain for non-live migration */
        snprintf(cmd, sizeof cmd, "%s", "stop");
4296
        qemudMonitorCommand (vm, cmd, &info);
4297 4298 4299
        DEBUG ("stop reply: %s", info);
        VIR_FREE(info);

4300 4301 4302 4303 4304 4305
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_SUSPENDED,
                                         VIR_DOMAIN_EVENT_SUSPENDED_MIGRATED);
        if (event)
            qemuDomainEventQueue(driver, event);
        event = NULL;
4306 4307
    }

D
Daniel Veillard 已提交
4308 4309 4310
    if (resource > 0) {
        /* Issue migrate_set_speed command.  Don't worry if it fails. */
        snprintf (cmd, sizeof cmd, "migrate_set_speed %lum", resource);
4311
        qemudMonitorCommand (vm, cmd, &info);
D
Daniel Veillard 已提交
4312 4313 4314 4315 4316 4317 4318 4319

        DEBUG ("migrate_set_speed reply: %s", info);
        VIR_FREE (info);
    }

    /* Issue the migrate command. */
    safe_uri = qemudEscapeMonitorArg (uri);
    if (!safe_uri) {
4320
        virReportOOMError (dom->conn);
4321
        goto cleanup;
D
Daniel Veillard 已提交
4322 4323 4324 4325
    }
    snprintf (cmd, sizeof cmd, "migrate \"%s\"", safe_uri);
    VIR_FREE (safe_uri);

4326
    if (qemudMonitorCommand (vm, cmd, &info) < 0) {
D
Daniel Veillard 已提交
4327 4328
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("migrate operation failed"));
4329
        goto cleanup;
D
Daniel Veillard 已提交
4330 4331 4332 4333 4334 4335 4336 4337
    }

    DEBUG ("migrate reply: %s", info);

    /* 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);
4338
        goto cleanup;
D
Daniel Veillard 已提交
4339 4340 4341 4342
    }

    /* Clean up the source domain. */
    qemudShutdownVMDaemon (dom->conn, driver, vm);
4343 4344 4345 4346

    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_MIGRATED);
4347
    if (!vm->persistent) {
D
Daniel Veillard 已提交
4348
        virDomainRemoveInactive(&driver->domains, vm);
4349 4350
        vm = NULL;
    }
4351
    ret = 0;
D
Daniel Veillard 已提交
4352

4353 4354
cleanup:
    VIR_FREE(info);
4355 4356
    if (vm)
        virDomainObjUnlock(vm);
4357 4358
    if (event)
        qemuDomainEventQueue(driver, event);
4359
    qemuDriverUnlock(driver);
4360
    return ret;
D
Daniel Veillard 已提交
4361 4362 4363 4364 4365 4366 4367 4368 4369 4370 4371 4372
}

/* 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)
{
4373 4374 4375
    struct qemud_driver *driver = dconn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
4376
    virDomainEventPtr event = NULL;
D
Daniel Veillard 已提交
4377 4378
    char *info = NULL;

4379
    qemuDriverLock(driver);
4380
    vm = virDomainFindByName(&driver->domains, dname);
D
Daniel Veillard 已提交
4381 4382 4383
    if (!vm) {
        qemudReportError (dconn, NULL, NULL, VIR_ERR_INVALID_DOMAIN,
                          _("no domain with matching name %s"), dname);
4384
        goto cleanup;
D
Daniel Veillard 已提交
4385 4386 4387 4388 4389 4390 4391 4392 4393
    }

    /* 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;
4394 4395 4396
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_RESUMED,
                                         VIR_DOMAIN_EVENT_RESUMED_MIGRATED);
D
Daniel Veillard 已提交
4397 4398
    } else {
        qemudShutdownVMDaemon (dconn, driver, vm);
4399 4400 4401
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_FAILED);
4402
        if (!vm->persistent) {
D
Daniel Veillard 已提交
4403
            virDomainRemoveInactive(&driver->domains, vm);
4404 4405
            vm = NULL;
        }
D
Daniel Veillard 已提交
4406
    }
4407 4408

cleanup:
4409 4410
    if (vm)
        virDomainObjUnlock(vm);
4411 4412
    if (event)
        qemuDomainEventQueue(driver, event);
4413
    qemuDriverUnlock(driver);
4414
    return dom;
D
Daniel Veillard 已提交
4415 4416
}

4417 4418 4419 4420 4421
static virDriver qemuDriver = {
    VIR_DRV_QEMU,
    "QEMU",
    qemudOpen, /* open */
    qemudClose, /* close */
D
Daniel Veillard 已提交
4422
    qemudSupportsFeature, /* supports_feature */
4423 4424
    qemudGetType, /* type */
    qemudGetVersion, /* version */
4425
    qemudGetHostname, /* hostname */
4426
    NULL, /* URI  */
4427 4428 4429 4430 4431
    qemudGetMaxVCPUs, /* getMaxVcpus */
    qemudGetNodeInfo, /* nodeGetInfo */
    qemudGetCapabilities, /* getCapabilities */
    qemudListDomains, /* listDomains */
    qemudNumDomains, /* numOfDomains */
4432
    qemudDomainCreate, /* domainCreateXML */
4433 4434 4435 4436 4437
    qemudDomainLookupByID, /* domainLookupByID */
    qemudDomainLookupByUUID, /* domainLookupByUUID */
    qemudDomainLookupByName, /* domainLookupByName */
    qemudDomainSuspend, /* domainSuspend */
    qemudDomainResume, /* domainResume */
4438
    qemudDomainShutdown, /* domainShutdown */
4439 4440 4441
    NULL, /* domainReboot */
    qemudDomainDestroy, /* domainDestroy */
    qemudDomainGetOSType, /* domainGetOSType */
4442 4443 4444
    qemudDomainGetMaxMemory, /* domainGetMaxMemory */
    qemudDomainSetMaxMemory, /* domainSetMaxMemory */
    qemudDomainSetMemory, /* domainSetMemory */
4445 4446 4447 4448
    qemudDomainGetInfo, /* domainGetInfo */
    qemudDomainSave, /* domainSave */
    qemudDomainRestore, /* domainRestore */
    NULL, /* domainCoreDump */
4449
    qemudDomainSetVcpus, /* domainSetVcpus */
4450 4451 4452 4453
#if HAVE_SCHED_GETAFFINITY
    qemudDomainPinVcpu, /* domainPinVcpu */
    qemudDomainGetVcpus, /* domainGetVcpus */
#else
4454 4455
    NULL, /* domainPinVcpu */
    NULL, /* domainGetVcpus */
4456
#endif
4457
    qemudDomainGetMaxVcpus, /* domainGetMaxVcpus */
4458 4459 4460 4461 4462 4463
    qemudDomainDumpXML, /* domainDumpXML */
    qemudListDefinedDomains, /* listDomains */
    qemudNumDefinedDomains, /* numOfDomains */
    qemudDomainStart, /* domainCreate */
    qemudDomainDefine, /* domainDefineXML */
    qemudDomainUndefine, /* domainUndefine */
4464
    qemudDomainAttachDevice, /* domainAttachDevice */
4465
    qemudDomainDetachDevice, /* domainDetachDevice */
4466 4467 4468 4469 4470
    qemudDomainGetAutostart, /* domainGetAutostart */
    qemudDomainSetAutostart, /* domainSetAutostart */
    NULL, /* domainGetSchedulerType */
    NULL, /* domainGetSchedulerParameters */
    NULL, /* domainSetSchedulerParameters */
D
Daniel Veillard 已提交
4471 4472
    NULL, /* domainMigratePrepare (v1) */
    qemudDomainMigratePerform, /* domainMigratePerform */
4473
    NULL, /* domainMigrateFinish */
4474
    qemudDomainBlockStats, /* domainBlockStats */
4475
    qemudDomainInterfaceStats, /* domainInterfaceStats */
4476
    qemudDomainBlockPeek, /* domainBlockPeek */
R
Richard W.M. Jones 已提交
4477
    qemudDomainMemoryPeek, /* domainMemoryPeek */
4478 4479 4480 4481
#if HAVE_NUMACTL
    qemudNodeGetCellsFreeMemory, /* nodeGetCellsFreeMemory */
    qemudNodeGetFreeMemory,  /* getFreeMemory */
#else
4482
    NULL, /* nodeGetCellsFreeMemory */
4483
    NULL, /* getFreeMemory */
4484
#endif
4485 4486
    qemudDomainEventRegister, /* domainEventRegister */
    qemudDomainEventDeregister, /* domainEventDeregister */
D
Daniel Veillard 已提交
4487 4488
    qemudDomainMigratePrepare2, /* domainMigratePrepare2 */
    qemudDomainMigrateFinish2, /* domainMigrateFinish2 */
4489 4490 4491
};


4492
static virStateDriver qemuStateDriver = {
4493 4494 4495 4496
    .initialize = qemudStartup,
    .cleanup = qemudShutdown,
    .reload = qemudReload,
    .active = qemudActive,
4497
};
4498

4499
int qemuRegister(void) {
4500 4501 4502 4503
    virRegisterDriver(&qemuDriver);
    virRegisterStateDriver(&qemuStateDriver);
    return 0;
}