1797ac58cSKevin Wolf /* 2797ac58cSKevin Wolf * Command line utility to exercise the QEMU I/O path. 3797ac58cSKevin Wolf * 4797ac58cSKevin Wolf * Copyright (C) 2009 Red Hat, Inc. 5797ac58cSKevin Wolf * Copyright (c) 2003-2005 Silicon Graphics, Inc. 6797ac58cSKevin Wolf * 7797ac58cSKevin Wolf * This work is licensed under the terms of the GNU GPL, version 2 or later. 8797ac58cSKevin Wolf * See the COPYING file in the top-level directory. 9797ac58cSKevin Wolf */ 10797ac58cSKevin Wolf 113d21994fSKevin Wolf #include "qemu-io.h" 12797ac58cSKevin Wolf #include "block/block_int.h" 13a8d8ecb7SMax Reitz #include "block/qapi.h" 146a1751b7SAlex Bligh #include "qemu/main-loop.h" 15cd33d02aSKevin Wolf #include "qemu/timer.h" 16797ac58cSKevin Wolf 17797ac58cSKevin Wolf #define CMD_NOFILE_OK 0x01 18797ac58cSKevin Wolf 19f9883880SStefan Weil bool qemuio_misalign; 20797ac58cSKevin Wolf 21c2cdf5c5SKevin Wolf static cmdinfo_t *cmdtab; 22c2cdf5c5SKevin Wolf static int ncmds; 23c2cdf5c5SKevin Wolf 24c2cdf5c5SKevin Wolf static int compare_cmdname(const void *a, const void *b) 25c2cdf5c5SKevin Wolf { 26c2cdf5c5SKevin Wolf return strcmp(((const cmdinfo_t *)a)->name, 27c2cdf5c5SKevin Wolf ((const cmdinfo_t *)b)->name); 28c2cdf5c5SKevin Wolf } 29c2cdf5c5SKevin Wolf 30c2cdf5c5SKevin Wolf void qemuio_add_command(const cmdinfo_t *ci) 31c2cdf5c5SKevin Wolf { 3202c4f26bSMarkus Armbruster cmdtab = g_renew(cmdinfo_t, cmdtab, ++ncmds); 33c2cdf5c5SKevin Wolf cmdtab[ncmds - 1] = *ci; 34c2cdf5c5SKevin Wolf qsort(cmdtab, ncmds, sizeof(*cmdtab), compare_cmdname); 35c2cdf5c5SKevin Wolf } 36c2cdf5c5SKevin Wolf 37c2cdf5c5SKevin Wolf int qemuio_command_usage(const cmdinfo_t *ci) 38c2cdf5c5SKevin Wolf { 39c2cdf5c5SKevin Wolf printf("%s %s -- %s\n", ci->name, ci->args, ci->oneline); 40c2cdf5c5SKevin Wolf return 0; 41c2cdf5c5SKevin Wolf } 42c2cdf5c5SKevin Wolf 43c2cdf5c5SKevin Wolf static int init_check_command(BlockDriverState *bs, const cmdinfo_t *ct) 44c2cdf5c5SKevin Wolf { 45c2cdf5c5SKevin Wolf if (ct->flags & CMD_FLAG_GLOBAL) { 46c2cdf5c5SKevin Wolf return 1; 47c2cdf5c5SKevin Wolf } 48c2cdf5c5SKevin Wolf if (!(ct->flags & CMD_NOFILE_OK) && !bs) { 49c2cdf5c5SKevin Wolf fprintf(stderr, "no file open, try 'help open'\n"); 50c2cdf5c5SKevin Wolf return 0; 51c2cdf5c5SKevin Wolf } 52c2cdf5c5SKevin Wolf return 1; 53c2cdf5c5SKevin Wolf } 54c2cdf5c5SKevin Wolf 553d21994fSKevin Wolf static int command(BlockDriverState *bs, const cmdinfo_t *ct, int argc, 563d21994fSKevin Wolf char **argv) 57c2cdf5c5SKevin Wolf { 58c2cdf5c5SKevin Wolf char *cmd = argv[0]; 59c2cdf5c5SKevin Wolf 603d21994fSKevin Wolf if (!init_check_command(bs, ct)) { 61c2cdf5c5SKevin Wolf return 0; 62c2cdf5c5SKevin Wolf } 63c2cdf5c5SKevin Wolf 64c2cdf5c5SKevin Wolf if (argc - 1 < ct->argmin || (ct->argmax != -1 && argc - 1 > ct->argmax)) { 65c2cdf5c5SKevin Wolf if (ct->argmax == -1) { 66c2cdf5c5SKevin Wolf fprintf(stderr, 67c2cdf5c5SKevin Wolf "bad argument count %d to %s, expected at least %d arguments\n", 68c2cdf5c5SKevin Wolf argc-1, cmd, ct->argmin); 69c2cdf5c5SKevin Wolf } else if (ct->argmin == ct->argmax) { 70c2cdf5c5SKevin Wolf fprintf(stderr, 71c2cdf5c5SKevin Wolf "bad argument count %d to %s, expected %d arguments\n", 72c2cdf5c5SKevin Wolf argc-1, cmd, ct->argmin); 73c2cdf5c5SKevin Wolf } else { 74c2cdf5c5SKevin Wolf fprintf(stderr, 75c2cdf5c5SKevin Wolf "bad argument count %d to %s, expected between %d and %d arguments\n", 76c2cdf5c5SKevin Wolf argc-1, cmd, ct->argmin, ct->argmax); 77c2cdf5c5SKevin Wolf } 78c2cdf5c5SKevin Wolf return 0; 79c2cdf5c5SKevin Wolf } 80c2cdf5c5SKevin Wolf optind = 0; 813d21994fSKevin Wolf return ct->cfunc(bs, argc, argv); 82c2cdf5c5SKevin Wolf } 83c2cdf5c5SKevin Wolf 84c2cdf5c5SKevin Wolf static const cmdinfo_t *find_command(const char *cmd) 85c2cdf5c5SKevin Wolf { 86c2cdf5c5SKevin Wolf cmdinfo_t *ct; 87c2cdf5c5SKevin Wolf 88c2cdf5c5SKevin Wolf for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { 89c2cdf5c5SKevin Wolf if (strcmp(ct->name, cmd) == 0 || 90c2cdf5c5SKevin Wolf (ct->altname && strcmp(ct->altname, cmd) == 0)) 91c2cdf5c5SKevin Wolf { 92c2cdf5c5SKevin Wolf return (const cmdinfo_t *)ct; 93c2cdf5c5SKevin Wolf } 94c2cdf5c5SKevin Wolf } 95c2cdf5c5SKevin Wolf return NULL; 96c2cdf5c5SKevin Wolf } 97c2cdf5c5SKevin Wolf 984694020dSStefan Hajnoczi /* Invoke fn() for commands with a matching prefix */ 994694020dSStefan Hajnoczi void qemuio_complete_command(const char *input, 1004694020dSStefan Hajnoczi void (*fn)(const char *cmd, void *opaque), 1014694020dSStefan Hajnoczi void *opaque) 1024694020dSStefan Hajnoczi { 1034694020dSStefan Hajnoczi cmdinfo_t *ct; 1044694020dSStefan Hajnoczi size_t input_len = strlen(input); 1054694020dSStefan Hajnoczi 1064694020dSStefan Hajnoczi for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { 1074694020dSStefan Hajnoczi if (strncmp(input, ct->name, input_len) == 0) { 1084694020dSStefan Hajnoczi fn(ct->name, opaque); 1094694020dSStefan Hajnoczi } 1104694020dSStefan Hajnoczi } 1114694020dSStefan Hajnoczi } 1124694020dSStefan Hajnoczi 113c2cdf5c5SKevin Wolf static char **breakline(char *input, int *count) 114c2cdf5c5SKevin Wolf { 115c2cdf5c5SKevin Wolf int c = 0; 116c2cdf5c5SKevin Wolf char *p; 1175839e53bSMarkus Armbruster char **rval = g_new0(char *, 1); 118c2cdf5c5SKevin Wolf 119c2cdf5c5SKevin Wolf while (rval && (p = qemu_strsep(&input, " ")) != NULL) { 120c2cdf5c5SKevin Wolf if (!*p) { 121c2cdf5c5SKevin Wolf continue; 122c2cdf5c5SKevin Wolf } 123c2cdf5c5SKevin Wolf c++; 124*08193dd5SMarkus Armbruster rval = g_renew(char *, rval, (c + 1)); 125c2cdf5c5SKevin Wolf rval[c - 1] = p; 126c2cdf5c5SKevin Wolf rval[c] = NULL; 127c2cdf5c5SKevin Wolf } 128c2cdf5c5SKevin Wolf *count = c; 129c2cdf5c5SKevin Wolf return rval; 130c2cdf5c5SKevin Wolf } 131c2cdf5c5SKevin Wolf 132797ac58cSKevin Wolf static int64_t cvtnum(const char *s) 133797ac58cSKevin Wolf { 134797ac58cSKevin Wolf char *end; 135797ac58cSKevin Wolf return strtosz_suffix(s, &end, STRTOSZ_DEFSUFFIX_B); 136797ac58cSKevin Wolf } 137797ac58cSKevin Wolf 1380b613881SKevin Wolf #define EXABYTES(x) ((long long)(x) << 60) 1390b613881SKevin Wolf #define PETABYTES(x) ((long long)(x) << 50) 1400b613881SKevin Wolf #define TERABYTES(x) ((long long)(x) << 40) 1410b613881SKevin Wolf #define GIGABYTES(x) ((long long)(x) << 30) 1420b613881SKevin Wolf #define MEGABYTES(x) ((long long)(x) << 20) 1430b613881SKevin Wolf #define KILOBYTES(x) ((long long)(x) << 10) 1440b613881SKevin Wolf 1450b613881SKevin Wolf #define TO_EXABYTES(x) ((x) / EXABYTES(1)) 1460b613881SKevin Wolf #define TO_PETABYTES(x) ((x) / PETABYTES(1)) 1470b613881SKevin Wolf #define TO_TERABYTES(x) ((x) / TERABYTES(1)) 1480b613881SKevin Wolf #define TO_GIGABYTES(x) ((x) / GIGABYTES(1)) 1490b613881SKevin Wolf #define TO_MEGABYTES(x) ((x) / MEGABYTES(1)) 1500b613881SKevin Wolf #define TO_KILOBYTES(x) ((x) / KILOBYTES(1)) 1510b613881SKevin Wolf 1520b613881SKevin Wolf static void cvtstr(double value, char *str, size_t size) 1530b613881SKevin Wolf { 1540b613881SKevin Wolf char *trim; 1550b613881SKevin Wolf const char *suffix; 1560b613881SKevin Wolf 1570b613881SKevin Wolf if (value >= EXABYTES(1)) { 1580b613881SKevin Wolf suffix = " EiB"; 1590b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_EXABYTES(value)); 1600b613881SKevin Wolf } else if (value >= PETABYTES(1)) { 1610b613881SKevin Wolf suffix = " PiB"; 1620b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_PETABYTES(value)); 1630b613881SKevin Wolf } else if (value >= TERABYTES(1)) { 1640b613881SKevin Wolf suffix = " TiB"; 1650b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_TERABYTES(value)); 1660b613881SKevin Wolf } else if (value >= GIGABYTES(1)) { 1670b613881SKevin Wolf suffix = " GiB"; 1680b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_GIGABYTES(value)); 1690b613881SKevin Wolf } else if (value >= MEGABYTES(1)) { 1700b613881SKevin Wolf suffix = " MiB"; 1710b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_MEGABYTES(value)); 1720b613881SKevin Wolf } else if (value >= KILOBYTES(1)) { 1730b613881SKevin Wolf suffix = " KiB"; 1740b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_KILOBYTES(value)); 1750b613881SKevin Wolf } else { 1760b613881SKevin Wolf suffix = " bytes"; 1770b613881SKevin Wolf snprintf(str, size - 6, "%f", value); 1780b613881SKevin Wolf } 1790b613881SKevin Wolf 1800b613881SKevin Wolf trim = strstr(str, ".000"); 1810b613881SKevin Wolf if (trim) { 1820b613881SKevin Wolf strcpy(trim, suffix); 1830b613881SKevin Wolf } else { 1840b613881SKevin Wolf strcat(str, suffix); 1850b613881SKevin Wolf } 1860b613881SKevin Wolf } 1870b613881SKevin Wolf 1880b613881SKevin Wolf 1890b613881SKevin Wolf 1900b613881SKevin Wolf static struct timeval tsub(struct timeval t1, struct timeval t2) 1910b613881SKevin Wolf { 1920b613881SKevin Wolf t1.tv_usec -= t2.tv_usec; 1930b613881SKevin Wolf if (t1.tv_usec < 0) { 1940b613881SKevin Wolf t1.tv_usec += 1000000; 1950b613881SKevin Wolf t1.tv_sec--; 1960b613881SKevin Wolf } 1970b613881SKevin Wolf t1.tv_sec -= t2.tv_sec; 1980b613881SKevin Wolf return t1; 1990b613881SKevin Wolf } 2000b613881SKevin Wolf 2010b613881SKevin Wolf static double tdiv(double value, struct timeval tv) 2020b613881SKevin Wolf { 2030b613881SKevin Wolf return value / ((double)tv.tv_sec + ((double)tv.tv_usec / 1000000.0)); 2040b613881SKevin Wolf } 2050b613881SKevin Wolf 2060b613881SKevin Wolf #define HOURS(sec) ((sec) / (60 * 60)) 2070b613881SKevin Wolf #define MINUTES(sec) (((sec) % (60 * 60)) / 60) 2080b613881SKevin Wolf #define SECONDS(sec) ((sec) % 60) 2090b613881SKevin Wolf 2100b613881SKevin Wolf enum { 2110b613881SKevin Wolf DEFAULT_TIME = 0x0, 2120b613881SKevin Wolf TERSE_FIXED_TIME = 0x1, 2130b613881SKevin Wolf VERBOSE_FIXED_TIME = 0x2, 2140b613881SKevin Wolf }; 2150b613881SKevin Wolf 2160b613881SKevin Wolf static void timestr(struct timeval *tv, char *ts, size_t size, int format) 2170b613881SKevin Wolf { 2180b613881SKevin Wolf double usec = (double)tv->tv_usec / 1000000.0; 2190b613881SKevin Wolf 2200b613881SKevin Wolf if (format & TERSE_FIXED_TIME) { 2210b613881SKevin Wolf if (!HOURS(tv->tv_sec)) { 2220b613881SKevin Wolf snprintf(ts, size, "%u:%02u.%02u", 2230b613881SKevin Wolf (unsigned int) MINUTES(tv->tv_sec), 2240b613881SKevin Wolf (unsigned int) SECONDS(tv->tv_sec), 2250b613881SKevin Wolf (unsigned int) (usec * 100)); 2260b613881SKevin Wolf return; 2270b613881SKevin Wolf } 2280b613881SKevin Wolf format |= VERBOSE_FIXED_TIME; /* fallback if hours needed */ 2290b613881SKevin Wolf } 2300b613881SKevin Wolf 2310b613881SKevin Wolf if ((format & VERBOSE_FIXED_TIME) || tv->tv_sec) { 2320b613881SKevin Wolf snprintf(ts, size, "%u:%02u:%02u.%02u", 2330b613881SKevin Wolf (unsigned int) HOURS(tv->tv_sec), 2340b613881SKevin Wolf (unsigned int) MINUTES(tv->tv_sec), 2350b613881SKevin Wolf (unsigned int) SECONDS(tv->tv_sec), 2360b613881SKevin Wolf (unsigned int) (usec * 100)); 2370b613881SKevin Wolf } else { 2380b613881SKevin Wolf snprintf(ts, size, "0.%04u sec", (unsigned int) (usec * 10000)); 2390b613881SKevin Wolf } 2400b613881SKevin Wolf } 2410b613881SKevin Wolf 242797ac58cSKevin Wolf /* 243797ac58cSKevin Wolf * Parse the pattern argument to various sub-commands. 244797ac58cSKevin Wolf * 245797ac58cSKevin Wolf * Because the pattern is used as an argument to memset it must evaluate 246797ac58cSKevin Wolf * to an unsigned integer that fits into a single byte. 247797ac58cSKevin Wolf */ 248797ac58cSKevin Wolf static int parse_pattern(const char *arg) 249797ac58cSKevin Wolf { 250797ac58cSKevin Wolf char *endptr = NULL; 251797ac58cSKevin Wolf long pattern; 252797ac58cSKevin Wolf 253797ac58cSKevin Wolf pattern = strtol(arg, &endptr, 0); 254797ac58cSKevin Wolf if (pattern < 0 || pattern > UCHAR_MAX || *endptr != '\0') { 255797ac58cSKevin Wolf printf("%s is not a valid pattern byte\n", arg); 256797ac58cSKevin Wolf return -1; 257797ac58cSKevin Wolf } 258797ac58cSKevin Wolf 259797ac58cSKevin Wolf return pattern; 260797ac58cSKevin Wolf } 261797ac58cSKevin Wolf 262797ac58cSKevin Wolf /* 263797ac58cSKevin Wolf * Memory allocation helpers. 264797ac58cSKevin Wolf * 265797ac58cSKevin Wolf * Make sure memory is aligned by default, or purposefully misaligned if 266797ac58cSKevin Wolf * that is specified on the command line. 267797ac58cSKevin Wolf */ 268797ac58cSKevin Wolf 269797ac58cSKevin Wolf #define MISALIGN_OFFSET 16 270797ac58cSKevin Wolf static void *qemu_io_alloc(BlockDriverState *bs, size_t len, int pattern) 271797ac58cSKevin Wolf { 272797ac58cSKevin Wolf void *buf; 273797ac58cSKevin Wolf 274797ac58cSKevin Wolf if (qemuio_misalign) { 275797ac58cSKevin Wolf len += MISALIGN_OFFSET; 276797ac58cSKevin Wolf } 277797ac58cSKevin Wolf buf = qemu_blockalign(bs, len); 278797ac58cSKevin Wolf memset(buf, pattern, len); 279797ac58cSKevin Wolf if (qemuio_misalign) { 280797ac58cSKevin Wolf buf += MISALIGN_OFFSET; 281797ac58cSKevin Wolf } 282797ac58cSKevin Wolf return buf; 283797ac58cSKevin Wolf } 284797ac58cSKevin Wolf 285797ac58cSKevin Wolf static void qemu_io_free(void *p) 286797ac58cSKevin Wolf { 287797ac58cSKevin Wolf if (qemuio_misalign) { 288797ac58cSKevin Wolf p -= MISALIGN_OFFSET; 289797ac58cSKevin Wolf } 290797ac58cSKevin Wolf qemu_vfree(p); 291797ac58cSKevin Wolf } 292797ac58cSKevin Wolf 293797ac58cSKevin Wolf static void dump_buffer(const void *buffer, int64_t offset, int len) 294797ac58cSKevin Wolf { 295797ac58cSKevin Wolf int i, j; 296797ac58cSKevin Wolf const uint8_t *p; 297797ac58cSKevin Wolf 298797ac58cSKevin Wolf for (i = 0, p = buffer; i < len; i += 16) { 299797ac58cSKevin Wolf const uint8_t *s = p; 300797ac58cSKevin Wolf 301797ac58cSKevin Wolf printf("%08" PRIx64 ": ", offset + i); 302797ac58cSKevin Wolf for (j = 0; j < 16 && i + j < len; j++, p++) { 303797ac58cSKevin Wolf printf("%02x ", *p); 304797ac58cSKevin Wolf } 305797ac58cSKevin Wolf printf(" "); 306797ac58cSKevin Wolf for (j = 0; j < 16 && i + j < len; j++, s++) { 307797ac58cSKevin Wolf if (isalnum(*s)) { 308797ac58cSKevin Wolf printf("%c", *s); 309797ac58cSKevin Wolf } else { 310797ac58cSKevin Wolf printf("."); 311797ac58cSKevin Wolf } 312797ac58cSKevin Wolf } 313797ac58cSKevin Wolf printf("\n"); 314797ac58cSKevin Wolf } 315797ac58cSKevin Wolf } 316797ac58cSKevin Wolf 317797ac58cSKevin Wolf static void print_report(const char *op, struct timeval *t, int64_t offset, 318797ac58cSKevin Wolf int count, int total, int cnt, int Cflag) 319797ac58cSKevin Wolf { 320797ac58cSKevin Wolf char s1[64], s2[64], ts[64]; 321797ac58cSKevin Wolf 322797ac58cSKevin Wolf timestr(t, ts, sizeof(ts), Cflag ? VERBOSE_FIXED_TIME : 0); 323797ac58cSKevin Wolf if (!Cflag) { 324797ac58cSKevin Wolf cvtstr((double)total, s1, sizeof(s1)); 325797ac58cSKevin Wolf cvtstr(tdiv((double)total, *t), s2, sizeof(s2)); 326797ac58cSKevin Wolf printf("%s %d/%d bytes at offset %" PRId64 "\n", 327797ac58cSKevin Wolf op, total, count, offset); 328797ac58cSKevin Wolf printf("%s, %d ops; %s (%s/sec and %.4f ops/sec)\n", 329797ac58cSKevin Wolf s1, cnt, ts, s2, tdiv((double)cnt, *t)); 330797ac58cSKevin Wolf } else {/* bytes,ops,time,bytes/sec,ops/sec */ 331797ac58cSKevin Wolf printf("%d,%d,%s,%.3f,%.3f\n", 332797ac58cSKevin Wolf total, cnt, ts, 333797ac58cSKevin Wolf tdiv((double)total, *t), 334797ac58cSKevin Wolf tdiv((double)cnt, *t)); 335797ac58cSKevin Wolf } 336797ac58cSKevin Wolf } 337797ac58cSKevin Wolf 338797ac58cSKevin Wolf /* 339797ac58cSKevin Wolf * Parse multiple length statements for vectored I/O, and construct an I/O 340797ac58cSKevin Wolf * vector matching it. 341797ac58cSKevin Wolf */ 342797ac58cSKevin Wolf static void * 343797ac58cSKevin Wolf create_iovec(BlockDriverState *bs, QEMUIOVector *qiov, char **argv, int nr_iov, 344797ac58cSKevin Wolf int pattern) 345797ac58cSKevin Wolf { 346797ac58cSKevin Wolf size_t *sizes = g_new0(size_t, nr_iov); 347797ac58cSKevin Wolf size_t count = 0; 348797ac58cSKevin Wolf void *buf = NULL; 349797ac58cSKevin Wolf void *p; 350797ac58cSKevin Wolf int i; 351797ac58cSKevin Wolf 352797ac58cSKevin Wolf for (i = 0; i < nr_iov; i++) { 353797ac58cSKevin Wolf char *arg = argv[i]; 354797ac58cSKevin Wolf int64_t len; 355797ac58cSKevin Wolf 356797ac58cSKevin Wolf len = cvtnum(arg); 357797ac58cSKevin Wolf if (len < 0) { 358797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", arg); 359797ac58cSKevin Wolf goto fail; 360797ac58cSKevin Wolf } 361797ac58cSKevin Wolf 362797ac58cSKevin Wolf /* should be SIZE_T_MAX, but that doesn't exist */ 363797ac58cSKevin Wolf if (len > INT_MAX) { 364797ac58cSKevin Wolf printf("too large length argument -- %s\n", arg); 365797ac58cSKevin Wolf goto fail; 366797ac58cSKevin Wolf } 367797ac58cSKevin Wolf 368797ac58cSKevin Wolf if (len & 0x1ff) { 369797ac58cSKevin Wolf printf("length argument %" PRId64 370797ac58cSKevin Wolf " is not sector aligned\n", len); 371797ac58cSKevin Wolf goto fail; 372797ac58cSKevin Wolf } 373797ac58cSKevin Wolf 374797ac58cSKevin Wolf sizes[i] = len; 375797ac58cSKevin Wolf count += len; 376797ac58cSKevin Wolf } 377797ac58cSKevin Wolf 378797ac58cSKevin Wolf qemu_iovec_init(qiov, nr_iov); 379797ac58cSKevin Wolf 380797ac58cSKevin Wolf buf = p = qemu_io_alloc(bs, count, pattern); 381797ac58cSKevin Wolf 382797ac58cSKevin Wolf for (i = 0; i < nr_iov; i++) { 383797ac58cSKevin Wolf qemu_iovec_add(qiov, p, sizes[i]); 384797ac58cSKevin Wolf p += sizes[i]; 385797ac58cSKevin Wolf } 386797ac58cSKevin Wolf 387797ac58cSKevin Wolf fail: 388797ac58cSKevin Wolf g_free(sizes); 389797ac58cSKevin Wolf return buf; 390797ac58cSKevin Wolf } 391797ac58cSKevin Wolf 392797ac58cSKevin Wolf static int do_read(BlockDriverState *bs, char *buf, int64_t offset, int count, 393797ac58cSKevin Wolf int *total) 394797ac58cSKevin Wolf { 395797ac58cSKevin Wolf int ret; 396797ac58cSKevin Wolf 397797ac58cSKevin Wolf ret = bdrv_read(bs, offset >> 9, (uint8_t *)buf, count >> 9); 398797ac58cSKevin Wolf if (ret < 0) { 399797ac58cSKevin Wolf return ret; 400797ac58cSKevin Wolf } 401797ac58cSKevin Wolf *total = count; 402797ac58cSKevin Wolf return 1; 403797ac58cSKevin Wolf } 404797ac58cSKevin Wolf 405797ac58cSKevin Wolf static int do_write(BlockDriverState *bs, char *buf, int64_t offset, int count, 406797ac58cSKevin Wolf int *total) 407797ac58cSKevin Wolf { 408797ac58cSKevin Wolf int ret; 409797ac58cSKevin Wolf 410797ac58cSKevin Wolf ret = bdrv_write(bs, offset >> 9, (uint8_t *)buf, count >> 9); 411797ac58cSKevin Wolf if (ret < 0) { 412797ac58cSKevin Wolf return ret; 413797ac58cSKevin Wolf } 414797ac58cSKevin Wolf *total = count; 415797ac58cSKevin Wolf return 1; 416797ac58cSKevin Wolf } 417797ac58cSKevin Wolf 418797ac58cSKevin Wolf static int do_pread(BlockDriverState *bs, char *buf, int64_t offset, int count, 419797ac58cSKevin Wolf int *total) 420797ac58cSKevin Wolf { 421797ac58cSKevin Wolf *total = bdrv_pread(bs, offset, (uint8_t *)buf, count); 422797ac58cSKevin Wolf if (*total < 0) { 423797ac58cSKevin Wolf return *total; 424797ac58cSKevin Wolf } 425797ac58cSKevin Wolf return 1; 426797ac58cSKevin Wolf } 427797ac58cSKevin Wolf 428797ac58cSKevin Wolf static int do_pwrite(BlockDriverState *bs, char *buf, int64_t offset, int count, 429797ac58cSKevin Wolf int *total) 430797ac58cSKevin Wolf { 431797ac58cSKevin Wolf *total = bdrv_pwrite(bs, offset, (uint8_t *)buf, count); 432797ac58cSKevin Wolf if (*total < 0) { 433797ac58cSKevin Wolf return *total; 434797ac58cSKevin Wolf } 435797ac58cSKevin Wolf return 1; 436797ac58cSKevin Wolf } 437797ac58cSKevin Wolf 438797ac58cSKevin Wolf typedef struct { 439797ac58cSKevin Wolf BlockDriverState *bs; 440797ac58cSKevin Wolf int64_t offset; 441797ac58cSKevin Wolf int count; 442797ac58cSKevin Wolf int *total; 443797ac58cSKevin Wolf int ret; 444797ac58cSKevin Wolf bool done; 445797ac58cSKevin Wolf } CoWriteZeroes; 446797ac58cSKevin Wolf 447797ac58cSKevin Wolf static void coroutine_fn co_write_zeroes_entry(void *opaque) 448797ac58cSKevin Wolf { 449797ac58cSKevin Wolf CoWriteZeroes *data = opaque; 450797ac58cSKevin Wolf 451797ac58cSKevin Wolf data->ret = bdrv_co_write_zeroes(data->bs, data->offset / BDRV_SECTOR_SIZE, 452aa7bfbffSPeter Lieven data->count / BDRV_SECTOR_SIZE, 0); 453797ac58cSKevin Wolf data->done = true; 454797ac58cSKevin Wolf if (data->ret < 0) { 455797ac58cSKevin Wolf *data->total = data->ret; 456797ac58cSKevin Wolf return; 457797ac58cSKevin Wolf } 458797ac58cSKevin Wolf 459797ac58cSKevin Wolf *data->total = data->count; 460797ac58cSKevin Wolf } 461797ac58cSKevin Wolf 462797ac58cSKevin Wolf static int do_co_write_zeroes(BlockDriverState *bs, int64_t offset, int count, 463797ac58cSKevin Wolf int *total) 464797ac58cSKevin Wolf { 465797ac58cSKevin Wolf Coroutine *co; 466797ac58cSKevin Wolf CoWriteZeroes data = { 467797ac58cSKevin Wolf .bs = bs, 468797ac58cSKevin Wolf .offset = offset, 469797ac58cSKevin Wolf .count = count, 470797ac58cSKevin Wolf .total = total, 471797ac58cSKevin Wolf .done = false, 472797ac58cSKevin Wolf }; 473797ac58cSKevin Wolf 474797ac58cSKevin Wolf co = qemu_coroutine_create(co_write_zeroes_entry); 475797ac58cSKevin Wolf qemu_coroutine_enter(co, &data); 476797ac58cSKevin Wolf while (!data.done) { 477b47ec2c4SPaolo Bonzini aio_poll(bdrv_get_aio_context(bs), true); 478797ac58cSKevin Wolf } 479797ac58cSKevin Wolf if (data.ret < 0) { 480797ac58cSKevin Wolf return data.ret; 481797ac58cSKevin Wolf } else { 482797ac58cSKevin Wolf return 1; 483797ac58cSKevin Wolf } 484797ac58cSKevin Wolf } 485797ac58cSKevin Wolf 486797ac58cSKevin Wolf static int do_write_compressed(BlockDriverState *bs, char *buf, int64_t offset, 487797ac58cSKevin Wolf int count, int *total) 488797ac58cSKevin Wolf { 489797ac58cSKevin Wolf int ret; 490797ac58cSKevin Wolf 491797ac58cSKevin Wolf ret = bdrv_write_compressed(bs, offset >> 9, (uint8_t *)buf, count >> 9); 492797ac58cSKevin Wolf if (ret < 0) { 493797ac58cSKevin Wolf return ret; 494797ac58cSKevin Wolf } 495797ac58cSKevin Wolf *total = count; 496797ac58cSKevin Wolf return 1; 497797ac58cSKevin Wolf } 498797ac58cSKevin Wolf 499797ac58cSKevin Wolf static int do_load_vmstate(BlockDriverState *bs, char *buf, int64_t offset, 500797ac58cSKevin Wolf int count, int *total) 501797ac58cSKevin Wolf { 502797ac58cSKevin Wolf *total = bdrv_load_vmstate(bs, (uint8_t *)buf, offset, count); 503797ac58cSKevin Wolf if (*total < 0) { 504797ac58cSKevin Wolf return *total; 505797ac58cSKevin Wolf } 506797ac58cSKevin Wolf return 1; 507797ac58cSKevin Wolf } 508797ac58cSKevin Wolf 509797ac58cSKevin Wolf static int do_save_vmstate(BlockDriverState *bs, char *buf, int64_t offset, 510797ac58cSKevin Wolf int count, int *total) 511797ac58cSKevin Wolf { 512797ac58cSKevin Wolf *total = bdrv_save_vmstate(bs, (uint8_t *)buf, offset, count); 513797ac58cSKevin Wolf if (*total < 0) { 514797ac58cSKevin Wolf return *total; 515797ac58cSKevin Wolf } 516797ac58cSKevin Wolf return 1; 517797ac58cSKevin Wolf } 518797ac58cSKevin Wolf 519797ac58cSKevin Wolf #define NOT_DONE 0x7fffffff 520797ac58cSKevin Wolf static void aio_rw_done(void *opaque, int ret) 521797ac58cSKevin Wolf { 522797ac58cSKevin Wolf *(int *)opaque = ret; 523797ac58cSKevin Wolf } 524797ac58cSKevin Wolf 525797ac58cSKevin Wolf static int do_aio_readv(BlockDriverState *bs, QEMUIOVector *qiov, 526797ac58cSKevin Wolf int64_t offset, int *total) 527797ac58cSKevin Wolf { 528797ac58cSKevin Wolf int async_ret = NOT_DONE; 529797ac58cSKevin Wolf 530797ac58cSKevin Wolf bdrv_aio_readv(bs, offset >> 9, qiov, qiov->size >> 9, 531797ac58cSKevin Wolf aio_rw_done, &async_ret); 532797ac58cSKevin Wolf while (async_ret == NOT_DONE) { 533797ac58cSKevin Wolf main_loop_wait(false); 534797ac58cSKevin Wolf } 535797ac58cSKevin Wolf 536797ac58cSKevin Wolf *total = qiov->size; 537797ac58cSKevin Wolf return async_ret < 0 ? async_ret : 1; 538797ac58cSKevin Wolf } 539797ac58cSKevin Wolf 540797ac58cSKevin Wolf static int do_aio_writev(BlockDriverState *bs, QEMUIOVector *qiov, 541797ac58cSKevin Wolf int64_t offset, int *total) 542797ac58cSKevin Wolf { 543797ac58cSKevin Wolf int async_ret = NOT_DONE; 544797ac58cSKevin Wolf 545797ac58cSKevin Wolf bdrv_aio_writev(bs, offset >> 9, qiov, qiov->size >> 9, 546797ac58cSKevin Wolf aio_rw_done, &async_ret); 547797ac58cSKevin Wolf while (async_ret == NOT_DONE) { 548797ac58cSKevin Wolf main_loop_wait(false); 549797ac58cSKevin Wolf } 550797ac58cSKevin Wolf 551797ac58cSKevin Wolf *total = qiov->size; 552797ac58cSKevin Wolf return async_ret < 0 ? async_ret : 1; 553797ac58cSKevin Wolf } 554797ac58cSKevin Wolf 555797ac58cSKevin Wolf struct multiwrite_async_ret { 556797ac58cSKevin Wolf int num_done; 557797ac58cSKevin Wolf int error; 558797ac58cSKevin Wolf }; 559797ac58cSKevin Wolf 560797ac58cSKevin Wolf static void multiwrite_cb(void *opaque, int ret) 561797ac58cSKevin Wolf { 562797ac58cSKevin Wolf struct multiwrite_async_ret *async_ret = opaque; 563797ac58cSKevin Wolf 564797ac58cSKevin Wolf async_ret->num_done++; 565797ac58cSKevin Wolf if (ret < 0) { 566797ac58cSKevin Wolf async_ret->error = ret; 567797ac58cSKevin Wolf } 568797ac58cSKevin Wolf } 569797ac58cSKevin Wolf 570797ac58cSKevin Wolf static int do_aio_multiwrite(BlockDriverState *bs, BlockRequest* reqs, 571797ac58cSKevin Wolf int num_reqs, int *total) 572797ac58cSKevin Wolf { 573797ac58cSKevin Wolf int i, ret; 574797ac58cSKevin Wolf struct multiwrite_async_ret async_ret = { 575797ac58cSKevin Wolf .num_done = 0, 576797ac58cSKevin Wolf .error = 0, 577797ac58cSKevin Wolf }; 578797ac58cSKevin Wolf 579797ac58cSKevin Wolf *total = 0; 580797ac58cSKevin Wolf for (i = 0; i < num_reqs; i++) { 581797ac58cSKevin Wolf reqs[i].cb = multiwrite_cb; 582797ac58cSKevin Wolf reqs[i].opaque = &async_ret; 583797ac58cSKevin Wolf *total += reqs[i].qiov->size; 584797ac58cSKevin Wolf } 585797ac58cSKevin Wolf 586797ac58cSKevin Wolf ret = bdrv_aio_multiwrite(bs, reqs, num_reqs); 587797ac58cSKevin Wolf if (ret < 0) { 588797ac58cSKevin Wolf return ret; 589797ac58cSKevin Wolf } 590797ac58cSKevin Wolf 591797ac58cSKevin Wolf while (async_ret.num_done < num_reqs) { 592797ac58cSKevin Wolf main_loop_wait(false); 593797ac58cSKevin Wolf } 594797ac58cSKevin Wolf 595797ac58cSKevin Wolf return async_ret.error < 0 ? async_ret.error : 1; 596797ac58cSKevin Wolf } 597797ac58cSKevin Wolf 598797ac58cSKevin Wolf static void read_help(void) 599797ac58cSKevin Wolf { 600797ac58cSKevin Wolf printf( 601797ac58cSKevin Wolf "\n" 602797ac58cSKevin Wolf " reads a range of bytes from the given offset\n" 603797ac58cSKevin Wolf "\n" 604797ac58cSKevin Wolf " Example:\n" 605797ac58cSKevin Wolf " 'read -v 512 1k' - dumps 1 kilobyte read from 512 bytes into the file\n" 606797ac58cSKevin Wolf "\n" 607797ac58cSKevin Wolf " Reads a segment of the currently open file, optionally dumping it to the\n" 608797ac58cSKevin Wolf " standard output stream (with -v option) for subsequent inspection.\n" 609797ac58cSKevin Wolf " -b, -- read from the VM state rather than the virtual disk\n" 610797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 611797ac58cSKevin Wolf " -l, -- length for pattern verification (only with -P)\n" 612797ac58cSKevin Wolf " -p, -- use bdrv_pread to read the file\n" 613797ac58cSKevin Wolf " -P, -- use a pattern to verify read data\n" 614797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 615797ac58cSKevin Wolf " -s, -- start offset for pattern verification (only with -P)\n" 616797ac58cSKevin Wolf " -v, -- dump buffer to standard output\n" 617797ac58cSKevin Wolf "\n"); 618797ac58cSKevin Wolf } 619797ac58cSKevin Wolf 620797ac58cSKevin Wolf static int read_f(BlockDriverState *bs, int argc, char **argv); 621797ac58cSKevin Wolf 622797ac58cSKevin Wolf static const cmdinfo_t read_cmd = { 623797ac58cSKevin Wolf .name = "read", 624797ac58cSKevin Wolf .altname = "r", 625797ac58cSKevin Wolf .cfunc = read_f, 626797ac58cSKevin Wolf .argmin = 2, 627797ac58cSKevin Wolf .argmax = -1, 628797ac58cSKevin Wolf .args = "[-abCpqv] [-P pattern [-s off] [-l len]] off len", 629797ac58cSKevin Wolf .oneline = "reads a number of bytes at a specified offset", 630797ac58cSKevin Wolf .help = read_help, 631797ac58cSKevin Wolf }; 632797ac58cSKevin Wolf 633797ac58cSKevin Wolf static int read_f(BlockDriverState *bs, int argc, char **argv) 634797ac58cSKevin Wolf { 635797ac58cSKevin Wolf struct timeval t1, t2; 636797ac58cSKevin Wolf int Cflag = 0, pflag = 0, qflag = 0, vflag = 0; 637797ac58cSKevin Wolf int Pflag = 0, sflag = 0, lflag = 0, bflag = 0; 638797ac58cSKevin Wolf int c, cnt; 639797ac58cSKevin Wolf char *buf; 640797ac58cSKevin Wolf int64_t offset; 641797ac58cSKevin Wolf int count; 642797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 643797ac58cSKevin Wolf int total = 0; 644797ac58cSKevin Wolf int pattern = 0, pattern_offset = 0, pattern_count = 0; 645797ac58cSKevin Wolf 646797ac58cSKevin Wolf while ((c = getopt(argc, argv, "bCl:pP:qs:v")) != EOF) { 647797ac58cSKevin Wolf switch (c) { 648797ac58cSKevin Wolf case 'b': 649797ac58cSKevin Wolf bflag = 1; 650797ac58cSKevin Wolf break; 651797ac58cSKevin Wolf case 'C': 652797ac58cSKevin Wolf Cflag = 1; 653797ac58cSKevin Wolf break; 654797ac58cSKevin Wolf case 'l': 655797ac58cSKevin Wolf lflag = 1; 656797ac58cSKevin Wolf pattern_count = cvtnum(optarg); 657797ac58cSKevin Wolf if (pattern_count < 0) { 658797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", optarg); 659797ac58cSKevin Wolf return 0; 660797ac58cSKevin Wolf } 661797ac58cSKevin Wolf break; 662797ac58cSKevin Wolf case 'p': 663797ac58cSKevin Wolf pflag = 1; 664797ac58cSKevin Wolf break; 665797ac58cSKevin Wolf case 'P': 666797ac58cSKevin Wolf Pflag = 1; 667797ac58cSKevin Wolf pattern = parse_pattern(optarg); 668797ac58cSKevin Wolf if (pattern < 0) { 669797ac58cSKevin Wolf return 0; 670797ac58cSKevin Wolf } 671797ac58cSKevin Wolf break; 672797ac58cSKevin Wolf case 'q': 673797ac58cSKevin Wolf qflag = 1; 674797ac58cSKevin Wolf break; 675797ac58cSKevin Wolf case 's': 676797ac58cSKevin Wolf sflag = 1; 677797ac58cSKevin Wolf pattern_offset = cvtnum(optarg); 678797ac58cSKevin Wolf if (pattern_offset < 0) { 679797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", optarg); 680797ac58cSKevin Wolf return 0; 681797ac58cSKevin Wolf } 682797ac58cSKevin Wolf break; 683797ac58cSKevin Wolf case 'v': 684797ac58cSKevin Wolf vflag = 1; 685797ac58cSKevin Wolf break; 686797ac58cSKevin Wolf default: 687c2cdf5c5SKevin Wolf return qemuio_command_usage(&read_cmd); 688797ac58cSKevin Wolf } 689797ac58cSKevin Wolf } 690797ac58cSKevin Wolf 691797ac58cSKevin Wolf if (optind != argc - 2) { 692c2cdf5c5SKevin Wolf return qemuio_command_usage(&read_cmd); 693797ac58cSKevin Wolf } 694797ac58cSKevin Wolf 695797ac58cSKevin Wolf if (bflag && pflag) { 696797ac58cSKevin Wolf printf("-b and -p cannot be specified at the same time\n"); 697797ac58cSKevin Wolf return 0; 698797ac58cSKevin Wolf } 699797ac58cSKevin Wolf 700797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 701797ac58cSKevin Wolf if (offset < 0) { 702797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", argv[optind]); 703797ac58cSKevin Wolf return 0; 704797ac58cSKevin Wolf } 705797ac58cSKevin Wolf 706797ac58cSKevin Wolf optind++; 707797ac58cSKevin Wolf count = cvtnum(argv[optind]); 708797ac58cSKevin Wolf if (count < 0) { 709797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", argv[optind]); 710797ac58cSKevin Wolf return 0; 711797ac58cSKevin Wolf } 712797ac58cSKevin Wolf 713797ac58cSKevin Wolf if (!Pflag && (lflag || sflag)) { 714c2cdf5c5SKevin Wolf return qemuio_command_usage(&read_cmd); 715797ac58cSKevin Wolf } 716797ac58cSKevin Wolf 717797ac58cSKevin Wolf if (!lflag) { 718797ac58cSKevin Wolf pattern_count = count - pattern_offset; 719797ac58cSKevin Wolf } 720797ac58cSKevin Wolf 721797ac58cSKevin Wolf if ((pattern_count < 0) || (pattern_count + pattern_offset > count)) { 722797ac58cSKevin Wolf printf("pattern verification range exceeds end of read data\n"); 723797ac58cSKevin Wolf return 0; 724797ac58cSKevin Wolf } 725797ac58cSKevin Wolf 726797ac58cSKevin Wolf if (!pflag) { 727797ac58cSKevin Wolf if (offset & 0x1ff) { 728797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 729797ac58cSKevin Wolf offset); 730797ac58cSKevin Wolf return 0; 731797ac58cSKevin Wolf } 732797ac58cSKevin Wolf if (count & 0x1ff) { 733797ac58cSKevin Wolf printf("count %d is not sector aligned\n", 734797ac58cSKevin Wolf count); 735797ac58cSKevin Wolf return 0; 736797ac58cSKevin Wolf } 737797ac58cSKevin Wolf } 738797ac58cSKevin Wolf 739797ac58cSKevin Wolf buf = qemu_io_alloc(bs, count, 0xab); 740797ac58cSKevin Wolf 741797ac58cSKevin Wolf gettimeofday(&t1, NULL); 742797ac58cSKevin Wolf if (pflag) { 743797ac58cSKevin Wolf cnt = do_pread(bs, buf, offset, count, &total); 744797ac58cSKevin Wolf } else if (bflag) { 745797ac58cSKevin Wolf cnt = do_load_vmstate(bs, buf, offset, count, &total); 746797ac58cSKevin Wolf } else { 747797ac58cSKevin Wolf cnt = do_read(bs, buf, offset, count, &total); 748797ac58cSKevin Wolf } 749797ac58cSKevin Wolf gettimeofday(&t2, NULL); 750797ac58cSKevin Wolf 751797ac58cSKevin Wolf if (cnt < 0) { 752797ac58cSKevin Wolf printf("read failed: %s\n", strerror(-cnt)); 753797ac58cSKevin Wolf goto out; 754797ac58cSKevin Wolf } 755797ac58cSKevin Wolf 756797ac58cSKevin Wolf if (Pflag) { 757797ac58cSKevin Wolf void *cmp_buf = g_malloc(pattern_count); 758797ac58cSKevin Wolf memset(cmp_buf, pattern, pattern_count); 759797ac58cSKevin Wolf if (memcmp(buf + pattern_offset, cmp_buf, pattern_count)) { 760797ac58cSKevin Wolf printf("Pattern verification failed at offset %" 761797ac58cSKevin Wolf PRId64 ", %d bytes\n", 762797ac58cSKevin Wolf offset + pattern_offset, pattern_count); 763797ac58cSKevin Wolf } 764797ac58cSKevin Wolf g_free(cmp_buf); 765797ac58cSKevin Wolf } 766797ac58cSKevin Wolf 767797ac58cSKevin Wolf if (qflag) { 768797ac58cSKevin Wolf goto out; 769797ac58cSKevin Wolf } 770797ac58cSKevin Wolf 771797ac58cSKevin Wolf if (vflag) { 772797ac58cSKevin Wolf dump_buffer(buf, offset, count); 773797ac58cSKevin Wolf } 774797ac58cSKevin Wolf 775797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 776797ac58cSKevin Wolf t2 = tsub(t2, t1); 777797ac58cSKevin Wolf print_report("read", &t2, offset, count, total, cnt, Cflag); 778797ac58cSKevin Wolf 779797ac58cSKevin Wolf out: 780797ac58cSKevin Wolf qemu_io_free(buf); 781797ac58cSKevin Wolf 782797ac58cSKevin Wolf return 0; 783797ac58cSKevin Wolf } 784797ac58cSKevin Wolf 785797ac58cSKevin Wolf static void readv_help(void) 786797ac58cSKevin Wolf { 787797ac58cSKevin Wolf printf( 788797ac58cSKevin Wolf "\n" 789797ac58cSKevin Wolf " reads a range of bytes from the given offset into multiple buffers\n" 790797ac58cSKevin Wolf "\n" 791797ac58cSKevin Wolf " Example:\n" 792797ac58cSKevin Wolf " 'readv -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n" 793797ac58cSKevin Wolf "\n" 794797ac58cSKevin Wolf " Reads a segment of the currently open file, optionally dumping it to the\n" 795797ac58cSKevin Wolf " standard output stream (with -v option) for subsequent inspection.\n" 796797ac58cSKevin Wolf " Uses multiple iovec buffers if more than one byte range is specified.\n" 797797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 798797ac58cSKevin Wolf " -P, -- use a pattern to verify read data\n" 799797ac58cSKevin Wolf " -v, -- dump buffer to standard output\n" 800797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 801797ac58cSKevin Wolf "\n"); 802797ac58cSKevin Wolf } 803797ac58cSKevin Wolf 804797ac58cSKevin Wolf static int readv_f(BlockDriverState *bs, int argc, char **argv); 805797ac58cSKevin Wolf 806797ac58cSKevin Wolf static const cmdinfo_t readv_cmd = { 807797ac58cSKevin Wolf .name = "readv", 808797ac58cSKevin Wolf .cfunc = readv_f, 809797ac58cSKevin Wolf .argmin = 2, 810797ac58cSKevin Wolf .argmax = -1, 811797ac58cSKevin Wolf .args = "[-Cqv] [-P pattern ] off len [len..]", 812797ac58cSKevin Wolf .oneline = "reads a number of bytes at a specified offset", 813797ac58cSKevin Wolf .help = readv_help, 814797ac58cSKevin Wolf }; 815797ac58cSKevin Wolf 816797ac58cSKevin Wolf static int readv_f(BlockDriverState *bs, int argc, char **argv) 817797ac58cSKevin Wolf { 818797ac58cSKevin Wolf struct timeval t1, t2; 819797ac58cSKevin Wolf int Cflag = 0, qflag = 0, vflag = 0; 820797ac58cSKevin Wolf int c, cnt; 821797ac58cSKevin Wolf char *buf; 822797ac58cSKevin Wolf int64_t offset; 823797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 824797ac58cSKevin Wolf int total = 0; 825797ac58cSKevin Wolf int nr_iov; 826797ac58cSKevin Wolf QEMUIOVector qiov; 827797ac58cSKevin Wolf int pattern = 0; 828797ac58cSKevin Wolf int Pflag = 0; 829797ac58cSKevin Wolf 830797ac58cSKevin Wolf while ((c = getopt(argc, argv, "CP:qv")) != EOF) { 831797ac58cSKevin Wolf switch (c) { 832797ac58cSKevin Wolf case 'C': 833797ac58cSKevin Wolf Cflag = 1; 834797ac58cSKevin Wolf break; 835797ac58cSKevin Wolf case 'P': 836797ac58cSKevin Wolf Pflag = 1; 837797ac58cSKevin Wolf pattern = parse_pattern(optarg); 838797ac58cSKevin Wolf if (pattern < 0) { 839797ac58cSKevin Wolf return 0; 840797ac58cSKevin Wolf } 841797ac58cSKevin Wolf break; 842797ac58cSKevin Wolf case 'q': 843797ac58cSKevin Wolf qflag = 1; 844797ac58cSKevin Wolf break; 845797ac58cSKevin Wolf case 'v': 846797ac58cSKevin Wolf vflag = 1; 847797ac58cSKevin Wolf break; 848797ac58cSKevin Wolf default: 849c2cdf5c5SKevin Wolf return qemuio_command_usage(&readv_cmd); 850797ac58cSKevin Wolf } 851797ac58cSKevin Wolf } 852797ac58cSKevin Wolf 853797ac58cSKevin Wolf if (optind > argc - 2) { 854c2cdf5c5SKevin Wolf return qemuio_command_usage(&readv_cmd); 855797ac58cSKevin Wolf } 856797ac58cSKevin Wolf 857797ac58cSKevin Wolf 858797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 859797ac58cSKevin Wolf if (offset < 0) { 860797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", argv[optind]); 861797ac58cSKevin Wolf return 0; 862797ac58cSKevin Wolf } 863797ac58cSKevin Wolf optind++; 864797ac58cSKevin Wolf 865797ac58cSKevin Wolf if (offset & 0x1ff) { 866797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 867797ac58cSKevin Wolf offset); 868797ac58cSKevin Wolf return 0; 869797ac58cSKevin Wolf } 870797ac58cSKevin Wolf 871797ac58cSKevin Wolf nr_iov = argc - optind; 872797ac58cSKevin Wolf buf = create_iovec(bs, &qiov, &argv[optind], nr_iov, 0xab); 873797ac58cSKevin Wolf if (buf == NULL) { 874797ac58cSKevin Wolf return 0; 875797ac58cSKevin Wolf } 876797ac58cSKevin Wolf 877797ac58cSKevin Wolf gettimeofday(&t1, NULL); 878797ac58cSKevin Wolf cnt = do_aio_readv(bs, &qiov, offset, &total); 879797ac58cSKevin Wolf gettimeofday(&t2, NULL); 880797ac58cSKevin Wolf 881797ac58cSKevin Wolf if (cnt < 0) { 882797ac58cSKevin Wolf printf("readv failed: %s\n", strerror(-cnt)); 883797ac58cSKevin Wolf goto out; 884797ac58cSKevin Wolf } 885797ac58cSKevin Wolf 886797ac58cSKevin Wolf if (Pflag) { 887797ac58cSKevin Wolf void *cmp_buf = g_malloc(qiov.size); 888797ac58cSKevin Wolf memset(cmp_buf, pattern, qiov.size); 889797ac58cSKevin Wolf if (memcmp(buf, cmp_buf, qiov.size)) { 890797ac58cSKevin Wolf printf("Pattern verification failed at offset %" 891797ac58cSKevin Wolf PRId64 ", %zd bytes\n", offset, qiov.size); 892797ac58cSKevin Wolf } 893797ac58cSKevin Wolf g_free(cmp_buf); 894797ac58cSKevin Wolf } 895797ac58cSKevin Wolf 896797ac58cSKevin Wolf if (qflag) { 897797ac58cSKevin Wolf goto out; 898797ac58cSKevin Wolf } 899797ac58cSKevin Wolf 900797ac58cSKevin Wolf if (vflag) { 901797ac58cSKevin Wolf dump_buffer(buf, offset, qiov.size); 902797ac58cSKevin Wolf } 903797ac58cSKevin Wolf 904797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 905797ac58cSKevin Wolf t2 = tsub(t2, t1); 906797ac58cSKevin Wolf print_report("read", &t2, offset, qiov.size, total, cnt, Cflag); 907797ac58cSKevin Wolf 908797ac58cSKevin Wolf out: 909797ac58cSKevin Wolf qemu_iovec_destroy(&qiov); 910797ac58cSKevin Wolf qemu_io_free(buf); 911797ac58cSKevin Wolf return 0; 912797ac58cSKevin Wolf } 913797ac58cSKevin Wolf 914797ac58cSKevin Wolf static void write_help(void) 915797ac58cSKevin Wolf { 916797ac58cSKevin Wolf printf( 917797ac58cSKevin Wolf "\n" 918797ac58cSKevin Wolf " writes a range of bytes from the given offset\n" 919797ac58cSKevin Wolf "\n" 920797ac58cSKevin Wolf " Example:\n" 921797ac58cSKevin Wolf " 'write 512 1k' - writes 1 kilobyte at 512 bytes into the open file\n" 922797ac58cSKevin Wolf "\n" 923797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 924797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd).\n" 925797ac58cSKevin Wolf " -b, -- write to the VM state rather than the virtual disk\n" 926797ac58cSKevin Wolf " -c, -- write compressed data with bdrv_write_compressed\n" 927797ac58cSKevin Wolf " -p, -- use bdrv_pwrite to write the file\n" 928797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 929797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 930797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 931797ac58cSKevin Wolf " -z, -- write zeroes using bdrv_co_write_zeroes\n" 932797ac58cSKevin Wolf "\n"); 933797ac58cSKevin Wolf } 934797ac58cSKevin Wolf 935797ac58cSKevin Wolf static int write_f(BlockDriverState *bs, int argc, char **argv); 936797ac58cSKevin Wolf 937797ac58cSKevin Wolf static const cmdinfo_t write_cmd = { 938797ac58cSKevin Wolf .name = "write", 939797ac58cSKevin Wolf .altname = "w", 940797ac58cSKevin Wolf .cfunc = write_f, 941797ac58cSKevin Wolf .argmin = 2, 942797ac58cSKevin Wolf .argmax = -1, 943797ac58cSKevin Wolf .args = "[-bcCpqz] [-P pattern ] off len", 944797ac58cSKevin Wolf .oneline = "writes a number of bytes at a specified offset", 945797ac58cSKevin Wolf .help = write_help, 946797ac58cSKevin Wolf }; 947797ac58cSKevin Wolf 948797ac58cSKevin Wolf static int write_f(BlockDriverState *bs, int argc, char **argv) 949797ac58cSKevin Wolf { 950797ac58cSKevin Wolf struct timeval t1, t2; 951797ac58cSKevin Wolf int Cflag = 0, pflag = 0, qflag = 0, bflag = 0, Pflag = 0, zflag = 0; 952797ac58cSKevin Wolf int cflag = 0; 953797ac58cSKevin Wolf int c, cnt; 954797ac58cSKevin Wolf char *buf = NULL; 955797ac58cSKevin Wolf int64_t offset; 956797ac58cSKevin Wolf int count; 957797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 958797ac58cSKevin Wolf int total = 0; 959797ac58cSKevin Wolf int pattern = 0xcd; 960797ac58cSKevin Wolf 961797ac58cSKevin Wolf while ((c = getopt(argc, argv, "bcCpP:qz")) != EOF) { 962797ac58cSKevin Wolf switch (c) { 963797ac58cSKevin Wolf case 'b': 964797ac58cSKevin Wolf bflag = 1; 965797ac58cSKevin Wolf break; 966797ac58cSKevin Wolf case 'c': 967797ac58cSKevin Wolf cflag = 1; 968797ac58cSKevin Wolf break; 969797ac58cSKevin Wolf case 'C': 970797ac58cSKevin Wolf Cflag = 1; 971797ac58cSKevin Wolf break; 972797ac58cSKevin Wolf case 'p': 973797ac58cSKevin Wolf pflag = 1; 974797ac58cSKevin Wolf break; 975797ac58cSKevin Wolf case 'P': 976797ac58cSKevin Wolf Pflag = 1; 977797ac58cSKevin Wolf pattern = parse_pattern(optarg); 978797ac58cSKevin Wolf if (pattern < 0) { 979797ac58cSKevin Wolf return 0; 980797ac58cSKevin Wolf } 981797ac58cSKevin Wolf break; 982797ac58cSKevin Wolf case 'q': 983797ac58cSKevin Wolf qflag = 1; 984797ac58cSKevin Wolf break; 985797ac58cSKevin Wolf case 'z': 986797ac58cSKevin Wolf zflag = 1; 987797ac58cSKevin Wolf break; 988797ac58cSKevin Wolf default: 989c2cdf5c5SKevin Wolf return qemuio_command_usage(&write_cmd); 990797ac58cSKevin Wolf } 991797ac58cSKevin Wolf } 992797ac58cSKevin Wolf 993797ac58cSKevin Wolf if (optind != argc - 2) { 994c2cdf5c5SKevin Wolf return qemuio_command_usage(&write_cmd); 995797ac58cSKevin Wolf } 996797ac58cSKevin Wolf 997797ac58cSKevin Wolf if (bflag + pflag + zflag > 1) { 998797ac58cSKevin Wolf printf("-b, -p, or -z cannot be specified at the same time\n"); 999797ac58cSKevin Wolf return 0; 1000797ac58cSKevin Wolf } 1001797ac58cSKevin Wolf 1002797ac58cSKevin Wolf if (zflag && Pflag) { 1003797ac58cSKevin Wolf printf("-z and -P cannot be specified at the same time\n"); 1004797ac58cSKevin Wolf return 0; 1005797ac58cSKevin Wolf } 1006797ac58cSKevin Wolf 1007797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1008797ac58cSKevin Wolf if (offset < 0) { 1009797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", argv[optind]); 1010797ac58cSKevin Wolf return 0; 1011797ac58cSKevin Wolf } 1012797ac58cSKevin Wolf 1013797ac58cSKevin Wolf optind++; 1014797ac58cSKevin Wolf count = cvtnum(argv[optind]); 1015797ac58cSKevin Wolf if (count < 0) { 1016797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", argv[optind]); 1017797ac58cSKevin Wolf return 0; 1018797ac58cSKevin Wolf } 1019797ac58cSKevin Wolf 1020797ac58cSKevin Wolf if (!pflag) { 1021797ac58cSKevin Wolf if (offset & 0x1ff) { 1022797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 1023797ac58cSKevin Wolf offset); 1024797ac58cSKevin Wolf return 0; 1025797ac58cSKevin Wolf } 1026797ac58cSKevin Wolf 1027797ac58cSKevin Wolf if (count & 0x1ff) { 1028797ac58cSKevin Wolf printf("count %d is not sector aligned\n", 1029797ac58cSKevin Wolf count); 1030797ac58cSKevin Wolf return 0; 1031797ac58cSKevin Wolf } 1032797ac58cSKevin Wolf } 1033797ac58cSKevin Wolf 1034797ac58cSKevin Wolf if (!zflag) { 1035797ac58cSKevin Wolf buf = qemu_io_alloc(bs, count, pattern); 1036797ac58cSKevin Wolf } 1037797ac58cSKevin Wolf 1038797ac58cSKevin Wolf gettimeofday(&t1, NULL); 1039797ac58cSKevin Wolf if (pflag) { 1040797ac58cSKevin Wolf cnt = do_pwrite(bs, buf, offset, count, &total); 1041797ac58cSKevin Wolf } else if (bflag) { 1042797ac58cSKevin Wolf cnt = do_save_vmstate(bs, buf, offset, count, &total); 1043797ac58cSKevin Wolf } else if (zflag) { 1044797ac58cSKevin Wolf cnt = do_co_write_zeroes(bs, offset, count, &total); 1045797ac58cSKevin Wolf } else if (cflag) { 1046797ac58cSKevin Wolf cnt = do_write_compressed(bs, buf, offset, count, &total); 1047797ac58cSKevin Wolf } else { 1048797ac58cSKevin Wolf cnt = do_write(bs, buf, offset, count, &total); 1049797ac58cSKevin Wolf } 1050797ac58cSKevin Wolf gettimeofday(&t2, NULL); 1051797ac58cSKevin Wolf 1052797ac58cSKevin Wolf if (cnt < 0) { 1053797ac58cSKevin Wolf printf("write failed: %s\n", strerror(-cnt)); 1054797ac58cSKevin Wolf goto out; 1055797ac58cSKevin Wolf } 1056797ac58cSKevin Wolf 1057797ac58cSKevin Wolf if (qflag) { 1058797ac58cSKevin Wolf goto out; 1059797ac58cSKevin Wolf } 1060797ac58cSKevin Wolf 1061797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1062797ac58cSKevin Wolf t2 = tsub(t2, t1); 1063797ac58cSKevin Wolf print_report("wrote", &t2, offset, count, total, cnt, Cflag); 1064797ac58cSKevin Wolf 1065797ac58cSKevin Wolf out: 1066797ac58cSKevin Wolf if (!zflag) { 1067797ac58cSKevin Wolf qemu_io_free(buf); 1068797ac58cSKevin Wolf } 1069797ac58cSKevin Wolf 1070797ac58cSKevin Wolf return 0; 1071797ac58cSKevin Wolf } 1072797ac58cSKevin Wolf 1073797ac58cSKevin Wolf static void 1074797ac58cSKevin Wolf writev_help(void) 1075797ac58cSKevin Wolf { 1076797ac58cSKevin Wolf printf( 1077797ac58cSKevin Wolf "\n" 1078797ac58cSKevin Wolf " writes a range of bytes from the given offset source from multiple buffers\n" 1079797ac58cSKevin Wolf "\n" 1080797ac58cSKevin Wolf " Example:\n" 10816e6507c0SMaria Kustova " 'writev 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n" 1082797ac58cSKevin Wolf "\n" 1083797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 1084797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd).\n" 1085797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 1086797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1087797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1088797ac58cSKevin Wolf "\n"); 1089797ac58cSKevin Wolf } 1090797ac58cSKevin Wolf 1091797ac58cSKevin Wolf static int writev_f(BlockDriverState *bs, int argc, char **argv); 1092797ac58cSKevin Wolf 1093797ac58cSKevin Wolf static const cmdinfo_t writev_cmd = { 1094797ac58cSKevin Wolf .name = "writev", 1095797ac58cSKevin Wolf .cfunc = writev_f, 1096797ac58cSKevin Wolf .argmin = 2, 1097797ac58cSKevin Wolf .argmax = -1, 1098797ac58cSKevin Wolf .args = "[-Cq] [-P pattern ] off len [len..]", 1099797ac58cSKevin Wolf .oneline = "writes a number of bytes at a specified offset", 1100797ac58cSKevin Wolf .help = writev_help, 1101797ac58cSKevin Wolf }; 1102797ac58cSKevin Wolf 1103797ac58cSKevin Wolf static int writev_f(BlockDriverState *bs, int argc, char **argv) 1104797ac58cSKevin Wolf { 1105797ac58cSKevin Wolf struct timeval t1, t2; 1106797ac58cSKevin Wolf int Cflag = 0, qflag = 0; 1107797ac58cSKevin Wolf int c, cnt; 1108797ac58cSKevin Wolf char *buf; 1109797ac58cSKevin Wolf int64_t offset; 1110797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 1111797ac58cSKevin Wolf int total = 0; 1112797ac58cSKevin Wolf int nr_iov; 1113797ac58cSKevin Wolf int pattern = 0xcd; 1114797ac58cSKevin Wolf QEMUIOVector qiov; 1115797ac58cSKevin Wolf 1116797ac58cSKevin Wolf while ((c = getopt(argc, argv, "CqP:")) != EOF) { 1117797ac58cSKevin Wolf switch (c) { 1118797ac58cSKevin Wolf case 'C': 1119797ac58cSKevin Wolf Cflag = 1; 1120797ac58cSKevin Wolf break; 1121797ac58cSKevin Wolf case 'q': 1122797ac58cSKevin Wolf qflag = 1; 1123797ac58cSKevin Wolf break; 1124797ac58cSKevin Wolf case 'P': 1125797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1126797ac58cSKevin Wolf if (pattern < 0) { 1127797ac58cSKevin Wolf return 0; 1128797ac58cSKevin Wolf } 1129797ac58cSKevin Wolf break; 1130797ac58cSKevin Wolf default: 1131c2cdf5c5SKevin Wolf return qemuio_command_usage(&writev_cmd); 1132797ac58cSKevin Wolf } 1133797ac58cSKevin Wolf } 1134797ac58cSKevin Wolf 1135797ac58cSKevin Wolf if (optind > argc - 2) { 1136c2cdf5c5SKevin Wolf return qemuio_command_usage(&writev_cmd); 1137797ac58cSKevin Wolf } 1138797ac58cSKevin Wolf 1139797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1140797ac58cSKevin Wolf if (offset < 0) { 1141797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", argv[optind]); 1142797ac58cSKevin Wolf return 0; 1143797ac58cSKevin Wolf } 1144797ac58cSKevin Wolf optind++; 1145797ac58cSKevin Wolf 1146797ac58cSKevin Wolf if (offset & 0x1ff) { 1147797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 1148797ac58cSKevin Wolf offset); 1149797ac58cSKevin Wolf return 0; 1150797ac58cSKevin Wolf } 1151797ac58cSKevin Wolf 1152797ac58cSKevin Wolf nr_iov = argc - optind; 1153797ac58cSKevin Wolf buf = create_iovec(bs, &qiov, &argv[optind], nr_iov, pattern); 1154797ac58cSKevin Wolf if (buf == NULL) { 1155797ac58cSKevin Wolf return 0; 1156797ac58cSKevin Wolf } 1157797ac58cSKevin Wolf 1158797ac58cSKevin Wolf gettimeofday(&t1, NULL); 1159797ac58cSKevin Wolf cnt = do_aio_writev(bs, &qiov, offset, &total); 1160797ac58cSKevin Wolf gettimeofday(&t2, NULL); 1161797ac58cSKevin Wolf 1162797ac58cSKevin Wolf if (cnt < 0) { 1163797ac58cSKevin Wolf printf("writev failed: %s\n", strerror(-cnt)); 1164797ac58cSKevin Wolf goto out; 1165797ac58cSKevin Wolf } 1166797ac58cSKevin Wolf 1167797ac58cSKevin Wolf if (qflag) { 1168797ac58cSKevin Wolf goto out; 1169797ac58cSKevin Wolf } 1170797ac58cSKevin Wolf 1171797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1172797ac58cSKevin Wolf t2 = tsub(t2, t1); 1173797ac58cSKevin Wolf print_report("wrote", &t2, offset, qiov.size, total, cnt, Cflag); 1174797ac58cSKevin Wolf out: 1175797ac58cSKevin Wolf qemu_iovec_destroy(&qiov); 1176797ac58cSKevin Wolf qemu_io_free(buf); 1177797ac58cSKevin Wolf return 0; 1178797ac58cSKevin Wolf } 1179797ac58cSKevin Wolf 1180797ac58cSKevin Wolf static void multiwrite_help(void) 1181797ac58cSKevin Wolf { 1182797ac58cSKevin Wolf printf( 1183797ac58cSKevin Wolf "\n" 1184797ac58cSKevin Wolf " writes a range of bytes from the given offset source from multiple buffers,\n" 1185797ac58cSKevin Wolf " in a batch of requests that may be merged by qemu\n" 1186797ac58cSKevin Wolf "\n" 1187797ac58cSKevin Wolf " Example:\n" 1188797ac58cSKevin Wolf " 'multiwrite 512 1k 1k ; 4k 1k'\n" 1189797ac58cSKevin Wolf " writes 2 kB at 512 bytes and 1 kB at 4 kB into the open file\n" 1190797ac58cSKevin Wolf "\n" 1191797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 1192797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd). The pattern byte is increased\n" 1193797ac58cSKevin Wolf " by one for each request contained in the multiwrite command.\n" 1194797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 1195797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1196797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1197797ac58cSKevin Wolf "\n"); 1198797ac58cSKevin Wolf } 1199797ac58cSKevin Wolf 1200797ac58cSKevin Wolf static int multiwrite_f(BlockDriverState *bs, int argc, char **argv); 1201797ac58cSKevin Wolf 1202797ac58cSKevin Wolf static const cmdinfo_t multiwrite_cmd = { 1203797ac58cSKevin Wolf .name = "multiwrite", 1204797ac58cSKevin Wolf .cfunc = multiwrite_f, 1205797ac58cSKevin Wolf .argmin = 2, 1206797ac58cSKevin Wolf .argmax = -1, 1207797ac58cSKevin Wolf .args = "[-Cq] [-P pattern ] off len [len..] [; off len [len..]..]", 1208797ac58cSKevin Wolf .oneline = "issues multiple write requests at once", 1209797ac58cSKevin Wolf .help = multiwrite_help, 1210797ac58cSKevin Wolf }; 1211797ac58cSKevin Wolf 1212797ac58cSKevin Wolf static int multiwrite_f(BlockDriverState *bs, int argc, char **argv) 1213797ac58cSKevin Wolf { 1214797ac58cSKevin Wolf struct timeval t1, t2; 1215797ac58cSKevin Wolf int Cflag = 0, qflag = 0; 1216797ac58cSKevin Wolf int c, cnt; 1217797ac58cSKevin Wolf char **buf; 1218797ac58cSKevin Wolf int64_t offset, first_offset = 0; 1219797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 1220797ac58cSKevin Wolf int total = 0; 1221797ac58cSKevin Wolf int nr_iov; 1222797ac58cSKevin Wolf int nr_reqs; 1223797ac58cSKevin Wolf int pattern = 0xcd; 1224797ac58cSKevin Wolf QEMUIOVector *qiovs; 1225797ac58cSKevin Wolf int i; 1226797ac58cSKevin Wolf BlockRequest *reqs; 1227797ac58cSKevin Wolf 1228797ac58cSKevin Wolf while ((c = getopt(argc, argv, "CqP:")) != EOF) { 1229797ac58cSKevin Wolf switch (c) { 1230797ac58cSKevin Wolf case 'C': 1231797ac58cSKevin Wolf Cflag = 1; 1232797ac58cSKevin Wolf break; 1233797ac58cSKevin Wolf case 'q': 1234797ac58cSKevin Wolf qflag = 1; 1235797ac58cSKevin Wolf break; 1236797ac58cSKevin Wolf case 'P': 1237797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1238797ac58cSKevin Wolf if (pattern < 0) { 1239797ac58cSKevin Wolf return 0; 1240797ac58cSKevin Wolf } 1241797ac58cSKevin Wolf break; 1242797ac58cSKevin Wolf default: 1243c2cdf5c5SKevin Wolf return qemuio_command_usage(&writev_cmd); 1244797ac58cSKevin Wolf } 1245797ac58cSKevin Wolf } 1246797ac58cSKevin Wolf 1247797ac58cSKevin Wolf if (optind > argc - 2) { 1248c2cdf5c5SKevin Wolf return qemuio_command_usage(&writev_cmd); 1249797ac58cSKevin Wolf } 1250797ac58cSKevin Wolf 1251797ac58cSKevin Wolf nr_reqs = 1; 1252797ac58cSKevin Wolf for (i = optind; i < argc; i++) { 1253797ac58cSKevin Wolf if (!strcmp(argv[i], ";")) { 1254797ac58cSKevin Wolf nr_reqs++; 1255797ac58cSKevin Wolf } 1256797ac58cSKevin Wolf } 1257797ac58cSKevin Wolf 125802c4f26bSMarkus Armbruster reqs = g_new0(BlockRequest, nr_reqs); 125902c4f26bSMarkus Armbruster buf = g_new0(char *, nr_reqs); 126002c4f26bSMarkus Armbruster qiovs = g_new(QEMUIOVector, nr_reqs); 1261797ac58cSKevin Wolf 1262797ac58cSKevin Wolf for (i = 0; i < nr_reqs && optind < argc; i++) { 1263797ac58cSKevin Wolf int j; 1264797ac58cSKevin Wolf 1265797ac58cSKevin Wolf /* Read the offset of the request */ 1266797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1267797ac58cSKevin Wolf if (offset < 0) { 1268797ac58cSKevin Wolf printf("non-numeric offset argument -- %s\n", argv[optind]); 1269797ac58cSKevin Wolf goto out; 1270797ac58cSKevin Wolf } 1271797ac58cSKevin Wolf optind++; 1272797ac58cSKevin Wolf 1273797ac58cSKevin Wolf if (offset & 0x1ff) { 1274797ac58cSKevin Wolf printf("offset %lld is not sector aligned\n", 1275797ac58cSKevin Wolf (long long)offset); 1276797ac58cSKevin Wolf goto out; 1277797ac58cSKevin Wolf } 1278797ac58cSKevin Wolf 1279797ac58cSKevin Wolf if (i == 0) { 1280797ac58cSKevin Wolf first_offset = offset; 1281797ac58cSKevin Wolf } 1282797ac58cSKevin Wolf 1283797ac58cSKevin Wolf /* Read lengths for qiov entries */ 1284797ac58cSKevin Wolf for (j = optind; j < argc; j++) { 1285797ac58cSKevin Wolf if (!strcmp(argv[j], ";")) { 1286797ac58cSKevin Wolf break; 1287797ac58cSKevin Wolf } 1288797ac58cSKevin Wolf } 1289797ac58cSKevin Wolf 1290797ac58cSKevin Wolf nr_iov = j - optind; 1291797ac58cSKevin Wolf 1292797ac58cSKevin Wolf /* Build request */ 1293797ac58cSKevin Wolf buf[i] = create_iovec(bs, &qiovs[i], &argv[optind], nr_iov, pattern); 1294797ac58cSKevin Wolf if (buf[i] == NULL) { 1295797ac58cSKevin Wolf goto out; 1296797ac58cSKevin Wolf } 1297797ac58cSKevin Wolf 1298797ac58cSKevin Wolf reqs[i].qiov = &qiovs[i]; 1299797ac58cSKevin Wolf reqs[i].sector = offset >> 9; 1300797ac58cSKevin Wolf reqs[i].nb_sectors = reqs[i].qiov->size >> 9; 1301797ac58cSKevin Wolf 1302797ac58cSKevin Wolf optind = j + 1; 1303797ac58cSKevin Wolf 1304797ac58cSKevin Wolf pattern++; 1305797ac58cSKevin Wolf } 1306797ac58cSKevin Wolf 1307797ac58cSKevin Wolf /* If there were empty requests at the end, ignore them */ 1308797ac58cSKevin Wolf nr_reqs = i; 1309797ac58cSKevin Wolf 1310797ac58cSKevin Wolf gettimeofday(&t1, NULL); 1311797ac58cSKevin Wolf cnt = do_aio_multiwrite(bs, reqs, nr_reqs, &total); 1312797ac58cSKevin Wolf gettimeofday(&t2, NULL); 1313797ac58cSKevin Wolf 1314797ac58cSKevin Wolf if (cnt < 0) { 1315797ac58cSKevin Wolf printf("aio_multiwrite failed: %s\n", strerror(-cnt)); 1316797ac58cSKevin Wolf goto out; 1317797ac58cSKevin Wolf } 1318797ac58cSKevin Wolf 1319797ac58cSKevin Wolf if (qflag) { 1320797ac58cSKevin Wolf goto out; 1321797ac58cSKevin Wolf } 1322797ac58cSKevin Wolf 1323797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1324797ac58cSKevin Wolf t2 = tsub(t2, t1); 1325797ac58cSKevin Wolf print_report("wrote", &t2, first_offset, total, total, cnt, Cflag); 1326797ac58cSKevin Wolf out: 1327797ac58cSKevin Wolf for (i = 0; i < nr_reqs; i++) { 1328797ac58cSKevin Wolf qemu_io_free(buf[i]); 1329797ac58cSKevin Wolf if (reqs[i].qiov != NULL) { 1330797ac58cSKevin Wolf qemu_iovec_destroy(&qiovs[i]); 1331797ac58cSKevin Wolf } 1332797ac58cSKevin Wolf } 1333797ac58cSKevin Wolf g_free(buf); 1334797ac58cSKevin Wolf g_free(reqs); 1335797ac58cSKevin Wolf g_free(qiovs); 1336797ac58cSKevin Wolf return 0; 1337797ac58cSKevin Wolf } 1338797ac58cSKevin Wolf 1339797ac58cSKevin Wolf struct aio_ctx { 1340797ac58cSKevin Wolf QEMUIOVector qiov; 1341797ac58cSKevin Wolf int64_t offset; 1342797ac58cSKevin Wolf char *buf; 1343797ac58cSKevin Wolf int qflag; 1344797ac58cSKevin Wolf int vflag; 1345797ac58cSKevin Wolf int Cflag; 1346797ac58cSKevin Wolf int Pflag; 1347797ac58cSKevin Wolf int pattern; 1348797ac58cSKevin Wolf struct timeval t1; 1349797ac58cSKevin Wolf }; 1350797ac58cSKevin Wolf 1351797ac58cSKevin Wolf static void aio_write_done(void *opaque, int ret) 1352797ac58cSKevin Wolf { 1353797ac58cSKevin Wolf struct aio_ctx *ctx = opaque; 1354797ac58cSKevin Wolf struct timeval t2; 1355797ac58cSKevin Wolf 1356797ac58cSKevin Wolf gettimeofday(&t2, NULL); 1357797ac58cSKevin Wolf 1358797ac58cSKevin Wolf 1359797ac58cSKevin Wolf if (ret < 0) { 1360797ac58cSKevin Wolf printf("aio_write failed: %s\n", strerror(-ret)); 1361797ac58cSKevin Wolf goto out; 1362797ac58cSKevin Wolf } 1363797ac58cSKevin Wolf 1364797ac58cSKevin Wolf if (ctx->qflag) { 1365797ac58cSKevin Wolf goto out; 1366797ac58cSKevin Wolf } 1367797ac58cSKevin Wolf 1368797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1369797ac58cSKevin Wolf t2 = tsub(t2, ctx->t1); 1370797ac58cSKevin Wolf print_report("wrote", &t2, ctx->offset, ctx->qiov.size, 1371797ac58cSKevin Wolf ctx->qiov.size, 1, ctx->Cflag); 1372797ac58cSKevin Wolf out: 1373797ac58cSKevin Wolf qemu_io_free(ctx->buf); 1374797ac58cSKevin Wolf qemu_iovec_destroy(&ctx->qiov); 1375797ac58cSKevin Wolf g_free(ctx); 1376797ac58cSKevin Wolf } 1377797ac58cSKevin Wolf 1378797ac58cSKevin Wolf static void aio_read_done(void *opaque, int ret) 1379797ac58cSKevin Wolf { 1380797ac58cSKevin Wolf struct aio_ctx *ctx = opaque; 1381797ac58cSKevin Wolf struct timeval t2; 1382797ac58cSKevin Wolf 1383797ac58cSKevin Wolf gettimeofday(&t2, NULL); 1384797ac58cSKevin Wolf 1385797ac58cSKevin Wolf if (ret < 0) { 1386797ac58cSKevin Wolf printf("readv failed: %s\n", strerror(-ret)); 1387797ac58cSKevin Wolf goto out; 1388797ac58cSKevin Wolf } 1389797ac58cSKevin Wolf 1390797ac58cSKevin Wolf if (ctx->Pflag) { 1391797ac58cSKevin Wolf void *cmp_buf = g_malloc(ctx->qiov.size); 1392797ac58cSKevin Wolf 1393797ac58cSKevin Wolf memset(cmp_buf, ctx->pattern, ctx->qiov.size); 1394797ac58cSKevin Wolf if (memcmp(ctx->buf, cmp_buf, ctx->qiov.size)) { 1395797ac58cSKevin Wolf printf("Pattern verification failed at offset %" 1396797ac58cSKevin Wolf PRId64 ", %zd bytes\n", ctx->offset, ctx->qiov.size); 1397797ac58cSKevin Wolf } 1398797ac58cSKevin Wolf g_free(cmp_buf); 1399797ac58cSKevin Wolf } 1400797ac58cSKevin Wolf 1401797ac58cSKevin Wolf if (ctx->qflag) { 1402797ac58cSKevin Wolf goto out; 1403797ac58cSKevin Wolf } 1404797ac58cSKevin Wolf 1405797ac58cSKevin Wolf if (ctx->vflag) { 1406797ac58cSKevin Wolf dump_buffer(ctx->buf, ctx->offset, ctx->qiov.size); 1407797ac58cSKevin Wolf } 1408797ac58cSKevin Wolf 1409797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1410797ac58cSKevin Wolf t2 = tsub(t2, ctx->t1); 1411797ac58cSKevin Wolf print_report("read", &t2, ctx->offset, ctx->qiov.size, 1412797ac58cSKevin Wolf ctx->qiov.size, 1, ctx->Cflag); 1413797ac58cSKevin Wolf out: 1414797ac58cSKevin Wolf qemu_io_free(ctx->buf); 1415797ac58cSKevin Wolf qemu_iovec_destroy(&ctx->qiov); 1416797ac58cSKevin Wolf g_free(ctx); 1417797ac58cSKevin Wolf } 1418797ac58cSKevin Wolf 1419797ac58cSKevin Wolf static void aio_read_help(void) 1420797ac58cSKevin Wolf { 1421797ac58cSKevin Wolf printf( 1422797ac58cSKevin Wolf "\n" 1423797ac58cSKevin Wolf " asynchronously reads a range of bytes from the given offset\n" 1424797ac58cSKevin Wolf "\n" 1425797ac58cSKevin Wolf " Example:\n" 1426797ac58cSKevin Wolf " 'aio_read -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n" 1427797ac58cSKevin Wolf "\n" 1428797ac58cSKevin Wolf " Reads a segment of the currently open file, optionally dumping it to the\n" 1429797ac58cSKevin Wolf " standard output stream (with -v option) for subsequent inspection.\n" 1430797ac58cSKevin Wolf " The read is performed asynchronously and the aio_flush command must be\n" 1431797ac58cSKevin Wolf " used to ensure all outstanding aio requests have been completed.\n" 1432797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1433797ac58cSKevin Wolf " -P, -- use a pattern to verify read data\n" 1434797ac58cSKevin Wolf " -v, -- dump buffer to standard output\n" 1435797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1436797ac58cSKevin Wolf "\n"); 1437797ac58cSKevin Wolf } 1438797ac58cSKevin Wolf 1439797ac58cSKevin Wolf static int aio_read_f(BlockDriverState *bs, int argc, char **argv); 1440797ac58cSKevin Wolf 1441797ac58cSKevin Wolf static const cmdinfo_t aio_read_cmd = { 1442797ac58cSKevin Wolf .name = "aio_read", 1443797ac58cSKevin Wolf .cfunc = aio_read_f, 1444797ac58cSKevin Wolf .argmin = 2, 1445797ac58cSKevin Wolf .argmax = -1, 1446797ac58cSKevin Wolf .args = "[-Cqv] [-P pattern ] off len [len..]", 1447797ac58cSKevin Wolf .oneline = "asynchronously reads a number of bytes", 1448797ac58cSKevin Wolf .help = aio_read_help, 1449797ac58cSKevin Wolf }; 1450797ac58cSKevin Wolf 1451797ac58cSKevin Wolf static int aio_read_f(BlockDriverState *bs, int argc, char **argv) 1452797ac58cSKevin Wolf { 1453797ac58cSKevin Wolf int nr_iov, c; 1454797ac58cSKevin Wolf struct aio_ctx *ctx = g_new0(struct aio_ctx, 1); 1455797ac58cSKevin Wolf 1456797ac58cSKevin Wolf while ((c = getopt(argc, argv, "CP:qv")) != EOF) { 1457797ac58cSKevin Wolf switch (c) { 1458797ac58cSKevin Wolf case 'C': 1459797ac58cSKevin Wolf ctx->Cflag = 1; 1460797ac58cSKevin Wolf break; 1461797ac58cSKevin Wolf case 'P': 1462797ac58cSKevin Wolf ctx->Pflag = 1; 1463797ac58cSKevin Wolf ctx->pattern = parse_pattern(optarg); 1464797ac58cSKevin Wolf if (ctx->pattern < 0) { 1465797ac58cSKevin Wolf g_free(ctx); 1466797ac58cSKevin Wolf return 0; 1467797ac58cSKevin Wolf } 1468797ac58cSKevin Wolf break; 1469797ac58cSKevin Wolf case 'q': 1470797ac58cSKevin Wolf ctx->qflag = 1; 1471797ac58cSKevin Wolf break; 1472797ac58cSKevin Wolf case 'v': 1473797ac58cSKevin Wolf ctx->vflag = 1; 1474797ac58cSKevin Wolf break; 1475797ac58cSKevin Wolf default: 1476797ac58cSKevin Wolf g_free(ctx); 1477c2cdf5c5SKevin Wolf return qemuio_command_usage(&aio_read_cmd); 1478797ac58cSKevin Wolf } 1479797ac58cSKevin Wolf } 1480797ac58cSKevin Wolf 1481797ac58cSKevin Wolf if (optind > argc - 2) { 1482797ac58cSKevin Wolf g_free(ctx); 1483c2cdf5c5SKevin Wolf return qemuio_command_usage(&aio_read_cmd); 1484797ac58cSKevin Wolf } 1485797ac58cSKevin Wolf 1486797ac58cSKevin Wolf ctx->offset = cvtnum(argv[optind]); 1487797ac58cSKevin Wolf if (ctx->offset < 0) { 1488797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", argv[optind]); 1489797ac58cSKevin Wolf g_free(ctx); 1490797ac58cSKevin Wolf return 0; 1491797ac58cSKevin Wolf } 1492797ac58cSKevin Wolf optind++; 1493797ac58cSKevin Wolf 1494797ac58cSKevin Wolf if (ctx->offset & 0x1ff) { 1495797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 1496797ac58cSKevin Wolf ctx->offset); 1497797ac58cSKevin Wolf g_free(ctx); 1498797ac58cSKevin Wolf return 0; 1499797ac58cSKevin Wolf } 1500797ac58cSKevin Wolf 1501797ac58cSKevin Wolf nr_iov = argc - optind; 1502797ac58cSKevin Wolf ctx->buf = create_iovec(bs, &ctx->qiov, &argv[optind], nr_iov, 0xab); 1503797ac58cSKevin Wolf if (ctx->buf == NULL) { 1504797ac58cSKevin Wolf g_free(ctx); 1505797ac58cSKevin Wolf return 0; 1506797ac58cSKevin Wolf } 1507797ac58cSKevin Wolf 1508797ac58cSKevin Wolf gettimeofday(&ctx->t1, NULL); 1509797ac58cSKevin Wolf bdrv_aio_readv(bs, ctx->offset >> 9, &ctx->qiov, 1510797ac58cSKevin Wolf ctx->qiov.size >> 9, aio_read_done, ctx); 1511797ac58cSKevin Wolf return 0; 1512797ac58cSKevin Wolf } 1513797ac58cSKevin Wolf 1514797ac58cSKevin Wolf static void aio_write_help(void) 1515797ac58cSKevin Wolf { 1516797ac58cSKevin Wolf printf( 1517797ac58cSKevin Wolf "\n" 1518797ac58cSKevin Wolf " asynchronously writes a range of bytes from the given offset source\n" 1519797ac58cSKevin Wolf " from multiple buffers\n" 1520797ac58cSKevin Wolf "\n" 1521797ac58cSKevin Wolf " Example:\n" 1522797ac58cSKevin Wolf " 'aio_write 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n" 1523797ac58cSKevin Wolf "\n" 1524797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 1525797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd).\n" 1526797ac58cSKevin Wolf " The write is performed asynchronously and the aio_flush command must be\n" 1527797ac58cSKevin Wolf " used to ensure all outstanding aio requests have been completed.\n" 1528797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 1529797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1530797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1531797ac58cSKevin Wolf "\n"); 1532797ac58cSKevin Wolf } 1533797ac58cSKevin Wolf 1534797ac58cSKevin Wolf static int aio_write_f(BlockDriverState *bs, int argc, char **argv); 1535797ac58cSKevin Wolf 1536797ac58cSKevin Wolf static const cmdinfo_t aio_write_cmd = { 1537797ac58cSKevin Wolf .name = "aio_write", 1538797ac58cSKevin Wolf .cfunc = aio_write_f, 1539797ac58cSKevin Wolf .argmin = 2, 1540797ac58cSKevin Wolf .argmax = -1, 1541797ac58cSKevin Wolf .args = "[-Cq] [-P pattern ] off len [len..]", 1542797ac58cSKevin Wolf .oneline = "asynchronously writes a number of bytes", 1543797ac58cSKevin Wolf .help = aio_write_help, 1544797ac58cSKevin Wolf }; 1545797ac58cSKevin Wolf 1546797ac58cSKevin Wolf static int aio_write_f(BlockDriverState *bs, int argc, char **argv) 1547797ac58cSKevin Wolf { 1548797ac58cSKevin Wolf int nr_iov, c; 1549797ac58cSKevin Wolf int pattern = 0xcd; 1550797ac58cSKevin Wolf struct aio_ctx *ctx = g_new0(struct aio_ctx, 1); 1551797ac58cSKevin Wolf 1552797ac58cSKevin Wolf while ((c = getopt(argc, argv, "CqP:")) != EOF) { 1553797ac58cSKevin Wolf switch (c) { 1554797ac58cSKevin Wolf case 'C': 1555797ac58cSKevin Wolf ctx->Cflag = 1; 1556797ac58cSKevin Wolf break; 1557797ac58cSKevin Wolf case 'q': 1558797ac58cSKevin Wolf ctx->qflag = 1; 1559797ac58cSKevin Wolf break; 1560797ac58cSKevin Wolf case 'P': 1561797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1562797ac58cSKevin Wolf if (pattern < 0) { 1563797ac58cSKevin Wolf g_free(ctx); 1564797ac58cSKevin Wolf return 0; 1565797ac58cSKevin Wolf } 1566797ac58cSKevin Wolf break; 1567797ac58cSKevin Wolf default: 1568797ac58cSKevin Wolf g_free(ctx); 1569c2cdf5c5SKevin Wolf return qemuio_command_usage(&aio_write_cmd); 1570797ac58cSKevin Wolf } 1571797ac58cSKevin Wolf } 1572797ac58cSKevin Wolf 1573797ac58cSKevin Wolf if (optind > argc - 2) { 1574797ac58cSKevin Wolf g_free(ctx); 1575c2cdf5c5SKevin Wolf return qemuio_command_usage(&aio_write_cmd); 1576797ac58cSKevin Wolf } 1577797ac58cSKevin Wolf 1578797ac58cSKevin Wolf ctx->offset = cvtnum(argv[optind]); 1579797ac58cSKevin Wolf if (ctx->offset < 0) { 1580797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", argv[optind]); 1581797ac58cSKevin Wolf g_free(ctx); 1582797ac58cSKevin Wolf return 0; 1583797ac58cSKevin Wolf } 1584797ac58cSKevin Wolf optind++; 1585797ac58cSKevin Wolf 1586797ac58cSKevin Wolf if (ctx->offset & 0x1ff) { 1587797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 1588797ac58cSKevin Wolf ctx->offset); 1589797ac58cSKevin Wolf g_free(ctx); 1590797ac58cSKevin Wolf return 0; 1591797ac58cSKevin Wolf } 1592797ac58cSKevin Wolf 1593797ac58cSKevin Wolf nr_iov = argc - optind; 1594797ac58cSKevin Wolf ctx->buf = create_iovec(bs, &ctx->qiov, &argv[optind], nr_iov, pattern); 1595797ac58cSKevin Wolf if (ctx->buf == NULL) { 1596797ac58cSKevin Wolf g_free(ctx); 1597797ac58cSKevin Wolf return 0; 1598797ac58cSKevin Wolf } 1599797ac58cSKevin Wolf 1600797ac58cSKevin Wolf gettimeofday(&ctx->t1, NULL); 1601797ac58cSKevin Wolf bdrv_aio_writev(bs, ctx->offset >> 9, &ctx->qiov, 1602797ac58cSKevin Wolf ctx->qiov.size >> 9, aio_write_done, ctx); 1603797ac58cSKevin Wolf return 0; 1604797ac58cSKevin Wolf } 1605797ac58cSKevin Wolf 1606797ac58cSKevin Wolf static int aio_flush_f(BlockDriverState *bs, int argc, char **argv) 1607797ac58cSKevin Wolf { 1608797ac58cSKevin Wolf bdrv_drain_all(); 1609797ac58cSKevin Wolf return 0; 1610797ac58cSKevin Wolf } 1611797ac58cSKevin Wolf 1612797ac58cSKevin Wolf static const cmdinfo_t aio_flush_cmd = { 1613797ac58cSKevin Wolf .name = "aio_flush", 1614797ac58cSKevin Wolf .cfunc = aio_flush_f, 1615797ac58cSKevin Wolf .oneline = "completes all outstanding aio requests" 1616797ac58cSKevin Wolf }; 1617797ac58cSKevin Wolf 1618797ac58cSKevin Wolf static int flush_f(BlockDriverState *bs, int argc, char **argv) 1619797ac58cSKevin Wolf { 1620797ac58cSKevin Wolf bdrv_flush(bs); 1621797ac58cSKevin Wolf return 0; 1622797ac58cSKevin Wolf } 1623797ac58cSKevin Wolf 1624797ac58cSKevin Wolf static const cmdinfo_t flush_cmd = { 1625797ac58cSKevin Wolf .name = "flush", 1626797ac58cSKevin Wolf .altname = "f", 1627797ac58cSKevin Wolf .cfunc = flush_f, 1628797ac58cSKevin Wolf .oneline = "flush all in-core file state to disk", 1629797ac58cSKevin Wolf }; 1630797ac58cSKevin Wolf 1631797ac58cSKevin Wolf static int truncate_f(BlockDriverState *bs, int argc, char **argv) 1632797ac58cSKevin Wolf { 1633797ac58cSKevin Wolf int64_t offset; 1634797ac58cSKevin Wolf int ret; 1635797ac58cSKevin Wolf 1636797ac58cSKevin Wolf offset = cvtnum(argv[1]); 1637797ac58cSKevin Wolf if (offset < 0) { 1638797ac58cSKevin Wolf printf("non-numeric truncate argument -- %s\n", argv[1]); 1639797ac58cSKevin Wolf return 0; 1640797ac58cSKevin Wolf } 1641797ac58cSKevin Wolf 1642797ac58cSKevin Wolf ret = bdrv_truncate(bs, offset); 1643797ac58cSKevin Wolf if (ret < 0) { 1644797ac58cSKevin Wolf printf("truncate: %s\n", strerror(-ret)); 1645797ac58cSKevin Wolf return 0; 1646797ac58cSKevin Wolf } 1647797ac58cSKevin Wolf 1648797ac58cSKevin Wolf return 0; 1649797ac58cSKevin Wolf } 1650797ac58cSKevin Wolf 1651797ac58cSKevin Wolf static const cmdinfo_t truncate_cmd = { 1652797ac58cSKevin Wolf .name = "truncate", 1653797ac58cSKevin Wolf .altname = "t", 1654797ac58cSKevin Wolf .cfunc = truncate_f, 1655797ac58cSKevin Wolf .argmin = 1, 1656797ac58cSKevin Wolf .argmax = 1, 1657797ac58cSKevin Wolf .args = "off", 1658797ac58cSKevin Wolf .oneline = "truncates the current file at the given offset", 1659797ac58cSKevin Wolf }; 1660797ac58cSKevin Wolf 1661797ac58cSKevin Wolf static int length_f(BlockDriverState *bs, int argc, char **argv) 1662797ac58cSKevin Wolf { 1663797ac58cSKevin Wolf int64_t size; 1664797ac58cSKevin Wolf char s1[64]; 1665797ac58cSKevin Wolf 1666797ac58cSKevin Wolf size = bdrv_getlength(bs); 1667797ac58cSKevin Wolf if (size < 0) { 1668797ac58cSKevin Wolf printf("getlength: %s\n", strerror(-size)); 1669797ac58cSKevin Wolf return 0; 1670797ac58cSKevin Wolf } 1671797ac58cSKevin Wolf 1672797ac58cSKevin Wolf cvtstr(size, s1, sizeof(s1)); 1673797ac58cSKevin Wolf printf("%s\n", s1); 1674797ac58cSKevin Wolf return 0; 1675797ac58cSKevin Wolf } 1676797ac58cSKevin Wolf 1677797ac58cSKevin Wolf 1678797ac58cSKevin Wolf static const cmdinfo_t length_cmd = { 1679797ac58cSKevin Wolf .name = "length", 1680797ac58cSKevin Wolf .altname = "l", 1681797ac58cSKevin Wolf .cfunc = length_f, 1682797ac58cSKevin Wolf .oneline = "gets the length of the current file", 1683797ac58cSKevin Wolf }; 1684797ac58cSKevin Wolf 1685797ac58cSKevin Wolf 1686797ac58cSKevin Wolf static int info_f(BlockDriverState *bs, int argc, char **argv) 1687797ac58cSKevin Wolf { 1688797ac58cSKevin Wolf BlockDriverInfo bdi; 1689a8d8ecb7SMax Reitz ImageInfoSpecific *spec_info; 1690797ac58cSKevin Wolf char s1[64], s2[64]; 1691797ac58cSKevin Wolf int ret; 1692797ac58cSKevin Wolf 1693797ac58cSKevin Wolf if (bs->drv && bs->drv->format_name) { 1694797ac58cSKevin Wolf printf("format name: %s\n", bs->drv->format_name); 1695797ac58cSKevin Wolf } 1696797ac58cSKevin Wolf if (bs->drv && bs->drv->protocol_name) { 1697797ac58cSKevin Wolf printf("format name: %s\n", bs->drv->protocol_name); 1698797ac58cSKevin Wolf } 1699797ac58cSKevin Wolf 1700797ac58cSKevin Wolf ret = bdrv_get_info(bs, &bdi); 1701797ac58cSKevin Wolf if (ret) { 1702797ac58cSKevin Wolf return 0; 1703797ac58cSKevin Wolf } 1704797ac58cSKevin Wolf 1705797ac58cSKevin Wolf cvtstr(bdi.cluster_size, s1, sizeof(s1)); 1706797ac58cSKevin Wolf cvtstr(bdi.vm_state_offset, s2, sizeof(s2)); 1707797ac58cSKevin Wolf 1708797ac58cSKevin Wolf printf("cluster size: %s\n", s1); 1709797ac58cSKevin Wolf printf("vm state offset: %s\n", s2); 1710797ac58cSKevin Wolf 1711a8d8ecb7SMax Reitz spec_info = bdrv_get_specific_info(bs); 1712a8d8ecb7SMax Reitz if (spec_info) { 1713a8d8ecb7SMax Reitz printf("Format specific information:\n"); 1714a8d8ecb7SMax Reitz bdrv_image_info_specific_dump(fprintf, stdout, spec_info); 1715a8d8ecb7SMax Reitz qapi_free_ImageInfoSpecific(spec_info); 1716a8d8ecb7SMax Reitz } 1717a8d8ecb7SMax Reitz 1718797ac58cSKevin Wolf return 0; 1719797ac58cSKevin Wolf } 1720797ac58cSKevin Wolf 1721797ac58cSKevin Wolf 1722797ac58cSKevin Wolf 1723797ac58cSKevin Wolf static const cmdinfo_t info_cmd = { 1724797ac58cSKevin Wolf .name = "info", 1725797ac58cSKevin Wolf .altname = "i", 1726797ac58cSKevin Wolf .cfunc = info_f, 1727797ac58cSKevin Wolf .oneline = "prints information about the current file", 1728797ac58cSKevin Wolf }; 1729797ac58cSKevin Wolf 1730797ac58cSKevin Wolf static void discard_help(void) 1731797ac58cSKevin Wolf { 1732797ac58cSKevin Wolf printf( 1733797ac58cSKevin Wolf "\n" 1734797ac58cSKevin Wolf " discards a range of bytes from the given offset\n" 1735797ac58cSKevin Wolf "\n" 1736797ac58cSKevin Wolf " Example:\n" 1737797ac58cSKevin Wolf " 'discard 512 1k' - discards 1 kilobyte from 512 bytes into the file\n" 1738797ac58cSKevin Wolf "\n" 1739797ac58cSKevin Wolf " Discards a segment of the currently open file.\n" 1740797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1741797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1742797ac58cSKevin Wolf "\n"); 1743797ac58cSKevin Wolf } 1744797ac58cSKevin Wolf 1745797ac58cSKevin Wolf static int discard_f(BlockDriverState *bs, int argc, char **argv); 1746797ac58cSKevin Wolf 1747797ac58cSKevin Wolf static const cmdinfo_t discard_cmd = { 1748797ac58cSKevin Wolf .name = "discard", 1749797ac58cSKevin Wolf .altname = "d", 1750797ac58cSKevin Wolf .cfunc = discard_f, 1751797ac58cSKevin Wolf .argmin = 2, 1752797ac58cSKevin Wolf .argmax = -1, 1753797ac58cSKevin Wolf .args = "[-Cq] off len", 1754797ac58cSKevin Wolf .oneline = "discards a number of bytes at a specified offset", 1755797ac58cSKevin Wolf .help = discard_help, 1756797ac58cSKevin Wolf }; 1757797ac58cSKevin Wolf 1758797ac58cSKevin Wolf static int discard_f(BlockDriverState *bs, int argc, char **argv) 1759797ac58cSKevin Wolf { 1760797ac58cSKevin Wolf struct timeval t1, t2; 1761797ac58cSKevin Wolf int Cflag = 0, qflag = 0; 1762797ac58cSKevin Wolf int c, ret; 1763797ac58cSKevin Wolf int64_t offset; 1764797ac58cSKevin Wolf int count; 1765797ac58cSKevin Wolf 1766797ac58cSKevin Wolf while ((c = getopt(argc, argv, "Cq")) != EOF) { 1767797ac58cSKevin Wolf switch (c) { 1768797ac58cSKevin Wolf case 'C': 1769797ac58cSKevin Wolf Cflag = 1; 1770797ac58cSKevin Wolf break; 1771797ac58cSKevin Wolf case 'q': 1772797ac58cSKevin Wolf qflag = 1; 1773797ac58cSKevin Wolf break; 1774797ac58cSKevin Wolf default: 1775c2cdf5c5SKevin Wolf return qemuio_command_usage(&discard_cmd); 1776797ac58cSKevin Wolf } 1777797ac58cSKevin Wolf } 1778797ac58cSKevin Wolf 1779797ac58cSKevin Wolf if (optind != argc - 2) { 1780c2cdf5c5SKevin Wolf return qemuio_command_usage(&discard_cmd); 1781797ac58cSKevin Wolf } 1782797ac58cSKevin Wolf 1783797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1784797ac58cSKevin Wolf if (offset < 0) { 1785797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", argv[optind]); 1786797ac58cSKevin Wolf return 0; 1787797ac58cSKevin Wolf } 1788797ac58cSKevin Wolf 1789797ac58cSKevin Wolf optind++; 1790797ac58cSKevin Wolf count = cvtnum(argv[optind]); 1791797ac58cSKevin Wolf if (count < 0) { 1792797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", argv[optind]); 1793797ac58cSKevin Wolf return 0; 1794797ac58cSKevin Wolf } 1795797ac58cSKevin Wolf 1796797ac58cSKevin Wolf gettimeofday(&t1, NULL); 1797797ac58cSKevin Wolf ret = bdrv_discard(bs, offset >> BDRV_SECTOR_BITS, 1798797ac58cSKevin Wolf count >> BDRV_SECTOR_BITS); 1799797ac58cSKevin Wolf gettimeofday(&t2, NULL); 1800797ac58cSKevin Wolf 1801797ac58cSKevin Wolf if (ret < 0) { 1802797ac58cSKevin Wolf printf("discard failed: %s\n", strerror(-ret)); 1803797ac58cSKevin Wolf goto out; 1804797ac58cSKevin Wolf } 1805797ac58cSKevin Wolf 1806797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1807797ac58cSKevin Wolf if (!qflag) { 1808797ac58cSKevin Wolf t2 = tsub(t2, t1); 1809797ac58cSKevin Wolf print_report("discard", &t2, offset, count, count, 1, Cflag); 1810797ac58cSKevin Wolf } 1811797ac58cSKevin Wolf 1812797ac58cSKevin Wolf out: 1813797ac58cSKevin Wolf return 0; 1814797ac58cSKevin Wolf } 1815797ac58cSKevin Wolf 1816797ac58cSKevin Wolf static int alloc_f(BlockDriverState *bs, int argc, char **argv) 1817797ac58cSKevin Wolf { 1818797ac58cSKevin Wolf int64_t offset, sector_num; 1819797ac58cSKevin Wolf int nb_sectors, remaining; 1820797ac58cSKevin Wolf char s1[64]; 1821797ac58cSKevin Wolf int num, sum_alloc; 1822797ac58cSKevin Wolf int ret; 1823797ac58cSKevin Wolf 1824797ac58cSKevin Wolf offset = cvtnum(argv[1]); 1825797ac58cSKevin Wolf if (offset < 0) { 1826797ac58cSKevin Wolf printf("non-numeric offset argument -- %s\n", argv[1]); 1827797ac58cSKevin Wolf return 0; 1828797ac58cSKevin Wolf } else if (offset & 0x1ff) { 1829797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 1830797ac58cSKevin Wolf offset); 1831797ac58cSKevin Wolf return 0; 1832797ac58cSKevin Wolf } 1833797ac58cSKevin Wolf 1834797ac58cSKevin Wolf if (argc == 3) { 1835797ac58cSKevin Wolf nb_sectors = cvtnum(argv[2]); 1836797ac58cSKevin Wolf if (nb_sectors < 0) { 1837797ac58cSKevin Wolf printf("non-numeric length argument -- %s\n", argv[2]); 1838797ac58cSKevin Wolf return 0; 1839797ac58cSKevin Wolf } 1840797ac58cSKevin Wolf } else { 1841797ac58cSKevin Wolf nb_sectors = 1; 1842797ac58cSKevin Wolf } 1843797ac58cSKevin Wolf 1844797ac58cSKevin Wolf remaining = nb_sectors; 1845797ac58cSKevin Wolf sum_alloc = 0; 1846797ac58cSKevin Wolf sector_num = offset >> 9; 1847797ac58cSKevin Wolf while (remaining) { 1848797ac58cSKevin Wolf ret = bdrv_is_allocated(bs, sector_num, remaining, &num); 1849d663640cSPaolo Bonzini if (ret < 0) { 1850d663640cSPaolo Bonzini printf("is_allocated failed: %s\n", strerror(-ret)); 1851d663640cSPaolo Bonzini return 0; 1852d663640cSPaolo Bonzini } 1853797ac58cSKevin Wolf sector_num += num; 1854797ac58cSKevin Wolf remaining -= num; 1855797ac58cSKevin Wolf if (ret) { 1856797ac58cSKevin Wolf sum_alloc += num; 1857797ac58cSKevin Wolf } 1858797ac58cSKevin Wolf if (num == 0) { 1859797ac58cSKevin Wolf nb_sectors -= remaining; 1860797ac58cSKevin Wolf remaining = 0; 1861797ac58cSKevin Wolf } 1862797ac58cSKevin Wolf } 1863797ac58cSKevin Wolf 1864797ac58cSKevin Wolf cvtstr(offset, s1, sizeof(s1)); 1865797ac58cSKevin Wolf 1866797ac58cSKevin Wolf printf("%d/%d sectors allocated at offset %s\n", 1867797ac58cSKevin Wolf sum_alloc, nb_sectors, s1); 1868797ac58cSKevin Wolf return 0; 1869797ac58cSKevin Wolf } 1870797ac58cSKevin Wolf 1871797ac58cSKevin Wolf static const cmdinfo_t alloc_cmd = { 1872797ac58cSKevin Wolf .name = "alloc", 1873797ac58cSKevin Wolf .altname = "a", 1874797ac58cSKevin Wolf .argmin = 1, 1875797ac58cSKevin Wolf .argmax = 2, 1876797ac58cSKevin Wolf .cfunc = alloc_f, 1877797ac58cSKevin Wolf .args = "off [sectors]", 1878797ac58cSKevin Wolf .oneline = "checks if a sector is present in the file", 1879797ac58cSKevin Wolf }; 1880797ac58cSKevin Wolf 1881797ac58cSKevin Wolf 1882797ac58cSKevin Wolf static int map_is_allocated(BlockDriverState *bs, int64_t sector_num, 1883797ac58cSKevin Wolf int64_t nb_sectors, int64_t *pnum) 1884797ac58cSKevin Wolf { 1885797ac58cSKevin Wolf int num, num_checked; 1886797ac58cSKevin Wolf int ret, firstret; 1887797ac58cSKevin Wolf 1888797ac58cSKevin Wolf num_checked = MIN(nb_sectors, INT_MAX); 1889797ac58cSKevin Wolf ret = bdrv_is_allocated(bs, sector_num, num_checked, &num); 1890797ac58cSKevin Wolf if (ret < 0) { 1891797ac58cSKevin Wolf return ret; 1892797ac58cSKevin Wolf } 1893797ac58cSKevin Wolf 1894797ac58cSKevin Wolf firstret = ret; 1895797ac58cSKevin Wolf *pnum = num; 1896797ac58cSKevin Wolf 1897797ac58cSKevin Wolf while (nb_sectors > 0 && ret == firstret) { 1898797ac58cSKevin Wolf sector_num += num; 1899797ac58cSKevin Wolf nb_sectors -= num; 1900797ac58cSKevin Wolf 1901797ac58cSKevin Wolf num_checked = MIN(nb_sectors, INT_MAX); 1902797ac58cSKevin Wolf ret = bdrv_is_allocated(bs, sector_num, num_checked, &num); 1903797ac58cSKevin Wolf if (ret == firstret) { 1904797ac58cSKevin Wolf *pnum += num; 1905797ac58cSKevin Wolf } else { 1906797ac58cSKevin Wolf break; 1907797ac58cSKevin Wolf } 1908797ac58cSKevin Wolf } 1909797ac58cSKevin Wolf 1910797ac58cSKevin Wolf return firstret; 1911797ac58cSKevin Wolf } 1912797ac58cSKevin Wolf 1913797ac58cSKevin Wolf static int map_f(BlockDriverState *bs, int argc, char **argv) 1914797ac58cSKevin Wolf { 1915797ac58cSKevin Wolf int64_t offset; 1916797ac58cSKevin Wolf int64_t nb_sectors; 1917797ac58cSKevin Wolf char s1[64]; 1918797ac58cSKevin Wolf int64_t num; 1919797ac58cSKevin Wolf int ret; 1920797ac58cSKevin Wolf const char *retstr; 1921797ac58cSKevin Wolf 1922797ac58cSKevin Wolf offset = 0; 1923797ac58cSKevin Wolf nb_sectors = bs->total_sectors; 1924797ac58cSKevin Wolf 1925797ac58cSKevin Wolf do { 1926797ac58cSKevin Wolf ret = map_is_allocated(bs, offset, nb_sectors, &num); 1927797ac58cSKevin Wolf if (ret < 0) { 1928797ac58cSKevin Wolf error_report("Failed to get allocation status: %s", strerror(-ret)); 1929797ac58cSKevin Wolf return 0; 1930797ac58cSKevin Wolf } 1931797ac58cSKevin Wolf 1932797ac58cSKevin Wolf retstr = ret ? " allocated" : "not allocated"; 1933797ac58cSKevin Wolf cvtstr(offset << 9ULL, s1, sizeof(s1)); 1934797ac58cSKevin Wolf printf("[% 24" PRId64 "] % 8" PRId64 "/% 8" PRId64 " sectors %s " 1935797ac58cSKevin Wolf "at offset %s (%d)\n", 1936797ac58cSKevin Wolf offset << 9ULL, num, nb_sectors, retstr, s1, ret); 1937797ac58cSKevin Wolf 1938797ac58cSKevin Wolf offset += num; 1939797ac58cSKevin Wolf nb_sectors -= num; 1940797ac58cSKevin Wolf } while (offset < bs->total_sectors); 1941797ac58cSKevin Wolf 1942797ac58cSKevin Wolf return 0; 1943797ac58cSKevin Wolf } 1944797ac58cSKevin Wolf 1945797ac58cSKevin Wolf static const cmdinfo_t map_cmd = { 1946797ac58cSKevin Wolf .name = "map", 1947797ac58cSKevin Wolf .argmin = 0, 1948797ac58cSKevin Wolf .argmax = 0, 1949797ac58cSKevin Wolf .cfunc = map_f, 1950797ac58cSKevin Wolf .args = "", 1951797ac58cSKevin Wolf .oneline = "prints the allocated areas of a file", 1952797ac58cSKevin Wolf }; 1953797ac58cSKevin Wolf 1954797ac58cSKevin Wolf static int break_f(BlockDriverState *bs, int argc, char **argv) 1955797ac58cSKevin Wolf { 1956797ac58cSKevin Wolf int ret; 1957797ac58cSKevin Wolf 1958797ac58cSKevin Wolf ret = bdrv_debug_breakpoint(bs, argv[1], argv[2]); 1959797ac58cSKevin Wolf if (ret < 0) { 1960797ac58cSKevin Wolf printf("Could not set breakpoint: %s\n", strerror(-ret)); 1961797ac58cSKevin Wolf } 1962797ac58cSKevin Wolf 1963797ac58cSKevin Wolf return 0; 1964797ac58cSKevin Wolf } 1965797ac58cSKevin Wolf 19664cc70e93SFam Zheng static int remove_break_f(BlockDriverState *bs, int argc, char **argv) 19674cc70e93SFam Zheng { 19684cc70e93SFam Zheng int ret; 19694cc70e93SFam Zheng 19704cc70e93SFam Zheng ret = bdrv_debug_remove_breakpoint(bs, argv[1]); 19714cc70e93SFam Zheng if (ret < 0) { 19724cc70e93SFam Zheng printf("Could not remove breakpoint %s: %s\n", argv[1], strerror(-ret)); 19734cc70e93SFam Zheng } 19744cc70e93SFam Zheng 19754cc70e93SFam Zheng return 0; 19764cc70e93SFam Zheng } 19774cc70e93SFam Zheng 1978797ac58cSKevin Wolf static const cmdinfo_t break_cmd = { 1979797ac58cSKevin Wolf .name = "break", 1980797ac58cSKevin Wolf .argmin = 2, 1981797ac58cSKevin Wolf .argmax = 2, 1982797ac58cSKevin Wolf .cfunc = break_f, 1983797ac58cSKevin Wolf .args = "event tag", 1984797ac58cSKevin Wolf .oneline = "sets a breakpoint on event and tags the stopped " 1985797ac58cSKevin Wolf "request as tag", 1986797ac58cSKevin Wolf }; 1987797ac58cSKevin Wolf 19884cc70e93SFam Zheng static const cmdinfo_t remove_break_cmd = { 19894cc70e93SFam Zheng .name = "remove_break", 19904cc70e93SFam Zheng .argmin = 1, 19914cc70e93SFam Zheng .argmax = 1, 19924cc70e93SFam Zheng .cfunc = remove_break_f, 19934cc70e93SFam Zheng .args = "tag", 19944cc70e93SFam Zheng .oneline = "remove a breakpoint by tag", 19954cc70e93SFam Zheng }; 19964cc70e93SFam Zheng 1997797ac58cSKevin Wolf static int resume_f(BlockDriverState *bs, int argc, char **argv) 1998797ac58cSKevin Wolf { 1999797ac58cSKevin Wolf int ret; 2000797ac58cSKevin Wolf 2001797ac58cSKevin Wolf ret = bdrv_debug_resume(bs, argv[1]); 2002797ac58cSKevin Wolf if (ret < 0) { 2003797ac58cSKevin Wolf printf("Could not resume request: %s\n", strerror(-ret)); 2004797ac58cSKevin Wolf } 2005797ac58cSKevin Wolf 2006797ac58cSKevin Wolf return 0; 2007797ac58cSKevin Wolf } 2008797ac58cSKevin Wolf 2009797ac58cSKevin Wolf static const cmdinfo_t resume_cmd = { 2010797ac58cSKevin Wolf .name = "resume", 2011797ac58cSKevin Wolf .argmin = 1, 2012797ac58cSKevin Wolf .argmax = 1, 2013797ac58cSKevin Wolf .cfunc = resume_f, 2014797ac58cSKevin Wolf .args = "tag", 2015797ac58cSKevin Wolf .oneline = "resumes the request tagged as tag", 2016797ac58cSKevin Wolf }; 2017797ac58cSKevin Wolf 2018797ac58cSKevin Wolf static int wait_break_f(BlockDriverState *bs, int argc, char **argv) 2019797ac58cSKevin Wolf { 2020797ac58cSKevin Wolf while (!bdrv_debug_is_suspended(bs, argv[1])) { 2021b47ec2c4SPaolo Bonzini aio_poll(bdrv_get_aio_context(bs), true); 2022797ac58cSKevin Wolf } 2023797ac58cSKevin Wolf 2024797ac58cSKevin Wolf return 0; 2025797ac58cSKevin Wolf } 2026797ac58cSKevin Wolf 2027797ac58cSKevin Wolf static const cmdinfo_t wait_break_cmd = { 2028797ac58cSKevin Wolf .name = "wait_break", 2029797ac58cSKevin Wolf .argmin = 1, 2030797ac58cSKevin Wolf .argmax = 1, 2031797ac58cSKevin Wolf .cfunc = wait_break_f, 2032797ac58cSKevin Wolf .args = "tag", 2033797ac58cSKevin Wolf .oneline = "waits for the suspension of a request", 2034797ac58cSKevin Wolf }; 2035797ac58cSKevin Wolf 2036797ac58cSKevin Wolf static int abort_f(BlockDriverState *bs, int argc, char **argv) 2037797ac58cSKevin Wolf { 2038797ac58cSKevin Wolf abort(); 2039797ac58cSKevin Wolf } 2040797ac58cSKevin Wolf 2041797ac58cSKevin Wolf static const cmdinfo_t abort_cmd = { 2042797ac58cSKevin Wolf .name = "abort", 2043797ac58cSKevin Wolf .cfunc = abort_f, 2044797ac58cSKevin Wolf .flags = CMD_NOFILE_OK, 2045797ac58cSKevin Wolf .oneline = "simulate a program crash using abort(3)", 2046797ac58cSKevin Wolf }; 2047797ac58cSKevin Wolf 2048cd33d02aSKevin Wolf static void sleep_cb(void *opaque) 2049cd33d02aSKevin Wolf { 2050cd33d02aSKevin Wolf bool *expired = opaque; 2051cd33d02aSKevin Wolf *expired = true; 2052cd33d02aSKevin Wolf } 2053cd33d02aSKevin Wolf 2054cd33d02aSKevin Wolf static int sleep_f(BlockDriverState *bs, int argc, char **argv) 2055cd33d02aSKevin Wolf { 2056cd33d02aSKevin Wolf char *endptr; 2057cd33d02aSKevin Wolf long ms; 2058cd33d02aSKevin Wolf struct QEMUTimer *timer; 2059cd33d02aSKevin Wolf bool expired = false; 2060cd33d02aSKevin Wolf 2061cd33d02aSKevin Wolf ms = strtol(argv[1], &endptr, 0); 2062cd33d02aSKevin Wolf if (ms < 0 || *endptr != '\0') { 2063cd33d02aSKevin Wolf printf("%s is not a valid number\n", argv[1]); 2064cd33d02aSKevin Wolf return 0; 2065cd33d02aSKevin Wolf } 2066cd33d02aSKevin Wolf 2067cd33d02aSKevin Wolf timer = timer_new_ns(QEMU_CLOCK_HOST, sleep_cb, &expired); 2068cd33d02aSKevin Wolf timer_mod(timer, qemu_clock_get_ns(QEMU_CLOCK_HOST) + SCALE_MS * ms); 2069cd33d02aSKevin Wolf 2070cd33d02aSKevin Wolf while (!expired) { 2071cd33d02aSKevin Wolf main_loop_wait(false); 2072cd33d02aSKevin Wolf } 2073cd33d02aSKevin Wolf 2074cd33d02aSKevin Wolf timer_free(timer); 2075cd33d02aSKevin Wolf 2076cd33d02aSKevin Wolf return 0; 2077cd33d02aSKevin Wolf } 2078cd33d02aSKevin Wolf 2079cd33d02aSKevin Wolf static const cmdinfo_t sleep_cmd = { 2080cd33d02aSKevin Wolf .name = "sleep", 2081cd33d02aSKevin Wolf .argmin = 1, 2082cd33d02aSKevin Wolf .argmax = 1, 2083cd33d02aSKevin Wolf .cfunc = sleep_f, 2084cd33d02aSKevin Wolf .flags = CMD_NOFILE_OK, 2085cd33d02aSKevin Wolf .oneline = "waits for the given value in milliseconds", 2086cd33d02aSKevin Wolf }; 2087cd33d02aSKevin Wolf 2088f18a834aSKevin Wolf static void help_oneline(const char *cmd, const cmdinfo_t *ct) 2089f18a834aSKevin Wolf { 2090f18a834aSKevin Wolf if (cmd) { 2091f18a834aSKevin Wolf printf("%s ", cmd); 2092f18a834aSKevin Wolf } else { 2093f18a834aSKevin Wolf printf("%s ", ct->name); 2094f18a834aSKevin Wolf if (ct->altname) { 2095f18a834aSKevin Wolf printf("(or %s) ", ct->altname); 2096f18a834aSKevin Wolf } 2097f18a834aSKevin Wolf } 2098f18a834aSKevin Wolf 2099f18a834aSKevin Wolf if (ct->args) { 2100f18a834aSKevin Wolf printf("%s ", ct->args); 2101f18a834aSKevin Wolf } 2102f18a834aSKevin Wolf printf("-- %s\n", ct->oneline); 2103f18a834aSKevin Wolf } 2104f18a834aSKevin Wolf 2105f18a834aSKevin Wolf static void help_onecmd(const char *cmd, const cmdinfo_t *ct) 2106f18a834aSKevin Wolf { 2107f18a834aSKevin Wolf help_oneline(cmd, ct); 2108f18a834aSKevin Wolf if (ct->help) { 2109f18a834aSKevin Wolf ct->help(); 2110f18a834aSKevin Wolf } 2111f18a834aSKevin Wolf } 2112f18a834aSKevin Wolf 2113f18a834aSKevin Wolf static void help_all(void) 2114f18a834aSKevin Wolf { 2115f18a834aSKevin Wolf const cmdinfo_t *ct; 2116f18a834aSKevin Wolf 2117f18a834aSKevin Wolf for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { 2118f18a834aSKevin Wolf help_oneline(ct->name, ct); 2119f18a834aSKevin Wolf } 2120f18a834aSKevin Wolf printf("\nUse 'help commandname' for extended help.\n"); 2121f18a834aSKevin Wolf } 2122f18a834aSKevin Wolf 2123f18a834aSKevin Wolf static int help_f(BlockDriverState *bs, int argc, char **argv) 2124f18a834aSKevin Wolf { 2125f18a834aSKevin Wolf const cmdinfo_t *ct; 2126f18a834aSKevin Wolf 2127f18a834aSKevin Wolf if (argc == 1) { 2128f18a834aSKevin Wolf help_all(); 2129f18a834aSKevin Wolf return 0; 2130f18a834aSKevin Wolf } 2131f18a834aSKevin Wolf 2132f18a834aSKevin Wolf ct = find_command(argv[1]); 2133f18a834aSKevin Wolf if (ct == NULL) { 2134f18a834aSKevin Wolf printf("command %s not found\n", argv[1]); 2135f18a834aSKevin Wolf return 0; 2136f18a834aSKevin Wolf } 2137f18a834aSKevin Wolf 2138f18a834aSKevin Wolf help_onecmd(argv[1], ct); 2139f18a834aSKevin Wolf return 0; 2140f18a834aSKevin Wolf } 2141f18a834aSKevin Wolf 2142f18a834aSKevin Wolf static const cmdinfo_t help_cmd = { 2143f18a834aSKevin Wolf .name = "help", 2144f18a834aSKevin Wolf .altname = "?", 2145f18a834aSKevin Wolf .cfunc = help_f, 2146f18a834aSKevin Wolf .argmin = 0, 2147f18a834aSKevin Wolf .argmax = 1, 2148f18a834aSKevin Wolf .flags = CMD_FLAG_GLOBAL, 2149f18a834aSKevin Wolf .args = "[command]", 2150f18a834aSKevin Wolf .oneline = "help for one or all commands", 2151f18a834aSKevin Wolf }; 2152f18a834aSKevin Wolf 21533d21994fSKevin Wolf bool qemuio_command(BlockDriverState *bs, const char *cmd) 2154dd583296SKevin Wolf { 2155dd583296SKevin Wolf char *input; 2156dd583296SKevin Wolf const cmdinfo_t *ct; 2157dd583296SKevin Wolf char **v; 2158dd583296SKevin Wolf int c; 2159dd583296SKevin Wolf bool done = false; 2160dd583296SKevin Wolf 2161dd583296SKevin Wolf input = g_strdup(cmd); 2162dd583296SKevin Wolf v = breakline(input, &c); 2163dd583296SKevin Wolf if (c) { 2164dd583296SKevin Wolf ct = find_command(v[0]); 2165dd583296SKevin Wolf if (ct) { 21663d21994fSKevin Wolf done = command(bs, ct, c, v); 2167dd583296SKevin Wolf } else { 2168dd583296SKevin Wolf fprintf(stderr, "command \"%s\" not found\n", v[0]); 2169dd583296SKevin Wolf } 2170dd583296SKevin Wolf } 2171dd583296SKevin Wolf g_free(input); 2172dd583296SKevin Wolf g_free(v); 2173dd583296SKevin Wolf 2174dd583296SKevin Wolf return done; 2175dd583296SKevin Wolf } 2176dd583296SKevin Wolf 2177797ac58cSKevin Wolf static void __attribute((constructor)) init_qemuio_commands(void) 2178797ac58cSKevin Wolf { 2179797ac58cSKevin Wolf /* initialize commands */ 2180c2cdf5c5SKevin Wolf qemuio_add_command(&help_cmd); 2181c2cdf5c5SKevin Wolf qemuio_add_command(&read_cmd); 2182c2cdf5c5SKevin Wolf qemuio_add_command(&readv_cmd); 2183c2cdf5c5SKevin Wolf qemuio_add_command(&write_cmd); 2184c2cdf5c5SKevin Wolf qemuio_add_command(&writev_cmd); 2185c2cdf5c5SKevin Wolf qemuio_add_command(&multiwrite_cmd); 2186c2cdf5c5SKevin Wolf qemuio_add_command(&aio_read_cmd); 2187c2cdf5c5SKevin Wolf qemuio_add_command(&aio_write_cmd); 2188c2cdf5c5SKevin Wolf qemuio_add_command(&aio_flush_cmd); 2189c2cdf5c5SKevin Wolf qemuio_add_command(&flush_cmd); 2190c2cdf5c5SKevin Wolf qemuio_add_command(&truncate_cmd); 2191c2cdf5c5SKevin Wolf qemuio_add_command(&length_cmd); 2192c2cdf5c5SKevin Wolf qemuio_add_command(&info_cmd); 2193c2cdf5c5SKevin Wolf qemuio_add_command(&discard_cmd); 2194c2cdf5c5SKevin Wolf qemuio_add_command(&alloc_cmd); 2195c2cdf5c5SKevin Wolf qemuio_add_command(&map_cmd); 2196c2cdf5c5SKevin Wolf qemuio_add_command(&break_cmd); 21974cc70e93SFam Zheng qemuio_add_command(&remove_break_cmd); 2198c2cdf5c5SKevin Wolf qemuio_add_command(&resume_cmd); 2199c2cdf5c5SKevin Wolf qemuio_add_command(&wait_break_cmd); 2200c2cdf5c5SKevin Wolf qemuio_add_command(&abort_cmd); 2201cd33d02aSKevin Wolf qemuio_add_command(&sleep_cmd); 2202797ac58cSKevin Wolf } 2203