]>
Commit | Line | Data |
---|---|---|
e3aff4f6 AL |
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 | |
8167ee88 | 15 | * along with this program; if not, see <http://www.gnu.org/licenses/>. |
e3aff4f6 AL |
16 | */ |
17 | ||
18 | #include <stdio.h> | |
19 | #include <stdlib.h> | |
20 | #include <string.h> | |
21 | #include <ctype.h> | |
22 | #include <errno.h> | |
c32d766a | 23 | #include <sys/time.h> |
22a2bdcb | 24 | #include <getopt.h> |
e3aff4f6 AL |
25 | |
26 | #include "cmd.h" | |
7d7d975c | 27 | #include "qemu-aio.h" |
e3aff4f6 AL |
28 | |
29 | #define _(x) x /* not gettext support yet */ | |
30 | ||
e3aff4f6 AL |
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 | ||
81beeec4 | 48 | void add_command(const cmdinfo_t *ci) |
e3aff4f6 | 49 | { |
ba7806ad | 50 | cmdtab = g_realloc((void *)cmdtab, ++ncmds * sizeof(*cmdtab)); |
81beeec4 PB |
51 | cmdtab[ncmds - 1] = *ci; |
52 | qsort(cmdtab, ncmds, sizeof(*cmdtab), compare); | |
e3aff4f6 AL |
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 | ||
81beeec4 | 123 | void add_user_command(char *optarg) |
e3aff4f6 | 124 | { |
ba7806ad | 125 | cmdline = g_realloc(cmdline, ++ncmdline * sizeof(char *)); |
81beeec4 | 126 | cmdline[ncmdline-1] = optarg; |
e3aff4f6 AL |
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 | ||
7d7d975c MK |
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 | ||
81beeec4 | 155 | void command_loop(void) |
e3aff4f6 | 156 | { |
81beeec4 PB |
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 | } | |
e3aff4f6 | 184 | } |
81beeec4 PB |
185 | doneline(input, v); |
186 | } | |
187 | if (cmdline) { | |
ba7806ad | 188 | g_free(cmdline); |
81beeec4 PB |
189 | return; |
190 | } | |
7d7d975c | 191 | |
81beeec4 | 192 | while (!done) { |
7d7d975c MK |
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 | } | |
81beeec4 PB |
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); | |
7d7d975c MK |
220 | |
221 | prompted = 0; | |
222 | fetchable = 0; | |
81beeec4 | 223 | } |
7d7d975c | 224 | qemu_aio_set_fd_handler(STDIN_FILENO, NULL, NULL, NULL, NULL, NULL); |
e3aff4f6 AL |
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 | ||
e3aff4f6 AL |
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; | |
e3aff4f6 AL |
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 | ||
c32d766a SW |
306 | static char *qemu_strsep(char **input, const char *delim) |
307 | { | |
308 | char *result = *input; | |
309 | if (result != NULL) { | |
88bf7950 BS |
310 | char *p; |
311 | ||
50da01ed SW |
312 | for (p = result; *p != '\0'; p++) { |
313 | if (strchr(delim, *p)) { | |
c32d766a SW |
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 | ||
81beeec4 | 327 | char **breakline(char *input, int *count) |
e3aff4f6 | 328 | { |
81beeec4 PB |
329 | int c = 0; |
330 | char *p; | |
331 | char **rval = calloc(sizeof(char *), 1); | |
47e8dd8f | 332 | char **tmp; |
81beeec4 PB |
333 | |
334 | while (rval && (p = qemu_strsep(&input, " ")) != NULL) { | |
335 | if (!*p) { | |
336 | continue; | |
337 | } | |
338 | c++; | |
47e8dd8f PB |
339 | tmp = realloc(rval, sizeof(*rval) * (c + 1)); |
340 | if (!tmp) { | |
341 | free(rval); | |
342 | rval = NULL; | |
81beeec4 PB |
343 | c = 0; |
344 | break; | |
47e8dd8f PB |
345 | } else { |
346 | rval = tmp; | |
81beeec4 PB |
347 | } |
348 | rval[c - 1] = p; | |
349 | rval[c] = NULL; | |
350 | } | |
351 | *count = c; | |
352 | return rval; | |
e3aff4f6 AL |
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 | ||
7b0a03a1 | 388 | c = qemu_tolower(*sp); |
e3aff4f6 AL |
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), | |
bcd2491a | 485 | (unsigned int) (usec * 100)); |
e3aff4f6 AL |
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), | |
bcd2491a | 496 | (unsigned int) (usec * 100)); |
e3aff4f6 | 497 | } else { |
bcd2491a | 498 | snprintf(ts, size, "0.%04u sec", (unsigned int) (usec * 10000)); |
e3aff4f6 AL |
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 | } |