virsh.c 69.1 KB
Newer Older
1
/*
2
 * virsh.c: a Xen shell used to exercise the libvir API
3 4 5 6 7 8
 *
 * Copyright (C) 2005 Red Hat, Inc.
 *
 * 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 11
 * Daniel P. Berrange <berrange@redhat.com>
 *
K
Karel Zak 已提交
12 13
 *
 * $Id$
14 15
 */

16
#define _GNU_SOURCE             /* isblank() */
K
Karel Zak 已提交
17

18 19
#include "libvirt/libvirt.h"
#include "libvirt/virterror.h"
20
#include <stdio.h>
K
Karel Zak 已提交
21 22 23
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
24
#include <unistd.h>
K
Karel Zak 已提交
25
#include <getopt.h>
26
#include <sys/types.h>
K
Karel Zak 已提交
27
#include <sys/time.h>
K
Karel Zak 已提交
28
#include <ctype.h>
29
#include <fcntl.h>
30
#include <locale.h>
K
Karel Zak 已提交
31 32 33 34 35

#include <readline/readline.h>
#include <readline/history.h>

#include "config.h"
36
#include "internal.h"
K
Karel Zak 已提交
37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52

static char *progname;

#ifndef TRUE
#define TRUE 1
#define FALSE 0
#endif

#define VSH_PROMPT_RW    "virsh # "
#define VSH_PROMPT_RO    "virsh > "

#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)

53 54 55 56
/*
 * The error handler for virtsh
 */
static void
57 58
virshErrorHandler(void *unused, virErrorPtr error)
{
59 60 61 62 63 64 65 66 67 68
    if ((unused != NULL) || (error == NULL))
        return;

    /* Suppress the VIR_ERR_NO_XEN error which fails as non-root */
    if ((error->code == VIR_ERR_NO_XEN) || (error->code == VIR_ERR_OK))
        return;

    virDefaultErrorFunc(error);
}

K
Karel Zak 已提交
69 70 71 72 73 74 75 76 77 78 79 80 81 82
/*
 * virsh command line grammar:
 *
 *    command_line    =     <command>\n | <command>; <command>; ...
 *
 *    command         =    <keyword> <option> <data>
 *
 *    option          =     <bool_option> | <int_option> | <string_option>
 *    data            =     <string>
 *
 *    bool_option     =     --optionname
 *    int_option      =     --optionname <number>
 *    string_option   =     --optionname <string>
 *        
83 84 85
 *    keyword         =     [a-zA-Z]
 *    number          =     [0-9]+
 *    string          =     [^[:blank:]] | "[[:alnum:]]"$
K
Karel Zak 已提交
86 87 88 89 90
 *
 */

/*
 * vshCmdOptType - command option type 
91
 */
K
Karel Zak 已提交
92
typedef enum {
93 94 95 96 97
    VSH_OT_NONE = 0,            /* none */
    VSH_OT_BOOL,                /* boolean option */
    VSH_OT_STRING,              /* string option */
    VSH_OT_INT,                 /* int option */
    VSH_OT_DATA                 /* string data (as non-option) */
K
Karel Zak 已提交
98 99 100 101 102
} vshCmdOptType;

/*
 * Command Option Flags
 */
103 104
#define VSH_OFLAG_NONE    0     /* without flags */
#define VSH_OFLAG_REQ    (1 << 1)       /* option required */
K
Karel Zak 已提交
105 106 107 108 109 110 111 112

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

/*
 * vshCmdInfo -- information about command
 */
113 114 115
typedef struct {
    const char *name;           /* name of information */
    const char *data;           /* information */
K
Karel Zak 已提交
116 117 118 119 120
} vshCmdInfo;

/*
 * vshCmdOptDef - command option definition
 */
121 122 123 124 125
typedef struct {
    const char *name;           /* the name of option */
    vshCmdOptType type;         /* option type */
    int flag;                   /* flags */
    const char *help;           /* help string */
K
Karel Zak 已提交
126 127 128 129 130 131
} vshCmdOptDef;

/*
 * vshCmdOpt - command options
 */
typedef struct vshCmdOpt {
132 133 134
    vshCmdOptDef *def;          /* pointer to relevant option */
    char *data;                 /* allocated data */
    struct vshCmdOpt *next;
K
Karel Zak 已提交
135 136 137 138 139
} vshCmdOpt;

/*
 * vshCmdDef - command definition
 */
140 141 142 143 144
typedef struct {
    const char *name;
    int (*handler) (vshControl *, vshCmd *);    /* command handler */
    vshCmdOptDef *opts;         /* definition of command options */
    vshCmdInfo *info;           /* details about command */
K
Karel Zak 已提交
145 146 147 148 149 150
} vshCmdDef;

/*
 * vshCmd - parsed command
 */
typedef struct __vshCmd {
151 152 153
    vshCmdDef *def;             /* command definition */
    vshCmdOpt *opts;            /* list of command arguments */
    struct __vshCmd *next;      /* next command */
K
Karel Zak 已提交
154 155 156 157 158 159
} __vshCmd;

/*
 * vshControl
 */
typedef struct __vshControl {
K
Karel Zak 已提交
160
    char *name;                 /* connection name */
161 162 163 164 165 166 167 168
    virConnectPtr conn;         /* connection to hypervisor */
    vshCmd *cmd;                /* the current command */
    char *cmdstr;               /* string with command */
    uid_t uid;                  /* process owner */
    int imode;                  /* interactive mode? */
    int quiet;                  /* quiet mode */
    int debug;                  /* print debug messages? */
    int timing;                 /* print timing info? */
K
Karel Zak 已提交
169
} __vshControl;
170

171

K
Karel Zak 已提交
172 173
static vshCmdDef commands[];

174 175 176 177 178
static void vshError(vshControl * ctl, int doexit, const char *format,
                     ...);
static int vshInit(vshControl * ctl);
static int vshDeinit(vshControl * ctl);
static void vshUsage(vshControl * ctl, const char *cmdname);
K
Karel Zak 已提交
179

180
static int vshParseArgv(vshControl * ctl, int argc, char **argv);
K
Karel Zak 已提交
181

182
static const char *vshCmddefGetInfo(vshCmdDef * cmd, const char *info);
K
Karel Zak 已提交
183
static vshCmdDef *vshCmddefSearch(const char *cmdname);
184
static int vshCmddefHelp(vshControl * ctl, const char *name, int withprog);
K
Karel Zak 已提交
185

186 187 188 189 190
static vshCmdOpt *vshCommandOpt(vshCmd * cmd, const char *name);
static int vshCommandOptInt(vshCmd * cmd, const char *name, int *found);
static char *vshCommandOptString(vshCmd * cmd, const char *name,
                                 int *found);
static int vshCommandOptBool(vshCmd * cmd, const char *name);
K
Karel Zak 已提交
191 192 193 194 195 196 197 198 199 200 201 202

#define VSH_DOMBYID     (1 << 1)
#define VSH_DOMBYUUID   (1 << 2)
#define VSH_DOMBYNAME   (1 << 3)

static virDomainPtr vshCommandOptDomainBy(vshControl * ctl, vshCmd * cmd,
                            const char *optname, char **name, int flag);

/* default is lookup by Id, Name and UUID */
#define vshCommandOptDomain(_ctl, _cmd, _optname, _name) \
                            vshCommandOptDomainBy(_ctl, _cmd, _optname, _name,\
                                        VSH_DOMBYID|VSH_DOMBYUUID|VSH_DOMBYNAME)
K
Karel Zak 已提交
203

K
Karel Zak 已提交
204 205
static void vshPrintExtra(vshControl * ctl, const char *format, ...);
static void vshDebug(vshControl * ctl, int level, const char *format, ...);
K
Karel Zak 已提交
206 207

/* XXX: add batch support */
K
Karel Zak 已提交
208
#define vshPrint(_ctl, ...)   fprintf(stdout, __VA_ARGS__)
K
Karel Zak 已提交
209

K
Karel Zak 已提交
210
static const char *vshDomainStateToString(int state);
211
static const char *vshDomainVcpuStateToString(int state);
212 213
static int vshConnectionUsability(vshControl * ctl, virConnectPtr conn,
                                  int showerror);
K
Karel Zak 已提交
214

215 216 217 218 219 220 221 222 223
static void *_vshMalloc(vshControl * ctl, size_t sz, const char *filename, int line);
#define vshMalloc(_ctl, _sz)    _vshMalloc(_ctl, _sz, __FILE__, __LINE__)

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

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

K
Karel Zak 已提交
224 225 226 227 228 229 230 231 232
/* ---------------
 * Commands
 * ---------------
 */

/*
 * "help" command 
 */
static vshCmdInfo info_help[] = {
233
    {"syntax", "help [<command>]"},
234 235 236
    {"help", gettext_noop("print help")},
    {"desc", gettext_noop("Prints global help or command specific help.")},

237
    {NULL, NULL}
K
Karel Zak 已提交
238 239 240
};

static vshCmdOptDef opts_help[] = {
241 242
    {"command", VSH_OT_DATA, 0, "name of command"},
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
243 244 245
};

static int
246 247
cmdHelp(vshControl * ctl, vshCmd * cmd)
{
K
Karel Zak 已提交
248
    const char *cmdname = vshCommandOptString(cmd, "command", NULL);
K
Karel Zak 已提交
249 250 251

    if (!cmdname) {
        vshCmdDef *def;
252

253
        vshPrint(ctl, _("Commands:\n\n"));
254
        for (def = commands; def->name; def++)
K
Karel Zak 已提交
255
            vshPrint(ctl, "    %-15s %s\n", def->name,
256
                     _N(vshCmddefGetInfo(def, "help")));
K
Karel Zak 已提交
257 258 259 260 261 262 263 264 265
        return TRUE;
    }
    return vshCmddefHelp(ctl, cmdname, FALSE);
}

/*
 * "connect" command 
 */
static vshCmdInfo info_connect[] = {
K
Karel Zak 已提交
266
    {"syntax", "connect [name] [--readonly]"},
267
    {"help", gettext_noop("(re)connect to hypervisor")},
268
    {"desc",
269
     gettext_noop("Connect to local hypervisor. This is built-in command after shell start up.")},
270
    {NULL, NULL}
K
Karel Zak 已提交
271 272 273
};

static vshCmdOptDef opts_connect[] = {
274 275
    {"name",     VSH_OT_DATA, 0, gettext_noop("hypervisor connection URI")},
    {"readonly", VSH_OT_BOOL, 0, gettext_noop("read-only connection")},
276
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
277 278 279
};

static int
280 281
cmdConnect(vshControl * ctl, vshCmd * cmd)
{
K
Karel Zak 已提交
282
    int ro = vshCommandOptBool(cmd, "readonly");
K
Karel Zak 已提交
283
    
K
Karel Zak 已提交
284
    if (ctl->conn) {
285 286
        if (virConnectClose(ctl->conn) != 0) {
            vshError(ctl, FALSE,
287
                     _("Failed to disconnect from the hypervisor"));
K
Karel Zak 已提交
288 289 290 291
            return FALSE;
        }
        ctl->conn = NULL;
    }
K
Karel Zak 已提交
292 293 294
    
    if (ctl->name)
        free(ctl->name);
295
    ctl->name = vshStrdup(ctl, vshCommandOptString(cmd, "name", NULL));
K
Karel Zak 已提交
296

K
Karel Zak 已提交
297
    if (!ro)
K
Karel Zak 已提交
298
        ctl->conn = virConnectOpen(ctl->name);
K
Karel Zak 已提交
299
    else
K
Karel Zak 已提交
300
        ctl->conn = virConnectOpenReadOnly(ctl->name);
K
Karel Zak 已提交
301 302

    if (!ctl->conn)
303
        vshError(ctl, FALSE, _("Failed to connect to the hypervisor"));
304

K
Karel Zak 已提交
305 306 307 308 309 310 311
    return ctl->conn ? TRUE : FALSE;
}

/*
 * "list" command
 */
static vshCmdInfo info_list[] = {
312
    {"syntax", "list"},
313 314
    {"help", gettext_noop("list domains")},
    {"desc", gettext_noop("Returns list of domains.")},
315
    {NULL, NULL}
K
Karel Zak 已提交
316 317
};

318
static vshCmdOptDef opts_list[] = {
319 320
    {"inactive", VSH_OT_BOOL, 0, gettext_noop("list inactive domains")},
    {"all", VSH_OT_BOOL, 0, gettext_noop("list inactive & active domains")},
321 322 323
    {NULL, 0, 0, NULL}
};

K
Karel Zak 已提交
324

325 326 327 328 329 330 331 332 333 334 335 336 337
static int domidsorter(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 domnamesorter(const void *a, const void *b) {
  const char **sa = (const char**)a;
  const char **sb = (const char**)b;
K
Karel Zak 已提交
338

339 340
  return strcasecmp(*sa, *sb);
}
K
Karel Zak 已提交
341
static int
342 343
cmdList(vshControl * ctl, vshCmd * cmd ATTRIBUTE_UNUSED)
{
344 345 346 347 348 349 350
    int inactive = vshCommandOptBool(cmd, "inactive");
    int all = vshCommandOptBool(cmd, "all");
    int active = !inactive || all ? 1 : 0;
    int *ids = NULL, maxid = 0, i;
    const char **names = NULL;
    int maxname = 0;
    inactive |= all;
K
Karel Zak 已提交
351 352 353

    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;
354 355 356 357
    
    if (active) {
      maxid = virConnectNumOfDomains(ctl->conn);
      if (maxid < 0) {
358
        vshError(ctl, FALSE, _("Failed to list active domains"));
K
Karel Zak 已提交
359
        return FALSE;
360 361
      }
      if (maxid) {
362
        ids = vshMalloc(ctl, sizeof(int) * maxid);
363
	
364
        if ((maxid = virConnectListDomains(ctl->conn, &ids[0], maxid)) < 0) {
365
	  vshError(ctl, FALSE, _("Failed to list active domains"));
366 367
	  free(ids);
	  return FALSE;
368
        }
369 370
	
	qsort(&ids[0], maxid, sizeof(int), domidsorter);
371 372 373 374 375
      }
    }
    if (inactive) {
      maxname = virConnectNumOfDefinedDomains(ctl->conn);
      if (maxname < 0) {
376
        vshError(ctl, FALSE, _("Failed to list inactive domains"));
377 378 379 380 381
	if (ids)
	  free(ids);
        return FALSE;
      }
      if (maxname) {
382
        names = vshMalloc(ctl, sizeof(char *) * maxname);
383
	
384
        if ((maxname = virConnectListDefinedDomains(ctl->conn, names, maxname)) < 0) {
385
	  vshError(ctl, FALSE, _("Failed to list inactive domains"));
386 387 388 389 390
	  if (ids)
	    free(ids);
	  free(names);
	  return FALSE;
        }
391 392

	qsort(&names[0], maxname, sizeof(char*), domnamesorter);
393
      }
394
    }
395
    vshPrintExtra(ctl, "%3s %-20s %s\n", _("Id"), _("Name"), _("State"));
K
Karel Zak 已提交
396
    vshPrintExtra(ctl, "----------------------------------\n");
397 398

    for (i = 0; i < maxid; i++) {
K
Karel Zak 已提交
399 400 401
        int ret;
        virDomainInfo info;
        virDomainPtr dom = virDomainLookupByID(ctl->conn, ids[i]);
402 403

        /* this kind of work with domains is not atomic operation */
K
Karel Zak 已提交
404 405 406
        if (!dom)
            continue;
        ret = virDomainGetInfo(dom, &info);
407

K
Karel Zak 已提交
408
        vshPrint(ctl, "%3d %-20s %s\n",
409 410 411
                 virDomainGetID(dom),
                 virDomainGetName(dom),
                 ret <
412
                 0 ? _("no state") : _N(vshDomainStateToString(info.state)));
413
        virDomainFree(dom);
K
Karel Zak 已提交
414
    }
415 416
    for (i = 0; i < maxname; i++) {
        int ret;
417
        unsigned int id;
418 419 420 421 422 423 424
        virDomainInfo info;
        virDomainPtr dom = virDomainLookupByName(ctl->conn, names[i]);

        /* this kind of work with domains is not atomic operation */
        if (!dom)
            continue;
        ret = virDomainGetInfo(dom, &info);
425 426 427 428 429 430 431 432 433 434 435 436 437 438 439
	id = virDomainGetID(dom);

	if (id == ((unsigned int)-1)) {
	  vshPrint(ctl, "%3s %-20s %s\n",
		   "-",
		   names[i],
		   ret <
		   0 ? "no state" : vshDomainStateToString(info.state));
	} else {
	  vshPrint(ctl, "%3d %-20s %s\n",
		   id,
		   names[i],
		   ret <
		   0 ? "no state" : vshDomainStateToString(info.state));
	}
440

441 442
        virDomainFree(dom);
    }
443 444
    if (ids)
        free(ids);
445 446
    if (names)
        free(names);
K
Karel Zak 已提交
447 448 449 450
    return TRUE;
}

/*
K
Karel Zak 已提交
451
 * "domstate" command
K
Karel Zak 已提交
452
 */
K
Karel Zak 已提交
453 454
static vshCmdInfo info_domstate[] = {
    {"syntax", "domstate <domain>"},
455 456
    {"help", gettext_noop("domain state")},
    {"desc", gettext_noop("Returns state about a running domain.")},
457
    {NULL, NULL}
K
Karel Zak 已提交
458 459
};

K
Karel Zak 已提交
460
static vshCmdOptDef opts_domstate[] = {
461
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
462
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
463 464 465
};

static int
K
Karel Zak 已提交
466
cmdDomstate(vshControl * ctl, vshCmd * cmd)
467
{
468
    virDomainInfo info;
K
Karel Zak 已提交
469
    virDomainPtr dom;
K
Karel Zak 已提交
470
    int ret = TRUE;
471

K
Karel Zak 已提交
472 473
    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;
474

K
Karel Zak 已提交
475
    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", NULL)))
K
Karel Zak 已提交
476
        return FALSE;
477 478

    if (virDomainGetInfo(dom, &info) == 0)
K
Karel Zak 已提交
479
        vshPrint(ctl, "%s\n",
480
                 _N(vshDomainStateToString(info.state)));
K
Karel Zak 已提交
481 482
    else
        ret = FALSE;
483

484 485 486 487 488 489 490 491
    virDomainFree(dom);
    return ret;
}

/*
 * "suspend" command
 */
static vshCmdInfo info_suspend[] = {
492
    {"syntax", "suspend <domain>"},
493 494
    {"help", gettext_noop("suspend a domain")},
    {"desc", gettext_noop("Suspend a running domain.")},
495
    {NULL, NULL}
496 497 498
};

static vshCmdOptDef opts_suspend[] = {
499
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
500
    {NULL, 0, 0, NULL}
501 502 503
};

static int
504 505
cmdSuspend(vshControl * ctl, vshCmd * cmd)
{
506
    virDomainPtr dom;
K
Karel Zak 已提交
507 508
    char *name;
    int ret = TRUE;
509

510 511 512
    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;

K
Karel Zak 已提交
513
    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", &name)))
514
        return FALSE;
515 516

    if (virDomainSuspend(dom) == 0) {
517
        vshPrint(ctl, _("Domain %s suspended\n"), name);
518
    } else {
519
        vshError(ctl, FALSE, _("Failed to suspend domain %s"), name);
520 521
        ret = FALSE;
    }
522

523 524 525 526
    virDomainFree(dom);
    return ret;
}

527 528 529 530 531
/*
 * "create" command
 */
static vshCmdInfo info_create[] = {
    {"syntax", "create a domain from an XML <file>"},
532 533
    {"help", gettext_noop("create a domain from an XML file")},
    {"desc", gettext_noop("Create a domain.")},
534 535 536 537
    {NULL, NULL}
};

static vshCmdOptDef opts_create[] = {
538
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("file conatining an XML domain description")},
539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560
    {NULL, 0, 0, NULL}
};

static int
cmdCreate(vshControl * ctl, vshCmd * cmd)
{
    virDomainPtr dom;
    char *from;
    int found;
    int ret = TRUE;
    char buffer[4096];
    int fd, l;

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

    from = vshCommandOptString(cmd, "file", &found);
    if (!found)
        return FALSE;

    fd = open(from, O_RDONLY);
    if (fd < 0) {
561
        vshError(ctl, FALSE, _("Failed to read description file %s"), from);
562 563 564 565
        return(FALSE);
    }
    l = read(fd, &buffer[0], sizeof(buffer));
    if ((l <= 0) || (l >= (int) sizeof(buffer))) {
566
        vshError(ctl, FALSE, _("Failed to read description file %s"), from);
567 568 569 570 571 572
        close(fd);
        return(FALSE);
    }
    buffer[l] = 0;
    dom = virDomainCreateLinux(ctl->conn, &buffer[0], 0);
    if (dom != NULL) {
573
        vshPrint(ctl, _("Domain %s created from %s\n"),
574 575
                 virDomainGetName(dom), from);
    } else {
576
        vshError(ctl, FALSE, _("Failed to create domain from %s"), from);
577 578 579 580 581
        ret = FALSE;
    }
    return ret;
}

582 583 584 585 586
/*
 * "define" command
 */
static vshCmdInfo info_define[] = {
    {"syntax", "define a domain from an XML <file>"},
587 588
    {"help", gettext_noop("define (but don't start) a domain from an XML file")},
    {"desc", gettext_noop("Define a domain.")},
589 590 591 592
    {NULL, NULL}
};

static vshCmdOptDef opts_define[] = {
593
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("file conatining an XML domain description")},
594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615
    {NULL, 0, 0, NULL}
};

static int
cmdDefine(vshControl * ctl, vshCmd * cmd)
{
    virDomainPtr dom;
    char *from;
    int found;
    int ret = TRUE;
    char buffer[4096];
    int fd, l;

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

    from = vshCommandOptString(cmd, "file", &found);
    if (!found)
        return FALSE;

    fd = open(from, O_RDONLY);
    if (fd < 0) {
616
        vshError(ctl, FALSE, _("Failed to read description file %s"), from);
617 618 619 620
        return(FALSE);
    }
    l = read(fd, &buffer[0], sizeof(buffer));
    if ((l <= 0) || (l >= (int) sizeof(buffer))) {
621
        vshError(ctl, FALSE, _("Failed to read description file %s"), from);
622 623 624 625 626 627
        close(fd);
        return(FALSE);
    }
    buffer[l] = 0;
    dom = virDomainDefineXML(ctl->conn, &buffer[0]);
    if (dom != NULL) {
628
        vshPrint(ctl, _("Domain %s defined from %s\n"),
629 630
                 virDomainGetName(dom), from);
    } else {
631
        vshError(ctl, FALSE, _("Failed to define domain from %s"), from);
632 633 634 635 636 637 638 639 640 641
        ret = FALSE;
    }
    return ret;
}

/*
 * "undefine" command
 */
static vshCmdInfo info_undefine[] = {
    {"syntax", "undefine <domain>"},
642 643
    {"help", gettext_noop("undefine an inactive domain")},
    {"desc", gettext_noop("Undefine the configuration for an inactive domain.")},
644 645 646 647
    {NULL, NULL}
};

static vshCmdOptDef opts_undefine[] = {
648
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name or uuid")},
649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665
    {NULL, 0, 0, NULL}
};

static int
cmdUndefine(vshControl * ctl, vshCmd * cmd)
{
    virDomainPtr dom;
    int ret = TRUE;
    char *name;

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

    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", &name)))
        return FALSE;

    if (virDomainUndefine(dom) == 0) {
666
        vshPrint(ctl, _("Domain %s has been undefined\n"), name);
667
    } else {
668
        vshError(ctl, FALSE, _("Failed to undefine domain %s"), name);
669 670 671 672 673 674 675 676 677 678 679 680
        ret = FALSE;
    }

    return ret;
}


/*
 * "start" command
 */
static vshCmdInfo info_start[] = {
    {"syntax", "start a domain "},
681 682
    {"help", gettext_noop("start a (previously defined) inactive domain")},
    {"desc", gettext_noop("Start a domain.")},
683 684 685 686
    {NULL, NULL}
};

static vshCmdOptDef opts_start[] = {
687
    {"name", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("name of the inactive domain")},
688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710
    {NULL, 0, 0, NULL}
};

static int
cmdStart(vshControl * ctl, vshCmd * cmd)
{
    virDomainPtr dom;
    char *name;
    int found;
    int ret = TRUE;

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

    name = vshCommandOptString(cmd, "name", &found);
    if (!found)
        return FALSE;

    dom = virDomainLookupByName(ctl->conn, name);
    if (!dom)
        return FALSE;

    if (virDomainGetID(dom) != (unsigned int)-1) {
711
        vshError(ctl, FALSE, _("Domain is already active"));
712 713 714 715
        return FALSE;
    }

    if (virDomainCreate(dom) == 0) {
716
        vshPrint(ctl, _("Domain %s started\n"),
717 718
                 name);
    } else {
719
      vshError(ctl, FALSE, _("Failed to start domain %s"), name);
720 721 722 723 724
        ret = FALSE;
    }
    return ret;
}

725 726 727 728
/*
 * "save" command
 */
static vshCmdInfo info_save[] = {
729
    {"syntax", "save <domain> <file>"},
730 731
    {"help", gettext_noop("save a domain state to a file")},
    {"desc", gettext_noop("Save a running domain.")},
732
    {NULL, NULL}
733 734 735
};

static vshCmdOptDef opts_save[] = {
736 737
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("where to save the data")},
738
    {NULL, 0, 0, NULL}
739 740 741
};

static int
742 743
cmdSave(vshControl * ctl, vshCmd * cmd)
{
744 745 746 747
    virDomainPtr dom;
    char *name;
    char *to;
    int ret = TRUE;
748

749 750 751
    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;

752
    if (!(to = vshCommandOptString(cmd, "file", NULL)))
753
        return FALSE;
754

755 756
    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", &name)))
        return FALSE;
757 758

    if (virDomainSave(dom, to) == 0) {
759
        vshPrint(ctl, _("Domain %s saved to %s\n"), name, to);
760
    } else {
761
        vshError(ctl, FALSE, _("Failed to save domain %s to %s"), name, to);
762 763
        ret = FALSE;
    }
764

765 766 767 768 769 770 771 772
    virDomainFree(dom);
    return ret;
}

/*
 * "restore" command
 */
static vshCmdInfo info_restore[] = {
773
    {"syntax", "restore a domain from <file>"},
774 775
    {"help", gettext_noop("restore a domain from a saved state in a file")},
    {"desc", gettext_noop("Restore a domain.")},
776
    {NULL, NULL}
777 778 779
};

static vshCmdOptDef opts_restore[] = {
780
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("the state to restore")},
781
    {NULL, 0, 0, NULL}
782 783 784
};

static int
785 786
cmdRestore(vshControl * ctl, vshCmd * cmd)
{
787 788 789
    char *from;
    int found;
    int ret = TRUE;
790

791 792 793 794 795 796
    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;

    from = vshCommandOptString(cmd, "file", &found);
    if (!found)
        return FALSE;
797 798

    if (virDomainRestore(ctl->conn, from) == 0) {
799
        vshPrint(ctl, _("Domain restored from %s\n"), from);
800
    } else {
801
        vshError(ctl, FALSE, _("Failed to restore domain from %s"), from);
802 803 804 805 806
        ret = FALSE;
    }
    return ret;
}

D
Daniel Veillard 已提交
807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851
/*
 * "dump" command
 */
static vshCmdInfo info_dump[] = {
    {"syntax", "dump <domain> <file>"},
    {"help", gettext_noop("dump the core of a domain to a file for analysis")},
    {"desc", gettext_noop("Core dump a domain.")},
    {NULL, NULL}
};

static vshCmdOptDef opts_dump[] = {
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
    {"file", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("where to dump the core")},
    {NULL, 0, 0, NULL}
};

static int
cmdDump(vshControl * ctl, vshCmd * cmd)
{
    virDomainPtr dom;
    char *name;
    char *to;
    int ret = TRUE;

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

    if (!(to = vshCommandOptString(cmd, "file", NULL)))
        return FALSE;

    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", &name)))
        return FALSE;

    if (virDomainCoreDump(dom, to, 0) == 0) {
        vshPrint(ctl, _("Domain %s dumpd to %s\n"), name, to);
    } else {
        vshError(ctl, FALSE, _("Failed to core dump domain %s to %s"),
                 name, to);
        ret = FALSE;
    }

    virDomainFree(dom);
    return ret;
}

852 853 854 855
/*
 * "resume" command
 */
static vshCmdInfo info_resume[] = {
856
    {"syntax", "resume <domain>"},
857 858
    {"help", gettext_noop("resume a domain")},
    {"desc", gettext_noop("Resume a previously suspended domain.")},
859
    {NULL, NULL}
860 861 862
};

static vshCmdOptDef opts_resume[] = {
863
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
864
    {NULL, 0, 0, NULL}
865 866 867
};

static int
868 869
cmdResume(vshControl * ctl, vshCmd * cmd)
{
870
    virDomainPtr dom;
K
Karel Zak 已提交
871 872
    int ret = TRUE;
    char *name;
873

874 875 876
    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;

K
Karel Zak 已提交
877
    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", &name)))
878
        return FALSE;
879 880

    if (virDomainResume(dom) == 0) {
881
        vshPrint(ctl, _("Domain %s resumed\n"), name);
882
    } else {
883
        vshError(ctl, FALSE, _("Failed to resume domain %s"), name);
884 885
        ret = FALSE;
    }
886

887 888 889 890
    virDomainFree(dom);
    return ret;
}

891 892 893 894
/*
 * "shutdown" command
 */
static vshCmdInfo info_shutdown[] = {
895
    {"syntax", "shutdown <domain>"},
896 897
    {"help", gettext_noop("gracefully shutdown a domain")},
    {"desc", gettext_noop("Run shutdown in the target domain.")},
898
    {NULL, NULL}
899 900 901
};

static vshCmdOptDef opts_shutdown[] = {
902
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
903
    {NULL, 0, 0, NULL}
904 905 906
};

static int
907 908
cmdShutdown(vshControl * ctl, vshCmd * cmd)
{
909 910 911
    virDomainPtr dom;
    int ret = TRUE;
    char *name;
912

913 914 915 916 917
    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;

    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", &name)))
        return FALSE;
918 919

    if (virDomainShutdown(dom) == 0) {
920
        vshPrint(ctl, _("Domain %s is being shutdown\n"), name);
921
    } else {
922
        vshError(ctl, FALSE, _("Failed to shutdown domain %s"), name);
923 924
        ret = FALSE;
    }
925

926 927 928 929
    virDomainFree(dom);
    return ret;
}

930 931 932 933 934
/*
 * "reboot" command
 */
static vshCmdInfo info_reboot[] = {
    {"syntax", "reboot <domain>"},
935 936
    {"help", gettext_noop("reboot a domain")},
    {"desc", gettext_noop("Run a reboot command in the target domain.")},
937 938 939 940
    {NULL, NULL}
};

static vshCmdOptDef opts_reboot[] = {
941
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958
    {NULL, 0, 0, NULL}
};

static int
cmdReboot(vshControl * ctl, vshCmd * cmd)
{
    virDomainPtr dom;
    int ret = TRUE;
    char *name;

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

    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", &name)))
        return FALSE;

    if (virDomainReboot(dom, 0) == 0) {
959
        vshPrint(ctl, _("Domain %s is being rebooted\n"), name);
960
    } else {
961
        vshError(ctl, FALSE, _("Failed to reboot domain %s"), name);
962 963 964 965 966 967 968
        ret = FALSE;
    }

    virDomainFree(dom);
    return ret;
}

969 970 971 972
/*
 * "destroy" command
 */
static vshCmdInfo info_destroy[] = {
973
    {"syntax", "destroy <domain>"},
974 975
    {"help", gettext_noop("destroy a domain")},
    {"desc", gettext_noop("Destroy a given domain.")},
976
    {NULL, NULL}
977 978 979
};

static vshCmdOptDef opts_destroy[] = {
980
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
981
    {NULL, 0, 0, NULL}
982 983 984
};

static int
985 986
cmdDestroy(vshControl * ctl, vshCmd * cmd)
{
987
    virDomainPtr dom;
K
Karel Zak 已提交
988 989
    int ret = TRUE;
    char *name;
990

991 992 993
    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;

K
Karel Zak 已提交
994
    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", &name)))
995
        return FALSE;
996 997

    if (virDomainDestroy(dom) == 0) {
998
        vshPrint(ctl, _("Domain %s destroyed\n"), name);
999
    } else {
1000
        vshError(ctl, FALSE, _("Failed to destroy domain %s"), name);
1001 1002 1003
        ret = FALSE;
        virDomainFree(dom);
    }
1004

K
Karel Zak 已提交
1005 1006 1007 1008
    return ret;
}

/*
1009
 * "dominfo" command
K
Karel Zak 已提交
1010
 */
1011 1012
static vshCmdInfo info_dominfo[] = {
    {"syntax", "dominfo <domain>"},
1013 1014
    {"help", gettext_noop("domain information")},
    {"desc", gettext_noop("Returns basic information about the domain.")},
1015
    {NULL, NULL}
K
Karel Zak 已提交
1016 1017
};

1018
static vshCmdOptDef opts_dominfo[] = {
1019
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
1020
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
1021 1022 1023
};

static int
1024
cmdDominfo(vshControl * ctl, vshCmd * cmd)
1025
{
K
Karel Zak 已提交
1026 1027
    virDomainInfo info;
    virDomainPtr dom;
K
Karel Zak 已提交
1028
    int ret = TRUE;
1029
    unsigned int id;
K
Karel Zak 已提交
1030
    char *str, uuid[37];
1031

K
Karel Zak 已提交
1032 1033 1034
    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;

K
Karel Zak 已提交
1035
    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", NULL)))
K
Karel Zak 已提交
1036
        return FALSE;
1037

1038 1039
    id = virDomainGetID(dom);
    if (id == ((unsigned int)-1))
1040
      vshPrint(ctl, "%-15s %s\n", _("Id:"), "-");
1041
    else
1042 1043 1044
      vshPrint(ctl, "%-15s %d\n", _("Id:"), id);
    vshPrint(ctl, "%-15s %s\n", _("Name:"), virDomainGetName(dom));

K
Karel Zak 已提交
1045
    if (virDomainGetUUIDString(dom, &uuid[0])==0)
1046
        vshPrint(ctl, "%-15s %s\n", _("UUID:"), uuid);
1047 1048

    if ((str = virDomainGetOSType(dom))) {
1049
        vshPrint(ctl, "%-15s %s\n", _("OS Type:"), str);
1050 1051 1052 1053
        free(str);
    }

    if (virDomainGetInfo(dom, &info) == 0) {
1054 1055
        vshPrint(ctl, "%-15s %s\n", _("State:"),
                 _N(vshDomainStateToString(info.state)));
1056

1057
        vshPrint(ctl, "%-15s %d\n", _("CPU(s):"), info.nrVirtCpu);
1058 1059

        if (info.cpuTime != 0) {
1060
	    double cpuUsed = info.cpuTime;
1061

1062
            cpuUsed /= 1000000000.0;
1063

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

1067
        vshPrint(ctl, "%-15s %lu kB\n", _("Max memory:"),
1068
                 info.maxMem);
1069
	vshPrint(ctl, "%-15s %lu kB\n", _("Used memory:"),
1070 1071
                 info.memory);

K
Karel Zak 已提交
1072 1073 1074
    } else {
        ret = FALSE;
    }
1075

1076
    virDomainFree(dom);
K
Karel Zak 已提交
1077 1078 1079
    return ret;
}

1080 1081 1082 1083 1084
/*
 * "vcpuinfo" command
 */
static vshCmdInfo info_vcpuinfo[] = {
    {"syntax", "vcpuinfo <domain>"},
1085 1086
    {"help", gettext_noop("domain vcpu information")},
    {"desc", gettext_noop("Returns basic information about the domain virtual CPUs.")},
1087 1088 1089 1090
    {NULL, NULL}
};

static vshCmdOptDef opts_vcpuinfo[] = {
1091
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132
    {NULL, 0, 0, NULL}
};

static int
cmdVcpuinfo(vshControl * ctl, vshCmd * cmd)
{
    virDomainInfo info;
    virDomainPtr dom;
    virNodeInfo nodeinfo;
    virVcpuInfoPtr cpuinfo;
    unsigned char *cpumap;
    int ncpus;
    size_t cpumaplen;
    int ret = TRUE;

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

    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", NULL)))
        return FALSE;

    if (virNodeGetInfo(ctl->conn, &nodeinfo) != 0) {
        virDomainFree(dom);
	return FALSE;
    }

    if (virDomainGetInfo(dom, &info) != 0) {
        virDomainFree(dom);
        return FALSE;
    }

    cpuinfo = malloc(sizeof(virVcpuInfo)*info.nrVirtCpu);
    cpumaplen = VIR_CPU_MAPLEN(VIR_NODEINFO_MAXCPUS(nodeinfo));
    cpumap = malloc(info.nrVirtCpu * cpumaplen);

    if ((ncpus = virDomainGetVcpus(dom, 
				   cpuinfo, info.nrVirtCpu,
				   cpumap, cpumaplen)) >= 0) {
        int n;
	for (n = 0 ; n < ncpus ; n++) {
	    unsigned int m;
1133 1134 1135 1136
	    vshPrint(ctl, "%-15s %d\n", _("VCPU:"), n);
	    vshPrint(ctl, "%-15s %d\n", _("CPU:"), cpuinfo[n].cpu);
	    vshPrint(ctl, "%-15s %s\n", _("State:"),
		     _N(vshDomainVcpuStateToString(cpuinfo[n].state)));
1137 1138 1139 1140 1141
	    if (cpuinfo[n].cpuTime != 0) {
	        double cpuUsed = cpuinfo[n].cpuTime;
		
		cpuUsed /= 1000000000.0;
		
1142
		vshPrint(ctl, "%-15s %.1lfs\n", _("CPU time:"), cpuUsed);
1143
	    }
1144
	    vshPrint(ctl, "%-15s ", _("CPU Affinity:"));
1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167
	    for (m = 0 ; m < VIR_NODEINFO_MAXCPUS(nodeinfo) ; m++) {
	        vshPrint(ctl, "%c", VIR_CPU_USABLE(cpumap, cpumaplen, n, m) ? 'y' : '-');
	    }
	    vshPrint(ctl, "\n");
	    if (n < (ncpus - 1)) {
	        vshPrint(ctl, "\n");
	    }
	}
    } else {
        ret = FALSE;
    }

    free(cpumap);
    free(cpuinfo);
    virDomainFree(dom);
    return ret;
}

/*
 * "vcpupin" command
 */
static vshCmdInfo info_vcpupin[] = {
    {"syntax", "vcpupin <domain>"},
1168 1169
    {"help", gettext_noop("control domain vcpu affinity")},
    {"desc", gettext_noop("Pin domain VCPUs to host physical CPUs.")},
1170 1171 1172 1173
    {NULL, NULL}
};

static vshCmdOptDef opts_vcpupin[] = {
1174 1175 1176
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
    {"vcpu", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("vcpu number")},
    {"cpulist", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("host cpu number(s) (comma separated)")},
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 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248
    {NULL, 0, 0, NULL}
};

static int
cmdVcpupin(vshControl * ctl, vshCmd * cmd)
{
    virDomainInfo info;
    virDomainPtr dom;
    virNodeInfo nodeinfo;
    int vcpu;
    char *cpulist;
    int ret = TRUE;
    int vcpufound = 0;
    unsigned char *cpumap;
    int cpumaplen;

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

    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", NULL)))
        return FALSE;

    vcpu = vshCommandOptInt(cmd, "vcpu", &vcpufound);
    if (!vcpufound) {
        virDomainFree(dom);
        return FALSE;
    }

    if (!(cpulist = vshCommandOptString(cmd, "cpulist", NULL))) {
        virDomainFree(dom);
        return FALSE;
    }
      
    if (virNodeGetInfo(ctl->conn, &nodeinfo) != 0) {
        virDomainFree(dom);
        return FALSE;
    }

    if (virDomainGetInfo(dom, &info) != 0) {
        virDomainFree(dom);
        return FALSE;
    }

    if (vcpu >= info.nrVirtCpu) {
        virDomainFree(dom);
        return FALSE;
    }

    cpumaplen = VIR_CPU_MAPLEN(VIR_NODEINFO_MAXCPUS(nodeinfo));
    cpumap = malloc(cpumaplen);
    memset(cpumap, 0, cpumaplen);

    do {
        unsigned int cpu = atoi(cpulist);

        if (cpu < VIR_NODEINFO_MAXCPUS(nodeinfo)) {
            VIR_USE_CPU(cpumap, cpu);
        }
        cpulist = index(cpulist, ',');
        if (cpulist)
            cpulist++;
    } while (cpulist);

    if (virDomainPinVcpu(dom, vcpu, cpumap, cpumaplen) != 0) {
        ret = FALSE;
    }

    free(cpumap);
    virDomainFree(dom);
    return ret;
}

1249 1250 1251 1252 1253
/*
 * "setvcpus" command
 */
static vshCmdInfo info_setvcpus[] = {
    {"syntax", "setvcpus <domain> <count>"},
1254 1255
    {"help", gettext_noop("change number of virtual CPUs")},
    {"desc", gettext_noop("Change the number of virtual CPUs active in the guest domain.")},
1256 1257 1258 1259
    {NULL, NULL}
};

static vshCmdOptDef opts_setvcpus[] = {
1260 1261
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
    {"count", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("number of virtual CPUs")},
1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296
    {NULL, 0, 0, NULL}
};

static int
cmdSetvcpus(vshControl * ctl, vshCmd * cmd)
{
    virDomainPtr dom;
    int count;
    int ret = TRUE;

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

    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", NULL)))
        return FALSE;

    count = vshCommandOptInt(cmd, "count", &count);
    if (!count) {
        virDomainFree(dom);
        return FALSE;
    }

    if (virDomainSetVcpus(dom, count) != 0) {
        ret = FALSE;
    }

    virDomainFree(dom);
    return ret;
}

/*
 * "setmemory" command
 */
static vshCmdInfo info_setmem[] = {
    {"syntax", "setmem <domain> <bytes>"},
1297 1298
    {"help", gettext_noop("change memory allocation")},
    {"desc", gettext_noop("Change the current memory allocation in the guest domain.")},
1299 1300 1301 1302
    {NULL, NULL}
};

static vshCmdOptDef opts_setmem[] = {
1303 1304
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
    {"bytes", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("number of bytes of memory")},
1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339
    {NULL, 0, 0, NULL}
};

static int
cmdSetmem(vshControl * ctl, vshCmd * cmd)
{
    virDomainPtr dom;
    int bytes;
    int ret = TRUE;

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

    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", NULL)))
        return FALSE;

    bytes = vshCommandOptInt(cmd, "bytes", &bytes);
    if (!bytes) {
        virDomainFree(dom);
        return FALSE;
    }

    if (virDomainSetMemory(dom, bytes) != 0) {
        ret = FALSE;
    }

    virDomainFree(dom);
    return ret;
}

/*
 * "setmaxmem" command
 */
static vshCmdInfo info_setmaxmem[] = {
    {"syntax", "setmaxmem <domain> <bytes>"},
1340 1341
    {"help", gettext_noop("change maximum memory limit")},
    {"desc", gettext_noop("Change the maximum memory allocation limit in the guest domain.")},
1342 1343 1344 1345
    {NULL, NULL}
};

static vshCmdOptDef opts_setmaxmem[] = {
1346 1347
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
    {"bytes", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("maxmimum memory limit in bytes")},
1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377
    {NULL, 0, 0, NULL}
};

static int
cmdSetmaxmem(vshControl * ctl, vshCmd * cmd)
{
    virDomainPtr dom;
    int bytes;
    int ret = TRUE;

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

    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", NULL)))
        return FALSE;

    bytes = vshCommandOptInt(cmd, "bytes", &bytes);
    if (!bytes) {
        virDomainFree(dom);
        return FALSE;
    }

    if (virDomainSetMaxMemory(dom, bytes) != 0) {
        ret = FALSE;
    }

    virDomainFree(dom);
    return ret;
}

1378 1379 1380 1381 1382
/*
 * "nodeinfo" command
 */
static vshCmdInfo info_nodeinfo[] = {
    {"syntax", "nodeinfo"},
1383 1384
    {"help", gettext_noop("node information")},
    {"desc", gettext_noop("Returns basic information about the node.")},
1385 1386 1387 1388 1389 1390 1391
    {NULL, NULL}
};

static int
cmdNodeinfo(vshControl * ctl, vshCmd * cmd ATTRIBUTE_UNUSED)
{
    virNodeInfo info;
1392

1393 1394 1395 1396
    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;

    if (virNodeGetInfo(ctl->conn, &info) < 0) {
1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408
        vshError(ctl, FALSE, _("failed to get node information"));
        return FALSE;
    }
    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);

1409 1410 1411
    return TRUE;
}

1412 1413 1414 1415
/*
 * "dumpxml" command
 */
static vshCmdInfo info_dumpxml[] = {
1416
    {"syntax", "dumpxml <name>"},
1417 1418
    {"help", gettext_noop("domain information in XML")},
    {"desc", gettext_noop("Ouput the domain information as an XML dump to stdout.")},
1419
    {NULL, NULL}
1420 1421 1422
};

static vshCmdOptDef opts_dumpxml[] = {
1423
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name, id or uuid")},
1424
    {NULL, 0, 0, NULL}
1425 1426 1427
};

static int
1428 1429
cmdDumpXML(vshControl * ctl, vshCmd * cmd)
{
1430
    virDomainPtr dom;
K
Karel Zak 已提交
1431
    int ret = TRUE;
1432
    char *dump;
1433

1434 1435 1436
    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;

K
Karel Zak 已提交
1437
    if (!(dom = vshCommandOptDomain(ctl, cmd, "domain", NULL)))
1438
        return FALSE;
1439

1440 1441 1442 1443 1444 1445 1446
    dump = virDomainGetXMLDesc(dom, 0);
    if (dump != NULL) {
        printf("%s", dump);
        free(dump);
    } else {
        ret = FALSE;
    }
1447

1448 1449 1450 1451
    virDomainFree(dom);
    return ret;
}

K
Karel Zak 已提交
1452
/*
K
Karel Zak 已提交
1453
 * "domname" command
K
Karel Zak 已提交
1454
 */
K
Karel Zak 已提交
1455
static vshCmdInfo info_domname[] = {
K
Karel Zak 已提交
1456
    {"syntax", "domname <domain>"},
1457
    {"help", gettext_noop("convert a domain id or UUID to domain name")},
1458
    {NULL, NULL}
K
Karel Zak 已提交
1459 1460
};

K
Karel Zak 已提交
1461
static vshCmdOptDef opts_domname[] = {
1462
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain id or uuid")},
1463
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
1464 1465 1466
};

static int
K
Karel Zak 已提交
1467
cmdDomname(vshControl * ctl, vshCmd * cmd)
1468
{
K
Karel Zak 已提交
1469 1470 1471 1472
    virDomainPtr dom;

    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;
K
Karel Zak 已提交
1473 1474
    if (!(dom = vshCommandOptDomainBy(ctl, cmd, "domain", NULL, 
                                    VSH_DOMBYID|VSH_DOMBYUUID)))
K
Karel Zak 已提交
1475
        return FALSE;
1476

K
Karel Zak 已提交
1477 1478
    vshPrint(ctl, "%s\n", virDomainGetName(dom));
    virDomainFree(dom);
K
Karel Zak 已提交
1479 1480 1481 1482
    return TRUE;
}

/*
K
Karel Zak 已提交
1483
 * "domid" command
K
Karel Zak 已提交
1484
 */
K
Karel Zak 已提交
1485
static vshCmdInfo info_domid[] = {
K
Karel Zak 已提交
1486
    {"syntax", "domid <domain>"},
1487
    {"help", gettext_noop("convert a domain name or UUID to domain id")},
1488
    {NULL, NULL}
K
Karel Zak 已提交
1489 1490
};

K
Karel Zak 已提交
1491
static vshCmdOptDef opts_domid[] = {
1492
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain name or uuid")},
1493
    {NULL, 0, 0, NULL}
K
Karel Zak 已提交
1494 1495 1496
};

static int
K
Karel Zak 已提交
1497
cmdDomid(vshControl * ctl, vshCmd * cmd)
1498
{
1499
    virDomainPtr dom;
1500
    unsigned int id;
K
Karel Zak 已提交
1501 1502 1503

    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;
K
Karel Zak 已提交
1504 1505
    if (!(dom = vshCommandOptDomainBy(ctl, cmd, "domain", NULL, 
                                    VSH_DOMBYNAME|VSH_DOMBYUUID)))
K
Karel Zak 已提交
1506
        return FALSE;
K
Karel Zak 已提交
1507
    
1508 1509 1510 1511 1512
    id = virDomainGetID(dom);
    if (id == ((unsigned int)-1))
      vshPrint(ctl, "%s\n", "-");
    else
      vshPrint(ctl, "%d\n", id);
K
Karel Zak 已提交
1513 1514 1515
    virDomainFree(dom);
    return TRUE;
}
1516

K
Karel Zak 已提交
1517 1518 1519 1520 1521
/*
 * "domuuid" command
 */
static vshCmdInfo info_domuuid[] = {
    {"syntax", "domuuid <domain>"},
1522
    {"help", gettext_noop("convert a domain name or id to domain UUID")},
K
Karel Zak 已提交
1523 1524 1525 1526
    {NULL, NULL}
};

static vshCmdOptDef opts_domuuid[] = {
1527
    {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, gettext_noop("domain id or name")},
K
Karel Zak 已提交
1528 1529 1530 1531 1532 1533 1534 1535 1536 1537
    {NULL, 0, 0, NULL}
};

static int
cmdDomuuid(vshControl * ctl, vshCmd * cmd)
{
    virDomainPtr dom;
    char uuid[37];

    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
K
Karel Zak 已提交
1538
        return FALSE;
1539
    if (!(dom = vshCommandOptDomainBy(ctl, cmd, "domain", NULL,
K
Karel Zak 已提交
1540 1541
                                    VSH_DOMBYNAME|VSH_DOMBYID)))
        return FALSE;
1542

K
Karel Zak 已提交
1543 1544 1545
    if (virDomainGetUUIDString(dom, uuid) != -1)
        vshPrint(ctl, "%s\n", uuid);
    else
1546 1547
        vshError(ctl, FALSE, _("failed to get domain UUID"));

K
Karel Zak 已提交
1548 1549 1550
    return TRUE;
}

K
Karel Zak 已提交
1551

1552 1553 1554 1555
/*
 * "version" command
 */
static vshCmdInfo info_version[] = {
1556
    {"syntax", "version"},
1557 1558
    {"help", gettext_noop("show version")},
    {"desc", gettext_noop("Display the system version information.")},
1559
    {NULL, NULL}
1560 1561 1562 1563
};


static int
1564 1565
cmdVersion(vshControl * ctl, vshCmd * cmd ATTRIBUTE_UNUSED)
{
1566 1567
    unsigned long hvVersion;
    const char *hvType;
1568 1569 1570 1571 1572 1573 1574
    unsigned long libVersion;
    unsigned long includeVersion;
    unsigned long apiVersion;
    int ret;
    unsigned int major;
    unsigned int minor;
    unsigned int rel;
1575 1576 1577

    if (!vshConnectionUsability(ctl, ctl->conn, TRUE))
        return FALSE;
1578

1579 1580
    hvType = virConnectGetType(ctl->conn);
    if (hvType == NULL) {
1581
        vshError(ctl, FALSE, _("failed to get hypervisor type"));
1582 1583 1584
        return FALSE;
    }

1585 1586 1587 1588 1589
    includeVersion = LIBVIR_VERSION_NUMBER;
    major = includeVersion / 1000000;
    includeVersion %= 1000000;
    minor = includeVersion / 1000;
    rel = includeVersion % 1000;
1590
    vshPrint(ctl, _("Compiled against library: libvir %d.%d.%d\n"),
1591 1592 1593 1594
             major, minor, rel);

    ret = virGetVersion(&libVersion, hvType, &apiVersion);
    if (ret < 0) {
1595
        vshError(ctl, FALSE, _("failed to get the library version"));
1596 1597 1598 1599 1600 1601
        return FALSE;
    }
    major = libVersion / 1000000;
    libVersion %= 1000000;
    minor = libVersion / 1000;
    rel = libVersion % 1000;
1602
    vshPrint(ctl, _("Using library: libvir %d.%d.%d\n"),
1603
             major, minor, rel);
1604

1605 1606 1607 1608
    major = apiVersion / 1000000;
    apiVersion %= 1000000;
    minor = apiVersion / 1000;
    rel = apiVersion % 1000;
1609
    vshPrint(ctl, _("Using API: %s %d.%d.%d\n"), hvType,
1610 1611
             major, minor, rel);

1612
    ret = virConnectGetVersion(ctl->conn, &hvVersion);
1613
    if (ret < 0) {
1614
        vshError(ctl, FALSE, _("failed to get the hypervisor version"));
1615 1616 1617
        return FALSE;
    }
    if (hvVersion == 0) {
K
Karel Zak 已提交
1618
        vshPrint(ctl,
1619
                 _("Cannot extract running %s hypervisor version\n"), hvType);
1620
    } else {
1621
        major = hvVersion / 1000000;
1622
        hvVersion %= 1000000;
1623 1624
        minor = hvVersion / 1000;
        rel = hvVersion % 1000;
1625

1626
        vshPrint(ctl, _("Running hypervisor: %s %d.%d.%d\n"),
1627
                 hvType, major, minor, rel);
1628 1629 1630 1631
    }
    return TRUE;
}

K
Karel Zak 已提交
1632 1633 1634 1635
/*
 * "quit" command
 */
static vshCmdInfo info_quit[] = {
1636
    {"syntax", "quit"},
1637
    {"help", gettext_noop("quit this interactive terminal")},
1638
    {NULL, NULL}
K
Karel Zak 已提交
1639 1640 1641
};

static int
1642 1643
cmdQuit(vshControl * ctl, vshCmd * cmd ATTRIBUTE_UNUSED)
{
K
Karel Zak 已提交
1644 1645 1646 1647 1648 1649 1650 1651
    ctl->imode = FALSE;
    return TRUE;
}

/*
 * Commands
 */
static vshCmdDef commands[] = {
1652
    {"connect", cmdConnect, opts_connect, info_connect},
1653
    {"create", cmdCreate, opts_create, info_create},
1654
    {"start", cmdStart, opts_start, info_start},
K
Karel Zak 已提交
1655
    {"destroy", cmdDestroy, opts_destroy, info_destroy},
1656
    {"define", cmdDefine, opts_define, info_define},
K
Karel Zak 已提交
1657
    {"domid", cmdDomid, opts_domid, info_domid},
K
Karel Zak 已提交
1658
    {"domuuid", cmdDomuuid, opts_domuuid, info_domuuid},
1659
    {"dominfo", cmdDominfo, opts_dominfo, info_dominfo},
K
Karel Zak 已提交
1660 1661
    {"domname", cmdDomname, opts_domname, info_domname},
    {"domstate", cmdDomstate, opts_domstate, info_domstate},
1662
    {"dumpxml", cmdDumpXML, opts_dumpxml, info_dumpxml},
K
Karel Zak 已提交
1663
    {"help", cmdHelp, opts_help, info_help},
1664
    {"list", cmdList, opts_list, info_list},
K
Karel Zak 已提交
1665 1666 1667 1668
    {"nodeinfo", cmdNodeinfo, NULL, info_nodeinfo},
    {"quit", cmdQuit, NULL, info_quit},
    {"reboot", cmdReboot, opts_reboot, info_reboot},
    {"restore", cmdRestore, opts_restore, info_restore},
1669 1670
    {"resume", cmdResume, opts_resume, info_resume},
    {"save", cmdSave, opts_save, info_save},
D
Daniel Veillard 已提交
1671
    {"dump", cmdDump, opts_dump, info_dump},
1672
    {"shutdown", cmdShutdown, opts_shutdown, info_shutdown},
1673 1674 1675
    {"setmem", cmdSetmem, opts_setmem, info_setmem},
    {"setmaxmem", cmdSetmaxmem, opts_setmaxmem, info_setmaxmem},
    {"setvcpus", cmdSetvcpus, opts_setvcpus, info_setvcpus},
K
Karel Zak 已提交
1676
    {"suspend", cmdSuspend, opts_suspend, info_suspend},
1677
    {"undefine", cmdUndefine, opts_undefine, info_undefine},
1678 1679
    {"vcpuinfo", cmdVcpuinfo, opts_vcpuinfo, info_vcpuinfo},
    {"vcpupin", cmdVcpupin, opts_vcpupin, info_vcpupin},
1680 1681
    {"version", cmdVersion, NULL, info_version},
    {NULL, NULL, NULL, NULL}
K
Karel Zak 已提交
1682 1683 1684 1685 1686 1687
};

/* ---------------
 * Utils for work with command definition
 * ---------------
 */
K
Karel Zak 已提交
1688
static const char *
1689 1690
vshCmddefGetInfo(vshCmdDef * cmd, const char *name)
{
K
Karel Zak 已提交
1691
    vshCmdInfo *info;
1692

K
Karel Zak 已提交
1693
    for (info = cmd->info; info && info->name; info++) {
1694
        if (strcmp(info->name, name) == 0)
K
Karel Zak 已提交
1695 1696 1697 1698 1699 1700
            return info->data;
    }
    return NULL;
}

static vshCmdOptDef *
1701 1702
vshCmddefGetOption(vshCmdDef * cmd, const char *name)
{
K
Karel Zak 已提交
1703
    vshCmdOptDef *opt;
1704

K
Karel Zak 已提交
1705
    for (opt = cmd->opts; opt && opt->name; opt++)
1706
        if (strcmp(opt->name, name) == 0)
K
Karel Zak 已提交
1707 1708 1709 1710 1711
            return opt;
    return NULL;
}

static vshCmdOptDef *
1712 1713
vshCmddefGetData(vshCmdDef * cmd, int data_ct)
{
K
Karel Zak 已提交
1714 1715
    vshCmdOptDef *opt;

1716
    for (opt = cmd->opts; opt && opt->name; opt++) {
1717 1718
        if (opt->type == VSH_OT_DATA) {
            if (data_ct == 0)
1719 1720 1721 1722 1723
                return opt;
            else
                data_ct--;
        }
    }
K
Karel Zak 已提交
1724 1725 1726
    return NULL;
}

1727 1728 1729
/*
 * Checks for required options
 */
1730 1731
static int
vshCommandCheckOpts(vshControl * ctl, vshCmd * cmd)
1732 1733 1734
{
    vshCmdDef *def = cmd->def;
    vshCmdOptDef *d;
1735
    int err = 0;
1736 1737 1738 1739

    for (d = def->opts; d && d->name; d++) {
        if (d->flag & VSH_OFLAG_REQ) {
            vshCmdOpt *o = cmd->opts;
1740 1741 1742
            int ok = 0;

            while (o && ok == 0) {
1743
                if (o->def == d)
1744
                    ok = 1;
1745 1746 1747
                o = o->next;
            }
            if (!ok) {
1748 1749
                vshError(ctl, FALSE,
                         d->type == VSH_OT_DATA ?
1750 1751
                         _("command '%s' requires <%s> option") :
			 _("command '%s' requires --%s option"),
1752
                         def->name, d->name);
1753 1754
                err = 1;
            }
1755

1756 1757 1758 1759 1760
        }
    }
    return !err;
}

K
Karel Zak 已提交
1761
static vshCmdDef *
1762 1763
vshCmddefSearch(const char *cmdname)
{
K
Karel Zak 已提交
1764
    vshCmdDef *c;
1765

K
Karel Zak 已提交
1766
    for (c = commands; c->name; c++)
1767
        if (strcmp(c->name, cmdname) == 0)
K
Karel Zak 已提交
1768 1769 1770 1771 1772
            return c;
    return NULL;
}

static int
1773 1774
vshCmddefHelp(vshControl * ctl, const char *cmdname, int withprog)
{
K
Karel Zak 已提交
1775
    vshCmdDef *def = vshCmddefSearch(cmdname);
1776

K
Karel Zak 已提交
1777
    if (!def) {
1778
        vshError(ctl, FALSE, _("command '%s' doesn't exist"), cmdname);
1779 1780
        return FALSE;
    } else {
K
Karel Zak 已提交
1781
        vshCmdOptDef *opt;
1782 1783
        const char *desc = _N(vshCmddefGetInfo(def, "desc"));
        const char *help = _N(vshCmddefGetInfo(def, "help"));
K
Karel Zak 已提交
1784
        const char *syntax = vshCmddefGetInfo(def, "syntax");
K
Karel Zak 已提交
1785

1786
        fputs(_("  NAME\n"), stdout);
1787 1788
        fprintf(stdout, "    %s - %s\n", def->name, help);

K
Karel Zak 已提交
1789
        if (syntax) {
1790
            fputs(("\n  SYNOPSIS\n"), stdout);
K
Karel Zak 已提交
1791 1792 1793 1794 1795 1796
            if (!withprog)
                fprintf(stdout, "    %s\n", syntax);
            else
                fprintf(stdout, "    %s %s\n", progname, syntax);
        }
        if (desc) {
1797
            fputs(_("\n  DESCRIPTION\n"), stdout);
K
Karel Zak 已提交
1798 1799 1800
            fprintf(stdout, "    %s\n", desc);
        }
        if (def->opts) {
1801
            fputs(_("\n  OPTIONS\n"), stdout);
1802
            for (opt = def->opts; opt->name; opt++) {
K
Karel Zak 已提交
1803
                char buf[256];
1804 1805

                if (opt->type == VSH_OT_BOOL)
K
Karel Zak 已提交
1806
                    snprintf(buf, sizeof(buf), "--%s", opt->name);
1807
                else if (opt->type == VSH_OT_INT)
1808
                    snprintf(buf, sizeof(buf), _("--%s <number>"), opt->name);
1809
                else if (opt->type == VSH_OT_STRING)
1810
                    snprintf(buf, sizeof(buf), _("--%s <string>"), opt->name);
1811
                else if (opt->type == VSH_OT_DATA)
K
Karel Zak 已提交
1812
                    snprintf(buf, sizeof(buf), "<%s>", opt->name);
1813

K
Karel Zak 已提交
1814
                fprintf(stdout, "    %-15s  %s\n", buf, opt->help);
1815
            }
K
Karel Zak 已提交
1816 1817 1818 1819 1820 1821 1822 1823 1824 1825
        }
        fputc('\n', stdout);
    }
    return TRUE;
}

/* ---------------
 * Utils for work with runtime commands data
 * ---------------
 */
1826 1827 1828
static void
vshCommandOptFree(vshCmdOpt * arg)
{
K
Karel Zak 已提交
1829 1830
    vshCmdOpt *a = arg;

1831
    while (a) {
K
Karel Zak 已提交
1832
        vshCmdOpt *tmp = a;
1833

K
Karel Zak 已提交
1834 1835 1836 1837 1838 1839 1840 1841 1842
        a = a->next;

        if (tmp->data)
            free(tmp->data);
        free(tmp);
    }
}

static void
1843 1844
vshCommandFree(vshCmd * cmd)
{
K
Karel Zak 已提交
1845 1846
    vshCmd *c = cmd;

1847
    while (c) {
K
Karel Zak 已提交
1848
        vshCmd *tmp = c;
1849

K
Karel Zak 已提交
1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861
        c = c->next;

        if (tmp->opts)
            vshCommandOptFree(tmp->opts);
        free(tmp);
    }
}

/*
 * Returns option by name
 */
static vshCmdOpt *
1862 1863
vshCommandOpt(vshCmd * cmd, const char *name)
{
K
Karel Zak 已提交
1864
    vshCmdOpt *opt = cmd->opts;
1865 1866 1867

    while (opt) {
        if (opt->def && strcmp(opt->def->name, name) == 0)
K
Karel Zak 已提交
1868 1869 1870 1871 1872 1873 1874 1875 1876 1877
            return opt;
        opt = opt->next;
    }
    return NULL;
}

/*
 * Returns option as INT
 */
static int
1878 1879
vshCommandOptInt(vshCmd * cmd, const char *name, int *found)
{
K
Karel Zak 已提交
1880 1881
    vshCmdOpt *arg = vshCommandOpt(cmd, name);
    int res = 0;
1882

K
Karel Zak 已提交
1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893
    if (arg)
        res = atoi(arg->data);
    if (found)
        *found = arg ? TRUE : FALSE;
    return res;
}

/*
 * Returns option as STRING
 */
static char *
1894 1895
vshCommandOptString(vshCmd * cmd, const char *name, int *found)
{
K
Karel Zak 已提交
1896
    vshCmdOpt *arg = vshCommandOpt(cmd, name);
1897

K
Karel Zak 已提交
1898 1899
    if (found)
        *found = arg ? TRUE : FALSE;
1900 1901

    return arg && arg->data && *arg->data ? arg->data : NULL;
K
Karel Zak 已提交
1902 1903 1904 1905 1906 1907
}

/*
 * Returns TRUE/FALSE if the option exists
 */
static int
1908 1909
vshCommandOptBool(vshCmd * cmd, const char *name)
{
K
Karel Zak 已提交
1910 1911 1912
    return vshCommandOpt(cmd, name) ? TRUE : FALSE;
}

1913

K
Karel Zak 已提交
1914
static virDomainPtr
K
Karel Zak 已提交
1915 1916
vshCommandOptDomainBy(vshControl * ctl, vshCmd * cmd, const char *optname,
                    char **name, int flag)
1917
{
K
Karel Zak 已提交
1918 1919 1920
    virDomainPtr dom = NULL;
    char *n, *end = NULL;
    int id;
1921

K
Karel Zak 已提交
1922
    if (!(n = vshCommandOptString(cmd, optname, NULL))) {
1923
        vshError(ctl, FALSE, _("undefined domain name or id"));
1924
        return NULL;
K
Karel Zak 已提交
1925
    }
1926

K
Karel Zak 已提交
1927
    vshDebug(ctl, 5, "%s: found option <%s>: %s\n",
1928 1929
             cmd->def->name, optname, n);

K
Karel Zak 已提交
1930 1931
    if (name)
        *name = n;
1932

K
Karel Zak 已提交
1933
    /* try it by ID */
K
Karel Zak 已提交
1934 1935 1936 1937 1938 1939 1940
    if (flag & VSH_DOMBYID) {
        id = (int) strtol(n, &end, 10);
        if (id >= 0 && end && *end == '\0') {
            vshDebug(ctl, 5, "%s: <%s> seems like domain ID\n",
                     cmd->def->name, optname);
            dom = virDomainLookupByID(ctl->conn, id);
        }
1941
    }
K
Karel Zak 已提交
1942
    /* try it by UUID */
K
Karel Zak 已提交
1943
    if (dom==NULL && (flag & VSH_DOMBYUUID) && strlen(n)==36) {
K
Karel Zak 已提交
1944 1945
        vshDebug(ctl, 5, "%s: <%s> tring as domain UUID\n",
                cmd->def->name, optname);
K
Karel Zak 已提交
1946
        dom = virDomainLookupByUUIDString(ctl->conn, n);
K
Karel Zak 已提交
1947
    }
K
Karel Zak 已提交
1948
    /* try it by NAME */
K
Karel Zak 已提交
1949
    if (dom==NULL && (flag & VSH_DOMBYNAME)) {
K
Karel Zak 已提交
1950
        vshDebug(ctl, 5, "%s: <%s> tring as domain NAME\n",
1951
                 cmd->def->name, optname);
K
Karel Zak 已提交
1952
        dom = virDomainLookupByName(ctl->conn, n);
1953
    }
K
Karel Zak 已提交
1954

1955
    if (!dom)
1956
        vshError(ctl, FALSE, _("failed to get domain '%s'"), n);
1957

K
Karel Zak 已提交
1958 1959 1960
    return dom;
}

K
Karel Zak 已提交
1961 1962 1963 1964
/*
 * Executes command(s) and returns return code from last command
 */
static int
1965 1966
vshCommandRun(vshControl * ctl, vshCmd * cmd)
{
K
Karel Zak 已提交
1967
    int ret = TRUE;
1968 1969

    while (cmd) {
K
Karel Zak 已提交
1970
        struct timeval before, after;
1971

K
Karel Zak 已提交
1972 1973
        if (ctl->timing)
            GETTIMEOFDAY(&before);
1974

K
Karel Zak 已提交
1975 1976 1977 1978
        ret = cmd->def->handler(ctl, cmd);

        if (ctl->timing)
            GETTIMEOFDAY(&after);
1979 1980

        if (strcmp(cmd->def->name, "quit") == 0)        /* hack ... */
K
Karel Zak 已提交
1981 1982 1983
            return ret;

        if (ctl->timing)
1984
            vshPrint(ctl, _("\n(Time: %.3f ms)\n\n"),
1985 1986
                     DIFF_MSEC(&after, &before));
        else
K
Karel Zak 已提交
1987
            vshPrintExtra(ctl, "\n");
K
Karel Zak 已提交
1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002
        cmd = cmd->next;
    }
    return ret;
}

/* ---------------
 * Command string parsing
 * ---------------
 */
#define VSH_TK_ERROR    -1
#define VSH_TK_NONE    0
#define VSH_TK_OPTION    1
#define VSH_TK_DATA    2
#define VSH_TK_END    3

2003 2004 2005
static int
vshCommandGetToken(vshControl * ctl, char *str, char **end, char **res)
{
K
Karel Zak 已提交
2006 2007 2008 2009 2010
    int tk = VSH_TK_NONE;
    int quote = FALSE;
    int sz = 0;
    char *p = str;
    char *tkstr = NULL;
2011

K
Karel Zak 已提交
2012
    *end = NULL;
2013 2014

    while (p && *p && isblank((unsigned char) *p))
K
Karel Zak 已提交
2015
        p++;
2016 2017

    if (p == NULL || *p == '\0')
K
Karel Zak 已提交
2018
        return VSH_TK_END;
2019 2020
    if (*p == ';') {
        *end = ++p;             /* = \0 or begi of next command */
K
Karel Zak 已提交
2021 2022
        return VSH_TK_END;
    }
2023
    while (*p) {
K
Karel Zak 已提交
2024
        /* end of token is blank space or ';' */
2025
        if ((quote == FALSE && isblank((unsigned char) *p)) || *p == ';')
K
Karel Zak 已提交
2026
            break;
2027

2028
        /* end of option name could be '=' */
2029 2030
        if (tk == VSH_TK_OPTION && *p == '=') {
            p++;                /* skip '=' */
2031 2032
            break;
        }
2033 2034 2035 2036

        if (tk == VSH_TK_NONE) {
            if (*p == '-' && *(p + 1) == '-' && *(p + 2)
                && isalnum((unsigned char) *(p + 2))) {
K
Karel Zak 已提交
2037
                tk = VSH_TK_OPTION;
2038
                p += 2;
K
Karel Zak 已提交
2039 2040
            } else {
                tk = VSH_TK_DATA;
2041 2042
                if (*p == '"') {
                    quote = TRUE;
K
Karel Zak 已提交
2043 2044 2045 2046 2047
                    p++;
                } else {
                    quote = FALSE;
                }
            }
2048 2049
            tkstr = p;          /* begin of token */
        } else if (quote && *p == '"') {
K
Karel Zak 已提交
2050 2051
            quote = FALSE;
            p++;
2052
            break;              /* end of "..." token */
K
Karel Zak 已提交
2053 2054 2055 2056 2057
        }
        p++;
        sz++;
    }
    if (quote) {
2058
        vshError(ctl, FALSE, _("missing \""));
K
Karel Zak 已提交
2059 2060
        return VSH_TK_ERROR;
    }
2061
    if (tkstr == NULL || *tkstr == '\0' || p == NULL)
K
Karel Zak 已提交
2062
        return VSH_TK_END;
2063
    if (sz == 0)
K
Karel Zak 已提交
2064
        return VSH_TK_END;
2065

2066
    *res = vshMalloc(ctl, sz + 1);
K
Karel Zak 已提交
2067
    memcpy(*res, tkstr, sz);
2068
    *(*res + sz) = '\0';
K
Karel Zak 已提交
2069 2070 2071 2072 2073 2074

    *end = p;
    return tk;
}

static int
2075 2076
vshCommandParse(vshControl * ctl, char *cmdstr)
{
K
Karel Zak 已提交
2077 2078 2079 2080
    char *str;
    char *tkdata = NULL;
    vshCmd *clast = NULL;
    vshCmdOpt *first = NULL;
2081

K
Karel Zak 已提交
2082 2083 2084 2085
    if (ctl->cmd) {
        vshCommandFree(ctl->cmd);
        ctl->cmd = NULL;
    }
2086 2087

    if (cmdstr == NULL || *cmdstr == '\0')
K
Karel Zak 已提交
2088
        return FALSE;
2089

K
Karel Zak 已提交
2090
    str = cmdstr;
2091
    while (str && *str) {
K
Karel Zak 已提交
2092 2093 2094
        vshCmdOpt *last = NULL;
        vshCmdDef *cmd = NULL;
        int tk = VSH_TK_NONE;
2095
        int data_ct = 0;
2096

K
Karel Zak 已提交
2097
        first = NULL;
2098 2099

        while (tk != VSH_TK_END) {
K
Karel Zak 已提交
2100 2101
            char *end = NULL;
            vshCmdOptDef *opt = NULL;
2102

K
Karel Zak 已提交
2103
            tkdata = NULL;
2104

K
Karel Zak 已提交
2105 2106
            /* get token */
            tk = vshCommandGetToken(ctl, str, &end, &tkdata);
2107

K
Karel Zak 已提交
2108
            str = end;
2109 2110

            if (tk == VSH_TK_END)
K
Karel Zak 已提交
2111
                break;
2112
            if (tk == VSH_TK_ERROR)
K
Karel Zak 已提交
2113
                goto syntaxError;
2114 2115

            if (cmd == NULL) {
K
Karel Zak 已提交
2116
                /* first token must be command name */
2117 2118
                if (tk != VSH_TK_DATA) {
                    vshError(ctl, FALSE,
2119
                             _("unexpected token (command name): '%s'"),
2120
                             tkdata);
K
Karel Zak 已提交
2121 2122 2123
                    goto syntaxError;
                }
                if (!(cmd = vshCmddefSearch(tkdata))) {
2124
                    vshError(ctl, FALSE, _("unknown command: '%s'"), tkdata);
2125
                    goto syntaxError;   /* ... or ignore this command only? */
K
Karel Zak 已提交
2126 2127
                }
                free(tkdata);
2128
            } else if (tk == VSH_TK_OPTION) {
K
Karel Zak 已提交
2129 2130
                if (!(opt = vshCmddefGetOption(cmd, tkdata))) {
                    vshError(ctl, FALSE,
2131
                             _("command '%s' doesn't support option --%s"),
2132
                             cmd->name, tkdata);
K
Karel Zak 已提交
2133 2134
                    goto syntaxError;
                }
2135
                free(tkdata);   /* option name */
K
Karel Zak 已提交
2136 2137 2138 2139 2140
                tkdata = NULL;

                if (opt->type != VSH_OT_BOOL) {
                    /* option data */
                    tk = vshCommandGetToken(ctl, str, &end, &tkdata);
2141 2142
                    str = end;
                    if (tk == VSH_TK_ERROR)
K
Karel Zak 已提交
2143
                        goto syntaxError;
2144
                    if (tk != VSH_TK_DATA) {
K
Karel Zak 已提交
2145
                        vshError(ctl, FALSE,
2146
                                 _("expected syntax: --%s <%s>"),
2147 2148
                                 opt->name,
                                 opt->type ==
2149
                                 VSH_OT_INT ? _("number") : _("string"));
K
Karel Zak 已提交
2150 2151 2152
                        goto syntaxError;
                    }
                }
2153
            } else if (tk == VSH_TK_DATA) {
2154
                if (!(opt = vshCmddefGetData(cmd, data_ct++))) {
2155
                    vshError(ctl, FALSE, _("unexpected data '%s'"), tkdata);
K
Karel Zak 已提交
2156 2157 2158 2159 2160
                    goto syntaxError;
                }
            }
            if (opt) {
                /* save option */
2161
                vshCmdOpt *arg = vshMalloc(ctl, sizeof(vshCmdOpt));
2162

K
Karel Zak 已提交
2163 2164 2165 2166
                arg->def = opt;
                arg->data = tkdata;
                arg->next = NULL;
                tkdata = NULL;
2167

K
Karel Zak 已提交
2168 2169 2170 2171 2172
                if (!first)
                    first = arg;
                if (last)
                    last->next = arg;
                last = arg;
2173

K
Karel Zak 已提交
2174
                vshDebug(ctl, 4, "%s: %s(%s): %s\n",
2175 2176
                         cmd->name,
                         opt->name,
2177
                         tk == VSH_TK_OPTION ? _("OPTION") : _("DATA"),
2178
                         arg->data);
K
Karel Zak 已提交
2179 2180 2181 2182
            }
            if (!str)
                break;
        }
2183

K
Karel Zak 已提交
2184 2185
        /* commad parsed -- allocate new struct for the command */
        if (cmd) {
2186
            vshCmd *c = vshMalloc(ctl, sizeof(vshCmd));
2187

K
Karel Zak 已提交
2188 2189 2190 2191
            c->opts = first;
            c->def = cmd;
            c->next = NULL;

2192 2193
            if (!vshCommandCheckOpts(ctl, c))
                goto syntaxError;
2194

K
Karel Zak 已提交
2195 2196 2197 2198 2199 2200 2201
            if (!ctl->cmd)
                ctl->cmd = c;
            if (clast)
                clast->next = c;
            clast = c;
        }
    }
2202

K
Karel Zak 已提交
2203 2204
    return TRUE;

2205
  syntaxError:
K
Karel Zak 已提交
2206 2207 2208 2209 2210 2211
    if (ctl->cmd)
        vshCommandFree(ctl->cmd);
    if (first)
        vshCommandOptFree(first);
    if (tkdata)
        free(tkdata);
2212
    return FALSE;
K
Karel Zak 已提交
2213 2214 2215 2216 2217 2218 2219
}


/* ---------------
 * Misc utils  
 * ---------------
 */
K
Karel Zak 已提交
2220
static const char *
2221 2222
vshDomainStateToString(int state)
{
K
Karel Zak 已提交
2223 2224
    switch (state) {
        case VIR_DOMAIN_RUNNING:
2225
            return gettext_noop("running");
K
Karel Zak 已提交
2226
        case VIR_DOMAIN_BLOCKED:
2227
            return gettext_noop("blocked");
K
Karel Zak 已提交
2228
        case VIR_DOMAIN_PAUSED:
2229
            return gettext_noop("paused");
K
Karel Zak 已提交
2230
        case VIR_DOMAIN_SHUTDOWN:
2231
            return gettext_noop("in shutdown");
K
Karel Zak 已提交
2232
        case VIR_DOMAIN_SHUTOFF:
2233
            return gettext_noop("shut off");
K
Karel Zak 已提交
2234
        case VIR_DOMAIN_CRASHED:
2235
            return gettext_noop("crashed");
K
Karel Zak 已提交
2236
        default:
2237
            return gettext_noop("no state");  /* = dom0 state */
K
Karel Zak 已提交
2238 2239 2240 2241
    }
    return NULL;
}

2242 2243 2244 2245 2246
static const char *
vshDomainVcpuStateToString(int state)
{
    switch (state) {
        case VIR_VCPU_OFFLINE:
2247
            return gettext_noop("offline");
2248
        case VIR_VCPU_BLOCKED:
2249
            return gettext_noop("blocked");
2250
        case VIR_VCPU_RUNNING:
2251
            return gettext_noop("running");
2252
        default:
2253
            return gettext_noop("no state");
2254 2255 2256 2257
    }
    return NULL;
}

K
Karel Zak 已提交
2258
static int
2259 2260
vshConnectionUsability(vshControl * ctl, virConnectPtr conn, int showerror)
{
K
Karel Zak 已提交
2261 2262 2263 2264 2265
    /* TODO: use something like virConnectionState() to 
     *       check usability of the connection 
     */
    if (!conn) {
        if (showerror)
2266
            vshError(ctl, FALSE, _("no valid connection"));
K
Karel Zak 已提交
2267 2268 2269 2270 2271
        return FALSE;
    }
    return TRUE;
}

K
Karel Zak 已提交
2272 2273
static void
vshDebug(vshControl * ctl, int level, const char *format, ...)
2274
{
K
Karel Zak 已提交
2275 2276 2277 2278 2279 2280 2281 2282
    va_list ap;

    if (level > ctl->debug)
        return;

    va_start(ap, format);
    vfprintf(stdout, format, ap);
    va_end(ap);
K
Karel Zak 已提交
2283 2284 2285
}

static void
K
Karel Zak 已提交
2286
vshPrintExtra(vshControl * ctl, const char *format, ...)
2287
{
K
Karel Zak 已提交
2288
    va_list ap;
2289

K
Karel Zak 已提交
2290
    if (ctl->quiet == TRUE)
K
Karel Zak 已提交
2291
        return;
2292

K
Karel Zak 已提交
2293
    va_start(ap, format);
2294
    vfprintf(stdout, format, ap);
K
Karel Zak 已提交
2295 2296 2297
    va_end(ap);
}

K
Karel Zak 已提交
2298

K
Karel Zak 已提交
2299
static void
2300 2301
vshError(vshControl * ctl, int doexit, const char *format, ...)
{
K
Karel Zak 已提交
2302
    va_list ap;
2303

K
Karel Zak 已提交
2304
    if (doexit)
2305
        fprintf(stderr, _("%s: error: "), progname);
K
Karel Zak 已提交
2306
    else
2307
        fputs(_("error: "), stderr);
2308

K
Karel Zak 已提交
2309 2310 2311 2312 2313
    va_start(ap, format);
    vfprintf(stderr, format, ap);
    va_end(ap);

    fputc('\n', stderr);
2314

K
Karel Zak 已提交
2315
    if (doexit) {
2316 2317
        if (ctl)
            vshDeinit(ctl);
K
Karel Zak 已提交
2318 2319 2320 2321
        exit(EXIT_FAILURE);
    }
}

2322 2323 2324 2325 2326 2327 2328
static void *
_vshMalloc(vshControl * ctl, size_t size, const char *filename, int line)
{
    void *x;

    if ((x = malloc(size)))
        return x;
2329 2330
    vshError(ctl, TRUE, _("%s: %d: failed to allocate %d bytes"),
	     filename, line, (int) size);
2331 2332 2333 2334 2335 2336 2337 2338 2339 2340
    return NULL;
}

static void *
_vshCalloc(vshControl * ctl, size_t nmemb, size_t size, const char *filename, int line)
{
    void *x;

    if ((x = calloc(nmemb, size)))
        return x;
2341 2342
    vshError(ctl, TRUE, _("%s: %d: failed to allocate %d bytes"),
	     filename, line, (int) (size*nmemb));
2343 2344 2345 2346 2347 2348 2349 2350 2351 2352
    return NULL;
}

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

    if ((x = strdup(s)))
        return x;
2353 2354
    vshError(ctl, TRUE, _("%s: %d: failed to allocate %d bytes"),
	     filename, line, strlen(s));
2355 2356 2357
    return NULL;
}

K
Karel Zak 已提交
2358 2359 2360 2361
/*
 * Initialize vistsh
 */
static int
2362 2363
vshInit(vshControl * ctl)
{
K
Karel Zak 已提交
2364 2365 2366 2367
    if (ctl->conn)
        return FALSE;

    ctl->uid = getuid();
2368

2369 2370
    /* set up the library error handler */
    virSetErrorFunc(NULL, virshErrorHandler);
2371

2372 2373 2374 2375
    /* basic connection to hypervisor, for Xen connections unless
       we're root open a read only connections. Allow 'test' HV
       to be RW all the time though */
    if (ctl->uid == 0 || (ctl->name && !strncmp(ctl->name, "test", 4)))
K
Karel Zak 已提交
2376
        ctl->conn = virConnectOpen(ctl->name);
K
Karel Zak 已提交
2377
    else
K
Karel Zak 已提交
2378
        ctl->conn = virConnectOpenReadOnly(ctl->name);
2379

K
Karel Zak 已提交
2380
    if (!ctl->conn)
2381
        vshError(ctl, TRUE, _("failed to connect to the hypervisor"));
K
Karel Zak 已提交
2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396

    return TRUE;
}

/* -----------------
 * Readline stuff
 * -----------------
 */

/* 
 * Generator function for command completion.  STATE lets us
 * know whether to start from scratch; without any state
 * (i.e. STATE == 0), then we start at the top of the list. 
 */
static char *
2397 2398
vshReadlineCommandGenerator(const char *text, int state)
{
K
Karel Zak 已提交
2399
    static int list_index, len;
K
Karel Zak 已提交
2400
    const char *name;
K
Karel Zak 已提交
2401 2402 2403 2404 2405 2406 2407

    /* If this is a new word to complete, initialize now.  This
     * includes saving the length of TEXT for efficiency, and
     * initializing the index variable to 0. 
     */
    if (!state) {
        list_index = 0;
2408
        len = strlen(text);
K
Karel Zak 已提交
2409 2410 2411 2412 2413
    }

    /* Return the next name which partially matches from the
     * command list. 
     */
K
Karel Zak 已提交
2414
    while ((name = commands[list_index].name)) {
K
Karel Zak 已提交
2415
        list_index++;
2416
        if (strncmp(name, text, len) == 0)
2417
            return vshStrdup(NULL, name);
K
Karel Zak 已提交
2418 2419 2420 2421 2422 2423 2424
    }

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

static char *
2425 2426
vshReadlineOptionsGenerator(const char *text, int state)
{
K
Karel Zak 已提交
2427 2428
    static int list_index, len;
    static vshCmdDef *cmd = NULL;
K
Karel Zak 已提交
2429
    const char *name;
K
Karel Zak 已提交
2430 2431 2432 2433 2434 2435 2436 2437 2438

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

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

2439
        cmdname = vshCalloc(NULL, (p - rl_line_buffer) + 1, 1);
2440
        memcpy(cmdname, rl_line_buffer, p - rl_line_buffer);
K
Karel Zak 已提交
2441 2442 2443

        cmd = vshCmddefSearch(cmdname);
        list_index = 0;
2444
        len = strlen(text);
K
Karel Zak 已提交
2445 2446 2447 2448 2449
        free(cmdname);
    }

    if (!cmd)
        return NULL;
2450

K
Karel Zak 已提交
2451
    while ((name = cmd->opts[list_index].name)) {
K
Karel Zak 已提交
2452 2453
        vshCmdOptDef *opt = &cmd->opts[list_index];
        char *res;
2454

K
Karel Zak 已提交
2455
        list_index++;
2456

K
Karel Zak 已提交
2457
        if (opt->type == VSH_OT_DATA)
K
Karel Zak 已提交
2458 2459
            /* ignore non --option */
            continue;
2460

K
Karel Zak 已提交
2461
        if (len > 2) {
2462
            if (strncmp(name, text + 2, len - 2))
K
Karel Zak 已提交
2463 2464
                continue;
        }
2465
        res = vshMalloc(NULL, strlen(name) + 3);
K
Karel Zak 已提交
2466 2467 2468 2469 2470 2471 2472 2473 2474
        sprintf(res, "--%s", name);
        return res;
    }

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

static char **
2475 2476 2477
vshReadlineCompletion(const char *text, int start,
                      int end ATTRIBUTE_UNUSED)
{
K
Karel Zak 已提交
2478 2479
    char **matches = (char **) NULL;

2480
    if (start == 0)
K
Karel Zak 已提交
2481
        /* command name generator */
2482
        matches = rl_completion_matches(text, vshReadlineCommandGenerator);
K
Karel Zak 已提交
2483 2484
    else
        /* commands options */
2485
        matches = rl_completion_matches(text, vshReadlineOptionsGenerator);
K
Karel Zak 已提交
2486 2487 2488 2489 2490
    return matches;
}


static void
2491 2492
vshReadlineInit(void)
{
K
Karel Zak 已提交
2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503
    /* 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;
}

/*
 * Deinitliaze virsh
 */
static int
2504 2505
vshDeinit(vshControl * ctl)
{
K
Karel Zak 已提交
2506
    if (ctl->conn) {
2507 2508 2509 2510
        if (virConnectClose(ctl->conn) != 0) {
            ctl->conn = NULL;   /* prevent recursive call from vshError() */
            vshError(ctl, TRUE,
                     "failed to disconnect from the hypervisor");
K
Karel Zak 已提交
2511 2512 2513 2514
        }
    }
    return TRUE;
}
2515

K
Karel Zak 已提交
2516 2517 2518 2519
/*
 * Print usage
 */
static void
2520 2521
vshUsage(vshControl * ctl, const char *cmdname)
{
K
Karel Zak 已提交
2522
    vshCmdDef *cmd;
2523

K
Karel Zak 已提交
2524 2525
    /* global help */
    if (!cmdname) {
2526 2527 2528 2529 2530 2531 2532 2533 2534
        fprintf(stdout, _("\n%s [options] [commands]\n\n"
			  "  options:\n"
			  "    -c | --connect <uri>    hypervisor connection URI\n"
			  "    -d | --debug <num>      debug level [0-5]\n"
			  "    -h | --help             this help\n"
			  "    -q | --quiet            quiet mode\n"
			  "    -t | --timing           print timing information\n"
			  "    -v | --version          program version\n\n"
			  "  commands (non interactive mode):\n"), progname);
2535 2536 2537

        for (cmd = commands; cmd->name; cmd++)
            fprintf(stdout,
2538 2539
                    "    %-15s %s\n", cmd->name, _N(vshCmddefGetInfo(cmd,
								     "help")));
2540 2541

        fprintf(stdout,
2542
                _("\n  (specify --help <command> for details about the command)\n\n"));
K
Karel Zak 已提交
2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553
        return;
    }
    if (!vshCmddefHelp(ctl, cmdname, TRUE))
        exit(EXIT_FAILURE);
}

/*
 * argv[]:  virsh [options] [command]
 *
 */
static int
2554 2555
vshParseArgv(vshControl * ctl, int argc, char **argv)
{
K
Karel Zak 已提交
2556 2557
    char *last = NULL;
    int i, end = 0, help = 0;
2558
    int arg, idx = 0;
K
Karel Zak 已提交
2559
    struct option opt[] = {
2560 2561 2562 2563 2564
        {"debug", 1, 0, 'd'},
        {"help", 0, 0, 'h'},
        {"quiet", 0, 0, 'q'},
        {"timing", 0, 0, 't'},
        {"version", 0, 0, 'v'},
K
Karel Zak 已提交
2565
        {"connect", 1, 0, 'c'},
K
Karel Zak 已提交
2566
        {0, 0, 0, 0}
2567 2568
    };

K
Karel Zak 已提交
2569 2570

    if (argc < 2)
K
Karel Zak 已提交
2571
        return TRUE;
2572

2573
    /* look for begin of the command, for example:
K
Karel Zak 已提交
2574 2575 2576 2577
     *   ./virsh --debug 5 -q command --cmdoption
     *                  <--- ^ --->
     *        getopt() stuff | command suff
     */
2578
    for (i = 1; i < argc; i++) {
K
Karel Zak 已提交
2579 2580
        if (*argv[i] != '-') {
            int valid = FALSE;
2581

K
Karel Zak 已提交
2582 2583 2584 2585
            /* non "--option" argv, is it command? */
            if (last) {
                struct option *o;
                int sz = strlen(last);
2586 2587 2588

                for (o = opt; o->name; o++) {
                    if (sz == 2 && *(last + 1) == o->val)
K
Karel Zak 已提交
2589 2590
                        /* valid virsh short option */
                        valid = TRUE;
2591
                    else if (sz > 2 && strcmp(o->name, last + 2) == 0)
K
Karel Zak 已提交
2592 2593 2594 2595 2596 2597 2598 2599 2600 2601 2602 2603
                        /* valid virsh long option */
                        valid = TRUE;
                }
            }
            if (!valid) {
                end = i;
                break;
            }
        }
        last = argv[i];
    }
    end = end ? : argc;
2604

K
Karel Zak 已提交
2605
    /* standard (non-command) options */
2606 2607
    while ((arg = getopt_long(end, argv, "d:hqtv", opt, &idx)) != -1) {
        switch (arg) {
K
Karel Zak 已提交
2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619
            case 'd':
                ctl->debug = atoi(optarg);
                break;
            case 'h':
                help = 1;
                break;
            case 'q':
                ctl->quiet = TRUE;
                break;
            case 't':
                ctl->timing = TRUE;
                break;
K
Karel Zak 已提交
2620 2621 2622
            case 'c':
                ctl->name = vshStrdup(ctl, optarg);
                break;
K
Karel Zak 已提交
2623 2624 2625 2626
            case 'v':
                fprintf(stdout, "%s\n", VERSION);
                exit(EXIT_SUCCESS);
            default:
2627
                vshError(ctl, TRUE,
2628
			 _("unsupported option '-%c'. See --help."), arg);
K
Karel Zak 已提交
2629 2630 2631 2632 2633 2634 2635 2636
                break;
        }
    }

    if (help) {
        /* global or command specific help */
        vshUsage(ctl, argc > end ? argv[end] : NULL);
        exit(EXIT_SUCCESS);
2637 2638
    }

K
Karel Zak 已提交
2639 2640 2641
    if (argc > end) {
        /* parse command */
        char *cmdstr;
2642 2643
        int sz = 0, ret;

K
Karel Zak 已提交
2644 2645
        ctl->imode = FALSE;

2646 2647 2648
        for (i = end; i < argc; i++)
            sz += strlen(argv[i]) + 1;  /* +1 is for blank space between items */

2649
        cmdstr = vshCalloc(ctl, sz + 1, 1);
2650 2651

        for (i = end; i < argc; i++) {
K
Karel Zak 已提交
2652 2653 2654 2655
            strncat(cmdstr, argv[i], sz);
            sz -= strlen(argv[i]);
            strncat(cmdstr, " ", sz--);
        }
K
Karel Zak 已提交
2656
        vshDebug(ctl, 2, "command: \"%s\"\n", cmdstr);
K
Karel Zak 已提交
2657
        ret = vshCommandParse(ctl, cmdstr);
2658

K
Karel Zak 已提交
2659 2660 2661 2662 2663 2664
        free(cmdstr);
        return ret;
    }
    return TRUE;
}

2665 2666 2667 2668
int
main(int argc, char **argv)
{
    vshControl _ctl, *ctl = &_ctl;
2669
    char *defaultConn;
K
Karel Zak 已提交
2670 2671
    int ret = TRUE;

2672 2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684
    if (!setlocale(LC_ALL, "")) {
        perror("setlocale");
	return -1;
    }
    if (!bindtextdomain(GETTEXT_PACKAGE, LOCALEBASEDIR)) {
        perror("bindtextdomain");
	return -1;
    }
    if (!textdomain(GETTEXT_PACKAGE)) {
        perror("textdomain");
	return -1;
    }

2685
    if (!(progname = strrchr(argv[0], '/')))
K
Karel Zak 已提交
2686 2687 2688
        progname = argv[0];
    else
        progname++;
2689

K
Karel Zak 已提交
2690
    memset(ctl, 0, sizeof(vshControl));
2691
    ctl->imode = TRUE;          /* default is interactive mode */
K
Karel Zak 已提交
2692

2693 2694 2695 2696
    if ((defaultConn = getenv("VIRSH_DEFAULT_CONNECT_URI"))) {
      ctl->name = strdup(defaultConn);
    }

K
Karel Zak 已提交
2697 2698
    if (!vshParseArgv(ctl, argc, argv))
        exit(EXIT_FAILURE);
2699

K
Karel Zak 已提交
2700 2701
    if (!vshInit(ctl))
        exit(EXIT_FAILURE);
2702

K
Karel Zak 已提交
2703
    if (!ctl->imode) {
2704
        ret = vshCommandRun(ctl, ctl->cmd);
2705
    } else {
K
Karel Zak 已提交
2706 2707
        /* interactive mode */
        if (!ctl->quiet) {
K
Karel Zak 已提交
2708
            vshPrint(ctl,
2709
                     _("Welcome to %s, the virtualization interactive terminal.\n\n"),
2710
                     progname);
K
Karel Zak 已提交
2711
            vshPrint(ctl,
2712 2713
                     _("Type:  'help' for help with commands\n"
		       "       'quit' to quit\n\n"));
K
Karel Zak 已提交
2714
        }
K
Karel Zak 已提交
2715
        vshReadlineInit();
K
Karel Zak 已提交
2716
        do {
2717 2718 2719 2720
            ctl->cmdstr =
                readline(ctl->uid == 0 ? VSH_PROMPT_RW : VSH_PROMPT_RO);
            if (ctl->cmdstr == NULL)
                break;          /* EOF */
K
Karel Zak 已提交
2721 2722 2723 2724 2725 2726 2727
            if (*ctl->cmdstr) {
                add_history(ctl->cmdstr);
                if (vshCommandParse(ctl, ctl->cmdstr))
                    vshCommandRun(ctl, ctl->cmd);
            }
            free(ctl->cmdstr);
            ctl->cmdstr = NULL;
2728
        } while (ctl->imode);
K
Karel Zak 已提交
2729

2730 2731
        if (ctl->cmdstr == NULL)
            fputc('\n', stdout);        /* line break after alone prompt */
K
Karel Zak 已提交
2732
    }
2733

K
Karel Zak 已提交
2734 2735
    vshDeinit(ctl);
    exit(ret ? EXIT_SUCCESS : EXIT_FAILURE);
2736
}
K
Karel Zak 已提交
2737 2738 2739 2740 2741 2742

/*
 * vim: set tabstop=4:
 * vim: set shiftwidth=4:
 * vim: set expandtab:
 */