Statistics
| Branch: | Revision:

root / cmd.c @ 371c6489

History | View | Annotate | Download (12.2 kB)

1
/*
2
 * Copyright (c) 2003-2005 Silicon Graphics, Inc.
3
 * All Rights Reserved.
4
 *
5
 * This program is free software; you can redistribute it and/or
6
 * modify it under the terms of the GNU General Public License as
7
 * published by the Free Software Foundation.
8
 *
9
 * This program is distributed in the hope that it would be useful,
10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 * GNU General Public License for more details.
13
 *
14
 * You should have received a copy of the GNU General Public License
15
 * along with this program; if not, see <http://www.gnu.org/licenses/>.
16
 */
17

    
18
#include <stdio.h>
19
#include <stdlib.h>
20
#include <string.h>
21
#include <ctype.h>
22
#include <errno.h>
23
#include <sys/time.h>
24
#include <getopt.h>
25

    
26
#include "cmd.h"
27
#include "qemu-aio.h"
28

    
29
#define _(x)        x        /* not gettext support yet */
30

    
31
/* from libxcmd/command.c */
32

    
33
cmdinfo_t        *cmdtab;
34
int                ncmds;
35

    
36
static argsfunc_t        args_func;
37
static checkfunc_t        check_func;
38
static int                ncmdline;
39
static char                **cmdline;
40

    
41
static int
42
compare(const void *a, const void *b)
43
{
44
        return strcmp(((const cmdinfo_t *)a)->name,
45
                      ((const cmdinfo_t *)b)->name);
46
}
47

    
48
void add_command(const cmdinfo_t *ci)
49
{
50
    cmdtab = g_realloc((void *)cmdtab, ++ncmds * sizeof(*cmdtab));
51
    cmdtab[ncmds - 1] = *ci;
52
    qsort(cmdtab, ncmds, sizeof(*cmdtab), compare);
53
}
54

    
55
static int
56
check_command(
57
        const cmdinfo_t        *ci)
58
{
59
        if (check_func)
60
                return check_func(ci);
61
        return 1;
62
}
63

    
64
void
65
add_check_command(
66
        checkfunc_t        cf)
67
{
68
        check_func = cf;
69
}
70

    
71
int
72
command_usage(
73
        const cmdinfo_t *ci)
74
{
75
        printf("%s %s -- %s\n", ci->name, ci->args, ci->oneline);
76
        return 0;
77
}
78

    
79
int
80
command(
81
        const cmdinfo_t        *ct,
82
        int                argc,
83
        char                **argv)
84
{
85
        char                *cmd = argv[0];
86

    
87
        if (!check_command(ct))
88
                return 0;
89

    
90
        if (argc-1 < ct->argmin || (ct->argmax != -1 && argc-1 > ct->argmax)) {
91
                if (ct->argmax == -1)
92
                        fprintf(stderr,
93
        _("bad argument count %d to %s, expected at least %d arguments\n"),
94
                                argc-1, cmd, ct->argmin);
95
                else if (ct->argmin == ct->argmax)
96
                        fprintf(stderr,
97
        _("bad argument count %d to %s, expected %d arguments\n"),
98
                                argc-1, cmd, ct->argmin);
99
                else
100
                        fprintf(stderr,
101
        _("bad argument count %d to %s, expected between %d and %d arguments\n"),
102
                        argc-1, cmd, ct->argmin, ct->argmax);
103
                return 0;
104
        }
105
        optind = 0;
106
        return ct->cfunc(argc, argv);
107
}
108

    
109
const cmdinfo_t *
110
find_command(
111
        const char        *cmd)
112
{
113
        cmdinfo_t        *ct;
114

    
115
        for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) {
116
                if (strcmp(ct->name, cmd) == 0 ||
117
                    (ct->altname && strcmp(ct->altname, cmd) == 0))
118
                        return (const cmdinfo_t *)ct;
119
        }
120
        return NULL;
121
}
122

    
123
void add_user_command(char *optarg)
124
{
125
    cmdline = g_realloc(cmdline, ++ncmdline * sizeof(char *));
126
    cmdline[ncmdline-1] = optarg;
127
}
128

    
129
static int
130
args_command(
131
        int        index)
132
{
133
        if (args_func)
134
                return args_func(index);
135
        return 0;
136
}
137

    
138
void
139
add_args_command(
140
        argsfunc_t        af)
141
{
142
        args_func = af;
143
}
144

    
145
static void prep_fetchline(void *opaque)
146
{
147
    int *fetchable = opaque;
148

    
149
    qemu_aio_set_fd_handler(STDIN_FILENO, NULL, NULL, NULL, NULL, NULL);
150
    *fetchable= 1;
151
}
152

    
153
static char *get_prompt(void);
154

    
155
void command_loop(void)
156
{
157
    int c, i, j = 0, done = 0, fetchable = 0, prompted = 0;
158
    char *input;
159
    char **v;
160
    const cmdinfo_t *ct;
161

    
162
    for (i = 0; !done && i < ncmdline; i++) {
163
        input = strdup(cmdline[i]);
164
        if (!input) {
165
            fprintf(stderr, _("cannot strdup command '%s': %s\n"),
166
                    cmdline[i], strerror(errno));
167
            exit(1);
168
        }
169
        v = breakline(input, &c);
170
        if (c) {
171
            ct = find_command(v[0]);
172
            if (ct) {
173
                if (ct->flags & CMD_FLAG_GLOBAL) {
174
                    done = command(ct, c, v);
175
                } else {
176
                    j = 0;
177
                    while (!done && (j = args_command(j))) {
178
                        done = command(ct, c, v);
179
                    }
180
                }
181
            } else {
182
                fprintf(stderr, _("command \"%s\" not found\n"), v[0]);
183
            }
184
        }
185
        doneline(input, v);
186
    }
187
    if (cmdline) {
188
        g_free(cmdline);
189
        return;
190
    }
191

    
192
    while (!done) {
193
        if (!prompted) {
194
            printf("%s", get_prompt());
195
            fflush(stdout);
196
            qemu_aio_set_fd_handler(STDIN_FILENO, prep_fetchline, NULL, NULL,
197
                                    NULL, &fetchable);
198
            prompted = 1;
199
        }
200

    
201
        qemu_aio_wait();
202

    
203
        if (!fetchable) {
204
            continue;
205
        }
206
        input = fetchline();
207
        if (input == NULL) {
208
            break;
209
        }
210
        v = breakline(input, &c);
211
        if (c) {
212
            ct = find_command(v[0]);
213
            if (ct) {
214
                done = command(ct, c, v);
215
            } else {
216
                fprintf(stderr, _("command \"%s\" not found\n"), v[0]);
217
            }
218
        }
219
        doneline(input, v);
220

    
221
        prompted = 0;
222
        fetchable = 0;
223
    }
224
    qemu_aio_set_fd_handler(STDIN_FILENO, NULL, NULL, NULL, NULL, NULL);
225
}
226

    
227
/* from libxcmd/input.c */
228

    
229
#if defined(ENABLE_READLINE)
230
# include <readline/history.h>
231
# include <readline/readline.h>
232
#elif defined(ENABLE_EDITLINE)
233
# include <histedit.h>
234
#endif
235

    
236
static char *
237
get_prompt(void)
238
{
239
        static char        prompt[FILENAME_MAX + 2 /*"> "*/ + 1 /*"\0"*/ ];
240

    
241
        if (!prompt[0])
242
                snprintf(prompt, sizeof(prompt), "%s> ", progname);
243
        return prompt;
244
}
245

    
246
#if defined(ENABLE_READLINE)
247
char *
248
fetchline(void)
249
{
250
        char        *line;
251

    
252
        line = readline(get_prompt());
253
        if (line && *line)
254
                add_history(line);
255
        return line;
256
}
257
#elif defined(ENABLE_EDITLINE)
258
static char *el_get_prompt(EditLine *e) { return get_prompt(); }
259
char *
260
fetchline(void)
261
{
262
        static EditLine        *el;
263
        static History        *hist;
264
        HistEvent        hevent;
265
        char                *line;
266
        int                count;
267

    
268
        if (!el) {
269
                hist = history_init();
270
                history(hist, &hevent, H_SETSIZE, 100);
271
                el = el_init(progname, stdin, stdout, stderr);
272
                el_source(el, NULL);
273
                el_set(el, EL_SIGNAL, 1);
274
                el_set(el, EL_PROMPT, el_get_prompt);
275
                el_set(el, EL_HIST, history, (const char *)hist);
276
        }
277
        line = strdup(el_gets(el, &count));
278
        if (line) {
279
                if (count > 0)
280
                        line[count-1] = '\0';
281
                if (*line)
282
                        history(hist, &hevent, H_ENTER, line);
283
        }
284
        return line;
285
}
286
#else
287
# define MAXREADLINESZ        1024
288
char *
289
fetchline(void)
290
{
291
        char        *p, *line = malloc(MAXREADLINESZ);
292

    
293
        if (!line)
294
                return NULL;
295
        if (!fgets(line, MAXREADLINESZ, stdin)) {
296
                free(line);
297
                return NULL;
298
        }
299
        p = line + strlen(line);
300
        if (p != line && p[-1] == '\n')
301
                p[-1] = '\0';
302
        return line;
303
}
304
#endif
305

    
306
static char *qemu_strsep(char **input, const char *delim)
307
{
308
    char *result = *input;
309
    if (result != NULL) {
310
        char *p;
311

    
312
        for (p = result; *p != '\0'; p++) {
313
            if (strchr(delim, *p)) {
314
                break;
315
            }
316
        }
317
        if (*p == '\0') {
318
            *input = NULL;
319
        } else {
320
            *p = '\0';
321
            *input = p + 1;
322
        }
323
    }
324
    return result;
325
}
326

    
327
char **breakline(char *input, int *count)
328
{
329
    int c = 0;
330
    char *p;
331
    char **rval = calloc(sizeof(char *), 1);
332
    char **tmp;
333

    
334
    while (rval && (p = qemu_strsep(&input, " ")) != NULL) {
335
        if (!*p) {
336
            continue;
337
        }
338
        c++;
339
        tmp = realloc(rval, sizeof(*rval) * (c + 1));
340
        if (!tmp) {
341
            free(rval);
342
            rval = NULL;
343
            c = 0;
344
            break;
345
        } else {
346
            rval = tmp;
347
        }
348
        rval[c - 1] = p;
349
        rval[c] = NULL;
350
    }
351
    *count = c;
352
    return rval;
353
}
354

    
355
void
356
doneline(
357
        char        *input,
358
        char        **vec)
359
{
360
        free(input);
361
        free(vec);
362
}
363

    
364
#define EXABYTES(x)        ((long long)(x) << 60)
365
#define PETABYTES(x)        ((long long)(x) << 50)
366
#define TERABYTES(x)        ((long long)(x) << 40)
367
#define GIGABYTES(x)        ((long long)(x) << 30)
368
#define MEGABYTES(x)        ((long long)(x) << 20)
369
#define KILOBYTES(x)        ((long long)(x) << 10)
370

    
371
long long
372
cvtnum(
373
        char                *s)
374
{
375
        long long        i;
376
        char                *sp;
377
        int                c;
378

    
379
        i = strtoll(s, &sp, 0);
380
        if (i == 0 && sp == s)
381
                return -1LL;
382
        if (*sp == '\0')
383
                return i;
384

    
385
        if (sp[1] != '\0')
386
                return -1LL;
387

    
388
        c = qemu_tolower(*sp);
389
        switch (c) {
390
        default:
391
                return i;
392
        case 'k':
393
                return KILOBYTES(i);
394
        case 'm':
395
                return MEGABYTES(i);
396
        case 'g':
397
                return GIGABYTES(i);
398
        case 't':
399
                return TERABYTES(i);
400
        case 'p':
401
                return PETABYTES(i);
402
        case 'e':
403
                return  EXABYTES(i);
404
        }
405
        return -1LL;
406
}
407

    
408
#define TO_EXABYTES(x)        ((x) / EXABYTES(1))
409
#define TO_PETABYTES(x)        ((x) / PETABYTES(1))
410
#define TO_TERABYTES(x)        ((x) / TERABYTES(1))
411
#define TO_GIGABYTES(x)        ((x) / GIGABYTES(1))
412
#define TO_MEGABYTES(x)        ((x) / MEGABYTES(1))
413
#define TO_KILOBYTES(x)        ((x) / KILOBYTES(1))
414

    
415
void
416
cvtstr(
417
        double                value,
418
        char                *str,
419
        size_t                size)
420
{
421
        const char        *fmt;
422
        int                precise;
423

    
424
        precise = ((double)value * 1000 == (double)(int)value * 1000);
425

    
426
        if (value >= EXABYTES(1)) {
427
                fmt = precise ? "%.f EiB" : "%.3f EiB";
428
                snprintf(str, size, fmt, TO_EXABYTES(value));
429
        } else if (value >= PETABYTES(1)) {
430
                fmt = precise ? "%.f PiB" : "%.3f PiB";
431
                snprintf(str, size, fmt, TO_PETABYTES(value));
432
        } else if (value >= TERABYTES(1)) {
433
                fmt = precise ? "%.f TiB" : "%.3f TiB";
434
                snprintf(str, size, fmt, TO_TERABYTES(value));
435
        } else if (value >= GIGABYTES(1)) {
436
                fmt = precise ? "%.f GiB" : "%.3f GiB";
437
                snprintf(str, size, fmt, TO_GIGABYTES(value));
438
        } else if (value >= MEGABYTES(1)) {
439
                fmt = precise ? "%.f MiB" : "%.3f MiB";
440
                snprintf(str, size, fmt, TO_MEGABYTES(value));
441
        } else if (value >= KILOBYTES(1)) {
442
                fmt = precise ? "%.f KiB" : "%.3f KiB";
443
                snprintf(str, size, fmt, TO_KILOBYTES(value));
444
        } else {
445
                snprintf(str, size, "%f bytes", value);
446
        }
447
}
448

    
449
struct timeval
450
tsub(struct timeval t1, struct timeval t2)
451
{
452
        t1.tv_usec -= t2.tv_usec;
453
        if (t1.tv_usec < 0) {
454
                t1.tv_usec += 1000000;
455
                t1.tv_sec--;
456
        }
457
        t1.tv_sec -= t2.tv_sec;
458
        return t1;
459
}
460

    
461
double
462
tdiv(double value, struct timeval tv)
463
{
464
        return value / ((double)tv.tv_sec + ((double)tv.tv_usec / 1000000.0));
465
}
466

    
467
#define HOURS(sec)        ((sec) / (60 * 60))
468
#define MINUTES(sec)        (((sec) % (60 * 60)) / 60)
469
#define SECONDS(sec)        ((sec) % 60)
470

    
471
void
472
timestr(
473
        struct timeval        *tv,
474
        char                *ts,
475
        size_t                size,
476
        int                format)
477
{
478
        double                usec = (double)tv->tv_usec / 1000000.0;
479

    
480
        if (format & TERSE_FIXED_TIME) {
481
                if (!HOURS(tv->tv_sec)) {
482
                        snprintf(ts, size, "%u:%02u.%02u",
483
                                (unsigned int) MINUTES(tv->tv_sec),
484
                                (unsigned int) SECONDS(tv->tv_sec),
485
                                (unsigned int) (usec * 100));
486
                        return;
487
                }
488
                format |= VERBOSE_FIXED_TIME;        /* fallback if hours needed */
489
        }
490

    
491
        if ((format & VERBOSE_FIXED_TIME) || tv->tv_sec) {
492
                snprintf(ts, size, "%u:%02u:%02u.%02u",
493
                        (unsigned int) HOURS(tv->tv_sec),
494
                        (unsigned int) MINUTES(tv->tv_sec),
495
                        (unsigned int) SECONDS(tv->tv_sec),
496
                        (unsigned int) (usec * 100));
497
        } else {
498
                snprintf(ts, size, "0.%04u sec", (unsigned int) (usec * 10000));
499
        }
500
}
501

    
502

    
503
/* from libxcmd/quit.c */
504

    
505
static cmdinfo_t quit_cmd;
506

    
507
/* ARGSUSED */
508
static int
509
quit_f(
510
        int        argc,
511
        char        **argv)
512
{
513
        return 1;
514
}
515

    
516
void
517
quit_init(void)
518
{
519
        quit_cmd.name = _("quit");
520
        quit_cmd.altname = _("q");
521
        quit_cmd.cfunc = quit_f;
522
        quit_cmd.argmin = -1;
523
        quit_cmd.argmax = -1;
524
        quit_cmd.flags = CMD_FLAG_GLOBAL;
525
        quit_cmd.oneline = _("exit the program");
526

    
527
        add_command(&quit_cmd);
528
}
529

    
530
/* from libxcmd/help.c */
531

    
532
static cmdinfo_t help_cmd;
533
static void help_onecmd(const char *cmd, const cmdinfo_t *ct);
534
static void help_oneline(const char *cmd, const cmdinfo_t *ct);
535

    
536
static void
537
help_all(void)
538
{
539
        const cmdinfo_t        *ct;
540

    
541
        for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++)
542
                help_oneline(ct->name, ct);
543
        printf(_("\nUse 'help commandname' for extended help.\n"));
544
}
545

    
546
static int
547
help_f(
548
        int                argc,
549
        char                **argv)
550
{
551
        const cmdinfo_t        *ct;
552

    
553
        if (argc == 1) {
554
                help_all();
555
                return 0;
556
        }
557
        ct = find_command(argv[1]);
558
        if (ct == NULL) {
559
                printf(_("command %s not found\n"), argv[1]);
560
                return 0;
561
        }
562
        help_onecmd(argv[1], ct);
563
        return 0;
564
}
565

    
566
static void
567
help_onecmd(
568
        const char        *cmd,
569
        const cmdinfo_t        *ct)
570
{
571
        help_oneline(cmd, ct);
572
        if (ct->help)
573
                ct->help();
574
}
575

    
576
static void
577
help_oneline(
578
        const char        *cmd,
579
        const cmdinfo_t        *ct)
580
{
581
        if (cmd)
582
                printf("%s ", cmd);
583
        else {
584
                printf("%s ", ct->name);
585
                if (ct->altname)
586
                        printf("(or %s) ", ct->altname);
587
        }
588
        if (ct->args)
589
                printf("%s ", ct->args);
590
        printf("-- %s\n", ct->oneline);
591
}
592

    
593
void
594
help_init(void)
595
{
596
        help_cmd.name = _("help");
597
        help_cmd.altname = _("?");
598
        help_cmd.cfunc = help_f;
599
        help_cmd.argmin = 0;
600
        help_cmd.argmax = 1;
601
        help_cmd.flags = CMD_FLAG_GLOBAL;
602
        help_cmd.args = _("[command]");
603
        help_cmd.oneline = _("help for one or all commands");
604

    
605
        add_command(&help_cmd);
606
}