qemu_migration.c 197.7 KB
Newer Older
1 2 3
/*
 * qemu_migration.c: QEMU migration handling
 *
4
 * Copyright (C) 2006-2015 Red Hat, Inc.
5 6 7 8 9 10 11 12 13 14 15 16
 *
 * 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
17
 * License along with this library.  If not, see
O
Osier Yang 已提交
18
 * <http://www.gnu.org/licenses/>.
19 20 21 22 23
 *
 */

#include <config.h>

J
Ján Tomko 已提交
24 25
#include <netdb.h>
#include <sys/socket.h>
26
#include <sys/time.h>
27
#include <fcntl.h>
28
#include <poll.h>
29 30

#include "qemu_migration.h"
31
#include "qemu_migration_cookie.h"
32 33 34 35
#include "qemu_monitor.h"
#include "qemu_domain.h"
#include "qemu_process.h"
#include "qemu_capabilities.h"
36
#include "qemu_alias.h"
37
#include "qemu_cgroup.h"
38
#include "qemu_hotplug.h"
39
#include "qemu_blockjob.h"
40
#include "qemu_security.h"
41

42
#include "domain_audit.h"
43
#include "virlog.h"
44
#include "virerror.h"
45
#include "viralloc.h"
E
Eric Blake 已提交
46
#include "virfile.h"
47
#include "virnetdevopenvswitch.h"
48
#include "datatypes.h"
49
#include "virfdstream.h"
50
#include "viruuid.h"
51
#include "virtime.h"
52
#include "locking/domain_lock.h"
53
#include "rpc/virnetsocket.h"
54
#include "virstoragefile.h"
M
Martin Kletzander 已提交
55
#include "viruri.h"
56
#include "virhook.h"
57
#include "virstring.h"
58
#include "virtypedparam.h"
M
Michael R. Hines 已提交
59
#include "virprocess.h"
60
#include "nwfilter_conf.h"
61
#include "storage/storage_driver.h"
62 63 64

#define VIR_FROM_THIS VIR_FROM_QEMU

65 66
VIR_LOG_INIT("qemu.qemu_migration");

67 68 69 70 71 72 73 74 75 76 77 78 79
VIR_ENUM_IMPL(qemuMigrationJobPhase, QEMU_MIGRATION_PHASE_LAST,
              "none",
              "perform2",
              "begin3",
              "perform3",
              "perform3_done",
              "confirm3_cancelled",
              "confirm3",
              "prepare",
              "finish2",
              "finish3",
);

80 81 82 83 84
VIR_ENUM_IMPL(qemuMigrationCompressMethod, QEMU_MIGRATION_COMPRESS_LAST,
              "xbzrle",
              "mt",
);

85 86
#define QEMU_MIGRATION_TLS_ALIAS_BASE "libvirt_migrate"

87

88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110
/* qemuMigrationCheckTLSCreds
 * @driver: pointer to qemu driver
 * @vm: domain object
 * @asyncJob: migration job to join
 *
 * Query the migration parameters looking for the 'tls-creds' parameter.
 * If found, then we can support setting or clearing the parameters and thus
 * can support TLS for migration.
 *
 * Returns 0 if we were able to successfully fetch the params and
 * additionally if the tls-creds parameter exists, saves it in the
 * private domain structure. Returns -1 on failure.
 */
static int
qemuMigrationCheckTLSCreds(virQEMUDriverPtr driver,
                           virDomainObjPtr vm,
                           qemuDomainAsyncJob asyncJob)
{
    int ret = -1;
    qemuDomainObjPrivatePtr priv = vm->privateData;
    qemuMonitorMigrationParams migParams = { 0 };

    if (qemuDomainObjEnterMonitorAsync(driver, vm, asyncJob) < 0)
111
        return -1;
112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215

    if (qemuMonitorGetMigrationParams(priv->mon, &migParams) < 0)
        goto cleanup;

    /* NB: Could steal NULL pointer too! Let caller decide what to do. */
    VIR_STEAL_PTR(priv->migTLSAlias, migParams.migrateTLSAlias);

    ret = 0;

 cleanup:
    if (qemuDomainObjExitMonitor(driver, vm) < 0)
        ret = -1;

    qemuMigrationParamsClear(&migParams);

    return ret;
}


/* qemuMigrationCheckSetupTLS
 * @conn: Connection pointer
 * @driver: pointer to qemu driver
 * @vm: domain object
 * @cfg: configuration pointer
 * @asyncJob: migration job to join
 *
 * Check if TLS is possible and set up the environment. Assumes the caller
 * desires to use TLS (e.g. caller found VIR_MIGRATE_TLS flag).
 *
 * Ensure the qemu.conf has been properly configured to add an entry for
 * "migrate_tls_x509_cert_dir". Also check if the "tls-creds" parameter
 * was present from a query of migration parameters
 *
 * Returns 0 on success, -1 on error/failure
 */
static int
qemuMigrationCheckSetupTLS(virConnectPtr conn,
                           virQEMUDriverPtr driver,
                           virQEMUDriverConfigPtr cfg,
                           virDomainObjPtr vm,
                           qemuDomainAsyncJob asyncJob)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;

    if (!cfg->migrateTLSx509certdir) {
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                       _("host migration TLS directory not configured"));
        return -1;
    }

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

    if (!priv->migTLSAlias) {
        virReportError(VIR_ERR_OPERATION_UNSUPPORTED, "%s",
                       _("TLS migration is not supported with this "
                         "QEMU binary"));
        return -1;
    }

    /* If there's a secret, then grab/store it now using the connection */
    if (cfg->migrateTLSx509secretUUID &&
        !(priv->migSecinfo =
          qemuDomainSecretInfoTLSNew(conn, priv, QEMU_MIGRATION_TLS_ALIAS_BASE,
                                     cfg->migrateTLSx509secretUUID)))
        return -1;

    return 0;
}


/* qemuMigrationAddTLSObjects
 * @driver: pointer to qemu driver
 * @vm: domain object
 * @cfg: configuration pointer
 * @tlsListen: server or client
 * @asyncJob: Migration job to join
 * @tlsAlias: alias to be generated for TLS object
 * @secAlias: alias to be generated for a secinfo object
 * @migParams: migration parameters to set
 *
 * Create the TLS objects for the migration and set the migParams value
 *
 * Returns 0 on success, -1 on failure
 */
static int
qemuMigrationAddTLSObjects(virQEMUDriverPtr driver,
                           virDomainObjPtr vm,
                           virQEMUDriverConfigPtr cfg,
                           bool tlsListen,
                           qemuDomainAsyncJob asyncJob,
                           char **tlsAlias,
                           char **secAlias,
                           qemuMonitorMigrationParamsPtr migParams)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    virJSONValuePtr tlsProps = NULL;
    virJSONValuePtr secProps = NULL;

    if (qemuDomainGetTLSObjects(priv->qemuCaps, priv->migSecinfo,
                                cfg->migrateTLSx509certdir, tlsListen,
                                cfg->migrateTLSx509verify,
                                QEMU_MIGRATION_TLS_ALIAS_BASE,
                                &tlsProps, tlsAlias, &secProps, secAlias) < 0)
216
        goto error;
217 218 219 220 221 222 223 224 225

    /* Ensure the domain doesn't already have the TLS objects defined...
     * This should prevent any issues just in case some cleanup wasn't
     * properly completed (both src and dst use the same alias) or
     * some other error path between now and perform . */
    qemuDomainDelTLSObjects(driver, vm, asyncJob, *secAlias, *tlsAlias);

    if (qemuDomainAddTLSObjects(driver, vm, asyncJob, *secAlias, &secProps,
                                *tlsAlias, &tlsProps) < 0)
226
        goto error;
227 228

    if (VIR_STRDUP(migParams->migrateTLSAlias, *tlsAlias) < 0)
229
        goto error;
230 231

    return 0;
232 233 234 235 236

 error:
    virJSONValueFree(tlsProps);
    virJSONValueFree(secProps);
    return -1;
237 238 239
}


240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255
static void
qemuMigrationStoreDomainState(virDomainObjPtr vm)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    priv->preMigrationState = virDomainObjGetState(vm, NULL);

    VIR_DEBUG("Storing pre-migration state=%d domain=%p",
              priv->preMigrationState, vm);
}

/* Returns true if the domain was resumed, false otherwise */
static bool
qemuMigrationRestoreDomainState(virConnectPtr conn, virDomainObjPtr vm)
{
    virQEMUDriverPtr driver = conn->privateData;
    qemuDomainObjPrivatePtr priv = vm->privateData;
256 257
    int reason;
    virDomainState state = virDomainObjGetState(vm, &reason);
258 259
    bool ret = false;

260 261 262 263 264
    VIR_DEBUG("driver=%p, vm=%p, pre-mig-state=%s, state=%s, reason=%s",
              driver, vm,
              virDomainStateTypeToString(priv->preMigrationState),
              virDomainStateTypeToString(state),
              virDomainStateReasonToString(state, reason));
265

266 267 268 269 270
    if (state != VIR_DOMAIN_PAUSED ||
        reason == VIR_DOMAIN_PAUSED_POSTCOPY_FAILED)
        goto cleanup;

    if (priv->preMigrationState == VIR_DOMAIN_RUNNING) {
271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293
        /* This is basically the only restore possibility that's safe
         * and we should attempt to do */

        VIR_DEBUG("Restoring pre-migration state due to migration error");

        /* we got here through some sort of failure; start the domain again */
        if (qemuProcessStartCPUs(driver, vm, conn,
                                 VIR_DOMAIN_RUNNING_MIGRATION_CANCELED,
                                 QEMU_ASYNC_JOB_MIGRATION_OUT) < 0) {
            /* Hm, we already know we are in error here.  We don't want to
             * overwrite the previous error, though, so we just throw something
             * to the logs and hope for the best */
            VIR_ERROR(_("Failed to resume guest %s after failure"), vm->def->name);
            goto cleanup;
        }
        ret = true;
    }

 cleanup:
    priv->preMigrationState = VIR_DOMAIN_NOSTATE;
    return ret;
}

294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347

static int
qemuMigrationPrecreateDisk(virConnectPtr conn,
                           virDomainDiskDefPtr disk,
                           unsigned long long capacity)
{
    int ret = -1;
    virStoragePoolPtr pool = NULL;
    virStorageVolPtr vol = NULL;
    char *volName = NULL, *basePath = NULL;
    char *volStr = NULL;
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    const char *format = NULL;
    unsigned int flags = 0;

    VIR_DEBUG("Precreate disk type=%s", virStorageTypeToString(disk->src->type));

    switch ((virStorageType) disk->src->type) {
    case VIR_STORAGE_TYPE_FILE:
        if (!virDomainDiskGetSource(disk)) {
            VIR_DEBUG("Dropping sourceless disk '%s'",
                      disk->dst);
            return 0;
        }

        if (VIR_STRDUP(basePath, disk->src->path) < 0)
            goto cleanup;

        if (!(volName = strrchr(basePath, '/'))) {
            virReportError(VIR_ERR_INVALID_ARG,
                           _("malformed disk path: %s"),
                           disk->src->path);
            goto cleanup;
        }

        *volName = '\0';
        volName++;

        if (!(pool = storagePoolLookupByTargetPath(conn, basePath)))
            goto cleanup;
        format = virStorageFileFormatTypeToString(disk->src->format);
        if (disk->src->format == VIR_STORAGE_FILE_QCOW2)
            flags |= VIR_STORAGE_VOL_CREATE_PREALLOC_METADATA;
        break;

    case VIR_STORAGE_TYPE_VOLUME:
        if (!(pool = virStoragePoolLookupByName(conn, disk->src->srcpool->pool)))
            goto cleanup;
        format = virStorageFileFormatTypeToString(disk->src->format);
        volName = disk->src->srcpool->volume;
        if (disk->src->format == VIR_STORAGE_FILE_QCOW2)
            flags |= VIR_STORAGE_VOL_CREATE_PREALLOC_METADATA;
        break;

348 349 350 351 352
    case VIR_STORAGE_TYPE_NETWORK:
        VIR_DEBUG("Skipping creation of network disk '%s'",
                  disk->dst);
        return 0;

353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399
    case VIR_STORAGE_TYPE_BLOCK:
    case VIR_STORAGE_TYPE_DIR:
    case VIR_STORAGE_TYPE_NONE:
    case VIR_STORAGE_TYPE_LAST:
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("cannot precreate storage for disk type '%s'"),
                       virStorageTypeToString(disk->src->type));
        goto cleanup;
    }

    if ((vol = virStorageVolLookupByName(pool, volName))) {
        VIR_DEBUG("Skipping creation of already existing volume of name '%s'",
                  volName);
        ret = 0;
        goto cleanup;
    }

    virBufferAddLit(&buf, "<volume>\n");
    virBufferAdjustIndent(&buf, 2);
    virBufferEscapeString(&buf, "<name>%s</name>\n", volName);
    virBufferAsprintf(&buf, "<capacity>%llu</capacity>\n", capacity);
    virBufferAddLit(&buf, "<target>\n");
    virBufferAdjustIndent(&buf, 2);
    virBufferAsprintf(&buf, "<format type='%s'/>\n", format);
    virBufferAdjustIndent(&buf, -2);
    virBufferAddLit(&buf, "</target>\n");
    virBufferAdjustIndent(&buf, -2);
    virBufferAddLit(&buf, "</volume>\n");

    if (!(volStr = virBufferContentAndReset(&buf))) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("unable to create volume XML"));
        goto cleanup;
    }

    if (!(vol = virStorageVolCreateXML(pool, volStr, flags)))
        goto cleanup;

    ret = 0;
 cleanup:
    VIR_FREE(basePath);
    VIR_FREE(volStr);
    virObjectUnref(vol);
    virObjectUnref(pool);
    return ret;
}

400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417
static bool
qemuMigrateDisk(virDomainDiskDef const *disk,
                size_t nmigrate_disks, const char **migrate_disks)
{
    size_t i;

    /* Check if the disk alias is in the list */
    if (nmigrate_disks) {
        for (i = 0; i < nmigrate_disks; i++) {
            if (STREQ(disk->dst, migrate_disks[i]))
                return true;
        }
        return false;
    }

    /* Default is to migrate only non-shared non-readonly disks
     * with source */
    return !disk->src->shared && !disk->src->readonly &&
418
           !virStorageSourceIsEmpty(disk->src);
419 420
}

421 422 423 424 425

static int
qemuMigrationPrecreateStorage(virConnectPtr conn,
                              virQEMUDriverPtr driver ATTRIBUTE_UNUSED,
                              virDomainObjPtr vm,
426 427
                              qemuMigrationCookieNBDPtr nbd,
                              size_t nmigrate_disks,
428 429
                              const char **migrate_disks,
                              bool incremental)
430 431 432 433 434 435 436 437 438 439 440
{
    int ret = -1;
    size_t i = 0;

    if (!nbd || !nbd->ndisks)
        return 0;

    for (i = 0; i < nbd->ndisks; i++) {
        virDomainDiskDefPtr disk;
        const char *diskSrcPath;

441
        VIR_DEBUG("Looking up disk target '%s' (capacity=%llu)",
442 443
                  nbd->disks[i].target, nbd->disks[i].capacity);

444 445
        if (!(disk = virDomainDiskByName(vm->def, nbd->disks[i].target,
                                         false))) {
446 447 448 449 450 451 452 453
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("unable to find disk by target: %s"),
                           nbd->disks[i].target);
            goto cleanup;
        }

        diskSrcPath = virDomainDiskGetSource(disk);

454 455
        /* Skip disks we don't want to migrate and already existing disks. */
        if (!qemuMigrateDisk(disk, nmigrate_disks, migrate_disks) ||
456 457 458 459
            (diskSrcPath && virFileExists(diskSrcPath))) {
            continue;
        }

460 461 462 463 464 465 466
        if (incremental) {
            virReportError(VIR_ERR_OPERATION_UNSUPPORTED, "%s",
                           _("pre-creation of storage targets for incremental "
                             "storage migration is not supported"));
            goto cleanup;
        }

467 468 469 470 471 472 473 474 475 476 477 478
        VIR_DEBUG("Proceeding with disk source %s", NULLSTR(diskSrcPath));

        if (qemuMigrationPrecreateDisk(conn, disk, nbd->disks[i].capacity) < 0)
            goto cleanup;
    }

    ret = 0;
 cleanup:
    return ret;
}


479 480 481 482 483 484 485 486 487 488 489 490 491 492
/**
 * qemuMigrationStartNBDServer:
 * @driver: qemu driver
 * @vm: domain
 *
 * Starts NBD server. This is a newer method to copy
 * storage during migration than using 'blk' and 'inc'
 * arguments in 'migrate' monitor command.
 * Error is reported here.
 *
 * Returns 0 on success, -1 otherwise.
 */
static int
qemuMigrationStartNBDServer(virQEMUDriverPtr driver,
J
Ján Tomko 已提交
493
                            virDomainObjPtr vm,
494 495
                            const char *listenAddr,
                            size_t nmigrate_disks,
496 497
                            const char **migrate_disks,
                            int nbdPort)
498 499 500 501 502 503 504
{
    int ret = -1;
    qemuDomainObjPrivatePtr priv = vm->privateData;
    unsigned short port = 0;
    char *diskAlias = NULL;
    size_t i;

505 506 507 508 509 510
    if (nbdPort < 0 || nbdPort > USHRT_MAX) {
        virReportError(VIR_ERR_INVALID_ARG, "%s",
                       _("nbd port must be in range 0-65535"));
        return -1;
    }

511 512 513
    for (i = 0; i < vm->def->ndisks; i++) {
        virDomainDiskDefPtr disk = vm->def->disks[i];

514 515
        /* check whether disk should be migrated */
        if (!qemuMigrateDisk(disk, nmigrate_disks, migrate_disks))
516 517
            continue;

518
        if (disk->src->readonly || virStorageSourceIsEmpty(disk->src)) {
519
            virReportError(VIR_ERR_OPERATION_UNSUPPORTED,
520 521
                           _("Cannot migrate empty or read-only disk %s"),
                           disk->dst);
522 523 524
            goto cleanup;
        }

525
        VIR_FREE(diskAlias);
526
        if (!(diskAlias = qemuAliasFromDisk(disk)))
527 528 529 530 531 532
            goto cleanup;

        if (qemuDomainObjEnterMonitorAsync(driver, vm,
                                           QEMU_ASYNC_JOB_MIGRATION_IN) < 0)
            goto cleanup;

533 534 535 536 537 538 539 540
        if (port == 0) {
            if (nbdPort)
                port = nbdPort;
            else if (virPortAllocatorAcquire(driver->migrationPorts, &port) < 0)
                goto exit_monitor;

            if (qemuMonitorNBDServerStart(priv->mon, listenAddr, port) < 0)
                goto exit_monitor;
541 542
        }

543 544 545
        if (qemuMonitorNBDServerAdd(priv->mon, diskAlias, true) < 0)
            goto exit_monitor;
        if (qemuDomainObjExitMonitor(driver, vm) < 0)
546 547 548 549 550 551
            goto cleanup;
    }

    priv->nbdPort = port;
    ret = 0;

552
 cleanup:
553
    VIR_FREE(diskAlias);
554
    if (ret < 0 && nbdPort == 0)
555
        virPortAllocatorRelease(driver->migrationPorts, port);
556
    return ret;
557 558 559 560

 exit_monitor:
    ignore_value(qemuDomainObjExitMonitor(driver, vm));
    goto cleanup;
561 562
}

563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589

static int
qemuMigrationStopNBDServer(virQEMUDriverPtr driver,
                           virDomainObjPtr vm,
                           qemuMigrationCookiePtr mig)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;

    if (!mig->nbd)
        return 0;

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

    if (qemuMonitorNBDServerStop(priv->mon) < 0)
        VIR_WARN("Unable to stop NBD server");
    if (qemuDomainObjExitMonitor(driver, vm) < 0)
        return -1;

    virPortAllocatorRelease(driver->migrationPorts, priv->nbdPort);
    priv->nbdPort = 0;
    return 0;
}


/**
590
 * qemuMigrationDriveMirrorReady:
591 592 593 594 595 596 597 598 599 600 601 602
 * @driver: qemu driver
 * @vm: domain
 *
 * Check the status of all drive-mirrors started by
 * qemuMigrationDriveMirror. Any pending block job events
 * for the mirrored disks will be processed.
 *
 * Returns 1 if all mirrors are "ready",
 *         0 if some mirrors are still performing initial sync,
 *        -1 on error.
 */
static int
603
qemuMigrationDriveMirrorReady(virQEMUDriverPtr driver,
604 605
                              virDomainObjPtr vm,
                              qemuDomainAsyncJob asyncJob)
606 607
{
    size_t i;
608 609
    size_t notReady = 0;
    int status;
610 611 612

    for (i = 0; i < vm->def->ndisks; i++) {
        virDomainDiskDefPtr disk = vm->def->disks[i];
613
        qemuDomainDiskPrivatePtr diskPriv = QEMU_DOMAIN_DISK_PRIVATE(disk);
614

615
        if (!diskPriv->migrating)
616 617
            continue;

618
        status = qemuBlockJobUpdate(driver, vm, asyncJob, disk);
619
        if (status == VIR_DOMAIN_BLOCK_JOB_FAILED) {
620 621 622 623 624
            virReportError(VIR_ERR_OPERATION_FAILED,
                           _("migration of disk %s failed"),
                           disk->dst);
            return -1;
        }
625 626 627

        if (disk->mirrorState != VIR_DOMAIN_DISK_MIRROR_STATE_READY)
            notReady++;
628 629
    }

630 631 632 633 634 635 636
    if (notReady) {
        VIR_DEBUG("Waiting for %zu disk mirrors to get ready", notReady);
        return 0;
    } else {
        VIR_DEBUG("All disk mirrors are ready");
        return 1;
    }
637 638 639
}


640 641 642 643
/*
 * If @check is true, the function will report an error and return a different
 * code in case a block job fails. This way we can properly abort migration in
 * case some block jobs failed once all memory has already been transferred.
644
 *
645 646 647 648
 * Returns 1 if all mirrors are gone,
 *         0 if some mirrors are still active,
 *         -1 some mirrors failed but some are still active,
 *         -2 all mirrors are gone but some of them failed.
649 650
 */
static int
651
qemuMigrationDriveMirrorCancelled(virQEMUDriverPtr driver,
652
                                  virDomainObjPtr vm,
653
                                  qemuDomainAsyncJob asyncJob,
654
                                  bool check)
655
{
656 657
    size_t i;
    size_t active = 0;
658
    size_t completed = 0;
659 660
    int status;
    bool failed = false;
661

662
 retry:
663 664 665
    for (i = 0; i < vm->def->ndisks; i++) {
        virDomainDiskDefPtr disk = vm->def->disks[i];
        qemuDomainDiskPrivatePtr diskPriv = QEMU_DOMAIN_DISK_PRIVATE(disk);
666

667 668
        if (!diskPriv->migrating)
            continue;
669

670
        status = qemuBlockJobUpdate(driver, vm, asyncJob, disk);
671 672 673
        switch (status) {
        case VIR_DOMAIN_BLOCK_JOB_FAILED:
            if (check) {
674
                virReportError(VIR_ERR_OPERATION_FAILED,
675
                               _("migration of disk %s failed"),
676
                               disk->dst);
677
                failed = true;
678
            }
M
Marc Hartmayer 已提交
679
            ATTRIBUTE_FALLTHROUGH;
680 681
        case VIR_DOMAIN_BLOCK_JOB_CANCELED:
        case VIR_DOMAIN_BLOCK_JOB_COMPLETED:
682
            qemuBlockJobSyncEnd(driver, vm, asyncJob, disk);
683 684
            diskPriv->migrating = false;
            break;
685

686 687
        default:
            active++;
688
        }
689 690 691 692 693 694 695 696 697 698 699 700 701

        if (status == VIR_DOMAIN_BLOCK_JOB_COMPLETED)
            completed++;
    }

    /* Updating completed block job drops the lock thus we have to recheck
     * block jobs for disks that reside before the disk(s) with completed
     * block job.
     */
    if (completed > 0) {
        completed = 0;
        active = 0;
        goto retry;
702
    }
703

704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719
    if (failed) {
        if (active) {
            VIR_DEBUG("Some disk mirrors failed; still waiting for %zu "
                      "disk mirrors to finish", active);
            return -1;
        } else {
            VIR_DEBUG("All disk mirrors are gone; some of them failed");
            return -2;
        }
    } else {
        if (active) {
            VIR_DEBUG("Waiting for %zu disk mirrors to finish", active);
            return 0;
        } else {
            VIR_DEBUG("All disk mirrors are gone");
            return 1;
720
        }
721
    }
722
}
723 724


725 726 727 728 729 730 731 732 733
/*
 * Returns 0 on success,
 *         1 when job is already completed or it failed and failNoJob is false,
 *         -1 on error or when job failed and failNoJob is true.
 */
static int
qemuMigrationCancelOneDriveMirror(virQEMUDriverPtr driver,
                                  virDomainObjPtr vm,
                                  virDomainDiskDefPtr disk,
734 735
                                  bool failNoJob,
                                  qemuDomainAsyncJob asyncJob)
736 737 738 739 740 741 742
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    char *diskAlias = NULL;
    int ret = -1;
    int status;
    int rv;

743
    status = qemuBlockJobUpdate(driver, vm, asyncJob, disk);
744 745 746 747 748 749 750 751 752 753 754 755 756 757 758
    switch (status) {
    case VIR_DOMAIN_BLOCK_JOB_FAILED:
    case VIR_DOMAIN_BLOCK_JOB_CANCELED:
        if (failNoJob) {
            virReportError(VIR_ERR_OPERATION_FAILED,
                           _("migration of disk %s failed"),
                           disk->dst);
            return -1;
        }
        return 1;

    case VIR_DOMAIN_BLOCK_JOB_COMPLETED:
        return 1;
    }

759
    if (!(diskAlias = qemuAliasFromDisk(disk)))
760 761
        return -1;

762
    if (qemuDomainObjEnterMonitorAsync(driver, vm, asyncJob) < 0)
763 764
        goto cleanup;

765
    rv = qemuMonitorBlockJobCancel(priv->mon, diskAlias);
766 767 768 769 770

    if (qemuDomainObjExitMonitor(driver, vm) < 0 || rv < 0)
        goto cleanup;

    ret = 0;
771 772 773 774 775 776 777 778 779 780 781

 cleanup:
    VIR_FREE(diskAlias);
    return ret;
}


/**
 * qemuMigrationCancelDriveMirror:
 * @driver: qemu driver
 * @vm: domain
782
 * @check: if true report an error when some of the mirrors fails
783 784 785 786 787 788 789 790 791
 *
 * Cancel all drive-mirrors started by qemuMigrationDriveMirror.
 * Any pending block job events for the affected disks will be
 * processed.
 *
 * Returns 0 on success, -1 otherwise.
 */
static int
qemuMigrationCancelDriveMirror(virQEMUDriverPtr driver,
792
                               virDomainObjPtr vm,
793
                               bool check,
794 795
                               qemuDomainAsyncJob asyncJob,
                               virConnectPtr dconn)
796
{
797
    virErrorPtr err = NULL;
798
    int ret = -1;
799
    size_t i;
800 801 802 803
    int rv;
    bool failed = false;

    VIR_DEBUG("Cancelling drive mirrors for domain %s", vm->def->name);
804 805 806

    for (i = 0; i < vm->def->ndisks; i++) {
        virDomainDiskDefPtr disk = vm->def->disks[i];
807
        qemuDomainDiskPrivatePtr diskPriv = QEMU_DOMAIN_DISK_PRIVATE(disk);
808

809
        if (!diskPriv->migrating)
810 811
            continue;

812 813
        rv = qemuMigrationCancelOneDriveMirror(driver, vm, disk,
                                               check, asyncJob);
814 815 816 817 818 819
        if (rv != 0) {
            if (rv < 0) {
                if (!err)
                    err = virSaveLastError();
                failed = true;
            }
820
            qemuBlockJobSyncEnd(driver, vm, asyncJob, disk);
821 822 823 824
            diskPriv->migrating = false;
        }
    }

825 826
    while ((rv = qemuMigrationDriveMirrorCancelled(driver, vm, asyncJob,
                                                   check)) != 1) {
827 828 829 830 831 832 833
        if (check && !failed &&
            dconn && virConnectIsAlive(dconn) <= 0) {
            virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                           _("Lost connection to destination host"));
            failed = true;
        }

834 835 836 837
        if (rv < 0) {
            failed = true;
            if (rv == -2)
                break;
838
        }
839

840 841 842 843 844
        if (failed && !err)
            err = virSaveLastError();

        if (virDomainObjWait(vm) < 0)
            goto cleanup;
845 846
    }

847 848 849
    ret = failed ? -1 : 0;

 cleanup:
850 851 852 853 854
    if (err) {
        virSetError(err);
        virFreeError(err);
    }
    return ret;
855 856 857
}


858 859 860 861 862 863
/**
 * qemuMigrationDriveMirror:
 * @driver: qemu driver
 * @vm: domain
 * @mig: migration cookie
 * @host: where are we migrating to
864
 * @speed: bandwidth limit in MiB/s
865 866 867 868
 * @migrate_flags: migrate monitor command flags
 *
 * Run drive-mirror to feed NBD server running on dst and wait
 * till the process switches into another phase where writes go
869 870 871 872 873
 * simultaneously to both source and destination. On success,
 * update @migrate_flags so we don't tell 'migrate' command
 * to do the very same operation. On failure, the caller is
 * expected to call qemuMigrationCancelDriveMirror to stop all
 * running mirrors.
874 875 876 877 878 879 880 881 882 883
 *
 * Returns 0 on success (@migrate_flags updated),
 *        -1 otherwise.
 */
static int
qemuMigrationDriveMirror(virQEMUDriverPtr driver,
                         virDomainObjPtr vm,
                         qemuMigrationCookiePtr mig,
                         const char *host,
                         unsigned long speed,
884 885
                         unsigned int *migrate_flags,
                         size_t nmigrate_disks,
886 887
                         const char **migrate_disks,
                         virConnectPtr dconn)
888 889 890 891
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret = -1;
    int port;
892
    size_t i;
893 894
    char *diskAlias = NULL;
    char *nbd_dest = NULL;
895
    char *hoststr = NULL;
896
    unsigned long long mirror_speed = speed;
897
    unsigned int mirror_flags = VIR_DOMAIN_BLOCK_REBASE_REUSE_EXT;
898
    int rv;
899
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
900 901

    VIR_DEBUG("Starting drive mirrors for domain %s", vm->def->name);
902

903 904 905 906 907 908 909 910
    if (mirror_speed > LLONG_MAX >> 20) {
        virReportError(VIR_ERR_OVERFLOW,
                       _("bandwidth must be less than %llu"),
                       LLONG_MAX >> 20);
        goto cleanup;
    }
    mirror_speed <<= 20;

911 912 913 914
    /* steal NBD port and thus prevent its propagation back to destination */
    port = mig->nbd->port;
    mig->nbd->port = 0;

915 916
    /* escape literal IPv6 address */
    if (strchr(host, ':')) {
917
        if (virAsprintf(&hoststr, "[%s]", host) < 0)
918
            goto cleanup;
919
    } else if (VIR_STRDUP(hoststr, host) < 0) {
920
        goto cleanup;
921 922
    }

923 924 925 926 927
    if (*migrate_flags & QEMU_MONITOR_MIGRATE_NON_SHARED_INC)
        mirror_flags |= VIR_DOMAIN_BLOCK_REBASE_SHALLOW;

    for (i = 0; i < vm->def->ndisks; i++) {
        virDomainDiskDefPtr disk = vm->def->disks[i];
928
        qemuDomainDiskPrivatePtr diskPriv = QEMU_DOMAIN_DISK_PRIVATE(disk);
929
        int mon_ret;
930

931 932
        /* check whether disk should be migrated */
        if (!qemuMigrateDisk(disk, nmigrate_disks, migrate_disks))
933 934
            continue;

935
        if (!(diskAlias = qemuAliasFromDisk(disk)) ||
936
            (virAsprintf(&nbd_dest, "nbd:%s:%d:exportname=%s",
937
                         hoststr, port, diskAlias) < 0))
938 939
            goto cleanup;

940
        if (qemuDomainObjEnterMonitorAsync(driver, vm,
941
                                           QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
942 943
            goto cleanup;

944
        qemuBlockJobSyncBegin(disk);
945
        /* Force "raw" format for NBD export */
946
        mon_ret = qemuMonitorDriveMirror(priv->mon, diskAlias, nbd_dest,
947
                                         "raw", mirror_speed, 0, 0, mirror_flags);
948 949
        VIR_FREE(diskAlias);
        VIR_FREE(nbd_dest);
950

951
        if (qemuDomainObjExitMonitor(driver, vm) < 0 || mon_ret < 0) {
952
            qemuBlockJobSyncEnd(driver, vm, QEMU_ASYNC_JOB_MIGRATION_OUT, disk);
953 954
            goto cleanup;
        }
955
        diskPriv->migrating = true;
956

957
        if (virDomainSaveStatus(driver->xmlopt, cfg->stateDir, vm, driver->caps) < 0) {
958 959 960
            VIR_WARN("Failed to save status on vm %s", vm->def->name);
            goto cleanup;
        }
961
    }
962

963 964
    while ((rv = qemuMigrationDriveMirrorReady(driver, vm,
                                               QEMU_ASYNC_JOB_MIGRATION_OUT)) != 1) {
965 966
        if (rv < 0)
            goto cleanup;
967

968
        if (priv->job.abortJob) {
969
            priv->job.current->status = QEMU_DOMAIN_JOB_STATUS_CANCELED;
970 971 972 973
            virReportError(VIR_ERR_OPERATION_ABORTED, _("%s: %s"),
                           qemuDomainAsyncJobTypeToString(priv->job.asyncJob),
                           _("canceled by client"));
            goto cleanup;
974
        }
975

976 977 978 979 980 981
        if (dconn && virConnectIsAlive(dconn) <= 0) {
            virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                           _("Lost connection to destination host"));
            goto cleanup;
        }

982
        if (virDomainObjWait(vm) < 0)
983
            goto cleanup;
984 985
    }

986 987 988
    qemuMigrationFetchMirrorStats(driver, vm, QEMU_ASYNC_JOB_MIGRATION_OUT,
                                  priv->job.current);

989
    /* Okay, all disks are ready. Modify migrate_flags */
990 991 992 993
    *migrate_flags &= ~(QEMU_MONITOR_MIGRATE_NON_SHARED_DISK |
                        QEMU_MONITOR_MIGRATE_NON_SHARED_INC);
    ret = 0;

994
 cleanup:
995
    virObjectUnref(cfg);
996 997
    VIR_FREE(diskAlias);
    VIR_FREE(nbd_dest);
998
    VIR_FREE(hoststr);
999 1000
    return ret;
}
1001

1002

1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030
/**
 * qemuMigrationIsAllowedHostdev:
 * @def: domain definition
 *
 * Checks that @def does not contain any host devices unsupported accross
 * migrations. Returns true if the vm is allowed to migrate.
 */
static bool
qemuMigrationIsAllowedHostdev(const virDomainDef *def)
{
    size_t i;

    /* Migration with USB host devices is allowed, all other devices are
     * forbidden. */
    for (i = 0; i < def->nhostdevs; i++) {
        virDomainHostdevDefPtr hostdev = def->hostdevs[i];
        if (hostdev->mode != VIR_DOMAIN_HOSTDEV_MODE_SUBSYS ||
            hostdev->source.subsys.type != VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB) {
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                           _("domain has assigned non-USB host devices"));
            return false;
        }
    }

    return true;
}


1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044
/**
 * qemuMigrationIsAllowed:
 * @driver: qemu driver struct
 * @vm: domain object
 * @remote: migration is remote
 * @flags: migration flags (see struct virDomainMigrateFlags)
 *
 * Validates that the configuration of @vm can be migrated in various
 * situations. If @remote is true, the migration happens to remote host. @flags
 * is used to check various special migration types according to the request.
 *
 * Returns true if migration is supported. Reports libvirt error and returns
 * false otherwise.
 */
1045
bool
1046 1047 1048
qemuMigrationIsAllowed(virQEMUDriverPtr driver,
                       virDomainObjPtr vm,
                       bool remote,
1049
                       unsigned int flags)
1050
{
1051
    int nsnapshots;
1052
    int pauseReason;
1053
    size_t i;
1054

1055 1056 1057 1058 1059
    /* perform these checks only when migrating to remote hosts */
    if (remote) {
        nsnapshots = virDomainSnapshotObjListNum(vm->snapshots, NULL, 0);
        if (nsnapshots < 0)
            return false;
1060

1061 1062 1063 1064 1065
        if (nsnapshots > 0) {
            virReportError(VIR_ERR_OPERATION_INVALID,
                           _("cannot migrate domain with %d snapshots"),
                           nsnapshots);
            return false;
1066
        }
1067

1068
        /* cancel migration if disk I/O error is emitted while migrating */
1069
        if (flags & VIR_MIGRATE_ABORT_ON_ERROR &&
1070
            !(flags & VIR_MIGRATE_OFFLINE) &&
1071 1072
            virDomainObjGetState(vm, &pauseReason) == VIR_DOMAIN_PAUSED &&
            pauseReason == VIR_DOMAIN_PAUSED_IOERROR) {
E
Eric Blake 已提交
1073
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
1074
                           _("cannot migrate domain with I/O error"));
E
Eric Blake 已提交
1075 1076
            return false;
        }
1077 1078

    }
1079

1080 1081 1082 1083 1084 1085 1086
    /* following checks don't make sense for offline migration */
    if (!(flags & VIR_MIGRATE_OFFLINE)) {
        if (qemuProcessAutoDestroyActive(driver, vm)) {
            virReportError(VIR_ERR_OPERATION_INVALID,
                           "%s", _("domain is marked for auto destroy"));
            return false;
        }
1087

1088

1089 1090
        if (qemuDomainHasBlockjob(vm, false)) {
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
1091
                           _("domain has active block job"));
1092 1093 1094 1095 1096 1097
            return false;
        }

        if (!qemuMigrationIsAllowedHostdev(vm->def))
            return false;

1098
        if (vm->def->cpu) {
1099 1100 1101
            /* QEMU blocks migration and save with invariant TSC enabled
             * unless TSC frequency is explicitly set.
             */
1102 1103
            if (virCPUCheckFeature(vm->def->os.arch, vm->def->cpu,
                                   "invtsc") == 1) {
1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121
                bool block = true;

                for (i = 0; i < vm->def->clock.ntimers; i++) {
                    virDomainTimerDefPtr timer = vm->def->clock.timers[i];

                    if (timer->name == VIR_DOMAIN_TIMER_NAME_TSC &&
                        timer->frequency > 0) {
                        block = false;
                        break;
                    }
                }

                if (block) {
                    virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                                   _("domain has 'invtsc' CPU feature but "
                                     "TSC frequency is not specified"));
                    return false;
                }
1122
            }
J
Ján Tomko 已提交
1123 1124
        }

1125 1126 1127
        /* Verify that memory device config can be transferred reliably */
        for (i = 0; i < vm->def->nmems; i++) {
            virDomainMemoryDefPtr mem = vm->def->mems[i];
1128

1129 1130 1131 1132 1133
            if (mem->model == VIR_DOMAIN_MEMORY_MODEL_DIMM &&
                mem->info.type != VIR_DOMAIN_DEVICE_ADDRESS_TYPE_DIMM) {
                virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                               _("domain's dimm info lacks slot ID "
                                 "or base address"));
1134

1135 1136
                return false;
            }
1137
        }
1138 1139 1140 1141 1142 1143

        if (vm->def->nshmems) {
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                           _("migration with shmem device is not supported"));
            return false;
        }
1144 1145
    }

1146 1147 1148
    return true;
}

1149
static bool
1150 1151
qemuMigrationIsSafe(virDomainDefPtr def,
                    size_t nmigrate_disks,
1152 1153 1154
                    const char **migrate_disks,
                    unsigned int flags)

1155
{
1156 1157
    bool storagemigration = flags & (VIR_MIGRATE_NON_SHARED_DISK |
                                     VIR_MIGRATE_NON_SHARED_INC);
1158
    size_t i;
1159
    int rc;
1160

1161
    for (i = 0; i < def->ndisks; i++) {
1162
        virDomainDiskDefPtr disk = def->disks[i];
1163
        const char *src = virDomainDiskGetSource(disk);
1164

1165
        /* Our code elsewhere guarantees shared disks are either readonly (in
1166
         * which case cache mode doesn't matter) or used with cache=none or used with cache=directsync */
1167 1168 1169
        if (virStorageSourceIsEmpty(disk->src) ||
            disk->src->readonly ||
            disk->src->shared ||
1170 1171
            disk->cachemode == VIR_DOMAIN_DISK_CACHE_DISABLE ||
            disk->cachemode == VIR_DOMAIN_DISK_CACHE_DIRECTSYNC)
1172
            continue;
1173

1174 1175 1176 1177
        /* disks which are migrated by qemu are safe too */
        if (storagemigration &&
            qemuMigrateDisk(disk, nmigrate_disks, migrate_disks))
            continue;
1178

1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190
        if (virDomainDiskGetType(disk) == VIR_STORAGE_TYPE_FILE) {
            if ((rc = virFileIsSharedFS(src)) < 0)
                return false;
            else if (rc == 0)
                continue;
            if ((rc = virStorageFileIsClusterFS(src)) < 0)
                return false;
            else if (rc == 1)
                continue;
        } else if (disk->src->type == VIR_STORAGE_TYPE_NETWORK &&
                   disk->src->protocol == VIR_STORAGE_NET_PROTOCOL_RBD) {
            continue;
1191
        }
1192 1193 1194

        virReportError(VIR_ERR_MIGRATE_UNSAFE, "%s",
                       _("Migration may lead to data corruption if disks"
1195
                         " use cache != none or cache != directsync"));
1196
        return false;
1197 1198 1199 1200 1201
    }

    return true;
}

1202 1203 1204 1205
/** qemuMigrationSetOffline
 * Pause domain for non-live migration.
 */
int
1206
qemuMigrationSetOffline(virQEMUDriverPtr driver,
1207 1208 1209
                        virDomainObjPtr vm)
{
    int ret;
1210
    VIR_DEBUG("driver=%p vm=%p", driver, vm);
1211 1212
    ret = qemuProcessStopCPUs(driver, vm, VIR_DOMAIN_PAUSED_MIGRATION,
                              QEMU_ASYNC_JOB_MIGRATION_OUT);
1213
    if (ret == 0) {
1214
        virObjectEventPtr event;
1215

1216
        event = virDomainEventLifecycleNewFromObj(vm,
1217 1218
                                         VIR_DOMAIN_EVENT_SUSPENDED,
                                         VIR_DOMAIN_EVENT_SUSPENDED_MIGRATED);
1219
        qemuDomainEventQueue(driver, event);
1220 1221 1222 1223 1224
    }

    return ret;
}

1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266

void
qemuMigrationPostcopyFailed(virQEMUDriverPtr driver,
                            virDomainObjPtr vm)
{
    virDomainState state;
    int reason;

    state = virDomainObjGetState(vm, &reason);

    if (state != VIR_DOMAIN_PAUSED &&
        state != VIR_DOMAIN_RUNNING)
        return;

    if (state == VIR_DOMAIN_PAUSED &&
        reason == VIR_DOMAIN_PAUSED_POSTCOPY_FAILED)
        return;

    VIR_WARN("Migration of domain %s failed during post-copy; "
             "leaving the domain paused", vm->def->name);

    if (state == VIR_DOMAIN_RUNNING) {
        virObjectEventPtr event;

        if (qemuProcessStopCPUs(driver, vm,
                                VIR_DOMAIN_PAUSED_POSTCOPY_FAILED,
                                QEMU_ASYNC_JOB_MIGRATION_IN) < 0) {
            VIR_WARN("Unable to pause guest CPUs for %s", vm->def->name);
            return;
        }

        event = virDomainEventLifecycleNewFromObj(vm,
                                VIR_DOMAIN_EVENT_SUSPENDED,
                                VIR_DOMAIN_EVENT_SUSPENDED_POSTCOPY_FAILED);
        qemuDomainEventQueue(driver, event);
    } else {
        virDomainObjSetState(vm, VIR_DOMAIN_PAUSED,
                             VIR_DOMAIN_PAUSED_POSTCOPY_FAILED);
    }
}


1267
static int
1268
qemuMigrationSetOption(virQEMUDriverPtr driver,
1269
                       virDomainObjPtr vm,
1270
                       qemuMonitorMigrationCaps capability,
1271
                       bool state,
1272 1273 1274 1275 1276
                       qemuDomainAsyncJob job)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret;

1277 1278 1279 1280 1281
    if (!qemuMigrationCapsGet(vm, capability)) {
        if (!state) {
            /* Unsupported but we want it off anyway */
            return 0;
        }
1282 1283

        if (job == QEMU_ASYNC_JOB_MIGRATION_IN) {
1284 1285 1286 1287
            virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED,
                           _("Migration option '%s' is not supported by "
                             "target QEMU binary"),
                           qemuMonitorMigrationCapsTypeToString(capability));
1288
        } else {
1289 1290 1291 1292
            virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED,
                           _("Migration option '%s' is not supported by "
                             "source QEMU binary"),
                           qemuMonitorMigrationCapsTypeToString(capability));
1293
        }
1294
        return -1;
1295 1296
    }

1297 1298 1299
    if (qemuDomainObjEnterMonitorAsync(driver, vm, job) < 0)
        return -1;

1300
    ret = qemuMonitorSetMigrationCapability(priv->mon, capability, state);
1301

1302 1303
    if (qemuDomainObjExitMonitor(driver, vm) < 0)
        ret = -1;
1304

1305 1306 1307
    return ret;
}

1308 1309 1310 1311 1312 1313 1314

static int
qemuMigrationSetPostCopy(virQEMUDriverPtr driver,
                         virDomainObjPtr vm,
                         bool state,
                         qemuDomainAsyncJob job)
{
1315 1316
    qemuDomainObjPrivatePtr priv = vm->privateData;

1317 1318 1319 1320 1321
    if (qemuMigrationSetOption(driver, vm,
                               QEMU_MONITOR_MIGRATION_CAPS_POSTCOPY,
                               state, job) < 0)
        return -1;

1322
    priv->job.postcopyEnabled = state;
1323 1324 1325 1326
    return 0;
}


1327
static int
1328
qemuMigrationWaitForSpice(virDomainObjPtr vm)
1329 1330 1331
{
    qemuDomainObjPrivatePtr priv = vm->privateData;

1332 1333
    if (!virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_SEAMLESS_MIGRATION) ||
        !priv->job.spiceMigration)
1334 1335
        return 0;

1336
    VIR_DEBUG("Waiting for SPICE to finish migration");
1337 1338
    while (!priv->job.spiceMigrated && !priv->job.abortJob) {
        if (virDomainObjWait(vm) < 0)
1339 1340 1341 1342
            return -1;
    }
    return 0;
}
1343

1344 1345 1346 1347

static void
qemuMigrationUpdateJobType(qemuDomainJobInfoPtr jobInfo)
{
1348
    switch ((qemuMonitorMigrationStatus) jobInfo->stats.status) {
1349 1350 1351 1352
    case QEMU_MONITOR_MIGRATION_STATUS_POSTCOPY:
        jobInfo->status = QEMU_DOMAIN_JOB_STATUS_POSTCOPY;
        break;

1353
    case QEMU_MONITOR_MIGRATION_STATUS_COMPLETED:
1354
        jobInfo->status = QEMU_DOMAIN_JOB_STATUS_QEMU_COMPLETED;
1355 1356 1357
        break;

    case QEMU_MONITOR_MIGRATION_STATUS_INACTIVE:
1358
        jobInfo->status = QEMU_DOMAIN_JOB_STATUS_NONE;
1359 1360 1361
        break;

    case QEMU_MONITOR_MIGRATION_STATUS_ERROR:
1362
        jobInfo->status = QEMU_DOMAIN_JOB_STATUS_FAILED;
1363 1364 1365
        break;

    case QEMU_MONITOR_MIGRATION_STATUS_CANCELLED:
1366
        jobInfo->status = QEMU_DOMAIN_JOB_STATUS_CANCELED;
1367 1368 1369 1370 1371
        break;

    case QEMU_MONITOR_MIGRATION_STATUS_SETUP:
    case QEMU_MONITOR_MIGRATION_STATUS_ACTIVE:
    case QEMU_MONITOR_MIGRATION_STATUS_CANCELLING:
1372
    case QEMU_MONITOR_MIGRATION_STATUS_LAST:
1373 1374 1375 1376 1377 1378
        break;
    }
}


int
1379 1380 1381
qemuMigrationFetchStats(virQEMUDriverPtr driver,
                        virDomainObjPtr vm,
                        qemuDomainAsyncJob asyncJob,
1382 1383
                        qemuDomainJobInfoPtr jobInfo,
                        char **error)
1384 1385
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
1386
    qemuMonitorMigrationStats stats;
1387 1388 1389 1390 1391
    int rv;

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

1392
    rv = qemuMonitorGetMigrationStats(priv->mon, &stats, error);
1393 1394 1395 1396

    if (qemuDomainObjExitMonitor(driver, vm) < 0 || rv < 0)
        return -1;

1397 1398 1399
    jobInfo->stats = stats;

    return 0;
1400 1401 1402
}


1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420
static const char *
qemuMigrationJobName(virDomainObjPtr vm)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;

    switch (priv->job.asyncJob) {
    case QEMU_ASYNC_JOB_MIGRATION_OUT:
        return _("migration job");
    case QEMU_ASYNC_JOB_SAVE:
        return _("domain save job");
    case QEMU_ASYNC_JOB_DUMP:
        return _("domain core dump job");
    default:
        return _("job");
    }
}


1421 1422 1423
static int
qemuMigrationCheckJobStatus(virQEMUDriverPtr driver,
                            virDomainObjPtr vm,
1424
                            qemuDomainAsyncJob asyncJob)
1425 1426 1427
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    qemuDomainJobInfoPtr jobInfo = priv->job.current;
1428
    char *error = NULL;
1429
    bool events = virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATION_EVENT);
1430
    int ret = -1;
1431

1432 1433 1434 1435 1436
    if (!events ||
        jobInfo->stats.status == QEMU_MONITOR_MIGRATION_STATUS_ERROR) {
        if (qemuMigrationFetchStats(driver, vm, asyncJob, jobInfo, &error) < 0)
            return -1;
    }
1437

1438 1439
    qemuMigrationUpdateJobType(jobInfo);

1440 1441
    switch (jobInfo->status) {
    case QEMU_DOMAIN_JOB_STATUS_NONE:
1442 1443
        virReportError(VIR_ERR_OPERATION_FAILED, _("%s: %s"),
                       qemuMigrationJobName(vm), _("is not active"));
1444
        goto cleanup;
1445

1446
    case QEMU_DOMAIN_JOB_STATUS_FAILED:
1447
        virReportError(VIR_ERR_OPERATION_FAILED, _("%s: %s"),
1448 1449 1450
                       qemuMigrationJobName(vm),
                       error ? error : _("unexpectedly failed"));
        goto cleanup;
1451

1452
    case QEMU_DOMAIN_JOB_STATUS_CANCELED:
1453 1454
        virReportError(VIR_ERR_OPERATION_ABORTED, _("%s: %s"),
                       qemuMigrationJobName(vm), _("canceled by client"));
1455
        goto cleanup;
1456

1457 1458
    case QEMU_DOMAIN_JOB_STATUS_COMPLETED:
    case QEMU_DOMAIN_JOB_STATUS_ACTIVE:
1459
    case QEMU_DOMAIN_JOB_STATUS_MIGRATING:
1460
    case QEMU_DOMAIN_JOB_STATUS_QEMU_COMPLETED:
1461
    case QEMU_DOMAIN_JOB_STATUS_POSTCOPY:
1462 1463
        break;
    }
1464 1465 1466 1467 1468 1469

    ret = 0;

 cleanup:
    VIR_FREE(error);
    return ret;
1470 1471 1472
}


1473 1474 1475
enum qemuMigrationCompletedFlags {
    QEMU_MIGRATION_COMPLETED_ABORT_ON_ERROR = (1 << 0),
    QEMU_MIGRATION_COMPLETED_CHECK_STORAGE  = (1 << 1),
1476
    QEMU_MIGRATION_COMPLETED_POSTCOPY       = (1 << 2),
1477 1478
};

1479

1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490
/**
 * Returns 1 if migration completed successfully,
 *         0 if the domain is still being migrated,
 *         -1 migration failed,
 *         -2 something else failed, we need to cancel migration.
 */
static int
qemuMigrationCompleted(virQEMUDriverPtr driver,
                       virDomainObjPtr vm,
                       qemuDomainAsyncJob asyncJob,
                       virConnectPtr dconn,
1491
                       unsigned int flags)
1492 1493 1494 1495 1496
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    qemuDomainJobInfoPtr jobInfo = priv->job.current;
    int pauseReason;

1497
    if (qemuMigrationCheckJobStatus(driver, vm, asyncJob) < 0)
1498 1499
        goto error;

1500
    if (flags & QEMU_MIGRATION_COMPLETED_CHECK_STORAGE &&
1501
        qemuMigrationDriveMirrorReady(driver, vm, asyncJob) < 0)
1502 1503
        goto error;

1504
    if (flags & QEMU_MIGRATION_COMPLETED_ABORT_ON_ERROR &&
1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517
        virDomainObjGetState(vm, &pauseReason) == VIR_DOMAIN_PAUSED &&
        pauseReason == VIR_DOMAIN_PAUSED_IOERROR) {
        virReportError(VIR_ERR_OPERATION_FAILED, _("%s: %s"),
                       qemuMigrationJobName(vm), _("failed due to I/O error"));
        goto error;
    }

    if (dconn && virConnectIsAlive(dconn) <= 0) {
        virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                       _("Lost connection to destination host"));
        goto error;
    }

1518 1519 1520 1521 1522
    /* In case of postcopy the source considers migration completed at the
     * moment it switched from active to postcopy-active state. The destination
     * will continue waiting until the migrate state changes to completed.
     */
    if (flags & QEMU_MIGRATION_COMPLETED_POSTCOPY &&
1523
        jobInfo->status == QEMU_DOMAIN_JOB_STATUS_POSTCOPY) {
1524 1525 1526 1527
        VIR_DEBUG("Migration switched to post-copy");
        return 1;
    }

1528
    if (jobInfo->status == QEMU_DOMAIN_JOB_STATUS_QEMU_COMPLETED)
1529 1530 1531 1532 1533
        return 1;
    else
        return 0;

 error:
1534 1535 1536
    switch (jobInfo->status) {
    case QEMU_DOMAIN_JOB_STATUS_MIGRATING:
    case QEMU_DOMAIN_JOB_STATUS_POSTCOPY:
1537
        /* The migration was aborted by us rather than QEMU itself. */
1538
        jobInfo->status = QEMU_DOMAIN_JOB_STATUS_FAILED;
1539
        return -2;
1540 1541 1542

    case QEMU_DOMAIN_JOB_STATUS_QEMU_COMPLETED:
        /* Something failed after QEMU already finished the migration. */
1543
        jobInfo->status = QEMU_DOMAIN_JOB_STATUS_FAILED;
1544
        return -1;
1545 1546 1547 1548

    case QEMU_DOMAIN_JOB_STATUS_FAILED:
    case QEMU_DOMAIN_JOB_STATUS_CANCELED:
        /* QEMU aborted the migration. */
1549
        return -1;
1550 1551 1552 1553 1554 1555

    case QEMU_DOMAIN_JOB_STATUS_ACTIVE:
    case QEMU_DOMAIN_JOB_STATUS_COMPLETED:
    case QEMU_DOMAIN_JOB_STATUS_NONE:
        /* Impossible. */
        break;
1556
    }
1557 1558

    return -1;
1559 1560 1561
}


1562 1563 1564
/* Returns 0 on success, -2 when migration needs to be cancelled, or -1 when
 * QEMU reports failed migration.
 */
1565
static int
J
Jiri Denemark 已提交
1566 1567
qemuMigrationWaitForCompletion(virQEMUDriverPtr driver,
                               virDomainObjPtr vm,
1568
                               qemuDomainAsyncJob asyncJob,
J
Jiri Denemark 已提交
1569
                               virConnectPtr dconn,
1570
                               unsigned int flags)
1571
{
1572
    qemuDomainObjPrivatePtr priv = vm->privateData;
J
Jiri Denemark 已提交
1573
    qemuDomainJobInfoPtr jobInfo = priv->job.current;
1574
    bool events = virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATION_EVENT);
1575
    int rv;
1576

1577 1578
    jobInfo->status = QEMU_DOMAIN_JOB_STATUS_MIGRATING;

1579 1580
    while ((rv = qemuMigrationCompleted(driver, vm, asyncJob,
                                        dconn, flags)) != 1) {
1581 1582
        if (rv < 0)
            return rv;
1583

1584 1585
        if (events) {
            if (virDomainObjWait(vm) < 0) {
1586
                jobInfo->status = QEMU_DOMAIN_JOB_STATUS_FAILED;
1587 1588 1589 1590 1591 1592 1593 1594 1595 1596
                return -2;
            }
        } else {
            /* Poll every 50ms for progress & to allow cancellation */
            struct timespec ts = { .tv_sec = 0, .tv_nsec = 50 * 1000 * 1000ull };

            virObjectUnlock(vm);
            nanosleep(&ts, NULL);
            virObjectLock(vm);
        }
1597 1598
    }

1599
    if (events)
1600
        ignore_value(qemuMigrationFetchStats(driver, vm, asyncJob, jobInfo, NULL));
1601

1602
    qemuDomainJobInfoUpdateTime(jobInfo);
1603 1604 1605 1606 1607
    qemuDomainJobInfoUpdateDowntime(jobInfo);
    VIR_FREE(priv->job.completed);
    if (VIR_ALLOC(priv->job.completed) == 0)
        *priv->job.completed = *jobInfo;

1608 1609 1610 1611
    if (asyncJob != QEMU_ASYNC_JOB_MIGRATION_OUT &&
        jobInfo->status == QEMU_DOMAIN_JOB_STATUS_QEMU_COMPLETED)
        jobInfo->status = QEMU_DOMAIN_JOB_STATUS_COMPLETED;

1612
    return 0;
1613 1614 1615
}


1616 1617 1618
static int
qemuMigrationWaitForDestCompletion(virQEMUDriverPtr driver,
                                   virDomainObjPtr vm,
1619 1620
                                   qemuDomainAsyncJob asyncJob,
                                   bool postcopy)
1621 1622
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
1623
    unsigned int flags = 0;
1624 1625 1626 1627 1628 1629 1630
    int rv;

    if (!virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATION_EVENT))
        return 0;

    VIR_DEBUG("Waiting for incoming migration to complete");

1631 1632 1633 1634 1635
    if (postcopy)
        flags = QEMU_MIGRATION_COMPLETED_POSTCOPY;

    while ((rv = qemuMigrationCompleted(driver, vm, asyncJob,
                                        NULL, flags)) != 1) {
1636 1637 1638 1639 1640 1641 1642 1643
        if (rv < 0 || virDomainObjWait(vm) < 0)
            return -1;
    }

    return 0;
}


1644
static int
1645
qemuDomainMigrateGraphicsRelocate(virQEMUDriverPtr driver,
1646
                                  virDomainObjPtr vm,
1647 1648
                                  qemuMigrationCookiePtr cookie,
                                  const char *graphicsuri)
1649 1650
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
1651 1652
    int ret = -1;
    const char *listenAddress = NULL;
1653
    virSocketAddr addr;
1654 1655 1656 1657 1658
    virURIPtr uri = NULL;
    int type = -1;
    int port = -1;
    int tlsPort = -1;
    const char *tlsSubject = NULL;
1659

1660
    if (!cookie || (!cookie->graphics && !graphicsuri))
1661 1662
        return 0;

1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681
    if (graphicsuri && !(uri = virURIParse(graphicsuri)))
        goto cleanup;

    if (cookie->graphics) {
        type = cookie->graphics->type;

        listenAddress = cookie->graphics->listen;

        if (!listenAddress ||
            (virSocketAddrParse(&addr, listenAddress, AF_UNSPEC) > 0 &&
             virSocketAddrIsWildcard(&addr)))
            listenAddress = cookie->remoteHostname;

        port = cookie->graphics->port;
        tlsPort = cookie->graphics->tlsPort;
        tlsSubject = cookie->graphics->tlsSubject;
    }

    if (uri) {
1682
        size_t i;
1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709

        if ((type = virDomainGraphicsTypeFromString(uri->scheme)) < 0) {
            virReportError(VIR_ERR_INVALID_ARG,
                           _("unknown graphics type %s"), uri->scheme);
            goto cleanup;
        }

        if (uri->server)
            listenAddress = uri->server;
        if (uri->port > 0)
            port = uri->port;

        for (i = 0; i < uri->paramsCount; i++) {
            virURIParamPtr param = uri->params + i;

            if (STRCASEEQ(param->name, "tlsPort")) {
                if (virStrToLong_i(param->value, NULL, 10, &tlsPort) < 0) {
                    virReportError(VIR_ERR_INVALID_ARG,
                                   _("invalid tlsPort number: %s"),
                                   param->value);
                    goto cleanup;
                }
            } else if (STRCASEEQ(param->name, "tlsSubject")) {
                tlsSubject = param->value;
            }
        }
    }
1710 1711 1712 1713

    /* QEMU doesn't support VNC relocation yet, so
     * skip it to avoid generating an error
     */
1714 1715 1716 1717
    if (type != VIR_DOMAIN_GRAPHICS_TYPE_SPICE) {
        ret = 0;
        goto cleanup;
    }
1718

1719 1720 1721 1722 1723 1724 1725 1726 1727
    /* Older libvirt sends port == 0 for listen type='none' graphics. It's
     * safe to ignore such requests since relocation to unknown port does
     * not make sense in general.
     */
    if (port <= 0 && tlsPort <= 0) {
        ret = 0;
        goto cleanup;
    }

1728 1729 1730 1731
    if (qemuDomainObjEnterMonitorAsync(driver, vm,
                                       QEMU_ASYNC_JOB_MIGRATION_OUT) == 0) {
        ret = qemuMonitorGraphicsRelocate(priv->mon, type, listenAddress,
                                          port, tlsPort, tlsSubject);
1732
        priv->job.spiceMigration = !ret;
1733 1734
        if (qemuDomainObjExitMonitor(driver, vm) < 0)
            ret = -1;
1735
    }
1736

1737
 cleanup:
1738
    virURIFree(uri);
1739 1740 1741 1742
    return ret;
}


1743
static int
1744
qemuDomainMigrateOPDRelocate(virQEMUDriverPtr driver ATTRIBUTE_UNUSED,
1745 1746 1747
                             virDomainObjPtr vm,
                             qemuMigrationCookiePtr cookie)
{
1748 1749
    virDomainNetDefPtr netptr;
    int ret = -1;
1750
    size_t i;
1751 1752 1753 1754 1755 1756 1757 1758

    for (i = 0; i < cookie->network->nnets; i++) {
        netptr = vm->def->nets[i];

        switch (cookie->network->net[i].vporttype) {
        case VIR_NETDEV_VPORT_PROFILE_NONE:
        case VIR_NETDEV_VPORT_PROFILE_8021QBG:
        case VIR_NETDEV_VPORT_PROFILE_8021QBH:
1759
           break;
1760
        case VIR_NETDEV_VPORT_PROFILE_OPENVSWITCH:
1761 1762
            if (virNetDevOpenvswitchSetMigrateData(cookie->network->net[i].portdata,
                                                   netptr->ifname) != 0) {
J
Jiri Denemark 已提交
1763 1764 1765
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("Unable to run command to set OVS port data for "
                                 "interface %s"), netptr->ifname);
1766 1767 1768
                goto cleanup;
            }
            break;
1769 1770 1771 1772 1773
        default:
            break;
        }
    }

1774
    ret = 0;
1775
 cleanup:
1776 1777 1778 1779
    return ret;
}


1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819
int
qemuMigrationCheckIncoming(virQEMUCapsPtr qemuCaps,
                           const char *migrateFrom)
{
    if (STRPREFIX(migrateFrom, "rdma")) {
        if (!virQEMUCapsGet(qemuCaps, QEMU_CAPS_MIGRATE_RDMA)) {
            virReportError(VIR_ERR_OPERATION_UNSUPPORTED, "%s",
                           _("incoming RDMA migration is not supported "
                             "with this QEMU binary"));
            return -1;
        }
    } else if (!STRPREFIX(migrateFrom, "tcp") &&
               !STRPREFIX(migrateFrom, "exec") &&
               !STRPREFIX(migrateFrom, "fd") &&
               !STRPREFIX(migrateFrom, "unix") &&
               STRNEQ(migrateFrom, "stdio")) {
        virReportError(VIR_ERR_OPERATION_UNSUPPORTED, "%s",
                       _("unknown migration protocol"));
        return -1;
    }

    return 0;
}


char *
qemuMigrationIncomingURI(const char *migrateFrom,
                         int migrateFd)
{
    char *uri = NULL;

    if (STREQ(migrateFrom, "stdio"))
        ignore_value(virAsprintf(&uri, "fd:%d", migrateFd));
    else
        ignore_value(VIR_STRDUP(uri, migrateFrom));

    return uri;
}


1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845
int
qemuMigrationRunIncoming(virQEMUDriverPtr driver,
                         virDomainObjPtr vm,
                         const char *uri,
                         qemuDomainAsyncJob asyncJob)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret = -1;
    int rv;

    VIR_DEBUG("Setting up incoming migration with URI %s", uri);

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

    rv = qemuMonitorMigrateIncoming(priv->mon, uri);

    if (qemuDomainObjExitMonitor(driver, vm) < 0 || rv < 0)
        goto cleanup;

    if (asyncJob == QEMU_ASYNC_JOB_MIGRATION_IN) {
        /* qemuMigrationWaitForDestCompletion is called from the Finish phase */
        ret = 0;
        goto cleanup;
    }

1846
    if (qemuMigrationWaitForDestCompletion(driver, vm, asyncJob, false) < 0)
1847 1848 1849 1850 1851 1852 1853 1854 1855
        goto cleanup;

    ret = 0;

 cleanup:
    return ret;
}


1856 1857 1858 1859 1860 1861
/* This is called for outgoing non-p2p migrations when a connection to the
 * client which initiated the migration was closed but we were waiting for it
 * to follow up with the next phase, that is, in between
 * qemuDomainMigrateBegin3 and qemuDomainMigratePerform3 or
 * qemuDomainMigratePerform3 and qemuDomainMigrateConfirm3.
 */
1862 1863 1864 1865
static virDomainObjPtr
qemuMigrationCleanup(virDomainObjPtr vm,
                     virConnectPtr conn,
                     void *opaque)
1866
{
1867
    virQEMUDriverPtr driver = opaque;
1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882
    qemuDomainObjPrivatePtr priv = vm->privateData;

    VIR_DEBUG("vm=%s, conn=%p, asyncJob=%s, phase=%s",
              vm->def->name, conn,
              qemuDomainAsyncJobTypeToString(priv->job.asyncJob),
              qemuDomainAsyncJobPhaseToString(priv->job.asyncJob,
                                              priv->job.phase));

    if (!qemuMigrationJobIsActive(vm, QEMU_ASYNC_JOB_MIGRATION_OUT))
        goto cleanup;

    VIR_DEBUG("The connection which started outgoing migration of domain %s"
              " was closed; canceling the migration",
              vm->def->name);

1883
    switch ((qemuMigrationJobPhase) priv->job.phase) {
1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913
    case QEMU_MIGRATION_PHASE_BEGIN3:
        /* just forget we were about to migrate */
        qemuDomainObjDiscardAsyncJob(driver, vm);
        break;

    case QEMU_MIGRATION_PHASE_PERFORM3_DONE:
        VIR_WARN("Migration of domain %s finished but we don't know if the"
                 " domain was successfully started on destination or not",
                 vm->def->name);
        /* clear the job and let higher levels decide what to do */
        qemuDomainObjDiscardAsyncJob(driver, vm);
        break;

    case QEMU_MIGRATION_PHASE_PERFORM3:
        /* cannot be seen without an active migration API; unreachable */
    case QEMU_MIGRATION_PHASE_CONFIRM3:
    case QEMU_MIGRATION_PHASE_CONFIRM3_CANCELLED:
        /* all done; unreachable */
    case QEMU_MIGRATION_PHASE_PREPARE:
    case QEMU_MIGRATION_PHASE_FINISH2:
    case QEMU_MIGRATION_PHASE_FINISH3:
        /* incoming migration; unreachable */
    case QEMU_MIGRATION_PHASE_PERFORM2:
        /* single phase outgoing migration; unreachable */
    case QEMU_MIGRATION_PHASE_NONE:
    case QEMU_MIGRATION_PHASE_LAST:
        /* unreachable */
        ;
    }

1914
 cleanup:
1915 1916 1917
    return vm;
}

1918

1919
/* The caller is supposed to lock the vm and start a migration job. */
1920 1921 1922 1923 1924 1925 1926
static char *
qemuMigrationBeginPhase(virQEMUDriverPtr driver,
                        virDomainObjPtr vm,
                        const char *xmlin,
                        const char *dname,
                        char **cookieout,
                        int *cookieoutlen,
1927 1928
                        size_t nmigrate_disks,
                        const char **migrate_disks,
1929
                        unsigned long flags)
1930 1931 1932
{
    char *rv = NULL;
    qemuMigrationCookiePtr mig = NULL;
1933
    virDomainDefPtr def = NULL;
1934
    qemuDomainObjPrivatePtr priv = vm->privateData;
1935
    virCapsPtr caps = NULL;
1936
    unsigned int cookieFlags = QEMU_MIGRATION_COOKIE_LOCKSTATE;
1937

1938
    VIR_DEBUG("driver=%p, vm=%p, xmlin=%s, dname=%s,"
1939
              " cookieout=%p, cookieoutlen=%p,"
1940
              " nmigrate_disks=%zu, migrate_disks=%p, flags=0x%lx",
1941
              driver, vm, NULLSTR(xmlin), NULLSTR(dname),
1942 1943
              cookieout, cookieoutlen, nmigrate_disks,
              migrate_disks, flags);
1944

1945 1946 1947
    if (!(caps = virQEMUDriverGetCapabilities(driver, false)))
        goto cleanup;

1948 1949 1950 1951 1952 1953
    /* Only set the phase if we are inside QEMU_ASYNC_JOB_MIGRATION_OUT.
     * Otherwise we will start the async job later in the perform phase losing
     * change protection.
     */
    if (priv->job.asyncJob == QEMU_ASYNC_JOB_MIGRATION_OUT)
        qemuMigrationJobSetPhase(driver, vm, QEMU_MIGRATION_PHASE_BEGIN3);
1954

1955
    if (!qemuMigrationIsAllowed(driver, vm, true, flags))
1956 1957
        goto cleanup;

1958
    if (!(flags & (VIR_MIGRATE_UNSAFE | VIR_MIGRATE_OFFLINE)) &&
1959
        !qemuMigrationIsSafe(vm->def, nmigrate_disks, migrate_disks, flags))
1960 1961
        goto cleanup;

1962 1963 1964 1965 1966 1967 1968 1969 1970
    if (flags & VIR_MIGRATE_POSTCOPY &&
        (!(flags & VIR_MIGRATE_LIVE) ||
         flags & VIR_MIGRATE_PAUSED)) {
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("post-copy migration is not supported with non-live "
                         "or paused migration"));
        goto cleanup;
    }

1971 1972 1973 1974 1975 1976
    if (flags & VIR_MIGRATE_POSTCOPY && flags & VIR_MIGRATE_TUNNELLED) {
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("post-copy is not supported with tunnelled migration"));
        goto cleanup;
    }

1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019
    if (flags & (VIR_MIGRATE_NON_SHARED_DISK | VIR_MIGRATE_NON_SHARED_INC)) {
        bool has_drive_mirror =  virQEMUCapsGet(priv->qemuCaps,
                                                QEMU_CAPS_DRIVE_MIRROR);

        if (nmigrate_disks) {
            if (has_drive_mirror) {
                size_t i, j;
                /* Check user requested only known disk targets. */
                for (i = 0; i < nmigrate_disks; i++) {
                    for (j = 0; j < vm->def->ndisks; j++) {
                        if (STREQ(vm->def->disks[j]->dst, migrate_disks[i]))
                            break;
                    }

                    if (j == vm->def->ndisks) {
                        virReportError(VIR_ERR_INVALID_ARG,
                                       _("disk target %s not found"),
                                       migrate_disks[i]);
                        goto cleanup;
                    }
                }

                if (flags & VIR_MIGRATE_TUNNELLED) {
                    virReportError(VIR_ERR_OPERATION_UNSUPPORTED, "%s",
                                   _("Selecting disks to migrate is not "
                                     "implemented for tunnelled migration"));
                    goto cleanup;
                }
            } else {
                virReportError(VIR_ERR_OPERATION_UNSUPPORTED, "%s",
                               _("qemu does not support drive-mirror command"));
                goto cleanup;
            }
        }

        if (has_drive_mirror) {
            /* TODO support NBD for TUNNELLED migration */
            if (flags & VIR_MIGRATE_TUNNELLED) {
                VIR_WARN("NBD in tunnelled migration is currently not supported");
            } else {
                cookieFlags |= QEMU_MIGRATION_COOKIE_NBD;
                priv->nbdPort = 0;
            }
2020 2021 2022
        }
    }

2023
    if (virDomainDefHasMemoryHotplug(vm->def) ||
2024
        ((flags & VIR_MIGRATE_PERSIST_DEST) &&
2025
         vm->newDef && virDomainDefHasMemoryHotplug(vm->newDef)))
2026 2027
        cookieFlags |= QEMU_MIGRATION_COOKIE_MEMORY_HOTPLUG;

2028 2029 2030 2031 2032
    if (!qemuDomainVcpuHotplugIsInOrder(vm->def) ||
        ((flags & VIR_MIGRATE_PERSIST_DEST) &&
         vm->newDef && !qemuDomainVcpuHotplugIsInOrder(vm->newDef)))
        cookieFlags |= QEMU_MIGRATION_COOKIE_CPU_HOTPLUG;

2033 2034 2035
    if (priv->origCPU)
        cookieFlags |= QEMU_MIGRATION_COOKIE_CPU;

2036 2037
    cookieFlags |= QEMU_MIGRATION_COOKIE_ALLOW_REBOOT;

2038
    if (!(mig = qemuMigrationEatCookie(driver, vm, NULL, 0, 0)))
2039 2040 2041 2042
        goto cleanup;

    if (qemuMigrationBakeCookie(mig, driver, vm,
                                cookieout, cookieoutlen,
2043
                                cookieFlags) < 0)
2044 2045
        goto cleanup;

L
liguang 已提交
2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067
    if (flags & VIR_MIGRATE_OFFLINE) {
        if (flags & (VIR_MIGRATE_NON_SHARED_DISK |
                     VIR_MIGRATE_NON_SHARED_INC)) {
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                           _("offline migration cannot handle "
                             "non-shared storage"));
            goto cleanup;
        }
        if (!(flags & VIR_MIGRATE_PERSIST_DEST)) {
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                           _("offline migration must be specified with "
                             "the persistent flag set"));
            goto cleanup;
        }
        if (flags & VIR_MIGRATE_TUNNELLED) {
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                           _("tunnelled offline migration does not "
                             "make sense"));
            goto cleanup;
        }
    }

2068
    if (xmlin) {
2069
        if (!(def = virDomainDefParseString(xmlin, caps, driver->xmlopt, priv->qemuCaps,
2070 2071
                                            VIR_DOMAIN_DEF_PARSE_INACTIVE |
                                            VIR_DOMAIN_DEF_PARSE_SKIP_VALIDATE)))
2072 2073
            goto cleanup;

2074
        if (!qemuDomainCheckABIStability(driver, vm, def))
2075 2076
            goto cleanup;

2077
        rv = qemuDomainDefFormatLive(driver, def, NULL, false, true);
2078
    } else {
2079 2080
        rv = qemuDomainDefFormatLive(driver, vm->def, priv->origCPU,
                                     false, true);
2081
    }
2082

2083
 cleanup:
2084
    qemuMigrationCookieFree(mig);
2085
    virObjectUnref(caps);
2086
    virDomainDefFree(def);
2087 2088 2089
    return rv;
}

2090 2091 2092 2093 2094 2095 2096
char *
qemuMigrationBegin(virConnectPtr conn,
                   virDomainObjPtr vm,
                   const char *xmlin,
                   const char *dname,
                   char **cookieout,
                   int *cookieoutlen,
2097 2098
                   size_t nmigrate_disks,
                   const char **migrate_disks,
2099 2100 2101
                   unsigned long flags)
{
    virQEMUDriverPtr driver = conn->privateData;
2102
    virQEMUDriverConfigPtr cfg = NULL;
2103
    char *xml = NULL;
2104
    qemuDomainAsyncJob asyncJob;
2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115

    if ((flags & VIR_MIGRATE_CHANGE_PROTECTION)) {
        if (qemuMigrationJobStart(driver, vm, QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
            goto cleanup;
        asyncJob = QEMU_ASYNC_JOB_MIGRATION_OUT;
    } else {
        if (qemuDomainObjBeginJob(driver, vm, QEMU_JOB_MODIFY) < 0)
            goto cleanup;
        asyncJob = QEMU_ASYNC_JOB_NONE;
    }

2116 2117
    qemuMigrationStoreDomainState(vm);

2118 2119 2120 2121 2122 2123 2124 2125 2126 2127
    if (!virDomainObjIsActive(vm) && !(flags & VIR_MIGRATE_OFFLINE)) {
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s", _("domain is not running"));
        goto endjob;
    }

    /* Check if there is any ejected media.
     * We don't want to require them on the destination.
     */
    if (!(flags & VIR_MIGRATE_OFFLINE) &&
2128
        qemuProcessRefreshDisks(driver, vm, asyncJob) < 0)
2129 2130 2131 2132
        goto endjob;

    if (!(xml = qemuMigrationBeginPhase(driver, vm, xmlin, dname,
                                        cookieout, cookieoutlen,
2133
                                        nmigrate_disks, migrate_disks, flags)))
2134 2135
        goto endjob;

2136 2137 2138 2139 2140 2141
    if (flags & VIR_MIGRATE_TLS) {
        cfg = virQEMUDriverGetConfig(driver);
        if (qemuMigrationCheckSetupTLS(conn, driver, cfg, vm, asyncJob) < 0)
            goto endjob;
    }

2142 2143 2144 2145 2146
    if ((flags & VIR_MIGRATE_CHANGE_PROTECTION)) {
        /* We keep the job active across API calls until the confirm() call.
         * This prevents any other APIs being invoked while migration is taking
         * place.
         */
2147
        if (virCloseCallbacksSet(driver->closeCallbacks, vm, conn,
2148 2149
                                 qemuMigrationCleanup) < 0) {
            VIR_FREE(xml);
2150
            goto endjob;
2151
        }
2152
        qemuMigrationJobContinue(vm);
2153 2154 2155 2156
    } else {
        goto endjob;
    }

2157
 cleanup:
2158
    virObjectUnref(cfg);
M
Michal Privoznik 已提交
2159
    virDomainObjEndAPI(&vm);
2160 2161
    return xml;

2162
 endjob:
2163 2164 2165 2166
    if (flags & VIR_MIGRATE_CHANGE_PROTECTION)
        qemuMigrationJobFinish(driver, vm);
    else
        qemuDomainObjEndJob(driver, vm);
2167 2168 2169
    goto cleanup;
}

2170

2171 2172
/* Prepare is the first step, and it runs on the destination host.
 */
2173

2174
static void
2175
qemuMigrationPrepareCleanup(virQEMUDriverPtr driver,
2176 2177 2178 2179 2180 2181 2182 2183 2184 2185
                            virDomainObjPtr vm)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;

    VIR_DEBUG("driver=%p, vm=%s, job=%s, asyncJob=%s",
              driver,
              vm->def->name,
              qemuDomainJobTypeToString(priv->job.active),
              qemuDomainAsyncJobTypeToString(priv->job.asyncJob));

2186 2187 2188
    virPortAllocatorRelease(driver->migrationPorts, priv->migrationPort);
    priv->migrationPort = 0;

2189 2190 2191 2192 2193
    if (!qemuMigrationJobIsActive(vm, QEMU_ASYNC_JOB_MIGRATION_IN))
        return;
    qemuDomainObjDiscardAsyncJob(driver, vm);
}

2194
static qemuProcessIncomingDefPtr
2195 2196 2197 2198
qemuMigrationPrepareIncoming(virDomainObjPtr vm,
                             bool tunnel,
                             const char *protocol,
                             const char *listenAddress,
2199 2200
                             unsigned short port,
                             int fd)
2201 2202
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
2203
    qemuProcessIncomingDefPtr inc = NULL;
2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265
    char *migrateFrom = NULL;

    if (tunnel) {
        if (VIR_STRDUP(migrateFrom, "stdio") < 0)
            goto cleanup;
    } else {
        bool encloseAddress = false;
        bool hostIPv6Capable = false;
        bool qemuIPv6Capable = false;
        struct addrinfo *info = NULL;
        struct addrinfo hints = { .ai_flags = AI_ADDRCONFIG,
                                  .ai_socktype = SOCK_STREAM };
        const char *incFormat;

        if (getaddrinfo("::", NULL, &hints, &info) == 0) {
            freeaddrinfo(info);
            hostIPv6Capable = true;
        }
        qemuIPv6Capable = virQEMUCapsGet(priv->qemuCaps,
                                         QEMU_CAPS_IPV6_MIGRATION);

        if (listenAddress) {
            if (virSocketAddrNumericFamily(listenAddress) == AF_INET6) {
                if (!qemuIPv6Capable) {
                    virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                                   _("qemu isn't capable of IPv6"));
                    goto cleanup;
                }
                if (!hostIPv6Capable) {
                    virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                                   _("host isn't capable of IPv6"));
                    goto cleanup;
                }
                /* IPv6 address must be escaped in brackets on the cmd line */
                encloseAddress = true;
            } else {
                /* listenAddress is a hostname or IPv4 */
            }
        } else if (qemuIPv6Capable && hostIPv6Capable) {
            /* Listen on :: instead of 0.0.0.0 if QEMU understands it
             * and there is at least one IPv6 address configured
             */
            listenAddress = "::";
            encloseAddress = true;
        } else {
            listenAddress = "0.0.0.0";
        }

        /* QEMU will be started with
         *   -incoming protocol:[<IPv6 addr>]:port,
         *   -incoming protocol:<IPv4 addr>:port, or
         *   -incoming protocol:<hostname>:port
         */
        if (encloseAddress)
            incFormat = "%s:[%s]:%d";
        else
            incFormat = "%s:%s:%d";
        if (virAsprintf(&migrateFrom, incFormat,
                        protocol, listenAddress, port) < 0)
            goto cleanup;
    }

2266 2267
    inc = qemuProcessIncomingDefNew(priv->qemuCaps, listenAddress,
                                    migrateFrom, fd, NULL);
2268

2269
 cleanup:
2270 2271
    VIR_FREE(migrateFrom);
    return inc;
2272 2273
}

2274 2275 2276 2277
static int
qemuMigrationSetCompression(virQEMUDriverPtr driver,
                            virDomainObjPtr vm,
                            qemuDomainAsyncJob job,
2278 2279
                            qemuMigrationCompressionPtr compression,
                            qemuMonitorMigrationParamsPtr migParams)
2280
{
2281 2282 2283
    int ret = -1;
    qemuDomainObjPrivatePtr priv = vm->privateData;

2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297
    if (qemuMigrationSetOption(driver, vm,
                               QEMU_MONITOR_MIGRATION_CAPS_XBZRLE,
                               compression->methods &
                                   (1ULL << QEMU_MIGRATION_COMPRESS_XBZRLE),
                               job) < 0)
        return -1;

    if (qemuMigrationSetOption(driver, vm,
                               QEMU_MONITOR_MIGRATION_CAPS_COMPRESS,
                               compression->methods &
                                   (1ULL << QEMU_MIGRATION_COMPRESS_MT),
                               job) < 0)
        return -1;

2298 2299 2300
    if (qemuDomainObjEnterMonitorAsync(driver, vm, job) < 0)
        return -1;

2301 2302
    migParams->compressLevel_set = compression->level_set;
    migParams->compressLevel = compression->level;
2303

2304 2305
    migParams->compressThreads_set = compression->threads_set;
    migParams->compressThreads = compression->threads;
2306

2307 2308
    migParams->decompressThreads_set = compression->dthreads_set;
    migParams->decompressThreads = compression->dthreads;
2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321

    if (compression->xbzrle_cache_set &&
        qemuMonitorSetMigrationCacheSize(priv->mon,
                                         compression->xbzrle_cache) < 0)
        goto cleanup;

    ret = 0;

 cleanup:
    if (qemuDomainObjExitMonitor(driver, vm) < 0)
        ret = -1;

    return ret;
2322 2323
}

2324

2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346
void
qemuMigrationParamsClear(qemuMonitorMigrationParamsPtr migParams)
{
    if (!migParams)
        return;

    VIR_FREE(migParams->migrateTLSAlias);
    VIR_FREE(migParams->migrateTLSHostname);
}


void
qemuMigrationParamsFree(qemuMonitorMigrationParamsPtr *migParams)
{
    if (!*migParams)
        return;

    qemuMigrationParamsClear(*migParams);
    VIR_FREE(*migParams);
}


2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380
/* qemuMigrationSetEmptyTLSParams
 * @driver: pointer to qemu driver
 * @vm: domain object
 * @asyncJob: migration job to join
 * @migParams: Pointer to a migration parameters block
 *
 * If we support setting the tls-creds, then set both tls-creds and
 * tls-hostname to the empty string ("") which indicates to not use
 * TLS on this migration.
 *
 * Returns 0 on success, -1 on failure
 */
static int
qemuMigrationSetEmptyTLSParams(virQEMUDriverPtr driver,
                               virDomainObjPtr vm,
                               qemuDomainAsyncJob asyncJob,
                               qemuMonitorMigrationParamsPtr migParams)
{
   qemuDomainObjPrivatePtr priv = vm->privateData;

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

   if (!priv->migTLSAlias)
       return 0;

   if (VIR_STRDUP(migParams->migrateTLSAlias, "") < 0 ||
       VIR_STRDUP(migParams->migrateTLSHostname, "") < 0)
       return -1;

    return 0;
}


2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421
qemuMonitorMigrationParamsPtr
qemuMigrationParams(virTypedParameterPtr params,
                    int nparams,
                    unsigned long flags)
{
    qemuMonitorMigrationParamsPtr migParams;

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

    if (!params)
        return migParams;

#define GET(PARAM, VAR)                                                     \
    do {                                                                    \
        int rc;                                                             \
        if ((rc = virTypedParamsGetInt(params, nparams,                     \
                                       VIR_MIGRATE_PARAM_ ## PARAM,         \
                                       &migParams->VAR)) < 0)               \
            goto error;                                                     \
                                                                            \
        if (rc == 1)                                                        \
            migParams->VAR ## _set = true;                                  \
    } while (0)

    GET(AUTO_CONVERGE_INITIAL, cpuThrottleInitial);
    GET(AUTO_CONVERGE_INCREMENT, cpuThrottleIncrement);

#undef GET

    if ((migParams->cpuThrottleInitial_set ||
         migParams->cpuThrottleIncrement_set) &&
        !(flags & VIR_MIGRATE_AUTO_CONVERGE)) {
        virReportError(VIR_ERR_INVALID_ARG, "%s",
                       _("Turn auto convergence on to tune it"));
        goto error;
    }

    return migParams;

 error:
2422
    qemuMigrationParamsFree(&migParams);
2423 2424 2425 2426
    return NULL;
}


2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451
static int
qemuMigrationSetParams(virQEMUDriverPtr driver,
                       virDomainObjPtr vm,
                       qemuDomainAsyncJob job,
                       qemuMonitorMigrationParamsPtr migParams)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret = -1;

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

    if (qemuMonitorSetMigrationParams(priv->mon, migParams) < 0)
        goto cleanup;

    ret = 0;

 cleanup:
    if (qemuDomainObjExitMonitor(driver, vm) < 0)
        ret = -1;

    return ret;
}


2452 2453 2454 2455 2456 2457
/* qemuMigrationResetTLS
 * @driver: pointer to qemu driver
 * @vm: domain object
 * @asyncJob: migration job to join
 *
 * Deconstruct all the setup possibly done for TLS - delete the TLS and
2458
 * security objects, free the secinfo, and reset the migration params to "".
2459 2460 2461
 *
 * Returns 0 on success, -1 on failure
 */
2462
static int
2463 2464
qemuMigrationResetTLS(virQEMUDriverPtr driver,
                      virDomainObjPtr vm,
2465
                      qemuDomainAsyncJob asyncJob)
2466 2467
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
2468 2469
    char *tlsAlias = NULL;
    char *secAlias = NULL;
2470
    qemuMonitorMigrationParams migParams = { 0 };
R
Roman Bogorodskiy 已提交
2471
    int ret = -1;
2472

2473
    if (qemuMigrationCheckTLSCreds(driver, vm, asyncJob) < 0)
2474 2475
        return -1;

2476 2477
    /* If the tls-creds doesn't exist or if they're set to "" then there's
     * nothing to do since we never set anything up */
2478 2479 2480 2481 2482 2483
    if (!priv->migTLSAlias || !*priv->migTLSAlias)
        return 0;

    /* NB: If either or both fail to allocate memory we can still proceed
     *     since the next time we migrate another deletion attempt will be
     *     made after successfully generating the aliases. */
2484 2485
    tlsAlias = qemuAliasTLSObjFromSrcAlias(QEMU_MIGRATION_TLS_ALIAS_BASE);
    secAlias = qemuDomainGetSecretAESAlias(QEMU_MIGRATION_TLS_ALIAS_BASE, false);
2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497

    qemuDomainDelTLSObjects(driver, vm, asyncJob, secAlias, tlsAlias);
    qemuDomainSecretInfoFree(&priv->migSecinfo);

    if (VIR_STRDUP(migParams.migrateTLSAlias, "") < 0 ||
        VIR_STRDUP(migParams.migrateTLSHostname, "") < 0 ||
        qemuMigrationSetParams(driver, vm, asyncJob, &migParams) < 0)
        goto cleanup;

    ret = 0;

 cleanup:
2498 2499
    VIR_FREE(tlsAlias);
    VIR_FREE(secAlias);
2500 2501 2502 2503 2504 2505
    qemuMigrationParamsClear(&migParams);

    return ret;
}


2506
static int
2507
qemuMigrationPrepareAny(virQEMUDriverPtr driver,
2508 2509 2510 2511 2512
                        virConnectPtr dconn,
                        const char *cookiein,
                        int cookieinlen,
                        char **cookieout,
                        int *cookieoutlen,
2513
                        virDomainDefPtr *def,
2514
                        const char *origname,
L
liguang 已提交
2515
                        virStreamPtr st,
2516
                        const char *protocol,
2517 2518
                        unsigned short port,
                        bool autoPort,
2519
                        const char *listenAddress,
2520 2521
                        size_t nmigrate_disks,
                        const char **migrate_disks,
2522
                        int nbdPort,
2523
                        qemuMigrationCompressionPtr compression,
L
liguang 已提交
2524
                        unsigned long flags)
2525 2526
{
    virDomainObjPtr vm = NULL;
2527
    virObjectEventPtr event = NULL;
2528
    virQEMUDriverConfigPtr cfg = NULL;
2529
    int ret = -1;
2530
    int dataFD[2] = { -1, -1 };
2531
    qemuDomainObjPrivatePtr priv = NULL;
2532
    qemuMigrationCookiePtr mig = NULL;
2533
    bool tunnel = !!st;
J
Jiri Denemark 已提交
2534
    char *xmlout = NULL;
L
liguang 已提交
2535
    unsigned int cookieFlags;
2536
    unsigned int startFlags;
2537
    virCapsPtr caps = NULL;
2538
    qemuProcessIncomingDefPtr incoming = NULL;
2539
    bool taint_hook = false;
2540 2541 2542
    bool stopProcess = false;
    bool relabel = false;
    int rv;
2543 2544
    char *tlsAlias = NULL;
    char *secAlias = NULL;
2545
    qemuMonitorMigrationParams migParams = { 0 };
2546

2547 2548
    virNWFilterReadLockFilterUpdates();

L
liguang 已提交
2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568
    if (flags & VIR_MIGRATE_OFFLINE) {
        if (flags & (VIR_MIGRATE_NON_SHARED_DISK |
                     VIR_MIGRATE_NON_SHARED_INC)) {
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                           _("offline migration cannot handle "
                             "non-shared storage"));
            goto cleanup;
        }
        if (!(flags & VIR_MIGRATE_PERSIST_DEST)) {
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                           _("offline migration must be specified with "
                             "the persistent flag set"));
            goto cleanup;
        }
        if (tunnel) {
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                           _("tunnelled offline migration does not "
                             "make sense"));
            goto cleanup;
        }
2569 2570 2571
        cookieFlags = 0;
    } else {
        cookieFlags = QEMU_MIGRATION_COOKIE_GRAPHICS;
L
liguang 已提交
2572 2573
    }

2574 2575 2576 2577 2578 2579 2580 2581 2582
    if (flags & VIR_MIGRATE_POSTCOPY &&
        (!(flags & VIR_MIGRATE_LIVE) ||
         flags & VIR_MIGRATE_PAUSED)) {
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("post-copy migration is not supported with non-live "
                         "or paused migration"));
        goto cleanup;
    }

2583 2584 2585 2586 2587 2588
    if (flags & VIR_MIGRATE_POSTCOPY && flags & VIR_MIGRATE_TUNNELLED) {
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("post-copy is not supported with tunnelled migration"));
        goto cleanup;
    }

2589 2590 2591
    if (!(caps = virQEMUDriverGetCapabilities(driver, false)))
        goto cleanup;

2592
    if (!qemuMigrationIsAllowedHostdev(*def))
2593 2594
        goto cleanup;

J
Jiri Denemark 已提交
2595 2596 2597 2598 2599
    /* Let migration hook filter domain XML */
    if (virHookPresent(VIR_HOOK_DRIVER_QEMU)) {
        char *xml;
        int hookret;

2600
        if (!(xml = qemuDomainDefFormatXML(driver, *def,
2601 2602
                                           VIR_DOMAIN_XML_SECURE |
                                           VIR_DOMAIN_XML_MIGRATABLE)))
J
Jiri Denemark 已提交
2603 2604
            goto cleanup;

2605
        hookret = virHookCall(VIR_HOOK_DRIVER_QEMU, (*def)->name,
J
Jiri Denemark 已提交
2606 2607 2608 2609 2610 2611 2612
                              VIR_HOOK_QEMU_OP_MIGRATE, VIR_HOOK_SUBOP_BEGIN,
                              NULL, xml, &xmlout);
        VIR_FREE(xml);

        if (hookret < 0) {
            goto cleanup;
        } else if (hookret == 0) {
2613
            if (virStringIsEmpty(xmlout)) {
J
Jiri Denemark 已提交
2614 2615 2616 2617 2618 2619
                VIR_DEBUG("Migrate hook filter returned nothing; using the"
                          " original XML");
            } else {
                virDomainDefPtr newdef;

                VIR_DEBUG("Using hook-filtered domain XML: %s", xmlout);
2620
                newdef = virDomainDefParseString(xmlout, caps, driver->xmlopt, NULL,
2621 2622
                                                 VIR_DOMAIN_DEF_PARSE_INACTIVE |
                                                 VIR_DOMAIN_DEF_PARSE_SKIP_VALIDATE);
J
Jiri Denemark 已提交
2623 2624 2625
                if (!newdef)
                    goto cleanup;

2626
                if (!qemuDomainDefCheckABIStability(driver, *def, newdef)) {
J
Jiri Denemark 已提交
2627 2628 2629 2630
                    virDomainDefFree(newdef);
                    goto cleanup;
                }

2631 2632
                virDomainDefFree(*def);
                *def = newdef;
2633 2634 2635 2636
                /* We should taint the domain here. However, @vm and therefore
                 * privateData too are still NULL, so just notice the fact and
                 * taint it later. */
                taint_hook = true;
J
Jiri Denemark 已提交
2637 2638 2639 2640
            }
        }
    }

2641
    if (!(vm = virDomainObjListAdd(driver->domains, *def,
2642
                                   driver->xmlopt,
2643 2644 2645
                                   VIR_DOMAIN_OBJ_LIST_ADD_LIVE |
                                   VIR_DOMAIN_OBJ_LIST_ADD_CHECK_LIVE,
                                   NULL)))
2646
        goto cleanup;
2647

2648
    virObjectRef(vm);
2649
    *def = NULL;
2650
    priv = vm->privateData;
2651 2652
    if (VIR_STRDUP(priv->origname, origname) < 0)
        goto cleanup;
2653

2654 2655 2656 2657 2658
    if (taint_hook) {
        /* Domain XML has been altered by a hook script. */
        priv->hookRun = true;
    }

2659
    if (!(mig = qemuMigrationEatCookie(driver, vm, cookiein, cookieinlen,
2660
                                       QEMU_MIGRATION_COOKIE_LOCKSTATE |
2661
                                       QEMU_MIGRATION_COOKIE_NBD |
2662
                                       QEMU_MIGRATION_COOKIE_MEMORY_HOTPLUG |
2663
                                       QEMU_MIGRATION_COOKIE_CPU_HOTPLUG |
2664 2665
                                       QEMU_MIGRATION_COOKIE_CPU |
                                       QEMU_MIGRATION_COOKIE_ALLOW_REBOOT)))
2666 2667
        goto cleanup;

2668 2669
    if (STREQ_NULLABLE(protocol, "rdma") &&
        !virMemoryLimitIsSet(vm->def->mem.hard_limit)) {
M
Michael R. Hines 已提交
2670 2671 2672 2673 2674 2675
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                       _("cannot start RDMA migration with no memory hard "
                         "limit set"));
        goto cleanup;
    }

2676
    if (qemuMigrationPrecreateStorage(dconn, driver, vm, mig->nbd,
2677 2678
                                      nmigrate_disks, migrate_disks,
                                      !!(flags & VIR_MIGRATE_NON_SHARED_INC)) < 0)
2679 2680
        goto cleanup;

2681
    if (qemuMigrationJobStart(driver, vm, QEMU_ASYNC_JOB_MIGRATION_IN) < 0)
2682
        goto cleanup;
2683
    qemuMigrationJobSetPhase(driver, vm, QEMU_MIGRATION_PHASE_PREPARE);
2684 2685 2686 2687

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

L
liguang 已提交
2688 2689 2690
    if (flags & VIR_MIGRATE_OFFLINE)
        goto done;

2691 2692
    if (tunnel &&
        (pipe(dataFD) < 0 || virSetCloseExec(dataFD[1]) < 0)) {
2693 2694
        virReportSystemError(errno, "%s",
                             _("cannot create pipe for tunnelled migration"));
2695
        goto stopjob;
2696 2697
    }

2698 2699
    startFlags = VIR_QEMU_PROCESS_START_AUTODESTROY;

2700
    if (qemuProcessInit(driver, vm, mig->cpu, QEMU_ASYNC_JOB_MIGRATION_IN,
2701
                        true, startFlags) < 0)
2702
        goto stopjob;
2703
    stopProcess = true;
2704

2705 2706
    priv->allowReboot = mig->allowReboot;

2707 2708 2709
    if (!(incoming = qemuMigrationPrepareIncoming(vm, tunnel, protocol,
                                                  listenAddress, port,
                                                  dataFD[0])))
2710
        goto stopjob;
2711

2712
    if (qemuProcessPrepareDomain(dconn, driver, vm, startFlags) < 0)
2713 2714
        goto stopjob;

2715
    if (qemuProcessPrepareHost(driver, vm, startFlags) < 0)
2716 2717
        goto stopjob;

2718 2719 2720
    rv = qemuProcessLaunch(dconn, driver, vm, QEMU_ASYNC_JOB_MIGRATION_IN,
                           incoming, NULL,
                           VIR_NETDEV_VPORT_PROFILE_OP_MIGRATE_IN_START,
2721
                           startFlags);
2722 2723 2724
    if (rv < 0) {
        if (rv == -2)
            relabel = true;
2725
        goto stopjob;
2726
    }
2727
    relabel = true;
2728

2729 2730 2731 2732
    if (tunnel) {
        if (virFDStreamOpen(st, dataFD[1]) < 0) {
            virReportSystemError(errno, "%s",
                                 _("cannot pass pipe for tunnelled migration"));
2733
            goto stopjob;
2734
        }
2735
        dataFD[1] = -1; /* 'st' owns the FD now & will close it */
2736 2737
    }

2738
    if (qemuMigrationSetCompression(driver, vm, QEMU_ASYNC_JOB_MIGRATION_IN,
2739
                                    compression, &migParams) < 0)
2740
        goto stopjob;
2741

2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765
    /* Migrations using TLS need to add the "tls-creds-x509" object and
     * set the migration TLS parameters */
    if (flags & VIR_MIGRATE_TLS) {
        cfg = virQEMUDriverGetConfig(driver);
        if (qemuMigrationCheckSetupTLS(dconn, driver, cfg, vm,
                                       QEMU_ASYNC_JOB_MIGRATION_IN) < 0)
            goto stopjob;

        if (qemuMigrationAddTLSObjects(driver, vm, cfg, true,
                                       QEMU_ASYNC_JOB_MIGRATION_IN,
                                       &tlsAlias, &secAlias, &migParams) < 0)
            goto stopjob;

        /* Force reset of 'tls-hostname', it's a source only parameter */
        if (VIR_STRDUP(migParams.migrateTLSHostname, "") < 0)
            goto stopjob;

    } else {
        if (qemuMigrationSetEmptyTLSParams(driver, vm,
                                           QEMU_ASYNC_JOB_MIGRATION_IN,
                                           &migParams) < 0)
            goto stopjob;
    }

2766
    if (STREQ_NULLABLE(protocol, "rdma") &&
M
Michael R. Hines 已提交
2767
        virProcessSetMaxMemLock(vm->pid, vm->def->mem.hard_limit << 10) < 0) {
2768
        goto stopjob;
M
Michael R. Hines 已提交
2769 2770
    }

2771 2772
    if (qemuMigrationSetOption(driver, vm,
                               QEMU_MONITOR_MIGRATION_CAPS_RDMA_PIN_ALL,
2773 2774
                               flags & VIR_MIGRATE_RDMA_PIN_ALL,
                               QEMU_ASYNC_JOB_MIGRATION_IN) < 0)
2775
        goto stopjob;
2776

2777 2778 2779 2780 2781
    if (qemuMigrationSetPostCopy(driver, vm,
                                 flags & VIR_MIGRATE_POSTCOPY,
                                 QEMU_ASYNC_JOB_MIGRATION_IN) < 0)
        goto stopjob;

2782 2783 2784 2785
    if (qemuMigrationSetParams(driver, vm, QEMU_ASYNC_JOB_MIGRATION_IN,
                               &migParams) < 0)
        goto stopjob;

2786 2787 2788
    if (mig->nbd &&
        flags & (VIR_MIGRATE_NON_SHARED_DISK | VIR_MIGRATE_NON_SHARED_INC) &&
        virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_NBD_SERVER)) {
2789
        if (qemuMigrationStartNBDServer(driver, vm, incoming->address,
2790 2791
                                        nmigrate_disks, migrate_disks,
                                        nbdPort) < 0) {
2792
            goto stopjob;
2793
        }
2794
        cookieFlags |= QEMU_MIGRATION_COOKIE_NBD;
2795 2796
    }

2797 2798 2799 2800 2801 2802 2803 2804 2805
    if (mig->lockState) {
        VIR_DEBUG("Received lockstate %s", mig->lockState);
        VIR_FREE(priv->lockState);
        priv->lockState = mig->lockState;
        mig->lockState = NULL;
    } else {
        VIR_DEBUG("Received no lockstate");
    }

2806 2807 2808 2809 2810 2811 2812 2813 2814
    if (incoming->deferredURI &&
        qemuMigrationRunIncoming(driver, vm, incoming->deferredURI,
                                 QEMU_ASYNC_JOB_MIGRATION_IN) < 0)
        goto stopjob;

    if (qemuProcessFinishStartup(dconn, driver, vm, QEMU_ASYNC_JOB_MIGRATION_IN,
                                 false, VIR_DOMAIN_PAUSED_MIGRATION) < 0)
        goto stopjob;

2815
 done:
2816 2817
    if (qemuMigrationBakeCookie(mig, driver, vm, cookieout,
                                cookieoutlen, cookieFlags) < 0) {
2818 2819 2820 2821 2822 2823 2824
        /* We could tear down the whole guest here, but
         * cookie data is (so far) non-critical, so that
         * seems a little harsh. We'll just warn for now.
         */
        VIR_WARN("Unable to encode migration cookie");
    }

2825
    if (qemuDomainCleanupAdd(vm, qemuMigrationPrepareCleanup) < 0)
2826
        goto stopjob;
2827

L
liguang 已提交
2828 2829
    if (!(flags & VIR_MIGRATE_OFFLINE)) {
        virDomainAuditStart(vm, "migrated", true);
2830
        event = virDomainEventLifecycleNewFromObj(vm,
L
liguang 已提交
2831 2832 2833
                                         VIR_DOMAIN_EVENT_STARTED,
                                         VIR_DOMAIN_EVENT_STARTED_MIGRATED);
    }
2834

2835 2836 2837 2838
    /* We keep the job active across API calls until the finish() call.
     * This prevents any other APIs being invoked while incoming
     * migration is taking place.
     */
2839
    qemuMigrationJobContinue(vm);
2840

2841 2842
    if (autoPort)
        priv->migrationPort = port;
2843 2844 2845 2846 2847
    /* in this case port is not auto selected and we don't need to manage it
     * anymore after cookie is baked
     */
    if (nbdPort != 0)
        priv->nbdPort = 0;
2848
    ret = 0;
2849

2850
 cleanup:
2851 2852 2853
    VIR_FREE(tlsAlias);
    VIR_FREE(secAlias);
    virObjectUnref(cfg);
2854
    qemuProcessIncomingDefFree(incoming);
J
Jiri Denemark 已提交
2855
    VIR_FREE(xmlout);
2856 2857
    VIR_FORCE_CLOSE(dataFD[0]);
    VIR_FORCE_CLOSE(dataFD[1]);
2858 2859 2860
    if (ret < 0 && priv) {
        /* priv is set right after vm is added to the list of domains
         * and there is no 'goto cleanup;' in the middle of those */
2861
        VIR_FREE(priv->origname);
2862 2863 2864 2865 2866
        /* release if port is auto selected which is not the case if
         * it is given in parameters
         */
        if (nbdPort == 0)
            virPortAllocatorRelease(driver->migrationPorts, priv->nbdPort);
2867
        priv->nbdPort = 0;
2868
        virDomainObjRemoveTransientDef(vm);
2869
        qemuDomainRemoveInactiveJob(driver, vm);
2870
    }
2871
    qemuMigrationParamsClear(&migParams);
M
Michal Privoznik 已提交
2872
    virDomainObjEndAPI(&vm);
2873
    qemuDomainEventQueue(driver, event);
2874
    qemuMigrationCookieFree(mig);
2875
    virObjectUnref(caps);
2876
    virNWFilterUnlockFilterUpdates();
2877
    return ret;
2878

2879
 stopjob:
2880
    qemuMigrationReset(driver, vm, QEMU_ASYNC_JOB_MIGRATION_IN);
2881

2882 2883 2884 2885
    if (stopProcess) {
        unsigned int stopFlags = VIR_QEMU_PROCESS_STOP_MIGRATED;
        if (!relabel)
            stopFlags |= VIR_QEMU_PROCESS_STOP_NO_RELABEL;
2886
        virDomainAuditStart(vm, "migrated", false);
2887 2888
        qemuProcessStop(driver, vm, VIR_DOMAIN_SHUTOFF_FAILED,
                        QEMU_ASYNC_JOB_MIGRATION_IN, stopFlags);
2889
    }
2890

2891
    qemuMigrationJobFinish(driver, vm);
2892
    goto cleanup;
2893 2894 2895
}


2896 2897 2898 2899 2900
/*
 * This version starts an empty VM listening on a localhost TCP port, and
 * sets up the corresponding virStream to handle the incoming data.
 */
int
2901
qemuMigrationPrepareTunnel(virQEMUDriverPtr driver,
2902 2903 2904 2905 2906 2907
                           virConnectPtr dconn,
                           const char *cookiein,
                           int cookieinlen,
                           char **cookieout,
                           int *cookieoutlen,
                           virStreamPtr st,
2908
                           virDomainDefPtr *def,
2909
                           const char *origname,
L
liguang 已提交
2910
                           unsigned long flags)
2911
{
2912
    qemuMigrationCompressionPtr compression = NULL;
2913 2914 2915
    int ret;

    VIR_DEBUG("driver=%p, dconn=%p, cookiein=%s, cookieinlen=%d, "
2916
              "cookieout=%p, cookieoutlen=%p, st=%p, def=%p, "
2917
              "origname=%s, flags=0x%lx",
2918
              driver, dconn, NULLSTR(cookiein), cookieinlen,
2919
              cookieout, cookieoutlen, st, *def, origname, flags);
2920

2921 2922 2923 2924 2925 2926
    if (st == NULL) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("tunnelled migration requested but NULL stream passed"));
        return -1;
    }

2927 2928 2929
    if (!(compression = qemuMigrationCompressionParse(NULL, 0, flags)))
        return -1;

2930
    ret = qemuMigrationPrepareAny(driver, dconn, cookiein, cookieinlen,
2931
                                  cookieout, cookieoutlen, def, origname,
2932 2933 2934
                                  st, NULL, 0, false, NULL, 0, NULL, 0,
                                  compression, flags);
    VIR_FREE(compression);
2935 2936 2937 2938
    return ret;
}


2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958 2959 2960 2961
static virURIPtr
qemuMigrationParseURI(const char *uri, bool *wellFormed)
{
    char *tmp = NULL;
    virURIPtr parsed;

    /* For compatibility reasons tcp://... URIs are sent as tcp:...
     * We need to transform them to a well-formed URI before parsing. */
    if (STRPREFIX(uri, "tcp:") && !STRPREFIX(uri + 4, "//")) {
        if (virAsprintf(&tmp, "tcp://%s", uri + 4) < 0)
            return NULL;
        uri = tmp;
    }

    parsed = virURIParse(uri);
    if (parsed && wellFormed)
        *wellFormed = !tmp;
    VIR_FREE(tmp);

    return parsed;
}


2962
int
2963
qemuMigrationPrepareDirect(virQEMUDriverPtr driver,
2964
                           virConnectPtr dconn,
2965 2966 2967 2968
                           const char *cookiein,
                           int cookieinlen,
                           char **cookieout,
                           int *cookieoutlen,
2969 2970
                           const char *uri_in,
                           char **uri_out,
2971
                           virDomainDefPtr *def,
2972
                           const char *origname,
2973
                           const char *listenAddress,
2974 2975
                           size_t nmigrate_disks,
                           const char **migrate_disks,
2976
                           int nbdPort,
2977
                           qemuMigrationCompressionPtr compression,
L
liguang 已提交
2978
                           unsigned long flags)
2979
{
2980 2981
    unsigned short port = 0;
    bool autoPort = true;
2982 2983
    char *hostname = NULL;
    int ret = -1;
2984
    virURIPtr uri = NULL;
2985 2986
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
    const char *migrateHost = cfg->migrateHost;
J
Jiri Denemark 已提交
2987

2988 2989
    VIR_DEBUG("driver=%p, dconn=%p, cookiein=%s, cookieinlen=%d, "
              "cookieout=%p, cookieoutlen=%p, uri_in=%s, uri_out=%p, "
2990
              "def=%p, origname=%s, listenAddress=%s, "
2991
              "nmigrate_disks=%zu, migrate_disks=%p, nbdPort=%d, flags=0x%lx",
2992 2993
              driver, dconn, NULLSTR(cookiein), cookieinlen,
              cookieout, cookieoutlen, NULLSTR(uri_in), uri_out,
2994
              *def, origname, NULLSTR(listenAddress),
2995
              nmigrate_disks, migrate_disks, nbdPort, flags);
2996

2997 2998
    *uri_out = NULL;

2999 3000 3001
    /* The URI passed in may be NULL or a string "tcp://somehostname:port".
     *
     * If the URI passed in is NULL then we allocate a port number
3002 3003 3004
     * from our pool of port numbers, and if the migrateHost is configured,
     * we return a URI of "tcp://migrateHost:port", otherwise return a URI
     * of "tcp://ourhostname:port".
3005 3006 3007 3008 3009 3010
     *
     * If the URI passed in is not NULL then we try to parse out the
     * port number and use that (note that the hostname is assumed
     * to be a correct hostname which refers to the target machine).
     */
    if (uri_in == NULL) {
3011 3012 3013
        bool encloseAddress = false;
        const char *incFormat;

3014
        if (virPortAllocatorAcquire(driver->migrationPorts, &port) < 0)
3015
            goto cleanup;
3016

3017
        if (migrateHost != NULL) {
3018 3019
            if (virSocketAddrNumericFamily(migrateHost) == AF_INET6)
                encloseAddress = true;
3020

3021
            if (VIR_STRDUP(hostname, migrateHost) < 0)
3022 3023 3024 3025 3026
                goto cleanup;
        } else {
            if ((hostname = virGetHostname()) == NULL)
                goto cleanup;
        }
3027 3028

        if (STRPREFIX(hostname, "localhost")) {
3029 3030 3031
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("hostname on destination resolved to localhost,"
                             " but migration requires an FQDN"));
3032 3033 3034 3035 3036
            goto cleanup;
        }

        /* XXX this really should have been a properly well-formed
         * URI, but we can't add in tcp:// now without breaking
3037
         * compatibility with old targets. We at least make the
3038 3039
         * new targets accept both syntaxes though.
         */
3040 3041 3042 3043 3044 3045
        if (encloseAddress)
            incFormat = "%s:[%s]:%d";
        else
            incFormat = "%s:%s:%d";

        if (virAsprintf(uri_out, incFormat, "tcp", hostname, port) < 0)
3046 3047
            goto cleanup;
    } else {
3048
        bool well_formed_uri;
J
Ján Tomko 已提交
3049

3050 3051
        if (!(uri = qemuMigrationParseURI(uri_in, &well_formed_uri)))
            goto cleanup;
J
Ján Tomko 已提交
3052

3053 3054 3055 3056 3057 3058 3059
        if (uri->scheme == NULL) {
            virReportError(VIR_ERR_INVALID_ARG,
                           _("missing scheme in migration URI: %s"),
                           uri_in);
            goto cleanup;
        }

M
Michael R. Hines 已提交
3060 3061
        if (STRNEQ(uri->scheme, "tcp") &&
            STRNEQ(uri->scheme, "rdma")) {
3062 3063 3064
            virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED,
                           _("unsupported scheme %s in migration URI %s"),
                           uri->scheme, uri_in);
J
Ján Tomko 已提交
3065 3066 3067 3068 3069 3070 3071 3072 3073 3074
            goto cleanup;
        }

        if (uri->server == NULL) {
            virReportError(VIR_ERR_INVALID_ARG, _("missing host in migration"
                                                  " URI: %s"), uri_in);
            goto cleanup;
        }

        if (uri->port == 0) {
3075
            if (virPortAllocatorAcquire(driver->migrationPorts, &port) < 0)
3076
                goto cleanup;
3077

3078
            /* Send well-formed URI only if uri_in was well-formed */
3079 3080 3081 3082 3083 3084 3085 3086
            if (well_formed_uri) {
                uri->port = port;
                if (!(*uri_out = virURIFormat(uri)))
                    goto cleanup;
            } else {
                if (virAsprintf(uri_out, "%s:%d", uri_in, port) < 0)
                    goto cleanup;
            }
3087
        } else {
3088 3089
            port = uri->port;
            autoPort = false;
3090 3091 3092 3093 3094 3095
        }
    }

    if (*uri_out)
        VIR_DEBUG("Generated uri_out=%s", *uri_out);

3096
    ret = qemuMigrationPrepareAny(driver, dconn, cookiein, cookieinlen,
3097
                                  cookieout, cookieoutlen, def, origname,
3098
                                  NULL, uri ? uri->scheme : "tcp",
3099
                                  port, autoPort, listenAddress,
3100 3101
                                  nmigrate_disks, migrate_disks, nbdPort,
                                  compression, flags);
3102
 cleanup:
3103
    virURIFree(uri);
3104
    VIR_FREE(hostname);
3105
    virObjectUnref(cfg);
3106
    if (ret != 0) {
3107
        VIR_FREE(*uri_out);
3108 3109 3110
        if (autoPort)
            virPortAllocatorRelease(driver->migrationPorts, port);
    }
3111 3112 3113 3114
    return ret;
}


3115 3116 3117
virDomainDefPtr
qemuMigrationPrepareDef(virQEMUDriverPtr driver,
                        const char *dom_xml,
3118 3119
                        const char *dname,
                        char **origname)
3120 3121 3122
{
    virCapsPtr caps = NULL;
    virDomainDefPtr def;
3123
    char *name = NULL;
3124 3125 3126 3127 3128 3129 3130 3131 3132 3133

    if (!dom_xml) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("no domain XML passed"));
        return NULL;
    }

    if (!(caps = virQEMUDriverGetCapabilities(driver, false)))
        return NULL;

3134
    if (!(def = virDomainDefParseString(dom_xml, caps, driver->xmlopt, NULL,
3135 3136
                                        VIR_DOMAIN_DEF_PARSE_INACTIVE |
                                        VIR_DOMAIN_DEF_PARSE_SKIP_VALIDATE)))
3137 3138 3139
        goto cleanup;

    if (dname) {
3140
        name = def->name;
3141 3142 3143 3144 3145 3146
        if (VIR_STRDUP(def->name, dname) < 0) {
            virDomainDefFree(def);
            def = NULL;
        }
    }

3147
 cleanup:
3148
    virObjectUnref(caps);
3149 3150 3151 3152
    if (def && origname)
        *origname = name;
    else
        VIR_FREE(name);
3153 3154 3155 3156
    return def;
}


3157 3158 3159 3160 3161 3162 3163 3164 3165 3166
static int
qemuMigrationConfirmPhase(virQEMUDriverPtr driver,
                          virConnectPtr conn,
                          virDomainObjPtr vm,
                          const char *cookiein,
                          int cookieinlen,
                          unsigned int flags,
                          int retcode)
{
    qemuMigrationCookiePtr mig;
3167
    virObjectEventPtr event;
3168 3169
    int rv = -1;
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
3170 3171
    qemuDomainObjPrivatePtr priv = vm->privateData;
    qemuDomainJobInfoPtr jobInfo = NULL;
3172 3173

    VIR_DEBUG("driver=%p, conn=%p, vm=%p, cookiein=%s, cookieinlen=%d, "
3174
              "flags=0x%x, retcode=%d",
3175 3176 3177 3178 3179 3180 3181 3182 3183 3184
              driver, conn, vm, NULLSTR(cookiein), cookieinlen,
              flags, retcode);

    virCheckFlags(QEMU_MIGRATION_FLAGS, -1);

    qemuMigrationJobSetPhase(driver, vm,
                             retcode == 0
                             ? QEMU_MIGRATION_PHASE_CONFIRM3
                             : QEMU_MIGRATION_PHASE_CONFIRM3_CANCELLED);

3185 3186
    if (!(mig = qemuMigrationEatCookie(driver, vm, cookiein, cookieinlen,
                                       QEMU_MIGRATION_COOKIE_STATS)))
3187 3188
        goto cleanup;

3189 3190 3191
    if (retcode == 0)
        jobInfo = priv->job.completed;
    else
3192
        VIR_FREE(priv->job.completed);
3193 3194 3195

    /* Update times with the values sent by the destination daemon */
    if (mig->jobInfo && jobInfo) {
3196 3197 3198 3199 3200 3201 3202 3203
        int reason;

        /* We need to refresh migration statistics after a completed post-copy
         * migration since priv->job.completed contains obsolete data from the
         * time we switched to post-copy mode.
         */
        if (virDomainObjGetState(vm, &reason) == VIR_DOMAIN_PAUSED &&
            reason == VIR_DOMAIN_PAUSED_POSTCOPY &&
3204
            qemuMigrationFetchStats(driver, vm, QEMU_ASYNC_JOB_MIGRATION_OUT,
3205
                                    jobInfo, NULL) < 0)
3206 3207
            VIR_WARN("Could not refresh migration statistics");

3208 3209 3210 3211 3212
        qemuDomainJobInfoUpdateTime(jobInfo);
        jobInfo->timeDeltaSet = mig->jobInfo->timeDeltaSet;
        jobInfo->timeDelta = mig->jobInfo->timeDelta;
        jobInfo->stats.downtime_set = mig->jobInfo->stats.downtime_set;
        jobInfo->stats.downtime = mig->jobInfo->stats.downtime;
3213 3214
    }

3215 3216 3217
    if (flags & VIR_MIGRATE_OFFLINE)
        goto done;

3218 3219
    /* Did the migration go as planned?  If yes, kill off the domain object.
     * If something failed, resume CPUs, but only if we didn't use post-copy.
3220 3221 3222 3223
     */
    if (retcode == 0) {
        /* If guest uses SPICE and supports seamless migration we have to hold
         * up domain shutdown until SPICE server transfers its data */
3224
        qemuMigrationWaitForSpice(vm);
3225 3226

        qemuProcessStop(driver, vm, VIR_DOMAIN_SHUTOFF_MIGRATED,
3227
                        QEMU_ASYNC_JOB_MIGRATION_OUT,
3228 3229 3230
                        VIR_QEMU_PROCESS_STOP_MIGRATED);
        virDomainAuditStop(vm, "migrated");

3231
        event = virDomainEventLifecycleNewFromObj(vm,
3232 3233
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_MIGRATED);
3234 3235
        qemuDomainEventQueue(driver, event);
        qemuDomainEventEmitJobCompleted(driver, vm);
3236
    } else {
3237
        virErrorPtr orig_err = virSaveLastError();
3238
        int reason;
3239 3240

        /* cancel any outstanding NBD jobs */
3241
        qemuMigrationCancelDriveMirror(driver, vm, false,
3242
                                       QEMU_ASYNC_JOB_MIGRATION_OUT, NULL);
3243 3244 3245

        virSetError(orig_err);
        virFreeError(orig_err);
3246

3247 3248 3249 3250
        if (virDomainObjGetState(vm, &reason) == VIR_DOMAIN_PAUSED &&
            reason == VIR_DOMAIN_PAUSED_POSTCOPY) {
            qemuMigrationPostcopyFailed(driver, vm);
        } else if (qemuMigrationRestoreDomainState(conn, vm)) {
3251 3252 3253
            event = virDomainEventLifecycleNewFromObj(vm,
                                                      VIR_DOMAIN_EVENT_RESUMED,
                                                      VIR_DOMAIN_EVENT_RESUMED_MIGRATED);
3254
            qemuDomainEventQueue(driver, event);
3255 3256
        }

3257
        qemuMigrationReset(driver, vm, QEMU_ASYNC_JOB_MIGRATION_OUT);
3258

3259
        if (virDomainSaveStatus(driver->xmlopt, cfg->stateDir, vm, driver->caps) < 0)
3260 3261 3262
            VIR_WARN("Failed to save status on vm %s", vm->def->name);
    }

3263
 done:
3264 3265 3266
    qemuMigrationCookieFree(mig);
    rv = 0;

3267
 cleanup:
3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280
    virObjectUnref(cfg);
    return rv;
}

int
qemuMigrationConfirm(virConnectPtr conn,
                     virDomainObjPtr vm,
                     const char *cookiein,
                     int cookieinlen,
                     unsigned int flags,
                     int cancelled)
{
    virQEMUDriverPtr driver = conn->privateData;
3281
    qemuMigrationJobPhase phase;
3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295
    virQEMUDriverConfigPtr cfg = NULL;
    int ret = -1;

    cfg = virQEMUDriverGetConfig(driver);

    if (!qemuMigrationJobIsActive(vm, QEMU_ASYNC_JOB_MIGRATION_OUT))
        goto cleanup;

    if (cancelled)
        phase = QEMU_MIGRATION_PHASE_CONFIRM3_CANCELLED;
    else
        phase = QEMU_MIGRATION_PHASE_CONFIRM3;

    qemuMigrationJobStartPhase(driver, vm, phase);
3296 3297
    virCloseCallbacksUnset(driver->closeCallbacks, vm,
                           qemuMigrationCleanup);
3298 3299 3300 3301 3302

    ret = qemuMigrationConfirmPhase(driver, conn, vm,
                                    cookiein, cookieinlen,
                                    flags, cancelled);

3303
    qemuMigrationJobFinish(driver, vm);
3304
    if (!virDomainObjIsActive(vm)) {
3305
        if (flags & VIR_MIGRATE_UNDEFINE_SOURCE) {
3306
            virDomainDeleteConfig(cfg->configDir, cfg->autostartDir, vm);
3307 3308
            vm->persistent = 0;
        }
3309
        qemuDomainRemoveInactiveJob(driver, vm);
3310 3311
    }

3312
 cleanup:
M
Michal Privoznik 已提交
3313
    virDomainObjEndAPI(&vm);
3314 3315 3316 3317 3318
    virObjectUnref(cfg);
    return ret;
}


3319 3320
enum qemuMigrationDestinationType {
    MIGRATION_DEST_HOST,
3321
    MIGRATION_DEST_CONNECT_HOST,
3322
    MIGRATION_DEST_FD,
3323
};
3324

3325 3326 3327 3328
enum qemuMigrationForwardType {
    MIGRATION_FWD_DIRECT,
    MIGRATION_FWD_STREAM,
};
3329

3330 3331 3332 3333 3334 3335
typedef struct _qemuMigrationSpec qemuMigrationSpec;
typedef qemuMigrationSpec *qemuMigrationSpecPtr;
struct _qemuMigrationSpec {
    enum qemuMigrationDestinationType destType;
    union {
        struct {
3336
            const char *protocol;
3337 3338 3339 3340
            const char *name;
            int port;
        } host;

3341 3342 3343 3344
        struct {
            int qemu;
            int local;
        } fd;
3345 3346 3347 3348 3349 3350 3351
    } dest;

    enum qemuMigrationForwardType fwdType;
    union {
        virStreamPtr stream;
    } fwd;
};
3352 3353 3354

#define TUNNEL_SEND_BUF_SIZE 65536

3355 3356 3357 3358 3359 3360 3361
typedef struct _qemuMigrationIOThread qemuMigrationIOThread;
typedef qemuMigrationIOThread *qemuMigrationIOThreadPtr;
struct _qemuMigrationIOThread {
    virThread thread;
    virStreamPtr st;
    int sock;
    virError err;
3362 3363
    int wakeupRecvFD;
    int wakeupSendFD;
3364 3365 3366
};

static void qemuMigrationIOFunc(void *arg)
3367
{
3368
    qemuMigrationIOThreadPtr data = arg;
3369 3370 3371 3372 3373 3374 3375
    char *buffer = NULL;
    struct pollfd fds[2];
    int timeout = -1;
    virErrorPtr err = NULL;

    VIR_DEBUG("Running migration tunnel; stream=%p, sock=%d",
              data->st, data->sock);
3376

3377
    if (VIR_ALLOC_N(buffer, TUNNEL_SEND_BUF_SIZE) < 0)
3378
        goto abrt;
3379

3380 3381 3382
    fds[0].fd = data->sock;
    fds[1].fd = data->wakeupRecvFD;

3383
    for (;;) {
3384 3385 3386 3387 3388 3389 3390 3391 3392 3393
        int ret;

        fds[0].events = fds[1].events = POLLIN;
        fds[0].revents = fds[1].revents = 0;

        ret = poll(fds, ARRAY_CARDINALITY(fds), timeout);

        if (ret < 0) {
            if (errno == EAGAIN || errno == EINTR)
                continue;
3394
            virReportSystemError(errno, "%s",
3395 3396
                                 _("poll failed in migration tunnel"));
            goto abrt;
3397
        }
3398 3399 3400 3401 3402 3403 3404

        if (ret == 0) {
            /* We were asked to gracefully stop but reading would block. This
             * can only happen if qemu told us migration finished but didn't
             * close the migration fd. We handle this in the same way as EOF.
             */
            VIR_DEBUG("QEMU forgot to close migration fd");
3405
            break;
3406
        }
3407

3408 3409 3410 3411 3412 3413 3414 3415 3416 3417 3418 3419 3420 3421 3422 3423
        if (fds[1].revents & (POLLIN | POLLERR | POLLHUP)) {
            char stop = 0;

            if (saferead(data->wakeupRecvFD, &stop, 1) != 1) {
                virReportSystemError(errno, "%s",
                                     _("failed to read from wakeup fd"));
                goto abrt;
            }

            VIR_DEBUG("Migration tunnel was asked to %s",
                      stop ? "abort" : "finish");
            if (stop) {
                goto abrt;
            } else {
                timeout = 0;
            }
3424 3425
        }

3426 3427 3428 3429 3430 3431 3432 3433 3434 3435 3436 3437 3438 3439 3440 3441 3442
        if (fds[0].revents & (POLLIN | POLLERR | POLLHUP)) {
            int nbytes;

            nbytes = saferead(data->sock, buffer, TUNNEL_SEND_BUF_SIZE);
            if (nbytes > 0) {
                if (virStreamSend(data->st, buffer, nbytes) < 0)
                    goto error;
            } else if (nbytes < 0) {
                virReportSystemError(errno, "%s",
                        _("tunnelled migration failed to read from qemu"));
                goto abrt;
            } else {
                /* EOF; get out of here */
                break;
            }
        }
    }
3443

3444 3445
    if (virStreamFinish(data->st) < 0)
        goto error;
3446

3447
    VIR_FORCE_CLOSE(data->sock);
3448 3449
    VIR_FREE(buffer);

3450 3451
    return;

3452
 abrt:
3453 3454 3455 3456 3457 3458 3459 3460 3461 3462 3463
    err = virSaveLastError();
    if (err && err->code == VIR_ERR_OK) {
        virFreeError(err);
        err = NULL;
    }
    virStreamAbort(data->st);
    if (err) {
        virSetError(err);
        virFreeError(err);
    }

3464
 error:
3465 3466 3467 3468 3469
    /* Let the source qemu know that the transfer cant continue anymore.
     * Don't copy the error for EPIPE as destination has the actual error. */
    VIR_FORCE_CLOSE(data->sock);
    if (!virLastErrorIsSystemErrno(EPIPE))
        virCopyLastError(&data->err);
3470
    virResetLastError();
3471
    VIR_FREE(buffer);
3472 3473 3474 3475 3476 3477 3478
}


static qemuMigrationIOThreadPtr
qemuMigrationStartTunnel(virStreamPtr st,
                         int sock)
{
3479 3480
    qemuMigrationIOThreadPtr io = NULL;
    int wakeupFD[2] = { -1, -1 };
3481

3482 3483 3484 3485
    if (pipe2(wakeupFD, O_CLOEXEC) < 0) {
        virReportSystemError(errno, "%s",
                             _("Unable to make pipe"));
        goto error;
3486 3487
    }

3488
    if (VIR_ALLOC(io) < 0)
3489
        goto error;
3490

3491 3492
    io->st = st;
    io->sock = sock;
3493 3494
    io->wakeupRecvFD = wakeupFD[0];
    io->wakeupSendFD = wakeupFD[1];
3495 3496 3497 3498 3499 3500

    if (virThreadCreate(&io->thread, true,
                        qemuMigrationIOFunc,
                        io) < 0) {
        virReportSystemError(errno, "%s",
                             _("Unable to create migration thread"));
3501
        goto error;
3502 3503 3504
    }

    return io;
3505

3506
 error:
3507 3508 3509 3510
    VIR_FORCE_CLOSE(wakeupFD[0]);
    VIR_FORCE_CLOSE(wakeupFD[1]);
    VIR_FREE(io);
    return NULL;
3511 3512 3513
}

static int
3514
qemuMigrationStopTunnel(qemuMigrationIOThreadPtr io, bool error)
3515 3516
{
    int rv = -1;
3517 3518 3519 3520 3521 3522 3523 3524 3525
    char stop = error ? 1 : 0;

    /* make sure the thread finishes its job and is joinable */
    if (safewrite(io->wakeupSendFD, &stop, 1) != 1) {
        virReportSystemError(errno, "%s",
                             _("failed to wakeup migration tunnel"));
        goto cleanup;
    }

3526 3527 3528 3529
    virThreadJoin(&io->thread);

    /* Forward error from the IO thread, to this thread */
    if (io->err.code != VIR_ERR_OK) {
3530 3531 3532 3533
        if (error)
            rv = 0;
        else
            virSetError(&io->err);
3534 3535 3536 3537 3538 3539
        virResetError(&io->err);
        goto cleanup;
    }

    rv = 0;

3540
 cleanup:
3541 3542
    VIR_FORCE_CLOSE(io->wakeupSendFD);
    VIR_FORCE_CLOSE(io->wakeupRecvFD);
3543 3544
    VIR_FREE(io);
    return rv;
3545 3546
}

3547
static int
3548
qemuMigrationConnect(virQEMUDriverPtr driver,
3549 3550 3551 3552 3553 3554 3555 3556 3557
                     virDomainObjPtr vm,
                     qemuMigrationSpecPtr spec)
{
    virNetSocketPtr sock;
    const char *host;
    char *port = NULL;
    int ret = -1;

    host = spec->dest.host.name;
3558
    if (virAsprintf(&port, "%d", spec->dest.host.port) < 0)
3559 3560 3561 3562 3563
        return -1;

    spec->destType = MIGRATION_DEST_FD;
    spec->dest.fd.qemu = -1;

3564
    if (qemuSecuritySetSocketLabel(driver->securityManager, vm->def) < 0)
3565
        goto cleanup;
3566 3567 3568
    if (virNetSocketNewConnectTCP(host, port,
                                  AF_UNSPEC,
                                  &sock) == 0) {
3569
        spec->dest.fd.qemu = virNetSocketDupFD(sock, true);
3570
        virObjectUnref(sock);
3571
    }
3572
    if (qemuSecurityClearSocketLabel(driver->securityManager, vm->def) < 0 ||
3573 3574 3575
        spec->dest.fd.qemu == -1)
        goto cleanup;

3576 3577 3578 3579 3580 3581 3582
    /* Migration expects a blocking FD */
    if (virSetBlocking(spec->dest.fd.qemu, true) < 0) {
        virReportSystemError(errno, _("Unable to set FD %d blocking"),
                             spec->dest.fd.qemu);
        goto cleanup;
    }

3583 3584
    ret = 0;

3585
 cleanup:
3586 3587 3588 3589 3590 3591
    VIR_FREE(port);
    if (ret < 0)
        VIR_FORCE_CLOSE(spec->dest.fd.qemu);
    return ret;
}

3592
static int
3593
qemuMigrationRun(virQEMUDriverPtr driver,
3594
                 virDomainObjPtr vm,
3595
                 const char *persist_xml,
3596 3597 3598 3599 3600 3601
                 const char *cookiein,
                 int cookieinlen,
                 char **cookieout,
                 int *cookieoutlen,
                 unsigned long flags,
                 unsigned long resource,
3602
                 qemuMigrationSpecPtr spec,
3603
                 virConnectPtr dconn,
3604 3605
                 const char *graphicsuri,
                 size_t nmigrate_disks,
3606
                 const char **migrate_disks,
3607 3608
                 qemuMigrationCompressionPtr compression,
                 qemuMonitorMigrationParamsPtr migParams)
3609
{
3610
    int ret = -1;
3611
    unsigned int migrate_flags = QEMU_MONITOR_MIGRATE_BACKGROUND;
3612
    virQEMUDriverConfigPtr cfg = NULL;
3613
    qemuDomainObjPrivatePtr priv = vm->privateData;
3614
    qemuMigrationCookiePtr mig = NULL;
3615 3616
    char *tlsAlias = NULL;
    char *secAlias = NULL;
3617
    qemuMigrationIOThreadPtr iothread = NULL;
3618
    int fd = -1;
3619
    unsigned long migrate_speed = resource ? resource : priv->migMaxBandwidth;
3620
    virErrorPtr orig_err = NULL;
3621
    unsigned int cookieFlags = 0;
3622
    bool abort_on_error = !!(flags & VIR_MIGRATE_ABORT_ON_ERROR);
3623
    bool events = virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATION_EVENT);
3624
    bool cancel = false;
3625
    unsigned int waitFlags;
3626
    virDomainDefPtr persistDef = NULL;
3627
    char *timestamp;
3628
    int rc;
3629 3630

    VIR_DEBUG("driver=%p, vm=%p, cookiein=%s, cookieinlen=%d, "
3631
              "cookieout=%p, cookieoutlen=%p, flags=0x%lx, resource=%lu, "
3632 3633
              "spec=%p (dest=%d, fwd=%d), dconn=%p, graphicsuri=%s, "
              "nmigrate_disks=%zu, migrate_disks=%p",
3634 3635
              driver, vm, NULLSTR(cookiein), cookieinlen,
              cookieout, cookieoutlen, flags, resource,
3636
              spec, spec->destType, spec->fwdType, dconn,
3637
              NULLSTR(graphicsuri), nmigrate_disks, migrate_disks);
3638

3639 3640 3641 3642 3643 3644 3645 3646 3647 3648
    if (flags & VIR_MIGRATE_NON_SHARED_DISK) {
        migrate_flags |= QEMU_MONITOR_MIGRATE_NON_SHARED_DISK;
        cookieFlags |= QEMU_MIGRATION_COOKIE_NBD;
    }

    if (flags & VIR_MIGRATE_NON_SHARED_INC) {
        migrate_flags |= QEMU_MONITOR_MIGRATE_NON_SHARED_INC;
        cookieFlags |= QEMU_MIGRATION_COOKIE_NBD;
    }

3649 3650
    if (virLockManagerPluginUsesState(driver->lockManager) &&
        !cookieout) {
3651 3652 3653 3654
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Migration with lock driver %s requires"
                         " cookie support"),
                       virLockManagerPluginGetName(driver->lockManager));
3655 3656 3657
        return -1;
    }

3658 3659 3660
    if (events)
        priv->signalIOError = abort_on_error;

3661 3662
    if (flags & VIR_MIGRATE_PERSIST_DEST) {
        if (persist_xml) {
3663 3664
            if (!(persistDef = qemuMigrationPrepareDef(driver, persist_xml,
                                                       NULL, NULL)))
3665
                goto error;
3666 3667 3668
        } else {
            virDomainDefPtr def = vm->newDef ? vm->newDef : vm->def;
            if (!(persistDef = qemuDomainDefCopy(driver, def,
3669 3670
                                                 VIR_DOMAIN_XML_SECURE |
                                                 VIR_DOMAIN_XML_MIGRATABLE)))
3671
                goto error;
3672 3673 3674
        }
    }

3675 3676 3677
    mig = qemuMigrationEatCookie(driver, vm, cookiein, cookieinlen,
                                 cookieFlags | QEMU_MIGRATION_COOKIE_GRAPHICS);
    if (!mig)
3678
        goto error;
3679

3680
    if (qemuDomainMigrateGraphicsRelocate(driver, vm, mig, graphicsuri) < 0)
3681 3682
        VIR_WARN("unable to provide data for graphics client relocation");

3683 3684 3685 3686 3687 3688 3689 3690
    if (flags & VIR_MIGRATE_TLS) {
        cfg = virQEMUDriverGetConfig(driver);

        /* Begin/CheckSetupTLS already set up migTLSAlias, the following
         * assumes that and adds the TLS objects to the domain. */
        if (qemuMigrationAddTLSObjects(driver, vm, cfg, false,
                                       QEMU_ASYNC_JOB_MIGRATION_OUT,
                                       &tlsAlias, &secAlias, migParams) < 0)
3691
            goto error;
3692 3693 3694 3695 3696 3697 3698

        /* We need to add tls-hostname whenever QEMU itself does not
         * connect directly to the destination. */
        if (spec->destType == MIGRATION_DEST_CONNECT_HOST ||
            spec->destType == MIGRATION_DEST_FD) {
            if (VIR_STRDUP(migParams->migrateTLSHostname,
                           spec->dest.host.name) < 0)
3699
                goto error;
3700 3701 3702
        } else {
            /* Be sure there's nothing from a previous migration */
            if (VIR_STRDUP(migParams->migrateTLSHostname, "") < 0)
3703
                goto error;
3704 3705 3706 3707 3708
        }
    } else {
        if (qemuMigrationSetEmptyTLSParams(driver, vm,
                                           QEMU_ASYNC_JOB_MIGRATION_OUT,
                                           migParams) < 0)
3709
            goto error;
3710 3711
    }

3712 3713 3714 3715 3716 3717 3718
    if (migrate_flags & (QEMU_MONITOR_MIGRATE_NON_SHARED_DISK |
                         QEMU_MONITOR_MIGRATE_NON_SHARED_INC)) {
        if (mig->nbd) {
            /* This will update migrate_flags on success */
            if (qemuMigrationDriveMirror(driver, vm, mig,
                                         spec->dest.host.name,
                                         migrate_speed,
3719 3720
                                         &migrate_flags,
                                         nmigrate_disks,
3721 3722
                                         migrate_disks,
                                         dconn) < 0) {
3723
                goto error;
3724 3725 3726 3727 3728 3729 3730
            }
        } else {
            /* Destination doesn't support NBD server.
             * Fall back to previous implementation. */
            VIR_DEBUG("Destination doesn't support NBD server "
                      "Falling back to previous implementation.");
        }
3731 3732
    }

3733
    /* Before EnterMonitor, since qemuMigrationSetOffline already does that */
3734 3735 3736
    if (!(flags & VIR_MIGRATE_LIVE) &&
        virDomainObjGetState(vm, NULL) == VIR_DOMAIN_RUNNING) {
        if (qemuMigrationSetOffline(driver, vm) < 0)
3737
            goto error;
3738 3739
    }

3740
    if (qemuMigrationSetCompression(driver, vm, QEMU_ASYNC_JOB_MIGRATION_OUT,
3741
                                    compression, migParams) < 0)
3742
        goto error;
3743

3744 3745 3746 3747
    if (qemuMigrationSetOption(driver, vm,
                               QEMU_MONITOR_MIGRATION_CAPS_AUTO_CONVERGE,
                               flags & VIR_MIGRATE_AUTO_CONVERGE,
                               QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
3748
        goto error;
3749

3750 3751
    if (qemuMigrationSetOption(driver, vm,
                               QEMU_MONITOR_MIGRATION_CAPS_RDMA_PIN_ALL,
3752
                               flags & VIR_MIGRATE_RDMA_PIN_ALL,
3753
                               QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
3754
        goto error;
3755

3756 3757 3758
    if (qemuMigrationSetPostCopy(driver, vm,
                                 flags & VIR_MIGRATE_POSTCOPY,
                                 QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
3759
        goto error;
3760

3761
    if (qemuMigrationSetParams(driver, vm, QEMU_ASYNC_JOB_MIGRATION_OUT,
3762
                               migParams) < 0)
3763
        goto error;
3764

3765 3766
    if (qemuDomainObjEnterMonitorAsync(driver, vm,
                                       QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
3767
        goto error;
3768

3769
    if (priv->job.abortJob) {
3770 3771
        /* explicitly do this *after* we entered the monitor,
         * as this is a critical section so we are guaranteed
3772
         * priv->job.abortJob will not change */
3773
        ignore_value(qemuDomainObjExitMonitor(driver, vm));
3774
        priv->job.current->status = QEMU_DOMAIN_JOB_STATUS_CANCELED;
3775 3776 3777
        virReportError(VIR_ERR_OPERATION_ABORTED, _("%s: %s"),
                       qemuDomainAsyncJobTypeToString(priv->job.asyncJob),
                       _("canceled by client"));
3778
        goto error;
3779 3780
    }

3781 3782
    if (qemuMonitorSetMigrationSpeed(priv->mon, migrate_speed) < 0)
        goto exit_monitor;
3783

3784 3785
    /* connect to the destination qemu if needed */
    if (spec->destType == MIGRATION_DEST_CONNECT_HOST &&
3786
        qemuMigrationConnect(driver, vm, spec) < 0) {
3787
        goto exit_monitor;
3788
    }
3789

3790 3791 3792 3793 3794 3795
    /* log start of migration */
    if ((timestamp = virTimeStringNow()) != NULL) {
        qemuDomainLogAppendMessage(driver, vm, "%s: initiating migration\n", timestamp);
        VIR_FREE(timestamp);
    }

3796
    rc = -1;
3797 3798
    switch (spec->destType) {
    case MIGRATION_DEST_HOST:
M
Michael R. Hines 已提交
3799 3800
        if (STREQ(spec->dest.host.protocol, "rdma") &&
            virProcessSetMaxMemLock(vm->pid, vm->def->mem.hard_limit << 10) < 0) {
3801
            goto exit_monitor;
M
Michael R. Hines 已提交
3802
        }
3803 3804 3805 3806
        rc = qemuMonitorMigrateToHost(priv->mon, migrate_flags,
                                      spec->dest.host.protocol,
                                      spec->dest.host.name,
                                      spec->dest.host.port);
3807 3808
        break;

3809 3810 3811 3812
    case MIGRATION_DEST_CONNECT_HOST:
        /* handled above and transformed into MIGRATION_DEST_FD */
        break;

3813
    case MIGRATION_DEST_FD:
3814
        if (spec->fwdType != MIGRATION_FWD_DIRECT) {
3815
            fd = spec->dest.fd.local;
3816 3817
            spec->dest.fd.local = -1;
        }
3818 3819
        rc = qemuMonitorMigrateToFd(priv->mon, migrate_flags,
                                    spec->dest.fd.qemu);
3820 3821
        VIR_FORCE_CLOSE(spec->dest.fd.qemu);
        break;
3822
    }
3823 3824

    if (qemuDomainObjExitMonitor(driver, vm) < 0 || rc < 0)
3825
        goto error;
3826 3827 3828

    /* From this point onwards we *must* call cancel to abort the
     * migration on source if anything goes wrong */
3829
    cancel = true;
3830

3831 3832
    if (spec->fwdType != MIGRATION_FWD_DIRECT) {
        if (!(iothread = qemuMigrationStartTunnel(spec->fwd.stream, fd)))
3833
            goto error;
3834 3835 3836 3837 3838
        /* If we've created a tunnel, then the 'fd' will be closed in the
         * qemuMigrationIOFunc as data->sock.
         */
        fd = -1;
    }
3839

3840 3841 3842 3843 3844 3845 3846 3847
    waitFlags = 0;
    if (abort_on_error)
        waitFlags |= QEMU_MIGRATION_COMPLETED_ABORT_ON_ERROR;
    if (mig->nbd)
        waitFlags |= QEMU_MIGRATION_COMPLETED_CHECK_STORAGE;
    if (flags & VIR_MIGRATE_POSTCOPY)
        waitFlags |= QEMU_MIGRATION_COMPLETED_POSTCOPY;

3848 3849
    rc = qemuMigrationWaitForCompletion(driver, vm,
                                        QEMU_ASYNC_JOB_MIGRATION_OUT,
3850
                                        dconn, waitFlags);
3851
    if (rc == -2) {
3852
        goto error;
3853 3854 3855 3856 3857
    } else if (rc == -1) {
        /* QEMU reported failed migration, nothing to cancel anymore */
        cancel = false;
        goto error;
    }
3858

3859 3860 3861
    /* When migration completed, QEMU will have paused the CPUs for us.
     * Wait for the STOP event to be processed or explicitly stop CPUs
     * (for old QEMU which does not send events) to release the lock state.
3862
     */
3863 3864 3865 3866 3867
    if (priv->monJSON) {
        while (virDomainObjGetState(vm, NULL) == VIR_DOMAIN_RUNNING) {
            priv->signalStop = true;
            rc = virDomainObjWait(vm);
            priv->signalStop = false;
3868
            if (rc < 0)
3869
                goto error;
3870
        }
3871 3872
    } else if (virDomainObjGetState(vm, NULL) == VIR_DOMAIN_RUNNING &&
               qemuMigrationSetOffline(driver, vm) < 0) {
3873
        goto error;
3874
    }
3875 3876 3877 3878 3879

    if (mig && mig->nbd &&
        qemuMigrationCancelDriveMirror(driver, vm, true,
                                       QEMU_ASYNC_JOB_MIGRATION_OUT,
                                       dconn) < 0)
3880
        goto error;
3881 3882 3883 3884 3885 3886

    if (iothread) {
        qemuMigrationIOThreadPtr io;

        VIR_STEAL_PTR(io, iothread);
        if (qemuMigrationStopTunnel(io, false) < 0)
3887
            goto error;
3888 3889 3890
    }

    if (priv->job.completed) {
3891
        priv->job.completed->stopped = priv->job.current->stopped;
3892 3893 3894 3895 3896 3897 3898 3899 3900 3901 3902 3903 3904
        qemuDomainJobInfoUpdateTime(priv->job.completed);
        qemuDomainJobInfoUpdateDowntime(priv->job.completed);
        ignore_value(virTimeMillisNow(&priv->job.completed->sent));
    }

    cookieFlags |= QEMU_MIGRATION_COOKIE_NETWORK |
                   QEMU_MIGRATION_COOKIE_STATS;

    if (qemuMigrationCookieAddPersistent(mig, &persistDef) < 0 ||
        qemuMigrationBakeCookie(mig, driver, vm, cookieout,
                                cookieoutlen, cookieFlags) < 0) {
        VIR_WARN("Unable to encode migration cookie");
    }
3905

3906
    ret = 0;
3907

3908
 cleanup:
3909 3910 3911
    VIR_FREE(tlsAlias);
    VIR_FREE(secAlias);
    virObjectUnref(cfg);
3912
    VIR_FORCE_CLOSE(fd);
3913
    virDomainDefFree(persistDef);
3914 3915
    qemuMigrationCookieFree(mig);

3916 3917 3918
    if (events)
        priv->signalIOError = false;

3919 3920 3921 3922 3923
    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }

3924 3925
    return ret;

3926
 error:
3927 3928 3929 3930 3931 3932 3933 3934 3935 3936
    orig_err = virSaveLastError();

    if (cancel &&
        priv->job.current->status != QEMU_DOMAIN_JOB_STATUS_QEMU_COMPLETED &&
        virDomainObjIsActive(vm) &&
        qemuDomainObjEnterMonitorAsync(driver, vm,
                                       QEMU_ASYNC_JOB_MIGRATION_OUT) == 0) {
        qemuMonitorMigrateCancel(priv->mon);
        ignore_value(qemuDomainObjExitMonitor(driver, vm));
    }
3937 3938 3939 3940 3941 3942 3943 3944 3945 3946 3947

    /* cancel any outstanding NBD jobs */
    if (mig && mig->nbd)
        qemuMigrationCancelDriveMirror(driver, vm, false,
                                       QEMU_ASYNC_JOB_MIGRATION_OUT,
                                       dconn);

    if (iothread)
        qemuMigrationStopTunnel(iothread, true);

    if (priv->job.current->status == QEMU_DOMAIN_JOB_STATUS_ACTIVE ||
3948 3949
        priv->job.current->status == QEMU_DOMAIN_JOB_STATUS_MIGRATING ||
        priv->job.current->status == QEMU_DOMAIN_JOB_STATUS_POSTCOPY)
3950 3951 3952 3953
        priv->job.current->status = QEMU_DOMAIN_JOB_STATUS_FAILED;

    goto cleanup;

3954 3955
 exit_monitor:
    ignore_value(qemuDomainObjExitMonitor(driver, vm));
3956
    goto error;
3957 3958
}

3959
/* Perform migration using QEMU's native migrate support,
3960 3961
 * not encrypted obviously
 */
3962
static int doNativeMigrate(virQEMUDriverPtr driver,
3963
                           virDomainObjPtr vm,
3964
                           const char *persist_xml,
3965 3966 3967 3968 3969 3970
                           const char *uri,
                           const char *cookiein,
                           int cookieinlen,
                           char **cookieout,
                           int *cookieoutlen,
                           unsigned long flags,
3971
                           unsigned long resource,
3972
                           virConnectPtr dconn,
3973 3974
                           const char *graphicsuri,
                           size_t nmigrate_disks,
3975
                           const char **migrate_disks,
3976 3977
                           qemuMigrationCompressionPtr compression,
                           qemuMonitorMigrationParamsPtr migParams)
3978
{
3979
    qemuDomainObjPrivatePtr priv = vm->privateData;
M
Martin Kletzander 已提交
3980
    virURIPtr uribits = NULL;
3981
    int ret = -1;
3982 3983 3984
    qemuMigrationSpec spec;

    VIR_DEBUG("driver=%p, vm=%p, uri=%s, cookiein=%s, cookieinlen=%d, "
3985
              "cookieout=%p, cookieoutlen=%p, flags=0x%lx, resource=%lu, "
3986
              "graphicsuri=%s, nmigrate_disks=%zu migrate_disks=%p",
3987
              driver, vm, uri, NULLSTR(cookiein), cookieinlen,
3988
              cookieout, cookieoutlen, flags, resource,
3989
              NULLSTR(graphicsuri), nmigrate_disks, migrate_disks);
3990

3991
    if (!(uribits = qemuMigrationParseURI(uri, NULL)))
3992 3993
        return -1;

3994 3995 3996 3997 3998 3999 4000
    if (uribits->scheme == NULL) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("missing scheme in migration URI: %s"),
                       uri);
        goto cleanup;
    }

M
Michael R. Hines 已提交
4001 4002 4003 4004 4005
    if (STREQ(uribits->scheme, "rdma")) {
        if (!virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATE_RDMA)) {
            virReportError(VIR_ERR_OPERATION_UNSUPPORTED, "%s",
                           _("outgoing RDMA migration is not supported "
                             "with this QEMU binary"));
4006
            goto cleanup;
M
Michael R. Hines 已提交
4007
        }
4008
        if (!virMemoryLimitIsSet(vm->def->mem.hard_limit)) {
M
Michael R. Hines 已提交
4009 4010 4011
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                           _("cannot start RDMA migration with no memory hard "
                             "limit set"));
4012
            goto cleanup;
M
Michael R. Hines 已提交
4013 4014 4015
        }
    }

4016
    if (STRNEQ(uribits->scheme, "rdma"))
4017 4018
        spec.destType = MIGRATION_DEST_CONNECT_HOST;
    else
4019
        spec.destType = MIGRATION_DEST_HOST;
4020
    spec.dest.host.protocol = uribits->scheme;
4021 4022 4023
    spec.dest.host.name = uribits->server;
    spec.dest.host.port = uribits->port;
    spec.fwdType = MIGRATION_FWD_DIRECT;
4024

4025
    ret = qemuMigrationRun(driver, vm, persist_xml, cookiein, cookieinlen, cookieout,
4026
                           cookieoutlen, flags, resource, &spec, dconn,
4027
                           graphicsuri, nmigrate_disks, migrate_disks,
4028
                           compression, migParams);
4029 4030 4031 4032

    if (spec.destType == MIGRATION_DEST_FD)
        VIR_FORCE_CLOSE(spec.dest.fd.qemu);

4033
 cleanup:
4034
    virURIFree(uribits);
4035 4036 4037 4038 4039

    return ret;
}


4040
static int doTunnelMigrate(virQEMUDriverPtr driver,
4041 4042
                           virDomainObjPtr vm,
                           virStreamPtr st,
4043
                           const char *persist_xml,
4044 4045 4046 4047 4048
                           const char *cookiein,
                           int cookieinlen,
                           char **cookieout,
                           int *cookieoutlen,
                           unsigned long flags,
4049
                           unsigned long resource,
4050
                           virConnectPtr dconn,
4051 4052
                           const char *graphicsuri,
                           size_t nmigrate_disks,
4053
                           const char **migrate_disks,
4054 4055
                           qemuMigrationCompressionPtr compression,
                           qemuMonitorMigrationParamsPtr migParams)
4056 4057 4058
{
    int ret = -1;
    qemuMigrationSpec spec;
4059
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
4060
    int fds[2] = { -1, -1 };
4061 4062

    VIR_DEBUG("driver=%p, vm=%p, st=%p, cookiein=%s, cookieinlen=%d, "
4063
              "cookieout=%p, cookieoutlen=%p, flags=0x%lx, resource=%lu, "
4064
              "graphicsuri=%s, nmigrate_disks=%zu, migrate_disks=%p",
4065
              driver, vm, st, NULLSTR(cookiein), cookieinlen,
4066
              cookieout, cookieoutlen, flags, resource,
4067
              NULLSTR(graphicsuri), nmigrate_disks, migrate_disks);
4068 4069 4070 4071

    spec.fwdType = MIGRATION_FWD_STREAM;
    spec.fwd.stream = st;

4072

4073 4074 4075
    spec.destType = MIGRATION_DEST_FD;
    spec.dest.fd.qemu = -1;
    spec.dest.fd.local = -1;
4076

4077 4078 4079 4080 4081
    if (pipe2(fds, O_CLOEXEC) == 0) {
        spec.dest.fd.qemu = fds[1];
        spec.dest.fd.local = fds[0];
    }
    if (spec.dest.fd.qemu == -1 ||
4082 4083
        qemuSecuritySetImageFDLabel(driver->securityManager, vm->def,
                                    spec.dest.fd.qemu) < 0) {
4084 4085 4086
        virReportSystemError(errno, "%s",
                             _("cannot create pipe for tunnelled migration"));
        goto cleanup;
4087 4088
    }

4089 4090 4091
    ret = qemuMigrationRun(driver, vm, persist_xml, cookiein, cookieinlen,
                           cookieout, cookieoutlen, flags, resource, &spec,
                           dconn, graphicsuri, nmigrate_disks, migrate_disks,
4092
                           compression, migParams);
4093

4094
 cleanup:
4095 4096
    VIR_FORCE_CLOSE(spec.dest.fd.qemu);
    VIR_FORCE_CLOSE(spec.dest.fd.local);
4097

4098
    virObjectUnref(cfg);
4099 4100 4101 4102
    return ret;
}


4103 4104 4105 4106
/* This is essentially a re-impl of virDomainMigrateVersion2
 * from libvirt.c, but running in source libvirtd context,
 * instead of client app context & also adding in tunnel
 * handling */
4107
static int doPeer2PeerMigrate2(virQEMUDriverPtr driver,
4108
                               virConnectPtr sconn ATTRIBUTE_UNUSED,
4109 4110
                               virConnectPtr dconn,
                               virDomainObjPtr vm,
4111
                               const char *dconnuri,
4112 4113 4114
                               unsigned long flags,
                               const char *dname,
                               unsigned long resource)
4115 4116 4117
{
    virDomainPtr ddomain = NULL;
    char *uri_out = NULL;
4118
    char *cookie = NULL;
4119 4120 4121
    char *dom_xml = NULL;
    int cookielen = 0, ret;
    virErrorPtr orig_err = NULL;
4122
    bool cancelled;
4123
    virStreamPtr st = NULL;
4124
    unsigned long destflags;
4125
    qemuMigrationCompressionPtr compression = NULL;
4126
    qemuMonitorMigrationParams migParams = { 0 };
4127

4128
    VIR_DEBUG("driver=%p, sconn=%p, dconn=%p, vm=%p, dconnuri=%s, "
4129
              "flags=0x%lx, dname=%s, resource=%lu",
4130 4131
              driver, sconn, dconn, vm, NULLSTR(dconnuri),
              flags, NULLSTR(dname), resource);
4132

4133 4134 4135 4136 4137
    /* In version 2 of the protocol, the prepare step is slightly
     * different.  We fetch the domain XML of the source domain
     * and pass it to Prepare2.
     */
    if (!(dom_xml = qemuDomainFormatXML(driver, vm,
4138 4139
                                        QEMU_DOMAIN_FORMAT_LIVE_FLAGS |
                                        VIR_DOMAIN_XML_MIGRATABLE)))
4140 4141 4142 4143 4144
        return -1;

    if (virDomainObjGetState(vm, NULL) == VIR_DOMAIN_PAUSED)
        flags |= VIR_MIGRATE_PAUSED;

4145 4146
    destflags = flags & ~(VIR_MIGRATE_ABORT_ON_ERROR |
                          VIR_MIGRATE_AUTO_CONVERGE);
4147

4148 4149 4150
    if (!(compression = qemuMigrationCompressionParse(NULL, 0, flags)))
        goto cleanup;

4151 4152 4153 4154 4155 4156 4157 4158 4159 4160
    VIR_DEBUG("Prepare2 %p", dconn);
    if (flags & VIR_MIGRATE_TUNNELLED) {
        /*
         * Tunnelled Migrate Version 2 does not support cookies
         * due to missing parameters in the prepareTunnel() API.
         */

        if (!(st = virStreamNew(dconn, 0)))
            goto cleanup;

4161
        qemuDomainObjEnterRemote(vm);
4162
        ret = dconn->driver->domainMigratePrepareTunnel
4163
            (dconn, st, destflags, dname, resource, dom_xml);
4164
        qemuDomainObjExitRemote(vm);
4165
    } else {
4166
        qemuDomainObjEnterRemote(vm);
4167 4168
        ret = dconn->driver->domainMigratePrepare2
            (dconn, &cookie, &cookielen, NULL, &uri_out,
4169
             destflags, dname, resource, dom_xml);
4170
        qemuDomainObjExitRemote(vm);
4171 4172 4173
    }
    VIR_FREE(dom_xml);
    if (ret == -1)
4174 4175 4176
        goto cleanup;

    /* the domain may have shutdown or crashed while we had the locks dropped
4177
     * in qemuDomainObjEnterRemote, so check again
4178 4179
     */
    if (!virDomainObjIsActive(vm)) {
4180 4181
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest unexpectedly quit"));
4182 4183 4184
        goto cleanup;
    }

4185 4186
    if (!(flags & VIR_MIGRATE_TUNNELLED) &&
        (uri_out == NULL)) {
4187 4188
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("domainMigratePrepare2 did not set uri"));
4189
        cancelled = true;
4190
        orig_err = virSaveLastError();
4191
        goto finish;
4192 4193
    }

4194 4195 4196 4197
    /* Perform the migration.  The driver isn't supposed to return
     * until the migration is complete.
     */
    VIR_DEBUG("Perform %p", sconn);
4198
    qemuMigrationJobSetPhase(driver, vm, QEMU_MIGRATION_PHASE_PERFORM2);
4199
    if (flags & VIR_MIGRATE_TUNNELLED)
4200
        ret = doTunnelMigrate(driver, vm, st, NULL,
4201
                              NULL, 0, NULL, NULL,
4202
                              flags, resource, dconn,
4203
                              NULL, 0, NULL, compression, &migParams);
4204
    else
4205
        ret = doNativeMigrate(driver, vm, NULL, uri_out,
4206 4207
                              cookie, cookielen,
                              NULL, NULL, /* No out cookie with v2 migration */
4208 4209
                              flags, resource, dconn, NULL, 0, NULL,
                              compression, &migParams);
4210 4211 4212 4213

    /* Perform failed. Make sure Finish doesn't overwrite the error */
    if (ret < 0)
        orig_err = virSaveLastError();
4214

4215 4216 4217
    /* If Perform returns < 0, then we need to cancel the VM
     * startup on the destination
     */
4218
    cancelled = ret < 0;
4219

4220
 finish:
4221 4222 4223 4224
    /* In version 2 of the migration protocol, we pass the
     * status code from the sender to the destination host,
     * so it can do any cleanup if the migration failed.
     */
4225
    dname = dname ? dname : vm->def->name;
4226
    VIR_DEBUG("Finish2 %p ret=%d", dconn, ret);
4227
    qemuDomainObjEnterRemote(vm);
4228
    ddomain = dconn->driver->domainMigrateFinish2
4229
        (dconn, dname, cookie, cookielen,
4230
         uri_out ? uri_out : dconnuri, destflags, cancelled);
4231
    qemuDomainObjExitRemote(vm);
4232 4233
    if (cancelled && ddomain)
        VIR_ERROR(_("finish step ignored that migration was cancelled"));
4234

4235
 cleanup:
4236
    if (ddomain) {
4237
        virObjectUnref(ddomain);
4238 4239 4240 4241
        ret = 0;
    } else {
        ret = -1;
    }
4242

4243
    virObjectUnref(st);
4244 4245 4246 4247 4248

    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }
4249
    qemuMigrationParamsClear(&migParams);
4250
    VIR_FREE(uri_out);
4251
    VIR_FREE(cookie);
4252
    VIR_FREE(compression);
4253 4254

    return ret;
4255 4256 4257
}


4258 4259 4260 4261
/* This is essentially a re-impl of virDomainMigrateVersion3
 * from libvirt.c, but running in source libvirtd context,
 * instead of client app context & also adding in tunnel
 * handling */
4262 4263 4264 4265 4266 4267 4268
static int
doPeer2PeerMigrate3(virQEMUDriverPtr driver,
                    virConnectPtr sconn,
                    virConnectPtr dconn,
                    const char *dconnuri,
                    virDomainObjPtr vm,
                    const char *xmlin,
4269
                    const char *persist_xml,
4270 4271
                    const char *dname,
                    const char *uri,
4272
                    const char *graphicsuri,
4273
                    const char *listenAddress,
4274 4275
                    size_t nmigrate_disks,
                    const char **migrate_disks,
4276
                    int nbdPort,
4277
                    qemuMigrationCompressionPtr compression,
4278
                    qemuMonitorMigrationParamsPtr migParams,
4279 4280 4281
                    unsigned long long bandwidth,
                    bool useParams,
                    unsigned long flags)
4282 4283 4284 4285 4286 4287 4288 4289 4290 4291
{
    virDomainPtr ddomain = NULL;
    char *uri_out = NULL;
    char *cookiein = NULL;
    char *cookieout = NULL;
    char *dom_xml = NULL;
    int cookieinlen = 0;
    int cookieoutlen = 0;
    int ret = -1;
    virErrorPtr orig_err = NULL;
4292
    bool cancelled = true;
4293
    virStreamPtr st = NULL;
4294
    unsigned long destflags;
4295 4296 4297
    virTypedParameterPtr params = NULL;
    int nparams = 0;
    int maxparams = 0;
4298
    size_t i;
4299 4300

    VIR_DEBUG("driver=%p, sconn=%p, dconn=%p, dconnuri=%s, vm=%p, xmlin=%s, "
4301
              "dname=%s, uri=%s, graphicsuri=%s, listenAddress=%s, "
4302
              "nmigrate_disks=%zu, migrate_disks=%p, nbdPort=%d, "
4303
              "bandwidth=%llu, useParams=%d, flags=0x%lx",
4304
              driver, sconn, dconn, NULLSTR(dconnuri), vm, NULLSTR(xmlin),
4305
              NULLSTR(dname), NULLSTR(uri), NULLSTR(graphicsuri),
4306
              NULLSTR(listenAddress), nmigrate_disks, migrate_disks, nbdPort,
4307
              bandwidth, useParams, flags);
4308

4309 4310 4311 4312 4313
    /* Unlike the virDomainMigrateVersion3 counterpart, we don't need
     * to worry about auto-setting the VIR_MIGRATE_CHANGE_PROTECTION
     * bit here, because we are already running inside the context of
     * a single job.  */

4314
    dom_xml = qemuMigrationBeginPhase(driver, vm, xmlin, dname,
4315 4316
                                      &cookieout, &cookieoutlen,
                                      nmigrate_disks, migrate_disks, flags);
4317 4318 4319
    if (!dom_xml)
        goto cleanup;

4320 4321 4322 4323 4324 4325 4326 4327 4328 4329 4330 4331 4332 4333 4334 4335 4336 4337 4338 4339
    if (useParams) {
        if (virTypedParamsAddString(&params, &nparams, &maxparams,
                                    VIR_MIGRATE_PARAM_DEST_XML, dom_xml) < 0)
            goto cleanup;

        if (dname &&
            virTypedParamsAddString(&params, &nparams, &maxparams,
                                    VIR_MIGRATE_PARAM_DEST_NAME, dname) < 0)
            goto cleanup;

        if (uri &&
            virTypedParamsAddString(&params, &nparams, &maxparams,
                                    VIR_MIGRATE_PARAM_URI, uri) < 0)
            goto cleanup;

        if (bandwidth &&
            virTypedParamsAddULLong(&params, &nparams, &maxparams,
                                    VIR_MIGRATE_PARAM_BANDWIDTH,
                                    bandwidth) < 0)
            goto cleanup;
4340 4341 4342 4343 4344 4345

        if (graphicsuri &&
            virTypedParamsAddString(&params, &nparams, &maxparams,
                                    VIR_MIGRATE_PARAM_GRAPHICS_URI,
                                    graphicsuri) < 0)
            goto cleanup;
4346 4347 4348 4349 4350
        if (listenAddress &&
            virTypedParamsAddString(&params, &nparams, &maxparams,
                                    VIR_MIGRATE_PARAM_LISTEN_ADDRESS,
                                    listenAddress) < 0)
            goto cleanup;
4351 4352 4353 4354 4355
        for (i = 0; i < nmigrate_disks; i++)
            if (virTypedParamsAddString(&params, &nparams, &maxparams,
                                        VIR_MIGRATE_PARAM_MIGRATE_DISKS,
                                        migrate_disks[i]) < 0)
                goto cleanup;
4356 4357 4358 4359 4360
        if (nbdPort &&
            virTypedParamsAddInt(&params, &nparams, &maxparams,
                                 VIR_MIGRATE_PARAM_DISKS_PORT,
                                 nbdPort) < 0)
            goto cleanup;
4361 4362 4363 4364

        if (qemuMigrationCompressionDump(compression, &params, &nparams,
                                         &maxparams, &flags) < 0)
            goto cleanup;
4365 4366
    }

4367 4368 4369
    if (virDomainObjGetState(vm, NULL) == VIR_DOMAIN_PAUSED)
        flags |= VIR_MIGRATE_PAUSED;

4370 4371
    destflags = flags & ~(VIR_MIGRATE_ABORT_ON_ERROR |
                          VIR_MIGRATE_AUTO_CONVERGE);
4372

4373 4374 4375 4376 4377 4378 4379 4380 4381
    VIR_DEBUG("Prepare3 %p", dconn);
    cookiein = cookieout;
    cookieinlen = cookieoutlen;
    cookieout = NULL;
    cookieoutlen = 0;
    if (flags & VIR_MIGRATE_TUNNELLED) {
        if (!(st = virStreamNew(dconn, 0)))
            goto cleanup;

4382
        qemuDomainObjEnterRemote(vm);
4383 4384 4385 4386 4387 4388 4389 4390 4391
        if (useParams) {
            ret = dconn->driver->domainMigratePrepareTunnel3Params
                (dconn, st, params, nparams, cookiein, cookieinlen,
                 &cookieout, &cookieoutlen, destflags);
        } else {
            ret = dconn->driver->domainMigratePrepareTunnel3
                (dconn, st, cookiein, cookieinlen, &cookieout, &cookieoutlen,
                 destflags, dname, bandwidth, dom_xml);
        }
4392
        qemuDomainObjExitRemote(vm);
4393
    } else {
4394
        qemuDomainObjEnterRemote(vm);
4395 4396 4397 4398 4399 4400 4401 4402 4403
        if (useParams) {
            ret = dconn->driver->domainMigratePrepare3Params
                (dconn, params, nparams, cookiein, cookieinlen,
                 &cookieout, &cookieoutlen, &uri_out, destflags);
        } else {
            ret = dconn->driver->domainMigratePrepare3
                (dconn, cookiein, cookieinlen, &cookieout, &cookieoutlen,
                 uri, &uri_out, destflags, dname, bandwidth, dom_xml);
        }
4404
        qemuDomainObjExitRemote(vm);
4405 4406 4407 4408 4409
    }
    VIR_FREE(dom_xml);
    if (ret == -1)
        goto cleanup;

L
liguang 已提交
4410 4411 4412 4413
    if (flags & VIR_MIGRATE_OFFLINE) {
        VIR_DEBUG("Offline migration, skipping Perform phase");
        VIR_FREE(cookieout);
        cookieoutlen = 0;
4414
        cancelled = false;
L
liguang 已提交
4415 4416 4417
        goto finish;
    }

4418 4419 4420 4421
    if (uri_out) {
        uri = uri_out;
        if (useParams &&
            virTypedParamsReplaceString(&params, &nparams,
4422 4423
                                        VIR_MIGRATE_PARAM_URI, uri_out) < 0) {
            orig_err = virSaveLastError();
4424
            goto finish;
4425
        }
4426
    } else if (!uri && !(flags & VIR_MIGRATE_TUNNELLED)) {
4427 4428
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("domainMigratePrepare3 did not set uri"));
4429
        orig_err = virSaveLastError();
4430 4431 4432 4433 4434 4435 4436 4437
        goto finish;
    }

    /* Perform the migration.  The driver isn't supposed to return
     * until the migration is complete. The src VM should remain
     * running, but in paused state until the destination can
     * confirm migration completion.
     */
4438
    VIR_DEBUG("Perform3 %p uri=%s", sconn, NULLSTR(uri));
4439
    qemuMigrationJobSetPhase(driver, vm, QEMU_MIGRATION_PHASE_PERFORM3);
4440 4441 4442 4443 4444
    VIR_FREE(cookiein);
    cookiein = cookieout;
    cookieinlen = cookieoutlen;
    cookieout = NULL;
    cookieoutlen = 0;
4445
    if (flags & VIR_MIGRATE_TUNNELLED) {
4446
        ret = doTunnelMigrate(driver, vm, st, persist_xml,
4447 4448
                              cookiein, cookieinlen,
                              &cookieout, &cookieoutlen,
4449
                              flags, bandwidth, dconn, graphicsuri,
4450 4451
                              nmigrate_disks, migrate_disks, compression,
                              migParams);
4452
    } else {
4453
        ret = doNativeMigrate(driver, vm, persist_xml, uri,
4454 4455
                              cookiein, cookieinlen,
                              &cookieout, &cookieoutlen,
4456
                              flags, bandwidth, dconn, graphicsuri,
4457 4458
                              nmigrate_disks, migrate_disks, compression,
                              migParams);
4459
    }
4460 4461

    /* Perform failed. Make sure Finish doesn't overwrite the error */
4462
    if (ret < 0) {
4463
        orig_err = virSaveLastError();
4464 4465 4466 4467
    } else {
        qemuMigrationJobSetPhase(driver, vm,
                                 QEMU_MIGRATION_PHASE_PERFORM3_DONE);
    }
4468 4469 4470 4471

    /* If Perform returns < 0, then we need to cancel the VM
     * startup on the destination
     */
4472
    cancelled = ret < 0;
4473

4474
 finish:
4475 4476 4477 4478 4479 4480 4481 4482 4483 4484 4485 4486
    /*
     * The status code from the source is passed to the destination.
     * The dest can cleanup in the source indicated it failed to
     * send all migration data. Returns NULL for ddomain if
     * the dest was unable to complete migration.
     */
    VIR_DEBUG("Finish3 %p ret=%d", dconn, ret);
    VIR_FREE(cookiein);
    cookiein = cookieout;
    cookieinlen = cookieoutlen;
    cookieout = NULL;
    cookieoutlen = 0;
4487 4488 4489 4490 4491 4492 4493 4494 4495 4496 4497 4498 4499 4500 4501 4502 4503 4504 4505 4506 4507 4508 4509

    if (useParams) {
        if (virTypedParamsGetString(params, nparams,
                                    VIR_MIGRATE_PARAM_DEST_NAME, NULL) <= 0 &&
            virTypedParamsReplaceString(&params, &nparams,
                                        VIR_MIGRATE_PARAM_DEST_NAME,
                                        vm->def->name) < 0) {
            ddomain = NULL;
        } else {
            qemuDomainObjEnterRemote(vm);
            ddomain = dconn->driver->domainMigrateFinish3Params
                (dconn, params, nparams, cookiein, cookieinlen,
                 &cookieout, &cookieoutlen, destflags, cancelled);
            qemuDomainObjExitRemote(vm);
        }
    } else {
        dname = dname ? dname : vm->def->name;
        qemuDomainObjEnterRemote(vm);
        ddomain = dconn->driver->domainMigrateFinish3
            (dconn, dname, cookiein, cookieinlen, &cookieout, &cookieoutlen,
             dconnuri, uri, destflags, cancelled);
        qemuDomainObjExitRemote(vm);
    }
4510 4511 4512 4513 4514 4515 4516 4517 4518 4519 4520 4521 4522 4523 4524 4525 4526 4527 4528 4529

    if (cancelled) {
        if (ddomain) {
            VIR_ERROR(_("finish step ignored that migration was cancelled"));
        } else {
            /* If Finish reported a useful error, use it instead of the
             * original "migration unexpectedly failed" error.
             *
             * This is ugly but we can't do better with the APIs we have. We
             * only replace the error if Finish was called with cancelled == 1
             * and reported a real error (old libvirt would report an error
             * from RPC instead of MIGRATE_FINISH_OK), which only happens when
             * the domain died on destination. To further reduce a possibility
             * of false positives we also check that Perform returned
             * VIR_ERR_OPERATION_FAILED.
             */
            if (orig_err &&
                orig_err->domain == VIR_FROM_QEMU &&
                orig_err->code == VIR_ERR_OPERATION_FAILED) {
                virErrorPtr err = virGetLastError();
4530 4531
                if (err &&
                    err->domain == VIR_FROM_QEMU &&
4532 4533 4534 4535 4536 4537 4538
                    err->code != VIR_ERR_MIGRATE_FINISH_OK) {
                    virFreeError(orig_err);
                    orig_err = NULL;
                }
            }
        }
    }
4539

4540 4541 4542 4543 4544 4545 4546
    /* If ddomain is NULL, then we were unable to start
     * the guest on the target, and must restart on the
     * source. There is a small chance that the ddomain
     * is NULL due to an RPC failure, in which case
     * ddomain could in fact be running on the dest.
     * The lock manager plugins should take care of
     * safety in this scenario.
4547
     */
4548
    cancelled = ddomain == NULL;
4549

4550 4551 4552 4553 4554 4555
    /* If finish3 set an error, and we don't have an earlier
     * one we need to preserve it in case confirm3 overwrites
     */
    if (!orig_err)
        orig_err = virSaveLastError();

4556 4557 4558 4559
    /*
     * If cancelled, then src VM will be restarted, else
     * it will be killed
     */
4560
    VIR_DEBUG("Confirm3 %p cancelled=%d vm=%p", sconn, cancelled, vm);
4561 4562 4563 4564 4565
    VIR_FREE(cookiein);
    cookiein = cookieout;
    cookieinlen = cookieoutlen;
    cookieout = NULL;
    cookieoutlen = 0;
4566 4567 4568
    ret = qemuMigrationConfirmPhase(driver, sconn, vm,
                                    cookiein, cookieinlen,
                                    flags, cancelled);
4569 4570 4571 4572
    /* If Confirm3 returns -1, there's nothing more we can
     * do, but fortunately worst case is that there is a
     * domain left in 'paused' state on source.
     */
4573 4574 4575
    if (ret < 0)
        VIR_WARN("Guest %s probably left in 'paused' state on source",
                 vm->def->name);
4576 4577 4578

 cleanup:
    if (ddomain) {
4579
        virObjectUnref(ddomain);
4580 4581 4582 4583 4584
        ret = 0;
    } else {
        ret = -1;
    }

4585
    virObjectUnref(st);
4586 4587 4588 4589 4590 4591 4592 4593

    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }
    VIR_FREE(uri_out);
    VIR_FREE(cookiein);
    VIR_FREE(cookieout);
4594
    virTypedParamsFree(params, nparams);
4595 4596 4597 4598
    return ret;
}


4599 4600 4601 4602 4603 4604 4605 4606 4607 4608 4609 4610
static void
qemuMigrationConnectionClosed(virConnectPtr conn,
                              int reason,
                              void *opaque)
{
    virDomainObjPtr vm = opaque;

    VIR_DEBUG("conn=%p, reason=%d, vm=%s", conn, reason, vm->def->name);
    virDomainObjBroadcast(vm);
}


4611 4612 4613 4614 4615 4616 4617 4618 4619 4620 4621 4622
static int virConnectCredType[] = {
    VIR_CRED_AUTHNAME,
    VIR_CRED_PASSPHRASE,
};


static virConnectAuth virConnectAuthConfig = {
    .credtype = virConnectCredType,
    .ncredtype = ARRAY_CARDINALITY(virConnectCredType),
};


4623
static int doPeer2PeerMigrate(virQEMUDriverPtr driver,
4624
                              virConnectPtr sconn,
4625
                              virDomainObjPtr vm,
4626
                              const char *xmlin,
4627
                              const char *persist_xml,
4628
                              const char *dconnuri,
4629
                              const char *uri,
4630
                              const char *graphicsuri,
4631
                              const char *listenAddress,
4632 4633
                              size_t nmigrate_disks,
                              const char **migrate_disks,
4634
                              int nbdPort,
4635
                              qemuMigrationCompressionPtr compression,
4636
                              qemuMonitorMigrationParamsPtr migParams,
4637 4638
                              unsigned long flags,
                              const char *dname,
4639 4640
                              unsigned long resource,
                              bool *v3proto)
4641 4642 4643 4644
{
    int ret = -1;
    virConnectPtr dconn = NULL;
    bool p2p;
4645
    virErrorPtr orig_err = NULL;
4646
    bool offline = false;
4647
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
4648
    bool useParams;
4649

4650 4651
    VIR_DEBUG("driver=%p, sconn=%p, vm=%p, xmlin=%s, dconnuri=%s, uri=%s, "
              "graphicsuri=%s, listenAddress=%s, nmigrate_disks=%zu, "
4652
              "migrate_disks=%p, nbdPort=%d, flags=0x%lx, dname=%s, "
4653
              "resource=%lu",
4654
              driver, sconn, vm, NULLSTR(xmlin), NULLSTR(dconnuri),
4655
              NULLSTR(uri), NULLSTR(graphicsuri), NULLSTR(listenAddress),
4656 4657
              nmigrate_disks, migrate_disks, nbdPort, flags, NULLSTR(dname),
              resource);
4658

4659 4660 4661 4662 4663 4664 4665 4666 4667 4668 4669 4670 4671 4672
    if (flags & VIR_MIGRATE_TUNNELLED && uri) {
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("migration URI is not supported by tunnelled "
                         "migration"));
        goto cleanup;
    }

    if (flags & VIR_MIGRATE_TUNNELLED && listenAddress) {
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("listen address is not supported by tunnelled "
                         "migration"));
        goto cleanup;
    }

4673 4674 4675 4676 4677 4678 4679
    if (flags & VIR_MIGRATE_TUNNELLED && nbdPort) {
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("disk port address is not supported by tunnelled "
                         "migration"));
        goto cleanup;
    }

4680 4681 4682 4683
    /* the order of operations is important here; we make sure the
     * destination side is completely setup before we touch the source
     */

4684
    qemuDomainObjEnterRemote(vm);
4685
    dconn = virConnectOpenAuth(dconnuri, &virConnectAuthConfig, 0);
4686
    qemuDomainObjExitRemote(vm);
4687
    if (dconn == NULL) {
4688
        virReportError(VIR_ERR_OPERATION_FAILED,
4689 4690
                       _("Failed to connect to remote libvirt URI %s: %s"),
                       dconnuri, virGetLastErrorMessage());
4691
        virObjectUnref(cfg);
4692 4693 4694
        return -1;
    }

4695 4696
    if (virConnectSetKeepAlive(dconn, cfg->keepAliveInterval,
                               cfg->keepAliveCount) < 0)
4697 4698
        goto cleanup;

4699 4700 4701 4702 4703
    if (virConnectRegisterCloseCallback(dconn, qemuMigrationConnectionClosed,
                                        vm, NULL) < 0) {
        goto cleanup;
    }

4704
    qemuDomainObjEnterRemote(vm);
4705 4706
    p2p = VIR_DRV_SUPPORTS_FEATURE(dconn->driver, dconn,
                                   VIR_DRV_FEATURE_MIGRATION_P2P);
4707
        /* v3proto reflects whether the caller used Perform3, but with
4708
         * p2p migrate, regardless of whether Perform2 or Perform3
4709 4710 4711 4712
         * were used, we decide protocol based on what target supports
         */
    *v3proto = VIR_DRV_SUPPORTS_FEATURE(dconn->driver, dconn,
                                        VIR_DRV_FEATURE_MIGRATION_V3);
4713 4714
    useParams = VIR_DRV_SUPPORTS_FEATURE(dconn->driver, dconn,
                                         VIR_DRV_FEATURE_MIGRATION_PARAMS);
L
liguang 已提交
4715 4716 4717
    if (flags & VIR_MIGRATE_OFFLINE)
        offline = VIR_DRV_SUPPORTS_FEATURE(dconn->driver, dconn,
                                           VIR_DRV_FEATURE_MIGRATION_OFFLINE);
4718
    qemuDomainObjExitRemote(vm);
4719

4720
    if (!p2p) {
4721 4722
        virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                       _("Destination libvirt does not support peer-to-peer migration protocol"));
4723 4724 4725
        goto cleanup;
    }

4726 4727
    /* Only xmlin, dname, uri, and bandwidth parameters can be used with
     * old-style APIs. */
4728
    if (!useParams && (graphicsuri || listenAddress || nmigrate_disks)) {
4729 4730 4731 4732 4733 4734
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("Migration APIs with extensible parameters are not "
                         "supported but extended parameters were passed"));
        goto cleanup;
    }

L
liguang 已提交
4735 4736 4737 4738 4739 4740 4741
    if (flags & VIR_MIGRATE_OFFLINE && !offline) {
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("offline migration is not supported by "
                         "the destination host"));
        goto cleanup;
    }

4742
    /* domain may have been stopped while we were talking to remote daemon */
L
liguang 已提交
4743
    if (!virDomainObjIsActive(vm) && !(flags & VIR_MIGRATE_OFFLINE)) {
4744 4745
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest unexpectedly quit"));
4746 4747 4748
        goto cleanup;
    }

4749 4750 4751 4752 4753 4754 4755
    /* Change protection is only required on the source side (us), and
     * only for v3 migration when begin and perform are separate jobs.
     * But peer-2-peer is already a single job, and we still want to
     * talk to older destinations that would reject the flag.
     * Therefore it is safe to clear the bit here.  */
    flags &= ~VIR_MIGRATE_CHANGE_PROTECTION;

4756 4757
    if (*v3proto) {
        ret = doPeer2PeerMigrate3(driver, sconn, dconn, dconnuri, vm, xmlin,
4758 4759
                                  persist_xml, dname, uri, graphicsuri,
                                  listenAddress, nmigrate_disks, migrate_disks,
4760 4761
                                  nbdPort, compression, migParams, resource,
                                  useParams, flags);
4762
    } else {
4763
        ret = doPeer2PeerMigrate2(driver, sconn, dconn, vm,
4764
                                  dconnuri, flags, dname, resource);
4765
    }
4766

4767
 cleanup:
4768
    orig_err = virSaveLastError();
4769
    qemuDomainObjEnterRemote(vm);
4770
    virConnectUnregisterCloseCallback(dconn, qemuMigrationConnectionClosed);
4771
    virObjectUnref(dconn);
4772
    qemuDomainObjExitRemote(vm);
4773 4774 4775 4776
    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }
4777
    virObjectUnref(cfg);
4778 4779 4780 4781
    return ret;
}


4782 4783 4784 4785 4786 4787
/*
 * This implements perform part of the migration protocol when migration job
 * does not need to be active across several APIs, i.e., peer2peer migration or
 * perform phase of v2 non-peer2peer migration.
 */
static int
4788
qemuMigrationPerformJob(virQEMUDriverPtr driver,
4789 4790 4791
                        virConnectPtr conn,
                        virDomainObjPtr vm,
                        const char *xmlin,
4792
                        const char *persist_xml,
4793 4794
                        const char *dconnuri,
                        const char *uri,
4795
                        const char *graphicsuri,
4796
                        const char *listenAddress,
4797 4798
                        size_t nmigrate_disks,
                        const char **migrate_disks,
4799
                        int nbdPort,
4800
                        qemuMigrationCompressionPtr compression,
4801
                        qemuMonitorMigrationParamsPtr migParams,
4802
                        const char *cookiein,
4803 4804 4805 4806 4807 4808 4809
                        int cookieinlen,
                        char **cookieout,
                        int *cookieoutlen,
                        unsigned long flags,
                        const char *dname,
                        unsigned long resource,
                        bool v3proto)
4810
{
4811
    virObjectEventPtr event = NULL;
4812
    int ret = -1;
4813
    virErrorPtr orig_err = NULL;
4814
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
4815

4816
    if (qemuMigrationJobStart(driver, vm, QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
4817 4818
        goto cleanup;

L
liguang 已提交
4819
    if (!virDomainObjIsActive(vm) && !(flags & VIR_MIGRATE_OFFLINE)) {
4820 4821
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s", _("domain is not running"));
4822 4823 4824
        goto endjob;
    }

4825
    if (!qemuMigrationIsAllowed(driver, vm, true, flags))
4826
        goto endjob;
4827

4828
    if (!(flags & (VIR_MIGRATE_UNSAFE | VIR_MIGRATE_OFFLINE)) &&
4829
        !qemuMigrationIsSafe(vm->def, nmigrate_disks, migrate_disks, flags))
4830
        goto endjob;
4831

4832
    qemuMigrationStoreDomainState(vm);
4833 4834

    if ((flags & (VIR_MIGRATE_TUNNELLED | VIR_MIGRATE_PEER2PEER))) {
4835
        ret = doPeer2PeerMigrate(driver, conn, vm, xmlin, persist_xml,
4836
                                 dconnuri, uri, graphicsuri, listenAddress,
4837
                                 nmigrate_disks, migrate_disks, nbdPort,
4838 4839
                                 compression, migParams, flags, dname, resource,
                                 &v3proto);
4840
    } else {
4841
        qemuMigrationJobSetPhase(driver, vm, QEMU_MIGRATION_PHASE_PERFORM2);
4842
        ret = doNativeMigrate(driver, vm, persist_xml, uri, cookiein, cookieinlen,
4843
                              cookieout, cookieoutlen,
4844
                              flags, resource, NULL, NULL, 0, NULL,
4845
                              compression, migParams);
4846
    }
4847 4848
    if (ret < 0)
        goto endjob;
4849

4850 4851 4852 4853
    /*
     * In v3 protocol, the source VM is not killed off until the
     * confirm step.
     */
4854
    if (!v3proto) {
4855
        qemuProcessStop(driver, vm, VIR_DOMAIN_SHUTOFF_MIGRATED,
4856
                        QEMU_ASYNC_JOB_MIGRATION_OUT,
4857
                        VIR_QEMU_PROCESS_STOP_MIGRATED);
4858
        virDomainAuditStop(vm, "migrated");
4859
        event = virDomainEventLifecycleNewFromObj(vm,
4860 4861
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_MIGRATED);
4862 4863
    }

4864
 endjob:
4865 4866 4867
    if (ret < 0)
        orig_err = virSaveLastError();

4868 4869 4870 4871
    /* v2 proto has no confirm phase so we need to reset migration parameters
     * here
     */
    if (!v3proto && ret < 0)
4872
        qemuMigrationReset(driver, vm, QEMU_ASYNC_JOB_MIGRATION_OUT);
4873

4874
    if (qemuMigrationRestoreDomainState(conn, vm)) {
4875
        event = virDomainEventLifecycleNewFromObj(vm,
4876 4877 4878
                                         VIR_DOMAIN_EVENT_RESUMED,
                                         VIR_DOMAIN_EVENT_RESUMED_MIGRATED);
    }
4879

4880
    qemuMigrationJobFinish(driver, vm);
4881
    if (!virDomainObjIsActive(vm) && ret == 0) {
4882
        if (flags & VIR_MIGRATE_UNDEFINE_SOURCE) {
4883
            virDomainDeleteConfig(cfg->configDir, cfg->autostartDir, vm);
4884 4885
            vm->persistent = 0;
        }
4886
        qemuDomainRemoveInactiveJob(driver, vm);
4887 4888
    }

4889 4890 4891 4892 4893
    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }

4894
 cleanup:
M
Michal Privoznik 已提交
4895
    virDomainObjEndAPI(&vm);
4896
    qemuDomainEventQueue(driver, event);
4897
    virObjectUnref(cfg);
4898 4899 4900 4901 4902 4903 4904
    return ret;
}

/*
 * This implements perform phase of v3 migration protocol.
 */
static int
4905
qemuMigrationPerformPhase(virQEMUDriverPtr driver,
4906 4907
                          virConnectPtr conn,
                          virDomainObjPtr vm,
4908
                          const char *persist_xml,
4909
                          const char *uri,
4910
                          const char *graphicsuri,
4911 4912
                          size_t nmigrate_disks,
                          const char **migrate_disks,
4913
                          qemuMigrationCompressionPtr compression,
4914
                          qemuMonitorMigrationParamsPtr migParams,
4915 4916 4917 4918 4919 4920 4921
                          const char *cookiein,
                          int cookieinlen,
                          char **cookieout,
                          int *cookieoutlen,
                          unsigned long flags,
                          unsigned long resource)
{
4922
    virObjectEventPtr event = NULL;
4923 4924 4925 4926 4927 4928 4929 4930 4931 4932 4933
    int ret = -1;

    /* If we didn't start the job in the begin phase, start it now. */
    if (!(flags & VIR_MIGRATE_CHANGE_PROTECTION)) {
        if (qemuMigrationJobStart(driver, vm, QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
            goto cleanup;
    } else if (!qemuMigrationJobIsActive(vm, QEMU_ASYNC_JOB_MIGRATION_OUT)) {
        goto cleanup;
    }

    qemuMigrationJobStartPhase(driver, vm, QEMU_MIGRATION_PHASE_PERFORM3);
4934 4935
    virCloseCallbacksUnset(driver->closeCallbacks, vm,
                           qemuMigrationCleanup);
4936

4937
    ret = doNativeMigrate(driver, vm, persist_xml, uri, cookiein, cookieinlen,
4938
                          cookieout, cookieoutlen,
4939
                          flags, resource, NULL, graphicsuri,
4940
                          nmigrate_disks, migrate_disks, compression, migParams);
4941

4942 4943 4944 4945 4946
    if (ret < 0) {
        if (qemuMigrationRestoreDomainState(conn, vm)) {
            event = virDomainEventLifecycleNewFromObj(vm,
                                                      VIR_DOMAIN_EVENT_RESUMED,
                                                      VIR_DOMAIN_EVENT_RESUMED_MIGRATED);
4947
        }
4948
        goto endjob;
4949
    }
4950 4951 4952

    qemuMigrationJobSetPhase(driver, vm, QEMU_MIGRATION_PHASE_PERFORM3_DONE);

4953 4954
    if (virCloseCallbacksSet(driver->closeCallbacks, vm, conn,
                             qemuMigrationCleanup) < 0)
4955 4956
        goto endjob;

4957
 endjob:
4958 4959
    if (ret < 0) {
        qemuMigrationReset(driver, vm, QEMU_ASYNC_JOB_MIGRATION_OUT);
4960
        qemuMigrationJobFinish(driver, vm);
4961
    } else {
4962
        qemuMigrationJobContinue(vm);
4963 4964
    }

4965
    if (!virDomainObjIsActive(vm))
4966
        qemuDomainRemoveInactiveJob(driver, vm);
4967

4968
 cleanup:
M
Michal Privoznik 已提交
4969
    virDomainObjEndAPI(&vm);
4970
    qemuDomainEventQueue(driver, event);
4971 4972 4973
    return ret;
}

4974
int
4975
qemuMigrationPerform(virQEMUDriverPtr driver,
4976 4977 4978
                     virConnectPtr conn,
                     virDomainObjPtr vm,
                     const char *xmlin,
4979
                     const char *persist_xml,
4980 4981
                     const char *dconnuri,
                     const char *uri,
4982
                     const char *graphicsuri,
4983
                     const char *listenAddress,
4984 4985
                     size_t nmigrate_disks,
                     const char **migrate_disks,
4986
                     int nbdPort,
4987
                     qemuMigrationCompressionPtr compression,
4988
                     qemuMonitorMigrationParamsPtr migParams,
4989 4990 4991 4992 4993 4994 4995 4996 4997 4998
                     const char *cookiein,
                     int cookieinlen,
                     char **cookieout,
                     int *cookieoutlen,
                     unsigned long flags,
                     const char *dname,
                     unsigned long resource,
                     bool v3proto)
{
    VIR_DEBUG("driver=%p, conn=%p, vm=%p, xmlin=%s, dconnuri=%s, "
4999
              "uri=%s, graphicsuri=%s, listenAddress=%s, "
5000
              "nmigrate_disks=%zu, migrate_disks=%p, nbdPort=%d, "
5001
              "cookiein=%s, cookieinlen=%d, cookieout=%p, cookieoutlen=%p, "
5002
              "flags=0x%lx, dname=%s, resource=%lu, v3proto=%d",
5003
              driver, conn, vm, NULLSTR(xmlin), NULLSTR(dconnuri),
5004
              NULLSTR(uri), NULLSTR(graphicsuri), NULLSTR(listenAddress),
5005 5006 5007
              nmigrate_disks, migrate_disks, nbdPort,
              NULLSTR(cookiein), cookieinlen, cookieout, cookieoutlen,
              flags, NULLSTR(dname), resource, v3proto);
5008 5009 5010

    if ((flags & (VIR_MIGRATE_TUNNELLED | VIR_MIGRATE_PEER2PEER))) {
        if (cookieinlen) {
5011 5012
            virReportError(VIR_ERR_OPERATION_INVALID,
                           "%s", _("received unexpected cookie with P2P migration"));
5013 5014 5015
            return -1;
        }

5016
        return qemuMigrationPerformJob(driver, conn, vm, xmlin, persist_xml, dconnuri, uri,
5017
                                       graphicsuri, listenAddress,
5018
                                       nmigrate_disks, migrate_disks, nbdPort,
5019 5020
                                       compression, migParams,
                                       cookiein, cookieinlen,
5021 5022
                                       cookieout, cookieoutlen,
                                       flags, dname, resource, v3proto);
5023 5024
    } else {
        if (dconnuri) {
5025 5026
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           "%s", _("Unexpected dconnuri parameter with non-peer2peer migration"));
5027 5028 5029 5030
            return -1;
        }

        if (v3proto) {
5031
            return qemuMigrationPerformPhase(driver, conn, vm, persist_xml, uri,
5032
                                             graphicsuri,
5033
                                             nmigrate_disks, migrate_disks,
5034 5035
                                             compression, migParams,
                                             cookiein, cookieinlen,
5036
                                             cookieout, cookieoutlen,
5037
                                             flags, resource);
5038
        } else {
5039
            return qemuMigrationPerformJob(driver, conn, vm, xmlin, persist_xml, NULL,
5040
                                           uri, graphicsuri, listenAddress,
5041
                                           nmigrate_disks, migrate_disks, nbdPort,
5042 5043
                                           compression, migParams,
                                           cookiein, cookieinlen,
5044 5045 5046 5047 5048
                                           cookieout, cookieoutlen, flags,
                                           dname, resource, v3proto);
        }
    }
}
5049

5050
static int
5051 5052
qemuMigrationVPAssociatePortProfiles(virDomainDefPtr def)
{
5053
    size_t i;
5054 5055 5056 5057 5058
    int last_good_net = -1;
    virDomainNetDefPtr net;

    for (i = 0; i < def->nnets; i++) {
        net = def->nets[i];
5059
        if (virDomainNetGetActualType(net) == VIR_DOMAIN_NET_TYPE_DIRECT) {
5060
            if (virNetDevVPortProfileAssociate(net->ifname,
5061
                                               virDomainNetGetActualVirtPortProfile(net),
5062
                                               &net->mac,
5063
                                               virDomainNetGetActualDirectDev(net),
5064
                                               -1,
5065
                                               def->uuid,
5066 5067
                                               VIR_NETDEV_VPORT_PROFILE_OP_MIGRATE_IN_FINISH,
                                               false) < 0) {
5068 5069 5070
                virReportError(VIR_ERR_OPERATION_FAILED,
                               _("Port profile Associate failed for %s"),
                               net->ifname);
5071
                goto err_exit;
5072
            }
5073
            last_good_net = i;
5074
            VIR_DEBUG("Port profile Associate succeeded for %s", net->ifname);
5075

5076
            if (virNetDevMacVLanVPortProfileRegisterCallback(net->ifname, &net->mac,
5077 5078 5079 5080
                                                             virDomainNetGetActualDirectDev(net), def->uuid,
                                                             virDomainNetGetActualVirtPortProfile(net),
                                                             VIR_NETDEV_VPORT_PROFILE_OP_CREATE))
                goto err_exit;
5081 5082 5083
        }
    }

5084
    return 0;
5085

5086
 err_exit:
5087
    for (i = 0; last_good_net != -1 && i <= last_good_net; i++) {
5088
        net = def->nets[i];
5089
        if (virDomainNetGetActualType(net) == VIR_DOMAIN_NET_TYPE_DIRECT) {
5090
            ignore_value(virNetDevVPortProfileDisassociate(net->ifname,
5091
                                                           virDomainNetGetActualVirtPortProfile(net),
5092
                                                           &net->mac,
5093
                                                           virDomainNetGetActualDirectDev(net),
5094
                                                           -1,
5095
                                                           VIR_NETDEV_VPORT_PROFILE_OP_MIGRATE_IN_FINISH));
5096 5097
        }
    }
5098
    return -1;
5099 5100 5101
}


J
Jiri Denemark 已提交
5102 5103 5104
static int
qemuMigrationPersist(virQEMUDriverPtr driver,
                     virDomainObjPtr vm,
5105 5106
                     qemuMigrationCookiePtr mig,
                     bool ignoreSaveError)
J
Jiri Denemark 已提交
5107 5108 5109 5110
{
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
    virCapsPtr caps = NULL;
    virDomainDefPtr vmdef;
5111 5112
    virDomainDefPtr oldDef = NULL;
    unsigned int oldPersist = vm->persistent;
J
Jiri Denemark 已提交
5113 5114 5115 5116 5117 5118 5119
    virObjectEventPtr event;
    int ret = -1;

    if (!(caps = virQEMUDriverGetCapabilities(driver, false)))
        goto cleanup;

    vm->persistent = 1;
5120 5121
    oldDef = vm->newDef;
    vm->newDef = qemuMigrationCookieGetPersistent(mig);
J
Jiri Denemark 已提交
5122

5123 5124
    if (!(vmdef = virDomainObjGetPersistentDef(caps, driver->xmlopt, vm)))
        goto error;
J
Jiri Denemark 已提交
5125

5126 5127
    if (virDomainSaveConfig(cfg->configDir, driver->caps, vmdef) < 0 &&
        !ignoreSaveError)
5128
        goto error;
J
Jiri Denemark 已提交
5129 5130 5131

    event = virDomainEventLifecycleNewFromObj(vm,
                                              VIR_DOMAIN_EVENT_DEFINED,
5132 5133 5134
                                              oldPersist ?
                                              VIR_DOMAIN_EVENT_DEFINED_UPDATED :
                                              VIR_DOMAIN_EVENT_DEFINED_ADDED);
5135
    qemuDomainEventQueue(driver, event);
J
Jiri Denemark 已提交
5136 5137 5138 5139

    ret = 0;

 cleanup:
5140
    virDomainDefFree(oldDef);
J
Jiri Denemark 已提交
5141 5142 5143
    virObjectUnref(caps);
    virObjectUnref(cfg);
    return ret;
5144 5145 5146 5147 5148 5149 5150

 error:
    virDomainDefFree(vm->newDef);
    vm->persistent = oldPersist;
    vm->newDef = oldDef;
    oldDef = NULL;
    goto cleanup;
J
Jiri Denemark 已提交
5151 5152 5153
}


5154
virDomainPtr
5155
qemuMigrationFinish(virQEMUDriverPtr driver,
5156 5157
                    virConnectPtr dconn,
                    virDomainObjPtr vm,
5158 5159 5160 5161
                    const char *cookiein,
                    int cookieinlen,
                    char **cookieout,
                    int *cookieoutlen,
5162
                    unsigned long flags,
5163 5164
                    int retcode,
                    bool v3proto)
5165 5166
{
    virDomainPtr dom = NULL;
5167
    qemuMigrationCookiePtr mig = NULL;
5168
    virErrorPtr orig_err = NULL;
5169
    int cookie_flags = 0;
J
Jiri Denemark 已提交
5170
    qemuDomainObjPrivatePtr priv = vm->privateData;
5171
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
5172
    unsigned short port;
5173 5174
    unsigned long long timeReceived = 0;
    virObjectEventPtr event;
5175
    qemuDomainJobInfoPtr jobInfo = NULL;
5176
    bool inPostCopy = false;
5177
    bool doKill = true;
5178

5179
    VIR_DEBUG("driver=%p, dconn=%p, vm=%p, cookiein=%s, cookieinlen=%d, "
5180
              "cookieout=%p, cookieoutlen=%p, flags=0x%lx, retcode=%d",
5181 5182
              driver, dconn, vm, NULLSTR(cookiein), cookieinlen,
              cookieout, cookieoutlen, flags, retcode);
5183

5184 5185 5186
    port = priv->migrationPort;
    priv->migrationPort = 0;

5187 5188
    if (!qemuMigrationJobIsActive(vm, QEMU_ASYNC_JOB_MIGRATION_IN)) {
        qemuMigrationErrorReport(driver, vm->def->name);
5189
        goto cleanup;
5190
    }
5191

5192 5193
    ignore_value(virTimeMillisNow(&timeReceived));

5194 5195 5196
    qemuMigrationJobStartPhase(driver, vm,
                               v3proto ? QEMU_MIGRATION_PHASE_FINISH3
                                       : QEMU_MIGRATION_PHASE_FINISH2);
5197

5198
    qemuDomainCleanupRemove(vm, qemuMigrationPrepareCleanup);
5199
    VIR_FREE(priv->job.completed);
5200

5201
    cookie_flags = QEMU_MIGRATION_COOKIE_NETWORK |
5202 5203
                   QEMU_MIGRATION_COOKIE_STATS |
                   QEMU_MIGRATION_COOKIE_NBD;
5204 5205 5206
    /* Some older versions of libvirt always send persistent XML in the cookie
     * even though VIR_MIGRATE_PERSIST_DEST was not used. */
    cookie_flags |= QEMU_MIGRATION_COOKIE_PERSISTENT;
5207 5208 5209

    if (!(mig = qemuMigrationEatCookie(driver, vm, cookiein,
                                       cookieinlen, cookie_flags)))
5210
        goto endjob;
5211

5212
    if (flags & VIR_MIGRATE_OFFLINE) {
5213 5214
        if (retcode == 0 &&
            qemuMigrationPersist(driver, vm, mig, false) == 0)
5215
            dom = virGetDomain(dconn, vm->def->name, vm->def->uuid, -1);
5216 5217
        goto endjob;
    }
5218

5219 5220 5221 5222
    if (retcode != 0) {
        /* Check for a possible error on the monitor in case Finish was called
         * earlier than monitor EOF handler got a chance to process the error
         */
5223
        qemuDomainCheckMonitor(driver, vm, QEMU_ASYNC_JOB_MIGRATION_IN);
5224 5225
        goto endjob;
    }
5226

5227 5228 5229 5230 5231 5232
    if (!virDomainObjIsActive(vm)) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest unexpectedly quit"));
        qemuMigrationErrorReport(driver, vm->def->name);
        goto endjob;
    }
5233

5234 5235
    if (qemuMigrationVPAssociatePortProfiles(vm->def) < 0)
        goto endjob;
5236

5237 5238
    if (mig->network && qemuDomainMigrateOPDRelocate(driver, vm, mig) < 0)
        VIR_WARN("unable to provide network data for relocation");
5239

5240 5241
    if (qemuMigrationStopNBDServer(driver, vm, mig) < 0)
        goto endjob;
5242

5243 5244
    if (qemuRefreshVirtioChannelState(driver, vm,
                                      QEMU_ASYNC_JOB_MIGRATION_IN) < 0)
5245 5246
        goto endjob;

5247 5248
    if (qemuConnectAgent(driver, vm) < 0)
        goto endjob;
5249

5250 5251 5252 5253 5254 5255 5256 5257 5258 5259 5260 5261 5262
    if (flags & VIR_MIGRATE_PERSIST_DEST) {
        if (qemuMigrationPersist(driver, vm, mig, !v3proto) < 0) {
            /* Hmpf.  Migration was successful, but making it persistent
             * was not.  If we report successful, then when this domain
             * shuts down, management tools are in for a surprise.  On the
             * other hand, if we report failure, then the management tools
             * might try to restart the domain on the source side, even
             * though the domain is actually running on the destination.
             * Pretend success and hope that this is a rare situation and
             * management tools are smart.
             *
             * However, in v3 protocol, the source VM is still available
             * to restart during confirm() step, so we kill it off now.
5263 5264 5265
             */
            if (v3proto)
                goto endjob;
5266
        }
5267
    }
5268

5269 5270 5271 5272
    /* We need to wait for QEMU to process all data sent by the source
     * before starting guest CPUs.
     */
    if (qemuMigrationWaitForDestCompletion(driver, vm,
5273 5274
                                           QEMU_ASYNC_JOB_MIGRATION_IN,
                                           !!(flags & VIR_MIGRATE_POSTCOPY)) < 0) {
5275 5276 5277 5278 5279 5280
        /* There's not much we can do for v2 protocol since the
         * original domain on the source host is already gone.
         */
        if (v3proto)
            goto endjob;
    }
5281

5282
    if (priv->job.current->status == QEMU_DOMAIN_JOB_STATUS_POSTCOPY)
5283 5284
        inPostCopy = true;

5285 5286 5287 5288 5289 5290
    if (!(flags & VIR_MIGRATE_PAUSED)) {
        /* run 'cont' on the destination, which allows migration on qemu
         * >= 0.10.6 to work properly.  This isn't strictly necessary on
         * older qemu's, but it also doesn't hurt anything there
         */
        if (qemuProcessStartCPUs(driver, vm, dconn,
5291 5292
                                 inPostCopy ? VIR_DOMAIN_RUNNING_POSTCOPY
                                            : VIR_DOMAIN_RUNNING_MIGRATED,
5293 5294 5295 5296 5297 5298 5299 5300
                                 QEMU_ASYNC_JOB_MIGRATION_IN) < 0) {
            if (virGetLastError() == NULL)
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               "%s", _("resume operation failed"));
            /* Need to save the current error, in case shutting
             * down the process overwrites it
             */
            orig_err = virSaveLastError();
5301

5302 5303 5304 5305 5306 5307 5308 5309 5310 5311
            /*
             * In v3 protocol, the source VM is still available to
             * restart during confirm() step, so we kill it off
             * now.
             * In v2 protocol, the source is dead, so we leave
             * target in paused state, in case admin can fix
             * things up.
             */
            if (v3proto)
                goto endjob;
5312
        }
5313 5314

        if (inPostCopy) {
5315
            doKill = false;
5316 5317 5318 5319 5320
            event = virDomainEventLifecycleNewFromObj(vm,
                                        VIR_DOMAIN_EVENT_RESUMED,
                                        VIR_DOMAIN_EVENT_RESUMED_POSTCOPY);
            qemuDomainEventQueue(driver, event);
        }
5321
    }
5322

5323
    if (mig->jobInfo) {
5324
        jobInfo = mig->jobInfo;
5325 5326 5327 5328 5329 5330
        mig->jobInfo = NULL;

        if (jobInfo->sent && timeReceived) {
            jobInfo->timeDelta = timeReceived - jobInfo->sent;
            jobInfo->received = timeReceived;
            jobInfo->timeDeltaSet = true;
5331
        }
5332 5333
        qemuDomainJobInfoUpdateTime(jobInfo);
        qemuDomainJobInfoUpdateDowntime(jobInfo);
5334
    }
L
liguang 已提交
5335

5336 5337 5338 5339 5340 5341 5342 5343 5344 5345 5346 5347 5348
    if (inPostCopy) {
        if (qemuMigrationWaitForDestCompletion(driver, vm,
                                               QEMU_ASYNC_JOB_MIGRATION_IN,
                                               false) < 0) {
            goto endjob;
        }
        if (virDomainObjGetState(vm, NULL) == VIR_DOMAIN_RUNNING) {
            virDomainObjSetState(vm,
                                 VIR_DOMAIN_RUNNING,
                                 VIR_DOMAIN_RUNNING_MIGRATED);
        }
    }

5349
    dom = virGetDomain(dconn, vm->def->name, vm->def->uuid, vm->def->id);
5350

5351 5352 5353 5354
    event = virDomainEventLifecycleNewFromObj(vm,
                                              VIR_DOMAIN_EVENT_RESUMED,
                                              VIR_DOMAIN_EVENT_RESUMED_MIGRATED);
    qemuDomainEventQueue(driver, event);
5355

5356 5357 5358 5359 5360 5361
    if (virDomainObjGetState(vm, NULL) == VIR_DOMAIN_PAUSED) {
        virDomainObjSetState(vm, VIR_DOMAIN_PAUSED, VIR_DOMAIN_PAUSED_USER);
        event = virDomainEventLifecycleNewFromObj(vm,
                                                  VIR_DOMAIN_EVENT_SUSPENDED,
                                                  VIR_DOMAIN_EVENT_SUSPENDED_PAUSED);
        qemuDomainEventQueue(driver, event);
5362
    }
5363

5364
    if (virDomainObjIsActive(vm) &&
5365
        virDomainSaveStatus(driver->xmlopt, cfg->stateDir, vm, driver->caps) < 0)
5366 5367 5368 5369 5370
        VIR_WARN("Failed to save status on vm %s", vm->def->name);

    /* Guest is successfully running, so cancel previous auto destroy */
    qemuProcessAutoDestroyRemove(driver, vm);

5371
 endjob:
5372
    if (!dom &&
5373 5374
        !(flags & VIR_MIGRATE_OFFLINE) &&
        virDomainObjIsActive(vm)) {
5375
        if (doKill) {
5376 5377 5378 5379 5380 5381 5382 5383 5384 5385 5386
            qemuProcessStop(driver, vm, VIR_DOMAIN_SHUTOFF_FAILED,
                            QEMU_ASYNC_JOB_MIGRATION_IN,
                            VIR_QEMU_PROCESS_STOP_MIGRATED);
            virDomainAuditStop(vm, "failed");
            event = virDomainEventLifecycleNewFromObj(vm,
                                VIR_DOMAIN_EVENT_STOPPED,
                                VIR_DOMAIN_EVENT_STOPPED_FAILED);
            qemuDomainEventQueue(driver, event);
        } else {
            qemuMigrationPostcopyFailed(driver, vm);
        }
5387 5388
    }

5389 5390 5391 5392 5393 5394
    if (dom) {
        priv->job.completed = jobInfo;
        jobInfo = NULL;
        if (qemuMigrationBakeCookie(mig, driver, vm, cookieout, cookieoutlen,
                                    QEMU_MIGRATION_COOKIE_STATS) < 0)
            VIR_WARN("Unable to encode migration cookie");
5395 5396 5397 5398 5399 5400

        /* Remove completed stats for post-copy, everything but timing fields
         * is obsolete anyway.
         */
        if (inPostCopy)
            VIR_FREE(priv->job.completed);
5401
    }
5402

5403
    qemuMigrationReset(driver, vm, QEMU_ASYNC_JOB_MIGRATION_IN);
5404

5405
    qemuMigrationJobFinish(driver, vm);
5406
    if (!virDomainObjIsActive(vm))
5407
        qemuDomainRemoveInactiveJob(driver, vm);
5408

5409
 cleanup:
5410
    VIR_FREE(jobInfo);
5411
    virPortAllocatorRelease(driver->migrationPorts, port);
5412
    if (priv->mon)
5413
        qemuMonitorSetDomainLog(priv->mon, NULL, NULL, NULL);
5414
    VIR_FREE(priv->origname);
M
Michal Privoznik 已提交
5415
    virDomainObjEndAPI(&vm);
5416
    qemuMigrationCookieFree(mig);
5417 5418 5419 5420
    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }
5421
    virObjectUnref(cfg);
5422 5423 5424 5425 5426 5427

    /* Set a special error if Finish is expected to return NULL as a result of
     * successful call with retcode != 0
     */
    if (retcode != 0 && !dom && !virGetLastError())
        virReportError(VIR_ERR_MIGRATE_FINISH_OK, NULL);
5428 5429
    return dom;
}
5430

5431

5432
/* Helper function called while vm is active.  */
5433
int
5434
qemuMigrationToFile(virQEMUDriverPtr driver, virDomainObjPtr vm,
5435
                    int fd,
5436
                    const char *compressor,
5437
                    qemuDomainAsyncJob asyncJob)
5438 5439 5440
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int rc;
5441
    int ret = -1;
5442 5443
    virCommandPtr cmd = NULL;
    int pipeFD[2] = { -1, -1 };
5444
    unsigned long saveMigBandwidth = priv->migMaxBandwidth;
5445
    char *errbuf = NULL;
5446
    virErrorPtr orig_err = NULL;
5447 5448 5449 5450 5451

    /* Increase migration bandwidth to unlimited since target is a file.
     * Failure to change migration speed is not fatal. */
    if (qemuDomainObjEnterMonitorAsync(driver, vm, asyncJob) == 0) {
        qemuMonitorSetMigrationSpeed(priv->mon,
5452 5453
                                     QEMU_DOMAIN_MIG_BANDWIDTH_MAX);
        priv->migMaxBandwidth = QEMU_DOMAIN_MIG_BANDWIDTH_MAX;
5454 5455
        if (qemuDomainObjExitMonitor(driver, vm) < 0)
            return -1;
5456
    }
5457

5458 5459 5460 5461 5462 5463 5464
    if (!virDomainObjIsActive(vm)) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest unexpectedly quit"));
        /* nothing to tear down */
        return -1;
    }

5465 5466 5467 5468
    if (compressor && pipe(pipeFD) < 0) {
        virReportSystemError(errno, "%s",
                             _("Failed to create pipe for migration"));
        return -1;
5469 5470
    }

5471 5472 5473 5474
    /* All right! We can use fd migration, which means that qemu
     * doesn't have to open() the file, so while we still have to
     * grant SELinux access, we can do it on fd and avoid cleanup
     * later, as well as skip futzing with cgroup.  */
5475 5476
    if (qemuSecuritySetImageFDLabel(driver->securityManager, vm->def,
                                    compressor ? pipeFD[1] : fd) < 0)
5477 5478
        goto cleanup;

5479
    if (qemuDomainObjEnterMonitorAsync(driver, vm, asyncJob) < 0)
5480 5481
        goto cleanup;

5482
    if (!compressor) {
5483 5484 5485
        rc = qemuMonitorMigrateToFd(priv->mon,
                                    QEMU_MONITOR_MIGRATE_BACKGROUND,
                                    fd);
5486 5487 5488 5489 5490 5491 5492
    } else {
        const char *prog = compressor;
        const char *args[] = {
            prog,
            "-c",
            NULL
        };
5493 5494 5495 5496 5497 5498 5499 5500 5501 5502 5503

        cmd = virCommandNewArgs(args);
        virCommandSetInputFD(cmd, pipeFD[0]);
        virCommandSetOutputFD(cmd, &fd);
        virCommandSetErrorBuffer(cmd, &errbuf);
        virCommandDoAsyncIO(cmd);
        if (virSetCloseExec(pipeFD[1]) < 0) {
            virReportSystemError(errno, "%s",
                                 _("Unable to set cloexec flag"));
            ignore_value(qemuDomainObjExitMonitor(driver, vm));
            goto cleanup;
5504
        }
5505 5506 5507 5508 5509 5510 5511 5512 5513 5514
        if (virCommandRunAsync(cmd, NULL) < 0) {
            ignore_value(qemuDomainObjExitMonitor(driver, vm));
            goto cleanup;
        }
        rc = qemuMonitorMigrateToFd(priv->mon,
                                    QEMU_MONITOR_MIGRATE_BACKGROUND,
                                    pipeFD[1]);
        if (VIR_CLOSE(pipeFD[0]) < 0 ||
            VIR_CLOSE(pipeFD[1]) < 0)
            VIR_WARN("failed to close intermediate pipe");
5515
    }
5516
    if (qemuDomainObjExitMonitor(driver, vm) < 0)
5517
        goto cleanup;
5518 5519 5520
    if (rc < 0)
        goto cleanup;

5521
    rc = qemuMigrationWaitForCompletion(driver, vm, asyncJob, NULL, 0);
5522

5523 5524 5525 5526
    if (rc < 0) {
        if (rc == -2) {
            orig_err = virSaveLastError();
            virCommandAbort(cmd);
5527 5528
            if (virDomainObjIsActive(vm) &&
                qemuDomainObjEnterMonitorAsync(driver, vm, asyncJob) == 0) {
5529
                qemuMonitorMigrateCancel(priv->mon);
5530
                ignore_value(qemuDomainObjExitMonitor(driver, vm));
5531 5532
            }
        }
5533
        goto cleanup;
5534
    }
5535

5536 5537 5538
    if (cmd && virCommandWait(cmd, NULL) < 0)
        goto cleanup;

5539
    qemuDomainEventEmitJobCompleted(driver, vm);
5540 5541
    ret = 0;

5542
 cleanup:
5543 5544 5545
    if (ret < 0 && !orig_err)
        orig_err = virSaveLastError();

5546
    /* Restore max migration bandwidth */
5547 5548
    if (virDomainObjIsActive(vm) &&
        qemuDomainObjEnterMonitorAsync(driver, vm, asyncJob) == 0) {
5549 5550
        qemuMonitorSetMigrationSpeed(priv->mon, saveMigBandwidth);
        priv->migMaxBandwidth = saveMigBandwidth;
5551
        ignore_value(qemuDomainObjExitMonitor(driver, vm));
5552 5553
    }

5554 5555
    VIR_FORCE_CLOSE(pipeFD[0]);
    VIR_FORCE_CLOSE(pipeFD[1]);
5556 5557 5558 5559 5560
    if (cmd) {
        VIR_DEBUG("Compression binary stderr: %s", NULLSTR(errbuf));
        VIR_FREE(errbuf);
        virCommandFree(cmd);
    }
5561 5562 5563 5564 5565 5566

    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }

5567 5568
    return ret;
}
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 5595 5596 5597 5598 5599 5600 5601 5602 5603 5604 5605 5606 5607 5608 5609 5610 5611 5612 5613 5614 5615 5616

int
qemuMigrationCancel(virQEMUDriverPtr driver,
                    virDomainObjPtr vm)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    virHashTablePtr blockJobs = NULL;
    bool storage = false;
    size_t i;
    int ret = -1;

    VIR_DEBUG("Canceling unfinished outgoing migration of domain %s",
              vm->def->name);

    for (i = 0; i < vm->def->ndisks; i++) {
        virDomainDiskDefPtr disk = vm->def->disks[i];
        if (QEMU_DOMAIN_DISK_PRIVATE(disk)->migrating) {
            qemuBlockJobSyncBegin(disk);
            storage = true;
        }
    }

    qemuDomainObjEnterMonitor(driver, vm);

    ignore_value(qemuMonitorMigrateCancel(priv->mon));
    if (storage)
        blockJobs = qemuMonitorGetAllBlockJobInfo(priv->mon);

    if (qemuDomainObjExitMonitor(driver, vm) < 0 || (storage && !blockJobs))
        goto endsyncjob;

    if (!storage) {
        ret = 0;
        goto cleanup;
    }

    for (i = 0; i < vm->def->ndisks; i++) {
        virDomainDiskDefPtr disk = vm->def->disks[i];
        qemuDomainDiskPrivatePtr diskPriv = QEMU_DOMAIN_DISK_PRIVATE(disk);

        if (!diskPriv->migrating)
            continue;

        if (virHashLookup(blockJobs, disk->info.alias)) {
            VIR_DEBUG("Drive mirror on disk %s is still running", disk->dst);
        } else {
            VIR_DEBUG("Drive mirror on disk %s is gone", disk->dst);
5617
            qemuBlockJobSyncEnd(driver, vm, QEMU_ASYNC_JOB_NONE, disk);
5618 5619 5620 5621 5622
            diskPriv->migrating = false;
        }
    }

    if (qemuMigrationCancelDriveMirror(driver, vm, false,
5623
                                       QEMU_ASYNC_JOB_NONE, NULL) < 0)
5624 5625 5626 5627 5628 5629 5630 5631 5632 5633 5634 5635 5636 5637 5638
        goto endsyncjob;

    ret = 0;

 cleanup:
    virHashFree(blockJobs);
    return ret;

 endsyncjob:
    if (storage) {
        for (i = 0; i < vm->def->ndisks; i++) {
            virDomainDiskDefPtr disk = vm->def->disks[i];
            qemuDomainDiskPrivatePtr diskPriv = QEMU_DOMAIN_DISK_PRIVATE(disk);

            if (diskPriv->migrating) {
5639
                qemuBlockJobSyncEnd(driver, vm, QEMU_ASYNC_JOB_NONE, disk);
5640 5641 5642 5643 5644 5645 5646 5647
                diskPriv->migrating = false;
            }
        }
    }
    goto cleanup;
}


5648
int
5649
qemuMigrationJobStart(virQEMUDriverPtr driver,
5650
                      virDomainObjPtr vm,
5651
                      qemuDomainAsyncJob job)
5652
{
5653 5654
    virDomainJobOperation op;
    unsigned long long mask;
5655

5656
    if (job == QEMU_ASYNC_JOB_MIGRATION_IN) {
5657 5658
        op = VIR_DOMAIN_JOB_OPERATION_MIGRATION_IN;
        mask = QEMU_JOB_NONE;
5659
    } else {
5660 5661 5662 5663
        op = VIR_DOMAIN_JOB_OPERATION_MIGRATION_OUT;
        mask = QEMU_JOB_DEFAULT_MASK |
               JOB_MASK(QEMU_JOB_SUSPEND) |
               JOB_MASK(QEMU_JOB_MIGRATION_OP);
5664
    }
5665

5666 5667 5668 5669
    if (qemuDomainObjBeginAsyncJob(driver, vm, job, op) < 0)
        return -1;

    qemuDomainObjSetAsyncJobMask(vm, mask);
5670 5671 5672 5673
    return 0;
}

void
5674
qemuMigrationJobSetPhase(virQEMUDriverPtr driver,
5675
                         virDomainObjPtr vm,
5676
                         qemuMigrationJobPhase phase)
5677 5678 5679 5680 5681 5682 5683 5684 5685 5686 5687 5688 5689 5690
{
    qemuDomainObjPrivatePtr priv = vm->privateData;

    if (phase < priv->job.phase) {
        VIR_ERROR(_("migration protocol going backwards %s => %s"),
                  qemuMigrationJobPhaseTypeToString(priv->job.phase),
                  qemuMigrationJobPhaseTypeToString(phase));
        return;
    }

    qemuDomainObjSetJobPhase(driver, vm, phase);
}

void
5691
qemuMigrationJobStartPhase(virQEMUDriverPtr driver,
5692
                           virDomainObjPtr vm,
5693
                           qemuMigrationJobPhase phase)
5694 5695 5696 5697
{
    qemuMigrationJobSetPhase(driver, vm, phase);
}

5698
void
5699 5700
qemuMigrationJobContinue(virDomainObjPtr vm)
{
5701
    qemuDomainObjReleaseAsyncJob(vm);
5702 5703 5704 5705
}

bool
qemuMigrationJobIsActive(virDomainObjPtr vm,
5706
                         qemuDomainAsyncJob job)
5707 5708 5709 5710 5711 5712 5713 5714 5715 5716 5717
{
    qemuDomainObjPrivatePtr priv = vm->privateData;

    if (priv->job.asyncJob != job) {
        const char *msg;

        if (job == QEMU_ASYNC_JOB_MIGRATION_IN)
            msg = _("domain '%s' is not processing incoming migration");
        else
            msg = _("domain '%s' is not being migrated");

5718
        virReportError(VIR_ERR_OPERATION_INVALID, msg, vm->def->name);
5719 5720 5721 5722 5723
        return false;
    }
    return true;
}

5724
void
5725
qemuMigrationJobFinish(virQEMUDriverPtr driver, virDomainObjPtr vm)
5726
{
5727
    qemuDomainObjEndAsyncJob(driver, vm);
5728
}
5729 5730 5731 5732 5733 5734 5735 5736 5737 5738 5739 5740 5741 5742 5743 5744 5745 5746 5747 5748 5749 5750 5751 5752 5753 5754 5755 5756 5757 5758 5759 5760 5761 5762 5763 5764 5765 5766 5767 5768 5769 5770 5771 5772 5773 5774 5775 5776 5777 5778 5779 5780 5781 5782


static void
qemuMigrationErrorFree(void *data,
                       const void *name ATTRIBUTE_UNUSED)
{
    virErrorPtr err = data;
    virFreeError(err);
}

int
qemuMigrationErrorInit(virQEMUDriverPtr driver)
{
    driver->migrationErrors = virHashAtomicNew(64, qemuMigrationErrorFree);
    if (driver->migrationErrors)
        return 0;
    else
        return -1;
}

/**
 * This function consumes @err; the caller should consider the @err pointer
 * invalid after calling this function.
 */
void
qemuMigrationErrorSave(virQEMUDriverPtr driver,
                       const char *name,
                       virErrorPtr err)
{
    if (!err)
        return;

    VIR_DEBUG("Saving incoming migration error for domain %s: %s",
              name, err->message);
    if (virHashAtomicUpdate(driver->migrationErrors, name, err) < 0) {
        VIR_WARN("Failed to save migration error for domain '%s'", name);
        virFreeError(err);
    }
}

void
qemuMigrationErrorReport(virQEMUDriverPtr driver,
                         const char *name)
{
    virErrorPtr err;

    if (!(err = virHashAtomicSteal(driver->migrationErrors, name)))
        return;

    VIR_DEBUG("Restoring saved incoming migration error for domain %s: %s",
              name, err->message);
    virSetError(err);
    virFreeError(err);
}
5783 5784 5785 5786 5787 5788 5789 5790 5791 5792 5793 5794 5795 5796 5797 5798 5799 5800 5801 5802 5803 5804 5805 5806 5807 5808 5809 5810 5811 5812 5813 5814 5815 5816 5817 5818 5819 5820


/* don't ever pass NULL params with non zero nparams */
qemuMigrationCompressionPtr
qemuMigrationCompressionParse(virTypedParameterPtr params,
                              int nparams,
                              unsigned long flags)
{
    size_t i;
    qemuMigrationCompressionPtr compression = NULL;

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

    for (i = 0; i < nparams; i++) {
        int method;

        if (STRNEQ(params[i].field, VIR_MIGRATE_PARAM_COMPRESSION))
            continue;

        method = qemuMigrationCompressMethodTypeFromString(params[i].value.s);
        if (method < 0) {
            virReportError(VIR_ERR_INVALID_ARG,
                           _("Unsupported compression method '%s'"),
                           params[i].value.s);
            goto error;
        }

        if (compression->methods & (1ULL << method)) {
            virReportError(VIR_ERR_INVALID_ARG,
                           _("Compression method '%s' is specified twice"),
                           params[i].value.s);
            goto error;
        }

        compression->methods |= 1ULL << method;
    }

5821
#define GET_PARAM(PARAM, TYPE, VALUE)                                       \
5822 5823
    do {                                                                    \
        int rc;                                                             \
5824
        const char *par = VIR_MIGRATE_PARAM_COMPRESSION_ ## PARAM;          \
5825 5826
                                                                            \
        if ((rc = virTypedParamsGet ## TYPE(params, nparams,                \
5827
                                            par, &compression->VALUE)) < 0) \
5828 5829 5830
            goto error;                                                     \
                                                                            \
        if (rc == 1)                                                        \
5831
            compression->VALUE ## _set = true;                              \
5832 5833 5834
    } while (0)

    if (params) {
5835 5836 5837 5838
        GET_PARAM(MT_LEVEL, Int, level);
        GET_PARAM(MT_THREADS, Int, threads);
        GET_PARAM(MT_DTHREADS, Int, dthreads);
        GET_PARAM(XBZRLE_CACHE, ULLong, xbzrle_cache);
5839 5840 5841 5842
    }

#undef GET_PARAM

5843 5844 5845
    if ((compression->level_set ||
         compression->threads_set ||
         compression->dthreads_set) &&
5846 5847 5848 5849 5850 5851 5852 5853 5854 5855 5856 5857 5858
        !(compression->methods & (1ULL << QEMU_MIGRATION_COMPRESS_MT))) {
        virReportError(VIR_ERR_INVALID_ARG, "%s",
                       _("Turn multithread compression on to tune it"));
        goto error;
    }

    if (compression->xbzrle_cache_set &&
        !(compression->methods & (1ULL << QEMU_MIGRATION_COMPRESS_XBZRLE))) {
        virReportError(VIR_ERR_INVALID_ARG, "%s",
                       _("Turn xbzrle compression on to tune it"));
        goto error;
    }

5859 5860 5861 5862 5863 5864 5865 5866 5867 5868 5869 5870 5871 5872 5873 5874 5875 5876 5877
    if (!compression->methods && (flags & VIR_MIGRATE_COMPRESSED))
        compression->methods = 1ULL << QEMU_MIGRATION_COMPRESS_XBZRLE;

    return compression;

 error:
    VIR_FREE(compression);
    return NULL;
}

int
qemuMigrationCompressionDump(qemuMigrationCompressionPtr compression,
                             virTypedParameterPtr *params,
                             int *nparams,
                             int *maxparams,
                             unsigned long *flags)
{
    size_t i;

5878 5879
    if (compression->methods == 1ULL << QEMU_MIGRATION_COMPRESS_XBZRLE &&
        !compression->xbzrle_cache_set) {
5880 5881 5882 5883 5884 5885 5886 5887 5888 5889 5890 5891
        *flags |= VIR_MIGRATE_COMPRESSED;
        return 0;
    }

    for (i = 0; i < QEMU_MIGRATION_COMPRESS_LAST; ++i) {
        if ((compression->methods & (1ULL << i)) &&
            virTypedParamsAddString(params, nparams, maxparams,
                                    VIR_MIGRATE_PARAM_COMPRESSION,
                                    qemuMigrationCompressMethodTypeToString(i)) < 0)
            return -1;
    }

5892
    if (compression->level_set &&
5893 5894
        virTypedParamsAddInt(params, nparams, maxparams,
                             VIR_MIGRATE_PARAM_COMPRESSION_MT_LEVEL,
5895
                             compression->level) < 0)
5896 5897
        return -1;

5898
    if (compression->threads_set &&
5899 5900
        virTypedParamsAddInt(params, nparams, maxparams,
                             VIR_MIGRATE_PARAM_COMPRESSION_MT_THREADS,
5901
                             compression->threads) < 0)
5902 5903
        return -1;

5904
    if (compression->dthreads_set &&
5905 5906
        virTypedParamsAddInt(params, nparams, maxparams,
                             VIR_MIGRATE_PARAM_COMPRESSION_MT_DTHREADS,
5907
                             compression->dthreads) < 0)
5908 5909 5910 5911 5912 5913 5914 5915
        return -1;

    if (compression->xbzrle_cache_set &&
        virTypedParamsAddULLong(params, nparams, maxparams,
                                VIR_MIGRATE_PARAM_COMPRESSION_XBZRLE_CACHE,
                                compression->xbzrle_cache) < 0)
        return -1;

5916 5917
    return 0;
}
5918 5919 5920 5921 5922 5923 5924 5925 5926 5927 5928 5929 5930


/*
 * qemuMigrationReset:
 *
 * Reset all migration parameters so that the next job which internally uses
 * migration (save, managedsave, snapshots, dump) will not try to use them.
 */
void
qemuMigrationReset(virQEMUDriverPtr driver,
                   virDomainObjPtr vm,
                   qemuDomainAsyncJob job)
{
5931
    qemuMonitorMigrationCaps cap;
5932
    virErrorPtr err = virSaveLastError();
5933

5934
    if (!virDomainObjIsActive(vm))
5935
        goto cleanup;
5936

5937
    if (qemuMigrationResetTLS(driver, vm, job) < 0)
5938
        goto cleanup;
5939 5940

    for (cap = 0; cap < QEMU_MONITOR_MIGRATION_CAPS_LAST; cap++) {
5941 5942
        if (qemuMigrationCapsGet(vm, cap) &&
            qemuMigrationSetOption(driver, vm, cap, false, job) < 0)
5943 5944 5945 5946 5947 5948 5949
            goto cleanup;
    }

 cleanup:
    if (err) {
        virSetError(err);
        virFreeError(err);
5950
    }
5951
}
5952 5953 5954 5955 5956 5957 5958 5959 5960 5961 5962 5963 5964 5965 5966 5967 5968 5969 5970 5971 5972 5973 5974 5975 5976 5977 5978 5979 5980 5981 5982 5983 5984 5985 5986 5987 5988 5989 5990 5991 5992 5993 5994 5995 5996 5997 5998 5999 6000 6001 6002


int
qemuMigrationFetchMirrorStats(virQEMUDriverPtr driver,
                              virDomainObjPtr vm,
                              qemuDomainAsyncJob asyncJob,
                              qemuDomainJobInfoPtr jobInfo)
{
    size_t i;
    qemuDomainObjPrivatePtr priv = vm->privateData;
    bool nbd = false;
    virHashTablePtr blockinfo = NULL;
    qemuDomainMirrorStatsPtr stats = &jobInfo->mirrorStats;

    for (i = 0; i < vm->def->ndisks; i++) {
        virDomainDiskDefPtr disk = vm->def->disks[i];
        if (QEMU_DOMAIN_DISK_PRIVATE(disk)->migrating) {
            nbd = true;
            break;
        }
    }

    if (!nbd)
        return 0;

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

    blockinfo = qemuMonitorGetAllBlockJobInfo(priv->mon);

    if (qemuDomainObjExitMonitor(driver, vm) < 0 || !blockinfo)
        return -1;

    memset(stats, 0, sizeof(*stats));

    for (i = 0; i < vm->def->ndisks; i++) {
        virDomainDiskDefPtr disk = vm->def->disks[i];
        qemuDomainDiskPrivatePtr diskPriv = QEMU_DOMAIN_DISK_PRIVATE(disk);
        qemuMonitorBlockJobInfoPtr data;

        if (!diskPriv->migrating ||
            !(data = virHashLookup(blockinfo, disk->info.alias)))
            continue;

        stats->transferred += data->cur;
        stats->total += data->end;
    }

    virHashFree(blockinfo);
    return 0;
}
6003 6004 6005 6006 6007 6008 6009 6010 6011 6012 6013 6014 6015 6016


bool
qemuMigrationCapsGet(virDomainObjPtr vm,
                     qemuMonitorMigrationCaps cap)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    bool enabled = false;

    if (priv->migrationCaps)
        ignore_value(virBitmapGetBit(priv->migrationCaps, cap, &enabled));

    return enabled;
}