qemu_agent.c 70.4 KB
Newer Older
D
Daniel P. Berrange 已提交
1
/*
2
 * qemu_agent.c: interaction with QEMU guest agent
D
Daniel P. Berrange 已提交
3
 *
4
 * Copyright (C) 2006-2014 Red Hat, Inc.
D
Daniel P. Berrange 已提交
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
18
 * License along with this library.  If not, see
O
Osier Yang 已提交
19
 * <http://www.gnu.org/licenses/>.
D
Daniel P. Berrange 已提交
20 21 22 23 24 25 26 27
 */

#include <config.h>

#include <poll.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/time.h>
28
#include <gio/gio.h>
D
Daniel P. Berrange 已提交
29 30

#include "qemu_agent.h"
31
#include "qemu_domain.h"
32
#include "viralloc.h"
33
#include "virlog.h"
34
#include "virerror.h"
35
#include "virjson.h"
D
Daniel P. Berrange 已提交
36
#include "virfile.h"
37
#include "virprocess.h"
38
#include "virtime.h"
39
#include "virobject.h"
40
#include "virstring.h"
41
#include "virenum.h"
42
#include "virsocket.h"
J
Ján Tomko 已提交
43
#include "virutil.h"
D
Daniel P. Berrange 已提交
44 45 46

#define VIR_FROM_THIS VIR_FROM_QEMU

47 48
VIR_LOG_INIT("qemu.qemu_agent");

D
Daniel P. Berrange 已提交
49 50 51 52 53
#define LINE_ENDING "\n"

#define DEBUG_IO 0
#define DEBUG_RAW_IO 0

54 55 56 57 58 59 60 61 62
/* We read from QEMU until seeing a \r\n pair to indicate a
 * completed reply or event. To avoid memory denial-of-service
 * though, we must have a size limit on amount of data we
 * buffer. 10 MB is large enough that it ought to cope with
 * normal QEMU replies, and small enough that we're not
 * consuming unreasonable mem.
 */
#define QEMU_AGENT_MAX_RESPONSE (10 * 1024 * 1024)

D
Daniel P. Berrange 已提交
63 64 65 66 67 68
/* When you are the first to uncomment this,
 * don't forget to uncomment the corresponding
 * part in qemuAgentIOProcessEvent as well.
 *
static struct {
    const char *type;
69
    void (*handler)(qemuAgentPtr agent, virJSONValuePtr data);
D
Daniel P. Berrange 已提交
70 71 72 73 74 75 76 77 78 79 80 81
} eventHandlers[] = {
};
*/

typedef struct _qemuAgentMessage qemuAgentMessage;
typedef qemuAgentMessage *qemuAgentMessagePtr;

struct _qemuAgentMessage {
    char *txBuffer;
    int txOffset;
    int txLength;

82
    /* Used by the JSON agent to hold reply / error */
D
Daniel P. Berrange 已提交
83 84 85 86 87
    char *rxBuffer;
    int rxLength;
    void *rxObject;

    /* True if rxBuffer / rxObject are ready, or a
88
     * fatal error occurred on the agent channel
D
Daniel P. Berrange 已提交
89 90
     */
    bool finished;
91 92
    /* true for sync command */
    bool sync;
93 94
    /* id of the issued sync comand */
    unsigned long long id;
95
    bool first;
D
Daniel P. Berrange 已提交
96 97 98 99
};


struct _qemuAgent {
100
    virObjectLockable parent;
101

D
Daniel P. Berrange 已提交
102 103 104
    virCond notify;

    int fd;
105 106 107 108

    GMainContext *context;
    GSocket *socket;
    GSource *watch;
D
Daniel P. Berrange 已提交
109

110
    bool running;
D
Daniel P. Berrange 已提交
111 112 113 114 115 116 117 118 119

    virDomainObjPtr vm;

    qemuAgentCallbacksPtr cb;

    /* If there's a command being processed this will be
     * non-NULL */
    qemuAgentMessagePtr msg;

120
    /* Buffer incoming data ready for agent
D
Daniel P. Berrange 已提交
121 122 123 124 125 126
     * code to process & find message boundaries */
    size_t bufferOffset;
    size_t bufferLength;
    char *buffer;

    /* If anything went wrong, this will be fed back
127
     * the next agent msg */
D
Daniel P. Berrange 已提交
128
    virError lastError;
129 130

    /* Some guest agent commands don't return anything
131
     * but fire up an event on qemu agent instead.
132 133
     * Take that as indication of successful completion */
    qemuAgentEvent await_event;
134
    int timeout;
D
Daniel P. Berrange 已提交
135 136
};

137 138 139 140 141
static virClassPtr qemuAgentClass;
static void qemuAgentDispose(void *obj);

static int qemuAgentOnceInit(void)
{
142
    if (!VIR_CLASS_NEW(qemuAgent, virClassForObjectLockable()))
143 144 145 146 147
        return -1;

    return 0;
}

148
VIR_ONCE_GLOBAL_INIT(qemuAgent);
149 150


D
Daniel P. Berrange 已提交
151 152 153 154
#if DEBUG_RAW_IO
static char *
qemuAgentEscapeNonPrintable(const char *text)
{
155
    size_t i;
D
Daniel P. Berrange 已提交
156
    virBuffer buf = VIR_BUFFER_INITIALIZER;
157
    for (i = 0; text[i] != '\0'; i++) {
D
Daniel P. Berrange 已提交
158 159
        if (text[i] == '\\')
            virBufferAddLit(&buf, "\\\\");
160
        else if (g_ascii_isprint(text[i]) || text[i] == '\n' ||
D
Daniel P. Berrange 已提交
161 162 163 164 165 166 167 168 169 170
                 (text[i] == '\r' && text[i+1] == '\n'))
            virBufferAddChar(&buf, text[i]);
        else
            virBufferAsprintf(&buf, "\\x%02x", text[i]);
    }
    return virBufferContentAndReset(&buf);
}
#endif


171
static void qemuAgentDispose(void *obj)
D
Daniel P. Berrange 已提交
172
{
173 174 175 176 177 178
    qemuAgentPtr agent = obj;
    VIR_DEBUG("agent=%p", agent);
    if (agent->cb && agent->cb->destroy)
        (agent->cb->destroy)(agent, agent->vm);
    virCondDestroy(&agent->notify);
    VIR_FREE(agent->buffer);
179
    g_main_context_unref(agent->context);
180
    virResetError(&agent->lastError);
D
Daniel P. Berrange 已提交
181 182 183
}

static int
184
qemuAgentOpenUnix(const char *socketpath)
D
Daniel P. Berrange 已提交
185 186
{
    struct sockaddr_un addr;
187
    int agentfd;
188
    int ret = -1;
D
Daniel P. Berrange 已提交
189

190
    if ((agentfd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
D
Daniel P. Berrange 已提交
191 192 193 194 195
        virReportSystemError(errno,
                             "%s", _("failed to create socket"));
        return -1;
    }

196
    if (virSetCloseExec(agentfd) < 0) {
197
        virReportSystemError(errno, "%s",
198
                             _("Unable to set agent "
199
                               "close-on-exec flag"));
D
Daniel P. Berrange 已提交
200 201 202 203 204
        goto error;
    }

    memset(&addr, 0, sizeof(addr));
    addr.sun_family = AF_UNIX;
205
    if (virStrcpyStatic(addr.sun_path, socketpath) < 0) {
206
        virReportError(VIR_ERR_INTERNAL_ERROR,
207
                       _("Socket path %s too big for destination"), socketpath);
D
Daniel P. Berrange 已提交
208 209 210
        goto error;
    }

211
    ret = connect(agentfd, (struct sockaddr *)&addr, sizeof(addr));
212
    if (ret < 0) {
D
Daniel P. Berrange 已提交
213
        virReportSystemError(errno, "%s",
214
                             _("failed to connect to agent socket"));
D
Daniel P. Berrange 已提交
215 216 217
        goto error;
    }

218
    return agentfd;
D
Daniel P. Berrange 已提交
219

220
 error:
221
    VIR_FORCE_CLOSE(agentfd);
D
Daniel P. Berrange 已提交
222 223 224 225 226
    return -1;
}


static int
227
qemuAgentIOProcessEvent(qemuAgentPtr agent,
D
Daniel P. Berrange 已提交
228 229 230
                        virJSONValuePtr obj)
{
    const char *type;
231
    VIR_DEBUG("agent=%p obj=%p", agent, obj);
D
Daniel P. Berrange 已提交
232 233 234 235 236 237 238 239 240

    type = virJSONValueObjectGetString(obj, "event");
    if (!type) {
        VIR_WARN("missing event type in message");
        errno = EINVAL;
        return -1;
    }

/*
241
    for (i = 0; i < G_N_ELEMENTS(eventHandlers); i++) {
D
Daniel P. Berrange 已提交
242 243 244 245
        if (STREQ(eventHandlers[i].type, type)) {
            virJSONValuePtr data = virJSONValueObjectGet(obj, "data");
            VIR_DEBUG("handle %s handler=%p data=%p", type,
                      eventHandlers[i].handler, data);
246
            (eventHandlers[i].handler)(agent, data);
D
Daniel P. Berrange 已提交
247 248 249 250 251 252 253 254
            break;
        }
    }
*/
    return 0;
}

static int
255
qemuAgentIOProcessLine(qemuAgentPtr agent,
D
Daniel P. Berrange 已提交
256 257 258 259 260 261 262 263
                       const char *line,
                       qemuAgentMessagePtr msg)
{
    virJSONValuePtr obj = NULL;
    int ret = -1;

    VIR_DEBUG("Line [%s]", line);

264
    if (!(obj = virJSONValueFromString(line))) {
265 266
        /* receiving garbage on first sync is regular situation */
        if (msg && msg->sync && msg->first) {
267 268 269 270 271
            VIR_DEBUG("Received garbage on sync");
            msg->finished = 1;
            return 0;
        }

D
Daniel P. Berrange 已提交
272
        goto cleanup;
273
    }
D
Daniel P. Berrange 已提交
274

275
    if (virJSONValueGetType(obj) != VIR_JSON_TYPE_OBJECT) {
276 277
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Parsed JSON reply '%s' isn't an object"), line);
D
Daniel P. Berrange 已提交
278 279 280 281 282 283
        goto cleanup;
    }

    if (virJSONValueObjectHasKey(obj, "QMP") == 1) {
        ret = 0;
    } else if (virJSONValueObjectHasKey(obj, "event") == 1) {
284
        ret = qemuAgentIOProcessEvent(agent, obj);
D
Daniel P. Berrange 已提交
285 286 287
    } else if (virJSONValueObjectHasKey(obj, "error") == 1 ||
               virJSONValueObjectHasKey(obj, "return") == 1) {
        if (msg) {
288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305
            if (msg->sync) {
                unsigned long long id;

                if (virJSONValueObjectGetNumberUlong(obj, "return", &id) < 0) {
                    VIR_DEBUG("Ignoring delayed reply on sync");
                    ret = 0;
                    goto cleanup;
                }

                VIR_DEBUG("Guest returned ID: %llu", id);

                if (msg->id != id) {
                    VIR_DEBUG("Guest agent returned ID: %llu instead of %llu",
                              id, msg->id);
                    ret = 0;
                    goto cleanup;
                }
            }
D
Daniel P. Berrange 已提交
306 307 308 309
            msg->rxObject = obj;
            msg->finished = 1;
            obj = NULL;
        } else {
310 311
            /* we are out of sync */
            VIR_DEBUG("Ignoring delayed reply");
D
Daniel P. Berrange 已提交
312
        }
313
        ret = 0;
D
Daniel P. Berrange 已提交
314
    } else {
315 316
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unknown JSON reply '%s'"), line);
D
Daniel P. Berrange 已提交
317 318
    }

319
 cleanup:
D
Daniel P. Berrange 已提交
320 321 322 323
    virJSONValueFree(obj);
    return ret;
}

324
static int qemuAgentIOProcessData(qemuAgentPtr agent,
D
Daniel P. Berrange 已提交
325 326 327 328 329
                                  char *data,
                                  size_t len,
                                  qemuAgentMessagePtr msg)
{
    int used = 0;
330
    size_t i = 0;
D
Daniel P. Berrange 已提交
331 332 333
#if DEBUG_IO
# if DEBUG_RAW_IO
    char *str1 = qemuAgentEscapeNonPrintable(data);
334
    VIR_ERROR(_("[%s]"), str1);
D
Daniel P. Berrange 已提交
335 336 337 338 339 340 341 342 343 344 345 346 347
    VIR_FREE(str1);
# else
    VIR_DEBUG("Data %zu bytes [%s]", len, data);
# endif
#endif

    while (used < len) {
        char *nl = strstr(data + used, LINE_ENDING);

        if (nl) {
            int got = nl - (data + used);
            for (i = 0; i < strlen(LINE_ENDING); i++)
                data[used + got + i] = '\0';
348
            if (qemuAgentIOProcessLine(agent, data + used, msg) < 0)
D
Daniel P. Berrange 已提交
349 350 351 352 353 354 355 356 357 358 359 360
                return -1;
            used += got + strlen(LINE_ENDING);
        } else {
            break;
        }
    }

    VIR_DEBUG("Total used %d bytes out of %zd available in buffer", used, len);
    return used;
}

/* This method processes data that has been received
361
 * from the agent. Looking for async events and
D
Daniel P. Berrange 已提交
362 363 364
 * replies/errors.
 */
static int
365
qemuAgentIOProcess(qemuAgentPtr agent)
D
Daniel P. Berrange 已提交
366 367 368 369 370 371 372
{
    int len;
    qemuAgentMessagePtr msg = NULL;

    /* See if there's a message ready for reply; that is,
     * one that has completed writing all its data.
     */
373 374
    if (agent->msg && agent->msg->txOffset == agent->msg->txLength)
        msg = agent->msg;
D
Daniel P. Berrange 已提交
375 376 377 378

#if DEBUG_IO
# if DEBUG_RAW_IO
    char *str1 = qemuAgentEscapeNonPrintable(msg ? msg->txBuffer : "");
379
    char *str2 = qemuAgentEscapeNonPrintable(agent->buffer);
D
Daniel P. Berrange 已提交
380
    VIR_ERROR(_("Process %zu %p %p [[[%s]]][[[%s]]]"),
381
              agent->bufferOffset, agent->msg, msg, str1, str2);
D
Daniel P. Berrange 已提交
382 383 384
    VIR_FREE(str1);
    VIR_FREE(str2);
# else
385
    VIR_DEBUG("Process %zu", agent->bufferOffset);
D
Daniel P. Berrange 已提交
386 387 388
# endif
#endif

389 390
    len = qemuAgentIOProcessData(agent,
                                 agent->buffer, agent->bufferOffset,
D
Daniel P. Berrange 已提交
391 392 393 394 395
                                 msg);

    if (len < 0)
        return -1;

396 397 398
    if (len < agent->bufferOffset) {
        memmove(agent->buffer, agent->buffer + len, agent->bufferOffset - len);
        agent->bufferOffset -= len;
D
Daniel P. Berrange 已提交
399
    } else {
400 401
        VIR_FREE(agent->buffer);
        agent->bufferOffset = agent->bufferLength = 0;
D
Daniel P. Berrange 已提交
402 403
    }
#if DEBUG_IO
404
    VIR_DEBUG("Process done %zu used %d", agent->bufferOffset, len);
D
Daniel P. Berrange 已提交
405 406
#endif
    if (msg && msg->finished)
407
        virCondBroadcast(&agent->notify);
D
Daniel P. Berrange 已提交
408 409 410 411 412
    return len;
}


/*
413 414
 * Called when the agent is able to write data
 * Call this function while holding the agent lock.
D
Daniel P. Berrange 已提交
415 416
 */
static int
417
qemuAgentIOWrite(qemuAgentPtr agent)
D
Daniel P. Berrange 已提交
418 419 420 421
{
    int done;

    /* If no active message, or fully transmitted, then no-op */
422
    if (!agent->msg || agent->msg->txOffset == agent->msg->txLength)
D
Daniel P. Berrange 已提交
423 424
        return 0;

425 426 427
    done = safewrite(agent->fd,
                     agent->msg->txBuffer + agent->msg->txOffset,
                     agent->msg->txLength - agent->msg->txOffset);
D
Daniel P. Berrange 已提交
428 429 430 431 432 433

    if (done < 0) {
        if (errno == EAGAIN)
            return 0;

        virReportSystemError(errno, "%s",
434
                             _("Unable to write to agent"));
D
Daniel P. Berrange 已提交
435 436
        return -1;
    }
437
    agent->msg->txOffset += done;
D
Daniel P. Berrange 已提交
438 439 440 441
    return done;
}

/*
442 443
 * Called when the agent has incoming data to read
 * Call this function while holding the agent lock.
D
Daniel P. Berrange 已提交
444 445 446 447
 *
 * Returns -1 on error, or number of bytes read
 */
static int
448
qemuAgentIORead(qemuAgentPtr agent)
D
Daniel P. Berrange 已提交
449
{
450
    size_t avail = agent->bufferLength - agent->bufferOffset;
D
Daniel P. Berrange 已提交
451 452 453
    int ret = 0;

    if (avail < 1024) {
454
        if (agent->bufferLength >= QEMU_AGENT_MAX_RESPONSE) {
455 456 457 458 459
            virReportSystemError(ERANGE,
                                 _("No complete agent response found in %d bytes"),
                                 QEMU_AGENT_MAX_RESPONSE);
            return -1;
        }
460 461
        if (VIR_REALLOC_N(agent->buffer,
                          agent->bufferLength + 1024) < 0)
D
Daniel P. Berrange 已提交
462
            return -1;
463
        agent->bufferLength += 1024;
D
Daniel P. Berrange 已提交
464 465 466 467 468 469 470
        avail += 1024;
    }

    /* Read as much as we can get into our buffer,
       until we block on EAGAIN, or hit EOF */
    while (avail > 1) {
        int got;
471 472
        got = read(agent->fd,
                   agent->buffer + agent->bufferOffset,
D
Daniel P. Berrange 已提交
473 474 475 476 477
                   avail - 1);
        if (got < 0) {
            if (errno == EAGAIN)
                break;
            virReportSystemError(errno, "%s",
478
                                 _("Unable to read from agent"));
D
Daniel P. Berrange 已提交
479 480 481 482 483 484 485 486
            ret = -1;
            break;
        }
        if (got == 0)
            break;

        ret += got;
        avail -= got;
487 488
        agent->bufferOffset += got;
        agent->buffer[agent->bufferOffset] = '\0';
D
Daniel P. Berrange 已提交
489 490 491
    }

#if DEBUG_IO
492
    VIR_DEBUG("Now read %zu bytes of data", agent->bufferOffset);
D
Daniel P. Berrange 已提交
493 494 495 496 497 498
#endif

    return ret;
}


499 500 501 502
static gboolean
qemuAgentIO(GSocket *socket,
            GIOCondition cond,
            gpointer opaque);
D
Daniel P. Berrange 已提交
503

504 505 506 507 508

static void
qemuAgentRegister(qemuAgentPtr agent)
{
    GIOCondition cond = 0;
509

510
    if (agent->lastError.code == VIR_ERR_OK) {
511
        cond |= G_IO_IN;
D
Daniel P. Berrange 已提交
512

513
        if (agent->msg && agent->msg->txOffset < agent->msg->txLength)
514
            cond |= G_IO_OUT;
D
Daniel P. Berrange 已提交
515 516
    }

517 518 519 520 521 522 523 524 525 526 527 528
    agent->watch = g_socket_create_source(agent->socket,
                                        cond,
                                        NULL);

    virObjectRef(agent);
    g_source_set_callback(agent->watch,
                          (GSourceFunc)qemuAgentIO,
                          agent,
                          (GDestroyNotify)virObjectUnref);

    g_source_attach(agent->watch,
                    agent->context);
D
Daniel P. Berrange 已提交
529 530 531 532
}


static void
533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554
qemuAgentUnregister(qemuAgentPtr agent)
{
    if (agent->watch) {
        g_source_destroy(agent->watch);
        g_source_unref(agent->watch);
        agent->watch = NULL;
    }
}


static void qemuAgentUpdateWatch(qemuAgentPtr agent)
{
    qemuAgentUnregister(agent);
    if (agent->socket)
        qemuAgentRegister(agent);
}


static gboolean
qemuAgentIO(GSocket *socket G_GNUC_UNUSED,
            GIOCondition cond,
            gpointer opaque)
555
{
556
    qemuAgentPtr agent = opaque;
D
Daniel P. Berrange 已提交
557 558 559
    bool error = false;
    bool eof = false;

560 561 562
    virObjectRef(agent);
    /* lock access to the agent and protect fd */
    virObjectLock(agent);
D
Daniel P. Berrange 已提交
563
#if DEBUG_IO
564
    VIR_DEBUG("Agent %p I/O on watch %d socket %p cond %d", agent, agent->socket, cond);
D
Daniel P. Berrange 已提交
565 566
#endif

567
    if (agent->fd == -1 || !agent->watch) {
568 569
        virObjectUnlock(agent);
        virObjectUnref(agent);
570
        return G_SOURCE_REMOVE;
571 572
    }

573 574
    if (agent->lastError.code != VIR_ERR_OK) {
        if (cond & (G_IO_HUP | G_IO_ERR))
D
Daniel P. Berrange 已提交
575 576 577
            eof = true;
        error = true;
    } else {
578
        if (cond & G_IO_OUT) {
579
            if (qemuAgentIOWrite(agent) < 0)
580
                error = true;
D
Daniel P. Berrange 已提交
581 582 583
        }

        if (!error &&
584
            cond & G_IO_IN) {
585
            int got = qemuAgentIORead(agent);
D
Daniel P. Berrange 已提交
586 587 588 589 590
            if (got < 0) {
                error = true;
            } else if (got == 0) {
                eof = true;
            } else {
591
                /* Ignore hangup/error cond if we read some data, to
D
Daniel P. Berrange 已提交
592
                 * give time for that data to be consumed */
593
                cond = 0;
D
Daniel P. Berrange 已提交
594

595
                if (qemuAgentIOProcess(agent) < 0)
D
Daniel P. Berrange 已提交
596 597 598 599 600
                    error = true;
            }
        }

        if (!error &&
601
            cond & G_IO_HUP) {
602
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
603
                           _("End of file from agent socket"));
604
            eof = true;
D
Daniel P. Berrange 已提交
605 606 607
        }

        if (!error && !eof &&
608
            cond & G_IO_ERR) {
609
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
610
                           _("Invalid file descriptor while waiting for agent"));
611
            eof = true;
D
Daniel P. Berrange 已提交
612 613 614 615
        }
    }

    if (error || eof) {
616
        if (agent->lastError.code != VIR_ERR_OK) {
D
Daniel P. Berrange 已提交
617 618 619
            /* Already have an error, so clear any new error */
            virResetLastError();
        } else {
620
            if (virGetLastErrorCode() == VIR_ERR_OK)
621
                virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
622 623
                               _("Error while processing agent IO"));
            virCopyLastError(&agent->lastError);
D
Daniel P. Berrange 已提交
624 625 626
            virResetLastError();
        }

627
        VIR_DEBUG("Error on agent %s", NULLSTR(agent->lastError.message));
D
Daniel P. Berrange 已提交
628 629
        /* If IO process resulted in an error & we have a message,
         * then wakeup that waiter */
630 631 632
        if (agent->msg && !agent->msg->finished) {
            agent->msg->finished = 1;
            virCondSignal(&agent->notify);
D
Daniel P. Berrange 已提交
633 634 635
        }
    }

636
    qemuAgentUpdateWatch(agent);
D
Daniel P. Berrange 已提交
637 638 639 640 641 642

    /* We have to unlock to avoid deadlock against command thread,
     * but is this safe ?  I think it is, because the callback
     * will try to acquire the virDomainObjPtr mutex next */
    if (eof) {
        void (*eofNotify)(qemuAgentPtr, virDomainObjPtr)
643 644
            = agent->cb->eofNotify;
        virDomainObjPtr vm = agent->vm;
D
Daniel P. Berrange 已提交
645 646

        /* Make sure anyone waiting wakes up now */
647 648 649
        virCondSignal(&agent->notify);
        virObjectUnlock(agent);
        virObjectUnref(agent);
D
Daniel P. Berrange 已提交
650
        VIR_DEBUG("Triggering EOF callback");
651
        (eofNotify)(agent, vm);
D
Daniel P. Berrange 已提交
652 653
    } else if (error) {
        void (*errorNotify)(qemuAgentPtr, virDomainObjPtr)
654 655
            = agent->cb->errorNotify;
        virDomainObjPtr vm = agent->vm;
D
Daniel P. Berrange 已提交
656 657

        /* Make sure anyone waiting wakes up now */
658 659 660
        virCondSignal(&agent->notify);
        virObjectUnlock(agent);
        virObjectUnref(agent);
D
Daniel P. Berrange 已提交
661
        VIR_DEBUG("Triggering error callback");
662
        (errorNotify)(agent, vm);
D
Daniel P. Berrange 已提交
663
    } else {
664 665
        virObjectUnlock(agent);
        virObjectUnref(agent);
D
Daniel P. Berrange 已提交
666
    }
667 668

    return G_SOURCE_REMOVE;
D
Daniel P. Berrange 已提交
669 670 671 672 673
}


qemuAgentPtr
qemuAgentOpen(virDomainObjPtr vm,
674
              const virDomainChrSourceDef *config,
675
              GMainContext *context,
D
Daniel P. Berrange 已提交
676 677
              qemuAgentCallbacksPtr cb)
{
678
    qemuAgentPtr agent;
679
    g_autoptr(GError) gerr = NULL;
D
Daniel P. Berrange 已提交
680 681

    if (!cb || !cb->eofNotify) {
682 683
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("EOF notify callback must be supplied"));
D
Daniel P. Berrange 已提交
684 685 686
        return NULL;
    }

687 688 689
    if (qemuAgentInitialize() < 0)
        return NULL;

690
    if (!(agent = virObjectLockableNew(qemuAgentClass)))
D
Daniel P. Berrange 已提交
691 692
        return NULL;

693 694 695
    agent->timeout = QEMU_DOMAIN_PRIVATE(vm)->agentTimeout;
    agent->fd = -1;
    if (virCondInit(&agent->notify) < 0) {
696
        virReportSystemError(errno, "%s",
697 698
                             _("cannot initialize agent condition"));
        virObjectUnref(agent);
D
Daniel P. Berrange 已提交
699 700
        return NULL;
    }
701 702
    agent->vm = vm;
    agent->cb = cb;
D
Daniel P. Berrange 已提交
703

704
    if (config->type != VIR_DOMAIN_CHR_TYPE_UNIX) {
705
        virReportError(VIR_ERR_INTERNAL_ERROR,
706
                       _("unable to handle agent type: %s"),
707
                       virDomainChrTypeToString(config->type));
D
Daniel P. Berrange 已提交
708 709 710
        goto cleanup;
    }

711 712
    agent->fd = qemuAgentOpenUnix(config->data.nix.path);
    if (agent->fd == -1)
D
Daniel P. Berrange 已提交
713 714
        goto cleanup;

715 716 717 718 719 720 721
    agent->context = g_main_context_ref(context);

    agent->socket = g_socket_new_from_fd(agent->fd, &gerr);
    if (!agent->socket) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unable to create socket object: %s"),
                       gerr->message);
D
Daniel P. Berrange 已提交
722 723 724
        goto cleanup;
    }

725 726
    qemuAgentRegister(agent);

727
    agent->running = true;
728
    VIR_DEBUG("New agent %p fd=%d", agent, agent->fd);
D
Daniel P. Berrange 已提交
729

730
    return agent;
D
Daniel P. Berrange 已提交
731

732
 cleanup:
D
Daniel P. Berrange 已提交
733 734 735 736 737
    /* We don't want the 'destroy' callback invoked during
     * cleanup from construction failure, because that can
     * give a double-unref on virDomainObjPtr in the caller,
     * so kill the callbacks now.
     */
738 739
    agent->cb = NULL;
    qemuAgentClose(agent);
D
Daniel P. Berrange 已提交
740 741 742 743
    return NULL;
}


744
static void
745
qemuAgentNotifyCloseLocked(qemuAgentPtr agent)
746
{
747 748
    if (agent) {
        agent->running = false;
749 750 751

        /* If there is somebody waiting for a message
         * wake him up. No message will arrive anyway. */
752 753 754
        if (agent->msg && !agent->msg->finished) {
            agent->msg->finished = 1;
            virCondSignal(&agent->notify);
755 756 757 758 759 760
        }
    }
}


void
761
qemuAgentNotifyClose(qemuAgentPtr agent)
762
{
763
    if (!agent)
764 765
        return;

766
    VIR_DEBUG("agent=%p", agent);
767

768 769 770
    virObjectLock(agent);
    qemuAgentNotifyCloseLocked(agent);
    virObjectUnlock(agent);
771 772 773
}


774
void qemuAgentClose(qemuAgentPtr agent)
D
Daniel P. Berrange 已提交
775
{
776
    if (!agent)
D
Daniel P. Berrange 已提交
777 778
        return;

779
    VIR_DEBUG("agent=%p", agent);
D
Daniel P. Berrange 已提交
780

781
    virObjectLock(agent);
D
Daniel P. Berrange 已提交
782

783 784 785 786 787
    if (agent->socket) {
        qemuAgentUnregister(agent);
        g_object_unref(agent->socket);
        agent->socket = NULL;
        agent->fd = -1;
D
Daniel P. Berrange 已提交
788 789
    }

790 791
    qemuAgentNotifyCloseLocked(agent);
    virObjectUnlock(agent);
792

793
    virObjectUnref(agent);
D
Daniel P. Berrange 已提交
794 795
}

796
#define QEMU_AGENT_WAIT_TIME 5
D
Daniel P. Berrange 已提交
797

798 799
/**
 * qemuAgentSend:
800
 * @agent: agent object
801
 * @msg: Message
802 803
 * @seconds: number of seconds to wait for the result, it can be either
 *           -2, -1, 0 or positive.
804
 *
805
 * Send @msg to agent @agent. If @seconds is equal to
806 807 808
 * VIR_DOMAIN_QEMU_AGENT_COMMAND_BLOCK(-2), this function will block forever
 * waiting for the result. The value of
 * VIR_DOMAIN_QEMU_AGENT_COMMAND_DEFAULT(-1) means use default timeout value
Y
Yuri Chornoivan 已提交
809
 * and VIR_DOMAIN_QEMU_AGENT_COMMAND_NOWAIT(0) makes this function return
810 811
 * immediately without waiting. Any positive value means the number of seconds
 * to wait for the result.
812 813 814 815 816
 *
 * Returns: 0 on success,
 *          -2 on timeout,
 *          -1 otherwise
 */
817
static int qemuAgentSend(qemuAgentPtr agent,
818
                         qemuAgentMessagePtr msg,
819
                         int seconds)
D
Daniel P. Berrange 已提交
820 821
{
    int ret = -1;
822
    unsigned long long then = 0;
D
Daniel P. Berrange 已提交
823 824

    /* Check whether qemu quit unexpectedly */
825
    if (agent->lastError.code != VIR_ERR_OK) {
D
Daniel P. Berrange 已提交
826
        VIR_DEBUG("Attempt to send command while error is set %s",
827 828
                  NULLSTR(agent->lastError.message));
        virSetError(&agent->lastError);
D
Daniel P. Berrange 已提交
829 830 831
        return -1;
    }

832 833
    if (seconds > VIR_DOMAIN_QEMU_AGENT_COMMAND_BLOCK) {
        unsigned long long now;
834 835
        if (virTimeMillisNow(&now) < 0)
            return -1;
836 837 838
        if (seconds == VIR_DOMAIN_QEMU_AGENT_COMMAND_DEFAULT)
            seconds = QEMU_AGENT_WAIT_TIME;
        then = now + seconds * 1000ull;
839 840
    }

841 842
    agent->msg = msg;
    qemuAgentUpdateWatch(agent);
D
Daniel P. Berrange 已提交
843

844 845 846
    while (!agent->msg->finished) {
        if ((then && virCondWaitUntil(&agent->notify, &agent->parent.lock, then) < 0) ||
            (!then && virCondWait(&agent->notify, &agent->parent.lock) < 0)) {
847
            if (errno == ETIMEDOUT) {
848
                virReportError(VIR_ERR_AGENT_UNRESPONSIVE, "%s",
849
                               _("Guest agent not available for now"));
850 851
                ret = -2;
            } else {
852
                virReportSystemError(errno, "%s",
853
                                     _("Unable to wait on agent socket "
854
                                       "condition"));
855
            }
D
Daniel P. Berrange 已提交
856 857 858 859
            goto cleanup;
        }
    }

860
    if (agent->lastError.code != VIR_ERR_OK) {
D
Daniel P. Berrange 已提交
861
        VIR_DEBUG("Send command resulted in error %s",
862 863
                  NULLSTR(agent->lastError.message));
        virSetError(&agent->lastError);
D
Daniel P. Berrange 已提交
864 865 866 867 868
        goto cleanup;
    }

    ret = 0;

869
 cleanup:
870 871
    agent->msg = NULL;
    qemuAgentUpdateWatch(agent);
D
Daniel P. Berrange 已提交
872 873 874 875 876

    return ret;
}


877 878
/**
 * qemuAgentGuestSync:
879
 * @agent: agent object
880 881 882 883 884 885 886 887 888
 *
 * Send guest-sync with unique ID
 * and wait for reply. If we get one, check if
 * received ID is equal to given.
 *
 * Returns: 0 on success,
 *          -1 otherwise
 */
static int
889
qemuAgentGuestSync(qemuAgentPtr agent)
890 891 892
{
    int ret = -1;
    int send_ret;
893
    unsigned long long id;
894
    qemuAgentMessage sync_msg;
895 896 897 898
    int timeout = VIR_DOMAIN_QEMU_AGENT_COMMAND_DEFAULT;

    /* if user specified a custom agent timeout that is lower than the
     * default timeout, use the shorter timeout instead */
899 900
    if ((agent->timeout >= 0) && (agent->timeout < timeout))
        timeout = agent->timeout;
901 902

    memset(&sync_msg, 0, sizeof(sync_msg));
903 904
    /* set only on first sync */
    sync_msg.first = true;
905

906
 retry:
907 908 909
    if (virTimeMillisNow(&id) < 0)
        return -1;

910 911
    sync_msg.txBuffer = g_strdup_printf("{\"execute\":\"guest-sync\", "
                                        "\"arguments\":{\"id\":%llu}}\n", id);
912 913

    sync_msg.txLength = strlen(sync_msg.txBuffer);
914
    sync_msg.sync = true;
915
    sync_msg.id = id;
916 917 918

    VIR_DEBUG("Sending guest-sync command with ID: %llu", id);

919
    send_ret = qemuAgentSend(agent, &sync_msg, timeout);
920 921 922

    VIR_DEBUG("qemuAgentSend returned: %d", send_ret);

923
    if (send_ret < 0)
924 925 926
        goto cleanup;

    if (!sync_msg.rxObject) {
927 928 929 930 931
        if (sync_msg.first) {
            VIR_FREE(sync_msg.txBuffer);
            memset(&sync_msg, 0, sizeof(sync_msg));
            goto retry;
        } else {
932
            if (agent->running)
933
                virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
934
                               _("Missing agent reply object"));
935 936 937
            else
                virReportError(VIR_ERR_AGENT_UNRESPONSIVE, "%s",
                               _("Guest agent disappeared while executing command"));
938 939
            goto cleanup;
        }
940 941 942 943
    }

    ret = 0;

944
 cleanup:
945 946 947 948 949
    virJSONValueFree(sync_msg.rxObject);
    VIR_FREE(sync_msg.txBuffer);
    return ret;
}

950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983
static const char *
qemuAgentStringifyErrorClass(const char *klass)
{
    if (STREQ_NULLABLE(klass, "BufferOverrun"))
        return "Buffer overrun";
    else if (STREQ_NULLABLE(klass, "CommandDisabled"))
        return "The command has been disabled for this instance";
    else if (STREQ_NULLABLE(klass, "CommandNotFound"))
        return "The command has not been found";
    else if (STREQ_NULLABLE(klass, "FdNotFound"))
        return "File descriptor not found";
    else if (STREQ_NULLABLE(klass, "InvalidParameter"))
        return "Invalid parameter";
    else if (STREQ_NULLABLE(klass, "InvalidParameterType"))
        return "Invalid parameter type";
    else if (STREQ_NULLABLE(klass, "InvalidParameterValue"))
        return "Invalid parameter value";
    else if (STREQ_NULLABLE(klass, "OpenFileFailed"))
        return "Cannot open file";
    else if (STREQ_NULLABLE(klass, "QgaCommandFailed"))
        return "Guest agent command failed";
    else if (STREQ_NULLABLE(klass, "QMPBadInputObjectMember"))
        return "Bad QMP input object member";
    else if (STREQ_NULLABLE(klass, "QMPExtraInputObjectMember"))
        return "Unexpected extra object member";
    else if (STREQ_NULLABLE(klass, "UndefinedError"))
        return "An undefined error has occurred";
    else if (STREQ_NULLABLE(klass, "Unsupported"))
        return "this feature or command is not currently supported";
    else if (klass)
        return klass;
    else
        return "unknown QEMU command error";
}
D
Daniel P. Berrange 已提交
984

985 986 987 988 989 990 991 992 993 994
/* Checks whether the agent reply msg is an error caused by an unsupported
 * command.
 *
 * Returns true when reply is CommandNotFound or CommandDisabled
 *         false otherwise
 */
static bool
qemuAgentErrorCommandUnsupported(virJSONValuePtr reply)
{
    const char *klass;
995 996 997 998 999 1000
    virJSONValuePtr error;

    if (!reply)
        return false;

    error = virJSONValueObjectGet(reply, "error");
1001 1002 1003 1004 1005 1006 1007 1008 1009

    if (!error)
        return false;

    klass = virJSONValueObjectGetString(error, "class");
    return STREQ_NULLABLE(klass, "CommandNotFound") ||
        STREQ_NULLABLE(klass, "CommandDisabled");
}

D
Daniel P. Berrange 已提交
1010 1011 1012 1013 1014 1015 1016 1017 1018
/* Ignoring OOM in this method, since we're already reporting
 * a more important error
 *
 * XXX see qerror.h for different klasses & fill out useful params
 */
static const char *
qemuAgentStringifyError(virJSONValuePtr error)
{
    const char *klass = virJSONValueObjectGetString(error, "class");
1019
    const char *detail = virJSONValueObjectGetString(error, "desc");
D
Daniel P. Berrange 已提交
1020 1021

    /* The QMP 'desc' field is usually sufficient for our generic
1022 1023
     * error reporting needs. However, if not present, translate
     * the class into something readable.
D
Daniel P. Berrange 已提交
1024 1025
     */
    if (!detail)
1026
        detail = qemuAgentStringifyErrorClass(klass);
D
Daniel P. Berrange 已提交
1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046

    return detail;
}

static const char *
qemuAgentCommandName(virJSONValuePtr cmd)
{
    const char *name = virJSONValueObjectGetString(cmd, "execute");
    if (name)
        return name;
    else
        return "<unknown>";
}

static int
qemuAgentCheckError(virJSONValuePtr cmd,
                    virJSONValuePtr reply)
{
    if (virJSONValueObjectHasKey(reply, "error")) {
        virJSONValuePtr error = virJSONValueObjectGet(reply, "error");
1047 1048
        char *cmdstr = virJSONValueToString(cmd, false);
        char *replystr = virJSONValueToString(reply, false);
D
Daniel P. Berrange 已提交
1049 1050

        /* Log the full JSON formatted command & error */
1051
        VIR_DEBUG("unable to execute QEMU agent command %s: %s",
1052
                  NULLSTR(cmdstr), NULLSTR(replystr));
D
Daniel P. Berrange 已提交
1053 1054 1055

        /* Only send the user the command name + friendly error */
        if (!error)
1056
            virReportError(VIR_ERR_INTERNAL_ERROR,
1057
                           _("unable to execute QEMU agent command '%s'"),
1058
                           qemuAgentCommandName(cmd));
D
Daniel P. Berrange 已提交
1059
        else
1060
            virReportError(VIR_ERR_INTERNAL_ERROR,
1061
                           _("unable to execute QEMU agent command '%s': %s"),
1062 1063
                           qemuAgentCommandName(cmd),
                           qemuAgentStringifyError(error));
D
Daniel P. Berrange 已提交
1064 1065 1066 1067 1068

        VIR_FREE(cmdstr);
        VIR_FREE(replystr);
        return -1;
    } else if (!virJSONValueObjectHasKey(reply, "return")) {
1069 1070
        char *cmdstr = virJSONValueToString(cmd, false);
        char *replystr = virJSONValueToString(reply, false);
D
Daniel P. Berrange 已提交
1071 1072

        VIR_DEBUG("Neither 'return' nor 'error' is set in the JSON reply %s: %s",
1073
                  NULLSTR(cmdstr), NULLSTR(replystr));
1074
        virReportError(VIR_ERR_INTERNAL_ERROR,
1075
                       _("unable to execute QEMU agent command '%s'"),
1076
                       qemuAgentCommandName(cmd));
D
Daniel P. Berrange 已提交
1077 1078 1079 1080 1081 1082 1083
        VIR_FREE(cmdstr);
        VIR_FREE(replystr);
        return -1;
    }
    return 0;
}

1084
static int
1085
qemuAgentCommand(qemuAgentPtr agent,
1086 1087
                 virJSONValuePtr cmd,
                 virJSONValuePtr *reply,
1088
                 bool needReply,
1089 1090 1091 1092 1093
                 int seconds)
{
    int ret = -1;
    qemuAgentMessage msg;
    char *cmdstr = NULL;
1094
    int await_event = agent->await_event;
1095 1096 1097

    *reply = NULL;

1098
    if (!agent->running) {
1099 1100 1101 1102 1103
        virReportError(VIR_ERR_AGENT_UNRESPONSIVE, "%s",
                       _("Guest agent disappeared while executing command"));
        return -1;
    }

1104
    if (qemuAgentGuestSync(agent) < 0)
1105 1106 1107 1108 1109 1110
        return -1;

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

    if (!(cmdstr = virJSONValueToString(cmd, false)))
        goto cleanup;
1111
    msg.txBuffer = g_strdup_printf("%s" LINE_ENDING, cmdstr);
1112 1113 1114 1115
    msg.txLength = strlen(msg.txBuffer);

    VIR_DEBUG("Send command '%s' for write, seconds = %d", cmdstr, seconds);

1116
    ret = qemuAgentSend(agent, &msg, seconds);
1117 1118 1119 1120 1121 1122 1123 1124

    VIR_DEBUG("Receive command reply ret=%d rxObject=%p",
              ret, msg.rxObject);

    if (ret == 0) {
        /* If we haven't obtained any reply but we wait for an
         * event, then don't report this as error */
        if (!msg.rxObject) {
1125
            if (await_event && !needReply) {
1126 1127
                VIR_DEBUG("Woken up by event %d", await_event);
            } else {
1128
                if (agent->running)
1129
                    virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
1130
                                   _("Missing agent reply object"));
1131 1132 1133
                else
                    virReportError(VIR_ERR_AGENT_UNRESPONSIVE, "%s",
                                   _("Guest agent disappeared while executing command"));
1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148
                ret = -1;
            }
        } else {
            *reply = msg.rxObject;
            ret = qemuAgentCheckError(cmd, *reply);
        }
    }

 cleanup:
    VIR_FREE(cmdstr);
    VIR_FREE(msg.txBuffer);

    return ret;
}

1149
static virJSONValuePtr G_GNUC_NULL_TERMINATED
D
Daniel P. Berrange 已提交
1150 1151 1152
qemuAgentMakeCommand(const char *cmdname,
                     ...)
{
1153
    virJSONValuePtr obj = virJSONValueNewObject();
D
Daniel P. Berrange 已提交
1154 1155 1156 1157 1158 1159
    virJSONValuePtr jargs = NULL;
    va_list args;

    va_start(args, cmdname);

    if (virJSONValueObjectAppendString(obj, "execute", cmdname) < 0)
1160
        goto error;
D
Daniel P. Berrange 已提交
1161

1162 1163
    if (virJSONValueObjectCreateVArgs(&jargs, args) < 0)
        goto error;
D
Daniel P. Berrange 已提交
1164 1165 1166

    if (jargs &&
        virJSONValueObjectAppend(obj, "arguments", jargs) < 0)
1167
        goto error;
D
Daniel P. Berrange 已提交
1168 1169 1170 1171 1172

    va_end(args);

    return obj;

1173
 error:
D
Daniel P. Berrange 已提交
1174 1175 1176 1177 1178 1179
    virJSONValueFree(obj);
    virJSONValueFree(jargs);
    va_end(args);
    return NULL;
}

1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202
static virJSONValuePtr
qemuAgentMakeStringsArray(const char **strings, unsigned int len)
{
    size_t i;
    virJSONValuePtr ret = virJSONValueNewArray(), str;

    for (i = 0; i < len; i++) {
        str = virJSONValueNewString(strings[i]);
        if (!str)
            goto error;

        if (virJSONValueArrayAppend(ret, str) < 0) {
            virJSONValueFree(str);
            goto error;
        }
    }
    return ret;

 error:
    virJSONValueFree(ret);
    return NULL;
}

1203
void qemuAgentNotifyEvent(qemuAgentPtr agent,
1204 1205
                          qemuAgentEvent event)
{
1206
    virObjectLock(agent);
1207

1208 1209 1210
    VIR_DEBUG("agent=%p event=%d await_event=%d", agent, event, agent->await_event);
    if (agent->await_event == event) {
        agent->await_event = QEMU_AGENT_EVENT_NONE;
1211
        /* somebody waiting for this event, wake him up. */
1212 1213 1214
        if (agent->msg && !agent->msg->finished) {
            agent->msg->finished = 1;
            virCondSignal(&agent->notify);
1215 1216
        }
    }
1217

1218
    virObjectUnlock(agent);
1219 1220
}

D
Daniel P. Berrange 已提交
1221 1222 1223 1224
VIR_ENUM_DECL(qemuAgentShutdownMode);

VIR_ENUM_IMPL(qemuAgentShutdownMode,
              QEMU_AGENT_SHUTDOWN_LAST,
1225 1226
              "powerdown", "reboot", "halt",
);
D
Daniel P. Berrange 已提交
1227

1228
int qemuAgentShutdown(qemuAgentPtr agent,
D
Daniel P. Berrange 已提交
1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240
                      qemuAgentShutdownMode mode)
{
    int ret = -1;
    virJSONValuePtr cmd;
    virJSONValuePtr reply = NULL;

    cmd = qemuAgentMakeCommand("guest-shutdown",
                               "s:mode", qemuAgentShutdownModeTypeToString(mode),
                               NULL);
    if (!cmd)
        return -1;

1241
    if (mode == QEMU_AGENT_SHUTDOWN_REBOOT)
1242
        agent->await_event = QEMU_AGENT_EVENT_RESET;
1243
    else
1244 1245
        agent->await_event = QEMU_AGENT_EVENT_SHUTDOWN;
    ret = qemuAgentCommand(agent, cmd, &reply, false,
1246
                           VIR_DOMAIN_QEMU_AGENT_COMMAND_SHUTDOWN);
D
Daniel P. Berrange 已提交
1247 1248 1249 1250 1251

    virJSONValueFree(cmd);
    virJSONValueFree(reply);
    return ret;
}
1252 1253 1254

/*
 * qemuAgentFSFreeze:
1255
 * @agent: agent object
1256 1257
 * @mountpoints: Array of mountpoint paths to be frozen, or NULL for all
 * @nmountpoints: Number of mountpoints to be frozen, or 0 for all
1258 1259
 *
 * Issue guest-fsfreeze-freeze command to guest agent,
1260 1261
 * which freezes file systems mounted on specified mountpoints
 * (or all file systems when @mountpoints is NULL), and returns
1262 1263 1264 1265 1266
 * number of frozen file systems on success.
 *
 * Returns: number of file system frozen on success,
 *          -1 on error.
 */
1267
int qemuAgentFSFreeze(qemuAgentPtr agent, const char **mountpoints,
1268
                      unsigned int nmountpoints)
1269 1270
{
    int ret = -1;
1271
    virJSONValuePtr cmd, arg = NULL;
1272 1273
    virJSONValuePtr reply = NULL;

1274 1275 1276 1277 1278
    if (mountpoints && nmountpoints) {
        arg = qemuAgentMakeStringsArray(mountpoints, nmountpoints);
        if (!arg)
            return -1;

1279
        cmd = qemuAgentMakeCommand("guest-fsfreeze-freeze-list",
1280
                                   "a:mountpoints", &arg, NULL);
1281 1282 1283
    } else {
        cmd = qemuAgentMakeCommand("guest-fsfreeze-freeze", NULL);
    }
1284 1285

    if (!cmd)
1286
        goto cleanup;
1287

1288
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0)
1289 1290 1291
        goto cleanup;

    if (virJSONValueObjectGetNumberInt(reply, "return", &ret) < 0) {
1292 1293
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("malformed return value"));
1294 1295
    }

1296
 cleanup:
1297
    virJSONValueFree(arg);
1298 1299 1300 1301 1302 1303 1304
    virJSONValueFree(cmd);
    virJSONValueFree(reply);
    return ret;
}

/*
 * qemuAgentFSThaw:
1305
 * @agent: agent object
1306 1307 1308 1309 1310 1311 1312 1313
 *
 * Issue guest-fsfreeze-thaw command to guest agent,
 * which unfreezes all mounted file systems and returns
 * number of thawed file systems on success.
 *
 * Returns: number of file system thawed on success,
 *          -1 on error.
 */
1314
int qemuAgentFSThaw(qemuAgentPtr agent)
1315 1316 1317 1318 1319 1320 1321 1322 1323 1324
{
    int ret = -1;
    virJSONValuePtr cmd;
    virJSONValuePtr reply = NULL;

    cmd = qemuAgentMakeCommand("guest-fsfreeze-thaw", NULL);

    if (!cmd)
        return -1;

1325
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0)
1326 1327 1328
        goto cleanup;

    if (virJSONValueObjectGetNumberInt(reply, "return", &ret) < 0) {
1329 1330
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("malformed return value"));
1331 1332
    }

1333
 cleanup:
1334 1335 1336 1337
    virJSONValueFree(cmd);
    virJSONValueFree(reply);
    return ret;
}
1338 1339 1340 1341 1342 1343 1344

VIR_ENUM_DECL(qemuAgentSuspendMode);

VIR_ENUM_IMPL(qemuAgentSuspendMode,
              VIR_NODE_SUSPEND_TARGET_LAST,
              "guest-suspend-ram",
              "guest-suspend-disk",
1345 1346
              "guest-suspend-hybrid",
);
1347 1348

int
1349
qemuAgentSuspend(qemuAgentPtr agent,
1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360
                 unsigned int target)
{
    int ret = -1;
    virJSONValuePtr cmd;
    virJSONValuePtr reply = NULL;

    cmd = qemuAgentMakeCommand(qemuAgentSuspendModeTypeToString(target),
                               NULL);
    if (!cmd)
        return -1;

1361 1362
    agent->await_event = QEMU_AGENT_EVENT_SUSPEND;
    ret = qemuAgentCommand(agent, cmd, &reply, false, agent->timeout);
1363 1364 1365 1366 1367

    virJSONValueFree(cmd);
    virJSONValueFree(reply);
    return ret;
}
1368 1369

int
1370
qemuAgentArbitraryCommand(qemuAgentPtr agent,
1371 1372 1373 1374 1375
                          const char *cmd_str,
                          char **result,
                          int timeout)
{
    int ret = -1;
1376
    virJSONValuePtr cmd = NULL;
1377 1378 1379
    virJSONValuePtr reply = NULL;

    *result = NULL;
1380 1381 1382 1383 1384 1385 1386
    if (timeout < VIR_DOMAIN_QEMU_AGENT_COMMAND_MIN) {
        virReportError(VIR_ERR_INVALID_ARG,
                       _("guest agent timeout '%d' is "
                         "less than the minimum '%d'"),
                       timeout, VIR_DOMAIN_QEMU_AGENT_COMMAND_MIN);
        goto cleanup;
    }
1387

1388 1389 1390
    if (!(cmd = virJSONValueFromString(cmd_str)))
        goto cleanup;

1391
    if ((ret = qemuAgentCommand(agent, cmd, &reply, true, timeout)) < 0)
1392
        goto cleanup;
1393

1394 1395
    if (!(*result = virJSONValueToString(reply, false)))
        ret = -1;
1396

1397

1398
 cleanup:
1399 1400 1401 1402
    virJSONValueFree(cmd);
    virJSONValueFree(reply);
    return ret;
}
M
Michal Privoznik 已提交
1403 1404

int
1405
qemuAgentFSTrim(qemuAgentPtr agent,
M
Michal Privoznik 已提交
1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417
                unsigned long long minimum)
{
    int ret = -1;
    virJSONValuePtr cmd;
    virJSONValuePtr reply = NULL;

    cmd = qemuAgentMakeCommand("guest-fstrim",
                               "U:minimum", minimum,
                               NULL);
    if (!cmd)
        return ret;

1418
    ret = qemuAgentCommand(agent, cmd, &reply, false, agent->timeout);
M
Michal Privoznik 已提交
1419 1420 1421 1422 1423

    virJSONValueFree(cmd);
    virJSONValueFree(reply);
    return ret;
}
1424 1425

int
1426
qemuAgentGetVCPUs(qemuAgentPtr agent,
1427 1428 1429
                  qemuAgentCPUInfoPtr *info)
{
    int ret = -1;
1430
    size_t i;
1431 1432 1433
    virJSONValuePtr cmd;
    virJSONValuePtr reply = NULL;
    virJSONValuePtr data = NULL;
1434
    size_t ndata;
1435 1436 1437 1438

    if (!(cmd = qemuAgentMakeCommand("guest-get-vcpus", NULL)))
        return -1;

1439
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0)
1440 1441
        goto cleanup;

1442
    if (!(data = virJSONValueObjectGetArray(reply, "return"))) {
1443 1444 1445 1446 1447
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest-get-vcpus reply was missing return data"));
        goto cleanup;
    }

1448 1449 1450 1451 1452 1453
    if (!virJSONValueIsArray(data)) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Malformed guest-get-vcpus data array"));
        goto cleanup;
    }

1454 1455
    ndata = virJSONValueArraySize(data);

1456
    if (VIR_ALLOC_N(*info, ndata) < 0)
1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491
        goto cleanup;

    for (i = 0; i < ndata; i++) {
        virJSONValuePtr entry = virJSONValueArrayGet(data, i);
        qemuAgentCPUInfoPtr in = *info + i;

        if (!entry) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("array element missing in guest-get-vcpus return "
                             "value"));
            goto cleanup;
        }

        if (virJSONValueObjectGetNumberUint(entry, "logical-id", &in->id) < 0) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("'logical-id' missing in reply of guest-get-vcpus"));
            goto cleanup;
        }

        if (virJSONValueObjectGetBoolean(entry, "online", &in->online) < 0) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("'online' missing in reply of guest-get-vcpus"));
            goto cleanup;
        }

        if (virJSONValueObjectGetBoolean(entry, "can-offline",
                                         &in->offlinable) < 0) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("'can-offline' missing in reply of guest-get-vcpus"));
            goto cleanup;
        }
    }

    ret = ndata;

1492
 cleanup:
1493 1494 1495 1496 1497
    virJSONValueFree(cmd);
    virJSONValueFree(reply);
    return ret;
}

1498 1499 1500

/* returns the value provided by the guest agent or -1 on internal error */
static int
1501
qemuAgentSetVCPUsCommand(qemuAgentPtr agent,
1502 1503 1504
                         qemuAgentCPUInfoPtr info,
                         size_t ninfo,
                         int *nmodified)
1505 1506 1507 1508 1509 1510 1511 1512
{
    int ret = -1;
    virJSONValuePtr cmd = NULL;
    virJSONValuePtr reply = NULL;
    virJSONValuePtr cpus = NULL;
    virJSONValuePtr cpu = NULL;
    size_t i;

1513 1514
    *nmodified = 0;

1515
    /* create the key data array */
1516
    cpus = virJSONValueNewArray();
1517 1518 1519 1520

    for (i = 0; i < ninfo; i++) {
        qemuAgentCPUInfoPtr in = &info[i];

1521 1522 1523 1524 1525 1526
        /* don't set state for cpus that were not touched */
        if (!in->modified)
            continue;

        (*nmodified)++;

1527
        /* create single cpu object */
1528
        cpu = virJSONValueNewObject();
1529 1530

        if (virJSONValueObjectAppendNumberInt(cpu, "logical-id", in->id) < 0)
1531
            goto cleanup;
1532 1533

        if (virJSONValueObjectAppendBoolean(cpu, "online", in->online) < 0)
1534
            goto cleanup;
1535 1536

        if (virJSONValueArrayAppend(cpus, cpu) < 0)
1537
            goto cleanup;
1538 1539 1540 1541

        cpu = NULL;
    }

1542 1543 1544 1545 1546
    if (*nmodified == 0) {
        ret = 0;
        goto cleanup;
    }

1547
    if (!(cmd = qemuAgentMakeCommand("guest-set-vcpus",
1548
                                     "a:vcpus", &cpus,
1549 1550 1551
                                     NULL)))
        goto cleanup;

1552
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0)
1553 1554
        goto cleanup;

1555 1556 1557 1558 1559
    /* All negative values are invalid. Return of 0 is bogus since we wouldn't
     * call the guest agent so that 0 cpus would be set successfully. Reporting
     * more successfully set vcpus that we've asked for is invalid. */
    if (virJSONValueObjectGetNumberInt(reply, "return", &ret) < 0 ||
        ret <= 0 || ret > *nmodified) {
1560
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
1561 1562
                       _("guest agent returned malformed or invalid return value"));
        ret = -1;
1563 1564
    }

1565
 cleanup:
1566 1567 1568 1569 1570 1571
    virJSONValueFree(cmd);
    virJSONValueFree(reply);
    virJSONValueFree(cpu);
    virJSONValueFree(cpus);
    return ret;
}
1572 1573


1574 1575 1576 1577 1578 1579 1580 1581 1582
/**
 * Set the VCPU state using guest agent.
 *
 * Attempts to set the guest agent state for all cpus or until a proper error is
 * reported by the guest agent. This may require multiple calls.
 *
 * Returns -1 on error, 0 on success.
 */
int
1583
qemuAgentSetVCPUs(qemuAgentPtr agent,
1584 1585 1586 1587 1588 1589 1590 1591
                  qemuAgentCPUInfoPtr info,
                  size_t ninfo)
{
    int rv;
    int nmodified;
    size_t i;

    do {
1592
        if ((rv = qemuAgentSetVCPUsCommand(agent, info, ninfo, &nmodified)) < 0)
1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612
            return -1;

        /* all vcpus were set successfully */
        if (rv == nmodified)
            return 0;

        /* un-mark vcpus that were already set */
        for (i = 0; i < ninfo && rv > 0; i++) {
            if (!info[i].modified)
                continue;

            info[i].modified = false;
            rv--;
        }
    } while (1);

    return 0;
}


1613 1614 1615 1616 1617 1618 1619 1620 1621
/* modify the cpu info structure to set the correct amount of cpus */
int
qemuAgentUpdateCPUInfo(unsigned int nvcpus,
                       qemuAgentCPUInfoPtr cpuinfo,
                       int ncpuinfo)
{
    size_t i;
    int nonline = 0;
    int nofflinable = 0;
1622
    ssize_t cpu0 = -1;
1623 1624 1625

    /* count the active and offlinable cpus */
    for (i = 0; i < ncpuinfo; i++) {
1626 1627 1628
        if (cpuinfo[i].id == 0)
            cpu0 = i;

1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642
        if (cpuinfo[i].online)
            nonline++;

        if (cpuinfo[i].offlinable && cpuinfo[i].online)
            nofflinable++;

        /* This shouldn't happen, but we can't trust the guest agent */
        if (!cpuinfo[i].online && !cpuinfo[i].offlinable) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("Invalid data provided by guest agent"));
            return -1;
        }
    }

1643 1644 1645 1646 1647 1648 1649 1650 1651
    /* CPU0 was made offlinable in linux a while ago, but certain parts (suspend
     * to ram) of the kernel still don't cope well with that. Make sure that if
     * all remaining vCPUs are offlinable, vCPU0 will not be selected to be
     * offlined automatically */
    if (nofflinable == nonline && cpu0 >= 0 && cpuinfo[cpu0].online) {
        cpuinfo[cpu0].offlinable = false;
        nofflinable--;
    }

1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670
    /* the guest agent reported less cpus than requested */
    if (nvcpus > ncpuinfo) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest agent reports less cpu than requested"));
        return -1;
    }

    /* not enough offlinable CPUs to support the request */
    if (nvcpus < nonline - nofflinable) {
        virReportError(VIR_ERR_INVALID_ARG, "%s",
                       _("Cannot offline enough CPUs"));
        return -1;
    }

    for (i = 0; i < ncpuinfo; i++) {
        if (nvcpus < nonline) {
            /* unplug */
            if (cpuinfo[i].offlinable && cpuinfo[i].online) {
                cpuinfo[i].online = false;
1671
                cpuinfo[i].modified = true;
1672 1673 1674 1675 1676 1677
                nonline--;
            }
        } else if (nvcpus > nonline) {
            /* plug */
            if (!cpuinfo[i].online) {
                cpuinfo[i].online = true;
1678
                cpuinfo[i].modified = true;
1679 1680 1681 1682 1683 1684 1685 1686 1687 1688
                nonline++;
            }
        } else {
            /* done */
            break;
        }
    }

    return 0;
}
1689 1690


1691
int
1692
qemuAgentGetHostname(qemuAgentPtr agent,
1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706
                     char **hostname)
{
    int ret = -1;
    virJSONValuePtr cmd;
    virJSONValuePtr reply = NULL;
    virJSONValuePtr data = NULL;
    const char *result = NULL;

    cmd = qemuAgentMakeCommand("guest-get-host-name",
                               NULL);

    if (!cmd)
        return ret;

1707
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0) {
1708 1709
        if (qemuAgentErrorCommandUnsupported(reply))
            ret = -2;
1710
        goto cleanup;
1711
    }
1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724

    if (!(data = virJSONValueObjectGet(reply, "return"))) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("malformed return value"));
        goto cleanup;
    }

    if (!(result = virJSONValueObjectGetString(data, "host-name"))) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("'host-name' missing in guest-get-host-name reply"));
        goto cleanup;
    }

1725
    *hostname = g_strdup(result);
1726 1727 1728 1729 1730 1731 1732 1733 1734 1735

    ret = 0;

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


1736
int
1737
qemuAgentGetTime(qemuAgentPtr agent,
1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750
                 long long *seconds,
                 unsigned int *nseconds)
{
    int ret = -1;
    unsigned long long json_time;
    virJSONValuePtr cmd;
    virJSONValuePtr reply = NULL;

    cmd = qemuAgentMakeCommand("guest-get-time",
                               NULL);
    if (!cmd)
        return ret;

1751
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0)
1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778
        goto cleanup;

    if (virJSONValueObjectGetNumberUlong(reply, "return", &json_time) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("malformed return value"));
        goto cleanup;
    }

    /* guest agent returns time in nanoseconds,
     * we need it in seconds here */
    *seconds = json_time / 1000000000LL;
    *nseconds = json_time % 1000000000LL;
    ret = 0;

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


/**
 * qemuAgentSetTime:
 * @setTime: time to set
 * @sync: let guest agent to read domain's RTC (@setTime is ignored)
 */
int
1779
qemuAgentSetTime(qemuAgentPtr agent,
1780 1781
                long long seconds,
                unsigned int nseconds,
P
Pavel Hrdina 已提交
1782
                bool rtcSync)
1783 1784 1785 1786 1787
{
    int ret = -1;
    virJSONValuePtr cmd;
    virJSONValuePtr reply = NULL;

P
Pavel Hrdina 已提交
1788
    if (rtcSync) {
1789 1790 1791 1792 1793 1794 1795
        cmd = qemuAgentMakeCommand("guest-set-time", NULL);
    } else {
        /* guest agent expect time with nanosecond granularity.
         * Impressing. */
        long long json_time;

        /* Check if we overflow. For some reason qemu doesn't handle unsigned
1796
         * long long on the agent well as it silently truncates numbers to
1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815
         * signed long long. Therefore we must check overflow against LLONG_MAX
         * not ULLONG_MAX. */
        if (seconds > LLONG_MAX / 1000000000LL) {
            virReportError(VIR_ERR_INVALID_ARG,
                           _("Time '%lld' is too big for guest agent"),
                           seconds);
            return ret;
        }

        json_time = seconds * 1000000000LL;
        json_time += nseconds;
        cmd = qemuAgentMakeCommand("guest-set-time",
                                   "I:time", json_time,
                                   NULL);
    }

    if (!cmd)
        return ret;

1816
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0)
1817 1818 1819 1820 1821 1822 1823 1824
        goto cleanup;

    ret = 0;
 cleanup:
    virJSONValueFree(cmd);
    virJSONValueFree(reply);
    return ret;
}
1825

1826 1827 1828 1829 1830 1831
static void
qemuAgentDiskInfoFree(qemuAgentDiskInfoPtr info)
{
    if (!info)
        return;

1832 1833 1834 1835
    g_free(info->serial);
    g_free(info->bus_type);
    g_free(info->devnode);
    g_free(info);
1836 1837
}

1838
void
1839 1840 1841 1842 1843 1844 1845
qemuAgentFSInfoFree(qemuAgentFSInfoPtr info)
{
    size_t i;

    if (!info)
        return;

1846 1847 1848
    g_free(info->mountpoint);
    g_free(info->name);
    g_free(info->fstype);
1849 1850 1851

    for (i = 0; i < info->ndisks; i++)
        qemuAgentDiskInfoFree(info->disks[i]);
1852
    g_free(info->disks);
1853

1854
    g_free(info);
1855 1856 1857
}

static int
1858
qemuAgentGetFSInfoFillDisks(virJSONValuePtr jsondisks,
1859
                            qemuAgentFSInfoPtr fsinfo)
1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871
{
    size_t ndisks;
    size_t i;

    if (!virJSONValueIsArray(jsondisks)) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Malformed guest-get-fsinfo 'disk' data array"));
        return -1;
    }

    ndisks = virJSONValueArraySize(jsondisks);

1872 1873
    if (ndisks)
        fsinfo->disks = g_new0(qemuAgentDiskInfoPtr, ndisks);
1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889
    fsinfo->ndisks = ndisks;

    for (i = 0; i < fsinfo->ndisks; i++) {
        virJSONValuePtr jsondisk = virJSONValueArrayGet(jsondisks, i);
        virJSONValuePtr pci;
        qemuAgentDiskInfoPtr disk;
        const char *val;

        if (!jsondisk) {
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("array element '%zd' of '%zd' missing in "
                             "guest-get-fsinfo 'disk' data"),
                           i, fsinfo->ndisks);
            return -1;
        }

1890
        fsinfo->disks[i] = g_new0(qemuAgentDiskInfo, 1);
1891 1892
        disk = fsinfo->disks[i];

1893 1894 1895
        if ((val = virJSONValueObjectGetString(jsondisk, "bus-type")))
            disk->bus_type = g_strdup(val);

1896 1897
        if ((val = virJSONValueObjectGetString(jsondisk, "serial")))
            disk->serial = g_strdup(val);
1898

1899 1900
        if ((val = virJSONValueObjectGetString(jsondisk, "dev")))
            disk->devnode = g_strdup(val);
1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911

#define GET_DISK_ADDR(jsonObject, var, name) \
        do { \
            if (virJSONValueObjectGetNumberUint(jsonObject, name, var) < 0) { \
                virReportError(VIR_ERR_INTERNAL_ERROR, \
                               _("'%s' missing in guest-get-fsinfo " \
                                 "'disk' data"), name); \
                return -1; \
            } \
        } while (0)

1912 1913 1914
        GET_DISK_ADDR(jsondisk, &disk->bus, "bus");
        GET_DISK_ADDR(jsondisk, &disk->target, "target");
        GET_DISK_ADDR(jsondisk, &disk->unit, "unit");
1915 1916 1917 1918 1919 1920 1921 1922

        if (!(pci = virJSONValueObjectGet(jsondisk, "pci-controller"))) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("'pci-controller' missing in guest-get-fsinfo "
                             "'disk' data"));
            return -1;
        }

1923 1924 1925 1926
        GET_DISK_ADDR(pci, &disk->pci_controller.domain, "domain");
        GET_DISK_ADDR(pci, &disk->pci_controller.bus, "bus");
        GET_DISK_ADDR(pci, &disk->pci_controller.slot, "slot");
        GET_DISK_ADDR(pci, &disk->pci_controller.function, "function");
1927 1928 1929 1930 1931 1932
#undef GET_DISK_ADDR
    }

    return 0;
}

1933 1934 1935 1936
/* Returns: 0 on success
 *          -2 when agent command is not supported by the agent
 *          -1 otherwise
 */
1937
int
1938
qemuAgentGetFSInfo(qemuAgentPtr agent,
1939
                   qemuAgentFSInfoPtr **info)
1940
{
1941
    size_t i;
1942
    int ret = -1;
J
Ján Tomko 已提交
1943 1944
    g_autoptr(virJSONValue) cmd = NULL;
    g_autoptr(virJSONValue) reply = NULL;
1945
    virJSONValuePtr data;
1946 1947
    size_t ndata = 0;
    qemuAgentFSInfoPtr *info_ret = NULL;
1948 1949 1950 1951 1952

    cmd = qemuAgentMakeCommand("guest-get-fsinfo", NULL);
    if (!cmd)
        return ret;

1953
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0) {
1954 1955
        if (qemuAgentErrorCommandUnsupported(reply))
            ret = -2;
1956
        goto cleanup;
1957
    }
1958 1959 1960 1961 1962 1963 1964

    if (!(data = virJSONValueObjectGet(reply, "return"))) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest-get-fsinfo reply was missing return data"));
        goto cleanup;
    }

1965
    if (!virJSONValueIsArray(data)) {
1966
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
1967
                       _("Malformed guest-get-fsinfo data array"));
1968 1969 1970 1971
        goto cleanup;
    }

    ndata = virJSONValueArraySize(data);
1972
    if (ndata == 0) {
1973
        ret = 0;
1974
        *info = NULL;
1975 1976
        goto cleanup;
    }
1977
    info_ret = g_new0(qemuAgentFSInfoPtr, ndata);
1978 1979 1980 1981

    for (i = 0; i < ndata; i++) {
        /* Reverse the order to arrange in mount order */
        virJSONValuePtr entry = virJSONValueArrayGet(data, ndata - 1 - i);
1982 1983 1984
        virJSONValuePtr disk;
        unsigned long long bytes_val;
        const char *result = NULL;
1985 1986 1987

        if (!entry) {
            virReportError(VIR_ERR_INTERNAL_ERROR,
1988
                           _("array element '%zd' of '%zd' missing in "
1989 1990 1991 1992 1993
                             "guest-get-fsinfo return data"),
                           i, ndata);
            goto cleanup;
        }

1994
        info_ret[i] = g_new0(qemuAgentFSInfo, 1);
1995

1996
        if (!(result = virJSONValueObjectGetString(entry, "mountpoint"))) {
1997 1998 1999 2000 2001 2002
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("'mountpoint' missing in reply of "
                             "guest-get-fsinfo"));
            goto cleanup;
        }

2003
        info_ret[i]->mountpoint = g_strdup(result);
2004 2005

        if (!(result = virJSONValueObjectGetString(entry, "name"))) {
2006 2007 2008 2009 2010
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("'name' missing in reply of guest-get-fsinfo"));
            goto cleanup;
        }

2011
        info_ret[i]->name = g_strdup(result);
2012 2013

        if (!(result = virJSONValueObjectGetString(entry, "type"))) {
2014 2015 2016 2017 2018
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("'type' missing in reply of guest-get-fsinfo"));
            goto cleanup;
        }

2019
        info_ret[i]->fstype = g_strdup(result);
2020

2021

2022 2023 2024 2025 2026 2027
        /* 'used-bytes' and 'total-bytes' were added in qemu-ga 3.0 */
        if (virJSONValueObjectHasKey(entry, "used-bytes")) {
            if (virJSONValueObjectGetNumberUlong(entry, "used-bytes",
                                                 &bytes_val) < 0) {
                virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                               _("Error getting 'used-bytes' in reply of guest-get-fsinfo"));
2028 2029
                goto cleanup;
            }
2030 2031 2032 2033
            info_ret[i]->used_bytes = bytes_val;
        } else {
            info_ret[i]->used_bytes = -1;
        }
2034

2035 2036 2037
        if (virJSONValueObjectHasKey(entry, "total-bytes")) {
            if (virJSONValueObjectGetNumberUlong(entry, "total-bytes",
                                                 &bytes_val) < 0) {
2038
                virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
2039
                               _("Error getting 'total-bytes' in reply of guest-get-fsinfo"));
2040 2041
                goto cleanup;
            }
2042 2043 2044 2045
            info_ret[i]->total_bytes = bytes_val;
        } else {
            info_ret[i]->total_bytes = -1;
        }
2046

2047 2048 2049 2050
        if (!(disk = virJSONValueObjectGet(entry, "disk"))) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("'disk' missing in reply of guest-get-fsinfo"));
            goto cleanup;
2051
        }
2052

2053
        if (qemuAgentGetFSInfoFillDisks(disk, info_ret[i]) < 0)
2054
            goto cleanup;
2055 2056
    }

2057
    *info = g_steal_pointer(&info_ret);
2058 2059 2060 2061 2062
    ret = ndata;

 cleanup:
    if (info_ret) {
        for (i = 0; i < ndata; i++)
2063
            qemuAgentFSInfoFree(info_ret[i]);
2064
        g_free(info_ret);
2065
    }
2066 2067 2068
    return ret;
}

2069 2070
/*
 * qemuAgentGetInterfaces:
2071
 * @agent: agent object
2072 2073 2074 2075 2076 2077 2078 2079 2080
 * @ifaces: pointer to an array of pointers pointing to interface objects
 *
 * Issue guest-network-get-interfaces to guest agent, which returns a
 * list of interfaces of a running domain along with their IP and MAC
 * addresses.
 *
 * Returns: number of interfaces on success, -1 on error.
 */
int
2081
qemuAgentGetInterfaces(qemuAgentPtr agent,
2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103
                       virDomainInterfacePtr **ifaces)
{
    int ret = -1;
    size_t i, j;
    virJSONValuePtr cmd = NULL;
    virJSONValuePtr reply = NULL;
    virJSONValuePtr ret_array = NULL;
    size_t ifaces_count = 0;
    size_t addrs_count = 0;
    virDomainInterfacePtr *ifaces_ret = NULL;
    virHashTablePtr ifaces_store = NULL;
    char **ifname = NULL;

    /* Hash table to handle the interface alias */
    if (!(ifaces_store = virHashCreate(ifaces_count, NULL))) {
        virHashFree(ifaces_store);
        return -1;
    }

    if (!(cmd = qemuAgentMakeCommand("guest-network-get-interfaces", NULL)))
        goto cleanup;

2104
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0)
2105 2106 2107 2108 2109 2110 2111 2112
        goto cleanup;

    if (!(ret_array = virJSONValueObjectGet(reply, "return"))) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("qemu agent didn't provide 'return' field"));
        goto cleanup;
    }

2113
    if (!virJSONValueIsArray(ret_array)) {
2114 2115 2116 2117 2118
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("qemu agent didn't return an array of interfaces"));
        goto cleanup;
    }

2119
    for (i = 0; i < virJSONValueArraySize(ret_array); i++) {
2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160
        virJSONValuePtr tmp_iface = virJSONValueArrayGet(ret_array, i);
        virJSONValuePtr ip_addr_arr = NULL;
        const char *hwaddr, *ifname_s, *name = NULL;
        virDomainInterfacePtr iface = NULL;

        /* Shouldn't happen but doesn't hurt to check neither */
        if (!tmp_iface) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("qemu agent reply missing interface entry in array"));
            goto error;
        }

        /* interface name is required to be presented */
        name = virJSONValueObjectGetString(tmp_iface, "name");
        if (!name) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("qemu agent didn't provide 'name' field"));
            goto error;
        }

        /* Handle interface alias (<ifname>:<alias>) */
        ifname = virStringSplit(name, ":", 2);
        ifname_s = ifname[0];

        iface = virHashLookup(ifaces_store, ifname_s);

        /* If the hash table doesn't contain this iface, add it */
        if (!iface) {
            if (VIR_EXPAND_N(ifaces_ret, ifaces_count, 1) < 0)
                goto error;

            if (VIR_ALLOC(ifaces_ret[ifaces_count - 1]) < 0)
                goto error;

            if (virHashAddEntry(ifaces_store, ifname_s,
                                ifaces_ret[ifaces_count - 1]) < 0)
                goto error;

            iface = ifaces_ret[ifaces_count - 1];
            iface->naddrs = 0;

2161
            iface->name = g_strdup(ifname_s);
2162 2163

            hwaddr = virJSONValueObjectGetString(tmp_iface, "hardware-address");
2164
            iface->hwaddr = g_strdup(hwaddr);
2165 2166 2167
        }

        /* Has to be freed for each interface. */
2168
        virStringListFree(ifname);
2169 2170 2171 2172 2173 2174 2175

        /* as well as IP address which - moreover -
         * can be presented multiple times */
        ip_addr_arr = virJSONValueObjectGet(tmp_iface, "ip-addresses");
        if (!ip_addr_arr)
            continue;

2176 2177 2178
        if (!virJSONValueIsArray(ip_addr_arr)) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("Malformed ip-addresses array"));
2179
            goto error;
2180
        }
2181 2182 2183 2184

        /* If current iface already exists, continue with the count */
        addrs_count = iface->naddrs;

2185
        for (j = 0; j < virJSONValueArraySize(ip_addr_arr); j++) {
2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225
            const char *type, *addr;
            virJSONValuePtr ip_addr_obj = virJSONValueArrayGet(ip_addr_arr, j);
            virDomainIPAddressPtr ip_addr;

            if (VIR_EXPAND_N(iface->addrs, addrs_count, 1)  < 0)
                goto error;

            ip_addr = &iface->addrs[addrs_count - 1];

            /* Shouldn't happen but doesn't hurt to check neither */
            if (!ip_addr_obj) {
                virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                               _("qemu agent reply missing IP addr in array"));
                goto error;
            }

            type = virJSONValueObjectGetString(ip_addr_obj, "ip-address-type");
            if (!type) {
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("qemu agent didn't provide 'ip-address-type'"
                                 " field for interface '%s'"), name);
                goto error;
            } else if (STREQ(type, "ipv4")) {
                ip_addr->type = VIR_IP_ADDR_TYPE_IPV4;
            } else if (STREQ(type, "ipv6")) {
                ip_addr->type = VIR_IP_ADDR_TYPE_IPV6;
            } else {
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("unknown ip address type '%s'"),
                               type);
                goto error;
            }

            addr = virJSONValueObjectGetString(ip_addr_obj, "ip-address");
            if (!addr) {
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("qemu agent didn't provide 'ip-address'"
                                 " field for interface '%s'"), name);
                goto error;
            }
2226
            ip_addr->addr = g_strdup(addr);
2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238

            if (virJSONValueObjectGetNumberUint(ip_addr_obj, "prefix",
                                                &ip_addr->prefix) < 0) {
                virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                               _("malformed 'prefix' field"));
                goto error;
            }
        }

        iface->naddrs = addrs_count;
    }

2239
    *ifaces = g_steal_pointer(&ifaces_ret);
2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253
    ret = ifaces_count;

 cleanup:
    virJSONValueFree(cmd);
    virJSONValueFree(reply);
    virHashFree(ifaces_store);
    return ret;

 error:
    if (ifaces_ret) {
        for (i = 0; i < ifaces_count; i++)
            virDomainInterfaceFree(ifaces_ret[i]);
    }
    VIR_FREE(ifaces_ret);
2254
    virStringListFree(ifname);
2255 2256 2257

    goto cleanup;
}
2258 2259 2260


int
2261
qemuAgentSetUserPassword(qemuAgentPtr agent,
2262 2263 2264 2265 2266 2267 2268 2269 2270
                         const char *user,
                         const char *password,
                         bool crypted)
{
    int ret = -1;
    virJSONValuePtr cmd = NULL;
    virJSONValuePtr reply = NULL;
    char *password64 = NULL;

2271 2272
    password64 = g_base64_encode((unsigned char *)password,
                                 strlen(password));
2273 2274 2275 2276 2277 2278 2279 2280

    if (!(cmd = qemuAgentMakeCommand("guest-set-user-password",
                                     "b:crypted", crypted,
                                     "s:username", user,
                                     "s:password", password64,
                                     NULL)))
        goto cleanup;

2281
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0)
2282 2283 2284 2285 2286 2287 2288 2289 2290 2291
        goto cleanup;

    ret = 0;

 cleanup:
    virJSONValueFree(cmd);
    virJSONValueFree(reply);
    VIR_FREE(password64);
    return ret;
}
2292

2293 2294 2295 2296
/* Returns: 0 on success
 *          -2 when agent command is not supported by the agent
 *          -1 otherwise
 */
2297
int
2298
qemuAgentGetUsers(qemuAgentPtr agent,
2299 2300 2301 2302
                  virTypedParameterPtr *params,
                  int *nparams,
                  int *maxparams)
{
J
Ján Tomko 已提交
2303 2304
    g_autoptr(virJSONValue) cmd = NULL;
    g_autoptr(virJSONValue) reply = NULL;
2305 2306 2307 2308 2309 2310 2311
    virJSONValuePtr data = NULL;
    size_t ndata;
    size_t i;

    if (!(cmd = qemuAgentMakeCommand("guest-get-users", NULL)))
        return -1;

2312
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0) {
2313 2314
        if (qemuAgentErrorCommandUnsupported(reply))
            return -2;
2315
        return -1;
2316
    }
2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354

    if (!(data = virJSONValueObjectGetArray(reply, "return"))) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest-get-users reply was missing return data"));
        return -1;
    }

    if (!virJSONValueIsArray(data)) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Malformed guest-get-users data array"));
        return -1;
    }

    ndata = virJSONValueArraySize(data);

    if (virTypedParamsAddUInt(params, nparams, maxparams,
                              "user.count", ndata) < 0)
        return -1;

    for (i = 0; i < ndata; i++) {
        virJSONValuePtr entry = virJSONValueArrayGet(data, i);
        char param_name[VIR_TYPED_PARAM_FIELD_LENGTH];
        const char *strvalue;
        double logintime;

        if (!entry) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("array element missing in guest-get-users return "
                             "value"));
            return -1;
        }

        if (!(strvalue = virJSONValueObjectGetString(entry, "user"))) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("'user' missing in reply of guest-get-users"));
            return -1;
        }

2355
        g_snprintf(param_name, VIR_TYPED_PARAM_FIELD_LENGTH, "user.%zu.name", i);
2356 2357 2358 2359 2360 2361
        if (virTypedParamsAddString(params, nparams, maxparams,
                                    param_name, strvalue) < 0)
            return -1;

        /* 'domain' is only present for windows guests */
        if ((strvalue = virJSONValueObjectGetString(entry, "domain"))) {
2362 2363
            g_snprintf(param_name, VIR_TYPED_PARAM_FIELD_LENGTH,
                       "user.%zu.domain", i);
2364 2365 2366 2367 2368 2369 2370 2371 2372 2373
            if (virTypedParamsAddString(params, nparams, maxparams,
                                        param_name, strvalue) < 0)
                return -1;
        }

        if (virJSONValueObjectGetNumberDouble(entry, "login-time", &logintime) < 0) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("'login-time' missing in reply of guest-get-users"));
            return -1;
        }
2374 2375
        g_snprintf(param_name, VIR_TYPED_PARAM_FIELD_LENGTH,
                   "user.%zu.login-time", i);
2376 2377 2378 2379 2380 2381 2382
        if (virTypedParamsAddULLong(params, nparams, maxparams,
                                    param_name, logintime * 1000) < 0)
            return -1;
    }

    return ndata;
}
2383

2384 2385 2386 2387
/* Returns: 0 on success
 *          -2 when agent command is not supported by the agent
 *          -1 otherwise
 */
2388
int
2389
qemuAgentGetOSInfo(qemuAgentPtr agent,
2390 2391 2392 2393
                   virTypedParameterPtr *params,
                   int *nparams,
                   int *maxparams)
{
J
Ján Tomko 已提交
2394 2395
    g_autoptr(virJSONValue) cmd = NULL;
    g_autoptr(virJSONValue) reply = NULL;
2396 2397 2398 2399 2400
    virJSONValuePtr data = NULL;

    if (!(cmd = qemuAgentMakeCommand("guest-get-osinfo", NULL)))
        return -1;

2401
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0) {
2402 2403
        if (qemuAgentErrorCommandUnsupported(reply))
            return -2;
2404
        return -1;
2405
    }
2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435

    if (!(data = virJSONValueObjectGetObject(reply, "return"))) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest-get-osinfo reply was missing return data"));
        return -1;
    }

#define OSINFO_ADD_PARAM(agent_string_, param_string_) \
    do { \
        const char *result; \
        if ((result = virJSONValueObjectGetString(data, agent_string_))) { \
            if (virTypedParamsAddString(params, nparams, maxparams, \
                                        param_string_, result) < 0) { \
                return -1; \
            } \
        } \
    } while (0)
    OSINFO_ADD_PARAM("id", "os.id");
    OSINFO_ADD_PARAM("name", "os.name");
    OSINFO_ADD_PARAM("pretty-name", "os.pretty-name");
    OSINFO_ADD_PARAM("version", "os.version");
    OSINFO_ADD_PARAM("version-id", "os.version-id");
    OSINFO_ADD_PARAM("machine", "os.machine");
    OSINFO_ADD_PARAM("variant", "os.variant");
    OSINFO_ADD_PARAM("variant-id", "os.variant-id");
    OSINFO_ADD_PARAM("kernel-release", "os.kernel-release");
    OSINFO_ADD_PARAM("kernel-version", "os.kernel-version");

    return 0;
}
2436

2437 2438 2439 2440
/* Returns: 0 on success
 *          -2 when agent command is not supported by the agent
 *          -1 otherwise
 */
2441
int
2442
qemuAgentGetTimezone(qemuAgentPtr agent,
2443 2444 2445 2446
                     virTypedParameterPtr *params,
                     int *nparams,
                     int *maxparams)
{
J
Ján Tomko 已提交
2447 2448
    g_autoptr(virJSONValue) cmd = NULL;
    g_autoptr(virJSONValue) reply = NULL;
2449 2450 2451 2452 2453 2454 2455
    virJSONValuePtr data = NULL;
    const char *name;
    int offset;

    if (!(cmd = qemuAgentMakeCommand("guest-get-timezone", NULL)))
        return -1;

2456
    if (qemuAgentCommand(agent, cmd, &reply, true, agent->timeout) < 0) {
2457 2458
        if (qemuAgentErrorCommandUnsupported(reply))
            return -2;
2459
        return -1;
2460
    }
2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484

    if (!(data = virJSONValueObjectGetObject(reply, "return"))) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("guest-get-timezone reply was missing return data"));
        return -1;
    }

    if ((name = virJSONValueObjectGetString(data, "zone")) &&
        virTypedParamsAddString(params, nparams, maxparams,
                                "timezone.name", name) < 0)
        return -1;

    if ((virJSONValueObjectGetNumberInt(data, "offset", &offset)) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("'offset' missing in reply of guest-get-timezone"));
        return -1;
    }

    if (virTypedParamsAddInt(params, nparams, maxparams,
                             "timezone.offset", offset) < 0)
        return -1;

    return 0;
}
2485 2486

/* qemuAgentSetResponseTimeout:
2487 2488
 * @agent: agent object
 * @timeout: number of seconds to wait for agent response
2489 2490 2491 2492
 *
 * The agent object must be locked prior to calling this function.
 */
void
2493
qemuAgentSetResponseTimeout(qemuAgentPtr agent,
2494 2495
                            int timeout)
{
2496
    agent->timeout = timeout;
2497
}