virsh.c 88.4 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 6 7 8
 *
 * See COPYING.LIB for the License of this software
 *
 * Daniel Veillard <veillard@redhat.com>
K
Karel Zak 已提交
9
 * Karel Zak <kzak@redhat.com>
K
Karel Zak 已提交
10
 * Daniel P. Berrange <berrange@redhat.com>
11 12
 */

13
#include <config.h>
14

15
#include <stdio.h>
K
Karel Zak 已提交
16 17 18
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
19
#include <unistd.h>
20
#include <errno.h>
K
Karel Zak 已提交
21
#include <getopt.h>
22
#include <sys/types.h>
K
Karel Zak 已提交
23
#include <sys/time.h>
E
Eric Blake 已提交
24
#include <sys/wait.h>
J
Jim Meyering 已提交
25
#include "c-ctype.h"
26
#include <fcntl.h>
27
#include <locale.h>
28
#include <time.h>
29
#include <limits.h>
30
#include <assert.h>
31
#include <sys/stat.h>
32
#include <inttypes.h>
33
#include <signal.h>
34
#include <poll.h>
E
Eric Blake 已提交
35
#include <strings.h>
36
#include <termios.h>
K
Karel Zak 已提交
37

38 39 40
#include <libxml/parser.h>
#include <libxml/tree.h>
#include <libxml/xpath.h>
41
#include <libxml/xmlsave.h>
42

43
#ifdef HAVE_READLINE_READLINE_H
44 45
# include <readline/readline.h>
# include <readline/history.h>
46
#endif
K
Karel Zak 已提交
47

48
#include "internal.h"
49
#include "virterror_internal.h"
50
#include "base64.h"
51
#include "buf.h"
52
#include "console.h"
53
#include "util.h"
54
#include "memory.h"
55
#include "xml.h"
56
#include "libvirt/libvirt-qemu.h"
E
Eric Blake 已提交
57
#include "virfile.h"
58
#include "event_poll.h"
59
#include "configmake.h"
60
#include "threads.h"
E
Eric Blake 已提交
61
#include "command.h"
62
#include "virkeycode.h"
63
#include "virnetdevbandwidth.h"
64
#include "util/bitmap.h"
H
Hu Tao 已提交
65
#include "conf/domain_conf.h"
66
#include "virtypedparam.h"
67
#include "conf/virdomainlist.h"
K
Karel Zak 已提交
68 69 70

static char *progname;

71 72
#define VIRSH_MAX_XML_FILE 10*1024*1024

K
Karel Zak 已提交
73 74 75
#define VSH_PROMPT_RW    "virsh # "
#define VSH_PROMPT_RO    "virsh > "

76 77
#define VIR_FROM_THIS VIR_FROM_NONE

K
Karel Zak 已提交
78 79 80 81 82
#define GETTIMEOFDAY(T) gettimeofday(T, NULL)
#define DIFF_MSEC(T, U) \
        ((((int) ((T)->tv_sec - (U)->tv_sec)) * 1000000.0 + \
          ((int) ((T)->tv_usec - (U)->tv_usec))) / 1000.0)

83 84 85
/* Default escape char Ctrl-] as per telnet */
#define CTRL_CLOSE_BRACKET "^]"

86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102
/**
 * The log configuration
 */
#define MSG_BUFFER    4096
#define SIGN_NAME     "virsh"
#define DIR_MODE      (S_IWUSR | S_IRUSR | S_IXUSR | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH)  /* 0755 */
#define FILE_MODE     (S_IWUSR | S_IRUSR | S_IRGRP | S_IROTH)                                /* 0644 */
#define LOCK_MODE     (S_IWUSR | S_IRUSR)                                                    /* 0600 */
#define LVL_DEBUG     "DEBUG"
#define LVL_INFO      "INFO"
#define LVL_NOTICE    "NOTICE"
#define LVL_WARNING   "WARNING"
#define LVL_ERROR     "ERROR"

/**
 * vshErrorLevel:
 *
J
Jim Meyering 已提交
103
 * Indicates the level of a log message
104 105 106 107 108 109 110 111 112
 */
typedef enum {
    VSH_ERR_DEBUG = 0,
    VSH_ERR_INFO,
    VSH_ERR_NOTICE,
    VSH_ERR_WARNING,
    VSH_ERR_ERROR
} vshErrorLevel;

J
Jiri Denemark 已提交
113 114
#define VSH_DEBUG_DEFAULT VSH_ERR_ERROR

K
Karel Zak 已提交
115 116 117 118 119
/*
 * virsh command line grammar:
 *
 *    command_line    =     <command>\n | <command>; <command>; ...
 *
E
Eric Blake 已提交
120
 *    command         =    <keyword> <option> [--] <data>
K
Karel Zak 已提交
121 122 123 124 125
 *
 *    option          =     <bool_option> | <int_option> | <string_option>
 *    data            =     <string>
 *
 *    bool_option     =     --optionname
E
Eric Blake 已提交
126 127
 *    int_option      =     --optionname <number> | --optionname=<number>
 *    string_option   =     --optionname <string> | --optionname=<string>
128
 *
E
Eric Blake 已提交
129
 *    keyword         =     [a-zA-Z][a-zA-Z-]*
130
 *    number          =     [0-9]+
E
Eric Blake 已提交
131
 *    string          =     ('[^']*'|"([^\\"]|\\.)*"|([^ \t\n\\'"]|\\.))+
K
Karel Zak 已提交
132 133 134 135
 *
 */

/*
136
 * vshCmdOptType - command option type
137
 */
K
Karel Zak 已提交
138
typedef enum {
139 140 141
    VSH_OT_BOOL,     /* optional boolean option */
    VSH_OT_STRING,   /* optional string option */
    VSH_OT_INT,      /* optional or mandatory int option */
142
    VSH_OT_DATA,     /* string data (as non-option) */
E
Eric Blake 已提交
143 144
    VSH_OT_ARGV,     /* remaining arguments */
    VSH_OT_ALIAS,    /* alternate spelling for a later argument */
K
Karel Zak 已提交
145 146
} vshCmdOptType;

147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162
/*
 * Command group types
 */
#define VSH_CMD_GRP_DOM_MANAGEMENT   "Domain Management"
#define VSH_CMD_GRP_DOM_MONITORING   "Domain Monitoring"
#define VSH_CMD_GRP_STORAGE_POOL     "Storage Pool"
#define VSH_CMD_GRP_STORAGE_VOL      "Storage Volume"
#define VSH_CMD_GRP_NETWORK          "Networking"
#define VSH_CMD_GRP_NODEDEV          "Node Device"
#define VSH_CMD_GRP_IFACE            "Interface"
#define VSH_CMD_GRP_NWFILTER         "Network Filter"
#define VSH_CMD_GRP_SECRET           "Secret"
#define VSH_CMD_GRP_SNAPSHOT         "Snapshot"
#define VSH_CMD_GRP_HOST_AND_HV      "Host and Hypervisor"
#define VSH_CMD_GRP_VIRSH            "Virsh itself"

K
Karel Zak 已提交
163 164 165
/*
 * Command Option Flags
 */
E
Eric Blake 已提交
166 167 168 169
enum {
    VSH_OFLAG_NONE     = 0,        /* without flags */
    VSH_OFLAG_REQ      = (1 << 0), /* option required */
    VSH_OFLAG_EMPTY_OK = (1 << 1), /* empty string option allowed */
L
Lai Jiangshan 已提交
170
    VSH_OFLAG_REQ_OPT  = (1 << 2), /* --optionname required */
E
Eric Blake 已提交
171
};
K
Karel Zak 已提交
172 173 174 175 176 177

/* dummy */
typedef struct __vshControl vshControl;
typedef struct __vshCmd vshCmd;

/*
E
Eric Blake 已提交
178 179 180 181 182
 * vshCmdInfo -- name/value pair for information about command
 *
 * Commands should have at least the following names:
 * "name" - command name
 * "desc" - description of command, or empty string
K
Karel Zak 已提交
183
 */
184
typedef struct {
E
Eric Blake 已提交
185 186
    const char *name;           /* name of information, or NULL for list end */
    const char *data;           /* non-NULL information */
K
Karel Zak 已提交
187 188 189 190 191
} vshCmdInfo;

/*
 * vshCmdOptDef - command option definition
 */
192
typedef struct {
E
Eric Blake 已提交
193
    const char *name;           /* the name of option, or NULL for list end */
194
    vshCmdOptType type;         /* option type */
E
Eric Blake 已提交
195
    unsigned int flags;         /* flags */
E
Eric Blake 已提交
196 197
    const char *help;           /* non-NULL help string; or for VSH_OT_ALIAS
                                 * the name of a later public option */
K
Karel Zak 已提交
198 199 200 201
} vshCmdOptDef;

/*
 * vshCmdOpt - command options
E
Eric Blake 已提交
202 203 204
 *
 * After parsing a command, all arguments to the command have been
 * collected into a list of these objects.
K
Karel Zak 已提交
205 206
 */
typedef struct vshCmdOpt {
E
Eric Blake 已提交
207 208
    const vshCmdOptDef *def;    /* non-NULL pointer to option definition */
    char *data;                 /* allocated data, or NULL for bool option */
209
    struct vshCmdOpt *next;
K
Karel Zak 已提交
210 211
} vshCmdOpt;

212 213 214 215 216
/*
 * Command Usage Flags
 */
enum {
    VSH_CMD_FLAG_NOCONNECT = (1 << 0),  /* no prior connection needed */
217
    VSH_CMD_FLAG_ALIAS     = (1 << 1),  /* command is an alias */
218 219
};

K
Karel Zak 已提交
220 221 222
/*
 * vshCmdDef - command definition
 */
223
typedef struct {
E
Eric Blake 已提交
224
    const char *name;           /* name of command, or NULL for list end */
E
Eric Blake 已提交
225
    bool (*handler) (vshControl *, const vshCmd *);    /* command handler */
226 227
    const vshCmdOptDef *opts;   /* definition of command options */
    const vshCmdInfo *info;     /* details about command */
228
    unsigned int flags;         /* bitwise OR of VSH_CMD_FLAG */
K
Karel Zak 已提交
229 230 231 232 233 234
} vshCmdDef;

/*
 * vshCmd - parsed command
 */
typedef struct __vshCmd {
235
    const vshCmdDef *def;       /* command definition */
236 237
    vshCmdOpt *opts;            /* list of command arguments */
    struct __vshCmd *next;      /* next command */
K
Karel Zak 已提交
238 239 240 241 242 243
} __vshCmd;

/*
 * vshControl
 */
typedef struct __vshControl {
K
Karel Zak 已提交
244
    char *name;                 /* connection name */
245
    virConnectPtr conn;         /* connection to hypervisor (MAY BE NULL) */
246 247
    vshCmd *cmd;                /* the current command */
    char *cmdstr;               /* string with command */
E
Eric Blake 已提交
248 249
    bool imode;                 /* interactive mode? */
    bool quiet;                 /* quiet mode */
250
    int debug;                  /* print debug messages? */
E
Eric Blake 已提交
251 252
    bool timing;                /* print timing info? */
    bool readonly;              /* connect readonly (first time only, not
253 254
                                 * during explicit connect command)
                                 */
255 256
    char *logfile;              /* log file name */
    int log_fd;                 /* log file descriptor */
257 258
    char *historydir;           /* readline history directory name */
    char *historyfile;          /* readline history file name */
259 260
    bool useGetInfo;            /* must use virDomainGetInfo, since
                                   virDomainGetState is not supported */
261 262
    bool useSnapshotOld;        /* cannot use virDomainSnapshotGetParent or
                                   virDomainSnapshotNumChildren */
J
Jiri Denemark 已提交
263
    virThread eventLoop;
264
    virMutex lock;
J
Jiri Denemark 已提交
265 266
    bool eventLoopStarted;
    bool quit;
267 268 269

    const char *escapeChar;     /* String representation of
                                   console escape character */
K
Karel Zak 已提交
270
} __vshControl;
271

272
typedef struct vshCmdGrp {
E
Eric Blake 已提交
273
    const char *name;    /* name of group, or NULL for list end */
274 275 276
    const char *keyword; /* help keyword */
    const vshCmdDef *commands;
} vshCmdGrp;
277

278
static const vshCmdGrp cmdGroups[];
K
Karel Zak 已提交
279

280 281
static void vshError(vshControl *ctl, const char *format, ...)
    ATTRIBUTE_FMT_PRINTF(2, 3);
E
Eric Blake 已提交
282 283
static bool vshInit(vshControl *ctl);
static bool vshDeinit(vshControl *ctl);
284
static void vshUsage(void);
285
static void vshOpenLogFile(vshControl *ctl);
286 287
static void vshOutputLogFile(vshControl *ctl, int log_level, const char *format, va_list ap)
    ATTRIBUTE_FMT_PRINTF(3, 0);
288
static void vshCloseLogFile(vshControl *ctl);
K
Karel Zak 已提交
289

E
Eric Blake 已提交
290
static bool vshParseArgv(vshControl *ctl, int argc, char **argv);
K
Karel Zak 已提交
291

292
static const char *vshCmddefGetInfo(const vshCmdDef *cmd, const char *info);
293
static const vshCmdDef *vshCmddefSearch(const char *cmdname);
E
Eric Blake 已提交
294
static bool vshCmddefHelp(vshControl *ctl, const char *name);
295
static const vshCmdGrp *vshCmdGrpSearch(const char *grpname);
E
Eric Blake 已提交
296
static bool vshCmdGrpHelp(vshControl *ctl, const char *name);
K
Karel Zak 已提交
297

E
Eric Blake 已提交
298 299 300
static int vshCommandOpt(const vshCmd *cmd, const char *name, vshCmdOpt **opt)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_NONNULL(2) ATTRIBUTE_NONNULL(3)
    ATTRIBUTE_RETURN_CHECK;
301 302
static int vshCommandOptInt(const vshCmd *cmd, const char *name, int *value)
    ATTRIBUTE_NONNULL(3) ATTRIBUTE_RETURN_CHECK;
303 304 305
static int vshCommandOptUInt(const vshCmd *cmd, const char *name,
                             unsigned int *value)
    ATTRIBUTE_NONNULL(3) ATTRIBUTE_RETURN_CHECK;
306 307 308 309 310 311 312 313 314
static int vshCommandOptUL(const vshCmd *cmd, const char *name,
                           unsigned long *value)
    ATTRIBUTE_NONNULL(3) ATTRIBUTE_RETURN_CHECK;
static int vshCommandOptString(const vshCmd *cmd, const char *name,
                               const char **value)
    ATTRIBUTE_NONNULL(3) ATTRIBUTE_RETURN_CHECK;
static int vshCommandOptLongLong(const vshCmd *cmd, const char *name,
                                 long long *value)
    ATTRIBUTE_NONNULL(3) ATTRIBUTE_RETURN_CHECK;
315 316 317
static int vshCommandOptULongLong(const vshCmd *cmd, const char *name,
                                  unsigned long long *value)
    ATTRIBUTE_NONNULL(3) ATTRIBUTE_RETURN_CHECK;
E
Eric Blake 已提交
318 319 320 321
static int vshCommandOptScaledInt(const vshCmd *cmd, const char *name,
                                  unsigned long long *value, int scale,
                                  unsigned long long max)
    ATTRIBUTE_NONNULL(3) ATTRIBUTE_RETURN_CHECK;
E
Eric Blake 已提交
322
static bool vshCommandOptBool(const vshCmd *cmd, const char *name);
323 324
static const vshCmdOpt *vshCommandOptArgv(const vshCmd *cmd,
                                          const vshCmdOpt *opt);
325 326 327
static char *vshGetDomainDescription(vshControl *ctl, virDomainPtr dom,
                                     bool title, unsigned int flags)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_NONNULL(2) ATTRIBUTE_RETURN_CHECK;
K
Karel Zak 已提交
328

329 330 331
#define VSH_BYID     (1 << 1)
#define VSH_BYUUID   (1 << 2)
#define VSH_BYNAME   (1 << 3)
332
#define VSH_BYMAC    (1 << 4)
K
Karel Zak 已提交
333

334
static virDomainPtr vshCommandOptDomainBy(vshControl *ctl, const vshCmd *cmd,
335
                                          const char **name, int flag);
K
Karel Zak 已提交
336 337

/* default is lookup by Id, Name and UUID */
J
Jim Meyering 已提交
338 339
#define vshCommandOptDomain(_ctl, _cmd, _name)                      \
    vshCommandOptDomainBy(_ctl, _cmd, _name, VSH_BYID|VSH_BYUUID|VSH_BYNAME)
340

341
static void vshPrintExtra(vshControl *ctl, const char *format, ...)
342
    ATTRIBUTE_FMT_PRINTF(2, 3);
343
static void vshDebug(vshControl *ctl, int level, const char *format, ...)
344
    ATTRIBUTE_FMT_PRINTF(3, 4);
K
Karel Zak 已提交
345 346

/* XXX: add batch support */
347
#define vshPrint(_ctl, ...)   vshPrintExtra(NULL, __VA_ARGS__)
K
Karel Zak 已提交
348

349
static int vshDomainState(vshControl *ctl, virDomainPtr dom, int *reason);
K
Karel Zak 已提交
350
static const char *vshDomainStateToString(int state);
351
static const char *vshDomainStateReasonToString(int state, int reason);
352
static const char *vshDomainControlStateToString(int state);
353
static const char *vshDomainVcpuStateToString(int state);
E
Eric Blake 已提交
354
static bool vshConnectionUsability(vshControl *ctl, virConnectPtr conn);
355 356 357
static virTypedParameterPtr vshFindTypedParamByName(const char *name,
                                                    virTypedParameterPtr list,
                                                    int count);
358 359
static char *vshGetTypedParamValue(vshControl *ctl, virTypedParameterPtr item)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_NONNULL(2);
K
Karel Zak 已提交
360

E
Eric Blake 已提交
361 362 363
static char *editWriteToTempFile(vshControl *ctl, const char *doc);
static int   editFile(vshControl *ctl, const char *filename);
static char *editReadBackFile(vshControl *ctl, const char *filename);
364

365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387
/* Typedefs, function prototypes for job progress reporting.
 * There are used by some long lingering commands like
 * migrate, dump, save, managedsave.
 */
typedef struct __vshCtrlData {
    vshControl *ctl;
    const vshCmd *cmd;
    int writefd;
} vshCtrlData;

typedef void (*jobWatchTimeoutFunc) (vshControl *ctl, virDomainPtr dom,
                                     void *opaque);

static bool
vshWatchJob(vshControl *ctl,
            virDomainPtr dom,
            bool verbose,
            int pipe_fd,
            int timeout,
            jobWatchTimeoutFunc timeout_func,
            void *opaque,
            const char *label);

388
static void *_vshMalloc(vshControl *ctl, size_t sz, const char *filename, int line);
389 390
#define vshMalloc(_ctl, _sz)    _vshMalloc(_ctl, _sz, __FILE__, __LINE__)

391
static void *_vshCalloc(vshControl *ctl, size_t nmemb, size_t sz, const char *filename, int line);
392 393
#define vshCalloc(_ctl, _nmemb, _sz)    _vshCalloc(_ctl, _nmemb, _sz, __FILE__, __LINE__)

394
static char *_vshStrdup(vshControl *ctl, const char *s, const char *filename, int line);
395 396
#define vshStrdup(_ctl, _s)    _vshStrdup(_ctl, _s, __FILE__, __LINE__)

397 398
static int parseRateStr(const char *rateStr, virNetDevBandwidthRatePtr rate);

E
Eric Blake 已提交
399 400 401
static void *
_vshMalloc(vshControl *ctl, size_t size, const char *filename, int line)
{
E
Eric Blake 已提交
402
    char *x;
E
Eric Blake 已提交
403

E
Eric Blake 已提交
404
    if (VIR_ALLOC_N(x, size) == 0)
E
Eric Blake 已提交
405 406 407 408 409 410 411 412 413
        return x;
    vshError(ctl, _("%s: %d: failed to allocate %d bytes"),
             filename, line, (int) size);
    exit(EXIT_FAILURE);
}

static void *
_vshCalloc(vshControl *ctl, size_t nmemb, size_t size, const char *filename, int line)
{
E
Eric Blake 已提交
414
    char *x;
E
Eric Blake 已提交
415

E
Eric Blake 已提交
416 417
    if (!xalloc_oversized(nmemb, size) &&
        VIR_ALLOC_N(x, nmemb * size) == 0)
E
Eric Blake 已提交
418 419 420 421 422 423 424 425 426 427 428 429
        return x;
    vshError(ctl, _("%s: %d: failed to allocate %d bytes"),
             filename, line, (int) (size*nmemb));
    exit(EXIT_FAILURE);
}

static char *
_vshStrdup(vshControl *ctl, const char *s, const char *filename, int line)
{
    char *x;

    if (s == NULL)
430
        return NULL;
E
Eric Blake 已提交
431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446
    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.  */
#undef malloc
#undef calloc
#undef realloc
#undef strdup
#define malloc use_vshMalloc_instead_of_malloc
#define calloc use_vshCalloc_instead_of_calloc
#define realloc use_vshRealloc_instead_of_realloc
#define strdup use_vshStrdup_instead_of_strdup
447

448 449 450 451 452
static int
vshNameSorter(const void *a, const void *b)
{
    const char **sa = (const char**)a;
    const char **sb = (const char**)b;
453

454 455
    /* User visible sort, so we want locale-specific case comparison. */
    return strcasecmp(*sa, *sb);
456 457
}

458 459 460 461 462 463 464
static double
prettyCapacity(unsigned long long val,
               const char **unit) {
    if (val < 1024) {
        *unit = "";
        return (double)val;
    } else if (val < (1024.0l * 1024.0l)) {
465
        *unit = "KiB";
466 467
        return (((double)val / 1024.0l));
    } else if (val < (1024.0l * 1024.0l * 1024.0l)) {
468
        *unit = "MiB";
469
        return (double)val / (1024.0l * 1024.0l);
470
    } else if (val < (1024.0l * 1024.0l * 1024.0l * 1024.0l)) {
471
        *unit = "GiB";
472
        return (double)val / (1024.0l * 1024.0l * 1024.0l);
473
    } else {
474
        *unit = "TiB";
475
        return (double)val / (1024.0l * 1024.0l * 1024.0l * 1024.0l);
476 477 478 479
    }
}


J
John Levon 已提交
480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504
static virErrorPtr last_error;

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

/*
 * 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.
 */
static void
virshReportError(vshControl *ctl)
{
505 506 507 508 509 510 511 512
    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)
513
            goto out;
514
    }
J
John Levon 已提交
515 516

    if (last_error->code == VIR_ERR_OK) {
517
        vshError(ctl, "%s", _("unknown error"));
J
John Levon 已提交
518 519 520
        goto out;
    }

521
    vshError(ctl, "%s", last_error->message);
J
John Levon 已提交
522 523 524 525 526 527

out:
    virFreeError(last_error);
    last_error = NULL;
}

528 529 530 531 532 533 534 535 536
static volatile sig_atomic_t intCaught = 0;

static void vshCatchInt(int sig ATTRIBUTE_UNUSED,
                        siginfo_t *siginfo ATTRIBUTE_UNUSED,
                        void *context ATTRIBUTE_UNUSED)
{
    intCaught = 1;
}

537 538 539 540 541
/*
 * Detection of disconnections and automatic reconnection support
 */
static int disconnected = 0; /* we may have been disconnected */

542 543 544 545 546
/* Gnulib doesn't guarantee SA_SIGINFO support.  */
#ifndef SA_SIGINFO
# define SA_SIGINFO 0
#endif

547 548 549 550 551 552
/*
 * vshCatchDisconnect:
 *
 * We get here when a SIGPIPE is being raised, we can't do much in the
 * handler, just save the fact it was raised
 */
553 554
static void vshCatchDisconnect(int sig, siginfo_t *siginfo,
                               void *context ATTRIBUTE_UNUSED) {
E
Eric Blake 已提交
555
    if (sig == SIGPIPE ||
556
        (SA_SIGINFO && siginfo->si_signo == SIGPIPE))
557 558 559 560 561 562 563 564 565
        disconnected++;
}

/*
 * vshSetupSignals:
 *
 * Catch SIGPIPE signals which may arise when disconnection
 * from libvirtd occurs
 */
L
Laine Stump 已提交
566
static void
567 568 569 570 571 572 573 574 575 576 577 578 579
vshSetupSignals(void) {
    struct sigaction sig_action;

    sig_action.sa_sigaction = vshCatchDisconnect;
    sig_action.sa_flags = SA_SIGINFO;
    sigemptyset(&sig_action.sa_mask);

    sigaction(SIGPIPE, &sig_action, NULL);
}

/*
 * vshReconnect:
 *
L
Laine Stump 已提交
580
 * Reconnect after a disconnect from libvirtd
581 582
 *
 */
L
Laine Stump 已提交
583
static void
584 585 586 587 588 589
vshReconnect(vshControl *ctl)
{
    bool connected = false;

    if (ctl->conn != NULL) {
        connected = true;
590
        virConnectClose(ctl->conn);
591
    }
592 593 594 595 596 597

    ctl->conn = virConnectOpenAuth(ctl->name,
                                   virConnectAuthPtrDefault,
                                   ctl->readonly ? VIR_CONNECT_RO : 0);
    if (!ctl->conn)
        vshError(ctl, "%s", _("Failed to reconnect to the hypervisor"));
598
    else if (connected)
599 600
        vshError(ctl, "%s", _("Reconnected to the hypervisor"));
    disconnected = 0;
601
    ctl->useGetInfo = false;
602
    ctl->useSnapshotOld = false;
603
}
604

605
#ifndef WIN32
606 607 608 609 610 611 612 613 614 615 616 617 618
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);
}

619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652
/**
 * 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
 */
static int
vshAskReedit(vshControl *ctl, const char *msg)
{
    int c = -1;
    struct termios ttyattr;

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

    virshReportError(ctl);

    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 == '?') {
653 654 655 656 657 658 659
            vshPrintRaw(ctl,
                        "",
                        _("y - yes, start editor again"),
                        _("n - no, throw away my changes"),
                        _("f - force, try to redefine again"),
                        _("? - print this help"),
                        NULL);
660 661 662 663 664 665 666 667 668 669
            continue;
        } else if (c == 'y' || c == 'n' || c == 'f') {
            break;
        }
    }

    tcsetattr(STDIN_FILENO, TCSAFLUSH, &ttyattr);

    vshPrint(ctl, "\r\n");
    return c;
670 671 672 673 674
}
#else /* WIN32 */
static int
vshAskReedit(vshControl *ctl, const char *msg ATTRIBUTE_UNUSED)
{
675 676 677 678
    vshDebug(ctl, VSH_ERR_WARNING, "%s", _("This function is not "
                                           "supported on WIN32 platform"));
    return 0;
}
679
#endif /* WIN32 */
680

681 682 683 684 685 686 687 688
static int vshStreamSink(virStreamPtr st ATTRIBUTE_UNUSED,
                         const char *bytes, size_t nbytes, void *opaque)
{
    int *fd = opaque;

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

K
Karel Zak 已提交
689 690 691 692 693 694
/* ---------------
 * Commands
 * ---------------
 */

/*
695
 * "help" command
K
Karel Zak 已提交
696
 */
697
static const vshCmdInfo info_help[] = {
698
    {"help", N_("print help")},
699 700
    {"desc", N_("Prints global help, command specific help, or help for a\n"
                "    group of related commands")},
701

702
    {NULL, NULL}
K
Karel Zak 已提交
703 704
};

705
static const vshCmdOptDef opts_help[] = {
706
    {"command", VSH_OT_DATA, 0, N_("Prints global help, command specific help, or help for a group of related commands")},
707
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
708 709
};

E
Eric Blake 已提交
710
static bool
711
cmdHelp(vshControl *ctl, const vshCmd *cmd)
712
 {
713
    const char *name = NULL;
714

715
    if (vshCommandOptString(cmd, "command", &name) <= 0) {
716
        const vshCmdGrp *grp;
717
        const vshCmdDef *def;
718

719 720 721 722 723 724
        vshPrint(ctl, "%s", _("Grouped commands:\n\n"));

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

725 726 727
            for (def = grp->commands; def->name; def++) {
                if (def->flags & VSH_CMD_FLAG_ALIAS)
                    continue;
728 729
                vshPrint(ctl, "    %-30s %s\n", def->name,
                         _(vshCmddefGetInfo(def, "help")));
730
            }
731 732 733 734

            vshPrint(ctl, "\n");
        }

E
Eric Blake 已提交
735
        return true;
736
    }
737

E
Eric Blake 已提交
738
    if (vshCmddefSearch(name)) {
739
        return vshCmddefHelp(ctl, name);
E
Eric Blake 已提交
740
    } else if (vshCmdGrpSearch(name)) {
741 742 743
        return vshCmdGrpHelp(ctl, name);
    } else {
        vshError(ctl, _("command or command group '%s' doesn't exist"), name);
E
Eric Blake 已提交
744
        return false;
K
Karel Zak 已提交
745 746 747
    }
}

748 749 750 751 752 753 754 755 756 757 758 759 760 761 762
/* Tree listing helpers.  */

/* Given an index, return either the name of that device (non-NULL) or
 * of its parent (NULL if a root).  */
typedef const char * (*vshTreeLookup)(int devid, bool parent, void *opaque);

static int
vshTreePrintInternal(vshControl *ctl,
                     vshTreeLookup lookup,
                     void *opaque,
                     int num_devices,
                     int devid,
                     int lastdev,
                     bool root,
                     virBufferPtr indent)
763
{
764 765 766 767
    int i;
    int nextlastdev = -1;
    int ret = -1;
    const char *dev = (lookup)(devid, false, opaque);
768

769
    if (virBufferError(indent))
770 771
        goto cleanup;

772 773 774 775 776 777 778 779 780 781
    /* 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;
782 783
    }

784 785 786
    /* Determine the index of the last child device */
    for (i = 0 ; i < num_devices ; i++) {
        const char *parent = (lookup)(i, true, opaque);
787

788 789 790
        if (parent && STREQ(parent, dev))
            nextlastdev = i;
    }
791

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

796 797 798 799
    /* Finally print all children */
    virBufferAddLit(indent, "  ");
    for (i = 0 ; i < num_devices ; i++) {
        const char *parent = (lookup)(i, true, opaque);
800

801 802 803 804 805
        if (parent && STREQ(parent, dev) &&
            vshTreePrintInternal(ctl, lookup, opaque,
                                 num_devices, i, nextlastdev,
                                 false, indent) < 0)
            goto cleanup;
806
    }
807
    virBufferTrim(indent, "  ", -1);
808

809 810 811 812
    /* 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));
813

814 815 816
    if (!root)
        virBufferTrim(indent, NULL, 2);
    ret = 0;
817 818 819 820
cleanup:
    return ret;
}

821 822 823
static int
vshTreePrint(vshControl *ctl, vshTreeLookup lookup, void *opaque,
             int num_devices, int devid)
824
{
825 826
    int ret;
    virBuffer indent = VIR_BUFFER_INITIALIZER;
827

828 829 830 831 832
    ret = vshTreePrintInternal(ctl, lookup, opaque, num_devices,
                               devid, devid, true, &indent);
    if (ret < 0)
        vshError(ctl, "%s", _("Failed to complete tree listing"));
    virBufferFreeAndReset(&indent);
833
    return ret;
834
}
835

836 837
/* Common code for the edit / net-edit / pool-edit functions which follow. */
static char *
838
editWriteToTempFile(vshControl *ctl, const char *doc)
839 840 841 842 843 844 845
{
    char *ret;
    const char *tmpdir;
    int fd;

    tmpdir = getenv ("TMPDIR");
    if (!tmpdir) tmpdir = "/tmp";
846 847 848 849
    if (virAsprintf(&ret, "%s/virshXXXXXX.xml", tmpdir) < 0) {
        vshError(ctl, "%s", _("out of memory"));
        return NULL;
    }
850
    fd = mkstemps(ret, 4);
851
    if (fd == -1) {
852
        vshError(ctl, _("mkstemps: failed to create temporary file: %s"),
853
                 strerror(errno));
854
        VIR_FREE(ret);
855 856 857
        return NULL;
    }

858
    if (safewrite(fd, doc, strlen(doc)) == -1) {
859 860
        vshError(ctl, _("write: %s: failed to write to temporary file: %s"),
                 ret, strerror(errno));
S
Stefan Berger 已提交
861
        VIR_FORCE_CLOSE(fd);
862
        unlink(ret);
863
        VIR_FREE(ret);
864 865
        return NULL;
    }
S
Stefan Berger 已提交
866
    if (VIR_CLOSE(fd) < 0) {
867 868
        vshError(ctl, _("close: %s: failed to write or close temporary file: %s"),
                 ret, strerror(errno));
869
        unlink(ret);
870
        VIR_FREE(ret);
871 872 873 874 875 876 877 878 879 880 881 882
        return NULL;
    }

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

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

static int
883
editFile(vshControl *ctl, const char *filename)
884 885
{
    const char *editor;
E
Eric Blake 已提交
886 887 888 889
    virCommandPtr cmd;
    int ret = -1;
    int outfd = STDOUT_FILENO;
    int errfd = STDERR_FILENO;
890

891
    editor = getenv("VISUAL");
E
Eric Blake 已提交
892
    if (!editor)
893
        editor = getenv("EDITOR");
E
Eric Blake 已提交
894 895
    if (!editor)
        editor = "vi"; /* could be cruel & default to ed(1) here */
896

897 898 899 900 901
    /* 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 已提交
902 903
     * is why sudo scrubs it by default).  Conversely, if the editor
     * is safe, we can run it directly rather than wasting a shell.
904
     */
905 906
    if (strspn(editor, ACCEPTED_CHARS) != strlen(editor)) {
        if (strspn(filename, ACCEPTED_CHARS) != strlen(filename)) {
E
Eric Blake 已提交
907 908 909 910 911 912 913 914 915 916
            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);
917 918
    }

E
Eric Blake 已提交
919 920 921 922 923 924 925
    virCommandSetInputFD(cmd, STDIN_FILENO);
    virCommandSetOutputFD(cmd, &outfd);
    virCommandSetErrorFD(cmd, &errfd);
    if (virCommandRunAsync(cmd, NULL) < 0 ||
        virCommandWait(cmd, NULL) < 0) {
        virshReportError(ctl);
        goto cleanup;
926
    }
E
Eric Blake 已提交
927
    ret = 0;
928

E
Eric Blake 已提交
929 930 931
cleanup:
    virCommandFree(cmd);
    return ret;
932 933 934
}

static char *
935
editReadBackFile(vshControl *ctl, const char *filename)
936 937 938
{
    char *ret;

E
Eric Blake 已提交
939
    if (virFileReadAll(filename, VIRSH_MAX_XML_FILE, &ret) == -1) {
940
        vshError(ctl,
941
                 _("%s: failed to read temporary file: %s"),
942
                 filename, strerror(errno));
943 944 945 946 947
        return NULL;
    }
    return ret;
}

948

P
Paolo Bonzini 已提交
949 950 951 952
/*
 * "cd" command
 */
static const vshCmdInfo info_cd[] = {
953 954
    {"help", N_("change the current directory")},
    {"desc", N_("Change the current directory.")},
P
Paolo Bonzini 已提交
955 956 957 958
    {NULL, NULL}
};

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

E
Eric Blake 已提交
963
static bool
964
cmdCd(vshControl *ctl, const vshCmd *cmd)
P
Paolo Bonzini 已提交
965
{
966
    const char *dir = NULL;
967
    char *dir_malloced = NULL;
E
Eric Blake 已提交
968
    bool ret = true;
P
Paolo Bonzini 已提交
969 970

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

975
    if (vshCommandOptString(cmd, "dir", &dir) <= 0) {
976
        dir = dir_malloced = virGetUserDirectory();
P
Paolo Bonzini 已提交
977 978 979 980
    }
    if (!dir)
        dir = "/";

P
Phil Petty 已提交
981
    if (chdir(dir) == -1) {
982
        vshError(ctl, _("cd: %s: %s"), strerror(errno), dir);
E
Eric Blake 已提交
983
        ret = false;
P
Paolo Bonzini 已提交
984 985
    }

986
    VIR_FREE(dir_malloced);
P
Phil Petty 已提交
987
    return ret;
P
Paolo Bonzini 已提交
988 989 990 991 992 993
}

/*
 * "pwd" command
 */
static const vshCmdInfo info_pwd[] = {
994 995
    {"help", N_("print the current directory")},
    {"desc", N_("Print the current directory.")},
P
Paolo Bonzini 已提交
996 997 998
    {NULL, NULL}
};

E
Eric Blake 已提交
999
static bool
P
Paolo Bonzini 已提交
1000 1001 1002
cmdPwd(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
{
    char *cwd;
1003
    bool ret = true;
P
Paolo Bonzini 已提交
1004

1005 1006
    cwd = getcwd(NULL, 0);
    if (!cwd) {
1007 1008
        vshError(ctl, _("pwd: cannot get current directory: %s"),
                 strerror(errno));
1009 1010
        ret = false;
    } else {
1011
        vshPrint(ctl, _("%s\n"), cwd);
1012 1013
        VIR_FREE(cwd);
    }
P
Paolo Bonzini 已提交
1014

1015
    return ret;
P
Paolo Bonzini 已提交
1016 1017
}

E
Eric Blake 已提交
1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029
/*
 * "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 已提交
1030
    {"str", VSH_OT_ALIAS, 0, "string"},
1031
    {"string", VSH_OT_ARGV, 0, N_("arguments to echo")},
E
Eric Blake 已提交
1032 1033 1034 1035 1036 1037
    {NULL, 0, 0, NULL}
};

/* Exists mainly for debugging virsh, but also handy for adding back
 * quotes for later evaluation.
 */
E
Eric Blake 已提交
1038
static bool
1039
cmdEcho(vshControl *ctl, const vshCmd *cmd)
E
Eric Blake 已提交
1040 1041 1042 1043
{
    bool shell = false;
    bool xml = false;
    int count = 0;
1044
    const vshCmdOpt *opt = NULL;
E
Eric Blake 已提交
1045 1046 1047 1048 1049 1050 1051 1052
    char *arg;
    virBuffer buf = VIR_BUFFER_INITIALIZER;

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

1053
    while ((opt = vshCommandOptArgv(cmd, opt))) {
1054 1055
        char *str;
        virBuffer xmlbuf = VIR_BUFFER_INITIALIZER;
E
Eric Blake 已提交
1056

1057
        arg = opt->data;
1058

E
Eric Blake 已提交
1059 1060
        if (count)
            virBufferAddChar(&buf, ' ');
1061

E
Eric Blake 已提交
1062
        if (xml) {
1063 1064 1065 1066
            virBufferEscapeString(&xmlbuf, "%s", arg);
            if (virBufferError(&buf)) {
                vshPrint(ctl, "%s", _("Failed to allocate XML buffer"));
                return false;
E
Eric Blake 已提交
1067
            }
1068 1069 1070
            str = virBufferContentAndReset(&xmlbuf);
        } else {
            str = vshStrdup(ctl, arg);
E
Eric Blake 已提交
1071
        }
1072 1073 1074 1075 1076

        if (shell)
            virBufferEscapeShell(&buf, str);
        else
            virBufferAdd(&buf, str, -1);
E
Eric Blake 已提交
1077
        count++;
1078
        VIR_FREE(str);
E
Eric Blake 已提交
1079 1080 1081 1082
    }

    if (virBufferError(&buf)) {
        vshPrint(ctl, "%s", _("Failed to allocate XML buffer"));
E
Eric Blake 已提交
1083
        return false;
E
Eric Blake 已提交
1084 1085 1086 1087 1088
    }
    arg = virBufferContentAndReset(&buf);
    if (arg)
        vshPrint(ctl, "%s", arg);
    VIR_FREE(arg);
E
Eric Blake 已提交
1089
    return true;
E
Eric Blake 已提交
1090 1091
}

K
Karel Zak 已提交
1092 1093 1094
/*
 * "quit" command
 */
1095
static const vshCmdInfo info_quit[] = {
1096
    {"help", N_("quit this interactive terminal")},
1097
    {"desc", ""},
1098
    {NULL, NULL}
K
Karel Zak 已提交
1099 1100
};

E
Eric Blake 已提交
1101
static bool
1102
cmdQuit(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
1103
{
E
Eric Blake 已提交
1104 1105
    ctl->imode = false;
    return true;
K
Karel Zak 已提交
1106 1107
}

1108 1109 1110 1111 1112 1113 1114 1115
/* ---------------
 * Utils for work with command definition
 * ---------------
 */
static const char *
vshCmddefGetInfo(const vshCmdDef * cmd, const char *name)
{
    const vshCmdInfo *info;
1116

1117 1118 1119 1120 1121 1122
    for (info = cmd->info; info && info->name; info++) {
        if (STREQ(info->name, name))
            return info->data;
    }
    return NULL;
}
1123

1124 1125 1126 1127 1128 1129 1130
/* 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;
1131

1132 1133
    *opts_need_arg = 0;
    *opts_required = 0;
1134

1135 1136
    if (!cmd->opts)
        return 0;
1137

1138 1139
    for (i = 0; cmd->opts[i].name; i++) {
        const vshCmdOptDef *opt = &cmd->opts[i];
1140 1141 1142 1143

        if (i > 31)
            return -1; /* too many options */
        if (opt->type == VSH_OT_BOOL) {
E
Eric Blake 已提交
1144
            if (opt->flags & VSH_OFLAG_REQ)
1145 1146 1147
                return -1; /* bool options can't be mandatory */
            continue;
        }
E
Eric Blake 已提交
1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159
        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 已提交
1160 1161
        if (opt->flags & VSH_OFLAG_REQ_OPT) {
            if (opt->flags & VSH_OFLAG_REQ)
L
Lai Jiangshan 已提交
1162 1163 1164 1165
                *opts_required |= 1 << i;
            continue;
        }

1166
        *opts_need_arg |= 1 << i;
E
Eric Blake 已提交
1167
        if (opt->flags & VSH_OFLAG_REQ) {
1168 1169 1170 1171 1172 1173
            if (optional)
                return -1; /* mandatory options must be listed first */
            *opts_required |= 1 << i;
        } else {
            optional = true;
        }
1174 1175 1176

        if (opt->type == VSH_OT_ARGV && cmd->opts[i + 1].name)
            return -1; /* argv option must be listed last */
1177 1178 1179 1180
    }
    return 0;
}

1181
static const vshCmdOptDef *
1182
vshCmddefGetOption(vshControl *ctl, const vshCmdDef *cmd, const char *name,
1183
                   uint32_t *opts_seen, int *opt_index)
1184
{
1185 1186 1187 1188
    int i;

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

1190
        if (STREQ(opt->name, name)) {
E
Eric Blake 已提交
1191 1192 1193 1194
            if (opt->type == VSH_OT_ALIAS) {
                name = opt->help;
                continue;
            }
1195
            if ((*opts_seen & (1 << i)) && opt->type != VSH_OT_ARGV) {
1196 1197 1198
                vshError(ctl, _("option --%s already seen"), name);
                return NULL;
            }
1199 1200
            *opts_seen |= 1 << i;
            *opt_index = i;
K
Karel Zak 已提交
1201
            return opt;
1202 1203 1204 1205 1206
        }
    }

    vshError(ctl, _("command '%s' doesn't support option --%s"),
             cmd->name, name);
K
Karel Zak 已提交
1207 1208 1209
    return NULL;
}

1210
static const vshCmdOptDef *
1211 1212
vshCmddefGetData(const vshCmdDef *cmd, uint32_t *opts_need_arg,
                 uint32_t *opts_seen)
1213
{
1214
    int i;
1215
    const vshCmdOptDef *opt;
K
Karel Zak 已提交
1216

1217 1218 1219 1220
    if (!*opts_need_arg)
        return NULL;

    /* Grab least-significant set bit */
E
Eric Blake 已提交
1221
    i = ffs(*opts_need_arg) - 1;
1222
    opt = &cmd->opts[i];
1223
    if (opt->type != VSH_OT_ARGV)
1224
        *opts_need_arg &= ~(1 << i);
1225
    *opts_seen |= 1 << i;
1226
    return opt;
K
Karel Zak 已提交
1227 1228
}

1229 1230 1231
/*
 * Checks for required options
 */
1232
static int
1233 1234
vshCommandCheckOpts(vshControl *ctl, const vshCmd *cmd, uint32_t opts_required,
                    uint32_t opts_seen)
1235
{
1236
    const vshCmdDef *def = cmd->def;
1237 1238 1239 1240 1241 1242 1243 1244 1245
    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];
1246

1247
            vshError(ctl,
1248
                     opt->type == VSH_OT_DATA || opt->type == VSH_OT_ARGV ?
1249 1250 1251
                     _("command '%s' requires <%s> option") :
                     _("command '%s' requires --%s option"),
                     def->name, opt->name);
1252 1253
        }
    }
1254
    return -1;
1255 1256
}

1257
static const vshCmdDef *
1258 1259
vshCmddefSearch(const char *cmdname)
{
1260
    const vshCmdGrp *g;
1261
    const vshCmdDef *c;
1262

1263 1264
    for (g = cmdGroups; g->name; g++) {
        for (c = g->commands; c->name; c++) {
1265
            if (STREQ(c->name, cmdname))
1266 1267 1268 1269
                return c;
        }
    }

K
Karel Zak 已提交
1270 1271 1272
    return NULL;
}

1273 1274 1275 1276 1277 1278
static const vshCmdGrp *
vshCmdGrpSearch(const char *grpname)
{
    const vshCmdGrp *g;

    for (g = cmdGroups; g->name; g++) {
1279
        if (STREQ(g->name, grpname) || STREQ(g->keyword, grpname))
1280 1281 1282 1283 1284 1285
            return g;
    }

    return NULL;
}

E
Eric Blake 已提交
1286
static bool
1287 1288 1289 1290 1291 1292 1293
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 已提交
1294
        return false;
1295 1296 1297 1298 1299 1300 1301 1302 1303 1304
    } 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 已提交
1305
    return true;
1306 1307
}

E
Eric Blake 已提交
1308
static bool
1309
vshCmddefHelp(vshControl *ctl, const char *cmdname)
1310
{
1311
    const vshCmdDef *def = vshCmddefSearch(cmdname);
1312

K
Karel Zak 已提交
1313
    if (!def) {
1314
        vshError(ctl, _("command '%s' doesn't exist"), cmdname);
E
Eric Blake 已提交
1315
        return false;
1316
    } else {
E
Eric Blake 已提交
1317 1318
        /* Don't translate desc if it is "".  */
        const char *desc = vshCmddefGetInfo(def, "desc");
E
Eric Blake 已提交
1319
        const char *help = _(vshCmddefGetInfo(def, "help"));
1320
        char buf[256];
1321 1322
        uint32_t opts_need_arg;
        uint32_t opts_required;
1323
        bool shortopt = false; /* true if 'arg' works instead of '--opt arg' */
1324 1325 1326 1327

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

1331
        fputs(_("  NAME\n"), stdout);
1332 1333
        fprintf(stdout, "    %s - %s\n", def->name, help);

1334 1335 1336 1337 1338
        fputs(_("\n  SYNOPSIS\n"), stdout);
        fprintf(stdout, "    %s", def->name);
        if (def->opts) {
            const vshCmdOptDef *opt;
            for (opt = def->opts; opt->name; opt++) {
1339
                const char *fmt = "%s";
1340 1341
                switch (opt->type) {
                case VSH_OT_BOOL:
1342
                    fmt = "[--%s]";
1343 1344
                    break;
                case VSH_OT_INT:
E
Eric Blake 已提交
1345
                    /* xgettext:c-format */
E
Eric Blake 已提交
1346
                    fmt = ((opt->flags & VSH_OFLAG_REQ) ? "<%s>"
1347
                           : _("[--%s <number>]"));
1348 1349
                    if (!(opt->flags & VSH_OFLAG_REQ_OPT))
                        shortopt = true;
1350 1351
                    break;
                case VSH_OT_STRING:
E
Eric Blake 已提交
1352 1353
                    /* xgettext:c-format */
                    fmt = _("[--%s <string>]");
1354 1355
                    if (!(opt->flags & VSH_OFLAG_REQ_OPT))
                        shortopt = true;
1356 1357
                    break;
                case VSH_OT_DATA:
E
Eric Blake 已提交
1358
                    fmt = ((opt->flags & VSH_OFLAG_REQ) ? "<%s>" : "[<%s>]");
1359 1360
                    if (!(opt->flags & VSH_OFLAG_REQ_OPT))
                        shortopt = true;
1361 1362 1363
                    break;
                case VSH_OT_ARGV:
                    /* xgettext:c-format */
1364 1365 1366 1367 1368 1369 1370 1371
                    if (shortopt) {
                        fmt = (opt->flags & VSH_OFLAG_REQ)
                            ? _("{[--%s] <string>}...")
                            : _("[[--%s] <string>]...");
                    } else {
                        fmt = (opt->flags & VSH_OFLAG_REQ) ? _("<%s>...")
                            : _("[<%s>]...");
                    }
1372
                    break;
E
Eric Blake 已提交
1373 1374 1375
                case VSH_OT_ALIAS:
                    /* aliases are intentionally undocumented */
                    continue;
1376
                default:
1377
                    assert(0);
1378
                }
1379
                fputc(' ', stdout);
E
Eric Blake 已提交
1380
                fprintf(stdout, fmt, opt->name);
1381
            }
K
Karel Zak 已提交
1382
        }
1383 1384 1385
        fputc('\n', stdout);

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

K
Karel Zak 已提交
1391
        if (def->opts) {
1392
            const vshCmdOptDef *opt;
1393
            fputs(_("\n  OPTIONS\n"), stdout);
1394
            for (opt = def->opts; opt->name; opt++) {
1395 1396
                switch (opt->type) {
                case VSH_OT_BOOL:
K
Karel Zak 已提交
1397
                    snprintf(buf, sizeof(buf), "--%s", opt->name);
1398 1399
                    break;
                case VSH_OT_INT:
1400
                    snprintf(buf, sizeof(buf),
E
Eric Blake 已提交
1401
                             (opt->flags & VSH_OFLAG_REQ) ? _("[--%s] <number>")
1402
                             : _("--%s <number>"), opt->name);
1403 1404
                    break;
                case VSH_OT_STRING:
1405
                    /* OT_STRING should never be VSH_OFLAG_REQ */
1406
                    snprintf(buf, sizeof(buf), _("--%s <string>"), opt->name);
1407 1408
                    break;
                case VSH_OT_DATA:
1409 1410
                    snprintf(buf, sizeof(buf), _("[--%s] <string>"),
                             opt->name);
1411 1412
                    break;
                case VSH_OT_ARGV:
1413 1414 1415
                    snprintf(buf, sizeof(buf),
                             shortopt ? _("[--%s] <string>") : _("<%s>"),
                             opt->name);
1416
                    break;
E
Eric Blake 已提交
1417 1418
                case VSH_OT_ALIAS:
                    continue;
1419 1420 1421
                default:
                    assert(0);
                }
1422

E
Eric Blake 已提交
1423
                fprintf(stdout, "    %-15s  %s\n", buf, _(opt->help));
1424
            }
K
Karel Zak 已提交
1425 1426 1427
        }
        fputc('\n', stdout);
    }
E
Eric Blake 已提交
1428
    return true;
K
Karel Zak 已提交
1429 1430 1431 1432 1433 1434
}

/* ---------------
 * Utils for work with runtime commands data
 * ---------------
 */
1435 1436 1437
static void
vshCommandOptFree(vshCmdOpt * arg)
{
K
Karel Zak 已提交
1438 1439
    vshCmdOpt *a = arg;

1440
    while (a) {
K
Karel Zak 已提交
1441
        vshCmdOpt *tmp = a;
1442

K
Karel Zak 已提交
1443 1444
        a = a->next;

1445 1446
        VIR_FREE(tmp->data);
        VIR_FREE(tmp);
K
Karel Zak 已提交
1447 1448 1449 1450
    }
}

static void
1451
vshCommandFree(vshCmd *cmd)
1452
{
K
Karel Zak 已提交
1453 1454
    vshCmd *c = cmd;

1455
    while (c) {
K
Karel Zak 已提交
1456
        vshCmd *tmp = c;
1457

K
Karel Zak 已提交
1458 1459 1460 1461
        c = c->next;

        if (tmp->opts)
            vshCommandOptFree(tmp->opts);
1462
        VIR_FREE(tmp);
K
Karel Zak 已提交
1463 1464 1465
    }
}

E
Eric Blake 已提交
1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476
/**
 * 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 已提交
1477
 */
E
Eric Blake 已提交
1478 1479
static int
vshCommandOpt(const vshCmd *cmd, const char *name, vshCmdOpt **opt)
1480
{
E
Eric Blake 已提交
1481 1482
    vshCmdOpt *candidate = cmd->opts;
    const vshCmdOptDef *valid = cmd->def->opts;
1483

E
Eric Blake 已提交
1484 1485 1486 1487 1488 1489 1490
    /* 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 已提交
1491
    }
E
Eric Blake 已提交
1492 1493 1494 1495 1496 1497 1498

    /* Option not present, see if command requires it.  */
    *opt = NULL;
    while (valid) {
        if (!valid->name)
            break;
        if (STREQ(name, valid->name))
E
Eric Blake 已提交
1499
            return (valid->flags & VSH_OFLAG_REQ) == 0 ? 0 : -1;
E
Eric Blake 已提交
1500 1501 1502 1503
        valid++;
    }
    /* If we got here, the name is unknown.  */
    return -2;
K
Karel Zak 已提交
1504 1505
}

E
Eric Blake 已提交
1506 1507
/**
 * vshCommandOptInt:
1508 1509 1510 1511 1512 1513 1514
 * @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 已提交
1515
 * 0 if option not found and not required (@value untouched)
1516
 * <0 in all other cases (@value untouched)
K
Karel Zak 已提交
1517 1518
 */
static int
1519
vshCommandOptInt(const vshCmd *cmd, const char *name, int *value)
1520
{
E
Eric Blake 已提交
1521 1522
    vshCmdOpt *arg;
    int ret;
1523

E
Eric Blake 已提交
1524 1525 1526 1527 1528 1529 1530
    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;
1531
    }
E
Eric Blake 已提交
1532

E
Eric Blake 已提交
1533 1534 1535
    if (virStrToLong_i(arg->data, NULL, 10, value) < 0)
        return -1;
    return 1;
K
Karel Zak 已提交
1536 1537
}

1538

E
Eric Blake 已提交
1539 1540 1541 1542 1543 1544
/**
 * vshCommandOptUInt:
 * @cmd command reference
 * @name option name
 * @value result
 *
1545 1546 1547 1548 1549 1550
 * Convert option to unsigned int
 * See vshCommandOptInt()
 */
static int
vshCommandOptUInt(const vshCmd *cmd, const char *name, unsigned int *value)
{
E
Eric Blake 已提交
1551 1552
    vshCmdOpt *arg;
    int ret;
1553

E
Eric Blake 已提交
1554 1555 1556 1557 1558 1559 1560
    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;
1561
    }
E
Eric Blake 已提交
1562

E
Eric Blake 已提交
1563 1564 1565
    if (virStrToLong_ui(arg->data, NULL, 10, value) < 0)
        return -1;
    return 1;
1566 1567 1568
}


1569
/*
E
Eric Blake 已提交
1570 1571 1572 1573 1574
 * vshCommandOptUL:
 * @cmd command reference
 * @name option name
 * @value result
 *
1575 1576 1577 1578 1579
 * Convert option to unsigned long
 * See vshCommandOptInt()
 */
static int
vshCommandOptUL(const vshCmd *cmd, const char *name, unsigned long *value)
1580
{
E
Eric Blake 已提交
1581 1582
    vshCmdOpt *arg;
    int ret;
1583

E
Eric Blake 已提交
1584 1585 1586 1587 1588 1589 1590
    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;
1591
    }
E
Eric Blake 已提交
1592

E
Eric Blake 已提交
1593 1594 1595
    if (virStrToLong_ul(arg->data, NULL, 10, value) < 0)
        return -1;
    return 1;
1596 1597
}

E
Eric Blake 已提交
1598 1599 1600 1601 1602 1603
/**
 * vshCommandOptString:
 * @cmd command reference
 * @name option name
 * @value result
 *
K
Karel Zak 已提交
1604
 * Returns option as STRING
E
Eric Blake 已提交
1605 1606 1607 1608
 * 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 已提交
1609
 */
1610 1611
static int
vshCommandOptString(const vshCmd *cmd, const char *name, const char **value)
1612
{
E
Eric Blake 已提交
1613 1614 1615 1616 1617 1618 1619 1620 1621 1622
    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;
1623
    }
1624

E
Eric Blake 已提交
1625
    if (!*arg->data && !(arg->def->flags & VSH_OFLAG_EMPTY_OK)) {
E
Eric Blake 已提交
1626 1627 1628 1629
        return -1;
    }
    *value = arg->data;
    return 1;
K
Karel Zak 已提交
1630 1631
}

E
Eric Blake 已提交
1632 1633 1634 1635 1636 1637
/**
 * vshCommandOptLongLong:
 * @cmd command reference
 * @name option name
 * @value result
 *
1638
 * Returns option as long long
1639
 * See vshCommandOptInt()
1640
 */
1641 1642 1643
static int
vshCommandOptLongLong(const vshCmd *cmd, const char *name,
                      long long *value)
1644
{
E
Eric Blake 已提交
1645 1646
    vshCmdOpt *arg;
    int ret;
1647

E
Eric Blake 已提交
1648 1649 1650 1651 1652 1653 1654
    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;
1655
    }
E
Eric Blake 已提交
1656

E
Eric Blake 已提交
1657 1658 1659
    if (virStrToLong_ll(arg->data, NULL, 10, value) < 0)
        return -1;
    return 1;
1660 1661
}

E
Eric Blake 已提交
1662 1663 1664 1665 1666 1667 1668 1669 1670
/**
 * vshCommandOptULongLong:
 * @cmd command reference
 * @name option name
 * @value result
 *
 * Returns option as long long
 * See vshCommandOptInt()
 */
1671 1672 1673 1674
static int
vshCommandOptULongLong(const vshCmd *cmd, const char *name,
                       unsigned long long *value)
{
E
Eric Blake 已提交
1675 1676
    vshCmdOpt *arg;
    int ret;
1677

E
Eric Blake 已提交
1678 1679 1680 1681 1682 1683 1684
    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;
1685
    }
E
Eric Blake 已提交
1686

E
Eric Blake 已提交
1687 1688 1689
    if (virStrToLong_ull(arg->data, NULL, 10, value) < 0)
        return -1;
    return 1;
1690 1691 1692
}


E
Eric Blake 已提交
1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722
/**
 * 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()
 */
static int
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 已提交
1723 1724 1725 1726 1727 1728 1729 1730 1731
/**
 * 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 已提交
1732
 */
E
Eric Blake 已提交
1733
static bool
1734
vshCommandOptBool(const vshCmd *cmd, const char *name)
1735
{
E
Eric Blake 已提交
1736 1737 1738
    vshCmdOpt *dummy;

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

E
Eric Blake 已提交
1741 1742 1743 1744 1745
/**
 * vshCommandOptArgv:
 * @cmd command reference
 * @opt starting point for the search
 *
1746 1747
 * Returns the next argv argument after OPT (or the first one if OPT
 * is NULL), or NULL if no more are present.
1748
 *
1749
 * Requires that a VSH_OT_ARGV option be last in the
1750 1751
 * list of supported options in CMD->def->opts.
 */
1752 1753
static const vshCmdOpt *
vshCommandOptArgv(const vshCmd *cmd, const vshCmdOpt *opt)
1754
{
1755
    opt = opt ? opt->next : cmd->opts;
1756 1757

    while (opt) {
E
Eric Blake 已提交
1758
        if (opt->def->type == VSH_OT_ARGV) {
1759
            return opt;
1760 1761 1762 1763 1764 1765
        }
        opt = opt->next;
    }
    return NULL;
}

J
Jim Meyering 已提交
1766 1767 1768 1769
/* Determine whether CMD->opts includes an option with name OPTNAME.
   If not, give a diagnostic and return false.
   If so, return true.  */
static bool
1770
cmd_has_option(vshControl *ctl, const vshCmd *cmd, const char *optname)
J
Jim Meyering 已提交
1771 1772 1773 1774 1775 1776
{
    /* 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) {
1777
        if (STREQ(opt->def->name, optname) && opt->def->type == VSH_OT_DATA) {
J
Jim Meyering 已提交
1778 1779 1780 1781 1782 1783
            found = true;
            break;
        }
    }

    if (!found)
1784
        vshError(ctl, _("internal error: virsh %s: no %s VSH_OT_DATA option"),
J
Jim Meyering 已提交
1785 1786 1787
                 cmd->def->name, optname);
    return found;
}
1788

K
Karel Zak 已提交
1789
static virDomainPtr
J
Jim Meyering 已提交
1790
vshCommandOptDomainBy(vshControl *ctl, const vshCmd *cmd,
1791
                      const char **name, int flag)
1792
{
K
Karel Zak 已提交
1793
    virDomainPtr dom = NULL;
1794
    const char *n = NULL;
K
Karel Zak 已提交
1795
    int id;
J
Jim Meyering 已提交
1796
    const char *optname = "domain";
1797
    if (!cmd_has_option(ctl, cmd, optname))
J
Jim Meyering 已提交
1798
        return NULL;
1799

1800
    if (vshCommandOptString(cmd, optname, &n) <= 0)
1801 1802
        return NULL;

1803
    vshDebug(ctl, VSH_ERR_INFO, "%s: found option <%s>: %s\n",
1804 1805
             cmd->def->name, optname, n);

K
Karel Zak 已提交
1806 1807
    if (name)
        *name = n;
1808

K
Karel Zak 已提交
1809
    /* try it by ID */
1810
    if (flag & VSH_BYID) {
1811
        if (virStrToLong_i(n, NULL, 10, &id) == 0 && id >= 0) {
1812 1813
            vshDebug(ctl, VSH_ERR_DEBUG,
                     "%s: <%s> seems like domain ID\n",
K
Karel Zak 已提交
1814 1815 1816
                     cmd->def->name, optname);
            dom = virDomainLookupByID(ctl->conn, id);
        }
1817
    }
K
Karel Zak 已提交
1818
    /* try it by UUID */
1819
    if (dom==NULL && (flag & VSH_BYUUID) && strlen(n)==VIR_UUID_STRING_BUFLEN-1) {
1820
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as domain UUID\n",
1821
                 cmd->def->name, optname);
K
Karel Zak 已提交
1822
        dom = virDomainLookupByUUIDString(ctl->conn, n);
K
Karel Zak 已提交
1823
    }
K
Karel Zak 已提交
1824
    /* try it by NAME */
1825
    if (dom==NULL && (flag & VSH_BYNAME)) {
1826
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as domain NAME\n",
1827
                 cmd->def->name, optname);
K
Karel Zak 已提交
1828
        dom = virDomainLookupByName(ctl->conn, n);
1829
    }
K
Karel Zak 已提交
1830

1831
    if (!dom)
1832
        vshError(ctl, _("failed to get domain '%s'"), n);
1833

K
Karel Zak 已提交
1834 1835 1836
    return dom;
}

K
Karel Zak 已提交
1837 1838 1839
/*
 * Executes command(s) and returns return code from last command
 */
E
Eric Blake 已提交
1840
static bool
1841
vshCommandRun(vshControl *ctl, const vshCmd *cmd)
1842
{
E
Eric Blake 已提交
1843
    bool ret = true;
1844 1845

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

1849 1850
        if ((ctl->conn == NULL || disconnected) &&
            !(cmd->def->flags & VSH_CMD_FLAG_NOCONNECT))
1851 1852
            vshReconnect(ctl);

1853
        if (enable_timing)
K
Karel Zak 已提交
1854
            GETTIMEOFDAY(&before);
1855

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

1858
        if (enable_timing)
K
Karel Zak 已提交
1859
            GETTIMEOFDAY(&after);
1860

1861 1862 1863 1864 1865 1866 1867 1868 1869 1870
        /* 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++;

1871
        if (!ret)
J
John Levon 已提交
1872 1873
            virshReportError(ctl);

1874
        if (!ret && disconnected != 0)
1875 1876
            vshReconnect(ctl);

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

1880
        if (enable_timing)
1881
            vshPrint(ctl, _("\n(Time: %.3f ms)\n\n"),
1882 1883
                     DIFF_MSEC(&after, &before));
        else
K
Karel Zak 已提交
1884
            vshPrintExtra(ctl, "\n");
K
Karel Zak 已提交
1885 1886 1887 1888 1889 1890
        cmd = cmd->next;
    }
    return ret;
}

/* ---------------
1891
 * Command parsing
K
Karel Zak 已提交
1892 1893 1894
 * ---------------
 */

1895 1896 1897 1898 1899 1900 1901 1902
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;

typedef struct __vshCommandParser {
1903
    vshCommandToken(*getNextArg)(vshControl *, struct __vshCommandParser *,
1904
                                  char **);
L
Lai Jiangshan 已提交
1905
    /* vshCommandStringGetArg() */
1906
    char *pos;
L
Lai Jiangshan 已提交
1907 1908 1909
    /* vshCommandArgvGetArg() */
    char **arg_pos;
    char **arg_end;
1910 1911
} vshCommandParser;

E
Eric Blake 已提交
1912
static bool
1913
vshCommandParse(vshControl *ctl, vshCommandParser *parser)
1914
{
K
Karel Zak 已提交
1915 1916 1917
    char *tkdata = NULL;
    vshCmd *clast = NULL;
    vshCmdOpt *first = NULL;
1918

K
Karel Zak 已提交
1919 1920 1921 1922
    if (ctl->cmd) {
        vshCommandFree(ctl->cmd);
        ctl->cmd = NULL;
    }
1923

1924
    while (1) {
K
Karel Zak 已提交
1925
        vshCmdOpt *last = NULL;
1926
        const vshCmdDef *cmd = NULL;
1927
        vshCommandToken tk;
L
Lai Jiangshan 已提交
1928
        bool data_only = false;
1929 1930 1931
        uint32_t opts_need_arg = 0;
        uint32_t opts_required = 0;
        uint32_t opts_seen = 0;
1932

K
Karel Zak 已提交
1933
        first = NULL;
1934

1935
        while (1) {
1936
            const vshCmdOptDef *opt = NULL;
1937

K
Karel Zak 已提交
1938
            tkdata = NULL;
1939
            tk = parser->getNextArg(ctl, parser, &tkdata);
1940 1941

            if (tk == VSH_TK_ERROR)
K
Karel Zak 已提交
1942
                goto syntaxError;
H
Hu Tao 已提交
1943 1944
            if (tk != VSH_TK_ARG) {
                VIR_FREE(tkdata);
1945
                break;
H
Hu Tao 已提交
1946
            }
1947 1948

            if (cmd == NULL) {
K
Karel Zak 已提交
1949 1950
                /* first token must be command name */
                if (!(cmd = vshCmddefSearch(tkdata))) {
1951
                    vshError(ctl, _("unknown command: '%s'"), tkdata);
1952
                    goto syntaxError;   /* ... or ignore this command only? */
K
Karel Zak 已提交
1953
                }
1954 1955 1956 1957 1958 1959 1960
                if (vshCmddefOptParse(cmd, &opts_need_arg,
                                      &opts_required) < 0) {
                    vshError(ctl,
                             _("internal error: bad options in command: '%s'"),
                             tkdata);
                    goto syntaxError;
                }
1961
                VIR_FREE(tkdata);
L
Lai Jiangshan 已提交
1962 1963 1964 1965
            } else if (data_only) {
                goto get_data;
            } else if (tkdata[0] == '-' && tkdata[1] == '-' &&
                       c_isalnum(tkdata[2])) {
1966
                char *optstr = strchr(tkdata + 2, '=');
1967 1968
                int opt_index;

1969 1970 1971 1972
                if (optstr) {
                    *optstr = '\0'; /* convert the '=' to '\0' */
                    optstr = vshStrdup(ctl, optstr + 1);
                }
1973
                if (!(opt = vshCmddefGetOption(ctl, cmd, tkdata + 2,
1974
                                               &opts_seen, &opt_index))) {
1975
                    VIR_FREE(optstr);
K
Karel Zak 已提交
1976 1977
                    goto syntaxError;
                }
1978
                VIR_FREE(tkdata);
K
Karel Zak 已提交
1979 1980 1981

                if (opt->type != VSH_OT_BOOL) {
                    /* option data */
1982 1983 1984
                    if (optstr)
                        tkdata = optstr;
                    else
1985
                        tk = parser->getNextArg(ctl, parser, &tkdata);
1986
                    if (tk == VSH_TK_ERROR)
K
Karel Zak 已提交
1987
                        goto syntaxError;
1988
                    if (tk != VSH_TK_ARG) {
1989
                        vshError(ctl,
1990
                                 _("expected syntax: --%s <%s>"),
1991 1992
                                 opt->name,
                                 opt->type ==
1993
                                 VSH_OT_INT ? _("number") : _("string"));
K
Karel Zak 已提交
1994 1995
                        goto syntaxError;
                    }
1996 1997
                    if (opt->type != VSH_OT_ARGV)
                        opts_need_arg &= ~(1 << opt_index);
1998 1999 2000 2001 2002 2003 2004 2005
                } else {
                    tkdata = NULL;
                    if (optstr) {
                        vshError(ctl, _("invalid '=' after option --%s"),
                                opt->name);
                        VIR_FREE(optstr);
                        goto syntaxError;
                    }
K
Karel Zak 已提交
2006
                }
L
Lai Jiangshan 已提交
2007 2008 2009 2010
            } else if (tkdata[0] == '-' && tkdata[1] == '-' &&
                       tkdata[2] == '\0') {
                data_only = true;
                continue;
2011
            } else {
L
Lai Jiangshan 已提交
2012
get_data:
2013 2014
                if (!(opt = vshCmddefGetData(cmd, &opts_need_arg,
                                             &opts_seen))) {
2015
                    vshError(ctl, _("unexpected data '%s'"), tkdata);
K
Karel Zak 已提交
2016 2017 2018 2019 2020
                    goto syntaxError;
                }
            }
            if (opt) {
                /* save option */
2021
                vshCmdOpt *arg = vshMalloc(ctl, sizeof(vshCmdOpt));
2022

K
Karel Zak 已提交
2023 2024 2025 2026
                arg->def = opt;
                arg->data = tkdata;
                arg->next = NULL;
                tkdata = NULL;
2027

K
Karel Zak 已提交
2028 2029 2030 2031 2032
                if (!first)
                    first = arg;
                if (last)
                    last->next = arg;
                last = arg;
2033

2034
                vshDebug(ctl, VSH_ERR_INFO, "%s: %s(%s): %s\n",
2035 2036
                         cmd->name,
                         opt->name,
2037 2038
                         opt->type != VSH_OT_BOOL ? _("optdata") : _("bool"),
                         opt->type != VSH_OT_BOOL ? arg->data : _("(none)"));
K
Karel Zak 已提交
2039 2040
            }
        }
2041

D
Daniel Veillard 已提交
2042
        /* command parsed -- allocate new struct for the command */
K
Karel Zak 已提交
2043
        if (cmd) {
2044
            vshCmd *c = vshMalloc(ctl, sizeof(vshCmd));
2045

K
Karel Zak 已提交
2046 2047 2048 2049
            c->opts = first;
            c->def = cmd;
            c->next = NULL;

2050
            if (vshCommandCheckOpts(ctl, c, opts_required, opts_seen) < 0) {
2051
                VIR_FREE(c);
2052
                goto syntaxError;
2053
            }
2054

K
Karel Zak 已提交
2055 2056 2057 2058 2059 2060
            if (!ctl->cmd)
                ctl->cmd = c;
            if (clast)
                clast->next = c;
            clast = c;
        }
2061 2062 2063

        if (tk == VSH_TK_END)
            break;
K
Karel Zak 已提交
2064
    }
2065

E
Eric Blake 已提交
2066
    return true;
K
Karel Zak 已提交
2067

2068
 syntaxError:
2069
    if (ctl->cmd) {
K
Karel Zak 已提交
2070
        vshCommandFree(ctl->cmd);
2071 2072
        ctl->cmd = NULL;
    }
K
Karel Zak 已提交
2073 2074
    if (first)
        vshCommandOptFree(first);
2075
    VIR_FREE(tkdata);
E
Eric Blake 已提交
2076
    return false;
K
Karel Zak 已提交
2077 2078
}

2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096
/* --------------------
 * 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 已提交
2097 2098
static bool
vshCommandArgvParse(vshControl *ctl, int nargs, char **argv)
2099 2100 2101 2102
{
    vshCommandParser parser;

    if (nargs <= 0)
E
Eric Blake 已提交
2103
        return false;
2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175

    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 已提交
2176 2177
static bool
vshCommandStringParse(vshControl *ctl, char *cmdstr)
2178 2179 2180 2181
{
    vshCommandParser parser;

    if (cmdstr == NULL || *cmdstr == '\0')
E
Eric Blake 已提交
2182
        return false;
2183 2184 2185 2186 2187 2188

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

K
Karel Zak 已提交
2189
/* ---------------
2190
 * Misc utils
K
Karel Zak 已提交
2191 2192
 * ---------------
 */
2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220
static int
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;
}

2221 2222
/* Return a non-NULL string representation of a typed parameter; exit
 * if we are out of memory.  */
2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253
static char *
vshGetTypedParamValue(vshControl *ctl, virTypedParameterPtr item)
{
    int ret = 0;
    char *str = NULL;

    switch(item->type) {
    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;

2254 2255 2256 2257
    case VIR_TYPED_PARAM_STRING:
        str = vshStrdup(ctl, item->value.s);
        break;

2258
    default:
2259
        vshError(ctl, _("unimplemented parameter type %d"), item->type);
2260 2261
    }

2262
    if (ret < 0) {
2263
        vshError(ctl, "%s", _("Out of memory"));
2264 2265
        exit(EXIT_FAILURE);
    }
2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288
    return str;
}

static virTypedParameterPtr
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 已提交
2289
static bool
2290
vshConnectionUsability(vshControl *ctl, virConnectPtr conn)
2291
{
2292 2293
    /* TODO: use something like virConnectionState() to
     *       check usability of the connection
K
Karel Zak 已提交
2294 2295
     */
    if (!conn) {
2296
        vshError(ctl, "%s", _("no valid connection"));
E
Eric Blake 已提交
2297
        return false;
K
Karel Zak 已提交
2298
    }
E
Eric Blake 已提交
2299
    return true;
K
Karel Zak 已提交
2300 2301
}

K
Karel Zak 已提交
2302
static void
2303
vshDebug(vshControl *ctl, int level, const char *format, ...)
2304
{
K
Karel Zak 已提交
2305
    va_list ap;
2306
    char *str;
K
Karel Zak 已提交
2307

2308 2309 2310 2311 2312 2313 2314
    /* Aligning log levels to that of libvirt.
     * Traces with levels >=  user-specified-level
     * gets logged into file
     */
    if (level < ctl->debug)
        return;

2315
    va_start(ap, format);
2316
    vshOutputLogFile(ctl, level, format, ap);
2317 2318
    va_end(ap);

K
Karel Zak 已提交
2319
    va_start(ap, format);
2320 2321 2322 2323 2324
    if (virVasprintf(&str, format, ap) < 0) {
        /* Skip debug messages on low memory */
        va_end(ap);
        return;
    }
K
Karel Zak 已提交
2325
    va_end(ap);
2326 2327
    fputs(str, stdout);
    VIR_FREE(str);
K
Karel Zak 已提交
2328 2329 2330
}

static void
2331
vshPrintExtra(vshControl *ctl, const char *format, ...)
2332
{
K
Karel Zak 已提交
2333
    va_list ap;
2334
    char *str;
2335

2336
    if (ctl && ctl->quiet)
K
Karel Zak 已提交
2337
        return;
2338

K
Karel Zak 已提交
2339
    va_start(ap, format);
2340 2341 2342 2343 2344
    if (virVasprintf(&str, format, ap) < 0) {
        vshError(ctl, "%s", _("Out of memory"));
        va_end(ap);
        return;
    }
K
Karel Zak 已提交
2345
    va_end(ap);
2346
    fputs(str, stdout);
2347
    VIR_FREE(str);
K
Karel Zak 已提交
2348 2349
}

K
Karel Zak 已提交
2350

K
Karel Zak 已提交
2351
static void
2352
vshError(vshControl *ctl, const char *format, ...)
2353
{
K
Karel Zak 已提交
2354
    va_list ap;
2355
    char *str;
2356

2357 2358 2359 2360 2361
    if (ctl != NULL) {
        va_start(ap, format);
        vshOutputLogFile(ctl, VSH_ERR_ERROR, format, ap);
        va_end(ap);
    }
2362

2363 2364 2365 2366
    /* 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);
2367
    fputs(_("error: "), stderr);
2368

K
Karel Zak 已提交
2369
    va_start(ap, format);
2370 2371 2372
    /* We can't recursively call vshError on an OOM situation, so ignore
       failure here. */
    ignore_value(virVasprintf(&str, format, ap));
K
Karel Zak 已提交
2373 2374
    va_end(ap);

2375
    fprintf(stderr, "%s\n", NULLSTR(str));
2376
    fflush(stderr);
2377
    VIR_FREE(str);
K
Karel Zak 已提交
2378 2379
}

2380

J
Jiri Denemark 已提交
2381 2382 2383 2384 2385
static void
vshEventLoop(void *opaque)
{
    vshControl *ctl = opaque;

2386 2387 2388 2389 2390 2391 2392 2393 2394 2395
    while (1) {
        bool quit;
        virMutexLock(&ctl->lock);
        quit = ctl->quit;
        virMutexUnlock(&ctl->lock);

        if (quit)
            break;

        if (virEventRunDefaultImpl() < 0)
J
Jiri Denemark 已提交
2396 2397 2398 2399 2400
            virshReportError(ctl);
    }
}


K
Karel Zak 已提交
2401
/*
2402
 * Initialize connection.
K
Karel Zak 已提交
2403
 */
E
Eric Blake 已提交
2404
static bool
2405
vshInit(vshControl *ctl)
2406
{
2407 2408
    char *debugEnv;

K
Karel Zak 已提交
2409
    if (ctl->conn)
E
Eric Blake 已提交
2410
        return false;
K
Karel Zak 已提交
2411

J
Jiri Denemark 已提交
2412
    if (ctl->debug == VSH_DEBUG_DEFAULT) {
2413 2414 2415
        /* log level not set from commandline, check env variable */
        debugEnv = getenv("VIRSH_DEBUG");
        if (debugEnv) {
J
Jiri Denemark 已提交
2416 2417 2418
            int debug;
            if (virStrToLong_i(debugEnv, NULL, 10, &debug) < 0 ||
                debug < VSH_ERR_DEBUG || debug > VSH_ERR_ERROR) {
2419 2420
                vshError(ctl, "%s",
                         _("VIRSH_DEBUG not set with a valid numeric value"));
J
Jiri Denemark 已提交
2421 2422
            } else {
                ctl->debug = debug;
2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434
            }
        }
    }

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

2435 2436
    vshOpenLogFile(ctl);

2437 2438
    /* set up the library error handler */
    virSetErrorFunc(NULL, virshErrorHandler);
2439

2440 2441 2442
    /* set up the signals handlers to catch disconnections */
    vshSetupSignals();

2443
    if (virEventRegisterDefaultImpl() < 0)
E
Eric Blake 已提交
2444
        return false;
2445

J
Jiri Denemark 已提交
2446 2447 2448 2449
    if (virThreadCreate(&ctl->eventLoop, true, vshEventLoop, ctl) < 0)
        return false;
    ctl->eventLoopStarted = true;

2450 2451 2452 2453
    if (ctl->name) {
        ctl->conn = virConnectOpenAuth(ctl->name,
                                       virConnectAuthPtrDefault,
                                       ctl->readonly ? VIR_CONNECT_RO : 0);
2454

2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465
        /* 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) {
            virshReportError(ctl);
            vshError(ctl, "%s", _("failed to connect to the hypervisor"));
            return false;
        }
2466
    }
K
Karel Zak 已提交
2467

E
Eric Blake 已提交
2468
    return true;
K
Karel Zak 已提交
2469 2470
}

2471 2472
#define LOGFILE_FLAGS (O_WRONLY | O_APPEND | O_CREAT | O_SYNC)

2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491
/**
 * vshOpenLogFile:
 *
 * Open log file.
 */
static void
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:
2492
                vshError(ctl, "%s",
J
Jim Meyering 已提交
2493
                         _("failed to get the log file information"));
2494
                exit(EXIT_FAILURE);
2495 2496 2497
        }
    } else {
        if (!S_ISREG(st.st_mode)) {
2498 2499
            vshError(ctl, "%s", _("the log path is not a file"));
            exit(EXIT_FAILURE);
2500 2501 2502 2503
        }
    }

    /* log file open */
2504
    if ((ctl->log_fd = open(ctl->logfile, LOGFILE_FLAGS, FILE_MODE)) < 0) {
2505
        vshError(ctl, "%s",
J
Jim Meyering 已提交
2506
                 _("failed to open the log file. check the log file path"));
2507
        exit(EXIT_FAILURE);
2508 2509 2510 2511 2512 2513 2514 2515 2516
    }
}

/**
 * vshOutputLogFile:
 *
 * Outputting an error to log file.
 */
static void
2517 2518
vshOutputLogFile(vshControl *ctl, int log_level, const char *msg_format,
                 va_list ap)
2519
{
2520 2521 2522
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    char *str;
    size_t len;
2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536
    const char *lvl = "";
    struct timeval stTimeval;
    struct tm *stTm;

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

    /**
     * create log format
     *
     * [YYYY.MM.DD HH:MM:SS SIGNATURE PID] LOG_LEVEL message
    */
    gettimeofday(&stTimeval, NULL);
    stTm = localtime(&stTimeval.tv_sec);
2537
    virBufferAsprintf(&buf, "[%d.%02d.%02d %02d:%02d:%02d %s %d] ",
2538 2539 2540 2541 2542 2543
                      (1900 + stTm->tm_year),
                      (1 + stTm->tm_mon),
                      stTm->tm_mday,
                      stTm->tm_hour,
                      stTm->tm_min,
                      stTm->tm_sec,
2544 2545
                      SIGN_NAME,
                      (int) getpid());
2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565
    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;
    }
2566 2567 2568
    virBufferAsprintf(&buf, "%s ", lvl);
    virBufferVasprintf(&buf, msg_format, ap);
    virBufferAddChar(&buf, '\n');
2569

2570 2571
    if (virBufferError(&buf))
        goto error;
2572

2573 2574 2575 2576 2577
    str = virBufferContentAndReset(&buf);
    len = strlen(str);
    if (len > 1 && str[len - 2] == '\n') {
        str[len - 1] = '\0';
        len--;
2578
    }
2579

2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590
    /* 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);
2591 2592 2593 2594 2595 2596 2597 2598 2599 2600 2601
}

/**
 * vshCloseLogFile:
 *
 * Close log file.
 */
static void
vshCloseLogFile(vshControl *ctl)
{
    /* log file close */
2602 2603 2604
    if (VIR_CLOSE(ctl->log_fd) < 0) {
        vshError(ctl, _("%s: failed to write log file: %s"),
                 ctl->logfile ? ctl->logfile : "?", strerror (errno));
2605 2606 2607
    }

    if (ctl->logfile) {
2608
        VIR_FREE(ctl->logfile);
2609 2610 2611 2612
        ctl->logfile = NULL;
    }
}

2613
#ifdef USE_READLINE
2614

K
Karel Zak 已提交
2615 2616 2617 2618 2619
/* -----------------
 * Readline stuff
 * -----------------
 */

2620
/*
K
Karel Zak 已提交
2621 2622
 * Generator function for command completion.  STATE lets us
 * know whether to start from scratch; without any state
2623
 * (i.e. STATE == 0), then we start at the top of the list.
K
Karel Zak 已提交
2624 2625
 */
static char *
2626 2627
vshReadlineCommandGenerator(const char *text, int state)
{
2628
    static int grp_list_index, cmd_list_index, len;
K
Karel Zak 已提交
2629
    const char *name;
2630 2631
    const vshCmdGrp *grp;
    const vshCmdDef *cmds;
K
Karel Zak 已提交
2632 2633

    if (!state) {
2634 2635
        grp_list_index = 0;
        cmd_list_index = 0;
2636
        len = strlen(text);
K
Karel Zak 已提交
2637 2638
    }

2639 2640
    grp = cmdGroups;

K
Karel Zak 已提交
2641
    /* Return the next name which partially matches from the
2642
     * command list.
K
Karel Zak 已提交
2643
     */
2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657
    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 已提交
2658 2659 2660 2661 2662 2663 2664
    }

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

static char *
2665 2666
vshReadlineOptionsGenerator(const char *text, int state)
{
K
Karel Zak 已提交
2667
    static int list_index, len;
2668
    static const vshCmdDef *cmd = NULL;
K
Karel Zak 已提交
2669
    const char *name;
K
Karel Zak 已提交
2670 2671 2672 2673 2674 2675 2676 2677 2678

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

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

2679
        cmdname = vshCalloc(NULL, (p - rl_line_buffer) + 1, 1);
2680
        memcpy(cmdname, rl_line_buffer, p - rl_line_buffer);
K
Karel Zak 已提交
2681 2682 2683

        cmd = vshCmddefSearch(cmdname);
        list_index = 0;
2684
        len = strlen(text);
2685
        VIR_FREE(cmdname);
K
Karel Zak 已提交
2686 2687 2688 2689
    }

    if (!cmd)
        return NULL;
2690

2691 2692 2693
    if (!cmd->opts)
        return NULL;

K
Karel Zak 已提交
2694
    while ((name = cmd->opts[list_index].name)) {
2695
        const vshCmdOptDef *opt = &cmd->opts[list_index];
K
Karel Zak 已提交
2696
        char *res;
2697

K
Karel Zak 已提交
2698
        list_index++;
2699

2700
        if (opt->type == VSH_OT_DATA || opt->type == VSH_OT_ARGV)
K
Karel Zak 已提交
2701 2702
            /* ignore non --option */
            continue;
2703

K
Karel Zak 已提交
2704
        if (len > 2) {
2705
            if (STRNEQLEN(name, text + 2, len - 2))
K
Karel Zak 已提交
2706 2707
                continue;
        }
2708
        res = vshMalloc(NULL, strlen(name) + 3);
2709
        snprintf(res, strlen(name) + 3,  "--%s", name);
K
Karel Zak 已提交
2710 2711 2712 2713 2714 2715 2716 2717
        return res;
    }

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

static char **
2718 2719 2720
vshReadlineCompletion(const char *text, int start,
                      int end ATTRIBUTE_UNUSED)
{
K
Karel Zak 已提交
2721 2722
    char **matches = (char **) NULL;

2723
    if (start == 0)
K
Karel Zak 已提交
2724
        /* command name generator */
2725
        matches = rl_completion_matches(text, vshReadlineCommandGenerator);
K
Karel Zak 已提交
2726 2727
    else
        /* commands options */
2728
        matches = rl_completion_matches(text, vshReadlineOptionsGenerator);
K
Karel Zak 已提交
2729 2730 2731 2732
    return matches;
}


2733 2734
static int
vshReadlineInit(vshControl *ctl)
2735
{
2736 2737
    char *userdir = NULL;

K
Karel Zak 已提交
2738 2739 2740 2741 2742
    /* 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;
2743 2744 2745

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

2747
    /* Prepare to read/write history from/to the $XDG_CACHE_HOME/virsh/history file */
2748
    userdir = virGetUserCacheDirectory();
2749

2750 2751
    if (userdir == NULL) {
        vshError(ctl, "%s", _("Could not determine home directory"));
2752
        return -1;
2753
    }
2754

2755
    if (virAsprintf(&ctl->historydir, "%s/virsh", userdir) < 0) {
2756
        vshError(ctl, "%s", _("Out of memory"));
2757
        VIR_FREE(userdir);
2758 2759 2760 2761 2762
        return -1;
    }

    if (virAsprintf(&ctl->historyfile, "%s/history", ctl->historydir) < 0) {
        vshError(ctl, "%s", _("Out of memory"));
2763
        VIR_FREE(userdir);
2764 2765 2766
        return -1;
    }

2767
    VIR_FREE(userdir);
2768 2769 2770 2771 2772 2773 2774

    read_history(ctl->historyfile);

    return 0;
}

static void
2775
vshReadlineDeinit(vshControl *ctl)
2776 2777
{
    if (ctl->historyfile != NULL) {
2778 2779
        if (virFileMakePathWithMode(ctl->historydir, 0755) < 0 &&
            errno != EEXIST) {
2780 2781
            char ebuf[1024];
            vshError(ctl, _("Failed to create '%s': %s"),
2782
                     ctl->historydir, virStrerror(errno, ebuf, sizeof(ebuf)));
E
Eric Blake 已提交
2783
        } else {
2784
            write_history(ctl->historyfile);
E
Eric Blake 已提交
2785
        }
2786 2787
    }

2788 2789
    VIR_FREE(ctl->historydir);
    VIR_FREE(ctl->historyfile);
K
Karel Zak 已提交
2790 2791
}

2792
static char *
2793
vshReadline(vshControl *ctl ATTRIBUTE_UNUSED, const char *prompt)
2794
{
2795
    return readline(prompt);
2796 2797
}

2798
#else /* !USE_READLINE */
2799

2800
static int
2801
vshReadlineInit(vshControl *ctl ATTRIBUTE_UNUSED)
2802 2803 2804 2805 2806
{
    /* empty */
    return 0;
}

2807
static void
2808
vshReadlineDeinit(vshControl *ctl ATTRIBUTE_UNUSED)
2809 2810 2811 2812 2813
{
    /* empty */
}

static char *
2814
vshReadline(vshControl *ctl, const char *prompt)
2815 2816 2817 2818 2819
{
    char line[1024];
    char *r;
    int len;

2820 2821
    fputs(prompt, stdout);
    r = fgets(line, sizeof(line), stdin);
2822 2823 2824
    if (r == NULL) return NULL; /* EOF */

    /* Chomp trailing \n */
2825
    len = strlen(r);
2826 2827 2828
    if (len > 0 && r[len-1] == '\n')
        r[len-1] = '\0';

2829
    return vshStrdup(ctl, r);
2830 2831
}

2832
#endif /* !USE_READLINE */
2833

2834 2835 2836 2837 2838 2839
static void
vshDeinitTimer(int timer ATTRIBUTE_UNUSED, void *opaque ATTRIBUTE_UNUSED)
{
    /* nothing to be done here */
}

K
Karel Zak 已提交
2840
/*
J
Jim Meyering 已提交
2841
 * Deinitialize virsh
K
Karel Zak 已提交
2842
 */
E
Eric Blake 已提交
2843
static bool
2844
vshDeinit(vshControl *ctl)
2845
{
2846
    vshReadlineDeinit(ctl);
2847
    vshCloseLogFile(ctl);
2848
    VIR_FREE(ctl->name);
K
Karel Zak 已提交
2849
    if (ctl->conn) {
2850 2851 2852
        int ret;
        if ((ret = virConnectClose(ctl->conn)) != 0) {
            vshError(ctl, _("Failed to disconnect from the hypervisor, %d leaked reference(s)"), ret);
K
Karel Zak 已提交
2853 2854
        }
    }
D
Daniel P. Berrange 已提交
2855 2856
    virResetLastError();

J
Jiri Denemark 已提交
2857
    if (ctl->eventLoopStarted) {
2858 2859 2860 2861
        int timer;

        virMutexLock(&ctl->lock);
        ctl->quit = true;
J
Jiri Denemark 已提交
2862
        /* HACK: Add a dummy timeout to break event loop */
2863 2864 2865 2866 2867
        timer = virEventAddTimeout(0, vshDeinitTimer, NULL, NULL);
        virMutexUnlock(&ctl->lock);

        virThreadJoin(&ctl->eventLoop);

J
Jiri Denemark 已提交
2868 2869 2870 2871 2872 2873
        if (timer != -1)
            virEventRemoveTimeout(timer);

        ctl->eventLoopStarted = false;
    }

2874 2875
    virMutexDestroy(&ctl->lock);

E
Eric Blake 已提交
2876
    return true;
K
Karel Zak 已提交
2877
}
2878

K
Karel Zak 已提交
2879 2880 2881 2882
/*
 * Print usage
 */
static void
2883
vshUsage(void)
2884
{
2885
    const vshCmdGrp *grp;
2886
    const vshCmdDef *cmd;
2887

L
Lai Jiangshan 已提交
2888 2889
    fprintf(stdout, _("\n%s [options]... [<command_string>]"
                      "\n%s [options]... <command> [args...]\n\n"
2890
                      "  options:\n"
2891
                      "    -c | --connect=URI      hypervisor connection URI\n"
2892
                      "    -r | --readonly         connect readonly\n"
2893
                      "    -d | --debug=NUM        debug level [0-4]\n"
2894 2895 2896
                      "    -h | --help             this help\n"
                      "    -q | --quiet            quiet mode\n"
                      "    -t | --timing           print timing information\n"
2897 2898 2899 2900 2901
                      "    -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"
2902
                      "  commands (non interactive mode):\n\n"), progname, progname);
2903

2904
    for (grp = cmdGroups; grp->name; grp++) {
E
Eric Blake 已提交
2905 2906 2907 2908 2909
        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;
2910
            fprintf(stdout,
E
Eric Blake 已提交
2911 2912 2913
                    "    %-30s %s\n", cmd->name,
                    _(vshCmddefGetInfo(cmd, "help")));
        }
2914 2915 2916 2917 2918
        fprintf(stdout, "\n");
    }

    fprintf(stdout, "%s",
            _("\n  (specify help <group> for details about the commands in the group)\n"));
2919 2920 2921
    fprintf(stdout, "%s",
            _("\n  (specify help <command> for details about the command)\n\n"));
    return;
K
Karel Zak 已提交
2922 2923
}

2924 2925 2926 2927 2928 2929 2930 2931 2932 2933
/*
 * 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 已提交
2934 2935
    vshPrint(ctl, "%s", _("Compiled with support for:\n"));
    vshPrint(ctl, "%s", _(" Hypervisors:"));
2936 2937 2938
#ifdef WITH_QEMU
    vshPrint(ctl, " QEmu/KVM");
#endif
D
Doug Goldstein 已提交
2939 2940 2941
#ifdef WITH_LXC
    vshPrint(ctl, " LXC");
#endif
2942 2943 2944
#ifdef WITH_UML
    vshPrint(ctl, " UML");
#endif
D
Doug Goldstein 已提交
2945 2946 2947 2948 2949 2950
#ifdef WITH_XEN
    vshPrint(ctl, " Xen");
#endif
#ifdef WITH_LIBXL
    vshPrint(ctl, " LibXL");
#endif
2951 2952 2953
#ifdef WITH_OPENVZ
    vshPrint(ctl, " OpenVZ");
#endif
D
Doug Goldstein 已提交
2954 2955
#ifdef WITH_VMWARE
    vshPrint(ctl, " VMWare");
2956
#endif
D
Doug Goldstein 已提交
2957 2958
#ifdef WITH_PHYP
    vshPrint(ctl, " PHYP");
2959
#endif
D
Doug Goldstein 已提交
2960 2961
#ifdef WITH_VBOX
    vshPrint(ctl, " VirtualBox");
2962 2963 2964 2965
#endif
#ifdef WITH_ESX
    vshPrint(ctl, " ESX");
#endif
D
Doug Goldstein 已提交
2966 2967
#ifdef WITH_HYPERV
    vshPrint(ctl, " Hyper-V");
2968
#endif
D
Doug Goldstein 已提交
2969 2970
#ifdef WITH_XENAPI
    vshPrint(ctl, " XenAPI");
2971 2972 2973 2974 2975 2976
#endif
#ifdef WITH_TEST
    vshPrint(ctl, " Test");
#endif
    vshPrint(ctl, "\n");

L
Laine Stump 已提交
2977
    vshPrint(ctl, "%s", _(" Networking:"));
2978 2979 2980 2981 2982 2983 2984 2985 2986 2987 2988 2989 2990
#ifdef WITH_REMOTE
    vshPrint(ctl, " Remote");
#endif
#ifdef WITH_LIBVIRTD
    vshPrint(ctl, " Daemon");
#endif
#ifdef WITH_NETWORK
    vshPrint(ctl, " Network");
#endif
#ifdef WITH_BRIDGE
    vshPrint(ctl, " Bridging");
#endif
#ifdef WITH_NETCF
D
Doug Goldstein 已提交
2991
    vshPrint(ctl, " Interface");
2992 2993 2994 2995 2996 2997 2998 2999 3000
#endif
#ifdef WITH_NWFILTER
    vshPrint(ctl, " Nwfilter");
#endif
#ifdef WITH_VIRTUALPORT
    vshPrint(ctl, " VirtualPort");
#endif
    vshPrint(ctl, "\n");

L
Laine Stump 已提交
3001
    vshPrint(ctl, "%s", _(" Storage:"));
3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015 3016 3017 3018 3019 3020 3021
#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");
3022 3023 3024
#endif
#ifdef WITH_STORAGE_RBD
    vshPrint(ctl, " RBD");
3025 3026 3027
#endif
#ifdef WITH_STORAGE_SHEEPDOG
    vshPrint(ctl, " Sheepdog");
3028 3029 3030
#endif
    vshPrint(ctl, "\n");

3031 3032 3033 3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056 3057 3058 3059 3060 3061 3062 3063 3064 3065 3066 3067 3068 3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079 3080 3081 3082 3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095 3096 3097 3098 3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158 3159 3160
    vshPrint(ctl, "%s", _(" Miscellaneous:"));
#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]
 *
 */
static bool
vshParseArgv(vshControl *ctl, int argc, char **argv)
{
    int arg, len;
    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':
            if (virStrToLong_i(optarg, NULL, 10, &ctl->debug) < 0) {
                vshError(ctl, "%s", _("option -d takes a numeric argument"));
                exit(EXIT_FAILURE);
            }
            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;
}

3161
#include "virsh-domain.c"
3162
#include "virsh-domain-monitor.c"
3163
#include "virsh-pool.c"
3164
#include "virsh-volume.c"
3165
#include "virsh-network.c"
3166
#include "virsh-nodedev.c"
3167
#include "virsh-interface.c"
3168
#include "virsh-nwfilter.c"
3169
#include "virsh-secret.c"
3170 3171
#include "virsh-snapshot.c"
#include "virsh-host.c"
3172

3173 3174 3175 3176 3177 3178 3179 3180 3181
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}
};
3182

3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197
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 已提交
3198

3199 3200 3201 3202
int
main(int argc, char **argv)
{
    vshControl _ctl, *ctl = &_ctl;
3203
    char *defaultConn;
E
Eric Blake 已提交
3204
    bool ret = true;
K
Karel Zak 已提交
3205

3206 3207 3208
    memset(ctl, 0, sizeof(vshControl));
    ctl->imode = true;          /* default is interactive mode */
    ctl->log_fd = -1;           /* Initialize log file descriptor */
J
Jiri Denemark 已提交
3209
    ctl->debug = VSH_DEBUG_DEFAULT;
3210 3211
    ctl->escapeChar = CTRL_CLOSE_BRACKET;

3212

3213 3214
    if (!setlocale(LC_ALL, "")) {
        perror("setlocale");
3215
        /* failure to setup locale is not fatal */
3216
    }
3217
    if (!bindtextdomain(PACKAGE, LOCALEDIR)) {
3218
        perror("bindtextdomain");
E
Eric Blake 已提交
3219
        return EXIT_FAILURE;
3220
    }
3221
    if (!textdomain(PACKAGE)) {
3222
        perror("textdomain");
E
Eric Blake 已提交
3223
        return EXIT_FAILURE;
3224 3225
    }

3226 3227 3228 3229 3230
    if (virMutexInit(&ctl->lock) < 0) {
        vshError(ctl, "%s", _("Failed to initialize mutex"));
        return EXIT_FAILURE;
    }

3231 3232 3233 3234 3235
    if (virInitialize() < 0) {
        vshError(ctl, "%s", _("Failed to initialize libvirt"));
        return EXIT_FAILURE;
    }

3236
    if (!(progname = strrchr(argv[0], '/')))
K
Karel Zak 已提交
3237 3238 3239
        progname = argv[0];
    else
        progname++;
3240

3241
    if ((defaultConn = getenv("VIRSH_DEFAULT_CONNECT_URI"))) {
E
Eric Blake 已提交
3242
        ctl->name = vshStrdup(ctl, defaultConn);
3243 3244
    }

D
Daniel P. Berrange 已提交
3245 3246
    if (!vshParseArgv(ctl, argc, argv)) {
        vshDeinit(ctl);
K
Karel Zak 已提交
3247
        exit(EXIT_FAILURE);
D
Daniel P. Berrange 已提交
3248
    }
3249

D
Daniel P. Berrange 已提交
3250 3251
    if (!vshInit(ctl)) {
        vshDeinit(ctl);
K
Karel Zak 已提交
3252
        exit(EXIT_FAILURE);
D
Daniel P. Berrange 已提交
3253
    }
3254

K
Karel Zak 已提交
3255
    if (!ctl->imode) {
3256
        ret = vshCommandRun(ctl, ctl->cmd);
3257
    } else {
K
Karel Zak 已提交
3258 3259
        /* interactive mode */
        if (!ctl->quiet) {
K
Karel Zak 已提交
3260
            vshPrint(ctl,
3261
                     _("Welcome to %s, the virtualization interactive terminal.\n\n"),
3262
                     progname);
J
Jim Meyering 已提交
3263
            vshPrint(ctl, "%s",
3264
                     _("Type:  'help' for help with commands\n"
3265
                       "       'quit' to quit\n\n"));
K
Karel Zak 已提交
3266
        }
3267 3268 3269 3270 3271 3272

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

K
Karel Zak 已提交
3273
        do {
3274
            const char *prompt = ctl->readonly ? VSH_PROMPT_RO : VSH_PROMPT_RW;
3275
            ctl->cmdstr =
3276
                vshReadline(ctl, prompt);
3277 3278
            if (ctl->cmdstr == NULL)
                break;          /* EOF */
K
Karel Zak 已提交
3279
            if (*ctl->cmdstr) {
3280
#if USE_READLINE
K
Karel Zak 已提交
3281
                add_history(ctl->cmdstr);
3282
#endif
3283
                if (vshCommandStringParse(ctl, ctl->cmdstr))
K
Karel Zak 已提交
3284 3285
                    vshCommandRun(ctl, ctl->cmd);
            }
3286
            VIR_FREE(ctl->cmdstr);
3287
        } while (ctl->imode);
K
Karel Zak 已提交
3288

3289 3290
        if (ctl->cmdstr == NULL)
            fputc('\n', stdout);        /* line break after alone prompt */
K
Karel Zak 已提交
3291
    }
3292

K
Karel Zak 已提交
3293 3294
    vshDeinit(ctl);
    exit(ret ? EXIT_SUCCESS : EXIT_FAILURE);
3295
}