You need to sign in or sign up before continuing.
qemumonitortestutils.c 24.1 KB
Newer Older
1
/*
2
 * Copyright (C) 2011-2013 Red Hat, Inc.
3 4 5 6 7 8 9 10 11 12 13 14
 *
 * 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
15
 * License along with this library.  If not, see
16 17 18 19 20 21 22 23 24 25 26 27 28
 * <http://www.gnu.org/licenses/>.
 *
 */

#include <config.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

#include "qemumonitortestutils.h"

29
#include "virthread.h"
30
#include "qemu/qemu_processpriv.h"
31
#include "qemu/qemu_monitor.h"
32
#include "qemu/qemu_agent.h"
33
#include "rpc/virnetsocket.h"
34
#include "viralloc.h"
35
#include "virlog.h"
36
#include "virerror.h"
37
#include "virstring.h"
38 39 40 41

#define VIR_FROM_THIS VIR_FROM_NONE

struct _qemuMonitorTestItem {
42 43 44
    qemuMonitorTestResponseCallback cb;
    void *opaque;
    virFreeCallback freecb;
45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66
};

struct _qemuMonitorTest {
    virMutex lock;
    virThread thread;

    bool json;
    bool quit;
    bool running;

    char *incoming;
    size_t incomingLength;
    size_t incomingCapacity;

    char *outgoing;
    size_t outgoingLength;
    size_t outgoingCapacity;

    virNetSocketPtr server;
    virNetSocketPtr client;

    qemuMonitorPtr mon;
67
    qemuAgentPtr agent;
68

69 70
    char *tmpdir;

71 72 73 74 75 76 77
    size_t nitems;
    qemuMonitorTestItemPtr *items;

    virDomainObjPtr vm;
};


78 79 80 81 82 83 84 85 86 87 88
static void
qemuMonitorTestItemFree(qemuMonitorTestItemPtr item)
{
    if (!item)
        return;

    if (item->freecb)
        (item->freecb)(item->opaque);

    VIR_FREE(item);
}
89

90

91
/*
E
Eric Blake 已提交
92
 * Appends data for a reply to the outgoing buffer
93
 */
94
int
95 96
qemuMonitorTestAddReponse(qemuMonitorTestPtr test,
                          const char *response)
97 98 99 100
{
    size_t want = strlen(response) + 2;
    size_t have = test->outgoingCapacity - test->outgoingLength;

101 102
    VIR_DEBUG("Adding response to monitor command: '%s", response);

103 104
    if (have < want) {
        size_t need = want - have;
105
        if (VIR_EXPAND_N(test->outgoing, test->outgoingCapacity, need) < 0)
106 107 108 109
            return -1;
    }

    want -= 2;
110 111
    memcpy(test->outgoing + test->outgoingLength, response, want);
    memcpy(test->outgoing + test->outgoingLength + want, "\r\n", 2);
112 113 114 115 116
    test->outgoingLength += want + 2;
    return 0;
}


117
int
118
qemuMonitorTestAddUnexpectedErrorResponse(qemuMonitorTestPtr test)
119
{
120
    if (test->agent || test->json) {
121 122 123 124
        return qemuMonitorTestAddReponse(test,
                                         "{ \"error\": "
                                         " { \"desc\": \"Unexpected command\", "
                                         "   \"class\": \"UnexpectedCommand\" } }");
125
    } else {
126
        return qemuMonitorTestAddReponse(test, "unexpected command");
127 128 129 130
    }
}


131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164
int ATTRIBUTE_FMT_PRINTF(2, 3)
qemuMonitorReportError(qemuMonitorTestPtr test, const char *errmsg, ...)
{
    va_list msgargs;
    char *msg = NULL;
    char *jsonmsg = NULL;
    int ret = -1;

    va_start(msgargs, errmsg);

    if (virVasprintf(&msg, errmsg, msgargs) < 0)
        goto cleanup;

    if (test->agent || test->json) {
        if (virAsprintf(&jsonmsg, "{ \"error\": "
                                  " { \"desc\": \"%s\", "
                                  "   \"class\": \"UnexpectedCommand\" } }",
                                  msg) < 0)
            goto cleanup;
    } else {
        if (virAsprintf(&jsonmsg, "error: '%s'", msg) < 0)
            goto cleanup;
    }

    ret = qemuMonitorTestAddReponse(test, jsonmsg);

cleanup:
    va_end(msgargs);
    VIR_FREE(msg);
    VIR_FREE(jsonmsg);
    return ret;
}


165
static int
166 167
qemuMonitorTestProcessCommand(qemuMonitorTestPtr test,
                              const char *cmdstr)
168
{
169
    int ret;
170

171 172
    VIR_DEBUG("Processing string from monitor handler: '%s", cmdstr);

173 174
    if (test->nitems == 0) {
        return qemuMonitorTestAddUnexpectedErrorResponse(test);
175
    } else {
176 177 178 179 180
        qemuMonitorTestItemPtr item = test->items[0];
        ret = (item->cb)(test, item, cmdstr);
        qemuMonitorTestItemFree(item);
        if (VIR_DELETE_ELEMENT(test->items, 0, test->nitems) < 0)
            return -1;
181 182 183 184 185
    }

    return ret;
}

186

187 188 189
/*
 * Handles read/write of monitor data on the monitor server side
 */
190 191 192 193
static void
qemuMonitorTestIO(virNetSocketPtr sock,
                  int events,
                  void *opaque)
194 195 196 197 198
{
    qemuMonitorTestPtr test = opaque;
    bool err = false;

    virMutexLock(&test->lock);
199 200 201 202
    if (test->quit) {
        virMutexUnlock(&test->lock);
        return;
    }
203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244
    if (events & VIR_EVENT_HANDLE_WRITABLE) {
        ssize_t ret;
        if ((ret = virNetSocketWrite(sock,
                                     test->outgoing,
                                     test->outgoingLength)) < 0) {
            err = true;
            goto cleanup;
        }

        memmove(test->outgoing,
                test->outgoing + ret,
                test->outgoingLength - ret);
        test->outgoingLength -= ret;

        if ((test->outgoingCapacity - test->outgoingLength) > 1024)
            VIR_SHRINK_N(test->outgoing, test->outgoingCapacity, 1024);
    }

    if (events & VIR_EVENT_HANDLE_READABLE) {
        ssize_t ret, used;
        char *t1, *t2;

        if ((test->incomingCapacity - test->incomingLength) < 1024) {
            if (VIR_EXPAND_N(test->incoming, test->incomingCapacity, 1024) < 0) {
                err = true;
                goto cleanup;
            }
        }

        if ((ret = virNetSocketRead(sock,
                                    test->incoming + test->incomingLength,
                                    (test->incomingCapacity - test->incomingLength) - 1)) < 0) {
            err = true;
            goto cleanup;
        }
        test->incomingLength += ret;
        test->incoming[test->incomingLength] = '\0';

        /* Look to see if we've got a complete line, and
         * if so, handle that command
         */
        t1 = test->incoming;
245
        while ((t2 = strstr(t1, "\n"))) {
246 247 248 249 250 251 252
            *t2 = '\0';

            if (qemuMonitorTestProcessCommand(test, t1) < 0) {
                err = true;
                goto cleanup;
            }

253
            t1 = t2 + 1;
254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286
        }
        used = t1 - test->incoming;
        memmove(test->incoming, t1, test->incomingLength - used);
        test->incomingLength -= used;
        if ((test->incomingCapacity - test->incomingLength) > 1024) {
            VIR_SHRINK_N(test->incoming,
                         test->incomingCapacity,
                         1024);
        }
    }

    if (events & (VIR_EVENT_HANDLE_HANGUP |
                  VIR_EVENT_HANDLE_ERROR))
        err = true;

cleanup:
    if (err) {
        virNetSocketRemoveIOCallback(sock);
        virNetSocketClose(sock);
        virObjectUnref(test->client);
        test->client = NULL;
    } else {
        events = VIR_EVENT_HANDLE_READABLE;

        if (test->outgoingLength)
            events |= VIR_EVENT_HANDLE_WRITABLE;

        virNetSocketUpdateIOCallback(sock, events);
    }
    virMutexUnlock(&test->lock);
}


287 288
static void
qemuMonitorTestWorker(void *opaque)
289 290 291 292 293 294 295 296 297
{
    qemuMonitorTestPtr test = opaque;

    virMutexLock(&test->lock);

    while (!test->quit) {
        virMutexUnlock(&test->lock);

        if (virEventRunDefaultImpl() < 0) {
298
            virMutexLock(&test->lock);
299 300 301 302 303 304 305 306 307 308 309 310 311
            test->quit = true;
            break;
        }

        virMutexLock(&test->lock);
    }

    test->running = false;

    virMutexUnlock(&test->lock);
    return;
}

312

313
static void
314 315
qemuMonitorTestFreeTimer(int timer ATTRIBUTE_UNUSED,
                         void *opaque ATTRIBUTE_UNUSED)
316 317 318 319 320
{
    /* nothing to be done here */
}


321 322
void
qemuMonitorTestFree(qemuMonitorTestPtr test)
323 324
{
    size_t i;
325
    int timer = -1;
326 327 328 329 330 331 332

    if (!test)
        return;

    virMutexLock(&test->lock);
    if (test->running) {
        test->quit = true;
333 334
        /* HACK: Add a dummy timeout to break event loop */
        timer = virEventAddTimeout(0, qemuMonitorTestFreeTimer, NULL, NULL);
335 336 337 338 339 340 341 342 343 344 345
    }
    virMutexUnlock(&test->lock);

    if (test->client) {
        virNetSocketRemoveIOCallback(test->client);
        virNetSocketClose(test->client);
        virObjectUnref(test->client);
    }

    virObjectUnref(test->server);
    if (test->mon) {
346
        virObjectUnlock(test->mon);
347 348 349
        qemuMonitorClose(test->mon);
    }

350 351 352 353 354
    if (test->agent) {
        virObjectUnlock(test->agent);
        qemuAgentClose(test->agent);
    }

355 356
    virObjectUnref(test->vm);

357 358
    if (test->running)
        virThreadJoin(&test->thread);
359

360 361 362
    if (timer != -1)
        virEventRemoveTimeout(timer);

363 364 365
    VIR_FREE(test->incoming);
    VIR_FREE(test->outgoing);

366
    for (i = 0; i < test->nitems; i++)
367 368 369
        qemuMonitorTestItemFree(test->items[i]);
    VIR_FREE(test->items);

370 371 372 373 374
    if (test->tmpdir && rmdir(test->tmpdir) < 0)
        VIR_WARN("Failed to remove tempdir: %s", strerror(errno));

    VIR_FREE(test->tmpdir);

375 376 377 378 379 380
    virMutexDestroy(&test->lock);
    VIR_FREE(test);
}


int
381 382 383 384
qemuMonitorTestAddHandler(qemuMonitorTestPtr test,
                          qemuMonitorTestResponseCallback cb,
                          void *opaque,
                          virFreeCallback freecb)
385 386 387 388
{
    qemuMonitorTestItemPtr item;

    if (VIR_ALLOC(item) < 0)
389
        goto error;
390

391 392 393
    item->cb = cb;
    item->freecb = freecb;
    item->opaque = opaque;
394 395

    virMutexLock(&test->lock);
396
    if (VIR_APPEND_ELEMENT(test->items, test->nitems, item) < 0) {
397
        virMutexUnlock(&test->lock);
398
        goto error;
399 400 401 402 403
    }
    virMutexUnlock(&test->lock);

    return 0;

404
error:
405 406 407
    if (freecb)
        (freecb)(opaque);
    VIR_FREE(item);
408 409 410
    return -1;
}

411 412 413 414 415 416 417
void *
qemuMonitorTestItemGetPrivateData(qemuMonitorTestItemPtr item)
{
    return item ? item->opaque : NULL;
}


418 419 420 421 422
typedef struct _qemuMonitorTestCommandArgs qemuMonitorTestCommandArgs;
typedef qemuMonitorTestCommandArgs *qemuMonitorTestCommandArgsPtr;
struct _qemuMonitorTestCommandArgs {
    char *argname;
    char *argval;
423 424 425
};


426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452
struct qemuMonitorTestHandlerData {
    char *command_name;
    char *response;
    size_t nargs;
    qemuMonitorTestCommandArgsPtr args;
};

static void
qemuMonitorTestHandlerDataFree(void *opaque)
{
    struct qemuMonitorTestHandlerData *data = opaque;
    size_t i;

    if (!data)
        return;

    for (i = 0; i < data->nargs; i++) {
        VIR_FREE(data->args[i].argname);
        VIR_FREE(data->args[i].argval);
    }

    VIR_FREE(data->command_name);
    VIR_FREE(data->response);
    VIR_FREE(data->args);
    VIR_FREE(data);
}

453 454 455 456 457
static int
qemuMonitorTestProcessCommandDefault(qemuMonitorTestPtr test,
                                     qemuMonitorTestItemPtr item,
                                     const char *cmdstr)
{
458
    struct qemuMonitorTestHandlerData *data = item->opaque;
459 460 461 462 463 464
    virJSONValuePtr val = NULL;
    char *cmdcopy = NULL;
    const char *cmdname;
    char *tmp;
    int ret = -1;

465
    if (test->agent || test->json) {
466 467 468 469
        if (!(val = virJSONValueFromString(cmdstr)))
            return -1;

        if (!(cmdname = virJSONValueObjectGetString(val, "execute"))) {
470
            ret = qemuMonitorReportError(test, "Missing command name in %s", cmdstr);
471 472 473 474 475 476 477 478 479
            goto cleanup;
        }
    } else {
        if (VIR_STRDUP(cmdcopy, cmdstr) < 0)
            return -1;

        cmdname = cmdcopy;

        if (!(tmp = strchr(cmdcopy, ' '))) {
480 481 482
            ret = qemuMonitorReportError(test,
                                         "Cannot find command name in '%s'",
                                         cmdstr);
483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504
            goto cleanup;
        }
        *tmp = '\0';
    }

    if (STRNEQ(data->command_name, cmdname))
        ret = qemuMonitorTestAddUnexpectedErrorResponse(test);
    else
        ret = qemuMonitorTestAddReponse(test, data->response);

cleanup:
    VIR_FREE(cmdcopy);
    virJSONValueFree(val);
    return ret;
}


int
qemuMonitorTestAddItem(qemuMonitorTestPtr test,
                       const char *command_name,
                       const char *response)
{
505
    struct qemuMonitorTestHandlerData *data;
506 507 508 509

    if (VIR_ALLOC(data) < 0)
        return -1;

510 511 512 513 514
    if (VIR_STRDUP(data->command_name, command_name) < 0 ||
        VIR_STRDUP(data->response, response) < 0) {
        qemuMonitorTestHandlerDataFree(data);
        return -1;
    }
515 516 517

    return qemuMonitorTestAddHandler(test,
                                     qemuMonitorTestProcessCommandDefault,
518
                                     data, qemuMonitorTestHandlerDataFree);
519 520
}

521

522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537
static int
qemuMonitorTestProcessGuestAgentSync(qemuMonitorTestPtr test,
                                     qemuMonitorTestItemPtr item ATTRIBUTE_UNUSED,
                                     const char *cmdstr)
{
    virJSONValuePtr val = NULL;
    virJSONValuePtr args;
    unsigned long long id;
    const char *cmdname;
    char *retmsg = NULL;
    int ret = -1;

    if (!(val = virJSONValueFromString(cmdstr)))
        return -1;

    if (!(cmdname = virJSONValueObjectGetString(val, "execute"))) {
538
        ret = qemuMonitorReportError(test, "Missing guest-sync command name");
539 540 541 542 543 544 545 546 547
        goto cleanup;
    }

    if (STRNEQ(cmdname, "guest-sync")) {
        ret = qemuMonitorTestAddUnexpectedErrorResponse(test);
        goto cleanup;
    }

    if (!(args = virJSONValueObjectGet(val, "arguments"))) {
548
        ret = qemuMonitorReportError(test, "Missing arguments for guest-sync");
549 550 551 552
        goto cleanup;
    }

    if (virJSONValueObjectGetNumberUlong(args, "id", &id)) {
553
        ret = qemuMonitorReportError(test, "Missing id for guest sync");
554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584
        goto cleanup;
    }

    if (virAsprintf(&retmsg, "{\"return\":%llu}", id) < 0)
        goto cleanup;


    ret = qemuMonitorTestAddReponse(test, retmsg);

cleanup:
    virJSONValueFree(val);
    VIR_FREE(retmsg);
    return ret;
}


int
qemuMonitorTestAddAgentSyncResponse(qemuMonitorTestPtr test)
{
    if (!test->agent) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       "This test is not an agent test");
        return -1;
    }

    return qemuMonitorTestAddHandler(test,
                                     qemuMonitorTestProcessGuestAgentSync,
                                     NULL, NULL);
}


585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672
static int
qemuMonitorTestProcessCommandWithArgs(qemuMonitorTestPtr test,
                                      qemuMonitorTestItemPtr item,
                                      const char *cmdstr)
{
    struct qemuMonitorTestHandlerData *data = item->opaque;
    virJSONValuePtr val = NULL;
    virJSONValuePtr args;
    virJSONValuePtr argobj;
    char *argstr = NULL;
    const char *cmdname;
    size_t i;
    int ret = -1;

    if (!(val = virJSONValueFromString(cmdstr)))
        return -1;

    if (!(cmdname = virJSONValueObjectGetString(val, "execute"))) {
        ret = qemuMonitorReportError(test, "Missing command name in %s", cmdstr);
        goto cleanup;
    }

    if (STRNEQ(data->command_name, cmdname)) {
        ret = qemuMonitorTestAddUnexpectedErrorResponse(test);
        goto cleanup;
    }

    if (!(args = virJSONValueObjectGet(val, "arguments"))) {
        ret = qemuMonitorReportError(test,
                                     "Missing arguments section for command '%s'",
                                     data->command_name);
        goto cleanup;
    }

    /* validate the args */
    for (i = 0; i < data->nargs; i++) {
        qemuMonitorTestCommandArgsPtr arg = &data->args[i];
        if (!(argobj = virJSONValueObjectGet(args, arg->argname))) {
            ret = qemuMonitorReportError(test,
                                         "Missing argument '%s' for command '%s'",
                                         arg->argname, data->command_name);
            goto cleanup;
        }

        /* convert the argument to string */
        if (!(argstr = virJSONValueToString(argobj, false)))
            goto cleanup;

        /* verify that the argument value is expected */
        if (STRNEQ(argstr, arg->argval)) {
            ret = qemuMonitorReportError(test,
                                         "Invalid value of argument '%s' "
                                         "of command '%s': "
                                         "expected '%s' got '%s'",
                                         arg->argname, data->command_name,
                                         arg->argval, argstr);
            goto cleanup;
        }

        VIR_FREE(argstr);
    }

    /* arguments checked out, return the response */
    ret = qemuMonitorTestAddReponse(test, data->response);

cleanup:
    VIR_FREE(argstr);
    virJSONValueFree(val);
    return ret;
}


/* this allows to add a responder that is able to check
 * a (shallow) structure of arguments for a command */
int
qemuMonitorTestAddItemParams(qemuMonitorTestPtr test,
                             const char *cmdname,
                             const char *response,
                             ...)
{
    struct qemuMonitorTestHandlerData *data;
    const char *argname;
    const char *argval;
    va_list args;

    va_start(args, response);

    if (VIR_ALLOC(data) < 0)
673
        goto error;
674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709

    if (VIR_STRDUP(data->command_name, cmdname) < 0 ||
        VIR_STRDUP(data->response, response) < 0)
        goto error;

    while ((argname = va_arg(args, char *))) {
        size_t i;
        if (!(argval = va_arg(args, char *))) {
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           "Missing argument value for argument '%s'",
                           argname);
            goto error;
        }

        i = data->nargs;
        if (VIR_EXPAND_N(data->args, data->nargs, 1))
            goto error;

        if (VIR_STRDUP(data->args[i].argname, argname) < 0 ||
            VIR_STRDUP(data->args[i].argval, argval) < 0)
            goto error;
    }

    va_end(args);

    return qemuMonitorTestAddHandler(test,
                                     qemuMonitorTestProcessCommandWithArgs,
                                     data, qemuMonitorTestHandlerDataFree);

error:
    va_end(args);
    qemuMonitorTestHandlerDataFree(data);
    return -1;
}


710 711
static void
qemuMonitorTestEOFNotify(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
712 713
                         virDomainObjPtr vm ATTRIBUTE_UNUSED,
                         void *opaque ATTRIBUTE_UNUSED)
714 715 716
{
}

717 718 719

static void
qemuMonitorTestErrorNotify(qemuMonitorPtr mon ATTRIBUTE_UNUSED,
720 721
                           virDomainObjPtr vm ATTRIBUTE_UNUSED,
                           void *opaque ATTRIBUTE_UNUSED)
722 723 724 725
{
}


726
static qemuMonitorCallbacks qemuMonitorTestCallbacks = {
727 728
    .eofNotify = qemuMonitorTestEOFNotify,
    .errorNotify = qemuMonitorTestErrorNotify,
729
    .domainDeviceDeleted = qemuProcessHandleDeviceDeleted,
730 731
};

732

733 734 735 736 737 738 739 740 741 742 743 744 745
static void
qemuMonitorTestAgentNotify(qemuAgentPtr agent ATTRIBUTE_UNUSED,
                           virDomainObjPtr vm ATTRIBUTE_UNUSED)
{
}


static qemuAgentCallbacks qemuMonitorTestAgentCallbacks = {
    .eofNotify = qemuMonitorTestAgentNotify,
    .errorNotify = qemuMonitorTestAgentNotify,
};


746 747
static qemuMonitorTestPtr
qemuMonitorCommonTestNew(virDomainXMLOptionPtr xmlopt,
748
                         virDomainObjPtr vm,
749
                         virDomainChrSourceDefPtr src)
750
{
G
Guido Günther 已提交
751
    qemuMonitorTestPtr test = NULL;
752 753
    char *path = NULL;
    char *tmpdir_template = NULL;
754

755
    if (VIR_ALLOC(test) < 0)
756
        goto error;
757 758 759 760 761 762 763 764

    if (virMutexInit(&test->lock) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       "Cannot initialize mutex");
        VIR_FREE(test);
        return NULL;
    }

765 766
    if (VIR_STRDUP(tmpdir_template, "/tmp/libvirt_XXXXXX") < 0)
        goto error;
767 768 769 770 771 772 773 774 775 776

    if (!(test->tmpdir = mkdtemp(tmpdir_template))) {
        virReportSystemError(errno, "%s",
                             "Failed to create temporary directory");
        goto error;
    }

    tmpdir_template = NULL;

    if (virAsprintf(&path, "%s/qemumonitorjsontest.sock", test->tmpdir) < 0)
777
        goto error;
778

779 780 781 782 783 784 785 786
    if (vm) {
        virObjectRef(vm);
        test->vm = vm;
    } else {
        test->vm = virDomainObjNew(xmlopt);
        if (!test->vm)
            goto error;
    }
787

788
    if (virNetSocketNewListenUNIX(path, 0700, getuid(), getgid(),
789 790 791
                                  &test->server) < 0)
        goto error;

792 793 794 795
    memset(src, 0, sizeof(*src));
    src->type = VIR_DOMAIN_CHR_TYPE_UNIX;
    src->data.nix.path = (char *)path;
    src->data.nix.listen = false;
796
    path = NULL;
797 798 799 800

    if (virNetSocketListen(test->server, 1) < 0)
        goto error;

801 802 803 804
cleanup:
    return test;

error:
805
    VIR_FREE(path);
806 807 808 809 810 811 812 813 814 815 816
    VIR_FREE(tmpdir_template);
    qemuMonitorTestFree(test);
    test = NULL;
    goto cleanup;

}


static int
qemuMonitorCommonTestInit(qemuMonitorTestPtr test)
{
817 818
    int events = VIR_EVENT_HANDLE_READABLE;

819 820
    if (!test)
        return -1;
821 822 823 824

    if (virNetSocketAccept(test->server, &test->client) < 0)
        goto error;

825
    if (!test->client)
826 827
        goto error;

828 829 830
    if (test->outgoingLength > 0)
        events = VIR_EVENT_HANDLE_WRITABLE;

831
    if (virNetSocketAddIOCallback(test->client,
832
                                  events,
833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848
                                  qemuMonitorTestIO,
                                  test,
                                  NULL) < 0)
        goto error;

    virMutexLock(&test->lock);
    if (virThreadCreate(&test->thread,
                        true,
                        qemuMonitorTestWorker,
                        test) < 0) {
        virMutexUnlock(&test->lock);
        goto error;
    }
    test->running = true;
    virMutexUnlock(&test->lock);

849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872
    return 0;

error:
    return -1;
}


#define QEMU_JSON_GREETING  "{\"QMP\":"\
                            "   {\"version\":"\
                            "       {\"qemu\":"\
                            "           {\"micro\": 1,"\
                            "            \"minor\": 0,"\
                            "            \"major\": 1"\
                            "           },"\
                            "        \"package\": \"(qemu-kvm-1.0.1)"\
                            "       \"},"\
                            "    \"capabilities\": []"\
                            "   }"\
                            "}"
/* We skip the normal handshake reply of "{\"execute\":\"qmp_capabilities\"}" */

#define QEMU_TEXT_GREETING "QEMU 1.0,1 monitor - type 'help' for more information"

qemuMonitorTestPtr
873 874
qemuMonitorTestNew(bool json,
                   virDomainXMLOptionPtr xmlopt,
875
                   virDomainObjPtr vm,
876 877
                   virQEMUDriverPtr driver,
                   const char *greeting)
878 879 880 881
{
    qemuMonitorTestPtr test = NULL;
    virDomainChrSourceDef src;

882 883
    memset(&src, 0, sizeof(src));

884
    if (!(test = qemuMonitorCommonTestNew(xmlopt, vm, &src)))
885 886 887 888 889 890
        goto error;

    test->json = json;
    if (!(test->mon = qemuMonitorOpen(test->vm,
                                      &src,
                                      json,
891
                                      &qemuMonitorTestCallbacks,
892
                                      driver)))
893 894 895 896
        goto error;

    virObjectLock(test->mon);

897 898 899 900
    if (!greeting)
        greeting = json ? QEMU_JSON_GREETING : QEMU_TEXT_GREETING;

    if (qemuMonitorTestAddReponse(test, greeting) < 0)
901 902 903 904 905 906 907
        goto error;

    if (qemuMonitorCommonTestInit(test) < 0)
        goto error;

    virDomainChrSourceDefClear(&src);

908 909 910
    return test;

error:
911 912 913 914 915 916 917 918 919 920 921
    virDomainChrSourceDefClear(&src);
    qemuMonitorTestFree(test);
    return NULL;
}

qemuMonitorTestPtr
qemuMonitorTestNewAgent(virDomainXMLOptionPtr xmlopt)
{
    qemuMonitorTestPtr test = NULL;
    virDomainChrSourceDef src;

922 923
    memset(&src, 0, sizeof(src));

924
    if (!(test = qemuMonitorCommonTestNew(xmlopt, NULL, &src)))
925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942
        goto error;

    if (!(test->agent = qemuAgentOpen(test->vm,
                                      &src,
                                      &qemuMonitorTestAgentCallbacks)))
        goto error;

    virObjectLock(test->agent);

    if (qemuMonitorCommonTestInit(test) < 0)
        goto error;

    virDomainChrSourceDefClear(&src);

    return test;

error:
    virDomainChrSourceDefClear(&src);
943
    qemuMonitorTestFree(test);
944
    return NULL;
945 946
}

947 948 949

qemuMonitorPtr
qemuMonitorTestGetMonitor(qemuMonitorTestPtr test)
950 951 952
{
    return test->mon;
}
953 954 955 956 957 958 959


qemuAgentPtr
qemuMonitorTestGetAgent(qemuMonitorTestPtr test)
{
    return test->agent;
}