qemu_driver.c 374.4 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, 2010 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
#include <sys/types.h>
#include <sys/poll.h>
28
#include <sys/time.h>
D
Daniel P. Berrange 已提交
29 30 31
#include <dirent.h>
#include <limits.h>
#include <string.h>
32
#include <stdbool.h>
D
Daniel P. Berrange 已提交
33 34 35 36 37
#include <stdio.h>
#include <stdarg.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
38
#include <sys/utsname.h>
39 40 41 42
#include <sys/stat.h>
#include <fcntl.h>
#include <signal.h>
#include <paths.h>
43 44
#include <pwd.h>
#include <stdio.h>
45
#include <sys/wait.h>
46
#include <sys/ioctl.h>
47
#include <sys/un.h>
D
Daniel P. Berrange 已提交
48

49
#ifdef __linux__
50 51 52 53
# include <sys/vfs.h>
# ifndef NFS_SUPER_MAGIC
#  define NFS_SUPER_MAGIC 0x6969
# endif /* NFS_SUPER_MAGIC */
54
#endif /* __linux__ */
55

56
#include "virterror_internal.h"
57
#include "logging.h"
58
#include "datatypes.h"
59 60
#include "qemu_driver.h"
#include "qemu_conf.h"
61
#include "qemu_monitor.h"
62
#include "qemu_bridge_filter.h"
J
Jim Meyering 已提交
63
#include "c-ctype.h"
64
#include "event.h"
65
#include "buf.h"
66
#include "util.h"
67
#include "nodeinfo.h"
68
#include "stats_linux.h"
69
#include "capabilities.h"
70
#include "memory.h"
71
#include "uuid.h"
72
#include "domain_conf.h"
73 74
#include "node_device_conf.h"
#include "pci.h"
75
#include "hostusb.h"
76
#include "processinfo.h"
77 78
#include "qemu_security_stacked.h"
#include "qemu_security_dac.h"
79
#include "cgroup.h"
C
Chris Lalancette 已提交
80
#include "libvirt_internal.h"
81
#include "xml.h"
82
#include "cpu/cpu.h"
83
#include "macvtap.h"
S
Stefan Berger 已提交
84
#include "nwfilter/nwfilter_gentech_driver.h"
85
#include "hooks.h"
86
#include "storage_file.h"
87

88

89 90
#define VIR_FROM_THIS VIR_FROM_QEMU

91 92 93
#define QEMU_VNC_PORT_MIN  5900
#define QEMU_VNC_PORT_MAX  65535

94 95 96 97 98 99 100 101 102 103 104 105
/* Only 1 job is allowed at any time
 * A job includes *all* monitor commands, even those just querying
 * information, not merely actions */
enum qemuDomainJob {
    QEMU_JOB_NONE = 0,  /* Always set to 0 for easy if (jobActive) conditions */
    QEMU_JOB_UNSPECIFIED,
    QEMU_JOB_MIGRATION,
};

enum qemuDomainJobSignals {
    QEMU_JOB_SIGNAL_CANCEL  = 1 << 0, /* Request job cancellation */
    QEMU_JOB_SIGNAL_SUSPEND = 1 << 1, /* Request VM suspend to finish live migration offline */
106 107 108 109 110
    QEMU_JOB_SIGNAL_MIGRATE_DOWNTIME = 1 << 2, /* Request migration downtime change */
};

struct qemuDomainJobSignalsData {
    unsigned long long migrateDowntime; /* Data for QEMU_JOB_SIGNAL_MIGRATE_DOWNTIME */
111 112
};

113 114 115
typedef struct _qemuDomainObjPrivate qemuDomainObjPrivate;
typedef qemuDomainObjPrivate *qemuDomainObjPrivatePtr;
struct _qemuDomainObjPrivate {
116
    virCond jobCond; /* Use in conjunction with main virDomainObjPtr lock */
117 118
    enum qemuDomainJob jobActive;   /* Currently running job */
    unsigned int jobSignals;        /* Signals for running job */
119
    struct qemuDomainJobSignalsData jobSignalsData; /* Signal specific data */
120 121
    virDomainJobInfo jobInfo;
    unsigned long long jobStart;
122

123
    qemuMonitorPtr mon;
124
    virDomainChrDefPtr monConfig;
D
Daniel P. Berrange 已提交
125
    int monJSON;
126 127 128

    int nvcpupids;
    int *vcpupids;
129 130

    qemuDomainPCIAddressSetPtr pciaddrs;
131
    int persistentAddrs;
132 133
};

134 135
static int qemudShutdown(void);

136 137
static void qemuDriverLock(struct qemud_driver *driver)
{
138
    virMutexLock(&driver->lock);
139 140 141
}
static void qemuDriverUnlock(struct qemud_driver *driver)
{
142
    virMutexUnlock(&driver->lock);
143 144
}

145 146 147
static void qemuDomainEventFlush(int timer, void *opaque);
static void qemuDomainEventQueue(struct qemud_driver *driver,
                                 virDomainEventPtr event);
148

J
Jiri Denemark 已提交
149 150
static int qemudDomainObjStart(virConnectPtr conn,
                               struct qemud_driver *driver,
151 152
                               virDomainObjPtr vm,
                               bool start_paused);
J
Jiri Denemark 已提交
153

154 155
static int qemudStartVMDaemon(virConnectPtr conn,
                              struct qemud_driver *driver,
156
                              virDomainObjPtr vm,
157
                              const char *migrateFrom,
158
                              bool start_paused,
159 160
                              int stdin_fd,
                              const char *stdin_path);
161

162
static void qemudShutdownVMDaemon(struct qemud_driver *driver,
163 164
                                  virDomainObjPtr vm,
                                  int migrated);
165

166
static int qemudDomainGetMaxVcpus(virDomainPtr dom);
167

168
static int qemuDetectVcpuPIDs(struct qemud_driver *driver,
169
                              virDomainObjPtr vm);
170

171 172 173
static int qemuUpdateActivePciHostdevs(struct qemud_driver *driver,
                                       virDomainDefPtr def);

J
Jim Meyering 已提交
174
static struct qemud_driver *qemu_driver = NULL;
175

176 177 178 179 180 181 182 183 184 185 186 187 188 189 190

static void *qemuDomainObjPrivateAlloc(void)
{
    qemuDomainObjPrivatePtr priv;

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

    return priv;
}

static void qemuDomainObjPrivateFree(void *data)
{
    qemuDomainObjPrivatePtr priv = data;

191
    qemuDomainPCIAddressSetFree(priv->pciaddrs);
192 193 194
    virDomainChrDefFree(priv->monConfig);
    VIR_FREE(priv->vcpupids);

195 196
    /* This should never be non-NULL if we get here, but just in case... */
    if (priv->mon) {
197
        VIR_ERROR0(_("Unexpected QEMU monitor still active during domain deletion"));
198 199 200 201 202 203
        qemuMonitorClose(priv->mon);
    }
    VIR_FREE(priv);
}


204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221
static int qemuDomainObjPrivateXMLFormat(virBufferPtr buf, void *data)
{
    qemuDomainObjPrivatePtr priv = data;
    const char *monitorpath;

    /* priv->monitor_chr is set only for qemu */
    if (priv->monConfig) {
        switch (priv->monConfig->type) {
        case VIR_DOMAIN_CHR_TYPE_UNIX:
            monitorpath = priv->monConfig->data.nix.path;
            break;
        default:
        case VIR_DOMAIN_CHR_TYPE_PTY:
            monitorpath = priv->monConfig->data.file.path;
            break;
        }

        virBufferEscapeString(buf, "  <monitor path='%s'", monitorpath);
D
Daniel P. Berrange 已提交
222 223
        if (priv->monJSON)
            virBufferAddLit(buf, " json='1'");
224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249
        virBufferVSprintf(buf, " type='%s'/>\n",
                          virDomainChrTypeToString(priv->monConfig->type));
    }


    if (priv->nvcpupids) {
        int i;
        virBufferAddLit(buf, "  <vcpus>\n");
        for (i = 0 ; i < priv->nvcpupids ; i++) {
            virBufferVSprintf(buf, "    <vcpu pid='%d'/>\n", priv->vcpupids[i]);
        }
        virBufferAddLit(buf, "  </vcpus>\n");
    }

    return 0;
}

static int qemuDomainObjPrivateXMLParse(xmlXPathContextPtr ctxt, void *data)
{
    qemuDomainObjPrivatePtr priv = data;
    char *monitorpath;
    char *tmp;
    int n, i;
    xmlNodePtr *nodes = NULL;

    if (VIR_ALLOC(priv->monConfig) < 0) {
250
        virReportOOMError();
251 252 253
        goto error;
    }

254
    if (!(priv->monConfig->info.alias = strdup("monitor"))) {
255
        virReportOOMError();
256 257 258
        goto error;
    }

259
    if (!(monitorpath =
260
          virXPathString("string(./monitor[1]/@path)", ctxt))) {
261 262
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("no monitor path"));
263 264 265
        goto error;
    }

266
    tmp = virXPathString("string(./monitor[1]/@type)", ctxt);
267 268 269 270 271 272
    if (tmp)
        priv->monConfig->type = virDomainChrTypeFromString(tmp);
    else
        priv->monConfig->type = VIR_DOMAIN_CHR_TYPE_PTY;
    VIR_FREE(tmp);

273
    if (virXPathBoolean("count(./monitor[@json = '1']) > 0", ctxt)) {
D
Daniel P. Berrange 已提交
274
        priv->monJSON = 1;
275 276 277
    } else {
        priv->monJSON = 0;
    }
D
Daniel P. Berrange 已提交
278

279 280 281 282 283 284 285 286 287
    switch (priv->monConfig->type) {
    case VIR_DOMAIN_CHR_TYPE_PTY:
        priv->monConfig->data.file.path = monitorpath;
        break;
    case VIR_DOMAIN_CHR_TYPE_UNIX:
        priv->monConfig->data.nix.path = monitorpath;
        break;
    default:
        VIR_FREE(monitorpath);
288 289 290
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("unsupported monitor type '%s'"),
                        virDomainChrTypeToString(priv->monConfig->type));
291 292 293
        goto error;
    }

294
    n = virXPathNodeSet("./vcpus/vcpu", ctxt, &nodes);
295 296 297 298 299
    if (n < 0)
        goto error;
    if (n) {
        priv->nvcpupids = n;
        if (VIR_REALLOC_N(priv->vcpupids, priv->nvcpupids) < 0) {
300
            virReportOOMError();
301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320
            goto error;
        }

        for (i = 0 ; i < n ; i++) {
            char *pidstr = virXMLPropString(nodes[i], "pid");
            if (!pidstr)
                goto error;

            if (virStrToLong_i(pidstr, NULL, 10, &(priv->vcpupids[i])) < 0) {
                VIR_FREE(pidstr);
                goto error;
            }
            VIR_FREE(pidstr);
        }
        VIR_FREE(nodes);
    }

    return 0;

error:
321 322
    virDomainChrDefFree(priv->monConfig);
    priv->monConfig = NULL;
323 324 325 326 327 328
    VIR_FREE(nodes);
    return -1;
}



329 330 331 332 333 334 335 336 337
/*
 * obj must be locked before calling, qemud_driver must NOT be locked
 *
 * This must be called by anything that will change the VM state
 * in any way, or anything that will use the QEMU monitor.
 *
 * Upon successful return, the object will have its ref count increased,
 * successful calls must be followed by EndJob eventually
 */
338 339 340 341

/* Give up waiting for mutex after 30 seconds */
#define QEMU_JOB_WAIT_TIME (1000ull * 30)

342 343 344 345
static int qemuDomainObjBeginJob(virDomainObjPtr obj) ATTRIBUTE_RETURN_CHECK;
static int qemuDomainObjBeginJob(virDomainObjPtr obj)
{
    qemuDomainObjPrivatePtr priv = obj->privateData;
346 347 348 349
    struct timeval now;
    unsigned long long then;

    if (gettimeofday(&now, NULL) < 0) {
350
        virReportSystemError(errno, "%s",
351 352 353 354 355
                             _("cannot get time of day"));
        return -1;
    }
    then = (now.tv_sec * 1000ull) + (now.tv_usec / 1000);
    then += QEMU_JOB_WAIT_TIME;
356 357 358 359

    virDomainObjRef(obj);

    while (priv->jobActive) {
360
        if (virCondWaitUntil(&priv->jobCond, &obj->lock, then) < 0) {
361
            virDomainObjUnref(obj);
362
            if (errno == ETIMEDOUT)
363 364
                qemuReportError(VIR_ERR_OPERATION_TIMEOUT,
                                "%s", _("cannot acquire state change lock"));
365
            else
366
                virReportSystemError(errno,
367
                                     "%s", _("cannot acquire job mutex"));
368 369 370
            return -1;
        }
    }
371 372
    priv->jobActive = QEMU_JOB_UNSPECIFIED;
    priv->jobSignals = 0;
373
    memset(&priv->jobSignalsData, 0, sizeof(priv->jobSignalsData));
374 375
    priv->jobStart = (now.tv_sec * 1000ull) + (now.tv_usec / 1000);
    memset(&priv->jobInfo, 0, sizeof(priv->jobInfo));
376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391

    return 0;
}

/*
 * obj must be locked before calling, qemud_driver must be locked
 *
 * This must be called by anything that will change the VM state
 * in any way, or anything that will use the QEMU monitor.
 */
static int qemuDomainObjBeginJobWithDriver(struct qemud_driver *driver,
                                           virDomainObjPtr obj) ATTRIBUTE_RETURN_CHECK;
static int qemuDomainObjBeginJobWithDriver(struct qemud_driver *driver,
                                           virDomainObjPtr obj)
{
    qemuDomainObjPrivatePtr priv = obj->privateData;
392 393 394 395
    struct timeval now;
    unsigned long long then;

    if (gettimeofday(&now, NULL) < 0) {
396
        virReportSystemError(errno, "%s",
397 398 399 400 401
                             _("cannot get time of day"));
        return -1;
    }
    then = (now.tv_sec * 1000ull) + (now.tv_usec / 1000);
    then += QEMU_JOB_WAIT_TIME;
402 403 404 405 406

    virDomainObjRef(obj);
    qemuDriverUnlock(driver);

    while (priv->jobActive) {
407
        if (virCondWaitUntil(&priv->jobCond, &obj->lock, then) < 0) {
408
            virDomainObjUnref(obj);
409
            if (errno == ETIMEDOUT)
410 411
                qemuReportError(VIR_ERR_OPERATION_TIMEOUT,
                                "%s", _("cannot acquire state change lock"));
412
            else
413
                virReportSystemError(errno,
414
                                     "%s", _("cannot acquire job mutex"));
M
Matthias Bolte 已提交
415
            qemuDriverLock(driver);
416 417 418
            return -1;
        }
    }
419 420
    priv->jobActive = QEMU_JOB_UNSPECIFIED;
    priv->jobSignals = 0;
421
    memset(&priv->jobSignalsData, 0, sizeof(priv->jobSignalsData));
422 423
    priv->jobStart = (now.tv_sec * 1000ull) + (now.tv_usec / 1000);
    memset(&priv->jobInfo, 0, sizeof(priv->jobInfo));
424 425 426 427 428 429 430 431 432 433 434 435 436

    virDomainObjUnlock(obj);
    qemuDriverLock(driver);
    virDomainObjLock(obj);

    return 0;
}

/*
 * obj must be locked before calling, qemud_driver does not matter
 *
 * To be called after completing the work associated with the
 * earlier  qemuDomainBeginJob() call
437 438 439
 *
 * Returns remaining refcount on 'obj', maybe 0 to indicated it
 * was deleted
440
 */
441
static int ATTRIBUTE_RETURN_CHECK qemuDomainObjEndJob(virDomainObjPtr obj)
442 443 444
{
    qemuDomainObjPrivatePtr priv = obj->privateData;

445 446
    priv->jobActive = QEMU_JOB_NONE;
    priv->jobSignals = 0;
447
    memset(&priv->jobSignalsData, 0, sizeof(priv->jobSignalsData));
448 449
    priv->jobStart = 0;
    memset(&priv->jobInfo, 0, sizeof(priv->jobInfo));
450 451
    virCondSignal(&priv->jobCond);

452
    return virDomainObjUnref(obj);
453 454 455 456 457 458 459 460 461 462 463
}


/*
 * obj must be locked before calling, qemud_driver must be unlocked
 *
 * To be called immediately before any QEMU monitor API call
 * Must have alrady called qemuDomainObjBeginJob().
 *
 * To be followed with qemuDomainObjExitMonitor() once complete
 */
464 465 466 467 468
static void qemuDomainObjEnterMonitor(virDomainObjPtr obj)
{
    qemuDomainObjPrivatePtr priv = obj->privateData;

    qemuMonitorLock(priv->mon);
469
    qemuMonitorRef(priv->mon);
470
    virDomainObjUnlock(obj);
471 472 473
}


474 475 476 477
/* obj must NOT be locked before calling, qemud_driver must be unlocked
 *
 * Should be paired with an earlier  qemuDomainObjEnterMonitor() call
 */
478 479 480
static void qemuDomainObjExitMonitor(virDomainObjPtr obj)
{
    qemuDomainObjPrivatePtr priv = obj->privateData;
481 482 483 484 485 486
    int refs;

    refs = qemuMonitorUnref(priv->mon);

    if (refs > 0)
        qemuMonitorUnlock(priv->mon);
487

488
    virDomainObjLock(obj);
489 490 491 492 493

    if (refs == 0) {
        virDomainObjUnref(obj);
        priv->mon = NULL;
    }
494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509
}


/*
 * obj must be locked before calling, qemud_driver must be locked
 *
 * To be called immediately before any QEMU monitor API call
 * Must have alrady called qemuDomainObjBeginJob().
 *
 * To be followed with qemuDomainObjExitMonitorWithDriver() once complete
 */
static void qemuDomainObjEnterMonitorWithDriver(struct qemud_driver *driver, virDomainObjPtr obj)
{
    qemuDomainObjPrivatePtr priv = obj->privateData;

    qemuMonitorLock(priv->mon);
510
    qemuMonitorRef(priv->mon);
511 512 513 514 515 516 517 518 519 520 521 522 523
    virDomainObjUnlock(obj);
    qemuDriverUnlock(driver);
}


/* obj must NOT be locked before calling, qemud_driver must be unlocked,
 * and will be locked after returning
 *
 * Should be paired with an earlier  qemuDomainObjEnterMonitor() call
 */
static void qemuDomainObjExitMonitorWithDriver(struct qemud_driver *driver, virDomainObjPtr obj)
{
    qemuDomainObjPrivatePtr priv = obj->privateData;
524 525 526 527 528 529
    int refs;

    refs = qemuMonitorUnref(priv->mon);

    if (refs > 0)
        qemuMonitorUnlock(priv->mon);
530 531 532

    qemuDriverLock(driver);
    virDomainObjLock(obj);
533 534 535 536 537

    if (refs == 0) {
        virDomainObjUnref(obj);
        priv->mon = NULL;
    }
538 539 540
}


541 542 543 544 545 546 547 548 549
static int qemuCgroupControllerActive(struct qemud_driver *driver,
                                      int controller)
{
    if (driver->cgroup == NULL)
        return 0;
    if (driver->cgroupControllers & (1 << controller))
        return 1;
    return 0;
}
550

551
static int
552
qemudLogFD(struct qemud_driver *driver, const char* name)
553 554 555
{
    char logfile[PATH_MAX];
    mode_t logmode;
G
Guido Günther 已提交
556
    int ret, fd = -1;
557

558 559
    if ((ret = snprintf(logfile, sizeof(logfile), "%s/%s.log",
                        driver->logDir, name))
G
Guido Günther 已提交
560
        < 0 || ret >= sizeof(logfile)) {
561
        virReportOOMError();
562 563 564 565
        return -1;
    }

    logmode = O_CREAT | O_WRONLY;
566 567
    /* Only logrotate files in /var/log, so only append if running privileged */
    if (driver->privileged)
568
        logmode |= O_APPEND;
569 570 571
    else
        logmode |= O_TRUNC;

572
    if ((fd = open(logfile, logmode, S_IRUSR | S_IWUSR)) < 0) {
573
        virReportSystemError(errno,
574 575
                             _("failed to create logfile %s"),
                             logfile);
576 577
        return -1;
    }
578
    if (virSetCloseExec(fd) < 0) {
579
        virReportSystemError(errno, "%s",
580
                             _("Unable to set VM logfile close-on-exec flag"));
581 582 583 584 585 586 587
        close(fd);
        return -1;
    }
    return fd;
}


588
static int
589
qemudLogReadFD(const char* logDir, const char* name, off_t pos)
590 591 592 593 594 595 596
{
    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)) {
597 598 599
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("failed to build logfile name %s/%s.log"),
                        logDir, name);
600 601 602 603 604
        return -1;
    }


    if ((fd = open(logfile, logmode)) < 0) {
605
        virReportSystemError(errno,
606 607
                             _("failed to create logfile %s"),
                             logfile);
608 609
        return -1;
    }
610
    if (virSetCloseExec(fd) < 0) {
611
        virReportSystemError(errno, "%s",
612
                             _("Unable to set VM logfile close-on-exec flag"));
613 614 615
        close(fd);
        return -1;
    }
616
    if (pos < 0 || lseek(fd, pos, SEEK_SET) < 0) {
617
      virReportSystemError(pos < 0 ? 0 : errno,
618 619
                             _("Unable to seek to %lld in %s"),
                             (long long) pos, logfile);
620 621 622 623 624 625
        close(fd);
    }
    return fd;
}


626 627 628 629 630 631 632 633 634
struct qemuAutostartData {
    struct qemud_driver *driver;
    virConnectPtr conn;
};
static void
qemuAutostartDomain(void *payload, const char *name ATTRIBUTE_UNUSED, void *opaque)
{
    virDomainObjPtr vm = payload;
    struct qemuAutostartData *data = opaque;
635
    virErrorPtr err;
636 637

    virDomainObjLock(vm);
638 639 640 641 642 643 644 645 646
    virResetLastError();
    if (qemuDomainObjBeginJobWithDriver(data->driver, vm) < 0) {
        err = virGetLastError();
        VIR_ERROR(_("Failed to start job on VM '%s': %s"),
                  vm->def->name,
                  err ? err->message : _("unknown error"));
    } else {
        if (vm->autostart &&
            !virDomainObjIsActive(vm) &&
647
            qemudDomainObjStart(data->conn, data->driver, vm, false) < 0) {
648
            err = virGetLastError();
649
            VIR_ERROR(_("Failed to autostart VM '%s': %s"),
650
                      vm->def->name,
651
                      err ? err->message : _("unknown error"));
652
        }
653 654 655

        if (qemuDomainObjEndJob(vm) == 0)
            vm = NULL;
656
    }
657 658 659

    if (vm)
        virDomainObjUnlock(vm);
660 661
}

662 663
static void
qemudAutostartConfigs(struct qemud_driver *driver) {
664 665 666 667 668
    /* 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
     */
669 670 671
    virConnectPtr conn = virConnectOpen(driver->privileged ?
                                        "qemu:///system" :
                                        "qemu:///session");
672
    /* Ignoring NULL conn which is mostly harmless here */
673
    struct qemuAutostartData data = { driver, conn };
674

675
    qemuDriverLock(driver);
676
    virHashForEach(driver->domains.objs, qemuAutostartDomain, &data);
677
    qemuDriverUnlock(driver);
678

679 680
    if (conn)
        virConnectClose(conn);
681 682
}

683 684 685 686 687 688 689 690 691

/**
 * qemudRemoveDomainStatus
 *
 * remove all state files of a domain from statedir
 *
 * Returns 0 on success
 */
static int
692
qemudRemoveDomainStatus(struct qemud_driver *driver,
693 694
                        virDomainObjPtr vm)
{
695
    char ebuf[1024];
696 697 698
    char *file = NULL;

    if (virAsprintf(&file, "%s/%s.xml", driver->stateDir, vm->def->name) < 0) {
699
        virReportOOMError();
D
Daniel Veillard 已提交
700
        return(-1);
701 702
    }

703
    if (unlink(file) < 0 && errno != ENOENT && errno != ENOTDIR)
704
        VIR_WARN("Failed to remove domain XML for %s: %s",
D
Daniel Veillard 已提交
705 706 707
                 vm->def->name, virStrerror(errno, ebuf, sizeof(ebuf)));
    VIR_FREE(file);

708
    if (virFileDeletePid(driver->stateDir, vm->def->name) != 0)
709
        VIR_WARN("Failed to remove PID file for %s: %s",
710
                 vm->def->name, virStrerror(errno, ebuf, sizeof(ebuf)));
711

D
Daniel Veillard 已提交
712

713
    return 0;
714 715
}

716 717 718 719 720 721 722 723 724 725 726 727 728 729

/*
 * This is a callback registered with a qemuMonitorPtr  instance,
 * and to be invoked when the monitor console hits an end of file
 * condition, or error, thus indicating VM shutdown should be
 * performed
 */
static void
qemuHandleMonitorEOF(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
                     virDomainObjPtr vm,
                     int hasError) {
    struct qemud_driver *driver = qemu_driver;
    virDomainEventPtr event = NULL;

730
    VIR_DEBUG("Received EOF on %p '%s'", vm, vm->def->name);
731 732 733 734 735 736 737 738
    virDomainObjLock(vm);

    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     hasError ?
                                     VIR_DOMAIN_EVENT_STOPPED_FAILED :
                                     VIR_DOMAIN_EVENT_STOPPED_SHUTDOWN);

739
    qemudShutdownVMDaemon(driver, vm, 0);
740 741 742 743 744 745 746 747 748 749 750 751 752
    if (!vm->persistent)
        virDomainRemoveInactive(&driver->domains, vm);
    else
        virDomainObjUnlock(vm);

    if (event) {
        qemuDriverLock(driver);
        qemuDomainEventQueue(driver, event);
        qemuDriverUnlock(driver);
    }
}


753 754 755
static virDomainDiskDefPtr
findDomainDiskByPath(virDomainObjPtr vm,
                     const char *path)
756 757 758 759 760 761 762
{
    int i;

    for (i = 0; i < vm->def->ndisks; i++) {
        virDomainDiskDefPtr disk;

        disk = vm->def->disks[i];
763 764
        if (disk->src != NULL && STREQ(disk->src, path))
            return disk;
765
    }
766 767 768 769

    qemuReportError(VIR_ERR_INTERNAL_ERROR,
                    _("no disk found with path %s"),
                    path);
770 771 772
    return NULL;
}

773 774 775 776 777 778
static virDomainDiskDefPtr
findDomainDiskByAlias(virDomainObjPtr vm,
                      const char *alias)
{
    int i;

779 780 781
    if (STRPREFIX(alias, QEMU_DRIVE_HOST_PREFIX))
        alias += strlen(QEMU_DRIVE_HOST_PREFIX);

782 783 784 785 786 787 788 789 790 791 792 793 794 795
    for (i = 0; i < vm->def->ndisks; i++) {
        virDomainDiskDefPtr disk;

        disk = vm->def->disks[i];
        if (disk->info.alias != NULL && STREQ(disk->info.alias, alias))
            return disk;
    }

    qemuReportError(VIR_ERR_INTERNAL_ERROR,
                    _("no disk found with alias %s"),
                    alias);
    return NULL;
}

796
static int
797 798 799 800
getVolumeQcowPassphrase(virConnectPtr conn,
                        virDomainDiskDefPtr disk,
                        char **secretRet,
                        size_t *secretLen)
801 802 803 804 805
{
    virSecretPtr secret;
    char *passphrase;
    unsigned char *data;
    size_t size;
806
    int ret = -1;
807
    virStorageEncryptionPtr enc;
808

809 810 811 812 813 814 815
    if (!disk->encryption) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("disk %s does not have any encryption information"),
                        disk->src);
        return -1;
    }
    enc = disk->encryption;
816 817

    if (!conn) {
818 819
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        "%s", _("cannot find secrets without a connection"));
820
        goto cleanup;
821 822 823 824 825
    }

    if (conn->secretDriver == NULL ||
        conn->secretDriver->lookupByUUID == NULL ||
        conn->secretDriver->getValue == NULL) {
826 827
        qemuReportError(VIR_ERR_NO_SUPPORT, "%s",
                        _("secret storage not supported"));
828
        goto cleanup;
829 830 831 832 833 834
    }

    if (enc->format != VIR_STORAGE_ENCRYPTION_FORMAT_QCOW ||
        enc->nsecrets != 1 ||
        enc->secrets[0]->type !=
        VIR_STORAGE_ENCRYPTION_SECRET_TYPE_PASSPHRASE) {
835
        qemuReportError(VIR_ERR_INVALID_DOMAIN,
836
                        _("invalid <encryption> for volume %s"), disk->src);
837
        goto cleanup;
838 839 840 841 842
    }

    secret = conn->secretDriver->lookupByUUID(conn,
                                              enc->secrets[0]->uuid);
    if (secret == NULL)
843
        goto cleanup;
844 845 846 847
    data = conn->secretDriver->getValue(secret, &size,
                                        VIR_SECRET_GET_VALUE_INTERNAL_CALL);
    virUnrefSecret(secret);
    if (data == NULL)
848
        goto cleanup;
849 850 851 852

    if (memchr(data, '\0', size) != NULL) {
        memset(data, 0, size);
        VIR_FREE(data);
853 854
        qemuReportError(VIR_ERR_INVALID_SECRET,
                        _("format='qcow' passphrase for %s must not contain a "
855
                          "'\\0'"), disk->src);
856
        goto cleanup;
857 858 859 860 861
    }

    if (VIR_ALLOC_N(passphrase, size + 1) < 0) {
        memset(data, 0, size);
        VIR_FREE(data);
862
        virReportOOMError();
863
        goto cleanup;
864 865 866 867 868 869 870 871 872 873
    }
    memcpy(passphrase, data, size);
    passphrase[size] = '\0';

    memset(data, 0, size);
    VIR_FREE(data);

    *secretRet = passphrase;
    *secretLen = size;

874 875 876
    ret = 0;

cleanup:
877 878
    return ret;
}
879

880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900
static int
findVolumeQcowPassphrase(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
                         virConnectPtr conn,
                         virDomainObjPtr vm,
                         const char *path,
                         char **secretRet,
                         size_t *secretLen)
{
    virDomainDiskDefPtr disk;
    int ret = -1;

    virDomainObjLock(vm);
    disk = findDomainDiskByPath(vm, path);

    if (!disk)
        goto cleanup;

    ret = getVolumeQcowPassphrase(conn, disk, secretRet, secretLen);

cleanup:
    virDomainObjUnlock(vm);
901
    return ret;
902 903
}

904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925

static int
qemuHandleDomainReset(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
                      virDomainObjPtr vm)
{
    struct qemud_driver *driver = qemu_driver;
    virDomainEventPtr event;

    virDomainObjLock(vm);
    event = virDomainEventRebootNewFromObj(vm);
    virDomainObjUnlock(vm);

    if (event) {
        qemuDriverLock(driver);
        qemuDomainEventQueue(driver, event);
        qemuDriverUnlock(driver);
    }

    return 0;
}


926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957
static int
qemuHandleDomainStop(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
                     virDomainObjPtr vm)
{
    struct qemud_driver *driver = qemu_driver;
    virDomainEventPtr event = NULL;

    virDomainObjLock(vm);
    if (vm->state == VIR_DOMAIN_RUNNING) {
        VIR_DEBUG("Transitioned guest %s to paused state due to unknown event", vm->def->name);

        vm->state = VIR_DOMAIN_PAUSED;
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_SUSPENDED,
                                         VIR_DOMAIN_EVENT_SUSPENDED_PAUSED);

        if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
            VIR_WARN("Unable to save status on vm %s after IO error", vm->def->name);
    }
    virDomainObjUnlock(vm);

    if (event) {
        qemuDriverLock(driver);
        if (event)
            qemuDomainEventQueue(driver, event);
        qemuDriverUnlock(driver);
    }

    return 0;
}


958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986
static int
qemuHandleDomainRTCChange(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
                          virDomainObjPtr vm,
                          long long offset)
{
    struct qemud_driver *driver = qemu_driver;
    virDomainEventPtr event;

    virDomainObjLock(vm);
    event = virDomainEventRTCChangeNewFromObj(vm, offset);

    if (vm->def->clock.offset == VIR_DOMAIN_CLOCK_OFFSET_VARIABLE)
        vm->def->clock.data.adjustment = offset;

    if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
        VIR_WARN0("unable to save domain status with RTC change");

    virDomainObjUnlock(vm);

    if (event) {
        qemuDriverLock(driver);
        qemuDomainEventQueue(driver, event);
        qemuDriverUnlock(driver);
    }

    return 0;
}


987 988 989 990 991 992
static int
qemuHandleDomainWatchdog(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
                         virDomainObjPtr vm,
                         int action)
{
    struct qemud_driver *driver = qemu_driver;
993 994
    virDomainEventPtr watchdogEvent = NULL;
    virDomainEventPtr lifecycleEvent = NULL;
995 996

    virDomainObjLock(vm);
997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010
    watchdogEvent = virDomainEventWatchdogNewFromObj(vm, action);

    if (action == VIR_DOMAIN_EVENT_WATCHDOG_PAUSE &&
        vm->state == VIR_DOMAIN_RUNNING) {
        VIR_DEBUG("Transitioned guest %s to paused state due to watchdog", vm->def->name);

        vm->state = VIR_DOMAIN_PAUSED;
        lifecycleEvent = virDomainEventNewFromObj(vm,
                                                  VIR_DOMAIN_EVENT_SUSPENDED,
                                                  VIR_DOMAIN_EVENT_SUSPENDED_WATCHDOG);

        if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
            VIR_WARN("Unable to save status on vm %s after IO error", vm->def->name);
    }
1011 1012
    virDomainObjUnlock(vm);

1013
    if (watchdogEvent || lifecycleEvent) {
1014
        qemuDriverLock(driver);
1015 1016 1017 1018
        if (watchdogEvent)
            qemuDomainEventQueue(driver, watchdogEvent);
        if (lifecycleEvent)
            qemuDomainEventQueue(driver, lifecycleEvent);
1019 1020 1021 1022 1023 1024 1025
        qemuDriverUnlock(driver);
    }

    return 0;
}


1026 1027 1028 1029
static int
qemuHandleDomainIOError(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
                        virDomainObjPtr vm,
                        const char *diskAlias,
1030 1031
                        int action,
                        const char *reason)
1032 1033
{
    struct qemud_driver *driver = qemu_driver;
1034
    virDomainEventPtr ioErrorEvent = NULL;
1035
    virDomainEventPtr ioErrorEvent2 = NULL;
1036
    virDomainEventPtr lifecycleEvent = NULL;
1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051
    const char *srcPath;
    const char *devAlias;
    virDomainDiskDefPtr disk;

    virDomainObjLock(vm);
    disk = findDomainDiskByAlias(vm, diskAlias);

    if (disk) {
        srcPath = disk->src;
        devAlias = disk->info.alias;
    } else {
        srcPath = "";
        devAlias = "";
    }

1052
    ioErrorEvent = virDomainEventIOErrorNewFromObj(vm, srcPath, devAlias, action);
1053
    ioErrorEvent2 = virDomainEventIOErrorReasonNewFromObj(vm, srcPath, devAlias, action, reason);
1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066

    if (action == VIR_DOMAIN_EVENT_IO_ERROR_PAUSE &&
        vm->state == VIR_DOMAIN_RUNNING) {
        VIR_DEBUG("Transitioned guest %s to paused state due to IO error", vm->def->name);

        vm->state = VIR_DOMAIN_PAUSED;
        lifecycleEvent = virDomainEventNewFromObj(vm,
                                                  VIR_DOMAIN_EVENT_SUSPENDED,
                                                  VIR_DOMAIN_EVENT_SUSPENDED_IOERROR);

        if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
            VIR_WARN("Unable to save status on vm %s after IO error", vm->def->name);
    }
1067 1068
    virDomainObjUnlock(vm);

1069
    if (ioErrorEvent || ioErrorEvent2 || lifecycleEvent) {
1070
        qemuDriverLock(driver);
1071 1072
        if (ioErrorEvent)
            qemuDomainEventQueue(driver, ioErrorEvent);
1073 1074
        if (ioErrorEvent2)
            qemuDomainEventQueue(driver, ioErrorEvent2);
1075 1076
        if (lifecycleEvent)
            qemuDomainEventQueue(driver, lifecycleEvent);
1077 1078 1079 1080 1081 1082 1083
        qemuDriverUnlock(driver);
    }

    return 0;
}


1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175
static int
qemuHandleDomainGraphics(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
                         virDomainObjPtr vm,
                         int phase,
                         int localFamily,
                         const char *localNode,
                         const char *localService,
                         int remoteFamily,
                         const char *remoteNode,
                         const char *remoteService,
                         const char *authScheme,
                         const char *x509dname,
                         const char *saslUsername)
{
    struct qemud_driver *driver = qemu_driver;
    virDomainEventPtr event;
    virDomainEventGraphicsAddressPtr localAddr = NULL;
    virDomainEventGraphicsAddressPtr remoteAddr = NULL;
    virDomainEventGraphicsSubjectPtr subject = NULL;
    int i;

    virDomainObjLock(vm);

    if (VIR_ALLOC(localAddr) < 0)
        goto no_memory;
    localAddr->family = localFamily;
    if (!(localAddr->service = strdup(localService)) ||
        !(localAddr->node = strdup(localNode)))
        goto no_memory;

    if (VIR_ALLOC(remoteAddr) < 0)
        goto no_memory;
    remoteAddr->family = remoteFamily;
    if (!(remoteAddr->service = strdup(remoteService)) ||
        !(remoteAddr->node = strdup(remoteNode)))
        goto no_memory;

    if (VIR_ALLOC(subject) < 0)
        goto no_memory;
    if (x509dname) {
        if (VIR_REALLOC_N(subject->identities, subject->nidentity+1) < 0)
            goto no_memory;
        if (!(subject->identities[subject->nidentity].type = strdup("x509dname")) ||
            !(subject->identities[subject->nidentity].name = strdup(x509dname)))
            goto no_memory;
        subject->nidentity++;
    }
    if (saslUsername) {
        if (VIR_REALLOC_N(subject->identities, subject->nidentity+1) < 0)
            goto no_memory;
        if (!(subject->identities[subject->nidentity].type = strdup("saslUsername")) ||
            !(subject->identities[subject->nidentity].name = strdup(saslUsername)))
            goto no_memory;
        subject->nidentity++;
    }

    event = virDomainEventGraphicsNewFromObj(vm, phase, localAddr, remoteAddr, authScheme, subject);
    virDomainObjUnlock(vm);

    if (event) {
        qemuDriverLock(driver);
        qemuDomainEventQueue(driver, event);
        qemuDriverUnlock(driver);
    }

    return 0;

no_memory:
    virReportOOMError();
    if (localAddr) {
        VIR_FREE(localAddr->service);
        VIR_FREE(localAddr->node);
        VIR_FREE(localAddr);
    }
    if (remoteAddr) {
        VIR_FREE(remoteAddr->service);
        VIR_FREE(remoteAddr->node);
        VIR_FREE(remoteAddr);
    }
    if (subject) {
        for (i = 0 ; i < subject->nidentity ; i++) {
            VIR_FREE(subject->identities[i].type);
            VIR_FREE(subject->identities[i].name);
        }
        VIR_FREE(subject->identities);
        VIR_FREE(subject);
    }

    return -1;
}


1176 1177 1178
static qemuMonitorCallbacks monitorCallbacks = {
    .eofNotify = qemuHandleMonitorEOF,
    .diskSecretLookup = findVolumeQcowPassphrase,
1179
    .domainStop = qemuHandleDomainStop,
1180
    .domainReset = qemuHandleDomainReset,
1181
    .domainRTCChange = qemuHandleDomainRTCChange,
1182
    .domainWatchdog = qemuHandleDomainWatchdog,
1183
    .domainIOError = qemuHandleDomainIOError,
1184
    .domainGraphics = qemuHandleDomainGraphics,
1185 1186
};

1187
static int
1188
qemuConnectMonitor(struct qemud_driver *driver, virDomainObjPtr vm)
1189
{
1190
    qemuDomainObjPrivatePtr priv = vm->privateData;
1191
    int ret = -1;
1192

1193 1194 1195 1196
    /* Hold an extra reference because we can't allow 'vm' to be
     * deleted while the monitor is active */
    virDomainObjRef(vm);

1197 1198 1199 1200 1201 1202 1203
    if ((driver->securityDriver &&
         driver->securityDriver->domainSetSecuritySocketLabel &&
         driver->securityDriver->domainSetSecuritySocketLabel(driver->securityDriver,vm)) < 0) {
        VIR_ERROR(_("Failed to set security context for monitor for %s"), vm->def->name);
        goto error;
    }

1204 1205
    if ((priv->mon = qemuMonitorOpen(vm,
                                     priv->monConfig,
D
Daniel P. Berrange 已提交
1206
                                     priv->monJSON,
1207
                                     &monitorCallbacks)) == NULL) {
1208
        VIR_ERROR(_("Failed to connect monitor for %s"), vm->def->name);
1209 1210 1211 1212 1213 1214 1215 1216
        goto error;
    }

    if ((driver->securityDriver &&
         driver->securityDriver->domainClearSecuritySocketLabel &&
         driver->securityDriver->domainClearSecuritySocketLabel(driver->securityDriver,vm)) < 0) {
        VIR_ERROR(_("Failed to set security context for monitor for %s"), vm->def->name);
        goto error;
1217
    }
1218

1219 1220 1221 1222
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
    ret = qemuMonitorSetCapabilities(priv->mon);
    qemuDomainObjExitMonitorWithDriver(driver, vm);

1223 1224
    ret = 0;
error:
1225 1226 1227
    if (ret < 0) {
        qemuMonitorClose(priv->mon);
        priv->mon = NULL;
1228
        virDomainObjUnref(vm);
1229 1230 1231
    }

    return ret;
1232
}
1233 1234 1235 1236

/*
 * Open an existing VM's monitor, re-detect VCPU threads
 * and re-reserve the security labels in use
1237
 */
1238 1239
static void
qemuReconnectDomain(void *payload, const char *name ATTRIBUTE_UNUSED, void *opaque)
1240
{
1241 1242
    virDomainObjPtr obj = payload;
    struct qemud_driver *driver = opaque;
1243
    qemuDomainObjPrivatePtr priv;
1244
    unsigned long long qemuCmdFlags;
1245 1246

    virDomainObjLock(obj);
1247

1248 1249
    VIR_DEBUG("Reconnect monitor to %p '%s'", obj, obj->def->name);

1250 1251
    priv = obj->privateData;

1252
    /* XXX check PID liveliness & EXE path */
1253
    if (qemuConnectMonitor(driver, obj) < 0)
1254
        goto error;
1255

1256 1257 1258 1259
    if (qemuUpdateActivePciHostdevs(driver, obj->def) < 0) {
        goto error;
    }

1260 1261 1262 1263 1264 1265 1266 1267 1268
    /* XXX we should be persisting the original flags in the XML
     * not re-detecting them, since the binary may have changed
     * since launch time */
    if (qemudExtractVersionInfo(obj->def->emulator,
                                NULL,
                                &qemuCmdFlags) >= 0 &&
        (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE))
        priv->persistentAddrs = 1;

1269 1270 1271
    if (!(priv->pciaddrs = qemuDomainPCIAddressSetCreate(obj->def)))
        goto error;

1272
    if (driver->securityDriver &&
1273
        driver->securityDriver->domainReserveSecurityLabel &&
1274
        driver->securityDriver->domainReserveSecurityLabel(obj) < 0)
1275
        goto error;
1276

1277 1278
    if (obj->def->id >= driver->nextvmid)
        driver->nextvmid = obj->def->id + 1;
1279

1280 1281
    virDomainObjUnlock(obj);
    return;
1282

1283
error:
1284 1285 1286
    /* We can't get the monitor back, so must kill the VM
     * to remove danger of it ending up running twice if
     * user tries to start it again later */
1287
    qemudShutdownVMDaemon(driver, obj, 0);
1288 1289 1290 1291
    if (!obj->persistent)
        virDomainRemoveInactive(&driver->domains, obj);
    else
        virDomainObjUnlock(obj);
1292
}
1293

1294
/**
1295
 * qemudReconnectDomains
1296 1297 1298 1299 1300 1301 1302
 *
 * Try to re-open the resources for live VMs that we care
 * about.
 */
static void
qemuReconnectDomains(struct qemud_driver *driver)
{
1303
    virHashForEach(driver->domains.objs, qemuReconnectDomain, driver);
1304 1305
}

1306

1307 1308 1309 1310 1311 1312
static int
qemudSecurityInit(struct qemud_driver *qemud_drv)
{
    int ret;
    virSecurityDriverPtr security_drv;

1313 1314 1315
    qemuSecurityStackedSetDriver(qemud_drv);
    qemuSecurityDACSetDriver(qemud_drv);

1316 1317 1318 1319 1320 1321
    ret = virSecurityDriverStartup(&security_drv,
                                   qemud_drv->securityDriverName);
    if (ret == -1) {
        VIR_ERROR0(_("Failed to start security driver"));
        return -1;
    }
1322 1323 1324

    /* No primary security driver wanted to be enabled: just setup
     * the DAC driver on its own */
1325
    if (ret == -2) {
1326
        qemud_drv->securityDriver = &qemuDACSecurityDriver;
1327
        VIR_INFO0(_("No security driver available"));
1328 1329 1330 1331 1332
    } else {
        qemud_drv->securityPrimaryDriver = security_drv;
        qemud_drv->securitySecondaryDriver = &qemuDACSecurityDriver;
        qemud_drv->securityDriver = &qemuStackedSecurityDriver;
        VIR_INFO("Initialized security driver %s", security_drv->name);
1333 1334
    }

1335
    return 0;
1336
}
1337 1338


1339 1340
static virCapsPtr
qemuCreateCapabilities(virCapsPtr oldcaps,
1341
                       struct qemud_driver *driver)
1342 1343 1344 1345 1346
{
    virCapsPtr caps;

    /* Basic host arch / guest machine capabilities */
    if (!(caps = qemudCapsInit(oldcaps))) {
1347
        virReportOOMError();
1348 1349 1350 1351 1352 1353 1354 1355 1356
        return NULL;
    }

    /* Domain XML parser hooks */
    caps->privateDataAllocFunc = qemuDomainObjPrivateAlloc;
    caps->privateDataFreeFunc = qemuDomainObjPrivateFree;
    caps->privateDataXMLFormat = qemuDomainObjPrivateXMLFormat;
    caps->privateDataXMLParse = qemuDomainObjPrivateXMLParse;

1357 1358 1359 1360 1361
    if (virGetHostUUID(caps->host.host_uuid)) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot get the host uuid"));
        goto err_exit;
    }
1362 1363

    /* Security driver data */
1364
    if (driver->securityPrimaryDriver) {
1365 1366
        const char *doi, *model;

1367 1368
        doi = virSecurityDriverGetDOI(driver->securityPrimaryDriver);
        model = virSecurityDriverGetModel(driver->securityPrimaryDriver);
1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381

        if (!(caps->host.secModel.model = strdup(model)))
            goto no_memory;
        if (!(caps->host.secModel.doi = strdup(doi)))
            goto no_memory;

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

    return caps;

no_memory:
1382
    virReportOOMError();
1383
err_exit:
1384 1385 1386
    virCapabilitiesFree(caps);
    return NULL;
}
1387

C
Chris Lalancette 已提交
1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404
static void qemuDomainSnapshotLoad(void *payload,
                                   const char *name ATTRIBUTE_UNUSED,
                                   void *data)
{
    virDomainObjPtr vm = (virDomainObjPtr)payload;
    char *baseDir = (char *)data;
    char *snapDir = NULL;
    DIR *dir = NULL;
    struct dirent *entry;
    char *xmlStr;
    int ret;
    char *fullpath;
    virDomainSnapshotDefPtr def = NULL;
    char ebuf[1024];

    virDomainObjLock(vm);
    if (virAsprintf(&snapDir, "%s/%s", baseDir, vm->def->name) < 0) {
1405
        VIR_ERROR(_("Failed to allocate memory for snapshot directory for domain %s"),
C
Chris Lalancette 已提交
1406 1407 1408 1409 1410 1411 1412 1413 1414
                   vm->def->name);
        goto cleanup;
    }

    VIR_INFO("Scanning for snapshots for domain %s in %s", vm->def->name,
             snapDir);

    if (!(dir = opendir(snapDir))) {
        if (errno != ENOENT)
1415
            VIR_ERROR(_("Failed to open snapshot directory %s for domain %s: %s"),
C
Chris Lalancette 已提交
1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429
                      snapDir, vm->def->name,
                      virStrerror(errno, ebuf, sizeof(ebuf)));
        goto cleanup;
    }

    while ((entry = readdir(dir))) {
        if (entry->d_name[0] == '.')
            continue;

        /* NB: ignoring errors, so one malformed config doesn't
           kill the whole process */
        VIR_INFO("Loading snapshot file '%s'", entry->d_name);

        if (virAsprintf(&fullpath, "%s/%s", snapDir, entry->d_name) < 0) {
1430
            VIR_ERROR0(_("Failed to allocate memory for path"));
C
Chris Lalancette 已提交
1431 1432 1433 1434 1435 1436
            continue;
        }

        ret = virFileReadAll(fullpath, 1024*1024*1, &xmlStr);
        if (ret < 0) {
            /* Nothing we can do here, skip this one */
1437
            VIR_ERROR(_("Failed to read snapshot file %s: %s"), fullpath,
C
Chris Lalancette 已提交
1438
                      virStrerror(errno, ebuf, sizeof(ebuf)));
1439
            VIR_FREE(fullpath);
C
Chris Lalancette 已提交
1440 1441 1442 1443 1444 1445
            continue;
        }

        def = virDomainSnapshotDefParseString(xmlStr, 0);
        if (def == NULL) {
            /* Nothing we can do here, skip this one */
1446
            VIR_ERROR(_("Failed to parse snapshot XML from file '%s'"), fullpath);
1447
            VIR_FREE(fullpath);
C
Chris Lalancette 已提交
1448 1449 1450 1451
            VIR_FREE(xmlStr);
            continue;
        }

1452
        virDomainSnapshotAssignDef(&vm->snapshots, def);
C
Chris Lalancette 已提交
1453

1454
        VIR_FREE(fullpath);
C
Chris Lalancette 已提交
1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475
        VIR_FREE(xmlStr);
    }

    /* FIXME: qemu keeps internal track of snapshots.  We can get access
     * to this info via the "info snapshots" monitor command for running
     * domains, or via "qemu-img snapshot -l" for shutoff domains.  It would
     * be nice to update our internal state based on that, but there is a
     * a problem.  qemu doesn't track all of the same metadata that we do.
     * In particular we wouldn't be able to fill in the <parent>, which is
     * pretty important in our metadata.
     */

    virResetLastError();

cleanup:
    if (dir)
        closedir(dir);
    VIR_FREE(snapDir);
    virDomainObjUnlock(vm);
}

1476 1477 1478 1479 1480 1481
/**
 * qemudStartup:
 *
 * Initialization function for the QEmu daemon
 */
static int
1482
qemudStartup(int privileged) {
1483
    char *base = NULL;
D
Daniel P. Berrange 已提交
1484
    char driverConf[PATH_MAX];
1485
    int rc;
1486

1487
    if (VIR_ALLOC(qemu_driver) < 0)
1488 1489
        return -1;

1490
    if (virMutexInit(&qemu_driver->lock) < 0) {
1491
        VIR_ERROR0(_("cannot initialize mutex"));
1492 1493 1494
        VIR_FREE(qemu_driver);
        return -1;
    }
1495
    qemuDriverLock(qemu_driver);
1496
    qemu_driver->privileged = privileged;
1497

1498 1499 1500
    /* Don't have a dom0 so start from 1 */
    qemu_driver->nextvmid = 1;

1501 1502 1503
    if (virDomainObjListInit(&qemu_driver->domains) < 0)
        goto out_of_memory;

1504
    /* Init callback list */
1505
    if (VIR_ALLOC(qemu_driver->domainEventCallbacks) < 0)
1506
        goto out_of_memory;
1507 1508 1509 1510 1511 1512
    if (!(qemu_driver->domainEventQueue = virDomainEventQueueNew()))
        goto out_of_memory;

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

1514 1515 1516 1517 1518
    /* Allocate bitmap for vnc port reservation */
    if ((qemu_driver->reservedVNCPorts =
         virBitmapAlloc(QEMU_VNC_PORT_MAX - QEMU_VNC_PORT_MIN)) == NULL)
        goto out_of_memory;

1519
    if (privileged) {
1520 1521
        if (virAsprintf(&qemu_driver->logDir,
                        "%s/log/libvirt/qemu", LOCAL_STATE_DIR) == -1)
1522
            goto out_of_memory;
1523

D
Daniel P. Berrange 已提交
1524
        if ((base = strdup (SYSCONF_DIR "/libvirt")) == NULL)
1525
            goto out_of_memory;
1526 1527

        if (virAsprintf(&qemu_driver->stateDir,
1528
                      "%s/run/libvirt/qemu", LOCAL_STATE_DIR) == -1)
1529
            goto out_of_memory;
1530 1531 1532 1533 1534 1535 1536 1537

        if (virAsprintf(&qemu_driver->libDir,
                      "%s/lib/libvirt/qemu", LOCAL_STATE_DIR) == -1)
            goto out_of_memory;

        if (virAsprintf(&qemu_driver->cacheDir,
                      "%s/cache/libvirt/qemu", LOCAL_STATE_DIR) == -1)
            goto out_of_memory;
1538 1539 1540
        if (virAsprintf(&qemu_driver->saveDir,
                      "%s/lib/libvirt/qemu/save/", LOCAL_STATE_DIR) == -1)
            goto out_of_memory;
C
Chris Lalancette 已提交
1541 1542 1543
        if (virAsprintf(&qemu_driver->snapshotDir,
                        "%s/lib/libvirt/qemu/snapshot", LOCAL_STATE_DIR) == -1)
            goto out_of_memory;
1544
    } else {
1545
        uid_t uid = geteuid();
1546
        char *userdir = virGetUserDirectory(uid);
1547
        if (!userdir)
1548
            goto error;
1549

1550
        if (virAsprintf(&qemu_driver->logDir,
1551 1552
                        "%s/.libvirt/qemu/log", userdir) == -1) {
            VIR_FREE(userdir);
1553
            goto out_of_memory;
1554
        }
1555

1556 1557
        if (virAsprintf(&base, "%s/.libvirt", userdir) == -1) {
            VIR_FREE(userdir);
1558
            goto out_of_memory;
1559 1560
        }
        VIR_FREE(userdir);
1561 1562 1563

        if (virAsprintf(&qemu_driver->stateDir, "%s/qemu/run", base) == -1)
            goto out_of_memory;
1564 1565 1566 1567
        if (virAsprintf(&qemu_driver->libDir, "%s/qemu/lib", base) == -1)
            goto out_of_memory;
        if (virAsprintf(&qemu_driver->cacheDir, "%s/qemu/cache", base) == -1)
            goto out_of_memory;
1568 1569
        if (virAsprintf(&qemu_driver->saveDir, "%s/qemu/save", base) == -1)
            goto out_of_memory;
C
Chris Lalancette 已提交
1570 1571
        if (virAsprintf(&qemu_driver->snapshotDir, "%s/qemu/snapshot", base) == -1)
            goto out_of_memory;
1572 1573
    }

L
Laine Stump 已提交
1574
    if (virFileMakePath(qemu_driver->stateDir) != 0) {
1575
        char ebuf[1024];
1576
        VIR_ERROR(_("Failed to create state dir '%s': %s"),
1577
                  qemu_driver->stateDir, virStrerror(errno, ebuf, sizeof ebuf));
1578
        goto error;
1579
    }
L
Laine Stump 已提交
1580
    if (virFileMakePath(qemu_driver->libDir) != 0) {
1581
        char ebuf[1024];
1582
        VIR_ERROR(_("Failed to create lib dir '%s': %s"),
1583 1584 1585
                  qemu_driver->libDir, virStrerror(errno, ebuf, sizeof ebuf));
        goto error;
    }
L
Laine Stump 已提交
1586
    if (virFileMakePath(qemu_driver->cacheDir) != 0) {
1587
        char ebuf[1024];
1588
        VIR_ERROR(_("Failed to create cache dir '%s': %s"),
1589 1590 1591
                  qemu_driver->cacheDir, virStrerror(errno, ebuf, sizeof ebuf));
        goto error;
    }
1592 1593 1594 1595 1596 1597
    if (virFileMakePath(qemu_driver->saveDir) != 0) {
        char ebuf[1024];
        VIR_ERROR(_("Failed to create save dir '%s': %s"),
                  qemu_driver->saveDir, virStrerror(errno, ebuf, sizeof ebuf));
        goto error;
    }
C
Chris Lalancette 已提交
1598 1599 1600 1601 1602 1603
    if (virFileMakePath(qemu_driver->snapshotDir) != 0) {
        char ebuf[1024];
        VIR_ERROR(_("Failed to create save dir '%s': %s"),
                  qemu_driver->snapshotDir, virStrerror(errno, ebuf, sizeof ebuf));
        goto error;
    }
1604 1605 1606 1607

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

1612
    if (virAsprintf(&qemu_driver->configDir, "%s/qemu", base) == -1)
1613 1614
        goto out_of_memory;

1615
    if (virAsprintf(&qemu_driver->autostartDir, "%s/qemu/autostart", base) == -1)
1616 1617
        goto out_of_memory;

1618
    VIR_FREE(base);
1619

1620 1621 1622 1623 1624 1625 1626
    rc = virCgroupForDriver("qemu", &qemu_driver->cgroup, privileged, 1);
    if (rc < 0) {
        char buf[1024];
        VIR_WARN("Unable to create cgroup for driver: %s",
                 virStrerror(-rc, buf, sizeof(buf)));
    }

1627 1628 1629 1630
    if (qemudLoadDriverConfig(qemu_driver, driverConf) < 0) {
        goto error;
    }

1631 1632
    if (qemudSecurityInit(qemu_driver) < 0)
        goto error;
D
Daniel P. Berrange 已提交
1633

1634
    if ((qemu_driver->caps = qemuCreateCapabilities(NULL,
1635
                                                    qemu_driver)) == NULL)
1636
        goto error;
1637

1638
    if ((qemu_driver->activePciHostdevs = pciDeviceListNew()) == NULL)
1639 1640
        goto error;

1641 1642
    if (privileged) {
        if (chown(qemu_driver->libDir, qemu_driver->user, qemu_driver->group) < 0) {
1643
            virReportSystemError(errno,
1644 1645 1646 1647 1648
                                 _("unable to set ownership of '%s' to user %d:%d"),
                                 qemu_driver->libDir, qemu_driver->user, qemu_driver->group);
            goto error;
        }
        if (chown(qemu_driver->cacheDir, qemu_driver->user, qemu_driver->group) < 0) {
1649
            virReportSystemError(errno,
1650 1651 1652 1653
                                 _("unable to set ownership of '%s' to %d:%d"),
                                 qemu_driver->cacheDir, qemu_driver->user, qemu_driver->group);
            goto error;
        }
1654 1655 1656 1657 1658 1659
        if (chown(qemu_driver->saveDir, qemu_driver->user, qemu_driver->group) < 0) {
            virReportSystemError(errno,
                                 _("unable to set ownership of '%s' to %d:%d"),
                                 qemu_driver->saveDir, qemu_driver->user, qemu_driver->group);
            goto error;
        }
C
Chris Lalancette 已提交
1660 1661 1662 1663 1664 1665
        if (chown(qemu_driver->snapshotDir, qemu_driver->user, qemu_driver->group) < 0) {
            virReportSystemError(errno,
                                 _("unable to set ownership of '%s' to %d:%d"),
                                 qemu_driver->snapshotDir, qemu_driver->user, qemu_driver->group);
            goto error;
        }
1666 1667
    }

1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681
    /* If hugetlbfs is present, then we need to create a sub-directory within
     * it, since we can't assume the root mount point has permissions that
     * will let our spawned QEMU instances use it.
     *
     * NB the check for '/', since user may config "" to disable hugepages
     * even when mounted
     */
    if (qemu_driver->hugetlbfs_mount &&
        qemu_driver->hugetlbfs_mount[0] == '/') {
        char *mempath = NULL;
        if (virAsprintf(&mempath, "%s/libvirt/qemu", qemu_driver->hugetlbfs_mount) < 0)
            goto out_of_memory;

        if ((rc = virFileMakePath(mempath)) != 0) {
1682
            virReportSystemError(rc,
1683 1684 1685 1686 1687 1688
                                 _("unable to create hugepage path %s"), mempath);
            VIR_FREE(mempath);
            goto error;
        }
        if (qemu_driver->privileged &&
            chown(mempath, qemu_driver->user, qemu_driver->group) < 0) {
1689
            virReportSystemError(errno,
1690 1691 1692 1693 1694 1695 1696 1697 1698
                                 _("unable to set ownership on %s to %d:%d"),
                                 mempath, qemu_driver->user, qemu_driver->group);
            VIR_FREE(mempath);
            goto error;
        }

        qemu_driver->hugepage_path = mempath;
    }

1699
    /* Get all the running persistent or transient configs first */
1700
    if (virDomainLoadAllConfigs(qemu_driver->caps,
1701 1702 1703 1704 1705 1706 1707 1708 1709
                                &qemu_driver->domains,
                                qemu_driver->stateDir,
                                NULL,
                                1, NULL, NULL) < 0)
        goto error;

    qemuReconnectDomains(qemu_driver);

    /* Then inactive persistent configs */
1710
    if (virDomainLoadAllConfigs(qemu_driver->caps,
1711 1712
                                &qemu_driver->domains,
                                qemu_driver->configDir,
1713
                                qemu_driver->autostartDir,
1714
                                0, NULL, NULL) < 0)
1715
        goto error;
C
Chris Lalancette 已提交
1716 1717 1718 1719 1720


    virHashForEach(qemu_driver->domains.objs, qemuDomainSnapshotLoad,
                   qemu_driver->snapshotDir);

1721 1722
    qemuDriverUnlock(qemu_driver);

1723 1724
    qemudAutostartConfigs(qemu_driver);

1725

1726 1727
    return 0;

1728
out_of_memory:
1729
    virReportOOMError();
1730 1731 1732
error:
    if (qemu_driver)
        qemuDriverUnlock(qemu_driver);
1733
    VIR_FREE(base);
1734
    qemudShutdown();
1735 1736 1737
    return -1;
}

1738 1739 1740 1741
static void qemudNotifyLoadDomain(virDomainObjPtr vm, int newVM, void *opaque)
{
    struct qemud_driver *driver = opaque;

1742 1743 1744 1745 1746 1747 1748 1749
    if (newVM) {
        virDomainEventPtr event =
            virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_DEFINED,
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED);
        if (event)
            qemuDomainEventQueue(driver, event);
    }
1750 1751
}

1752 1753 1754 1755 1756 1757 1758 1759
/**
 * qemudReload:
 *
 * Function to restart the QEmu daemon, it will recheck the configuration
 * files and update its state and the networking
 */
static int
qemudReload(void) {
1760 1761 1762
    if (!qemu_driver)
        return 0;

1763
    qemuDriverLock(qemu_driver);
1764
    virDomainLoadAllConfigs(qemu_driver->caps,
1765 1766
                            &qemu_driver->domains,
                            qemu_driver->configDir,
1767
                            qemu_driver->autostartDir,
1768
                            0, qemudNotifyLoadDomain, qemu_driver);
1769
    qemuDriverUnlock(qemu_driver);
1770

1771
    qemudAutostartConfigs(qemu_driver);
1772 1773

    return 0;
1774 1775
}

1776 1777 1778 1779 1780 1781 1782 1783 1784 1785
/**
 * 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) {
1786
    int active = 0;
1787

1788 1789 1790
    if (!qemu_driver)
        return 0;

1791
    /* XXX having to iterate here is not great because it requires many locks */
1792
    qemuDriverLock(qemu_driver);
1793
    active = virDomainObjListNumOfDomains(&qemu_driver->domains, 1);
1794 1795
    qemuDriverUnlock(qemu_driver);
    return active;
1796 1797
}

1798 1799 1800 1801 1802 1803 1804
/**
 * qemudShutdown:
 *
 * Shutdown the QEmu daemon, it will stop all active domains and networks
 */
static int
qemudShutdown(void) {
1805
    int i;
1806

1807
    if (!qemu_driver)
1808
        return -1;
1809

1810
    qemuDriverLock(qemu_driver);
1811
    pciDeviceListFree(qemu_driver->activePciHostdevs);
1812 1813
    virCapabilitiesFree(qemu_driver->caps);

1814
    virDomainObjListDeinit(&qemu_driver->domains);
1815
    virBitmapFree(qemu_driver->reservedVNCPorts);
1816

1817
    VIR_FREE(qemu_driver->securityDriverName);
1818
    VIR_FREE(qemu_driver->logDir);
1819 1820
    VIR_FREE(qemu_driver->configDir);
    VIR_FREE(qemu_driver->autostartDir);
1821
    VIR_FREE(qemu_driver->stateDir);
1822 1823
    VIR_FREE(qemu_driver->libDir);
    VIR_FREE(qemu_driver->cacheDir);
1824
    VIR_FREE(qemu_driver->saveDir);
C
Chris Lalancette 已提交
1825
    VIR_FREE(qemu_driver->snapshotDir);
1826
    VIR_FREE(qemu_driver->vncTLSx509certdir);
J
Jim Meyering 已提交
1827
    VIR_FREE(qemu_driver->vncListen);
1828
    VIR_FREE(qemu_driver->vncPassword);
1829
    VIR_FREE(qemu_driver->vncSASLdir);
1830
    VIR_FREE(qemu_driver->saveImageFormat);
1831 1832
    VIR_FREE(qemu_driver->hugetlbfs_mount);
    VIR_FREE(qemu_driver->hugepage_path);
D
Daniel P. Berrange 已提交
1833

1834 1835 1836 1837 1838 1839
    if (qemu_driver->cgroupDeviceACL) {
        for (i = 0 ; qemu_driver->cgroupDeviceACL[i] != NULL ; i++)
            VIR_FREE(qemu_driver->cgroupDeviceACL[i]);
        VIR_FREE(qemu_driver->cgroupDeviceACL);
    }

1840 1841
    /* Free domain callback list */
    virDomainEventCallbackListFree(qemu_driver->domainEventCallbacks);
1842 1843 1844 1845
    virDomainEventQueueFree(qemu_driver->domainEventQueue);

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

1847 1848 1849
    if (qemu_driver->brctl)
        brShutdown(qemu_driver->brctl);

1850 1851
    virCgroupFree(&qemu_driver->cgroup);

1852
    qemuDriverUnlock(qemu_driver);
1853
    virMutexDestroy(&qemu_driver->lock);
1854
    VIR_FREE(qemu_driver);
1855 1856

    return 0;
1857 1858
}

1859
typedef int qemuLogHandleOutput(virDomainObjPtr vm,
1860 1861
                                const char *output,
                                int fd);
1862 1863 1864 1865 1866

/*
 * Returns -1 for error, 0 on success
 */
static int
1867
qemudReadLogOutput(virDomainObjPtr vm,
1868 1869
                   int fd,
                   char *buf,
G
Guido Günther 已提交
1870
                   size_t buflen,
1871
                   qemuLogHandleOutput func,
1872 1873 1874
                   const char *what,
                   int timeout)
{
1875
    int retries = (timeout*10);
1876
    int got = 0;
1877 1878 1879
    buf[0] = '\0';

    while (retries) {
1880
        ssize_t func_ret, ret;
1881
        int isdead = 0;
G
Guido Günther 已提交
1882

1883
        func_ret = func(vm, buf, fd);
1884

1885 1886
        if (kill(vm->pid, 0) == -1 && errno == ESRCH)
            isdead = 1;
1887

1888 1889
        /* Any failures should be detected before we read the log, so we
         * always have something useful to report on failure. */
1890 1891
        ret = saferead(fd, buf+got, buflen-got-1);
        if (ret < 0) {
1892
            virReportSystemError(errno,
1893 1894 1895 1896 1897
                                 _("Failure while reading %s log output"),
                                 what);
            return -1;
        }

1898 1899 1900
        got += ret;
        buf[got] = '\0';
        if (got == buflen-1) {
1901
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
1902 1903
                            _("Out of space while reading %s log output: %s"),
                            what, buf);
1904 1905 1906 1907
            return -1;
        }

        if (isdead) {
1908
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
1909 1910
                            _("Process exited while reading %s log output: %s"),
                            what, buf);
1911 1912 1913
            return -1;
        }

1914 1915
        if (func_ret <= 0)
            return func_ret;
1916 1917 1918 1919

        usleep(100*1000);
        retries--;
    }
1920

1921
    qemuReportError(VIR_ERR_INTERNAL_ERROR,
1922 1923
                    _("Timed out while reading %s log output: %s"),
                    what, buf);
1924 1925 1926
    return -1;
}

1927

1928 1929 1930 1931 1932 1933 1934 1935
/*
 * Look at a chunk of data from the QEMU stdout logs and try to
 * find a TTY device, as indicated by a line like
 *
 * char device redirected to /dev/pts/3
 *
 * Returns -1 for error, 0 success, 1 continue reading
 */
1936
static int
1937
qemudExtractTTYPath(const char *haystack,
1938 1939
                    size_t *offset,
                    char **path)
1940
{
1941
    static const char needle[] = "char device redirected to";
1942
    char *tmp, *dev;
1943

1944
    VIR_FREE(*path);
1945
    /* First look for our magic string */
1946 1947 1948 1949 1950
    if (!(tmp = strstr(haystack + *offset, needle))) {
        return 1;
    }
    tmp += sizeof(needle);
    dev = tmp;
1951

1952 1953 1954 1955 1956
    /*
     * And look for first whitespace character and nul terminate
     * to mark end of the pty path
     */
    while (*tmp) {
1957
        if (c_isspace(*tmp)) {
1958 1959
            *path = strndup(dev, tmp-dev);
            if (*path == NULL) {
1960
                virReportOOMError();
1961 1962
                return -1;
            }
1963

1964
            /* ... now further update offset till we get EOL */
1965
            *offset = tmp - haystack;
1966 1967
            return 0;
        }
1968
        tmp++;
1969 1970 1971 1972 1973
    }

    /*
     * We found a path, but didn't find any whitespace,
     * so it must be still incomplete - we should at
1974 1975
     * least see a \n - indicate that we want to carry
     * on trying again
1976
     */
1977
    return 1;
1978 1979
}

1980
static int
1981
qemudFindCharDevicePTYsMonitor(virDomainObjPtr vm,
1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996
                               virHashTablePtr paths)
{
    int i;

#define LOOKUP_PTYS(array, arraylen, idprefix)                            \
    for (i = 0 ; i < (arraylen) ; i++) {                                  \
        virDomainChrDefPtr chr = (array)[i];                              \
        if (chr->type == VIR_DOMAIN_CHR_TYPE_PTY) {                       \
            char id[16];                                                  \
                                                                          \
            if (snprintf(id, sizeof(id), idprefix "%i", i) >= sizeof(id)) \
                return -1;                                                \
                                                                          \
            const char *path = (const char *) virHashLookup(paths, id);   \
            if (path == NULL) {                                           \
1997 1998 1999
                if (chr->data.file.path == NULL) {                        \
                    /* neither the log output nor 'info chardev' had a */ \
                    /* pty path for this chardev, report an error */      \
2000 2001 2002
                    qemuReportError(VIR_ERR_INTERNAL_ERROR,               \
                                    _("no assigned pty for device %s"), id); \
                    return -1;                                            \
2003 2004 2005 2006 2007
                } else {                                                  \
                    /* 'info chardev' had no pty path for this chardev, */\
                    /* but the log output had, so we're fine */           \
                    continue;                                             \
                }                                                         \
2008 2009
            }                                                             \
                                                                          \
2010
            VIR_FREE(chr->data.file.path);                                \
2011
            chr->data.file.path = strdup(path);                           \
2012 2013
                                                                          \
            if (chr->data.file.path == NULL) {                            \
2014
                virReportOOMError();                                      \
2015 2016
                return -1;                                                \
            }                                                             \
2017 2018 2019 2020 2021 2022
        }                                                                 \
    }

    LOOKUP_PTYS(vm->def->serials,   vm->def->nserials,   "serial");
    LOOKUP_PTYS(vm->def->parallels, vm->def->nparallels, "parallel");
    LOOKUP_PTYS(vm->def->channels,  vm->def->nchannels,  "channel");
2023
#undef LOOKUP_PTYS
2024 2025 2026 2027

    return 0;
}

2028
static int
2029
qemudFindCharDevicePTYs(virDomainObjPtr vm,
2030 2031
                        const char *output,
                        int fd ATTRIBUTE_UNUSED)
2032
{
2033
    size_t offset = 0;
2034
    int ret, i;
2035 2036

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

2040
    /* first comes the serial devices */
2041 2042
    for (i = 0 ; i < vm->def->nserials ; i++) {
        virDomainChrDefPtr chr = vm->def->serials[i];
2043
        if (chr->type == VIR_DOMAIN_CHR_TYPE_PTY) {
2044
            if ((ret = qemudExtractTTYPath(output, &offset,
2045
                                           &chr->data.file.path)) != 0)
2046
                return ret;
2047 2048 2049
        }
    }

2050
    /* then the parallel devices */
2051 2052
    for (i = 0 ; i < vm->def->nparallels ; i++) {
        virDomainChrDefPtr chr = vm->def->parallels[i];
2053
        if (chr->type == VIR_DOMAIN_CHR_TYPE_PTY) {
2054
            if ((ret = qemudExtractTTYPath(output, &offset,
2055
                                           &chr->data.file.path)) != 0)
2056
                return ret;
2057 2058 2059
        }
    }

2060 2061 2062 2063
    /* then the channel devices */
    for (i = 0 ; i < vm->def->nchannels ; i++) {
        virDomainChrDefPtr chr = vm->def->channels[i];
        if (chr->type == VIR_DOMAIN_CHR_TYPE_PTY) {
2064
            if ((ret = qemudExtractTTYPath(output, &offset,
2065 2066 2067 2068 2069
                                           &chr->data.file.path)) != 0)
                return ret;
        }
    }

2070
    return 0;
2071 2072
}

2073 2074 2075 2076 2077
static void qemudFreePtyPath(void *payload, const char *name ATTRIBUTE_UNUSED)
{
    VIR_FREE(payload);
}

2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091
static void
qemuReadLogFD(int logfd, char *buf, int maxlen, int off)
{
    int ret;
    char *tmpbuf = buf + off;

    ret = saferead(logfd, tmpbuf, maxlen - off - 1);
    if (ret < 0) {
        ret = 0;
    }

    tmpbuf[ret] = '\0';
}

2092
static int
2093
qemudWaitForMonitor(struct qemud_driver* driver,
2094
                    virDomainObjPtr vm, off_t pos)
2095
{
2096
    char buf[4096] = ""; /* Plenty of space to get startup greeting */
2097
    int logfd;
2098
    int ret = -1;
2099
    virHashTablePtr paths = NULL;
2100

2101
    if ((logfd = qemudLogReadFD(driver->logDir, vm->def->name, pos)) < 0)
2102
        return -1;
2103

2104 2105 2106 2107
    if (qemudReadLogOutput(vm, logfd, buf, sizeof(buf),
                           qemudFindCharDevicePTYs,
                           "console", 30) < 0)
        goto closelog;
2108

2109
    VIR_DEBUG("Connect monitor to %p '%s'", vm, vm->def->name);
2110 2111 2112
    if (qemuConnectMonitor(driver, vm) < 0) {
        goto cleanup;
    }
2113

2114 2115 2116 2117
    /* Try to get the pty path mappings again via the monitor. This is much more
     * reliable if it's available.
     * Note that the monitor itself can be on a pty, so we still need to try the
     * log output method. */
2118
    paths = virHashCreate(0);
2119
    if (paths == NULL) {
2120
        virReportOOMError();
2121 2122 2123
        goto cleanup;
    }

2124
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
2125 2126
    qemuDomainObjPrivatePtr priv = vm->privateData;
    ret = qemuMonitorGetPtyPaths(priv->mon, paths);
2127
    qemuDomainObjExitMonitorWithDriver(driver, vm);
2128 2129 2130

    VIR_DEBUG("qemuMonitorGetPtyPaths returned %i", ret);
    if (ret == 0) {
2131
        ret = qemudFindCharDevicePTYsMonitor(vm, paths);
2132 2133 2134 2135 2136 2137 2138
    }

cleanup:
    if (paths) {
        virHashFree(paths, qemudFreePtyPath);
    }

2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151
    if (kill(vm->pid, 0) == -1 && errno == ESRCH) {
        /* VM is dead, any other error raised in the interim is probably
         * not as important as the qemu cmdline output */
        qemuReadLogFD(logfd, buf, sizeof(buf), strlen(buf));
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("process exited while connecting to monitor: %s"),
                        buf);
        ret = -1;
    }

closelog:
    if (close(logfd) < 0) {
        char ebuf[4096];
2152
        VIR_WARN("Unable to close logfile: %s",
2153 2154 2155
                 virStrerror(errno, ebuf, sizeof ebuf));
    }

2156
    return ret;
2157 2158
}

2159
static int
2160
qemuDetectVcpuPIDs(struct qemud_driver *driver,
2161 2162 2163
                   virDomainObjPtr vm) {
    pid_t *cpupids = NULL;
    int ncpupids;
2164
    qemuDomainObjPrivatePtr priv = vm->privateData;
2165

2166
    if (vm->def->virtType != VIR_DOMAIN_VIRT_KVM) {
2167 2168
        priv->nvcpupids = 1;
        if (VIR_ALLOC_N(priv->vcpupids, priv->nvcpupids) < 0) {
2169
            virReportOOMError();
2170 2171
            return -1;
        }
2172
        priv->vcpupids[0] = vm->pid;
2173 2174 2175
        return 0;
    }

2176
    /* What follows is now all KVM specific */
2177

2178
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
2179
    if ((ncpupids = qemuMonitorGetCPUInfo(priv->mon, &cpupids)) < 0) {
2180
        qemuDomainObjExitMonitorWithDriver(driver, vm);
2181
        return -1;
2182
    }
2183
    qemuDomainObjExitMonitorWithDriver(driver, vm);
2184

2185 2186 2187
    /* Treat failure to get VCPU<->PID mapping as non-fatal */
    if (ncpupids == 0)
        return 0;
2188

2189
    if (ncpupids != vm->def->vcpus) {
2190 2191 2192
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("got wrong number of vCPU pids from QEMU monitor. got %d, wanted %d"),
                        ncpupids, (int)vm->def->vcpus);
2193 2194 2195
        VIR_FREE(cpupids);
        return -1;
    }
2196

2197 2198
    priv->nvcpupids = ncpupids;
    priv->vcpupids = cpupids;
2199 2200 2201
    return 0;
}

2202 2203 2204
/*
 * To be run between fork/exec of QEMU only
 */
2205
static int
2206 2207
qemudInitCpuAffinity(virDomainObjPtr vm)
{
2208
    int i, hostcpus, maxcpu = QEMUD_CPUMASK_LEN;
2209
    virNodeInfo nodeinfo;
2210 2211
    unsigned char *cpumap;
    int cpumaplen;
2212 2213

    DEBUG0("Setting CPU affinity");
2214

2215
    if (nodeGetInfo(NULL, &nodeinfo) < 0)
2216 2217 2218 2219
        return -1;

    /* setaffinity fails if you set bits for CPUs which
     * aren't present, so we have to limit ourselves */
2220 2221 2222
    hostcpus = VIR_NODEINFO_MAXCPUS(nodeinfo);
    if (maxcpu > hostcpus)
        maxcpu = hostcpus;
2223

2224 2225
    cpumaplen = VIR_CPU_MAPLEN(maxcpu);
    if (VIR_ALLOC_N(cpumap, cpumaplen) < 0) {
2226
        virReportOOMError();
2227 2228 2229
        return -1;
    }

D
Daniel P. Berrange 已提交
2230
    if (vm->def->cpumask) {
2231 2232 2233
        /* XXX why don't we keep 'cpumask' in the libvirt cpumap
         * format to start with ?!?! */
        for (i = 0 ; i < maxcpu && i < vm->def->cpumasklen ; i++)
D
Daniel P. Berrange 已提交
2234
            if (vm->def->cpumask[i])
2235
                VIR_USE_CPU(cpumap, i);
D
Daniel P. Berrange 已提交
2236
    } else {
2237 2238 2239 2240
        /* You may think this is redundant, but we can't assume libvirtd
         * itself is running on all pCPUs, so we need to explicitly set
         * the spawned QEMU instance to all pCPUs if no map is given in
         * its config file */
D
Daniel P. Berrange 已提交
2241
        for (i = 0 ; i < maxcpu ; i++)
2242
            VIR_USE_CPU(cpumap, i);
D
Daniel P. Berrange 已提交
2243
    }
2244

2245 2246 2247 2248 2249 2250 2251 2252
    /* We are pressuming we are running between fork/exec of QEMU
     * so use '0' to indicate our own process ID. No threads are
     * running at this point
     */
    if (virProcessInfoSetAffinity(0, /* Self */
                                  cpumap, cpumaplen, maxcpu) < 0) {
        VIR_FREE(cpumap);
        return -1;
2253
    }
2254
    VIR_FREE(cpumap);
2255 2256 2257 2258 2259

    return 0;
}


2260
static int
2261 2262 2263 2264
qemuInitPasswords(virConnectPtr conn,
                  struct qemud_driver *driver,
                  virDomainObjPtr vm,
                  unsigned long long qemuCmdFlags) {
2265
    int ret = 0;
2266
    qemuDomainObjPrivatePtr priv = vm->privateData;
2267

2268 2269 2270
    if ((vm->def->ngraphics == 1) &&
        vm->def->graphics[0]->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
        (vm->def->graphics[0]->data.vnc.passwd || driver->vncPassword)) {
2271

2272
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
2273
        ret = qemuMonitorSetVNCPassword(priv->mon,
2274 2275 2276
                                        vm->def->graphics[0]->data.vnc.passwd ?
                                        vm->def->graphics[0]->data.vnc.passwd :
                                        driver->vncPassword);
2277
        qemuDomainObjExitMonitorWithDriver(driver, vm);
2278 2279
    }

2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302
    if (ret < 0)
        goto cleanup;

    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        int i;

        for (i = 0 ; i < vm->def->ndisks ; i++) {
            char *secret;
            size_t secretLen;

            if (!vm->def->disks[i]->encryption ||
                !vm->def->disks[i]->src)
                continue;

            if (getVolumeQcowPassphrase(conn,
                                        vm->def->disks[i],
                                        &secret, &secretLen) < 0)
                goto cleanup;

            qemuDomainObjEnterMonitorWithDriver(driver, vm);
            ret = qemuMonitorSetDrivePassphrase(priv->mon,
                                                vm->def->disks[i]->info.alias,
                                                secret);
2303
            VIR_FREE(secret);
2304 2305 2306 2307 2308 2309 2310
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            if (ret < 0)
                goto cleanup;
        }
    }

cleanup:
2311
    return ret;
2312 2313 2314
}


2315 2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542 2543
#define QEMU_PCI_VENDOR_INTEL     0x8086
#define QEMU_PCI_VENDOR_LSI_LOGIC 0x1000
#define QEMU_PCI_VENDOR_REDHAT    0x1af4
#define QEMU_PCI_VENDOR_CIRRUS    0x1013
#define QEMU_PCI_VENDOR_REALTEK   0x10ec
#define QEMU_PCI_VENDOR_AMD       0x1022
#define QEMU_PCI_VENDOR_ENSONIQ   0x1274
#define QEMU_PCI_VENDOR_VMWARE    0x15ad
#define QEMU_PCI_VENDOR_QEMU      0x1234

#define QEMU_PCI_PRODUCT_DISK_VIRTIO 0x1001

#define QEMU_PCI_PRODUCT_NIC_NE2K     0x8029
#define QEMU_PCI_PRODUCT_NIC_PCNET    0x2000
#define QEMU_PCI_PRODUCT_NIC_RTL8139  0x8139
#define QEMU_PCI_PRODUCT_NIC_E1000    0x100E
#define QEMU_PCI_PRODUCT_NIC_VIRTIO   0x1000

#define QEMU_PCI_PRODUCT_VGA_CIRRUS 0x00b8
#define QEMU_PCI_PRODUCT_VGA_VMWARE 0x0405
#define QEMU_PCI_PRODUCT_VGA_STDVGA 0x1111

#define QEMU_PCI_PRODUCT_AUDIO_AC97    0x2415
#define QEMU_PCI_PRODUCT_AUDIO_ES1370  0x5000

#define QEMU_PCI_PRODUCT_CONTROLLER_PIIX 0x7010
#define QEMU_PCI_PRODUCT_CONTROLLER_LSI  0x0012

#define QEMU_PCI_PRODUCT_WATCHDOG_I63000ESB 0x25ab

static int
qemuAssignNextPCIAddress(virDomainDeviceInfo *info,
                         int vendor,
                         int product,
                         qemuMonitorPCIAddress *addrs,
                         int naddrs)
{
    int found = 0;
    int i;

    VIR_DEBUG("Look for %x:%x out of %d", vendor, product, naddrs);

    for (i = 0 ; (i < naddrs) && !found; i++) {
        VIR_DEBUG("Maybe %x:%x", addrs[i].vendor, addrs[i].product);
        if (addrs[i].vendor == vendor &&
            addrs[i].product == product) {
            VIR_DEBUG("Match %d", i);
            found = 1;
            break;
        }
    }
    if (!found) {
        return -1;
    }

    /* Blank it out so this device isn't matched again */
    addrs[i].vendor = 0;
    addrs[i].product = 0;

    if (info->type == VIR_DOMAIN_DEVICE_ADDRESS_TYPE_NONE)
        info->type = VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI;

    if (info->type == VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI) {
        info->addr.pci.domain = addrs[i].addr.domain;
        info->addr.pci.bus = addrs[i].addr.bus;
        info->addr.pci.slot = addrs[i].addr.slot;
        info->addr.pci.function = addrs[i].addr.function;
    }

    return 0;
}

static int
qemuGetPCIDiskVendorProduct(virDomainDiskDefPtr def,
                            unsigned *vendor,
                            unsigned *product)
{
    switch (def->bus) {
    case VIR_DOMAIN_DISK_BUS_VIRTIO:
        *vendor = QEMU_PCI_VENDOR_REDHAT;
        *product = QEMU_PCI_PRODUCT_DISK_VIRTIO;
        break;

    default:
        return -1;
    }

    return 0;
}

static int
qemuGetPCINetVendorProduct(virDomainNetDefPtr def,
                            unsigned *vendor,
                            unsigned *product)
{
    if (!def->model)
        return -1;

    if (STREQ(def->model, "ne2k_pci")) {
        *vendor = QEMU_PCI_VENDOR_REALTEK;
        *product = QEMU_PCI_PRODUCT_NIC_NE2K;
    } else if (STREQ(def->model, "pcnet")) {
        *vendor = QEMU_PCI_VENDOR_AMD;
        *product = QEMU_PCI_PRODUCT_NIC_PCNET;
    } else if (STREQ(def->model, "rtl8139")) {
        *vendor = QEMU_PCI_VENDOR_REALTEK;
        *product = QEMU_PCI_PRODUCT_NIC_RTL8139;
    } else if (STREQ(def->model, "e1000")) {
        *vendor = QEMU_PCI_VENDOR_INTEL;
        *product = QEMU_PCI_PRODUCT_NIC_E1000;
    } else if (STREQ(def->model, "virtio")) {
        *vendor = QEMU_PCI_VENDOR_REDHAT;
        *product = QEMU_PCI_PRODUCT_NIC_VIRTIO;
    } else {
        VIR_INFO("Unexpected NIC model %s, cannot get PCI address",
                 def->model);
        return -1;
    }
    return 0;
}

static int
qemuGetPCIControllerVendorProduct(virDomainControllerDefPtr def,
                                  unsigned *vendor,
                                  unsigned *product)
{
    switch (def->type) {
    case VIR_DOMAIN_CONTROLLER_TYPE_SCSI:
        *vendor = QEMU_PCI_VENDOR_LSI_LOGIC;
        *product = QEMU_PCI_PRODUCT_CONTROLLER_LSI;
        break;

    case VIR_DOMAIN_CONTROLLER_TYPE_FDC:
        /* XXX we could put in the ISA bridge address, but
           that's not technically the FDC's address */
        return -1;

    case VIR_DOMAIN_CONTROLLER_TYPE_IDE:
        *vendor = QEMU_PCI_VENDOR_INTEL;
        *product = QEMU_PCI_PRODUCT_CONTROLLER_PIIX;
        break;

    default:
        VIR_INFO("Unexpected controller type %s, cannot get PCI address",
                 virDomainControllerTypeToString(def->type));
        return -1;
    }

    return 0;
}

static int
qemuGetPCIVideoVendorProduct(virDomainVideoDefPtr def,
                             unsigned *vendor,
                             unsigned *product)
{
    switch (def->type) {
    case VIR_DOMAIN_VIDEO_TYPE_CIRRUS:
        *vendor = QEMU_PCI_VENDOR_CIRRUS;
        *product = QEMU_PCI_PRODUCT_VGA_CIRRUS;
        break;

    case VIR_DOMAIN_VIDEO_TYPE_VGA:
        *vendor = QEMU_PCI_VENDOR_QEMU;
        *product = QEMU_PCI_PRODUCT_VGA_STDVGA;
        break;

    case VIR_DOMAIN_VIDEO_TYPE_VMVGA:
        *vendor = QEMU_PCI_VENDOR_VMWARE;
        *product = QEMU_PCI_PRODUCT_VGA_VMWARE;
        break;

    default:
        return -1;
    }
    return 0;
}

static int
qemuGetPCISoundVendorProduct(virDomainSoundDefPtr def,
                             unsigned *vendor,
                             unsigned *product)
{
    switch (def->model) {
    case VIR_DOMAIN_SOUND_MODEL_ES1370:
        *vendor = QEMU_PCI_VENDOR_ENSONIQ;
        *product = QEMU_PCI_PRODUCT_AUDIO_ES1370;
        break;

    case VIR_DOMAIN_SOUND_MODEL_AC97:
        *vendor = QEMU_PCI_VENDOR_INTEL;
        *product = QEMU_PCI_PRODUCT_AUDIO_AC97;
        break;

    default:
        return -1;
    }

    return 0;
}

static int
qemuGetPCIWatchdogVendorProduct(virDomainWatchdogDefPtr def,
                                unsigned *vendor,
                                unsigned *product)
{
    switch (def->model) {
    case VIR_DOMAIN_WATCHDOG_MODEL_I6300ESB:
        *vendor = QEMU_PCI_VENDOR_INTEL;
        *product = QEMU_PCI_PRODUCT_WATCHDOG_I63000ESB;
        break;

    default:
        return -1;
    }

    return 0;
}


/*
 * This entire method assumes that PCI devices in 'info pci'
 * match ordering of devices specified on the command line
 * wrt to devices of matching vendor+product
 *
 * XXXX this might not be a valid assumption if we assign
 * some static addrs on CLI. Have to check that...
 */
static int
2544
qemuDetectPCIAddresses(virDomainObjPtr vm,
2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561
                       qemuMonitorPCIAddress *addrs,
                       int naddrs)
{
    unsigned int vendor = 0, product = 0;
    int i;

    /* XXX should all these vendor/product IDs be kept in the
     * actual device data structure instead ?
     */

    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (qemuGetPCIDiskVendorProduct(vm->def->disks[i], &vendor, &product) < 0)
            continue;

        if (qemuAssignNextPCIAddress(&(vm->def->disks[i]->info),
                                     vendor, product,
                                     addrs, naddrs) < 0) {
2562 2563 2564
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("cannot find PCI address for VirtIO disk %s"),
                            vm->def->disks[i]->dst);
2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575
            return -1;
        }
    }

    for (i = 0 ; i < vm->def->nnets ; i++) {
        if (qemuGetPCINetVendorProduct(vm->def->nets[i], &vendor, &product) < 0)
            continue;

        if (qemuAssignNextPCIAddress(&(vm->def->nets[i]->info),
                                     vendor, product,
                                     addrs,  naddrs) < 0) {
2576 2577 2578
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("cannot find PCI address for %s NIC"),
                            vm->def->nets[i]->model);
2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589
            return -1;
        }
    }

    for (i = 0 ; i < vm->def->ncontrollers ; i++) {
        if (qemuGetPCIControllerVendorProduct(vm->def->controllers[i], &vendor, &product) < 0)
            continue;

        if (qemuAssignNextPCIAddress(&(vm->def->controllers[i]->info),
                                     vendor, product,
                                     addrs,  naddrs) < 0) {
2590 2591 2592
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("cannot find PCI address for controller %s"),
                            virDomainControllerTypeToString(vm->def->controllers[i]->type));
2593 2594 2595 2596 2597 2598 2599 2600 2601 2602 2603
            return -1;
        }
    }

    for (i = 0 ; i < vm->def->nvideos ; i++) {
        if (qemuGetPCIVideoVendorProduct(vm->def->videos[i], &vendor, &product) < 0)
            continue;

        if (qemuAssignNextPCIAddress(&(vm->def->videos[i]->info),
                                     vendor, product,
                                     addrs,  naddrs) < 0) {
2604 2605 2606
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("cannot find PCI address for video adapter %s"),
                            virDomainVideoTypeToString(vm->def->videos[i]->type));
2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617
            return -1;
        }
    }

    for (i = 0 ; i < vm->def->nsounds ; i++) {
        if (qemuGetPCISoundVendorProduct(vm->def->sounds[i], &vendor, &product) < 0)
            continue;

        if (qemuAssignNextPCIAddress(&(vm->def->sounds[i]->info),
                                     vendor, product,
                                     addrs,  naddrs) < 0) {
2618 2619 2620
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("cannot find PCI address for sound adapter %s"),
                            virDomainSoundModelTypeToString(vm->def->sounds[i]->model));
2621 2622 2623 2624 2625 2626 2627 2628 2629 2630
            return -1;
        }
    }


    if (vm->def->watchdog &&
        qemuGetPCIWatchdogVendorProduct(vm->def->watchdog, &vendor, &product) == 0) {
        if (qemuAssignNextPCIAddress(&(vm->def->watchdog->info),
                                     vendor, product,
                                     addrs,  naddrs) < 0) {
2631 2632 2633
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("cannot find PCI address for watchdog %s"),
                            virDomainWatchdogModelTypeToString(vm->def->watchdog->model));
2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665
            return -1;
        }
    }

    /* XXX console (virtio) */


    /* ... and now things we don't have in our xml */

    /* XXX USB controller ? */

    /* XXXX virtio balloon ? */

    /* XXX what about other PCI devices (ie bridges) */

    return 0;
}

static int
qemuInitPCIAddresses(struct qemud_driver *driver,
                     virDomainObjPtr vm)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int naddrs;
    int ret;
    qemuMonitorPCIAddress *addrs = NULL;

    qemuDomainObjEnterMonitorWithDriver(driver, vm);
    naddrs = qemuMonitorGetAllPCIAddresses(priv->mon,
                                           &addrs);
    qemuDomainObjExitMonitorWithDriver(driver, vm);

2666
    ret = qemuDetectPCIAddresses(vm, addrs, naddrs);
2667 2668 2669 2670 2671 2672

    VIR_FREE(addrs);

    return ret;
}

2673
static int qemudNextFreeVNCPort(struct qemud_driver *driver) {
2674 2675
    int i;

2676
    for (i = QEMU_VNC_PORT_MIN; i < QEMU_VNC_PORT_MAX; i++) {
2677 2678 2679
        int fd;
        int reuse = 1;
        struct sockaddr_in addr;
2680 2681 2682 2683 2684 2685 2686 2687 2688
        bool used = false;

        if (virBitmapGetBit(driver->reservedVNCPorts,
                            i - QEMU_VNC_PORT_MIN, &used) < 0)
            VIR_DEBUG("virBitmapGetBit failed on bit %d", i - QEMU_VNC_PORT_MIN);

        if (used)
            continue;

2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703
        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);
2704 2705 2706 2707 2708 2709
            /* Add port to bitmap of reserved ports */
            if (virBitmapSetBit(driver->reservedVNCPorts,
                                i - QEMU_VNC_PORT_MIN) < 0) {
                VIR_DEBUG("virBitmapSetBit failed on bit %d",
                          i - QEMU_VNC_PORT_MIN);
            }
2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723
            return i;
        }
        close(fd);

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

2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761

static int
qemuAssignPCIAddresses(virDomainDefPtr def)
{
    int ret = -1;
    unsigned long long qemuCmdFlags = 0;
    qemuDomainPCIAddressSetPtr addrs = NULL;
    struct stat sb;

    if (stat(def->emulator, &sb) < 0) {
        virReportSystemError(errno,
                             _("Cannot find QEMU binary %s"),
                             def->emulator);
        goto cleanup;
    }

    if (qemudExtractVersionInfo(def->emulator,
                                NULL,
                                &qemuCmdFlags) < 0)
        goto cleanup;

    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        if (!(addrs = qemuDomainPCIAddressSetCreate(def)))
            goto cleanup;

        if (qemuAssignDevicePCISlots(def, addrs) < 0)
            goto cleanup;
    }

    ret = 0;

cleanup:
    qemuDomainPCIAddressSetFree(addrs);

    return ret;
}


2762
static pciDeviceList *
2763
qemuGetPciHostDeviceList(virDomainDefPtr def)
2764 2765
{
    pciDeviceList *list;
2766 2767
    int i;

2768
    if (!(list = pciDeviceListNew()))
2769
        return NULL;
2770 2771 2772 2773 2774 2775 2776 2777 2778 2779

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

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

2780
        dev = pciGetDevice(hostdev->source.subsys.u.pci.domain,
2781 2782 2783
                           hostdev->source.subsys.u.pci.bus,
                           hostdev->source.subsys.u.pci.slot,
                           hostdev->source.subsys.u.pci.function);
2784
        if (!dev) {
2785
            pciDeviceListFree(list);
2786 2787
            return NULL;
        }
2788

2789 2790 2791
        if (pciDeviceListAdd(list, dev) < 0) {
            pciFreeDevice(dev);
            pciDeviceListFree(list);
2792
            return NULL;
2793 2794
        }

2795
        pciDeviceSetManaged(dev, hostdev->managed);
2796 2797
    }

2798 2799 2800 2801
    return list;
}

static int
2802 2803 2804 2805
qemuUpdateActivePciHostdevs(struct qemud_driver *driver,
                            virDomainDefPtr def)
{
    pciDeviceList *pcidevs;
2806
    int i;
2807
    int ret = -1;
2808 2809 2810 2811

    if (!def->nhostdevs)
        return 0;

2812
    if (!(pcidevs = qemuGetPciHostDeviceList(def)))
2813 2814
        return -1;

2815 2816
    for (i = 0; i < pciDeviceListCount(pcidevs); i++) {
        pciDevice *dev = pciDeviceListGet(pcidevs, i);
2817 2818 2819
        pciDeviceListSteal(pcidevs, dev);
        if (pciDeviceListAdd(driver->activePciHostdevs, dev) < 0) {
            pciFreeDevice(dev);
2820
            goto cleanup;
2821 2822 2823
        }
    }

2824 2825 2826
    ret = 0;

cleanup:
2827
    pciDeviceListFree(pcidevs);
2828 2829 2830
    return ret;
}

2831

2832
static int
2833 2834
qemuPrepareHostPCIDevices(struct qemud_driver *driver,
                          virDomainDefPtr def)
2835 2836 2837
{
    pciDeviceList *pcidevs;
    int i;
2838
    int ret = -1;
2839

2840
    if (!(pcidevs = qemuGetPciHostDeviceList(def)))
2841 2842
        return -1;

2843
    /* We have to use 3 loops here. *All* devices must
2844 2845
     * be detached before we reset any of them, because
     * in some cases you have to reset the whole PCI,
2846 2847
     * which impacts all devices on it. Also, all devices
     * must be reset before being marked as active.
2848 2849 2850 2851 2852 2853 2854
     */

    /* XXX validate that non-managed device isn't in use, eg
     * by checking that device is either un-bound, or bound
     * to pci-stub.ko
     */

2855 2856
    for (i = 0; i < pciDeviceListCount(pcidevs); i++) {
        pciDevice *dev = pciDeviceListGet(pcidevs, i);
2857
        if (!pciDeviceIsAssignable(dev, !driver->relaxedACS))
2858 2859
            goto cleanup;

2860
        if (pciDeviceGetManaged(dev) &&
2861
            pciDettachDevice(dev) < 0)
2862 2863
            goto cleanup;
    }
2864 2865 2866

    /* Now that all the PCI hostdevs have be dettached, we can safely
     * reset them */
2867 2868
    for (i = 0; i < pciDeviceListCount(pcidevs); i++) {
        pciDevice *dev = pciDeviceListGet(pcidevs, i);
2869
        if (pciResetDevice(dev, driver->activePciHostdevs) < 0)
2870 2871
            goto cleanup;
    }
2872

2873
    /* Now mark all the devices as active */
2874 2875
    for (i = 0; i < pciDeviceListCount(pcidevs); i++) {
        pciDevice *dev = pciDeviceListGet(pcidevs, i);
2876 2877 2878
        pciDeviceListSteal(pcidevs, dev);
        if (pciDeviceListAdd(driver->activePciHostdevs, dev) < 0) {
            pciFreeDevice(dev);
2879 2880
            goto cleanup;
        }
2881 2882
    }

2883
    ret = 0;
2884

2885
cleanup:
2886
    pciDeviceListFree(pcidevs);
2887
    return ret;
2888 2889
}

2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900

static int
qemuPrepareHostUSBDevices(struct qemud_driver *driver ATTRIBUTE_UNUSED,
                          virDomainDefPtr def)
{
    int i;
    for (i = 0 ; i < def->nhostdevs ; i++) {
        virDomainHostdevDefPtr hostdev = def->hostdevs[i];

        if (hostdev->mode != VIR_DOMAIN_HOSTDEV_MODE_SUBSYS)
            continue;
2901
        if (hostdev->source.subsys.type != VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB)
2902 2903 2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939
            continue;

        /* Resolve a vendor/product to bus/device */
        if (hostdev->source.subsys.u.usb.vendor) {
            usbDevice *usb
                = usbFindDevice(hostdev->source.subsys.u.usb.vendor,
                                hostdev->source.subsys.u.usb.product);

            if (!usb)
                return -1;

            hostdev->source.subsys.u.usb.bus = usbDeviceGetBus(usb);
            hostdev->source.subsys.u.usb.device = usbDeviceGetDevno(usb);

            usbFreeDevice(usb);
        }
    }

    return 0;
}

static int
qemuPrepareHostDevices(struct qemud_driver *driver,
                       virDomainDefPtr def)
{
    if (!def->nhostdevs)
        return 0;

    if (qemuPrepareHostPCIDevices(driver, def) < 0)
        return -1;

    if (qemuPrepareHostUSBDevices(driver, def) < 0)
        return -1;

    return 0;
}


2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950
static void
qemudReattachManagedDevice(pciDevice *dev)
{
    int retries = 100;

    if (pciDeviceGetManaged(dev)) {
        while (pciWaitForDeviceCleanup(dev, "kvm_assigned_device")
               && retries) {
            usleep(100*1000);
            retries--;
        }
2951
        if (pciReAttachDevice(dev) < 0) {
2952 2953
            virErrorPtr err = virGetLastError();
            VIR_ERROR(_("Failed to re-attach PCI device: %s"),
2954
                      err ? err->message : _("unknown error"));
2955 2956 2957 2958 2959
            virResetError(err);
        }
    }
}

2960
static void
2961
qemuDomainReAttachHostDevices(struct qemud_driver *driver,
2962
                              virDomainDefPtr def)
2963
{
2964
    pciDeviceList *pcidevs;
2965 2966
    int i;

2967 2968
    if (!def->nhostdevs)
        return;
2969

2970
    if (!(pcidevs = qemuGetPciHostDeviceList(def))) {
2971
        virErrorPtr err = virGetLastError();
2972
        VIR_ERROR(_("Failed to allocate pciDeviceList: %s"),
2973
                  err ? err->message : _("unknown error"));
2974 2975
        virResetError(err);
        return;
2976 2977
    }

2978 2979
    /* Again 3 loops; mark all devices as inactive before reset
     * them and reset all the devices before re-attach */
2980

2981 2982
    for (i = 0; i < pciDeviceListCount(pcidevs); i++) {
        pciDevice *dev = pciDeviceListGet(pcidevs, i);
2983
        pciDeviceListDel(driver->activePciHostdevs, dev);
2984
    }
2985

2986 2987
    for (i = 0; i < pciDeviceListCount(pcidevs); i++) {
        pciDevice *dev = pciDeviceListGet(pcidevs, i);
2988
        if (pciResetDevice(dev, driver->activePciHostdevs) < 0) {
2989
            virErrorPtr err = virGetLastError();
2990
            VIR_ERROR(_("Failed to reset PCI device: %s"),
2991
                      err ? err->message : _("unknown error"));
2992 2993
            virResetError(err);
        }
2994
    }
2995

2996 2997
    for (i = 0; i < pciDeviceListCount(pcidevs); i++) {
        pciDevice *dev = pciDeviceListGet(pcidevs, i);
2998
        qemudReattachManagedDevice(dev);
2999
    }
3000

3001
    pciDeviceListFree(pcidevs);
3002 3003
}

3004 3005 3006 3007 3008 3009 3010 3011 3012 3013
static const char *const defaultDeviceACL[] = {
    "/dev/null", "/dev/full", "/dev/zero",
    "/dev/random", "/dev/urandom",
    "/dev/ptmx", "/dev/kvm", "/dev/kqemu",
    "/dev/rtc", "/dev/hpet", "/dev/net/tun",
    NULL,
};
#define DEVICE_PTY_MAJOR 136
#define DEVICE_SND_MAJOR 116

3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030
static int qemuSetupDiskCgroup(virCgroupPtr cgroup,
                               virDomainObjPtr vm,
                               virDomainDiskDefPtr disk)
{
    char *path = disk->src;
    int ret = -1;

    while (path != NULL) {
        virStorageFileMetadata meta;
        int rc;

        VIR_DEBUG("Process path %s for disk", path);
        rc = virCgroupAllowDevicePath(cgroup, path);
        if (rc != 0) {
            /* Get this for non-block devices */
            if (rc == -EINVAL) {
                VIR_DEBUG("Ignoring EINVAL for %s", path);
3031 3032
            } else if (rc == -EACCES) { /* Get this for root squash NFS */
                VIR_DEBUG("Ignoring EACCES for %s", path);
3033 3034 3035 3036 3037 3038 3039 3040 3041 3042 3043
            } else {
                virReportSystemError(-rc,
                                     _("Unable to allow device %s for %s"),
                                     path, vm->def->name);
                if (path != disk->src)
                    VIR_FREE(path);
                goto cleanup;
            }
        }

        rc = virStorageFileGetMetadata(path, &meta);
3044 3045
        if (rc < 0)
            VIR_WARN("Unable to lookup parent image for %s", path);
3046 3047 3048 3049 3050 3051

        if (path != disk->src)
            VIR_FREE(path);
        path = NULL;

        if (rc < 0)
3052
            break; /* Treating as non fatal */
3053 3054

        path = meta.backingStore;
3055
    }
3056 3057 3058 3059 3060 3061 3062 3063 3064 3065 3066 3067 3068 3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079 3080

    ret = 0;

cleanup:
    return ret;
}


static int qemuTeardownDiskCgroup(virCgroupPtr cgroup,
                                  virDomainObjPtr vm,
                                  virDomainDiskDefPtr disk)
{
    char *path = disk->src;
    int ret = -1;

    while (path != NULL) {
        virStorageFileMetadata meta;
        int rc;

        VIR_DEBUG("Process path %s for disk", path);
        rc = virCgroupDenyDevicePath(cgroup, path);
        if (rc != 0) {
            /* Get this for non-block devices */
            if (rc == -EINVAL) {
                VIR_DEBUG("Ignoring EINVAL for %s", path);
3081 3082
            } else if (rc == -EACCES) { /* Get this for root squash NFS */
                VIR_DEBUG("Ignoring EACCES for %s", path);
3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093
            } else {
                virReportSystemError(-rc,
                                     _("Unable to deny device %s for %s"),
                                     path, vm->def->name);
                if (path != disk->src)
                    VIR_FREE(path);
                goto cleanup;
            }
        }

        rc = virStorageFileGetMetadata(path, &meta);
3094 3095
        if (rc < 0)
            VIR_WARN("Unable to lookup parent image for %s", path);
3096 3097 3098 3099 3100 3101

        if (path != disk->src)
            VIR_FREE(path);
        path = NULL;

        if (rc < 0)
3102
            break; /* Treating as non fatal */
3103 3104

        path = meta.backingStore;
3105
    }
3106 3107 3108 3109 3110 3111 3112 3113

    ret = 0;

cleanup:
    return ret;
}


3114
static int qemuSetupCgroup(struct qemud_driver *driver,
3115 3116 3117 3118
                           virDomainObjPtr vm)
{
    virCgroupPtr cgroup = NULL;
    int rc;
3119
    unsigned int i;
3120 3121 3122 3123
    const char *const *deviceACL =
        driver->cgroupDeviceACL ?
        (const char *const *)driver->cgroupDeviceACL :
        defaultDeviceACL;
3124 3125 3126 3127 3128 3129

    if (driver->cgroup == NULL)
        return 0; /* Not supported, so claim success */

    rc = virCgroupForDomain(driver->cgroup, vm->def->name, &cgroup, 1);
    if (rc != 0) {
3130
        virReportSystemError(-rc,
3131 3132 3133 3134 3135
                             _("Unable to create cgroup for %s"),
                             vm->def->name);
        goto cleanup;
    }

3136 3137
    if (qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_DEVICES)) {
        rc = virCgroupDenyAllDevices(cgroup);
3138
        if (rc != 0) {
3139 3140 3141 3142 3143
            if (rc == -EPERM) {
                VIR_WARN0("Group devices ACL is not accessible, disabling whitelisting");
                goto done;
            }

3144
            virReportSystemError(-rc,
3145
                                 _("Unable to deny all devices for %s"), vm->def->name);
3146 3147 3148
            goto cleanup;
        }

3149
        for (i = 0; i < vm->def->ndisks ; i++) {
3150
            if (qemuSetupDiskCgroup(cgroup, vm, vm->def->disks[i]) < 0)
3151 3152
                goto cleanup;
        }
3153

3154
        rc = virCgroupAllowDeviceMajor(cgroup, 'c', DEVICE_PTY_MAJOR);
3155
        if (rc != 0) {
3156
            virReportSystemError(-rc, "%s",
3157
                                 _("unable to allow /dev/pts/ devices"));
3158 3159 3160
            goto cleanup;
        }

3161 3162 3163
        if (vm->def->nsounds) {
            rc = virCgroupAllowDeviceMajor(cgroup, 'c', DEVICE_SND_MAJOR);
            if (rc != 0) {
3164
                virReportSystemError(-rc, "%s",
3165 3166 3167 3168 3169 3170 3171 3172 3173 3174
                                     _("unable to allow /dev/snd/ devices"));
                goto cleanup;
            }
        }

        for (i = 0; deviceACL[i] != NULL ; i++) {
            rc = virCgroupAllowDevicePath(cgroup,
                                          deviceACL[i]);
            if (rc < 0 &&
                rc != -ENOENT) {
3175
                virReportSystemError(-rc,
3176 3177 3178 3179
                                     _("unable to allow device %s"),
                                     deviceACL[i]);
                goto cleanup;
            }
3180 3181 3182 3183
        }
    }

done:
3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195
    virCgroupFree(&cgroup);
    return 0;

cleanup:
    if (cgroup) {
        virCgroupRemove(cgroup);
        virCgroupFree(&cgroup);
    }
    return -1;
}


3196
static int qemuRemoveCgroup(struct qemud_driver *driver,
3197 3198
                            virDomainObjPtr vm,
                            int quiet)
3199 3200 3201 3202 3203 3204 3205 3206 3207
{
    virCgroupPtr cgroup;
    int rc;

    if (driver->cgroup == NULL)
        return 0; /* Not supported, so claim success */

    rc = virCgroupForDomain(driver->cgroup, vm->def->name, &cgroup, 0);
    if (rc != 0) {
3208
        if (!quiet)
3209 3210 3211
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("Unable to find cgroup for %s\n"),
                            vm->def->name);
3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226 3227 3228 3229 3230 3231
        return rc;
    }

    rc = virCgroupRemove(cgroup);
    virCgroupFree(&cgroup);
    return rc;
}

static int qemuAddToCgroup(struct qemud_driver *driver,
                           virDomainDefPtr def)
{
    virCgroupPtr cgroup = NULL;
    int ret = -1;
    int rc;

    if (driver->cgroup == NULL)
        return 0; /* Not supported, so claim success */

    rc = virCgroupForDomain(driver->cgroup, def->name, &cgroup, 0);
    if (rc != 0) {
3232
        virReportSystemError(-rc,
3233 3234 3235 3236 3237 3238 3239
                             _("unable to find cgroup for domain %s"),
                             def->name);
        goto cleanup;
    }

    rc = virCgroupAddTask(cgroup, getpid());
    if (rc != 0) {
3240
        virReportSystemError(-rc,
3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257
                             _("unable to add domain %s task %d to cgroup"),
                             def->name, getpid());
        goto cleanup;
    }

    ret = 0;

cleanup:
    virCgroupFree(&cgroup);
    return ret;
}


struct qemudHookData {
    virConnectPtr conn;
    virDomainObjPtr vm;
    struct qemud_driver *driver;
3258 3259 3260
};

static int qemudSecurityHook(void *data) {
3261 3262
    struct qemudHookData *h = data;

3263 3264 3265
    /* This must take place before exec(), so that all QEMU
     * memory allocation is on the correct NUMA node
     */
3266
    if (qemuAddToCgroup(h->driver, h->vm->def) < 0)
3267 3268
        return -1;

3269 3270 3271
    /* This must be done after cgroup placement to avoid resetting CPU
     * affinity */
    if (qemudInitCpuAffinity(h->vm) < 0)
3272
        return -1;
3273

3274 3275
    if (h->driver->securityDriver &&
        h->driver->securityDriver->domainSetSecurityProcessLabel &&
3276
        h->driver->securityDriver->domainSetSecurityProcessLabel(h->driver->securityDriver, h->vm) < 0)
3277 3278 3279
        return -1;

    return 0;
3280 3281
}

3282
static int
3283
qemuPrepareMonitorChr(struct qemud_driver *driver,
3284
                      virDomainChrDefPtr monConfig,
3285 3286
                      const char *vm)
{
3287
    monConfig->targetType = VIR_DOMAIN_CHR_TARGET_TYPE_MONITOR;
3288

3289 3290
    monConfig->type = VIR_DOMAIN_CHR_TYPE_UNIX;
    monConfig->data.nix.listen = 1;
3291

D
Daniel P. Berrange 已提交
3292
    if (!(monConfig->info.alias = strdup("monitor"))) {
3293
        virReportOOMError();
D
Daniel P. Berrange 已提交
3294 3295 3296
        return -1;
    }

3297
    if (virAsprintf(&monConfig->data.nix.path, "%s/%s.monitor",
3298
                    driver->libDir, vm) < 0) {
3299
        virReportOOMError();
3300 3301 3302 3303 3304 3305
        return -1;
    }

    return 0;
}

C
Chris Lalancette 已提交
3306 3307 3308 3309 3310
static int qemuDomainSnapshotSetActive(virDomainObjPtr vm,
                                       char *snapshotDir);
static int qemuDomainSnapshotSetInactive(virDomainObjPtr vm,
                                         char *snapshotDir);

3311 3312
static int qemudStartVMDaemon(virConnectPtr conn,
                              struct qemud_driver *driver,
3313
                              virDomainObjPtr vm,
3314
                              const char *migrateFrom,
3315
                              bool start_paused,
3316 3317
                              int stdin_fd,
                              const char *stdin_path) {
3318
    const char **argv = NULL, **tmp;
3319
    const char **progenv = NULL;
3320
    int i, ret, runflags;
3321
    struct stat sb;
3322 3323
    int *vmfds = NULL;
    int nvmfds = 0;
3324
    unsigned long long qemuCmdFlags;
3325
    fd_set keepfd;
3326
    const char *emulator;
G
Guido Günther 已提交
3327
    pid_t child;
3328
    int pos = -1;
3329
    char ebuf[1024];
3330
    char *pidfile = NULL;
3331
    int logfile = -1;
3332
    qemuDomainObjPrivatePtr priv = vm->privateData;
3333

3334
    struct qemudHookData hookData;
3335 3336 3337 3338
    hookData.conn = conn;
    hookData.vm = vm;
    hookData.driver = driver;

3339
    FD_ZERO(&keepfd);
3340

3341 3342
    DEBUG0("Beginning VM startup process");

D
Daniel P. Berrange 已提交
3343
    if (virDomainObjIsActive(vm)) {
3344 3345
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("VM is already active"));
3346 3347 3348
        return -1;
    }

3349 3350 3351 3352 3353
    /* Must be run before security labelling */
    DEBUG0("Preparing host devices");
    if (qemuPrepareHostDevices(driver, vm->def) < 0)
        goto cleanup;

3354 3355
    /* If you are using a SecurityDriver with dynamic labelling,
       then generate a security label for isolation */
3356
    DEBUG0("Generating domain security label (if required)");
3357
    if (driver->securityDriver &&
3358
        driver->securityDriver->domainGenSecurityLabel &&
3359
        driver->securityDriver->domainGenSecurityLabel(vm) < 0)
3360
        goto cleanup;
3361

3362
    DEBUG0("Generating setting domain security labels (if required)");
3363 3364
    if (driver->securityDriver &&
        driver->securityDriver->domainSetSecurityAllLabel &&
3365
        driver->securityDriver->domainSetSecurityAllLabel(vm, stdin_path) < 0)
3366 3367
        goto cleanup;

3368 3369 3370
    /* Ensure no historical cgroup for this VM is lying around bogus
     * settings */
    DEBUG0("Ensuring no historical cgroup is lying around");
3371
    qemuRemoveCgroup(driver, vm, 1);
3372

3373 3374 3375
    if ((vm->def->ngraphics == 1) &&
        vm->def->graphics[0]->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
        vm->def->graphics[0]->data.vnc.autoport) {
3376
        DEBUG0("Determining VNC port");
3377
        int port = qemudNextFreeVNCPort(driver);
3378
        if (port < 0) {
3379 3380
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            "%s", _("Unable to find an unused VNC port"));
3381
            goto cleanup;
3382
        }
3383
        vm->def->graphics[0]->data.vnc.port = port;
3384
    }
3385

L
Laine Stump 已提交
3386
    if (virFileMakePath(driver->logDir) != 0) {
3387
        virReportSystemError(errno,
3388 3389
                             _("cannot create log directory %s"),
                             driver->logDir);
3390
        goto cleanup;
3391 3392
    }

3393
    DEBUG0("Creating domain log file");
3394
    if ((logfile = qemudLogFD(driver, vm->def->name)) < 0)
3395
        goto cleanup;
3396

3397 3398
    emulator = vm->def->emulator;

3399 3400 3401 3402
    /* 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
     */
3403
    if (stat(emulator, &sb) < 0) {
3404
        virReportSystemError(errno,
3405 3406
                             _("Cannot find QEMU binary %s"),
                             emulator);
3407
        goto cleanup;
3408 3409
    }

3410
    DEBUG0("Determing emulator version");
3411
    if (qemudExtractVersionInfo(emulator,
3412
                                NULL,
3413
                                &qemuCmdFlags) < 0)
3414
        goto cleanup;
3415

3416
    DEBUG0("Setting up domain cgroup (if required)");
3417
    if (qemuSetupCgroup(driver, vm) < 0)
3418 3419
        goto cleanup;

3420
    if (VIR_ALLOC(priv->monConfig) < 0) {
3421
        virReportOOMError();
3422 3423 3424
        goto cleanup;
    }

3425
    DEBUG0("Preparing monitor state");
3426
    if (qemuPrepareMonitorChr(driver, priv->monConfig, vm->def->name) < 0)
3427
        goto cleanup;
3428

D
Daniel P. Berrange 已提交
3429 3430 3431
#if HAVE_YAJL
    if (qemuCmdFlags & QEMUD_CMD_FLAG_MONITOR_JSON)
        priv->monJSON = 1;
3432
    else
D
Daniel P. Berrange 已提交
3433
#endif
3434
        priv->monJSON = 0;
D
Daniel P. Berrange 已提交
3435

D
Daniel P. Berrange 已提交
3436
    if ((ret = virFileDeletePid(driver->stateDir, vm->def->name)) != 0) {
3437
        virReportSystemError(ret,
D
Daniel P. Berrange 已提交
3438 3439 3440 3441 3442
                             _("Cannot remove stale PID file for %s"),
                             vm->def->name);
        goto cleanup;
    }

3443
    if (!(pidfile = virFilePid(driver->stateDir, vm->def->name))) {
3444
        virReportSystemError(errno,
3445 3446 3447
                             "%s", _("Failed to build pidfile path."));
        goto cleanup;
    }
D
Daniel P. Berrange 已提交
3448

3449
    /*
M
Matthew Booth 已提交
3450
     * Normally PCI addresses are assigned in the virDomainCreate
3451 3452 3453 3454 3455
     * or virDomainDefine methods. We might still need to assign
     * some here to cope with the question of upgrades. Regardless
     * we also need to populate the PCi address set cache for later
     * use in hotplug
     */
3456
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
3457
        DEBUG0("Assigning domain PCI addresses");
3458
        /* Populate cache with current addresses */
3459 3460 3461 3462 3463 3464 3465
        if (priv->pciaddrs) {
            qemuDomainPCIAddressSetFree(priv->pciaddrs);
            priv->pciaddrs = NULL;
        }
        if (!(priv->pciaddrs = qemuDomainPCIAddressSetCreate(vm->def)))
            goto cleanup;

3466 3467

        /* Assign any remaining addresses */
3468 3469
        if (qemuAssignDevicePCISlots(vm->def, priv->pciaddrs) < 0)
            goto cleanup;
3470 3471 3472 3473

        priv->persistentAddrs = 1;
    } else {
        priv->persistentAddrs = 0;
3474 3475
    }

3476
    DEBUG0("Building emulator command line");
3477
    vm->def->id = driver->nextvmid++;
3478
    if (qemudBuildCommandLine(conn, driver, vm->def, priv->monConfig,
D
Daniel P. Berrange 已提交
3479
                              priv->monJSON, qemuCmdFlags, &argv, &progenv,
3480
                              &vmfds, &nvmfds, migrateFrom,
C
Chris Lalancette 已提交
3481 3482 3483 3484
                              vm->current_snapshot) < 0)
        goto cleanup;

    if (qemuDomainSnapshotSetInactive(vm, driver->snapshotDir) < 0)
3485
        goto cleanup;
3486

3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502
    /* now that we know it is about to start call the hook if present */
    if (virHookPresent(VIR_HOOK_DRIVER_QEMU)) {
        char *xml = virDomainDefFormat(vm->def, 0);
        int hookret;

        hookret = virHookCall(VIR_HOOK_DRIVER_QEMU, vm->def->name,
                    VIR_HOOK_QEMU_OP_START, VIR_HOOK_SUBOP_BEGIN, NULL, xml);
        VIR_FREE(xml);

        /*
         * If the script raised an error abort the launch
         */
        if (hookret < 0)
            goto cleanup;
    }

3503 3504
    tmp = progenv;
    while (*tmp) {
3505
        if (safewrite(logfile, *tmp, strlen(*tmp)) < 0)
3506
            VIR_WARN("Unable to write envv to logfile: %s",
3507
                     virStrerror(errno, ebuf, sizeof ebuf));
3508
        if (safewrite(logfile, " ", 1) < 0)
3509
            VIR_WARN("Unable to write envv to logfile: %s",
3510
                     virStrerror(errno, ebuf, sizeof ebuf));
3511 3512
        tmp++;
    }
3513 3514
    tmp = argv;
    while (*tmp) {
3515
        if (safewrite(logfile, *tmp, strlen(*tmp)) < 0)
3516
            VIR_WARN("Unable to write argv to logfile: %s",
3517
                     virStrerror(errno, ebuf, sizeof ebuf));
3518
        if (safewrite(logfile, " ", 1) < 0)
3519
            VIR_WARN("Unable to write argv to logfile: %s",
3520
                     virStrerror(errno, ebuf, sizeof ebuf));
3521 3522
        tmp++;
    }
3523
    if (safewrite(logfile, "\n", 1) < 0)
3524
        VIR_WARN("Unable to write argv to logfile: %s",
3525
                 virStrerror(errno, ebuf, sizeof ebuf));
3526

3527
    if ((pos = lseek(logfile, 0, SEEK_END)) < 0)
3528
        VIR_WARN("Unable to seek to end of logfile: %s",
3529
                 virStrerror(errno, ebuf, sizeof ebuf));
3530

3531 3532
    for (i = 0 ; i < nvmfds ; i++)
        FD_SET(vmfds[i], &keepfd);
3533

3534 3535 3536 3537 3538 3539 3540
    VIR_DEBUG("Clear emulator capabilities: %d",
              driver->clearEmulatorCapabilities);
    runflags = VIR_EXEC_NONBLOCK;
    if (driver->clearEmulatorCapabilities) {
        runflags |= VIR_EXEC_CLEAR_CAPS;
    }

3541
    ret = virExecDaemonize(argv, progenv, &keepfd, &child,
3542
                           stdin_fd, &logfile, &logfile,
3543
                           runflags,
3544 3545 3546
                           qemudSecurityHook, &hookData,
                           pidfile);
    VIR_FREE(pidfile);
G
Guido Günther 已提交
3547 3548 3549

    /* wait for qemu process to to show up */
    if (ret == 0) {
3550
        if (virFileReadPid(driver->stateDir, vm->def->name, &vm->pid)) {
3551 3552
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("Domain %s didn't show up\n"), vm->def->name);
3553
            ret = -1;
G
Guido Günther 已提交
3554
        }
3555 3556 3557 3558 3559 3560 3561 3562 3563 3564 3565
    } else if (ret == -2) {
        /* The virExec process that launches the daemon failed. Pending on
         * when it failed (we can't determine for sure), there may be
         * extra info in the domain log (if the hook failed for example).
         *
         * Pretend like things succeeded, and let 'WaitForMonitor' report
         * the log contents for us.
         */
        vm->pid = child;
        ret = 0;
    }
3566

3567 3568 3569
    if (migrateFrom)
        start_paused = true;
    vm->state = start_paused ? VIR_DOMAIN_PAUSED : VIR_DOMAIN_RUNNING;
3570

3571
    for (i = 0 ; argv[i] ; i++)
3572 3573
        VIR_FREE(argv[i]);
    VIR_FREE(argv);
3574

3575 3576 3577 3578
    for (i = 0 ; progenv[i] ; i++)
        VIR_FREE(progenv[i]);
    VIR_FREE(progenv);

3579 3580 3581
    if (ret == -1) /* The VM failed to start; tear filters before taps */
        virNWFilterTearVMNWFilters(vm);

3582 3583 3584
    if (vmfds) {
        for (i = 0 ; i < nvmfds ; i++) {
            close(vmfds[i]);
3585
        }
3586
        VIR_FREE(vmfds);
3587 3588
    }

3589
    if (ret == -1) /* The VM failed to start */
3590 3591
        goto cleanup;

3592
    DEBUG0("Waiting for monitor to show up");
3593
    if (qemudWaitForMonitor(driver, vm, pos) < 0)
3594
        goto cleanup;
3595

3596
    DEBUG0("Detecting VCPU PIDs");
3597
    if (qemuDetectVcpuPIDs(driver, vm) < 0)
3598
        goto cleanup;
3599

3600
    DEBUG0("Setting any required VM passwords");
3601
    if (qemuInitPasswords(conn, driver, vm, qemuCmdFlags) < 0)
3602
        goto cleanup;
3603

D
Daniel P. Berrange 已提交
3604 3605 3606
    /* If we have -device, then addresses are assigned explicitly.
     * If not, then we have to detect dynamic ones here */
    if (!(qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE)) {
3607
        DEBUG0("Determining domain device PCI addresses");
D
Daniel P. Berrange 已提交
3608
        if (qemuInitPCIAddresses(driver, vm) < 0)
3609
            goto cleanup;
D
Daniel P. Berrange 已提交
3610
    }
3611

3612
    DEBUG0("Setting initial memory amount");
3613
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
3614
    if (qemuMonitorSetBalloon(priv->mon, vm->def->memory) < 0) {
3615
        qemuDomainObjExitMonitorWithDriver(driver, vm);
3616
        goto cleanup;
3617
    }
3618

3619
    if (!start_paused) {
3620
        DEBUG0("Starting domain CPUs");
3621 3622 3623
        /* Allow the CPUS to start executing */
        if (qemuMonitorStartCPUs(priv->mon, conn) < 0) {
            if (virGetLastError() == NULL)
3624 3625
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                "%s", _("resume operation failed"));
3626
            qemuDomainObjExitMonitorWithDriver(driver, vm);
3627
            goto cleanup;
3628 3629
        }
    }
3630
    qemuDomainObjExitMonitorWithDriver(driver, vm);
3631

3632

3633
    DEBUG0("Writing domain status to disk");
3634
    if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
3635
        goto cleanup;
3636

3637 3638 3639
    if (logfile != -1)
        close(logfile);

3640
    return 0;
3641 3642

cleanup:
3643 3644 3645
    /* We jump here if we failed to start the VM for any reason, or
     * if we failed to initialize the now running VM. kill it off and
     * pretend we never started it */
3646
    qemudShutdownVMDaemon(driver, vm, 0);
3647 3648 3649 3650 3651

    if (logfile != -1)
        close(logfile);

    return -1;
3652 3653 3654
}


3655
static void qemudShutdownVMDaemon(struct qemud_driver *driver,
3656 3657
                                  virDomainObjPtr vm,
                                  int migrated) {
D
Daniel P. Berrange 已提交
3658
    int ret;
3659
    int retries = 0;
3660
    qemuDomainObjPrivatePtr priv = vm->privateData;
3661
    virErrorPtr orig_err;
3662 3663
    virDomainDefPtr def;
    int i;
D
Daniel P. Berrange 已提交
3664

3665 3666
    VIR_DEBUG("Shutting down VM '%s' pid=%d migrated=%d",
              vm->def->name, vm->pid, migrated);
3667

3668 3669 3670 3671
    /* This method is routinely used in clean up paths. Disable error
     * reporting so we don't squash a legit error. */
    orig_err = virSaveLastError();

3672
    virNWFilterTearVMNWFilters(vm);
S
Stefan Berger 已提交
3673

3674
    if (driver->macFilter) {
3675
        def = vm->def;
3676 3677 3678 3679
        for (i = 0 ; i < def->nnets ; i++) {
            virDomainNetDefPtr net = def->nets[i];
            if (net->ifname == NULL)
                continue;
3680
            if ((errno = networkDisallowMacOnPort(driver, net->ifname,
3681
                                                  net->mac))) {
3682
                virReportSystemError(errno,
3683 3684 3685 3686 3687 3688
             _("failed to remove ebtables rule to allow MAC address on  '%s'"),
                                     net->ifname);
            }
        }
    }

3689
    /* This will safely handle a non-running guest with pid=0 or pid=-1*/
G
Guido Günther 已提交
3690 3691
    if (virKillProcess(vm->pid, 0) == 0 &&
        virKillProcess(vm->pid, SIGTERM) < 0)
3692
        virReportSystemError(errno,
3693 3694
                             _("Failed to send SIGTERM to %s (%d)"),
                             vm->def->name, vm->pid);
3695

3696 3697 3698
    if (priv->mon &&
        qemuMonitorClose(priv->mon) == 0) {
        virDomainObjUnref(vm);
3699
        priv->mon = NULL;
3700
    }
3701

3702 3703 3704 3705 3706
    if (priv->monConfig) {
        if (priv->monConfig->type == VIR_DOMAIN_CHR_TYPE_UNIX)
            unlink(priv->monConfig->data.nix.path);
        virDomainChrDefFree(priv->monConfig);
        priv->monConfig = NULL;
3707 3708
    }

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

3712 3713 3714 3715 3716 3717 3718 3719 3720 3721
    /* now that we know it's stopped call the hook if present */
    if (virHookPresent(VIR_HOOK_DRIVER_QEMU)) {
        char *xml = virDomainDefFormat(vm->def, 0);

        /* we can't stop the operation even if the script raised an error */
        virHookCall(VIR_HOOK_DRIVER_QEMU, vm->def->name,
                    VIR_HOOK_QEMU_OP_STOPPED, VIR_HOOK_SUBOP_END, NULL, xml);
        VIR_FREE(xml);
    }

3722
    /* Reset Security Labels */
3723
    if (driver->securityDriver &&
3724
        driver->securityDriver->domainRestoreSecurityAllLabel)
3725
        driver->securityDriver->domainRestoreSecurityAllLabel(vm, migrated);
3726 3727
    if (driver->securityDriver &&
        driver->securityDriver->domainReleaseSecurityLabel)
3728
        driver->securityDriver->domainReleaseSecurityLabel(vm);
3729

3730 3731 3732 3733 3734 3735 3736
    /* Clear out dynamically assigned labels */
    if (vm->def->seclabel.type == VIR_DOMAIN_SECLABEL_DYNAMIC) {
        VIR_FREE(vm->def->seclabel.model);
        VIR_FREE(vm->def->seclabel.label);
        VIR_FREE(vm->def->seclabel.imagelabel);
    }

D
Daniel P. Berrange 已提交
3737
    virDomainDefClearDeviceAliases(vm->def);
3738 3739 3740 3741 3742
    if (!priv->persistentAddrs) {
        virDomainDefClearPCIAddresses(vm->def);
        qemuDomainPCIAddressSetFree(priv->pciaddrs);
        priv->pciaddrs = NULL;
    }
3743

3744
    qemuDomainReAttachHostDevices(driver, vm->def);
3745

S
Stefan Berger 已提交
3746 3747 3748 3749
#if WITH_MACVTAP
    def = vm->def;
    for (i = 0; i < def->nnets; i++) {
        virDomainNetDefPtr net = def->nets[i];
3750
        if (net->type == VIR_DOMAIN_NET_TYPE_DIRECT) {
3751
            delMacvtap(net->ifname, net->mac, net->data.direct.linkdev,
3752
                       &net->data.direct.virtPortProfile);
3753 3754
            VIR_FREE(net->ifname);
        }
S
Stefan Berger 已提交
3755 3756 3757
    }
#endif

3758
retry:
3759
    if ((ret = qemuRemoveCgroup(driver, vm, 0)) < 0) {
3760 3761 3762 3763 3764 3765 3766 3767
        if (ret == -EBUSY && (retries++ < 5)) {
            usleep(200*1000);
            goto retry;
        }
        VIR_WARN("Failed to remove cgroup for %s",
                 vm->def->name);
    }

3768
    qemudRemoveDomainStatus(driver, vm);
D
Daniel P. Berrange 已提交
3769

3770 3771 3772 3773 3774 3775 3776 3777 3778 3779 3780 3781 3782 3783 3784
    /* Remove VNC port from port reservation bitmap, but only if it was
       reserved by the driver (autoport=yes)
    */
    if ((vm->def->ngraphics == 1) &&
        vm->def->graphics[0]->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
        vm->def->graphics[0]->data.vnc.autoport &&
        vm->def->graphics[0]->data.vnc.port != -1) {
        if (virBitmapClearBit(driver->reservedVNCPorts,
                              vm->def->graphics[0]->data.vnc.port - \
                              QEMU_VNC_PORT_MIN) < 0) {
            VIR_DEBUG("virBitmapClearBit failed on bit %d",
                      vm->def->graphics[0]->data.vnc.port - QEMU_VNC_PORT_MIN);
       }
    }

3785
    vm->pid = -1;
3786
    vm->def->id = -1;
3787
    vm->state = VIR_DOMAIN_SHUTOFF;
3788 3789
    VIR_FREE(priv->vcpupids);
    priv->nvcpupids = 0;
3790 3791

    if (vm->newDef) {
3792
        virDomainDefFree(vm->def);
3793
        vm->def = vm->newDef;
3794
        vm->def->id = -1;
3795 3796
        vm->newDef = NULL;
    }
3797 3798 3799 3800 3801

    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }
3802 3803 3804
}


3805
static virDrvOpenStatus qemudOpen(virConnectPtr conn,
3806
                                  virConnectAuthPtr auth ATTRIBUTE_UNUSED,
3807
                                  int flags ATTRIBUTE_UNUSED) {
3808
    if (conn->uri == NULL) {
3809 3810 3811
        if (qemu_driver == NULL)
            return VIR_DRV_OPEN_DECLINED;

3812
        conn->uri = xmlParseURI(qemu_driver->privileged ?
3813 3814
                                "qemu:///system" :
                                "qemu:///session");
3815
        if (!conn->uri) {
3816
            virReportOOMError();
3817 3818
            return VIR_DRV_OPEN_ERROR;
        }
3819 3820 3821 3822 3823 3824 3825 3826 3827 3828
    } else {
        /* If URI isn't 'qemu' its definitely not for us */
        if (conn->uri->scheme == NULL ||
            STRNEQ(conn->uri->scheme, "qemu"))
            return VIR_DRV_OPEN_DECLINED;

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

3829
        if (qemu_driver == NULL) {
3830 3831
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("qemu state driver is not active"));
3832 3833 3834
            return VIR_DRV_OPEN_ERROR;
        }

3835
        if (conn->uri->path == NULL) {
3836 3837 3838 3839 3840
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("no QEMU URI path given, try %s"),
                            qemu_driver->privileged
                            ? "qemu:///system"
                            : "qemu:///session");
3841 3842 3843
                return VIR_DRV_OPEN_ERROR;
        }

3844
        if (qemu_driver->privileged) {
3845 3846
            if (STRNEQ (conn->uri->path, "/system") &&
                STRNEQ (conn->uri->path, "/session")) {
3847 3848 3849
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                _("unexpected QEMU URI path '%s', try qemu:///system"),
                                conn->uri->path);
3850 3851 3852 3853
                return VIR_DRV_OPEN_ERROR;
            }
        } else {
            if (STRNEQ (conn->uri->path, "/session")) {
3854 3855 3856
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                _("unexpected QEMU URI path '%s', try qemu:///session"),
                                conn->uri->path);
3857 3858 3859
                return VIR_DRV_OPEN_ERROR;
            }
        }
3860 3861 3862 3863 3864 3865 3866
    }
    conn->privateData = qemu_driver;

    return VIR_DRV_OPEN_SUCCESS;
}

static int qemudClose(virConnectPtr conn) {
3867
    struct qemud_driver *driver = conn->privateData;
3868 3869

    /* Get rid of callbacks registered for this conn */
3870
    qemuDriverLock(driver);
3871
    virDomainEventCallbackListRemoveConn(conn, driver->domainEventCallbacks);
3872
    qemuDriverUnlock(driver);
3873 3874 3875 3876 3877 3878

    conn->privateData = NULL;

    return 0;
}

D
Daniel Veillard 已提交
3879 3880 3881 3882 3883
/* Which features are supported by this driver? */
static int
qemudSupportsFeature (virConnectPtr conn ATTRIBUTE_UNUSED, int feature)
{
    switch (feature) {
3884 3885 3886 3887 3888
    case VIR_DRV_FEATURE_MIGRATION_V2:
    case VIR_DRV_FEATURE_MIGRATION_P2P:
        return 1;
    default:
        return 0;
D
Daniel Veillard 已提交
3889 3890 3891
    }
}

3892
static const char *qemudGetType(virConnectPtr conn ATTRIBUTE_UNUSED) {
3893
    return "QEMU";
3894 3895
}

3896

3897 3898 3899 3900 3901 3902 3903 3904 3905 3906 3907 3908 3909
static int qemuIsSecure(virConnectPtr conn ATTRIBUTE_UNUSED)
{
    /* Trivially secure, since always inside the daemon */
    return 1;
}

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


3910 3911 3912 3913
static int kvmGetMaxVCPUs(void) {
    int maxvcpus = 1;

    int r, fd;
3914

3915 3916
    fd = open(KVM_DEVICE, O_RDONLY);
    if (fd < 0) {
3917
        virReportSystemError(errno, _("Unable to open %s"), KVM_DEVICE);
3918
        return -1;
3919 3920 3921 3922 3923 3924 3925 3926 3927 3928 3929
    }

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

    close(fd);
    return maxvcpus;
}


3930
static int qemudGetMaxVCPUs(virConnectPtr conn ATTRIBUTE_UNUSED, const char *type) {
3931 3932 3933
    if (!type)
        return 16;

3934
    if (STRCASEEQ(type, "qemu"))
3935 3936
        return 16;

3937
    if (STRCASEEQ(type, "kvm"))
3938
        return kvmGetMaxVCPUs();
3939

3940
    if (STRCASEEQ(type, "kqemu"))
3941
        return 1;
3942

3943 3944
    qemuReportError(VIR_ERR_INVALID_ARG,
                    _("unknown type '%s'"), type);
3945 3946 3947
    return -1;
}

3948

3949
static char *qemudGetCapabilities(virConnectPtr conn) {
3950
    struct qemud_driver *driver = conn->privateData;
3951
    virCapsPtr caps = NULL;
3952
    char *xml = NULL;
3953

3954
    qemuDriverLock(driver);
3955

3956
    if ((caps = qemuCreateCapabilities(qemu_driver->caps,
3957
                                       qemu_driver)) == NULL) {
3958 3959 3960
        virCapabilitiesFree(caps);
        goto cleanup;
    }
3961

3962
    virCapabilitiesFree(qemu_driver->caps);
3963 3964 3965
    qemu_driver->caps = caps;

    if ((xml = virCapabilitiesFormatXML(driver->caps)) == NULL)
3966
        virReportOOMError();
3967 3968

cleanup:
3969
    qemuDriverUnlock(driver);
3970

3971
    return xml;
3972 3973 3974
}


3975
static int qemudGetProcessInfo(unsigned long long *cpuTime, int *lastCpu, int pid, int tid) {
D
Daniel P. Berrange 已提交
3976 3977
    char proc[PATH_MAX];
    FILE *pidinfo;
3978
    unsigned long long usertime, systime;
3979 3980
    int cpu;
    int ret;
D
Daniel P. Berrange 已提交
3981

3982 3983 3984 3985 3986 3987
    if (tid)
        ret = snprintf(proc, sizeof(proc), "/proc/%d/task/%d/stat", pid, tid);
    else
        ret = snprintf(proc, sizeof(proc), "/proc/%d/stat", pid);
    if (ret >= (int)sizeof(proc)) {
        errno = E2BIG;
D
Daniel P. Berrange 已提交
3988 3989 3990 3991
        return -1;
    }

    if (!(pidinfo = fopen(proc, "r"))) {
3992
        /*printf("cannot read pid info");*/
D
Daniel P. Berrange 已提交
3993
        /* VM probably shut down, so fake 0 */
3994 3995 3996 3997
        if (cpuTime)
            *cpuTime = 0;
        if (lastCpu)
            *lastCpu = 0;
D
Daniel P. Berrange 已提交
3998 3999 4000
        return 0;
    }

4001 4002 4003 4004 4005 4006 4007 4008 4009 4010
    /* See 'man proc' for information about what all these fields are. We're
     * only interested in a very few of them */
    if (fscanf(pidinfo,
               /* pid -> stime */
               "%*d %*s %*c %*d %*d %*d %*d %*d %*u %*u %*u %*u %*u %llu %llu"
               /* cutime -> endcode */
               "%*d %*d %*d %*d %*d %*u %*u %*d %*u %*u %*u %*u"
               /* startstack -> processor */
               "%*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*d %d",
               &usertime, &systime, &cpu) != 3) {
4011
        fclose(pidinfo);
4012 4013
        VIR_WARN0("cannot parse process status data");
        errno = -EINVAL;
D
Daniel P. Berrange 已提交
4014 4015 4016 4017 4018 4019 4020 4021
        return -1;
    }

    /* We got jiffies
     * We want nanoseconds
     * _SC_CLK_TCK is jiffies per second
     * So calulate thus....
     */
4022 4023 4024 4025 4026
    if (cpuTime)
        *cpuTime = 1000ull * 1000ull * 1000ull * (usertime + systime) / (unsigned long long)sysconf(_SC_CLK_TCK);
    if (lastCpu)
        *lastCpu = cpu;

D
Daniel P. Berrange 已提交
4027

4028 4029
    VIR_DEBUG("Got status for %d/%d user=%llu sys=%llu cpu=%d",
              pid, tid, usertime, systime, cpu);
D
Daniel P. Berrange 已提交
4030 4031 4032 4033 4034 4035 4036

    fclose(pidinfo);

    return 0;
}


4037
static virDomainPtr qemudDomainLookupByID(virConnectPtr conn,
4038
                                          int id) {
4039 4040 4041 4042
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;

4043
    qemuDriverLock(driver);
4044
    vm  = virDomainFindByID(&driver->domains, id);
4045
    qemuDriverUnlock(driver);
4046 4047

    if (!vm) {
4048 4049
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching id %d"), id);
4050
        goto cleanup;
4051 4052
    }

4053
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
4054
    if (dom) dom->id = vm->def->id;
4055 4056

cleanup:
4057 4058
    if (vm)
        virDomainObjUnlock(vm);
4059 4060
    return dom;
}
4061

4062
static virDomainPtr qemudDomainLookupByUUID(virConnectPtr conn,
4063
                                            const unsigned char *uuid) {
4064 4065 4066
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
4067

4068
    qemuDriverLock(driver);
4069
    vm = virDomainFindByUUID(&driver->domains, uuid);
4070 4071
    qemuDriverUnlock(driver);

4072
    if (!vm) {
4073 4074
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(uuid, uuidstr);
4075 4076
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4077
        goto cleanup;
4078 4079
    }

4080
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
4081
    if (dom) dom->id = vm->def->id;
4082 4083

cleanup:
4084 4085
    if (vm)
        virDomainObjUnlock(vm);
4086 4087
    return dom;
}
4088

4089
static virDomainPtr qemudDomainLookupByName(virConnectPtr conn,
4090
                                            const char *name) {
4091 4092 4093
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
4094

4095
    qemuDriverLock(driver);
4096
    vm = virDomainFindByName(&driver->domains, name);
4097 4098
    qemuDriverUnlock(driver);

4099
    if (!vm) {
4100 4101
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching name '%s'"), name);
4102
        goto cleanup;
4103 4104
    }

4105
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
4106
    if (dom) dom->id = vm->def->id;
4107 4108

cleanup:
4109 4110
    if (vm)
        virDomainObjUnlock(vm);
4111 4112 4113
    return dom;
}

4114 4115 4116 4117 4118 4119 4120 4121 4122 4123 4124

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

    qemuDriverLock(driver);
    obj = virDomainFindByUUID(&driver->domains, dom->uuid);
    qemuDriverUnlock(driver);
    if (!obj) {
4125
        qemuReportError(VIR_ERR_NO_DOMAIN, NULL);
4126 4127 4128 4129 4130 4131 4132 4133 4134 4135 4136 4137 4138 4139 4140 4141 4142 4143 4144 4145
        goto cleanup;
    }
    ret = virDomainObjIsActive(obj);

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

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

    qemuDriverLock(driver);
    obj = virDomainFindByUUID(&driver->domains, dom->uuid);
    qemuDriverUnlock(driver);
    if (!obj) {
4146
        qemuReportError(VIR_ERR_NO_DOMAIN, NULL);
4147 4148 4149 4150 4151 4152 4153 4154 4155 4156 4157
        goto cleanup;
    }
    ret = obj->persistent;

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


4158
static int qemudGetVersion(virConnectPtr conn, unsigned long *version) {
4159 4160 4161
    struct qemud_driver *driver = conn->privateData;
    int ret = -1;

4162
    qemuDriverLock(driver);
4163
    if (qemudExtractVersion(driver) < 0)
4164
        goto cleanup;
4165

4166
    *version = qemu_driver->qemuVersion;
4167 4168 4169
    ret = 0;

cleanup:
4170
    qemuDriverUnlock(driver);
4171
    return ret;
D
Daniel P. Berrange 已提交
4172 4173
}

4174
static int qemudListDomains(virConnectPtr conn, int *ids, int nids) {
4175
    struct qemud_driver *driver = conn->privateData;
4176
    int n;
4177

4178
    qemuDriverLock(driver);
4179
    n = virDomainObjListGetActiveIDs(&driver->domains, ids, nids);
4180
    qemuDriverUnlock(driver);
4181

4182
    return n;
D
Daniel P. Berrange 已提交
4183
}
4184

4185
static int qemudNumDomains(virConnectPtr conn) {
4186
    struct qemud_driver *driver = conn->privateData;
4187
    int n;
4188

4189
    qemuDriverLock(driver);
4190
    n = virDomainObjListNumOfDomains(&driver->domains, 1);
4191
    qemuDriverUnlock(driver);
4192

4193
    return n;
D
Daniel P. Berrange 已提交
4194
}
4195

4196
static virDomainPtr qemudDomainCreate(virConnectPtr conn, const char *xml,
4197
                                      unsigned int flags) {
4198
    struct qemud_driver *driver = conn->privateData;
4199
    virDomainDefPtr def;
4200
    virDomainObjPtr vm = NULL;
4201
    virDomainPtr dom = NULL;
4202
    virDomainEventPtr event = NULL;
D
Daniel P. Berrange 已提交
4203

4204
    virCheckFlags(VIR_DOMAIN_START_PAUSED, NULL);
4205

4206
    qemuDriverLock(driver);
4207
    if (!(def = virDomainDefParseString(driver->caps, xml,
4208
                                        VIR_DOMAIN_XML_INACTIVE)))
4209
        goto cleanup;
4210

4211
    if (virSecurityDriverVerify(def) < 0)
4212 4213
        goto cleanup;

4214 4215
    if (virDomainObjIsDuplicate(&driver->domains, def, 1) < 0)
        goto cleanup;
4216

4217 4218 4219 4220 4221 4222
    if (qemudCanonicalizeMachine(driver, def) < 0)
        goto cleanup;

    if (qemuAssignPCIAddresses(def) < 0)
        goto cleanup;

4223
    if (!(vm = virDomainAssignDef(driver->caps,
4224
                                  &driver->domains,
4225
                                  def, false)))
4226 4227 4228
        goto cleanup;

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

4230 4231 4232
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup; /* XXXX free the 'vm' we created ? */

4233 4234 4235
    if (qemudStartVMDaemon(conn, driver, vm, NULL,
                           (flags & VIR_DOMAIN_START_PAUSED) != 0,
                           -1, NULL) < 0) {
4236 4237 4238
        if (qemuDomainObjEndJob(vm) > 0)
            virDomainRemoveInactive(&driver->domains,
                                    vm);
4239
        vm = NULL;
4240
        goto cleanup;
D
Daniel P. Berrange 已提交
4241
    }
4242 4243 4244 4245

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

4247
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
4248
    if (dom) dom->id = vm->def->id;
4249

4250 4251 4252
    if (vm &&
        qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
4253

4254 4255
cleanup:
    virDomainDefFree(def);
4256 4257
    if (vm)
        virDomainObjUnlock(vm);
4258 4259
    if (event)
        qemuDomainEventQueue(driver, event);
4260
    qemuDriverUnlock(driver);
4261
    return dom;
D
Daniel P. Berrange 已提交
4262 4263 4264
}


4265
static int qemudDomainSuspend(virDomainPtr dom) {
4266 4267 4268
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
4269
    virDomainEventPtr event = NULL;
4270
    qemuDomainObjPrivatePtr priv;
4271

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

D
Daniel P. Berrange 已提交
4275
    if (!vm) {
4276 4277
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4278 4279
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4280
        goto cleanup;
D
Daniel P. Berrange 已提交
4281
    }
D
Daniel P. Berrange 已提交
4282
    if (!virDomainObjIsActive(vm)) {
4283 4284
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
4285
        goto cleanup;
D
Daniel P. Berrange 已提交
4286
    }
4287 4288 4289 4290 4291 4292 4293 4294 4295 4296 4297 4298 4299 4300 4301 4302 4303 4304

    priv = vm->privateData;

    if (priv->jobActive == QEMU_JOB_MIGRATION) {
        if (vm->state != VIR_DOMAIN_PAUSED) {
            VIR_DEBUG("Requesting domain pause on %s",
                      vm->def->name);
            priv->jobSignals |= QEMU_JOB_SIGNAL_SUSPEND;
        }
        ret = 0;
        goto cleanup;
    } else {
        if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
            goto cleanup;

        if (!virDomainObjIsActive(vm)) {
            qemuReportError(VIR_ERR_OPERATION_INVALID,
                            "%s", _("domain is not running"));
4305
            goto endjob;
4306
        }
4307 4308
        if (vm->state != VIR_DOMAIN_PAUSED) {
            int rc;
4309
            int state = vm->state;
4310

4311
            vm->state = VIR_DOMAIN_PAUSED;
4312 4313 4314
            qemuDomainObjEnterMonitorWithDriver(driver, vm);
            rc = qemuMonitorStopCPUs(priv->mon);
            qemuDomainObjExitMonitorWithDriver(driver, vm);
4315 4316
            if (rc < 0) {
                vm->state = state;
4317
                goto endjob;
4318
            }
4319 4320 4321 4322 4323 4324 4325
            event = virDomainEventNewFromObj(vm,
                                             VIR_DOMAIN_EVENT_SUSPENDED,
                                             VIR_DOMAIN_EVENT_SUSPENDED_PAUSED);
        }
        if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
            goto endjob;
        ret = 0;
D
Daniel P. Berrange 已提交
4326
    }
4327

4328
endjob:
4329 4330
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
4331

4332
cleanup:
4333 4334
    if (vm)
        virDomainObjUnlock(vm);
4335

4336
    if (event)
4337
        qemuDomainEventQueue(driver, event);
4338
    qemuDriverUnlock(driver);
4339
    return ret;
D
Daniel P. Berrange 已提交
4340 4341 4342
}


4343
static int qemudDomainResume(virDomainPtr dom) {
4344 4345 4346
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
4347
    virDomainEventPtr event = NULL;
4348

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

D
Daniel P. Berrange 已提交
4352
    if (!vm) {
4353 4354
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4355 4356
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4357
        goto cleanup;
D
Daniel P. Berrange 已提交
4358
    }
4359 4360 4361 4362

    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
4363
    if (!virDomainObjIsActive(vm)) {
4364 4365
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
4366
        goto endjob;
D
Daniel P. Berrange 已提交
4367
    }
4368
    if (vm->state == VIR_DOMAIN_PAUSED) {
4369
        qemuDomainObjPrivatePtr priv = vm->privateData;
4370
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
4371
        if (qemuMonitorStartCPUs(priv->mon, dom->conn) < 0) {
4372
            qemuDomainObjExitMonitorWithDriver(driver, vm);
4373
            if (virGetLastError() == NULL)
4374 4375
                qemuReportError(VIR_ERR_OPERATION_FAILED,
                                "%s", _("resume operation failed"));
4376
            goto endjob;
4377
        }
4378
        qemuDomainObjExitMonitorWithDriver(driver, vm);
4379
        vm->state = VIR_DOMAIN_RUNNING;
4380 4381 4382
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_RESUMED,
                                         VIR_DOMAIN_EVENT_RESUMED_UNPAUSED);
D
Daniel P. Berrange 已提交
4383
    }
4384
    if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
4385
        goto endjob;
4386 4387
    ret = 0;

4388
endjob:
4389 4390
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
4391

4392
cleanup:
4393 4394
    if (vm)
        virDomainObjUnlock(vm);
4395
    if (event)
4396
        qemuDomainEventQueue(driver, event);
4397
    qemuDriverUnlock(driver);
4398
    return ret;
D
Daniel P. Berrange 已提交
4399 4400 4401
}


4402
static int qemudDomainShutdown(virDomainPtr dom) {
4403 4404 4405
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
4406

4407
    qemuDriverLock(driver);
4408
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4409 4410
    qemuDriverUnlock(driver);

4411
    if (!vm) {
4412 4413
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4414 4415
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4416
        goto cleanup;
4417 4418
    }

4419 4420 4421
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
4422
    if (!virDomainObjIsActive(vm)) {
4423 4424
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
4425
        goto endjob;
4426 4427
    }

4428
    qemuDomainObjPrivatePtr priv = vm->privateData;
4429 4430 4431
    qemuDomainObjEnterMonitor(vm);
    ret = qemuMonitorSystemPowerdown(priv->mon);
    qemuDomainObjExitMonitor(vm);
4432

4433
endjob:
4434 4435
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
4436

4437
cleanup:
4438 4439
    if (vm)
        virDomainObjUnlock(vm);
4440
    return ret;
4441 4442 4443
}


4444
static int qemudDomainDestroy(virDomainPtr dom) {
4445 4446 4447
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
4448
    virDomainEventPtr event = NULL;
4449

4450
    qemuDriverLock(driver);
4451
    vm  = virDomainFindByUUID(&driver->domains, dom->uuid);
D
Daniel P. Berrange 已提交
4452
    if (!vm) {
4453 4454
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4455 4456
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4457
        goto cleanup;
D
Daniel P. Berrange 已提交
4458
    }
4459 4460 4461 4462

    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
4463
    if (!virDomainObjIsActive(vm)) {
4464 4465
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
4466
        goto endjob;
4467
    }
4468

4469
    qemudShutdownVMDaemon(driver, vm, 0);
4470 4471 4472
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_DESTROYED);
4473
    if (!vm->persistent) {
4474 4475 4476
        if (qemuDomainObjEndJob(vm) > 0)
            virDomainRemoveInactive(&driver->domains,
                                    vm);
4477 4478
        vm = NULL;
    }
4479 4480
    ret = 0;

4481
endjob:
4482 4483 4484
    if (vm &&
        qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
4485

4486
cleanup:
4487 4488
    if (vm)
        virDomainObjUnlock(vm);
4489 4490
    if (event)
        qemuDomainEventQueue(driver, event);
4491
    qemuDriverUnlock(driver);
4492
    return ret;
D
Daniel P. Berrange 已提交
4493 4494 4495
}


4496
static char *qemudDomainGetOSType(virDomainPtr dom) {
4497 4498 4499
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *type = NULL;
4500

4501
    qemuDriverLock(driver);
4502
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4503
    qemuDriverUnlock(driver);
4504
    if (!vm) {
4505 4506
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4507 4508
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4509
        goto cleanup;
4510 4511
    }

4512
    if (!(type = strdup(vm->def->os.type)))
4513
        virReportOOMError();
4514 4515

cleanup:
4516 4517
    if (vm)
        virDomainObjUnlock(vm);
4518 4519 4520
    return type;
}

4521 4522
/* Returns max memory in kb, 0 if error */
static unsigned long qemudDomainGetMaxMemory(virDomainPtr dom) {
4523 4524 4525
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    unsigned long ret = 0;
4526

4527
    qemuDriverLock(driver);
4528
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4529 4530
    qemuDriverUnlock(driver);

4531
    if (!vm) {
4532 4533
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4534 4535
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4536
        goto cleanup;
4537 4538
    }

4539 4540 4541
    ret = vm->def->maxmem;

cleanup:
4542 4543
    if (vm)
        virDomainObjUnlock(vm);
4544
    return ret;
4545 4546 4547
}

static int qemudDomainSetMemory(virDomainPtr dom, unsigned long newmem) {
4548
    struct qemud_driver *driver = dom->conn->privateData;
4549
    qemuDomainObjPrivatePtr priv;
4550
    virDomainObjPtr vm;
4551
    int ret = -1, r;
4552

4553
    qemuDriverLock(driver);
4554
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4555
    qemuDriverUnlock(driver);
4556
    if (!vm) {
4557 4558
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4559 4560
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4561
        goto cleanup;
4562 4563
    }

4564 4565 4566 4567 4568 4569
    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
    }

4570
    if (newmem > vm->def->maxmem) {
4571 4572
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("cannot set memory higher than max memory"));
4573
        goto cleanup;
4574 4575
    }

4576 4577 4578
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

4579 4580 4581 4582 4583 4584
    priv = vm->privateData;
    qemuDomainObjEnterMonitor(vm);
    r = qemuMonitorSetBalloon(priv->mon, newmem);
    qemuDomainObjExitMonitor(vm);
    if (r < 0)
        goto endjob;
4585

4586 4587 4588 4589 4590
    /* Lack of balloon support is a fatal error */
    if (r == 0) {
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        "%s", _("cannot set memory of an active domain"));
        goto endjob;
4591
    }
4592

4593
    ret = 0;
4594
endjob:
4595 4596
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
4597

4598
cleanup:
4599 4600
    if (vm)
        virDomainObjUnlock(vm);
4601
    return ret;
4602 4603
}

4604
static int qemudDomainGetInfo(virDomainPtr dom,
4605
                              virDomainInfoPtr info) {
4606 4607 4608
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
4609 4610
    int err;
    unsigned long balloon;
4611

4612
    qemuDriverLock(driver);
4613
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4614
    qemuDriverUnlock(driver);
D
Daniel P. Berrange 已提交
4615
    if (!vm) {
4616 4617
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4618 4619
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4620
        goto cleanup;
D
Daniel P. Berrange 已提交
4621 4622
    }

4623
    info->state = vm->state;
D
Daniel P. Berrange 已提交
4624

D
Daniel P. Berrange 已提交
4625
    if (!virDomainObjIsActive(vm)) {
4626
        info->cpuTime = 0;
D
Daniel P. Berrange 已提交
4627
    } else {
4628
        if (qemudGetProcessInfo(&(info->cpuTime), NULL, vm->pid, 0) < 0) {
4629
            qemuReportError(VIR_ERR_OPERATION_FAILED, ("cannot read cputime for domain"));
4630
            goto cleanup;
D
Daniel P. Berrange 已提交
4631 4632 4633
        }
    }

4634
    info->maxMem = vm->def->maxmem;
4635

D
Daniel P. Berrange 已提交
4636
    if (virDomainObjIsActive(vm)) {
4637
        qemuDomainObjPrivatePtr priv = vm->privateData;
4638 4639 4640
        if (!priv->jobActive) {
            if (qemuDomainObjBeginJob(vm) < 0)
                goto cleanup;
4641

4642 4643 4644 4645
            qemuDomainObjEnterMonitor(vm);
            err = qemuMonitorGetBalloonInfo(priv->mon, &balloon);
            qemuDomainObjExitMonitor(vm);
            if (err < 0) {
4646 4647
                if (qemuDomainObjEndJob(vm) == 0)
                    vm = NULL;
4648 4649 4650 4651 4652 4653 4654 4655 4656
                goto cleanup;
            }

            if (err == 0)
                /* Balloon not supported, so maxmem is always the allocation */
                info->memory = vm->def->maxmem;
            else
                info->memory = balloon;

4657 4658 4659 4660
            if (qemuDomainObjEndJob(vm) == 0) {
                vm = NULL;
                goto cleanup;
            }
4661 4662 4663
        } else {
            info->memory = vm->def->memory;
        }
4664 4665 4666 4667
    } else {
        info->memory = vm->def->memory;
    }

4668
    info->nrVirtCpu = vm->def->vcpus;
4669 4670 4671
    ret = 0;

cleanup:
4672 4673
    if (vm)
        virDomainObjUnlock(vm);
4674
    return ret;
D
Daniel P. Berrange 已提交
4675 4676 4677
}


4678 4679 4680 4681 4682 4683 4684 4685
/** qemuDomainMigrateOffline:
 * Pause domain for non-live migration.
 */
static int
qemuDomainMigrateOffline(struct qemud_driver *driver,
                         virDomainObjPtr vm)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
4686
    int state = vm->state;
4687 4688
    int ret;

4689
    vm->state = VIR_DOMAIN_PAUSED;
4690 4691 4692 4693 4694 4695 4696 4697 4698 4699 4700 4701
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
    ret = qemuMonitorStopCPUs(priv->mon);
    qemuDomainObjExitMonitorWithDriver(driver, vm);

    if (ret == 0) {
        virDomainEventPtr event;

        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_SUSPENDED,
                                         VIR_DOMAIN_EVENT_SUSPENDED_MIGRATED);
        if (event)
            qemuDomainEventQueue(driver, event);
4702 4703
    } else
        vm->state = state;
4704 4705 4706 4707 4708

    return ret;
}


4709 4710 4711 4712 4713 4714 4715 4716 4717 4718 4719 4720 4721 4722
static int
qemuDomainWaitForMigrationComplete(struct qemud_driver *driver, virDomainObjPtr vm)
{
    int ret = -1;
    int status;
    unsigned long long memProcessed;
    unsigned long long memRemaining;
    unsigned long long memTotal;
    qemuDomainObjPrivatePtr priv = vm->privateData;

    priv->jobInfo.type = VIR_DOMAIN_JOB_UNBOUNDED;

    while (priv->jobInfo.type == VIR_DOMAIN_JOB_UNBOUNDED) {
        /* Poll every 50ms for progress & to allow cancellation */
4723
        struct timespec ts = { .tv_sec = 0, .tv_nsec = 50 * 1000 * 1000ull };
4724 4725 4726
        struct timeval now;
        int rc;

4727 4728 4729 4730 4731 4732
        if (!virDomainObjIsActive(vm)) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("guest unexpectedly quit during migration"));
            goto cleanup;
        }

4733 4734
        if (priv->jobSignals & QEMU_JOB_SIGNAL_CANCEL) {
            priv->jobSignals ^= QEMU_JOB_SIGNAL_CANCEL;
4735 4736 4737 4738 4739 4740 4741
            VIR_DEBUG0("Cancelling migration at client request");
            qemuDomainObjEnterMonitorWithDriver(driver, vm);
            rc = qemuMonitorMigrateCancel(priv->mon);
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            if (rc < 0) {
                VIR_WARN0("Unable to cancel migration");
            }
4742 4743 4744 4745 4746
        } else if (priv->jobSignals & QEMU_JOB_SIGNAL_SUSPEND) {
            priv->jobSignals ^= QEMU_JOB_SIGNAL_SUSPEND;
            VIR_DEBUG0("Pausing domain for non-live migration");
            if (qemuDomainMigrateOffline(driver, vm) < 0)
                VIR_WARN0("Unable to pause domain");
4747 4748 4749 4750 4751 4752 4753 4754 4755 4756 4757
        } else if (priv->jobSignals & QEMU_JOB_SIGNAL_MIGRATE_DOWNTIME) {
            unsigned long long ms = priv->jobSignalsData.migrateDowntime;

            priv->jobSignals ^= QEMU_JOB_SIGNAL_MIGRATE_DOWNTIME;
            priv->jobSignalsData.migrateDowntime = 0;
            VIR_DEBUG("Setting migration downtime to %llums", ms);
            qemuDomainObjEnterMonitorWithDriver(driver, vm);
            rc = qemuMonitorSetMigrationDowntime(priv->mon, ms);
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            if (rc < 0)
                VIR_WARN0("Unable to set migration downtime");
4758 4759
        }

4760 4761 4762 4763 4764 4765 4766 4767 4768
        /* Repeat check because the job signals might have caused
         * guest to die
         */
        if (!virDomainObjIsActive(vm)) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("guest unexpectedly quit during migration"));
            goto cleanup;
        }

4769 4770 4771 4772 4773 4774 4775 4776 4777 4778 4779 4780 4781 4782 4783 4784 4785 4786 4787 4788 4789 4790 4791 4792 4793 4794 4795 4796 4797 4798 4799 4800 4801 4802 4803 4804 4805 4806 4807 4808 4809 4810 4811 4812 4813 4814 4815 4816 4817 4818 4819 4820 4821 4822 4823 4824 4825 4826 4827 4828 4829 4830 4831 4832 4833 4834 4835 4836 4837 4838 4839 4840
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
        rc = qemuMonitorGetMigrationStatus(priv->mon,
                                           &status,
                                           &memProcessed,
                                           &memRemaining,
                                           &memTotal);
        qemuDomainObjExitMonitorWithDriver(driver, vm);

        if (rc < 0) {
            priv->jobInfo.type = VIR_DOMAIN_JOB_FAILED;
            goto cleanup;
        }

        if (gettimeofday(&now, NULL) < 0) {
            priv->jobInfo.type = VIR_DOMAIN_JOB_FAILED;
            virReportSystemError(errno, "%s",
                                 _("cannot get time of day"));
            goto cleanup;
        }
        priv->jobInfo.timeElapsed =
            ((now.tv_sec * 1000ull) + (now.tv_usec / 1000)) -
            priv->jobStart;

        switch (status) {
        case QEMU_MONITOR_MIGRATION_STATUS_INACTIVE:
            priv->jobInfo.type = VIR_DOMAIN_JOB_NONE;
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("Migration is not active"));
            break;

        case QEMU_MONITOR_MIGRATION_STATUS_ACTIVE:
            priv->jobInfo.dataTotal = memTotal;
            priv->jobInfo.dataRemaining = memRemaining;
            priv->jobInfo.dataProcessed = memProcessed;

            priv->jobInfo.memTotal = memTotal;
            priv->jobInfo.memRemaining = memRemaining;
            priv->jobInfo.memProcessed = memProcessed;
            break;

        case QEMU_MONITOR_MIGRATION_STATUS_COMPLETED:
            priv->jobInfo.type = VIR_DOMAIN_JOB_COMPLETED;
            ret = 0;
            break;

        case QEMU_MONITOR_MIGRATION_STATUS_ERROR:
            priv->jobInfo.type = VIR_DOMAIN_JOB_FAILED;
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("Migration unexpectedly failed"));
            break;

        case QEMU_MONITOR_MIGRATION_STATUS_CANCELLED:
            priv->jobInfo.type = VIR_DOMAIN_JOB_CANCELLED;
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("Migration was cancelled by client"));
            break;
        }

        virDomainObjUnlock(vm);
        qemuDriverUnlock(driver);

        nanosleep(&ts, NULL);

        qemuDriverLock(driver);
        virDomainObjLock(vm);
    }

cleanup:
    return ret;
}


4841
#define QEMUD_SAVE_MAGIC "LibvirtQemudSave"
4842 4843 4844
#define QEMUD_SAVE_VERSION 2

enum qemud_save_formats {
4845 4846 4847
    QEMUD_SAVE_FORMAT_RAW = 0,
    QEMUD_SAVE_FORMAT_GZIP = 1,
    QEMUD_SAVE_FORMAT_BZIP2 = 2,
4848 4849
    /*
     * Deprecated by xz and never used as part of a release
4850
     * QEMUD_SAVE_FORMAT_LZMA
4851 4852
     */
    QEMUD_SAVE_FORMAT_XZ = 3,
4853
    QEMUD_SAVE_FORMAT_LZOP = 4,
4854 4855 4856
    /* Note: add new members only at the end.
       These values are used in the on-disk format.
       Do not change or re-use numbers. */
4857 4858

    QEMUD_SAVE_FORMAT_LAST
4859
};
4860

4861 4862 4863 4864 4865
VIR_ENUM_DECL(qemudSaveCompression)
VIR_ENUM_IMPL(qemudSaveCompression, QEMUD_SAVE_FORMAT_LAST,
              "raw",
              "gzip",
              "bzip2",
4866 4867
              "xz",
              "lzop")
4868

4869 4870 4871 4872 4873
struct qemud_save_header {
    char magic[sizeof(QEMUD_SAVE_MAGIC)-1];
    int version;
    int xml_len;
    int was_running;
4874 4875
    int compressed;
    int unused[15];
4876 4877
};

4878 4879 4880 4881 4882 4883 4884 4885 4886 4887 4888 4889 4890 4891
struct fileOpHookData {
    virDomainPtr dom;
    const char *path;
    char *xml;
    struct qemud_save_header *header;
};

static int qemudDomainSaveFileOpHook(int fd, void *data) {
    struct fileOpHookData *hdata = data;
    int ret = 0;

    if (safewrite(fd, hdata->header, sizeof(*hdata->header)) != sizeof(*hdata->header)) {
        ret = errno;
        qemuReportError(VIR_ERR_OPERATION_FAILED,
4892 4893
                        _("failed to write header to domain save file '%s'"),
                        hdata->path);
4894 4895 4896 4897 4898 4899 4900 4901 4902 4903 4904 4905 4906 4907
        goto endjob;
    }

    if (safewrite(fd, hdata->xml, hdata->header->xml_len) != hdata->header->xml_len) {
        ret = errno;
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                         _("failed to write xml to '%s'"), hdata->path);
        goto endjob;
    }
endjob:
    return ret;
}


4908 4909
static int qemudDomainSaveFlag(virDomainPtr dom, const char *path,
                               int compressed)
4910
{
4911
    struct qemud_driver *driver = dom->conn->privateData;
4912
    virDomainObjPtr vm = NULL;
4913
    char *xml = NULL;
4914
    struct qemud_save_header header;
4915 4916
    struct fileOpHookData hdata;
    int bypassSecurityDriver = 0;
4917
    int ret = -1;
4918
    int rc;
4919
    virDomainEventPtr event = NULL;
4920
    qemuDomainObjPrivatePtr priv;
4921 4922
    struct stat sb;
    int is_reg = 0;
4923
    unsigned long long offset;
4924
    virCgroupPtr cgroup = NULL;
4925 4926 4927 4928 4929

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

4930
    qemuDriverLock(driver);
4931 4932

    header.compressed = compressed;
4933

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

D
Daniel P. Berrange 已提交
4936
    if (!vm) {
4937 4938
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4939 4940
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4941
        goto cleanup;
D
Daniel P. Berrange 已提交
4942
    }
4943
    priv = vm->privateData;
4944

4945 4946 4947
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
4948
    if (!virDomainObjIsActive(vm)) {
4949 4950
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
4951
        goto endjob;
D
Daniel P. Berrange 已提交
4952
    }
4953

4954 4955 4956
    memset(&priv->jobInfo, 0, sizeof(priv->jobInfo));
    priv->jobInfo.type = VIR_DOMAIN_JOB_UNBOUNDED;

4957 4958 4959
    /* Pause */
    if (vm->state == VIR_DOMAIN_RUNNING) {
        header.was_running = 1;
4960
        vm->state = VIR_DOMAIN_PAUSED;
4961
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
4962
        if (qemuMonitorStopCPUs(priv->mon) < 0) {
4963
            qemuDomainObjExitMonitorWithDriver(driver, vm);
4964
            vm->state = VIR_DOMAIN_RUNNING;
4965
            goto endjob;
4966
        }
4967
        qemuDomainObjExitMonitorWithDriver(driver, vm);
4968 4969 4970 4971 4972 4973

        if (!virDomainObjIsActive(vm)) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("guest unexpectedly quit"));
            goto endjob;
        }
4974 4975 4976
    }

    /* Get XML for the domain */
4977
    xml = virDomainDefFormat(vm->def, VIR_DOMAIN_XML_SECURE);
4978
    if (!xml) {
4979 4980
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to get domain xml"));
4981
        goto endjob;
4982 4983 4984
    }
    header.xml_len = strlen(xml) + 1;

4985 4986 4987 4988 4989 4990 4991 4992 4993 4994 4995 4996 4997
    /* path might be a pre-existing block dev, in which case
     * we need to skip the create step, and also avoid unlink
     * in the failure case */
    if (stat(path, &sb) < 0) {
        /* Avoid throwing an error here, since it is possible
         * that with NFS we can't actually stat() the file.
         * The subsequent codepaths will still raise an error
         * if a truely fatal problem is hit */
        is_reg = 1;
    } else {
        is_reg = S_ISREG(sb.st_mode);
    }

4998 4999 5000 5001 5002
    offset = sizeof(header) + header.xml_len;

    /* Due to way we append QEMU state on our header with dd,
     * we need to ensure there's a 512 byte boundary. Unfortunately
     * we don't have an explicit offset in the header, so we fake
5003 5004
     * it by padding the XML string with NULLs.
     */
5005 5006 5007 5008 5009 5010 5011 5012 5013 5014 5015 5016 5017
    if (offset % QEMU_MONITOR_MIGRATE_TO_FILE_BS) {
        unsigned long long pad =
            QEMU_MONITOR_MIGRATE_TO_FILE_BS -
            (offset % QEMU_MONITOR_MIGRATE_TO_FILE_BS);

        if (VIR_REALLOC_N(xml, header.xml_len + pad) < 0) {
            virReportOOMError();
            goto endjob;
        }
        memset(xml + header.xml_len, 0, pad);
        offset += pad;
        header.xml_len += pad;
    }
5018

5019 5020 5021 5022 5023 5024
    /* Setup hook data needed by virFileOperation hook function */
    hdata.dom = dom;
    hdata.path = path;
    hdata.xml = xml;
    hdata.header = &header;

5025 5026
    /* Write header to file, followed by XML */

5027
    /* First try creating the file as root */
5028 5029 5030 5031
    if (!is_reg) {
        int fd = open(path, O_WRONLY | O_TRUNC);
        if (fd < 0) {
            virReportSystemError(errno, _("unable to open %s"), path);
5032 5033
            goto endjob;
        }
5034
        if (qemudDomainSaveFileOpHook(fd, &hdata) != 0) {
5035 5036 5037 5038 5039
            close(fd);
            goto endjob;
        }
        if (close(fd) < 0) {
            virReportSystemError(errno, _("unable to close %s"), path);
5040 5041
            goto endjob;
        }
5042 5043 5044 5045 5046 5047 5048 5049 5050 5051 5052 5053 5054 5055 5056 5057 5058 5059 5060
    } else {
        if ((rc = virFileOperation(path, O_CREAT|O_TRUNC|O_WRONLY,
                                  S_IRUSR|S_IWUSR,
                                  getuid(), getgid(),
                                  qemudDomainSaveFileOpHook, &hdata,
                                  0)) != 0) {
            /* If we failed as root, and the error was permission-denied
               (EACCES), assume it's on a network-connected share where
               root access is restricted (eg, root-squashed NFS). If the
               qemu user (driver->user) is non-root, just set a flag to
               bypass security driver shenanigans, and retry the operation
               after doing setuid to qemu user */

            if ((rc != EACCES) ||
                driver->user == getuid()) {
                virReportSystemError(rc, _("Failed to create domain save file '%s'"),
                                     path);
                goto endjob;
            }
5061

5062 5063 5064 5065 5066
#ifdef __linux__
            /* On Linux we can also verify the FS-type of the directory. */
            char *dirpath, *p;
            struct statfs st;
            int statfs_ret;
5067

5068 5069
            if ((dirpath = strdup(path)) == NULL) {
                virReportOOMError();
5070 5071 5072
                goto endjob;
            }

5073 5074 5075 5076 5077 5078 5079 5080 5081 5082 5083 5084 5085 5086 5087
            do {
                // Try less and less of the path until we get to a
                // directory we can stat. Even if we don't have 'x'
                // permission on any directory in the path on the NFS
                // server (assuming it's NFS), we will be able to stat the
                // mount point, and that will properly tell us if the
                // fstype is NFS.

                if ((p = strrchr(dirpath, '/')) == NULL) {
                    qemuReportError(VIR_ERR_INVALID_ARG,
                                    _("Invalid relative path '%s' for domain save file"),
                                    path);
                    VIR_FREE(dirpath);
                    goto endjob;
                }
5088

5089 5090 5091 5092
                if (p == dirpath)
                    *(p+1) = '\0';
                else
                    *p = '\0';
5093

5094
                statfs_ret = statfs(dirpath, &st);
5095

5096
            } while ((statfs_ret == -1) && (p != dirpath));
5097

5098 5099
            if (statfs_ret == -1) {
                virReportSystemError(errno,
5100 5101 5102
                                     _("Failed to create domain save file "
                                       "'%s': statfs of all elements of path "
                                       "failed"),
5103 5104 5105 5106 5107 5108 5109 5110 5111 5112 5113 5114 5115
                                     path);
                VIR_FREE(dirpath);
                goto endjob;
            }

            if (st.f_type != NFS_SUPER_MAGIC) {
                virReportSystemError(rc,
                                     _("Failed to create domain save file '%s'"
                                       " (fstype of '%s' is 0x%X"),
                                     path, dirpath, (unsigned int) st.f_type);
                VIR_FREE(dirpath);
                goto endjob;
            }
5116 5117 5118
            VIR_FREE(dirpath);
#endif

5119
            /* Retry creating the file as driver->user */
5120

5121 5122 5123 5124 5125 5126
            if ((rc = virFileOperation(path, O_CREAT|O_TRUNC|O_WRONLY,
                                       S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP,
                                       driver->user, driver->group,
                                       qemudDomainSaveFileOpHook, &hdata,
                                       VIR_FILE_OP_AS_UID)) != 0) {
                virReportSystemError(rc, _("Error from child process creating '%s'"),
5127
                                 path);
5128 5129
                goto endjob;
            }
5130

5131 5132 5133
            /* Since we had to setuid to create the file, and the fstype
               is NFS, we assume it's a root-squashing NFS share, and that
               the security driver stuff would have failed anyway */
5134

5135 5136
            bypassSecurityDriver = 1;
        }
5137
    }
5138

5139

5140 5141 5142 5143 5144 5145 5146 5147 5148 5149 5150 5151 5152 5153 5154 5155 5156
    if (!is_reg &&
        qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_DEVICES)) {
        if (virCgroupForDomain(driver->cgroup, vm->def->name, &cgroup, 0) != 0) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("Unable to find cgroup for %s\n"),
                            vm->def->name);
            goto endjob;
        }
        rc = virCgroupAllowDevicePath(cgroup, path);
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to allow device %s for %s"),
                                 path, vm->def->name);
            goto endjob;
        }
    }

5157 5158
    if ((!bypassSecurityDriver) &&
        driver->securityDriver &&
5159
        driver->securityDriver->domainSetSavedStateLabel &&
5160
        driver->securityDriver->domainSetSavedStateLabel(vm, path) == -1)
5161 5162
        goto endjob;

5163 5164
    if (header.compressed == QEMUD_SAVE_FORMAT_RAW) {
        const char *args[] = { "cat", NULL };
M
Matthias Bolte 已提交
5165
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
5166 5167 5168
        rc = qemuMonitorMigrateToFile(priv->mon,
                                      QEMU_MONITOR_MIGRATE_BACKGROUND,
                                      args, path, offset);
M
Matthias Bolte 已提交
5169
        qemuDomainObjExitMonitorWithDriver(driver, vm);
5170
    } else {
5171
        const char *prog = qemudSaveCompressionTypeToString(header.compressed);
5172 5173 5174 5175 5176
        const char *args[] = {
            prog,
            "-c",
            NULL
        };
M
Matthias Bolte 已提交
5177
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
5178 5179 5180
        rc = qemuMonitorMigrateToFile(priv->mon,
                                      QEMU_MONITOR_MIGRATE_BACKGROUND,
                                      args, path, offset);
M
Matthias Bolte 已提交
5181
        qemuDomainObjExitMonitorWithDriver(driver, vm);
5182 5183
    }

5184 5185 5186 5187 5188
    if (rc < 0)
        goto endjob;

    rc = qemuDomainWaitForMigrationComplete(driver, vm);

5189
    if (rc < 0)
5190
        goto endjob;
5191

5192 5193
    if ((!bypassSecurityDriver) &&
        driver->securityDriver &&
5194
        driver->securityDriver->domainRestoreSavedStateLabel &&
5195
        driver->securityDriver->domainRestoreSavedStateLabel(vm, path) == -1)
5196
        VIR_WARN("failed to restore save state label on %s", path);
5197

5198 5199
    if (cgroup != NULL) {
        rc = virCgroupDenyDevicePath(cgroup, path);
5200 5201 5202
        if (rc != 0)
            VIR_WARN("Unable to deny device %s for %s %d",
                     path, vm->def->name, rc);
5203 5204
    }

5205 5206
    ret = 0;

5207
    /* Shut it down */
5208
    qemudShutdownVMDaemon(driver, vm, 0);
5209 5210 5211
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_SAVED);
5212
    if (!vm->persistent) {
5213 5214 5215
        if (qemuDomainObjEndJob(vm) > 0)
            virDomainRemoveInactive(&driver->domains,
                                    vm);
5216 5217
        vm = NULL;
    }
5218

5219
endjob:
5220
    if (vm) {
5221 5222 5223 5224 5225 5226 5227 5228 5229 5230
        if (ret != 0) {
            if (header.was_running && priv->mon) {
                qemuDomainObjEnterMonitorWithDriver(driver, vm);
                rc = qemuMonitorStartCPUs(priv->mon, dom->conn);
                qemuDomainObjExitMonitorWithDriver(driver, vm);
                if (rc < 0)
                    VIR_WARN0("Unable to resume guest CPUs after save failure");
                else
                    vm->state = VIR_DOMAIN_RUNNING;
            }
5231

5232 5233 5234 5235 5236
            if (cgroup != NULL) {
                rc = virCgroupDenyDevicePath(cgroup, path);
                if (rc != 0)
                    VIR_WARN("Unable to deny device %s for %s: %d",
                             path, vm->def->name, rc);
5237
            }
5238 5239 5240 5241 5242 5243

            if ((!bypassSecurityDriver) &&
                driver->securityDriver &&
                driver->securityDriver->domainRestoreSavedStateLabel &&
                driver->securityDriver->domainRestoreSavedStateLabel(vm, path) == -1)
                VIR_WARN("failed to restore save state label on %s", path);
5244 5245
        }

5246
        if (qemuDomainObjEndJob(vm) == 0)
5247
            vm = NULL;
5248
    }
5249

5250 5251
cleanup:
    VIR_FREE(xml);
5252
    if (ret != 0 && is_reg)
5253
        unlink(path);
5254 5255
    if (vm)
        virDomainObjUnlock(vm);
5256 5257
    if (event)
        qemuDomainEventQueue(driver, event);
5258
    virCgroupFree(&cgroup);
5259
    qemuDriverUnlock(driver);
5260
    return ret;
D
Daniel P. Berrange 已提交
5261 5262
}

5263 5264 5265 5266 5267 5268 5269 5270 5271 5272 5273 5274 5275 5276 5277 5278 5279 5280 5281 5282 5283 5284 5285 5286 5287 5288 5289 5290 5291 5292 5293 5294 5295 5296 5297 5298 5299 5300 5301 5302 5303 5304
static int qemudDomainSave(virDomainPtr dom, const char *path)
{
    struct qemud_driver *driver = dom->conn->privateData;
    int compressed;

    /* Hm, is this safe against qemudReload? */
    if (driver->saveImageFormat == NULL)
        compressed = QEMUD_SAVE_FORMAT_RAW;
    else {
        compressed = qemudSaveCompressionTypeFromString(driver->saveImageFormat);
        if (compressed < 0) {
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("Invalid save image format specified "
                                    "in configuration file"));
            return -1;
        }
    }

    return qemudDomainSaveFlag(dom, path, compressed);
}

static char *
qemuDomainManagedSavePath(struct qemud_driver *driver, virDomainObjPtr vm) {
    char *ret;

    if (virAsprintf(&ret, "%s/%s.save", driver->saveDir, vm->def->name) < 0) {
        virReportOOMError();
        return(NULL);
    }

    return(ret);
}

static int
qemuDomainManagedSave(virDomainPtr dom, unsigned int flags)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm = NULL;
    char *name = NULL;
    int ret = -1;
    int compressed;

5305
    virCheckFlags(0, -1);
5306 5307 5308 5309 5310 5311 5312 5313 5314 5315 5316 5317 5318 5319 5320 5321 5322 5323 5324 5325 5326 5327 5328 5329 5330 5331 5332 5333 5334 5335 5336 5337 5338 5339 5340 5341 5342 5343 5344 5345 5346 5347 5348 5349 5350 5351 5352 5353 5354 5355 5356 5357 5358 5359 5360 5361 5362 5363 5364 5365

    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto error;
    }

    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto error;
    }

    name = qemuDomainManagedSavePath(driver, vm);
    if (name == NULL)
        goto error;

    VIR_DEBUG("Saving state to %s", name);

    /* FIXME: we should take the flags parameter, and use bits out
     * of there to control whether we are compressing or not
     */
    compressed = QEMUD_SAVE_FORMAT_RAW;

    /* we have to drop our locks here because qemudDomainSaveFlag is
     * going to pick them back up.  Unfortunately it opens a race window
     * between us dropping and qemudDomainSaveFlag picking it back up, but
     * if we want to allow other operations to be able to happen while
     * qemuDomainSaveFlag is running (possibly for a long time), I'm not
     * sure if there is a better solution
     */
    virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);

    ret = qemudDomainSaveFlag(dom, name, compressed);

cleanup:
    VIR_FREE(name);

    return ret;

error:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    goto cleanup;
}

static int
qemuDomainHasManagedSaveImage(virDomainPtr dom, unsigned int flags)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm = NULL;
    int ret = -1;
    char *name = NULL;

5366
    virCheckFlags(0, -1);
5367 5368 5369 5370 5371 5372 5373 5374 5375 5376 5377 5378 5379 5380 5381 5382 5383 5384 5385 5386 5387 5388 5389 5390 5391 5392 5393 5394 5395 5396 5397 5398 5399

    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    name = qemuDomainManagedSavePath(driver, vm);
    if (name == NULL)
        goto cleanup;

    ret = virFileExists(name);

cleanup:
    VIR_FREE(name);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}

static int
qemuDomainManagedSaveRemove(virDomainPtr dom, unsigned int flags)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm = NULL;
    int ret = -1;
    char *name = NULL;

5400
    virCheckFlags(0, -1);
5401 5402 5403 5404 5405 5406 5407 5408 5409 5410 5411 5412 5413 5414 5415 5416 5417 5418 5419 5420 5421 5422 5423 5424

    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    name = qemuDomainManagedSavePath(driver, vm);
    if (name == NULL)
        goto cleanup;

    ret = unlink(name);

cleanup:
    VIR_FREE(name);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}
D
Daniel P. Berrange 已提交
5425

P
Paolo Bonzini 已提交
5426 5427 5428 5429 5430 5431
static int qemudDomainCoreDump(virDomainPtr dom,
                               const char *path,
                               int flags ATTRIBUTE_UNUSED) {
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int resume = 0, paused = 0;
5432
    int ret = -1, fd = -1;
5433
    virDomainEventPtr event = NULL;
5434 5435 5436 5437
    const char *args[] = {
        "cat",
        NULL,
    };
5438
    qemuDomainObjPrivatePtr priv;
P
Paolo Bonzini 已提交
5439 5440 5441 5442 5443 5444 5445

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

    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
5446 5447
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
P
Paolo Bonzini 已提交
5448 5449
        goto cleanup;
    }
5450
    priv = vm->privateData;
P
Paolo Bonzini 已提交
5451

5452
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
5453 5454
        goto cleanup;

D
Daniel P. Berrange 已提交
5455
    if (!virDomainObjIsActive(vm)) {
5456 5457
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
5458
        goto endjob;
P
Paolo Bonzini 已提交
5459 5460
    }

5461 5462
    /* Create an empty file with appropriate ownership.  */
    if ((fd = open(path, O_CREAT|O_TRUNC|O_WRONLY, S_IRUSR|S_IWUSR)) < 0) {
5463 5464
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("failed to create '%s'"), path);
5465 5466 5467 5468
        goto endjob;
    }

    if (close(fd) < 0) {
5469
        virReportSystemError(errno,
5470 5471 5472 5473 5474 5475 5476
                             _("unable to save file %s"),
                             path);
        goto endjob;
    }

    if (driver->securityDriver &&
        driver->securityDriver->domainSetSavedStateLabel &&
5477
        driver->securityDriver->domainSetSavedStateLabel(vm, path) == -1)
5478 5479
        goto endjob;

P
Paolo Bonzini 已提交
5480 5481
    /* Migrate will always stop the VM, so the resume condition is
       independent of whether the stop command is issued.  */
P
Paolo Bonzini 已提交
5482 5483 5484
    resume = (vm->state == VIR_DOMAIN_RUNNING);

    /* Pause domain for non-live dump */
P
Paolo Bonzini 已提交
5485
    if (!(flags & VIR_DUMP_LIVE) && vm->state == VIR_DOMAIN_RUNNING) {
5486
        vm->state = VIR_DOMAIN_PAUSED;
5487
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
5488
        if (qemuMonitorStopCPUs(priv->mon) < 0) {
5489
            qemuDomainObjExitMonitorWithDriver(driver, vm);
5490
            vm->state = VIR_DOMAIN_RUNNING;
5491
            goto endjob;
5492
        }
5493
        qemuDomainObjExitMonitorWithDriver(driver, vm);
P
Paolo Bonzini 已提交
5494
        paused = 1;
5495 5496 5497 5498 5499 5500

        if (!virDomainObjIsActive(vm)) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("guest unexpectedly quit"));
            goto endjob;
        }
P
Paolo Bonzini 已提交
5501 5502
    }

5503
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
5504 5505 5506
    ret = qemuMonitorMigrateToFile(priv->mon,
                                   QEMU_MONITOR_MIGRATE_BACKGROUND,
                                   args, path, 0);
5507
    qemuDomainObjExitMonitorWithDriver(driver, vm);
5508 5509 5510 5511 5512 5513 5514 5515 5516
    if (ret < 0)
        goto endjob;

    ret = qemuDomainWaitForMigrationComplete(driver, vm);

    if (ret < 0)
        goto endjob;

    paused = 1;
5517 5518 5519

    if (driver->securityDriver &&
        driver->securityDriver->domainRestoreSavedStateLabel &&
5520
        driver->securityDriver->domainRestoreSavedStateLabel(vm, path) == -1)
5521 5522 5523
        goto endjob;

endjob:
5524
    if ((ret == 0) && (flags & VIR_DUMP_CRASH)) {
5525
        qemudShutdownVMDaemon(driver, vm, 0);
5526 5527 5528 5529 5530
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_CRASHED);
    }

P
Paolo Bonzini 已提交
5531 5532 5533
    /* Since the monitor is always attached to a pty for libvirt, it
       will support synchronous operations so we always get here after
       the migration is complete.  */
5534
    else if (resume && paused && priv->mon) {
5535
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
5536
        if (qemuMonitorStartCPUs(priv->mon, dom->conn) < 0) {
5537
            if (virGetLastError() == NULL)
5538 5539
                qemuReportError(VIR_ERR_OPERATION_FAILED,
                                "%s", _("resuming after dump failed"));
P
Paolo Bonzini 已提交
5540
        }
5541
        qemuDomainObjExitMonitorWithDriver(driver, vm);
5542
        vm->state = VIR_DOMAIN_RUNNING;
P
Paolo Bonzini 已提交
5543
    }
5544

5545 5546
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
5547
    else if ((ret == 0) && (flags & VIR_DUMP_CRASH) && !vm->persistent) {
5548 5549 5550 5551
        virDomainRemoveInactive(&driver->domains,
                                vm);
        vm = NULL;
    }
5552 5553

cleanup:
5554 5555
    if (ret != 0)
        unlink(path);
P
Paolo Bonzini 已提交
5556 5557
    if (vm)
        virDomainObjUnlock(vm);
5558 5559
    if (event)
        qemuDomainEventQueue(driver, event);
5560
    qemuDriverUnlock(driver);
P
Paolo Bonzini 已提交
5561 5562 5563 5564
    return ret;
}


5565 5566 5567 5568 5569 5570
static int qemudDomainHotplugVcpus(virDomainObjPtr vm, unsigned int nvcpus)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int i, rc;
    int ret = -1;

5571 5572
    qemuDomainObjEnterMonitor(vm);

5573 5574 5575 5576 5577 5578 5579 5580 5581 5582 5583 5584 5585 5586 5587 5588 5589 5590 5591 5592 5593 5594 5595 5596 5597 5598 5599 5600 5601 5602
    /* We need different branches here, because we want to offline
     * in reverse order to onlining, so any partial fail leaves us in a
     * reasonably sensible state */
    if (nvcpus > vm->def->vcpus) {
        for (i = vm->def->vcpus ; i < nvcpus ; i++) {
            /* Online new CPU */
            rc = qemuMonitorSetCPU(priv->mon, i, 1);
            if (rc == 0)
                goto unsupported;
            if (rc < 0)
                goto cleanup;

            vm->def->vcpus++;
        }
    } else {
        for (i = vm->def->vcpus - 1 ; i >= nvcpus ; i--) {
            /* Offline old CPU */
            rc = qemuMonitorSetCPU(priv->mon, i, 0);
            if (rc == 0)
                goto unsupported;
            if (rc < 0)
                goto cleanup;

            vm->def->vcpus--;
        }
    }

    ret = 0;

cleanup:
5603
    qemuDomainObjExitMonitor(vm);
5604 5605 5606 5607 5608 5609 5610 5611 5612 5613
    return ret;

unsupported:
    qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                    _("cannot change vcpu count of this domain"));
    goto cleanup;
}


static int qemudDomainSetVcpus(virDomainPtr dom, unsigned int nvcpus) {
5614 5615
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
5616 5617
    const char * type;
    int max;
5618
    int ret = -1;
5619

5620
    qemuDriverLock(driver);
5621
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
5622 5623
    qemuDriverUnlock(driver);

5624
    if (!vm) {
5625 5626
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
5627 5628
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
5629
        goto cleanup;
5630 5631
    }

5632 5633 5634
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

5635 5636 5637
    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                         "%s", _("domain is not running"));
5638
        goto endjob;
5639 5640
    }

5641 5642 5643 5644 5645 5646 5647 5648 5649 5650 5651 5652 5653 5654 5655 5656 5657 5658 5659 5660 5661 5662 5663 5664 5665
    if (!(type = virDomainVirtTypeToString(vm->def->virtType))) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("unknown virt type in domain definition '%d'"),
                        vm->def->virtType);
        goto endjob;
    }

    if ((max = qemudGetMaxVCPUs(NULL, type)) < 0) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                        _("could not determine max vcpus for the domain"));
        goto endjob;
    }

    if (nvcpus > max) {
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("requested vcpus is greater than max allowable"
                          " vcpus for the domain: %d > %d"), nvcpus, max);
        goto endjob;
    }

    ret = qemudDomainHotplugVcpus(vm, nvcpus);

endjob:
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
5666

5667
cleanup:
5668 5669
    if (vm)
        virDomainObjUnlock(vm);
5670
    return ret;
5671 5672
}

5673 5674 5675 5676 5677 5678

static int
qemudDomainPinVcpu(virDomainPtr dom,
                   unsigned int vcpu,
                   unsigned char *cpumap,
                   int maplen) {
5679 5680
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
5681
    int maxcpu, hostcpus;
5682
    virNodeInfo nodeinfo;
5683
    int ret = -1;
5684
    qemuDomainObjPrivatePtr priv;
5685

5686
    qemuDriverLock(driver);
5687
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
5688 5689
    qemuDriverUnlock(driver);

5690 5691 5692
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
5693 5694
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
5695 5696 5697
        goto cleanup;
    }

D
Daniel P. Berrange 已提交
5698
    if (!virDomainObjIsActive(vm)) {
5699 5700
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s",_("cannot pin vcpus on an inactive domain"));
5701
        goto cleanup;
5702 5703
    }

5704 5705 5706
    priv = vm->privateData;

    if (vcpu > (priv->nvcpupids-1)) {
5707 5708 5709
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("vcpu number out of range %d > %d"),
                        vcpu, priv->nvcpupids);
5710
        goto cleanup;
5711 5712
    }

5713
    if (nodeGetInfo(dom->conn, &nodeinfo) < 0)
5714
        goto cleanup;
5715

5716
    hostcpus = VIR_NODEINFO_MAXCPUS(nodeinfo);
5717
    maxcpu = maplen * 8;
5718 5719
    if (maxcpu > hostcpus)
        maxcpu = hostcpus;
5720

5721 5722
    if (priv->vcpupids != NULL) {
        if (virProcessInfoSetAffinity(priv->vcpupids[vcpu],
5723
                                      cpumap, maplen, maxcpu) < 0)
5724
            goto cleanup;
5725
    } else {
5726 5727
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        "%s", _("cpu affinity is not supported"));
5728
        goto cleanup;
5729
    }
5730
    ret = 0;
5731

5732
cleanup:
5733 5734
    if (vm)
        virDomainObjUnlock(vm);
5735
    return ret;
5736 5737 5738 5739 5740 5741 5742 5743
}

static int
qemudDomainGetVcpus(virDomainPtr dom,
                    virVcpuInfoPtr info,
                    int maxinfo,
                    unsigned char *cpumaps,
                    int maplen) {
5744 5745
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
5746
    virNodeInfo nodeinfo;
5747
    int i, v, maxcpu, hostcpus;
5748
    int ret = -1;
5749
    qemuDomainObjPrivatePtr priv;
5750

5751
    qemuDriverLock(driver);
5752
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
5753 5754
    qemuDriverUnlock(driver);

5755 5756 5757
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
5758 5759
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
5760 5761 5762
        goto cleanup;
    }

D
Daniel P. Berrange 已提交
5763
    if (!virDomainObjIsActive(vm)) {
5764 5765 5766
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s",
                        _("cannot list vcpu pinning for an inactive domain"));
5767
        goto cleanup;
5768 5769
    }

5770 5771
    priv = vm->privateData;

5772
    if (nodeGetInfo(dom->conn, &nodeinfo) < 0)
5773
        goto cleanup;
5774

5775
    hostcpus = VIR_NODEINFO_MAXCPUS(nodeinfo);
5776
    maxcpu = maplen * 8;
5777 5778
    if (maxcpu > hostcpus)
        maxcpu = hostcpus;
5779 5780

    /* Clamp to actual number of vcpus */
5781 5782
    if (maxinfo > priv->nvcpupids)
        maxinfo = priv->nvcpupids;
5783

5784 5785 5786 5787 5788 5789
    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;
5790

5791
                if (priv->vcpupids != NULL &&
5792 5793 5794
                    qemudGetProcessInfo(&(info[i].cpuTime),
                                        &(info[i].cpu),
                                        vm->pid,
5795
                                        priv->vcpupids[i]) < 0) {
5796
                    virReportSystemError(errno, "%s",
5797 5798 5799
                                         _("cannot get vCPU placement & pCPU time"));
                    goto cleanup;
                }
5800
            }
5801 5802
        }

5803 5804
        if (cpumaps != NULL) {
            memset(cpumaps, 0, maplen * maxinfo);
5805
            if (priv->vcpupids != NULL) {
5806 5807 5808
                for (v = 0 ; v < maxinfo ; v++) {
                    unsigned char *cpumap = VIR_GET_CPUMAP(cpumaps, maplen, v);

5809
                    if (virProcessInfoGetAffinity(priv->vcpupids[v],
5810
                                                  cpumap, maplen, maxcpu) < 0)
5811
                        goto cleanup;
5812
                }
5813
            } else {
5814 5815
                qemuReportError(VIR_ERR_NO_SUPPORT,
                                "%s", _("cpu affinity is not available"));
5816
                goto cleanup;
5817 5818 5819
            }
        }
    }
5820
    ret = maxinfo;
5821

5822
cleanup:
5823 5824
    if (vm)
        virDomainObjUnlock(vm);
5825
    return ret;
5826 5827 5828
}


5829
static int qemudDomainGetMaxVcpus(virDomainPtr dom) {
5830 5831
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
5832
    const char *type;
5833
    int ret = -1;
5834

5835
    qemuDriverLock(driver);
5836
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
5837 5838
    qemuDriverUnlock(driver);

5839
    if (!vm) {
5840 5841
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
5842 5843
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
5844
        goto cleanup;
5845 5846
    }

5847
    if (!(type = virDomainVirtTypeToString(vm->def->virtType))) {
5848 5849 5850
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("unknown virt type in domain definition '%d'"),
                        vm->def->virtType);
5851
        goto cleanup;
5852 5853
    }

5854
    ret = qemudGetMaxVCPUs(NULL, type);
5855

5856
cleanup:
5857 5858
    if (vm)
        virDomainObjUnlock(vm);
5859 5860 5861
    return ret;
}

5862 5863 5864 5865 5866 5867 5868 5869 5870
static int qemudDomainGetSecurityLabel(virDomainPtr dom, virSecurityLabelPtr seclabel)
{
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;

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

5871 5872
    memset(seclabel, 0, sizeof(*seclabel));

5873 5874 5875
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
5876 5877
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
5878 5879 5880
        goto cleanup;
    }

5881
    if (!virDomainVirtTypeToString(vm->def->virtType)) {
5882 5883 5884
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("unknown virt type in domain definition '%d'"),
                        vm->def->virtType);
5885 5886 5887 5888 5889 5890 5891 5892 5893 5894 5895 5896 5897 5898 5899 5900 5901
        goto cleanup;
    }

    /*
     * Theoretically, the pid can be replaced during this operation and
     * return the label of a different process.  If atomicity is needed,
     * further validation will be required.
     *
     * Comment from Dan Berrange:
     *
     *   Well the PID as stored in the virDomainObjPtr can't be changed
     *   because you've got a locked object.  The OS level PID could have
     *   exited, though and in extreme circumstances have cycled through all
     *   PIDs back to ours. We could sanity check that our PID still exists
     *   after reading the label, by checking that our FD connecting to the
     *   QEMU monitor hasn't seen SIGHUP/ERR on poll().
     */
D
Daniel P. Berrange 已提交
5902
    if (virDomainObjIsActive(vm)) {
5903
        if (driver->securityDriver && driver->securityDriver->domainGetSecurityProcessLabel) {
5904
            if (driver->securityDriver->domainGetSecurityProcessLabel(vm, seclabel) == -1) {
5905 5906
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                "%s", _("Failed to get security label"));
5907 5908 5909 5910 5911 5912 5913 5914 5915 5916
                goto cleanup;
            }
        }
    }

    ret = 0;

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
5917
    qemuDriverUnlock(driver);
5918 5919 5920
    return ret;
}

5921 5922
static int qemudNodeGetSecurityModel(virConnectPtr conn,
                                     virSecurityModelPtr secmodel)
5923 5924 5925
{
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
    char *p;
5926
    int ret = 0;
5927

5928
    qemuDriverLock(driver);
5929
    if (!driver->securityPrimaryDriver) {
5930
        memset(secmodel, 0, sizeof (*secmodel));
5931 5932
        goto cleanup;
    }
5933

5934 5935
    p = driver->caps->host.secModel.model;
    if (strlen(p) >= VIR_SECURITY_MODEL_BUFLEN-1) {
5936 5937 5938
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("security model string exceeds max %d bytes"),
                        VIR_SECURITY_MODEL_BUFLEN-1);
5939 5940
        ret = -1;
        goto cleanup;
5941 5942 5943 5944 5945
    }
    strcpy(secmodel->model, p);

    p = driver->caps->host.secModel.doi;
    if (strlen(p) >= VIR_SECURITY_DOI_BUFLEN-1) {
5946 5947 5948
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("security DOI string exceeds max %d bytes"),
                        VIR_SECURITY_DOI_BUFLEN-1);
5949 5950
        ret = -1;
        goto cleanup;
5951 5952
    }
    strcpy(secmodel->doi, p);
5953 5954 5955 5956

cleanup:
    qemuDriverUnlock(driver);
    return ret;
5957 5958
}

5959 5960 5961 5962 5963 5964 5965 5966 5967 5968 5969 5970 5971 5972 5973 5974 5975 5976 5977 5978 5979 5980 5981 5982 5983 5984 5985 5986 5987 5988 5989 5990 5991 5992 5993 5994 5995 5996 5997 5998 5999 6000 6001 6002 6003 6004 6005 6006 6007 6008 6009 6010 6011 6012 6013 6014 6015 6016 6017 6018 6019 6020 6021 6022 6023 6024 6025 6026 6027 6028 6029 6030 6031 6032 6033 6034 6035 6036 6037 6038 6039 6040 6041 6042 6043 6044 6045 6046 6047 6048 6049 6050 6051 6052 6053 6054 6055 6056 6057 6058 6059 6060 6061 6062 6063 6064 6065 6066 6067 6068 6069 6070 6071 6072 6073 6074 6075 6076 6077 6078 6079 6080 6081 6082 6083 6084 6085 6086 6087 6088 6089 6090
/* qemudOpenAsUID() - pipe/fork/setuid/open a file, and return the
   pipe fd to caller, so that it can read from the file. Also return
   the pid of the child process, so the caller can wait for it to exit
   after it's finished reading (to avoid a zombie, if nothing
   else). */

static int qemudOpenAsUID(const char *path, uid_t uid, pid_t *child_pid) {
    int pipefd[2];
    int fd = -1;

    *child_pid = -1;

    if (pipe(pipefd) < 0) {
        virReportSystemError(errno,
                             _("failed to create pipe to read '%s'"),
                             path);
        pipefd[0] = pipefd[1] = -1;
        goto parent_cleanup;
    }

    int forkRet = virFork(child_pid);

    if (*child_pid < 0) {
        virReportSystemError(errno,
                             _("failed to fork child to read '%s'"),
                             path);
        goto parent_cleanup;
    }

    if (*child_pid > 0) {

        /* parent */

        /* parent doesn't need the write side of the pipe */
        close(pipefd[1]);
        pipefd[1] = -1;

        if (forkRet < 0) {
            virReportSystemError(errno,
                                 _("failed in parent after forking child to read '%s'"),
                                 path);
            goto parent_cleanup;
        }
        /* caller gets the read side of the pipe */
        fd = pipefd[0];
        pipefd[0] = -1;
parent_cleanup:
        if (pipefd[0] != -1)
            close(pipefd[0]);
        if (pipefd[1] != -1)
            close(pipefd[1]);
        if ((fd < 0) && (*child_pid > 0)) {
            /* a child process was started and subsequently an error
               occurred in the parent, so we need to wait for it to
               exit, but its status is inconsequential. */
            while ((waitpid(*child_pid, NULL, 0) == -1)
                   && (errno == EINTR)) {
                /* empty */
            }
            *child_pid = -1;
        }
        return fd;
    }

    /* child */

    /* setuid to the qemu user, then open the file, read it,
       and stuff it into the pipe for the parent process to
       read */
    int exit_code;
    char *buf = NULL;
    size_t bufsize = 1024 * 1024;
    int bytesread;

    /* child doesn't need the read side of the pipe */
    close(pipefd[0]);

    if (forkRet < 0) {
        exit_code = errno;
        virReportSystemError(errno,
                             _("failed in child after forking to read '%s'"),
                             path);
        goto child_cleanup;
    }

    if (setuid(uid) != 0) {
        exit_code = errno;
        virReportSystemError(errno,
                             _("cannot setuid(%d) to read '%s'"),
                             uid, path);
        goto child_cleanup;
    }
    if ((fd = open(path, O_RDONLY)) < 0) {
        exit_code = errno;
        virReportSystemError(errno,
                             _("cannot open '%s' as uid %d"),
                             path, uid);
        goto child_cleanup;
    }
    if (VIR_ALLOC_N(buf, bufsize) < 0) {
        exit_code = ENOMEM;
        virReportOOMError();
        goto child_cleanup;
    }

    /* read from fd and write to pipefd[1] until EOF */
    do {
        if ((bytesread = saferead(fd, buf, bufsize)) < 0) {
            exit_code = errno;
            virReportSystemError(errno,
                                 _("child failed reading from '%s'"),
                                 path);
            goto child_cleanup;
        }
        if (safewrite(pipefd[1], buf, bytesread) != bytesread) {
            exit_code = errno;
            virReportSystemError(errno, "%s",
                                 _("child failed writing to pipe"));
            goto child_cleanup;
        }
    } while (bytesread > 0);
    exit_code = 0;

child_cleanup:
    VIR_FREE(buf);
    if (fd != -1)
        close(fd);
    if (pipefd[1] != -1)
        close(pipefd[1]);
    _exit(exit_code);
}

J
Jiri Denemark 已提交
6091 6092 6093 6094 6095 6096 6097 6098 6099 6100 6101 6102 6103 6104 6105 6106 6107 6108 6109 6110 6111 6112 6113 6114 6115 6116 6117 6118
static int qemudDomainSaveImageClose(int fd, pid_t read_pid, int *status)
{
    int ret = 0;

    if (fd != -1)
        close(fd);

    if (read_pid != -1) {
        /* reap the process that read the file */
        while ((ret = waitpid(read_pid, status, 0)) == -1
               && errno == EINTR) {
            /* empty */
        }
    } else if (status) {
        *status = 0;
    }

    return ret;
}

static int ATTRIBUTE_NONNULL(3) ATTRIBUTE_NONNULL(4) ATTRIBUTE_NONNULL(5)
qemudDomainSaveImageOpen(struct qemud_driver *driver,
                                    const char *path,
                                    virDomainDefPtr *ret_def,
                                    struct qemud_save_header *ret_header,
                                    pid_t *ret_read_pid)
{
    int fd;
6119
    pid_t read_pid = -1;
6120
    struct qemud_save_header header;
J
Jiri Denemark 已提交
6121 6122
    char *xml = NULL;
    virDomainDefPtr def = NULL;
6123 6124

    if ((fd = open(path, O_RDONLY)) < 0) {
6125 6126 6127
        if ((driver->user == 0) || (getuid() != 0)) {
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("cannot read domain image"));
J
Jiri Denemark 已提交
6128
            goto error;
6129 6130 6131 6132 6133 6134 6135 6136
        }

        /* Opening as root failed, but qemu runs as a different user
           that might have better luck. Create a pipe, then fork a
           child process to run as the qemu user, which will hopefully
           have the necessary authority to read the file. */
        if ((fd = qemudOpenAsUID(path, driver->user, &read_pid)) < 0) {
            /* error already reported */
J
Jiri Denemark 已提交
6137
            goto error;
6138
        }
6139 6140 6141
    }

    if (saferead(fd, &header, sizeof(header)) != sizeof(header)) {
6142 6143
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to read qemu header"));
J
Jiri Denemark 已提交
6144
        goto error;
6145 6146 6147
    }

    if (memcmp(header.magic, QEMUD_SAVE_MAGIC, sizeof(header.magic)) != 0) {
6148 6149
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("image magic is incorrect"));
J
Jiri Denemark 已提交
6150
        goto error;
6151 6152 6153
    }

    if (header.version > QEMUD_SAVE_VERSION) {
6154 6155 6156
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("image version is not supported (%d > %d)"),
                        header.version, QEMUD_SAVE_VERSION);
J
Jiri Denemark 已提交
6157
        goto error;
6158 6159
    }

6160 6161 6162
    if (header.xml_len <= 0) {
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("invalid XML length: %d"), header.xml_len);
J
Jiri Denemark 已提交
6163
        goto error;
6164 6165
    }

6166 6167
    if (VIR_ALLOC_N(xml, header.xml_len) < 0) {
        virReportOOMError();
J
Jiri Denemark 已提交
6168
        goto error;
6169 6170 6171
    }

    if (saferead(fd, xml, header.xml_len) != header.xml_len) {
6172 6173
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to read XML"));
J
Jiri Denemark 已提交
6174
        goto error;
6175 6176 6177
    }

    /* Create a domain from this XML */
6178
    if (!(def = virDomainDefParseString(driver->caps, xml,
6179
                                        VIR_DOMAIN_XML_INACTIVE))) {
6180 6181
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to parse XML"));
J
Jiri Denemark 已提交
6182
        goto error;
6183 6184
    }

J
Jiri Denemark 已提交
6185
    VIR_FREE(xml);
6186

J
Jiri Denemark 已提交
6187 6188 6189
    *ret_def = def;
    *ret_header = header;
    *ret_read_pid = read_pid;
6190

J
Jiri Denemark 已提交
6191
    return fd;
6192

J
Jiri Denemark 已提交
6193 6194 6195 6196 6197 6198 6199 6200 6201 6202 6203 6204 6205 6206 6207 6208 6209 6210 6211 6212 6213 6214 6215 6216 6217 6218 6219
error:
    virDomainDefFree(def);
    VIR_FREE(xml);
    qemudDomainSaveImageClose(fd, read_pid, NULL);

    return -1;
}

/* TODO: check seclabel restore */
static int ATTRIBUTE_NONNULL(6)
qemudDomainSaveImageStartVM(virConnectPtr conn,
                            struct qemud_driver *driver,
                            virDomainObjPtr vm,
                            int fd,
                            pid_t read_pid,
                            const struct qemud_save_header *header,
                            const char *path)
{
    int ret = -1;
    virDomainEventPtr event;
    int intermediatefd = -1;
    pid_t intermediate_pid = -1;
    int childstat;
    int wait_ret;
    int status;

    if (header->version == 2) {
6220
        const char *intermediate_argv[3] = { NULL, "-dc", NULL };
J
Jiri Denemark 已提交
6221
        const char *prog = qemudSaveCompressionTypeToString(header->compressed);
6222
        if (prog == NULL) {
6223 6224
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            _("Invalid compressed save format %d"),
J
Jiri Denemark 已提交
6225 6226
                            header->compressed);
            goto out;
6227
        }
6228

J
Jiri Denemark 已提交
6229
        if (header->compressed != QEMUD_SAVE_FORMAT_RAW) {
6230
            intermediate_argv[0] = prog;
6231 6232
            intermediatefd = fd;
            fd = -1;
6233
            if (virExec(intermediate_argv, NULL, NULL,
6234
                        &intermediate_pid, intermediatefd, &fd, NULL, 0) < 0) {
6235 6236 6237
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                _("Failed to start decompression binary %s"),
                                intermediate_argv[0]);
J
Jiri Denemark 已提交
6238
                goto out;
6239 6240 6241
            }
        }
    }
J
Jiri Denemark 已提交
6242

6243
    /* Set the migration source and start it up. */
6244
    ret = qemudStartVMDaemon(conn, driver, vm, "stdio", true, fd, path);
J
Jiri Denemark 已提交
6245

6246 6247 6248
    if (intermediate_pid != -1) {
        /* Wait for intermediate process to exit */
        while (waitpid(intermediate_pid, &childstat, 0) == -1 &&
J
Jiri Denemark 已提交
6249 6250 6251
               errno == EINTR) {
            /* empty */
        }
6252 6253 6254
    }
    if (intermediatefd != -1)
        close(intermediatefd);
J
Jiri Denemark 已提交
6255 6256

    wait_ret = qemudDomainSaveImageClose(fd, read_pid, &status);
6257
    fd = -1;
6258 6259 6260 6261 6262 6263 6264 6265 6266 6267 6268 6269 6270 6271 6272 6273 6274 6275 6276 6277 6278 6279
    if (read_pid != -1) {
        read_pid = -1;
        if (wait_ret == -1) {
            virReportSystemError(errno,
                                 _("failed to wait for process reading '%s'"),
                                 path);
            ret = -1;
        } else if (!WIFEXITED(status)) {
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            _("child process exited abnormally reading '%s'"),
                            path);
            ret = -1;
        } else {
            int exit_status = WEXITSTATUS(status);
            if (exit_status != 0) {
                virReportSystemError(exit_status,
                                     _("child process returned error reading '%s'"),
                                     path);
                ret = -1;
            }
        }
    }
J
Jiri Denemark 已提交
6280 6281 6282

    if (ret < 0)
        goto out;
6283

6284 6285 6286
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_RESTORED);
J
Jiri Denemark 已提交
6287 6288 6289
    if (event)
        qemuDomainEventQueue(driver, event);

6290

6291
    /* If it was running before, resume it now. */
J
Jiri Denemark 已提交
6292
    if (header->was_running) {
6293
        qemuDomainObjPrivatePtr priv = vm->privateData;
6294
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
6295
        if (qemuMonitorStartCPUs(priv->mon, conn) < 0) {
6296
            if (virGetLastError() == NULL)
6297 6298
                qemuReportError(VIR_ERR_OPERATION_FAILED,
                                "%s", _("failed to resume domain"));
6299
            qemuDomainObjExitMonitorWithDriver(driver,vm);
J
Jiri Denemark 已提交
6300
            goto out;
6301
        }
6302
        qemuDomainObjExitMonitorWithDriver(driver, vm);
6303
        vm->state = VIR_DOMAIN_RUNNING;
6304 6305
        if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0) {
            VIR_WARN("Failed to save status on vm %s", vm->def->name);
J
Jiri Denemark 已提交
6306
            goto out;
6307
        }
6308
    }
J
Jiri Denemark 已提交
6309

6310
    ret = 0;
6311

J
Jiri Denemark 已提交
6312 6313 6314 6315 6316 6317 6318 6319 6320 6321 6322 6323 6324 6325 6326 6327 6328 6329 6330 6331 6332 6333 6334 6335 6336 6337
out:
    return ret;
}

static int qemudDomainRestore(virConnectPtr conn,
                              const char *path) {
    struct qemud_driver *driver = conn->privateData;
    virDomainDefPtr def = NULL;
    virDomainObjPtr vm = NULL;
    int fd = -1;
    pid_t read_pid = -1;
    int ret = -1;
    struct qemud_save_header header;

    qemuDriverLock(driver);

    fd = qemudDomainSaveImageOpen(driver, path, &def, &header, &read_pid);
    if (fd < 0)
        goto cleanup;

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

    if (!(vm = virDomainAssignDef(driver->caps,
                                  &driver->domains,
                                  def, true))) {
6338
        /* virDomainAssignDef already set the error */
J
Jiri Denemark 已提交
6339 6340 6341 6342 6343 6344 6345 6346 6347 6348 6349
        goto cleanup;
    }
    def = NULL;

    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

    ret = qemudDomainSaveImageStartVM(conn, driver, vm, fd,
                                      read_pid, &header, path);

    if (qemuDomainObjEndJob(vm) == 0)
6350
        vm = NULL;
J
Jiri Denemark 已提交
6351 6352 6353 6354
    else if (ret < 0 && !vm->persistent) {
        virDomainRemoveInactive(&driver->domains, vm);
        vm = NULL;
    }
6355

6356 6357
cleanup:
    virDomainDefFree(def);
J
Jiri Denemark 已提交
6358
    qemudDomainSaveImageClose(fd, read_pid, NULL);
6359 6360 6361
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
6362
    return ret;
D
Daniel P. Berrange 已提交
6363 6364
}

J
Jiri Denemark 已提交
6365 6366 6367 6368 6369 6370 6371 6372 6373 6374 6375 6376 6377 6378 6379 6380 6381 6382 6383 6384 6385 6386 6387 6388 6389 6390 6391 6392 6393 6394 6395 6396 6397 6398 6399 6400 6401 6402 6403 6404 6405
static int qemudDomainObjRestore(virConnectPtr conn,
                                 struct qemud_driver *driver,
                                 virDomainObjPtr vm,
                                 const char *path)
{
    virDomainDefPtr def = NULL;
    int fd = -1;
    pid_t read_pid = -1;
    int ret = -1;
    struct qemud_save_header header;

    fd = qemudDomainSaveImageOpen(driver, path, &def, &header, &read_pid);
    if (fd < 0)
        goto cleanup;

    if (STRNEQ(vm->def->name, def->name) ||
        memcmp(vm->def->uuid, def->uuid, VIR_UUID_BUFLEN)) {
        char vm_uuidstr[VIR_UUID_STRING_BUFLEN];
        char def_uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(vm->def->uuid, vm_uuidstr);
        virUUIDFormat(def->uuid, def_uuidstr);
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("cannot restore domain '%s' uuid %s from a file"
                          " which belongs to domain '%s' uuid %s"),
                        vm->def->name, vm_uuidstr,
                        def->name, def_uuidstr);
        goto cleanup;
    }

    virDomainObjAssignDef(vm, def, true);
    def = NULL;

    ret = qemudDomainSaveImageStartVM(conn, driver, vm, fd,
                                      read_pid, &header, path);

cleanup:
    virDomainDefFree(def);
    qemudDomainSaveImageClose(fd, read_pid, NULL);
    return ret;
}

D
Daniel P. Berrange 已提交
6406

6407 6408 6409 6410 6411 6412 6413 6414 6415 6416 6417 6418 6419 6420 6421 6422 6423 6424 6425 6426 6427 6428 6429 6430 6431 6432 6433 6434 6435 6436 6437 6438 6439 6440 6441 6442 6443 6444
static char *qemudVMDumpXML(struct qemud_driver *driver,
                            virDomainObjPtr vm,
                            int flags)
{
    char *ret = NULL;
    virCPUDefPtr cpu = NULL;
    virDomainDefPtr def;
    virCPUDefPtr def_cpu;

    if ((flags & VIR_DOMAIN_XML_INACTIVE) && vm->newDef)
        def = vm->newDef;
    else
        def = vm->def;
    def_cpu = def->cpu;

    /* Update guest CPU requirements according to host CPU */
    if ((flags & VIR_DOMAIN_XML_UPDATE_CPU) && def_cpu && def_cpu->model) {
        if (!driver->caps || !driver->caps->host.cpu) {
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("cannot get host CPU capabilities"));
            goto cleanup;
        }

        if (!(cpu = virCPUDefCopy(def_cpu))
            || cpuUpdate(cpu, driver->caps->host.cpu))
            goto cleanup;
        def->cpu = cpu;
    }

    ret = virDomainDefFormat(def, flags);

cleanup:
    def->cpu = def_cpu;
    virCPUDefFree(cpu);
    return ret;
}


6445
static char *qemudDomainDumpXML(virDomainPtr dom,
6446
                                int flags) {
6447 6448 6449
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *ret = NULL;
6450 6451
    unsigned long balloon;
    int err;
6452

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

D
Daniel P. Berrange 已提交
6456
    if (!vm) {
6457 6458
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
6459 6460
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
6461
        goto cleanup;
D
Daniel P. Berrange 已提交
6462 6463
    }

6464
    /* Refresh current memory based on balloon info */
D
Daniel P. Berrange 已提交
6465
    if (virDomainObjIsActive(vm)) {
6466
        qemuDomainObjPrivatePtr priv = vm->privateData;
6467 6468 6469
        /* Don't delay if someone's using the monitor, just use
         * existing most recent data instead */
        if (!priv->jobActive) {
6470
            if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
6471 6472
                goto cleanup;

6473
            qemuDomainObjEnterMonitorWithDriver(driver, vm);
6474
            err = qemuMonitorGetBalloonInfo(priv->mon, &balloon);
6475
            qemuDomainObjExitMonitorWithDriver(driver, vm);
6476 6477 6478 6479
            if (qemuDomainObjEndJob(vm) == 0) {
                vm = NULL;
                goto cleanup;
            }
6480 6481 6482 6483 6484 6485
            if (err < 0)
                goto cleanup;
            if (err > 0)
                vm->def->memory = balloon;
            /* err == 0 indicates no balloon support, so ignore it */
        }
6486
    }
6487

6488
    ret = qemudVMDumpXML(driver, vm, flags);
6489 6490

cleanup:
6491 6492
    if (vm)
        virDomainObjUnlock(vm);
6493
    qemuDriverUnlock(driver);
6494
    return ret;
D
Daniel P. Berrange 已提交
6495 6496 6497
}


6498 6499 6500 6501
static char *qemuDomainXMLFromNative(virConnectPtr conn,
                                     const char *format,
                                     const char *config,
                                     unsigned int flags ATTRIBUTE_UNUSED) {
6502
    struct qemud_driver *driver = conn->privateData;
6503 6504 6505 6506
    virDomainDefPtr def = NULL;
    char *xml = NULL;

    if (STRNEQ(format, QEMU_CONFIG_FORMAT_ARGV)) {
6507 6508
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("unsupported config type %s"), format);
6509 6510 6511
        goto cleanup;
    }

6512
    qemuDriverLock(driver);
6513
    def = qemuParseCommandLineString(driver->caps, config);
6514
    qemuDriverUnlock(driver);
6515 6516 6517
    if (!def)
        goto cleanup;

6518
    xml = virDomainDefFormat(def, VIR_DOMAIN_XML_INACTIVE);
6519 6520 6521 6522 6523 6524

cleanup:
    virDomainDefFree(def);
    return xml;
}

6525 6526 6527 6528 6529 6530
static char *qemuDomainXMLToNative(virConnectPtr conn,
                                   const char *format,
                                   const char *xmlData,
                                   unsigned int flags ATTRIBUTE_UNUSED) {
    struct qemud_driver *driver = conn->privateData;
    virDomainDefPtr def = NULL;
6531
    virDomainChrDef monConfig;
6532
    const char *emulator;
6533
    unsigned long long qemuCmdFlags;
6534 6535 6536 6537 6538 6539 6540 6541
    struct stat sb;
    const char **retargv = NULL;
    const char **retenv = NULL;
    const char **tmp;
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    char *ret = NULL;
    int i;

6542 6543
    qemuDriverLock(driver);

6544
    if (STRNEQ(format, QEMU_CONFIG_FORMAT_ARGV)) {
6545 6546
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("unsupported config type %s"), format);
6547 6548 6549
        goto cleanup;
    }

6550
    def = virDomainDefParseString(driver->caps, xmlData, 0);
6551 6552 6553 6554 6555 6556 6557 6558 6559 6560 6561 6562 6563 6564 6565 6566 6567 6568 6569 6570 6571 6572 6573 6574 6575 6576 6577 6578 6579 6580 6581 6582 6583 6584 6585 6586 6587 6588 6589 6590 6591 6592 6593 6594
    if (!def)
        goto cleanup;

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

            memset(net, 0, sizeof *net);

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

            memset(net, 0, sizeof *net);

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

    /* Make sure the binary we are about to try exec'ing exists.
     * Technically we could catch the exec() failure, but that's
     * in a sub-process so its hard to feed back a useful error
     */
    if (stat(emulator, &sb) < 0) {
6595
        virReportSystemError(errno,
6596 6597 6598 6599 6600 6601 6602 6603
                             _("Cannot find QEMU binary %s"),
                             emulator);
        goto cleanup;
    }

    if (qemudExtractVersionInfo(emulator,
                                NULL,
                                &qemuCmdFlags) < 0) {
6604 6605 6606
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("Cannot determine QEMU argv syntax %s"),
                        emulator);
6607 6608 6609
        goto cleanup;
    }

6610
    if (qemuPrepareMonitorChr(driver, &monConfig, def->name) < 0)
6611
        goto cleanup;
6612 6613

    if (qemudBuildCommandLine(conn, driver, def,
D
Daniel P. Berrange 已提交
6614
                              &monConfig, 0, qemuCmdFlags,
6615 6616
                              &retargv, &retenv,
                              NULL, NULL, /* Don't want it to create TAP devices */
C
Chris Lalancette 已提交
6617
                              NULL, NULL) < 0) {
6618 6619 6620 6621 6622 6623 6624 6625 6626 6627 6628 6629 6630 6631 6632 6633
        goto cleanup;
    }

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

6634 6635
    if (virBufferError(&buf)) {
        virBufferFreeAndReset(&buf);
6636
        virReportOOMError();
6637
        goto cleanup;
6638
    }
6639 6640 6641 6642

    ret = virBufferContentAndReset(&buf);

cleanup:
6643
    qemuDriverUnlock(driver);
6644 6645 6646 6647 6648 6649 6650 6651 6652 6653 6654 6655 6656
    for (tmp = retargv ; tmp && *tmp ; tmp++)
        VIR_FREE(*tmp);
    VIR_FREE(retargv);

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

    virDomainDefFree(def);
    return ret;
}


6657
static int qemudListDefinedDomains(virConnectPtr conn,
6658
                            char **const names, int nnames) {
6659
    struct qemud_driver *driver = conn->privateData;
6660
    int n;
6661

6662
    qemuDriverLock(driver);
6663
    n = virDomainObjListGetInactiveNames(&driver->domains, names, nnames);
6664
    qemuDriverUnlock(driver);
6665
    return n;
D
Daniel P. Berrange 已提交
6666 6667
}

6668
static int qemudNumDefinedDomains(virConnectPtr conn) {
6669
    struct qemud_driver *driver = conn->privateData;
6670
    int n;
6671

6672
    qemuDriverLock(driver);
6673
    n = virDomainObjListNumOfDomains(&driver->domains, 0);
6674
    qemuDriverUnlock(driver);
6675

6676
    return n;
D
Daniel P. Berrange 已提交
6677 6678 6679
}


J
Jiri Denemark 已提交
6680 6681
static int qemudDomainObjStart(virConnectPtr conn,
                               struct qemud_driver *driver,
6682 6683
                               virDomainObjPtr vm,
                               bool start_paused)
J
Jiri Denemark 已提交
6684 6685 6686 6687 6688 6689 6690 6691 6692 6693 6694 6695 6696 6697 6698 6699 6700 6701 6702 6703
{
    int ret = -1;
    char *managed_save;

    /*
     * If there is a managed saved state restore it instead of starting
     * from scratch. In any case the old state is removed.
     */
    managed_save = qemuDomainManagedSavePath(driver, vm);
    if ((managed_save) && (virFileExists(managed_save))) {
        ret = qemudDomainObjRestore(conn, driver, vm, managed_save);

        if (unlink(managed_save) < 0) {
            VIR_WARN("Failed to remove the managed state %s", managed_save);
        }

        if (ret == 0)
            goto cleanup;
    }

6704
    ret = qemudStartVMDaemon(conn, driver, vm, NULL, start_paused, -1, NULL);
J
Jiri Denemark 已提交
6705 6706 6707 6708 6709 6710 6711 6712 6713 6714 6715 6716 6717 6718
    if (ret != -1) {
        virDomainEventPtr event =
            virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_BOOTED);
        if (event)
            qemuDomainEventQueue(driver, event);
    }

cleanup:
    VIR_FREE(managed_save);
    return ret;
}

6719 6720 6721
static int
qemudDomainStartWithFlags(virDomainPtr dom, unsigned int flags)
{
6722 6723 6724
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
6725

6726
    virCheckFlags(VIR_DOMAIN_START_PAUSED, -1);
6727

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

6731
    if (!vm) {
6732 6733
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
6734 6735
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
6736
        goto cleanup;
6737 6738
    }

6739 6740 6741 6742
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

    if (virDomainObjIsActive(vm)) {
6743 6744
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is already running"));
6745 6746 6747
        goto endjob;
    }

6748 6749
    ret = qemudDomainObjStart(dom->conn, driver, vm,
                              (flags & VIR_DOMAIN_START_PAUSED) != 0);
6750

6751
endjob:
6752 6753
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
6754

6755
cleanup:
6756 6757
    if (vm)
        virDomainObjUnlock(vm);
6758
    qemuDriverUnlock(driver);
6759
    return ret;
D
Daniel P. Berrange 已提交
6760 6761
}

6762 6763 6764 6765 6766 6767
static int
qemudDomainStart(virDomainPtr dom)
{
    return qemudDomainStartWithFlags(dom, 0);
}

6768 6769 6770 6771 6772 6773 6774 6775 6776 6777 6778 6779 6780 6781 6782
static int
qemudCanonicalizeMachineFromInfo(virDomainDefPtr def,
                                 virCapsGuestDomainInfoPtr info,
                                 char **canonical)
{
    int i;

    *canonical = NULL;

    for (i = 0; i < info->nmachines; i++) {
        virCapsGuestMachinePtr machine = info->machines[i];

        if (!machine->canonical)
            continue;

6783
        if (STRNEQ(def->os.machine, machine->name))
6784 6785 6786
            continue;

        if (!(*canonical = strdup(machine->canonical))) {
6787
            virReportOOMError();
6788 6789 6790 6791 6792 6793 6794 6795 6796
            return -1;
        }

        break;
    }

    return 0;
}

6797 6798 6799 6800 6801 6802 6803
static int
qemudCanonicalizeMachineDirect(virDomainDefPtr def, char **canonical)
{
    virCapsGuestMachinePtr *machines = NULL;
    int i, nmachines = 0;

    if (qemudProbeMachineTypes(def->emulator, &machines, &nmachines) < 0) {
6804
        virReportOOMError();
6805 6806 6807 6808 6809 6810 6811
        return -1;
    }

    for (i = 0; i < nmachines; i++) {
        if (!machines[i]->canonical)
            continue;

6812
        if (STRNEQ(def->os.machine, machines[i]->name))
6813 6814 6815 6816 6817 6818 6819 6820 6821 6822 6823 6824
            continue;

        *canonical = machines[i]->canonical;
        machines[i]->canonical = NULL;
        break;
    }

    virCapabilitiesFreeMachines(machines, nmachines);

    return 0;
}

6825 6826
int
qemudCanonicalizeMachine(struct qemud_driver *driver, virDomainDefPtr def)
6827 6828 6829 6830 6831 6832
{
    char *canonical = NULL;
    int i;

    for (i = 0; i < driver->caps->nguests; i++) {
        virCapsGuestPtr guest = driver->caps->guests[i];
6833
        virCapsGuestDomainInfoPtr info;
6834 6835 6836
        int j;

        for (j = 0; j < guest->arch.ndomains; j++) {
6837
            info = &guest->arch.domains[j]->info;
6838

6839 6840 6841 6842 6843 6844 6845 6846 6847
            if (!info->emulator || !STREQ(info->emulator, def->emulator))
                continue;

            if (!info->nmachines)
                info = &guest->arch.defaultInfo;

            if (qemudCanonicalizeMachineFromInfo(def, info, &canonical) < 0)
                return -1;
            goto out;
6848 6849
        }

6850 6851 6852 6853
        info = &guest->arch.defaultInfo;

        if (info->emulator && STREQ(info->emulator, def->emulator)) {
            if (qemudCanonicalizeMachineFromInfo(def, info, &canonical) < 0)
6854 6855 6856 6857
                return -1;
            goto out;
        }
    }
6858 6859 6860 6861

    if (qemudCanonicalizeMachineDirect(def, &canonical) < 0)
        return -1;

6862 6863 6864 6865 6866 6867 6868
out:
    if (canonical) {
        VIR_FREE(def->os.machine);
        def->os.machine = canonical;
    }
    return 0;
}
D
Daniel P. Berrange 已提交
6869

6870
static virDomainPtr qemudDomainDefine(virConnectPtr conn, const char *xml) {
6871
    struct qemud_driver *driver = conn->privateData;
6872
    virDomainDefPtr def;
6873
    virDomainObjPtr vm = NULL;
6874
    virDomainPtr dom = NULL;
6875
    virDomainEventPtr event = NULL;
6876
    int dupVM;
6877

6878
    qemuDriverLock(driver);
6879
    if (!(def = virDomainDefParseString(driver->caps, xml,
6880
                                        VIR_DOMAIN_XML_INACTIVE)))
6881
        goto cleanup;
6882

6883
    if (virSecurityDriverVerify(def) < 0)
6884 6885
        goto cleanup;

6886 6887
    if ((dupVM = virDomainObjIsDuplicate(&driver->domains, def, 0)) < 0)
        goto cleanup;
6888

6889
    if (qemudCanonicalizeMachine(driver, def) < 0)
6890 6891
        goto cleanup;

6892 6893 6894
    if (qemuAssignPCIAddresses(def) < 0)
        goto cleanup;

6895
    if (!(vm = virDomainAssignDef(driver->caps,
6896
                                  &driver->domains,
6897
                                  def, false))) {
6898
        goto cleanup;
6899
    }
6900
    def = NULL;
6901
    vm->persistent = 1;
6902

6903
    if (virDomainSaveConfig(driver->configDir,
6904
                            vm->newDef ? vm->newDef : vm->def) < 0) {
6905 6906
        virDomainRemoveInactive(&driver->domains,
                                vm);
6907
        vm = NULL;
6908
        goto cleanup;
6909 6910
    }

6911 6912
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_DEFINED,
6913
                                     !dupVM ?
6914 6915
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED :
                                     VIR_DOMAIN_EVENT_DEFINED_UPDATED);
6916

6917
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
6918
    if (dom) dom->id = vm->def->id;
6919 6920

cleanup:
6921
    virDomainDefFree(def);
6922 6923
    if (vm)
        virDomainObjUnlock(vm);
6924 6925
    if (event)
        qemuDomainEventQueue(driver, event);
6926
    qemuDriverUnlock(driver);
6927
    return dom;
D
Daniel P. Berrange 已提交
6928 6929
}

6930
static int qemudDomainUndefine(virDomainPtr dom) {
6931 6932
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
6933
    virDomainEventPtr event = NULL;
6934
    int ret = -1;
D
Daniel P. Berrange 已提交
6935

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

D
Daniel P. Berrange 已提交
6939
    if (!vm) {
6940 6941
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
6942 6943
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
6944
        goto cleanup;
D
Daniel P. Berrange 已提交
6945 6946
    }

D
Daniel P. Berrange 已提交
6947
    if (virDomainObjIsActive(vm)) {
6948 6949
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot delete active domain"));
6950
        goto cleanup;
D
Daniel P. Berrange 已提交
6951 6952
    }

6953
    if (!vm->persistent) {
6954 6955
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("cannot undefine transient domain"));
6956
        goto cleanup;
6957 6958
    }

6959
    if (virDomainDeleteConfig(driver->configDir, driver->autostartDir, vm) < 0)
6960
        goto cleanup;
D
Daniel P. Berrange 已提交
6961

6962 6963 6964
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_UNDEFINED,
                                     VIR_DOMAIN_EVENT_UNDEFINED_REMOVED);
6965

6966 6967
    virDomainRemoveInactive(&driver->domains,
                            vm);
6968
    vm = NULL;
6969
    ret = 0;
D
Daniel P. Berrange 已提交
6970

6971
cleanup:
6972 6973
    if (vm)
        virDomainObjUnlock(vm);
6974 6975
    if (event)
        qemuDomainEventQueue(driver, event);
6976
    qemuDriverUnlock(driver);
6977
    return ret;
D
Daniel P. Berrange 已提交
6978 6979
}

6980

6981
static int qemudDomainChangeEjectableMedia(struct qemud_driver *driver,
6982
                                           virDomainObjPtr vm,
6983 6984
                                           virDomainDiskDefPtr disk,
                                           unsigned long long qemuCmdFlags)
6985
{
6986
    virDomainDiskDefPtr origdisk = NULL;
6987
    int i;
6988
    int ret;
6989
    char *driveAlias = NULL;
6990

6991
    origdisk = NULL;
6992
    for (i = 0 ; i < vm->def->ndisks ; i++) {
6993 6994
        if (vm->def->disks[i]->bus == disk->bus &&
            STREQ(vm->def->disks[i]->dst, disk->dst)) {
6995
            origdisk = vm->def->disks[i];
6996
            break;
6997
        }
6998 6999 7000
    }

    if (!origdisk) {
7001 7002 7003 7004
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("No device with bus '%s' and target '%s'"),
                        virDomainDiskBusTypeToString(disk->bus),
                        disk->dst);
7005 7006 7007
        return -1;
    }

7008
    if (!origdisk->info.alias) {
7009 7010
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("missing disk device alias name for %s"), origdisk->dst);
7011 7012
        return -1;
    }
7013

7014 7015
    if (origdisk->device != VIR_DOMAIN_DISK_DEVICE_FLOPPY &&
        origdisk->device != VIR_DOMAIN_DISK_DEVICE_CDROM) {
7016 7017 7018
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("Removable media not supported for %s device"),
                        virDomainDiskDeviceTypeToString(disk->device));
7019
        return -1;
7020
    }
7021

7022 7023
    if (driver->securityDriver &&
        driver->securityDriver->domainSetSecurityImageLabel &&
7024
        driver->securityDriver->domainSetSecurityImageLabel(vm, disk) < 0)
7025 7026
        return -1;

7027 7028 7029
    if (!(driveAlias = qemuDeviceDriveHostAlias(origdisk, qemuCmdFlags)))
        goto error;

7030
    qemuDomainObjPrivatePtr priv = vm->privateData;
7031
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
7032
    if (disk->src) {
7033
        const char *format = NULL;
7034 7035 7036
        if (disk->type != VIR_DOMAIN_DISK_TYPE_DIR) {
            if (disk->driverType)
                format = disk->driverType;
7037 7038 7039
            else if (origdisk->driverType)
                format = origdisk->driverType;
        }
7040
        ret = qemuMonitorChangeMedia(priv->mon,
7041
                                     driveAlias,
7042
                                     disk->src, format);
7043
    } else {
7044
        ret = qemuMonitorEjectMedia(priv->mon, driveAlias);
7045
    }
7046
    qemuDomainObjExitMonitorWithDriver(driver, vm);
7047

7048 7049 7050 7051 7052
    if (ret < 0)
        goto error;

    if (driver->securityDriver &&
        driver->securityDriver->domainRestoreSecurityImageLabel &&
7053
        driver->securityDriver->domainRestoreSecurityImageLabel(vm, origdisk) < 0)
7054 7055 7056
        VIR_WARN("Unable to restore security label on ejected image %s", origdisk->src);

    VIR_FREE(origdisk->src);
7057 7058 7059 7060
    origdisk->src = disk->src;
    disk->src = NULL;
    origdisk->type = disk->type;

7061 7062
    VIR_FREE(driveAlias);

7063
    virDomainDiskDefFree(disk);
7064

7065
    return ret;
7066 7067

error:
7068
    VIR_FREE(driveAlias);
7069 7070
    if (driver->securityDriver &&
        driver->securityDriver->domainRestoreSecurityImageLabel &&
7071
        driver->securityDriver->domainRestoreSecurityImageLabel(vm, disk) < 0)
7072
        VIR_WARN("Unable to restore security label on new media %s", disk->src);
7073
    return -1;
7074 7075
}

7076

7077
static int qemudDomainAttachPciDiskDevice(struct qemud_driver *driver,
7078
                                          virDomainObjPtr vm,
7079
                                          virDomainDiskDefPtr disk,
7080
                                          unsigned long long qemuCmdFlags)
7081
{
7082
    int i, ret;
7083
    const char* type = virDomainDiskBusTypeToString(disk->bus);
7084
    qemuDomainObjPrivatePtr priv = vm->privateData;
7085 7086
    char *devstr = NULL;
    char *drivestr = NULL;
7087 7088

    for (i = 0 ; i < vm->def->ndisks ; i++) {
7089
        if (STREQ(vm->def->disks[i]->dst, disk->dst)) {
7090 7091
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            _("target %s already exists"), disk->dst);
7092 7093 7094 7095
            return -1;
        }
    }

7096 7097
    if (driver->securityDriver &&
        driver->securityDriver->domainSetSecurityImageLabel &&
7098
        driver->securityDriver->domainSetSecurityImageLabel(vm, disk) < 0)
7099 7100
        return -1;

7101
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
7102 7103
        if (qemuDomainPCIAddressEnsureAddr(priv->pciaddrs, &disk->info) < 0)
            goto error;
7104 7105
        if (qemuAssignDeviceDiskAlias(disk, qemuCmdFlags) < 0)
            goto error;
7106

7107 7108 7109
        if (!(drivestr = qemuBuildDriveStr(disk, 0, qemuCmdFlags)))
            goto error;

7110
        if (!(devstr = qemuBuildDriveDevStr(disk)))
7111 7112 7113
            goto error;
    }

7114
    if (VIR_REALLOC_N(vm->def->disks, vm->def->ndisks+1) < 0) {
7115
        virReportOOMError();
7116
        goto error;
7117 7118
    }

7119
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
7120 7121
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        ret = qemuMonitorAddDrive(priv->mon, drivestr);
7122 7123 7124
        if (ret == 0) {
            ret = qemuMonitorAddDevice(priv->mon, devstr);
            if (ret < 0) {
7125
                VIR_WARN("qemuMonitorAddDevice failed on %s (%s)",
7126 7127 7128 7129 7130
                         drivestr, devstr);
                /* XXX should call 'drive_del' on error but this does not
                   exist yet */
            }
        }
7131 7132 7133 7134 7135 7136 7137 7138 7139 7140 7141
    } else {
        virDomainDevicePCIAddress guestAddr;
        ret = qemuMonitorAddPCIDisk(priv->mon,
                                    disk->src,
                                    type,
                                    &guestAddr);
        if (ret == 0) {
            disk->info.type = VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI;
            memcpy(&disk->info.addr.pci, &guestAddr, sizeof(guestAddr));
        }
    }
7142
    qemuDomainObjExitMonitorWithDriver(driver, vm);
7143

7144 7145
    if (ret < 0)
        goto error;
7146

7147
    virDomainDiskInsertPreAlloced(vm->def, disk);
7148

7149 7150 7151
    VIR_FREE(devstr);
    VIR_FREE(drivestr);

7152 7153 7154
    return 0;

error:
7155 7156 7157
    VIR_FREE(devstr);
    VIR_FREE(drivestr);

7158 7159 7160 7161 7162
    if ((qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) &&
        (disk->info.type == VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI) &&
        qemuDomainPCIAddressReleaseAddr(priv->pciaddrs, &disk->info) < 0)
        VIR_WARN("Unable to release PCI address on %s", disk->src);

7163 7164
    if (driver->securityDriver &&
        driver->securityDriver->domainRestoreSecurityImageLabel &&
7165
        driver->securityDriver->domainRestoreSecurityImageLabel(vm, disk) < 0)
7166
        VIR_WARN("Unable to restore security label on %s", disk->src);
7167 7168

    return -1;
7169
}
7170

7171

7172
static int qemudDomainAttachPciControllerDevice(struct qemud_driver *driver,
7173
                                                virDomainObjPtr vm,
7174
                                                virDomainControllerDefPtr controller,
7175
                                                unsigned long long qemuCmdFlags)
7176
{
7177 7178
    int i;
    int ret = -1;
7179
    const char* type = virDomainControllerTypeToString(controller->type);
7180
    char *devstr = NULL;
7181 7182 7183
    qemuDomainObjPrivatePtr priv = vm->privateData;

    for (i = 0 ; i < vm->def->ncontrollers ; i++) {
7184 7185
        if ((vm->def->controllers[i]->type == controller->type) &&
            (vm->def->controllers[i]->idx == controller->idx)) {
7186 7187 7188
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            _("target %s:%d already exists"),
                            type, controller->idx);
7189 7190 7191 7192
            return -1;
        }
    }

7193 7194 7195 7196 7197 7198
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        if (qemuDomainPCIAddressEnsureAddr(priv->pciaddrs, &controller->info) < 0)
            goto cleanup;
        if (qemuAssignDeviceControllerAlias(controller) < 0)
            goto cleanup;
    }
7199

7200
    if (!(devstr = qemuBuildControllerDevStr(controller))) {
7201
        virReportOOMError();
7202 7203 7204
        goto cleanup;
    }

7205
    if (VIR_REALLOC_N(vm->def->controllers, vm->def->ncontrollers+1) < 0) {
7206
        virReportOOMError();
7207
        goto cleanup;
7208 7209 7210
    }

    qemuDomainObjEnterMonitorWithDriver(driver, vm);
7211 7212 7213 7214 7215 7216 7217
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        ret = qemuMonitorAddDevice(priv->mon, devstr);
    } else {
        ret = qemuMonitorAttachPCIDiskController(priv->mon,
                                                 type,
                                                 &controller->info.addr.pci);
    }
7218 7219 7220
    qemuDomainObjExitMonitorWithDriver(driver, vm);

    if (ret == 0) {
7221 7222
        controller->info.type = VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI;
        virDomainControllerInsertPreAlloced(vm->def, controller);
7223 7224
    }

7225
cleanup:
7226 7227 7228 7229 7230 7231
    if ((ret != 0) &&
        (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) &&
        (controller->info.type == VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI) &&
        qemuDomainPCIAddressReleaseAddr(priv->pciaddrs, &controller->info) < 0)
        VIR_WARN0("Unable to release PCI address on controller");

7232
    VIR_FREE(devstr);
7233 7234 7235
    return ret;
}

7236

7237
static virDomainControllerDefPtr
7238
qemuDomainFindOrCreateSCSIDiskController(struct qemud_driver *driver,
7239
                                         virDomainObjPtr vm,
7240
                                         int controller,
7241
                                         unsigned long long qemuCmdFlags)
7242 7243 7244 7245 7246 7247 7248 7249 7250 7251 7252 7253 7254
{
    int i;
    virDomainControllerDefPtr cont;
    for (i = 0 ; i < vm->def->ncontrollers ; i++) {
        cont = vm->def->controllers[i];

        if (cont->type != VIR_DOMAIN_CONTROLLER_TYPE_SCSI)
            continue;

        if (cont->idx == controller)
            return cont;
    }

W
Wolfgang Mauerer 已提交
7255
    /* No SCSI controller present, for backward compatibility we
7256 7257
     * now hotplug a controller */
    if (VIR_ALLOC(cont) < 0) {
7258
        virReportOOMError();
7259 7260 7261 7262 7263 7264
        return NULL;
    }
    cont->type = VIR_DOMAIN_CONTROLLER_TYPE_SCSI;
    cont->idx = 0;

    VIR_INFO0("No SCSI controller present, hotplugging one");
7265
    if (qemudDomainAttachPciControllerDevice(driver,
7266
                                             vm, cont, qemuCmdFlags) < 0) {
7267 7268 7269
        VIR_FREE(cont);
        return NULL;
    }
7270 7271 7272 7273 7274 7275 7276 7277 7278

    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                        _("guest unexpectedly quit"));
        /* cont doesn't need freeing here, since the reference
         * now held in def->controllers */
        return NULL;
    }

7279 7280 7281
    return cont;
}

7282

7283
static int qemudDomainAttachSCSIDisk(struct qemud_driver *driver,
7284
                                     virDomainObjPtr vm,
7285
                                     virDomainDiskDefPtr disk,
7286
                                     unsigned long long qemuCmdFlags)
7287 7288 7289
{
    int i;
    qemuDomainObjPrivatePtr priv = vm->privateData;
7290
    virDomainControllerDefPtr cont = NULL;
7291
    char *drivestr = NULL;
7292
    char *devstr = NULL;
7293 7294 7295
    int ret = -1;

    for (i = 0 ; i < vm->def->ndisks ; i++) {
7296
        if (STREQ(vm->def->disks[i]->dst, disk->dst)) {
7297 7298
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            _("target %s already exists"), disk->dst);
7299
            return -1;
7300 7301 7302
        }
    }

7303 7304 7305

    if (driver->securityDriver &&
        driver->securityDriver->domainSetSecurityImageLabel &&
7306
        driver->securityDriver->domainSetSecurityImageLabel(vm, disk) < 0)
7307 7308
        return -1;

7309
    /* We should have an address already, so make sure */
7310
    if (disk->info.type != VIR_DOMAIN_DEVICE_ADDRESS_TYPE_DRIVE) {
7311 7312 7313
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("unexpected disk address type %s"),
                        virDomainDeviceAddressTypeToString(disk->info.type));
7314
        goto error;
7315 7316
    }

7317 7318 7319
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        if (qemuAssignDeviceDiskAlias(disk, qemuCmdFlags) < 0)
            goto error;
7320
        if (!(devstr = qemuBuildDriveDevStr(disk)))
7321 7322 7323 7324 7325 7326
            goto error;
    }

    if (!(drivestr = qemuBuildDriveStr(disk, 0, qemuCmdFlags)))
        goto error;

7327
    for (i = 0 ; i <= disk->info.addr.drive.controller ; i++) {
7328
        cont = qemuDomainFindOrCreateSCSIDiskController(driver, vm, i, qemuCmdFlags);
7329
        if (!cont)
7330
            goto error;
7331 7332
    }

7333 7334 7335 7336 7337
    /* Tell clang that "cont" is non-NULL.
       This is because disk->info.addr.driver.controller is unsigned,
       and hence the above loop must iterate at least once.  */
    sa_assert (cont);

7338
    if (cont->info.type != VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI) {
7339 7340
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("SCSI controller %d was missing its PCI address"), cont->idx);
7341
        goto error;
7342 7343 7344
    }

    if (VIR_REALLOC_N(vm->def->disks, vm->def->ndisks+1) < 0) {
7345
        virReportOOMError();
7346
        goto error;
7347 7348 7349
    }

    qemuDomainObjEnterMonitorWithDriver(driver, vm);
7350
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
7351 7352 7353 7354
        ret = qemuMonitorAddDrive(priv->mon, drivestr);
        if (ret == 0) {
            ret = qemuMonitorAddDevice(priv->mon, devstr);
            if (ret < 0) {
7355
                VIR_WARN("qemuMonitorAddDevice failed on %s (%s)",
7356 7357 7358 7359 7360
                         drivestr, devstr);
                /* XXX should call 'drive_del' on error but this does not
                   exist yet */
            }
        }
7361 7362 7363 7364 7365 7366 7367 7368 7369 7370 7371 7372 7373
    } else {
        virDomainDeviceDriveAddress driveAddr;
        ret = qemuMonitorAttachDrive(priv->mon,
                                     drivestr,
                                     &cont->info.addr.pci,
                                     &driveAddr);
        if (ret == 0) {
            /* XXX we should probably validate that the addr matches
             * our existing defined addr instead of overwriting */
            disk->info.type = VIR_DOMAIN_DEVICE_ADDRESS_TYPE_DRIVE;
            memcpy(&disk->info.addr.drive, &driveAddr, sizeof(driveAddr));
        }
    }
7374 7375
    qemuDomainObjExitMonitorWithDriver(driver, vm);

7376 7377
    if (ret < 0)
        goto error;
7378

7379
    virDomainDiskInsertPreAlloced(vm->def, disk);
7380 7381

    VIR_FREE(devstr);
7382
    VIR_FREE(drivestr);
7383 7384 7385 7386

    return 0;

error:
7387
    VIR_FREE(devstr);
7388
    VIR_FREE(drivestr);
7389

7390 7391
    if (driver->securityDriver &&
        driver->securityDriver->domainRestoreSecurityImageLabel &&
7392
        driver->securityDriver->domainRestoreSecurityImageLabel(vm, disk) < 0)
7393
        VIR_WARN("Unable to restore security label on %s", disk->src);
7394 7395

    return -1;
7396 7397
}

7398

7399
static int qemudDomainAttachUsbMassstorageDevice(struct qemud_driver *driver,
7400
                                                 virDomainObjPtr vm,
7401
                                                 virDomainDiskDefPtr disk,
7402
                                                 unsigned long long qemuCmdFlags)
7403
{
7404
    qemuDomainObjPrivatePtr priv = vm->privateData;
7405
    int i, ret;
7406 7407
    char *drivestr = NULL;
    char *devstr = NULL;
7408

7409
    for (i = 0 ; i < vm->def->ndisks ; i++) {
7410
        if (STREQ(vm->def->disks[i]->dst, disk->dst)) {
7411 7412
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            _("target %s already exists"), disk->dst);
7413 7414 7415 7416
            return -1;
        }
    }

7417 7418
    if (driver->securityDriver &&
        driver->securityDriver->domainSetSecurityImageLabel &&
7419
        driver->securityDriver->domainSetSecurityImageLabel(vm, disk) < 0)
7420 7421
        return -1;

7422
    if (!disk->src) {
7423 7424
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("disk source path is missing"));
7425
        goto error;
7426 7427
    }

7428
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
7429 7430
        if (qemuAssignDeviceDiskAlias(disk, qemuCmdFlags) < 0)
            goto error;
7431 7432
        if (!(drivestr = qemuBuildDriveStr(disk, 0, qemuCmdFlags)))
            goto error;
7433
        if (!(devstr = qemuBuildDriveDevStr(disk)))
7434 7435 7436
            goto error;
    }

7437
    if (VIR_REALLOC_N(vm->def->disks, vm->def->ndisks+1) < 0) {
7438
        virReportOOMError();
7439
        goto error;
7440 7441
    }

7442
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
7443
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
7444 7445 7446 7447
        ret = qemuMonitorAddDrive(priv->mon, drivestr);
        if (ret == 0) {
            ret = qemuMonitorAddDevice(priv->mon, devstr);
            if (ret < 0) {
7448
                VIR_WARN("qemuMonitorAddDevice failed on %s (%s)",
7449 7450 7451 7452 7453
                         drivestr, devstr);
                /* XXX should call 'drive_del' on error but this does not
                   exist yet */
            }
        }
7454 7455 7456
    } else {
        ret = qemuMonitorAddUSBDisk(priv->mon, disk->src);
    }
7457
    qemuDomainObjExitMonitorWithDriver(driver, vm);
7458

7459 7460 7461
    if (ret < 0)
        goto error;

7462
    virDomainDiskInsertPreAlloced(vm->def, disk);
7463

7464 7465 7466
    VIR_FREE(devstr);
    VIR_FREE(drivestr);

7467 7468 7469
    return 0;

error:
7470 7471 7472
    VIR_FREE(devstr);
    VIR_FREE(drivestr);

7473 7474
    if (driver->securityDriver &&
        driver->securityDriver->domainRestoreSecurityImageLabel &&
7475
        driver->securityDriver->domainRestoreSecurityImageLabel(vm, disk) < 0)
7476
        VIR_WARN("Unable to restore security label on %s", disk->src);
7477 7478

    return -1;
7479 7480
}

7481

7482
/* XXX conn required for network -> bridge resolution */
M
Mark McLoughlin 已提交
7483
static int qemudDomainAttachNetDevice(virConnectPtr conn,
7484
                                      struct qemud_driver *driver,
M
Mark McLoughlin 已提交
7485
                                      virDomainObjPtr vm,
7486
                                      virDomainNetDefPtr net,
7487
                                      unsigned long long qemuCmdFlags)
M
Mark McLoughlin 已提交
7488
{
7489
    qemuDomainObjPrivatePtr priv = vm->privateData;
7490
    char *tapfd_name = NULL;
7491
    int tapfd = -1;
7492
    char *nicstr = NULL;
7493
    char *netstr = NULL;
7494
    int ret = -1;
7495
    virDomainDevicePCIAddress guestAddr;
7496
    int vlan;
M
Mark McLoughlin 已提交
7497 7498

    if (!(qemuCmdFlags & QEMUD_CMD_FLAG_HOST_NET_ADD)) {
7499 7500
        qemuReportError(VIR_ERR_NO_SUPPORT, "%s",
                        _("installed qemu version does not support host_net_add"));
M
Mark McLoughlin 已提交
7501 7502 7503 7504 7505
        return -1;
    }

    if (net->type == VIR_DOMAIN_NET_TYPE_BRIDGE ||
        net->type == VIR_DOMAIN_NET_TYPE_NETWORK) {
7506
        if (priv->monConfig->type != VIR_DOMAIN_CHR_TYPE_UNIX) {
7507 7508 7509 7510
            qemuReportError(VIR_ERR_NO_SUPPORT,
                            _("network device type '%s' cannot be attached: "
                              "qemu is not using a unix socket monitor"),
                            virDomainNetTypeToString(net->type));
7511 7512 7513 7514 7515
            return -1;
        }

        if ((tapfd = qemudNetworkIfaceConnect(conn, driver, net, qemuCmdFlags)) < 0)
            return -1;
7516 7517 7518 7519 7520 7521 7522 7523 7524
    } else if (net->type == VIR_DOMAIN_NET_TYPE_DIRECT) {
        if (priv->monConfig->type != VIR_DOMAIN_CHR_TYPE_UNIX) {
            qemuReportError(VIR_ERR_NO_SUPPORT,
                            _("network device type '%s' cannot be attached: "
                            "qemu is not using a unix socket monitor"),
                            virDomainNetTypeToString(net->type));
            return -1;
        }

S
Stefan Berger 已提交
7525
        if ((tapfd = qemudPhysIfaceConnect(conn, driver, net,
7526 7527
                                           qemuCmdFlags,
                                           vm->def->uuid)) < 0)
7528
            return -1;
M
Mark McLoughlin 已提交
7529 7530
    }

7531 7532
    if (VIR_REALLOC_N(vm->def->nets, vm->def->nnets+1) < 0)
        goto no_memory;
M
Mark McLoughlin 已提交
7533

7534 7535 7536 7537 7538
    if ((qemuCmdFlags & QEMUD_CMD_FLAG_NET_NAME) ||
        (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE)) {
        if (qemuAssignDeviceNetAlias(vm->def, net, -1) < 0)
            goto cleanup;
    }
M
Mark McLoughlin 已提交
7539

7540 7541 7542 7543
    if ((qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) &&
        qemuDomainPCIAddressEnsureAddr(priv->pciaddrs, &net->info) < 0)
        goto cleanup;

7544 7545 7546 7547 7548
    if ((qemuCmdFlags & QEMUD_CMD_FLAG_NETDEV) &&
        (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE)) {
        vlan = -1;
    } else {
        vlan = qemuDomainNetVLAN(net);
M
Mark McLoughlin 已提交
7549

7550 7551 7552 7553 7554
        if (vlan < 0) {
            qemuReportError(VIR_ERR_NO_SUPPORT, "%s",
                            _("Unable to attach network devices without vlan"));
            goto cleanup;
        }
7555 7556
    }

7557
    if (tapfd != -1) {
7558
        if (virAsprintf(&tapfd_name, "fd-%s", net->info.alias) < 0)
7559 7560
            goto no_memory;

7561
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
7562
        if (qemuMonitorSendFileHandle(priv->mon, tapfd_name, tapfd) < 0) {
7563
            qemuDomainObjExitMonitorWithDriver(driver, vm);
7564
            goto cleanup;
7565
        }
7566
        qemuDomainObjExitMonitorWithDriver(driver, vm);
7567 7568 7569 7570 7571 7572

        if (!virDomainObjIsActive(vm)) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("guest unexpectedly quit"));
            goto cleanup;
        }
7573 7574
    }

7575
    /* FIXME - need to support vhost-net here (5th arg) */
7576 7577 7578 7579 7580 7581 7582 7583 7584 7585
    if ((qemuCmdFlags & QEMUD_CMD_FLAG_NETDEV) &&
        (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE)) {
        if (!(netstr = qemuBuildHostNetStr(net, ',',
                                           -1, tapfd_name, 0)))
            goto try_tapfd_close;
    } else {
        if (!(netstr = qemuBuildHostNetStr(net, ' ',
                                           vlan, tapfd_name, 0)))
            goto try_tapfd_close;
    }
M
Mark McLoughlin 已提交
7586

7587
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
7588 7589 7590 7591 7592 7593 7594 7595 7596 7597 7598
    if ((qemuCmdFlags & QEMUD_CMD_FLAG_NETDEV) &&
        (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE)) {
        if (qemuMonitorAddNetdev(priv->mon, netstr) < 0) {
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            goto try_tapfd_close;
        }
    } else {
        if (qemuMonitorAddHostNetwork(priv->mon, netstr) < 0) {
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            goto try_tapfd_close;
        }
7599
    }
7600
    qemuDomainObjExitMonitorWithDriver(driver, vm);
M
Mark McLoughlin 已提交
7601

7602 7603 7604
    if (tapfd != -1)
        close(tapfd);
    tapfd = -1;
M
Mark McLoughlin 已提交
7605

7606 7607 7608 7609 7610 7611
    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                        _("guest unexpectedly quit"));
        goto cleanup;
    }

7612 7613 7614 7615 7616 7617 7618
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        if (!(nicstr = qemuBuildNicDevStr(net, vlan)))
            goto try_remove;
    } else {
        if (!(nicstr = qemuBuildNicStr(net, NULL, vlan)))
            goto try_remove;
    }
M
Mark McLoughlin 已提交
7619

7620
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
7621 7622 7623 7624 7625 7626 7627 7628 7629 7630 7631 7632 7633
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        if (qemuMonitorAddDevice(priv->mon, nicstr) < 0) {
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            goto try_remove;
        }
    } else {
        if (qemuMonitorAddPCINetwork(priv->mon, nicstr,
                                     &guestAddr) < 0) {
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            goto try_remove;
        }
        net->info.type = VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI;
        memcpy(&net->info.addr.pci, &guestAddr, sizeof(guestAddr));
7634
    }
7635
    qemuDomainObjExitMonitorWithDriver(driver, vm);
7636

7637
    ret = 0;
M
Mark McLoughlin 已提交
7638 7639 7640

    vm->def->nets[vm->def->nnets++] = net;

7641
cleanup:
7642 7643 7644 7645 7646 7647
    if ((ret != 0) &&
        (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) &&
        (net->info.type == VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI) &&
        qemuDomainPCIAddressReleaseAddr(priv->pciaddrs, &net->info) < 0)
        VIR_WARN0("Unable to release PCI address on NIC");

7648 7649
    if (ret != 0)
        virNWFilterTearNWFilter(net);
S
Stefan Berger 已提交
7650

7651 7652 7653 7654 7655
    VIR_FREE(nicstr);
    VIR_FREE(netstr);
    VIR_FREE(tapfd_name);
    if (tapfd != -1)
        close(tapfd);
7656

7657
    return ret;
7658

7659
try_remove:
7660 7661 7662
    if (!priv->mon)
        goto cleanup;

7663
    if (vlan < 0) {
7664 7665 7666 7667 7668 7669 7670
        if ((qemuCmdFlags & QEMUD_CMD_FLAG_NETDEV) &&
            (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE)) {
            char *netdev_name;
            if (virAsprintf(&netdev_name, "host%s", net->info.alias) < 0)
                goto no_memory;
            qemuDomainObjEnterMonitorWithDriver(driver, vm);
            if (qemuMonitorRemoveNetdev(priv->mon, netdev_name) < 0)
7671
                VIR_WARN("Failed to remove network backend for netdev %s",
7672 7673 7674 7675
                         netdev_name);
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            VIR_FREE(netdev_name);
        } else {
7676
            VIR_WARN0("Unable to remove network backend");
7677
        }
7678 7679 7680 7681
    } else {
        char *hostnet_name;
        if (virAsprintf(&hostnet_name, "host%s", net->info.alias) < 0)
            goto no_memory;
7682
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
7683
        if (qemuMonitorRemoveHostNetwork(priv->mon, vlan, hostnet_name) < 0)
7684
            VIR_WARN("Failed to remove network backend for vlan %d, net %s",
7685
                     vlan, hostnet_name);
7686
        qemuDomainObjExitMonitorWithDriver(driver, vm);
7687
        VIR_FREE(hostnet_name);
7688
    }
7689
    goto cleanup;
7690

7691
try_tapfd_close:
7692 7693 7694
    if (!priv->mon)
        goto cleanup;

7695
    if (tapfd_name) {
7696
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
7697
        if (qemuMonitorCloseFileHandle(priv->mon, tapfd_name) < 0)
7698
            VIR_WARN("Failed to close tapfd with '%s'", tapfd_name);
7699
        qemuDomainObjExitMonitorWithDriver(driver, vm);
7700
    }
7701

7702 7703
    goto cleanup;

7704
no_memory:
7705
    virReportOOMError();
7706
    goto cleanup;
M
Mark McLoughlin 已提交
7707 7708
}

7709

7710
static int qemudDomainAttachHostPciDevice(struct qemud_driver *driver,
7711
                                          virDomainObjPtr vm,
7712
                                          virDomainHostdevDefPtr hostdev,
7713
                                          unsigned long long qemuCmdFlags)
7714
{
7715
    qemuDomainObjPrivatePtr priv = vm->privateData;
7716
    pciDevice *pci;
7717
    int ret;
7718
    char *devstr = NULL;
7719 7720
    int configfd = -1;
    char *configfd_name = NULL;
7721 7722

    if (VIR_REALLOC_N(vm->def->hostdevs, vm->def->nhostdevs+1) < 0) {
7723
        virReportOOMError();
7724 7725 7726
        return -1;
    }

7727
    pci = pciGetDevice(hostdev->source.subsys.u.pci.domain,
7728 7729 7730
                       hostdev->source.subsys.u.pci.bus,
                       hostdev->source.subsys.u.pci.slot,
                       hostdev->source.subsys.u.pci.function);
7731
    if (!pci)
7732
        return -1;
7733

7734 7735 7736 7737
    if (!pciDeviceIsAssignable(pci, !driver->relaxedACS) ||
        (hostdev->managed && pciDettachDevice(pci) < 0) ||
        pciResetDevice(pci, driver->activePciHostdevs) < 0) {
        pciFreeDevice(pci);
7738 7739 7740
        return -1;
    }

7741 7742
    if (pciDeviceListAdd(driver->activePciHostdevs, pci) < 0) {
        pciFreeDevice(pci);
7743
        return -1;
7744
    }
7745
    pci = NULL; /* activePciHostdevs owns the 'pci' reference now */
7746

7747
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
7748 7749
        if (qemuAssignDeviceHostdevAlias(vm->def, hostdev, -1) < 0)
            goto error;
7750 7751
        if (qemuDomainPCIAddressEnsureAddr(priv->pciaddrs, &hostdev->info) < 0)
            goto error;
7752 7753 7754 7755 7756 7757 7758 7759 7760 7761 7762 7763 7764 7765 7766 7767 7768 7769 7770 7771 7772 7773 7774 7775
        if (qemuCmdFlags & QEMUD_CMD_FLAG_PCI_CONFIGFD) {
            configfd = qemudOpenPCIConfig(hostdev);
            if (configfd >= 0) {
                if (virAsprintf(&configfd_name, "fd-%s",
                                hostdev->info.alias) < 0) {
                    virReportOOMError();
                    goto error;
                }

                qemuDomainObjEnterMonitorWithDriver(driver, vm);
                if (qemuMonitorSendFileHandle(priv->mon, configfd_name,
                                              configfd) < 0) {
                    qemuDomainObjExitMonitorWithDriver(driver, vm);
                    goto error;
                }
                qemuDomainObjExitMonitorWithDriver(driver, vm);
            }
        }

        if (!virDomainObjIsActive(vm)) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("guest unexpectedly quit during hotplug"));
            goto error;
        }
7776

7777
        if (!(devstr = qemuBuildPCIHostdevDevStr(hostdev, configfd_name)))
7778
            goto error;
7779

7780
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
7781
        ret = qemuMonitorAddDevice(priv->mon, devstr);
7782 7783 7784 7785 7786
        qemuDomainObjExitMonitorWithDriver(driver, vm);
    } else {
        virDomainDevicePCIAddress guestAddr;

        qemuDomainObjEnterMonitorWithDriver(driver, vm);
7787 7788 7789
        ret = qemuMonitorAddPCIHostDevice(priv->mon,
                                          &hostdev->source.subsys.u.pci,
                                          &guestAddr);
7790 7791 7792 7793 7794
        qemuDomainObjExitMonitorWithDriver(driver, vm);

        hostdev->info.type = VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI;
        memcpy(&hostdev->info.addr.pci, &guestAddr, sizeof(guestAddr));
    }
7795
    if (ret < 0)
7796
        goto error;
7797 7798 7799

    vm->def->hostdevs[vm->def->nhostdevs++] = hostdev;

7800
    VIR_FREE(devstr);
7801 7802 7803
    VIR_FREE(configfd_name);
    if (configfd >= 0)
        close(configfd);
7804

7805
    return 0;
7806 7807

error:
7808 7809 7810 7811 7812
    if ((qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) &&
        (hostdev->info.type == VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI) &&
        qemuDomainPCIAddressReleaseAddr(priv->pciaddrs, &hostdev->info) < 0)
        VIR_WARN0("Unable to release PCI address on host device");

7813 7814 7815 7816 7817
    pci = pciGetDevice(hostdev->source.subsys.u.pci.domain,
                       hostdev->source.subsys.u.pci.bus,
                       hostdev->source.subsys.u.pci.slot,
                       hostdev->source.subsys.u.pci.function);

7818
    pciDeviceListDel(driver->activePciHostdevs, pci);
7819 7820 7821 7822 7823 7824 7825 7826 7827 7828

    if (pciResetDevice(pci, driver->activePciHostdevs) < 0)
        VIR_WARN0("Unable to reset PCI device after assign failure");
    else if (hostdev->managed &&
             pciReAttachDevice(pci) < 0)
        VIR_WARN0("Unable to re-attach PCI device after assign failure");
    pciFreeDevice(pci);


    VIR_FREE(devstr);
7829 7830 7831
    VIR_FREE(configfd_name);
    if (configfd >= 0)
        close(configfd);
7832 7833

    return -1;
7834 7835
}

7836

7837
static int qemudDomainAttachHostUsbDevice(struct qemud_driver *driver,
M
Mark McLoughlin 已提交
7838
                                          virDomainObjPtr vm,
7839
                                          virDomainHostdevDefPtr hostdev,
7840
                                          unsigned long long qemuCmdFlags)
7841 7842
{
    int ret;
7843
    qemuDomainObjPrivatePtr priv = vm->privateData;
7844 7845
    char *devstr = NULL;

7846 7847 7848 7849 7850 7851
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        if (qemuAssignDeviceHostdevAlias(vm->def, hostdev, -1) < 0)
            goto error;
        if (!(devstr = qemuBuildUSBHostdevDevStr(hostdev)))
            goto error;
    }
7852

7853
    if (VIR_REALLOC_N(vm->def->hostdevs, vm->def->nhostdevs+1) < 0) {
7854
        virReportOOMError();
7855
        goto error;
7856
    }
7857

7858
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
7859 7860 7861
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE)
        ret = qemuMonitorAddDevice(priv->mon, devstr);
    else
7862
        ret = qemuMonitorAddUSBDeviceExact(priv->mon,
7863 7864
                                           hostdev->source.subsys.u.usb.bus,
                                           hostdev->source.subsys.u.usb.device);
7865
    qemuDomainObjExitMonitorWithDriver(driver, vm);
7866 7867 7868 7869
    if (ret < 0)
        goto error;

    vm->def->hostdevs[vm->def->nhostdevs++] = hostdev;
7870

7871
    VIR_FREE(devstr);
7872

7873 7874 7875 7876 7877
    return 0;

error:
    VIR_FREE(devstr);
    return -1;
7878 7879
}

7880

7881
static int qemudDomainAttachHostDevice(struct qemud_driver *driver,
M
Mark McLoughlin 已提交
7882
                                       virDomainObjPtr vm,
7883
                                       virDomainHostdevDefPtr hostdev,
7884
                                       unsigned long long qemuCmdFlags)
M
Mark McLoughlin 已提交
7885 7886
{
    if (hostdev->mode != VIR_DOMAIN_HOSTDEV_MODE_SUBSYS) {
7887 7888 7889
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        _("hostdev mode '%s' not supported"),
                        virDomainHostdevModeTypeToString(hostdev->mode));
M
Mark McLoughlin 已提交
7890 7891 7892
        return -1;
    }

7893 7894 7895 7896 7897 7898 7899 7900 7901 7902 7903 7904 7905 7906 7907 7908 7909
    /* Resolve USB product/vendor to bus/device */
    if (hostdev->source.subsys.type == VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB &&
        hostdev->source.subsys.u.usb.vendor) {
        usbDevice *usb
            = usbFindDevice(hostdev->source.subsys.u.usb.vendor,
                            hostdev->source.subsys.u.usb.product);

        if (!usb)
            return -1;

        hostdev->source.subsys.u.usb.bus = usbDeviceGetBus(usb);
        hostdev->source.subsys.u.usb.device = usbDeviceGetDevno(usb);

        usbFreeDevice(usb);
    }


7910
    if (driver->securityDriver &&
7911
        driver->securityDriver->domainSetSecurityHostdevLabel &&
7912
        driver->securityDriver->domainSetSecurityHostdevLabel(vm, hostdev) < 0)
7913
        return -1;
M
Mark McLoughlin 已提交
7914 7915

    switch (hostdev->source.subsys.type) {
7916
    case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_PCI:
7917
        if (qemudDomainAttachHostPciDevice(driver, vm,
7918
                                           hostdev, qemuCmdFlags) < 0)
7919 7920 7921
            goto error;
        break;

M
Mark McLoughlin 已提交
7922
    case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB:
7923
        if (qemudDomainAttachHostUsbDevice(driver, vm,
7924
                                           hostdev, qemuCmdFlags) < 0)
7925 7926 7927
            goto error;
        break;

M
Mark McLoughlin 已提交
7928
    default:
7929 7930 7931
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        _("hostdev subsys type '%s' not supported"),
                        virDomainHostdevSubsysTypeToString(hostdev->source.subsys.type));
7932
        goto error;
M
Mark McLoughlin 已提交
7933
    }
7934 7935 7936 7937 7938 7939

    return 0;

error:
    if (driver->securityDriver &&
        driver->securityDriver->domainRestoreSecurityHostdevLabel &&
7940
        driver->securityDriver->domainRestoreSecurityHostdevLabel(vm, hostdev) < 0)
7941 7942 7943
        VIR_WARN0("Unable to restore host device labelling on hotplug fail");

    return -1;
M
Mark McLoughlin 已提交
7944 7945
}

7946

7947
static int qemudDomainAttachDevice(virDomainPtr dom,
7948 7949
                                   const char *xml)
{
7950 7951 7952
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    virDomainDeviceDefPtr dev = NULL;
7953
    unsigned long long qemuCmdFlags;
7954
    virCgroupPtr cgroup = NULL;
7955
    int ret = -1;
7956

7957
    qemuDriverLock(driver);
7958
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
7959
    if (!vm) {
7960 7961
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
7962 7963
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
7964
        goto cleanup;
7965 7966
    }

7967 7968 7969
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
7970
    if (!virDomainObjIsActive(vm)) {
7971 7972
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot attach device on inactive domain"));
7973
        goto endjob;
7974 7975
    }

7976
    dev = virDomainDeviceDefParse(driver->caps, vm->def, xml,
7977
                                  VIR_DOMAIN_XML_INACTIVE);
7978
    if (dev == NULL)
7979
        goto endjob;
7980

7981 7982 7983
    if (qemudExtractVersionInfo(vm->def->emulator,
                                NULL,
                                &qemuCmdFlags) < 0)
7984
        goto endjob;
7985

7986
    if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
7987
        if (qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_DEVICES)) {
7988
            if (virCgroupForDomain(driver->cgroup, vm->def->name, &cgroup, 0) !=0 ) {
7989 7990 7991
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                _("Unable to find cgroup for %s\n"),
                                vm->def->name);
7992
                goto endjob;
7993
            }
7994
            if (qemuSetupDiskCgroup(cgroup, vm, dev->data.disk) < 0)
7995
                goto endjob;
7996 7997
        }

7998
        switch (dev->data.disk->device) {
7999 8000
        case VIR_DOMAIN_DISK_DEVICE_CDROM:
        case VIR_DOMAIN_DISK_DEVICE_FLOPPY:
8001 8002 8003
            ret = qemudDomainChangeEjectableMedia(driver, vm,
                                                  dev->data.disk,
                                                  qemuCmdFlags);
8004 8005
            if (ret == 0)
                dev->data.disk = NULL;
8006
            break;
8007

8008 8009
        case VIR_DOMAIN_DISK_DEVICE_DISK:
            if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_USB) {
8010
                ret = qemudDomainAttachUsbMassstorageDevice(driver, vm,
8011
                                                            dev->data.disk, qemuCmdFlags);
8012 8013
                if (ret == 0)
                    dev->data.disk = NULL;
8014
            } else if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_VIRTIO) {
8015
                ret = qemudDomainAttachPciDiskDevice(driver, vm,
8016
                                                     dev->data.disk, qemuCmdFlags);
8017 8018
                if (ret == 0)
                    dev->data.disk = NULL;
8019
            } else if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_SCSI) {
8020
                ret = qemudDomainAttachSCSIDisk(driver, vm,
8021
                                                dev->data.disk, qemuCmdFlags);
8022 8023
                if (ret == 0)
                    dev->data.disk = NULL;
8024
            } else {
8025 8026 8027
                qemuReportError(VIR_ERR_NO_SUPPORT,
                                _("disk bus '%s' cannot be hotplugged."),
                                virDomainDiskBusTypeToString(dev->data.disk->bus));
8028
                /* fallthrough */
8029 8030
            }
            break;
8031

8032
        default:
8033 8034 8035
            qemuReportError(VIR_ERR_NO_SUPPORT,
                            _("disk device type '%s' cannot be hotplugged"),
                            virDomainDiskDeviceTypeToString(dev->data.disk->device));
8036 8037
            /* Fallthrough */
        }
8038
        if (ret != 0 && cgroup) {
8039 8040 8041
            if (qemuTeardownDiskCgroup(cgroup, vm, dev->data.disk) < 0)
                VIR_WARN("Failed to teardown cgroup for disk path %s",
                         NULLSTR(dev->data.disk->src));
8042
        }
8043 8044
    } else if (dev->type == VIR_DOMAIN_DEVICE_CONTROLLER) {
        if (dev->data.controller->type == VIR_DOMAIN_CONTROLLER_TYPE_SCSI) {
8045
            ret = qemudDomainAttachPciControllerDevice(driver, vm,
8046
                                                       dev->data.controller, qemuCmdFlags);
8047 8048
            if (ret == 0)
                dev->data.controller = NULL;
8049
        } else {
8050 8051 8052
            qemuReportError(VIR_ERR_NO_SUPPORT,
                            _("disk controller bus '%s' cannot be hotplugged."),
                            virDomainControllerTypeToString(dev->data.controller->type));
8053 8054
            /* fallthrough */
        }
M
Mark McLoughlin 已提交
8055
    } else if (dev->type == VIR_DOMAIN_DEVICE_NET) {
8056 8057
        ret = qemudDomainAttachNetDevice(dom->conn, driver, vm,
                                         dev->data.net, qemuCmdFlags);
8058 8059
        if (ret == 0)
            dev->data.net = NULL;
M
Mark McLoughlin 已提交
8060
    } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV) {
8061
        ret = qemudDomainAttachHostDevice(driver, vm,
8062
                                          dev->data.hostdev, qemuCmdFlags);
8063 8064
        if (ret == 0)
            dev->data.hostdev = NULL;
8065
    } else {
8066 8067 8068
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        _("device type '%s' cannot be attached"),
                        virDomainDeviceTypeToString(dev->type));
8069
        goto endjob;
8070 8071
    }

8072
    if (!ret && virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
8073 8074
        ret = -1;

8075
endjob:
8076 8077
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
8078

8079
cleanup:
8080 8081 8082
    if (cgroup)
        virCgroupFree(&cgroup);

8083
    virDomainDeviceDefFree(dev);
8084 8085
    if (vm)
        virDomainObjUnlock(vm);
8086
    qemuDriverUnlock(driver);
8087 8088 8089
    return ret;
}

8090 8091 8092 8093
static int qemudDomainAttachDeviceFlags(virDomainPtr dom,
                                        const char *xml,
                                        unsigned int flags) {
    if (flags & VIR_DOMAIN_DEVICE_MODIFY_CONFIG) {
8094 8095
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot modify the persistent configuration of a domain"));
8096 8097 8098 8099 8100 8101
        return -1;
    }

    return qemudDomainAttachDevice(dom, xml);
}

8102

8103 8104 8105 8106 8107 8108 8109 8110 8111 8112 8113 8114 8115 8116 8117 8118 8119 8120 8121 8122 8123 8124 8125 8126 8127 8128 8129 8130 8131 8132 8133 8134 8135 8136 8137 8138 8139 8140 8141 8142 8143 8144 8145 8146 8147 8148 8149 8150 8151 8152 8153 8154 8155 8156 8157 8158 8159 8160 8161 8162 8163 8164 8165 8166 8167 8168 8169 8170 8171 8172 8173 8174 8175 8176 8177 8178 8179
static virDomainGraphicsDefPtr qemuDomainFindGraphics(virDomainObjPtr vm,
                                                      virDomainGraphicsDefPtr dev)
{
    int i;

    for (i = 0 ; i < vm->def->ngraphics ; i++) {
        if (vm->def->graphics[i]->type == dev->type)
            return vm->def->graphics[i];
    }

    return NULL;
}


static int
qemuDomainChangeGraphics(struct qemud_driver *driver,
                         virDomainObjPtr vm,
                         virDomainGraphicsDefPtr dev)
{
    virDomainGraphicsDefPtr olddev = qemuDomainFindGraphics(vm, dev);
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret = -1;

    if (!olddev) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                        _("cannot find existing graphics device to modify"));
        return -1;
    }

    switch (dev->type) {
    case VIR_DOMAIN_GRAPHICS_TYPE_VNC:
        if ((olddev->data.vnc.autoport != dev->data.vnc.autoport) ||
            (!dev->data.vnc.autoport && (olddev->data.vnc.port != dev->data.vnc.port))) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("cannot change port settings on vnc graphics"));
            return -1;
        }
        if (STRNEQ_NULLABLE(olddev->data.vnc.listenAddr, dev->data.vnc.listenAddr)) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("cannot change listen address setting on vnc graphics"));
            return -1;
        }
        if (STRNEQ_NULLABLE(olddev->data.vnc.keymap, dev->data.vnc.keymap)) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("cannot change keymap setting on vnc graphics"));
            return -1;
        }

        if (STRNEQ_NULLABLE(olddev->data.vnc.passwd, dev->data.vnc.passwd)) {
            VIR_DEBUG("Updating password on VNC server %p %p", dev->data.vnc.passwd, driver->vncPassword);
            qemuDomainObjEnterMonitorWithDriver(driver, vm);
            ret = qemuMonitorSetVNCPassword(priv->mon,
                                            dev->data.vnc.passwd ?
                                            dev->data.vnc.passwd :
                                            driver->vncPassword);
            qemuDomainObjExitMonitorWithDriver(driver, vm);

            /* Steal the new dev's  char * reference */
            VIR_FREE(olddev->data.vnc.passwd);
            olddev->data.vnc.passwd = dev->data.vnc.passwd;
            dev->data.vnc.passwd = NULL;
        } else {
            ret = 0;
        }
        break;

    default:
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("unable to change config on '%s' graphics type"),
                        virDomainGraphicsTypeToString(dev->type));
        break;
    }

    return ret;
}


8180 8181 8182 8183 8184 8185 8186 8187 8188 8189 8190
static int qemuDomainUpdateDeviceFlags(virDomainPtr dom,
                                       const char *xml,
                                       unsigned int flags)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    virDomainDeviceDefPtr dev = NULL;
    unsigned long long qemuCmdFlags;
    virCgroupPtr cgroup = NULL;
    int ret = -1;

8191 8192 8193 8194
    virCheckFlags(VIR_DOMAIN_DEVICE_MODIFY_CURRENT |
                  VIR_DOMAIN_DEVICE_MODIFY_LIVE |
                  VIR_DOMAIN_DEVICE_MODIFY_CONFIG, -1);

8195 8196 8197 8198 8199 8200 8201 8202 8203 8204 8205 8206 8207 8208 8209 8210 8211 8212 8213 8214 8215 8216 8217 8218 8219 8220 8221 8222 8223 8224 8225 8226 8227 8228 8229 8230 8231 8232 8233 8234 8235 8236 8237 8238
    if (flags & VIR_DOMAIN_DEVICE_MODIFY_CONFIG) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot modify the persistent configuration of a domain"));
        return -1;
    }

    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot attach device on inactive domain"));
        goto endjob;
    }

    dev = virDomainDeviceDefParse(driver->caps, vm->def, xml,
                                  VIR_DOMAIN_XML_INACTIVE);
    if (dev == NULL)
        goto endjob;

    if (qemudExtractVersionInfo(vm->def->emulator,
                                NULL,
                                &qemuCmdFlags) < 0)
        goto endjob;

    switch (dev->type) {
    case VIR_DOMAIN_DEVICE_DISK:
        if (qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_DEVICES)) {
            if (virCgroupForDomain(driver->cgroup, vm->def->name, &cgroup, 0) !=0 ) {
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                _("Unable to find cgroup for %s\n"),
                                vm->def->name);
                goto endjob;
            }
8239
            if (qemuSetupDiskCgroup(cgroup, vm, dev->data.disk) < 0)
8240 8241 8242 8243 8244 8245
                goto endjob;
        }

        switch (dev->data.disk->device) {
        case VIR_DOMAIN_DISK_DEVICE_CDROM:
        case VIR_DOMAIN_DISK_DEVICE_FLOPPY:
8246 8247 8248
            ret = qemudDomainChangeEjectableMedia(driver, vm,
                                                  dev->data.disk,
                                                  qemuCmdFlags);
8249 8250 8251 8252 8253 8254 8255 8256 8257 8258 8259 8260 8261
            if (ret == 0)
                dev->data.disk = NULL;
            break;


        default:
            qemuReportError(VIR_ERR_NO_SUPPORT,
                            _("disk bus '%s' cannot be updated."),
                            virDomainDiskBusTypeToString(dev->data.disk->bus));
            break;
        }

        if (ret != 0 && cgroup) {
8262 8263 8264
            if (qemuTeardownDiskCgroup(cgroup, vm, dev->data.disk) < 0)
                VIR_WARN("Failed to teardown cgroup for disk path %s",
                         NULLSTR(dev->data.disk->src));
8265 8266 8267
        }
        break;

8268 8269 8270 8271
    case VIR_DOMAIN_DEVICE_GRAPHICS:
        ret = qemuDomainChangeGraphics(driver, vm, dev->data.graphics);
        break;

8272 8273 8274 8275 8276 8277 8278 8279 8280 8281 8282 8283 8284 8285 8286 8287 8288 8289 8290 8291 8292 8293 8294 8295 8296 8297
    default:
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        _("disk device type '%s' cannot be updated"),
                        virDomainDiskDeviceTypeToString(dev->data.disk->device));
        break;
    }

    if (!ret && virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
        ret = -1;

endjob:
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;

cleanup:
    if (cgroup)
        virCgroupFree(&cgroup);

    virDomainDeviceDefFree(dev);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}


E
Eric Blake 已提交
8298
static inline int qemudFindDisk(virDomainDefPtr def, const char *dst)
W
Wolfgang Mauerer 已提交
8299 8300 8301 8302 8303 8304 8305 8306 8307 8308 8309 8310
{
    int i;

    for (i = 0 ; i < def->ndisks ; i++) {
        if (STREQ(def->disks[i]->dst, dst)) {
            return i;
        }
    }

    return -1;
}

E
Eric Blake 已提交
8311
static inline void qemudShrinkDisks(virDomainDefPtr def, size_t i)
W
Wolfgang Mauerer 已提交
8312 8313 8314 8315 8316 8317 8318 8319 8320 8321 8322 8323 8324 8325 8326 8327
{
    if (def->ndisks > 1) {
        memmove(def->disks + i,
                def->disks + i + 1,
                sizeof(*def->disks) *
                (def->ndisks - (i + 1)));
        def->ndisks--;
        if (VIR_REALLOC_N(def->disks, def->ndisks) < 0) {
            /* ignore, harmless */
        }
    } else {
        VIR_FREE(def->disks);
        def->ndisks = 0;
    }
}

8328
static int qemudDomainDetachPciDiskDevice(struct qemud_driver *driver,
8329
                                          virDomainObjPtr vm,
8330 8331
                                          virDomainDeviceDefPtr dev,
                                          unsigned long long qemuCmdFlags)
8332 8333 8334
{
    int i, ret = -1;
    virDomainDiskDefPtr detach = NULL;
8335
    qemuDomainObjPrivatePtr priv = vm->privateData;
8336
    virCgroupPtr cgroup = NULL;
8337

W
Wolfgang Mauerer 已提交
8338
    i = qemudFindDisk(vm->def, dev->data.disk->dst);
8339

W
Wolfgang Mauerer 已提交
8340
    if (i < 0) {
8341 8342
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("disk %s not found"), dev->data.disk->dst);
8343
        goto cleanup;
8344 8345
    }

W
Wolfgang Mauerer 已提交
8346 8347
    detach = vm->def->disks[i];

8348 8349 8350 8351 8352 8353 8354 8355 8356
    if (qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_DEVICES)) {
        if (virCgroupForDomain(driver->cgroup, vm->def->name, &cgroup, 0) != 0) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("Unable to find cgroup for %s\n"),
                            vm->def->name);
            goto cleanup;
        }
    }

8357 8358
    if (!virDomainDeviceAddressIsValid(&detach->info,
                                       VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI)) {
8359 8360
        qemuReportError(VIR_ERR_OPERATION_FAILED, "%s",
                        _("device cannot be detached without a PCI address"));
8361
        goto cleanup;
8362 8363
    }

8364
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
8365 8366 8367 8368 8369 8370 8371 8372 8373 8374 8375
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        if (qemuMonitorDelDevice(priv->mon, detach->info.alias) < 0) {
            qemuDomainObjExitMonitor(vm);
            goto cleanup;
        }
    } else {
        if (qemuMonitorRemovePCIDevice(priv->mon,
                                       &detach->info.addr.pci) < 0) {
            qemuDomainObjExitMonitor(vm);
            goto cleanup;
        }
8376
    }
8377
    qemuDomainObjExitMonitorWithDriver(driver, vm);
8378

W
Wolfgang Mauerer 已提交
8379 8380
    qemudShrinkDisks(vm->def, i);

8381 8382
    virDomainDiskDefFree(detach);

8383 8384 8385 8386 8387
    if (driver->securityDriver &&
        driver->securityDriver->domainRestoreSecurityImageLabel &&
        driver->securityDriver->domainRestoreSecurityImageLabel(vm, dev->data.disk) < 0)
        VIR_WARN("Unable to restore security label on %s", dev->data.disk->src);

8388 8389 8390 8391 8392 8393
    if (cgroup != NULL) {
        if (qemuTeardownDiskCgroup(cgroup, vm, dev->data.disk) < 0)
            VIR_WARN("Failed to teardown cgroup for disk path %s",
                     NULLSTR(dev->data.disk->src));
    }

8394 8395 8396 8397 8398 8399 8400 8401 8402 8403 8404 8405 8406 8407
    ret = 0;

cleanup:
    return ret;
}

static int qemudDomainDetachSCSIDiskDevice(struct qemud_driver *driver,
                                           virDomainObjPtr vm,
                                           virDomainDeviceDefPtr dev,
                                           unsigned long long qemuCmdFlags)
{
    int i, ret = -1;
    virDomainDiskDefPtr detach = NULL;
    qemuDomainObjPrivatePtr priv = vm->privateData;
8408
    virCgroupPtr cgroup = NULL;
8409 8410 8411 8412 8413 8414 8415 8416 8417 8418

    i = qemudFindDisk(vm->def, dev->data.disk->dst);

    if (i < 0) {
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("disk %s not found"), dev->data.disk->dst);
        goto cleanup;
    }

    if (!(qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE)) {
8419
        qemuReportError(VIR_ERR_OPERATION_FAILED, "%s",
8420 8421 8422 8423 8424 8425
                        _("Underlying qemu does not support SCSI disk removal"));
        goto cleanup;
    }

    detach = vm->def->disks[i];

8426 8427 8428 8429 8430 8431 8432 8433 8434
    if (qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_DEVICES)) {
        if (virCgroupForDomain(driver->cgroup, vm->def->name, &cgroup, 0) != 0) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("Unable to find cgroup for %s\n"),
                            vm->def->name);
            goto cleanup;
        }
    }

8435 8436 8437 8438 8439 8440 8441 8442 8443
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
    if (qemuMonitorDelDevice(priv->mon, detach->info.alias) < 0) {
        qemuDomainObjExitMonitor(vm);
        goto cleanup;
    }
    qemuDomainObjExitMonitorWithDriver(driver, vm);

    qemudShrinkDisks(vm->def, i);

8444
    virDomainDiskDefFree(detach);
8445

8446 8447
    if (driver->securityDriver &&
        driver->securityDriver->domainRestoreSecurityImageLabel &&
8448
        driver->securityDriver->domainRestoreSecurityImageLabel(vm, dev->data.disk) < 0)
8449 8450
        VIR_WARN("Unable to restore security label on %s", dev->data.disk->src);

8451 8452 8453 8454 8455 8456
    if (cgroup != NULL) {
        if (qemuTeardownDiskCgroup(cgroup, vm, dev->data.disk) < 0)
            VIR_WARN("Failed to teardown cgroup for disk path %s",
                     NULLSTR(dev->data.disk->src));
    }

8457
    ret = 0;
8458 8459

cleanup:
8460
    virCgroupFree(&cgroup);
8461 8462 8463
    return ret;
}

8464
static int qemudDomainDetachPciControllerDevice(struct qemud_driver *driver,
8465
                                                virDomainObjPtr vm,
8466 8467
                                                virDomainDeviceDefPtr dev,
                                                unsigned long long qemuCmdFlags)
8468 8469 8470 8471 8472 8473 8474 8475 8476 8477 8478 8479 8480 8481
{
    int i, ret = -1;
    virDomainControllerDefPtr detach = NULL;
    qemuDomainObjPrivatePtr priv = vm->privateData;

    for (i = 0 ; i < vm->def->ncontrollers ; i++) {
        if ((vm->def->controllers[i]->type == dev->data.controller->type) &&
            (vm->def->controllers[i]->idx == dev->data.controller->idx)) {
            detach = vm->def->controllers[i];
            break;
        }
    }

    if (!detach) {
8482 8483 8484 8485
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("disk controller %s:%d not found"),
                        virDomainControllerTypeToString(dev->data.controller->type),
                        dev->data.controller->idx);
8486 8487 8488 8489 8490
        goto cleanup;
    }

    if (!virDomainDeviceAddressIsValid(&detach->info,
                                       VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI)) {
8491 8492
        qemuReportError(VIR_ERR_OPERATION_FAILED, "%s",
                        _("device cannot be detached without a PCI address"));
8493 8494 8495
        goto cleanup;
    }

8496 8497 8498 8499 8500
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        if (qemuAssignDeviceControllerAlias(detach) < 0)
            goto cleanup;
    }

8501
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
8502 8503 8504 8505 8506 8507 8508 8509 8510 8511 8512
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        if (qemuMonitorDelDevice(priv->mon, detach->info.alias)) {
            qemuDomainObjExitMonitor(vm);
            goto cleanup;
        }
    } else {
        if (qemuMonitorRemovePCIDevice(priv->mon,
                                       &detach->info.addr.pci) < 0) {
            qemuDomainObjExitMonitor(vm);
            goto cleanup;
        }
8513 8514 8515 8516 8517 8518 8519 8520 8521 8522 8523 8524 8525 8526 8527 8528
    }
    qemuDomainObjExitMonitorWithDriver(driver, vm);

    if (vm->def->ncontrollers > 1) {
        memmove(vm->def->controllers + i,
                vm->def->controllers + i + 1,
                sizeof(*vm->def->controllers) *
                (vm->def->ncontrollers - (i + 1)));
        vm->def->ncontrollers--;
        if (VIR_REALLOC_N(vm->def->controllers, vm->def->ncontrollers) < 0) {
            /* ignore, harmless */
        }
    } else {
        VIR_FREE(vm->def->controllers);
        vm->def->ncontrollers = 0;
    }
8529 8530 8531 8532 8533

    if (qemuDomainPCIAddressReleaseAddr(priv->pciaddrs, &detach->info) < 0) {
        VIR_WARN0("Unable to release PCI address on controller");
    }

8534 8535 8536 8537 8538 8539 8540 8541
    virDomainControllerDefFree(detach);

    ret = 0;

cleanup:
    return ret;
}

8542
static int
8543
qemudDomainDetachNetDevice(struct qemud_driver *driver,
8544
                           virDomainObjPtr vm,
8545 8546
                           virDomainDeviceDefPtr dev,
                           unsigned long long qemuCmdFlags)
8547 8548 8549
{
    int i, ret = -1;
    virDomainNetDefPtr detach = NULL;
8550
    qemuDomainObjPrivatePtr priv = vm->privateData;
8551 8552
    int vlan;
    char *hostnet_name = NULL;
8553 8554 8555 8556 8557 8558 8559 8560 8561 8562 8563

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

        if (!memcmp(net->mac, dev->data.net->mac,  sizeof(net->mac))) {
            detach = net;
            break;
        }
    }

    if (!detach) {
8564 8565 8566 8567 8568
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("network device %02x:%02x:%02x:%02x:%02x:%02x not found"),
                        dev->data.net->mac[0], dev->data.net->mac[1],
                        dev->data.net->mac[2], dev->data.net->mac[3],
                        dev->data.net->mac[4], dev->data.net->mac[5]);
8569 8570 8571
        goto cleanup;
    }

8572 8573
    if (!virDomainDeviceAddressIsValid(&detach->info,
                                       VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI)) {
8574 8575
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("device cannot be detached without a PCI address"));
8576 8577 8578
        goto cleanup;
    }

8579
    if ((vlan = qemuDomainNetVLAN(detach)) < 0) {
8580 8581
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("unable to determine original VLAN"));
8582 8583 8584 8585
        goto cleanup;
    }

    if (virAsprintf(&hostnet_name, "host%s", detach->info.alias) < 0) {
8586
        virReportOOMError();
8587 8588 8589
        goto cleanup;
    }

8590
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
8591 8592 8593 8594 8595 8596 8597 8598 8599 8600 8601
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        if (qemuMonitorDelDevice(priv->mon, detach->info.alias) < 0) {
            qemuDomainObjExitMonitor(vm);
            goto cleanup;
        }
    } else {
        if (qemuMonitorRemovePCIDevice(priv->mon,
                                       &detach->info.addr.pci) < 0) {
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            goto cleanup;
        }
8602
    }
8603

8604 8605 8606 8607 8608 8609 8610 8611 8612 8613 8614
    if ((qemuCmdFlags & QEMUD_CMD_FLAG_NETDEV) &&
        (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE)) {
        if (qemuMonitorRemoveNetdev(priv->mon, hostnet_name) < 0) {
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            goto cleanup;
        }
    } else {
        if (qemuMonitorRemoveHostNetwork(priv->mon, vlan, hostnet_name) < 0) {
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            goto cleanup;
        }
8615
    }
8616
    qemuDomainObjExitMonitorWithDriver(driver, vm);
8617

8618 8619
    virNWFilterTearNWFilter(detach);

8620
#if WITH_MACVTAP
8621
    if (detach->type == VIR_DOMAIN_NET_TYPE_DIRECT) {
8622
        delMacvtap(detach->ifname, detach->mac, detach->data.direct.linkdev,
8623
                   &detach->data.direct.virtPortProfile);
8624 8625
        VIR_FREE(detach->ifname);
    }
8626 8627
#endif

8628
    if ((driver->macFilter) && (detach->ifname != NULL)) {
8629
        if ((errno = networkDisallowMacOnPort(driver,
8630 8631
                                              detach->ifname,
                                              detach->mac))) {
8632
            virReportSystemError(errno,
8633 8634 8635 8636 8637
             _("failed to remove ebtables rule on  '%s'"),
                                 detach->ifname);
        }
    }

8638 8639 8640 8641 8642 8643 8644 8645 8646 8647
    if (vm->def->nnets > 1) {
        memmove(vm->def->nets + i,
                vm->def->nets + i + 1,
                sizeof(*vm->def->nets) *
                (vm->def->nnets - (i + 1)));
        vm->def->nnets--;
        if (VIR_REALLOC_N(vm->def->nets, vm->def->nnets) < 0) {
            /* ignore, harmless */
        }
    } else {
8648
        VIR_FREE(vm->def->nets);
8649
        vm->def->nnets = 0;
8650
    }
8651
    virDomainNetDefFree(detach);
8652

8653 8654 8655
    ret = 0;

cleanup:
8656
    VIR_FREE(hostnet_name);
8657 8658 8659
    return ret;
}

8660
static int qemudDomainDetachHostPciDevice(struct qemud_driver *driver,
8661
                                          virDomainObjPtr vm,
8662 8663
                                          virDomainDeviceDefPtr dev,
                                          unsigned long long qemuCmdFlags)
8664
{
8665
    virDomainHostdevDefPtr detach = NULL;
8666
    qemuDomainObjPrivatePtr priv = vm->privateData;
8667
    int i, ret;
8668
    pciDevice *pci;
8669 8670

    for (i = 0 ; i < vm->def->nhostdevs ; i++) {
8671 8672 8673 8674
        if (vm->def->hostdevs[i]->mode != VIR_DOMAIN_HOSTDEV_MODE_SUBSYS ||
            vm->def->hostdevs[i]->source.subsys.type != VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_PCI)
            continue;

8675 8676 8677 8678 8679 8680 8681 8682 8683 8684 8685 8686 8687 8688 8689
        unsigned domain   = vm->def->hostdevs[i]->source.subsys.u.pci.domain;
        unsigned bus      = vm->def->hostdevs[i]->source.subsys.u.pci.bus;
        unsigned slot     = vm->def->hostdevs[i]->source.subsys.u.pci.slot;
        unsigned function = vm->def->hostdevs[i]->source.subsys.u.pci.function;

        if (dev->data.hostdev->source.subsys.u.pci.domain   == domain &&
            dev->data.hostdev->source.subsys.u.pci.bus      == bus &&
            dev->data.hostdev->source.subsys.u.pci.slot     == slot &&
            dev->data.hostdev->source.subsys.u.pci.function == function) {
            detach = vm->def->hostdevs[i];
            break;
        }
    }

    if (!detach) {
8690 8691 8692 8693 8694 8695
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("host pci device %.4x:%.2x:%.2x.%.1x not found"),
                        dev->data.hostdev->source.subsys.u.pci.domain,
                        dev->data.hostdev->source.subsys.u.pci.bus,
                        dev->data.hostdev->source.subsys.u.pci.slot,
                        dev->data.hostdev->source.subsys.u.pci.function);
8696 8697 8698
        return -1;
    }

8699 8700
    if (!virDomainDeviceAddressIsValid(&detach->info,
                                       VIR_DOMAIN_DEVICE_ADDRESS_TYPE_PCI)) {
8701 8702
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("device cannot be detached without a PCI address"));
8703 8704 8705
        return -1;
    }

8706
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
8707 8708 8709 8710 8711 8712 8713 8714 8715 8716 8717
    if (qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE) {
        if (qemuMonitorDelDevice(priv->mon, detach->info.alias) < 0) {
            qemuDomainObjExitMonitor(vm);
            return -1;
        }
    } else {
        if (qemuMonitorRemovePCIDevice(priv->mon,
                                       &detach->info.addr.pci) < 0) {
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            return -1;
        }
8718
    }
8719
    qemuDomainObjExitMonitorWithDriver(driver, vm);
8720 8721 8722

    ret = 0;

8723
    pci = pciGetDevice(detach->source.subsys.u.pci.domain,
8724 8725 8726 8727 8728 8729
                       detach->source.subsys.u.pci.bus,
                       detach->source.subsys.u.pci.slot,
                       detach->source.subsys.u.pci.function);
    if (!pci)
        ret = -1;
    else {
8730
        pciDeviceSetManaged(pci, detach->managed);
8731 8732
        pciDeviceListDel(driver->activePciHostdevs, pci);
        if (pciResetDevice(pci, driver->activePciHostdevs) < 0)
8733
            ret = -1;
8734
        qemudReattachManagedDevice(pci);
8735
        pciFreeDevice(pci);
8736 8737
    }

8738 8739 8740 8741
    if (qemuDomainPCIAddressReleaseAddr(priv->pciaddrs, &detach->info) < 0) {
        VIR_WARN0("Unable to release PCI address on controller");
    }

8742 8743 8744 8745 8746 8747 8748 8749 8750 8751 8752 8753
    if (vm->def->nhostdevs > 1) {
        memmove(vm->def->hostdevs + i,
                vm->def->hostdevs + i + 1,
                sizeof(*vm->def->hostdevs) *
                (vm->def->nhostdevs - (i + 1)));
        vm->def->nhostdevs--;
        if (VIR_REALLOC_N(vm->def->hostdevs, vm->def->nhostdevs) < 0) {
            /* ignore, harmless */
        }
    } else {
        VIR_FREE(vm->def->hostdevs);
        vm->def->nhostdevs = 0;
8754
    }
8755
    virDomainHostdevDefFree(detach);
8756 8757 8758 8759

    return ret;
}

8760 8761 8762 8763 8764 8765 8766 8767 8768 8769 8770 8771 8772 8773 8774 8775
static int qemudDomainDetachHostUsbDevice(struct qemud_driver *driver,
                                          virDomainObjPtr vm,
                                          virDomainDeviceDefPtr dev,
                                          unsigned long long qemuCmdFlags)
{
    virDomainHostdevDefPtr detach = NULL;
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int i, ret;

    for (i = 0 ; i < vm->def->nhostdevs ; i++) {
        if (vm->def->hostdevs[i]->mode != VIR_DOMAIN_HOSTDEV_MODE_SUBSYS ||
            vm->def->hostdevs[i]->source.subsys.type != VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB)
            continue;

        unsigned bus = vm->def->hostdevs[i]->source.subsys.u.usb.bus;
        unsigned device = vm->def->hostdevs[i]->source.subsys.u.usb.device;
8776 8777 8778 8779 8780 8781 8782 8783 8784 8785 8786 8787 8788 8789 8790 8791
        unsigned product = vm->def->hostdevs[i]->source.subsys.u.usb.product;
        unsigned vendor = vm->def->hostdevs[i]->source.subsys.u.usb.vendor;

        if (dev->data.hostdev->source.subsys.u.usb.bus &&
            dev->data.hostdev->source.subsys.u.usb.device) {
            if (dev->data.hostdev->source.subsys.u.usb.bus == bus &&
                dev->data.hostdev->source.subsys.u.usb.device == device) {
                detach = vm->def->hostdevs[i];
                break;
            }
        } else {
            if (dev->data.hostdev->source.subsys.u.usb.product == product &&
                dev->data.hostdev->source.subsys.u.usb.vendor == vendor) {
                detach = vm->def->hostdevs[i];
                break;
            }
8792 8793 8794 8795 8796 8797 8798 8799 8800 8801 8802 8803 8804 8805 8806 8807 8808 8809 8810 8811 8812 8813 8814 8815 8816 8817 8818 8819 8820 8821 8822 8823 8824 8825 8826 8827 8828 8829 8830 8831 8832 8833 8834 8835 8836 8837 8838 8839 8840 8841
        }
    }

    if (!detach) {
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("host usb device %03d.%03d not found"),
                        dev->data.hostdev->source.subsys.u.usb.bus,
                        dev->data.hostdev->source.subsys.u.usb.device);
        return -1;
    }

    if (!detach->info.alias) {
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("device cannot be detached without a device alias"));
        return -1;
    }

    if (!(qemuCmdFlags & QEMUD_CMD_FLAG_DEVICE)) {
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("device cannot be detached with this QEMU version"));
        return -1;
    }

    qemuDomainObjEnterMonitorWithDriver(driver, vm);
    if (qemuMonitorDelDevice(priv->mon, detach->info.alias) < 0) {
        qemuDomainObjExitMonitorWithDriver(driver, vm);
        return -1;
    }
    qemuDomainObjExitMonitorWithDriver(driver, vm);

    ret = 0;

    if (vm->def->nhostdevs > 1) {
        memmove(vm->def->hostdevs + i,
                vm->def->hostdevs + i + 1,
                sizeof(*vm->def->hostdevs) *
                (vm->def->nhostdevs - (i + 1)));
        vm->def->nhostdevs--;
        if (VIR_REALLOC_N(vm->def->hostdevs, vm->def->nhostdevs) < 0) {
            /* ignore, harmless */
        }
    } else {
        VIR_FREE(vm->def->hostdevs);
        vm->def->nhostdevs = 0;
    }
    virDomainHostdevDefFree(detach);

    return ret;
}

8842
static int qemudDomainDetachHostDevice(struct qemud_driver *driver,
8843
                                       virDomainObjPtr vm,
8844 8845
                                       virDomainDeviceDefPtr dev,
                                       unsigned long long qemuCmdFlags)
8846 8847 8848 8849 8850
{
    virDomainHostdevDefPtr hostdev = dev->data.hostdev;
    int ret;

    if (hostdev->mode != VIR_DOMAIN_HOSTDEV_MODE_SUBSYS) {
8851 8852 8853
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        _("hostdev mode '%s' not supported"),
                        virDomainHostdevModeTypeToString(hostdev->mode));
8854 8855 8856 8857 8858
        return -1;
    }

    switch (hostdev->source.subsys.type) {
    case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_PCI:
8859
        ret = qemudDomainDetachHostPciDevice(driver, vm, dev, qemuCmdFlags);
8860
        break;
8861 8862 8863
    case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB:
        ret = qemudDomainDetachHostUsbDevice(driver, vm, dev, qemuCmdFlags);
        break;
8864
    default:
8865 8866 8867
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        _("hostdev subsys type '%s' not supported"),
                        virDomainHostdevSubsysTypeToString(hostdev->source.subsys.type));
8868 8869 8870
        return -1;
    }

8871
    if (driver->securityDriver &&
8872
        driver->securityDriver->domainRestoreSecurityHostdevLabel &&
8873
        driver->securityDriver->domainRestoreSecurityHostdevLabel(vm, dev->data.hostdev) < 0)
8874
        VIR_WARN0("Failed to restore host device labelling");
8875

8876 8877 8878
    return ret;
}

8879 8880
static int qemudDomainDetachDevice(virDomainPtr dom,
                                   const char *xml) {
8881 8882
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
8883
    unsigned long long qemuCmdFlags;
8884 8885
    virDomainDeviceDefPtr dev = NULL;
    int ret = -1;
8886

8887
    qemuDriverLock(driver);
8888
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
8889
    if (!vm) {
8890 8891
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
8892 8893
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
8894
        goto cleanup;
8895 8896
    }

8897 8898 8899
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
8900
    if (!virDomainObjIsActive(vm)) {
8901 8902
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot detach device on inactive domain"));
8903
        goto endjob;
8904 8905
    }

8906
    dev = virDomainDeviceDefParse(driver->caps, vm->def, xml,
8907
                                  VIR_DOMAIN_XML_INACTIVE);
8908
    if (dev == NULL)
8909
        goto endjob;
8910

8911 8912 8913 8914
    if (qemudExtractVersionInfo(vm->def->emulator,
                                NULL,
                                &qemuCmdFlags) < 0)
        goto endjob;
8915 8916

    if (dev->type == VIR_DOMAIN_DEVICE_DISK &&
8917 8918 8919 8920 8921 8922 8923 8924 8925 8926 8927 8928
        dev->data.disk->device == VIR_DOMAIN_DISK_DEVICE_DISK) {
        if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_VIRTIO) {
            ret = qemudDomainDetachPciDiskDevice(driver, vm, dev, qemuCmdFlags);
        }
        else if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_SCSI) {
            ret = qemudDomainDetachSCSIDiskDevice(driver, vm, dev,
                                                  qemuCmdFlags);
        }
        else {
            qemuReportError(VIR_ERR_NO_SUPPORT, "%s",
                            _("This type of disk cannot be hot unplugged"));
        }
8929
    } else if (dev->type == VIR_DOMAIN_DEVICE_NET) {
8930
        ret = qemudDomainDetachNetDevice(driver, vm, dev, qemuCmdFlags);
8931 8932
    } else if (dev->type == VIR_DOMAIN_DEVICE_CONTROLLER) {
        if (dev->data.controller->type == VIR_DOMAIN_CONTROLLER_TYPE_SCSI) {
8933 8934
            ret = qemudDomainDetachPciControllerDevice(driver, vm, dev,
                                                       qemuCmdFlags);
8935
        } else {
8936 8937 8938
            qemuReportError(VIR_ERR_NO_SUPPORT,
                            _("disk controller bus '%s' cannot be hotunplugged."),
                            virDomainControllerTypeToString(dev->data.controller->type));
8939 8940
            /* fallthrough */
        }
8941
    } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV) {
8942
        ret = qemudDomainDetachHostDevice(driver, vm, dev, qemuCmdFlags);
8943
    } else {
8944 8945
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        "%s", _("This type of device cannot be hot unplugged"));
8946
    }
8947

8948
    if (!ret && virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
8949 8950
        ret = -1;

8951
endjob:
8952 8953
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
8954

8955 8956
cleanup:
    virDomainDeviceDefFree(dev);
8957 8958
    if (vm)
        virDomainObjUnlock(vm);
8959
    qemuDriverUnlock(driver);
8960 8961 8962
    return ret;
}

8963 8964 8965 8966
static int qemudDomainDetachDeviceFlags(virDomainPtr dom,
                                        const char *xml,
                                        unsigned int flags) {
    if (flags & VIR_DOMAIN_DEVICE_MODIFY_CONFIG) {
8967 8968
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot modify the persistent configuration of a domain"));
8969 8970 8971 8972 8973 8974
        return -1;
    }

    return qemudDomainDetachDevice(dom, xml);
}

8975
static int qemudDomainGetAutostart(virDomainPtr dom,
8976
                                   int *autostart) {
8977 8978 8979
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
8980

8981
    qemuDriverLock(driver);
8982
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
8983 8984
    qemuDriverUnlock(driver);

8985
    if (!vm) {
8986 8987
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
8988 8989
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
8990
        goto cleanup;
8991 8992 8993
    }

    *autostart = vm->autostart;
8994
    ret = 0;
8995

8996
cleanup:
8997 8998
    if (vm)
        virDomainObjUnlock(vm);
8999
    return ret;
9000 9001
}

9002
static int qemudDomainSetAutostart(virDomainPtr dom,
9003
                                   int autostart) {
9004 9005
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
9006 9007
    char *configFile = NULL, *autostartLink = NULL;
    int ret = -1;
9008

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

9012
    if (!vm) {
9013 9014
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
9015 9016
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
9017
        goto cleanup;
9018 9019
    }

9020
    if (!vm->persistent) {
9021 9022
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("cannot set autostart for transient domain"));
9023
        goto cleanup;
9024 9025
    }

9026 9027
    autostart = (autostart != 0);

9028
    if (vm->autostart != autostart) {
9029
        if ((configFile = virDomainConfigFile(driver->configDir, vm->def->name)) == NULL)
9030
            goto cleanup;
9031
        if ((autostartLink = virDomainConfigFile(driver->autostartDir, vm->def->name)) == NULL)
9032
            goto cleanup;
9033

9034 9035
        if (autostart) {
            int err;
9036

9037
            if ((err = virFileMakePath(driver->autostartDir))) {
9038
                virReportSystemError(err,
9039 9040
                                     _("cannot create autostart directory %s"),
                                     driver->autostartDir);
9041 9042
                goto cleanup;
            }
9043

9044
            if (symlink(configFile, autostartLink) < 0) {
9045
                virReportSystemError(errno,
9046 9047
                                     _("Failed to create symlink '%s to '%s'"),
                                     autostartLink, configFile);
9048 9049 9050 9051
                goto cleanup;
            }
        } else {
            if (unlink(autostartLink) < 0 && errno != ENOENT && errno != ENOTDIR) {
9052
                virReportSystemError(errno,
9053 9054
                                     _("Failed to delete symlink '%s'"),
                                     autostartLink);
9055 9056
                goto cleanup;
            }
9057 9058
        }

9059
        vm->autostart = autostart;
9060
    }
9061
    ret = 0;
9062

9063 9064 9065
cleanup:
    VIR_FREE(configFile);
    VIR_FREE(autostartLink);
9066 9067
    if (vm)
        virDomainObjUnlock(vm);
9068
    qemuDriverUnlock(driver);
9069
    return ret;
9070 9071
}

9072 9073 9074 9075 9076

static char *qemuGetSchedulerType(virDomainPtr dom,
                                  int *nparams)
{
    struct qemud_driver *driver = dom->conn->privateData;
9077
    char *ret = NULL;
9078

9079
    qemuDriverLock(driver);
9080
    if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_CPU)) {
9081 9082
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        __FUNCTION__);
9083
        goto cleanup;
9084 9085 9086 9087 9088 9089 9090
    }

    if (nparams)
        *nparams = 1;

    ret = strdup("posix");
    if (!ret)
9091
        virReportOOMError();
9092 9093 9094

cleanup:
    qemuDriverUnlock(driver);
9095 9096 9097 9098 9099 9100 9101 9102 9103 9104 9105 9106 9107
    return ret;
}

static int qemuSetSchedulerParameters(virDomainPtr dom,
                                      virSchedParameterPtr params,
                                      int nparams)
{
    struct qemud_driver *driver = dom->conn->privateData;
    int i;
    virCgroupPtr group = NULL;
    virDomainObjPtr vm = NULL;
    int ret = -1;

9108
    qemuDriverLock(driver);
9109
    if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_CPU)) {
9110 9111
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        __FUNCTION__);
9112
        goto cleanup;
9113 9114 9115 9116 9117
    }

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

    if (vm == NULL) {
9118 9119
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("No such domain %s"), dom->uuid);
9120 9121 9122 9123
        goto cleanup;
    }

    if (virCgroupForDomain(driver->cgroup, vm->def->name, &group, 0) != 0) {
9124 9125
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("cannot find cgroup for domain %s"), vm->def->name);
9126 9127 9128 9129 9130 9131 9132 9133 9134
        goto cleanup;
    }

    for (i = 0; i < nparams; i++) {
        virSchedParameterPtr param = &params[i];

        if (STREQ(param->field, "cpu_shares")) {
            int rc;
            if (param->type != VIR_DOMAIN_SCHED_FIELD_ULLONG) {
9135 9136
                qemuReportError(VIR_ERR_INVALID_ARG, "%s",
                                _("invalid type for cpu_shares tunable, expected a 'ullong'"));
9137 9138 9139 9140 9141
                goto cleanup;
            }

            rc = virCgroupSetCpuShares(group, params[i].value.ul);
            if (rc != 0) {
9142
                virReportSystemError(-rc, "%s",
9143 9144 9145 9146
                                     _("unable to set cpu shares tunable"));
                goto cleanup;
            }
        } else {
9147 9148
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("Invalid parameter `%s'"), param->field);
9149 9150 9151 9152 9153 9154 9155 9156 9157
            goto cleanup;
        }
    }
    ret = 0;

cleanup:
    virCgroupFree(&group);
    if (vm)
        virDomainObjUnlock(vm);
9158
    qemuDriverUnlock(driver);
9159 9160 9161 9162 9163 9164 9165 9166 9167 9168 9169 9170 9171 9172
    return ret;
}

static int qemuGetSchedulerParameters(virDomainPtr dom,
                                      virSchedParameterPtr params,
                                      int *nparams)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virCgroupPtr group = NULL;
    virDomainObjPtr vm = NULL;
    unsigned long long val;
    int ret = -1;
    int rc;

9173
    qemuDriverLock(driver);
9174
    if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_CPU)) {
9175 9176
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        __FUNCTION__);
9177
        goto cleanup;
9178 9179 9180
    }

    if ((*nparams) != 1) {
9181 9182
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("Invalid parameter count"));
9183
        goto cleanup;
9184 9185 9186 9187 9188
    }

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

    if (vm == NULL) {
9189 9190
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("No such domain %s"), dom->uuid);
9191 9192 9193 9194
        goto cleanup;
    }

    if (virCgroupForDomain(driver->cgroup, vm->def->name, &group, 0) != 0) {
9195 9196
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("cannot find cgroup for domain %s"), vm->def->name);
9197 9198 9199 9200 9201
        goto cleanup;
    }

    rc = virCgroupGetCpuShares(group, &val);
    if (rc != 0) {
9202
        virReportSystemError(-rc, "%s",
9203 9204 9205 9206 9207
                             _("unable to get cpu shares tunable"));
        goto cleanup;
    }
    params[0].value.ul = val;
    params[0].type = VIR_DOMAIN_SCHED_FIELD_ULLONG;
C
Chris Lalancette 已提交
9208
    if (virStrcpyStatic(params[0].field, "cpu_shares") == NULL) {
9209 9210
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("Field cpu_shares too long for destination"));
C
Chris Lalancette 已提交
9211 9212
        goto cleanup;
    }
9213 9214 9215 9216 9217 9218 9219

    ret = 0;

cleanup:
    virCgroupFree(&group);
    if (vm)
        virDomainObjUnlock(vm);
9220
    qemuDriverUnlock(driver);
9221 9222 9223 9224
    return ret;
}


9225 9226 9227 9228 9229 9230 9231 9232 9233
/* 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)
{
9234
    struct qemud_driver *driver = dom->conn->privateData;
9235
    int i, ret = -1;
9236
    virDomainObjPtr vm;
9237
    virDomainDiskDefPtr disk = NULL;
9238

9239
    qemuDriverLock(driver);
9240
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
9241
    qemuDriverUnlock(driver);
9242
    if (!vm) {
9243 9244
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
9245 9246
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
9247
        goto cleanup;
9248
    }
9249 9250 9251 9252

    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
9253
    if (!virDomainObjIsActive (vm)) {
9254 9255
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
9256
        goto endjob;
9257 9258
    }

9259 9260 9261 9262 9263 9264 9265 9266
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (STREQ(path, vm->def->disks[i]->dst)) {
            disk = vm->def->disks[i];
            break;
        }
    }

    if (!disk) {
9267 9268
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("invalid path: %s"), path);
9269
        goto endjob;
9270 9271
    }

9272
    if (!disk->info.alias) {
9273 9274
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("missing disk device alias name for %s"), disk->dst);
9275
        goto endjob;
9276
    }
9277

9278
    qemuDomainObjPrivatePtr priv = vm->privateData;
9279 9280
    qemuDomainObjEnterMonitor(vm);
    ret = qemuMonitorGetBlockStatsInfo(priv->mon,
9281
                                       disk->info.alias,
9282 9283 9284 9285 9286 9287
                                       &stats->rd_req,
                                       &stats->rd_bytes,
                                       &stats->wr_req,
                                       &stats->wr_bytes,
                                       &stats->errs);
    qemuDomainObjExitMonitor(vm);
9288

9289
endjob:
9290 9291
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
9292

9293
cleanup:
9294 9295
    if (vm)
        virDomainObjUnlock(vm);
9296
    return ret;
9297 9298
}

9299
#ifdef __linux__
9300 9301 9302 9303 9304
static int
qemudDomainInterfaceStats (virDomainPtr dom,
                           const char *path,
                           struct _virDomainInterfaceStats *stats)
{
9305 9306
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
9307
    int i;
9308
    int ret = -1;
9309

9310
    qemuDriverLock(driver);
9311
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
9312 9313
    qemuDriverUnlock(driver);

9314
    if (!vm) {
9315 9316
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
9317 9318
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
9319
        goto cleanup;
9320 9321
    }

D
Daniel P. Berrange 已提交
9322
    if (!virDomainObjIsActive(vm)) {
9323 9324
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
9325
        goto cleanup;
9326 9327 9328
    }

    /* Check the path is one of the domain's network interfaces. */
9329 9330
    for (i = 0 ; i < vm->def->nnets ; i++) {
        if (vm->def->nets[i]->ifname &&
9331 9332 9333 9334
            STREQ (vm->def->nets[i]->ifname, path)) {
            ret = 0;
            break;
        }
9335 9336
    }

9337
    if (ret == 0)
9338
        ret = linuxDomainInterfaceStats(path, stats);
9339
    else
9340 9341
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("invalid path, '%s' is not a known interface"), path);
9342

9343
cleanup:
9344 9345
    if (vm)
        virDomainObjUnlock(vm);
9346 9347
    return ret;
}
9348
#else
9349 9350 9351 9352
static int
qemudDomainInterfaceStats (virDomainPtr dom,
                           const char *path ATTRIBUTE_UNUSED,
                           struct _virDomainInterfaceStats *stats ATTRIBUTE_UNUSED)
9353 9354
    qemuReportError(VIR_ERR_NO_SUPPORT,
                    "%s", __FUNCTION__);
9355 9356
    return -1;
}
9357
#endif
9358

9359 9360 9361 9362 9363 9364 9365 9366 9367 9368 9369 9370 9371 9372 9373 9374
static int
qemudDomainMemoryStats (virDomainPtr dom,
                        struct _virDomainMemoryStat *stats,
                        unsigned int nr_stats)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    unsigned int ret = -1;

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

    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
9375 9376
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
9377 9378 9379
        goto cleanup;
    }

9380 9381 9382
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

9383 9384 9385
    if (virDomainObjIsActive(vm)) {
        qemuDomainObjPrivatePtr priv = vm->privateData;
        qemuDomainObjEnterMonitor(vm);
9386
        ret = qemuMonitorGetMemoryStats(priv->mon, stats, nr_stats);
9387 9388
        qemuDomainObjExitMonitor(vm);
    } else {
9389 9390
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
9391 9392
    }

9393 9394 9395
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;

9396 9397 9398 9399 9400 9401
cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    return ret;
}

9402 9403 9404 9405 9406 9407 9408
static int
qemudDomainBlockPeek (virDomainPtr dom,
                      const char *path,
                      unsigned long long offset, size_t size,
                      void *buffer,
                      unsigned int flags ATTRIBUTE_UNUSED)
{
9409 9410 9411
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int fd = -1, ret = -1, i;
9412

9413
    qemuDriverLock(driver);
9414
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
9415 9416
    qemuDriverUnlock(driver);

9417
    if (!vm) {
9418 9419
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
9420 9421
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
9422
        goto cleanup;
9423 9424 9425
    }

    if (!path || path[0] == '\0') {
9426 9427
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("NULL or empty path"));
9428
        goto cleanup;
9429 9430 9431
    }

    /* Check the path belongs to this domain. */
9432 9433
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (vm->def->disks[i]->src != NULL &&
9434 9435 9436 9437
            STREQ (vm->def->disks[i]->src, path)) {
            ret = 0;
            break;
        }
9438 9439
    }

9440 9441 9442 9443 9444
    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) {
9445 9446
            virReportSystemError(errno,
                                 _("%s: failed to open"), path);
9447 9448
            goto cleanup;
        }
9449

9450 9451 9452 9453 9454 9455
        /* 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) {
9456 9457
            virReportSystemError(errno,
                                 _("%s: failed to seek or read"), path);
9458 9459 9460 9461 9462
            goto cleanup;
        }

        ret = 0;
    } else {
9463 9464
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("invalid path"));
9465 9466
    }

9467 9468 9469
cleanup:
    if (fd >= 0)
        close (fd);
9470 9471
    if (vm)
        virDomainObjUnlock(vm);
9472 9473 9474
    return ret;
}

R
Richard W.M. Jones 已提交
9475 9476 9477 9478 9479 9480
static int
qemudDomainMemoryPeek (virDomainPtr dom,
                       unsigned long long offset, size_t size,
                       void *buffer,
                       unsigned int flags)
{
9481 9482
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
9483
    char *tmp = NULL;
R
Richard W.M. Jones 已提交
9484 9485
    int fd = -1, ret = -1;

9486
    qemuDriverLock(driver);
9487
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
9488
    qemuDriverUnlock(driver);
R
Richard W.M. Jones 已提交
9489 9490

    if (!vm) {
9491 9492
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
9493 9494
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
9495 9496 9497
        goto cleanup;
    }

9498
    if (flags != VIR_MEMORY_VIRTUAL && flags != VIR_MEMORY_PHYSICAL) {
9499 9500
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("flags parameter must be VIR_MEMORY_VIRTUAL or VIR_MEMORY_PHYSICAL"));
9501
        goto cleanup;
R
Richard W.M. Jones 已提交
9502 9503
    }

9504 9505 9506
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
9507
    if (!virDomainObjIsActive(vm)) {
9508 9509
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
9510
        goto endjob;
R
Richard W.M. Jones 已提交
9511 9512
    }

9513
    if (virAsprintf(&tmp, "%s/qemu.mem.XXXXXX", driver->cacheDir) < 0) {
9514
        virReportOOMError();
9515
        goto endjob;
9516 9517
    }

R
Richard W.M. Jones 已提交
9518 9519
    /* Create a temporary filename. */
    if ((fd = mkstemp (tmp)) == -1) {
9520 9521
        virReportSystemError(errno,
                             _("mkstemp(\"%s\") failed"), tmp);
9522
        goto endjob;
R
Richard W.M. Jones 已提交
9523 9524
    }

9525
    qemuDomainObjPrivatePtr priv = vm->privateData;
9526
    qemuDomainObjEnterMonitor(vm);
9527
    if (flags == VIR_MEMORY_VIRTUAL) {
9528 9529
        if (qemuMonitorSaveVirtualMemory(priv->mon, offset, size, tmp) < 0) {
            qemuDomainObjExitMonitor(vm);
9530
            goto endjob;
9531
        }
9532
    } else {
9533 9534
        if (qemuMonitorSavePhysicalMemory(priv->mon, offset, size, tmp) < 0) {
            qemuDomainObjExitMonitor(vm);
9535
            goto endjob;
9536
        }
R
Richard W.M. Jones 已提交
9537
    }
9538
    qemuDomainObjExitMonitor(vm);
R
Richard W.M. Jones 已提交
9539 9540 9541

    /* Read the memory file into buffer. */
    if (saferead (fd, buffer, size) == (ssize_t) -1) {
9542 9543 9544
        virReportSystemError(errno,
                             _("failed to read temporary file "
                               "created with template %s"), tmp);
9545
        goto endjob;
R
Richard W.M. Jones 已提交
9546 9547 9548
    }

    ret = 0;
9549

9550
endjob:
9551 9552
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
9553

9554
cleanup:
9555
    VIR_FREE(tmp);
R
Richard W.M. Jones 已提交
9556 9557
    if (fd >= 0) close (fd);
    unlink (tmp);
9558 9559
    if (vm)
        virDomainObjUnlock(vm);
R
Richard W.M. Jones 已提交
9560 9561 9562
    return ret;
}

9563

9564 9565 9566 9567 9568 9569 9570 9571 9572 9573
static int qemuDomainGetBlockInfo(virDomainPtr dom,
                                  const char *path,
                                  virDomainBlockInfoPtr info,
                                  unsigned int flags) {
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
    int fd = -1;
    off_t end;
    virStorageFileMetadata meta;
9574
    virDomainDiskDefPtr disk = NULL;
9575 9576 9577 9578 9579 9580 9581 9582 9583 9584 9585 9586 9587 9588 9589 9590 9591 9592 9593 9594 9595 9596 9597 9598 9599 9600
    struct stat sb;
    int i;

    virCheckFlags(0, -1);

    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
    qemuDriverUnlock(driver);
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    if (!path || path[0] == '\0') {
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("NULL or empty path"));
        goto cleanup;
    }

    /* Check the path belongs to this domain. */
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (vm->def->disks[i]->src != NULL &&
            STREQ (vm->def->disks[i]->src, path)) {
9601
            disk = vm->def->disks[i];
9602 9603 9604 9605
            break;
        }
    }

9606
    if (!disk) {
9607 9608 9609 9610 9611 9612 9613 9614 9615 9616 9617 9618 9619 9620 9621 9622 9623 9624 9625 9626 9627 9628 9629 9630 9631
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("invalid path %s not assigned to domain"), path);
        goto cleanup;
    }

    /* The path is correct, now try to open it and get its size. */
    fd = open (path, O_RDONLY);
    if (fd == -1) {
        virReportSystemError(errno,
                             _("failed to open path '%s'"), path);
        goto cleanup;
    }

    /* Probe for magic formats */
    if (virStorageFileGetMetadataFromFD(path, fd, &meta) < 0)
        goto cleanup;

    /* Get info for normal formats */
    if (fstat(fd, &sb) < 0) {
        virReportSystemError(errno,
                             _("cannot stat file '%s'"), path);
        goto cleanup;
    }

    if (S_ISREG(sb.st_mode)) {
9632
#ifndef WIN32
9633 9634 9635 9636 9637 9638 9639 9640 9641 9642 9643 9644 9645 9646 9647 9648 9649 9650 9651 9652 9653 9654 9655 9656 9657 9658 9659 9660
        info->physical = (unsigned long long)sb.st_blocks *
            (unsigned long long)DEV_BSIZE;
#else
        info->physical = sb.st_size;
#endif
        /* Regular files may be sparse, so logical size (capacity) is not same
         * as actual physical above
         */
        info->capacity = sb.st_size;
    } else {
        /* NB. Because we configure with AC_SYS_LARGEFILE, off_t should
         * be 64 bits on all platforms.
         */
        end = lseek (fd, 0, SEEK_END);
        if (end == (off_t)-1) {
            virReportSystemError(errno,
                                 _("failed to seek to end of %s"), path);
            goto cleanup;
        }
        info->physical = end;
        info->capacity = end;
    }

    /* If the file we probed has a capacity set, then override
     * what we calculated from file/block extents */
    if (meta.capacity)
        info->capacity = meta.capacity;

9661
    /* Set default value .. */
9662 9663
    info->allocation = info->physical;

9664 9665 9666 9667 9668 9669 9670 9671 9672 9673 9674 9675 9676 9677 9678 9679 9680 9681 9682 9683 9684 9685
    /* ..but if guest is running & not using raw
       disk format and on a block device, then query
       highest allocated extent from QEMU */
    if (virDomainObjIsActive(vm) &&
        disk->type == VIR_DOMAIN_DISK_TYPE_BLOCK &&
        meta.format != VIR_STORAGE_FILE_RAW &&
        S_ISBLK(sb.st_mode)) {
        qemuDomainObjPrivatePtr priv = vm->privateData;
        if (qemuDomainObjBeginJob(vm) < 0)
            goto cleanup;

        qemuDomainObjEnterMonitor(vm);
        ret = qemuMonitorGetBlockExtent(priv->mon,
                                        disk->info.alias,
                                        &info->allocation);
        qemuDomainObjExitMonitor(vm);

        if (qemuDomainObjEndJob(vm) == 0)
            vm = NULL;
    } else {
        ret = 0;
    }
9686 9687 9688 9689 9690 9691 9692 9693 9694 9695

cleanup:
    if (fd != -1)
        close(fd);
    if (vm)
        virDomainObjUnlock(vm);
    return ret;
}


9696
static int
9697 9698 9699 9700
qemuDomainEventRegister(virConnectPtr conn,
                        virConnectDomainEventCallback callback,
                        void *opaque,
                        virFreeCallback freecb)
9701
{
9702 9703 9704
    struct qemud_driver *driver = conn->privateData;
    int ret;

9705
    qemuDriverLock(driver);
9706 9707
    ret = virDomainEventCallbackListAdd(conn, driver->domainEventCallbacks,
                                        callback, opaque, freecb);
9708
    qemuDriverUnlock(driver);
9709

9710
    return ret;
9711 9712
}

9713

9714
static int
9715 9716
qemuDomainEventDeregister(virConnectPtr conn,
                          virConnectDomainEventCallback callback)
9717
{
9718 9719 9720
    struct qemud_driver *driver = conn->privateData;
    int ret;

9721
    qemuDriverLock(driver);
9722 9723 9724 9725 9726 9727
    if (driver->domainEventDispatching)
        ret = virDomainEventCallbackListMarkDelete(conn, driver->domainEventCallbacks,
                                                   callback);
    else
        ret = virDomainEventCallbackListRemove(conn, driver->domainEventCallbacks,
                                               callback);
9728
    qemuDriverUnlock(driver);
9729

9730
    return ret;
9731 9732
}

9733 9734 9735 9736 9737 9738 9739 9740 9741 9742 9743 9744 9745 9746 9747 9748 9749 9750 9751 9752 9753 9754 9755 9756 9757 9758 9759 9760 9761 9762 9763 9764 9765 9766 9767 9768 9769 9770 9771 9772 9773 9774 9775

static int
qemuDomainEventRegisterAny(virConnectPtr conn,
                           virDomainPtr dom,
                           int eventID,
                           virConnectDomainEventGenericCallback callback,
                           void *opaque,
                           virFreeCallback freecb)
{
    struct qemud_driver *driver = conn->privateData;
    int ret;

    qemuDriverLock(driver);
    ret = virDomainEventCallbackListAddID(conn,
                                          driver->domainEventCallbacks,
                                          dom, eventID,
                                          callback, opaque, freecb);
    qemuDriverUnlock(driver);

    return ret;
}


static int
qemuDomainEventDeregisterAny(virConnectPtr conn,
                             int callbackID)
{
    struct qemud_driver *driver = conn->privateData;
    int ret;

    qemuDriverLock(driver);
    if (driver->domainEventDispatching)
        ret = virDomainEventCallbackListMarkDeleteID(conn, driver->domainEventCallbacks,
                                                     callbackID);
    else
        ret = virDomainEventCallbackListRemoveID(conn, driver->domainEventCallbacks,
                                                 callbackID);
    qemuDriverUnlock(driver);

    return ret;
}


9776 9777
static void qemuDomainEventDispatchFunc(virConnectPtr conn,
                                        virDomainEventPtr event,
9778
                                        virConnectDomainEventGenericCallback cb,
9779 9780
                                        void *cbopaque,
                                        void *opaque)
9781
{
9782
    struct qemud_driver *driver = opaque;
9783

9784 9785 9786 9787 9788 9789 9790 9791 9792 9793 9794 9795 9796 9797 9798 9799 9800 9801 9802 9803 9804 9805 9806 9807 9808 9809 9810 9811 9812 9813 9814 9815 9816 9817 9818 9819 9820 9821 9822 9823 9824 9825 9826 9827
    /* 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);
9828 9829
}

D
Daniel Veillard 已提交
9830 9831
/* Migration support. */

C
Chris Lalancette 已提交
9832 9833 9834 9835 9836 9837 9838 9839 9840 9841 9842 9843 9844 9845 9846 9847 9848 9849 9850
/* Tunnelled migration stream support */
struct qemuStreamMigFile {
    int fd;

    int watch;
    unsigned int cbRemoved;
    unsigned int dispatching;
    virStreamEventCallback cb;
    void *opaque;
    virFreeCallback ff;
};

static int qemuStreamMigRemoveCallback(virStreamPtr stream)
{
    struct qemud_driver *driver = stream->conn->privateData;
    struct qemuStreamMigFile *qemust = stream->privateData;
    int ret = -1;

    if (!qemust) {
9851 9852
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("stream is not open"));
C
Chris Lalancette 已提交
9853 9854 9855 9856 9857
        return -1;
    }

    qemuDriverLock(driver);
    if (qemust->watch == 0) {
9858 9859
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("stream does not have a callback registered"));
C
Chris Lalancette 已提交
9860 9861 9862 9863 9864 9865 9866 9867 9868 9869 9870 9871 9872 9873 9874 9875 9876 9877 9878 9879 9880 9881 9882 9883 9884 9885 9886 9887
        goto cleanup;
    }

    virEventRemoveHandle(qemust->watch);
    if (qemust->dispatching)
        qemust->cbRemoved = 1;
    else if (qemust->ff)
        (qemust->ff)(qemust->opaque);

    qemust->watch = 0;
    qemust->ff = NULL;
    qemust->cb = NULL;
    qemust->opaque = NULL;

    ret = 0;

cleanup:
    qemuDriverUnlock(driver);
    return ret;
}

static int qemuStreamMigUpdateCallback(virStreamPtr stream, int events)
{
    struct qemud_driver *driver = stream->conn->privateData;
    struct qemuStreamMigFile *qemust = stream->privateData;
    int ret = -1;

    if (!qemust) {
9888 9889
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("stream is not open"));
C
Chris Lalancette 已提交
9890 9891 9892 9893 9894
        return -1;
    }

    qemuDriverLock(driver);
    if (qemust->watch == 0) {
9895 9896
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("stream does not have a callback registered"));
C
Chris Lalancette 已提交
9897 9898 9899 9900 9901 9902 9903 9904 9905 9906 9907 9908 9909 9910 9911 9912 9913 9914 9915 9916 9917 9918 9919 9920 9921 9922 9923 9924 9925 9926 9927 9928 9929 9930 9931 9932 9933 9934 9935 9936 9937 9938 9939 9940 9941 9942 9943 9944 9945 9946 9947 9948 9949 9950 9951 9952 9953
        goto cleanup;
    }

    virEventUpdateHandle(qemust->watch, events);

    ret = 0;

cleanup:
    qemuDriverUnlock(driver);
    return ret;
}

static void qemuStreamMigEvent(int watch ATTRIBUTE_UNUSED,
                               int fd ATTRIBUTE_UNUSED,
                               int events,
                               void *opaque)
{
    virStreamPtr stream = opaque;
    struct qemud_driver *driver = stream->conn->privateData;
    struct qemuStreamMigFile *qemust = stream->privateData;
    virStreamEventCallback cb;
    void *cbopaque;
    virFreeCallback ff;

    qemuDriverLock(driver);
    if (!qemust || !qemust->cb) {
        qemuDriverUnlock(driver);
        return;
    }

    cb = qemust->cb;
    cbopaque = qemust->opaque;
    ff = qemust->ff;
    qemust->dispatching = 1;
    qemuDriverUnlock(driver);

    cb(stream, events, cbopaque);

    qemuDriverLock(driver);
    qemust->dispatching = 0;
    if (qemust->cbRemoved && ff)
        (ff)(cbopaque);
    qemuDriverUnlock(driver);
}

static int
qemuStreamMigAddCallback(virStreamPtr st,
                         int events,
                         virStreamEventCallback cb,
                         void *opaque,
                         virFreeCallback ff)
{
    struct qemud_driver *driver = st->conn->privateData;
    struct qemuStreamMigFile *qemust = st->privateData;
    int ret = -1;

    if (!qemust) {
9954 9955
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("stream is not open"));
C
Chris Lalancette 已提交
9956 9957 9958 9959 9960
        return -1;
    }

    qemuDriverLock(driver);
    if (qemust->watch != 0) {
9961 9962
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("stream already has a callback registered"));
C
Chris Lalancette 已提交
9963 9964 9965 9966 9967 9968 9969 9970
        goto cleanup;
    }

    if ((qemust->watch = virEventAddHandle(qemust->fd,
                                           events,
                                           qemuStreamMigEvent,
                                           st,
                                           NULL)) < 0) {
9971 9972
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("cannot register file watch on stream"));
C
Chris Lalancette 已提交
9973 9974 9975 9976 9977 9978 9979 9980 9981 9982 9983 9984 9985 9986 9987 9988 9989 9990 9991 9992 9993 9994 9995 9996 9997 9998 9999 10000 10001 10002 10003 10004
        goto cleanup;
    }

    qemust->cbRemoved = 0;
    qemust->cb = cb;
    qemust->opaque = opaque;
    qemust->ff = ff;
    virStreamRef(st);

    ret = 0;

cleanup:
    qemuDriverUnlock(driver);
    return ret;
}

static void qemuStreamMigFree(struct qemuStreamMigFile *qemust)
{
    if (qemust->fd != -1)
        close(qemust->fd);
    VIR_FREE(qemust);
}

static struct qemuStreamMigFile *qemuStreamMigOpen(virStreamPtr st,
                                                   const char *unixfile)
{
    struct qemuStreamMigFile *qemust = NULL;
    struct sockaddr_un sa_qemu;
    int i = 0;
    int timeout = 3;
    int ret;

10005
    if (VIR_ALLOC(qemust) < 0) {
10006
        virReportOOMError();
C
Chris Lalancette 已提交
10007
        return NULL;
10008
    }
C
Chris Lalancette 已提交
10009 10010 10011 10012 10013 10014 10015 10016 10017 10018 10019 10020 10021 10022 10023 10024 10025 10026 10027 10028 10029 10030 10031 10032 10033 10034 10035 10036 10037 10038 10039 10040 10041 10042 10043 10044 10045 10046 10047 10048 10049 10050 10051 10052 10053 10054 10055 10056 10057 10058 10059 10060 10061 10062 10063 10064 10065 10066 10067 10068 10069

    qemust->fd = socket(AF_UNIX, SOCK_STREAM, 0);
    if (qemust->fd < 0)
        goto cleanup;

    memset(&sa_qemu, 0, sizeof(sa_qemu));
    sa_qemu.sun_family = AF_UNIX;
    if (virStrcpy(sa_qemu.sun_path, unixfile, sizeof(sa_qemu.sun_path)) == NULL)
        goto cleanup;

    do {
        ret = connect(qemust->fd, (struct sockaddr *)&sa_qemu, sizeof(sa_qemu));
        if (ret == 0)
            break;

        if (errno == ENOENT || errno == ECONNREFUSED) {
            /* ENOENT       : Socket may not have shown up yet
             * ECONNREFUSED : Leftover socket hasn't been removed yet */
            continue;
        }

        goto cleanup;
    } while ((++i <= timeout*5) && (usleep(.2 * 1000000) <= 0));

    if ((st->flags & VIR_STREAM_NONBLOCK) && virSetNonBlock(qemust->fd) < 0)
        goto cleanup;

    return qemust;

cleanup:
    qemuStreamMigFree(qemust);
    return NULL;
}

static int
qemuStreamMigClose(virStreamPtr st)
{
    struct qemud_driver *driver = st->conn->privateData;
    struct qemuStreamMigFile *qemust = st->privateData;

    if (!qemust)
        return 0;

    qemuDriverLock(driver);

    qemuStreamMigFree(qemust);

    st->privateData = NULL;

    qemuDriverUnlock(driver);

    return 0;
}

static int qemuStreamMigWrite(virStreamPtr st, const char *bytes, size_t nbytes)
{
    struct qemud_driver *driver = st->conn->privateData;
    struct qemuStreamMigFile *qemust = st->privateData;
    int ret;

    if (!qemust) {
10070 10071
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("stream is not open"));
C
Chris Lalancette 已提交
10072 10073 10074 10075 10076 10077 10078 10079 10080 10081 10082 10083 10084 10085
        return -1;
    }

    qemuDriverLock(driver);

retry:
    ret = write(qemust->fd, bytes, nbytes);
    if (ret < 0) {
        if (errno == EAGAIN || errno == EWOULDBLOCK) {
            ret = -2;
        } else if (errno == EINTR) {
            goto retry;
        } else {
            ret = -1;
10086
            virReportSystemError(errno, "%s",
C
Chris Lalancette 已提交
10087 10088 10089 10090 10091 10092 10093 10094 10095 10096 10097 10098 10099 10100 10101 10102 10103 10104 10105 10106 10107 10108 10109 10110 10111 10112 10113 10114 10115 10116 10117 10118 10119 10120 10121 10122 10123 10124
                                 _("cannot write to stream"));
        }
    }

    qemuDriverUnlock(driver);
    return ret;
}

static virStreamDriver qemuStreamMigDrv = {
    .streamSend = qemuStreamMigWrite,
    .streamFinish = qemuStreamMigClose,
    .streamAbort = qemuStreamMigClose,
    .streamAddCallback = qemuStreamMigAddCallback,
    .streamUpdateCallback = qemuStreamMigUpdateCallback,
    .streamRemoveCallback = qemuStreamMigRemoveCallback
};

/* Prepare is the first step, and it runs on the destination host.
 *
 * This version starts an empty VM listening on a localhost TCP port, and
 * sets up the corresponding virStream to handle the incoming data.
 */
static int
qemudDomainMigratePrepareTunnel(virConnectPtr dconn,
                                virStreamPtr st,
                                unsigned long flags,
                                const char *dname,
                                unsigned long resource ATTRIBUTE_UNUSED,
                                const char *dom_xml)
{
    struct qemud_driver *driver = dconn->privateData;
    virDomainDefPtr def = NULL;
    virDomainObjPtr vm = NULL;
    char *migrateFrom;
    virDomainEventPtr event = NULL;
    int ret = -1;
    int internalret;
    char *unixfile = NULL;
10125
    unsigned long long qemuCmdFlags;
C
Chris Lalancette 已提交
10126 10127 10128 10129
    struct qemuStreamMigFile *qemust = NULL;

    qemuDriverLock(driver);
    if (!dom_xml) {
10130 10131
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("no domain XML passed"));
C
Chris Lalancette 已提交
10132 10133 10134
        goto cleanup;
    }
    if (!(flags & VIR_MIGRATE_TUNNELLED)) {
10135
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
C
Chris Lalancette 已提交
10136 10137 10138 10139
                         "%s", _("PrepareTunnel called but no TUNNELLED flag set"));
        goto cleanup;
    }
    if (st == NULL) {
10140 10141
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("tunnelled migration requested but NULL stream passed"));
C
Chris Lalancette 已提交
10142 10143 10144 10145
        goto cleanup;
    }

    /* Parse the domain XML. */
10146
    if (!(def = virDomainDefParseString(driver->caps, dom_xml,
C
Chris Lalancette 已提交
10147
                                        VIR_DOMAIN_XML_INACTIVE))) {
10148 10149
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to parse XML"));
C
Chris Lalancette 已提交
10150 10151 10152 10153
        goto cleanup;
    }

    /* Target domain name, maybe renamed. */
10154 10155 10156 10157 10158 10159
    if (dname) {
        VIR_FREE(def->name);
        def->name = strdup(dname);
        if (def->name == NULL)
            goto cleanup;
    }
C
Chris Lalancette 已提交
10160

10161 10162
    if (virDomainObjIsDuplicate(&driver->domains, def, 1) < 0)
        goto cleanup;
C
Chris Lalancette 已提交
10163

10164
    if (!(vm = virDomainAssignDef(driver->caps,
C
Chris Lalancette 已提交
10165
                                  &driver->domains,
10166
                                  def, true))) {
10167
        /* virDomainAssignDef already set the error */
C
Chris Lalancette 已提交
10168 10169 10170 10171
        goto cleanup;
    }
    def = NULL;

10172 10173 10174
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

C
Chris Lalancette 已提交
10175 10176 10177 10178 10179
    /* Domain starts inactive, even if the domain XML had an id field. */
    vm->def->id = -1;

    if (virAsprintf(&unixfile, "%s/qemu.tunnelmigrate.dest.%s",
                    driver->stateDir, vm->def->name) < 0) {
10180
        virReportOOMError();
10181
        goto endjob;
C
Chris Lalancette 已提交
10182 10183 10184 10185 10186
    }
    unlink(unixfile);

    /* check that this qemu version supports the interactive exec */
    if (qemudExtractVersionInfo(vm->def->emulator, NULL, &qemuCmdFlags) < 0) {
10187 10188 10189
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("Cannot determine QEMU argv syntax %s"),
                        vm->def->emulator);
10190
        goto endjob;
C
Chris Lalancette 已提交
10191 10192 10193 10194 10195 10196
    }
    if (qemuCmdFlags & QEMUD_CMD_FLAG_MIGRATE_QEMU_UNIX)
        internalret = virAsprintf(&migrateFrom, "unix:%s", unixfile);
    else if (qemuCmdFlags & QEMUD_CMD_FLAG_MIGRATE_QEMU_EXEC)
        internalret = virAsprintf(&migrateFrom, "exec:nc -U -l %s", unixfile);
    else {
10197 10198
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("Destination qemu is too old to support tunnelled migration"));
10199
        goto endjob;
C
Chris Lalancette 已提交
10200 10201
    }
    if (internalret < 0) {
10202
        virReportOOMError();
10203
        goto endjob;
C
Chris Lalancette 已提交
10204 10205 10206 10207
    }
    /* Start the QEMU daemon, with the same command-line arguments plus
     * -incoming unix:/path/to/file or exec:nc -U /path/to/file
     */
10208 10209
    internalret = qemudStartVMDaemon(dconn, driver, vm, migrateFrom, true,
                                     -1, NULL);
C
Chris Lalancette 已提交
10210 10211 10212 10213 10214 10215 10216 10217 10218
    VIR_FREE(migrateFrom);
    if (internalret < 0) {
        /* Note that we don't set an error here because qemudStartVMDaemon
         * should have already done that.
         */
        if (!vm->persistent) {
            virDomainRemoveInactive(&driver->domains, vm);
            vm = NULL;
        }
10219
        goto endjob;
C
Chris Lalancette 已提交
10220 10221 10222 10223
    }

    qemust = qemuStreamMigOpen(st, unixfile);
    if (qemust == NULL) {
10224
        qemudShutdownVMDaemon(driver, vm, 0);
10225
        if (!vm->persistent) {
10226 10227
            if (qemuDomainObjEndJob(vm) > 0)
                virDomainRemoveInactive(&driver->domains, vm);
10228 10229
            vm = NULL;
        }
10230
        virReportSystemError(errno,
C
Chris Lalancette 已提交
10231 10232
                             _("cannot open unix socket '%s' for tunnelled migration"),
                             unixfile);
10233
        goto endjob;
C
Chris Lalancette 已提交
10234 10235 10236 10237 10238 10239 10240 10241 10242 10243
    }

    st->driver = &qemuStreamMigDrv;
    st->privateData = qemust;

    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_MIGRATED);
    ret = 0;

10244
endjob:
10245 10246 10247
    if (vm &&
        qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
10248

C
Chris Lalancette 已提交
10249 10250
cleanup:
    virDomainDefFree(def);
10251 10252
    if (unixfile)
        unlink(unixfile);
C
Chris Lalancette 已提交
10253 10254 10255 10256 10257 10258 10259 10260 10261
    VIR_FREE(unixfile);
    if (vm)
        virDomainObjUnlock(vm);
    if (event)
        qemuDomainEventQueue(driver, event);
    qemuDriverUnlock(driver);
    return ret;
}

D
Daniel Veillard 已提交
10262 10263 10264 10265
/* Prepare is the first step, and it runs on the destination host.
 *
 * This starts an empty VM listening on a TCP port.
 */
10266
static int ATTRIBUTE_NONNULL (5)
D
Daniel Veillard 已提交
10267 10268 10269 10270 10271
qemudDomainMigratePrepare2 (virConnectPtr dconn,
                            char **cookie ATTRIBUTE_UNUSED,
                            int *cookielen ATTRIBUTE_UNUSED,
                            const char *uri_in,
                            char **uri_out,
C
Chris Lalancette 已提交
10272
                            unsigned long flags,
D
Daniel Veillard 已提交
10273 10274 10275 10276 10277
                            const char *dname,
                            unsigned long resource ATTRIBUTE_UNUSED,
                            const char *dom_xml)
{
    static int port = 0;
10278 10279
    struct qemud_driver *driver = dconn->privateData;
    virDomainDefPtr def = NULL;
D
Daniel Veillard 已提交
10280 10281
    virDomainObjPtr vm = NULL;
    int this_port;
C
Chris Lalancette 已提交
10282
    char *hostname = NULL;
D
Daniel Veillard 已提交
10283 10284
    char migrateFrom [64];
    const char *p;
10285
    virDomainEventPtr event = NULL;
10286
    int ret = -1;
10287
    int internalret;
10288

C
Chris Lalancette 已提交
10289 10290 10291 10292 10293 10294 10295 10296 10297
    virCheckFlags(VIR_MIGRATE_LIVE |
                  VIR_MIGRATE_PEER2PEER |
                  VIR_MIGRATE_TUNNELLED |
                  VIR_MIGRATE_PERSIST_DEST |
                  VIR_MIGRATE_UNDEFINE_SOURCE |
                  VIR_MIGRATE_PAUSED |
                  VIR_MIGRATE_NON_SHARED_DISK |
                  VIR_MIGRATE_NON_SHARED_INC, -1);

10298
    *uri_out = NULL;
D
Daniel Veillard 已提交
10299

10300
    qemuDriverLock(driver);
C
Chris Lalancette 已提交
10301 10302 10303 10304
    if (flags & VIR_MIGRATE_TUNNELLED) {
        /* this is a logical error; we never should have gotten here with
         * VIR_MIGRATE_TUNNELLED set
         */
10305 10306
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("Tunnelled migration requested but invalid RPC method called"));
C
Chris Lalancette 已提交
10307 10308 10309
        goto cleanup;
    }

D
Daniel Veillard 已提交
10310
    if (!dom_xml) {
10311 10312
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("no domain XML passed"));
10313
        goto cleanup;
D
Daniel Veillard 已提交
10314 10315 10316 10317 10318 10319 10320 10321 10322 10323 10324 10325 10326 10327 10328 10329 10330
    }

    /* 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 */
C
Chris Lalancette 已提交
10331
        if ((hostname = virGetHostname(NULL)) == NULL)
10332
            goto cleanup;
D
Daniel Veillard 已提交
10333

C
Chris Lalancette 已提交
10334 10335 10336 10337 10338 10339
        if (STRPREFIX(hostname, "localhost")) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("hostname on destination resolved to localhost, but migration requires an FQDN"));
            goto cleanup;
        }

10340 10341 10342 10343 10344
        /* XXX this really should have been a properly well-formed
         * URI, but we can't add in tcp:// now without breaking
         * compatability with old targets. We at least make the
         * new targets accept both syntaxes though.
         */
D
Daniel Veillard 已提交
10345
        /* Caller frees */
10346 10347
        internalret = virAsprintf(uri_out, "tcp:%s:%d", hostname, this_port);
        if (internalret < 0) {
10348
            virReportOOMError();
10349
            goto cleanup;
D
Daniel Veillard 已提交
10350 10351 10352 10353 10354 10355
        }
    } 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.
         */
10356
        if (!STRPREFIX (uri_in, "tcp:")) {
10357 10358
            qemuReportError (VIR_ERR_INVALID_ARG,
                             "%s", _("only tcp URIs are supported for KVM/QEMU migrations"));
10359
            goto cleanup;
D
Daniel Veillard 已提交
10360 10361 10362 10363
        }

        /* Get the port number. */
        p = strrchr (uri_in, ':');
10364 10365 10366 10367 10368 10369 10370 10371
        if (p == strchr(uri_in, ':')) {
            /* Generate a port */
            this_port = QEMUD_MIGRATION_FIRST_PORT + port++;
            if (port == QEMUD_MIGRATION_NUM_PORTS)
                port = 0;

            /* Caller frees */
            if (virAsprintf(uri_out, "%s:%d", uri_in, this_port) < 0) {
10372
                virReportOOMError();
10373 10374 10375 10376 10377 10378 10379
                goto cleanup;
            }

        } else {
            p++; /* definitely has a ':' in it, see above */
            this_port = virParseNumber (&p);
            if (this_port == -1 || p-uri_in != strlen (uri_in)) {
10380 10381
                qemuReportError(VIR_ERR_INVALID_ARG,
                                "%s", _("URI ended with incorrect ':port'"));
10382 10383
                goto cleanup;
            }
D
Daniel Veillard 已提交
10384 10385 10386
        }
    }

10387
    if (*uri_out)
10388 10389
        VIR_DEBUG("Generated uri_out=%s", *uri_out);

D
Daniel Veillard 已提交
10390
    /* Parse the domain XML. */
10391
    if (!(def = virDomainDefParseString(driver->caps, dom_xml,
10392
                                        VIR_DOMAIN_XML_INACTIVE))) {
10393 10394
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to parse XML"));
10395
        goto cleanup;
D
Daniel Veillard 已提交
10396 10397 10398
    }

    /* Target domain name, maybe renamed. */
10399 10400 10401 10402 10403 10404
    if (dname) {
        VIR_FREE(def->name);
        def->name = strdup(dname);
        if (def->name == NULL)
            goto cleanup;
    }
D
Daniel Veillard 已提交
10405

10406 10407
    if (virDomainObjIsDuplicate(&driver->domains, def, 1) < 0)
        goto cleanup;
D
Daniel Veillard 已提交
10408

10409
    if (!(vm = virDomainAssignDef(driver->caps,
D
Daniel Veillard 已提交
10410
                                  &driver->domains,
10411
                                  def, true))) {
10412
        /* virDomainAssignDef already set the error */
10413
        goto cleanup;
D
Daniel Veillard 已提交
10414
    }
10415
    def = NULL;
D
Daniel Veillard 已提交
10416

10417 10418 10419
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

D
Daniel Veillard 已提交
10420 10421 10422 10423 10424 10425 10426
    /* 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);
10427 10428
    if (qemudStartVMDaemon (dconn, driver, vm, migrateFrom, true,
                            -1, NULL) < 0) {
10429 10430 10431
        /* Note that we don't set an error here because qemudStartVMDaemon
         * should have already done that.
         */
10432
        if (!vm->persistent) {
10433 10434
            if (qemuDomainObjEndJob(vm) > 0)
                virDomainRemoveInactive(&driver->domains, vm);
10435 10436
            vm = NULL;
        }
10437
        goto endjob;
D
Daniel Veillard 已提交
10438
    }
10439 10440 10441 10442

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

10445
endjob:
10446 10447 10448
    if (vm &&
        qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
10449

10450
cleanup:
C
Chris Lalancette 已提交
10451
    VIR_FREE(hostname);
10452
    virDomainDefFree(def);
C
Chris Lalancette 已提交
10453
    if (ret != 0)
10454
        VIR_FREE(*uri_out);
10455 10456
    if (vm)
        virDomainObjUnlock(vm);
10457 10458
    if (event)
        qemuDomainEventQueue(driver, event);
10459
    qemuDriverUnlock(driver);
10460
    return ret;
C
Chris Lalancette 已提交
10461 10462 10463

}

10464 10465 10466 10467

/* Perform migration using QEMU's native TCP migrate support,
 * not encrypted obviously
 */
10468
static int doNativeMigrate(struct qemud_driver *driver,
10469 10470
                           virDomainObjPtr vm,
                           const char *uri,
10471
                           unsigned int flags,
10472 10473 10474 10475
                           const char *dname ATTRIBUTE_UNUSED,
                           unsigned long resource)
{
    int ret = -1;
10476
    xmlURIPtr uribits = NULL;
10477
    qemuDomainObjPrivatePtr priv = vm->privateData;
10478 10479
    unsigned int background_flags = 0;

10480 10481 10482 10483 10484
    /* Issue the migrate command. */
    if (STRPREFIX(uri, "tcp:") && !STRPREFIX(uri, "tcp://")) {
        /* HACK: source host generates bogus URIs, so fix them up */
        char *tmpuri;
        if (virAsprintf(&tmpuri, "tcp://%s", uri + strlen("tcp:")) < 0) {
10485
            virReportOOMError();
10486 10487 10488 10489 10490 10491 10492 10493
            goto cleanup;
        }
        uribits = xmlParseURI(tmpuri);
        VIR_FREE(tmpuri);
    } else {
        uribits = xmlParseURI(uri);
    }
    if (!uribits) {
10494 10495
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("cannot parse URI %s"), uri);
10496 10497 10498
        goto cleanup;
    }

10499
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
10500
    if (resource > 0 &&
10501
        qemuMonitorSetMigrationSpeed(priv->mon, resource) < 0) {
10502
        qemuDomainObjExitMonitorWithDriver(driver, vm);
10503
        goto cleanup;
10504
    }
10505

10506 10507 10508 10509 10510 10511 10512 10513
    if (flags & VIR_MIGRATE_NON_SHARED_DISK)
        background_flags |= QEMU_MONITOR_MIGRATE_NON_SHARED_DISK;

    if (flags & VIR_MIGRATE_NON_SHARED_INC)
        background_flags |= QEMU_MONITOR_MIGRATE_NON_SHARED_INC;

    if (qemuMonitorMigrateToHost(priv->mon, background_flags, uribits->server,
                                 uribits->port) < 0) {
10514
        qemuDomainObjExitMonitorWithDriver(driver, vm);
10515 10516
        goto cleanup;
    }
10517
    qemuDomainObjExitMonitorWithDriver(driver, vm);
10518

10519
    if (qemuDomainWaitForMigrationComplete(driver, vm) < 0)
10520 10521 10522 10523 10524 10525 10526 10527 10528 10529
        goto cleanup;

    ret = 0;

cleanup:
    xmlFreeURI(uribits);
    return ret;
}


10530 10531
#define TUNNEL_SEND_BUF_SIZE 65536

10532
static int doTunnelSendAll(virStreamPtr st,
10533 10534
                           int sock)
{
10535 10536 10537 10538 10539 10540 10541 10542
    char *buffer;
    int nbytes = TUNNEL_SEND_BUF_SIZE;

    if (VIR_ALLOC_N(buffer, TUNNEL_SEND_BUF_SIZE) < 0) {
        virReportOOMError();
        virStreamAbort(st);
        return -1;
    }
10543 10544 10545 10546 10547

    /* XXX should honour the 'resource' parameter here */
    for (;;) {
        nbytes = saferead(sock, buffer, nbytes);
        if (nbytes < 0) {
10548
            virReportSystemError(errno, "%s",
10549
                                 _("tunnelled migration failed to read from qemu"));
10550 10551
            virStreamAbort(st);
            VIR_FREE(buffer);
10552 10553 10554 10555 10556 10557 10558
            return -1;
        }
        else if (nbytes == 0)
            /* EOF; get out of here */
            break;

        if (virStreamSend(st, buffer, nbytes) < 0) {
10559 10560
            qemuReportError(VIR_ERR_OPERATION_FAILED, "%s",
                            _("Failed to write migration data to remote libvirtd"));
10561
            VIR_FREE(buffer);
10562 10563 10564 10565
            return -1;
        }
    }

10566 10567
    VIR_FREE(buffer);

10568 10569 10570 10571 10572 10573 10574
    if (virStreamFinish(st) < 0)
        /* virStreamFinish set the error for us */
        return -1;

    return 0;
}

C
Chris Lalancette 已提交
10575
static int doTunnelMigrate(virDomainPtr dom,
10576
                           struct qemud_driver *driver,
10577
                           virConnectPtr dconn,
C
Chris Lalancette 已提交
10578
                           virDomainObjPtr vm,
10579
                           const char *dom_xml,
C
Chris Lalancette 已提交
10580 10581 10582 10583 10584
                           const char *uri,
                           unsigned long flags,
                           const char *dname,
                           unsigned long resource)
{
10585
    qemuDomainObjPrivatePtr priv = vm->privateData;
10586 10587
    int client_sock = -1;
    int qemu_sock = -1;
C
Chris Lalancette 已提交
10588 10589
    struct sockaddr_un sa_qemu, sa_client;
    socklen_t addrlen;
10590
    virDomainPtr ddomain = NULL;
C
Chris Lalancette 已提交
10591
    int retval = -1;
10592
    virStreamPtr st = NULL;
C
Chris Lalancette 已提交
10593 10594
    char *unixfile = NULL;
    int internalret;
10595
    unsigned long long qemuCmdFlags;
C
Chris Lalancette 已提交
10596 10597
    int status;
    unsigned long long transferred, remaining, total;
10598
    unsigned int background_flags = QEMU_MONITOR_MIGRATE_BACKGROUND;
C
Chris Lalancette 已提交
10599

10600 10601 10602 10603 10604 10605 10606 10607
    /*
     * The order of operations is important here to avoid touching
     * the source VM until we are very sure we can successfully
     * start the migration operation.
     *
     *   1. setup local support infrastructure (eg sockets)
     *   2. setup destination fully
     *   3. start migration on source
C
Chris Lalancette 已提交
10608 10609
     */

10610

10611
    /* Stage 1. setup local support infrastructure */
C
Chris Lalancette 已提交
10612 10613 10614

    if (virAsprintf(&unixfile, "%s/qemu.tunnelmigrate.src.%s",
                    driver->stateDir, vm->def->name) < 0) {
10615
        virReportOOMError();
10616
        goto cleanup;
C
Chris Lalancette 已提交
10617 10618 10619 10620
    }

    qemu_sock = socket(AF_UNIX, SOCK_STREAM, 0);
    if (qemu_sock < 0) {
10621
        virReportSystemError(errno, "%s",
C
Chris Lalancette 已提交
10622
                             _("cannot open tunnelled migration socket"));
10623
        goto cleanup;
C
Chris Lalancette 已提交
10624 10625 10626 10627 10628
    }
    memset(&sa_qemu, 0, sizeof(sa_qemu));
    sa_qemu.sun_family = AF_UNIX;
    if (virStrcpy(sa_qemu.sun_path, unixfile,
                  sizeof(sa_qemu.sun_path)) == NULL) {
10629 10630 10631
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("Unix socket '%s' too big for destination"),
                        unixfile);
10632
        goto cleanup;
C
Chris Lalancette 已提交
10633 10634 10635
    }
    unlink(unixfile);
    if (bind(qemu_sock, (struct sockaddr *)&sa_qemu, sizeof(sa_qemu)) < 0) {
10636
        virReportSystemError(errno,
C
Chris Lalancette 已提交
10637 10638
                             _("Cannot bind to unix socket '%s' for tunnelled migration"),
                             unixfile);
10639
        goto cleanup;
C
Chris Lalancette 已提交
10640 10641
    }
    if (listen(qemu_sock, 1) < 0) {
10642
        virReportSystemError(errno,
C
Chris Lalancette 已提交
10643 10644
                             _("Cannot listen on unix socket '%s' for tunnelled migration"),
                             unixfile);
10645
        goto cleanup;
C
Chris Lalancette 已提交
10646 10647 10648 10649
    }

    /* check that this qemu version supports the unix migration */
    if (qemudExtractVersionInfo(vm->def->emulator, NULL, &qemuCmdFlags) < 0) {
10650 10651 10652
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("Cannot extract Qemu version from '%s'"),
                        vm->def->emulator);
10653 10654 10655 10656 10657
        goto cleanup;
    }

    if (!(qemuCmdFlags & QEMUD_CMD_FLAG_MIGRATE_QEMU_UNIX) &&
        !(qemuCmdFlags & QEMUD_CMD_FLAG_MIGRATE_QEMU_EXEC)) {
10658 10659
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("Source qemu is too old to support tunnelled migration"));
10660
        goto cleanup;
C
Chris Lalancette 已提交
10661
    }
10662 10663 10664 10665 10666 10667 10668 10669 10670 10671 10672 10673 10674 10675 10676 10677 10678 10679 10680 10681 10682


    /* Stage 2. setup destination fully
     *
     * Once stage 2 has completed successfully, we *must* call finish
     * to cleanup the target whether we succeed or fail
     */
    st = virStreamNew(dconn, 0);
    if (st == NULL)
        /* virStreamNew only fails on OOM, and it reports the error itself */
        goto cleanup;

    internalret = dconn->driver->domainMigratePrepareTunnel(dconn, st,
                                                            flags, dname,
                                                            resource, dom_xml);

    if (internalret < 0)
        /* domainMigratePrepareTunnel sets the error for us */
        goto cleanup;

    /*   3. start migration on source */
10683
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
10684 10685 10686 10687 10688 10689 10690 10691
    if (flags & VIR_MIGRATE_NON_SHARED_DISK)
        background_flags |= QEMU_MONITOR_MIGRATE_NON_SHARED_DISK;
    if (flags & VIR_MIGRATE_NON_SHARED_INC)
        background_flags |= QEMU_MONITOR_MIGRATE_NON_SHARED_INC;
    if (qemuCmdFlags & QEMUD_CMD_FLAG_MIGRATE_QEMU_UNIX){
        internalret = qemuMonitorMigrateToUnix(priv->mon, background_flags,
                                               unixfile);
    }
C
Chris Lalancette 已提交
10692 10693
    else if (qemuCmdFlags & QEMUD_CMD_FLAG_MIGRATE_QEMU_EXEC) {
        const char *args[] = { "nc", "-U", unixfile, NULL };
10694
        internalret = qemuMonitorMigrateToCommand(priv->mon, QEMU_MONITOR_MIGRATE_BACKGROUND, args);
10695 10696
    } else {
        internalret = -1;
C
Chris Lalancette 已提交
10697
    }
10698
    qemuDomainObjExitMonitorWithDriver(driver, vm);
C
Chris Lalancette 已提交
10699
    if (internalret < 0) {
10700 10701
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("tunnelled migration monitor command failed"));
10702
        goto finish;
C
Chris Lalancette 已提交
10703 10704
    }

10705 10706 10707 10708 10709 10710
    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                        _("guest unexpectedly quit"));
        goto cleanup;
    }

10711 10712 10713
    /* From this point onwards we *must* call cancel to abort the
     * migration on source if anything goes wrong */

C
Chris Lalancette 已提交
10714 10715 10716
    /* it is also possible that the migrate didn't fail initially, but
     * rather failed later on.  Check the output of "info migrate"
     */
10717
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
10718 10719
    if (qemuMonitorGetMigrationStatus(priv->mon,
                                      &status,
C
Chris Lalancette 已提交
10720 10721 10722
                                      &transferred,
                                      &remaining,
                                      &total) < 0) {
10723
        qemuDomainObjExitMonitorWithDriver(driver, vm);
10724
        goto cancel;
C
Chris Lalancette 已提交
10725
    }
10726
    qemuDomainObjExitMonitorWithDriver(driver, vm);
C
Chris Lalancette 已提交
10727 10728

    if (status == QEMU_MONITOR_MIGRATION_STATUS_ERROR) {
10729 10730
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s",_("migrate failed"));
10731
        goto cancel;
C
Chris Lalancette 已提交
10732 10733 10734 10735 10736 10737
    }

    addrlen = sizeof(sa_client);
    while ((client_sock = accept(qemu_sock, (struct sockaddr *)&sa_client, &addrlen)) < 0) {
        if (errno == EAGAIN || errno == EINTR)
            continue;
10738
        virReportSystemError(errno, "%s",
C
Chris Lalancette 已提交
10739
                             _("tunnelled migration failed to accept from qemu"));
10740
        goto cancel;
C
Chris Lalancette 已提交
10741 10742
    }

10743
    retval = doTunnelSendAll(st, client_sock);
10744

10745
cancel:
10746
    if (retval != 0 && priv->mon) {
10747
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
10748
        qemuMonitorMigrateCancel(priv->mon);
10749
        qemuDomainObjExitMonitorWithDriver(driver, vm);
10750
    }
C
Chris Lalancette 已提交
10751

10752
finish:
C
Chris Lalancette 已提交
10753 10754 10755
    dname = dname ? dname : dom->name;
    ddomain = dconn->driver->domainMigrateFinish2
        (dconn, dname, NULL, 0, uri, flags, retval);
10756 10757 10758 10759 10760 10761 10762

cleanup:
    if (client_sock != -1)
        close(client_sock);
    if (qemu_sock != -1)
        close(qemu_sock);

C
Chris Lalancette 已提交
10763 10764 10765
    if (ddomain)
        virUnrefDomain(ddomain);

10766 10767 10768 10769
    if (unixfile) {
        unlink(unixfile);
        VIR_FREE(unixfile);
    }
C
Chris Lalancette 已提交
10770

10771 10772 10773
    if (st)
        /* don't call virStreamFree(), because that resets any pending errors */
        virUnrefStream(st);
10774 10775 10776 10777
    return retval;
}


10778 10779 10780 10781
/* This is essentially a simplified re-impl of
 * virDomainMigrateVersion2 from libvirt.c, but running in source
 * libvirtd context, instead of client app context */
static int doNonTunnelMigrate(virDomainPtr dom,
10782
                              struct qemud_driver *driver,
10783 10784 10785 10786 10787 10788 10789 10790 10791 10792 10793 10794 10795 10796 10797 10798 10799 10800 10801 10802 10803 10804 10805 10806 10807
                              virConnectPtr dconn,
                              virDomainObjPtr vm,
                              const char *dom_xml,
                              const char *uri ATTRIBUTE_UNUSED,
                              unsigned long flags,
                              const char *dname,
                              unsigned long resource)
{
    virDomainPtr ddomain = NULL;
    int retval = -1;
    char *uri_out = NULL;

    /* NB we don't pass 'uri' into this, since that's the libvirtd
     * URI in this context - so we let dest pick it */
    if (dconn->driver->domainMigratePrepare2(dconn,
                                             NULL, /* cookie */
                                             0, /* cookielen */
                                             NULL, /* uri */
                                             &uri_out,
                                             flags, dname,
                                             resource, dom_xml) < 0)
        /* domainMigratePrepare2 sets the error for us */
        goto cleanup;

    if (uri_out == NULL) {
10808 10809
        qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                        _("domainMigratePrepare2 did not set uri"));
10810
        goto cleanup;
10811 10812
    }

10813
    if (doNativeMigrate(driver, vm, uri_out, flags, dname, resource) < 0)
10814 10815 10816 10817 10818 10819 10820 10821 10822 10823 10824 10825 10826 10827 10828 10829 10830
        goto finish;

    retval = 0;

finish:
    dname = dname ? dname : dom->name;
    ddomain = dconn->driver->domainMigrateFinish2
        (dconn, dname, NULL, 0, uri_out, flags, retval);

    if (ddomain)
        virUnrefDomain(ddomain);

cleanup:
    return retval;
}


10831
static int doPeer2PeerMigrate(virDomainPtr dom,
10832
                              struct qemud_driver *driver,
10833 10834 10835 10836 10837 10838 10839 10840 10841 10842 10843 10844 10845 10846 10847 10848
                              virDomainObjPtr vm,
                              const char *uri,
                              unsigned long flags,
                              const char *dname,
                              unsigned long resource)
{
    int ret = -1;
    virConnectPtr dconn = NULL;
    char *dom_xml;

    /* the order of operations is important here; we make sure the
     * destination side is completely setup before we touch the source
     */

    dconn = virConnectOpen(uri);
    if (dconn == NULL) {
10849 10850
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("Failed to connect to remote libvirt URI %s"), uri);
10851 10852 10853
        return -1;
    }
    if (!VIR_DRV_SUPPORTS_FEATURE(dconn->driver, dconn,
10854
                                  VIR_DRV_FEATURE_MIGRATION_P2P)) {
10855 10856
        qemuReportError(VIR_ERR_OPERATION_FAILED, "%s",
                        _("Destination libvirt does not support peer-to-peer migration protocol"));
10857 10858 10859
        goto cleanup;
    }

10860 10861 10862
    dom_xml = qemudVMDumpXML(driver, vm,
                             VIR_DOMAIN_XML_SECURE |
                             VIR_DOMAIN_XML_UPDATE_CPU);
10863
    if (!dom_xml) {
10864 10865
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to get domain xml"));
10866 10867 10868
        goto cleanup;
    }

10869
    if (flags & VIR_MIGRATE_TUNNELLED)
10870
        ret = doTunnelMigrate(dom, driver, dconn, vm, dom_xml, uri, flags, dname, resource);
10871
    else
10872
        ret = doNonTunnelMigrate(dom, driver, dconn, vm, dom_xml, uri, flags, dname, resource);
10873 10874 10875

cleanup:
    VIR_FREE(dom_xml);
C
Chris Lalancette 已提交
10876 10877 10878
    /* don't call virConnectClose(), because that resets any pending errors */
    virUnrefConnect(dconn);

10879
    return ret;
D
Daniel Veillard 已提交
10880 10881
}

10882

D
Daniel Veillard 已提交
10883 10884 10885 10886 10887 10888
/* 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,
10889
                           unsigned long flags,
10890
                           const char *dname,
D
Daniel Veillard 已提交
10891 10892
                           unsigned long resource)
{
10893 10894
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
10895
    virDomainEventPtr event = NULL;
10896
    int ret = -1;
10897
    int resume = 0;
10898
    qemuDomainObjPrivatePtr priv;
D
Daniel Veillard 已提交
10899

C
Chris Lalancette 已提交
10900 10901 10902 10903 10904 10905 10906 10907 10908
    virCheckFlags(VIR_MIGRATE_LIVE |
                  VIR_MIGRATE_PEER2PEER |
                  VIR_MIGRATE_TUNNELLED |
                  VIR_MIGRATE_PERSIST_DEST |
                  VIR_MIGRATE_UNDEFINE_SOURCE |
                  VIR_MIGRATE_PAUSED |
                  VIR_MIGRATE_NON_SHARED_DISK |
                  VIR_MIGRATE_NON_SHARED_INC, -1);

10909
    qemuDriverLock(driver);
10910
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
D
Daniel Veillard 已提交
10911
    if (!vm) {
10912 10913
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
10914 10915
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
10916
        goto cleanup;
D
Daniel Veillard 已提交
10917
    }
10918
    priv = vm->privateData;
D
Daniel Veillard 已提交
10919

10920 10921
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;
10922
    priv->jobActive = QEMU_JOB_MIGRATION;
10923

D
Daniel P. Berrange 已提交
10924
    if (!virDomainObjIsActive(vm)) {
10925 10926
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
10927
        goto endjob;
D
Daniel Veillard 已提交
10928 10929
    }

10930 10931 10932
    memset(&priv->jobInfo, 0, sizeof(priv->jobInfo));
    priv->jobInfo.type = VIR_DOMAIN_JOB_UNBOUNDED;

10933
    resume = vm->state == VIR_DOMAIN_RUNNING;
10934
    if (!(flags & VIR_MIGRATE_LIVE) && vm->state == VIR_DOMAIN_RUNNING) {
10935
        if (qemuDomainMigrateOffline(driver, vm) < 0)
10936
            goto endjob;
10937 10938
    }

10939
    if ((flags & (VIR_MIGRATE_TUNNELLED | VIR_MIGRATE_PEER2PEER))) {
10940
        if (doPeer2PeerMigrate(dom, driver, vm, uri, flags, dname, resource) < 0)
10941
            /* doPeer2PeerMigrate already set the error, so just get out */
10942
            goto endjob;
10943
    } else {
10944
        if (doNativeMigrate(driver, vm, uri, flags, dname, resource) < 0)
10945
            goto endjob;
10946 10947
    }

D
Daniel Veillard 已提交
10948
    /* Clean up the source domain. */
10949
    qemudShutdownVMDaemon(driver, vm, 1);
10950
    resume = 0;
10951 10952 10953 10954

    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_MIGRATED);
C
Chris Lalancette 已提交
10955
    if (!vm->persistent || (flags & VIR_MIGRATE_UNDEFINE_SOURCE)) {
10956
        virDomainDeleteConfig(driver->configDir, driver->autostartDir, vm);
10957 10958
        if (qemuDomainObjEndJob(vm) > 0)
            virDomainRemoveInactive(&driver->domains, vm);
10959 10960
        vm = NULL;
    }
10961
    ret = 0;
D
Daniel Veillard 已提交
10962

10963
endjob:
10964
    if (resume && vm->state == VIR_DOMAIN_PAUSED) {
10965
        /* we got here through some sort of failure; start the domain again */
10966
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
10967
        if (qemuMonitorStartCPUs(priv->mon, dom->conn) < 0) {
10968 10969 10970 10971
            /* Hm, we already know we are in error here.  We don't want to
             * overwrite the previous error, though, so we just throw something
             * to the logs and hope for the best
             */
10972
            VIR_ERROR(_("Failed to resume guest %s after failure"),
10973
                      vm->def->name);
10974
        }
10975
        qemuDomainObjExitMonitorWithDriver(driver, vm);
10976

10977
        vm->state = VIR_DOMAIN_RUNNING;
10978 10979 10980 10981
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_RESUMED,
                                         VIR_DOMAIN_EVENT_RESUMED_MIGRATED);
    }
10982 10983 10984
    if (vm &&
        qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
10985

10986
cleanup:
10987 10988
    if (vm)
        virDomainObjUnlock(vm);
10989 10990
    if (event)
        qemuDomainEventQueue(driver, event);
10991
    qemuDriverUnlock(driver);
10992
    return ret;
D
Daniel Veillard 已提交
10993 10994 10995 10996 10997 10998 10999 11000 11001
}

/* 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,
C
Chris Lalancette 已提交
11002
                           unsigned long flags,
D
Daniel Veillard 已提交
11003 11004
                           int retcode)
{
11005 11006 11007
    struct qemud_driver *driver = dconn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
11008
    virDomainEventPtr event = NULL;
11009
    virErrorPtr orig_err;
C
Chris Lalancette 已提交
11010
    int newVM = 1;
D
Daniel Veillard 已提交
11011

C
Chris Lalancette 已提交
11012 11013 11014 11015 11016 11017 11018 11019 11020
    virCheckFlags(VIR_MIGRATE_LIVE |
                  VIR_MIGRATE_PEER2PEER |
                  VIR_MIGRATE_TUNNELLED |
                  VIR_MIGRATE_PERSIST_DEST |
                  VIR_MIGRATE_UNDEFINE_SOURCE |
                  VIR_MIGRATE_PAUSED |
                  VIR_MIGRATE_NON_SHARED_DISK |
                  VIR_MIGRATE_NON_SHARED_INC, NULL);

11021 11022 11023
    /* Migration failed. Save the current error so nothing squashes it */
    orig_err = virSaveLastError();

11024
    qemuDriverLock(driver);
11025
    vm = virDomainFindByName(&driver->domains, dname);
D
Daniel Veillard 已提交
11026
    if (!vm) {
11027 11028
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching name '%s'"), dname);
11029
        goto cleanup;
D
Daniel Veillard 已提交
11030 11031
    }

11032 11033 11034
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

D
Daniel Veillard 已提交
11035 11036 11037 11038
    /* Did the migration go as planned?  If yes, return the domain
     * object, but if no, clean up the empty qemu process.
     */
    if (retcode == 0) {
11039 11040 11041 11042 11043 11044
        if (!virDomainObjIsActive(vm)) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("guest unexpectedly quit"));
            goto cleanup;
        }

C
Chris Lalancette 已提交
11045 11046 11047 11048 11049
        if (flags & VIR_MIGRATE_PERSIST_DEST) {
            if (vm->persistent)
                newVM = 0;
            vm->persistent = 1;

11050
            if (virDomainSaveConfig(driver->configDir, vm->def) < 0) {
C
Chris Lalancette 已提交
11051 11052 11053 11054 11055 11056 11057 11058 11059 11060
                /* Hmpf.  Migration was successful, but making it persistent
                 * was not.  If we report successful, then when this domain
                 * shuts down, management tools are in for a surprise.  On the
                 * other hand, if we report failure, then the management tools
                 * might try to restart the domain on the source side, even
                 * though the domain is actually running on the destination.
                 * Return a NULL dom pointer, and hope that this is a rare
                 * situation and management tools are smart.
                 */
                vm = NULL;
11061
                goto endjob;
C
Chris Lalancette 已提交
11062 11063 11064 11065 11066 11067 11068 11069 11070
            }

            event = virDomainEventNewFromObj(vm,
                                             VIR_DOMAIN_EVENT_DEFINED,
                                             newVM ?
                                             VIR_DOMAIN_EVENT_DEFINED_ADDED :
                                             VIR_DOMAIN_EVENT_DEFINED_UPDATED);
            if (event)
                qemuDomainEventQueue(driver, event);
11071
            event = NULL;
C
Chris Lalancette 已提交
11072 11073

        }
11074
        qemuDomainObjPrivatePtr priv = vm->privateData;
D
Daniel Veillard 已提交
11075
        dom = virGetDomain (dconn, vm->def->name, vm->def->uuid);
11076

11077 11078 11079 11080 11081 11082 11083 11084
        if (!(flags & VIR_MIGRATE_PAUSED)) {
            /* run 'cont' on the destination, which allows migration on qemu
             * >= 0.10.6 to work properly.  This isn't strictly necessary on
             * older qemu's, but it also doesn't hurt anything there
             */
            qemuDomainObjEnterMonitorWithDriver(driver, vm);
            if (qemuMonitorStartCPUs(priv->mon, dconn) < 0) {
                if (virGetLastError() == NULL)
11085 11086
                    qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                    "%s", _("resume operation failed"));
11087 11088 11089
                qemuDomainObjExitMonitorWithDriver(driver, vm);
                goto endjob;
            }
11090
            qemuDomainObjExitMonitorWithDriver(driver, vm);
11091 11092

            vm->state = VIR_DOMAIN_RUNNING;
11093 11094
        }

11095 11096 11097
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_RESUMED,
                                         VIR_DOMAIN_EVENT_RESUMED_MIGRATED);
11098 11099 11100 11101 11102 11103
        if (vm->state == VIR_DOMAIN_PAUSED) {
            qemuDomainEventQueue(driver, event);
            event = virDomainEventNewFromObj(vm,
                                             VIR_DOMAIN_EVENT_SUSPENDED,
                                             VIR_DOMAIN_EVENT_SUSPENDED_PAUSED);
        }
11104 11105 11106 11107
        if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0) {
            VIR_WARN("Failed to save status on vm %s", vm->def->name);
            goto endjob;
        }
D
Daniel Veillard 已提交
11108
    } else {
11109
        qemudShutdownVMDaemon(driver, vm, 0);
11110 11111 11112
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_FAILED);
11113
        if (!vm->persistent) {
11114 11115
            if (qemuDomainObjEndJob(vm) > 0)
                virDomainRemoveInactive(&driver->domains, vm);
11116 11117
            vm = NULL;
        }
D
Daniel Veillard 已提交
11118
    }
11119

11120
endjob:
11121 11122 11123
    if (vm &&
        qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
11124

11125
cleanup:
11126 11127 11128 11129
    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }
11130 11131
    if (vm)
        virDomainObjUnlock(vm);
11132 11133
    if (event)
        qemuDomainEventQueue(driver, event);
11134
    qemuDriverUnlock(driver);
11135
    return dom;
D
Daniel Veillard 已提交
11136 11137
}

11138 11139 11140 11141 11142 11143 11144 11145 11146 11147 11148 11149 11150 11151 11152 11153
static int
qemudNodeDeviceGetPciInfo (virNodeDevicePtr dev,
                           unsigned *domain,
                           unsigned *bus,
                           unsigned *slot,
                           unsigned *function)
{
    virNodeDeviceDefPtr def = NULL;
    virNodeDevCapsDefPtr cap;
    char *xml = NULL;
    int ret = -1;

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

11154
    def = virNodeDeviceDefParseString(xml, EXISTING_DEVICE);
11155 11156 11157 11158 11159 11160 11161 11162 11163 11164 11165 11166 11167 11168 11169 11170 11171
    if (!def)
        goto out;

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

        cap = cap->next;
    }

    if (!cap) {
11172 11173
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("device %s is not a PCI device"), dev->name);
11174 11175 11176 11177 11178 11179 11180 11181 11182 11183 11184 11185 11186 11187 11188 11189 11190 11191 11192 11193
        goto out;
    }

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

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

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

11194
    pci = pciGetDevice(domain, bus, slot, function);
11195 11196 11197
    if (!pci)
        return -1;

11198
    if (pciDettachDevice(pci) < 0)
11199 11200 11201 11202
        goto out;

    ret = 0;
out:
11203
    pciFreeDevice(pci);
11204 11205 11206 11207 11208 11209 11210 11211 11212 11213 11214 11215 11216
    return ret;
}

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

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

11217
    pci = pciGetDevice(domain, bus, slot, function);
11218 11219 11220
    if (!pci)
        return -1;

11221
    if (pciReAttachDevice(pci) < 0)
11222 11223 11224 11225
        goto out;

    ret = 0;
out:
11226
    pciFreeDevice(pci);
11227 11228 11229 11230 11231 11232
    return ret;
}

static int
qemudNodeDeviceReset (virNodeDevicePtr dev)
{
11233
    struct qemud_driver *driver = dev->conn->privateData;
11234 11235 11236 11237 11238 11239 11240
    pciDevice *pci;
    unsigned domain, bus, slot, function;
    int ret = -1;

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

11241
    pci = pciGetDevice(domain, bus, slot, function);
11242 11243 11244
    if (!pci)
        return -1;

11245 11246
    qemuDriverLock(driver);

11247
    if (pciResetDevice(pci, driver->activePciHostdevs) < 0)
11248 11249 11250 11251
        goto out;

    ret = 0;
out:
11252
    qemuDriverUnlock(driver);
11253
    pciFreeDevice(pci);
11254 11255 11256
    return ret;
}

11257 11258 11259 11260 11261 11262 11263 11264 11265 11266 11267
static int
qemuCPUCompare(virConnectPtr conn,
               const char *xmlDesc,
               unsigned int flags ATTRIBUTE_UNUSED)
{
    struct qemud_driver *driver = conn->privateData;
    int ret = VIR_CPU_COMPARE_ERROR;

    qemuDriverLock(driver);

    if (!driver->caps || !driver->caps->host.cpu) {
11268 11269
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        "%s", _("cannot get host CPU capabilities"));
11270 11271
    }
    else
11272
        ret = cpuCompareXML(driver->caps->host.cpu, xmlDesc);
11273 11274 11275 11276 11277 11278

    qemuDriverUnlock(driver);

    return ret;
}

11279

11280 11281 11282 11283 11284 11285 11286 11287 11288 11289 11290 11291 11292
static char *
qemuCPUBaseline(virConnectPtr conn ATTRIBUTE_UNUSED,
                const char **xmlCPUs,
                unsigned int ncpus,
                unsigned int flags ATTRIBUTE_UNUSED)
{
    char *cpu;

    cpu = cpuBaselineXML(xmlCPUs, ncpus, NULL, 0);

    return cpu;
}

11293 11294 11295 11296 11297 11298 11299 11300 11301 11302 11303 11304 11305 11306 11307 11308 11309 11310 11311 11312 11313 11314 11315 11316 11317 11318 11319 11320 11321 11322 11323 11324 11325 11326 11327 11328 11329 11330 11331 11332 11333 11334 11335

static int qemuDomainGetJobInfo(virDomainPtr dom,
                                virDomainJobInfoPtr info) {
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
    qemuDomainObjPrivatePtr priv;

    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
    qemuDriverUnlock(driver);
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    priv = vm->privateData;

    if (virDomainObjIsActive(vm)) {
        if (priv->jobActive) {
            memcpy(info, &priv->jobInfo, sizeof(*info));
        } else {
            memset(info, 0, sizeof(*info));
            info->type = VIR_DOMAIN_JOB_NONE;
        }
    } else {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
    }

    ret = 0;

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


11336 11337 11338 11339 11340 11341 11342 11343 11344 11345 11346 11347 11348 11349 11350 11351 11352 11353 11354 11355 11356 11357
static int qemuDomainAbortJob(virDomainPtr dom) {
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
    qemuDomainObjPrivatePtr priv;

    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
    qemuDriverUnlock(driver);
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    priv = vm->privateData;

    if (virDomainObjIsActive(vm)) {
        if (priv->jobActive) {
            VIR_DEBUG("Requesting cancellation of job on vm %s", vm->def->name);
11358
            priv->jobSignals |= QEMU_JOB_SIGNAL_CANCEL;
11359 11360 11361 11362 11363 11364 11365 11366 11367 11368 11369 11370 11371 11372 11373 11374 11375 11376 11377 11378
        } else {
            qemuReportError(VIR_ERR_OPERATION_INVALID,
                            "%s", _("no job is active on the domain"));
            goto cleanup;
        }
    } else {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
    }

    ret = 0;

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


11379 11380 11381 11382 11383 11384 11385 11386 11387 11388
static int
qemuDomainMigrateSetMaxDowntime(virDomainPtr dom,
                                unsigned long long downtime,
                                unsigned int flags)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    qemuDomainObjPrivatePtr priv;
    int ret = -1;

11389
    virCheckFlags(0, -1);
11390 11391 11392 11393 11394 11395 11396 11397 11398 11399 11400 11401 11402 11403 11404 11405 11406 11407 11408 11409 11410 11411 11412 11413 11414 11415 11416 11417 11418 11419 11420 11421 11422 11423 11424 11425 11426 11427

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

    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
    }

    priv = vm->privateData;

    if (priv->jobActive != QEMU_JOB_MIGRATION) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not being migrated"));
        goto cleanup;
    }

    VIR_DEBUG("Requesting migration downtime change to %llums", downtime);
    priv->jobSignals |= QEMU_JOB_SIGNAL_MIGRATE_DOWNTIME;
    priv->jobSignalsData.migrateDowntime = downtime;
    ret = 0;

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}

C
Chris Lalancette 已提交
11428 11429 11430 11431 11432 11433 11434 11435 11436 11437 11438 11439 11440 11441 11442 11443 11444 11445 11446 11447 11448 11449 11450 11451 11452 11453 11454 11455 11456 11457 11458 11459 11460 11461 11462 11463 11464 11465 11466 11467 11468 11469 11470 11471 11472 11473 11474 11475 11476 11477 11478 11479 11480 11481 11482 11483 11484 11485 11486 11487 11488 11489 11490 11491 11492 11493 11494 11495 11496 11497 11498 11499 11500 11501 11502 11503 11504 11505 11506 11507 11508 11509 11510 11511 11512 11513 11514 11515 11516 11517 11518 11519 11520 11521 11522 11523 11524 11525 11526 11527 11528 11529 11530 11531 11532 11533 11534 11535 11536
static char *qemuFindQemuImgBinary(void)
{
    char *ret;

    ret = virFindFileInPath("kvm-img");
    if (ret == NULL)
        ret = virFindFileInPath("qemu-img");
    if (ret == NULL)
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("unable to find kvm-img or qemu-img"));

    return ret;
}

static int qemuDomainSnapshotWriteSnapshotMetadata(virDomainObjPtr vm,
                                                   char *snapshotDir)
{
    int fd = -1;
    char *newxml = NULL;
    int ret = -1;
    char *snapDir = NULL;
    char *snapFile = NULL;
    int err;
    char uuidstr[VIR_UUID_STRING_BUFLEN];

    virUUIDFormat(vm->def->uuid, uuidstr);
    newxml = virDomainSnapshotDefFormat(uuidstr, vm->current_snapshot->def, 1);
    if (newxml == NULL) {
        virReportOOMError();
        return -1;
    }

    if (virAsprintf(&snapDir, "%s/%s", snapshotDir, vm->def->name) < 0) {
        virReportOOMError();
        goto cleanup;
    }
    err = virFileMakePath(snapDir);
    if (err < 0) {
        virReportSystemError(err, _("cannot create snapshot directory '%s'"),
                             snapDir);
        goto cleanup;
    }

    if (virAsprintf(&snapFile, "%s/%s.xml", snapDir,
                    vm->current_snapshot->def->name) < 0) {
        virReportOOMError();
        goto cleanup;
    }
    fd = open(snapFile, O_CREAT|O_TRUNC|O_WRONLY, S_IRUSR|S_IWUSR);
    if (fd < 0) {
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("failed to create snapshot file '%s'"), snapFile);
        goto cleanup;
    }
    if (safewrite(fd, newxml, strlen(newxml)) != strlen(newxml)) {
        virReportSystemError(errno, _("Failed to write snapshot data to %s"),
                             snapFile);
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(snapFile);
    VIR_FREE(snapDir);
    VIR_FREE(newxml);
    if (fd != -1)
        close(fd);
    return ret;
}

static int qemuDomainSnapshotSetActive(virDomainObjPtr vm,
                                       char *snapshotDir)
{
    if (vm->current_snapshot) {
        vm->current_snapshot->def->active = 1;

        return qemuDomainSnapshotWriteSnapshotMetadata(vm, snapshotDir);
    }

    return 0;
}

static int qemuDomainSnapshotSetInactive(virDomainObjPtr vm,
                                         char *snapshotDir)
{
    if (vm->current_snapshot) {
        vm->current_snapshot->def->active = 0;

        return qemuDomainSnapshotWriteSnapshotMetadata(vm, snapshotDir);
    }

    return 0;
}


static int qemuDomainSnapshotIsAllowed(virDomainObjPtr vm)
{
    int i;

    /* FIXME: we need to figure out what else here might succeed; in
     * particular, if it's a raw device but on LVM, we could probably make
     * that succeed as well
     */
    for (i = 0; i < vm->def->ndisks; i++) {
        if (vm->def->disks[i]->device == VIR_DOMAIN_DISK_DEVICE_DISK &&
            (!vm->def->disks[i]->driverType ||
             STRNEQ(vm->def->disks[i]->driverType, "qcow2"))) {
            qemuReportError(VIR_ERR_OPERATION_INVALID,
11537 11538
                            _("Disk '%s' does not support snapshotting"),
                            vm->def->disks[i]->src);
C
Chris Lalancette 已提交
11539 11540 11541 11542 11543 11544 11545 11546 11547
            return 0;
        }
    }

    return 1;
}

static virDomainSnapshotPtr qemuDomainSnapshotCreateXML(virDomainPtr domain,
                                                        const char *xmlDesc,
11548
                                                        unsigned int flags)
C
Chris Lalancette 已提交
11549 11550 11551 11552 11553 11554 11555 11556 11557 11558
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    virDomainSnapshotObjPtr snap = NULL;
    virDomainSnapshotPtr snapshot = NULL;
    char uuidstr[VIR_UUID_STRING_BUFLEN];
    virDomainSnapshotDefPtr def;
    const char *qemuimgarg[] = { NULL, "snapshot", "-c", NULL, NULL, NULL };
    int i;

11559 11560
    virCheckFlags(0, NULL);

C
Chris Lalancette 已提交
11561 11562 11563 11564 11565 11566 11567 11568 11569 11570 11571 11572 11573 11574 11575 11576 11577 11578 11579 11580 11581 11582 11583 11584 11585 11586 11587 11588 11589 11590 11591 11592 11593 11594 11595 11596 11597 11598 11599 11600 11601 11602 11603 11604 11605 11606 11607 11608 11609 11610 11611 11612 11613 11614 11615 11616 11617 11618 11619 11620 11621
    qemuDriverLock(driver);
    virUUIDFormat(domain->uuid, uuidstr);
    vm = virDomainFindByUUID(&driver->domains, domain->uuid);
    if (!vm) {
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    /* in a perfect world, we would allow qemu to tell us this.  The problem
     * is that qemu only does this check device-by-device; so if you had a
     * domain that booted from a large qcow2 device, but had a secondary raw
     * device attached, you wouldn't find out that you can't snapshot your
     * guest until *after* it had spent the time to snapshot the boot device.
     * This is probably a bug in qemu, but we'll work around it here for now.
     */
    if (!qemuDomainSnapshotIsAllowed(vm))
        goto cleanup;

    if (!(def = virDomainSnapshotDefParseString(xmlDesc, 1)))
        goto cleanup;

    if (!(snap = virDomainSnapshotAssignDef(&vm->snapshots, def)))
        goto cleanup;

    /* actually do the snapshot */
    if (!virDomainObjIsActive(vm)) {
        qemuimgarg[0] = qemuFindQemuImgBinary();
        if (qemuimgarg[0] == NULL)
            /* qemuFindQemuImgBinary set the error */
            goto cleanup;

        qemuimgarg[3] = snap->def->name;

        for (i = 0; i < vm->def->ndisks; i++) {
            /* FIXME: we also need to handle LVM here */
            /* FIXME: if we fail halfway through this loop, we are in an
             * inconsistent state.  I'm not quite sure what to do about that
             */
            if (vm->def->disks[i]->device == VIR_DOMAIN_DISK_DEVICE_DISK) {
                if (!vm->def->disks[i]->driverType ||
                    STRNEQ(vm->def->disks[i]->driverType, "qcow2")) {
                    qemuReportError(VIR_ERR_OPERATION_INVALID,
                                    _("Disk device '%s' does not support snapshotting"),
                                    vm->def->disks[i]->info.alias);
                    goto cleanup;
                }

                qemuimgarg[4] = vm->def->disks[i]->src;

                if (virRun(qemuimgarg, NULL) < 0) {
                    virReportSystemError(errno,
                                         _("Failed to run '%s' to create snapshot '%s' from disk '%s'"),
                                         qemuimgarg[0], snap->def->name,
                                         vm->def->disks[i]->src);
                    goto cleanup;
                }
            }
        }
    }
    else {
11622 11623 11624 11625 11626
        qemuDomainObjPrivatePtr priv;
        int ret;

        if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
            goto cleanup;
C
Chris Lalancette 已提交
11627 11628
        priv = vm->privateData;
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
11629
        ret = qemuMonitorCreateSnapshot(priv->mon, def->name);
C
Chris Lalancette 已提交
11630
        qemuDomainObjExitMonitorWithDriver(driver, vm);
11631
        if (qemuDomainObjEndJob(vm) == 0) {
11632
            vm = NULL;
11633 11634
            goto cleanup;
        }
11635 11636
        if (ret < 0)
            goto cleanup;
C
Chris Lalancette 已提交
11637 11638 11639 11640 11641 11642 11643 11644 11645 11646 11647 11648 11649 11650 11651 11652 11653 11654 11655 11656 11657 11658 11659 11660 11661 11662 11663 11664 11665 11666 11667 11668 11669 11670 11671 11672
    }

    snap->def->state = vm->state;

    /* FIXME: if we fail after this point, there's not a whole lot we can
     * do; we've successfully taken the snapshot, and we are now running
     * on it, so we have to go forward the best we can
     */

    if (vm->current_snapshot) {
        def->parent = strdup(vm->current_snapshot->def->name);
        if (def->parent == NULL) {
            virReportOOMError();
            goto cleanup;
        }
    }

    /* Now we set the new current_snapshot for the domain */
    vm->current_snapshot = snap;

    if (qemuDomainSnapshotWriteSnapshotMetadata(vm, driver->snapshotDir) < 0)
        /* qemuDomainSnapshotWriteSnapshotMetadata set the error */
        goto cleanup;

    snapshot = virGetDomainSnapshot(domain, snap->def->name);

cleanup:
    VIR_FREE(qemuimgarg[0]);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return snapshot;
}

static int qemuDomainSnapshotListNames(virDomainPtr domain, char **names,
                                       int nameslen,
11673
                                       unsigned int flags)
C
Chris Lalancette 已提交
11674 11675 11676 11677 11678
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    int n = -1;

11679 11680
    virCheckFlags(0, -1);

C
Chris Lalancette 已提交
11681 11682 11683 11684 11685 11686 11687 11688 11689 11690 11691 11692 11693 11694 11695 11696 11697 11698 11699 11700
    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, domain->uuid);
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(domain->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    n = virDomainSnapshotObjListGetNames(&vm->snapshots, names, nameslen);

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return n;
}

static int qemuDomainSnapshotNum(virDomainPtr domain,
11701
                                 unsigned int flags)
C
Chris Lalancette 已提交
11702 11703 11704 11705 11706
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    int n = -1;

11707 11708
    virCheckFlags(0, -1);

C
Chris Lalancette 已提交
11709 11710 11711 11712 11713 11714 11715 11716 11717 11718 11719 11720 11721 11722 11723 11724 11725 11726 11727 11728 11729
    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, domain->uuid);
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(domain->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    n = virDomainSnapshotObjListNum(&vm->snapshots);

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return n;
}

static virDomainSnapshotPtr qemuDomainSnapshotLookupByName(virDomainPtr domain,
                                                           const char *name,
11730
                                                           unsigned int flags)
C
Chris Lalancette 已提交
11731 11732 11733 11734 11735 11736
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm;
    virDomainSnapshotObjPtr snap = NULL;
    virDomainSnapshotPtr snapshot = NULL;

11737 11738
    virCheckFlags(0, NULL);

C
Chris Lalancette 已提交
11739 11740 11741 11742 11743 11744 11745 11746 11747 11748 11749 11750 11751 11752 11753 11754 11755 11756 11757 11758 11759 11760 11761 11762 11763 11764 11765
    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, domain->uuid);
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(domain->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    snap = virDomainSnapshotFindByName(&vm->snapshots, name);
    if (!snap) {
        qemuReportError(VIR_ERR_NO_DOMAIN_SNAPSHOT,
                        _("no snapshot with matching name '%s'"), name);
        goto cleanup;
    }

    snapshot = virGetDomainSnapshot(domain, snap->def->name);

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return snapshot;
}

static int qemuDomainHasCurrentSnapshot(virDomainPtr domain,
11766
                                        unsigned int flags)
C
Chris Lalancette 已提交
11767 11768 11769 11770 11771
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;

11772 11773
    virCheckFlags(0, -1);

C
Chris Lalancette 已提交
11774 11775 11776 11777 11778 11779 11780 11781 11782 11783 11784 11785 11786 11787 11788 11789 11790 11791 11792 11793
    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, domain->uuid);
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(domain->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    ret = (vm->current_snapshot != NULL);

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}

static virDomainSnapshotPtr qemuDomainSnapshotCurrent(virDomainPtr domain,
11794
                                                      unsigned int flags)
C
Chris Lalancette 已提交
11795 11796 11797 11798 11799
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm;
    virDomainSnapshotPtr snapshot = NULL;

11800 11801
    virCheckFlags(0, NULL);

C
Chris Lalancette 已提交
11802 11803 11804 11805 11806 11807 11808 11809 11810 11811 11812 11813 11814 11815 11816 11817 11818 11819 11820 11821 11822 11823 11824 11825 11826 11827
    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, domain->uuid);
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(domain->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    if (!vm->current_snapshot) {
        qemuReportError(VIR_ERR_NO_DOMAIN_SNAPSHOT, "%s",
                        _("the domain does not have a current snapshot"));
        goto cleanup;
    }

    snapshot = virGetDomainSnapshot(domain, vm->current_snapshot->def->name);

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return snapshot;
}

static char *qemuDomainSnapshotDumpXML(virDomainSnapshotPtr snapshot,
11828
                                       unsigned int flags)
C
Chris Lalancette 已提交
11829 11830 11831 11832 11833 11834 11835
{
    struct qemud_driver *driver = snapshot->domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    char *xml = NULL;
    virDomainSnapshotObjPtr snap = NULL;
    char uuidstr[VIR_UUID_STRING_BUFLEN];

11836 11837
    virCheckFlags(0, NULL);

C
Chris Lalancette 已提交
11838 11839 11840 11841 11842 11843 11844 11845 11846 11847 11848 11849 11850 11851 11852 11853 11854 11855 11856 11857 11858 11859 11860 11861 11862 11863 11864
    qemuDriverLock(driver);
    virUUIDFormat(snapshot->domain->uuid, uuidstr);
    vm = virDomainFindByUUID(&driver->domains, snapshot->domain->uuid);
    if (!vm) {
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    snap = virDomainSnapshotFindByName(&vm->snapshots, snapshot->name);
    if (!snap) {
        qemuReportError(VIR_ERR_NO_DOMAIN_SNAPSHOT,
                        _("no domain snapshot with matching name '%s'"),
                        snapshot->name);
        goto cleanup;
    }

    xml = virDomainSnapshotDefFormat(uuidstr, snap->def, 0);

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return xml;
}

static int qemuDomainRevertToSnapshot(virDomainSnapshotPtr snapshot,
11865
                                      unsigned int flags)
C
Chris Lalancette 已提交
11866 11867 11868 11869 11870 11871 11872 11873 11874 11875
{
    struct qemud_driver *driver = snapshot->domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    int ret = -1;
    virDomainSnapshotObjPtr snap = NULL;
    char uuidstr[VIR_UUID_STRING_BUFLEN];
    virDomainEventPtr event = NULL;
    qemuDomainObjPrivatePtr priv;
    int rc;

11876 11877
    virCheckFlags(0, -1);

C
Chris Lalancette 已提交
11878 11879 11880 11881 11882 11883 11884 11885 11886 11887 11888 11889 11890 11891 11892 11893 11894 11895 11896 11897 11898 11899 11900 11901 11902 11903 11904 11905 11906 11907 11908
    qemuDriverLock(driver);
    virUUIDFormat(snapshot->domain->uuid, uuidstr);
    vm = virDomainFindByUUID(&driver->domains, snapshot->domain->uuid);
    if (!vm) {
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    snap = virDomainSnapshotFindByName(&vm->snapshots, snapshot->name);
    if (!snap) {
        qemuReportError(VIR_ERR_NO_DOMAIN_SNAPSHOT,
                        _("no domain snapshot with matching name '%s'"),
                        snapshot->name);
        goto cleanup;
    }

    vm->current_snapshot = snap;

    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

    if (snap->def->state == VIR_DOMAIN_RUNNING
        || snap->def->state == VIR_DOMAIN_PAUSED) {

        if (virDomainObjIsActive(vm)) {
            priv = vm->privateData;
            qemuDomainObjEnterMonitorWithDriver(driver, vm);
            rc = qemuMonitorLoadSnapshot(priv->mon, snap->def->name);
            qemuDomainObjExitMonitorWithDriver(driver, vm);
            if (rc < 0)
11909
                goto endjob;
C
Chris Lalancette 已提交
11910 11911 11912
        }
        else {
            if (qemuDomainSnapshotSetActive(vm, driver->snapshotDir) < 0)
11913
                goto endjob;
C
Chris Lalancette 已提交
11914 11915

            rc = qemudStartVMDaemon(snapshot->domain->conn, driver, vm, NULL,
11916
                                    false, -1, NULL);
C
Chris Lalancette 已提交
11917
            if (qemuDomainSnapshotSetInactive(vm, driver->snapshotDir) < 0)
11918
                goto endjob;
C
Chris Lalancette 已提交
11919
            if (rc < 0)
11920
                goto endjob;
C
Chris Lalancette 已提交
11921 11922 11923 11924 11925 11926
        }

        if (snap->def->state == VIR_DOMAIN_PAUSED) {
            /* qemu unconditionally starts the domain running again after
             * loadvm, so let's pause it to keep consistency
             */
11927
            int state = vm->state;
C
Chris Lalancette 已提交
11928
            priv = vm->privateData;
11929
            vm->state = VIR_DOMAIN_PAUSED;
C
Chris Lalancette 已提交
11930 11931 11932
            qemuDomainObjEnterMonitorWithDriver(driver, vm);
            rc = qemuMonitorStopCPUs(priv->mon);
            qemuDomainObjExitMonitorWithDriver(driver, vm);
11933 11934
            if (rc < 0) {
                vm->state = state;
11935
                goto endjob;
11936
            }
C
Chris Lalancette 已提交
11937 11938 11939 11940 11941 11942 11943 11944 11945 11946 11947 11948 11949 11950 11951 11952 11953 11954
        }

        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_STARTED,
                                         VIR_DOMAIN_EVENT_STARTED_FROM_SNAPSHOT);
    }
    else {
        /* qemu is a little funny with running guests and the restoration
         * of snapshots.  If the snapshot was taken online,
         * then after a "loadvm" monitor command, the VM is set running
         * again.  If the snapshot was taken offline, then after a "loadvm"
         * monitor command the VM is left paused.  Unpausing it leads to
         * the memory state *before* the loadvm with the disk *after* the
         * loadvm, which obviously is bound to corrupt something.
         * Therefore we destroy the domain and set it to "off" in this case.
         */

        if (virDomainObjIsActive(vm)) {
11955
            qemudShutdownVMDaemon(driver, vm, 0);
C
Chris Lalancette 已提交
11956 11957 11958
            event = virDomainEventNewFromObj(vm,
                                             VIR_DOMAIN_EVENT_STOPPED,
                                             VIR_DOMAIN_EVENT_STOPPED_FROM_SNAPSHOT);
11959 11960 11961 11962
            if (!vm->persistent) {
                if (qemuDomainObjEndJob(vm) > 0)
                    virDomainRemoveInactive(&driver->domains, vm);
                vm = NULL;
11963
                goto cleanup;
11964
            }
C
Chris Lalancette 已提交
11965 11966 11967
        }

        if (qemuDomainSnapshotSetActive(vm, driver->snapshotDir) < 0)
11968
            goto endjob;
C
Chris Lalancette 已提交
11969 11970 11971 11972 11973 11974
    }

    vm->state = snap->def->state;

    ret = 0;

11975
endjob:
C
Chris Lalancette 已提交
11976 11977 11978
    if (vm && qemuDomainObjEndJob(vm) == 0)
        vm = NULL;

11979
cleanup:
C
Chris Lalancette 已提交
11980 11981 11982 11983 11984 11985 11986 11987 11988 11989 11990 11991 11992 11993 11994 11995 11996 11997 11998 11999 12000 12001 12002 12003 12004 12005 12006 12007 12008 12009 12010 12011 12012 12013 12014 12015 12016 12017 12018 12019 12020 12021 12022 12023 12024 12025 12026 12027 12028 12029 12030 12031 12032 12033 12034 12035 12036 12037 12038 12039 12040 12041 12042 12043 12044 12045 12046 12047 12048 12049 12050 12051 12052 12053 12054 12055 12056 12057 12058 12059 12060 12061 12062 12063 12064 12065 12066 12067 12068 12069 12070 12071 12072 12073 12074 12075 12076 12077 12078 12079 12080 12081 12082 12083 12084 12085 12086 12087 12088 12089 12090 12091 12092 12093 12094 12095 12096 12097 12098 12099 12100 12101 12102 12103 12104 12105 12106 12107 12108 12109 12110 12111 12112 12113
    if (event)
        qemuDomainEventQueue(driver, event);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);

    return ret;
}

static int qemuDomainSnapshotDiscard(struct qemud_driver *driver,
                                     virDomainObjPtr vm,
                                     virDomainSnapshotObjPtr snap)
{
    const char *qemuimgarg[] = { NULL, "snapshot", "-d", NULL, NULL, NULL };
    char *snapFile = NULL;
    int ret = -1;
    int i;
    qemuDomainObjPrivatePtr priv;
    virDomainSnapshotObjPtr parentsnap;

    if (!virDomainObjIsActive(vm)) {
        qemuimgarg[0] = qemuFindQemuImgBinary();
        if (qemuimgarg[0] == NULL)
            /* qemuFindQemuImgBinary set the error */
            goto cleanup;

        qemuimgarg[3] = snap->def->name;

        for (i = 0; i < vm->def->ndisks; i++) {
            /* FIXME: we also need to handle LVM here */
            if (vm->def->disks[i]->device == VIR_DOMAIN_DISK_DEVICE_DISK) {
                if (!vm->def->disks[i]->driverType ||
                    STRNEQ(vm->def->disks[i]->driverType, "qcow2")) {
                    /* we continue on even in the face of error, since other
                     * disks in this VM may have this snapshot in place
                     */
                    continue;
                }

                qemuimgarg[4] = vm->def->disks[i]->src;

                if (virRun(qemuimgarg, NULL) < 0) {
                    /* we continue on even in the face of error, since other
                     * disks in this VM may have this snapshot in place
                     */
                    continue;
                }
            }
        }
    }
    else {
        priv = vm->privateData;
        qemuDomainObjEnterMonitorWithDriver(driver, vm);
        /* we continue on even in the face of error */
        qemuMonitorDeleteSnapshot(priv->mon, snap->def->name);
        qemuDomainObjExitMonitorWithDriver(driver, vm);
    }

    if (snap == vm->current_snapshot) {
        if (snap->def->parent) {
            parentsnap = virDomainSnapshotFindByName(&vm->snapshots,
                                                     snap->def->parent);
            if (!parentsnap) {
                qemuReportError(VIR_ERR_NO_DOMAIN_SNAPSHOT,
                                _("no domain snapshot parent with matching name '%s'"),
                                snap->def->parent);
                goto cleanup;
            }

            /* Now we set the new current_snapshot for the domain */
            vm->current_snapshot = parentsnap;
        }
        else
            vm->current_snapshot = NULL;
    }

    if (virAsprintf(&snapFile, "%s/%s/%s.xml", driver->snapshotDir,
                    vm->def->name, snap->def->name) < 0) {
        virReportOOMError();
        goto cleanup;
    }
    unlink(snapFile);

    virDomainSnapshotObjListRemove(&vm->snapshots, snap);

    ret = 0;

cleanup:
    VIR_FREE(snapFile);
    VIR_FREE(qemuimgarg[0]);

    return ret;
}

struct snap_remove {
    struct qemud_driver *driver;
    virDomainObjPtr vm;
    char *parent;
    int err;
};

static void qemuDomainSnapshotDiscardChildren(void *payload,
                                              const char *name ATTRIBUTE_UNUSED,
                                              void *data)
{
    virDomainSnapshotObjPtr snap = payload;
    struct snap_remove *curr = data;
    struct snap_remove this;

    if (snap->def->parent && STREQ(snap->def->parent, curr->parent)) {
        this.driver = curr->driver;
        this.vm = curr->vm;
        this.parent = snap->def->name;
        this.err = 0;
        virHashForEach(curr->vm->snapshots.objs,
                       qemuDomainSnapshotDiscardChildren, &this);

        if (this.err)
            curr->err = this.err;
        else
            this.err = qemuDomainSnapshotDiscard(curr->driver, curr->vm, snap);
    }
}

static int qemuDomainSnapshotDelete(virDomainSnapshotPtr snapshot,
                                    unsigned int flags)
{
    struct qemud_driver *driver = snapshot->domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    int ret = -1;
    virDomainSnapshotObjPtr snap = NULL;
    char uuidstr[VIR_UUID_STRING_BUFLEN];
    struct snap_remove rem;

12114 12115
    virCheckFlags(VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN, -1);

C
Chris Lalancette 已提交
12116 12117 12118 12119 12120 12121 12122 12123 12124 12125 12126 12127 12128 12129 12130 12131 12132
    qemuDriverLock(driver);
    virUUIDFormat(snapshot->domain->uuid, uuidstr);
    vm = virDomainFindByUUID(&driver->domains, snapshot->domain->uuid);
    if (!vm) {
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    snap = virDomainSnapshotFindByName(&vm->snapshots, snapshot->name);
    if (!snap) {
        qemuReportError(VIR_ERR_NO_DOMAIN_SNAPSHOT,
                        _("no domain snapshot with matching name '%s'"),
                        snapshot->name);
        goto cleanup;
    }

12133 12134 12135
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

C
Chris Lalancette 已提交
12136 12137 12138 12139 12140 12141 12142 12143
    if (flags & VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN) {
        rem.driver = driver;
        rem.vm = vm;
        rem.parent = snap->def->name;
        rem.err = 0;
        virHashForEach(vm->snapshots.objs, qemuDomainSnapshotDiscardChildren,
                       &rem);
        if (rem.err < 0)
12144
            goto endjob;
C
Chris Lalancette 已提交
12145 12146 12147 12148
    }

    ret = qemuDomainSnapshotDiscard(driver, vm, snap);

12149 12150 12151 12152
endjob:
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;

C
Chris Lalancette 已提交
12153 12154 12155 12156 12157 12158
cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}
12159

12160 12161 12162 12163 12164
static virDriver qemuDriver = {
    VIR_DRV_QEMU,
    "QEMU",
    qemudOpen, /* open */
    qemudClose, /* close */
D
Daniel Veillard 已提交
12165
    qemudSupportsFeature, /* supports_feature */
12166 12167
    qemudGetType, /* type */
    qemudGetVersion, /* version */
12168
    NULL, /* libvirtVersion (impl. in libvirt.c) */
12169
    virGetHostname, /* getHostname */
12170
    qemudGetMaxVCPUs, /* getMaxVcpus */
12171
    nodeGetInfo, /* nodeGetInfo */
12172 12173 12174
    qemudGetCapabilities, /* getCapabilities */
    qemudListDomains, /* listDomains */
    qemudNumDomains, /* numOfDomains */
12175
    qemudDomainCreate, /* domainCreateXML */
12176 12177 12178 12179 12180
    qemudDomainLookupByID, /* domainLookupByID */
    qemudDomainLookupByUUID, /* domainLookupByUUID */
    qemudDomainLookupByName, /* domainLookupByName */
    qemudDomainSuspend, /* domainSuspend */
    qemudDomainResume, /* domainResume */
12181
    qemudDomainShutdown, /* domainShutdown */
12182 12183 12184
    NULL, /* domainReboot */
    qemudDomainDestroy, /* domainDestroy */
    qemudDomainGetOSType, /* domainGetOSType */
12185
    qemudDomainGetMaxMemory, /* domainGetMaxMemory */
12186
    NULL, /* domainSetMaxMemory */
12187
    qemudDomainSetMemory, /* domainSetMemory */
12188 12189 12190
    qemudDomainGetInfo, /* domainGetInfo */
    qemudDomainSave, /* domainSave */
    qemudDomainRestore, /* domainRestore */
P
Paolo Bonzini 已提交
12191
    qemudDomainCoreDump, /* domainCoreDump */
12192
    qemudDomainSetVcpus, /* domainSetVcpus */
12193 12194
    qemudDomainPinVcpu, /* domainPinVcpu */
    qemudDomainGetVcpus, /* domainGetVcpus */
12195
    qemudDomainGetMaxVcpus, /* domainGetMaxVcpus */
12196 12197
    qemudDomainGetSecurityLabel, /* domainGetSecurityLabel */
    qemudNodeGetSecurityModel, /* nodeGetSecurityModel */
12198
    qemudDomainDumpXML, /* domainDumpXML */
12199
    qemuDomainXMLFromNative, /* domainXmlFromNative */
12200
    qemuDomainXMLToNative, /* domainXMLToNative */
12201 12202
    qemudListDefinedDomains, /* listDefinedDomains */
    qemudNumDefinedDomains, /* numOfDefinedDomains */
12203
    qemudDomainStart, /* domainCreate */
12204
    qemudDomainStartWithFlags, /* domainCreateWithFlags */
12205 12206
    qemudDomainDefine, /* domainDefineXML */
    qemudDomainUndefine, /* domainUndefine */
12207
    qemudDomainAttachDevice, /* domainAttachDevice */
12208
    qemudDomainAttachDeviceFlags, /* domainAttachDeviceFlags */
12209
    qemudDomainDetachDevice, /* domainDetachDevice */
12210
    qemudDomainDetachDeviceFlags, /* domainDetachDeviceFlags */
12211
    qemuDomainUpdateDeviceFlags, /* domainUpdateDeviceFlags */
12212 12213
    qemudDomainGetAutostart, /* domainGetAutostart */
    qemudDomainSetAutostart, /* domainSetAutostart */
12214 12215 12216
    qemuGetSchedulerType, /* domainGetSchedulerType */
    qemuGetSchedulerParameters, /* domainGetSchedulerParameters */
    qemuSetSchedulerParameters, /* domainSetSchedulerParameters */
D
Daniel Veillard 已提交
12217 12218
    NULL, /* domainMigratePrepare (v1) */
    qemudDomainMigratePerform, /* domainMigratePerform */
12219
    NULL, /* domainMigrateFinish */
12220
    qemudDomainBlockStats, /* domainBlockStats */
12221
    qemudDomainInterfaceStats, /* domainInterfaceStats */
12222
    qemudDomainMemoryStats, /* domainMemoryStats */
12223
    qemudDomainBlockPeek, /* domainBlockPeek */
R
Richard W.M. Jones 已提交
12224
    qemudDomainMemoryPeek, /* domainMemoryPeek */
12225
    qemuDomainGetBlockInfo, /* domainGetBlockInfo */
12226 12227
    nodeGetCellsFreeMemory, /* nodeGetCellsFreeMemory */
    nodeGetFreeMemory,  /* getFreeMemory */
12228 12229
    qemuDomainEventRegister, /* domainEventRegister */
    qemuDomainEventDeregister, /* domainEventDeregister */
D
Daniel Veillard 已提交
12230 12231
    qemudDomainMigratePrepare2, /* domainMigratePrepare2 */
    qemudDomainMigrateFinish2, /* domainMigrateFinish2 */
12232 12233 12234
    qemudNodeDeviceDettach, /* nodeDeviceDettach */
    qemudNodeDeviceReAttach, /* nodeDeviceReAttach */
    qemudNodeDeviceReset, /* nodeDeviceReset */
C
Chris Lalancette 已提交
12235
    qemudDomainMigratePrepareTunnel, /* domainMigratePrepareTunnel */
12236 12237 12238 12239
    qemuIsEncrypted, /* isEncrypted */
    qemuIsSecure, /* isSecure */
    qemuDomainIsActive, /* domainIsActive */
    qemuDomainIsPersistent, /* domainIsPersistent */
12240
    qemuCPUCompare, /* cpuCompare */
12241
    qemuCPUBaseline, /* cpuBaseline */
12242
    qemuDomainGetJobInfo, /* domainGetJobInfo */
12243
    qemuDomainAbortJob, /* domainAbortJob */
12244
    qemuDomainMigrateSetMaxDowntime, /* domainMigrateSetMaxDowntime */
12245 12246
    qemuDomainEventRegisterAny, /* domainEventRegisterAny */
    qemuDomainEventDeregisterAny, /* domainEventDeregisterAny */
12247 12248 12249
    qemuDomainManagedSave, /* domainManagedSave */
    qemuDomainHasManagedSaveImage, /* domainHasManagedSaveImage */
    qemuDomainManagedSaveRemove, /* domainManagedSaveRemove */
C
Chris Lalancette 已提交
12250 12251 12252 12253 12254 12255 12256 12257 12258
    qemuDomainSnapshotCreateXML, /* domainSnapshotCreateXML */
    qemuDomainSnapshotDumpXML, /* domainSnapshotDumpXML */
    qemuDomainSnapshotNum, /* domainSnapshotNum */
    qemuDomainSnapshotListNames, /* domainSnapshotListNames */
    qemuDomainSnapshotLookupByName, /* domainSnapshotLookupByName */
    qemuDomainHasCurrentSnapshot, /* domainHasCurrentSnapshot */
    qemuDomainSnapshotCurrent, /* domainSnapshotCurrent */
    qemuDomainRevertToSnapshot, /* domainRevertToSnapshot */
    qemuDomainSnapshotDelete, /* domainSnapshotDelete */
12259 12260 12261
};


12262
static virStateDriver qemuStateDriver = {
12263
    .name = "QEMU",
12264 12265 12266 12267
    .initialize = qemudStartup,
    .cleanup = qemudShutdown,
    .reload = qemudReload,
    .active = qemudActive,
12268
};
12269

S
Stefan Berger 已提交
12270
static int
12271
qemudVMFilterRebuild(virConnectPtr conn ATTRIBUTE_UNUSED,
S
Stefan Berger 已提交
12272 12273
                     virHashIterator iter, void *data)
{
12274 12275 12276
    struct qemud_driver *driver = qemu_driver;

    qemuDriverLock(driver);
S
Stefan Berger 已提交
12277
    virHashForEach(qemu_driver->domains.objs, iter, data);
12278 12279
    qemuDriverUnlock(driver);

S
Stefan Berger 已提交
12280 12281 12282 12283 12284 12285 12286 12287 12288
    return 0;
}


static virNWFilterCallbackDriver qemuCallbackDriver = {
    .name = "QEMU",
    .vmFilterRebuild = qemudVMFilterRebuild,
};

12289
int qemuRegister(void) {
12290 12291
    virRegisterDriver(&qemuDriver);
    virRegisterStateDriver(&qemuStateDriver);
S
Stefan Berger 已提交
12292
    virNWFilterRegisterCallbackDriver(&qemuCallbackDriver);
12293 12294
    return 0;
}