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

25
#include <config.h>
E
Eric Blake 已提交
26
#include "virsh.h"
27

28
#include <stdio.h>
K
Karel Zak 已提交
29 30 31
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
32
#include <unistd.h>
33
#include <errno.h>
K
Karel Zak 已提交
34
#include <getopt.h>
K
Karel Zak 已提交
35
#include <sys/time.h>
J
Jim Meyering 已提交
36
#include "c-ctype.h"
37
#include <fcntl.h>
38
#include <locale.h>
39
#include <time.h>
40
#include <limits.h>
41
#include <assert.h>
42
#include <sys/stat.h>
43
#include <inttypes.h>
E
Eric Blake 已提交
44
#include <strings.h>
45
#include <termios.h>
K
Karel Zak 已提交
46

47 48 49
#include <libxml/parser.h>
#include <libxml/tree.h>
#include <libxml/xpath.h>
50
#include <libxml/xmlsave.h>
51

52
#ifdef HAVE_READLINE_READLINE_H
53 54
# include <readline/readline.h>
# include <readline/history.h>
55
#endif
K
Karel Zak 已提交
56

57
#include "internal.h"
58
#include "virterror_internal.h"
59
#include "base64.h"
60
#include "buf.h"
61
#include "console.h"
62
#include "util.h"
63
#include "memory.h"
64
#include "xml.h"
65
#include "libvirt/libvirt-qemu.h"
E
Eric Blake 已提交
66
#include "virfile.h"
67
#include "event_poll.h"
68
#include "configmake.h"
69
#include "threads.h"
E
Eric Blake 已提交
70
#include "command.h"
71
#include "virkeycode.h"
72
#include "virnetdevbandwidth.h"
73
#include "util/bitmap.h"
H
Hu Tao 已提交
74
#include "conf/domain_conf.h"
75
#include "virtypedparam.h"
K
Karel Zak 已提交
76

E
Eric Blake 已提交
77
#include "virsh-domain.h"
78
#include "virsh-domain-monitor.h"
E
Eric Blake 已提交
79
#include "virsh-host.h"
E
Eric Blake 已提交
80
#include "virsh-interface.h"
E
Eric Blake 已提交
81
#include "virsh-network.h"
E
Eric Blake 已提交
82
#include "virsh-nodedev.h"
E
Eric Blake 已提交
83
#include "virsh-nwfilter.h"
E
Eric Blake 已提交
84
#include "virsh-pool.h"
E
Eric Blake 已提交
85
#include "virsh-secret.h"
E
Eric Blake 已提交
86
#include "virsh-snapshot.h"
E
Eric Blake 已提交
87
#include "virsh-volume.h"
E
Eric Blake 已提交
88

K
Karel Zak 已提交
89 90
static char *progname;

91
static const vshCmdGrp cmdGroups[];
K
Karel Zak 已提交
92

E
Eric Blake 已提交
93 94
/* Bypass header poison */
#undef strdup
95

E
Eric Blake 已提交
96
void *
E
Eric Blake 已提交
97 98
_vshMalloc(vshControl *ctl, size_t size, const char *filename, int line)
{
E
Eric Blake 已提交
99
    char *x;
E
Eric Blake 已提交
100

E
Eric Blake 已提交
101
    if (VIR_ALLOC_N(x, size) == 0)
E
Eric Blake 已提交
102 103 104 105 106 107
        return x;
    vshError(ctl, _("%s: %d: failed to allocate %d bytes"),
             filename, line, (int) size);
    exit(EXIT_FAILURE);
}

E
Eric Blake 已提交
108 109 110
void *
_vshCalloc(vshControl *ctl, size_t nmemb, size_t size, const char *filename,
           int line)
E
Eric Blake 已提交
111
{
E
Eric Blake 已提交
112
    char *x;
E
Eric Blake 已提交
113

E
Eric Blake 已提交
114 115
    if (!xalloc_oversized(nmemb, size) &&
        VIR_ALLOC_N(x, nmemb * size) == 0)
E
Eric Blake 已提交
116 117 118 119 120 121
        return x;
    vshError(ctl, _("%s: %d: failed to allocate %d bytes"),
             filename, line, (int) (size*nmemb));
    exit(EXIT_FAILURE);
}

E
Eric Blake 已提交
122
char *
E
Eric Blake 已提交
123 124 125 126 127
_vshStrdup(vshControl *ctl, const char *s, const char *filename, int line)
{
    char *x;

    if (s == NULL)
128
        return NULL;
E
Eric Blake 已提交
129 130 131 132 133 134 135 136 137
    if ((x = strdup(s)))
        return x;
    vshError(ctl, _("%s: %d: failed to allocate %lu bytes"),
             filename, line, (unsigned long)strlen(s));
    exit(EXIT_FAILURE);
}

/* Poison the raw allocating identifiers in favor of our vsh variants.  */
#define strdup use_vshStrdup_instead_of_strdup
138

139
int
140 141 142 143
vshNameSorter(const void *a, const void *b)
{
    const char **sa = (const char**)a;
    const char **sb = (const char**)b;
144

145
    return vshStrcasecmp(*sa, *sb);
146 147
}

E
Eric Blake 已提交
148
double
E
Eric Blake 已提交
149
vshPrettyCapacity(unsigned long long val, const char **unit)
E
Eric Blake 已提交
150
{
151 152 153 154
    if (val < 1024) {
        *unit = "";
        return (double)val;
    } else if (val < (1024.0l * 1024.0l)) {
155
        *unit = "KiB";
156 157
        return (((double)val / 1024.0l));
    } else if (val < (1024.0l * 1024.0l * 1024.0l)) {
158
        *unit = "MiB";
159
        return (double)val / (1024.0l * 1024.0l);
160
    } else if (val < (1024.0l * 1024.0l * 1024.0l * 1024.0l)) {
161
        *unit = "GiB";
162
        return (double)val / (1024.0l * 1024.0l * 1024.0l);
163
    } else {
164
        *unit = "TiB";
165
        return (double)val / (1024.0l * 1024.0l * 1024.0l * 1024.0l);
166 167 168
    }
}

169 170
/*
 * Convert the strings separated by ',' into array. The caller
E
Eric Blake 已提交
171 172 173
 * must free the first array element and the returned array after
 * use (all other array elements belong to the memory allocated
 * for the first array element).
174 175 176 177 178
 *
 * Returns the length of the filled array on success, or -1
 * on error.
 */
int
179
vshStringToArray(const char *str,
180 181
                 char ***array)
{
182
    char *str_copied = vshStrdup(NULL, str);
183
    char *str_tok = NULL;
E
Eric Blake 已提交
184
    char *tmp;
185 186
    unsigned int nstr_tokens = 0;
    char **arr = NULL;
E
Eric Blake 已提交
187
    size_t len = strlen(str_copied);
188

E
Eric Blake 已提交
189 190
    /* tokenize the string from user and save its parts into an array */
    nstr_tokens = 1;
191

E
Eric Blake 已提交
192 193 194 195 196
    /* count the delimiters, recognizing ,, as an escape for a
     * literal comma */
    str_tok = str_copied;
    while ((str_tok = strchr(str_tok, ','))) {
        if (str_tok[1] == ',')
197
            str_tok++;
E
Eric Blake 已提交
198 199 200 201
        else
            nstr_tokens++;
        str_tok++;
    }
202

E
Eric Blake 已提交
203 204 205 206 207
    if (VIR_ALLOC_N(arr, nstr_tokens) < 0) {
        virReportOOMError();
        VIR_FREE(str_copied);
        return -1;
    }
208

E
Eric Blake 已提交
209 210 211 212 213 214 215 216 217 218 219 220 221
    /* tokenize the input string, while treating ,, as a literal comma */
    nstr_tokens = 0;
    tmp = str_tok = str_copied;
    while ((tmp = strchr(tmp, ','))) {
        if (tmp[1] == ',') {
            memmove(&tmp[1], &tmp[2], len - (tmp - str_copied) - 2 + 1);
            len--;
            tmp++;
            continue;
        }
        *tmp++ = '\0';
        arr[nstr_tokens++] = str_tok;
        str_tok = tmp;
222
    }
E
Eric Blake 已提交
223
    arr[nstr_tokens++] = str_tok;
224 225 226 227

    *array = arr;
    return nstr_tokens;
}
228

E
Eric Blake 已提交
229
virErrorPtr last_error;
J
John Levon 已提交
230 231 232 233 234 235 236 237 238 239 240 241 242

/*
 * Quieten libvirt until we're done with the command.
 */
static void
virshErrorHandler(void *unused ATTRIBUTE_UNUSED, virErrorPtr error)
{
    virFreeError(last_error);
    last_error = virSaveLastError();
    if (getenv("VIRSH_DEBUG") != NULL)
        virDefaultErrorFunc(error);
}

243 244 245
/*
 * Reset libvirt error on graceful fallback paths
 */
E
Eric Blake 已提交
246
void
247 248 249 250 251 252
vshResetLibvirtError(void)
{
    virFreeError(last_error);
    last_error = NULL;
}

J
John Levon 已提交
253 254 255 256 257 258 259 260
/*
 * Report an error when a command finishes.  This is better than before
 * (when correct operation would report errors), but it has some
 * problems: we lose the smarter formatting of virDefaultErrorFunc(),
 * and it can become harder to debug problems, if errors get reported
 * twice during one command.  This case shouldn't really happen anyway,
 * and it's IMHO a bug that libvirt does that sometimes.
 */
E
Eric Blake 已提交
261
void
E
Eric Blake 已提交
262
vshReportError(vshControl *ctl)
J
John Levon 已提交
263
{
264 265 266 267 268 269 270 271
    if (last_error == NULL) {
        /* Calling directly into libvirt util functions won't trigger the
         * error callback (which sets last_error), so check it ourselves.
         *
         * If the returned error has CODE_OK, this most likely means that
         * no error was ever raised, so just ignore */
        last_error = virSaveLastError();
        if (!last_error || last_error->code == VIR_ERR_OK)
272
            goto out;
273
    }
J
John Levon 已提交
274 275

    if (last_error->code == VIR_ERR_OK) {
276
        vshError(ctl, "%s", _("unknown error"));
J
John Levon 已提交
277 278 279
        goto out;
    }

280
    vshError(ctl, "%s", last_error->message);
J
John Levon 已提交
281 282

out:
283
    vshResetLibvirtError();
J
John Levon 已提交
284 285
}

286 287 288 289 290 291 292 293
/*
 * Detection of disconnections and automatic reconnection support
 */
static int disconnected = 0; /* we may have been disconnected */

/*
 * vshCatchDisconnect:
 *
294 295
 * We get here when the connection was closed.  We can't do much in the
 * handler, just save the fact it was raised.
296
 */
L
Laine Stump 已提交
297
static void
298 299 300 301 302 303
vshCatchDisconnect(virConnectPtr conn ATTRIBUTE_UNUSED,
                   int reason,
                   void *opaque ATTRIBUTE_UNUSED)
{
    if (reason != VIR_CONNECT_CLOSE_REASON_CLIENT)
        disconnected++;
304 305 306 307 308
}

/*
 * vshReconnect:
 *
L
Laine Stump 已提交
309
 * Reconnect after a disconnect from libvirtd
310 311
 *
 */
L
Laine Stump 已提交
312
static void
313 314 315 316 317 318
vshReconnect(vshControl *ctl)
{
    bool connected = false;

    if (ctl->conn != NULL) {
        connected = true;
319
        virConnectClose(ctl->conn);
320
    }
321 322 323 324

    ctl->conn = virConnectOpenAuth(ctl->name,
                                   virConnectAuthPtrDefault,
                                   ctl->readonly ? VIR_CONNECT_RO : 0);
325
    if (!ctl->conn) {
326
        vshError(ctl, "%s", _("Failed to reconnect to the hypervisor"));
327 328 329 330 331 332 333
    } else {
        if (virConnectRegisterCloseCallback(ctl->conn, vshCatchDisconnect,
                                            NULL, NULL) < 0)
            vshError(ctl, "%s", _("Unable to register disconnect callback"));
        if (connected)
            vshError(ctl, "%s", _("Reconnected to the hypervisor"));
    }
334
    disconnected = 0;
335
    ctl->useGetInfo = false;
336
    ctl->useSnapshotOld = false;
337
}
338

339
#ifndef WIN32
340 341 342 343 344 345 346 347 348 349 350 351 352
static void
vshPrintRaw(vshControl *ctl, ...)
{
    va_list ap;
    char *key;

    va_start(ap, ctl);
    while ((key = va_arg(ap, char *)) != NULL) {
        vshPrint(ctl, "%s\r\n", key);
    }
    va_end(ap);
}

353 354 355 356 357 358 359 360 361 362 363 364 365
/**
 * vshAskReedit:
 * @msg: Question to ask user
 *
 * Ask user if he wants to return to previously
 * edited file.
 *
 * Returns 'y' if he wants to
 *         'f' if he forcibly wants to
 *         'n' if he doesn't want to
 *         -1  on error
 *          0  otherwise
 */
E
Eric Blake 已提交
366
int
367 368 369 370 371 372 373 374
vshAskReedit(vshControl *ctl, const char *msg)
{
    int c = -1;
    struct termios ttyattr;

    if (!isatty(STDIN_FILENO))
        return -1;

E
Eric Blake 已提交
375
    vshReportError(ctl);
376 377 378 379 380 381 382 383 384 385 386

    if (vshMakeStdinRaw(&ttyattr, false) < 0)
        return -1;

    while (true) {
        /* TRANSLATORS: For now, we aren't using LC_MESSAGES, and the user
         * choices really are limited to just 'y', 'n', 'f' and '?'  */
        vshPrint(ctl, "\r%s %s", msg, _("Try again? [y,n,f,?]:"));
        c = c_tolower(getchar());

        if (c == '?') {
387 388 389 390 391 392 393
            vshPrintRaw(ctl,
                        "",
                        _("y - yes, start editor again"),
                        _("n - no, throw away my changes"),
                        _("f - force, try to redefine again"),
                        _("? - print this help"),
                        NULL);
394 395 396 397 398 399 400 401 402 403
            continue;
        } else if (c == 'y' || c == 'n' || c == 'f') {
            break;
        }
    }

    tcsetattr(STDIN_FILENO, TCSAFLUSH, &ttyattr);

    vshPrint(ctl, "\r\n");
    return c;
404 405
}
#else /* WIN32 */
406
int
407 408
vshAskReedit(vshControl *ctl, const char *msg ATTRIBUTE_UNUSED)
{
409 410 411 412
    vshDebug(ctl, VSH_ERR_WARNING, "%s", _("This function is not "
                                           "supported on WIN32 platform"));
    return 0;
}
413
#endif /* WIN32 */
414

E
Eric Blake 已提交
415 416
int vshStreamSink(virStreamPtr st ATTRIBUTE_UNUSED,
                  const char *bytes, size_t nbytes, void *opaque)
417 418 419 420 421 422
{
    int *fd = opaque;

    return safewrite(*fd, bytes, nbytes);
}

K
Karel Zak 已提交
423 424 425 426 427 428
/* ---------------
 * Commands
 * ---------------
 */

/*
429
 * "help" command
K
Karel Zak 已提交
430
 */
431
static const vshCmdInfo info_help[] = {
432
    {"help", N_("print help")},
433 434
    {"desc", N_("Prints global help, command specific help, or help for a\n"
                "    group of related commands")},
435

436
    {NULL, NULL}
K
Karel Zak 已提交
437 438
};

439
static const vshCmdOptDef opts_help[] = {
440
    {"command", VSH_OT_DATA, 0, N_("Prints global help, command specific help, or help for a group of related commands")},
441
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
442 443
};

E
Eric Blake 已提交
444
static bool
445
cmdHelp(vshControl *ctl, const vshCmd *cmd)
446
 {
447
    const char *name = NULL;
448

449
    if (vshCommandOptString(cmd, "command", &name) <= 0) {
450
        const vshCmdGrp *grp;
451
        const vshCmdDef *def;
452

453 454 455 456 457 458
        vshPrint(ctl, "%s", _("Grouped commands:\n\n"));

        for (grp = cmdGroups; grp->name; grp++) {
            vshPrint(ctl, _(" %s (help keyword '%s'):\n"), grp->name,
                     grp->keyword);

459 460 461
            for (def = grp->commands; def->name; def++) {
                if (def->flags & VSH_CMD_FLAG_ALIAS)
                    continue;
462 463
                vshPrint(ctl, "    %-30s %s\n", def->name,
                         _(vshCmddefGetInfo(def, "help")));
464
            }
465 466 467 468

            vshPrint(ctl, "\n");
        }

E
Eric Blake 已提交
469
        return true;
470
    }
471

E
Eric Blake 已提交
472
    if (vshCmddefSearch(name)) {
473
        return vshCmddefHelp(ctl, name);
E
Eric Blake 已提交
474
    } else if (vshCmdGrpSearch(name)) {
475 476 477
        return vshCmdGrpHelp(ctl, name);
    } else {
        vshError(ctl, _("command or command group '%s' doesn't exist"), name);
E
Eric Blake 已提交
478
        return false;
K
Karel Zak 已提交
479 480 481
    }
}

482 483 484 485 486 487 488 489 490 491 492
/* Tree listing helpers.  */

static int
vshTreePrintInternal(vshControl *ctl,
                     vshTreeLookup lookup,
                     void *opaque,
                     int num_devices,
                     int devid,
                     int lastdev,
                     bool root,
                     virBufferPtr indent)
493
{
494 495 496 497
    int i;
    int nextlastdev = -1;
    int ret = -1;
    const char *dev = (lookup)(devid, false, opaque);
498

499
    if (virBufferError(indent))
500 501
        goto cleanup;

502 503 504 505 506 507 508 509 510 511
    /* Print this device, with indent if not at root */
    vshPrint(ctl, "%s%s%s\n", virBufferCurrentContent(indent),
             root ? "" : "+- ", dev);

    /* Update indent to show '|' or ' ' for child devices */
    if (!root) {
        virBufferAddChar(indent, devid == lastdev ? ' ' : '|');
        virBufferAddChar(indent, ' ');
        if (virBufferError(indent))
            goto cleanup;
512 513
    }

514 515 516
    /* Determine the index of the last child device */
    for (i = 0 ; i < num_devices ; i++) {
        const char *parent = (lookup)(i, true, opaque);
517

518 519 520
        if (parent && STREQ(parent, dev))
            nextlastdev = i;
    }
521

522 523 524
    /* If there is a child device, then print another blank line */
    if (nextlastdev != -1)
        vshPrint(ctl, "%s  |\n", virBufferCurrentContent(indent));
525

526 527 528 529
    /* Finally print all children */
    virBufferAddLit(indent, "  ");
    for (i = 0 ; i < num_devices ; i++) {
        const char *parent = (lookup)(i, true, opaque);
530

531 532 533 534 535
        if (parent && STREQ(parent, dev) &&
            vshTreePrintInternal(ctl, lookup, opaque,
                                 num_devices, i, nextlastdev,
                                 false, indent) < 0)
            goto cleanup;
536
    }
537
    virBufferTrim(indent, "  ", -1);
538

539 540 541 542
    /* If there was no child device, and we're the last in
     * a list of devices, then print another blank line */
    if (nextlastdev == -1 && devid == lastdev)
        vshPrint(ctl, "%s\n", virBufferCurrentContent(indent));
543

544 545 546
    if (!root)
        virBufferTrim(indent, NULL, 2);
    ret = 0;
547 548 549 550
cleanup:
    return ret;
}

E
Eric Blake 已提交
551
int
552 553
vshTreePrint(vshControl *ctl, vshTreeLookup lookup, void *opaque,
             int num_devices, int devid)
554
{
555 556
    int ret;
    virBuffer indent = VIR_BUFFER_INITIALIZER;
557

558 559 560 561 562
    ret = vshTreePrintInternal(ctl, lookup, opaque, num_devices,
                               devid, devid, true, &indent);
    if (ret < 0)
        vshError(ctl, "%s", _("Failed to complete tree listing"));
    virBufferFreeAndReset(&indent);
563
    return ret;
564
}
565

566
/* Common code for the edit / net-edit / pool-edit functions which follow. */
E
Eric Blake 已提交
567
char *
E
Eric Blake 已提交
568
vshEditWriteToTempFile(vshControl *ctl, const char *doc)
569 570 571 572
{
    char *ret;
    const char *tmpdir;
    int fd;
573
    char ebuf[1024];
574

575
    tmpdir = getenv("TMPDIR");
576
    if (!tmpdir) tmpdir = "/tmp";
577 578 579 580
    if (virAsprintf(&ret, "%s/virshXXXXXX.xml", tmpdir) < 0) {
        vshError(ctl, "%s", _("out of memory"));
        return NULL;
    }
581
    fd = mkostemps(ret, 4, O_CLOEXEC);
582
    if (fd == -1) {
583
        vshError(ctl, _("mkostemps: failed to create temporary file: %s"),
584
                 virStrerror(errno, ebuf, sizeof(ebuf)));
585
        VIR_FREE(ret);
586 587 588
        return NULL;
    }

589
    if (safewrite(fd, doc, strlen(doc)) == -1) {
590
        vshError(ctl, _("write: %s: failed to write to temporary file: %s"),
591
                 ret, virStrerror(errno, ebuf, sizeof(ebuf)));
S
Stefan Berger 已提交
592
        VIR_FORCE_CLOSE(fd);
593
        unlink(ret);
594
        VIR_FREE(ret);
595 596
        return NULL;
    }
S
Stefan Berger 已提交
597
    if (VIR_CLOSE(fd) < 0) {
598
        vshError(ctl, _("close: %s: failed to write or close temporary file: %s"),
599
                 ret, virStrerror(errno, ebuf, sizeof(ebuf)));
600
        unlink(ret);
601
        VIR_FREE(ret);
602 603 604 605 606 607 608 609 610 611 612
        return NULL;
    }

    /* Temporary filename: caller frees. */
    return ret;
}

/* Characters permitted in $EDITOR environment variable and temp filename. */
#define ACCEPTED_CHARS \
  "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-/_.:@"

E
Eric Blake 已提交
613
int
E
Eric Blake 已提交
614
vshEditFile(vshControl *ctl, const char *filename)
615 616
{
    const char *editor;
E
Eric Blake 已提交
617 618 619 620
    virCommandPtr cmd;
    int ret = -1;
    int outfd = STDOUT_FILENO;
    int errfd = STDERR_FILENO;
621

622
    editor = getenv("VISUAL");
E
Eric Blake 已提交
623
    if (!editor)
624
        editor = getenv("EDITOR");
E
Eric Blake 已提交
625 626
    if (!editor)
        editor = "vi"; /* could be cruel & default to ed(1) here */
627

628 629 630 631 632
    /* Check that filename doesn't contain shell meta-characters, and
     * if it does, refuse to run.  Follow the Unix conventions for
     * EDITOR: the user can intentionally specify command options, so
     * we don't protect any shell metacharacters there.  Lots more
     * than virsh will misbehave if EDITOR has bogus contents (which
E
Eric Blake 已提交
633 634
     * is why sudo scrubs it by default).  Conversely, if the editor
     * is safe, we can run it directly rather than wasting a shell.
635
     */
636 637
    if (strspn(editor, ACCEPTED_CHARS) != strlen(editor)) {
        if (strspn(filename, ACCEPTED_CHARS) != strlen(filename)) {
E
Eric Blake 已提交
638 639 640 641 642 643 644 645 646 647
            vshError(ctl,
                     _("%s: temporary filename contains shell meta or other "
                       "unacceptable characters (is $TMPDIR wrong?)"),
                     filename);
            return -1;
        }
        cmd = virCommandNewArgList("sh", "-c", NULL);
        virCommandAddArgFormat(cmd, "%s %s", editor, filename);
    } else {
        cmd = virCommandNewArgList(editor, filename, NULL);
648 649
    }

E
Eric Blake 已提交
650 651 652 653 654
    virCommandSetInputFD(cmd, STDIN_FILENO);
    virCommandSetOutputFD(cmd, &outfd);
    virCommandSetErrorFD(cmd, &errfd);
    if (virCommandRunAsync(cmd, NULL) < 0 ||
        virCommandWait(cmd, NULL) < 0) {
E
Eric Blake 已提交
655
        vshReportError(ctl);
E
Eric Blake 已提交
656
        goto cleanup;
657
    }
E
Eric Blake 已提交
658
    ret = 0;
659

E
Eric Blake 已提交
660 661 662
cleanup:
    virCommandFree(cmd);
    return ret;
663 664
}

E
Eric Blake 已提交
665
char *
E
Eric Blake 已提交
666
vshEditReadBackFile(vshControl *ctl, const char *filename)
667 668
{
    char *ret;
669
    char ebuf[1024];
670

E
Eric Blake 已提交
671
    if (virFileReadAll(filename, VSH_MAX_XML_FILE, &ret) == -1) {
672
        vshError(ctl,
673
                 _("%s: failed to read temporary file: %s"),
674
                 filename, virStrerror(errno, ebuf, sizeof(ebuf)));
675 676 677 678 679
        return NULL;
    }
    return ret;
}

680

P
Paolo Bonzini 已提交
681 682 683 684
/*
 * "cd" command
 */
static const vshCmdInfo info_cd[] = {
685 686
    {"help", N_("change the current directory")},
    {"desc", N_("Change the current directory.")},
P
Paolo Bonzini 已提交
687 688 689 690
    {NULL, NULL}
};

static const vshCmdOptDef opts_cd[] = {
691
    {"dir", VSH_OT_DATA, 0, N_("directory to switch to (default: home or else root)")},
P
Paolo Bonzini 已提交
692 693 694
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
695
static bool
696
cmdCd(vshControl *ctl, const vshCmd *cmd)
P
Paolo Bonzini 已提交
697
{
698
    const char *dir = NULL;
699
    char *dir_malloced = NULL;
E
Eric Blake 已提交
700
    bool ret = true;
701
    char ebuf[1024];
P
Paolo Bonzini 已提交
702 703

    if (!ctl->imode) {
704
        vshError(ctl, "%s", _("cd: command valid only in interactive mode"));
E
Eric Blake 已提交
705
        return false;
P
Paolo Bonzini 已提交
706 707
    }

708
    if (vshCommandOptString(cmd, "dir", &dir) <= 0) {
709
        dir = dir_malloced = virGetUserDirectory();
P
Paolo Bonzini 已提交
710 711 712 713
    }
    if (!dir)
        dir = "/";

P
Phil Petty 已提交
714
    if (chdir(dir) == -1) {
715 716
        vshError(ctl, _("cd: %s: %s"),
                 virStrerror(errno, ebuf, sizeof(ebuf)), dir);
E
Eric Blake 已提交
717
        ret = false;
P
Paolo Bonzini 已提交
718 719
    }

720
    VIR_FREE(dir_malloced);
P
Phil Petty 已提交
721
    return ret;
P
Paolo Bonzini 已提交
722 723 724 725 726 727
}

/*
 * "pwd" command
 */
static const vshCmdInfo info_pwd[] = {
728 729
    {"help", N_("print the current directory")},
    {"desc", N_("Print the current directory.")},
P
Paolo Bonzini 已提交
730 731 732
    {NULL, NULL}
};

E
Eric Blake 已提交
733
static bool
P
Paolo Bonzini 已提交
734 735 736
cmdPwd(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
{
    char *cwd;
737
    bool ret = true;
738
    char ebuf[1024];
P
Paolo Bonzini 已提交
739

740 741
    cwd = getcwd(NULL, 0);
    if (!cwd) {
742
        vshError(ctl, _("pwd: cannot get current directory: %s"),
743
                 virStrerror(errno, ebuf, sizeof(ebuf)));
744 745
        ret = false;
    } else {
746
        vshPrint(ctl, _("%s\n"), cwd);
747 748
        VIR_FREE(cwd);
    }
P
Paolo Bonzini 已提交
749

750
    return ret;
P
Paolo Bonzini 已提交
751 752
}

E
Eric Blake 已提交
753 754 755 756 757 758 759 760 761 762 763 764
/*
 * "echo" command
 */
static const vshCmdInfo info_echo[] = {
    {"help", N_("echo arguments")},
    {"desc", N_("Echo back arguments, possibly with quoting.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_echo[] = {
    {"shell", VSH_OT_BOOL, 0, N_("escape for shell use")},
    {"xml", VSH_OT_BOOL, 0, N_("escape for XML use")},
E
Eric Blake 已提交
765
    {"str", VSH_OT_ALIAS, 0, "string"},
766
    {"string", VSH_OT_ARGV, 0, N_("arguments to echo")},
E
Eric Blake 已提交
767 768 769 770 771 772
    {NULL, 0, 0, NULL}
};

/* Exists mainly for debugging virsh, but also handy for adding back
 * quotes for later evaluation.
 */
E
Eric Blake 已提交
773
static bool
774
cmdEcho(vshControl *ctl, const vshCmd *cmd)
E
Eric Blake 已提交
775 776 777 778
{
    bool shell = false;
    bool xml = false;
    int count = 0;
779
    const vshCmdOpt *opt = NULL;
E
Eric Blake 已提交
780 781 782 783 784 785 786 787
    char *arg;
    virBuffer buf = VIR_BUFFER_INITIALIZER;

    if (vshCommandOptBool(cmd, "shell"))
        shell = true;
    if (vshCommandOptBool(cmd, "xml"))
        xml = true;

788
    while ((opt = vshCommandOptArgv(cmd, opt))) {
789 790
        char *str;
        virBuffer xmlbuf = VIR_BUFFER_INITIALIZER;
E
Eric Blake 已提交
791

792
        arg = opt->data;
793

E
Eric Blake 已提交
794 795
        if (count)
            virBufferAddChar(&buf, ' ');
796

E
Eric Blake 已提交
797
        if (xml) {
798 799 800 801
            virBufferEscapeString(&xmlbuf, "%s", arg);
            if (virBufferError(&buf)) {
                vshPrint(ctl, "%s", _("Failed to allocate XML buffer"));
                return false;
E
Eric Blake 已提交
802
            }
803 804 805
            str = virBufferContentAndReset(&xmlbuf);
        } else {
            str = vshStrdup(ctl, arg);
E
Eric Blake 已提交
806
        }
807 808 809 810 811

        if (shell)
            virBufferEscapeShell(&buf, str);
        else
            virBufferAdd(&buf, str, -1);
E
Eric Blake 已提交
812
        count++;
813
        VIR_FREE(str);
E
Eric Blake 已提交
814 815 816 817
    }

    if (virBufferError(&buf)) {
        vshPrint(ctl, "%s", _("Failed to allocate XML buffer"));
E
Eric Blake 已提交
818
        return false;
E
Eric Blake 已提交
819 820 821 822 823
    }
    arg = virBufferContentAndReset(&buf);
    if (arg)
        vshPrint(ctl, "%s", arg);
    VIR_FREE(arg);
E
Eric Blake 已提交
824
    return true;
E
Eric Blake 已提交
825 826
}

K
Karel Zak 已提交
827 828 829
/*
 * "quit" command
 */
830
static const vshCmdInfo info_quit[] = {
831
    {"help", N_("quit this interactive terminal")},
832
    {"desc", ""},
833
    {NULL, NULL}
K
Karel Zak 已提交
834 835
};

E
Eric Blake 已提交
836
static bool
837
cmdQuit(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
838
{
E
Eric Blake 已提交
839 840
    ctl->imode = false;
    return true;
K
Karel Zak 已提交
841 842
}

843 844 845 846
/* ---------------
 * Utils for work with command definition
 * ---------------
 */
E
Eric Blake 已提交
847
const char *
848 849 850
vshCmddefGetInfo(const vshCmdDef * cmd, const char *name)
{
    const vshCmdInfo *info;
851

852 853 854 855 856 857
    for (info = cmd->info; info && info->name; info++) {
        if (STREQ(info->name, name))
            return info->data;
    }
    return NULL;
}
858

859 860 861 862 863 864 865
/* Validate that the options associated with cmd can be parsed.  */
static int
vshCmddefOptParse(const vshCmdDef *cmd, uint32_t *opts_need_arg,
                  uint32_t *opts_required)
{
    int i;
    bool optional = false;
866

867 868
    *opts_need_arg = 0;
    *opts_required = 0;
869

870 871
    if (!cmd->opts)
        return 0;
872

873 874
    for (i = 0; cmd->opts[i].name; i++) {
        const vshCmdOptDef *opt = &cmd->opts[i];
875 876 877 878

        if (i > 31)
            return -1; /* too many options */
        if (opt->type == VSH_OT_BOOL) {
E
Eric Blake 已提交
879
            if (opt->flags & VSH_OFLAG_REQ)
880 881 882
                return -1; /* bool options can't be mandatory */
            continue;
        }
E
Eric Blake 已提交
883 884 885 886 887 888 889 890 891 892 893 894
        if (opt->type == VSH_OT_ALIAS) {
            int j;
            if (opt->flags || !opt->help)
                return -1; /* alias options are tracked by the original name */
            for (j = i + 1; cmd->opts[j].name; j++) {
                if (STREQ(opt->help, cmd->opts[j].name))
                    break;
            }
            if (!cmd->opts[j].name)
                return -1; /* alias option must map to a later option name */
            continue;
        }
E
Eric Blake 已提交
895 896
        if (opt->flags & VSH_OFLAG_REQ_OPT) {
            if (opt->flags & VSH_OFLAG_REQ)
L
Lai Jiangshan 已提交
897 898 899 900
                *opts_required |= 1 << i;
            continue;
        }

901
        *opts_need_arg |= 1 << i;
E
Eric Blake 已提交
902
        if (opt->flags & VSH_OFLAG_REQ) {
903 904 905 906 907 908
            if (optional)
                return -1; /* mandatory options must be listed first */
            *opts_required |= 1 << i;
        } else {
            optional = true;
        }
909 910 911

        if (opt->type == VSH_OT_ARGV && cmd->opts[i + 1].name)
            return -1; /* argv option must be listed last */
912 913 914 915
    }
    return 0;
}

916 917
static vshCmdOptDef helpopt = {"help", VSH_OT_BOOL, 0,
                               N_("print help for this function")};
918
static const vshCmdOptDef *
919
vshCmddefGetOption(vshControl *ctl, const vshCmdDef *cmd, const char *name,
920
                   uint32_t *opts_seen, int *opt_index)
921
{
922 923
    int i;

924 925 926 927
    if (STREQ(name, helpopt.name)) {
        return &helpopt;
    }

928 929
    for (i = 0; cmd->opts && cmd->opts[i].name; i++) {
        const vshCmdOptDef *opt = &cmd->opts[i];
930

931
        if (STREQ(opt->name, name)) {
E
Eric Blake 已提交
932 933 934 935
            if (opt->type == VSH_OT_ALIAS) {
                name = opt->help;
                continue;
            }
936
            if ((*opts_seen & (1 << i)) && opt->type != VSH_OT_ARGV) {
937 938 939
                vshError(ctl, _("option --%s already seen"), name);
                return NULL;
            }
940 941
            *opts_seen |= 1 << i;
            *opt_index = i;
K
Karel Zak 已提交
942
            return opt;
943 944 945
        }
    }

946 947 948 949
    if (STRNEQ(cmd->name, "help")) {
        vshError(ctl, _("command '%s' doesn't support option --%s"),
                 cmd->name, name);
    }
K
Karel Zak 已提交
950 951 952
    return NULL;
}

953
static const vshCmdOptDef *
954 955
vshCmddefGetData(const vshCmdDef *cmd, uint32_t *opts_need_arg,
                 uint32_t *opts_seen)
956
{
957
    int i;
958
    const vshCmdOptDef *opt;
K
Karel Zak 已提交
959

960 961 962 963
    if (!*opts_need_arg)
        return NULL;

    /* Grab least-significant set bit */
E
Eric Blake 已提交
964
    i = ffs(*opts_need_arg) - 1;
965
    opt = &cmd->opts[i];
966
    if (opt->type != VSH_OT_ARGV)
967
        *opts_need_arg &= ~(1 << i);
968
    *opts_seen |= 1 << i;
969
    return opt;
K
Karel Zak 已提交
970 971
}

972 973 974
/*
 * Checks for required options
 */
975
static int
976 977
vshCommandCheckOpts(vshControl *ctl, const vshCmd *cmd, uint32_t opts_required,
                    uint32_t opts_seen)
978
{
979
    const vshCmdDef *def = cmd->def;
980 981 982 983 984 985 986 987 988
    int i;

    opts_required &= ~opts_seen;
    if (!opts_required)
        return 0;

    for (i = 0; def->opts[i].name; i++) {
        if (opts_required & (1 << i)) {
            const vshCmdOptDef *opt = &def->opts[i];
989

990
            vshError(ctl,
991
                     opt->type == VSH_OT_DATA || opt->type == VSH_OT_ARGV ?
992 993 994
                     _("command '%s' requires <%s> option") :
                     _("command '%s' requires --%s option"),
                     def->name, opt->name);
995 996
        }
    }
997
    return -1;
998 999
}

E
Eric Blake 已提交
1000
const vshCmdDef *
1001 1002
vshCmddefSearch(const char *cmdname)
{
1003
    const vshCmdGrp *g;
1004
    const vshCmdDef *c;
1005

1006 1007
    for (g = cmdGroups; g->name; g++) {
        for (c = g->commands; c->name; c++) {
1008
            if (STREQ(c->name, cmdname))
1009 1010 1011 1012
                return c;
        }
    }

K
Karel Zak 已提交
1013 1014 1015
    return NULL;
}

E
Eric Blake 已提交
1016
const vshCmdGrp *
1017 1018 1019 1020 1021
vshCmdGrpSearch(const char *grpname)
{
    const vshCmdGrp *g;

    for (g = cmdGroups; g->name; g++) {
1022
        if (STREQ(g->name, grpname) || STREQ(g->keyword, grpname))
1023 1024 1025 1026 1027 1028
            return g;
    }

    return NULL;
}

E
Eric Blake 已提交
1029
bool
1030 1031 1032 1033 1034 1035 1036
vshCmdGrpHelp(vshControl *ctl, const char *grpname)
{
    const vshCmdGrp *grp = vshCmdGrpSearch(grpname);
    const vshCmdDef *cmd = NULL;

    if (!grp) {
        vshError(ctl, _("command group '%s' doesn't exist"), grpname);
E
Eric Blake 已提交
1037
        return false;
1038 1039 1040 1041 1042 1043 1044 1045 1046 1047
    } else {
        vshPrint(ctl, _(" %s (help keyword '%s'):\n"), grp->name,
                 grp->keyword);

        for (cmd = grp->commands; cmd->name; cmd++) {
            vshPrint(ctl, "    %-30s %s\n", cmd->name,
                     _(vshCmddefGetInfo(cmd, "help")));
        }
    }

E
Eric Blake 已提交
1048
    return true;
1049 1050
}

E
Eric Blake 已提交
1051
bool
1052
vshCmddefHelp(vshControl *ctl, const char *cmdname)
1053
{
1054
    const vshCmdDef *def = vshCmddefSearch(cmdname);
1055

K
Karel Zak 已提交
1056
    if (!def) {
1057
        vshError(ctl, _("command '%s' doesn't exist"), cmdname);
E
Eric Blake 已提交
1058
        return false;
1059
    } else {
E
Eric Blake 已提交
1060 1061
        /* Don't translate desc if it is "".  */
        const char *desc = vshCmddefGetInfo(def, "desc");
E
Eric Blake 已提交
1062
        const char *help = _(vshCmddefGetInfo(def, "help"));
1063
        char buf[256];
1064 1065
        uint32_t opts_need_arg;
        uint32_t opts_required;
1066
        bool shortopt = false; /* true if 'arg' works instead of '--opt arg' */
1067 1068 1069 1070

        if (vshCmddefOptParse(def, &opts_need_arg, &opts_required)) {
            vshError(ctl, _("internal error: bad options in command: '%s'"),
                     def->name);
E
Eric Blake 已提交
1071
            return false;
1072
        }
K
Karel Zak 已提交
1073

1074
        fputs(_("  NAME\n"), stdout);
1075 1076
        fprintf(stdout, "    %s - %s\n", def->name, help);

1077 1078 1079 1080 1081
        fputs(_("\n  SYNOPSIS\n"), stdout);
        fprintf(stdout, "    %s", def->name);
        if (def->opts) {
            const vshCmdOptDef *opt;
            for (opt = def->opts; opt->name; opt++) {
1082
                const char *fmt = "%s";
1083 1084
                switch (opt->type) {
                case VSH_OT_BOOL:
1085
                    fmt = "[--%s]";
1086 1087
                    break;
                case VSH_OT_INT:
E
Eric Blake 已提交
1088
                    /* xgettext:c-format */
E
Eric Blake 已提交
1089
                    fmt = ((opt->flags & VSH_OFLAG_REQ) ? "<%s>"
1090
                           : _("[--%s <number>]"));
1091 1092
                    if (!(opt->flags & VSH_OFLAG_REQ_OPT))
                        shortopt = true;
1093 1094
                    break;
                case VSH_OT_STRING:
E
Eric Blake 已提交
1095 1096
                    /* xgettext:c-format */
                    fmt = _("[--%s <string>]");
1097 1098
                    if (!(opt->flags & VSH_OFLAG_REQ_OPT))
                        shortopt = true;
1099 1100
                    break;
                case VSH_OT_DATA:
E
Eric Blake 已提交
1101
                    fmt = ((opt->flags & VSH_OFLAG_REQ) ? "<%s>" : "[<%s>]");
1102 1103
                    if (!(opt->flags & VSH_OFLAG_REQ_OPT))
                        shortopt = true;
1104 1105 1106
                    break;
                case VSH_OT_ARGV:
                    /* xgettext:c-format */
1107 1108 1109 1110 1111 1112 1113 1114
                    if (shortopt) {
                        fmt = (opt->flags & VSH_OFLAG_REQ)
                            ? _("{[--%s] <string>}...")
                            : _("[[--%s] <string>]...");
                    } else {
                        fmt = (opt->flags & VSH_OFLAG_REQ) ? _("<%s>...")
                            : _("[<%s>]...");
                    }
1115
                    break;
E
Eric Blake 已提交
1116 1117 1118
                case VSH_OT_ALIAS:
                    /* aliases are intentionally undocumented */
                    continue;
1119
                default:
1120
                    assert(0);
1121
                }
1122
                fputc(' ', stdout);
E
Eric Blake 已提交
1123
                fprintf(stdout, fmt, opt->name);
1124
            }
K
Karel Zak 已提交
1125
        }
1126 1127 1128
        fputc('\n', stdout);

        if (desc[0]) {
1129
            /* Print the description only if it's not empty.  */
1130
            fputs(_("\n  DESCRIPTION\n"), stdout);
E
Eric Blake 已提交
1131
            fprintf(stdout, "    %s\n", _(desc));
K
Karel Zak 已提交
1132
        }
1133

K
Karel Zak 已提交
1134
        if (def->opts) {
1135
            const vshCmdOptDef *opt;
1136
            fputs(_("\n  OPTIONS\n"), stdout);
1137
            for (opt = def->opts; opt->name; opt++) {
1138 1139
                switch (opt->type) {
                case VSH_OT_BOOL:
K
Karel Zak 已提交
1140
                    snprintf(buf, sizeof(buf), "--%s", opt->name);
1141 1142
                    break;
                case VSH_OT_INT:
1143
                    snprintf(buf, sizeof(buf),
E
Eric Blake 已提交
1144
                             (opt->flags & VSH_OFLAG_REQ) ? _("[--%s] <number>")
1145
                             : _("--%s <number>"), opt->name);
1146 1147
                    break;
                case VSH_OT_STRING:
1148
                    /* OT_STRING should never be VSH_OFLAG_REQ */
1149
                    snprintf(buf, sizeof(buf), _("--%s <string>"), opt->name);
1150 1151
                    break;
                case VSH_OT_DATA:
1152 1153
                    snprintf(buf, sizeof(buf), _("[--%s] <string>"),
                             opt->name);
1154 1155
                    break;
                case VSH_OT_ARGV:
1156 1157 1158
                    snprintf(buf, sizeof(buf),
                             shortopt ? _("[--%s] <string>") : _("<%s>"),
                             opt->name);
1159
                    break;
E
Eric Blake 已提交
1160 1161
                case VSH_OT_ALIAS:
                    continue;
1162 1163 1164
                default:
                    assert(0);
                }
1165

E
Eric Blake 已提交
1166
                fprintf(stdout, "    %-15s  %s\n", buf, _(opt->help));
1167
            }
K
Karel Zak 已提交
1168 1169 1170
        }
        fputc('\n', stdout);
    }
E
Eric Blake 已提交
1171
    return true;
K
Karel Zak 已提交
1172 1173 1174 1175 1176 1177
}

/* ---------------
 * Utils for work with runtime commands data
 * ---------------
 */
1178 1179 1180
static void
vshCommandOptFree(vshCmdOpt * arg)
{
K
Karel Zak 已提交
1181 1182
    vshCmdOpt *a = arg;

1183
    while (a) {
K
Karel Zak 已提交
1184
        vshCmdOpt *tmp = a;
1185

K
Karel Zak 已提交
1186 1187
        a = a->next;

1188 1189
        VIR_FREE(tmp->data);
        VIR_FREE(tmp);
K
Karel Zak 已提交
1190 1191 1192 1193
    }
}

static void
1194
vshCommandFree(vshCmd *cmd)
1195
{
K
Karel Zak 已提交
1196 1197
    vshCmd *c = cmd;

1198
    while (c) {
K
Karel Zak 已提交
1199
        vshCmd *tmp = c;
1200

K
Karel Zak 已提交
1201 1202 1203 1204
        c = c->next;

        if (tmp->opts)
            vshCommandOptFree(tmp->opts);
1205
        VIR_FREE(tmp);
K
Karel Zak 已提交
1206 1207 1208
    }
}

E
Eric Blake 已提交
1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219
/**
 * vshCommandOpt:
 * @cmd: parsed command line to search
 * @name: option name to search for
 * @opt: result of the search
 *
 * Look up an option passed to CMD by NAME.  Returns 1 with *OPT set
 * to the option if found, 0 with *OPT set to NULL if the name is
 * valid and the option is not required, -1 with *OPT set to NULL if
 * the option is required but not present, and -2 if NAME is not valid
 * (-2 indicates a programming error).  No error messages are issued.
K
Karel Zak 已提交
1220
 */
E
Eric Blake 已提交
1221
int
E
Eric Blake 已提交
1222
vshCommandOpt(const vshCmd *cmd, const char *name, vshCmdOpt **opt)
1223
{
E
Eric Blake 已提交
1224 1225
    vshCmdOpt *candidate = cmd->opts;
    const vshCmdOptDef *valid = cmd->def->opts;
1226

E
Eric Blake 已提交
1227 1228 1229 1230 1231 1232 1233
    /* See if option is present on command line.  */
    while (candidate) {
        if (STREQ(candidate->def->name, name)) {
            *opt = candidate;
            return 1;
        }
        candidate = candidate->next;
K
Karel Zak 已提交
1234
    }
E
Eric Blake 已提交
1235 1236 1237 1238 1239 1240 1241

    /* Option not present, see if command requires it.  */
    *opt = NULL;
    while (valid) {
        if (!valid->name)
            break;
        if (STREQ(name, valid->name))
E
Eric Blake 已提交
1242
            return (valid->flags & VSH_OFLAG_REQ) == 0 ? 0 : -1;
E
Eric Blake 已提交
1243 1244 1245 1246
        valid++;
    }
    /* If we got here, the name is unknown.  */
    return -2;
K
Karel Zak 已提交
1247 1248
}

E
Eric Blake 已提交
1249 1250
/**
 * vshCommandOptInt:
1251 1252 1253 1254 1255 1256 1257
 * @cmd command reference
 * @name option name
 * @value result
 *
 * Convert option to int
 * Return value:
 * >0 if option found and valid (@value updated)
E
Eric Blake 已提交
1258
 * 0 if option not found and not required (@value untouched)
1259
 * <0 in all other cases (@value untouched)
K
Karel Zak 已提交
1260
 */
E
Eric Blake 已提交
1261
int
1262
vshCommandOptInt(const vshCmd *cmd, const char *name, int *value)
1263
{
E
Eric Blake 已提交
1264 1265
    vshCmdOpt *arg;
    int ret;
1266

E
Eric Blake 已提交
1267 1268 1269 1270 1271 1272 1273
    ret = vshCommandOpt(cmd, name, &arg);
    if (ret <= 0)
        return ret;
    if (!arg->data) {
        /* only possible on bool, but if name is bool, this is a
         * programming bug */
        return -2;
1274
    }
E
Eric Blake 已提交
1275

E
Eric Blake 已提交
1276 1277 1278
    if (virStrToLong_i(arg->data, NULL, 10, value) < 0)
        return -1;
    return 1;
K
Karel Zak 已提交
1279 1280
}

1281

E
Eric Blake 已提交
1282 1283 1284 1285 1286 1287
/**
 * vshCommandOptUInt:
 * @cmd command reference
 * @name option name
 * @value result
 *
1288 1289 1290
 * Convert option to unsigned int
 * See vshCommandOptInt()
 */
E
Eric Blake 已提交
1291
int
1292 1293
vshCommandOptUInt(const vshCmd *cmd, const char *name, unsigned int *value)
{
E
Eric Blake 已提交
1294 1295
    vshCmdOpt *arg;
    int ret;
1296

E
Eric Blake 已提交
1297 1298 1299 1300 1301 1302 1303
    ret = vshCommandOpt(cmd, name, &arg);
    if (ret <= 0)
        return ret;
    if (!arg->data) {
        /* only possible on bool, but if name is bool, this is a
         * programming bug */
        return -2;
1304
    }
E
Eric Blake 已提交
1305

E
Eric Blake 已提交
1306 1307 1308
    if (virStrToLong_ui(arg->data, NULL, 10, value) < 0)
        return -1;
    return 1;
1309 1310 1311
}


1312
/*
E
Eric Blake 已提交
1313 1314 1315 1316 1317
 * vshCommandOptUL:
 * @cmd command reference
 * @name option name
 * @value result
 *
1318 1319 1320
 * Convert option to unsigned long
 * See vshCommandOptInt()
 */
E
Eric Blake 已提交
1321
int
1322
vshCommandOptUL(const vshCmd *cmd, const char *name, unsigned long *value)
1323
{
E
Eric Blake 已提交
1324 1325
    vshCmdOpt *arg;
    int ret;
1326

E
Eric Blake 已提交
1327 1328 1329 1330 1331 1332 1333
    ret = vshCommandOpt(cmd, name, &arg);
    if (ret <= 0)
        return ret;
    if (!arg->data) {
        /* only possible on bool, but if name is bool, this is a
         * programming bug */
        return -2;
1334
    }
E
Eric Blake 已提交
1335

E
Eric Blake 已提交
1336 1337 1338
    if (virStrToLong_ul(arg->data, NULL, 10, value) < 0)
        return -1;
    return 1;
1339 1340
}

E
Eric Blake 已提交
1341 1342 1343 1344 1345 1346
/**
 * vshCommandOptString:
 * @cmd command reference
 * @name option name
 * @value result
 *
K
Karel Zak 已提交
1347
 * Returns option as STRING
E
Eric Blake 已提交
1348 1349 1350 1351
 * Return value:
 * >0 if option found and valid (@value updated)
 * 0 if option not found and not required (@value untouched)
 * <0 in all other cases (@value untouched)
K
Karel Zak 已提交
1352
 */
E
Eric Blake 已提交
1353
int
1354
vshCommandOptString(const vshCmd *cmd, const char *name, const char **value)
1355
{
E
Eric Blake 已提交
1356 1357 1358 1359 1360 1361 1362 1363 1364 1365
    vshCmdOpt *arg;
    int ret;

    ret = vshCommandOpt(cmd, name, &arg);
    if (ret <= 0)
        return ret;
    if (!arg->data) {
        /* only possible on bool, but if name is bool, this is a
         * programming bug */
        return -2;
1366
    }
1367

E
Eric Blake 已提交
1368
    if (!*arg->data && !(arg->def->flags & VSH_OFLAG_EMPTY_OK)) {
E
Eric Blake 已提交
1369 1370 1371 1372
        return -1;
    }
    *value = arg->data;
    return 1;
K
Karel Zak 已提交
1373 1374
}

E
Eric Blake 已提交
1375 1376 1377 1378 1379 1380
/**
 * vshCommandOptLongLong:
 * @cmd command reference
 * @name option name
 * @value result
 *
1381
 * Returns option as long long
1382
 * See vshCommandOptInt()
1383
 */
E
Eric Blake 已提交
1384
int
1385 1386
vshCommandOptLongLong(const vshCmd *cmd, const char *name,
                      long long *value)
1387
{
E
Eric Blake 已提交
1388 1389
    vshCmdOpt *arg;
    int ret;
1390

E
Eric Blake 已提交
1391 1392 1393 1394 1395 1396 1397
    ret = vshCommandOpt(cmd, name, &arg);
    if (ret <= 0)
        return ret;
    if (!arg->data) {
        /* only possible on bool, but if name is bool, this is a
         * programming bug */
        return -2;
1398
    }
E
Eric Blake 已提交
1399

E
Eric Blake 已提交
1400 1401 1402
    if (virStrToLong_ll(arg->data, NULL, 10, value) < 0)
        return -1;
    return 1;
1403 1404
}

E
Eric Blake 已提交
1405 1406 1407 1408 1409 1410 1411 1412 1413
/**
 * vshCommandOptULongLong:
 * @cmd command reference
 * @name option name
 * @value result
 *
 * Returns option as long long
 * See vshCommandOptInt()
 */
E
Eric Blake 已提交
1414
int
1415 1416 1417
vshCommandOptULongLong(const vshCmd *cmd, const char *name,
                       unsigned long long *value)
{
E
Eric Blake 已提交
1418 1419
    vshCmdOpt *arg;
    int ret;
1420

E
Eric Blake 已提交
1421 1422 1423 1424 1425 1426 1427
    ret = vshCommandOpt(cmd, name, &arg);
    if (ret <= 0)
        return ret;
    if (!arg->data) {
        /* only possible on bool, but if name is bool, this is a
         * programming bug */
        return -2;
1428
    }
E
Eric Blake 已提交
1429

E
Eric Blake 已提交
1430 1431 1432
    if (virStrToLong_ull(arg->data, NULL, 10, value) < 0)
        return -1;
    return 1;
1433 1434 1435
}


E
Eric Blake 已提交
1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446
/**
 * vshCommandOptScaledInt:
 * @cmd command reference
 * @name option name
 * @value result
 * @scale default of 1 or 1024, if no suffix is present
 * @max maximum value permitted
 *
 * Returns option as long long, scaled according to suffix
 * See vshCommandOptInt()
 */
E
Eric Blake 已提交
1447
int
E
Eric Blake 已提交
1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465
vshCommandOptScaledInt(const vshCmd *cmd, const char *name,
                       unsigned long long *value, int scale,
                       unsigned long long max)
{
    const char *str;
    int ret;
    char *end;

    ret = vshCommandOptString(cmd, name, &str);
    if (ret <= 0)
        return ret;
    if (virStrToLong_ull(str, &end, 10, value) < 0 ||
        virScaleInteger(value, end, scale, max) < 0)
        return -1;
    return 1;
}


E
Eric Blake 已提交
1466 1467 1468 1469 1470 1471 1472 1473 1474
/**
 * vshCommandOptBool:
 * @cmd command reference
 * @name option name
 *
 * Returns true/false if the option exists.  Note that this does NOT
 * validate whether the option is actually boolean, or even whether
 * name is legal; so that this can be used to probe whether a data
 * option is present without actually using that data.
K
Karel Zak 已提交
1475
 */
E
Eric Blake 已提交
1476
bool
1477
vshCommandOptBool(const vshCmd *cmd, const char *name)
1478
{
E
Eric Blake 已提交
1479 1480 1481
    vshCmdOpt *dummy;

    return vshCommandOpt(cmd, name, &dummy) == 1;
K
Karel Zak 已提交
1482 1483
}

E
Eric Blake 已提交
1484 1485 1486 1487 1488
/**
 * vshCommandOptArgv:
 * @cmd command reference
 * @opt starting point for the search
 *
1489 1490
 * Returns the next argv argument after OPT (or the first one if OPT
 * is NULL), or NULL if no more are present.
1491
 *
1492
 * Requires that a VSH_OT_ARGV option be last in the
1493 1494
 * list of supported options in CMD->def->opts.
 */
E
Eric Blake 已提交
1495
const vshCmdOpt *
1496
vshCommandOptArgv(const vshCmd *cmd, const vshCmdOpt *opt)
1497
{
1498
    opt = opt ? opt->next : cmd->opts;
1499 1500

    while (opt) {
E
Eric Blake 已提交
1501
        if (opt->def->type == VSH_OT_ARGV) {
1502
            return opt;
1503 1504 1505 1506 1507 1508
        }
        opt = opt->next;
    }
    return NULL;
}

J
Jim Meyering 已提交
1509 1510 1511
/* Determine whether CMD->opts includes an option with name OPTNAME.
   If not, give a diagnostic and return false.
   If so, return true.  */
1512 1513
bool
vshCmdHasOption(vshControl *ctl, const vshCmd *cmd, const char *optname)
J
Jim Meyering 已提交
1514 1515 1516 1517 1518 1519
{
    /* Iterate through cmd->opts, to ensure that there is an entry
       with name OPTNAME and type VSH_OT_DATA. */
    bool found = false;
    const vshCmdOpt *opt;
    for (opt = cmd->opts; opt; opt = opt->next) {
1520
        if (STREQ(opt->def->name, optname) && opt->def->type == VSH_OT_DATA) {
J
Jim Meyering 已提交
1521 1522 1523 1524 1525 1526
            found = true;
            break;
        }
    }

    if (!found)
1527
        vshError(ctl, _("internal error: virsh %s: no %s VSH_OT_DATA option"),
J
Jim Meyering 已提交
1528 1529 1530
                 cmd->def->name, optname);
    return found;
}
1531

1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548
static bool
vshConnectionUsability(vshControl *ctl, virConnectPtr conn)
{
    if (!conn ||
        virConnectIsAlive(conn) == 0) {
        vshError(ctl, "%s", _("no valid connection"));
        return false;
    }

    /* The connection is considered dead only if
     * virConnectIsAlive() successfuly says so.
     */
    vshResetLibvirtError();

    return true;
}

K
Karel Zak 已提交
1549 1550 1551
/*
 * Executes command(s) and returns return code from last command
 */
E
Eric Blake 已提交
1552
static bool
1553
vshCommandRun(vshControl *ctl, const vshCmd *cmd)
1554
{
E
Eric Blake 已提交
1555
    bool ret = true;
1556 1557

    while (cmd) {
K
Karel Zak 已提交
1558
        struct timeval before, after;
1559
        bool enable_timing = ctl->timing;
1560

1561 1562
        if ((ctl->conn == NULL || disconnected) &&
            !(cmd->def->flags & VSH_CMD_FLAG_NOCONNECT))
1563 1564
            vshReconnect(ctl);

1565 1566 1567 1568
        if ((cmd->def->flags & VSH_CMD_FLAG_NOCONNECT) ||
            vshConnectionUsability(ctl, ctl->conn)) {
            if (enable_timing)
                GETTIMEOFDAY(&before);
1569

1570
            ret = cmd->def->handler(ctl, cmd);
K
Karel Zak 已提交
1571

1572 1573 1574 1575 1576 1577
            if (enable_timing)
                GETTIMEOFDAY(&after);
        } else {
            /* connection is not usable, return error */
            ret = false;
        }
1578

1579 1580 1581 1582 1583 1584 1585 1586 1587 1588
        /* try to automatically catch disconnections */
        if (!ret &&
            ((last_error != NULL) &&
             (((last_error->code == VIR_ERR_SYSTEM_ERROR) &&
               (last_error->domain == VIR_FROM_REMOTE)) ||
              (last_error->code == VIR_ERR_RPC) ||
              (last_error->code == VIR_ERR_NO_CONNECT) ||
              (last_error->code == VIR_ERR_INVALID_CONN))))
            disconnected++;

1589
        if (!ret)
E
Eric Blake 已提交
1590
            vshReportError(ctl);
J
John Levon 已提交
1591

1592
        if (!ret && disconnected != 0)
1593 1594
            vshReconnect(ctl);

1595
        if (STREQ(cmd->def->name, "quit"))        /* hack ... */
K
Karel Zak 已提交
1596 1597
            return ret;

E
Eric Blake 已提交
1598
        if (enable_timing) {
1599
            double diff_ms = (((after.tv_sec - before.tv_sec) * 1000.0) +
E
Eric Blake 已提交
1600 1601 1602 1603
                              ((after.tv_usec - before.tv_usec) / 1000.0));

            vshPrint(ctl, _("\n(Time: %.3f ms)\n\n"), diff_ms);
        } else {
K
Karel Zak 已提交
1604
            vshPrintExtra(ctl, "\n");
E
Eric Blake 已提交
1605
        }
K
Karel Zak 已提交
1606 1607 1608 1609 1610 1611
        cmd = cmd->next;
    }
    return ret;
}

/* ---------------
1612
 * Command parsing
K
Karel Zak 已提交
1613 1614 1615
 * ---------------
 */

1616 1617 1618 1619 1620 1621 1622
typedef enum {
    VSH_TK_ERROR, /* Failed to parse a token */
    VSH_TK_ARG, /* Arbitrary argument, might be option or empty */
    VSH_TK_SUBCMD_END, /* Separation between commands */
    VSH_TK_END /* No more commands */
} vshCommandToken;

E
Eric Blake 已提交
1623 1624 1625 1626
typedef struct _vshCommandParser vshCommandParser;
struct _vshCommandParser {
    vshCommandToken(*getNextArg)(vshControl *, vshCommandParser *,
                                 char **);
L
Lai Jiangshan 已提交
1627
    /* vshCommandStringGetArg() */
1628
    char *pos;
L
Lai Jiangshan 已提交
1629 1630 1631
    /* vshCommandArgvGetArg() */
    char **arg_pos;
    char **arg_end;
E
Eric Blake 已提交
1632
};
1633

E
Eric Blake 已提交
1634
static bool
1635
vshCommandParse(vshControl *ctl, vshCommandParser *parser)
1636
{
K
Karel Zak 已提交
1637 1638 1639
    char *tkdata = NULL;
    vshCmd *clast = NULL;
    vshCmdOpt *first = NULL;
1640

K
Karel Zak 已提交
1641 1642 1643 1644
    if (ctl->cmd) {
        vshCommandFree(ctl->cmd);
        ctl->cmd = NULL;
    }
1645

1646
    while (1) {
K
Karel Zak 已提交
1647
        vshCmdOpt *last = NULL;
1648
        const vshCmdDef *cmd = NULL;
1649
        vshCommandToken tk;
L
Lai Jiangshan 已提交
1650
        bool data_only = false;
1651 1652 1653
        uint32_t opts_need_arg = 0;
        uint32_t opts_required = 0;
        uint32_t opts_seen = 0;
1654

K
Karel Zak 已提交
1655
        first = NULL;
1656

1657
        while (1) {
1658
            const vshCmdOptDef *opt = NULL;
1659

K
Karel Zak 已提交
1660
            tkdata = NULL;
1661
            tk = parser->getNextArg(ctl, parser, &tkdata);
1662 1663

            if (tk == VSH_TK_ERROR)
K
Karel Zak 已提交
1664
                goto syntaxError;
H
Hu Tao 已提交
1665 1666
            if (tk != VSH_TK_ARG) {
                VIR_FREE(tkdata);
1667
                break;
H
Hu Tao 已提交
1668
            }
1669 1670

            if (cmd == NULL) {
K
Karel Zak 已提交
1671 1672
                /* first token must be command name */
                if (!(cmd = vshCmddefSearch(tkdata))) {
1673
                    vshError(ctl, _("unknown command: '%s'"), tkdata);
1674
                    goto syntaxError;   /* ... or ignore this command only? */
K
Karel Zak 已提交
1675
                }
1676 1677 1678 1679 1680 1681 1682
                if (vshCmddefOptParse(cmd, &opts_need_arg,
                                      &opts_required) < 0) {
                    vshError(ctl,
                             _("internal error: bad options in command: '%s'"),
                             tkdata);
                    goto syntaxError;
                }
1683
                VIR_FREE(tkdata);
L
Lai Jiangshan 已提交
1684 1685 1686 1687
            } else if (data_only) {
                goto get_data;
            } else if (tkdata[0] == '-' && tkdata[1] == '-' &&
                       c_isalnum(tkdata[2])) {
1688
                char *optstr = strchr(tkdata + 2, '=');
C
Cole Robinson 已提交
1689
                int opt_index = 0;
1690

1691 1692 1693 1694
                if (optstr) {
                    *optstr = '\0'; /* convert the '=' to '\0' */
                    optstr = vshStrdup(ctl, optstr + 1);
                }
1695
                /* Special case 'help' to ignore all spurious options */
1696
                if (!(opt = vshCmddefGetOption(ctl, cmd, tkdata + 2,
1697
                                               &opts_seen, &opt_index))) {
1698
                    VIR_FREE(optstr);
1699 1700
                    if (STREQ(cmd->name, "help"))
                        continue;
K
Karel Zak 已提交
1701 1702
                    goto syntaxError;
                }
1703
                VIR_FREE(tkdata);
K
Karel Zak 已提交
1704 1705 1706

                if (opt->type != VSH_OT_BOOL) {
                    /* option data */
1707 1708 1709
                    if (optstr)
                        tkdata = optstr;
                    else
1710
                        tk = parser->getNextArg(ctl, parser, &tkdata);
1711
                    if (tk == VSH_TK_ERROR)
K
Karel Zak 已提交
1712
                        goto syntaxError;
1713
                    if (tk != VSH_TK_ARG) {
1714
                        vshError(ctl,
1715
                                 _("expected syntax: --%s <%s>"),
1716 1717
                                 opt->name,
                                 opt->type ==
1718
                                 VSH_OT_INT ? _("number") : _("string"));
K
Karel Zak 已提交
1719 1720
                        goto syntaxError;
                    }
1721 1722
                    if (opt->type != VSH_OT_ARGV)
                        opts_need_arg &= ~(1 << opt_index);
1723 1724 1725 1726 1727 1728 1729 1730
                } else {
                    tkdata = NULL;
                    if (optstr) {
                        vshError(ctl, _("invalid '=' after option --%s"),
                                opt->name);
                        VIR_FREE(optstr);
                        goto syntaxError;
                    }
K
Karel Zak 已提交
1731
                }
L
Lai Jiangshan 已提交
1732 1733 1734 1735
            } else if (tkdata[0] == '-' && tkdata[1] == '-' &&
                       tkdata[2] == '\0') {
                data_only = true;
                continue;
1736
            } else {
L
Lai Jiangshan 已提交
1737
get_data:
1738
                /* Special case 'help' to ignore spurious data */
1739
                if (!(opt = vshCmddefGetData(cmd, &opts_need_arg,
1740 1741
                                             &opts_seen)) &&
                     STRNEQ(cmd->name, "help")) {
1742
                    vshError(ctl, _("unexpected data '%s'"), tkdata);
K
Karel Zak 已提交
1743 1744 1745 1746 1747
                    goto syntaxError;
                }
            }
            if (opt) {
                /* save option */
1748
                vshCmdOpt *arg = vshMalloc(ctl, sizeof(vshCmdOpt));
1749

K
Karel Zak 已提交
1750 1751 1752 1753
                arg->def = opt;
                arg->data = tkdata;
                arg->next = NULL;
                tkdata = NULL;
1754

K
Karel Zak 已提交
1755 1756 1757 1758 1759
                if (!first)
                    first = arg;
                if (last)
                    last->next = arg;
                last = arg;
1760

1761
                vshDebug(ctl, VSH_ERR_INFO, "%s: %s(%s): %s\n",
1762 1763
                         cmd->name,
                         opt->name,
1764 1765
                         opt->type != VSH_OT_BOOL ? _("optdata") : _("bool"),
                         opt->type != VSH_OT_BOOL ? arg->data : _("(none)"));
K
Karel Zak 已提交
1766 1767
            }
        }
1768

D
Daniel Veillard 已提交
1769
        /* command parsed -- allocate new struct for the command */
K
Karel Zak 已提交
1770
        if (cmd) {
1771
            vshCmd *c = vshMalloc(ctl, sizeof(vshCmd));
1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790
            vshCmdOpt *tmpopt = first;

            /* if we encountered --help, replace parsed command with
             * 'help <cmdname>' */
            for (tmpopt = first; tmpopt; tmpopt = tmpopt->next) {
                if (STRNEQ(tmpopt->def->name, "help"))
                    continue;

                vshCommandOptFree(first);
                first = vshMalloc(ctl, sizeof(vshCmdOpt));
                first->def = &(opts_help[0]);
                first->data = vshStrdup(ctl, cmd->name);
                first->next = NULL;

                cmd = vshCmddefSearch("help");
                opts_required = 0;
                opts_seen = 0;
                break;
            }
1791

K
Karel Zak 已提交
1792 1793 1794 1795
            c->opts = first;
            c->def = cmd;
            c->next = NULL;

1796
            if (vshCommandCheckOpts(ctl, c, opts_required, opts_seen) < 0) {
1797
                VIR_FREE(c);
1798
                goto syntaxError;
1799
            }
1800

K
Karel Zak 已提交
1801 1802 1803 1804 1805 1806
            if (!ctl->cmd)
                ctl->cmd = c;
            if (clast)
                clast->next = c;
            clast = c;
        }
1807 1808 1809

        if (tk == VSH_TK_END)
            break;
K
Karel Zak 已提交
1810
    }
1811

E
Eric Blake 已提交
1812
    return true;
K
Karel Zak 已提交
1813

1814
 syntaxError:
1815
    if (ctl->cmd) {
K
Karel Zak 已提交
1816
        vshCommandFree(ctl->cmd);
1817 1818
        ctl->cmd = NULL;
    }
K
Karel Zak 已提交
1819 1820
    if (first)
        vshCommandOptFree(first);
1821
    VIR_FREE(tkdata);
E
Eric Blake 已提交
1822
    return false;
K
Karel Zak 已提交
1823 1824
}

1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842
/* --------------------
 * Command argv parsing
 * --------------------
 */

static vshCommandToken ATTRIBUTE_NONNULL(2) ATTRIBUTE_NONNULL(3)
vshCommandArgvGetArg(vshControl *ctl, vshCommandParser *parser, char **res)
{
    if (parser->arg_pos == parser->arg_end) {
        *res = NULL;
        return VSH_TK_END;
    }

    *res = vshStrdup(ctl, *parser->arg_pos);
    parser->arg_pos++;
    return VSH_TK_ARG;
}

E
Eric Blake 已提交
1843 1844
static bool
vshCommandArgvParse(vshControl *ctl, int nargs, char **argv)
1845 1846 1847 1848
{
    vshCommandParser parser;

    if (nargs <= 0)
E
Eric Blake 已提交
1849
        return false;
1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921

    parser.arg_pos = argv;
    parser.arg_end = argv + nargs;
    parser.getNextArg = vshCommandArgvGetArg;
    return vshCommandParse(ctl, &parser);
}

/* ----------------------
 * Command string parsing
 * ----------------------
 */

static vshCommandToken ATTRIBUTE_NONNULL(2) ATTRIBUTE_NONNULL(3)
vshCommandStringGetArg(vshControl *ctl, vshCommandParser *parser, char **res)
{
    bool single_quote = false;
    bool double_quote = false;
    int sz = 0;
    char *p = parser->pos;
    char *q = vshStrdup(ctl, p);

    *res = q;

    while (*p && (*p == ' ' || *p == '\t'))
        p++;

    if (*p == '\0')
        return VSH_TK_END;
    if (*p == ';') {
        parser->pos = ++p;             /* = \0 or begin of next command */
        return VSH_TK_SUBCMD_END;
    }

    while (*p) {
        /* end of token is blank space or ';' */
        if (!double_quote && !single_quote &&
            (*p == ' ' || *p == '\t' || *p == ';'))
            break;

        if (!double_quote && *p == '\'') { /* single quote */
            single_quote = !single_quote;
            p++;
            continue;
        } else if (!single_quote && *p == '\\') { /* escape */
            /*
             * The same as the bash, a \ in "" is an escaper,
             * but a \ in '' is not an escaper.
             */
            p++;
            if (*p == '\0') {
                vshError(ctl, "%s", _("dangling \\"));
                return VSH_TK_ERROR;
            }
        } else if (!single_quote && *p == '"') { /* double quote */
            double_quote = !double_quote;
            p++;
            continue;
        }

        *q++ = *p++;
        sz++;
    }
    if (double_quote) {
        vshError(ctl, "%s", _("missing \""));
        return VSH_TK_ERROR;
    }

    *q = '\0';
    parser->pos = p;
    return VSH_TK_ARG;
}

E
Eric Blake 已提交
1922 1923
static bool
vshCommandStringParse(vshControl *ctl, char *cmdstr)
1924 1925 1926 1927
{
    vshCommandParser parser;

    if (cmdstr == NULL || *cmdstr == '\0')
E
Eric Blake 已提交
1928
        return false;
1929 1930 1931 1932 1933 1934

    parser.pos = cmdstr;
    parser.getNextArg = vshCommandStringGetArg;
    return vshCommandParse(ctl, &parser);
}

K
Karel Zak 已提交
1935
/* ---------------
1936
 * Misc utils
K
Karel Zak 已提交
1937 1938
 * ---------------
 */
E
Eric Blake 已提交
1939
int
1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966
vshDomainState(vshControl *ctl, virDomainPtr dom, int *reason)
{
    virDomainInfo info;

    if (reason)
        *reason = -1;

    if (!ctl->useGetInfo) {
        int state;
        if (virDomainGetState(dom, &state, reason, 0) < 0) {
            virErrorPtr err = virGetLastError();
            if (err && err->code == VIR_ERR_NO_SUPPORT)
                ctl->useGetInfo = true;
            else
                return -1;
        } else {
            return state;
        }
    }

    /* fall back to virDomainGetInfo if virDomainGetState is not supported */
    if (virDomainGetInfo(dom, &info) < 0)
        return -1;
    else
        return info.state;
}

1967 1968
/* Return a non-NULL string representation of a typed parameter; exit
 * if we are out of memory.  */
E
Eric Blake 已提交
1969
char *
1970 1971 1972 1973 1974
vshGetTypedParamValue(vshControl *ctl, virTypedParameterPtr item)
{
    int ret = 0;
    char *str = NULL;

1975
    switch (item->type) {
1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999
    case VIR_TYPED_PARAM_INT:
        ret = virAsprintf(&str, "%d", item->value.i);
        break;

    case VIR_TYPED_PARAM_UINT:
        ret = virAsprintf(&str, "%u", item->value.ui);
        break;

    case VIR_TYPED_PARAM_LLONG:
        ret = virAsprintf(&str, "%lld", item->value.l);
        break;

    case VIR_TYPED_PARAM_ULLONG:
        ret = virAsprintf(&str, "%llu", item->value.ul);
        break;

    case VIR_TYPED_PARAM_DOUBLE:
        ret = virAsprintf(&str, "%f", item->value.d);
        break;

    case VIR_TYPED_PARAM_BOOLEAN:
        ret = virAsprintf(&str, "%s", item->value.b ? _("yes") : _("no"));
        break;

2000 2001 2002 2003
    case VIR_TYPED_PARAM_STRING:
        str = vshStrdup(ctl, item->value.s);
        break;

2004
    default:
2005
        vshError(ctl, _("unimplemented parameter type %d"), item->type);
2006 2007
    }

2008
    if (ret < 0) {
2009
        vshError(ctl, "%s", _("Out of memory"));
2010 2011
        exit(EXIT_FAILURE);
    }
2012 2013 2014
    return str;
}

E
Eric Blake 已提交
2015
virTypedParameterPtr
2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034
vshFindTypedParamByName(const char *name, virTypedParameterPtr list, int count)
{
    int i = count;
    virTypedParameterPtr found = list;

    if (!list || !name)
        return NULL;

    while (i-- > 0) {
        if (STREQ(name, found->field))
            return found;

        found++; /* go to next struct in array */
    }

    /* not found */
    return NULL;
}

E
Eric Blake 已提交
2035
void
2036
vshDebug(vshControl *ctl, int level, const char *format, ...)
2037
{
K
Karel Zak 已提交
2038
    va_list ap;
2039
    char *str;
K
Karel Zak 已提交
2040

2041 2042 2043 2044 2045 2046 2047
    /* Aligning log levels to that of libvirt.
     * Traces with levels >=  user-specified-level
     * gets logged into file
     */
    if (level < ctl->debug)
        return;

2048
    va_start(ap, format);
2049
    vshOutputLogFile(ctl, level, format, ap);
2050 2051
    va_end(ap);

K
Karel Zak 已提交
2052
    va_start(ap, format);
2053 2054 2055 2056 2057
    if (virVasprintf(&str, format, ap) < 0) {
        /* Skip debug messages on low memory */
        va_end(ap);
        return;
    }
K
Karel Zak 已提交
2058
    va_end(ap);
2059 2060
    fputs(str, stdout);
    VIR_FREE(str);
K
Karel Zak 已提交
2061 2062
}

E
Eric Blake 已提交
2063
void
2064
vshPrintExtra(vshControl *ctl, const char *format, ...)
2065
{
K
Karel Zak 已提交
2066
    va_list ap;
2067
    char *str;
2068

2069
    if (ctl && ctl->quiet)
K
Karel Zak 已提交
2070
        return;
2071

K
Karel Zak 已提交
2072
    va_start(ap, format);
2073 2074 2075 2076 2077
    if (virVasprintf(&str, format, ap) < 0) {
        vshError(ctl, "%s", _("Out of memory"));
        va_end(ap);
        return;
    }
K
Karel Zak 已提交
2078
    va_end(ap);
2079
    fputs(str, stdout);
2080
    VIR_FREE(str);
K
Karel Zak 已提交
2081 2082
}

K
Karel Zak 已提交
2083

E
Eric Blake 已提交
2084
void
2085
vshError(vshControl *ctl, const char *format, ...)
2086
{
K
Karel Zak 已提交
2087
    va_list ap;
2088
    char *str;
2089

2090 2091 2092 2093 2094
    if (ctl != NULL) {
        va_start(ap, format);
        vshOutputLogFile(ctl, VSH_ERR_ERROR, format, ap);
        va_end(ap);
    }
2095

2096 2097 2098 2099
    /* Most output is to stdout, but if someone ran virsh 2>&1, then
     * printing to stderr will not interleave correctly with stdout
     * unless we flush between every transition between streams.  */
    fflush(stdout);
2100
    fputs(_("error: "), stderr);
2101

K
Karel Zak 已提交
2102
    va_start(ap, format);
2103 2104 2105
    /* We can't recursively call vshError on an OOM situation, so ignore
       failure here. */
    ignore_value(virVasprintf(&str, format, ap));
K
Karel Zak 已提交
2106 2107
    va_end(ap);

2108
    fprintf(stderr, "%s\n", NULLSTR(str));
2109
    fflush(stderr);
2110
    VIR_FREE(str);
K
Karel Zak 已提交
2111 2112
}

2113

J
Jiri Denemark 已提交
2114 2115 2116 2117 2118
static void
vshEventLoop(void *opaque)
{
    vshControl *ctl = opaque;

2119 2120 2121 2122 2123 2124 2125 2126 2127 2128
    while (1) {
        bool quit;
        virMutexLock(&ctl->lock);
        quit = ctl->quit;
        virMutexUnlock(&ctl->lock);

        if (quit)
            break;

        if (virEventRunDefaultImpl() < 0)
E
Eric Blake 已提交
2129
            vshReportError(ctl);
J
Jiri Denemark 已提交
2130 2131 2132 2133
    }
}


K
Karel Zak 已提交
2134
/*
2135
 * Initialize connection.
K
Karel Zak 已提交
2136
 */
E
Eric Blake 已提交
2137
static bool
2138
vshInit(vshControl *ctl)
2139
{
2140 2141
    char *debugEnv;

K
Karel Zak 已提交
2142
    if (ctl->conn)
E
Eric Blake 已提交
2143
        return false;
K
Karel Zak 已提交
2144

J
Jiri Denemark 已提交
2145
    if (ctl->debug == VSH_DEBUG_DEFAULT) {
2146 2147 2148
        /* log level not set from commandline, check env variable */
        debugEnv = getenv("VIRSH_DEBUG");
        if (debugEnv) {
J
Jiri Denemark 已提交
2149 2150 2151
            int debug;
            if (virStrToLong_i(debugEnv, NULL, 10, &debug) < 0 ||
                debug < VSH_ERR_DEBUG || debug > VSH_ERR_ERROR) {
2152 2153
                vshError(ctl, "%s",
                         _("VIRSH_DEBUG not set with a valid numeric value"));
J
Jiri Denemark 已提交
2154 2155
            } else {
                ctl->debug = debug;
2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167
            }
        }
    }

    if (ctl->logfile == NULL) {
        /* log file not set from cmdline */
        debugEnv = getenv("VIRSH_LOG_FILE");
        if (debugEnv && *debugEnv) {
            ctl->logfile = vshStrdup(ctl, debugEnv);
        }
    }

2168 2169
    vshOpenLogFile(ctl);

2170 2171
    /* set up the library error handler */
    virSetErrorFunc(NULL, virshErrorHandler);
2172

2173
    if (virEventRegisterDefaultImpl() < 0)
E
Eric Blake 已提交
2174
        return false;
2175

J
Jiri Denemark 已提交
2176 2177 2178 2179
    if (virThreadCreate(&ctl->eventLoop, true, vshEventLoop, ctl) < 0)
        return false;
    ctl->eventLoopStarted = true;

2180 2181 2182 2183
    if (ctl->name) {
        ctl->conn = virConnectOpenAuth(ctl->name,
                                       virConnectAuthPtrDefault,
                                       ctl->readonly ? VIR_CONNECT_RO : 0);
2184

2185 2186 2187 2188 2189 2190 2191
        /* Connecting to a named connection must succeed, but we delay
         * connecting to the default connection until we need it
         * (since the first command might be 'connect' which allows a
         * non-default connection, or might be 'help' which needs no
         * connection).
         */
        if (!ctl->conn) {
E
Eric Blake 已提交
2192
            vshReportError(ctl);
2193 2194 2195
            vshError(ctl, "%s", _("failed to connect to the hypervisor"));
            return false;
        }
2196
    }
K
Karel Zak 已提交
2197

E
Eric Blake 已提交
2198
    return true;
K
Karel Zak 已提交
2199 2200
}

2201 2202
#define LOGFILE_FLAGS (O_WRONLY | O_APPEND | O_CREAT | O_SYNC)

2203 2204 2205 2206 2207
/**
 * vshOpenLogFile:
 *
 * Open log file.
 */
E
Eric Blake 已提交
2208
void
2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221
vshOpenLogFile(vshControl *ctl)
{
    struct stat st;

    if (ctl->logfile == NULL)
        return;

    /* check log file */
    if (stat(ctl->logfile, &st) == -1) {
        switch (errno) {
            case ENOENT:
                break;
            default:
2222
                vshError(ctl, "%s",
J
Jim Meyering 已提交
2223
                         _("failed to get the log file information"));
2224
                exit(EXIT_FAILURE);
2225 2226 2227
        }
    } else {
        if (!S_ISREG(st.st_mode)) {
2228 2229
            vshError(ctl, "%s", _("the log path is not a file"));
            exit(EXIT_FAILURE);
2230 2231 2232 2233
        }
    }

    /* log file open */
2234
    if ((ctl->log_fd = open(ctl->logfile, LOGFILE_FLAGS, FILE_MODE)) < 0) {
2235
        vshError(ctl, "%s",
J
Jim Meyering 已提交
2236
                 _("failed to open the log file. check the log file path"));
2237
        exit(EXIT_FAILURE);
2238 2239 2240 2241 2242 2243 2244 2245
    }
}

/**
 * vshOutputLogFile:
 *
 * Outputting an error to log file.
 */
E
Eric Blake 已提交
2246
void
2247 2248
vshOutputLogFile(vshControl *ctl, int log_level, const char *msg_format,
                 va_list ap)
2249
{
2250 2251 2252
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    char *str;
    size_t len;
2253
    const char *lvl = "";
2254
    time_t stTime;
2255
    struct tm stTm;
2256 2257 2258 2259 2260 2261 2262 2263 2264

    if (ctl->log_fd == -1)
        return;

    /**
     * create log format
     *
     * [YYYY.MM.DD HH:MM:SS SIGNATURE PID] LOG_LEVEL message
    */
2265 2266
    time(&stTime);
    localtime_r(&stTime, &stTm);
2267
    virBufferAsprintf(&buf, "[%d.%02d.%02d %02d:%02d:%02d %s %d] ",
2268 2269 2270 2271 2272 2273
                      (1900 + stTm.tm_year),
                      (1 + stTm.tm_mon),
                      stTm.tm_mday,
                      stTm.tm_hour,
                      stTm.tm_min,
                      stTm.tm_sec,
2274 2275
                      SIGN_NAME,
                      (int) getpid());
2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295
    switch (log_level) {
        case VSH_ERR_DEBUG:
            lvl = LVL_DEBUG;
            break;
        case VSH_ERR_INFO:
            lvl = LVL_INFO;
            break;
        case VSH_ERR_NOTICE:
            lvl = LVL_INFO;
            break;
        case VSH_ERR_WARNING:
            lvl = LVL_WARNING;
            break;
        case VSH_ERR_ERROR:
            lvl = LVL_ERROR;
            break;
        default:
            lvl = LVL_DEBUG;
            break;
    }
2296 2297 2298
    virBufferAsprintf(&buf, "%s ", lvl);
    virBufferVasprintf(&buf, msg_format, ap);
    virBufferAddChar(&buf, '\n');
2299

2300 2301
    if (virBufferError(&buf))
        goto error;
2302

2303 2304 2305 2306 2307
    str = virBufferContentAndReset(&buf);
    len = strlen(str);
    if (len > 1 && str[len - 2] == '\n') {
        str[len - 1] = '\0';
        len--;
2308
    }
2309

2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320
    /* write log */
    if (safewrite(ctl->log_fd, str, len) < 0)
        goto error;

    return;

error:
    vshCloseLogFile(ctl);
    vshError(ctl, "%s", _("failed to write the log file"));
    virBufferFreeAndReset(&buf);
    VIR_FREE(str);
2321 2322 2323 2324 2325 2326 2327
}

/**
 * vshCloseLogFile:
 *
 * Close log file.
 */
E
Eric Blake 已提交
2328
void
2329 2330
vshCloseLogFile(vshControl *ctl)
{
2331 2332
    char ebuf[1024];

2333
    /* log file close */
2334 2335
    if (VIR_CLOSE(ctl->log_fd) < 0) {
        vshError(ctl, _("%s: failed to write log file: %s"),
2336 2337
                 ctl->logfile ? ctl->logfile : "?",
                 virStrerror(errno, ebuf, sizeof(ebuf)));
2338 2339 2340
    }

    if (ctl->logfile) {
2341
        VIR_FREE(ctl->logfile);
2342 2343 2344 2345
        ctl->logfile = NULL;
    }
}

2346
#ifdef USE_READLINE
2347

K
Karel Zak 已提交
2348 2349 2350 2351 2352
/* -----------------
 * Readline stuff
 * -----------------
 */

2353
/*
K
Karel Zak 已提交
2354 2355
 * Generator function for command completion.  STATE lets us
 * know whether to start from scratch; without any state
2356
 * (i.e. STATE == 0), then we start at the top of the list.
K
Karel Zak 已提交
2357 2358
 */
static char *
2359 2360
vshReadlineCommandGenerator(const char *text, int state)
{
2361
    static int grp_list_index, cmd_list_index, len;
K
Karel Zak 已提交
2362
    const char *name;
2363 2364
    const vshCmdGrp *grp;
    const vshCmdDef *cmds;
K
Karel Zak 已提交
2365 2366

    if (!state) {
2367 2368
        grp_list_index = 0;
        cmd_list_index = 0;
2369
        len = strlen(text);
K
Karel Zak 已提交
2370 2371
    }

2372 2373
    grp = cmdGroups;

K
Karel Zak 已提交
2374
    /* Return the next name which partially matches from the
2375
     * command list.
K
Karel Zak 已提交
2376
     */
2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390
    while (grp[grp_list_index].name) {
        cmds = grp[grp_list_index].commands;

        if (cmds[cmd_list_index].name) {
            while ((name = cmds[cmd_list_index].name)) {
                cmd_list_index++;

                if (STREQLEN(name, text, len))
                    return vshStrdup(NULL, name);
            }
        } else {
            cmd_list_index = 0;
            grp_list_index++;
        }
K
Karel Zak 已提交
2391 2392 2393 2394 2395 2396 2397
    }

    /* If no names matched, then return NULL. */
    return NULL;
}

static char *
2398 2399
vshReadlineOptionsGenerator(const char *text, int state)
{
K
Karel Zak 已提交
2400
    static int list_index, len;
2401
    static const vshCmdDef *cmd = NULL;
K
Karel Zak 已提交
2402
    const char *name;
K
Karel Zak 已提交
2403 2404 2405 2406 2407 2408 2409 2410 2411

    if (!state) {
        /* determine command name */
        char *p;
        char *cmdname;

        if (!(p = strchr(rl_line_buffer, ' ')))
            return NULL;

2412
        cmdname = vshCalloc(NULL, (p - rl_line_buffer) + 1, 1);
2413
        memcpy(cmdname, rl_line_buffer, p - rl_line_buffer);
K
Karel Zak 已提交
2414 2415 2416

        cmd = vshCmddefSearch(cmdname);
        list_index = 0;
2417
        len = strlen(text);
2418
        VIR_FREE(cmdname);
K
Karel Zak 已提交
2419 2420 2421 2422
    }

    if (!cmd)
        return NULL;
2423

2424 2425 2426
    if (!cmd->opts)
        return NULL;

K
Karel Zak 已提交
2427
    while ((name = cmd->opts[list_index].name)) {
2428
        const vshCmdOptDef *opt = &cmd->opts[list_index];
K
Karel Zak 已提交
2429
        char *res;
2430

K
Karel Zak 已提交
2431
        list_index++;
2432

2433
        if (opt->type == VSH_OT_DATA || opt->type == VSH_OT_ARGV)
K
Karel Zak 已提交
2434 2435
            /* ignore non --option */
            continue;
2436

K
Karel Zak 已提交
2437
        if (len > 2) {
2438
            if (STRNEQLEN(name, text + 2, len - 2))
K
Karel Zak 已提交
2439 2440
                continue;
        }
2441
        res = vshMalloc(NULL, strlen(name) + 3);
2442
        snprintf(res, strlen(name) + 3,  "--%s", name);
K
Karel Zak 已提交
2443 2444 2445 2446 2447 2448 2449 2450
        return res;
    }

    /* If no names matched, then return NULL. */
    return NULL;
}

static char **
2451 2452 2453
vshReadlineCompletion(const char *text, int start,
                      int end ATTRIBUTE_UNUSED)
{
K
Karel Zak 已提交
2454 2455
    char **matches = (char **) NULL;

2456
    if (start == 0)
K
Karel Zak 已提交
2457
        /* command name generator */
2458
        matches = rl_completion_matches(text, vshReadlineCommandGenerator);
K
Karel Zak 已提交
2459 2460
    else
        /* commands options */
2461
        matches = rl_completion_matches(text, vshReadlineOptionsGenerator);
K
Karel Zak 已提交
2462 2463 2464 2465
    return matches;
}


2466 2467
static int
vshReadlineInit(vshControl *ctl)
2468
{
2469 2470
    char *userdir = NULL;

K
Karel Zak 已提交
2471 2472 2473 2474 2475
    /* Allow conditional parsing of the ~/.inputrc file. */
    rl_readline_name = "virsh";

    /* Tell the completer that we want a crack first. */
    rl_attempted_completion_function = vshReadlineCompletion;
2476 2477 2478

    /* Limit the total size of the history buffer */
    stifle_history(500);
2479

2480
    /* Prepare to read/write history from/to the $XDG_CACHE_HOME/virsh/history file */
2481
    userdir = virGetUserCacheDirectory();
2482

2483 2484
    if (userdir == NULL) {
        vshError(ctl, "%s", _("Could not determine home directory"));
2485
        return -1;
2486
    }
2487

2488
    if (virAsprintf(&ctl->historydir, "%s/virsh", userdir) < 0) {
2489
        vshError(ctl, "%s", _("Out of memory"));
2490
        VIR_FREE(userdir);
2491 2492 2493 2494 2495
        return -1;
    }

    if (virAsprintf(&ctl->historyfile, "%s/history", ctl->historydir) < 0) {
        vshError(ctl, "%s", _("Out of memory"));
2496
        VIR_FREE(userdir);
2497 2498 2499
        return -1;
    }

2500
    VIR_FREE(userdir);
2501 2502 2503 2504 2505 2506 2507

    read_history(ctl->historyfile);

    return 0;
}

static void
2508
vshReadlineDeinit(vshControl *ctl)
2509 2510
{
    if (ctl->historyfile != NULL) {
2511 2512
        if (virFileMakePathWithMode(ctl->historydir, 0755) < 0 &&
            errno != EEXIST) {
2513 2514
            char ebuf[1024];
            vshError(ctl, _("Failed to create '%s': %s"),
2515
                     ctl->historydir, virStrerror(errno, ebuf, sizeof(ebuf)));
E
Eric Blake 已提交
2516
        } else {
2517
            write_history(ctl->historyfile);
E
Eric Blake 已提交
2518
        }
2519 2520
    }

2521 2522
    VIR_FREE(ctl->historydir);
    VIR_FREE(ctl->historyfile);
K
Karel Zak 已提交
2523 2524
}

2525
static char *
2526
vshReadline(vshControl *ctl ATTRIBUTE_UNUSED, const char *prompt)
2527
{
2528
    return readline(prompt);
2529 2530
}

2531
#else /* !USE_READLINE */
2532

2533
static int
2534
vshReadlineInit(vshControl *ctl ATTRIBUTE_UNUSED)
2535 2536 2537 2538 2539
{
    /* empty */
    return 0;
}

2540
static void
2541
vshReadlineDeinit(vshControl *ctl ATTRIBUTE_UNUSED)
2542 2543 2544 2545 2546
{
    /* empty */
}

static char *
2547
vshReadline(vshControl *ctl, const char *prompt)
2548 2549 2550 2551 2552
{
    char line[1024];
    char *r;
    int len;

2553 2554
    fputs(prompt, stdout);
    r = fgets(line, sizeof(line), stdin);
2555 2556 2557
    if (r == NULL) return NULL; /* EOF */

    /* Chomp trailing \n */
2558
    len = strlen(r);
2559 2560 2561
    if (len > 0 && r[len-1] == '\n')
        r[len-1] = '\0';

2562
    return vshStrdup(ctl, r);
2563 2564
}

2565
#endif /* !USE_READLINE */
2566

2567 2568 2569 2570 2571 2572
static void
vshDeinitTimer(int timer ATTRIBUTE_UNUSED, void *opaque ATTRIBUTE_UNUSED)
{
    /* nothing to be done here */
}

K
Karel Zak 已提交
2573
/*
J
Jim Meyering 已提交
2574
 * Deinitialize virsh
K
Karel Zak 已提交
2575
 */
E
Eric Blake 已提交
2576
static bool
2577
vshDeinit(vshControl *ctl)
2578
{
2579
    vshReadlineDeinit(ctl);
2580
    vshCloseLogFile(ctl);
2581
    VIR_FREE(ctl->name);
K
Karel Zak 已提交
2582
    if (ctl->conn) {
2583 2584 2585
        int ret;
        if ((ret = virConnectClose(ctl->conn)) != 0) {
            vshError(ctl, _("Failed to disconnect from the hypervisor, %d leaked reference(s)"), ret);
K
Karel Zak 已提交
2586 2587
        }
    }
D
Daniel P. Berrange 已提交
2588 2589
    virResetLastError();

J
Jiri Denemark 已提交
2590
    if (ctl->eventLoopStarted) {
2591 2592 2593 2594
        int timer;

        virMutexLock(&ctl->lock);
        ctl->quit = true;
J
Jiri Denemark 已提交
2595
        /* HACK: Add a dummy timeout to break event loop */
2596 2597 2598 2599 2600
        timer = virEventAddTimeout(0, vshDeinitTimer, NULL, NULL);
        virMutexUnlock(&ctl->lock);

        virThreadJoin(&ctl->eventLoop);

J
Jiri Denemark 已提交
2601 2602 2603 2604 2605 2606
        if (timer != -1)
            virEventRemoveTimeout(timer);

        ctl->eventLoopStarted = false;
    }

2607 2608
    virMutexDestroy(&ctl->lock);

E
Eric Blake 已提交
2609
    return true;
K
Karel Zak 已提交
2610
}
2611

K
Karel Zak 已提交
2612 2613 2614
/*
 * Print usage
 */
E
Eric Blake 已提交
2615
static void
2616
vshUsage(void)
2617
{
2618
    const vshCmdGrp *grp;
2619
    const vshCmdDef *cmd;
2620

L
Lai Jiangshan 已提交
2621 2622
    fprintf(stdout, _("\n%s [options]... [<command_string>]"
                      "\n%s [options]... <command> [args...]\n\n"
2623
                      "  options:\n"
2624
                      "    -c | --connect=URI      hypervisor connection URI\n"
2625
                      "    -r | --readonly         connect readonly\n"
2626
                      "    -d | --debug=NUM        debug level [0-4]\n"
2627 2628 2629
                      "    -h | --help             this help\n"
                      "    -q | --quiet            quiet mode\n"
                      "    -t | --timing           print timing information\n"
2630 2631 2632 2633 2634
                      "    -l | --log=FILE         output logging to file\n"
                      "    -v                      short version\n"
                      "    -V                      long version\n"
                      "         --version[=TYPE]   version, TYPE is short or long (default short)\n"
                      "    -e | --escape <char>    set escape sequence for console\n\n"
2635
                      "  commands (non interactive mode):\n\n"), progname, progname);
2636

2637
    for (grp = cmdGroups; grp->name; grp++) {
E
Eric Blake 已提交
2638 2639 2640 2641 2642
        fprintf(stdout, _(" %s (help keyword '%s')\n"),
                grp->name, grp->keyword);
        for (cmd = grp->commands; cmd->name; cmd++) {
            if (cmd->flags & VSH_CMD_FLAG_ALIAS)
                continue;
2643
            fprintf(stdout,
E
Eric Blake 已提交
2644 2645 2646
                    "    %-30s %s\n", cmd->name,
                    _(vshCmddefGetInfo(cmd, "help")));
        }
2647 2648 2649 2650 2651
        fprintf(stdout, "\n");
    }

    fprintf(stdout, "%s",
            _("\n  (specify help <group> for details about the commands in the group)\n"));
2652 2653 2654
    fprintf(stdout, "%s",
            _("\n  (specify help <command> for details about the command)\n\n"));
    return;
K
Karel Zak 已提交
2655 2656
}

2657 2658 2659 2660 2661 2662 2663 2664 2665 2666
/*
 * Show version and options compiled in
 */
static void
vshShowVersion(vshControl *ctl ATTRIBUTE_UNUSED)
{
    /* FIXME - list a copyright blurb, as in GNU programs?  */
    vshPrint(ctl, _("Virsh command line tool of libvirt %s\n"), VERSION);
    vshPrint(ctl, _("See web site at %s\n\n"), "http://libvirt.org/");

L
Laine Stump 已提交
2667 2668
    vshPrint(ctl, "%s", _("Compiled with support for:\n"));
    vshPrint(ctl, "%s", _(" Hypervisors:"));
2669
#ifdef WITH_QEMU
2670
    vshPrint(ctl, " QEMU/KVM");
2671
#endif
D
Doug Goldstein 已提交
2672 2673 2674
#ifdef WITH_LXC
    vshPrint(ctl, " LXC");
#endif
2675 2676 2677
#ifdef WITH_UML
    vshPrint(ctl, " UML");
#endif
D
Doug Goldstein 已提交
2678 2679 2680 2681 2682 2683
#ifdef WITH_XEN
    vshPrint(ctl, " Xen");
#endif
#ifdef WITH_LIBXL
    vshPrint(ctl, " LibXL");
#endif
2684 2685 2686
#ifdef WITH_OPENVZ
    vshPrint(ctl, " OpenVZ");
#endif
D
Doug Goldstein 已提交
2687 2688
#ifdef WITH_VMWARE
    vshPrint(ctl, " VMWare");
2689
#endif
D
Doug Goldstein 已提交
2690 2691
#ifdef WITH_PHYP
    vshPrint(ctl, " PHYP");
2692
#endif
D
Doug Goldstein 已提交
2693 2694
#ifdef WITH_VBOX
    vshPrint(ctl, " VirtualBox");
2695 2696 2697 2698
#endif
#ifdef WITH_ESX
    vshPrint(ctl, " ESX");
#endif
D
Doug Goldstein 已提交
2699 2700
#ifdef WITH_HYPERV
    vshPrint(ctl, " Hyper-V");
2701
#endif
D
Doug Goldstein 已提交
2702 2703
#ifdef WITH_XENAPI
    vshPrint(ctl, " XenAPI");
2704 2705 2706 2707 2708 2709
#endif
#ifdef WITH_TEST
    vshPrint(ctl, " Test");
#endif
    vshPrint(ctl, "\n");

L
Laine Stump 已提交
2710
    vshPrint(ctl, "%s", _(" Networking:"));
2711 2712 2713 2714 2715 2716 2717 2718 2719
#ifdef WITH_REMOTE
    vshPrint(ctl, " Remote");
#endif
#ifdef WITH_NETWORK
    vshPrint(ctl, " Network");
#endif
#ifdef WITH_BRIDGE
    vshPrint(ctl, " Bridging");
#endif
2720
#if defined(WITH_INTERFACE)
D
Doug Goldstein 已提交
2721
    vshPrint(ctl, " Interface");
2722 2723
# if defined(WITH_NETCF)
    vshPrint(ctl, " netcf");
2724 2725
# elif defined(HAVE_UDEV)
    vshPrint(ctl, " udev");
2726
# endif
2727 2728 2729 2730 2731 2732 2733 2734 2735
#endif
#ifdef WITH_NWFILTER
    vshPrint(ctl, " Nwfilter");
#endif
#ifdef WITH_VIRTUALPORT
    vshPrint(ctl, " VirtualPort");
#endif
    vshPrint(ctl, "\n");

L
Laine Stump 已提交
2736
    vshPrint(ctl, "%s", _(" Storage:"));
2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756
#ifdef WITH_STORAGE_DIR
    vshPrint(ctl, " Dir");
#endif
#ifdef WITH_STORAGE_DISK
    vshPrint(ctl, " Disk");
#endif
#ifdef WITH_STORAGE_FS
    vshPrint(ctl, " Filesystem");
#endif
#ifdef WITH_STORAGE_SCSI
    vshPrint(ctl, " SCSI");
#endif
#ifdef WITH_STORAGE_MPATH
    vshPrint(ctl, " Multipath");
#endif
#ifdef WITH_STORAGE_ISCSI
    vshPrint(ctl, " iSCSI");
#endif
#ifdef WITH_STORAGE_LVM
    vshPrint(ctl, " LVM");
2757 2758 2759
#endif
#ifdef WITH_STORAGE_RBD
    vshPrint(ctl, " RBD");
2760 2761 2762
#endif
#ifdef WITH_STORAGE_SHEEPDOG
    vshPrint(ctl, " Sheepdog");
2763 2764 2765
#endif
    vshPrint(ctl, "\n");

2766
    vshPrint(ctl, "%s", _(" Miscellaneous:"));
2767 2768 2769
#ifdef WITH_LIBVIRTD
    vshPrint(ctl, " Daemon");
#endif
2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805 2806 2807 2808 2809 2810
#ifdef WITH_NODE_DEVICES
    vshPrint(ctl, " Nodedev");
#endif
#ifdef WITH_SECDRIVER_APPARMOR
    vshPrint(ctl, " AppArmor");
#endif
#ifdef WITH_SECDRIVER_SELINUX
    vshPrint(ctl, " SELinux");
#endif
#ifdef WITH_SECRETS
    vshPrint(ctl, " Secrets");
#endif
#ifdef ENABLE_DEBUG
    vshPrint(ctl, " Debug");
#endif
#ifdef WITH_DTRACE_PROBES
    vshPrint(ctl, " DTrace");
#endif
#ifdef USE_READLINE
    vshPrint(ctl, " Readline");
#endif
#ifdef WITH_DRIVER_MODULES
    vshPrint(ctl, " Modular");
#endif
    vshPrint(ctl, "\n");
}

static bool
vshAllowedEscapeChar(char c)
{
    /* Allowed escape characters:
     * a-z A-Z @ [ \ ] ^ _
     */
    return ('a' <= c && c <= 'z') ||
        ('@' <= c && c <= '_');
}

/*
 * argv[]:  virsh [options] [command]
 *
 */
E
Eric Blake 已提交
2811
static bool
2812 2813
vshParseArgv(vshControl *ctl, int argc, char **argv)
{
2814
    int arg, len, debug;
2815 2816 2817 2818 2819 2820 2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833
    struct option opt[] = {
        {"debug", required_argument, NULL, 'd'},
        {"help", no_argument, NULL, 'h'},
        {"quiet", no_argument, NULL, 'q'},
        {"timing", no_argument, NULL, 't'},
        {"version", optional_argument, NULL, 'v'},
        {"connect", required_argument, NULL, 'c'},
        {"readonly", no_argument, NULL, 'r'},
        {"log", required_argument, NULL, 'l'},
        {"escape", required_argument, NULL, 'e'},
        {NULL, 0, NULL, 0}
    };

    /* Standard (non-command) options. The leading + ensures that no
     * argument reordering takes place, so that command options are
     * not confused with top-level virsh options. */
    while ((arg = getopt_long(argc, argv, "+d:hqtc:vVrl:e:", opt, NULL)) != -1) {
        switch (arg) {
        case 'd':
2834
            if (virStrToLong_i(optarg, NULL, 10, &debug) < 0) {
2835 2836 2837
                vshError(ctl, "%s", _("option -d takes a numeric argument"));
                exit(EXIT_FAILURE);
            }
2838 2839 2840 2841 2842
            if (debug < VSH_ERR_DEBUG || debug > VSH_ERR_ERROR)
                vshError(ctl, _("ignoring debug level %d out of range [%d-%d]"),
                         debug, VSH_ERR_DEBUG, VSH_ERR_ERROR);
            else
                ctl->debug = debug;
2843 2844 2845 2846 2847 2848 2849 2850 2851 2852 2853 2854 2855 2856 2857 2858 2859 2860 2861 2862 2863 2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910 2911 2912
            break;
        case 'h':
            vshUsage();
            exit(EXIT_SUCCESS);
            break;
        case 'q':
            ctl->quiet = true;
            break;
        case 't':
            ctl->timing = true;
            break;
        case 'c':
            ctl->name = vshStrdup(ctl, optarg);
            break;
        case 'v':
            if (STRNEQ_NULLABLE(optarg, "long")) {
                puts(VERSION);
                exit(EXIT_SUCCESS);
            }
            /* fall through */
        case 'V':
            vshShowVersion(ctl);
            exit(EXIT_SUCCESS);
        case 'r':
            ctl->readonly = true;
            break;
        case 'l':
            ctl->logfile = vshStrdup(ctl, optarg);
            break;
        case 'e':
            len = strlen(optarg);

            if ((len == 2 && *optarg == '^' &&
                 vshAllowedEscapeChar(optarg[1])) ||
                (len == 1 && *optarg != '^')) {
                ctl->escapeChar = optarg;
            } else {
                vshError(ctl, _("Invalid string '%s' for escape sequence"),
                         optarg);
                exit(EXIT_FAILURE);
            }
            break;
        default:
            vshError(ctl, _("unsupported option '-%c'. See --help."), arg);
            exit(EXIT_FAILURE);
        }
    }

    if (argc > optind) {
        /* parse command */
        ctl->imode = false;
        if (argc - optind == 1) {
            vshDebug(ctl, VSH_ERR_INFO, "commands: \"%s\"\n", argv[optind]);
            return vshCommandStringParse(ctl, argv[optind]);
        } else {
            return vshCommandArgvParse(ctl, argc - optind, argv + optind);
        }
    }
    return true;
}

static const vshCmdDef virshCmds[] = {
    {"cd", cmdCd, opts_cd, info_cd, VSH_CMD_FLAG_NOCONNECT},
    {"echo", cmdEcho, opts_echo, info_echo, VSH_CMD_FLAG_NOCONNECT},
    {"exit", cmdQuit, NULL, info_quit, VSH_CMD_FLAG_NOCONNECT},
    {"help", cmdHelp, opts_help, info_help, VSH_CMD_FLAG_NOCONNECT},
    {"pwd", cmdPwd, NULL, info_pwd, VSH_CMD_FLAG_NOCONNECT},
    {"quit", cmdQuit, NULL, info_quit, VSH_CMD_FLAG_NOCONNECT},
    {NULL, NULL, NULL, NULL, 0}
};
2913

2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928
static const vshCmdGrp cmdGroups[] = {
    {VSH_CMD_GRP_DOM_MANAGEMENT, "domain", domManagementCmds},
    {VSH_CMD_GRP_DOM_MONITORING, "monitor", domMonitoringCmds},
    {VSH_CMD_GRP_HOST_AND_HV, "host", hostAndHypervisorCmds},
    {VSH_CMD_GRP_IFACE, "interface", ifaceCmds},
    {VSH_CMD_GRP_NWFILTER, "filter", nwfilterCmds},
    {VSH_CMD_GRP_NETWORK, "network", networkCmds},
    {VSH_CMD_GRP_NODEDEV, "nodedev", nodedevCmds},
    {VSH_CMD_GRP_SECRET, "secret", secretCmds},
    {VSH_CMD_GRP_SNAPSHOT, "snapshot", snapshotCmds},
    {VSH_CMD_GRP_STORAGE_POOL, "pool", storagePoolCmds},
    {VSH_CMD_GRP_STORAGE_VOL, "volume", storageVolCmds},
    {VSH_CMD_GRP_VIRSH, "virsh", virshCmds},
    {NULL, NULL, NULL}
};
K
Karel Zak 已提交
2929

2930 2931 2932 2933
int
main(int argc, char **argv)
{
    vshControl _ctl, *ctl = &_ctl;
2934
    char *defaultConn;
E
Eric Blake 已提交
2935
    bool ret = true;
K
Karel Zak 已提交
2936

2937 2938 2939
    memset(ctl, 0, sizeof(vshControl));
    ctl->imode = true;          /* default is interactive mode */
    ctl->log_fd = -1;           /* Initialize log file descriptor */
J
Jiri Denemark 已提交
2940
    ctl->debug = VSH_DEBUG_DEFAULT;
E
Eric Blake 已提交
2941
    ctl->escapeChar = "^]";     /* Same default as telnet */
2942

2943

2944 2945
    if (!setlocale(LC_ALL, "")) {
        perror("setlocale");
2946
        /* failure to setup locale is not fatal */
2947
    }
2948
    if (!bindtextdomain(PACKAGE, LOCALEDIR)) {
2949
        perror("bindtextdomain");
E
Eric Blake 已提交
2950
        return EXIT_FAILURE;
2951
    }
2952
    if (!textdomain(PACKAGE)) {
2953
        perror("textdomain");
E
Eric Blake 已提交
2954
        return EXIT_FAILURE;
2955 2956
    }

2957 2958 2959 2960 2961
    if (virMutexInit(&ctl->lock) < 0) {
        vshError(ctl, "%s", _("Failed to initialize mutex"));
        return EXIT_FAILURE;
    }

2962 2963 2964 2965 2966
    if (virInitialize() < 0) {
        vshError(ctl, "%s", _("Failed to initialize libvirt"));
        return EXIT_FAILURE;
    }

2967
    if (!(progname = strrchr(argv[0], '/')))
K
Karel Zak 已提交
2968 2969 2970
        progname = argv[0];
    else
        progname++;
2971

2972
    if ((defaultConn = getenv("VIRSH_DEFAULT_CONNECT_URI"))) {
E
Eric Blake 已提交
2973
        ctl->name = vshStrdup(ctl, defaultConn);
2974 2975
    }

D
Daniel P. Berrange 已提交
2976 2977
    if (!vshParseArgv(ctl, argc, argv)) {
        vshDeinit(ctl);
K
Karel Zak 已提交
2978
        exit(EXIT_FAILURE);
D
Daniel P. Berrange 已提交
2979
    }
2980

D
Daniel P. Berrange 已提交
2981 2982
    if (!vshInit(ctl)) {
        vshDeinit(ctl);
K
Karel Zak 已提交
2983
        exit(EXIT_FAILURE);
D
Daniel P. Berrange 已提交
2984
    }
2985

K
Karel Zak 已提交
2986
    if (!ctl->imode) {
2987
        ret = vshCommandRun(ctl, ctl->cmd);
2988
    } else {
K
Karel Zak 已提交
2989 2990
        /* interactive mode */
        if (!ctl->quiet) {
K
Karel Zak 已提交
2991
            vshPrint(ctl,
2992
                     _("Welcome to %s, the virtualization interactive terminal.\n\n"),
2993
                     progname);
J
Jim Meyering 已提交
2994
            vshPrint(ctl, "%s",
2995
                     _("Type:  'help' for help with commands\n"
2996
                       "       'quit' to quit\n\n"));
K
Karel Zak 已提交
2997
        }
2998 2999 3000 3001 3002 3003

        if (vshReadlineInit(ctl) < 0) {
            vshDeinit(ctl);
            exit(EXIT_FAILURE);
        }

K
Karel Zak 已提交
3004
        do {
3005
            const char *prompt = ctl->readonly ? VSH_PROMPT_RO : VSH_PROMPT_RW;
3006
            ctl->cmdstr =
3007
                vshReadline(ctl, prompt);
3008 3009
            if (ctl->cmdstr == NULL)
                break;          /* EOF */
K
Karel Zak 已提交
3010
            if (*ctl->cmdstr) {
3011
#if USE_READLINE
K
Karel Zak 已提交
3012
                add_history(ctl->cmdstr);
3013
#endif
3014
                if (vshCommandStringParse(ctl, ctl->cmdstr))
K
Karel Zak 已提交
3015 3016
                    vshCommandRun(ctl, ctl->cmd);
            }
3017
            VIR_FREE(ctl->cmdstr);
3018
        } while (ctl->imode);
K
Karel Zak 已提交
3019

3020 3021
        if (ctl->cmdstr == NULL)
            fputc('\n', stdout);        /* line break after alone prompt */
K
Karel Zak 已提交
3022
    }
3023

K
Karel Zak 已提交
3024 3025
    vshDeinit(ctl);
    exit(ret ? EXIT_SUCCESS : EXIT_FAILURE);
3026
}