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 1180c71a24SPeter Maydell #include "qemu/osdep.h" 12da34e65cSMarkus Armbruster #include "qapi/error.h" 133d21994fSKevin Wolf #include "qemu-io.h" 144c7b7e9bSMax Reitz #include "sysemu/block-backend.h" 154c7b7e9bSMax Reitz #include "block/block.h" 164c7b7e9bSMax Reitz #include "block/block_int.h" /* for info_f() */ 17a8d8ecb7SMax Reitz #include "block/qapi.h" 18d49b6836SMarkus Armbruster #include "qemu/error-report.h" 196a1751b7SAlex Bligh #include "qemu/main-loop.h" 20cd33d02aSKevin Wolf #include "qemu/timer.h" 21a91f9584SFam Zheng #include "sysemu/block-backend.h" 22f348b6d1SVeronia Bahaa #include "qemu/cutils.h" 23797ac58cSKevin Wolf 24797ac58cSKevin Wolf #define CMD_NOFILE_OK 0x01 25797ac58cSKevin Wolf 26f9883880SStefan Weil bool qemuio_misalign; 27797ac58cSKevin Wolf 28c2cdf5c5SKevin Wolf static cmdinfo_t *cmdtab; 29c2cdf5c5SKevin Wolf static int ncmds; 30c2cdf5c5SKevin Wolf 31c2cdf5c5SKevin Wolf static int compare_cmdname(const void *a, const void *b) 32c2cdf5c5SKevin Wolf { 33c2cdf5c5SKevin Wolf return strcmp(((const cmdinfo_t *)a)->name, 34c2cdf5c5SKevin Wolf ((const cmdinfo_t *)b)->name); 35c2cdf5c5SKevin Wolf } 36c2cdf5c5SKevin Wolf 37c2cdf5c5SKevin Wolf void qemuio_add_command(const cmdinfo_t *ci) 38c2cdf5c5SKevin Wolf { 3902c4f26bSMarkus Armbruster cmdtab = g_renew(cmdinfo_t, cmdtab, ++ncmds); 40c2cdf5c5SKevin Wolf cmdtab[ncmds - 1] = *ci; 41c2cdf5c5SKevin Wolf qsort(cmdtab, ncmds, sizeof(*cmdtab), compare_cmdname); 42c2cdf5c5SKevin Wolf } 43c2cdf5c5SKevin Wolf 44c2cdf5c5SKevin Wolf int qemuio_command_usage(const cmdinfo_t *ci) 45c2cdf5c5SKevin Wolf { 46c2cdf5c5SKevin Wolf printf("%s %s -- %s\n", ci->name, ci->args, ci->oneline); 47c2cdf5c5SKevin Wolf return 0; 48c2cdf5c5SKevin Wolf } 49c2cdf5c5SKevin Wolf 504c7b7e9bSMax Reitz static int init_check_command(BlockBackend *blk, const cmdinfo_t *ct) 51c2cdf5c5SKevin Wolf { 52c2cdf5c5SKevin Wolf if (ct->flags & CMD_FLAG_GLOBAL) { 53c2cdf5c5SKevin Wolf return 1; 54c2cdf5c5SKevin Wolf } 554c7b7e9bSMax Reitz if (!(ct->flags & CMD_NOFILE_OK) && !blk) { 56c2cdf5c5SKevin Wolf fprintf(stderr, "no file open, try 'help open'\n"); 57c2cdf5c5SKevin Wolf return 0; 58c2cdf5c5SKevin Wolf } 59c2cdf5c5SKevin Wolf return 1; 60c2cdf5c5SKevin Wolf } 61c2cdf5c5SKevin Wolf 624c7b7e9bSMax Reitz static int command(BlockBackend *blk, const cmdinfo_t *ct, int argc, 633d21994fSKevin Wolf char **argv) 64c2cdf5c5SKevin Wolf { 65c2cdf5c5SKevin Wolf char *cmd = argv[0]; 66c2cdf5c5SKevin Wolf 674c7b7e9bSMax Reitz if (!init_check_command(blk, ct)) { 68c2cdf5c5SKevin Wolf return 0; 69c2cdf5c5SKevin Wolf } 70c2cdf5c5SKevin Wolf 71c2cdf5c5SKevin Wolf if (argc - 1 < ct->argmin || (ct->argmax != -1 && argc - 1 > ct->argmax)) { 72c2cdf5c5SKevin Wolf if (ct->argmax == -1) { 73c2cdf5c5SKevin Wolf fprintf(stderr, 74c2cdf5c5SKevin Wolf "bad argument count %d to %s, expected at least %d arguments\n", 75c2cdf5c5SKevin Wolf argc-1, cmd, ct->argmin); 76c2cdf5c5SKevin Wolf } else if (ct->argmin == ct->argmax) { 77c2cdf5c5SKevin Wolf fprintf(stderr, 78c2cdf5c5SKevin Wolf "bad argument count %d to %s, expected %d arguments\n", 79c2cdf5c5SKevin Wolf argc-1, cmd, ct->argmin); 80c2cdf5c5SKevin Wolf } else { 81c2cdf5c5SKevin Wolf fprintf(stderr, 82c2cdf5c5SKevin Wolf "bad argument count %d to %s, expected between %d and %d arguments\n", 83c2cdf5c5SKevin Wolf argc-1, cmd, ct->argmin, ct->argmax); 84c2cdf5c5SKevin Wolf } 85c2cdf5c5SKevin Wolf return 0; 86c2cdf5c5SKevin Wolf } 87c2cdf5c5SKevin Wolf optind = 0; 884c7b7e9bSMax Reitz return ct->cfunc(blk, argc, argv); 89c2cdf5c5SKevin Wolf } 90c2cdf5c5SKevin Wolf 91c2cdf5c5SKevin Wolf static const cmdinfo_t *find_command(const char *cmd) 92c2cdf5c5SKevin Wolf { 93c2cdf5c5SKevin Wolf cmdinfo_t *ct; 94c2cdf5c5SKevin Wolf 95c2cdf5c5SKevin Wolf for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { 96c2cdf5c5SKevin Wolf if (strcmp(ct->name, cmd) == 0 || 97c2cdf5c5SKevin Wolf (ct->altname && strcmp(ct->altname, cmd) == 0)) 98c2cdf5c5SKevin Wolf { 99c2cdf5c5SKevin Wolf return (const cmdinfo_t *)ct; 100c2cdf5c5SKevin Wolf } 101c2cdf5c5SKevin Wolf } 102c2cdf5c5SKevin Wolf return NULL; 103c2cdf5c5SKevin Wolf } 104c2cdf5c5SKevin Wolf 1054694020dSStefan Hajnoczi /* Invoke fn() for commands with a matching prefix */ 1064694020dSStefan Hajnoczi void qemuio_complete_command(const char *input, 1074694020dSStefan Hajnoczi void (*fn)(const char *cmd, void *opaque), 1084694020dSStefan Hajnoczi void *opaque) 1094694020dSStefan Hajnoczi { 1104694020dSStefan Hajnoczi cmdinfo_t *ct; 1114694020dSStefan Hajnoczi size_t input_len = strlen(input); 1124694020dSStefan Hajnoczi 1134694020dSStefan Hajnoczi for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { 1144694020dSStefan Hajnoczi if (strncmp(input, ct->name, input_len) == 0) { 1154694020dSStefan Hajnoczi fn(ct->name, opaque); 1164694020dSStefan Hajnoczi } 1174694020dSStefan Hajnoczi } 1184694020dSStefan Hajnoczi } 1194694020dSStefan Hajnoczi 120c2cdf5c5SKevin Wolf static char **breakline(char *input, int *count) 121c2cdf5c5SKevin Wolf { 122c2cdf5c5SKevin Wolf int c = 0; 123c2cdf5c5SKevin Wolf char *p; 1245839e53bSMarkus Armbruster char **rval = g_new0(char *, 1); 125c2cdf5c5SKevin Wolf 126c2cdf5c5SKevin Wolf while (rval && (p = qemu_strsep(&input, " ")) != NULL) { 127c2cdf5c5SKevin Wolf if (!*p) { 128c2cdf5c5SKevin Wolf continue; 129c2cdf5c5SKevin Wolf } 130c2cdf5c5SKevin Wolf c++; 13108193dd5SMarkus Armbruster rval = g_renew(char *, rval, (c + 1)); 132c2cdf5c5SKevin Wolf rval[c - 1] = p; 133c2cdf5c5SKevin Wolf rval[c] = NULL; 134c2cdf5c5SKevin Wolf } 135c2cdf5c5SKevin Wolf *count = c; 136c2cdf5c5SKevin Wolf return rval; 137c2cdf5c5SKevin Wolf } 138c2cdf5c5SKevin Wolf 139797ac58cSKevin Wolf static int64_t cvtnum(const char *s) 140797ac58cSKevin Wolf { 141797ac58cSKevin Wolf char *end; 142ef5a7885SJohn Snow int64_t ret; 143ef5a7885SJohn Snow 144ef5a7885SJohn Snow ret = qemu_strtosz_suffix(s, &end, QEMU_STRTOSZ_DEFSUFFIX_B); 145ef5a7885SJohn Snow if (*end != '\0') { 146ef5a7885SJohn Snow /* Detritus at the end of the string */ 147ef5a7885SJohn Snow return -EINVAL; 148ef5a7885SJohn Snow } 149ef5a7885SJohn Snow return ret; 150797ac58cSKevin Wolf } 151797ac58cSKevin Wolf 152a9ecfa00SJohn Snow static void print_cvtnum_err(int64_t rc, const char *arg) 153a9ecfa00SJohn Snow { 154a9ecfa00SJohn Snow switch (rc) { 155a9ecfa00SJohn Snow case -EINVAL: 156a9ecfa00SJohn Snow printf("Parsing error: non-numeric argument," 157a9ecfa00SJohn Snow " or extraneous/unrecognized suffix -- %s\n", arg); 158a9ecfa00SJohn Snow break; 159a9ecfa00SJohn Snow case -ERANGE: 160a9ecfa00SJohn Snow printf("Parsing error: argument too large -- %s\n", arg); 161a9ecfa00SJohn Snow break; 162a9ecfa00SJohn Snow default: 163a9ecfa00SJohn Snow printf("Parsing error: %s\n", arg); 164a9ecfa00SJohn Snow } 165a9ecfa00SJohn Snow } 166a9ecfa00SJohn Snow 1670b613881SKevin Wolf #define EXABYTES(x) ((long long)(x) << 60) 1680b613881SKevin Wolf #define PETABYTES(x) ((long long)(x) << 50) 1690b613881SKevin Wolf #define TERABYTES(x) ((long long)(x) << 40) 1700b613881SKevin Wolf #define GIGABYTES(x) ((long long)(x) << 30) 1710b613881SKevin Wolf #define MEGABYTES(x) ((long long)(x) << 20) 1720b613881SKevin Wolf #define KILOBYTES(x) ((long long)(x) << 10) 1730b613881SKevin Wolf 1740b613881SKevin Wolf #define TO_EXABYTES(x) ((x) / EXABYTES(1)) 1750b613881SKevin Wolf #define TO_PETABYTES(x) ((x) / PETABYTES(1)) 1760b613881SKevin Wolf #define TO_TERABYTES(x) ((x) / TERABYTES(1)) 1770b613881SKevin Wolf #define TO_GIGABYTES(x) ((x) / GIGABYTES(1)) 1780b613881SKevin Wolf #define TO_MEGABYTES(x) ((x) / MEGABYTES(1)) 1790b613881SKevin Wolf #define TO_KILOBYTES(x) ((x) / KILOBYTES(1)) 1800b613881SKevin Wolf 1810b613881SKevin Wolf static void cvtstr(double value, char *str, size_t size) 1820b613881SKevin Wolf { 1830b613881SKevin Wolf char *trim; 1840b613881SKevin Wolf const char *suffix; 1850b613881SKevin Wolf 1860b613881SKevin Wolf if (value >= EXABYTES(1)) { 1870b613881SKevin Wolf suffix = " EiB"; 1880b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_EXABYTES(value)); 1890b613881SKevin Wolf } else if (value >= PETABYTES(1)) { 1900b613881SKevin Wolf suffix = " PiB"; 1910b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_PETABYTES(value)); 1920b613881SKevin Wolf } else if (value >= TERABYTES(1)) { 1930b613881SKevin Wolf suffix = " TiB"; 1940b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_TERABYTES(value)); 1950b613881SKevin Wolf } else if (value >= GIGABYTES(1)) { 1960b613881SKevin Wolf suffix = " GiB"; 1970b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_GIGABYTES(value)); 1980b613881SKevin Wolf } else if (value >= MEGABYTES(1)) { 1990b613881SKevin Wolf suffix = " MiB"; 2000b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_MEGABYTES(value)); 2010b613881SKevin Wolf } else if (value >= KILOBYTES(1)) { 2020b613881SKevin Wolf suffix = " KiB"; 2030b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_KILOBYTES(value)); 2040b613881SKevin Wolf } else { 2050b613881SKevin Wolf suffix = " bytes"; 2060b613881SKevin Wolf snprintf(str, size - 6, "%f", value); 2070b613881SKevin Wolf } 2080b613881SKevin Wolf 2090b613881SKevin Wolf trim = strstr(str, ".000"); 2100b613881SKevin Wolf if (trim) { 2110b613881SKevin Wolf strcpy(trim, suffix); 2120b613881SKevin Wolf } else { 2130b613881SKevin Wolf strcat(str, suffix); 2140b613881SKevin Wolf } 2150b613881SKevin Wolf } 2160b613881SKevin Wolf 2170b613881SKevin Wolf 2180b613881SKevin Wolf 2190b613881SKevin Wolf static struct timeval tsub(struct timeval t1, struct timeval t2) 2200b613881SKevin Wolf { 2210b613881SKevin Wolf t1.tv_usec -= t2.tv_usec; 2220b613881SKevin Wolf if (t1.tv_usec < 0) { 2230b613881SKevin Wolf t1.tv_usec += 1000000; 2240b613881SKevin Wolf t1.tv_sec--; 2250b613881SKevin Wolf } 2260b613881SKevin Wolf t1.tv_sec -= t2.tv_sec; 2270b613881SKevin Wolf return t1; 2280b613881SKevin Wolf } 2290b613881SKevin Wolf 2300b613881SKevin Wolf static double tdiv(double value, struct timeval tv) 2310b613881SKevin Wolf { 2320b613881SKevin Wolf return value / ((double)tv.tv_sec + ((double)tv.tv_usec / 1000000.0)); 2330b613881SKevin Wolf } 2340b613881SKevin Wolf 2350b613881SKevin Wolf #define HOURS(sec) ((sec) / (60 * 60)) 2360b613881SKevin Wolf #define MINUTES(sec) (((sec) % (60 * 60)) / 60) 2370b613881SKevin Wolf #define SECONDS(sec) ((sec) % 60) 2380b613881SKevin Wolf 2390b613881SKevin Wolf enum { 2400b613881SKevin Wolf DEFAULT_TIME = 0x0, 2410b613881SKevin Wolf TERSE_FIXED_TIME = 0x1, 2420b613881SKevin Wolf VERBOSE_FIXED_TIME = 0x2, 2430b613881SKevin Wolf }; 2440b613881SKevin Wolf 2450b613881SKevin Wolf static void timestr(struct timeval *tv, char *ts, size_t size, int format) 2460b613881SKevin Wolf { 2470b613881SKevin Wolf double usec = (double)tv->tv_usec / 1000000.0; 2480b613881SKevin Wolf 2490b613881SKevin Wolf if (format & TERSE_FIXED_TIME) { 2500b613881SKevin Wolf if (!HOURS(tv->tv_sec)) { 2510b613881SKevin Wolf snprintf(ts, size, "%u:%02u.%02u", 2520b613881SKevin Wolf (unsigned int) MINUTES(tv->tv_sec), 2530b613881SKevin Wolf (unsigned int) SECONDS(tv->tv_sec), 2540b613881SKevin Wolf (unsigned int) (usec * 100)); 2550b613881SKevin Wolf return; 2560b613881SKevin Wolf } 2570b613881SKevin Wolf format |= VERBOSE_FIXED_TIME; /* fallback if hours needed */ 2580b613881SKevin Wolf } 2590b613881SKevin Wolf 2600b613881SKevin Wolf if ((format & VERBOSE_FIXED_TIME) || tv->tv_sec) { 2610b613881SKevin Wolf snprintf(ts, size, "%u:%02u:%02u.%02u", 2620b613881SKevin Wolf (unsigned int) HOURS(tv->tv_sec), 2630b613881SKevin Wolf (unsigned int) MINUTES(tv->tv_sec), 2640b613881SKevin Wolf (unsigned int) SECONDS(tv->tv_sec), 2650b613881SKevin Wolf (unsigned int) (usec * 100)); 2660b613881SKevin Wolf } else { 2670b613881SKevin Wolf snprintf(ts, size, "0.%04u sec", (unsigned int) (usec * 10000)); 2680b613881SKevin Wolf } 2690b613881SKevin Wolf } 2700b613881SKevin Wolf 271797ac58cSKevin Wolf /* 272797ac58cSKevin Wolf * Parse the pattern argument to various sub-commands. 273797ac58cSKevin Wolf * 274797ac58cSKevin Wolf * Because the pattern is used as an argument to memset it must evaluate 275797ac58cSKevin Wolf * to an unsigned integer that fits into a single byte. 276797ac58cSKevin Wolf */ 277797ac58cSKevin Wolf static int parse_pattern(const char *arg) 278797ac58cSKevin Wolf { 279797ac58cSKevin Wolf char *endptr = NULL; 280797ac58cSKevin Wolf long pattern; 281797ac58cSKevin Wolf 282797ac58cSKevin Wolf pattern = strtol(arg, &endptr, 0); 283797ac58cSKevin Wolf if (pattern < 0 || pattern > UCHAR_MAX || *endptr != '\0') { 284797ac58cSKevin Wolf printf("%s is not a valid pattern byte\n", arg); 285797ac58cSKevin Wolf return -1; 286797ac58cSKevin Wolf } 287797ac58cSKevin Wolf 288797ac58cSKevin Wolf return pattern; 289797ac58cSKevin Wolf } 290797ac58cSKevin Wolf 291797ac58cSKevin Wolf /* 292797ac58cSKevin Wolf * Memory allocation helpers. 293797ac58cSKevin Wolf * 294797ac58cSKevin Wolf * Make sure memory is aligned by default, or purposefully misaligned if 295797ac58cSKevin Wolf * that is specified on the command line. 296797ac58cSKevin Wolf */ 297797ac58cSKevin Wolf 298797ac58cSKevin Wolf #define MISALIGN_OFFSET 16 2994c7b7e9bSMax Reitz static void *qemu_io_alloc(BlockBackend *blk, size_t len, int pattern) 300797ac58cSKevin Wolf { 301797ac58cSKevin Wolf void *buf; 302797ac58cSKevin Wolf 303797ac58cSKevin Wolf if (qemuio_misalign) { 304797ac58cSKevin Wolf len += MISALIGN_OFFSET; 305797ac58cSKevin Wolf } 3064c7b7e9bSMax Reitz buf = blk_blockalign(blk, len); 307797ac58cSKevin Wolf memset(buf, pattern, len); 308797ac58cSKevin Wolf if (qemuio_misalign) { 309797ac58cSKevin Wolf buf += MISALIGN_OFFSET; 310797ac58cSKevin Wolf } 311797ac58cSKevin Wolf return buf; 312797ac58cSKevin Wolf } 313797ac58cSKevin Wolf 314797ac58cSKevin Wolf static void qemu_io_free(void *p) 315797ac58cSKevin Wolf { 316797ac58cSKevin Wolf if (qemuio_misalign) { 317797ac58cSKevin Wolf p -= MISALIGN_OFFSET; 318797ac58cSKevin Wolf } 319797ac58cSKevin Wolf qemu_vfree(p); 320797ac58cSKevin Wolf } 321797ac58cSKevin Wolf 3229b0beaf3SJohn Snow static void dump_buffer(const void *buffer, int64_t offset, int64_t len) 323797ac58cSKevin Wolf { 3249b0beaf3SJohn Snow uint64_t i; 3259b0beaf3SJohn Snow int j; 326797ac58cSKevin Wolf const uint8_t *p; 327797ac58cSKevin Wolf 328797ac58cSKevin Wolf for (i = 0, p = buffer; i < len; i += 16) { 329797ac58cSKevin Wolf const uint8_t *s = p; 330797ac58cSKevin Wolf 331797ac58cSKevin Wolf printf("%08" PRIx64 ": ", offset + i); 332797ac58cSKevin Wolf for (j = 0; j < 16 && i + j < len; j++, p++) { 333797ac58cSKevin Wolf printf("%02x ", *p); 334797ac58cSKevin Wolf } 335797ac58cSKevin Wolf printf(" "); 336797ac58cSKevin Wolf for (j = 0; j < 16 && i + j < len; j++, s++) { 337797ac58cSKevin Wolf if (isalnum(*s)) { 338797ac58cSKevin Wolf printf("%c", *s); 339797ac58cSKevin Wolf } else { 340797ac58cSKevin Wolf printf("."); 341797ac58cSKevin Wolf } 342797ac58cSKevin Wolf } 343797ac58cSKevin Wolf printf("\n"); 344797ac58cSKevin Wolf } 345797ac58cSKevin Wolf } 346797ac58cSKevin Wolf 347797ac58cSKevin Wolf static void print_report(const char *op, struct timeval *t, int64_t offset, 3489b0beaf3SJohn Snow int64_t count, int64_t total, int cnt, int Cflag) 349797ac58cSKevin Wolf { 350797ac58cSKevin Wolf char s1[64], s2[64], ts[64]; 351797ac58cSKevin Wolf 352797ac58cSKevin Wolf timestr(t, ts, sizeof(ts), Cflag ? VERBOSE_FIXED_TIME : 0); 353797ac58cSKevin Wolf if (!Cflag) { 354797ac58cSKevin Wolf cvtstr((double)total, s1, sizeof(s1)); 355797ac58cSKevin Wolf cvtstr(tdiv((double)total, *t), s2, sizeof(s2)); 3569b0beaf3SJohn Snow printf("%s %"PRId64"/%"PRId64" bytes at offset %" PRId64 "\n", 357797ac58cSKevin Wolf op, total, count, offset); 358797ac58cSKevin Wolf printf("%s, %d ops; %s (%s/sec and %.4f ops/sec)\n", 359797ac58cSKevin Wolf s1, cnt, ts, s2, tdiv((double)cnt, *t)); 360797ac58cSKevin Wolf } else {/* bytes,ops,time,bytes/sec,ops/sec */ 3619b0beaf3SJohn Snow printf("%"PRId64",%d,%s,%.3f,%.3f\n", 362797ac58cSKevin Wolf total, cnt, ts, 363797ac58cSKevin Wolf tdiv((double)total, *t), 364797ac58cSKevin Wolf tdiv((double)cnt, *t)); 365797ac58cSKevin Wolf } 366797ac58cSKevin Wolf } 367797ac58cSKevin Wolf 368797ac58cSKevin Wolf /* 369797ac58cSKevin Wolf * Parse multiple length statements for vectored I/O, and construct an I/O 370797ac58cSKevin Wolf * vector matching it. 371797ac58cSKevin Wolf */ 372797ac58cSKevin Wolf static void * 3734c7b7e9bSMax Reitz create_iovec(BlockBackend *blk, QEMUIOVector *qiov, char **argv, int nr_iov, 374797ac58cSKevin Wolf int pattern) 375797ac58cSKevin Wolf { 376797ac58cSKevin Wolf size_t *sizes = g_new0(size_t, nr_iov); 377797ac58cSKevin Wolf size_t count = 0; 378797ac58cSKevin Wolf void *buf = NULL; 379797ac58cSKevin Wolf void *p; 380797ac58cSKevin Wolf int i; 381797ac58cSKevin Wolf 382797ac58cSKevin Wolf for (i = 0; i < nr_iov; i++) { 383797ac58cSKevin Wolf char *arg = argv[i]; 384797ac58cSKevin Wolf int64_t len; 385797ac58cSKevin Wolf 386797ac58cSKevin Wolf len = cvtnum(arg); 387797ac58cSKevin Wolf if (len < 0) { 388a9ecfa00SJohn Snow print_cvtnum_err(len, arg); 389797ac58cSKevin Wolf goto fail; 390797ac58cSKevin Wolf } 391797ac58cSKevin Wolf 392797ac58cSKevin Wolf /* should be SIZE_T_MAX, but that doesn't exist */ 393797ac58cSKevin Wolf if (len > INT_MAX) { 394a9ecfa00SJohn Snow printf("Argument '%s' exceeds maximum size %d\n", arg, INT_MAX); 395797ac58cSKevin Wolf goto fail; 396797ac58cSKevin Wolf } 397797ac58cSKevin Wolf 398797ac58cSKevin Wolf if (len & 0x1ff) { 399797ac58cSKevin Wolf printf("length argument %" PRId64 400797ac58cSKevin Wolf " is not sector aligned\n", len); 401797ac58cSKevin Wolf goto fail; 402797ac58cSKevin Wolf } 403797ac58cSKevin Wolf 404797ac58cSKevin Wolf sizes[i] = len; 405797ac58cSKevin Wolf count += len; 406797ac58cSKevin Wolf } 407797ac58cSKevin Wolf 408797ac58cSKevin Wolf qemu_iovec_init(qiov, nr_iov); 409797ac58cSKevin Wolf 4104c7b7e9bSMax Reitz buf = p = qemu_io_alloc(blk, count, pattern); 411797ac58cSKevin Wolf 412797ac58cSKevin Wolf for (i = 0; i < nr_iov; i++) { 413797ac58cSKevin Wolf qemu_iovec_add(qiov, p, sizes[i]); 414797ac58cSKevin Wolf p += sizes[i]; 415797ac58cSKevin Wolf } 416797ac58cSKevin Wolf 417797ac58cSKevin Wolf fail: 418797ac58cSKevin Wolf g_free(sizes); 419797ac58cSKevin Wolf return buf; 420797ac58cSKevin Wolf } 421797ac58cSKevin Wolf 4229b0beaf3SJohn Snow static int do_read(BlockBackend *blk, char *buf, int64_t offset, int64_t count, 4239b0beaf3SJohn Snow int64_t *total) 424797ac58cSKevin Wolf { 425797ac58cSKevin Wolf int ret; 426797ac58cSKevin Wolf 4279b0beaf3SJohn Snow if (count >> 9 > INT_MAX) { 4289b0beaf3SJohn Snow return -ERANGE; 4299b0beaf3SJohn Snow } 4309b0beaf3SJohn Snow 4314c7b7e9bSMax Reitz ret = blk_read(blk, offset >> 9, (uint8_t *)buf, count >> 9); 432797ac58cSKevin Wolf if (ret < 0) { 433797ac58cSKevin Wolf return ret; 434797ac58cSKevin Wolf } 435797ac58cSKevin Wolf *total = count; 436797ac58cSKevin Wolf return 1; 437797ac58cSKevin Wolf } 438797ac58cSKevin Wolf 4399b0beaf3SJohn Snow static int do_write(BlockBackend *blk, char *buf, int64_t offset, int64_t count, 4409b0beaf3SJohn Snow int64_t *total) 441797ac58cSKevin Wolf { 442797ac58cSKevin Wolf int ret; 443797ac58cSKevin Wolf 4449b0beaf3SJohn Snow if (count >> 9 > INT_MAX) { 4459b0beaf3SJohn Snow return -ERANGE; 4469b0beaf3SJohn Snow } 4479b0beaf3SJohn Snow 4484c7b7e9bSMax Reitz ret = blk_write(blk, offset >> 9, (uint8_t *)buf, count >> 9); 449797ac58cSKevin Wolf if (ret < 0) { 450797ac58cSKevin Wolf return ret; 451797ac58cSKevin Wolf } 452797ac58cSKevin Wolf *total = count; 453797ac58cSKevin Wolf return 1; 454797ac58cSKevin Wolf } 455797ac58cSKevin Wolf 4569b0beaf3SJohn Snow static int do_pread(BlockBackend *blk, char *buf, int64_t offset, 4579b0beaf3SJohn Snow int64_t count, int64_t *total) 458797ac58cSKevin Wolf { 4599b0beaf3SJohn Snow if (count > INT_MAX) { 4609b0beaf3SJohn Snow return -ERANGE; 4619b0beaf3SJohn Snow } 4629b0beaf3SJohn Snow 4634c7b7e9bSMax Reitz *total = blk_pread(blk, offset, (uint8_t *)buf, count); 464797ac58cSKevin Wolf if (*total < 0) { 465797ac58cSKevin Wolf return *total; 466797ac58cSKevin Wolf } 467797ac58cSKevin Wolf return 1; 468797ac58cSKevin Wolf } 469797ac58cSKevin Wolf 4709b0beaf3SJohn Snow static int do_pwrite(BlockBackend *blk, char *buf, int64_t offset, 4719b0beaf3SJohn Snow int64_t count, int64_t *total) 472797ac58cSKevin Wolf { 4739b0beaf3SJohn Snow if (count > INT_MAX) { 4749b0beaf3SJohn Snow return -ERANGE; 4759b0beaf3SJohn Snow } 4769b0beaf3SJohn Snow 4774c7b7e9bSMax Reitz *total = blk_pwrite(blk, offset, (uint8_t *)buf, count); 478797ac58cSKevin Wolf if (*total < 0) { 479797ac58cSKevin Wolf return *total; 480797ac58cSKevin Wolf } 481797ac58cSKevin Wolf return 1; 482797ac58cSKevin Wolf } 483797ac58cSKevin Wolf 484797ac58cSKevin Wolf typedef struct { 4854c7b7e9bSMax Reitz BlockBackend *blk; 486797ac58cSKevin Wolf int64_t offset; 4879b0beaf3SJohn Snow int64_t count; 4889b0beaf3SJohn Snow int64_t *total; 489797ac58cSKevin Wolf int ret; 490797ac58cSKevin Wolf bool done; 491797ac58cSKevin Wolf } CoWriteZeroes; 492797ac58cSKevin Wolf 493797ac58cSKevin Wolf static void coroutine_fn co_write_zeroes_entry(void *opaque) 494797ac58cSKevin Wolf { 495797ac58cSKevin Wolf CoWriteZeroes *data = opaque; 496797ac58cSKevin Wolf 4974c7b7e9bSMax Reitz data->ret = blk_co_write_zeroes(data->blk, data->offset / BDRV_SECTOR_SIZE, 498aa7bfbffSPeter Lieven data->count / BDRV_SECTOR_SIZE, 0); 499797ac58cSKevin Wolf data->done = true; 500797ac58cSKevin Wolf if (data->ret < 0) { 501797ac58cSKevin Wolf *data->total = data->ret; 502797ac58cSKevin Wolf return; 503797ac58cSKevin Wolf } 504797ac58cSKevin Wolf 505797ac58cSKevin Wolf *data->total = data->count; 506797ac58cSKevin Wolf } 507797ac58cSKevin Wolf 5089b0beaf3SJohn Snow static int do_co_write_zeroes(BlockBackend *blk, int64_t offset, int64_t count, 5099b0beaf3SJohn Snow int64_t *total) 510797ac58cSKevin Wolf { 511797ac58cSKevin Wolf Coroutine *co; 512797ac58cSKevin Wolf CoWriteZeroes data = { 5134c7b7e9bSMax Reitz .blk = blk, 514797ac58cSKevin Wolf .offset = offset, 515797ac58cSKevin Wolf .count = count, 516797ac58cSKevin Wolf .total = total, 517797ac58cSKevin Wolf .done = false, 518797ac58cSKevin Wolf }; 519797ac58cSKevin Wolf 5209b0beaf3SJohn Snow if (count >> BDRV_SECTOR_BITS > INT_MAX) { 5219b0beaf3SJohn Snow return -ERANGE; 5229b0beaf3SJohn Snow } 5239b0beaf3SJohn Snow 524797ac58cSKevin Wolf co = qemu_coroutine_create(co_write_zeroes_entry); 525797ac58cSKevin Wolf qemu_coroutine_enter(co, &data); 526797ac58cSKevin Wolf while (!data.done) { 5274c7b7e9bSMax Reitz aio_poll(blk_get_aio_context(blk), true); 528797ac58cSKevin Wolf } 529797ac58cSKevin Wolf if (data.ret < 0) { 530797ac58cSKevin Wolf return data.ret; 531797ac58cSKevin Wolf } else { 532797ac58cSKevin Wolf return 1; 533797ac58cSKevin Wolf } 534797ac58cSKevin Wolf } 535797ac58cSKevin Wolf 5364c7b7e9bSMax Reitz static int do_write_compressed(BlockBackend *blk, char *buf, int64_t offset, 5379b0beaf3SJohn Snow int64_t count, int64_t *total) 538797ac58cSKevin Wolf { 539797ac58cSKevin Wolf int ret; 540797ac58cSKevin Wolf 5419b0beaf3SJohn Snow if (count >> 9 > INT_MAX) { 5429b0beaf3SJohn Snow return -ERANGE; 5439b0beaf3SJohn Snow } 5449b0beaf3SJohn Snow 5454c7b7e9bSMax Reitz ret = blk_write_compressed(blk, offset >> 9, (uint8_t *)buf, count >> 9); 546797ac58cSKevin Wolf if (ret < 0) { 547797ac58cSKevin Wolf return ret; 548797ac58cSKevin Wolf } 549797ac58cSKevin Wolf *total = count; 550797ac58cSKevin Wolf return 1; 551797ac58cSKevin Wolf } 552797ac58cSKevin Wolf 5534c7b7e9bSMax Reitz static int do_load_vmstate(BlockBackend *blk, char *buf, int64_t offset, 5549b0beaf3SJohn Snow int64_t count, int64_t *total) 555797ac58cSKevin Wolf { 5569b0beaf3SJohn Snow if (count > INT_MAX) { 5579b0beaf3SJohn Snow return -ERANGE; 5589b0beaf3SJohn Snow } 5599b0beaf3SJohn Snow 5604c7b7e9bSMax Reitz *total = blk_load_vmstate(blk, (uint8_t *)buf, offset, count); 561797ac58cSKevin Wolf if (*total < 0) { 562797ac58cSKevin Wolf return *total; 563797ac58cSKevin Wolf } 564797ac58cSKevin Wolf return 1; 565797ac58cSKevin Wolf } 566797ac58cSKevin Wolf 5674c7b7e9bSMax Reitz static int do_save_vmstate(BlockBackend *blk, char *buf, int64_t offset, 5689b0beaf3SJohn Snow int64_t count, int64_t *total) 569797ac58cSKevin Wolf { 5709b0beaf3SJohn Snow if (count > INT_MAX) { 5719b0beaf3SJohn Snow return -ERANGE; 5729b0beaf3SJohn Snow } 5739b0beaf3SJohn Snow 5744c7b7e9bSMax Reitz *total = blk_save_vmstate(blk, (uint8_t *)buf, offset, count); 575797ac58cSKevin Wolf if (*total < 0) { 576797ac58cSKevin Wolf return *total; 577797ac58cSKevin Wolf } 578797ac58cSKevin Wolf return 1; 579797ac58cSKevin Wolf } 580797ac58cSKevin Wolf 581797ac58cSKevin Wolf #define NOT_DONE 0x7fffffff 582797ac58cSKevin Wolf static void aio_rw_done(void *opaque, int ret) 583797ac58cSKevin Wolf { 584797ac58cSKevin Wolf *(int *)opaque = ret; 585797ac58cSKevin Wolf } 586797ac58cSKevin Wolf 5874c7b7e9bSMax Reitz static int do_aio_readv(BlockBackend *blk, QEMUIOVector *qiov, 588797ac58cSKevin Wolf int64_t offset, int *total) 589797ac58cSKevin Wolf { 590797ac58cSKevin Wolf int async_ret = NOT_DONE; 591797ac58cSKevin Wolf 5924c7b7e9bSMax Reitz blk_aio_readv(blk, offset >> 9, qiov, qiov->size >> 9, 593797ac58cSKevin Wolf aio_rw_done, &async_ret); 594797ac58cSKevin Wolf while (async_ret == NOT_DONE) { 595797ac58cSKevin Wolf main_loop_wait(false); 596797ac58cSKevin Wolf } 597797ac58cSKevin Wolf 598797ac58cSKevin Wolf *total = qiov->size; 599797ac58cSKevin Wolf return async_ret < 0 ? async_ret : 1; 600797ac58cSKevin Wolf } 601797ac58cSKevin Wolf 6024c7b7e9bSMax Reitz static int do_aio_writev(BlockBackend *blk, QEMUIOVector *qiov, 603797ac58cSKevin Wolf int64_t offset, int *total) 604797ac58cSKevin Wolf { 605797ac58cSKevin Wolf int async_ret = NOT_DONE; 606797ac58cSKevin Wolf 6074c7b7e9bSMax Reitz blk_aio_writev(blk, offset >> 9, qiov, qiov->size >> 9, 608797ac58cSKevin Wolf aio_rw_done, &async_ret); 609797ac58cSKevin Wolf while (async_ret == NOT_DONE) { 610797ac58cSKevin Wolf main_loop_wait(false); 611797ac58cSKevin Wolf } 612797ac58cSKevin Wolf 613797ac58cSKevin Wolf *total = qiov->size; 614797ac58cSKevin Wolf return async_ret < 0 ? async_ret : 1; 615797ac58cSKevin Wolf } 616797ac58cSKevin Wolf 617797ac58cSKevin Wolf struct multiwrite_async_ret { 618797ac58cSKevin Wolf int num_done; 619797ac58cSKevin Wolf int error; 620797ac58cSKevin Wolf }; 621797ac58cSKevin Wolf 622797ac58cSKevin Wolf static void multiwrite_cb(void *opaque, int ret) 623797ac58cSKevin Wolf { 624797ac58cSKevin Wolf struct multiwrite_async_ret *async_ret = opaque; 625797ac58cSKevin Wolf 626797ac58cSKevin Wolf async_ret->num_done++; 627797ac58cSKevin Wolf if (ret < 0) { 628797ac58cSKevin Wolf async_ret->error = ret; 629797ac58cSKevin Wolf } 630797ac58cSKevin Wolf } 631797ac58cSKevin Wolf 6324c7b7e9bSMax Reitz static int do_aio_multiwrite(BlockBackend *blk, BlockRequest* reqs, 633797ac58cSKevin Wolf int num_reqs, int *total) 634797ac58cSKevin Wolf { 635797ac58cSKevin Wolf int i, ret; 636797ac58cSKevin Wolf struct multiwrite_async_ret async_ret = { 637797ac58cSKevin Wolf .num_done = 0, 638797ac58cSKevin Wolf .error = 0, 639797ac58cSKevin Wolf }; 640797ac58cSKevin Wolf 641797ac58cSKevin Wolf *total = 0; 642797ac58cSKevin Wolf for (i = 0; i < num_reqs; i++) { 643797ac58cSKevin Wolf reqs[i].cb = multiwrite_cb; 644797ac58cSKevin Wolf reqs[i].opaque = &async_ret; 645797ac58cSKevin Wolf *total += reqs[i].qiov->size; 646797ac58cSKevin Wolf } 647797ac58cSKevin Wolf 6484c7b7e9bSMax Reitz ret = blk_aio_multiwrite(blk, reqs, num_reqs); 649797ac58cSKevin Wolf if (ret < 0) { 650797ac58cSKevin Wolf return ret; 651797ac58cSKevin Wolf } 652797ac58cSKevin Wolf 653797ac58cSKevin Wolf while (async_ret.num_done < num_reqs) { 654797ac58cSKevin Wolf main_loop_wait(false); 655797ac58cSKevin Wolf } 656797ac58cSKevin Wolf 657797ac58cSKevin Wolf return async_ret.error < 0 ? async_ret.error : 1; 658797ac58cSKevin Wolf } 659797ac58cSKevin Wolf 660797ac58cSKevin Wolf static void read_help(void) 661797ac58cSKevin Wolf { 662797ac58cSKevin Wolf printf( 663797ac58cSKevin Wolf "\n" 664797ac58cSKevin Wolf " reads a range of bytes from the given offset\n" 665797ac58cSKevin Wolf "\n" 666797ac58cSKevin Wolf " Example:\n" 667797ac58cSKevin Wolf " 'read -v 512 1k' - dumps 1 kilobyte read from 512 bytes into the file\n" 668797ac58cSKevin Wolf "\n" 669797ac58cSKevin Wolf " Reads a segment of the currently open file, optionally dumping it to the\n" 670797ac58cSKevin Wolf " standard output stream (with -v option) for subsequent inspection.\n" 671797ac58cSKevin Wolf " -b, -- read from the VM state rather than the virtual disk\n" 672797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 673797ac58cSKevin Wolf " -l, -- length for pattern verification (only with -P)\n" 6744c7b7e9bSMax Reitz " -p, -- use blk_pread to read the file\n" 675797ac58cSKevin Wolf " -P, -- use a pattern to verify read data\n" 676797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 677797ac58cSKevin Wolf " -s, -- start offset for pattern verification (only with -P)\n" 678797ac58cSKevin Wolf " -v, -- dump buffer to standard output\n" 679797ac58cSKevin Wolf "\n"); 680797ac58cSKevin Wolf } 681797ac58cSKevin Wolf 6824c7b7e9bSMax Reitz static int read_f(BlockBackend *blk, int argc, char **argv); 683797ac58cSKevin Wolf 684797ac58cSKevin Wolf static const cmdinfo_t read_cmd = { 685797ac58cSKevin Wolf .name = "read", 686797ac58cSKevin Wolf .altname = "r", 687797ac58cSKevin Wolf .cfunc = read_f, 688797ac58cSKevin Wolf .argmin = 2, 689797ac58cSKevin Wolf .argmax = -1, 690797ac58cSKevin Wolf .args = "[-abCpqv] [-P pattern [-s off] [-l len]] off len", 691797ac58cSKevin Wolf .oneline = "reads a number of bytes at a specified offset", 692797ac58cSKevin Wolf .help = read_help, 693797ac58cSKevin Wolf }; 694797ac58cSKevin Wolf 6954c7b7e9bSMax Reitz static int read_f(BlockBackend *blk, int argc, char **argv) 696797ac58cSKevin Wolf { 697797ac58cSKevin Wolf struct timeval t1, t2; 698797ac58cSKevin Wolf int Cflag = 0, pflag = 0, qflag = 0, vflag = 0; 699797ac58cSKevin Wolf int Pflag = 0, sflag = 0, lflag = 0, bflag = 0; 700797ac58cSKevin Wolf int c, cnt; 701797ac58cSKevin Wolf char *buf; 702797ac58cSKevin Wolf int64_t offset; 7039b0beaf3SJohn Snow int64_t count; 704797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 7059b0beaf3SJohn Snow int64_t total = 0; 7069b0beaf3SJohn Snow int pattern = 0; 7079b0beaf3SJohn Snow int64_t pattern_offset = 0, pattern_count = 0; 708797ac58cSKevin Wolf 709b062ad86SEric Blake while ((c = getopt(argc, argv, "bCl:pP:qs:v")) != -1) { 710797ac58cSKevin Wolf switch (c) { 711797ac58cSKevin Wolf case 'b': 712797ac58cSKevin Wolf bflag = 1; 713797ac58cSKevin Wolf break; 714797ac58cSKevin Wolf case 'C': 715797ac58cSKevin Wolf Cflag = 1; 716797ac58cSKevin Wolf break; 717797ac58cSKevin Wolf case 'l': 718797ac58cSKevin Wolf lflag = 1; 719797ac58cSKevin Wolf pattern_count = cvtnum(optarg); 720797ac58cSKevin Wolf if (pattern_count < 0) { 721a9ecfa00SJohn Snow print_cvtnum_err(pattern_count, optarg); 722797ac58cSKevin Wolf return 0; 723797ac58cSKevin Wolf } 724797ac58cSKevin Wolf break; 725797ac58cSKevin Wolf case 'p': 726797ac58cSKevin Wolf pflag = 1; 727797ac58cSKevin Wolf break; 728797ac58cSKevin Wolf case 'P': 729797ac58cSKevin Wolf Pflag = 1; 730797ac58cSKevin Wolf pattern = parse_pattern(optarg); 731797ac58cSKevin Wolf if (pattern < 0) { 732797ac58cSKevin Wolf return 0; 733797ac58cSKevin Wolf } 734797ac58cSKevin Wolf break; 735797ac58cSKevin Wolf case 'q': 736797ac58cSKevin Wolf qflag = 1; 737797ac58cSKevin Wolf break; 738797ac58cSKevin Wolf case 's': 739797ac58cSKevin Wolf sflag = 1; 740797ac58cSKevin Wolf pattern_offset = cvtnum(optarg); 741797ac58cSKevin Wolf if (pattern_offset < 0) { 742a9ecfa00SJohn Snow print_cvtnum_err(pattern_offset, optarg); 743797ac58cSKevin Wolf return 0; 744797ac58cSKevin Wolf } 745797ac58cSKevin Wolf break; 746797ac58cSKevin Wolf case 'v': 747797ac58cSKevin Wolf vflag = 1; 748797ac58cSKevin Wolf break; 749797ac58cSKevin Wolf default: 750c2cdf5c5SKevin Wolf return qemuio_command_usage(&read_cmd); 751797ac58cSKevin Wolf } 752797ac58cSKevin Wolf } 753797ac58cSKevin Wolf 754797ac58cSKevin Wolf if (optind != argc - 2) { 755c2cdf5c5SKevin Wolf return qemuio_command_usage(&read_cmd); 756797ac58cSKevin Wolf } 757797ac58cSKevin Wolf 758797ac58cSKevin Wolf if (bflag && pflag) { 759797ac58cSKevin Wolf printf("-b and -p cannot be specified at the same time\n"); 760797ac58cSKevin Wolf return 0; 761797ac58cSKevin Wolf } 762797ac58cSKevin Wolf 763797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 764797ac58cSKevin Wolf if (offset < 0) { 765a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 766797ac58cSKevin Wolf return 0; 767797ac58cSKevin Wolf } 768797ac58cSKevin Wolf 769797ac58cSKevin Wolf optind++; 770797ac58cSKevin Wolf count = cvtnum(argv[optind]); 771797ac58cSKevin Wolf if (count < 0) { 772a9ecfa00SJohn Snow print_cvtnum_err(count, argv[optind]); 773797ac58cSKevin Wolf return 0; 7749b0beaf3SJohn Snow } else if (count > SIZE_MAX) { 7759b0beaf3SJohn Snow printf("length cannot exceed %" PRIu64 ", given %s\n", 7769b0beaf3SJohn Snow (uint64_t) SIZE_MAX, argv[optind]); 7779b0beaf3SJohn Snow return 0; 778797ac58cSKevin Wolf } 779797ac58cSKevin Wolf 780797ac58cSKevin Wolf if (!Pflag && (lflag || sflag)) { 781c2cdf5c5SKevin Wolf return qemuio_command_usage(&read_cmd); 782797ac58cSKevin Wolf } 783797ac58cSKevin Wolf 784797ac58cSKevin Wolf if (!lflag) { 785797ac58cSKevin Wolf pattern_count = count - pattern_offset; 786797ac58cSKevin Wolf } 787797ac58cSKevin Wolf 788797ac58cSKevin Wolf if ((pattern_count < 0) || (pattern_count + pattern_offset > count)) { 789797ac58cSKevin Wolf printf("pattern verification range exceeds end of read data\n"); 790797ac58cSKevin Wolf return 0; 791797ac58cSKevin Wolf } 792797ac58cSKevin Wolf 793797ac58cSKevin Wolf if (!pflag) { 794797ac58cSKevin Wolf if (offset & 0x1ff) { 795797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 796797ac58cSKevin Wolf offset); 797797ac58cSKevin Wolf return 0; 798797ac58cSKevin Wolf } 799797ac58cSKevin Wolf if (count & 0x1ff) { 8009b0beaf3SJohn Snow printf("count %"PRId64" is not sector aligned\n", 801797ac58cSKevin Wolf count); 802797ac58cSKevin Wolf return 0; 803797ac58cSKevin Wolf } 804797ac58cSKevin Wolf } 805797ac58cSKevin Wolf 8064c7b7e9bSMax Reitz buf = qemu_io_alloc(blk, count, 0xab); 807797ac58cSKevin Wolf 808797ac58cSKevin Wolf gettimeofday(&t1, NULL); 809797ac58cSKevin Wolf if (pflag) { 8104c7b7e9bSMax Reitz cnt = do_pread(blk, buf, offset, count, &total); 811797ac58cSKevin Wolf } else if (bflag) { 8124c7b7e9bSMax Reitz cnt = do_load_vmstate(blk, buf, offset, count, &total); 813797ac58cSKevin Wolf } else { 8144c7b7e9bSMax Reitz cnt = do_read(blk, buf, offset, count, &total); 815797ac58cSKevin Wolf } 816797ac58cSKevin Wolf gettimeofday(&t2, NULL); 817797ac58cSKevin Wolf 818797ac58cSKevin Wolf if (cnt < 0) { 819797ac58cSKevin Wolf printf("read failed: %s\n", strerror(-cnt)); 820797ac58cSKevin Wolf goto out; 821797ac58cSKevin Wolf } 822797ac58cSKevin Wolf 823797ac58cSKevin Wolf if (Pflag) { 824797ac58cSKevin Wolf void *cmp_buf = g_malloc(pattern_count); 825797ac58cSKevin Wolf memset(cmp_buf, pattern, pattern_count); 826797ac58cSKevin Wolf if (memcmp(buf + pattern_offset, cmp_buf, pattern_count)) { 827797ac58cSKevin Wolf printf("Pattern verification failed at offset %" 8289b0beaf3SJohn Snow PRId64 ", %"PRId64" bytes\n", 829797ac58cSKevin Wolf offset + pattern_offset, pattern_count); 830797ac58cSKevin Wolf } 831797ac58cSKevin Wolf g_free(cmp_buf); 832797ac58cSKevin Wolf } 833797ac58cSKevin Wolf 834797ac58cSKevin Wolf if (qflag) { 835797ac58cSKevin Wolf goto out; 836797ac58cSKevin Wolf } 837797ac58cSKevin Wolf 838797ac58cSKevin Wolf if (vflag) { 839797ac58cSKevin Wolf dump_buffer(buf, offset, count); 840797ac58cSKevin Wolf } 841797ac58cSKevin Wolf 842797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 843797ac58cSKevin Wolf t2 = tsub(t2, t1); 844797ac58cSKevin Wolf print_report("read", &t2, offset, count, total, cnt, Cflag); 845797ac58cSKevin Wolf 846797ac58cSKevin Wolf out: 847797ac58cSKevin Wolf qemu_io_free(buf); 848797ac58cSKevin Wolf 849797ac58cSKevin Wolf return 0; 850797ac58cSKevin Wolf } 851797ac58cSKevin Wolf 852797ac58cSKevin Wolf static void readv_help(void) 853797ac58cSKevin Wolf { 854797ac58cSKevin Wolf printf( 855797ac58cSKevin Wolf "\n" 856797ac58cSKevin Wolf " reads a range of bytes from the given offset into multiple buffers\n" 857797ac58cSKevin Wolf "\n" 858797ac58cSKevin Wolf " Example:\n" 859797ac58cSKevin Wolf " 'readv -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n" 860797ac58cSKevin Wolf "\n" 861797ac58cSKevin Wolf " Reads a segment of the currently open file, optionally dumping it to the\n" 862797ac58cSKevin Wolf " standard output stream (with -v option) for subsequent inspection.\n" 863797ac58cSKevin Wolf " Uses multiple iovec buffers if more than one byte range is specified.\n" 864797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 865797ac58cSKevin Wolf " -P, -- use a pattern to verify read data\n" 866797ac58cSKevin Wolf " -v, -- dump buffer to standard output\n" 867797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 868797ac58cSKevin Wolf "\n"); 869797ac58cSKevin Wolf } 870797ac58cSKevin Wolf 8714c7b7e9bSMax Reitz static int readv_f(BlockBackend *blk, int argc, char **argv); 872797ac58cSKevin Wolf 873797ac58cSKevin Wolf static const cmdinfo_t readv_cmd = { 874797ac58cSKevin Wolf .name = "readv", 875797ac58cSKevin Wolf .cfunc = readv_f, 876797ac58cSKevin Wolf .argmin = 2, 877797ac58cSKevin Wolf .argmax = -1, 878797ac58cSKevin Wolf .args = "[-Cqv] [-P pattern ] off len [len..]", 879797ac58cSKevin Wolf .oneline = "reads a number of bytes at a specified offset", 880797ac58cSKevin Wolf .help = readv_help, 881797ac58cSKevin Wolf }; 882797ac58cSKevin Wolf 8834c7b7e9bSMax Reitz static int readv_f(BlockBackend *blk, int argc, char **argv) 884797ac58cSKevin Wolf { 885797ac58cSKevin Wolf struct timeval t1, t2; 886797ac58cSKevin Wolf int Cflag = 0, qflag = 0, vflag = 0; 887797ac58cSKevin Wolf int c, cnt; 888797ac58cSKevin Wolf char *buf; 889797ac58cSKevin Wolf int64_t offset; 890797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 891797ac58cSKevin Wolf int total = 0; 892797ac58cSKevin Wolf int nr_iov; 893797ac58cSKevin Wolf QEMUIOVector qiov; 894797ac58cSKevin Wolf int pattern = 0; 895797ac58cSKevin Wolf int Pflag = 0; 896797ac58cSKevin Wolf 897b062ad86SEric Blake while ((c = getopt(argc, argv, "CP:qv")) != -1) { 898797ac58cSKevin Wolf switch (c) { 899797ac58cSKevin Wolf case 'C': 900797ac58cSKevin Wolf Cflag = 1; 901797ac58cSKevin Wolf break; 902797ac58cSKevin Wolf case 'P': 903797ac58cSKevin Wolf Pflag = 1; 904797ac58cSKevin Wolf pattern = parse_pattern(optarg); 905797ac58cSKevin Wolf if (pattern < 0) { 906797ac58cSKevin Wolf return 0; 907797ac58cSKevin Wolf } 908797ac58cSKevin Wolf break; 909797ac58cSKevin Wolf case 'q': 910797ac58cSKevin Wolf qflag = 1; 911797ac58cSKevin Wolf break; 912797ac58cSKevin Wolf case 'v': 913797ac58cSKevin Wolf vflag = 1; 914797ac58cSKevin Wolf break; 915797ac58cSKevin Wolf default: 916c2cdf5c5SKevin Wolf return qemuio_command_usage(&readv_cmd); 917797ac58cSKevin Wolf } 918797ac58cSKevin Wolf } 919797ac58cSKevin Wolf 920797ac58cSKevin Wolf if (optind > argc - 2) { 921c2cdf5c5SKevin Wolf return qemuio_command_usage(&readv_cmd); 922797ac58cSKevin Wolf } 923797ac58cSKevin Wolf 924797ac58cSKevin Wolf 925797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 926797ac58cSKevin Wolf if (offset < 0) { 927a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 928797ac58cSKevin Wolf return 0; 929797ac58cSKevin Wolf } 930797ac58cSKevin Wolf optind++; 931797ac58cSKevin Wolf 932797ac58cSKevin Wolf if (offset & 0x1ff) { 933797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 934797ac58cSKevin Wolf offset); 935797ac58cSKevin Wolf return 0; 936797ac58cSKevin Wolf } 937797ac58cSKevin Wolf 938797ac58cSKevin Wolf nr_iov = argc - optind; 9394c7b7e9bSMax Reitz buf = create_iovec(blk, &qiov, &argv[optind], nr_iov, 0xab); 940797ac58cSKevin Wolf if (buf == NULL) { 941797ac58cSKevin Wolf return 0; 942797ac58cSKevin Wolf } 943797ac58cSKevin Wolf 944797ac58cSKevin Wolf gettimeofday(&t1, NULL); 9454c7b7e9bSMax Reitz cnt = do_aio_readv(blk, &qiov, offset, &total); 946797ac58cSKevin Wolf gettimeofday(&t2, NULL); 947797ac58cSKevin Wolf 948797ac58cSKevin Wolf if (cnt < 0) { 949797ac58cSKevin Wolf printf("readv failed: %s\n", strerror(-cnt)); 950797ac58cSKevin Wolf goto out; 951797ac58cSKevin Wolf } 952797ac58cSKevin Wolf 953797ac58cSKevin Wolf if (Pflag) { 954797ac58cSKevin Wolf void *cmp_buf = g_malloc(qiov.size); 955797ac58cSKevin Wolf memset(cmp_buf, pattern, qiov.size); 956797ac58cSKevin Wolf if (memcmp(buf, cmp_buf, qiov.size)) { 957797ac58cSKevin Wolf printf("Pattern verification failed at offset %" 958797ac58cSKevin Wolf PRId64 ", %zd bytes\n", offset, qiov.size); 959797ac58cSKevin Wolf } 960797ac58cSKevin Wolf g_free(cmp_buf); 961797ac58cSKevin Wolf } 962797ac58cSKevin Wolf 963797ac58cSKevin Wolf if (qflag) { 964797ac58cSKevin Wolf goto out; 965797ac58cSKevin Wolf } 966797ac58cSKevin Wolf 967797ac58cSKevin Wolf if (vflag) { 968797ac58cSKevin Wolf dump_buffer(buf, offset, qiov.size); 969797ac58cSKevin Wolf } 970797ac58cSKevin Wolf 971797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 972797ac58cSKevin Wolf t2 = tsub(t2, t1); 973797ac58cSKevin Wolf print_report("read", &t2, offset, qiov.size, total, cnt, Cflag); 974797ac58cSKevin Wolf 975797ac58cSKevin Wolf out: 976797ac58cSKevin Wolf qemu_iovec_destroy(&qiov); 977797ac58cSKevin Wolf qemu_io_free(buf); 978797ac58cSKevin Wolf return 0; 979797ac58cSKevin Wolf } 980797ac58cSKevin Wolf 981797ac58cSKevin Wolf static void write_help(void) 982797ac58cSKevin Wolf { 983797ac58cSKevin Wolf printf( 984797ac58cSKevin Wolf "\n" 985797ac58cSKevin Wolf " writes a range of bytes from the given offset\n" 986797ac58cSKevin Wolf "\n" 987797ac58cSKevin Wolf " Example:\n" 988797ac58cSKevin Wolf " 'write 512 1k' - writes 1 kilobyte at 512 bytes into the open file\n" 989797ac58cSKevin Wolf "\n" 990797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 991797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd).\n" 992797ac58cSKevin Wolf " -b, -- write to the VM state rather than the virtual disk\n" 9934c7b7e9bSMax Reitz " -c, -- write compressed data with blk_write_compressed\n" 9944c7b7e9bSMax Reitz " -p, -- use blk_pwrite to write the file\n" 995797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 996797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 997797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 9984c7b7e9bSMax Reitz " -z, -- write zeroes using blk_co_write_zeroes\n" 999797ac58cSKevin Wolf "\n"); 1000797ac58cSKevin Wolf } 1001797ac58cSKevin Wolf 10024c7b7e9bSMax Reitz static int write_f(BlockBackend *blk, int argc, char **argv); 1003797ac58cSKevin Wolf 1004797ac58cSKevin Wolf static const cmdinfo_t write_cmd = { 1005797ac58cSKevin Wolf .name = "write", 1006797ac58cSKevin Wolf .altname = "w", 1007797ac58cSKevin Wolf .cfunc = write_f, 1008797ac58cSKevin Wolf .argmin = 2, 1009797ac58cSKevin Wolf .argmax = -1, 1010797ac58cSKevin Wolf .args = "[-bcCpqz] [-P pattern ] off len", 1011797ac58cSKevin Wolf .oneline = "writes a number of bytes at a specified offset", 1012797ac58cSKevin Wolf .help = write_help, 1013797ac58cSKevin Wolf }; 1014797ac58cSKevin Wolf 10154c7b7e9bSMax Reitz static int write_f(BlockBackend *blk, int argc, char **argv) 1016797ac58cSKevin Wolf { 1017797ac58cSKevin Wolf struct timeval t1, t2; 1018797ac58cSKevin Wolf int Cflag = 0, pflag = 0, qflag = 0, bflag = 0, Pflag = 0, zflag = 0; 1019797ac58cSKevin Wolf int cflag = 0; 1020797ac58cSKevin Wolf int c, cnt; 1021797ac58cSKevin Wolf char *buf = NULL; 1022797ac58cSKevin Wolf int64_t offset; 10239b0beaf3SJohn Snow int64_t count; 1024797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 10259b0beaf3SJohn Snow int64_t total = 0; 1026797ac58cSKevin Wolf int pattern = 0xcd; 1027797ac58cSKevin Wolf 1028b062ad86SEric Blake while ((c = getopt(argc, argv, "bcCpP:qz")) != -1) { 1029797ac58cSKevin Wolf switch (c) { 1030797ac58cSKevin Wolf case 'b': 1031797ac58cSKevin Wolf bflag = 1; 1032797ac58cSKevin Wolf break; 1033797ac58cSKevin Wolf case 'c': 1034797ac58cSKevin Wolf cflag = 1; 1035797ac58cSKevin Wolf break; 1036797ac58cSKevin Wolf case 'C': 1037797ac58cSKevin Wolf Cflag = 1; 1038797ac58cSKevin Wolf break; 1039797ac58cSKevin Wolf case 'p': 1040797ac58cSKevin Wolf pflag = 1; 1041797ac58cSKevin Wolf break; 1042797ac58cSKevin Wolf case 'P': 1043797ac58cSKevin Wolf Pflag = 1; 1044797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1045797ac58cSKevin Wolf if (pattern < 0) { 1046797ac58cSKevin Wolf return 0; 1047797ac58cSKevin Wolf } 1048797ac58cSKevin Wolf break; 1049797ac58cSKevin Wolf case 'q': 1050797ac58cSKevin Wolf qflag = 1; 1051797ac58cSKevin Wolf break; 1052797ac58cSKevin Wolf case 'z': 1053797ac58cSKevin Wolf zflag = 1; 1054797ac58cSKevin Wolf break; 1055797ac58cSKevin Wolf default: 1056c2cdf5c5SKevin Wolf return qemuio_command_usage(&write_cmd); 1057797ac58cSKevin Wolf } 1058797ac58cSKevin Wolf } 1059797ac58cSKevin Wolf 1060797ac58cSKevin Wolf if (optind != argc - 2) { 1061c2cdf5c5SKevin Wolf return qemuio_command_usage(&write_cmd); 1062797ac58cSKevin Wolf } 1063797ac58cSKevin Wolf 1064797ac58cSKevin Wolf if (bflag + pflag + zflag > 1) { 1065797ac58cSKevin Wolf printf("-b, -p, or -z cannot be specified at the same time\n"); 1066797ac58cSKevin Wolf return 0; 1067797ac58cSKevin Wolf } 1068797ac58cSKevin Wolf 1069797ac58cSKevin Wolf if (zflag && Pflag) { 1070797ac58cSKevin Wolf printf("-z and -P cannot be specified at the same time\n"); 1071797ac58cSKevin Wolf return 0; 1072797ac58cSKevin Wolf } 1073797ac58cSKevin Wolf 1074797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1075797ac58cSKevin Wolf if (offset < 0) { 1076a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 1077797ac58cSKevin Wolf return 0; 1078797ac58cSKevin Wolf } 1079797ac58cSKevin Wolf 1080797ac58cSKevin Wolf optind++; 1081797ac58cSKevin Wolf count = cvtnum(argv[optind]); 1082797ac58cSKevin Wolf if (count < 0) { 1083a9ecfa00SJohn Snow print_cvtnum_err(count, argv[optind]); 1084797ac58cSKevin Wolf return 0; 10859b0beaf3SJohn Snow } else if (count > SIZE_MAX) { 10869b0beaf3SJohn Snow printf("length cannot exceed %" PRIu64 ", given %s\n", 10879b0beaf3SJohn Snow (uint64_t) SIZE_MAX, argv[optind]); 10889b0beaf3SJohn Snow return 0; 1089797ac58cSKevin Wolf } 1090797ac58cSKevin Wolf 1091797ac58cSKevin Wolf if (!pflag) { 1092797ac58cSKevin Wolf if (offset & 0x1ff) { 1093797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 1094797ac58cSKevin Wolf offset); 1095797ac58cSKevin Wolf return 0; 1096797ac58cSKevin Wolf } 1097797ac58cSKevin Wolf 1098797ac58cSKevin Wolf if (count & 0x1ff) { 10999b0beaf3SJohn Snow printf("count %"PRId64" is not sector aligned\n", 1100797ac58cSKevin Wolf count); 1101797ac58cSKevin Wolf return 0; 1102797ac58cSKevin Wolf } 1103797ac58cSKevin Wolf } 1104797ac58cSKevin Wolf 1105797ac58cSKevin Wolf if (!zflag) { 11064c7b7e9bSMax Reitz buf = qemu_io_alloc(blk, count, pattern); 1107797ac58cSKevin Wolf } 1108797ac58cSKevin Wolf 1109797ac58cSKevin Wolf gettimeofday(&t1, NULL); 1110797ac58cSKevin Wolf if (pflag) { 11114c7b7e9bSMax Reitz cnt = do_pwrite(blk, buf, offset, count, &total); 1112797ac58cSKevin Wolf } else if (bflag) { 11134c7b7e9bSMax Reitz cnt = do_save_vmstate(blk, buf, offset, count, &total); 1114797ac58cSKevin Wolf } else if (zflag) { 11154c7b7e9bSMax Reitz cnt = do_co_write_zeroes(blk, offset, count, &total); 1116797ac58cSKevin Wolf } else if (cflag) { 11174c7b7e9bSMax Reitz cnt = do_write_compressed(blk, buf, offset, count, &total); 1118797ac58cSKevin Wolf } else { 11194c7b7e9bSMax Reitz cnt = do_write(blk, buf, offset, count, &total); 1120797ac58cSKevin Wolf } 1121797ac58cSKevin Wolf gettimeofday(&t2, NULL); 1122797ac58cSKevin Wolf 1123797ac58cSKevin Wolf if (cnt < 0) { 1124797ac58cSKevin Wolf printf("write failed: %s\n", strerror(-cnt)); 1125797ac58cSKevin Wolf goto out; 1126797ac58cSKevin Wolf } 1127797ac58cSKevin Wolf 1128797ac58cSKevin Wolf if (qflag) { 1129797ac58cSKevin Wolf goto out; 1130797ac58cSKevin Wolf } 1131797ac58cSKevin Wolf 1132797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1133797ac58cSKevin Wolf t2 = tsub(t2, t1); 1134797ac58cSKevin Wolf print_report("wrote", &t2, offset, count, total, cnt, Cflag); 1135797ac58cSKevin Wolf 1136797ac58cSKevin Wolf out: 1137797ac58cSKevin Wolf if (!zflag) { 1138797ac58cSKevin Wolf qemu_io_free(buf); 1139797ac58cSKevin Wolf } 1140797ac58cSKevin Wolf 1141797ac58cSKevin Wolf return 0; 1142797ac58cSKevin Wolf } 1143797ac58cSKevin Wolf 1144797ac58cSKevin Wolf static void 1145797ac58cSKevin Wolf writev_help(void) 1146797ac58cSKevin Wolf { 1147797ac58cSKevin Wolf printf( 1148797ac58cSKevin Wolf "\n" 1149797ac58cSKevin Wolf " writes a range of bytes from the given offset source from multiple buffers\n" 1150797ac58cSKevin Wolf "\n" 1151797ac58cSKevin Wolf " Example:\n" 11526e6507c0SMaria Kustova " 'writev 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n" 1153797ac58cSKevin Wolf "\n" 1154797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 1155797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd).\n" 1156797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 1157797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1158797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1159797ac58cSKevin Wolf "\n"); 1160797ac58cSKevin Wolf } 1161797ac58cSKevin Wolf 11624c7b7e9bSMax Reitz static int writev_f(BlockBackend *blk, int argc, char **argv); 1163797ac58cSKevin Wolf 1164797ac58cSKevin Wolf static const cmdinfo_t writev_cmd = { 1165797ac58cSKevin Wolf .name = "writev", 1166797ac58cSKevin Wolf .cfunc = writev_f, 1167797ac58cSKevin Wolf .argmin = 2, 1168797ac58cSKevin Wolf .argmax = -1, 1169797ac58cSKevin Wolf .args = "[-Cq] [-P pattern ] off len [len..]", 1170797ac58cSKevin Wolf .oneline = "writes a number of bytes at a specified offset", 1171797ac58cSKevin Wolf .help = writev_help, 1172797ac58cSKevin Wolf }; 1173797ac58cSKevin Wolf 11744c7b7e9bSMax Reitz static int writev_f(BlockBackend *blk, int argc, char **argv) 1175797ac58cSKevin Wolf { 1176797ac58cSKevin Wolf struct timeval t1, t2; 1177797ac58cSKevin Wolf int Cflag = 0, qflag = 0; 1178797ac58cSKevin Wolf int c, cnt; 1179797ac58cSKevin Wolf char *buf; 1180797ac58cSKevin Wolf int64_t offset; 1181797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 1182797ac58cSKevin Wolf int total = 0; 1183797ac58cSKevin Wolf int nr_iov; 1184797ac58cSKevin Wolf int pattern = 0xcd; 1185797ac58cSKevin Wolf QEMUIOVector qiov; 1186797ac58cSKevin Wolf 1187b062ad86SEric Blake while ((c = getopt(argc, argv, "CqP:")) != -1) { 1188797ac58cSKevin Wolf switch (c) { 1189797ac58cSKevin Wolf case 'C': 1190797ac58cSKevin Wolf Cflag = 1; 1191797ac58cSKevin Wolf break; 1192797ac58cSKevin Wolf case 'q': 1193797ac58cSKevin Wolf qflag = 1; 1194797ac58cSKevin Wolf break; 1195797ac58cSKevin Wolf case 'P': 1196797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1197797ac58cSKevin Wolf if (pattern < 0) { 1198797ac58cSKevin Wolf return 0; 1199797ac58cSKevin Wolf } 1200797ac58cSKevin Wolf break; 1201797ac58cSKevin Wolf default: 1202c2cdf5c5SKevin Wolf return qemuio_command_usage(&writev_cmd); 1203797ac58cSKevin Wolf } 1204797ac58cSKevin Wolf } 1205797ac58cSKevin Wolf 1206797ac58cSKevin Wolf if (optind > argc - 2) { 1207c2cdf5c5SKevin Wolf return qemuio_command_usage(&writev_cmd); 1208797ac58cSKevin Wolf } 1209797ac58cSKevin Wolf 1210797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1211797ac58cSKevin Wolf if (offset < 0) { 1212a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 1213797ac58cSKevin Wolf return 0; 1214797ac58cSKevin Wolf } 1215797ac58cSKevin Wolf optind++; 1216797ac58cSKevin Wolf 1217797ac58cSKevin Wolf if (offset & 0x1ff) { 1218797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 1219797ac58cSKevin Wolf offset); 1220797ac58cSKevin Wolf return 0; 1221797ac58cSKevin Wolf } 1222797ac58cSKevin Wolf 1223797ac58cSKevin Wolf nr_iov = argc - optind; 12244c7b7e9bSMax Reitz buf = create_iovec(blk, &qiov, &argv[optind], nr_iov, pattern); 1225797ac58cSKevin Wolf if (buf == NULL) { 1226797ac58cSKevin Wolf return 0; 1227797ac58cSKevin Wolf } 1228797ac58cSKevin Wolf 1229797ac58cSKevin Wolf gettimeofday(&t1, NULL); 12304c7b7e9bSMax Reitz cnt = do_aio_writev(blk, &qiov, offset, &total); 1231797ac58cSKevin Wolf gettimeofday(&t2, NULL); 1232797ac58cSKevin Wolf 1233797ac58cSKevin Wolf if (cnt < 0) { 1234797ac58cSKevin Wolf printf("writev failed: %s\n", strerror(-cnt)); 1235797ac58cSKevin Wolf goto out; 1236797ac58cSKevin Wolf } 1237797ac58cSKevin Wolf 1238797ac58cSKevin Wolf if (qflag) { 1239797ac58cSKevin Wolf goto out; 1240797ac58cSKevin Wolf } 1241797ac58cSKevin Wolf 1242797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1243797ac58cSKevin Wolf t2 = tsub(t2, t1); 1244797ac58cSKevin Wolf print_report("wrote", &t2, offset, qiov.size, total, cnt, Cflag); 1245797ac58cSKevin Wolf out: 1246797ac58cSKevin Wolf qemu_iovec_destroy(&qiov); 1247797ac58cSKevin Wolf qemu_io_free(buf); 1248797ac58cSKevin Wolf return 0; 1249797ac58cSKevin Wolf } 1250797ac58cSKevin Wolf 1251797ac58cSKevin Wolf static void multiwrite_help(void) 1252797ac58cSKevin Wolf { 1253797ac58cSKevin Wolf printf( 1254797ac58cSKevin Wolf "\n" 1255797ac58cSKevin Wolf " writes a range of bytes from the given offset source from multiple buffers,\n" 1256797ac58cSKevin Wolf " in a batch of requests that may be merged by qemu\n" 1257797ac58cSKevin Wolf "\n" 1258797ac58cSKevin Wolf " Example:\n" 1259797ac58cSKevin Wolf " 'multiwrite 512 1k 1k ; 4k 1k'\n" 1260797ac58cSKevin Wolf " writes 2 kB at 512 bytes and 1 kB at 4 kB into the open file\n" 1261797ac58cSKevin Wolf "\n" 1262797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 1263797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd). The pattern byte is increased\n" 1264797ac58cSKevin Wolf " by one for each request contained in the multiwrite command.\n" 1265797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 1266797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1267797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1268797ac58cSKevin Wolf "\n"); 1269797ac58cSKevin Wolf } 1270797ac58cSKevin Wolf 12714c7b7e9bSMax Reitz static int multiwrite_f(BlockBackend *blk, int argc, char **argv); 1272797ac58cSKevin Wolf 1273797ac58cSKevin Wolf static const cmdinfo_t multiwrite_cmd = { 1274797ac58cSKevin Wolf .name = "multiwrite", 1275797ac58cSKevin Wolf .cfunc = multiwrite_f, 1276797ac58cSKevin Wolf .argmin = 2, 1277797ac58cSKevin Wolf .argmax = -1, 1278797ac58cSKevin Wolf .args = "[-Cq] [-P pattern ] off len [len..] [; off len [len..]..]", 1279797ac58cSKevin Wolf .oneline = "issues multiple write requests at once", 1280797ac58cSKevin Wolf .help = multiwrite_help, 1281797ac58cSKevin Wolf }; 1282797ac58cSKevin Wolf 12834c7b7e9bSMax Reitz static int multiwrite_f(BlockBackend *blk, int argc, char **argv) 1284797ac58cSKevin Wolf { 1285797ac58cSKevin Wolf struct timeval t1, t2; 1286797ac58cSKevin Wolf int Cflag = 0, qflag = 0; 1287797ac58cSKevin Wolf int c, cnt; 1288797ac58cSKevin Wolf char **buf; 1289797ac58cSKevin Wolf int64_t offset, first_offset = 0; 1290797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 1291797ac58cSKevin Wolf int total = 0; 1292797ac58cSKevin Wolf int nr_iov; 1293797ac58cSKevin Wolf int nr_reqs; 1294797ac58cSKevin Wolf int pattern = 0xcd; 1295797ac58cSKevin Wolf QEMUIOVector *qiovs; 1296797ac58cSKevin Wolf int i; 1297797ac58cSKevin Wolf BlockRequest *reqs; 1298797ac58cSKevin Wolf 1299b062ad86SEric Blake while ((c = getopt(argc, argv, "CqP:")) != -1) { 1300797ac58cSKevin Wolf switch (c) { 1301797ac58cSKevin Wolf case 'C': 1302797ac58cSKevin Wolf Cflag = 1; 1303797ac58cSKevin Wolf break; 1304797ac58cSKevin Wolf case 'q': 1305797ac58cSKevin Wolf qflag = 1; 1306797ac58cSKevin Wolf break; 1307797ac58cSKevin Wolf case 'P': 1308797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1309797ac58cSKevin Wolf if (pattern < 0) { 1310797ac58cSKevin Wolf return 0; 1311797ac58cSKevin Wolf } 1312797ac58cSKevin Wolf break; 1313797ac58cSKevin Wolf default: 1314c2cdf5c5SKevin Wolf return qemuio_command_usage(&writev_cmd); 1315797ac58cSKevin Wolf } 1316797ac58cSKevin Wolf } 1317797ac58cSKevin Wolf 1318797ac58cSKevin Wolf if (optind > argc - 2) { 1319c2cdf5c5SKevin Wolf return qemuio_command_usage(&writev_cmd); 1320797ac58cSKevin Wolf } 1321797ac58cSKevin Wolf 1322797ac58cSKevin Wolf nr_reqs = 1; 1323797ac58cSKevin Wolf for (i = optind; i < argc; i++) { 1324797ac58cSKevin Wolf if (!strcmp(argv[i], ";")) { 1325797ac58cSKevin Wolf nr_reqs++; 1326797ac58cSKevin Wolf } 1327797ac58cSKevin Wolf } 1328797ac58cSKevin Wolf 132902c4f26bSMarkus Armbruster reqs = g_new0(BlockRequest, nr_reqs); 133002c4f26bSMarkus Armbruster buf = g_new0(char *, nr_reqs); 133102c4f26bSMarkus Armbruster qiovs = g_new(QEMUIOVector, nr_reqs); 1332797ac58cSKevin Wolf 1333797ac58cSKevin Wolf for (i = 0; i < nr_reqs && optind < argc; i++) { 1334797ac58cSKevin Wolf int j; 1335797ac58cSKevin Wolf 1336797ac58cSKevin Wolf /* Read the offset of the request */ 1337797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1338797ac58cSKevin Wolf if (offset < 0) { 1339a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 1340797ac58cSKevin Wolf goto out; 1341797ac58cSKevin Wolf } 1342797ac58cSKevin Wolf optind++; 1343797ac58cSKevin Wolf 1344797ac58cSKevin Wolf if (offset & 0x1ff) { 1345797ac58cSKevin Wolf printf("offset %lld is not sector aligned\n", 1346797ac58cSKevin Wolf (long long)offset); 1347797ac58cSKevin Wolf goto out; 1348797ac58cSKevin Wolf } 1349797ac58cSKevin Wolf 1350797ac58cSKevin Wolf if (i == 0) { 1351797ac58cSKevin Wolf first_offset = offset; 1352797ac58cSKevin Wolf } 1353797ac58cSKevin Wolf 1354797ac58cSKevin Wolf /* Read lengths for qiov entries */ 1355797ac58cSKevin Wolf for (j = optind; j < argc; j++) { 1356797ac58cSKevin Wolf if (!strcmp(argv[j], ";")) { 1357797ac58cSKevin Wolf break; 1358797ac58cSKevin Wolf } 1359797ac58cSKevin Wolf } 1360797ac58cSKevin Wolf 1361797ac58cSKevin Wolf nr_iov = j - optind; 1362797ac58cSKevin Wolf 1363797ac58cSKevin Wolf /* Build request */ 13644c7b7e9bSMax Reitz buf[i] = create_iovec(blk, &qiovs[i], &argv[optind], nr_iov, pattern); 1365797ac58cSKevin Wolf if (buf[i] == NULL) { 1366797ac58cSKevin Wolf goto out; 1367797ac58cSKevin Wolf } 1368797ac58cSKevin Wolf 1369797ac58cSKevin Wolf reqs[i].qiov = &qiovs[i]; 1370797ac58cSKevin Wolf reqs[i].sector = offset >> 9; 1371797ac58cSKevin Wolf reqs[i].nb_sectors = reqs[i].qiov->size >> 9; 1372797ac58cSKevin Wolf 1373797ac58cSKevin Wolf optind = j + 1; 1374797ac58cSKevin Wolf 1375797ac58cSKevin Wolf pattern++; 1376797ac58cSKevin Wolf } 1377797ac58cSKevin Wolf 1378797ac58cSKevin Wolf /* If there were empty requests at the end, ignore them */ 1379797ac58cSKevin Wolf nr_reqs = i; 1380797ac58cSKevin Wolf 1381797ac58cSKevin Wolf gettimeofday(&t1, NULL); 13824c7b7e9bSMax Reitz cnt = do_aio_multiwrite(blk, reqs, nr_reqs, &total); 1383797ac58cSKevin Wolf gettimeofday(&t2, NULL); 1384797ac58cSKevin Wolf 1385797ac58cSKevin Wolf if (cnt < 0) { 1386797ac58cSKevin Wolf printf("aio_multiwrite failed: %s\n", strerror(-cnt)); 1387797ac58cSKevin Wolf goto out; 1388797ac58cSKevin Wolf } 1389797ac58cSKevin Wolf 1390797ac58cSKevin Wolf if (qflag) { 1391797ac58cSKevin Wolf goto out; 1392797ac58cSKevin Wolf } 1393797ac58cSKevin Wolf 1394797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1395797ac58cSKevin Wolf t2 = tsub(t2, t1); 1396797ac58cSKevin Wolf print_report("wrote", &t2, first_offset, total, total, cnt, Cflag); 1397797ac58cSKevin Wolf out: 1398797ac58cSKevin Wolf for (i = 0; i < nr_reqs; i++) { 1399797ac58cSKevin Wolf qemu_io_free(buf[i]); 1400797ac58cSKevin Wolf if (reqs[i].qiov != NULL) { 1401797ac58cSKevin Wolf qemu_iovec_destroy(&qiovs[i]); 1402797ac58cSKevin Wolf } 1403797ac58cSKevin Wolf } 1404797ac58cSKevin Wolf g_free(buf); 1405797ac58cSKevin Wolf g_free(reqs); 1406797ac58cSKevin Wolf g_free(qiovs); 1407797ac58cSKevin Wolf return 0; 1408797ac58cSKevin Wolf } 1409797ac58cSKevin Wolf 1410797ac58cSKevin Wolf struct aio_ctx { 14114c7b7e9bSMax Reitz BlockBackend *blk; 1412797ac58cSKevin Wolf QEMUIOVector qiov; 1413797ac58cSKevin Wolf int64_t offset; 1414797ac58cSKevin Wolf char *buf; 1415797ac58cSKevin Wolf int qflag; 1416797ac58cSKevin Wolf int vflag; 1417797ac58cSKevin Wolf int Cflag; 1418797ac58cSKevin Wolf int Pflag; 14195ceb7765SKevin Wolf int zflag; 1420a91f9584SFam Zheng BlockAcctCookie acct; 1421797ac58cSKevin Wolf int pattern; 1422797ac58cSKevin Wolf struct timeval t1; 1423797ac58cSKevin Wolf }; 1424797ac58cSKevin Wolf 1425797ac58cSKevin Wolf static void aio_write_done(void *opaque, int ret) 1426797ac58cSKevin Wolf { 1427797ac58cSKevin Wolf struct aio_ctx *ctx = opaque; 1428797ac58cSKevin Wolf struct timeval t2; 1429797ac58cSKevin Wolf 1430797ac58cSKevin Wolf gettimeofday(&t2, NULL); 1431797ac58cSKevin Wolf 1432797ac58cSKevin Wolf 1433797ac58cSKevin Wolf if (ret < 0) { 1434797ac58cSKevin Wolf printf("aio_write failed: %s\n", strerror(-ret)); 1435556c2b60SAlberto Garcia block_acct_failed(blk_get_stats(ctx->blk), &ctx->acct); 1436797ac58cSKevin Wolf goto out; 1437797ac58cSKevin Wolf } 1438797ac58cSKevin Wolf 14394c7b7e9bSMax Reitz block_acct_done(blk_get_stats(ctx->blk), &ctx->acct); 1440a91f9584SFam Zheng 1441797ac58cSKevin Wolf if (ctx->qflag) { 1442797ac58cSKevin Wolf goto out; 1443797ac58cSKevin Wolf } 1444797ac58cSKevin Wolf 1445797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1446797ac58cSKevin Wolf t2 = tsub(t2, ctx->t1); 1447797ac58cSKevin Wolf print_report("wrote", &t2, ctx->offset, ctx->qiov.size, 1448797ac58cSKevin Wolf ctx->qiov.size, 1, ctx->Cflag); 1449797ac58cSKevin Wolf out: 14505ceb7765SKevin Wolf if (!ctx->zflag) { 1451797ac58cSKevin Wolf qemu_io_free(ctx->buf); 1452797ac58cSKevin Wolf qemu_iovec_destroy(&ctx->qiov); 14535ceb7765SKevin Wolf } 1454797ac58cSKevin Wolf g_free(ctx); 1455797ac58cSKevin Wolf } 1456797ac58cSKevin Wolf 1457797ac58cSKevin Wolf static void aio_read_done(void *opaque, int ret) 1458797ac58cSKevin Wolf { 1459797ac58cSKevin Wolf struct aio_ctx *ctx = opaque; 1460797ac58cSKevin Wolf struct timeval t2; 1461797ac58cSKevin Wolf 1462797ac58cSKevin Wolf gettimeofday(&t2, NULL); 1463797ac58cSKevin Wolf 1464797ac58cSKevin Wolf if (ret < 0) { 1465797ac58cSKevin Wolf printf("readv failed: %s\n", strerror(-ret)); 1466556c2b60SAlberto Garcia block_acct_failed(blk_get_stats(ctx->blk), &ctx->acct); 1467797ac58cSKevin Wolf goto out; 1468797ac58cSKevin Wolf } 1469797ac58cSKevin Wolf 1470797ac58cSKevin Wolf if (ctx->Pflag) { 1471797ac58cSKevin Wolf void *cmp_buf = g_malloc(ctx->qiov.size); 1472797ac58cSKevin Wolf 1473797ac58cSKevin Wolf memset(cmp_buf, ctx->pattern, ctx->qiov.size); 1474797ac58cSKevin Wolf if (memcmp(ctx->buf, cmp_buf, ctx->qiov.size)) { 1475797ac58cSKevin Wolf printf("Pattern verification failed at offset %" 1476797ac58cSKevin Wolf PRId64 ", %zd bytes\n", ctx->offset, ctx->qiov.size); 1477797ac58cSKevin Wolf } 1478797ac58cSKevin Wolf g_free(cmp_buf); 1479797ac58cSKevin Wolf } 1480797ac58cSKevin Wolf 14814c7b7e9bSMax Reitz block_acct_done(blk_get_stats(ctx->blk), &ctx->acct); 1482a91f9584SFam Zheng 1483797ac58cSKevin Wolf if (ctx->qflag) { 1484797ac58cSKevin Wolf goto out; 1485797ac58cSKevin Wolf } 1486797ac58cSKevin Wolf 1487797ac58cSKevin Wolf if (ctx->vflag) { 1488797ac58cSKevin Wolf dump_buffer(ctx->buf, ctx->offset, ctx->qiov.size); 1489797ac58cSKevin Wolf } 1490797ac58cSKevin Wolf 1491797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1492797ac58cSKevin Wolf t2 = tsub(t2, ctx->t1); 1493797ac58cSKevin Wolf print_report("read", &t2, ctx->offset, ctx->qiov.size, 1494797ac58cSKevin Wolf ctx->qiov.size, 1, ctx->Cflag); 1495797ac58cSKevin Wolf out: 1496797ac58cSKevin Wolf qemu_io_free(ctx->buf); 1497797ac58cSKevin Wolf qemu_iovec_destroy(&ctx->qiov); 1498797ac58cSKevin Wolf g_free(ctx); 1499797ac58cSKevin Wolf } 1500797ac58cSKevin Wolf 1501797ac58cSKevin Wolf static void aio_read_help(void) 1502797ac58cSKevin Wolf { 1503797ac58cSKevin Wolf printf( 1504797ac58cSKevin Wolf "\n" 1505797ac58cSKevin Wolf " asynchronously reads a range of bytes from the given offset\n" 1506797ac58cSKevin Wolf "\n" 1507797ac58cSKevin Wolf " Example:\n" 1508797ac58cSKevin Wolf " 'aio_read -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n" 1509797ac58cSKevin Wolf "\n" 1510797ac58cSKevin Wolf " Reads a segment of the currently open file, optionally dumping it to the\n" 1511797ac58cSKevin Wolf " standard output stream (with -v option) for subsequent inspection.\n" 1512797ac58cSKevin Wolf " The read is performed asynchronously and the aio_flush command must be\n" 1513797ac58cSKevin Wolf " used to ensure all outstanding aio requests have been completed.\n" 1514797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1515797ac58cSKevin Wolf " -P, -- use a pattern to verify read data\n" 1516797ac58cSKevin Wolf " -v, -- dump buffer to standard output\n" 1517797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1518797ac58cSKevin Wolf "\n"); 1519797ac58cSKevin Wolf } 1520797ac58cSKevin Wolf 15214c7b7e9bSMax Reitz static int aio_read_f(BlockBackend *blk, int argc, char **argv); 1522797ac58cSKevin Wolf 1523797ac58cSKevin Wolf static const cmdinfo_t aio_read_cmd = { 1524797ac58cSKevin Wolf .name = "aio_read", 1525797ac58cSKevin Wolf .cfunc = aio_read_f, 1526797ac58cSKevin Wolf .argmin = 2, 1527797ac58cSKevin Wolf .argmax = -1, 1528797ac58cSKevin Wolf .args = "[-Cqv] [-P pattern ] off len [len..]", 1529797ac58cSKevin Wolf .oneline = "asynchronously reads a number of bytes", 1530797ac58cSKevin Wolf .help = aio_read_help, 1531797ac58cSKevin Wolf }; 1532797ac58cSKevin Wolf 15334c7b7e9bSMax Reitz static int aio_read_f(BlockBackend *blk, int argc, char **argv) 1534797ac58cSKevin Wolf { 1535797ac58cSKevin Wolf int nr_iov, c; 1536797ac58cSKevin Wolf struct aio_ctx *ctx = g_new0(struct aio_ctx, 1); 1537797ac58cSKevin Wolf 15384c7b7e9bSMax Reitz ctx->blk = blk; 1539b062ad86SEric Blake while ((c = getopt(argc, argv, "CP:qv")) != -1) { 1540797ac58cSKevin Wolf switch (c) { 1541797ac58cSKevin Wolf case 'C': 1542797ac58cSKevin Wolf ctx->Cflag = 1; 1543797ac58cSKevin Wolf break; 1544797ac58cSKevin Wolf case 'P': 1545797ac58cSKevin Wolf ctx->Pflag = 1; 1546797ac58cSKevin Wolf ctx->pattern = parse_pattern(optarg); 1547797ac58cSKevin Wolf if (ctx->pattern < 0) { 1548797ac58cSKevin Wolf g_free(ctx); 1549797ac58cSKevin Wolf return 0; 1550797ac58cSKevin Wolf } 1551797ac58cSKevin Wolf break; 1552797ac58cSKevin Wolf case 'q': 1553797ac58cSKevin Wolf ctx->qflag = 1; 1554797ac58cSKevin Wolf break; 1555797ac58cSKevin Wolf case 'v': 1556797ac58cSKevin Wolf ctx->vflag = 1; 1557797ac58cSKevin Wolf break; 1558797ac58cSKevin Wolf default: 1559797ac58cSKevin Wolf g_free(ctx); 1560c2cdf5c5SKevin Wolf return qemuio_command_usage(&aio_read_cmd); 1561797ac58cSKevin Wolf } 1562797ac58cSKevin Wolf } 1563797ac58cSKevin Wolf 1564797ac58cSKevin Wolf if (optind > argc - 2) { 1565797ac58cSKevin Wolf g_free(ctx); 1566c2cdf5c5SKevin Wolf return qemuio_command_usage(&aio_read_cmd); 1567797ac58cSKevin Wolf } 1568797ac58cSKevin Wolf 1569797ac58cSKevin Wolf ctx->offset = cvtnum(argv[optind]); 1570797ac58cSKevin Wolf if (ctx->offset < 0) { 1571a9ecfa00SJohn Snow print_cvtnum_err(ctx->offset, argv[optind]); 1572797ac58cSKevin Wolf g_free(ctx); 1573797ac58cSKevin Wolf return 0; 1574797ac58cSKevin Wolf } 1575797ac58cSKevin Wolf optind++; 1576797ac58cSKevin Wolf 1577797ac58cSKevin Wolf if (ctx->offset & 0x1ff) { 1578797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 1579797ac58cSKevin Wolf ctx->offset); 1580556c2b60SAlberto Garcia block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_READ); 1581797ac58cSKevin Wolf g_free(ctx); 1582797ac58cSKevin Wolf return 0; 1583797ac58cSKevin Wolf } 1584797ac58cSKevin Wolf 1585797ac58cSKevin Wolf nr_iov = argc - optind; 15864c7b7e9bSMax Reitz ctx->buf = create_iovec(blk, &ctx->qiov, &argv[optind], nr_iov, 0xab); 1587797ac58cSKevin Wolf if (ctx->buf == NULL) { 1588556c2b60SAlberto Garcia block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_READ); 1589797ac58cSKevin Wolf g_free(ctx); 1590797ac58cSKevin Wolf return 0; 1591797ac58cSKevin Wolf } 1592797ac58cSKevin Wolf 1593797ac58cSKevin Wolf gettimeofday(&ctx->t1, NULL); 15944c7b7e9bSMax Reitz block_acct_start(blk_get_stats(blk), &ctx->acct, ctx->qiov.size, 15954c7b7e9bSMax Reitz BLOCK_ACCT_READ); 15964c7b7e9bSMax Reitz blk_aio_readv(blk, ctx->offset >> 9, &ctx->qiov, 1597797ac58cSKevin Wolf ctx->qiov.size >> 9, aio_read_done, ctx); 1598797ac58cSKevin Wolf return 0; 1599797ac58cSKevin Wolf } 1600797ac58cSKevin Wolf 1601797ac58cSKevin Wolf static void aio_write_help(void) 1602797ac58cSKevin Wolf { 1603797ac58cSKevin Wolf printf( 1604797ac58cSKevin Wolf "\n" 1605797ac58cSKevin Wolf " asynchronously writes a range of bytes from the given offset source\n" 1606797ac58cSKevin Wolf " from multiple buffers\n" 1607797ac58cSKevin Wolf "\n" 1608797ac58cSKevin Wolf " Example:\n" 1609797ac58cSKevin Wolf " 'aio_write 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n" 1610797ac58cSKevin Wolf "\n" 1611797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 1612797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd).\n" 1613797ac58cSKevin Wolf " The write is performed asynchronously and the aio_flush command must be\n" 1614797ac58cSKevin Wolf " used to ensure all outstanding aio requests have been completed.\n" 1615797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 1616797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1617797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 16185ceb7765SKevin Wolf " -z, -- write zeroes using blk_aio_write_zeroes\n" 1619797ac58cSKevin Wolf "\n"); 1620797ac58cSKevin Wolf } 1621797ac58cSKevin Wolf 16224c7b7e9bSMax Reitz static int aio_write_f(BlockBackend *blk, int argc, char **argv); 1623797ac58cSKevin Wolf 1624797ac58cSKevin Wolf static const cmdinfo_t aio_write_cmd = { 1625797ac58cSKevin Wolf .name = "aio_write", 1626797ac58cSKevin Wolf .cfunc = aio_write_f, 1627797ac58cSKevin Wolf .argmin = 2, 1628797ac58cSKevin Wolf .argmax = -1, 16295ceb7765SKevin Wolf .args = "[-Cqz] [-P pattern ] off len [len..]", 1630797ac58cSKevin Wolf .oneline = "asynchronously writes a number of bytes", 1631797ac58cSKevin Wolf .help = aio_write_help, 1632797ac58cSKevin Wolf }; 1633797ac58cSKevin Wolf 16344c7b7e9bSMax Reitz static int aio_write_f(BlockBackend *blk, int argc, char **argv) 1635797ac58cSKevin Wolf { 1636797ac58cSKevin Wolf int nr_iov, c; 1637797ac58cSKevin Wolf int pattern = 0xcd; 1638797ac58cSKevin Wolf struct aio_ctx *ctx = g_new0(struct aio_ctx, 1); 1639797ac58cSKevin Wolf 16404c7b7e9bSMax Reitz ctx->blk = blk; 16415ceb7765SKevin Wolf while ((c = getopt(argc, argv, "CqP:z")) != -1) { 1642797ac58cSKevin Wolf switch (c) { 1643797ac58cSKevin Wolf case 'C': 1644797ac58cSKevin Wolf ctx->Cflag = 1; 1645797ac58cSKevin Wolf break; 1646797ac58cSKevin Wolf case 'q': 1647797ac58cSKevin Wolf ctx->qflag = 1; 1648797ac58cSKevin Wolf break; 1649797ac58cSKevin Wolf case 'P': 1650797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1651797ac58cSKevin Wolf if (pattern < 0) { 1652797ac58cSKevin Wolf g_free(ctx); 1653797ac58cSKevin Wolf return 0; 1654797ac58cSKevin Wolf } 1655797ac58cSKevin Wolf break; 16565ceb7765SKevin Wolf case 'z': 16575ceb7765SKevin Wolf ctx->zflag = 1; 16585ceb7765SKevin Wolf break; 1659797ac58cSKevin Wolf default: 1660797ac58cSKevin Wolf g_free(ctx); 1661c2cdf5c5SKevin Wolf return qemuio_command_usage(&aio_write_cmd); 1662797ac58cSKevin Wolf } 1663797ac58cSKevin Wolf } 1664797ac58cSKevin Wolf 1665797ac58cSKevin Wolf if (optind > argc - 2) { 1666797ac58cSKevin Wolf g_free(ctx); 1667c2cdf5c5SKevin Wolf return qemuio_command_usage(&aio_write_cmd); 1668797ac58cSKevin Wolf } 1669797ac58cSKevin Wolf 16705ceb7765SKevin Wolf if (ctx->zflag && optind != argc - 2) { 16715ceb7765SKevin Wolf printf("-z supports only a single length parameter\n"); 16725ceb7765SKevin Wolf g_free(ctx); 16735ceb7765SKevin Wolf return 0; 16745ceb7765SKevin Wolf } 16755ceb7765SKevin Wolf 16765ceb7765SKevin Wolf if (ctx->zflag && ctx->Pflag) { 16775ceb7765SKevin Wolf printf("-z and -P cannot be specified at the same time\n"); 16785ceb7765SKevin Wolf g_free(ctx); 16795ceb7765SKevin Wolf return 0; 16805ceb7765SKevin Wolf } 16815ceb7765SKevin Wolf 1682797ac58cSKevin Wolf ctx->offset = cvtnum(argv[optind]); 1683797ac58cSKevin Wolf if (ctx->offset < 0) { 1684a9ecfa00SJohn Snow print_cvtnum_err(ctx->offset, argv[optind]); 1685797ac58cSKevin Wolf g_free(ctx); 1686797ac58cSKevin Wolf return 0; 1687797ac58cSKevin Wolf } 1688797ac58cSKevin Wolf optind++; 1689797ac58cSKevin Wolf 1690797ac58cSKevin Wolf if (ctx->offset & 0x1ff) { 1691797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 1692797ac58cSKevin Wolf ctx->offset); 1693556c2b60SAlberto Garcia block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_WRITE); 1694797ac58cSKevin Wolf g_free(ctx); 1695797ac58cSKevin Wolf return 0; 1696797ac58cSKevin Wolf } 1697797ac58cSKevin Wolf 16985ceb7765SKevin Wolf if (ctx->zflag) { 16995ceb7765SKevin Wolf int64_t count = cvtnum(argv[optind]); 17005ceb7765SKevin Wolf if (count < 0) { 17015ceb7765SKevin Wolf print_cvtnum_err(count, argv[optind]); 1702*0e01b76eSKevin Wolf g_free(ctx); 17035ceb7765SKevin Wolf return 0; 17045ceb7765SKevin Wolf } 17055ceb7765SKevin Wolf 17065ceb7765SKevin Wolf ctx->qiov.size = count; 17075ceb7765SKevin Wolf blk_aio_write_zeroes(blk, ctx->offset >> 9, count >> 9, 0, 17085ceb7765SKevin Wolf aio_write_done, ctx); 17095ceb7765SKevin Wolf } else { 1710797ac58cSKevin Wolf nr_iov = argc - optind; 17115ceb7765SKevin Wolf ctx->buf = create_iovec(blk, &ctx->qiov, &argv[optind], nr_iov, 17125ceb7765SKevin Wolf pattern); 1713797ac58cSKevin Wolf if (ctx->buf == NULL) { 1714556c2b60SAlberto Garcia block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_WRITE); 1715797ac58cSKevin Wolf g_free(ctx); 1716797ac58cSKevin Wolf return 0; 1717797ac58cSKevin Wolf } 1718797ac58cSKevin Wolf 1719797ac58cSKevin Wolf gettimeofday(&ctx->t1, NULL); 17204c7b7e9bSMax Reitz block_acct_start(blk_get_stats(blk), &ctx->acct, ctx->qiov.size, 17214c7b7e9bSMax Reitz BLOCK_ACCT_WRITE); 17225ceb7765SKevin Wolf 17234c7b7e9bSMax Reitz blk_aio_writev(blk, ctx->offset >> 9, &ctx->qiov, 1724797ac58cSKevin Wolf ctx->qiov.size >> 9, aio_write_done, ctx); 17255ceb7765SKevin Wolf } 1726797ac58cSKevin Wolf return 0; 1727797ac58cSKevin Wolf } 1728797ac58cSKevin Wolf 17294c7b7e9bSMax Reitz static int aio_flush_f(BlockBackend *blk, int argc, char **argv) 1730797ac58cSKevin Wolf { 1731556c2b60SAlberto Garcia BlockAcctCookie cookie; 1732556c2b60SAlberto Garcia block_acct_start(blk_get_stats(blk), &cookie, 0, BLOCK_ACCT_FLUSH); 17334c7b7e9bSMax Reitz blk_drain_all(); 1734556c2b60SAlberto Garcia block_acct_done(blk_get_stats(blk), &cookie); 1735797ac58cSKevin Wolf return 0; 1736797ac58cSKevin Wolf } 1737797ac58cSKevin Wolf 1738797ac58cSKevin Wolf static const cmdinfo_t aio_flush_cmd = { 1739797ac58cSKevin Wolf .name = "aio_flush", 1740797ac58cSKevin Wolf .cfunc = aio_flush_f, 1741797ac58cSKevin Wolf .oneline = "completes all outstanding aio requests" 1742797ac58cSKevin Wolf }; 1743797ac58cSKevin Wolf 17444c7b7e9bSMax Reitz static int flush_f(BlockBackend *blk, int argc, char **argv) 1745797ac58cSKevin Wolf { 17464c7b7e9bSMax Reitz blk_flush(blk); 1747797ac58cSKevin Wolf return 0; 1748797ac58cSKevin Wolf } 1749797ac58cSKevin Wolf 1750797ac58cSKevin Wolf static const cmdinfo_t flush_cmd = { 1751797ac58cSKevin Wolf .name = "flush", 1752797ac58cSKevin Wolf .altname = "f", 1753797ac58cSKevin Wolf .cfunc = flush_f, 1754797ac58cSKevin Wolf .oneline = "flush all in-core file state to disk", 1755797ac58cSKevin Wolf }; 1756797ac58cSKevin Wolf 17574c7b7e9bSMax Reitz static int truncate_f(BlockBackend *blk, int argc, char **argv) 1758797ac58cSKevin Wolf { 1759797ac58cSKevin Wolf int64_t offset; 1760797ac58cSKevin Wolf int ret; 1761797ac58cSKevin Wolf 1762797ac58cSKevin Wolf offset = cvtnum(argv[1]); 1763797ac58cSKevin Wolf if (offset < 0) { 1764a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[1]); 1765797ac58cSKevin Wolf return 0; 1766797ac58cSKevin Wolf } 1767797ac58cSKevin Wolf 17684c7b7e9bSMax Reitz ret = blk_truncate(blk, offset); 1769797ac58cSKevin Wolf if (ret < 0) { 1770797ac58cSKevin Wolf printf("truncate: %s\n", strerror(-ret)); 1771797ac58cSKevin Wolf return 0; 1772797ac58cSKevin Wolf } 1773797ac58cSKevin Wolf 1774797ac58cSKevin Wolf return 0; 1775797ac58cSKevin Wolf } 1776797ac58cSKevin Wolf 1777797ac58cSKevin Wolf static const cmdinfo_t truncate_cmd = { 1778797ac58cSKevin Wolf .name = "truncate", 1779797ac58cSKevin Wolf .altname = "t", 1780797ac58cSKevin Wolf .cfunc = truncate_f, 1781797ac58cSKevin Wolf .argmin = 1, 1782797ac58cSKevin Wolf .argmax = 1, 1783797ac58cSKevin Wolf .args = "off", 1784797ac58cSKevin Wolf .oneline = "truncates the current file at the given offset", 1785797ac58cSKevin Wolf }; 1786797ac58cSKevin Wolf 17874c7b7e9bSMax Reitz static int length_f(BlockBackend *blk, int argc, char **argv) 1788797ac58cSKevin Wolf { 1789797ac58cSKevin Wolf int64_t size; 1790797ac58cSKevin Wolf char s1[64]; 1791797ac58cSKevin Wolf 17924c7b7e9bSMax Reitz size = blk_getlength(blk); 1793797ac58cSKevin Wolf if (size < 0) { 1794797ac58cSKevin Wolf printf("getlength: %s\n", strerror(-size)); 1795797ac58cSKevin Wolf return 0; 1796797ac58cSKevin Wolf } 1797797ac58cSKevin Wolf 1798797ac58cSKevin Wolf cvtstr(size, s1, sizeof(s1)); 1799797ac58cSKevin Wolf printf("%s\n", s1); 1800797ac58cSKevin Wolf return 0; 1801797ac58cSKevin Wolf } 1802797ac58cSKevin Wolf 1803797ac58cSKevin Wolf 1804797ac58cSKevin Wolf static const cmdinfo_t length_cmd = { 1805797ac58cSKevin Wolf .name = "length", 1806797ac58cSKevin Wolf .altname = "l", 1807797ac58cSKevin Wolf .cfunc = length_f, 1808797ac58cSKevin Wolf .oneline = "gets the length of the current file", 1809797ac58cSKevin Wolf }; 1810797ac58cSKevin Wolf 1811797ac58cSKevin Wolf 18124c7b7e9bSMax Reitz static int info_f(BlockBackend *blk, int argc, char **argv) 1813797ac58cSKevin Wolf { 18144c7b7e9bSMax Reitz BlockDriverState *bs = blk_bs(blk); 1815797ac58cSKevin Wolf BlockDriverInfo bdi; 1816a8d8ecb7SMax Reitz ImageInfoSpecific *spec_info; 1817797ac58cSKevin Wolf char s1[64], s2[64]; 1818797ac58cSKevin Wolf int ret; 1819797ac58cSKevin Wolf 1820797ac58cSKevin Wolf if (bs->drv && bs->drv->format_name) { 1821797ac58cSKevin Wolf printf("format name: %s\n", bs->drv->format_name); 1822797ac58cSKevin Wolf } 1823797ac58cSKevin Wolf if (bs->drv && bs->drv->protocol_name) { 1824797ac58cSKevin Wolf printf("format name: %s\n", bs->drv->protocol_name); 1825797ac58cSKevin Wolf } 1826797ac58cSKevin Wolf 1827797ac58cSKevin Wolf ret = bdrv_get_info(bs, &bdi); 1828797ac58cSKevin Wolf if (ret) { 1829797ac58cSKevin Wolf return 0; 1830797ac58cSKevin Wolf } 1831797ac58cSKevin Wolf 1832797ac58cSKevin Wolf cvtstr(bdi.cluster_size, s1, sizeof(s1)); 1833797ac58cSKevin Wolf cvtstr(bdi.vm_state_offset, s2, sizeof(s2)); 1834797ac58cSKevin Wolf 1835797ac58cSKevin Wolf printf("cluster size: %s\n", s1); 1836797ac58cSKevin Wolf printf("vm state offset: %s\n", s2); 1837797ac58cSKevin Wolf 1838a8d8ecb7SMax Reitz spec_info = bdrv_get_specific_info(bs); 1839a8d8ecb7SMax Reitz if (spec_info) { 1840a8d8ecb7SMax Reitz printf("Format specific information:\n"); 1841a8d8ecb7SMax Reitz bdrv_image_info_specific_dump(fprintf, stdout, spec_info); 1842a8d8ecb7SMax Reitz qapi_free_ImageInfoSpecific(spec_info); 1843a8d8ecb7SMax Reitz } 1844a8d8ecb7SMax Reitz 1845797ac58cSKevin Wolf return 0; 1846797ac58cSKevin Wolf } 1847797ac58cSKevin Wolf 1848797ac58cSKevin Wolf 1849797ac58cSKevin Wolf 1850797ac58cSKevin Wolf static const cmdinfo_t info_cmd = { 1851797ac58cSKevin Wolf .name = "info", 1852797ac58cSKevin Wolf .altname = "i", 1853797ac58cSKevin Wolf .cfunc = info_f, 1854797ac58cSKevin Wolf .oneline = "prints information about the current file", 1855797ac58cSKevin Wolf }; 1856797ac58cSKevin Wolf 1857797ac58cSKevin Wolf static void discard_help(void) 1858797ac58cSKevin Wolf { 1859797ac58cSKevin Wolf printf( 1860797ac58cSKevin Wolf "\n" 1861797ac58cSKevin Wolf " discards a range of bytes from the given offset\n" 1862797ac58cSKevin Wolf "\n" 1863797ac58cSKevin Wolf " Example:\n" 1864797ac58cSKevin Wolf " 'discard 512 1k' - discards 1 kilobyte from 512 bytes into the file\n" 1865797ac58cSKevin Wolf "\n" 1866797ac58cSKevin Wolf " Discards a segment of the currently open file.\n" 1867797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1868797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1869797ac58cSKevin Wolf "\n"); 1870797ac58cSKevin Wolf } 1871797ac58cSKevin Wolf 18724c7b7e9bSMax Reitz static int discard_f(BlockBackend *blk, int argc, char **argv); 1873797ac58cSKevin Wolf 1874797ac58cSKevin Wolf static const cmdinfo_t discard_cmd = { 1875797ac58cSKevin Wolf .name = "discard", 1876797ac58cSKevin Wolf .altname = "d", 1877797ac58cSKevin Wolf .cfunc = discard_f, 1878797ac58cSKevin Wolf .argmin = 2, 1879797ac58cSKevin Wolf .argmax = -1, 1880797ac58cSKevin Wolf .args = "[-Cq] off len", 1881797ac58cSKevin Wolf .oneline = "discards a number of bytes at a specified offset", 1882797ac58cSKevin Wolf .help = discard_help, 1883797ac58cSKevin Wolf }; 1884797ac58cSKevin Wolf 18854c7b7e9bSMax Reitz static int discard_f(BlockBackend *blk, int argc, char **argv) 1886797ac58cSKevin Wolf { 1887797ac58cSKevin Wolf struct timeval t1, t2; 1888797ac58cSKevin Wolf int Cflag = 0, qflag = 0; 1889797ac58cSKevin Wolf int c, ret; 18909b0beaf3SJohn Snow int64_t offset, count; 1891797ac58cSKevin Wolf 1892b062ad86SEric Blake while ((c = getopt(argc, argv, "Cq")) != -1) { 1893797ac58cSKevin Wolf switch (c) { 1894797ac58cSKevin Wolf case 'C': 1895797ac58cSKevin Wolf Cflag = 1; 1896797ac58cSKevin Wolf break; 1897797ac58cSKevin Wolf case 'q': 1898797ac58cSKevin Wolf qflag = 1; 1899797ac58cSKevin Wolf break; 1900797ac58cSKevin Wolf default: 1901c2cdf5c5SKevin Wolf return qemuio_command_usage(&discard_cmd); 1902797ac58cSKevin Wolf } 1903797ac58cSKevin Wolf } 1904797ac58cSKevin Wolf 1905797ac58cSKevin Wolf if (optind != argc - 2) { 1906c2cdf5c5SKevin Wolf return qemuio_command_usage(&discard_cmd); 1907797ac58cSKevin Wolf } 1908797ac58cSKevin Wolf 1909797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1910797ac58cSKevin Wolf if (offset < 0) { 1911a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 1912797ac58cSKevin Wolf return 0; 1913797ac58cSKevin Wolf } 1914797ac58cSKevin Wolf 1915797ac58cSKevin Wolf optind++; 1916797ac58cSKevin Wolf count = cvtnum(argv[optind]); 1917797ac58cSKevin Wolf if (count < 0) { 1918a9ecfa00SJohn Snow print_cvtnum_err(count, argv[optind]); 1919797ac58cSKevin Wolf return 0; 19209b0beaf3SJohn Snow } else if (count >> BDRV_SECTOR_BITS > INT_MAX) { 19219b0beaf3SJohn Snow printf("length cannot exceed %"PRIu64", given %s\n", 19229b0beaf3SJohn Snow (uint64_t)INT_MAX << BDRV_SECTOR_BITS, 19239b0beaf3SJohn Snow argv[optind]); 19249b0beaf3SJohn Snow return 0; 1925797ac58cSKevin Wolf } 1926797ac58cSKevin Wolf 1927797ac58cSKevin Wolf gettimeofday(&t1, NULL); 19284c7b7e9bSMax Reitz ret = blk_discard(blk, offset >> BDRV_SECTOR_BITS, 1929797ac58cSKevin Wolf count >> BDRV_SECTOR_BITS); 1930797ac58cSKevin Wolf gettimeofday(&t2, NULL); 1931797ac58cSKevin Wolf 1932797ac58cSKevin Wolf if (ret < 0) { 1933797ac58cSKevin Wolf printf("discard failed: %s\n", strerror(-ret)); 1934797ac58cSKevin Wolf goto out; 1935797ac58cSKevin Wolf } 1936797ac58cSKevin Wolf 1937797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1938797ac58cSKevin Wolf if (!qflag) { 1939797ac58cSKevin Wolf t2 = tsub(t2, t1); 1940797ac58cSKevin Wolf print_report("discard", &t2, offset, count, count, 1, Cflag); 1941797ac58cSKevin Wolf } 1942797ac58cSKevin Wolf 1943797ac58cSKevin Wolf out: 1944797ac58cSKevin Wolf return 0; 1945797ac58cSKevin Wolf } 1946797ac58cSKevin Wolf 19474c7b7e9bSMax Reitz static int alloc_f(BlockBackend *blk, int argc, char **argv) 1948797ac58cSKevin Wolf { 19494c7b7e9bSMax Reitz BlockDriverState *bs = blk_bs(blk); 19509b0beaf3SJohn Snow int64_t offset, sector_num, nb_sectors, remaining; 1951797ac58cSKevin Wolf char s1[64]; 19529b0beaf3SJohn Snow int num, ret; 19539b0beaf3SJohn Snow int64_t sum_alloc; 1954797ac58cSKevin Wolf 1955797ac58cSKevin Wolf offset = cvtnum(argv[1]); 1956797ac58cSKevin Wolf if (offset < 0) { 1957a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[1]); 1958797ac58cSKevin Wolf return 0; 1959797ac58cSKevin Wolf } else if (offset & 0x1ff) { 1960797ac58cSKevin Wolf printf("offset %" PRId64 " is not sector aligned\n", 1961797ac58cSKevin Wolf offset); 1962797ac58cSKevin Wolf return 0; 1963797ac58cSKevin Wolf } 1964797ac58cSKevin Wolf 1965797ac58cSKevin Wolf if (argc == 3) { 1966797ac58cSKevin Wolf nb_sectors = cvtnum(argv[2]); 1967797ac58cSKevin Wolf if (nb_sectors < 0) { 1968a9ecfa00SJohn Snow print_cvtnum_err(nb_sectors, argv[2]); 1969797ac58cSKevin Wolf return 0; 19709b0beaf3SJohn Snow } else if (nb_sectors > INT_MAX) { 19719b0beaf3SJohn Snow printf("length argument cannot exceed %d, given %s\n", 19729b0beaf3SJohn Snow INT_MAX, argv[2]); 19739b0beaf3SJohn Snow return 0; 1974797ac58cSKevin Wolf } 1975797ac58cSKevin Wolf } else { 1976797ac58cSKevin Wolf nb_sectors = 1; 1977797ac58cSKevin Wolf } 1978797ac58cSKevin Wolf 1979797ac58cSKevin Wolf remaining = nb_sectors; 1980797ac58cSKevin Wolf sum_alloc = 0; 1981797ac58cSKevin Wolf sector_num = offset >> 9; 1982797ac58cSKevin Wolf while (remaining) { 1983797ac58cSKevin Wolf ret = bdrv_is_allocated(bs, sector_num, remaining, &num); 1984d663640cSPaolo Bonzini if (ret < 0) { 1985d663640cSPaolo Bonzini printf("is_allocated failed: %s\n", strerror(-ret)); 1986d663640cSPaolo Bonzini return 0; 1987d663640cSPaolo Bonzini } 1988797ac58cSKevin Wolf sector_num += num; 1989797ac58cSKevin Wolf remaining -= num; 1990797ac58cSKevin Wolf if (ret) { 1991797ac58cSKevin Wolf sum_alloc += num; 1992797ac58cSKevin Wolf } 1993797ac58cSKevin Wolf if (num == 0) { 1994797ac58cSKevin Wolf nb_sectors -= remaining; 1995797ac58cSKevin Wolf remaining = 0; 1996797ac58cSKevin Wolf } 1997797ac58cSKevin Wolf } 1998797ac58cSKevin Wolf 1999797ac58cSKevin Wolf cvtstr(offset, s1, sizeof(s1)); 2000797ac58cSKevin Wolf 20019b0beaf3SJohn Snow printf("%"PRId64"/%"PRId64" sectors allocated at offset %s\n", 2002797ac58cSKevin Wolf sum_alloc, nb_sectors, s1); 2003797ac58cSKevin Wolf return 0; 2004797ac58cSKevin Wolf } 2005797ac58cSKevin Wolf 2006797ac58cSKevin Wolf static const cmdinfo_t alloc_cmd = { 2007797ac58cSKevin Wolf .name = "alloc", 2008797ac58cSKevin Wolf .altname = "a", 2009797ac58cSKevin Wolf .argmin = 1, 2010797ac58cSKevin Wolf .argmax = 2, 2011797ac58cSKevin Wolf .cfunc = alloc_f, 2012797ac58cSKevin Wolf .args = "off [sectors]", 2013797ac58cSKevin Wolf .oneline = "checks if a sector is present in the file", 2014797ac58cSKevin Wolf }; 2015797ac58cSKevin Wolf 2016797ac58cSKevin Wolf 2017797ac58cSKevin Wolf static int map_is_allocated(BlockDriverState *bs, int64_t sector_num, 2018797ac58cSKevin Wolf int64_t nb_sectors, int64_t *pnum) 2019797ac58cSKevin Wolf { 2020797ac58cSKevin Wolf int num, num_checked; 2021797ac58cSKevin Wolf int ret, firstret; 2022797ac58cSKevin Wolf 2023797ac58cSKevin Wolf num_checked = MIN(nb_sectors, INT_MAX); 2024797ac58cSKevin Wolf ret = bdrv_is_allocated(bs, sector_num, num_checked, &num); 2025797ac58cSKevin Wolf if (ret < 0) { 2026797ac58cSKevin Wolf return ret; 2027797ac58cSKevin Wolf } 2028797ac58cSKevin Wolf 2029797ac58cSKevin Wolf firstret = ret; 2030797ac58cSKevin Wolf *pnum = num; 2031797ac58cSKevin Wolf 2032797ac58cSKevin Wolf while (nb_sectors > 0 && ret == firstret) { 2033797ac58cSKevin Wolf sector_num += num; 2034797ac58cSKevin Wolf nb_sectors -= num; 2035797ac58cSKevin Wolf 2036797ac58cSKevin Wolf num_checked = MIN(nb_sectors, INT_MAX); 2037797ac58cSKevin Wolf ret = bdrv_is_allocated(bs, sector_num, num_checked, &num); 20384b25bbc4SMax Reitz if (ret == firstret && num) { 2039797ac58cSKevin Wolf *pnum += num; 2040797ac58cSKevin Wolf } else { 2041797ac58cSKevin Wolf break; 2042797ac58cSKevin Wolf } 2043797ac58cSKevin Wolf } 2044797ac58cSKevin Wolf 2045797ac58cSKevin Wolf return firstret; 2046797ac58cSKevin Wolf } 2047797ac58cSKevin Wolf 20484c7b7e9bSMax Reitz static int map_f(BlockBackend *blk, int argc, char **argv) 2049797ac58cSKevin Wolf { 2050797ac58cSKevin Wolf int64_t offset; 20514c7b7e9bSMax Reitz int64_t nb_sectors, total_sectors; 2052797ac58cSKevin Wolf char s1[64]; 2053797ac58cSKevin Wolf int64_t num; 2054797ac58cSKevin Wolf int ret; 2055797ac58cSKevin Wolf const char *retstr; 2056797ac58cSKevin Wolf 2057797ac58cSKevin Wolf offset = 0; 20584c7b7e9bSMax Reitz total_sectors = blk_nb_sectors(blk); 20594c7b7e9bSMax Reitz if (total_sectors < 0) { 20604c7b7e9bSMax Reitz error_report("Failed to query image length: %s", 20614c7b7e9bSMax Reitz strerror(-total_sectors)); 20624c7b7e9bSMax Reitz return 0; 20634c7b7e9bSMax Reitz } 20644c7b7e9bSMax Reitz 20654c7b7e9bSMax Reitz nb_sectors = total_sectors; 2066797ac58cSKevin Wolf 2067797ac58cSKevin Wolf do { 20684c7b7e9bSMax Reitz ret = map_is_allocated(blk_bs(blk), offset, nb_sectors, &num); 2069797ac58cSKevin Wolf if (ret < 0) { 2070797ac58cSKevin Wolf error_report("Failed to get allocation status: %s", strerror(-ret)); 2071797ac58cSKevin Wolf return 0; 20724b25bbc4SMax Reitz } else if (!num) { 20734b25bbc4SMax Reitz error_report("Unexpected end of image"); 20744b25bbc4SMax Reitz return 0; 2075797ac58cSKevin Wolf } 2076797ac58cSKevin Wolf 2077797ac58cSKevin Wolf retstr = ret ? " allocated" : "not allocated"; 2078797ac58cSKevin Wolf cvtstr(offset << 9ULL, s1, sizeof(s1)); 2079797ac58cSKevin Wolf printf("[% 24" PRId64 "] % 8" PRId64 "/% 8" PRId64 " sectors %s " 2080797ac58cSKevin Wolf "at offset %s (%d)\n", 2081797ac58cSKevin Wolf offset << 9ULL, num, nb_sectors, retstr, s1, ret); 2082797ac58cSKevin Wolf 2083797ac58cSKevin Wolf offset += num; 2084797ac58cSKevin Wolf nb_sectors -= num; 20854c7b7e9bSMax Reitz } while (offset < total_sectors); 2086797ac58cSKevin Wolf 2087797ac58cSKevin Wolf return 0; 2088797ac58cSKevin Wolf } 2089797ac58cSKevin Wolf 2090797ac58cSKevin Wolf static const cmdinfo_t map_cmd = { 2091797ac58cSKevin Wolf .name = "map", 2092797ac58cSKevin Wolf .argmin = 0, 2093797ac58cSKevin Wolf .argmax = 0, 2094797ac58cSKevin Wolf .cfunc = map_f, 2095797ac58cSKevin Wolf .args = "", 2096797ac58cSKevin Wolf .oneline = "prints the allocated areas of a file", 2097797ac58cSKevin Wolf }; 2098797ac58cSKevin Wolf 20995bbd2e59SKevin Wolf static void reopen_help(void) 21005bbd2e59SKevin Wolf { 21015bbd2e59SKevin Wolf printf( 21025bbd2e59SKevin Wolf "\n" 21035bbd2e59SKevin Wolf " Changes the open options of an already opened image\n" 21045bbd2e59SKevin Wolf "\n" 21055bbd2e59SKevin Wolf " Example:\n" 21065bbd2e59SKevin Wolf " 'reopen -o lazy-refcounts=on' - activates lazy refcount writeback on a qcow2 image\n" 21075bbd2e59SKevin Wolf "\n" 21085bbd2e59SKevin Wolf " -r, -- Reopen the image read-only\n" 21095bbd2e59SKevin Wolf " -c, -- Change the cache mode to the given value\n" 21105bbd2e59SKevin Wolf " -o, -- Changes block driver options (cf. 'open' command)\n" 21115bbd2e59SKevin Wolf "\n"); 21125bbd2e59SKevin Wolf } 21135bbd2e59SKevin Wolf 21145bbd2e59SKevin Wolf static int reopen_f(BlockBackend *blk, int argc, char **argv); 21155bbd2e59SKevin Wolf 21165bbd2e59SKevin Wolf static QemuOptsList reopen_opts = { 21175bbd2e59SKevin Wolf .name = "reopen", 21185bbd2e59SKevin Wolf .merge_lists = true, 21195bbd2e59SKevin Wolf .head = QTAILQ_HEAD_INITIALIZER(reopen_opts.head), 21205bbd2e59SKevin Wolf .desc = { 21215bbd2e59SKevin Wolf /* no elements => accept any params */ 21225bbd2e59SKevin Wolf { /* end of list */ } 21235bbd2e59SKevin Wolf }, 21245bbd2e59SKevin Wolf }; 21255bbd2e59SKevin Wolf 21265bbd2e59SKevin Wolf static const cmdinfo_t reopen_cmd = { 21275bbd2e59SKevin Wolf .name = "reopen", 21285bbd2e59SKevin Wolf .argmin = 0, 21295bbd2e59SKevin Wolf .argmax = -1, 21305bbd2e59SKevin Wolf .cfunc = reopen_f, 21315bbd2e59SKevin Wolf .args = "[-r] [-c cache] [-o options]", 21325bbd2e59SKevin Wolf .oneline = "reopens an image with new options", 21335bbd2e59SKevin Wolf .help = reopen_help, 21345bbd2e59SKevin Wolf }; 21355bbd2e59SKevin Wolf 21365bbd2e59SKevin Wolf static int reopen_f(BlockBackend *blk, int argc, char **argv) 21375bbd2e59SKevin Wolf { 21385bbd2e59SKevin Wolf BlockDriverState *bs = blk_bs(blk); 21395bbd2e59SKevin Wolf QemuOpts *qopts; 21405bbd2e59SKevin Wolf QDict *opts; 21415bbd2e59SKevin Wolf int c; 21425bbd2e59SKevin Wolf int flags = bs->open_flags; 214319dbecdcSKevin Wolf bool writethrough = !blk_enable_write_cache(blk); 21445bbd2e59SKevin Wolf 21455bbd2e59SKevin Wolf BlockReopenQueue *brq; 21465bbd2e59SKevin Wolf Error *local_err = NULL; 21475bbd2e59SKevin Wolf 21485bbd2e59SKevin Wolf while ((c = getopt(argc, argv, "c:o:r")) != -1) { 21495bbd2e59SKevin Wolf switch (c) { 21505bbd2e59SKevin Wolf case 'c': 215119dbecdcSKevin Wolf if (bdrv_parse_cache_mode(optarg, &flags, &writethrough) < 0) { 21525bbd2e59SKevin Wolf error_report("Invalid cache option: %s", optarg); 21535bbd2e59SKevin Wolf return 0; 21545bbd2e59SKevin Wolf } 21555bbd2e59SKevin Wolf break; 21565bbd2e59SKevin Wolf case 'o': 21575bbd2e59SKevin Wolf if (!qemu_opts_parse_noisily(&reopen_opts, optarg, 0)) { 21585bbd2e59SKevin Wolf qemu_opts_reset(&reopen_opts); 21595bbd2e59SKevin Wolf return 0; 21605bbd2e59SKevin Wolf } 21615bbd2e59SKevin Wolf break; 21625bbd2e59SKevin Wolf case 'r': 21635bbd2e59SKevin Wolf flags &= ~BDRV_O_RDWR; 21645bbd2e59SKevin Wolf break; 21655bbd2e59SKevin Wolf default: 21665bbd2e59SKevin Wolf qemu_opts_reset(&reopen_opts); 21675bbd2e59SKevin Wolf return qemuio_command_usage(&reopen_cmd); 21685bbd2e59SKevin Wolf } 21695bbd2e59SKevin Wolf } 21705bbd2e59SKevin Wolf 21715bbd2e59SKevin Wolf if (optind != argc) { 21725bbd2e59SKevin Wolf qemu_opts_reset(&reopen_opts); 21735bbd2e59SKevin Wolf return qemuio_command_usage(&reopen_cmd); 21745bbd2e59SKevin Wolf } 21755bbd2e59SKevin Wolf 217619dbecdcSKevin Wolf if (writethrough != blk_enable_write_cache(blk) && 217719dbecdcSKevin Wolf blk_get_attached_dev(blk)) 217819dbecdcSKevin Wolf { 217919dbecdcSKevin Wolf error_report("Cannot change cache.writeback: Device attached"); 218019dbecdcSKevin Wolf qemu_opts_reset(&reopen_opts); 218119dbecdcSKevin Wolf return 0; 218219dbecdcSKevin Wolf } 218319dbecdcSKevin Wolf 21845bbd2e59SKevin Wolf qopts = qemu_opts_find(&reopen_opts, NULL); 21855bbd2e59SKevin Wolf opts = qopts ? qemu_opts_to_qdict(qopts, NULL) : NULL; 21865bbd2e59SKevin Wolf qemu_opts_reset(&reopen_opts); 21875bbd2e59SKevin Wolf 21885bbd2e59SKevin Wolf brq = bdrv_reopen_queue(NULL, bs, opts, flags); 21895bbd2e59SKevin Wolf bdrv_reopen_multiple(brq, &local_err); 21905bbd2e59SKevin Wolf if (local_err) { 21915bbd2e59SKevin Wolf error_report_err(local_err); 219219dbecdcSKevin Wolf } else { 219319dbecdcSKevin Wolf blk_set_enable_write_cache(blk, !writethrough); 21945bbd2e59SKevin Wolf } 21955bbd2e59SKevin Wolf 21965bbd2e59SKevin Wolf return 0; 21975bbd2e59SKevin Wolf } 21985bbd2e59SKevin Wolf 21994c7b7e9bSMax Reitz static int break_f(BlockBackend *blk, int argc, char **argv) 2200797ac58cSKevin Wolf { 2201797ac58cSKevin Wolf int ret; 2202797ac58cSKevin Wolf 22034c7b7e9bSMax Reitz ret = bdrv_debug_breakpoint(blk_bs(blk), argv[1], argv[2]); 2204797ac58cSKevin Wolf if (ret < 0) { 2205797ac58cSKevin Wolf printf("Could not set breakpoint: %s\n", strerror(-ret)); 2206797ac58cSKevin Wolf } 2207797ac58cSKevin Wolf 2208797ac58cSKevin Wolf return 0; 2209797ac58cSKevin Wolf } 2210797ac58cSKevin Wolf 22114c7b7e9bSMax Reitz static int remove_break_f(BlockBackend *blk, int argc, char **argv) 22124cc70e93SFam Zheng { 22134cc70e93SFam Zheng int ret; 22144cc70e93SFam Zheng 22154c7b7e9bSMax Reitz ret = bdrv_debug_remove_breakpoint(blk_bs(blk), argv[1]); 22164cc70e93SFam Zheng if (ret < 0) { 22174cc70e93SFam Zheng printf("Could not remove breakpoint %s: %s\n", argv[1], strerror(-ret)); 22184cc70e93SFam Zheng } 22194cc70e93SFam Zheng 22204cc70e93SFam Zheng return 0; 22214cc70e93SFam Zheng } 22224cc70e93SFam Zheng 2223797ac58cSKevin Wolf static const cmdinfo_t break_cmd = { 2224797ac58cSKevin Wolf .name = "break", 2225797ac58cSKevin Wolf .argmin = 2, 2226797ac58cSKevin Wolf .argmax = 2, 2227797ac58cSKevin Wolf .cfunc = break_f, 2228797ac58cSKevin Wolf .args = "event tag", 2229797ac58cSKevin Wolf .oneline = "sets a breakpoint on event and tags the stopped " 2230797ac58cSKevin Wolf "request as tag", 2231797ac58cSKevin Wolf }; 2232797ac58cSKevin Wolf 22334cc70e93SFam Zheng static const cmdinfo_t remove_break_cmd = { 22344cc70e93SFam Zheng .name = "remove_break", 22354cc70e93SFam Zheng .argmin = 1, 22364cc70e93SFam Zheng .argmax = 1, 22374cc70e93SFam Zheng .cfunc = remove_break_f, 22384cc70e93SFam Zheng .args = "tag", 22394cc70e93SFam Zheng .oneline = "remove a breakpoint by tag", 22404cc70e93SFam Zheng }; 22414cc70e93SFam Zheng 22424c7b7e9bSMax Reitz static int resume_f(BlockBackend *blk, int argc, char **argv) 2243797ac58cSKevin Wolf { 2244797ac58cSKevin Wolf int ret; 2245797ac58cSKevin Wolf 22464c7b7e9bSMax Reitz ret = bdrv_debug_resume(blk_bs(blk), argv[1]); 2247797ac58cSKevin Wolf if (ret < 0) { 2248797ac58cSKevin Wolf printf("Could not resume request: %s\n", strerror(-ret)); 2249797ac58cSKevin Wolf } 2250797ac58cSKevin Wolf 2251797ac58cSKevin Wolf return 0; 2252797ac58cSKevin Wolf } 2253797ac58cSKevin Wolf 2254797ac58cSKevin Wolf static const cmdinfo_t resume_cmd = { 2255797ac58cSKevin Wolf .name = "resume", 2256797ac58cSKevin Wolf .argmin = 1, 2257797ac58cSKevin Wolf .argmax = 1, 2258797ac58cSKevin Wolf .cfunc = resume_f, 2259797ac58cSKevin Wolf .args = "tag", 2260797ac58cSKevin Wolf .oneline = "resumes the request tagged as tag", 2261797ac58cSKevin Wolf }; 2262797ac58cSKevin Wolf 22634c7b7e9bSMax Reitz static int wait_break_f(BlockBackend *blk, int argc, char **argv) 2264797ac58cSKevin Wolf { 22654c7b7e9bSMax Reitz while (!bdrv_debug_is_suspended(blk_bs(blk), argv[1])) { 22664c7b7e9bSMax Reitz aio_poll(blk_get_aio_context(blk), true); 2267797ac58cSKevin Wolf } 2268797ac58cSKevin Wolf 2269797ac58cSKevin Wolf return 0; 2270797ac58cSKevin Wolf } 2271797ac58cSKevin Wolf 2272797ac58cSKevin Wolf static const cmdinfo_t wait_break_cmd = { 2273797ac58cSKevin Wolf .name = "wait_break", 2274797ac58cSKevin Wolf .argmin = 1, 2275797ac58cSKevin Wolf .argmax = 1, 2276797ac58cSKevin Wolf .cfunc = wait_break_f, 2277797ac58cSKevin Wolf .args = "tag", 2278797ac58cSKevin Wolf .oneline = "waits for the suspension of a request", 2279797ac58cSKevin Wolf }; 2280797ac58cSKevin Wolf 22814c7b7e9bSMax Reitz static int abort_f(BlockBackend *blk, int argc, char **argv) 2282797ac58cSKevin Wolf { 2283797ac58cSKevin Wolf abort(); 2284797ac58cSKevin Wolf } 2285797ac58cSKevin Wolf 2286797ac58cSKevin Wolf static const cmdinfo_t abort_cmd = { 2287797ac58cSKevin Wolf .name = "abort", 2288797ac58cSKevin Wolf .cfunc = abort_f, 2289797ac58cSKevin Wolf .flags = CMD_NOFILE_OK, 2290797ac58cSKevin Wolf .oneline = "simulate a program crash using abort(3)", 2291797ac58cSKevin Wolf }; 2292797ac58cSKevin Wolf 22930e82dc7bSMax Reitz static void sigraise_help(void) 22940e82dc7bSMax Reitz { 22950e82dc7bSMax Reitz printf( 22960e82dc7bSMax Reitz "\n" 22970e82dc7bSMax Reitz " raises the given signal\n" 22980e82dc7bSMax Reitz "\n" 22990e82dc7bSMax Reitz " Example:\n" 23000e82dc7bSMax Reitz " 'sigraise %i' - raises SIGTERM\n" 23010e82dc7bSMax Reitz "\n" 23020e82dc7bSMax Reitz " Invokes raise(signal), where \"signal\" is the mandatory integer argument\n" 23030e82dc7bSMax Reitz " given to sigraise.\n" 23040e82dc7bSMax Reitz "\n", SIGTERM); 23050e82dc7bSMax Reitz } 23060e82dc7bSMax Reitz 23074c7b7e9bSMax Reitz static int sigraise_f(BlockBackend *blk, int argc, char **argv); 23080e82dc7bSMax Reitz 23090e82dc7bSMax Reitz static const cmdinfo_t sigraise_cmd = { 23100e82dc7bSMax Reitz .name = "sigraise", 23110e82dc7bSMax Reitz .cfunc = sigraise_f, 23120e82dc7bSMax Reitz .argmin = 1, 23130e82dc7bSMax Reitz .argmax = 1, 23140e82dc7bSMax Reitz .flags = CMD_NOFILE_OK, 23150e82dc7bSMax Reitz .args = "signal", 23160e82dc7bSMax Reitz .oneline = "raises a signal", 23170e82dc7bSMax Reitz .help = sigraise_help, 23180e82dc7bSMax Reitz }; 23190e82dc7bSMax Reitz 23204c7b7e9bSMax Reitz static int sigraise_f(BlockBackend *blk, int argc, char **argv) 23210e82dc7bSMax Reitz { 23229b0beaf3SJohn Snow int64_t sig = cvtnum(argv[1]); 23230e82dc7bSMax Reitz if (sig < 0) { 2324a9ecfa00SJohn Snow print_cvtnum_err(sig, argv[1]); 23250e82dc7bSMax Reitz return 0; 23269b0beaf3SJohn Snow } else if (sig > NSIG) { 23279b0beaf3SJohn Snow printf("signal argument '%s' is too large to be a valid signal\n", 23289b0beaf3SJohn Snow argv[1]); 23299b0beaf3SJohn Snow return 0; 23300e82dc7bSMax Reitz } 23310e82dc7bSMax Reitz 23320e82dc7bSMax Reitz /* Using raise() to kill this process does not necessarily flush all open 23330e82dc7bSMax Reitz * streams. At least stdout and stderr (although the latter should be 23340e82dc7bSMax Reitz * non-buffered anyway) should be flushed, though. */ 23350e82dc7bSMax Reitz fflush(stdout); 23360e82dc7bSMax Reitz fflush(stderr); 23370e82dc7bSMax Reitz 23380e82dc7bSMax Reitz raise(sig); 23390e82dc7bSMax Reitz return 0; 23400e82dc7bSMax Reitz } 23410e82dc7bSMax Reitz 2342cd33d02aSKevin Wolf static void sleep_cb(void *opaque) 2343cd33d02aSKevin Wolf { 2344cd33d02aSKevin Wolf bool *expired = opaque; 2345cd33d02aSKevin Wolf *expired = true; 2346cd33d02aSKevin Wolf } 2347cd33d02aSKevin Wolf 23484c7b7e9bSMax Reitz static int sleep_f(BlockBackend *blk, int argc, char **argv) 2349cd33d02aSKevin Wolf { 2350cd33d02aSKevin Wolf char *endptr; 2351cd33d02aSKevin Wolf long ms; 2352cd33d02aSKevin Wolf struct QEMUTimer *timer; 2353cd33d02aSKevin Wolf bool expired = false; 2354cd33d02aSKevin Wolf 2355cd33d02aSKevin Wolf ms = strtol(argv[1], &endptr, 0); 2356cd33d02aSKevin Wolf if (ms < 0 || *endptr != '\0') { 2357cd33d02aSKevin Wolf printf("%s is not a valid number\n", argv[1]); 2358cd33d02aSKevin Wolf return 0; 2359cd33d02aSKevin Wolf } 2360cd33d02aSKevin Wolf 2361cd33d02aSKevin Wolf timer = timer_new_ns(QEMU_CLOCK_HOST, sleep_cb, &expired); 2362cd33d02aSKevin Wolf timer_mod(timer, qemu_clock_get_ns(QEMU_CLOCK_HOST) + SCALE_MS * ms); 2363cd33d02aSKevin Wolf 2364cd33d02aSKevin Wolf while (!expired) { 2365cd33d02aSKevin Wolf main_loop_wait(false); 2366cd33d02aSKevin Wolf } 2367cd33d02aSKevin Wolf 2368cd33d02aSKevin Wolf timer_free(timer); 2369cd33d02aSKevin Wolf 2370cd33d02aSKevin Wolf return 0; 2371cd33d02aSKevin Wolf } 2372cd33d02aSKevin Wolf 2373cd33d02aSKevin Wolf static const cmdinfo_t sleep_cmd = { 2374cd33d02aSKevin Wolf .name = "sleep", 2375cd33d02aSKevin Wolf .argmin = 1, 2376cd33d02aSKevin Wolf .argmax = 1, 2377cd33d02aSKevin Wolf .cfunc = sleep_f, 2378cd33d02aSKevin Wolf .flags = CMD_NOFILE_OK, 2379cd33d02aSKevin Wolf .oneline = "waits for the given value in milliseconds", 2380cd33d02aSKevin Wolf }; 2381cd33d02aSKevin Wolf 2382f18a834aSKevin Wolf static void help_oneline(const char *cmd, const cmdinfo_t *ct) 2383f18a834aSKevin Wolf { 2384f18a834aSKevin Wolf if (cmd) { 2385f18a834aSKevin Wolf printf("%s ", cmd); 2386f18a834aSKevin Wolf } else { 2387f18a834aSKevin Wolf printf("%s ", ct->name); 2388f18a834aSKevin Wolf if (ct->altname) { 2389f18a834aSKevin Wolf printf("(or %s) ", ct->altname); 2390f18a834aSKevin Wolf } 2391f18a834aSKevin Wolf } 2392f18a834aSKevin Wolf 2393f18a834aSKevin Wolf if (ct->args) { 2394f18a834aSKevin Wolf printf("%s ", ct->args); 2395f18a834aSKevin Wolf } 2396f18a834aSKevin Wolf printf("-- %s\n", ct->oneline); 2397f18a834aSKevin Wolf } 2398f18a834aSKevin Wolf 2399f18a834aSKevin Wolf static void help_onecmd(const char *cmd, const cmdinfo_t *ct) 2400f18a834aSKevin Wolf { 2401f18a834aSKevin Wolf help_oneline(cmd, ct); 2402f18a834aSKevin Wolf if (ct->help) { 2403f18a834aSKevin Wolf ct->help(); 2404f18a834aSKevin Wolf } 2405f18a834aSKevin Wolf } 2406f18a834aSKevin Wolf 2407f18a834aSKevin Wolf static void help_all(void) 2408f18a834aSKevin Wolf { 2409f18a834aSKevin Wolf const cmdinfo_t *ct; 2410f18a834aSKevin Wolf 2411f18a834aSKevin Wolf for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { 2412f18a834aSKevin Wolf help_oneline(ct->name, ct); 2413f18a834aSKevin Wolf } 2414f18a834aSKevin Wolf printf("\nUse 'help commandname' for extended help.\n"); 2415f18a834aSKevin Wolf } 2416f18a834aSKevin Wolf 24174c7b7e9bSMax Reitz static int help_f(BlockBackend *blk, int argc, char **argv) 2418f18a834aSKevin Wolf { 2419f18a834aSKevin Wolf const cmdinfo_t *ct; 2420f18a834aSKevin Wolf 2421f18a834aSKevin Wolf if (argc == 1) { 2422f18a834aSKevin Wolf help_all(); 2423f18a834aSKevin Wolf return 0; 2424f18a834aSKevin Wolf } 2425f18a834aSKevin Wolf 2426f18a834aSKevin Wolf ct = find_command(argv[1]); 2427f18a834aSKevin Wolf if (ct == NULL) { 2428f18a834aSKevin Wolf printf("command %s not found\n", argv[1]); 2429f18a834aSKevin Wolf return 0; 2430f18a834aSKevin Wolf } 2431f18a834aSKevin Wolf 2432f18a834aSKevin Wolf help_onecmd(argv[1], ct); 2433f18a834aSKevin Wolf return 0; 2434f18a834aSKevin Wolf } 2435f18a834aSKevin Wolf 2436f18a834aSKevin Wolf static const cmdinfo_t help_cmd = { 2437f18a834aSKevin Wolf .name = "help", 2438f18a834aSKevin Wolf .altname = "?", 2439f18a834aSKevin Wolf .cfunc = help_f, 2440f18a834aSKevin Wolf .argmin = 0, 2441f18a834aSKevin Wolf .argmax = 1, 2442f18a834aSKevin Wolf .flags = CMD_FLAG_GLOBAL, 2443f18a834aSKevin Wolf .args = "[command]", 2444f18a834aSKevin Wolf .oneline = "help for one or all commands", 2445f18a834aSKevin Wolf }; 2446f18a834aSKevin Wolf 24474c7b7e9bSMax Reitz bool qemuio_command(BlockBackend *blk, const char *cmd) 2448dd583296SKevin Wolf { 2449dd583296SKevin Wolf char *input; 2450dd583296SKevin Wolf const cmdinfo_t *ct; 2451dd583296SKevin Wolf char **v; 2452dd583296SKevin Wolf int c; 2453dd583296SKevin Wolf bool done = false; 2454dd583296SKevin Wolf 2455dd583296SKevin Wolf input = g_strdup(cmd); 2456dd583296SKevin Wolf v = breakline(input, &c); 2457dd583296SKevin Wolf if (c) { 2458dd583296SKevin Wolf ct = find_command(v[0]); 2459dd583296SKevin Wolf if (ct) { 24604c7b7e9bSMax Reitz done = command(blk, ct, c, v); 2461dd583296SKevin Wolf } else { 2462dd583296SKevin Wolf fprintf(stderr, "command \"%s\" not found\n", v[0]); 2463dd583296SKevin Wolf } 2464dd583296SKevin Wolf } 2465dd583296SKevin Wolf g_free(input); 2466dd583296SKevin Wolf g_free(v); 2467dd583296SKevin Wolf 2468dd583296SKevin Wolf return done; 2469dd583296SKevin Wolf } 2470dd583296SKevin Wolf 2471797ac58cSKevin Wolf static void __attribute((constructor)) init_qemuio_commands(void) 2472797ac58cSKevin Wolf { 2473797ac58cSKevin Wolf /* initialize commands */ 2474c2cdf5c5SKevin Wolf qemuio_add_command(&help_cmd); 2475c2cdf5c5SKevin Wolf qemuio_add_command(&read_cmd); 2476c2cdf5c5SKevin Wolf qemuio_add_command(&readv_cmd); 2477c2cdf5c5SKevin Wolf qemuio_add_command(&write_cmd); 2478c2cdf5c5SKevin Wolf qemuio_add_command(&writev_cmd); 2479c2cdf5c5SKevin Wolf qemuio_add_command(&multiwrite_cmd); 2480c2cdf5c5SKevin Wolf qemuio_add_command(&aio_read_cmd); 2481c2cdf5c5SKevin Wolf qemuio_add_command(&aio_write_cmd); 2482c2cdf5c5SKevin Wolf qemuio_add_command(&aio_flush_cmd); 2483c2cdf5c5SKevin Wolf qemuio_add_command(&flush_cmd); 2484c2cdf5c5SKevin Wolf qemuio_add_command(&truncate_cmd); 2485c2cdf5c5SKevin Wolf qemuio_add_command(&length_cmd); 2486c2cdf5c5SKevin Wolf qemuio_add_command(&info_cmd); 2487c2cdf5c5SKevin Wolf qemuio_add_command(&discard_cmd); 2488c2cdf5c5SKevin Wolf qemuio_add_command(&alloc_cmd); 2489c2cdf5c5SKevin Wolf qemuio_add_command(&map_cmd); 24905bbd2e59SKevin Wolf qemuio_add_command(&reopen_cmd); 2491c2cdf5c5SKevin Wolf qemuio_add_command(&break_cmd); 24924cc70e93SFam Zheng qemuio_add_command(&remove_break_cmd); 2493c2cdf5c5SKevin Wolf qemuio_add_command(&resume_cmd); 2494c2cdf5c5SKevin Wolf qemuio_add_command(&wait_break_cmd); 2495c2cdf5c5SKevin Wolf qemuio_add_command(&abort_cmd); 2496cd33d02aSKevin Wolf qemuio_add_command(&sleep_cmd); 24970e82dc7bSMax Reitz qemuio_add_command(&sigraise_cmd); 2498797ac58cSKevin Wolf } 2499