qemu_migration.c 203.5 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
#ifdef WITH_GNUTLS
28 29 30
# include <gnutls/gnutls.h>
# include <gnutls/x509.h>
#endif
31
#include <fcntl.h>
32
#include <poll.h>
33 34 35 36 37 38

#include "qemu_migration.h"
#include "qemu_monitor.h"
#include "qemu_domain.h"
#include "qemu_process.h"
#include "qemu_capabilities.h"
39
#include "qemu_command.h"
40
#include "qemu_cgroup.h"
41
#include "qemu_hotplug.h"
42
#include "qemu_blockjob.h"
43

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

#define VIR_FROM_THIS VIR_FROM_QEMU

67 68
VIR_LOG_INIT("qemu.qemu_migration");

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

82
enum qemuMigrationCookieFlags {
83
    QEMU_MIGRATION_COOKIE_FLAG_GRAPHICS,
84
    QEMU_MIGRATION_COOKIE_FLAG_LOCKSTATE,
85
    QEMU_MIGRATION_COOKIE_FLAG_PERSISTENT,
86
    QEMU_MIGRATION_COOKIE_FLAG_NETWORK,
87
    QEMU_MIGRATION_COOKIE_FLAG_NBD,
88
    QEMU_MIGRATION_COOKIE_FLAG_STATS,
89
    QEMU_MIGRATION_COOKIE_FLAG_MEMORY_HOTPLUG,
90 91 92 93 94 95 96

    QEMU_MIGRATION_COOKIE_FLAG_LAST
};

VIR_ENUM_DECL(qemuMigrationCookieFlag);
VIR_ENUM_IMPL(qemuMigrationCookieFlag,
              QEMU_MIGRATION_COOKIE_FLAG_LAST,
97 98 99 100
              "graphics",
              "lockstate",
              "persistent",
              "network",
101
              "nbd",
102 103
              "statistics",
              "memory-hotplug");
104 105 106

enum qemuMigrationCookieFeatures {
    QEMU_MIGRATION_COOKIE_GRAPHICS  = (1 << QEMU_MIGRATION_COOKIE_FLAG_GRAPHICS),
107
    QEMU_MIGRATION_COOKIE_LOCKSTATE = (1 << QEMU_MIGRATION_COOKIE_FLAG_LOCKSTATE),
108
    QEMU_MIGRATION_COOKIE_PERSISTENT = (1 << QEMU_MIGRATION_COOKIE_FLAG_PERSISTENT),
109
    QEMU_MIGRATION_COOKIE_NETWORK = (1 << QEMU_MIGRATION_COOKIE_FLAG_NETWORK),
110
    QEMU_MIGRATION_COOKIE_NBD = (1 << QEMU_MIGRATION_COOKIE_FLAG_NBD),
111
    QEMU_MIGRATION_COOKIE_STATS = (1 << QEMU_MIGRATION_COOKIE_FLAG_STATS),
112
    QEMU_MIGRATION_COOKIE_MEMORY_HOTPLUG = (1 << QEMU_MIGRATION_COOKIE_FLAG_MEMORY_HOTPLUG),
113 114 115 116 117 118 119 120 121 122 123 124
};

typedef struct _qemuMigrationCookieGraphics qemuMigrationCookieGraphics;
typedef qemuMigrationCookieGraphics *qemuMigrationCookieGraphicsPtr;
struct _qemuMigrationCookieGraphics {
    int type;
    int port;
    int tlsPort;
    char *listen;
    char *tlsSubject;
};

125 126 127 128 129 130
typedef struct _qemuMigrationCookieNetData qemuMigrationCookieNetData;
typedef qemuMigrationCookieNetData *qemuMigrationCookieNetDataPtr;
struct _qemuMigrationCookieNetData {
    int vporttype; /* enum virNetDevVPortProfile */

    /*
131
     * Array of pointers to saved data. Each VIF will have its own
132 133 134 135 136 137 138 139 140 141 142 143 144 145
     * data to transfer.
     */
    char *portdata;
};

typedef struct _qemuMigrationCookieNetwork qemuMigrationCookieNetwork;
typedef qemuMigrationCookieNetwork *qemuMigrationCookieNetworkPtr;
struct _qemuMigrationCookieNetwork {
    /* How many virtual NICs are we saving data for? */
    int nnets;

    qemuMigrationCookieNetDataPtr net;
};

146 147 148 149
typedef struct _qemuMigrationCookieNBD qemuMigrationCookieNBD;
typedef qemuMigrationCookieNBD *qemuMigrationCookieNBDPtr;
struct _qemuMigrationCookieNBD {
    int port; /* on which port does NBD server listen for incoming data */
150 151 152 153 154 155

    size_t ndisks;  /* Number of items in @disk array */
    struct {
        char *target;                   /* Disk target */
        unsigned long long capacity;    /* And its capacity */
    } *disks;
156 157
};

158 159 160
typedef struct _qemuMigrationCookie qemuMigrationCookie;
typedef qemuMigrationCookie *qemuMigrationCookiePtr;
struct _qemuMigrationCookie {
E
Eric Blake 已提交
161 162
    unsigned int flags;
    unsigned int flagsMandatory;
163 164

    /* Host properties */
165 166 167 168
    unsigned char localHostuuid[VIR_UUID_BUFLEN];
    unsigned char remoteHostuuid[VIR_UUID_BUFLEN];
    char *localHostname;
    char *remoteHostname;
169 170 171 172

    /* Guest properties */
    unsigned char uuid[VIR_UUID_BUFLEN];
    char *name;
173

174 175 176 177
    /* If (flags & QEMU_MIGRATION_COOKIE_LOCKSTATE) */
    char *lockState;
    char *lockDriver;

178 179
    /* If (flags & QEMU_MIGRATION_COOKIE_GRAPHICS) */
    qemuMigrationCookieGraphicsPtr graphics;
180 181 182

    /* If (flags & QEMU_MIGRATION_COOKIE_PERSISTENT) */
    virDomainDefPtr persistent;
183 184 185

    /* If (flags & QEMU_MIGRATION_COOKIE_NETWORK) */
    qemuMigrationCookieNetworkPtr network;
186 187 188

    /* If (flags & QEMU_MIGRATION_COOKIE_NBD) */
    qemuMigrationCookieNBDPtr nbd;
189 190 191

    /* If (flags & QEMU_MIGRATION_COOKIE_STATS) */
    qemuDomainJobInfoPtr jobInfo;
192 193
};

194 195 196 197 198 199 200 201 202
static void qemuMigrationCookieGraphicsFree(qemuMigrationCookieGraphicsPtr grap)
{
    if (!grap)
        return;
    VIR_FREE(grap->listen);
    VIR_FREE(grap->tlsSubject);
    VIR_FREE(grap);
}

203

204 205 206
static void
qemuMigrationCookieNetworkFree(qemuMigrationCookieNetworkPtr network)
{
207
    size_t i;
208 209 210 211 212 213 214 215 216 217 218 219 220

    if (!network)
        return;

    if (network->net) {
        for (i = 0; i < network->nnets; i++)
            VIR_FREE(network->net[i].portdata);
    }
    VIR_FREE(network->net);
    VIR_FREE(network);
}


221 222 223 224 225 226 227 228 229 230 231 232
static void qemuMigrationCookieNBDFree(qemuMigrationCookieNBDPtr nbd)
{
    if (!nbd)
        return;

    while (nbd->ndisks)
        VIR_FREE(nbd->disks[--nbd->ndisks].target);
    VIR_FREE(nbd->disks);
    VIR_FREE(nbd);
}


233 234 235 236 237
static void qemuMigrationCookieFree(qemuMigrationCookiePtr mig)
{
    if (!mig)
        return;

238 239
    qemuMigrationCookieGraphicsFree(mig->graphics);
    qemuMigrationCookieNetworkFree(mig->network);
240
    qemuMigrationCookieNBDFree(mig->nbd);
241

242 243
    VIR_FREE(mig->localHostname);
    VIR_FREE(mig->remoteHostname);
244
    VIR_FREE(mig->name);
245 246
    VIR_FREE(mig->lockState);
    VIR_FREE(mig->lockDriver);
247
    VIR_FREE(mig->jobInfo);
248 249 250 251
    VIR_FREE(mig);
}


252
#ifdef WITH_GNUTLS
253 254 255 256 257 258 259 260 261 262 263 264
static char *
qemuDomainExtractTLSSubject(const char *certdir)
{
    char *certfile = NULL;
    char *subject = NULL;
    char *pemdata = NULL;
    gnutls_datum_t pemdatum;
    gnutls_x509_crt_t cert;
    int ret;
    size_t subjectlen;

    if (virAsprintf(&certfile, "%s/server-cert.pem", certdir) < 0)
265
        goto error;
266 267

    if (virFileReadAll(certfile, 8192, &pemdata) < 0) {
268 269
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("unable to read server cert %s"), certfile);
270 271 272 273 274
        goto error;
    }

    ret = gnutls_x509_crt_init(&cert);
    if (ret < 0) {
275 276 277
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("cannot initialize cert object: %s"),
                       gnutls_strerror(ret));
278 279 280 281 282 283 284 285
        goto error;
    }

    pemdatum.data = (unsigned char *)pemdata;
    pemdatum.size = strlen(pemdata);

    ret = gnutls_x509_crt_import(cert, &pemdatum, GNUTLS_X509_FMT_PEM);
    if (ret < 0) {
286 287 288
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("cannot load cert data from %s: %s"),
                       certfile, gnutls_strerror(ret));
289 290 291 292 293
        goto error;
    }

    subjectlen = 1024;
    if (VIR_ALLOC_N(subject, subjectlen+1) < 0)
294
        goto error;
295 296 297 298 299 300 301 302 303

    gnutls_x509_crt_get_dn(cert, subject, &subjectlen);
    subject[subjectlen] = '\0';

    VIR_FREE(certfile);
    VIR_FREE(pemdata);

    return subject;

304
 error:
305 306 307 308
    VIR_FREE(certfile);
    VIR_FREE(pemdata);
    return NULL;
}
309
#endif
310 311

static qemuMigrationCookieGraphicsPtr
312
qemuMigrationCookieGraphicsAlloc(virQEMUDriverPtr driver,
313 314 315 316
                                 virDomainGraphicsDefPtr def)
{
    qemuMigrationCookieGraphicsPtr mig = NULL;
    const char *listenAddr;
317
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
318 319

    if (VIR_ALLOC(mig) < 0)
320
        goto error;
321 322 323 324

    mig->type = def->type;
    if (mig->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC) {
        mig->port = def->data.vnc.port;
325
        listenAddr = virDomainGraphicsListenGetAddress(def, 0);
326
        if (!listenAddr)
327
            listenAddr = cfg->vncListen;
328

329
#ifdef WITH_GNUTLS
330 331
        if (cfg->vncTLS &&
            !(mig->tlsSubject = qemuDomainExtractTLSSubject(cfg->vncTLSx509certdir)))
332
            goto error;
333
#endif
334 335
    } else {
        mig->port = def->data.spice.port;
336
        if (cfg->spiceTLS)
337 338 339
            mig->tlsPort = def->data.spice.tlsPort;
        else
            mig->tlsPort = -1;
340
        listenAddr = virDomainGraphicsListenGetAddress(def, 0);
341
        if (!listenAddr)
342
            listenAddr = cfg->spiceListen;
343

344
#ifdef WITH_GNUTLS
345 346
        if (cfg->spiceTLS &&
            !(mig->tlsSubject = qemuDomainExtractTLSSubject(cfg->spiceTLSx509certdir)))
347
            goto error;
348
#endif
349
    }
350 351
    if (VIR_STRDUP(mig->listen, listenAddr) < 0)
        goto error;
352

353
    virObjectUnref(cfg);
354 355
    return mig;

356
 error:
357
    qemuMigrationCookieGraphicsFree(mig);
358
    virObjectUnref(cfg);
359 360 361 362
    return NULL;
}


363
static qemuMigrationCookieNetworkPtr
364
qemuMigrationCookieNetworkAlloc(virQEMUDriverPtr driver ATTRIBUTE_UNUSED,
365 366 367
                                virDomainDefPtr def)
{
    qemuMigrationCookieNetworkPtr mig;
368
    size_t i;
369 370

    if (VIR_ALLOC(mig) < 0)
371
        goto error;
372 373 374 375

    mig->nnets = def->nnets;

    if (VIR_ALLOC_N(mig->net, def->nnets) <0)
376
        goto error;
377 378 379 380 381 382 383 384 385 386 387 388 389 390 391

    for (i = 0; i < def->nnets; i++) {
        virDomainNetDefPtr netptr;
        virNetDevVPortProfilePtr vport;

        netptr = def->nets[i];
        vport = virDomainNetGetActualVirtPortProfile(netptr);

        if (vport) {
            mig->net[i].vporttype = vport->virtPortType;

            switch (vport->virtPortType) {
            case VIR_NETDEV_VPORT_PROFILE_NONE:
            case VIR_NETDEV_VPORT_PROFILE_8021QBG:
            case VIR_NETDEV_VPORT_PROFILE_8021QBH:
392
               break;
393
            case VIR_NETDEV_VPORT_PROFILE_OPENVSWITCH:
394 395
                if (virNetDevOpenvswitchGetMigrateData(&mig->net[i].portdata,
                                                       netptr->ifname) != 0) {
J
Jiri Denemark 已提交
396 397 398
                        virReportError(VIR_ERR_INTERNAL_ERROR,
                                       _("Unable to run command to get OVS port data for "
                                         "interface %s"), netptr->ifname);
399 400 401
                        goto error;
                }
                break;
402 403 404 405 406 407 408
            default:
                break;
            }
        }
    }
    return mig;

409
 error:
410 411 412 413
    qemuMigrationCookieNetworkFree(mig);
    return NULL;
}

414 415 416
static qemuMigrationCookiePtr
qemuMigrationCookieNew(virDomainObjPtr dom)
{
J
Jiri Denemark 已提交
417
    qemuDomainObjPrivatePtr priv = dom->privateData;
418
    qemuMigrationCookiePtr mig = NULL;
J
Jiri Denemark 已提交
419
    const char *name;
420 421

    if (VIR_ALLOC(mig) < 0)
422
        goto error;
423

J
Jiri Denemark 已提交
424 425 426 427
    if (priv->origname)
        name = priv->origname;
    else
        name = dom->def->name;
428 429
    if (VIR_STRDUP(mig->name, name) < 0)
        goto error;
430 431
    memcpy(mig->uuid, dom->def->uuid, VIR_UUID_BUFLEN);

432
    if (!(mig->localHostname = virGetHostname()))
433
        goto error;
434
    if (virGetHostUUID(mig->localHostuuid) < 0) {
435 436
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Unable to obtain host UUID"));
437 438 439 440 441
        goto error;
    }

    return mig;

442
 error:
443 444 445 446 447
    qemuMigrationCookieFree(mig);
    return NULL;
}


448 449
static int
qemuMigrationCookieAddGraphics(qemuMigrationCookiePtr mig,
450
                               virQEMUDriverPtr driver,
451 452
                               virDomainObjPtr dom)
{
453 454
    size_t i = 0;

455
    if (mig->flags & QEMU_MIGRATION_COOKIE_GRAPHICS) {
456 457
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Migration graphics data already present"));
458 459 460
        return -1;
    }

461 462 463 464 465 466 467 468
    for (i = 0; i < dom->def->ngraphics; i++) {
       if (dom->def->graphics[i]->type == VIR_DOMAIN_GRAPHICS_TYPE_SPICE) {
           if (!(mig->graphics =
                 qemuMigrationCookieGraphicsAlloc(driver, dom->def->graphics[i])))
               return -1;
           mig->flags |= QEMU_MIGRATION_COOKIE_GRAPHICS;
           break;
       }
469
    }
470 471 472 473 474

    return 0;
}


475 476
static int
qemuMigrationCookieAddLockstate(qemuMigrationCookiePtr mig,
477
                                virQEMUDriverPtr driver,
478 479 480 481 482
                                virDomainObjPtr dom)
{
    qemuDomainObjPrivatePtr priv = dom->privateData;

    if (mig->flags & QEMU_MIGRATION_COOKIE_LOCKSTATE) {
483 484
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Migration lockstate data already present"));
485 486 487 488
        return -1;
    }

    if (virDomainObjGetState(dom, NULL) == VIR_DOMAIN_PAUSED) {
489
        if (VIR_STRDUP(mig->lockState, priv->lockState) < 0)
490 491 492 493 494 495
            return -1;
    } else {
        if (virDomainLockProcessInquire(driver->lockManager, dom, &mig->lockState) < 0)
            return -1;
    }

496
    if (VIR_STRDUP(mig->lockDriver, virLockManagerPluginGetName(driver->lockManager)) < 0) {
497 498 499 500 501 502 503 504 505 506 507
        VIR_FREE(mig->lockState);
        return -1;
    }

    mig->flags |= QEMU_MIGRATION_COOKIE_LOCKSTATE;
    mig->flagsMandatory |= QEMU_MIGRATION_COOKIE_LOCKSTATE;

    return 0;
}


508 509 510 511 512
static int
qemuMigrationCookieAddPersistent(qemuMigrationCookiePtr mig,
                                 virDomainObjPtr dom)
{
    if (mig->flags & QEMU_MIGRATION_COOKIE_PERSISTENT) {
513 514
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Migration persistent data already present"));
515 516 517 518 519 520 521 522 523 524 525 526 527
        return -1;
    }

    if (!dom->newDef)
        return 0;

    mig->persistent = dom->newDef;
    mig->flags |= QEMU_MIGRATION_COOKIE_PERSISTENT;
    mig->flagsMandatory |= QEMU_MIGRATION_COOKIE_PERSISTENT;
    return 0;
}


528 529 530 531 532 533 534 535 536 537 538 539 540
static virDomainDefPtr
qemuMigrationCookieGetPersistent(qemuMigrationCookiePtr mig)
{
    virDomainDefPtr def = mig->persistent;

    mig->persistent = NULL;
    mig->flags &= ~QEMU_MIGRATION_COOKIE_PERSISTENT;
    mig->flagsMandatory &= ~QEMU_MIGRATION_COOKIE_PERSISTENT;

    return def;
}


541 542
static int
qemuMigrationCookieAddNetwork(qemuMigrationCookiePtr mig,
543
                              virQEMUDriverPtr driver,
544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561
                              virDomainObjPtr dom)
{
    if (mig->flags & QEMU_MIGRATION_COOKIE_NETWORK) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Network migration data already present"));
        return -1;
    }

    if (dom->def->nnets > 0) {
        mig->network = qemuMigrationCookieNetworkAlloc(driver, dom->def);
        if (!mig->network)
            return -1;
        mig->flags |= QEMU_MIGRATION_COOKIE_NETWORK;
    }

    return 0;
}

562

563 564
static int
qemuMigrationCookieAddNBD(qemuMigrationCookiePtr mig,
565
                          virQEMUDriverPtr driver,
566 567 568
                          virDomainObjPtr vm)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
569 570
    virHashTablePtr stats = NULL;
    size_t i;
571
    int ret = -1, rc;
572 573 574

    /* It is not a bug if there already is a NBD data */
    if (!mig->nbd &&
575
        VIR_ALLOC(mig->nbd) < 0)
576 577
        return -1;

578 579 580 581 582 583 584 585 586 587 588 589 590
    if (vm->def->ndisks &&
        VIR_ALLOC_N(mig->nbd->disks, vm->def->ndisks) < 0)
        return -1;
    mig->nbd->ndisks = 0;

    for (i = 0; i < vm->def->ndisks; i++) {
        virDomainDiskDefPtr disk = vm->def->disks[i];
        qemuBlockStats *entry;

        if (!stats) {
            if (!(stats = virHashCreate(10, virHashValueFree)))
                goto cleanup;

591 592 593
            if (qemuDomainObjEnterMonitorAsync(driver, vm,
                                               priv->job.asyncJob) < 0)
                goto cleanup;
594 595
            rc = qemuMonitorBlockStatsUpdateCapacity(priv->mon, stats, false);
            if (qemuDomainObjExitMonitor(driver, vm) < 0)
596
                goto cleanup;
597
            if (rc < 0)
598 599 600 601 602 603 604 605 606 607 608 609 610 611
                goto cleanup;
        }

        if (!disk->info.alias ||
            !(entry = virHashLookup(stats, disk->info.alias)))
            continue;

        if (VIR_STRDUP(mig->nbd->disks[mig->nbd->ndisks].target,
                       disk->dst) < 0)
            goto cleanup;
        mig->nbd->disks[mig->nbd->ndisks].capacity = entry->capacity;
        mig->nbd->ndisks++;
    }

612 613 614
    mig->nbd->port = priv->nbdPort;
    mig->flags |= QEMU_MIGRATION_COOKIE_NBD;

615 616 617 618
    ret = 0;
 cleanup:
    virHashFree(stats);
    return ret;
619 620 621
}


622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640
static int
qemuMigrationCookieAddStatistics(qemuMigrationCookiePtr mig,
                                 virDomainObjPtr vm)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;

    if (!priv->job.completed)
        return 0;

    if (!mig->jobInfo && VIR_ALLOC(mig->jobInfo) < 0)
        return -1;

    *mig->jobInfo = *priv->job.completed;
    mig->flags |= QEMU_MIGRATION_COOKIE_STATS;

    return 0;
}


641 642 643
static void qemuMigrationCookieGraphicsXMLFormat(virBufferPtr buf,
                                                 qemuMigrationCookieGraphicsPtr grap)
{
644
    virBufferAsprintf(buf, "<graphics type='%s' port='%d' listen='%s'",
645 646 647 648 649 650
                      virDomainGraphicsTypeToString(grap->type),
                      grap->port, grap->listen);
    if (grap->type == VIR_DOMAIN_GRAPHICS_TYPE_SPICE)
        virBufferAsprintf(buf, " tlsPort='%d'", grap->tlsPort);
    if (grap->tlsSubject) {
        virBufferAddLit(buf, ">\n");
651 652 653 654
        virBufferAdjustIndent(buf, 2);
        virBufferEscapeString(buf, "<cert info='subject' value='%s'/>\n", grap->tlsSubject);
        virBufferAdjustIndent(buf, -2);
        virBufferAddLit(buf, "</graphics>\n");
655 656 657 658 659 660
    } else {
        virBufferAddLit(buf, "/>\n");
    }
}


661 662 663 664
static void
qemuMigrationCookieNetworkXMLFormat(virBufferPtr buf,
                                    qemuMigrationCookieNetworkPtr optr)
{
665
    size_t i;
666 667 668 669 670 671
    bool empty = true;

    for (i = 0; i < optr->nnets; i++) {
        /* If optr->net[i].vporttype is not set, there is nothing to transfer */
        if (optr->net[i].vporttype != VIR_NETDEV_VPORT_PROFILE_NONE) {
            if (empty) {
672 673
                virBufferAddLit(buf, "<network>\n");
                virBufferAdjustIndent(buf, 2);
674 675
                empty = false;
            }
676
            virBufferAsprintf(buf, "<interface index='%zu' vporttype='%s'",
677 678 679
                              i, virNetDevVPortTypeToString(optr->net[i].vporttype));
            if (optr->net[i].portdata) {
                virBufferAddLit(buf, ">\n");
680 681
                virBufferAdjustIndent(buf, 2);
                virBufferEscapeString(buf, "<portdata>%s</portdata>\n",
682
                                      optr->net[i].portdata);
683 684
                virBufferAdjustIndent(buf, -2);
                virBufferAddLit(buf, "</interface>\n");
685 686 687 688 689
            } else {
                virBufferAddLit(buf, "/>\n");
            }
        }
    }
690 691 692 693
    if (!empty) {
        virBufferAdjustIndent(buf, -2);
        virBufferAddLit(buf, "</network>\n");
    }
694 695 696
}


697 698 699 700 701 702 703 704 705
static void
qemuMigrationCookieStatisticsXMLFormat(virBufferPtr buf,
                                       qemuDomainJobInfoPtr jobInfo)
{
    qemuMonitorMigrationStatus *status = &jobInfo->status;

    virBufferAddLit(buf, "<statistics>\n");
    virBufferAdjustIndent(buf, 2);

706 707
    virBufferAsprintf(buf, "<started>%llu</started>\n", jobInfo->started);
    virBufferAsprintf(buf, "<stopped>%llu</stopped>\n", jobInfo->stopped);
708 709 710
    virBufferAsprintf(buf, "<sent>%llu</sent>\n", jobInfo->sent);
    if (jobInfo->timeDeltaSet)
        virBufferAsprintf(buf, "<delta>%lld</delta>\n", jobInfo->timeDelta);
711

712 713 714 715 716 717 718 719 720 721
    virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                      VIR_DOMAIN_JOB_TIME_ELAPSED,
                      jobInfo->timeElapsed);
    virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                      VIR_DOMAIN_JOB_TIME_REMAINING,
                      jobInfo->timeRemaining);
    if (status->downtime_set)
        virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                          VIR_DOMAIN_JOB_DOWNTIME,
                          status->downtime);
722 723 724 725
    if (status->setup_time_set)
        virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                          VIR_DOMAIN_JOB_SETUP_TIME,
                          status->setup_time);
726 727 728 729 730 731 732 733 734 735

    virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                      VIR_DOMAIN_JOB_MEMORY_TOTAL,
                      status->ram_total);
    virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                      VIR_DOMAIN_JOB_MEMORY_PROCESSED,
                      status->ram_transferred);
    virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                      VIR_DOMAIN_JOB_MEMORY_REMAINING,
                      status->ram_remaining);
736 737 738
    virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                      VIR_DOMAIN_JOB_MEMORY_BPS,
                      status->ram_bps);
739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760

    if (status->ram_duplicate_set) {
        virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                          VIR_DOMAIN_JOB_MEMORY_CONSTANT,
                          status->ram_duplicate);
        virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                          VIR_DOMAIN_JOB_MEMORY_NORMAL,
                          status->ram_normal);
        virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                          VIR_DOMAIN_JOB_MEMORY_NORMAL_BYTES,
                          status->ram_normal_bytes);
    }

    virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                      VIR_DOMAIN_JOB_DISK_TOTAL,
                      status->disk_total);
    virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                      VIR_DOMAIN_JOB_DISK_PROCESSED,
                      status->disk_transferred);
    virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                      VIR_DOMAIN_JOB_DISK_REMAINING,
                      status->disk_remaining);
761 762 763
    virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                      VIR_DOMAIN_JOB_DISK_BPS,
                      status->disk_bps);
764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787

    if (status->xbzrle_set) {
        virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                          VIR_DOMAIN_JOB_COMPRESSION_CACHE,
                          status->xbzrle_cache_size);
        virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                          VIR_DOMAIN_JOB_COMPRESSION_BYTES,
                          status->xbzrle_bytes);
        virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                          VIR_DOMAIN_JOB_COMPRESSION_PAGES,
                          status->xbzrle_pages);
        virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                          VIR_DOMAIN_JOB_COMPRESSION_CACHE_MISSES,
                          status->xbzrle_cache_miss);
        virBufferAsprintf(buf, "<%1$s>%2$llu</%1$s>\n",
                          VIR_DOMAIN_JOB_COMPRESSION_OVERFLOW,
                          status->xbzrle_overflow);
    }

    virBufferAdjustIndent(buf, -2);
    virBufferAddLit(buf, "</statistics>\n");
}


788
static int
789
qemuMigrationCookieXMLFormat(virQEMUDriverPtr driver,
790
                             virBufferPtr buf,
791
                             qemuMigrationCookiePtr mig)
792 793 794
{
    char uuidstr[VIR_UUID_STRING_BUFLEN];
    char hostuuidstr[VIR_UUID_STRING_BUFLEN];
795
    size_t i;
796 797

    virUUIDFormat(mig->uuid, uuidstr);
798
    virUUIDFormat(mig->localHostuuid, hostuuidstr);
799

800
    virBufferAddLit(buf, "<qemu-migration>\n");
801 802 803 804 805
    virBufferAdjustIndent(buf, 2);
    virBufferEscapeString(buf, "<name>%s</name>\n", mig->name);
    virBufferAsprintf(buf, "<uuid>%s</uuid>\n", uuidstr);
    virBufferEscapeString(buf, "<hostname>%s</hostname>\n", mig->localHostname);
    virBufferAsprintf(buf, "<hostuuid>%s</hostuuid>\n", hostuuidstr);
806

807
    for (i = 0; i < QEMU_MIGRATION_COOKIE_FLAG_LAST; i++) {
808
        if (mig->flagsMandatory & (1 << i))
809
            virBufferAsprintf(buf, "<feature name='%s'/>\n",
810 811 812
                              qemuMigrationCookieFlagTypeToString(i));
    }

813 814
    if ((mig->flags & QEMU_MIGRATION_COOKIE_GRAPHICS) &&
        mig->graphics)
815 816
        qemuMigrationCookieGraphicsXMLFormat(buf, mig->graphics);

817 818
    if ((mig->flags & QEMU_MIGRATION_COOKIE_LOCKSTATE) &&
        mig->lockState) {
819
        virBufferAsprintf(buf, "<lockstate driver='%s'>\n",
820
                          mig->lockDriver);
821 822
        virBufferAdjustIndent(buf, 2);
        virBufferAsprintf(buf, "<leases>%s</leases>\n",
823
                          mig->lockState);
824 825
        virBufferAdjustIndent(buf, -2);
        virBufferAddLit(buf, "</lockstate>\n");
826 827
    }

828 829
    if ((mig->flags & QEMU_MIGRATION_COOKIE_PERSISTENT) &&
        mig->persistent) {
830 831 832
        if (qemuDomainDefFormatBuf(driver,
                                   mig->persistent,
                                   VIR_DOMAIN_XML_INACTIVE |
833 834
                                   VIR_DOMAIN_XML_SECURE |
                                   VIR_DOMAIN_XML_MIGRATABLE,
835
                                   buf) < 0)
836
            return -1;
837 838
    }

839 840 841
    if ((mig->flags & QEMU_MIGRATION_COOKIE_NETWORK) && mig->network)
        qemuMigrationCookieNetworkXMLFormat(buf, mig->network);

842
    if ((mig->flags & QEMU_MIGRATION_COOKIE_NBD) && mig->nbd) {
843
        virBufferAddLit(buf, "<nbd");
844 845
        if (mig->nbd->port)
            virBufferAsprintf(buf, " port='%d'", mig->nbd->port);
846 847 848 849 850 851 852 853 854 855 856 857 858 859
        if (mig->nbd->ndisks) {
            virBufferAddLit(buf, ">\n");
            virBufferAdjustIndent(buf, 2);
            for (i = 0; i < mig->nbd->ndisks; i++) {
                virBufferEscapeString(buf, "<disk target='%s'",
                                      mig->nbd->disks[i].target);
                virBufferAsprintf(buf, " capacity='%llu'/>\n",
                                  mig->nbd->disks[i].capacity);
            }
            virBufferAdjustIndent(buf, -2);
            virBufferAddLit(buf, "</nbd>\n");
        } else {
            virBufferAddLit(buf, "/>\n");
        }
860 861
    }

862 863 864
    if (mig->flags & QEMU_MIGRATION_COOKIE_STATS && mig->jobInfo)
        qemuMigrationCookieStatisticsXMLFormat(buf, mig->jobInfo);

865
    virBufferAdjustIndent(buf, -2);
866
    virBufferAddLit(buf, "</qemu-migration>\n");
867
    return 0;
868 869 870
}


871
static char *qemuMigrationCookieXMLFormatStr(virQEMUDriverPtr driver,
872
                                             qemuMigrationCookiePtr mig)
873 874 875
{
    virBuffer buf = VIR_BUFFER_INITIALIZER;

876
    if (qemuMigrationCookieXMLFormat(driver, &buf, mig) < 0) {
877 878 879
        virBufferFreeAndReset(&buf);
        return NULL;
    }
880

881
    if (virBufferCheckError(&buf) < 0)
882 883 884 885 886 887
        return NULL;

    return virBufferContentAndReset(&buf);
}


888 889 890 891 892 893 894
static qemuMigrationCookieGraphicsPtr
qemuMigrationCookieGraphicsXMLParse(xmlXPathContextPtr ctxt)
{
    qemuMigrationCookieGraphicsPtr grap;
    char *tmp;

    if (VIR_ALLOC(grap) < 0)
895
        goto error;
896 897

    if (!(tmp = virXPathString("string(./graphics/@type)", ctxt))) {
898 899
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("missing type attribute in migration data"));
900 901 902
        goto error;
    }
    if ((grap->type = virDomainGraphicsTypeFromString(tmp)) < 0) {
903 904
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("unknown graphics type %s"), tmp);
905 906 907
        VIR_FREE(tmp);
        goto error;
    }
E
Eric Blake 已提交
908
    VIR_FREE(tmp);
909
    if (virXPathInt("string(./graphics/@port)", ctxt, &grap->port) < 0) {
910 911
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("missing port attribute in migration data"));
912 913 914 915
        goto error;
    }
    if (grap->type == VIR_DOMAIN_GRAPHICS_TYPE_SPICE) {
        if (virXPathInt("string(./graphics/@tlsPort)", ctxt, &grap->tlsPort) < 0) {
916 917
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           "%s", _("missing tlsPort attribute in migration data"));
918 919 920 921
            goto error;
        }
    }
    if (!(grap->listen = virXPathString("string(./graphics/@listen)", ctxt))) {
922 923
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("missing listen attribute in migration data"));
924 925 926
        goto error;
    }
    /* Optional */
927
    grap->tlsSubject = virXPathString("string(./graphics/cert[@info='subject']/@value)", ctxt);
928 929 930

    return grap;

931
 error:
932 933 934 935 936
    qemuMigrationCookieGraphicsFree(grap);
    return NULL;
}


937 938 939 940
static qemuMigrationCookieNetworkPtr
qemuMigrationCookieNetworkXMLParse(xmlXPathContextPtr ctxt)
{
    qemuMigrationCookieNetworkPtr optr;
941
    size_t i;
942 943 944 945 946 947
    int n;
    xmlNodePtr *interfaces = NULL;
    char *vporttype;
    xmlNodePtr save_ctxt = ctxt->node;

    if (VIR_ALLOC(optr) < 0)
948
        goto error;
949 950 951 952 953 954 955 956

    if ((n = virXPathNodeSet("./network/interface", ctxt, &interfaces)) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("missing interface information"));
        goto error;
    }

    optr->nnets = n;
957
    if (VIR_ALLOC_N(optr->net, optr->nnets) < 0)
958
        goto error;
959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974

    for (i = 0; i < n; i++) {
        /* portdata is optional, and may not exist */
        ctxt->node = interfaces[i];
        optr->net[i].portdata = virXPathString("string(./portdata[1])", ctxt);

        if (!(vporttype = virXMLPropString(interfaces[i], "vporttype"))) {
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           "%s", _("missing vporttype attribute in migration data"));
            goto error;
        }
        optr->net[i].vporttype = virNetDevVPortTypeFromString(vporttype);
    }

    VIR_FREE(interfaces);

975
 cleanup:
976 977 978
    ctxt->node = save_ctxt;
    return optr;

979
 error:
980 981 982 983 984 985 986
    VIR_FREE(interfaces);
    qemuMigrationCookieNetworkFree(optr);
    optr = NULL;
    goto cleanup;
}


987 988 989 990 991 992 993 994 995 996 997 998 999 1000 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 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050
static qemuMigrationCookieNBDPtr
qemuMigrationCookieNBDXMLParse(xmlXPathContextPtr ctxt)
{
    qemuMigrationCookieNBDPtr ret = NULL;
    char *port = NULL, *capacity = NULL;
    size_t i;
    int n;
    xmlNodePtr *disks = NULL;
    xmlNodePtr save_ctxt = ctxt->node;

    if (VIR_ALLOC(ret) < 0)
        goto error;

    port = virXPathString("string(./nbd/@port)", ctxt);
    if (port && virStrToLong_i(port, NULL, 10, &ret->port) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Malformed nbd port '%s'"),
                       port);
        goto error;
    }

    /* Now check if source sent a list of disks to prealloc. We might be
     * talking to an older server, so it's not an error if the list is
     * missing. */
    if ((n = virXPathNodeSet("./nbd/disk", ctxt, &disks)) > 0) {
        if (VIR_ALLOC_N(ret->disks, n) < 0)
            goto error;
        ret->ndisks = n;

        for (i = 0; i < n; i++) {
            ctxt->node = disks[i];
            VIR_FREE(capacity);

            if (!(ret->disks[i].target = virXPathString("string(./@target)", ctxt))) {
                virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                               _("Malformed disk target"));
                goto error;
            }

            capacity = virXPathString("string(./@capacity)", ctxt);
            if (!capacity ||
                virStrToLong_ull(capacity, NULL, 10,
                                 &ret->disks[i].capacity) < 0) {
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("Malformed disk capacity: '%s'"),
                               NULLSTR(capacity));
                goto error;
            }
        }
    }

 cleanup:
    VIR_FREE(port);
    VIR_FREE(capacity);
    VIR_FREE(disks);
    ctxt->node = save_ctxt;
    return ret;
 error:
    qemuMigrationCookieNBDFree(ret);
    ret = NULL;
    goto cleanup;
}


1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066
static qemuDomainJobInfoPtr
qemuMigrationCookieStatisticsXMLParse(xmlXPathContextPtr ctxt)
{
    qemuDomainJobInfoPtr jobInfo = NULL;
    qemuMonitorMigrationStatus *status;
    xmlNodePtr save_ctxt = ctxt->node;

    if (!(ctxt->node = virXPathNode("./statistics", ctxt)))
        goto cleanup;

    if (VIR_ALLOC(jobInfo) < 0)
        goto cleanup;

    status = &jobInfo->status;
    jobInfo->type = VIR_DOMAIN_JOB_COMPLETED;

1067 1068
    virXPathULongLong("string(./started[1])", ctxt, &jobInfo->started);
    virXPathULongLong("string(./stopped[1])", ctxt, &jobInfo->stopped);
1069 1070 1071
    virXPathULongLong("string(./sent[1])", ctxt, &jobInfo->sent);
    if (virXPathLongLong("string(./delta[1])", ctxt, &jobInfo->timeDelta) == 0)
        jobInfo->timeDeltaSet = true;
1072

1073 1074 1075 1076
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_TIME_ELAPSED "[1])",
                      ctxt, &jobInfo->timeElapsed);
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_TIME_REMAINING "[1])",
                      ctxt, &jobInfo->timeRemaining);
1077

1078 1079 1080
    if (virXPathULongLong("string(./" VIR_DOMAIN_JOB_DOWNTIME "[1])",
                          ctxt, &status->downtime) == 0)
        status->downtime_set = true;
1081 1082 1083
    if (virXPathULongLong("string(./" VIR_DOMAIN_JOB_SETUP_TIME "[1])",
                          ctxt, &status->setup_time) == 0)
        status->setup_time_set = true;
1084 1085 1086 1087 1088 1089 1090

    virXPathULongLong("string(./" VIR_DOMAIN_JOB_MEMORY_TOTAL "[1])",
                      ctxt, &status->ram_total);
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_MEMORY_PROCESSED "[1])",
                      ctxt, &status->ram_transferred);
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_MEMORY_REMAINING "[1])",
                      ctxt, &status->ram_remaining);
1091 1092
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_MEMORY_BPS "[1])",
                      ctxt, &status->ram_bps);
1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107

    if (virXPathULongLong("string(./" VIR_DOMAIN_JOB_MEMORY_CONSTANT "[1])",
                          ctxt, &status->ram_duplicate) == 0)
        status->ram_duplicate_set = true;
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_MEMORY_NORMAL "[1])",
                      ctxt, &status->ram_normal);
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_MEMORY_NORMAL_BYTES "[1])",
                      ctxt, &status->ram_normal_bytes);

    virXPathULongLong("string(./" VIR_DOMAIN_JOB_DISK_TOTAL "[1])",
                      ctxt, &status->disk_total);
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_DISK_PROCESSED "[1])",
                      ctxt, &status->disk_transferred);
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_DISK_REMAINING "[1])",
                      ctxt, &status->disk_remaining);
1108 1109
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_DISK_BPS "[1])",
                      ctxt, &status->disk_bps);
1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128

    if (virXPathULongLong("string(./" VIR_DOMAIN_JOB_COMPRESSION_CACHE "[1])",
                          ctxt, &status->xbzrle_cache_size) == 0)
        status->xbzrle_set = true;
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_COMPRESSION_BYTES "[1])",
                      ctxt, &status->xbzrle_bytes);
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_COMPRESSION_PAGES "[1])",
                      ctxt, &status->xbzrle_pages);
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_COMPRESSION_CACHE_MISSES "[1])",
                      ctxt, &status->xbzrle_cache_miss);
    virXPathULongLong("string(./" VIR_DOMAIN_JOB_COMPRESSION_OVERFLOW "[1])",
                      ctxt, &status->xbzrle_overflow);

 cleanup:
    ctxt->node = save_ctxt;
    return jobInfo;
}


1129 1130
static int
qemuMigrationCookieXMLParse(qemuMigrationCookiePtr mig,
1131
                            virQEMUDriverPtr driver,
1132
                            xmlDocPtr doc,
1133
                            xmlXPathContextPtr ctxt,
E
Eric Blake 已提交
1134
                            unsigned int flags)
1135 1136
{
    char uuidstr[VIR_UUID_STRING_BUFLEN];
1137
    char *tmp = NULL;
1138
    xmlNodePtr *nodes = NULL;
1139 1140
    size_t i;
    int n;
1141 1142 1143 1144
    virCapsPtr caps = NULL;

    if (!(caps = virQEMUDriverGetCapabilities(driver, false)))
        goto error;
1145 1146 1147 1148 1149 1150 1151 1152

    /* We don't store the uuid, name, hostname, or hostuuid
     * values. We just compare them to local data to do some
     * sanity checking on migration operation
     */

    /* Extract domain name */
    if (!(tmp = virXPathString("string(./name[1])", ctxt))) {
1153 1154
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("missing name element in migration data"));
1155 1156 1157
        goto error;
    }
    if (STRNEQ(tmp, mig->name)) {
1158 1159 1160
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Incoming cookie data had unexpected name %s vs %s"),
                       tmp, mig->name);
1161 1162 1163 1164 1165 1166 1167
        goto error;
    }
    VIR_FREE(tmp);

    /* Extract domain uuid */
    tmp = virXPathString("string(./uuid[1])", ctxt);
    if (!tmp) {
1168 1169
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("missing uuid element in migration data"));
1170 1171 1172 1173
        goto error;
    }
    virUUIDFormat(mig->uuid, uuidstr);
    if (STRNEQ(tmp, uuidstr)) {
1174 1175 1176
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Incoming cookie data had unexpected UUID %s vs %s"),
                       tmp, uuidstr);
1177
        goto error;
1178 1179 1180 1181
    }
    VIR_FREE(tmp);

    /* Check & forbid "localhost" migration */
1182
    if (!(mig->remoteHostname = virXPathString("string(./hostname[1])", ctxt))) {
1183 1184
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("missing hostname element in migration data"));
1185 1186
        goto error;
    }
1187
    if (STREQ(mig->remoteHostname, mig->localHostname)) {
1188 1189 1190
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Attempt to migrate guest to the same host %s"),
                       mig->remoteHostname);
1191 1192 1193 1194
        goto error;
    }

    if (!(tmp = virXPathString("string(./hostuuid[1])", ctxt))) {
1195 1196 1197
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("missing hostuuid element in migration data"));
        goto error;
1198
    }
1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210
    if (virUUIDParse(tmp, mig->remoteHostuuid) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("malformed hostuuid element in migration data"));
        goto error;
    }
    if (memcmp(mig->remoteHostuuid, mig->localHostuuid, VIR_UUID_BUFLEN) == 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Attempt to migrate guest to the same host %s"),
                       tmp);
        goto error;
    }
    VIR_FREE(tmp);
1211

1212 1213
    /* Check to ensure all mandatory features from XML are also
     * present in 'flags' */
1214
    if ((n = virXPathNodeSet("./feature", ctxt, &nodes)) < 0)
1215 1216
        goto error;

1217
    for (i = 0; i < n; i++) {
1218 1219 1220
        int val;
        char *str = virXMLPropString(nodes[i], "name");
        if (!str) {
1221 1222
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           "%s", _("missing feature name"));
1223 1224 1225 1226
            goto error;
        }

        if ((val = qemuMigrationCookieFlagTypeFromString(str)) < 0) {
1227 1228 1229
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Unknown migration cookie feature %s"),
                           str);
1230 1231 1232 1233 1234
            VIR_FREE(str);
            goto error;
        }

        if ((flags & (1 << val)) == 0) {
1235 1236 1237
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Unsupported migration cookie feature %s"),
                           str);
1238 1239 1240 1241 1242 1243
            VIR_FREE(str);
        }
        VIR_FREE(str);
    }
    VIR_FREE(nodes);

1244 1245 1246 1247 1248
    if ((flags & QEMU_MIGRATION_COOKIE_GRAPHICS) &&
        virXPathBoolean("count(./graphics) > 0", ctxt) &&
        (!(mig->graphics = qemuMigrationCookieGraphicsXMLParse(ctxt))))
        goto error;

1249 1250 1251 1252
    if ((flags & QEMU_MIGRATION_COOKIE_LOCKSTATE) &&
        virXPathBoolean("count(./lockstate) > 0", ctxt)) {
        mig->lockDriver = virXPathString("string(./lockstate[1]/@driver)", ctxt);
        if (!mig->lockDriver) {
1253 1254
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("Missing lock driver name in migration cookie"));
1255 1256 1257 1258 1259 1260 1261
            goto error;
        }
        mig->lockState = virXPathString("string(./lockstate[1]/leases[1])", ctxt);
        if (mig->lockState && STREQ(mig->lockState, ""))
            VIR_FREE(mig->lockState);
    }

1262 1263 1264
    if ((flags & QEMU_MIGRATION_COOKIE_PERSISTENT) &&
        virXPathBoolean("count(./domain) > 0", ctxt)) {
        if ((n = virXPathNodeSet("./domain", ctxt, &nodes)) > 1) {
1265 1266 1267 1268
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Too many domain elements in "
                             "migration cookie: %d"),
                           n);
1269 1270
            goto error;
        }
1271 1272
        mig->persistent = virDomainDefParseNode(doc, nodes[0],
                                                caps, driver->xmlopt,
1273
                                                VIR_DOMAIN_DEF_PARSE_INACTIVE);
1274 1275 1276 1277 1278 1279 1280 1281
        if (!mig->persistent) {
            /* virDomainDefParseNode already reported
             * an error for us */
            goto error;
        }
        VIR_FREE(nodes);
    }

1282 1283 1284 1285 1286
    if ((flags & QEMU_MIGRATION_COOKIE_NETWORK) &&
        virXPathBoolean("count(./network) > 0", ctxt) &&
        (!(mig->network = qemuMigrationCookieNetworkXMLParse(ctxt))))
        goto error;

1287
    if (flags & QEMU_MIGRATION_COOKIE_NBD &&
1288 1289 1290
        virXPathBoolean("boolean(./nbd)", ctxt) &&
        (!(mig->nbd = qemuMigrationCookieNBDXMLParse(ctxt))))
        goto error;
1291

1292 1293 1294 1295 1296
    if (flags & QEMU_MIGRATION_COOKIE_STATS &&
        virXPathBoolean("boolean(./statistics)", ctxt) &&
        (!(mig->jobInfo = qemuMigrationCookieStatisticsXMLParse(ctxt))))
        goto error;

1297
    virObjectUnref(caps);
1298 1299
    return 0;

1300
 error:
1301
    VIR_FREE(tmp);
1302
    VIR_FREE(nodes);
1303
    virObjectUnref(caps);
1304 1305 1306 1307 1308 1309
    return -1;
}


static int
qemuMigrationCookieXMLParseStr(qemuMigrationCookiePtr mig,
1310
                               virQEMUDriverPtr driver,
1311
                               const char *xml,
E
Eric Blake 已提交
1312
                               unsigned int flags)
1313 1314 1315
{
    xmlDocPtr doc = NULL;
    xmlXPathContextPtr ctxt = NULL;
1316
    int ret = -1;
1317 1318 1319

    VIR_DEBUG("xml=%s", NULLSTR(xml));

1320
    if (!(doc = virXMLParseStringCtxt(xml, _("(qemu_migration_cookie)"), &ctxt)))
1321 1322
        goto cleanup;

1323
    ret = qemuMigrationCookieXMLParse(mig, driver, doc, ctxt, flags);
1324

1325
 cleanup:
1326 1327 1328 1329 1330 1331 1332 1333 1334
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(doc);

    return ret;
}


static int
qemuMigrationBakeCookie(qemuMigrationCookiePtr mig,
1335
                        virQEMUDriverPtr driver,
1336
                        virDomainObjPtr dom,
1337 1338
                        char **cookieout,
                        int *cookieoutlen,
E
Eric Blake 已提交
1339
                        unsigned int flags)
1340
{
1341 1342
    if (!cookieout || !cookieoutlen)
        return 0;
1343 1344 1345

    *cookieoutlen = 0;

1346 1347 1348 1349
    if (flags & QEMU_MIGRATION_COOKIE_GRAPHICS &&
        qemuMigrationCookieAddGraphics(mig, driver, dom) < 0)
        return -1;

1350 1351 1352 1353
    if (flags & QEMU_MIGRATION_COOKIE_LOCKSTATE &&
        qemuMigrationCookieAddLockstate(mig, driver, dom) < 0)
        return -1;

1354 1355 1356 1357
    if (flags & QEMU_MIGRATION_COOKIE_PERSISTENT &&
        qemuMigrationCookieAddPersistent(mig, dom) < 0)
        return -1;

1358 1359 1360 1361 1362
    if (flags & QEMU_MIGRATION_COOKIE_NETWORK &&
        qemuMigrationCookieAddNetwork(mig, driver, dom) < 0) {
        return -1;
    }

1363 1364 1365 1366
    if ((flags & QEMU_MIGRATION_COOKIE_NBD) &&
        qemuMigrationCookieAddNBD(mig, driver, dom) < 0)
        return -1;

1367 1368 1369 1370
    if (flags & QEMU_MIGRATION_COOKIE_STATS &&
        qemuMigrationCookieAddStatistics(mig, dom) < 0)
        return -1;

1371 1372 1373
    if (flags & QEMU_MIGRATION_COOKIE_MEMORY_HOTPLUG)
        mig->flagsMandatory |= QEMU_MIGRATION_COOKIE_MEMORY_HOTPLUG;

1374
    if (!(*cookieout = qemuMigrationCookieXMLFormatStr(driver, mig)))
1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385
        return -1;

    *cookieoutlen = strlen(*cookieout) + 1;

    VIR_DEBUG("cookielen=%d cookie=%s", *cookieoutlen, *cookieout);

    return 0;
}


static qemuMigrationCookiePtr
1386
qemuMigrationEatCookie(virQEMUDriverPtr driver,
1387
                       virDomainObjPtr dom,
1388 1389
                       const char *cookiein,
                       int cookieinlen,
E
Eric Blake 已提交
1390
                       unsigned int flags)
1391 1392 1393 1394 1395 1396
{
    qemuMigrationCookiePtr mig = NULL;

    /* Parse & validate incoming cookie (if any) */
    if (cookiein && cookieinlen &&
        cookiein[cookieinlen-1] != '\0') {
1397 1398
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Migration cookie was not NULL terminated"));
1399 1400 1401 1402 1403 1404 1405 1406 1407 1408
        goto error;
    }

    VIR_DEBUG("cookielen=%d cookie='%s'", cookieinlen, NULLSTR(cookiein));

    if (!(mig = qemuMigrationCookieNew(dom)))
        return NULL;

    if (cookiein && cookieinlen &&
        qemuMigrationCookieXMLParseStr(mig,
1409
                                       driver,
1410 1411 1412 1413
                                       cookiein,
                                       flags) < 0)
        goto error;

1414 1415 1416 1417 1418 1419 1420 1421
    if (flags & QEMU_MIGRATION_COOKIE_PERSISTENT &&
        mig->persistent &&
        STRNEQ(dom->def->name, mig->persistent->name)) {
        VIR_FREE(mig->persistent->name);
        if (VIR_STRDUP(mig->persistent->name, dom->def->name) < 0)
            goto error;
    }

1422 1423 1424
    if (mig->flags & QEMU_MIGRATION_COOKIE_LOCKSTATE) {
        if (!mig->lockDriver) {
            if (virLockManagerPluginUsesState(driver->lockManager)) {
1425 1426 1427
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("Missing %s lock state for migration cookie"),
                               virLockManagerPluginGetName(driver->lockManager));
1428 1429 1430 1431
                goto error;
            }
        } else if (STRNEQ(mig->lockDriver,
                          virLockManagerPluginGetName(driver->lockManager))) {
1432 1433 1434 1435
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Source host lock driver %s different from target %s"),
                           mig->lockDriver,
                           virLockManagerPluginGetName(driver->lockManager));
1436 1437 1438 1439
            goto error;
        }
    }

1440 1441
    return mig;

1442
 error:
1443 1444 1445
    qemuMigrationCookieFree(mig);
    return NULL;
}
1446

1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493
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;
    int state = virDomainObjGetState(vm, NULL);
    bool ret = false;

    VIR_DEBUG("driver=%p, vm=%p, pre-mig-state=%d, state=%d",
              driver, vm, priv->preMigrationState, state);

    if (state == VIR_DOMAIN_PAUSED &&
        priv->preMigrationState == VIR_DOMAIN_RUNNING) {
        /* 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;
}

1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547

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;

1548 1549 1550 1551 1552
    case VIR_STORAGE_TYPE_NETWORK:
        VIR_DEBUG("Skipping creation of network disk '%s'",
                  disk->dst);
        return 0;

1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600
    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;
        break;
    }

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

1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621
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 &&
           virDomainDiskGetSource(disk);
}

1622 1623 1624 1625 1626

static int
qemuMigrationPrecreateStorage(virConnectPtr conn,
                              virQEMUDriverPtr driver ATTRIBUTE_UNUSED,
                              virDomainObjPtr vm,
1627 1628
                              qemuMigrationCookieNBDPtr nbd,
                              size_t nmigrate_disks,
1629 1630
                              const char **migrate_disks,
                              bool incremental)
1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641
{
    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;

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

1645 1646
        if (!(disk = virDomainDiskByName(vm->def, nbd->disks[i].target,
                                         false))) {
1647 1648 1649 1650 1651 1652 1653 1654
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("unable to find disk by target: %s"),
                           nbd->disks[i].target);
            goto cleanup;
        }

        diskSrcPath = virDomainDiskGetSource(disk);

1655 1656
        /* Skip disks we don't want to migrate and already existing disks. */
        if (!qemuMigrateDisk(disk, nmigrate_disks, migrate_disks) ||
1657 1658 1659 1660
            (diskSrcPath && virFileExists(diskSrcPath))) {
            continue;
        }

1661 1662 1663 1664 1665 1666 1667
        if (incremental) {
            virReportError(VIR_ERR_OPERATION_UNSUPPORTED, "%s",
                           _("pre-creation of storage targets for incremental "
                             "storage migration is not supported"));
            goto cleanup;
        }

1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679
        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;
}


1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693
/**
 * 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 已提交
1694
                            virDomainObjPtr vm,
1695 1696 1697
                            const char *listenAddr,
                            size_t nmigrate_disks,
                            const char **migrate_disks)
1698 1699 1700 1701 1702 1703 1704 1705 1706 1707
{
    int ret = -1;
    qemuDomainObjPrivatePtr priv = vm->privateData;
    unsigned short port = 0;
    char *diskAlias = NULL;
    size_t i;

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

1708 1709
        /* check whether disk should be migrated */
        if (!qemuMigrateDisk(disk, nmigrate_disks, migrate_disks))
1710 1711 1712 1713
            continue;

        VIR_FREE(diskAlias);
        if (virAsprintf(&diskAlias, "%s%s",
1714
                        QEMU_DRIVE_HOST_PREFIX, disk->info.alias) < 0)
1715 1716 1717 1718 1719 1720 1721
            goto cleanup;

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

        if (!port &&
1722
            ((virPortAllocatorAcquire(driver->migrationPorts, &port) < 0) ||
1723
             (qemuMonitorNBDServerStart(priv->mon, listenAddr, port) < 0))) {
1724
            goto exit_monitor;
1725 1726
        }

1727 1728 1729
        if (qemuMonitorNBDServerAdd(priv->mon, diskAlias, true) < 0)
            goto exit_monitor;
        if (qemuDomainObjExitMonitor(driver, vm) < 0)
1730 1731 1732 1733 1734 1735
            goto cleanup;
    }

    priv->nbdPort = port;
    ret = 0;

1736
 cleanup:
1737
    VIR_FREE(diskAlias);
1738
    if (ret < 0)
1739
        virPortAllocatorRelease(driver->migrationPorts, port);
1740
    return ret;
1741 1742 1743 1744

 exit_monitor:
    ignore_value(qemuDomainObjExitMonitor(driver, vm));
    goto cleanup;
1745 1746
}

1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773

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


/**
1774
 * qemuMigrationDriveMirrorReady:
1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786
 * @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
1787
qemuMigrationDriveMirrorReady(virQEMUDriverPtr driver,
1788 1789 1790
                              virDomainObjPtr vm)
{
    size_t i;
1791 1792
    size_t notReady = 0;
    int status;
1793 1794 1795

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

1798
        if (!diskPriv->migrating)
1799 1800
            continue;

1801 1802
        status = qemuBlockJobUpdate(driver, vm, disk);
        if (status == VIR_DOMAIN_BLOCK_JOB_FAILED) {
1803 1804 1805 1806 1807
            virReportError(VIR_ERR_OPERATION_FAILED,
                           _("migration of disk %s failed"),
                           disk->dst);
            return -1;
        }
1808 1809 1810

        if (disk->mirrorState != VIR_DOMAIN_DISK_MIRROR_STATE_READY)
            notReady++;
1811 1812
    }

1813 1814 1815 1816 1817 1818 1819
    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;
    }
1820 1821 1822
}


1823 1824 1825 1826
/*
 * 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.
1827
 *
1828 1829 1830 1831
 * 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.
1832 1833
 */
static int
1834
qemuMigrationDriveMirrorCancelled(virQEMUDriverPtr driver,
1835
                                  virDomainObjPtr vm,
1836
                                  bool check)
1837
{
1838 1839 1840 1841
    size_t i;
    size_t active = 0;
    int status;
    bool failed = false;
1842

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

1847 1848
        if (!diskPriv->migrating)
            continue;
1849

1850 1851 1852 1853
        status = qemuBlockJobUpdate(driver, vm, disk);
        switch (status) {
        case VIR_DOMAIN_BLOCK_JOB_FAILED:
            if (check) {
1854
                virReportError(VIR_ERR_OPERATION_FAILED,
1855
                               _("migration of disk %s failed"),
1856
                               disk->dst);
1857
                failed = true;
1858
            }
1859 1860 1861 1862 1863 1864
            /* fallthrough */
        case VIR_DOMAIN_BLOCK_JOB_CANCELED:
        case VIR_DOMAIN_BLOCK_JOB_COMPLETED:
            qemuBlockJobSyncEnd(driver, vm, disk);
            diskPriv->migrating = false;
            break;
1865

1866 1867
        default:
            active++;
1868
        }
1869
    }
1870

1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886
    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;
1887
        }
1888
    }
1889
}
1890 1891


1892 1893 1894 1895 1896 1897 1898 1899 1900
/*
 * 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,
1901 1902
                                  bool failNoJob,
                                  qemuDomainAsyncJob asyncJob)
1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    char *diskAlias = NULL;
    int ret = -1;
    int status;
    int rv;

    status = qemuBlockJobUpdate(driver, vm, disk);
    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;
    }

    if (virAsprintf(&diskAlias, "%s%s",
                    QEMU_DRIVE_HOST_PREFIX, disk->info.alias) < 0)
        return -1;

1930
    if (qemuDomainObjEnterMonitorAsync(driver, vm, asyncJob) < 0)
1931 1932 1933 1934 1935 1936 1937 1938
        goto cleanup;

    rv = qemuMonitorBlockJobCancel(priv->mon, diskAlias, true);

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

    ret = 0;
1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949

 cleanup:
    VIR_FREE(diskAlias);
    return ret;
}


/**
 * qemuMigrationCancelDriveMirror:
 * @driver: qemu driver
 * @vm: domain
1950
 * @check: if true report an error when some of the mirrors fails
1951 1952 1953 1954 1955 1956 1957 1958 1959
 *
 * 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,
1960
                               virDomainObjPtr vm,
1961
                               bool check,
1962 1963
                               qemuDomainAsyncJob asyncJob,
                               virConnectPtr dconn)
1964
{
1965
    virErrorPtr err = NULL;
1966
    int ret = -1;
1967
    size_t i;
1968 1969 1970 1971
    int rv;
    bool failed = false;

    VIR_DEBUG("Cancelling drive mirrors for domain %s", vm->def->name);
1972 1973 1974

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

1977
        if (!diskPriv->migrating)
1978 1979
            continue;

1980 1981
        rv = qemuMigrationCancelOneDriveMirror(driver, vm, disk,
                                               check, asyncJob);
1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993
        if (rv != 0) {
            if (rv < 0) {
                if (!err)
                    err = virSaveLastError();
                failed = true;
            }
            qemuBlockJobSyncEnd(driver, vm, disk);
            diskPriv->migrating = false;
        }
    }

    while ((rv = qemuMigrationDriveMirrorCancelled(driver, vm, check)) != 1) {
1994 1995 1996 1997 1998 1999 2000
        if (check && !failed &&
            dconn && virConnectIsAlive(dconn) <= 0) {
            virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                           _("Lost connection to destination host"));
            failed = true;
        }

2001 2002 2003 2004
        if (rv < 0) {
            failed = true;
            if (rv == -2)
                break;
2005
        }
2006

2007 2008 2009 2010 2011
        if (failed && !err)
            err = virSaveLastError();

        if (virDomainObjWait(vm) < 0)
            goto cleanup;
2012 2013
    }

2014 2015 2016
    ret = failed ? -1 : 0;

 cleanup:
2017 2018 2019 2020 2021
    if (err) {
        virSetError(err);
        virFreeError(err);
    }
    return ret;
2022 2023 2024
}


2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035
/**
 * qemuMigrationDriveMirror:
 * @driver: qemu driver
 * @vm: domain
 * @mig: migration cookie
 * @host: where are we migrating to
 * @speed: how much should the copying be limited
 * @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
2036 2037 2038 2039 2040
 * 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.
2041 2042 2043 2044 2045 2046 2047 2048 2049 2050
 *
 * Returns 0 on success (@migrate_flags updated),
 *        -1 otherwise.
 */
static int
qemuMigrationDriveMirror(virQEMUDriverPtr driver,
                         virDomainObjPtr vm,
                         qemuMigrationCookiePtr mig,
                         const char *host,
                         unsigned long speed,
2051 2052
                         unsigned int *migrate_flags,
                         size_t nmigrate_disks,
2053 2054
                         const char **migrate_disks,
                         virConnectPtr dconn)
2055 2056 2057 2058
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret = -1;
    int port;
2059
    size_t i;
2060 2061
    char *diskAlias = NULL;
    char *nbd_dest = NULL;
2062
    char *hoststr = NULL;
2063
    unsigned int mirror_flags = VIR_DOMAIN_BLOCK_REBASE_REUSE_EXT;
2064
    int rv;
2065
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
2066 2067

    VIR_DEBUG("Starting drive mirrors for domain %s", vm->def->name);
2068 2069 2070 2071 2072

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

2073 2074
    /* escape literal IPv6 address */
    if (strchr(host, ':')) {
2075
        if (virAsprintf(&hoststr, "[%s]", host) < 0)
2076
            goto cleanup;
2077
    } else if (VIR_STRDUP(hoststr, host) < 0) {
2078
        goto cleanup;
2079 2080
    }

2081 2082 2083 2084 2085
    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];
2086
        qemuDomainDiskPrivatePtr diskPriv = QEMU_DOMAIN_DISK_PRIVATE(disk);
2087
        int mon_ret;
2088

2089 2090
        /* check whether disk should be migrated */
        if (!qemuMigrateDisk(disk, nmigrate_disks, migrate_disks))
2091 2092 2093 2094 2095
            continue;

        if ((virAsprintf(&diskAlias, "%s%s",
                         QEMU_DRIVE_HOST_PREFIX, disk->info.alias) < 0) ||
            (virAsprintf(&nbd_dest, "nbd:%s:%d:exportname=%s",
2096
                         hoststr, port, diskAlias) < 0))
2097 2098
            goto cleanup;

2099
        if (qemuDomainObjEnterMonitorAsync(driver, vm,
2100
                                           QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
2101 2102
            goto cleanup;

2103
        qemuBlockJobSyncBegin(disk);
2104
        /* Force "raw" format for NBD export */
2105
        mon_ret = qemuMonitorDriveMirror(priv->mon, diskAlias, nbd_dest,
2106
                                         "raw", speed, 0, 0, mirror_flags);
2107 2108
        VIR_FREE(diskAlias);
        VIR_FREE(nbd_dest);
2109

2110
        if (qemuDomainObjExitMonitor(driver, vm) < 0 || mon_ret < 0) {
2111
            qemuBlockJobSyncEnd(driver, vm, disk);
2112 2113
            goto cleanup;
        }
2114
        diskPriv->migrating = true;
2115 2116 2117 2118 2119

        if (virDomainSaveStatus(driver->xmlopt, cfg->stateDir, vm) < 0) {
            VIR_WARN("Failed to save status on vm %s", vm->def->name);
            goto cleanup;
        }
2120
    }
2121

2122 2123 2124
    while ((rv = qemuMigrationDriveMirrorReady(driver, vm)) != 1) {
        if (rv < 0)
            goto cleanup;
2125

2126
        if (priv->job.abortJob) {
2127 2128 2129 2130 2131
            priv->job.current->type = VIR_DOMAIN_JOB_CANCELLED;
            virReportError(VIR_ERR_OPERATION_ABORTED, _("%s: %s"),
                           qemuDomainAsyncJobTypeToString(priv->job.asyncJob),
                           _("canceled by client"));
            goto cleanup;
2132
        }
2133

2134 2135 2136 2137 2138 2139
        if (dconn && virConnectIsAlive(dconn) <= 0) {
            virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                           _("Lost connection to destination host"));
            goto cleanup;
        }

2140
        if (virDomainObjWait(vm) < 0)
2141
            goto cleanup;
2142 2143
    }

2144
    /* Okay, all disks are ready. Modify migrate_flags */
2145 2146 2147 2148
    *migrate_flags &= ~(QEMU_MONITOR_MIGRATE_NON_SHARED_DISK |
                        QEMU_MONITOR_MIGRATE_NON_SHARED_INC);
    ret = 0;

2149
 cleanup:
2150
    virObjectUnref(cfg);
2151 2152
    VIR_FREE(diskAlias);
    VIR_FREE(nbd_dest);
2153
    VIR_FREE(hoststr);
2154 2155
    return ret;
}
2156

2157

2158 2159 2160 2161 2162 2163 2164 2165 2166
/* Validate whether the domain is safe to migrate.  If vm is NULL,
 * then this is being run in the v2 Prepare stage on the destination
 * (where we only have the target xml); if vm is provided, then this
 * is being run in either v2 Perform or v3 Begin (where we also have
 * access to all of the domain's metadata, such as whether it is
 * marked autodestroy or has snapshots).  While it would be nice to
 * assume that checking on source is sufficient to prevent ever
 * talking to the destination in the first place, we are stuck with
 * the fact that older servers did not do checks on the source. */
2167
bool
2168
qemuMigrationIsAllowed(virQEMUDriverPtr driver, virDomainObjPtr vm,
2169
                       virDomainDefPtr def, bool remote, bool abort_on_error)
2170
{
2171
    int nsnapshots;
2172
    int pauseReason;
2173
    size_t i;
2174 2175 2176

    if (vm) {
        if (qemuProcessAutoDestroyActive(driver, vm)) {
2177 2178
            virReportError(VIR_ERR_OPERATION_INVALID,
                           "%s", _("domain is marked for auto destroy"));
2179 2180
            return false;
        }
2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193

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

            if (nsnapshots > 0) {
                virReportError(VIR_ERR_OPERATION_INVALID,
                               _("cannot migrate domain with %d snapshots"),
                               nsnapshots);
                return false;
            }
2194 2195

            /* cancel migration if disk I/O error is emitted while migrating */
2196 2197
            if (abort_on_error &&
                virDomainObjGetState(vm, &pauseReason) == VIR_DOMAIN_PAUSED &&
2198 2199 2200 2201 2202 2203
                pauseReason == VIR_DOMAIN_PAUSED_IOERROR) {
                virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                               _("cannot migrate domain with I/O error"));
                return false;
            }

2204
        }
2205

2206
        if (qemuDomainHasBlockjob(vm, false)) {
E
Eric Blake 已提交
2207
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
2208
                           _("domain has an active block job"));
E
Eric Blake 已提交
2209 2210
            return false;
        }
2211 2212 2213

        def = vm->def;
    }
2214 2215

    /* Migration with USB host devices is allowed, all other devices are
2216
     * forbidden. */
2217 2218 2219 2220
    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) {
2221 2222 2223
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                           _("domain has assigned non-USB host devices"));
            return false;
2224 2225
        }
    }
2226

2227 2228 2229
    if (def->cpu && def->cpu->mode != VIR_CPU_MODE_HOST_PASSTHROUGH) {
        for (i = 0; i < def->cpu->nfeatures; i++) {
            virCPUFeatureDefPtr feature = &def->cpu->features[i];
J
Ján Tomko 已提交
2230

2231 2232
            if (feature->policy != VIR_CPU_FEATURE_REQUIRE)
                continue;
J
Ján Tomko 已提交
2233

2234 2235 2236 2237 2238 2239 2240
            /* QEMU blocks migration and save with invariant TSC enabled */
            if (STREQ(feature->name, "invtsc")) {
                virReportError(VIR_ERR_OPERATION_INVALID,
                               _("domain has CPU feature: %s"),
                               feature->name);
                return false;
            }
J
Ján Tomko 已提交
2241 2242 2243
        }
    }

2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257
    /* Verify that memory device config can be transferred reliably */
    for (i = 0; i < def->nmems; i++) {
        virDomainMemoryDefPtr mem = def->mems[i];

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

            return false;
        }
    }

2258 2259 2260
    return true;
}

2261
static bool
2262 2263 2264
qemuMigrationIsSafe(virDomainDefPtr def,
                    size_t nmigrate_disks,
                    const char **migrate_disks)
2265
{
2266
    size_t i;
2267

2268
    for (i = 0; i < def->ndisks; i++) {
2269
        virDomainDiskDefPtr disk = def->disks[i];
2270
        const char *src = virDomainDiskGetSource(disk);
2271

2272 2273
        /* Our code elsewhere guarantees shared disks are either readonly (in
         * which case cache mode doesn't matter) or used with cache=none */
2274
        if (qemuMigrateDisk(disk, nmigrate_disks, migrate_disks) &&
2275
            disk->cachemode != VIR_DOMAIN_DISK_CACHE_DISABLE) {
2276
            int rc;
2277

E
Eric Blake 已提交
2278
            if (virDomainDiskGetType(disk) == VIR_STORAGE_TYPE_FILE) {
2279
                if ((rc = virFileIsSharedFS(src)) < 0)
2280 2281
                    return false;
                else if (rc == 0)
2282
                    continue;
2283
                if ((rc = virStorageFileIsClusterFS(src)) < 0)
2284
                    return false;
2285 2286
                else if (rc == 1)
                    continue;
2287 2288
            } else if (disk->src->type == VIR_STORAGE_TYPE_NETWORK &&
                       disk->src->protocol == VIR_STORAGE_NET_PROTOCOL_RBD) {
2289
                continue;
2290
            }
2291

2292 2293 2294
            virReportError(VIR_ERR_MIGRATE_UNSAFE, "%s",
                           _("Migration may lead to data corruption if disks"
                             " use cache != none"));
2295 2296 2297 2298 2299 2300 2301
            return false;
        }
    }

    return true;
}

2302 2303 2304 2305
/** qemuMigrationSetOffline
 * Pause domain for non-live migration.
 */
int
2306
qemuMigrationSetOffline(virQEMUDriverPtr driver,
2307 2308 2309
                        virDomainObjPtr vm)
{
    int ret;
2310
    VIR_DEBUG("driver=%p vm=%p", driver, vm);
2311 2312
    ret = qemuProcessStopCPUs(driver, vm, VIR_DOMAIN_PAUSED_MIGRATION,
                              QEMU_ASYNC_JOB_MIGRATION_OUT);
2313
    if (ret == 0) {
2314
        virObjectEventPtr event;
2315

2316
        event = virDomainEventLifecycleNewFromObj(vm,
2317 2318
                                         VIR_DOMAIN_EVENT_SUSPENDED,
                                         VIR_DOMAIN_EVENT_SUSPENDED_MIGRATED);
2319
        qemuDomainEventQueue(driver, event);
2320 2321 2322 2323 2324 2325
    }

    return ret;
}


2326 2327 2328
static int
qemuMigrationSetCompression(virQEMUDriverPtr driver,
                            virDomainObjPtr vm,
2329
                            bool state,
2330
                            qemuDomainAsyncJob job)
2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret;

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

    ret = qemuMonitorGetMigrationCapability(
                priv->mon,
                QEMU_MONITOR_MIGRATION_CAPS_XBZRLE);

    if (ret < 0) {
        goto cleanup;
2344 2345 2346
    } else if (ret == 0 && !state) {
        /* Unsupported but we want it off anyway */
        goto cleanup;
2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362
    } else if (ret == 0) {
        if (job == QEMU_ASYNC_JOB_MIGRATION_IN) {
            virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                           _("Compressed migration is not supported by "
                             "target QEMU binary"));
        } else {
            virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                           _("Compressed migration is not supported by "
                             "source QEMU binary"));
        }
        ret = -1;
        goto cleanup;
    }

    ret = qemuMonitorSetMigrationCapability(
                priv->mon,
2363 2364
                QEMU_MONITOR_MIGRATION_CAPS_XBZRLE,
                state);
2365

2366
 cleanup:
2367 2368
    if (qemuDomainObjExitMonitor(driver, vm) < 0)
        ret = -1;
2369 2370 2371
    return ret;
}

2372 2373 2374
static int
qemuMigrationSetAutoConverge(virQEMUDriverPtr driver,
                             virDomainObjPtr vm,
2375
                             bool state,
2376
                             qemuDomainAsyncJob job)
2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret;

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

    ret = qemuMonitorGetMigrationCapability(
                priv->mon,
                QEMU_MONITOR_MIGRATION_CAPS_AUTO_CONVERGE);

    if (ret < 0) {
        goto cleanup;
2390 2391 2392
    } else if (ret == 0 && !state) {
        /* Unsupported but we want it off anyway */
        goto cleanup;
2393 2394 2395 2396 2397 2398 2399 2400 2401 2402
    } else if (ret == 0) {
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("Auto-Converge is not supported by "
                         "QEMU binary"));
        ret = -1;
        goto cleanup;
    }

    ret = qemuMonitorSetMigrationCapability(
                priv->mon,
2403 2404
                QEMU_MONITOR_MIGRATION_CAPS_AUTO_CONVERGE,
                state);
2405

2406
 cleanup:
2407 2408
    if (qemuDomainObjExitMonitor(driver, vm) < 0)
        ret = -1;
2409 2410 2411 2412
    return ret;
}


2413 2414 2415
static int
qemuMigrationSetPinAll(virQEMUDriverPtr driver,
                       virDomainObjPtr vm,
2416
                       bool state,
2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430
                       qemuDomainAsyncJob job)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret;

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

    ret = qemuMonitorGetMigrationCapability(
                priv->mon,
                QEMU_MONITOR_MIGRATION_CAPS_RDMA_PIN_ALL);

    if (ret < 0) {
        goto cleanup;
2431 2432 2433
    } else if (ret == 0 && !state) {
        /* Unsupported but we want it off anyway */
        goto cleanup;
2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449
    } else if (ret == 0) {
        if (job == QEMU_ASYNC_JOB_MIGRATION_IN) {
            virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                           _("rdma pinning migration is not supported by "
                             "target QEMU binary"));
        } else {
            virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                           _("rdma pinning migration is not supported by "
                             "source QEMU binary"));
        }
        ret = -1;
        goto cleanup;
    }

    ret = qemuMonitorSetMigrationCapability(
                priv->mon,
2450 2451
                QEMU_MONITOR_MIGRATION_CAPS_RDMA_PIN_ALL,
                state);
2452 2453

 cleanup:
2454 2455
    if (qemuDomainObjExitMonitor(driver, vm) < 0)
        ret = -1;
2456 2457 2458
    return ret;
}

2459
static int
2460
qemuMigrationWaitForSpice(virDomainObjPtr vm)
2461 2462 2463
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    bool wait_for_spice = false;
2464
    size_t i = 0;
2465

2466 2467 2468 2469 2470 2471 2472
    if (!virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_SEAMLESS_MIGRATION))
        return 0;

    for (i = 0; i < vm->def->ngraphics; i++) {
        if (vm->def->graphics[i]->type == VIR_DOMAIN_GRAPHICS_TYPE_SPICE) {
            wait_for_spice = true;
            break;
2473 2474
        }
    }
2475 2476 2477 2478

    if (!wait_for_spice)
        return 0;

2479 2480
    while (!priv->job.spiceMigrated && !priv->job.abortJob) {
        if (virDomainObjWait(vm) < 0)
2481 2482 2483 2484
            return -1;
    }
    return 0;
}
2485

2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537

static void
qemuMigrationUpdateJobType(qemuDomainJobInfoPtr jobInfo)
{
    switch (jobInfo->status.status) {
    case QEMU_MONITOR_MIGRATION_STATUS_COMPLETED:
        jobInfo->type = VIR_DOMAIN_JOB_COMPLETED;
        break;

    case QEMU_MONITOR_MIGRATION_STATUS_INACTIVE:
        jobInfo->type = VIR_DOMAIN_JOB_NONE;
        break;

    case QEMU_MONITOR_MIGRATION_STATUS_ERROR:
        jobInfo->type = VIR_DOMAIN_JOB_FAILED;
        break;

    case QEMU_MONITOR_MIGRATION_STATUS_CANCELLED:
        jobInfo->type = VIR_DOMAIN_JOB_CANCELLED;
        break;

    case QEMU_MONITOR_MIGRATION_STATUS_SETUP:
    case QEMU_MONITOR_MIGRATION_STATUS_ACTIVE:
    case QEMU_MONITOR_MIGRATION_STATUS_CANCELLING:
        break;
    }
}


int
qemuMigrationFetchJobStatus(virQEMUDriverPtr driver,
                            virDomainObjPtr vm,
                            qemuDomainAsyncJob asyncJob,
                            qemuDomainJobInfoPtr jobInfo)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int rv;

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

    memset(&jobInfo->status, 0, sizeof(jobInfo->status));
    rv = qemuMonitorGetMigrationStatus(priv->mon, &jobInfo->status);

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

    qemuMigrationUpdateJobType(jobInfo);
    return qemuDomainJobInfoUpdateTime(jobInfo);
}


2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555
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");
    }
}


2556
static int
2557
qemuMigrationUpdateJobStatus(virQEMUDriverPtr driver,
2558
                             virDomainObjPtr vm,
2559
                             qemuDomainAsyncJob asyncJob)
2560
{
2561
    qemuDomainObjPrivatePtr priv = vm->privateData;
2562 2563
    qemuDomainJobInfoPtr jobInfo = priv->job.current;
    qemuDomainJobInfo newInfo = *jobInfo;
2564

2565 2566
    if (qemuMigrationFetchJobStatus(driver, vm, asyncJob, &newInfo) < 0)
        return -1;
2567

2568 2569 2570
    *jobInfo = newInfo;
    return 0;
}
2571

2572

2573 2574 2575 2576 2577 2578 2579
static int
qemuMigrationCheckJobStatus(virQEMUDriverPtr driver,
                            virDomainObjPtr vm,
                            qemuDomainAsyncJob asyncJob)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    qemuDomainJobInfoPtr jobInfo = priv->job.current;
2580

2581 2582 2583 2584 2585
    bool events = virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATION_EVENT);

    if (events)
        qemuMigrationUpdateJobType(jobInfo);
    else if (qemuMigrationUpdateJobStatus(driver, vm, asyncJob) < 0)
2586
        return -1;
2587

2588 2589
    switch (jobInfo->type) {
    case VIR_DOMAIN_JOB_NONE:
2590 2591
        virReportError(VIR_ERR_OPERATION_FAILED, _("%s: %s"),
                       qemuMigrationJobName(vm), _("is not active"));
2592
        return -1;
2593

2594
    case VIR_DOMAIN_JOB_FAILED:
2595 2596
        virReportError(VIR_ERR_OPERATION_FAILED, _("%s: %s"),
                       qemuMigrationJobName(vm), _("unexpectedly failed"));
2597
        return -1;
2598

2599
    case VIR_DOMAIN_JOB_CANCELLED:
2600 2601
        virReportError(VIR_ERR_OPERATION_ABORTED, _("%s: %s"),
                       qemuMigrationJobName(vm), _("canceled by client"));
2602 2603
        return -1;

2604 2605 2606 2607 2608 2609 2610
    case VIR_DOMAIN_JOB_COMPLETED:
        /* Fetch statistics of a completed migration */
        if (events &&
            qemuMigrationUpdateJobStatus(driver, vm, asyncJob) < 0)
            return -1;
        break;

2611 2612 2613
    case VIR_DOMAIN_JOB_BOUNDED:
    case VIR_DOMAIN_JOB_UNBOUNDED:
    case VIR_DOMAIN_JOB_LAST:
2614 2615
        break;
    }
2616
    return 0;
2617 2618 2619
}


2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676
/**
 * 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,
                       bool abort_on_error,
                       bool storage)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    qemuDomainJobInfoPtr jobInfo = priv->job.current;
    int pauseReason;

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

    if (storage && qemuMigrationDriveMirrorReady(driver, vm) < 0)
        goto error;

    if (abort_on_error &&
        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;
    }

    if (jobInfo->type == VIR_DOMAIN_JOB_COMPLETED)
        return 1;
    else
        return 0;

 error:
    if (jobInfo->type == VIR_DOMAIN_JOB_UNBOUNDED) {
        /* The migration was aborted by us rather than QEMU itself. */
        jobInfo->type = VIR_DOMAIN_JOB_FAILED;
        return -2;
    } else if (jobInfo->type == VIR_DOMAIN_JOB_COMPLETED) {
        jobInfo->type = VIR_DOMAIN_JOB_FAILED;
        return -1;
    } else {
        return -1;
    }
}


2677 2678 2679
/* Returns 0 on success, -2 when migration needs to be cancelled, or -1 when
 * QEMU reports failed migration.
 */
2680
static int
J
Jiri Denemark 已提交
2681 2682
qemuMigrationWaitForCompletion(virQEMUDriverPtr driver,
                               virDomainObjPtr vm,
2683
                               qemuDomainAsyncJob asyncJob,
J
Jiri Denemark 已提交
2684
                               virConnectPtr dconn,
2685 2686
                               bool abort_on_error,
                               bool storage)
2687
{
2688
    qemuDomainObjPrivatePtr priv = vm->privateData;
J
Jiri Denemark 已提交
2689
    qemuDomainJobInfoPtr jobInfo = priv->job.current;
2690
    bool events = virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATION_EVENT);
2691
    int rv;
2692

J
Jiri Denemark 已提交
2693
    jobInfo->type = VIR_DOMAIN_JOB_UNBOUNDED;
2694 2695 2696 2697
    while ((rv = qemuMigrationCompleted(driver, vm, asyncJob, dconn,
                                        abort_on_error, storage)) != 1) {
        if (rv < 0)
            return rv;
2698

2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711
        if (events) {
            if (virDomainObjWait(vm) < 0) {
                jobInfo->type = VIR_DOMAIN_JOB_FAILED;
                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);
        }
2712 2713
    }

2714 2715 2716 2717 2718
    qemuDomainJobInfoUpdateDowntime(jobInfo);
    VIR_FREE(priv->job.completed);
    if (VIR_ALLOC(priv->job.completed) == 0)
        *priv->job.completed = *jobInfo;

2719
    return 0;
2720 2721 2722
}


2723
static int
2724
qemuDomainMigrateGraphicsRelocate(virQEMUDriverPtr driver,
2725
                                  virDomainObjPtr vm,
2726 2727
                                  qemuMigrationCookiePtr cookie,
                                  const char *graphicsuri)
2728 2729
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
2730 2731
    int ret = -1;
    const char *listenAddress = NULL;
2732
    virSocketAddr addr;
2733 2734 2735 2736 2737
    virURIPtr uri = NULL;
    int type = -1;
    int port = -1;
    int tlsPort = -1;
    const char *tlsSubject = NULL;
2738

2739
    if (!cookie || (!cookie->graphics && !graphicsuri))
2740 2741
        return 0;

2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760
    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) {
2761
        size_t i;
2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788

        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;
            }
        }
    }
2789 2790 2791 2792

    /* QEMU doesn't support VNC relocation yet, so
     * skip it to avoid generating an error
     */
2793 2794 2795 2796
    if (type != VIR_DOMAIN_GRAPHICS_TYPE_SPICE) {
        ret = 0;
        goto cleanup;
    }
2797

2798 2799 2800 2801
    if (qemuDomainObjEnterMonitorAsync(driver, vm,
                                       QEMU_ASYNC_JOB_MIGRATION_OUT) == 0) {
        ret = qemuMonitorGraphicsRelocate(priv->mon, type, listenAddress,
                                          port, tlsPort, tlsSubject);
2802 2803
        if (qemuDomainObjExitMonitor(driver, vm) < 0)
            ret = -1;
2804
    }
2805

2806
 cleanup:
2807
    virURIFree(uri);
2808 2809 2810 2811
    return ret;
}


2812
static int
2813
qemuDomainMigrateOPDRelocate(virQEMUDriverPtr driver ATTRIBUTE_UNUSED,
2814 2815 2816
                             virDomainObjPtr vm,
                             qemuMigrationCookiePtr cookie)
{
2817 2818
    virDomainNetDefPtr netptr;
    int ret = -1;
2819
    size_t i;
2820 2821 2822 2823 2824 2825 2826 2827

    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:
2828
           break;
2829
        case VIR_NETDEV_VPORT_PROFILE_OPENVSWITCH:
2830 2831
            if (virNetDevOpenvswitchSetMigrateData(cookie->network->net[i].portdata,
                                                   netptr->ifname) != 0) {
J
Jiri Denemark 已提交
2832 2833 2834
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("Unable to run command to set OVS port data for "
                                 "interface %s"), netptr->ifname);
2835 2836 2837
                goto cleanup;
            }
            break;
2838 2839 2840 2841 2842
        default:
            break;
        }
    }

2843
    ret = 0;
2844
 cleanup:
2845 2846 2847 2848
    return ret;
}


2849 2850 2851 2852 2853 2854
/* 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.
 */
2855 2856 2857 2858
static virDomainObjPtr
qemuMigrationCleanup(virDomainObjPtr vm,
                     virConnectPtr conn,
                     void *opaque)
2859
{
2860
    virQEMUDriverPtr driver = opaque;
2861 2862 2863 2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875
    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);

2876
    switch ((qemuMigrationJobPhase) priv->job.phase) {
2877 2878 2879 2880 2881 2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900 2901 2902 2903 2904 2905 2906
    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 */
        ;
    }

2907
 cleanup:
2908 2909 2910
    return vm;
}

2911

2912
/* The caller is supposed to lock the vm and start a migration job. */
2913 2914 2915 2916 2917 2918 2919
static char *
qemuMigrationBeginPhase(virQEMUDriverPtr driver,
                        virDomainObjPtr vm,
                        const char *xmlin,
                        const char *dname,
                        char **cookieout,
                        int *cookieoutlen,
2920 2921
                        size_t nmigrate_disks,
                        const char **migrate_disks,
2922
                        unsigned long flags)
2923 2924 2925
{
    char *rv = NULL;
    qemuMigrationCookiePtr mig = NULL;
2926
    virDomainDefPtr def = NULL;
2927
    qemuDomainObjPrivatePtr priv = vm->privateData;
2928
    virCapsPtr caps = NULL;
2929
    unsigned int cookieFlags = QEMU_MIGRATION_COOKIE_LOCKSTATE;
2930
    bool abort_on_error = !!(flags & VIR_MIGRATE_ABORT_ON_ERROR);
2931

2932
    VIR_DEBUG("driver=%p, vm=%p, xmlin=%s, dname=%s,"
2933 2934
              " cookieout=%p, cookieoutlen=%p,"
              " nmigrate_disks=%zu, migrate_disks=%p, flags=%lx",
2935
              driver, vm, NULLSTR(xmlin), NULLSTR(dname),
2936 2937
              cookieout, cookieoutlen, nmigrate_disks,
              migrate_disks, flags);
2938

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

2942 2943 2944 2945 2946 2947
    /* 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);
2948

2949
    if (!qemuMigrationIsAllowed(driver, vm, NULL, true, abort_on_error))
2950 2951
        goto cleanup;

2952 2953
    if (!(flags & VIR_MIGRATE_UNSAFE) &&
        !qemuMigrationIsSafe(vm->def, nmigrate_disks, migrate_disks))
2954 2955
        goto cleanup;

2956 2957 2958 2959 2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974 2975 2976 2977 2978 2979 2980 2981 2982 2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998 2999
    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;
            }
3000 3001 3002
        }
    }

3003
    if (virDomainDefHasMemoryHotplug(vm->def) ||
3004
        ((flags & VIR_MIGRATE_PERSIST_DEST) &&
3005
         vm->newDef && virDomainDefHasMemoryHotplug(vm->newDef)))
3006 3007
        cookieFlags |= QEMU_MIGRATION_COOKIE_MEMORY_HOTPLUG;

3008
    if (!(mig = qemuMigrationEatCookie(driver, vm, NULL, 0, 0)))
3009 3010 3011 3012
        goto cleanup;

    if (qemuMigrationBakeCookie(mig, driver, vm,
                                cookieout, cookieoutlen,
3013
                                cookieFlags) < 0)
3014 3015
        goto cleanup;

L
liguang 已提交
3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031 3032 3033 3034 3035 3036 3037
    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;
        }
    }

3038
    if (xmlin) {
3039
        if (!(def = virDomainDefParseString(xmlin, caps, driver->xmlopt,
3040
                                            VIR_DOMAIN_DEF_PARSE_INACTIVE)))
3041 3042
            goto cleanup;

3043
        if (!qemuDomainDefCheckABIStability(driver, vm->def, def))
3044 3045
            goto cleanup;

3046
        rv = qemuDomainDefFormatLive(driver, def, false, true);
3047
    } else {
3048
        rv = qemuDomainDefFormatLive(driver, vm->def, false, true);
3049
    }
3050

3051
 cleanup:
3052
    qemuMigrationCookieFree(mig);
3053
    virObjectUnref(caps);
3054
    virDomainDefFree(def);
3055 3056 3057
    return rv;
}

3058 3059 3060 3061 3062 3063 3064
char *
qemuMigrationBegin(virConnectPtr conn,
                   virDomainObjPtr vm,
                   const char *xmlin,
                   const char *dname,
                   char **cookieout,
                   int *cookieoutlen,
3065 3066
                   size_t nmigrate_disks,
                   const char **migrate_disks,
3067 3068 3069 3070
                   unsigned long flags)
{
    virQEMUDriverPtr driver = conn->privateData;
    char *xml = NULL;
3071
    qemuDomainAsyncJob asyncJob;
3072 3073 3074 3075 3076 3077 3078 3079 3080 3081 3082

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

3083 3084
    qemuMigrationStoreDomainState(vm);

3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095 3096 3097 3098 3099
    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) &&
        qemuDomainCheckEjectableMedia(driver, vm, asyncJob) < 0)
        goto endjob;

    if (!(xml = qemuMigrationBeginPhase(driver, vm, xmlin, dname,
                                        cookieout, cookieoutlen,
3100
                                        nmigrate_disks, migrate_disks, flags)))
3101 3102 3103 3104 3105 3106 3107
        goto endjob;

    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.
         */
3108
        if (virCloseCallbacksSet(driver->closeCallbacks, vm, conn,
3109 3110
                                 qemuMigrationCleanup) < 0) {
            VIR_FREE(xml);
3111
            goto endjob;
3112
        }
3113
        qemuMigrationJobContinue(vm);
3114 3115 3116 3117
    } else {
        goto endjob;
    }

3118
 cleanup:
M
Michal Privoznik 已提交
3119
    virDomainObjEndAPI(&vm);
3120 3121
    return xml;

3122
 endjob:
3123 3124 3125 3126
    if (flags & VIR_MIGRATE_CHANGE_PROTECTION)
        qemuMigrationJobFinish(driver, vm);
    else
        qemuDomainObjEndJob(driver, vm);
3127 3128 3129
    goto cleanup;
}

3130

3131 3132
/* Prepare is the first step, and it runs on the destination host.
 */
3133

3134
static void
3135
qemuMigrationPrepareCleanup(virQEMUDriverPtr driver,
3136 3137 3138 3139 3140 3141 3142 3143 3144 3145
                            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));

3146 3147 3148
    virPortAllocatorRelease(driver->migrationPorts, priv->migrationPort);
    priv->migrationPort = 0;

3149 3150 3151 3152 3153
    if (!qemuMigrationJobIsActive(vm, QEMU_ASYNC_JOB_MIGRATION_IN))
        return;
    qemuDomainObjDiscardAsyncJob(driver, vm);
}

3154
static int
3155
qemuMigrationPrepareAny(virQEMUDriverPtr driver,
3156 3157 3158 3159 3160
                        virConnectPtr dconn,
                        const char *cookiein,
                        int cookieinlen,
                        char **cookieout,
                        int *cookieoutlen,
3161
                        virDomainDefPtr *def,
3162
                        const char *origname,
L
liguang 已提交
3163
                        virStreamPtr st,
3164
                        const char *protocol,
3165 3166
                        unsigned short port,
                        bool autoPort,
3167
                        const char *listenAddress,
3168 3169
                        size_t nmigrate_disks,
                        const char **migrate_disks,
L
liguang 已提交
3170
                        unsigned long flags)
3171 3172
{
    virDomainObjPtr vm = NULL;
3173
    virObjectEventPtr event = NULL;
3174
    int ret = -1;
3175
    int dataFD[2] = { -1, -1 };
3176
    qemuDomainObjPrivatePtr priv = NULL;
3177
    qemuMigrationCookiePtr mig = NULL;
3178
    bool tunnel = !!st;
J
Jiri Denemark 已提交
3179
    char *xmlout = NULL;
L
liguang 已提交
3180
    unsigned int cookieFlags;
3181
    virCapsPtr caps = NULL;
J
Ján Tomko 已提交
3182
    char *migrateFrom = NULL;
3183
    bool abort_on_error = !!(flags & VIR_MIGRATE_ABORT_ON_ERROR);
3184
    bool taint_hook = false;
3185

3186 3187
    virNWFilterReadLockFilterUpdates();

L
liguang 已提交
3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208 3209
    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;
        }
    }

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

3213
    if (!qemuMigrationIsAllowed(driver, NULL, *def, true, abort_on_error))
3214 3215
        goto cleanup;

J
Jiri Denemark 已提交
3216 3217 3218 3219 3220
    /* Let migration hook filter domain XML */
    if (virHookPresent(VIR_HOOK_DRIVER_QEMU)) {
        char *xml;
        int hookret;

3221
        if (!(xml = qemuDomainDefFormatXML(driver, *def,
3222 3223
                                           VIR_DOMAIN_XML_SECURE |
                                           VIR_DOMAIN_XML_MIGRATABLE)))
J
Jiri Denemark 已提交
3224 3225
            goto cleanup;

3226
        hookret = virHookCall(VIR_HOOK_DRIVER_QEMU, (*def)->name,
J
Jiri Denemark 已提交
3227 3228 3229 3230 3231 3232 3233
                              VIR_HOOK_QEMU_OP_MIGRATE, VIR_HOOK_SUBOP_BEGIN,
                              NULL, xml, &xmlout);
        VIR_FREE(xml);

        if (hookret < 0) {
            goto cleanup;
        } else if (hookret == 0) {
3234
            if (virStringIsEmpty(xmlout)) {
J
Jiri Denemark 已提交
3235 3236 3237 3238 3239 3240
                VIR_DEBUG("Migrate hook filter returned nothing; using the"
                          " original XML");
            } else {
                virDomainDefPtr newdef;

                VIR_DEBUG("Using hook-filtered domain XML: %s", xmlout);
3241
                newdef = virDomainDefParseString(xmlout, caps, driver->xmlopt,
3242
                                                 VIR_DOMAIN_DEF_PARSE_INACTIVE);
J
Jiri Denemark 已提交
3243 3244 3245
                if (!newdef)
                    goto cleanup;

3246
                if (!qemuDomainDefCheckABIStability(driver, *def, newdef)) {
J
Jiri Denemark 已提交
3247 3248 3249 3250
                    virDomainDefFree(newdef);
                    goto cleanup;
                }

3251 3252
                virDomainDefFree(*def);
                *def = newdef;
3253 3254 3255 3256
                /* 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 已提交
3257 3258 3259 3260
            }
        }
    }

J
Ján Tomko 已提交
3261 3262 3263 3264
    if (tunnel) {
        /* QEMU will be started with -incoming stdio
         * (which qemu_command might convert to exec:cat or fd:n)
         */
3265
        if (VIR_STRDUP(migrateFrom, "stdio") < 0)
J
Ján Tomko 已提交
3266 3267
            goto cleanup;
    } else {
3268 3269 3270
        bool encloseAddress = false;
        bool hostIPv6Capable = false;
        bool qemuIPv6Capable = false;
J
Ján Tomko 已提交
3271 3272 3273 3274
        virQEMUCapsPtr qemuCaps = NULL;
        struct addrinfo *info = NULL;
        struct addrinfo hints = { .ai_flags = AI_ADDRCONFIG,
                                  .ai_socktype = SOCK_STREAM };
3275
        const char *incFormat;
J
Ján Tomko 已提交
3276

3277 3278 3279 3280
        if (getaddrinfo("::", NULL, &hints, &info) == 0) {
            freeaddrinfo(info);
            hostIPv6Capable = true;
        }
J
Ján Tomko 已提交
3281
        if (!(qemuCaps = virQEMUCapsCacheLookupCopy(driver->qemuCapsCache,
3282 3283
                                                    (*def)->emulator,
                                                    (*def)->os.machine)))
J
Ján Tomko 已提交
3284 3285
            goto cleanup;

3286 3287 3288 3289
        qemuIPv6Capable = virQEMUCapsGet(qemuCaps, QEMU_CAPS_IPV6_MIGRATION);
        virObjectUnref(qemuCaps);

        if (listenAddress) {
3290 3291 3292 3293 3294 3295 3296 3297 3298
            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"));
3299 3300
                    goto cleanup;
                }
3301 3302
                /* IPv6 address must be escaped in brackets on the cmd line */
                encloseAddress = true;
3303
            } else {
3304
                /* listenAddress is a hostname or IPv4 */
3305
            }
3306
        } else if (qemuIPv6Capable && hostIPv6Capable) {
3307 3308 3309
            /* Listen on :: instead of 0.0.0.0 if QEMU understands it
             * and there is at least one IPv6 address configured
             */
3310 3311 3312 3313
            listenAddress = "::";
            encloseAddress = true;
        } else {
            listenAddress = "0.0.0.0";
J
Ján Tomko 已提交
3314 3315
        }

3316 3317 3318 3319
        /* QEMU will be started with
         *   -incoming protocol:[<IPv6 addr>]:port,
         *   -incoming protocol:<IPv4 addr>:port, or
         *   -incoming protocol:<hostname>:port
J
Ján Tomko 已提交
3320
         */
3321 3322 3323 3324 3325 3326
        if (encloseAddress)
            incFormat = "%s:[%s]:%d";
        else
            incFormat = "%s:%s:%d";
        if (virAsprintf(&migrateFrom, incFormat,
                        protocol, listenAddress, port) < 0)
J
Ján Tomko 已提交
3327 3328 3329
            goto cleanup;
    }

3330
    if (!(vm = virDomainObjListAdd(driver->domains, *def,
3331
                                   driver->xmlopt,
3332 3333 3334
                                   VIR_DOMAIN_OBJ_LIST_ADD_LIVE |
                                   VIR_DOMAIN_OBJ_LIST_ADD_CHECK_LIVE,
                                   NULL)))
3335
        goto cleanup;
3336

3337
    virObjectRef(vm);
3338
    *def = NULL;
3339
    priv = vm->privateData;
3340 3341
    if (VIR_STRDUP(priv->origname, origname) < 0)
        goto cleanup;
3342

3343 3344 3345 3346 3347
    if (taint_hook) {
        /* Domain XML has been altered by a hook script. */
        priv->hookRun = true;
    }

3348
    if (!(mig = qemuMigrationEatCookie(driver, vm, cookiein, cookieinlen,
3349 3350
                                       QEMU_MIGRATION_COOKIE_LOCKSTATE |
                                       QEMU_MIGRATION_COOKIE_NBD)))
3351 3352
        goto cleanup;

3353 3354
    if (STREQ_NULLABLE(protocol, "rdma") &&
        !virMemoryLimitIsSet(vm->def->mem.hard_limit)) {
M
Michael R. Hines 已提交
3355 3356 3357 3358 3359 3360
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                       _("cannot start RDMA migration with no memory hard "
                         "limit set"));
        goto cleanup;
    }

3361
    if (qemuMigrationPrecreateStorage(dconn, driver, vm, mig->nbd,
3362 3363
                                      nmigrate_disks, migrate_disks,
                                      !!(flags & VIR_MIGRATE_NON_SHARED_INC)) < 0)
3364 3365
        goto cleanup;

3366
    if (qemuMigrationJobStart(driver, vm, QEMU_ASYNC_JOB_MIGRATION_IN) < 0)
3367
        goto cleanup;
3368
    qemuMigrationJobSetPhase(driver, vm, QEMU_MIGRATION_PHASE_PREPARE);
3369 3370 3371 3372

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

L
liguang 已提交
3373 3374 3375
    if (flags & VIR_MIGRATE_OFFLINE)
        goto done;

3376 3377
    if (tunnel &&
        (pipe(dataFD) < 0 || virSetCloseExec(dataFD[1]) < 0)) {
3378 3379
        virReportSystemError(errno, "%s",
                             _("cannot create pipe for tunnelled migration"));
3380 3381 3382 3383
        goto endjob;
    }

    /* Start the QEMU daemon, with the same command-line arguments plus
3384
     * -incoming $migrateFrom
3385
     */
3386 3387
    if (qemuProcessStart(dconn, driver, vm, QEMU_ASYNC_JOB_MIGRATION_IN,
                         migrateFrom, dataFD[0], NULL, NULL,
3388 3389
                         VIR_NETDEV_VPORT_PROFILE_OP_MIGRATE_IN_START,
                         VIR_QEMU_PROCESS_START_PAUSED |
3390
                         VIR_QEMU_PROCESS_START_AUTODESTROY) < 0) {
3391
        virDomainAuditStart(vm, "migrated", false);
3392 3393 3394
        goto endjob;
    }

3395 3396 3397 3398
    if (tunnel) {
        if (virFDStreamOpen(st, dataFD[1]) < 0) {
            virReportSystemError(errno, "%s",
                                 _("cannot pass pipe for tunnelled migration"));
3399
            goto stop;
3400
        }
3401
        dataFD[1] = -1; /* 'st' owns the FD now & will close it */
3402 3403
    }

3404 3405
    if (qemuMigrationSetCompression(driver, vm,
                                    flags & VIR_MIGRATE_COMPRESSED,
3406 3407 3408
                                    QEMU_ASYNC_JOB_MIGRATION_IN) < 0)
        goto stop;

3409
    if (STREQ_NULLABLE(protocol, "rdma") &&
M
Michael R. Hines 已提交
3410 3411 3412 3413
        virProcessSetMaxMemLock(vm->pid, vm->def->mem.hard_limit << 10) < 0) {
        goto stop;
    }

3414 3415 3416
    if (qemuMigrationSetPinAll(driver, vm,
                               flags & VIR_MIGRATE_RDMA_PIN_ALL,
                               QEMU_ASYNC_JOB_MIGRATION_IN) < 0)
3417 3418
        goto stop;

3419 3420 3421 3422 3423 3424 3425 3426
    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");
    }
3427

3428
 done:
L
liguang 已提交
3429 3430 3431 3432 3433
    if (flags & VIR_MIGRATE_OFFLINE)
        cookieFlags = 0;
    else
        cookieFlags = QEMU_MIGRATION_COOKIE_GRAPHICS;

3434 3435 3436
    if (mig->nbd &&
        flags & (VIR_MIGRATE_NON_SHARED_DISK | VIR_MIGRATE_NON_SHARED_INC) &&
        virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_NBD_SERVER)) {
3437 3438
        if (qemuMigrationStartNBDServer(driver, vm, listenAddress,
                                        nmigrate_disks, migrate_disks) < 0) {
3439 3440
            /* error already reported */
            goto endjob;
3441
        }
3442
        cookieFlags |= QEMU_MIGRATION_COOKIE_NBD;
3443 3444 3445 3446
    }

    if (qemuMigrationBakeCookie(mig, driver, vm, cookieout,
                                cookieoutlen, cookieFlags) < 0) {
3447 3448 3449 3450 3451 3452 3453
        /* 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");
    }

3454 3455 3456
    if (qemuDomainCleanupAdd(vm, qemuMigrationPrepareCleanup) < 0)
        goto endjob;

L
liguang 已提交
3457 3458
    if (!(flags & VIR_MIGRATE_OFFLINE)) {
        virDomainAuditStart(vm, "migrated", true);
3459
        event = virDomainEventLifecycleNewFromObj(vm,
L
liguang 已提交
3460 3461 3462
                                         VIR_DOMAIN_EVENT_STARTED,
                                         VIR_DOMAIN_EVENT_STARTED_MIGRATED);
    }
3463

3464 3465 3466 3467
    /* 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.
     */
3468
    qemuMigrationJobContinue(vm);
3469

3470 3471
    if (autoPort)
        priv->migrationPort = port;
3472
    ret = 0;
3473

3474
 cleanup:
J
Ján Tomko 已提交
3475
    VIR_FREE(migrateFrom);
J
Jiri Denemark 已提交
3476
    VIR_FREE(xmlout);
3477 3478
    VIR_FORCE_CLOSE(dataFD[0]);
    VIR_FORCE_CLOSE(dataFD[1]);
3479 3480 3481
    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 */
3482
        VIR_FREE(priv->origname);
3483 3484
        virPortAllocatorRelease(driver->migrationPorts, priv->nbdPort);
        priv->nbdPort = 0;
3485 3486
        if (!vm->persistent)
            qemuDomainRemoveInactive(driver, vm);
3487
    }
M
Michal Privoznik 已提交
3488
    virDomainObjEndAPI(&vm);
3489
    qemuDomainEventQueue(driver, event);
3490
    qemuMigrationCookieFree(mig);
3491
    virObjectUnref(caps);
3492
    virNWFilterUnlockFilterUpdates();
3493
    return ret;
3494

3495
 stop:
3496
    virDomainAuditStart(vm, "migrated", false);
3497 3498
    qemuProcessStop(driver, vm, VIR_DOMAIN_SHUTOFF_FAILED,
                    VIR_QEMU_PROCESS_STOP_MIGRATED);
3499

3500
 endjob:
3501
    qemuMigrationJobFinish(driver, vm);
3502
    goto cleanup;
3503 3504 3505
}


3506 3507 3508 3509 3510
/*
 * This version starts an empty VM listening on a localhost TCP port, and
 * sets up the corresponding virStream to handle the incoming data.
 */
int
3511
qemuMigrationPrepareTunnel(virQEMUDriverPtr driver,
3512 3513 3514 3515 3516 3517
                           virConnectPtr dconn,
                           const char *cookiein,
                           int cookieinlen,
                           char **cookieout,
                           int *cookieoutlen,
                           virStreamPtr st,
3518
                           virDomainDefPtr *def,
3519
                           const char *origname,
L
liguang 已提交
3520
                           unsigned long flags)
3521 3522 3523 3524
{
    int ret;

    VIR_DEBUG("driver=%p, dconn=%p, cookiein=%s, cookieinlen=%d, "
3525
              "cookieout=%p, cookieoutlen=%p, st=%p, def=%p, "
3526
              "origname=%s, flags=%lx",
3527
              driver, dconn, NULLSTR(cookiein), cookieinlen,
3528
              cookieout, cookieoutlen, st, *def, origname, flags);
3529

3530 3531 3532 3533 3534 3535
    if (st == NULL) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("tunnelled migration requested but NULL stream passed"));
        return -1;
    }

3536
    ret = qemuMigrationPrepareAny(driver, dconn, cookiein, cookieinlen,
3537
                                  cookieout, cookieoutlen, def, origname,
3538
                                  st, NULL, 0, false, NULL, 0, NULL, flags);
3539 3540 3541 3542
    return ret;
}


3543 3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555 3556 3557 3558 3559 3560 3561 3562 3563 3564 3565
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;
}


3566
int
3567
qemuMigrationPrepareDirect(virQEMUDriverPtr driver,
3568
                           virConnectPtr dconn,
3569 3570 3571 3572
                           const char *cookiein,
                           int cookieinlen,
                           char **cookieout,
                           int *cookieoutlen,
3573 3574
                           const char *uri_in,
                           char **uri_out,
3575
                           virDomainDefPtr *def,
3576
                           const char *origname,
3577
                           const char *listenAddress,
3578 3579
                           size_t nmigrate_disks,
                           const char **migrate_disks,
L
liguang 已提交
3580
                           unsigned long flags)
3581
{
3582 3583
    unsigned short port = 0;
    bool autoPort = true;
3584 3585
    char *hostname = NULL;
    int ret = -1;
3586
    virURIPtr uri = NULL;
3587 3588
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
    const char *migrateHost = cfg->migrateHost;
J
Jiri Denemark 已提交
3589

3590 3591
    VIR_DEBUG("driver=%p, dconn=%p, cookiein=%s, cookieinlen=%d, "
              "cookieout=%p, cookieoutlen=%p, uri_in=%s, uri_out=%p, "
3592 3593
              "def=%p, origname=%s, listenAddress=%s, "
              "nmigrate_disks=%zu, migrate_disks=%p, flags=%lx",
3594 3595
              driver, dconn, NULLSTR(cookiein), cookieinlen,
              cookieout, cookieoutlen, NULLSTR(uri_in), uri_out,
3596 3597
              *def, origname, NULLSTR(listenAddress),
              nmigrate_disks, migrate_disks, flags);
3598

3599 3600
    *uri_out = NULL;

3601 3602 3603
    /* 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
3604 3605 3606
     * 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".
3607 3608 3609 3610 3611 3612
     *
     * 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) {
3613 3614 3615
        bool encloseAddress = false;
        const char *incFormat;

3616
        if (virPortAllocatorAcquire(driver->migrationPorts, &port) < 0)
3617
            goto cleanup;
3618

3619
        if (migrateHost != NULL) {
3620 3621
            if (virSocketAddrNumericFamily(migrateHost) == AF_INET6)
                encloseAddress = true;
3622

3623
            if (VIR_STRDUP(hostname, migrateHost) < 0)
3624 3625 3626 3627 3628
                goto cleanup;
        } else {
            if ((hostname = virGetHostname()) == NULL)
                goto cleanup;
        }
3629 3630

        if (STRPREFIX(hostname, "localhost")) {
3631 3632 3633
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("hostname on destination resolved to localhost,"
                             " but migration requires an FQDN"));
3634 3635 3636 3637 3638
            goto cleanup;
        }

        /* XXX this really should have been a properly well-formed
         * URI, but we can't add in tcp:// now without breaking
3639
         * compatibility with old targets. We at least make the
3640 3641
         * new targets accept both syntaxes though.
         */
3642 3643 3644 3645 3646 3647
        if (encloseAddress)
            incFormat = "%s:[%s]:%d";
        else
            incFormat = "%s:%s:%d";

        if (virAsprintf(uri_out, incFormat, "tcp", hostname, port) < 0)
3648 3649
            goto cleanup;
    } else {
3650
        bool well_formed_uri;
J
Ján Tomko 已提交
3651

3652 3653
        if (!(uri = qemuMigrationParseURI(uri_in, &well_formed_uri)))
            goto cleanup;
J
Ján Tomko 已提交
3654

3655 3656 3657 3658 3659 3660 3661
        if (uri->scheme == NULL) {
            virReportError(VIR_ERR_INVALID_ARG,
                           _("missing scheme in migration URI: %s"),
                           uri_in);
            goto cleanup;
        }

M
Michael R. Hines 已提交
3662 3663
        if (STRNEQ(uri->scheme, "tcp") &&
            STRNEQ(uri->scheme, "rdma")) {
3664 3665 3666
            virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED,
                           _("unsupported scheme %s in migration URI %s"),
                           uri->scheme, uri_in);
J
Ján Tomko 已提交
3667 3668 3669 3670 3671 3672 3673 3674 3675 3676
            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) {
3677
            if (virPortAllocatorAcquire(driver->migrationPorts, &port) < 0)
3678
                goto cleanup;
3679

3680
            /* Send well-formed URI only if uri_in was well-formed */
3681 3682 3683 3684 3685 3686 3687 3688
            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;
            }
3689
        } else {
3690 3691
            port = uri->port;
            autoPort = false;
3692 3693 3694 3695 3696 3697
        }
    }

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

3698
    ret = qemuMigrationPrepareAny(driver, dconn, cookiein, cookieinlen,
3699
                                  cookieout, cookieoutlen, def, origname,
3700
                                  NULL, uri ? uri->scheme : "tcp",
3701 3702
                                  port, autoPort, listenAddress,
                                  nmigrate_disks, migrate_disks, flags);
3703
 cleanup:
3704
    virURIFree(uri);
3705
    VIR_FREE(hostname);
3706
    virObjectUnref(cfg);
3707
    if (ret != 0) {
3708
        VIR_FREE(*uri_out);
3709 3710 3711
        if (autoPort)
            virPortAllocatorRelease(driver->migrationPorts, port);
    }
3712 3713 3714 3715
    return ret;
}


3716 3717 3718
virDomainDefPtr
qemuMigrationPrepareDef(virQEMUDriverPtr driver,
                        const char *dom_xml,
3719 3720
                        const char *dname,
                        char **origname)
3721 3722 3723
{
    virCapsPtr caps = NULL;
    virDomainDefPtr def;
3724
    char *name = NULL;
3725 3726 3727 3728 3729 3730 3731 3732 3733 3734 3735

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

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

    if (!(def = virDomainDefParseString(dom_xml, caps, driver->xmlopt,
3736
                                        VIR_DOMAIN_DEF_PARSE_INACTIVE)))
3737 3738 3739
        goto cleanup;

    if (dname) {
3740
        name = def->name;
3741 3742 3743 3744 3745 3746
        if (VIR_STRDUP(def->name, dname) < 0) {
            virDomainDefFree(def);
            def = NULL;
        }
    }

3747
 cleanup:
3748
    virObjectUnref(caps);
3749 3750 3751 3752
    if (def && origname)
        *origname = name;
    else
        VIR_FREE(name);
3753 3754 3755 3756
    return def;
}


3757 3758 3759 3760 3761 3762 3763 3764 3765 3766
static int
qemuMigrationConfirmPhase(virQEMUDriverPtr driver,
                          virConnectPtr conn,
                          virDomainObjPtr vm,
                          const char *cookiein,
                          int cookieinlen,
                          unsigned int flags,
                          int retcode)
{
    qemuMigrationCookiePtr mig;
3767
    virObjectEventPtr event = NULL;
3768 3769 3770 3771 3772 3773 3774 3775 3776 3777 3778 3779 3780 3781 3782
    int rv = -1;
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);

    VIR_DEBUG("driver=%p, conn=%p, vm=%p, cookiein=%s, cookieinlen=%d, "
              "flags=%x, retcode=%d",
              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);

3783 3784
    if (!(mig = qemuMigrationEatCookie(driver, vm, cookiein, cookieinlen,
                                       QEMU_MIGRATION_COOKIE_STATS)))
3785 3786
        goto cleanup;

3787 3788 3789
    /* Update total times with the values sent by the destination daemon */
    if (mig->jobInfo) {
        qemuDomainObjPrivatePtr priv = vm->privateData;
3790 3791 3792
        VIR_FREE(priv->job.completed);
        priv->job.completed = mig->jobInfo;
        mig->jobInfo = NULL;
3793 3794
    }

3795 3796 3797 3798 3799 3800 3801 3802 3803
    if (flags & VIR_MIGRATE_OFFLINE)
        goto done;

    /* Did the migration go as planned?  If yes, kill off the
     * domain object, but if no, resume CPUs
     */
    if (retcode == 0) {
        /* If guest uses SPICE and supports seamless migration we have to hold
         * up domain shutdown until SPICE server transfers its data */
3804
        qemuMigrationWaitForSpice(vm);
3805 3806 3807 3808 3809

        qemuProcessStop(driver, vm, VIR_DOMAIN_SHUTOFF_MIGRATED,
                        VIR_QEMU_PROCESS_STOP_MIGRATED);
        virDomainAuditStop(vm, "migrated");

3810
        event = virDomainEventLifecycleNewFromObj(vm,
3811 3812 3813
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_MIGRATED);
    } else {
3814
        virErrorPtr orig_err = virSaveLastError();
3815 3816

        /* cancel any outstanding NBD jobs */
3817
        qemuMigrationCancelDriveMirror(driver, vm, false,
3818
                                       QEMU_ASYNC_JOB_MIGRATION_OUT, NULL);
3819 3820 3821

        virSetError(orig_err);
        virFreeError(orig_err);
3822

3823 3824 3825 3826
        if (qemuMigrationRestoreDomainState(conn, vm)) {
            event = virDomainEventLifecycleNewFromObj(vm,
                                                      VIR_DOMAIN_EVENT_RESUMED,
                                                      VIR_DOMAIN_EVENT_RESUMED_MIGRATED);
3827 3828
        }

3829
        if (virDomainSaveStatus(driver->xmlopt, cfg->stateDir, vm) < 0)
3830 3831 3832
            VIR_WARN("Failed to save status on vm %s", vm->def->name);
    }

3833
 done:
3834 3835 3836
    qemuMigrationCookieFree(mig);
    rv = 0;

3837
 cleanup:
3838
    qemuDomainEventQueue(driver, event);
3839 3840 3841 3842 3843 3844 3845 3846 3847 3848 3849 3850 3851
    virObjectUnref(cfg);
    return rv;
}

int
qemuMigrationConfirm(virConnectPtr conn,
                     virDomainObjPtr vm,
                     const char *cookiein,
                     int cookieinlen,
                     unsigned int flags,
                     int cancelled)
{
    virQEMUDriverPtr driver = conn->privateData;
3852
    qemuMigrationJobPhase phase;
3853 3854 3855 3856 3857 3858 3859 3860 3861 3862 3863 3864 3865 3866
    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);
3867 3868
    virCloseCallbacksUnset(driver->closeCallbacks, vm,
                           qemuMigrationCleanup);
3869 3870 3871 3872 3873

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

3874 3875 3876
    qemuMigrationJobFinish(driver, vm);
    if (!virDomainObjIsActive(vm) &&
        (!vm->persistent || (flags & VIR_MIGRATE_UNDEFINE_SOURCE))) {
3877 3878 3879 3880 3881
        if (flags & VIR_MIGRATE_UNDEFINE_SOURCE)
            virDomainDeleteConfig(cfg->configDir, cfg->autostartDir, vm);
        qemuDomainRemoveInactive(driver, vm);
    }

3882
 cleanup:
M
Michal Privoznik 已提交
3883
    virDomainObjEndAPI(&vm);
3884 3885 3886 3887 3888
    virObjectUnref(cfg);
    return ret;
}


3889 3890
enum qemuMigrationDestinationType {
    MIGRATION_DEST_HOST,
3891
    MIGRATION_DEST_CONNECT_HOST,
3892
    MIGRATION_DEST_UNIX,
3893
    MIGRATION_DEST_FD,
3894
};
3895

3896 3897 3898 3899
enum qemuMigrationForwardType {
    MIGRATION_FWD_DIRECT,
    MIGRATION_FWD_STREAM,
};
3900

3901 3902 3903 3904 3905 3906
typedef struct _qemuMigrationSpec qemuMigrationSpec;
typedef qemuMigrationSpec *qemuMigrationSpecPtr;
struct _qemuMigrationSpec {
    enum qemuMigrationDestinationType destType;
    union {
        struct {
3907
            const char *protocol;
3908 3909 3910 3911 3912
            const char *name;
            int port;
        } host;

        struct {
3913
            char *file;
3914 3915
            int sock;
        } unix_socket;
3916 3917 3918 3919 3920

        struct {
            int qemu;
            int local;
        } fd;
3921 3922 3923 3924 3925 3926 3927
    } dest;

    enum qemuMigrationForwardType fwdType;
    union {
        virStreamPtr stream;
    } fwd;
};
3928 3929 3930

#define TUNNEL_SEND_BUF_SIZE 65536

3931 3932 3933 3934 3935 3936 3937
typedef struct _qemuMigrationIOThread qemuMigrationIOThread;
typedef qemuMigrationIOThread *qemuMigrationIOThreadPtr;
struct _qemuMigrationIOThread {
    virThread thread;
    virStreamPtr st;
    int sock;
    virError err;
3938 3939
    int wakeupRecvFD;
    int wakeupSendFD;
3940 3941 3942
};

static void qemuMigrationIOFunc(void *arg)
3943
{
3944
    qemuMigrationIOThreadPtr data = arg;
3945 3946 3947 3948 3949 3950 3951
    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);
3952

3953
    if (VIR_ALLOC_N(buffer, TUNNEL_SEND_BUF_SIZE) < 0)
3954
        goto abrt;
3955

3956 3957 3958
    fds[0].fd = data->sock;
    fds[1].fd = data->wakeupRecvFD;

3959
    for (;;) {
3960 3961 3962 3963 3964 3965 3966 3967 3968 3969
        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;
3970
            virReportSystemError(errno, "%s",
3971 3972
                                 _("poll failed in migration tunnel"));
            goto abrt;
3973
        }
3974 3975 3976 3977 3978 3979 3980

        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");
3981
            break;
3982
        }
3983

3984 3985 3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997 3998 3999
        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;
            }
4000 4001
        }

4002 4003 4004 4005 4006 4007 4008 4009 4010 4011 4012 4013 4014 4015 4016 4017 4018
        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;
            }
        }
    }
4019

4020 4021
    if (virStreamFinish(data->st) < 0)
        goto error;
4022

4023 4024
    VIR_FREE(buffer);

4025 4026
    return;

4027
 abrt:
4028 4029 4030 4031 4032 4033 4034 4035 4036 4037 4038
    err = virSaveLastError();
    if (err && err->code == VIR_ERR_OK) {
        virFreeError(err);
        err = NULL;
    }
    virStreamAbort(data->st);
    if (err) {
        virSetError(err);
        virFreeError(err);
    }

4039
 error:
4040 4041
    virCopyLastError(&data->err);
    virResetLastError();
4042
    VIR_FREE(buffer);
4043 4044 4045 4046 4047 4048 4049
}


static qemuMigrationIOThreadPtr
qemuMigrationStartTunnel(virStreamPtr st,
                         int sock)
{
4050 4051
    qemuMigrationIOThreadPtr io = NULL;
    int wakeupFD[2] = { -1, -1 };
4052

4053 4054 4055 4056
    if (pipe2(wakeupFD, O_CLOEXEC) < 0) {
        virReportSystemError(errno, "%s",
                             _("Unable to make pipe"));
        goto error;
4057 4058
    }

4059
    if (VIR_ALLOC(io) < 0)
4060
        goto error;
4061

4062 4063
    io->st = st;
    io->sock = sock;
4064 4065
    io->wakeupRecvFD = wakeupFD[0];
    io->wakeupSendFD = wakeupFD[1];
4066 4067 4068 4069 4070 4071

    if (virThreadCreate(&io->thread, true,
                        qemuMigrationIOFunc,
                        io) < 0) {
        virReportSystemError(errno, "%s",
                             _("Unable to create migration thread"));
4072
        goto error;
4073 4074 4075
    }

    return io;
4076

4077
 error:
4078 4079 4080 4081
    VIR_FORCE_CLOSE(wakeupFD[0]);
    VIR_FORCE_CLOSE(wakeupFD[1]);
    VIR_FREE(io);
    return NULL;
4082 4083 4084
}

static int
4085
qemuMigrationStopTunnel(qemuMigrationIOThreadPtr io, bool error)
4086 4087
{
    int rv = -1;
4088 4089 4090 4091 4092 4093 4094 4095 4096
    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;
    }

4097 4098 4099 4100
    virThreadJoin(&io->thread);

    /* Forward error from the IO thread, to this thread */
    if (io->err.code != VIR_ERR_OK) {
4101 4102 4103 4104
        if (error)
            rv = 0;
        else
            virSetError(&io->err);
4105 4106 4107 4108 4109 4110
        virResetError(&io->err);
        goto cleanup;
    }

    rv = 0;

4111
 cleanup:
4112 4113
    VIR_FORCE_CLOSE(io->wakeupSendFD);
    VIR_FORCE_CLOSE(io->wakeupRecvFD);
4114 4115
    VIR_FREE(io);
    return rv;
4116 4117
}

4118
static int
4119
qemuMigrationConnect(virQEMUDriverPtr driver,
4120 4121 4122 4123 4124 4125 4126 4127 4128
                     virDomainObjPtr vm,
                     qemuMigrationSpecPtr spec)
{
    virNetSocketPtr sock;
    const char *host;
    char *port = NULL;
    int ret = -1;

    host = spec->dest.host.name;
4129
    if (virAsprintf(&port, "%d", spec->dest.host.port) < 0)
4130 4131 4132 4133 4134 4135 4136
        return -1;

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

    if (virSecurityManagerSetSocketLabel(driver->securityManager, vm->def) < 0)
        goto cleanup;
4137 4138 4139
    if (virNetSocketNewConnectTCP(host, port,
                                  AF_UNSPEC,
                                  &sock) == 0) {
4140
        spec->dest.fd.qemu = virNetSocketDupFD(sock, true);
4141
        virObjectUnref(sock);
4142 4143 4144 4145 4146
    }
    if (virSecurityManagerClearSocketLabel(driver->securityManager, vm->def) < 0 ||
        spec->dest.fd.qemu == -1)
        goto cleanup;

4147 4148 4149 4150 4151 4152 4153
    /* 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;
    }

4154 4155
    ret = 0;

4156
 cleanup:
4157 4158 4159 4160 4161 4162
    VIR_FREE(port);
    if (ret < 0)
        VIR_FORCE_CLOSE(spec->dest.fd.qemu);
    return ret;
}

4163
static int
4164
qemuMigrationRun(virQEMUDriverPtr driver,
4165 4166 4167 4168 4169 4170 4171
                 virDomainObjPtr vm,
                 const char *cookiein,
                 int cookieinlen,
                 char **cookieout,
                 int *cookieoutlen,
                 unsigned long flags,
                 unsigned long resource,
4172
                 qemuMigrationSpecPtr spec,
4173
                 virConnectPtr dconn,
4174 4175 4176
                 const char *graphicsuri,
                 size_t nmigrate_disks,
                 const char **migrate_disks)
4177
{
4178
    int ret = -1;
4179 4180
    unsigned int migrate_flags = QEMU_MONITOR_MIGRATE_BACKGROUND;
    qemuDomainObjPrivatePtr priv = vm->privateData;
4181
    qemuMigrationCookiePtr mig = NULL;
4182
    qemuMigrationIOThreadPtr iothread = NULL;
4183
    int fd = -1;
4184
    unsigned long migrate_speed = resource ? resource : priv->migMaxBandwidth;
4185
    virErrorPtr orig_err = NULL;
4186
    unsigned int cookieFlags = 0;
4187
    bool abort_on_error = !!(flags & VIR_MIGRATE_ABORT_ON_ERROR);
4188
    bool events = virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATION_EVENT);
4189
    int rc;
4190 4191 4192

    VIR_DEBUG("driver=%p, vm=%p, cookiein=%s, cookieinlen=%d, "
              "cookieout=%p, cookieoutlen=%p, flags=%lx, resource=%lu, "
4193 4194
              "spec=%p (dest=%d, fwd=%d), dconn=%p, graphicsuri=%s, "
              "nmigrate_disks=%zu, migrate_disks=%p",
4195 4196
              driver, vm, NULLSTR(cookiein), cookieinlen,
              cookieout, cookieoutlen, flags, resource,
4197
              spec, spec->destType, spec->fwdType, dconn,
4198
              NULLSTR(graphicsuri), nmigrate_disks, migrate_disks);
4199

4200 4201 4202 4203 4204 4205 4206 4207 4208 4209
    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;
    }

4210 4211
    if (virLockManagerPluginUsesState(driver->lockManager) &&
        !cookieout) {
4212 4213 4214 4215
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Migration with lock driver %s requires"
                         " cookie support"),
                       virLockManagerPluginGetName(driver->lockManager));
4216 4217 4218
        return -1;
    }

4219 4220 4221
    if (events)
        priv->signalIOError = abort_on_error;

4222 4223 4224
    mig = qemuMigrationEatCookie(driver, vm, cookiein, cookieinlen,
                                 cookieFlags | QEMU_MIGRATION_COOKIE_GRAPHICS);
    if (!mig)
4225 4226
        goto cleanup;

4227
    if (qemuDomainMigrateGraphicsRelocate(driver, vm, mig, graphicsuri) < 0)
4228 4229
        VIR_WARN("unable to provide data for graphics client relocation");

4230 4231 4232 4233 4234 4235 4236
    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,
4237 4238
                                         &migrate_flags,
                                         nmigrate_disks,
4239 4240
                                         migrate_disks,
                                         dconn) < 0) {
4241 4242 4243 4244 4245 4246 4247 4248
                goto cleanup;
            }
        } 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.");
        }
4249 4250
    }

4251
    /* Before EnterMonitor, since qemuMigrationSetOffline already does that */
4252 4253 4254 4255 4256 4257
    if (!(flags & VIR_MIGRATE_LIVE) &&
        virDomainObjGetState(vm, NULL) == VIR_DOMAIN_RUNNING) {
        if (qemuMigrationSetOffline(driver, vm) < 0)
            goto cleanup;
    }

4258 4259
    if (qemuMigrationSetCompression(driver, vm,
                                    flags & VIR_MIGRATE_COMPRESSED,
4260 4261 4262
                                    QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
        goto cleanup;

4263 4264
    if (qemuMigrationSetAutoConverge(driver, vm,
                                     flags & VIR_MIGRATE_AUTO_CONVERGE,
4265 4266 4267
                                     QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
        goto cleanup;

4268 4269
    if (qemuMigrationSetPinAll(driver, vm,
                               flags & VIR_MIGRATE_RDMA_PIN_ALL,
4270 4271 4272
                               QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
        goto cleanup;

4273 4274
    if (qemuDomainObjEnterMonitorAsync(driver, vm,
                                       QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
4275 4276
        goto cleanup;

4277
    if (priv->job.abortJob) {
4278 4279
        /* explicitly do this *after* we entered the monitor,
         * as this is a critical section so we are guaranteed
4280
         * priv->job.abortJob will not change */
4281
        ignore_value(qemuDomainObjExitMonitor(driver, vm));
4282
        priv->job.current->type = VIR_DOMAIN_JOB_CANCELLED;
4283 4284 4285 4286 4287 4288
        virReportError(VIR_ERR_OPERATION_ABORTED, _("%s: %s"),
                       qemuDomainAsyncJobTypeToString(priv->job.asyncJob),
                       _("canceled by client"));
        goto cleanup;
    }

4289 4290
    if (qemuMonitorSetMigrationSpeed(priv->mon, migrate_speed) < 0)
        goto exit_monitor;
4291

4292 4293
    /* connect to the destination qemu if needed */
    if (spec->destType == MIGRATION_DEST_CONNECT_HOST &&
4294
        qemuMigrationConnect(driver, vm, spec) < 0) {
4295
        goto exit_monitor;
4296
    }
4297

4298 4299
    switch (spec->destType) {
    case MIGRATION_DEST_HOST:
M
Michael R. Hines 已提交
4300 4301
        if (STREQ(spec->dest.host.protocol, "rdma") &&
            virProcessSetMaxMemLock(vm->pid, vm->def->mem.hard_limit << 10) < 0) {
4302
            goto exit_monitor;
M
Michael R. Hines 已提交
4303
        }
4304
        ret = qemuMonitorMigrateToHost(priv->mon, migrate_flags,
4305
                                       spec->dest.host.protocol,
4306 4307 4308 4309
                                       spec->dest.host.name,
                                       spec->dest.host.port);
        break;

4310 4311 4312 4313
    case MIGRATION_DEST_CONNECT_HOST:
        /* handled above and transformed into MIGRATION_DEST_FD */
        break;

4314
    case MIGRATION_DEST_UNIX:
4315
        if (virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATE_QEMU_UNIX)) {
4316 4317 4318 4319 4320 4321 4322 4323 4324
            ret = qemuMonitorMigrateToUnix(priv->mon, migrate_flags,
                                           spec->dest.unix_socket.file);
        } else {
            const char *args[] = {
                "nc", "-U", spec->dest.unix_socket.file, NULL
            };
            ret = qemuMonitorMigrateToCommand(priv->mon, migrate_flags, args);
        }
        break;
4325 4326

    case MIGRATION_DEST_FD:
4327
        if (spec->fwdType != MIGRATION_FWD_DIRECT) {
4328
            fd = spec->dest.fd.local;
4329 4330
            spec->dest.fd.local = -1;
        }
4331 4332 4333 4334
        ret = qemuMonitorMigrateToFd(priv->mon, migrate_flags,
                                     spec->dest.fd.qemu);
        VIR_FORCE_CLOSE(spec->dest.fd.qemu);
        break;
4335
    }
4336 4337
    if (qemuDomainObjExitMonitor(driver, vm) < 0)
        ret = -1;
4338
    if (ret < 0)
4339 4340
        goto cleanup;
    ret = -1;
4341 4342 4343 4344

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

4345 4346 4347 4348 4349
    if (spec->destType == MIGRATION_DEST_UNIX) {
        /* It is also possible that the migrate didn't fail initially, but
         * rather failed later on.  Check its status before waiting for a
         * connection from qemu which may never be initiated.
         */
4350
        if (qemuMigrationCheckJobStatus(driver, vm,
4351
                                        QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
4352
            goto cancel;
4353

4354 4355 4356 4357 4358 4359 4360
        while ((fd = accept(spec->dest.unix_socket.sock, NULL, NULL)) < 0) {
            if (errno == EAGAIN || errno == EINTR)
                continue;
            virReportSystemError(errno, "%s",
                                 _("failed to accept connection from qemu"));
            goto cancel;
        }
4361 4362
    }

4363 4364
    if (spec->fwdType != MIGRATION_FWD_DIRECT &&
        !(iothread = qemuMigrationStartTunnel(spec->fwd.stream, fd)))
4365 4366
        goto cancel;

4367 4368
    rc = qemuMigrationWaitForCompletion(driver, vm,
                                        QEMU_ASYNC_JOB_MIGRATION_OUT,
4369
                                        dconn, abort_on_error, !!mig->nbd);
4370 4371 4372
    if (rc == -2)
        goto cancel;
    else if (rc == -1)
4373
        goto cleanup;
4374

4375 4376 4377 4378 4379 4380 4381 4382 4383
    /* When migration completed, QEMU will have paused the
     * CPUs for us, but unless we're using the JSON monitor
     * we won't have been notified of this, so might still
     * think we're running. For v2 protocol this doesn't
     * matter because we'll kill the VM soon, but for v3
     * this is important because we stay paused until the
     * confirm3 step, but need to release the lock state
     */
    if (virDomainObjGetState(vm, NULL) == VIR_DOMAIN_RUNNING) {
4384 4385
        if (qemuMigrationSetOffline(driver, vm) < 0) {
            priv->job.current->type = VIR_DOMAIN_JOB_FAILED;
4386
            goto cleanup;
4387
        }
4388 4389
    }

4390
    ret = 0;
4391

4392
 cleanup:
4393 4394 4395
    if (ret < 0 && !orig_err)
        orig_err = virSaveLastError();

4396
    /* cancel any outstanding NBD jobs */
4397
    if (mig && mig->nbd) {
4398
        if (qemuMigrationCancelDriveMirror(driver, vm, ret == 0,
4399 4400
                                           QEMU_ASYNC_JOB_MIGRATION_OUT,
                                           dconn) < 0)
4401 4402
            ret = -1;
    }
4403

4404
    if (spec->fwdType != MIGRATION_FWD_DIRECT) {
4405
        if (iothread && qemuMigrationStopTunnel(iothread, ret < 0) < 0)
4406 4407
            ret = -1;
    }
4408
    VIR_FORCE_CLOSE(fd);
4409

4410
    if (priv->job.completed) {
4411
        qemuDomainJobInfoUpdateTime(priv->job.completed);
4412
        qemuDomainJobInfoUpdateDowntime(priv->job.completed);
4413
        ignore_value(virTimeMillisNow(&priv->job.completed->sent));
4414
    }
4415

4416 4417 4418
    if (priv->job.current->type == VIR_DOMAIN_JOB_UNBOUNDED)
        priv->job.current->type = VIR_DOMAIN_JOB_FAILED;

4419 4420
    cookieFlags |= QEMU_MIGRATION_COOKIE_NETWORK |
                   QEMU_MIGRATION_COOKIE_STATS;
4421 4422
    if (flags & VIR_MIGRATE_PERSIST_DEST)
        cookieFlags |= QEMU_MIGRATION_COOKIE_PERSISTENT;
4423
    if (ret == 0 &&
4424 4425
        qemuMigrationBakeCookie(mig, driver, vm, cookieout,
                                cookieoutlen, cookieFlags) < 0) {
4426
        VIR_WARN("Unable to encode migration cookie");
4427
    }
4428

4429 4430
    qemuMigrationCookieFree(mig);

4431 4432 4433
    if (events)
        priv->signalIOError = false;

4434 4435 4436 4437 4438
    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }

4439 4440
    return ret;

4441 4442 4443 4444
 exit_monitor:
    ignore_value(qemuDomainObjExitMonitor(driver, vm));
    goto cleanup;

4445
 cancel:
4446 4447
    orig_err = virSaveLastError();

4448
    if (virDomainObjIsActive(vm)) {
4449 4450
        if (qemuDomainObjEnterMonitorAsync(driver, vm,
                                           QEMU_ASYNC_JOB_MIGRATION_OUT) == 0) {
4451
            qemuMonitorMigrateCancel(priv->mon);
4452
            ignore_value(qemuDomainObjExitMonitor(driver, vm));
4453
        }
4454
    }
4455 4456 4457
    goto cleanup;
}

4458
/* Perform migration using QEMU's native migrate support,
4459 4460
 * not encrypted obviously
 */
4461
static int doNativeMigrate(virQEMUDriverPtr driver,
4462 4463 4464 4465 4466 4467 4468
                           virDomainObjPtr vm,
                           const char *uri,
                           const char *cookiein,
                           int cookieinlen,
                           char **cookieout,
                           int *cookieoutlen,
                           unsigned long flags,
4469
                           unsigned long resource,
4470
                           virConnectPtr dconn,
4471 4472 4473
                           const char *graphicsuri,
                           size_t nmigrate_disks,
                           const char **migrate_disks)
4474
{
4475
    qemuDomainObjPrivatePtr priv = vm->privateData;
M
Martin Kletzander 已提交
4476
    virURIPtr uribits = NULL;
4477
    int ret = -1;
4478 4479 4480
    qemuMigrationSpec spec;

    VIR_DEBUG("driver=%p, vm=%p, uri=%s, cookiein=%s, cookieinlen=%d, "
4481
              "cookieout=%p, cookieoutlen=%p, flags=%lx, resource=%lu, "
4482
              "graphicsuri=%s, nmigrate_disks=%zu migrate_disks=%p",
4483
              driver, vm, uri, NULLSTR(cookiein), cookieinlen,
4484
              cookieout, cookieoutlen, flags, resource,
4485
              NULLSTR(graphicsuri), nmigrate_disks, migrate_disks);
4486

4487
    if (!(uribits = qemuMigrationParseURI(uri, NULL)))
4488 4489
        return -1;

4490 4491 4492 4493 4494 4495 4496
    if (uribits->scheme == NULL) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("missing scheme in migration URI: %s"),
                       uri);
        goto cleanup;
    }

M
Michael R. Hines 已提交
4497 4498 4499 4500 4501
    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"));
4502
            goto cleanup;
M
Michael R. Hines 已提交
4503
        }
4504
        if (!virMemoryLimitIsSet(vm->def->mem.hard_limit)) {
M
Michael R. Hines 已提交
4505 4506 4507
            virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                           _("cannot start RDMA migration with no memory hard "
                             "limit set"));
4508
            goto cleanup;
M
Michael R. Hines 已提交
4509 4510 4511 4512 4513
        }
    }

    if (virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATE_QEMU_FD) &&
        STRNEQ(uribits->scheme, "rdma"))
4514 4515
        spec.destType = MIGRATION_DEST_CONNECT_HOST;
    else
4516
        spec.destType = MIGRATION_DEST_HOST;
4517
    spec.dest.host.protocol = uribits->scheme;
4518 4519 4520
    spec.dest.host.name = uribits->server;
    spec.dest.host.port = uribits->port;
    spec.fwdType = MIGRATION_FWD_DIRECT;
4521

4522
    ret = qemuMigrationRun(driver, vm, cookiein, cookieinlen, cookieout,
4523
                           cookieoutlen, flags, resource, &spec, dconn,
4524
                           graphicsuri, nmigrate_disks, migrate_disks);
4525 4526 4527 4528

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

4529
 cleanup:
4530
    virURIFree(uribits);
4531 4532 4533 4534 4535

    return ret;
}


4536
static int doTunnelMigrate(virQEMUDriverPtr driver,
4537 4538 4539 4540 4541 4542 4543
                           virDomainObjPtr vm,
                           virStreamPtr st,
                           const char *cookiein,
                           int cookieinlen,
                           char **cookieout,
                           int *cookieoutlen,
                           unsigned long flags,
4544
                           unsigned long resource,
4545
                           virConnectPtr dconn,
4546 4547 4548
                           const char *graphicsuri,
                           size_t nmigrate_disks,
                           const char **migrate_disks)
4549 4550
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
4551
    virNetSocketPtr sock = NULL;
4552 4553
    int ret = -1;
    qemuMigrationSpec spec;
4554
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
4555 4556

    VIR_DEBUG("driver=%p, vm=%p, st=%p, cookiein=%s, cookieinlen=%d, "
4557
              "cookieout=%p, cookieoutlen=%p, flags=%lx, resource=%lu, "
4558
              "graphicsuri=%s, nmigrate_disks=%zu, migrate_disks=%p",
4559
              driver, vm, st, NULLSTR(cookiein), cookieinlen,
4560
              cookieout, cookieoutlen, flags, resource,
4561
              NULLSTR(graphicsuri), nmigrate_disks, migrate_disks);
4562

4563 4564 4565
    if (!virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATE_QEMU_FD) &&
        !virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATE_QEMU_UNIX) &&
        !virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATE_QEMU_EXEC)) {
4566 4567
        virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                       _("Source qemu is too old to support tunnelled migration"));
4568
        virObjectUnref(cfg);
4569
        return -1;
4570 4571 4572 4573 4574
    }

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

4575
    if (virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATE_QEMU_FD)) {
4576 4577 4578 4579 4580 4581
        int fds[2];

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

4582
        if (pipe2(fds, O_CLOEXEC) == 0) {
4583 4584 4585 4586
            spec.dest.fd.qemu = fds[1];
            spec.dest.fd.local = fds[0];
        }
        if (spec.dest.fd.qemu == -1 ||
4587
            virSecurityManagerSetImageFDLabel(driver->securityManager, vm->def,
4588
                                              spec.dest.fd.qemu) < 0) {
4589 4590 4591 4592 4593 4594 4595 4596 4597 4598 4599
            virReportSystemError(errno, "%s",
                        _("cannot create pipe for tunnelled migration"));
            goto cleanup;
        }
    } else {
        spec.destType = MIGRATION_DEST_UNIX;
        spec.dest.unix_socket.sock = -1;
        spec.dest.unix_socket.file = NULL;

        if (virAsprintf(&spec.dest.unix_socket.file,
                        "%s/qemu.tunnelmigrate.src.%s",
4600
                        cfg->libDir, vm->def->name) < 0)
4601 4602 4603
            goto cleanup;

        if (virNetSocketNewListenUNIX(spec.dest.unix_socket.file, 0700,
4604
                                      cfg->user, cfg->group,
4605 4606 4607 4608 4609 4610 4611
                                      &sock) < 0 ||
            virNetSocketListen(sock, 1) < 0)
            goto cleanup;

        spec.dest.unix_socket.sock = virNetSocketGetFD(sock);
    }

4612
    ret = qemuMigrationRun(driver, vm, cookiein, cookieinlen, cookieout,
4613
                           cookieoutlen, flags, resource, &spec, dconn,
4614
                           graphicsuri, nmigrate_disks, migrate_disks);
4615

4616
 cleanup:
4617 4618 4619 4620
    if (spec.destType == MIGRATION_DEST_FD) {
        VIR_FORCE_CLOSE(spec.dest.fd.qemu);
        VIR_FORCE_CLOSE(spec.dest.fd.local);
    } else {
4621
        virObjectUnref(sock);
4622 4623
        VIR_FREE(spec.dest.unix_socket.file);
    }
4624

4625
    virObjectUnref(cfg);
4626 4627 4628 4629
    return ret;
}


4630 4631 4632 4633
/* 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 */
4634
static int doPeer2PeerMigrate2(virQEMUDriverPtr driver,
4635
                               virConnectPtr sconn ATTRIBUTE_UNUSED,
4636 4637
                               virConnectPtr dconn,
                               virDomainObjPtr vm,
4638
                               const char *dconnuri,
4639 4640 4641
                               unsigned long flags,
                               const char *dname,
                               unsigned long resource)
4642 4643 4644
{
    virDomainPtr ddomain = NULL;
    char *uri_out = NULL;
4645
    char *cookie = NULL;
4646 4647 4648
    char *dom_xml = NULL;
    int cookielen = 0, ret;
    virErrorPtr orig_err = NULL;
4649
    bool cancelled;
4650
    virStreamPtr st = NULL;
4651
    unsigned long destflags;
4652

4653
    VIR_DEBUG("driver=%p, sconn=%p, dconn=%p, vm=%p, dconnuri=%s, "
4654
              "flags=%lx, dname=%s, resource=%lu",
4655 4656
              driver, sconn, dconn, vm, NULLSTR(dconnuri),
              flags, NULLSTR(dname), resource);
4657

4658 4659 4660 4661 4662
    /* 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,
4663 4664
                                        QEMU_DOMAIN_FORMAT_LIVE_FLAGS |
                                        VIR_DOMAIN_XML_MIGRATABLE)))
4665 4666 4667 4668 4669
        return -1;

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

4670 4671
    destflags = flags & ~(VIR_MIGRATE_ABORT_ON_ERROR |
                          VIR_MIGRATE_AUTO_CONVERGE);
4672

4673 4674 4675 4676 4677 4678 4679 4680 4681 4682
    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;

4683
        qemuDomainObjEnterRemote(vm);
4684
        ret = dconn->driver->domainMigratePrepareTunnel
4685
            (dconn, st, destflags, dname, resource, dom_xml);
4686
        qemuDomainObjExitRemote(vm);
4687
    } else {
4688
        qemuDomainObjEnterRemote(vm);
4689 4690
        ret = dconn->driver->domainMigratePrepare2
            (dconn, &cookie, &cookielen, NULL, &uri_out,
4691
             destflags, dname, resource, dom_xml);
4692
        qemuDomainObjExitRemote(vm);
4693 4694 4695
    }
    VIR_FREE(dom_xml);
    if (ret == -1)
4696 4697 4698
        goto cleanup;

    /* the domain may have shutdown or crashed while we had the locks dropped
4699
     * in qemuDomainObjEnterRemote, so check again
4700 4701
     */
    if (!virDomainObjIsActive(vm)) {
4702 4703
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest unexpectedly quit"));
4704 4705 4706
        goto cleanup;
    }

4707 4708
    if (!(flags & VIR_MIGRATE_TUNNELLED) &&
        (uri_out == NULL)) {
4709 4710
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("domainMigratePrepare2 did not set uri"));
4711
        cancelled = true;
4712
        orig_err = virSaveLastError();
4713
        goto finish;
4714 4715
    }

4716 4717 4718 4719
    /* Perform the migration.  The driver isn't supposed to return
     * until the migration is complete.
     */
    VIR_DEBUG("Perform %p", sconn);
4720
    qemuMigrationJobSetPhase(driver, vm, QEMU_MIGRATION_PHASE_PERFORM2);
4721
    if (flags & VIR_MIGRATE_TUNNELLED)
4722 4723
        ret = doTunnelMigrate(driver, vm, st,
                              NULL, 0, NULL, NULL,
4724 4725
                              flags, resource, dconn,
                              NULL, 0, NULL);
4726 4727 4728 4729
    else
        ret = doNativeMigrate(driver, vm, uri_out,
                              cookie, cookielen,
                              NULL, NULL, /* No out cookie with v2 migration */
4730
                              flags, resource, dconn, NULL, 0, NULL);
4731 4732 4733 4734

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

4736 4737 4738
    /* If Perform returns < 0, then we need to cancel the VM
     * startup on the destination
     */
4739
    cancelled = ret < 0;
4740

4741
 finish:
4742 4743 4744 4745
    /* 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.
     */
4746
    dname = dname ? dname : vm->def->name;
4747
    VIR_DEBUG("Finish2 %p ret=%d", dconn, ret);
4748
    qemuDomainObjEnterRemote(vm);
4749
    ddomain = dconn->driver->domainMigrateFinish2
4750
        (dconn, dname, cookie, cookielen,
4751
         uri_out ? uri_out : dconnuri, destflags, cancelled);
4752
    qemuDomainObjExitRemote(vm);
4753 4754
    if (cancelled && ddomain)
        VIR_ERROR(_("finish step ignored that migration was cancelled"));
4755

4756
 cleanup:
4757
    if (ddomain) {
4758
        virObjectUnref(ddomain);
4759 4760 4761 4762
        ret = 0;
    } else {
        ret = -1;
    }
4763

4764
    virObjectUnref(st);
4765 4766 4767 4768 4769 4770

    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }
    VIR_FREE(uri_out);
4771
    VIR_FREE(cookie);
4772 4773

    return ret;
4774 4775 4776
}


4777 4778 4779 4780
/* 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 */
4781 4782 4783 4784 4785 4786 4787 4788 4789
static int
doPeer2PeerMigrate3(virQEMUDriverPtr driver,
                    virConnectPtr sconn,
                    virConnectPtr dconn,
                    const char *dconnuri,
                    virDomainObjPtr vm,
                    const char *xmlin,
                    const char *dname,
                    const char *uri,
4790
                    const char *graphicsuri,
4791
                    const char *listenAddress,
4792 4793
                    size_t nmigrate_disks,
                    const char **migrate_disks,
4794 4795 4796
                    unsigned long long bandwidth,
                    bool useParams,
                    unsigned long flags)
4797 4798 4799 4800 4801 4802 4803 4804 4805 4806
{
    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;
4807
    bool cancelled = true;
4808
    virStreamPtr st = NULL;
4809
    unsigned long destflags;
4810 4811 4812
    virTypedParameterPtr params = NULL;
    int nparams = 0;
    int maxparams = 0;
4813
    size_t i;
4814 4815

    VIR_DEBUG("driver=%p, sconn=%p, dconn=%p, dconnuri=%s, vm=%p, xmlin=%s, "
4816
              "dname=%s, uri=%s, graphicsuri=%s, listenAddress=%s, "
4817 4818
              "nmigrate_disks=%zu, migrate_disks=%p, bandwidth=%llu, "
              "useParams=%d, flags=%lx",
4819
              driver, sconn, dconn, NULLSTR(dconnuri), vm, NULLSTR(xmlin),
4820
              NULLSTR(dname), NULLSTR(uri), NULLSTR(graphicsuri),
4821 4822
              NULLSTR(listenAddress), nmigrate_disks, migrate_disks,
              bandwidth, useParams, flags);
4823

4824 4825 4826 4827 4828
    /* 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.  */

4829
    dom_xml = qemuMigrationBeginPhase(driver, vm, xmlin, dname,
4830 4831
                                      &cookieout, &cookieoutlen,
                                      nmigrate_disks, migrate_disks, flags);
4832 4833 4834
    if (!dom_xml)
        goto cleanup;

4835 4836 4837 4838 4839 4840 4841 4842 4843 4844 4845 4846 4847 4848 4849 4850 4851 4852 4853 4854
    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;
4855 4856 4857 4858 4859 4860

        if (graphicsuri &&
            virTypedParamsAddString(&params, &nparams, &maxparams,
                                    VIR_MIGRATE_PARAM_GRAPHICS_URI,
                                    graphicsuri) < 0)
            goto cleanup;
4861 4862 4863 4864 4865
        if (listenAddress &&
            virTypedParamsAddString(&params, &nparams, &maxparams,
                                    VIR_MIGRATE_PARAM_LISTEN_ADDRESS,
                                    listenAddress) < 0)
            goto cleanup;
4866 4867 4868 4869 4870
        for (i = 0; i < nmigrate_disks; i++)
            if (virTypedParamsAddString(&params, &nparams, &maxparams,
                                        VIR_MIGRATE_PARAM_MIGRATE_DISKS,
                                        migrate_disks[i]) < 0)
                goto cleanup;
4871 4872
    }

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

4876 4877
    destflags = flags & ~(VIR_MIGRATE_ABORT_ON_ERROR |
                          VIR_MIGRATE_AUTO_CONVERGE);
4878

4879 4880 4881 4882 4883 4884 4885 4886 4887
    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;

4888
        qemuDomainObjEnterRemote(vm);
4889 4890 4891 4892 4893 4894 4895 4896 4897
        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);
        }
4898
        qemuDomainObjExitRemote(vm);
4899
    } else {
4900
        qemuDomainObjEnterRemote(vm);
4901 4902 4903 4904 4905 4906 4907 4908 4909
        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);
        }
4910
        qemuDomainObjExitRemote(vm);
4911 4912 4913 4914 4915
    }
    VIR_FREE(dom_xml);
    if (ret == -1)
        goto cleanup;

L
liguang 已提交
4916 4917 4918 4919
    if (flags & VIR_MIGRATE_OFFLINE) {
        VIR_DEBUG("Offline migration, skipping Perform phase");
        VIR_FREE(cookieout);
        cookieoutlen = 0;
4920
        cancelled = false;
L
liguang 已提交
4921 4922 4923
        goto finish;
    }

4924 4925 4926 4927
    if (uri_out) {
        uri = uri_out;
        if (useParams &&
            virTypedParamsReplaceString(&params, &nparams,
4928 4929
                                        VIR_MIGRATE_PARAM_URI, uri_out) < 0) {
            orig_err = virSaveLastError();
4930
            goto finish;
4931
        }
4932
    } else if (!uri && !(flags & VIR_MIGRATE_TUNNELLED)) {
4933 4934
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("domainMigratePrepare3 did not set uri"));
4935
        orig_err = virSaveLastError();
4936 4937 4938 4939 4940 4941 4942 4943
        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.
     */
4944
    VIR_DEBUG("Perform3 %p uri=%s", sconn, NULLSTR(uri));
4945
    qemuMigrationJobSetPhase(driver, vm, QEMU_MIGRATION_PHASE_PERFORM3);
4946 4947 4948 4949 4950
    VIR_FREE(cookiein);
    cookiein = cookieout;
    cookieinlen = cookieoutlen;
    cookieout = NULL;
    cookieoutlen = 0;
4951
    if (flags & VIR_MIGRATE_TUNNELLED) {
4952 4953 4954
        ret = doTunnelMigrate(driver, vm, st,
                              cookiein, cookieinlen,
                              &cookieout, &cookieoutlen,
4955 4956
                              flags, bandwidth, dconn, graphicsuri,
                              nmigrate_disks, migrate_disks);
4957 4958
    } else {
        ret = doNativeMigrate(driver, vm, uri,
4959 4960
                              cookiein, cookieinlen,
                              &cookieout, &cookieoutlen,
4961 4962
                              flags, bandwidth, dconn, graphicsuri,
                              nmigrate_disks, migrate_disks);
4963
    }
4964 4965

    /* Perform failed. Make sure Finish doesn't overwrite the error */
4966
    if (ret < 0) {
4967
        orig_err = virSaveLastError();
4968 4969 4970 4971
    } else {
        qemuMigrationJobSetPhase(driver, vm,
                                 QEMU_MIGRATION_PHASE_PERFORM3_DONE);
    }
4972 4973 4974 4975

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

4978
 finish:
4979 4980 4981 4982 4983 4984 4985 4986 4987 4988 4989 4990
    /*
     * 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;
4991 4992 4993 4994 4995 4996 4997 4998 4999 5000 5001 5002 5003 5004 5005 5006 5007 5008 5009 5010 5011 5012 5013

    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);
    }
5014 5015 5016 5017 5018 5019 5020 5021 5022 5023 5024 5025 5026 5027 5028 5029 5030 5031 5032 5033

    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();
5034 5035
                if (err &&
                    err->domain == VIR_FROM_QEMU &&
5036 5037 5038 5039 5040 5041 5042
                    err->code != VIR_ERR_MIGRATE_FINISH_OK) {
                    virFreeError(orig_err);
                    orig_err = NULL;
                }
            }
        }
    }
5043

5044 5045 5046 5047 5048 5049 5050
    /* 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.
5051
     */
5052
    cancelled = ddomain == NULL;
5053

5054 5055 5056 5057 5058 5059
    /* 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();

5060 5061 5062 5063
    /*
     * If cancelled, then src VM will be restarted, else
     * it will be killed
     */
5064
    VIR_DEBUG("Confirm3 %p cancelled=%d vm=%p", sconn, cancelled, vm);
5065 5066 5067 5068 5069
    VIR_FREE(cookiein);
    cookiein = cookieout;
    cookieinlen = cookieoutlen;
    cookieout = NULL;
    cookieoutlen = 0;
5070 5071 5072
    ret = qemuMigrationConfirmPhase(driver, sconn, vm,
                                    cookiein, cookieinlen,
                                    flags, cancelled);
5073 5074 5075 5076
    /* 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.
     */
5077 5078 5079
    if (ret < 0)
        VIR_WARN("Guest %s probably left in 'paused' state on source",
                 vm->def->name);
5080 5081 5082

 cleanup:
    if (ddomain) {
5083
        virObjectUnref(ddomain);
5084 5085 5086 5087 5088
        ret = 0;
    } else {
        ret = -1;
    }

5089
    virObjectUnref(st);
5090 5091 5092 5093 5094 5095 5096 5097

    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }
    VIR_FREE(uri_out);
    VIR_FREE(cookiein);
    VIR_FREE(cookieout);
5098
    virTypedParamsFree(params, nparams);
5099 5100 5101 5102
    return ret;
}


5103 5104 5105 5106 5107 5108 5109 5110 5111 5112 5113 5114
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);
}


5115 5116 5117 5118 5119 5120 5121 5122 5123 5124 5125 5126
static int virConnectCredType[] = {
    VIR_CRED_AUTHNAME,
    VIR_CRED_PASSPHRASE,
};


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


5127
static int doPeer2PeerMigrate(virQEMUDriverPtr driver,
5128
                              virConnectPtr sconn,
5129
                              virDomainObjPtr vm,
5130
                              const char *xmlin,
5131
                              const char *dconnuri,
5132
                              const char *uri,
5133
                              const char *graphicsuri,
5134
                              const char *listenAddress,
5135 5136
                              size_t nmigrate_disks,
                              const char **migrate_disks,
5137 5138
                              unsigned long flags,
                              const char *dname,
5139 5140
                              unsigned long resource,
                              bool *v3proto)
5141 5142 5143 5144
{
    int ret = -1;
    virConnectPtr dconn = NULL;
    bool p2p;
5145
    virErrorPtr orig_err = NULL;
5146
    bool offline = false;
5147
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
5148
    bool useParams;
5149

5150 5151 5152
    VIR_DEBUG("driver=%p, sconn=%p, vm=%p, xmlin=%s, dconnuri=%s, uri=%s, "
              "graphicsuri=%s, listenAddress=%s, nmigrate_disks=%zu, "
              "migrate_disks=%p, flags=%lx, dname=%s, resource=%lu",
5153
              driver, sconn, vm, NULLSTR(xmlin), NULLSTR(dconnuri),
5154
              NULLSTR(uri), NULLSTR(graphicsuri), NULLSTR(listenAddress),
5155
              nmigrate_disks, migrate_disks, flags, NULLSTR(dname), resource);
5156

5157 5158 5159 5160 5161 5162 5163 5164 5165 5166 5167 5168 5169 5170
    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;
    }

5171 5172 5173 5174
    /* the order of operations is important here; we make sure the
     * destination side is completely setup before we touch the source
     */

5175
    qemuDomainObjEnterRemote(vm);
5176
    dconn = virConnectOpenAuth(dconnuri, &virConnectAuthConfig, 0);
5177
    qemuDomainObjExitRemote(vm);
5178
    if (dconn == NULL) {
5179
        virReportError(VIR_ERR_OPERATION_FAILED,
5180 5181
                       _("Failed to connect to remote libvirt URI %s: %s"),
                       dconnuri, virGetLastErrorMessage());
5182
        virObjectUnref(cfg);
5183 5184 5185
        return -1;
    }

5186 5187
    if (virConnectSetKeepAlive(dconn, cfg->keepAliveInterval,
                               cfg->keepAliveCount) < 0)
5188 5189
        goto cleanup;

5190 5191 5192 5193 5194
    if (virConnectRegisterCloseCallback(dconn, qemuMigrationConnectionClosed,
                                        vm, NULL) < 0) {
        goto cleanup;
    }

5195
    qemuDomainObjEnterRemote(vm);
5196 5197
    p2p = VIR_DRV_SUPPORTS_FEATURE(dconn->driver, dconn,
                                   VIR_DRV_FEATURE_MIGRATION_P2P);
5198
        /* v3proto reflects whether the caller used Perform3, but with
5199
         * p2p migrate, regardless of whether Perform2 or Perform3
5200 5201 5202 5203
         * were used, we decide protocol based on what target supports
         */
    *v3proto = VIR_DRV_SUPPORTS_FEATURE(dconn->driver, dconn,
                                        VIR_DRV_FEATURE_MIGRATION_V3);
5204 5205
    useParams = VIR_DRV_SUPPORTS_FEATURE(dconn->driver, dconn,
                                         VIR_DRV_FEATURE_MIGRATION_PARAMS);
L
liguang 已提交
5206 5207 5208
    if (flags & VIR_MIGRATE_OFFLINE)
        offline = VIR_DRV_SUPPORTS_FEATURE(dconn->driver, dconn,
                                           VIR_DRV_FEATURE_MIGRATION_OFFLINE);
5209
    qemuDomainObjExitRemote(vm);
5210

5211
    if (!p2p) {
5212 5213
        virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                       _("Destination libvirt does not support peer-to-peer migration protocol"));
5214 5215 5216
        goto cleanup;
    }

5217 5218
    /* Only xmlin, dname, uri, and bandwidth parameters can be used with
     * old-style APIs. */
5219
    if (!useParams && (graphicsuri || listenAddress || nmigrate_disks)) {
5220 5221 5222 5223 5224 5225
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("Migration APIs with extensible parameters are not "
                         "supported but extended parameters were passed"));
        goto cleanup;
    }

L
liguang 已提交
5226 5227 5228 5229 5230 5231 5232
    if (flags & VIR_MIGRATE_OFFLINE && !offline) {
        virReportError(VIR_ERR_ARGUMENT_UNSUPPORTED, "%s",
                       _("offline migration is not supported by "
                         "the destination host"));
        goto cleanup;
    }

5233
    /* domain may have been stopped while we were talking to remote daemon */
L
liguang 已提交
5234
    if (!virDomainObjIsActive(vm) && !(flags & VIR_MIGRATE_OFFLINE)) {
5235 5236
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest unexpectedly quit"));
5237 5238 5239
        goto cleanup;
    }

5240 5241 5242 5243 5244 5245 5246
    /* 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;

5247 5248
    if (*v3proto) {
        ret = doPeer2PeerMigrate3(driver, sconn, dconn, dconnuri, vm, xmlin,
5249
                                  dname, uri, graphicsuri, listenAddress,
5250 5251
                                  nmigrate_disks, migrate_disks, resource,
                                  useParams, flags);
5252
    } else {
5253
        ret = doPeer2PeerMigrate2(driver, sconn, dconn, vm,
5254
                                  dconnuri, flags, dname, resource);
5255
    }
5256

5257
 cleanup:
5258
    orig_err = virSaveLastError();
5259
    qemuDomainObjEnterRemote(vm);
5260
    virConnectUnregisterCloseCallback(dconn, qemuMigrationConnectionClosed);
5261
    virObjectUnref(dconn);
5262
    qemuDomainObjExitRemote(vm);
5263 5264 5265 5266
    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }
5267
    virObjectUnref(cfg);
5268 5269 5270 5271
    return ret;
}


5272 5273 5274 5275 5276 5277
/*
 * 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
5278
qemuMigrationPerformJob(virQEMUDriverPtr driver,
5279 5280 5281 5282 5283
                        virConnectPtr conn,
                        virDomainObjPtr vm,
                        const char *xmlin,
                        const char *dconnuri,
                        const char *uri,
5284
                        const char *graphicsuri,
5285
                        const char *listenAddress,
5286 5287
                        size_t nmigrate_disks,
                        const char **migrate_disks,
5288
                        const char *cookiein,
5289 5290 5291 5292 5293 5294 5295
                        int cookieinlen,
                        char **cookieout,
                        int *cookieoutlen,
                        unsigned long flags,
                        const char *dname,
                        unsigned long resource,
                        bool v3proto)
5296
{
5297
    virObjectEventPtr event = NULL;
5298
    int ret = -1;
5299
    virErrorPtr orig_err = NULL;
5300
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
5301
    bool abort_on_error = !!(flags & VIR_MIGRATE_ABORT_ON_ERROR);
5302

5303
    if (qemuMigrationJobStart(driver, vm, QEMU_ASYNC_JOB_MIGRATION_OUT) < 0)
5304 5305
        goto cleanup;

L
liguang 已提交
5306
    if (!virDomainObjIsActive(vm) && !(flags & VIR_MIGRATE_OFFLINE)) {
5307 5308
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s", _("domain is not running"));
5309 5310 5311
        goto endjob;
    }

5312
    if (!qemuMigrationIsAllowed(driver, vm, NULL, true, abort_on_error))
5313
        goto endjob;
5314

5315 5316
    if (!(flags & VIR_MIGRATE_UNSAFE) &&
        !qemuMigrationIsSafe(vm->def, nmigrate_disks, migrate_disks))
5317
        goto endjob;
5318

5319
    qemuMigrationStoreDomainState(vm);
5320 5321

    if ((flags & (VIR_MIGRATE_TUNNELLED | VIR_MIGRATE_PEER2PEER))) {
5322
        ret = doPeer2PeerMigrate(driver, conn, vm, xmlin,
5323
                                 dconnuri, uri, graphicsuri, listenAddress,
5324
                                 nmigrate_disks, migrate_disks,
5325
                                 flags, dname, resource, &v3proto);
5326
    } else {
5327 5328 5329
        qemuMigrationJobSetPhase(driver, vm, QEMU_MIGRATION_PHASE_PERFORM2);
        ret = doNativeMigrate(driver, vm, uri, cookiein, cookieinlen,
                              cookieout, cookieoutlen,
5330
                              flags, resource, NULL, NULL, 0, NULL);
5331
    }
5332 5333
    if (ret < 0)
        goto endjob;
5334

5335 5336 5337 5338
    /*
     * In v3 protocol, the source VM is not killed off until the
     * confirm step.
     */
5339
    if (!v3proto) {
5340 5341
        qemuProcessStop(driver, vm, VIR_DOMAIN_SHUTOFF_MIGRATED,
                        VIR_QEMU_PROCESS_STOP_MIGRATED);
5342
        virDomainAuditStop(vm, "migrated");
5343
        event = virDomainEventLifecycleNewFromObj(vm,
5344 5345
                                         VIR_DOMAIN_EVENT_STOPPED,
                                         VIR_DOMAIN_EVENT_STOPPED_MIGRATED);
5346 5347
    }

5348
 endjob:
5349 5350 5351
    if (ret < 0)
        orig_err = virSaveLastError();

5352
    if (qemuMigrationRestoreDomainState(conn, vm)) {
5353
        event = virDomainEventLifecycleNewFromObj(vm,
5354 5355 5356
                                         VIR_DOMAIN_EVENT_RESUMED,
                                         VIR_DOMAIN_EVENT_RESUMED_MIGRATED);
    }
5357

5358 5359 5360 5361
    qemuMigrationJobFinish(driver, vm);
    if (!virDomainObjIsActive(vm) &&
        (!vm->persistent ||
         (ret == 0 && (flags & VIR_MIGRATE_UNDEFINE_SOURCE)))) {
5362
        if (flags & VIR_MIGRATE_UNDEFINE_SOURCE)
5363
            virDomainDeleteConfig(cfg->configDir, cfg->autostartDir, vm);
5364
        qemuDomainRemoveInactive(driver, vm);
5365 5366
    }

5367 5368 5369 5370 5371
    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }

5372
 cleanup:
M
Michal Privoznik 已提交
5373
    virDomainObjEndAPI(&vm);
5374
    qemuDomainEventQueue(driver, event);
5375
    virObjectUnref(cfg);
5376 5377 5378 5379 5380 5381 5382
    return ret;
}

/*
 * This implements perform phase of v3 migration protocol.
 */
static int
5383
qemuMigrationPerformPhase(virQEMUDriverPtr driver,
5384 5385 5386
                          virConnectPtr conn,
                          virDomainObjPtr vm,
                          const char *uri,
5387
                          const char *graphicsuri,
5388 5389
                          size_t nmigrate_disks,
                          const char **migrate_disks,
5390 5391 5392 5393 5394 5395 5396
                          const char *cookiein,
                          int cookieinlen,
                          char **cookieout,
                          int *cookieoutlen,
                          unsigned long flags,
                          unsigned long resource)
{
5397
    virObjectEventPtr event = NULL;
5398 5399 5400 5401 5402 5403 5404 5405 5406 5407 5408
    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);
5409 5410
    virCloseCallbacksUnset(driver->closeCallbacks, vm,
                           qemuMigrationCleanup);
5411 5412 5413

    ret = doNativeMigrate(driver, vm, uri, cookiein, cookieinlen,
                          cookieout, cookieoutlen,
5414 5415
                          flags, resource, NULL, graphicsuri,
                          nmigrate_disks, migrate_disks);
5416

5417 5418 5419 5420 5421
    if (ret < 0) {
        if (qemuMigrationRestoreDomainState(conn, vm)) {
            event = virDomainEventLifecycleNewFromObj(vm,
                                                      VIR_DOMAIN_EVENT_RESUMED,
                                                      VIR_DOMAIN_EVENT_RESUMED_MIGRATED);
5422
        }
5423
        goto endjob;
5424
    }
5425 5426 5427

    qemuMigrationJobSetPhase(driver, vm, QEMU_MIGRATION_PHASE_PERFORM3_DONE);

5428 5429
    if (virCloseCallbacksSet(driver->closeCallbacks, vm, conn,
                             qemuMigrationCleanup) < 0)
5430 5431
        goto endjob;

5432
 endjob:
5433
    if (ret < 0)
5434
        qemuMigrationJobFinish(driver, vm);
5435
    else
5436 5437
        qemuMigrationJobContinue(vm);
    if (!virDomainObjIsActive(vm) && !vm->persistent)
5438
        qemuDomainRemoveInactive(driver, vm);
5439

5440
 cleanup:
M
Michal Privoznik 已提交
5441
    virDomainObjEndAPI(&vm);
5442
    qemuDomainEventQueue(driver, event);
5443 5444 5445
    return ret;
}

5446
int
5447
qemuMigrationPerform(virQEMUDriverPtr driver,
5448 5449 5450 5451 5452
                     virConnectPtr conn,
                     virDomainObjPtr vm,
                     const char *xmlin,
                     const char *dconnuri,
                     const char *uri,
5453
                     const char *graphicsuri,
5454
                     const char *listenAddress,
5455 5456
                     size_t nmigrate_disks,
                     const char **migrate_disks,
5457 5458 5459 5460 5461 5462 5463 5464 5465 5466
                     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, "
5467 5468
              "uri=%s, graphicsuri=%s, listenAddress=%s, "
              "nmigrate_disks=%zu, migrate_disks=%p, "
5469 5470
              "cookiein=%s, cookieinlen=%d, cookieout=%p, cookieoutlen=%p, "
              "flags=%lx, dname=%s, resource=%lu, v3proto=%d",
5471
              driver, conn, vm, NULLSTR(xmlin), NULLSTR(dconnuri),
5472
              NULLSTR(uri), NULLSTR(graphicsuri), NULLSTR(listenAddress),
5473 5474
              nmigrate_disks, migrate_disks, NULLSTR(cookiein), cookieinlen,
              cookieout, cookieoutlen, flags, NULLSTR(dname), resource, v3proto);
5475 5476 5477

    if ((flags & (VIR_MIGRATE_TUNNELLED | VIR_MIGRATE_PEER2PEER))) {
        if (cookieinlen) {
5478 5479
            virReportError(VIR_ERR_OPERATION_INVALID,
                           "%s", _("received unexpected cookie with P2P migration"));
5480 5481 5482 5483
            return -1;
        }

        return qemuMigrationPerformJob(driver, conn, vm, xmlin, dconnuri, uri,
5484
                                       graphicsuri, listenAddress,
5485
                                       nmigrate_disks, migrate_disks,
5486
                                       cookiein, cookieinlen,
5487 5488
                                       cookieout, cookieoutlen,
                                       flags, dname, resource, v3proto);
5489 5490
    } else {
        if (dconnuri) {
5491 5492
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           "%s", _("Unexpected dconnuri parameter with non-peer2peer migration"));
5493 5494 5495 5496 5497
            return -1;
        }

        if (v3proto) {
            return qemuMigrationPerformPhase(driver, conn, vm, uri,
5498
                                             graphicsuri,
5499
                                             nmigrate_disks, migrate_disks,
5500 5501
                                             cookiein, cookieinlen,
                                             cookieout, cookieoutlen,
5502
                                             flags, resource);
5503 5504
        } else {
            return qemuMigrationPerformJob(driver, conn, vm, xmlin, dconnuri,
5505
                                           uri, graphicsuri, listenAddress,
5506
                                           nmigrate_disks, migrate_disks,
5507
                                           cookiein, cookieinlen,
5508 5509 5510 5511 5512
                                           cookieout, cookieoutlen, flags,
                                           dname, resource, v3proto);
        }
    }
}
5513

5514
static int
5515 5516
qemuMigrationVPAssociatePortProfiles(virDomainDefPtr def)
{
5517
    size_t i;
5518 5519 5520 5521 5522
    int last_good_net = -1;
    virDomainNetDefPtr net;

    for (i = 0; i < def->nnets; i++) {
        net = def->nets[i];
5523
        if (virDomainNetGetActualType(net) == VIR_DOMAIN_NET_TYPE_DIRECT) {
5524
            if (virNetDevVPortProfileAssociate(net->ifname,
5525
                                               virDomainNetGetActualVirtPortProfile(net),
5526
                                               &net->mac,
5527
                                               virDomainNetGetActualDirectDev(net),
5528
                                               -1,
5529
                                               def->uuid,
5530 5531
                                               VIR_NETDEV_VPORT_PROFILE_OP_MIGRATE_IN_FINISH,
                                               false) < 0) {
5532 5533 5534
                virReportError(VIR_ERR_OPERATION_FAILED,
                               _("Port profile Associate failed for %s"),
                               net->ifname);
5535
                goto err_exit;
5536
            }
5537
            last_good_net = i;
5538
            VIR_DEBUG("Port profile Associate succeeded for %s", net->ifname);
5539

5540
            if (virNetDevMacVLanVPortProfileRegisterCallback(net->ifname, &net->mac,
5541 5542 5543 5544
                                                             virDomainNetGetActualDirectDev(net), def->uuid,
                                                             virDomainNetGetActualVirtPortProfile(net),
                                                             VIR_NETDEV_VPORT_PROFILE_OP_CREATE))
                goto err_exit;
5545 5546 5547
        }
    }

5548
    return 0;
5549

5550
 err_exit:
5551
    for (i = 0; last_good_net != -1 && i <= last_good_net; i++) {
5552
        net = def->nets[i];
5553
        if (virDomainNetGetActualType(net) == VIR_DOMAIN_NET_TYPE_DIRECT) {
5554
            ignore_value(virNetDevVPortProfileDisassociate(net->ifname,
5555
                                                           virDomainNetGetActualVirtPortProfile(net),
5556
                                                           &net->mac,
5557
                                                           virDomainNetGetActualDirectDev(net),
5558
                                                           -1,
5559
                                                           VIR_NETDEV_VPORT_PROFILE_OP_MIGRATE_IN_FINISH));
5560 5561
        }
    }
5562
    return -1;
5563 5564 5565
}


J
Jiri Denemark 已提交
5566 5567 5568
static int
qemuMigrationPersist(virQEMUDriverPtr driver,
                     virDomainObjPtr vm,
5569 5570
                     qemuMigrationCookiePtr mig,
                     bool ignoreSaveError)
J
Jiri Denemark 已提交
5571 5572 5573 5574
{
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
    virCapsPtr caps = NULL;
    virDomainDefPtr vmdef;
5575 5576
    virDomainDefPtr oldDef = NULL;
    unsigned int oldPersist = vm->persistent;
J
Jiri Denemark 已提交
5577 5578 5579 5580 5581 5582 5583
    virObjectEventPtr event;
    int ret = -1;

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

    vm->persistent = 1;
5584 5585
    oldDef = vm->newDef;
    vm->newDef = qemuMigrationCookieGetPersistent(mig);
J
Jiri Denemark 已提交
5586

5587 5588
    if (!(vmdef = virDomainObjGetPersistentDef(caps, driver->xmlopt, vm)))
        goto error;
J
Jiri Denemark 已提交
5589

5590 5591
    if (virDomainSaveConfig(cfg->configDir, vmdef) < 0 && !ignoreSaveError)
        goto error;
J
Jiri Denemark 已提交
5592 5593 5594

    event = virDomainEventLifecycleNewFromObj(vm,
                                              VIR_DOMAIN_EVENT_DEFINED,
5595 5596 5597
                                              oldPersist ?
                                              VIR_DOMAIN_EVENT_DEFINED_UPDATED :
                                              VIR_DOMAIN_EVENT_DEFINED_ADDED);
5598
    qemuDomainEventQueue(driver, event);
J
Jiri Denemark 已提交
5599 5600 5601 5602

    ret = 0;

 cleanup:
5603
    virDomainDefFree(oldDef);
J
Jiri Denemark 已提交
5604 5605 5606
    virObjectUnref(caps);
    virObjectUnref(cfg);
    return ret;
5607 5608 5609 5610 5611 5612 5613

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


5617
virDomainPtr
5618
qemuMigrationFinish(virQEMUDriverPtr driver,
5619 5620
                    virConnectPtr dconn,
                    virDomainObjPtr vm,
5621 5622 5623 5624
                    const char *cookiein,
                    int cookieinlen,
                    char **cookieout,
                    int *cookieoutlen,
5625
                    unsigned long flags,
5626 5627
                    int retcode,
                    bool v3proto)
5628 5629
{
    virDomainPtr dom = NULL;
5630
    qemuMigrationCookiePtr mig = NULL;
5631
    virErrorPtr orig_err = NULL;
5632
    int cookie_flags = 0;
J
Jiri Denemark 已提交
5633
    qemuDomainObjPrivatePtr priv = vm->privateData;
5634
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
5635
    unsigned short port;
5636

5637
    VIR_DEBUG("driver=%p, dconn=%p, vm=%p, cookiein=%s, cookieinlen=%d, "
5638
              "cookieout=%p, cookieoutlen=%p, flags=%lx, retcode=%d",
5639 5640
              driver, dconn, vm, NULLSTR(cookiein), cookieinlen,
              cookieout, cookieoutlen, flags, retcode);
5641

5642 5643 5644
    port = priv->migrationPort;
    priv->migrationPort = 0;

5645 5646
    if (!qemuMigrationJobIsActive(vm, QEMU_ASYNC_JOB_MIGRATION_IN)) {
        qemuMigrationErrorReport(driver, vm->def->name);
5647
        goto cleanup;
5648
    }
5649

5650 5651 5652
    qemuMigrationJobStartPhase(driver, vm,
                               v3proto ? QEMU_MIGRATION_PHASE_FINISH3
                                       : QEMU_MIGRATION_PHASE_FINISH2);
5653

5654
    qemuDomainCleanupRemove(vm, qemuMigrationPrepareCleanup);
5655
    VIR_FREE(priv->job.completed);
5656

5657
    cookie_flags = QEMU_MIGRATION_COOKIE_NETWORK |
5658 5659
                   QEMU_MIGRATION_COOKIE_STATS |
                   QEMU_MIGRATION_COOKIE_NBD;
5660 5661 5662 5663 5664
    if (flags & VIR_MIGRATE_PERSIST_DEST)
        cookie_flags |= QEMU_MIGRATION_COOKIE_PERSISTENT;

    if (!(mig = qemuMigrationEatCookie(driver, vm, cookiein,
                                       cookieinlen, cookie_flags)))
5665
        goto endjob;
5666 5667 5668 5669

    /* Did the migration go as planned?  If yes, return the domain
     * object, but if no, clean up the empty qemu process.
     */
5670 5671
    if (flags & VIR_MIGRATE_OFFLINE) {
        if (retcode != 0 ||
5672
            qemuMigrationPersist(driver, vm, mig, false) < 0)
5673 5674 5675 5676 5677
            goto endjob;

        dom = virGetDomain(dconn, vm->def->name, vm->def->uuid);
    } else if (retcode == 0) {
        if (!virDomainObjIsActive(vm)) {
5678 5679
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("guest unexpectedly quit"));
5680
            qemuMigrationErrorReport(driver, vm->def->name);
5681
            goto endjob;
5682 5683
        }

5684
        if (mig->jobInfo) {
5685 5686
            qemuDomainJobInfoPtr jobInfo = mig->jobInfo;
            priv->job.completed = jobInfo;
5687
            mig->jobInfo = NULL;
5688 5689 5690 5691
            if (jobInfo->sent && virTimeMillisNow(&jobInfo->received) == 0) {
                jobInfo->timeDelta = jobInfo->received - jobInfo->sent;
                jobInfo->timeDeltaSet = true;
            }
5692 5693
        }

5694
        if (qemuMigrationVPAssociatePortProfiles(vm->def) < 0)
5695
            goto endjob;
5696

5697 5698
        if (mig->network && qemuDomainMigrateOPDRelocate(driver, vm, mig) < 0)
            VIR_WARN("unable to provide network data for relocation");
5699

5700 5701
        if (qemuMigrationStopNBDServer(driver, vm, mig) < 0)
            goto endjob;
5702

5703
        if (flags & VIR_MIGRATE_PERSIST_DEST) {
5704
            if (qemuMigrationPersist(driver, vm, mig, !v3proto) < 0) {
5705 5706 5707 5708 5709 5710
                /* 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.
5711 5712 5713
                 * Pretend success and hope that this is a rare situation and
                 * management tools are smart.
                 *
5714 5715
                 * However, in v3 protocol, the source VM is still available
                 * to restart during confirm() step, so we kill it off now.
5716
                 */
5717 5718
                if (v3proto)
                    goto endjob;
5719 5720 5721
            }
        }

5722
        if (!(flags & VIR_MIGRATE_PAUSED)) {
5723 5724 5725 5726
            /* 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
             */
J
Jiri Denemark 已提交
5727
            if (qemuProcessStartCPUs(driver, vm, dconn,
5728 5729
                                     VIR_DOMAIN_RUNNING_MIGRATED,
                                     QEMU_ASYNC_JOB_MIGRATION_IN) < 0) {
5730
                if (virGetLastError() == NULL)
5731 5732
                    virReportError(VIR_ERR_INTERNAL_ERROR,
                                   "%s", _("resume operation failed"));
5733 5734 5735 5736 5737
                /* Need to save the current error, in case shutting
                 * down the process overwrites it
                 */
                orig_err = virSaveLastError();

5738 5739 5740 5741 5742 5743 5744 5745
                /*
                 * 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
                 */
5746 5747
                if (v3proto)
                    goto endjob;
5748
            }
5749 5750 5751 5752
            if (priv->job.completed) {
                qemuDomainJobInfoUpdateTime(priv->job.completed);
                qemuDomainJobInfoUpdateDowntime(priv->job.completed);
            }
5753 5754
        }

5755
        dom = virGetDomain(dconn, vm->def->name, vm->def->uuid);
5756

5757 5758 5759 5760
        qemuDomainEventQueue(driver,
                             virDomainEventLifecycleNewFromObj(vm,
                                    VIR_DOMAIN_EVENT_RESUMED,
                                    VIR_DOMAIN_EVENT_RESUMED_MIGRATED));
5761 5762 5763
        if (virDomainObjGetState(vm, NULL) == VIR_DOMAIN_PAUSED) {
            virDomainObjSetState(vm, VIR_DOMAIN_PAUSED,
                                 VIR_DOMAIN_PAUSED_USER);
5764 5765 5766 5767
            qemuDomainEventQueue(driver,
                                 virDomainEventLifecycleNewFromObj(vm,
                                        VIR_DOMAIN_EVENT_SUSPENDED,
                                        VIR_DOMAIN_EVENT_SUSPENDED_PAUSED));
5768
        }
L
liguang 已提交
5769 5770

        if (virDomainObjIsActive(vm) &&
5771
            virDomainSaveStatus(driver->xmlopt, cfg->stateDir, vm) < 0)
5772
            VIR_WARN("Failed to save status on vm %s", vm->def->name);
5773 5774 5775

        /* Guest is successfully running, so cancel previous auto destroy */
        qemuProcessAutoDestroyRemove(driver, vm);
5776
    } else {
5777 5778 5779 5780 5781 5782 5783 5784
        qemuDomainJobInfo info;

        /* 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
         */
        qemuMigrationFetchJobStatus(driver, vm,
                                    QEMU_ASYNC_JOB_MIGRATION_IN,
                                    &info);
5785
    }
5786

5787
 endjob:
5788
    if (!dom &&
5789 5790
        !(flags & VIR_MIGRATE_OFFLINE) &&
        virDomainObjIsActive(vm)) {
5791 5792
        qemuProcessStop(driver, vm, VIR_DOMAIN_SHUTOFF_FAILED,
                        VIR_QEMU_PROCESS_STOP_MIGRATED);
5793
        virDomainAuditStop(vm, "failed");
5794 5795 5796 5797
        qemuDomainEventQueue(driver,
                             virDomainEventLifecycleNewFromObj(vm,
                                    VIR_DOMAIN_EVENT_STOPPED,
                                    VIR_DOMAIN_EVENT_STOPPED_FAILED));
5798 5799
    }

5800 5801
    if (dom &&
        qemuMigrationBakeCookie(mig, driver, vm, cookieout, cookieoutlen,
5802
                                QEMU_MIGRATION_COOKIE_STATS) < 0)
5803 5804
        VIR_WARN("Unable to encode migration cookie");

5805 5806
    qemuMigrationJobFinish(driver, vm);
    if (!vm->persistent && !virDomainObjIsActive(vm))
5807
        qemuDomainRemoveInactive(driver, vm);
5808

5809
 cleanup:
5810
    virPortAllocatorRelease(driver->migrationPorts, port);
5811 5812 5813
    if (priv->mon)
        qemuMonitorSetDomainLog(priv->mon, -1);
    VIR_FREE(priv->origname);
M
Michal Privoznik 已提交
5814
    virDomainObjEndAPI(&vm);
5815 5816 5817 5818
    if (mig) {
        virDomainDefFree(qemuMigrationCookieGetPersistent(mig));
        qemuMigrationCookieFree(mig);
    }
5819 5820 5821 5822
    if (orig_err) {
        virSetError(orig_err);
        virFreeError(orig_err);
    }
5823
    virObjectUnref(cfg);
5824 5825 5826 5827 5828 5829

    /* 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);
5830 5831
    return dom;
}
5832

5833

5834
/* Helper function called while vm is active.  */
5835
int
5836
qemuMigrationToFile(virQEMUDriverPtr driver, virDomainObjPtr vm,
5837 5838
                    int fd, off_t offset, const char *path,
                    const char *compressor,
E
Eric Blake 已提交
5839
                    bool bypassSecurityDriver,
5840
                    qemuDomainAsyncJob asyncJob)
5841 5842 5843
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int rc;
5844
    int ret = -1;
5845
    bool restoreLabel = false;
5846 5847
    virCommandPtr cmd = NULL;
    int pipeFD[2] = { -1, -1 };
5848
    unsigned long saveMigBandwidth = priv->migMaxBandwidth;
5849
    char *errbuf = NULL;
5850
    virErrorPtr orig_err = NULL;
5851 5852 5853 5854 5855

    /* 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,
5856 5857
                                     QEMU_DOMAIN_MIG_BANDWIDTH_MAX);
        priv->migMaxBandwidth = QEMU_DOMAIN_MIG_BANDWIDTH_MAX;
5858 5859
        if (qemuDomainObjExitMonitor(driver, vm) < 0)
            return -1;
5860
    }
5861

5862 5863 5864 5865 5866 5867 5868
    if (!virDomainObjIsActive(vm)) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest unexpectedly quit"));
        /* nothing to tear down */
        return -1;
    }

5869
    if (virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATE_QEMU_FD) &&
5870
        (!compressor || pipe(pipeFD) == 0)) {
5871
        /* All right! We can use fd migration, which means that qemu
5872 5873 5874
         * 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.  */
5875
        if (virSecurityManagerSetImageFDLabel(driver->securityManager, vm->def,
5876
                                              compressor ? pipeFD[1] : fd) < 0)
5877
            goto cleanup;
5878 5879 5880
        bypassSecurityDriver = true;
    } else {
        /* Phooey - we have to fall back on exec migration, where qemu
E
Eric Blake 已提交
5881 5882
         * has to popen() the file by name, and block devices have to be
         * given cgroup ACL permission.  We might also stumble on
5883 5884
         * a race present in some qemu versions where it does a wait()
         * that botches pclose.  */
5885 5886
        if (virCgroupHasController(priv->cgroup,
                                   VIR_CGROUP_CONTROLLER_DEVICES)) {
5887 5888 5889 5890
            int rv = virCgroupAllowDevicePath(priv->cgroup, path,
                                              VIR_CGROUP_DEVICE_RW);
            virDomainAuditCgroupPath(vm, priv->cgroup, "allow", path, "rw", rv == 0);
            if (rv == 1) {
E
Eric Blake 已提交
5891
                /* path was not a device, no further need for cgroup */
5892
            } else if (rv < 0) {
5893 5894
                goto cleanup;
            }
5895
        }
5896 5897
        if ((!bypassSecurityDriver) &&
            virSecurityManagerSetSavedStateLabel(driver->securityManager,
5898
                                                 vm->def, path) < 0)
5899
            goto cleanup;
5900
        restoreLabel = true;
5901 5902
    }

5903
    if (qemuDomainObjEnterMonitorAsync(driver, vm, asyncJob) < 0)
5904 5905
        goto cleanup;

5906 5907 5908
    if (!compressor) {
        const char *args[] = { "cat", NULL };

5909
        if (virQEMUCapsGet(priv->qemuCaps, QEMU_CAPS_MIGRATE_QEMU_FD) &&
5910 5911 5912 5913 5914 5915 5916 5917 5918 5919 5920 5921 5922 5923 5924 5925
            priv->monConfig->type == VIR_DOMAIN_CHR_TYPE_UNIX) {
            rc = qemuMonitorMigrateToFd(priv->mon,
                                        QEMU_MONITOR_MIGRATE_BACKGROUND,
                                        fd);
        } else {
            rc = qemuMonitorMigrateToFile(priv->mon,
                                          QEMU_MONITOR_MIGRATE_BACKGROUND,
                                          args, path, offset);
        }
    } else {
        const char *prog = compressor;
        const char *args[] = {
            prog,
            "-c",
            NULL
        };
5926 5927 5928 5929
        if (pipeFD[0] != -1) {
            cmd = virCommandNewArgs(args);
            virCommandSetInputFD(cmd, pipeFD[0]);
            virCommandSetOutputFD(cmd, &fd);
5930 5931
            virCommandSetErrorBuffer(cmd, &errbuf);
            virCommandDoAsyncIO(cmd);
5932 5933 5934
            if (virSetCloseExec(pipeFD[1]) < 0) {
                virReportSystemError(errno, "%s",
                                     _("Unable to set cloexec flag"));
5935
                ignore_value(qemuDomainObjExitMonitor(driver, vm));
5936 5937 5938
                goto cleanup;
            }
            if (virCommandRunAsync(cmd, NULL) < 0) {
5939
                ignore_value(qemuDomainObjExitMonitor(driver, vm));
5940 5941 5942 5943 5944 5945 5946
                goto cleanup;
            }
            rc = qemuMonitorMigrateToFd(priv->mon,
                                        QEMU_MONITOR_MIGRATE_BACKGROUND,
                                        pipeFD[1]);
            if (VIR_CLOSE(pipeFD[0]) < 0 ||
                VIR_CLOSE(pipeFD[1]) < 0)
5947
                VIR_WARN("failed to close intermediate pipe");
5948 5949 5950 5951 5952
        } else {
            rc = qemuMonitorMigrateToFile(priv->mon,
                                          QEMU_MONITOR_MIGRATE_BACKGROUND,
                                          args, path, offset);
        }
5953
    }
5954
    if (qemuDomainObjExitMonitor(driver, vm) < 0)
5955
        goto cleanup;
5956 5957 5958
    if (rc < 0)
        goto cleanup;

5959 5960
    rc = qemuMigrationWaitForCompletion(driver, vm, asyncJob,
                                        NULL, false, false);
5961

5962 5963 5964 5965
    if (rc < 0) {
        if (rc == -2) {
            orig_err = virSaveLastError();
            virCommandAbort(cmd);
5966 5967
            if (virDomainObjIsActive(vm) &&
                qemuDomainObjEnterMonitorAsync(driver, vm, asyncJob) == 0) {
5968
                qemuMonitorMigrateCancel(priv->mon);
5969
                ignore_value(qemuDomainObjExitMonitor(driver, vm));
5970 5971
            }
        }
5972
        goto cleanup;
5973
    }
5974

5975 5976 5977
    if (cmd && virCommandWait(cmd, NULL) < 0)
        goto cleanup;

5978 5979
    ret = 0;

5980
 cleanup:
5981 5982 5983
    if (ret < 0 && !orig_err)
        orig_err = virSaveLastError();

5984
    /* Restore max migration bandwidth */
5985 5986
    if (virDomainObjIsActive(vm) &&
        qemuDomainObjEnterMonitorAsync(driver, vm, asyncJob) == 0) {
5987 5988
        qemuMonitorSetMigrationSpeed(priv->mon, saveMigBandwidth);
        priv->migMaxBandwidth = saveMigBandwidth;
5989
        ignore_value(qemuDomainObjExitMonitor(driver, vm));
5990 5991
    }

5992 5993
    VIR_FORCE_CLOSE(pipeFD[0]);
    VIR_FORCE_CLOSE(pipeFD[1]);
5994 5995 5996 5997 5998
    if (cmd) {
        VIR_DEBUG("Compression binary stderr: %s", NULLSTR(errbuf));
        VIR_FREE(errbuf);
        virCommandFree(cmd);
    }
5999 6000
    if (restoreLabel && (!bypassSecurityDriver) &&
        virSecurityManagerRestoreSavedStateLabel(driver->securityManager,
6001
                                                 vm->def, path) < 0)
6002 6003
        VIR_WARN("failed to restore save state label on %s", path);

6004 6005
    if (virCgroupHasController(priv->cgroup,
                               VIR_CGROUP_CONTROLLER_DEVICES)) {
6006 6007 6008
        int rv = virCgroupDenyDevicePath(priv->cgroup, path,
                                         VIR_CGROUP_DEVICE_RWM);
        virDomainAuditCgroupPath(vm, priv->cgroup, "deny", path, "rwm", rv == 0);
6009
    }
6010 6011 6012 6013 6014 6015

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

6016 6017
    return ret;
}
6018

6019 6020 6021 6022 6023 6024 6025 6026 6027 6028 6029 6030 6031 6032 6033 6034 6035 6036 6037 6038 6039 6040 6041 6042 6043 6044 6045 6046 6047 6048 6049 6050 6051 6052 6053 6054 6055 6056 6057 6058 6059 6060 6061 6062 6063 6064 6065 6066 6067 6068 6069 6070 6071

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);
            qemuBlockJobSyncEnd(driver, vm, disk);
            diskPriv->migrating = false;
        }
    }

    if (qemuMigrationCancelDriveMirror(driver, vm, false,
6072
                                       QEMU_ASYNC_JOB_NONE, NULL) < 0)
6073 6074 6075 6076 6077 6078 6079 6080 6081 6082 6083 6084 6085 6086 6087 6088 6089 6090 6091 6092 6093 6094 6095 6096
        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) {
                qemuBlockJobSyncEnd(driver, vm, disk);
                diskPriv->migrating = false;
            }
        }
    }
    goto cleanup;
}


6097
int
6098
qemuMigrationJobStart(virQEMUDriverPtr driver,
6099
                      virDomainObjPtr vm,
6100
                      qemuDomainAsyncJob job)
6101 6102 6103
{
    qemuDomainObjPrivatePtr priv = vm->privateData;

6104
    if (qemuDomainObjBeginAsyncJob(driver, vm, job) < 0)
6105 6106
        return -1;

6107
    if (job == QEMU_ASYNC_JOB_MIGRATION_IN) {
6108
        qemuDomainObjSetAsyncJobMask(vm, QEMU_JOB_NONE);
6109
    } else {
6110 6111 6112
        qemuDomainObjSetAsyncJobMask(vm, (QEMU_JOB_DEFAULT_MASK |
                                          JOB_MASK(QEMU_JOB_SUSPEND) |
                                          JOB_MASK(QEMU_JOB_MIGRATION_OP)));
6113
    }
6114

J
Jiri Denemark 已提交
6115
    priv->job.current->type = VIR_DOMAIN_JOB_UNBOUNDED;
6116 6117 6118 6119 6120

    return 0;
}

void
6121
qemuMigrationJobSetPhase(virQEMUDriverPtr driver,
6122
                         virDomainObjPtr vm,
6123
                         qemuMigrationJobPhase phase)
6124 6125 6126 6127 6128 6129 6130 6131 6132 6133 6134 6135 6136 6137
{
    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
6138
qemuMigrationJobStartPhase(virQEMUDriverPtr driver,
6139
                           virDomainObjPtr vm,
6140
                           qemuMigrationJobPhase phase)
6141 6142 6143 6144
{
    qemuMigrationJobSetPhase(driver, vm, phase);
}

6145
void
6146 6147
qemuMigrationJobContinue(virDomainObjPtr vm)
{
6148
    qemuDomainObjReleaseAsyncJob(vm);
6149 6150 6151 6152
}

bool
qemuMigrationJobIsActive(virDomainObjPtr vm,
6153
                         qemuDomainAsyncJob job)
6154 6155 6156 6157 6158 6159 6160 6161 6162 6163 6164
{
    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");

6165
        virReportError(VIR_ERR_OPERATION_INVALID, msg, vm->def->name);
6166 6167 6168 6169 6170
        return false;
    }
    return true;
}

6171
void
6172
qemuMigrationJobFinish(virQEMUDriverPtr driver, virDomainObjPtr vm)
6173
{
6174
    qemuDomainObjEndAsyncJob(driver, vm);
6175
}
6176 6177 6178 6179 6180 6181 6182 6183 6184 6185 6186 6187 6188 6189 6190 6191 6192 6193 6194 6195 6196 6197 6198 6199 6200 6201 6202 6203 6204 6205 6206 6207 6208 6209 6210 6211 6212 6213 6214 6215 6216 6217 6218 6219 6220 6221 6222 6223 6224 6225 6226 6227 6228 6229


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