qemu_monitor_text.c 89.6 KB
Newer Older
1 2 3
/*
 * qemu_monitor_text.c: interaction with QEMU monitor console
 *
E
Eric Blake 已提交
4
 * Copyright (C) 2006-2012 Red Hat, Inc.
5 6 7 8 9 10 11 12 13 14 15 16 17
 * Copyright (C) 2006 Daniel P. Berrange
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
O
Osier Yang 已提交
18 19
 * License along with this library;  If not, see
 * <http://www.gnu.org/licenses/>.
20 21 22 23 24 25 26 27 28 29 30
 *
 * Author: Daniel P. Berrange <berrange@redhat.com>
 */

#include <config.h>

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <poll.h>
#include <unistd.h>
31
#include <string.h>
32 33

#include "qemu_monitor_text.h"
34
#include "qemu_command.h"
35
#include "c-ctype.h"
36
#include "c-strcasestr.h"
37 38 39 40 41
#include "memory.h"
#include "logging.h"
#include "driver.h"
#include "datatypes.h"
#include "virterror_internal.h"
42
#include "buf.h"
43

44 45 46 47
#ifdef WITH_DTRACE_PROBES
# include "libvirt_qemu_probes.h"
#endif

48 49
#define VIR_FROM_THIS VIR_FROM_QEMU

50 51 52
#define QEMU_CMD_PROMPT "\n(qemu) "
#define QEMU_PASSWD_PROMPT "Password: "

53 54
#define DEBUG_IO 0

55
/* Return -1 for error, 0 for success */
56
typedef int qemuMonitorExtraPromptHandler(qemuMonitorPtr mon,
57 58 59 60 61
                                          const char *buf,
                                          const char *prompt,
                                          void *data);


62 63 64 65 66
/* When connecting to a monitor, QEMU will print a greeting like
 *
 * QEMU 0.11.0 monitor - type 'help' for more information
 *
 * Don't expect the version number bit to be stable :-)
67
 */
68 69 70 71 72 73 74 75 76 77
#define GREETING_PREFIX "QEMU "
#define GREETING_POSTFIX "type 'help' for more information\r\n(qemu) "
#define BASIC_PROMPT "(qemu) "
#define PASSWORD_PROMPT "Password:"
#define DISK_ENCRYPTION_PREFIX "("
#define DISK_ENCRYPTION_POSTFIX ") is encrypted."
#define LINE_ENDING "\r\n"

int qemuMonitorTextIOProcess(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
                             const char *data,
78
                             size_t len ATTRIBUTE_UNUSED,
79 80 81
                             qemuMonitorMessagePtr msg)
{
    int used = 0;
82

83 84 85
    /* Check for & discard greeting */
    if (STRPREFIX(data, GREETING_PREFIX)) {
        const char *offset = strstr(data, GREETING_POSTFIX);
86

87 88 89
        /* We see the greeting prefix, but not postfix, so pretend we've
           not consumed anything. We'll restart when more data arrives. */
        if (!offset) {
90
#if DEBUG_IO
91
            VIR_DEBUG("Partial greeting seen, getting out & waiting for more");
92
#endif
93 94
            return 0;
        }
95

96 97
        used = offset - data + strlen(GREETING_POSTFIX);

98
#if DEBUG_IO
99
        VIR_DEBUG("Discarded monitor greeting");
100
#endif
101
    }
102

103 104
    /* Don't print raw data in debug because its full of control chars */
    /*VIR_DEBUG("Process data %d byts of data [%s]", len - used, data + used);*/
105
#if DEBUG_IO
106
    VIR_DEBUG("Process data %d byts of data", (int)(len - used));
107
#endif
108 109 110

    /* Look for a non-zero reply followed by prompt */
    if (msg && !msg->finished) {
111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145
        char *start = NULL;
        char *end = NULL;
        char *skip;

        /* If we're here, we've already sent the command. We now
         * strip the trailing '\r' because it makes the matching
         * code that follows a little easier ie we can just strstr()
         * for the original command
         */
        if (msg->txLength > 0) {
            char *tmp;
            if ((tmp = strchr(msg->txBuffer, '\r'))) {
                *tmp = '\0';
            }
        }

        /* QEMU echos the command back to us, full of control
         * character junk that we don't want. We have to skip
         * over this junk by looking for the first complete
         * repetition of our command. Then we can look for
         * the prompt that is supposed to follow
         *
         * NB, we can't optimize by immediately looking for
         * LINE_ENDING, because QEMU 0.10 has bad problems
         * when initially connecting where it might write a
         * prompt in the wrong place. So we must not look
         * for LINE_ENDING, or BASIC_PROMPT until we've
         * seen our original command echod.
         */
        skip = strstr(data + used, msg->txBuffer);

        /* After the junk we should have a line ending... */
        if (skip) {
            start = strstr(skip + strlen(msg->txBuffer), LINE_ENDING);
        }
146

147 148 149 150 151 152 153 154
        /* ... then our command reply data, following by a (qemu) prompt */
        if (start) {
            char *passwd;
            start += strlen(LINE_ENDING);

            /* We might get a prompt for a password before the (qemu) prompt */
            passwd = strstr(start, PASSWORD_PROMPT);
            if (passwd) {
155
#if DEBUG_IO
156
                VIR_DEBUG("Seen a password prompt [%s]", data + used);
157
#endif
158 159
                if (msg->passwordHandler) {
                    int i;
160 161
                    /* Try and handle the prompt. The handler is required
                     * to report a normal libvirt error */
162 163 164 165 166 167 168 169 170 171 172 173 174 175
                    if (msg->passwordHandler(mon, msg,
                                             start,
                                             passwd - start + strlen(PASSWORD_PROMPT),
                                             msg->passwordOpaque) < 0)
                        return -1;

                    /* Blank out the password prompt so we don't re-trigger
                     * if we have to go back to sleep for more I/O */
                    for (i = 0 ; i < strlen(PASSWORD_PROMPT) ; i++)
                        start[i] = ' ';

                    /* Handled, so skip forward over password prompt */
                    start = passwd;
                } else {
176 177
                    virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                                   _("Password request seen, but no handler available"));
178 179
                    return -1;
                }
180
            }
181 182

            end = strstr(start, BASIC_PROMPT);
183
        }
184

185
        if (start && end) {
186 187 188 189 190 191
            int want = end - start;
            /* Annoyingly some commands may not have any reply data
             * at all upon success, but since we've detected the
             * BASIC_PROMPT we can reasonably reliably cope */
            if (want) {
                if (VIR_REALLOC_N(msg->rxBuffer,
192 193
                                  msg->rxLength + want + 1) < 0) {
                    virReportOOMError();
194
                    return -1;
195
                }
196 197 198
                memcpy(msg->rxBuffer + msg->rxLength, start, want);
                msg->rxLength += want;
                msg->rxBuffer[msg->rxLength] = '\0';
199
#if DEBUG_IO
200 201
                VIR_DEBUG("Finished %d byte reply [%s]", want, msg->rxBuffer);
            } else {
202
                VIR_DEBUG("Finished 0 byte reply");
203
#endif
204
            }
205 206 207
            PROBE(QEMU_MONITOR_RECV_REPLY,
                  "mon=%p reply=%s",
                  mon, msg->rxBuffer);
208 209 210 211 212 213
            msg->finished = 1;
            used += end - (data + used);
            used += strlen(BASIC_PROMPT);
        }
    }

214
#if DEBUG_IO
215
    VIR_DEBUG("Total used %d", used);
216
#endif
217
    return used;
218 219
}

220
static int
221 222 223 224 225 226 227
qemuMonitorTextCommandWithHandler(qemuMonitorPtr mon,
                                  const char *cmd,
                                  qemuMonitorPasswordHandler passwordHandler,
                                  void *passwordOpaque,
                                  int scm_fd,
                                  char **reply)
{
228 229
    int ret;
    qemuMonitorMessage msg;
230

231
    *reply = NULL;
232

233
    memset(&msg, 0, sizeof(msg));
234

235
    if (virAsprintf(&msg.txBuffer, "%s\r", cmd) < 0) {
236
        virReportOOMError();
237 238 239 240 241 242
        return -1;
    }
    msg.txLength = strlen(msg.txBuffer);
    msg.txFD = scm_fd;
    msg.passwordHandler = passwordHandler;
    msg.passwordOpaque = passwordOpaque;
243

244
    VIR_DEBUG("Send command '%s' for write with FD %d", cmd, scm_fd);
245

246
    ret = qemuMonitorSend(mon, &msg);
247

248 249
    VIR_DEBUG("Receive command reply ret=%d rxLength=%d rxBuffer='%s'",
              ret, msg.rxLength, msg.rxBuffer);
250

251 252 253
    /* Just in case buffer had some passwords in */
    memset(msg.txBuffer, 0, msg.txLength);
    VIR_FREE(msg.txBuffer);
254

255 256 257 258 259 260 261 262 263 264
    if (ret >= 0) {
        /* To make life safer for callers, already ensure there's at least an empty string */
        if (msg.rxBuffer) {
            *reply = msg.rxBuffer;
        } else {
            *reply = strdup("");
            if (!*reply) {
                virReportOOMError();
                return -1;
            }
265 266 267
        }
    }

268
    return ret;
269 270
}

271 272 273 274 275 276
int
qemuMonitorTextCommandWithFd(qemuMonitorPtr mon,
                             const char *cmd,
                             int scm_fd,
                             char **reply)
{
277 278
    return qemuMonitorTextCommandWithHandler(mon, cmd, NULL, NULL,
                                             scm_fd, reply);
279 280
}

281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297
/* Check monitor output for evidence that the command was not recognized.
 * For 'info' commands, qemu returns help text.  For other commands, qemu
 * returns 'unknown command:'.
 */
static int
qemuMonitorTextCommandNotFound(const char *cmd, const char *reply)
{
    if (STRPREFIX(cmd, "info ")) {
        if (strstr(reply, "info version"))
            return 1;
    } else {
        if (strstr(reply, "unknown command:"))
            return 1;
    }

    return 0;
}
298 299

static int
300 301 302 303 304
qemuMonitorSendDiskPassphrase(qemuMonitorPtr mon,
                              qemuMonitorMessagePtr msg,
                              const char *data,
                              size_t len ATTRIBUTE_UNUSED,
                              void *opaque)
305
{
306 307 308 309
    virConnectPtr conn = opaque;
    char *path;
    char *passphrase = NULL;
    size_t passphrase_len = 0;
310
    int res;
311 312
    const char *pathStart;
    const char *pathEnd;
313

314 315 316 317 318 319 320 321 322 323
    /*
     * For disk passwords:
     *
     *    ide0-hd0 (/path/to/volume) is encrypted.
     *    Password:
     *
     */
    pathStart = strstr(data, DISK_ENCRYPTION_PREFIX);
    pathEnd = strstr(data, DISK_ENCRYPTION_POSTFIX);
    if (!pathStart || !pathEnd || pathStart >= pathEnd) {
324 325 326
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unable to extract disk path from %s"),
                       data);
327
        return -1;
328 329 330 331
    }

    /* Extra the path */
    pathStart += strlen(DISK_ENCRYPTION_PREFIX);
332 333
    if (!(path = strndup(pathStart, pathEnd - pathStart))) {
        virReportOOMError();
334
        return -1;
335
    }
336

337 338 339 340 341 342
    /* Fetch the disk password if possible */
    res = qemuMonitorGetDiskSecret(mon,
                                   conn,
                                   path,
                                   &passphrase,
                                   &passphrase_len);
343
    VIR_FREE(path);
344
    if (res < 0)
345 346
        return -1;

347 348 349 350 351 352
    /* Enlarge transmit buffer to allow for the extra data
     * to be sent back */
    if (VIR_REALLOC_N(msg->txBuffer,
                      msg->txLength + passphrase_len + 1 + 1) < 0) {
        memset(passphrase, 0, passphrase_len);
        VIR_FREE(passphrase);
353
        virReportOOMError();
354 355 356 357 358 359 360 361 362 363
        return -1;
    }

    /* Queue the password for sending */
    memcpy(msg->txBuffer + msg->txLength,
           passphrase, passphrase_len);
    msg->txLength += passphrase_len;
    msg->txBuffer[msg->txLength] = '\r';
    msg->txLength++;
    msg->txBuffer[msg->txLength] = '\0';
364 365 366 367

    memset(passphrase, 0, passphrase_len);
    VIR_FREE(passphrase);

368
    return 0;
369 370 371
}

int
372 373
qemuMonitorTextStartCPUs(qemuMonitorPtr mon,
                         virConnectPtr conn) {
374 375
    char *reply;

376 377 378 379
    if (qemuMonitorTextCommandWithHandler(mon, "cont",
                                          qemuMonitorSendDiskPassphrase,
                                          conn,
                                          -1, &reply) < 0)
380
        return -1;
381

382 383 384
    VIR_FREE(reply);
    return 0;
}
385 386


387
int
388
qemuMonitorTextStopCPUs(qemuMonitorPtr mon) {
389
    char *info;
390
    int ret;
391

392
    ret = qemuMonitorHMPCommand(mon, "stop", &info);
393
    VIR_FREE(info);
394
    return ret;
395 396
}

397

398
int
399 400 401
qemuMonitorTextGetStatus(qemuMonitorPtr mon,
                         bool *running,
                         virDomainPausedReason *reason)
402 403 404 405
{
    char *reply;
    int ret = -1;

406 407 408
    if (reason)
        *reason = VIR_DOMAIN_PAUSED_UNKNOWN;

409
    if (qemuMonitorHMPCommand(mon, "info status", &reply) < 0)
410 411 412 413 414
        return -1;

    if (strstr(reply, "running")) {
        *running = true;
    } else if (strstr(reply, "paused")) {
415 416 417 418 419 420 421 422 423 424 425 426 427
        char *status;

        if ((status = strchr(reply, '('))) {
            char *end = strchr(status, ')');
            if (end)
                *end = '\0';
            else
                status = NULL;
        }
        if (!status)
            VIR_DEBUG("info status was missing status details");
        else if (reason)
            *reason = qemuMonitorVMStatusToPausedReason(status);
428 429
        *running = false;
    } else {
430 431
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("unexpected reply from info status: %s"), reply);
432 433 434 435 436 437 438 439 440 441 442
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(reply);
    return ret;
}


443
int qemuMonitorTextSystemPowerdown(qemuMonitorPtr mon) {
444
    char *info;
445 446 447
    int ret;

    ret = qemuMonitorHMPCommand(mon, "system_powerdown", &info);
448 449

    VIR_FREE(info);
450
    return ret;
451 452
}

453 454 455 456 457 458 459 460 461 462 463 464 465 466 467
int qemuMonitorTextSetLink(qemuMonitorPtr mon, const char *name, enum virDomainNetInterfaceLinkState state) {
    char *info = NULL;
    char *cmd = NULL;
    const char *st_str = NULL;

    /* determine state */
    if (state == VIR_DOMAIN_NET_INTERFACE_LINK_STATE_DOWN)
        st_str = "off";
    else
        st_str = "on";

    if (virAsprintf(&cmd, "set_link %s %s", name, st_str) < 0) {
        virReportOOMError();
        goto error;
    }
468
    if (qemuMonitorHMPCommand(mon, cmd, &info) < 0)
469 470 471 472
        goto error;

    /* check if set_link command is supported */
    if (strstr(info, "\nunknown ")) {
473 474 475
        virReportError(VIR_ERR_NO_SUPPORT,
                       "%s",
                       _("\'set_link\' not supported by this qemu"));
476 477 478 479 480
        goto error;
    }

    /* check if qemu didn't reject device name */
    if (strstr(info, "\nDevice ")) {
481 482
        virReportError(VIR_ERR_OPERATION_FAILED,
                       "%s", _("device name rejected"));
483 484 485 486 487 488 489 490 491 492 493 494 495
        goto error;
    }

    VIR_FREE(info);
    VIR_FREE(cmd);
    return 0;

error:
    VIR_FREE(info);
    VIR_FREE(cmd);

    return -1;
}
496

497 498
int qemuMonitorTextSystemReset(qemuMonitorPtr mon) {
    char *info;
499 500 501
    int ret;

    ret = qemuMonitorHMPCommand(mon, "system_reset", &info);
502 503

    VIR_FREE(info);
504
    return ret;
505 506 507
}


508 509
int qemuMonitorTextGetCPUInfo(qemuMonitorPtr mon,
                              int **pids)
510 511 512 513 514 515 516
{
    char *qemucpus = NULL;
    char *line;
    int lastVcpu = -1;
    pid_t *cpupids = NULL;
    size_t ncpupids = 0;

517
    if (qemuMonitorHMPCommand(mon, "info cpus", &qemucpus) < 0)
518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560
        return -1;

    /*
     * This is the gross format we're about to parse :-{
     *
     * (qemu) info cpus
     * * CPU #0: pc=0x00000000000f0c4a thread_id=30019
     *   CPU #1: pc=0x00000000fffffff0 thread_id=30020
     *   CPU #2: pc=0x00000000fffffff0 thread_id=30021
     *
     */
    line = qemucpus;
    do {
        char *offset = strchr(line, '#');
        char *end = NULL;
        int vcpu = 0, tid = 0;

        /* See if we're all done */
        if (offset == NULL)
            break;

        /* Extract VCPU number */
        if (virStrToLong_i(offset + 1, &end, 10, &vcpu) < 0)
            goto error;

        if (end == NULL || *end != ':')
            goto error;

        /* Extract host Thread ID */
        if ((offset = strstr(line, "thread_id=")) == NULL)
            goto error;

        if (virStrToLong_i(offset + strlen("thread_id="), &end, 10, &tid) < 0)
            goto error;
        if (end == NULL || !c_isspace(*end))
            goto error;

        if (vcpu != (lastVcpu + 1))
            goto error;

        if (VIR_REALLOC_N(cpupids, ncpupids+1) < 0)
            goto error;

561
        VIR_DEBUG("vcpu=%d pid=%d", vcpu, tid);
562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586
        cpupids[ncpupids++] = tid;
        lastVcpu = vcpu;

        /* Skip to next data line */
        line = strchr(offset, '\r');
        if (line == NULL)
            line = strchr(offset, '\n');
    } while (line != NULL);

    /* Validate we got data for all VCPUs we expected */
    VIR_FREE(qemucpus);
    *pids = cpupids;
    return ncpupids;

error:
    VIR_FREE(qemucpus);
    VIR_FREE(cpupids);

    /* Returning 0 to indicate non-fatal failure, since
     * older QEMU does not have VCPU<->PID mapping and
     * we don't want to fail on that
     */
    return 0;
}

587 588 589 590 591 592 593 594

int qemuMonitorTextGetVirtType(qemuMonitorPtr mon,
                               int *virtType)
{
    char *reply = NULL;

    *virtType = VIR_DOMAIN_VIRT_QEMU;

595
    if (qemuMonitorHMPCommand(mon, "info kvm", &reply) < 0)
596 597 598 599 600 601 602 603 604 605
        return -1;

    if (strstr(reply, "enabled"))
        *virtType = VIR_DOMAIN_VIRT_KVM;

    VIR_FREE(reply);
    return 0;
}


606 607 608 609 610 611 612 613 614
static int parseMemoryStat(char **text, unsigned int tag,
                           const char *search, virDomainMemoryStatPtr stat)
{
    char *dummy;
    unsigned long long value;

    if (STRPREFIX (*text, search)) {
        *text += strlen(search);
        if (virStrToLong_ull (*text, &dummy, 10, &value)) {
615
            VIR_DEBUG ("error reading %s: %s", search, *text);
616 617 618 619
            return 0;
        }

        switch (tag) {
620 621 622 623 624
            /* Convert megabytes to kilobytes for libvirt */
            case VIR_DOMAIN_MEMORY_STAT_ACTUAL_BALLOON:
                value <<= 10;
                break;
            /* Convert bytes to kilobytes for libvirt */
625 626 627 628
            case VIR_DOMAIN_MEMORY_STAT_SWAP_IN:
            case VIR_DOMAIN_MEMORY_STAT_SWAP_OUT:
            case VIR_DOMAIN_MEMORY_STAT_UNUSED:
            case VIR_DOMAIN_MEMORY_STAT_AVAILABLE:
629
                value >>= 10;
630 631 632 633 634 635 636 637 638
        }
        stat->tag = tag;
        stat->val = value;
        return 1;
    }
    return 0;
}

/* The reply from the 'info balloon' command may contain additional memory
639
 * statistics in the form: 'actual=<val> [,<tag>=<val>]*'
640
 */
641 642 643
static int qemuMonitorParseBalloonInfo(char *text,
                                       virDomainMemoryStatPtr stats,
                                       unsigned int nr_stats)
644 645 646 647
{
    char *p = text;
    unsigned int nr_stats_found = 0;

648 649 650 651 652 653 654 655 656 657
    /* Since "actual=" always comes first in the returned string,
     * and sometime we only care about the value of "actual", such
     * as qemuMonitorGetBalloonInfo, we parse it outside of the
     * loop.
     */
    if (parseMemoryStat(&p, VIR_DOMAIN_MEMORY_STAT_ACTUAL_BALLOON,
                        "actual=", &stats[nr_stats_found]) == 1) {
        nr_stats_found++;
    }

658 659 660 661 662 663 664 665 666 667 668 669
    while (*p && nr_stats_found < nr_stats) {
        if (parseMemoryStat(&p, VIR_DOMAIN_MEMORY_STAT_SWAP_IN,
                            ",mem_swapped_in=", &stats[nr_stats_found]) ||
            parseMemoryStat(&p, VIR_DOMAIN_MEMORY_STAT_SWAP_OUT,
                            ",mem_swapped_out=", &stats[nr_stats_found]) ||
            parseMemoryStat(&p, VIR_DOMAIN_MEMORY_STAT_MAJOR_FAULT,
                            ",major_page_faults=", &stats[nr_stats_found]) ||
            parseMemoryStat(&p, VIR_DOMAIN_MEMORY_STAT_MINOR_FAULT,
                            ",minor_page_faults=", &stats[nr_stats_found]) ||
            parseMemoryStat(&p, VIR_DOMAIN_MEMORY_STAT_UNUSED,
                            ",free_mem=", &stats[nr_stats_found]) ||
            parseMemoryStat(&p, VIR_DOMAIN_MEMORY_STAT_AVAILABLE,
670
                            ",total_mem=", &stats[nr_stats_found]))
671 672
            nr_stats_found++;

673 674 675 676 677
        /* Skip to the next label.  When *p is ',' the last match attempt
         * failed so try to match the next ','.
         */
        if (*p == ',')
            p++;
678 679 680 681 682
        p = strchr (p, ',');
        if (!p) break;
    }
    return nr_stats_found;
}
683

684 685

/* The reply from QEMU contains 'ballon: actual=421' where value is in MB */
686
#define BALLOON_PREFIX "balloon: "
687

688 689 690 691
/*
 * Returns: 0 if balloon not supported, +1 if balloon query worked
 * or -1 on failure
 */
692
int qemuMonitorTextGetBalloonInfo(qemuMonitorPtr mon,
693
                                  unsigned long long *currmem)
694 695 696 697 698
{
    char *reply = NULL;
    int ret = -1;
    char *offset;

699
    if (qemuMonitorHMPCommand(mon, "info balloon", &reply) < 0)
700 701 702 703
        return -1;

    if ((offset = strstr(reply, BALLOON_PREFIX)) != NULL) {
        offset += strlen(BALLOON_PREFIX);
704 705 706
        struct _virDomainMemoryStat stats[1];

        if (qemuMonitorParseBalloonInfo(offset, stats, 1) == 0) {
707 708
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("unexpected balloon information '%s'"), reply);
709 710
            goto cleanup;
        }
711 712

        if (stats[0].tag != VIR_DOMAIN_MEMORY_STAT_ACTUAL_BALLOON) {
713 714
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("unexpected balloon information '%s'"), reply);
715 716 717 718
            goto cleanup;
        }

        *currmem = stats[0].val;
719 720 721 722 723 724 725 726 727 728 729 730 731
        ret = 1;
    } else {
        /* We don't raise an error here, since its to be expected that
         * many QEMU's don't support ballooning
         */
        ret = 0;
    }

cleanup:
    VIR_FREE(reply);
    return ret;
}

732 733 734 735 736 737 738 739
int qemuMonitorTextGetMemoryStats(qemuMonitorPtr mon,
                                  virDomainMemoryStatPtr stats,
                                  unsigned int nr_stats)
{
    char *reply = NULL;
    int ret = 0;
    char *offset;

740
    if (qemuMonitorHMPCommand(mon, "info balloon", &reply) < 0)
741 742 743
        return -1;

    if ((offset = strstr(reply, BALLOON_PREFIX)) != NULL) {
744
        offset += strlen(BALLOON_PREFIX);
745
        ret = qemuMonitorParseBalloonInfo(offset, stats, nr_stats);
746 747 748 749 750 751
    }

    VIR_FREE(reply);
    return ret;
}

752

753
int qemuMonitorTextGetBlockInfo(qemuMonitorPtr mon,
754
                                virHashTablePtr table)
755
{
756
    struct qemuDomainDiskInfo *info = NULL;
757 758 759
    char *reply = NULL;
    int ret = -1;
    char *dummy;
760 761
    char *p, *eol;
    char *dev;
762 763
    int tmp;

764
    if (qemuMonitorHMPCommand(mon, "info block", &reply) < 0)
765 766 767
        goto cleanup;

    if (strstr(reply, "\ninfo ")) {
768 769 770
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s",
                       _("info block not supported by this qemu"));
771 772 773 774 775 776 777 778 779 780 781 782 783
        goto cleanup;
    }

    /* The output looks like this:
     * drive-ide0-0-0: removable=0 file=<path> ro=0 drv=raw encrypted=0
     * drive-ide0-1-0: removable=1 locked=0 file=<path> ro=1 drv=raw encrypted=0
     */
    p = reply;

    while (*p) {
        if (STRPREFIX(p, QEMU_DRIVE_HOST_PREFIX))
            p += strlen(QEMU_DRIVE_HOST_PREFIX);

784 785 786
        eol = strchr(p, '\n');
        if (!eol)
            eol = p + strlen(p) - 1;
787

788 789 790 791 792 793 794
        dev = p;
        p = strchr(p, ':');
        if (p && p < eol && *(p + 1) == ' ') {
            if (VIR_ALLOC(info) < 0) {
                virReportOOMError();
                goto cleanup;
            }
795

796 797
            *p = '\0';
            p += 2;
798

799
            while (p < eol) {
800 801 802 803 804
                if (STRPREFIX(p, "removable=")) {
                    p += strlen("removable=");
                    if (virStrToLong_i(p, &dummy, 10, &tmp) == -1)
                        VIR_DEBUG("error reading removable: %s", p);
                    else
E
Eric Blake 已提交
805
                        info->removable = (tmp != 0);
806 807 808 809 810
                } else if (STRPREFIX(p, "locked=")) {
                    p += strlen("locked=");
                    if (virStrToLong_i(p, &dummy, 10, &tmp) == -1)
                        VIR_DEBUG("error reading locked: %s", p);
                    else
E
Eric Blake 已提交
811
                        info->locked = (tmp != 0);
812 813 814 815 816
                } else if (STRPREFIX(p, "tray_open=")) {
                    p += strlen("tray_open=");
                    if (virStrToLong_i(p, &dummy, 10, &tmp) == -1)
                        VIR_DEBUG("error reading tray_open: %s", p);
                    else
E
Eric Blake 已提交
817
                        info->tray_open = (tmp != 0);
818 819 820 821 822 823 824 825 826 827 828 829 830 831 832
                } else if (STRPREFIX(p, "io-status=")) {
                    char *end;
                    char c;

                    p += strlen("io-status=");
                    end = strchr(p, ' ');
                    if (!end || end > eol)
                        end = eol;

                    c = *end;
                    *end = '\0';
                    info->io_status = qemuMonitorBlockIOStatusToError(p);
                    *end = c;
                    if (info->io_status < 0)
                        goto cleanup;
833 834 835 836 837 838
                } else {
                    /* ignore because we don't parse all options */
                }

                /* skip to next label */
                p = strchr(p, ' ');
839 840
                if (!p)
                    break;
841 842 843
                p++;
            }

844 845 846 847
            if (virHashAddEntry(table, dev, info) < 0)
                goto cleanup;
            else
                info = NULL;
848 849
        }

850 851
        /* skip to the next line */
        p = eol + 1;
852 853
    }

854
    ret = 0;
855 856

cleanup:
857
    VIR_FREE(info);
858 859 860 861
    VIR_FREE(reply);
    return ret;
}

862
int qemuMonitorTextGetBlockStatsInfo(qemuMonitorPtr mon,
863
                                     const char *dev_name,
864 865
                                     long long *rd_req,
                                     long long *rd_bytes,
866
                                     long long *rd_total_times,
867 868
                                     long long *wr_req,
                                     long long *wr_bytes,
869 870 871
                                     long long *wr_total_times,
                                     long long *flush_req,
                                     long long *flush_total_times,
872
                                     long long *errs)
873 874 875 876 877
{
    char *info = NULL;
    int ret = -1;
    char *dummy;
    const char *p, *eol;
878
    int devnamelen = strlen(dev_name);
879

880
    if (qemuMonitorHMPCommand (mon, "info blockstats", &info) < 0)
881 882 883 884 885 886 887 888
        goto cleanup;

    /* If the command isn't supported then qemu prints the supported
     * info commands, so the output starts "info ".  Since this is
     * unlikely to be the name of a block device, we can use this
     * to detect if qemu supports the command.
     */
    if (strstr(info, "\ninfo ")) {
889 890 891
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s",
                       _("'info blockstats' not supported by this qemu"));
892 893 894
        goto cleanup;
    }

895 896 897 898 899 900 901 902 903 904 905
    *rd_req = *rd_bytes = -1;
    *wr_req = *wr_bytes = *errs = -1;

    if (rd_total_times)
        *rd_total_times = -1;
    if (wr_total_times)
        *wr_total_times = -1;
    if (flush_req)
        *flush_req = -1;
    if (flush_total_times)
        *flush_total_times = -1;
906 907 908 909 910 911 912 913 914

    /* The output format for both qemu & KVM is:
     *   blockdevice: rd_bytes=% wr_bytes=% rd_operations=% wr_operations=%
     *   (repeated for each block device)
     * where '%' is a 64 bit number.
     */
    p = info;

    while (*p) {
915 916
        /* New QEMU has separate names for host & guest side of the disk
         * and libvirt gives the host side a 'drive-' prefix. The passed
917
         * in dev_name is the guest side though
918 919 920 921
         */
        if (STRPREFIX(p, QEMU_DRIVE_HOST_PREFIX))
            p += strlen(QEMU_DRIVE_HOST_PREFIX);

922
        if (STREQLEN (p, dev_name, devnamelen)
923 924 925 926 927 928 929 930 931 932
            && p[devnamelen] == ':' && p[devnamelen+1] == ' ') {

            eol = strchr (p, '\n');
            if (!eol)
                eol = p + strlen (p);

            p += devnamelen+2;         /* Skip to first label. */

            while (*p) {
                if (STRPREFIX (p, "rd_bytes=")) {
933
                    p += strlen("rd_bytes=");
934
                    if (virStrToLong_ll (p, &dummy, 10, rd_bytes) == -1)
935
                        VIR_DEBUG ("error reading rd_bytes: %s", p);
936
                } else if (STRPREFIX (p, "wr_bytes=")) {
937
                    p += strlen("wr_bytes=");
938
                    if (virStrToLong_ll (p, &dummy, 10, wr_bytes) == -1)
939
                        VIR_DEBUG ("error reading wr_bytes: %s", p);
940
                } else if (STRPREFIX (p, "rd_operations=")) {
941
                    p += strlen("rd_operations=");
942
                    if (virStrToLong_ll (p, &dummy, 10, rd_req) == -1)
943
                        VIR_DEBUG ("error reading rd_req: %s", p);
944
                } else if (STRPREFIX (p, "wr_operations=")) {
945
                    p += strlen("wr_operations=");
946
                    if (virStrToLong_ll (p, &dummy, 10, wr_req) == -1)
947
                        VIR_DEBUG ("error reading wr_req: %s", p);
948
                } else if (rd_total_times &&
949 950
                           STRPREFIX (p, "rd_total_time_ns=")) {
                    p += strlen("rd_total_time_ns=");
951 952 953
                    if (virStrToLong_ll (p, &dummy, 10, rd_total_times) == -1)
                        VIR_DEBUG ("error reading rd_total_times: %s", p);
                } else if (wr_total_times &&
954 955
                           STRPREFIX (p, "wr_total_time_ns=")) {
                    p += strlen("wr_total_time_ns=");
956 957 958 959 960 961 962 963
                    if (virStrToLong_ll (p, &dummy, 10, wr_total_times) == -1)
                        VIR_DEBUG ("error reading wr_total_times: %s", p);
                } else if (flush_req &&
                           STRPREFIX (p, "flush_operations=")) {
                    p += strlen("flush_operations=");
                    if (virStrToLong_ll (p, &dummy, 10, flush_req) == -1)
                        VIR_DEBUG ("error reading flush_req: %s", p);
                } else if (flush_total_times &&
964 965
                           STRPREFIX (p, "flush_total_time_ns=")) {
                    p += strlen("flush_total_time_ns=");
966 967 968
                    if (virStrToLong_ll (p, &dummy, 10, flush_total_times) == -1)
                        VIR_DEBUG ("error reading flush_total_times: %s", p);
                } else {
969
                    VIR_DEBUG ("unknown block stat near %s", p);
970
                }
971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987

                /* Skip to next label. */
                p = strchr (p, ' ');
                if (!p || p >= eol) break;
                p++;
            }
            ret = 0;
            goto cleanup;
        }

        /* Skip to next line. */
        p = strchr (p, '\n');
        if (!p) break;
        p++;
    }

    /* If we reach here then the device was not found. */
988 989
    virReportError (VIR_ERR_INVALID_ARG,
                    _("no stats found for device %s"), dev_name);
990 991 992 993 994 995

 cleanup:
    VIR_FREE(info);
    return ret;
}

996 997 998 999 1000 1001 1002 1003
int qemuMonitorTextGetBlockStatsParamsNumber(qemuMonitorPtr mon,
                                             int *nparams)
{
    char *info = NULL;
    int ret = -1;
    int num = 0;
    const char *p, *eol;

1004
    if (qemuMonitorHMPCommand (mon, "info blockstats", &info) < 0)
1005 1006 1007 1008 1009 1010 1011 1012
        goto cleanup;

    /* If the command isn't supported then qemu prints the supported
     * info commands, so the output starts "info ".  Since this is
     * unlikely to be the name of a block device, we can use this
     * to detect if qemu supports the command.
     */
    if (strstr(info, "\ninfo ")) {
1013 1014 1015
        virReportError(VIR_ERR_OPERATION_INVALID,
                       "%s",
                       _("'info blockstats' not supported by this qemu"));
1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034
        goto cleanup;
    }

    /* The output format for both qemu & KVM is:
     *   blockdevice: rd_bytes=% wr_bytes=% rd_operations=% wr_operations=%
     *   (repeated for each block device)
     * where '%' is a 64 bit number.
     */
    p = info;

    eol = strchr (p, '\n');
    if (!eol)
        eol = p + strlen (p);

    /* Skip the device name and following ":", and spaces (e.g.
     * "floppy0: ")
     */
    p = strchr(p, ' ');

E
Eric Blake 已提交
1035 1036 1037 1038 1039
    while (p && p < eol) {
        if (STRPREFIX (p, " rd_bytes=") ||
            STRPREFIX (p, " wr_bytes=") ||
            STRPREFIX (p, " rd_operations=") ||
            STRPREFIX (p, " wr_operations=") ||
1040 1041
            STRPREFIX (p, " rd_total_time_ns=") ||
            STRPREFIX (p, " wr_total_time_ns=") ||
E
Eric Blake 已提交
1042
            STRPREFIX (p, " flush_operations=") ||
1043
            STRPREFIX (p, " flush_total_time_ns=")) {
E
Eric Blake 已提交
1044 1045 1046 1047
            num++;
        } else {
            VIR_DEBUG ("unknown block stat near %s", p);
        }
1048

E
Eric Blake 已提交
1049 1050
        /* Skip to next label. */
        p = strchr(p + 1, ' ');
1051 1052 1053 1054 1055 1056 1057 1058 1059
    }

    *nparams = num;
    ret = 0;

 cleanup:
    VIR_FREE(info);
    return ret;
}
1060

1061
int qemuMonitorTextGetBlockExtent(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
1062
                                  const char *dev_name ATTRIBUTE_UNUSED,
1063 1064
                                  unsigned long long *extent ATTRIBUTE_UNUSED)
{
1065 1066
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                   _("unable to query block extent with this QEMU"));
1067 1068 1069
    return -1;
}

E
Eric Blake 已提交
1070 1071
/* Return 0 on success, -1 on failure, or -2 if not supported.  Size
 * is in bytes. */
1072 1073 1074 1075 1076 1077 1078 1079
int qemuMonitorTextBlockResize(qemuMonitorPtr mon,
                               const char *device,
                               unsigned long long size)
{
    char *cmd = NULL;
    char *reply = NULL;
    int ret = -1;

E
Eric Blake 已提交
1080
    if (virAsprintf(&cmd, "block_resize %s %lluB", device, size) < 0) {
1081 1082 1083 1084
        virReportOOMError();
        goto cleanup;
    }

1085
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099
        goto cleanup;

    if (strstr(reply, "unknown command:")) {
        ret = -2;
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}
1100

1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114
static int
qemuMonitorSendVNCPassphrase(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
                             qemuMonitorMessagePtr msg,
                             const char *data ATTRIBUTE_UNUSED,
                             size_t len ATTRIBUTE_UNUSED,
                             void *opaque)
{
    char *passphrase = opaque;
    size_t passphrase_len = strlen(passphrase);

    /* Enlarge transmit buffer to allow for the extra data
     * to be sent back */
    if (VIR_REALLOC_N(msg->txBuffer,
                      msg->txLength + passphrase_len + 1 + 1) < 0) {
1115
        virReportOOMError();
1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129
        return -1;
    }

    /* Queue the password for sending */
    memcpy(msg->txBuffer + msg->txLength,
           passphrase, passphrase_len);
    msg->txLength += passphrase_len;
    msg->txBuffer[msg->txLength] = '\r';
    msg->txLength++;
    msg->txBuffer[msg->txLength] = '\0';

    return 0;
}

1130 1131
int qemuMonitorTextSetVNCPassword(qemuMonitorPtr mon,
                                  const char *password)
1132 1133
{
    char *info = NULL;
1134

1135 1136 1137 1138
    if (qemuMonitorTextCommandWithHandler(mon, "change vnc password",
                                          qemuMonitorSendVNCPassphrase,
                                          (char *)password,
                                          -1, &info) < 0) {
1139 1140
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("setting VNC password failed"));
1141 1142 1143 1144 1145
        return -1;
    }
    VIR_FREE(info);
    return 0;
}
1146

1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162
/* Returns -1 on error, -2 if not supported */
int qemuMonitorTextSetPassword(qemuMonitorPtr mon,
                               const char *protocol,
                               const char *password,
                               const char *action_if_connected)
{
    char *cmd = NULL;
    char *reply = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "set_password %s \"%s\" %s",
                    protocol, password, action_if_connected) < 0) {
        virReportOOMError();
        goto cleanup;
    }

1163
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178
        goto cleanup;

    if (strstr(reply, "unknown command:")) {
        ret = -2;
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(reply);
    VIR_FREE(cmd);
    return ret;
}

1179
/* Returns -1 on error, -2 if not supported */
1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193
int qemuMonitorTextExpirePassword(qemuMonitorPtr mon,
                                  const char *protocol,
                                  const char *expire_time)
{
    char *cmd = NULL;
    char *reply = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "expire_password %s %s",
                    protocol, expire_time) < 0) {
        virReportOOMError();
        goto cleanup;
    }

1194
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
1195 1196 1197
        goto cleanup;

    if (strstr(reply, "unknown command:")) {
1198
        ret = -2;
1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(reply);
    VIR_FREE(cmd);
    return ret;
}

1210 1211 1212 1213
/*
 * Returns: 0 if balloon not supported, +1 if balloon adjust worked
 * or -1 on failure
 */
1214 1215
int qemuMonitorTextSetBalloon(qemuMonitorPtr mon,
                              unsigned long newmem)
1216 1217 1218 1219 1220 1221 1222 1223 1224
{
    char *cmd;
    char *reply = NULL;
    int ret = -1;

    /*
     * 'newmem' is in KB, QEMU monitor works in MB, and we all wish
     * we just worked in bytes with unsigned long long everywhere.
     */
1225
    if (virAsprintf(&cmd, "balloon %lu", VIR_DIV_UP(newmem, 1024)) < 0) {
1226
        virReportOOMError();
1227 1228 1229
        return -1;
    }

1230
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0) {
1231 1232 1233 1234 1235 1236 1237
        VIR_FREE(cmd);
        return -1;
    }
    VIR_FREE(cmd);

    /* If the command failed qemu prints: 'unknown command'
     * No message is printed on success it seems */
1238
    if (strstr(reply, "unknown command:")) {
1239 1240 1241 1242 1243 1244 1245 1246 1247 1248
        /* Don't set error - it is expected memory balloon fails on many qemu */
        ret = 0;
    } else {
        ret = 1;
    }

    VIR_FREE(reply);
    return ret;
}

1249 1250

/*
1251
 * Returns: 0 if CPU hotplug not supported, +1 if CPU hotplug worked
1252 1253 1254 1255 1256 1257 1258 1259
 * or -1 on failure
 */
int qemuMonitorTextSetCPU(qemuMonitorPtr mon, int cpu, int online)
{
    char *cmd;
    char *reply = NULL;
    int ret = -1;

1260
    if (virAsprintf(&cmd, "cpu_set %d %s", cpu, online ? "online" : "offline") < 0) {
1261 1262 1263 1264
        virReportOOMError();
        return -1;
    }

1265
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0) {
1266 1267 1268 1269 1270 1271 1272
        VIR_FREE(cmd);
        return -1;
    }
    VIR_FREE(cmd);

    /* If the command failed qemu prints: 'unknown command'
     * No message is printed on success it seems */
1273
    if (strstr(reply, "unknown command:")) {
1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284
        /* Don't set error - it is expected CPU onlining fails on many qemu - caller will handle */
        ret = 0;
    } else {
        ret = 1;
    }

    VIR_FREE(reply);
    return ret;
}


1285
int qemuMonitorTextEjectMedia(qemuMonitorPtr mon,
1286
                              const char *dev_name,
1287
                              bool force)
1288 1289 1290 1291 1292
{
    char *cmd = NULL;
    char *reply = NULL;
    int ret = -1;

1293
    if (virAsprintf(&cmd, "eject %s%s", force ? "-f " : "", dev_name) < 0) {
1294
        virReportOOMError();
1295 1296 1297
        goto cleanup;
    }

1298
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
1299 1300 1301 1302 1303
        goto cleanup;

    /* If the command failed qemu prints:
     * device not found, device is locked ...
     * No message is printed on success it seems */
1304
    if (c_strcasestr(reply, "device ")) {
1305 1306
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("could not eject media on %s: %s"), dev_name, reply);
1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(reply);
    VIR_FREE(cmd);
    return ret;
}


1319
int qemuMonitorTextChangeMedia(qemuMonitorPtr mon,
1320
                               const char *dev_name,
1321 1322
                               const char *newmedia,
                               const char *format ATTRIBUTE_UNUSED)
1323 1324 1325 1326 1327 1328
{
    char *cmd = NULL;
    char *reply = NULL;
    char *safepath = NULL;
    int ret = -1;

1329
    if (!(safepath = qemuMonitorEscapeArg(newmedia))) {
1330
        virReportOOMError();
1331 1332 1333
        goto cleanup;
    }

1334
    if (virAsprintf(&cmd, "change %s \"%s\"", dev_name, safepath) < 0) {
1335
        virReportOOMError();
1336 1337 1338
        goto cleanup;
    }

1339
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
1340 1341 1342 1343 1344
        goto cleanup;

    /* If the command failed qemu prints:
     * device not found, device is locked ...
     * No message is printed on success it seems */
1345
    if (c_strcasestr(reply, "device ")) {
1346 1347
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("could not change media on %s: %s"), dev_name, reply);
1348 1349 1350
        goto cleanup;
    }

R
Ryan Harper 已提交
1351
    /* Could not open message indicates bad filename */
1352
    if (strstr(reply, "Could not open ")) {
1353 1354
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("could not change media on %s: %s"), dev_name, reply);
R
Ryan Harper 已提交
1355 1356 1357
        goto cleanup;
    }

1358 1359 1360 1361 1362 1363 1364 1365 1366
    ret = 0;

cleanup:
    VIR_FREE(reply);
    VIR_FREE(cmd);
    VIR_FREE(safepath);
    return ret;
}

1367 1368 1369 1370 1371
static int qemuMonitorTextSaveMemory(qemuMonitorPtr mon,
                                     const char *cmdtype,
                                     unsigned long long offset,
                                     size_t length,
                                     const char *path)
1372 1373 1374 1375 1376 1377
{
    char *cmd = NULL;
    char *reply = NULL;
    char *safepath = NULL;
    int ret = -1;

1378
    if (!(safepath = qemuMonitorEscapeArg(path))) {
1379
        virReportOOMError();
1380 1381 1382 1383
        goto cleanup;
    }

    if (virAsprintf(&cmd, "%s %llu %zi \"%s\"", cmdtype, offset, length, safepath) < 0) {
1384
        virReportOOMError();
1385 1386 1387
        goto cleanup;
    }

1388
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402
        goto cleanup;

    /* XXX what is printed on failure ? */

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    VIR_FREE(safepath);
    return ret;
}


1403 1404 1405 1406
int qemuMonitorTextSaveVirtualMemory(qemuMonitorPtr mon,
                                     unsigned long long offset,
                                     size_t length,
                                     const char *path)
1407
{
1408
    return qemuMonitorTextSaveMemory(mon, "memsave", offset, length, path);
1409 1410
}

1411 1412 1413 1414
int qemuMonitorTextSavePhysicalMemory(qemuMonitorPtr mon,
                                      unsigned long long offset,
                                      size_t length,
                                      const char *path)
1415
{
1416
    return qemuMonitorTextSaveMemory(mon, "pmemsave", offset, length, path);
1417
}
1418 1419


1420 1421
int qemuMonitorTextSetMigrationSpeed(qemuMonitorPtr mon,
                                     unsigned long bandwidth)
1422 1423 1424 1425 1426 1427
{
    char *cmd = NULL;
    char *info = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "migrate_set_speed %lum", bandwidth) < 0) {
1428
        virReportOOMError();
1429 1430 1431
        goto cleanup;
    }

1432
    if (qemuMonitorHMPCommand(mon, cmd, &info) < 0)
1433 1434 1435 1436 1437 1438 1439 1440 1441
        goto cleanup;

    ret = 0;

cleanup:
    VIR_FREE(info);
    VIR_FREE(cmd);
    return ret;
}
1442 1443


1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455
int qemuMonitorTextSetMigrationDowntime(qemuMonitorPtr mon,
                                        unsigned long long downtime)
{
    char *cmd = NULL;
    char *info = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "migrate_set_downtime %llums", downtime) < 0) {
        virReportOOMError();
        goto cleanup;
    }

1456
    if (qemuMonitorHMPCommand(mon, cmd, &info) < 0)
1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467
        goto cleanup;

    ret = 0;

cleanup:
    VIR_FREE(info);
    VIR_FREE(cmd);
    return ret;
}


1468 1469 1470 1471
#define MIGRATION_PREFIX "Migration status: "
#define MIGRATION_TRANSFER_PREFIX "transferred ram: "
#define MIGRATION_REMAINING_PREFIX "remaining ram: "
#define MIGRATION_TOTAL_PREFIX "total ram: "
1472 1473 1474
#define MIGRATION_DISK_TRANSFER_PREFIX "transferred disk: "
#define MIGRATION_DISK_REMAINING_PREFIX "remaining disk: "
#define MIGRATION_DISK_TOTAL_PREFIX "total disk: "
1475

1476 1477 1478 1479 1480
int qemuMonitorTextGetMigrationStatus(qemuMonitorPtr mon,
                                      int *status,
                                      unsigned long long *transferred,
                                      unsigned long long *remaining,
                                      unsigned long long *total) {
1481 1482 1483
    char *reply;
    char *tmp;
    char *end;
1484 1485 1486
    unsigned long long disk_transferred = 0;
    unsigned long long disk_remaining = 0;
    unsigned long long disk_total = 0;
1487 1488 1489 1490 1491 1492 1493
    int ret = -1;

    *status = QEMU_MONITOR_MIGRATION_STATUS_INACTIVE;
    *transferred = 0;
    *remaining = 0;
    *total = 0;

1494
    if (qemuMonitorHMPCommand(mon, "info migrate", &reply) < 0)
1495 1496 1497 1498
        return -1;

    if ((tmp = strstr(reply, MIGRATION_PREFIX)) != NULL) {
        tmp += strlen(MIGRATION_PREFIX);
P
Paolo Bonzini 已提交
1499
        end = strchr(tmp, '\r');
1500
        if (end == NULL) {
1501 1502
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("unexpected migration status in %s"), reply);
1503 1504
            goto cleanup;
        }
1505 1506 1507
        *end = '\0';

        if ((*status = qemuMonitorMigrationStatusTypeFromString(tmp)) < 0) {
1508 1509
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("unexpected migration status in %s"), reply);
1510 1511 1512 1513 1514 1515 1516 1517 1518 1519
            goto cleanup;
        }

        if (*status == QEMU_MONITOR_MIGRATION_STATUS_ACTIVE) {
            tmp = end + 1;

            if (!(tmp = strstr(tmp, MIGRATION_TRANSFER_PREFIX)))
                goto done;
            tmp += strlen(MIGRATION_TRANSFER_PREFIX);

1520
            if (virStrToLong_ull(tmp, &end, 10, transferred) < 0) {
1521 1522 1523
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("cannot parse migration data transferred "
                                 "statistic %s"), tmp);
1524 1525 1526
                goto cleanup;
            }
            *transferred *= 1024;
1527
            tmp = end;
1528 1529 1530 1531 1532

            if (!(tmp = strstr(tmp, MIGRATION_REMAINING_PREFIX)))
                goto done;
            tmp += strlen(MIGRATION_REMAINING_PREFIX);

1533
            if (virStrToLong_ull(tmp, &end, 10, remaining) < 0) {
1534 1535 1536
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("cannot parse migration data remaining "
                                 "statistic %s"), tmp);
1537 1538 1539
                goto cleanup;
            }
            *remaining *= 1024;
1540
            tmp = end;
1541 1542 1543 1544 1545

            if (!(tmp = strstr(tmp, MIGRATION_TOTAL_PREFIX)))
                goto done;
            tmp += strlen(MIGRATION_TOTAL_PREFIX);

1546
            if (virStrToLong_ull(tmp, &end, 10, total) < 0) {
1547 1548 1549
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("cannot parse migration data total "
                                 "statistic %s"), tmp);
1550 1551 1552
                goto cleanup;
            }
            *total *= 1024;
1553 1554 1555 1556 1557 1558 1559 1560 1561 1562
            tmp = end;

            /*
             * Check for Optional Disk Migration status
             */
            if (!(tmp = strstr(tmp, MIGRATION_DISK_TRANSFER_PREFIX)))
                goto done;
            tmp += strlen(MIGRATION_DISK_TRANSFER_PREFIX);

            if (virStrToLong_ull(tmp, &end, 10, &disk_transferred) < 0) {
1563 1564 1565
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("cannot parse disk migration data "
                                 "transferred statistic %s"), tmp);
1566 1567 1568 1569 1570 1571 1572 1573
                goto cleanup;
            }
            *transferred += disk_transferred * 1024;
            tmp = end;

            if (!(tmp = strstr(tmp, MIGRATION_DISK_REMAINING_PREFIX)))
                goto done;
            tmp += strlen(MIGRATION_DISK_REMAINING_PREFIX);
1574

1575
            if (virStrToLong_ull(tmp, &end, 10, &disk_remaining) < 0) {
1576 1577 1578
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("cannot parse disk migration data remaining "
                                 "statistic %s"), tmp);
1579 1580 1581 1582 1583 1584 1585 1586 1587 1588
                goto cleanup;
            }
            *remaining += disk_remaining * 1024;
            tmp = end;

            if (!(tmp = strstr(tmp, MIGRATION_DISK_TOTAL_PREFIX)))
                goto done;
            tmp += strlen(MIGRATION_DISK_TOTAL_PREFIX);

            if (virStrToLong_ull(tmp, &end, 10, &disk_total) < 0) {
1589 1590 1591
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("cannot parse disk migration data total "
                                 "statistic %s"), tmp);
1592 1593 1594
                goto cleanup;
            }
            *total += disk_total * 1024;
1595 1596 1597 1598 1599 1600 1601 1602 1603 1604
        }
    }

done:
    ret = 0;

cleanup:
    VIR_FREE(reply);
    return ret;
}
1605 1606


1607 1608 1609
int qemuMonitorTextMigrate(qemuMonitorPtr mon,
                           unsigned int flags,
                           const char *dest)
1610 1611 1612 1613
{
    char *cmd = NULL;
    char *info = NULL;
    int ret = -1;
1614
    char *safedest = qemuMonitorEscapeArg(dest);
1615
    virBuffer extra = VIR_BUFFER_INITIALIZER;
C
Chris Lalancette 已提交
1616
    char *extrastr = NULL;
1617

1618
    if (!safedest) {
1619
        virReportOOMError();
1620 1621 1622
        return -1;
    }

1623
    if (flags & QEMU_MONITOR_MIGRATE_BACKGROUND)
1624
        virBufferAddLit(&extra, " -d");
1625
    if (flags & QEMU_MONITOR_MIGRATE_NON_SHARED_DISK)
1626
        virBufferAddLit(&extra, " -b");
1627
    if (flags & QEMU_MONITOR_MIGRATE_NON_SHARED_INC)
1628 1629 1630 1631
        virBufferAddLit(&extra, " -i");
    if (virBufferError(&extra)) {
        virBufferFreeAndReset(&extra);
        virReportOOMError();
C
Chris Lalancette 已提交
1632
        goto cleanup;
1633
    }
C
Chris Lalancette 已提交
1634 1635 1636 1637

    extrastr = virBufferContentAndReset(&extra);
    if (virAsprintf(&cmd, "migrate %s\"%s\"", extrastr ? extrastr : "",
                    safedest) < 0) {
1638
        virReportOOMError();
1639 1640 1641
        goto cleanup;
    }

1642
    if (qemuMonitorHMPCommand(mon, cmd, &info) < 0)
1643 1644 1645 1646
        goto cleanup;

    /* Now check for "fail" in the output string */
    if (strstr(info, "fail") != NULL) {
1647 1648
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("migration to '%s' failed: %s"), dest, info);
1649 1650 1651 1652 1653
        goto cleanup;
    }
    /* If the command isn't supported then qemu prints:
     * unknown command: migrate" */
    if (strstr(info, "unknown command:")) {
1654 1655
        virReportError(VIR_ERR_OPERATION_INVALID,
                       _("migration to '%s' not supported by this qemu: %s"), dest, info);
1656 1657 1658 1659 1660 1661 1662
        goto cleanup;
    }


    ret = 0;

cleanup:
C
Chris Lalancette 已提交
1663
    VIR_FREE(extrastr);
1664
    VIR_FREE(safedest);
1665 1666 1667 1668 1669
    VIR_FREE(info);
    VIR_FREE(cmd);
    return ret;
}

1670
int qemuMonitorTextMigrateCancel(qemuMonitorPtr mon)
1671 1672
{
    char *info = NULL;
1673
    int ret;
1674

1675
    ret = qemuMonitorHMPCommand(mon, "migrate_cancel", &info);
1676

1677 1678
    VIR_FREE(info);
    return ret;
1679
}
1680

1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709

int qemuMonitorTextGraphicsRelocate(qemuMonitorPtr mon,
                                    int type,
                                    const char *hostname,
                                    int port,
                                    int tlsPort,
                                    const char *tlsSubject)
{
    char *cmd;
    char *info = NULL;

    if (virAsprintf(&cmd, "client_migrate_info %s %s %d %d %s",
                    type == VIR_DOMAIN_GRAPHICS_TYPE_SPICE ? "spice" : "vnc",
                    hostname, port, tlsPort, tlsSubject ? tlsSubject : "") < 0) {
        virReportOOMError();
        return -1;
    }

    if (qemuMonitorHMPCommand(mon, cmd, &info) < 0) {
        VIR_FREE(cmd);
        return -1;
    }
    VIR_FREE(cmd);
    VIR_FREE(info);

    return 0;
}


1710 1711
int qemuMonitorTextAddUSBDisk(qemuMonitorPtr mon,
                              const char *path)
1712 1713 1714 1715 1716 1717
{
    char *cmd = NULL;
    char *safepath;
    int ret = -1;
    char *info = NULL;

1718
    safepath = qemuMonitorEscapeArg(path);
1719
    if (!safepath) {
1720
        virReportOOMError();
1721 1722 1723 1724
        return -1;
    }

    if (virAsprintf(&cmd, "usb_add disk:%s", safepath) < 0) {
1725
        virReportOOMError();
1726 1727 1728
        goto cleanup;
    }

1729
    if (qemuMonitorHMPCommand(mon, cmd, &info) < 0)
1730 1731 1732 1733 1734
        goto cleanup;

    /* If the command failed qemu prints:
     * Could not add ... */
    if (strstr(info, "Could not add ")) {
1735 1736
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("unable to add USB disk %s: %s"), path, info);
1737 1738 1739 1740 1741 1742 1743 1744
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(safepath);
1745
    VIR_FREE(info);
1746 1747
    return ret;
}
1748 1749


1750 1751
static int qemuMonitorTextAddUSBDevice(qemuMonitorPtr mon,
                                       const char *addr)
1752 1753 1754 1755 1756 1757
{
    char *cmd;
    char *reply = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "usb_add %s", addr) < 0) {
1758
        virReportOOMError();
1759 1760 1761
        return -1;
    }

1762
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
1763 1764 1765 1766 1767
        goto cleanup;

    /* If the command failed qemu prints:
     * Could not add ... */
    if (strstr(reply, "Could not add ")) {
1768 1769
        virReportError(VIR_ERR_OPERATION_FAILED,
                       "%s", _("adding usb device failed"));
1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}


1782 1783 1784
int qemuMonitorTextAddUSBDeviceExact(qemuMonitorPtr mon,
                                     int bus,
                                     int dev)
1785 1786 1787 1788 1789
{
    int ret;
    char *addr;

    if (virAsprintf(&addr, "host:%.3d.%.3d", bus, dev) < 0) {
1790
        virReportOOMError();
1791 1792 1793
        return -1;
    }

1794
    ret = qemuMonitorTextAddUSBDevice(mon, addr);
1795 1796 1797 1798 1799

    VIR_FREE(addr);
    return ret;
}

1800 1801 1802
int qemuMonitorTextAddUSBDeviceMatch(qemuMonitorPtr mon,
                                     int vendor,
                                     int product)
1803 1804 1805 1806 1807
{
    int ret;
    char *addr;

    if (virAsprintf(&addr, "host:%.4x:%.4x", vendor, product) < 0) {
1808
        virReportOOMError();
1809 1810 1811
        return -1;
    }

1812
    ret = qemuMonitorTextAddUSBDevice(mon, addr);
1813 1814 1815 1816

    VIR_FREE(addr);
    return ret;
}
1817 1818 1819


static int
1820 1821
qemuMonitorTextParsePciAddReply(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
                                const char *reply,
1822
                                virDevicePCIAddress *addr)
1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838
{
    char *s, *e;

    /* If the command succeeds qemu prints:
     * OK bus 0, slot XXX...
     * or
     * OK domain 0, bus 0, slot XXX
     */
    if (!(s = strstr(reply, "OK ")))
        return -1;

    s += 3;

    if (STRPREFIX(s, "domain ")) {
        s += strlen("domain ");

1839
        if (virStrToLong_ui(s, &e, 10, &addr->domain) == -1) {
1840
            VIR_WARN("Unable to parse domain number '%s'", s);
1841 1842 1843 1844
            return -1;
        }

        if (!STRPREFIX(e, ", ")) {
1845
            VIR_WARN("Expected ', ' parsing pci_add reply '%s'", s);
1846 1847 1848 1849 1850 1851
            return -1;
        }
        s = e + 2;
    }

    if (!STRPREFIX(s, "bus ")) {
1852
        VIR_WARN("Expected 'bus ' parsing pci_add reply '%s'", s);
1853 1854 1855 1856
        return -1;
    }
    s += strlen("bus ");

1857
    if (virStrToLong_ui(s, &e, 10, &addr->bus) == -1) {
1858
        VIR_WARN("Unable to parse bus number '%s'", s);
1859 1860 1861 1862
        return -1;
    }

    if (!STRPREFIX(e, ", ")) {
1863
        VIR_WARN("Expected ', ' parsing pci_add reply '%s'", s);
1864 1865 1866 1867 1868
        return -1;
    }
    s = e + 2;

    if (!STRPREFIX(s, "slot ")) {
1869
        VIR_WARN("Expected 'slot ' parsing pci_add reply '%s'", s);
1870 1871 1872 1873
        return -1;
    }
    s += strlen("slot ");

1874
    if (virStrToLong_ui(s, &e, 10, &addr->slot) == -1) {
1875
        VIR_WARN("Unable to parse slot number '%s'", s);
1876 1877 1878 1879 1880 1881 1882
        return -1;
    }

    return 0;
}


1883
int qemuMonitorTextAddPCIHostDevice(qemuMonitorPtr mon,
1884 1885
                                    virDevicePCIAddress *hostAddr,
                                    virDevicePCIAddress *guestAddr)
1886 1887 1888 1889 1890
{
    char *cmd;
    char *reply = NULL;
    int ret = -1;

1891
    memset(guestAddr, 0, sizeof(*guestAddr));
1892

1893
    /* XXX hostAddr->domain */
1894
    if (virAsprintf(&cmd, "pci_add pci_addr=auto host host=%.2x:%.2x.%.1x",
1895
                    hostAddr->bus, hostAddr->slot, hostAddr->function) < 0) {
1896
        virReportOOMError();
1897 1898 1899
        goto cleanup;
    }

1900
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
1901 1902 1903
        goto cleanup;

    if (strstr(reply, "invalid type: host")) {
1904 1905
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                       _("PCI device assignment is not supported by this version of qemu"));
1906 1907 1908
        goto cleanup;
    }

1909
    if (qemuMonitorTextParsePciAddReply(mon, reply, guestAddr) < 0) {
1910 1911
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("parsing pci_add reply failed: %s"), reply);
1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}

1923

1924 1925 1926
int qemuMonitorTextAddPCIDisk(qemuMonitorPtr mon,
                              const char *path,
                              const char *bus,
1927
                              virDevicePCIAddress *guestAddr)
1928
{
1929 1930 1931 1932 1933 1934
    char *cmd = NULL;
    char *reply = NULL;
    char *safe_path = NULL;
    int tryOldSyntax = 0;
    int ret = -1;

1935
    safe_path = qemuMonitorEscapeArg(path);
1936
    if (!safe_path) {
1937
        virReportOOMError();
1938 1939 1940 1941 1942 1943
        return -1;
    }

try_command:
    if (virAsprintf(&cmd, "pci_add %s storage file=%s,if=%s",
                    (tryOldSyntax ? "0": "pci_addr=auto"), safe_path, bus) < 0) {
1944
        virReportOOMError();
1945 1946 1947
        goto cleanup;
    }

1948
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
1949 1950
        goto cleanup;

1951
    if (qemuMonitorTextParsePciAddReply(mon, reply, guestAddr) < 0) {
1952 1953 1954 1955 1956 1957 1958
        if (!tryOldSyntax && strstr(reply, "invalid char in expression")) {
            VIR_FREE(reply);
            VIR_FREE(cmd);
            tryOldSyntax = 1;
            goto try_command;
        }

1959 1960
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("adding %s disk failed %s: %s"), bus, path, reply);
1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(safe_path);
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}


1974 1975
int qemuMonitorTextAddPCINetwork(qemuMonitorPtr mon,
                                 const char *nicstr,
1976
                                 virDevicePCIAddress *guestAddr)
1977 1978 1979 1980 1981 1982
{
    char *cmd;
    char *reply = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "pci_add pci_addr=auto nic %s", nicstr) < 0) {
1983
        virReportOOMError();
1984 1985 1986
        return -1;
    }

1987
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
1988 1989
        goto cleanup;

1990
    if (qemuMonitorTextParsePciAddReply(mon, reply, guestAddr) < 0) {
1991 1992
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("parsing pci_add reply failed: %s"), reply);
1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(reply);
    VIR_FREE(cmd);
    return ret;
}

2004

2005
int qemuMonitorTextRemovePCIDevice(qemuMonitorPtr mon,
2006
                                   virDevicePCIAddress *guestAddr)
2007 2008 2009 2010 2011 2012 2013 2014
{
    char *cmd = NULL;
    char *reply = NULL;
    int tryOldSyntax = 0;
    int ret = -1;

try_command:
    if (tryOldSyntax) {
2015
        if (virAsprintf(&cmd, "pci_del 0 %.2x", guestAddr->slot) < 0) {
2016
            virReportOOMError();
2017 2018 2019
            goto cleanup;
        }
    } else {
2020
        /* XXX function ? */
2021
        if (virAsprintf(&cmd, "pci_del pci_addr=%.4x:%.2x:%.2x",
2022
                        guestAddr->domain, guestAddr->bus, guestAddr->slot) < 0) {
2023
            virReportOOMError();
2024 2025 2026 2027
            goto cleanup;
        }
    }

2028
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2029 2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044
        goto cleanup;

    /* Syntax changed when KVM merged PCI hotplug upstream to QEMU,
     * so check for an error message from old KVM indicating the
     * need to try the old syntax */
    if (!tryOldSyntax &&
        strstr(reply, "extraneous characters")) {
        tryOldSyntax = 1;
        VIR_FREE(reply);
        VIR_FREE(cmd);
        goto try_command;
    }
    /* If the command fails due to a wrong slot qemu prints: invalid slot,
     * nothing is printed on success */
    if (strstr(reply, "invalid slot") ||
        strstr(reply, "Invalid pci address")) {
2045 2046 2047
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("failed to detach PCI device, invalid address %.4x:%.2x:%.2x: %s"),
                       guestAddr->domain, guestAddr->bus, guestAddr->slot, reply);
2048 2049 2050 2051 2052 2053 2054 2055 2056 2057
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}
2058 2059


2060 2061 2062
int qemuMonitorTextSendFileHandle(qemuMonitorPtr mon,
                                  const char *fdname,
                                  int fd)
2063 2064 2065 2066 2067 2068
{
    char *cmd;
    char *reply = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "getfd %s", fdname) < 0) {
2069
        virReportOOMError();
2070 2071 2072
        return -1;
    }

2073
    if (qemuMonitorHMPCommandWithFd(mon, cmd, fd, &reply) < 0)
2074 2075 2076 2077 2078
        goto cleanup;

    /* If the command isn't supported then qemu prints:
     * unknown command: getfd" */
    if (strstr(reply, "unknown command:")) {
2079 2080 2081
        virReportError(VIR_ERR_OPERATION_INVALID,
                       _("qemu does not support sending of file handles: %s"),
                       reply);
2082 2083 2084
        goto cleanup;
    }

2085
    if (STRNEQ(reply, "")) {
2086 2087
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("unable to send file handle '%s': %s"),
2088
                        fdname, reply);
2089 2090 2091
        goto cleanup;
    }

2092 2093 2094 2095 2096 2097 2098 2099 2100
    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}


2101 2102
int qemuMonitorTextCloseFileHandle(qemuMonitorPtr mon,
                                   const char *fdname)
2103 2104 2105 2106 2107 2108
{
    char *cmd;
    char *reply = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "closefd %s", fdname) < 0) {
2109
        virReportOOMError();
2110 2111 2112
        return -1;
    }

2113
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2114 2115 2116 2117 2118
        goto cleanup;

    /* If the command isn't supported then qemu prints:
     * unknown command: getfd" */
    if (strstr(reply, "unknown command:")) {
2119 2120 2121
        virReportError(VIR_ERR_OPERATION_INVALID,
                       _("qemu does not support closing of file handles: %s"),
                       reply);
2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}

2133

2134 2135
int qemuMonitorTextAddHostNetwork(qemuMonitorPtr mon,
                                  const char *netstr)
2136 2137 2138 2139 2140 2141
{
    char *cmd;
    char *reply = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "host_net_add %s", netstr) < 0) {
2142
        virReportOOMError();
2143 2144 2145
        return -1;
    }

2146
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2147 2148
        goto cleanup;

2149
    if (STRNEQ(reply, "")) {
2150 2151 2152
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("unable to add host net: %s"),
                       reply);
2153 2154
        goto cleanup;
    }
2155 2156 2157 2158 2159 2160 2161 2162

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}
2163 2164


2165 2166 2167
int qemuMonitorTextRemoveHostNetwork(qemuMonitorPtr mon,
                                     int vlan,
                                     const char *netname)
2168 2169 2170 2171 2172 2173
{
    char *cmd;
    char *reply = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "host_net_remove %d %s", vlan, netname) < 0) {
2174
        virReportOOMError();
2175 2176 2177
        return -1;
    }

2178
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189
        goto cleanup;

    /* XXX error messages here ? */

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}
2190 2191


2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203
int qemuMonitorTextAddNetdev(qemuMonitorPtr mon,
                             const char *netdevstr)
{
    char *cmd;
    char *reply = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "netdev_add %s", netdevstr) < 0) {
        virReportOOMError();
        return -1;
    }

2204
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229
        goto cleanup;

    /* XXX error messages here ? */

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}


int qemuMonitorTextRemoveNetdev(qemuMonitorPtr mon,
                                const char *alias)
{
    char *cmd;
    char *reply = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "netdev_del %s", alias) < 0) {
        virReportOOMError();
        return -1;
    }

2230
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243
        goto cleanup;

    /* XXX error messages here ? */

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}


2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261
/* Parse the output of "info chardev" and return a hash of pty paths.
 *
 * Output is:
 * foo: filename=pty:/dev/pts/7
 * monitor: filename=stdio
 * serial0: filename=vc
 * parallel0: filename=vc
 *
 * Non-pty lines are ignored. In the above example, key is 'foo', value is
 * '/dev/pty/7'. The hash will contain only a single value.
 */

int qemuMonitorTextGetPtyPaths(qemuMonitorPtr mon,
                               virHashTablePtr paths)
{
    char *reply = NULL;
    int ret = -1;

2262
    if (qemuMonitorHMPCommand(mon, "info chardev", &reply) < 0)
2263
        return -1;
2264

2265 2266
    char *pos;                          /* The current start of searching */
    char *next = reply;                 /* The start of the next line */
2267
    char *eol;                   /* The character which ends the current line */
2268 2269 2270 2271
    char *end = reply + strlen(reply);  /* The end of the reply string */

    while (next) {
        pos = next;
2272 2273 2274

        /* Split the output into lines */
        eol = memchr(pos, '\n', end - pos);
2275
        if (eol == NULL) {
2276
            eol = end;
2277 2278 2279 2280 2281 2282 2283 2284
            next = NULL;
        } else {
            next = eol + 1;
        }

        /* Ignore all whitespace immediately before eol */
        while (eol > pos && c_isspace(*(eol-1)))
            eol -= 1;
2285 2286 2287 2288 2289 2290 2291

        /* Look for 'filename=pty:' */
#define NEEDLE "filename=pty:"
        char *needle = memmem(pos, eol - pos, NEEDLE, strlen(NEEDLE));

        /* If it's not there we can ignore this line */
        if (!needle)
2292
            continue;
2293 2294 2295 2296 2297

        /* id is everthing from the beginning of the line to the ':'
         * find ':' and turn it into a terminator */
        char *colon = memchr(pos, ':', needle - pos);
        if (colon == NULL)
2298
            continue;
2299 2300 2301 2302 2303
        *colon = '\0';
        char *id = pos;

        /* Path is everything after needle to the end of the line */
        *eol = '\0';
2304 2305
        char *path = strdup(needle + strlen(NEEDLE));
        if (path == NULL) {
2306
            virReportOOMError();
2307 2308
            goto cleanup;
        }
2309

2310
        if (virHashAddEntry(paths, id, path) < 0) {
2311 2312 2313
            virReportError(VIR_ERR_OPERATION_FAILED,
                           _("failed to save chardev path '%s'"),
                           path);
2314 2315 2316
            VIR_FREE(path);
            goto cleanup;
        }
2317 2318 2319 2320 2321 2322
#undef NEEDLE
    }

    ret = 0;

cleanup:
2323 2324 2325 2326 2327 2328 2329
    VIR_FREE(reply);
    return ret;
}


int qemuMonitorTextAttachPCIDiskController(qemuMonitorPtr mon,
                                           const char *bus,
2330
                                           virDevicePCIAddress *guestAddr)
2331 2332 2333 2334 2335 2336 2337 2338 2339
{
    char *cmd = NULL;
    char *reply = NULL;
    int tryOldSyntax = 0;
    int ret = -1;

try_command:
    if (virAsprintf(&cmd, "pci_add %s storage if=%s",
                    (tryOldSyntax ? "0": "pci_addr=auto"), bus) < 0) {
2340
        virReportOOMError();
2341 2342 2343
        goto cleanup;
    }

2344
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2345 2346 2347 2348 2349 2350 2351 2352 2353 2354
        goto cleanup;

    if (qemuMonitorTextParsePciAddReply(mon, reply, guestAddr) < 0) {
        if (!tryOldSyntax && strstr(reply, "invalid char in expression")) {
            VIR_FREE(reply);
            VIR_FREE(cmd);
            tryOldSyntax = 1;
            goto try_command;
        }

2355 2356
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("adding %s disk controller failed: %s"), bus, reply);
2357 2358 2359 2360 2361 2362 2363
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(cmd);
2364 2365 2366
    VIR_FREE(reply);
    return ret;
}
2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387


static int
qemudParseDriveAddReply(const char *reply,
                        virDomainDeviceDriveAddressPtr addr)
{
    char *s, *e;

    /* If the command succeeds qemu prints:
     * OK bus X, unit Y
     */

    if (!(s = strstr(reply, "OK ")))
        return -1;

    s += 3;

    if (STRPREFIX(s, "bus ")) {
        s += strlen("bus ");

        if (virStrToLong_ui(s, &e, 10, &addr->bus) == -1) {
2388
            VIR_WARN("Unable to parse bus '%s'", s);
2389 2390 2391 2392
            return -1;
        }

        if (!STRPREFIX(e, ", ")) {
2393
            VIR_WARN("Expected ', ' parsing drive_add reply '%s'", s);
2394 2395 2396 2397 2398 2399
            return -1;
        }
        s = e + 2;
    }

    if (!STRPREFIX(s, "unit ")) {
2400
        VIR_WARN("Expected 'unit ' parsing drive_add reply '%s'", s);
2401 2402 2403 2404 2405
        return -1;
    }
    s += strlen("bus ");

    if (virStrToLong_ui(s, &e, 10, &addr->unit) == -1) {
2406
        VIR_WARN("Unable to parse unit number '%s'", s);
2407 2408 2409 2410 2411 2412 2413 2414 2415
        return -1;
    }

    return 0;
}


int qemuMonitorTextAttachDrive(qemuMonitorPtr mon,
                               const char *drivestr,
2416
                               virDevicePCIAddress *controllerAddr,
2417 2418 2419 2420 2421 2422 2423 2424 2425 2426
                               virDomainDeviceDriveAddress *driveAddr)
{
    char *cmd = NULL;
    char *reply = NULL;
    int ret = -1;
    char *safe_str;
    int tryOldSyntax = 0;

    safe_str = qemuMonitorEscapeArg(drivestr);
    if (!safe_str) {
2427
        virReportOOMError();
2428 2429 2430 2431
        return -1;
    }

try_command:
2432 2433 2434 2435
    if (virAsprintf(&cmd, "drive_add %s%.2x:%.2x:%.2x %s",
                    (tryOldSyntax ? "" : "pci_addr="),
                    controllerAddr->domain, controllerAddr->bus,
                    controllerAddr->slot, safe_str) < 0) {
2436
        virReportOOMError();
2437 2438 2439
        goto cleanup;
    }

2440
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2441 2442
        goto cleanup;

2443
    if (strstr(reply, "unknown command:")) {
2444 2445
        virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                       _("drive hotplug is not supported"));
2446 2447 2448
        goto cleanup;
    }

2449 2450 2451
    if (qemudParseDriveAddReply(reply, driveAddr) < 0) {
        if (!tryOldSyntax && strstr(reply, "invalid char in expression")) {
            VIR_FREE(reply);
2452
            VIR_FREE(cmd);
2453 2454 2455
            tryOldSyntax = 1;
            goto try_command;
        }
2456 2457
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("adding %s disk failed: %s"), drivestr, reply);
2458
        goto cleanup;
2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470
    }

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    VIR_FREE(safe_str);
    return ret;
}


2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510
/*
 * The format we're after looks like this
 *
 *   (qemu) info pci
 *   Bus  0, device   0, function 0:
 *     Host bridge: PCI device 8086:1237
 *       id ""
 *   Bus  0, device   1, function 0:
 *     ISA bridge: PCI device 8086:7000
 *       id ""
 *   Bus  0, device   1, function 1:
 *     IDE controller: PCI device 8086:7010
 *       BAR4: I/O at 0xc000 [0xc00f].
 *       id ""
 *   Bus  0, device   1, function 3:
 *     Bridge: PCI device 8086:7113
 *       IRQ 9.
 *       id ""
 *   Bus  0, device   2, function 0:
 *     VGA controller: PCI device 1013:00b8
 *       BAR0: 32 bit prefetchable memory at 0xf0000000 [0xf1ffffff].
 *       BAR1: 32 bit memory at 0xf2000000 [0xf2000fff].
 *       id ""
 *   Bus  0, device   3, function 0:
 *     Ethernet controller: PCI device 8086:100e
 *      IRQ 11.
 *      BAR0: 32 bit memory at 0xf2020000 [0xf203ffff].
 *      BAR1: I/O at 0xc040 [0xc07f].
 *       id ""
 *
 * Of this, we're interesting in the vendor/product ID
 * and the bus/device/function data.
 */
#define CHECK_END(p) if (!(p)) break;
#define SKIP_TO(p, lbl)                                            \
    (p) = strstr((p), (lbl));                                      \
    if (p)                                                         \
        (p) += strlen(lbl);
#define GET_INT(p, base, val)                                           \
    if (virStrToLong_ui((p), &(p), (base), &(val)) < 0) {               \
2511 2512
        virReportError(VIR_ERR_OPERATION_FAILED,                       \
                       _("cannot parse value for %s"), #val);           \
2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525 2526 2527
        break;                                                          \
    }
#define SKIP_SPACE(p)                           \
    while (*(p) == ' ') (p)++;

int qemuMonitorTextGetAllPCIAddresses(qemuMonitorPtr mon,
                                      qemuMonitorPCIAddress **retaddrs)
{
    char *reply;
    qemuMonitorPCIAddress *addrs = NULL;
    int naddrs = 0;
    char *p;

    *retaddrs = NULL;

2528
    if (qemuMonitorHMPCommand(mon, "info pci", &reply) < 0)
2529 2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566
        return -1;

    p = reply;


    while (p) {
        unsigned int bus, slot, func, vendor, product;

        SKIP_TO(p, "  Bus");
        CHECK_END(p);
        SKIP_SPACE(p);
        GET_INT(p, 10, bus);
        CHECK_END(p);

        SKIP_TO(p, ", device");
        CHECK_END(p);
        SKIP_SPACE(p);
        GET_INT(p, 10, slot);
        CHECK_END(p);

        SKIP_TO(p, ", function");
        CHECK_END(p);
        SKIP_SPACE(p);
        GET_INT(p, 10, func);
        CHECK_END(p);

        SKIP_TO(p, "PCI device");
        CHECK_END(p);
        SKIP_SPACE(p);
        GET_INT(p, 16, vendor);
        CHECK_END(p);

        if (*p != ':')
            break;
        p++;
        GET_INT(p, 16, product);

        if (VIR_REALLOC_N(addrs, naddrs+1) < 0) {
2567
            virReportOOMError();
2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596
            goto error;
        }

        addrs[naddrs].addr.domain = 0;
        addrs[naddrs].addr.bus = bus;
        addrs[naddrs].addr.slot = slot;
        addrs[naddrs].addr.function = func;
        addrs[naddrs].vendor = vendor;
        addrs[naddrs].product = product;
        naddrs++;

        VIR_DEBUG("Got dev %d:%d:%d   %x:%x", bus, slot, func, vendor, product);
    }

    VIR_FREE(reply);

    *retaddrs = addrs;

    return naddrs;

error:
    VIR_FREE(addrs);
    VIR_FREE(reply);
    return -1;
}
#undef GET_INT
#undef SKIP_SPACE
#undef CHECK_END
#undef SKIP_TO
2597 2598


2599
int qemuMonitorTextDelDevice(qemuMonitorPtr mon,
2600
                             const char *devalias)
2601 2602 2603 2604 2605 2606
{
    char *cmd = NULL;
    char *reply = NULL;
    char *safedev;
    int ret = -1;

2607
    if (!(safedev = qemuMonitorEscapeArg(devalias))) {
2608 2609 2610 2611 2612 2613 2614 2615 2616
        virReportOOMError();
        goto cleanup;
    }

    if (virAsprintf(&cmd, "device_del %s", safedev) < 0) {
        virReportOOMError();
        goto cleanup;
    }

2617
    VIR_DEBUG("TextDelDevice devalias=%s", devalias);
2618
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2619 2620 2621
        goto cleanup;

    if (STRNEQ(reply, "")) {
2622 2623
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("detaching %s device failed: %s"), devalias, reply);
2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    VIR_FREE(safedev);
    return ret;
}


2637 2638 2639 2640 2641 2642 2643 2644 2645
int qemuMonitorTextAddDevice(qemuMonitorPtr mon,
                             const char *devicestr)
{
    char *cmd = NULL;
    char *reply = NULL;
    char *safedev;
    int ret = -1;

    if (!(safedev = qemuMonitorEscapeArg(devicestr))) {
2646
        virReportOOMError();
2647 2648 2649 2650
        goto cleanup;
    }

    if (virAsprintf(&cmd, "device_add %s", safedev) < 0) {
2651
        virReportOOMError();
2652 2653 2654
        goto cleanup;
    }

2655
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2656 2657
        goto cleanup;

2658 2659 2660 2661 2662 2663 2664 2665 2666
    /* If the host device is hotpluged first time, qemu will output
     * husb: using %s file-system with %s if the command succeeds.
     */
    if (STRPREFIX(reply, "husb: using")) {
        ret = 0;
        goto cleanup;
    }

    /* Otherwise, if the command succeeds, no output is sent. So
2667 2668
     * any non-empty string shows an error */
    if (STRNEQ(reply, "")) {
2669 2670
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("adding %s device failed: %s"), devicestr, reply);
2671 2672 2673 2674 2675 2676 2677 2678
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
2679
    VIR_FREE(safedev);
2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690 2691 2692 2693
    return ret;
}


int qemuMonitorTextAddDrive(qemuMonitorPtr mon,
                            const char *drivestr)
{
    char *cmd = NULL;
    char *reply = NULL;
    int ret = -1;
    char *safe_str;

    safe_str = qemuMonitorEscapeArg(drivestr);
    if (!safe_str) {
2694
        virReportOOMError();
2695 2696 2697 2698 2699
        return -1;
    }

    /* 'dummy' here is just a placeholder since there is no PCI
     * address required when attaching drives to a controller */
2700
    if (virAsprintf(&cmd, "drive_add dummy %s", safe_str) < 0) {
2701
        virReportOOMError();
2702 2703 2704
        goto cleanup;
    }

2705
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2706 2707
        goto cleanup;

2708
    if (strstr(reply, "unknown command:")) {
2709 2710
        virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                       _("drive hotplug is not supported"));
2711 2712 2713
        goto cleanup;
    }

2714
    if (strstr(reply, "could not open disk image")) {
2715 2716
        virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                       _("open disk image file failed"));
2717 2718 2719
        goto cleanup;
    }

2720 2721 2722 2723 2724 2725 2726 2727
    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    VIR_FREE(safe_str);
    return ret;
}
2728

2729 2730 2731 2732
/* Attempts to remove a host drive.
 * Returns 1 if unsupported, 0 if ok, and -1 on other failure */
int qemuMonitorTextDriveDel(qemuMonitorPtr mon,
                            const char *drivestr)
2733 2734 2735 2736 2737
{
    char *cmd = NULL;
    char *reply = NULL;
    char *safedev;
    int ret = -1;
2738
    VIR_DEBUG("TextDriveDel drivestr=%s", drivestr);
2739 2740 2741 2742 2743 2744

    if (!(safedev = qemuMonitorEscapeArg(drivestr))) {
        virReportOOMError();
        goto cleanup;
    }

2745
    if (virAsprintf(&cmd, "drive_del %s", safedev) < 0) {
2746 2747 2748 2749
        virReportOOMError();
        goto cleanup;
    }

2750
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2751 2752 2753
        goto cleanup;

    if (strstr(reply, "unknown command:")) {
2754
        VIR_ERROR(_("deleting drive is not supported.  "
2755 2756 2757
                    "This may leak data if disk is reassigned"));
        ret = 1;
        goto cleanup;
2758 2759 2760 2761 2762 2763

    /* (qemu) drive_del wark
     * Device 'wark' not found */
    } else if (STRPREFIX(reply, "Device '") && (strstr(reply, "not found"))) {
        /* NB: device not found errors mean the drive was auto-deleted and we
         * ignore the error */
2764
    } else if (STRNEQ(reply, "")) {
2765 2766
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("deleting %s drive failed: %s"), drivestr, reply);
2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    VIR_FREE(safedev);
    return ret;
}
2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793

int qemuMonitorTextSetDrivePassphrase(qemuMonitorPtr mon,
                                      const char *alias,
                                      const char *passphrase)
{
    char *cmd = NULL;
    char *reply = NULL;
    int ret = -1;
    char *safe_str;

    safe_str = qemuMonitorEscapeArg(passphrase);
    if (!safe_str) {
        virReportOOMError();
        return -1;
    }

2794 2795
    if (virAsprintf(&cmd, "block_passwd %s%s \"%s\"",
                    QEMU_DRIVE_HOST_PREFIX, alias, safe_str) < 0) {
2796 2797 2798 2799
        virReportOOMError();
        goto cleanup;
    }

2800
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
2801 2802
        goto cleanup;

2803
    if (strstr(reply, "unknown command:")) {
2804 2805
        virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                       _("setting disk password is not supported"));
2806 2807
        goto cleanup;
    } else if (strstr(reply, "The entered password is invalid")) {
2808 2809
        virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                       _("the disk password is incorrect"));
2810 2811 2812 2813 2814 2815 2816 2817 2818 2819 2820
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(reply);
    VIR_FREE(safe_str);
    return ret;
}
C
Chris Lalancette 已提交
2821 2822 2823

int qemuMonitorTextCreateSnapshot(qemuMonitorPtr mon, const char *name)
{
2824
    char *cmd = NULL;
C
Chris Lalancette 已提交
2825 2826
    char *reply = NULL;
    int ret = -1;
2827
    char *safename;
C
Chris Lalancette 已提交
2828

2829 2830
    if (!(safename = qemuMonitorEscapeArg(name)) ||
        virAsprintf(&cmd, "savevm \"%s\"", safename) < 0) {
C
Chris Lalancette 已提交
2831
        virReportOOMError();
2832
        goto cleanup;
C
Chris Lalancette 已提交
2833 2834
    }

2835
    if (qemuMonitorHMPCommand(mon, cmd, &reply))
C
Chris Lalancette 已提交
2836 2837 2838
        goto cleanup;

    if (strstr(reply, "Error while creating snapshot") != NULL) {
2839 2840
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("Failed to take snapshot: %s"), reply);
C
Chris Lalancette 已提交
2841 2842 2843
        goto cleanup;
    }
    else if (strstr(reply, "No block device can accept snapshots") != NULL) {
2844 2845
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                       _("this domain does not have a device to take snapshots"));
C
Chris Lalancette 已提交
2846 2847 2848
        goto cleanup;
    }
    else if (strstr(reply, "Could not open VM state file") != NULL) {
2849
        virReportError(VIR_ERR_OPERATION_FAILED, "%s", reply);
C
Chris Lalancette 已提交
2850 2851 2852 2853
        goto cleanup;
    }
    else if (strstr(reply, "Error") != NULL
             && strstr(reply, "while writing VM") != NULL) {
2854
        virReportError(VIR_ERR_OPERATION_FAILED, "%s", reply);
C
Chris Lalancette 已提交
2855 2856 2857 2858 2859 2860
        goto cleanup;
    }

    ret = 0;

cleanup:
2861
    VIR_FREE(safename);
C
Chris Lalancette 已提交
2862 2863 2864 2865 2866 2867 2868
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}

int qemuMonitorTextLoadSnapshot(qemuMonitorPtr mon, const char *name)
{
2869
    char *cmd = NULL;
C
Chris Lalancette 已提交
2870 2871
    char *reply = NULL;
    int ret = -1;
2872
    char *safename;
C
Chris Lalancette 已提交
2873

2874 2875
    if (!(safename = qemuMonitorEscapeArg(name)) ||
        virAsprintf(&cmd, "loadvm \"%s\"", safename) < 0) {
C
Chris Lalancette 已提交
2876
        virReportOOMError();
2877
        goto cleanup;
C
Chris Lalancette 已提交
2878 2879
    }

2880
    if (qemuMonitorHMPCommand(mon, cmd, &reply))
C
Chris Lalancette 已提交
2881 2882 2883
        goto cleanup;

    if (strstr(reply, "No block device supports snapshots") != NULL) {
2884 2885
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                       _("this domain does not have a device to load snapshots"));
C
Chris Lalancette 已提交
2886 2887 2888
        goto cleanup;
    }
    else if (strstr(reply, "Could not find snapshot") != NULL) {
2889 2890 2891
        virReportError(VIR_ERR_OPERATION_INVALID,
                       _("the snapshot '%s' does not exist, and was not loaded"),
                       name);
C
Chris Lalancette 已提交
2892 2893 2894
        goto cleanup;
    }
    else if (strstr(reply, "Snapshots not supported on device") != NULL) {
2895
        virReportError(VIR_ERR_OPERATION_INVALID, "%s", reply);
C
Chris Lalancette 已提交
2896 2897 2898
        goto cleanup;
    }
    else if (strstr(reply, "Could not open VM state file") != NULL) {
2899
        virReportError(VIR_ERR_OPERATION_FAILED, "%s", reply);
C
Chris Lalancette 已提交
2900 2901 2902 2903
        goto cleanup;
    }
    else if (strstr(reply, "Error") != NULL
             && strstr(reply, "while loading VM state") != NULL) {
2904
        virReportError(VIR_ERR_OPERATION_FAILED, "%s", reply);
C
Chris Lalancette 已提交
2905 2906 2907 2908
        goto cleanup;
    }
    else if (strstr(reply, "Error") != NULL
             && strstr(reply, "while activating snapshot on") != NULL) {
2909
        virReportError(VIR_ERR_OPERATION_FAILED, "%s", reply);
C
Chris Lalancette 已提交
2910 2911 2912 2913 2914 2915
        goto cleanup;
    }

    ret = 0;

cleanup:
2916
    VIR_FREE(safename);
C
Chris Lalancette 已提交
2917 2918 2919 2920 2921 2922 2923
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}

int qemuMonitorTextDeleteSnapshot(qemuMonitorPtr mon, const char *name)
{
2924
    char *cmd = NULL;
C
Chris Lalancette 已提交
2925 2926
    char *reply = NULL;
    int ret = -1;
2927
    char *safename;
C
Chris Lalancette 已提交
2928

2929 2930
    if (!(safename = qemuMonitorEscapeArg(name)) ||
        virAsprintf(&cmd, "delvm \"%s\"", safename) < 0) {
C
Chris Lalancette 已提交
2931
        virReportOOMError();
2932
        goto cleanup;
C
Chris Lalancette 已提交
2933
    }
2934
    if (qemuMonitorHMPCommand(mon, cmd, &reply))
C
Chris Lalancette 已提交
2935 2936 2937
        goto cleanup;

    if (strstr(reply, "No block device supports snapshots") != NULL) {
2938 2939
        virReportError(VIR_ERR_OPERATION_INVALID, "%s",
                       _("this domain does not have a device to delete snapshots"));
C
Chris Lalancette 已提交
2940 2941 2942
        goto cleanup;
    }
    else if (strstr(reply, "Snapshots not supported on device") != NULL) {
2943
        virReportError(VIR_ERR_OPERATION_INVALID, "%s", reply);
C
Chris Lalancette 已提交
2944 2945 2946 2947
        goto cleanup;
    }
    else if (strstr(reply, "Error") != NULL
             && strstr(reply, "while deleting snapshot") != NULL) {
2948
        virReportError(VIR_ERR_OPERATION_FAILED, "%s", reply);
C
Chris Lalancette 已提交
2949 2950 2951 2952 2953 2954
        goto cleanup;
    }

    ret = 0;

cleanup:
2955
    VIR_FREE(safename);
C
Chris Lalancette 已提交
2956 2957 2958 2959
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}
2960

2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974 2975
int
qemuMonitorTextDiskSnapshot(qemuMonitorPtr mon, const char *device,
                            const char *file)
{
    char *cmd = NULL;
    char *reply = NULL;
    int ret = -1;
    char *safename;

    if (!(safename = qemuMonitorEscapeArg(file)) ||
        virAsprintf(&cmd, "snapshot_blkdev %s \"%s\"", device, safename) < 0) {
        virReportOOMError();
        goto cleanup;
    }

2976
    if (qemuMonitorHMPCommand(mon, cmd, &reply))
2977 2978
        goto cleanup;

2979 2980
    if (strstr(reply, "error while creating qcow2") != NULL ||
        strstr(reply, "unknown command:") != NULL) {
2981 2982
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("Failed to take snapshot: %s"), reply);
2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998
        goto cleanup;
    }

    /* XXX Should we scrape 'info block' output for
     * 'device:... file=name backing_file=oldname' to make sure the
     * command succeeded?  */

    ret = 0;

cleanup:
    VIR_FREE(safename);
    VIR_FREE(cmd);
    VIR_FREE(reply);
    return ret;
}

2999 3000 3001 3002 3003 3004 3005 3006 3007 3008 3009
int qemuMonitorTextArbitraryCommand(qemuMonitorPtr mon, const char *cmd,
                                    char **reply)
{
    char *safecmd = NULL;
    int ret;

    if (!(safecmd = qemuMonitorEscapeArg(cmd))) {
        virReportOOMError();
        return -1;
    }

3010
    ret = qemuMonitorHMPCommand(mon, safecmd, reply);
3011 3012 3013 3014 3015

    VIR_FREE(safecmd);

    return ret;
}
3016 3017 3018 3019 3020

int qemuMonitorTextInjectNMI(qemuMonitorPtr mon)
{
    char *reply = NULL;

3021 3022
    if (qemuMonitorHMPCommand(mon, "inject-nmi", &reply) < 0)
        return -1;
3023 3024 3025 3026 3027

    if (strstr(reply, "unknown command") != NULL) {
        VIR_FREE(reply);

        /* fallback to 'nmi' if qemu has not supported "inject-nmi" yet. */
3028 3029
        if (qemuMonitorHMPCommand(mon, "nmi 0", &reply) < 0)
            return -1;
3030 3031 3032 3033 3034
    }

    VIR_FREE(reply);
    return 0;
}
3035

3036 3037 3038 3039 3040 3041 3042 3043
int qemuMonitorTextSendKey(qemuMonitorPtr mon,
                           unsigned int holdtime,
                           unsigned int *keycodes,
                           unsigned int nkeycodes)
{
    int i;
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    char *cmd, *reply = NULL;
3044
    int ret = -1;
3045 3046 3047 3048 3049 3050 3051

    if (nkeycodes > VIR_DOMAIN_SEND_KEY_MAX_KEYS || nkeycodes == 0)
        return -1;

    virBufferAddLit(&buf, "sendkey ");
    for (i = 0; i < nkeycodes; i++) {
        if (keycodes[i] > 0xffff) {
3052 3053 3054
            virReportError(VIR_ERR_OPERATION_FAILED,
                           _("keycode %d is invalid: 0x%X"),
                           i, keycodes[i]);
3055 3056 3057 3058 3059 3060 3061 3062 3063 3064 3065 3066 3067 3068 3069 3070 3071 3072
            virBufferFreeAndReset(&buf);
            return -1;
        }

        if (i)
            virBufferAddChar(&buf, '-');
        virBufferAsprintf(&buf, "0x%02X", keycodes[i]);
    }

    if (holdtime)
        virBufferAsprintf(&buf, " %u", holdtime);

    if (virBufferError(&buf)) {
        virReportOOMError();
        return -1;
    }

    cmd = virBufferContentAndReset(&buf);
3073
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
3074 3075 3076
        goto cleanup;

    if (STRNEQ(reply, "")) {
3077 3078
        virReportError(VIR_ERR_OPERATION_FAILED,
                       _("failed to send key '%s'"), reply);
3079
        goto cleanup;
3080 3081
    }

3082 3083 3084
    ret = 0;

cleanup:
3085 3086
    VIR_FREE(cmd);
    VIR_FREE(reply);
3087
    return ret;
3088 3089
}

3090 3091 3092 3093 3094 3095 3096 3097 3098 3099 3100 3101
/* Returns -1 on error, -2 if not supported */
int qemuMonitorTextScreendump(qemuMonitorPtr mon, const char *file)
{
    char *cmd = NULL;
    char *reply = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "screendump %s", file) < 0){
        virReportOOMError();
        goto cleanup;
    }

3102
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116
        goto cleanup;

    if (strstr(reply, "unknown command:")) {
        ret = -2;
        goto cleanup;
    }

    ret = 0;

cleanup:
    VIR_FREE(reply);
    VIR_FREE(cmd);
    return ret;
}
3117

3118 3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132

int qemuMonitorTextOpenGraphics(qemuMonitorPtr mon,
                                const char *protocol,
                                const char *fdname,
                                bool skipauth)
{
    char *cmd = NULL;
    char *reply = NULL;
    int ret = -1;

    if (virAsprintf(&cmd, "add_client %s %s %d", protocol, fdname, skipauth ? 0 : 1) < 0){
        virReportOOMError();
        goto cleanup;
    }

3133
    if (qemuMonitorHMPCommand(mon, cmd, &reply) < 0)
3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145
        goto cleanup;

    if (STRNEQ(reply, ""))
        goto cleanup;

    ret = 0;

cleanup:
    VIR_FREE(reply);
    VIR_FREE(cmd);
    return ret;
}
3146 3147 3148 3149 3150 3151 3152 3153


int qemuMonitorTextSetBlockIoThrottle(qemuMonitorPtr mon,
                                      const char *device,
                                      virDomainBlockIoTuneInfoPtr info)
{
    char *cmd = NULL;
    char *result = NULL;
3154
    int ret = -1;
3155 3156 3157 3158
    const char *cmd_name = NULL;

    /* For the not specified fields, 0 by default */
    cmd_name = "block_set_io_throttle";
3159 3160 3161 3162
    if (virAsprintf(&cmd, "%s %s %llu %llu %llu %llu %llu %llu", cmd_name,
                    device, info->total_bytes_sec, info->read_bytes_sec,
                    info->write_bytes_sec, info->total_iops_sec,
                    info->read_iops_sec, info->write_iops_sec) < 0) {
3163
        virReportOOMError();
3164
        goto cleanup;
3165 3166
    }

3167
    if (qemuMonitorHMPCommand(mon, cmd, &result) < 0)
3168 3169 3170
        goto cleanup;

    if (qemuMonitorTextCommandNotFound(cmd_name, result)) {
3171 3172
        virReportError(VIR_ERR_OPERATION_INVALID,
                       _("Command '%s' is not found"), cmd_name);
3173 3174
        goto cleanup;
    }
3175
    ret = 0;
3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226 3227 3228 3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253

cleanup:
    VIR_FREE(cmd);
    VIR_FREE(result);
    return ret;
}

static int
qemuMonitorTextParseBlockIoThrottle(const char *result,
                                    const char *device,
                                    virDomainBlockIoTuneInfoPtr reply)
{
    char *dummy = NULL;
    int ret = -1;
    const char *p, *eol;
    int devnamelen = strlen(device);

    p = result;

    while (*p) {
        if (STRPREFIX(p, QEMU_DRIVE_HOST_PREFIX))
            p += strlen(QEMU_DRIVE_HOST_PREFIX);

        if (STREQLEN(p, device, devnamelen) &&
            p[devnamelen] == ':' && p[devnamelen+1] == ' ') {

            eol = strchr(p, '\n');
            if (!eol)
                eol = p + strlen(p);

            p += devnamelen + 2; /* Skip to first label. */

            while (*p) {
                if (STRPREFIX(p, "bps=")) {
                    p += strlen("bps=");
                    if (virStrToLong_ull(p, &dummy, 10, &reply->total_bytes_sec) == -1)
                        VIR_DEBUG("error reading total_bytes_sec: %s", p);
                } else if (STRPREFIX(p, "bps_rd=")) {
                    p += strlen("bps_rd=");
                    if (virStrToLong_ull(p, &dummy, 10, &reply->read_bytes_sec)  == -1)
                        VIR_DEBUG("error reading read_bytes_sec: %s", p);
                } else if (STRPREFIX(p, "bps_wr=")) {
                    p += strlen("bps_wr=");
                    if (virStrToLong_ull(p, &dummy, 10, &reply->write_bytes_sec) == -1)
                        VIR_DEBUG("error reading write_bytes_sec: %s", p);
                } else if (STRPREFIX(p, "iops=")) {
                    p += strlen("iops=");
                    if (virStrToLong_ull(p, &dummy, 10, &reply->total_iops_sec) == -1)
                        VIR_DEBUG("error reading total_iops_sec: %s", p);
                } else if (STRPREFIX(p, "iops_rd=")) {
                    p += strlen("iops_rd=");
                    if (virStrToLong_ull(p, &dummy, 10, &reply->read_iops_sec) == -1)
                        VIR_DEBUG("error reading read_iops_sec: %s", p);
                } else if (STRPREFIX(p, "iops_wr=")) {
                    p += strlen("iops_wr=");
                    if (virStrToLong_ull(p, &dummy, 10, &reply->write_iops_sec) == -1)
                        VIR_DEBUG("error reading write_iops_sec: %s", p);
                } else {
                    VIR_DEBUG(" unknown block info %s", p);
                }

                /* Skip to next label. */
                p = strchr (p, ' ');
                if (!p || p >= eol)
                    break;
                p++;
            }
            ret = 0;
            goto cleanup;
        }

        /* Skip to next line. */
        p = strchr (p, '\n');
        if (!p)
            break;
        p++;
    }

3254 3255
    virReportError(VIR_ERR_INVALID_ARG,
                   _("No info for device '%s'"), device);
3256 3257 3258 3259 3260 3261 3262 3263 3264 3265

cleanup:
    return ret;
}

int qemuMonitorTextGetBlockIoThrottle(qemuMonitorPtr mon,
                                      const char *device,
                                      virDomainBlockIoTuneInfoPtr reply)
{
    char *result = NULL;
3266
    int ret = -1;
3267 3268
    const char *cmd_name = "info block";

3269
    if (qemuMonitorHMPCommand(mon, cmd_name, &result) < 0)
3270 3271 3272
        goto cleanup;

    if (qemuMonitorTextCommandNotFound(cmd_name, result)) {
3273 3274
        virReportError(VIR_ERR_OPERATION_INVALID,
                       _("Command '%s' is not found"), cmd_name);
3275 3276 3277 3278 3279 3280 3281 3282 3283
        goto cleanup;
    }

    ret = qemuMonitorTextParseBlockIoThrottle(result, device, reply);

cleanup:
    VIR_FREE(result);
    return ret;
}