qemu_driver.c 247.6 KB
Newer Older
D
Daniel P. Berrange 已提交
1
/*
2
 * qemu_driver.c: core driver methods for managing qemu guests
D
Daniel P. Berrange 已提交
3
 *
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>
46
#include <byteswap.h>
D
Daniel P. Berrange 已提交
47

48

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

#include "virterror_internal.h"
#include "logging.h"
#include "datatypes.h"
65
#include "event.h"
66
#include "buf.h"
67
#include "util.h"
68
#include "nodeinfo.h"
69
#include "stats_linux.h"
70
#include "capabilities.h"
71
#include "memory.h"
72
#include "uuid.h"
73
#include "domain_conf.h"
74 75
#include "node_device_conf.h"
#include "pci.h"
76
#include "hostusb.h"
77
#include "processinfo.h"
C
Chris Lalancette 已提交
78
#include "libvirt_internal.h"
79
#include "xml.h"
80
#include "cpu/cpu.h"
81
#include "macvtap.h"
82
#include "sysinfo.h"
83
#include "domain_nwfilter.h"
84
#include "hooks.h"
85
#include "storage_file.h"
86
#include "files.h"
87
#include "fdstream.h"
88
#include "configmake.h"
H
Hu Tao 已提交
89
#include "threadpool.h"
90
#include "locking/lock_manager.h"
91

92 93
#define VIR_FROM_THIS VIR_FROM_QEMU

94 95
#define QEMU_NB_MEM_PARAM  3

96 97 98
#if HAVE_LINUX_KVM_H
# include <linux/kvm.h>
#endif
99

100 101
/* device for kvm ioctls */
#define KVM_DEVICE "/dev/kvm"
102

103 104 105 106 107 108 109 110 111 112 113
/* 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

114
#define QEMU_NB_BLKIO_PARAM  1
115 116

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

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

120 121
static int qemudShutdown(void);

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

127
static int qemudDomainGetMaxVcpus(virDomainPtr dom);
128

129
struct qemud_driver *qemu_driver = NULL;
130 131


132 133 134 135
struct qemuAutostartData {
    struct qemud_driver *driver;
    virConnectPtr conn;
};
136

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

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

        if (qemuDomainObjEndJob(vm) == 0)
            vm = NULL;
163
    }
164 165 166

    if (vm)
        virDomainObjUnlock(vm);
167 168
}

169

170
static void
171 172
qemuAutostartDomains(struct qemud_driver *driver)
{
173 174 175 176 177
    /* 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
     */
178 179 180
    virConnectPtr conn = virConnectOpen(driver->privileged ?
                                        "qemu:///system" :
                                        "qemu:///session");
181
    /* Ignoring NULL conn which is mostly harmless here */
182
    struct qemuAutostartData data = { driver, conn };
183

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

188 189
    if (conn)
        virConnectClose(conn);
190 191
}

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

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

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

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

218
    return 0;
219

220
error:
221
    VIR_ERROR(_("Failed to initialize security drivers"));
222 223 224
    virSecurityManagerFree(mgr);
    return -1;
}
225

226

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

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

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

247 248
    qemuDomainSetPrivateDataHooks(caps);
    qemuDomainSetNamespaceHooks(caps);
249

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

256 257
    /* Security driver data */
    const char *doi, *model;
258

259 260 261 262 263 264 265
    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;
266
    }
267

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

271
    return caps;
272

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

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

295 296 297 298
    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);
299
        goto cleanup;
300 301
    }

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

305 306 307 308 309
    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)));
310
        goto cleanup;
311 312
    }

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

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

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

326 327 328 329 330 331 332 333
        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;
        }
334

335 336 337 338 339 340 341 342
        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;
        }
343

344
        virDomainSnapshotAssignDef(&vm->snapshots, def);
345

346 347
        VIR_FREE(fullpath);
        VIR_FREE(xmlStr);
348 349
    }

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

359
    virResetLastError();
360

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

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

380 381
    if (VIR_ALLOC(qemu_driver) < 0)
        return -1;
382

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

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

394 395
    if (virDomainObjListInit(&qemu_driver->domains) < 0)
        goto out_of_memory;
396

397 398 399 400 401 402
    /* Init domain events */
    qemu_driver->domainEventState = virDomainEventStateNew(qemuDomainEventFlush,
                                                           qemu_driver,
                                                           NULL,
                                                           true);
    if (!qemu_driver->domainEventState)
403
        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 535 536 537 538 539 540
    /* We should always at least have the 'nop' manager, so
     * NULLs here are a fatal error
     */
    if (!qemu_driver->lockManager) {
        VIR_ERROR(_("Missing lock manager implementation"));
        goto error;
    }

541 542
    if (qemuSecurityInit(qemu_driver) < 0)
        goto error;
543

544 545 546
    if ((qemu_driver->caps = qemuCreateCapabilities(NULL,
                                                    qemu_driver)) == NULL)
        goto error;
547

548
    if ((qemu_driver->activePciHostdevs = pciDeviceListNew()) == NULL)
549
        goto error;
550

551 552 553 554 555 556
    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;
557
        }
558
        if (chown(qemu_driver->cacheDir, qemu_driver->user, qemu_driver->group) < 0) {
559
            virReportSystemError(errno,
560 561 562
                                 _("unable to set ownership of '%s' to %d:%d"),
                                 qemu_driver->cacheDir, qemu_driver->user, qemu_driver->group);
            goto error;
563
        }
564 565 566 567 568 569 570 571 572 573 574
        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;
575
        }
576
    }
577

578 579 580 581 582 583
    /* 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
584
     */
585 586 587 588 589
    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;
590

591 592 593 594 595
        if ((rc = virFileMakePath(mempath)) != 0) {
            virReportSystemError(rc,
                                 _("unable to create hugepage path %s"), mempath);
            VIR_FREE(mempath);
            goto error;
596
        }
597 598 599 600 601 602 603
        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 已提交
604
        }
E
Eric Blake 已提交
605

606
        qemu_driver->hugepage_path = mempath;
607
    }
608

609 610 611 612 613 614 615
    /* 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;
616

617 618 619
    conn = virConnectOpen(qemu_driver->privileged ?
                          "qemu:///system" :
                          "qemu:///session");
620

621
    qemuProcessReconnectAll(conn, qemu_driver);
622

623 624 625 626 627 628 629
    /* Then inactive persistent configs */
    if (virDomainLoadAllConfigs(qemu_driver->caps,
                                &qemu_driver->domains,
                                qemu_driver->configDir,
                                qemu_driver->autostartDir,
                                0, NULL, NULL) < 0)
        goto error;
630

631

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

635 636 637
    qemu_driver->workerPool = virThreadPoolNew(0, 1, processWatchdogEvent, qemu_driver);
    if (!qemu_driver->workerPool)
        goto error;
638

639 640 641 642
    qemuDriverUnlock(qemu_driver);

    qemuAutostartDomains(qemu_driver);

643 644
    if (conn)
        virConnectClose(conn);
645

646
    return 0;
647

648 649 650 651 652 653 654 655 656 657
out_of_memory:
    virReportOOMError();
error:
    if (qemu_driver)
        qemuDriverUnlock(qemu_driver);
    if (conn)
        virConnectClose(conn);
    VIR_FREE(base);
    VIR_FREE(driverConf);
    qemudShutdown();
658
    return -1;
659 660
}

661 662 663
static void qemudNotifyLoadDomain(virDomainObjPtr vm, int newVM, void *opaque)
{
    struct qemud_driver *driver = opaque;
664

665 666 667 668 669 670 671
    if (newVM) {
        virDomainEventPtr event =
            virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_DEFINED,
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED);
        if (event)
            qemuDomainEventQueue(driver, event);
E
Eric Blake 已提交
672
    }
673
}
E
Eric Blake 已提交
674

675 676 677 678 679 680 681 682 683 684
/**
 * 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;
685

686 687 688 689 690 691 692
    qemuDriverLock(qemu_driver);
    virDomainLoadAllConfigs(qemu_driver->caps,
                            &qemu_driver->domains,
                            qemu_driver->configDir,
                            qemu_driver->autostartDir,
                            0, qemudNotifyLoadDomain, qemu_driver);
    qemuDriverUnlock(qemu_driver);
693

694
    qemuAutostartDomains(qemu_driver);
695

696 697
    return 0;
}
S
Stefan Berger 已提交
698

699 700 701 702 703 704 705 706 707 708 709
/**
 * 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;
710

711 712
    if (!qemu_driver)
        return 0;
713

714 715 716 717 718 719
    /* 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;
}
720

721 722 723 724 725 726 727 728
/**
 * qemudShutdown:
 *
 * Shutdown the QEmu daemon, it will stop all active domains and networks
 */
static int
qemudShutdown(void) {
    int i;
729

730 731
    if (!qemu_driver)
        return -1;
732

733 734 735
    qemuDriverLock(qemu_driver);
    pciDeviceListFree(qemu_driver->activePciHostdevs);
    virCapabilitiesFree(qemu_driver->caps);
736

737 738
    virDomainObjListDeinit(&qemu_driver->domains);
    virBitmapFree(qemu_driver->reservedVNCPorts);
739

740
    virSysinfoDefFree(qemu_driver->hostsysinfo);
741

742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761
    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);
762

763
    virSecurityManagerFree(qemu_driver->securityManager);
764

765
    ebtablesContextFree(qemu_driver->ebtables);
766

767 768 769 770
    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 已提交
771 772
    }

773
    /* Free domain callback list */
774
    virDomainEventStateFree(qemu_driver->domainEventState);
D
Daniel P. Berrange 已提交
775

776 777
    if (qemu_driver->brctl)
        brShutdown(qemu_driver->brctl);
778

779
    virCgroupFree(&qemu_driver->cgroup);
780

781 782
    virLockManagerPluginUnref(qemu_driver->lockManager);

783 784 785 786
    qemuDriverUnlock(qemu_driver);
    virMutexDestroy(&qemu_driver->lock);
    virThreadPoolFree(qemu_driver->workerPool);
    VIR_FREE(qemu_driver);
787

788
    return 0;
789 790
}

791 792 793 794 795 796 797

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


798
static virDrvOpenStatus qemudOpen(virConnectPtr conn,
799
                                  virConnectAuthPtr auth ATTRIBUTE_UNUSED,
800
                                  int flags ATTRIBUTE_UNUSED) {
801
    if (conn->uri == NULL) {
802 803 804
        if (qemu_driver == NULL)
            return VIR_DRV_OPEN_DECLINED;

805
        conn->uri = xmlParseURI(qemu_driver->privileged ?
806 807
                                "qemu:///system" :
                                "qemu:///session");
808
        if (!conn->uri) {
809
            virReportOOMError();
810 811
            return VIR_DRV_OPEN_ERROR;
        }
812 813 814 815 816 817 818 819 820 821
    } 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;

822
        if (qemu_driver == NULL) {
823 824
            qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                            _("qemu state driver is not active"));
825 826 827
            return VIR_DRV_OPEN_ERROR;
        }

828
        if (conn->uri->path == NULL) {
829 830 831 832 833
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("no QEMU URI path given, try %s"),
                            qemu_driver->privileged
                            ? "qemu:///system"
                            : "qemu:///session");
834 835 836
                return VIR_DRV_OPEN_ERROR;
        }

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

    return VIR_DRV_OPEN_SUCCESS;
}

static int qemudClose(virConnectPtr conn) {
860
    struct qemud_driver *driver = conn->privateData;
861 862

    /* Get rid of callbacks registered for this conn */
863
    qemuDriverLock(driver);
864 865
    virDomainEventCallbackListRemoveConn(conn,
                                         driver->domainEventState->callbacks);
866
    qemuDriverUnlock(driver);
867 868 869 870 871 872

    conn->privateData = NULL;

    return 0;
}

D
Daniel Veillard 已提交
873 874 875 876 877
/* Which features are supported by this driver? */
static int
qemudSupportsFeature (virConnectPtr conn ATTRIBUTE_UNUSED, int feature)
{
    switch (feature) {
878
    case VIR_DRV_FEATURE_MIGRATION_V2:
879
    case VIR_DRV_FEATURE_MIGRATION_V3:
880 881 882 883
    case VIR_DRV_FEATURE_MIGRATION_P2P:
        return 1;
    default:
        return 0;
D
Daniel Veillard 已提交
884 885 886
    }
}

887
static const char *qemudGetType(virConnectPtr conn ATTRIBUTE_UNUSED) {
888
    return "QEMU";
889 890
}

891

892 893 894 895 896 897 898 899 900 901 902 903 904
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;
}


905 906 907 908
static int kvmGetMaxVCPUs(void) {
    int maxvcpus = 1;

    int r, fd;
909

910 911
    fd = open(KVM_DEVICE, O_RDONLY);
    if (fd < 0) {
912
        virReportSystemError(errno, _("Unable to open %s"), KVM_DEVICE);
913
        return -1;
914 915 916 917 918 919
    }

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

920
    VIR_FORCE_CLOSE(fd);
921 922 923 924
    return maxvcpus;
}


E
Eric Blake 已提交
925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940
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, "");
}

941
static int qemudGetMaxVCPUs(virConnectPtr conn ATTRIBUTE_UNUSED, const char *type) {
942 943 944
    if (!type)
        return 16;

945
    if (STRCASEEQ(type, "qemu"))
946 947
        return 16;

948
    if (STRCASEEQ(type, "kvm"))
949
        return kvmGetMaxVCPUs();
950

951
    if (STRCASEEQ(type, "kqemu"))
952
        return 1;
953

954 955
    qemuReportError(VIR_ERR_INVALID_ARG,
                    _("unknown type '%s'"), type);
956 957 958
    return -1;
}

959

960
static char *qemudGetCapabilities(virConnectPtr conn) {
961
    struct qemud_driver *driver = conn->privateData;
962
    virCapsPtr caps = NULL;
963
    char *xml = NULL;
964

965
    qemuDriverLock(driver);
966

967
    if ((caps = qemuCreateCapabilities(qemu_driver->caps,
968
                                       qemu_driver)) == NULL) {
969 970 971
        virCapabilitiesFree(caps);
        goto cleanup;
    }
972

973
    virCapabilitiesFree(qemu_driver->caps);
974 975 976
    qemu_driver->caps = caps;

    if ((xml = virCapabilitiesFormatXML(driver->caps)) == NULL)
977
        virReportOOMError();
978 979

cleanup:
980
    qemuDriverUnlock(driver);
981

982
    return xml;
983 984 985
}


986 987 988 989 990
static int
qemudGetProcessInfo(unsigned long long *cpuTime, int *lastCpu, int pid,
                    int tid)
{
    char *proc;
D
Daniel P. Berrange 已提交
991
    FILE *pidinfo;
992
    unsigned long long usertime, systime;
993 994
    int cpu;
    int ret;
D
Daniel P. Berrange 已提交
995

996
    if (tid)
997
        ret = virAsprintf(&proc, "/proc/%d/task/%d/stat", pid, tid);
998
    else
999 1000
        ret = virAsprintf(&proc, "/proc/%d/stat", pid);
    if (ret < 0)
D
Daniel P. Berrange 已提交
1001 1002 1003 1004
        return -1;

    if (!(pidinfo = fopen(proc, "r"))) {
        /* VM probably shut down, so fake 0 */
1005 1006 1007 1008
        if (cpuTime)
            *cpuTime = 0;
        if (lastCpu)
            *lastCpu = 0;
1009
        VIR_FREE(proc);
D
Daniel P. Berrange 已提交
1010 1011
        return 0;
    }
1012
    VIR_FREE(proc);
D
Daniel P. Berrange 已提交
1013

1014 1015 1016 1017 1018 1019 1020 1021 1022 1023
    /* 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) {
1024
        VIR_FORCE_FCLOSE(pidinfo);
1025
        VIR_WARN("cannot parse process status data");
1026
        errno = -EINVAL;
D
Daniel P. Berrange 已提交
1027 1028 1029 1030 1031 1032 1033 1034
        return -1;
    }

    /* We got jiffies
     * We want nanoseconds
     * _SC_CLK_TCK is jiffies per second
     * So calulate thus....
     */
1035 1036 1037 1038 1039
    if (cpuTime)
        *cpuTime = 1000ull * 1000ull * 1000ull * (usertime + systime) / (unsigned long long)sysconf(_SC_CLK_TCK);
    if (lastCpu)
        *lastCpu = cpu;

D
Daniel P. Berrange 已提交
1040

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

1044
    VIR_FORCE_FCLOSE(pidinfo);
D
Daniel P. Berrange 已提交
1045 1046 1047 1048 1049

    return 0;
}


1050
static virDomainPtr qemudDomainLookupByID(virConnectPtr conn,
1051
                                          int id) {
1052 1053 1054 1055
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;

1056
    qemuDriverLock(driver);
1057
    vm  = virDomainFindByID(&driver->domains, id);
1058
    qemuDriverUnlock(driver);
1059 1060

    if (!vm) {
1061 1062
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching id %d"), id);
1063
        goto cleanup;
1064 1065
    }

1066
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1067
    if (dom) dom->id = vm->def->id;
1068 1069

cleanup:
1070 1071
    if (vm)
        virDomainObjUnlock(vm);
1072 1073
    return dom;
}
1074

1075
static virDomainPtr qemudDomainLookupByUUID(virConnectPtr conn,
1076
                                            const unsigned char *uuid) {
1077 1078 1079
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
1080

1081
    qemuDriverLock(driver);
1082
    vm = virDomainFindByUUID(&driver->domains, uuid);
1083 1084
    qemuDriverUnlock(driver);

1085
    if (!vm) {
1086 1087
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(uuid, uuidstr);
1088 1089
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1090
        goto cleanup;
1091 1092
    }

1093
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1094
    if (dom) dom->id = vm->def->id;
1095 1096

cleanup:
1097 1098
    if (vm)
        virDomainObjUnlock(vm);
1099 1100
    return dom;
}
1101

1102
static virDomainPtr qemudDomainLookupByName(virConnectPtr conn,
1103
                                            const char *name) {
1104 1105 1106
    struct qemud_driver *driver = conn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
1107

1108
    qemuDriverLock(driver);
1109
    vm = virDomainFindByName(&driver->domains, name);
1110 1111
    qemuDriverUnlock(driver);

1112
    if (!vm) {
1113 1114
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching name '%s'"), name);
1115
        goto cleanup;
1116 1117
    }

1118
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1119
    if (dom) dom->id = vm->def->id;
1120 1121

cleanup:
1122 1123
    if (vm)
        virDomainObjUnlock(vm);
1124 1125 1126
    return dom;
}

1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137

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 已提交
1138 1139 1140 1141
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161
        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 已提交
1162 1163 1164 1165
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1166 1167 1168 1169 1170 1171 1172 1173 1174 1175
        goto cleanup;
    }
    ret = obj->persistent;

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

1176 1177 1178 1179 1180 1181 1182 1183 1184 1185
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 已提交
1186 1187 1188 1189
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1190 1191 1192 1193 1194 1195 1196 1197 1198
        goto cleanup;
    }
    ret = obj->updated;

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

1200
static int qemudGetVersion(virConnectPtr conn, unsigned long *version) {
1201 1202 1203
    struct qemud_driver *driver = conn->privateData;
    int ret = -1;

1204
    qemuDriverLock(driver);
1205
    if (qemuCapsExtractVersion(driver->caps, &driver->qemuVersion) < 0)
1206
        goto cleanup;
1207

1208
    *version = driver->qemuVersion;
1209 1210 1211
    ret = 0;

cleanup:
1212
    qemuDriverUnlock(driver);
1213
    return ret;
D
Daniel P. Berrange 已提交
1214 1215
}

1216
static int qemudListDomains(virConnectPtr conn, int *ids, int nids) {
1217
    struct qemud_driver *driver = conn->privateData;
1218
    int n;
1219

1220
    qemuDriverLock(driver);
1221
    n = virDomainObjListGetActiveIDs(&driver->domains, ids, nids);
1222
    qemuDriverUnlock(driver);
1223

1224
    return n;
D
Daniel P. Berrange 已提交
1225
}
1226

1227
static int qemudNumDomains(virConnectPtr conn) {
1228
    struct qemud_driver *driver = conn->privateData;
1229
    int n;
1230

1231
    qemuDriverLock(driver);
1232
    n = virDomainObjListNumOfDomains(&driver->domains, 1);
1233
    qemuDriverUnlock(driver);
1234

1235
    return n;
D
Daniel P. Berrange 已提交
1236
}
1237

1238
static virDomainPtr qemudDomainCreate(virConnectPtr conn, const char *xml,
1239
                                      unsigned int flags) {
1240
    struct qemud_driver *driver = conn->privateData;
1241
    virDomainDefPtr def;
1242
    virDomainObjPtr vm = NULL;
1243
    virDomainPtr dom = NULL;
1244
    virDomainEventPtr event = NULL;
D
Daniel P. Berrange 已提交
1245

1246
    virCheckFlags(VIR_DOMAIN_START_PAUSED, NULL);
1247

1248
    qemuDriverLock(driver);
1249
    if (!(def = virDomainDefParseString(driver->caps, xml,
1250
                                        VIR_DOMAIN_XML_INACTIVE)))
1251
        goto cleanup;
1252

1253
    if (virSecurityManagerVerify(driver->securityManager, def) < 0)
1254 1255
        goto cleanup;

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

1259 1260 1261
    if (qemudCanonicalizeMachine(driver, def) < 0)
        goto cleanup;

1262
    if (qemuDomainAssignPCIAddresses(def) < 0)
1263 1264
        goto cleanup;

1265
    if (!(vm = virDomainAssignDef(driver->caps,
1266
                                  &driver->domains,
1267
                                  def, false)))
1268 1269 1270
        goto cleanup;

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

1272 1273 1274
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup; /* XXXX free the 'vm' we created ? */

1275 1276 1277
    if (qemuProcessStart(conn, driver, vm, NULL,
                         (flags & VIR_DOMAIN_START_PAUSED) != 0,
                         -1, NULL, VIR_VM_OP_CREATE) < 0) {
1278
        qemuAuditDomainStart(vm, "booted", false);
1279 1280 1281
        if (qemuDomainObjEndJob(vm) > 0)
            virDomainRemoveInactive(&driver->domains,
                                    vm);
1282
        vm = NULL;
1283
        goto cleanup;
D
Daniel P. Berrange 已提交
1284
    }
1285 1286 1287 1288

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

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

1294 1295 1296
    if (vm &&
        qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
1297

1298 1299
cleanup:
    virDomainDefFree(def);
1300 1301
    if (vm)
        virDomainObjUnlock(vm);
1302 1303
    if (event)
        qemuDomainEventQueue(driver, event);
1304
    qemuDriverUnlock(driver);
1305
    return dom;
D
Daniel P. Berrange 已提交
1306 1307 1308
}


1309
static int qemudDomainSuspend(virDomainPtr dom) {
1310 1311 1312
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1313
    virDomainEventPtr event = NULL;
1314
    qemuDomainObjPrivatePtr priv;
1315

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

D
Daniel P. Berrange 已提交
1319
    if (!vm) {
1320 1321
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1322 1323
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1324
        goto cleanup;
D
Daniel P. Berrange 已提交
1325
    }
D
Daniel P. Berrange 已提交
1326
    if (!virDomainObjIsActive(vm)) {
1327 1328
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
1329
        goto cleanup;
D
Daniel P. Berrange 已提交
1330
    }
1331 1332 1333

    priv = vm->privateData;

1334
    if (priv->jobActive == QEMU_JOB_MIGRATION_OUT) {
J
Jiri Denemark 已提交
1335
        if (virDomainObjGetState(vm, NULL) != VIR_DOMAIN_PAUSED) {
1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348
            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"));
1349
            goto endjob;
1350
        }
J
Jiri Denemark 已提交
1351 1352
        if (virDomainObjGetState(vm, NULL) != VIR_DOMAIN_PAUSED) {
            if (qemuProcessStopCPUs(driver, vm, VIR_DOMAIN_PAUSED_USER) < 0) {
1353
                goto endjob;
1354
            }
1355 1356 1357 1358 1359 1360 1361
            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 已提交
1362
    }
1363

1364
endjob:
1365 1366
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
1367

1368
cleanup:
1369 1370
    if (vm)
        virDomainObjUnlock(vm);
1371

1372
    if (event)
1373
        qemuDomainEventQueue(driver, event);
1374
    qemuDriverUnlock(driver);
1375
    return ret;
D
Daniel P. Berrange 已提交
1376 1377 1378
}


1379
static int qemudDomainResume(virDomainPtr dom) {
1380 1381 1382
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1383
    virDomainEventPtr event = NULL;
1384

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

D
Daniel P. Berrange 已提交
1388
    if (!vm) {
1389 1390
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1391 1392
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1393
        goto cleanup;
D
Daniel P. Berrange 已提交
1394
    }
1395 1396 1397 1398

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

D
Daniel P. Berrange 已提交
1399
    if (!virDomainObjIsActive(vm)) {
1400 1401
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
1402
        goto endjob;
D
Daniel P. Berrange 已提交
1403
    }
J
Jiri Denemark 已提交
1404 1405 1406
    if (virDomainObjGetState(vm, NULL) == VIR_DOMAIN_PAUSED) {
        if (qemuProcessStartCPUs(driver, vm, dom->conn,
                                 VIR_DOMAIN_RUNNING_UNPAUSED) < 0) {
1407
            if (virGetLastError() == NULL)
1408 1409
                qemuReportError(VIR_ERR_OPERATION_FAILED,
                                "%s", _("resume operation failed"));
1410
            goto endjob;
1411
        }
1412 1413 1414
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_RESUMED,
                                         VIR_DOMAIN_EVENT_RESUMED_UNPAUSED);
D
Daniel P. Berrange 已提交
1415
    }
1416
    if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
1417
        goto endjob;
1418 1419
    ret = 0;

1420
endjob:
1421 1422
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
1423

1424
cleanup:
1425 1426
    if (vm)
        virDomainObjUnlock(vm);
1427
    if (event)
1428
        qemuDomainEventQueue(driver, event);
1429
    qemuDriverUnlock(driver);
1430
    return ret;
D
Daniel P. Berrange 已提交
1431 1432 1433
}


1434
static int qemudDomainShutdown(virDomainPtr dom) {
1435 1436 1437
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1438
    qemuDomainObjPrivatePtr priv;
1439

1440
    qemuDriverLock(driver);
1441
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1442 1443
    qemuDriverUnlock(driver);

1444
    if (!vm) {
1445 1446
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1447 1448
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1449
        goto cleanup;
1450 1451
    }

1452 1453 1454
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
1455
    if (!virDomainObjIsActive(vm)) {
1456 1457
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
1458
        goto endjob;
1459 1460
    }

1461
    priv = vm->privateData;
1462 1463 1464
    qemuDomainObjEnterMonitor(vm);
    ret = qemuMonitorSystemPowerdown(priv->mon);
    qemuDomainObjExitMonitor(vm);
1465

1466
endjob:
1467 1468
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
1469

1470
cleanup:
1471 1472
    if (vm)
        virDomainObjUnlock(vm);
1473
    return ret;
1474 1475 1476
}


1477
static int qemudDomainDestroy(virDomainPtr dom) {
1478 1479 1480
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1481
    virDomainEventPtr event = NULL;
1482

1483
    qemuDriverLock(driver);
1484
    vm  = virDomainFindByUUID(&driver->domains, dom->uuid);
D
Daniel P. Berrange 已提交
1485
    if (!vm) {
1486 1487
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1488 1489
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1490
        goto cleanup;
D
Daniel P. Berrange 已提交
1491
    }
1492

1493 1494 1495 1496 1497 1498 1499
    /* Although qemuProcessStop does this already, there may
     * be an outstanding job active. We want to make sure we
     * can kill the process even if a job is active. Killing
     * it now means the job will be released
     */
    qemuProcessKill(vm);

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

D
Daniel P. Berrange 已提交
1503
    if (!virDomainObjIsActive(vm)) {
1504 1505
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
1506
        goto endjob;
1507
    }
1508

J
Jiri Denemark 已提交
1509
    qemuProcessStop(driver, vm, 0, VIR_DOMAIN_SHUTOFF_DESTROYED);
1510 1511 1512
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_DESTROYED);
1513
    qemuAuditDomainStop(vm, "destroyed");
1514

1515
    if (!vm->persistent) {
1516 1517 1518
        if (qemuDomainObjEndJob(vm) > 0)
            virDomainRemoveInactive(&driver->domains,
                                    vm);
1519 1520
        vm = NULL;
    }
1521 1522
    ret = 0;

1523
endjob:
1524 1525 1526
    if (vm &&
        qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
1527

1528
cleanup:
1529 1530
    if (vm)
        virDomainObjUnlock(vm);
1531 1532
    if (event)
        qemuDomainEventQueue(driver, event);
1533
    qemuDriverUnlock(driver);
1534
    return ret;
D
Daniel P. Berrange 已提交
1535 1536 1537
}


1538
static char *qemudDomainGetOSType(virDomainPtr dom) {
1539 1540 1541
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *type = NULL;
1542

1543
    qemuDriverLock(driver);
1544
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1545
    qemuDriverUnlock(driver);
1546
    if (!vm) {
1547 1548
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1549 1550
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1551
        goto cleanup;
1552 1553
    }

1554
    if (!(type = strdup(vm->def->os.type)))
1555
        virReportOOMError();
1556 1557

cleanup:
1558 1559
    if (vm)
        virDomainObjUnlock(vm);
1560 1561 1562
    return type;
}

1563 1564
/* Returns max memory in kb, 0 if error */
static unsigned long qemudDomainGetMaxMemory(virDomainPtr dom) {
1565 1566 1567
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    unsigned long ret = 0;
1568

1569
    qemuDriverLock(driver);
1570
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1571 1572
    qemuDriverUnlock(driver);

1573
    if (!vm) {
1574 1575
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1576 1577
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1578
        goto cleanup;
1579 1580
    }

1581
    ret = vm->def->mem.max_balloon;
1582 1583

cleanup:
1584 1585
    if (vm)
        virDomainObjUnlock(vm);
1586
    return ret;
1587 1588
}

1589 1590
static int qemudDomainSetMemoryFlags(virDomainPtr dom, unsigned long newmem,
                                     unsigned int flags) {
1591
    struct qemud_driver *driver = dom->conn->privateData;
1592
    qemuDomainObjPrivatePtr priv;
1593
    virDomainObjPtr vm;
1594
    virDomainDefPtr persistentDef = NULL;
1595
    int ret = -1, r;
1596
    bool isActive;
1597

1598
    virCheckFlags(VIR_DOMAIN_MEM_LIVE |
1599 1600
                  VIR_DOMAIN_MEM_CONFIG |
                  VIR_DOMAIN_MEM_MAXIMUM, -1);
1601

1602
    qemuDriverLock(driver);
1603
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1604
    qemuDriverUnlock(driver);
1605
    if (!vm) {
1606 1607
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1608 1609
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1610
        goto cleanup;
1611 1612
    }

1613 1614 1615
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

1616 1617 1618 1619 1620 1621 1622 1623
    isActive = virDomainObjIsActive(vm);

    if (flags == VIR_DOMAIN_MEM_CURRENT) {
        if (isActive)
            flags = VIR_DOMAIN_MEM_LIVE;
        else
            flags = VIR_DOMAIN_MEM_CONFIG;
    }
1624 1625 1626 1627 1628 1629
    if (flags == VIR_DOMAIN_MEM_MAXIMUM) {
        if (isActive)
            flags = VIR_DOMAIN_MEM_LIVE | VIR_DOMAIN_MEM_MAXIMUM;
        else
            flags = VIR_DOMAIN_MEM_CONFIG | VIR_DOMAIN_MEM_MAXIMUM;
    }
1630 1631

    if (!isActive && (flags & VIR_DOMAIN_MEM_LIVE)) {
1632 1633 1634 1635 1636
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto endjob;
    }

1637 1638 1639 1640 1641 1642 1643 1644 1645
    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;
    }
1646

1647 1648 1649 1650 1651 1652 1653
    if (flags & VIR_DOMAIN_MEM_MAXIMUM) {
        /* resize the maximum memory */

        if (flags & VIR_DOMAIN_MEM_LIVE) {
            qemuReportError(VIR_ERR_OPERATION_INVALID, "%s",
                            _("cannot resize the maximum memory on an "
                              "active domain"));
1654
            goto endjob;
1655
        }
1656

1657
        if (flags & VIR_DOMAIN_MEM_CONFIG) {
E
Eric Blake 已提交
1658 1659
            /* Help clang 2.8 decipher the logic flow.  */
            sa_assert(persistentDef);
1660 1661 1662 1663
            persistentDef->mem.max_balloon = newmem;
            if (persistentDef->mem.cur_balloon > newmem)
                persistentDef->mem.cur_balloon = newmem;
            ret = virDomainSaveConfig(driver->configDir, persistentDef);
1664 1665 1666
            goto endjob;
        }

1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694
    } else {
        /* resize the current memory */

        if (newmem > vm->def->mem.max_balloon) {
            qemuReportError(VIR_ERR_INVALID_ARG, "%s",
                            _("cannot set memory higher than max memory"));
            goto endjob;
        }

        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) {
E
Eric Blake 已提交
1695
            sa_assert(persistentDef);
1696 1697 1698 1699
            persistentDef->mem.cur_balloon = newmem;
            ret = virDomainSaveConfig(driver->configDir, persistentDef);
            goto endjob;
        }
1700
    }
1701

1702
    ret = 0;
1703
endjob:
1704 1705
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
1706

1707
cleanup:
1708 1709
    if (vm)
        virDomainObjUnlock(vm);
1710
    return ret;
1711 1712
}

1713 1714
static int qemudDomainSetMemory(virDomainPtr dom, unsigned long newmem)
{
1715 1716 1717
    return qemudDomainSetMemoryFlags(dom, newmem, VIR_DOMAIN_MEM_LIVE);
}

1718 1719 1720 1721 1722
static int qemudDomainSetMaxMemory(virDomainPtr dom, unsigned long memory)
{
    return qemudDomainSetMemoryFlags(dom, memory, VIR_DOMAIN_MEM_MAXIMUM);
}

1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766
static int qemuDomainInjectNMI(virDomainPtr domain, unsigned int flags)
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    int ret = -1;
    qemuDomainObjPrivatePtr priv;

    virCheckFlags(0, -1);

    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 (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
    ret = qemuMonitorInjectNMI(priv->mon);
    qemuDomainObjExitMonitorWithDriver(driver, vm);
    if (qemuDomainObjEndJob(vm) == 0) {
        vm = NULL;
        goto cleanup;
    }

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

1767
static int qemudDomainGetInfo(virDomainPtr dom,
1768 1769
                              virDomainInfoPtr info)
{
1770 1771 1772
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
1773 1774
    int err;
    unsigned long balloon;
1775

1776
    qemuDriverLock(driver);
1777
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
1778
    qemuDriverUnlock(driver);
D
Daniel P. Berrange 已提交
1779
    if (!vm) {
1780 1781
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
1782 1783
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
1784
        goto cleanup;
D
Daniel P. Berrange 已提交
1785 1786
    }

J
Jiri Denemark 已提交
1787
    info->state = virDomainObjGetState(vm, NULL);
D
Daniel P. Berrange 已提交
1788

D
Daniel P. Berrange 已提交
1789
    if (!virDomainObjIsActive(vm)) {
1790
        info->cpuTime = 0;
D
Daniel P. Berrange 已提交
1791
    } else {
1792
        if (qemudGetProcessInfo(&(info->cpuTime), NULL, vm->pid, 0) < 0) {
E
Eric Blake 已提交
1793 1794
            qemuReportError(VIR_ERR_OPERATION_FAILED, "%s",
                            _("cannot read cputime for domain"));
1795
            goto cleanup;
D
Daniel P. Berrange 已提交
1796 1797 1798
        }
    }

1799
    info->maxMem = vm->def->mem.max_balloon;
1800

D
Daniel P. Berrange 已提交
1801
    if (virDomainObjIsActive(vm)) {
1802
        qemuDomainObjPrivatePtr priv = vm->privateData;
1803 1804 1805

        if ((vm->def->memballoon != NULL) &&
            (vm->def->memballoon->model == VIR_DOMAIN_MEMBALLOON_MODEL_NONE)) {
1806
            info->memory = vm->def->mem.max_balloon;
1807
        } else if (!priv->jobActive) {
1808 1809
            if (qemuDomainObjBeginJob(vm) < 0)
                goto cleanup;
1810 1811 1812 1813 1814 1815 1816 1817 1818
            if (!virDomainObjIsActive(vm))
                err = 0;
            else {
                qemuDomainObjEnterMonitor(vm);
                err = qemuMonitorGetBalloonInfo(priv->mon, &balloon);
                qemuDomainObjExitMonitor(vm);
            }
            if (qemuDomainObjEndJob(vm) == 0) {
                vm = NULL;
1819 1820 1821
                goto cleanup;
            }

1822 1823
            if (err < 0)
                goto cleanup;
1824 1825
            if (err == 0)
                /* Balloon not supported, so maxmem is always the allocation */
1826
                info->memory = vm->def->mem.max_balloon;
1827 1828 1829
            else
                info->memory = balloon;
        } else {
1830
            info->memory = vm->def->mem.cur_balloon;
1831
        }
1832
    } else {
1833
        info->memory = vm->def->mem.cur_balloon;
1834 1835
    }

1836
    info->nrVirtCpu = vm->def->vcpus;
1837 1838 1839
    ret = 0;

cleanup:
1840 1841
    if (vm)
        virDomainObjUnlock(vm);
1842
    return ret;
D
Daniel P. Berrange 已提交
1843 1844
}

1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868
static int
qemuDomainGetState(virDomainPtr dom,
                   int *state,
                   int *reason,
                   unsigned int flags)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;

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

J
Jiri Denemark 已提交
1869
    *state = virDomainObjGetState(vm, reason);
1870 1871 1872 1873 1874 1875 1876 1877
    ret = 0;

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

D
Daniel P. Berrange 已提交
1878

1879
#define QEMUD_SAVE_MAGIC "LibvirtQemudSave"
1880 1881 1882
#define QEMUD_SAVE_VERSION 2

enum qemud_save_formats {
1883 1884 1885
    QEMUD_SAVE_FORMAT_RAW = 0,
    QEMUD_SAVE_FORMAT_GZIP = 1,
    QEMUD_SAVE_FORMAT_BZIP2 = 2,
1886 1887
    /*
     * Deprecated by xz and never used as part of a release
1888
     * QEMUD_SAVE_FORMAT_LZMA
1889 1890
     */
    QEMUD_SAVE_FORMAT_XZ = 3,
1891
    QEMUD_SAVE_FORMAT_LZOP = 4,
1892 1893 1894
    /* Note: add new members only at the end.
       These values are used in the on-disk format.
       Do not change or re-use numbers. */
1895 1896

    QEMUD_SAVE_FORMAT_LAST
1897
};
1898

1899 1900 1901 1902 1903
VIR_ENUM_DECL(qemudSaveCompression)
VIR_ENUM_IMPL(qemudSaveCompression, QEMUD_SAVE_FORMAT_LAST,
              "raw",
              "gzip",
              "bzip2",
1904 1905
              "xz",
              "lzop")
1906

1907 1908
struct qemud_save_header {
    char magic[sizeof(QEMUD_SAVE_MAGIC)-1];
1909 1910 1911 1912 1913
    uint32_t version;
    uint32_t xml_len;
    uint32_t was_running;
    uint32_t compressed;
    uint32_t unused[15];
1914 1915
};

1916 1917 1918 1919 1920 1921 1922 1923 1924
static inline void
bswap_header(struct qemud_save_header *hdr) {
    hdr->version = bswap_32(hdr->version);
    hdr->xml_len = bswap_32(hdr->xml_len);
    hdr->was_running = bswap_32(hdr->was_running);
    hdr->compressed = bswap_32(hdr->compressed);
}


1925
/* return -errno on failure, or 0 on success */
E
Eric Blake 已提交
1926 1927 1928 1929
static int
qemuDomainSaveHeader(int fd, const char *path, char *xml,
                     struct qemud_save_header *header)
{
1930 1931
    int ret = 0;

E
Eric Blake 已提交
1932
    if (safewrite(fd, header, sizeof(*header)) != sizeof(*header)) {
1933
        ret = -errno;
1934
        qemuReportError(VIR_ERR_OPERATION_FAILED,
1935
                        _("failed to write header to domain save file '%s'"),
E
Eric Blake 已提交
1936
                        path);
1937 1938 1939
        goto endjob;
    }

E
Eric Blake 已提交
1940
    if (safewrite(fd, xml, header->xml_len) != header->xml_len) {
1941
        ret = -errno;
1942
        qemuReportError(VIR_ERR_OPERATION_FAILED,
E
Eric Blake 已提交
1943
                         _("failed to write xml to '%s'"), path);
1944 1945 1946 1947 1948 1949
        goto endjob;
    }
endjob:
    return ret;
}

1950 1951 1952 1953 1954 1955 1956 1957 1958
/* 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));
}

1959 1960 1961
/* This internal function expects the driver lock to already be held on
 * entry and the vm must be active.
 */
1962 1963
static int qemudDomainSaveFlag(struct qemud_driver *driver, virDomainPtr dom,
                               virDomainObjPtr vm, const char *path,
1964
                               int compressed)
1965
{
1966
    char *xml = NULL;
1967
    struct qemud_save_header header;
1968
    bool bypassSecurityDriver = false;
1969
    int ret = -1;
1970
    int rc;
1971
    virDomainEventPtr event = NULL;
1972
    qemuDomainObjPrivatePtr priv;
1973
    struct stat sb;
1974
    bool is_reg = false;
1975
    unsigned long long offset;
1976
    int fd = -1;
1977 1978
    uid_t uid = getuid();
    gid_t gid = getgid();
1979 1980 1981 1982 1983

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

1984
    header.compressed = compressed;
1985

1986
    priv = vm->privateData;
1987

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

O
Osier Yang 已提交
1991 1992
    priv->jobActive = QEMU_JOB_SAVE;

1993 1994 1995
    memset(&priv->jobInfo, 0, sizeof(priv->jobInfo));
    priv->jobInfo.type = VIR_DOMAIN_JOB_UNBOUNDED;

1996
    /* Pause */
J
Jiri Denemark 已提交
1997
    if (virDomainObjGetState(vm, NULL) == VIR_DOMAIN_RUNNING) {
1998
        header.was_running = 1;
J
Jiri Denemark 已提交
1999
        if (qemuProcessStopCPUs(driver, vm, VIR_DOMAIN_PAUSED_SAVE) < 0)
2000
            goto endjob;
2001 2002 2003 2004 2005 2006

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

    /* Get XML for the domain */
2010
    xml = virDomainDefFormat(vm->def, VIR_DOMAIN_XML_SECURE);
2011
    if (!xml) {
2012 2013
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to get domain xml"));
2014
        goto endjob;
2015 2016 2017
    }
    header.xml_len = strlen(xml) + 1;

2018 2019 2020 2021 2022 2023 2024 2025
    /* 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 */
2026
        is_reg = true;
2027
    } else {
2028
        is_reg = !!S_ISREG(sb.st_mode);
2029 2030 2031 2032 2033 2034 2035 2036
        /* If the path is regular local file which exists
         * already and dynamic_ownership is off, we don't
         * want to change it's ownership, just open it as-is */
        if (is_reg && !driver->dynamicOwnership &&
            virStorageFileIsSharedFS(path) == 0) {
            uid=sb.st_uid;
            gid=sb.st_gid;
        }
2037 2038
    }

2039 2040 2041 2042 2043
    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
2044 2045
     * it by padding the XML string with NULLs.
     */
2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058
    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;
    }
2059

2060
    /* Obtain the file handle.  */
2061

2062
    /* First try creating the file as root */
2063
    if (!is_reg) {
2064
        fd = open(path, O_WRONLY | O_TRUNC);
2065 2066
        if (fd < 0) {
            virReportSystemError(errno, _("unable to open %s"), path);
2067 2068
            goto endjob;
        }
2069
    } else {
2070 2071
        if ((fd = virFileOpenAs(path, O_CREAT|O_TRUNC|O_WRONLY,
                                S_IRUSR|S_IWUSR,
2072
                                uid, gid, 0)) < 0) {
2073
            /* If we failed as root, and the error was permission-denied
2074 2075
               (EACCES or EPERM), assume it's on a network-connected share
               where root access is restricted (eg, root-squashed NFS). If the
2076 2077 2078
               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 */
2079
            rc = fd;
2080
            if (((rc != -EACCES) && (rc != -EPERM)) ||
2081
                driver->user == getuid()) {
2082 2083
                virReportSystemError(-rc,
                                    _("Failed to create domain save file '%s'"),
2084 2085 2086
                                     path);
                goto endjob;
            }
2087

2088
            /* On Linux we can also verify the FS-type of the directory. */
2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105
            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:
2106
                   /* local file - log the error returned by virFileOpenAs */
2107
                   virReportSystemError(-rc,
2108
                                    _("Failed to create domain save file '%s'"),
2109 2110 2111
                                        path);
                   goto endjob;
                   break;
2112 2113 2114

            }

2115
            /* Retry creating the file as driver->user */
2116

2117 2118 2119 2120
            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) {
2121 2122 2123
                virReportSystemError(-fd,
                                   _("Error from child process creating '%s'"),
                                     path);
2124 2125
                goto endjob;
            }
2126

2127 2128 2129
            /* 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 */
2130

2131
            bypassSecurityDriver = true;
2132
        }
2133
    }
2134

2135
    /* Write header to file, followed by XML */
E
Eric Blake 已提交
2136
    if (qemuDomainSaveHeader(fd, path, xml, &header) < 0) {
2137 2138 2139 2140
        VIR_FORCE_CLOSE(fd);
        goto endjob;
    }

2141
    /* Perform the migration */
2142
    if (qemuMigrationToFile(driver, vm, fd, offset, path,
2143 2144
                            qemuCompressProgramName(compressed),
                            is_reg, bypassSecurityDriver) < 0)
2145
        goto endjob;
2146 2147
    if (VIR_CLOSE(fd) < 0) {
        virReportSystemError(errno, _("unable to close %s"), path);
2148
        goto endjob;
2149 2150
    }

2151 2152
    ret = 0;

2153
    /* Shut it down */
J
Jiri Denemark 已提交
2154
    qemuProcessStop(driver, vm, 0, VIR_DOMAIN_SHUTOFF_SAVED);
2155
    qemuAuditDomainStop(vm, "saved");
2156 2157 2158
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STOPPED,
                                     VIR_DOMAIN_EVENT_STOPPED_SAVED);
2159
    if (!vm->persistent) {
2160 2161 2162
        if (qemuDomainObjEndJob(vm) > 0)
            virDomainRemoveInactive(&driver->domains,
                                    vm);
2163 2164
        vm = NULL;
    }
2165

2166
endjob:
2167
    if (vm) {
2168
        if (ret != 0) {
2169
            if (header.was_running && virDomainObjIsActive(vm)) {
J
Jiri Denemark 已提交
2170 2171
                rc = qemuProcessStartCPUs(driver, vm, dom->conn,
                                          VIR_DOMAIN_RUNNING_SAVE_CANCELED);
2172
                if (rc < 0)
2173
                    VIR_WARN("Unable to resume guest CPUs after save failure");
2174
            }
2175
        }
2176
        if (qemuDomainObjEndJob(vm) == 0)
2177
            vm = NULL;
2178
    }
2179

2180
cleanup:
2181
    VIR_FORCE_CLOSE(fd);
2182
    VIR_FREE(xml);
2183
    if (ret != 0 && is_reg)
2184
        unlink(path);
2185 2186
    if (event)
        qemuDomainEventQueue(driver, event);
2187
    return ret;
D
Daniel P. Berrange 已提交
2188 2189
}

2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205
/* 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;
}

2206 2207 2208 2209
static int qemudDomainSave(virDomainPtr dom, const char *path)
{
    struct qemud_driver *driver = dom->conn->privateData;
    int compressed;
2210 2211 2212 2213
    int ret = -1;
    virDomainObjPtr vm = NULL;

    qemuDriverLock(driver);
2214 2215 2216 2217 2218 2219 2220 2221 2222

    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"));
2223
            goto cleanup;
2224
        }
2225 2226 2227 2228
        if (!qemudCompressProgramAvailable(compressed)) {
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("Compression program for image format "
                                    "in configuration file isn't available"));
2229
            goto cleanup;
2230
        }
2231 2232
    }

2233 2234 2235 2236 2237 2238 2239 2240 2241
    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;
    }

2242 2243 2244 2245 2246 2247
    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
    }

2248 2249 2250 2251 2252 2253 2254 2255
    ret = qemudDomainSaveFlag(driver, dom, vm, path, compressed);

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

    return ret;
2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278
}

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;

2279
    virCheckFlags(0, -1);
2280 2281 2282 2283 2284 2285 2286 2287

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

2291 2292 2293 2294 2295 2296
    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
    }

2297 2298
    name = qemuDomainManagedSavePath(driver, vm);
    if (name == NULL)
2299
        goto cleanup;
2300

2301
    VIR_INFO("Saving state to %s", name);
2302 2303

    compressed = QEMUD_SAVE_FORMAT_RAW;
2304
    ret = qemudDomainSaveFlag(driver, dom, vm, name, compressed);
2305 2306 2307 2308 2309

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
2310 2311 2312
    VIR_FREE(name);

    return ret;
2313 2314 2315 2316 2317 2318 2319 2320 2321 2322
}

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

2323
    virCheckFlags(0, -1);
2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356

    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;

2357
    virCheckFlags(0, -1);
2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381

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

H
Hu Tao 已提交
2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397
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;
    }

2398
    if (qemuMigrationToFile(driver, vm, fd, 0, path,
2399 2400 2401
                            qemuCompressProgramName(compress), true, false) < 0)
        goto cleanup;

H
Hu Tao 已提交
2402 2403 2404 2405 2406 2407 2408
    if (VIR_CLOSE(fd) < 0) {
        virReportSystemError(errno,
                             _("unable to save file %s"),
                             path);
        goto cleanup;
    }

2409
    ret = 0;
H
Hu Tao 已提交
2410 2411 2412 2413 2414 2415 2416

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

2417 2418 2419 2420 2421
static enum qemud_save_formats
getCompressionType(struct qemud_driver *driver)
{
    int compress = QEMUD_SAVE_FORMAT_RAW;

2422 2423 2424 2425 2426 2427 2428
    /*
     * 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) {
2429 2430 2431 2432
            qemuReportError(VIR_ERR_OPERATION_FAILED, "%s",
                            _("Invalid dump image format specified in "
                              "configuration file, using raw"));
            return QEMUD_SAVE_FORMAT_RAW;
2433
        }
2434 2435 2436
        if (!qemudCompressProgramAvailable(compress)) {
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("Compression program for dump image format "
2437 2438 2439
                                    "in configuration file isn't available, "
                                    "using raw"));
            return QEMUD_SAVE_FORMAT_RAW;
2440
        }
2441
    }
2442 2443 2444 2445 2446
    return compress;
}

static int qemudDomainCoreDump(virDomainPtr dom,
                               const char *path,
2447 2448
                               int flags)
{
2449 2450 2451
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int resume = 0, paused = 0;
H
Hu Tao 已提交
2452
    int ret = -1;
2453 2454 2455
    virDomainEventPtr event = NULL;
    qemuDomainObjPrivatePtr priv;

2456 2457
    virCheckFlags(VIR_DUMP_LIVE | VIR_DUMP_CRASH, -1);

P
Paolo Bonzini 已提交
2458 2459 2460 2461 2462 2463
    qemuDriverLock(driver);
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);

    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2464 2465
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
P
Paolo Bonzini 已提交
2466 2467
        goto cleanup;
    }
2468
    priv = vm->privateData;
P
Paolo Bonzini 已提交
2469

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

D
Daniel P. Berrange 已提交
2473
    if (!virDomainObjIsActive(vm)) {
2474 2475
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
2476
        goto endjob;
P
Paolo Bonzini 已提交
2477 2478
    }

O
Osier Yang 已提交
2479 2480
    priv->jobActive = QEMU_JOB_DUMP;

P
Paolo Bonzini 已提交
2481 2482
    /* Migrate will always stop the VM, so the resume condition is
       independent of whether the stop command is issued.  */
J
Jiri Denemark 已提交
2483
    resume = virDomainObjGetState(vm, NULL) == VIR_DOMAIN_RUNNING;
P
Paolo Bonzini 已提交
2484 2485

    /* Pause domain for non-live dump */
J
Jiri Denemark 已提交
2486 2487 2488
    if (!(flags & VIR_DUMP_LIVE) &&
        virDomainObjGetState(vm, NULL) == VIR_DOMAIN_RUNNING) {
        if (qemuProcessStopCPUs(driver, vm, VIR_DOMAIN_PAUSED_DUMP) < 0)
2489
            goto endjob;
P
Paolo Bonzini 已提交
2490
        paused = 1;
2491 2492 2493 2494 2495 2496

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

H
Hu Tao 已提交
2499
    ret = doCoreDump(driver, vm, path, getCompressionType(driver));
2500 2501 2502 2503
    if (ret < 0)
        goto endjob;

    paused = 1;
2504 2505

endjob:
2506
    if ((ret == 0) && (flags & VIR_DUMP_CRASH)) {
J
Jiri Denemark 已提交
2507
        qemuProcessStop(driver, vm, 0, VIR_DOMAIN_SHUTOFF_CRASHED);
2508
        qemuAuditDomainStop(vm, "crashed");
2509 2510 2511 2512 2513
        event = virDomainEventNewFromObj(vm,
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_CRASHED);
    }

P
Paolo Bonzini 已提交
2514 2515 2516
    /* 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.  */
2517
    else if (resume && paused && virDomainObjIsActive(vm)) {
J
Jiri Denemark 已提交
2518 2519
        if (qemuProcessStartCPUs(driver, vm, dom->conn,
                                 VIR_DOMAIN_RUNNING_UNPAUSED) < 0) {
2520
            if (virGetLastError() == NULL)
2521 2522
                qemuReportError(VIR_ERR_OPERATION_FAILED,
                                "%s", _("resuming after dump failed"));
P
Paolo Bonzini 已提交
2523 2524
        }
    }
2525

2526 2527
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
2528
    else if ((ret == 0) && (flags & VIR_DUMP_CRASH) && !vm->persistent) {
2529 2530 2531 2532
        virDomainRemoveInactive(&driver->domains,
                                vm);
        vm = NULL;
    }
2533 2534

cleanup:
P
Paolo Bonzini 已提交
2535 2536
    if (vm)
        virDomainObjUnlock(vm);
2537 2538
    if (event)
        qemuDomainEventQueue(driver, event);
2539
    qemuDriverUnlock(driver);
P
Paolo Bonzini 已提交
2540 2541 2542
    return ret;
}

2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596 2597 2598 2599 2600 2601 2602 2603 2604 2605 2606 2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630
static char *
qemuDomainScreenshot(virDomainPtr dom,
                     virStreamPtr st,
                     unsigned int screen,
                     unsigned int flags ATTRIBUTE_UNUSED)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    qemuDomainObjPrivatePtr priv;
    char *tmp = NULL;
    int tmp_fd = -1;
    char *ret = NULL;

    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 matching uuid '%s'"), uuidstr);
        goto cleanup;
    }

    priv = vm->privateData;

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

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

    /* Well, even if qemu allows multiple graphic cards, heads, whatever,
     * screenshot command does not */
    if (screen) {
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("currently is supported only taking "
                                "screenshots of screen ID 0"));
        goto endjob;
    }

    if (virAsprintf(&tmp, "%s/qemu.screendump.XXXXXX", driver->cacheDir) < 0) {
        virReportOOMError();
        goto endjob;
    }

    if ((tmp_fd = mkstemp(tmp)) == -1) {
        virReportSystemError(errno, _("mkstemp(\"%s\") failed"), tmp);
        goto endjob;
    }

    qemuDomainObjEnterMonitor(vm);
    if (qemuMonitorScreendump(priv->mon, tmp) < 0) {
        qemuDomainObjExitMonitor(vm);
        goto endjob;
    }
    qemuDomainObjExitMonitor(vm);

    if (VIR_CLOSE(tmp_fd) < 0) {
        virReportSystemError(errno, _("unable to close %s"), tmp);
        goto endjob;
    }

    if (virFDStreamOpenFile(st, tmp, 0, 0, O_RDONLY, true) < 0) {
        qemuReportError(VIR_ERR_OPERATION_FAILED, "%s",
                        _("unable to open stream"));
        goto endjob;
    }

    ret = strdup("image/x-portable-pixmap");

endjob:
    VIR_FORCE_CLOSE(tmp_fd);
    VIR_FREE(tmp);

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

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

H
Hu Tao 已提交
2631 2632 2633
static void processWatchdogEvent(void *data, void *opaque)
{
    int ret;
2634
    struct qemuDomainWatchdogEvent *wdEvent = data;
H
Hu Tao 已提交
2635 2636
    struct qemud_driver *driver = opaque;

W
Wen Congyang 已提交
2637 2638 2639
    qemuDriverLock(driver);
    virDomainObjLock(wdEvent->vm);

H
Hu Tao 已提交
2640 2641 2642 2643 2644
    switch (wdEvent->action) {
    case VIR_DOMAIN_WATCHDOG_ACTION_DUMP:
        {
            char *dumpfile;

E
Eric Blake 已提交
2645
            if (virAsprintf(&dumpfile, "%s/%s-%u",
H
Hu Tao 已提交
2646 2647
                            driver->autoDumpPath,
                            wdEvent->vm->def->name,
E
Eric Blake 已提交
2648 2649
                            (unsigned int)time(NULL)) < 0) {
                virReportOOMError();
W
Wen Congyang 已提交
2650
                goto unlock;
E
Eric Blake 已提交
2651
            }
H
Hu Tao 已提交
2652

W
Wen Congyang 已提交
2653 2654 2655 2656
            if (qemuDomainObjBeginJobWithDriver(driver, wdEvent->vm) < 0) {
                VIR_FREE(dumpfile);
                goto unlock;
            }
H
Hu Tao 已提交
2657 2658 2659 2660

            if (!virDomainObjIsActive(wdEvent->vm)) {
                qemuReportError(VIR_ERR_OPERATION_INVALID,
                                "%s", _("domain is not running"));
W
Wen Congyang 已提交
2661 2662
                VIR_FREE(dumpfile);
                goto endjob;
H
Hu Tao 已提交
2663 2664 2665 2666 2667 2668 2669 2670 2671 2672
            }

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

J
Jiri Denemark 已提交
2673 2674
            ret = qemuProcessStartCPUs(driver, wdEvent->vm, NULL,
                                       VIR_DOMAIN_RUNNING_UNPAUSED);
H
Hu Tao 已提交
2675 2676 2677 2678 2679 2680 2681 2682

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

            VIR_FREE(dumpfile);
        }
        break;
W
Wen Congyang 已提交
2683 2684
    default:
        goto unlock;
H
Hu Tao 已提交
2685 2686
    }

W
Wen Congyang 已提交
2687 2688 2689 2690 2691 2692 2693 2694 2695 2696
endjob:
    /* Safe to ignore value since ref count was incremented in
     * qemuProcessHandleWatchdog().
     */
    ignore_value(qemuDomainObjEndJob(wdEvent->vm));

unlock:
    if (virDomainObjUnref(wdEvent->vm) > 0)
        virDomainObjUnlock(wdEvent->vm);
    qemuDriverUnlock(driver);
H
Hu Tao 已提交
2697 2698
    VIR_FREE(wdEvent);
}
P
Paolo Bonzini 已提交
2699

2700 2701 2702
static int qemudDomainHotplugVcpus(virDomainObjPtr vm, unsigned int nvcpus)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
2703
    int i, rc = 1;
2704
    int ret = -1;
2705
    int oldvcpus = vm->def->vcpus;
E
Eric Blake 已提交
2706
    int vcpus = oldvcpus;
2707

2708 2709
    qemuDomainObjEnterMonitor(vm);

2710 2711 2712
    /* 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 已提交
2713 2714
    if (nvcpus > vcpus) {
        for (i = vcpus ; i < nvcpus ; i++) {
2715 2716 2717 2718 2719 2720 2721
            /* Online new CPU */
            rc = qemuMonitorSetCPU(priv->mon, i, 1);
            if (rc == 0)
                goto unsupported;
            if (rc < 0)
                goto cleanup;

E
Eric Blake 已提交
2722
            vcpus++;
2723 2724
        }
    } else {
E
Eric Blake 已提交
2725
        for (i = vcpus - 1 ; i >= nvcpus ; i--) {
2726 2727 2728 2729 2730 2731 2732
            /* Offline old CPU */
            rc = qemuMonitorSetCPU(priv->mon, i, 0);
            if (rc == 0)
                goto unsupported;
            if (rc < 0)
                goto cleanup;

E
Eric Blake 已提交
2733
            vcpus--;
2734 2735 2736 2737 2738 2739
        }
    }

    ret = 0;

cleanup:
2740
    qemuDomainObjExitMonitor(vm);
E
Eric Blake 已提交
2741
    vm->def->vcpus = vcpus;
2742
    qemuAuditVcpu(vm, oldvcpus, nvcpus, "update", rc == 1);
2743 2744 2745 2746 2747 2748 2749 2750 2751
    return ret;

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


2752 2753 2754 2755
static int
qemudDomainSetVcpusFlags(virDomainPtr dom, unsigned int nvcpus,
                         unsigned int flags)
{
2756 2757
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2758
    virDomainDefPtr persistentDef;
2759 2760
    const char * type;
    int max;
2761
    int ret = -1;
2762

2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778
    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);
2779 2780 2781
        return -1;
    }

2782
    qemuDriverLock(driver);
2783
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2784 2785
    qemuDriverUnlock(driver);

2786
    if (!vm) {
2787 2788
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2789 2790
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
2791
        goto cleanup;
2792 2793
    }

2794 2795 2796
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

2797
    if (!virDomainObjIsActive(vm) && (flags & VIR_DOMAIN_VCPU_LIVE)) {
2798 2799
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                         "%s", _("domain is not running"));
2800
        goto endjob;
2801 2802
    }

2803 2804 2805 2806 2807 2808
    if (!vm->persistent && (flags & VIR_DOMAIN_VCPU_CONFIG)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID, "%s",
                        _("cannot change persistent config of a transient domain"));
        goto endjob;
    }

2809 2810 2811 2812 2813 2814 2815 2816 2817 2818 2819 2820 2821
    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 已提交
2822
    if (!(flags & VIR_DOMAIN_VCPU_MAXIMUM) && vm->def->maxvcpus < max) {
2823 2824 2825
        max = vm->def->maxvcpus;
    }

2826 2827 2828 2829 2830 2831 2832
    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;
    }

2833 2834 2835
    if (!(persistentDef = virDomainObjGetPersistentDef(driver->caps, vm)))
        goto endjob;

2836 2837
    switch (flags) {
    case VIR_DOMAIN_VCPU_MAXIMUM | VIR_DOMAIN_VCPU_CONFIG:
2838 2839 2840
        persistentDef->maxvcpus = nvcpus;
        if (nvcpus < persistentDef->vcpus)
            persistentDef->vcpus = nvcpus;
2841 2842 2843 2844
        ret = 0;
        break;

    case VIR_DOMAIN_VCPU_CONFIG:
2845
        persistentDef->vcpus = nvcpus;
2846 2847 2848 2849 2850 2851 2852 2853 2854
        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);
2855 2856 2857
        if (ret == 0) {
            persistentDef->vcpus = nvcpus;
        }
2858 2859
        break;
    }
2860

2861 2862 2863 2864
    /* Save the persistent config to disk */
    if (flags & VIR_DOMAIN_VCPU_CONFIG)
        ret = virDomainSaveConfig(driver->configDir, persistentDef);

2865 2866 2867
endjob:
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
2868

2869
cleanup:
2870 2871
    if (vm)
        virDomainObjUnlock(vm);
2872
    return ret;
2873 2874
}

2875 2876 2877 2878 2879 2880
static int
qemudDomainSetVcpus(virDomainPtr dom, unsigned int nvcpus)
{
    return qemudDomainSetVcpusFlags(dom, nvcpus, VIR_DOMAIN_VCPU_LIVE);
}

2881 2882 2883 2884 2885 2886

static int
qemudDomainPinVcpu(virDomainPtr dom,
                   unsigned int vcpu,
                   unsigned char *cpumap,
                   int maplen) {
2887 2888
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2889
    int maxcpu, hostcpus;
2890
    virNodeInfo nodeinfo;
2891
    int ret = -1;
2892
    qemuDomainObjPrivatePtr priv;
2893

2894
    qemuDriverLock(driver);
2895
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2896 2897
    qemuDriverUnlock(driver);

2898 2899 2900
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2901 2902
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
2903 2904 2905
        goto cleanup;
    }

D
Daniel P. Berrange 已提交
2906
    if (!virDomainObjIsActive(vm)) {
2907 2908
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s",_("cannot pin vcpus on an inactive domain"));
2909
        goto cleanup;
2910 2911
    }

2912 2913 2914
    priv = vm->privateData;

    if (vcpu > (priv->nvcpupids-1)) {
2915 2916 2917
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("vcpu number out of range %d > %d"),
                        vcpu, priv->nvcpupids);
2918
        goto cleanup;
2919 2920
    }

2921
    if (nodeGetInfo(dom->conn, &nodeinfo) < 0)
2922
        goto cleanup;
2923

2924
    hostcpus = VIR_NODEINFO_MAXCPUS(nodeinfo);
2925
    maxcpu = maplen * 8;
2926 2927
    if (maxcpu > hostcpus)
        maxcpu = hostcpus;
2928

2929 2930
    if (priv->vcpupids != NULL) {
        if (virProcessInfoSetAffinity(priv->vcpupids[vcpu],
2931
                                      cpumap, maplen, maxcpu) < 0)
2932
            goto cleanup;
2933
    } else {
2934 2935
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        "%s", _("cpu affinity is not supported"));
2936
        goto cleanup;
2937
    }
2938 2939 2940 2941 2942 2943 2944

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

2945
    ret = 0;
2946

2947
cleanup:
2948 2949
    if (vm)
        virDomainObjUnlock(vm);
2950
    return ret;
2951 2952 2953 2954 2955 2956 2957 2958
}

static int
qemudDomainGetVcpus(virDomainPtr dom,
                    virVcpuInfoPtr info,
                    int maxinfo,
                    unsigned char *cpumaps,
                    int maplen) {
2959 2960
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
2961
    virNodeInfo nodeinfo;
2962
    int i, v, maxcpu, hostcpus;
2963
    int ret = -1;
2964
    qemuDomainObjPrivatePtr priv;
2965

2966
    qemuDriverLock(driver);
2967
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
2968 2969
    qemuDriverUnlock(driver);

2970 2971 2972
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
2973 2974
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
2975 2976 2977
        goto cleanup;
    }

D
Daniel P. Berrange 已提交
2978
    if (!virDomainObjIsActive(vm)) {
2979 2980 2981
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s",
                        _("cannot list vcpu pinning for an inactive domain"));
2982
        goto cleanup;
2983 2984
    }

2985 2986
    priv = vm->privateData;

2987
    if (nodeGetInfo(dom->conn, &nodeinfo) < 0)
2988
        goto cleanup;
2989

2990
    hostcpus = VIR_NODEINFO_MAXCPUS(nodeinfo);
2991
    maxcpu = maplen * 8;
2992 2993
    if (maxcpu > hostcpus)
        maxcpu = hostcpus;
2994 2995

    /* Clamp to actual number of vcpus */
2996 2997
    if (maxinfo > priv->nvcpupids)
        maxinfo = priv->nvcpupids;
2998

2999 3000 3001 3002 3003 3004
    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;
3005

3006
                if (priv->vcpupids != NULL &&
3007 3008 3009
                    qemudGetProcessInfo(&(info[i].cpuTime),
                                        &(info[i].cpu),
                                        vm->pid,
3010
                                        priv->vcpupids[i]) < 0) {
3011
                    virReportSystemError(errno, "%s",
3012 3013 3014
                                         _("cannot get vCPU placement & pCPU time"));
                    goto cleanup;
                }
3015
            }
3016 3017
        }

3018 3019
        if (cpumaps != NULL) {
            memset(cpumaps, 0, maplen * maxinfo);
3020
            if (priv->vcpupids != NULL) {
3021 3022 3023
                for (v = 0 ; v < maxinfo ; v++) {
                    unsigned char *cpumap = VIR_GET_CPUMAP(cpumaps, maplen, v);

3024
                    if (virProcessInfoGetAffinity(priv->vcpupids[v],
3025
                                                  cpumap, maplen, maxcpu) < 0)
3026
                        goto cleanup;
3027
                }
3028
            } else {
3029 3030
                qemuReportError(VIR_ERR_NO_SUPPORT,
                                "%s", _("cpu affinity is not available"));
3031
                goto cleanup;
3032 3033 3034
            }
        }
    }
3035
    ret = maxinfo;
3036

3037
cleanup:
3038 3039
    if (vm)
        virDomainObjUnlock(vm);
3040
    return ret;
3041 3042 3043
}


3044 3045 3046
static int
qemudDomainGetVcpusFlags(virDomainPtr dom, unsigned int flags)
{
3047 3048
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
3049
    virDomainDefPtr def;
3050
    int ret = -1;
3051

3052 3053 3054 3055 3056 3057 3058 3059
    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);
3060 3061 3062
        return -1;
    }

3063
    qemuDriverLock(driver);
3064
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
3065 3066
    qemuDriverUnlock(driver);

3067
    if (!vm) {
3068 3069
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
3070 3071
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
3072
        goto cleanup;
3073 3074
    }

3075 3076 3077 3078 3079 3080 3081 3082 3083
    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;
3084 3085
    }

3086
    ret = (flags & VIR_DOMAIN_VCPU_MAXIMUM) ? def->maxvcpus : def->vcpus;
3087

3088
cleanup:
3089 3090
    if (vm)
        virDomainObjUnlock(vm);
3091 3092 3093
    return ret;
}

3094 3095 3096 3097 3098 3099 3100
static int
qemudDomainGetMaxVcpus(virDomainPtr dom)
{
    return qemudDomainGetVcpusFlags(dom, (VIR_DOMAIN_VCPU_LIVE |
                                          VIR_DOMAIN_VCPU_MAXIMUM));
}

3101 3102 3103 3104 3105 3106 3107 3108 3109
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);

3110 3111
    memset(seclabel, 0, sizeof(*seclabel));

3112 3113 3114
    if (!vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
3115 3116
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
3117 3118 3119
        goto cleanup;
    }

3120
    if (!virDomainVirtTypeToString(vm->def->virtType)) {
3121 3122 3123
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("unknown virt type in domain definition '%d'"),
                        vm->def->virtType);
3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139 3140
        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 已提交
3141
    if (virDomainObjIsActive(vm)) {
3142 3143
        if (virSecurityManagerGetProcessLabel(driver->securityManager,
                                              vm, seclabel) < 0) {
3144 3145 3146
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            "%s", _("Failed to get security label"));
            goto cleanup;
3147 3148 3149 3150 3151 3152 3153 3154
        }
    }

    ret = 0;

cleanup:
    if (vm)
        virDomainObjUnlock(vm);
3155
    qemuDriverUnlock(driver);
3156 3157 3158
    return ret;
}

3159 3160
static int qemudNodeGetSecurityModel(virConnectPtr conn,
                                     virSecurityModelPtr secmodel)
3161 3162 3163
{
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
    char *p;
3164
    int ret = 0;
3165

3166
    qemuDriverLock(driver);
3167 3168 3169 3170 3171
    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)
3172
        goto cleanup;
3173

3174 3175
    p = driver->caps->host.secModel.model;
    if (strlen(p) >= VIR_SECURITY_MODEL_BUFLEN-1) {
3176 3177 3178
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("security model string exceeds max %d bytes"),
                        VIR_SECURITY_MODEL_BUFLEN-1);
3179 3180
        ret = -1;
        goto cleanup;
3181 3182 3183 3184 3185
    }
    strcpy(secmodel->model, p);

    p = driver->caps->host.secModel.doi;
    if (strlen(p) >= VIR_SECURITY_DOI_BUFLEN-1) {
3186 3187 3188
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("security DOI string exceeds max %d bytes"),
                        VIR_SECURITY_DOI_BUFLEN-1);
3189 3190
        ret = -1;
        goto cleanup;
3191 3192
    }
    strcpy(secmodel->doi, p);
3193 3194 3195 3196

cleanup:
    qemuDriverUnlock(driver);
    return ret;
3197 3198
}

3199 3200 3201 3202 3203
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 已提交
3204 3205
{
    int fd;
3206
    struct qemud_save_header header;
J
Jiri Denemark 已提交
3207 3208
    char *xml = NULL;
    virDomainDefPtr def = NULL;
3209

3210 3211 3212
    if ((fd = virFileOpenAs(path, O_RDONLY, 0, getuid(), getgid(), 0)) < 0) {
        if ((fd != -EACCES && fd != -EPERM) ||
            driver->user == getuid()) {
3213 3214
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            "%s", _("cannot read domain image"));
J
Jiri Denemark 已提交
3215
            goto error;
3216 3217 3218
        }

        /* Opening as root failed, but qemu runs as a different user
3219 3220 3221 3222 3223 3224
         * 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 已提交
3225
            goto error;
3226
        }
3227 3228 3229
    }

    if (saferead(fd, &header, sizeof(header)) != sizeof(header)) {
3230 3231
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to read qemu header"));
J
Jiri Denemark 已提交
3232
        goto error;
3233 3234 3235
    }

    if (memcmp(header.magic, QEMUD_SAVE_MAGIC, sizeof(header.magic)) != 0) {
3236 3237
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("image magic is incorrect"));
J
Jiri Denemark 已提交
3238
        goto error;
3239 3240 3241
    }

    if (header.version > QEMUD_SAVE_VERSION) {
3242 3243 3244 3245 3246
        /* convert endianess and try again */
        bswap_header(&header);
    }

    if (header.version > QEMUD_SAVE_VERSION) {
3247 3248 3249
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("image version is not supported (%d > %d)"),
                        header.version, QEMUD_SAVE_VERSION);
J
Jiri Denemark 已提交
3250
        goto error;
3251 3252
    }

3253 3254 3255
    if (header.xml_len <= 0) {
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        _("invalid XML length: %d"), header.xml_len);
J
Jiri Denemark 已提交
3256
        goto error;
3257 3258
    }

3259 3260
    if (VIR_ALLOC_N(xml, header.xml_len) < 0) {
        virReportOOMError();
J
Jiri Denemark 已提交
3261
        goto error;
3262 3263 3264
    }

    if (saferead(fd, xml, header.xml_len) != header.xml_len) {
3265 3266
        qemuReportError(VIR_ERR_OPERATION_FAILED,
                        "%s", _("failed to read XML"));
J
Jiri Denemark 已提交
3267
        goto error;
3268 3269 3270
    }

    /* Create a domain from this XML */
3271
    if (!(def = virDomainDefParseString(driver->caps, xml,
3272
                                        VIR_DOMAIN_XML_INACTIVE)))
J
Jiri Denemark 已提交
3273
        goto error;
3274

J
Jiri Denemark 已提交
3275
    VIR_FREE(xml);
3276

J
Jiri Denemark 已提交
3277 3278
    *ret_def = def;
    *ret_header = header;
3279

J
Jiri Denemark 已提交
3280
    return fd;
3281

J
Jiri Denemark 已提交
3282 3283 3284
error:
    virDomainDefFree(def);
    VIR_FREE(xml);
3285
    VIR_FORCE_CLOSE(fd);
J
Jiri Denemark 已提交
3286 3287 3288 3289

    return -1;
}

3290 3291 3292 3293 3294 3295 3296
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 已提交
3297 3298 3299 3300 3301 3302 3303 3304
{
    int ret = -1;
    virDomainEventPtr event;
    int intermediatefd = -1;
    pid_t intermediate_pid = -1;
    int childstat;

    if (header->version == 2) {
3305
        const char *intermediate_argv[3] = { NULL, "-dc", NULL };
J
Jiri Denemark 已提交
3306
        const char *prog = qemudSaveCompressionTypeToString(header->compressed);
3307
        if (prog == NULL) {
3308 3309
            qemuReportError(VIR_ERR_OPERATION_FAILED,
                            _("Invalid compressed save format %d"),
J
Jiri Denemark 已提交
3310 3311
                            header->compressed);
            goto out;
3312
        }
3313

J
Jiri Denemark 已提交
3314
        if (header->compressed != QEMUD_SAVE_FORMAT_RAW) {
3315
            intermediate_argv[0] = prog;
3316 3317
            intermediatefd = *fd;
            *fd = -1;
3318
            if (virExec(intermediate_argv, NULL, NULL,
3319
                        &intermediate_pid, intermediatefd, fd, NULL, 0) < 0) {
3320 3321 3322
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                _("Failed to start decompression binary %s"),
                                intermediate_argv[0]);
3323
                *fd = intermediatefd;
J
Jiri Denemark 已提交
3324
                goto out;
3325 3326 3327
            }
        }
    }
J
Jiri Denemark 已提交
3328

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

3333
    if (intermediate_pid != -1) {
3334
        if (ret < 0) {
3335 3336 3337
            /* if there was an error setting up qemu, the intermediate
             * process will wait forever to write to stdout, so we
             * must manually kill it.
3338 3339
             */
            VIR_FORCE_CLOSE(intermediatefd);
3340
            VIR_FORCE_CLOSE(*fd);
3341 3342 3343
            kill(intermediate_pid, SIGTERM);
        }

3344 3345
        /* Wait for intermediate process to exit */
        while (waitpid(intermediate_pid, &childstat, 0) == -1 &&
J
Jiri Denemark 已提交
3346 3347 3348
               errno == EINTR) {
            /* empty */
        }
3349
    }
3350
    VIR_FORCE_CLOSE(intermediatefd);
J
Jiri Denemark 已提交
3351

3352 3353 3354
    if (VIR_CLOSE(*fd) < 0) {
        virReportSystemError(errno, _("cannot close file: %s"), path);
        ret = -1;
3355
    }
J
Jiri Denemark 已提交
3356

3357
    if (ret < 0) {
3358
        qemuAuditDomainStart(vm, "restored", false);
J
Jiri Denemark 已提交
3359
        goto out;
3360
    }
3361

3362 3363 3364
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_STARTED,
                                     VIR_DOMAIN_EVENT_STARTED_RESTORED);
3365
    qemuAuditDomainStart(vm, "restored", true);
J
Jiri Denemark 已提交
3366 3367 3368
    if (event)
        qemuDomainEventQueue(driver, event);

3369

3370
    /* If it was running before, resume it now. */
J
Jiri Denemark 已提交
3371
    if (header->was_running) {
J
Jiri Denemark 已提交
3372 3373
        if (qemuProcessStartCPUs(driver, vm, conn,
                                 VIR_DOMAIN_RUNNING_RESTORED) < 0) {
3374
            if (virGetLastError() == NULL)
3375 3376
                qemuReportError(VIR_ERR_OPERATION_FAILED,
                                "%s", _("failed to resume domain"));
J
Jiri Denemark 已提交
3377
            goto out;
3378
        }
3379 3380
        if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0) {
            VIR_WARN("Failed to save status on vm %s", vm->def->name);
J
Jiri Denemark 已提交
3381
            goto out;
3382
        }
3383
    }
J
Jiri Denemark 已提交
3384

3385
    ret = 0;
3386

J
Jiri Denemark 已提交
3387
out:
3388 3389
    if (virSecurityManagerRestoreSavedStateLabel(driver->securityManager,
                                                 vm, path) < 0)
3390 3391
        VIR_WARN("failed to restore save state label on %s", path);

J
Jiri Denemark 已提交
3392 3393 3394
    return ret;
}

3395 3396 3397 3398
static int
qemuDomainRestore(virConnectPtr conn,
                  const char *path)
{
J
Jiri Denemark 已提交
3399 3400 3401 3402 3403 3404 3405 3406 3407
    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);

3408
    fd = qemuDomainSaveImageOpen(driver, path, &def, &header);
J
Jiri Denemark 已提交
3409 3410 3411 3412 3413 3414 3415 3416 3417
    if (fd < 0)
        goto cleanup;

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

    if (!(vm = virDomainAssignDef(driver->caps,
                                  &driver->domains,
                                  def, true))) {
3418
        /* virDomainAssignDef already set the error */
J
Jiri Denemark 已提交
3419 3420 3421 3422 3423 3424 3425
        goto cleanup;
    }
    def = NULL;

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

3426
    ret = qemuDomainSaveImageStartVM(conn, driver, vm, &fd, &header, path);
J
Jiri Denemark 已提交
3427 3428

    if (qemuDomainObjEndJob(vm) == 0)
3429
        vm = NULL;
J
Jiri Denemark 已提交
3430 3431 3432 3433
    else if (ret < 0 && !vm->persistent) {
        virDomainRemoveInactive(&driver->domains, vm);
        vm = NULL;
    }
3434

3435 3436
cleanup:
    virDomainDefFree(def);
3437
    VIR_FORCE_CLOSE(fd);
3438 3439 3440
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
3441
    return ret;
D
Daniel P. Berrange 已提交
3442 3443
}

3444 3445 3446 3447 3448
static int
qemuDomainObjRestore(virConnectPtr conn,
                     struct qemud_driver *driver,
                     virDomainObjPtr vm,
                     const char *path)
J
Jiri Denemark 已提交
3449 3450 3451 3452 3453 3454
{
    virDomainDefPtr def = NULL;
    int fd = -1;
    int ret = -1;
    struct qemud_save_header header;

3455
    fd = qemuDomainSaveImageOpen(driver, path, &def, &header);
J
Jiri Denemark 已提交
3456 3457 3458 3459 3460 3461 3462 3463 3464 3465 3466 3467 3468 3469 3470 3471 3472 3473 3474 3475
    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;

3476
    ret = qemuDomainSaveImageStartVM(conn, driver, vm, &fd, &header, path);
J
Jiri Denemark 已提交
3477 3478 3479

cleanup:
    virDomainDefFree(def);
3480
    VIR_FORCE_CLOSE(fd);
J
Jiri Denemark 已提交
3481 3482 3483
    return ret;
}

D
Daniel P. Berrange 已提交
3484

3485 3486
static char *qemuDomainGetXMLDesc(virDomainPtr dom,
                                  int flags) {
3487 3488 3489
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    char *ret = NULL;
3490 3491
    unsigned long balloon;
    int err;
3492

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

D
Daniel P. Berrange 已提交
3496
    if (!vm) {
3497 3498
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
3499 3500
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
3501
        goto cleanup;
D
Daniel P. Berrange 已提交
3502 3503
    }

3504 3505 3506 3507
    /* 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))) {
3508
        qemuDomainObjPrivatePtr priv = vm->privateData;
3509 3510 3511
        /* Don't delay if someone's using the monitor, just use
         * existing most recent data instead */
        if (!priv->jobActive) {
3512
            if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
3513 3514
                goto cleanup;

3515
            qemuDomainObjEnterMonitorWithDriver(driver, vm);
3516
            err = qemuMonitorGetBalloonInfo(priv->mon, &balloon);
3517
            qemuDomainObjExitMonitorWithDriver(driver, vm);
3518 3519 3520 3521
            if (qemuDomainObjEndJob(vm) == 0) {
                vm = NULL;
                goto cleanup;
            }
3522 3523 3524
            if (err < 0)
                goto cleanup;
            if (err > 0)
3525
                vm->def->mem.cur_balloon = balloon;
3526 3527
            /* err == 0 indicates no balloon support, so ignore it */
        }
3528
    }
3529

3530
    ret = qemuDomainFormatXML(driver, vm, flags);
3531 3532

cleanup:
3533 3534
    if (vm)
        virDomainObjUnlock(vm);
3535
    qemuDriverUnlock(driver);
3536
    return ret;
D
Daniel P. Berrange 已提交
3537 3538 3539
}


3540 3541 3542 3543
static char *qemuDomainXMLFromNative(virConnectPtr conn,
                                     const char *format,
                                     const char *config,
                                     unsigned int flags ATTRIBUTE_UNUSED) {
3544
    struct qemud_driver *driver = conn->privateData;
3545 3546 3547 3548
    virDomainDefPtr def = NULL;
    char *xml = NULL;

    if (STRNEQ(format, QEMU_CONFIG_FORMAT_ARGV)) {
3549 3550
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("unsupported config type %s"), format);
3551 3552 3553
        goto cleanup;
    }

3554
    qemuDriverLock(driver);
3555
    def = qemuParseCommandLineString(driver->caps, config);
3556
    qemuDriverUnlock(driver);
3557 3558 3559
    if (!def)
        goto cleanup;

3560
    xml = virDomainDefFormat(def, VIR_DOMAIN_XML_INACTIVE);
3561 3562 3563 3564 3565 3566

cleanup:
    virDomainDefFree(def);
    return xml;
}

3567 3568 3569 3570 3571 3572
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;
3573
    virDomainChrSourceDef monConfig;
3574
    virBitmapPtr qemuCaps = NULL;
E
Eric Blake 已提交
3575
    virCommandPtr cmd = NULL;
3576 3577 3578
    char *ret = NULL;
    int i;

3579 3580
    qemuDriverLock(driver);

3581
    if (STRNEQ(format, QEMU_CONFIG_FORMAT_ARGV)) {
3582 3583
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("unsupported config type %s"), format);
3584 3585 3586
        goto cleanup;
    }

3587
    def = virDomainDefParseString(driver->caps, xmlData, 0);
3588 3589 3590
    if (!def)
        goto cleanup;

3591 3592
    /* Since we're just exporting args, we can't do bridge/network/direct
     * setups, since libvirt will normally create TAP/macvtap devices
3593 3594 3595 3596 3597
     * 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];
3598
        int bootIndex = net->bootIndex;
3599 3600
        if (net->type == VIR_DOMAIN_NET_TYPE_NETWORK ||
            net->type == VIR_DOMAIN_NET_TYPE_DIRECT) {
3601 3602 3603 3604 3605 3606 3607 3608 3609 3610 3611 3612 3613 3614 3615 3616 3617 3618 3619 3620
            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;
        }
3621
        net->bootIndex = bootIndex;
3622 3623 3624 3625
    }
    for (i = 0 ; i < def->ngraphics ; i++) {
        if (def->graphics[i]->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
            def->graphics[i]->data.vnc.autoport)
3626
            def->graphics[i]->data.vnc.port = QEMU_VNC_PORT_MIN;
3627 3628
    }

3629
    if (qemuCapsExtractVersionInfo(def->emulator, def->os.arch,
3630
                                   NULL,
3631
                                   &qemuCaps) < 0)
3632 3633
        goto cleanup;

3634
    if (qemuProcessPrepareMonitorChr(driver, &monConfig, def->name) < 0)
3635
        goto cleanup;
3636

3637
    if (!(cmd = qemuBuildCommandLine(conn, driver, def,
3638
                                     &monConfig, false, qemuCaps,
3639
                                     NULL, -1, NULL, VIR_VM_OP_NO_OP)))
3640 3641
        goto cleanup;

E
Eric Blake 已提交
3642
    ret = virCommandToString(cmd);
3643 3644

cleanup:
3645
    qemuDriverUnlock(driver);
3646

3647
    qemuCapsFree(qemuCaps);
E
Eric Blake 已提交
3648
    virCommandFree(cmd);
3649 3650 3651 3652 3653
    virDomainDefFree(def);
    return ret;
}


3654
static int qemudListDefinedDomains(virConnectPtr conn,
3655
                            char **const names, int nnames) {
3656
    struct qemud_driver *driver = conn->privateData;
3657
    int n;
3658

3659
    qemuDriverLock(driver);
3660
    n = virDomainObjListGetInactiveNames(&driver->domains, names, nnames);
3661
    qemuDriverUnlock(driver);
3662
    return n;
D
Daniel P. Berrange 已提交
3663 3664
}

3665
static int qemudNumDefinedDomains(virConnectPtr conn) {
3666
    struct qemud_driver *driver = conn->privateData;
3667
    int n;
3668

3669
    qemuDriverLock(driver);
3670
    n = virDomainObjListNumOfDomains(&driver->domains, 0);
3671
    qemuDriverUnlock(driver);
3672

3673
    return n;
D
Daniel P. Berrange 已提交
3674 3675 3676
}


J
Jiri Denemark 已提交
3677 3678
static int qemudDomainObjStart(virConnectPtr conn,
                               struct qemud_driver *driver,
3679 3680
                               virDomainObjPtr vm,
                               bool start_paused)
J
Jiri Denemark 已提交
3681 3682 3683 3684 3685 3686
{
    int ret = -1;
    char *managed_save;

    /*
     * If there is a managed saved state restore it instead of starting
3687
     * from scratch. The old state is removed once the restoring succeeded.
J
Jiri Denemark 已提交
3688 3689
     */
    managed_save = qemuDomainManagedSavePath(driver, vm);
3690 3691 3692 3693

    if (!managed_save)
        goto cleanup;

E
Eric Blake 已提交
3694
    if (virFileExists(managed_save)) {
3695
        ret = qemuDomainObjRestore(conn, driver, vm, managed_save);
J
Jiri Denemark 已提交
3696

3697
        if ((ret == 0) && (unlink(managed_save) < 0))
J
Jiri Denemark 已提交
3698 3699
            VIR_WARN("Failed to remove the managed state %s", managed_save);

3700
        goto cleanup;
J
Jiri Denemark 已提交
3701 3702
    }

3703 3704
    ret = qemuProcessStart(conn, driver, vm, NULL, start_paused, -1, NULL,
                           VIR_VM_OP_CREATE);
3705
    qemuAuditDomainStart(vm, "booted", ret >= 0);
3706
    if (ret >= 0) {
J
Jiri Denemark 已提交
3707 3708 3709 3710 3711 3712 3713 3714 3715 3716 3717 3718 3719
        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;
}

3720 3721 3722
static int
qemudDomainStartWithFlags(virDomainPtr dom, unsigned int flags)
{
3723 3724 3725
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
3726

3727
    virCheckFlags(VIR_DOMAIN_START_PAUSED, -1);
3728

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

3732
    if (!vm) {
3733 3734
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
3735 3736
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
3737
        goto cleanup;
3738 3739
    }

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

    if (virDomainObjIsActive(vm)) {
3744 3745
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is already running"));
3746 3747 3748
        goto endjob;
    }

3749 3750
    ret = qemudDomainObjStart(dom->conn, driver, vm,
                              (flags & VIR_DOMAIN_START_PAUSED) != 0);
3751

3752
endjob:
3753 3754
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
3755

3756
cleanup:
3757 3758
    if (vm)
        virDomainObjUnlock(vm);
3759
    qemuDriverUnlock(driver);
3760
    return ret;
D
Daniel P. Berrange 已提交
3761 3762
}

3763 3764 3765 3766 3767 3768
static int
qemudDomainStart(virDomainPtr dom)
{
    return qemudDomainStartWithFlags(dom, 0);
}

3769 3770 3771 3772 3773 3774 3775 3776 3777 3778 3779 3780 3781 3782 3783
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;

3784
        if (STRNEQ(def->os.machine, machine->name))
3785 3786 3787
            continue;

        if (!(*canonical = strdup(machine->canonical))) {
3788
            virReportOOMError();
3789 3790 3791 3792 3793 3794 3795 3796 3797
            return -1;
        }

        break;
    }

    return 0;
}

3798 3799 3800 3801 3802 3803
static int
qemudCanonicalizeMachineDirect(virDomainDefPtr def, char **canonical)
{
    virCapsGuestMachinePtr *machines = NULL;
    int i, nmachines = 0;

3804
    if (qemuCapsProbeMachineTypes(def->emulator, &machines, &nmachines) < 0)
3805 3806 3807 3808 3809 3810
        return -1;

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

3811
        if (STRNEQ(def->os.machine, machines[i]->name))
3812 3813 3814 3815 3816 3817 3818 3819 3820 3821 3822 3823
            continue;

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

    virCapabilitiesFreeMachines(machines, nmachines);

    return 0;
}

3824 3825
int
qemudCanonicalizeMachine(struct qemud_driver *driver, virDomainDefPtr def)
3826 3827 3828 3829 3830 3831
{
    char *canonical = NULL;
    int i;

    for (i = 0; i < driver->caps->nguests; i++) {
        virCapsGuestPtr guest = driver->caps->guests[i];
3832
        virCapsGuestDomainInfoPtr info;
3833 3834 3835
        int j;

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

3838 3839 3840 3841 3842 3843 3844 3845 3846
            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;
3847 3848
        }

3849 3850 3851 3852
        info = &guest->arch.defaultInfo;

        if (info->emulator && STREQ(info->emulator, def->emulator)) {
            if (qemudCanonicalizeMachineFromInfo(def, info, &canonical) < 0)
3853 3854 3855 3856
                return -1;
            goto out;
        }
    }
3857 3858 3859 3860

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

3861 3862 3863 3864 3865 3866 3867
out:
    if (canonical) {
        VIR_FREE(def->os.machine);
        def->os.machine = canonical;
    }
    return 0;
}
D
Daniel P. Berrange 已提交
3868

3869
static virDomainPtr qemudDomainDefine(virConnectPtr conn, const char *xml) {
3870
    struct qemud_driver *driver = conn->privateData;
3871
    virDomainDefPtr def;
3872
    virDomainObjPtr vm = NULL;
3873
    virDomainPtr dom = NULL;
3874
    virDomainEventPtr event = NULL;
3875
    int dupVM;
3876

3877
    qemuDriverLock(driver);
3878
    if (!(def = virDomainDefParseString(driver->caps, xml,
3879
                                        VIR_DOMAIN_XML_INACTIVE)))
3880
        goto cleanup;
3881

3882
    if (virSecurityManagerVerify(driver->securityManager, def) < 0)
3883 3884
        goto cleanup;

3885 3886
    if ((dupVM = virDomainObjIsDuplicate(&driver->domains, def, 0)) < 0)
        goto cleanup;
3887

3888
    if (qemudCanonicalizeMachine(driver, def) < 0)
3889 3890
        goto cleanup;

3891
    if (qemuDomainAssignPCIAddresses(def) < 0)
3892 3893
        goto cleanup;

3894
    if (!(vm = virDomainAssignDef(driver->caps,
3895
                                  &driver->domains,
3896
                                  def, false))) {
3897
        goto cleanup;
3898
    }
3899
    def = NULL;
3900
    vm->persistent = 1;
3901

3902
    if (virDomainSaveConfig(driver->configDir,
3903
                            vm->newDef ? vm->newDef : vm->def) < 0) {
3904
        VIR_INFO("Defining domain '%s'", vm->def->name);
3905 3906
        virDomainRemoveInactive(&driver->domains,
                                vm);
3907
        vm = NULL;
3908
        goto cleanup;
3909 3910
    }

3911 3912
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_DEFINED,
3913
                                     !dupVM ?
3914 3915
                                     VIR_DOMAIN_EVENT_DEFINED_ADDED :
                                     VIR_DOMAIN_EVENT_DEFINED_UPDATED);
3916

3917
    VIR_INFO("Creating domain '%s'", vm->def->name);
3918
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
3919
    if (dom) dom->id = vm->def->id;
3920 3921

cleanup:
3922
    virDomainDefFree(def);
3923 3924
    if (vm)
        virDomainObjUnlock(vm);
3925 3926
    if (event)
        qemuDomainEventQueue(driver, event);
3927
    qemuDriverUnlock(driver);
3928
    return dom;
D
Daniel P. Berrange 已提交
3929 3930
}

3931
static int qemudDomainUndefine(virDomainPtr dom) {
3932 3933
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
3934
    virDomainEventPtr event = NULL;
3935
    int ret = -1;
D
Daniel P. Berrange 已提交
3936

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

D
Daniel P. Berrange 已提交
3940
    if (!vm) {
3941 3942
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
3943 3944
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
3945
        goto cleanup;
D
Daniel P. Berrange 已提交
3946 3947
    }

D
Daniel P. Berrange 已提交
3948
    if (virDomainObjIsActive(vm)) {
3949 3950
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cannot delete active domain"));
3951
        goto cleanup;
D
Daniel P. Berrange 已提交
3952 3953
    }

3954
    if (!vm->persistent) {
3955
        qemuReportError(VIR_ERR_OPERATION_INVALID,
3956
                        "%s", _("cannot undefine transient domain"));
3957
        goto cleanup;
3958 3959
    }

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

3963 3964 3965
    event = virDomainEventNewFromObj(vm,
                                     VIR_DOMAIN_EVENT_UNDEFINED,
                                     VIR_DOMAIN_EVENT_UNDEFINED_REMOVED);
3966

3967
    VIR_INFO("Undefining domain '%s'", vm->def->name);
3968 3969
    virDomainRemoveInactive(&driver->domains,
                            vm);
3970
    vm = NULL;
3971
    ret = 0;
D
Daniel P. Berrange 已提交
3972

3973
cleanup:
3974 3975
    if (vm)
        virDomainObjUnlock(vm);
3976 3977
    if (event)
        qemuDomainEventQueue(driver, event);
3978
    qemuDriverUnlock(driver);
3979
    return ret;
D
Daniel P. Berrange 已提交
3980 3981
}

3982 3983 3984
static int
qemuDomainAttachDeviceDiskLive(struct qemud_driver *driver,
                               virDomainObjPtr vm,
3985
                               virDomainDeviceDefPtr dev)
3986 3987 3988 3989
{
    virDomainDiskDefPtr disk = dev->data.disk;
    virCgroupPtr cgroup = NULL;
    int ret = -1;
3990

3991 3992 3993 3994 3995 3996 3997 3998 3999 4000 4001 4002 4003 4004 4005 4006 4007 4008 4009 4010
    if (disk->driverName != NULL && !STREQ(disk->driverName, "qemu")) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("unsupported driver name '%s' for disk '%s'"),
                        disk->driverName, disk->src);
        goto end;
    }

    if (qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_DEVICES)) {
        if (virCgroupForDomain(driver->cgroup, vm->def->name, &cgroup, 0)) {
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("Unable to find cgroup for %s"),
                            vm->def->name);
            goto end;
        }
        if (qemuSetupDiskCgroup(driver, vm, cgroup, disk) < 0)
            goto end;
    }
    switch (disk->device)  {
    case VIR_DOMAIN_DISK_DEVICE_CDROM:
    case VIR_DOMAIN_DISK_DEVICE_FLOPPY:
4011
        ret = qemuDomainChangeEjectableMedia(driver, vm, disk, false);
4012 4013 4014 4015
        break;
    case VIR_DOMAIN_DISK_DEVICE_DISK:
        if (disk->bus == VIR_DOMAIN_DISK_BUS_USB)
            ret = qemuDomainAttachUsbMassstorageDevice(driver, vm,
4016
                                                       disk);
4017
        else if (disk->bus == VIR_DOMAIN_DISK_BUS_VIRTIO)
4018
            ret = qemuDomainAttachPciDiskDevice(driver, vm, disk);
4019
        else if (disk->bus == VIR_DOMAIN_DISK_BUS_SCSI)
4020
            ret = qemuDomainAttachSCSIDisk(driver, vm, disk);
4021 4022 4023 4024 4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039 4040 4041 4042 4043 4044 4045 4046
        else
            qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                            _("disk bus '%s' cannot be hotplugged."),
                            virDomainDiskBusTypeToString(disk->bus));
        break;
    default:
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                        _("disk device type '%s' cannot be hotplugged"),
                        virDomainDiskDeviceTypeToString(disk->device));
        break;
    }

    if (ret != 0 && cgroup) {
        if (qemuTeardownDiskCgroup(driver, vm, cgroup, disk) < 0)
            VIR_WARN("Failed to teardown cgroup for disk path %s",
                     NULLSTR(disk->src));
    }
end:
    if (cgroup)
        virCgroupFree(&cgroup);
    return ret;
}

static int
qemuDomainAttachDeviceControllerLive(struct qemud_driver *driver,
                                     virDomainObjPtr vm,
4047
                                     virDomainDeviceDefPtr dev)
4048 4049 4050 4051 4052 4053
{
    virDomainControllerDefPtr cont = dev->data.controller;
    int ret = -1;

    switch (cont->type) {
    case VIR_DOMAIN_CONTROLLER_TYPE_SCSI:
4054
        ret = qemuDomainAttachPciControllerDevice(driver, vm, cont);
4055 4056 4057 4058 4059 4060 4061 4062 4063 4064 4065 4066 4067
        break;
    default:
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                        _("disk controller bus '%s' cannot be hotplugged."),
                        virDomainControllerTypeToString(cont->type));
        break;
    }
    return ret;
}

static int
qemuDomainAttachDeviceLive(virDomainObjPtr vm,
                           virDomainDeviceDefPtr dev,
4068
                           virDomainPtr dom)
4069 4070 4071 4072 4073 4074
{
    struct qemud_driver *driver = dom->conn->privateData;
    int ret = -1;

    switch (dev->type) {
    case VIR_DOMAIN_DEVICE_DISK:
4075
        qemuDomainObjCheckDiskTaint(driver, vm, dev->data.disk, -1);
4076
        ret = qemuDomainAttachDeviceDiskLive(driver, vm, dev);
4077 4078 4079 4080 4081
        if (!ret)
            dev->data.disk = NULL;
        break;

    case VIR_DOMAIN_DEVICE_CONTROLLER:
4082
        ret = qemuDomainAttachDeviceControllerLive(driver, vm, dev);
4083 4084 4085 4086
        if (!ret)
            dev->data.controller = NULL;
        break;

4087 4088 4089 4090 4091 4092 4093
    case VIR_DOMAIN_DEVICE_LEASE:
        ret = qemuDomainAttachLease(driver, vm,
                                    dev->data.lease);
        if (ret == 0)
            dev->data.lease = NULL;
        break;

4094
    case VIR_DOMAIN_DEVICE_NET:
4095
        qemuDomainObjCheckNetTaint(driver, vm, dev->data.net, -1);
4096
        ret = qemuDomainAttachNetDevice(dom->conn, driver, vm,
4097
                                        dev->data.net);
4098 4099 4100 4101 4102 4103
        if (!ret)
            dev->data.net = NULL;
        break;

    case VIR_DOMAIN_DEVICE_HOSTDEV:
        ret = qemuDomainAttachHostDevice(driver, vm,
4104
                                         dev->data.hostdev);
4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119 4120 4121
        if (!ret)
            dev->data.hostdev = NULL;
        break;

    default:
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                        _("device type '%s' cannot be attached"),
                        virDomainDeviceTypeToString(dev->type));
        break;
    }

    return ret;
}

static int
qemuDomainDetachDeviceDiskLive(struct qemud_driver *driver,
                               virDomainObjPtr vm,
4122
                               virDomainDeviceDefPtr dev)
4123 4124 4125 4126 4127 4128 4129
{
    virDomainDiskDefPtr disk = dev->data.disk;
    int ret = -1;

    switch (disk->device) {
    case VIR_DOMAIN_DISK_DEVICE_DISK:
        if (disk->bus == VIR_DOMAIN_DISK_BUS_VIRTIO)
4130
            ret = qemuDomainDetachPciDiskDevice(driver, vm, dev);
4131
        else if (disk->bus == VIR_DOMAIN_DISK_BUS_SCSI)
4132
            ret =  qemuDomainDetachDiskDevice(driver, vm, dev);
4133
        else if (dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_USB)
4134
            ret = qemuDomainDetachDiskDevice(driver, vm, dev);
4135 4136 4137 4138 4139 4140 4141 4142 4143 4144 4145 4146 4147 4148 4149 4150
        else
            qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                            _("This type of disk cannot be hot unplugged"));
        break;
    default:
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                        _("device type '%s' cannot be detached"),
                        virDomainDeviceTypeToString(dev->type));
        break;
    }
    return ret;
}

static int
qemuDomainDetachDeviceControllerLive(struct qemud_driver *driver,
                                     virDomainObjPtr vm,
4151
                                     virDomainDeviceDefPtr dev)
4152 4153 4154 4155 4156 4157
{
    virDomainControllerDefPtr cont = dev->data.controller;
    int ret = -1;

    switch (cont->type) {
    case VIR_DOMAIN_CONTROLLER_TYPE_SCSI:
4158
        ret = qemuDomainDetachPciControllerDevice(driver, vm, dev);
4159 4160 4161 4162 4163 4164 4165 4166 4167 4168 4169 4170
        break;
    default :
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                        _("disk controller bus '%s' cannot be hotunplugged."),
                        virDomainControllerTypeToString(cont->type));
    }
    return ret;
}

static int
qemuDomainDetachDeviceLive(virDomainObjPtr vm,
                           virDomainDeviceDefPtr dev,
4171
                           virDomainPtr dom)
4172 4173 4174 4175 4176 4177
{
    struct qemud_driver *driver = dom->conn->privateData;
    int ret = -1;

    switch (dev->type) {
    case VIR_DOMAIN_DEVICE_DISK:
4178
        ret = qemuDomainDetachDeviceDiskLive(driver, vm, dev);
4179 4180
        break;
    case VIR_DOMAIN_DEVICE_CONTROLLER:
4181
        ret = qemuDomainDetachDeviceControllerLive(driver, vm, dev);
4182
        break;
4183 4184 4185
    case VIR_DOMAIN_DEVICE_LEASE:
        ret = qemuDomainDetachLease(driver, vm, dev->data.lease);
        break;
4186
    case VIR_DOMAIN_DEVICE_NET:
4187
        ret = qemuDomainDetachNetDevice(driver, vm, dev);
4188 4189
        break;
    case VIR_DOMAIN_DEVICE_HOSTDEV:
4190
        ret = qemuDomainDetachHostDevice(driver, vm, dev);
4191 4192 4193 4194 4195 4196 4197 4198 4199 4200
        break;
    default:
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                        "%s", _("This type of device cannot be hot unplugged"));
        break;
    }

    return ret;
}

4201 4202 4203 4204 4205 4206 4207 4208
static int
qemuDomainChangeDiskMediaLive(virDomainObjPtr vm,
                              virDomainDeviceDefPtr dev,
                              struct qemud_driver *driver,
                              bool force)
{
    virDomainDiskDefPtr disk = dev->data.disk;
    virCgroupPtr cgroup = NULL;
4209
    int ret = -1;
4210 4211 4212

    if (qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_DEVICES)) {
        if (virCgroupForDomain(driver->cgroup,
4213
                               vm->def->name, &cgroup, 0) != 0) {
4214 4215 4216 4217 4218 4219 4220 4221 4222 4223 4224 4225
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("Unable to find cgroup for %s"),
                            vm->def->name);
            goto end;
        }
        if (qemuSetupDiskCgroup(driver, vm, cgroup, disk) < 0)
            goto end;
    }

    switch (disk->device) {
    case VIR_DOMAIN_DISK_DEVICE_CDROM:
    case VIR_DOMAIN_DISK_DEVICE_FLOPPY:
4226
        ret = qemuDomainChangeEjectableMedia(driver, vm, disk, force);
4227 4228 4229 4230 4231 4232 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
        if (ret == 0)
            dev->data.disk = NULL;
        break;
    default:
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                        _("disk bus '%s' cannot be updated."),
                        virDomainDiskBusTypeToString(disk->bus));
        break;
    }

    if (ret != 0 && cgroup) {
        if (qemuTeardownDiskCgroup(driver, vm, cgroup, disk) < 0)
             VIR_WARN("Failed to teardown cgroup for disk path %s",
                      NULLSTR(disk->src));
    }
end:
    if (cgroup)
        virCgroupFree(&cgroup);
    return ret;
}

static int
qemuDomainUpdateDeviceLive(virDomainObjPtr vm,
                           virDomainDeviceDefPtr dev,
                           virDomainPtr dom,
                           bool force)
{
    struct qemud_driver *driver = dom->conn->privateData;
    int ret = -1;

    switch (dev->type) {
    case VIR_DOMAIN_DEVICE_DISK:
4259
        ret = qemuDomainChangeDiskMediaLive(vm, dev, driver, force);
4260 4261 4262 4263 4264 4265 4266 4267 4268 4269 4270 4271 4272 4273
        break;
    case VIR_DOMAIN_DEVICE_GRAPHICS:
        ret = qemuDomainChangeGraphics(driver, vm, dev->data.graphics);
        break;
    default:
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                        _("device type '%s' cannot be updated"),
                        virDomainDeviceTypeToString(dev->type));
        break;
    }

    return ret;
}

4274
static int
4275
qemuDomainAttachDeviceConfig(virDomainDefPtr vmdef,
4276 4277
                             virDomainDeviceDefPtr dev)
{
4278
    virDomainDiskDefPtr disk;
4279
    virDomainNetDefPtr net;
4280
    virDomainLeaseDefPtr lease;
4281

4282
    switch (dev->type) {
4283 4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294 4295 4296 4297 4298 4299 4300 4301 4302
    case VIR_DOMAIN_DEVICE_DISK:
        disk = dev->data.disk;
        if (virDomainDiskIndexByName(vmdef, disk->dst) >= 0) {
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("target %s already exists."), disk->dst);
            return -1;
        }
        if (virDomainDiskInsert(vmdef, disk)) {
            virReportOOMError();
            return -1;
        }
        /* vmdef has the pointer. Generic codes for vmdef will do all jobs */
        dev->data.disk = NULL;
        if (disk->bus != VIR_DOMAIN_DISK_BUS_VIRTIO)
            if (virDomainDefAddImplicitControllers(vmdef) < 0)
                return -1;
        if (qemuDomainAssignPCIAddresses(vmdef) < 0)
            return -1;
        break;

4303 4304 4305 4306 4307 4308 4309 4310 4311 4312 4313 4314 4315 4316 4317 4318 4319
    case VIR_DOMAIN_DEVICE_NET:
        net = dev->data.net;
        if (virDomainNetIndexByMac(vmdef, net->mac) >= 0) {
            char macbuf[VIR_MAC_STRING_BUFLEN];
            virFormatMacAddr(net->mac, macbuf);
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("mac %s already exists"), macbuf);
            return -1;
        }
        if (virDomainNetInsert(vmdef, net)) {
            virReportOOMError();
            return -1;
        }
        dev->data.net = NULL;
        if (qemuDomainAssignPCIAddresses(vmdef) < 0)
            return -1;
        break;
4320 4321 4322 4323 4324 4325 4326 4327 4328 4329 4330 4331 4332 4333 4334 4335

    case VIR_DOMAIN_DEVICE_LEASE:
        lease = dev->data.lease;
        if (virDomainLeaseIndex(vmdef, lease) >= 0) {
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("Lease %s in lockspace %s already exists"),
                            lease->key, NULLSTR(lease->lockspace));
            return -1;
        }
        if (virDomainLeaseInsert(vmdef, lease) < 0)
            return -1;

        /* vmdef has the pointer. Generic codes for vmdef will do all jobs */
        dev->data.lease = NULL;
        break;

4336 4337 4338 4339 4340 4341 4342 4343 4344 4345
    default:
         qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                         _("persistent attach of device is not supported"));
         return -1;
    }
    return 0;
}


static int
4346
qemuDomainDetachDeviceConfig(virDomainDefPtr vmdef,
4347 4348
                             virDomainDeviceDefPtr dev)
{
4349
    virDomainDiskDefPtr disk;
4350
    virDomainNetDefPtr net;
4351
    virDomainLeaseDefPtr lease;
4352

4353
    switch (dev->type) {
4354 4355 4356 4357 4358 4359 4360 4361
    case VIR_DOMAIN_DEVICE_DISK:
        disk = dev->data.disk;
        if (virDomainDiskRemoveByName(vmdef, disk->dst)) {
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("no target device %s"), disk->dst);
            return -1;
        }
        break;
4362

4363 4364 4365 4366 4367 4368 4369 4370 4371 4372 4373
    case VIR_DOMAIN_DEVICE_NET:
        net = dev->data.net;
        if (virDomainNetRemoveByMac(vmdef, net->mac)) {
            char macbuf[VIR_MAC_STRING_BUFLEN];

            virFormatMacAddr(net->mac, macbuf);
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("no nic of mac %s"), macbuf);
            return -1;
        }
        break;
4374 4375 4376 4377 4378 4379 4380 4381 4382 4383 4384

    case VIR_DOMAIN_DEVICE_LEASE:
        lease = dev->data.lease;
        if (virDomainLeaseRemove(vmdef, lease) < 0) {
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("Lease %s in lockspace %s does not exist"),
                            lease->key, NULLSTR(lease->lockspace));
            return -1;
        }
        break;

4385 4386 4387 4388 4389 4390 4391 4392 4393
    default:
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                        _("persistent detach of device is not supported"));
        return -1;
    }
    return 0;
}

static int
4394
qemuDomainUpdateDeviceConfig(virDomainDefPtr vmdef,
4395 4396
                             virDomainDeviceDefPtr dev)
{
4397 4398 4399
    virDomainDiskDefPtr orig, disk;
    int pos;

4400
    switch (dev->type) {
4401 4402 4403 4404 4405 4406 4407 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
    case VIR_DOMAIN_DEVICE_DISK:
        disk = dev->data.disk;
        pos = virDomainDiskIndexByName(vmdef, disk->dst);
        if (pos < 0) {
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("target %s doesn't exists."), disk->dst);
            return -1;
        }
        orig = vmdef->disks[pos];
        if (!(orig->device == VIR_DOMAIN_DISK_DEVICE_CDROM) &&
            !(orig->device == VIR_DOMAIN_DISK_DEVICE_FLOPPY)) {
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("this disk doesn't support update"));
            return -1;
        }
        /*
         * Update 'orig'
         * We allow updating src/type//driverType/cachemode/
         */
        VIR_FREE(orig->src);
        orig->src = disk->src;
        orig->type = disk->type;
        orig->cachemode = disk->cachemode;
        if (disk->driverName) {
            VIR_FREE(orig->driverName);
            orig->driverName = disk->driverName;
            disk->driverName = NULL;
        }
        if (disk->driverType) {
            VIR_FREE(orig->driverType);
            orig->driverType = disk->driverType;
            disk->driverType = NULL;
        }
        disk->src = NULL;
        break;
4436 4437 4438 4439 4440 4441 4442 4443
    default:
        qemuReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                         _("persistent update of device is not supported"));
        return -1;
    }
    return 0;
}

4444 4445 4446 4447 4448 4449 4450 4451 4452 4453 4454
/* Actions for qemuDomainModifyDeviceFlags */
enum {
    QEMU_DEVICE_ATTACH,
    QEMU_DEVICE_DETACH,
    QEMU_DEVICE_UPDATE,
};


static int
qemuDomainModifyDeviceFlags(virDomainPtr dom, const char *xml,
                            unsigned int flags, int action)
4455
{
4456
    struct qemud_driver *driver = dom->conn->privateData;
4457
    virDomainObjPtr vm = NULL;
4458
    virDomainDefPtr vmdef = NULL;
4459
    virDomainDeviceDefPtr dev = NULL;
4460
    bool force = (flags & VIR_DOMAIN_DEVICE_MODIFY_FORCE) != 0;
4461
    int ret = -1;
4462

4463
    virCheckFlags(VIR_DOMAIN_DEVICE_MODIFY_LIVE |
4464 4465 4466 4467
                  VIR_DOMAIN_DEVICE_MODIFY_CONFIG |
                  (action == QEMU_DEVICE_UPDATE ?
                   VIR_DOMAIN_DEVICE_MODIFY_FORCE : 0), -1);

4468 4469 4470 4471 4472 4473 4474
    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);
4475 4476 4477
        goto cleanup;
    }

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

4481 4482 4483 4484 4485 4486 4487 4488 4489 4490 4491 4492 4493 4494
    if (virDomainObjIsActive(vm)) {
        if (flags == VIR_DOMAIN_DEVICE_MODIFY_CURRENT)
            flags |= VIR_DOMAIN_DEVICE_MODIFY_LIVE;
    } else {
        if (flags == VIR_DOMAIN_DEVICE_MODIFY_CURRENT)
            flags |= VIR_DOMAIN_DEVICE_MODIFY_CONFIG;
        /* check consistency between flags and the vm state */
        if (flags & VIR_DOMAIN_DEVICE_MODIFY_LIVE) {
            qemuReportError(VIR_ERR_OPERATION_INVALID,
                            "%s",
                            _("cannot do live update a device on "
                              "inactive domain"));
            goto endjob;
        }
4495
    }
4496

4497 4498 4499 4500 4501
    if ((flags & VIR_DOMAIN_DEVICE_MODIFY_CONFIG) && !vm->persistent) {
         qemuReportError(VIR_ERR_OPERATION_INVALID,
                         "%s", _("cannot modify device on transient domain"));
         goto endjob;
    }
4502

4503
    if (flags & VIR_DOMAIN_DEVICE_MODIFY_CONFIG) {
4504 4505 4506 4507 4508
        dev = virDomainDeviceDefParse(driver->caps, vm->def, xml,
                                      VIR_DOMAIN_XML_INACTIVE);
        if (dev == NULL)
            goto endjob;

4509 4510 4511 4512 4513 4514 4515 4516 4517 4518 4519 4520 4521 4522 4523 4524 4525 4526 4527 4528 4529
        /* Make a copy for updated domain. */
        vmdef = virDomainObjCopyPersistentDef(driver->caps, vm);
        if (!vmdef)
            goto endjob;
        switch (action) {
        case QEMU_DEVICE_ATTACH:
            ret = qemuDomainAttachDeviceConfig(vmdef, dev);
            break;
        case QEMU_DEVICE_DETACH:
            ret = qemuDomainDetachDeviceConfig(vmdef, dev);
            break;
        case QEMU_DEVICE_UPDATE:
            ret = qemuDomainUpdateDeviceConfig(vmdef, dev);
            break;
        default:
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("unknown domain modify action %d"), action);
            break;
        }
    } else
        ret = 0;
4530

4531
    if (!ret && (flags & VIR_DOMAIN_DEVICE_MODIFY_LIVE)) {
4532 4533 4534 4535 4536 4537 4538
        /* If dev exists it was created to modify the domain config. Free it. */
        virDomainDeviceDefFree(dev);
        dev = virDomainDeviceDefParse(driver->caps, vm->def, xml,
                                      VIR_DOMAIN_XML_INACTIVE);
        if (dev == NULL)
            goto endjob;

4539 4540
        switch (action) {
        case QEMU_DEVICE_ATTACH:
4541
            ret = qemuDomainAttachDeviceLive(vm, dev, dom);
4542 4543
            break;
        case QEMU_DEVICE_DETACH:
4544
            ret = qemuDomainDetachDeviceLive(vm, dev, dom);
4545 4546
            break;
        case QEMU_DEVICE_UPDATE:
4547
            ret = qemuDomainUpdateDeviceLive(vm, dev, dom, force);
4548 4549 4550 4551 4552 4553 4554 4555 4556 4557 4558 4559 4560 4561 4562 4563 4564 4565 4566 4567 4568 4569
            break;
        default:
            qemuReportError(VIR_ERR_INTERNAL_ERROR,
                            _("unknown domain modify action %d"), action);
            break;
        }
        /*
         * update domain status forcibly because the domain status may be
         * changed even if we attach the device failed. For example, a
         * For example, a new controller may be created.
         */
        if (virDomainSaveStatus(driver->caps, driver->stateDir, vm) < 0)
            ret = -1;
    }
    /* Finally, if no error until here, we can save config. */
    if (!ret && (flags & VIR_DOMAIN_DEVICE_MODIFY_CONFIG)) {
        ret = virDomainSaveConfig(driver->configDir, vmdef);
        if (!ret) {
            virDomainObjAssignDef(vm, vmdef, false);
            vmdef = NULL;
        }
    }
4570 4571 4572 4573 4574 4575

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

cleanup:
4576
    virDomainDefFree(vmdef);
4577 4578 4579 4580
    virDomainDeviceDefFree(dev);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
4581 4582 4583
    return ret;
}

4584 4585 4586 4587 4588 4589
static int qemuDomainAttachDeviceFlags(virDomainPtr dom, const char *xml,
                                       unsigned int flags)
{
    return qemuDomainModifyDeviceFlags(dom, xml, flags, QEMU_DEVICE_ATTACH);
}

4590 4591 4592 4593
static int qemuDomainAttachDevice(virDomainPtr dom, const char *xml)
{
    return qemuDomainAttachDeviceFlags(dom, xml,
                                       VIR_DOMAIN_DEVICE_MODIFY_LIVE);
4594
}
4595

4596

4597 4598 4599 4600
static int qemuDomainUpdateDeviceFlags(virDomainPtr dom,
                                       const char *xml,
                                       unsigned int flags)
{
4601
    return qemuDomainModifyDeviceFlags(dom, xml, flags, QEMU_DEVICE_UPDATE);
4602 4603
}

4604 4605 4606
static int qemuDomainDetachDeviceFlags(virDomainPtr dom, const char *xml,
                                       unsigned int flags)
{
4607
    return qemuDomainModifyDeviceFlags(dom, xml, flags, QEMU_DEVICE_DETACH);
4608 4609
}

4610 4611 4612 4613
static int qemuDomainDetachDevice(virDomainPtr dom, const char *xml)
{
    return qemuDomainDetachDeviceFlags(dom, xml,
                                       VIR_DOMAIN_DEVICE_MODIFY_LIVE);
4614 4615
}

4616
static int qemudDomainGetAutostart(virDomainPtr dom,
4617
                                   int *autostart) {
4618 4619 4620
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
4621

4622
    qemuDriverLock(driver);
4623
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
4624 4625
    qemuDriverUnlock(driver);

4626
    if (!vm) {
4627 4628
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
4629 4630
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
4631
        goto cleanup;
4632 4633 4634
    }

    *autostart = vm->autostart;
4635
    ret = 0;
4636

4637
cleanup:
4638 4639
    if (vm)
        virDomainObjUnlock(vm);
4640
    return ret;
4641 4642
}

4643
static int qemudDomainSetAutostart(virDomainPtr dom,
4644
                                   int autostart) {
4645 4646
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
4647 4648
    char *configFile = NULL, *autostartLink = NULL;
    int ret = -1;
4649

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

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

4661
    if (!vm->persistent) {
4662
        qemuReportError(VIR_ERR_OPERATION_INVALID,
4663
                        "%s", _("cannot set autostart for transient domain"));
4664
        goto cleanup;
4665 4666
    }

4667 4668
    autostart = (autostart != 0);

4669
    if (vm->autostart != autostart) {
4670
        if ((configFile = virDomainConfigFile(driver->configDir, vm->def->name)) == NULL)
4671
            goto cleanup;
4672
        if ((autostartLink = virDomainConfigFile(driver->autostartDir, vm->def->name)) == NULL)
4673
            goto cleanup;
4674

4675 4676
        if (autostart) {
            int err;
4677

4678
            if ((err = virFileMakePath(driver->autostartDir))) {
4679
                virReportSystemError(err,
4680 4681
                                     _("cannot create autostart directory %s"),
                                     driver->autostartDir);
4682 4683
                goto cleanup;
            }
4684

4685
            if (symlink(configFile, autostartLink) < 0) {
4686
                virReportSystemError(errno,
4687 4688
                                     _("Failed to create symlink '%s to '%s'"),
                                     autostartLink, configFile);
4689 4690 4691 4692
                goto cleanup;
            }
        } else {
            if (unlink(autostartLink) < 0 && errno != ENOENT && errno != ENOTDIR) {
4693
                virReportSystemError(errno,
4694 4695
                                     _("Failed to delete symlink '%s'"),
                                     autostartLink);
4696 4697
                goto cleanup;
            }
4698 4699
        }

4700
        vm->autostart = autostart;
4701
    }
4702
    ret = 0;
4703

4704 4705 4706
cleanup:
    VIR_FREE(configFile);
    VIR_FREE(autostartLink);
4707 4708
    if (vm)
        virDomainObjUnlock(vm);
4709
    qemuDriverUnlock(driver);
4710
    return ret;
4711 4712
}

4713 4714 4715 4716 4717

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

4720
    qemuDriverLock(driver);
4721
    if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_CPU)) {
4722 4723
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cgroup CPU controller is not mounted"));
4724
        goto cleanup;
4725 4726 4727 4728 4729 4730 4731
    }

    if (nparams)
        *nparams = 1;

    ret = strdup("posix");
    if (!ret)
4732
        virReportOOMError();
4733 4734 4735

cleanup:
    qemuDriverUnlock(driver);
4736 4737 4738
    return ret;
}

4739
static int qemuDomainSetBlkioParameters(virDomainPtr dom,
4740
                                         virTypedParameterPtr params,
4741 4742 4743 4744 4745 4746 4747 4748 4749 4750 4751 4752 4753 4754 4755 4756 4757 4758 4759 4760 4761 4762 4763 4764
                                         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;
    }

4765 4766 4767 4768 4769 4770
    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
    }

4771 4772 4773 4774 4775 4776 4777 4778
    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++) {
4779
        virTypedParameterPtr param = &params[i];
4780 4781 4782

        if (STREQ(param->field, VIR_DOMAIN_BLKIO_WEIGHT)) {
            int rc;
4783
            if (param->type != VIR_TYPED_PARAM_UINT) {
4784 4785 4786 4787 4788 4789 4790 4791 4792 4793 4794 4795 4796 4797 4798 4799 4800 4801 4802 4803 4804 4805 4806 4807 4808 4809 4810 4811 4812 4813 4814 4815 4816 4817 4818
                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,
4819
                                         virTypedParameterPtr params,
4820 4821 4822 4823 4824 4825 4826 4827 4828 4829 4830 4831 4832 4833 4834 4835 4836 4837 4838 4839 4840 4841 4842 4843 4844 4845 4846
                                         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;
    }

4847 4848 4849 4850 4851 4852
    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
    }

4853 4854 4855 4856 4857 4858 4859 4860 4861 4862 4863 4864 4865 4866 4867 4868 4869 4870 4871 4872
    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++) {
4873
        virTypedParameterPtr param = &params[i];
4874 4875
        val = 0;
        param->value.ui = 0;
4876
        param->type = VIR_TYPED_PARAM_UINT;
4877

4878
        switch (i) {
4879 4880 4881 4882 4883 4884 4885 4886 4887 4888 4889 4890 4891 4892 4893 4894 4895 4896 4897 4898 4899 4900 4901 4902 4903 4904 4905 4906 4907 4908 4909
        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;
}
4910 4911

static int qemuDomainSetMemoryParameters(virDomainPtr dom,
4912
                                         virTypedParameterPtr params,
4913
                                         int nparams,
4914
                                         unsigned int flags)
4915 4916 4917
{
    struct qemud_driver *driver = dom->conn->privateData;
    int i;
4918
    virDomainDefPtr persistentDef = NULL;
4919 4920 4921
    virCgroupPtr group = NULL;
    virDomainObjPtr vm = NULL;
    int ret = -1;
4922 4923 4924 4925
    bool isActive;

    virCheckFlags(VIR_DOMAIN_MEMORY_PARAM_LIVE |
                  VIR_DOMAIN_MEMORY_PARAM_CONFIG, -1);
4926 4927 4928 4929 4930 4931 4932 4933 4934 4935 4936

    qemuDriverLock(driver);

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

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

4937 4938 4939 4940 4941 4942 4943
    isActive = virDomainObjIsActive(vm);

    if (flags == VIR_DOMAIN_MEMORY_PARAM_CURRENT) {
        if (isActive)
            flags = VIR_DOMAIN_MEMORY_PARAM_LIVE;
        else
            flags = VIR_DOMAIN_MEMORY_PARAM_CONFIG;
4944 4945
    }

4946 4947 4948 4949 4950 4951 4952 4953 4954 4955 4956 4957 4958 4959 4960 4961 4962 4963 4964 4965 4966 4967 4968 4969 4970 4971 4972 4973
    if (flags & VIR_DOMAIN_MEMORY_PARAM_LIVE) {
        if (!isActive) {
            qemuReportError(VIR_ERR_OPERATION_INVALID,
                            "%s", _("domain is not running"));
            goto cleanup;
        }

        if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_MEMORY)) {
            qemuReportError(VIR_ERR_OPERATION_INVALID,
                            "%s", _("cgroup memory controller is not mounted"));
            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;
        }
    }

    if (flags & VIR_DOMAIN_MEMORY_PARAM_CONFIG) {
        if (!vm->persistent) {
            qemuReportError(VIR_ERR_OPERATION_INVALID, "%s",
                            _("cannot change persistent config of a transient domain"));
            goto cleanup;
        }
        if (!(persistentDef = virDomainObjGetPersistentDef(driver->caps, vm)))
            goto cleanup;
4974 4975 4976 4977
    }

    ret = 0;
    for (i = 0; i < nparams; i++) {
4978
        virTypedParameterPtr param = &params[i];
4979 4980 4981

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

4989 4990 4991 4992 4993 4994 4995 4996 4997 4998 4999
            if (flags & VIR_DOMAIN_MEMORY_PARAM_LIVE) {
                rc = virCgroupSetMemoryHardLimit(group, params[i].value.ul);
                if (rc != 0) {
                    virReportSystemError(-rc, "%s",
                                         _("unable to set memory hard_limit tunable"));
                    ret = -1;
                }
            }

            if (flags & VIR_DOMAIN_MEMORY_PARAM_CONFIG) {
                persistentDef->mem.hard_limit = params[i].value.ul;
5000 5001 5002
            }
        } else if (STREQ(param->field, VIR_DOMAIN_MEMORY_SOFT_LIMIT)) {
            int rc;
5003
            if (param->type != VIR_TYPED_PARAM_ULLONG) {
5004 5005 5006 5007 5008 5009
                qemuReportError(VIR_ERR_INVALID_ARG, "%s",
                                _("invalid type for memory soft_limit tunable, expected a 'ullong'"));
                ret = -1;
                continue;
            }

5010 5011 5012 5013 5014 5015 5016 5017 5018 5019 5020
            if (flags & VIR_DOMAIN_MEMORY_PARAM_LIVE) {
                rc = virCgroupSetMemorySoftLimit(group, params[i].value.ul);
                if (rc != 0) {
                    virReportSystemError(-rc, "%s",
                                         _("unable to set memory soft_limit tunable"));
                    ret = -1;
                }
            }

            if (flags & VIR_DOMAIN_MEMORY_PARAM_CONFIG) {
                persistentDef->mem.soft_limit = params[i].value.ul;
5021
            }
5022
        } else if (STREQ(param->field, VIR_DOMAIN_MEMORY_SWAP_HARD_LIMIT)) {
5023
            int rc;
5024
            if (param->type != VIR_TYPED_PARAM_ULLONG) {
5025 5026 5027 5028 5029 5030
                qemuReportError(VIR_ERR_INVALID_ARG, "%s",
                                _("invalid type for swap_hard_limit tunable, expected a 'ullong'"));
                ret = -1;
                continue;
            }

5031 5032 5033 5034 5035 5036 5037 5038 5039 5040
            if (flags & VIR_DOMAIN_MEMORY_PARAM_LIVE) {
                rc = virCgroupSetMemSwapHardLimit(group, params[i].value.ul);
                if (rc != 0) {
                    virReportSystemError(-rc, "%s",
                                         _("unable to set swap_hard_limit tunable"));
                    ret = -1;
                }
            }
            if (flags & VIR_DOMAIN_MEMORY_PARAM_CONFIG) {
                persistentDef->mem.swap_hard_limit = params[i].value.ul;
5041 5042 5043 5044 5045 5046 5047 5048 5049 5050 5051 5052
            }
        } 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;
        }
    }

5053 5054 5055 5056
    if (flags & VIR_DOMAIN_MEMORY_PARAM_CONFIG) {
        ret = virDomainSaveConfig(driver->configDir, persistentDef);
    }

5057 5058 5059 5060 5061 5062 5063 5064
cleanup:
    virCgroupFree(&group);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}

5065
static int qemuDomainGetMemoryParameters(virDomainPtr dom,
5066
                                         virTypedParameterPtr params,
5067
                                         int *nparams,
5068
                                         unsigned int flags)
5069 5070 5071 5072 5073
{
    struct qemud_driver *driver = dom->conn->privateData;
    int i;
    virCgroupPtr group = NULL;
    virDomainObjPtr vm = NULL;
5074
    virDomainDefPtr persistentDef = NULL;
5075
    unsigned long long val;
5076 5077
    int ret = -1;
    int rc;
5078
    bool isActive;
5079

5080 5081
    virCheckFlags(VIR_DOMAIN_MEMORY_PARAM_LIVE |
                  VIR_DOMAIN_MEMORY_PARAM_CONFIG, -1);
5082

5083
    qemuDriverLock(driver);
5084 5085 5086 5087 5088 5089 5090 5091 5092

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

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

5093 5094 5095 5096 5097 5098 5099 5100 5101 5102 5103 5104 5105 5106 5107 5108 5109 5110 5111 5112 5113 5114 5115 5116 5117 5118 5119 5120 5121 5122 5123 5124 5125 5126 5127 5128 5129
    isActive = virDomainObjIsActive(vm);

    if (flags == VIR_DOMAIN_MEMORY_PARAM_CURRENT) {
        if (isActive)
            flags = VIR_DOMAIN_MEMORY_PARAM_LIVE;
        else
            flags = VIR_DOMAIN_MEMORY_PARAM_CONFIG;
    }

    if (flags & VIR_DOMAIN_MEMORY_PARAM_LIVE) {
        if (!isActive) {
            qemuReportError(VIR_ERR_OPERATION_INVALID,
                            "%s", _("domain is not running"));
            goto cleanup;
        }

        if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_MEMORY)) {
            qemuReportError(VIR_ERR_OPERATION_INVALID,
                            "%s", _("cgroup memory controller is not mounted"));
            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;
        }
    }

    if (flags & VIR_DOMAIN_MEMORY_PARAM_CONFIG) {
        if (!vm->persistent) {
            qemuReportError(VIR_ERR_OPERATION_INVALID, "%s",
                            _("cannot change persistent config of a transient domain"));
            goto cleanup;
        }
        if (!(persistentDef = virDomainObjGetPersistentDef(driver->caps, vm)))
            goto cleanup;
5130 5131
    }

5132 5133 5134 5135 5136 5137 5138
    if ((*nparams) == 0) {
        /* Current number of memory parameters supported by cgroups */
        *nparams = QEMU_NB_MEM_PARAM;
        ret = 0;
        goto cleanup;
    }

5139
    if ((*nparams) < QEMU_NB_MEM_PARAM) {
5140 5141 5142 5143 5144
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("Invalid parameter count"));
        goto cleanup;
    }

5145 5146 5147 5148 5149 5150 5151 5152 5153 5154 5155 5156 5157 5158 5159 5160 5161 5162 5163 5164 5165 5166 5167 5168 5169 5170 5171 5172 5173 5174 5175 5176 5177 5178 5179 5180 5181 5182 5183 5184 5185
    if (flags & VIR_DOMAIN_MEMORY_PARAM_CONFIG) {
        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 */
                if (virStrcpyStatic(param->field, VIR_DOMAIN_MEMORY_HARD_LIMIT) == NULL) {
                    qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                    "%s", _("Field memory hard limit too long for destination"));
                    goto cleanup;
                }
                param->value.ul = persistentDef->mem.hard_limit;
                break;

            case 1: /* fill memory soft limit here */
                if (virStrcpyStatic(param->field, VIR_DOMAIN_MEMORY_SOFT_LIMIT) == NULL) {
                    qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                    "%s", _("Field memory soft limit too long for destination"));
                    goto cleanup;
                }
                param->value.ul = persistentDef->mem.soft_limit;
                break;

            case 2: /* fill swap hard limit here */
                if (virStrcpyStatic(param->field, VIR_DOMAIN_MEMORY_SWAP_HARD_LIMIT) == NULL) {
                    qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                    "%s", _("Field swap hard limit too long for destination"));
                    goto cleanup;
                }
                param->value.ul = persistentDef->mem.swap_hard_limit;
                break;

            default:
                break;
                /* should not hit here */
            }
        }
        goto out;
5186 5187
    }

5188
    for (i = 0; i < QEMU_NB_MEM_PARAM; i++) {
5189
        virTypedParameterPtr param = &params[i];
5190 5191
        val = 0;
        param->value.ul = 0;
5192
        param->type = VIR_TYPED_PARAM_ULLONG;
5193

5194
        switch (i) {
5195 5196 5197 5198 5199
        case 0: /* fill memory hard limit here */
            rc = virCgroupGetMemoryHardLimit(group, &val);
            if (rc != 0) {
                virReportSystemError(-rc, "%s",
                                     _("unable to get memory hard limit"));
5200
                goto cleanup;
5201 5202 5203 5204
            }
            if (virStrcpyStatic(param->field, VIR_DOMAIN_MEMORY_HARD_LIMIT) == NULL) {
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                "%s", _("Field memory hard limit too long for destination"));
5205
                goto cleanup;
5206 5207 5208 5209 5210 5211 5212 5213 5214
            }
            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"));
5215
                goto cleanup;
5216 5217 5218 5219
            }
            if (virStrcpyStatic(param->field, VIR_DOMAIN_MEMORY_SOFT_LIMIT) == NULL) {
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                "%s", _("Field memory soft limit too long for destination"));
5220
                goto cleanup;
5221 5222 5223 5224 5225
            }
            param->value.ul = val;
            break;

        case 2: /* fill swap hard limit here */
5226
            rc = virCgroupGetMemSwapHardLimit(group, &val);
5227 5228 5229
            if (rc != 0) {
                virReportSystemError(-rc, "%s",
                                     _("unable to get swap hard limit"));
5230
                goto cleanup;
5231
            }
5232
            if (virStrcpyStatic(param->field, VIR_DOMAIN_MEMORY_SWAP_HARD_LIMIT) == NULL) {
5233 5234
                qemuReportError(VIR_ERR_INTERNAL_ERROR,
                                "%s", _("Field swap hard limit too long for destination"));
5235
                goto cleanup;
5236 5237 5238 5239 5240 5241 5242 5243 5244 5245
            }
            param->value.ul = val;
            break;

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

5246
out:
5247
    *nparams = QEMU_NB_MEM_PARAM;
5248 5249
    ret = 0;

5250 5251 5252 5253 5254 5255 5256 5257 5258
cleanup:
    if (group)
        virCgroupFree(&group);
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}

5259
static int qemuSetSchedulerParametersFlags(virDomainPtr dom,
5260
                                           virTypedParameterPtr params,
5261 5262
                                           int nparams,
                                           unsigned int flags)
5263 5264 5265 5266 5267
{
    struct qemud_driver *driver = dom->conn->privateData;
    int i;
    virCgroupPtr group = NULL;
    virDomainObjPtr vm = NULL;
5268
    virDomainDefPtr persistentDef = NULL;
5269
    int ret = -1;
5270 5271 5272 5273
    bool isActive;

    virCheckFlags(VIR_DOMAIN_SCHEDPARAM_LIVE |
                  VIR_DOMAIN_SCHEDPARAM_CONFIG, -1);
5274

5275
    qemuDriverLock(driver);
5276 5277 5278 5279

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

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

5285 5286 5287 5288 5289 5290 5291
    isActive = virDomainObjIsActive(vm);

    if (flags == VIR_DOMAIN_SCHEDPARAM_CURRENT) {
        if (isActive)
            flags = VIR_DOMAIN_SCHEDPARAM_LIVE;
        else
            flags = VIR_DOMAIN_SCHEDPARAM_CONFIG;
5292 5293
    }

5294
    if ((flags & VIR_DOMAIN_SCHEDPARAM_CONFIG) && !vm->persistent) {
5295 5296
        qemuReportError(VIR_ERR_OPERATION_INVALID, "%s",
                        _("cannot change persistent config of a transient domain"));
5297 5298 5299
        goto cleanup;
    }

5300 5301 5302 5303 5304 5305 5306 5307 5308 5309 5310 5311 5312 5313 5314 5315 5316 5317 5318 5319
    if (flags & VIR_DOMAIN_SCHEDPARAM_LIVE) {
        if (!isActive) {
            qemuReportError(VIR_ERR_OPERATION_INVALID,
                            "%s", _("domain is not running"));
            goto cleanup;
        }

        if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_CPU)) {
            qemuReportError(VIR_ERR_OPERATION_INVALID,
                            "%s", _("cgroup CPU controller is not mounted"));
            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;
        }
    }

5320
    for (i = 0; i < nparams; i++) {
5321
        virTypedParameterPtr param = &params[i];
5322 5323 5324

        if (STREQ(param->field, "cpu_shares")) {
            int rc;
5325
            if (param->type != VIR_TYPED_PARAM_ULLONG) {
5326 5327
                qemuReportError(VIR_ERR_INVALID_ARG, "%s",
                                _("invalid type for cpu_shares tunable, expected a 'ullong'"));
5328 5329 5330
                goto cleanup;
            }

5331 5332 5333 5334 5335 5336 5337 5338 5339
            if (flags & VIR_DOMAIN_SCHEDPARAM_LIVE) {
                rc = virCgroupSetCpuShares(group, params[i].value.ul);
                if (rc != 0) {
                    virReportSystemError(-rc, "%s",
                                         _("unable to set cpu shares tunable"));
                    goto cleanup;
                }

                vm->def->cputune.shares = params[i].value.ul;
5340
            }
5341

5342 5343 5344 5345 5346 5347 5348 5349 5350 5351 5352 5353 5354 5355 5356
            if (flags & VIR_DOMAIN_SCHEDPARAM_CONFIG) {
                persistentDef = virDomainObjGetPersistentDef(driver->caps, vm);
                if (!persistentDef) {
                    qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                                    _("can't get persistentDef"));
                    goto cleanup;
                }
                persistentDef->cputune.shares = params[i].value.ul;
                rc = virDomainSaveConfig(driver->configDir, persistentDef);
                if (rc) {
                    qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                                    _("can't save config"));
                    goto cleanup;
                }
            }
5357
        } else {
5358 5359
            qemuReportError(VIR_ERR_INVALID_ARG,
                            _("Invalid parameter `%s'"), param->field);
5360 5361 5362
            goto cleanup;
        }
    }
5363

5364 5365 5366 5367 5368 5369
    ret = 0;

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

5374
static int qemuSetSchedulerParameters(virDomainPtr dom,
5375
                                      virTypedParameterPtr params,
5376 5377 5378 5379 5380 5381 5382 5383
                                      int nparams)
{
    return qemuSetSchedulerParametersFlags(dom,
                                           params,
                                           nparams,
                                           VIR_DOMAIN_SCHEDPARAM_LIVE);
}

5384 5385 5386 5387 5388
static int
qemuGetSchedulerParametersFlags(virDomainPtr dom,
                                virTypedParameterPtr params,
                                int *nparams,
                                unsigned int flags)
5389 5390 5391 5392 5393 5394 5395
{
    struct qemud_driver *driver = dom->conn->privateData;
    virCgroupPtr group = NULL;
    virDomainObjPtr vm = NULL;
    unsigned long long val;
    int ret = -1;
    int rc;
5396
    bool isActive;
5397

5398 5399 5400 5401 5402 5403 5404
    virCheckFlags(VIR_DOMAIN_SCHEDPARAM_LIVE |
                  VIR_DOMAIN_SCHEDPARAM_CONFIG, -1);

    if ((flags & (VIR_DOMAIN_SCHEDPARAM_LIVE | VIR_DOMAIN_SCHEDPARAM_CONFIG)) ==
        (VIR_DOMAIN_SCHEDPARAM_LIVE | VIR_DOMAIN_SCHEDPARAM_CONFIG)) {
        qemuReportError(VIR_ERR_INVALID_ARG, "%s",
                        _("cannot query live and config together"));
5405
        goto cleanup;
5406 5407
    }

5408
    if (*nparams < 1) {
5409 5410
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("Invalid parameter count"));
5411
        goto cleanup;
5412 5413
    }

5414
    qemuDriverLock(driver);
5415 5416 5417
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);

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

5423 5424 5425 5426 5427 5428 5429 5430 5431 5432 5433 5434 5435 5436 5437 5438 5439 5440 5441 5442 5443 5444 5445 5446 5447 5448 5449 5450 5451
    isActive = virDomainObjIsActive(vm);

    if (flags == VIR_DOMAIN_SCHEDPARAM_CURRENT) {
        if (isActive)
            flags = VIR_DOMAIN_SCHEDPARAM_LIVE;
        else
            flags = VIR_DOMAIN_SCHEDPARAM_CONFIG;
    }

    if (flags & VIR_DOMAIN_SCHEDPARAM_CONFIG) {
        if (!vm->persistent) {
            qemuReportError(VIR_ERR_OPERATION_INVALID, "%s",
                            _("cannot query persistent config of a transient domain"));
            goto cleanup;
        }

        if (isActive) {
            virDomainDefPtr persistentDef;

            persistentDef = virDomainObjGetPersistentDef(driver->caps, vm);
            if (!persistentDef) {
                qemuReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                                _("can't get persistentDef"));
                goto cleanup;
            }
            val = persistentDef->cputune.shares;
        } else {
            val = vm->def->cputune.shares;
        }
5452
        goto out;
5453 5454
    }

5455 5456 5457 5458 5459 5460 5461 5462 5463 5464 5465 5466
    if (!isActive) {
        qemuReportError(VIR_ERR_OPERATION_INVALID, "%s",
                        _("domain is not running"));
        goto cleanup;
    }

    if (!qemuCgroupControllerActive(driver, VIR_CGROUP_CONTROLLER_CPU)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("cgroup CPU controller is not mounted"));
        goto cleanup;
    }

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

    rc = virCgroupGetCpuShares(group, &val);
    if (rc != 0) {
5475
        virReportSystemError(-rc, "%s",
5476 5477 5478
                             _("unable to get cpu shares tunable"));
        goto cleanup;
    }
5479
out:
5480
    params[0].value.ul = val;
5481
    params[0].type = VIR_TYPED_PARAM_ULLONG;
C
Chris Lalancette 已提交
5482
    if (virStrcpyStatic(params[0].field, "cpu_shares") == NULL) {
5483 5484
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("Field cpu_shares too long for destination"));
C
Chris Lalancette 已提交
5485 5486
        goto cleanup;
    }
5487

5488
    *nparams = 1;
5489 5490 5491 5492 5493 5494
    ret = 0;

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

5499 5500 5501 5502 5503 5504 5505 5506
static int
qemuGetSchedulerParameters(virDomainPtr dom,
                           virTypedParameterPtr params,
                           int *nparams)
{
    return qemuGetSchedulerParametersFlags(dom, params, nparams,
                                           VIR_DOMAIN_SCHEDPARAM_CURRENT);
}
5507

5508 5509 5510 5511 5512 5513 5514 5515 5516
/* 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)
{
5517
    struct qemud_driver *driver = dom->conn->privateData;
5518
    int i, ret = -1;
5519
    virDomainObjPtr vm;
5520
    virDomainDiskDefPtr disk = NULL;
5521
    qemuDomainObjPrivatePtr priv;
5522

5523
    qemuDriverLock(driver);
5524
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
5525
    qemuDriverUnlock(driver);
5526
    if (!vm) {
5527 5528
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
5529 5530
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
5531
        goto cleanup;
5532
    }
5533

5534 5535 5536 5537 5538 5539
    if (!virDomainObjIsActive(vm)) {
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
        goto cleanup;
    }

5540 5541 5542 5543 5544 5545 5546 5547
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (STREQ(path, vm->def->disks[i]->dst)) {
            disk = vm->def->disks[i];
            break;
        }
    }

    if (!disk) {
5548 5549
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("invalid path: %s"), path);
5550
        goto cleanup;
5551 5552
    }

5553
    if (!disk->info.alias) {
5554 5555
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("missing disk device alias name for %s"), disk->dst);
5556
        goto cleanup;
5557
    }
5558

5559
    priv = vm->privateData;
5560 5561 5562 5563 5564 5565 5566 5567 5568 5569 5570 5571 5572 5573 5574 5575 5576 5577 5578 5579 5580 5581 5582 5583 5584 5585 5586 5587 5588 5589 5590 5591 5592 5593 5594
    if ((priv->jobActive == QEMU_JOB_MIGRATION_OUT)
        || (priv->jobActive == QEMU_JOB_SAVE)) {
        virDomainObjRef(vm);
        while (priv->jobSignals & QEMU_JOB_SIGNAL_BLKSTAT)
            ignore_value(virCondWait(&priv->signalCond, &vm->lock));

        priv->jobSignalsData.statDevName = disk->info.alias;
        priv->jobSignalsData.blockStat = stats;
        priv->jobSignalsData.statRetCode = &ret;
        priv->jobSignals |= QEMU_JOB_SIGNAL_BLKSTAT;

        while (priv->jobSignals & QEMU_JOB_SIGNAL_BLKSTAT)
            ignore_value(virCondWait(&priv->signalCond, &vm->lock));

        if (virDomainObjUnref(vm) == 0)
            vm = NULL;
    } else {
        if (qemuDomainObjBeginJob(vm) < 0)
            goto cleanup;

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

        qemuDomainObjEnterMonitor(vm);
        ret = qemuMonitorGetBlockStatsInfo(priv->mon,
                                           disk->info.alias,
                                           &stats->rd_req,
                                           &stats->rd_bytes,
                                           &stats->wr_req,
                                           &stats->wr_bytes,
                                           &stats->errs);
        qemuDomainObjExitMonitor(vm);
5595

5596
endjob:
5597 5598 5599
        if (qemuDomainObjEndJob(vm) == 0)
            vm = NULL;
    }
5600

5601
cleanup:
5602 5603
    if (vm)
        virDomainObjUnlock(vm);
5604
    return ret;
5605 5606
}

5607
#ifdef __linux__
5608 5609 5610 5611 5612
static int
qemudDomainInterfaceStats (virDomainPtr dom,
                           const char *path,
                           struct _virDomainInterfaceStats *stats)
{
5613 5614
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
5615
    int i;
5616
    int ret = -1;
5617

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

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

D
Daniel P. Berrange 已提交
5630
    if (!virDomainObjIsActive(vm)) {
5631 5632
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
5633
        goto cleanup;
5634 5635 5636
    }

    /* Check the path is one of the domain's network interfaces. */
5637 5638
    for (i = 0 ; i < vm->def->nnets ; i++) {
        if (vm->def->nets[i]->ifname &&
5639 5640 5641 5642
            STREQ (vm->def->nets[i]->ifname, path)) {
            ret = 0;
            break;
        }
5643 5644
    }

5645
    if (ret == 0)
5646
        ret = linuxDomainInterfaceStats(path, stats);
5647
    else
5648 5649
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("invalid path, '%s' is not a known interface"), path);
5650

5651
cleanup:
5652 5653
    if (vm)
        virDomainObjUnlock(vm);
5654 5655
    return ret;
}
5656
#else
5657 5658 5659 5660
static int
qemudDomainInterfaceStats (virDomainPtr dom,
                           const char *path ATTRIBUTE_UNUSED,
                           struct _virDomainInterfaceStats *stats ATTRIBUTE_UNUSED)
5661
{
5662 5663
    qemuReportError(VIR_ERR_NO_SUPPORT,
                    "%s", __FUNCTION__);
5664 5665
    return -1;
}
5666
#endif
5667

5668 5669 5670 5671 5672 5673 5674 5675 5676 5677 5678 5679 5680 5681 5682 5683
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);
5684 5685
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
5686 5687 5688
        goto cleanup;
    }

5689 5690 5691
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

5692 5693 5694
    if (virDomainObjIsActive(vm)) {
        qemuDomainObjPrivatePtr priv = vm->privateData;
        qemuDomainObjEnterMonitor(vm);
5695
        ret = qemuMonitorGetMemoryStats(priv->mon, stats, nr_stats);
5696 5697
        qemuDomainObjExitMonitor(vm);
    } else {
5698 5699
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
5700 5701
    }

5702 5703 5704
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;

5705 5706 5707 5708 5709 5710
cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    return ret;
}

5711 5712 5713 5714 5715 5716 5717
static int
qemudDomainBlockPeek (virDomainPtr dom,
                      const char *path,
                      unsigned long long offset, size_t size,
                      void *buffer,
                      unsigned int flags ATTRIBUTE_UNUSED)
{
5718 5719 5720
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int fd = -1, ret = -1, i;
5721

5722
    qemuDriverLock(driver);
5723
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
5724 5725
    qemuDriverUnlock(driver);

5726
    if (!vm) {
5727 5728
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
5729 5730
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
5731
        goto cleanup;
5732 5733 5734
    }

    if (!path || path[0] == '\0') {
5735 5736
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("NULL or empty path"));
5737
        goto cleanup;
5738 5739 5740
    }

    /* Check the path belongs to this domain. */
5741 5742
    for (i = 0 ; i < vm->def->ndisks ; i++) {
        if (vm->def->disks[i]->src != NULL &&
5743 5744 5745 5746
            STREQ (vm->def->disks[i]->src, path)) {
            ret = 0;
            break;
        }
5747 5748
    }

5749 5750 5751 5752 5753
    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) {
5754 5755
            virReportSystemError(errno,
                                 _("%s: failed to open"), path);
5756 5757
            goto cleanup;
        }
5758

5759 5760 5761 5762 5763 5764
        /* 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) {
5765 5766
            virReportSystemError(errno,
                                 _("%s: failed to seek or read"), path);
5767 5768 5769 5770 5771
            goto cleanup;
        }

        ret = 0;
    } else {
5772 5773
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("invalid path"));
5774 5775
    }

5776
cleanup:
5777
    VIR_FORCE_CLOSE(fd);
5778 5779
    if (vm)
        virDomainObjUnlock(vm);
5780 5781 5782
    return ret;
}

R
Richard W.M. Jones 已提交
5783 5784 5785 5786 5787 5788
static int
qemudDomainMemoryPeek (virDomainPtr dom,
                       unsigned long long offset, size_t size,
                       void *buffer,
                       unsigned int flags)
{
5789 5790
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
5791
    char *tmp = NULL;
R
Richard W.M. Jones 已提交
5792
    int fd = -1, ret = -1;
5793
    qemuDomainObjPrivatePtr priv;
R
Richard W.M. Jones 已提交
5794

5795
    qemuDriverLock(driver);
5796
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
5797
    qemuDriverUnlock(driver);
R
Richard W.M. Jones 已提交
5798 5799

    if (!vm) {
5800 5801
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
5802 5803
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
5804 5805 5806
        goto cleanup;
    }

5807
    if (flags != VIR_MEMORY_VIRTUAL && flags != VIR_MEMORY_PHYSICAL) {
5808 5809
        qemuReportError(VIR_ERR_INVALID_ARG,
                        "%s", _("flags parameter must be VIR_MEMORY_VIRTUAL or VIR_MEMORY_PHYSICAL"));
5810
        goto cleanup;
R
Richard W.M. Jones 已提交
5811 5812
    }

5813 5814 5815
    if (qemuDomainObjBeginJob(vm) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
5816
    if (!virDomainObjIsActive(vm)) {
5817 5818
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not running"));
5819
        goto endjob;
R
Richard W.M. Jones 已提交
5820 5821
    }

5822
    if (virAsprintf(&tmp, "%s/qemu.mem.XXXXXX", driver->cacheDir) < 0) {
5823
        virReportOOMError();
5824
        goto endjob;
5825 5826
    }

R
Richard W.M. Jones 已提交
5827 5828
    /* Create a temporary filename. */
    if ((fd = mkstemp (tmp)) == -1) {
5829 5830
        virReportSystemError(errno,
                             _("mkstemp(\"%s\") failed"), tmp);
5831
        goto endjob;
R
Richard W.M. Jones 已提交
5832 5833
    }

5834 5835
    virSecurityManagerSetSavedStateLabel(qemu_driver->securityManager, vm, tmp);

5836
    priv = vm->privateData;
5837
    qemuDomainObjEnterMonitor(vm);
5838
    if (flags == VIR_MEMORY_VIRTUAL) {
5839 5840
        if (qemuMonitorSaveVirtualMemory(priv->mon, offset, size, tmp) < 0) {
            qemuDomainObjExitMonitor(vm);
5841
            goto endjob;
5842
        }
5843
    } else {
5844 5845
        if (qemuMonitorSavePhysicalMemory(priv->mon, offset, size, tmp) < 0) {
            qemuDomainObjExitMonitor(vm);
5846
            goto endjob;
5847
        }
R
Richard W.M. Jones 已提交
5848
    }
5849
    qemuDomainObjExitMonitor(vm);
R
Richard W.M. Jones 已提交
5850 5851 5852

    /* Read the memory file into buffer. */
    if (saferead (fd, buffer, size) == (ssize_t) -1) {
5853 5854 5855
        virReportSystemError(errno,
                             _("failed to read temporary file "
                               "created with template %s"), tmp);
5856
        goto endjob;
R
Richard W.M. Jones 已提交
5857 5858 5859
    }

    ret = 0;
5860

5861
endjob:
5862 5863
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;
5864

5865
cleanup:
5866
    VIR_FORCE_CLOSE(fd);
5867 5868
    if (tmp)
        unlink(tmp);
W
Wen Congyang 已提交
5869
    VIR_FREE(tmp);
5870 5871
    if (vm)
        virDomainObjUnlock(vm);
R
Richard W.M. Jones 已提交
5872 5873 5874
    return ret;
}

5875

5876 5877 5878 5879 5880 5881 5882 5883 5884 5885
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;
5886
    virDomainDiskDefPtr disk = NULL;
5887 5888
    struct stat sb;
    int i;
5889
    int format;
5890 5891 5892 5893 5894 5895 5896 5897 5898 5899 5900 5901 5902 5903 5904 5905 5906 5907 5908 5909 5910 5911 5912 5913

    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)) {
5914
            disk = vm->def->disks[i];
5915 5916 5917 5918
            break;
        }
    }

5919
    if (!disk) {
5920 5921 5922 5923 5924 5925 5926 5927 5928 5929 5930 5931 5932 5933
        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 */
5934 5935 5936 5937 5938 5939 5940 5941
    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 {
5942 5943 5944 5945 5946 5947 5948
        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);
5949
            goto cleanup;
5950
        }
5951 5952 5953 5954 5955
    }

    if (virStorageFileGetMetadataFromFD(path, fd,
                                        format,
                                        &meta) < 0)
5956 5957 5958 5959 5960 5961 5962 5963 5964 5965
        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)) {
5966
#ifndef WIN32
5967 5968 5969 5970 5971 5972 5973 5974 5975 5976 5977 5978 5979 5980 5981 5982 5983 5984 5985 5986 5987 5988 5989 5990 5991 5992 5993 5994
        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;

5995
    /* Set default value .. */
5996 5997
    info->allocation = info->physical;

5998 5999 6000
    /* ..but if guest is running & not using raw
       disk format and on a block device, then query
       highest allocated extent from QEMU */
6001
    if (disk->type == VIR_DOMAIN_DISK_TYPE_BLOCK &&
6002
        format != VIR_STORAGE_FILE_RAW &&
6003 6004
        S_ISBLK(sb.st_mode) &&
        virDomainObjIsActive(vm)) {
6005
        qemuDomainObjPrivatePtr priv = vm->privateData;
6006 6007 6008 6009 6010 6011 6012 6013 6014 6015 6016 6017 6018 6019 6020 6021 6022 6023 6024 6025 6026

        if ((priv->jobActive == QEMU_JOB_MIGRATION_OUT)
            || (priv->jobActive == QEMU_JOB_SAVE)) {
            virDomainObjRef(vm);
            while (priv->jobSignals & QEMU_JOB_SIGNAL_BLKINFO)
                ignore_value(virCondWait(&priv->signalCond, &vm->lock));

            priv->jobSignalsData.infoDevName = disk->info.alias;
            priv->jobSignalsData.blockInfo = info;
            priv->jobSignalsData.infoRetCode = &ret;
            priv->jobSignals |= QEMU_JOB_SIGNAL_BLKINFO;

            while (priv->jobSignals & QEMU_JOB_SIGNAL_BLKINFO)
                ignore_value(virCondWait(&priv->signalCond, &vm->lock));

            if (virDomainObjUnref(vm) == 0)
                vm = NULL;
        } else {
            if (qemuDomainObjBeginJob(vm) < 0)
                goto cleanup;

6027 6028 6029 6030 6031 6032 6033 6034
            if (virDomainObjIsActive(vm)) {
                qemuDomainObjEnterMonitor(vm);
                ret = qemuMonitorGetBlockExtent(priv->mon,
                                                disk->info.alias,
                                                &info->allocation);
                qemuDomainObjExitMonitor(vm);
            } else {
                ret = 0;
6035 6036 6037 6038 6039
            }

            if (qemuDomainObjEndJob(vm) == 0)
                vm = NULL;
        }
6040 6041 6042
    } else {
        ret = 0;
    }
6043 6044

cleanup:
6045
    VIR_FORCE_CLOSE(fd);
6046 6047 6048 6049 6050 6051
    if (vm)
        virDomainObjUnlock(vm);
    return ret;
}


6052
static int
6053 6054 6055 6056
qemuDomainEventRegister(virConnectPtr conn,
                        virConnectDomainEventCallback callback,
                        void *opaque,
                        virFreeCallback freecb)
6057
{
6058 6059 6060
    struct qemud_driver *driver = conn->privateData;
    int ret;

6061
    qemuDriverLock(driver);
6062 6063
    ret = virDomainEventCallbackListAdd(conn,
                                        driver->domainEventState->callbacks,
6064
                                        callback, opaque, freecb);
6065
    qemuDriverUnlock(driver);
6066

6067
    return ret;
6068 6069
}

6070

6071
static int
6072 6073
qemuDomainEventDeregister(virConnectPtr conn,
                          virConnectDomainEventCallback callback)
6074
{
6075 6076 6077
    struct qemud_driver *driver = conn->privateData;
    int ret;

6078
    qemuDriverLock(driver);
6079 6080 6081
    ret = virDomainEventStateDeregister(conn,
                                        driver->domainEventState,
                                        callback);
6082
    qemuDriverUnlock(driver);
6083

6084
    return ret;
6085 6086
}

6087 6088 6089 6090 6091 6092 6093 6094 6095 6096 6097 6098 6099 6100

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,
6101
                                          driver->domainEventState->callbacks,
6102 6103 6104 6105 6106 6107 6108 6109 6110 6111 6112 6113 6114 6115 6116 6117
                                          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);
6118 6119 6120
    ret = virDomainEventStateDeregisterAny(conn,
                                           driver->domainEventState,
                                           callbackID);
6121 6122 6123 6124 6125 6126
    qemuDriverUnlock(driver);

    return ret;
}


6127 6128 6129
/*******************************************************************
 * Migration Protocol Version 2
 *******************************************************************/
D
Daniel Veillard 已提交
6130

C
Chris Lalancette 已提交
6131 6132 6133 6134 6135 6136 6137 6138 6139 6140 6141 6142 6143 6144 6145
/* 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;
6146

6147 6148 6149 6150 6151 6152 6153 6154 6155
    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);

6156 6157
    qemuDriverLock(driver);

C
Chris Lalancette 已提交
6158
    if (!dom_xml) {
6159 6160
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("no domain XML passed"));
C
Chris Lalancette 已提交
6161 6162 6163
        goto cleanup;
    }
    if (!(flags & VIR_MIGRATE_TUNNELLED)) {
6164
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
C
Chris Lalancette 已提交
6165 6166 6167 6168
                         "%s", _("PrepareTunnel called but no TUNNELLED flag set"));
        goto cleanup;
    }
    if (st == NULL) {
6169 6170
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("tunnelled migration requested but NULL stream passed"));
C
Chris Lalancette 已提交
6171 6172 6173
        goto cleanup;
    }

6174 6175 6176 6177 6178 6179 6180
    if (virLockManagerPluginUsesState(driver->lockManager)) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("Cannot use migrate v2 protocol with lock manager %s"),
                        virLockManagerPluginGetName(driver->lockManager));
        goto cleanup;
    }

6181 6182 6183
    ret = qemuMigrationPrepareTunnel(driver, dconn,
                                     NULL, 0, NULL, NULL, /* No cookies in v2 */
                                     st, dname, dom_xml);
6184

C
Chris Lalancette 已提交
6185
cleanup:
6186
    qemuDriverUnlock(driver);
C
Chris Lalancette 已提交
6187 6188 6189
    return ret;
}

D
Daniel Veillard 已提交
6190 6191 6192 6193
/* Prepare is the first step, and it runs on the destination host.
 *
 * This starts an empty VM listening on a TCP port.
 */
6194
static int ATTRIBUTE_NONNULL (5)
D
Daniel Veillard 已提交
6195
qemudDomainMigratePrepare2 (virConnectPtr dconn,
6196 6197
                            char **cookie ATTRIBUTE_UNUSED,
                            int *cookielen ATTRIBUTE_UNUSED,
D
Daniel Veillard 已提交
6198 6199
                            const char *uri_in,
                            char **uri_out,
C
Chris Lalancette 已提交
6200
                            unsigned long flags,
D
Daniel Veillard 已提交
6201 6202 6203 6204
                            const char *dname,
                            unsigned long resource ATTRIBUTE_UNUSED,
                            const char *dom_xml)
{
6205
    struct qemud_driver *driver = dconn->privateData;
6206
    int ret = -1;
6207

C
Chris Lalancette 已提交
6208 6209 6210 6211 6212 6213 6214 6215 6216
    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);

6217
    *uri_out = NULL;
D
Daniel Veillard 已提交
6218

6219
    qemuDriverLock(driver);
6220 6221 6222 6223 6224 6225 6226 6227

    if (virLockManagerPluginUsesState(driver->lockManager)) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("Cannot use migrate v2 protocol with lock manager %s"),
                        virLockManagerPluginGetName(driver->lockManager));
        goto cleanup;
    }

C
Chris Lalancette 已提交
6228 6229 6230 6231
    if (flags & VIR_MIGRATE_TUNNELLED) {
        /* this is a logical error; we never should have gotten here with
         * VIR_MIGRATE_TUNNELLED set
         */
6232 6233
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("Tunnelled migration requested but invalid RPC method called"));
C
Chris Lalancette 已提交
6234 6235 6236
        goto cleanup;
    }

D
Daniel Veillard 已提交
6237
    if (!dom_xml) {
6238 6239
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("no domain XML passed"));
6240
        goto cleanup;
D
Daniel Veillard 已提交
6241 6242
    }

6243 6244 6245 6246
    /* Do not use cookies in v2 protocol, since the cookie
     * length was not sufficiently large, causing failures
     * migrating between old & new libvirtd
     */
6247
    ret = qemuMigrationPrepareDirect(driver, dconn,
6248
                                     NULL, 0, NULL, NULL, /* No cookies */
6249 6250
                                     uri_in, uri_out,
                                     dname, dom_xml);
D
Daniel Veillard 已提交
6251

6252 6253 6254 6255
cleanup:
    qemuDriverUnlock(driver);
    return ret;
}
C
Chris Lalancette 已提交
6256

D
Daniel Veillard 已提交
6257

6258 6259 6260
/* Perform is the second step, and it runs on the source host. */
static int
qemudDomainMigratePerform (virDomainPtr dom,
6261 6262
                           const char *cookie,
                           int cookielen,
6263 6264 6265 6266 6267 6268 6269 6270
                           const char *uri,
                           unsigned long flags,
                           const char *dname,
                           unsigned long resource)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;
6271
    const char *dconnuri = NULL;
6272

6273 6274 6275 6276 6277 6278 6279 6280
    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 已提交
6281

6282
    qemuDriverLock(driver);
6283 6284 6285 6286 6287 6288 6289
    if (virLockManagerPluginUsesState(driver->lockManager)) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("Cannot use migrate v2 protocol with lock manager %s"),
                        virLockManagerPluginGetName(driver->lockManager));
        goto cleanup;
    }

6290
    vm = virDomainFindByUUID(&driver->domains, dom->uuid);
D
Daniel Veillard 已提交
6291
    if (!vm) {
6292 6293
        char uuidstr[VIR_UUID_STRING_BUFLEN];
        virUUIDFormat(dom->uuid, uuidstr);
6294 6295
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching uuid '%s'"), uuidstr);
6296
        goto cleanup;
D
Daniel Veillard 已提交
6297 6298
    }

6299 6300 6301 6302 6303
    if (flags & VIR_MIGRATE_PEER2PEER) {
        dconnuri = uri;
        uri = NULL;
    }

6304 6305 6306 6307 6308 6309
    /* Do not output cookies in v2 protocol, since the cookie
     * length was not sufficiently large, causing failures
     * migrating between old & new libvirtd.
     *
     * Consume any cookie we were able to decode though
     */
6310
    ret = qemuMigrationPerform(driver, dom->conn, vm,
6311
                               NULL, dconnuri, uri, cookie, cookielen,
6312
                               NULL, NULL, /* No output cookies in v2 */
6313
                               flags, dname, resource, false);
6314

6315
cleanup:
6316
    qemuDriverUnlock(driver);
6317
    return ret;
D
Daniel Veillard 已提交
6318 6319
}

6320

D
Daniel Veillard 已提交
6321 6322 6323 6324 6325 6326 6327
/* 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 已提交
6328
                           unsigned long flags,
D
Daniel Veillard 已提交
6329 6330
                           int retcode)
{
6331 6332 6333
    struct qemud_driver *driver = dconn->privateData;
    virDomainObjPtr vm;
    virDomainPtr dom = NULL;
D
Daniel Veillard 已提交
6334

C
Chris Lalancette 已提交
6335 6336 6337 6338 6339 6340 6341 6342 6343
    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);

6344
    qemuDriverLock(driver);
6345
    vm = virDomainFindByName(&driver->domains, dname);
D
Daniel Veillard 已提交
6346
    if (!vm) {
6347 6348
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching name '%s'"), dname);
6349
        goto cleanup;
D
Daniel Veillard 已提交
6350 6351
    }

6352 6353 6354 6355
    /* Do not use cookies in v2 protocol, since the cookie
     * length was not sufficiently large, causing failures
     * migrating between old & new libvirtd
     */
6356
    dom = qemuMigrationFinish(driver, dconn, vm,
6357
                              NULL, 0, NULL, NULL, /* No cookies */
6358
                              flags, retcode, false);
6359

6360
cleanup:
6361
    qemuDriverUnlock(driver);
6362
    return dom;
D
Daniel Veillard 已提交
6363 6364
}

6365

6366 6367 6368 6369 6370 6371
/*******************************************************************
 * Migration Protocol Version 3
 *******************************************************************/

static char *
qemuDomainMigrateBegin3(virDomainPtr domain,
6372
                        const char *xmlin,
6373 6374 6375 6376 6377 6378 6379 6380 6381 6382 6383 6384 6385 6386 6387 6388 6389 6390 6391 6392 6393 6394 6395 6396 6397 6398 6399 6400 6401
                        char **cookieout,
                        int *cookieoutlen,
                        unsigned long flags,
                        const char *dname ATTRIBUTE_UNUSED,
                        unsigned long resource ATTRIBUTE_UNUSED)
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm;
    char *xml = NULL;

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

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

6402
    xml = qemuMigrationBegin(driver, vm, xmlin,
6403 6404 6405 6406 6407 6408 6409 6410 6411 6412 6413 6414 6415 6416 6417 6418 6419 6420 6421 6422 6423 6424 6425 6426 6427 6428 6429 6430 6431 6432 6433 6434 6435 6436 6437 6438 6439 6440 6441 6442 6443 6444 6445 6446 6447 6448 6449 6450 6451 6452 6453 6454 6455 6456 6457 6458 6459 6460 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
                             cookieout, cookieoutlen);

cleanup:
    qemuDriverUnlock(driver);
    return xml;
}

static int
qemuDomainMigratePrepare3(virConnectPtr dconn,
                          const char *cookiein,
                          int cookieinlen,
                          char **cookieout,
                          int *cookieoutlen,
                          const char *uri_in,
                          char **uri_out,
                          unsigned long flags,
                          const char *dname,
                          unsigned long resource ATTRIBUTE_UNUSED,
                          const char *dom_xml)
{
    struct qemud_driver *driver = dconn->privateData;
    int ret = -1;

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

    *uri_out = NULL;

    qemuDriverLock(driver);
    if (flags & VIR_MIGRATE_TUNNELLED) {
        /* this is a logical error; we never should have gotten here with
         * VIR_MIGRATE_TUNNELLED set
         */
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("Tunnelled migration requested but invalid RPC method called"));
        goto cleanup;
    }

    if (!dom_xml) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("no domain XML passed"));
        goto cleanup;
    }

    ret = qemuMigrationPrepareDirect(driver, dconn,
                                     cookiein, cookieinlen,
                                     cookieout, cookieoutlen,
                                     uri_in, uri_out,
                                     dname, dom_xml);

cleanup:
    qemuDriverUnlock(driver);
    return ret;
}


static int
qemuDomainMigratePrepareTunnel3(virConnectPtr dconn,
                                virStreamPtr st,
                                const char *cookiein,
                                int cookieinlen,
                                char **cookieout,
                                int *cookieoutlen,
                                unsigned long flags,
                                const char *dname,
                                unsigned long resource ATTRIBUTE_UNUSED,
                                const char *dom_xml)
{
    struct qemud_driver *driver = dconn->privateData;
    int ret = -1;

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

    if (!dom_xml) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("no domain XML passed"));
        goto cleanup;
    }
    if (!(flags & VIR_MIGRATE_TUNNELLED)) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                         "%s", _("PrepareTunnel called but no TUNNELLED flag set"));
        goto cleanup;
    }
    if (st == NULL) {
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        "%s", _("tunnelled migration requested but NULL stream passed"));
        goto cleanup;
    }

    qemuDriverLock(driver);
    ret = qemuMigrationPrepareTunnel(driver, dconn,
                                     cookiein, cookieinlen,
                                     cookieout, cookieoutlen,
                                     st, dname, dom_xml);
    qemuDriverUnlock(driver);

cleanup:
    return ret;
}


static int
qemuDomainMigratePerform3(virDomainPtr dom,
6519
                          const char *xmlin,
6520 6521 6522 6523
                          const char *cookiein,
                          int cookieinlen,
                          char **cookieout,
                          int *cookieoutlen,
6524
                          const char *dconnuri,
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
                          const char *uri,
                          unsigned long flags,
                          const char *dname,
                          unsigned long resource)
{
    struct qemud_driver *driver = dom->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;

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

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

6553
    ret = qemuMigrationPerform(driver, dom->conn, vm, xmlin,
6554
                               dconnuri, uri, cookiein, cookieinlen,
6555
                               cookieout, cookieoutlen,
6556
                               flags, dname, resource, true);
6557 6558 6559 6560 6561 6562 6563

cleanup:
    qemuDriverUnlock(driver);
    return ret;
}


6564
static virDomainPtr
6565 6566 6567 6568 6569 6570
qemuDomainMigrateFinish3(virConnectPtr dconn,
                         const char *dname,
                         const char *cookiein,
                         int cookieinlen,
                         char **cookieout,
                         int *cookieoutlen,
6571
                         const char *dconnuri ATTRIBUTE_UNUSED,
6572 6573
                         const char *uri ATTRIBUTE_UNUSED,
                         unsigned long flags,
6574
                         int cancelled)
6575 6576 6577
{
    struct qemud_driver *driver = dconn->privateData;
    virDomainObjPtr vm;
6578
    virDomainPtr dom = NULL;
6579 6580 6581 6582 6583 6584 6585 6586

    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 |
6587
                  VIR_MIGRATE_NON_SHARED_INC, NULL);
6588 6589 6590 6591 6592 6593 6594 6595 6596

    qemuDriverLock(driver);
    vm = virDomainFindByName(&driver->domains, dname);
    if (!vm) {
        qemuReportError(VIR_ERR_NO_DOMAIN,
                        _("no domain with matching name '%s'"), dname);
        goto cleanup;
    }

6597 6598 6599 6600
    dom = qemuMigrationFinish(driver, dconn, vm,
                              cookiein, cookieinlen,
                              cookieout, cookieoutlen,
                              flags, cancelled, true);
6601 6602 6603

cleanup:
    qemuDriverUnlock(driver);
6604
    return dom;
6605 6606 6607 6608 6609 6610 6611 6612 6613 6614 6615 6616 6617 6618 6619 6620 6621 6622 6623 6624 6625 6626 6627 6628 6629 6630 6631 6632 6633 6634 6635 6636
}

static int
qemuDomainMigrateConfirm3(virDomainPtr domain,
                          const char *cookiein,
                          int cookieinlen,
                          unsigned long flags,
                          int cancelled)
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;

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

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

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

6640 6641
    ret = qemuMigrationConfirm(driver, domain->conn, vm,
                               cookiein, cookieinlen,
6642 6643 6644 6645 6646 6647 6648 6649 6650 6651 6652 6653
                               flags, cancelled);

    if (qemuDomainObjEndJob(vm) == 0) {
        vm = NULL;
    } else if (!virDomainObjIsActive(vm) &&
               (!vm->persistent || (flags & VIR_MIGRATE_UNDEFINE_SOURCE))) {
        if (flags & VIR_MIGRATE_UNDEFINE_SOURCE)
            virDomainDeleteConfig(driver->configDir, driver->autostartDir, vm);
        virDomainRemoveInactive(&driver->domains, vm);
        vm = NULL;
    }

6654
cleanup:
6655 6656
    if (vm)
        virDomainObjUnlock(vm);
6657 6658 6659 6660 6661
    qemuDriverUnlock(driver);
    return ret;
}


6662 6663 6664 6665 6666 6667 6668 6669 6670 6671 6672 6673 6674 6675 6676 6677
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;

6678
    def = virNodeDeviceDefParseString(xml, EXISTING_DEVICE);
6679 6680 6681 6682 6683 6684 6685 6686 6687 6688 6689 6690 6691 6692 6693 6694 6695
    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) {
6696 6697
        qemuReportError(VIR_ERR_INVALID_ARG,
                        _("device %s is not a PCI device"), dev->name);
6698 6699 6700 6701 6702 6703 6704 6705 6706 6707 6708 6709 6710
        goto out;
    }

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

static int
qemudNodeDeviceDettach (virNodeDevicePtr dev)
{
6711
    struct qemud_driver *driver = dev->conn->privateData;
6712 6713 6714 6715 6716 6717 6718
    pciDevice *pci;
    unsigned domain, bus, slot, function;
    int ret = -1;

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

6719
    pci = pciGetDevice(domain, bus, slot, function);
6720 6721 6722
    if (!pci)
        return -1;

6723 6724
    qemuDriverLock(driver);
    if (pciDettachDevice(pci, driver->activePciHostdevs) < 0)
6725 6726 6727 6728
        goto out;

    ret = 0;
out:
6729
    qemuDriverUnlock(driver);
6730
    pciFreeDevice(pci);
6731 6732 6733 6734 6735 6736
    return ret;
}

static int
qemudNodeDeviceReAttach (virNodeDevicePtr dev)
{
6737
    struct qemud_driver *driver = dev->conn->privateData;
6738 6739 6740 6741 6742 6743 6744
    pciDevice *pci;
    unsigned domain, bus, slot, function;
    int ret = -1;

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

6745
    pci = pciGetDevice(domain, bus, slot, function);
6746 6747 6748
    if (!pci)
        return -1;

6749 6750
    qemuDriverLock(driver);
    if (pciReAttachDevice(pci, driver->activePciHostdevs) < 0)
6751 6752 6753 6754
        goto out;

    ret = 0;
out:
6755
    qemuDriverUnlock(driver);
6756
    pciFreeDevice(pci);
6757 6758 6759 6760 6761 6762
    return ret;
}

static int
qemudNodeDeviceReset (virNodeDevicePtr dev)
{
6763
    struct qemud_driver *driver = dev->conn->privateData;
6764 6765 6766 6767 6768 6769 6770
    pciDevice *pci;
    unsigned domain, bus, slot, function;
    int ret = -1;

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

6771
    pci = pciGetDevice(domain, bus, slot, function);
6772 6773 6774
    if (!pci)
        return -1;

6775 6776
    qemuDriverLock(driver);

6777
    if (pciResetDevice(pci, driver->activePciHostdevs, NULL) < 0)
6778 6779 6780 6781
        goto out;

    ret = 0;
out:
6782
    qemuDriverUnlock(driver);
6783
    pciFreeDevice(pci);
6784 6785 6786
    return ret;
}

6787 6788 6789 6790 6791 6792 6793 6794 6795 6796 6797
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) {
6798 6799
        qemuReportError(VIR_ERR_NO_SUPPORT,
                        "%s", _("cannot get host CPU capabilities"));
6800 6801
    }
    else
6802
        ret = cpuCompareXML(driver->caps->host.cpu, xmlDesc);
6803 6804 6805 6806 6807 6808

    qemuDriverUnlock(driver);

    return ret;
}

6809

6810 6811 6812 6813 6814 6815 6816 6817 6818 6819 6820 6821 6822
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;
}

6823 6824 6825 6826 6827 6828 6829 6830 6831 6832 6833 6834 6835 6836 6837 6838 6839 6840 6841 6842 6843 6844 6845

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) {
6846 6847
            struct timeval now;

6848
            memcpy(info, &priv->jobInfo, sizeof(*info));
6849 6850 6851 6852 6853 6854 6855 6856 6857 6858 6859

            /* 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;
            }
6860
            info->timeElapsed = timeval_to_ms(now) - priv->jobStart;
6861 6862 6863 6864 6865 6866 6867 6868 6869 6870 6871 6872 6873 6874 6875 6876 6877 6878 6879
        } 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;
}


6880 6881 6882 6883 6884 6885 6886 6887 6888 6889 6890 6891 6892 6893 6894 6895 6896 6897 6898 6899 6900 6901
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);
6902
            priv->jobSignals |= QEMU_JOB_SIGNAL_CANCEL;
6903 6904 6905 6906 6907 6908 6909 6910 6911 6912 6913 6914 6915 6916 6917 6918 6919 6920 6921 6922
        } 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;
}


6923 6924 6925 6926 6927 6928 6929 6930 6931 6932
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;

6933
    virCheckFlags(0, -1);
6934 6935 6936 6937 6938 6939 6940 6941 6942 6943 6944 6945 6946 6947 6948 6949 6950 6951 6952 6953

    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;

6954
    if (priv->jobActive != QEMU_JOB_MIGRATION_OUT) {
6955 6956 6957 6958 6959 6960 6961
        qemuReportError(VIR_ERR_OPERATION_INVALID,
                        "%s", _("domain is not being migrated"));
        goto cleanup;
    }

    VIR_DEBUG("Requesting migration downtime change to %llums", downtime);
    priv->jobSignalsData.migrateDowntime = downtime;
6962
    priv->jobSignals |= QEMU_JOB_SIGNAL_MIGRATE_DOWNTIME;
6963 6964 6965 6966 6967 6968 6969 6970 6971
    ret = 0;

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

6972 6973 6974 6975 6976 6977 6978 6979 6980 6981 6982 6983 6984 6985 6986 6987 6988 6989 6990 6991 6992 6993 6994 6995 6996 6997 6998 6999 7000 7001 7002 7003 7004 7005 7006 7007 7008 7009 7010
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->jobSignalsData.migrateBandwidth = bandwidth;
7011
    priv->jobSignals |= QEMU_JOB_SIGNAL_MIGRATE_SPEED;
7012 7013 7014 7015 7016 7017 7018 7019 7020
    ret = 0;

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

C
Chris Lalancette 已提交
7021 7022 7023 7024 7025 7026 7027 7028 7029 7030 7031 7032 7033 7034
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;
}

7035 7036 7037
static int qemuDomainSnapshotWriteMetadata(virDomainObjPtr vm,
                                           virDomainSnapshotObjPtr snapshot,
                                           char *snapshotDir)
C
Chris Lalancette 已提交
7038 7039 7040 7041 7042 7043 7044 7045 7046 7047
{
    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);
7048
    newxml = virDomainSnapshotDefFormat(uuidstr, snapshot->def, 1);
C
Chris Lalancette 已提交
7049 7050 7051 7052 7053 7054 7055 7056 7057 7058 7059 7060 7061 7062 7063 7064
    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;
    }

7065
    if (virAsprintf(&snapFile, "%s/%s.xml", snapDir, snapshot->def->name) < 0) {
C
Chris Lalancette 已提交
7066 7067 7068 7069 7070 7071 7072 7073 7074 7075 7076 7077 7078 7079 7080 7081 7082 7083 7084 7085 7086
        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);
7087
    VIR_FORCE_CLOSE(fd);
C
Chris Lalancette 已提交
7088 7089 7090
    return ret;
}

7091 7092
static int qemuDomainSnapshotSetCurrentActive(virDomainObjPtr vm,
                                              char *snapshotDir)
C
Chris Lalancette 已提交
7093 7094 7095 7096
{
    if (vm->current_snapshot) {
        vm->current_snapshot->def->active = 1;

7097 7098
        return qemuDomainSnapshotWriteMetadata(vm, vm->current_snapshot,
                                               snapshotDir);
C
Chris Lalancette 已提交
7099 7100 7101 7102 7103
    }

    return 0;
}

7104 7105
static int qemuDomainSnapshotSetCurrentInactive(virDomainObjPtr vm,
                                                char *snapshotDir)
C
Chris Lalancette 已提交
7106 7107 7108 7109
{
    if (vm->current_snapshot) {
        vm->current_snapshot->def->active = 0;

7110 7111
        return qemuDomainSnapshotWriteMetadata(vm, vm->current_snapshot,
                                               snapshotDir);
C
Chris Lalancette 已提交
7112 7113 7114 7115 7116 7117 7118 7119 7120 7121 7122 7123 7124 7125 7126 7127 7128 7129 7130
    }

    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,
7131 7132
                            _("Disk '%s' does not support snapshotting"),
                            vm->def->disks[i]->src);
C
Chris Lalancette 已提交
7133 7134 7135 7136 7137 7138 7139
            return 0;
        }
    }

    return 1;
}

7140 7141 7142 7143 7144 7145 7146 7147 7148 7149 7150 7151 7152 7153 7154 7155 7156 7157 7158 7159 7160 7161 7162 7163 7164 7165 7166 7167 7168 7169 7170 7171 7172 7173 7174 7175 7176 7177 7178 7179 7180 7181 7182 7183 7184 7185 7186 7187
/* 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
7188 7189
qemuDomainSnapshotCreateActive(virConnectPtr conn,
                               struct qemud_driver *driver,
7190 7191 7192 7193 7194
                               virDomainObjPtr *vmptr,
                               virDomainSnapshotObjPtr snap)
{
    virDomainObjPtr vm = *vmptr;
    qemuDomainObjPrivatePtr priv = vm->privateData;
7195 7196
    bool resume = false;
    int ret = -1;
7197 7198 7199 7200

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

J
Jiri Denemark 已提交
7201
    if (virDomainObjGetState(vm, NULL) == VIR_DOMAIN_RUNNING) {
7202 7203 7204 7205
        /* 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.
         */
J
Jiri Denemark 已提交
7206
        if (qemuProcessStopCPUs(driver, vm, VIR_DOMAIN_PAUSED_SAVE) < 0)
7207 7208 7209 7210 7211 7212 7213 7214 7215 7216
            goto cleanup;

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

7217 7218 7219 7220
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
    ret = qemuMonitorCreateSnapshot(priv->mon, snap->def->name);
    qemuDomainObjExitMonitorWithDriver(driver, vm);

7221 7222
cleanup:
    if (resume && virDomainObjIsActive(vm) &&
J
Jiri Denemark 已提交
7223 7224
        qemuProcessStartCPUs(driver, vm, conn,
                             VIR_DOMAIN_RUNNING_UNPAUSED) < 0 &&
7225 7226 7227 7228 7229
        virGetLastError() == NULL) {
        qemuReportError(VIR_ERR_OPERATION_FAILED, "%s",
                        _("resuming after snapshot failed"));
    }

7230 7231 7232 7233 7234 7235
    if (qemuDomainObjEndJob(vm) == 0)
        *vmptr = NULL;

    return ret;
}

C
Chris Lalancette 已提交
7236 7237
static virDomainSnapshotPtr qemuDomainSnapshotCreateXML(virDomainPtr domain,
                                                        const char *xmlDesc,
7238
                                                        unsigned int flags)
C
Chris Lalancette 已提交
7239 7240 7241 7242 7243 7244 7245 7246
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    virDomainSnapshotObjPtr snap = NULL;
    virDomainSnapshotPtr snapshot = NULL;
    char uuidstr[VIR_UUID_STRING_BUFLEN];
    virDomainSnapshotDefPtr def;

7247 7248
    virCheckFlags(0, NULL);

C
Chris Lalancette 已提交
7249 7250 7251 7252 7253 7254 7255 7256 7257 7258 7259 7260 7261 7262 7263 7264 7265 7266 7267 7268 7269 7270 7271 7272 7273
    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;

J
Jiri Denemark 已提交
7274
    snap->def->state = virDomainObjGetState(vm, NULL);
7275

C
Chris Lalancette 已提交
7276 7277
    /* actually do the snapshot */
    if (!virDomainObjIsActive(vm)) {
7278
        if (qemuDomainSnapshotCreateInactive(vm, snap) < 0)
C
Chris Lalancette 已提交
7279 7280 7281
            goto cleanup;
    }
    else {
7282 7283
        if (qemuDomainSnapshotCreateActive(domain->conn, driver,
                                           &vm, snap) < 0)
7284
            goto cleanup;
C
Chris Lalancette 已提交
7285 7286 7287 7288 7289 7290 7291 7292 7293 7294 7295 7296 7297 7298 7299 7300 7301 7302
    }

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

7303 7304 7305
    if (qemuDomainSnapshotWriteMetadata(vm, vm->current_snapshot,
                                        driver->snapshotDir) < 0)
        /* qemuDomainSnapshotWriteMetadata set the error */
C
Chris Lalancette 已提交
7306 7307 7308 7309 7310 7311 7312 7313 7314 7315 7316 7317 7318
        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,
7319
                                       unsigned int flags)
C
Chris Lalancette 已提交
7320 7321 7322 7323 7324
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    int n = -1;

7325 7326
    virCheckFlags(0, -1);

C
Chris Lalancette 已提交
7327 7328 7329 7330 7331 7332 7333 7334 7335 7336 7337 7338 7339 7340 7341 7342 7343 7344 7345 7346
    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,
7347
                                 unsigned int flags)
C
Chris Lalancette 已提交
7348 7349 7350 7351 7352
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    int n = -1;

7353 7354
    virCheckFlags(0, -1);

C
Chris Lalancette 已提交
7355 7356 7357 7358 7359 7360 7361 7362 7363 7364 7365 7366 7367 7368 7369 7370 7371 7372 7373 7374 7375
    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,
7376
                                                           unsigned int flags)
C
Chris Lalancette 已提交
7377 7378 7379 7380 7381 7382
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm;
    virDomainSnapshotObjPtr snap = NULL;
    virDomainSnapshotPtr snapshot = NULL;

7383 7384
    virCheckFlags(0, NULL);

C
Chris Lalancette 已提交
7385 7386 7387 7388 7389 7390 7391 7392 7393 7394 7395 7396 7397 7398 7399 7400 7401 7402 7403 7404 7405 7406 7407 7408 7409 7410 7411
    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,
7412
                                        unsigned int flags)
C
Chris Lalancette 已提交
7413 7414 7415 7416 7417
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm;
    int ret = -1;

7418 7419
    virCheckFlags(0, -1);

C
Chris Lalancette 已提交
7420 7421 7422 7423 7424 7425 7426 7427 7428 7429 7430 7431 7432 7433 7434 7435 7436 7437 7438 7439
    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,
7440
                                                      unsigned int flags)
C
Chris Lalancette 已提交
7441 7442 7443 7444 7445
{
    struct qemud_driver *driver = domain->conn->privateData;
    virDomainObjPtr vm;
    virDomainSnapshotPtr snapshot = NULL;

7446 7447
    virCheckFlags(0, NULL);

C
Chris Lalancette 已提交
7448 7449 7450 7451 7452 7453 7454 7455 7456 7457 7458 7459 7460 7461 7462 7463 7464 7465 7466 7467 7468 7469 7470 7471 7472
    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;
}

7473 7474
static char *qemuDomainSnapshotGetXMLDesc(virDomainSnapshotPtr snapshot,
                                          unsigned int flags)
C
Chris Lalancette 已提交
7475 7476 7477 7478 7479 7480 7481
{
    struct qemud_driver *driver = snapshot->domain->conn->privateData;
    virDomainObjPtr vm = NULL;
    char *xml = NULL;
    virDomainSnapshotObjPtr snap = NULL;
    char uuidstr[VIR_UUID_STRING_BUFLEN];

7482 7483
    virCheckFlags(0, NULL);

C
Chris Lalancette 已提交
7484 7485 7486 7487 7488 7489 7490 7491 7492 7493 7494 7495 7496 7497 7498 7499 7500 7501 7502 7503 7504 7505 7506 7507 7508 7509 7510
    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,
7511
                                      unsigned int flags)
C
Chris Lalancette 已提交
7512 7513 7514 7515 7516 7517 7518 7519 7520 7521
{
    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;

7522 7523
    virCheckFlags(0, -1);

C
Chris Lalancette 已提交
7524 7525 7526 7527 7528 7529 7530 7531 7532 7533 7534 7535 7536 7537 7538 7539 7540 7541 7542 7543 7544 7545 7546 7547 7548 7549 7550 7551 7552 7553 7554
    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)
7555
                goto endjob;
C
Chris Lalancette 已提交
7556 7557
        }
        else {
7558
            if (qemuDomainSnapshotSetCurrentActive(vm, driver->snapshotDir) < 0)
7559
                goto endjob;
C
Chris Lalancette 已提交
7560

7561 7562
            rc = qemuProcessStart(snapshot->domain->conn, driver, vm, NULL,
                                  false, -1, NULL, VIR_VM_OP_CREATE);
7563
            qemuAuditDomainStart(vm, "from-snapshot", rc >= 0);
7564
            if (qemuDomainSnapshotSetCurrentInactive(vm, driver->snapshotDir) < 0)
7565
                goto endjob;
C
Chris Lalancette 已提交
7566
            if (rc < 0)
7567
                goto endjob;
C
Chris Lalancette 已提交
7568 7569 7570 7571 7572 7573
        }

        if (snap->def->state == VIR_DOMAIN_PAUSED) {
            /* qemu unconditionally starts the domain running again after
             * loadvm, so let's pause it to keep consistency
             */
J
Jiri Denemark 已提交
7574 7575
            rc = qemuProcessStopCPUs(driver, vm,
                                     VIR_DOMAIN_PAUSED_FROM_SNAPSHOT);
H
Hu Tao 已提交
7576
            if (rc < 0)
7577
                goto endjob;
J
Jiri Denemark 已提交
7578 7579 7580
        } else {
            virDomainObjSetState(vm, VIR_DOMAIN_RUNNING,
                                 VIR_DOMAIN_RUNNING_FROM_SNAPSHOT);
C
Chris Lalancette 已提交
7581 7582 7583 7584 7585 7586 7587 7588 7589 7590 7591 7592 7593 7594 7595 7596 7597 7598
        }

        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)) {
J
Jiri Denemark 已提交
7599
            qemuProcessStop(driver, vm, 0, VIR_DOMAIN_SHUTOFF_FROM_SNAPSHOT);
7600
            qemuAuditDomainStop(vm, "from-snapshot");
C
Chris Lalancette 已提交
7601 7602 7603
            event = virDomainEventNewFromObj(vm,
                                             VIR_DOMAIN_EVENT_STOPPED,
                                             VIR_DOMAIN_EVENT_STOPPED_FROM_SNAPSHOT);
7604 7605 7606 7607
            if (!vm->persistent) {
                if (qemuDomainObjEndJob(vm) > 0)
                    virDomainRemoveInactive(&driver->domains, vm);
                vm = NULL;
7608
                goto cleanup;
7609
            }
C
Chris Lalancette 已提交
7610 7611
        }

7612
        if (qemuDomainSnapshotSetCurrentActive(vm, driver->snapshotDir) < 0)
7613
            goto endjob;
C
Chris Lalancette 已提交
7614 7615 7616 7617
    }

    ret = 0;

7618
endjob:
C
Chris Lalancette 已提交
7619 7620 7621
    if (vm && qemuDomainObjEndJob(vm) == 0)
        vm = NULL;

7622
cleanup:
C
Chris Lalancette 已提交
7623 7624 7625 7626 7627 7628 7629 7630 7631 7632 7633 7634 7635 7636 7637 7638 7639 7640 7641 7642 7643 7644 7645 7646 7647 7648 7649 7650 7651 7652 7653 7654 7655 7656 7657 7658 7659 7660 7661 7662 7663 7664 7665 7666 7667 7668 7669 7670 7671 7672 7673 7674 7675 7676 7677 7678 7679 7680 7681 7682 7683 7684 7685 7686 7687 7688 7689 7690 7691 7692 7693 7694 7695 7696 7697 7698 7699 7700 7701 7702 7703 7704 7705 7706 7707 7708 7709 7710 7711 7712 7713 7714 7715 7716 7717 7718 7719 7720 7721 7722 7723 7724
    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,
7725
                                              const void *name ATTRIBUTE_UNUSED,
C
Chris Lalancette 已提交
7726 7727 7728 7729 7730 7731 7732 7733 7734 7735 7736 7737 7738 7739 7740 7741 7742 7743 7744 7745 7746
                                              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);
    }
}

7747 7748 7749 7750 7751 7752 7753 7754 7755
struct snap_reparent {
    struct qemud_driver *driver;
    virDomainSnapshotObjPtr snap;
    virDomainObjPtr vm;
    int err;
};

static void
qemuDomainSnapshotReparentChildren(void *payload,
7756
                                   const void *name ATTRIBUTE_UNUSED,
7757 7758 7759 7760 7761 7762 7763 7764 7765 7766 7767 7768 7769 7770 7771 7772 7773 7774 7775 7776 7777 7778 7779 7780 7781 7782 7783
                                   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 已提交
7784 7785 7786 7787 7788 7789 7790 7791 7792
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;
7793
    struct snap_reparent rep;
C
Chris Lalancette 已提交
7794

7795 7796
    virCheckFlags(VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN, -1);

C
Chris Lalancette 已提交
7797 7798 7799 7800 7801 7802 7803 7804 7805 7806 7807 7808 7809 7810 7811 7812 7813
    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;
    }

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

C
Chris Lalancette 已提交
7817 7818 7819 7820 7821 7822 7823 7824
    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)
7825
            goto endjob;
7826 7827 7828 7829 7830 7831 7832 7833 7834
    } 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 已提交
7835 7836 7837 7838
    }

    ret = qemuDomainSnapshotDiscard(driver, vm, snap);

7839 7840 7841 7842
endjob:
    if (qemuDomainObjEndJob(vm) == 0)
        vm = NULL;

C
Chris Lalancette 已提交
7843 7844 7845 7846 7847 7848
cleanup:
    if (vm)
        virDomainObjUnlock(vm);
    qemuDriverUnlock(driver);
    return ret;
}
7849

7850 7851 7852 7853 7854 7855 7856
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;
7857
    bool hmp;
7858

7859
    virCheckFlags(VIR_DOMAIN_QEMU_MONITOR_COMMAND_HMP, -1);
7860 7861 7862 7863 7864 7865 7866 7867 7868 7869 7870 7871 7872 7873 7874 7875 7876 7877 7878

    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;

7879
    qemuDomainObjTaint(driver, vm, VIR_DOMAIN_TAINT_CUSTOM_MONITOR, -1);
7880

7881 7882
    hmp = !!(flags & VIR_DOMAIN_QEMU_MONITOR_COMMAND_HMP);

7883 7884 7885
    if (qemuDomainObjBeginJobWithDriver(driver, vm) < 0)
        goto cleanup;
    qemuDomainObjEnterMonitorWithDriver(driver, vm);
7886
    ret = qemuMonitorArbitraryCommand(priv->mon, cmd, result, hmp);
7887 7888 7889 7890 7891 7892 7893 7894 7895 7896 7897 7898 7899
    qemuDomainObjExitMonitorWithDriver(driver, vm);
    if (qemuDomainObjEndJob(vm) == 0) {
        vm = NULL;
        goto cleanup;
    }

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

7900 7901 7902 7903 7904 7905 7906 7907 7908 7909 7910 7911 7912 7913 7914 7915 7916 7917 7918 7919 7920 7921 7922 7923 7924 7925 7926 7927 7928 7929 7930 7931 7932 7933 7934 7935 7936 7937 7938 7939 7940 7941 7942 7943 7944 7945 7946 7947 7948 7949 7950 7951 7952 7953 7954 7955 7956

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

7957
    if (chr->source.type != VIR_DOMAIN_CHR_TYPE_PTY) {
7958 7959 7960 7961 7962 7963
        qemuReportError(VIR_ERR_INTERNAL_ERROR,
                        _("character device %s is not using a PTY"),
                        NULLSTR(devname));
        goto cleanup;
    }

7964 7965
    if (virFDStreamOpenFile(st, chr->source.data.file.path,
                            0, 0, O_RDWR, false) < 0)
7966 7967 7968 7969 7970 7971 7972 7973 7974 7975 7976
        goto cleanup;

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


7977
static virDriver qemuDriver = {
7978 7979
    .no = VIR_DRV_QEMU,
    .name = "QEMU",
7980 7981 7982 7983 7984 7985 7986 7987 7988 7989 7990 7991 7992 7993 7994 7995 7996 7997 7998 7999 8000 8001 8002 8003 8004 8005 8006 8007 8008 8009 8010 8011 8012 8013 8014 8015 8016 8017 8018 8019 8020 8021 8022 8023 8024 8025 8026 8027 8028 8029 8030 8031 8032 8033 8034 8035 8036 8037 8038 8039 8040
    .open = qemudOpen, /* 0.2.0 */
    .close = qemudClose, /* 0.2.0 */
    .supports_feature = qemudSupportsFeature, /* 0.5.0 */
    .type = qemudGetType, /* 0.2.0 */
    .version = qemudGetVersion, /* 0.2.0 */
    .getHostname = virGetHostname, /* 0.3.3 */
    .getSysinfo = qemuGetSysinfo, /* 0.8.8 */
    .getMaxVcpus = qemudGetMaxVCPUs, /* 0.2.1 */
    .nodeGetInfo = nodeGetInfo, /* 0.2.0 */
    .getCapabilities = qemudGetCapabilities, /* 0.2.1 */
    .listDomains = qemudListDomains, /* 0.2.0 */
    .numOfDomains = qemudNumDomains, /* 0.2.0 */
    .domainCreateXML = qemudDomainCreate, /* 0.2.0 */
    .domainLookupByID = qemudDomainLookupByID, /* 0.2.0 */
    .domainLookupByUUID = qemudDomainLookupByUUID, /* 0.2.0 */
    .domainLookupByName = qemudDomainLookupByName, /* 0.2.0 */
    .domainSuspend = qemudDomainSuspend, /* 0.2.0 */
    .domainResume = qemudDomainResume, /* 0.2.0 */
    .domainShutdown = qemudDomainShutdown, /* 0.2.0 */
    .domainDestroy = qemudDomainDestroy, /* 0.2.0 */
    .domainGetOSType = qemudDomainGetOSType, /* 0.2.2 */
    .domainGetMaxMemory = qemudDomainGetMaxMemory, /* 0.4.2 */
    .domainSetMaxMemory = qemudDomainSetMaxMemory, /* 0.4.2 */
    .domainSetMemory = qemudDomainSetMemory, /* 0.4.2 */
    .domainSetMemoryFlags = qemudDomainSetMemoryFlags, /* 0.9.0 */
    .domainSetMemoryParameters = qemuDomainSetMemoryParameters, /* 0.8.5 */
    .domainGetMemoryParameters = qemuDomainGetMemoryParameters, /* 0.8.5 */
    .domainSetBlkioParameters = qemuDomainSetBlkioParameters, /* 0.9.0 */
    .domainGetBlkioParameters = qemuDomainGetBlkioParameters, /* 0.9.0 */
    .domainGetInfo = qemudDomainGetInfo, /* 0.2.0 */
    .domainGetState = qemuDomainGetState, /* 0.9.2 */
    .domainSave = qemudDomainSave, /* 0.2.0 */
    .domainRestore = qemuDomainRestore, /* 0.2.0 */
    .domainCoreDump = qemudDomainCoreDump, /* 0.7.0 */
    .domainScreenshot = qemuDomainScreenshot, /* 0.9.2 */
    .domainSetVcpus = qemudDomainSetVcpus, /* 0.4.4 */
    .domainSetVcpusFlags = qemudDomainSetVcpusFlags, /* 0.8.5 */
    .domainGetVcpusFlags = qemudDomainGetVcpusFlags, /* 0.8.5 */
    .domainPinVcpu = qemudDomainPinVcpu, /* 0.4.4 */
    .domainGetVcpus = qemudDomainGetVcpus, /* 0.4.4 */
    .domainGetMaxVcpus = qemudDomainGetMaxVcpus, /* 0.4.4 */
    .domainGetSecurityLabel = qemudDomainGetSecurityLabel, /* 0.6.1 */
    .nodeGetSecurityModel = qemudNodeGetSecurityModel, /* 0.6.1 */
    .domainGetXMLDesc = qemuDomainGetXMLDesc, /* 0.2.0 */
    .domainXMLFromNative = qemuDomainXMLFromNative, /* 0.6.4 */
    .domainXMLToNative = qemuDomainXMLToNative, /* 0.6.4 */
    .listDefinedDomains = qemudListDefinedDomains, /* 0.2.0 */
    .numOfDefinedDomains = qemudNumDefinedDomains, /* 0.2.0 */
    .domainCreate = qemudDomainStart, /* 0.2.0 */
    .domainCreateWithFlags = qemudDomainStartWithFlags, /* 0.8.2 */
    .domainDefineXML = qemudDomainDefine, /* 0.2.0 */
    .domainUndefine = qemudDomainUndefine, /* 0.2.0 */
    .domainAttachDevice = qemuDomainAttachDevice, /* 0.4.1 */
    .domainAttachDeviceFlags = qemuDomainAttachDeviceFlags, /* 0.7.7 */
    .domainDetachDevice = qemuDomainDetachDevice, /* 0.5.0 */
    .domainDetachDeviceFlags = qemuDomainDetachDeviceFlags, /* 0.7.7 */
    .domainUpdateDeviceFlags = qemuDomainUpdateDeviceFlags, /* 0.8.0 */
    .domainGetAutostart = qemudDomainGetAutostart, /* 0.2.1 */
    .domainSetAutostart = qemudDomainSetAutostart, /* 0.2.1 */
    .domainGetSchedulerType = qemuGetSchedulerType, /* 0.7.0 */
    .domainGetSchedulerParameters = qemuGetSchedulerParameters, /* 0.7.0 */
8041
    .domainGetSchedulerParametersFlags = qemuGetSchedulerParametersFlags, /* 0.9.2 */
8042
    .domainSetSchedulerParameters = qemuSetSchedulerParameters, /* 0.7.0 */
8043
    .domainSetSchedulerParametersFlags = qemuSetSchedulerParametersFlags, /* 0.9.2 */
8044 8045 8046 8047 8048 8049 8050 8051 8052 8053 8054 8055 8056 8057 8058 8059 8060 8061 8062 8063 8064 8065 8066 8067 8068 8069 8070 8071 8072 8073 8074 8075 8076 8077 8078 8079 8080 8081 8082 8083 8084 8085 8086 8087 8088
    .domainMigratePerform = qemudDomainMigratePerform, /* 0.5.0 */
    .domainBlockStats = qemudDomainBlockStats, /* 0.4.1 */
    .domainInterfaceStats = qemudDomainInterfaceStats, /* 0.4.1 */
    .domainMemoryStats = qemudDomainMemoryStats, /* 0.7.5 */
    .domainBlockPeek = qemudDomainBlockPeek, /* 0.4.4 */
    .domainMemoryPeek = qemudDomainMemoryPeek, /* 0.4.4 */
    .domainGetBlockInfo = qemuDomainGetBlockInfo, /* 0.8.1 */
    .nodeGetCellsFreeMemory = nodeGetCellsFreeMemory, /* 0.4.4 */
    .nodeGetFreeMemory = nodeGetFreeMemory, /* 0.4.4 */
    .domainEventRegister = qemuDomainEventRegister, /* 0.5.0 */
    .domainEventDeregister = qemuDomainEventDeregister, /* 0.5.0 */
    .domainMigratePrepare2 = qemudDomainMigratePrepare2, /* 0.5.0 */
    .domainMigrateFinish2 = qemudDomainMigrateFinish2, /* 0.5.0 */
    .nodeDeviceDettach = qemudNodeDeviceDettach, /* 0.6.1 */
    .nodeDeviceReAttach = qemudNodeDeviceReAttach, /* 0.6.1 */
    .nodeDeviceReset = qemudNodeDeviceReset, /* 0.6.1 */
    .domainMigratePrepareTunnel = qemudDomainMigratePrepareTunnel, /* 0.7.2 */
    .isEncrypted = qemuIsEncrypted, /* 0.7.3 */
    .isSecure = qemuIsSecure, /* 0.7.3 */
    .domainIsActive = qemuDomainIsActive, /* 0.7.3 */
    .domainIsPersistent = qemuDomainIsPersistent, /* 0.7.3 */
    .domainIsUpdated = qemuDomainIsUpdated, /* 0.8.6 */
    .cpuCompare = qemuCPUCompare, /* 0.7.5 */
    .cpuBaseline = qemuCPUBaseline, /* 0.7.7 */
    .domainGetJobInfo = qemuDomainGetJobInfo, /* 0.7.7 */
    .domainAbortJob = qemuDomainAbortJob, /* 0.7.7 */
    .domainMigrateSetMaxDowntime = qemuDomainMigrateSetMaxDowntime, /* 0.8.0 */
    .domainMigrateSetMaxSpeed = qemuDomainMigrateSetMaxSpeed, /* 0.9.0 */
    .domainEventRegisterAny = qemuDomainEventRegisterAny, /* 0.8.0 */
    .domainEventDeregisterAny = qemuDomainEventDeregisterAny, /* 0.8.0 */
    .domainManagedSave = qemuDomainManagedSave, /* 0.8.0 */
    .domainHasManagedSaveImage = qemuDomainHasManagedSaveImage, /* 0.8.0 */
    .domainManagedSaveRemove = qemuDomainManagedSaveRemove, /* 0.8.0 */
    .domainSnapshotCreateXML = qemuDomainSnapshotCreateXML, /* 0.8.0 */
    .domainSnapshotGetXMLDesc = qemuDomainSnapshotGetXMLDesc, /* 0.8.0 */
    .domainSnapshotNum = qemuDomainSnapshotNum, /* 0.8.0 */
    .domainSnapshotListNames = qemuDomainSnapshotListNames, /* 0.8.0 */
    .domainSnapshotLookupByName = qemuDomainSnapshotLookupByName, /* 0.8.0 */
    .domainHasCurrentSnapshot = qemuDomainHasCurrentSnapshot, /* 0.8.0 */
    .domainSnapshotCurrent = qemuDomainSnapshotCurrent, /* 0.8.0 */
    .domainRevertToSnapshot = qemuDomainRevertToSnapshot, /* 0.8.0 */
    .domainSnapshotDelete = qemuDomainSnapshotDelete, /* 0.8.0 */
    .qemuDomainMonitorCommand = qemuDomainMonitorCommand, /* 0.8.3 */
    .domainOpenConsole = qemuDomainOpenConsole, /* 0.8.6 */
    .domainInjectNMI = qemuDomainInjectNMI, /* 0.9.2 */
8089 8090 8091 8092 8093 8094
    .domainMigrateBegin3 = qemuDomainMigrateBegin3, /* 0.9.2 */
    .domainMigratePrepare3 = qemuDomainMigratePrepare3, /* 0.9.2 */
    .domainMigratePrepareTunnel3 = qemuDomainMigratePrepareTunnel3, /* 0.9.2 */
    .domainMigratePerform3 = qemuDomainMigratePerform3, /* 0.9.2 */
    .domainMigrateFinish3 = qemuDomainMigrateFinish3, /* 0.9.2 */
    .domainMigrateConfirm3 = qemuDomainMigrateConfirm3, /* 0.9.2 */
8095 8096 8097
};


8098
static virStateDriver qemuStateDriver = {
8099
    .name = "QEMU",
8100 8101 8102 8103
    .initialize = qemudStartup,
    .cleanup = qemudShutdown,
    .reload = qemudReload,
    .active = qemudActive,
8104
};
8105

8106
static void
8107
qemuVMDriverLock(void) {
8108 8109 8110 8111 8112
    qemuDriverLock(qemu_driver);
};


static void
8113
qemuVMDriverUnlock(void) {
8114 8115 8116 8117
    qemuDriverUnlock(qemu_driver);
};


8118 8119 8120 8121 8122 8123 8124 8125 8126
static int
qemuVMFilterRebuild(virConnectPtr conn ATTRIBUTE_UNUSED,
                    virHashIterator iter, void *data)
{
    virHashForEach(qemu_driver->domains.objs, iter, data);

    return 0;
}

S
Stefan Berger 已提交
8127 8128
static virNWFilterCallbackDriver qemuCallbackDriver = {
    .name = "QEMU",
8129 8130 8131
    .vmFilterRebuild = qemuVMFilterRebuild,
    .vmDriverLock = qemuVMDriverLock,
    .vmDriverUnlock = qemuVMDriverUnlock,
S
Stefan Berger 已提交
8132 8133
};

8134
int qemuRegister(void) {
8135 8136
    virRegisterDriver(&qemuDriver);
    virRegisterStateDriver(&qemuStateDriver);
S
Stefan Berger 已提交
8137
    virNWFilterRegisterCallbackDriver(&qemuCallbackDriver);
8138 8139
    return 0;
}