virsh.c 560.8 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>
K
Karel Zak 已提交
36

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

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

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

static char *progname;

69 70
#define VIRSH_MAX_XML_FILE 10*1024*1024

K
Karel Zak 已提交
71 72 73
#define VSH_PROMPT_RW    "virsh # "
#define VSH_PROMPT_RO    "virsh > "

74 75
#define VIR_FROM_THIS VIR_FROM_NONE

K
Karel Zak 已提交
76 77 78 79 80
#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)

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

84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100
/**
 * 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 已提交
101
 * Indicates the level of a log message
102 103 104 105 106 107 108 109 110
 */
typedef enum {
    VSH_ERR_DEBUG = 0,
    VSH_ERR_INFO,
    VSH_ERR_NOTICE,
    VSH_ERR_WARNING,
    VSH_ERR_ERROR
} vshErrorLevel;

J
Jiri Denemark 已提交
111 112
#define VSH_DEBUG_DEFAULT VSH_ERR_ERROR

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

/*
134
 * vshCmdOptType - command option type
135
 */
K
Karel Zak 已提交
136
typedef enum {
137 138 139
    VSH_OT_BOOL,     /* optional boolean option */
    VSH_OT_STRING,   /* optional string option */
    VSH_OT_INT,      /* optional or mandatory int option */
140
    VSH_OT_DATA,     /* string data (as non-option) */
141
    VSH_OT_ARGV      /* remaining arguments */
K
Karel Zak 已提交
142 143
} vshCmdOptType;

144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159
/*
 * 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 已提交
160 161 162
/*
 * Command Option Flags
 */
E
Eric Blake 已提交
163 164 165 166
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 已提交
167
    VSH_OFLAG_REQ_OPT  = (1 << 2), /* --optionname required */
E
Eric Blake 已提交
168
};
K
Karel Zak 已提交
169 170 171 172 173 174

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

/*
E
Eric Blake 已提交
175 176 177 178 179
 * 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 已提交
180
 */
181
typedef struct {
E
Eric Blake 已提交
182 183
    const char *name;           /* name of information, or NULL for list end */
    const char *data;           /* non-NULL information */
K
Karel Zak 已提交
184 185 186 187 188
} vshCmdInfo;

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

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

208 209 210 211 212 213 214
/*
 * Command Usage Flags
 */
enum {
    VSH_CMD_FLAG_NOCONNECT = (1 << 0),  /* no prior connection needed */
};

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

/*
 * vshCmd - parsed command
 */
typedef struct __vshCmd {
230
    const vshCmdDef *def;       /* command definition */
231 232
    vshCmdOpt *opts;            /* list of command arguments */
    struct __vshCmd *next;      /* next command */
K
Karel Zak 已提交
233 234 235 236 237 238
} __vshCmd;

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

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

267
typedef struct vshCmdGrp {
E
Eric Blake 已提交
268
    const char *name;    /* name of group, or NULL for list end */
269 270 271
    const char *keyword; /* help keyword */
    const vshCmdDef *commands;
} vshCmdGrp;
272

273
static const vshCmdGrp cmdGroups[];
K
Karel Zak 已提交
274

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

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

287
static const char *vshCmddefGetInfo(const vshCmdDef *cmd, const char *info);
288
static const vshCmdDef *vshCmddefSearch(const char *cmdname);
E
Eric Blake 已提交
289
static bool vshCmddefHelp(vshControl *ctl, const char *name);
290
static const vshCmdGrp *vshCmdGrpSearch(const char *grpname);
E
Eric Blake 已提交
291
static bool vshCmdGrpHelp(vshControl *ctl, const char *name);
K
Karel Zak 已提交
292

E
Eric Blake 已提交
293 294 295
static int vshCommandOpt(const vshCmd *cmd, const char *name, vshCmdOpt **opt)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_NONNULL(2) ATTRIBUTE_NONNULL(3)
    ATTRIBUTE_RETURN_CHECK;
296 297
static int vshCommandOptInt(const vshCmd *cmd, const char *name, int *value)
    ATTRIBUTE_NONNULL(3) ATTRIBUTE_RETURN_CHECK;
298 299 300
static int vshCommandOptUInt(const vshCmd *cmd, const char *name,
                             unsigned int *value)
    ATTRIBUTE_NONNULL(3) ATTRIBUTE_RETURN_CHECK;
301 302 303 304 305 306 307 308 309
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;
310 311 312
static int vshCommandOptULongLong(const vshCmd *cmd, const char *name,
                                  unsigned long long *value)
    ATTRIBUTE_NONNULL(3) ATTRIBUTE_RETURN_CHECK;
E
Eric Blake 已提交
313
static bool vshCommandOptBool(const vshCmd *cmd, const char *name);
314 315
static const vshCmdOpt *vshCommandOptArgv(const vshCmd *cmd,
                                          const vshCmdOpt *opt);
316 317 318
static char *vshGetDomainDescription(vshControl *ctl, virDomainPtr dom,
                                     bool title, unsigned int flags)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_NONNULL(2) ATTRIBUTE_RETURN_CHECK;
K
Karel Zak 已提交
319

320 321 322
#define VSH_BYID     (1 << 1)
#define VSH_BYUUID   (1 << 2)
#define VSH_BYNAME   (1 << 3)
323
#define VSH_BYMAC    (1 << 4)
K
Karel Zak 已提交
324

325
static virDomainPtr vshCommandOptDomainBy(vshControl *ctl, const vshCmd *cmd,
326
                                          const char **name, int flag);
K
Karel Zak 已提交
327 328

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

332
static virNetworkPtr vshCommandOptNetworkBy(vshControl *ctl, const vshCmd *cmd,
333
                                            const char **name, int flag);
334 335

/* default is lookup by Name and UUID */
J
Jim Meyering 已提交
336 337
#define vshCommandOptNetwork(_ctl, _cmd, _name)                    \
    vshCommandOptNetworkBy(_ctl, _cmd, _name,                      \
338 339
                           VSH_BYUUID|VSH_BYNAME)

340
static virNWFilterPtr vshCommandOptNWFilterBy(vshControl *ctl, const vshCmd *cmd,
341
                                                  const char **name, int flag);
342 343 344 345 346 347

/* default is lookup by Name and UUID */
#define vshCommandOptNWFilter(_ctl, _cmd, _name)                    \
    vshCommandOptNWFilterBy(_ctl, _cmd, _name,                      \
                            VSH_BYUUID|VSH_BYNAME)

348
static virInterfacePtr vshCommandOptInterfaceBy(vshControl *ctl, const vshCmd *cmd,
349
                                                const char *optname,
350
                                                const char **name, int flag);
351 352 353

/* default is lookup by Name and MAC */
#define vshCommandOptInterface(_ctl, _cmd, _name)                    \
354 355
    vshCommandOptInterfaceBy(_ctl, _cmd, NULL, _name,                \
                             VSH_BYMAC|VSH_BYNAME)
356

357
static virStoragePoolPtr vshCommandOptPoolBy(vshControl *ctl, const vshCmd *cmd,
358
                            const char *optname, const char **name, int flag);
359 360 361 362 363 364

/* default is lookup by Name and UUID */
#define vshCommandOptPool(_ctl, _cmd, _optname, _name)           \
    vshCommandOptPoolBy(_ctl, _cmd, _optname, _name,             \
                           VSH_BYUUID|VSH_BYNAME)

365
static virStorageVolPtr vshCommandOptVolBy(vshControl *ctl, const vshCmd *cmd,
366 367
                                           const char *optname,
                                           const char *pooloptname,
368
                                           const char **name, int flag);
369 370

/* default is lookup by Name and UUID */
371
#define vshCommandOptVol(_ctl, _cmd, _optname, _pooloptname, _name)   \
372 373 374
    vshCommandOptVolBy(_ctl, _cmd, _optname, _pooloptname, _name,     \
                           VSH_BYUUID|VSH_BYNAME)

375
static virSecretPtr vshCommandOptSecret(vshControl *ctl, const vshCmd *cmd,
376
                                        const char **name);
377

378
static void vshPrintExtra(vshControl *ctl, const char *format, ...)
379
    ATTRIBUTE_FMT_PRINTF(2, 3);
380
static void vshDebug(vshControl *ctl, int level, const char *format, ...)
381
    ATTRIBUTE_FMT_PRINTF(3, 4);
K
Karel Zak 已提交
382 383

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

386
static int vshDomainState(vshControl *ctl, virDomainPtr dom, int *reason);
K
Karel Zak 已提交
387
static const char *vshDomainStateToString(int state);
388
static const char *vshDomainStateReasonToString(int state, int reason);
389
static const char *vshDomainControlStateToString(int state);
390
static const char *vshDomainVcpuStateToString(int state);
E
Eric Blake 已提交
391
static bool vshConnectionUsability(vshControl *ctl, virConnectPtr conn);
392 393 394
static virTypedParameterPtr vshFindTypedParamByName(const char *name,
                                                    virTypedParameterPtr list,
                                                    int count);
395 396
static char *vshGetTypedParamValue(vshControl *ctl, virTypedParameterPtr item)
    ATTRIBUTE_NONNULL(1) ATTRIBUTE_NONNULL(2);
K
Karel Zak 已提交
397

398 399 400 401
static char *editWriteToTempFile (vshControl *ctl, const char *doc);
static int   editFile (vshControl *ctl, const char *filename);
static char *editReadBackFile (vshControl *ctl, const char *filename);

402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424
/* 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);

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

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

431
static char *_vshStrdup(vshControl *ctl, const char *s, const char *filename, int line);
432 433
#define vshStrdup(_ctl, _s)    _vshStrdup(_ctl, _s, __FILE__, __LINE__)

434 435
static int parseRateStr(const char *rateStr, virNetDevBandwidthRatePtr rate);

E
Eric Blake 已提交
436 437 438
static void *
_vshMalloc(vshControl *ctl, size_t size, const char *filename, int line)
{
E
Eric Blake 已提交
439
    char *x;
E
Eric Blake 已提交
440

E
Eric Blake 已提交
441
    if (VIR_ALLOC_N(x, size) == 0)
E
Eric Blake 已提交
442 443 444 445 446 447 448 449 450
        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 已提交
451
    char *x;
E
Eric Blake 已提交
452

E
Eric Blake 已提交
453 454
    if (!xalloc_oversized(nmemb, size) &&
        VIR_ALLOC_N(x, nmemb * size) == 0)
E
Eric Blake 已提交
455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483
        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)
        return(NULL);
    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
484 485 486 487 488 489 490 491 492 493 494 495 496 497 498

static int idsorter(const void *a, const void *b) {
  const int *ia = (const int *)a;
  const int *ib = (const int *)b;

  if (*ia > *ib)
    return 1;
  else if (*ia < *ib)
    return -1;
  return 0;
}
static int namesorter(const void *a, const void *b) {
  const char **sa = (const char**)a;
  const char **sb = (const char**)b;

499
  /* User visible sort, so we want locale-specific case comparison.  */
500 501 502
  return strcasecmp(*sa, *sb);
}

503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524
static double
prettyCapacity(unsigned long long val,
               const char **unit) {
    if (val < 1024) {
        *unit = "";
        return (double)val;
    } else if (val < (1024.0l * 1024.0l)) {
        *unit = "KB";
        return (((double)val / 1024.0l));
    } else if (val < (1024.0l * 1024.0l * 1024.0l)) {
        *unit = "MB";
        return ((double)val / (1024.0l * 1024.0l));
    } else if (val < (1024.0l * 1024.0l * 1024.0l * 1024.0l)) {
        *unit = "GB";
        return ((double)val / (1024.0l * 1024.0l * 1024.0l));
    } else {
        *unit = "TB";
        return ((double)val / (1024.0l * 1024.0l * 1024.0l * 1024.0l));
    }
}


J
John Levon 已提交
525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549
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)
{
550 551 552 553 554 555 556 557
    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)
558
            goto out;
559
    }
J
John Levon 已提交
560 561

    if (last_error->code == VIR_ERR_OK) {
562
        vshError(ctl, "%s", _("unknown error"));
J
John Levon 已提交
563 564 565
        goto out;
    }

566
    vshError(ctl, "%s", last_error->message);
J
John Levon 已提交
567 568 569 570 571 572

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

573 574 575 576 577 578 579 580 581
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;
}

582 583 584 585 586
/*
 * Detection of disconnections and automatic reconnection support
 */
static int disconnected = 0; /* we may have been disconnected */

587 588 589 590 591
/* Gnulib doesn't guarantee SA_SIGINFO support.  */
#ifndef SA_SIGINFO
# define SA_SIGINFO 0
#endif

592 593 594 595 596 597
/*
 * 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
 */
598 599 600 601
static void vshCatchDisconnect(int sig, siginfo_t *siginfo,
                               void *context ATTRIBUTE_UNUSED) {
    if ((sig == SIGPIPE) ||
        (SA_SIGINFO && siginfo->si_signo == SIGPIPE))
602 603 604 605 606 607 608 609 610
        disconnected++;
}

/*
 * vshSetupSignals:
 *
 * Catch SIGPIPE signals which may arise when disconnection
 * from libvirtd occurs
 */
L
Laine Stump 已提交
611
static void
612 613 614 615 616 617 618 619 620 621 622 623 624
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 已提交
625
 * Reconnect after a disconnect from libvirtd
626 627
 *
 */
L
Laine Stump 已提交
628
static void
629 630 631 632 633 634
vshReconnect(vshControl *ctl)
{
    bool connected = false;

    if (ctl->conn != NULL) {
        connected = true;
635
        virConnectClose(ctl->conn);
636
    }
637 638 639 640 641 642

    ctl->conn = virConnectOpenAuth(ctl->name,
                                   virConnectAuthPtrDefault,
                                   ctl->readonly ? VIR_CONNECT_RO : 0);
    if (!ctl->conn)
        vshError(ctl, "%s", _("Failed to reconnect to the hypervisor"));
643
    else if (connected)
644 645
        vshError(ctl, "%s", _("Reconnected to the hypervisor"));
    disconnected = 0;
646
    ctl->useGetInfo = false;
647
    ctl->useSnapshotOld = false;
648
}
649

K
Karel Zak 已提交
650 651 652 653 654 655
/* ---------------
 * Commands
 * ---------------
 */

/*
656
 * "help" command
K
Karel Zak 已提交
657
 */
658
static const vshCmdInfo info_help[] = {
659
    {"help", N_("print help")},
660 661
    {"desc", N_("Prints global help, command specific help, or help for a\n"
                "    group of related commands")},
662

663
    {NULL, NULL}
K
Karel Zak 已提交
664 665
};

666
static const vshCmdOptDef opts_help[] = {
667
    {"command", VSH_OT_DATA, 0, N_("Prints global help, command specific help, or help for a group of related commands")},
668
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
669 670
};

E
Eric Blake 已提交
671
static bool
672
cmdHelp(vshControl *ctl, const vshCmd *cmd)
673
 {
674
    const char *name = NULL;
675

676
    if (vshCommandOptString(cmd, "command", &name) <= 0) {
677
        const vshCmdGrp *grp;
678
        const vshCmdDef *def;
679

680 681 682 683 684 685 686 687 688 689 690 691 692
        vshPrint(ctl, "%s", _("Grouped commands:\n\n"));

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

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

            vshPrint(ctl, "\n");
        }

E
Eric Blake 已提交
693
        return true;
694
    }
695

E
Eric Blake 已提交
696
    if (vshCmddefSearch(name)) {
697
        return vshCmddefHelp(ctl, name);
E
Eric Blake 已提交
698
    } else if (vshCmdGrpSearch(name)) {
699 700 701
        return vshCmdGrpHelp(ctl, name);
    } else {
        vshError(ctl, _("command or command group '%s' doesn't exist"), name);
E
Eric Blake 已提交
702
        return false;
K
Karel Zak 已提交
703 704 705
    }
}

706 707 708
/*
 * "autostart" command
 */
709
static const vshCmdInfo info_autostart[] = {
710
    {"help", N_("autostart a domain")},
711
    {"desc",
712
     N_("Configure a domain to be automatically started at boot.")},
713 714 715
    {NULL, NULL}
};

716
static const vshCmdOptDef opts_autostart[] = {
717 718
    {"domain",  VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"disable", VSH_OT_BOOL, 0, N_("disable autostarting")},
719 720 721
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
722
static bool
723
cmdAutostart(vshControl *ctl, const vshCmd *cmd)
724 725
{
    virDomainPtr dom;
726
    const char *name;
727 728
    int autostart;

729
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
730
        return false;
731

J
Jim Meyering 已提交
732
    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
E
Eric Blake 已提交
733
        return false;
734 735 736 737

    autostart = !vshCommandOptBool(cmd, "disable");

    if (virDomainSetAutostart(dom, autostart) < 0) {
738
        if (autostart)
739
            vshError(ctl, _("Failed to mark domain %s as autostarted"), name);
740
        else
741
            vshError(ctl, _("Failed to unmark domain %s as autostarted"), name);
742
        virDomainFree(dom);
E
Eric Blake 已提交
743
        return false;
744 745
    }

746
    if (autostart)
747
        vshPrint(ctl, _("Domain %s marked as autostarted\n"), name);
748
    else
749
        vshPrint(ctl, _("Domain %s unmarked as autostarted\n"), name);
750

751
    virDomainFree(dom);
E
Eric Blake 已提交
752
    return true;
753 754
}

K
Karel Zak 已提交
755
/*
756
 * "connect" command
K
Karel Zak 已提交
757
 */
758
static const vshCmdInfo info_connect[] = {
759
    {"help", N_("(re)connect to hypervisor")},
760
    {"desc",
761
     N_("Connect to local hypervisor. This is built-in command after shell start up.")},
762
    {NULL, NULL}
K
Karel Zak 已提交
763 764
};

765
static const vshCmdOptDef opts_connect[] = {
E
Eric Blake 已提交
766 767
    {"name",     VSH_OT_DATA, VSH_OFLAG_EMPTY_OK,
     N_("hypervisor connection URI")},
768
    {"readonly", VSH_OT_BOOL, 0, N_("read-only connection")},
769
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
770 771
};

E
Eric Blake 已提交
772
static bool
773
cmdConnect(vshControl *ctl, const vshCmd *cmd)
774
{
E
Eric Blake 已提交
775
    bool ro = vshCommandOptBool(cmd, "readonly");
776
    const char *name = NULL;
777

K
Karel Zak 已提交
778
    if (ctl->conn) {
779 780 781
        int ret;
        if ((ret = virConnectClose(ctl->conn)) != 0) {
            vshError(ctl, _("Failed to disconnect from the hypervisor, %d leaked reference(s)"), ret);
E
Eric Blake 已提交
782
            return false;
K
Karel Zak 已提交
783 784 785
        }
        ctl->conn = NULL;
    }
786

787
    VIR_FREE(ctl->name);
788 789
    if (vshCommandOptString(cmd, "name", &name) < 0) {
        vshError(ctl, "%s", _("Please specify valid connection URI"));
E
Eric Blake 已提交
790
        return false;
791
    }
792
    ctl->name = vshStrdup(ctl, name);
K
Karel Zak 已提交
793

794
    ctl->useGetInfo = false;
795
    ctl->useSnapshotOld = false;
E
Eric Blake 已提交
796
    ctl->readonly = ro;
K
Karel Zak 已提交
797

798 799 800
    ctl->conn = virConnectOpenAuth(ctl->name, virConnectAuthPtrDefault,
                                   ctl->readonly ? VIR_CONNECT_RO : 0);

K
Karel Zak 已提交
801
    if (!ctl->conn)
802
        vshError(ctl, "%s", _("Failed to connect to the hypervisor"));
803

E
Eric Blake 已提交
804
    return !!ctl->conn;
K
Karel Zak 已提交
805 806
}

807 808
#ifndef WIN32

809
/*
810
 * "console" command
811
 */
812
static const vshCmdInfo info_console[] = {
813
    {"help", N_("connect to the guest console")},
814
    {"desc",
815
     N_("Connect the virtual serial console for the guest")},
816 817 818
    {NULL, NULL}
};

819
static const vshCmdOptDef opts_console[] = {
820
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
821
    {"devname", VSH_OT_STRING, 0, N_("character device name")},
822 823 824
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
825
static bool
826
cmdRunConsole(vshControl *ctl, virDomainPtr dom, const char *name)
827
{
E
Eric Blake 已提交
828
    bool ret = false;
829
    int state;
830

831
    if ((state = vshDomainState(ctl, dom, NULL)) < 0) {
832 833 834 835
        vshError(ctl, "%s", _("Unable to get domain status"));
        goto cleanup;
    }

836
    if (state == VIR_DOMAIN_SHUTOFF) {
837 838 839 840
        vshError(ctl, "%s", _("The domain is not running"));
        goto cleanup;
    }

841 842 843 844 845
    if (!isatty(STDIN_FILENO)) {
        vshError(ctl, "%s", _("Cannot run interactive console without a controlling TTY"));
        goto cleanup;
    }

846
    vshPrintExtra(ctl, _("Connected to domain %s\n"), virDomainGetName(dom));
847
    vshPrintExtra(ctl, _("Escape character is %s\n"), ctl->escapeChar);
848
    fflush(stdout);
849
    if (vshRunConsole(dom, name, ctl->escapeChar) == 0)
E
Eric Blake 已提交
850
        ret = true;
851 852

 cleanup:
853

854 855 856
    return ret;
}

E
Eric Blake 已提交
857
static bool
858 859 860
cmdConsole(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
E
Eric Blake 已提交
861
    bool ret = false;
862
    const char *name = NULL;
863

864
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
865
        return false;
866 867

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
868
        return false;
869

870 871 872 873
    if (vshCommandOptString(cmd, "devname", &name) < 0) {
        vshError(ctl, "%s", _("Invalid devname"));
        goto cleanup;
    }
874

875
    ret = cmdRunConsole(ctl, dom, name);
876

877
cleanup:
878 879 880 881
    virDomainFree(dom);
    return ret;
}

882 883 884
#endif /* WIN32 */


K
Karel Zak 已提交
885 886 887
/*
 * "list" command
 */
888
static const vshCmdInfo info_list[] = {
889 890
    {"help", N_("list domains")},
    {"desc", N_("Returns list of domains.")},
891
    {NULL, NULL}
K
Karel Zak 已提交
892 893
};

894
static const vshCmdOptDef opts_list[] = {
895 896
    {"inactive", VSH_OT_BOOL, 0, N_("list inactive domains")},
    {"all", VSH_OT_BOOL, 0, N_("list inactive & active domains")},
897 898 899 900 901
    {"transient", VSH_OT_BOOL, 0, N_("list transient domains")},
    {"persistent", VSH_OT_BOOL, 0, N_("list persistent domains")},
    {"uuid", VSH_OT_BOOL, 0, N_("list uuid's only")},
    {"name", VSH_OT_BOOL, 0, N_("list domain names only")},
    {"table", VSH_OT_BOOL, 0, N_("list table (default)")},
E
Eric Blake 已提交
902 903
    {"managed-save", VSH_OT_BOOL, 0,
     N_("mark domains with managed save state")},
904
    {"title", VSH_OT_BOOL, 0, N_("show short domain description")},
905 906 907
    {NULL, 0, 0, NULL}
};

K
Karel Zak 已提交
908

E
Eric Blake 已提交
909
static bool
910
cmdList(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
911
{
912 913 914 915
    int inactive = vshCommandOptBool(cmd, "inactive");
    int all = vshCommandOptBool(cmd, "all");
    int active = !inactive || all ? 1 : 0;
    int *ids = NULL, maxid = 0, i;
916
    char **names = NULL;
917
    int maxname = 0;
E
Eric Blake 已提交
918
    bool managed = vshCommandOptBool(cmd, "managed-save");
919 920 921 922 923 924 925 926
    bool optTitle = vshCommandOptBool(cmd, "title");
    bool optTable = vshCommandOptBool(cmd, "table");
    bool optUUID = vshCommandOptBool(cmd, "uuid");
    bool optName = vshCommandOptBool(cmd, "name");
    bool optPersistent = vshCommandOptBool(cmd, "persistent");
    bool optTransient = vshCommandOptBool(cmd, "transient");
    bool persistUsed = true;
    virDomainPtr dom = NULL;
927
    char *title;
928
    char uuid[VIR_UUID_STRING_BUFLEN];
E
Eric Blake 已提交
929
    int state;
930
    int persistent;
931
    bool ret = false;
E
Eric Blake 已提交
932

933
    inactive |= all;
K
Karel Zak 已提交
934

935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951
    /* process arguments */
    if (!optPersistent && !optTransient) {
        optPersistent = true;
        optTransient = true;
        persistUsed = false;
    }

    if (optTable + optName + optUUID > 1) {
        vshError(ctl, "%s",
                 _("Only one argument from --table, --name and --uuid"
                   "may be specified."));
        return false;
    }

    if (!optUUID && !optName)
        optTable = true;

952
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
953
        return false;
954

955
    if (active) {
956 957
        maxid = virConnectNumOfDomains(ctl->conn);
        if (maxid < 0) {
958
            vshError(ctl, "%s", _("Failed to list active domains"));
E
Eric Blake 已提交
959
            return false;
960 961 962 963
        }
        if (maxid) {
            ids = vshMalloc(ctl, sizeof(int) * maxid);

964
            if ((maxid = virConnectListDomains(ctl->conn, ids, maxid)) < 0) {
965
                vshError(ctl, "%s", _("Failed to list active domains"));
966
                goto cleanup;
967 968
            }

969
            qsort(ids, maxid, sizeof(int), idsorter);
970
        }
971
    }
972

973
    if (inactive) {
974 975
        maxname = virConnectNumOfDefinedDomains(ctl->conn);
        if (maxname < 0) {
976
            vshError(ctl, "%s", _("Failed to list inactive domains"));
977
            goto cleanup;
978
        }
979 980 981
        if (maxname) {
            names = vshMalloc(ctl, sizeof(char *) * maxname);

982 983 984
            if ((maxname = virConnectListDefinedDomains(ctl->conn,
                                                        names,
                                                        maxname)) < 0) {
985
                vshError(ctl, "%s", _("Failed to list inactive domains"));
986
                goto cleanup;
987
            }
988

989
            qsort(&names[0], maxname, sizeof(char*), namesorter);
990
        }
991
    }
992

993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005
    /* print table header in legacy mode */
    if (optTable) {
        if (optTitle) {
            vshPrintExtra(ctl, " %-5s %-30s %-10s %-20s\n%s\n",
                          _("Id"), _("Name"), _("State"), _("Title"),
                          "-----------------------------------------"
                          "-----------------------------------------");
        } else {
            vshPrintExtra(ctl, " %-5s %-30s %s\n%s\n",
                          _("Id"), _("Name"), _("State"),
                          "-----------------------------------------"
                          "-----------");
        }
1006
    }
1007 1008

    for (i = 0; i < maxid; i++) {
1009
         dom = virDomainLookupByID(ctl->conn, ids[i]);
1010 1011

        /* this kind of work with domains is not atomic operation */
K
Karel Zak 已提交
1012 1013
        if (!dom)
            continue;
1014

1015 1016 1017 1018 1019
        if (persistUsed) {
            persistent = virDomainIsPersistent(dom);
            if (persistent < 0) {
                vshError(ctl, "%s",
                         _("Failed to determine domain's persistent state"));
1020
                goto cleanup;
1021
            }
1022

1023 1024 1025 1026 1027 1028
            if ((persistent && !optPersistent) ||
                (!persistent && !optTransient)) {
                virDomainFree(dom);
                dom = NULL;
                continue;
            }
1029
       }
1030

1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057
       if (optTable) {
           if (optTitle) {
               if (!(title = vshGetDomainDescription(ctl, dom, true, 0)))
                   goto cleanup;

               vshPrint(ctl, " %-5d %-30s %-10s %-20s\n",
                        virDomainGetID(dom),
                        virDomainGetName(dom),
                        _(vshDomainStateToString(vshDomainState(ctl, dom, NULL))),
                        title);
               VIR_FREE(title);
           } else {
               vshPrint(ctl, " %-5d %-30s %s\n",
                        virDomainGetID(dom),
                        virDomainGetName(dom),
                        _(vshDomainStateToString(vshDomainState(ctl, dom, NULL))));
           }
       } else if (optUUID) {
           if (virDomainGetUUIDString(dom, uuid) < 0) {
               vshError(ctl, "%s",
                        _("Failed to get domain's UUID"));
               goto cleanup;
           }
           vshPrint(ctl, "%s\n", uuid);
       } else if (optName) {
           vshPrint(ctl, "%s\n", virDomainGetName(dom));
       }
E
Eric Blake 已提交
1058

1059 1060 1061
       virDomainFree(dom);
       dom = NULL;
    }
1062

1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105
    if (optPersistent) {
        for (i = 0; i < maxname; i++) {
            dom = virDomainLookupByName(ctl->conn, names[i]);

            /* this kind of work with domains is not atomic operation */
            if (!dom)
                continue;

            if (optTable) {
                state = vshDomainState(ctl, dom, NULL);
                if (managed && state == VIR_DOMAIN_SHUTOFF &&
                    virDomainHasManagedSaveImage(dom, 0) > 0)
                    state = -2;

                if (optTitle) {
                    if (!(title = vshGetDomainDescription(ctl, dom, true, 0)))
                        goto cleanup;

                    vshPrint(ctl, " %-5s %-30s %-10s %-20s\n",
                             "-",
                             names[i],
                             state == -2 ? _("saved") : _(vshDomainStateToString(state)),
                             title);
                    VIR_FREE(title);
                } else {
                    vshPrint(ctl, " %-5s %-30s %s\n",
                             "-",
                             names[i],
                             state == -2 ? _("saved") : _(vshDomainStateToString(state)));
                }
           } else if (optUUID) {
               if (virDomainGetUUIDString(dom, uuid) < 0) {
                   vshError(ctl, "%s",
                            _("Failed to get domain's UUID"));
                   goto cleanup;
               }
               vshPrint(ctl, "%s\n", uuid);
           } else if (optName) {
               vshPrint(ctl, "%s\n", names[i]);
           }

           virDomainFree(dom);
           dom = NULL;
1106
        }
1107
    }
1108 1109 1110

    ret = true;
cleanup:
1111 1112
    if (dom)
        virDomainFree(dom);
1113
    VIR_FREE(ids);
1114 1115
    for (i = 0; i < maxname; i++)
        VIR_FREE(names[i]);
1116
    VIR_FREE(names);
1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145
    return ret;
}

/*
 * "desc" command for managing domain description and title
 */
static const vshCmdInfo info_desc[] = {
    {"help", N_("show or set domain's description or title")},
    {"desc", N_("Allows to show or modify description or title of a domain.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_desc[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"live", VSH_OT_BOOL, 0, N_("modify/get running state")},
    {"config", VSH_OT_BOOL, 0, N_("modify/get persistent configuration")},
    {"current", VSH_OT_BOOL, 0, N_("modify/get current state configuration")},
    {"title", VSH_OT_BOOL, 0, N_("modify the title instead of description")},
    {"edit", VSH_OT_BOOL, 0, N_("open an editor to modify the description")},
    {"new-desc", VSH_OT_ARGV, 0, N_("message")},
    {NULL, 0, 0, NULL}
};

static bool
cmdDesc(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
{
    virDomainPtr dom;
    bool config = vshCommandOptBool(cmd, "config");
    bool live = vshCommandOptBool(cmd, "live");
1146
    bool current = vshCommandOptBool(cmd, "current");
1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162

    bool title = vshCommandOptBool(cmd, "title");
    bool edit = vshCommandOptBool(cmd, "edit");

    int state;
    int type;
    char *desc = NULL;
    char *desc_edited = NULL;
    char *tmp = NULL;
    char *tmpstr;
    const vshCmdOpt *opt = NULL;
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    bool pad = false;
    bool ret = false;
    unsigned int flags = VIR_DOMAIN_AFFECT_CURRENT;

1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175
    if (current) {
        if (live || config) {
            vshError(ctl, "%s", _("--current must be specified exclusively"));
            return false;
        }
        flags = VIR_DOMAIN_AFFECT_CURRENT;
    } else {
        if (config)
            flags |= VIR_DOMAIN_AFFECT_CONFIG;
        if (live)
            flags |= VIR_DOMAIN_AFFECT_LIVE;
    }

1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225
    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

    if ((state = vshDomainState(ctl, dom, NULL)) < 0)
        goto cleanup;

    while ((opt = vshCommandOptArgv(cmd, opt))) {
        if (pad)
            virBufferAddChar(&buf, ' ');
        pad = true;
        virBufferAdd(&buf, opt->data, -1);
    }

    if (title)
        type = VIR_DOMAIN_METADATA_TITLE;
    else
        type = VIR_DOMAIN_METADATA_DESCRIPTION;

    if (virBufferError(&buf)) {
        vshPrint(ctl, "%s", _("Failed to collect new description/title"));
        goto cleanup;
    }
    desc = virBufferContentAndReset(&buf);

    if (edit || desc) {
        if (!desc) {
                desc = vshGetDomainDescription(ctl, dom, title,
                                           config?VIR_DOMAIN_XML_INACTIVE:0);
                if (!desc)
                    goto cleanup;
        }

        if (edit) {
            /* Create and open the temporary file. */
            if (!(tmp = editWriteToTempFile(ctl, desc)))
                goto cleanup;

            /* Start the editor. */
            if (editFile(ctl, tmp) == -1)
                goto cleanup;

            /* Read back the edited file. */
            if (!(desc_edited = editReadBackFile(ctl, tmp)))
                goto cleanup;

            /* strip a possible newline at the end of file; some
             * editors enforce a newline, this makes editing the title
1226
             * more convenient */
1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270
            if (title &&
                (tmpstr = strrchr(desc_edited, '\n')) &&
                *(tmpstr+1) == '\0')
                *tmpstr = '\0';

            /* Compare original XML with edited.  Has it changed at all? */
            if (STREQ(desc, desc_edited)) {
                vshPrint(ctl, _("Domain description not changed.\n"));
                ret = true;
                goto cleanup;
            }

            VIR_FREE(desc);
            desc = desc_edited;
            desc_edited = NULL;
        }

        if (virDomainSetMetadata(dom, type, desc, NULL, NULL, flags) < 0) {
            vshError(ctl, "%s",
                     _("Failed to set new domain description"));
            goto cleanup;
        }
        vshPrint(ctl, "%s", _("Domain description updated successfully"));
    } else {
        desc = vshGetDomainDescription(ctl, dom, title,
                                       config?VIR_DOMAIN_XML_INACTIVE:0);
        if (!desc)
            goto cleanup;

        if (strlen(desc) > 0)
            vshPrint(ctl, "%s", desc);
        else
            vshPrint(ctl, _("No description for domain: %s"),
                     virDomainGetName(dom));
    }

    ret = true;
cleanup:
    VIR_FREE(desc_edited);
    VIR_FREE(desc);
    if (tmp) {
        unlink(tmp);
        VIR_FREE(tmp);
    }
A
Alex Jia 已提交
1271 1272
    if (dom)
        virDomainFree(dom);
1273
    return ret;
K
Karel Zak 已提交
1274 1275 1276
}

/*
K
Karel Zak 已提交
1277
 * "domstate" command
K
Karel Zak 已提交
1278
 */
1279
static const vshCmdInfo info_domstate[] = {
1280 1281
    {"help", N_("domain state")},
    {"desc", N_("Returns state about a domain.")},
1282
    {NULL, NULL}
K
Karel Zak 已提交
1283 1284
};

1285
static const vshCmdOptDef opts_domstate[] = {
1286
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
1287
    {"reason", VSH_OT_BOOL, 0, N_("also print reason for the state")},
1288
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
1289 1290
};

E
Eric Blake 已提交
1291
static bool
1292
cmdDomstate(vshControl *ctl, const vshCmd *cmd)
1293
{
K
Karel Zak 已提交
1294
    virDomainPtr dom;
E
Eric Blake 已提交
1295
    bool ret = true;
1296 1297
    int showReason = vshCommandOptBool(cmd, "reason");
    int state, reason;
1298

1299
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
1300
        return false;
1301

J
Jim Meyering 已提交
1302
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
1303
        return false;
1304

1305
    if ((state = vshDomainState(ctl, dom, &reason)) < 0) {
E
Eric Blake 已提交
1306
        ret = false;
1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317
        goto cleanup;
    }

    if (showReason) {
        vshPrint(ctl, "%s (%s)\n",
                 _(vshDomainStateToString(state)),
                 vshDomainStateReasonToString(state, reason));
    } else {
        vshPrint(ctl, "%s\n",
                 _(vshDomainStateToString(state)));
    }
1318

1319
cleanup:
1320 1321 1322 1323
    virDomainFree(dom);
    return ret;
}

1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370
/*
 * "domcontrol" command
 */
static const vshCmdInfo info_domcontrol[] = {
    {"help", N_("domain control interface state")},
    {"desc", N_("Returns state of a control interface to the domain.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_domcontrol[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {NULL, 0, 0, NULL}
};

static bool
cmdDomControl(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    bool ret = true;
    virDomainControlInfo info;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

    if (virDomainGetControlInfo(dom, &info, 0) < 0) {
        ret = false;
        goto cleanup;
    }

    if (info.state != VIR_DOMAIN_CONTROL_OK &&
        info.state != VIR_DOMAIN_CONTROL_ERROR) {
        vshPrint(ctl, "%s (%0.3fs)\n",
                 _(vshDomainControlStateToString(info.state)),
                 info.stateTime / 1000.0);
    } else {
        vshPrint(ctl, "%s\n",
                 _(vshDomainControlStateToString(info.state)));
    }

cleanup:
    virDomainFree(dom);
    return ret;
}

1371 1372
/* "domblkstat" command
 */
1373
static const vshCmdInfo info_domblkstat[] = {
1374
    {"help", N_("get device block stats for a domain")},
1375 1376
    {"desc", N_("Get device block stats for a running domain. See man page or "
                "use --human for explanation of fields")},
1377 1378 1379
    {NULL,NULL}
};

1380
static const vshCmdOptDef opts_domblkstat[] = {
1381 1382
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"device", VSH_OT_DATA, VSH_OFLAG_REQ, N_("block device")},
1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396
    {"human",  VSH_OT_BOOL, 0, N_("print a more human readable output")},
    {NULL, 0, 0, NULL}
};

struct _domblkstat_sequence {
    const char *field;  /* field name */
    const char *legacy; /* legacy name from previous releases */
    const char *human;  /* human-friendly explanation */
};

/* sequence of values for output to honor legacy format from previous
 * versions */
static const struct _domblkstat_sequence domblkstat_output[] = {
    { VIR_DOMAIN_BLOCK_STATS_READ_REQ,          "rd_req",
1397
      N_("number of read operations:") }, /* 0 */
1398
    { VIR_DOMAIN_BLOCK_STATS_READ_BYTES,        "rd_bytes",
1399
      N_("number of bytes read:") }, /* 1 */
1400
    { VIR_DOMAIN_BLOCK_STATS_WRITE_REQ,         "wr_req",
1401
      N_("number of write operations:") }, /* 2 */
1402
    { VIR_DOMAIN_BLOCK_STATS_WRITE_BYTES,       "wr_bytes",
1403
      N_("number of bytes written:") }, /* 3 */
1404
    { VIR_DOMAIN_BLOCK_STATS_ERRS,              "errs",
1405
      N_("error count:") }, /* 4 */
1406
    { VIR_DOMAIN_BLOCK_STATS_FLUSH_REQ,         NULL,
1407
      N_("number of flush operations:") }, /* 5 */
1408
    { VIR_DOMAIN_BLOCK_STATS_READ_TOTAL_TIMES,  NULL,
1409
      N_("total duration of reads (ns):") }, /* 6 */
1410
    { VIR_DOMAIN_BLOCK_STATS_WRITE_TOTAL_TIMES, NULL,
1411
      N_("total duration of writes (ns):") }, /* 7 */
1412 1413 1414 1415 1416 1417 1418
    { VIR_DOMAIN_BLOCK_STATS_FLUSH_TOTAL_TIMES, NULL,
      N_("total duration of flushes (ns):") }, /* 8 */
    { NULL, NULL, NULL }
};

#define DOMBLKSTAT_LEGACY_PRINT(ID, VALUE)              \
    if (VALUE >= 0)                                     \
1419 1420
        vshPrint(ctl, "%s %-*s %lld\n", device,         \
                 human ? 31 : 0,                        \
1421 1422 1423
                 human ? _(domblkstat_output[ID].human) \
                 : domblkstat_output[ID].legacy,        \
                 VALUE);
1424

E
Eric Blake 已提交
1425
static bool
1426
cmdDomblkstat (vshControl *ctl, const vshCmd *cmd)
1427 1428
{
    virDomainPtr dom;
1429
    const char *name = NULL, *device = NULL;
1430
    struct _virDomainBlockStats stats;
1431
    virTypedParameterPtr params = NULL;
1432 1433 1434
    virTypedParameterPtr par = NULL;
    char *value = NULL;
    const char *field = NULL;
1435
    int rc, nparams = 0;
1436
    int i = 0;
1437
    bool ret = false;
1438
    bool human = vshCommandOptBool(cmd, "human"); /* human readable output */
1439

1440
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
1441
        return false;
1442

1443
    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
E
Eric Blake 已提交
1444
        return false;
1445

1446
    if (vshCommandOptString(cmd, "device", &device) <= 0)
1447
        goto cleanup;
1448

1449 1450 1451 1452 1453 1454 1455
    rc = virDomainBlockStatsFlags(dom, device, NULL, &nparams, 0);

    /* It might fail when virDomainBlockStatsFlags is not
     * supported on older libvirt, fallback to use virDomainBlockStats
     * then.
     */
    if (rc < 0) {
1456 1457
        /* try older API if newer is not supported */
        if (last_error->code != VIR_ERR_NO_SUPPORT)
1458 1459
            goto cleanup;

1460 1461
        virFreeError(last_error);
        last_error = NULL;
1462

1463 1464 1465 1466 1467 1468
        if (virDomainBlockStats(dom, device, &stats,
                                sizeof stats) == -1) {
            vshError(ctl, _("Failed to get block stats %s %s"),
                     name, device);
            goto cleanup;
        }
1469

1470 1471 1472 1473
        /* human friendly output */
        if (human) {
            vshPrint(ctl, N_("Device: %s\n"), device);
            device = "";
1474
        }
1475 1476 1477 1478 1479 1480

        DOMBLKSTAT_LEGACY_PRINT(0, stats.rd_req);
        DOMBLKSTAT_LEGACY_PRINT(1, stats.rd_bytes);
        DOMBLKSTAT_LEGACY_PRINT(2, stats.wr_req);
        DOMBLKSTAT_LEGACY_PRINT(3, stats.wr_bytes);
        DOMBLKSTAT_LEGACY_PRINT(4, stats.errs);
1481
    } else {
1482
        params = vshCalloc(ctl, nparams, sizeof(*params));
1483

1484 1485 1486 1487
        if (virDomainBlockStatsFlags (dom, device, params, &nparams, 0) < 0) {
            vshError(ctl, _("Failed to get block stats %s %s"), name, device);
            goto cleanup;
        }
1488

1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501
        /* set for prettier output */
        if (human) {
            vshPrint(ctl, N_("Device: %s\n"), device);
            device = "";
        }

        /* at first print all known values in desired order */
        for (i = 0; domblkstat_output[i].field != NULL; i++) {
            if (!(par = vshFindTypedParamByName(domblkstat_output[i].field,
                                                params,
                                                nparams)))
                continue;

1502
            value = vshGetTypedParamValue(ctl, par);
1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517

            /* to print other not supported fields, mark the already printed */
            par->field[0] = '\0'; /* set the name to empty string */

            /* translate into human readable or legacy spelling */
            field = NULL;
            if (human)
                field = _(domblkstat_output[i].human);
            else
                field = domblkstat_output[i].legacy;

            /* use the provided spelling if no translation is available */
            if (!field)
                field = domblkstat_output[i].field;

1518 1519
            vshPrint(ctl, "%s %-*s %s\n", device,
                     human ? 31 : 0, field, value);
1520 1521 1522 1523 1524

            VIR_FREE(value);
        }

        /* go through the fields again, for remaining fields */
1525
        for (i = 0; i < nparams; i++) {
1526 1527
            if (!*params[i].field)
                continue;
1528

1529
            value = vshGetTypedParamValue(ctl, params+i);
1530 1531
            vshPrint(ctl, "%s %s %s\n", device, params[i].field, value);
            VIR_FREE(value);
1532 1533 1534 1535 1536 1537 1538
        }
    }

    ret = true;

cleanup:
    VIR_FREE(params);
1539
    virDomainFree(dom);
1540
    return ret;
1541
}
1542
#undef DOMBLKSTAT_LEGACY_PRINT
1543 1544 1545

/* "domifstat" command
 */
1546
static const vshCmdInfo info_domifstat[] = {
1547 1548
    {"help", N_("get network interface stats for a domain")},
    {"desc", N_("Get network interface stats for a running domain.")},
1549 1550 1551
    {NULL,NULL}
};

1552
static const vshCmdOptDef opts_domifstat[] = {
1553 1554
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"interface", VSH_OT_DATA, VSH_OFLAG_REQ, N_("interface device")},
1555 1556 1557
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
1558
static bool
1559
cmdDomIfstat (vshControl *ctl, const vshCmd *cmd)
1560 1561
{
    virDomainPtr dom;
1562
    const char *name = NULL, *device = NULL;
1563 1564
    struct _virDomainInterfaceStats stats;

1565
    if (!vshConnectionUsability (ctl, ctl->conn))
E
Eric Blake 已提交
1566
        return false;
1567

J
Jim Meyering 已提交
1568
    if (!(dom = vshCommandOptDomain (ctl, cmd, &name)))
E
Eric Blake 已提交
1569
        return false;
1570

1571
    if (vshCommandOptString (cmd, "interface", &device) <= 0) {
L
Laine Stump 已提交
1572
        virDomainFree(dom);
E
Eric Blake 已提交
1573
        return false;
L
Laine Stump 已提交
1574
    }
1575 1576

    if (virDomainInterfaceStats (dom, device, &stats, sizeof stats) == -1) {
1577
        vshError(ctl, _("Failed to get interface stats %s %s"), name, device);
1578
        virDomainFree(dom);
E
Eric Blake 已提交
1579
        return false;
1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606
    }

    if (stats.rx_bytes >= 0)
        vshPrint (ctl, "%s rx_bytes %lld\n", device, stats.rx_bytes);

    if (stats.rx_packets >= 0)
        vshPrint (ctl, "%s rx_packets %lld\n", device, stats.rx_packets);

    if (stats.rx_errs >= 0)
        vshPrint (ctl, "%s rx_errs %lld\n", device, stats.rx_errs);

    if (stats.rx_drop >= 0)
        vshPrint (ctl, "%s rx_drop %lld\n", device, stats.rx_drop);

    if (stats.tx_bytes >= 0)
        vshPrint (ctl, "%s tx_bytes %lld\n", device, stats.tx_bytes);

    if (stats.tx_packets >= 0)
        vshPrint (ctl, "%s tx_packets %lld\n", device, stats.tx_packets);

    if (stats.tx_errs >= 0)
        vshPrint (ctl, "%s tx_errs %lld\n", device, stats.tx_errs);

    if (stats.tx_drop >= 0)
        vshPrint (ctl, "%s tx_drop %lld\n", device, stats.tx_drop);

    virDomainFree(dom);
E
Eric Blake 已提交
1607
    return true;
1608 1609
}

1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629
/* "domif-setlink" command
 */
static const vshCmdInfo info_domif_setlink[] = {
    {"help", N_("set link state of a virtual interface")},
    {"desc", N_("Set link state of a domain's virtual interface. This command wraps usage of update-device command.")},
    {NULL,NULL}
};

static const vshCmdOptDef opts_domif_setlink[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"interface", VSH_OT_DATA, VSH_OFLAG_REQ, N_("interface device (MAC Address)")},
    {"state", VSH_OT_DATA, VSH_OFLAG_REQ, N_("new state of the device")},
    {"persistent", VSH_OT_BOOL, 0, N_("persist interface state")},
    {NULL, 0, 0, NULL}
};

static bool
cmdDomIfSetLink (vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
1630
    const char *iface;
1631
    const char *state;
1632
    const char *value;
1633
    const char *desc;
1634 1635 1636
    unsigned char macaddr[VIR_MAC_BUFLEN];
    const char *element;
    const char *attr;
1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652
    bool persistent;
    bool ret = false;
    unsigned int flags = 0;
    int i;
    xmlDocPtr xml = NULL;
    xmlXPathContextPtr ctxt = NULL;
    xmlXPathObjectPtr obj = NULL;
    xmlNodePtr cur = NULL;
    xmlBufferPtr xml_buf = NULL;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

1653
    if (vshCommandOptString(cmd, "interface", &iface) <= 0)
1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681
        goto cleanup;

    if (vshCommandOptString(cmd, "state", &state) <= 0)
        goto cleanup;

    persistent = vshCommandOptBool(cmd, "persistent");

    if (STRNEQ(state, "up") && STRNEQ(state, "down")) {
        vshError(ctl, _("invalid link state '%s'"), state);
        goto cleanup;
    }

    /* get persistent or live description of network device */
    desc = virDomainGetXMLDesc(dom, persistent?VIR_DOMAIN_XML_INACTIVE:0);
    if (desc == NULL) {
        vshError(ctl, _("Failed to get domain description xml"));
        goto cleanup;
    }

    if (persistent)
        flags = VIR_DOMAIN_AFFECT_CONFIG;
    else
        flags = VIR_DOMAIN_AFFECT_LIVE;

    if (virDomainIsActive(dom) == 0)
        flags = VIR_DOMAIN_AFFECT_CONFIG;

    /* extract current network device description */
1682
    xml = virXMLParseStringCtxt(desc, _("(domain_definition)"), &ctxt);
1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695
    VIR_FREE(desc);
    if (!xml) {
        vshError(ctl, _("Failed to parse domain description xml"));
        goto cleanup;
    }

    obj = xmlXPathEval(BAD_CAST "/domain/devices/interface", ctxt);
    if (obj == NULL || obj->type != XPATH_NODESET ||
        obj->nodesetval == NULL || obj->nodesetval->nodeNr == 0) {
        vshError(ctl, _("Failed to extract interface information or no interfaces found"));
        goto cleanup;
    }

1696
    if (virMacAddrParse(iface, macaddr) == 0) {
1697 1698 1699 1700 1701 1702 1703
        element = "mac";
        attr = "address";
    } else {
        element = "target";
        attr = "dev";
    }

1704 1705 1706 1707 1708 1709
    /* find interface with matching mac addr */
    for (i = 0; i < obj->nodesetval->nodeNr; i++) {
        cur = obj->nodesetval->nodeTab[i]->children;

        while (cur) {
            if (cur->type == XML_ELEMENT_NODE &&
1710 1711
                xmlStrEqual(cur->name, BAD_CAST element)) {
                value = virXMLPropString(cur, attr);
1712

1713 1714
                if (STRCASEEQ(value, iface)) {
                    VIR_FREE(value);
1715 1716
                    goto hit;
                }
1717
                VIR_FREE(value);
1718 1719 1720 1721 1722
            }
            cur = cur->next;
        }
    }

1723
    vshError(ctl, _("interface (%s: %s) not found"), element, iface);
1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804
    goto cleanup;

hit:
    /* find and modify/add link state node */
    /* try to find <link> element */
    cur = obj->nodesetval->nodeTab[i]->children;

    while (cur) {
        if (cur->type == XML_ELEMENT_NODE &&
            xmlStrEqual(cur->name, BAD_CAST "link")) {
            /* found, just modify the property */
            xmlSetProp(cur, BAD_CAST "state", BAD_CAST state);

            break;
        }
        cur = cur->next;
    }

    if (!cur) {
        /* element <link> not found, add one */
        cur = xmlNewChild(obj->nodesetval->nodeTab[i],
                          NULL,
                          BAD_CAST "link",
                          NULL);
        if (!cur)
            goto cleanup;

        if (xmlNewProp(cur, BAD_CAST "state", BAD_CAST state) == NULL)
            goto cleanup;
    }

    xml_buf = xmlBufferCreate();
    if (!xml_buf) {
        vshError(ctl, _("Failed to allocate memory"));
        goto cleanup;
    }

    if (xmlNodeDump(xml_buf, xml, obj->nodesetval->nodeTab[i], 0, 0) < 0 ) {
        vshError(ctl, _("Failed to create XML"));
        goto cleanup;
    }

    if (virDomainUpdateDeviceFlags(dom, (char *)xmlBufferContent(xml_buf), flags) < 0) {
        vshError(ctl, _("Failed to update interface link state"));
        goto cleanup;
    } else {
        vshPrint(ctl, "%s", _("Device updated successfully\n"));
        ret = true;
    }

cleanup:
    xmlXPathFreeObject(obj);
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(xml);
    xmlBufferFree(xml_buf);

    if (dom)
        virDomainFree(dom);

    return ret;
}

/* "domif-getlink" command
 */
static const vshCmdInfo info_domif_getlink[] = {
    {"help", N_("get link state of a virtual interface")},
    {"desc", N_("Get link state of a domain's virtual interface.")},
    {NULL,NULL}
};

static const vshCmdOptDef opts_domif_getlink[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"interface", VSH_OT_DATA, VSH_OFLAG_REQ, N_("interface device (MAC Address)")},
    {"persistent", VSH_OT_BOOL, 0, N_("Get persistent interface state")},
    {NULL, 0, 0, NULL}
};

static bool
cmdDomIfGetLink (vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
1805
    const char *iface = NULL;
1806 1807
    int flags = 0;
    char *state = NULL;
1808 1809 1810 1811
    char *value = NULL;
    unsigned char macaddr[VIR_MAC_BUFLEN];
    const char *element;
    const char *attr;
1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825
    bool ret = false;
    int i;
    char *desc;
    xmlDocPtr xml = NULL;
    xmlXPathContextPtr ctxt = NULL;
    xmlNodePtr cur = NULL;
    xmlXPathObjectPtr obj = NULL;

    if (!vshConnectionUsability (ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain (ctl, cmd, NULL)))
        return false;

1826
    if (vshCommandOptString (cmd, "interface", &iface) <= 0) {
1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839
        virDomainFree(dom);
        return false;
    }

    if (vshCommandOptBool(cmd, "persistent"))
        flags = VIR_DOMAIN_XML_INACTIVE;

    desc = virDomainGetXMLDesc(dom, flags);
    if (desc == NULL) {
        vshError(ctl, _("Failed to get domain description xml"));
        goto cleanup;
    }

1840
    xml = virXMLParseStringCtxt(desc, _("(domain_definition)"), &ctxt);
1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853
    VIR_FREE(desc);
    if (!xml) {
        vshError(ctl, _("Failed to parse domain description xml"));
        goto cleanup;
    }

    obj = xmlXPathEval(BAD_CAST "/domain/devices/interface", ctxt);
    if (obj == NULL || obj->type != XPATH_NODESET ||
        obj->nodesetval == NULL || obj->nodesetval->nodeNr == 0) {
        vshError(ctl, _("Failed to extract interface information or no interfaces found"));
        goto cleanup;
    }

1854
    if (virMacAddrParse(iface, macaddr) == 0) {
1855 1856 1857 1858 1859 1860 1861
        element = "mac";
        attr = "address";
    } else {
        element = "target";
        attr = "dev";
    }

1862 1863 1864 1865 1866 1867
    /* find interface with matching mac addr */
    for (i = 0; i < obj->nodesetval->nodeNr; i++) {
        cur = obj->nodesetval->nodeTab[i]->children;

        while (cur) {
            if (cur->type == XML_ELEMENT_NODE &&
1868
                xmlStrEqual(cur->name, BAD_CAST element)) {
1869

1870
                value = virXMLPropString(cur, attr);
1871

1872 1873
                if (STRCASEEQ(value, iface)) {
                    VIR_FREE(value);
1874 1875
                    goto hit;
                }
1876
                VIR_FREE(value);
1877 1878 1879 1880 1881
            }
            cur = cur->next;
        }
    }

1882
    vshError(ctl, _("Interface (%s: %s) not found."), element, iface);
1883 1884 1885 1886 1887 1888 1889 1890 1891
    goto cleanup;

hit:
    cur = obj->nodesetval->nodeTab[i]->children;
    while (cur) {
        if (cur->type == XML_ELEMENT_NODE &&
            xmlStrEqual(cur->name, BAD_CAST "link")) {

            state = virXMLPropString(cur, "state");
1892
            vshPrint(ctl, "%s %s", iface, state);
1893 1894 1895 1896 1897 1898 1899 1900
            VIR_FREE(state);

            goto cleanup;
        }
        cur = cur->next;
    }

    /* attribute not found */
1901
    vshPrint(ctl, "%s default", iface);
1902

1903
    ret = true;
1904 1905 1906 1907 1908 1909 1910 1911 1912 1913
cleanup:
    xmlXPathFreeObject(obj);
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(xml);
    if (dom)
        virDomainFree(dom);

    return ret;
}

1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108
/* "domiftune" command
 */
static const vshCmdInfo info_domiftune[] = {
    {"help", N_("get/set parameters of a virtual interface")},
    {"desc", N_("Get/set parameters of a domain's virtual interface.")},
    {NULL,NULL}
};

static const vshCmdOptDef opts_domiftune[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"interface", VSH_OT_DATA, VSH_OFLAG_REQ, N_("interface device")},
    {"inbound", VSH_OT_DATA, VSH_OFLAG_NONE, N_("control domain's incoming traffics")},
    {"outbound", VSH_OT_DATA, VSH_OFLAG_NONE, N_("control domain's outgoing traffics")},
    {"config", VSH_OT_BOOL, VSH_OFLAG_NONE, N_("affect next boot")},
    {"live", VSH_OT_BOOL, VSH_OFLAG_NONE, N_("affect running domain")},
    {"current", VSH_OT_BOOL, VSH_OFLAG_NONE, N_("affect current domain")},
    {NULL, 0, 0, NULL}
};

static bool
cmdDomIftune(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    const char *name = NULL, *device = NULL,
               *inboundStr = NULL, *outboundStr = NULL;
    unsigned int flags = 0;
    int nparams = 0;
    virTypedParameterPtr params = NULL;
    bool ret = false;
    int current = vshCommandOptBool(cmd, "current");
    int config = vshCommandOptBool(cmd, "config");
    int live = vshCommandOptBool(cmd, "live");
    virNetDevBandwidthRate inbound, outbound;
    int i;

    if (current) {
        if (live || config) {
            vshError(ctl, "%s", _("--current must be specified exclusively"));
            return false;
        }
        flags = VIR_DOMAIN_AFFECT_CURRENT;
    } else {
        if (config)
            flags |= VIR_DOMAIN_AFFECT_CONFIG;
        if (live)
            flags |= VIR_DOMAIN_AFFECT_LIVE;
    }

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
        return false;

    if (vshCommandOptString(cmd, "interface", &device) <= 0) {
        virDomainFree(dom);
        return false;
    }

    if (vshCommandOptString(cmd, "inbound", &inboundStr) < 0 ||
        vshCommandOptString(cmd, "outbound", &outboundStr) < 0) {
        vshError(ctl, "missing argument");
        goto cleanup;
    }

    memset(&inbound, 0, sizeof(inbound));
    memset(&outbound, 0, sizeof(outbound));

    if (inboundStr) {
        if (parseRateStr(inboundStr, &inbound) < 0) {
            vshError(ctl, _("inbound format is incorrect"));
            goto cleanup;
        }
        if (inbound.average == 0) {
            vshError(ctl, _("inbound average is mandatory"));
            goto cleanup;
        }
        nparams++; /* average */
        if (inbound.peak) nparams++;
        if (inbound.burst) nparams++;
    }
    if (outboundStr) {
        if (parseRateStr(outboundStr, &outbound) < 0) {
            vshError(ctl, _("outbound format is incorrect"));
            goto cleanup;
        }
        if (outbound.average == 0) {
            vshError(ctl, _("outbound average is mandatory"));
            goto cleanup;
        }
        nparams++; /* average */
        if (outbound.peak) nparams++;
        if (outbound.burst) nparams++;
    }

    if (nparams == 0) {
        /* get the number of interface parameters */
        if (virDomainGetInterfaceParameters(dom, device, NULL, &nparams, flags) != 0) {
            vshError(ctl, "%s",
                     _("Unable to get number of interface parameters"));
            goto cleanup;
        }

        if (nparams == 0) {
            /* nothing to output */
            ret = true;
            goto cleanup;
        }

        /* get all interface parameters */
        params = vshCalloc(ctl, nparams, sizeof(*params));
        if (!params) {
            virReportOOMError();
            goto cleanup;
        }
        if (virDomainGetInterfaceParameters(dom, device, params, &nparams, flags) != 0) {
            vshError(ctl, "%s", _("Unable to get interface parameters"));
            goto cleanup;
        }

        for (i = 0; i < nparams; i++) {
            char *str = vshGetTypedParamValue(ctl, &params[i]);
            vshPrint(ctl, "%-15s: %s\n", params[i].field, str);
            VIR_FREE(str);
        }
    } else {
        /* set the interface parameters */
        params = vshCalloc(ctl, nparams, sizeof(*params));
        if (!params) {
            virReportOOMError();
            goto cleanup;
        }

        for (i = 0; i < nparams; i++)
            params[i].type = VIR_TYPED_PARAM_UINT;

        i = 0;
        if (inbound.average && i < nparams) {
            if (!virStrcpy(params[i].field, VIR_DOMAIN_BANDWIDTH_IN_AVERAGE,
                           sizeof(params[i].field)))
                goto cleanup;
            params[i].value.ui = inbound.average;
            i++;
        }
        if (inbound.peak && i < nparams) {
            if (!virStrcpy(params[i].field, VIR_DOMAIN_BANDWIDTH_IN_PEAK,
                           sizeof(params[i].field)))
                goto cleanup;
            params[i].value.ui = inbound.peak;
            i++;
        }
        if (inbound.burst && i < nparams) {
            if (!virStrcpy(params[i].field, VIR_DOMAIN_BANDWIDTH_IN_BURST,
                           sizeof(params[i].field)))
                goto cleanup;
            params[i].value.ui = inbound.burst;
            i++;
        }
        if (outbound.average && i < nparams) {
            if (!virStrcpy(params[i].field, VIR_DOMAIN_BANDWIDTH_OUT_AVERAGE,
                           sizeof(params[i].field)))
                goto cleanup;
            params[i].value.ui = outbound.average;
            i++;
        }
        if (outbound.peak && i < nparams) {
            if (!virStrcpy(params[i].field, VIR_DOMAIN_BANDWIDTH_OUT_PEAK,
                           sizeof(params[i].field)))
                goto cleanup;
            params[i].value.ui = outbound.peak;
            i++;
        }
        if (outbound.burst && i < nparams) {
            if (!virStrcpy(params[i].field, VIR_DOMAIN_BANDWIDTH_OUT_BURST,
                           sizeof(params[i].field)))
                goto cleanup;
            params[i].value.ui = outbound.burst;
            i++;
        }

        if (virDomainSetInterfaceParameters(dom, device, params, nparams, flags) != 0) {
            vshError(ctl, "%s", _("Unable to set interface parameters"));
            goto cleanup;
        }
    }

    ret = true;

cleanup:
    virTypedParameterArrayClear(params, nparams);
    VIR_FREE(params);
    virDomainFree(dom);
    return ret;
}

2109 2110 2111
/*
 * "dommemstats" command
 */
2112
static const vshCmdInfo info_dommemstat[] = {
2113
    {"help", N_("get memory statistics for a domain")},
Y
Yuri Chornoivan 已提交
2114
    {"desc", N_("Get memory statistics for a running domain.")},
2115 2116 2117
    {NULL,NULL}
};

2118
static const vshCmdOptDef opts_dommemstat[] = {
2119
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
2120 2121 2122
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
2123
static bool
2124
cmdDomMemStat(vshControl *ctl, const vshCmd *cmd)
2125 2126
{
    virDomainPtr dom;
2127
    const char *name;
2128 2129 2130
    struct _virDomainMemoryStat stats[VIR_DOMAIN_MEMORY_STAT_NR];
    unsigned int nr_stats, i;

2131
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
2132
        return false;
2133 2134

    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
E
Eric Blake 已提交
2135
        return false;
2136 2137 2138 2139 2140

    nr_stats = virDomainMemoryStats (dom, stats, VIR_DOMAIN_MEMORY_STAT_NR, 0);
    if (nr_stats == -1) {
        vshError(ctl, _("Failed to get memory statistics for domain %s"), name);
        virDomainFree(dom);
E
Eric Blake 已提交
2141
        return false;
2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156
    }

    for (i = 0; i < nr_stats; i++) {
        if (stats[i].tag == VIR_DOMAIN_MEMORY_STAT_SWAP_IN)
            vshPrint (ctl, "swap_in %llu\n", stats[i].val);
        if (stats[i].tag == VIR_DOMAIN_MEMORY_STAT_SWAP_OUT)
            vshPrint (ctl, "swap_out %llu\n", stats[i].val);
        if (stats[i].tag == VIR_DOMAIN_MEMORY_STAT_MAJOR_FAULT)
            vshPrint (ctl, "major_fault %llu\n", stats[i].val);
        if (stats[i].tag == VIR_DOMAIN_MEMORY_STAT_MINOR_FAULT)
            vshPrint (ctl, "minor_fault %llu\n", stats[i].val);
        if (stats[i].tag == VIR_DOMAIN_MEMORY_STAT_UNUSED)
            vshPrint (ctl, "unused %llu\n", stats[i].val);
        if (stats[i].tag == VIR_DOMAIN_MEMORY_STAT_AVAILABLE)
            vshPrint (ctl, "available %llu\n", stats[i].val);
2157 2158
        if (stats[i].tag == VIR_DOMAIN_MEMORY_STAT_ACTUAL_BALLOON)
            vshPrint (ctl, "actual %llu\n", stats[i].val);
M
Martin Kletzander 已提交
2159 2160
        if (stats[i].tag == VIR_DOMAIN_MEMORY_STAT_RSS)
            vshPrint (ctl, "rss %llu\n", stats[i].val);
2161 2162 2163
    }

    virDomainFree(dom);
E
Eric Blake 已提交
2164
    return true;
2165 2166
}

2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181
/*
 * "domblkinfo" command
 */
static const vshCmdInfo info_domblkinfo[] = {
    {"help", N_("domain block device size information")},
    {"desc", N_("Get block device size info for a domain.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_domblkinfo[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"device", VSH_OT_DATA, VSH_OFLAG_REQ, N_("block device")},
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
2182
static bool
2183 2184 2185 2186
cmdDomblkinfo(vshControl *ctl, const vshCmd *cmd)
{
    virDomainBlockInfo info;
    virDomainPtr dom;
E
Eric Blake 已提交
2187
    bool ret = true;
2188
    const char *device = NULL;
2189

2190
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
2191
        return false;
2192 2193

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
2194
        return false;
2195

2196
    if (vshCommandOptString (cmd, "device", &device) <= 0) {
2197
        virDomainFree(dom);
E
Eric Blake 已提交
2198
        return false;
2199 2200 2201 2202
    }

    if (virDomainGetBlockInfo(dom, device, &info, 0) < 0) {
        virDomainFree(dom);
E
Eric Blake 已提交
2203
        return false;
2204 2205 2206 2207 2208 2209 2210 2211 2212 2213
    }

    vshPrint(ctl, "%-15s %llu\n", _("Capacity:"), info.capacity);
    vshPrint(ctl, "%-15s %llu\n", _("Allocation:"), info.allocation);
    vshPrint(ctl, "%-15s %llu\n", _("Physical:"), info.physical);

    virDomainFree(dom);
    return ret;
}

2214 2215 2216 2217 2218
/*
 * "domblklist" command
 */
static const vshCmdInfo info_domblklist[] = {
    {"help", N_("list all domain blocks")},
2219
    {"desc", N_("Get the summary of block devices for a domain.")},
2220 2221 2222 2223 2224 2225 2226
    {NULL, NULL}
};

static const vshCmdOptDef opts_domblklist[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"inactive", VSH_OT_BOOL, 0,
     N_("get inactive rather than running configuration")},
2227 2228
    {"details", VSH_OT_BOOL, 0,
     N_("additionally display the type and device value")},
2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243
    {NULL, 0, 0, NULL}
};

static bool
cmdDomblklist(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    bool ret = false;
    unsigned int flags = 0;
    char *xml = NULL;
    xmlDocPtr xmldoc = NULL;
    xmlXPathContextPtr ctxt = NULL;
    int ndisks;
    xmlNodePtr *disks = NULL;
    int i;
2244
    bool details = false;
2245 2246 2247 2248

    if (vshCommandOptBool(cmd, "inactive"))
        flags |= VIR_DOMAIN_XML_INACTIVE;

2249 2250
    details = vshCommandOptBool(cmd, "details");

2251 2252 2253 2254 2255 2256 2257 2258 2259 2260
    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

    xml = virDomainGetXMLDesc(dom, flags);
    if (!xml)
        goto cleanup;

2261
    xmldoc = virXMLParseStringCtxt(xml, _("(domain_definition)"), &ctxt);
2262 2263 2264 2265 2266 2267 2268
    if (!xmldoc)
        goto cleanup;

    ndisks = virXPathNodeSet("./devices/disk", ctxt, &disks);
    if (ndisks < 0)
        goto cleanup;

2269 2270 2271 2272 2273 2274
    if (details)
        vshPrint(ctl, "%-10s %-10s %-10s %s\n", _("Type"),
                 _("Device"), _("Target"), _("Source"));
    else
        vshPrint(ctl, "%-10s %s\n", _("Target"), _("Source"));

2275 2276 2277
    vshPrint(ctl, "------------------------------------------------\n");

    for (i = 0; i < ndisks; i++) {
2278 2279
        char *type;
        char *device;
2280 2281 2282 2283
        char *target;
        char *source;

        ctxt->node = disks[i];
2284 2285 2286 2287 2288 2289

        if (details) {
            type = virXPathString("string(./@type)", ctxt);
            device = virXPathString("string(./@device)", ctxt);
        }

2290 2291 2292 2293 2294 2295 2296 2297 2298
        target = virXPathString("string(./target/@dev)", ctxt);
        if (!target) {
            vshError(ctl, "unable to query block list");
            goto cleanup;
        }
        source = virXPathString("string(./source/@file"
                                "|./source/@dev"
                                "|./source/@dir"
                                "|./source/@name)", ctxt);
2299 2300 2301 2302 2303 2304 2305 2306 2307
        if (details) {
            vshPrint(ctl, "%-10s %-10s %-10s %s\n", type, device,
                     target, source ? source : "-");
            VIR_FREE(type);
            VIR_FREE(device);
        } else {
            vshPrint(ctl, "%-10s %s\n", target, source ? source : "-");
        }

2308 2309 2310 2311
        VIR_FREE(target);
        VIR_FREE(source);
    }

A
Alex Jia 已提交
2312
    ret = true;
2313 2314 2315 2316

cleanup:
    VIR_FREE(disks);
    virDomainFree(dom);
2317 2318 2319
    VIR_FREE(xml);
    xmlFreeDoc(xmldoc);
    xmlXPathFreeContext(ctxt);
2320 2321 2322
    return ret;
}

2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372
/*
 * "domiflist" command
 */
static const vshCmdInfo info_domiflist[] = {
    {"help", N_("list all domain virtual interfaces")},
    {"desc", N_("Get the summary of virtual interfaces for a domain.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_domiflist[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"inactive", VSH_OT_BOOL, 0,
     N_("get inactive rather than running configuration")},
    {NULL, 0, 0, NULL}
};

static bool
cmdDomiflist(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    bool ret = false;
    unsigned int flags = 0;
    char *xml = NULL;
    xmlDocPtr xmldoc = NULL;
    xmlXPathContextPtr ctxt = NULL;
    int ninterfaces;
    xmlNodePtr *interfaces = NULL;
    int i;

    if (vshCommandOptBool(cmd, "inactive"))
        flags |= VIR_DOMAIN_XML_INACTIVE;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

    xml = virDomainGetXMLDesc(dom, flags);
    if (!xml)
        goto cleanup;

    xmldoc = virXMLParseStringCtxt(xml, _("(domain_definition)"), &ctxt);
    if (!xmldoc)
        goto cleanup;

    ninterfaces = virXPathNodeSet("./devices/interface", ctxt, &interfaces);
    if (ninterfaces < 0)
        goto cleanup;

T
Taku Izumi 已提交
2373 2374
    vshPrint(ctl, "%-10s %-10s %-10s %-11s %s\n", _("Interface"), _("Type"),
             _("Source"), _("Model"), _("MAC"));
2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395
    vshPrint(ctl, "-------------------------------------------------------\n");

    for (i = 0; i < ninterfaces; i++) {
        char *type = NULL;
        char *source = NULL;
        char *target = NULL;
        char *model = NULL;
        char *mac = NULL;

        ctxt->node = interfaces[i];
        type = virXPathString("string(./@type)", ctxt);

        source = virXPathString("string(./source/@bridge"
                                "|./source/@dev"
                                "|./source/@network"
                                "|./source/@name)", ctxt);

        target = virXPathString("string(./target/@dev)", ctxt);
        model = virXPathString("string(./model/@type)", ctxt);
        mac = virXPathString("string(./mac/@address)", ctxt);

T
Taku Izumi 已提交
2396
        vshPrint(ctl, "%-10s %-10s %-10s %-11s %-10s\n",
2397
                 target ? target : "-",
T
Taku Izumi 已提交
2398 2399
                 type,
                 source ? source : "-",
2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420
                 model ? model : "-",
                 mac ? mac : "-");

        VIR_FREE(type);
        VIR_FREE(source);
        VIR_FREE(target);
        VIR_FREE(model);
        VIR_FREE(mac);
    }

    ret = true;

cleanup:
    VIR_FREE(interfaces);
    virDomainFree(dom);
    VIR_FREE(xml);
    xmlFreeDoc(xmldoc);
    xmlXPathFreeContext(ctxt);
    return ret;
}

2421 2422 2423
/*
 * "suspend" command
 */
2424
static const vshCmdInfo info_suspend[] = {
2425 2426
    {"help", N_("suspend a domain")},
    {"desc", N_("Suspend a running domain.")},
2427
    {NULL, NULL}
2428 2429
};

2430
static const vshCmdOptDef opts_suspend[] = {
2431
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
2432
    {NULL, 0, 0, NULL}
2433 2434
};

E
Eric Blake 已提交
2435
static bool
2436
cmdSuspend(vshControl *ctl, const vshCmd *cmd)
2437
{
2438
    virDomainPtr dom;
2439
    const char *name;
E
Eric Blake 已提交
2440
    bool ret = true;
2441

2442
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
2443
        return false;
2444

J
Jim Meyering 已提交
2445
    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
E
Eric Blake 已提交
2446
        return false;
2447 2448

    if (virDomainSuspend(dom) == 0) {
2449
        vshPrint(ctl, _("Domain %s suspended\n"), name);
2450
    } else {
2451
        vshError(ctl, _("Failed to suspend domain %s"), name);
E
Eric Blake 已提交
2452
        ret = false;
2453
    }
2454

2455 2456 2457 2458
    virDomainFree(dom);
    return ret;
}

2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529
/*
 * "dompmsuspend" command
 */
static const vshCmdInfo info_dom_pm_suspend[] = {
    {"help", N_("suspend a domain for a given time duration")},
    {"desc", N_("Suspend a running domain for a given time duration.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_dom_pm_suspend[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"duration", VSH_OT_INT, VSH_OFLAG_REQ_OPT, N_("duration in seconds")},
    {"target", VSH_OT_STRING, VSH_OFLAG_REQ, N_("mem(Suspend-to-RAM), "
                                                "disk(Suspend-to-Disk), "
                                                "hybrid(Hybrid-Suspend)")},
    {NULL, 0, 0, NULL}
};

static bool
cmdDomPMSuspend(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    const char *name;
    bool ret = false;
    const char *target = NULL;
    unsigned int suspendTarget;
    unsigned long long duration = 0;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
        return false;

    if (vshCommandOptULongLong(cmd, "duration", &duration) < 0) {
        vshError(ctl, _("Invalid duration argument"));
        goto cleanup;
    }

    if (vshCommandOptString(cmd, "target", &target) < 0) {
        vshError(ctl, _("Invalid target argument"));
        goto cleanup;
    }

    if (STREQ(target, "mem"))
        suspendTarget = VIR_NODE_SUSPEND_TARGET_MEM;
    else if (STREQ(target, "disk"))
        suspendTarget = VIR_NODE_SUSPEND_TARGET_DISK;
    else if (STREQ(target, "hybrid"))
        suspendTarget = VIR_NODE_SUSPEND_TARGET_HYBRID;
    else {
        vshError(ctl, "%s", _("Invalid target"));
        goto cleanup;
    }

    if (virDomainPMSuspendForDuration(dom, suspendTarget, duration, 0) < 0) {
        vshError(ctl, _("Domain %s could not be suspended"),
                 virDomainGetName(dom));
        goto cleanup;
    }

    vshPrint(ctl, _("Domain %s successfully suspended"),
             virDomainGetName(dom));

    ret = true;

cleanup:
    virDomainFree(dom);
    return ret;
}

2530 2531 2532
/*
 * "create" command
 */
2533
static const vshCmdInfo info_create[] = {
2534 2535
    {"help", N_("create a domain from an XML file")},
    {"desc", N_("Create a domain.")},
2536 2537 2538
    {NULL, NULL}
};

2539
static const vshCmdOptDef opts_create[] = {
2540
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file containing an XML domain description")},
2541
#ifndef WIN32
2542
    {"console", VSH_OT_BOOL, 0, N_("attach to console after creation")},
2543
#endif
2544
    {"paused", VSH_OT_BOOL, 0, N_("leave the guest paused after creation")},
2545
    {"autodestroy", VSH_OT_BOOL, 0, N_("automatically destroy the guest when virsh disconnects")},
2546 2547 2548
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
2549
static bool
2550
cmdCreate(vshControl *ctl, const vshCmd *cmd)
2551 2552
{
    virDomainPtr dom;
2553
    const char *from = NULL;
E
Eric Blake 已提交
2554
    bool ret = true;
2555
    char *buffer;
2556
#ifndef WIN32
2557
    int console = vshCommandOptBool(cmd, "console");
2558
#endif
2559
    unsigned int flags = VIR_DOMAIN_NONE;
2560

2561
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
2562
        return false;
2563

2564
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
2565
        return false;
2566

2567
    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0)
E
Eric Blake 已提交
2568
        return false;
2569

2570 2571
    if (vshCommandOptBool(cmd, "paused"))
        flags |= VIR_DOMAIN_START_PAUSED;
2572 2573
    if (vshCommandOptBool(cmd, "autodestroy"))
        flags |= VIR_DOMAIN_START_AUTODESTROY;
2574 2575

    dom = virDomainCreateXML(ctl->conn, buffer, flags);
2576
    VIR_FREE(buffer);
2577

2578
    if (dom != NULL) {
2579
        vshPrint(ctl, _("Domain %s created from %s\n"),
2580
                 virDomainGetName(dom), from);
2581
#ifndef WIN32
2582
        if (console)
2583
            cmdRunConsole(ctl, dom, NULL);
2584
#endif
2585
        virDomainFree(dom);
2586
    } else {
2587
        vshError(ctl, _("Failed to create domain from %s"), from);
E
Eric Blake 已提交
2588
        ret = false;
2589 2590 2591 2592
    }
    return ret;
}

2593 2594 2595
/*
 * "define" command
 */
2596
static const vshCmdInfo info_define[] = {
2597 2598
    {"help", N_("define (but don't start) a domain from an XML file")},
    {"desc", N_("Define a domain.")},
2599 2600 2601
    {NULL, NULL}
};

2602
static const vshCmdOptDef opts_define[] = {
2603
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file containing an XML domain description")},
2604 2605 2606
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
2607
static bool
2608
cmdDefine(vshControl *ctl, const vshCmd *cmd)
2609 2610
{
    virDomainPtr dom;
2611
    const char *from = NULL;
E
Eric Blake 已提交
2612
    bool ret = true;
2613
    char *buffer;
2614

2615
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
2616
        return false;
2617

2618
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
2619
        return false;
2620

2621
    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0)
E
Eric Blake 已提交
2622
        return false;
2623 2624

    dom = virDomainDefineXML(ctl->conn, buffer);
2625
    VIR_FREE(buffer);
2626

2627
    if (dom != NULL) {
2628
        vshPrint(ctl, _("Domain %s defined from %s\n"),
2629
                 virDomainGetName(dom), from);
2630
        virDomainFree(dom);
2631
    } else {
2632
        vshError(ctl, _("Failed to define domain from %s"), from);
E
Eric Blake 已提交
2633
        ret = false;
2634 2635 2636 2637 2638 2639 2640
    }
    return ret;
}

/*
 * "undefine" command
 */
2641
static const vshCmdInfo info_undefine[] = {
2642 2643 2644
    {"help", N_("undefine a domain")},
    {"desc",
     N_("Undefine an inactive domain, or convert persistent to transient.")},
2645 2646 2647
    {NULL, NULL}
};

2648
static const vshCmdOptDef opts_undefine[] = {
2649
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name or uuid")},
2650
    {"managed-save", VSH_OT_BOOL, 0, N_("remove domain managed state file")},
2651 2652 2653 2654 2655 2656 2657
    {"storage", VSH_OT_DATA, VSH_OFLAG_NONE,
     N_("remove associated storage volumes (comma separated list of targets "
        "or source paths) (see domblklist)")},
    {"remove-all-storage", VSH_OT_BOOL, 0,
     N_("remove all associated storage volumes (use with caution)")},
    {"wipe-storage", VSH_OT_BOOL, VSH_OFLAG_NONE,
     N_("wipe data on the removed volumes")},
2658 2659
    {"snapshots-metadata", VSH_OT_BOOL, 0,
     N_("remove all domain snapshot metadata, if inactive")},
2660 2661 2662
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
2663
static bool
2664
cmdUndefine(vshControl *ctl, const vshCmd *cmd)
2665 2666
{
    virDomainPtr dom;
2667
    bool ret = false;
2668
    const char *name = NULL;
2669
    /* Flags to attempt.  */
E
Eric Blake 已提交
2670
    unsigned int flags = 0;
2671 2672 2673
    /* User-requested actions.  */
    bool managed_save = vshCommandOptBool(cmd, "managed-save");
    bool snapshots_metadata = vshCommandOptBool(cmd, "snapshots-metadata");
2674 2675 2676
    bool wipe_storage = vshCommandOptBool(cmd, "wipe-storage");
    bool remove_storage = false;
    bool remove_all_storage = vshCommandOptBool(cmd, "remove-all-storage");
2677
    /* Positive if these items exist.  */
2678
    int has_managed_save = 0;
2679 2680 2681 2682 2683
    int has_snapshots_metadata = 0;
    int has_snapshots = 0;
    /* True if undefine will not strand data, even on older servers.  */
    bool managed_save_safe = false;
    bool snapshots_safe = false;
2684
    int rc = -1;
2685
    int running;
2686 2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702
    /* list of volumes to remove along with this domain */
    const char *volumes_arg = NULL;
    char *volumes = NULL;
    char **volume_tokens = NULL;
    char *volume_tok = NULL;
    int nvolume_tokens = 0;
    char *def = NULL;
    char *source = NULL;
    char *target = NULL;
    int vol_i;
    int tok_i;
    xmlDocPtr doc = NULL;
    xmlXPathContextPtr ctxt = NULL;
    xmlNodePtr *vol_nodes = NULL;
    int nvolumes = 0;
    virStorageVolPtr vol = NULL;
    bool vol_del_failed = false;
2703

2704
    if (managed_save) {
2705
        flags |= VIR_DOMAIN_UNDEFINE_MANAGED_SAVE;
2706 2707 2708 2709 2710 2711
        managed_save_safe = true;
    }
    if (snapshots_metadata) {
        flags |= VIR_DOMAIN_UNDEFINE_SNAPSHOTS_METADATA;
        snapshots_safe = true;
    }
2712

2713
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
2714
        return false;
2715

2716
    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
E
Eric Blake 已提交
2717
        return false;
2718

2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729
    /* check if a string that should contain list of volumes to remove is present */
    if (vshCommandOptString(cmd, "storage", &volumes_arg) > 0) {
        volumes = vshStrdup(ctl, volumes_arg);

        if (remove_all_storage) {
            vshError(ctl, _("Specified both --storage and --remove-all-storage"));
            goto cleanup;
        }
        remove_storage = true;
   }

2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748
    /* Do some flag manipulation.  The goal here is to disable bits
     * from flags to reduce the likelihood of a server rejecting
     * unknown flag bits, as well as to track conditions which are
     * safe by default for the given hypervisor and server version.  */
    running = virDomainIsActive(dom);
    if (running < 0) {
        virshReportError(ctl);
        goto cleanup;
    }
    if (!running) {
        /* Undefine with snapshots only fails for inactive domains,
         * and managed save only exists on inactive domains; if
         * running, then we don't want to remove anything.  */
        has_managed_save = virDomainHasManagedSaveImage(dom, 0);
        if (has_managed_save < 0) {
            if (last_error->code != VIR_ERR_NO_SUPPORT) {
                virshReportError(ctl);
                goto cleanup;
            }
2749 2750
            virFreeError(last_error);
            last_error = NULL;
2751
            has_managed_save = 0;
2752 2753
        }

2754 2755
        has_snapshots = virDomainSnapshotNum(dom, 0);
        if (has_snapshots < 0) {
2756 2757
            if (last_error->code != VIR_ERR_NO_SUPPORT) {
                virshReportError(ctl);
2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769
                goto cleanup;
            }
            virFreeError(last_error);
            last_error = NULL;
            has_snapshots = 0;
        }
        if (has_snapshots) {
            has_snapshots_metadata
                = virDomainSnapshotNum(dom, VIR_DOMAIN_SNAPSHOT_LIST_METADATA);
            if (has_snapshots_metadata < 0) {
                /* The server did not know the new flag, assume that all
                   snapshots have metadata.  */
2770 2771
                virFreeError(last_error);
                last_error = NULL;
2772 2773 2774 2775 2776
                has_snapshots_metadata = has_snapshots;
            } else {
                /* The server knew the new flag, all aspects of
                 * undefineFlags are safe.  */
                managed_save_safe = snapshots_safe = true;
2777
            }
2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790
        }
    }
    if (!has_managed_save) {
        flags &= ~VIR_DOMAIN_UNDEFINE_MANAGED_SAVE;
        managed_save_safe = true;
    }
    if (has_snapshots == 0) {
        snapshots_safe = true;
    }
    if (has_snapshots_metadata == 0) {
        flags &= ~VIR_DOMAIN_UNDEFINE_SNAPSHOTS_METADATA;
        snapshots_safe = true;
    }
2791

2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804
    /* Stash domain description for later use */
    if (remove_storage || remove_all_storage) {
        if (running) {
            vshError(ctl, _("Storage volume deletion is supported only on stopped domains"));
            goto cleanup;
        }

        if (!(def = virDomainGetXMLDesc(dom, 0))) {
            vshError(ctl, _("Could not retrieve domain XML description"));
            goto cleanup;
        }
    }

2805 2806 2807 2808 2809 2810 2811 2812 2813 2814 2815 2816 2817 2818
    /* Generally we want to try the new API first.  However, while
     * virDomainUndefineFlags was introduced at the same time as
     * VIR_DOMAIN_UNDEFINE_MANAGED_SAVE in 0.9.4, the
     * VIR_DOMAIN_UNDEFINE_SNAPSHOTS_METADATA flag was not present
     * until 0.9.5; skip to piecewise emulation if we couldn't prove
     * above that the new API is safe.  */
    if (managed_save_safe && snapshots_safe) {
        rc = virDomainUndefineFlags(dom, flags);
        if (rc == 0 || (last_error->code != VIR_ERR_NO_SUPPORT &&
                        last_error->code != VIR_ERR_INVALID_ARG))
            goto out;
        virFreeError(last_error);
        last_error = NULL;
    }
2819

2820 2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831
    /* The new API is unsupported or unsafe; fall back to doing things
     * piecewise.  */
    if (has_managed_save) {
        if (!managed_save) {
            vshError(ctl, "%s",
                     _("Refusing to undefine while domain managed save "
                       "image exists"));
            goto cleanup;
        }
        if (virDomainManagedSaveRemove(dom, 0) < 0) {
            virshReportError(ctl);
            goto cleanup;
2832 2833 2834
        }
    }

2835 2836 2837 2838 2839 2840 2841 2842 2843 2844 2845 2846 2847 2848
    /* No way to emulate deletion of just snapshot metadata
     * without support for the newer flags.  Oh well.  */
    if (has_snapshots_metadata) {
        vshError(ctl,
                 snapshots_metadata ?
                 _("Unable to remove metadata of %d snapshots") :
                 _("Refusing to undefine while %d snapshots exist"),
                 has_snapshots_metadata);
        goto cleanup;
    }

    rc = virDomainUndefine(dom);

out:
2849
    if (rc == 0) {
2850
        vshPrint(ctl, _("Domain %s has been undefined\n"), name);
2851
        ret = true;
2852
    } else {
2853
        vshError(ctl, _("Failed to undefine domain %s"), name);
2854 2855 2856 2857 2858 2859 2860 2861 2862 2863 2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958 2959 2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974
        goto cleanup;
    }

    /* try to undefine storage volumes associated with this domain, if it's requested */
    if (remove_storage || remove_all_storage) {
        ret = false;

        /* tokenize the string from user and save it's parts into an array */
        if (volumes) {
            /* count the delimiters */
            volume_tok = volumes;
            nvolume_tokens = 1; /* we need at least one member */
            while (*volume_tok) {
                if (*volume_tok == ',')
                    nvolume_tokens++;
                volume_tok++;
            }

            volume_tokens = vshCalloc(ctl, nvolume_tokens,  sizeof(char *));

            /* tokenize the input string */
            nvolume_tokens = 0;
            volume_tok = volumes;
            do {
                volume_tokens[nvolume_tokens] = strsep(&volume_tok, ",");
                nvolume_tokens++;
            } while (volume_tok);
        }

        doc = virXMLParseStringCtxt(def, _("(domain_definition)"), &ctxt);
        if (!doc)
            goto cleanup;

        nvolumes = virXPathNodeSet("./devices/disk", ctxt, &vol_nodes);

        if (nvolumes < 0)
            goto cleanup;

        for (vol_i = 0; vol_i < nvolumes; vol_i++) {
            ctxt->node = vol_nodes[vol_i];
            VIR_FREE(target);
            VIR_FREE(source);
            if (vol) {
                virStorageVolFree(vol);
                vol = NULL;
            }

            /* get volume source and target paths */
            if (!(target = virXPathString("string(./target/@dev)", ctxt))) {
                vshError(ctl, _("Failed to enumerate devices"));
                goto cleanup;
            }

            if (!(source = virXPathString("string("
                                          "./source/@file|"
                                          "./source/@dir|"
                                          "./source/@name|"
                                          "./source/@dev)", ctxt)) &&
                virGetLastError())
                goto cleanup;

            /* lookup if volume was selected by user */
            if (volumes) {
                volume_tok = NULL;
                for (tok_i = 0; tok_i < nvolume_tokens; tok_i++) {
                    if (volume_tokens[tok_i] &&
                        (STREQ_NULLABLE(volume_tokens[tok_i], target) ||
                         STREQ_NULLABLE(volume_tokens[tok_i], source))) {
                        volume_tok = volume_tokens[tok_i];
                        volume_tokens[tok_i] = NULL;
                        break;
                    }
                }
                if (!volume_tok)
                    continue;
            }

            if (!source)
                continue;

            if (!(vol = virStorageVolLookupByPath(ctl->conn, source))) {
                vshPrint(ctl,
                         _("Storage volume '%s'(%s) is not managed by libvirt. "
                           "Remove it manually.\n"), target, source);
                virResetLastError();
                continue;
            }

            if (wipe_storage) {
                vshPrint(ctl, _("Wiping volume '%s'(%s) ... "), target, source);
                fflush(stdout);
                if (virStorageVolWipe(vol, 0) < 0) {
                    vshError(ctl, _("Failed! Volume not removed."));
                    vol_del_failed = true;
                    continue;
                } else {
                    vshPrint(ctl, _("Done.\n"));
                }
            }

            /* delete the volume */
            if (virStorageVolDelete(vol, 0) < 0) {
                vshError(ctl, _("Failed to remove storage volume '%s'(%s)"),
                         target, source);
                vol_del_failed = true;
            }
            vshPrint(ctl, _("Volume '%s' removed.\n"), volume_tok?volume_tok:source);
        }

        /* print volumes specified by user that were not found in domain definition */
        if (volumes) {
            for (tok_i = 0; tok_i < nvolume_tokens; tok_i++) {
                if (volume_tokens[tok_i])
                    vshPrint(ctl, _("Volume '%s' was not found in domain's "
                                    "definition.\n"),
                             volume_tokens[tok_i]);
            }
        }

        if (!vol_del_failed)
            ret = true;
2975 2976
    }

2977
cleanup:
2978 2979 2980 2981 2982 2983
    VIR_FREE(source);
    VIR_FREE(target);
    VIR_FREE(volumes);
    VIR_FREE(volume_tokens);
    VIR_FREE(def);
    VIR_FREE(vol_nodes);
2984 2985
    if (vol)
        virStorageVolFree(vol);
2986 2987
    xmlFreeDoc(doc);
    xmlXPathFreeContext(ctxt);
2988
    virDomainFree(dom);
2989 2990 2991 2992 2993 2994 2995
    return ret;
}


/*
 * "start" command
 */
2996
static const vshCmdInfo info_start[] = {
2997
    {"help", N_("start a (previously defined) inactive domain")},
2998 2999 3000
    {"desc", N_("Start a domain, either from the last managedsave\n"
                "    state, or via a fresh boot if no managedsave state\n"
                "    is present.")},
3001 3002 3003
    {NULL, NULL}
};

3004
static const vshCmdOptDef opts_start[] = {
3005
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("name of the inactive domain")},
3006
#ifndef WIN32
3007
    {"console", VSH_OT_BOOL, 0, N_("attach to console after creation")},
3008
#endif
E
Eric Blake 已提交
3009
    {"paused", VSH_OT_BOOL, 0, N_("leave the guest paused after creation")},
3010 3011
    {"autodestroy", VSH_OT_BOOL, 0,
     N_("automatically destroy the guest when virsh disconnects")},
3012 3013
    {"bypass-cache", VSH_OT_BOOL, 0,
     N_("avoid file system cache when loading")},
3014 3015
    {"force-boot", VSH_OT_BOOL, 0,
     N_("force fresh boot by discarding any managed save")},
3016 3017 3018
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
3019
static bool
3020
cmdStart(vshControl *ctl, const vshCmd *cmd)
3021 3022
{
    virDomainPtr dom;
3023
    bool ret = false;
3024
#ifndef WIN32
3025
    int console = vshCommandOptBool(cmd, "console");
3026
#endif
E
Eric Blake 已提交
3027
    unsigned int flags = VIR_DOMAIN_NONE;
3028
    int rc;
3029

3030
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
3031
        return false;
3032

3033 3034
    if (!(dom = vshCommandOptDomainBy(ctl, cmd, NULL,
                                      VSH_BYNAME | VSH_BYUUID)))
E
Eric Blake 已提交
3035
        return false;
3036 3037

    if (virDomainGetID(dom) != (unsigned int)-1) {
3038
        vshError(ctl, "%s", _("Domain is already active"));
3039
        virDomainFree(dom);
E
Eric Blake 已提交
3040
        return false;
3041 3042
    }

E
Eric Blake 已提交
3043 3044
    if (vshCommandOptBool(cmd, "paused"))
        flags |= VIR_DOMAIN_START_PAUSED;
3045 3046
    if (vshCommandOptBool(cmd, "autodestroy"))
        flags |= VIR_DOMAIN_START_AUTODESTROY;
3047 3048
    if (vshCommandOptBool(cmd, "bypass-cache"))
        flags |= VIR_DOMAIN_START_BYPASS_CACHE;
3049 3050
    if (vshCommandOptBool(cmd, "force-boot"))
        flags |= VIR_DOMAIN_START_FORCE_BOOT;
E
Eric Blake 已提交
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
    /* We can emulate force boot, even for older servers that reject it.  */
    if (flags & VIR_DOMAIN_START_FORCE_BOOT) {
        if (virDomainCreateWithFlags(dom, flags) == 0)
            goto started;
        if (last_error->code != VIR_ERR_NO_SUPPORT &&
            last_error->code != VIR_ERR_INVALID_ARG) {
            virshReportError(ctl);
            goto cleanup;
        }
        virFreeError(last_error);
        last_error = NULL;
        rc = virDomainHasManagedSaveImage(dom, 0);
        if (rc < 0) {
            /* No managed save image to remove */
            virFreeError(last_error);
            last_error = NULL;
        } else if (rc > 0) {
            if (virDomainManagedSaveRemove(dom, 0) < 0) {
                virshReportError(ctl);
                goto cleanup;
            }
        }
        flags &= ~VIR_DOMAIN_START_FORCE_BOOT;
    }

E
Eric Blake 已提交
3077 3078
    /* Prefer older API unless we have to pass a flag.  */
    if ((flags ? virDomainCreateWithFlags(dom, flags)
3079
         : virDomainCreate(dom)) < 0) {
3080
        vshError(ctl, _("Failed to start domain %s"), virDomainGetName(dom));
3081
        goto cleanup;
3082
    }
3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094

started:
    vshPrint(ctl, _("Domain %s started\n"),
             virDomainGetName(dom));
#ifndef WIN32
    if (console && !cmdRunConsole(ctl, dom, NULL))
        goto cleanup;
#endif

    ret = true;

cleanup:
3095
    virDomainFree(dom);
3096 3097 3098
    return ret;
}

3099 3100 3101
/*
 * "save" command
 */
3102
static const vshCmdInfo info_save[] = {
3103
    {"help", N_("save a domain state to a file")},
E
Eric Blake 已提交
3104
    {"desc", N_("Save the RAM state of a running domain.")},
3105
    {NULL, NULL}
3106 3107
};

3108
static const vshCmdOptDef opts_save[] = {
3109
    {"bypass-cache", VSH_OT_BOOL, 0, N_("avoid file system cache when saving")},
3110 3111
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("where to save the data")},
3112 3113
    {"xml", VSH_OT_STRING, 0,
     N_("filename containing updated XML for the target")},
3114 3115
    {"running", VSH_OT_BOOL, 0, N_("set domain to be running on restore")},
    {"paused", VSH_OT_BOOL, 0, N_("set domain to be paused on restore")},
3116
    {"verbose", VSH_OT_BOOL, 0, N_("display the progress of save")},
3117
    {NULL, 0, 0, NULL}
3118 3119
};

3120 3121
static void
doSave(void *opaque)
3122
{
3123 3124 3125 3126 3127
    vshCtrlData *data = opaque;
    vshControl *ctl = data->ctl;
    const vshCmd *cmd = data->cmd;
    char ret = '1';
    virDomainPtr dom = NULL;
3128 3129
    const char *name = NULL;
    const char *to = NULL;
E
Eric Blake 已提交
3130
    unsigned int flags = 0;
3131 3132
    const char *xmlfile = NULL;
    char *xml = NULL;
3133 3134 3135 3136 3137 3138
    sigset_t sigmask, oldsigmask;

    sigemptyset(&sigmask);
    sigaddset(&sigmask, SIGINT);
    if (pthread_sigmask(SIG_BLOCK, &sigmask, &oldsigmask) < 0)
        goto out_sig;
3139

3140
    if (!vshConnectionUsability(ctl, ctl->conn))
3141
        goto out;
3142

3143
    if (vshCommandOptString(cmd, "file", &to) <= 0)
3144
        goto out;
3145

3146 3147
    if (vshCommandOptBool(cmd, "bypass-cache"))
        flags |= VIR_DOMAIN_SAVE_BYPASS_CACHE;
3148 3149 3150 3151
    if (vshCommandOptBool(cmd, "running"))
        flags |= VIR_DOMAIN_SAVE_RUNNING;
    if (vshCommandOptBool(cmd, "paused"))
        flags |= VIR_DOMAIN_SAVE_PAUSED;
3152

3153 3154
    if (vshCommandOptString(cmd, "xml", &xmlfile) < 0) {
        vshError(ctl, "%s", _("malformed xml argument"));
3155
        goto out;
3156 3157
    }

J
Jim Meyering 已提交
3158
    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
3159
        goto out;
3160

3161 3162
    if (xmlfile &&
        virFileReadAll(xmlfile, 8192, &xml) < 0)
3163
        goto out;
3164 3165 3166

    if (((flags || xml)
         ? virDomainSaveFlags(dom, to, xml, flags)
3167
         : virDomainSave(dom, to)) < 0) {
3168
        vshError(ctl, _("Failed to save domain %s to %s"), name, to);
3169
        goto out;
3170
    }
3171

3172
    ret = '0';
3173

3174 3175 3176 3177
out:
    pthread_sigmask(SIG_SETMASK, &oldsigmask, NULL);
out_sig:
    if (dom) virDomainFree (dom);
3178
    VIR_FREE(xml);
3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225
    ignore_value(safewrite(data->writefd, &ret, sizeof(ret)));
}

static bool
cmdSave(vshControl *ctl, const vshCmd *cmd)
{
    bool ret = false;
    virDomainPtr dom = NULL;
    int p[2] = {-1. -1};
    virThread workerThread;
    bool verbose = false;
    vshCtrlData data;
    const char *to = NULL;
    const char *name = NULL;

    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
        return false;

    if (vshCommandOptString(cmd, "file", &to) <= 0)
        goto cleanup;

    if (vshCommandOptBool (cmd, "verbose"))
        verbose = true;

    if (pipe(p) < 0)
        goto cleanup;

    data.ctl = ctl;
    data.cmd = cmd;
    data.writefd = p[1];

    if (virThreadCreate(&workerThread,
                        true,
                        doSave,
                        &data) < 0)
        goto cleanup;

    ret = vshWatchJob(ctl, dom, verbose, p[0], 0, NULL, NULL, _("Save"));

    virThreadJoin(&workerThread);

    if (ret)
        vshPrint(ctl, _("\nDomain %s saved to %s\n"), name, to);

cleanup:
    if (dom)
        virDomainFree(dom);
3226 3227 3228
    return ret;
}

3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249
/*
 * "save-image-dumpxml" command
 */
static const vshCmdInfo info_save_image_dumpxml[] = {
    {"help", N_("saved state domain information in XML")},
    {"desc", N_("Output the domain information for a saved state file,\n"
                "as an XML dump to stdout.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_save_image_dumpxml[] = {
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("saved state file to read")},
    {"security-info", VSH_OT_BOOL, 0, N_("include security sensitive information in XML dump")},
    {NULL, 0, 0, NULL}
};

static bool
cmdSaveImageDumpxml(vshControl *ctl, const vshCmd *cmd)
{
    const char *file = NULL;
    bool ret = false;
E
Eric Blake 已提交
3250
    unsigned int flags = 0;
3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283 3284 3285 3286
    char *xml = NULL;

    if (vshCommandOptBool(cmd, "security-info"))
        flags |= VIR_DOMAIN_XML_SECURE;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (vshCommandOptString(cmd, "file", &file) <= 0)
        return false;

    xml = virDomainSaveImageGetXMLDesc(ctl->conn, file, flags);
    if (!xml)
        goto cleanup;

    vshPrint(ctl, "%s", xml);
    ret = true;

cleanup:
    VIR_FREE(xml);
    return ret;
}

/*
 * "save-image-define" command
 */
static const vshCmdInfo info_save_image_define[] = {
    {"help", N_("redefine the XML for a domain's saved state file")},
    {"desc", N_("Replace the domain XML associated with a saved state file")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_save_image_define[] = {
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("saved state file to modify")},
    {"xml", VSH_OT_STRING, VSH_OFLAG_REQ,
     N_("filename containing updated XML for the target")},
3287 3288
    {"running", VSH_OT_BOOL, 0, N_("set domain to be running on restore")},
    {"paused", VSH_OT_BOOL, 0, N_("set domain to be paused on restore")},
3289 3290 3291 3292 3293 3294 3295 3296 3297 3298
    {NULL, 0, 0, NULL}
};

static bool
cmdSaveImageDefine(vshControl *ctl, const vshCmd *cmd)
{
    const char *file = NULL;
    bool ret = false;
    const char *xmlfile = NULL;
    char *xml = NULL;
3299 3300 3301 3302 3303 3304
    unsigned int flags = 0;

    if (vshCommandOptBool(cmd, "running"))
        flags |= VIR_DOMAIN_SAVE_RUNNING;
    if (vshCommandOptBool(cmd, "paused"))
        flags |= VIR_DOMAIN_SAVE_PAUSED;
3305 3306 3307 3308 3309 3310 3311 3312 3313 3314 3315 3316 3317 3318 3319

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (vshCommandOptString(cmd, "file", &file) <= 0)
        return false;

    if (vshCommandOptString(cmd, "xml", &xmlfile) <= 0) {
        vshError(ctl, "%s", _("malformed or missing xml argument"));
        return false;
    }

    if (virFileReadAll(xmlfile, 8192, &xml) < 0)
        goto cleanup;

3320
    if (virDomainSaveImageDefineXML(ctl->conn, file, xml, flags) < 0) {
3321 3322 3323 3324 3325 3326 3327 3328 3329 3330 3331 3332 3333 3334 3335 3336 3337 3338 3339 3340 3341 3342 3343
        vshError(ctl, _("Failed to update %s"), file);
        goto cleanup;
    }

    vshPrint(ctl, _("State file %s updated.\n"), file);
    ret = true;

cleanup:
    VIR_FREE(xml);
    return ret;
}

/*
 * "save-image-edit" command
 */
static const vshCmdInfo info_save_image_edit[] = {
    {"help", N_("edit XML for a domain's saved state file")},
    {"desc", N_("Edit the domain XML associated with a saved state file")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_save_image_edit[] = {
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("saved state file to edit")},
3344 3345
    {"running", VSH_OT_BOOL, 0, N_("set domain to be running on restore")},
    {"paused", VSH_OT_BOOL, 0, N_("set domain to be paused on restore")},
3346 3347 3348 3349 3350 3351 3352 3353 3354 3355 3356
    {NULL, 0, 0, NULL}
};

static bool
cmdSaveImageEdit(vshControl *ctl, const vshCmd *cmd)
{
    const char *file = NULL;
    bool ret = false;
    char *tmp = NULL;
    char *doc = NULL;
    char *doc_edited = NULL;
3357 3358 3359 3360 3361 3362 3363 3364 3365 3366 3367 3368 3369 3370 3371 3372
    unsigned int getxml_flags = VIR_DOMAIN_XML_SECURE;
    unsigned int define_flags = 0;

    if (vshCommandOptBool(cmd, "running"))
        define_flags |= VIR_DOMAIN_SAVE_RUNNING;
    if (vshCommandOptBool(cmd, "paused"))
        define_flags |= VIR_DOMAIN_SAVE_PAUSED;

    /* Normally, we let the API reject mutually exclusive flags.
     * However, in the edit cycle, we let the user retry if the define
     * step fails, but the define step will always fail on invalid
     * flags, so we reject it up front to avoid looping.  */
    if (define_flags == (VIR_DOMAIN_SAVE_RUNNING | VIR_DOMAIN_SAVE_PAUSED)) {
        vshError(ctl, "%s", _("--running and --saved are mutually exclusive"));
        return false;
    }
3373 3374 3375 3376 3377 3378 3379 3380

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (vshCommandOptString(cmd, "file", &file) <= 0)
        return false;

    /* Get the XML configuration of the saved image.  */
3381
    doc = virDomainSaveImageGetXMLDesc(ctl->conn, file, getxml_flags);
3382 3383 3384 3385 3386 3387 3388 3389 3390 3391 3392 3393 3394 3395 3396 3397 3398
    if (!doc)
        goto cleanup;

    /* Create and open the temporary file.  */
    tmp = editWriteToTempFile(ctl, doc);
    if (!tmp)
        goto cleanup;

    /* Start the editor.  */
    if (editFile(ctl, tmp) == -1)
        goto cleanup;

    /* Read back the edited file.  */
    doc_edited = editReadBackFile(ctl, tmp);
    if (!doc_edited)
        goto cleanup;

3399 3400 3401
    /* Compare original XML with edited.  Short-circuit if it did not
     * change, and we do not have any flags.  */
    if (STREQ(doc, doc_edited) && !define_flags) {
3402 3403 3404 3405 3406 3407 3408
        vshPrint(ctl, _("Saved image %s XML configuration not changed.\n"),
                 file);
        ret = true;
        goto cleanup;
    }

    /* Everything checks out, so redefine the xml.  */
3409 3410
    if (virDomainSaveImageDefineXML(ctl->conn, file, doc_edited,
                                    define_flags) < 0) {
3411 3412 3413 3414 3415 3416 3417 3418 3419 3420 3421 3422 3423 3424 3425 3426 3427
        vshError(ctl, _("Failed to update %s"), file);
        goto cleanup;
    }

    vshPrint(ctl, _("State file %s edited.\n"), file);
    ret = true;

cleanup:
    VIR_FREE(doc);
    VIR_FREE(doc_edited);
    if (tmp) {
        unlink(tmp);
        VIR_FREE(tmp);
    }
    return ret;
}

3428 3429 3430 3431 3432
/*
 * "managedsave" command
 */
static const vshCmdInfo info_managedsave[] = {
    {"help", N_("managed save of a domain state")},
3433 3434 3435 3436
    {"desc", N_("Save and destroy a running domain, so it can be restarted from\n"
                "    the same state at a later time.  When the virsh 'start'\n"
                "    command is next run for the domain, it will automatically\n"
                "    be started from this saved state.")},
3437 3438 3439 3440
    {NULL, NULL}
};

static const vshCmdOptDef opts_managedsave[] = {
3441
    {"bypass-cache", VSH_OT_BOOL, 0, N_("avoid file system cache when saving")},
3442
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
3443 3444
    {"running", VSH_OT_BOOL, 0, N_("set domain to be running on next start")},
    {"paused", VSH_OT_BOOL, 0, N_("set domain to be paused on next start")},
3445
    {"verbose", VSH_OT_BOOL, 0, N_("display the progress of save")},
3446 3447 3448
    {NULL, 0, 0, NULL}
};

3449 3450
static void
doManagedsave(void *opaque)
3451
{
3452 3453 3454 3455 3456
    char ret = '1';
    vshCtrlData *data = opaque;
    vshControl *ctl = data->ctl;
    const vshCmd *cmd = data->cmd;
    virDomainPtr dom = NULL;
3457
    const char *name;
E
Eric Blake 已提交
3458
    unsigned int flags = 0;
3459 3460 3461 3462 3463 3464
    sigset_t sigmask, oldsigmask;

    sigemptyset(&sigmask);
    sigaddset(&sigmask, SIGINT);
    if (pthread_sigmask(SIG_BLOCK, &sigmask, &oldsigmask) < 0)
        goto out_sig;
3465

3466
    if (!vshConnectionUsability(ctl, ctl->conn))
3467
        goto out;
3468

3469 3470
    if (vshCommandOptBool(cmd, "bypass-cache"))
        flags |= VIR_DOMAIN_SAVE_BYPASS_CACHE;
3471 3472 3473 3474
    if (vshCommandOptBool(cmd, "running"))
        flags |= VIR_DOMAIN_SAVE_RUNNING;
    if (vshCommandOptBool(cmd, "paused"))
        flags |= VIR_DOMAIN_SAVE_PAUSED;
3475

3476
    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
3477
        goto out;
3478

3479
    if (virDomainManagedSave(dom, flags) < 0) {
3480
        vshError(ctl, _("Failed to save domain %s state"), name);
3481
        goto out;
3482 3483
    }

3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502 3503 3504 3505 3506 3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529
    ret = '0';
out:
    pthread_sigmask(SIG_SETMASK, &oldsigmask, NULL);
out_sig:
    if (dom)
        virDomainFree (dom);
    ignore_value(safewrite(data->writefd, &ret, sizeof(ret)));
}

static bool
cmdManagedSave(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    int p[2] = { -1, -1};
    bool ret = false;
    bool verbose = false;
    const char *name = NULL;
    vshCtrlData data;
    virThread workerThread;

    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
        return false;

    if (vshCommandOptBool (cmd, "verbose"))
        verbose = true;

    if (pipe(p) < 0)
        goto cleanup;

    data.ctl = ctl;
    data.cmd = cmd;
    data.writefd = p[1];

    if (virThreadCreate(&workerThread,
                        true,
                        doManagedsave,
                        &data) < 0)
        goto cleanup;

    ret = vshWatchJob(ctl, dom, verbose, p[0], 0,
                      NULL, NULL, _("Managedsave"));

    virThreadJoin(&workerThread);

    if (ret)
        vshPrint(ctl, _("\nDomain %s state saved by libvirt\n"), name);
3530 3531

cleanup:
3532
    virDomainFree(dom);
3533 3534
    VIR_FORCE_CLOSE(p[0]);
    VIR_FORCE_CLOSE(p[1]);
3535 3536 3537
    return ret;
}

3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551
/*
 * "managedsave-remove" command
 */
static const vshCmdInfo info_managedsaveremove[] = {
    {"help", N_("Remove managed save of a domain")},
    {"desc", N_("Remove an existing managed save state file from a domain")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_managedsaveremove[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
3552
static bool
3553 3554 3555
cmdManagedSaveRemove(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
3556
    const char *name;
E
Eric Blake 已提交
3557
    bool ret = false;
3558 3559
    int hassave;

3560
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
3561
        return false;
3562 3563

    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
E
Eric Blake 已提交
3564
        return false;
3565 3566 3567

    hassave = virDomainHasManagedSaveImage(dom, 0);
    if (hassave < 0) {
3568
        vshError(ctl, "%s", _("Failed to check for domain managed save image"));
3569 3570 3571 3572 3573 3574 3575 3576 3577 3578 3579 3580 3581 3582 3583 3584
        goto cleanup;
    }

    if (hassave) {
        if (virDomainManagedSaveRemove(dom, 0) < 0) {
            vshError(ctl, _("Failed to remove managed save image for domain %s"),
                     name);
            goto cleanup;
        }
        else
            vshPrint(ctl, _("Removed managedsave image for domain %s"), name);
    }
    else
        vshPrint(ctl, _("Domain %s has no manage save image; removal skipped"),
                 name);

E
Eric Blake 已提交
3585
    ret = true;
3586 3587 3588 3589 3590 3591

cleanup:
    virDomainFree(dom);
    return ret;
}

3592 3593 3594
/*
 * "schedinfo" command
 */
3595
static const vshCmdInfo info_schedinfo[] = {
3596 3597
    {"help", N_("show/set scheduler parameters")},
    {"desc", N_("Show/Set scheduler parameters.")},
3598 3599 3600
    {NULL, NULL}
};

3601
static const vshCmdOptDef opts_schedinfo[] = {
3602 3603 3604 3605
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"set", VSH_OT_STRING, VSH_OFLAG_NONE, N_("parameter=value")},
    {"weight", VSH_OT_INT, VSH_OFLAG_NONE, N_("weight for XEN_CREDIT")},
    {"cap", VSH_OT_INT, VSH_OFLAG_NONE, N_("cap for XEN_CREDIT")},
3606 3607 3608
    {"current", VSH_OT_BOOL, 0, N_("get/set current scheduler info")},
    {"config", VSH_OT_BOOL, 0, N_("get/set value to be used on next boot")},
    {"live", VSH_OT_BOOL, 0, N_("get/set value from running domain")},
3609 3610 3611
    {NULL, 0, 0, NULL}
};

W
Wen Congyang 已提交
3612
static int
3613
cmdSchedInfoUpdate(vshControl *ctl, const vshCmd *cmd,
3614
                   virTypedParameterPtr param)
3615
{
3616
    const char *data = NULL;
3617 3618 3619

    /* Legacy 'weight' parameter */
    if (STREQ(param->field, "weight") &&
3620
        param->type == VIR_TYPED_PARAM_UINT &&
3621 3622
        vshCommandOptBool(cmd, "weight")) {
        int val;
3623
        if (vshCommandOptInt(cmd, "weight", &val) <= 0) {
3624
            vshError(ctl, "%s", _("Invalid value of weight"));
3625
            return -1;
3626
        } else {
3627
            param->value.ui = val;
3628
        }
3629
        return 1;
3630 3631
    }

3632 3633
    /* Legacy 'cap' parameter */
    if (STREQ(param->field, "cap") &&
3634
        param->type == VIR_TYPED_PARAM_UINT &&
3635 3636
        vshCommandOptBool(cmd, "cap")) {
        int val;
3637
        if (vshCommandOptInt(cmd, "cap", &val) <= 0) {
3638
            vshError(ctl, "%s", _("Invalid value of cap"));
3639
            return -1;
3640
        } else {
3641
            param->value.ui = val;
3642
        }
3643
        return 1;
3644
    }
3645

3646
    if (vshCommandOptString(cmd, "set", &data) > 0) {
3647 3648 3649
        char *val = strchr(data, '=');
        int match = 0;
        if (!val) {
3650
            vshError(ctl, "%s", _("Invalid syntax for --set, expecting name=value"));
3651
            return -1;
3652
        }
3653 3654 3655 3656 3657 3658 3659 3660 3661
        *val = '\0';
        match = STREQ(data, param->field);
        *val = '=';
        val++;

        if (!match)
            return 0;

        switch (param->type) {
3662
        case VIR_TYPED_PARAM_INT:
3663
            if (virStrToLong_i(val, NULL, 10, &param->value.i) < 0) {
3664
                vshError(ctl, "%s",
3665 3666 3667 3668
                         _("Invalid value for parameter, expecting an int"));
                return -1;
            }
            break;
3669
        case VIR_TYPED_PARAM_UINT:
3670
            if (virStrToLong_ui(val, NULL, 10, &param->value.ui) < 0) {
3671
                vshError(ctl, "%s",
3672 3673 3674 3675
                         _("Invalid value for parameter, expecting an unsigned int"));
                return -1;
            }
            break;
3676
        case VIR_TYPED_PARAM_LLONG:
3677
            if (virStrToLong_ll(val, NULL, 10, &param->value.l) < 0) {
3678
                vshError(ctl, "%s",
J
Jim Meyering 已提交
3679
                         _("Invalid value for parameter, expecting a long long"));
3680 3681 3682
                return -1;
            }
            break;
3683
        case VIR_TYPED_PARAM_ULLONG:
3684
            if (virStrToLong_ull(val, NULL, 10, &param->value.ul) < 0) {
3685
                vshError(ctl, "%s",
3686 3687 3688 3689
                         _("Invalid value for parameter, expecting an unsigned long long"));
                return -1;
            }
            break;
3690
        case VIR_TYPED_PARAM_DOUBLE:
3691
            if (virStrToDouble(val, NULL, &param->value.d) < 0) {
3692
                vshError(ctl, "%s", _("Invalid value for parameter, expecting a double"));
3693 3694 3695
                return -1;
            }
            break;
3696
        case VIR_TYPED_PARAM_BOOLEAN:
3697
            param->value.b = STREQ(val, "0") ? 0 : 1;
3698
        }
3699
        return 1;
3700
    }
3701

3702 3703
    return 0;
}
3704

3705

E
Eric Blake 已提交
3706
static bool
3707 3708 3709 3710
cmdSchedinfo(vshControl *ctl, const vshCmd *cmd)
{
    char *schedulertype;
    virDomainPtr dom;
3711
    virTypedParameterPtr params = NULL;
3712 3713 3714
    int nparams = 0;
    int update = 0;
    int i, ret;
W
Wen Congyang 已提交
3715
    bool ret_val = false;
3716 3717 3718 3719 3720 3721 3722 3723 3724 3725
    unsigned int flags = 0;
    int current = vshCommandOptBool(cmd, "current");
    int config = vshCommandOptBool(cmd, "config");
    int live = vshCommandOptBool(cmd, "live");

    if (current) {
        if (live || config) {
            vshError(ctl, "%s", _("--current must be specified exclusively"));
            return false;
        }
3726
        flags = VIR_DOMAIN_AFFECT_CURRENT;
3727 3728
    } else {
        if (config)
3729
            flags |= VIR_DOMAIN_AFFECT_CONFIG;
3730
        if (live)
3731
            flags |= VIR_DOMAIN_AFFECT_LIVE;
3732
    }
3733

3734
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
3735
        return false;
3736

3737
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
3738
        return false;
3739 3740 3741

    /* Print SchedulerType */
    schedulertype = virDomainGetSchedulerType(dom, &nparams);
3742
    if (schedulertype != NULL) {
3743
        vshPrint(ctl, "%-15s: %s\n", _("Scheduler"),
3744
             schedulertype);
3745
        VIR_FREE(schedulertype);
3746
    } else {
3747
        vshPrint(ctl, "%-15s: %s\n", _("Scheduler"), _("Unknown"));
3748
        goto cleanup;
3749 3750
    }

3751
    if (nparams) {
3752
        params = vshMalloc(ctl, sizeof(*params) * nparams);
3753

3754
        memset(params, 0, sizeof(*params) * nparams);
3755 3756 3757 3758 3759 3760 3761 3762 3763 3764 3765
        if (flags || current) {
            /* We cannot query both live and config at once, so settle
               on current in that case.  If we are setting, then the
               two values should match when we re-query; otherwise, we
               report the error later.  */
            ret = virDomainGetSchedulerParametersFlags(dom, params, &nparams,
                                                       ((live && config) ? 0
                                                        : flags));
        } else {
            ret = virDomainGetSchedulerParameters(dom, params, &nparams);
        }
3766 3767 3768 3769
        if (ret == -1)
            goto cleanup;

        /* See if any params are being set */
3770
        for (i = 0; i < nparams; i++) {
3771 3772 3773 3774 3775 3776 3777 3778 3779 3780
            ret = cmdSchedInfoUpdate(ctl, cmd, &(params[i]));
            if (ret == -1)
                goto cleanup;

            if (ret == 1)
                update = 1;
        }

        /* Update parameters & refresh data */
        if (update) {
3781 3782 3783 3784 3785
            if (flags || current)
                ret = virDomainSetSchedulerParametersFlags(dom, params,
                                                           nparams, flags);
            else
                ret = virDomainSetSchedulerParameters(dom, params, nparams);
3786 3787 3788
            if (ret == -1)
                goto cleanup;

3789 3790 3791 3792 3793 3794 3795
            if (flags || current)
                ret = virDomainGetSchedulerParametersFlags(dom, params,
                                                           &nparams,
                                                           ((live && config) ? 0
                                                            : flags));
            else
                ret = virDomainGetSchedulerParameters(dom, params, &nparams);
3796 3797
            if (ret == -1)
                goto cleanup;
3798 3799 3800 3801
        } else {
            /* See if we've tried to --set var=val.  If so, the fact that
               we reach this point (with update == 0) means that "var" did
               not match any of the settable parameters.  Report the error.  */
3802 3803
            const char *var_value_pair = NULL;
            if (vshCommandOptString(cmd, "set", &var_value_pair) > 0) {
3804 3805 3806 3807
                vshError(ctl, _("invalid scheduler option: %s"),
                         var_value_pair);
                goto cleanup;
            }
3808 3809 3810 3811 3812 3813
            /* When not doing --set, --live and --config do not mix.  */
            if (live && config) {
                vshError(ctl, "%s",
                         _("cannot query both live and config at once"));
                goto cleanup;
            }
3814 3815
        }

E
Eric Blake 已提交
3816
        ret_val = true;
3817
        for (i = 0; i < nparams; i++) {
3818 3819 3820
            char *str = vshGetTypedParamValue(ctl, &params[i]);
            vshPrint(ctl, "%-15s: %s\n", params[i].field, str);
            VIR_FREE(str);
3821 3822
        }
    }
3823

3824
 cleanup:
3825
    VIR_FREE(params);
3826
    virDomainFree(dom);
3827
    return ret_val;
3828 3829
}

3830 3831 3832
/*
 * "restore" command
 */
3833
static const vshCmdInfo info_restore[] = {
3834 3835
    {"help", N_("restore a domain from a saved state in a file")},
    {"desc", N_("Restore a domain.")},
3836
    {NULL, NULL}
3837 3838
};

3839
static const vshCmdOptDef opts_restore[] = {
3840
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("the state to restore")},
3841 3842
    {"bypass-cache", VSH_OT_BOOL, 0,
     N_("avoid file system cache when restoring")},
3843 3844
    {"xml", VSH_OT_STRING, 0,
     N_("filename containing updated XML for the target")},
3845 3846
    {"running", VSH_OT_BOOL, 0, N_("restore domain into running state")},
    {"paused", VSH_OT_BOOL, 0, N_("restore domain into paused state")},
3847
    {NULL, 0, 0, NULL}
3848 3849
};

E
Eric Blake 已提交
3850
static bool
3851
cmdRestore(vshControl *ctl, const vshCmd *cmd)
3852
{
3853
    const char *from = NULL;
3854
    bool ret = false;
E
Eric Blake 已提交
3855
    unsigned int flags = 0;
3856 3857
    const char *xmlfile = NULL;
    char *xml = NULL;
3858

3859
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
3860
        return false;
3861

3862
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
3863
        return false;
3864

3865 3866
    if (vshCommandOptBool(cmd, "bypass-cache"))
        flags |= VIR_DOMAIN_SAVE_BYPASS_CACHE;
3867 3868 3869 3870
    if (vshCommandOptBool(cmd, "running"))
        flags |= VIR_DOMAIN_SAVE_RUNNING;
    if (vshCommandOptBool(cmd, "paused"))
        flags |= VIR_DOMAIN_SAVE_PAUSED;
3871

3872 3873 3874 3875 3876 3877 3878 3879 3880 3881 3882
    if (vshCommandOptString(cmd, "xml", &xmlfile) < 0) {
        vshError(ctl, "%s", _("malformed xml argument"));
        return false;
    }

    if (xmlfile &&
        virFileReadAll(xmlfile, 8192, &xml) < 0)
        goto cleanup;

    if (((flags || xml)
         ? virDomainRestoreFlags(ctl->conn, from, xml, flags)
3883
         : virDomainRestore(ctl->conn, from)) < 0) {
3884
        vshError(ctl, _("Failed to restore domain from %s"), from);
3885
        goto cleanup;
3886
    }
3887 3888 3889 3890 3891

    vshPrint(ctl, _("Domain restored from %s\n"), from);
    ret = true;

cleanup:
3892
    VIR_FREE(xml);
3893 3894 3895
    return ret;
}

D
Daniel Veillard 已提交
3896 3897 3898
/*
 * "dump" command
 */
3899
static const vshCmdInfo info_dump[] = {
3900 3901
    {"help", N_("dump the core of a domain to a file for analysis")},
    {"desc", N_("Core dump a domain.")},
D
Daniel Veillard 已提交
3902 3903 3904
    {NULL, NULL}
};

3905
static const vshCmdOptDef opts_dump[] = {
3906 3907
    {"live", VSH_OT_BOOL, 0, N_("perform a live core dump if supported")},
    {"crash", VSH_OT_BOOL, 0, N_("crash the domain after core dump")},
3908 3909
    {"bypass-cache", VSH_OT_BOOL, 0,
     N_("avoid file system cache when saving")},
3910
    {"reset", VSH_OT_BOOL, 0, N_("reset the domain after core dump")},
3911 3912
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("where to dump the core")},
3913
    {"verbose", VSH_OT_BOOL, 0, N_("display the progress of dump")},
D
Daniel Veillard 已提交
3914 3915 3916
    {NULL, 0, 0, NULL}
};

3917 3918
static void
doDump(void *opaque)
D
Daniel Veillard 已提交
3919
{
3920 3921 3922 3923 3924 3925
    char ret = '1';
    vshCtrlData *data = opaque;
    vshControl *ctl = data->ctl;
    const vshCmd *cmd = data->cmd;
    virDomainPtr dom = NULL;
    sigset_t sigmask, oldsigmask;
3926 3927
    const char *name = NULL;
    const char *to = NULL;
E
Eric Blake 已提交
3928
    unsigned int flags = 0;
D
Daniel Veillard 已提交
3929

3930 3931 3932 3933 3934
    sigemptyset(&sigmask);
    sigaddset(&sigmask, SIGINT);
    if (pthread_sigmask(SIG_BLOCK, &sigmask, &oldsigmask) < 0)
        goto out_sig;

3935
    if (!vshConnectionUsability(ctl, ctl->conn))
3936
        goto out;
D
Daniel Veillard 已提交
3937

3938
    if (vshCommandOptString(cmd, "file", &to) <= 0)
3939
        goto out;
D
Daniel Veillard 已提交
3940

J
Jim Meyering 已提交
3941
    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
3942
        goto out;
D
Daniel Veillard 已提交
3943

P
Paolo Bonzini 已提交
3944 3945
    if (vshCommandOptBool (cmd, "live"))
        flags |= VIR_DUMP_LIVE;
3946 3947
    if (vshCommandOptBool (cmd, "crash"))
        flags |= VIR_DUMP_CRASH;
3948 3949
    if (vshCommandOptBool(cmd, "bypass-cache"))
        flags |= VIR_DUMP_BYPASS_CACHE;
3950 3951
    if (vshCommandOptBool(cmd, "reset"))
        flags |= VIR_DUMP_RESET;
3952

3953
    if (virDomainCoreDump(dom, to, flags) < 0) {
3954
        vshError(ctl, _("Failed to core dump domain %s to %s"), name, to);
3955
        goto out;
D
Daniel Veillard 已提交
3956 3957
    }

3958 3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978 3979 3980 3981 3982 3983 3984 3985 3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997 3998 3999 4000 4001 4002 4003 4004 4005 4006
    ret = '0';
out:
    pthread_sigmask(SIG_SETMASK, &oldsigmask, NULL);
out_sig:
    if (dom)
        virDomainFree (dom);
    ignore_value(safewrite(data->writefd, &ret, sizeof(ret)));
}

static bool
cmdDump(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    int p[2] = { -1, -1};
    bool ret = false;
    bool verbose = false;
    const char *name = NULL;
    const char *to = NULL;
    vshCtrlData data;
    virThread workerThread;

    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
        return false;

    if (vshCommandOptString(cmd, "file", &to) <= 0)
        return false;

    if (vshCommandOptBool (cmd, "verbose"))
        verbose = true;

    if (pipe(p) < 0)
        goto cleanup;

    data.ctl = ctl;
    data.cmd = cmd;
    data.writefd = p[1];

    if (virThreadCreate(&workerThread,
                        true,
                        doDump,
                        &data) < 0)
        goto cleanup;

    ret = vshWatchJob(ctl, dom, verbose, p[0], 0, NULL, NULL, _("Dump"));

    virThreadJoin(&workerThread);

    if (ret)
        vshPrint(ctl, _("\nDomain %s dumped to %s\n"), name, to);
4007 4008

cleanup:
D
Daniel Veillard 已提交
4009
    virDomainFree(dom);
4010 4011
    VIR_FORCE_CLOSE(p[0]);
    VIR_FORCE_CLOSE(p[1]);
D
Daniel Veillard 已提交
4012 4013 4014
    return ret;
}

4015 4016 4017 4018 4019 4020 4021 4022 4023 4024 4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039 4040 4041 4042 4043 4044 4045 4046 4047 4048 4049 4050 4051 4052 4053 4054 4055 4056 4057 4058 4059 4060 4061 4062 4063 4064 4065 4066 4067 4068 4069 4070 4071 4072 4073 4074 4075 4076 4077 4078 4079 4080 4081 4082 4083 4084 4085 4086 4087 4088
static const vshCmdInfo info_screenshot[] = {
    {"help", N_("take a screenshot of a current domain console and store it "
                "into a file")},
    {"desc", N_("screenshot of a current domain console")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_screenshot[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"file", VSH_OT_DATA, VSH_OFLAG_NONE, N_("where to store the screenshot")},
    {"screen", VSH_OT_INT, VSH_OFLAG_NONE, N_("ID of a screen to take screenshot of")},
    {NULL, 0, 0, NULL}
};

static int vshStreamSink(virStreamPtr st ATTRIBUTE_UNUSED,
                         const char *bytes, size_t nbytes, void *opaque)
{
    int *fd = opaque;

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

/**
 * Generate string: '<domain name>-<timestamp>[<extension>]'
 */
static char *
vshGenFileName(vshControl *ctl, virDomainPtr dom, const char *mime)
{
    char timestr[100];
    struct timeval cur_time;
    struct tm time_info;
    const char *ext = NULL;
    char *ret = NULL;

    /* We should be already connected, but doesn't
     * hurt to check */
    if (!vshConnectionUsability(ctl, ctl->conn))
        return NULL;

    if (!dom) {
        vshError(ctl, "%s", _("Invalid domain supplied"));
        return NULL;
    }

    if (STREQ(mime, "image/x-portable-pixmap"))
        ext = ".ppm";
    else if (STREQ(mime, "image/png"))
        ext = ".png";
    /* add mime type here */

    gettimeofday(&cur_time, NULL);
    localtime_r(&cur_time.tv_sec, &time_info);
    strftime(timestr, sizeof(timestr), "%Y-%m-%d-%H:%M:%S", &time_info);

    if (virAsprintf(&ret, "%s-%s%s", virDomainGetName(dom),
                    timestr, ext ? ext : "") < 0) {
        vshError(ctl, "%s", _("Out of memory"));
        return NULL;
    }

    return ret;
}

static bool
cmdScreenshot(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    const char *name = NULL;
    char *file = NULL;
    int fd = -1;
    virStreamPtr st = NULL;
    unsigned int screen = 0;
    unsigned int flags = 0; /* currently unused */
    int ret = false;
4089
    bool created = false;
4090 4091 4092 4093 4094 4095 4096 4097 4098 4099 4100 4101 4102 4103 4104 4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119 4120 4121 4122 4123 4124 4125 4126 4127 4128
    bool generated = false;
    char *mime = NULL;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (vshCommandOptString(cmd, "file", (const char **) &file) < 0) {
        vshError(ctl, "%s", _("file must not be empty"));
        return false;
    }

    if (vshCommandOptUInt(cmd, "screen", &screen) < 0) {
        vshError(ctl, "%s", _("invalid screen ID"));
        return false;
    }

    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
        return false;

    st = virStreamNew(ctl->conn, 0);

    mime = virDomainScreenshot(dom, st, screen, flags);
    if (!mime) {
        vshError(ctl, _("could not take a screenshot of %s"), name);
        goto cleanup;
    }

    if (!file) {
        if (!(file=vshGenFileName(ctl, dom, mime)))
            return false;
        generated = true;
    }

    if ((fd = open(file, O_WRONLY|O_CREAT|O_EXCL, 0666)) < 0) {
        if (errno != EEXIST ||
            (fd = open(file, O_WRONLY|O_TRUNC, 0666)) < 0) {
            vshError(ctl, _("cannot create file %s"), file);
            goto cleanup;
        }
4129 4130
    } else {
        created = true;
4131 4132 4133 4134 4135 4136 4137 4138 4139 4140 4141 4142 4143 4144 4145 4146 4147 4148 4149 4150 4151 4152 4153 4154 4155 4156 4157 4158 4159
    }

    if (virStreamRecvAll(st, vshStreamSink, &fd) < 0) {
        vshError(ctl, _("could not receive data from domain %s"), name);
        goto cleanup;
    }

    if (VIR_CLOSE(fd) < 0) {
        vshError(ctl, _("cannot close file %s"), file);
        goto cleanup;
    }

    if (virStreamFinish(st) < 0) {
        vshError(ctl, _("cannot close stream on domain %s"), name);
        goto cleanup;
    }

    vshPrint(ctl, _("Screenshot saved to %s, with type of %s"), file, mime);
    ret = true;

cleanup:
    if (!ret && created)
        unlink(file);
    if (generated)
        VIR_FREE(file);
    virDomainFree(dom);
    if (st)
        virStreamFree(st);
    VIR_FORCE_CLOSE(fd);
4160
    VIR_FREE(mime);
4161 4162 4163
    return ret;
}

4164 4165 4166
/*
 * "resume" command
 */
4167
static const vshCmdInfo info_resume[] = {
4168 4169
    {"help", N_("resume a domain")},
    {"desc", N_("Resume a previously suspended domain.")},
4170
    {NULL, NULL}
4171 4172
};

4173
static const vshCmdOptDef opts_resume[] = {
4174
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
4175
    {NULL, 0, 0, NULL}
4176 4177
};

E
Eric Blake 已提交
4178
static bool
4179
cmdResume(vshControl *ctl, const vshCmd *cmd)
4180
{
4181
    virDomainPtr dom;
E
Eric Blake 已提交
4182
    bool ret = true;
4183
    const char *name;
4184

4185
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
4186
        return false;
4187

J
Jim Meyering 已提交
4188
    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
E
Eric Blake 已提交
4189
        return false;
4190 4191

    if (virDomainResume(dom) == 0) {
4192
        vshPrint(ctl, _("Domain %s resumed\n"), name);
4193
    } else {
4194
        vshError(ctl, _("Failed to resume domain %s"), name);
E
Eric Blake 已提交
4195
        ret = false;
4196
    }
4197

4198 4199 4200 4201
    virDomainFree(dom);
    return ret;
}

4202 4203 4204
/*
 * "shutdown" command
 */
4205
static const vshCmdInfo info_shutdown[] = {
4206 4207
    {"help", N_("gracefully shutdown a domain")},
    {"desc", N_("Run shutdown in the target domain.")},
4208
    {NULL, NULL}
4209 4210
};

4211
static const vshCmdOptDef opts_shutdown[] = {
4212
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
4213
    {"mode", VSH_OT_STRING, VSH_OFLAG_NONE, N_("shutdown mode: acpi|agent")},
4214
    {NULL, 0, 0, NULL}
4215 4216
};

E
Eric Blake 已提交
4217
static bool
4218
cmdShutdown(vshControl *ctl, const vshCmd *cmd)
4219
{
4220
    virDomainPtr dom;
E
Eric Blake 已提交
4221
    bool ret = true;
4222
    const char *name;
4223 4224 4225
    const char *mode = NULL;
    int flags = 0;
    int rv;
4226

4227
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
4228
        return false;
4229

4230 4231 4232 4233 4234 4235 4236 4237 4238 4239 4240 4241 4242 4243 4244 4245
    if (vshCommandOptString(cmd, "mode", &mode) < 0) {
        vshError(ctl, "%s", _("Invalid type"));
        return false;
    }

    if (mode) {
        if (STREQ(mode, "acpi")) {
            flags |= VIR_DOMAIN_SHUTDOWN_ACPI_POWER_BTN;
        } else if (STREQ(mode, "agent")) {
            flags |= VIR_DOMAIN_SHUTDOWN_GUEST_AGENT;
        } else {
            vshError(ctl, _("Unknown mode %s value, expecting 'acpi' or 'agent'"), mode);
            return false;
        }
    }

J
Jim Meyering 已提交
4246
    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
E
Eric Blake 已提交
4247
        return false;
4248

4249 4250 4251 4252 4253
    if (flags)
        rv = virDomainShutdownFlags(dom, flags);
    else
        rv = virDomainShutdown(dom);
    if (rv == 0) {
4254
        vshPrint(ctl, _("Domain %s is being shutdown\n"), name);
4255
    } else {
4256
        vshError(ctl, _("Failed to shutdown domain %s"), name);
E
Eric Blake 已提交
4257
        ret = false;
4258
    }
4259

4260 4261 4262 4263
    virDomainFree(dom);
    return ret;
}

4264 4265 4266
/*
 * "reboot" command
 */
4267
static const vshCmdInfo info_reboot[] = {
4268 4269
    {"help", N_("reboot a domain")},
    {"desc", N_("Run a reboot command in the target domain.")},
4270 4271 4272
    {NULL, NULL}
};

4273
static const vshCmdOptDef opts_reboot[] = {
4274
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
4275
    {"mode", VSH_OT_STRING, VSH_OFLAG_NONE, N_("shutdown mode: acpi|agent")},
4276 4277 4278
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
4279
static bool
4280
cmdReboot(vshControl *ctl, const vshCmd *cmd)
4281 4282
{
    virDomainPtr dom;
E
Eric Blake 已提交
4283
    bool ret = true;
4284
    const char *name;
4285 4286
    const char *mode = NULL;
    int flags = 0;
4287

4288
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
4289
        return false;
4290

4291 4292 4293 4294 4295 4296 4297 4298 4299 4300 4301 4302 4303 4304 4305 4306
    if (vshCommandOptString(cmd, "mode", &mode) < 0) {
        vshError(ctl, "%s", _("Invalid type"));
        return false;
    }

    if (mode) {
        if (STREQ(mode, "acpi")) {
            flags |= VIR_DOMAIN_SHUTDOWN_ACPI_POWER_BTN;
        } else if (STREQ(mode, "agent")) {
            flags |= VIR_DOMAIN_SHUTDOWN_GUEST_AGENT;
        } else {
            vshError(ctl, _("Unknown mode %s value, expecting 'acpi' or 'agent'"), mode);
            return false;
        }
    }

J
Jim Meyering 已提交
4307
    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
E
Eric Blake 已提交
4308
        return false;
4309

4310
    if (virDomainReboot(dom, flags) == 0) {
4311
        vshPrint(ctl, _("Domain %s is being rebooted\n"), name);
4312
    } else {
4313
        vshError(ctl, _("Failed to reboot domain %s"), name);
E
Eric Blake 已提交
4314
        ret = false;
4315 4316 4317 4318 4319 4320
    }

    virDomainFree(dom);
    return ret;
}

X
Xu He Jie 已提交
4321 4322 4323 4324 4325 4326 4327 4328 4329 4330 4331 4332 4333 4334 4335 4336 4337 4338 4339 4340 4341 4342 4343 4344 4345 4346 4347 4348 4349 4350 4351 4352 4353 4354 4355 4356 4357 4358
/*
 * "reset" command
 */
static const vshCmdInfo info_reset[] = {
    {"help", N_("reset a domain")},
    {"desc", N_("Reset the target domain as if by power button")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_reset[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {NULL, 0, 0, NULL}
};

static bool
cmdReset(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    bool ret = true;
    const char *name;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
        return false;

    if (virDomainReset(dom, 0) == 0) {
        vshPrint(ctl, _("Domain %s was reset\n"), name);
    } else {
        vshError(ctl, _("Failed to reset domain %s"), name);
        ret = false;
    }

    virDomainFree(dom);
    return ret;
}

4359 4360 4361
/*
 * "destroy" command
 */
4362
static const vshCmdInfo info_destroy[] = {
4363 4364 4365
    {"help", N_("destroy (stop) a domain")},
    {"desc",
     N_("Forcefully stop a given domain, but leave its resources intact.")},
4366
    {NULL, NULL}
4367 4368
};

4369
static const vshCmdOptDef opts_destroy[] = {
4370
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
4371
    {"graceful", VSH_OT_BOOL, VSH_OFLAG_NONE, N_("terminate gracefully")},
4372
    {NULL, 0, 0, NULL}
4373 4374
};

E
Eric Blake 已提交
4375
static bool
4376
cmdDestroy(vshControl *ctl, const vshCmd *cmd)
4377
{
4378
    virDomainPtr dom;
E
Eric Blake 已提交
4379
    bool ret = true;
4380
    const char *name;
4381 4382
    unsigned int flags = 0;
    int result;
4383

4384
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
4385
        return false;
4386

J
Jim Meyering 已提交
4387
    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
E
Eric Blake 已提交
4388
        return false;
4389

4390 4391 4392 4393 4394 4395 4396 4397 4398
    if (vshCommandOptBool(cmd, "graceful"))
       flags |= VIR_DOMAIN_DESTROY_GRACEFUL;

    if (flags)
       result = virDomainDestroyFlags(dom, VIR_DOMAIN_DESTROY_GRACEFUL);
    else
       result = virDomainDestroy(dom);

    if (result == 0) {
4399
        vshPrint(ctl, _("Domain %s destroyed\n"), name);
4400
    } else {
4401
        vshError(ctl, _("Failed to destroy domain %s"), name);
E
Eric Blake 已提交
4402
        ret = false;
4403
    }
4404

4405
    virDomainFree(dom);
K
Karel Zak 已提交
4406 4407 4408 4409
    return ret;
}

/*
4410
 * "dominfo" command
K
Karel Zak 已提交
4411
 */
4412
static const vshCmdInfo info_dominfo[] = {
4413 4414
    {"help", N_("domain information")},
    {"desc", N_("Returns basic information about the domain.")},
4415
    {NULL, NULL}
K
Karel Zak 已提交
4416 4417
};

4418
static const vshCmdOptDef opts_dominfo[] = {
4419
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
4420
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
4421 4422
};

E
Eric Blake 已提交
4423
static bool
4424
cmdDominfo(vshControl *ctl, const vshCmd *cmd)
4425
{
K
Karel Zak 已提交
4426 4427
    virDomainInfo info;
    virDomainPtr dom;
4428
    virSecurityModel secmodel;
4429
    virSecurityLabelPtr seclabel;
4430
    int persistent = 0;
E
Eric Blake 已提交
4431 4432
    bool ret = true;
    int autostart;
4433
    unsigned int id;
4434
    char *str, uuid[VIR_UUID_STRING_BUFLEN];
4435
    int has_managed_save = 0;
4436

4437
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
4438
        return false;
K
Karel Zak 已提交
4439

J
Jim Meyering 已提交
4440
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
4441
        return false;
4442

4443 4444
    id = virDomainGetID(dom);
    if (id == ((unsigned int)-1))
4445
        vshPrint(ctl, "%-15s %s\n", _("Id:"), "-");
4446
    else
4447
        vshPrint(ctl, "%-15s %d\n", _("Id:"), id);
4448 4449
    vshPrint(ctl, "%-15s %s\n", _("Name:"), virDomainGetName(dom));

K
Karel Zak 已提交
4450
    if (virDomainGetUUIDString(dom, &uuid[0])==0)
4451
        vshPrint(ctl, "%-15s %s\n", _("UUID:"), uuid);
4452 4453

    if ((str = virDomainGetOSType(dom))) {
4454
        vshPrint(ctl, "%-15s %s\n", _("OS Type:"), str);
4455
        VIR_FREE(str);
4456 4457 4458
    }

    if (virDomainGetInfo(dom, &info) == 0) {
4459
        vshPrint(ctl, "%-15s %s\n", _("State:"),
E
Eric Blake 已提交
4460
                 _(vshDomainStateToString(info.state)));
4461

4462
        vshPrint(ctl, "%-15s %d\n", _("CPU(s):"), info.nrVirtCpu);
4463 4464

        if (info.cpuTime != 0) {
4465
            double cpuUsed = info.cpuTime;
4466

4467
            cpuUsed /= 1000000000.0;
4468

4469
            vshPrint(ctl, "%-15s %.1lfs\n", _("CPU time:"), cpuUsed);
K
Karel Zak 已提交
4470
        }
4471

4472 4473
        if (info.maxMem != UINT_MAX)
            vshPrint(ctl, "%-15s %lu kB\n", _("Max memory:"),
4474
                 info.maxMem);
4475
        else
4476
            vshPrint(ctl, "%-15s %s\n", _("Max memory:"),
4477 4478
                 _("no limit"));

4479
        vshPrint(ctl, "%-15s %lu kB\n", _("Used memory:"),
4480 4481
                 info.memory);

K
Karel Zak 已提交
4482
    } else {
E
Eric Blake 已提交
4483
        ret = false;
K
Karel Zak 已提交
4484
    }
4485

4486 4487
    /* Check and display whether the domain is persistent or not */
    persistent = virDomainIsPersistent(dom);
4488 4489
    vshDebug(ctl, VSH_ERR_DEBUG, "Domain persistent flag value: %d\n",
             persistent);
4490 4491 4492 4493 4494 4495
    if (persistent < 0)
        vshPrint(ctl, "%-15s %s\n", _("Persistent:"), _("unknown"));
    else
        vshPrint(ctl, "%-15s %s\n", _("Persistent:"), persistent ? _("yes") : _("no"));

    /* Check and display whether the domain autostarts or not */
4496
    if (!virDomainGetAutostart(dom, &autostart)) {
4497
        vshPrint(ctl, "%-15s %s\n", _("Autostart:"),
4498 4499 4500
                 autostart ? _("enable") : _("disable") );
    }

4501 4502 4503 4504 4505 4506 4507
    has_managed_save = virDomainHasManagedSaveImage(dom, 0);
    if (has_managed_save < 0)
        vshPrint(ctl, "%-15s %s\n", _("Managed save:"), _("unknown"));
    else
        vshPrint(ctl, "%-15s %s\n", _("Managed save:"),
                 has_managed_save ? _("yes") : _("no"));

4508 4509 4510
    /* Security model and label information */
    memset(&secmodel, 0, sizeof secmodel);
    if (virNodeGetSecurityModel(ctl->conn, &secmodel) == -1) {
4511 4512
        if (last_error->code != VIR_ERR_NO_SUPPORT) {
            virDomainFree(dom);
E
Eric Blake 已提交
4513
            return false;
4514 4515 4516
        } else {
            virFreeError(last_error);
            last_error = NULL;
4517
        }
4518 4519 4520 4521 4522 4523 4524
    } else {
        /* Only print something if a security model is active */
        if (secmodel.model[0] != '\0') {
            vshPrint(ctl, "%-15s %s\n", _("Security model:"), secmodel.model);
            vshPrint(ctl, "%-15s %s\n", _("Security DOI:"), secmodel.doi);

            /* Security labels are only valid for active domains */
4525
            if (VIR_ALLOC(seclabel) < 0) {
4526
                virDomainFree(dom);
E
Eric Blake 已提交
4527
                return false;
4528 4529 4530 4531 4532
            }

            if (virDomainGetSecurityLabel(dom, seclabel) == -1) {
                virDomainFree(dom);
                VIR_FREE(seclabel);
E
Eric Blake 已提交
4533
                return false;
4534
            } else {
4535
                if (seclabel->label[0] != '\0')
4536
                    vshPrint(ctl, "%-15s %s (%s)\n", _("Security label:"),
4537
                             seclabel->label, seclabel->enforcing ? "enforcing" : "permissive");
4538
            }
4539 4540

            VIR_FREE(seclabel);
4541 4542
        }
    }
4543
    virDomainFree(dom);
K
Karel Zak 已提交
4544 4545 4546
    return ret;
}

4547 4548 4549 4550
/*
 * "domjobinfo" command
 */
static const vshCmdInfo info_domjobinfo[] = {
4551 4552
    {"help", N_("domain job information")},
    {"desc", N_("Returns information about jobs running on a domain.")},
4553 4554 4555 4556
    {NULL, NULL}
};

static const vshCmdOptDef opts_domjobinfo[] = {
4557
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
4558 4559 4560 4561
    {NULL, 0, 0, NULL}
};


E
Eric Blake 已提交
4562
static bool
4563 4564 4565 4566
cmdDomjobinfo(vshControl *ctl, const vshCmd *cmd)
{
    virDomainJobInfo info;
    virDomainPtr dom;
E
Eric Blake 已提交
4567
    bool ret = true;
4568

4569
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
4570
        return false;
4571 4572

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
4573
        return false;
4574 4575 4576 4577 4578 4579 4580 4581 4582 4583 4584 4585 4586 4587 4588 4589 4590 4591 4592 4593 4594 4595 4596 4597 4598 4599

    if (virDomainGetJobInfo(dom, &info) == 0) {
        const char *unit;
        double val;

        vshPrint(ctl, "%-17s ", _("Job type:"));
        switch (info.type) {
        case VIR_DOMAIN_JOB_BOUNDED:
            vshPrint(ctl, "%-12s\n", _("Bounded"));
            break;

        case VIR_DOMAIN_JOB_UNBOUNDED:
            vshPrint(ctl, "%-12s\n", _("Unbounded"));
            break;

        case VIR_DOMAIN_JOB_NONE:
        default:
            vshPrint(ctl, "%-12s\n", _("None"));
            goto cleanup;
        }

        vshPrint(ctl, "%-17s %-12llu ms\n", _("Time elapsed:"), info.timeElapsed);
        if (info.type == VIR_DOMAIN_JOB_BOUNDED)
            vshPrint(ctl, "%-17s %-12llu ms\n", _("Time remaining:"), info.timeRemaining);
        if (info.dataTotal || info.dataRemaining || info.dataProcessed) {
            val = prettyCapacity(info.dataProcessed, &unit);
E
Eric Blake 已提交
4600
            vshPrint(ctl, "%-17s %-.3lf %s\n", _("Data processed:"), val, unit);
4601
            val = prettyCapacity(info.dataRemaining, &unit);
E
Eric Blake 已提交
4602
            vshPrint(ctl, "%-17s %-.3lf %s\n", _("Data remaining:"), val, unit);
4603
            val = prettyCapacity(info.dataTotal, &unit);
E
Eric Blake 已提交
4604
            vshPrint(ctl, "%-17s %-.3lf %s\n", _("Data total:"), val, unit);
4605 4606 4607
        }
        if (info.memTotal || info.memRemaining || info.memProcessed) {
            val = prettyCapacity(info.memProcessed, &unit);
E
Eric Blake 已提交
4608
            vshPrint(ctl, "%-17s %-.3lf %s\n", _("Memory processed:"), val, unit);
4609
            val = prettyCapacity(info.memRemaining, &unit);
E
Eric Blake 已提交
4610
            vshPrint(ctl, "%-17s %-.3lf %s\n", _("Memory remaining:"), val, unit);
4611
            val = prettyCapacity(info.memTotal, &unit);
E
Eric Blake 已提交
4612
            vshPrint(ctl, "%-17s %-.3lf %s\n", _("Memory total:"), val, unit);
4613 4614 4615
        }
        if (info.fileTotal || info.fileRemaining || info.fileProcessed) {
            val = prettyCapacity(info.fileProcessed, &unit);
E
Eric Blake 已提交
4616
            vshPrint(ctl, "%-17s %-.3lf %s\n", _("File processed:"), val, unit);
4617
            val = prettyCapacity(info.fileRemaining, &unit);
E
Eric Blake 已提交
4618
            vshPrint(ctl, "%-17s %-.3lf %s\n", _("File remaining:"), val, unit);
4619
            val = prettyCapacity(info.fileTotal, &unit);
E
Eric Blake 已提交
4620
            vshPrint(ctl, "%-17s %-.3lf %s\n", _("File total:"), val, unit);
4621 4622
        }
    } else {
E
Eric Blake 已提交
4623
        ret = false;
4624 4625 4626 4627 4628 4629
    }
cleanup:
    virDomainFree(dom);
    return ret;
}

4630 4631 4632 4633
/*
 * "domjobabort" command
 */
static const vshCmdInfo info_domjobabort[] = {
4634 4635
    {"help", N_("abort active domain job")},
    {"desc", N_("Aborts the currently running domain job")},
4636 4637 4638 4639
    {NULL, NULL}
};

static const vshCmdOptDef opts_domjobabort[] = {
4640
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
4641 4642 4643
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
4644
static bool
4645 4646 4647
cmdDomjobabort(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
E
Eric Blake 已提交
4648
    bool ret = true;
4649

4650
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
4651
        return false;
4652 4653

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
4654
        return false;
4655 4656

    if (virDomainAbortJob(dom) < 0)
E
Eric Blake 已提交
4657
        ret = false;
4658 4659 4660 4661 4662

    virDomainFree(dom);
    return ret;
}

4663 4664 4665
/*
 * "freecell" command
 */
4666
static const vshCmdInfo info_freecell[] = {
4667 4668
    {"help", N_("NUMA free memory")},
    {"desc", N_("display available free memory for the NUMA cell.")},
4669 4670 4671
    {NULL, NULL}
};

4672
static const vshCmdOptDef opts_freecell[] = {
4673
    {"cellno", VSH_OT_INT, 0, N_("NUMA cell number")},
4674
    {"all", VSH_OT_BOOL, 0, N_("show free memory for all NUMA cells")},
4675 4676 4677
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
4678
static bool
4679
cmdFreecell(vshControl *ctl, const vshCmd *cmd)
4680
{
W
Wen Congyang 已提交
4681
    bool func_ret = false;
4682
    int ret;
4683
    int cell = -1, cell_given;
4684
    unsigned long long memory;
4685 4686 4687 4688
    xmlNodePtr *nodes = NULL;
    unsigned long nodes_cnt;
    unsigned long *nodes_id = NULL;
    unsigned long long *nodes_free = NULL;
4689
    int all_given;
4690 4691 4692 4693
    int i;
    char *cap_xml = NULL;
    xmlDocPtr xml = NULL;
    xmlXPathContextPtr ctxt = NULL;
4694

4695

4696
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
4697
        return false;
4698

4699 4700 4701 4702
    if ( (cell_given = vshCommandOptInt(cmd, "cellno", &cell)) < 0) {
        vshError(ctl, "%s", _("cell number has to be a number"));
        goto cleanup;
    }
4703 4704 4705 4706 4707 4708 4709 4710 4711
    all_given = vshCommandOptBool(cmd, "all");

    if (all_given && cell_given) {
        vshError(ctl, "%s", _("--cellno and --all are mutually exclusive. "
                              "Please choose only one."));
        goto cleanup;
    }

    if (all_given) {
4712 4713 4714
        cap_xml = virConnectGetCapabilities(ctl->conn);
        if (!cap_xml) {
            vshError(ctl, "%s", _("unable to get node capabilities"));
4715 4716 4717
            goto cleanup;
        }

4718
        xml = virXMLParseStringCtxt(cap_xml, _("(capabilities)"), &ctxt);
4719 4720
        if (!xml) {
            vshError(ctl, "%s", _("unable to get node capabilities"));
4721 4722
            goto cleanup;
        }
4723 4724 4725 4726
        nodes_cnt = virXPathNodeSet("/capabilities/host/topology/cells/cell",
                                    ctxt, &nodes);

        if (nodes_cnt == -1) {
4727
            vshError(ctl, "%s", _("could not get information about "
4728
                                  "NUMA topology"));
4729 4730 4731
            goto cleanup;
        }

4732 4733 4734 4735 4736 4737 4738 4739 4740 4741 4742 4743 4744 4745 4746 4747 4748 4749 4750 4751 4752
        nodes_free = vshCalloc(ctl, nodes_cnt, sizeof(*nodes_free));
        nodes_id = vshCalloc(ctl, nodes_cnt, sizeof(*nodes_id));

        for (i = 0; i < nodes_cnt; i++) {
            unsigned long id;
            char *val = virXMLPropString(nodes[i], "id");
            if (virStrToLong_ul(val, NULL, 10, &id)) {
                vshError(ctl, "%s", _("conversion from string failed"));
                VIR_FREE(val);
                goto cleanup;
            }
            VIR_FREE(val);
            nodes_id[i]=id;
            ret = virNodeGetCellsFreeMemory(ctl->conn, &(nodes_free[i]), id, 1);
            if (ret != 1) {
                vshError(ctl, _("failed to get free memory for NUMA node "
                                "number: %lu"), id);
                goto cleanup;
            }
        }

4753
        memory = 0;
4754 4755 4756 4757
        for (cell = 0; cell < nodes_cnt; cell++) {
            vshPrint(ctl, "%5lu: %10llu kB\n", nodes_id[cell],
                    (nodes_free[cell]/1024));
            memory += nodes_free[cell];
4758 4759 4760 4761
        }

        vshPrintExtra(ctl, "--------------------\n");
        vshPrintExtra(ctl, "%5s: %10llu kB\n", _("Total"), memory/1024);
4762
    } else {
4763 4764 4765 4766 4767 4768 4769 4770 4771 4772 4773 4774 4775 4776
        if (!cell_given) {
            memory = virNodeGetFreeMemory(ctl->conn);
            if (memory == 0)
                goto cleanup;
        } else {
            ret = virNodeGetCellsFreeMemory(ctl->conn, &memory, cell, 1);
            if (ret != 1)
                goto cleanup;
        }

        if (cell == -1)
            vshPrint(ctl, "%s: %llu kB\n", _("Total"), (memory/1024));
        else
            vshPrint(ctl, "%d: %llu kB\n", cell, (memory/1024));
4777 4778
    }

E
Eric Blake 已提交
4779
    func_ret = true;
4780

4781
cleanup:
4782
    xmlXPathFreeContext(ctxt);
4783
    xmlFreeDoc(xml);
4784
    VIR_FREE(nodes);
4785 4786 4787
    VIR_FREE(nodes_free);
    VIR_FREE(nodes_id);
    VIR_FREE(cap_xml);
4788
    return func_ret;
4789 4790
}

E
Eric Blake 已提交
4791 4792 4793 4794 4795 4796 4797 4798 4799 4800 4801 4802 4803 4804
/*
 * "maxvcpus" command
 */
static const vshCmdInfo info_maxvcpus[] = {
    {"help", N_("connection vcpu maximum")},
    {"desc", N_("Show maximum number of virtual CPUs for guests on this connection.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_maxvcpus[] = {
    {"type", VSH_OT_STRING, 0, N_("domain type")},
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
4805
static bool
E
Eric Blake 已提交
4806 4807
cmdMaxvcpus(vshControl *ctl, const vshCmd *cmd)
{
4808
    const char *type = NULL;
E
Eric Blake 已提交
4809 4810
    int vcpus;

4811 4812
    if (vshCommandOptString(cmd, "type", &type) < 0) {
        vshError(ctl, "%s", _("Invalid type"));
E
Eric Blake 已提交
4813
        return false;
4814
    }
E
Eric Blake 已提交
4815 4816

    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
4817
        return false;
E
Eric Blake 已提交
4818 4819 4820

    vcpus = virConnectGetMaxVcpus(ctl->conn, type);
    if (vcpus < 0)
E
Eric Blake 已提交
4821
        return false;
E
Eric Blake 已提交
4822 4823
    vshPrint(ctl, "%d\n", vcpus);

E
Eric Blake 已提交
4824
    return true;
E
Eric Blake 已提交
4825 4826 4827 4828 4829 4830 4831 4832 4833 4834 4835 4836 4837 4838
}

/*
 * "vcpucount" command
 */
static const vshCmdInfo info_vcpucount[] = {
    {"help", N_("domain vcpu counts")},
    {"desc", N_("Returns the number of virtual CPUs used by the domain.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_vcpucount[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"maximum", VSH_OT_BOOL, 0, N_("get maximum cap on vcpus")},
4839
    {"active", VSH_OT_BOOL, 0, N_("get number of currently active vcpus")},
E
Eric Blake 已提交
4840
    {"live", VSH_OT_BOOL, 0, N_("get value from running domain")},
4841 4842 4843
    {"config", VSH_OT_BOOL, 0, N_("get value to be used on next boot")},
    {"current", VSH_OT_BOOL, 0,
     N_("get value according to current domain state")},
E
Eric Blake 已提交
4844 4845 4846
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
4847
static bool
E
Eric Blake 已提交
4848 4849 4850
cmdVcpucount(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
E
Eric Blake 已提交
4851
    bool ret = true;
E
Eric Blake 已提交
4852
    int maximum = vshCommandOptBool(cmd, "maximum");
4853
    int active = vshCommandOptBool(cmd, "active");
E
Eric Blake 已提交
4854 4855
    int config = vshCommandOptBool(cmd, "config");
    int live = vshCommandOptBool(cmd, "live");
4856 4857
    int current = vshCommandOptBool(cmd, "current");
    bool all = maximum + active + current + config + live == 0;
E
Eric Blake 已提交
4858 4859
    int count;

4860 4861 4862 4863 4864 4865 4866 4867 4868 4869 4870 4871 4872 4873 4874
    /* We want one of each pair of mutually exclusive options; that
     * is, use of flags requires exactly two options.  We reject the
     * use of more than 2 flags later on.  */
    if (maximum + active + current + config + live == 1) {
        if (maximum || active) {
            vshError(ctl,
                     _("when using --%s, one of --config, --live, or --current "
                       "must be specified"),
                     maximum ? "maximum" : "active");
        } else {
            vshError(ctl,
                     _("when using --%s, either --maximum or --active must be "
                       "specified"),
                     (current ? "current" : config ? "config" : "live"));
        }
E
Eric Blake 已提交
4875
        return false;
E
Eric Blake 已提交
4876
    }
4877 4878 4879 4880 4881 4882 4883 4884 4885 4886 4887 4888 4889

    /* Backwards compatibility: prior to 0.9.4,
     * VIR_DOMAIN_AFFECT_CURRENT was unsupported, and --current meant
     * the opposite of --maximum.  Translate the old '--current
     * --live' into the new '--active --live', while treating the new
     * '--maximum --current' correctly rather than rejecting it as
     * '--maximum --active'.  */
    if (!maximum && !active && current) {
        current = false;
        active = true;
    }

    if (maximum && active) {
E
Eric Blake 已提交
4890
        vshError(ctl, "%s",
4891
                 _("--maximum and --active cannot both be specified"));
E
Eric Blake 已提交
4892
        return false;
E
Eric Blake 已提交
4893
    }
4894 4895 4896
    if (current + config + live > 1) {
        vshError(ctl, "%s",
                 _("--config, --live, and --current are mutually exclusive"));
E
Eric Blake 已提交
4897
        return false;
E
Eric Blake 已提交
4898 4899 4900
    }

    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
4901
        return false;
E
Eric Blake 已提交
4902 4903

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
4904
        return false;
E
Eric Blake 已提交
4905 4906

    /* In all cases, try the new API first; if it fails because we are
4907 4908 4909 4910 4911 4912 4913 4914 4915 4916 4917 4918 4919 4920
     * talking to an older client, generally we try a fallback API
     * before giving up.  --current requires the new API, since we
     * don't know whether the domain is running or inactive.  */
    if (current) {
        count = virDomainGetVcpusFlags(dom,
                                       maximum ? VIR_DOMAIN_VCPU_MAXIMUM : 0);
        if (count < 0) {
            virshReportError(ctl);
            ret = false;
        } else {
            vshPrint(ctl, "%d\n", count);
        }
    }

E
Eric Blake 已提交
4921 4922
    if (all || (maximum && config)) {
        count = virDomainGetVcpusFlags(dom, (VIR_DOMAIN_VCPU_MAXIMUM |
4923
                                             VIR_DOMAIN_AFFECT_CONFIG));
E
Eric Blake 已提交
4924 4925 4926 4927 4928 4929 4930 4931 4932
        if (count < 0 && (last_error->code == VIR_ERR_NO_SUPPORT
                          || last_error->code == VIR_ERR_INVALID_ARG)) {
            char *tmp;
            char *xml = virDomainGetXMLDesc(dom, VIR_DOMAIN_XML_INACTIVE);
            if (xml && (tmp = strstr(xml, "<vcpu"))) {
                tmp = strchr(tmp, '>');
                if (!tmp || virStrToLong_i(tmp + 1, &tmp, 10, &count) < 0)
                    count = -1;
            }
4933 4934
            virFreeError(last_error);
            last_error = NULL;
E
Eric Blake 已提交
4935 4936 4937 4938 4939
            VIR_FREE(xml);
        }

        if (count < 0) {
            virshReportError(ctl);
E
Eric Blake 已提交
4940
            ret = false;
E
Eric Blake 已提交
4941 4942 4943 4944 4945 4946 4947 4948 4949 4950 4951 4952
        } else if (all) {
            vshPrint(ctl, "%-12s %-12s %3d\n", _("maximum"), _("config"),
                     count);
        } else {
            vshPrint(ctl, "%d\n", count);
        }
        virFreeError(last_error);
        last_error = NULL;
    }

    if (all || (maximum && live)) {
        count = virDomainGetVcpusFlags(dom, (VIR_DOMAIN_VCPU_MAXIMUM |
4953
                                             VIR_DOMAIN_AFFECT_LIVE));
E
Eric Blake 已提交
4954 4955 4956 4957 4958 4959 4960
        if (count < 0 && (last_error->code == VIR_ERR_NO_SUPPORT
                          || last_error->code == VIR_ERR_INVALID_ARG)) {
            count = virDomainGetMaxVcpus(dom);
        }

        if (count < 0) {
            virshReportError(ctl);
E
Eric Blake 已提交
4961
            ret = false;
E
Eric Blake 已提交
4962 4963 4964 4965 4966 4967 4968 4969 4970 4971
        } else if (all) {
            vshPrint(ctl, "%-12s %-12s %3d\n", _("maximum"), _("live"),
                     count);
        } else {
            vshPrint(ctl, "%d\n", count);
        }
        virFreeError(last_error);
        last_error = NULL;
    }

4972
    if (all || (active && config)) {
4973
        count = virDomainGetVcpusFlags(dom, VIR_DOMAIN_AFFECT_CONFIG);
E
Eric Blake 已提交
4974 4975 4976 4977 4978 4979 4980 4981 4982 4983 4984 4985 4986 4987 4988 4989 4990 4991 4992 4993 4994 4995 4996 4997
        if (count < 0 && (last_error->code == VIR_ERR_NO_SUPPORT
                          || last_error->code == VIR_ERR_INVALID_ARG)) {
            char *tmp, *end;
            char *xml = virDomainGetXMLDesc(dom, VIR_DOMAIN_XML_INACTIVE);
            if (xml && (tmp = strstr(xml, "<vcpu"))) {
                end = strchr(tmp, '>');
                if (end) {
                    *end = '\0';
                    tmp = strstr(tmp, "current=");
                    if (!tmp)
                        tmp = end + 1;
                    else {
                        tmp += strlen("current=");
                        tmp += *tmp == '\'' || *tmp == '"';
                    }
                }
                if (!tmp || virStrToLong_i(tmp, &tmp, 10, &count) < 0)
                    count = -1;
            }
            VIR_FREE(xml);
        }

        if (count < 0) {
            virshReportError(ctl);
E
Eric Blake 已提交
4998
            ret = false;
E
Eric Blake 已提交
4999 5000 5001 5002 5003 5004 5005 5006 5007 5008
        } else if (all) {
            vshPrint(ctl, "%-12s %-12s %3d\n", _("current"), _("config"),
                     count);
        } else {
            vshPrint(ctl, "%d\n", count);
        }
        virFreeError(last_error);
        last_error = NULL;
    }

5009
    if (all || (active && live)) {
5010
        count = virDomainGetVcpusFlags(dom, VIR_DOMAIN_AFFECT_LIVE);
E
Eric Blake 已提交
5011 5012 5013 5014 5015 5016 5017 5018 5019
        if (count < 0 && (last_error->code == VIR_ERR_NO_SUPPORT
                          || last_error->code == VIR_ERR_INVALID_ARG)) {
            virDomainInfo info;
            if (virDomainGetInfo(dom, &info) == 0)
                count = info.nrVirtCpu;
        }

        if (count < 0) {
            virshReportError(ctl);
E
Eric Blake 已提交
5020
            ret = false;
E
Eric Blake 已提交
5021 5022 5023 5024 5025 5026 5027 5028 5029 5030 5031 5032 5033 5034
        } else if (all) {
            vshPrint(ctl, "%-12s %-12s %3d\n", _("current"), _("live"),
                     count);
        } else {
            vshPrint(ctl, "%d\n", count);
        }
        virFreeError(last_error);
        last_error = NULL;
    }

    virDomainFree(dom);
    return ret;
}

5035 5036 5037
/*
 * "vcpuinfo" command
 */
5038
static const vshCmdInfo info_vcpuinfo[] = {
E
Eric Blake 已提交
5039
    {"help", N_("detailed domain vcpu information")},
5040
    {"desc", N_("Returns basic information about the domain virtual CPUs.")},
5041 5042 5043
    {NULL, NULL}
};

5044
static const vshCmdOptDef opts_vcpuinfo[] = {
5045
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
5046 5047 5048
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
5049
static bool
5050
cmdVcpuinfo(vshControl *ctl, const vshCmd *cmd)
5051 5052 5053 5054 5055
{
    virDomainInfo info;
    virDomainPtr dom;
    virNodeInfo nodeinfo;
    virVcpuInfoPtr cpuinfo;
5056 5057
    unsigned char *cpumaps;
    int ncpus, maxcpu;
5058
    size_t cpumaplen;
E
Eric Blake 已提交
5059
    bool ret = true;
5060
    int n, m;
5061

5062
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
5063
        return false;
5064

J
Jim Meyering 已提交
5065
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
5066
        return false;
5067 5068 5069

    if (virNodeGetInfo(ctl->conn, &nodeinfo) != 0) {
        virDomainFree(dom);
E
Eric Blake 已提交
5070
        return false;
5071 5072 5073 5074
    }

    if (virDomainGetInfo(dom, &info) != 0) {
        virDomainFree(dom);
E
Eric Blake 已提交
5075
        return false;
5076 5077
    }

5078
    cpuinfo = vshMalloc(ctl, sizeof(virVcpuInfo)*info.nrVirtCpu);
5079 5080 5081
    maxcpu = VIR_NODEINFO_MAXCPUS(nodeinfo);
    cpumaplen = VIR_CPU_MAPLEN(maxcpu);
    cpumaps = vshMalloc(ctl, info.nrVirtCpu * cpumaplen);
5082

5083 5084
    if ((ncpus = virDomainGetVcpus(dom,
                                   cpuinfo, info.nrVirtCpu,
5085
                                   cpumaps, cpumaplen)) >= 0) {
5086 5087 5088 5089
        for (n = 0 ; n < ncpus ; n++) {
            vshPrint(ctl, "%-15s %d\n", _("VCPU:"), n);
            vshPrint(ctl, "%-15s %d\n", _("CPU:"), cpuinfo[n].cpu);
            vshPrint(ctl, "%-15s %s\n", _("State:"),
E
Eric Blake 已提交
5090
                     _(vshDomainVcpuStateToString(cpuinfo[n].state)));
5091 5092 5093 5094 5095 5096 5097 5098
            if (cpuinfo[n].cpuTime != 0) {
                double cpuUsed = cpuinfo[n].cpuTime;

                cpuUsed /= 1000000000.0;

                vshPrint(ctl, "%-15s %.1lfs\n", _("CPU time:"), cpuUsed);
            }
            vshPrint(ctl, "%-15s ", _("CPU Affinity:"));
5099 5100
            for (m = 0; m < maxcpu; m++) {
                vshPrint(ctl, "%c", VIR_CPU_USABLE(cpumaps, cpumaplen, n, m) ? 'y' : '-');
5101 5102 5103 5104 5105 5106
            }
            vshPrint(ctl, "\n");
            if (n < (ncpus - 1)) {
                vshPrint(ctl, "\n");
            }
        }
5107
    } else {
5108
        if (info.state == VIR_DOMAIN_SHUTOFF &&
E
Eric Blake 已提交
5109
            (ncpus = virDomainGetVcpuPinInfo(dom, info.nrVirtCpu,
5110 5111 5112
                                             cpumaps, cpumaplen,
                                             VIR_DOMAIN_AFFECT_CONFIG)) >= 0) {

E
Eric Blake 已提交
5113
            /* fallback plan to use virDomainGetVcpuPinInfo */
5114 5115 5116 5117 5118 5119 5120 5121 5122 5123 5124 5125 5126 5127 5128 5129 5130 5131

            for (n = 0; n < ncpus; n++) {
                vshPrint(ctl, "%-15s %d\n", _("VCPU:"), n);
                vshPrint(ctl, "%-15s %s\n", _("CPU:"), _("N/A"));
                vshPrint(ctl, "%-15s %s\n", _("State:"), _("N/A"));
                vshPrint(ctl, "%-15s %s\n", _("CPU time"), _("N/A"));
                vshPrint(ctl, "%-15s ", _("CPU Affinity:"));
                for (m = 0; m < maxcpu; m++) {
                    vshPrint(ctl, "%c",
                             VIR_CPU_USABLE(cpumaps, cpumaplen, n, m) ? 'y' : '-');
                }
                vshPrint(ctl, "\n");
                if (n < (ncpus - 1)) {
                    vshPrint(ctl, "\n");
                }
            }
        } else {
            ret = false;
5132
        }
5133 5134
    }

5135
    VIR_FREE(cpumaps);
5136
    VIR_FREE(cpuinfo);
5137 5138 5139 5140 5141 5142 5143
    virDomainFree(dom);
    return ret;
}

/*
 * "vcpupin" command
 */
5144
static const vshCmdInfo info_vcpupin[] = {
5145
    {"help", N_("control or query domain vcpu affinity")},
5146
    {"desc", N_("Pin domain VCPUs to host physical CPUs.")},
5147 5148 5149
    {NULL, NULL}
};

5150
static const vshCmdOptDef opts_vcpupin[] = {
5151
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
5152 5153 5154
    {"vcpu", VSH_OT_INT, 0, N_("vcpu number")},
    {"cpulist", VSH_OT_DATA, VSH_OFLAG_EMPTY_OK,
     N_("host cpu number(s) to set, or omit option to query")},
5155 5156 5157
    {"config", VSH_OT_BOOL, 0, N_("affect next boot")},
    {"live", VSH_OT_BOOL, 0, N_("affect running domain")},
    {"current", VSH_OT_BOOL, 0, N_("affect current domain")},
5158 5159 5160
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
5161
static bool
E
Eric Blake 已提交
5162
cmdVcpuPin(vshControl *ctl, const vshCmd *cmd)
5163 5164 5165 5166
{
    virDomainInfo info;
    virDomainPtr dom;
    virNodeInfo nodeinfo;
5167
    int vcpu = -1;
5168
    const char *cpulist = NULL;
E
Eric Blake 已提交
5169
    bool ret = true;
5170 5171
    unsigned char *cpumap = NULL;
    unsigned char *cpumaps = NULL;
E
Eric Blake 已提交
5172
    size_t cpumaplen;
5173 5174
    bool bit, lastbit, isInvert;
    int i, cpu, lastcpu, maxcpu, ncpus;
5175
    bool unuse = false;
5176
    const char *cur;
5177 5178 5179
    int config = vshCommandOptBool(cmd, "config");
    int live = vshCommandOptBool(cmd, "live");
    int current = vshCommandOptBool(cmd, "current");
5180
    bool query = false; /* Query mode if no cpulist */
E
Eric Blake 已提交
5181
    unsigned int flags = 0;
5182 5183 5184 5185 5186 5187 5188 5189 5190 5191 5192 5193 5194 5195 5196 5197

    if (current) {
        if (live || config) {
            vshError(ctl, "%s", _("--current must be specified exclusively"));
            return false;
        }
        flags = VIR_DOMAIN_AFFECT_CURRENT;
    } else {
        if (config)
            flags |= VIR_DOMAIN_AFFECT_CONFIG;
        if (live)
            flags |= VIR_DOMAIN_AFFECT_LIVE;
        /* neither option is specified */
        if (!live && !config)
            flags = -1;
    }
5198

5199
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
5200
        return false;
5201

J
Jim Meyering 已提交
5202
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
5203
        return false;
5204

5205 5206
    if (vshCommandOptString(cmd, "cpulist", &cpulist) < 0) {
        vshError(ctl, "%s", _("vcpupin: Missing cpulist."));
5207
        virDomainFree(dom);
E
Eric Blake 已提交
5208
        return false;
5209
    }
5210
    query = !cpulist;
5211

5212 5213 5214 5215
    /* In query mode, "vcpu" is optional */
    if (vshCommandOptInt(cmd, "vcpu", &vcpu) < !query) {
        vshError(ctl, "%s",
                 _("vcpupin: Invalid or missing vCPU number."));
5216
        virDomainFree(dom);
E
Eric Blake 已提交
5217
        return false;
5218
    }
5219

5220 5221
    if (virNodeGetInfo(ctl->conn, &nodeinfo) != 0) {
        virDomainFree(dom);
E
Eric Blake 已提交
5222
        return false;
5223 5224 5225
    }

    if (virDomainGetInfo(dom, &info) != 0) {
5226
        vshError(ctl, "%s", _("vcpupin: failed to get domain information."));
5227
        virDomainFree(dom);
E
Eric Blake 已提交
5228
        return false;
5229 5230 5231
    }

    if (vcpu >= info.nrVirtCpu) {
5232
        vshError(ctl, "%s", _("vcpupin: Invalid vCPU number."));
5233
        virDomainFree(dom);
E
Eric Blake 已提交
5234
        return false;
5235 5236
    }

5237 5238
    maxcpu = VIR_NODEINFO_MAXCPUS(nodeinfo);
    cpumaplen = VIR_CPU_MAPLEN(maxcpu);
5239

5240 5241 5242 5243 5244 5245 5246 5247
    /* Query mode: show CPU affinity information then exit.*/
    if (query) {
        /* When query mode and neither "live", "config" nor "current"
         * is specified, set VIR_DOMAIN_AFFECT_CURRENT as flags */
        if (flags == -1)
            flags = VIR_DOMAIN_AFFECT_CURRENT;

        cpumaps = vshMalloc(ctl, info.nrVirtCpu * cpumaplen);
E
Eric Blake 已提交
5248
        if ((ncpus = virDomainGetVcpuPinInfo(dom, info.nrVirtCpu,
5249 5250 5251 5252 5253 5254 5255 5256 5257 5258 5259 5260 5261 5262 5263 5264 5265 5266 5267 5268 5269 5270 5271 5272 5273 5274 5275 5276 5277 5278 5279 5280 5281 5282 5283 5284 5285 5286 5287 5288 5289 5290 5291 5292
                                             cpumaps, cpumaplen, flags)) >= 0) {

            vshPrint(ctl, "%s %s\n", _("VCPU:"), _("CPU Affinity"));
            vshPrint(ctl, "----------------------------------\n");
            for (i = 0; i < ncpus; i++) {

               if (vcpu != -1 && i != vcpu)
                   continue;

               bit = lastbit = isInvert = false;
               lastcpu = -1;

               vshPrint(ctl, "%4d: ", i);
               for (cpu = 0; cpu < maxcpu; cpu++) {

                   bit = VIR_CPU_USABLE(cpumaps, cpumaplen, i, cpu);

                   isInvert = (bit ^ lastbit);
                   if (bit && isInvert) {
                       if (lastcpu == -1)
                           vshPrint(ctl, "%d", cpu);
                       else
                           vshPrint(ctl, ",%d", cpu);
                       lastcpu = cpu;
                   }
                   if (!bit && isInvert && lastcpu != cpu - 1)
                       vshPrint(ctl, "-%d", cpu - 1);
                   lastbit = bit;
               }
               if (bit && !isInvert) {
                  vshPrint(ctl, "-%d", maxcpu - 1);
               }
               vshPrint(ctl, "\n");
            }

        } else {
            ret = false;
        }
        VIR_FREE(cpumaps);
        goto cleanup;
    }

    /* Pin mode: pinning specified vcpu to specified physical cpus*/

5293
    cpumap = vshCalloc(ctl, cpumaplen, sizeof(cpumap));
5294 5295
    /* Parse cpulist */
    cur = cpulist;
5296
    if (*cur == 0) {
5297
        goto parse_error;
5298 5299 5300
    } else if (*cur == 'r') {
        for (cpu = 0; cpu < maxcpu; cpu++)
            VIR_USE_CPU(cpumap, cpu);
5301
        cur = "";
5302
    }
5303 5304 5305 5306 5307 5308 5309

    while (*cur != 0) {

        /* the char '^' denotes exclusive */
        if (*cur == '^') {
            cur++;
            unuse = true;
5310 5311
        }

5312 5313 5314 5315 5316 5317 5318 5319 5320 5321 5322 5323
        /* parse physical CPU number */
        if (!c_isdigit(*cur))
            goto parse_error;
        cpu  = virParseNumber(&cur);
        if (cpu < 0) {
            goto parse_error;
        }
        if (cpu >= maxcpu) {
            vshError(ctl, _("Physical CPU %d doesn't exist."), cpu);
            goto parse_error;
        }
        virSkipSpaces(&cur);
5324

5325 5326 5327 5328 5329 5330 5331 5332 5333 5334 5335 5336 5337 5338 5339 5340 5341 5342 5343 5344 5345 5346 5347 5348 5349 5350 5351
        if ((*cur == ',') || (*cur == 0)) {
            if (unuse) {
                VIR_UNUSE_CPU(cpumap, cpu);
            } else {
                VIR_USE_CPU(cpumap, cpu);
            }
        } else if (*cur == '-') {
            /* the char '-' denotes range */
            if (unuse) {
                goto parse_error;
            }
            cur++;
            virSkipSpaces(&cur);
            /* parse the end of range */
            lastcpu = virParseNumber(&cur);
            if (lastcpu < cpu) {
                goto parse_error;
            }
            if (lastcpu >= maxcpu) {
                vshError(ctl, _("Physical CPU %d doesn't exist."), maxcpu);
                goto parse_error;
            }
            for (i = cpu; i <= lastcpu; i++) {
                VIR_USE_CPU(cpumap, i);
            }
            virSkipSpaces(&cur);
        }
5352

5353 5354 5355 5356 5357 5358
        if (*cur == ',') {
            cur++;
            virSkipSpaces(&cur);
            unuse = false;
        } else if (*cur == 0) {
            break;
5359
        } else {
5360
            goto parse_error;
5361
        }
5362
    }
5363

5364 5365 5366 5367 5368 5369 5370 5371
    if (flags == -1) {
        if (virDomainPinVcpu(dom, vcpu, cpumap, cpumaplen) != 0) {
            ret = false;
        }
    } else {
        if (virDomainPinVcpuFlags(dom, vcpu, cpumap, cpumaplen, flags) != 0) {
            ret = false;
        }
5372 5373
    }

5374
cleanup:
5375
    VIR_FREE(cpumap);
5376 5377
    virDomainFree(dom);
    return ret;
5378 5379 5380 5381 5382

parse_error:
    vshError(ctl, "%s", _("cpulist: Invalid format."));
    ret = false;
    goto cleanup;
5383 5384
}

5385 5386 5387
/*
 * "setvcpus" command
 */
5388
static const vshCmdInfo info_setvcpus[] = {
5389 5390
    {"help", N_("change number of virtual CPUs")},
    {"desc", N_("Change the number of virtual CPUs in the guest domain.")},
5391 5392 5393
    {NULL, NULL}
};

5394
static const vshCmdOptDef opts_setvcpus[] = {
5395
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
5396
    {"count", VSH_OT_INT, VSH_OFLAG_REQ, N_("number of virtual CPUs")},
E
Eric Blake 已提交
5397 5398 5399
    {"maximum", VSH_OT_BOOL, 0, N_("set maximum limit on next boot")},
    {"config", VSH_OT_BOOL, 0, N_("affect next boot")},
    {"live", VSH_OT_BOOL, 0, N_("affect running domain")},
5400
    {"current", VSH_OT_BOOL, 0, N_("affect current domain")},
5401 5402 5403
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
5404
static bool
5405
cmdSetvcpus(vshControl *ctl, const vshCmd *cmd)
5406 5407
{
    virDomainPtr dom;
5408
    int count = 0;
E
Eric Blake 已提交
5409
    bool ret = true;
E
Eric Blake 已提交
5410 5411 5412
    int maximum = vshCommandOptBool(cmd, "maximum");
    int config = vshCommandOptBool(cmd, "config");
    int live = vshCommandOptBool(cmd, "live");
5413
    int current = vshCommandOptBool(cmd, "current");
E
Eric Blake 已提交
5414
    unsigned int flags = 0;
5415 5416 5417 5418 5419 5420 5421 5422 5423 5424 5425 5426 5427 5428 5429 5430

    if (current) {
        if (live || config) {
            vshError(ctl, "%s", _("--current must be specified exclusively"));
            return false;
        }
        flags = VIR_DOMAIN_AFFECT_CURRENT;
    } else {
        if (config)
            flags |= VIR_DOMAIN_AFFECT_CONFIG;
        if (live)
            flags |= VIR_DOMAIN_AFFECT_LIVE;
        /* neither option is specified */
        if (!live && !config && !maximum)
            flags = -1;
    }
5431

5432
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
5433
        return false;
5434

J
Jim Meyering 已提交
5435
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
5436
        return false;
5437

5438 5439 5440 5441
    if (vshCommandOptInt(cmd, "count", &count) < 0) {
        vshError(ctl, "%s", _("Invalid number of virtual CPUs"));
        goto cleanup;
    }
5442

5443
    if (flags == -1) {
E
Eric Blake 已提交
5444
        if (virDomainSetVcpus(dom, count) != 0) {
E
Eric Blake 已提交
5445
            ret = false;
E
Eric Blake 已提交
5446 5447
        }
    } else {
5448 5449 5450
        /* If the --maximum flag was given, we need to ensure only the
           --config flag is in effect as well */
        if (maximum) {
5451
            vshDebug(ctl, VSH_ERR_DEBUG, "--maximum flag was given\n");
5452

5453 5454
            flags |= VIR_DOMAIN_VCPU_MAXIMUM;

5455 5456 5457 5458 5459 5460 5461 5462
            /* If neither the --config nor --live flags were given, OR
               if just the --live flag was given, we need to error out
               warning the user that the --maximum flag can only be used
               with the --config flag */
            if (live || !config) {

                /* Warn the user about the invalid flag combination */
                vshError(ctl, _("--maximum must be used with --config only"));
E
Eric Blake 已提交
5463
                ret = false;
5464 5465 5466 5467 5468
                goto cleanup;
            }
        }

        /* Apply the virtual cpu changes */
E
Eric Blake 已提交
5469
        if (virDomainSetVcpusFlags(dom, count, flags) < 0) {
E
Eric Blake 已提交
5470
            ret = false;
E
Eric Blake 已提交
5471
        }
5472 5473
    }

5474
  cleanup:
5475 5476 5477 5478
    virDomainFree(dom);
    return ret;
}

5479 5480 5481 5482 5483 5484 5485 5486 5487 5488 5489 5490 5491 5492 5493 5494 5495 5496 5497 5498 5499 5500 5501 5502 5503 5504 5505 5506 5507 5508 5509 5510 5511 5512
/*
 * "inject-nmi" command
 */
static const vshCmdInfo info_inject_nmi[] = {
    {"help", N_("Inject NMI to the guest")},
    {"desc", N_("Inject NMI to the guest domain.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_inject_nmi[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {NULL, 0, 0, NULL}
};


static bool
cmdInjectNMI(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    int ret = true;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

    if (virDomainInjectNMI(dom, 0) < 0)
            ret = false;

    virDomainFree(dom);
    return ret;
}

5513 5514 5515 5516 5517
/*
 * "send-key" command
 */
static const vshCmdInfo info_send_key[] = {
    {"help", N_("Send keycodes to the guest")},
5518
    {"desc", N_("Send keycodes (integers or symbolic names) to the guest")},
5519 5520 5521 5522 5523
    {NULL, NULL}
};

static const vshCmdOptDef opts_send_key[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
5524 5525
    {"codeset", VSH_OT_STRING, VSH_OFLAG_REQ_OPT,
     N_("the codeset of keycodes, default:linux")},
5526
    {"holdtime", VSH_OT_INT, VSH_OFLAG_REQ_OPT,
A
Alex Jia 已提交
5527
     N_("the time (in milliseconds) how long the keys will be held")},
5528 5529 5530 5531 5532 5533 5534 5535 5536 5537 5538 5539 5540 5541 5542 5543 5544 5545 5546 5547 5548 5549 5550 5551 5552 5553 5554 5555 5556 5557 5558 5559 5560 5561 5562 5563 5564 5565 5566 5567 5568 5569 5570 5571 5572 5573 5574 5575 5576 5577 5578 5579 5580 5581 5582 5583 5584 5585 5586 5587 5588 5589 5590 5591 5592 5593 5594 5595 5596 5597 5598 5599
    {"keycode", VSH_OT_ARGV, VSH_OFLAG_REQ, N_("the key code")},
    {NULL, 0, 0, NULL}
};

static int get_integer_keycode(const char *key_name)
{
    long val;
    char *endptr;

    val = strtol(key_name, &endptr, 0);
    if (*endptr != '\0' || val > 0xffff || val <= 0)
         return -1;

    return val;
}

static bool
cmdSendKey(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    int ret = false;
    const char *codeset_option;
    int codeset;
    int holdtime;
    int count = 0;
    const vshCmdOpt *opt = NULL;
    int keycode;
    unsigned int keycodes[VIR_DOMAIN_SEND_KEY_MAX_KEYS];

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

    if (vshCommandOptString(cmd, "codeset", &codeset_option) <= 0)
        codeset_option = "linux";

    if (vshCommandOptInt(cmd, "holdtime", &holdtime) <= 0)
        holdtime = 0;

    codeset = virKeycodeSetTypeFromString(codeset_option);
    if ((int)codeset < 0) {
        vshError(ctl, _("unknown codeset: '%s'"), codeset_option);
        goto cleanup;
    }

    while ((opt = vshCommandOptArgv(cmd, opt))) {
        if (count == VIR_DOMAIN_SEND_KEY_MAX_KEYS) {
            vshError(ctl, _("too many keycodes"));
            goto cleanup;
        }

        if ((keycode = get_integer_keycode(opt->data)) <= 0) {
            if ((keycode = virKeycodeValueFromString(codeset, opt->data)) <= 0) {
                vshError(ctl, _("invalid keycode: '%s'"), opt->data);
                goto cleanup;
            }
        }

        keycodes[count] = keycode;
        count++;
    }

    if (!(virDomainSendKey(dom, codeset, holdtime, keycodes, count, 0) < 0))
        ret = true;

cleanup:
    virDomainFree(dom);
    return ret;
}

5600 5601 5602
/*
 * "setmemory" command
 */
5603
static const vshCmdInfo info_setmem[] = {
5604 5605
    {"help", N_("change memory allocation")},
    {"desc", N_("Change the current memory allocation in the guest domain.")},
5606 5607 5608
    {NULL, NULL}
};

5609
static const vshCmdOptDef opts_setmem[] = {
5610
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
5611
    {"kilobytes", VSH_OT_INT, VSH_OFLAG_REQ, N_("number of kilobytes of memory")},
5612 5613
    {"config", VSH_OT_BOOL, 0, N_("affect next boot")},
    {"live", VSH_OT_BOOL, 0, N_("affect running domain")},
5614
    {"current", VSH_OT_BOOL, 0, N_("affect current domain")},
5615 5616 5617
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
5618
static bool
5619
cmdSetmem(vshControl *ctl, const vshCmd *cmd)
5620 5621
{
    virDomainPtr dom;
5622
    virDomainInfo info;
5623
    unsigned long kilobytes = 0;
E
Eric Blake 已提交
5624
    bool ret = true;
5625 5626
    int config = vshCommandOptBool(cmd, "config");
    int live = vshCommandOptBool(cmd, "live");
5627
    int current = vshCommandOptBool(cmd, "current");
E
Eric Blake 已提交
5628
    unsigned int flags = 0;
5629

5630 5631 5632
    if (current) {
        if (live || config) {
            vshError(ctl, "%s", _("--current must be specified exclusively"));
E
Eric Blake 已提交
5633
            return false;
5634
        }
5635
        flags = VIR_DOMAIN_AFFECT_CURRENT;
5636 5637
    } else {
        if (config)
5638
            flags |= VIR_DOMAIN_AFFECT_CONFIG;
5639
        if (live)
5640
            flags |= VIR_DOMAIN_AFFECT_LIVE;
5641 5642 5643
        /* neither option is specified */
        if (!live && !config)
            flags = -1;
5644
    }
5645

5646
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
5647
        return false;
5648

J
Jim Meyering 已提交
5649
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
5650
        return false;
5651

5652 5653
    if (vshCommandOptUL(cmd, "kilobytes", &kilobytes) < 0) {
        vshError(ctl, "%s", _("memory size has to be a number"));
E
Eric Blake 已提交
5654
        return false;
5655 5656
    }

5657
    if (kilobytes <= 0) {
5658
        virDomainFree(dom);
5659
        vshError(ctl, _("Invalid value of %lu for memory size"), kilobytes);
E
Eric Blake 已提交
5660
        return false;
5661 5662
    }

5663 5664
    if (virDomainGetInfo(dom, &info) != 0) {
        virDomainFree(dom);
5665
        vshError(ctl, "%s", _("Unable to verify MaxMemorySize"));
E
Eric Blake 已提交
5666
        return false;
5667 5668 5669 5670
    }

    if (kilobytes > info.maxMem) {
        virDomainFree(dom);
5671
        vshError(ctl, _("Requested memory size %lu kb is larger than maximum of %lu kb"),
5672
                 kilobytes, info.maxMem);
E
Eric Blake 已提交
5673
        return false;
5674 5675
    }

5676
    if (flags == -1) {
5677
        if (virDomainSetMemory(dom, kilobytes) != 0) {
E
Eric Blake 已提交
5678
            ret = false;
5679 5680 5681
        }
    } else {
        if (virDomainSetMemoryFlags(dom, kilobytes, flags) < 0) {
E
Eric Blake 已提交
5682
            ret = false;
5683
        }
5684 5685 5686 5687 5688 5689 5690 5691 5692
    }

    virDomainFree(dom);
    return ret;
}

/*
 * "setmaxmem" command
 */
5693
static const vshCmdInfo info_setmaxmem[] = {
5694 5695
    {"help", N_("change maximum memory limit")},
    {"desc", N_("Change the maximum memory allocation limit in the guest domain.")},
5696 5697 5698
    {NULL, NULL}
};

5699
static const vshCmdOptDef opts_setmaxmem[] = {
5700
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
5701
    {"kilobytes", VSH_OT_INT, VSH_OFLAG_REQ, N_("maximum memory limit in kilobytes")},
5702 5703 5704
    {"config", VSH_OT_BOOL, 0, N_("affect next boot")},
    {"live", VSH_OT_BOOL, 0, N_("affect running domain")},
    {"current", VSH_OT_BOOL, 0, N_("affect current domain")},
5705 5706 5707
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
5708
static bool
5709
cmdSetmaxmem(vshControl *ctl, const vshCmd *cmd)
5710 5711
{
    virDomainPtr dom;
5712
    int kilobytes = 0;
E
Eric Blake 已提交
5713
    bool ret = true;
5714 5715 5716
    int config = vshCommandOptBool(cmd, "config");
    int live = vshCommandOptBool(cmd, "live");
    int current = vshCommandOptBool(cmd, "current");
E
Eric Blake 已提交
5717
    unsigned int flags = VIR_DOMAIN_MEM_MAXIMUM;
5718 5719 5720 5721

    if (current) {
        if (live || config) {
            vshError(ctl, "%s", _("--current must be specified exclusively"));
E
Eric Blake 已提交
5722
            return false;
5723 5724 5725
        }
    } else {
        if (config)
5726
            flags |= VIR_DOMAIN_AFFECT_CONFIG;
5727
        if (live)
5728
            flags |= VIR_DOMAIN_AFFECT_LIVE;
5729 5730 5731 5732
        /* neither option is specified */
        if (!live && !config)
            flags = -1;
    }
5733

5734
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
5735
        return false;
5736

J
Jim Meyering 已提交
5737
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
5738
        return false;
5739

5740 5741
    if (vshCommandOptInt(cmd, "kilobytes", &kilobytes) < 0) {
        vshError(ctl, "%s", _("memory size has to be a number"));
E
Eric Blake 已提交
5742
        return false;
5743 5744
    }

5745
    if (kilobytes <= 0) {
5746
        virDomainFree(dom);
5747
        vshError(ctl, _("Invalid value of %d for memory size"), kilobytes);
E
Eric Blake 已提交
5748
        return false;
5749 5750
    }

5751 5752 5753
    if (flags == -1) {
        if (virDomainSetMaxMemory(dom, kilobytes) != 0) {
            vshError(ctl, "%s", _("Unable to change MaxMemorySize"));
E
Eric Blake 已提交
5754
            ret = false;
5755 5756 5757 5758
        }
    } else {
        if (virDomainSetMemoryFlags(dom, kilobytes, flags) < 0) {
            vshError(ctl, "%s", _("Unable to change MaxMemorySize"));
E
Eric Blake 已提交
5759
            ret = false;
5760
        }
5761 5762
    }

5763 5764 5765 5766
    virDomainFree(dom);
    return ret;
}

5767 5768 5769 5770 5771 5772 5773 5774 5775 5776 5777 5778 5779 5780 5781 5782
/*
 * "blkiotune" command
 */
static const vshCmdInfo info_blkiotune[] = {
    {"help", N_("Get or set blkio parameters")},
    {"desc", N_("Get or set the current blkio parameters for a guest" \
                " domain.\n" \
                "    To get the blkio parameters use following command: \n\n" \
                "    virsh # blkiotune <domain>")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_blkiotune[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"weight", VSH_OT_INT, VSH_OFLAG_NONE,
     N_("IO Weight in range [100, 1000]")},
5783 5784
    {"device-weights", VSH_OT_STRING, VSH_OFLAG_NONE,
     N_("per-device IO Weights, in the form of /path/to/device,weight,...")},
H
Hu Tao 已提交
5785 5786 5787
    {"config", VSH_OT_BOOL, 0, N_("affect next boot")},
    {"live", VSH_OT_BOOL, 0, N_("affect running domain")},
    {"current", VSH_OT_BOOL, 0, N_("affect current domain")},
5788 5789 5790
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
5791
static bool
5792 5793 5794
cmdBlkiotune(vshControl * ctl, const vshCmd * cmd)
{
    virDomainPtr dom;
5795
    const char *device_weight = NULL;
5796 5797
    int weight = 0;
    int nparams = 0;
5798
    int rv = 0;
5799
    unsigned int i = 0;
5800
    virTypedParameterPtr params = NULL, temp = NULL;
E
Eric Blake 已提交
5801
    bool ret = false;
H
Hu Tao 已提交
5802 5803 5804 5805 5806 5807 5808 5809 5810 5811 5812 5813 5814 5815 5816 5817 5818
    unsigned int flags = 0;
    int current = vshCommandOptBool(cmd, "current");
    int config = vshCommandOptBool(cmd, "config");
    int live = vshCommandOptBool(cmd, "live");

    if (current) {
        if (live || config) {
            vshError(ctl, "%s", _("--current must be specified exclusively"));
            return false;
        }
        flags = VIR_DOMAIN_AFFECT_CURRENT;
    } else {
        if (config)
            flags |= VIR_DOMAIN_AFFECT_CONFIG;
        if (live)
            flags |= VIR_DOMAIN_AFFECT_LIVE;
    }
5819 5820

    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
5821
        return false;
5822 5823

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
5824
        return false;
5825

5826
    if ((rv = vshCommandOptInt(cmd, "weight", &weight)) < 0) {
5827 5828 5829 5830 5831
        vshError(ctl, "%s",
                 _("Unable to parse integer parameter"));
        goto cleanup;
    }

5832
    if (rv > 0) {
5833
        nparams++;
5834
        if (weight <= 0) {
5835 5836 5837 5838 5839
            vshError(ctl, _("Invalid value of %d for I/O weight"), weight);
            goto cleanup;
        }
    }

5840 5841 5842 5843 5844 5845 5846 5847 5848 5849
    rv = vshCommandOptString(cmd, "device-weights", &device_weight);
    if (rv < 0) {
        vshError(ctl, "%s",
                 _("Unable to parse string parameter"));
        goto cleanup;
    }
    if (rv > 0) {
        nparams++;
    }

5850 5851
    if (nparams == 0) {
        /* get the number of blkio parameters */
H
Hu Tao 已提交
5852
        if (virDomainGetBlkioParameters(dom, NULL, &nparams, flags) != 0) {
5853 5854 5855 5856 5857 5858 5859
            vshError(ctl, "%s",
                     _("Unable to get number of blkio parameters"));
            goto cleanup;
        }

        if (nparams == 0) {
            /* nothing to output */
E
Eric Blake 已提交
5860
            ret = true;
5861 5862 5863 5864 5865
            goto cleanup;
        }

        /* now go get all the blkio parameters */
        params = vshCalloc(ctl, nparams, sizeof(*params));
H
Hu Tao 已提交
5866
        if (virDomainGetBlkioParameters(dom, params, &nparams, flags) != 0) {
5867 5868 5869 5870 5871
            vshError(ctl, "%s", _("Unable to get blkio parameters"));
            goto cleanup;
        }

        for (i = 0; i < nparams; i++) {
5872 5873 5874
            char *str = vshGetTypedParamValue(ctl, &params[i]);
            vshPrint(ctl, "%-15s: %s\n", params[i].field, str);
            VIR_FREE(str);
5875 5876 5877 5878 5879 5880 5881
        }
    } else {
        /* set the blkio parameters */
        params = vshCalloc(ctl, nparams, sizeof(*params));

        for (i = 0; i < nparams; i++) {
            temp = &params[i];
5882
            temp->type = VIR_TYPED_PARAM_UINT;
5883 5884 5885

            if (weight) {
                temp->value.ui = weight;
5886 5887 5888
                if (!virStrcpy(temp->field, VIR_DOMAIN_BLKIO_WEIGHT,
                               sizeof(temp->field)))
                    goto cleanup;
5889 5890
                weight = 0;
            } else if (device_weight) {
5891 5892 5893 5894 5895
                temp->value.s = vshStrdup(ctl, device_weight);
                temp->type = VIR_TYPED_PARAM_STRING;
                if (!virStrcpy(temp->field, VIR_DOMAIN_BLKIO_DEVICE_WEIGHT,
                               sizeof(temp->field)))
                    goto cleanup;
5896
                device_weight = NULL;
5897 5898
            }
        }
5899 5900

        if (virDomainSetBlkioParameters(dom, params, nparams, flags) < 0) {
5901
            vshError(ctl, "%s", _("Unable to change blkio parameters"));
5902 5903
            goto cleanup;
        }
5904 5905
    }

5906 5907
    ret = true;

5908
  cleanup:
5909
    virTypedParameterArrayClear(params, nparams);
5910 5911 5912 5913 5914
    VIR_FREE(params);
    virDomainFree(dom);
    return ret;
}

5915 5916 5917 5918
/*
 * "memtune" command
 */
static const vshCmdInfo info_memtune[] = {
5919 5920 5921
    {"help", N_("Get or set memory parameters")},
    {"desc", N_("Get or set the current memory parameters for a guest" \
                " domain.\n" \
5922 5923 5924 5925 5926 5927 5928
                "    To get the memory parameters use following command: \n\n" \
                "    virsh # memtune <domain>")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_memtune[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
5929
    {"hard-limit", VSH_OT_INT, VSH_OFLAG_NONE,
5930
     N_("Max memory in kilobytes")},
5931
    {"soft-limit", VSH_OT_INT, VSH_OFLAG_NONE,
5932
     N_("Memory during contention in kilobytes")},
5933
    {"swap-hard-limit", VSH_OT_INT, VSH_OFLAG_NONE,
5934
     N_("Max memory plus swap in kilobytes")},
5935
    {"min-guarantee", VSH_OT_INT, VSH_OFLAG_NONE,
5936
     N_("Min guaranteed memory in kilobytes")},
5937 5938 5939
    {"config", VSH_OT_BOOL, 0, N_("affect next boot")},
    {"live", VSH_OT_BOOL, 0, N_("affect running domain")},
    {"current", VSH_OT_BOOL, 0, N_("affect current domain")},
5940 5941 5942
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
5943
static bool
5944 5945 5946
cmdMemtune(vshControl * ctl, const vshCmd * cmd)
{
    virDomainPtr dom;
5947 5948
    long long hard_limit = 0, soft_limit = 0, swap_hard_limit = 0;
    long long min_guarantee = 0;
5949 5950
    int nparams = 0;
    unsigned int i = 0;
5951
    virTypedParameterPtr params = NULL, temp = NULL;
E
Eric Blake 已提交
5952
    bool ret = false;
5953 5954 5955 5956 5957 5958 5959 5960 5961 5962
    unsigned int flags = 0;
    int current = vshCommandOptBool(cmd, "current");
    int config = vshCommandOptBool(cmd, "config");
    int live = vshCommandOptBool(cmd, "live");

    if (current) {
        if (live || config) {
            vshError(ctl, "%s", _("--current must be specified exclusively"));
            return false;
        }
5963
        flags = VIR_DOMAIN_AFFECT_CURRENT;
5964 5965
    } else {
        if (config)
5966
            flags |= VIR_DOMAIN_AFFECT_CONFIG;
5967
        if (live)
5968
            flags |= VIR_DOMAIN_AFFECT_LIVE;
5969
    }
5970 5971

    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
5972
        return false;
5973 5974

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
5975
        return false;
5976

5977 5978 5979 5980 5981 5982 5983 5984 5985
    if (vshCommandOptLongLong(cmd, "hard-limit", &hard_limit) < 0 ||
        vshCommandOptLongLong(cmd, "soft-limit", &soft_limit) < 0 ||
        vshCommandOptLongLong(cmd, "swap-hard-limit", &swap_hard_limit) < 0 ||
        vshCommandOptLongLong(cmd, "min-guarantee", &min_guarantee) < 0) {
        vshError(ctl, "%s",
                 _("Unable to parse integer parameter"));
        goto cleanup;
    }

5986 5987 5988 5989 5990 5991 5992 5993 5994
    if (hard_limit)
        nparams++;

    if (soft_limit)
        nparams++;

    if (swap_hard_limit)
        nparams++;

5995 5996 5997
    if (min_guarantee)
        nparams++;

5998 5999
    if (nparams == 0) {
        /* get the number of memory parameters */
6000
        if (virDomainGetMemoryParameters(dom, NULL, &nparams, flags) != 0) {
6001 6002 6003 6004 6005
            vshError(ctl, "%s",
                     _("Unable to get number of memory parameters"));
            goto cleanup;
        }

6006 6007
        if (nparams == 0) {
            /* nothing to output */
E
Eric Blake 已提交
6008
            ret = true;
6009 6010 6011
            goto cleanup;
        }

6012
        /* now go get all the memory parameters */
E
Eric Blake 已提交
6013
        params = vshCalloc(ctl, nparams, sizeof(*params));
6014
        if (virDomainGetMemoryParameters(dom, params, &nparams, flags) != 0) {
6015 6016 6017 6018 6019
            vshError(ctl, "%s", _("Unable to get memory parameters"));
            goto cleanup;
        }

        for (i = 0; i < nparams; i++) {
6020 6021 6022 6023 6024 6025 6026
            if (params[i].type == VIR_TYPED_PARAM_ULLONG &&
                params[i].value.ul == VIR_DOMAIN_MEMORY_PARAM_UNLIMITED) {
                vshPrint(ctl, "%-15s: %s\n", params[i].field, _("unlimited"));
            } else {
                char *str = vshGetTypedParamValue(ctl, &params[i]);
                vshPrint(ctl, "%-15s: %s\n", params[i].field, str);
                VIR_FREE(str);
6027 6028 6029
            }
        }

E
Eric Blake 已提交
6030
        ret = true;
6031 6032
    } else {
        /* set the memory parameters */
E
Eric Blake 已提交
6033
        params = vshCalloc(ctl, nparams, sizeof(*params));
6034 6035 6036

        for (i = 0; i < nparams; i++) {
            temp = &params[i];
6037
            temp->type = VIR_TYPED_PARAM_ULLONG;
6038 6039 6040 6041 6042 6043 6044 6045 6046 6047 6048 6049 6050 6051 6052 6053 6054 6055 6056

            /*
             * Some magic here, this is used to fill the params structure with
             * the valid arguments passed, after filling the particular
             * argument we purposely make them 0, so on the next pass it goes
             * to the next valid argument and so on.
             */
            if (soft_limit) {
                temp->value.ul = soft_limit;
                strncpy(temp->field, VIR_DOMAIN_MEMORY_SOFT_LIMIT,
                        sizeof(temp->field));
                soft_limit = 0;
            } else if (hard_limit) {
                temp->value.ul = hard_limit;
                strncpy(temp->field, VIR_DOMAIN_MEMORY_HARD_LIMIT,
                        sizeof(temp->field));
                hard_limit = 0;
            } else if (swap_hard_limit) {
                temp->value.ul = swap_hard_limit;
6057
                strncpy(temp->field, VIR_DOMAIN_MEMORY_SWAP_HARD_LIMIT,
6058 6059
                        sizeof(temp->field));
                swap_hard_limit = 0;
6060 6061 6062 6063 6064
            } else if (min_guarantee) {
                temp->value.ul = min_guarantee;
                strncpy(temp->field, VIR_DOMAIN_MEMORY_MIN_GUARANTEE,
                        sizeof(temp->field));
                min_guarantee = 0;
6065
            }
6066 6067 6068 6069

            /* If the user has passed -1, we interpret it as unlimited */
            if (temp->value.ul == -1)
                temp->value.ul = VIR_DOMAIN_MEMORY_PARAM_UNLIMITED;
6070
        }
6071
        if (virDomainSetMemoryParameters(dom, params, nparams, flags) != 0)
6072
            vshError(ctl, "%s", _("Unable to change memory parameters"));
6073
        else
E
Eric Blake 已提交
6074
            ret = true;
6075 6076 6077
    }

  cleanup:
E
Eric Blake 已提交
6078
    VIR_FREE(params);
6079 6080 6081 6082
    virDomainFree(dom);
    return ret;
}

H
Hu Tao 已提交
6083 6084 6085 6086 6087 6088 6089 6090 6091 6092 6093 6094 6095 6096 6097 6098 6099 6100 6101 6102 6103 6104 6105 6106 6107 6108 6109 6110 6111 6112 6113 6114 6115 6116 6117 6118 6119 6120 6121 6122 6123 6124 6125 6126 6127 6128 6129 6130 6131 6132 6133 6134 6135 6136 6137 6138 6139 6140 6141 6142 6143 6144 6145 6146 6147 6148 6149 6150 6151 6152 6153 6154 6155 6156 6157 6158 6159 6160 6161 6162 6163 6164 6165 6166 6167 6168 6169 6170 6171 6172 6173 6174 6175 6176 6177 6178 6179 6180 6181 6182 6183 6184 6185 6186 6187 6188 6189 6190 6191 6192 6193 6194 6195 6196 6197 6198 6199 6200 6201 6202 6203 6204 6205 6206 6207 6208 6209 6210 6211 6212 6213 6214 6215 6216 6217 6218 6219 6220 6221 6222 6223 6224 6225 6226 6227 6228 6229 6230 6231 6232 6233 6234 6235 6236 6237 6238 6239 6240
/*
 * "numatune" command
 */
static const vshCmdInfo info_numatune[] = {
    {"help", N_("Get or set numa parameters")},
    {"desc", N_("Get or set the current numa parameters for a guest" \
                " domain.\n" \
                "    To get the numa parameters use following command: \n\n" \
                "    virsh # numatune <domain>")},
    {NULL, NULL}

};

static const vshCmdOptDef opts_numatune[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"mode", VSH_OT_DATA, VSH_OFLAG_NONE,
     N_("NUMA mode, one of strict, preferred and interleave")},
    {"nodeset", VSH_OT_DATA, VSH_OFLAG_NONE,
     N_("NUMA node selections to set")},
    {"config", VSH_OT_BOOL, 0, N_("affect next boot")},
    {"live", VSH_OT_BOOL, 0, N_("affect running domain")},
    {"current", VSH_OT_BOOL, 0, N_("affect current domain")},
    {NULL, 0, 0, NULL}
};

static bool
cmdNumatune(vshControl * ctl, const vshCmd * cmd)
{
    virDomainPtr dom;
    int nparams = 0;
    unsigned int i = 0;
    virTypedParameterPtr params = NULL, temp = NULL;
    const char *nodeset = NULL;
    bool ret = false;
    unsigned int flags = 0;
    int current = vshCommandOptBool(cmd, "current");
    int config = vshCommandOptBool(cmd, "config");
    int live = vshCommandOptBool(cmd, "live");
    const char *mode = NULL;

    if (current) {
        if (live || config) {
            vshError(ctl, "%s", _("--current must be specified exclusively"));
            return false;
        }
        flags = VIR_DOMAIN_AFFECT_CURRENT;
    } else {
        if (config)
            flags |= VIR_DOMAIN_AFFECT_CONFIG;
        if (live)
            flags |= VIR_DOMAIN_AFFECT_LIVE;
    }

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

    if (vshCommandOptString(cmd, "nodeset", &nodeset) < 0) {
        vshError(ctl, "%s", _("Unable to parse nodeset."));
        virDomainFree(dom);
        return false;
    }
    if (nodeset)
        nparams++;
    if (vshCommandOptString(cmd, "mode", &mode) < 0) {
        vshError(ctl, "%s", _("Unable to parse mode."));
        virDomainFree(dom);
        return false;
    }
    if (mode)
        nparams++;

    if (nparams == 0) {
        /* get the number of numa parameters */
        if (virDomainGetNumaParameters(dom, NULL, &nparams, flags) != 0) {
            vshError(ctl, "%s",
                     _("Unable to get number of memory parameters"));
            goto cleanup;
        }

        if (nparams == 0) {
            /* nothing to output */
            ret = true;
            goto cleanup;
        }

        /* now go get all the numa parameters */
        params = vshCalloc(ctl, nparams, sizeof(*params));
        if (virDomainGetNumaParameters(dom, params, &nparams, flags) != 0) {
            vshError(ctl, "%s", _("Unable to get numa parameters"));
            goto cleanup;
        }

        for (i = 0; i < nparams; i++) {
            if (params[i].type == VIR_TYPED_PARAM_INT &&
                STREQ(params[i].field, VIR_DOMAIN_NUMA_MODE)) {
                vshPrint(ctl, "%-15s: %s\n", params[i].field,
                         virDomainNumatuneMemModeTypeToString(params[i].value.i));
            } else {
                char *str = vshGetTypedParamValue(ctl, &params[i]);
                vshPrint(ctl, "%-15s: %s\n", params[i].field, str);
                VIR_FREE(str);
            }
        }

        ret = true;
    } else {
        /* set the numa parameters */
        params = vshCalloc(ctl, nparams, sizeof(*params));

        for (i = 0; i < nparams; i++) {
            temp = &params[i];

            /*
             * Some magic here, this is used to fill the params structure with
             * the valid arguments passed, after filling the particular
             * argument we purposely make them 0, so on the next pass it goes
             * to the next valid argument and so on.
             */
            if (mode) {
                /* Accept string or integer, in case server
                 * understands newer integer than what strings we were
                 * compiled with */
                if ((temp->value.i =
                    virDomainNumatuneMemModeTypeFromString(mode)) < 0 &&
                    virStrToLong_i(mode, NULL, 0, &temp->value.i) < 0) {
                    vshError(ctl, _("Invalid mode: %s"), mode);
                    goto cleanup;
                }
                if (!virStrcpy(temp->field, VIR_DOMAIN_NUMA_MODE,
                               sizeof(temp->field)))
                    goto cleanup;
                temp->type = VIR_TYPED_PARAM_INT;
                mode = NULL;
            } else if (nodeset) {
                temp->value.s = vshStrdup(ctl, nodeset);
                temp->type = VIR_TYPED_PARAM_STRING;
                if (!virStrcpy(temp->field, VIR_DOMAIN_NUMA_NODESET,
                               sizeof(temp->field)))
                    goto cleanup;
                nodeset = NULL;
            }
        }
        if (virDomainSetNumaParameters(dom, params, nparams, flags) != 0)
            vshError(ctl, "%s", _("Unable to change numa parameters"));
        else
            ret = true;
    }

  cleanup:
    virTypedParameterArrayClear(params, nparams);
    VIR_FREE(params);
    virDomainFree(dom);
    return ret;
}

6241 6242 6243
/*
 * "nodeinfo" command
 */
6244
static const vshCmdInfo info_nodeinfo[] = {
6245 6246
    {"help", N_("node information")},
    {"desc", N_("Returns basic information about the node.")},
6247 6248 6249
    {NULL, NULL}
};

E
Eric Blake 已提交
6250
static bool
6251
cmdNodeinfo(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
6252 6253
{
    virNodeInfo info;
6254

6255
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
6256
        return false;
6257 6258

    if (virNodeGetInfo(ctl->conn, &info) < 0) {
6259
        vshError(ctl, "%s", _("failed to get node information"));
E
Eric Blake 已提交
6260
        return false;
6261 6262 6263 6264 6265 6266 6267 6268 6269 6270
    }
    vshPrint(ctl, "%-20s %s\n", _("CPU model:"), info.model);
    vshPrint(ctl, "%-20s %d\n", _("CPU(s):"), info.cpus);
    vshPrint(ctl, "%-20s %d MHz\n", _("CPU frequency:"), info.mhz);
    vshPrint(ctl, "%-20s %d\n", _("CPU socket(s):"), info.sockets);
    vshPrint(ctl, "%-20s %d\n", _("Core(s) per socket:"), info.cores);
    vshPrint(ctl, "%-20s %d\n", _("Thread(s) per core:"), info.threads);
    vshPrint(ctl, "%-20s %d\n", _("NUMA cell(s):"), info.nodes);
    vshPrint(ctl, "%-20s %lu kB\n", _("Memory size:"), info.memory);

E
Eric Blake 已提交
6271
    return true;
6272 6273
}

6274 6275 6276 6277 6278
/*
 * "nodecpustats" command
 */
static const vshCmdInfo info_nodecpustats[] = {
    {"help", N_("Prints cpu stats of the node.")},
6279
    {"desc", N_("Returns cpu stats of the node, in nanoseconds.")},
6280 6281 6282 6283 6284 6285 6286 6287 6288 6289
    {NULL, NULL}
};

static const vshCmdOptDef opts_node_cpustats[] = {
    {"cpu", VSH_OT_INT, 0, N_("prints specified cpu statistics only.")},
    {"percent", VSH_OT_BOOL, 0, N_("prints by percentage during 1 second.")},
    {NULL, 0, 0, NULL}
};

static bool
6290
cmdNodeCpuStats(vshControl *ctl, const vshCmd *cmd)
6291 6292 6293 6294
{
    int i, j;
    bool flag_utilization = false;
    bool flag_percent = vshCommandOptBool(cmd, "percent");
6295 6296
    int cpuNum = VIR_NODE_CPU_STATS_ALL_CPUS;
    virNodeCPUStatsPtr params;
6297 6298 6299 6300 6301 6302 6303 6304 6305 6306 6307 6308 6309 6310 6311 6312 6313 6314 6315 6316 6317 6318 6319 6320 6321 6322 6323 6324 6325 6326 6327 6328 6329 6330 6331 6332 6333 6334 6335 6336 6337 6338 6339
    int nparams = 0;
    bool ret = false;
    struct cpu_stats {
        unsigned long long user;
        unsigned long long sys;
        unsigned long long idle;
        unsigned long long iowait;
        unsigned long long util;
    } cpu_stats[2];
    double user_time, sys_time, idle_time, iowait_time, total_time;
    double usage;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (vshCommandOptInt(cmd, "cpu", &cpuNum) < 0) {
        vshError(ctl, "%s", _("Invalid value of cpuNum"));
        return false;
    }

    if (virNodeGetCPUStats(ctl->conn, cpuNum, NULL, &nparams, 0) != 0) {
        vshError(ctl, "%s",
                 _("Unable to get number of cpu stats"));
        return false;
    }
    if (nparams == 0) {
        /* nothing to output */
        return true;
    }

    memset(cpu_stats, 0, sizeof(cpu_stats));
    params = vshCalloc(ctl, nparams, sizeof(*params));

    i = 0;
    do {
        if (virNodeGetCPUStats(ctl->conn, cpuNum, params, &nparams, 0) != 0) {
            vshError(ctl, "%s", _("Unable to get node cpu stats"));
            goto cleanup;
        }

        for (j = 0; j < nparams; j++) {
            unsigned long long value = params[j].value;

6340
            if (STREQ(params[j].field, VIR_NODE_CPU_STATS_KERNEL)) {
6341
                cpu_stats[i].sys = value;
6342
            } else if (STREQ(params[j].field, VIR_NODE_CPU_STATS_USER)) {
6343
                cpu_stats[i].user = value;
6344
            } else if (STREQ(params[j].field, VIR_NODE_CPU_STATS_IDLE)) {
6345
                cpu_stats[i].idle = value;
6346
            } else if (STREQ(params[j].field, VIR_NODE_CPU_STATS_IOWAIT)) {
6347
                cpu_stats[i].iowait = value;
6348
            } else if (STREQ(params[j].field, VIR_NODE_CPU_STATS_UTILIZATION)) {
6349 6350 6351 6352 6353 6354 6355 6356 6357 6358 6359 6360 6361 6362
                cpu_stats[i].util = value;
                flag_utilization = true;
            }
        }

        if (flag_utilization || !flag_percent)
            break;

        i++;
        sleep(1);
    } while (i < 2);

    if (!flag_percent) {
        if (!flag_utilization) {
6363
            vshPrint(ctl, "%-15s %20llu\n", _("user:"), cpu_stats[0].user);
6364
            vshPrint(ctl, "%-15s %20llu\n", _("system:"), cpu_stats[0].sys);
6365
            vshPrint(ctl, "%-15s %20llu\n", _("idle:"), cpu_stats[0].idle);
6366 6367 6368 6369 6370 6371 6372
            vshPrint(ctl, "%-15s %20llu\n", _("iowait:"), cpu_stats[0].iowait);
        }
    } else {
        if (flag_utilization) {
            usage = cpu_stats[0].util;

            vshPrint(ctl, "%-15s %5.1lf%%\n", _("usage:"), usage);
6373
            vshPrint(ctl, "%-15s %5.1lf%%\n", _("idle:"), 100 - usage);
6374 6375 6376 6377 6378 6379 6380 6381 6382 6383 6384 6385
        } else {
            user_time   = cpu_stats[1].user   - cpu_stats[0].user;
            sys_time    = cpu_stats[1].sys    - cpu_stats[0].sys;
            idle_time   = cpu_stats[1].idle   - cpu_stats[0].idle;
            iowait_time = cpu_stats[1].iowait - cpu_stats[0].iowait;
            total_time  = user_time + sys_time + idle_time + iowait_time;

            usage = (user_time + sys_time) / total_time * 100;

            vshPrint(ctl, "%-15s %5.1lf%%\n",
                     _("usage:"), usage);
            vshPrint(ctl, "%-15s %5.1lf%%\n",
6386
                     _("user:"), user_time / total_time * 100);
6387
            vshPrint(ctl, "%-15s %5.1lf%%\n",
6388
                     _("system:"), sys_time  / total_time * 100);
6389
            vshPrint(ctl, "%-15s %5.1lf%%\n",
6390
                     _("idle:"), idle_time     / total_time * 100);
6391 6392 6393 6394 6395 6396 6397 6398 6399 6400 6401 6402
            vshPrint(ctl, "%-15s %5.1lf%%\n",
                     _("iowait:"), iowait_time   / total_time * 100);
        }
    }

    ret = true;

  cleanup:
    VIR_FREE(params);
    return ret;
}

6403 6404 6405 6406 6407 6408 6409 6410 6411 6412 6413 6414 6415 6416 6417 6418 6419 6420 6421
/*
 * "nodememstats" command
 */
static const vshCmdInfo info_nodememstats[] = {
    {"help", N_("Prints memory stats of the node.")},
    {"desc", N_("Returns memory stats of the node, in kilobytes.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_node_memstats[] = {
    {"cell", VSH_OT_INT, 0, N_("prints specified cell statistics only.")},
    {NULL, 0, 0, NULL}
};

static bool
cmdNodeMemStats(vshControl *ctl, const vshCmd *cmd)
{
    int nparams = 0;
    unsigned int i = 0;
6422 6423
    int cellNum = VIR_NODE_MEMORY_STATS_ALL_CELLS;
    virNodeMemoryStatsPtr params = NULL;
6424 6425 6426 6427 6428 6429 6430 6431 6432 6433 6434 6435 6436 6437 6438 6439 6440 6441 6442 6443 6444 6445 6446 6447 6448 6449 6450 6451 6452 6453 6454 6455 6456 6457 6458 6459 6460 6461 6462 6463
    bool ret = false;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (vshCommandOptInt(cmd, "cell", &cellNum) < 0) {
        vshError(ctl, "%s", _("Invalid value of cellNum"));
        return false;
    }

    /* get the number of memory parameters */
    if (virNodeGetMemoryStats(ctl->conn, cellNum, NULL, &nparams, 0) != 0) {
        vshError(ctl, "%s",
                 _("Unable to get number of memory stats"));
        goto cleanup;
    }

    if (nparams == 0) {
        /* nothing to output */
        ret = true;
        goto cleanup;
    }

    /* now go get all the memory parameters */
    params = vshCalloc(ctl, nparams, sizeof(*params));
    if (virNodeGetMemoryStats(ctl->conn, cellNum, params, &nparams, 0) != 0) {
        vshError(ctl, "%s", _("Unable to get memory stats"));
        goto cleanup;
    }

    for (i = 0; i < nparams; i++)
        vshPrint(ctl, "%-7s: %20llu kB\n", params[i].field, params[i].value);

    ret = true;

  cleanup:
    VIR_FREE(params);
    return ret;
}

6464 6465 6466 6467 6468 6469 6470 6471 6472 6473 6474 6475 6476 6477 6478 6479 6480 6481 6482 6483 6484 6485 6486 6487 6488 6489 6490 6491 6492
/*
 * "nodesuspend" command
 */
static const vshCmdInfo info_nodesuspend[] = {
    {"help", N_("suspend the host node for a given time duration")},
    {"desc", N_("Suspend the host node for a given time duration "
                               "and attempt to resume thereafter.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_node_suspend[] = {
    {"target", VSH_OT_DATA, VSH_OFLAG_REQ, N_("mem(Suspend-to-RAM), "
                                               "disk(Suspend-to-Disk), hybrid(Hybrid-Suspend)")},
    {"duration", VSH_OT_INT, VSH_OFLAG_REQ, N_("Suspend duration in seconds")},
    {"flags", VSH_OT_INT, VSH_OFLAG_NONE, N_("Suspend flags, 0 for default")},
    {NULL, 0, 0, NULL}
};

static bool
cmdNodeSuspend(vshControl *ctl, const vshCmd *cmd)
{
    const char *target = NULL;
    unsigned int suspendTarget;
    long long duration;
    unsigned int flags = 0;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

6493 6494
    if (vshCommandOptString(cmd, "target", &target) < 0) {
        vshError(ctl, _("Invalid target argument"));
6495
        return false;
6496
    }
6497

6498 6499
    if (vshCommandOptLongLong(cmd, "duration", &duration) < 0) {
        vshError(ctl, _("Invalid duration argument"));
6500
        return false;
6501
    }
6502

6503 6504
    if (vshCommandOptUInt(cmd, "flags", &flags) < 0) {
        vshError(ctl, _("Invalid flags argument"));
6505
        return false;
6506
    }
6507 6508 6509 6510 6511 6512 6513 6514 6515 6516 6517 6518 6519 6520 6521 6522 6523 6524 6525 6526 6527 6528 6529 6530 6531 6532

    if (STREQ(target, "mem"))
        suspendTarget = VIR_NODE_SUSPEND_TARGET_MEM;
    else if (STREQ(target, "disk"))
        suspendTarget = VIR_NODE_SUSPEND_TARGET_DISK;
    else if (STREQ(target, "hybrid"))
        suspendTarget = VIR_NODE_SUSPEND_TARGET_HYBRID;
    else {
        vshError(ctl, "%s", _("Invalid target"));
        return false;
    }

    if (duration <= 0) {
        vshError(ctl, "%s", _("Invalid duration"));
        return false;
    }

    if (virNodeSuspendForDuration(ctl->conn, suspendTarget, duration,
                                  flags) < 0) {
        vshError(ctl, "%s", _("The host was not suspended"));
        return false;
    }
    return true;
}


6533 6534 6535
/*
 * "capabilities" command
 */
6536
static const vshCmdInfo info_capabilities[] = {
6537 6538
    {"help", N_("capabilities")},
    {"desc", N_("Returns capabilities of hypervisor/driver.")},
6539 6540 6541
    {NULL, NULL}
};

E
Eric Blake 已提交
6542
static bool
6543
cmdCapabilities (vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
6544 6545 6546
{
    char *caps;

6547
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
6548
        return false;
6549 6550

    if ((caps = virConnectGetCapabilities (ctl->conn)) == NULL) {
6551
        vshError(ctl, "%s", _("failed to get capabilities"));
E
Eric Blake 已提交
6552
        return false;
6553 6554
    }
    vshPrint (ctl, "%s\n", caps);
6555
    VIR_FREE(caps);
6556

E
Eric Blake 已提交
6557
    return true;
6558 6559
}

6560 6561 6562
/*
 * "dumpxml" command
 */
6563
static const vshCmdInfo info_dumpxml[] = {
6564 6565
    {"help", N_("domain information in XML")},
    {"desc", N_("Output the domain information as an XML dump to stdout.")},
6566
    {NULL, NULL}
6567 6568
};

6569
static const vshCmdOptDef opts_dumpxml[] = {
6570 6571 6572
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"inactive", VSH_OT_BOOL, 0, N_("show inactive defined XML")},
    {"security-info", VSH_OT_BOOL, 0, N_("include security sensitive information in XML dump")},
6573
    {"update-cpu", VSH_OT_BOOL, 0, N_("update guest CPU according to host CPU")},
6574
    {NULL, 0, 0, NULL}
6575 6576
};

E
Eric Blake 已提交
6577
static bool
6578
cmdDumpXML(vshControl *ctl, const vshCmd *cmd)
6579
{
6580
    virDomainPtr dom;
E
Eric Blake 已提交
6581
    bool ret = true;
6582
    char *dump;
E
Eric Blake 已提交
6583
    unsigned int flags = 0;
6584 6585
    int inactive = vshCommandOptBool(cmd, "inactive");
    int secure = vshCommandOptBool(cmd, "security-info");
6586
    int update = vshCommandOptBool(cmd, "update-cpu");
6587 6588 6589 6590 6591

    if (inactive)
        flags |= VIR_DOMAIN_XML_INACTIVE;
    if (secure)
        flags |= VIR_DOMAIN_XML_SECURE;
6592 6593
    if (update)
        flags |= VIR_DOMAIN_XML_UPDATE_CPU;
6594

6595
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
6596
        return false;
6597

J
Jim Meyering 已提交
6598
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
6599
        return false;
6600

6601
    dump = virDomainGetXMLDesc(dom, flags);
6602
    if (dump != NULL) {
6603
        vshPrint(ctl, "%s", dump);
6604
        VIR_FREE(dump);
6605
    } else {
E
Eric Blake 已提交
6606
        ret = false;
6607
    }
6608

6609 6610 6611 6612
    virDomainFree(dom);
    return ret;
}

6613 6614 6615 6616
/*
 * "domxml-from-native" command
 */
static const vshCmdInfo info_domxmlfromnative[] = {
6617 6618
    {"help", N_("Convert native config to domain XML")},
    {"desc", N_("Convert native guest configuration format to domain XML format.")},
6619 6620 6621 6622
    {NULL, NULL}
};

static const vshCmdOptDef opts_domxmlfromnative[] = {
6623 6624
    {"format", VSH_OT_DATA, VSH_OFLAG_REQ, N_("source config data format")},
    {"config", VSH_OT_DATA, VSH_OFLAG_REQ, N_("config data file to import from")},
6625 6626 6627
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
6628
static bool
6629 6630
cmdDomXMLFromNative(vshControl *ctl, const vshCmd *cmd)
{
E
Eric Blake 已提交
6631
    bool ret = true;
6632 6633
    const char *format = NULL;
    const char *configFile = NULL;
6634 6635
    char *configData;
    char *xmlData;
E
Eric Blake 已提交
6636
    unsigned int flags = 0;
6637

6638
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
6639
        return false;
6640

6641 6642
    if (vshCommandOptString(cmd, "format", &format) < 0 ||
        vshCommandOptString(cmd, "config", &configFile) < 0)
E
Eric Blake 已提交
6643
        return false;
6644

6645
    if (virFileReadAll(configFile, 1024*1024, &configData) < 0)
E
Eric Blake 已提交
6646
        return false;
6647 6648 6649

    xmlData = virConnectDomainXMLFromNative(ctl->conn, format, configData, flags);
    if (xmlData != NULL) {
6650
        vshPrint(ctl, "%s", xmlData);
6651
        VIR_FREE(xmlData);
6652
    } else {
E
Eric Blake 已提交
6653
        ret = false;
6654 6655
    }

6656
    VIR_FREE(configData);
6657 6658 6659 6660 6661 6662 6663
    return ret;
}

/*
 * "domxml-to-native" command
 */
static const vshCmdInfo info_domxmltonative[] = {
6664 6665
    {"help", N_("Convert domain XML to native config")},
    {"desc", N_("Convert domain XML config to a native guest configuration format.")},
6666 6667 6668 6669
    {NULL, NULL}
};

static const vshCmdOptDef opts_domxmltonative[] = {
6670 6671
    {"format", VSH_OT_DATA, VSH_OFLAG_REQ, N_("target config data type format")},
    {"xml", VSH_OT_DATA, VSH_OFLAG_REQ, N_("xml data file to export from")},
6672 6673 6674
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
6675
static bool
6676 6677
cmdDomXMLToNative(vshControl *ctl, const vshCmd *cmd)
{
E
Eric Blake 已提交
6678
    bool ret = true;
6679 6680
    const char *format = NULL;
    const char *xmlFile = NULL;
6681 6682
    char *configData;
    char *xmlData;
E
Eric Blake 已提交
6683
    unsigned int flags = 0;
6684

6685
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
6686
        return false;
6687

6688 6689
    if (vshCommandOptString(cmd, "format", &format) < 0
        || vshCommandOptString(cmd, "xml", &xmlFile) < 0)
E
Eric Blake 已提交
6690
        return false;
6691

6692
    if (virFileReadAll(xmlFile, 1024*1024, &xmlData) < 0)
E
Eric Blake 已提交
6693
        return false;
6694 6695 6696

    configData = virConnectDomainXMLToNative(ctl->conn, format, xmlData, flags);
    if (configData != NULL) {
6697
        vshPrint(ctl, "%s", configData);
6698
        VIR_FREE(configData);
6699
    } else {
E
Eric Blake 已提交
6700
        ret = false;
6701 6702
    }

6703
    VIR_FREE(xmlData);
6704 6705 6706
    return ret;
}

K
Karel Zak 已提交
6707
/*
K
Karel Zak 已提交
6708
 * "domname" command
K
Karel Zak 已提交
6709
 */
6710
static const vshCmdInfo info_domname[] = {
6711
    {"help", N_("convert a domain id or UUID to domain name")},
6712
    {"desc", ""},
6713
    {NULL, NULL}
K
Karel Zak 已提交
6714 6715
};

6716
static const vshCmdOptDef opts_domname[] = {
6717
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain id or uuid")},
6718
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
6719 6720
};

E
Eric Blake 已提交
6721
static bool
6722
cmdDomname(vshControl *ctl, const vshCmd *cmd)
6723
{
K
Karel Zak 已提交
6724 6725
    virDomainPtr dom;

6726
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
6727
        return false;
J
Jim Meyering 已提交
6728
    if (!(dom = vshCommandOptDomainBy(ctl, cmd, NULL,
6729
                                      VSH_BYID|VSH_BYUUID)))
E
Eric Blake 已提交
6730
        return false;
6731

K
Karel Zak 已提交
6732 6733
    vshPrint(ctl, "%s\n", virDomainGetName(dom));
    virDomainFree(dom);
E
Eric Blake 已提交
6734
    return true;
K
Karel Zak 已提交
6735 6736 6737
}

/*
K
Karel Zak 已提交
6738
 * "domid" command
K
Karel Zak 已提交
6739
 */
6740
static const vshCmdInfo info_domid[] = {
6741
    {"help", N_("convert a domain name or UUID to domain id")},
6742
    {"desc", ""},
6743
    {NULL, NULL}
K
Karel Zak 已提交
6744 6745
};

6746
static const vshCmdOptDef opts_domid[] = {
6747
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name or uuid")},
6748
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
6749 6750
};

E
Eric Blake 已提交
6751
static bool
6752
cmdDomid(vshControl *ctl, const vshCmd *cmd)
6753
{
6754
    virDomainPtr dom;
6755
    unsigned int id;
K
Karel Zak 已提交
6756

6757
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
6758
        return false;
J
Jim Meyering 已提交
6759
    if (!(dom = vshCommandOptDomainBy(ctl, cmd, NULL,
6760
                                      VSH_BYNAME|VSH_BYUUID)))
E
Eric Blake 已提交
6761
        return false;
6762

6763 6764
    id = virDomainGetID(dom);
    if (id == ((unsigned int)-1))
6765
        vshPrint(ctl, "%s\n", "-");
6766
    else
6767
        vshPrint(ctl, "%d\n", id);
K
Karel Zak 已提交
6768
    virDomainFree(dom);
E
Eric Blake 已提交
6769
    return true;
K
Karel Zak 已提交
6770
}
6771

K
Karel Zak 已提交
6772 6773 6774
/*
 * "domuuid" command
 */
6775
static const vshCmdInfo info_domuuid[] = {
6776
    {"help", N_("convert a domain name or id to domain UUID")},
6777
    {"desc", ""},
K
Karel Zak 已提交
6778 6779 6780
    {NULL, NULL}
};

6781
static const vshCmdOptDef opts_domuuid[] = {
6782
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain id or name")},
K
Karel Zak 已提交
6783 6784 6785
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
6786
static bool
6787
cmdDomuuid(vshControl *ctl, const vshCmd *cmd)
K
Karel Zak 已提交
6788 6789
{
    virDomainPtr dom;
6790
    char uuid[VIR_UUID_STRING_BUFLEN];
K
Karel Zak 已提交
6791

6792
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
6793
        return false;
J
Jim Meyering 已提交
6794
    if (!(dom = vshCommandOptDomainBy(ctl, cmd, NULL,
6795
                                      VSH_BYNAME|VSH_BYID)))
E
Eric Blake 已提交
6796
        return false;
6797

K
Karel Zak 已提交
6798 6799 6800
    if (virDomainGetUUIDString(dom, uuid) != -1)
        vshPrint(ctl, "%s\n", uuid);
    else
6801
        vshError(ctl, "%s", _("failed to get domain UUID"));
6802

6803
    virDomainFree(dom);
E
Eric Blake 已提交
6804
    return true;
K
Karel Zak 已提交
6805 6806
}

6807 6808 6809
/*
 * "migrate" command
 */
6810
static const vshCmdInfo info_migrate[] = {
6811 6812
    {"help", N_("migrate domain to another host")},
    {"desc", N_("Migrate domain to another host.  Add --live for live migration.")},
6813 6814 6815
    {NULL, NULL}
};

6816
static const vshCmdOptDef opts_migrate[] = {
6817 6818 6819 6820 6821 6822 6823
    {"live", VSH_OT_BOOL, 0, N_("live migration")},
    {"p2p", VSH_OT_BOOL, 0, N_("peer-2-peer migration")},
    {"direct", VSH_OT_BOOL, 0, N_("direct migration")},
    {"tunnelled", VSH_OT_BOOL, 0, N_("tunnelled migration")},
    {"persistent", VSH_OT_BOOL, 0, N_("persist VM on destination")},
    {"undefinesource", VSH_OT_BOOL, 0, N_("undefine VM on source")},
    {"suspend", VSH_OT_BOOL, 0, N_("do not restart the domain on the destination host")},
6824 6825
    {"copy-storage-all", VSH_OT_BOOL, 0, N_("migration with non-shared storage with full disk copy")},
    {"copy-storage-inc", VSH_OT_BOOL, 0, N_("migration with non-shared storage with incremental copy (same base image shared between source and destination)")},
6826 6827
    {"change-protection", VSH_OT_BOOL, 0,
     N_("prevent any configuration changes to domain until migration ends)")},
6828
    {"unsafe", VSH_OT_BOOL, 0, N_("force migration even if it may be unsafe")},
W
Wen Congyang 已提交
6829
    {"verbose", VSH_OT_BOOL, 0, N_("display the progress of migration")},
6830
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
6831
    {"desturi", VSH_OT_DATA, VSH_OFLAG_REQ, N_("connection URI of the destination host as seen from the client(normal migration) or source(p2p migration)")},
6832 6833
    {"migrateuri", VSH_OT_DATA, 0, N_("migration URI, usually can be omitted")},
    {"dname", VSH_OT_DATA, 0, N_("rename to new name during migration (if supported)")},
W
Wen Congyang 已提交
6834
    {"timeout", VSH_OT_INT, 0, N_("force guest to suspend if live migration exceeds timeout (in seconds)")},
6835
    {"xml", VSH_OT_STRING, 0, N_("filename containing updated XML for the target")},
6836 6837 6838
    {NULL, 0, 0, NULL}
};

6839 6840
static void
doMigrate (void *opaque)
6841
{
6842
    char ret = '1';
6843
    virDomainPtr dom = NULL;
6844 6845 6846
    const char *desturi = NULL;
    const char *migrateuri = NULL;
    const char *dname = NULL;
E
Eric Blake 已提交
6847
    unsigned int flags = 0;
6848 6849 6850
    vshCtrlData *data = opaque;
    vshControl *ctl = data->ctl;
    const vshCmd *cmd = data->cmd;
6851 6852
    const char *xmlfile = NULL;
    char *xml = NULL;
6853
    sigset_t sigmask, oldsigmask;
6854 6855 6856 6857 6858

    sigemptyset(&sigmask);
    sigaddset(&sigmask, SIGINT);
    if (pthread_sigmask(SIG_BLOCK, &sigmask, &oldsigmask) < 0)
        goto out_sig;
6859

6860
    if (!vshConnectionUsability (ctl, ctl->conn))
6861
        goto out;
6862

J
Jim Meyering 已提交
6863
    if (!(dom = vshCommandOptDomain (ctl, cmd, NULL)))
6864
        goto out;
6865

6866 6867
    if (vshCommandOptString(cmd, "desturi", &desturi) <= 0 ||
        vshCommandOptString(cmd, "migrateuri", &migrateuri) < 0 ||
6868 6869
        vshCommandOptString(cmd, "dname", &dname) < 0) {
        vshError(ctl, "%s", _("missing argument"));
6870
        goto out;
6871
    }
6872

6873 6874 6875 6876 6877
    if (vshCommandOptString(cmd, "xml", &xmlfile) < 0) {
        vshError(ctl, "%s", _("malformed xml argument"));
        goto out;
    }

6878 6879
    if (vshCommandOptBool (cmd, "live"))
        flags |= VIR_MIGRATE_LIVE;
6880 6881
    if (vshCommandOptBool (cmd, "p2p"))
        flags |= VIR_MIGRATE_PEER2PEER;
C
Chris Lalancette 已提交
6882
    if (vshCommandOptBool (cmd, "tunnelled"))
6883
        flags |= VIR_MIGRATE_TUNNELLED;
C
Chris Lalancette 已提交
6884

C
Chris Lalancette 已提交
6885 6886 6887 6888 6889
    if (vshCommandOptBool (cmd, "persistent"))
        flags |= VIR_MIGRATE_PERSIST_DEST;
    if (vshCommandOptBool (cmd, "undefinesource"))
        flags |= VIR_MIGRATE_UNDEFINE_SOURCE;

6890 6891 6892
    if (vshCommandOptBool (cmd, "suspend"))
        flags |= VIR_MIGRATE_PAUSED;

6893 6894 6895 6896 6897 6898
    if (vshCommandOptBool (cmd, "copy-storage-all"))
        flags |= VIR_MIGRATE_NON_SHARED_DISK;

    if (vshCommandOptBool (cmd, "copy-storage-inc"))
        flags |= VIR_MIGRATE_NON_SHARED_INC;

6899 6900
    if (vshCommandOptBool (cmd, "change-protection"))
        flags |= VIR_MIGRATE_CHANGE_PROTECTION;
6901

6902 6903 6904
    if (vshCommandOptBool(cmd, "unsafe"))
        flags |= VIR_MIGRATE_UNSAFE;

6905
    if (xmlfile &&
6906 6907
        virFileReadAll(xmlfile, 8192, &xml) < 0) {
        vshError(ctl, _("file '%s' doesn't exist"), xmlfile);
6908
        goto out;
6909
    }
6910

6911 6912 6913 6914
    if ((flags & VIR_MIGRATE_PEER2PEER) ||
        vshCommandOptBool (cmd, "direct")) {
        /* For peer2peer migration or direct migration we only expect one URI
         * a libvirt URI, or a hypervisor specific URI. */
6915

6916
        if (migrateuri != NULL) {
J
Jim Fehlig 已提交
6917
            vshError(ctl, "%s", _("migrate: Unexpected migrateuri for peer2peer/direct migration"));
6918
            goto out;
6919
        }
6920

6921
        if (virDomainMigrateToURI2(dom, desturi, NULL, xml, flags, dname, 0) == 0)
6922
            ret = '0';
6923 6924 6925 6926 6927 6928
    } else {
        /* For traditional live migration, connect to the destination host directly. */
        virConnectPtr dconn = NULL;
        virDomainPtr ddom = NULL;

        dconn = virConnectOpenAuth (desturi, virConnectAuthPtrDefault, 0);
6929
        if (!dconn) goto out;
6930

6931
        ddom = virDomainMigrate2(dom, dconn, xml, flags, dname, migrateuri, 0);
6932 6933
        if (ddom) {
            virDomainFree(ddom);
6934
            ret = '0';
6935 6936 6937
        }
        virConnectClose (dconn);
    }
6938

6939 6940 6941
out:
    pthread_sigmask(SIG_SETMASK, &oldsigmask, NULL);
out_sig:
6942
    if (dom) virDomainFree (dom);
6943
    VIR_FREE(xml);
6944 6945 6946
    ignore_value(safewrite(data->writefd, &ret, sizeof(ret)));
}

W
Wen Congyang 已提交
6947
static void
6948 6949
print_job_progress(const char *label, unsigned long long remaining,
                   unsigned long long total)
W
Wen Congyang 已提交
6950 6951 6952 6953 6954 6955 6956 6957 6958 6959 6960 6961 6962 6963 6964 6965 6966 6967 6968
{
    int progress;

    if (total == 0)
        /* migration has not been started */
        return;

    if (remaining == 0) {
        /* migration has completed */
        progress = 100;
    } else {
        /* use float to avoid overflow */
        progress = (int)(100.0 - remaining * 100.0 / total);
        if (progress >= 100) {
            /* migration has not completed, do not print [100 %] */
            progress = 99;
        }
    }

6969 6970
    /* see comments in vshError about why we must flush */
    fflush(stdout);
6971
    fprintf(stderr, "\r%s: [%3d %%]", label, progress);
6972
    fflush(stderr);
W
Wen Congyang 已提交
6973 6974
}

6975 6976 6977 6978 6979 6980 6981 6982 6983 6984
static void
vshMigrationTimeout(vshControl *ctl,
                    virDomainPtr dom,
                    void *opaque ATTRIBUTE_UNUSED)
{
    vshDebug(ctl, VSH_ERR_DEBUG, "suspending the domain, "
             "since migration timed out\n");
    virDomainSuspend(dom);
}

E
Eric Blake 已提交
6985
static bool
6986 6987 6988 6989 6990 6991 6992 6993
vshWatchJob(vshControl *ctl,
            virDomainPtr dom,
            bool verbose,
            int pipe_fd,
            int timeout,
            jobWatchTimeoutFunc timeout_func,
            void *opaque,
            const char *label)
6994 6995 6996
{
    struct sigaction sig_action;
    struct sigaction old_sig_action;
6997
    struct pollfd pollfd;
W
Wen Congyang 已提交
6998
    struct timeval start, curr;
6999 7000 7001 7002
    virDomainJobInfo jobinfo;
    int ret = -1;
    char retchar;
    bool functionReturn = false;
7003 7004 7005 7006
    sigset_t sigmask, oldsigmask;

    sigemptyset(&sigmask);
    sigaddset(&sigmask, SIGINT);
7007 7008 7009 7010 7011 7012 7013

    intCaught = 0;
    sig_action.sa_sigaction = vshCatchInt;
    sig_action.sa_flags = SA_SIGINFO;
    sigemptyset(&sig_action.sa_mask);
    sigaction(SIGINT, &sig_action, &old_sig_action);

7014
    pollfd.fd = pipe_fd;
7015 7016 7017
    pollfd.events = POLLIN;
    pollfd.revents = 0;

W
Wen Congyang 已提交
7018
    GETTIMEOFDAY(&start);
W
Wen Congyang 已提交
7019
    while (1) {
7020
repoll:
W
Wen Congyang 已提交
7021 7022
        ret = poll(&pollfd, 1, 500);
        if (ret > 0) {
7023 7024 7025
            if (pollfd.revents & POLLIN &&
                saferead(pipe_fd, &retchar, sizeof(retchar)) > 0 &&
                retchar == '0') {
W
Wen Congyang 已提交
7026 7027
                    if (verbose) {
                        /* print [100 %] */
7028
                        print_job_progress(label, 0, 1);
W
Wen Congyang 已提交
7029
                    }
7030 7031 7032
                    break;
            }
            goto cleanup;
W
Wen Congyang 已提交
7033 7034 7035 7036 7037 7038 7039 7040 7041 7042
        }

        if (ret < 0) {
            if (errno == EINTR) {
                if (intCaught) {
                    virDomainAbortJob(dom);
                    intCaught = 0;
                } else
                    goto repoll;
            }
7043
            goto cleanup;
W
Wen Congyang 已提交
7044 7045
        }

W
Wen Congyang 已提交
7046 7047 7048 7049
        GETTIMEOFDAY(&curr);
        if ( timeout && ((int)(curr.tv_sec - start.tv_sec)  * 1000 + \
                         (int)(curr.tv_usec - start.tv_usec) / 1000) > timeout * 1000 ) {
            /* suspend the domain when migration timeouts. */
7050 7051 7052
            vshDebug(ctl, VSH_ERR_DEBUG, "%s timeout", label);
            if (timeout_func)
                (timeout_func)(ctl, dom, opaque);
W
Wen Congyang 已提交
7053 7054 7055
            timeout = 0;
        }

W
Wen Congyang 已提交
7056 7057 7058 7059 7060
        if (verbose) {
            pthread_sigmask(SIG_BLOCK, &sigmask, &oldsigmask);
            ret = virDomainGetJobInfo(dom, &jobinfo);
            pthread_sigmask(SIG_SETMASK, &oldsigmask, NULL);
            if (ret == 0)
7061
                print_job_progress(label, jobinfo.dataRemaining,
7062
                                   jobinfo.dataTotal);
7063 7064 7065
        }
    }

7066 7067 7068
    functionReturn = true;

cleanup:
7069
    sigaction(SIGINT, &old_sig_action, NULL);
7070 7071 7072 7073 7074 7075 7076 7077 7078 7079 7080 7081 7082 7083 7084 7085 7086 7087 7088 7089 7090 7091 7092 7093 7094 7095 7096 7097 7098 7099 7100 7101 7102 7103 7104 7105 7106 7107 7108 7109 7110 7111 7112 7113 7114 7115 7116 7117 7118 7119 7120 7121 7122 7123 7124
    return functionReturn;
}

static bool
cmdMigrate (vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
    int p[2] = {-1, -1};
    virThread workerThread;
    bool verbose = false;
    bool functionReturn = false;
    int timeout = 0;
    bool live_flag = false;
    vshCtrlData data;

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

    if (vshCommandOptBool (cmd, "verbose"))
        verbose = true;

    if (vshCommandOptBool (cmd, "live"))
        live_flag = true;
    if (vshCommandOptInt(cmd, "timeout", &timeout) > 0) {
        if (! live_flag) {
            vshError(ctl, "%s", _("migrate: Unexpected timeout for offline migration"));
            goto cleanup;
        }

        if (timeout < 1) {
            vshError(ctl, "%s", _("migrate: Invalid timeout"));
            goto cleanup;
        }

        /* Ensure that we can multiply by 1000 without overflowing. */
        if (timeout > INT_MAX / 1000) {
            vshError(ctl, "%s", _("migrate: Timeout is too big"));
            goto cleanup;
        }
    }

    if (pipe(p) < 0)
        goto cleanup;

    data.ctl = ctl;
    data.cmd = cmd;
    data.writefd = p[1];

    if (virThreadCreate(&workerThread,
                        true,
                        doMigrate,
                        &data) < 0)
        goto cleanup;
    functionReturn = vshWatchJob(ctl, dom, verbose, p[0], timeout,
                                 vshMigrationTimeout, NULL, _("Migration"));
7125 7126 7127 7128 7129 7130 7131

    virThreadJoin(&workerThread);

cleanup:
    virDomainFree(dom);
    VIR_FORCE_CLOSE(p[0]);
    VIR_FORCE_CLOSE(p[1]);
W
Wen Congyang 已提交
7132
    return functionReturn;
7133 7134
}

7135 7136 7137 7138 7139 7140 7141 7142 7143 7144 7145
/*
 * "migrate-setmaxdowntime" command
 */
static const vshCmdInfo info_migrate_setmaxdowntime[] = {
    {"help", N_("set maximum tolerable downtime")},
    {"desc", N_("Set maximum tolerable downtime of a domain which is being live-migrated to another host.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_migrate_setmaxdowntime[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
7146
    {"downtime", VSH_OT_INT, VSH_OFLAG_REQ, N_("maximum tolerable downtime (in milliseconds) for migration")},
7147 7148 7149
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
7150
static bool
7151 7152 7153
cmdMigrateSetMaxDowntime(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
7154
    long long downtime = 0;
E
Eric Blake 已提交
7155
    bool ret = false;
7156

7157
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
7158
        return false;
7159 7160

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
7161
        return false;
7162

7163 7164
    if (vshCommandOptLongLong(cmd, "downtime", &downtime) < 0 ||
        downtime < 1) {
7165 7166 7167 7168 7169 7170 7171
        vshError(ctl, "%s", _("migrate: Invalid downtime"));
        goto done;
    }

    if (virDomainMigrateSetMaxDowntime(dom, downtime, 0))
        goto done;

E
Eric Blake 已提交
7172
    ret = true;
7173 7174 7175 7176 7177 7178

done:
    virDomainFree(dom);
    return ret;
}

7179 7180 7181 7182 7183 7184 7185 7186 7187 7188 7189 7190 7191 7192 7193 7194 7195 7196 7197 7198 7199 7200 7201 7202 7203 7204 7205 7206 7207 7208 7209 7210 7211 7212 7213 7214 7215 7216 7217 7218 7219 7220 7221 7222
/*
 * "migrate-setspeed" command
 */
static const vshCmdInfo info_migrate_setspeed[] = {
    {"help", N_("Set the maximum migration bandwidth")},
    {"desc", N_("Set the maximum migration bandwidth (in Mbps) for a domain "
                "which is being migrated to another host.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_migrate_setspeed[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"bandwidth", VSH_OT_INT, VSH_OFLAG_REQ, N_("migration bandwidth limit in Mbps")},
    {NULL, 0, 0, NULL}
};

static bool
cmdMigrateSetMaxSpeed(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
    unsigned long bandwidth = 0;
    bool ret = false;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

    if (vshCommandOptUL(cmd, "bandwidth", &bandwidth) < 0) {
        vshError(ctl, "%s", _("migrate: Invalid bandwidth"));
        goto done;
    }

    if (virDomainMigrateSetMaxSpeed(dom, bandwidth, 0) < 0)
        goto done;

    ret = true;

done:
    virDomainFree(dom);
    return ret;
}

7223 7224 7225 7226 7227 7228 7229 7230 7231 7232 7233 7234 7235 7236 7237 7238 7239 7240 7241 7242 7243 7244 7245 7246 7247 7248 7249 7250 7251 7252 7253 7254 7255 7256 7257 7258 7259 7260 7261
/*
 * "migrate-getspeed" command
 */
static const vshCmdInfo info_migrate_getspeed[] = {
    {"help", N_("Get the maximum migration bandwidth")},
    {"desc", N_("Get the maximum migration bandwidth (in Mbps) for a domain.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_migrate_getspeed[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {NULL, 0, 0, NULL}
};

static bool
cmdMigrateGetMaxSpeed(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
    unsigned long bandwidth;
    bool ret = false;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

    if (virDomainMigrateGetMaxSpeed(dom, &bandwidth, 0) < 0)
        goto done;

    vshPrint(ctl, "%lu\n", bandwidth);

    ret = true;

done:
    virDomainFree(dom);
    return ret;
}

7262 7263 7264 7265 7266 7267 7268 7269 7270 7271 7272 7273 7274 7275 7276 7277 7278
typedef enum {
    VSH_CMD_BLOCK_JOB_ABORT = 0,
    VSH_CMD_BLOCK_JOB_INFO = 1,
    VSH_CMD_BLOCK_JOB_SPEED = 2,
    VSH_CMD_BLOCK_JOB_PULL = 3,
} VSH_CMD_BLOCK_JOB_MODE;

static int
blockJobImpl(vshControl *ctl, const vshCmd *cmd,
              virDomainBlockJobInfoPtr info, int mode)
{
    virDomainPtr dom = NULL;
    const char *name, *path;
    unsigned long bandwidth = 0;
    int ret = -1;

    if (!vshConnectionUsability(ctl, ctl->conn))
7279
        goto cleanup;
7280 7281

    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
7282
        goto cleanup;
7283 7284

    if (vshCommandOptString(cmd, "path", &path) < 0)
7285
        goto cleanup;
7286

7287 7288
    if (vshCommandOptUL(cmd, "bandwidth", &bandwidth) < 0) {
        vshError(ctl, "%s", _("bandwidth must be a number"));
7289
        goto cleanup;
7290
    }
7291 7292 7293 7294 7295 7296 7297 7298 7299 7300

    if (mode == VSH_CMD_BLOCK_JOB_ABORT)
        ret = virDomainBlockJobAbort(dom, path, 0);
    else if (mode == VSH_CMD_BLOCK_JOB_INFO)
        ret = virDomainGetBlockJobInfo(dom, path, info, 0);
    else if (mode == VSH_CMD_BLOCK_JOB_SPEED)
        ret = virDomainBlockJobSetSpeed(dom, path, bandwidth, 0);
    else if (mode == VSH_CMD_BLOCK_JOB_PULL)
        ret = virDomainBlockPull(dom, path, bandwidth, 0);

7301
cleanup:
7302 7303
    if (dom)
        virDomainFree(dom);
7304 7305 7306 7307 7308 7309 7310 7311 7312 7313 7314 7315 7316 7317 7318 7319 7320 7321 7322 7323 7324 7325 7326 7327 7328 7329 7330 7331 7332 7333 7334 7335 7336 7337 7338 7339 7340 7341 7342
    return ret;
}

/*
 * "blockpull" command
 */
static const vshCmdInfo info_block_pull[] = {
    {"help", N_("Populate a disk from its backing image.")},
    {"desc", N_("Populate a disk from its backing image.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_block_pull[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"path", VSH_OT_DATA, VSH_OFLAG_REQ, N_("Fully-qualified path of disk")},
    {"bandwidth", VSH_OT_DATA, VSH_OFLAG_NONE, N_("Bandwidth limit in MB/s")},
    {NULL, 0, 0, NULL}
};

static bool
cmdBlockPull(vshControl *ctl, const vshCmd *cmd)
{
    if (blockJobImpl(ctl, cmd, NULL, VSH_CMD_BLOCK_JOB_PULL) != 0)
        return false;
    return true;
}

/*
 * "blockjobinfo" command
 */
static const vshCmdInfo info_block_job[] = {
    {"help", N_("Manage active block operations.")},
    {"desc", N_("Manage active block operations.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_block_job[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"path", VSH_OT_DATA, VSH_OFLAG_REQ, N_("Fully-qualified path of disk")},
Y
Yuri Chornoivan 已提交
7343
    {"abort", VSH_OT_BOOL, VSH_OFLAG_NONE, N_("Abort the active job on the specified disk")},
7344 7345 7346 7347 7348 7349 7350 7351 7352 7353 7354 7355 7356 7357 7358 7359 7360 7361 7362 7363 7364 7365 7366 7367 7368 7369 7370 7371 7372 7373 7374 7375 7376 7377 7378 7379 7380 7381 7382 7383 7384 7385 7386
    {"info", VSH_OT_BOOL, VSH_OFLAG_NONE, N_("Get active job information for the specified disk")},
    {"bandwidth", VSH_OT_DATA, VSH_OFLAG_NONE, N_("Set the Bandwidth limit in MB/s")},
    {NULL, 0, 0, NULL}
};

static bool
cmdBlockJob(vshControl *ctl, const vshCmd *cmd)
{
    int mode;
    virDomainBlockJobInfo info;
    const char *type;
    int ret;

    if (vshCommandOptBool (cmd, "abort")) {
        mode = VSH_CMD_BLOCK_JOB_ABORT;
    } else if (vshCommandOptBool (cmd, "info")) {
        mode = VSH_CMD_BLOCK_JOB_INFO;
    } else if (vshCommandOptBool (cmd, "bandwidth")) {
        mode = VSH_CMD_BLOCK_JOB_SPEED;
    } else {
        vshError(ctl, "%s",
                 _("One of --abort, --info, or --bandwidth is required"));
        return false;
    }

    ret = blockJobImpl(ctl, cmd, &info, mode);
    if (ret < 0)
        return false;

    if (ret == 0 || mode != VSH_CMD_BLOCK_JOB_INFO)
        return true;

    if (info.type == VIR_DOMAIN_BLOCK_JOB_TYPE_PULL)
        type = "Block Pull";
    else
        type = "Unknown job";

    print_job_progress(type, info.end - info.cur, info.end);
    if (info.bandwidth != 0)
        vshPrint(ctl, "    Bandwidth limit: %lu MB/s\n", info.bandwidth);
    return true;
}

7387 7388 7389 7390 7391 7392 7393 7394 7395 7396 7397 7398 7399 7400 7401 7402 7403 7404 7405 7406 7407 7408 7409 7410 7411 7412 7413 7414 7415 7416 7417 7418 7419 7420 7421 7422 7423 7424 7425 7426 7427 7428 7429 7430 7431 7432 7433 7434 7435 7436 7437 7438 7439 7440 7441 7442 7443
/*
 * "blockresize" command
 */
static const vshCmdInfo info_block_resize[] = {
    {"help", N_("Resize block device of domain.")},
    {"desc", N_("Resize block device of domain.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_block_resize[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"path", VSH_OT_DATA, VSH_OFLAG_REQ, N_("Fully-qualified path of block device")},
    {"size", VSH_OT_INT, VSH_OFLAG_REQ, N_("New size of the block device in kilobytes, "
                                           "the size must be integer")},
    {NULL, 0, 0, NULL}
};

static bool
cmdBlockResize(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    const char *path = NULL;
    unsigned long long size = 0;
    unsigned int flags = 0;
    int ret = false;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (vshCommandOptString(cmd, "path", (const char **) &path) < 0) {
        vshError(ctl, "%s", _("Path must not be empty"));
        return false;
    }

    if (vshCommandOptULongLong(cmd, "size", &size) < 0) {
        vshError(ctl, "%s", _("Unable to parse integer"));
        return false;
    }

    if (size > ULLONG_MAX / 1024) {
        vshError(ctl, _("Size must be less than %llu"), ULLONG_MAX / 1024);
        return false;
    }

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

    if (virDomainBlockResize(dom, path, size, flags) < 0) {
        vshError(ctl, _("Failed to resize block device '%s'"), path);
    } else {
        vshPrint(ctl, _("Block device '%s' is resized"), path);
        ret = true;
    }

    virDomainFree(dom);
    return ret;
}
7444

7445 7446 7447
/*
 * "net-autostart" command
 */
7448
static const vshCmdInfo info_network_autostart[] = {
7449
    {"help", N_("autostart a network")},
7450
    {"desc",
7451
     N_("Configure a network to be automatically started at boot.")},
7452 7453 7454
    {NULL, NULL}
};

7455
static const vshCmdOptDef opts_network_autostart[] = {
7456 7457
    {"network",  VSH_OT_DATA, VSH_OFLAG_REQ, N_("network name or uuid")},
    {"disable", VSH_OT_BOOL, 0, N_("disable autostarting")},
7458 7459 7460
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
7461
static bool
7462
cmdNetworkAutostart(vshControl *ctl, const vshCmd *cmd)
7463 7464
{
    virNetworkPtr network;
7465
    const char *name;
7466 7467
    int autostart;

7468
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
7469
        return false;
7470

J
Jim Meyering 已提交
7471
    if (!(network = vshCommandOptNetwork(ctl, cmd, &name)))
E
Eric Blake 已提交
7472
        return false;
7473 7474 7475 7476

    autostart = !vshCommandOptBool(cmd, "disable");

    if (virNetworkSetAutostart(network, autostart) < 0) {
7477
        if (autostart)
7478
            vshError(ctl, _("failed to mark network %s as autostarted"), name);
7479
        else
7480
            vshError(ctl, _("failed to unmark network %s as autostarted"), name);
7481
        virNetworkFree(network);
E
Eric Blake 已提交
7482
        return false;
7483 7484
    }

7485
    if (autostart)
7486
        vshPrint(ctl, _("Network %s marked as autostarted\n"), name);
7487
    else
7488
        vshPrint(ctl, _("Network %s unmarked as autostarted\n"), name);
7489

L
Laine Stump 已提交
7490
    virNetworkFree(network);
E
Eric Blake 已提交
7491
    return true;
7492
}
K
Karel Zak 已提交
7493

L
Lei Li 已提交
7494 7495 7496 7497 7498 7499 7500 7501 7502 7503 7504 7505 7506 7507 7508 7509 7510 7511 7512 7513 7514 7515 7516 7517 7518 7519 7520 7521 7522 7523 7524 7525 7526 7527 7528 7529 7530 7531 7532 7533 7534 7535 7536 7537 7538 7539 7540 7541 7542 7543 7544 7545 7546 7547 7548 7549 7550 7551 7552 7553 7554 7555 7556 7557 7558 7559 7560 7561 7562 7563 7564 7565 7566 7567 7568 7569 7570 7571 7572 7573 7574 7575 7576 7577 7578 7579 7580 7581 7582 7583 7584 7585 7586 7587 7588 7589 7590 7591 7592 7593 7594 7595 7596 7597 7598 7599 7600 7601 7602 7603 7604 7605 7606 7607 7608 7609 7610 7611 7612 7613 7614 7615 7616 7617 7618 7619 7620 7621 7622 7623 7624 7625 7626 7627 7628 7629 7630 7631
/*
 * "blkdeviotune" command
 */
static const vshCmdInfo info_blkdeviotune[] = {
    {"help", N_("Set or query a block device I/O tuning parameters.")},
    {"desc", N_("Set or query disk I/O parameters such as block throttling.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_blkdeviotune[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"device", VSH_OT_DATA, VSH_OFLAG_REQ, N_("block device")},
    {"total_bytes_sec", VSH_OT_INT, VSH_OFLAG_NONE,
     N_("total throughput limit in bytes per second")},
    {"read_bytes_sec", VSH_OT_INT, VSH_OFLAG_NONE,
     N_("read throughput limit in bytes per second")},
    {"write_bytes_sec", VSH_OT_INT, VSH_OFLAG_NONE,
     N_("write throughput limit in bytes per second")},
    {"total_iops_sec", VSH_OT_INT, VSH_OFLAG_NONE,
     N_("total I/O operations limit per second")},
    {"read_iops_sec", VSH_OT_INT, VSH_OFLAG_NONE,
     N_("read I/O operations limit per second")},
    {"write_iops_sec", VSH_OT_INT, VSH_OFLAG_NONE,
     N_("write I/O operations limit per second")},
    {"config", VSH_OT_BOOL, 0, N_("affect next boot")},
    {"live", VSH_OT_BOOL, 0, N_("affect running domain")},
    {"current", VSH_OT_BOOL, 0, N_("affect current domain")},
    {NULL, 0, 0, NULL}
};

static bool
cmdBlkdeviotune(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
    const char *name, *disk;
    unsigned long long total_bytes_sec = 0, read_bytes_sec = 0, write_bytes_sec = 0;
    unsigned long long total_iops_sec = 0, read_iops_sec = 0, write_iops_sec = 0;
    int nparams = 0;
    virTypedParameterPtr params = NULL, temp = NULL;
    unsigned int flags = 0, i = 0;
    int rv = 0;
    int current = vshCommandOptBool(cmd, "current");
    int config = vshCommandOptBool(cmd, "config");
    int live = vshCommandOptBool(cmd, "live");
    bool ret = false;

    if (current) {
        if (live || config) {
            vshError(ctl, "%s", _("--current must be specified exclusively"));
            return false;
        }
        flags = VIR_DOMAIN_AFFECT_CURRENT;
    } else {
        if (config)
            flags |= VIR_DOMAIN_AFFECT_CONFIG;
        if (live)
            flags |= VIR_DOMAIN_AFFECT_LIVE;
    }

    if (!vshConnectionUsability(ctl, ctl->conn))
        goto cleanup;

    if (!(dom = vshCommandOptDomain(ctl, cmd, &name)))
        goto cleanup;

    if (vshCommandOptString(cmd, "device", &disk) < 0)
        goto cleanup;

    if ((rv = vshCommandOptULongLong(cmd, "total_bytes_sec", &total_bytes_sec)) < 0) {
        vshError(ctl, "%s",
                 _("Unable to parse integer parameter"));
        goto cleanup;
    } else if (rv > 0) {
        nparams++;
    }

    if ((rv = vshCommandOptULongLong(cmd, "read_bytes_sec", &read_bytes_sec)) < 0) {
        vshError(ctl, "%s",
                 _("Unable to parse integer parameter"));
        goto cleanup;
    } else if (rv > 0) {
        nparams++;
    }

    if ((rv = vshCommandOptULongLong(cmd, "write_bytes_sec", &write_bytes_sec)) < 0) {
        vshError(ctl, "%s",
                 _("Unable to parse integer parameter"));
        goto cleanup;
    } else if (rv > 0) {
        nparams++;
    }

    if ((rv = vshCommandOptULongLong(cmd, "total_iops_sec", &total_iops_sec)) < 0) {
        vshError(ctl, "%s",
                 _("Unable to parse integer parameter"));
        goto cleanup;
    } else if (rv > 0) {
        nparams++;
    }

    if ((rv = vshCommandOptULongLong(cmd, "read_iops_sec", &read_iops_sec)) < 0) {
        vshError(ctl, "%s",
                 _("Unable to parse integer parameter"));
        goto cleanup;
    } else if (rv > 0) {
        nparams++;
    }

    if ((rv = vshCommandOptULongLong(cmd, "write_iops_sec", &write_iops_sec)) < 0) {
        vshError(ctl, "%s",
                 _("Unable to parse integer parameter"));
        goto cleanup;
    } else if (rv > 0) {
        nparams++;
    }

    if (nparams == 0) {

        if ((virDomainGetBlockIoTune(dom, NULL, NULL, &nparams, flags)) != 0) {
            vshError(ctl, "%s",
                     _("Unable to get number of block I/O throttle parameters"));
            goto cleanup;
        }

        if (nparams == 0) {
            ret = true;
            goto cleanup;
        }

        params = vshCalloc(ctl, nparams, sizeof(*params));

        if ((virDomainGetBlockIoTune(dom, disk, params, &nparams, flags)) != 0) {
            vshError(ctl, "%s",
                     _("Unable to get block I/O throttle parameters"));
            goto cleanup;
        }

        for (i = 0; i < nparams; i++) {
7632 7633 7634
            char *str = vshGetTypedParamValue(ctl, &params[i]);
            vshPrint(ctl, "%-15s: %s\n", params[i].field, str);
            VIR_FREE(str);
L
Lei Li 已提交
7635 7636
        }

7637
        ret = true;
7638
        goto cleanup;
L
Lei Li 已提交
7639 7640 7641 7642 7643 7644 7645 7646 7647 7648 7649 7650 7651 7652 7653 7654 7655 7656 7657 7658 7659 7660 7661 7662 7663 7664 7665 7666 7667 7668 7669 7670 7671 7672 7673 7674 7675 7676 7677 7678 7679 7680 7681 7682 7683 7684 7685 7686 7687 7688 7689 7690 7691 7692 7693 7694 7695 7696 7697 7698 7699 7700 7701 7702 7703 7704 7705 7706 7707 7708 7709 7710 7711
    } else {
        /* Set the block I/O throttle, match by opt since parameters can be 0 */
        params = vshCalloc(ctl, nparams, sizeof(*params));
        i = 0;

        if ((i < nparams) && (vshCommandOptBool(cmd, "total_bytes_sec"))) {
            temp = &params[i];
            temp->type = VIR_TYPED_PARAM_ULLONG;
            strncpy(temp->field, VIR_DOMAIN_BLOCK_IOTUNE_TOTAL_BYTES_SEC,
                    sizeof(temp->field));
            temp->value.ul = total_bytes_sec;
            i++;
        }

        if ((i < nparams) && (vshCommandOptBool(cmd, "read_bytes_sec"))) {
            temp = &params[i];
            temp->type = VIR_TYPED_PARAM_ULLONG;
            strncpy(temp->field, VIR_DOMAIN_BLOCK_IOTUNE_READ_BYTES_SEC,
                    sizeof(temp->field));
            temp->value.ul = read_bytes_sec;
            i++;
        }

        if ((i < nparams) && (vshCommandOptBool(cmd, "write_bytes_sec"))) {
            temp = &params[i];
            temp->type = VIR_TYPED_PARAM_ULLONG;
            strncpy(temp->field, VIR_DOMAIN_BLOCK_IOTUNE_WRITE_BYTES_SEC,
                    sizeof(temp->field));
            temp->value.ul = write_bytes_sec;
            i++;
        }

        if ((i < nparams) && (vshCommandOptBool(cmd, "total_iops_sec"))) {
            temp = &params[i];
            temp->type = VIR_TYPED_PARAM_ULLONG;
            strncpy(temp->field, VIR_DOMAIN_BLOCK_IOTUNE_TOTAL_IOPS_SEC,
                    sizeof(temp->field));
            temp->value.ul = total_iops_sec;
            i++;
        }

        if ((i < nparams) && (vshCommandOptBool(cmd, "read_iops_sec"))) {
            temp = &params[i];
            temp->type = VIR_TYPED_PARAM_ULLONG;
            strncpy(temp->field, VIR_DOMAIN_BLOCK_IOTUNE_READ_IOPS_SEC,
                    sizeof(temp->field));
            temp->value.ul = read_iops_sec;
            i++;
        }

        if ((i < nparams) && (vshCommandOptBool(cmd, "write_iops_sec"))) {
            temp = &params[i];
            temp->type = VIR_TYPED_PARAM_ULLONG;
            strncpy(temp->field, VIR_DOMAIN_BLOCK_IOTUNE_WRITE_IOPS_SEC,
                    sizeof(temp->field));
            temp->value.ul = write_iops_sec;
        }

        if ((virDomainSetBlockIoTune(dom, disk, params, nparams, flags)) < 0) {
            vshError(ctl, "%s",
                     _("Unable to change block I/O throttle"));
            goto cleanup;
        }
    }

    ret = true;

cleanup:
    VIR_FREE(params);
    virDomainFree(dom);
    return ret;
}

7712 7713 7714
/*
 * "net-create" command
 */
7715
static const vshCmdInfo info_network_create[] = {
7716 7717
    {"help", N_("create a network from an XML file")},
    {"desc", N_("Create a network.")},
7718 7719 7720
    {NULL, NULL}
};

7721
static const vshCmdOptDef opts_network_create[] = {
7722
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file containing an XML network description")},
7723 7724 7725
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
7726
static bool
7727
cmdNetworkCreate(vshControl *ctl, const vshCmd *cmd)
7728 7729
{
    virNetworkPtr network;
7730
    const char *from = NULL;
E
Eric Blake 已提交
7731
    bool ret = true;
7732
    char *buffer;
7733

7734
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
7735
        return false;
7736

7737
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
7738
        return false;
7739

7740
    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0)
E
Eric Blake 已提交
7741
        return false;
7742 7743

    network = virNetworkCreateXML(ctl->conn, buffer);
7744
    VIR_FREE(buffer);
7745

7746 7747 7748
    if (network != NULL) {
        vshPrint(ctl, _("Network %s created from %s\n"),
                 virNetworkGetName(network), from);
L
Laine Stump 已提交
7749
        virNetworkFree(network);
7750
    } else {
7751
        vshError(ctl, _("Failed to create network from %s"), from);
E
Eric Blake 已提交
7752
        ret = false;
7753 7754 7755 7756 7757 7758 7759 7760
    }
    return ret;
}


/*
 * "net-define" command
 */
7761
static const vshCmdInfo info_network_define[] = {
7762 7763
    {"help", N_("define (but don't start) a network from an XML file")},
    {"desc", N_("Define a network.")},
7764 7765 7766
    {NULL, NULL}
};

7767
static const vshCmdOptDef opts_network_define[] = {
7768
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file containing an XML network description")},
7769 7770 7771
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
7772
static bool
7773
cmdNetworkDefine(vshControl *ctl, const vshCmd *cmd)
7774 7775
{
    virNetworkPtr network;
7776
    const char *from = NULL;
E
Eric Blake 已提交
7777
    bool ret = true;
7778
    char *buffer;
7779

7780
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
7781
        return false;
7782

7783
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
7784
        return false;
7785

7786
    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0)
E
Eric Blake 已提交
7787
        return false;
7788 7789

    network = virNetworkDefineXML(ctl->conn, buffer);
7790
    VIR_FREE(buffer);
7791

7792 7793 7794
    if (network != NULL) {
        vshPrint(ctl, _("Network %s defined from %s\n"),
                 virNetworkGetName(network), from);
L
Laine Stump 已提交
7795
        virNetworkFree(network);
7796
    } else {
7797
        vshError(ctl, _("Failed to define network from %s"), from);
E
Eric Blake 已提交
7798
        ret = false;
7799 7800 7801 7802 7803 7804 7805 7806
    }
    return ret;
}


/*
 * "net-destroy" command
 */
7807
static const vshCmdInfo info_network_destroy[] = {
7808 7809
    {"help", N_("destroy (stop) a network")},
    {"desc", N_("Forcefully stop a given network.")},
7810 7811 7812
    {NULL, NULL}
};

7813
static const vshCmdOptDef opts_network_destroy[] = {
7814
    {"network", VSH_OT_DATA, VSH_OFLAG_REQ, N_("network name or uuid")},
7815 7816 7817
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
7818
static bool
7819
cmdNetworkDestroy(vshControl *ctl, const vshCmd *cmd)
7820 7821
{
    virNetworkPtr network;
E
Eric Blake 已提交
7822
    bool ret = true;
7823
    const char *name;
7824

7825
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
7826
        return false;
7827

J
Jim Meyering 已提交
7828
    if (!(network = vshCommandOptNetwork(ctl, cmd, &name)))
E
Eric Blake 已提交
7829
        return false;
7830 7831 7832 7833

    if (virNetworkDestroy(network) == 0) {
        vshPrint(ctl, _("Network %s destroyed\n"), name);
    } else {
7834
        vshError(ctl, _("Failed to destroy network %s"), name);
E
Eric Blake 已提交
7835
        ret = false;
7836 7837
    }

7838
    virNetworkFree(network);
7839 7840 7841 7842 7843 7844 7845
    return ret;
}


/*
 * "net-dumpxml" command
 */
7846
static const vshCmdInfo info_network_dumpxml[] = {
7847 7848
    {"help", N_("network information in XML")},
    {"desc", N_("Output the network information as an XML dump to stdout.")},
7849 7850 7851
    {NULL, NULL}
};

7852
static const vshCmdOptDef opts_network_dumpxml[] = {
7853
    {"network", VSH_OT_DATA, VSH_OFLAG_REQ, N_("network name or uuid")},
7854
    {"inactive", VSH_OT_BOOL, VSH_OFLAG_NONE, N_("network information of an inactive domain")},
7855 7856 7857
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
7858
static bool
7859
cmdNetworkDumpXML(vshControl *ctl, const vshCmd *cmd)
7860 7861
{
    virNetworkPtr network;
E
Eric Blake 已提交
7862
    bool ret = true;
7863
    char *dump;
7864 7865
    unsigned int flags = 0;
    int inactive;
7866

7867
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
7868
        return false;
7869

J
Jim Meyering 已提交
7870
    if (!(network = vshCommandOptNetwork(ctl, cmd, NULL)))
E
Eric Blake 已提交
7871
        return false;
7872

7873 7874 7875 7876 7877 7878
    inactive = vshCommandOptBool (cmd, "inactive");
    if (inactive)
        flags |= VIR_NETWORK_XML_INACTIVE;

    dump = virNetworkGetXMLDesc(network, flags);

7879
    if (dump != NULL) {
7880
        vshPrint(ctl, "%s", dump);
7881
        VIR_FREE(dump);
7882
    } else {
E
Eric Blake 已提交
7883
        ret = false;
7884 7885 7886 7887 7888 7889
    }

    virNetworkFree(network);
    return ret;
}

O
Osier Yang 已提交
7890 7891 7892 7893 7894
/*
 * "net-info" command
 */
static const vshCmdInfo info_network_info[] = {
    {"help", N_("network information")},
E
Eric Blake 已提交
7895
    {"desc", N_("Returns basic information about the network")},
O
Osier Yang 已提交
7896 7897 7898 7899 7900 7901 7902 7903
    {NULL, NULL}
};

static const vshCmdOptDef opts_network_info[] = {
    {"network", VSH_OT_DATA, VSH_OFLAG_REQ, N_("network name")},
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
7904
static bool
O
Osier Yang 已提交
7905 7906 7907 7908 7909 7910 7911 7912 7913 7914
cmdNetworkInfo(vshControl *ctl, const vshCmd *cmd)
{
    virNetworkPtr network;
    char uuid[VIR_UUID_STRING_BUFLEN];
    int autostart;
    int persistent = -1;
    int active = -1;
    char *bridge = NULL;

    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
7915
        return false;
O
Osier Yang 已提交
7916 7917 7918

    if (!(network = vshCommandOptNetworkBy(ctl, cmd, NULL,
                                           VSH_BYNAME)))
E
Eric Blake 已提交
7919
        return false;
O
Osier Yang 已提交
7920 7921 7922 7923 7924 7925 7926 7927 7928 7929 7930 7931 7932 7933 7934 7935 7936 7937 7938 7939 7940 7941 7942 7943 7944

    vshPrint(ctl, "%-15s %s\n", _("Name"), virNetworkGetName(network));

    if (virNetworkGetUUIDString(network, uuid) == 0)
        vshPrint(ctl, "%-15s %s\n", _("UUID"), uuid);

    active = virNetworkIsActive(network);
    if (active >= 0)
        vshPrint(ctl, "%-15s %s\n", _("Active:"), active? _("yes") : _("no"));

    persistent = virNetworkIsPersistent(network);
    if (persistent < 0)
        vshPrint(ctl, "%-15s %s\n", _("Persistent:"), _("unknown"));
    else
        vshPrint(ctl, "%-15s %s\n", _("Persistent:"), persistent ? _("yes") : _("no"));

    if (virNetworkGetAutostart(network, &autostart) < 0)
        vshPrint(ctl, "%-15s %s\n", _("Autostart:"), _("no autostart"));
    else
        vshPrint(ctl, "%-15s %s\n", _("Autostart:"), autostart ? _("yes") : _("no"));

    bridge = virNetworkGetBridgeName(network);
    if (bridge)
        vshPrint(ctl, "%-15s %s\n", _("Bridge:"), bridge);

7945
    VIR_FREE(bridge);
O
Osier Yang 已提交
7946
    virNetworkFree(network);
E
Eric Blake 已提交
7947
    return true;
O
Osier Yang 已提交
7948
}
7949

7950 7951 7952 7953
/*
 * "iface-edit" command
 */
static const vshCmdInfo info_interface_edit[] = {
7954 7955
    {"help", N_("edit XML configuration for a physical host interface")},
    {"desc", N_("Edit the XML configuration for a physical host interface.")},
7956 7957 7958 7959
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_edit[] = {
7960
    {"interface", VSH_OT_DATA, VSH_OFLAG_REQ, N_("interface name or MAC address")},
7961 7962 7963
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
7964
static bool
7965 7966
cmdInterfaceEdit (vshControl *ctl, const vshCmd *cmd)
{
E
Eric Blake 已提交
7967
    bool ret = false;
7968 7969 7970 7971 7972
    virInterfacePtr iface = NULL;
    char *tmp = NULL;
    char *doc = NULL;
    char *doc_edited = NULL;
    char *doc_reread = NULL;
E
Eric Blake 已提交
7973
    unsigned int flags = VIR_INTERFACE_XML_INACTIVE;
7974

7975
    if (!vshConnectionUsability(ctl, ctl->conn))
7976 7977 7978 7979 7980 7981 7982 7983 7984 7985 7986 7987 7988 7989 7990 7991 7992 7993 7994 7995 7996 7997 7998 7999 8000 8001
        goto cleanup;

    iface = vshCommandOptInterface (ctl, cmd, NULL);
    if (iface == NULL)
        goto cleanup;

    /* Get the XML configuration of the interface. */
    doc = virInterfaceGetXMLDesc (iface, flags);
    if (!doc)
        goto cleanup;

    /* Create and open the temporary file. */
    tmp = editWriteToTempFile (ctl, doc);
    if (!tmp) goto cleanup;

    /* Start the editor. */
    if (editFile (ctl, tmp) == -1) goto cleanup;

    /* Read back the edited file. */
    doc_edited = editReadBackFile (ctl, tmp);
    if (!doc_edited) goto cleanup;

    /* Compare original XML with edited.  Has it changed at all? */
    if (STREQ (doc, doc_edited)) {
        vshPrint (ctl, _("Interface %s XML configuration not changed.\n"),
                  virInterfaceGetName (iface));
E
Eric Blake 已提交
8002
        ret = true;
8003 8004 8005 8006 8007 8008 8009 8010 8011 8012 8013 8014
        goto cleanup;
    }

    /* Now re-read the interface XML.  Did someone else change it while
     * it was being edited?  This also catches problems such as us
     * losing a connection or the interface going away.
     */
    doc_reread = virInterfaceGetXMLDesc (iface, flags);
    if (!doc_reread)
        goto cleanup;

    if (STRNEQ (doc, doc_reread)) {
8015 8016
        vshError(ctl, "%s",
                 _("ERROR: the XML configuration was changed by another user"));
8017 8018 8019 8020 8021 8022 8023 8024 8025 8026 8027 8028
        goto cleanup;
    }

    /* Everything checks out, so redefine the interface. */
    virInterfaceFree (iface);
    iface = virInterfaceDefineXML (ctl->conn, doc_edited, 0);
    if (!iface)
        goto cleanup;

    vshPrint (ctl, _("Interface %s XML configuration edited.\n"),
              virInterfaceGetName(iface));

E
Eric Blake 已提交
8029
    ret = true;
8030 8031 8032 8033 8034

cleanup:
    if (iface)
        virInterfaceFree (iface);

8035 8036 8037
    VIR_FREE(doc);
    VIR_FREE(doc_edited);
    VIR_FREE(doc_reread);
8038 8039 8040

    if (tmp) {
        unlink (tmp);
8041
        VIR_FREE(tmp);
8042 8043 8044 8045 8046
    }

    return ret;
}

8047 8048 8049
/*
 * "net-list" command
 */
8050
static const vshCmdInfo info_network_list[] = {
8051 8052
    {"help", N_("list networks")},
    {"desc", N_("Returns list of networks.")},
8053 8054 8055
    {NULL, NULL}
};

8056
static const vshCmdOptDef opts_network_list[] = {
8057 8058
    {"inactive", VSH_OT_BOOL, 0, N_("list inactive networks")},
    {"all", VSH_OT_BOOL, 0, N_("list inactive & active networks")},
8059 8060 8061
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
8062
static bool
8063
cmdNetworkList(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
8064 8065 8066 8067 8068
{
    int inactive = vshCommandOptBool(cmd, "inactive");
    int all = vshCommandOptBool(cmd, "all");
    int active = !inactive || all ? 1 : 0;
    int maxactive = 0, maxinactive = 0, i;
8069
    char **activeNames = NULL, **inactiveNames = NULL;
8070 8071
    inactive |= all;

8072
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8073
        return false;
8074 8075

    if (active) {
8076 8077
        maxactive = virConnectNumOfNetworks(ctl->conn);
        if (maxactive < 0) {
8078
            vshError(ctl, "%s", _("Failed to list active networks"));
E
Eric Blake 已提交
8079
            return false;
8080
        }
8081
        if (maxactive) {
8082
            activeNames = vshMalloc(ctl, sizeof(char *) * maxactive);
8083

8084
            if ((maxactive = virConnectListNetworks(ctl->conn, activeNames,
8085
                                                    maxactive)) < 0) {
8086
                vshError(ctl, "%s", _("Failed to list active networks"));
8087
                VIR_FREE(activeNames);
E
Eric Blake 已提交
8088
                return false;
8089
            }
8090

8091
            qsort(&activeNames[0], maxactive, sizeof(char *), namesorter);
8092
        }
8093 8094
    }
    if (inactive) {
8095 8096
        maxinactive = virConnectNumOfDefinedNetworks(ctl->conn);
        if (maxinactive < 0) {
8097
            vshError(ctl, "%s", _("Failed to list inactive networks"));
8098
            VIR_FREE(activeNames);
E
Eric Blake 已提交
8099
            return false;
8100
        }
8101 8102 8103
        if (maxinactive) {
            inactiveNames = vshMalloc(ctl, sizeof(char *) * maxinactive);

8104 8105 8106
            if ((maxinactive =
                     virConnectListDefinedNetworks(ctl->conn, inactiveNames,
                                                   maxinactive)) < 0) {
8107
                vshError(ctl, "%s", _("Failed to list inactive networks"));
8108 8109
                VIR_FREE(activeNames);
                VIR_FREE(inactiveNames);
E
Eric Blake 已提交
8110
                return false;
8111
            }
8112

8113 8114
            qsort(&inactiveNames[0], maxinactive, sizeof(char*), namesorter);
        }
8115
    }
8116 8117
    vshPrintExtra(ctl, "%-20s %-10s %s\n", _("Name"), _("State"),
                  _("Autostart"));
8118
    vshPrintExtra(ctl, "-----------------------------------------\n");
8119 8120

    for (i = 0; i < maxactive; i++) {
8121 8122
        virNetworkPtr network =
            virNetworkLookupByName(ctl->conn, activeNames[i]);
8123 8124
        const char *autostartStr;
        int autostart = 0;
8125 8126 8127

        /* this kind of work with networks is not atomic operation */
        if (!network) {
8128
            VIR_FREE(activeNames[i]);
8129
            continue;
8130
        }
8131

8132 8133 8134
        if (virNetworkGetAutostart(network, &autostart) < 0)
            autostartStr = _("no autostart");
        else
8135
            autostartStr = autostart ? _("yes") : _("no");
8136 8137 8138 8139 8140

        vshPrint(ctl, "%-20s %-10s %-10s\n",
                 virNetworkGetName(network),
                 _("active"),
                 autostartStr);
8141
        virNetworkFree(network);
8142
        VIR_FREE(activeNames[i]);
8143 8144 8145
    }
    for (i = 0; i < maxinactive; i++) {
        virNetworkPtr network = virNetworkLookupByName(ctl->conn, inactiveNames[i]);
8146 8147
        const char *autostartStr;
        int autostart = 0;
8148 8149 8150

        /* this kind of work with networks is not atomic operation */
        if (!network) {
8151
            VIR_FREE(inactiveNames[i]);
8152
            continue;
8153
        }
8154

8155 8156 8157
        if (virNetworkGetAutostart(network, &autostart) < 0)
            autostartStr = _("no autostart");
        else
8158
            autostartStr = autostart ? _("yes") : _("no");
8159

8160
        vshPrint(ctl, "%-20s %-10s %-10s\n",
8161 8162 8163
                 inactiveNames[i],
                 _("inactive"),
                 autostartStr);
8164 8165

        virNetworkFree(network);
8166
        VIR_FREE(inactiveNames[i]);
8167
    }
8168 8169
    VIR_FREE(activeNames);
    VIR_FREE(inactiveNames);
E
Eric Blake 已提交
8170
    return true;
8171 8172 8173 8174 8175 8176
}


/*
 * "net-name" command
 */
8177
static const vshCmdInfo info_network_name[] = {
8178
    {"help", N_("convert a network UUID to network name")},
8179
    {"desc", ""},
8180 8181 8182
    {NULL, NULL}
};

8183
static const vshCmdOptDef opts_network_name[] = {
8184
    {"network", VSH_OT_DATA, VSH_OFLAG_REQ, N_("network uuid")},
8185 8186 8187
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
8188
static bool
8189
cmdNetworkName(vshControl *ctl, const vshCmd *cmd)
8190 8191 8192
{
    virNetworkPtr network;

8193
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8194
        return false;
J
Jim Meyering 已提交
8195
    if (!(network = vshCommandOptNetworkBy(ctl, cmd, NULL,
8196
                                           VSH_BYUUID)))
E
Eric Blake 已提交
8197
        return false;
8198 8199 8200

    vshPrint(ctl, "%s\n", virNetworkGetName(network));
    virNetworkFree(network);
E
Eric Blake 已提交
8201
    return true;
8202 8203 8204 8205 8206 8207
}


/*
 * "net-start" command
 */
8208
static const vshCmdInfo info_network_start[] = {
8209 8210
    {"help", N_("start a (previously defined) inactive network")},
    {"desc", N_("Start a network.")},
8211 8212 8213
    {NULL, NULL}
};

8214
static const vshCmdOptDef opts_network_start[] = {
8215
    {"network", VSH_OT_DATA, VSH_OFLAG_REQ, N_("name of the inactive network")},
8216 8217 8218
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
8219
static bool
8220
cmdNetworkStart(vshControl *ctl, const vshCmd *cmd)
8221 8222
{
    virNetworkPtr network;
E
Eric Blake 已提交
8223
    bool ret = true;
8224

8225
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8226
        return false;
8227

J
Jim Meyering 已提交
8228
    if (!(network = vshCommandOptNetworkBy(ctl, cmd, NULL, VSH_BYNAME)))
E
Eric Blake 已提交
8229
         return false;
8230 8231 8232

    if (virNetworkCreate(network) == 0) {
        vshPrint(ctl, _("Network %s started\n"),
8233
                 virNetworkGetName(network));
8234
    } else {
8235
        vshError(ctl, _("Failed to start network %s"),
8236
                 virNetworkGetName(network));
E
Eric Blake 已提交
8237
        ret = false;
8238
    }
L
Laine Stump 已提交
8239
    virNetworkFree(network);
8240 8241 8242 8243 8244 8245 8246
    return ret;
}


/*
 * "net-undefine" command
 */
8247
static const vshCmdInfo info_network_undefine[] = {
8248 8249
    {"help", N_("undefine an inactive network")},
    {"desc", N_("Undefine the configuration for an inactive network.")},
8250 8251 8252
    {NULL, NULL}
};

8253
static const vshCmdOptDef opts_network_undefine[] = {
8254
    {"network", VSH_OT_DATA, VSH_OFLAG_REQ, N_("network name or uuid")},
8255 8256 8257
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
8258
static bool
8259
cmdNetworkUndefine(vshControl *ctl, const vshCmd *cmd)
8260 8261
{
    virNetworkPtr network;
E
Eric Blake 已提交
8262
    bool ret = true;
8263
    const char *name;
8264

8265
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8266
        return false;
8267

J
Jim Meyering 已提交
8268
    if (!(network = vshCommandOptNetwork(ctl, cmd, &name)))
E
Eric Blake 已提交
8269
        return false;
8270 8271 8272 8273

    if (virNetworkUndefine(network) == 0) {
        vshPrint(ctl, _("Network %s has been undefined\n"), name);
    } else {
8274
        vshError(ctl, _("Failed to undefine network %s"), name);
E
Eric Blake 已提交
8275
        ret = false;
8276 8277
    }

L
Laine Stump 已提交
8278
    virNetworkFree(network);
8279 8280 8281 8282 8283 8284 8285
    return ret;
}


/*
 * "net-uuid" command
 */
8286
static const vshCmdInfo info_network_uuid[] = {
8287
    {"help", N_("convert a network name to network UUID")},
8288
    {"desc", ""},
8289 8290 8291
    {NULL, NULL}
};

8292
static const vshCmdOptDef opts_network_uuid[] = {
8293
    {"network", VSH_OT_DATA, VSH_OFLAG_REQ, N_("network name")},
8294 8295 8296
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
8297
static bool
8298
cmdNetworkUuid(vshControl *ctl, const vshCmd *cmd)
8299 8300 8301 8302
{
    virNetworkPtr network;
    char uuid[VIR_UUID_STRING_BUFLEN];

8303
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8304
        return false;
8305

J
Jim Meyering 已提交
8306
    if (!(network = vshCommandOptNetworkBy(ctl, cmd, NULL,
8307
                                           VSH_BYNAME)))
E
Eric Blake 已提交
8308
        return false;
8309 8310 8311 8312

    if (virNetworkGetUUIDString(network, uuid) != -1)
        vshPrint(ctl, "%s\n", uuid);
    else
8313
        vshError(ctl, "%s", _("failed to get network UUID"));
8314

L
Laine Stump 已提交
8315
    virNetworkFree(network);
E
Eric Blake 已提交
8316
    return true;
8317 8318 8319
}


8320 8321 8322 8323 8324
/**************************************************************************/
/*
 * "iface-list" command
 */
static const vshCmdInfo info_interface_list[] = {
8325 8326
    {"help", N_("list physical host interfaces")},
    {"desc", N_("Returns list of physical host interfaces.")},
8327 8328 8329 8330
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_list[] = {
8331 8332
    {"inactive", VSH_OT_BOOL, 0, N_("list inactive interfaces")},
    {"all", VSH_OT_BOOL, 0, N_("list inactive & active interfaces")},
8333 8334
    {NULL, 0, 0, NULL}
};
E
Eric Blake 已提交
8335
static bool
8336 8337 8338 8339 8340 8341 8342 8343 8344
cmdInterfaceList(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
{
    int inactive = vshCommandOptBool(cmd, "inactive");
    int all = vshCommandOptBool(cmd, "all");
    int active = !inactive || all ? 1 : 0;
    int maxactive = 0, maxinactive = 0, i;
    char **activeNames = NULL, **inactiveNames = NULL;
    inactive |= all;

8345
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8346
        return false;
8347 8348 8349 8350

    if (active) {
        maxactive = virConnectNumOfInterfaces(ctl->conn);
        if (maxactive < 0) {
8351
            vshError(ctl, "%s", _("Failed to list active interfaces"));
E
Eric Blake 已提交
8352
            return false;
8353 8354 8355 8356 8357 8358
        }
        if (maxactive) {
            activeNames = vshMalloc(ctl, sizeof(char *) * maxactive);

            if ((maxactive = virConnectListInterfaces(ctl->conn, activeNames,
                                                    maxactive)) < 0) {
8359
                vshError(ctl, "%s", _("Failed to list active interfaces"));
8360
                VIR_FREE(activeNames);
E
Eric Blake 已提交
8361
                return false;
8362 8363 8364 8365 8366 8367 8368 8369
            }

            qsort(&activeNames[0], maxactive, sizeof(char *), namesorter);
        }
    }
    if (inactive) {
        maxinactive = virConnectNumOfDefinedInterfaces(ctl->conn);
        if (maxinactive < 0) {
8370
            vshError(ctl, "%s", _("Failed to list inactive interfaces"));
8371
            VIR_FREE(activeNames);
E
Eric Blake 已提交
8372
            return false;
8373 8374 8375 8376
        }
        if (maxinactive) {
            inactiveNames = vshMalloc(ctl, sizeof(char *) * maxinactive);

8377 8378 8379
            if ((maxinactive =
                     virConnectListDefinedInterfaces(ctl->conn, inactiveNames,
                                                     maxinactive)) < 0) {
8380
                vshError(ctl, "%s", _("Failed to list inactive interfaces"));
8381 8382
                VIR_FREE(activeNames);
                VIR_FREE(inactiveNames);
E
Eric Blake 已提交
8383
                return false;
8384 8385 8386 8387 8388
            }

            qsort(&inactiveNames[0], maxinactive, sizeof(char*), namesorter);
        }
    }
8389 8390
    vshPrintExtra(ctl, "%-20s %-10s %s\n", _("Name"), _("State"),
                  _("MAC Address"));
8391 8392 8393
    vshPrintExtra(ctl, "--------------------------------------------\n");

    for (i = 0; i < maxactive; i++) {
8394 8395
        virInterfacePtr iface =
            virInterfaceLookupByName(ctl->conn, activeNames[i]);
8396 8397 8398

        /* this kind of work with interfaces is not atomic */
        if (!iface) {
8399
            VIR_FREE(activeNames[i]);
8400 8401 8402 8403 8404 8405 8406 8407
            continue;
        }

        vshPrint(ctl, "%-20s %-10s %s\n",
                 virInterfaceGetName(iface),
                 _("active"),
                 virInterfaceGetMACString(iface));
        virInterfaceFree(iface);
8408
        VIR_FREE(activeNames[i]);
8409 8410
    }
    for (i = 0; i < maxinactive; i++) {
8411 8412
        virInterfacePtr iface =
            virInterfaceLookupByName(ctl->conn, inactiveNames[i]);
8413 8414 8415

        /* this kind of work with interfaces is not atomic */
        if (!iface) {
8416
            VIR_FREE(inactiveNames[i]);
8417 8418 8419 8420 8421 8422 8423 8424
            continue;
        }

        vshPrint(ctl, "%-20s %-10s %s\n",
                 virInterfaceGetName(iface),
                 _("inactive"),
                 virInterfaceGetMACString(iface));
        virInterfaceFree(iface);
8425
        VIR_FREE(inactiveNames[i]);
8426
    }
8427 8428
    VIR_FREE(activeNames);
    VIR_FREE(inactiveNames);
E
Eric Blake 已提交
8429
    return true;
8430 8431 8432 8433 8434 8435 8436

}

/*
 * "iface-name" command
 */
static const vshCmdInfo info_interface_name[] = {
8437
    {"help", N_("convert an interface MAC address to interface name")},
8438 8439 8440 8441 8442
    {"desc", ""},
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_name[] = {
8443
    {"interface", VSH_OT_DATA, VSH_OFLAG_REQ, N_("interface mac")},
8444 8445 8446
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
8447
static bool
8448 8449 8450 8451
cmdInterfaceName(vshControl *ctl, const vshCmd *cmd)
{
    virInterfacePtr iface;

8452
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8453
        return false;
8454
    if (!(iface = vshCommandOptInterfaceBy(ctl, cmd, NULL, NULL,
8455
                                           VSH_BYMAC)))
E
Eric Blake 已提交
8456
        return false;
8457 8458 8459

    vshPrint(ctl, "%s\n", virInterfaceGetName(iface));
    virInterfaceFree(iface);
E
Eric Blake 已提交
8460
    return true;
8461 8462 8463 8464 8465 8466
}

/*
 * "iface-mac" command
 */
static const vshCmdInfo info_interface_mac[] = {
8467
    {"help", N_("convert an interface name to interface MAC address")},
8468 8469 8470 8471 8472
    {"desc", ""},
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_mac[] = {
8473
    {"interface", VSH_OT_DATA, VSH_OFLAG_REQ, N_("interface name")},
8474 8475 8476
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
8477
static bool
8478 8479 8480 8481
cmdInterfaceMAC(vshControl *ctl, const vshCmd *cmd)
{
    virInterfacePtr iface;

8482
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8483
        return false;
8484
    if (!(iface = vshCommandOptInterfaceBy(ctl, cmd, NULL, NULL,
8485
                                           VSH_BYNAME)))
E
Eric Blake 已提交
8486
        return false;
8487 8488 8489

    vshPrint(ctl, "%s\n", virInterfaceGetMACString(iface));
    virInterfaceFree(iface);
E
Eric Blake 已提交
8490
    return true;
8491 8492 8493 8494 8495 8496
}

/*
 * "iface-dumpxml" command
 */
static const vshCmdInfo info_interface_dumpxml[] = {
8497 8498
    {"help", N_("interface information in XML")},
    {"desc", N_("Output the physical host interface information as an XML dump to stdout.")},
8499 8500 8501 8502
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_dumpxml[] = {
8503 8504
    {"interface", VSH_OT_DATA, VSH_OFLAG_REQ, N_("interface name or MAC address")},
    {"inactive", VSH_OT_BOOL, 0, N_("show inactive defined XML")},
8505 8506 8507
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
8508
static bool
8509 8510 8511
cmdInterfaceDumpXML(vshControl *ctl, const vshCmd *cmd)
{
    virInterfacePtr iface;
E
Eric Blake 已提交
8512
    bool ret = true;
8513
    char *dump;
E
Eric Blake 已提交
8514
    unsigned int flags = 0;
8515 8516 8517 8518
    int inactive = vshCommandOptBool(cmd, "inactive");

    if (inactive)
        flags |= VIR_INTERFACE_XML_INACTIVE;
8519

8520
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8521
        return false;
8522 8523

    if (!(iface = vshCommandOptInterface(ctl, cmd, NULL)))
E
Eric Blake 已提交
8524
        return false;
8525

8526
    dump = virInterfaceGetXMLDesc(iface, flags);
8527
    if (dump != NULL) {
8528
        vshPrint(ctl, "%s", dump);
8529
        VIR_FREE(dump);
8530
    } else {
E
Eric Blake 已提交
8531
        ret = false;
8532 8533 8534 8535 8536 8537 8538 8539 8540 8541
    }

    virInterfaceFree(iface);
    return ret;
}

/*
 * "iface-define" command
 */
static const vshCmdInfo info_interface_define[] = {
8542 8543
    {"help", N_("define (but don't start) a physical host interface from an XML file")},
    {"desc", N_("Define a physical host interface.")},
8544 8545 8546 8547
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_define[] = {
8548
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file containing an XML interface description")},
8549 8550 8551
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
8552
static bool
8553 8554
cmdInterfaceDefine(vshControl *ctl, const vshCmd *cmd)
{
8555
    virInterfacePtr iface;
8556
    const char *from = NULL;
E
Eric Blake 已提交
8557
    bool ret = true;
8558 8559
    char *buffer;

8560
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8561
        return false;
8562

8563
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
8564
        return false;
8565 8566

    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0)
E
Eric Blake 已提交
8567
        return false;
8568

8569
    iface = virInterfaceDefineXML(ctl->conn, buffer, 0);
8570
    VIR_FREE(buffer);
8571

8572
    if (iface != NULL) {
8573
        vshPrint(ctl, _("Interface %s defined from %s\n"),
8574
                 virInterfaceGetName(iface), from);
L
Laine Stump 已提交
8575
        virInterfaceFree (iface);
8576
    } else {
8577
        vshError(ctl, _("Failed to define interface from %s"), from);
E
Eric Blake 已提交
8578
        ret = false;
8579 8580 8581 8582 8583 8584 8585 8586
    }
    return ret;
}

/*
 * "iface-undefine" command
 */
static const vshCmdInfo info_interface_undefine[] = {
8587 8588
    {"help", N_("undefine a physical host interface (remove it from configuration)")},
    {"desc", N_("undefine an interface.")},
8589 8590 8591 8592
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_undefine[] = {
8593
    {"interface", VSH_OT_DATA, VSH_OFLAG_REQ, N_("interface name or MAC address")},
8594 8595 8596
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
8597
static bool
8598 8599 8600
cmdInterfaceUndefine(vshControl *ctl, const vshCmd *cmd)
{
    virInterfacePtr iface;
E
Eric Blake 已提交
8601
    bool ret = true;
8602
    const char *name;
8603

8604
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8605
        return false;
8606 8607

    if (!(iface = vshCommandOptInterface(ctl, cmd, &name)))
E
Eric Blake 已提交
8608
        return false;
8609 8610 8611 8612

    if (virInterfaceUndefine(iface) == 0) {
        vshPrint(ctl, _("Interface %s undefined\n"), name);
    } else {
8613
        vshError(ctl, _("Failed to undefine interface %s"), name);
E
Eric Blake 已提交
8614
        ret = false;
8615 8616 8617 8618 8619 8620 8621 8622 8623 8624
    }

    virInterfaceFree(iface);
    return ret;
}

/*
 * "iface-start" command
 */
static const vshCmdInfo info_interface_start[] = {
8625 8626
    {"help", N_("start a physical host interface (enable it / \"if-up\")")},
    {"desc", N_("start a physical host interface.")},
8627 8628 8629 8630
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_start[] = {
8631
    {"interface", VSH_OT_DATA, VSH_OFLAG_REQ, N_("interface name or MAC address")},
8632 8633 8634
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
8635
static bool
8636 8637 8638
cmdInterfaceStart(vshControl *ctl, const vshCmd *cmd)
{
    virInterfacePtr iface;
E
Eric Blake 已提交
8639
    bool ret = true;
8640
    const char *name;
8641

8642
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8643
        return false;
8644 8645

    if (!(iface = vshCommandOptInterface(ctl, cmd, &name)))
E
Eric Blake 已提交
8646
        return false;
8647 8648 8649 8650

    if (virInterfaceCreate(iface, 0) == 0) {
        vshPrint(ctl, _("Interface %s started\n"), name);
    } else {
8651
        vshError(ctl, _("Failed to start interface %s"), name);
E
Eric Blake 已提交
8652
        ret = false;
8653 8654 8655 8656 8657 8658 8659 8660 8661 8662
    }

    virInterfaceFree(iface);
    return ret;
}

/*
 * "iface-destroy" command
 */
static const vshCmdInfo info_interface_destroy[] = {
8663
    {"help", N_("destroy a physical host interface (disable it / \"if-down\")")},
8664
    {"desc", N_("forcefully stop a physical host interface.")},
8665 8666 8667 8668
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_destroy[] = {
8669
    {"interface", VSH_OT_DATA, VSH_OFLAG_REQ, N_("interface name or MAC address")},
8670 8671 8672
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
8673
static bool
8674 8675 8676
cmdInterfaceDestroy(vshControl *ctl, const vshCmd *cmd)
{
    virInterfacePtr iface;
E
Eric Blake 已提交
8677
    bool ret = true;
8678
    const char *name;
8679

8680
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
8681
        return false;
8682 8683

    if (!(iface = vshCommandOptInterface(ctl, cmd, &name)))
E
Eric Blake 已提交
8684
        return false;
8685 8686 8687 8688

    if (virInterfaceDestroy(iface, 0) == 0) {
        vshPrint(ctl, _("Interface %s destroyed\n"), name);
    } else {
8689
        vshError(ctl, _("Failed to destroy interface %s"), name);
E
Eric Blake 已提交
8690
        ret = false;
8691 8692 8693 8694 8695 8696
    }

    virInterfaceFree(iface);
    return ret;
}

8697 8698 8699 8700 8701
/*
 * "iface-begin" command
 */
static const vshCmdInfo info_interface_begin[] = {
    {"help", N_("create a snapshot of current interfaces settings, "
Y
Yuri Chornoivan 已提交
8702
                "which can be later committed (iface-commit) or "
8703 8704 8705 8706 8707 8708 8709 8710 8711 8712 8713 8714 8715 8716 8717 8718 8719 8720 8721 8722 8723 8724 8725 8726 8727 8728 8729 8730 8731 8732 8733 8734 8735 8736 8737 8738 8739 8740 8741 8742 8743 8744 8745 8746 8747 8748 8749 8750 8751 8752 8753 8754 8755 8756 8757 8758 8759 8760 8761 8762 8763 8764 8765 8766 8767 8768 8769 8770 8771 8772 8773 8774 8775 8776 8777 8778 8779 8780 8781
                "restored (iface-rollback)")},
    {"desc", N_("Create a restore point for interfaces settings")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_begin[] = {
    {NULL, 0, 0, NULL}
};

static bool
cmdInterfaceBegin(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
{
    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (virInterfaceChangeBegin(ctl->conn, 0) < 0) {
        vshError(ctl, "%s", _("Failed to begin network config change transaction"));
        return false;
    }

    vshPrint(ctl, "%s", _("Network config change transaction started\n"));
    return true;
}

/*
 * "iface-commit" command
 */
static const vshCmdInfo info_interface_commit[] = {
    {"help", N_("commit changes made since iface-begin and free restore point")},
    {"desc", N_("commit changes and free restore point")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_commit[] = {
    {NULL, 0, 0, NULL}
};

static bool
cmdInterfaceCommit(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
{
    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (virInterfaceChangeCommit(ctl->conn, 0) < 0) {
        vshError(ctl, "%s", _("Failed to commit network config change transaction"));
        return false;
    }

    vshPrint(ctl, "%s", _("Network config change transaction committed\n"));
    return true;
}

/*
 * "iface-rollback" command
 */
static const vshCmdInfo info_interface_rollback[] = {
    {"help", N_("rollback to previous saved configuration created via iface-begin")},
    {"desc", N_("rollback to previous restore point")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_rollback[] = {
    {NULL, 0, 0, NULL}
};

static bool
cmdInterfaceRollback(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
{
    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (virInterfaceChangeRollback(ctl->conn, 0) < 0) {
        vshError(ctl, "%s", _("Failed to rollback network config change transaction"));
        return false;
    }

    vshPrint(ctl, "%s", _("Network config change transaction rolled back\n"));
    return true;
}
8782

8783 8784 8785 8786 8787 8788 8789 8790 8791 8792 8793 8794 8795 8796 8797 8798 8799 8800 8801 8802 8803 8804 8805 8806 8807 8808 8809 8810 8811 8812 8813 8814 8815 8816 8817 8818 8819 8820 8821 8822 8823 8824 8825 8826 8827 8828 8829 8830 8831 8832 8833 8834 8835 8836 8837 8838 8839 8840 8841 8842 8843 8844 8845 8846 8847 8848 8849 8850 8851 8852 8853 8854 8855 8856 8857 8858 8859 8860 8861 8862 8863 8864 8865 8866 8867 8868 8869 8870 8871 8872 8873 8874 8875 8876 8877 8878 8879 8880 8881 8882 8883 8884 8885 8886 8887 8888 8889 8890 8891 8892 8893 8894 8895 8896 8897 8898 8899 8900 8901 8902 8903 8904 8905 8906 8907 8908 8909 8910 8911 8912 8913 8914 8915 8916 8917 8918 8919 8920 8921 8922 8923 8924 8925 8926 8927 8928 8929 8930 8931 8932 8933 8934 8935 8936 8937 8938 8939 8940 8941 8942 8943 8944 8945 8946 8947 8948 8949 8950 8951 8952 8953 8954 8955 8956 8957 8958 8959 8960 8961 8962 8963 8964 8965 8966 8967 8968 8969 8970 8971 8972 8973 8974 8975 8976
/*
 * "iface-bridge" command
 */
static const vshCmdInfo info_interface_bridge[] = {
    {"help", N_("create a bridge device and attach an existing network device to it")},
    {"desc", N_("bridge an existing network device")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_bridge[] = {
    {"interface", VSH_OT_DATA, VSH_OFLAG_REQ, N_("existing interface name")},
    {"bridge", VSH_OT_DATA, VSH_OFLAG_REQ, N_("new bridge device name")},
    {"no-stp", VSH_OT_BOOL, 0, N_("do not enable STP for this bridge")},
    {"delay", VSH_OT_INT, 0,
     N_("number of seconds to squelch traffic on newly connected ports")},
    {"no-start", VSH_OT_BOOL, 0, N_("don't start the bridge immediately")},
    {NULL, 0, 0, NULL}
};

static bool
cmdInterfaceBridge(vshControl *ctl, const vshCmd *cmd)
{
    bool ret = false;
    virInterfacePtr if_handle = NULL, br_handle = NULL;
    const char *if_name, *br_name;
    char *if_type = NULL, *if2_name = NULL, *delay_str = NULL;
    bool stp = false, nostart = false;
    unsigned int delay = 0;
    char *if_xml = NULL;
    xmlChar *br_xml = NULL;
    int br_xml_size;
    xmlDocPtr xml_doc = NULL;
    xmlXPathContextPtr ctxt = NULL;
    xmlNodePtr top_node, br_node, if_node, cur;

    if (!vshConnectionUsability(ctl, ctl->conn))
        goto cleanup;

    /* Get a handle to the original device */
    if (!(if_handle = vshCommandOptInterfaceBy(ctl, cmd, "interface",
                                               &if_name, VSH_BYNAME))) {
        goto cleanup;
    }

    /* Name for new bridge device */
    if (vshCommandOptString(cmd, "bridge", &br_name) <= 0) {
        vshError(ctl, "%s", _("Missing bridge device name in command"));
        goto cleanup;
    }

    /* make sure "new" device doesn't already exist */
    if ((br_handle = virInterfaceLookupByName(ctl->conn, br_name))) {
        vshError(ctl, _("Network device %s already exists"), br_name);
        goto cleanup;
    }

    /* use "no-stp" because we want "stp" to default true */
    stp = !vshCommandOptBool(cmd, "no-stp");

    if (vshCommandOptUInt(cmd, "delay", &delay) < 0) {
        vshError(ctl, "%s", _("Unable to parse delay parameter"));
        goto cleanup;
    }

    nostart = vshCommandOptBool(cmd, "no-start");

    /* Get the original interface into an xmlDoc */
    if (!(if_xml = virInterfaceGetXMLDesc(if_handle, VIR_INTERFACE_XML_INACTIVE)))
        goto cleanup;
    if (!(xml_doc = virXMLParseStringCtxt(if_xml,
                                          _("(interface definition)"), &ctxt))) {
        vshError(ctl, _("Failed to parse configuration of %s"), if_name);
        goto cleanup;
    }
    top_node = ctxt->node;

    /* Verify that the original device isn't already a bridge. */
    if (!(if_type = virXMLPropString(top_node, "type"))) {
        vshError(ctl, _("Existing device %s has no type"), if_name);
        goto cleanup;
    }

    if (STREQ(if_type, "bridge")) {
        vshError(ctl, _("Existing device %s is already a bridge"), if_name);
        goto cleanup;
    }

    /* verify the name in the XML matches the device name */
    if (!(if2_name = virXMLPropString(top_node, "name")) ||
        STRNEQ(if2_name, if_name)) {
        vshError(ctl, _("Interface name from config %s doesn't match given supplied name %s"),
                 if2_name, if_name);
        goto cleanup;
    }

    /* Create a <bridge> node under <interface>. */
    if (!(br_node = xmlNewChild(top_node, NULL, BAD_CAST "bridge", NULL))) {
        vshError(ctl, "%s", _("Failed to create bridge node in xml document"));
        goto cleanup;
    }

    /* Set stp and delay attributes in <bridge> according to the
     * commandline options.
     */
    if (!xmlSetProp(br_node, BAD_CAST "stp", BAD_CAST (stp ? "on" : "off"))) {
        vshError(ctl, "%s", _("Failed to set stp attribute in xml document"));
        goto cleanup;
    }

    if ((delay || stp) &&
        ((virAsprintf(&delay_str, "%d", delay) < 0) ||
         !xmlSetProp(br_node, BAD_CAST "delay", BAD_CAST delay_str))) {
        vshError(ctl, _("Failed to set bridge delay %d in xml document"), delay);
        goto cleanup;
    }

    /* Change the type of the outer/master interface to "bridge" and the
     * name to the provided bridge name.
     */
    if (!xmlSetProp(top_node, BAD_CAST "type", BAD_CAST "bridge")) {
        vshError(ctl, "%s", _("Failed to set bridge interface type to 'bridge' in xml document"));
        goto cleanup;
    }

    if (!xmlSetProp(top_node, BAD_CAST "name", BAD_CAST br_name)) {
        vshError(ctl, _("Failed to set master bridge interface name to '%s' in xml document"),
            br_name);
        goto cleanup;
    }

    /* Create an <interface> node under <bridge> that uses the
     * original interface's type and name.
     */
    if (!(if_node = xmlNewChild(br_node, NULL, BAD_CAST "interface", NULL))) {
        vshError(ctl, "%s", _("Failed to create interface node under bridge node in xml document"));
        goto cleanup;
    }

    /* set the type of the inner/slave interface to the original
     * if_type, and the name to the original if_name.
     */
    if (!xmlSetProp(if_node, BAD_CAST "type", BAD_CAST if_type)) {
        vshError(ctl, _("Failed to set new slave interface type to '%s' in xml document"),
                 if_name);
        goto cleanup;
    }

    if (!xmlSetProp(if_node, BAD_CAST "name", BAD_CAST if_name)) {
        vshError(ctl, _("Failed to set new slave interface name to '%s' in xml document"),
            br_name);
        goto cleanup;
    }

    /* Cycle through all the nodes under the original <interface>,
     * moving all <mac>, <bond> and <vlan> nodes down into the new
     * lower level <interface>.
     */
    cur = top_node->children;
    while (cur) {
        xmlNodePtr old = cur;

        cur = cur->next;
        if ((old->type == XML_ELEMENT_NODE) &&
            (xmlStrEqual(old->name, BAD_CAST "mac") ||  /* ethernet stuff to move down */
             xmlStrEqual(old->name, BAD_CAST "bond") || /* bond stuff to move down */
             xmlStrEqual(old->name, BAD_CAST "vlan"))) { /* vlan stuff to move down */
            xmlUnlinkNode(old);
            if (!xmlAddChild(if_node, old)) {
                vshError(ctl, _("Failed to move '%s' element in xml document"), old->name);
                xmlFreeNode(old);
                goto cleanup;
            }
        }
    }

    /* The document should now be fully converted; write it out to a string. */
    xmlDocDumpMemory(xml_doc, &br_xml, &br_xml_size);

    if (!br_xml || br_xml_size <= 0) {
        vshError(ctl, _("Failed to format new xml document for bridge %s"), br_name);
        goto cleanup;
    }


    /* br_xml is the new interface to define. It will automatically undefine the
     * independent original interface.
     */
    if (!(br_handle = virInterfaceDefineXML(ctl->conn, (char *) br_xml, 0))) {
        vshError(ctl, _("Failed to define new bridge interface %s"),
                 br_name);
        goto cleanup;
    }

    vshPrint(ctl, _("Created bridge %s with attached device %s\n"),
8977
             br_name, if_name);
8978 8979 8980 8981 8982 8983 8984 8985 8986 8987 8988 8989 8990 8991 8992 8993 8994 8995 8996 8997 8998 8999 9000 9001 9002 9003 9004 9005 9006 9007 9008 9009 9010 9011 9012 9013 9014 9015 9016 9017 9018 9019 9020 9021 9022 9023 9024 9025 9026 9027 9028 9029 9030 9031 9032 9033 9034 9035 9036 9037 9038 9039 9040 9041 9042 9043 9044 9045 9046 9047 9048 9049 9050 9051 9052 9053 9054 9055 9056 9057 9058 9059 9060 9061 9062 9063 9064 9065 9066 9067 9068 9069 9070 9071 9072 9073 9074 9075 9076 9077 9078 9079 9080 9081 9082 9083 9084

    /* start it up unless requested not to */
    if (!nostart) {
        if (virInterfaceCreate(br_handle, 0) < 0) {
            vshError(ctl, _("Failed to start bridge interface %s"), br_name);
            goto cleanup;
        }
        vshPrint(ctl, _("Bridge interface %s started\n"), br_name);
    }

    ret = true;
 cleanup:
    if (if_handle)
       virInterfaceFree(if_handle);
    if (br_handle)
       virInterfaceFree(br_handle);
    VIR_FREE(if_xml);
    VIR_FREE(br_xml);
    VIR_FREE(if_type);
    VIR_FREE(if2_name);
    VIR_FREE(delay_str);
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(xml_doc);
    return ret;
}

/*
 * "iface-unbridge" command
 */
static const vshCmdInfo info_interface_unbridge[] = {
    {"help", N_("undefine a bridge device after detaching its slave device")},
    {"desc", N_("unbridge a network device")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_interface_unbridge[] = {
    {"bridge", VSH_OT_DATA, VSH_OFLAG_REQ, N_("current bridge device name")},
    {"no-start", VSH_OT_BOOL, 0,
     N_("don't start the un-slaved interface immediately (not recommended)")},
    {NULL, 0, 0, NULL}
};

static bool
cmdInterfaceUnbridge(vshControl *ctl, const vshCmd *cmd)
{
    bool ret = false;
    virInterfacePtr if_handle = NULL, br_handle = NULL;
    const char *br_name;
    char *if_type = NULL, *if_name = NULL;
    bool nostart = false;
    char *br_xml = NULL;
    xmlChar *if_xml = NULL;
    int if_xml_size;
    xmlDocPtr xml_doc = NULL;
    xmlXPathContextPtr ctxt = NULL;
    xmlNodePtr top_node, br_node, if_node, cur;

    if (!vshConnectionUsability(ctl, ctl->conn))
        goto cleanup;

    /* Get a handle to the original device */
    if (!(br_handle = vshCommandOptInterfaceBy(ctl, cmd, "bridge",
                                               &br_name, VSH_BYNAME))) {
        goto cleanup;
    }

    nostart = vshCommandOptBool(cmd, "no-start");

    /* Get the bridge xml into an xmlDoc */
    if (!(br_xml = virInterfaceGetXMLDesc(br_handle, VIR_INTERFACE_XML_INACTIVE)))
        goto cleanup;
    if (!(xml_doc = virXMLParseStringCtxt(br_xml,
                                          _("(bridge interface definition)"),
                                          &ctxt))) {
        vshError(ctl, _("Failed to parse configuration of %s"), br_name);
        goto cleanup;
    }
    top_node = ctxt->node;

    /* Verify that the device really is a bridge. */
    if (!(if_type = virXMLPropString(top_node, "type"))) {
        vshError(ctl, _("Existing device %s has no type"), br_name);
        goto cleanup;
    }

    if (STRNEQ(if_type, "bridge")) {
        vshError(ctl, _("Device %s is not a bridge"), br_name);
        goto cleanup;
    }
    VIR_FREE(if_type);

    /* verify the name in the XML matches the device name */
    if (!(if_name = virXMLPropString(top_node, "name")) ||
        STRNEQ(if_name, br_name)) {
        vshError(ctl, _("Interface name from config %s doesn't match given supplied name %s"),
                 if_name, br_name);
        goto cleanup;
    }
    VIR_FREE(if_name);

    /* Find the <bridge> node under <interface>. */
    if (!(br_node = virXPathNode("./bridge", ctxt))) {
        vshError(ctl, "%s", _("No bridge node in xml document"));
        goto cleanup;
    }

    if ((if_node = virXPathNode("./bridge/interface[2]", ctxt))) {
Y
Yuri Chornoivan 已提交
9085
        vshError(ctl, "%s", _("Multiple interfaces attached to bridge"));
9086 9087 9088 9089 9090 9091 9092 9093 9094 9095 9096 9097 9098 9099 9100 9101 9102 9103 9104 9105 9106 9107 9108 9109 9110 9111 9112 9113 9114 9115 9116 9117 9118 9119 9120 9121 9122 9123 9124 9125 9126 9127 9128 9129 9130 9131 9132 9133 9134 9135 9136 9137 9138 9139 9140 9141 9142 9143 9144 9145 9146 9147 9148 9149 9150 9151 9152 9153 9154 9155 9156 9157 9158 9159 9160 9161 9162 9163 9164 9165 9166 9167 9168 9169 9170 9171 9172 9173 9174 9175 9176 9177 9178 9179 9180 9181 9182 9183 9184 9185 9186 9187 9188 9189 9190 9191 9192 9193 9194 9195
        goto cleanup;
    }

    if (!(if_node = virXPathNode("./bridge/interface", ctxt))) {
        vshError(ctl, "%s", _("No interface attached to bridge"));
        goto cleanup;
    }

    /* Change the type and name of the outer/master interface to
     * the type/name of the attached slave interface.
     */
    if (!(if_name = virXMLPropString(if_node, "name"))) {
        vshError(ctl, _("Device attached to bridge %s has no name"), br_name);
        goto cleanup;
    }

    if (!(if_type = virXMLPropString(if_node, "type"))) {
        vshError(ctl, _("Attached device %s has no type"), if_name);
        goto cleanup;
    }

    if (!xmlSetProp(top_node, BAD_CAST "type", BAD_CAST if_type)) {
        vshError(ctl, _("Failed to set interface type to '%s' in xml document"),
                 if_type);
        goto cleanup;
    }

    if (!xmlSetProp(top_node, BAD_CAST "name", BAD_CAST if_name)) {
        vshError(ctl, _("Failed to set interface name to '%s' in xml document"),
                 if_name);
        goto cleanup;
    }

    /* Cycle through all the nodes under the attached <interface>,
     * moving all <mac>, <bond> and <vlan> nodes up into the toplevel
     * <interface>.
     */
    cur = if_node->children;
    while (cur) {
        xmlNodePtr old = cur;

        cur = cur->next;
        if ((old->type == XML_ELEMENT_NODE) &&
            (xmlStrEqual(old->name, BAD_CAST "mac") ||  /* ethernet stuff to move down */
             xmlStrEqual(old->name, BAD_CAST "bond") || /* bond stuff to move down */
             xmlStrEqual(old->name, BAD_CAST "vlan"))) { /* vlan stuff to move down */
            xmlUnlinkNode(old);
            if (!xmlAddChild(top_node, old)) {
                vshError(ctl, _("Failed to move '%s' element in xml document"), old->name);
                xmlFreeNode(old);
                goto cleanup;
            }
        }
    }

    /* The document should now be fully converted; write it out to a string. */
    xmlDocDumpMemory(xml_doc, &if_xml, &if_xml_size);

    if (!if_xml || if_xml_size <= 0) {
        vshError(ctl, _("Failed to format new xml document for un-enslaved interface %s"),
                 if_name);
        goto cleanup;
    }

    /* Destroy and Undefine the bridge device, since we otherwise
     * can't safely define the unattached device.
     */
    if (virInterfaceDestroy(br_handle, 0) < 0) {
        vshError(ctl, _("Failed to destroy bridge interface %s"), br_name);
        goto cleanup;
    }
    if (virInterfaceUndefine(br_handle) < 0) {
        vshError(ctl, _("Failed to undefine bridge interface %s"), br_name);
        goto cleanup;
    }

    /* if_xml is the new interface to define.
     */
    if (!(if_handle = virInterfaceDefineXML(ctl->conn, (char *) if_xml, 0))) {
        vshError(ctl, _("Failed to define new interface %s"), if_name);
        goto cleanup;
    }

    vshPrint(ctl, _("Device %s un-attached from bridge %s\n"),
             if_name, br_name);

    /* unless requested otherwise, undefine the bridge device */
    if (!nostart) {
        if (virInterfaceCreate(if_handle, 0) < 0) {
            vshError(ctl, _("Failed to start interface %s"), if_name);
            goto cleanup;
        }
        vshPrint(ctl, _("Interface %s started\n"), if_name);
    }

    ret = true;
 cleanup:
    if (if_handle)
       virInterfaceFree(if_handle);
    if (br_handle)
       virInterfaceFree(br_handle);
    VIR_FREE(if_xml);
    VIR_FREE(br_xml);
    VIR_FREE(if_type);
    VIR_FREE(if_name);
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(xml_doc);
    return ret;
}

9196 9197 9198 9199 9200 9201 9202 9203 9204 9205 9206 9207 9208 9209
/*
 * "nwfilter-define" command
 */
static const vshCmdInfo info_nwfilter_define[] = {
    {"help", N_("define or update a network filter from an XML file")},
    {"desc", N_("Define a new network filter or update an existing one.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_nwfilter_define[] = {
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file containing an XML network filter description")},
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
9210
static bool
9211 9212 9213
cmdNWFilterDefine(vshControl *ctl, const vshCmd *cmd)
{
    virNWFilterPtr nwfilter;
9214
    const char *from = NULL;
E
Eric Blake 已提交
9215
    bool ret = true;
9216 9217
    char *buffer;

9218
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9219
        return false;
9220

9221
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
9222
        return false;
9223 9224

    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0)
E
Eric Blake 已提交
9225
        return false;
9226 9227 9228 9229 9230 9231 9232 9233 9234 9235

    nwfilter = virNWFilterDefineXML(ctl->conn, buffer);
    VIR_FREE(buffer);

    if (nwfilter != NULL) {
        vshPrint(ctl, _("Network filter %s defined from %s\n"),
                 virNWFilterGetName(nwfilter), from);
        virNWFilterFree(nwfilter);
    } else {
        vshError(ctl, _("Failed to define network filter from %s"), from);
E
Eric Blake 已提交
9236
        ret = false;
9237 9238 9239 9240 9241 9242 9243 9244 9245 9246 9247 9248 9249 9250 9251 9252 9253 9254 9255
    }
    return ret;
}


/*
 * "nwfilter-undefine" command
 */
static const vshCmdInfo info_nwfilter_undefine[] = {
    {"help", N_("undefine a network filter")},
    {"desc", N_("Undefine a given network filter.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_nwfilter_undefine[] = {
    {"nwfilter", VSH_OT_DATA, VSH_OFLAG_REQ, N_("network filter name or uuid")},
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
9256
static bool
9257 9258 9259
cmdNWFilterUndefine(vshControl *ctl, const vshCmd *cmd)
{
    virNWFilterPtr nwfilter;
E
Eric Blake 已提交
9260
    bool ret = true;
9261
    const char *name;
9262

9263
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9264
        return false;
9265 9266

    if (!(nwfilter = vshCommandOptNWFilter(ctl, cmd, &name)))
E
Eric Blake 已提交
9267
        return false;
9268 9269 9270 9271 9272

    if (virNWFilterUndefine(nwfilter) == 0) {
        vshPrint(ctl, _("Network filter %s undefined\n"), name);
    } else {
        vshError(ctl, _("Failed to undefine network filter %s"), name);
E
Eric Blake 已提交
9273
        ret = false;
9274 9275 9276 9277 9278 9279 9280 9281 9282 9283 9284 9285 9286 9287 9288 9289 9290 9291 9292 9293 9294
    }

    virNWFilterFree(nwfilter);
    return ret;
}


/*
 * "nwfilter-dumpxml" command
 */
static const vshCmdInfo info_nwfilter_dumpxml[] = {
    {"help", N_("network filter information in XML")},
    {"desc", N_("Output the network filter information as an XML dump to stdout.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_nwfilter_dumpxml[] = {
    {"nwfilter", VSH_OT_DATA, VSH_OFLAG_REQ, N_("network filter name or uuid")},
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
9295
static bool
9296 9297 9298
cmdNWFilterDumpXML(vshControl *ctl, const vshCmd *cmd)
{
    virNWFilterPtr nwfilter;
E
Eric Blake 已提交
9299
    bool ret = true;
9300 9301
    char *dump;

9302
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9303
        return false;
9304 9305

    if (!(nwfilter = vshCommandOptNWFilter(ctl, cmd, NULL)))
E
Eric Blake 已提交
9306
        return false;
9307 9308 9309

    dump = virNWFilterGetXMLDesc(nwfilter, 0);
    if (dump != NULL) {
9310
        vshPrint(ctl, "%s", dump);
9311 9312
        VIR_FREE(dump);
    } else {
E
Eric Blake 已提交
9313
        ret = false;
9314 9315 9316 9317 9318 9319 9320 9321 9322 9323 9324 9325 9326 9327 9328 9329 9330 9331 9332
    }

    virNWFilterFree(nwfilter);
    return ret;
}

/*
 * "nwfilter-list" command
 */
static const vshCmdInfo info_nwfilter_list[] = {
    {"help", N_("list network filters")},
    {"desc", N_("Returns list of network filters.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_nwfilter_list[] = {
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
9333
static bool
9334 9335 9336 9337 9338 9339
cmdNWFilterList(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
{
    int numfilters, i;
    char **names;
    char uuid[VIR_UUID_STRING_BUFLEN];

9340
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9341
        return false;
9342 9343 9344 9345

    numfilters = virConnectNumOfNWFilters(ctl->conn);
    if (numfilters < 0) {
        vshError(ctl, "%s", _("Failed to list network filters"));
E
Eric Blake 已提交
9346
        return false;
9347 9348 9349 9350 9351 9352 9353 9354
    }

    names = vshMalloc(ctl, sizeof(char *) * numfilters);

    if ((numfilters = virConnectListNWFilters(ctl->conn, names,
                                              numfilters)) < 0) {
        vshError(ctl, "%s", _("Failed to list network filters"));
        VIR_FREE(names);
E
Eric Blake 已提交
9355
        return false;
9356 9357 9358 9359 9360 9361 9362 9363 9364 9365 9366 9367 9368 9369 9370 9371 9372 9373 9374 9375 9376 9377 9378 9379 9380 9381 9382
    }

    qsort(&names[0], numfilters, sizeof(char *), namesorter);

    vshPrintExtra(ctl, "%-36s  %-20s \n", _("UUID"), _("Name"));
    vshPrintExtra(ctl,
       "----------------------------------------------------------------\n");

    for (i = 0; i < numfilters; i++) {
        virNWFilterPtr nwfilter =
            virNWFilterLookupByName(ctl->conn, names[i]);

        /* this kind of work with networks is not atomic operation */
        if (!nwfilter) {
            VIR_FREE(names[i]);
            continue;
        }

        virNWFilterGetUUIDString(nwfilter, uuid);
        vshPrint(ctl, "%-36s  %-20s\n",
                 uuid,
                 virNWFilterGetName(nwfilter));
        virNWFilterFree(nwfilter);
        VIR_FREE(names[i]);
    }

    VIR_FREE(names);
E
Eric Blake 已提交
9383
    return true;
9384 9385 9386 9387 9388 9389 9390 9391 9392 9393 9394 9395 9396 9397 9398 9399 9400
}


/*
 * "nwfilter-edit" command
 */
static const vshCmdInfo info_nwfilter_edit[] = {
    {"help", N_("edit XML configuration for a network filter")},
    {"desc", N_("Edit the XML configuration for a network filter.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_nwfilter_edit[] = {
    {"nwfilter", VSH_OT_DATA, VSH_OFLAG_REQ, N_("network filter name or uuid")},
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
9401
static bool
9402 9403
cmdNWFilterEdit (vshControl *ctl, const vshCmd *cmd)
{
E
Eric Blake 已提交
9404
    bool ret = false;
9405 9406 9407 9408 9409 9410
    virNWFilterPtr nwfilter = NULL;
    char *tmp = NULL;
    char *doc = NULL;
    char *doc_edited = NULL;
    char *doc_reread = NULL;

9411
    if (!vshConnectionUsability(ctl, ctl->conn))
9412 9413 9414 9415 9416 9417 9418 9419 9420 9421 9422 9423 9424 9425 9426 9427 9428 9429 9430 9431 9432 9433 9434 9435 9436 9437
        goto cleanup;

    nwfilter = vshCommandOptNWFilter (ctl, cmd, NULL);
    if (nwfilter == NULL)
        goto cleanup;

    /* Get the XML configuration of the interface. */
    doc = virNWFilterGetXMLDesc (nwfilter, 0);
    if (!doc)
        goto cleanup;

    /* Create and open the temporary file. */
    tmp = editWriteToTempFile (ctl, doc);
    if (!tmp) goto cleanup;

    /* Start the editor. */
    if (editFile (ctl, tmp) == -1) goto cleanup;

    /* Read back the edited file. */
    doc_edited = editReadBackFile (ctl, tmp);
    if (!doc_edited) goto cleanup;

    /* Compare original XML with edited.  Has it changed at all? */
    if (STREQ (doc, doc_edited)) {
        vshPrint (ctl, _("Network filter %s XML configuration not changed.\n"),
                  virNWFilterGetName (nwfilter));
E
Eric Blake 已提交
9438
        ret = true;
9439 9440 9441 9442 9443 9444 9445 9446 9447 9448 9449 9450 9451 9452 9453 9454 9455 9456 9457 9458 9459 9460 9461 9462 9463 9464
        goto cleanup;
    }

    /* Now re-read the network filter XML.  Did someone else change it while
     * it was being edited?  This also catches problems such as us
     * losing a connection or the interface going away.
     */
    doc_reread = virNWFilterGetXMLDesc (nwfilter, 0);
    if (!doc_reread)
        goto cleanup;

    if (STRNEQ (doc, doc_reread)) {
        vshError(ctl, "%s",
                 _("ERROR: the XML configuration was changed by another user"));
        goto cleanup;
    }

    /* Everything checks out, so redefine the interface. */
    virNWFilterFree (nwfilter);
    nwfilter = virNWFilterDefineXML (ctl->conn, doc_edited);
    if (!nwfilter)
        goto cleanup;

    vshPrint (ctl, _("Network filter %s XML configuration edited.\n"),
              virNWFilterGetName(nwfilter));

E
Eric Blake 已提交
9465
    ret = true;
9466 9467 9468 9469 9470 9471 9472 9473 9474 9475 9476 9477 9478 9479 9480 9481 9482 9483

cleanup:
    if (nwfilter)
        virNWFilterFree (nwfilter);

    VIR_FREE(doc);
    VIR_FREE(doc_edited);
    VIR_FREE(doc_reread);

    if (tmp) {
        unlink (tmp);
        VIR_FREE(tmp);
    }

    return ret;
}


9484
/**************************************************************************/
9485
/*
9486
 * "pool-autostart" command
9487
 */
9488
static const vshCmdInfo info_pool_autostart[] = {
9489
    {"help", N_("autostart a pool")},
9490
    {"desc",
9491
     N_("Configure a pool to be automatically started at boot.")},
9492
    {NULL, NULL}
9493 9494
};

9495
static const vshCmdOptDef opts_pool_autostart[] = {
9496 9497
    {"pool",  VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name or uuid")},
    {"disable", VSH_OT_BOOL, 0, N_("disable autostarting")},
9498 9499
    {NULL, 0, 0, NULL}
};
9500

E
Eric Blake 已提交
9501
static bool
9502
cmdPoolAutostart(vshControl *ctl, const vshCmd *cmd)
9503
{
9504
    virStoragePoolPtr pool;
9505
    const char *name;
9506
    int autostart;
9507

9508
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9509
        return false;
9510

9511
    if (!(pool = vshCommandOptPool(ctl, cmd, "pool", &name)))
E
Eric Blake 已提交
9512
        return false;
9513

9514
    autostart = !vshCommandOptBool(cmd, "disable");
9515

9516 9517
    if (virStoragePoolSetAutostart(pool, autostart) < 0) {
        if (autostart)
9518
            vshError(ctl, _("failed to mark pool %s as autostarted"), name);
9519
        else
9520
            vshError(ctl, _("failed to unmark pool %s as autostarted"), name);
9521
        virStoragePoolFree(pool);
E
Eric Blake 已提交
9522
        return false;
9523 9524
    }

9525
    if (autostart)
9526
        vshPrint(ctl, _("Pool %s marked as autostarted\n"), name);
9527
    else
9528
        vshPrint(ctl, _("Pool %s unmarked as autostarted\n"), name);
9529

L
Laine Stump 已提交
9530
    virStoragePoolFree(pool);
E
Eric Blake 已提交
9531
    return true;
9532 9533
}

9534
/*
9535
 * "pool-create" command
9536
 */
9537
static const vshCmdInfo info_pool_create[] = {
9538 9539
    {"help", N_("create a pool from an XML file")},
    {"desc", N_("Create a pool.")},
9540 9541 9542
    {NULL, NULL}
};

9543
static const vshCmdOptDef opts_pool_create[] = {
J
Jim Meyering 已提交
9544
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ,
9545
     N_("file containing an XML pool description")},
9546 9547 9548
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
9549
static bool
9550
cmdPoolCreate(vshControl *ctl, const vshCmd *cmd)
9551
{
9552
    virStoragePoolPtr pool;
9553
    const char *from = NULL;
E
Eric Blake 已提交
9554
    bool ret = true;
9555
    char *buffer;
9556

9557
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9558
        return false;
9559

9560
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
9561
        return false;
9562

9563
    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0)
E
Eric Blake 已提交
9564
        return false;
9565

9566
    pool = virStoragePoolCreateXML(ctl->conn, buffer, 0);
9567
    VIR_FREE(buffer);
9568 9569 9570 9571

    if (pool != NULL) {
        vshPrint(ctl, _("Pool %s created from %s\n"),
                 virStoragePoolGetName(pool), from);
L
Laine Stump 已提交
9572
        virStoragePoolFree(pool);
9573
    } else {
9574
        vshError(ctl, _("Failed to create pool from %s"), from);
E
Eric Blake 已提交
9575
        ret = false;
9576 9577
    }
    return ret;
9578 9579
}

9580

9581 9582 9583 9584
/*
 * "nodedev-create" command
 */
static const vshCmdInfo info_node_device_create[] = {
9585
    {"help", N_("create a device defined "
9586
                          "by an XML file on the node")},
9587
    {"desc", N_("Create a device on the node.  Note that this "
9588 9589 9590 9591 9592 9593 9594
                          "command creates devices on the physical host "
                          "that can then be assigned to a virtual machine.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_node_device_create[] = {
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ,
9595
     N_("file containing an XML description of the device")},
9596 9597 9598
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
9599
static bool
9600 9601 9602
cmdNodeDeviceCreate(vshControl *ctl, const vshCmd *cmd)
{
    virNodeDevicePtr dev = NULL;
9603
    const char *from = NULL;
E
Eric Blake 已提交
9604
    bool ret = true;
9605 9606
    char *buffer;

9607
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9608
        return false;
9609

9610
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
9611
        return false;
9612

9613
    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0)
E
Eric Blake 已提交
9614
        return false;
9615 9616

    dev = virNodeDeviceCreateXML(ctl->conn, buffer, 0);
9617
    VIR_FREE(buffer);
9618 9619 9620 9621

    if (dev != NULL) {
        vshPrint(ctl, _("Node device %s created from %s\n"),
                 virNodeDeviceGetName(dev), from);
L
Laine Stump 已提交
9622
        virNodeDeviceFree(dev);
9623
    } else {
9624
        vshError(ctl, _("Failed to create node device from %s"), from);
E
Eric Blake 已提交
9625
        ret = false;
9626 9627 9628 9629 9630 9631 9632 9633 9634 9635
    }

    return ret;
}


/*
 * "nodedev-destroy" command
 */
static const vshCmdInfo info_node_device_destroy[] = {
9636
    {"help", N_("destroy (stop) a device on the node")},
9637
    {"desc", N_("Destroy a device on the node.  Note that this "
9638
                "command destroys devices on the physical host")},
9639 9640 9641 9642 9643
    {NULL, NULL}
};

static const vshCmdOptDef opts_node_device_destroy[] = {
    {"name", VSH_OT_DATA, VSH_OFLAG_REQ,
9644
     N_("name of the device to be destroyed")},
9645 9646 9647
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
9648
static bool
9649 9650 9651
cmdNodeDeviceDestroy(vshControl *ctl, const vshCmd *cmd)
{
    virNodeDevicePtr dev = NULL;
E
Eric Blake 已提交
9652
    bool ret = true;
9653
    const char *name = NULL;
9654

9655
    if (!vshConnectionUsability(ctl, ctl->conn)) {
E
Eric Blake 已提交
9656
        return false;
9657 9658
    }

9659
    if (vshCommandOptString(cmd, "name", &name) <= 0)
E
Eric Blake 已提交
9660
        return false;
9661 9662 9663 9664 9665 9666

    dev = virNodeDeviceLookupByName(ctl->conn, name);

    if (virNodeDeviceDestroy(dev) == 0) {
        vshPrint(ctl, _("Destroyed node device '%s'\n"), name);
    } else {
9667
        vshError(ctl, _("Failed to destroy node device '%s'"), name);
E
Eric Blake 已提交
9668
        ret = false;
9669 9670 9671 9672 9673 9674 9675
    }

    virNodeDeviceFree(dev);
    return ret;
}


9676
/*
9677
 * XML Building helper for pool-define-as and pool-create-as
9678
 */
9679
static const vshCmdOptDef opts_pool_X_as[] = {
9680 9681 9682 9683 9684 9685 9686 9687
    {"name", VSH_OT_DATA, VSH_OFLAG_REQ, N_("name of the pool")},
    {"print-xml", VSH_OT_BOOL, 0, N_("print XML document, but don't define/create")},
    {"type", VSH_OT_DATA, VSH_OFLAG_REQ, N_("type of the pool")},
    {"source-host", VSH_OT_DATA, 0, N_("source-host for underlying storage")},
    {"source-path", VSH_OT_DATA, 0, N_("source path for underlying storage")},
    {"source-dev", VSH_OT_DATA, 0, N_("source device for underlying storage")},
    {"source-name", VSH_OT_DATA, 0, N_("source name for underlying storage")},
    {"target", VSH_OT_DATA, 0, N_("target for underlying storage")},
9688
    {"source-format", VSH_OT_STRING, 0, N_("format for underlying storage")},
9689 9690 9691
    {NULL, 0, 0, NULL}
};

9692
static int buildPoolXML(const vshCmd *cmd, const char **retname, char **xml) {
9693

9694 9695
    const char *name = NULL, *type = NULL, *srcHost = NULL, *srcPath = NULL,
               *srcDev = NULL, *srcName = NULL, *srcFormat = NULL, *target = NULL;
9696
    virBuffer buf = VIR_BUFFER_INITIALIZER;
9697

9698
    if (vshCommandOptString(cmd, "name", &name) <= 0)
9699
        goto cleanup;
9700
    if (vshCommandOptString(cmd, "type", &type) <= 0)
9701 9702
        goto cleanup;

9703 9704 9705 9706 9707
    if (vshCommandOptString(cmd, "source-host", &srcHost) < 0 ||
        vshCommandOptString(cmd, "source-path", &srcPath) < 0 ||
        vshCommandOptString(cmd, "source-dev", &srcDev) < 0 ||
        vshCommandOptString(cmd, "source-name", &srcName) < 0 ||
        vshCommandOptString(cmd, "source-format", &srcFormat) < 0 ||
9708 9709
        vshCommandOptString(cmd, "target", &target) < 0) {
        vshError(NULL, "%s", _("missing argument"));
9710
        goto cleanup;
9711
    }
9712

9713 9714
    virBufferAsprintf(&buf, "<pool type='%s'>\n", type);
    virBufferAsprintf(&buf, "  <name>%s</name>\n", name);
9715
    if (srcHost || srcPath || srcDev || srcFormat || srcName) {
9716
        virBufferAddLit(&buf, "  <source>\n");
9717

9718
        if (srcHost)
9719
            virBufferAsprintf(&buf, "    <host name='%s'/>\n", srcHost);
9720
        if (srcPath)
9721
            virBufferAsprintf(&buf, "    <dir path='%s'/>\n", srcPath);
9722
        if (srcDev)
9723
            virBufferAsprintf(&buf, "    <device path='%s'/>\n", srcDev);
9724
        if (srcFormat)
9725
            virBufferAsprintf(&buf, "    <format type='%s'/>\n", srcFormat);
9726
        if (srcName)
9727
            virBufferAsprintf(&buf, "    <name>%s</name>\n", srcName);
9728 9729

        virBufferAddLit(&buf, "  </source>\n");
9730 9731
    }
    if (target) {
9732
        virBufferAddLit(&buf, "  <target>\n");
9733
        virBufferAsprintf(&buf, "    <path>%s</path>\n", target);
9734
        virBufferAddLit(&buf, "  </target>\n");
9735
    }
9736 9737 9738 9739
    virBufferAddLit(&buf, "</pool>\n");

    if (virBufferError(&buf)) {
        vshPrint(ctl, "%s", _("Failed to allocate XML buffer"));
E
Eric Blake 已提交
9740
        return false;
9741
    }
9742 9743 9744

    *xml = virBufferContentAndReset(&buf);
    *retname = name;
E
Eric Blake 已提交
9745
    return true;
9746 9747

cleanup:
9748
    virBufferFreeAndReset(&buf);
E
Eric Blake 已提交
9749
    return false;
9750 9751 9752 9753 9754 9755
}

/*
 * "pool-create-as" command
 */
static const vshCmdInfo info_pool_create_as[] = {
9756 9757
    {"help", N_("create a pool from a set of args")},
    {"desc", N_("Create a pool.")},
9758 9759 9760
    {NULL, NULL}
};

E
Eric Blake 已提交
9761
static bool
9762 9763 9764
cmdPoolCreateAs(vshControl *ctl, const vshCmd *cmd)
{
    virStoragePoolPtr pool;
9765 9766
    const char *name;
    char *xml;
9767
    int printXML = vshCommandOptBool(cmd, "print-xml");
9768

9769
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9770
        return false;
9771 9772

    if (!buildPoolXML(cmd, &name, &xml))
E
Eric Blake 已提交
9773
        return false;
9774

9775
    if (printXML) {
9776
        vshPrint(ctl, "%s", xml);
9777
        VIR_FREE(xml);
9778
    } else {
9779
        pool = virStoragePoolCreateXML(ctl->conn, xml, 0);
9780
        VIR_FREE(xml);
9781

9782 9783 9784 9785
        if (pool != NULL) {
            vshPrint(ctl, _("Pool %s created\n"), name);
            virStoragePoolFree(pool);
        } else {
9786
            vshError(ctl, _("Failed to create pool %s"), name);
E
Eric Blake 已提交
9787
            return false;
9788 9789
        }
    }
E
Eric Blake 已提交
9790
    return true;
9791 9792
}

9793

9794
/*
9795
 * "pool-define" command
9796
 */
9797
static const vshCmdInfo info_pool_define[] = {
9798 9799
    {"help", N_("define (but don't start) a pool from an XML file")},
    {"desc", N_("Define a pool.")},
9800 9801 9802
    {NULL, NULL}
};

9803
static const vshCmdOptDef opts_pool_define[] = {
9804
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file containing an XML pool description")},
9805 9806 9807
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
9808
static bool
9809
cmdPoolDefine(vshControl *ctl, const vshCmd *cmd)
9810
{
9811
    virStoragePoolPtr pool;
9812
    const char *from = NULL;
E
Eric Blake 已提交
9813
    bool ret = true;
9814
    char *buffer;
9815

9816
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9817
        return false;
9818

9819
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
9820
        return false;
9821

9822
    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0)
E
Eric Blake 已提交
9823
        return false;
9824

9825
    pool = virStoragePoolDefineXML(ctl->conn, buffer, 0);
9826
    VIR_FREE(buffer);
9827 9828 9829 9830

    if (pool != NULL) {
        vshPrint(ctl, _("Pool %s defined from %s\n"),
                 virStoragePoolGetName(pool), from);
L
Laine Stump 已提交
9831
        virStoragePoolFree(pool);
9832
    } else {
9833
        vshError(ctl, _("Failed to define pool from %s"), from);
E
Eric Blake 已提交
9834
        ret = false;
9835 9836
    }
    return ret;
9837 9838
}

9839

9840 9841 9842
/*
 * "pool-define-as" command
 */
9843
static const vshCmdInfo info_pool_define_as[] = {
9844 9845
    {"help", N_("define a pool from a set of args")},
    {"desc", N_("Define a pool.")},
9846 9847 9848
    {NULL, NULL}
};

E
Eric Blake 已提交
9849
static bool
9850
cmdPoolDefineAs(vshControl *ctl, const vshCmd *cmd)
9851 9852
{
    virStoragePoolPtr pool;
9853 9854
    const char *name;
    char *xml;
9855
    int printXML = vshCommandOptBool(cmd, "print-xml");
9856

9857
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9858
        return false;
9859

9860
    if (!buildPoolXML(cmd, &name, &xml))
E
Eric Blake 已提交
9861
        return false;
9862

9863
    if (printXML) {
9864
        vshPrint(ctl, "%s", xml);
9865
        VIR_FREE(xml);
9866
    } else {
9867
        pool = virStoragePoolDefineXML(ctl->conn, xml, 0);
9868
        VIR_FREE(xml);
9869

9870 9871 9872 9873
        if (pool != NULL) {
            vshPrint(ctl, _("Pool %s defined\n"), name);
            virStoragePoolFree(pool);
        } else {
9874
            vshError(ctl, _("Failed to define pool %s"), name);
E
Eric Blake 已提交
9875
            return false;
9876 9877
        }
    }
E
Eric Blake 已提交
9878
    return true;
9879 9880 9881
}


9882
/*
9883
 * "pool-build" command
9884
 */
9885
static const vshCmdInfo info_pool_build[] = {
9886 9887
    {"help", N_("build a pool")},
    {"desc", N_("Build a given pool.")},
9888 9889 9890
    {NULL, NULL}
};

9891
static const vshCmdOptDef opts_pool_build[] = {
9892
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name or uuid")},
9893 9894
    {"no-overwrite", VSH_OT_BOOL, 0, N_("do not overwrite an existing pool of this type")},
    {"overwrite", VSH_OT_BOOL, 0, N_("overwrite any existing data")},
9895 9896 9897
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
9898
static bool
9899
cmdPoolBuild(vshControl *ctl, const vshCmd *cmd)
9900
{
9901
    virStoragePoolPtr pool;
E
Eric Blake 已提交
9902
    bool ret = true;
9903
    const char *name;
9904
    unsigned int flags = 0;
9905

9906
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9907
        return false;
9908

9909
    if (!(pool = vshCommandOptPool(ctl, cmd, "pool", &name)))
E
Eric Blake 已提交
9910
        return false;
9911

9912 9913 9914 9915 9916 9917 9918 9919 9920
    if (vshCommandOptBool (cmd, "no-overwrite")) {
        flags |= VIR_STORAGE_POOL_BUILD_NO_OVERWRITE;
    }

    if (vshCommandOptBool (cmd, "overwrite")) {
        flags |= VIR_STORAGE_POOL_BUILD_OVERWRITE;
    }

    if (virStoragePoolBuild(pool, flags) == 0) {
9921
        vshPrint(ctl, _("Pool %s built\n"), name);
9922
    } else {
9923
        vshError(ctl, _("Failed to build pool %s"), name);
E
Eric Blake 已提交
9924
        ret = false;
9925 9926
    }

9927 9928
    virStoragePoolFree(pool);

9929 9930 9931
    return ret;
}

9932
/*
9933
 * "pool-destroy" command
9934
 */
9935
static const vshCmdInfo info_pool_destroy[] = {
9936 9937 9938
    {"help", N_("destroy (stop) a pool")},
    {"desc",
     N_("Forcefully stop a given pool. Raw data in the pool is untouched")},
9939 9940 9941
    {NULL, NULL}
};

9942
static const vshCmdOptDef opts_pool_destroy[] = {
9943
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name or uuid")},
9944 9945 9946
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
9947
static bool
9948
cmdPoolDestroy(vshControl *ctl, const vshCmd *cmd)
9949
{
9950
    virStoragePoolPtr pool;
E
Eric Blake 已提交
9951
    bool ret = true;
9952
    const char *name;
9953

9954
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9955
        return false;
9956

9957
    if (!(pool = vshCommandOptPool(ctl, cmd, "pool", &name)))
E
Eric Blake 已提交
9958
        return false;
9959

9960 9961 9962
    if (virStoragePoolDestroy(pool) == 0) {
        vshPrint(ctl, _("Pool %s destroyed\n"), name);
    } else {
9963
        vshError(ctl, _("Failed to destroy pool %s"), name);
E
Eric Blake 已提交
9964
        ret = false;
9965 9966
    }

9967
    virStoragePoolFree(pool);
9968 9969 9970
    return ret;
}

9971

9972
/*
9973 9974
 * "pool-delete" command
 */
9975
static const vshCmdInfo info_pool_delete[] = {
9976 9977
    {"help", N_("delete a pool")},
    {"desc", N_("Delete a given pool.")},
9978 9979 9980
    {NULL, NULL}
};

9981
static const vshCmdOptDef opts_pool_delete[] = {
9982
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name or uuid")},
9983 9984 9985
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
9986
static bool
9987
cmdPoolDelete(vshControl *ctl, const vshCmd *cmd)
9988 9989
{
    virStoragePoolPtr pool;
E
Eric Blake 已提交
9990
    bool ret = true;
9991
    const char *name;
9992

9993
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
9994
        return false;
9995 9996

    if (!(pool = vshCommandOptPool(ctl, cmd, "pool", &name)))
E
Eric Blake 已提交
9997
        return false;
9998 9999

    if (virStoragePoolDelete(pool, 0) == 0) {
D
Daniel Veillard 已提交
10000
        vshPrint(ctl, _("Pool %s deleted\n"), name);
10001
    } else {
10002
        vshError(ctl, _("Failed to delete pool %s"), name);
E
Eric Blake 已提交
10003
        ret = false;
10004 10005
    }

L
Laine Stump 已提交
10006
    virStoragePoolFree(pool);
10007 10008 10009 10010 10011 10012 10013
    return ret;
}


/*
 * "pool-refresh" command
 */
10014
static const vshCmdInfo info_pool_refresh[] = {
10015 10016
    {"help", N_("refresh a pool")},
    {"desc", N_("Refresh a given pool.")},
10017 10018 10019
    {NULL, NULL}
};

10020
static const vshCmdOptDef opts_pool_refresh[] = {
10021
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name or uuid")},
10022 10023 10024
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
10025
static bool
10026
cmdPoolRefresh(vshControl *ctl, const vshCmd *cmd)
10027 10028
{
    virStoragePoolPtr pool;
E
Eric Blake 已提交
10029
    bool ret = true;
10030
    const char *name;
10031

10032
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
10033
        return false;
10034 10035

    if (!(pool = vshCommandOptPool(ctl, cmd, "pool", &name)))
E
Eric Blake 已提交
10036
        return false;
10037 10038 10039 10040

    if (virStoragePoolRefresh(pool, 0) == 0) {
        vshPrint(ctl, _("Pool %s refreshed\n"), name);
    } else {
10041
        vshError(ctl, _("Failed to refresh pool %s"), name);
E
Eric Blake 已提交
10042
        ret = false;
10043 10044 10045 10046 10047 10048 10049 10050 10051 10052
    }
    virStoragePoolFree(pool);

    return ret;
}


/*
 * "pool-dumpxml" command
 */
10053
static const vshCmdInfo info_pool_dumpxml[] = {
10054 10055
    {"help", N_("pool information in XML")},
    {"desc", N_("Output the pool information as an XML dump to stdout.")},
10056 10057 10058
    {NULL, NULL}
};

10059
static const vshCmdOptDef opts_pool_dumpxml[] = {
10060
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name or uuid")},
10061 10062 10063
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
10064
static bool
10065
cmdPoolDumpXML(vshControl *ctl, const vshCmd *cmd)
10066 10067
{
    virStoragePoolPtr pool;
E
Eric Blake 已提交
10068
    bool ret = true;
10069 10070
    char *dump;

10071
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
10072
        return false;
10073 10074

    if (!(pool = vshCommandOptPool(ctl, cmd, "pool", NULL)))
E
Eric Blake 已提交
10075
        return false;
10076 10077 10078

    dump = virStoragePoolGetXMLDesc(pool, 0);
    if (dump != NULL) {
10079
        vshPrint(ctl, "%s", dump);
10080
        VIR_FREE(dump);
10081
    } else {
E
Eric Blake 已提交
10082
        ret = false;
10083 10084 10085 10086 10087 10088 10089 10090 10091 10092
    }

    virStoragePoolFree(pool);
    return ret;
}


/*
 * "pool-list" command
 */
10093
static const vshCmdInfo info_pool_list[] = {
10094 10095
    {"help", N_("list pools")},
    {"desc", N_("Returns list of pools.")},
10096 10097 10098
    {NULL, NULL}
};

10099
static const vshCmdOptDef opts_pool_list[] = {
10100 10101
    {"inactive", VSH_OT_BOOL, 0, N_("list inactive pools")},
    {"all", VSH_OT_BOOL, 0, N_("list inactive & active pools")},
10102
    {"details", VSH_OT_BOOL, 0, N_("display extended details for pools")},
10103 10104 10105
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
10106
static bool
10107
cmdPoolList(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
10108
{
10109 10110
    virStoragePoolInfo info;
    char **poolNames = NULL;
W
Wen Congyang 已提交
10111 10112
    int i, ret;
    bool functionReturn;
10113 10114 10115 10116 10117 10118 10119 10120 10121 10122 10123 10124 10125 10126 10127 10128
    int numActivePools = 0, numInactivePools = 0, numAllPools = 0;
    size_t stringLength = 0, nameStrLength = 0;
    size_t autostartStrLength = 0, persistStrLength = 0;
    size_t stateStrLength = 0, capStrLength = 0;
    size_t allocStrLength = 0, availStrLength = 0;
    struct poolInfoText {
        char *state;
        char *autostart;
        char *persistent;
        char *capacity;
        char *allocation;
        char *available;
    };
    struct poolInfoText *poolInfoTexts = NULL;

    /* Determine the options passed by the user */
10129
    int all = vshCommandOptBool(cmd, "all");
10130 10131
    int details = vshCommandOptBool(cmd, "details");
    int inactive = vshCommandOptBool(cmd, "inactive");
10132 10133 10134
    int active = !inactive || all ? 1 : 0;
    inactive |= all;

10135
    /* Check the connection to libvirtd daemon is still working */
10136
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
10137
        return false;
10138

10139
    /* Retrieve the number of active storage pools */
10140
    if (active) {
10141 10142
        numActivePools = virConnectNumOfStoragePools(ctl->conn);
        if (numActivePools < 0) {
10143
            vshError(ctl, "%s", _("Failed to list active pools"));
E
Eric Blake 已提交
10144
            return false;
10145 10146
        }
    }
10147 10148

    /* Retrieve the number of inactive storage pools */
10149
    if (inactive) {
10150 10151
        numInactivePools = virConnectNumOfDefinedStoragePools(ctl->conn);
        if (numInactivePools < 0) {
10152
            vshError(ctl, "%s", _("Failed to list inactive pools"));
E
Eric Blake 已提交
10153
            return false;
10154
        }
10155
    }
10156

10157 10158
    /* Determine the total number of pools to list */
    numAllPools = numActivePools + numInactivePools;
10159

10160 10161 10162 10163 10164 10165 10166 10167 10168 10169 10170 10171
    /* Allocate memory for arrays of storage pool names and info */
    poolNames = vshCalloc(ctl, numAllPools, sizeof(*poolNames));
    poolInfoTexts =
        vshCalloc(ctl, numAllPools, sizeof(*poolInfoTexts));

    /* Retrieve a list of active storage pool names */
    if (active) {
        if ((virConnectListStoragePools(ctl->conn,
                                        poolNames, numActivePools)) < 0) {
            vshError(ctl, "%s", _("Failed to list active pools"));
            VIR_FREE(poolInfoTexts);
            VIR_FREE(poolNames);
E
Eric Blake 已提交
10172
            return false;
10173 10174 10175
        }
    }

10176 10177 10178 10179 10180 10181 10182 10183
    /* Add the inactive storage pools to the end of the name list */
    if (inactive) {
        if ((virConnectListDefinedStoragePools(ctl->conn,
                                               &poolNames[numActivePools],
                                               numInactivePools)) < 0) {
            vshError(ctl, "%s", _("Failed to list inactive pools"));
            VIR_FREE(poolInfoTexts);
            VIR_FREE(poolNames);
E
Eric Blake 已提交
10184
            return false;
10185 10186
        }
    }
10187

10188 10189 10190 10191 10192 10193 10194 10195 10196 10197
    /* Sort the storage pool names */
    qsort(poolNames, numAllPools, sizeof(*poolNames), namesorter);

    /* Collect the storage pool information for display */
    for (i = 0; i < numAllPools; i++) {
        int autostart = 0, persistent = 0;

        /* Retrieve a pool object, looking it up by name */
        virStoragePoolPtr pool = virStoragePoolLookupByName(ctl->conn,
                                                            poolNames[i]);
10198
        if (!pool) {
10199
            VIR_FREE(poolNames[i]);
10200 10201 10202
            continue;
        }

10203
        /* Retrieve the autostart status of the pool */
10204
        if (virStoragePoolGetAutostart(pool, &autostart) < 0)
10205
            poolInfoTexts[i].autostart = vshStrdup(ctl, _("no autostart"));
10206
        else
10207 10208 10209 10210 10211 10212
            poolInfoTexts[i].autostart = vshStrdup(ctl, autostart ?
                                                    _("yes") : _("no"));

        /* Retrieve the persistence status of the pool */
        if (details) {
            persistent = virStoragePoolIsPersistent(pool);
10213 10214
            vshDebug(ctl, VSH_ERR_DEBUG, "Persistent flag value: %d\n",
                     persistent);
10215 10216 10217 10218 10219 10220 10221 10222 10223 10224 10225
            if (persistent < 0)
                poolInfoTexts[i].persistent = vshStrdup(ctl, _("unknown"));
            else
                poolInfoTexts[i].persistent = vshStrdup(ctl, persistent ?
                                                         _("yes") : _("no"));

            /* Keep the length of persistent string if longest so far */
            stringLength = strlen(poolInfoTexts[i].persistent);
            if (stringLength > persistStrLength)
                persistStrLength = stringLength;
        }
10226

10227 10228 10229 10230 10231 10232 10233 10234 10235 10236 10237 10238 10239 10240 10241 10242 10243 10244 10245 10246 10247 10248 10249 10250 10251 10252 10253 10254 10255 10256 10257 10258 10259 10260 10261 10262 10263 10264 10265 10266 10267 10268 10269 10270 10271 10272 10273 10274 10275 10276 10277 10278 10279 10280 10281 10282 10283 10284 10285 10286 10287 10288 10289 10290 10291 10292 10293 10294 10295 10296 10297 10298 10299 10300 10301 10302 10303 10304 10305 10306 10307 10308 10309 10310 10311 10312 10313 10314 10315 10316 10317 10318 10319 10320 10321 10322 10323 10324 10325 10326 10327 10328 10329 10330 10331 10332 10333 10334 10335 10336 10337 10338
        /* Collect further extended information about the pool */
        if (virStoragePoolGetInfo(pool, &info) != 0) {
            /* Something went wrong retrieving pool info, cope with it */
            vshError(ctl, "%s", _("Could not retrieve pool information"));
            poolInfoTexts[i].state = vshStrdup(ctl, _("unknown"));
            if (details) {
                poolInfoTexts[i].capacity = vshStrdup(ctl, _("unknown"));
                poolInfoTexts[i].allocation = vshStrdup(ctl, _("unknown"));
                poolInfoTexts[i].available = vshStrdup(ctl, _("unknown"));
            }
        } else {
            /* Decide which state string to display */
            if (details) {
                /* --details option was specified, we're using detailed state
                 * strings */
                switch (info.state) {
                case VIR_STORAGE_POOL_INACTIVE:
                    poolInfoTexts[i].state = vshStrdup(ctl, _("inactive"));
                    break;
                case VIR_STORAGE_POOL_BUILDING:
                    poolInfoTexts[i].state = vshStrdup(ctl, _("building"));
                    break;
                case VIR_STORAGE_POOL_RUNNING:
                    poolInfoTexts[i].state = vshStrdup(ctl, _("running"));
                    break;
                case VIR_STORAGE_POOL_DEGRADED:
                    poolInfoTexts[i].state = vshStrdup(ctl, _("degraded"));
                    break;
                case VIR_STORAGE_POOL_INACCESSIBLE:
                    poolInfoTexts[i].state = vshStrdup(ctl, _("inaccessible"));
                    break;
                }

                /* Create the pool size related strings */
                if (info.state == VIR_STORAGE_POOL_RUNNING ||
                    info.state == VIR_STORAGE_POOL_DEGRADED) {
                    double val;
                    const char *unit;

                    /* Create the capacity output string */
                    val = prettyCapacity(info.capacity, &unit);
                    ret = virAsprintf(&poolInfoTexts[i].capacity,
                                      "%.2lf %s", val, unit);
                    if (ret < 0) {
                        /* An error occurred creating the string, return */
                        goto asprintf_failure;
                    }

                    /* Create the allocation output string */
                    val = prettyCapacity(info.allocation, &unit);
                    ret = virAsprintf(&poolInfoTexts[i].allocation,
                                      "%.2lf %s", val, unit);
                    if (ret < 0) {
                        /* An error occurred creating the string, return */
                        goto asprintf_failure;
                    }

                    /* Create the available space output string */
                    val = prettyCapacity(info.available, &unit);
                    ret = virAsprintf(&poolInfoTexts[i].available,
                                      "%.2lf %s", val, unit);
                    if (ret < 0) {
                        /* An error occurred creating the string, return */
                        goto asprintf_failure;
                    }
                } else {
                    /* Capacity related information isn't available */
                    poolInfoTexts[i].capacity = vshStrdup(ctl, _("-"));
                    poolInfoTexts[i].allocation = vshStrdup(ctl, _("-"));
                    poolInfoTexts[i].available = vshStrdup(ctl, _("-"));
                }

                /* Keep the length of capacity string if longest so far */
                stringLength = strlen(poolInfoTexts[i].capacity);
                if (stringLength > capStrLength)
                    capStrLength = stringLength;

                /* Keep the length of allocation string if longest so far */
                stringLength = strlen(poolInfoTexts[i].allocation);
                if (stringLength > allocStrLength)
                    allocStrLength = stringLength;

                /* Keep the length of available string if longest so far */
                stringLength = strlen(poolInfoTexts[i].available);
                if (stringLength > availStrLength)
                    availStrLength = stringLength;
            } else {
                /* --details option was not specified, only active/inactive
                * state strings are used */
                if (info.state == VIR_STORAGE_POOL_INACTIVE)
                    poolInfoTexts[i].state = vshStrdup(ctl, _("inactive"));
                else
                    poolInfoTexts[i].state = vshStrdup(ctl, _("active"));
            }
        }

        /* Keep the length of name string if longest so far */
        stringLength = strlen(poolNames[i]);
        if (stringLength > nameStrLength)
            nameStrLength = stringLength;

        /* Keep the length of state string if longest so far */
        stringLength = strlen(poolInfoTexts[i].state);
        if (stringLength > stateStrLength)
            stateStrLength = stringLength;

        /* Keep the length of autostart string if longest so far */
        stringLength = strlen(poolInfoTexts[i].autostart);
        if (stringLength > autostartStrLength)
            autostartStrLength = stringLength;

        /* Free the pool object */
10339 10340 10341
        virStoragePoolFree(pool);
    }

10342 10343 10344 10345 10346 10347 10348 10349 10350 10351 10352 10353 10354 10355 10356 10357 10358 10359
    /* If the --details option wasn't selected, we output the pool
     * info using the fixed string format from previous versions to
     * maintain backward compatibility.
     */

    /* Output basic info then return if --details option not selected */
    if (!details) {
        /* Output old style header */
        vshPrintExtra(ctl, "%-20s %-10s %-10s\n", _("Name"), _("State"),
                      _("Autostart"));
        vshPrintExtra(ctl, "-----------------------------------------\n");

        /* Output old style pool info */
        for (i = 0; i < numAllPools; i++) {
            vshPrint(ctl, "%-20s %-10s %-10s\n",
                 poolNames[i],
                 poolInfoTexts[i].state,
                 poolInfoTexts[i].autostart);
10360 10361
        }

10362
        /* Cleanup and return */
E
Eric Blake 已提交
10363
        functionReturn = true;
10364 10365
        goto cleanup;
    }
10366

10367 10368 10369 10370 10371 10372 10373 10374 10375 10376 10377 10378 10379 10380 10381 10382 10383 10384 10385 10386 10387 10388 10389 10390 10391 10392 10393 10394 10395 10396 10397 10398 10399 10400 10401 10402 10403 10404
    /* We only get here if the --details option was selected. */

    /* Use the length of name header string if it's longest */
    stringLength = strlen(_("Name"));
    if (stringLength > nameStrLength)
        nameStrLength = stringLength;

    /* Use the length of state header string if it's longest */
    stringLength = strlen(_("State"));
    if (stringLength > stateStrLength)
        stateStrLength = stringLength;

    /* Use the length of autostart header string if it's longest */
    stringLength = strlen(_("Autostart"));
    if (stringLength > autostartStrLength)
        autostartStrLength = stringLength;

    /* Use the length of persistent header string if it's longest */
    stringLength = strlen(_("Persistent"));
    if (stringLength > persistStrLength)
        persistStrLength = stringLength;

    /* Use the length of capacity header string if it's longest */
    stringLength = strlen(_("Capacity"));
    if (stringLength > capStrLength)
        capStrLength = stringLength;

    /* Use the length of allocation header string if it's longest */
    stringLength = strlen(_("Allocation"));
    if (stringLength > allocStrLength)
        allocStrLength = stringLength;

    /* Use the length of available header string if it's longest */
    stringLength = strlen(_("Available"));
    if (stringLength > availStrLength)
        availStrLength = stringLength;

    /* Display the string lengths for debugging. */
10405
    vshDebug(ctl, VSH_ERR_DEBUG, "Longest name string = %lu chars\n",
10406
             (unsigned long) nameStrLength);
10407
    vshDebug(ctl, VSH_ERR_DEBUG, "Longest state string = %lu chars\n",
10408
             (unsigned long) stateStrLength);
10409
    vshDebug(ctl, VSH_ERR_DEBUG, "Longest autostart string = %lu chars\n",
10410
             (unsigned long) autostartStrLength);
10411
    vshDebug(ctl, VSH_ERR_DEBUG, "Longest persistent string = %lu chars\n",
10412
             (unsigned long) persistStrLength);
10413
    vshDebug(ctl, VSH_ERR_DEBUG, "Longest capacity string = %lu chars\n",
10414
             (unsigned long) capStrLength);
10415
    vshDebug(ctl, VSH_ERR_DEBUG, "Longest allocation string = %lu chars\n",
10416
             (unsigned long) allocStrLength);
10417
    vshDebug(ctl, VSH_ERR_DEBUG, "Longest available string = %lu chars\n",
10418 10419 10420 10421 10422 10423 10424 10425 10426 10427 10428 10429 10430 10431 10432 10433 10434 10435 10436 10437 10438 10439 10440 10441 10442 10443 10444 10445 10446 10447 10448 10449 10450 10451 10452 10453 10454 10455 10456 10457 10458 10459 10460
             (unsigned long) availStrLength);

    /* Create the output template.  Each column is sized according to
     * the longest string.
     */
    char *outputStr;
    ret = virAsprintf(&outputStr,
              "%%-%lus  %%-%lus  %%-%lus  %%-%lus  %%%lus  %%%lus  %%%lus\n",
              (unsigned long) nameStrLength,
              (unsigned long) stateStrLength,
              (unsigned long) autostartStrLength,
              (unsigned long) persistStrLength,
              (unsigned long) capStrLength,
              (unsigned long) allocStrLength,
              (unsigned long) availStrLength);
    if (ret < 0) {
        /* An error occurred creating the string, return */
        goto asprintf_failure;
    }

    /* Display the header */
    vshPrint(ctl, outputStr, _("Name"), _("State"), _("Autostart"),
             _("Persistent"), _("Capacity"), _("Allocation"), _("Available"));
    for (i = nameStrLength + stateStrLength + autostartStrLength
                           + persistStrLength + capStrLength
                           + allocStrLength + availStrLength
                           + 12; i > 0; i--)
        vshPrintExtra(ctl, "-");
    vshPrintExtra(ctl, "\n");

    /* Display the pool info rows */
    for (i = 0; i < numAllPools; i++) {
        vshPrint(ctl, outputStr,
                 poolNames[i],
                 poolInfoTexts[i].state,
                 poolInfoTexts[i].autostart,
                 poolInfoTexts[i].persistent,
                 poolInfoTexts[i].capacity,
                 poolInfoTexts[i].allocation,
                 poolInfoTexts[i].available);
    }

    /* Cleanup and return */
E
Eric Blake 已提交
10461
    functionReturn = true;
10462 10463 10464 10465 10466 10467 10468 10469 10470 10471 10472 10473 10474 10475
    goto cleanup;

asprintf_failure:

    /* Display an appropriate error message then cleanup and return */
    switch (errno) {
    case ENOMEM:
        /* Couldn't allocate memory */
        vshError(ctl, "%s", _("Out of memory"));
        break;
    default:
        /* Some other error */
        vshError(ctl, _("virAsprintf failed (errno %d)"), errno);
    }
E
Eric Blake 已提交
10476
    functionReturn = false;
10477

10478 10479 10480 10481 10482 10483 10484 10485 10486 10487 10488 10489
cleanup:

    /* Safely free the memory allocated in this function */
    for (i = 0; i < numAllPools; i++) {
        /* Cleanup the memory for one pool info structure */
        VIR_FREE(poolInfoTexts[i].state);
        VIR_FREE(poolInfoTexts[i].autostart);
        VIR_FREE(poolInfoTexts[i].persistent);
        VIR_FREE(poolInfoTexts[i].capacity);
        VIR_FREE(poolInfoTexts[i].allocation);
        VIR_FREE(poolInfoTexts[i].available);
        VIR_FREE(poolNames[i]);
10490
    }
10491 10492 10493 10494 10495 10496 10497

    /* Cleanup the memory for the initial arrays*/
    VIR_FREE(poolInfoTexts);
    VIR_FREE(poolNames);

    /* Return the desired value */
    return functionReturn;
10498 10499
}

10500 10501 10502 10503
/*
 * "find-storage-pool-sources-as" command
 */
static const vshCmdInfo info_find_storage_pool_sources_as[] = {
10504 10505
    {"help", N_("find potential storage pool sources")},
    {"desc", N_("Returns XML <sources> document.")},
10506 10507 10508 10509 10510
    {NULL, NULL}
};

static const vshCmdOptDef opts_find_storage_pool_sources_as[] = {
    {"type", VSH_OT_DATA, VSH_OFLAG_REQ,
10511 10512 10513
     N_("type of storage pool sources to find")},
    {"host", VSH_OT_DATA, VSH_OFLAG_NONE, N_("optional host to query")},
    {"port", VSH_OT_DATA, VSH_OFLAG_NONE, N_("optional port to query")},
10514
    {"initiator", VSH_OT_DATA, VSH_OFLAG_NONE, N_("optional initiator IQN to use for query")},
10515 10516 10517
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
10518
static bool
10519 10520
cmdPoolDiscoverSourcesAs(vshControl * ctl, const vshCmd * cmd ATTRIBUTE_UNUSED)
{
10521
    const char *type = NULL, *host = NULL;
10522 10523
    char *srcSpec = NULL;
    char *srcList;
10524
    const char *initiator = NULL;
10525

10526 10527
    if (vshCommandOptString(cmd, "type", &type) <= 0 ||
        vshCommandOptString(cmd, "host", &host) < 0 ||
10528 10529
        vshCommandOptString(cmd, "initiator", &initiator) < 0) {
        vshError(ctl,"%s", _("missing argument"));
E
Eric Blake 已提交
10530
        return false;
10531
    }
10532

10533
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
10534
        return false;
10535 10536

    if (host) {
10537
        const char *port = NULL;
10538
        virBuffer buf = VIR_BUFFER_INITIALIZER;
10539 10540 10541 10542

        if (vshCommandOptString(cmd, "port", &port) < 0) {
            vshError(ctl, "%s", _("missing argument"));
            virBufferFreeAndReset(&buf);
E
Eric Blake 已提交
10543
            return false;
10544
        }
10545
        virBufferAddLit(&buf, "<source>\n");
10546
        virBufferAsprintf(&buf, "  <host name='%s'", host);
10547
        if (port)
10548
            virBufferAsprintf(&buf, " port='%s'", port);
10549
        virBufferAddLit(&buf, "/>\n");
10550 10551
        if (initiator) {
            virBufferAddLit(&buf, "  <initiator>\n");
10552
            virBufferAsprintf(&buf, "    <iqn name='%s'/>\n", initiator);
10553 10554
            virBufferAddLit(&buf, "  </initiator>\n");
        }
10555 10556 10557
        virBufferAddLit(&buf, "</source>\n");
        if (virBufferError(&buf)) {
            vshError(ctl, "%s", _("Out of memory"));
E
Eric Blake 已提交
10558
            return false;
10559
        }
10560
        srcSpec = virBufferContentAndReset(&buf);
10561 10562 10563
    }

    srcList = virConnectFindStoragePoolSources(ctl->conn, type, srcSpec, 0);
10564
    VIR_FREE(srcSpec);
10565
    if (srcList == NULL) {
10566
        vshError(ctl, _("Failed to find any %s pool sources"), type);
E
Eric Blake 已提交
10567
        return false;
10568 10569
    }
    vshPrint(ctl, "%s", srcList);
10570
    VIR_FREE(srcList);
10571

E
Eric Blake 已提交
10572
    return true;
10573 10574 10575 10576 10577 10578 10579
}


/*
 * "find-storage-pool-sources" command
 */
static const vshCmdInfo info_find_storage_pool_sources[] = {
10580 10581
    {"help", N_("discover potential storage pool sources")},
    {"desc", N_("Returns XML <sources> document.")},
10582 10583 10584 10585 10586
    {NULL, NULL}
};

static const vshCmdOptDef opts_find_storage_pool_sources[] = {
    {"type", VSH_OT_DATA, VSH_OFLAG_REQ,
10587
     N_("type of storage pool sources to discover")},
10588
    {"srcSpec", VSH_OT_DATA, VSH_OFLAG_NONE,
10589
     N_("optional file of source xml to query for pools")},
10590 10591 10592
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
10593
static bool
10594 10595
cmdPoolDiscoverSources(vshControl * ctl, const vshCmd * cmd ATTRIBUTE_UNUSED)
{
10596
    const char *type = NULL, *srcSpecFile = NULL;
10597
    char *srcSpec = NULL, *srcList;
10598

10599
    if (vshCommandOptString(cmd, "type", &type) <= 0)
E
Eric Blake 已提交
10600
        return false;
10601

10602 10603
    if (vshCommandOptString(cmd, "srcSpec", &srcSpecFile) < 0) {
        vshError(ctl, "%s", _("missing option"));
E
Eric Blake 已提交
10604
        return false;
10605
    }
10606

10607
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
10608
        return false;
10609 10610

    if (srcSpecFile && virFileReadAll(srcSpecFile, VIRSH_MAX_XML_FILE, &srcSpec) < 0)
E
Eric Blake 已提交
10611
        return false;
10612 10613

    srcList = virConnectFindStoragePoolSources(ctl->conn, type, srcSpec, 0);
10614
    VIR_FREE(srcSpec);
10615
    if (srcList == NULL) {
10616
        vshError(ctl, _("Failed to find any %s pool sources"), type);
E
Eric Blake 已提交
10617
        return false;
10618 10619
    }
    vshPrint(ctl, "%s", srcList);
10620
    VIR_FREE(srcList);
10621

E
Eric Blake 已提交
10622
    return true;
10623 10624 10625
}


10626 10627 10628
/*
 * "pool-info" command
 */
10629
static const vshCmdInfo info_pool_info[] = {
10630 10631
    {"help", N_("storage pool information")},
    {"desc", N_("Returns basic information about the storage pool.")},
10632 10633 10634
    {NULL, NULL}
};

10635
static const vshCmdOptDef opts_pool_info[] = {
10636
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name or uuid")},
10637 10638 10639
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
10640
static bool
10641
cmdPoolInfo(vshControl *ctl, const vshCmd *cmd)
10642 10643 10644
{
    virStoragePoolInfo info;
    virStoragePoolPtr pool;
10645 10646
    int autostart = 0;
    int persistent = 0;
E
Eric Blake 已提交
10647
    bool ret = true;
10648 10649
    char uuid[VIR_UUID_STRING_BUFLEN];

10650
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
10651
        return false;
10652 10653

    if (!(pool = vshCommandOptPool(ctl, cmd, "pool", NULL)))
E
Eric Blake 已提交
10654
        return false;
10655 10656 10657 10658 10659 10660 10661 10662 10663 10664 10665 10666 10667 10668 10669 10670 10671 10672 10673 10674 10675 10676 10677 10678 10679 10680

    vshPrint(ctl, "%-15s %s\n", _("Name:"), virStoragePoolGetName(pool));

    if (virStoragePoolGetUUIDString(pool, &uuid[0])==0)
        vshPrint(ctl, "%-15s %s\n", _("UUID:"), uuid);

    if (virStoragePoolGetInfo(pool, &info) == 0) {
        double val;
        const char *unit;
        switch (info.state) {
        case VIR_STORAGE_POOL_INACTIVE:
            vshPrint(ctl, "%-15s %s\n", _("State:"),
                     _("inactive"));
            break;
        case VIR_STORAGE_POOL_BUILDING:
            vshPrint(ctl, "%-15s %s\n", _("State:"),
                     _("building"));
            break;
        case VIR_STORAGE_POOL_RUNNING:
            vshPrint(ctl, "%-15s %s\n", _("State:"),
                     _("running"));
            break;
        case VIR_STORAGE_POOL_DEGRADED:
            vshPrint(ctl, "%-15s %s\n", _("State:"),
                     _("degraded"));
            break;
10681 10682 10683 10684
        case VIR_STORAGE_POOL_INACCESSIBLE:
            vshPrint(ctl, "%-15s %s\n", _("State:"),
                     _("inaccessible"));
            break;
10685 10686
        }

10687 10688
        /* Check and display whether the pool is persistent or not */
        persistent = virStoragePoolIsPersistent(pool);
10689 10690
        vshDebug(ctl, VSH_ERR_DEBUG, "Pool persistent flag value: %d\n",
                 persistent);
10691 10692 10693 10694 10695 10696 10697
        if (persistent < 0)
            vshPrint(ctl, "%-15s %s\n", _("Persistent:"),  _("unknown"));
        else
            vshPrint(ctl, "%-15s %s\n", _("Persistent:"), persistent ? _("yes") : _("no"));

        /* Check and display whether the pool is autostarted or not */
        virStoragePoolGetAutostart(pool, &autostart);
10698 10699
        vshDebug(ctl, VSH_ERR_DEBUG, "Pool autostart flag value: %d\n",
                 autostart);
10700 10701 10702 10703 10704
        if (autostart < 0)
            vshPrint(ctl, "%-15s %s\n", _("Autostart:"), _("no autostart"));
        else
            vshPrint(ctl, "%-15s %s\n", _("Autostart:"), autostart ? _("yes") : _("no"));

10705 10706 10707 10708 10709 10710 10711 10712 10713 10714 10715 10716
        if (info.state == VIR_STORAGE_POOL_RUNNING ||
            info.state == VIR_STORAGE_POOL_DEGRADED) {
            val = prettyCapacity(info.capacity, &unit);
            vshPrint(ctl, "%-15s %2.2lf %s\n", _("Capacity:"), val, unit);

            val = prettyCapacity(info.allocation, &unit);
            vshPrint(ctl, "%-15s %2.2lf %s\n", _("Allocation:"), val, unit);

            val = prettyCapacity(info.available, &unit);
            vshPrint(ctl, "%-15s %2.2lf %s\n", _("Available:"), val, unit);
        }
    } else {
E
Eric Blake 已提交
10717
        ret = false;
10718 10719 10720 10721 10722 10723 10724 10725 10726 10727
    }

    virStoragePoolFree(pool);
    return ret;
}


/*
 * "pool-name" command
 */
10728
static const vshCmdInfo info_pool_name[] = {
10729
    {"help", N_("convert a pool UUID to pool name")},
10730
    {"desc", ""},
10731 10732 10733
    {NULL, NULL}
};

10734
static const vshCmdOptDef opts_pool_name[] = {
10735
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool uuid")},
10736 10737 10738
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
10739
static bool
10740
cmdPoolName(vshControl *ctl, const vshCmd *cmd)
10741 10742 10743
{
    virStoragePoolPtr pool;

10744
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
10745
        return false;
10746
    if (!(pool = vshCommandOptPoolBy(ctl, cmd, "pool", NULL,
10747
                                           VSH_BYUUID)))
E
Eric Blake 已提交
10748
        return false;
10749 10750 10751

    vshPrint(ctl, "%s\n", virStoragePoolGetName(pool));
    virStoragePoolFree(pool);
E
Eric Blake 已提交
10752
    return true;
10753 10754 10755 10756 10757 10758
}


/*
 * "pool-start" command
 */
10759
static const vshCmdInfo info_pool_start[] = {
10760 10761
    {"help", N_("start a (previously defined) inactive pool")},
    {"desc", N_("Start a pool.")},
10762 10763 10764
    {NULL, NULL}
};

10765
static const vshCmdOptDef opts_pool_start[] = {
10766
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("name of the inactive pool")},
10767 10768 10769
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
10770
static bool
10771
cmdPoolStart(vshControl *ctl, const vshCmd *cmd)
10772 10773
{
    virStoragePoolPtr pool;
E
Eric Blake 已提交
10774
    bool ret = true;
10775

10776
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
10777
        return false;
10778

10779
    if (!(pool = vshCommandOptPoolBy(ctl, cmd, "pool", NULL, VSH_BYNAME)))
E
Eric Blake 已提交
10780
         return false;
10781 10782 10783 10784 10785

    if (virStoragePoolCreate(pool, 0) == 0) {
        vshPrint(ctl, _("Pool %s started\n"),
                 virStoragePoolGetName(pool));
    } else {
10786
        vshError(ctl, _("Failed to start pool %s"), virStoragePoolGetName(pool));
E
Eric Blake 已提交
10787
        ret = false;
10788
    }
L
Laine Stump 已提交
10789 10790

    virStoragePoolFree(pool);
10791 10792 10793 10794
    return ret;
}


10795 10796 10797
/*
 * "vol-create-as" command
 */
10798
static const vshCmdInfo info_vol_create_as[] = {
10799 10800
    {"help", N_("create a volume from a set of args")},
    {"desc", N_("Create a vol.")},
10801 10802 10803
    {NULL, NULL}
};

10804
static const vshCmdOptDef opts_vol_create_as[] = {
10805 10806 10807 10808 10809
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name")},
    {"name", VSH_OT_DATA, VSH_OFLAG_REQ, N_("name of the volume")},
    {"capacity", VSH_OT_DATA, VSH_OFLAG_REQ, N_("size of the vol with optional k,M,G,T suffix")},
    {"allocation", VSH_OT_STRING, 0, N_("initial allocation size with optional k,M,G,T suffix")},
    {"format", VSH_OT_STRING, 0, N_("file format type raw,bochs,qcow,qcow2,vmdk")},
10810 10811
    {"backing-vol", VSH_OT_STRING, 0, N_("the backing volume if taking a snapshot")},
    {"backing-vol-format", VSH_OT_STRING, 0, N_("format of backing volume if taking a snapshot")},
10812 10813 10814 10815 10816 10817 10818 10819 10820 10821
    {NULL, 0, 0, NULL}
};

static int cmdVolSize(const char *data, unsigned long long *val)
{
    char *end;
    if (virStrToLong_ull(data, &end, 10, val) < 0)
        return -1;

    if (end && *end) {
D
Daniel Veillard 已提交
10822
        /* Deliberate fallthrough cases here :-) */
10823 10824 10825
        switch (*end) {
        case 'T':
            *val *= 1024;
10826
            /* fallthrough */
10827 10828
        case 'G':
            *val *= 1024;
10829
            /* fallthrough */
10830 10831
        case 'M':
            *val *= 1024;
10832
            /* fallthrough */
10833 10834 10835 10836 10837 10838 10839 10840 10841 10842 10843 10844 10845
        case 'k':
            *val *= 1024;
            break;
        default:
            return -1;
        }
        end++;
        if (*end)
            return -1;
    }
    return 0;
}

E
Eric Blake 已提交
10846
static bool
10847
cmdVolCreateAs(vshControl *ctl, const vshCmd *cmd)
10848 10849 10850
{
    virStoragePoolPtr pool;
    virStorageVolPtr vol;
10851
    char *xml;
10852 10853
    const char *name, *capacityStr = NULL, *allocationStr = NULL, *format = NULL;
    const char *snapshotStrVol = NULL, *snapshotStrFormat = NULL;
10854
    unsigned long long capacity, allocation = 0;
10855
    virBuffer buf = VIR_BUFFER_INITIALIZER;
10856

10857
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
10858
        return false;
10859 10860 10861

    if (!(pool = vshCommandOptPoolBy(ctl, cmd, "pool", NULL,
                                     VSH_BYNAME)))
E
Eric Blake 已提交
10862
        return false;
10863

10864
    if (vshCommandOptString(cmd, "name", &name) <= 0)
10865 10866
        goto cleanup;

10867
    if (vshCommandOptString(cmd, "capacity", &capacityStr) <= 0)
10868 10869
        goto cleanup;
    if (cmdVolSize(capacityStr, &capacity) < 0)
10870
        vshError(ctl, _("Malformed size %s"), capacityStr);
10871

10872 10873
    if ((vshCommandOptString(cmd, "allocation", &allocationStr) > 0) &&
        (cmdVolSize(allocationStr, &allocation) < 0))
10874
        vshError(ctl, _("Malformed size %s"), allocationStr);
10875

10876 10877 10878 10879 10880 10881 10882
    if (vshCommandOptString(cmd, "format", &format) < 0 ||
        vshCommandOptString(cmd, "backing-vol", &snapshotStrVol) < 0 ||
        vshCommandOptString(cmd, "backing-vol-format",
                            &snapshotStrFormat) < 0) {
        vshError(ctl, "%s", _("missing argument"));
    }

10883

10884
    virBufferAddLit(&buf, "<volume>\n");
10885 10886
    virBufferAsprintf(&buf, "  <name>%s</name>\n", name);
    virBufferAsprintf(&buf, "  <capacity>%llu</capacity>\n", capacity);
10887
    if (allocationStr)
10888
        virBufferAsprintf(&buf, "  <allocation>%llu</allocation>\n", allocation);
10889 10890

    if (format) {
10891
        virBufferAddLit(&buf, "  <target>\n");
10892
        virBufferAsprintf(&buf, "    <format type='%s'/>\n",format);
10893
        virBufferAddLit(&buf, "  </target>\n");
10894
    }
10895

10896 10897 10898 10899
    /* Convert the snapshot parameters into backingStore XML */
    if (snapshotStrVol) {
        /* Lookup snapshot backing volume.  Try the backing-vol
         *  parameter as a name */
10900 10901
        vshDebug(ctl, VSH_ERR_DEBUG,
                 "%s: Look up backing store volume '%s' as name\n",
10902 10903 10904
                 cmd->def->name, snapshotStrVol);
        virStorageVolPtr snapVol = virStorageVolLookupByName(pool, snapshotStrVol);
        if (snapVol)
10905 10906
                vshDebug(ctl, VSH_ERR_DEBUG,
                         "%s: Backing store volume found using '%s' as name\n",
10907 10908 10909 10910 10911
                         cmd->def->name, snapshotStrVol);

        if (snapVol == NULL) {
            /* Snapshot backing volume not found by name.  Try the
             *  backing-vol parameter as a key */
10912 10913
            vshDebug(ctl, VSH_ERR_DEBUG,
                     "%s: Look up backing store volume '%s' as key\n",
10914 10915 10916
                     cmd->def->name, snapshotStrVol);
            snapVol = virStorageVolLookupByKey(ctl->conn, snapshotStrVol);
            if (snapVol)
10917 10918
                vshDebug(ctl, VSH_ERR_DEBUG,
                         "%s: Backing store volume found using '%s' as key\n",
10919 10920 10921 10922 10923
                         cmd->def->name, snapshotStrVol);
        }
        if (snapVol == NULL) {
            /* Snapshot backing volume not found by key.  Try the
             *  backing-vol parameter as a path */
10924 10925
            vshDebug(ctl, VSH_ERR_DEBUG,
                     "%s: Look up backing store volume '%s' as path\n",
10926 10927 10928
                     cmd->def->name, snapshotStrVol);
            snapVol = virStorageVolLookupByPath(ctl->conn, snapshotStrVol);
            if (snapVol)
10929 10930
                vshDebug(ctl, VSH_ERR_DEBUG,
                         "%s: Backing store volume found using '%s' as path\n",
10931 10932 10933 10934
                         cmd->def->name, snapshotStrVol);
        }
        if (snapVol == NULL) {
            vshError(ctl, _("failed to get vol '%s'"), snapshotStrVol);
10935
            goto cleanup;
10936 10937 10938 10939 10940
        }

        char *snapshotStrVolPath;
        if ((snapshotStrVolPath = virStorageVolGetPath(snapVol)) == NULL) {
            virStorageVolFree(snapVol);
10941
            goto cleanup;
10942 10943 10944 10945
        }

        /* Create XML for the backing store */
        virBufferAddLit(&buf, "  <backingStore>\n");
10946
        virBufferAsprintf(&buf, "    <path>%s</path>\n",snapshotStrVolPath);
10947
        if (snapshotStrFormat)
10948
            virBufferAsprintf(&buf, "    <format type='%s'/>\n",snapshotStrFormat);
10949 10950 10951 10952 10953 10954 10955 10956
        virBufferAddLit(&buf, "  </backingStore>\n");

        /* Cleanup snapshot allocations */
        VIR_FREE(snapshotStrVolPath);
        virStorageVolFree(snapVol);
    }

    virBufferAddLit(&buf, "</volume>\n");
10957

10958 10959
    if (virBufferError(&buf)) {
        vshPrint(ctl, "%s", _("Failed to allocate XML buffer"));
10960
        goto cleanup;
10961 10962 10963
    }
    xml = virBufferContentAndReset(&buf);
    vol = virStorageVolCreateXML(pool, xml, 0);
10964
    VIR_FREE(xml);
10965 10966 10967 10968 10969
    virStoragePoolFree(pool);

    if (vol != NULL) {
        vshPrint(ctl, _("Vol %s created\n"), name);
        virStorageVolFree(vol);
E
Eric Blake 已提交
10970
        return true;
10971
    } else {
10972
        vshError(ctl, _("Failed to create vol %s"), name);
E
Eric Blake 已提交
10973
        return false;
10974 10975 10976
    }

 cleanup:
10977
    virBufferFreeAndReset(&buf);
10978
    virStoragePoolFree(pool);
E
Eric Blake 已提交
10979
    return false;
10980 10981 10982
}


10983 10984 10985
/*
 * "pool-undefine" command
 */
10986
static const vshCmdInfo info_pool_undefine[] = {
10987 10988
    {"help", N_("undefine an inactive pool")},
    {"desc", N_("Undefine the configuration for an inactive pool.")},
10989 10990 10991
    {NULL, NULL}
};

10992
static const vshCmdOptDef opts_pool_undefine[] = {
10993
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name or uuid")},
10994 10995 10996
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
10997
static bool
10998
cmdPoolUndefine(vshControl *ctl, const vshCmd *cmd)
10999 11000
{
    virStoragePoolPtr pool;
E
Eric Blake 已提交
11001
    bool ret = true;
11002
    const char *name;
11003

11004
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
11005
        return false;
11006 11007

    if (!(pool = vshCommandOptPool(ctl, cmd, "pool", &name)))
E
Eric Blake 已提交
11008
        return false;
11009 11010 11011 11012

    if (virStoragePoolUndefine(pool) == 0) {
        vshPrint(ctl, _("Pool %s has been undefined\n"), name);
    } else {
11013
        vshError(ctl, _("Failed to undefine pool %s"), name);
E
Eric Blake 已提交
11014
        ret = false;
11015 11016
    }

L
Laine Stump 已提交
11017
    virStoragePoolFree(pool);
11018 11019 11020 11021 11022 11023 11024
    return ret;
}


/*
 * "pool-uuid" command
 */
11025
static const vshCmdInfo info_pool_uuid[] = {
11026
    {"help", N_("convert a pool name to pool UUID")},
11027
    {"desc", ""},
11028 11029 11030
    {NULL, NULL}
};

11031
static const vshCmdOptDef opts_pool_uuid[] = {
11032
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name")},
11033 11034 11035
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
11036
static bool
11037
cmdPoolUuid(vshControl *ctl, const vshCmd *cmd)
11038 11039 11040 11041
{
    virStoragePoolPtr pool;
    char uuid[VIR_UUID_STRING_BUFLEN];

11042
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
11043
        return false;
11044 11045

    if (!(pool = vshCommandOptPoolBy(ctl, cmd, "pool", NULL,
11046
                                           VSH_BYNAME)))
E
Eric Blake 已提交
11047
        return false;
11048 11049 11050 11051

    if (virStoragePoolGetUUIDString(pool, uuid) != -1)
        vshPrint(ctl, "%s\n", uuid);
    else
11052
        vshError(ctl, "%s", _("failed to get pool UUID"));
11053

L
Laine Stump 已提交
11054
    virStoragePoolFree(pool);
E
Eric Blake 已提交
11055
    return true;
11056 11057 11058 11059 11060 11061
}


/*
 * "vol-create" command
 */
11062
static const vshCmdInfo info_vol_create[] = {
11063 11064
    {"help", N_("create a vol from an XML file")},
    {"desc", N_("Create a vol.")},
11065 11066 11067
    {NULL, NULL}
};

11068
static const vshCmdOptDef opts_vol_create[] = {
11069 11070
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name")},
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file containing an XML vol description")},
11071 11072 11073
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
11074
static bool
11075
cmdVolCreate(vshControl *ctl, const vshCmd *cmd)
11076 11077 11078
{
    virStoragePoolPtr pool;
    virStorageVolPtr vol;
11079
    const char *from = NULL;
E
Eric Blake 已提交
11080
    bool ret = true;
11081 11082
    char *buffer;

11083
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
11084
        return false;
11085 11086

    if (!(pool = vshCommandOptPoolBy(ctl, cmd, "pool", NULL,
11087
                                           VSH_BYNAME)))
E
Eric Blake 已提交
11088
        return false;
11089

11090
    if (vshCommandOptString(cmd, "file", &from) <= 0) {
11091
        virStoragePoolFree(pool);
E
Eric Blake 已提交
11092
        return false;
11093 11094 11095
    }

    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0) {
11096
        virshReportError(ctl);
11097
        virStoragePoolFree(pool);
E
Eric Blake 已提交
11098
        return false;
11099 11100 11101
    }

    vol = virStorageVolCreateXML(pool, buffer, 0);
11102
    VIR_FREE(buffer);
11103 11104 11105 11106 11107 11108 11109
    virStoragePoolFree(pool);

    if (vol != NULL) {
        vshPrint(ctl, _("Vol %s created from %s\n"),
                 virStorageVolGetName(vol), from);
        virStorageVolFree(vol);
    } else {
11110
        vshError(ctl, _("Failed to create vol from %s"), from);
E
Eric Blake 已提交
11111
        ret = false;
11112 11113 11114 11115
    }
    return ret;
}

11116 11117 11118 11119
/*
 * "vol-create-from" command
 */
static const vshCmdInfo info_vol_create_from[] = {
11120 11121
    {"help", N_("create a vol, using another volume as input")},
    {"desc", N_("Create a vol from an existing volume.")},
11122 11123 11124 11125
    {NULL, NULL}
};

static const vshCmdOptDef opts_vol_create_from[] = {
11126 11127 11128
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name")},
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file containing an XML vol description")},
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("input vol name or key")},
E
Eric Blake 已提交
11129
    {"inputpool", VSH_OT_STRING, 0, N_("pool name or uuid of the input volume's pool")},
11130 11131 11132
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
11133
static bool
11134 11135 11136 11137
cmdVolCreateFrom(vshControl *ctl, const vshCmd *cmd)
{
    virStoragePoolPtr pool = NULL;
    virStorageVolPtr newvol = NULL, inputvol = NULL;
11138
    const char *from = NULL;
E
Eric Blake 已提交
11139
    bool ret = false;
11140 11141
    char *buffer = NULL;

11142
    if (!vshConnectionUsability(ctl, ctl->conn))
11143 11144 11145 11146 11147
        goto cleanup;

    if (!(pool = vshCommandOptPoolBy(ctl, cmd, "pool", NULL, VSH_BYNAME)))
        goto cleanup;

11148
    if (vshCommandOptString(cmd, "file", &from) <= 0) {
11149 11150 11151 11152 11153 11154 11155
        goto cleanup;
    }

    if (!(inputvol = vshCommandOptVol(ctl, cmd, "vol", "inputpool", NULL)))
        goto cleanup;

    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0) {
11156
        virshReportError(ctl);
11157 11158 11159 11160 11161 11162 11163 11164 11165
        goto cleanup;
    }

    newvol = virStorageVolCreateXMLFrom(pool, buffer, inputvol, 0);

    if (newvol != NULL) {
        vshPrint(ctl, _("Vol %s created from input vol %s\n"),
                 virStorageVolGetName(newvol), virStorageVolGetName(inputvol));
    } else {
11166
        vshError(ctl, _("Failed to create vol from %s"), from);
11167 11168 11169
        goto cleanup;
    }

E
Eric Blake 已提交
11170
    ret = true;
11171
cleanup:
11172
    VIR_FREE(buffer);
11173 11174 11175 11176
    if (pool)
        virStoragePoolFree(pool);
    if (inputvol)
        virStorageVolFree(inputvol);
L
Laine Stump 已提交
11177 11178
    if (newvol)
        virStorageVolFree(newvol);
11179 11180 11181 11182
    return ret;
}

static xmlChar *
11183 11184
makeCloneXML(const char *origxml, const char *newname)
{
11185

11186 11187 11188
    xmlDocPtr doc = NULL;
    xmlXPathContextPtr ctxt = NULL;
    xmlXPathObjectPtr obj = NULL;
11189 11190 11191
    xmlChar *newxml = NULL;
    int size;

11192
    doc = virXMLParseStringCtxt(origxml, _("(volume_definition)"), &ctxt);
11193 11194 11195 11196 11197 11198 11199 11200 11201 11202 11203 11204 11205 11206 11207 11208 11209 11210 11211 11212 11213 11214
    if (!doc)
        goto cleanup;

    obj = xmlXPathEval(BAD_CAST "/volume/name", ctxt);
    if ((obj == NULL) || (obj->nodesetval == NULL) ||
        (obj->nodesetval->nodeTab == NULL))
        goto cleanup;

    xmlNodeSetContent(obj->nodesetval->nodeTab[0], (const xmlChar *)newname);
    xmlDocDumpMemory(doc, &newxml, &size);

cleanup:
    xmlXPathFreeObject(obj);
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(doc);
    return newxml;
}

/*
 * "vol-clone" command
 */
static const vshCmdInfo info_vol_clone[] = {
11215 11216
    {"help", N_("clone a volume.")},
    {"desc", N_("Clone an existing volume.")},
11217 11218 11219 11220
    {NULL, NULL}
};

static const vshCmdOptDef opts_vol_clone[] = {
11221 11222
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("orig vol name or key")},
    {"newname", VSH_OT_DATA, VSH_OFLAG_REQ, N_("clone name")},
E
Eric Blake 已提交
11223
    {"pool", VSH_OT_STRING, 0, N_("pool name or uuid")},
11224 11225 11226
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
11227
static bool
11228 11229 11230 11231
cmdVolClone(vshControl *ctl, const vshCmd *cmd)
{
    virStoragePoolPtr origpool = NULL;
    virStorageVolPtr origvol = NULL, newvol = NULL;
11232
    const char *name = NULL;
11233
    char *origxml = NULL;
11234
    xmlChar *newxml = NULL;
E
Eric Blake 已提交
11235
    bool ret = false;
11236

11237
    if (!vshConnectionUsability(ctl, ctl->conn))
11238 11239 11240 11241 11242 11243 11244
        goto cleanup;

    if (!(origvol = vshCommandOptVol(ctl, cmd, "vol", "pool", NULL)))
        goto cleanup;

    origpool = virStoragePoolLookupByVolume(origvol);
    if (!origpool) {
11245
        vshError(ctl, "%s", _("failed to get parent pool"));
11246 11247 11248
        goto cleanup;
    }

11249
    if (vshCommandOptString(cmd, "newname", &name) <= 0)
11250 11251 11252 11253 11254 11255 11256 11257 11258 11259 11260 11261 11262 11263 11264 11265 11266 11267
        goto cleanup;

    origxml = virStorageVolGetXMLDesc(origvol, 0);
    if (!origxml)
        goto cleanup;

    newxml = makeCloneXML(origxml, name);
    if (!newxml) {
        vshPrint(ctl, "%s", _("Failed to allocate XML buffer"));
        goto cleanup;
    }

    newvol = virStorageVolCreateXMLFrom(origpool, (char *) newxml, origvol, 0);

    if (newvol != NULL) {
        vshPrint(ctl, _("Vol %s cloned from %s\n"),
                 virStorageVolGetName(newvol), virStorageVolGetName(origvol));
    } else {
11268
        vshError(ctl, _("Failed to clone vol from %s"),
11269 11270 11271 11272
                 virStorageVolGetName(origvol));
        goto cleanup;
    }

E
Eric Blake 已提交
11273
    ret = true;
11274 11275

cleanup:
11276
    VIR_FREE(origxml);
11277 11278 11279
    xmlFree(newxml);
    if (origvol)
        virStorageVolFree(origvol);
L
Laine Stump 已提交
11280 11281
    if (newvol)
        virStorageVolFree(newvol);
11282 11283 11284 11285 11286
    if (origpool)
        virStoragePoolFree(origpool);
    return ret;
}

11287 11288 11289 11290 11291 11292 11293 11294 11295 11296 11297 11298 11299

/*
 * "vol-upload" command
 */
static const vshCmdInfo info_vol_upload[] = {
    {"help", N_("upload a file into a volume")},
    {"desc", N_("Upload a file into a volume")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_vol_upload[] = {
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("vol name, key or path")},
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file")},
E
Eric Blake 已提交
11300
    {"pool", VSH_OT_STRING, 0, N_("pool name or uuid")},
11301 11302 11303 11304 11305 11306 11307 11308 11309 11310 11311 11312 11313 11314
    {"offset", VSH_OT_INT, 0, N_("volume offset to upload to") },
    {"length", VSH_OT_INT, 0, N_("amount of data to upload") },
    {NULL, 0, 0, NULL}
};

static int
cmdVolUploadSource(virStreamPtr st ATTRIBUTE_UNUSED,
                   char *bytes, size_t nbytes, void *opaque)
{
    int *fd = opaque;

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

E
Eric Blake 已提交
11315
static bool
11316 11317 11318 11319
cmdVolUpload (vshControl *ctl, const vshCmd *cmd)
{
    const char *file = NULL;
    virStorageVolPtr vol = NULL;
E
Eric Blake 已提交
11320
    bool ret = false;
11321 11322 11323 11324 11325 11326 11327 11328 11329 11330
    int fd = -1;
    virStreamPtr st = NULL;
    const char *name = NULL;
    unsigned long long offset = 0, length = 0;

    if (!vshConnectionUsability(ctl, ctl->conn))
        goto cleanup;

    if (vshCommandOptULongLong(cmd, "offset", &offset) < 0) {
        vshError(ctl, _("Unable to parse integer"));
E
Eric Blake 已提交
11331
        return false;
11332 11333 11334 11335
    }

    if (vshCommandOptULongLong(cmd, "length", &length) < 0) {
        vshError(ctl, _("Unable to parse integer"));
E
Eric Blake 已提交
11336
        return false;
11337 11338 11339
    }

    if (!(vol = vshCommandOptVol(ctl, cmd, "vol", "pool", &name))) {
E
Eric Blake 已提交
11340
        return false;
11341 11342 11343 11344 11345 11346 11347 11348 11349 11350 11351 11352 11353 11354 11355 11356 11357 11358 11359 11360 11361 11362 11363 11364 11365 11366 11367 11368 11369 11370 11371 11372 11373 11374
    }

    if (vshCommandOptString(cmd, "file", &file) < 0) {
        vshError(ctl, _("file must not be empty"));
        goto cleanup;
    }

    if ((fd = open(file, O_RDONLY)) < 0) {
        vshError(ctl, _("cannot read %s"), file);
        goto cleanup;
    }

    st = virStreamNew(ctl->conn, 0);
    if (virStorageVolUpload(vol, st, offset, length, 0) < 0) {
        vshError(ctl, _("cannot upload to volume %s"), name);
        goto cleanup;
    }

    if (virStreamSendAll(st, cmdVolUploadSource, &fd) < 0) {
        vshError(ctl, _("cannot send data to volume %s"), name);
        goto cleanup;
    }

    if (VIR_CLOSE(fd) < 0) {
        vshError(ctl, _("cannot close file %s"), file);
        virStreamAbort(st);
        goto cleanup;
    }

    if (virStreamFinish(st) < 0) {
        vshError(ctl, _("cannot close volume %s"), name);
        goto cleanup;
    }

E
Eric Blake 已提交
11375
    ret = true;
11376 11377 11378 11379 11380 11381 11382 11383 11384 11385 11386 11387 11388 11389 11390 11391 11392 11393 11394 11395 11396 11397 11398 11399

cleanup:
    if (vol)
        virStorageVolFree(vol);
    if (st)
        virStreamFree(st);
    VIR_FORCE_CLOSE(fd);
    return ret;
}



/*
 * "vol-download" command
 */
static const vshCmdInfo info_vol_download[] = {
    {"help", N_("Download a volume to a file")},
    {"desc", N_("Download a volume to a file")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_vol_download[] = {
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("vol name, key or path")},
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file")},
E
Eric Blake 已提交
11400
    {"pool", VSH_OT_STRING, 0, N_("pool name or uuid")},
11401 11402 11403 11404 11405
    {"offset", VSH_OT_INT, 0, N_("volume offset to download from") },
    {"length", VSH_OT_INT, 0, N_("amount of data to download") },
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
11406
static bool
11407 11408 11409 11410
cmdVolDownload (vshControl *ctl, const vshCmd *cmd)
{
    const char *file = NULL;
    virStorageVolPtr vol = NULL;
E
Eric Blake 已提交
11411
    bool ret = false;
11412 11413 11414 11415
    int fd = -1;
    virStreamPtr st = NULL;
    const char *name = NULL;
    unsigned long long offset = 0, length = 0;
11416
    bool created = false;
11417 11418

    if (!vshConnectionUsability(ctl, ctl->conn))
11419
        return false;
11420 11421 11422

    if (vshCommandOptULongLong(cmd, "offset", &offset) < 0) {
        vshError(ctl, _("Unable to parse integer"));
E
Eric Blake 已提交
11423
        return false;
11424 11425 11426 11427
    }

    if (vshCommandOptULongLong(cmd, "length", &length) < 0) {
        vshError(ctl, _("Unable to parse integer"));
E
Eric Blake 已提交
11428
        return false;
11429 11430 11431
    }

    if (!(vol = vshCommandOptVol(ctl, cmd, "vol", "pool", &name)))
E
Eric Blake 已提交
11432
        return false;
11433 11434 11435 11436 11437 11438 11439 11440 11441 11442 11443 11444

    if (vshCommandOptString(cmd, "file", &file) < 0) {
        vshError(ctl, _("file must not be empty"));
        goto cleanup;
    }

    if ((fd = open(file, O_WRONLY|O_CREAT|O_EXCL, 0666)) < 0) {
        if (errno != EEXIST ||
            (fd = open(file, O_WRONLY|O_TRUNC, 0666)) < 0) {
            vshError(ctl, _("cannot create %s"), file);
            goto cleanup;
        }
11445 11446
    } else {
        created = true;
11447 11448 11449 11450 11451 11452 11453 11454
    }

    st = virStreamNew(ctl->conn, 0);
    if (virStorageVolDownload(vol, st, offset, length, 0) < 0) {
        vshError(ctl, _("cannot download from volume %s"), name);
        goto cleanup;
    }

11455
    if (virStreamRecvAll(st, vshStreamSink, &fd) < 0) {
11456 11457 11458 11459 11460 11461 11462 11463 11464 11465 11466 11467 11468 11469 11470
        vshError(ctl, _("cannot receive data from volume %s"), name);
        goto cleanup;
    }

    if (VIR_CLOSE(fd) < 0) {
        vshError(ctl, _("cannot close file %s"), file);
        virStreamAbort(st);
        goto cleanup;
    }

    if (virStreamFinish(st) < 0) {
        vshError(ctl, _("cannot close volume %s"), name);
        goto cleanup;
    }

E
Eric Blake 已提交
11471
    ret = true;
11472 11473

cleanup:
11474
    VIR_FORCE_CLOSE(fd);
11475
    if (!ret && created)
11476 11477 11478 11479 11480 11481 11482 11483 11484
        unlink(file);
    if (vol)
        virStorageVolFree(vol);
    if (st)
        virStreamFree(st);
    return ret;
}


11485 11486 11487
/*
 * "vol-delete" command
 */
11488
static const vshCmdInfo info_vol_delete[] = {
11489 11490
    {"help", N_("delete a vol")},
    {"desc", N_("Delete a given vol.")},
11491 11492 11493
    {NULL, NULL}
};

11494
static const vshCmdOptDef opts_vol_delete[] = {
11495
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("vol name, key or path")},
E
Eric Blake 已提交
11496
    {"pool", VSH_OT_STRING, 0, N_("pool name or uuid")},
11497 11498 11499
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
11500
static bool
11501
cmdVolDelete(vshControl *ctl, const vshCmd *cmd)
11502 11503
{
    virStorageVolPtr vol;
E
Eric Blake 已提交
11504
    bool ret = true;
11505
    const char *name;
11506

11507
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
11508
        return false;
11509 11510

    if (!(vol = vshCommandOptVol(ctl, cmd, "vol", "pool", &name))) {
E
Eric Blake 已提交
11511
        return false;
11512 11513 11514
    }

    if (virStorageVolDelete(vol, 0) == 0) {
D
Daniel Veillard 已提交
11515
        vshPrint(ctl, _("Vol %s deleted\n"), name);
11516
    } else {
11517
        vshError(ctl, _("Failed to delete vol %s"), name);
E
Eric Blake 已提交
11518
        ret = false;
11519 11520
    }

L
Laine Stump 已提交
11521
    virStorageVolFree(vol);
11522 11523 11524 11525
    return ret;
}


D
David Allan 已提交
11526 11527 11528 11529 11530 11531 11532 11533 11534 11535 11536
/*
 * "vol-wipe" command
 */
static const vshCmdInfo info_vol_wipe[] = {
    {"help", N_("wipe a vol")},
    {"desc", N_("Ensure data previously on a volume is not accessible to future reads")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_vol_wipe[] = {
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("vol name, key or path")},
E
Eric Blake 已提交
11537
    {"pool", VSH_OT_STRING, 0, N_("pool name or uuid")},
11538
    {"algorithm", VSH_OT_STRING, 0, N_("perform selected wiping algorithm")},
D
David Allan 已提交
11539 11540 11541
    {NULL, 0, 0, NULL}
};

11542 11543 11544 11545 11546
VIR_ENUM_DECL(virStorageVolWipeAlgorithm)
VIR_ENUM_IMPL(virStorageVolWipeAlgorithm, VIR_STORAGE_VOL_WIPE_ALG_LAST,
              "zero", "nnsa", "dod", "bsi", "gutmann", "schneier",
              "pfitzner7", "pfitzner33", "random");

E
Eric Blake 已提交
11547
static bool
D
David Allan 已提交
11548 11549 11550
cmdVolWipe(vshControl *ctl, const vshCmd *cmd)
{
    virStorageVolPtr vol;
11551
    bool ret = false;
11552
    const char *name;
11553 11554 11555
    const char *algorithm_str = NULL;
    int algorithm = VIR_STORAGE_VOL_WIPE_ALG_ZERO;
    int funcRet;
D
David Allan 已提交
11556

11557
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
11558
        return false;
D
David Allan 已提交
11559 11560

    if (!(vol = vshCommandOptVol(ctl, cmd, "vol", "pool", &name))) {
E
Eric Blake 已提交
11561
        return false;
D
David Allan 已提交
11562 11563
    }

11564 11565 11566 11567 11568 11569 11570 11571 11572 11573 11574 11575 11576 11577 11578 11579 11580 11581
    if (vshCommandOptString(cmd, "algorithm", &algorithm_str) < 0) {
        vshError(ctl, "%s", _("missing argument"));
        goto out;
    }

    if (algorithm_str &&
        (algorithm = virStorageVolWipeAlgorithmTypeFromString(algorithm_str)) < 0) {
        vshError(ctl, _("Unsupported algorithm '%s'"), algorithm_str);
        goto out;
    }

    if ((funcRet = virStorageVolWipePattern(vol, algorithm, 0)) < 0) {
        if (last_error->code == VIR_ERR_NO_SUPPORT &&
            algorithm == VIR_STORAGE_VOL_WIPE_ALG_ZERO)
            funcRet = virStorageVolWipe(vol, 0);
    }

    if (funcRet < 0) {
D
David Allan 已提交
11582
        vshError(ctl, _("Failed to wipe vol %s"), name);
11583
        goto out;
D
David Allan 已提交
11584 11585
    }

11586 11587 11588
    vshPrint(ctl, _("Vol %s wiped\n"), name);
    ret = true;
out:
D
David Allan 已提交
11589 11590 11591 11592 11593
    virStorageVolFree(vol);
    return ret;
}


11594 11595 11596
/*
 * "vol-info" command
 */
11597
static const vshCmdInfo info_vol_info[] = {
11598 11599
    {"help", N_("storage vol information")},
    {"desc", N_("Returns basic information about the storage vol.")},
11600 11601 11602
    {NULL, NULL}
};

11603
static const vshCmdOptDef opts_vol_info[] = {
11604
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("vol name, key or path")},
E
Eric Blake 已提交
11605
    {"pool", VSH_OT_STRING, 0, N_("pool name or uuid")},
11606 11607 11608
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
11609
static bool
11610
cmdVolInfo(vshControl *ctl, const vshCmd *cmd)
11611 11612 11613
{
    virStorageVolInfo info;
    virStorageVolPtr vol;
E
Eric Blake 已提交
11614
    bool ret = true;
11615

11616
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
11617
        return false;
11618 11619

    if (!(vol = vshCommandOptVol(ctl, cmd, "vol", "pool", NULL)))
E
Eric Blake 已提交
11620
        return false;
11621 11622 11623 11624 11625 11626

    vshPrint(ctl, "%-15s %s\n", _("Name:"), virStorageVolGetName(vol));

    if (virStorageVolGetInfo(vol, &info) == 0) {
        double val;
        const char *unit;
R
Ryota Ozaki 已提交
11627 11628 11629 11630 11631 11632 11633 11634 11635 11636 11637 11638 11639 11640 11641 11642
        switch(info.type) {
        case VIR_STORAGE_VOL_FILE:
            vshPrint(ctl, "%-15s %s\n", _("Type:"), _("file"));
            break;

        case VIR_STORAGE_VOL_BLOCK:
            vshPrint(ctl, "%-15s %s\n", _("Type:"), _("block"));
            break;

        case VIR_STORAGE_VOL_DIR:
            vshPrint(ctl, "%-15s %s\n", _("Type:"), _("dir"));
            break;

        default:
            vshPrint(ctl, "%-15s %s\n", _("Type:"), _("unknown"));
        }
11643 11644 11645 11646 11647 11648 11649

        val = prettyCapacity(info.capacity, &unit);
        vshPrint(ctl, "%-15s %2.2lf %s\n", _("Capacity:"), val, unit);

        val = prettyCapacity(info.allocation, &unit);
        vshPrint(ctl, "%-15s %2.2lf %s\n", _("Allocation:"), val, unit);
    } else {
E
Eric Blake 已提交
11650
        ret = false;
11651 11652 11653 11654 11655 11656
    }

    virStorageVolFree(vol);
    return ret;
}

11657 11658 11659 11660 11661 11662 11663 11664 11665 11666 11667 11668 11669 11670 11671 11672 11673 11674 11675 11676 11677 11678 11679 11680 11681 11682 11683 11684 11685 11686 11687 11688 11689 11690 11691 11692 11693 11694 11695 11696 11697 11698 11699 11700 11701 11702 11703 11704 11705 11706 11707 11708 11709 11710 11711 11712 11713 11714 11715 11716 11717 11718 11719 11720 11721 11722 11723 11724 11725 11726 11727 11728 11729 11730 11731 11732 11733 11734 11735 11736 11737
/*
 * "vol-resize" command
 */
static const vshCmdInfo info_vol_resize[] = {
    {"help", N_("resize a vol")},
    {"desc", N_("Resizes a storage volume.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_vol_resize[] = {
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("vol name, key or path")},
    {"capacity", VSH_OT_DATA, VSH_OFLAG_REQ,
     N_("new capacity for the vol with optional k,M,G,T suffix")},
    {"pool", VSH_OT_STRING, 0, N_("pool name or uuid")},
    {"allocate", VSH_OT_BOOL, 0,
     N_("allocate the new capacity, rather than leaving it sparse")},
    {"delta", VSH_OT_BOOL, 0,
     N_("use capacity as a delta to current size, rather than the new size")},
    {"shrink", VSH_OT_BOOL, 0, N_("allow the resize to shrink the volume")},
    {NULL, 0, 0, NULL}
};

static bool
cmdVolResize(vshControl *ctl, const vshCmd *cmd)
{
    virStorageVolPtr vol;
    const char *capacityStr = NULL;
    unsigned long long capacity = 0;
    unsigned int flags = 0;
    bool ret = false;
    bool delta = false;

    if (vshCommandOptBool(cmd, "allocate"))
        flags |= VIR_STORAGE_VOL_RESIZE_ALLOCATE;
    if (vshCommandOptBool(cmd, "delta")) {
        delta = true;
        flags |= VIR_STORAGE_VOL_RESIZE_DELTA;
    }
    if (vshCommandOptBool(cmd, "shrink"))
        flags |= VIR_STORAGE_VOL_RESIZE_SHRINK;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(vol = vshCommandOptVol(ctl, cmd, "vol", "pool", NULL)))
        return false;

    if (vshCommandOptString(cmd, "capacity", &capacityStr) <= 0)
        goto cleanup;
    if (delta && *capacityStr == '-') {
        if (cmdVolSize(capacityStr + 1, &capacity) < 0) {
            vshError(ctl, _("Malformed size %s"), capacityStr);
            goto cleanup;
        }
        capacity = -capacity;
    } else {
        if (cmdVolSize(capacityStr, &capacity) < 0) {
            vshError(ctl, _("Malformed size %s"), capacityStr);
            goto cleanup;
        }
    }

    if (virStorageVolResize(vol, capacity, flags) == 0) {
        vshPrint(ctl,
                 delta ? _("Size of volume '%s' successfully changed by %s\n")
                 : _("Size of volume '%s' successfully changed to %s\n"),
                 virStorageVolGetName(vol), capacityStr);
        ret = true;
    } else {
        vshError(ctl,
                 delta ? _("Failed to change size of volume '%s' by %s\n")
                 : _("Failed to change size of volume '%s' to %s\n"),
                 virStorageVolGetName(vol), capacityStr);
        ret = false;
    }

cleanup:
    virStorageVolFree(vol);
    return ret;
}

11738 11739 11740 11741

/*
 * "vol-dumpxml" command
 */
11742
static const vshCmdInfo info_vol_dumpxml[] = {
11743 11744
    {"help", N_("vol information in XML")},
    {"desc", N_("Output the vol information as an XML dump to stdout.")},
11745 11746 11747
    {NULL, NULL}
};

11748
static const vshCmdOptDef opts_vol_dumpxml[] = {
11749
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("vol name, key or path")},
E
Eric Blake 已提交
11750
    {"pool", VSH_OT_STRING, 0, N_("pool name or uuid")},
11751 11752 11753
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
11754
static bool
11755
cmdVolDumpXML(vshControl *ctl, const vshCmd *cmd)
11756 11757
{
    virStorageVolPtr vol;
E
Eric Blake 已提交
11758
    bool ret = true;
11759 11760
    char *dump;

11761
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
11762
        return false;
11763 11764

    if (!(vol = vshCommandOptVol(ctl, cmd, "vol", "pool", NULL)))
E
Eric Blake 已提交
11765
        return false;
11766 11767 11768

    dump = virStorageVolGetXMLDesc(vol, 0);
    if (dump != NULL) {
11769
        vshPrint(ctl, "%s", dump);
11770
        VIR_FREE(dump);
11771
    } else {
E
Eric Blake 已提交
11772
        ret = false;
11773 11774 11775 11776 11777 11778 11779 11780 11781 11782
    }

    virStorageVolFree(vol);
    return ret;
}


/*
 * "vol-list" command
 */
11783
static const vshCmdInfo info_vol_list[] = {
11784 11785
    {"help", N_("list vols")},
    {"desc", N_("Returns list of vols by pool.")},
11786 11787 11788
    {NULL, NULL}
};

11789
static const vshCmdOptDef opts_vol_list[] = {
11790
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name or uuid")},
11791
    {"details", VSH_OT_BOOL, 0, N_("display extended details for volumes")},
11792 11793 11794
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
11795
static bool
11796
cmdVolList(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
11797
{
11798
    virStorageVolInfo volumeInfo;
11799 11800
    virStoragePoolPtr pool;
    char **activeNames = NULL;
11801 11802 11803 11804 11805
    char *outputStr = NULL;
    const char *unit;
    double val;
    int details = vshCommandOptBool(cmd, "details");
    int numVolumes = 0, i;
W
Wen Congyang 已提交
11806 11807
    int ret;
    bool functionReturn;
11808 11809 11810 11811 11812 11813 11814 11815 11816 11817 11818
    int stringLength = 0;
    size_t allocStrLength = 0, capStrLength = 0;
    size_t nameStrLength = 0, pathStrLength = 0;
    size_t typeStrLength = 0;
    struct volInfoText {
        char *allocation;
        char *capacity;
        char *path;
        char *type;
    };
    struct volInfoText *volInfoTexts = NULL;
11819

11820
    /* Check the connection to libvirtd daemon is still working */
11821
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
11822
        return false;
11823

11824
    /* Look up the pool information given to us by the user */
11825
    if (!(pool = vshCommandOptPool(ctl, cmd, "pool", NULL)))
E
Eric Blake 已提交
11826
        return false;
11827

11828 11829
    /* Determine the number of volumes in the pool */
    numVolumes = virStoragePoolNumOfVolumes(pool);
11830

11831 11832 11833
    if (numVolumes < 0) {
        vshError(ctl, "%s", _("Failed to list storage volumes"));
        virStoragePoolFree(pool);
E
Eric Blake 已提交
11834
        return false;
11835 11836
    }

11837 11838 11839 11840 11841
    /* Retrieve the list of volume names in the pool */
    if (numVolumes > 0) {
        activeNames = vshCalloc(ctl, numVolumes, sizeof(*activeNames));
        if ((numVolumes = virStoragePoolListVolumes(pool, activeNames,
                                                    numVolumes)) < 0) {
11842
            vshError(ctl, "%s", _("Failed to list active vols"));
11843
            VIR_FREE(activeNames);
11844
            virStoragePoolFree(pool);
E
Eric Blake 已提交
11845
            return false;
11846 11847
        }

11848 11849 11850 11851 11852
        /* Sort the volume names */
        qsort(&activeNames[0], numVolumes, sizeof(*activeNames), namesorter);

        /* Set aside memory for volume information pointers */
        volInfoTexts = vshCalloc(ctl, numVolumes, sizeof(*volInfoTexts));
11853 11854
    }

11855 11856 11857 11858 11859
    /* Collect the rest of the volume information for display */
    for (i = 0; i < numVolumes; i++) {
        /* Retrieve volume info */
        virStorageVolPtr vol = virStorageVolLookupByName(pool,
                                                         activeNames[i]);
11860

11861 11862 11863 11864
        /* Retrieve the volume path */
        if ((volInfoTexts[i].path = virStorageVolGetPath(vol)) == NULL) {
            /* Something went wrong retrieving a volume path, cope with it */
            volInfoTexts[i].path = vshStrdup(ctl, _("unknown"));
11865 11866
        }

11867 11868 11869 11870 11871 11872 11873 11874 11875
        /* If requested, retrieve volume type and sizing information */
        if (details) {
            if (virStorageVolGetInfo(vol, &volumeInfo) != 0) {
                /* Something went wrong retrieving volume info, cope with it */
                volInfoTexts[i].allocation = vshStrdup(ctl, _("unknown"));
                volInfoTexts[i].capacity = vshStrdup(ctl, _("unknown"));
                volInfoTexts[i].type = vshStrdup(ctl, _("unknown"));
            } else {
                /* Convert the returned volume info into output strings */
11876

11877
                /* Volume type */
11878 11879 11880 11881 11882 11883 11884 11885 11886 11887 11888 11889 11890
                switch (volumeInfo.type) {
                        case VIR_STORAGE_VOL_FILE:
                            volInfoTexts[i].type = vshStrdup(ctl, _("file"));
                            break;
                        case VIR_STORAGE_VOL_BLOCK:
                            volInfoTexts[i].type = vshStrdup(ctl, _("block"));
                            break;
                        case VIR_STORAGE_VOL_DIR:
                            volInfoTexts[i].type = vshStrdup(ctl, _("dir"));
                            break;
                        default:
                            volInfoTexts[i].type = vshStrdup(ctl, _("unknown"));
                }
11891 11892 11893 11894 11895 11896 11897 11898 11899 11900 11901 11902 11903 11904 11905 11906 11907 11908 11909 11910 11911 11912 11913 11914 11915 11916 11917 11918 11919 11920 11921 11922 11923 11924 11925 11926 11927 11928 11929 11930 11931 11932 11933 11934 11935 11936 11937 11938 11939 11940

                /* Create the capacity output string */
                val = prettyCapacity(volumeInfo.capacity, &unit);
                ret = virAsprintf(&volInfoTexts[i].capacity,
                                  "%.2lf %s", val, unit);
                if (ret < 0) {
                    /* An error occurred creating the string, return */
                    goto asprintf_failure;
                }

                /* Create the allocation output string */
                val = prettyCapacity(volumeInfo.allocation, &unit);
                ret = virAsprintf(&volInfoTexts[i].allocation,
                                  "%.2lf %s", val, unit);
                if (ret < 0) {
                    /* An error occurred creating the string, return */
                    goto asprintf_failure;
                }
            }

            /* Remember the largest length for each output string.
             * This lets us displaying header and volume information rows
             * using a single, properly sized, printf style output string.
             */

            /* Keep the length of name string if longest so far */
            stringLength = strlen(activeNames[i]);
            if (stringLength > nameStrLength)
                nameStrLength = stringLength;

            /* Keep the length of path string if longest so far */
            stringLength = strlen(volInfoTexts[i].path);
            if (stringLength > pathStrLength)
                pathStrLength = stringLength;

            /* Keep the length of type string if longest so far */
            stringLength = strlen(volInfoTexts[i].type);
            if (stringLength > typeStrLength)
                typeStrLength = stringLength;

            /* Keep the length of capacity string if longest so far */
            stringLength = strlen(volInfoTexts[i].capacity);
            if (stringLength > capStrLength)
                capStrLength = stringLength;

            /* Keep the length of allocation string if longest so far */
            stringLength = strlen(volInfoTexts[i].allocation);
            if (stringLength > allocStrLength)
                allocStrLength = stringLength;
        }
11941

11942
        /* Cleanup memory allocation */
11943
        virStorageVolFree(vol);
11944 11945 11946 11947 11948 11949 11950 11951 11952 11953 11954 11955 11956 11957 11958 11959 11960 11961
    }

    /* If the --details option wasn't selected, we output the volume
     * info using the fixed string format from previous versions to
     * maintain backward compatibility.
     */

    /* Output basic info then return if --details option not selected */
    if (!details) {
        /* The old output format */
        vshPrintExtra(ctl, "%-20s %-40s\n", _("Name"), _("Path"));
        vshPrintExtra(ctl, "-----------------------------------------\n");
        for (i = 0; i < numVolumes; i++) {
            vshPrint(ctl, "%-20s %-40s\n", activeNames[i],
                     volInfoTexts[i].path);
        }

        /* Cleanup and return */
E
Eric Blake 已提交
11962
        functionReturn = true;
11963 11964 11965 11966 11967 11968 11969 11970 11971 11972 11973 11974 11975 11976 11977 11978 11979 11980 11981 11982 11983 11984 11985 11986 11987 11988 11989 11990 11991 11992 11993
        goto cleanup;
    }

    /* We only get here if the --details option was selected. */

    /* Use the length of name header string if it's longest */
    stringLength = strlen(_("Name"));
    if (stringLength > nameStrLength)
        nameStrLength = stringLength;

    /* Use the length of path header string if it's longest */
    stringLength = strlen(_("Path"));
    if (stringLength > pathStrLength)
        pathStrLength = stringLength;

    /* Use the length of type header string if it's longest */
    stringLength = strlen(_("Type"));
    if (stringLength > typeStrLength)
        typeStrLength = stringLength;

    /* Use the length of capacity header string if it's longest */
    stringLength = strlen(_("Capacity"));
    if (stringLength > capStrLength)
        capStrLength = stringLength;

    /* Use the length of allocation header string if it's longest */
    stringLength = strlen(_("Allocation"));
    if (stringLength > allocStrLength)
        allocStrLength = stringLength;

    /* Display the string lengths for debugging */
11994 11995 11996 11997 11998 11999 12000 12001 12002 12003
    vshDebug(ctl, VSH_ERR_DEBUG,
             "Longest name string = %zu chars\n", nameStrLength);
    vshDebug(ctl, VSH_ERR_DEBUG,
             "Longest path string = %zu chars\n", pathStrLength);
    vshDebug(ctl, VSH_ERR_DEBUG,
             "Longest type string = %zu chars\n", typeStrLength);
    vshDebug(ctl, VSH_ERR_DEBUG,
             "Longest capacity string = %zu chars\n", capStrLength);
    vshDebug(ctl, VSH_ERR_DEBUG,
             "Longest allocation string = %zu chars\n", allocStrLength);
12004 12005 12006 12007 12008 12009 12010 12011 12012 12013 12014 12015 12016 12017 12018 12019 12020 12021 12022 12023 12024 12025 12026 12027 12028 12029 12030 12031 12032 12033 12034 12035 12036 12037

    /* Create the output template */
    ret = virAsprintf(&outputStr,
                      "%%-%lus  %%-%lus  %%-%lus  %%%lus  %%%lus\n",
                      (unsigned long) nameStrLength,
                      (unsigned long) pathStrLength,
                      (unsigned long) typeStrLength,
                      (unsigned long) capStrLength,
                      (unsigned long) allocStrLength);
    if (ret < 0) {
        /* An error occurred creating the string, return */
        goto asprintf_failure;
    }

    /* Display the header */
    vshPrint(ctl, outputStr, _("Name"), _("Path"), _("Type"),
             ("Capacity"), _("Allocation"));
    for (i = nameStrLength + pathStrLength + typeStrLength
                           + capStrLength + allocStrLength
                           + 8; i > 0; i--)
        vshPrintExtra(ctl, "-");
    vshPrintExtra(ctl, "\n");

    /* Display the volume info rows */
    for (i = 0; i < numVolumes; i++) {
        vshPrint(ctl, outputStr,
                 activeNames[i],
                 volInfoTexts[i].path,
                 volInfoTexts[i].type,
                 volInfoTexts[i].capacity,
                 volInfoTexts[i].allocation);
    }

    /* Cleanup and return */
E
Eric Blake 已提交
12038
    functionReturn = true;
12039 12040 12041 12042 12043 12044 12045 12046 12047 12048 12049 12050 12051 12052
    goto cleanup;

asprintf_failure:

    /* Display an appropriate error message then cleanup and return */
    switch (errno) {
    case ENOMEM:
        /* Couldn't allocate memory */
        vshError(ctl, "%s", _("Out of memory"));
        break;
    default:
        /* Some other error */
        vshError(ctl, _("virAsprintf failed (errno %d)"), errno);
    }
E
Eric Blake 已提交
12053
    functionReturn = false;
12054 12055 12056 12057 12058 12059 12060 12061 12062 12063

cleanup:

    /* Safely free the memory allocated in this function */
    for (i = 0; i < numVolumes; i++) {
        /* Cleanup the memory for one volume info structure per loop */
        VIR_FREE(volInfoTexts[i].path);
        VIR_FREE(volInfoTexts[i].type);
        VIR_FREE(volInfoTexts[i].capacity);
        VIR_FREE(volInfoTexts[i].allocation);
12064
        VIR_FREE(activeNames[i]);
12065
    }
12066 12067 12068 12069

    /* Cleanup remaining memory */
    VIR_FREE(outputStr);
    VIR_FREE(volInfoTexts);
12070
    VIR_FREE(activeNames);
12071
    virStoragePoolFree(pool);
12072 12073 12074

    /* Return the desired value */
    return functionReturn;
12075 12076 12077 12078 12079 12080
}


/*
 * "vol-name" command
 */
12081
static const vshCmdInfo info_vol_name[] = {
12082
    {"help", N_("returns the volume name for a given volume key or path")},
12083
    {"desc", ""},
12084 12085 12086
    {NULL, NULL}
};

12087
static const vshCmdOptDef opts_vol_name[] = {
12088
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("volume key or path")},
12089 12090 12091
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
12092
static bool
12093
cmdVolName(vshControl *ctl, const vshCmd *cmd)
12094 12095 12096
{
    virStorageVolPtr vol;

12097
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12098
        return false;
12099

12100
    if (!(vol = vshCommandOptVolBy(ctl, cmd, "vol", NULL, NULL,
12101
                                   VSH_BYUUID)))
E
Eric Blake 已提交
12102
        return false;
12103 12104 12105

    vshPrint(ctl, "%s\n", virStorageVolGetName(vol));
    virStorageVolFree(vol);
E
Eric Blake 已提交
12106
    return true;
12107 12108 12109
}


J
Justin Clift 已提交
12110 12111 12112 12113 12114 12115 12116 12117 12118 12119
/*
 * "vol-pool" command
 */
static const vshCmdInfo info_vol_pool[] = {
    {"help", N_("returns the storage pool for a given volume key or path")},
    {"desc", ""},
    {NULL, NULL}
};

static const vshCmdOptDef opts_vol_pool[] = {
12120
    {"uuid", VSH_OT_BOOL, 0, N_("return the pool uuid rather than pool name")},
J
Justin Clift 已提交
12121 12122 12123 12124
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("volume key or path")},
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
12125
static bool
J
Justin Clift 已提交
12126 12127 12128 12129
cmdVolPool(vshControl *ctl, const vshCmd *cmd)
{
    virStoragePoolPtr pool;
    virStorageVolPtr vol;
12130
    char uuid[VIR_UUID_STRING_BUFLEN];
J
Justin Clift 已提交
12131 12132

    /* Check the connection to libvirtd daemon is still working */
12133
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12134
        return false;
J
Justin Clift 已提交
12135 12136

    /* Use the supplied string to locate the volume */
12137
    if (!(vol = vshCommandOptVolBy(ctl, cmd, "vol", NULL, NULL,
J
Justin Clift 已提交
12138
                                   VSH_BYUUID))) {
E
Eric Blake 已提交
12139
        return false;
J
Justin Clift 已提交
12140 12141 12142 12143 12144 12145 12146
    }

    /* Look up the parent storage pool for the volume */
    pool = virStoragePoolLookupByVolume(vol);
    if (pool == NULL) {
        vshError(ctl, "%s", _("failed to get parent pool"));
        virStorageVolFree(vol);
E
Eric Blake 已提交
12147
        return false;
J
Justin Clift 已提交
12148 12149
    }

12150 12151 12152 12153 12154 12155 12156 12157 12158
    /* Return the requested details of the parent storage pool */
    if (vshCommandOptBool(cmd, "uuid")) {
        /* Retrieve and return pool UUID string */
        if (virStoragePoolGetUUIDString(pool, &uuid[0]) == 0)
            vshPrint(ctl, "%s\n", uuid);
    } else {
        /* Return the storage pool name */
        vshPrint(ctl, "%s\n", virStoragePoolGetName(pool));
    }
J
Justin Clift 已提交
12159 12160 12161 12162

    /* Cleanup */
    virStorageVolFree(vol);
    virStoragePoolFree(pool);
E
Eric Blake 已提交
12163
    return true;
J
Justin Clift 已提交
12164 12165
}

12166 12167 12168 12169

/*
 * "vol-key" command
 */
12170
static const vshCmdInfo info_vol_key[] = {
12171
    {"help", N_("returns the volume key for a given volume name or path")},
12172
    {"desc", ""},
12173 12174 12175
    {NULL, NULL}
};

12176
static const vshCmdOptDef opts_vol_key[] = {
12177
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("volume name or path")},
E
Eric Blake 已提交
12178
    {"pool", VSH_OT_STRING, 0, N_("pool name or uuid")},
12179 12180 12181
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
12182
static bool
12183
cmdVolKey(vshControl *ctl, const vshCmd *cmd)
12184 12185 12186
{
    virStorageVolPtr vol;

12187
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12188
        return false;
12189

12190
    if (!(vol = vshCommandOptVol(ctl, cmd, "vol", "pool", NULL)))
E
Eric Blake 已提交
12191
        return false;
12192 12193 12194

    vshPrint(ctl, "%s\n", virStorageVolGetKey(vol));
    virStorageVolFree(vol);
E
Eric Blake 已提交
12195
    return true;
12196 12197 12198 12199 12200 12201 12202
}



/*
 * "vol-path" command
 */
12203
static const vshCmdInfo info_vol_path[] = {
12204
    {"help", N_("returns the volume path for a given volume name or key")},
12205
    {"desc", ""},
12206 12207 12208
    {NULL, NULL}
};

12209
static const vshCmdOptDef opts_vol_path[] = {
12210
    {"vol", VSH_OT_DATA, VSH_OFLAG_REQ, N_("volume name or key")},
E
Eric Blake 已提交
12211
    {"pool", VSH_OT_STRING, 0, N_("pool name or uuid")},
12212 12213 12214
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
12215
static bool
12216
cmdVolPath(vshControl *ctl, const vshCmd *cmd)
12217 12218
{
    virStorageVolPtr vol;
12219
    char * StorageVolPath;
12220

12221
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12222
        return false;
12223

12224
    if (!(vol = vshCommandOptVol(ctl, cmd, "vol", "pool", NULL))) {
E
Eric Blake 已提交
12225
        return false;
12226
    }
12227

12228 12229 12230 12231 12232 12233 12234
    if ((StorageVolPath = virStorageVolGetPath(vol)) == NULL) {
        virStorageVolFree(vol);
        return false;
    }

    vshPrint(ctl, "%s\n", StorageVolPath);
    VIR_FREE(StorageVolPath);
12235
    virStorageVolFree(vol);
E
Eric Blake 已提交
12236
    return true;
12237 12238 12239
}


12240 12241 12242 12243
/*
 * "secret-define" command
 */
static const vshCmdInfo info_secret_define[] = {
12244 12245
    {"help", N_("define or modify a secret from an XML file")},
    {"desc", N_("Define or modify a secret.")},
12246 12247
    {NULL, NULL}
};
12248

12249
static const vshCmdOptDef opts_secret_define[] = {
12250
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file containing secret attributes in XML")},
12251 12252
    {NULL, 0, 0, NULL}
};
12253

E
Eric Blake 已提交
12254
static bool
12255 12256
cmdSecretDefine(vshControl *ctl, const vshCmd *cmd)
{
12257
    const char *from = NULL;
12258
    char *buffer;
12259
    virSecretPtr res;
12260
    char uuid[VIR_UUID_STRING_BUFLEN];
12261

12262
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12263
        return false;
12264

12265
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
12266
        return false;
12267 12268

    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0)
E
Eric Blake 已提交
12269
        return false;
12270 12271

    res = virSecretDefineXML(ctl->conn, buffer, 0);
12272
    VIR_FREE(buffer);
12273 12274

    if (res == NULL) {
12275
        vshError(ctl, _("Failed to set attributes from %s"), from);
E
Eric Blake 已提交
12276
        return false;
12277
    }
12278
    if (virSecretGetUUIDString(res, &(uuid[0])) < 0) {
12279
        vshError(ctl, "%s", _("Failed to get UUID of created secret"));
12280
        virSecretFree(res);
E
Eric Blake 已提交
12281
        return false;
12282 12283 12284
    }
    vshPrint(ctl, _("Secret %s created\n"), uuid);
    virSecretFree(res);
E
Eric Blake 已提交
12285
    return true;
12286 12287 12288 12289 12290 12291
}

/*
 * "secret-dumpxml" command
 */
static const vshCmdInfo info_secret_dumpxml[] = {
12292 12293
    {"help", N_("secret attributes in XML")},
    {"desc", N_("Output attributes of a secret as an XML dump to stdout.")},
12294 12295 12296 12297
    {NULL, NULL}
};

static const vshCmdOptDef opts_secret_dumpxml[] = {
12298
    {"secret", VSH_OT_DATA, VSH_OFLAG_REQ, N_("secret UUID")},
12299 12300 12301
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
12302
static bool
12303 12304 12305
cmdSecretDumpXML(vshControl *ctl, const vshCmd *cmd)
{
    virSecretPtr secret;
E
Eric Blake 已提交
12306
    bool ret = false;
12307 12308
    char *xml;

12309
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12310
        return false;
12311 12312 12313

    secret = vshCommandOptSecret(ctl, cmd, NULL);
    if (secret == NULL)
E
Eric Blake 已提交
12314
        return false;
12315 12316 12317 12318

    xml = virSecretGetXMLDesc(secret, 0);
    if (xml == NULL)
        goto cleanup;
12319
    vshPrint(ctl, "%s", xml);
12320
    VIR_FREE(xml);
E
Eric Blake 已提交
12321
    ret = true;
12322 12323 12324 12325 12326 12327 12328 12329 12330 12331

cleanup:
    virSecretFree(secret);
    return ret;
}

/*
 * "secret-set-value" command
 */
static const vshCmdInfo info_secret_set_value[] = {
12332 12333
    {"help", N_("set a secret value")},
    {"desc", N_("Set a secret value.")},
12334 12335 12336 12337
    {NULL, NULL}
};

static const vshCmdOptDef opts_secret_set_value[] = {
12338 12339
    {"secret", VSH_OT_DATA, VSH_OFLAG_REQ, N_("secret UUID")},
    {"base64", VSH_OT_DATA, VSH_OFLAG_REQ, N_("base64-encoded secret value")},
12340 12341 12342
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
12343
static bool
12344 12345 12346 12347
cmdSecretSetValue(vshControl *ctl, const vshCmd *cmd)
{
    virSecretPtr secret;
    size_t value_size;
12348
    const char *base64 = NULL;
12349
    char *value;
E
Eric Blake 已提交
12350 12351
    int res;
    bool ret = false;
12352

12353
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12354
        return false;
12355 12356 12357

    secret = vshCommandOptSecret(ctl, cmd, NULL);
    if (secret == NULL)
E
Eric Blake 已提交
12358
        return false;
12359

12360
    if (vshCommandOptString(cmd, "base64", &base64) <= 0)
12361 12362 12363
        goto cleanup;

    if (!base64_decode_alloc(base64, strlen(base64), &value, &value_size)) {
J
Jim Meyering 已提交
12364
        vshError(ctl, "%s", _("Invalid base64 data"));
12365 12366 12367
        goto cleanup;
    }
    if (value == NULL) {
12368
        vshError(ctl, "%s", _("Failed to allocate memory"));
E
Eric Blake 已提交
12369
        return false;
12370 12371 12372 12373
    }

    res = virSecretSetValue(secret, (unsigned char *)value, value_size, 0);
    memset(value, 0, value_size);
12374
    VIR_FREE(value);
12375 12376

    if (res != 0) {
12377
        vshError(ctl, "%s", _("Failed to set secret value"));
12378 12379 12380
        goto cleanup;
    }
    vshPrint(ctl, "%s", _("Secret value set\n"));
E
Eric Blake 已提交
12381
    ret = true;
12382 12383 12384 12385 12386 12387 12388 12389 12390 12391

cleanup:
    virSecretFree(secret);
    return ret;
}

/*
 * "secret-get-value" command
 */
static const vshCmdInfo info_secret_get_value[] = {
12392 12393
    {"help", N_("Output a secret value")},
    {"desc", N_("Output a secret value to stdout.")},
12394 12395 12396 12397
    {NULL, NULL}
};

static const vshCmdOptDef opts_secret_get_value[] = {
12398
    {"secret", VSH_OT_DATA, VSH_OFLAG_REQ, N_("secret UUID")},
12399 12400 12401
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
12402
static bool
12403 12404 12405 12406 12407 12408
cmdSecretGetValue(vshControl *ctl, const vshCmd *cmd)
{
    virSecretPtr secret;
    char *base64;
    unsigned char *value;
    size_t value_size;
E
Eric Blake 已提交
12409
    bool ret = false;
12410

12411
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12412
        return false;
12413 12414 12415

    secret = vshCommandOptSecret(ctl, cmd, NULL);
    if (secret == NULL)
E
Eric Blake 已提交
12416
        return false;
12417 12418 12419 12420 12421 12422 12423

    value = virSecretGetValue(secret, &value_size, 0);
    if (value == NULL)
        goto cleanup;

    base64_encode_alloc((char *)value, value_size, &base64);
    memset(value, 0, value_size);
12424
    VIR_FREE(value);
12425 12426

    if (base64 == NULL) {
12427
        vshError(ctl, "%s", _("Failed to allocate memory"));
12428 12429
        goto cleanup;
    }
12430
    vshPrint(ctl, "%s", base64);
12431
    memset(base64, 0, strlen(base64));
12432
    VIR_FREE(base64);
E
Eric Blake 已提交
12433
    ret = true;
12434 12435 12436 12437 12438 12439 12440 12441 12442 12443

cleanup:
    virSecretFree(secret);
    return ret;
}

/*
 * "secret-undefine" command
 */
static const vshCmdInfo info_secret_undefine[] = {
12444 12445
    {"help", N_("undefine a secret")},
    {"desc", N_("Undefine a secret.")},
12446 12447 12448 12449
    {NULL, NULL}
};

static const vshCmdOptDef opts_secret_undefine[] = {
12450
    {"secret", VSH_OT_DATA, VSH_OFLAG_REQ, N_("secret UUID")},
12451 12452 12453
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
12454
static bool
12455 12456 12457
cmdSecretUndefine(vshControl *ctl, const vshCmd *cmd)
{
    virSecretPtr secret;
E
Eric Blake 已提交
12458
    bool ret = false;
12459
    const char *uuid;
12460

12461
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12462
        return false;
12463 12464 12465

    secret = vshCommandOptSecret(ctl, cmd, &uuid);
    if (secret == NULL)
E
Eric Blake 已提交
12466
        return false;
12467 12468

    if (virSecretUndefine(secret) < 0) {
12469
        vshError(ctl, _("Failed to delete secret %s"), uuid);
12470 12471 12472
        goto cleanup;
    }
    vshPrint(ctl, _("Secret %s deleted\n"), uuid);
E
Eric Blake 已提交
12473
    ret = true;
12474 12475 12476 12477 12478 12479 12480 12481 12482 12483

cleanup:
    virSecretFree(secret);
    return ret;
}

/*
 * "secret-list" command
 */
static const vshCmdInfo info_secret_list[] = {
12484 12485
    {"help", N_("list secrets")},
    {"desc", N_("Returns a list of secrets")},
12486 12487 12488
    {NULL, NULL}
};

E
Eric Blake 已提交
12489
static bool
12490 12491 12492 12493 12494
cmdSecretList(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
{
    int maxuuids = 0, i;
    char **uuids = NULL;

12495
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12496
        return false;
12497 12498 12499

    maxuuids = virConnectNumOfSecrets(ctl->conn);
    if (maxuuids < 0) {
12500
        vshError(ctl, "%s", _("Failed to list secrets"));
E
Eric Blake 已提交
12501
        return false;
12502 12503 12504 12505 12506
    }
    uuids = vshMalloc(ctl, sizeof(*uuids) * maxuuids);

    maxuuids = virConnectListSecrets(ctl->conn, uuids, maxuuids);
    if (maxuuids < 0) {
12507
        vshError(ctl, "%s", _("Failed to list secrets"));
12508
        VIR_FREE(uuids);
E
Eric Blake 已提交
12509
        return false;
12510 12511 12512 12513
    }

    qsort(uuids, maxuuids, sizeof(char *), namesorter);

12514 12515
    vshPrintExtra(ctl, "%-36s %s\n", _("UUID"), _("Usage"));
    vshPrintExtra(ctl, "-----------------------------------------------------------\n");
12516 12517

    for (i = 0; i < maxuuids; i++) {
12518 12519 12520 12521
        virSecretPtr sec = virSecretLookupByUUIDString(ctl->conn, uuids[i]);
        const char *usageType = NULL;

        if (!sec) {
12522
            VIR_FREE(uuids[i]);
12523 12524 12525 12526 12527 12528 12529 12530 12531 12532 12533 12534 12535 12536 12537 12538 12539 12540
            continue;
        }

        switch (virSecretGetUsageType(sec)) {
        case VIR_SECRET_USAGE_TYPE_VOLUME:
            usageType = _("Volume");
            break;
        }

        if (usageType) {
            vshPrint(ctl, "%-36s %s %s\n",
                     uuids[i], usageType,
                     virSecretGetUsageID(sec));
        } else {
            vshPrint(ctl, "%-36s %s\n",
                     uuids[i], _("Unused"));
        }
        virSecretFree(sec);
12541
        VIR_FREE(uuids[i]);
12542
    }
12543
    VIR_FREE(uuids);
E
Eric Blake 已提交
12544
    return true;
12545
}
12546 12547 12548 12549 12550


/*
 * "version" command
 */
12551
static const vshCmdInfo info_version[] = {
12552 12553
    {"help", N_("show version")},
    {"desc", N_("Display the system version information.")},
12554 12555 12556
    {NULL, NULL}
};

12557 12558 12559 12560
static const vshCmdOptDef opts_version[] = {
    {"daemon", VSH_OT_BOOL, VSH_OFLAG_NONE, N_("report daemon version too")},
    {NULL, 0, 0, NULL}
};
12561

E
Eric Blake 已提交
12562
static bool
12563
cmdVersion(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
12564 12565 12566 12567 12568 12569
{
    unsigned long hvVersion;
    const char *hvType;
    unsigned long libVersion;
    unsigned long includeVersion;
    unsigned long apiVersion;
12570
    unsigned long daemonVersion;
12571 12572 12573 12574 12575
    int ret;
    unsigned int major;
    unsigned int minor;
    unsigned int rel;

12576
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12577
        return false;
12578 12579 12580

    hvType = virConnectGetType(ctl->conn);
    if (hvType == NULL) {
12581
        vshError(ctl, "%s", _("failed to get hypervisor type"));
E
Eric Blake 已提交
12582
        return false;
12583 12584 12585 12586 12587 12588 12589 12590 12591 12592 12593 12594
    }

    includeVersion = LIBVIR_VERSION_NUMBER;
    major = includeVersion / 1000000;
    includeVersion %= 1000000;
    minor = includeVersion / 1000;
    rel = includeVersion % 1000;
    vshPrint(ctl, _("Compiled against library: libvir %d.%d.%d\n"),
             major, minor, rel);

    ret = virGetVersion(&libVersion, hvType, &apiVersion);
    if (ret < 0) {
12595
        vshError(ctl, "%s", _("failed to get the library version"));
E
Eric Blake 已提交
12596
        return false;
12597 12598 12599 12600 12601 12602 12603 12604 12605 12606 12607 12608 12609 12610 12611 12612 12613
    }
    major = libVersion / 1000000;
    libVersion %= 1000000;
    minor = libVersion / 1000;
    rel = libVersion % 1000;
    vshPrint(ctl, _("Using library: libvir %d.%d.%d\n"),
             major, minor, rel);

    major = apiVersion / 1000000;
    apiVersion %= 1000000;
    minor = apiVersion / 1000;
    rel = apiVersion % 1000;
    vshPrint(ctl, _("Using API: %s %d.%d.%d\n"), hvType,
             major, minor, rel);

    ret = virConnectGetVersion(ctl->conn, &hvVersion);
    if (ret < 0) {
12614
        vshError(ctl, "%s", _("failed to get the hypervisor version"));
E
Eric Blake 已提交
12615
        return false;
12616 12617 12618 12619 12620 12621 12622 12623 12624 12625 12626 12627 12628
    }
    if (hvVersion == 0) {
        vshPrint(ctl,
                 _("Cannot extract running %s hypervisor version\n"), hvType);
    } else {
        major = hvVersion / 1000000;
        hvVersion %= 1000000;
        minor = hvVersion / 1000;
        rel = hvVersion % 1000;

        vshPrint(ctl, _("Running hypervisor: %s %d.%d.%d\n"),
                 hvType, major, minor, rel);
    }
12629 12630 12631 12632 12633 12634 12635 12636 12637 12638 12639 12640 12641 12642 12643

    if (vshCommandOptBool(cmd, "daemon")) {
        ret = virConnectGetLibVersion(ctl->conn, &daemonVersion);
        if (ret < 0) {
            vshError(ctl, "%s", _("failed to get the daemon version"));
        } else {
            major = daemonVersion / 1000000;
            daemonVersion %= 1000000;
            minor = daemonVersion / 1000;
            rel = daemonVersion % 1000;
            vshPrint(ctl, _("Running against daemon: %d.%d.%d\n"),
                     major, minor, rel);
        }
    }

E
Eric Blake 已提交
12644
    return true;
12645 12646
}

12647 12648 12649 12650
/*
 * "nodedev-list" command
 */
static const vshCmdInfo info_node_list_devices[] = {
12651
    {"help", N_("enumerate devices on this host")},
12652
    {"desc", ""},
12653 12654 12655 12656
    {NULL, NULL}
};

static const vshCmdOptDef opts_node_list_devices[] = {
12657 12658
    {"tree", VSH_OT_BOOL, 0, N_("list devices in a tree")},
    {"cap", VSH_OT_STRING, VSH_OFLAG_NONE, N_("capability name")},
12659 12660 12661
    {NULL, 0, 0, NULL}
};

12662 12663 12664 12665 12666 12667 12668 12669 12670 12671 12672 12673 12674 12675 12676 12677 12678 12679 12680 12681 12682 12683
#define MAX_DEPTH 100
#define INDENT_SIZE 4
#define INDENT_BUFLEN ((MAX_DEPTH * INDENT_SIZE) + 1)

static void
cmdNodeListDevicesPrint(vshControl *ctl,
                        char **devices,
                        char **parents,
                        int num_devices,
                        int devid,
                        int lastdev,
                        unsigned int depth,
                        unsigned int indentIdx,
                        char *indentBuf)
{
    int i;
    int nextlastdev = -1;

    /* Prepare indent for this device, but not if at root */
    if (depth && depth < MAX_DEPTH) {
        indentBuf[indentIdx] = '+';
        indentBuf[indentIdx+1] = '-';
12684 12685
        indentBuf[indentIdx+2] = ' ';
        indentBuf[indentIdx+3] = '\0';
12686 12687 12688
    }

    /* Print this device */
12689
    vshPrint(ctl, "%s", indentBuf);
12690 12691 12692 12693 12694 12695 12696 12697 12698 12699 12700 12701 12702 12703 12704 12705 12706 12707 12708 12709 12710 12711 12712
    vshPrint(ctl, "%s\n", devices[devid]);


    /* Update indent to show '|' or ' ' for child devices */
    if (depth && depth < MAX_DEPTH) {
        if (devid == lastdev)
            indentBuf[indentIdx] = ' ';
        else
            indentBuf[indentIdx] = '|';
        indentBuf[indentIdx+1] = ' ';
        indentIdx+=2;
    }

    /* Determine the index of the last child device */
    for (i = 0 ; i < num_devices ; i++) {
        if (parents[i] &&
            STREQ(parents[i], devices[devid])) {
            nextlastdev = i;
        }
    }

    /* If there is a child device, then print another blank line */
    if (nextlastdev != -1) {
12713
        vshPrint(ctl, "%s", indentBuf);
12714
        vshPrint(ctl, " |\n");
12715 12716 12717 12718 12719 12720 12721 12722 12723 12724 12725 12726 12727 12728 12729 12730 12731 12732 12733 12734 12735 12736
    }

    /* Finally print all children */
    if (depth < MAX_DEPTH)
        indentBuf[indentIdx] = ' ';
    for (i = 0 ; i < num_devices ; i++) {
        if (depth < MAX_DEPTH) {
            indentBuf[indentIdx] = ' ';
            indentBuf[indentIdx+1] = ' ';
        }
        if (parents[i] &&
            STREQ(parents[i], devices[devid]))
            cmdNodeListDevicesPrint(ctl, devices, parents,
                                    num_devices, i, nextlastdev,
                                    depth + 1, indentIdx + 2, indentBuf);
        if (depth < MAX_DEPTH)
            indentBuf[indentIdx] = '\0';
    }

    /* 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) {
12737
        vshPrint(ctl, "%s", indentBuf);
12738 12739 12740 12741
        vshPrint(ctl, "\n");
    }
}

E
Eric Blake 已提交
12742
static bool
12743 12744
cmdNodeListDevices (vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
{
12745
    const char *cap = NULL;
12746
    char **devices;
12747
    int num_devices, i;
12748
    int tree = vshCommandOptBool(cmd, "tree");
12749

12750
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12751
        return false;
12752

12753
    if (vshCommandOptString(cmd, "cap", &cap) <= 0)
12754 12755 12756 12757
        cap = NULL;

    num_devices = virNodeNumOfDevices(ctl->conn, cap, 0);
    if (num_devices < 0) {
12758
        vshError(ctl, "%s", _("Failed to count node devices"));
E
Eric Blake 已提交
12759
        return false;
12760
    } else if (num_devices == 0) {
E
Eric Blake 已提交
12761
        return true;
12762 12763 12764 12765 12766 12767
    }

    devices = vshMalloc(ctl, sizeof(char *) * num_devices);
    num_devices =
        virNodeListDevices(ctl->conn, cap, devices, num_devices, 0);
    if (num_devices < 0) {
12768
        vshError(ctl, "%s", _("Failed to list node devices"));
12769
        VIR_FREE(devices);
E
Eric Blake 已提交
12770
        return false;
12771
    }
12772
    qsort(&devices[0], num_devices, sizeof(char*), namesorter);
12773 12774 12775 12776 12777 12778 12779
    if (tree) {
        char indentBuf[INDENT_BUFLEN];
        char **parents = vshMalloc(ctl, sizeof(char *) * num_devices);
        for (i = 0; i < num_devices; i++) {
            virNodeDevicePtr dev = virNodeDeviceLookupByName(ctl->conn, devices[i]);
            if (dev && STRNEQ(devices[i], "computer")) {
                const char *parent = virNodeDeviceGetParent(dev);
E
Eric Blake 已提交
12780
                parents[i] = parent ? vshStrdup(ctl, parent) : NULL;
12781 12782 12783 12784 12785 12786 12787 12788 12789 12790 12791 12792 12793 12794 12795 12796 12797 12798 12799
            } else {
                parents[i] = NULL;
            }
            virNodeDeviceFree(dev);
        }
        for (i = 0 ; i < num_devices ; i++) {
            memset(indentBuf, '\0', sizeof indentBuf);
            if (parents[i] == NULL)
                cmdNodeListDevicesPrint(ctl,
                                        devices,
                                        parents,
                                        num_devices,
                                        i,
                                        i,
                                        0,
                                        0,
                                        indentBuf);
        }
        for (i = 0 ; i < num_devices ; i++) {
12800 12801
            VIR_FREE(devices[i]);
            VIR_FREE(parents[i]);
12802
        }
12803
        VIR_FREE(parents);
12804 12805 12806
    } else {
        for (i = 0; i < num_devices; i++) {
            vshPrint(ctl, "%s\n", devices[i]);
12807
            VIR_FREE(devices[i]);
12808
        }
12809
    }
12810
    VIR_FREE(devices);
E
Eric Blake 已提交
12811
    return true;
12812 12813 12814 12815 12816 12817
}

/*
 * "nodedev-dumpxml" command
 */
static const vshCmdInfo info_node_device_dumpxml[] = {
12818 12819
    {"help", N_("node device details in XML")},
    {"desc", N_("Output the node device details as an XML dump to stdout.")},
12820 12821 12822 12823 12824
    {NULL, NULL}
};


static const vshCmdOptDef opts_node_device_dumpxml[] = {
12825
    {"device", VSH_OT_DATA, VSH_OFLAG_REQ, N_("device key")},
12826 12827 12828
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
12829
static bool
12830 12831
cmdNodeDeviceDumpXML (vshControl *ctl, const vshCmd *cmd)
{
12832
    const char *name = NULL;
12833
    virNodeDevicePtr device;
L
Laine Stump 已提交
12834
    char *xml;
12835

12836
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12837
        return false;
12838
    if (vshCommandOptString(cmd, "device", &name) <= 0)
E
Eric Blake 已提交
12839
        return false;
12840
    if (!(device = virNodeDeviceLookupByName(ctl->conn, name))) {
12841
        vshError(ctl, "%s '%s'", _("Could not find matching device"), name);
E
Eric Blake 已提交
12842
        return false;
12843 12844
    }

L
Laine Stump 已提交
12845 12846 12847
    xml = virNodeDeviceGetXMLDesc(device, 0);
    if (!xml) {
        virNodeDeviceFree(device);
E
Eric Blake 已提交
12848
        return false;
L
Laine Stump 已提交
12849 12850 12851
    }

    vshPrint(ctl, "%s\n", xml);
12852
    VIR_FREE(xml);
12853
    virNodeDeviceFree(device);
E
Eric Blake 已提交
12854
    return true;
12855 12856
}

12857 12858 12859 12860
/*
 * "nodedev-dettach" command
 */
static const vshCmdInfo info_node_device_dettach[] = {
12861 12862
    {"help", N_("dettach node device from its device driver")},
    {"desc", N_("Dettach node device from its device driver before assigning to a domain.")},
12863 12864 12865 12866 12867
    {NULL, NULL}
};


static const vshCmdOptDef opts_node_device_dettach[] = {
12868
    {"device", VSH_OT_DATA, VSH_OFLAG_REQ, N_("device key")},
12869 12870 12871
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
12872
static bool
12873 12874
cmdNodeDeviceDettach (vshControl *ctl, const vshCmd *cmd)
{
12875
    const char *name = NULL;
12876
    virNodeDevicePtr device;
E
Eric Blake 已提交
12877
    bool ret = true;
12878

12879
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12880
        return false;
12881
    if (vshCommandOptString(cmd, "device", &name) <= 0)
E
Eric Blake 已提交
12882
        return false;
12883
    if (!(device = virNodeDeviceLookupByName(ctl->conn, name))) {
12884
        vshError(ctl, "%s '%s'", _("Could not find matching device"), name);
E
Eric Blake 已提交
12885
        return false;
12886 12887 12888 12889 12890
    }

    if (virNodeDeviceDettach(device) == 0) {
        vshPrint(ctl, _("Device %s dettached\n"), name);
    } else {
12891
        vshError(ctl, _("Failed to dettach device %s"), name);
E
Eric Blake 已提交
12892
        ret = false;
12893 12894 12895 12896 12897 12898 12899 12900 12901
    }
    virNodeDeviceFree(device);
    return ret;
}

/*
 * "nodedev-reattach" command
 */
static const vshCmdInfo info_node_device_reattach[] = {
12902 12903
    {"help", N_("reattach node device to its device driver")},
    {"desc", N_("Reattach node device to its device driver once released by the domain.")},
12904 12905 12906 12907 12908
    {NULL, NULL}
};


static const vshCmdOptDef opts_node_device_reattach[] = {
12909
    {"device", VSH_OT_DATA, VSH_OFLAG_REQ, N_("device key")},
12910 12911 12912
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
12913
static bool
12914 12915
cmdNodeDeviceReAttach (vshControl *ctl, const vshCmd *cmd)
{
12916
    const char *name = NULL;
12917
    virNodeDevicePtr device;
E
Eric Blake 已提交
12918
    bool ret = true;
12919

12920
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12921
        return false;
12922
    if (vshCommandOptString(cmd, "device", &name) <= 0)
E
Eric Blake 已提交
12923
        return false;
12924
    if (!(device = virNodeDeviceLookupByName(ctl->conn, name))) {
12925
        vshError(ctl, "%s '%s'", _("Could not find matching device"), name);
E
Eric Blake 已提交
12926
        return false;
12927 12928 12929 12930 12931
    }

    if (virNodeDeviceReAttach(device) == 0) {
        vshPrint(ctl, _("Device %s re-attached\n"), name);
    } else {
12932
        vshError(ctl, _("Failed to re-attach device %s"), name);
E
Eric Blake 已提交
12933
        ret = false;
12934 12935 12936 12937 12938 12939 12940 12941 12942
    }
    virNodeDeviceFree(device);
    return ret;
}

/*
 * "nodedev-reset" command
 */
static const vshCmdInfo info_node_device_reset[] = {
12943 12944
    {"help", N_("reset node device")},
    {"desc", N_("Reset node device before or after assigning to a domain.")},
12945 12946 12947 12948 12949
    {NULL, NULL}
};


static const vshCmdOptDef opts_node_device_reset[] = {
12950
    {"device", VSH_OT_DATA, VSH_OFLAG_REQ, N_("device key")},
12951 12952 12953
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
12954
static bool
12955 12956
cmdNodeDeviceReset (vshControl *ctl, const vshCmd *cmd)
{
12957
    const char *name = NULL;
12958
    virNodeDevicePtr device;
E
Eric Blake 已提交
12959
    bool ret = true;
12960

12961
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12962
        return false;
12963
    if (vshCommandOptString(cmd, "device", &name) <= 0)
E
Eric Blake 已提交
12964
        return false;
12965
    if (!(device = virNodeDeviceLookupByName(ctl->conn, name))) {
12966
        vshError(ctl, "%s '%s'", _("Could not find matching device"), name);
E
Eric Blake 已提交
12967
        return false;
12968 12969 12970 12971 12972
    }

    if (virNodeDeviceReset(device) == 0) {
        vshPrint(ctl, _("Device %s reset\n"), name);
    } else {
12973
        vshError(ctl, _("Failed to reset device %s"), name);
E
Eric Blake 已提交
12974
        ret = false;
12975 12976 12977 12978 12979
    }
    virNodeDeviceFree(device);
    return ret;
}

12980
/*
E
Eric Blake 已提交
12981
 * "hostname" command
12982
 */
12983
static const vshCmdInfo info_hostname[] = {
12984
    {"help", N_("print the hypervisor hostname")},
12985
    {"desc", ""},
12986 12987 12988
    {NULL, NULL}
};

E
Eric Blake 已提交
12989
static bool
12990
cmdHostname (vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
12991 12992 12993
{
    char *hostname;

12994
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
12995
        return false;
12996 12997 12998

    hostname = virConnectGetHostname (ctl->conn);
    if (hostname == NULL) {
12999
        vshError(ctl, "%s", _("failed to get hostname"));
E
Eric Blake 已提交
13000
        return false;
13001 13002 13003
    }

    vshPrint (ctl, "%s\n", hostname);
13004
    VIR_FREE(hostname);
13005

E
Eric Blake 已提交
13006
    return true;
13007 13008 13009 13010 13011
}

/*
 * "uri" command
 */
13012
static const vshCmdInfo info_uri[] = {
13013
    {"help", N_("print the hypervisor canonical URI")},
13014
    {"desc", ""},
13015 13016 13017
    {NULL, NULL}
};

E
Eric Blake 已提交
13018
static bool
13019
cmdURI (vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
13020 13021 13022
{
    char *uri;

13023
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
13024
        return false;
13025 13026 13027

    uri = virConnectGetURI (ctl->conn);
    if (uri == NULL) {
13028
        vshError(ctl, "%s", _("failed to get URI"));
E
Eric Blake 已提交
13029
        return false;
13030 13031 13032
    }

    vshPrint (ctl, "%s\n", uri);
13033
    VIR_FREE(uri);
13034

E
Eric Blake 已提交
13035
    return true;
13036 13037
}

E
Eric Blake 已提交
13038 13039 13040 13041 13042 13043 13044 13045 13046 13047
/*
 * "sysinfo" command
 */
static const vshCmdInfo info_sysinfo[] = {
    {"help", N_("print the hypervisor sysinfo")},
    {"desc",
     N_("output an XML string for the hypervisor sysinfo, if available")},
    {NULL, NULL}
};

E
Eric Blake 已提交
13048
static bool
E
Eric Blake 已提交
13049 13050 13051 13052 13053
cmdSysinfo (vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
{
    char *sysinfo;

    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
13054
        return false;
E
Eric Blake 已提交
13055 13056 13057 13058

    sysinfo = virConnectGetSysinfo (ctl->conn, 0);
    if (sysinfo == NULL) {
        vshError(ctl, "%s", _("failed to get sysinfo"));
E
Eric Blake 已提交
13059
        return false;
E
Eric Blake 已提交
13060 13061 13062 13063 13064
    }

    vshPrint (ctl, "%s", sysinfo);
    VIR_FREE(sysinfo);

E
Eric Blake 已提交
13065
    return true;
E
Eric Blake 已提交
13066 13067
}

13068 13069 13070
/*
 * "vncdisplay" command
 */
13071
static const vshCmdInfo info_vncdisplay[] = {
13072 13073
    {"help", N_("vnc display")},
    {"desc", N_("Output the IP address and port number for the VNC display.")},
13074 13075 13076
    {NULL, NULL}
};

13077
static const vshCmdOptDef opts_vncdisplay[] = {
13078
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
13079 13080 13081
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
13082
static bool
13083
cmdVNCDisplay(vshControl *ctl, const vshCmd *cmd)
13084 13085 13086 13087 13088
{
    xmlDocPtr xml = NULL;
    xmlXPathObjectPtr obj = NULL;
    xmlXPathContextPtr ctxt = NULL;
    virDomainPtr dom;
E
Eric Blake 已提交
13089
    bool ret = false;
13090 13091 13092
    int port = 0;
    char *doc;

13093
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
13094
        return false;
13095

J
Jim Meyering 已提交
13096
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
13097
        return false;
13098 13099 13100 13101 13102

    doc = virDomainGetXMLDesc(dom, 0);
    if (!doc)
        goto cleanup;

13103
    xml = virXMLParseStringCtxt(doc, _("(domain_definition)"), &ctxt);
13104
    VIR_FREE(doc);
13105 13106 13107 13108 13109 13110 13111 13112 13113 13114 13115 13116 13117 13118 13119
    if (!xml)
        goto cleanup;

    obj = xmlXPathEval(BAD_CAST "string(/domain/devices/graphics[@type='vnc']/@port)", ctxt);
    if ((obj == NULL) || (obj->type != XPATH_STRING) ||
        (obj->stringval == NULL) || (obj->stringval[0] == 0)) {
        goto cleanup;
    }
    if (virStrToLong_i((const char *)obj->stringval, NULL, 10, &port) || port < 0)
        goto cleanup;
    xmlXPathFreeObject(obj);

    obj = xmlXPathEval(BAD_CAST "string(/domain/devices/graphics[@type='vnc']/@listen)", ctxt);
    if ((obj == NULL) || (obj->type != XPATH_STRING) ||
        (obj->stringval == NULL) || (obj->stringval[0] == 0) ||
13120
        STREQ((const char*)obj->stringval, "0.0.0.0")) {
13121 13122 13123 13124 13125 13126
        vshPrint(ctl, ":%d\n", port-5900);
    } else {
        vshPrint(ctl, "%s:%d\n", (const char *)obj->stringval, port-5900);
    }
    xmlXPathFreeObject(obj);
    obj = NULL;
E
Eric Blake 已提交
13127
    ret = true;
13128 13129 13130 13131

 cleanup:
    xmlXPathFreeObject(obj);
    xmlXPathFreeContext(ctxt);
13132
    xmlFreeDoc(xml);
13133 13134 13135 13136 13137 13138 13139
    virDomainFree(dom);
    return ret;
}

/*
 * "ttyconsole" command
 */
13140
static const vshCmdInfo info_ttyconsole[] = {
13141 13142
    {"help", N_("tty console")},
    {"desc", N_("Output the device for the TTY console.")},
13143 13144 13145
    {NULL, NULL}
};

13146
static const vshCmdOptDef opts_ttyconsole[] = {
13147
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
13148 13149 13150
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
13151
static bool
13152
cmdTTYConsole(vshControl *ctl, const vshCmd *cmd)
13153 13154 13155 13156 13157
{
    xmlDocPtr xml = NULL;
    xmlXPathObjectPtr obj = NULL;
    xmlXPathContextPtr ctxt = NULL;
    virDomainPtr dom;
E
Eric Blake 已提交
13158
    bool ret = false;
13159 13160
    char *doc;

13161
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
13162
        return false;
13163

J
Jim Meyering 已提交
13164
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
13165
        return false;
13166 13167 13168 13169 13170

    doc = virDomainGetXMLDesc(dom, 0);
    if (!doc)
        goto cleanup;

13171
    xml = virXMLParseStringCtxt(doc, _("(domain_definition)"), &ctxt);
13172
    VIR_FREE(doc);
13173 13174 13175 13176 13177 13178 13179 13180 13181
    if (!xml)
        goto cleanup;

    obj = xmlXPathEval(BAD_CAST "string(/domain/devices/console/@tty)", ctxt);
    if ((obj == NULL) || (obj->type != XPATH_STRING) ||
        (obj->stringval == NULL) || (obj->stringval[0] == 0)) {
        goto cleanup;
    }
    vshPrint(ctl, "%s\n", (const char *)obj->stringval);
E
Eric Blake 已提交
13182
    ret = true;
13183 13184 13185 13186

 cleanup:
    xmlXPathFreeObject(obj);
    xmlXPathFreeContext(ctxt);
13187
    xmlFreeDoc(xml);
13188 13189 13190 13191 13192 13193
    virDomainFree(dom);
    return ret;
}

/*
 * "attach-device" command
13194
 */
13195
static const vshCmdInfo info_attach_device[] = {
13196 13197
    {"help", N_("attach device from an XML file")},
    {"desc", N_("Attach device from an XML <file>.")},
13198 13199 13200
    {NULL, NULL}
};

13201
static const vshCmdOptDef opts_attach_device[] = {
13202 13203 13204
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"file",   VSH_OT_DATA, VSH_OFLAG_REQ, N_("XML file")},
    {"persistent", VSH_OT_BOOL, 0, N_("persist device attachment")},
13205 13206 13207
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
13208
static bool
13209
cmdAttachDevice(vshControl *ctl, const vshCmd *cmd)
13210 13211
{
    virDomainPtr dom;
13212
    const char *from = NULL;
13213
    char *buffer;
W
Wen Congyang 已提交
13214
    int ret;
J
Jim Fehlig 已提交
13215
    unsigned int flags;
13216

13217
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
13218
        return false;
13219

J
Jim Meyering 已提交
13220
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
13221
        return false;
13222

13223
    if (vshCommandOptString(cmd, "file", &from) <= 0) {
13224
        virDomainFree(dom);
E
Eric Blake 已提交
13225
        return false;
13226 13227
    }

13228
    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0) {
13229
        virshReportError(ctl);
13230
        virDomainFree(dom);
E
Eric Blake 已提交
13231
        return false;
13232
    }
13233

J
Jim Fehlig 已提交
13234
    if (vshCommandOptBool(cmd, "persistent")) {
13235
        flags = VIR_DOMAIN_AFFECT_CONFIG;
J
Jim Fehlig 已提交
13236
        if (virDomainIsActive(dom) == 1)
13237
           flags |= VIR_DOMAIN_AFFECT_LIVE;
J
Jim Fehlig 已提交
13238 13239 13240 13241
        ret = virDomainAttachDeviceFlags(dom, buffer, flags);
    } else {
        ret = virDomainAttachDevice(dom, buffer);
    }
13242
    VIR_FREE(buffer);
13243 13244

    if (ret < 0) {
13245
        vshError(ctl, _("Failed to attach device from %s"), from);
13246
        virDomainFree(dom);
E
Eric Blake 已提交
13247
        return false;
13248
    } else {
J
Jim Meyering 已提交
13249
        vshPrint(ctl, "%s", _("Device attached successfully\n"));
13250 13251 13252
    }

    virDomainFree(dom);
E
Eric Blake 已提交
13253
    return true;
13254 13255
}

13256 13257 13258 13259 13260 13261 13262
/**
 * Check if n1 is superset of n2, meaning n1 contains all elements and
 * attributes as n2 at least. Including children.
 * @n1 first node
 * @n2 second node
 * returns true in case n1 covers n2, false otherwise.
 */
13263
ATTRIBUTE_UNUSED
13264 13265 13266 13267 13268 13269 13270 13271 13272
static bool
vshNodeIsSuperset(xmlNodePtr n1, xmlNodePtr n2)
{
    xmlNodePtr child1, child2;
    xmlAttrPtr attr;
    char *prop1, *prop2;
    bool found;
    bool visited;
    bool ret = false;
E
Eric Blake 已提交
13273
    long n1_child_size, n2_child_size, n1_iter;
13274 13275 13276 13277 13278 13279 13280 13281 13282 13283 13284 13285 13286 13287 13288 13289 13290 13291 13292 13293 13294 13295 13296 13297 13298 13299 13300 13301
    virBitmapPtr bitmap;

    if (!n1 && !n2)
        return true;

    if (!n1 || !n2)
        return false;

    if (!xmlStrEqual(n1->name, n2->name))
        return false;

    /* Iterate over n2 attributes and check if n1 contains them*/
    attr = n2->properties;
    while (attr) {
        if (attr->type == XML_ATTRIBUTE_NODE) {
            prop1 = virXMLPropString(n1, (const char *) attr->name);
            prop2 = virXMLPropString(n2, (const char *) attr->name);
            if (STRNEQ_NULLABLE(prop1, prop2)) {
                xmlFree(prop1);
                xmlFree(prop2);
                return false;
            }
            xmlFree(prop1);
            xmlFree(prop2);
        }
        attr = attr->next;
    }

E
Eric Blake 已提交
13302 13303 13304 13305
    n1_child_size = virXMLChildElementCount(n1);
    n2_child_size = virXMLChildElementCount(n2);
    if (n1_child_size < 0 || n2_child_size < 0 ||
        n1_child_size < n2_child_size)
13306 13307
        return false;

13308 13309 13310
    if (n1_child_size == 0 && n2_child_size == 0)
        return true;

13311 13312 13313 13314 13315 13316 13317 13318 13319 13320 13321 13322 13323 13324 13325 13326 13327 13328 13329 13330 13331 13332 13333 13334 13335 13336 13337 13338 13339 13340 13341 13342 13343 13344 13345 13346 13347 13348 13349 13350 13351 13352 13353 13354 13355 13356 13357 13358 13359 13360 13361 13362 13363 13364 13365 13366 13367 13368 13369 13370 13371 13372 13373 13374 13375 13376 13377 13378 13379 13380 13381 13382 13383 13384 13385 13386 13387 13388
    if (!(bitmap = virBitmapAlloc(n1_child_size))) {
        virReportOOMError();
        return false;
    }

    child2 = n2->children;
    while (child2) {
        if (child2->type != XML_ELEMENT_NODE) {
            child2 = child2->next;
            continue;
        }

        child1 = n1->children;
        n1_iter = 0;
        found = false;
        while (child1) {
            if (child1->type != XML_ELEMENT_NODE) {
                child1 = child1->next;
                continue;
            }

            if (virBitmapGetBit(bitmap, n1_iter, &visited) < 0) {
                vshError(NULL, "%s", _("Bad child elements counting."));
                goto cleanup;
            }

            if (visited) {
                child1 = child1->next;
                n1_iter++;
                continue;
            }

            if (xmlStrEqual(child1->name, child2->name)) {
                found = true;
                if (virBitmapSetBit(bitmap, n1_iter) < 0) {
                    vshError(NULL, "%s", _("Bad child elements counting."));
                    goto cleanup;
                }

                if (!vshNodeIsSuperset(child1, child2))
                    goto cleanup;

                break;
            }

            child1 = child1->next;
            n1_iter++;
        }

        if (!found)
            goto cleanup;

        child2 = child2->next;
    }

    ret = true;

cleanup:
    virBitmapFree(bitmap);
    return ret;
}

/**
 * vshCompleteXMLFromDomain:
 * @ctl vshControl for error messages printing
 * @dom domain
 * @oldXML device XML before
 * @newXML and after completion
 *
 * For given domain and (probably incomplete) device XML specification try to
 * find such device in domain and complete missing parts. This is however
 * possible only when given device XML is sufficiently precise so it addresses
 * only one device.
 *
 * Returns -2 when no such device exists in domain, -3 when given XML selects many
 *          (is too ambiguous), 0 in case of success. Otherwise returns -1. @newXML
 *          is touched only in case of success.
 */
13389
ATTRIBUTE_UNUSED
13390 13391 13392 13393 13394 13395 13396 13397 13398 13399 13400 13401 13402 13403 13404 13405 13406 13407 13408 13409 13410 13411 13412 13413 13414 13415 13416 13417 13418 13419 13420 13421 13422 13423 13424 13425 13426 13427 13428 13429 13430 13431 13432 13433 13434 13435 13436 13437 13438 13439 13440 13441 13442 13443 13444 13445 13446 13447 13448 13449 13450 13451 13452 13453 13454 13455 13456 13457 13458 13459 13460 13461 13462 13463 13464 13465 13466 13467 13468 13469 13470 13471 13472 13473 13474 13475 13476 13477 13478 13479 13480 13481 13482 13483 13484 13485 13486 13487 13488 13489 13490 13491 13492 13493 13494 13495 13496 13497 13498 13499 13500 13501
static int
vshCompleteXMLFromDomain(vshControl *ctl, virDomainPtr dom, char *oldXML,
                         char **newXML)
{
    int funcRet = -1;
    char *domXML = NULL;
    xmlDocPtr domDoc = NULL, devDoc = NULL;
    xmlNodePtr node = NULL;
    xmlXPathContextPtr domCtxt = NULL, devCtxt = NULL;
    xmlNodePtr *devices = NULL;
    xmlSaveCtxtPtr sctxt = NULL;
    int devices_size;
    char *xpath = NULL;
    xmlBufferPtr buf = NULL;
    int i = 0;
    int indx = -1;

    if (!(domXML = virDomainGetXMLDesc(dom, 0))) {
        vshError(ctl, _("couldn't get XML description of domain %s"),
                 virDomainGetName(dom));
        goto cleanup;
    }

    domDoc = virXMLParseStringCtxt(domXML, _("(domain_definition)"), &domCtxt);
    if (!domDoc) {
        vshError(ctl, _("Failed to parse domain definition xml"));
        goto cleanup;
    }

    devDoc = virXMLParseStringCtxt(oldXML, _("(device_definition)"), &devCtxt);
    if (!devDoc) {
        vshError(ctl, _("Failed to parse device definition xml"));
        goto cleanup;
    }

    node = xmlDocGetRootElement(devDoc);

    buf = xmlBufferCreate();
    if (!buf) {
        vshError(ctl, "%s", _("out of memory"));
        goto cleanup;
    }

    /* Get all possible devices */
    virAsprintf(&xpath, "/domain/devices/%s", node->name);
    if (!xpath) {
        virReportOOMError();
        goto cleanup;
    }
    devices_size = virXPathNodeSet(xpath, domCtxt, &devices);

    if (devices_size < 0) {
        /* error */
        vshError(ctl, "%s", _("error when selecting nodes"));
        goto cleanup;
    } else if (devices_size == 0) {
        /* no such device */
        funcRet = -2;
        goto cleanup;
    }

    /* and refine */
    for (i = 0; i < devices_size; i++) {
        if (vshNodeIsSuperset(devices[i], node)) {
            if (indx >= 0) {
                funcRet = -3; /* ambiguous */
                goto cleanup;
            }
            indx = i;
        }
    }

    if (indx < 0) {
        funcRet = -2; /* no such device */
        goto cleanup;
    }

    vshDebug(ctl, VSH_ERR_DEBUG, "Found device at pos %d\n", indx);

    if (newXML) {
        sctxt = xmlSaveToBuffer(buf, NULL, 0);
        if (!sctxt) {
            vshError(ctl, "%s", _("failed to create document saving context"));
            goto cleanup;
        }

        xmlSaveTree(sctxt, devices[indx]);
        xmlSaveClose(sctxt);
        *newXML = (char *) xmlBufferContent(buf);
        if (!*newXML) {
            virReportOOMError();
            goto cleanup;
        }
        buf->content = NULL;
    }

    vshDebug(ctl, VSH_ERR_DEBUG, "Old xml:\n%s\nNew xml:\n%s\n", oldXML,
             newXML ? NULLSTR(*newXML) : "(null)");

    funcRet = 0;

cleanup:
    xmlBufferFree(buf);
    VIR_FREE(devices);
    xmlXPathFreeContext(devCtxt);
    xmlXPathFreeContext(domCtxt);
    xmlFreeDoc(devDoc);
    xmlFreeDoc(domDoc);
    VIR_FREE(domXML);
    VIR_FREE(xpath);
    return funcRet;
}
13502 13503 13504 13505

/*
 * "detach-device" command
 */
13506
static const vshCmdInfo info_detach_device[] = {
13507 13508
    {"help", N_("detach device from an XML file")},
    {"desc", N_("Detach device from an XML <file>")},
13509 13510 13511
    {NULL, NULL}
};

13512
static const vshCmdOptDef opts_detach_device[] = {
13513 13514 13515
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"file",   VSH_OT_DATA, VSH_OFLAG_REQ, N_("XML file")},
    {"persistent", VSH_OT_BOOL, 0, N_("persist device detachment")},
13516 13517 13518
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
13519
static bool
13520
cmdDetachDevice(vshControl *ctl, const vshCmd *cmd)
13521
{
13522
    virDomainPtr dom = NULL;
13523
    const char *from = NULL;
13524
    char *buffer = NULL;
W
Wen Congyang 已提交
13525
    int ret;
13526
    bool funcRet = false;
J
Jim Fehlig 已提交
13527
    unsigned int flags;
13528

13529
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
13530
        return false;
13531

J
Jim Meyering 已提交
13532
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
13533
        return false;
13534

13535 13536
    if (vshCommandOptString(cmd, "file", &from) <= 0)
        goto cleanup;
13537

13538
    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0) {
13539
        virshReportError(ctl);
13540 13541 13542
        goto cleanup;
    }

J
Jim Fehlig 已提交
13543
    if (vshCommandOptBool(cmd, "persistent")) {
13544
        flags = VIR_DOMAIN_AFFECT_CONFIG;
J
Jim Fehlig 已提交
13545
        if (virDomainIsActive(dom) == 1)
13546
           flags |= VIR_DOMAIN_AFFECT_LIVE;
13547
        ret = virDomainDetachDeviceFlags(dom, buffer, flags);
J
Jim Fehlig 已提交
13548
    } else {
13549
        ret = virDomainDetachDevice(dom, buffer);
J
Jim Fehlig 已提交
13550
    }
13551 13552

    if (ret < 0) {
13553
        vshError(ctl, _("Failed to detach device from %s"), from);
13554
        goto cleanup;
13555 13556
    }

13557 13558 13559 13560 13561
    vshPrint(ctl, "%s", _("Device detached successfully\n"));
    funcRet = true;

cleanup:
    VIR_FREE(buffer);
13562
    virDomainFree(dom);
13563
    return funcRet;
13564 13565
}

13566

13567 13568 13569 13570 13571 13572 13573 13574 13575 13576 13577 13578 13579
/*
 * "update-device" command
 */
static const vshCmdInfo info_update_device[] = {
    {"help", N_("update device from an XML file")},
    {"desc", N_("Update device from an XML <file>.")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_update_device[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"file",   VSH_OT_DATA, VSH_OFLAG_REQ, N_("XML file")},
    {"persistent", VSH_OT_BOOL, 0, N_("persist device update")},
13580
    {"force",  VSH_OT_BOOL, 0, N_("force device update")},
13581 13582 13583
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
13584
static bool
13585 13586 13587
cmdUpdateDevice(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
13588
    const char *from = NULL;
13589
    char *buffer;
W
Wen Congyang 已提交
13590
    int ret;
13591 13592
    unsigned int flags;

13593
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
13594
        return false;
13595 13596

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
E
Eric Blake 已提交
13597
        return false;
13598

13599
    if (vshCommandOptString(cmd, "file", &from) <= 0) {
13600
        virDomainFree(dom);
E
Eric Blake 已提交
13601
        return false;
13602 13603 13604
    }

    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0) {
13605
        virshReportError(ctl);
13606
        virDomainFree(dom);
E
Eric Blake 已提交
13607
        return false;
13608 13609 13610
    }

    if (vshCommandOptBool(cmd, "persistent")) {
13611
        flags = VIR_DOMAIN_AFFECT_CONFIG;
13612
        if (virDomainIsActive(dom) == 1)
13613
           flags |= VIR_DOMAIN_AFFECT_LIVE;
13614
    } else {
13615
        flags = VIR_DOMAIN_AFFECT_LIVE;
13616
    }
13617 13618 13619 13620

    if (vshCommandOptBool(cmd, "force"))
        flags |= VIR_DOMAIN_DEVICE_MODIFY_FORCE;

13621 13622 13623 13624 13625 13626
    ret = virDomainUpdateDeviceFlags(dom, buffer, flags);
    VIR_FREE(buffer);

    if (ret < 0) {
        vshError(ctl, _("Failed to update device from %s"), from);
        virDomainFree(dom);
E
Eric Blake 已提交
13627
        return false;
13628 13629 13630 13631 13632
    } else {
        vshPrint(ctl, "%s", _("Device updated successfully\n"));
    }

    virDomainFree(dom);
E
Eric Blake 已提交
13633
    return true;
13634 13635 13636
}


13637 13638 13639
/*
 * "attach-interface" command
 */
13640
static const vshCmdInfo info_attach_interface[] = {
13641 13642
    {"help", N_("attach network interface")},
    {"desc", N_("Attach new network interface.")},
13643 13644 13645
    {NULL, NULL}
};

13646
static const vshCmdOptDef opts_attach_interface[] = {
13647 13648 13649 13650 13651 13652
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"type",   VSH_OT_DATA, VSH_OFLAG_REQ, N_("network interface type")},
    {"source", VSH_OT_DATA, VSH_OFLAG_REQ, N_("source of network interface")},
    {"target", VSH_OT_DATA, 0, N_("target network name")},
    {"mac",    VSH_OT_DATA, 0, N_("MAC address")},
    {"script", VSH_OT_DATA, 0, N_("script used to bridge network interface")},
13653
    {"model", VSH_OT_DATA, 0, N_("model type")},
13654
    {"persistent", VSH_OT_BOOL, 0, N_("persist interface attachment")},
13655 13656
    {"inbound", VSH_OT_DATA, VSH_OFLAG_NONE, N_("control domain's incoming traffics")},
    {"outbound", VSH_OT_DATA, VSH_OFLAG_NONE, N_("control domain's outgoing traffics")},
13657 13658 13659
    {NULL, 0, 0, NULL}
};

13660 13661 13662
/* parse inbound and outbound which are in the format of
 * 'average,peak,burst', in which peak and burst are optional,
 * thus 'average,,burst' and 'average,peak' are also legal. */
13663
static int parseRateStr(const char *rateStr, virNetDevBandwidthRatePtr rate)
13664 13665 13666 13667 13668 13669 13670 13671 13672 13673 13674 13675 13676 13677 13678 13679 13680 13681 13682 13683 13684 13685 13686 13687 13688 13689 13690 13691 13692 13693 13694
{
    const char *average = NULL;
    char *peak = NULL, *burst = NULL;

    average = rateStr;
    if (!average)
        return -1;
    if (virStrToLong_ull(average, &peak, 10, &rate->average) < 0)
        return -1;

    /* peak will be updated to point to the end of rateStr in case
     * of 'average' */
    if (peak && *peak != '\0') {
        burst = strchr(peak + 1, ',');
        if (!(burst && (burst - peak == 1))) {
            if (virStrToLong_ull(peak + 1, &burst, 10, &rate->peak) < 0)
                return -1;
        }

        /* burst will be updated to point to the end of rateStr in case
         * of 'average,peak' */
        if (burst && *burst != '\0') {
            if (virStrToLong_ull(burst + 1, NULL, 10, &rate->burst) < 0)
                return -1;
        }
    }


    return 0;
}

E
Eric Blake 已提交
13695
static bool
13696
cmdAttachInterface(vshControl *ctl, const vshCmd *cmd)
13697 13698
{
    virDomainPtr dom = NULL;
13699
    const char *mac = NULL, *target = NULL, *script = NULL,
13700 13701
                *type = NULL, *source = NULL, *model = NULL,
                *inboundStr = NULL, *outboundStr = NULL;
13702
    virNetDevBandwidthRate inbound, outbound;
E
Eric Blake 已提交
13703
    int typ;
W
Wen Congyang 已提交
13704 13705
    int ret;
    bool functionReturn = false;
J
Jim Fehlig 已提交
13706
    unsigned int flags;
13707 13708
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    char *xml;
13709

13710
    if (!vshConnectionUsability(ctl, ctl->conn))
13711 13712
        goto cleanup;

J
Jim Meyering 已提交
13713
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
13714 13715
        goto cleanup;

13716
    if (vshCommandOptString(cmd, "type", &type) <= 0)
13717 13718
        goto cleanup;

13719 13720 13721 13722
    if (vshCommandOptString(cmd, "source", &source) < 0 ||
        vshCommandOptString(cmd, "target", &target) < 0 ||
        vshCommandOptString(cmd, "mac", &mac) < 0 ||
        vshCommandOptString(cmd, "script", &script) < 0 ||
13723 13724 13725
        vshCommandOptString(cmd, "model", &model) < 0 ||
        vshCommandOptString(cmd, "inbound", &inboundStr) < 0 ||
        vshCommandOptString(cmd, "outbound", &outboundStr) < 0) {
13726
        vshError(ctl, "missing argument");
13727
        goto cleanup;
13728
    }
13729 13730

    /* check interface type */
13731
    if (STREQ(type, "network")) {
13732
        typ = 1;
13733
    } else if (STREQ(type, "bridge")) {
13734 13735
        typ = 2;
    } else {
E
Eric Blake 已提交
13736 13737
        vshError(ctl, _("No support for %s in command 'attach-interface'"),
                 type);
13738 13739 13740
        goto cleanup;
    }

13741 13742 13743 13744 13745 13746 13747 13748 13749 13750 13751 13752 13753 13754 13755 13756 13757 13758 13759 13760 13761 13762 13763
    if (inboundStr) {
        memset(&inbound, 0, sizeof(inbound));
        if (parseRateStr(inboundStr, &inbound) < 0) {
            vshError(ctl, _("inbound format is incorrect"));
            goto cleanup;
        }
        if (inbound.average == 0) {
            vshError(ctl, _("inbound average is mandatory"));
            goto cleanup;
        }
    }
    if (outboundStr) {
        memset(&outbound, 0, sizeof(outbound));
        if (parseRateStr(outboundStr, &outbound) < 0) {
            vshError(ctl, _("outbound format is incorrect"));
            goto cleanup;
        }
        if (outbound.average == 0) {
            vshError(ctl, _("outbound average is mandatory"));
            goto cleanup;
        }
    }

13764
    /* Make XML of interface */
13765
    virBufferAsprintf(&buf, "<interface type='%s'>\n", type);
13766

13767
    if (typ == 1)
13768
        virBufferAsprintf(&buf, "  <source network='%s'/>\n", source);
13769
    else if (typ == 2)
13770
        virBufferAsprintf(&buf, "  <source bridge='%s'/>\n", source);
13771

13772
    if (target != NULL)
13773
        virBufferAsprintf(&buf, "  <target dev='%s'/>\n", target);
13774
    if (mac != NULL)
13775
        virBufferAsprintf(&buf, "  <mac address='%s'/>\n", mac);
13776
    if (script != NULL)
13777
        virBufferAsprintf(&buf, "  <script path='%s'/>\n", script);
13778
    if (model != NULL)
13779
        virBufferAsprintf(&buf, "  <model type='%s'/>\n", model);
13780

13781 13782 13783 13784 13785 13786 13787 13788 13789 13790 13791 13792 13793 13794 13795 13796 13797 13798 13799 13800 13801
    if (inboundStr || outboundStr) {
        virBufferAsprintf(&buf, "  <bandwidth>\n");
        if (inboundStr && inbound.average > 0) {
            virBufferAsprintf(&buf, "    <inbound average='%llu'", inbound.average);
            if (inbound.peak > 0)
                virBufferAsprintf(&buf, " peak='%llu'", inbound.peak);
            if (inbound.burst > 0)
                virBufferAsprintf(&buf, " burst='%llu'", inbound.burst);
            virBufferAsprintf(&buf, "/>\n");
        }
        if (outboundStr && outbound.average > 0) {
            virBufferAsprintf(&buf, "    <outbound average='%llu'", outbound.average);
            if (outbound.peak > 0)
                virBufferAsprintf(&buf, " peak='%llu'", outbound.peak);
            if (outbound.burst > 0)
                virBufferAsprintf(&buf, " burst='%llu'", outbound.burst);
            virBufferAsprintf(&buf, "/>\n");
        }
        virBufferAsprintf(&buf, "  </bandwidth>\n");
    }

13802
    virBufferAddLit(&buf, "</interface>\n");
13803

13804 13805
    if (virBufferError(&buf)) {
        vshPrint(ctl, "%s", _("Failed to allocate XML buffer"));
W
Wen Congyang 已提交
13806
        goto cleanup;
13807 13808
    }

13809
    xml = virBufferContentAndReset(&buf);
13810

J
Jim Fehlig 已提交
13811
    if (vshCommandOptBool(cmd, "persistent")) {
13812
        flags = VIR_DOMAIN_AFFECT_CONFIG;
J
Jim Fehlig 已提交
13813
        if (virDomainIsActive(dom) == 1)
13814
            flags |= VIR_DOMAIN_AFFECT_LIVE;
13815
        ret = virDomainAttachDeviceFlags(dom, xml, flags);
13816
    } else {
13817
        ret = virDomainAttachDevice(dom, xml);
13818
    }
13819

13820 13821
    VIR_FREE(xml);

J
Jim Fehlig 已提交
13822
    if (ret != 0) {
L
Laine Stump 已提交
13823
        vshError(ctl, "%s", _("Failed to attach interface"));
J
Jim Fehlig 已提交
13824 13825
    } else {
        vshPrint(ctl, "%s", _("Interface attached successfully\n"));
W
Wen Congyang 已提交
13826
        functionReturn = true;
J
Jim Fehlig 已提交
13827
    }
13828 13829 13830 13831

 cleanup:
    if (dom)
        virDomainFree(dom);
13832
    virBufferFreeAndReset(&buf);
W
Wen Congyang 已提交
13833
    return functionReturn;
13834 13835 13836 13837 13838
}

/*
 * "detach-interface" command
 */
13839
static const vshCmdInfo info_detach_interface[] = {
13840 13841
    {"help", N_("detach network interface")},
    {"desc", N_("Detach network interface.")},
13842 13843 13844
    {NULL, NULL}
};

13845
static const vshCmdOptDef opts_detach_interface[] = {
13846 13847 13848 13849
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"type",   VSH_OT_DATA, VSH_OFLAG_REQ, N_("network interface type")},
    {"mac",    VSH_OT_STRING, 0, N_("MAC address")},
    {"persistent", VSH_OT_BOOL, 0, N_("persist interface detachment")},
13850 13851 13852
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
13853
static bool
13854
cmdDetachInterface(vshControl *ctl, const vshCmd *cmd)
13855 13856 13857 13858 13859 13860 13861
{
    virDomainPtr dom = NULL;
    xmlDocPtr xml = NULL;
    xmlXPathObjectPtr obj=NULL;
    xmlXPathContextPtr ctxt = NULL;
    xmlNodePtr cur = NULL;
    xmlBufferPtr xml_buf = NULL;
13862
    const char *mac =NULL, *type = NULL;
13863
    char *doc;
13864
    char buf[64];
E
Eric Blake 已提交
13865
    int i = 0, diff_mac;
W
Wen Congyang 已提交
13866 13867
    int ret;
    int functionReturn = false;
J
Jim Fehlig 已提交
13868
    unsigned int flags;
13869

13870
    if (!vshConnectionUsability(ctl, ctl->conn))
13871 13872
        goto cleanup;

J
Jim Meyering 已提交
13873
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
13874 13875
        goto cleanup;

13876
    if (vshCommandOptString(cmd, "type", &type) <= 0)
13877 13878
        goto cleanup;

13879 13880
    if (vshCommandOptString(cmd, "mac", &mac) < 0) {
        vshError(ctl, "%s", _("missing option"));
13881
        goto cleanup;
13882
    }
13883 13884 13885 13886 13887

    doc = virDomainGetXMLDesc(dom, 0);
    if (!doc)
        goto cleanup;

13888
    xml = virXMLParseStringCtxt(doc, _("(domain_definition)"), &ctxt);
13889
    VIR_FREE(doc);
13890
    if (!xml) {
13891
        vshError(ctl, "%s", _("Failed to get interface information"));
13892 13893 13894
        goto cleanup;
    }

E
Eric Blake 已提交
13895
    snprintf(buf, sizeof(buf), "/domain/devices/interface[@type='%s']", type);
13896 13897 13898
    obj = xmlXPathEval(BAD_CAST buf, ctxt);
    if ((obj == NULL) || (obj->type != XPATH_NODESET) ||
        (obj->nodesetval == NULL) || (obj->nodesetval->nodeNr == 0)) {
13899
        vshError(ctl, _("No found interface whose type is %s"), type);
13900 13901 13902
        goto cleanup;
    }

13903 13904 13905 13906 13907 13908
    if ((!mac) && (obj->nodesetval->nodeNr > 1)) {
        vshError(ctl, _("Domain has %d interfaces. Please specify which one "
                        "to detach using --mac"), obj->nodesetval->nodeNr);
        goto cleanup;
    }

13909 13910 13911 13912 13913 13914 13915
    if (!mac)
        goto hit;

    /* search mac */
    for (; i < obj->nodesetval->nodeNr; i++) {
        cur = obj->nodesetval->nodeTab[i]->children;
        while (cur != NULL) {
13916 13917 13918
            if (cur->type == XML_ELEMENT_NODE &&
                xmlStrEqual(cur->name, BAD_CAST "mac")) {
                char *tmp_mac = virXMLPropString(cur, "address");
13919
                diff_mac = virMacAddrCompare(tmp_mac, mac);
13920
                VIR_FREE(tmp_mac);
13921 13922 13923 13924 13925 13926 13927
                if (!diff_mac) {
                    goto hit;
                }
            }
            cur = cur->next;
        }
    }
13928
    vshError(ctl, _("No found interface whose MAC address is %s"), mac);
13929 13930 13931 13932 13933
    goto cleanup;

 hit:
    xml_buf = xmlBufferCreate();
    if (!xml_buf) {
13934
        vshError(ctl, "%s", _("Failed to allocate memory"));
13935 13936 13937
        goto cleanup;
    }

13938
    if (xmlNodeDump(xml_buf, xml, obj->nodesetval->nodeTab[i], 0, 0) < 0) {
13939
        vshError(ctl, "%s", _("Failed to create XML"));
13940 13941 13942
        goto cleanup;
    }

J
Jim Fehlig 已提交
13943
    if (vshCommandOptBool(cmd, "persistent")) {
13944
        flags = VIR_DOMAIN_AFFECT_CONFIG;
J
Jim Fehlig 已提交
13945
        if (virDomainIsActive(dom) == 1)
13946
            flags |= VIR_DOMAIN_AFFECT_LIVE;
J
Jim Fehlig 已提交
13947 13948 13949 13950 13951 13952 13953 13954
        ret = virDomainDetachDeviceFlags(dom,
                                         (char *)xmlBufferContent(xml_buf),
                                         flags);
    } else {
        ret = virDomainDetachDevice(dom, (char *)xmlBufferContent(xml_buf));
    }

    if (ret != 0) {
L
Laine Stump 已提交
13955
        vshError(ctl, "%s", _("Failed to detach interface"));
J
Jim Fehlig 已提交
13956
    } else {
J
Jim Meyering 已提交
13957
        vshPrint(ctl, "%s", _("Interface detached successfully\n"));
W
Wen Congyang 已提交
13958
        functionReturn = true;
13959
    }
13960 13961 13962 13963

 cleanup:
    if (dom)
        virDomainFree(dom);
13964
    xmlXPathFreeObject(obj);
13965
    xmlXPathFreeContext(ctxt);
13966 13967
    xmlFreeDoc(xml);
    xmlBufferFree(xml_buf);
W
Wen Congyang 已提交
13968
    return functionReturn;
13969 13970 13971 13972 13973
}

/*
 * "attach-disk" command
 */
13974
static const vshCmdInfo info_attach_disk[] = {
13975 13976
    {"help", N_("attach disk device")},
    {"desc", N_("Attach new disk device.")},
13977 13978 13979
    {NULL, NULL}
};

13980
static const vshCmdOptDef opts_attach_disk[] = {
13981
    {"domain",  VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
13982 13983
    {"source",  VSH_OT_DATA, VSH_OFLAG_REQ | VSH_OFLAG_EMPTY_OK,
     N_("source of disk device")},
13984 13985 13986
    {"target",  VSH_OT_DATA, VSH_OFLAG_REQ, N_("target of disk device")},
    {"driver",    VSH_OT_STRING, 0, N_("driver of disk device")},
    {"subdriver", VSH_OT_STRING, 0, N_("subdriver of disk device")},
13987
    {"cache",     VSH_OT_STRING, 0, N_("cache mode of disk device")},
13988 13989 13990
    {"type",    VSH_OT_STRING, 0, N_("target device type")},
    {"mode",    VSH_OT_STRING, 0, N_("mode of device reading and writing")},
    {"persistent", VSH_OT_BOOL, 0, N_("persist disk attachment")},
13991
    {"sourcetype", VSH_OT_STRING, 0, N_("type of source (block|file)")},
13992 13993
    {"serial", VSH_OT_STRING, 0, N_("serial of disk device")},
    {"shareable", VSH_OT_BOOL, 0, N_("shareable between domains")},
13994
    {"rawio", VSH_OT_BOOL, 0, N_("needs rawio capability")},
13995
    {"address", VSH_OT_STRING, 0, N_("address of disk device")},
13996 13997
    {"multifunction", VSH_OT_BOOL, 0,
     N_("use multifunction pci under specified address")},
13998 13999 14000
    {NULL, 0, 0, NULL}
};

14001 14002 14003 14004 14005 14006 14007 14008 14009 14010 14011 14012 14013 14014 14015 14016 14017 14018 14019 14020 14021 14022 14023 14024 14025 14026 14027 14028 14029 14030 14031 14032 14033 14034 14035 14036 14037 14038 14039 14040 14041 14042 14043 14044 14045 14046 14047 14048 14049 14050 14051 14052 14053 14054 14055 14056 14057 14058 14059 14060 14061 14062 14063 14064 14065 14066 14067 14068 14069 14070 14071 14072 14073 14074 14075 14076 14077 14078 14079 14080 14081 14082 14083 14084 14085 14086 14087 14088 14089 14090 14091 14092 14093 14094 14095 14096 14097 14098 14099 14100 14101 14102 14103 14104 14105 14106 14107 14108 14109 14110 14111 14112 14113 14114 14115 14116 14117 14118 14119 14120 14121 14122 14123 14124 14125 14126 14127 14128 14129 14130 14131 14132 14133 14134 14135 14136 14137 14138 14139 14140 14141 14142 14143 14144 14145 14146 14147
enum {
    DISK_ADDR_TYPE_INVALID,
    DISK_ADDR_TYPE_PCI,
    DISK_ADDR_TYPE_SCSI,
    DISK_ADDR_TYPE_IDE,
};

struct PCIAddress {
    unsigned int domain;
    unsigned int bus;
    unsigned int slot;
    unsigned int function;
};

struct SCSIAddress {
    unsigned int controller;
    unsigned int bus;
    unsigned int unit;
};

struct IDEAddress {
    unsigned int controller;
    unsigned int bus;
    unsigned int unit;
};

struct DiskAddress {
    int type;
    union {
        struct PCIAddress pci;
        struct SCSIAddress scsi;
        struct IDEAddress ide;
    } addr;
};

static int str2PCIAddress(const char *str, struct PCIAddress *pciAddr)
{
    char *domain, *bus, *slot, *function;

    if (!pciAddr)
        return -1;
    if (!str)
        return -1;

    domain = (char *)str;

    if (virStrToLong_ui(domain, &bus, 0, &pciAddr->domain) != 0)
        return -1;

    bus++;
    if (virStrToLong_ui(bus, &slot, 0, &pciAddr->bus) != 0)
        return -1;

    slot++;
    if (virStrToLong_ui(slot, &function, 0, &pciAddr->slot) != 0)
        return -1;

    function++;
    if (virStrToLong_ui(function, NULL, 0, &pciAddr->function) != 0)
        return -1;

    return 0;
}

static int str2SCSIAddress(const char *str, struct SCSIAddress *scsiAddr)
{
    char *controller, *bus, *unit;

    if (!scsiAddr)
        return -1;
    if (!str)
        return -1;

    controller = (char *)str;

    if (virStrToLong_ui(controller, &bus, 0, &scsiAddr->controller) != 0)
        return -1;

    bus++;
    if (virStrToLong_ui(bus, &unit, 0, &scsiAddr->bus) != 0)
        return -1;

    unit++;
    if (virStrToLong_ui(unit, NULL, 0, &scsiAddr->unit) != 0)
        return -1;

    return 0;
}

static int str2IDEAddress(const char *str, struct IDEAddress *ideAddr)
{
    char *controller, *bus, *unit;

    if (!ideAddr)
        return -1;
    if (!str)
        return -1;

    controller = (char *)str;

    if (virStrToLong_ui(controller, &bus, 0, &ideAddr->controller) != 0)
        return -1;

    bus++;
    if (virStrToLong_ui(bus, &unit, 0, &ideAddr->bus) != 0)
        return -1;

    unit++;
    if (virStrToLong_ui(unit, NULL, 0, &ideAddr->unit) != 0)
        return -1;

    return 0;
}

/* pci address pci:0000.00.0x0a.0 (domain:bus:slot:function)
 * ide disk address: ide:00.00.0 (controller:bus:unit)
 * scsi disk address: scsi:00.00.0 (controller:bus:unit)
 */

static int str2DiskAddress(const char *str, struct DiskAddress *diskAddr)
{
    char *type, *addr;

    if (!diskAddr)
        return -1;
    if (!str)
        return -1;

    type = (char *)str;
    addr = strchr(type, ':');
    if (!addr)
        return -1;

    if (STREQLEN(type, "pci", addr - type)) {
        diskAddr->type = DISK_ADDR_TYPE_PCI;
        return str2PCIAddress(addr + 1, &diskAddr->addr.pci);
    } else if (STREQLEN(type, "scsi", addr - type)) {
        diskAddr->type = DISK_ADDR_TYPE_SCSI;
        return str2SCSIAddress(addr + 1, &diskAddr->addr.scsi);
    } else if (STREQLEN(type, "ide", addr - type)) {
        diskAddr->type = DISK_ADDR_TYPE_IDE;
        return str2IDEAddress(addr + 1, &diskAddr->addr.ide);
    }

    return -1;
}

E
Eric Blake 已提交
14148
static bool
14149
cmdAttachDisk(vshControl *ctl, const vshCmd *cmd)
14150 14151
{
    virDomainPtr dom = NULL;
14152
    const char *source = NULL, *target = NULL, *driver = NULL,
14153 14154 14155
                *subdriver = NULL, *type = NULL, *mode = NULL,
                *cache = NULL, *serial = NULL, *straddr = NULL;
    struct DiskAddress diskAddr;
W
Wen Congyang 已提交
14156 14157
    bool isFile = false, functionReturn = false;
    int ret;
J
Jim Fehlig 已提交
14158
    unsigned int flags;
14159
    const char *stype = NULL;
14160 14161
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    char *xml;
14162

14163
    if (!vshConnectionUsability(ctl, ctl->conn))
14164 14165
        goto cleanup;

J
Jim Meyering 已提交
14166
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
14167 14168
        goto cleanup;

14169
    if (vshCommandOptString(cmd, "source", &source) <= 0)
14170
        goto cleanup;
14171 14172 14173 14174
    /* Allow empty string as a placeholder that implies no source, for
     * use in adding a cdrom drive with no disk.  */
    if (!*source)
        source = NULL;
14175

14176
    if (vshCommandOptString(cmd, "target", &target) <= 0)
14177 14178
        goto cleanup;

14179 14180 14181 14182
    if (vshCommandOptString(cmd, "driver", &driver) < 0 ||
        vshCommandOptString(cmd, "subdriver", &subdriver) < 0 ||
        vshCommandOptString(cmd, "type", &type) < 0 ||
        vshCommandOptString(cmd, "mode", &mode) < 0 ||
14183 14184 14185
        vshCommandOptString(cmd, "cache", &cache) < 0 ||
        vshCommandOptString(cmd, "serial", &serial) < 0 ||
        vshCommandOptString(cmd, "address", &straddr) < 0 ||
14186
        vshCommandOptString(cmd, "sourcetype", &stype) < 0) {
14187
        vshError(ctl, "%s", _("missing option"));
14188 14189
        goto cleanup;
    }
14190

14191 14192
    if (!stype) {
        if (driver && (STREQ(driver, "file") || STREQ(driver, "tap")))
E
Eric Blake 已提交
14193
            isFile = true;
14194
    } else if (STREQ(stype, "file")) {
E
Eric Blake 已提交
14195
        isFile = true;
14196 14197 14198
    } else if (STRNEQ(stype, "block")) {
        vshError(ctl, _("Unknown source type: '%s'"), stype);
        goto cleanup;
14199 14200 14201
    }

    if (mode) {
14202
        if (STRNEQ(mode, "readonly") && STRNEQ(mode, "shareable")) {
E
Eric Blake 已提交
14203 14204
            vshError(ctl, _("No support for %s in command 'attach-disk'"),
                     mode);
14205 14206 14207 14208 14209
            goto cleanup;
        }
    }

    /* Make XML of disk */
14210
    virBufferAsprintf(&buf, "<disk type='%s'",
14211 14212
                      (isFile) ? "file" : "block");
    if (type)
14213
        virBufferAsprintf(&buf, " device='%s'", type);
14214 14215
    if (vshCommandOptBool (cmd, "rawio"))
        virBufferAddLit(&buf, " rawio='yes'");
14216 14217
    virBufferAddLit(&buf, ">\n");

14218
    if (driver || subdriver)
14219
        virBufferAsprintf(&buf, "  <driver");
14220 14221

    if (driver)
14222
        virBufferAsprintf(&buf, " name='%s'", driver);
14223
    if (subdriver)
14224
        virBufferAsprintf(&buf, " type='%s'", subdriver);
14225 14226
    if (cache)
        virBufferAsprintf(&buf, " cache='%s'", cache);
14227

14228
    if (driver || subdriver || cache)
14229
        virBufferAddLit(&buf, "/>\n");
14230

14231 14232 14233 14234
    if (source)
        virBufferAsprintf(&buf, "  <source %s='%s'/>\n",
                          (isFile) ? "file" : "dev",
                          source);
14235
    virBufferAsprintf(&buf, "  <target dev='%s'/>\n", target);
14236
    if (mode)
14237
        virBufferAsprintf(&buf, "  <%s/>\n", mode);
14238

14239 14240 14241 14242 14243 14244 14245 14246 14247 14248 14249 14250 14251 14252 14253 14254
    if (serial)
        virBufferAsprintf(&buf, "  <serial>%s</serial>\n", serial);

    if (vshCommandOptBool(cmd, "shareable"))
        virBufferAsprintf(&buf, "  <shareable/>\n");

    if (straddr) {
        if (str2DiskAddress(straddr, &diskAddr) != 0) {
            vshError(ctl, _("Invalid address."));
            goto cleanup;
        }

        if (STRPREFIX((const char *)target, "vd")) {
            if (diskAddr.type == DISK_ADDR_TYPE_PCI) {
                virBufferAsprintf(&buf,
                                  "  <address type='pci' domain='0x%04x'"
14255
                                  " bus ='0x%02x' slot='0x%02x' function='0x%0x'",
14256 14257
                                  diskAddr.addr.pci.domain, diskAddr.addr.pci.bus,
                                  diskAddr.addr.pci.slot, diskAddr.addr.pci.function);
14258 14259 14260
                if (vshCommandOptBool(cmd, "multifunction"))
                    virBufferAddLit(&buf, " multifunction='on'");
                virBufferAddLit(&buf, "/>\n");
14261 14262 14263 14264 14265 14266 14267 14268 14269 14270 14271 14272 14273 14274 14275 14276 14277 14278 14279 14280 14281 14282 14283 14284 14285 14286 14287 14288 14289
            } else {
                vshError(ctl, "%s", _("expecting a pci:0000.00.00.00 address."));
                goto cleanup;
            }
        } else if (STRPREFIX((const char *)target, "sd")) {
            if (diskAddr.type == DISK_ADDR_TYPE_SCSI) {
                virBufferAsprintf(&buf,
                                  "  <address type='drive' controller='%d'"
                                  " bus='%d' unit='%d' />\n",
                                  diskAddr.addr.scsi.controller, diskAddr.addr.scsi.bus,
                                  diskAddr.addr.scsi.unit);
            } else {
                vshError(ctl, "%s", _("expecting a scsi:00.00.00 address."));
                goto cleanup;
            }
        } else if (STRPREFIX((const char *)target, "hd")) {
            if (diskAddr.type == DISK_ADDR_TYPE_IDE) {
                virBufferAsprintf(&buf,
                                  "  <address type='drive' controller='%d'"
                                  " bus='%d' unit='%d' />\n",
                                  diskAddr.addr.ide.controller, diskAddr.addr.ide.bus,
                                  diskAddr.addr.ide.unit);
            } else {
                vshError(ctl, "%s", _("expecting an ide:00.00.00 address."));
                goto cleanup;
            }
        }
    }

14290
    virBufferAddLit(&buf, "</disk>\n");
14291

14292 14293
    if (virBufferError(&buf)) {
        vshPrint(ctl, "%s", _("Failed to allocate XML buffer"));
E
Eric Blake 已提交
14294
        return false;
14295 14296
    }

14297
    xml = virBufferContentAndReset(&buf);
14298

J
Jim Fehlig 已提交
14299
    if (vshCommandOptBool(cmd, "persistent")) {
14300
        flags = VIR_DOMAIN_AFFECT_CONFIG;
J
Jim Fehlig 已提交
14301
        if (virDomainIsActive(dom) == 1)
14302
            flags |= VIR_DOMAIN_AFFECT_LIVE;
14303
        ret = virDomainAttachDeviceFlags(dom, xml, flags);
J
Jim Fehlig 已提交
14304
    } else {
14305
        ret = virDomainAttachDevice(dom, xml);
J
Jim Fehlig 已提交
14306
    }
14307

14308 14309
    VIR_FREE(xml);

J
Jim Fehlig 已提交
14310
    if (ret != 0) {
L
Laine Stump 已提交
14311
        vshError(ctl, "%s", _("Failed to attach disk"));
J
Jim Fehlig 已提交
14312 14313
    } else {
        vshPrint(ctl, "%s", _("Disk attached successfully\n"));
W
Wen Congyang 已提交
14314
        functionReturn = true;
J
Jim Fehlig 已提交
14315
    }
14316 14317 14318 14319

 cleanup:
    if (dom)
        virDomainFree(dom);
14320
    virBufferFreeAndReset(&buf);
W
Wen Congyang 已提交
14321
    return functionReturn;
14322 14323 14324 14325 14326
}

/*
 * "detach-disk" command
 */
14327
static const vshCmdInfo info_detach_disk[] = {
14328 14329
    {"help", N_("detach disk device")},
    {"desc", N_("Detach disk device.")},
14330 14331 14332
    {NULL, NULL}
};

14333
static const vshCmdOptDef opts_detach_disk[] = {
14334 14335 14336
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"target", VSH_OT_DATA, VSH_OFLAG_REQ, N_("target of disk device")},
    {"persistent", VSH_OT_BOOL, 0, N_("persist disk detachment")},
14337 14338 14339
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
14340
static bool
14341
cmdDetachDisk(vshControl *ctl, const vshCmd *cmd)
14342 14343 14344 14345 14346 14347 14348
{
    xmlDocPtr xml = NULL;
    xmlXPathObjectPtr obj=NULL;
    xmlXPathContextPtr ctxt = NULL;
    xmlNodePtr cur = NULL;
    xmlBufferPtr xml_buf = NULL;
    virDomainPtr dom = NULL;
14349
    const char *target = NULL;
14350
    char *doc;
E
Eric Blake 已提交
14351
    int i = 0, diff_tgt;
W
Wen Congyang 已提交
14352 14353
    int ret;
    bool functionReturn = false;
J
Jim Fehlig 已提交
14354
    unsigned int flags;
14355

14356
    if (!vshConnectionUsability(ctl, ctl->conn))
14357 14358
        goto cleanup;

J
Jim Meyering 已提交
14359
    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
14360 14361
        goto cleanup;

14362
    if (vshCommandOptString(cmd, "target", &target) <= 0)
14363 14364 14365 14366 14367 14368
        goto cleanup;

    doc = virDomainGetXMLDesc(dom, 0);
    if (!doc)
        goto cleanup;

14369
    xml = virXMLParseStringCtxt(doc, _("(domain_definition)"), &ctxt);
14370
    VIR_FREE(doc);
14371
    if (!xml) {
14372
        vshError(ctl, "%s", _("Failed to get disk information"));
14373 14374 14375 14376 14377 14378
        goto cleanup;
    }

    obj = xmlXPathEval(BAD_CAST "/domain/devices/disk", ctxt);
    if ((obj == NULL) || (obj->type != XPATH_NODESET) ||
        (obj->nodesetval == NULL) || (obj->nodesetval->nodeNr == 0)) {
14379
        vshError(ctl, "%s", _("Failed to get disk information"));
14380 14381 14382 14383 14384 14385 14386
        goto cleanup;
    }

    /* search target */
    for (; i < obj->nodesetval->nodeNr; i++) {
        cur = obj->nodesetval->nodeTab[i]->children;
        while (cur != NULL) {
14387 14388 14389 14390 14391
            if (cur->type == XML_ELEMENT_NODE &&
                xmlStrEqual(cur->name, BAD_CAST "target")) {
                char *tmp_tgt = virXMLPropString(cur, "dev");
                diff_tgt = STREQ(tmp_tgt, target);
                VIR_FREE(tmp_tgt);
14392 14393 14394 14395 14396 14397 14398
                if (diff_tgt) {
                    goto hit;
                }
            }
            cur = cur->next;
        }
    }
14399
    vshError(ctl, _("No found disk whose target is %s"), target);
14400 14401 14402 14403 14404
    goto cleanup;

 hit:
    xml_buf = xmlBufferCreate();
    if (!xml_buf) {
14405
        vshError(ctl, "%s", _("Failed to allocate memory"));
14406 14407 14408
        goto cleanup;
    }

14409
    if (xmlNodeDump(xml_buf, xml, obj->nodesetval->nodeTab[i], 0, 0) < 0) {
14410
        vshError(ctl, "%s", _("Failed to create XML"));
14411 14412 14413
        goto cleanup;
    }

J
Jim Fehlig 已提交
14414
    if (vshCommandOptBool(cmd, "persistent")) {
14415
        flags = VIR_DOMAIN_AFFECT_CONFIG;
J
Jim Fehlig 已提交
14416
        if (virDomainIsActive(dom) == 1)
14417
            flags |= VIR_DOMAIN_AFFECT_LIVE;
J
Jim Fehlig 已提交
14418 14419 14420 14421 14422 14423 14424 14425
        ret = virDomainDetachDeviceFlags(dom,
                                         (char *)xmlBufferContent(xml_buf),
                                         flags);
    } else {
        ret = virDomainDetachDevice(dom, (char *)xmlBufferContent(xml_buf));
    }

    if (ret != 0) {
L
Laine Stump 已提交
14426
        vshError(ctl, "%s", _("Failed to detach disk"));
J
Jim Fehlig 已提交
14427
    } else {
J
Jim Meyering 已提交
14428
        vshPrint(ctl, "%s", _("Disk detached successfully\n"));
W
Wen Congyang 已提交
14429
        functionReturn = true;
14430
    }
14431 14432

 cleanup:
14433
    xmlXPathFreeObject(obj);
14434
    xmlXPathFreeContext(ctxt);
14435 14436
    xmlFreeDoc(xml);
    xmlBufferFree(xml_buf);
14437 14438
    if (dom)
        virDomainFree(dom);
W
Wen Congyang 已提交
14439
    return functionReturn;
14440 14441
}

14442 14443 14444 14445
/*
 * "cpu-compare" command
 */
static const vshCmdInfo info_cpu_compare[] = {
14446 14447
    {"help", N_("compare host CPU with a CPU described by an XML file")},
    {"desc", N_("compare CPU with host CPU")},
14448 14449 14450 14451
    {NULL, NULL}
};

static const vshCmdOptDef opts_cpu_compare[] = {
14452
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file containing an XML CPU description")},
14453 14454 14455
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
14456
static bool
14457 14458
cmdCPUCompare(vshControl *ctl, const vshCmd *cmd)
{
14459
    const char *from = NULL;
14460
    bool ret = false;
14461 14462
    char *buffer;
    int result;
14463 14464 14465 14466 14467 14468
    const char *snippet;

    xmlDocPtr xml = NULL;
    xmlXPathContextPtr ctxt = NULL;
    xmlBufferPtr xml_buf = NULL;
    xmlNodePtr node;
14469

14470
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
14471
        return false;
14472

14473
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
14474
        return false;
14475

14476 14477 14478
    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0) {
        vshError(ctl, _("Failed to read file '%s' to compare"),
                 from);
E
Eric Blake 已提交
14479
        return false;
14480
    }
14481

14482 14483 14484 14485 14486 14487 14488 14489 14490 14491 14492 14493 14494 14495 14496 14497 14498 14499 14500 14501 14502 14503 14504 14505 14506
    /* try to extract the CPU element from as it would appear in a domain XML*/
    if (!(xml = virXMLParseStringCtxt(buffer, from, &ctxt)))
        goto cleanup;

    if ((node = virXPathNode("/cpu|"
                             "/domain/cpu|"
                              "/capabilities/host/cpu", ctxt))) {
        if (!(xml_buf = xmlBufferCreate())) {
            vshError(ctl, _("Can't create XML buffer to extract CPU element."));
            goto cleanup;
        }

        if (xmlNodeDump(xml_buf, xml, node, 0, 0) < 0) {
            vshError(ctl, _("Failed to extract CPU element snippet from domain XML."));
            goto cleanup;
        }

        snippet = (const char *) xmlBufferContent(xml_buf);
    } else {
        vshError(ctl, _("File '%s' does not contain a <cpu> element or is not "
                        "a valid domain or capabilities XML"), from);
        goto cleanup;
    }

    result = virConnectCompareCPU(ctl->conn, snippet, 0);
14507 14508 14509 14510 14511

    switch (result) {
    case VIR_CPU_COMPARE_INCOMPATIBLE:
        vshPrint(ctl, _("CPU described in %s is incompatible with host CPU\n"),
                 from);
14512
        goto cleanup;
14513 14514 14515 14516 14517 14518 14519 14520 14521 14522 14523 14524 14525 14526 14527
        break;

    case VIR_CPU_COMPARE_IDENTICAL:
        vshPrint(ctl, _("CPU described in %s is identical to host CPU\n"),
                 from);
        break;

    case VIR_CPU_COMPARE_SUPERSET:
        vshPrint(ctl, _("Host CPU is a superset of CPU described in %s\n"),
                 from);
        break;

    case VIR_CPU_COMPARE_ERROR:
    default:
        vshError(ctl, _("Failed to compare host CPU with %s"), from);
14528
        goto cleanup;
14529 14530
    }

14531 14532 14533 14534 14535 14536 14537 14538
    ret = true;

cleanup:
    VIR_FREE(buffer);
    xmlBufferFree(xml_buf);
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(xml);

14539 14540 14541
    return ret;
}

14542 14543 14544 14545
/*
 * "cpu-baseline" command
 */
static const vshCmdInfo info_cpu_baseline[] = {
14546 14547
    {"help", N_("compute baseline CPU")},
    {"desc", N_("Compute baseline CPU for a set of given CPUs.")},
14548 14549 14550 14551
    {NULL, NULL}
};

static const vshCmdOptDef opts_cpu_baseline[] = {
14552
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, N_("file containing XML CPU descriptions")},
14553 14554 14555
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
14556
static bool
14557 14558
cmdCPUBaseline(vshControl *ctl, const vshCmd *cmd)
{
14559
    const char *from = NULL;
14560
    bool ret = false;
14561 14562 14563
    char *buffer;
    char *result = NULL;
    const char **list = NULL;
14564 14565 14566 14567
    int count = 0;

    xmlDocPtr xml = NULL;
    xmlNodePtr *node_list = NULL;
14568
    xmlXPathContextPtr ctxt = NULL;
14569 14570 14571
    xmlBufferPtr xml_buf = NULL;
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    int i;
14572

14573
    if (!vshConnectionUsability(ctl, ctl->conn))
E
Eric Blake 已提交
14574
        return false;
14575

14576
    if (vshCommandOptString(cmd, "file", &from) <= 0)
E
Eric Blake 已提交
14577
        return false;
14578 14579

    if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0)
E
Eric Blake 已提交
14580
        return false;
14581

E
Eric Blake 已提交
14582
    /* add a separate container around the xml */
14583 14584
    virBufferStrcat(&buf, "<container>", buffer, "</container>", NULL);
    if (virBufferError(&buf))
14585 14586
        goto no_memory;

14587 14588 14589 14590 14591 14592 14593 14594 14595 14596 14597 14598 14599
    VIR_FREE(buffer);
    buffer = virBufferContentAndReset(&buf);


    if (!(xml = virXMLParseStringCtxt(buffer, from, &ctxt)))
        goto cleanup;

    if ((count = virXPathNodeSet("//cpu[not(ancestor::cpus)]",
                                 ctxt, &node_list)) == -1)
        goto cleanup;

    if (count == 0) {
        vshError(ctl, _("No host CPU specified in '%s'"), from);
14600 14601 14602
        goto cleanup;
    }

14603
    list = vshCalloc(ctl, count, sizeof(const char *));
14604

14605
    if (!(xml_buf = xmlBufferCreate()))
14606 14607
        goto no_memory;

14608 14609
    for (i = 0; i < count; i++) {
        xmlBufferEmpty(xml_buf);
14610

14611 14612 14613
        if (xmlNodeDump(xml_buf, xml,  node_list[i], 0, 0) < 0) {
            vshError(ctl, _("Failed to extract <cpu> element"));
            goto cleanup;
14614 14615
        }

14616
        list[i] = vshStrdup(ctl, (const char *)xmlBufferContent(xml_buf));
14617 14618 14619 14620
    }

    result = virConnectBaselineCPU(ctl->conn, list, count, 0);

14621
    if (result) {
14622
        vshPrint(ctl, "%s", result);
14623 14624
        ret = true;
    }
14625 14626 14627

cleanup:
    xmlXPathFreeContext(ctxt);
14628 14629
    xmlFreeDoc(xml);
    xmlBufferFree(xml_buf);
14630 14631
    VIR_FREE(result);
    if ((list != NULL) && (count > 0)) {
14632
        for (i = 0; i < count; i++)
14633 14634 14635 14636 14637 14638 14639 14640 14641
            VIR_FREE(list[i]);
    }
    VIR_FREE(list);
    VIR_FREE(buffer);

    return ret;

no_memory:
    vshError(ctl, "%s", _("Out of memory"));
E
Eric Blake 已提交
14642
    ret = false;
14643
    goto cleanup;
14644 14645
}

14646 14647 14648 14649 14650 14651 14652 14653
/* Common code for the edit / net-edit / pool-edit functions which follow. */
static char *
editWriteToTempFile (vshControl *ctl, const char *doc)
{
    char *ret;
    const char *tmpdir;
    int fd;

14654
    ret = vshMalloc(ctl, PATH_MAX);
14655 14656 14657

    tmpdir = getenv ("TMPDIR");
    if (!tmpdir) tmpdir = "/tmp";
14658 14659
    snprintf (ret, PATH_MAX, "%s/virshXXXXXX.xml", tmpdir);
    fd = mkstemps(ret, 4);
14660
    if (fd == -1) {
14661
        vshError(ctl, _("mkstemps: failed to create temporary file: %s"),
14662
                 strerror(errno));
14663
        VIR_FREE(ret);
14664 14665 14666 14667
        return NULL;
    }

    if (safewrite (fd, doc, strlen (doc)) == -1) {
14668 14669
        vshError(ctl, _("write: %s: failed to write to temporary file: %s"),
                 ret, strerror(errno));
S
Stefan Berger 已提交
14670
        VIR_FORCE_CLOSE(fd);
14671
        unlink (ret);
14672
        VIR_FREE(ret);
14673 14674
        return NULL;
    }
S
Stefan Berger 已提交
14675
    if (VIR_CLOSE(fd) < 0) {
14676 14677
        vshError(ctl, _("close: %s: failed to write or close temporary file: %s"),
                 ret, strerror(errno));
14678
        unlink (ret);
14679
        VIR_FREE(ret);
14680 14681 14682 14683 14684 14685 14686 14687 14688 14689 14690 14691 14692 14693 14694
        return NULL;
    }

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

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

static int
editFile (vshControl *ctl, const char *filename)
{
    const char *editor;
E
Eric Blake 已提交
14695 14696 14697 14698
    virCommandPtr cmd;
    int ret = -1;
    int outfd = STDOUT_FILENO;
    int errfd = STDERR_FILENO;
14699

14700
    editor = getenv ("VISUAL");
E
Eric Blake 已提交
14701 14702 14703 14704
    if (!editor)
        editor = getenv ("EDITOR");
    if (!editor)
        editor = "vi"; /* could be cruel & default to ed(1) here */
14705

14706 14707 14708 14709 14710
    /* 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 已提交
14711 14712
     * is why sudo scrubs it by default).  Conversely, if the editor
     * is safe, we can run it directly rather than wasting a shell.
14713
     */
E
Eric Blake 已提交
14714 14715 14716 14717 14718 14719 14720 14721 14722 14723 14724 14725
    if (strspn (editor, ACCEPTED_CHARS) != strlen (editor)) {
        if (strspn (filename, ACCEPTED_CHARS) != strlen (filename)) {
            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);
14726 14727
    }

E
Eric Blake 已提交
14728 14729 14730 14731 14732 14733 14734
    virCommandSetInputFD(cmd, STDIN_FILENO);
    virCommandSetOutputFD(cmd, &outfd);
    virCommandSetErrorFD(cmd, &errfd);
    if (virCommandRunAsync(cmd, NULL) < 0 ||
        virCommandWait(cmd, NULL) < 0) {
        virshReportError(ctl);
        goto cleanup;
14735
    }
E
Eric Blake 已提交
14736
    ret = 0;
14737

E
Eric Blake 已提交
14738 14739 14740
cleanup:
    virCommandFree(cmd);
    return ret;
14741 14742 14743 14744 14745 14746 14747 14748
}

static char *
editReadBackFile (vshControl *ctl, const char *filename)
{
    char *ret;

    if (virFileReadAll (filename, VIRSH_MAX_XML_FILE, &ret) == -1) {
14749
        vshError(ctl,
14750
                 _("%s: failed to read temporary file: %s"),
14751
                 filename, strerror(errno));
14752 14753 14754 14755 14756
        return NULL;
    }
    return ret;
}

14757

P
Paolo Bonzini 已提交
14758 14759 14760 14761
/*
 * "cd" command
 */
static const vshCmdInfo info_cd[] = {
14762 14763
    {"help", N_("change the current directory")},
    {"desc", N_("Change the current directory.")},
P
Paolo Bonzini 已提交
14764 14765 14766 14767
    {NULL, NULL}
};

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

E
Eric Blake 已提交
14772
static bool
14773
cmdCd(vshControl *ctl, const vshCmd *cmd)
P
Paolo Bonzini 已提交
14774
{
14775
    const char *dir = NULL;
14776
    char *dir_malloced = NULL;
E
Eric Blake 已提交
14777
    bool ret = true;
P
Paolo Bonzini 已提交
14778 14779

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

14784
    if (vshCommandOptString(cmd, "dir", &dir) <= 0) {
P
Paolo Bonzini 已提交
14785
        uid_t uid = geteuid();
14786
        dir = dir_malloced = virGetUserDirectory(uid);
P
Paolo Bonzini 已提交
14787 14788 14789 14790
    }
    if (!dir)
        dir = "/";

P
Phil Petty 已提交
14791
    if (chdir(dir) == -1) {
14792
        vshError(ctl, _("cd: %s: %s"), strerror(errno), dir);
E
Eric Blake 已提交
14793
        ret = false;
P
Paolo Bonzini 已提交
14794 14795
    }

14796
    VIR_FREE(dir_malloced);
P
Phil Petty 已提交
14797
    return ret;
P
Paolo Bonzini 已提交
14798 14799 14800 14801 14802 14803
}

/*
 * "pwd" command
 */
static const vshCmdInfo info_pwd[] = {
14804 14805
    {"help", N_("print the current directory")},
    {"desc", N_("Print the current directory.")},
P
Paolo Bonzini 已提交
14806 14807 14808
    {NULL, NULL}
};

E
Eric Blake 已提交
14809
static bool
P
Paolo Bonzini 已提交
14810 14811 14812
cmdPwd(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
{
    char *cwd;
14813
    bool ret = true;
P
Paolo Bonzini 已提交
14814

14815 14816
    cwd = getcwd(NULL, 0);
    if (!cwd) {
14817 14818
        vshError(ctl, _("pwd: cannot get current directory: %s"),
                 strerror(errno));
14819 14820
        ret = false;
    } else {
P
Paolo Bonzini 已提交
14821
        vshPrint (ctl, _("%s\n"), cwd);
14822 14823
        VIR_FREE(cwd);
    }
P
Paolo Bonzini 已提交
14824

14825
    return ret;
P
Paolo Bonzini 已提交
14826 14827
}

E
Eric Blake 已提交
14828 14829 14830 14831 14832 14833 14834 14835 14836 14837 14838 14839
/*
 * "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")},
14840
    {"string", VSH_OT_ARGV, 0, N_("arguments to echo")},
E
Eric Blake 已提交
14841 14842 14843 14844 14845 14846
    {NULL, 0, 0, NULL}
};

/* Exists mainly for debugging virsh, but also handy for adding back
 * quotes for later evaluation.
 */
E
Eric Blake 已提交
14847
static bool
14848
cmdEcho (vshControl *ctl, const vshCmd *cmd)
E
Eric Blake 已提交
14849 14850 14851 14852
{
    bool shell = false;
    bool xml = false;
    int count = 0;
14853
    const vshCmdOpt *opt = NULL;
E
Eric Blake 已提交
14854 14855 14856 14857 14858 14859 14860 14861
    char *arg;
    virBuffer buf = VIR_BUFFER_INITIALIZER;

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

14862
    while ((opt = vshCommandOptArgv(cmd, opt))) {
14863 14864
        char *str;
        virBuffer xmlbuf = VIR_BUFFER_INITIALIZER;
E
Eric Blake 已提交
14865

14866
        arg = opt->data;
14867

E
Eric Blake 已提交
14868 14869
        if (count)
            virBufferAddChar(&buf, ' ');
14870

E
Eric Blake 已提交
14871
        if (xml) {
14872 14873 14874 14875
            virBufferEscapeString(&xmlbuf, "%s", arg);
            if (virBufferError(&buf)) {
                vshPrint(ctl, "%s", _("Failed to allocate XML buffer"));
                return false;
E
Eric Blake 已提交
14876
            }
14877 14878 14879
            str = virBufferContentAndReset(&xmlbuf);
        } else {
            str = vshStrdup(ctl, arg);
E
Eric Blake 已提交
14880
        }
14881 14882 14883 14884 14885

        if (shell)
            virBufferEscapeShell(&buf, str);
        else
            virBufferAdd(&buf, str, -1);
E
Eric Blake 已提交
14886
        count++;
14887
        VIR_FREE(str);
E
Eric Blake 已提交
14888 14889 14890 14891
    }

    if (virBufferError(&buf)) {
        vshPrint(ctl, "%s", _("Failed to allocate XML buffer"));
E
Eric Blake 已提交
14892
        return false;
E
Eric Blake 已提交
14893 14894 14895 14896 14897
    }
    arg = virBufferContentAndReset(&buf);
    if (arg)
        vshPrint(ctl, "%s", arg);
    VIR_FREE(arg);
E
Eric Blake 已提交
14898
    return true;
E
Eric Blake 已提交
14899 14900
}

14901 14902 14903 14904
/*
 * "edit" command
 */
static const vshCmdInfo info_edit[] = {
14905 14906
    {"help", N_("edit XML configuration for a domain")},
    {"desc", N_("Edit the XML configuration for a domain.")},
14907 14908 14909 14910
    {NULL, NULL}
};

static const vshCmdOptDef opts_edit[] = {
14911
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
14912 14913 14914 14915 14916 14917
    {NULL, 0, 0, NULL}
};

/* This function also acts as a template to generate cmdNetworkEdit
 * and cmdPoolEdit functions (below) using a sed script in the Makefile.
 */
E
Eric Blake 已提交
14918
static bool
14919 14920
cmdEdit (vshControl *ctl, const vshCmd *cmd)
{
E
Eric Blake 已提交
14921
    bool ret = false;
14922 14923 14924 14925 14926
    virDomainPtr dom = NULL;
    char *tmp = NULL;
    char *doc = NULL;
    char *doc_edited = NULL;
    char *doc_reread = NULL;
E
Eric Blake 已提交
14927
    unsigned int flags = VIR_DOMAIN_XML_SECURE | VIR_DOMAIN_XML_INACTIVE;
14928

14929
    if (!vshConnectionUsability(ctl, ctl->conn))
14930 14931
        goto cleanup;

J
Jim Meyering 已提交
14932
    dom = vshCommandOptDomain (ctl, cmd, NULL);
14933 14934 14935 14936
    if (dom == NULL)
        goto cleanup;

    /* Get the XML configuration of the domain. */
14937
    doc = virDomainGetXMLDesc (dom, flags);
14938 14939 14940 14941 14942 14943 14944 14945 14946 14947 14948 14949 14950 14951 14952 14953 14954 14955
    if (!doc)
        goto cleanup;

    /* Create and open the temporary file. */
    tmp = editWriteToTempFile (ctl, doc);
    if (!tmp) goto cleanup;

    /* Start the editor. */
    if (editFile (ctl, tmp) == -1) goto cleanup;

    /* Read back the edited file. */
    doc_edited = editReadBackFile (ctl, tmp);
    if (!doc_edited) goto cleanup;

    /* Compare original XML with edited.  Has it changed at all? */
    if (STREQ (doc, doc_edited)) {
        vshPrint (ctl, _("Domain %s XML configuration not changed.\n"),
                  virDomainGetName (dom));
E
Eric Blake 已提交
14956
        ret = true;
14957 14958 14959 14960 14961 14962 14963
        goto cleanup;
    }

    /* Now re-read the domain XML.  Did someone else change it while
     * it was being edited?  This also catches problems such as us
     * losing a connection or the domain going away.
     */
14964
    doc_reread = virDomainGetXMLDesc (dom, flags);
14965 14966 14967 14968
    if (!doc_reread)
        goto cleanup;

    if (STRNEQ (doc, doc_reread)) {
14969 14970
        vshError(ctl,
                 "%s", _("ERROR: the XML configuration was changed by another user"));
14971 14972 14973 14974 14975 14976 14977 14978 14979 14980 14981 14982
        goto cleanup;
    }

    /* Everything checks out, so redefine the domain. */
    virDomainFree (dom);
    dom = virDomainDefineXML (ctl->conn, doc_edited);
    if (!dom)
        goto cleanup;

    vshPrint (ctl, _("Domain %s XML configuration edited.\n"),
              virDomainGetName(dom));

E
Eric Blake 已提交
14983
    ret = true;
14984 14985 14986 14987 14988

 cleanup:
    if (dom)
        virDomainFree (dom);

14989 14990 14991
    VIR_FREE(doc);
    VIR_FREE(doc_edited);
    VIR_FREE(doc_reread);
14992 14993 14994

    if (tmp) {
        unlink (tmp);
14995
        VIR_FREE(tmp);
14996 14997 14998 14999 15000
    }

    return ret;
}

15001

15002 15003 15004 15005
/*
 * "net-edit" command
 */
static const vshCmdInfo info_network_edit[] = {
15006 15007
    {"help", N_("edit XML configuration for a network")},
    {"desc", N_("Edit the XML configuration for a network.")},
15008 15009 15010 15011
    {NULL, NULL}
};

static const vshCmdOptDef opts_network_edit[] = {
15012
    {"network", VSH_OT_DATA, VSH_OFLAG_REQ, N_("network name or uuid")},
15013 15014 15015 15016 15017 15018 15019 15020 15021 15022
    {NULL, 0, 0, NULL}
};

/* This is generated from this file by a sed script in the Makefile. */
#include "virsh-net-edit.c"

/*
 * "pool-edit" command
 */
static const vshCmdInfo info_pool_edit[] = {
15023 15024
    {"help", N_("edit XML configuration for a storage pool")},
    {"desc", N_("Edit the XML configuration for a storage pool.")},
15025 15026 15027 15028
    {NULL, NULL}
};

static const vshCmdOptDef opts_pool_edit[] = {
15029
    {"pool", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pool name or uuid")},
15030 15031 15032 15033 15034 15035
    {NULL, 0, 0, NULL}
};

/* This is generated from this file by a sed script in the Makefile. */
#include "virsh-pool-edit.c"

K
Karel Zak 已提交
15036 15037 15038
/*
 * "quit" command
 */
15039
static const vshCmdInfo info_quit[] = {
15040
    {"help", N_("quit this interactive terminal")},
15041
    {"desc", ""},
15042
    {NULL, NULL}
K
Karel Zak 已提交
15043 15044
};

E
Eric Blake 已提交
15045
static bool
15046
cmdQuit(vshControl *ctl, const vshCmd *cmd ATTRIBUTE_UNUSED)
15047
{
E
Eric Blake 已提交
15048 15049
    ctl->imode = false;
    return true;
K
Karel Zak 已提交
15050 15051
}

15052 15053 15054 15055 15056 15057 15058
/* Helper for snapshot-create and snapshot-create-as */
static bool
vshSnapshotCreate(vshControl *ctl, virDomainPtr dom, const char *buffer,
                  unsigned int flags, const char *from)
{
    bool ret = false;
    virDomainSnapshotPtr snapshot;
15059
    bool halt = false;
15060 15061 15062
    char *doc = NULL;
    xmlDocPtr xml = NULL;
    xmlXPathContextPtr ctxt = NULL;
E
Eric Blake 已提交
15063
    const char *name = NULL;
15064 15065

    snapshot = virDomainSnapshotCreateXML(dom, buffer, flags);
15066 15067 15068 15069 15070 15071 15072 15073 15074 15075 15076 15077 15078 15079 15080 15081 15082 15083 15084 15085 15086 15087 15088 15089

    /* Emulate --halt on older servers.  */
    if (!snapshot && last_error->code == VIR_ERR_INVALID_ARG &&
        (flags & VIR_DOMAIN_SNAPSHOT_CREATE_HALT)) {
        int persistent;

        virFreeError(last_error);
        last_error = NULL;
        persistent = virDomainIsPersistent(dom);
        if (persistent < 0) {
            virshReportError(ctl);
            goto cleanup;
        }
        if (!persistent) {
            vshError(ctl, "%s",
                     _("cannot halt after snapshot of transient domain"));
            goto cleanup;
        }
        if (virDomainIsActive(dom) == 1)
            halt = true;
        flags &= ~VIR_DOMAIN_SNAPSHOT_CREATE_HALT;
        snapshot = virDomainSnapshotCreateXML(dom, buffer, flags);
    }

15090 15091 15092
    if (snapshot == NULL)
        goto cleanup;

15093 15094 15095 15096 15097
    if (halt && virDomainDestroy(dom) < 0) {
        virshReportError(ctl);
        goto cleanup;
    }

E
Eric Blake 已提交
15098
    name = virDomainSnapshotGetName(snapshot);
15099
    if (!name) {
E
Eric Blake 已提交
15100
        vshError(ctl, "%s", _("Could not get snapshot name"));
15101 15102 15103 15104 15105 15106 15107 15108 15109 15110 15111 15112 15113 15114 15115 15116 15117 15118 15119
        goto cleanup;
    }

    if (from)
        vshPrint(ctl, _("Domain snapshot %s created from '%s'"), name, from);
    else
        vshPrint(ctl, _("Domain snapshot %s created"), name);

    ret = true;

cleanup:
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(xml);
    if (snapshot)
        virDomainSnapshotFree(snapshot);
    VIR_FREE(doc);
    return ret;
}

15120 15121 15122 15123
/*
 * "snapshot-create" command
 */
static const vshCmdInfo info_snapshot_create[] = {
E
Eric Blake 已提交
15124 15125
    {"help", N_("Create a snapshot from XML")},
    {"desc", N_("Create a snapshot (disk and RAM) from XML")},
15126 15127 15128 15129 15130 15131
    {NULL, NULL}
};

static const vshCmdOptDef opts_snapshot_create[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"xmlfile", VSH_OT_DATA, 0, N_("domain snapshot XML")},
15132 15133 15134
    {"redefine", VSH_OT_BOOL, 0, N_("redefine metadata for existing snapshot")},
    {"current", VSH_OT_BOOL, 0, N_("with redefine, set current snapshot")},
    {"no-metadata", VSH_OT_BOOL, 0, N_("take snapshot but create no metadata")},
15135
    {"halt", VSH_OT_BOOL, 0, N_("halt domain after snapshot is created")},
15136
    {"disk-only", VSH_OT_BOOL, 0, N_("capture disk state but not vm state")},
15137
    {"reuse-external", VSH_OT_BOOL, 0, N_("reuse any existing external files")},
15138
    {"quiesce", VSH_OT_BOOL, 0, N_("quiesce guest's file systems")},
15139 15140 15141
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
15142
static bool
15143 15144 15145
cmdSnapshotCreate(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
E
Eric Blake 已提交
15146
    bool ret = false;
15147
    const char *from = NULL;
15148
    char *buffer = NULL;
15149 15150 15151 15152 15153 15154 15155 15156
    unsigned int flags = 0;

    if (vshCommandOptBool(cmd, "redefine"))
        flags |= VIR_DOMAIN_SNAPSHOT_CREATE_REDEFINE;
    if (vshCommandOptBool(cmd, "current"))
        flags |= VIR_DOMAIN_SNAPSHOT_CREATE_CURRENT;
    if (vshCommandOptBool(cmd, "no-metadata"))
        flags |= VIR_DOMAIN_SNAPSHOT_CREATE_NO_METADATA;
15157 15158
    if (vshCommandOptBool(cmd, "halt"))
        flags |= VIR_DOMAIN_SNAPSHOT_CREATE_HALT;
15159 15160
    if (vshCommandOptBool(cmd, "disk-only"))
        flags |= VIR_DOMAIN_SNAPSHOT_CREATE_DISK_ONLY;
15161 15162
    if (vshCommandOptBool(cmd, "reuse-external"))
        flags |= VIR_DOMAIN_SNAPSHOT_CREATE_REUSE_EXT;
15163 15164
    if (vshCommandOptBool(cmd, "quiesce"))
        flags |= VIR_DOMAIN_SNAPSHOT_CREATE_QUIESCE;
15165

15166
    if (!vshConnectionUsability(ctl, ctl->conn))
15167 15168 15169 15170 15171 15172
        goto cleanup;

    dom = vshCommandOptDomain(ctl, cmd, NULL);
    if (dom == NULL)
        goto cleanup;

15173
    if (vshCommandOptString(cmd, "xmlfile", &from) <= 0)
E
Eric Blake 已提交
15174
        buffer = vshStrdup(ctl, "<domainsnapshot/>");
15175 15176 15177 15178 15179 15180 15181 15182 15183 15184 15185 15186 15187 15188 15189
    else {
        if (virFileReadAll(from, VIRSH_MAX_XML_FILE, &buffer) < 0) {
            /* we have to report the error here because during cleanup
             * we'll run through virDomainFree(), which loses the
             * last error
             */
            virshReportError(ctl);
            goto cleanup;
        }
    }
    if (buffer == NULL) {
        vshError(ctl, "%s", _("Out of memory"));
        goto cleanup;
    }

15190
    ret = vshSnapshotCreate(ctl, dom, buffer, flags, from);
15191 15192 15193 15194 15195 15196 15197 15198 15199

cleanup:
    VIR_FREE(buffer);
    if (dom)
        virDomainFree(dom);

    return ret;
}

15200 15201 15202
/*
 * "snapshot-create-as" command
 */
15203 15204 15205 15206 15207 15208 15209 15210 15211 15212 15213 15214 15215 15216 15217 15218 15219 15220 15221 15222 15223 15224 15225 15226 15227 15228 15229 15230 15231 15232 15233 15234 15235 15236 15237 15238 15239 15240 15241 15242 15243 15244 15245 15246 15247 15248 15249 15250 15251 15252 15253 15254 15255 15256 15257 15258
static int
vshParseSnapshotDiskspec(vshControl *ctl, virBufferPtr buf, const char *str)
{
    int ret = -1;
    char *name = NULL;
    char *snapshot = NULL;
    char *driver = NULL;
    char *file = NULL;
    char *spec = vshStrdup(ctl, str);
    char *tmp = spec;
    size_t len = strlen(str);

    if (*str == ',')
        goto cleanup;
    name = tmp;
    while ((tmp = strchr(tmp, ','))) {
        if (tmp[1] == ',') {
            /* Recognize ,, as an escape for a literal comma */
            memmove(&tmp[1], &tmp[2], len - (tmp - spec) + 2);
            len--;
            tmp++;
            continue;
        }
        /* Terminate previous string, look for next recognized one */
        *tmp++ = '\0';
        if (!snapshot && STRPREFIX(tmp, "snapshot="))
            snapshot = tmp + strlen("snapshot=");
        else if (!driver && STRPREFIX(tmp, "driver="))
            driver = tmp + strlen("driver=");
        else if (!file && STRPREFIX(tmp, "file="))
            file = tmp + strlen("file=");
        else
            goto cleanup;
    }

    virBufferEscapeString(buf, "    <disk name='%s'", name);
    if (snapshot)
        virBufferAsprintf(buf, " snapshot='%s'", snapshot);
    if (driver || file) {
        virBufferAddLit(buf, ">\n");
        if (driver)
            virBufferAsprintf(buf, "      <driver type='%s'/>\n", driver);
        if (file)
            virBufferEscapeString(buf, "      <source file='%s'/>\n", file);
        virBufferAddLit(buf, "    </disk>\n");
    } else {
        virBufferAddLit(buf, "/>\n");
    }
    ret = 0;
cleanup:
    if (ret < 0)
        vshError(ctl, _("unable to parse diskspec: %s"), str);
    VIR_FREE(spec);
    return ret;
}

15259 15260 15261 15262 15263 15264 15265 15266 15267 15268
static const vshCmdInfo info_snapshot_create_as[] = {
    {"help", N_("Create a snapshot from a set of args")},
    {"desc", N_("Create a snapshot (disk and RAM) from arguments")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_snapshot_create_as[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"name", VSH_OT_DATA, 0, N_("name of snapshot")},
    {"description", VSH_OT_DATA, 0, N_("description of snapshot")},
E
Eric Blake 已提交
15269
    {"print-xml", VSH_OT_BOOL, 0, N_("print XML document rather than create")},
15270
    {"no-metadata", VSH_OT_BOOL, 0, N_("take snapshot but create no metadata")},
15271
    {"halt", VSH_OT_BOOL, 0, N_("halt domain after snapshot is created")},
15272
    {"disk-only", VSH_OT_BOOL, 0, N_("capture disk state but not vm state")},
15273
    {"reuse-external", VSH_OT_BOOL, 0, N_("reuse any existing external files")},
15274
    {"quiesce", VSH_OT_BOOL, 0, N_("quiesce guest's file systems")},
15275 15276
    {"diskspec", VSH_OT_ARGV, 0,
     N_("disk attributes: disk[,snapshot=type][,driver=type][,file=name]")},
15277 15278 15279 15280 15281 15282 15283 15284 15285 15286 15287 15288
    {NULL, 0, 0, NULL}
};

static bool
cmdSnapshotCreateAs(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
    bool ret = false;
    char *buffer = NULL;
    const char *name = NULL;
    const char *desc = NULL;
    virBuffer buf = VIR_BUFFER_INITIALIZER;
15289
    unsigned int flags = 0;
15290
    const vshCmdOpt *opt = NULL;
15291 15292 15293

    if (vshCommandOptBool(cmd, "no-metadata"))
        flags |= VIR_DOMAIN_SNAPSHOT_CREATE_NO_METADATA;
15294 15295
    if (vshCommandOptBool(cmd, "halt"))
        flags |= VIR_DOMAIN_SNAPSHOT_CREATE_HALT;
15296 15297
    if (vshCommandOptBool(cmd, "disk-only"))
        flags |= VIR_DOMAIN_SNAPSHOT_CREATE_DISK_ONLY;
15298 15299
    if (vshCommandOptBool(cmd, "reuse-external"))
        flags |= VIR_DOMAIN_SNAPSHOT_CREATE_REUSE_EXT;
15300 15301
    if (vshCommandOptBool(cmd, "quiesce"))
        flags |= VIR_DOMAIN_SNAPSHOT_CREATE_QUIESCE;
15302 15303 15304 15305 15306 15307 15308 15309 15310 15311 15312 15313 15314 15315 15316 15317

    if (!vshConnectionUsability(ctl, ctl->conn))
        goto cleanup;

    dom = vshCommandOptDomain(ctl, cmd, NULL);
    if (dom == NULL)
        goto cleanup;

    if (vshCommandOptString(cmd, "name", &name) < 0 ||
        vshCommandOptString(cmd, "description", &desc) < 0) {
        vshError(ctl, _("argument must not be empty"));
        goto cleanup;
    }

    virBufferAddLit(&buf, "<domainsnapshot>\n");
    if (name)
15318
        virBufferEscapeString(&buf, "  <name>%s</name>\n", name);
15319
    if (desc)
15320
        virBufferEscapeString(&buf, "  <description>%s</description>\n", desc);
15321 15322 15323 15324 15325 15326 15327 15328 15329 15330
    if (vshCommandOptBool(cmd, "diskspec")) {
        virBufferAddLit(&buf, "  <disks>\n");
        while ((opt = vshCommandOptArgv(cmd, opt))) {
            if (vshParseSnapshotDiskspec(ctl, &buf, opt->data) < 0) {
                virBufferFreeAndReset(&buf);
                goto cleanup;
            }
        }
        virBufferAddLit(&buf, "  </disks>\n");
    }
15331 15332 15333 15334 15335 15336 15337 15338
    virBufferAddLit(&buf, "</domainsnapshot>\n");

    buffer = virBufferContentAndReset(&buf);
    if (buffer == NULL) {
        vshError(ctl, "%s", _("Out of memory"));
        goto cleanup;
    }

E
Eric Blake 已提交
15339 15340 15341 15342 15343 15344
    if (vshCommandOptBool(cmd, "print-xml")) {
        vshPrint(ctl, "%s\n",  buffer);
        ret = true;
        goto cleanup;
    }

15345
    ret = vshSnapshotCreate(ctl, dom, buffer, flags, NULL);
15346 15347 15348 15349 15350 15351 15352 15353 15354

cleanup:
    VIR_FREE(buffer);
    if (dom)
        virDomainFree(dom);

    return ret;
}

15355 15356 15357 15358 15359 15360 15361 15362 15363 15364 15365 15366 15367 15368 15369 15370 15371 15372 15373 15374 15375 15376 15377 15378 15379 15380 15381 15382 15383 15384 15385 15386 15387 15388 15389 15390 15391 15392 15393
/* Helper for resolving {--current | --ARG name} into a snapshot
 * belonging to DOM.  If EXCLUSIVE, fail if both --current and arg are
 * present.  On success, populate *SNAP and *NAME, before returning 0.
 * On failure, return -1 after issuing an error message.  */
static int
vshLookupSnapshot(vshControl *ctl, const vshCmd *cmd,
                  const char *arg, bool exclusive, virDomainPtr dom,
                  virDomainSnapshotPtr *snap, const char **name)
{
    bool current = vshCommandOptBool(cmd, "current");
    const char *snapname = NULL;

    if (vshCommandOptString(cmd, arg, &snapname) < 0) {
        vshError(ctl, _("invalid argument for --%s"), arg);
        return -1;
    }

    if (exclusive && current && snapname) {
        vshError(ctl, _("--%s and --current are mutually exclusive"), arg);
        return -1;
    }

    if (snapname) {
        *snap = virDomainSnapshotLookupByName(dom, snapname, 0);
    } else if (current) {
        *snap = virDomainSnapshotCurrent(dom, 0);
    } else {
        vshError(ctl, _("--%s or --current is required"), arg);
        return -1;
    }
    if (!*snap) {
        virshReportError(ctl);
        return -1;
    }

    *name = virDomainSnapshotGetName(*snap);
    return 0;
}

15394 15395 15396 15397 15398 15399 15400 15401 15402 15403 15404
/*
 * "snapshot-edit" command
 */
static const vshCmdInfo info_snapshot_edit[] = {
    {"help", N_("edit XML for a snapshot")},
    {"desc", N_("Edit the domain snapshot XML for a named snapshot")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_snapshot_edit[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
15405
    {"snapshotname", VSH_OT_DATA, 0, N_("snapshot name")},
15406
    {"current", VSH_OT_BOOL, 0, N_("also set edited snapshot as current")},
15407 15408
    {"rename", VSH_OT_BOOL, 0, N_("allow renaming an existing snapshot")},
    {"clone", VSH_OT_BOOL, 0, N_("allow cloning to new name")},
15409 15410 15411 15412 15413 15414 15415 15416
    {NULL, 0, 0, NULL}
};

static bool
cmdSnapshotEdit(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
    virDomainSnapshotPtr snapshot = NULL;
15417
    virDomainSnapshotPtr edited = NULL;
15418
    const char *name;
15419
    const char *edited_name;
15420 15421 15422 15423 15424 15425
    bool ret = false;
    char *tmp = NULL;
    char *doc = NULL;
    char *doc_edited = NULL;
    unsigned int getxml_flags = VIR_DOMAIN_XML_SECURE;
    unsigned int define_flags = VIR_DOMAIN_SNAPSHOT_CREATE_REDEFINE;
15426 15427 15428 15429 15430 15431 15432 15433
    bool rename_okay = vshCommandOptBool(cmd, "rename");
    bool clone_okay = vshCommandOptBool(cmd, "clone");

    if (rename_okay && clone_okay) {
        vshError(ctl, "%s",
                 _("--rename and --clone are mutually exclusive"));
        return false;
    }
15434

15435 15436
    if (vshCommandOptBool(cmd, "current") &&
        vshCommandOptBool(cmd, "snapshotname"))
15437 15438 15439 15440 15441 15442 15443 15444 15445
        define_flags |= VIR_DOMAIN_SNAPSHOT_CREATE_CURRENT;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    dom = vshCommandOptDomain(ctl, cmd, NULL);
    if (dom == NULL)
        goto cleanup;

15446 15447
    if (vshLookupSnapshot(ctl, cmd, "snapshotname", false, dom,
                          &snapshot, &name) < 0)
15448 15449 15450 15451 15452 15453 15454
        goto cleanup;

    /* Get the XML configuration of the snapshot.  */
    doc = virDomainSnapshotGetXMLDesc(snapshot, getxml_flags);
    if (!doc)
        goto cleanup;

15455 15456 15457 15458
    /* strstr is safe here, since xml came from libvirt API and not user */
    if (strstr(doc, "<state>disk-snapshot</state>"))
        define_flags |= VIR_DOMAIN_SNAPSHOT_CREATE_DISK_ONLY;

15459 15460 15461 15462 15463 15464 15465 15466 15467 15468 15469 15470 15471 15472 15473 15474 15475 15476 15477 15478 15479 15480 15481 15482 15483
    /* Create and open the temporary file.  */
    tmp = editWriteToTempFile(ctl, doc);
    if (!tmp)
        goto cleanup;

    /* Start the editor.  */
    if (editFile(ctl, tmp) == -1)
        goto cleanup;

    /* Read back the edited file.  */
    doc_edited = editReadBackFile(ctl, tmp);
    if (!doc_edited)
        goto cleanup;

    /* Compare original XML with edited.  Short-circuit if it did not
     * change, and we do not have any flags.  */
    if (STREQ(doc, doc_edited) &&
        !(define_flags & VIR_DOMAIN_SNAPSHOT_CREATE_CURRENT)) {
        vshPrint(ctl, _("Snapshot %s XML configuration not changed.\n"),
                 name);
        ret = true;
        goto cleanup;
    }

    /* Everything checks out, so redefine the xml.  */
15484 15485
    edited = virDomainSnapshotCreateXML(dom, doc_edited, define_flags);
    if (!edited) {
15486 15487 15488 15489
        vshError(ctl, _("Failed to update %s"), name);
        goto cleanup;
    }

15490 15491 15492 15493 15494 15495 15496 15497 15498 15499 15500 15501 15502 15503 15504 15505 15506 15507 15508 15509 15510 15511 15512 15513
    edited_name = virDomainSnapshotGetName(edited);
    if (STREQ(name, edited_name)) {
        vshPrint(ctl, _("Snapshot %s edited.\n"), name);
    } else if (clone_okay) {
        vshPrint(ctl, _("Snapshot %s cloned to %s.\n"), name,
                 edited_name);
    } else {
        unsigned int delete_flags;

        delete_flags = VIR_DOMAIN_SNAPSHOT_DELETE_METADATA_ONLY;
        if (virDomainSnapshotDelete(rename_okay ? snapshot : edited,
                                    delete_flags) < 0) {
            virshReportError(ctl);
            vshError(ctl, _("Failed to clean up %s"),
                     rename_okay ? name : edited_name);
            goto cleanup;
        }
        if (!rename_okay) {
            vshError(ctl, _("Must use --rename or --clone to change %s to %s"),
                     name, edited_name);
            goto cleanup;
        }
    }

15514 15515 15516 15517 15518 15519 15520 15521 15522 15523 15524
    ret = true;

cleanup:
    VIR_FREE(doc);
    VIR_FREE(doc_edited);
    if (tmp) {
        unlink(tmp);
        VIR_FREE(tmp);
    }
    if (snapshot)
        virDomainSnapshotFree(snapshot);
15525 15526
    if (edited)
        virDomainSnapshotFree(edited);
15527 15528 15529 15530 15531
    if (dom)
        virDomainFree(dom);
    return ret;
}

15532 15533 15534 15535
/*
 * "snapshot-current" command
 */
static const vshCmdInfo info_snapshot_current[] = {
15536 15537
    {"help", N_("Get or set the current snapshot")},
    {"desc", N_("Get or set the current snapshot")},
15538 15539 15540 15541 15542
    {NULL, NULL}
};

static const vshCmdOptDef opts_snapshot_current[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
15543
    {"name", VSH_OT_BOOL, 0, N_("list the name, rather than the full xml")},
15544 15545
    {"security-info", VSH_OT_BOOL, 0,
     N_("include security sensitive information in XML dump")},
15546 15547
    {"snapshotname", VSH_OT_DATA, 0,
     N_("name of existing snapshot to make current")},
15548 15549 15550
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
15551
static bool
15552 15553 15554
cmdSnapshotCurrent(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
E
Eric Blake 已提交
15555
    bool ret = false;
15556 15557
    int current;
    virDomainSnapshotPtr snapshot = NULL;
15558
    char *xml = NULL;
15559
    const char *snapshotname = NULL;
15560
    unsigned int flags = 0;
15561
    const char *domname;
15562 15563 15564

    if (vshCommandOptBool(cmd, "security-info"))
        flags |= VIR_DOMAIN_XML_SECURE;
15565

15566
    if (!vshConnectionUsability(ctl, ctl->conn))
15567 15568
        goto cleanup;

15569
    dom = vshCommandOptDomain(ctl, cmd, &domname);
15570 15571 15572
    if (dom == NULL)
        goto cleanup;

15573 15574 15575 15576 15577 15578 15579 15580 15581 15582 15583 15584 15585 15586 15587 15588 15589 15590 15591 15592
    if (vshCommandOptString(cmd, "snapshotname", &snapshotname) < 0) {
        vshError(ctl, _("invalid snapshotname argument '%s'"), snapshotname);
        goto cleanup;
    }
    if (snapshotname) {
        virDomainSnapshotPtr snapshot2 = NULL;
        flags = (VIR_DOMAIN_SNAPSHOT_CREATE_REDEFINE |
                 VIR_DOMAIN_SNAPSHOT_CREATE_CURRENT);

        if (vshCommandOptBool(cmd, "name")) {
            vshError(ctl, "%s",
                     _("--name and snapshotname are mutually exclusive"));
            goto cleanup;
        }
        snapshot = virDomainSnapshotLookupByName(dom, snapshotname, 0);
        if (snapshot == NULL)
            goto cleanup;
        xml = virDomainSnapshotGetXMLDesc(snapshot, VIR_DOMAIN_XML_SECURE);
        if (!xml)
            goto cleanup;
15593 15594 15595
        /* strstr is safe here, since xml came from libvirt API and not user */
        if (strstr(xml, "<state>disk-snapshot</state>"))
            flags |= VIR_DOMAIN_SNAPSHOT_CREATE_DISK_ONLY;
15596 15597 15598 15599 15600 15601 15602 15603 15604
        snapshot2 = virDomainSnapshotCreateXML(dom, xml, flags);
        if (snapshot2 == NULL)
            goto cleanup;
        virDomainSnapshotFree(snapshot2);
        vshPrint(ctl, _("Snapshot %s set as current"), snapshotname);
        ret = true;
        goto cleanup;
    }

15605
    current = virDomainHasCurrentSnapshot(dom, 0);
15606 15607 15608 15609
    if (current < 0) {
        goto cleanup;
    } else if (!current) {
        vshError(ctl, _("domain '%s' has no current snapshot"), domname);
15610
        goto cleanup;
15611
    } else {
E
Eric Blake 已提交
15612
        const char *name = NULL;
15613 15614 15615 15616

        if (!(snapshot = virDomainSnapshotCurrent(dom, 0)))
            goto cleanup;

15617
        if (vshCommandOptBool(cmd, "name")) {
E
Eric Blake 已提交
15618
            name = virDomainSnapshotGetName(snapshot);
15619 15620
            if (!name)
                goto cleanup;
E
Eric Blake 已提交
15621 15622 15623 15624
        } else {
            xml = virDomainSnapshotGetXMLDesc(snapshot, flags);
            if (!xml)
                goto cleanup;
15625 15626 15627
        }

        vshPrint(ctl, "%s", name ? name : xml);
15628 15629
    }

E
Eric Blake 已提交
15630
    ret = true;
15631 15632

cleanup:
15633 15634
    if (!ret)
        virshReportError(ctl);
15635
    VIR_FREE(xml);
15636 15637 15638 15639 15640 15641 15642 15643
    if (snapshot)
        virDomainSnapshotFree(snapshot);
    if (dom)
        virDomainFree(dom);

    return ret;
}

15644
/* Helper function to get the name of a snapshot's parent.  Caller
15645 15646 15647 15648 15649 15650
 * must free the result.  Returns 0 on success (including when it was
 * proven no parent exists), and -1 on failure with error reported
 * (such as no snapshot support or domain deleted in meantime).  */
static int
vshGetSnapshotParent(vshControl *ctl, virDomainSnapshotPtr snapshot,
                     char **parent_name)
15651 15652 15653 15654 15655
{
    virDomainSnapshotPtr parent = NULL;
    char *xml = NULL;
    xmlDocPtr xmldoc = NULL;
    xmlXPathContextPtr ctxt = NULL;
15656 15657 15658
    int ret = -1;

    *parent_name = NULL;
15659 15660

    /* Try new API, since it is faster. */
15661
    if (!ctl->useSnapshotOld) {
15662 15663 15664
        parent = virDomainSnapshotGetParent(snapshot, 0);
        if (parent) {
            /* API works, and virDomainSnapshotGetName will succeed */
15665 15666
            *parent_name = vshStrdup(ctl, virDomainSnapshotGetName(parent));
            ret = 0;
15667 15668 15669 15670
            goto cleanup;
        }
        if (last_error->code == VIR_ERR_NO_DOMAIN_SNAPSHOT) {
            /* API works, and we found a root with no parent */
15671
            ret = 0;
15672 15673 15674
            goto cleanup;
        }
        /* API didn't work, fall back to XML scraping. */
15675
        ctl->useSnapshotOld = true;
15676 15677 15678 15679 15680 15681 15682 15683 15684 15685
    }

    xml = virDomainSnapshotGetXMLDesc(snapshot, 0);
    if (!xml)
        goto cleanup;

    xmldoc = virXMLParseStringCtxt(xml, _("(domain_snapshot)"), &ctxt);
    if (!xmldoc)
        goto cleanup;

15686 15687
    *parent_name = virXPathString("string(/domainsnapshot/parent/name)", ctxt);
    ret = 0;
15688 15689

cleanup:
15690 15691 15692 15693 15694 15695 15696
    if (ret < 0) {
        virshReportError(ctl);
        vshError(ctl, "%s", _("unable to determine if snapshot has parent"));
    } else {
        virFreeError(last_error);
        last_error = NULL;
    }
15697 15698 15699 15700 15701
    if (parent)
        virDomainSnapshotFree(parent);
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(xmldoc);
    VIR_FREE(xml);
15702
    return ret;
15703 15704
}

15705 15706 15707 15708 15709 15710 15711 15712 15713 15714 15715
/*
 * "snapshot-list" command
 */
static const vshCmdInfo info_snapshot_list[] = {
    {"help", N_("List snapshots for a domain")},
    {"desc", N_("Snapshot List")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_snapshot_list[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
15716
    {"parent", VSH_OT_BOOL, 0, N_("add a column showing parent snapshot")},
15717
    {"roots", VSH_OT_BOOL, 0, N_("list only snapshots without parents")},
15718
    {"leaves", VSH_OT_BOOL, 0, N_("list only snapshots without children")},
15719 15720
    {"metadata", VSH_OT_BOOL, 0,
     N_("list only snapshots that have metadata that would prevent undefine")},
15721
    {"tree", VSH_OT_BOOL, 0, N_("list snapshots in a tree")},
15722
    {"from", VSH_OT_DATA, 0, N_("limit list to children of given snapshot")},
15723 15724
    {"current", VSH_OT_BOOL, 0,
     N_("limit list to children of current snapshot")},
15725
    {"descendants", VSH_OT_BOOL, 0, N_("with --from, list all descendants")},
15726 15727 15728
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
15729
static bool
15730 15731 15732
cmdSnapshotList(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
E
Eric Blake 已提交
15733
    bool ret = false;
15734
    unsigned int flags = 0;
15735 15736
    int parent_filter = 0; /* -1 for roots filtering, 0 for no parent
                              information needed, 1 for parent column */
15737 15738
    int numsnaps;
    char **names = NULL;
15739
    char **parents = NULL;
15740
    int actual = 0;
15741 15742 15743 15744 15745 15746
    int i;
    xmlDocPtr xml = NULL;
    xmlXPathContextPtr ctxt = NULL;
    char *doc = NULL;
    virDomainSnapshotPtr snapshot = NULL;
    char *state = NULL;
15747
    char *parent = NULL;
15748 15749
    long long creation_longlong;
    time_t creation_time_t;
15750 15751
    char timestr[100];
    struct tm time_info;
15752
    bool tree = vshCommandOptBool(cmd, "tree");
15753
    bool leaves = vshCommandOptBool(cmd, "leaves");
15754 15755
    const char *from = NULL;
    virDomainSnapshotPtr start = NULL;
15756
    int start_index = -1;
15757
    bool descendants = false;
15758

15759 15760 15761 15762 15763 15764 15765 15766 15767 15768
    if (!vshConnectionUsability(ctl, ctl->conn))
        goto cleanup;

    dom = vshCommandOptDomain(ctl, cmd, NULL);
    if (dom == NULL)
        goto cleanup;

    if ((vshCommandOptBool(cmd, "from") ||
         vshCommandOptBool(cmd, "current")) &&
        vshLookupSnapshot(ctl, cmd, "from", true, dom, &start, &from) < 0)
15769
        goto cleanup;
15770

15771
    if (vshCommandOptBool(cmd, "parent")) {
15772 15773
        if (vshCommandOptBool(cmd, "roots")) {
            vshError(ctl, "%s",
15774
                     _("--parent and --roots are mutually exclusive"));
15775 15776
            return false;
        }
15777 15778
        if (tree) {
            vshError(ctl, "%s",
15779
                     _("--parent and --tree are mutually exclusive"));
15780 15781
            return false;
        }
15782
        parent_filter = 1;
15783
    } else if (vshCommandOptBool(cmd, "roots")) {
15784 15785
        if (tree) {
            vshError(ctl, "%s",
15786
                     _("--roots and --tree are mutually exclusive"));
15787 15788
            return false;
        }
15789 15790 15791 15792
        if (from) {
            vshError(ctl, "%s",
                     _("--roots and --from are mutually exclusive"));
        }
15793 15794
        flags |= VIR_DOMAIN_SNAPSHOT_LIST_ROOTS;
    }
15795 15796 15797 15798 15799 15800 15801 15802
    if (leaves) {
        if (tree) {
            vshError(ctl, "%s",
                     _("--leaves and --tree are mutually exclusive"));
            return false;
        }
        flags |= VIR_DOMAIN_SNAPSHOT_LIST_LEAVES;
    }
15803 15804 15805

    if (vshCommandOptBool(cmd, "metadata")) {
        flags |= VIR_DOMAIN_SNAPSHOT_LIST_METADATA;
15806 15807
    }

15808
    if (from) {
15809 15810
        descendants = vshCommandOptBool(cmd, "descendants");
        if (descendants || tree) {
15811 15812
            flags |= VIR_DOMAIN_SNAPSHOT_LIST_DESCENDANTS;
        }
15813 15814
        numsnaps = ctl->useSnapshotOld ? -1 :
            virDomainSnapshotNumChildren(start, flags);
15815
        if (numsnaps < 0) {
15816 15817
            if (ctl->useSnapshotOld ||
                last_error->code == VIR_ERR_NO_SUPPORT) {
15818
                /* We can emulate --from.  */
15819
                /* XXX can we also emulate --leaves? */
15820 15821 15822 15823 15824 15825
                virFreeError(last_error);
                last_error = NULL;
                ctl->useSnapshotOld = true;
                flags &= ~VIR_DOMAIN_SNAPSHOT_LIST_DESCENDANTS;
                numsnaps = virDomainSnapshotNum(dom, flags);
            }
15826 15827
        } else if (tree) {
            numsnaps++;
15828
        }
15829
    } else {
15830
        numsnaps = virDomainSnapshotNum(dom, flags);
15831 15832

        /* Fall back to simulation if --roots was unsupported. */
15833
        /* XXX can we also emulate --leaves? */
15834 15835 15836 15837 15838 15839 15840 15841
        if (numsnaps < 0 && last_error->code == VIR_ERR_INVALID_ARG &&
            (flags & VIR_DOMAIN_SNAPSHOT_LIST_ROOTS)) {
            virFreeError(last_error);
            last_error = NULL;
            parent_filter = -1;
            flags &= ~VIR_DOMAIN_SNAPSHOT_LIST_ROOTS;
            numsnaps = virDomainSnapshotNum(dom, flags);
        }
15842 15843
    }

15844 15845 15846
    if (numsnaps < 0) {
        if (!last_error)
            vshError(ctl, _("missing support"));
15847
        goto cleanup;
15848
    }
15849

15850 15851 15852 15853 15854 15855 15856 15857 15858
    if (!tree) {
        if (parent_filter > 0)
            vshPrintExtra(ctl, " %-20s %-25s %-15s %s",
                          _("Name"), _("Creation Time"), _("State"),
                          _("Parent"));
        else
            vshPrintExtra(ctl, " %-20s %-25s %s",
                          _("Name"), _("Creation Time"), _("State"));
        vshPrintExtra(ctl, "\n\
15859
------------------------------------------------------------\n");
15860
    }
15861

15862 15863 15864 15865
    if (!numsnaps) {
        ret = true;
        goto cleanup;
    }
15866

15867 15868 15869
    if (VIR_ALLOC_N(names, numsnaps) < 0)
        goto cleanup;

15870
    if (from && !ctl->useSnapshotOld) {
15871 15872 15873 15874 15875 15876 15877 15878 15879 15880 15881 15882 15883 15884 15885 15886 15887 15888
        /* When mixing --from and --tree, we want to start the tree at the
         * given snapshot.  Without --tree, only list the children.  */
        if (tree) {
            if (numsnaps)
                actual = virDomainSnapshotListChildrenNames(start, names + 1,
                                                            numsnaps - 1,
                                                            flags);
            if (actual >= 0) {
                actual++;
                names[0] = vshStrdup(ctl, from);
            }
        } else {
            actual = virDomainSnapshotListChildrenNames(start, names,
                                                        numsnaps, flags);
        }
    } else {
        actual = virDomainSnapshotListNames(dom, names, numsnaps, flags);
    }
15889 15890
    if (actual < 0)
        goto cleanup;
15891

E
Eric Blake 已提交
15892 15893
    qsort(&names[0], actual, sizeof(char*), namesorter);

15894 15895
    if (tree || ctl->useSnapshotOld) {
        parents = vshCalloc(ctl, sizeof(char *), actual);
15896
        for (i = (from && !ctl->useSnapshotOld); i < actual; i++) {
15897 15898 15899 15900 15901
            if (ctl->useSnapshotOld && STREQ(names[i], from)) {
                start_index = i;
                continue;
            }

15902 15903 15904 15905
            /* free up memory from previous iterations of the loop */
            if (snapshot)
                virDomainSnapshotFree(snapshot);
            snapshot = virDomainSnapshotLookupByName(dom, names[i], 0);
15906 15907
            if (!snapshot ||
                vshGetSnapshotParent(ctl, snapshot, &parents[i]) < 0) {
15908 15909 15910 15911 15912 15913
                while (--i >= 0)
                    VIR_FREE(parents[i]);
                VIR_FREE(parents);
                goto cleanup;
            }
        }
15914 15915 15916
    }
    if (tree) {
        char indentBuf[INDENT_BUFLEN];
15917 15918
        for (i = 0 ; i < actual ; i++) {
            memset(indentBuf, '\0', sizeof indentBuf);
15919
            if (ctl->useSnapshotOld ? STREQ(names[i], from) : !parents[i])
15920 15921 15922 15923 15924 15925 15926 15927 15928 15929 15930 15931 15932 15933
                cmdNodeListDevicesPrint(ctl,
                                        names,
                                        parents,
                                        actual,
                                        i,
                                        i,
                                        0,
                                        0,
                                        indentBuf);
        }

        ret = true;
        goto cleanup;
    } else {
15934
        if (ctl->useSnapshotOld && descendants) {
15935 15936 15937 15938 15939 15940 15941 15942 15943 15944 15945 15946 15947 15948 15949 15950 15951 15952 15953 15954 15955 15956 15957 15958 15959 15960 15961 15962 15963 15964 15965 15966 15967 15968 15969 15970 15971 15972 15973 15974 15975 15976 15977 15978 15979 15980 15981 15982 15983 15984
            bool changed = false;

            /* Make multiple passes over the list - first pass NULLs
             * out all roots except start, remaining passes NULL out
             * any entry whose parent is not still in list.  Also, we
             * NULL out parent when name is known to be in list.
             * Sorry, this is O(n^3) - hope your hierarchy isn't huge.  */
            if (start_index < 0) {
                vshError(ctl, _("snapshot %s disappeared from list"), from);
                goto cleanup;
            }
            for (i = 0; i < actual; i++) {
                if (i == start_index)
                    continue;
                if (!parents[i]) {
                    VIR_FREE(names[i]);
                } else if (STREQ(parents[i], from)) {
                    VIR_FREE(parents[i]);
                    changed = true;
                }
            }
            if (!changed) {
                ret = true;
                goto cleanup;
            }
            while (changed) {
                changed = false;
                for (i = 0; i < actual; i++) {
                    bool found = false;
                    int j;

                    if (!names[i] || !parents[i])
                        continue;
                    for (j = 0; j < actual; j++) {
                        if (!names[j] || i == j)
                            continue;
                        if (STREQ(parents[i], names[j])) {
                            found = true;
                            if (!parents[j])
                                VIR_FREE(parents[i]);
                            break;
                        }
                    }
                    if (!found) {
                        changed = true;
                        VIR_FREE(names[i]);
                    }
                }
            }
            VIR_FREE(names[start_index]);
15985 15986
        }

15987
        for (i = 0; i < actual; i++) {
15988 15989
            if (ctl->useSnapshotOld &&
                (descendants ? !names[i] : STRNEQ_NULLABLE(parents[i], from)))
15990 15991
                continue;

15992
            /* free up memory from previous iterations of the loop */
15993
            VIR_FREE(parent);
15994 15995 15996 15997
            VIR_FREE(state);
            if (snapshot)
                virDomainSnapshotFree(snapshot);
            xmlXPathFreeContext(ctxt);
15998
            xmlFreeDoc(xml);
15999 16000 16001 16002 16003 16004 16005 16006 16007 16008
            VIR_FREE(doc);

            snapshot = virDomainSnapshotLookupByName(dom, names[i], 0);
            if (snapshot == NULL)
                continue;

            doc = virDomainSnapshotGetXMLDesc(snapshot, 0);
            if (!doc)
                continue;

16009
            xml = virXMLParseStringCtxt(doc, _("(domain_snapshot)"), &ctxt);
16010 16011 16012
            if (!xml)
                continue;

16013 16014 16015
            if (parent_filter) {
                parent = virXPathString("string(/domainsnapshot/parent/name)",
                                        ctxt);
16016 16017
                if (!parent && parent_filter < 0)
                    continue;
16018 16019
            }

16020 16021 16022
            state = virXPathString("string(/domainsnapshot/state)", ctxt);
            if (state == NULL)
                continue;
16023 16024
            if (virXPathLongLong("string(/domainsnapshot/creationTime)", ctxt,
                                 &creation_longlong) < 0)
16025
                continue;
16026 16027 16028 16029 16030 16031
            creation_time_t = creation_longlong;
            if (creation_time_t != creation_longlong) {
                vshError(ctl, "%s", _("time_t overflow"));
                continue;
            }
            localtime_r(&creation_time_t, &time_info);
16032 16033
            strftime(timestr, sizeof(timestr), "%Y-%m-%d %H:%M:%S %z",
                     &time_info);
16034

16035 16036 16037 16038 16039
            if (parent)
                vshPrint(ctl, " %-20s %-25s %-15s %s\n",
                         names[i], timestr, state, parent);
            else
                vshPrint(ctl, " %-20s %-25s %s\n", names[i], timestr, state);
16040 16041 16042
        }
    }

E
Eric Blake 已提交
16043
    ret = true;
16044 16045 16046

cleanup:
    /* this frees up memory from the last iteration of the loop */
16047
    VIR_FREE(parent);
16048 16049 16050
    VIR_FREE(state);
    if (snapshot)
        virDomainSnapshotFree(snapshot);
16051 16052
    if (start)
        virDomainSnapshotFree(start);
16053
    xmlXPathFreeContext(ctxt);
16054
    xmlFreeDoc(xml);
16055
    VIR_FREE(doc);
16056
    for (i = 0; i < actual; i++) {
16057
        VIR_FREE(names[i]);
16058 16059 16060
        if (parents)
            VIR_FREE(parents[i]);
    }
16061
    VIR_FREE(names);
16062
    VIR_FREE(parents);
16063 16064 16065 16066 16067 16068 16069 16070 16071 16072 16073 16074 16075 16076 16077 16078 16079 16080
    if (dom)
        virDomainFree(dom);

    return ret;
}

/*
 * "snapshot-dumpxml" command
 */
static const vshCmdInfo info_snapshot_dumpxml[] = {
    {"help", N_("Dump XML for a domain snapshot")},
    {"desc", N_("Snapshot Dump XML")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_snapshot_dumpxml[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
    {"snapshotname", VSH_OT_DATA, VSH_OFLAG_REQ, N_("snapshot name")},
16081 16082
    {"security-info", VSH_OT_BOOL, 0,
     N_("include security sensitive information in XML dump")},
16083 16084 16085
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
16086
static bool
16087 16088 16089
cmdSnapshotDumpXML(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
E
Eric Blake 已提交
16090
    bool ret = false;
16091
    const char *name = NULL;
16092 16093
    virDomainSnapshotPtr snapshot = NULL;
    char *xml = NULL;
16094 16095 16096 16097
    unsigned int flags = 0;

    if (vshCommandOptBool(cmd, "security-info"))
        flags |= VIR_DOMAIN_XML_SECURE;
16098

16099
    if (!vshConnectionUsability(ctl, ctl->conn))
16100 16101 16102 16103 16104 16105
        goto cleanup;

    dom = vshCommandOptDomain(ctl, cmd, NULL);
    if (dom == NULL)
        goto cleanup;

16106
    if (vshCommandOptString(cmd, "snapshotname", &name) <= 0)
16107 16108 16109 16110 16111 16112
        goto cleanup;

    snapshot = virDomainSnapshotLookupByName(dom, name, 0);
    if (snapshot == NULL)
        goto cleanup;

16113
    xml = virDomainSnapshotGetXMLDesc(snapshot, flags);
16114 16115 16116
    if (!xml)
        goto cleanup;

16117
    vshPrint(ctl, "%s", xml);
16118

E
Eric Blake 已提交
16119
    ret = true;
16120 16121 16122 16123 16124 16125 16126 16127 16128 16129 16130

cleanup:
    VIR_FREE(xml);
    if (snapshot)
        virDomainSnapshotFree(snapshot);
    if (dom)
        virDomainFree(dom);

    return ret;
}

E
Eric Blake 已提交
16131 16132 16133 16134
/*
 * "snapshot-parent" command
 */
static const vshCmdInfo info_snapshot_parent[] = {
E
Eric Blake 已提交
16135
    {"help", N_("Get the name of the parent of a snapshot")},
E
Eric Blake 已提交
16136 16137 16138 16139 16140 16141
    {"desc", N_("Extract the snapshot's parent, if any")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_snapshot_parent[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
16142 16143
    {"snapshotname", VSH_OT_DATA, 0, N_("find parent of snapshot name")},
    {"current", VSH_OT_BOOL, 0, N_("find parent of current snapshot")},
E
Eric Blake 已提交
16144 16145 16146 16147 16148 16149 16150 16151 16152 16153 16154 16155 16156 16157 16158 16159 16160 16161 16162
    {NULL, 0, 0, NULL}
};

static bool
cmdSnapshotParent(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
    bool ret = false;
    const char *name = NULL;
    virDomainSnapshotPtr snapshot = NULL;
    char *parent = NULL;

    if (!vshConnectionUsability(ctl, ctl->conn))
        goto cleanup;

    dom = vshCommandOptDomain(ctl, cmd, NULL);
    if (dom == NULL)
        goto cleanup;

16163 16164
    if (vshLookupSnapshot(ctl, cmd, "snapshotname", true, dom,
                          &snapshot, &name) < 0)
E
Eric Blake 已提交
16165 16166
        goto cleanup;

16167
    if (vshGetSnapshotParent(ctl, snapshot, &parent) < 0)
E
Eric Blake 已提交
16168
        goto cleanup;
16169 16170 16171 16172
    if (!parent) {
        vshError(ctl, _("snapshot '%s' has no parent"), name);
        goto cleanup;
    }
E
Eric Blake 已提交
16173 16174 16175 16176 16177 16178 16179 16180 16181 16182 16183 16184 16185 16186 16187

    vshPrint(ctl, "%s", parent);

    ret = true;

cleanup:
    VIR_FREE(parent);
    if (snapshot)
        virDomainSnapshotFree(snapshot);
    if (dom)
        virDomainFree(dom);

    return ret;
}

16188
/*
16189
 * "snapshot-revert" command
16190
 */
16191
static const vshCmdInfo info_snapshot_revert[] = {
16192 16193 16194 16195 16196
    {"help", N_("Revert a domain to a snapshot")},
    {"desc", N_("Revert domain to snapshot")},
    {NULL, NULL}
};

16197
static const vshCmdOptDef opts_snapshot_revert[] = {
16198
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
16199 16200
    {"snapshotname", VSH_OT_DATA, 0, N_("snapshot name")},
    {"current", VSH_OT_BOOL, 0, N_("revert to current snapshot")},
16201 16202
    {"running", VSH_OT_BOOL, 0, N_("after reverting, change state to running")},
    {"paused", VSH_OT_BOOL, 0, N_("after reverting, change state to paused")},
E
Eric Blake 已提交
16203
    {"force", VSH_OT_BOOL, 0, N_("try harder on risky reverts")},
16204 16205 16206
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
16207
static bool
16208
cmdDomainSnapshotRevert(vshControl *ctl, const vshCmd *cmd)
16209 16210
{
    virDomainPtr dom = NULL;
E
Eric Blake 已提交
16211
    bool ret = false;
16212
    const char *name = NULL;
16213
    virDomainSnapshotPtr snapshot = NULL;
16214
    unsigned int flags = 0;
E
Eric Blake 已提交
16215 16216
    bool force = false;
    int result;
16217 16218 16219 16220 16221

    if (vshCommandOptBool(cmd, "running"))
        flags |= VIR_DOMAIN_SNAPSHOT_REVERT_RUNNING;
    if (vshCommandOptBool(cmd, "paused"))
        flags |= VIR_DOMAIN_SNAPSHOT_REVERT_PAUSED;
E
Eric Blake 已提交
16222 16223 16224 16225 16226 16227
    /* We want virsh snapshot-revert --force to work even when talking
     * to older servers that did the unsafe revert by default but
     * reject the flag, so we probe without the flag, and only use it
     * when the error says it will make a difference.  */
    if (vshCommandOptBool(cmd, "force"))
        force = true;
16228

16229
    if (!vshConnectionUsability(ctl, ctl->conn))
16230 16231 16232 16233 16234 16235
        goto cleanup;

    dom = vshCommandOptDomain(ctl, cmd, NULL);
    if (dom == NULL)
        goto cleanup;

16236 16237
    if (vshLookupSnapshot(ctl, cmd, "snapshotname", true, dom,
                          &snapshot, &name) < 0)
16238 16239
        goto cleanup;

E
Eric Blake 已提交
16240 16241 16242 16243 16244 16245 16246 16247 16248
    result = virDomainRevertToSnapshot(snapshot, flags);
    if (result < 0 && force &&
        last_error->code == VIR_ERR_SNAPSHOT_REVERT_RISKY) {
        flags |= VIR_DOMAIN_SNAPSHOT_REVERT_FORCE;
        virFreeError(last_error);
        last_error = NULL;
        result = virDomainRevertToSnapshot(snapshot, flags);
    }
    if (result < 0)
16249 16250
        goto cleanup;

E
Eric Blake 已提交
16251
    ret = true;
16252 16253 16254 16255 16256 16257 16258 16259 16260 16261 16262 16263 16264 16265 16266 16267 16268 16269 16270 16271 16272

cleanup:
    if (snapshot)
        virDomainSnapshotFree(snapshot);
    if (dom)
        virDomainFree(dom);

    return ret;
}

/*
 * "snapshot-delete" command
 */
static const vshCmdInfo info_snapshot_delete[] = {
    {"help", N_("Delete a domain snapshot")},
    {"desc", N_("Snapshot Delete")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_snapshot_delete[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
16273 16274
    {"snapshotname", VSH_OT_DATA, 0, N_("snapshot name")},
    {"current", VSH_OT_BOOL, 0, N_("delete current snapshot")},
16275
    {"children", VSH_OT_BOOL, 0, N_("delete snapshot and all children")},
16276 16277 16278
    {"children-only", VSH_OT_BOOL, 0, N_("delete children but not snapshot")},
    {"metadata", VSH_OT_BOOL, 0,
     N_("delete only libvirt metadata, leaving snapshot contents behind")},
16279 16280 16281
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
16282
static bool
16283 16284 16285
cmdSnapshotDelete(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
E
Eric Blake 已提交
16286
    bool ret = false;
16287
    const char *name = NULL;
16288 16289 16290
    virDomainSnapshotPtr snapshot = NULL;
    unsigned int flags = 0;

16291
    if (!vshConnectionUsability(ctl, ctl->conn))
16292 16293 16294 16295 16296 16297
        goto cleanup;

    dom = vshCommandOptDomain(ctl, cmd, NULL);
    if (dom == NULL)
        goto cleanup;

16298 16299
    if (vshLookupSnapshot(ctl, cmd, "snapshotname", true, dom,
                          &snapshot, &name) < 0)
16300 16301 16302 16303
        goto cleanup;

    if (vshCommandOptBool(cmd, "children"))
        flags |= VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN;
16304 16305 16306 16307
    if (vshCommandOptBool(cmd, "children-only"))
        flags |= VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN_ONLY;
    if (vshCommandOptBool(cmd, "metadata"))
        flags |= VIR_DOMAIN_SNAPSHOT_DELETE_METADATA_ONLY;
16308

16309 16310 16311
    /* XXX If we wanted, we could emulate DELETE_CHILDREN_ONLY even on
     * older servers that reject the flag, by manually computing the
     * list of descendants.  But that's a lot of code to maintain.  */
16312
    if (virDomainSnapshotDelete(snapshot, flags) == 0) {
16313 16314 16315 16316
        if (flags & VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN_ONLY)
            vshPrint(ctl, _("Domain snapshot %s children deleted\n"), name);
        else
            vshPrint(ctl, _("Domain snapshot %s deleted\n"), name);
16317 16318
    } else {
        vshError(ctl, _("Failed to delete snapshot %s"), name);
16319
        goto cleanup;
16320
    }
16321

E
Eric Blake 已提交
16322
    ret = true;
16323 16324 16325 16326 16327 16328 16329 16330 16331 16332

cleanup:
    if (snapshot)
        virDomainSnapshotFree(snapshot);
    if (dom)
        virDomainFree(dom);

    return ret;
}

16333 16334 16335 16336 16337 16338 16339 16340 16341 16342 16343 16344 16345 16346 16347 16348 16349 16350 16351 16352 16353 16354 16355 16356 16357 16358 16359 16360 16361 16362 16363 16364 16365 16366 16367 16368 16369 16370 16371 16372 16373 16374 16375 16376 16377 16378 16379 16380 16381 16382 16383 16384 16385 16386 16387 16388 16389 16390 16391 16392 16393 16394 16395 16396 16397 16398 16399 16400 16401 16402 16403 16404 16405 16406 16407 16408 16409
/*
 * "domblkerror" command
 */
static const char *
vshDomainIOErrorToString(int error)
{
    switch ((virDomainDiskErrorCode) error) {
    case VIR_DOMAIN_DISK_ERROR_NONE:
        return _("no error");
    case VIR_DOMAIN_DISK_ERROR_UNSPEC:
        return _("unspecified error");
    case VIR_DOMAIN_DISK_ERROR_NO_SPACE:
        return _("no space");
    case VIR_DOMAIN_DISK_ERROR_LAST:
        ;
    }

    return _("unknown error");
}

static const vshCmdInfo info_domblkerror[] = {
    {"help", N_("Show errors on block devices")},
    {"desc", N_("Show block device errors")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_domblkerror[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id, or uuid")},
    {NULL, 0, 0, NULL}
};

static bool
cmdDomBlkError(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom;
    virDomainDiskErrorPtr disks = NULL;
    unsigned int ndisks;
    int i;
    int count;
    bool ret = false;

    if (!vshConnectionUsability(ctl, ctl->conn))
        return false;

    if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
        return false;

    if ((count = virDomainGetDiskErrors(dom, NULL, 0, 0)) < 0)
        goto cleanup;
    ndisks = count;

    if (ndisks) {
        if (VIR_ALLOC_N(disks, ndisks) < 0)
            goto cleanup;

        if ((count = virDomainGetDiskErrors(dom, disks, ndisks, 0)) == -1)
            goto cleanup;
    }

    if (count == 0) {
        vshPrint(ctl, _("No errors found\n"));
    } else {
        for (i = 0; i < count; i++) {
            vshPrint(ctl, "%s: %s\n",
                     disks[i].disk,
                     vshDomainIOErrorToString(disks[i].error));
        }
    }

    ret = true;

cleanup:
    VIR_FREE(disks);
    virDomainFree(dom);
    return ret;
}

16410 16411 16412 16413
/*
 * "qemu-monitor-command" command
 */
static const vshCmdInfo info_qemu_monitor_command[] = {
16414 16415
    {"help", N_("QEMU Monitor Command")},
    {"desc", N_("QEMU Monitor Command")},
16416 16417 16418 16419 16420
    {NULL, NULL}
};

static const vshCmdOptDef opts_qemu_monitor_command[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or uuid")},
16421
    {"hmp", VSH_OT_BOOL, 0, N_("command is in human monitor protocol")},
16422
    {"cmd", VSH_OT_ARGV, VSH_OFLAG_REQ, N_("command")},
16423 16424 16425
    {NULL, 0, 0, NULL}
};

E
Eric Blake 已提交
16426
static bool
16427 16428 16429
cmdQemuMonitorCommand(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
E
Eric Blake 已提交
16430
    bool ret = false;
16431
    char *monitor_cmd = NULL;
16432
    char *result = NULL;
16433
    unsigned int flags = 0;
16434 16435 16436
    const vshCmdOpt *opt = NULL;
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    bool pad = false;
16437 16438 16439 16440 16441 16442 16443 16444

    if (!vshConnectionUsability(ctl, ctl->conn))
        goto cleanup;

    dom = vshCommandOptDomain(ctl, cmd, NULL);
    if (dom == NULL)
        goto cleanup;

16445 16446 16447 16448 16449 16450 16451 16452
    while ((opt = vshCommandOptArgv(cmd, opt))) {
        if (pad)
            virBufferAddChar(&buf, ' ');
        pad = true;
        virBufferAdd(&buf, opt->data, -1);
    }
    if (virBufferError(&buf)) {
        vshPrint(ctl, "%s", _("Failed to collect command"));
16453 16454
        goto cleanup;
    }
16455
    monitor_cmd = virBufferContentAndReset(&buf);
16456

16457 16458 16459 16460
    if (vshCommandOptBool(cmd, "hmp"))
        flags |= VIR_DOMAIN_QEMU_MONITOR_COMMAND_HMP;

    if (virDomainQemuMonitorCommand(dom, monitor_cmd, &result, flags) < 0)
16461 16462 16463 16464
        goto cleanup;

    printf("%s\n", result);

E
Eric Blake 已提交
16465
    ret = true;
16466 16467 16468

cleanup:
    VIR_FREE(result);
16469
    VIR_FREE(monitor_cmd);
16470 16471 16472 16473 16474 16475
    if (dom)
        virDomainFree(dom);

    return ret;
}

16476 16477 16478 16479 16480 16481 16482 16483 16484 16485 16486 16487 16488 16489 16490 16491 16492 16493 16494 16495 16496 16497 16498 16499 16500 16501 16502 16503 16504 16505 16506 16507 16508 16509 16510 16511 16512 16513 16514 16515 16516 16517 16518 16519 16520 16521
/*
 * "qemu-attach" command
 */
static const vshCmdInfo info_qemu_attach[] = {
    {"help", N_("QEMU Attach")},
    {"desc", N_("QEMU Attach")},
    {NULL, NULL}
};

static const vshCmdOptDef opts_qemu_attach[] = {
    {"pid", VSH_OT_DATA, VSH_OFLAG_REQ, N_("pid")},
    {NULL, 0, 0, NULL}
};

static bool
cmdQemuAttach(vshControl *ctl, const vshCmd *cmd)
{
    virDomainPtr dom = NULL;
    bool ret = false;
    unsigned int flags = 0;
    unsigned int pid;

    if (!vshConnectionUsability(ctl, ctl->conn))
        goto cleanup;

    if (vshCommandOptUInt(cmd, "pid", &pid) <= 0) {
        vshError(ctl, "%s", _("missing pid value"));
        goto cleanup;
    }

    if (!(dom = virDomainQemuAttach(ctl->conn, pid, flags)))
        goto cleanup;

    if (dom != NULL) {
        vshPrint(ctl, _("Domain %s attached to pid %u\n"),
                 virDomainGetName(dom), pid);
        virDomainFree(dom);
        ret = true;
    } else {
        vshError(ctl, _("Failed to attach to pid %u"), pid);
    }

cleanup:
    return ret;
}

16522
static const vshCmdDef domManagementCmds[] = {
16523 16524 16525 16526 16527 16528 16529
    {"attach-device", cmdAttachDevice, opts_attach_device,
     info_attach_device, 0},
    {"attach-disk", cmdAttachDisk, opts_attach_disk,
     info_attach_disk, 0},
    {"attach-interface", cmdAttachInterface, opts_attach_interface,
     info_attach_interface, 0},
    {"autostart", cmdAutostart, opts_autostart, info_autostart, 0},
L
Lei Li 已提交
16530
    {"blkdeviotune", cmdBlkdeviotune, opts_blkdeviotune, info_blkdeviotune, 0},
16531
    {"blkiotune", cmdBlkiotune, opts_blkiotune, info_blkiotune, 0},
16532 16533
    {"blockpull", cmdBlockPull, opts_block_pull, info_block_pull, 0},
    {"blockjob", cmdBlockJob, opts_block_job, info_block_job, 0},
16534
    {"blockresize", cmdBlockResize, opts_block_resize, info_block_resize, 0},
16535
#ifndef WIN32
16536
    {"console", cmdConsole, opts_console, info_console, 0},
16537
#endif
16538 16539 16540 16541
    {"cpu-baseline", cmdCPUBaseline, opts_cpu_baseline, info_cpu_baseline, 0},
    {"cpu-compare", cmdCPUCompare, opts_cpu_compare, info_cpu_compare, 0},
    {"create", cmdCreate, opts_create, info_create, 0},
    {"define", cmdDefine, opts_define, info_define, 0},
16542
    {"desc", cmdDesc, opts_desc, info_desc, 0},
16543 16544 16545 16546 16547 16548 16549
    {"destroy", cmdDestroy, opts_destroy, info_destroy, 0},
    {"detach-device", cmdDetachDevice, opts_detach_device,
     info_detach_device, 0},
    {"detach-disk", cmdDetachDisk, opts_detach_disk, info_detach_disk, 0},
    {"detach-interface", cmdDetachInterface, opts_detach_interface,
     info_detach_interface, 0},
    {"domid", cmdDomid, opts_domid, info_domid, 0},
16550
    {"domif-setlink", cmdDomIfSetLink, opts_domif_setlink, info_domif_setlink, 0},
16551
    {"domiftune", cmdDomIftune, opts_domiftune, info_domiftune, 0},
16552 16553 16554
    {"domjobabort", cmdDomjobabort, opts_domjobabort, info_domjobabort, 0},
    {"domjobinfo", cmdDomjobinfo, opts_domjobinfo, info_domjobinfo, 0},
    {"domname", cmdDomname, opts_domname, info_domname, 0},
16555 16556
    {"dompmsuspend", cmdDomPMSuspend,
     opts_dom_pm_suspend, info_dom_pm_suspend, 0},
16557 16558 16559 16560 16561 16562 16563 16564 16565
    {"domuuid", cmdDomuuid, opts_domuuid, info_domuuid, 0},
    {"domxml-from-native", cmdDomXMLFromNative, opts_domxmlfromnative,
     info_domxmlfromnative, 0},
    {"domxml-to-native", cmdDomXMLToNative, opts_domxmltonative,
     info_domxmltonative, 0},
    {"dump", cmdDump, opts_dump, info_dump, 0},
    {"dumpxml", cmdDumpXML, opts_dumpxml, info_dumpxml, 0},
    {"edit", cmdEdit, opts_edit, info_edit, 0},
    {"inject-nmi", cmdInjectNMI, opts_inject_nmi, info_inject_nmi, 0},
16566
    {"send-key", cmdSendKey, opts_send_key, info_send_key},
16567 16568 16569 16570 16571 16572 16573 16574
    {"managedsave", cmdManagedSave, opts_managedsave, info_managedsave, 0},
    {"managedsave-remove", cmdManagedSaveRemove, opts_managedsaveremove,
     info_managedsaveremove, 0},
    {"maxvcpus", cmdMaxvcpus, opts_maxvcpus, info_maxvcpus, 0},
    {"memtune", cmdMemtune, opts_memtune, info_memtune, 0},
    {"migrate", cmdMigrate, opts_migrate, info_migrate, 0},
    {"migrate-setmaxdowntime", cmdMigrateSetMaxDowntime,
     opts_migrate_setmaxdowntime, info_migrate_setmaxdowntime, 0},
16575 16576
    {"migrate-setspeed", cmdMigrateSetMaxSpeed,
     opts_migrate_setspeed, info_migrate_setspeed, 0},
16577 16578
    {"migrate-getspeed", cmdMigrateGetMaxSpeed,
     opts_migrate_getspeed, info_migrate_getspeed, 0},
H
Hu Tao 已提交
16579
    {"numatune", cmdNumatune, opts_numatune, info_numatune, 0},
16580
    {"reboot", cmdReboot, opts_reboot, info_reboot, 0},
X
Xu He Jie 已提交
16581
    {"reset", cmdReset, opts_reset, info_reset, 0},
16582 16583 16584
    {"restore", cmdRestore, opts_restore, info_restore, 0},
    {"resume", cmdResume, opts_resume, info_resume, 0},
    {"save", cmdSave, opts_save, info_save, 0},
16585 16586 16587 16588 16589 16590
    {"save-image-define", cmdSaveImageDefine, opts_save_image_define,
     info_save_image_define, 0},
    {"save-image-dumpxml", cmdSaveImageDumpxml, opts_save_image_dumpxml,
     info_save_image_dumpxml, 0},
    {"save-image-edit", cmdSaveImageEdit, opts_save_image_edit,
     info_save_image_edit, 0},
16591
    {"schedinfo", cmdSchedinfo, opts_schedinfo, info_schedinfo, 0},
16592
    {"screenshot", cmdScreenshot, opts_screenshot, info_screenshot, 0},
16593 16594 16595 16596 16597 16598 16599 16600 16601 16602 16603 16604
    {"setmaxmem", cmdSetmaxmem, opts_setmaxmem, info_setmaxmem, 0},
    {"setmem", cmdSetmem, opts_setmem, info_setmem, 0},
    {"setvcpus", cmdSetvcpus, opts_setvcpus, info_setvcpus, 0},
    {"shutdown", cmdShutdown, opts_shutdown, info_shutdown, 0},
    {"start", cmdStart, opts_start, info_start, 0},
    {"suspend", cmdSuspend, opts_suspend, info_suspend, 0},
    {"ttyconsole", cmdTTYConsole, opts_ttyconsole, info_ttyconsole, 0},
    {"undefine", cmdUndefine, opts_undefine, info_undefine, 0},
    {"update-device", cmdUpdateDevice, opts_update_device,
     info_update_device, 0},
    {"vcpucount", cmdVcpucount, opts_vcpucount, info_vcpucount, 0},
    {"vcpuinfo", cmdVcpuinfo, opts_vcpuinfo, info_vcpuinfo, 0},
E
Eric Blake 已提交
16605
    {"vcpupin", cmdVcpuPin, opts_vcpupin, info_vcpupin, 0},
16606 16607
    {"vncdisplay", cmdVNCDisplay, opts_vncdisplay, info_vncdisplay, 0},
    {NULL, NULL, NULL, NULL, 0}
16608 16609 16610
};

static const vshCmdDef domMonitoringCmds[] = {
16611
    {"domblkerror", cmdDomBlkError, opts_domblkerror, info_domblkerror, 0},
16612
    {"domblkinfo", cmdDomblkinfo, opts_domblkinfo, info_domblkinfo, 0},
16613
    {"domblklist", cmdDomblklist, opts_domblklist, info_domblklist, 0},
16614
    {"domblkstat", cmdDomblkstat, opts_domblkstat, info_domblkstat, 0},
16615
    {"domcontrol", cmdDomControl, opts_domcontrol, info_domcontrol, 0},
16616
    {"domif-getlink", cmdDomIfGetLink, opts_domif_getlink, info_domif_getlink, 0},
16617
    {"domiflist", cmdDomiflist, opts_domiflist, info_domiflist, 0},
16618 16619 16620 16621 16622 16623
    {"domifstat", cmdDomIfstat, opts_domifstat, info_domifstat, 0},
    {"dominfo", cmdDominfo, opts_dominfo, info_dominfo, 0},
    {"dommemstat", cmdDomMemStat, opts_dommemstat, info_dommemstat, 0},
    {"domstate", cmdDomstate, opts_domstate, info_domstate, 0},
    {"list", cmdList, opts_list, info_list, 0},
    {NULL, NULL, NULL, NULL, 0}
16624 16625 16626
};

static const vshCmdDef storagePoolCmds[] = {
16627
    {"find-storage-pool-sources-as", cmdPoolDiscoverSourcesAs,
16628
     opts_find_storage_pool_sources_as, info_find_storage_pool_sources_as, 0},
16629
    {"find-storage-pool-sources", cmdPoolDiscoverSources,
16630 16631 16632 16633 16634 16635 16636 16637 16638 16639 16640 16641 16642 16643 16644 16645 16646 16647 16648 16649 16650
     opts_find_storage_pool_sources, info_find_storage_pool_sources, 0},
    {"pool-autostart", cmdPoolAutostart, opts_pool_autostart,
     info_pool_autostart, 0},
    {"pool-build", cmdPoolBuild, opts_pool_build, info_pool_build, 0},
    {"pool-create-as", cmdPoolCreateAs, opts_pool_X_as, info_pool_create_as, 0},
    {"pool-create", cmdPoolCreate, opts_pool_create, info_pool_create, 0},
    {"pool-define-as", cmdPoolDefineAs, opts_pool_X_as, info_pool_define_as, 0},
    {"pool-define", cmdPoolDefine, opts_pool_define, info_pool_define, 0},
    {"pool-delete", cmdPoolDelete, opts_pool_delete, info_pool_delete, 0},
    {"pool-destroy", cmdPoolDestroy, opts_pool_destroy, info_pool_destroy, 0},
    {"pool-dumpxml", cmdPoolDumpXML, opts_pool_dumpxml, info_pool_dumpxml, 0},
    {"pool-edit", cmdPoolEdit, opts_pool_edit, info_pool_edit, 0},
    {"pool-info", cmdPoolInfo, opts_pool_info, info_pool_info, 0},
    {"pool-list", cmdPoolList, opts_pool_list, info_pool_list, 0},
    {"pool-name", cmdPoolName, opts_pool_name, info_pool_name, 0},
    {"pool-refresh", cmdPoolRefresh, opts_pool_refresh, info_pool_refresh, 0},
    {"pool-start", cmdPoolStart, opts_pool_start, info_pool_start, 0},
    {"pool-undefine", cmdPoolUndefine, opts_pool_undefine,
     info_pool_undefine, 0},
    {"pool-uuid", cmdPoolUuid, opts_pool_uuid, info_pool_uuid, 0},
    {NULL, NULL, NULL, NULL, 0}
16651 16652 16653
};

static const vshCmdDef storageVolCmds[] = {
16654 16655 16656 16657 16658 16659 16660 16661 16662 16663 16664 16665 16666 16667 16668
    {"vol-clone", cmdVolClone, opts_vol_clone, info_vol_clone, 0},
    {"vol-create-as", cmdVolCreateAs, opts_vol_create_as,
     info_vol_create_as, 0},
    {"vol-create", cmdVolCreate, opts_vol_create, info_vol_create, 0},
    {"vol-create-from", cmdVolCreateFrom, opts_vol_create_from,
     info_vol_create_from, 0},
    {"vol-delete", cmdVolDelete, opts_vol_delete, info_vol_delete, 0},
    {"vol-download", cmdVolDownload, opts_vol_download, info_vol_download, 0},
    {"vol-dumpxml", cmdVolDumpXML, opts_vol_dumpxml, info_vol_dumpxml, 0},
    {"vol-info", cmdVolInfo, opts_vol_info, info_vol_info, 0},
    {"vol-key", cmdVolKey, opts_vol_key, info_vol_key, 0},
    {"vol-list", cmdVolList, opts_vol_list, info_vol_list, 0},
    {"vol-name", cmdVolName, opts_vol_name, info_vol_name, 0},
    {"vol-path", cmdVolPath, opts_vol_path, info_vol_path, 0},
    {"vol-pool", cmdVolPool, opts_vol_pool, info_vol_pool, 0},
16669
    {"vol-resize", cmdVolResize, opts_vol_resize, info_vol_resize, 0},
16670 16671 16672
    {"vol-upload", cmdVolUpload, opts_vol_upload, info_vol_upload, 0},
    {"vol-wipe", cmdVolWipe, opts_vol_wipe, info_vol_wipe, 0},
    {NULL, NULL, NULL, NULL, 0}
16673
};
16674

16675
static const vshCmdDef networkCmds[] = {
16676 16677 16678 16679 16680 16681 16682 16683 16684 16685 16686 16687 16688 16689 16690 16691 16692 16693 16694
    {"net-autostart", cmdNetworkAutostart, opts_network_autostart,
     info_network_autostart, 0},
    {"net-create", cmdNetworkCreate, opts_network_create,
     info_network_create, 0},
    {"net-define", cmdNetworkDefine, opts_network_define,
     info_network_define, 0},
    {"net-destroy", cmdNetworkDestroy, opts_network_destroy,
     info_network_destroy, 0},
    {"net-dumpxml", cmdNetworkDumpXML, opts_network_dumpxml,
     info_network_dumpxml, 0},
    {"net-edit", cmdNetworkEdit, opts_network_edit, info_network_edit, 0},
    {"net-info", cmdNetworkInfo, opts_network_info, info_network_info, 0},
    {"net-list", cmdNetworkList, opts_network_list, info_network_list, 0},
    {"net-name", cmdNetworkName, opts_network_name, info_network_name, 0},
    {"net-start", cmdNetworkStart, opts_network_start, info_network_start, 0},
    {"net-undefine", cmdNetworkUndefine, opts_network_undefine,
     info_network_undefine, 0},
    {"net-uuid", cmdNetworkUuid, opts_network_uuid, info_network_uuid, 0},
    {NULL, NULL, NULL, NULL, 0}
16695
};
16696

16697
static const vshCmdDef nodedevCmds[] = {
16698 16699 16700 16701 16702 16703 16704 16705 16706 16707 16708 16709 16710 16711 16712
    {"nodedev-create", cmdNodeDeviceCreate, opts_node_device_create,
     info_node_device_create, 0},
    {"nodedev-destroy", cmdNodeDeviceDestroy, opts_node_device_destroy,
     info_node_device_destroy, 0},
    {"nodedev-dettach", cmdNodeDeviceDettach, opts_node_device_dettach,
     info_node_device_dettach, 0},
    {"nodedev-dumpxml", cmdNodeDeviceDumpXML, opts_node_device_dumpxml,
     info_node_device_dumpxml, 0},
    {"nodedev-list", cmdNodeListDevices, opts_node_list_devices,
     info_node_list_devices, 0},
    {"nodedev-reattach", cmdNodeDeviceReAttach, opts_node_device_reattach,
     info_node_device_reattach, 0},
    {"nodedev-reset", cmdNodeDeviceReset, opts_node_device_reset,
     info_node_device_reset, 0},
    {NULL, NULL, NULL, NULL, 0}
16713 16714 16715
};

static const vshCmdDef ifaceCmds[] = {
E
Eric Blake 已提交
16716 16717
    {"iface-begin", cmdInterfaceBegin, opts_interface_begin,
     info_interface_begin, 0},
16718 16719
    {"iface-bridge", cmdInterfaceBridge, opts_interface_bridge,
     info_interface_bridge, 0},
E
Eric Blake 已提交
16720 16721
    {"iface-commit", cmdInterfaceCommit, opts_interface_commit,
     info_interface_commit, 0},
16722 16723 16724 16725 16726 16727 16728 16729 16730 16731 16732 16733 16734 16735
    {"iface-define", cmdInterfaceDefine, opts_interface_define,
     info_interface_define, 0},
    {"iface-destroy", cmdInterfaceDestroy, opts_interface_destroy,
     info_interface_destroy, 0},
    {"iface-dumpxml", cmdInterfaceDumpXML, opts_interface_dumpxml,
     info_interface_dumpxml, 0},
    {"iface-edit", cmdInterfaceEdit, opts_interface_edit,
     info_interface_edit, 0},
    {"iface-list", cmdInterfaceList, opts_interface_list,
     info_interface_list, 0},
    {"iface-mac", cmdInterfaceMAC, opts_interface_mac,
     info_interface_mac, 0},
    {"iface-name", cmdInterfaceName, opts_interface_name,
     info_interface_name, 0},
E
Eric Blake 已提交
16736 16737
    {"iface-rollback", cmdInterfaceRollback, opts_interface_rollback,
     info_interface_rollback, 0},
16738 16739
    {"iface-start", cmdInterfaceStart, opts_interface_start,
     info_interface_start, 0},
16740 16741
    {"iface-unbridge", cmdInterfaceUnbridge, opts_interface_unbridge,
     info_interface_unbridge, 0},
16742 16743 16744
    {"iface-undefine", cmdInterfaceUndefine, opts_interface_undefine,
     info_interface_undefine, 0},
    {NULL, NULL, NULL, NULL, 0}
16745
};
16746

16747
static const vshCmdDef nwfilterCmds[] = {
16748 16749 16750 16751 16752 16753 16754 16755 16756 16757 16758
    {"nwfilter-define", cmdNWFilterDefine, opts_nwfilter_define,
     info_nwfilter_define, 0},
    {"nwfilter-dumpxml", cmdNWFilterDumpXML, opts_nwfilter_dumpxml,
     info_nwfilter_dumpxml, 0},
    {"nwfilter-edit", cmdNWFilterEdit, opts_nwfilter_edit,
     info_nwfilter_edit, 0},
    {"nwfilter-list", cmdNWFilterList, opts_nwfilter_list,
     info_nwfilter_list, 0},
    {"nwfilter-undefine", cmdNWFilterUndefine, opts_nwfilter_undefine,
     info_nwfilter_undefine, 0},
    {NULL, NULL, NULL, NULL, 0}
16759
};
16760

16761
static const vshCmdDef secretCmds[] = {
16762 16763 16764 16765 16766 16767 16768 16769 16770 16771 16772 16773
    {"secret-define", cmdSecretDefine, opts_secret_define,
     info_secret_define, 0},
    {"secret-dumpxml", cmdSecretDumpXML, opts_secret_dumpxml,
     info_secret_dumpxml, 0},
    {"secret-get-value", cmdSecretGetValue, opts_secret_get_value,
     info_secret_get_value, 0},
    {"secret-list", cmdSecretList, NULL, info_secret_list, 0},
    {"secret-set-value", cmdSecretSetValue, opts_secret_set_value,
     info_secret_set_value, 0},
    {"secret-undefine", cmdSecretUndefine, opts_secret_undefine,
     info_secret_undefine, 0},
    {NULL, NULL, NULL, NULL, 0}
16774
};
16775

16776
static const vshCmdDef virshCmds[] = {
16777 16778 16779 16780 16781 16782 16783
    {"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}
16784
};
16785

16786
static const vshCmdDef snapshotCmds[] = {
16787 16788
    {"snapshot-create", cmdSnapshotCreate, opts_snapshot_create,
     info_snapshot_create, 0},
16789 16790
    {"snapshot-create-as", cmdSnapshotCreateAs, opts_snapshot_create_as,
     info_snapshot_create_as, 0},
16791 16792 16793 16794 16795 16796
    {"snapshot-current", cmdSnapshotCurrent, opts_snapshot_current,
     info_snapshot_current, 0},
    {"snapshot-delete", cmdSnapshotDelete, opts_snapshot_delete,
     info_snapshot_delete, 0},
    {"snapshot-dumpxml", cmdSnapshotDumpXML, opts_snapshot_dumpxml,
     info_snapshot_dumpxml, 0},
16797 16798
    {"snapshot-edit", cmdSnapshotEdit, opts_snapshot_edit,
     info_snapshot_edit, 0},
16799 16800
    {"snapshot-list", cmdSnapshotList, opts_snapshot_list,
     info_snapshot_list, 0},
E
Eric Blake 已提交
16801 16802
    {"snapshot-parent", cmdSnapshotParent, opts_snapshot_parent,
     info_snapshot_parent, 0},
16803 16804 16805
    {"snapshot-revert", cmdDomainSnapshotRevert, opts_snapshot_revert,
     info_snapshot_revert, 0},
    {NULL, NULL, NULL, NULL, 0}
16806
};
16807

16808
static const vshCmdDef hostAndHypervisorCmds[] = {
16809 16810 16811 16812 16813
    {"capabilities", cmdCapabilities, NULL, info_capabilities, 0},
    {"connect", cmdConnect, opts_connect, info_connect,
     VSH_CMD_FLAG_NOCONNECT},
    {"freecell", cmdFreecell, opts_freecell, info_freecell, 0},
    {"hostname", cmdHostname, NULL, info_hostname, 0},
16814
    {"nodecpustats", cmdNodeCpuStats, opts_node_cpustats, info_nodecpustats, 0},
16815
    {"nodeinfo", cmdNodeinfo, NULL, info_nodeinfo, 0},
16816
    {"nodememstats", cmdNodeMemStats, opts_node_memstats, info_nodememstats, 0},
16817
    {"nodesuspend", cmdNodeSuspend, opts_node_suspend, info_nodesuspend, 0},
16818
    {"qemu-attach", cmdQemuAttach, opts_qemu_attach, info_qemu_attach},
16819 16820 16821 16822
    {"qemu-monitor-command", cmdQemuMonitorCommand, opts_qemu_monitor_command,
     info_qemu_monitor_command, 0},
    {"sysinfo", cmdSysinfo, NULL, info_sysinfo, 0},
    {"uri", cmdURI, NULL, info_uri, 0},
16823
    {"version", cmdVersion, opts_version, info_version, 0},
16824
    {NULL, NULL, NULL, NULL, 0}
K
Karel Zak 已提交
16825 16826
};

16827 16828 16829 16830 16831 16832 16833 16834 16835 16836 16837 16838 16839 16840 16841 16842 16843
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 已提交
16844 16845 16846 16847
/* ---------------
 * Utils for work with command definition
 * ---------------
 */
K
Karel Zak 已提交
16848
static const char *
16849
vshCmddefGetInfo(const vshCmdDef * cmd, const char *name)
16850
{
16851
    const vshCmdInfo *info;
16852

K
Karel Zak 已提交
16853
    for (info = cmd->info; info && info->name; info++) {
16854
        if (STREQ(info->name, name))
K
Karel Zak 已提交
16855 16856 16857 16858 16859
            return info->data;
    }
    return NULL;
}

16860
/* Validate that the options associated with cmd can be parsed.  */
16861
static int
E
Eric Blake 已提交
16862
vshCmddefOptParse(const vshCmdDef *cmd, uint32_t *opts_need_arg,
16863 16864 16865 16866 16867
                  uint32_t *opts_required)
{
    int i;
    bool optional = false;

E
Eric Blake 已提交
16868 16869 16870
    *opts_need_arg = 0;
    *opts_required = 0;

16871 16872 16873 16874 16875 16876 16877 16878 16879
    if (!cmd->opts)
        return 0;

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

        if (i > 31)
            return -1; /* too many options */
        if (opt->type == VSH_OT_BOOL) {
E
Eric Blake 已提交
16880
            if (opt->flags & VSH_OFLAG_REQ)
16881 16882 16883
                return -1; /* bool options can't be mandatory */
            continue;
        }
E
Eric Blake 已提交
16884 16885
        if (opt->flags & VSH_OFLAG_REQ_OPT) {
            if (opt->flags & VSH_OFLAG_REQ)
L
Lai Jiangshan 已提交
16886 16887 16888 16889
                *opts_required |= 1 << i;
            continue;
        }

16890
        *opts_need_arg |= 1 << i;
E
Eric Blake 已提交
16891
        if (opt->flags & VSH_OFLAG_REQ) {
16892 16893 16894 16895 16896 16897
            if (optional)
                return -1; /* mandatory options must be listed first */
            *opts_required |= 1 << i;
        } else {
            optional = true;
        }
16898 16899 16900

        if (opt->type == VSH_OT_ARGV && cmd->opts[i + 1].name)
            return -1; /* argv option must be listed last */
16901 16902 16903 16904
    }
    return 0;
}

16905
static const vshCmdOptDef *
16906
vshCmddefGetOption(vshControl *ctl, const vshCmdDef *cmd, const char *name,
16907
                   uint32_t *opts_seen, int *opt_index)
16908
{
16909 16910 16911 16912
    int i;

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

16914
        if (STREQ(opt->name, name)) {
16915
            if ((*opts_seen & (1 << i)) && opt->type != VSH_OT_ARGV) {
16916 16917 16918
                vshError(ctl, _("option --%s already seen"), name);
                return NULL;
            }
16919 16920
            *opts_seen |= 1 << i;
            *opt_index = i;
K
Karel Zak 已提交
16921
            return opt;
16922 16923 16924 16925 16926
        }
    }

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

16930
static const vshCmdOptDef *
16931 16932
vshCmddefGetData(const vshCmdDef *cmd, uint32_t *opts_need_arg,
                 uint32_t *opts_seen)
16933
{
16934
    int i;
16935
    const vshCmdOptDef *opt;
K
Karel Zak 已提交
16936

16937 16938 16939 16940
    if (!*opts_need_arg)
        return NULL;

    /* Grab least-significant set bit */
E
Eric Blake 已提交
16941
    i = ffs(*opts_need_arg) - 1;
16942
    opt = &cmd->opts[i];
16943
    if (opt->type != VSH_OT_ARGV)
16944
        *opts_need_arg &= ~(1 << i);
16945
    *opts_seen |= 1 << i;
16946
    return opt;
K
Karel Zak 已提交
16947 16948
}

16949 16950 16951
/*
 * Checks for required options
 */
16952
static int
16953 16954
vshCommandCheckOpts(vshControl *ctl, const vshCmd *cmd, uint32_t opts_required,
                    uint32_t opts_seen)
16955
{
16956
    const vshCmdDef *def = cmd->def;
16957 16958 16959 16960 16961 16962 16963 16964 16965
    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];
16966

16967
            vshError(ctl,
16968
                     opt->type == VSH_OT_DATA || opt->type == VSH_OT_ARGV ?
16969 16970 16971
                     _("command '%s' requires <%s> option") :
                     _("command '%s' requires --%s option"),
                     def->name, opt->name);
16972 16973
        }
    }
16974
    return -1;
16975 16976
}

16977
static const vshCmdDef *
16978 16979
vshCmddefSearch(const char *cmdname)
{
16980
    const vshCmdGrp *g;
16981
    const vshCmdDef *c;
16982

16983 16984
    for (g = cmdGroups; g->name; g++) {
        for (c = g->commands; c->name; c++) {
16985
            if (STREQ(c->name, cmdname))
16986 16987 16988 16989
                return c;
        }
    }

K
Karel Zak 已提交
16990 16991 16992
    return NULL;
}

16993 16994 16995 16996 16997 16998
static const vshCmdGrp *
vshCmdGrpSearch(const char *grpname)
{
    const vshCmdGrp *g;

    for (g = cmdGroups; g->name; g++) {
16999
        if (STREQ(g->name, grpname) || STREQ(g->keyword, grpname))
17000 17001 17002 17003 17004 17005
            return g;
    }

    return NULL;
}

E
Eric Blake 已提交
17006
static bool
17007 17008 17009 17010 17011 17012 17013
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 已提交
17014
        return false;
17015 17016 17017 17018 17019 17020 17021 17022 17023 17024
    } 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 已提交
17025
    return true;
17026 17027
}

E
Eric Blake 已提交
17028
static bool
17029
vshCmddefHelp(vshControl *ctl, const char *cmdname)
17030
{
17031
    const vshCmdDef *def = vshCmddefSearch(cmdname);
17032

K
Karel Zak 已提交
17033
    if (!def) {
17034
        vshError(ctl, _("command '%s' doesn't exist"), cmdname);
E
Eric Blake 已提交
17035
        return false;
17036
    } else {
E
Eric Blake 已提交
17037 17038
        /* Don't translate desc if it is "".  */
        const char *desc = vshCmddefGetInfo(def, "desc");
E
Eric Blake 已提交
17039
        const char *help = _(vshCmddefGetInfo(def, "help"));
17040
        char buf[256];
17041 17042
        uint32_t opts_need_arg;
        uint32_t opts_required;
17043
        bool shortopt = false; /* true if 'arg' works instead of '--opt arg' */
17044 17045 17046 17047

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

17051
        fputs(_("  NAME\n"), stdout);
17052 17053
        fprintf(stdout, "    %s - %s\n", def->name, help);

17054 17055 17056 17057 17058
        fputs(_("\n  SYNOPSIS\n"), stdout);
        fprintf(stdout, "    %s", def->name);
        if (def->opts) {
            const vshCmdOptDef *opt;
            for (opt = def->opts; opt->name; opt++) {
17059
                const char *fmt = "%s";
17060 17061
                switch (opt->type) {
                case VSH_OT_BOOL:
17062
                    fmt = "[--%s]";
17063 17064
                    break;
                case VSH_OT_INT:
E
Eric Blake 已提交
17065
                    /* xgettext:c-format */
E
Eric Blake 已提交
17066
                    fmt = ((opt->flags & VSH_OFLAG_REQ) ? "<%s>"
17067
                           : _("[--%s <number>]"));
17068 17069
                    if (!(opt->flags & VSH_OFLAG_REQ_OPT))
                        shortopt = true;
17070 17071
                    break;
                case VSH_OT_STRING:
E
Eric Blake 已提交
17072 17073
                    /* xgettext:c-format */
                    fmt = _("[--%s <string>]");
17074 17075
                    if (!(opt->flags & VSH_OFLAG_REQ_OPT))
                        shortopt = true;
17076 17077
                    break;
                case VSH_OT_DATA:
E
Eric Blake 已提交
17078
                    fmt = ((opt->flags & VSH_OFLAG_REQ) ? "<%s>" : "[<%s>]");
17079 17080
                    if (!(opt->flags & VSH_OFLAG_REQ_OPT))
                        shortopt = true;
17081 17082 17083
                    break;
                case VSH_OT_ARGV:
                    /* xgettext:c-format */
17084 17085 17086 17087 17088 17089 17090 17091
                    if (shortopt) {
                        fmt = (opt->flags & VSH_OFLAG_REQ)
                            ? _("{[--%s] <string>}...")
                            : _("[[--%s] <string>]...");
                    } else {
                        fmt = (opt->flags & VSH_OFLAG_REQ) ? _("<%s>...")
                            : _("[<%s>]...");
                    }
17092 17093
                    break;
                default:
17094
                    assert(0);
17095
                }
17096
                fputc(' ', stdout);
E
Eric Blake 已提交
17097
                fprintf(stdout, fmt, opt->name);
17098
            }
K
Karel Zak 已提交
17099
        }
17100 17101 17102
        fputc('\n', stdout);

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

K
Karel Zak 已提交
17108
        if (def->opts) {
17109
            const vshCmdOptDef *opt;
17110
            fputs(_("\n  OPTIONS\n"), stdout);
17111
            for (opt = def->opts; opt->name; opt++) {
17112 17113
                switch (opt->type) {
                case VSH_OT_BOOL:
K
Karel Zak 已提交
17114
                    snprintf(buf, sizeof(buf), "--%s", opt->name);
17115 17116
                    break;
                case VSH_OT_INT:
17117
                    snprintf(buf, sizeof(buf),
E
Eric Blake 已提交
17118
                             (opt->flags & VSH_OFLAG_REQ) ? _("[--%s] <number>")
17119
                             : _("--%s <number>"), opt->name);
17120 17121
                    break;
                case VSH_OT_STRING:
17122
                    /* OT_STRING should never be VSH_OFLAG_REQ */
17123
                    snprintf(buf, sizeof(buf), _("--%s <string>"), opt->name);
17124 17125
                    break;
                case VSH_OT_DATA:
17126 17127
                    snprintf(buf, sizeof(buf), _("[--%s] <string>"),
                             opt->name);
17128 17129
                    break;
                case VSH_OT_ARGV:
17130 17131 17132
                    snprintf(buf, sizeof(buf),
                             shortopt ? _("[--%s] <string>") : _("<%s>"),
                             opt->name);
17133
                    break;
17134 17135 17136
                default:
                    assert(0);
                }
17137

E
Eric Blake 已提交
17138
                fprintf(stdout, "    %-15s  %s\n", buf, _(opt->help));
17139
            }
K
Karel Zak 已提交
17140 17141 17142
        }
        fputc('\n', stdout);
    }
E
Eric Blake 已提交
17143
    return true;
K
Karel Zak 已提交
17144 17145 17146 17147 17148 17149
}

/* ---------------
 * Utils for work with runtime commands data
 * ---------------
 */
17150 17151 17152
static void
vshCommandOptFree(vshCmdOpt * arg)
{
K
Karel Zak 已提交
17153 17154
    vshCmdOpt *a = arg;

17155
    while (a) {
K
Karel Zak 已提交
17156
        vshCmdOpt *tmp = a;
17157

K
Karel Zak 已提交
17158 17159
        a = a->next;

17160 17161
        VIR_FREE(tmp->data);
        VIR_FREE(tmp);
K
Karel Zak 已提交
17162 17163 17164 17165
    }
}

static void
17166
vshCommandFree(vshCmd *cmd)
17167
{
K
Karel Zak 已提交
17168 17169
    vshCmd *c = cmd;

17170
    while (c) {
K
Karel Zak 已提交
17171
        vshCmd *tmp = c;
17172

K
Karel Zak 已提交
17173 17174 17175 17176
        c = c->next;

        if (tmp->opts)
            vshCommandOptFree(tmp->opts);
17177
        VIR_FREE(tmp);
K
Karel Zak 已提交
17178 17179 17180
    }
}

E
Eric Blake 已提交
17181 17182 17183 17184 17185 17186 17187 17188 17189 17190 17191
/**
 * 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 已提交
17192
 */
E
Eric Blake 已提交
17193 17194
static int
vshCommandOpt(const vshCmd *cmd, const char *name, vshCmdOpt **opt)
17195
{
E
Eric Blake 已提交
17196 17197
    vshCmdOpt *candidate = cmd->opts;
    const vshCmdOptDef *valid = cmd->def->opts;
17198

E
Eric Blake 已提交
17199 17200 17201 17202 17203 17204 17205
    /* 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 已提交
17206
    }
E
Eric Blake 已提交
17207 17208 17209 17210 17211 17212 17213

    /* Option not present, see if command requires it.  */
    *opt = NULL;
    while (valid) {
        if (!valid->name)
            break;
        if (STREQ(name, valid->name))
E
Eric Blake 已提交
17214
            return (valid->flags & VSH_OFLAG_REQ) == 0 ? 0 : -1;
E
Eric Blake 已提交
17215 17216 17217 17218
        valid++;
    }
    /* If we got here, the name is unknown.  */
    return -2;
K
Karel Zak 已提交
17219 17220
}

E
Eric Blake 已提交
17221 17222
/**
 * vshCommandOptInt:
17223 17224 17225 17226 17227 17228 17229
 * @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 已提交
17230
 * 0 if option not found and not required (@value untouched)
17231
 * <0 in all other cases (@value untouched)
K
Karel Zak 已提交
17232 17233
 */
static int
17234
vshCommandOptInt(const vshCmd *cmd, const char *name, int *value)
17235
{
E
Eric Blake 已提交
17236 17237 17238
    vshCmdOpt *arg;
    int ret;
    int num;
17239
    char *end_p = NULL;
17240

E
Eric Blake 已提交
17241 17242 17243 17244 17245 17246 17247
    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;
17248
    }
E
Eric Blake 已提交
17249 17250 17251 17252 17253 17254 17255

    num = strtol(arg->data, &end_p, 10);
    if ((arg->data != end_p) && (*end_p == 0)) {
        *value = num;
        return 1;
    }
    return -1;
K
Karel Zak 已提交
17256 17257
}

17258

E
Eric Blake 已提交
17259 17260 17261 17262 17263 17264
/**
 * vshCommandOptUInt:
 * @cmd command reference
 * @name option name
 * @value result
 *
17265 17266 17267 17268 17269 17270
 * Convert option to unsigned int
 * See vshCommandOptInt()
 */
static int
vshCommandOptUInt(const vshCmd *cmd, const char *name, unsigned int *value)
{
E
Eric Blake 已提交
17271 17272 17273
    vshCmdOpt *arg;
    int ret;
    unsigned int num;
17274 17275
    char *end_p = NULL;

E
Eric Blake 已提交
17276 17277 17278 17279 17280 17281 17282
    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;
17283
    }
E
Eric Blake 已提交
17284 17285 17286 17287 17288 17289 17290

    num = strtoul(arg->data, &end_p, 10);
    if ((arg->data != end_p) && (*end_p == 0)) {
        *value = num;
        return 1;
    }
    return -1;
17291 17292 17293
}


17294
/*
E
Eric Blake 已提交
17295 17296 17297 17298 17299
 * vshCommandOptUL:
 * @cmd command reference
 * @name option name
 * @value result
 *
17300 17301 17302 17303 17304
 * Convert option to unsigned long
 * See vshCommandOptInt()
 */
static int
vshCommandOptUL(const vshCmd *cmd, const char *name, unsigned long *value)
17305
{
E
Eric Blake 已提交
17306 17307
    vshCmdOpt *arg;
    int ret;
17308
    unsigned long num;
17309 17310
    char *end_p = NULL;

E
Eric Blake 已提交
17311 17312 17313 17314 17315 17316 17317
    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;
17318
    }
E
Eric Blake 已提交
17319 17320 17321 17322 17323 17324 17325

    num = strtoul(arg->data, &end_p, 10);
    if ((arg->data != end_p) && (*end_p == 0)) {
        *value = num;
        return 1;
    }
    return -1;
17326 17327
}

E
Eric Blake 已提交
17328 17329 17330 17331 17332 17333
/**
 * vshCommandOptString:
 * @cmd command reference
 * @name option name
 * @value result
 *
K
Karel Zak 已提交
17334
 * Returns option as STRING
E
Eric Blake 已提交
17335 17336 17337 17338
 * 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 已提交
17339
 */
17340 17341
static int
vshCommandOptString(const vshCmd *cmd, const char *name, const char **value)
17342
{
E
Eric Blake 已提交
17343 17344 17345 17346 17347 17348 17349 17350 17351 17352
    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;
17353
    }
17354

E
Eric Blake 已提交
17355
    if (!*arg->data && !(arg->def->flags & VSH_OFLAG_EMPTY_OK)) {
E
Eric Blake 已提交
17356 17357 17358 17359
        return -1;
    }
    *value = arg->data;
    return 1;
K
Karel Zak 已提交
17360 17361
}

E
Eric Blake 已提交
17362 17363 17364 17365 17366 17367
/**
 * vshCommandOptLongLong:
 * @cmd command reference
 * @name option name
 * @value result
 *
17368
 * Returns option as long long
17369
 * See vshCommandOptInt()
17370
 */
17371 17372 17373
static int
vshCommandOptLongLong(const vshCmd *cmd, const char *name,
                      long long *value)
17374
{
E
Eric Blake 已提交
17375 17376
    vshCmdOpt *arg;
    int ret;
17377
    long long num;
17378 17379
    char *end_p = NULL;

E
Eric Blake 已提交
17380 17381 17382 17383 17384 17385 17386
    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;
17387
    }
E
Eric Blake 已提交
17388 17389 17390 17391 17392 17393 17394

    num = strtoll(arg->data, &end_p, 10);
    if ((arg->data != end_p) && (*end_p == 0)) {
        *value = num;
        return 1;
    }
    return -1;
17395 17396
}

E
Eric Blake 已提交
17397 17398 17399 17400 17401 17402 17403 17404 17405
/**
 * vshCommandOptULongLong:
 * @cmd command reference
 * @name option name
 * @value result
 *
 * Returns option as long long
 * See vshCommandOptInt()
 */
17406 17407 17408 17409
static int
vshCommandOptULongLong(const vshCmd *cmd, const char *name,
                       unsigned long long *value)
{
E
Eric Blake 已提交
17410 17411
    vshCmdOpt *arg;
    int ret;
17412 17413 17414
    unsigned long long num;
    char *end_p = NULL;

E
Eric Blake 已提交
17415 17416 17417 17418 17419 17420 17421
    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;
17422
    }
E
Eric Blake 已提交
17423 17424 17425 17426 17427 17428 17429

    num = strtoull(arg->data, &end_p, 10);
    if ((arg->data != end_p) && (*end_p == 0)) {
        *value = num;
        return 1;
    }
    return -1;
17430 17431 17432
}


E
Eric Blake 已提交
17433 17434 17435 17436 17437 17438 17439 17440 17441
/**
 * 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 已提交
17442
 */
E
Eric Blake 已提交
17443
static bool
17444
vshCommandOptBool(const vshCmd *cmd, const char *name)
17445
{
E
Eric Blake 已提交
17446 17447 17448
    vshCmdOpt *dummy;

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

E
Eric Blake 已提交
17451 17452 17453 17454 17455
/**
 * vshCommandOptArgv:
 * @cmd command reference
 * @opt starting point for the search
 *
17456 17457
 * Returns the next argv argument after OPT (or the first one if OPT
 * is NULL), or NULL if no more are present.
17458
 *
17459
 * Requires that a VSH_OT_ARGV option be last in the
17460 17461
 * list of supported options in CMD->def->opts.
 */
17462 17463
static const vshCmdOpt *
vshCommandOptArgv(const vshCmd *cmd, const vshCmdOpt *opt)
17464
{
17465
    opt = opt ? opt->next : cmd->opts;
17466 17467

    while (opt) {
E
Eric Blake 已提交
17468
        if (opt->def->type == VSH_OT_ARGV) {
17469
            return opt;
17470 17471 17472 17473 17474 17475
        }
        opt = opt->next;
    }
    return NULL;
}

J
Jim Meyering 已提交
17476 17477 17478 17479 17480 17481 17482 17483 17484 17485 17486 17487 17488 17489 17490 17491 17492 17493
/* Determine whether CMD->opts includes an option with name OPTNAME.
   If not, give a diagnostic and return false.
   If so, return true.  */
static bool
cmd_has_option (vshControl *ctl, const vshCmd *cmd, const char *optname)
{
    /* 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) {
        if (STREQ (opt->def->name, optname) && opt->def->type == VSH_OT_DATA) {
            found = true;
            break;
        }
    }

    if (!found)
17494
        vshError(ctl, _("internal error: virsh %s: no %s VSH_OT_DATA option"),
J
Jim Meyering 已提交
17495 17496 17497
                 cmd->def->name, optname);
    return found;
}
17498

K
Karel Zak 已提交
17499
static virDomainPtr
J
Jim Meyering 已提交
17500
vshCommandOptDomainBy(vshControl *ctl, const vshCmd *cmd,
17501
                      const char **name, int flag)
17502
{
K
Karel Zak 已提交
17503
    virDomainPtr dom = NULL;
17504
    const char *n = NULL;
K
Karel Zak 已提交
17505
    int id;
J
Jim Meyering 已提交
17506 17507 17508
    const char *optname = "domain";
    if (!cmd_has_option (ctl, cmd, optname))
        return NULL;
17509

17510
    if (vshCommandOptString(cmd, optname, &n) <= 0)
17511 17512
        return NULL;

17513
    vshDebug(ctl, VSH_ERR_INFO, "%s: found option <%s>: %s\n",
17514 17515
             cmd->def->name, optname, n);

K
Karel Zak 已提交
17516 17517
    if (name)
        *name = n;
17518

K
Karel Zak 已提交
17519
    /* try it by ID */
17520
    if (flag & VSH_BYID) {
17521
        if (virStrToLong_i(n, NULL, 10, &id) == 0 && id >= 0) {
17522 17523
            vshDebug(ctl, VSH_ERR_DEBUG,
                     "%s: <%s> seems like domain ID\n",
K
Karel Zak 已提交
17524 17525 17526
                     cmd->def->name, optname);
            dom = virDomainLookupByID(ctl->conn, id);
        }
17527
    }
K
Karel Zak 已提交
17528
    /* try it by UUID */
17529
    if (dom==NULL && (flag & VSH_BYUUID) && strlen(n)==VIR_UUID_STRING_BUFLEN-1) {
17530
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as domain UUID\n",
17531
                 cmd->def->name, optname);
K
Karel Zak 已提交
17532
        dom = virDomainLookupByUUIDString(ctl->conn, n);
K
Karel Zak 已提交
17533
    }
K
Karel Zak 已提交
17534
    /* try it by NAME */
17535
    if (dom==NULL && (flag & VSH_BYNAME)) {
17536
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as domain NAME\n",
17537
                 cmd->def->name, optname);
K
Karel Zak 已提交
17538
        dom = virDomainLookupByName(ctl->conn, n);
17539
    }
K
Karel Zak 已提交
17540

17541
    if (!dom)
17542
        vshError(ctl, _("failed to get domain '%s'"), n);
17543

K
Karel Zak 已提交
17544 17545 17546
    return dom;
}

17547
static virNetworkPtr
J
Jim Meyering 已提交
17548
vshCommandOptNetworkBy(vshControl *ctl, const vshCmd *cmd,
17549
                       const char **name, int flag)
17550 17551
{
    virNetworkPtr network = NULL;
17552
    const char *n = NULL;
J
Jim Meyering 已提交
17553 17554 17555
    const char *optname = "network";
    if (!cmd_has_option (ctl, cmd, optname))
        return NULL;
17556

17557
    if (vshCommandOptString(cmd, optname, &n) <= 0)
17558 17559
        return NULL;

17560
    vshDebug(ctl, VSH_ERR_INFO, "%s: found option <%s>: %s\n",
17561 17562 17563 17564 17565 17566
             cmd->def->name, optname, n);

    if (name)
        *name = n;

    /* try it by UUID */
17567
    if ((flag & VSH_BYUUID) && (strlen(n) == VIR_UUID_STRING_BUFLEN-1)) {
17568
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as network UUID\n",
17569
                 cmd->def->name, optname);
17570 17571 17572 17573
        network = virNetworkLookupByUUIDString(ctl->conn, n);
    }
    /* try it by NAME */
    if (network==NULL && (flag & VSH_BYNAME)) {
17574
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as network NAME\n",
17575 17576 17577 17578 17579
                 cmd->def->name, optname);
        network = virNetworkLookupByName(ctl->conn, n);
    }

    if (!network)
17580
        vshError(ctl, _("failed to get network '%s'"), n);
17581 17582 17583 17584

    return network;
}

17585 17586 17587

static virNWFilterPtr
vshCommandOptNWFilterBy(vshControl *ctl, const vshCmd *cmd,
17588
                        const char **name, int flag)
17589 17590
{
    virNWFilterPtr nwfilter = NULL;
17591
    const char *n = NULL;
17592 17593 17594 17595
    const char *optname = "nwfilter";
    if (!cmd_has_option (ctl, cmd, optname))
        return NULL;

17596
    if (vshCommandOptString(cmd, optname, &n) <= 0)
17597 17598
        return NULL;

17599
    vshDebug(ctl, VSH_ERR_INFO, "%s: found option <%s>: %s\n",
17600 17601 17602 17603 17604 17605 17606
             cmd->def->name, optname, n);

    if (name)
        *name = n;

    /* try it by UUID */
    if ((flag & VSH_BYUUID) && (strlen(n) == VIR_UUID_STRING_BUFLEN-1)) {
17607
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as nwfilter UUID\n",
17608 17609 17610 17611 17612
                 cmd->def->name, optname);
        nwfilter = virNWFilterLookupByUUIDString(ctl->conn, n);
    }
    /* try it by NAME */
    if (nwfilter == NULL && (flag & VSH_BYNAME)) {
17613
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as nwfilter NAME\n",
17614 17615 17616 17617 17618 17619 17620 17621 17622 17623
                 cmd->def->name, optname);
        nwfilter = virNWFilterLookupByName(ctl->conn, n);
    }

    if (!nwfilter)
        vshError(ctl, _("failed to get nwfilter '%s'"), n);

    return nwfilter;
}

17624 17625
static virInterfacePtr
vshCommandOptInterfaceBy(vshControl *ctl, const vshCmd *cmd,
17626
                         const char *optname,
17627
                         const char **name, int flag)
17628 17629
{
    virInterfacePtr iface = NULL;
17630
    const char *n = NULL;
17631 17632 17633

    if (!optname)
       optname = "interface";
17634 17635 17636
    if (!cmd_has_option (ctl, cmd, optname))
        return NULL;

17637
    if (vshCommandOptString(cmd, optname, &n) <= 0)
17638 17639
        return NULL;

17640
    vshDebug(ctl, VSH_ERR_INFO, "%s: found option <%s>: %s\n",
17641 17642 17643 17644 17645 17646
             cmd->def->name, optname, n);

    if (name)
        *name = n;

    /* try it by NAME */
17647
    if ((flag & VSH_BYNAME)) {
17648
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as interface NAME\n",
17649 17650 17651 17652 17653
                 cmd->def->name, optname);
        iface = virInterfaceLookupByName(ctl->conn, n);
    }
    /* try it by MAC */
    if ((iface == NULL) && (flag & VSH_BYMAC)) {
17654
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as interface MAC\n",
17655 17656 17657 17658 17659
                 cmd->def->name, optname);
        iface = virInterfaceLookupByMACString(ctl->conn, n);
    }

    if (!iface)
17660
        vshError(ctl, _("failed to get interface '%s'"), n);
17661 17662 17663 17664

    return iface;
}

17665
static virStoragePoolPtr
17666
vshCommandOptPoolBy(vshControl *ctl, const vshCmd *cmd, const char *optname,
17667
                    const char **name, int flag)
17668 17669
{
    virStoragePoolPtr pool = NULL;
17670
    const char *n = NULL;
17671

17672
    if (vshCommandOptString(cmd, optname, &n) <= 0)
17673 17674
        return NULL;

17675
    vshDebug(ctl, VSH_ERR_INFO, "%s: found option <%s>: %s\n",
17676 17677 17678 17679 17680 17681
             cmd->def->name, optname, n);

    if (name)
        *name = n;

    /* try it by UUID */
17682
    if ((flag & VSH_BYUUID) && (strlen(n) == VIR_UUID_STRING_BUFLEN-1)) {
17683
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as pool UUID\n",
17684
                 cmd->def->name, optname);
17685 17686 17687
        pool = virStoragePoolLookupByUUIDString(ctl->conn, n);
    }
    /* try it by NAME */
17688
    if (pool == NULL && (flag & VSH_BYNAME)) {
17689
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as pool NAME\n",
17690 17691 17692 17693 17694
                 cmd->def->name, optname);
        pool = virStoragePoolLookupByName(ctl->conn, n);
    }

    if (!pool)
17695
        vshError(ctl, _("failed to get pool '%s'"), n);
17696 17697 17698 17699 17700

    return pool;
}

static virStorageVolPtr
17701
vshCommandOptVolBy(vshControl *ctl, const vshCmd *cmd,
17702 17703
                   const char *optname,
                   const char *pooloptname,
17704
                   const char **name, int flag)
17705 17706 17707
{
    virStorageVolPtr vol = NULL;
    virStoragePoolPtr pool = NULL;
17708
    const char *n = NULL, *p = NULL;
17709

17710
    if (vshCommandOptString(cmd, optname, &n) <= 0)
17711 17712
        return NULL;

17713
    if (pooloptname != NULL && vshCommandOptString(cmd, pooloptname, &p) < 0) {
17714
        vshError(ctl, "%s", _("missing option"));
17715
        return NULL;
17716
    }
17717 17718 17719 17720

    if (p)
        pool = vshCommandOptPoolBy(ctl, cmd, pooloptname, name, flag);

17721
    vshDebug(ctl, VSH_ERR_DEBUG, "%s: found option <%s>: %s\n",
17722 17723 17724 17725 17726
             cmd->def->name, optname, n);

    if (name)
        *name = n;

17727
    /* try it by name */
17728
    if (pool && (flag & VSH_BYNAME)) {
17729
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as vol name\n",
17730 17731 17732
                 cmd->def->name, optname);
        vol = virStorageVolLookupByName(pool, n);
    }
17733
    /* try it by key */
17734
    if (vol == NULL && (flag & VSH_BYUUID)) {
17735
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as vol key\n",
17736 17737 17738
                 cmd->def->name, optname);
        vol = virStorageVolLookupByKey(ctl->conn, n);
    }
17739
    /* try it by path */
17740
    if (vol == NULL && (flag & VSH_BYUUID)) {
17741
        vshDebug(ctl, VSH_ERR_DEBUG, "%s: <%s> trying as vol path\n",
17742 17743 17744 17745
                 cmd->def->name, optname);
        vol = virStorageVolLookupByPath(ctl->conn, n);
    }

17746 17747 17748 17749
    if (!vol) {
        if (pool)
            vshError(ctl, _("failed to get vol '%s'"), n);
        else
17750 17751
            vshError(ctl, _("failed to get vol '%s', specifying --%s "
                            "might help"), n, pooloptname);
17752
    }
17753 17754 17755 17756 17757 17758 17759

    if (pool)
        virStoragePoolFree(pool);

    return vol;
}

17760
static virSecretPtr
17761
vshCommandOptSecret(vshControl *ctl, const vshCmd *cmd, const char **name)
17762 17763
{
    virSecretPtr secret = NULL;
17764
    const char *n = NULL;
17765 17766 17767 17768 17769
    const char *optname = "secret";

    if (!cmd_has_option (ctl, cmd, optname))
        return NULL;

17770
    if (vshCommandOptString(cmd, optname, &n) <= 0)
17771 17772
        return NULL;

17773 17774
    vshDebug(ctl, VSH_ERR_DEBUG,
             "%s: found option <%s>: %s\n", cmd->def->name, optname, n);
17775 17776 17777 17778 17779 17780 17781

    if (name != NULL)
        *name = n;

    secret = virSecretLookupByUUIDString(ctl->conn, n);

    if (secret == NULL)
17782
        vshError(ctl, _("failed to get secret '%s'"), n);
17783 17784 17785 17786

    return secret;
}

K
Karel Zak 已提交
17787 17788 17789
/*
 * Executes command(s) and returns return code from last command
 */
E
Eric Blake 已提交
17790
static bool
17791
vshCommandRun(vshControl *ctl, const vshCmd *cmd)
17792
{
E
Eric Blake 已提交
17793
    bool ret = true;
17794 17795

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

17799 17800
        if ((ctl->conn == NULL || disconnected) &&
            !(cmd->def->flags & VSH_CMD_FLAG_NOCONNECT))
17801 17802
            vshReconnect(ctl);

17803
        if (enable_timing)
K
Karel Zak 已提交
17804
            GETTIMEOFDAY(&before);
17805

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

17808
        if (enable_timing)
K
Karel Zak 已提交
17809
            GETTIMEOFDAY(&after);
17810

17811 17812 17813 17814 17815 17816 17817 17818 17819 17820
        /* 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++;

17821
        if (!ret)
J
John Levon 已提交
17822 17823
            virshReportError(ctl);

17824
        if (!ret && disconnected != 0)
17825 17826
            vshReconnect(ctl);

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

17830
        if (enable_timing)
17831
            vshPrint(ctl, _("\n(Time: %.3f ms)\n\n"),
17832 17833
                     DIFF_MSEC(&after, &before));
        else
K
Karel Zak 已提交
17834
            vshPrintExtra(ctl, "\n");
K
Karel Zak 已提交
17835 17836 17837 17838 17839 17840
        cmd = cmd->next;
    }
    return ret;
}

/* ---------------
17841
 * Command parsing
K
Karel Zak 已提交
17842 17843 17844
 * ---------------
 */

17845 17846 17847 17848 17849 17850 17851 17852 17853 17854
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 {
    vshCommandToken (*getNextArg)(vshControl *, struct __vshCommandParser *,
                                  char **);
L
Lai Jiangshan 已提交
17855
    /* vshCommandStringGetArg() */
17856
    char *pos;
L
Lai Jiangshan 已提交
17857 17858 17859
    /* vshCommandArgvGetArg() */
    char **arg_pos;
    char **arg_end;
17860 17861
} vshCommandParser;

E
Eric Blake 已提交
17862
static bool
17863
vshCommandParse(vshControl *ctl, vshCommandParser *parser)
17864
{
K
Karel Zak 已提交
17865 17866 17867
    char *tkdata = NULL;
    vshCmd *clast = NULL;
    vshCmdOpt *first = NULL;
17868

K
Karel Zak 已提交
17869 17870 17871 17872
    if (ctl->cmd) {
        vshCommandFree(ctl->cmd);
        ctl->cmd = NULL;
    }
17873

17874
    while (1) {
K
Karel Zak 已提交
17875
        vshCmdOpt *last = NULL;
17876
        const vshCmdDef *cmd = NULL;
17877
        vshCommandToken tk;
L
Lai Jiangshan 已提交
17878
        bool data_only = false;
17879 17880 17881
        uint32_t opts_need_arg = 0;
        uint32_t opts_required = 0;
        uint32_t opts_seen = 0;
17882

K
Karel Zak 已提交
17883
        first = NULL;
17884

17885
        while (1) {
17886
            const vshCmdOptDef *opt = NULL;
17887

K
Karel Zak 已提交
17888
            tkdata = NULL;
17889
            tk = parser->getNextArg(ctl, parser, &tkdata);
17890 17891

            if (tk == VSH_TK_ERROR)
K
Karel Zak 已提交
17892
                goto syntaxError;
H
Hu Tao 已提交
17893 17894
            if (tk != VSH_TK_ARG) {
                VIR_FREE(tkdata);
17895
                break;
H
Hu Tao 已提交
17896
            }
17897 17898

            if (cmd == NULL) {
K
Karel Zak 已提交
17899 17900
                /* first token must be command name */
                if (!(cmd = vshCmddefSearch(tkdata))) {
17901
                    vshError(ctl, _("unknown command: '%s'"), tkdata);
17902
                    goto syntaxError;   /* ... or ignore this command only? */
K
Karel Zak 已提交
17903
                }
17904 17905 17906 17907 17908 17909 17910
                if (vshCmddefOptParse(cmd, &opts_need_arg,
                                      &opts_required) < 0) {
                    vshError(ctl,
                             _("internal error: bad options in command: '%s'"),
                             tkdata);
                    goto syntaxError;
                }
17911
                VIR_FREE(tkdata);
L
Lai Jiangshan 已提交
17912 17913 17914 17915
            } else if (data_only) {
                goto get_data;
            } else if (tkdata[0] == '-' && tkdata[1] == '-' &&
                       c_isalnum(tkdata[2])) {
17916
                char *optstr = strchr(tkdata + 2, '=');
17917 17918
                int opt_index;

17919 17920 17921 17922
                if (optstr) {
                    *optstr = '\0'; /* convert the '=' to '\0' */
                    optstr = vshStrdup(ctl, optstr + 1);
                }
17923
                if (!(opt = vshCmddefGetOption(ctl, cmd, tkdata + 2,
17924
                                               &opts_seen, &opt_index))) {
17925
                    VIR_FREE(optstr);
K
Karel Zak 已提交
17926 17927
                    goto syntaxError;
                }
17928
                VIR_FREE(tkdata);
K
Karel Zak 已提交
17929 17930 17931

                if (opt->type != VSH_OT_BOOL) {
                    /* option data */
17932 17933 17934
                    if (optstr)
                        tkdata = optstr;
                    else
17935
                        tk = parser->getNextArg(ctl, parser, &tkdata);
17936
                    if (tk == VSH_TK_ERROR)
K
Karel Zak 已提交
17937
                        goto syntaxError;
17938
                    if (tk != VSH_TK_ARG) {
17939
                        vshError(ctl,
17940
                                 _("expected syntax: --%s <%s>"),
17941 17942
                                 opt->name,
                                 opt->type ==
17943
                                 VSH_OT_INT ? _("number") : _("string"));
K
Karel Zak 已提交
17944 17945
                        goto syntaxError;
                    }
17946 17947
                    if (opt->type != VSH_OT_ARGV)
                        opts_need_arg &= ~(1 << opt_index);
17948 17949 17950 17951 17952 17953 17954 17955
                } else {
                    tkdata = NULL;
                    if (optstr) {
                        vshError(ctl, _("invalid '=' after option --%s"),
                                opt->name);
                        VIR_FREE(optstr);
                        goto syntaxError;
                    }
K
Karel Zak 已提交
17956
                }
L
Lai Jiangshan 已提交
17957 17958 17959 17960
            } else if (tkdata[0] == '-' && tkdata[1] == '-' &&
                       tkdata[2] == '\0') {
                data_only = true;
                continue;
17961
            } else {
L
Lai Jiangshan 已提交
17962
get_data:
17963 17964
                if (!(opt = vshCmddefGetData(cmd, &opts_need_arg,
                                             &opts_seen))) {
17965
                    vshError(ctl, _("unexpected data '%s'"), tkdata);
K
Karel Zak 已提交
17966 17967 17968 17969 17970
                    goto syntaxError;
                }
            }
            if (opt) {
                /* save option */
17971
                vshCmdOpt *arg = vshMalloc(ctl, sizeof(vshCmdOpt));
17972

K
Karel Zak 已提交
17973 17974 17975 17976
                arg->def = opt;
                arg->data = tkdata;
                arg->next = NULL;
                tkdata = NULL;
17977

K
Karel Zak 已提交
17978 17979 17980 17981 17982
                if (!first)
                    first = arg;
                if (last)
                    last->next = arg;
                last = arg;
17983

17984
                vshDebug(ctl, VSH_ERR_INFO, "%s: %s(%s): %s\n",
17985 17986
                         cmd->name,
                         opt->name,
17987 17988
                         opt->type != VSH_OT_BOOL ? _("optdata") : _("bool"),
                         opt->type != VSH_OT_BOOL ? arg->data : _("(none)"));
K
Karel Zak 已提交
17989 17990
            }
        }
17991

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

K
Karel Zak 已提交
17996 17997 17998 17999
            c->opts = first;
            c->def = cmd;
            c->next = NULL;

18000
            if (vshCommandCheckOpts(ctl, c, opts_required, opts_seen) < 0) {
18001
                VIR_FREE(c);
18002
                goto syntaxError;
18003
            }
18004

K
Karel Zak 已提交
18005 18006 18007 18008 18009 18010
            if (!ctl->cmd)
                ctl->cmd = c;
            if (clast)
                clast->next = c;
            clast = c;
        }
18011 18012 18013

        if (tk == VSH_TK_END)
            break;
K
Karel Zak 已提交
18014
    }
18015

E
Eric Blake 已提交
18016
    return true;
K
Karel Zak 已提交
18017

18018
 syntaxError:
18019
    if (ctl->cmd) {
K
Karel Zak 已提交
18020
        vshCommandFree(ctl->cmd);
18021 18022
        ctl->cmd = NULL;
    }
K
Karel Zak 已提交
18023 18024
    if (first)
        vshCommandOptFree(first);
18025
    VIR_FREE(tkdata);
E
Eric Blake 已提交
18026
    return false;
K
Karel Zak 已提交
18027 18028
}

18029 18030 18031 18032 18033 18034 18035 18036 18037 18038 18039 18040 18041 18042 18043 18044 18045 18046
/* --------------------
 * 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 已提交
18047 18048
static bool
vshCommandArgvParse(vshControl *ctl, int nargs, char **argv)
18049 18050 18051 18052
{
    vshCommandParser parser;

    if (nargs <= 0)
E
Eric Blake 已提交
18053
        return false;
18054 18055 18056 18057 18058 18059 18060 18061 18062 18063 18064 18065 18066 18067 18068 18069 18070 18071 18072 18073 18074 18075 18076 18077 18078 18079 18080 18081 18082 18083 18084 18085 18086 18087 18088 18089 18090 18091 18092 18093 18094 18095 18096 18097 18098 18099 18100 18101 18102 18103 18104 18105 18106 18107 18108 18109 18110 18111 18112 18113 18114 18115 18116 18117 18118 18119 18120 18121 18122 18123 18124 18125

    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 已提交
18126 18127
static bool
vshCommandStringParse(vshControl *ctl, char *cmdstr)
18128 18129 18130 18131
{
    vshCommandParser parser;

    if (cmdstr == NULL || *cmdstr == '\0')
E
Eric Blake 已提交
18132
        return false;
18133 18134 18135 18136 18137 18138

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

K
Karel Zak 已提交
18139
/* ---------------
18140
 * Misc utils
K
Karel Zak 已提交
18141 18142
 * ---------------
 */
18143 18144 18145 18146 18147 18148 18149 18150 18151 18152 18153 18154 18155 18156 18157 18158 18159 18160 18161 18162 18163 18164 18165 18166 18167 18168 18169 18170
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;
}

K
Karel Zak 已提交
18171
static const char *
18172 18173
vshDomainStateToString(int state)
{
18174 18175
    /* Can't use virDomainStateTypeToString, because we want to mark
     * strings for translation.  */
18176
    switch ((virDomainState) state) {
18177
    case VIR_DOMAIN_RUNNING:
18178
        return N_("running");
18179
    case VIR_DOMAIN_BLOCKED:
18180
        return N_("idle");
18181
    case VIR_DOMAIN_PAUSED:
18182
        return N_("paused");
18183
    case VIR_DOMAIN_SHUTDOWN:
18184
        return N_("in shutdown");
18185
    case VIR_DOMAIN_SHUTOFF:
18186
        return N_("shut off");
18187
    case VIR_DOMAIN_CRASHED:
18188
        return N_("crashed");
18189
    case VIR_DOMAIN_NOSTATE:
18190
    default:
18191
        ;/*FALLTHROUGH*/
K
Karel Zak 已提交
18192
    }
18193
    return N_("no state");  /* = dom0 state */
K
Karel Zak 已提交
18194 18195
}

18196 18197 18198 18199 18200 18201 18202
static const char *
vshDomainStateReasonToString(int state, int reason)
{
    switch ((virDomainState) state) {
    case VIR_DOMAIN_NOSTATE:
        switch ((virDomainNostateReason) reason) {
        case VIR_DOMAIN_NOSTATE_UNKNOWN:
18203
        case VIR_DOMAIN_NOSTATE_LAST:
18204 18205 18206 18207 18208 18209 18210 18211 18212 18213 18214 18215 18216 18217 18218 18219 18220 18221 18222 18223 18224
            ;
        }
        break;

    case VIR_DOMAIN_RUNNING:
        switch ((virDomainRunningReason) reason) {
        case VIR_DOMAIN_RUNNING_BOOTED:
            return N_("booted");
        case VIR_DOMAIN_RUNNING_MIGRATED:
            return N_("migrated");
        case VIR_DOMAIN_RUNNING_RESTORED:
            return N_("restored");
        case VIR_DOMAIN_RUNNING_FROM_SNAPSHOT:
            return N_("from snapshot");
        case VIR_DOMAIN_RUNNING_UNPAUSED:
            return N_("unpaused");
        case VIR_DOMAIN_RUNNING_MIGRATION_CANCELED:
            return N_("migration canceled");
        case VIR_DOMAIN_RUNNING_SAVE_CANCELED:
            return N_("save canceled");
        case VIR_DOMAIN_RUNNING_UNKNOWN:
18225
        case VIR_DOMAIN_RUNNING_LAST:
18226 18227 18228 18229 18230 18231 18232
            ;
        }
        break;

    case VIR_DOMAIN_BLOCKED:
        switch ((virDomainBlockedReason) reason) {
        case VIR_DOMAIN_BLOCKED_UNKNOWN:
18233
        case VIR_DOMAIN_BLOCKED_LAST:
18234 18235 18236 18237 18238 18239 18240 18241 18242 18243 18244 18245 18246 18247 18248 18249 18250 18251 18252 18253
            ;
        }
        break;

    case VIR_DOMAIN_PAUSED:
        switch ((virDomainPausedReason) reason) {
        case VIR_DOMAIN_PAUSED_USER:
            return N_("user");
        case VIR_DOMAIN_PAUSED_MIGRATION:
            return N_("migrating");
        case VIR_DOMAIN_PAUSED_SAVE:
            return N_("saving");
        case VIR_DOMAIN_PAUSED_DUMP:
            return N_("dumping");
        case VIR_DOMAIN_PAUSED_IOERROR:
            return N_("I/O error");
        case VIR_DOMAIN_PAUSED_WATCHDOG:
            return N_("watchdog");
        case VIR_DOMAIN_PAUSED_FROM_SNAPSHOT:
            return N_("from snapshot");
18254 18255
        case VIR_DOMAIN_PAUSED_SHUTTING_DOWN:
            return N_("shutting down");
18256
        case VIR_DOMAIN_PAUSED_UNKNOWN:
18257
        case VIR_DOMAIN_PAUSED_LAST:
18258 18259 18260 18261 18262 18263 18264 18265 18266
            ;
        }
        break;

    case VIR_DOMAIN_SHUTDOWN:
        switch ((virDomainShutdownReason) reason) {
        case VIR_DOMAIN_SHUTDOWN_USER:
            return N_("user");
        case VIR_DOMAIN_SHUTDOWN_UNKNOWN:
18267
        case VIR_DOMAIN_SHUTDOWN_LAST:
18268 18269 18270 18271 18272 18273 18274 18275 18276 18277 18278 18279 18280 18281 18282 18283 18284 18285 18286 18287 18288
            ;
        }
        break;

    case VIR_DOMAIN_SHUTOFF:
        switch ((virDomainShutoffReason) reason) {
        case VIR_DOMAIN_SHUTOFF_SHUTDOWN:
            return N_("shutdown");
        case VIR_DOMAIN_SHUTOFF_DESTROYED:
            return N_("destroyed");
        case VIR_DOMAIN_SHUTOFF_CRASHED:
            return N_("crashed");
        case VIR_DOMAIN_SHUTOFF_MIGRATED:
            return N_("migrated");
        case VIR_DOMAIN_SHUTOFF_SAVED:
            return N_("saved");
        case VIR_DOMAIN_SHUTOFF_FAILED:
            return N_("failed");
        case VIR_DOMAIN_SHUTOFF_FROM_SNAPSHOT:
            return N_("from snapshot");
        case VIR_DOMAIN_SHUTOFF_UNKNOWN:
18289
        case VIR_DOMAIN_SHUTOFF_LAST:
18290 18291 18292 18293 18294 18295 18296
            ;
        }
        break;

    case VIR_DOMAIN_CRASHED:
        switch ((virDomainCrashedReason) reason) {
        case VIR_DOMAIN_CRASHED_UNKNOWN:
18297
        case VIR_DOMAIN_CRASHED_LAST:
18298 18299 18300
            ;
        }
        break;
18301

18302
    case VIR_DOMAIN_LAST:
18303
        ;
18304 18305 18306 18307 18308
    }

    return N_("unknown");
}

18309 18310 18311 18312 18313 18314 18315 18316 18317 18318 18319 18320 18321 18322 18323 18324 18325 18326 18327 18328 18329 18330 18331 18332 18333 18334 18335 18336 18337 18338 18339 18340 18341 18342 18343 18344 18345 18346 18347 18348 18349 18350 18351 18352 18353 18354 18355 18356 18357 18358 18359 18360 18361 18362 18363 18364 18365 18366 18367
/* extract description or title from domain xml */
static char *
vshGetDomainDescription(vshControl *ctl, virDomainPtr dom, bool title,
                        unsigned int flags)
{
    char *desc = NULL;
    char *domxml = NULL;
    virErrorPtr err = NULL;
    xmlDocPtr doc = NULL;
    xmlXPathContextPtr ctxt = NULL;
    int type;

    if (title)
        type = VIR_DOMAIN_METADATA_TITLE;
    else
        type = VIR_DOMAIN_METADATA_DESCRIPTION;

    if ((desc = virDomainGetMetadata(dom, type, NULL, flags))) {
        return desc;
    } else {
        err = virGetLastError();

        if (err && err->code == VIR_ERR_NO_DOMAIN_METADATA) {
            desc = vshStrdup(ctl, "");
            virResetLastError();
            return desc;
        }

        if (err && err->code != VIR_ERR_NO_SUPPORT)
            return desc;
    }

    /* fall back to xml */
    /* get domain's xml description and extract the title/description */
    if (!(domxml = virDomainGetXMLDesc(dom, flags))) {
        vshError(ctl, "%s", _("Failed to retrieve domain XML"));
        goto cleanup;
    }
    doc = virXMLParseStringCtxt(domxml, _("(domain_definition)"), &ctxt);
    if (!doc) {
        vshError(ctl, "%s", _("Couldn't parse domain XML"));
        goto cleanup;
    }
    if (title)
        desc = virXPathString("string(./title[1])", ctxt);
    else
        desc = virXPathString("string(./description[1])", ctxt);

    if (!desc)
        desc = vshStrdup(ctl, "");

cleanup:
    VIR_FREE(domxml);
    xmlXPathFreeContext(ctxt);
    xmlFreeDoc(doc);

    return desc;
}

18368 18369
/* Return a non-NULL string representation of a typed parameter; exit
 * if we are out of memory.  */
18370 18371 18372 18373 18374 18375 18376 18377 18378 18379 18380 18381 18382 18383 18384 18385 18386 18387 18388 18389 18390 18391 18392 18393 18394 18395 18396 18397 18398 18399 18400
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;

18401 18402 18403 18404
    case VIR_TYPED_PARAM_STRING:
        str = vshStrdup(ctl, item->value.s);
        break;

18405
    default:
18406
        vshError(ctl, _("unimplemented parameter type %d"), item->type);
18407 18408
    }

18409
    if (ret < 0) {
18410
        vshError(ctl, "%s", _("Out of memory"));
18411 18412
        exit(EXIT_FAILURE);
    }
18413 18414 18415 18416 18417 18418 18419 18420 18421 18422 18423 18424 18425 18426 18427 18428 18429 18430 18431 18432 18433 18434 18435
    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;
}

18436 18437 18438 18439 18440 18441 18442 18443 18444 18445 18446 18447
static const char *
vshDomainControlStateToString(int state)
{
    switch ((virDomainControlState) state) {
    case VIR_DOMAIN_CONTROL_OK:
        return N_("ok");
    case VIR_DOMAIN_CONTROL_JOB:
        return N_("background job");
    case VIR_DOMAIN_CONTROL_OCCUPIED:
        return N_("occupied");
    case VIR_DOMAIN_CONTROL_ERROR:
        return N_("error");
18448 18449
    default:
        ;
18450 18451 18452 18453 18454
    }

    return N_("unknown");
}

18455 18456 18457 18458
static const char *
vshDomainVcpuStateToString(int state)
{
    switch (state) {
18459
    case VIR_VCPU_OFFLINE:
18460
        return N_("offline");
18461
    case VIR_VCPU_BLOCKED:
18462
        return N_("idle");
18463
    case VIR_VCPU_RUNNING:
18464
        return N_("running");
18465
    default:
18466
        ;/*FALLTHROUGH*/
18467
    }
18468
    return N_("no state");
18469 18470
}

E
Eric Blake 已提交
18471
static bool
18472
vshConnectionUsability(vshControl *ctl, virConnectPtr conn)
18473
{
18474 18475
    /* TODO: use something like virConnectionState() to
     *       check usability of the connection
K
Karel Zak 已提交
18476 18477
     */
    if (!conn) {
18478
        vshError(ctl, "%s", _("no valid connection"));
E
Eric Blake 已提交
18479
        return false;
K
Karel Zak 已提交
18480
    }
E
Eric Blake 已提交
18481
    return true;
K
Karel Zak 已提交
18482 18483
}

K
Karel Zak 已提交
18484
static void
18485
vshDebug(vshControl *ctl, int level, const char *format, ...)
18486
{
K
Karel Zak 已提交
18487
    va_list ap;
18488
    char *str;
K
Karel Zak 已提交
18489

18490 18491 18492 18493 18494 18495 18496
    /* Aligning log levels to that of libvirt.
     * Traces with levels >=  user-specified-level
     * gets logged into file
     */
    if (level < ctl->debug)
        return;

18497
    va_start(ap, format);
18498
    vshOutputLogFile(ctl, level, format, ap);
18499 18500
    va_end(ap);

K
Karel Zak 已提交
18501
    va_start(ap, format);
18502 18503 18504 18505 18506
    if (virVasprintf(&str, format, ap) < 0) {
        /* Skip debug messages on low memory */
        va_end(ap);
        return;
    }
K
Karel Zak 已提交
18507
    va_end(ap);
18508 18509
    fputs(str, stdout);
    VIR_FREE(str);
K
Karel Zak 已提交
18510 18511 18512
}

static void
18513
vshPrintExtra(vshControl *ctl, const char *format, ...)
18514
{
K
Karel Zak 已提交
18515
    va_list ap;
18516
    char *str;
18517

18518
    if (ctl && ctl->quiet)
K
Karel Zak 已提交
18519
        return;
18520

K
Karel Zak 已提交
18521
    va_start(ap, format);
18522 18523 18524 18525 18526
    if (virVasprintf(&str, format, ap) < 0) {
        vshError(ctl, "%s", _("Out of memory"));
        va_end(ap);
        return;
    }
K
Karel Zak 已提交
18527
    va_end(ap);
18528
    fputs(str, stdout);
18529
    VIR_FREE(str);
K
Karel Zak 已提交
18530 18531
}

K
Karel Zak 已提交
18532

K
Karel Zak 已提交
18533
static void
18534
vshError(vshControl *ctl, const char *format, ...)
18535
{
K
Karel Zak 已提交
18536
    va_list ap;
18537
    char *str;
18538

18539 18540 18541 18542 18543
    if (ctl != NULL) {
        va_start(ap, format);
        vshOutputLogFile(ctl, VSH_ERR_ERROR, format, ap);
        va_end(ap);
    }
18544

18545 18546 18547 18548
    /* 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);
18549
    fputs(_("error: "), stderr);
18550

K
Karel Zak 已提交
18551
    va_start(ap, format);
18552 18553 18554
    /* We can't recursively call vshError on an OOM situation, so ignore
       failure here. */
    ignore_value(virVasprintf(&str, format, ap));
K
Karel Zak 已提交
18555 18556
    va_end(ap);

18557
    fprintf(stderr, "%s\n", NULLSTR(str));
18558
    fflush(stderr);
18559
    VIR_FREE(str);
K
Karel Zak 已提交
18560 18561
}

18562

J
Jiri Denemark 已提交
18563 18564 18565 18566 18567
static void
vshEventLoop(void *opaque)
{
    vshControl *ctl = opaque;

18568 18569 18570 18571 18572 18573 18574 18575 18576 18577
    while (1) {
        bool quit;
        virMutexLock(&ctl->lock);
        quit = ctl->quit;
        virMutexUnlock(&ctl->lock);

        if (quit)
            break;

        if (virEventRunDefaultImpl() < 0)
J
Jiri Denemark 已提交
18578 18579 18580 18581 18582
            virshReportError(ctl);
    }
}


K
Karel Zak 已提交
18583
/*
18584
 * Initialize connection.
K
Karel Zak 已提交
18585
 */
E
Eric Blake 已提交
18586
static bool
18587
vshInit(vshControl *ctl)
18588
{
18589 18590
    char *debugEnv;

K
Karel Zak 已提交
18591
    if (ctl->conn)
E
Eric Blake 已提交
18592
        return false;
K
Karel Zak 已提交
18593

J
Jiri Denemark 已提交
18594
    if (ctl->debug == VSH_DEBUG_DEFAULT) {
18595 18596 18597
        /* log level not set from commandline, check env variable */
        debugEnv = getenv("VIRSH_DEBUG");
        if (debugEnv) {
J
Jiri Denemark 已提交
18598 18599 18600
            int debug;
            if (virStrToLong_i(debugEnv, NULL, 10, &debug) < 0 ||
                debug < VSH_ERR_DEBUG || debug > VSH_ERR_ERROR) {
18601 18602
                vshError(ctl, "%s",
                         _("VIRSH_DEBUG not set with a valid numeric value"));
J
Jiri Denemark 已提交
18603 18604
            } else {
                ctl->debug = debug;
18605 18606 18607 18608 18609 18610 18611 18612 18613 18614 18615 18616
            }
        }
    }

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

18617 18618
    vshOpenLogFile(ctl);

18619 18620
    /* set up the library error handler */
    virSetErrorFunc(NULL, virshErrorHandler);
18621

18622 18623 18624
    /* set up the signals handlers to catch disconnections */
    vshSetupSignals();

18625
    if (virEventRegisterDefaultImpl() < 0)
E
Eric Blake 已提交
18626
        return false;
18627

J
Jiri Denemark 已提交
18628 18629 18630 18631
    if (virThreadCreate(&ctl->eventLoop, true, vshEventLoop, ctl) < 0)
        return false;
    ctl->eventLoopStarted = true;

18632 18633 18634 18635
    if (ctl->name) {
        ctl->conn = virConnectOpenAuth(ctl->name,
                                       virConnectAuthPtrDefault,
                                       ctl->readonly ? VIR_CONNECT_RO : 0);
18636

18637 18638 18639 18640 18641 18642 18643 18644 18645 18646 18647
        /* 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;
        }
18648
    }
K
Karel Zak 已提交
18649

E
Eric Blake 已提交
18650
    return true;
K
Karel Zak 已提交
18651 18652
}

18653 18654
#define LOGFILE_FLAGS (O_WRONLY | O_APPEND | O_CREAT | O_SYNC)

18655 18656 18657 18658 18659 18660 18661 18662 18663 18664 18665 18666 18667 18668 18669 18670 18671 18672 18673
/**
 * 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:
18674
                vshError(ctl, "%s",
J
Jim Meyering 已提交
18675
                         _("failed to get the log file information"));
18676
                exit(EXIT_FAILURE);
18677 18678 18679
        }
    } else {
        if (!S_ISREG(st.st_mode)) {
18680 18681
            vshError(ctl, "%s", _("the log path is not a file"));
            exit(EXIT_FAILURE);
18682 18683 18684 18685
        }
    }

    /* log file open */
18686
    if ((ctl->log_fd = open(ctl->logfile, LOGFILE_FLAGS, FILE_MODE)) < 0) {
18687
        vshError(ctl, "%s",
J
Jim Meyering 已提交
18688
                 _("failed to open the log file. check the log file path"));
18689
        exit(EXIT_FAILURE);
18690 18691 18692 18693 18694 18695 18696 18697 18698
    }
}

/**
 * vshOutputLogFile:
 *
 * Outputting an error to log file.
 */
static void
18699 18700
vshOutputLogFile(vshControl *ctl, int log_level, const char *msg_format,
                 va_list ap)
18701
{
18702 18703 18704
    virBuffer buf = VIR_BUFFER_INITIALIZER;
    char *str;
    size_t len;
18705 18706 18707 18708 18709 18710 18711 18712 18713 18714 18715 18716 18717 18718
    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);
18719
    virBufferAsprintf(&buf, "[%d.%02d.%02d %02d:%02d:%02d %s %d] ",
18720 18721 18722 18723 18724 18725
                      (1900 + stTm->tm_year),
                      (1 + stTm->tm_mon),
                      stTm->tm_mday,
                      stTm->tm_hour,
                      stTm->tm_min,
                      stTm->tm_sec,
18726 18727
                      SIGN_NAME,
                      (int) getpid());
18728 18729 18730 18731 18732 18733 18734 18735 18736 18737 18738 18739 18740 18741 18742 18743 18744 18745 18746 18747
    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;
    }
18748 18749 18750
    virBufferAsprintf(&buf, "%s ", lvl);
    virBufferVasprintf(&buf, msg_format, ap);
    virBufferAddChar(&buf, '\n');
18751

18752 18753
    if (virBufferError(&buf))
        goto error;
18754

18755 18756 18757 18758 18759
    str = virBufferContentAndReset(&buf);
    len = strlen(str);
    if (len > 1 && str[len - 2] == '\n') {
        str[len - 1] = '\0';
        len--;
18760
    }
18761

18762 18763 18764 18765 18766 18767 18768 18769 18770 18771 18772
    /* 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);
18773 18774 18775 18776 18777 18778 18779 18780 18781 18782 18783
}

/**
 * vshCloseLogFile:
 *
 * Close log file.
 */
static void
vshCloseLogFile(vshControl *ctl)
{
    /* log file close */
18784 18785 18786
    if (VIR_CLOSE(ctl->log_fd) < 0) {
        vshError(ctl, _("%s: failed to write log file: %s"),
                 ctl->logfile ? ctl->logfile : "?", strerror (errno));
18787 18788 18789
    }

    if (ctl->logfile) {
18790
        VIR_FREE(ctl->logfile);
18791 18792 18793 18794
        ctl->logfile = NULL;
    }
}

18795
#ifdef USE_READLINE
18796

K
Karel Zak 已提交
18797 18798 18799 18800 18801
/* -----------------
 * Readline stuff
 * -----------------
 */

18802
/*
K
Karel Zak 已提交
18803 18804
 * Generator function for command completion.  STATE lets us
 * know whether to start from scratch; without any state
18805
 * (i.e. STATE == 0), then we start at the top of the list.
K
Karel Zak 已提交
18806 18807
 */
static char *
18808 18809
vshReadlineCommandGenerator(const char *text, int state)
{
18810
    static int grp_list_index, cmd_list_index, len;
K
Karel Zak 已提交
18811
    const char *name;
18812 18813
    const vshCmdGrp *grp;
    const vshCmdDef *cmds;
K
Karel Zak 已提交
18814 18815

    if (!state) {
18816 18817
        grp_list_index = 0;
        cmd_list_index = 0;
18818
        len = strlen(text);
K
Karel Zak 已提交
18819 18820
    }

18821 18822
    grp = cmdGroups;

K
Karel Zak 已提交
18823
    /* Return the next name which partially matches from the
18824
     * command list.
K
Karel Zak 已提交
18825
     */
18826 18827 18828 18829 18830 18831 18832 18833 18834 18835 18836 18837 18838 18839
    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 已提交
18840 18841 18842 18843 18844 18845 18846
    }

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

static char *
18847 18848
vshReadlineOptionsGenerator(const char *text, int state)
{
K
Karel Zak 已提交
18849
    static int list_index, len;
18850
    static const vshCmdDef *cmd = NULL;
K
Karel Zak 已提交
18851
    const char *name;
K
Karel Zak 已提交
18852 18853 18854 18855 18856 18857 18858 18859 18860

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

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

18861
        cmdname = vshCalloc(NULL, (p - rl_line_buffer) + 1, 1);
18862
        memcpy(cmdname, rl_line_buffer, p - rl_line_buffer);
K
Karel Zak 已提交
18863 18864 18865

        cmd = vshCmddefSearch(cmdname);
        list_index = 0;
18866
        len = strlen(text);
18867
        VIR_FREE(cmdname);
K
Karel Zak 已提交
18868 18869 18870 18871
    }

    if (!cmd)
        return NULL;
18872

18873 18874 18875
    if (!cmd->opts)
        return NULL;

K
Karel Zak 已提交
18876
    while ((name = cmd->opts[list_index].name)) {
18877
        const vshCmdOptDef *opt = &cmd->opts[list_index];
K
Karel Zak 已提交
18878
        char *res;
18879

K
Karel Zak 已提交
18880
        list_index++;
18881

18882
        if (opt->type == VSH_OT_DATA || opt->type == VSH_OT_ARGV)
K
Karel Zak 已提交
18883 18884
            /* ignore non --option */
            continue;
18885

K
Karel Zak 已提交
18886
        if (len > 2) {
18887
            if (STRNEQLEN(name, text + 2, len - 2))
K
Karel Zak 已提交
18888 18889
                continue;
        }
18890
        res = vshMalloc(NULL, strlen(name) + 3);
18891
        snprintf(res, strlen(name) + 3,  "--%s", name);
K
Karel Zak 已提交
18892 18893 18894 18895 18896 18897 18898 18899
        return res;
    }

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

static char **
18900 18901 18902
vshReadlineCompletion(const char *text, int start,
                      int end ATTRIBUTE_UNUSED)
{
K
Karel Zak 已提交
18903 18904
    char **matches = (char **) NULL;

18905
    if (start == 0)
K
Karel Zak 已提交
18906
        /* command name generator */
18907
        matches = rl_completion_matches(text, vshReadlineCommandGenerator);
K
Karel Zak 已提交
18908 18909
    else
        /* commands options */
18910
        matches = rl_completion_matches(text, vshReadlineOptionsGenerator);
K
Karel Zak 已提交
18911 18912 18913 18914
    return matches;
}


18915 18916
static int
vshReadlineInit(vshControl *ctl)
18917
{
18918 18919
    char *userdir = NULL;

K
Karel Zak 已提交
18920 18921 18922 18923 18924
    /* 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;
18925 18926 18927

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

    /* Prepare to read/write history from/to the ~/.virsh/history file */
18930
    userdir = virGetUserDirectory(getuid());
18931

18932 18933
    if (userdir == NULL) {
        vshError(ctl, "%s", _("Could not determine home directory"));
18934
        return -1;
18935
    }
18936 18937 18938

    if (virAsprintf(&ctl->historydir, "%s/.virsh", userdir) < 0) {
        vshError(ctl, "%s", _("Out of memory"));
18939
        VIR_FREE(userdir);
18940 18941 18942 18943 18944
        return -1;
    }

    if (virAsprintf(&ctl->historyfile, "%s/history", ctl->historydir) < 0) {
        vshError(ctl, "%s", _("Out of memory"));
18945
        VIR_FREE(userdir);
18946 18947 18948
        return -1;
    }

18949
    VIR_FREE(userdir);
18950 18951 18952 18953 18954 18955 18956 18957 18958 18959 18960 18961 18962 18963 18964 18965 18966 18967

    read_history(ctl->historyfile);

    return 0;
}

static void
vshReadlineDeinit (vshControl *ctl)
{
    if (ctl->historyfile != NULL) {
        if (mkdir(ctl->historydir, 0755) < 0 && errno != EEXIST) {
            char ebuf[1024];
            vshError(ctl, _("Failed to create '%s': %s"),
                     ctl->historydir, virStrerror(errno, ebuf, sizeof ebuf));
        } else
            write_history(ctl->historyfile);
    }

18968 18969
    VIR_FREE(ctl->historydir);
    VIR_FREE(ctl->historyfile);
K
Karel Zak 已提交
18970 18971
}

18972 18973 18974 18975 18976 18977
static char *
vshReadline (vshControl *ctl ATTRIBUTE_UNUSED, const char *prompt)
{
    return readline (prompt);
}

18978
#else /* !USE_READLINE */
18979

18980 18981 18982 18983 18984 18985 18986
static int
vshReadlineInit (vshControl *ctl ATTRIBUTE_UNUSED)
{
    /* empty */
    return 0;
}

18987
static void
18988
vshReadlineDeinit (vshControl *ctl ATTRIBUTE_UNUSED)
18989 18990 18991 18992 18993 18994 18995 18996 18997 18998 18999 19000 19001 19002 19003 19004 19005 19006 19007 19008 19009 19010 19011
{
    /* empty */
}

static char *
vshReadline (vshControl *ctl, const char *prompt)
{
    char line[1024];
    char *r;
    int len;

    fputs (prompt, stdout);
    r = fgets (line, sizeof line, stdin);
    if (r == NULL) return NULL; /* EOF */

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

    return vshStrdup (ctl, r);
}

19012
#endif /* !USE_READLINE */
19013

19014 19015 19016 19017 19018 19019
static void
vshDeinitTimer(int timer ATTRIBUTE_UNUSED, void *opaque ATTRIBUTE_UNUSED)
{
    /* nothing to be done here */
}

K
Karel Zak 已提交
19020
/*
J
Jim Meyering 已提交
19021
 * Deinitialize virsh
K
Karel Zak 已提交
19022
 */
E
Eric Blake 已提交
19023
static bool
19024
vshDeinit(vshControl *ctl)
19025
{
19026
    vshReadlineDeinit(ctl);
19027
    vshCloseLogFile(ctl);
19028
    VIR_FREE(ctl->name);
K
Karel Zak 已提交
19029
    if (ctl->conn) {
19030 19031 19032
        int ret;
        if ((ret = virConnectClose(ctl->conn)) != 0) {
            vshError(ctl, _("Failed to disconnect from the hypervisor, %d leaked reference(s)"), ret);
K
Karel Zak 已提交
19033 19034
        }
    }
D
Daniel P. Berrange 已提交
19035 19036
    virResetLastError();

J
Jiri Denemark 已提交
19037
    if (ctl->eventLoopStarted) {
19038 19039 19040 19041
        int timer;

        virMutexLock(&ctl->lock);
        ctl->quit = true;
J
Jiri Denemark 已提交
19042
        /* HACK: Add a dummy timeout to break event loop */
19043 19044 19045 19046 19047
        timer = virEventAddTimeout(0, vshDeinitTimer, NULL, NULL);
        virMutexUnlock(&ctl->lock);

        virThreadJoin(&ctl->eventLoop);

J
Jiri Denemark 已提交
19048 19049 19050 19051 19052 19053
        if (timer != -1)
            virEventRemoveTimeout(timer);

        ctl->eventLoopStarted = false;
    }

19054 19055
    virMutexDestroy(&ctl->lock);

E
Eric Blake 已提交
19056
    return true;
K
Karel Zak 已提交
19057
}
19058

K
Karel Zak 已提交
19059 19060 19061 19062
/*
 * Print usage
 */
static void
19063
vshUsage(void)
19064
{
19065
    const vshCmdGrp *grp;
19066
    const vshCmdDef *cmd;
19067

L
Lai Jiangshan 已提交
19068 19069
    fprintf(stdout, _("\n%s [options]... [<command_string>]"
                      "\n%s [options]... <command> [args...]\n\n"
19070
                      "  options:\n"
19071
                      "    -c | --connect=URI      hypervisor connection URI\n"
19072
                      "    -r | --readonly         connect readonly\n"
19073
                      "    -d | --debug=NUM        debug level [0-4]\n"
19074 19075 19076
                      "    -h | --help             this help\n"
                      "    -q | --quiet            quiet mode\n"
                      "    -t | --timing           print timing information\n"
19077 19078 19079 19080 19081
                      "    -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"
19082
                      "  commands (non interactive mode):\n\n"), progname, progname);
19083

19084 19085
    for (grp = cmdGroups; grp->name; grp++) {
        fprintf(stdout, _(" %s (help keyword '%s')\n"), grp->name, grp->keyword);
19086

19087 19088 19089 19090 19091 19092 19093 19094 19095
        for (cmd = grp->commands; cmd->name; cmd++)
            fprintf(stdout,
                    "    %-30s %s\n", cmd->name, _(vshCmddefGetInfo(cmd, "help")));

        fprintf(stdout, "\n");
    }

    fprintf(stdout, "%s",
            _("\n  (specify help <group> for details about the commands in the group)\n"));
19096 19097 19098
    fprintf(stdout, "%s",
            _("\n  (specify help <command> for details about the command)\n\n"));
    return;
K
Karel Zak 已提交
19099 19100
}

19101 19102 19103 19104 19105 19106 19107 19108 19109 19110
/*
 * 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 已提交
19111 19112
    vshPrint(ctl, "%s", _("Compiled with support for:\n"));
    vshPrint(ctl, "%s", _(" Hypervisors:"));
19113 19114 19115 19116 19117 19118 19119 19120 19121 19122 19123 19124 19125 19126 19127 19128 19129 19130 19131 19132 19133 19134 19135 19136 19137 19138 19139 19140 19141 19142 19143 19144 19145 19146 19147
#ifdef WITH_XEN
    vshPrint(ctl, " Xen");
#endif
#ifdef WITH_QEMU
    vshPrint(ctl, " QEmu/KVM");
#endif
#ifdef WITH_UML
    vshPrint(ctl, " UML");
#endif
#ifdef WITH_OPENVZ
    vshPrint(ctl, " OpenVZ");
#endif
#ifdef WITH_VBOX
    vshPrint(ctl, " VirtualBox");
#endif
#ifdef WITH_XENAPI
    vshPrint(ctl, " XenAPI");
#endif
#ifdef WITH_LXC
    vshPrint(ctl, " LXC");
#endif
#ifdef WITH_ESX
    vshPrint(ctl, " ESX");
#endif
#ifdef WITH_PHYP
    vshPrint(ctl, " PHYP");
#endif
#ifdef WITH_ONE
    vshPrint(ctl, " ONE");
#endif
#ifdef WITH_TEST
    vshPrint(ctl, " Test");
#endif
    vshPrint(ctl, "\n");

L
Laine Stump 已提交
19148
    vshPrint(ctl, "%s", _(" Networking:"));
19149 19150 19151 19152 19153 19154 19155 19156 19157 19158 19159 19160 19161 19162 19163 19164 19165 19166 19167 19168 19169 19170 19171 19172 19173 19174
#ifdef WITH_REMOTE
    vshPrint(ctl, " Remote");
#endif
#ifdef WITH_PROXY
    vshPrint(ctl, " Proxy");
#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
    vshPrint(ctl, " Netcf");
#endif
#ifdef WITH_NWFILTER
    vshPrint(ctl, " Nwfilter");
#endif
#ifdef WITH_VIRTUALPORT
    vshPrint(ctl, " VirtualPort");
#endif
    vshPrint(ctl, "\n");

L
Laine Stump 已提交
19175
    vshPrint(ctl, "%s", _(" Storage:"));
19176 19177 19178 19179 19180 19181 19182 19183 19184 19185 19186 19187 19188 19189 19190 19191 19192 19193 19194 19195 19196 19197 19198
#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");
#endif
    vshPrint(ctl, "\n");

L
Laine Stump 已提交
19199
    vshPrint(ctl, "%s", _(" Miscellaneous:"));
19200
#ifdef WITH_SECDRIVER_APPARMOR
19201 19202 19203 19204 19205 19206 19207 19208 19209 19210 19211 19212 19213 19214 19215 19216 19217 19218 19219 19220 19221 19222 19223
    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
    vshPrint(ctl, " DTrace");
#endif
#ifdef USE_READLINE
    vshPrint(ctl, " Readline");
#endif
#ifdef WITH_DRIVER_MODULES
    vshPrint(ctl, " Modular");
#endif
    vshPrint(ctl, "\n");
}

K
Karel Zak 已提交
19224 19225 19226 19227
/*
 * argv[]:  virsh [options] [command]
 *
 */
E
Eric Blake 已提交
19228
static bool
19229
vshParseArgv(vshControl *ctl, int argc, char **argv)
19230
{
19231
    bool help = false;
19232
    int arg, len;
K
Karel Zak 已提交
19233
    struct option opt[] = {
E
Eric Blake 已提交
19234 19235 19236 19237 19238 19239 19240 19241
        {"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'},
19242
        {"escape", required_argument, NULL, 'e'},
E
Eric Blake 已提交
19243
        {NULL, 0, NULL, 0}
19244 19245
    };

19246 19247 19248
    /* 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. */
19249
    while ((arg = getopt_long(argc, argv, "+d:hqtc:vVrl:e:", opt, NULL)) != -1) {
19250
        switch (arg) {
19251
        case 'd':
D
Daniel Veillard 已提交
19252
            if (virStrToLong_i(optarg, NULL, 10, &ctl->debug) < 0) {
L
Laine Stump 已提交
19253
                vshError(ctl, "%s", _("option -d takes a numeric argument"));
D
Daniel Veillard 已提交
19254 19255
                exit(EXIT_FAILURE);
            }
19256 19257
            break;
        case 'h':
19258
            help = true;
19259 19260
            break;
        case 'q':
E
Eric Blake 已提交
19261
            ctl->quiet = true;
19262 19263
            break;
        case 't':
E
Eric Blake 已提交
19264
            ctl->timing = true;
19265 19266 19267 19268
            break;
        case 'c':
            ctl->name = vshStrdup(ctl, optarg);
            break;
E
Eric Blake 已提交
19269 19270 19271 19272 19273 19274
        case 'v':
            if (STRNEQ_NULLABLE(optarg, "long")) {
                puts(VERSION);
                exit(EXIT_SUCCESS);
            }
            /* fall through */
19275 19276 19277
        case 'V':
            vshShowVersion(ctl);
            exit(EXIT_SUCCESS);
19278
        case 'r':
E
Eric Blake 已提交
19279
            ctl->readonly = true;
19280
            break;
19281 19282 19283
        case 'l':
            ctl->logfile = vshStrdup(ctl, optarg);
            break;
19284 19285 19286 19287 19288 19289 19290 19291 19292 19293 19294 19295
        case 'e':
            len = strlen(optarg);

            if ((len == 2 && *optarg == '^') ||
                (len == 1 && *optarg != '^')) {
                ctl->escapeChar = optarg;
            } else {
                vshError(ctl, _("Invalid string '%s' for escape sequence"),
                         optarg);
                exit(EXIT_FAILURE);
            }
            break;
19296
        default:
19297 19298
            vshError(ctl, _("unsupported option '-%c'. See --help."), arg);
            exit(EXIT_FAILURE);
K
Karel Zak 已提交
19299 19300 19301 19302
        }
    }

    if (help) {
19303 19304
        if (optind < argc) {
            vshError(ctl, _("extra argument '%s'. See --help."), argv[optind]);
19305 19306
            exit(EXIT_FAILURE);
        }
19307 19308 19309

        /* list all command */
        vshUsage();
K
Karel Zak 已提交
19310
        exit(EXIT_SUCCESS);
19311 19312
    }

19313
    if (argc > optind) {
K
Karel Zak 已提交
19314
        /* parse command */
E
Eric Blake 已提交
19315
        ctl->imode = false;
19316
        if (argc - optind == 1) {
19317
            vshDebug(ctl, VSH_ERR_INFO, "commands: \"%s\"\n", argv[optind]);
19318
            return vshCommandStringParse(ctl, argv[optind]);
L
Lai Jiangshan 已提交
19319
        } else {
19320
            return vshCommandArgvParse(ctl, argc - optind, argv + optind);
K
Karel Zak 已提交
19321 19322
        }
    }
E
Eric Blake 已提交
19323
    return true;
K
Karel Zak 已提交
19324 19325
}

19326 19327 19328 19329
int
main(int argc, char **argv)
{
    vshControl _ctl, *ctl = &_ctl;
19330
    char *defaultConn;
E
Eric Blake 已提交
19331
    bool ret = true;
K
Karel Zak 已提交
19332

19333 19334 19335
    memset(ctl, 0, sizeof(vshControl));
    ctl->imode = true;          /* default is interactive mode */
    ctl->log_fd = -1;           /* Initialize log file descriptor */
J
Jiri Denemark 已提交
19336
    ctl->debug = VSH_DEBUG_DEFAULT;
19337 19338
    ctl->escapeChar = CTRL_CLOSE_BRACKET;

19339

19340 19341
    if (!setlocale(LC_ALL, "")) {
        perror("setlocale");
19342
        /* failure to setup locale is not fatal */
19343
    }
19344
    if (!bindtextdomain(PACKAGE, LOCALEDIR)) {
19345
        perror("bindtextdomain");
E
Eric Blake 已提交
19346
        return EXIT_FAILURE;
19347
    }
19348
    if (!textdomain(PACKAGE)) {
19349
        perror("textdomain");
E
Eric Blake 已提交
19350
        return EXIT_FAILURE;
19351 19352
    }

19353 19354 19355 19356 19357
    if (virMutexInit(&ctl->lock) < 0) {
        vshError(ctl, "%s", _("Failed to initialize mutex"));
        return EXIT_FAILURE;
    }

19358 19359 19360 19361 19362
    if (virInitialize() < 0) {
        vshError(ctl, "%s", _("Failed to initialize libvirt"));
        return EXIT_FAILURE;
    }

19363
    if (!(progname = strrchr(argv[0], '/')))
K
Karel Zak 已提交
19364 19365 19366
        progname = argv[0];
    else
        progname++;
19367

19368
    if ((defaultConn = getenv("VIRSH_DEFAULT_CONNECT_URI"))) {
E
Eric Blake 已提交
19369
        ctl->name = vshStrdup(ctl, defaultConn);
19370 19371
    }

D
Daniel P. Berrange 已提交
19372 19373
    if (!vshParseArgv(ctl, argc, argv)) {
        vshDeinit(ctl);
K
Karel Zak 已提交
19374
        exit(EXIT_FAILURE);
D
Daniel P. Berrange 已提交
19375
    }
19376

D
Daniel P. Berrange 已提交
19377 19378
    if (!vshInit(ctl)) {
        vshDeinit(ctl);
K
Karel Zak 已提交
19379
        exit(EXIT_FAILURE);
D
Daniel P. Berrange 已提交
19380
    }
19381

K
Karel Zak 已提交
19382
    if (!ctl->imode) {
19383
        ret = vshCommandRun(ctl, ctl->cmd);
19384
    } else {
K
Karel Zak 已提交
19385 19386
        /* interactive mode */
        if (!ctl->quiet) {
K
Karel Zak 已提交
19387
            vshPrint(ctl,
19388
                     _("Welcome to %s, the virtualization interactive terminal.\n\n"),
19389
                     progname);
J
Jim Meyering 已提交
19390
            vshPrint(ctl, "%s",
19391
                     _("Type:  'help' for help with commands\n"
19392
                       "       'quit' to quit\n\n"));
K
Karel Zak 已提交
19393
        }
19394 19395 19396 19397 19398 19399

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

K
Karel Zak 已提交
19400
        do {
19401
            const char *prompt = ctl->readonly ? VSH_PROMPT_RO : VSH_PROMPT_RW;
19402
            ctl->cmdstr =
19403
                vshReadline(ctl, prompt);
19404 19405
            if (ctl->cmdstr == NULL)
                break;          /* EOF */
K
Karel Zak 已提交
19406
            if (*ctl->cmdstr) {
19407
#if USE_READLINE
K
Karel Zak 已提交
19408
                add_history(ctl->cmdstr);
19409
#endif
19410
                if (vshCommandStringParse(ctl, ctl->cmdstr))
K
Karel Zak 已提交
19411 19412
                    vshCommandRun(ctl, ctl->cmd);
            }
19413
            VIR_FREE(ctl->cmdstr);
19414
        } while (ctl->imode);
K
Karel Zak 已提交
19415

19416 19417
        if (ctl->cmdstr == NULL)
            fputc('\n', stdout);        /* line break after alone prompt */
K
Karel Zak 已提交
19418
    }
19419

K
Karel Zak 已提交
19420 19421
    vshDeinit(ctl);
    exit(ret ? EXIT_SUCCESS : EXIT_FAILURE);
19422
}