Merge remote-tracking branch 'remotes/armbru/tags/pull-monitor-2015-02-18' into staging
[qemu/qmp-unstable.git] / qemu-io-cmds.c
blob1afcfc01a57ca2a877b38b4ee87f144d4682d7d7
1 /*
2 * Command line utility to exercise the QEMU I/O path.
4 * Copyright (C) 2009 Red Hat, Inc.
5 * Copyright (c) 2003-2005 Silicon Graphics, Inc.
7 * This work is licensed under the terms of the GNU GPL, version 2 or later.
8 * See the COPYING file in the top-level directory.
9 */
11 #include "qemu-io.h"
12 #include "sysemu/block-backend.h"
13 #include "block/block.h"
14 #include "block/block_int.h" /* for info_f() */
15 #include "block/qapi.h"
16 #include "qemu/main-loop.h"
17 #include "qemu/timer.h"
18 #include "sysemu/block-backend.h"
20 #define CMD_NOFILE_OK 0x01
22 bool qemuio_misalign;
24 static cmdinfo_t *cmdtab;
25 static int ncmds;
27 static int compare_cmdname(const void *a, const void *b)
29 return strcmp(((const cmdinfo_t *)a)->name,
30 ((const cmdinfo_t *)b)->name);
33 void qemuio_add_command(const cmdinfo_t *ci)
35 cmdtab = g_renew(cmdinfo_t, cmdtab, ++ncmds);
36 cmdtab[ncmds - 1] = *ci;
37 qsort(cmdtab, ncmds, sizeof(*cmdtab), compare_cmdname);
40 int qemuio_command_usage(const cmdinfo_t *ci)
42 printf("%s %s -- %s\n", ci->name, ci->args, ci->oneline);
43 return 0;
46 static int init_check_command(BlockBackend *blk, const cmdinfo_t *ct)
48 if (ct->flags & CMD_FLAG_GLOBAL) {
49 return 1;
51 if (!(ct->flags & CMD_NOFILE_OK) && !blk) {
52 fprintf(stderr, "no file open, try 'help open'\n");
53 return 0;
55 return 1;
58 static int command(BlockBackend *blk, const cmdinfo_t *ct, int argc,
59 char **argv)
61 char *cmd = argv[0];
63 if (!init_check_command(blk, ct)) {
64 return 0;
67 if (argc - 1 < ct->argmin || (ct->argmax != -1 && argc - 1 > ct->argmax)) {
68 if (ct->argmax == -1) {
69 fprintf(stderr,
70 "bad argument count %d to %s, expected at least %d arguments\n",
71 argc-1, cmd, ct->argmin);
72 } else if (ct->argmin == ct->argmax) {
73 fprintf(stderr,
74 "bad argument count %d to %s, expected %d arguments\n",
75 argc-1, cmd, ct->argmin);
76 } else {
77 fprintf(stderr,
78 "bad argument count %d to %s, expected between %d and %d arguments\n",
79 argc-1, cmd, ct->argmin, ct->argmax);
81 return 0;
83 optind = 0;
84 return ct->cfunc(blk, argc, argv);
87 static const cmdinfo_t *find_command(const char *cmd)
89 cmdinfo_t *ct;
91 for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) {
92 if (strcmp(ct->name, cmd) == 0 ||
93 (ct->altname && strcmp(ct->altname, cmd) == 0))
95 return (const cmdinfo_t *)ct;
98 return NULL;
101 /* Invoke fn() for commands with a matching prefix */
102 void qemuio_complete_command(const char *input,
103 void (*fn)(const char *cmd, void *opaque),
104 void *opaque)
106 cmdinfo_t *ct;
107 size_t input_len = strlen(input);
109 for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) {
110 if (strncmp(input, ct->name, input_len) == 0) {
111 fn(ct->name, opaque);
116 static char **breakline(char *input, int *count)
118 int c = 0;
119 char *p;
120 char **rval = g_new0(char *, 1);
122 while (rval && (p = qemu_strsep(&input, " ")) != NULL) {
123 if (!*p) {
124 continue;
126 c++;
127 rval = g_renew(char *, rval, (c + 1));
128 rval[c - 1] = p;
129 rval[c] = NULL;
131 *count = c;
132 return rval;
135 static int64_t cvtnum(const char *s)
137 char *end;
138 return strtosz_suffix(s, &end, STRTOSZ_DEFSUFFIX_B);
141 #define EXABYTES(x) ((long long)(x) << 60)
142 #define PETABYTES(x) ((long long)(x) << 50)
143 #define TERABYTES(x) ((long long)(x) << 40)
144 #define GIGABYTES(x) ((long long)(x) << 30)
145 #define MEGABYTES(x) ((long long)(x) << 20)
146 #define KILOBYTES(x) ((long long)(x) << 10)
148 #define TO_EXABYTES(x) ((x) / EXABYTES(1))
149 #define TO_PETABYTES(x) ((x) / PETABYTES(1))
150 #define TO_TERABYTES(x) ((x) / TERABYTES(1))
151 #define TO_GIGABYTES(x) ((x) / GIGABYTES(1))
152 #define TO_MEGABYTES(x) ((x) / MEGABYTES(1))
153 #define TO_KILOBYTES(x) ((x) / KILOBYTES(1))
155 static void cvtstr(double value, char *str, size_t size)
157 char *trim;
158 const char *suffix;
160 if (value >= EXABYTES(1)) {
161 suffix = " EiB";
162 snprintf(str, size - 4, "%.3f", TO_EXABYTES(value));
163 } else if (value >= PETABYTES(1)) {
164 suffix = " PiB";
165 snprintf(str, size - 4, "%.3f", TO_PETABYTES(value));
166 } else if (value >= TERABYTES(1)) {
167 suffix = " TiB";
168 snprintf(str, size - 4, "%.3f", TO_TERABYTES(value));
169 } else if (value >= GIGABYTES(1)) {
170 suffix = " GiB";
171 snprintf(str, size - 4, "%.3f", TO_GIGABYTES(value));
172 } else if (value >= MEGABYTES(1)) {
173 suffix = " MiB";
174 snprintf(str, size - 4, "%.3f", TO_MEGABYTES(value));
175 } else if (value >= KILOBYTES(1)) {
176 suffix = " KiB";
177 snprintf(str, size - 4, "%.3f", TO_KILOBYTES(value));
178 } else {
179 suffix = " bytes";
180 snprintf(str, size - 6, "%f", value);
183 trim = strstr(str, ".000");
184 if (trim) {
185 strcpy(trim, suffix);
186 } else {
187 strcat(str, suffix);
193 static struct timeval tsub(struct timeval t1, struct timeval t2)
195 t1.tv_usec -= t2.tv_usec;
196 if (t1.tv_usec < 0) {
197 t1.tv_usec += 1000000;
198 t1.tv_sec--;
200 t1.tv_sec -= t2.tv_sec;
201 return t1;
204 static double tdiv(double value, struct timeval tv)
206 return value / ((double)tv.tv_sec + ((double)tv.tv_usec / 1000000.0));
209 #define HOURS(sec) ((sec) / (60 * 60))
210 #define MINUTES(sec) (((sec) % (60 * 60)) / 60)
211 #define SECONDS(sec) ((sec) % 60)
213 enum {
214 DEFAULT_TIME = 0x0,
215 TERSE_FIXED_TIME = 0x1,
216 VERBOSE_FIXED_TIME = 0x2,
219 static void timestr(struct timeval *tv, char *ts, size_t size, int format)
221 double usec = (double)tv->tv_usec / 1000000.0;
223 if (format & TERSE_FIXED_TIME) {
224 if (!HOURS(tv->tv_sec)) {
225 snprintf(ts, size, "%u:%02u.%02u",
226 (unsigned int) MINUTES(tv->tv_sec),
227 (unsigned int) SECONDS(tv->tv_sec),
228 (unsigned int) (usec * 100));
229 return;
231 format |= VERBOSE_FIXED_TIME; /* fallback if hours needed */
234 if ((format & VERBOSE_FIXED_TIME) || tv->tv_sec) {
235 snprintf(ts, size, "%u:%02u:%02u.%02u",
236 (unsigned int) HOURS(tv->tv_sec),
237 (unsigned int) MINUTES(tv->tv_sec),
238 (unsigned int) SECONDS(tv->tv_sec),
239 (unsigned int) (usec * 100));
240 } else {
241 snprintf(ts, size, "0.%04u sec", (unsigned int) (usec * 10000));
246 * Parse the pattern argument to various sub-commands.
248 * Because the pattern is used as an argument to memset it must evaluate
249 * to an unsigned integer that fits into a single byte.
251 static int parse_pattern(const char *arg)
253 char *endptr = NULL;
254 long pattern;
256 pattern = strtol(arg, &endptr, 0);
257 if (pattern < 0 || pattern > UCHAR_MAX || *endptr != '\0') {
258 printf("%s is not a valid pattern byte\n", arg);
259 return -1;
262 return pattern;
266 * Memory allocation helpers.
268 * Make sure memory is aligned by default, or purposefully misaligned if
269 * that is specified on the command line.
272 #define MISALIGN_OFFSET 16
273 static void *qemu_io_alloc(BlockBackend *blk, size_t len, int pattern)
275 void *buf;
277 if (qemuio_misalign) {
278 len += MISALIGN_OFFSET;
280 buf = blk_blockalign(blk, len);
281 memset(buf, pattern, len);
282 if (qemuio_misalign) {
283 buf += MISALIGN_OFFSET;
285 return buf;
288 static void qemu_io_free(void *p)
290 if (qemuio_misalign) {
291 p -= MISALIGN_OFFSET;
293 qemu_vfree(p);
296 static void dump_buffer(const void *buffer, int64_t offset, int len)
298 int i, j;
299 const uint8_t *p;
301 for (i = 0, p = buffer; i < len; i += 16) {
302 const uint8_t *s = p;
304 printf("%08" PRIx64 ": ", offset + i);
305 for (j = 0; j < 16 && i + j < len; j++, p++) {
306 printf("%02x ", *p);
308 printf(" ");
309 for (j = 0; j < 16 && i + j < len; j++, s++) {
310 if (isalnum(*s)) {
311 printf("%c", *s);
312 } else {
313 printf(".");
316 printf("\n");
320 static void print_report(const char *op, struct timeval *t, int64_t offset,
321 int count, int total, int cnt, int Cflag)
323 char s1[64], s2[64], ts[64];
325 timestr(t, ts, sizeof(ts), Cflag ? VERBOSE_FIXED_TIME : 0);
326 if (!Cflag) {
327 cvtstr((double)total, s1, sizeof(s1));
328 cvtstr(tdiv((double)total, *t), s2, sizeof(s2));
329 printf("%s %d/%d bytes at offset %" PRId64 "\n",
330 op, total, count, offset);
331 printf("%s, %d ops; %s (%s/sec and %.4f ops/sec)\n",
332 s1, cnt, ts, s2, tdiv((double)cnt, *t));
333 } else {/* bytes,ops,time,bytes/sec,ops/sec */
334 printf("%d,%d,%s,%.3f,%.3f\n",
335 total, cnt, ts,
336 tdiv((double)total, *t),
337 tdiv((double)cnt, *t));
342 * Parse multiple length statements for vectored I/O, and construct an I/O
343 * vector matching it.
345 static void *
346 create_iovec(BlockBackend *blk, QEMUIOVector *qiov, char **argv, int nr_iov,
347 int pattern)
349 size_t *sizes = g_new0(size_t, nr_iov);
350 size_t count = 0;
351 void *buf = NULL;
352 void *p;
353 int i;
355 for (i = 0; i < nr_iov; i++) {
356 char *arg = argv[i];
357 int64_t len;
359 len = cvtnum(arg);
360 if (len < 0) {
361 printf("non-numeric length argument -- %s\n", arg);
362 goto fail;
365 /* should be SIZE_T_MAX, but that doesn't exist */
366 if (len > INT_MAX) {
367 printf("too large length argument -- %s\n", arg);
368 goto fail;
371 if (len & 0x1ff) {
372 printf("length argument %" PRId64
373 " is not sector aligned\n", len);
374 goto fail;
377 sizes[i] = len;
378 count += len;
381 qemu_iovec_init(qiov, nr_iov);
383 buf = p = qemu_io_alloc(blk, count, pattern);
385 for (i = 0; i < nr_iov; i++) {
386 qemu_iovec_add(qiov, p, sizes[i]);
387 p += sizes[i];
390 fail:
391 g_free(sizes);
392 return buf;
395 static int do_read(BlockBackend *blk, char *buf, int64_t offset, int count,
396 int *total)
398 int ret;
400 ret = blk_read(blk, offset >> 9, (uint8_t *)buf, count >> 9);
401 if (ret < 0) {
402 return ret;
404 *total = count;
405 return 1;
408 static int do_write(BlockBackend *blk, char *buf, int64_t offset, int count,
409 int *total)
411 int ret;
413 ret = blk_write(blk, offset >> 9, (uint8_t *)buf, count >> 9);
414 if (ret < 0) {
415 return ret;
417 *total = count;
418 return 1;
421 static int do_pread(BlockBackend *blk, char *buf, int64_t offset, int count,
422 int *total)
424 *total = blk_pread(blk, offset, (uint8_t *)buf, count);
425 if (*total < 0) {
426 return *total;
428 return 1;
431 static int do_pwrite(BlockBackend *blk, char *buf, int64_t offset, int count,
432 int *total)
434 *total = blk_pwrite(blk, offset, (uint8_t *)buf, count);
435 if (*total < 0) {
436 return *total;
438 return 1;
441 typedef struct {
442 BlockBackend *blk;
443 int64_t offset;
444 int count;
445 int *total;
446 int ret;
447 bool done;
448 } CoWriteZeroes;
450 static void coroutine_fn co_write_zeroes_entry(void *opaque)
452 CoWriteZeroes *data = opaque;
454 data->ret = blk_co_write_zeroes(data->blk, data->offset / BDRV_SECTOR_SIZE,
455 data->count / BDRV_SECTOR_SIZE, 0);
456 data->done = true;
457 if (data->ret < 0) {
458 *data->total = data->ret;
459 return;
462 *data->total = data->count;
465 static int do_co_write_zeroes(BlockBackend *blk, int64_t offset, int count,
466 int *total)
468 Coroutine *co;
469 CoWriteZeroes data = {
470 .blk = blk,
471 .offset = offset,
472 .count = count,
473 .total = total,
474 .done = false,
477 co = qemu_coroutine_create(co_write_zeroes_entry);
478 qemu_coroutine_enter(co, &data);
479 while (!data.done) {
480 aio_poll(blk_get_aio_context(blk), true);
482 if (data.ret < 0) {
483 return data.ret;
484 } else {
485 return 1;
489 static int do_write_compressed(BlockBackend *blk, char *buf, int64_t offset,
490 int count, int *total)
492 int ret;
494 ret = blk_write_compressed(blk, offset >> 9, (uint8_t *)buf, count >> 9);
495 if (ret < 0) {
496 return ret;
498 *total = count;
499 return 1;
502 static int do_load_vmstate(BlockBackend *blk, char *buf, int64_t offset,
503 int count, int *total)
505 *total = blk_load_vmstate(blk, (uint8_t *)buf, offset, count);
506 if (*total < 0) {
507 return *total;
509 return 1;
512 static int do_save_vmstate(BlockBackend *blk, char *buf, int64_t offset,
513 int count, int *total)
515 *total = blk_save_vmstate(blk, (uint8_t *)buf, offset, count);
516 if (*total < 0) {
517 return *total;
519 return 1;
522 #define NOT_DONE 0x7fffffff
523 static void aio_rw_done(void *opaque, int ret)
525 *(int *)opaque = ret;
528 static int do_aio_readv(BlockBackend *blk, QEMUIOVector *qiov,
529 int64_t offset, int *total)
531 int async_ret = NOT_DONE;
533 blk_aio_readv(blk, offset >> 9, qiov, qiov->size >> 9,
534 aio_rw_done, &async_ret);
535 while (async_ret == NOT_DONE) {
536 main_loop_wait(false);
539 *total = qiov->size;
540 return async_ret < 0 ? async_ret : 1;
543 static int do_aio_writev(BlockBackend *blk, QEMUIOVector *qiov,
544 int64_t offset, int *total)
546 int async_ret = NOT_DONE;
548 blk_aio_writev(blk, offset >> 9, qiov, qiov->size >> 9,
549 aio_rw_done, &async_ret);
550 while (async_ret == NOT_DONE) {
551 main_loop_wait(false);
554 *total = qiov->size;
555 return async_ret < 0 ? async_ret : 1;
558 struct multiwrite_async_ret {
559 int num_done;
560 int error;
563 static void multiwrite_cb(void *opaque, int ret)
565 struct multiwrite_async_ret *async_ret = opaque;
567 async_ret->num_done++;
568 if (ret < 0) {
569 async_ret->error = ret;
573 static int do_aio_multiwrite(BlockBackend *blk, BlockRequest* reqs,
574 int num_reqs, int *total)
576 int i, ret;
577 struct multiwrite_async_ret async_ret = {
578 .num_done = 0,
579 .error = 0,
582 *total = 0;
583 for (i = 0; i < num_reqs; i++) {
584 reqs[i].cb = multiwrite_cb;
585 reqs[i].opaque = &async_ret;
586 *total += reqs[i].qiov->size;
589 ret = blk_aio_multiwrite(blk, reqs, num_reqs);
590 if (ret < 0) {
591 return ret;
594 while (async_ret.num_done < num_reqs) {
595 main_loop_wait(false);
598 return async_ret.error < 0 ? async_ret.error : 1;
601 static void read_help(void)
603 printf(
604 "\n"
605 " reads a range of bytes from the given offset\n"
606 "\n"
607 " Example:\n"
608 " 'read -v 512 1k' - dumps 1 kilobyte read from 512 bytes into the file\n"
609 "\n"
610 " Reads a segment of the currently open file, optionally dumping it to the\n"
611 " standard output stream (with -v option) for subsequent inspection.\n"
612 " -b, -- read from the VM state rather than the virtual disk\n"
613 " -C, -- report statistics in a machine parsable format\n"
614 " -l, -- length for pattern verification (only with -P)\n"
615 " -p, -- use blk_pread to read the file\n"
616 " -P, -- use a pattern to verify read data\n"
617 " -q, -- quiet mode, do not show I/O statistics\n"
618 " -s, -- start offset for pattern verification (only with -P)\n"
619 " -v, -- dump buffer to standard output\n"
620 "\n");
623 static int read_f(BlockBackend *blk, int argc, char **argv);
625 static const cmdinfo_t read_cmd = {
626 .name = "read",
627 .altname = "r",
628 .cfunc = read_f,
629 .argmin = 2,
630 .argmax = -1,
631 .args = "[-abCpqv] [-P pattern [-s off] [-l len]] off len",
632 .oneline = "reads a number of bytes at a specified offset",
633 .help = read_help,
636 static int read_f(BlockBackend *blk, int argc, char **argv)
638 struct timeval t1, t2;
639 int Cflag = 0, pflag = 0, qflag = 0, vflag = 0;
640 int Pflag = 0, sflag = 0, lflag = 0, bflag = 0;
641 int c, cnt;
642 char *buf;
643 int64_t offset;
644 int count;
645 /* Some compilers get confused and warn if this is not initialized. */
646 int total = 0;
647 int pattern = 0, pattern_offset = 0, pattern_count = 0;
649 while ((c = getopt(argc, argv, "bCl:pP:qs:v")) != EOF) {
650 switch (c) {
651 case 'b':
652 bflag = 1;
653 break;
654 case 'C':
655 Cflag = 1;
656 break;
657 case 'l':
658 lflag = 1;
659 pattern_count = cvtnum(optarg);
660 if (pattern_count < 0) {
661 printf("non-numeric length argument -- %s\n", optarg);
662 return 0;
664 break;
665 case 'p':
666 pflag = 1;
667 break;
668 case 'P':
669 Pflag = 1;
670 pattern = parse_pattern(optarg);
671 if (pattern < 0) {
672 return 0;
674 break;
675 case 'q':
676 qflag = 1;
677 break;
678 case 's':
679 sflag = 1;
680 pattern_offset = cvtnum(optarg);
681 if (pattern_offset < 0) {
682 printf("non-numeric length argument -- %s\n", optarg);
683 return 0;
685 break;
686 case 'v':
687 vflag = 1;
688 break;
689 default:
690 return qemuio_command_usage(&read_cmd);
694 if (optind != argc - 2) {
695 return qemuio_command_usage(&read_cmd);
698 if (bflag && pflag) {
699 printf("-b and -p cannot be specified at the same time\n");
700 return 0;
703 offset = cvtnum(argv[optind]);
704 if (offset < 0) {
705 printf("non-numeric length argument -- %s\n", argv[optind]);
706 return 0;
709 optind++;
710 count = cvtnum(argv[optind]);
711 if (count < 0) {
712 printf("non-numeric length argument -- %s\n", argv[optind]);
713 return 0;
716 if (!Pflag && (lflag || sflag)) {
717 return qemuio_command_usage(&read_cmd);
720 if (!lflag) {
721 pattern_count = count - pattern_offset;
724 if ((pattern_count < 0) || (pattern_count + pattern_offset > count)) {
725 printf("pattern verification range exceeds end of read data\n");
726 return 0;
729 if (!pflag) {
730 if (offset & 0x1ff) {
731 printf("offset %" PRId64 " is not sector aligned\n",
732 offset);
733 return 0;
735 if (count & 0x1ff) {
736 printf("count %d is not sector aligned\n",
737 count);
738 return 0;
742 buf = qemu_io_alloc(blk, count, 0xab);
744 gettimeofday(&t1, NULL);
745 if (pflag) {
746 cnt = do_pread(blk, buf, offset, count, &total);
747 } else if (bflag) {
748 cnt = do_load_vmstate(blk, buf, offset, count, &total);
749 } else {
750 cnt = do_read(blk, buf, offset, count, &total);
752 gettimeofday(&t2, NULL);
754 if (cnt < 0) {
755 printf("read failed: %s\n", strerror(-cnt));
756 goto out;
759 if (Pflag) {
760 void *cmp_buf = g_malloc(pattern_count);
761 memset(cmp_buf, pattern, pattern_count);
762 if (memcmp(buf + pattern_offset, cmp_buf, pattern_count)) {
763 printf("Pattern verification failed at offset %"
764 PRId64 ", %d bytes\n",
765 offset + pattern_offset, pattern_count);
767 g_free(cmp_buf);
770 if (qflag) {
771 goto out;
774 if (vflag) {
775 dump_buffer(buf, offset, count);
778 /* Finally, report back -- -C gives a parsable format */
779 t2 = tsub(t2, t1);
780 print_report("read", &t2, offset, count, total, cnt, Cflag);
782 out:
783 qemu_io_free(buf);
785 return 0;
788 static void readv_help(void)
790 printf(
791 "\n"
792 " reads a range of bytes from the given offset into multiple buffers\n"
793 "\n"
794 " Example:\n"
795 " 'readv -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n"
796 "\n"
797 " Reads a segment of the currently open file, optionally dumping it to the\n"
798 " standard output stream (with -v option) for subsequent inspection.\n"
799 " Uses multiple iovec buffers if more than one byte range is specified.\n"
800 " -C, -- report statistics in a machine parsable format\n"
801 " -P, -- use a pattern to verify read data\n"
802 " -v, -- dump buffer to standard output\n"
803 " -q, -- quiet mode, do not show I/O statistics\n"
804 "\n");
807 static int readv_f(BlockBackend *blk, int argc, char **argv);
809 static const cmdinfo_t readv_cmd = {
810 .name = "readv",
811 .cfunc = readv_f,
812 .argmin = 2,
813 .argmax = -1,
814 .args = "[-Cqv] [-P pattern ] off len [len..]",
815 .oneline = "reads a number of bytes at a specified offset",
816 .help = readv_help,
819 static int readv_f(BlockBackend *blk, int argc, char **argv)
821 struct timeval t1, t2;
822 int Cflag = 0, qflag = 0, vflag = 0;
823 int c, cnt;
824 char *buf;
825 int64_t offset;
826 /* Some compilers get confused and warn if this is not initialized. */
827 int total = 0;
828 int nr_iov;
829 QEMUIOVector qiov;
830 int pattern = 0;
831 int Pflag = 0;
833 while ((c = getopt(argc, argv, "CP:qv")) != EOF) {
834 switch (c) {
835 case 'C':
836 Cflag = 1;
837 break;
838 case 'P':
839 Pflag = 1;
840 pattern = parse_pattern(optarg);
841 if (pattern < 0) {
842 return 0;
844 break;
845 case 'q':
846 qflag = 1;
847 break;
848 case 'v':
849 vflag = 1;
850 break;
851 default:
852 return qemuio_command_usage(&readv_cmd);
856 if (optind > argc - 2) {
857 return qemuio_command_usage(&readv_cmd);
861 offset = cvtnum(argv[optind]);
862 if (offset < 0) {
863 printf("non-numeric length argument -- %s\n", argv[optind]);
864 return 0;
866 optind++;
868 if (offset & 0x1ff) {
869 printf("offset %" PRId64 " is not sector aligned\n",
870 offset);
871 return 0;
874 nr_iov = argc - optind;
875 buf = create_iovec(blk, &qiov, &argv[optind], nr_iov, 0xab);
876 if (buf == NULL) {
877 return 0;
880 gettimeofday(&t1, NULL);
881 cnt = do_aio_readv(blk, &qiov, offset, &total);
882 gettimeofday(&t2, NULL);
884 if (cnt < 0) {
885 printf("readv failed: %s\n", strerror(-cnt));
886 goto out;
889 if (Pflag) {
890 void *cmp_buf = g_malloc(qiov.size);
891 memset(cmp_buf, pattern, qiov.size);
892 if (memcmp(buf, cmp_buf, qiov.size)) {
893 printf("Pattern verification failed at offset %"
894 PRId64 ", %zd bytes\n", offset, qiov.size);
896 g_free(cmp_buf);
899 if (qflag) {
900 goto out;
903 if (vflag) {
904 dump_buffer(buf, offset, qiov.size);
907 /* Finally, report back -- -C gives a parsable format */
908 t2 = tsub(t2, t1);
909 print_report("read", &t2, offset, qiov.size, total, cnt, Cflag);
911 out:
912 qemu_iovec_destroy(&qiov);
913 qemu_io_free(buf);
914 return 0;
917 static void write_help(void)
919 printf(
920 "\n"
921 " writes a range of bytes from the given offset\n"
922 "\n"
923 " Example:\n"
924 " 'write 512 1k' - writes 1 kilobyte at 512 bytes into the open file\n"
925 "\n"
926 " Writes into a segment of the currently open file, using a buffer\n"
927 " filled with a set pattern (0xcdcdcdcd).\n"
928 " -b, -- write to the VM state rather than the virtual disk\n"
929 " -c, -- write compressed data with blk_write_compressed\n"
930 " -p, -- use blk_pwrite to write the file\n"
931 " -P, -- use different pattern to fill file\n"
932 " -C, -- report statistics in a machine parsable format\n"
933 " -q, -- quiet mode, do not show I/O statistics\n"
934 " -z, -- write zeroes using blk_co_write_zeroes\n"
935 "\n");
938 static int write_f(BlockBackend *blk, int argc, char **argv);
940 static const cmdinfo_t write_cmd = {
941 .name = "write",
942 .altname = "w",
943 .cfunc = write_f,
944 .argmin = 2,
945 .argmax = -1,
946 .args = "[-bcCpqz] [-P pattern ] off len",
947 .oneline = "writes a number of bytes at a specified offset",
948 .help = write_help,
951 static int write_f(BlockBackend *blk, int argc, char **argv)
953 struct timeval t1, t2;
954 int Cflag = 0, pflag = 0, qflag = 0, bflag = 0, Pflag = 0, zflag = 0;
955 int cflag = 0;
956 int c, cnt;
957 char *buf = NULL;
958 int64_t offset;
959 int count;
960 /* Some compilers get confused and warn if this is not initialized. */
961 int total = 0;
962 int pattern = 0xcd;
964 while ((c = getopt(argc, argv, "bcCpP:qz")) != EOF) {
965 switch (c) {
966 case 'b':
967 bflag = 1;
968 break;
969 case 'c':
970 cflag = 1;
971 break;
972 case 'C':
973 Cflag = 1;
974 break;
975 case 'p':
976 pflag = 1;
977 break;
978 case 'P':
979 Pflag = 1;
980 pattern = parse_pattern(optarg);
981 if (pattern < 0) {
982 return 0;
984 break;
985 case 'q':
986 qflag = 1;
987 break;
988 case 'z':
989 zflag = 1;
990 break;
991 default:
992 return qemuio_command_usage(&write_cmd);
996 if (optind != argc - 2) {
997 return qemuio_command_usage(&write_cmd);
1000 if (bflag + pflag + zflag > 1) {
1001 printf("-b, -p, or -z cannot be specified at the same time\n");
1002 return 0;
1005 if (zflag && Pflag) {
1006 printf("-z and -P cannot be specified at the same time\n");
1007 return 0;
1010 offset = cvtnum(argv[optind]);
1011 if (offset < 0) {
1012 printf("non-numeric length argument -- %s\n", argv[optind]);
1013 return 0;
1016 optind++;
1017 count = cvtnum(argv[optind]);
1018 if (count < 0) {
1019 printf("non-numeric length argument -- %s\n", argv[optind]);
1020 return 0;
1023 if (!pflag) {
1024 if (offset & 0x1ff) {
1025 printf("offset %" PRId64 " is not sector aligned\n",
1026 offset);
1027 return 0;
1030 if (count & 0x1ff) {
1031 printf("count %d is not sector aligned\n",
1032 count);
1033 return 0;
1037 if (!zflag) {
1038 buf = qemu_io_alloc(blk, count, pattern);
1041 gettimeofday(&t1, NULL);
1042 if (pflag) {
1043 cnt = do_pwrite(blk, buf, offset, count, &total);
1044 } else if (bflag) {
1045 cnt = do_save_vmstate(blk, buf, offset, count, &total);
1046 } else if (zflag) {
1047 cnt = do_co_write_zeroes(blk, offset, count, &total);
1048 } else if (cflag) {
1049 cnt = do_write_compressed(blk, buf, offset, count, &total);
1050 } else {
1051 cnt = do_write(blk, buf, offset, count, &total);
1053 gettimeofday(&t2, NULL);
1055 if (cnt < 0) {
1056 printf("write failed: %s\n", strerror(-cnt));
1057 goto out;
1060 if (qflag) {
1061 goto out;
1064 /* Finally, report back -- -C gives a parsable format */
1065 t2 = tsub(t2, t1);
1066 print_report("wrote", &t2, offset, count, total, cnt, Cflag);
1068 out:
1069 if (!zflag) {
1070 qemu_io_free(buf);
1073 return 0;
1076 static void
1077 writev_help(void)
1079 printf(
1080 "\n"
1081 " writes a range of bytes from the given offset source from multiple buffers\n"
1082 "\n"
1083 " Example:\n"
1084 " 'writev 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n"
1085 "\n"
1086 " Writes into a segment of the currently open file, using a buffer\n"
1087 " filled with a set pattern (0xcdcdcdcd).\n"
1088 " -P, -- use different pattern to fill file\n"
1089 " -C, -- report statistics in a machine parsable format\n"
1090 " -q, -- quiet mode, do not show I/O statistics\n"
1091 "\n");
1094 static int writev_f(BlockBackend *blk, int argc, char **argv);
1096 static const cmdinfo_t writev_cmd = {
1097 .name = "writev",
1098 .cfunc = writev_f,
1099 .argmin = 2,
1100 .argmax = -1,
1101 .args = "[-Cq] [-P pattern ] off len [len..]",
1102 .oneline = "writes a number of bytes at a specified offset",
1103 .help = writev_help,
1106 static int writev_f(BlockBackend *blk, int argc, char **argv)
1108 struct timeval t1, t2;
1109 int Cflag = 0, qflag = 0;
1110 int c, cnt;
1111 char *buf;
1112 int64_t offset;
1113 /* Some compilers get confused and warn if this is not initialized. */
1114 int total = 0;
1115 int nr_iov;
1116 int pattern = 0xcd;
1117 QEMUIOVector qiov;
1119 while ((c = getopt(argc, argv, "CqP:")) != EOF) {
1120 switch (c) {
1121 case 'C':
1122 Cflag = 1;
1123 break;
1124 case 'q':
1125 qflag = 1;
1126 break;
1127 case 'P':
1128 pattern = parse_pattern(optarg);
1129 if (pattern < 0) {
1130 return 0;
1132 break;
1133 default:
1134 return qemuio_command_usage(&writev_cmd);
1138 if (optind > argc - 2) {
1139 return qemuio_command_usage(&writev_cmd);
1142 offset = cvtnum(argv[optind]);
1143 if (offset < 0) {
1144 printf("non-numeric length argument -- %s\n", argv[optind]);
1145 return 0;
1147 optind++;
1149 if (offset & 0x1ff) {
1150 printf("offset %" PRId64 " is not sector aligned\n",
1151 offset);
1152 return 0;
1155 nr_iov = argc - optind;
1156 buf = create_iovec(blk, &qiov, &argv[optind], nr_iov, pattern);
1157 if (buf == NULL) {
1158 return 0;
1161 gettimeofday(&t1, NULL);
1162 cnt = do_aio_writev(blk, &qiov, offset, &total);
1163 gettimeofday(&t2, NULL);
1165 if (cnt < 0) {
1166 printf("writev failed: %s\n", strerror(-cnt));
1167 goto out;
1170 if (qflag) {
1171 goto out;
1174 /* Finally, report back -- -C gives a parsable format */
1175 t2 = tsub(t2, t1);
1176 print_report("wrote", &t2, offset, qiov.size, total, cnt, Cflag);
1177 out:
1178 qemu_iovec_destroy(&qiov);
1179 qemu_io_free(buf);
1180 return 0;
1183 static void multiwrite_help(void)
1185 printf(
1186 "\n"
1187 " writes a range of bytes from the given offset source from multiple buffers,\n"
1188 " in a batch of requests that may be merged by qemu\n"
1189 "\n"
1190 " Example:\n"
1191 " 'multiwrite 512 1k 1k ; 4k 1k'\n"
1192 " writes 2 kB at 512 bytes and 1 kB at 4 kB into the open file\n"
1193 "\n"
1194 " Writes into a segment of the currently open file, using a buffer\n"
1195 " filled with a set pattern (0xcdcdcdcd). The pattern byte is increased\n"
1196 " by one for each request contained in the multiwrite command.\n"
1197 " -P, -- use different pattern to fill file\n"
1198 " -C, -- report statistics in a machine parsable format\n"
1199 " -q, -- quiet mode, do not show I/O statistics\n"
1200 "\n");
1203 static int multiwrite_f(BlockBackend *blk, int argc, char **argv);
1205 static const cmdinfo_t multiwrite_cmd = {
1206 .name = "multiwrite",
1207 .cfunc = multiwrite_f,
1208 .argmin = 2,
1209 .argmax = -1,
1210 .args = "[-Cq] [-P pattern ] off len [len..] [; off len [len..]..]",
1211 .oneline = "issues multiple write requests at once",
1212 .help = multiwrite_help,
1215 static int multiwrite_f(BlockBackend *blk, int argc, char **argv)
1217 struct timeval t1, t2;
1218 int Cflag = 0, qflag = 0;
1219 int c, cnt;
1220 char **buf;
1221 int64_t offset, first_offset = 0;
1222 /* Some compilers get confused and warn if this is not initialized. */
1223 int total = 0;
1224 int nr_iov;
1225 int nr_reqs;
1226 int pattern = 0xcd;
1227 QEMUIOVector *qiovs;
1228 int i;
1229 BlockRequest *reqs;
1231 while ((c = getopt(argc, argv, "CqP:")) != EOF) {
1232 switch (c) {
1233 case 'C':
1234 Cflag = 1;
1235 break;
1236 case 'q':
1237 qflag = 1;
1238 break;
1239 case 'P':
1240 pattern = parse_pattern(optarg);
1241 if (pattern < 0) {
1242 return 0;
1244 break;
1245 default:
1246 return qemuio_command_usage(&writev_cmd);
1250 if (optind > argc - 2) {
1251 return qemuio_command_usage(&writev_cmd);
1254 nr_reqs = 1;
1255 for (i = optind; i < argc; i++) {
1256 if (!strcmp(argv[i], ";")) {
1257 nr_reqs++;
1261 reqs = g_new0(BlockRequest, nr_reqs);
1262 buf = g_new0(char *, nr_reqs);
1263 qiovs = g_new(QEMUIOVector, nr_reqs);
1265 for (i = 0; i < nr_reqs && optind < argc; i++) {
1266 int j;
1268 /* Read the offset of the request */
1269 offset = cvtnum(argv[optind]);
1270 if (offset < 0) {
1271 printf("non-numeric offset argument -- %s\n", argv[optind]);
1272 goto out;
1274 optind++;
1276 if (offset & 0x1ff) {
1277 printf("offset %lld is not sector aligned\n",
1278 (long long)offset);
1279 goto out;
1282 if (i == 0) {
1283 first_offset = offset;
1286 /* Read lengths for qiov entries */
1287 for (j = optind; j < argc; j++) {
1288 if (!strcmp(argv[j], ";")) {
1289 break;
1293 nr_iov = j - optind;
1295 /* Build request */
1296 buf[i] = create_iovec(blk, &qiovs[i], &argv[optind], nr_iov, pattern);
1297 if (buf[i] == NULL) {
1298 goto out;
1301 reqs[i].qiov = &qiovs[i];
1302 reqs[i].sector = offset >> 9;
1303 reqs[i].nb_sectors = reqs[i].qiov->size >> 9;
1305 optind = j + 1;
1307 pattern++;
1310 /* If there were empty requests at the end, ignore them */
1311 nr_reqs = i;
1313 gettimeofday(&t1, NULL);
1314 cnt = do_aio_multiwrite(blk, reqs, nr_reqs, &total);
1315 gettimeofday(&t2, NULL);
1317 if (cnt < 0) {
1318 printf("aio_multiwrite failed: %s\n", strerror(-cnt));
1319 goto out;
1322 if (qflag) {
1323 goto out;
1326 /* Finally, report back -- -C gives a parsable format */
1327 t2 = tsub(t2, t1);
1328 print_report("wrote", &t2, first_offset, total, total, cnt, Cflag);
1329 out:
1330 for (i = 0; i < nr_reqs; i++) {
1331 qemu_io_free(buf[i]);
1332 if (reqs[i].qiov != NULL) {
1333 qemu_iovec_destroy(&qiovs[i]);
1336 g_free(buf);
1337 g_free(reqs);
1338 g_free(qiovs);
1339 return 0;
1342 struct aio_ctx {
1343 BlockBackend *blk;
1344 QEMUIOVector qiov;
1345 int64_t offset;
1346 char *buf;
1347 int qflag;
1348 int vflag;
1349 int Cflag;
1350 int Pflag;
1351 BlockAcctCookie acct;
1352 int pattern;
1353 struct timeval t1;
1356 static void aio_write_done(void *opaque, int ret)
1358 struct aio_ctx *ctx = opaque;
1359 struct timeval t2;
1361 gettimeofday(&t2, NULL);
1364 if (ret < 0) {
1365 printf("aio_write failed: %s\n", strerror(-ret));
1366 goto out;
1369 block_acct_done(blk_get_stats(ctx->blk), &ctx->acct);
1371 if (ctx->qflag) {
1372 goto out;
1375 /* Finally, report back -- -C gives a parsable format */
1376 t2 = tsub(t2, ctx->t1);
1377 print_report("wrote", &t2, ctx->offset, ctx->qiov.size,
1378 ctx->qiov.size, 1, ctx->Cflag);
1379 out:
1380 qemu_io_free(ctx->buf);
1381 qemu_iovec_destroy(&ctx->qiov);
1382 g_free(ctx);
1385 static void aio_read_done(void *opaque, int ret)
1387 struct aio_ctx *ctx = opaque;
1388 struct timeval t2;
1390 gettimeofday(&t2, NULL);
1392 if (ret < 0) {
1393 printf("readv failed: %s\n", strerror(-ret));
1394 goto out;
1397 if (ctx->Pflag) {
1398 void *cmp_buf = g_malloc(ctx->qiov.size);
1400 memset(cmp_buf, ctx->pattern, ctx->qiov.size);
1401 if (memcmp(ctx->buf, cmp_buf, ctx->qiov.size)) {
1402 printf("Pattern verification failed at offset %"
1403 PRId64 ", %zd bytes\n", ctx->offset, ctx->qiov.size);
1405 g_free(cmp_buf);
1408 block_acct_done(blk_get_stats(ctx->blk), &ctx->acct);
1410 if (ctx->qflag) {
1411 goto out;
1414 if (ctx->vflag) {
1415 dump_buffer(ctx->buf, ctx->offset, ctx->qiov.size);
1418 /* Finally, report back -- -C gives a parsable format */
1419 t2 = tsub(t2, ctx->t1);
1420 print_report("read", &t2, ctx->offset, ctx->qiov.size,
1421 ctx->qiov.size, 1, ctx->Cflag);
1422 out:
1423 qemu_io_free(ctx->buf);
1424 qemu_iovec_destroy(&ctx->qiov);
1425 g_free(ctx);
1428 static void aio_read_help(void)
1430 printf(
1431 "\n"
1432 " asynchronously reads a range of bytes from the given offset\n"
1433 "\n"
1434 " Example:\n"
1435 " 'aio_read -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n"
1436 "\n"
1437 " Reads a segment of the currently open file, optionally dumping it to the\n"
1438 " standard output stream (with -v option) for subsequent inspection.\n"
1439 " The read is performed asynchronously and the aio_flush command must be\n"
1440 " used to ensure all outstanding aio requests have been completed.\n"
1441 " -C, -- report statistics in a machine parsable format\n"
1442 " -P, -- use a pattern to verify read data\n"
1443 " -v, -- dump buffer to standard output\n"
1444 " -q, -- quiet mode, do not show I/O statistics\n"
1445 "\n");
1448 static int aio_read_f(BlockBackend *blk, int argc, char **argv);
1450 static const cmdinfo_t aio_read_cmd = {
1451 .name = "aio_read",
1452 .cfunc = aio_read_f,
1453 .argmin = 2,
1454 .argmax = -1,
1455 .args = "[-Cqv] [-P pattern ] off len [len..]",
1456 .oneline = "asynchronously reads a number of bytes",
1457 .help = aio_read_help,
1460 static int aio_read_f(BlockBackend *blk, int argc, char **argv)
1462 int nr_iov, c;
1463 struct aio_ctx *ctx = g_new0(struct aio_ctx, 1);
1465 ctx->blk = blk;
1466 while ((c = getopt(argc, argv, "CP:qv")) != EOF) {
1467 switch (c) {
1468 case 'C':
1469 ctx->Cflag = 1;
1470 break;
1471 case 'P':
1472 ctx->Pflag = 1;
1473 ctx->pattern = parse_pattern(optarg);
1474 if (ctx->pattern < 0) {
1475 g_free(ctx);
1476 return 0;
1478 break;
1479 case 'q':
1480 ctx->qflag = 1;
1481 break;
1482 case 'v':
1483 ctx->vflag = 1;
1484 break;
1485 default:
1486 g_free(ctx);
1487 return qemuio_command_usage(&aio_read_cmd);
1491 if (optind > argc - 2) {
1492 g_free(ctx);
1493 return qemuio_command_usage(&aio_read_cmd);
1496 ctx->offset = cvtnum(argv[optind]);
1497 if (ctx->offset < 0) {
1498 printf("non-numeric length argument -- %s\n", argv[optind]);
1499 g_free(ctx);
1500 return 0;
1502 optind++;
1504 if (ctx->offset & 0x1ff) {
1505 printf("offset %" PRId64 " is not sector aligned\n",
1506 ctx->offset);
1507 g_free(ctx);
1508 return 0;
1511 nr_iov = argc - optind;
1512 ctx->buf = create_iovec(blk, &ctx->qiov, &argv[optind], nr_iov, 0xab);
1513 if (ctx->buf == NULL) {
1514 g_free(ctx);
1515 return 0;
1518 gettimeofday(&ctx->t1, NULL);
1519 block_acct_start(blk_get_stats(blk), &ctx->acct, ctx->qiov.size,
1520 BLOCK_ACCT_READ);
1521 blk_aio_readv(blk, ctx->offset >> 9, &ctx->qiov,
1522 ctx->qiov.size >> 9, aio_read_done, ctx);
1523 return 0;
1526 static void aio_write_help(void)
1528 printf(
1529 "\n"
1530 " asynchronously writes a range of bytes from the given offset source\n"
1531 " from multiple buffers\n"
1532 "\n"
1533 " Example:\n"
1534 " 'aio_write 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n"
1535 "\n"
1536 " Writes into a segment of the currently open file, using a buffer\n"
1537 " filled with a set pattern (0xcdcdcdcd).\n"
1538 " The write is performed asynchronously and the aio_flush command must be\n"
1539 " used to ensure all outstanding aio requests have been completed.\n"
1540 " -P, -- use different pattern to fill file\n"
1541 " -C, -- report statistics in a machine parsable format\n"
1542 " -q, -- quiet mode, do not show I/O statistics\n"
1543 "\n");
1546 static int aio_write_f(BlockBackend *blk, int argc, char **argv);
1548 static const cmdinfo_t aio_write_cmd = {
1549 .name = "aio_write",
1550 .cfunc = aio_write_f,
1551 .argmin = 2,
1552 .argmax = -1,
1553 .args = "[-Cq] [-P pattern ] off len [len..]",
1554 .oneline = "asynchronously writes a number of bytes",
1555 .help = aio_write_help,
1558 static int aio_write_f(BlockBackend *blk, int argc, char **argv)
1560 int nr_iov, c;
1561 int pattern = 0xcd;
1562 struct aio_ctx *ctx = g_new0(struct aio_ctx, 1);
1564 ctx->blk = blk;
1565 while ((c = getopt(argc, argv, "CqP:")) != EOF) {
1566 switch (c) {
1567 case 'C':
1568 ctx->Cflag = 1;
1569 break;
1570 case 'q':
1571 ctx->qflag = 1;
1572 break;
1573 case 'P':
1574 pattern = parse_pattern(optarg);
1575 if (pattern < 0) {
1576 g_free(ctx);
1577 return 0;
1579 break;
1580 default:
1581 g_free(ctx);
1582 return qemuio_command_usage(&aio_write_cmd);
1586 if (optind > argc - 2) {
1587 g_free(ctx);
1588 return qemuio_command_usage(&aio_write_cmd);
1591 ctx->offset = cvtnum(argv[optind]);
1592 if (ctx->offset < 0) {
1593 printf("non-numeric length argument -- %s\n", argv[optind]);
1594 g_free(ctx);
1595 return 0;
1597 optind++;
1599 if (ctx->offset & 0x1ff) {
1600 printf("offset %" PRId64 " is not sector aligned\n",
1601 ctx->offset);
1602 g_free(ctx);
1603 return 0;
1606 nr_iov = argc - optind;
1607 ctx->buf = create_iovec(blk, &ctx->qiov, &argv[optind], nr_iov, pattern);
1608 if (ctx->buf == NULL) {
1609 g_free(ctx);
1610 return 0;
1613 gettimeofday(&ctx->t1, NULL);
1614 block_acct_start(blk_get_stats(blk), &ctx->acct, ctx->qiov.size,
1615 BLOCK_ACCT_WRITE);
1616 blk_aio_writev(blk, ctx->offset >> 9, &ctx->qiov,
1617 ctx->qiov.size >> 9, aio_write_done, ctx);
1618 return 0;
1621 static int aio_flush_f(BlockBackend *blk, int argc, char **argv)
1623 blk_drain_all();
1624 return 0;
1627 static const cmdinfo_t aio_flush_cmd = {
1628 .name = "aio_flush",
1629 .cfunc = aio_flush_f,
1630 .oneline = "completes all outstanding aio requests"
1633 static int flush_f(BlockBackend *blk, int argc, char **argv)
1635 blk_flush(blk);
1636 return 0;
1639 static const cmdinfo_t flush_cmd = {
1640 .name = "flush",
1641 .altname = "f",
1642 .cfunc = flush_f,
1643 .oneline = "flush all in-core file state to disk",
1646 static int truncate_f(BlockBackend *blk, int argc, char **argv)
1648 int64_t offset;
1649 int ret;
1651 offset = cvtnum(argv[1]);
1652 if (offset < 0) {
1653 printf("non-numeric truncate argument -- %s\n", argv[1]);
1654 return 0;
1657 ret = blk_truncate(blk, offset);
1658 if (ret < 0) {
1659 printf("truncate: %s\n", strerror(-ret));
1660 return 0;
1663 return 0;
1666 static const cmdinfo_t truncate_cmd = {
1667 .name = "truncate",
1668 .altname = "t",
1669 .cfunc = truncate_f,
1670 .argmin = 1,
1671 .argmax = 1,
1672 .args = "off",
1673 .oneline = "truncates the current file at the given offset",
1676 static int length_f(BlockBackend *blk, int argc, char **argv)
1678 int64_t size;
1679 char s1[64];
1681 size = blk_getlength(blk);
1682 if (size < 0) {
1683 printf("getlength: %s\n", strerror(-size));
1684 return 0;
1687 cvtstr(size, s1, sizeof(s1));
1688 printf("%s\n", s1);
1689 return 0;
1693 static const cmdinfo_t length_cmd = {
1694 .name = "length",
1695 .altname = "l",
1696 .cfunc = length_f,
1697 .oneline = "gets the length of the current file",
1701 static int info_f(BlockBackend *blk, int argc, char **argv)
1703 BlockDriverState *bs = blk_bs(blk);
1704 BlockDriverInfo bdi;
1705 ImageInfoSpecific *spec_info;
1706 char s1[64], s2[64];
1707 int ret;
1709 if (bs->drv && bs->drv->format_name) {
1710 printf("format name: %s\n", bs->drv->format_name);
1712 if (bs->drv && bs->drv->protocol_name) {
1713 printf("format name: %s\n", bs->drv->protocol_name);
1716 ret = bdrv_get_info(bs, &bdi);
1717 if (ret) {
1718 return 0;
1721 cvtstr(bdi.cluster_size, s1, sizeof(s1));
1722 cvtstr(bdi.vm_state_offset, s2, sizeof(s2));
1724 printf("cluster size: %s\n", s1);
1725 printf("vm state offset: %s\n", s2);
1727 spec_info = bdrv_get_specific_info(bs);
1728 if (spec_info) {
1729 printf("Format specific information:\n");
1730 bdrv_image_info_specific_dump(fprintf, stdout, spec_info);
1731 qapi_free_ImageInfoSpecific(spec_info);
1734 return 0;
1739 static const cmdinfo_t info_cmd = {
1740 .name = "info",
1741 .altname = "i",
1742 .cfunc = info_f,
1743 .oneline = "prints information about the current file",
1746 static void discard_help(void)
1748 printf(
1749 "\n"
1750 " discards a range of bytes from the given offset\n"
1751 "\n"
1752 " Example:\n"
1753 " 'discard 512 1k' - discards 1 kilobyte from 512 bytes into the file\n"
1754 "\n"
1755 " Discards a segment of the currently open file.\n"
1756 " -C, -- report statistics in a machine parsable format\n"
1757 " -q, -- quiet mode, do not show I/O statistics\n"
1758 "\n");
1761 static int discard_f(BlockBackend *blk, int argc, char **argv);
1763 static const cmdinfo_t discard_cmd = {
1764 .name = "discard",
1765 .altname = "d",
1766 .cfunc = discard_f,
1767 .argmin = 2,
1768 .argmax = -1,
1769 .args = "[-Cq] off len",
1770 .oneline = "discards a number of bytes at a specified offset",
1771 .help = discard_help,
1774 static int discard_f(BlockBackend *blk, int argc, char **argv)
1776 struct timeval t1, t2;
1777 int Cflag = 0, qflag = 0;
1778 int c, ret;
1779 int64_t offset;
1780 int count;
1782 while ((c = getopt(argc, argv, "Cq")) != EOF) {
1783 switch (c) {
1784 case 'C':
1785 Cflag = 1;
1786 break;
1787 case 'q':
1788 qflag = 1;
1789 break;
1790 default:
1791 return qemuio_command_usage(&discard_cmd);
1795 if (optind != argc - 2) {
1796 return qemuio_command_usage(&discard_cmd);
1799 offset = cvtnum(argv[optind]);
1800 if (offset < 0) {
1801 printf("non-numeric length argument -- %s\n", argv[optind]);
1802 return 0;
1805 optind++;
1806 count = cvtnum(argv[optind]);
1807 if (count < 0) {
1808 printf("non-numeric length argument -- %s\n", argv[optind]);
1809 return 0;
1812 gettimeofday(&t1, NULL);
1813 ret = blk_discard(blk, offset >> BDRV_SECTOR_BITS,
1814 count >> BDRV_SECTOR_BITS);
1815 gettimeofday(&t2, NULL);
1817 if (ret < 0) {
1818 printf("discard failed: %s\n", strerror(-ret));
1819 goto out;
1822 /* Finally, report back -- -C gives a parsable format */
1823 if (!qflag) {
1824 t2 = tsub(t2, t1);
1825 print_report("discard", &t2, offset, count, count, 1, Cflag);
1828 out:
1829 return 0;
1832 static int alloc_f(BlockBackend *blk, int argc, char **argv)
1834 BlockDriverState *bs = blk_bs(blk);
1835 int64_t offset, sector_num;
1836 int nb_sectors, remaining;
1837 char s1[64];
1838 int num, sum_alloc;
1839 int ret;
1841 offset = cvtnum(argv[1]);
1842 if (offset < 0) {
1843 printf("non-numeric offset argument -- %s\n", argv[1]);
1844 return 0;
1845 } else if (offset & 0x1ff) {
1846 printf("offset %" PRId64 " is not sector aligned\n",
1847 offset);
1848 return 0;
1851 if (argc == 3) {
1852 nb_sectors = cvtnum(argv[2]);
1853 if (nb_sectors < 0) {
1854 printf("non-numeric length argument -- %s\n", argv[2]);
1855 return 0;
1857 } else {
1858 nb_sectors = 1;
1861 remaining = nb_sectors;
1862 sum_alloc = 0;
1863 sector_num = offset >> 9;
1864 while (remaining) {
1865 ret = bdrv_is_allocated(bs, sector_num, remaining, &num);
1866 if (ret < 0) {
1867 printf("is_allocated failed: %s\n", strerror(-ret));
1868 return 0;
1870 sector_num += num;
1871 remaining -= num;
1872 if (ret) {
1873 sum_alloc += num;
1875 if (num == 0) {
1876 nb_sectors -= remaining;
1877 remaining = 0;
1881 cvtstr(offset, s1, sizeof(s1));
1883 printf("%d/%d sectors allocated at offset %s\n",
1884 sum_alloc, nb_sectors, s1);
1885 return 0;
1888 static const cmdinfo_t alloc_cmd = {
1889 .name = "alloc",
1890 .altname = "a",
1891 .argmin = 1,
1892 .argmax = 2,
1893 .cfunc = alloc_f,
1894 .args = "off [sectors]",
1895 .oneline = "checks if a sector is present in the file",
1899 static int map_is_allocated(BlockDriverState *bs, int64_t sector_num,
1900 int64_t nb_sectors, int64_t *pnum)
1902 int num, num_checked;
1903 int ret, firstret;
1905 num_checked = MIN(nb_sectors, INT_MAX);
1906 ret = bdrv_is_allocated(bs, sector_num, num_checked, &num);
1907 if (ret < 0) {
1908 return ret;
1911 firstret = ret;
1912 *pnum = num;
1914 while (nb_sectors > 0 && ret == firstret) {
1915 sector_num += num;
1916 nb_sectors -= num;
1918 num_checked = MIN(nb_sectors, INT_MAX);
1919 ret = bdrv_is_allocated(bs, sector_num, num_checked, &num);
1920 if (ret == firstret && num) {
1921 *pnum += num;
1922 } else {
1923 break;
1927 return firstret;
1930 static int map_f(BlockBackend *blk, int argc, char **argv)
1932 int64_t offset;
1933 int64_t nb_sectors, total_sectors;
1934 char s1[64];
1935 int64_t num;
1936 int ret;
1937 const char *retstr;
1939 offset = 0;
1940 total_sectors = blk_nb_sectors(blk);
1941 if (total_sectors < 0) {
1942 error_report("Failed to query image length: %s",
1943 strerror(-total_sectors));
1944 return 0;
1947 nb_sectors = total_sectors;
1949 do {
1950 ret = map_is_allocated(blk_bs(blk), offset, nb_sectors, &num);
1951 if (ret < 0) {
1952 error_report("Failed to get allocation status: %s", strerror(-ret));
1953 return 0;
1954 } else if (!num) {
1955 error_report("Unexpected end of image");
1956 return 0;
1959 retstr = ret ? " allocated" : "not allocated";
1960 cvtstr(offset << 9ULL, s1, sizeof(s1));
1961 printf("[% 24" PRId64 "] % 8" PRId64 "/% 8" PRId64 " sectors %s "
1962 "at offset %s (%d)\n",
1963 offset << 9ULL, num, nb_sectors, retstr, s1, ret);
1965 offset += num;
1966 nb_sectors -= num;
1967 } while (offset < total_sectors);
1969 return 0;
1972 static const cmdinfo_t map_cmd = {
1973 .name = "map",
1974 .argmin = 0,
1975 .argmax = 0,
1976 .cfunc = map_f,
1977 .args = "",
1978 .oneline = "prints the allocated areas of a file",
1981 static int break_f(BlockBackend *blk, int argc, char **argv)
1983 int ret;
1985 ret = bdrv_debug_breakpoint(blk_bs(blk), argv[1], argv[2]);
1986 if (ret < 0) {
1987 printf("Could not set breakpoint: %s\n", strerror(-ret));
1990 return 0;
1993 static int remove_break_f(BlockBackend *blk, int argc, char **argv)
1995 int ret;
1997 ret = bdrv_debug_remove_breakpoint(blk_bs(blk), argv[1]);
1998 if (ret < 0) {
1999 printf("Could not remove breakpoint %s: %s\n", argv[1], strerror(-ret));
2002 return 0;
2005 static const cmdinfo_t break_cmd = {
2006 .name = "break",
2007 .argmin = 2,
2008 .argmax = 2,
2009 .cfunc = break_f,
2010 .args = "event tag",
2011 .oneline = "sets a breakpoint on event and tags the stopped "
2012 "request as tag",
2015 static const cmdinfo_t remove_break_cmd = {
2016 .name = "remove_break",
2017 .argmin = 1,
2018 .argmax = 1,
2019 .cfunc = remove_break_f,
2020 .args = "tag",
2021 .oneline = "remove a breakpoint by tag",
2024 static int resume_f(BlockBackend *blk, int argc, char **argv)
2026 int ret;
2028 ret = bdrv_debug_resume(blk_bs(blk), argv[1]);
2029 if (ret < 0) {
2030 printf("Could not resume request: %s\n", strerror(-ret));
2033 return 0;
2036 static const cmdinfo_t resume_cmd = {
2037 .name = "resume",
2038 .argmin = 1,
2039 .argmax = 1,
2040 .cfunc = resume_f,
2041 .args = "tag",
2042 .oneline = "resumes the request tagged as tag",
2045 static int wait_break_f(BlockBackend *blk, int argc, char **argv)
2047 while (!bdrv_debug_is_suspended(blk_bs(blk), argv[1])) {
2048 aio_poll(blk_get_aio_context(blk), true);
2051 return 0;
2054 static const cmdinfo_t wait_break_cmd = {
2055 .name = "wait_break",
2056 .argmin = 1,
2057 .argmax = 1,
2058 .cfunc = wait_break_f,
2059 .args = "tag",
2060 .oneline = "waits for the suspension of a request",
2063 static int abort_f(BlockBackend *blk, int argc, char **argv)
2065 abort();
2068 static const cmdinfo_t abort_cmd = {
2069 .name = "abort",
2070 .cfunc = abort_f,
2071 .flags = CMD_NOFILE_OK,
2072 .oneline = "simulate a program crash using abort(3)",
2075 static void sigraise_help(void)
2077 printf(
2078 "\n"
2079 " raises the given signal\n"
2080 "\n"
2081 " Example:\n"
2082 " 'sigraise %i' - raises SIGTERM\n"
2083 "\n"
2084 " Invokes raise(signal), where \"signal\" is the mandatory integer argument\n"
2085 " given to sigraise.\n"
2086 "\n", SIGTERM);
2089 static int sigraise_f(BlockBackend *blk, int argc, char **argv);
2091 static const cmdinfo_t sigraise_cmd = {
2092 .name = "sigraise",
2093 .cfunc = sigraise_f,
2094 .argmin = 1,
2095 .argmax = 1,
2096 .flags = CMD_NOFILE_OK,
2097 .args = "signal",
2098 .oneline = "raises a signal",
2099 .help = sigraise_help,
2102 static int sigraise_f(BlockBackend *blk, int argc, char **argv)
2104 int sig = cvtnum(argv[1]);
2105 if (sig < 0) {
2106 printf("non-numeric signal number argument -- %s\n", argv[1]);
2107 return 0;
2110 /* Using raise() to kill this process does not necessarily flush all open
2111 * streams. At least stdout and stderr (although the latter should be
2112 * non-buffered anyway) should be flushed, though. */
2113 fflush(stdout);
2114 fflush(stderr);
2116 raise(sig);
2117 return 0;
2120 static void sleep_cb(void *opaque)
2122 bool *expired = opaque;
2123 *expired = true;
2126 static int sleep_f(BlockBackend *blk, int argc, char **argv)
2128 char *endptr;
2129 long ms;
2130 struct QEMUTimer *timer;
2131 bool expired = false;
2133 ms = strtol(argv[1], &endptr, 0);
2134 if (ms < 0 || *endptr != '\0') {
2135 printf("%s is not a valid number\n", argv[1]);
2136 return 0;
2139 timer = timer_new_ns(QEMU_CLOCK_HOST, sleep_cb, &expired);
2140 timer_mod(timer, qemu_clock_get_ns(QEMU_CLOCK_HOST) + SCALE_MS * ms);
2142 while (!expired) {
2143 main_loop_wait(false);
2146 timer_free(timer);
2148 return 0;
2151 static const cmdinfo_t sleep_cmd = {
2152 .name = "sleep",
2153 .argmin = 1,
2154 .argmax = 1,
2155 .cfunc = sleep_f,
2156 .flags = CMD_NOFILE_OK,
2157 .oneline = "waits for the given value in milliseconds",
2160 static void help_oneline(const char *cmd, const cmdinfo_t *ct)
2162 if (cmd) {
2163 printf("%s ", cmd);
2164 } else {
2165 printf("%s ", ct->name);
2166 if (ct->altname) {
2167 printf("(or %s) ", ct->altname);
2171 if (ct->args) {
2172 printf("%s ", ct->args);
2174 printf("-- %s\n", ct->oneline);
2177 static void help_onecmd(const char *cmd, const cmdinfo_t *ct)
2179 help_oneline(cmd, ct);
2180 if (ct->help) {
2181 ct->help();
2185 static void help_all(void)
2187 const cmdinfo_t *ct;
2189 for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) {
2190 help_oneline(ct->name, ct);
2192 printf("\nUse 'help commandname' for extended help.\n");
2195 static int help_f(BlockBackend *blk, int argc, char **argv)
2197 const cmdinfo_t *ct;
2199 if (argc == 1) {
2200 help_all();
2201 return 0;
2204 ct = find_command(argv[1]);
2205 if (ct == NULL) {
2206 printf("command %s not found\n", argv[1]);
2207 return 0;
2210 help_onecmd(argv[1], ct);
2211 return 0;
2214 static const cmdinfo_t help_cmd = {
2215 .name = "help",
2216 .altname = "?",
2217 .cfunc = help_f,
2218 .argmin = 0,
2219 .argmax = 1,
2220 .flags = CMD_FLAG_GLOBAL,
2221 .args = "[command]",
2222 .oneline = "help for one or all commands",
2225 bool qemuio_command(BlockBackend *blk, const char *cmd)
2227 char *input;
2228 const cmdinfo_t *ct;
2229 char **v;
2230 int c;
2231 bool done = false;
2233 input = g_strdup(cmd);
2234 v = breakline(input, &c);
2235 if (c) {
2236 ct = find_command(v[0]);
2237 if (ct) {
2238 done = command(blk, ct, c, v);
2239 } else {
2240 fprintf(stderr, "command \"%s\" not found\n", v[0]);
2243 g_free(input);
2244 g_free(v);
2246 return done;
2249 static void __attribute((constructor)) init_qemuio_commands(void)
2251 /* initialize commands */
2252 qemuio_add_command(&help_cmd);
2253 qemuio_add_command(&read_cmd);
2254 qemuio_add_command(&readv_cmd);
2255 qemuio_add_command(&write_cmd);
2256 qemuio_add_command(&writev_cmd);
2257 qemuio_add_command(&multiwrite_cmd);
2258 qemuio_add_command(&aio_read_cmd);
2259 qemuio_add_command(&aio_write_cmd);
2260 qemuio_add_command(&aio_flush_cmd);
2261 qemuio_add_command(&flush_cmd);
2262 qemuio_add_command(&truncate_cmd);
2263 qemuio_add_command(&length_cmd);
2264 qemuio_add_command(&info_cmd);
2265 qemuio_add_command(&discard_cmd);
2266 qemuio_add_command(&alloc_cmd);
2267 qemuio_add_command(&map_cmd);
2268 qemuio_add_command(&break_cmd);
2269 qemuio_add_command(&remove_break_cmd);
2270 qemuio_add_command(&resume_cmd);
2271 qemuio_add_command(&wait_break_cmd);
2272 qemuio_add_command(&abort_cmd);
2273 qemuio_add_command(&sleep_cmd);
2274 qemuio_add_command(&sigraise_cmd);