qemu_driver.c 209.4 KB
Newer Older
D
Daniel P. Berrange 已提交
1 2 3
/*
 * driver.c: core driver methods for managing qemu guests
 *
4
 * Copyright (C) 2006-2011 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 32 33 34 35 36
#include <dirent.h>
#include <limits.h>
#include <string.h>
#include <stdio.h>
#include <stdarg.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
37
#include <sys/utsname.h>
38 39 40 41
#include <sys/stat.h>
#include <fcntl.h>
#include <signal.h>
#include <paths.h>
42
#include <stdio.h>
43
#include <sys/wait.h>
44
#include <sys/ioctl.h>
45
#include <sys/un.h>
D
Daniel P. Berrange 已提交
46

47

48 49
#include "qemu_driver.h"
#include "qemu_conf.h"
50
#include "qemu_capabilities.h"
51
#include "qemu_command.h"
52
#include "qemu_cgroup.h"
53
#include "qemu_hostdev.h"
54
#include "qemu_hotplug.h"
55
#include "qemu_monitor.h"
56
#include "qemu_bridge_filter.h"
57
#include "qemu_audit.h"
58
#include "qemu_process.h"
59
#include "qemu_migration.h"
60 61 62 63

#include "virterror_internal.h"
#include "logging.h"
#include "datatypes.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"
C
Chris Lalancette 已提交
77
#include "libvirt_internal.h"
78
#include "xml.h"
79
#include "cpu/cpu.h"
80
#include "macvtap.h"
81
#include "sysinfo.h"
82
#include "domain_nwfilter.h"
83
#include "hooks.h"
84
#include "storage_file.h"
85
#include "files.h"
86
#include "fdstream.h"
87
#include "configmake.h"
H
Hu Tao 已提交
88
#include "threadpool.h"
89

90 91
#define VIR_FROM_THIS VIR_FROM_QEMU

92 93
#define QEMU_NB_MEM_PARAM  3

94 95 96
#if HAVE_LINUX_KVM_H
# include <linux/kvm.h>
#endif
97

98 99
/* device for kvm ioctls */
#define KVM_DEVICE "/dev/kvm"
100

101 102 103 104 105 106 107 108 109 110 111
/* add definitions missing in older linux/kvm.h */
#ifndef KVMIO
# define KVMIO 0xAE
#endif
#ifndef KVM_CHECK_EXTENSION
# define KVM_CHECK_EXTENSION       _IO(KVMIO,   0x03)
#endif
#ifndef KVM_CAP_NR_VCPUS
# define KVM_CAP_NR_VCPUS 9       /* returns max vcpus per vm */
#endif

112
#define QEMU_NB_BLKIO_PARAM  1
113 114

#define timeval_to_ms(tv)       (((tv).tv_sec * 1000ull) + ((tv).tv_usec / 1000))
H
Hu Tao 已提交
115 116 117

static void processWatchdogEvent(void *data, void *opaque);

118 119
static int qemudShutdown(void);

J
Jiri Denemark 已提交
120 121
static int qemudDomainObjStart(virConnectPtr conn,
                               struct qemud_driver *driver,
122 123
                               virDomainObjPtr vm,
                               bool start_paused);
J
Jiri Denemark 已提交
124

125
static int qemudDomainGetMaxVcpus(virDomainPtr dom);
126

127
struct qemud_driver *qemu_driver = NULL;
128 129


130 131 132 133
struct qemuAutostartData {
    struct qemud_driver *driver;
    virConnectPtr conn;
};
134

135
static void
136
qemuAutostartDomain(void *payload, const void *name ATTRIBUTE_UNUSED, void *opaque)
137 138 139
{
    virDomainObjPtr vm = payload;
    struct qemuAutostartData *data = opaque;
140
    virErrorPtr err;
141 142

    virDomainObjLock(vm);
143 144 145 146 147 148 149 150 151
    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) &&
152
            qemudDomainObjStart(data->conn, data->driver, vm, false) < 0) {
153
            err = virGetLastError();
154
            VIR_ERROR(_("Failed to autostart VM '%s': %s"),
155
                      vm->def->name,
156
                      err ? err->message : _("unknown error"));
157
        }
158 159 160

        if (qemuDomainObjEndJob(vm) == 0)
            vm = NULL;
161
    }
162 163 164

    if (vm)
        virDomainObjUnlock(vm);
165 166
}

167

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

182
    qemuDriverLock(driver);
183
    virHashForEach(driver->domains.objs, qemuAutostartDomain, &data);
184
    qemuDriverUnlock(driver);
185

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

190
static int
191
qemuSecurityInit(struct qemud_driver *driver)
192
{
193 194 195 196
    virSecurityManagerPtr mgr = virSecurityManagerNew(driver->securityDriverName,
                                                      driver->allowDiskFormatProbing);
    if (!mgr)
        goto error;
D
Daniel Veillard 已提交
197

198 199 200 201 202 203 204
    if (driver->privileged) {
        virSecurityManagerPtr dac = virSecurityManagerNewDAC(driver->user,
                                                             driver->group,
                                                             driver->allowDiskFormatProbing,
                                                             driver->dynamicOwnership);
        if (!dac)
            goto error;
205

206
        if (!(driver->securityManager = virSecurityManagerNewStack(mgr,
E
Eric Blake 已提交
207 208 209
                                                                   dac))) {

            virSecurityManagerFree(dac);
210
            goto error;
E
Eric Blake 已提交
211
        }
212 213 214
    } else {
        driver->securityManager = mgr;
    }
D
Daniel Veillard 已提交
215

216
    return 0;
217

218 219 220 221 222
error:
    VIR_ERROR0(_("Failed to initialize security drivers"));
    virSecurityManagerFree(mgr);
    return -1;
}
223

224

225 226 227 228 229
static virCapsPtr
qemuCreateCapabilities(virCapsPtr oldcaps,
                       struct qemud_driver *driver)
{
    virCapsPtr caps;
230

231 232 233 234
    /* Basic host arch / guest machine capabilities */
    if (!(caps = qemuCapsInit(oldcaps))) {
        virReportOOMError();
        return NULL;
235 236
    }

237 238 239 240 241 242
    if (driver->allowDiskFormatProbing) {
        caps->defaultDiskDriverName = NULL;
        caps->defaultDiskDriverType = NULL;
    } else {
        caps->defaultDiskDriverName = "qemu";
        caps->defaultDiskDriverType = "raw";
243 244
    }

245 246
    qemuDomainSetPrivateDataHooks(caps);
    qemuDomainSetNamespaceHooks(caps);
247

248 249 250 251
    if (virGetHostUUID(caps->host.host_uuid)) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot get the host uuid"));
        goto err_exit;
252
    }
253

254 255
    /* Security driver data */
    const char *doi, *model;
256

257 258 259 260 261 262 263
    doi = virSecurityManagerGetDOI(driver->securityManager);
    model = virSecurityManagerGetModel(driver->securityManager);
    if (STRNEQ(model, "none")) {
        if (!(caps->host.secModel.model = strdup(model)))
            goto no_memory;
        if (!(caps->host.secModel.doi = strdup(doi)))
            goto no_memory;
264
    }
265

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

269
    return caps;
270

271 272 273 274
no_memory:
    virReportOOMError();
err_exit:
    virCapabilitiesFree(caps);
275 276 277
    return NULL;
}

278
static void qemuDomainSnapshotLoad(void *payload,
279
                                   const void *name ATTRIBUTE_UNUSED,
280
                                   void *data)
281
{
282 283 284 285 286 287 288 289 290 291
    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];
292

293 294 295 296
    virDomainObjLock(vm);
    if (virAsprintf(&snapDir, "%s/%s", baseDir, vm->def->name) < 0) {
        VIR_ERROR(_("Failed to allocate memory for snapshot directory for domain %s"),
                   vm->def->name);
297
        goto cleanup;
298 299
    }

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

303 304 305 306 307
    if (!(dir = opendir(snapDir))) {
        if (errno != ENOENT)
            VIR_ERROR(_("Failed to open snapshot directory %s for domain %s: %s"),
                      snapDir, vm->def->name,
                      virStrerror(errno, ebuf, sizeof(ebuf)));
308
        goto cleanup;
309 310
    }

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

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

319 320 321 322
        if (virAsprintf(&fullpath, "%s/%s", snapDir, entry->d_name) < 0) {
            VIR_ERROR0(_("Failed to allocate memory for path"));
            continue;
        }
323

324 325 326 327 328 329 330 331
        ret = virFileReadAll(fullpath, 1024*1024*1, &xmlStr);
        if (ret < 0) {
            /* Nothing we can do here, skip this one */
            VIR_ERROR(_("Failed to read snapshot file %s: %s"), fullpath,
                      virStrerror(errno, ebuf, sizeof(ebuf)));
            VIR_FREE(fullpath);
            continue;
        }
332

333 334 335 336 337 338 339 340
        def = virDomainSnapshotDefParseString(xmlStr, 0);
        if (def == NULL) {
            /* Nothing we can do here, skip this one */
            VIR_ERROR(_("Failed to parse snapshot XML from file '%s'"), fullpath);
            VIR_FREE(fullpath);
            VIR_FREE(xmlStr);
            continue;
        }
341

342
        virDomainSnapshotAssignDef(&vm->snapshots, def);
343

344 345
        VIR_FREE(fullpath);
        VIR_FREE(xmlStr);
346 347
    }

348 349 350 351 352 353 354 355
    /* 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.
     */
356

357
    virResetLastError();
358

359 360 361 362
cleanup:
    if (dir)
        closedir(dir);
    VIR_FREE(snapDir);
363 364 365
    virDomainObjUnlock(vm);
}

366 367 368 369 370
/**
 * qemudStartup:
 *
 * Initialization function for the QEmu daemon
 */
371
static int
372 373 374 375 376
qemudStartup(int privileged) {
    char *base = NULL;
    char *driverConf = NULL;
    int rc;
    virConnectPtr conn = NULL;
377

378 379
    if (VIR_ALLOC(qemu_driver) < 0)
        return -1;
380

381 382 383 384
    if (virMutexInit(&qemu_driver->lock) < 0) {
        VIR_ERROR0(_("cannot initialize mutex"));
        VIR_FREE(qemu_driver);
        return -1;
385
    }
386 387
    qemuDriverLock(qemu_driver);
    qemu_driver->privileged = privileged;
388

389 390
    /* Don't have a dom0 so start from 1 */
    qemu_driver->nextvmid = 1;
391

392 393
    if (virDomainObjListInit(&qemu_driver->domains) < 0)
        goto out_of_memory;
394

395 396 397 398 399
    /* Init callback list */
    if (VIR_ALLOC(qemu_driver->domainEventCallbacks) < 0)
        goto out_of_memory;
    if (!(qemu_driver->domainEventQueue = virDomainEventQueueNew()))
        goto out_of_memory;
400

401 402 403
    if ((qemu_driver->domainEventTimer =
         virEventAddTimeout(-1, qemuDomainEventFlush, qemu_driver, NULL)) < 0)
        goto error;
404

405 406 407 408
    /* Allocate bitmap for vnc port reservation */
    if ((qemu_driver->reservedVNCPorts =
         virBitmapAlloc(QEMU_VNC_PORT_MAX - QEMU_VNC_PORT_MIN)) == NULL)
        goto out_of_memory;
409

410 411 412
    /* read the host sysinfo */
    if (privileged)
        qemu_driver->hostsysinfo = virSysinfoRead();
413

414 415 416 417
    if (privileged) {
        if (virAsprintf(&qemu_driver->logDir,
                        "%s/log/libvirt/qemu", LOCALSTATEDIR) == -1)
            goto out_of_memory;
418

419 420
        if ((base = strdup (SYSCONFDIR "/libvirt")) == NULL)
            goto out_of_memory;
421

422 423 424
        if (virAsprintf(&qemu_driver->stateDir,
                      "%s/run/libvirt/qemu", LOCALSTATEDIR) == -1)
            goto out_of_memory;
425

426 427 428
        if (virAsprintf(&qemu_driver->libDir,
                      "%s/lib/libvirt/qemu", LOCALSTATEDIR) == -1)
            goto out_of_memory;
429

430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446
        if (virAsprintf(&qemu_driver->cacheDir,
                      "%s/cache/libvirt/qemu", LOCALSTATEDIR) == -1)
            goto out_of_memory;
        if (virAsprintf(&qemu_driver->saveDir,
                      "%s/lib/libvirt/qemu/save", LOCALSTATEDIR) == -1)
            goto out_of_memory;
        if (virAsprintf(&qemu_driver->snapshotDir,
                        "%s/lib/libvirt/qemu/snapshot", LOCALSTATEDIR) == -1)
            goto out_of_memory;
        if (virAsprintf(&qemu_driver->autoDumpPath,
                        "%s/lib/libvirt/qemu/dump", LOCALSTATEDIR) == -1)
            goto out_of_memory;
    } else {
        uid_t uid = geteuid();
        char *userdir = virGetUserDirectory(uid);
        if (!userdir)
            goto error;
447

448 449 450 451 452
        if (virAsprintf(&qemu_driver->logDir,
                        "%s/.libvirt/qemu/log", userdir) == -1) {
            VIR_FREE(userdir);
            goto out_of_memory;
        }
453

454 455 456 457 458
        if (virAsprintf(&base, "%s/.libvirt", userdir) == -1) {
            VIR_FREE(userdir);
            goto out_of_memory;
        }
        VIR_FREE(userdir);
459

460 461 462 463 464 465 466 467 468 469 470 471
        if (virAsprintf(&qemu_driver->stateDir, "%s/qemu/run", base) == -1)
            goto out_of_memory;
        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;
        if (virAsprintf(&qemu_driver->saveDir, "%s/qemu/save", base) == -1)
            goto out_of_memory;
        if (virAsprintf(&qemu_driver->snapshotDir, "%s/qemu/snapshot", base) == -1)
            goto out_of_memory;
        if (virAsprintf(&qemu_driver->autoDumpPath, "%s/qemu/dump", base) == -1)
            goto out_of_memory;
472
    }
H
Hu Tao 已提交
473

474 475 476 477 478
    if (virFileMakePath(qemu_driver->stateDir) != 0) {
        char ebuf[1024];
        VIR_ERROR(_("Failed to create state dir '%s': %s"),
                  qemu_driver->stateDir, virStrerror(errno, ebuf, sizeof ebuf));
        goto error;
H
Hu Tao 已提交
479
    }
480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508
    if (virFileMakePath(qemu_driver->libDir) != 0) {
        char ebuf[1024];
        VIR_ERROR(_("Failed to create lib dir '%s': %s"),
                  qemu_driver->libDir, virStrerror(errno, ebuf, sizeof ebuf));
        goto error;
    }
    if (virFileMakePath(qemu_driver->cacheDir) != 0) {
        char ebuf[1024];
        VIR_ERROR(_("Failed to create cache dir '%s': %s"),
                  qemu_driver->cacheDir, virStrerror(errno, ebuf, sizeof ebuf));
        goto error;
    }
    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;
    }
    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;
    }
    if (virFileMakePath(qemu_driver->autoDumpPath) != 0) {
        char ebuf[1024];
        VIR_ERROR(_("Failed to create dump dir '%s': %s"),
                  qemu_driver->autoDumpPath, virStrerror(errno, ebuf, sizeof ebuf));
        goto error;
509 510
    }

511 512 513 514 515 516 517
    /* Configuration paths are either ~/.libvirt/qemu/... (session) or
     * /etc/libvirt/qemu/... (system).
     */
    if (virAsprintf(&driverConf, "%s/qemu.conf", base) < 0 ||
        virAsprintf(&qemu_driver->configDir, "%s/qemu", base) < 0 ||
        virAsprintf(&qemu_driver->autostartDir, "%s/qemu/autostart", base) < 0)
        goto out_of_memory;
518

519
    VIR_FREE(base);
520

521 522 523 524 525
    rc = virCgroupForDriver("qemu", &qemu_driver->cgroup, privileged, 1);
    if (rc < 0) {
        char buf[1024];
        VIR_INFO("Unable to create cgroup for driver: %s",
                 virStrerror(-rc, buf, sizeof(buf)));
526 527
    }

528 529 530 531
    if (qemudLoadDriverConfig(qemu_driver, driverConf) < 0) {
        goto error;
    }
    VIR_FREE(driverConf);
532

533 534
    if (qemuSecurityInit(qemu_driver) < 0)
        goto error;
535

536 537 538
    if ((qemu_driver->caps = qemuCreateCapabilities(NULL,
                                                    qemu_driver)) == NULL)
        goto error;
539

540
    if ((qemu_driver->activePciHostdevs = pciDeviceListNew()) == NULL)
541
        goto error;
542

543 544 545 546 547 548
    if (privileged) {
        if (chown(qemu_driver->libDir, qemu_driver->user, qemu_driver->group) < 0) {
            virReportSystemError(errno,
                                 _("unable to set ownership of '%s' to user %d:%d"),
                                 qemu_driver->libDir, qemu_driver->user, qemu_driver->group);
            goto error;
549
        }
550
        if (chown(qemu_driver->cacheDir, qemu_driver->user, qemu_driver->group) < 0) {
551
            virReportSystemError(errno,
552 553 554
                                 _("unable to set ownership of '%s' to %d:%d"),
                                 qemu_driver->cacheDir, qemu_driver->user, qemu_driver->group);
            goto error;
555
        }
556 557 558 559 560 561 562 563 564 565 566
        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;
        }
        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;
567
        }
568
    }
569

570 571 572 573 574 575
    /* 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
576
     */
577 578 579 580 581
    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;
582

583 584 585 586 587
        if ((rc = virFileMakePath(mempath)) != 0) {
            virReportSystemError(rc,
                                 _("unable to create hugepage path %s"), mempath);
            VIR_FREE(mempath);
            goto error;
588
        }
589 590 591 592 593 594 595
        if (qemu_driver->privileged &&
            chown(mempath, qemu_driver->user, qemu_driver->group) < 0) {
            virReportSystemError(errno,
                                 _("unable to set ownership on %s to %d:%d"),
                                 mempath, qemu_driver->user, qemu_driver->group);
            VIR_FREE(mempath);
            goto error;
G
Guido Günther 已提交
596
        }
E
Eric Blake 已提交
597

598
        qemu_driver->hugepage_path = mempath;
599
    }
600

601 602 603 604 605 606 607
    /* Get all the running persistent or transient configs first */
    if (virDomainLoadAllConfigs(qemu_driver->caps,
                                &qemu_driver->domains,
                                qemu_driver->stateDir,
                                NULL,
                                1, NULL, NULL) < 0)
        goto error;
608

609 610 611
    conn = virConnectOpen(qemu_driver->privileged ?
                          "qemu:///system" :
                          "qemu:///session");
612

613
    qemuProcessReconnectAll(conn, qemu_driver);
614

615 616 617 618 619 620 621
    /* Then inactive persistent configs */
    if (virDomainLoadAllConfigs(qemu_driver->caps,
                                &qemu_driver->domains,
                                qemu_driver->configDir,
                                qemu_driver->autostartDir,
                                0, NULL, NULL) < 0)
        goto error;
622

623

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

627
    qemuDriverUnlock(qemu_driver);
628

629
    qemuAutostartDomains(qemu_driver);
630

631 632 633
    qemu_driver->workerPool = virThreadPoolNew(0, 1, processWatchdogEvent, qemu_driver);
    if (!qemu_driver->workerPool)
        goto error;
634

635 636
    if (conn)
        virConnectClose(conn);
637

638
    return 0;
639

640 641 642 643 644 645 646 647 648 649
out_of_memory:
    virReportOOMError();
error:
    if (qemu_driver)
        qemuDriverUnlock(qemu_driver);
    if (conn)
        virConnectClose(conn);
    VIR_FREE(base);
    VIR_FREE(driverConf);
    qemudShutdown();
650
    return -1;
651 652
}

653 654 655
static void qemudNotifyLoadDomain(virDomainObjPtr vm, int newVM, void *opaque)
{
    struct qemud_driver *driver = opaque;
656

657 658 659 660 661 662 663
    if (newVM) {
        virDomainEventPtr event =
            virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_DEFINED,
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED);
        if (event)
            qemuDomainEventQueue(driver, event);
E
Eric Blake 已提交
664
    }
665
}
E
Eric Blake 已提交
666

667 668 669 670 671 672 673 674 675 676
/**
 * qemudReload:
 *
 * Function to restart the QEmu daemon, it will recheck the configuration
 * files and update its state and the networking
 */
static int
qemudReload(void) {
    if (!qemu_driver)
        return 0;
677

678 679 680 681 682 683 684
    qemuDriverLock(qemu_driver);
    virDomainLoadAllConfigs(qemu_driver->caps,
                            &qemu_driver->domains,
                            qemu_driver->configDir,
                            qemu_driver->autostartDir,
                            0, qemudNotifyLoadDomain, qemu_driver);
    qemuDriverUnlock(qemu_driver);
685

686
    qemuAutostartDomains(qemu_driver);
687

688 689
    return 0;
}
S
Stefan Berger 已提交
690

691 692 693 694 695 696 697 698 699 700 701
/**
 * 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) {
    int active = 0;
702

703 704
    if (!qemu_driver)
        return 0;
705

706 707 708 709 710 711
    /* XXX having to iterate here is not great because it requires many locks */
    qemuDriverLock(qemu_driver);
    active = virDomainObjListNumOfDomains(&qemu_driver->domains, 1);
    qemuDriverUnlock(qemu_driver);
    return active;
}
712

713 714 715 716 717 718 719 720
/**
 * qemudShutdown:
 *
 * Shutdown the QEmu daemon, it will stop all active domains and networks
 */
static int
qemudShutdown(void) {
    int i;
721

722 723
    if (!qemu_driver)
        return -1;
724

725 726 727
    qemuDriverLock(qemu_driver);
    pciDeviceListFree(qemu_driver->activePciHostdevs);
    virCapabilitiesFree(qemu_driver->caps);
728

729 730
    virDomainObjListDeinit(&qemu_driver->domains);
    virBitmapFree(qemu_driver->reservedVNCPorts);
731

732
    virSysinfoDefFree(qemu_driver->hostsysinfo);
733

734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753
    VIR_FREE(qemu_driver->configDir);
    VIR_FREE(qemu_driver->autostartDir);
    VIR_FREE(qemu_driver->logDir);
    VIR_FREE(qemu_driver->stateDir);
    VIR_FREE(qemu_driver->libDir);
    VIR_FREE(qemu_driver->cacheDir);
    VIR_FREE(qemu_driver->saveDir);
    VIR_FREE(qemu_driver->snapshotDir);
    VIR_FREE(qemu_driver->autoDumpPath);
    VIR_FREE(qemu_driver->vncTLSx509certdir);
    VIR_FREE(qemu_driver->vncListen);
    VIR_FREE(qemu_driver->vncPassword);
    VIR_FREE(qemu_driver->vncSASLdir);
    VIR_FREE(qemu_driver->spiceTLSx509certdir);
    VIR_FREE(qemu_driver->spiceListen);
    VIR_FREE(qemu_driver->spicePassword);
    VIR_FREE(qemu_driver->hugetlbfs_mount);
    VIR_FREE(qemu_driver->hugepage_path);
    VIR_FREE(qemu_driver->saveImageFormat);
    VIR_FREE(qemu_driver->dumpImageFormat);
754

755
    virSecurityManagerFree(qemu_driver->securityManager);
756

757
    ebtablesContextFree(qemu_driver->ebtables);
758

759 760 761 762
    if (qemu_driver->cgroupDeviceACL) {
        for (i = 0 ; qemu_driver->cgroupDeviceACL[i] != NULL ; i++)
            VIR_FREE(qemu_driver->cgroupDeviceACL[i]);
        VIR_FREE(qemu_driver->cgroupDeviceACL);
S
Stefan Berger 已提交
763 764
    }

765 766 767
    /* Free domain callback list */
    virDomainEventCallbackListFree(qemu_driver->domainEventCallbacks);
    virDomainEventQueueFree(qemu_driver->domainEventQueue);
768

769 770
    if (qemu_driver->domainEventTimer != -1)
        virEventRemoveTimeout(qemu_driver->domainEventTimer);
D
Daniel P. Berrange 已提交
771

772 773
    if (qemu_driver->brctl)
        brShutdown(qemu_driver->brctl);
774

775
    virCgroupFree(&qemu_driver->cgroup);
776

777 778 779 780
    qemuDriverUnlock(qemu_driver);
    virMutexDestroy(&qemu_driver->lock);
    virThreadPoolFree(qemu_driver->workerPool);
    VIR_FREE(qemu_driver);
781

782
    return 0;
783 784
}

785 786 787 788 789 790 791

static int qemuDomainSnapshotSetCurrentActive(virDomainObjPtr vm,
                                              char *snapshotDir);
static int qemuDomainSnapshotSetCurrentInactive(virDomainObjPtr vm,
                                                char *snapshotDir);


792
static virDrvOpenStatus qemudOpen(virConnectPtr conn,
793
                                  virConnectAuthPtr auth ATTRIBUTE_UNUSED,
794
                                  int flags ATTRIBUTE_UNUSED) {
795
    if (conn->uri == NULL) {
796 797 798
        if (qemu_driver == NULL)
            return VIR_DRV_OPEN_DECLINED;

799
        conn->uri = xmlParseURI(qemu_driver->privileged ?
800 801
                                "qemu:///system" :
                                "qemu:///session");
802
        if (!conn->uri) {
803
            virReportOOMError();
804 805
            return VIR_DRV_OPEN_ERROR;
        }
806 807 808 809 810 811 812 813 814 815
    } 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;

816
        if (qemu_driver == NULL) {
817 818
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("qemu state driver is not active"));
819 820 821
            return VIR_DRV_OPEN_ERROR;
        }

822
        if (conn->uri->path == NULL) {
823 824 825 826 827
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("no QEMU URI path given, try %s"),
                            qemu_driver->privileged
                            ? "qemu:///system"
                            : "qemu:///session");
828 829 830
                return VIR_DRV_OPEN_ERROR;
        }

831
        if (qemu_driver->privileged) {
832 833
            if (STRNEQ (conn->uri->path, "/system") &&
                STRNEQ (conn->uri->path, "/session")) {
834 835 836
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                _("unexpected QEMU URI path '%s', try qemu:///system"),
                                conn->uri->path);
837 838 839 840
                return VIR_DRV_OPEN_ERROR;
            }
        } else {
            if (STRNEQ (conn->uri->path, "/session")) {
841 842 843
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                _("unexpected QEMU URI path '%s', try qemu:///session"),
                                conn->uri->path);
844 845 846
                return VIR_DRV_OPEN_ERROR;
            }
        }
847 848 849 850 851 852 853
    }
    conn->privateData = qemu_driver;

    return VIR_DRV_OPEN_SUCCESS;
}

static int qemudClose(virConnectPtr conn) {
854
    struct qemud_driver *driver = conn->privateData;
855 856

    /* Get rid of callbacks registered for this conn */
857
    qemuDriverLock(driver);
858
    virDomainEventCallbackListRemoveConn(conn, driver->domainEventCallbacks);
859
    qemuDriverUnlock(driver);
860 861 862 863 864 865

    conn->privateData = NULL;

    return 0;
}

D
Daniel Veillard 已提交
866 867 868 869 870
/* Which features are supported by this driver? */
static int
qemudSupportsFeature (virConnectPtr conn ATTRIBUTE_UNUSED, int feature)
{
    switch (feature) {
871 872 873 874 875
    case VIR_DRV_FEATURE_MIGRATION_V2:
    case VIR_DRV_FEATURE_MIGRATION_P2P:
        return 1;
    default:
        return 0;
D
Daniel Veillard 已提交
876 877 878
    }
}

879
static const char *qemudGetType(virConnectPtr conn ATTRIBUTE_UNUSED) {
880
    return "QEMU";
881 882
}

883

884 885 886 887 888 889 890 891 892 893 894 895 896
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;
}


897 898 899 900
static int kvmGetMaxVCPUs(void) {
    int maxvcpus = 1;

    int r, fd;
901

902 903
    fd = open(KVM_DEVICE, O_RDONLY);
    if (fd < 0) {
904
        virReportSystemError(errno, _("Unable to open %s"), KVM_DEVICE);
905
        return -1;
906 907 908 909 910 911
    }

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

912
    VIR_FORCE_CLOSE(fd);
913 914 915 916
    return maxvcpus;
}


E
Eric Blake 已提交
917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932
static char *
qemuGetSysinfo(virConnectPtr conn, unsigned int flags)
{
    struct qemud_driver *driver = conn->privateData;

    virCheckFlags(0, NULL);

    if (!driver->hostsysinfo) {
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                        _("Host SMBIOS information is not available"));
        return NULL;
    }

    return virSysinfoFormat(driver->hostsysinfo, "");
}

933
static int qemudGetMaxVCPUs(virConnectPtr conn ATTRIBUTE_UNUSED, const char *type) {
934 935 936
    if (!type)
        return 16;

937
    if (STRCASEEQ(type, "qemu"))
938 939
        return 16;

940
    if (STRCASEEQ(type, "kvm"))
941
        return kvmGetMaxVCPUs();
942

943
    if (STRCASEEQ(type, "kqemu"))
944
        return 1;
945

946 947
    qemuReportError(VIR_ERR_INVALID_ARG,
                    _("unknown type '%s'"), type);
948 949 950
    return -1;
}

951

952
static char *qemudGetCapabilities(virConnectPtr conn) {
953
    struct qemud_driver *driver = conn->privateData;
954
    virCapsPtr caps = NULL;
955
    char *xml = NULL;
956

957
    qemuDriverLock(driver);
958

959
    if ((caps = qemuCreateCapabilities(qemu_driver->caps,
960
                                       qemu_driver)) == NULL) {
961 962 963
        virCapabilitiesFree(caps);
        goto cleanup;
    }
964

965
    virCapabilitiesFree(qemu_driver->caps);
966 967 968
    qemu_driver->caps = caps;

    if ((xml = virCapabilitiesFormatXML(driver->caps)) == NULL)
969
        virReportOOMError();
970 971

cleanup:
972
    qemuDriverUnlock(driver);
973

974
    return xml;
975 976 977
}


978 979 980 981 982
static int
qemudGetProcessInfo(unsigned long long *cpuTime, int *lastCpu, int pid,
                    int tid)
{
    char *proc;
D
Daniel P. Berrange 已提交
983
    FILE *pidinfo;
984
    unsigned long long usertime, systime;
985 986
    int cpu;
    int ret;
D
Daniel P. Berrange 已提交
987

988
    if (tid)
989
        ret = virAsprintf(&proc, "/proc/%d/task/%d/stat", pid, tid);
990
    else
991 992
        ret = virAsprintf(&proc, "/proc/%d/stat", pid);
    if (ret < 0)
D
Daniel P. Berrange 已提交
993 994 995 996
        return -1;

    if (!(pidinfo = fopen(proc, "r"))) {
        /* VM probably shut down, so fake 0 */
997 998 999 1000
        if (cpuTime)
            *cpuTime = 0;
        if (lastCpu)
            *lastCpu = 0;
1001
        VIR_FREE(proc);
D
Daniel P. Berrange 已提交
1002 1003
        return 0;
    }
1004
    VIR_FREE(proc);
D
Daniel P. Berrange 已提交
1005

1006 1007 1008 1009 1010 1011 1012 1013 1014 1015
    /* 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) {
1016
        VIR_FORCE_FCLOSE(pidinfo);
1017 1018
        VIR_WARN0("cannot parse process status data");
        errno = -EINVAL;
D
Daniel P. Berrange 已提交
1019 1020 1021 1022 1023 1024 1025 1026
        return -1;
    }

    /* We got jiffies
     * We want nanoseconds
     * _SC_CLK_TCK is jiffies per second
     * So calulate thus....
     */
1027 1028 1029 1030 1031
    if (cpuTime)
        *cpuTime = 1000ull * 1000ull * 1000ull * (usertime + systime) / (unsigned long long)sysconf(_SC_CLK_TCK);
    if (lastCpu)
        *lastCpu = cpu;

D
Daniel P. Berrange 已提交
1032

1033 1034
    VIR_DEBUG("Got status for %d/%d user=%llu sys=%llu cpu=%d",
              pid, tid, usertime, systime, cpu);
D
Daniel P. Berrange 已提交
1035

1036
    VIR_FORCE_FCLOSE(pidinfo);
D
Daniel P. Berrange 已提交
1037 1038 1039 1040 1041

    return 0;
}


1042
static virDomainPtr qemudDomainLookupByID(virConnectPtr conn,
1043
                                          int id) {
1044 1045 1046 1047
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;

1048
    qemuDriverLock(driver);
1049
    vm  = virDomainFindByID(&driver->domains, id);
1050
    qemuDriverUnlock(driver);
1051 1052

    if (!vm) {
1053 1054
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching id %d"), id);
1055
        goto cleanup;
1056 1057
    }

1058
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1059
    if (dom) dom->id = vm->def->id;
1060 1061

cleanup:
1062 1063
    if (vm)
        virDomainObjUnlock(vm);
1064 1065
    return dom;
}
1066

1067
static virDomainPtr qemudDomainLookupByUUID(virConnectPtr conn,
1068
                                            const unsigned char *uuid) {
1069 1070 1071
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
1072

1073
    qemuDriverLock(driver);
1074
    vm = virDomainFindByUUID(&driver->domains, uuid);
1075 1076
    qemuDriverUnlock(driver);

1077
    if (!vm) {
1078 1079
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(uuid, uuidstr);
1080 1081
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1082
        goto cleanup;
1083 1084
    }

1085
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1086
    if (dom) dom->id = vm->def->id;
1087 1088

cleanup:
1089 1090
    if (vm)
        virDomainObjUnlock(vm);
1091 1092
    return dom;
}
1093

1094
static virDomainPtr qemudDomainLookupByName(virConnectPtr conn,
1095
                                            const char *name) {
1096 1097 1098
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
1099

1100
    qemuDriverLock(driver);
1101
    vm = virDomainFindByName(&driver->domains, name);
1102 1103
    qemuDriverUnlock(driver);

1104
    if (!vm) {
1105 1106
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching name '%s'"), name);
1107
        goto cleanup;
1108 1109
    }

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

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

1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129

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) {
E
Eric Blake 已提交
1130 1131 1132 1133
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153
        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) {
E
Eric Blake 已提交
1154 1155 1156 1157
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1158 1159 1160 1161 1162 1163 1164 1165 1166 1167
        goto cleanup;
    }
    ret = obj->persistent;

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

1168 1169 1170 1171 1172 1173 1174 1175 1176 1177
static int qemuDomainIsUpdated(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) {
E
Eric Blake 已提交
1178 1179 1180 1181
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1182 1183 1184 1185 1186 1187 1188 1189 1190
        goto cleanup;
    }
    ret = obj->updated;

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

1192
static int qemudGetVersion(virConnectPtr conn, unsigned long *version) {
1193 1194 1195
    struct qemud_driver *driver = conn->privateData;
    int ret = -1;

1196
    qemuDriverLock(driver);
1197
    if (qemuCapsExtractVersion(driver->caps, &driver->qemuVersion) < 0)
1198
        goto cleanup;
1199

1200
    *version = driver->qemuVersion;
1201 1202 1203
    ret = 0;

cleanup:
1204
    qemuDriverUnlock(driver);
1205
    return ret;
D
Daniel P. Berrange 已提交
1206 1207
}

1208
static int qemudListDomains(virConnectPtr conn, int *ids, int nids) {
1209
    struct qemud_driver *driver = conn->privateData;
1210
    int n;
1211

1212
    qemuDriverLock(driver);
1213
    n = virDomainObjListGetActiveIDs(&driver->domains, ids, nids);
1214
    qemuDriverUnlock(driver);
1215

1216
    return n;
D
Daniel P. Berrange 已提交
1217
}
1218

1219
static int qemudNumDomains(virConnectPtr conn) {
1220
    struct qemud_driver *driver = conn->privateData;
1221
    int n;
1222

1223
    qemuDriverLock(driver);
1224
    n = virDomainObjListNumOfDomains(&driver->domains, 1);
1225
    qemuDriverUnlock(driver);
1226

1227
    return n;
D
Daniel P. Berrange 已提交
1228
}
1229

1230
static virDomainPtr qemudDomainCreate(virConnectPtr conn, const char *xml,
1231
                                      unsigned int flags) {
1232
    struct qemud_driver *driver = conn->privateData;
1233
    virDomainDefPtr def;
1234
    virDomainObjPtr vm = NULL;
1235
    virDomainPtr dom = NULL;
1236
    virDomainEventPtr event = NULL;
D
Daniel P. Berrange 已提交
1237

1238
    virCheckFlags(VIR_DOMAIN_START_PAUSED, NULL);
1239

1240
    qemuDriverLock(driver);
1241
    if (!(def = virDomainDefParseString(driver->caps, xml,
1242
                                        VIR_DOMAIN_XML_INACTIVE)))
1243
        goto cleanup;
1244

1245
    if (virSecurityManagerVerify(driver->securityManager, def) < 0)
1246 1247
        goto cleanup;

1248 1249
    if (virDomainObjIsDuplicate(&driver->domains, def, 1) < 0)
        goto cleanup;
1250

1251 1252 1253
    if (qemudCanonicalizeMachine(driver, def) < 0)
        goto cleanup;

1254
    if (qemuDomainAssignPCIAddresses(def) < 0)
1255 1256
        goto cleanup;

1257
    if (!(vm = virDomainAssignDef(driver->caps,
1258
                                  &driver->domains,
1259
                                  def, false)))
1260 1261 1262
        goto cleanup;

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

1264 1265 1266
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup; /* XXXX free the 'vm' we created ? */

1267 1268 1269
    if (qemuProcessStart(conn, driver, vm, NULL,
                         (flags & VIR_DOMAIN_START_PAUSED) != 0,
                         -1, NULL, VIR_VM_OP_CREATE) < 0) {
1270
        qemuAuditDomainStart(vm, "booted", false);
1271 1272 1273
        if (qemuDomainObjEndJob(vm) > 0)
            virDomainRemoveInactive(&driver->domains,
                                    vm);
1274
        vm = NULL;
1275
        goto cleanup;
D
Daniel P. Berrange 已提交
1276
    }
1277 1278 1279 1280

    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_BOOTED);
1281
    qemuAuditDomainStart(vm, "booted", true);
D
Daniel P. Berrange 已提交
1282

1283
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1284
    if (dom) dom->id = vm->def->id;
1285

1286 1287 1288
    if (vm &&
        qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
1289

1290 1291
cleanup:
    virDomainDefFree(def);
1292 1293
    if (vm)
        virDomainObjUnlock(vm);
1294 1295
    if (event)
        qemuDomainEventQueue(driver, event);
1296
    qemuDriverUnlock(driver);
1297
    return dom;
D
Daniel P. Berrange 已提交
1298 1299 1300
}


1301
static int qemudDomainSuspend(virDomainPtr dom) {
1302 1303 1304
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1305
    virDomainEventPtr event = NULL;
1306
    qemuDomainObjPrivatePtr priv;
1307

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

D
Daniel P. Berrange 已提交
1311
    if (!vm) {
1312 1313
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1314 1315
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1316
        goto cleanup;
D
Daniel P. Berrange 已提交
1317
    }
D
Daniel P. Berrange 已提交
1318
    if (!virDomainObjIsActive(vm)) {
1319 1320
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
1321
        goto cleanup;
D
Daniel P. Berrange 已提交
1322
    }
1323 1324 1325

    priv = vm->privateData;

1326
    if (priv->jobActive == QEMU_JOB_MIGRATION_OUT) {
1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340
        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"));
1341
            goto endjob;
1342
        }
1343
        if (vm->state != VIR_DOMAIN_PAUSED) {
1344
            if (qemuProcessStopCPUs(driver, vm) < 0) {
1345
                goto endjob;
1346
            }
1347 1348 1349 1350 1351 1352 1353
            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 已提交
1354
    }
1355

1356
endjob:
1357 1358
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
1359

1360
cleanup:
1361 1362
    if (vm)
        virDomainObjUnlock(vm);
1363

1364
    if (event)
1365
        qemuDomainEventQueue(driver, event);
1366
    qemuDriverUnlock(driver);
1367
    return ret;
D
Daniel P. Berrange 已提交
1368 1369 1370
}


1371
static int qemudDomainResume(virDomainPtr dom) {
1372 1373 1374
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1375
    virDomainEventPtr event = NULL;
1376

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

D
Daniel P. Berrange 已提交
1380
    if (!vm) {
1381 1382
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1383 1384
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1385
        goto cleanup;
D
Daniel P. Berrange 已提交
1386
    }
1387 1388 1389 1390

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

D
Daniel P. Berrange 已提交
1391
    if (!virDomainObjIsActive(vm)) {
1392 1393
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
1394
        goto endjob;
D
Daniel P. Berrange 已提交
1395
    }
1396
    if (vm->state == VIR_DOMAIN_PAUSED) {
1397
        if (qemuProcessStartCPUs(driver, vm, dom->conn) < 0) {
1398
            if (virGetLastError() == NULL)
1399 1400
                qemuReportError(VIR_ERR_OPERATION_FAILED,
                                "%s", _("resume operation failed"));
1401
            goto endjob;
1402
        }
1403 1404 1405
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_RESUMED,
                                         VIR_DOMAIN_EVENT_RESUMED_UNPAUSED);
D
Daniel P. Berrange 已提交
1406
    }
1407
    if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
1408
        goto endjob;
1409 1410
    ret = 0;

1411
endjob:
1412 1413
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
1414

1415
cleanup:
1416 1417
    if (vm)
        virDomainObjUnlock(vm);
1418
    if (event)
1419
        qemuDomainEventQueue(driver, event);
1420
    qemuDriverUnlock(driver);
1421
    return ret;
D
Daniel P. Berrange 已提交
1422 1423 1424
}


1425
static int qemudDomainShutdown(virDomainPtr dom) {
1426 1427 1428
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1429
    qemuDomainObjPrivatePtr priv;
1430

1431
    qemuDriverLock(driver);
1432
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1433 1434
    qemuDriverUnlock(driver);

1435
    if (!vm) {
1436 1437
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1438 1439
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1440
        goto cleanup;
1441 1442
    }

1443 1444 1445
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
1446
    if (!virDomainObjIsActive(vm)) {
1447 1448
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
1449
        goto endjob;
1450 1451
    }

1452
    priv = vm->privateData;
1453 1454 1455
    qemuDomainObjEnterMonitor(vm);
    ret = qemuMonitorSystemPowerdown(priv->mon);
    qemuDomainObjExitMonitor(vm);
1456

1457
endjob:
1458 1459
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
1460

1461
cleanup:
1462 1463
    if (vm)
        virDomainObjUnlock(vm);
1464
    return ret;
1465 1466 1467
}


1468
static int qemudDomainDestroy(virDomainPtr dom) {
1469 1470 1471
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1472
    virDomainEventPtr event = NULL;
1473

1474
    qemuDriverLock(driver);
1475
    vm  = virDomainFindByUUID(&driver->domains, dom->uuid);
D
Daniel P. Berrange 已提交
1476
    if (!vm) {
1477 1478
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1479 1480
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1481
        goto cleanup;
D
Daniel P. Berrange 已提交
1482
    }
1483 1484 1485 1486

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

D
Daniel P. Berrange 已提交
1487
    if (!virDomainObjIsActive(vm)) {
1488 1489
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
1490
        goto endjob;
1491
    }
1492

1493
    qemuProcessStop(driver, vm, 0);
1494 1495 1496
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_DESTROYED);
1497
    qemuAuditDomainStop(vm, "destroyed");
1498

1499
    if (!vm->persistent) {
1500 1501 1502
        if (qemuDomainObjEndJob(vm) > 0)
            virDomainRemoveInactive(&driver->domains,
                                    vm);
1503 1504
        vm = NULL;
    }
1505 1506
    ret = 0;

1507
endjob:
1508 1509 1510
    if (vm &&
        qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
1511

1512
cleanup:
1513 1514
    if (vm)
        virDomainObjUnlock(vm);
1515 1516
    if (event)
        qemuDomainEventQueue(driver, event);
1517
    qemuDriverUnlock(driver);
1518
    return ret;
D
Daniel P. Berrange 已提交
1519 1520 1521
}


1522
static char *qemudDomainGetOSType(virDomainPtr dom) {
1523 1524 1525
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *type = NULL;
1526

1527
    qemuDriverLock(driver);
1528
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1529
    qemuDriverUnlock(driver);
1530
    if (!vm) {
1531 1532
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1533 1534
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1535
        goto cleanup;
1536 1537
    }

1538
    if (!(type = strdup(vm->def->os.type)))
1539
        virReportOOMError();
1540 1541

cleanup:
1542 1543
    if (vm)
        virDomainObjUnlock(vm);
1544 1545 1546
    return type;
}

1547 1548
/* Returns max memory in kb, 0 if error */
static unsigned long qemudDomainGetMaxMemory(virDomainPtr dom) {
1549 1550 1551
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    unsigned long ret = 0;
1552

1553
    qemuDriverLock(driver);
1554
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1555 1556
    qemuDriverUnlock(driver);

1557
    if (!vm) {
1558 1559
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1560 1561
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1562
        goto cleanup;
1563 1564
    }

1565
    ret = vm->def->mem.max_balloon;
1566 1567

cleanup:
1568 1569
    if (vm)
        virDomainObjUnlock(vm);
1570
    return ret;
1571 1572
}

1573 1574
static int qemudDomainSetMemoryFlags(virDomainPtr dom, unsigned long newmem,
                                     unsigned int flags) {
1575
    struct qemud_driver *driver = dom->conn->privateData;
1576
    qemuDomainObjPrivatePtr priv;
1577
    virDomainObjPtr vm;
1578
    virDomainDefPtr persistentDef = NULL;
1579
    int ret = -1, r;
1580

1581 1582 1583 1584 1585 1586 1587 1588
    virCheckFlags(VIR_DOMAIN_MEM_LIVE |
                  VIR_DOMAIN_MEM_CONFIG, -1);

    if ((flags & (VIR_DOMAIN_MEM_LIVE | VIR_DOMAIN_MEM_CONFIG)) == 0) {
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("invalid flag combination: (0x%x)"), flags);
    }

1589
    qemuDriverLock(driver);
1590
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1591
    qemuDriverUnlock(driver);
1592
    if (!vm) {
1593 1594
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1595 1596
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1597
        goto cleanup;
1598 1599
    }

1600
    if (newmem > vm->def->mem.max_balloon) {
1601 1602
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("cannot set memory higher than max memory"));
1603
        goto cleanup;
1604 1605
    }

1606 1607 1608
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

1609
    if (!virDomainObjIsActive(vm) && (flags & VIR_DOMAIN_MEM_LIVE)) {
1610 1611 1612 1613 1614
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto endjob;
    }

1615 1616 1617 1618 1619 1620 1621 1622 1623
    if (flags & VIR_DOMAIN_MEM_CONFIG) {
        if (!vm->persistent) {
            qemuReportError(VIR_ERR_OPERATION_INVALID, "%s",
                            _("cannot change persistent config of a transient domain"));
            goto endjob;
        }
        if (!(persistentDef = virDomainObjGetPersistentDef(driver->caps, vm)))
            goto endjob;
    }
1624

1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644
    if (flags & VIR_DOMAIN_MEM_LIVE) {
        priv = vm->privateData;
        qemuDomainObjEnterMonitor(vm);
        r = qemuMonitorSetBalloon(priv->mon, newmem);
        qemuDomainObjExitMonitor(vm);
        qemuAuditMemory(vm, vm->def->mem.cur_balloon, newmem, "update", r == 1);
        if (r < 0)
            goto endjob;

        /* Lack of balloon support is a fatal error */
        if (r == 0) {
            qemuReportError(VIR_ERR_OPERATION_INVALID,
                            "%s", _("cannot set memory of an active domain"));
            goto endjob;
        }
    }

    if (flags& VIR_DOMAIN_MEM_CONFIG) {
        persistentDef->mem.cur_balloon = newmem;
        ret = virDomainSaveConfig(driver->configDir, persistentDef);
1645
        goto endjob;
1646
    }
1647

1648
    ret = 0;
1649
endjob:
1650 1651
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
1652

1653
cleanup:
1654 1655
    if (vm)
        virDomainObjUnlock(vm);
1656
    return ret;
1657 1658
}

1659 1660 1661 1662
static int qemudDomainSetMemory(virDomainPtr dom, unsigned long newmem) {
    return qemudDomainSetMemoryFlags(dom, newmem, VIR_DOMAIN_MEM_LIVE);
}

1663
static int qemudDomainGetInfo(virDomainPtr dom,
1664
                              virDomainInfoPtr info) {
1665 1666 1667
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1668 1669
    int err;
    unsigned long balloon;
1670

1671
    qemuDriverLock(driver);
1672
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1673
    qemuDriverUnlock(driver);
D
Daniel P. Berrange 已提交
1674
    if (!vm) {
1675 1676
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1677 1678
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1679
        goto cleanup;
D
Daniel P. Berrange 已提交
1680 1681
    }

1682
    info->state = vm->state;
D
Daniel P. Berrange 已提交
1683

D
Daniel P. Berrange 已提交
1684
    if (!virDomainObjIsActive(vm)) {
1685
        info->cpuTime = 0;
D
Daniel P. Berrange 已提交
1686
    } else {
1687
        if (qemudGetProcessInfo(&(info->cpuTime), NULL, vm->pid, 0) < 0) {
E
Eric Blake 已提交
1688 1689
            qemuReportError(VIR_ERR_OPERATION_FAILED, "%s",
                            _("cannot read cputime for domain"));
1690
            goto cleanup;
D
Daniel P. Berrange 已提交
1691 1692 1693
        }
    }

1694
    info->maxMem = vm->def->mem.max_balloon;
1695

D
Daniel P. Berrange 已提交
1696
    if (virDomainObjIsActive(vm)) {
1697
        qemuDomainObjPrivatePtr priv = vm->privateData;
1698 1699 1700

        if ((vm->def->memballoon != NULL) &&
            (vm->def->memballoon->model == VIR_DOMAIN_MEMBALLOON_MODEL_NONE)) {
1701
            info->memory = vm->def->mem.max_balloon;
1702
        } else if (!priv->jobActive) {
1703 1704
            if (qemuDomainObjBeginJob(vm) < 0)
                goto cleanup;
1705 1706 1707 1708 1709 1710 1711 1712 1713
            if (!virDomainObjIsActive(vm))
                err = 0;
            else {
                qemuDomainObjEnterMonitor(vm);
                err = qemuMonitorGetBalloonInfo(priv->mon, &balloon);
                qemuDomainObjExitMonitor(vm);
            }
            if (qemuDomainObjEndJob(vm) == 0) {
                vm = NULL;
1714 1715 1716
                goto cleanup;
            }

1717 1718
            if (err < 0)
                goto cleanup;
1719 1720
            if (err == 0)
                /* Balloon not supported, so maxmem is always the allocation */
1721
                info->memory = vm->def->mem.max_balloon;
1722 1723 1724
            else
                info->memory = balloon;
        } else {
1725
            info->memory = vm->def->mem.cur_balloon;
1726
        }
1727
    } else {
1728
        info->memory = vm->def->mem.cur_balloon;
1729 1730
    }

1731
    info->nrVirtCpu = vm->def->vcpus;
1732 1733 1734
    ret = 0;

cleanup:
1735 1736
    if (vm)
        virDomainObjUnlock(vm);
1737
    return ret;
D
Daniel P. Berrange 已提交
1738 1739 1740
}


1741
#define QEMUD_SAVE_MAGIC "LibvirtQemudSave"
1742 1743 1744
#define QEMUD_SAVE_VERSION 2

enum qemud_save_formats {
1745 1746 1747
    QEMUD_SAVE_FORMAT_RAW = 0,
    QEMUD_SAVE_FORMAT_GZIP = 1,
    QEMUD_SAVE_FORMAT_BZIP2 = 2,
1748 1749
    /*
     * Deprecated by xz and never used as part of a release
1750
     * QEMUD_SAVE_FORMAT_LZMA
1751 1752
     */
    QEMUD_SAVE_FORMAT_XZ = 3,
1753
    QEMUD_SAVE_FORMAT_LZOP = 4,
1754 1755 1756
    /* Note: add new members only at the end.
       These values are used in the on-disk format.
       Do not change or re-use numbers. */
1757 1758

    QEMUD_SAVE_FORMAT_LAST
1759
};
1760

1761 1762 1763 1764 1765
VIR_ENUM_DECL(qemudSaveCompression)
VIR_ENUM_IMPL(qemudSaveCompression, QEMUD_SAVE_FORMAT_LAST,
              "raw",
              "gzip",
              "bzip2",
1766 1767
              "xz",
              "lzop")
1768

1769 1770 1771 1772 1773
struct qemud_save_header {
    char magic[sizeof(QEMUD_SAVE_MAGIC)-1];
    int version;
    int xml_len;
    int was_running;
1774 1775
    int compressed;
    int unused[15];
1776 1777
};

1778
/* return -errno on failure, or 0 on success */
E
Eric Blake 已提交
1779 1780 1781 1782
static int
qemuDomainSaveHeader(int fd, const char *path, char *xml,
                     struct qemud_save_header *header)
{
1783 1784
    int ret = 0;

E
Eric Blake 已提交
1785
    if (safewrite(fd, header, sizeof(*header)) != sizeof(*header)) {
1786
        ret = -errno;
1787
        qemuReportError(VIR_ERR_OPERATION_FAILED,
1788
                        _("failed to write header to domain save file '%s'"),
E
Eric Blake 已提交
1789
                        path);
1790 1791 1792
        goto endjob;
    }

E
Eric Blake 已提交
1793
    if (safewrite(fd, xml, header->xml_len) != header->xml_len) {
1794
        ret = -errno;
1795
        qemuReportError(VIR_ERR_OPERATION_FAILED,
E
Eric Blake 已提交
1796
                         _("failed to write xml to '%s'"), path);
1797 1798 1799 1800 1801 1802
        goto endjob;
    }
endjob:
    return ret;
}

1803 1804 1805 1806 1807 1808 1809 1810 1811
/* Given a enum qemud_save_formats compression level, return the name
 * of the program to run, or NULL if no program is needed.  */
static const char *
qemuCompressProgramName(int compress)
{
    return (compress == QEMUD_SAVE_FORMAT_RAW ? NULL :
            qemudSaveCompressionTypeToString(compress));
}

1812 1813 1814
/* This internal function expects the driver lock to already be held on
 * entry and the vm must be active.
 */
1815 1816
static int qemudDomainSaveFlag(struct qemud_driver *driver, virDomainPtr dom,
                               virDomainObjPtr vm, const char *path,
1817
                               int compressed)
1818
{
1819
    char *xml = NULL;
1820
    struct qemud_save_header header;
1821
    bool bypassSecurityDriver = false;
1822
    int ret = -1;
1823
    int rc;
1824
    virDomainEventPtr event = NULL;
1825
    qemuDomainObjPrivatePtr priv;
1826
    struct stat sb;
1827
    bool is_reg = false;
1828
    unsigned long long offset;
1829
    virBitmapPtr qemuCaps = NULL;
1830
    int fd = -1;
1831 1832 1833 1834 1835

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

1836
    header.compressed = compressed;
1837

1838
    priv = vm->privateData;
1839

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

O
Osier Yang 已提交
1843 1844
    priv->jobActive = QEMU_JOB_SAVE;

1845 1846 1847
    memset(&priv->jobInfo, 0, sizeof(priv->jobInfo));
    priv->jobInfo.type = VIR_DOMAIN_JOB_UNBOUNDED;

1848 1849 1850
    /* Pause */
    if (vm->state == VIR_DOMAIN_RUNNING) {
        header.was_running = 1;
1851
        if (qemuProcessStopCPUs(driver, vm) < 0)
1852
            goto endjob;
1853 1854 1855 1856 1857 1858

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

1861 1862 1863 1864 1865
    if (qemuCapsExtractVersionInfo(vm->def->emulator, vm->def->os.arch,
                                   NULL,
                                   &qemuCaps) < 0)
        goto endjob;

1866
    /* Get XML for the domain */
1867
    xml = virDomainDefFormat(vm->def, VIR_DOMAIN_XML_SECURE);
1868
    if (!xml) {
1869 1870
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to get domain xml"));
1871
        goto endjob;
1872 1873 1874
    }
    header.xml_len = strlen(xml) + 1;

1875 1876 1877 1878 1879 1880 1881 1882
    /* 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 */
1883
        is_reg = true;
1884
    } else {
1885
        is_reg = !!S_ISREG(sb.st_mode);
1886 1887
    }

1888 1889 1890 1891 1892
    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
1893 1894
     * it by padding the XML string with NULLs.
     */
1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907
    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;
    }
1908

1909
    /* Obtain the file handle.  */
1910

1911
    /* First try creating the file as root */
1912
    if (!is_reg) {
1913
        fd = open(path, O_WRONLY | O_TRUNC);
1914 1915
        if (fd < 0) {
            virReportSystemError(errno, _("unable to open %s"), path);
1916 1917
            goto endjob;
        }
1918
    } else {
1919 1920 1921
        if ((fd = virFileOpenAs(path, O_CREAT|O_TRUNC|O_WRONLY,
                                S_IRUSR|S_IWUSR,
                                getuid(), getgid(), 0)) < 0) {
1922
            /* If we failed as root, and the error was permission-denied
1923 1924
               (EACCES or EPERM), assume it's on a network-connected share
               where root access is restricted (eg, root-squashed NFS). If the
1925 1926 1927
               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 */
1928
            rc = fd;
1929
            if (((rc != -EACCES) && (rc != -EPERM)) ||
1930
                driver->user == getuid()) {
1931 1932
                virReportSystemError(-rc,
                                    _("Failed to create domain save file '%s'"),
1933 1934 1935
                                     path);
                goto endjob;
            }
1936

1937
            /* On Linux we can also verify the FS-type of the directory. */
1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954
            switch (virStorageFileIsSharedFS(path)) {
                case 1:
                   /* it was on a network share, so we'll continue
                    * as outlined above
                    */
                   break;

                case -1:
                   virReportSystemError(errno,
                                        _("Failed to create domain save file "
                                          "'%s': couldn't determine fs type"),
                                        path);
                   goto endjob;
                   break;

                case 0:
                default:
1955
                   /* local file - log the error returned by virFileOpenAs */
1956
                   virReportSystemError(-rc,
1957
                                    _("Failed to create domain save file '%s'"),
1958 1959 1960
                                        path);
                   goto endjob;
                   break;
1961 1962 1963

            }

1964
            /* Retry creating the file as driver->user */
1965

1966 1967 1968 1969
            if ((fd = virFileOpenAs(path, O_CREAT|O_TRUNC|O_WRONLY,
                                    S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP,
                                    driver->user, driver->group,
                                    VIR_FILE_OPEN_AS_UID)) < 0) {
1970 1971 1972
                virReportSystemError(-fd,
                                   _("Error from child process creating '%s'"),
                                     path);
1973 1974
                goto endjob;
            }
1975

1976 1977 1978
            /* 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 */
1979

1980
            bypassSecurityDriver = true;
1981
        }
1982
    }
1983

1984
    /* Write header to file, followed by XML */
E
Eric Blake 已提交
1985
    if (qemuDomainSaveHeader(fd, path, xml, &header) < 0) {
1986 1987 1988 1989
        VIR_FORCE_CLOSE(fd);
        goto endjob;
    }

1990 1991 1992 1993
    /* Perform the migration */
    if (qemuMigrationToFile(driver, vm, qemuCaps, fd, offset, path,
                            qemuCompressProgramName(compressed),
                            is_reg, bypassSecurityDriver) < 0)
1994
        goto endjob;
1995 1996
    if (VIR_CLOSE(fd) < 0) {
        virReportSystemError(errno, _("unable to close %s"), path);
1997
        goto endjob;
1998 1999
    }

2000 2001
    ret = 0;

2002
    /* Shut it down */
2003
    qemuProcessStop(driver, vm, 0);
2004
    qemuAuditDomainStop(vm, "saved");
2005 2006 2007
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_SAVED);
2008
    if (!vm->persistent) {
2009 2010 2011
        if (qemuDomainObjEndJob(vm) > 0)
            virDomainRemoveInactive(&driver->domains,
                                    vm);
2012 2013
        vm = NULL;
    }
2014

2015
endjob:
2016
    if (vm) {
2017
        if (ret != 0) {
2018
            if (header.was_running && virDomainObjIsActive(vm)) {
2019
                rc = qemuProcessStartCPUs(driver, vm, dom->conn);
2020 2021 2022
                if (rc < 0)
                    VIR_WARN0("Unable to resume guest CPUs after save failure");
            }
2023
        }
2024
        if (qemuDomainObjEndJob(vm) == 0)
2025
            vm = NULL;
2026
    }
2027

2028
cleanup:
2029
    qemuCapsFree(qemuCaps);
2030
    VIR_FORCE_CLOSE(fd);
2031
    VIR_FREE(xml);
2032
    if (ret != 0 && is_reg)
2033
        unlink(path);
2034 2035
    if (event)
        qemuDomainEventQueue(driver, event);
2036
    return ret;
D
Daniel P. Berrange 已提交
2037 2038
}

2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054
/* Returns true if a compression program is available in PATH */
static bool qemudCompressProgramAvailable(enum qemud_save_formats compress)
{
    const char *prog;
    char *c;

    if (compress == QEMUD_SAVE_FORMAT_RAW)
        return true;
    prog = qemudSaveCompressionTypeToString(compress);
    c = virFindFileInPath(prog);
    if (!c)
        return false;
    VIR_FREE(c);
    return true;
}

2055 2056 2057 2058
static int qemudDomainSave(virDomainPtr dom, const char *path)
{
    struct qemud_driver *driver = dom->conn->privateData;
    int compressed;
2059 2060 2061 2062
    int ret = -1;
    virDomainObjPtr vm = NULL;

    qemuDriverLock(driver);
2063 2064 2065 2066 2067 2068 2069 2070 2071

    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"));
2072
            goto cleanup;
2073
        }
2074 2075 2076 2077
        if (!qemudCompressProgramAvailable(compressed)) {
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("Compression program for image format "
                                    "in configuration file isn't available"));
2078
            goto cleanup;
2079
        }
2080 2081
    }

2082 2083 2084 2085 2086 2087 2088 2089 2090
    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;
    }

2091 2092 2093 2094 2095 2096
    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
    }

2097 2098 2099 2100 2101 2102 2103 2104
    ret = qemudDomainSaveFlag(driver, dom, vm, path, compressed);

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);

    return ret;
2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127
}

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;

2128
    virCheckFlags(0, -1);
2129 2130 2131 2132 2133 2134 2135 2136

    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);
2137
        goto cleanup;
2138 2139
    }

2140 2141 2142 2143 2144 2145
    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
    }

2146 2147
    name = qemuDomainManagedSavePath(driver, vm);
    if (name == NULL)
2148
        goto cleanup;
2149

2150
    VIR_INFO("Saving state to %s", name);
2151 2152

    compressed = QEMUD_SAVE_FORMAT_RAW;
2153
    ret = qemudDomainSaveFlag(driver, dom, vm, name, compressed);
2154 2155 2156 2157 2158

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
2159 2160 2161
    VIR_FREE(name);

    return ret;
2162 2163 2164 2165 2166 2167 2168 2169 2170 2171
}

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

2172
    virCheckFlags(0, -1);
2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205

    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;

2206
    virCheckFlags(0, -1);
2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230

    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 已提交
2231

H
Hu Tao 已提交
2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246
static int doCoreDump(struct qemud_driver *driver,
                      virDomainObjPtr vm,
                      const char *path,
                      enum qemud_save_formats compress)
{
    int fd = -1;
    int ret = -1;

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

2247 2248 2249 2250
    if (qemuMigrationToFile(driver, vm, NULL, fd, 0, path,
                            qemuCompressProgramName(compress), true, false) < 0)
        goto cleanup;

H
Hu Tao 已提交
2251 2252 2253 2254 2255 2256 2257
    if (VIR_CLOSE(fd) < 0) {
        virReportSystemError(errno,
                             _("unable to save file %s"),
                             path);
        goto cleanup;
    }

2258
    ret = 0;
H
Hu Tao 已提交
2259 2260 2261 2262 2263 2264 2265

cleanup:
    if (ret != 0)
        unlink(path);
    return ret;
}

2266 2267 2268 2269 2270
static enum qemud_save_formats
getCompressionType(struct qemud_driver *driver)
{
    int compress = QEMUD_SAVE_FORMAT_RAW;

2271 2272 2273 2274 2275 2276 2277
    /*
     * We reuse "save" flag for "dump" here. Then, we can support the same
     * format in "save" and "dump".
     */
    if (driver->dumpImageFormat) {
        compress = qemudSaveCompressionTypeFromString(driver->dumpImageFormat);
        if (compress < 0) {
2278 2279 2280 2281
            qemuReportError(VIR_ERR_OPERATION_FAILED, "%s",
                            _("Invalid dump image format specified in "
                              "configuration file, using raw"));
            return QEMUD_SAVE_FORMAT_RAW;
2282
        }
2283 2284 2285
        if (!qemudCompressProgramAvailable(compress)) {
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("Compression program for dump image format "
2286 2287 2288
                                    "in configuration file isn't available, "
                                    "using raw"));
            return QEMUD_SAVE_FORMAT_RAW;
2289
        }
2290
    }
2291 2292 2293 2294 2295 2296 2297 2298 2299
    return compress;
}

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;
H
Hu Tao 已提交
2300
    int ret = -1;
2301 2302 2303
    virDomainEventPtr event = NULL;
    qemuDomainObjPrivatePtr priv;

P
Paolo Bonzini 已提交
2304 2305 2306 2307 2308 2309
    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);

    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2310 2311
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
P
Paolo Bonzini 已提交
2312 2313
        goto cleanup;
    }
2314
    priv = vm->privateData;
P
Paolo Bonzini 已提交
2315

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

D
Daniel P. Berrange 已提交
2319
    if (!virDomainObjIsActive(vm)) {
2320 2321
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
2322
        goto endjob;
P
Paolo Bonzini 已提交
2323 2324
    }

O
Osier Yang 已提交
2325 2326
    priv->jobActive = QEMU_JOB_DUMP;

P
Paolo Bonzini 已提交
2327 2328
    /* Migrate will always stop the VM, so the resume condition is
       independent of whether the stop command is issued.  */
P
Paolo Bonzini 已提交
2329 2330 2331
    resume = (vm->state == VIR_DOMAIN_RUNNING);

    /* Pause domain for non-live dump */
P
Paolo Bonzini 已提交
2332
    if (!(flags & VIR_DUMP_LIVE) && vm->state == VIR_DOMAIN_RUNNING) {
2333
        if (qemuProcessStopCPUs(driver, vm) < 0)
2334
            goto endjob;
P
Paolo Bonzini 已提交
2335
        paused = 1;
2336 2337 2338 2339 2340 2341

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

H
Hu Tao 已提交
2344
    ret = doCoreDump(driver, vm, path, getCompressionType(driver));
2345 2346 2347 2348
    if (ret < 0)
        goto endjob;

    paused = 1;
2349 2350

endjob:
2351
    if ((ret == 0) && (flags & VIR_DUMP_CRASH)) {
2352
        qemuProcessStop(driver, vm, 0);
2353
        qemuAuditDomainStop(vm, "crashed");
2354 2355 2356 2357 2358
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_CRASHED);
    }

P
Paolo Bonzini 已提交
2359 2360 2361
    /* 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.  */
2362
    else if (resume && paused && virDomainObjIsActive(vm)) {
2363
        if (qemuProcessStartCPUs(driver, vm, dom->conn) < 0) {
2364
            if (virGetLastError() == NULL)
2365 2366
                qemuReportError(VIR_ERR_OPERATION_FAILED,
                                "%s", _("resuming after dump failed"));
P
Paolo Bonzini 已提交
2367 2368
        }
    }
2369

2370 2371
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
2372
    else if ((ret == 0) && (flags & VIR_DUMP_CRASH) && !vm->persistent) {
2373 2374 2375 2376
        virDomainRemoveInactive(&driver->domains,
                                vm);
        vm = NULL;
    }
2377 2378

cleanup:
P
Paolo Bonzini 已提交
2379 2380
    if (vm)
        virDomainObjUnlock(vm);
2381 2382
    if (event)
        qemuDomainEventQueue(driver, event);
2383
    qemuDriverUnlock(driver);
P
Paolo Bonzini 已提交
2384 2385 2386
    return ret;
}

H
Hu Tao 已提交
2387 2388 2389
static void processWatchdogEvent(void *data, void *opaque)
{
    int ret;
2390
    struct qemuDomainWatchdogEvent *wdEvent = data;
H
Hu Tao 已提交
2391 2392 2393 2394 2395 2396 2397
    struct qemud_driver *driver = opaque;

    switch (wdEvent->action) {
    case VIR_DOMAIN_WATCHDOG_ACTION_DUMP:
        {
            char *dumpfile;

E
Eric Blake 已提交
2398
            if (virAsprintf(&dumpfile, "%s/%s-%u",
H
Hu Tao 已提交
2399 2400
                            driver->autoDumpPath,
                            wdEvent->vm->def->name,
E
Eric Blake 已提交
2401 2402 2403 2404
                            (unsigned int)time(NULL)) < 0) {
                virReportOOMError();
                break;
            }
H
Hu Tao 已提交
2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425

            qemuDriverLock(driver);
            virDomainObjLock(wdEvent->vm);

            if (qemuDomainObjBeginJobWithDriver(driver, wdEvent->vm) < 0)
                break;

            if (!virDomainObjIsActive(wdEvent->vm)) {
                qemuReportError(VIR_ERR_OPERATION_INVALID,
                                "%s", _("domain is not running"));
                break;
            }

            ret = doCoreDump(driver,
                             wdEvent->vm,
                             dumpfile,
                             getCompressionType(driver));
            if (ret < 0)
                qemuReportError(VIR_ERR_OPERATION_FAILED,
                                "%s", _("Dump failed"));

2426
            ret = qemuProcessStartCPUs(driver, wdEvent->vm, NULL);
H
Hu Tao 已提交
2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443

            if (ret < 0)
                qemuReportError(VIR_ERR_OPERATION_FAILED,
                                "%s", _("Resuming after dump failed"));

            if (qemuDomainObjEndJob(wdEvent->vm) > 0)
                virDomainObjUnlock(wdEvent->vm);

            qemuDriverUnlock(driver);

            VIR_FREE(dumpfile);
        }
        break;
    }

    VIR_FREE(wdEvent);
}
P
Paolo Bonzini 已提交
2444

2445 2446 2447
static int qemudDomainHotplugVcpus(virDomainObjPtr vm, unsigned int nvcpus)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
2448
    int i, rc = 1;
2449
    int ret = -1;
2450
    int oldvcpus = vm->def->vcpus;
E
Eric Blake 已提交
2451
    int vcpus = oldvcpus;
2452

2453 2454
    qemuDomainObjEnterMonitor(vm);

2455 2456 2457
    /* 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 */
E
Eric Blake 已提交
2458 2459
    if (nvcpus > vcpus) {
        for (i = vcpus ; i < nvcpus ; i++) {
2460 2461 2462 2463 2464 2465 2466
            /* Online new CPU */
            rc = qemuMonitorSetCPU(priv->mon, i, 1);
            if (rc == 0)
                goto unsupported;
            if (rc < 0)
                goto cleanup;

E
Eric Blake 已提交
2467
            vcpus++;
2468 2469
        }
    } else {
E
Eric Blake 已提交
2470
        for (i = vcpus - 1 ; i >= nvcpus ; i--) {
2471 2472 2473 2474 2475 2476 2477
            /* Offline old CPU */
            rc = qemuMonitorSetCPU(priv->mon, i, 0);
            if (rc == 0)
                goto unsupported;
            if (rc < 0)
                goto cleanup;

E
Eric Blake 已提交
2478
            vcpus--;
2479 2480 2481 2482 2483 2484
        }
    }

    ret = 0;

cleanup:
2485
    qemuDomainObjExitMonitor(vm);
E
Eric Blake 已提交
2486
    vm->def->vcpus = vcpus;
2487
    qemuAuditVcpu(vm, oldvcpus, nvcpus, "update", rc == 1);
2488 2489 2490 2491 2492 2493 2494 2495 2496
    return ret;

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


2497 2498 2499 2500
static int
qemudDomainSetVcpusFlags(virDomainPtr dom, unsigned int nvcpus,
                         unsigned int flags)
{
2501 2502
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2503
    virDomainDefPtr persistentDef;
2504 2505
    const char * type;
    int max;
2506
    int ret = -1;
2507

2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523
    virCheckFlags(VIR_DOMAIN_VCPU_LIVE |
                  VIR_DOMAIN_VCPU_CONFIG |
                  VIR_DOMAIN_VCPU_MAXIMUM, -1);

    /* At least one of LIVE or CONFIG must be set.  MAXIMUM cannot be
     * mixed with LIVE.  */
    if ((flags & (VIR_DOMAIN_VCPU_LIVE | VIR_DOMAIN_VCPU_CONFIG)) == 0 ||
        (flags & (VIR_DOMAIN_VCPU_MAXIMUM | VIR_DOMAIN_VCPU_LIVE)) ==
         (VIR_DOMAIN_VCPU_MAXIMUM | VIR_DOMAIN_VCPU_LIVE)) {
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("invalid flag combination: (0x%x)"), flags);
        return -1;
    }
    if (!nvcpus || (unsigned short) nvcpus != nvcpus) {
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("argument out of range: %d"), nvcpus);
2524 2525 2526
        return -1;
    }

2527
    qemuDriverLock(driver);
2528
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2529 2530
    qemuDriverUnlock(driver);

2531
    if (!vm) {
2532 2533
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2534 2535
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
2536
        goto cleanup;
2537 2538
    }

2539 2540 2541
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

2542
    if (!virDomainObjIsActive(vm) && (flags & VIR_DOMAIN_VCPU_LIVE)) {
2543 2544
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                         "%s", _("domain is not running"));
2545
        goto endjob;
2546 2547
    }

2548 2549 2550 2551 2552 2553
    if (!vm->persistent && (flags & VIR_DOMAIN_VCPU_CONFIG)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID, "%s",
                        _("cannot change persistent config of a transient domain"));
        goto endjob;
    }

2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566
    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;
    }

C
Cole Robinson 已提交
2567
    if (!(flags & VIR_DOMAIN_VCPU_MAXIMUM) && vm->def->maxvcpus < max) {
2568 2569 2570
        max = vm->def->maxvcpus;
    }

2571 2572 2573 2574 2575 2576 2577
    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;
    }

2578 2579 2580
    if (!(persistentDef = virDomainObjGetPersistentDef(driver->caps, vm)))
        goto endjob;

2581 2582
    switch (flags) {
    case VIR_DOMAIN_VCPU_MAXIMUM | VIR_DOMAIN_VCPU_CONFIG:
2583 2584 2585
        persistentDef->maxvcpus = nvcpus;
        if (nvcpus < persistentDef->vcpus)
            persistentDef->vcpus = nvcpus;
2586 2587 2588 2589
        ret = 0;
        break;

    case VIR_DOMAIN_VCPU_CONFIG:
2590
        persistentDef->vcpus = nvcpus;
2591 2592 2593 2594 2595 2596 2597 2598 2599
        ret = 0;
        break;

    case VIR_DOMAIN_VCPU_LIVE:
        ret = qemudDomainHotplugVcpus(vm, nvcpus);
        break;

    case VIR_DOMAIN_VCPU_LIVE | VIR_DOMAIN_VCPU_CONFIG:
        ret = qemudDomainHotplugVcpus(vm, nvcpus);
2600 2601 2602
        if (ret == 0) {
            persistentDef->vcpus = nvcpus;
        }
2603 2604
        break;
    }
2605

2606 2607 2608 2609
    /* Save the persistent config to disk */
    if (flags & VIR_DOMAIN_VCPU_CONFIG)
        ret = virDomainSaveConfig(driver->configDir, persistentDef);

2610 2611 2612
endjob:
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
2613

2614
cleanup:
2615 2616
    if (vm)
        virDomainObjUnlock(vm);
2617
    return ret;
2618 2619
}

2620 2621 2622 2623 2624 2625
static int
qemudDomainSetVcpus(virDomainPtr dom, unsigned int nvcpus)
{
    return qemudDomainSetVcpusFlags(dom, nvcpus, VIR_DOMAIN_VCPU_LIVE);
}

2626 2627 2628 2629 2630 2631

static int
qemudDomainPinVcpu(virDomainPtr dom,
                   unsigned int vcpu,
                   unsigned char *cpumap,
                   int maplen) {
2632 2633
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2634
    int maxcpu, hostcpus;
2635
    virNodeInfo nodeinfo;
2636
    int ret = -1;
2637
    qemuDomainObjPrivatePtr priv;
2638

2639
    qemuDriverLock(driver);
2640
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2641 2642
    qemuDriverUnlock(driver);

2643 2644 2645
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2646 2647
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
2648 2649 2650
        goto cleanup;
    }

D
Daniel P. Berrange 已提交
2651
    if (!virDomainObjIsActive(vm)) {
2652 2653
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s",_("cannot pin vcpus on an inactive domain"));
2654
        goto cleanup;
2655 2656
    }

2657 2658 2659
    priv = vm->privateData;

    if (vcpu > (priv->nvcpupids-1)) {
2660 2661 2662
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("vcpu number out of range %d > %d"),
                        vcpu, priv->nvcpupids);
2663
        goto cleanup;
2664 2665
    }

2666
    if (nodeGetInfo(dom->conn, &nodeinfo) < 0)
2667
        goto cleanup;
2668

2669
    hostcpus = VIR_NODEINFO_MAXCPUS(nodeinfo);
2670
    maxcpu = maplen * 8;
2671 2672
    if (maxcpu > hostcpus)
        maxcpu = hostcpus;
2673

2674 2675
    if (priv->vcpupids != NULL) {
        if (virProcessInfoSetAffinity(priv->vcpupids[vcpu],
2676
                                      cpumap, maplen, maxcpu) < 0)
2677
            goto cleanup;
2678
    } else {
2679 2680
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        "%s", _("cpu affinity is not supported"));
2681
        goto cleanup;
2682
    }
2683 2684 2685 2686 2687 2688 2689

    if (virDomainVcpupinAdd(vm->def, cpumap, maplen, vcpu) < 0) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("failed to update or add vcpupin xml"));
        goto cleanup;
    }

2690
    ret = 0;
2691

2692
cleanup:
2693 2694
    if (vm)
        virDomainObjUnlock(vm);
2695
    return ret;
2696 2697 2698 2699 2700 2701 2702 2703
}

static int
qemudDomainGetVcpus(virDomainPtr dom,
                    virVcpuInfoPtr info,
                    int maxinfo,
                    unsigned char *cpumaps,
                    int maplen) {
2704 2705
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2706
    virNodeInfo nodeinfo;
2707
    int i, v, maxcpu, hostcpus;
2708
    int ret = -1;
2709
    qemuDomainObjPrivatePtr priv;
2710

2711
    qemuDriverLock(driver);
2712
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2713 2714
    qemuDriverUnlock(driver);

2715 2716 2717
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2718 2719
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
2720 2721 2722
        goto cleanup;
    }

D
Daniel P. Berrange 已提交
2723
    if (!virDomainObjIsActive(vm)) {
2724 2725 2726
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s",
                        _("cannot list vcpu pinning for an inactive domain"));
2727
        goto cleanup;
2728 2729
    }

2730 2731
    priv = vm->privateData;

2732
    if (nodeGetInfo(dom->conn, &nodeinfo) < 0)
2733
        goto cleanup;
2734

2735
    hostcpus = VIR_NODEINFO_MAXCPUS(nodeinfo);
2736
    maxcpu = maplen * 8;
2737 2738
    if (maxcpu > hostcpus)
        maxcpu = hostcpus;
2739 2740

    /* Clamp to actual number of vcpus */
2741 2742
    if (maxinfo > priv->nvcpupids)
        maxinfo = priv->nvcpupids;
2743

2744 2745 2746 2747 2748 2749
    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;
2750

2751
                if (priv->vcpupids != NULL &&
2752 2753 2754
                    qemudGetProcessInfo(&(info[i].cpuTime),
                                        &(info[i].cpu),
                                        vm->pid,
2755
                                        priv->vcpupids[i]) < 0) {
2756
                    virReportSystemError(errno, "%s",
2757 2758 2759
                                         _("cannot get vCPU placement & pCPU time"));
                    goto cleanup;
                }
2760
            }
2761 2762
        }

2763 2764
        if (cpumaps != NULL) {
            memset(cpumaps, 0, maplen * maxinfo);
2765
            if (priv->vcpupids != NULL) {
2766 2767 2768
                for (v = 0 ; v < maxinfo ; v++) {
                    unsigned char *cpumap = VIR_GET_CPUMAP(cpumaps, maplen, v);

2769
                    if (virProcessInfoGetAffinity(priv->vcpupids[v],
2770
                                                  cpumap, maplen, maxcpu) < 0)
2771
                        goto cleanup;
2772
                }
2773
            } else {
2774 2775
                qemuReportError(VIR_ERR_NO_SUPPORT,
                                "%s", _("cpu affinity is not available"));
2776
                goto cleanup;
2777 2778 2779
            }
        }
    }
2780
    ret = maxinfo;
2781

2782
cleanup:
2783 2784
    if (vm)
        virDomainObjUnlock(vm);
2785
    return ret;
2786 2787 2788
}


2789 2790 2791
static int
qemudDomainGetVcpusFlags(virDomainPtr dom, unsigned int flags)
{
2792 2793
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2794
    virDomainDefPtr def;
2795
    int ret = -1;
2796

2797 2798 2799 2800 2801 2802 2803 2804
    virCheckFlags(VIR_DOMAIN_VCPU_LIVE |
                  VIR_DOMAIN_VCPU_CONFIG |
                  VIR_DOMAIN_VCPU_MAXIMUM, -1);

    /* Exactly one of LIVE or CONFIG must be set.  */
    if (!(flags & VIR_DOMAIN_VCPU_LIVE) == !(flags & VIR_DOMAIN_VCPU_CONFIG)) {
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("invalid flag combination: (0x%x)"), flags);
2805 2806 2807
        return -1;
    }

2808
    qemuDriverLock(driver);
2809
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2810 2811
    qemuDriverUnlock(driver);

2812
    if (!vm) {
2813 2814
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2815 2816
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
2817
        goto cleanup;
2818 2819
    }

2820 2821 2822 2823 2824 2825 2826 2827 2828
    if (flags & VIR_DOMAIN_VCPU_LIVE) {
        if (!virDomainObjIsActive(vm)) {
            qemuReportError(VIR_ERR_OPERATION_INVALID, "%s",
                            _("domain not active"));
            goto cleanup;
        }
        def = vm->def;
    } else {
        def = vm->newDef ? vm->newDef : vm->def;
2829 2830
    }

2831
    ret = (flags & VIR_DOMAIN_VCPU_MAXIMUM) ? def->maxvcpus : def->vcpus;
2832

2833
cleanup:
2834 2835
    if (vm)
        virDomainObjUnlock(vm);
2836 2837 2838
    return ret;
}

2839 2840 2841 2842 2843 2844 2845
static int
qemudDomainGetMaxVcpus(virDomainPtr dom)
{
    return qemudDomainGetVcpusFlags(dom, (VIR_DOMAIN_VCPU_LIVE |
                                          VIR_DOMAIN_VCPU_MAXIMUM));
}

2846 2847 2848 2849 2850 2851 2852 2853 2854
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);

2855 2856
    memset(seclabel, 0, sizeof(*seclabel));

2857 2858 2859
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2860 2861
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
2862 2863 2864
        goto cleanup;
    }

2865
    if (!virDomainVirtTypeToString(vm->def->virtType)) {
2866 2867 2868
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("unknown virt type in domain definition '%d'"),
                        vm->def->virtType);
2869 2870 2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883 2884 2885
        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 已提交
2886
    if (virDomainObjIsActive(vm)) {
2887 2888
        if (virSecurityManagerGetProcessLabel(driver->securityManager,
                                              vm, seclabel) < 0) {
2889 2890 2891
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            "%s", _("Failed to get security label"));
            goto cleanup;
2892 2893 2894 2895 2896 2897 2898 2899
        }
    }

    ret = 0;

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
2900
    qemuDriverUnlock(driver);
2901 2902 2903
    return ret;
}

2904 2905
static int qemudNodeGetSecurityModel(virConnectPtr conn,
                                     virSecurityModelPtr secmodel)
2906 2907 2908
{
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
    char *p;
2909
    int ret = 0;
2910

2911
    qemuDriverLock(driver);
2912 2913 2914 2915 2916
    memset(secmodel, 0, sizeof(*secmodel));

    /* NULL indicates no driver, which we treat as
     * success, but simply return no data in *secmodel */
    if (driver->caps->host.secModel.model == NULL)
2917
        goto cleanup;
2918

2919 2920
    p = driver->caps->host.secModel.model;
    if (strlen(p) >= VIR_SECURITY_MODEL_BUFLEN-1) {
2921 2922 2923
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("security model string exceeds max %d bytes"),
                        VIR_SECURITY_MODEL_BUFLEN-1);
2924 2925
        ret = -1;
        goto cleanup;
2926 2927 2928 2929 2930
    }
    strcpy(secmodel->model, p);

    p = driver->caps->host.secModel.doi;
    if (strlen(p) >= VIR_SECURITY_DOI_BUFLEN-1) {
2931 2932 2933
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("security DOI string exceeds max %d bytes"),
                        VIR_SECURITY_DOI_BUFLEN-1);
2934 2935
        ret = -1;
        goto cleanup;
2936 2937
    }
    strcpy(secmodel->doi, p);
2938 2939 2940 2941

cleanup:
    qemuDriverUnlock(driver);
    return ret;
2942 2943
}

2944 2945 2946 2947 2948
static int ATTRIBUTE_NONNULL(3) ATTRIBUTE_NONNULL(4)
qemuDomainSaveImageOpen(struct qemud_driver *driver,
                        const char *path,
                        virDomainDefPtr *ret_def,
                        struct qemud_save_header *ret_header)
J
Jiri Denemark 已提交
2949 2950
{
    int fd;
2951
    struct qemud_save_header header;
J
Jiri Denemark 已提交
2952 2953
    char *xml = NULL;
    virDomainDefPtr def = NULL;
2954

2955 2956 2957
    if ((fd = virFileOpenAs(path, O_RDONLY, 0, getuid(), getgid(), 0)) < 0) {
        if ((fd != -EACCES && fd != -EPERM) ||
            driver->user == getuid()) {
2958 2959
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("cannot read domain image"));
J
Jiri Denemark 已提交
2960
            goto error;
2961 2962 2963
        }

        /* Opening as root failed, but qemu runs as a different user
2964 2965 2966 2967 2968 2969
         * that might have better luck. */
        if ((fd = virFileOpenAs(path, O_RDONLY, 0,
                                driver->user, driver->group,
                                VIR_FILE_OPEN_AS_UID)) < 0) {
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("cannot read domain image"));
J
Jiri Denemark 已提交
2970
            goto error;
2971
        }
2972 2973 2974
    }

    if (saferead(fd, &header, sizeof(header)) != sizeof(header)) {
2975 2976
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to read qemu header"));
J
Jiri Denemark 已提交
2977
        goto error;
2978 2979 2980
    }

    if (memcmp(header.magic, QEMUD_SAVE_MAGIC, sizeof(header.magic)) != 0) {
2981 2982
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("image magic is incorrect"));
J
Jiri Denemark 已提交
2983
        goto error;
2984 2985 2986
    }

    if (header.version > QEMUD_SAVE_VERSION) {
2987 2988 2989
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("image version is not supported (%d > %d)"),
                        header.version, QEMUD_SAVE_VERSION);
J
Jiri Denemark 已提交
2990
        goto error;
2991 2992
    }

2993 2994 2995
    if (header.xml_len <= 0) {
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("invalid XML length: %d"), header.xml_len);
J
Jiri Denemark 已提交
2996
        goto error;
2997 2998
    }

2999 3000
    if (VIR_ALLOC_N(xml, header.xml_len) < 0) {
        virReportOOMError();
J
Jiri Denemark 已提交
3001
        goto error;
3002 3003 3004
    }

    if (saferead(fd, xml, header.xml_len) != header.xml_len) {
3005 3006
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to read XML"));
J
Jiri Denemark 已提交
3007
        goto error;
3008 3009 3010
    }

    /* Create a domain from this XML */
3011
    if (!(def = virDomainDefParseString(driver->caps, xml,
3012
                                        VIR_DOMAIN_XML_INACTIVE)))
J
Jiri Denemark 已提交
3013
        goto error;
3014

J
Jiri Denemark 已提交
3015
    VIR_FREE(xml);
3016

J
Jiri Denemark 已提交
3017 3018
    *ret_def = def;
    *ret_header = header;
3019

J
Jiri Denemark 已提交
3020
    return fd;
3021

J
Jiri Denemark 已提交
3022 3023 3024
error:
    virDomainDefFree(def);
    VIR_FREE(xml);
3025
    VIR_FORCE_CLOSE(fd);
J
Jiri Denemark 已提交
3026 3027 3028 3029

    return -1;
}

3030 3031 3032 3033 3034 3035 3036
static int ATTRIBUTE_NONNULL(4) ATTRIBUTE_NONNULL(5) ATTRIBUTE_NONNULL(6)
qemuDomainSaveImageStartVM(virConnectPtr conn,
                           struct qemud_driver *driver,
                           virDomainObjPtr vm,
                           int *fd,
                           const struct qemud_save_header *header,
                           const char *path)
J
Jiri Denemark 已提交
3037 3038 3039 3040 3041 3042 3043 3044
{
    int ret = -1;
    virDomainEventPtr event;
    int intermediatefd = -1;
    pid_t intermediate_pid = -1;
    int childstat;

    if (header->version == 2) {
3045
        const char *intermediate_argv[3] = { NULL, "-dc", NULL };
J
Jiri Denemark 已提交
3046
        const char *prog = qemudSaveCompressionTypeToString(header->compressed);
3047
        if (prog == NULL) {
3048 3049
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            _("Invalid compressed save format %d"),
J
Jiri Denemark 已提交
3050 3051
                            header->compressed);
            goto out;
3052
        }
3053

J
Jiri Denemark 已提交
3054
        if (header->compressed != QEMUD_SAVE_FORMAT_RAW) {
3055
            intermediate_argv[0] = prog;
3056 3057
            intermediatefd = *fd;
            *fd = -1;
3058
            if (virExec(intermediate_argv, NULL, NULL,
3059
                        &intermediate_pid, intermediatefd, fd, NULL, 0) < 0) {
3060 3061 3062
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                _("Failed to start decompression binary %s"),
                                intermediate_argv[0]);
3063
                *fd = intermediatefd;
J
Jiri Denemark 已提交
3064
                goto out;
3065 3066 3067
            }
        }
    }
J
Jiri Denemark 已提交
3068

3069
    /* Set the migration source and start it up. */
3070
    ret = qemuProcessStart(conn, driver, vm, "stdio", true, *fd, path,
3071
                           VIR_VM_OP_RESTORE);
J
Jiri Denemark 已提交
3072

3073
    if (intermediate_pid != -1) {
3074
        if (ret < 0) {
3075 3076 3077
            /* if there was an error setting up qemu, the intermediate
             * process will wait forever to write to stdout, so we
             * must manually kill it.
3078 3079
             */
            VIR_FORCE_CLOSE(intermediatefd);
3080
            VIR_FORCE_CLOSE(*fd);
3081 3082 3083
            kill(intermediate_pid, SIGTERM);
        }

3084 3085
        /* Wait for intermediate process to exit */
        while (waitpid(intermediate_pid, &childstat, 0) == -1 &&
J
Jiri Denemark 已提交
3086 3087 3088
               errno == EINTR) {
            /* empty */
        }
3089
    }
3090
    VIR_FORCE_CLOSE(intermediatefd);
J
Jiri Denemark 已提交
3091

3092 3093 3094
    if (VIR_CLOSE(*fd) < 0) {
        virReportSystemError(errno, _("cannot close file: %s"), path);
        ret = -1;
3095
    }
J
Jiri Denemark 已提交
3096

3097
    if (ret < 0) {
3098
        qemuAuditDomainStart(vm, "restored", false);
J
Jiri Denemark 已提交
3099
        goto out;
3100
    }
3101

3102 3103 3104
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_RESTORED);
3105
    qemuAuditDomainStart(vm, "restored", true);
J
Jiri Denemark 已提交
3106 3107 3108
    if (event)
        qemuDomainEventQueue(driver, event);

3109

3110
    /* If it was running before, resume it now. */
J
Jiri Denemark 已提交
3111
    if (header->was_running) {
3112
        if (qemuProcessStartCPUs(driver, vm, conn) < 0) {
3113
            if (virGetLastError() == NULL)
3114 3115
                qemuReportError(VIR_ERR_OPERATION_FAILED,
                                "%s", _("failed to resume domain"));
J
Jiri Denemark 已提交
3116
            goto out;
3117
        }
3118 3119
        if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0) {
            VIR_WARN("Failed to save status on vm %s", vm->def->name);
J
Jiri Denemark 已提交
3120
            goto out;
3121
        }
3122
    }
J
Jiri Denemark 已提交
3123

3124
    ret = 0;
3125

J
Jiri Denemark 已提交
3126
out:
3127 3128
    if (virSecurityManagerRestoreSavedStateLabel(driver->securityManager,
                                                 vm, path) < 0)
3129 3130
        VIR_WARN("failed to restore save state label on %s", path);

J
Jiri Denemark 已提交
3131 3132 3133
    return ret;
}

3134 3135 3136 3137
static int
qemuDomainRestore(virConnectPtr conn,
                  const char *path)
{
J
Jiri Denemark 已提交
3138 3139 3140 3141 3142 3143 3144 3145 3146
    struct qemud_driver *driver = conn->privateData;
    virDomainDefPtr def = NULL;
    virDomainObjPtr vm = NULL;
    int fd = -1;
    int ret = -1;
    struct qemud_save_header header;

    qemuDriverLock(driver);

3147
    fd = qemuDomainSaveImageOpen(driver, path, &def, &header);
J
Jiri Denemark 已提交
3148 3149 3150 3151 3152 3153 3154 3155 3156
    if (fd < 0)
        goto cleanup;

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

    if (!(vm = virDomainAssignDef(driver->caps,
                                  &driver->domains,
                                  def, true))) {
3157
        /* virDomainAssignDef already set the error */
J
Jiri Denemark 已提交
3158 3159 3160 3161 3162 3163 3164
        goto cleanup;
    }
    def = NULL;

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

3165
    ret = qemuDomainSaveImageStartVM(conn, driver, vm, &fd, &header, path);
J
Jiri Denemark 已提交
3166 3167

    if (qemuDomainObjEndJob(vm) == 0)
3168
        vm = NULL;
J
Jiri Denemark 已提交
3169 3170 3171 3172
    else if (ret < 0 && !vm->persistent) {
        virDomainRemoveInactive(&driver->domains, vm);
        vm = NULL;
    }
3173

3174 3175
cleanup:
    virDomainDefFree(def);
3176
    VIR_FORCE_CLOSE(fd);
3177 3178 3179
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
3180
    return ret;
D
Daniel P. Berrange 已提交
3181 3182
}

3183 3184 3185 3186 3187
static int
qemuDomainObjRestore(virConnectPtr conn,
                     struct qemud_driver *driver,
                     virDomainObjPtr vm,
                     const char *path)
J
Jiri Denemark 已提交
3188 3189 3190 3191 3192 3193
{
    virDomainDefPtr def = NULL;
    int fd = -1;
    int ret = -1;
    struct qemud_save_header header;

3194
    fd = qemuDomainSaveImageOpen(driver, path, &def, &header);
J
Jiri Denemark 已提交
3195 3196 3197 3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208 3209 3210 3211 3212 3213 3214
    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;

3215
    ret = qemuDomainSaveImageStartVM(conn, driver, vm, &fd, &header, path);
J
Jiri Denemark 已提交
3216 3217 3218

cleanup:
    virDomainDefFree(def);
3219
    VIR_FORCE_CLOSE(fd);
J
Jiri Denemark 已提交
3220 3221 3222
    return ret;
}

D
Daniel P. Berrange 已提交
3223

3224
static char *qemudDomainDumpXML(virDomainPtr dom,
3225
                                int flags) {
3226 3227 3228
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *ret = NULL;
3229 3230
    unsigned long balloon;
    int err;
3231

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

D
Daniel P. Berrange 已提交
3235
    if (!vm) {
3236 3237
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
3238 3239
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
3240
        goto cleanup;
D
Daniel P. Berrange 已提交
3241 3242
    }

3243 3244 3245 3246
    /* Refresh current memory based on balloon info if supported */
    if ((vm->def->memballoon != NULL) &&
        (vm->def->memballoon->model != VIR_DOMAIN_MEMBALLOON_MODEL_NONE) &&
        (virDomainObjIsActive(vm))) {
3247
        qemuDomainObjPrivatePtr priv = vm->privateData;
3248 3249 3250
        /* Don't delay if someone's using the monitor, just use
         * existing most recent data instead */
        if (!priv->jobActive) {
3251
            if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
3252 3253
                goto cleanup;

3254
            qemuDomainObjEnterMonitorWithDriver(driver, vm);
3255
            err = qemuMonitorGetBalloonInfo(priv->mon, &balloon);
3256
            qemuDomainObjExitMonitorWithDriver(driver, vm);
3257 3258 3259 3260
            if (qemuDomainObjEndJob(vm) == 0) {
                vm = NULL;
                goto cleanup;
            }
3261 3262 3263
            if (err < 0)
                goto cleanup;
            if (err > 0)
3264
                vm->def->mem.cur_balloon = balloon;
3265 3266
            /* err == 0 indicates no balloon support, so ignore it */
        }
3267
    }
3268

3269
    ret = qemuDomainFormatXML(driver, vm, flags);
3270 3271

cleanup:
3272 3273
    if (vm)
        virDomainObjUnlock(vm);
3274
    qemuDriverUnlock(driver);
3275
    return ret;
D
Daniel P. Berrange 已提交
3276 3277 3278
}


3279 3280 3281 3282
static char *qemuDomainXMLFromNative(virConnectPtr conn,
                                     const char *format,
                                     const char *config,
                                     unsigned int flags ATTRIBUTE_UNUSED) {
3283
    struct qemud_driver *driver = conn->privateData;
3284 3285 3286 3287
    virDomainDefPtr def = NULL;
    char *xml = NULL;

    if (STRNEQ(format, QEMU_CONFIG_FORMAT_ARGV)) {
3288 3289
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("unsupported config type %s"), format);
3290 3291 3292
        goto cleanup;
    }

3293
    qemuDriverLock(driver);
3294
    def = qemuParseCommandLineString(driver->caps, config);
3295
    qemuDriverUnlock(driver);
3296 3297 3298
    if (!def)
        goto cleanup;

3299
    xml = virDomainDefFormat(def, VIR_DOMAIN_XML_INACTIVE);
3300 3301 3302 3303 3304 3305

cleanup:
    virDomainDefFree(def);
    return xml;
}

3306 3307 3308 3309 3310 3311
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;
3312
    virDomainChrSourceDef monConfig;
3313
    virBitmapPtr qemuCaps = NULL;
E
Eric Blake 已提交
3314
    virCommandPtr cmd = NULL;
3315 3316 3317
    char *ret = NULL;
    int i;

3318 3319
    qemuDriverLock(driver);

3320
    if (STRNEQ(format, QEMU_CONFIG_FORMAT_ARGV)) {
3321 3322
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("unsupported config type %s"), format);
3323 3324 3325
        goto cleanup;
    }

3326
    def = virDomainDefParseString(driver->caps, xmlData, 0);
3327 3328 3329
    if (!def)
        goto cleanup;

3330 3331
    /* Since we're just exporting args, we can't do bridge/network/direct
     * setups, since libvirt will normally create TAP/macvtap devices
3332 3333 3334 3335 3336
     * 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];
3337
        int bootIndex = net->bootIndex;
3338 3339
        if (net->type == VIR_DOMAIN_NET_TYPE_NETWORK ||
            net->type == VIR_DOMAIN_NET_TYPE_DIRECT) {
3340 3341 3342 3343 3344 3345 3346 3347 3348 3349 3350 3351 3352 3353 3354 3355 3356 3357 3358 3359
            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;
        }
3360
        net->bootIndex = bootIndex;
3361 3362 3363 3364
    }
    for (i = 0 ; i < def->ngraphics ; i++) {
        if (def->graphics[i]->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
            def->graphics[i]->data.vnc.autoport)
3365
            def->graphics[i]->data.vnc.port = QEMU_VNC_PORT_MIN;
3366 3367
    }

3368
    if (qemuCapsExtractVersionInfo(def->emulator, def->os.arch,
3369
                                   NULL,
3370
                                   &qemuCaps) < 0)
3371 3372
        goto cleanup;

3373
    if (qemuProcessPrepareMonitorChr(driver, &monConfig, def->name) < 0)
3374
        goto cleanup;
3375

3376
    if (!(cmd = qemuBuildCommandLine(conn, driver, def,
3377
                                     &monConfig, false, qemuCaps,
3378
                                     NULL, -1, NULL, VIR_VM_OP_NO_OP)))
3379 3380
        goto cleanup;

E
Eric Blake 已提交
3381
    ret = virCommandToString(cmd);
3382 3383

cleanup:
3384
    qemuDriverUnlock(driver);
3385

3386
    qemuCapsFree(qemuCaps);
E
Eric Blake 已提交
3387
    virCommandFree(cmd);
3388 3389 3390 3391 3392
    virDomainDefFree(def);
    return ret;
}


3393
static int qemudListDefinedDomains(virConnectPtr conn,
3394
                            char **const names, int nnames) {
3395
    struct qemud_driver *driver = conn->privateData;
3396
    int n;
3397

3398
    qemuDriverLock(driver);
3399
    n = virDomainObjListGetInactiveNames(&driver->domains, names, nnames);
3400
    qemuDriverUnlock(driver);
3401
    return n;
D
Daniel P. Berrange 已提交
3402 3403
}

3404
static int qemudNumDefinedDomains(virConnectPtr conn) {
3405
    struct qemud_driver *driver = conn->privateData;
3406
    int n;
3407

3408
    qemuDriverLock(driver);
3409
    n = virDomainObjListNumOfDomains(&driver->domains, 0);
3410
    qemuDriverUnlock(driver);
3411

3412
    return n;
D
Daniel P. Berrange 已提交
3413 3414 3415
}


J
Jiri Denemark 已提交
3416 3417
static int qemudDomainObjStart(virConnectPtr conn,
                               struct qemud_driver *driver,
3418 3419
                               virDomainObjPtr vm,
                               bool start_paused)
J
Jiri Denemark 已提交
3420 3421 3422 3423 3424 3425 3426 3427 3428 3429
{
    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))) {
3430
        ret = qemuDomainObjRestore(conn, driver, vm, managed_save);
J
Jiri Denemark 已提交
3431 3432 3433 3434 3435 3436 3437 3438 3439

        if (unlink(managed_save) < 0) {
            VIR_WARN("Failed to remove the managed state %s", managed_save);
        }

        if (ret == 0)
            goto cleanup;
    }

3440 3441
    ret = qemuProcessStart(conn, driver, vm, NULL, start_paused, -1, NULL,
                           VIR_VM_OP_CREATE);
3442
    qemuAuditDomainStart(vm, "booted", ret >= 0);
3443
    if (ret >= 0) {
J
Jiri Denemark 已提交
3444 3445 3446 3447 3448 3449 3450 3451 3452 3453 3454 3455 3456
        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;
}

3457 3458 3459
static int
qemudDomainStartWithFlags(virDomainPtr dom, unsigned int flags)
{
3460 3461 3462
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
3463

3464
    virCheckFlags(VIR_DOMAIN_START_PAUSED, -1);
3465

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

3469
    if (!vm) {
3470 3471
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
3472 3473
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
3474
        goto cleanup;
3475 3476
    }

3477 3478 3479 3480
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

    if (virDomainObjIsActive(vm)) {
3481 3482
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is already running"));
3483 3484 3485
        goto endjob;
    }

3486 3487
    ret = qemudDomainObjStart(dom->conn, driver, vm,
                              (flags & VIR_DOMAIN_START_PAUSED) != 0);
3488

3489
endjob:
3490 3491
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
3492

3493
cleanup:
3494 3495
    if (vm)
        virDomainObjUnlock(vm);
3496
    qemuDriverUnlock(driver);
3497
    return ret;
D
Daniel P. Berrange 已提交
3498 3499
}

3500 3501 3502 3503 3504 3505
static int
qemudDomainStart(virDomainPtr dom)
{
    return qemudDomainStartWithFlags(dom, 0);
}

3506 3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519 3520
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;

3521
        if (STRNEQ(def->os.machine, machine->name))
3522 3523 3524
            continue;

        if (!(*canonical = strdup(machine->canonical))) {
3525
            virReportOOMError();
3526 3527 3528 3529 3530 3531 3532 3533 3534
            return -1;
        }

        break;
    }

    return 0;
}

3535 3536 3537 3538 3539 3540
static int
qemudCanonicalizeMachineDirect(virDomainDefPtr def, char **canonical)
{
    virCapsGuestMachinePtr *machines = NULL;
    int i, nmachines = 0;

3541
    if (qemuCapsProbeMachineTypes(def->emulator, &machines, &nmachines) < 0)
3542 3543 3544 3545 3546 3547
        return -1;

    for (i = 0; i < nmachines; i++) {
        if (!machines[i]->canonical)
            continue;

3548
        if (STRNEQ(def->os.machine, machines[i]->name))
3549 3550 3551 3552 3553 3554 3555 3556 3557 3558 3559 3560
            continue;

        *canonical = machines[i]->canonical;
        machines[i]->canonical = NULL;
        break;
    }

    virCapabilitiesFreeMachines(machines, nmachines);

    return 0;
}

3561 3562
int
qemudCanonicalizeMachine(struct qemud_driver *driver, virDomainDefPtr def)
3563 3564 3565 3566 3567 3568
{
    char *canonical = NULL;
    int i;

    for (i = 0; i < driver->caps->nguests; i++) {
        virCapsGuestPtr guest = driver->caps->guests[i];
3569
        virCapsGuestDomainInfoPtr info;
3570 3571 3572
        int j;

        for (j = 0; j < guest->arch.ndomains; j++) {
3573
            info = &guest->arch.domains[j]->info;
3574

3575 3576 3577 3578 3579 3580 3581 3582 3583
            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;
3584 3585
        }

3586 3587 3588 3589
        info = &guest->arch.defaultInfo;

        if (info->emulator && STREQ(info->emulator, def->emulator)) {
            if (qemudCanonicalizeMachineFromInfo(def, info, &canonical) < 0)
3590 3591 3592 3593
                return -1;
            goto out;
        }
    }
3594 3595 3596 3597

    if (qemudCanonicalizeMachineDirect(def, &canonical) < 0)
        return -1;

3598 3599 3600 3601 3602 3603 3604
out:
    if (canonical) {
        VIR_FREE(def->os.machine);
        def->os.machine = canonical;
    }
    return 0;
}
D
Daniel P. Berrange 已提交
3605

3606
static virDomainPtr qemudDomainDefine(virConnectPtr conn, const char *xml) {
3607
    struct qemud_driver *driver = conn->privateData;
3608
    virDomainDefPtr def;
3609
    virDomainObjPtr vm = NULL;
3610
    virDomainPtr dom = NULL;
3611
    virDomainEventPtr event = NULL;
3612
    int dupVM;
3613

3614
    qemuDriverLock(driver);
3615
    if (!(def = virDomainDefParseString(driver->caps, xml,
3616
                                        VIR_DOMAIN_XML_INACTIVE)))
3617
        goto cleanup;
3618

3619
    if (virSecurityManagerVerify(driver->securityManager, def) < 0)
3620 3621
        goto cleanup;

3622 3623
    if ((dupVM = virDomainObjIsDuplicate(&driver->domains, def, 0)) < 0)
        goto cleanup;
3624

3625
    if (qemudCanonicalizeMachine(driver, def) < 0)
3626 3627
        goto cleanup;

3628
    if (qemuDomainAssignPCIAddresses(def) < 0)
3629 3630
        goto cleanup;

3631
    if (!(vm = virDomainAssignDef(driver->caps,
3632
                                  &driver->domains,
3633
                                  def, false))) {
3634
        goto cleanup;
3635
    }
3636
    def = NULL;
3637
    vm->persistent = 1;
3638

3639
    if (virDomainSaveConfig(driver->configDir,
3640
                            vm->newDef ? vm->newDef : vm->def) < 0) {
3641
        VIR_INFO(_("Defining domain '%s'"), vm->def->name);
3642 3643
        virDomainRemoveInactive(&driver->domains,
                                vm);
3644
        vm = NULL;
3645
        goto cleanup;
3646 3647
    }

3648 3649
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_DEFINED,
3650
                                     !dupVM ?
3651 3652
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED :
                                     VIR_DOMAIN_EVENT_DEFINED_UPDATED);
3653

3654
    VIR_INFO(_("Creating domain '%s'"), vm->def->name);
3655
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
3656
    if (dom) dom->id = vm->def->id;
3657 3658

cleanup:
3659
    virDomainDefFree(def);
3660 3661
    if (vm)
        virDomainObjUnlock(vm);
3662 3663
    if (event)
        qemuDomainEventQueue(driver, event);
3664
    qemuDriverUnlock(driver);
3665
    return dom;
D
Daniel P. Berrange 已提交
3666 3667
}

3668
static int qemudDomainUndefine(virDomainPtr dom) {
3669 3670
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
3671
    virDomainEventPtr event = NULL;
3672
    int ret = -1;
D
Daniel P. Berrange 已提交
3673

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

D
Daniel P. Berrange 已提交
3677
    if (!vm) {
3678 3679
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
3680 3681
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
3682
        goto cleanup;
D
Daniel P. Berrange 已提交
3683 3684
    }

D
Daniel P. Berrange 已提交
3685
    if (virDomainObjIsActive(vm)) {
3686 3687
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot delete active domain"));
3688
        goto cleanup;
D
Daniel P. Berrange 已提交
3689 3690
    }

3691
    if (!vm->persistent) {
3692
        qemuReportError(VIR_ERR_OPERATION_INVALID,
3693
                        "%s", _("cannot undefine transient domain"));
3694
        goto cleanup;
3695 3696
    }

3697
    if (virDomainDeleteConfig(driver->configDir, driver->autostartDir, vm) < 0)
3698
        goto cleanup;
D
Daniel P. Berrange 已提交
3699

3700 3701 3702
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_UNDEFINED,
                                     VIR_DOMAIN_EVENT_UNDEFINED_REMOVED);
3703

3704
    VIR_INFO(_("Undefining domain '%s'"), vm->def->name);
3705 3706
    virDomainRemoveInactive(&driver->domains,
                            vm);
3707
    vm = NULL;
3708
    ret = 0;
D
Daniel P. Berrange 已提交
3709

3710
cleanup:
3711 3712
    if (vm)
        virDomainObjUnlock(vm);
3713 3714
    if (event)
        qemuDomainEventQueue(driver, event);
3715
    qemuDriverUnlock(driver);
3716
    return ret;
D
Daniel P. Berrange 已提交
3717 3718
}

3719

3720 3721
static int qemudDomainAttachDevice(virDomainPtr dom,
                                   const char *xml)
3722
{
3723 3724 3725
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    virDomainDeviceDefPtr dev = NULL;
3726
    virBitmapPtr qemuCaps = NULL;
3727
    virCgroupPtr cgroup = NULL;
3728
    int ret = -1;
3729

3730 3731 3732 3733 3734 3735 3736
    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);
3737 3738 3739
        goto cleanup;
    }

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

3743 3744 3745 3746 3747
    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot attach device on inactive domain"));
        goto endjob;
    }
3748

3749 3750 3751 3752
    dev = virDomainDeviceDefParse(driver->caps, vm->def, xml,
                                  VIR_DOMAIN_XML_INACTIVE);
    if (dev == NULL)
        goto endjob;
3753

3754
    if (qemuCapsExtractVersionInfo(vm->def->emulator, vm->def->os.arch,
3755
                                   NULL,
3756
                                   &qemuCaps) < 0)
3757
        goto endjob;
3758

3759
    if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
3760 3761 3762 3763 3764 3765 3766 3767
        if (dev->data.disk->driverName != NULL &&
            !STREQ(dev->data.disk->driverName, "qemu")) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("unsupported driver name '%s' for disk '%s'"),
                            dev->data.disk->driverName, dev->data.disk->src);
            goto endjob;
        }

3768 3769 3770
        if (qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_DEVICES)) {
            if (virCgroupForDomain(driver->cgroup, vm->def->name, &cgroup, 0) !=0 ) {
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
E
Eric Blake 已提交
3771
                                _("Unable to find cgroup for %s"),
3772 3773 3774
                                vm->def->name);
                goto endjob;
            }
3775
            if (qemuSetupDiskCgroup(driver, vm, cgroup, dev->data.disk) < 0)
3776
                goto endjob;
3777 3778
        }

3779 3780 3781 3782 3783
        switch (dev->data.disk->device) {
        case VIR_DOMAIN_DISK_DEVICE_CDROM:
        case VIR_DOMAIN_DISK_DEVICE_FLOPPY:
            ret = qemuDomainChangeEjectableMedia(driver, vm,
                                                 dev->data.disk,
3784
                                                 qemuCaps,
3785 3786 3787 3788
                                                 false);
            if (ret == 0)
                dev->data.disk = NULL;
            break;
3789

3790 3791 3792
        case VIR_DOMAIN_DISK_DEVICE_DISK:
            if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_USB) {
                ret = qemuDomainAttachUsbMassstorageDevice(driver, vm,
3793
                                                           dev->data.disk, qemuCaps);
3794 3795 3796 3797
                if (ret == 0)
                    dev->data.disk = NULL;
            } else if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_VIRTIO) {
                ret = qemuDomainAttachPciDiskDevice(driver, vm,
3798
                                                    dev->data.disk, qemuCaps);
3799 3800 3801 3802
                if (ret == 0)
                    dev->data.disk = NULL;
            } else if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_SCSI) {
                ret = qemuDomainAttachSCSIDisk(driver, vm,
3803
                                               dev->data.disk, qemuCaps);
3804 3805 3806 3807 3808 3809 3810 3811 3812
                if (ret == 0)
                    dev->data.disk = NULL;
            } else {
                qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                                _("disk bus '%s' cannot be hotplugged."),
                                virDomainDiskBusTypeToString(dev->data.disk->bus));
                /* fallthrough */
            }
            break;
3813

3814 3815 3816 3817 3818
        default:
            qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                            _("disk device type '%s' cannot be hotplugged"),
                            virDomainDiskDeviceTypeToString(dev->data.disk->device));
            /* Fallthrough */
3819
        }
3820
        if (ret != 0 && cgroup) {
3821
            if (qemuTeardownDiskCgroup(driver, vm, cgroup, dev->data.disk) < 0)
3822 3823
                VIR_WARN("Failed to teardown cgroup for disk path %s",
                         NULLSTR(dev->data.disk->src));
3824
        }
3825 3826 3827
    } else if (dev->type == VIR_DOMAIN_DEVICE_CONTROLLER) {
        if (dev->data.controller->type == VIR_DOMAIN_CONTROLLER_TYPE_SCSI) {
            ret = qemuDomainAttachPciControllerDevice(driver, vm,
3828
                                                      dev->data.controller, qemuCaps);
3829 3830 3831 3832 3833 3834 3835
            if (ret == 0)
                dev->data.controller = NULL;
        } else {
            qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                            _("disk controller bus '%s' cannot be hotplugged."),
                            virDomainControllerTypeToString(dev->data.controller->type));
            /* fallthrough */
3836
        }
3837 3838
    } else if (dev->type == VIR_DOMAIN_DEVICE_NET) {
        ret = qemuDomainAttachNetDevice(dom->conn, driver, vm,
3839
                                        dev->data.net, qemuCaps);
3840 3841 3842 3843
        if (ret == 0)
            dev->data.net = NULL;
    } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV) {
        ret = qemuDomainAttachHostDevice(driver, vm,
3844
                                         dev->data.hostdev, qemuCaps);
3845 3846
        if (ret == 0)
            dev->data.hostdev = NULL;
3847
    } else {
3848 3849 3850 3851
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                        _("device type '%s' cannot be attached"),
                        virDomainDeviceTypeToString(dev->type));
        goto endjob;
3852 3853
    }

3854 3855 3856 3857 3858
    /* update domain status forcibly because the domain status may be changed
     * even if we attach the device failed. For example, a new controller may
     * be created.
     */
    if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
3859 3860 3861 3862 3863 3864 3865 3866 3867 3868
        ret = -1;

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

cleanup:
    if (cgroup)
        virCgroupFree(&cgroup);

3869
    qemuCapsFree(qemuCaps);
3870 3871 3872 3873
    virDomainDeviceDefFree(dev);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
3874 3875 3876
    return ret;
}

3877 3878 3879 3880 3881 3882 3883 3884
static int qemudDomainAttachDeviceFlags(virDomainPtr dom,
                                        const char *xml,
                                        unsigned int flags) {
    if (flags & VIR_DOMAIN_DEVICE_MODIFY_CONFIG) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot modify the persistent configuration of a domain"));
        return -1;
    }
3885

3886 3887
    return qemudDomainAttachDevice(dom, xml);
}
3888

3889

3890 3891 3892 3893 3894 3895 3896
static int qemuDomainUpdateDeviceFlags(virDomainPtr dom,
                                       const char *xml,
                                       unsigned int flags)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    virDomainDeviceDefPtr dev = NULL;
3897
    virBitmapPtr qemuCaps = NULL;
3898 3899 3900
    virCgroupPtr cgroup = NULL;
    int ret = -1;
    bool force = (flags & VIR_DOMAIN_DEVICE_MODIFY_FORCE) != 0;
3901

3902 3903 3904 3905
    virCheckFlags(VIR_DOMAIN_DEVICE_MODIFY_CURRENT |
                  VIR_DOMAIN_DEVICE_MODIFY_LIVE |
                  VIR_DOMAIN_DEVICE_MODIFY_CONFIG |
                  VIR_DOMAIN_DEVICE_MODIFY_FORCE, -1);
3906

3907 3908 3909
    if (flags & VIR_DOMAIN_DEVICE_MODIFY_CONFIG) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot modify the persistent configuration of a domain"));
3910 3911 3912
        return -1;
    }

3913 3914 3915 3916 3917 3918 3919 3920
    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;
3921 3922
    }

3923 3924 3925 3926 3927 3928 3929
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;

    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot attach device on inactive domain"));
        goto endjob;
3930 3931
    }

3932 3933 3934 3935 3936
    dev = virDomainDeviceDefParse(driver->caps, vm->def, xml,
                                  VIR_DOMAIN_XML_INACTIVE);
    if (dev == NULL)
        goto endjob;

3937
    if (qemuCapsExtractVersionInfo(vm->def->emulator, vm->def->os.arch,
3938
                                   NULL,
3939
                                   &qemuCaps) < 0)
3940
        goto endjob;
3941

3942 3943 3944 3945 3946
    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,
E
Eric Blake 已提交
3947
                                _("Unable to find cgroup for %s"),
3948 3949 3950
                                vm->def->name);
                goto endjob;
            }
3951
            if (qemuSetupDiskCgroup(driver, vm, cgroup, dev->data.disk) < 0)
3952
                goto endjob;
3953 3954
        }

3955 3956 3957 3958 3959
        switch (dev->data.disk->device) {
        case VIR_DOMAIN_DISK_DEVICE_CDROM:
        case VIR_DOMAIN_DISK_DEVICE_FLOPPY:
            ret = qemuDomainChangeEjectableMedia(driver, vm,
                                                 dev->data.disk,
3960
                                                 qemuCaps,
3961 3962 3963 3964
                                                 force);
            if (ret == 0)
                dev->data.disk = NULL;
            break;
3965

3966

3967 3968 3969 3970 3971 3972
        default:
            qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                            _("disk bus '%s' cannot be updated."),
                            virDomainDiskBusTypeToString(dev->data.disk->bus));
            break;
        }
3973

3974
        if (ret != 0 && cgroup) {
3975
            if (qemuTeardownDiskCgroup(driver, vm, cgroup, dev->data.disk) < 0)
3976 3977 3978
                VIR_WARN("Failed to teardown cgroup for disk path %s",
                         NULLSTR(dev->data.disk->src));
        }
3979
        break;
3980 3981 3982

    case VIR_DOMAIN_DEVICE_GRAPHICS:
        ret = qemuDomainChangeGraphics(driver, vm, dev->data.graphics);
3983
        break;
3984

3985
    default:
3986
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
3987 3988
                        _("device type '%s' cannot be updated"),
                        virDomainDeviceTypeToString(dev->type));
3989
        break;
3990 3991
    }

3992 3993 3994 3995 3996 3997 3998 3999 4000 4001
    if (!ret && virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
        ret = -1;

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

cleanup:
    if (cgroup)
        virCgroupFree(&cgroup);
4002

4003
    qemuCapsFree(qemuCaps);
4004 4005 4006 4007
    virDomainDeviceDefFree(dev);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
4008 4009 4010
    return ret;
}

4011

4012 4013
static int qemudDomainDetachDevice(virDomainPtr dom,
                                   const char *xml) {
4014 4015
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
4016
    virBitmapPtr qemuCaps = NULL;
4017 4018
    virDomainDeviceDefPtr dev = NULL;
    int ret = -1;
4019

4020
    qemuDriverLock(driver);
4021
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4022
    if (!vm) {
4023 4024
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4025 4026
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4027
        goto cleanup;
4028 4029
    }

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

D
Daniel P. Berrange 已提交
4033
    if (!virDomainObjIsActive(vm)) {
4034 4035
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot detach device on inactive domain"));
4036
        goto endjob;
4037 4038
    }

4039
    dev = virDomainDeviceDefParse(driver->caps, vm->def, xml,
4040
                                  VIR_DOMAIN_XML_INACTIVE);
4041
    if (dev == NULL)
4042
        goto endjob;
4043

4044
    if (qemuCapsExtractVersionInfo(vm->def->emulator, vm->def->os.arch,
4045
                                   NULL,
4046
                                   &qemuCaps) < 0)
4047
        goto endjob;
4048 4049

    if (dev->type == VIR_DOMAIN_DEVICE_DISK &&
4050 4051
        dev->data.disk->device == VIR_DOMAIN_DISK_DEVICE_DISK) {
        if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_VIRTIO) {
4052
            ret = qemuDomainDetachPciDiskDevice(driver, vm, dev, qemuCaps);
4053 4054
        }
        else if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_SCSI) {
4055
            ret = qemuDomainDetachDiskDevice(driver, vm, dev, qemuCaps);
W
Wen Congyang 已提交
4056 4057
        } else if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_USB) {
            ret = qemuDomainDetachDiskDevice(driver, vm, dev, qemuCaps);
4058 4059
        }
        else {
4060
            qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
4061 4062
                            _("This type of disk cannot be hot unplugged"));
        }
4063
    } else if (dev->type == VIR_DOMAIN_DEVICE_NET) {
4064
        ret = qemuDomainDetachNetDevice(driver, vm, dev, qemuCaps);
4065 4066
    } else if (dev->type == VIR_DOMAIN_DEVICE_CONTROLLER) {
        if (dev->data.controller->type == VIR_DOMAIN_CONTROLLER_TYPE_SCSI) {
4067
            ret = qemuDomainDetachPciControllerDevice(driver, vm, dev,
4068
                                                      qemuCaps);
4069
        } else {
4070
            qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
4071 4072
                            _("disk controller bus '%s' cannot be hotunplugged."),
                            virDomainControllerTypeToString(dev->data.controller->type));
4073 4074
            /* fallthrough */
        }
4075
    } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV) {
4076
        ret = qemuDomainDetachHostDevice(driver, vm, dev, qemuCaps);
4077
    } else {
4078
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
4079
                        "%s", _("This type of device cannot be hot unplugged"));
4080
    }
4081

4082
    if (!ret && virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
4083 4084
        ret = -1;

4085
endjob:
4086 4087
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
4088

4089
cleanup:
4090
    qemuCapsFree(qemuCaps);
4091
    virDomainDeviceDefFree(dev);
4092 4093
    if (vm)
        virDomainObjUnlock(vm);
4094
    qemuDriverUnlock(driver);
4095 4096 4097
    return ret;
}

4098 4099 4100 4101
static int qemudDomainDetachDeviceFlags(virDomainPtr dom,
                                        const char *xml,
                                        unsigned int flags) {
    if (flags & VIR_DOMAIN_DEVICE_MODIFY_CONFIG) {
4102 4103
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot modify the persistent configuration of a domain"));
4104 4105 4106 4107 4108 4109
        return -1;
    }

    return qemudDomainDetachDevice(dom, xml);
}

4110
static int qemudDomainGetAutostart(virDomainPtr dom,
4111
                                   int *autostart) {
4112 4113 4114
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
4115

4116
    qemuDriverLock(driver);
4117
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4118 4119
    qemuDriverUnlock(driver);

4120
    if (!vm) {
4121 4122
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4123 4124
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4125
        goto cleanup;
4126 4127 4128
    }

    *autostart = vm->autostart;
4129
    ret = 0;
4130

4131
cleanup:
4132 4133
    if (vm)
        virDomainObjUnlock(vm);
4134
    return ret;
4135 4136
}

4137
static int qemudDomainSetAutostart(virDomainPtr dom,
4138
                                   int autostart) {
4139 4140
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
4141 4142
    char *configFile = NULL, *autostartLink = NULL;
    int ret = -1;
4143

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

4147
    if (!vm) {
4148 4149
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4150 4151
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4152
        goto cleanup;
4153 4154
    }

4155
    if (!vm->persistent) {
4156
        qemuReportError(VIR_ERR_OPERATION_INVALID,
4157
                        "%s", _("cannot set autostart for transient domain"));
4158
        goto cleanup;
4159 4160
    }

4161 4162
    autostart = (autostart != 0);

4163
    if (vm->autostart != autostart) {
4164
        if ((configFile = virDomainConfigFile(driver->configDir, vm->def->name)) == NULL)
4165
            goto cleanup;
4166
        if ((autostartLink = virDomainConfigFile(driver->autostartDir, vm->def->name)) == NULL)
4167
            goto cleanup;
4168

4169 4170
        if (autostart) {
            int err;
4171

4172
            if ((err = virFileMakePath(driver->autostartDir))) {
4173
                virReportSystemError(err,
4174 4175
                                     _("cannot create autostart directory %s"),
                                     driver->autostartDir);
4176 4177
                goto cleanup;
            }
4178

4179
            if (symlink(configFile, autostartLink) < 0) {
4180
                virReportSystemError(errno,
4181 4182
                                     _("Failed to create symlink '%s to '%s'"),
                                     autostartLink, configFile);
4183 4184 4185 4186
                goto cleanup;
            }
        } else {
            if (unlink(autostartLink) < 0 && errno != ENOENT && errno != ENOTDIR) {
4187
                virReportSystemError(errno,
4188 4189
                                     _("Failed to delete symlink '%s'"),
                                     autostartLink);
4190 4191
                goto cleanup;
            }
4192 4193
        }

4194
        vm->autostart = autostart;
4195
    }
4196
    ret = 0;
4197

4198 4199 4200
cleanup:
    VIR_FREE(configFile);
    VIR_FREE(autostartLink);
4201 4202
    if (vm)
        virDomainObjUnlock(vm);
4203
    qemuDriverUnlock(driver);
4204
    return ret;
4205 4206
}

4207 4208 4209 4210 4211

static char *qemuGetSchedulerType(virDomainPtr dom,
                                  int *nparams)
{
    struct qemud_driver *driver = dom->conn->privateData;
4212
    char *ret = NULL;
4213

4214
    qemuDriverLock(driver);
4215
    if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_CPU)) {
4216 4217
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cgroup CPU controller is not mounted"));
4218
        goto cleanup;
4219 4220 4221 4222 4223 4224 4225
    }

    if (nparams)
        *nparams = 1;

    ret = strdup("posix");
    if (!ret)
4226
        virReportOOMError();
4227 4228 4229

cleanup:
    qemuDriverUnlock(driver);
4230 4231 4232
    return ret;
}

4233 4234 4235 4236 4237 4238 4239 4240 4241 4242 4243 4244 4245 4246 4247 4248 4249 4250 4251 4252 4253 4254 4255 4256 4257 4258 4259 4260 4261 4262 4263 4264 4265 4266 4267 4268 4269 4270 4271 4272 4273 4274 4275 4276 4277 4278 4279 4280 4281 4282 4283 4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294 4295 4296 4297 4298 4299 4300 4301 4302 4303 4304 4305 4306 4307 4308 4309 4310 4311 4312 4313 4314 4315 4316 4317 4318 4319 4320 4321 4322 4323 4324 4325 4326 4327 4328 4329 4330 4331 4332 4333 4334 4335 4336 4337 4338 4339 4340 4341 4342 4343 4344 4345 4346 4347 4348 4349 4350 4351 4352 4353 4354 4355 4356 4357 4358 4359 4360 4361 4362 4363 4364 4365 4366 4367 4368 4369 4370 4371 4372 4373 4374 4375 4376 4377 4378 4379 4380 4381 4382 4383 4384 4385 4386 4387 4388 4389 4390 4391
static int qemuDomainSetBlkioParameters(virDomainPtr dom,
                                         virBlkioParameterPtr params,
                                         int nparams,
                                         unsigned int flags)
{
    struct qemud_driver *driver = dom->conn->privateData;
    int i;
    virCgroupPtr group = NULL;
    virDomainObjPtr vm = NULL;
    int ret = -1;

    virCheckFlags(0, -1);
    qemuDriverLock(driver);
    if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_BLKIO)) {
        qemuReportError(VIR_ERR_NO_SUPPORT, _("blkio cgroup isn't mounted"));
        goto cleanup;
    }

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

    if (vm == NULL) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("No such domain %s"), dom->uuid);
        goto cleanup;
    }

    if (virCgroupForDomain(driver->cgroup, vm->def->name, &group, 0) != 0) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("cannot find cgroup for domain %s"), vm->def->name);
        goto cleanup;
    }

    ret = 0;
    for (i = 0; i < nparams; i++) {
        virBlkioParameterPtr param = &params[i];

        if (STREQ(param->field, VIR_DOMAIN_BLKIO_WEIGHT)) {
            int rc;
            if (param->type != VIR_DOMAIN_BLKIO_PARAM_UINT) {
                qemuReportError(VIR_ERR_INVALID_ARG, "%s",
                                _("invalid type for blkio weight tunable, expected a 'unsigned int'"));
                ret = -1;
                continue;
            }

            if (params[i].value.ui > 1000 || params[i].value.ui < 100) {
                qemuReportError(VIR_ERR_INVALID_ARG, "%s",
                                _("out of blkio weight range."));
                ret = -1;
                continue;
            }

            rc = virCgroupSetBlkioWeight(group, params[i].value.ui);
            if (rc != 0) {
                virReportSystemError(-rc, "%s",
                                     _("unable to set blkio weight tunable"));
                ret = -1;
            }
        } else {
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("Parameter `%s' not supported"), param->field);
            ret = -1;
        }
    }

cleanup:
    virCgroupFree(&group);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}

static int qemuDomainGetBlkioParameters(virDomainPtr dom,
                                         virBlkioParameterPtr params,
                                         int *nparams,
                                         unsigned int flags)
{
    struct qemud_driver *driver = dom->conn->privateData;
    int i;
    virCgroupPtr group = NULL;
    virDomainObjPtr vm = NULL;
    unsigned int val;
    int ret = -1;
    int rc;

    virCheckFlags(0, -1);
    qemuDriverLock(driver);

    if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_BLKIO)) {
        qemuReportError(VIR_ERR_NO_SUPPORT, _("blkio cgroup isn't mounted"));
        goto cleanup;
    }

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

    if (vm == NULL) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("No such domain %s"), dom->uuid);
        goto cleanup;
    }

    if ((*nparams) == 0) {
        /* Current number of blkio parameters supported by cgroups */
        *nparams = QEMU_NB_BLKIO_PARAM;
        ret = 0;
        goto cleanup;
    }

    if ((*nparams) != QEMU_NB_BLKIO_PARAM) {
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("Invalid parameter count"));
        goto cleanup;
    }

    if (virCgroupForDomain(driver->cgroup, vm->def->name, &group, 0) != 0) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("cannot find cgroup for domain %s"), vm->def->name);
        goto cleanup;
    }

    for (i = 0; i < *nparams; i++) {
        virBlkioParameterPtr param = &params[i];
        val = 0;
        param->value.ui = 0;
        param->type = VIR_DOMAIN_BLKIO_PARAM_UINT;

        switch(i) {
        case 0: /* fill blkio weight here */
            rc = virCgroupGetBlkioWeight(group, &val);
            if (rc != 0) {
                virReportSystemError(-rc, "%s",
                                     _("unable to get blkio weight"));
                goto cleanup;
            }
            if (virStrcpyStatic(param->field, VIR_DOMAIN_BLKIO_WEIGHT) == NULL) {
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                "%s", _("Field blkio weight too long for destination"));
                goto cleanup;
            }
            param->value.ui = val;
            break;

        default:
            break;
            /* should not hit here */
        }
    }

    ret = 0;

cleanup:
    if (group)
        virCgroupFree(&group);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}
4392 4393 4394 4395 4396 4397 4398 4399 4400 4401 4402 4403 4404 4405

static int qemuDomainSetMemoryParameters(virDomainPtr dom,
                                         virMemoryParameterPtr params,
                                         int nparams,
                                         unsigned int flags ATTRIBUTE_UNUSED)
{
    struct qemud_driver *driver = dom->conn->privateData;
    int i;
    virCgroupPtr group = NULL;
    virDomainObjPtr vm = NULL;
    int ret = -1;

    qemuDriverLock(driver);
    if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_MEMORY)) {
4406 4407
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cgroup memory controller is not mounted"));
4408 4409 4410 4411 4412 4413 4414 4415 4416 4417 4418 4419 4420 4421 4422 4423 4424 4425 4426 4427 4428 4429 4430 4431 4432 4433 4434 4435 4436 4437 4438 4439 4440 4441 4442 4443 4444 4445 4446 4447 4448 4449 4450 4451 4452 4453 4454 4455 4456 4457 4458
        goto cleanup;
    }

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

    if (vm == NULL) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("No such domain %s"), dom->uuid);
        goto cleanup;
    }

    if (virCgroupForDomain(driver->cgroup, vm->def->name, &group, 0) != 0) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("cannot find cgroup for domain %s"), vm->def->name);
        goto cleanup;
    }

    ret = 0;
    for (i = 0; i < nparams; i++) {
        virMemoryParameterPtr param = &params[i];

        if (STREQ(param->field, VIR_DOMAIN_MEMORY_HARD_LIMIT)) {
            int rc;
            if (param->type != VIR_DOMAIN_MEMORY_PARAM_ULLONG) {
                qemuReportError(VIR_ERR_INVALID_ARG, "%s",
                                _("invalid type for memory hard_limit tunable, expected a 'ullong'"));
                ret = -1;
                continue;
            }

            rc = virCgroupSetMemoryHardLimit(group, params[i].value.ul);
            if (rc != 0) {
                virReportSystemError(-rc, "%s",
                                     _("unable to set memory hard_limit tunable"));
                ret = -1;
            }
        } else if (STREQ(param->field, VIR_DOMAIN_MEMORY_SOFT_LIMIT)) {
            int rc;
            if (param->type != VIR_DOMAIN_MEMORY_PARAM_ULLONG) {
                qemuReportError(VIR_ERR_INVALID_ARG, "%s",
                                _("invalid type for memory soft_limit tunable, expected a 'ullong'"));
                ret = -1;
                continue;
            }

            rc = virCgroupSetMemorySoftLimit(group, params[i].value.ul);
            if (rc != 0) {
                virReportSystemError(-rc, "%s",
                                     _("unable to set memory soft_limit tunable"));
                ret = -1;
            }
4459
        } else if (STREQ(param->field, VIR_DOMAIN_MEMORY_SWAP_HARD_LIMIT)) {
4460 4461 4462 4463 4464 4465 4466 4467
            int rc;
            if (param->type != VIR_DOMAIN_MEMORY_PARAM_ULLONG) {
                qemuReportError(VIR_ERR_INVALID_ARG, "%s",
                                _("invalid type for swap_hard_limit tunable, expected a 'ullong'"));
                ret = -1;
                continue;
            }

4468
            rc = virCgroupSetMemSwapHardLimit(group, params[i].value.ul);
4469 4470 4471 4472 4473 4474 4475 4476 4477 4478 4479 4480 4481 4482 4483 4484 4485 4486 4487 4488 4489 4490 4491 4492
            if (rc != 0) {
                virReportSystemError(-rc, "%s",
                                     _("unable to set swap_hard_limit tunable"));
                ret = -1;
            }
        } else if (STREQ(param->field, VIR_DOMAIN_MEMORY_MIN_GUARANTEE)) {
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("Memory tunable `%s' not implemented"), param->field);
            ret = -1;
        } else {
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("Parameter `%s' not supported"), param->field);
            ret = -1;
        }
    }

cleanup:
    virCgroupFree(&group);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}

4493 4494 4495 4496 4497 4498 4499 4500 4501
static int qemuDomainGetMemoryParameters(virDomainPtr dom,
                                         virMemoryParameterPtr params,
                                         int *nparams,
                                         unsigned int flags ATTRIBUTE_UNUSED)
{
    struct qemud_driver *driver = dom->conn->privateData;
    int i;
    virCgroupPtr group = NULL;
    virDomainObjPtr vm = NULL;
4502
    unsigned long long val;
4503 4504 4505 4506 4507 4508
    int ret = -1;
    int rc;

    qemuDriverLock(driver);

    if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_MEMORY)) {
4509 4510
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cgroup memory controller is not mounted"));
4511 4512 4513 4514 4515 4516 4517 4518 4519 4520 4521 4522 4523 4524 4525 4526 4527 4528 4529 4530 4531 4532 4533 4534 4535 4536 4537 4538 4539 4540 4541 4542 4543 4544 4545 4546 4547 4548 4549 4550 4551 4552
        goto cleanup;
    }

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

    if (vm == NULL) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("No such domain %s"), dom->uuid);
        goto cleanup;
    }

    if ((*nparams) == 0) {
        /* Current number of memory parameters supported by cgroups */
        *nparams = QEMU_NB_MEM_PARAM;
        ret = 0;
        goto cleanup;
    }

    if ((*nparams) != QEMU_NB_MEM_PARAM) {
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("Invalid parameter count"));
        goto cleanup;
    }

    if (virCgroupForDomain(driver->cgroup, vm->def->name, &group, 0) != 0) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("cannot find cgroup for domain %s"), vm->def->name);
        goto cleanup;
    }

    for (i = 0; i < *nparams; i++) {
        virMemoryParameterPtr param = &params[i];
        val = 0;
        param->value.ul = 0;
        param->type = VIR_DOMAIN_MEMORY_PARAM_ULLONG;

        switch(i) {
        case 0: /* fill memory hard limit here */
            rc = virCgroupGetMemoryHardLimit(group, &val);
            if (rc != 0) {
                virReportSystemError(-rc, "%s",
                                     _("unable to get memory hard limit"));
4553
                goto cleanup;
4554 4555 4556 4557
            }
            if (virStrcpyStatic(param->field, VIR_DOMAIN_MEMORY_HARD_LIMIT) == NULL) {
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                "%s", _("Field memory hard limit too long for destination"));
4558
                goto cleanup;
4559 4560 4561 4562 4563 4564 4565 4566 4567
            }
            param->value.ul = val;
            break;

        case 1: /* fill memory soft limit here */
            rc = virCgroupGetMemorySoftLimit(group, &val);
            if (rc != 0) {
                virReportSystemError(-rc, "%s",
                                     _("unable to get memory soft limit"));
4568
                goto cleanup;
4569 4570 4571 4572
            }
            if (virStrcpyStatic(param->field, VIR_DOMAIN_MEMORY_SOFT_LIMIT) == NULL) {
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                "%s", _("Field memory soft limit too long for destination"));
4573
                goto cleanup;
4574 4575 4576 4577 4578
            }
            param->value.ul = val;
            break;

        case 2: /* fill swap hard limit here */
4579
            rc = virCgroupGetMemSwapHardLimit(group, &val);
4580 4581 4582
            if (rc != 0) {
                virReportSystemError(-rc, "%s",
                                     _("unable to get swap hard limit"));
4583
                goto cleanup;
4584
            }
4585
            if (virStrcpyStatic(param->field, VIR_DOMAIN_MEMORY_SWAP_HARD_LIMIT) == NULL) {
4586 4587
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                "%s", _("Field swap hard limit too long for destination"));
4588
                goto cleanup;
4589 4590 4591 4592 4593 4594 4595 4596 4597 4598
            }
            param->value.ul = val;
            break;

        default:
            break;
            /* should not hit here */
        }
    }

4599 4600
    ret = 0;

4601 4602 4603 4604 4605 4606 4607 4608 4609
cleanup:
    if (group)
        virCgroupFree(&group);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}

4610 4611 4612 4613 4614 4615 4616 4617 4618 4619
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;

4620
    qemuDriverLock(driver);
4621
    if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_CPU)) {
4622 4623
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cgroup CPU controller is not mounted"));
4624
        goto cleanup;
4625 4626 4627 4628 4629
    }

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

    if (vm == NULL) {
4630 4631
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("No such domain %s"), dom->uuid);
4632 4633 4634 4635
        goto cleanup;
    }

    if (virCgroupForDomain(driver->cgroup, vm->def->name, &group, 0) != 0) {
4636 4637
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("cannot find cgroup for domain %s"), vm->def->name);
4638 4639 4640 4641 4642 4643 4644 4645 4646
        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) {
4647 4648
                qemuReportError(VIR_ERR_INVALID_ARG, "%s",
                                _("invalid type for cpu_shares tunable, expected a 'ullong'"));
4649 4650 4651 4652 4653
                goto cleanup;
            }

            rc = virCgroupSetCpuShares(group, params[i].value.ul);
            if (rc != 0) {
4654
                virReportSystemError(-rc, "%s",
4655 4656 4657
                                     _("unable to set cpu shares tunable"));
                goto cleanup;
            }
4658 4659

            vm->def->cputune.shares = params[i].value.ul;
4660
        } else {
4661 4662
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("Invalid parameter `%s'"), param->field);
4663 4664 4665 4666 4667 4668 4669 4670 4671
            goto cleanup;
        }
    }
    ret = 0;

cleanup:
    virCgroupFree(&group);
    if (vm)
        virDomainObjUnlock(vm);
4672
    qemuDriverUnlock(driver);
4673 4674 4675 4676 4677 4678 4679 4680 4681 4682 4683 4684 4685 4686
    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;

4687
    qemuDriverLock(driver);
4688
    if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_CPU)) {
4689 4690
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cgroup CPU controller is not mounted"));
4691
        goto cleanup;
4692 4693 4694
    }

    if ((*nparams) != 1) {
4695 4696
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("Invalid parameter count"));
4697
        goto cleanup;
4698 4699 4700 4701 4702
    }

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

    if (vm == NULL) {
4703 4704
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("No such domain %s"), dom->uuid);
4705 4706 4707 4708
        goto cleanup;
    }

    if (virCgroupForDomain(driver->cgroup, vm->def->name, &group, 0) != 0) {
4709 4710
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("cannot find cgroup for domain %s"), vm->def->name);
4711 4712 4713 4714 4715
        goto cleanup;
    }

    rc = virCgroupGetCpuShares(group, &val);
    if (rc != 0) {
4716
        virReportSystemError(-rc, "%s",
4717 4718 4719 4720 4721
                             _("unable to get cpu shares tunable"));
        goto cleanup;
    }
    params[0].value.ul = val;
    params[0].type = VIR_DOMAIN_SCHED_FIELD_ULLONG;
C
Chris Lalancette 已提交
4722
    if (virStrcpyStatic(params[0].field, "cpu_shares") == NULL) {
4723 4724
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("Field cpu_shares too long for destination"));
C
Chris Lalancette 已提交
4725 4726
        goto cleanup;
    }
4727 4728 4729 4730 4731 4732 4733

    ret = 0;

cleanup:
    virCgroupFree(&group);
    if (vm)
        virDomainObjUnlock(vm);
4734
    qemuDriverUnlock(driver);
4735 4736 4737 4738
    return ret;
}


4739 4740 4741 4742 4743 4744 4745 4746 4747
/* 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)
{
4748
    struct qemud_driver *driver = dom->conn->privateData;
4749
    int i, ret = -1;
4750
    virDomainObjPtr vm;
4751
    virDomainDiskDefPtr disk = NULL;
4752
    qemuDomainObjPrivatePtr priv;
4753

4754
    qemuDriverLock(driver);
4755
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4756
    qemuDriverUnlock(driver);
4757
    if (!vm) {
4758 4759
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4760 4761
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4762
        goto cleanup;
4763
    }
4764 4765 4766 4767

    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
4768
    if (!virDomainObjIsActive (vm)) {
4769 4770
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
4771
        goto endjob;
4772 4773
    }

4774 4775 4776 4777 4778 4779 4780 4781
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (STREQ(path, vm->def->disks[i]->dst)) {
            disk = vm->def->disks[i];
            break;
        }
    }

    if (!disk) {
4782 4783
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("invalid path: %s"), path);
4784
        goto endjob;
4785 4786
    }

4787
    if (!disk->info.alias) {
4788 4789
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("missing disk device alias name for %s"), disk->dst);
4790
        goto endjob;
4791
    }
4792

4793
    priv = vm->privateData;
4794 4795
    qemuDomainObjEnterMonitor(vm);
    ret = qemuMonitorGetBlockStatsInfo(priv->mon,
4796
                                       disk->info.alias,
4797 4798 4799 4800 4801 4802
                                       &stats->rd_req,
                                       &stats->rd_bytes,
                                       &stats->wr_req,
                                       &stats->wr_bytes,
                                       &stats->errs);
    qemuDomainObjExitMonitor(vm);
4803

4804
endjob:
4805 4806
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
4807

4808
cleanup:
4809 4810
    if (vm)
        virDomainObjUnlock(vm);
4811
    return ret;
4812 4813
}

4814
#ifdef __linux__
4815 4816 4817 4818 4819
static int
qemudDomainInterfaceStats (virDomainPtr dom,
                           const char *path,
                           struct _virDomainInterfaceStats *stats)
{
4820 4821
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
4822
    int i;
4823
    int ret = -1;
4824

4825
    qemuDriverLock(driver);
4826
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4827 4828
    qemuDriverUnlock(driver);

4829
    if (!vm) {
4830 4831
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4832 4833
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4834
        goto cleanup;
4835 4836
    }

D
Daniel P. Berrange 已提交
4837
    if (!virDomainObjIsActive(vm)) {
4838 4839
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
4840
        goto cleanup;
4841 4842 4843
    }

    /* Check the path is one of the domain's network interfaces. */
4844 4845
    for (i = 0 ; i < vm->def->nnets ; i++) {
        if (vm->def->nets[i]->ifname &&
4846 4847 4848 4849
            STREQ (vm->def->nets[i]->ifname, path)) {
            ret = 0;
            break;
        }
4850 4851
    }

4852
    if (ret == 0)
4853
        ret = linuxDomainInterfaceStats(path, stats);
4854
    else
4855 4856
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("invalid path, '%s' is not a known interface"), path);
4857

4858
cleanup:
4859 4860
    if (vm)
        virDomainObjUnlock(vm);
4861 4862
    return ret;
}
4863
#else
4864 4865 4866 4867
static int
qemudDomainInterfaceStats (virDomainPtr dom,
                           const char *path ATTRIBUTE_UNUSED,
                           struct _virDomainInterfaceStats *stats ATTRIBUTE_UNUSED)
4868
{
4869 4870
    qemuReportError(VIR_ERR_NO_SUPPORT,
                    "%s", __FUNCTION__);
4871 4872
    return -1;
}
4873
#endif
4874

4875 4876 4877 4878 4879 4880 4881 4882 4883 4884 4885 4886 4887 4888 4889 4890
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);
4891 4892
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4893 4894 4895
        goto cleanup;
    }

4896 4897 4898
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

4899 4900 4901
    if (virDomainObjIsActive(vm)) {
        qemuDomainObjPrivatePtr priv = vm->privateData;
        qemuDomainObjEnterMonitor(vm);
4902
        ret = qemuMonitorGetMemoryStats(priv->mon, stats, nr_stats);
4903 4904
        qemuDomainObjExitMonitor(vm);
    } else {
4905 4906
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
4907 4908
    }

4909 4910 4911
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;

4912 4913 4914 4915 4916 4917
cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    return ret;
}

4918 4919 4920 4921 4922 4923 4924
static int
qemudDomainBlockPeek (virDomainPtr dom,
                      const char *path,
                      unsigned long long offset, size_t size,
                      void *buffer,
                      unsigned int flags ATTRIBUTE_UNUSED)
{
4925 4926 4927
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int fd = -1, ret = -1, i;
4928

4929
    qemuDriverLock(driver);
4930
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4931 4932
    qemuDriverUnlock(driver);

4933
    if (!vm) {
4934 4935
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4936 4937
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4938
        goto cleanup;
4939 4940 4941
    }

    if (!path || path[0] == '\0') {
4942 4943
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("NULL or empty path"));
4944
        goto cleanup;
4945 4946 4947
    }

    /* Check the path belongs to this domain. */
4948 4949
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (vm->def->disks[i]->src != NULL &&
4950 4951 4952 4953
            STREQ (vm->def->disks[i]->src, path)) {
            ret = 0;
            break;
        }
4954 4955
    }

4956 4957 4958 4959 4960
    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) {
4961 4962
            virReportSystemError(errno,
                                 _("%s: failed to open"), path);
4963 4964
            goto cleanup;
        }
4965

4966 4967 4968 4969 4970 4971
        /* 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) {
4972 4973
            virReportSystemError(errno,
                                 _("%s: failed to seek or read"), path);
4974 4975 4976 4977 4978
            goto cleanup;
        }

        ret = 0;
    } else {
4979 4980
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("invalid path"));
4981 4982
    }

4983
cleanup:
4984
    VIR_FORCE_CLOSE(fd);
4985 4986
    if (vm)
        virDomainObjUnlock(vm);
4987 4988 4989
    return ret;
}

R
Richard W.M. Jones 已提交
4990 4991 4992 4993 4994 4995
static int
qemudDomainMemoryPeek (virDomainPtr dom,
                       unsigned long long offset, size_t size,
                       void *buffer,
                       unsigned int flags)
{
4996 4997
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
4998
    char *tmp = NULL;
R
Richard W.M. Jones 已提交
4999
    int fd = -1, ret = -1;
5000
    qemuDomainObjPrivatePtr priv;
R
Richard W.M. Jones 已提交
5001

5002
    qemuDriverLock(driver);
5003
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
5004
    qemuDriverUnlock(driver);
R
Richard W.M. Jones 已提交
5005 5006

    if (!vm) {
5007 5008
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
5009 5010
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
5011 5012 5013
        goto cleanup;
    }

5014
    if (flags != VIR_MEMORY_VIRTUAL && flags != VIR_MEMORY_PHYSICAL) {
5015 5016
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("flags parameter must be VIR_MEMORY_VIRTUAL or VIR_MEMORY_PHYSICAL"));
5017
        goto cleanup;
R
Richard W.M. Jones 已提交
5018 5019
    }

5020 5021 5022
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
5023
    if (!virDomainObjIsActive(vm)) {
5024 5025
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
5026
        goto endjob;
R
Richard W.M. Jones 已提交
5027 5028
    }

5029
    if (virAsprintf(&tmp, "%s/qemu.mem.XXXXXX", driver->cacheDir) < 0) {
5030
        virReportOOMError();
5031
        goto endjob;
5032 5033
    }

R
Richard W.M. Jones 已提交
5034 5035
    /* Create a temporary filename. */
    if ((fd = mkstemp (tmp)) == -1) {
5036 5037
        virReportSystemError(errno,
                             _("mkstemp(\"%s\") failed"), tmp);
5038
        goto endjob;
R
Richard W.M. Jones 已提交
5039 5040
    }

5041
    priv = vm->privateData;
5042
    qemuDomainObjEnterMonitor(vm);
5043
    if (flags == VIR_MEMORY_VIRTUAL) {
5044 5045
        if (qemuMonitorSaveVirtualMemory(priv->mon, offset, size, tmp) < 0) {
            qemuDomainObjExitMonitor(vm);
5046
            goto endjob;
5047
        }
5048
    } else {
5049 5050
        if (qemuMonitorSavePhysicalMemory(priv->mon, offset, size, tmp) < 0) {
            qemuDomainObjExitMonitor(vm);
5051
            goto endjob;
5052
        }
R
Richard W.M. Jones 已提交
5053
    }
5054
    qemuDomainObjExitMonitor(vm);
R
Richard W.M. Jones 已提交
5055 5056 5057

    /* Read the memory file into buffer. */
    if (saferead (fd, buffer, size) == (ssize_t) -1) {
5058 5059 5060
        virReportSystemError(errno,
                             _("failed to read temporary file "
                               "created with template %s"), tmp);
5061
        goto endjob;
R
Richard W.M. Jones 已提交
5062 5063 5064
    }

    ret = 0;
5065

5066
endjob:
5067 5068
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
5069

5070
cleanup:
5071
    VIR_FORCE_CLOSE(fd);
R
Richard W.M. Jones 已提交
5072
    unlink (tmp);
W
Wen Congyang 已提交
5073
    VIR_FREE(tmp);
5074 5075
    if (vm)
        virDomainObjUnlock(vm);
R
Richard W.M. Jones 已提交
5076 5077 5078
    return ret;
}

5079

5080 5081 5082 5083 5084 5085 5086 5087 5088 5089
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;
5090
    virDomainDiskDefPtr disk = NULL;
5091 5092
    struct stat sb;
    int i;
5093
    int format;
5094 5095 5096 5097 5098 5099 5100 5101 5102 5103 5104 5105 5106 5107 5108 5109 5110 5111 5112 5113 5114 5115 5116 5117

    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)) {
5118
            disk = vm->def->disks[i];
5119 5120 5121 5122
            break;
        }
    }

5123
    if (!disk) {
5124 5125 5126 5127 5128 5129 5130 5131 5132 5133 5134 5135 5136 5137
        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 */
5138 5139 5140 5141 5142 5143 5144 5145
    if (disk->driverType) {
        if ((format = virStorageFileFormatTypeFromString(disk->driverType)) < 0) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("unknown disk format %s for %s"),
                            disk->driverType, disk->src);
            goto cleanup;
        }
    } else {
5146 5147 5148 5149 5150 5151 5152
        if (driver->allowDiskFormatProbing) {
            if ((format = virStorageFileProbeFormat(disk->src)) < 0)
                goto cleanup;
        } else {
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("no disk format for %s and probing is disabled"),
                            disk->src);
5153
            goto cleanup;
5154
        }
5155 5156 5157 5158 5159
    }

    if (virStorageFileGetMetadataFromFD(path, fd,
                                        format,
                                        &meta) < 0)
5160 5161 5162 5163 5164 5165 5166 5167 5168 5169
        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)) {
5170
#ifndef WIN32
5171 5172 5173 5174 5175 5176 5177 5178 5179 5180 5181 5182 5183 5184 5185 5186 5187 5188 5189 5190 5191 5192 5193 5194 5195 5196 5197 5198
        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;

5199
    /* Set default value .. */
5200 5201
    info->allocation = info->physical;

5202 5203 5204
    /* ..but if guest is running & not using raw
       disk format and on a block device, then query
       highest allocated extent from QEMU */
5205
    if (disk->type == VIR_DOMAIN_DISK_TYPE_BLOCK &&
5206
        format != VIR_STORAGE_FILE_RAW &&
5207 5208 5209 5210
        S_ISBLK(sb.st_mode)) {
        qemuDomainObjPrivatePtr priv = vm->privateData;
        if (qemuDomainObjBeginJob(vm) < 0)
            goto cleanup;
5211 5212 5213 5214 5215 5216 5217 5218 5219
        if (!virDomainObjIsActive(vm))
            ret = 0;
        else {
            qemuDomainObjEnterMonitor(vm);
            ret = qemuMonitorGetBlockExtent(priv->mon,
                                            disk->info.alias,
                                            &info->allocation);
            qemuDomainObjExitMonitor(vm);
        }
5220 5221 5222 5223 5224 5225

        if (qemuDomainObjEndJob(vm) == 0)
            vm = NULL;
    } else {
        ret = 0;
    }
5226 5227

cleanup:
5228
    VIR_FORCE_CLOSE(fd);
5229 5230 5231 5232 5233 5234
    if (vm)
        virDomainObjUnlock(vm);
    return ret;
}


5235
static int
5236 5237 5238 5239
qemuDomainEventRegister(virConnectPtr conn,
                        virConnectDomainEventCallback callback,
                        void *opaque,
                        virFreeCallback freecb)
5240
{
5241 5242 5243
    struct qemud_driver *driver = conn->privateData;
    int ret;

5244
    qemuDriverLock(driver);
5245 5246
    ret = virDomainEventCallbackListAdd(conn, driver->domainEventCallbacks,
                                        callback, opaque, freecb);
5247
    qemuDriverUnlock(driver);
5248

5249
    return ret;
5250 5251
}

5252

5253
static int
5254 5255
qemuDomainEventDeregister(virConnectPtr conn,
                          virConnectDomainEventCallback callback)
5256
{
5257 5258 5259
    struct qemud_driver *driver = conn->privateData;
    int ret;

5260
    qemuDriverLock(driver);
5261 5262 5263 5264 5265 5266
    if (driver->domainEventDispatching)
        ret = virDomainEventCallbackListMarkDelete(conn, driver->domainEventCallbacks,
                                                   callback);
    else
        ret = virDomainEventCallbackListRemove(conn, driver->domainEventCallbacks,
                                               callback);
5267
    qemuDriverUnlock(driver);
5268

5269
    return ret;
5270 5271
}

5272 5273 5274 5275 5276 5277 5278 5279 5280 5281 5282 5283 5284 5285 5286 5287 5288 5289 5290 5291 5292 5293 5294 5295 5296 5297 5298 5299 5300 5301 5302 5303 5304 5305 5306 5307 5308 5309 5310 5311 5312 5313 5314

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


D
Daniel Veillard 已提交
5315 5316
/* Migration support. */

C
Chris Lalancette 已提交
5317 5318 5319 5320 5321 5322 5323 5324 5325 5326 5327 5328 5329 5330 5331
/* 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;
    int ret = -1;
5332

C
Chris Lalancette 已提交
5333
    if (!dom_xml) {
5334 5335
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("no domain XML passed"));
C
Chris Lalancette 已提交
5336 5337 5338
        goto cleanup;
    }
    if (!(flags & VIR_MIGRATE_TUNNELLED)) {
5339
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
C
Chris Lalancette 已提交
5340 5341 5342 5343
                         "%s", _("PrepareTunnel called but no TUNNELLED flag set"));
        goto cleanup;
    }
    if (st == NULL) {
5344 5345
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("tunnelled migration requested but NULL stream passed"));
C
Chris Lalancette 已提交
5346 5347 5348
        goto cleanup;
    }

5349 5350 5351 5352
    qemuDriverLock(driver);
    ret = qemuMigrationPrepareTunnel(driver, dconn, st,
                                     dname, dom_xml);
    qemuDriverUnlock(driver);
5353

C
Chris Lalancette 已提交
5354 5355 5356 5357
cleanup:
    return ret;
}

D
Daniel Veillard 已提交
5358 5359 5360 5361
/* Prepare is the first step, and it runs on the destination host.
 *
 * This starts an empty VM listening on a TCP port.
 */
5362
static int ATTRIBUTE_NONNULL (5)
D
Daniel Veillard 已提交
5363 5364 5365 5366 5367
qemudDomainMigratePrepare2 (virConnectPtr dconn,
                            char **cookie ATTRIBUTE_UNUSED,
                            int *cookielen ATTRIBUTE_UNUSED,
                            const char *uri_in,
                            char **uri_out,
C
Chris Lalancette 已提交
5368
                            unsigned long flags,
D
Daniel Veillard 已提交
5369 5370 5371 5372
                            const char *dname,
                            unsigned long resource ATTRIBUTE_UNUSED,
                            const char *dom_xml)
{
5373
    struct qemud_driver *driver = dconn->privateData;
5374
    int ret = -1;
5375

C
Chris Lalancette 已提交
5376 5377 5378 5379 5380 5381 5382 5383 5384
    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);

5385
    *uri_out = NULL;
D
Daniel Veillard 已提交
5386

5387
    qemuDriverLock(driver);
C
Chris Lalancette 已提交
5388 5389 5390 5391
    if (flags & VIR_MIGRATE_TUNNELLED) {
        /* this is a logical error; we never should have gotten here with
         * VIR_MIGRATE_TUNNELLED set
         */
5392 5393
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("Tunnelled migration requested but invalid RPC method called"));
C
Chris Lalancette 已提交
5394 5395 5396
        goto cleanup;
    }

D
Daniel Veillard 已提交
5397
    if (!dom_xml) {
5398 5399
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("no domain XML passed"));
5400
        goto cleanup;
D
Daniel Veillard 已提交
5401 5402
    }

5403 5404 5405
    ret = qemuMigrationPrepareDirect(driver, dconn,
                                     uri_in, uri_out,
                                     dname, dom_xml);
D
Daniel Veillard 已提交
5406

5407 5408 5409 5410
cleanup:
    qemuDriverUnlock(driver);
    return ret;
}
C
Chris Lalancette 已提交
5411

D
Daniel Veillard 已提交
5412

5413 5414 5415 5416 5417 5418 5419 5420 5421 5422 5423 5424 5425
/* Perform is the second step, and it runs on the source host. */
static int
qemudDomainMigratePerform (virDomainPtr dom,
                           const char *cookie ATTRIBUTE_UNUSED,
                           int cookielen ATTRIBUTE_UNUSED,
                           const char *uri,
                           unsigned long flags,
                           const char *dname,
                           unsigned long resource)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
5426

5427 5428 5429 5430 5431 5432 5433 5434
    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);
C
Chris Lalancette 已提交
5435

5436
    qemuDriverLock(driver);
5437
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
D
Daniel Veillard 已提交
5438
    if (!vm) {
5439 5440
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
5441 5442
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
5443
        goto cleanup;
D
Daniel Veillard 已提交
5444 5445
    }

5446 5447 5448
    ret = qemuMigrationPerform(driver, dom->conn, vm,
                               uri, flags,
                               dname, resource);
5449

5450
cleanup:
5451
    qemuDriverUnlock(driver);
5452
    return ret;
D
Daniel Veillard 已提交
5453 5454
}

5455

D
Daniel Veillard 已提交
5456 5457 5458 5459 5460 5461 5462
/* 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 已提交
5463
                           unsigned long flags,
D
Daniel Veillard 已提交
5464 5465
                           int retcode)
{
5466 5467 5468
    struct qemud_driver *driver = dconn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
5469
    virErrorPtr orig_err;
D
Daniel Veillard 已提交
5470

C
Chris Lalancette 已提交
5471 5472 5473 5474 5475 5476 5477 5478 5479
    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);

5480 5481 5482
    /* Migration failed. Save the current error so nothing squashes it */
    orig_err = virSaveLastError();

5483
    qemuDriverLock(driver);
5484
    vm = virDomainFindByName(&driver->domains, dname);
D
Daniel Veillard 已提交
5485
    if (!vm) {
5486 5487
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching name '%s'"), dname);
5488
        goto cleanup;
D
Daniel Veillard 已提交
5489 5490
    }

5491
    dom = qemuMigrationFinish(driver, dconn, vm, flags, retcode);
5492

5493
cleanup:
5494 5495 5496 5497
    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }
5498
    qemuDriverUnlock(driver);
5499
    return dom;
D
Daniel Veillard 已提交
5500 5501
}

5502

5503 5504 5505 5506 5507 5508 5509 5510 5511 5512 5513 5514 5515 5516 5517 5518
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;

5519
    def = virNodeDeviceDefParseString(xml, EXISTING_DEVICE);
5520 5521 5522 5523 5524 5525 5526 5527 5528 5529 5530 5531 5532 5533 5534 5535 5536
    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) {
5537 5538
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("device %s is not a PCI device"), dev->name);
5539 5540 5541 5542 5543 5544 5545 5546 5547 5548 5549 5550 5551
        goto out;
    }

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

static int
qemudNodeDeviceDettach (virNodeDevicePtr dev)
{
5552
    struct qemud_driver *driver = dev->conn->privateData;
5553 5554 5555 5556 5557 5558 5559
    pciDevice *pci;
    unsigned domain, bus, slot, function;
    int ret = -1;

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

5560
    pci = pciGetDevice(domain, bus, slot, function);
5561 5562 5563
    if (!pci)
        return -1;

5564 5565
    qemuDriverLock(driver);
    if (pciDettachDevice(pci, driver->activePciHostdevs) < 0)
5566 5567 5568 5569
        goto out;

    ret = 0;
out:
5570
    qemuDriverUnlock(driver);
5571
    pciFreeDevice(pci);
5572 5573 5574 5575 5576 5577
    return ret;
}

static int
qemudNodeDeviceReAttach (virNodeDevicePtr dev)
{
5578
    struct qemud_driver *driver = dev->conn->privateData;
5579 5580 5581 5582 5583 5584 5585
    pciDevice *pci;
    unsigned domain, bus, slot, function;
    int ret = -1;

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

5586
    pci = pciGetDevice(domain, bus, slot, function);
5587 5588 5589
    if (!pci)
        return -1;

5590 5591
    qemuDriverLock(driver);
    if (pciReAttachDevice(pci, driver->activePciHostdevs) < 0)
5592 5593 5594 5595
        goto out;

    ret = 0;
out:
5596
    qemuDriverUnlock(driver);
5597
    pciFreeDevice(pci);
5598 5599 5600 5601 5602 5603
    return ret;
}

static int
qemudNodeDeviceReset (virNodeDevicePtr dev)
{
5604
    struct qemud_driver *driver = dev->conn->privateData;
5605 5606 5607 5608 5609 5610 5611
    pciDevice *pci;
    unsigned domain, bus, slot, function;
    int ret = -1;

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

5612
    pci = pciGetDevice(domain, bus, slot, function);
5613 5614 5615
    if (!pci)
        return -1;

5616 5617
    qemuDriverLock(driver);

5618
    if (pciResetDevice(pci, driver->activePciHostdevs, NULL) < 0)
5619 5620 5621 5622
        goto out;

    ret = 0;
out:
5623
    qemuDriverUnlock(driver);
5624
    pciFreeDevice(pci);
5625 5626 5627
    return ret;
}

5628 5629 5630 5631 5632 5633 5634 5635 5636 5637 5638
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) {
5639 5640
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        "%s", _("cannot get host CPU capabilities"));
5641 5642
    }
    else
5643
        ret = cpuCompareXML(driver->caps->host.cpu, xmlDesc);
5644 5645 5646 5647 5648 5649

    qemuDriverUnlock(driver);

    return ret;
}

5650

5651 5652 5653 5654 5655 5656 5657 5658 5659 5660 5661 5662 5663
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;
}

5664 5665 5666 5667 5668 5669 5670 5671 5672 5673 5674 5675 5676 5677 5678 5679 5680 5681 5682 5683 5684 5685 5686

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) {
5687 5688
            struct timeval now;

5689
            memcpy(info, &priv->jobInfo, sizeof(*info));
5690 5691 5692 5693 5694 5695 5696 5697 5698 5699 5700

            /* Refresh elapsed time again just to ensure it
             * is fully updated. This is primarily for benefit
             * of incoming migration which we don't currently
             * monitor actively in the background thread
             */
            if (gettimeofday(&now, NULL) < 0) {
                virReportSystemError(errno, "%s",
                                     _("cannot get time of day"));
                goto cleanup;
            }
5701
            info->timeElapsed = timeval_to_ms(now) - priv->jobStart;
5702 5703 5704 5705 5706 5707 5708 5709 5710 5711 5712 5713 5714 5715 5716 5717 5718 5719 5720
        } 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;
}


5721 5722 5723 5724 5725 5726 5727 5728 5729 5730 5731 5732 5733 5734 5735 5736 5737 5738 5739 5740 5741 5742
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);
5743
            priv->jobSignals |= QEMU_JOB_SIGNAL_CANCEL;
5744 5745 5746 5747 5748 5749 5750 5751 5752 5753 5754 5755 5756 5757 5758 5759 5760 5761 5762 5763
        } 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;
}


5764 5765 5766 5767 5768 5769 5770 5771 5772 5773
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;

5774
    virCheckFlags(0, -1);
5775 5776 5777 5778 5779 5780 5781 5782 5783 5784 5785 5786 5787 5788 5789 5790 5791 5792 5793 5794

    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;

5795
    if (priv->jobActive != QEMU_JOB_MIGRATION_OUT) {
5796 5797 5798 5799 5800 5801 5802 5803 5804 5805 5806 5807 5808 5809 5810 5811 5812
        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;
}

5813 5814 5815 5816 5817 5818 5819 5820 5821 5822 5823 5824 5825 5826 5827 5828 5829 5830 5831 5832 5833 5834 5835 5836 5837 5838 5839 5840 5841 5842 5843 5844 5845 5846 5847 5848 5849 5850 5851 5852 5853 5854 5855 5856 5857 5858 5859 5860 5861
static int
qemuDomainMigrateSetMaxSpeed(virDomainPtr dom,
                             unsigned long bandwidth,
                             unsigned int flags)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    qemuDomainObjPrivatePtr priv;
    int ret = -1;

    virCheckFlags(0, -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 (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
    }

    priv = vm->privateData;

    if (priv->jobActive != QEMU_JOB_MIGRATION_OUT) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not being migrated"));
        goto cleanup;
    }

    VIR_DEBUG("Requesting migration speed change to %luMbs", bandwidth);
    priv->jobSignals |= QEMU_JOB_SIGNAL_MIGRATE_SPEED;
    priv->jobSignalsData.migrateBandwidth = bandwidth;
    ret = 0;

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

C
Chris Lalancette 已提交
5862 5863 5864 5865 5866 5867 5868 5869 5870 5871 5872 5873 5874 5875
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;
}

5876 5877 5878
static int qemuDomainSnapshotWriteMetadata(virDomainObjPtr vm,
                                           virDomainSnapshotObjPtr snapshot,
                                           char *snapshotDir)
C
Chris Lalancette 已提交
5879 5880 5881 5882 5883 5884 5885 5886 5887 5888
{
    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);
5889
    newxml = virDomainSnapshotDefFormat(uuidstr, snapshot->def, 1);
C
Chris Lalancette 已提交
5890 5891 5892 5893 5894 5895 5896 5897 5898 5899 5900 5901 5902 5903 5904 5905
    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;
    }

5906
    if (virAsprintf(&snapFile, "%s/%s.xml", snapDir, snapshot->def->name) < 0) {
C
Chris Lalancette 已提交
5907 5908 5909 5910 5911 5912 5913 5914 5915 5916 5917 5918 5919 5920 5921 5922 5923 5924 5925 5926 5927
        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);
5928
    VIR_FORCE_CLOSE(fd);
C
Chris Lalancette 已提交
5929 5930 5931
    return ret;
}

5932 5933
static int qemuDomainSnapshotSetCurrentActive(virDomainObjPtr vm,
                                              char *snapshotDir)
C
Chris Lalancette 已提交
5934 5935 5936 5937
{
    if (vm->current_snapshot) {
        vm->current_snapshot->def->active = 1;

5938 5939
        return qemuDomainSnapshotWriteMetadata(vm, vm->current_snapshot,
                                               snapshotDir);
C
Chris Lalancette 已提交
5940 5941 5942 5943 5944
    }

    return 0;
}

5945 5946
static int qemuDomainSnapshotSetCurrentInactive(virDomainObjPtr vm,
                                                char *snapshotDir)
C
Chris Lalancette 已提交
5947 5948 5949 5950
{
    if (vm->current_snapshot) {
        vm->current_snapshot->def->active = 0;

5951 5952
        return qemuDomainSnapshotWriteMetadata(vm, vm->current_snapshot,
                                               snapshotDir);
C
Chris Lalancette 已提交
5953 5954 5955 5956 5957 5958 5959 5960 5961 5962 5963 5964 5965 5966 5967 5968 5969 5970 5971
    }

    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,
5972 5973
                            _("Disk '%s' does not support snapshotting"),
                            vm->def->disks[i]->src);
C
Chris Lalancette 已提交
5974 5975 5976 5977 5978 5979 5980
            return 0;
        }
    }

    return 1;
}

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
/* The domain is expected to be locked and inactive. */
static int
qemuDomainSnapshotCreateInactive(virDomainObjPtr vm,
                                 virDomainSnapshotObjPtr snap)
{
    const char *qemuimgarg[] = { NULL, "snapshot", "-c", NULL, NULL, NULL };
    int ret = -1;
    int i;

    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)
                goto cleanup;
        }
    }

    ret = 0;

cleanup:
    VIR_FREE(qemuimgarg[0]);
    return ret;
}

/* The domain is expected to be locked and active. */
static int
6029 6030
qemuDomainSnapshotCreateActive(virConnectPtr conn,
                               struct qemud_driver *driver,
6031 6032 6033 6034 6035
                               virDomainObjPtr *vmptr,
                               virDomainSnapshotObjPtr snap)
{
    virDomainObjPtr vm = *vmptr;
    qemuDomainObjPrivatePtr priv = vm->privateData;
6036 6037
    bool resume = false;
    int ret = -1;
6038 6039 6040 6041

    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        return -1;

6042 6043 6044 6045 6046 6047 6048 6049 6050 6051 6052 6053 6054 6055 6056 6057
    if (vm->state == VIR_DOMAIN_RUNNING) {
        /* savevm monitor command pauses the domain emitting an event which
         * confuses libvirt since it's not notified when qemu resumes the
         * domain. Thus we stop and start CPUs ourselves.
         */
        if (qemuProcessStopCPUs(driver, vm) < 0)
            goto cleanup;

        resume = true;
        if (!virDomainObjIsActive(vm)) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("guest unexpectedly quit"));
            goto cleanup;
        }
    }

6058 6059 6060 6061
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
    ret = qemuMonitorCreateSnapshot(priv->mon, snap->def->name);
    qemuDomainObjExitMonitorWithDriver(driver, vm);

6062 6063 6064 6065 6066 6067 6068 6069
cleanup:
    if (resume && virDomainObjIsActive(vm) &&
        qemuProcessStartCPUs(driver, vm, conn) < 0 &&
        virGetLastError() == NULL) {
        qemuReportError(VIR_ERR_OPERATION_FAILED, "%s",
                        _("resuming after snapshot failed"));
    }

6070 6071 6072 6073 6074 6075
    if (qemuDomainObjEndJob(vm) == 0)
        *vmptr = NULL;

    return ret;
}

C
Chris Lalancette 已提交
6076 6077
static virDomainSnapshotPtr qemuDomainSnapshotCreateXML(virDomainPtr domain,
                                                        const char *xmlDesc,
6078
                                                        unsigned int flags)
C
Chris Lalancette 已提交
6079 6080 6081 6082 6083 6084 6085 6086
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    virDomainSnapshotObjPtr snap = NULL;
    virDomainSnapshotPtr snapshot = NULL;
    char uuidstr[VIR_UUID_STRING_BUFLEN];
    virDomainSnapshotDefPtr def;

6087 6088
    virCheckFlags(0, NULL);

C
Chris Lalancette 已提交
6089 6090 6091 6092 6093 6094 6095 6096 6097 6098 6099 6100 6101 6102 6103 6104 6105 6106 6107 6108 6109 6110 6111 6112 6113
    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;

6114 6115
    snap->def->state = vm->state;

C
Chris Lalancette 已提交
6116 6117
    /* actually do the snapshot */
    if (!virDomainObjIsActive(vm)) {
6118
        if (qemuDomainSnapshotCreateInactive(vm, snap) < 0)
C
Chris Lalancette 已提交
6119 6120 6121
            goto cleanup;
    }
    else {
6122 6123
        if (qemuDomainSnapshotCreateActive(domain->conn, driver,
                                           &vm, snap) < 0)
6124
            goto cleanup;
C
Chris Lalancette 已提交
6125 6126 6127 6128 6129 6130 6131 6132 6133 6134 6135 6136 6137 6138 6139 6140 6141 6142
    }

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

6143 6144 6145
    if (qemuDomainSnapshotWriteMetadata(vm, vm->current_snapshot,
                                        driver->snapshotDir) < 0)
        /* qemuDomainSnapshotWriteMetadata set the error */
C
Chris Lalancette 已提交
6146 6147 6148 6149 6150 6151 6152 6153 6154 6155 6156 6157 6158
        goto cleanup;

    snapshot = virGetDomainSnapshot(domain, snap->def->name);

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return snapshot;
}

static int qemuDomainSnapshotListNames(virDomainPtr domain, char **names,
                                       int nameslen,
6159
                                       unsigned int flags)
C
Chris Lalancette 已提交
6160 6161 6162 6163 6164
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    int n = -1;

6165 6166
    virCheckFlags(0, -1);

C
Chris Lalancette 已提交
6167 6168 6169 6170 6171 6172 6173 6174 6175 6176 6177 6178 6179 6180 6181 6182 6183 6184 6185 6186
    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,
6187
                                 unsigned int flags)
C
Chris Lalancette 已提交
6188 6189 6190 6191 6192
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    int n = -1;

6193 6194
    virCheckFlags(0, -1);

C
Chris Lalancette 已提交
6195 6196 6197 6198 6199 6200 6201 6202 6203 6204 6205 6206 6207 6208 6209 6210 6211 6212 6213 6214 6215
    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,
6216
                                                           unsigned int flags)
C
Chris Lalancette 已提交
6217 6218 6219 6220 6221 6222
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm;
    virDomainSnapshotObjPtr snap = NULL;
    virDomainSnapshotPtr snapshot = NULL;

6223 6224
    virCheckFlags(0, NULL);

C
Chris Lalancette 已提交
6225 6226 6227 6228 6229 6230 6231 6232 6233 6234 6235 6236 6237 6238 6239 6240 6241 6242 6243 6244 6245 6246 6247 6248 6249 6250 6251
    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,
6252
                                        unsigned int flags)
C
Chris Lalancette 已提交
6253 6254 6255 6256 6257
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;

6258 6259
    virCheckFlags(0, -1);

C
Chris Lalancette 已提交
6260 6261 6262 6263 6264 6265 6266 6267 6268 6269 6270 6271 6272 6273 6274 6275 6276 6277 6278 6279
    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,
6280
                                                      unsigned int flags)
C
Chris Lalancette 已提交
6281 6282 6283 6284 6285
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm;
    virDomainSnapshotPtr snapshot = NULL;

6286 6287
    virCheckFlags(0, NULL);

C
Chris Lalancette 已提交
6288 6289 6290 6291 6292 6293 6294 6295 6296 6297 6298 6299 6300 6301 6302 6303 6304 6305 6306 6307 6308 6309 6310 6311 6312 6313
    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,
6314
                                       unsigned int flags)
C
Chris Lalancette 已提交
6315 6316 6317 6318 6319 6320 6321
{
    struct qemud_driver *driver = snapshot->domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    char *xml = NULL;
    virDomainSnapshotObjPtr snap = NULL;
    char uuidstr[VIR_UUID_STRING_BUFLEN];

6322 6323
    virCheckFlags(0, NULL);

C
Chris Lalancette 已提交
6324 6325 6326 6327 6328 6329 6330 6331 6332 6333 6334 6335 6336 6337 6338 6339 6340 6341 6342 6343 6344 6345 6346 6347 6348 6349 6350
    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,
6351
                                      unsigned int flags)
C
Chris Lalancette 已提交
6352 6353 6354 6355 6356 6357 6358 6359 6360 6361
{
    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;

6362 6363
    virCheckFlags(0, -1);

C
Chris Lalancette 已提交
6364 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
    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)
6395
                goto endjob;
C
Chris Lalancette 已提交
6396 6397
        }
        else {
6398
            if (qemuDomainSnapshotSetCurrentActive(vm, driver->snapshotDir) < 0)
6399
                goto endjob;
C
Chris Lalancette 已提交
6400

6401 6402
            rc = qemuProcessStart(snapshot->domain->conn, driver, vm, NULL,
                                  false, -1, NULL, VIR_VM_OP_CREATE);
6403
            qemuAuditDomainStart(vm, "from-snapshot", rc >= 0);
6404
            if (qemuDomainSnapshotSetCurrentInactive(vm, driver->snapshotDir) < 0)
6405
                goto endjob;
C
Chris Lalancette 已提交
6406
            if (rc < 0)
6407
                goto endjob;
C
Chris Lalancette 已提交
6408 6409 6410 6411 6412 6413
        }

        if (snap->def->state == VIR_DOMAIN_PAUSED) {
            /* qemu unconditionally starts the domain running again after
             * loadvm, so let's pause it to keep consistency
             */
6414
            rc = qemuProcessStopCPUs(driver, vm);
H
Hu Tao 已提交
6415
            if (rc < 0)
6416
                goto endjob;
C
Chris Lalancette 已提交
6417 6418 6419 6420 6421 6422 6423 6424 6425 6426 6427 6428 6429 6430 6431 6432 6433 6434
        }

        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)) {
6435
            qemuProcessStop(driver, vm, 0);
6436
            qemuAuditDomainStop(vm, "from-snapshot");
C
Chris Lalancette 已提交
6437 6438 6439
            event = virDomainEventNewFromObj(vm,
                                             VIR_DOMAIN_EVENT_STOPPED,
                                             VIR_DOMAIN_EVENT_STOPPED_FROM_SNAPSHOT);
6440 6441 6442 6443
            if (!vm->persistent) {
                if (qemuDomainObjEndJob(vm) > 0)
                    virDomainRemoveInactive(&driver->domains, vm);
                vm = NULL;
6444
                goto cleanup;
6445
            }
C
Chris Lalancette 已提交
6446 6447
        }

6448
        if (qemuDomainSnapshotSetCurrentActive(vm, driver->snapshotDir) < 0)
6449
            goto endjob;
C
Chris Lalancette 已提交
6450 6451 6452 6453 6454 6455
    }

    vm->state = snap->def->state;

    ret = 0;

6456
endjob:
C
Chris Lalancette 已提交
6457 6458 6459
    if (vm && qemuDomainObjEndJob(vm) == 0)
        vm = NULL;

6460
cleanup:
C
Chris Lalancette 已提交
6461 6462 6463 6464 6465 6466 6467 6468 6469 6470 6471 6472 6473 6474 6475 6476 6477 6478 6479 6480 6481 6482 6483 6484 6485 6486 6487 6488 6489 6490 6491 6492 6493 6494 6495 6496 6497 6498 6499 6500 6501 6502 6503 6504 6505 6506 6507 6508 6509 6510 6511 6512 6513 6514 6515 6516 6517 6518 6519 6520 6521 6522 6523 6524 6525 6526 6527 6528 6529 6530 6531 6532 6533 6534 6535 6536 6537 6538 6539 6540 6541 6542 6543 6544 6545 6546 6547 6548 6549 6550 6551 6552 6553 6554 6555 6556 6557 6558 6559 6560 6561 6562
    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,
6563
                                              const void *name ATTRIBUTE_UNUSED,
C
Chris Lalancette 已提交
6564 6565 6566 6567 6568 6569 6570 6571 6572 6573 6574 6575 6576 6577 6578 6579 6580 6581 6582 6583 6584
                                              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);
    }
}

6585 6586 6587 6588 6589 6590 6591 6592 6593
struct snap_reparent {
    struct qemud_driver *driver;
    virDomainSnapshotObjPtr snap;
    virDomainObjPtr vm;
    int err;
};

static void
qemuDomainSnapshotReparentChildren(void *payload,
6594
                                   const void *name ATTRIBUTE_UNUSED,
6595 6596 6597 6598 6599 6600 6601 6602 6603 6604 6605 6606 6607 6608 6609 6610 6611 6612 6613 6614 6615 6616 6617 6618 6619 6620 6621
                                   void *data)
{
    virDomainSnapshotObjPtr snap = payload;
    struct snap_reparent *rep = data;

    if (rep->err < 0) {
        return;
    }

    if (snap->def->parent && STREQ(snap->def->parent, rep->snap->def->name)) {
        VIR_FREE(snap->def->parent);

        if (rep->snap->def->parent != NULL) {
            snap->def->parent = strdup(rep->snap->def->parent);

            if (snap->def->parent == NULL) {
                virReportOOMError();
                rep->err = -1;
                return;
            }
        }

        rep->err = qemuDomainSnapshotWriteMetadata(rep->vm, snap,
                                                   rep->driver->snapshotDir);
    }
}

C
Chris Lalancette 已提交
6622 6623 6624 6625 6626 6627 6628 6629 6630
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;
6631
    struct snap_reparent rep;
C
Chris Lalancette 已提交
6632

6633 6634
    virCheckFlags(VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN, -1);

C
Chris Lalancette 已提交
6635 6636 6637 6638 6639 6640 6641 6642 6643 6644 6645 6646 6647 6648 6649 6650 6651
    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;
    }

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

C
Chris Lalancette 已提交
6655 6656 6657 6658 6659 6660 6661 6662
    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)
6663
            goto endjob;
6664 6665 6666 6667 6668 6669 6670 6671 6672
    } else {
        rep.driver = driver;
        rep.snap = snap;
        rep.vm = vm;
        rep.err = 0;
        virHashForEach(vm->snapshots.objs, qemuDomainSnapshotReparentChildren,
                       &rep);
        if (rep.err < 0)
            goto endjob;
C
Chris Lalancette 已提交
6673 6674 6675 6676
    }

    ret = qemuDomainSnapshotDiscard(driver, vm, snap);

6677 6678 6679 6680
endjob:
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;

C
Chris Lalancette 已提交
6681 6682 6683 6684 6685 6686
cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}
6687

6688 6689 6690 6691 6692 6693 6694
static int qemuDomainMonitorCommand(virDomainPtr domain, const char *cmd,
                                    char **result, unsigned int flags)
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    int ret = -1;
    qemuDomainObjPrivatePtr priv;
6695
    bool hmp;
6696

6697
    virCheckFlags(VIR_DOMAIN_QEMU_MONITOR_COMMAND_HMP, -1);
6698 6699 6700 6701 6702 6703 6704 6705 6706 6707 6708 6709 6710 6711 6712 6713 6714 6715 6716 6717 6718 6719 6720 6721 6722

    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 (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
   }

    priv = vm->privateData;

    if (!priv->monitor_warned) {
        VIR_INFO("Qemu monitor command '%s' executed; libvirt results may be unpredictable!",
                 cmd);
        priv->monitor_warned = 1;
    }

6723 6724
    hmp = !!(flags & VIR_DOMAIN_QEMU_MONITOR_COMMAND_HMP);

6725 6726 6727
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
6728
    ret = qemuMonitorArbitraryCommand(priv->mon, cmd, result, hmp);
6729 6730 6731 6732 6733 6734 6735 6736 6737 6738 6739 6740 6741
    qemuDomainObjExitMonitorWithDriver(driver, vm);
    if (qemuDomainObjEndJob(vm) == 0) {
        vm = NULL;
        goto cleanup;
    }

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

6742 6743 6744 6745 6746 6747 6748 6749 6750 6751 6752 6753 6754 6755 6756 6757 6758 6759 6760 6761 6762 6763 6764 6765 6766 6767 6768 6769 6770 6771 6772 6773 6774 6775 6776 6777 6778 6779 6780 6781 6782 6783 6784 6785 6786 6787 6788 6789 6790 6791 6792 6793 6794 6795 6796 6797 6798

static int
qemuDomainOpenConsole(virDomainPtr dom,
                      const char *devname,
                      virStreamPtr st,
                      unsigned int flags)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm = NULL;
    char uuidstr[VIR_UUID_STRING_BUFLEN];
    int ret = -1;
    int i;
    virDomainChrDefPtr chr = NULL;

    virCheckFlags(0, -1);

    qemuDriverLock(driver);
    virUUIDFormat(dom->uuid, uuidstr);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
    if (!vm) {
        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;
    }

    if (devname) {
        if (vm->def->console &&
            STREQ(devname, vm->def->console->info.alias))
            chr = vm->def->console;
        for (i = 0 ; !chr && i < vm->def->nserials ; i++) {
            if (STREQ(devname, vm->def->serials[i]->info.alias))
                chr = vm->def->serials[i];
        }
        for (i = 0 ; !chr && i < vm->def->nparallels ; i++) {
            if (STREQ(devname, vm->def->parallels[i]->info.alias))
                chr = vm->def->parallels[i];
        }
    } else {
        if (vm->def->console)
            chr = vm->def->console;
        else if (vm->def->nserials)
            chr = vm->def->serials[0];
    }

    if (!chr) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("cannot find character device %s"),
                        NULLSTR(devname));
        goto cleanup;
    }

6799
    if (chr->source.type != VIR_DOMAIN_CHR_TYPE_PTY) {
6800 6801 6802 6803 6804 6805
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("character device %s is not using a PTY"),
                        NULLSTR(devname));
        goto cleanup;
    }

6806
    if (virFDStreamOpenFile(st, chr->source.data.file.path, 0, 0, O_RDWR) < 0)
6807 6808 6809 6810 6811 6812 6813 6814 6815 6816 6817
        goto cleanup;

    ret = 0;
cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}


6818 6819 6820 6821 6822
static virDriver qemuDriver = {
    VIR_DRV_QEMU,
    "QEMU",
    qemudOpen, /* open */
    qemudClose, /* close */
D
Daniel Veillard 已提交
6823
    qemudSupportsFeature, /* supports_feature */
6824 6825
    qemudGetType, /* type */
    qemudGetVersion, /* version */
6826
    NULL, /* libvirtVersion (impl. in libvirt.c) */
6827
    virGetHostname, /* getHostname */
E
Eric Blake 已提交
6828
    qemuGetSysinfo, /* getSysinfo */
6829
    qemudGetMaxVCPUs, /* getMaxVcpus */
6830
    nodeGetInfo, /* nodeGetInfo */
6831 6832 6833
    qemudGetCapabilities, /* getCapabilities */
    qemudListDomains, /* listDomains */
    qemudNumDomains, /* numOfDomains */
6834
    qemudDomainCreate, /* domainCreateXML */
6835 6836 6837 6838 6839
    qemudDomainLookupByID, /* domainLookupByID */
    qemudDomainLookupByUUID, /* domainLookupByUUID */
    qemudDomainLookupByName, /* domainLookupByName */
    qemudDomainSuspend, /* domainSuspend */
    qemudDomainResume, /* domainResume */
6840
    qemudDomainShutdown, /* domainShutdown */
6841 6842 6843
    NULL, /* domainReboot */
    qemudDomainDestroy, /* domainDestroy */
    qemudDomainGetOSType, /* domainGetOSType */
6844
    qemudDomainGetMaxMemory, /* domainGetMaxMemory */
6845
    NULL, /* domainSetMaxMemory */
6846
    qemudDomainSetMemory, /* domainSetMemory */
6847
    qemudDomainSetMemoryFlags, /* domainSetMemoryFlags */
6848 6849
    qemuDomainSetMemoryParameters, /* domainSetMemoryParameters */
    qemuDomainGetMemoryParameters, /* domainGetMemoryParameters */
6850 6851
    qemuDomainSetBlkioParameters, /* domainSetBlkioParameters */
    qemuDomainGetBlkioParameters, /* domainGetBlkioParameters */
6852 6853
    qemudDomainGetInfo, /* domainGetInfo */
    qemudDomainSave, /* domainSave */
6854
    qemuDomainRestore, /* domainRestore */
P
Paolo Bonzini 已提交
6855
    qemudDomainCoreDump, /* domainCoreDump */
6856
    qemudDomainSetVcpus, /* domainSetVcpus */
6857 6858
    qemudDomainSetVcpusFlags, /* domainSetVcpusFlags */
    qemudDomainGetVcpusFlags, /* domainGetVcpusFlags */
6859 6860
    qemudDomainPinVcpu, /* domainPinVcpu */
    qemudDomainGetVcpus, /* domainGetVcpus */
6861
    qemudDomainGetMaxVcpus, /* domainGetMaxVcpus */
6862 6863
    qemudDomainGetSecurityLabel, /* domainGetSecurityLabel */
    qemudNodeGetSecurityModel, /* nodeGetSecurityModel */
6864
    qemudDomainDumpXML, /* domainDumpXML */
6865
    qemuDomainXMLFromNative, /* domainXmlFromNative */
6866
    qemuDomainXMLToNative, /* domainXMLToNative */
6867 6868
    qemudListDefinedDomains, /* listDefinedDomains */
    qemudNumDefinedDomains, /* numOfDefinedDomains */
6869
    qemudDomainStart, /* domainCreate */
6870
    qemudDomainStartWithFlags, /* domainCreateWithFlags */
6871 6872
    qemudDomainDefine, /* domainDefineXML */
    qemudDomainUndefine, /* domainUndefine */
6873
    qemudDomainAttachDevice, /* domainAttachDevice */
6874
    qemudDomainAttachDeviceFlags, /* domainAttachDeviceFlags */
6875
    qemudDomainDetachDevice, /* domainDetachDevice */
6876
    qemudDomainDetachDeviceFlags, /* domainDetachDeviceFlags */
6877
    qemuDomainUpdateDeviceFlags, /* domainUpdateDeviceFlags */
6878 6879
    qemudDomainGetAutostart, /* domainGetAutostart */
    qemudDomainSetAutostart, /* domainSetAutostart */
6880 6881 6882
    qemuGetSchedulerType, /* domainGetSchedulerType */
    qemuGetSchedulerParameters, /* domainGetSchedulerParameters */
    qemuSetSchedulerParameters, /* domainSetSchedulerParameters */
D
Daniel Veillard 已提交
6883 6884
    NULL, /* domainMigratePrepare (v1) */
    qemudDomainMigratePerform, /* domainMigratePerform */
6885
    NULL, /* domainMigrateFinish */
6886
    qemudDomainBlockStats, /* domainBlockStats */
6887
    qemudDomainInterfaceStats, /* domainInterfaceStats */
6888
    qemudDomainMemoryStats, /* domainMemoryStats */
6889
    qemudDomainBlockPeek, /* domainBlockPeek */
R
Richard W.M. Jones 已提交
6890
    qemudDomainMemoryPeek, /* domainMemoryPeek */
6891
    qemuDomainGetBlockInfo, /* domainGetBlockInfo */
6892 6893
    nodeGetCellsFreeMemory, /* nodeGetCellsFreeMemory */
    nodeGetFreeMemory,  /* getFreeMemory */
6894 6895
    qemuDomainEventRegister, /* domainEventRegister */
    qemuDomainEventDeregister, /* domainEventDeregister */
D
Daniel Veillard 已提交
6896 6897
    qemudDomainMigratePrepare2, /* domainMigratePrepare2 */
    qemudDomainMigrateFinish2, /* domainMigrateFinish2 */
6898 6899 6900
    qemudNodeDeviceDettach, /* nodeDeviceDettach */
    qemudNodeDeviceReAttach, /* nodeDeviceReAttach */
    qemudNodeDeviceReset, /* nodeDeviceReset */
C
Chris Lalancette 已提交
6901
    qemudDomainMigratePrepareTunnel, /* domainMigratePrepareTunnel */
6902 6903 6904 6905
    qemuIsEncrypted, /* isEncrypted */
    qemuIsSecure, /* isSecure */
    qemuDomainIsActive, /* domainIsActive */
    qemuDomainIsPersistent, /* domainIsPersistent */
6906
    qemuDomainIsUpdated, /* domainIsUpdated */
6907
    qemuCPUCompare, /* cpuCompare */
6908
    qemuCPUBaseline, /* cpuBaseline */
6909
    qemuDomainGetJobInfo, /* domainGetJobInfo */
6910
    qemuDomainAbortJob, /* domainAbortJob */
6911
    qemuDomainMigrateSetMaxDowntime, /* domainMigrateSetMaxDowntime */
6912
    qemuDomainMigrateSetMaxSpeed, /* domainMigrateSetMaxSpeed */
6913 6914
    qemuDomainEventRegisterAny, /* domainEventRegisterAny */
    qemuDomainEventDeregisterAny, /* domainEventDeregisterAny */
6915 6916 6917
    qemuDomainManagedSave, /* domainManagedSave */
    qemuDomainHasManagedSaveImage, /* domainHasManagedSaveImage */
    qemuDomainManagedSaveRemove, /* domainManagedSaveRemove */
C
Chris Lalancette 已提交
6918 6919 6920 6921 6922 6923 6924 6925 6926
    qemuDomainSnapshotCreateXML, /* domainSnapshotCreateXML */
    qemuDomainSnapshotDumpXML, /* domainSnapshotDumpXML */
    qemuDomainSnapshotNum, /* domainSnapshotNum */
    qemuDomainSnapshotListNames, /* domainSnapshotListNames */
    qemuDomainSnapshotLookupByName, /* domainSnapshotLookupByName */
    qemuDomainHasCurrentSnapshot, /* domainHasCurrentSnapshot */
    qemuDomainSnapshotCurrent, /* domainSnapshotCurrent */
    qemuDomainRevertToSnapshot, /* domainRevertToSnapshot */
    qemuDomainSnapshotDelete, /* domainSnapshotDelete */
6927
    qemuDomainMonitorCommand, /* qemuDomainMonitorCommand */
6928
    qemuDomainOpenConsole, /* domainOpenConsole */
6929 6930 6931
};


6932
static virStateDriver qemuStateDriver = {
6933
    .name = "QEMU",
6934 6935 6936 6937
    .initialize = qemudStartup,
    .cleanup = qemudShutdown,
    .reload = qemudReload,
    .active = qemudActive,
6938
};
6939

6940
static void
6941
qemuVMDriverLock(void) {
6942 6943 6944 6945 6946
    qemuDriverLock(qemu_driver);
};


static void
6947
qemuVMDriverUnlock(void) {
6948 6949 6950 6951
    qemuDriverUnlock(qemu_driver);
};


6952 6953 6954 6955 6956 6957 6958 6959 6960
static int
qemuVMFilterRebuild(virConnectPtr conn ATTRIBUTE_UNUSED,
                    virHashIterator iter, void *data)
{
    virHashForEach(qemu_driver->domains.objs, iter, data);

    return 0;
}

S
Stefan Berger 已提交
6961 6962
static virNWFilterCallbackDriver qemuCallbackDriver = {
    .name = "QEMU",
6963 6964 6965
    .vmFilterRebuild = qemuVMFilterRebuild,
    .vmDriverLock = qemuVMDriverLock,
    .vmDriverUnlock = qemuVMDriverUnlock,
S
Stefan Berger 已提交
6966 6967
};

6968
int qemuRegister(void) {
6969 6970
    virRegisterDriver(&qemuDriver);
    virRegisterStateDriver(&qemuStateDriver);
S
Stefan Berger 已提交
6971
    virNWFilterRegisterCallbackDriver(&qemuCallbackDriver);
6972 6973
    return 0;
}