1797ac58cSKevin Wolf /* 2797ac58cSKevin Wolf * Command line utility to exercise the QEMU I/O path. 3797ac58cSKevin Wolf * 4093ea232SEric Blake * Copyright (C) 2009-2016 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" 13dc900c35SAlberto Garcia #include "qapi/qmp/qdict.h" 143d21994fSKevin Wolf #include "qemu-io.h" 154c7b7e9bSMax Reitz #include "sysemu/block-backend.h" 164c7b7e9bSMax Reitz #include "block/block.h" 174c7b7e9bSMax Reitz #include "block/block_int.h" /* for info_f() */ 18a8d8ecb7SMax Reitz #include "block/qapi.h" 19d49b6836SMarkus Armbruster #include "qemu/error-report.h" 206a1751b7SAlex Bligh #include "qemu/main-loop.h" 21922a01a0SMarkus Armbruster #include "qemu/option.h" 22cd33d02aSKevin Wolf #include "qemu/timer.h" 23f348b6d1SVeronia Bahaa #include "qemu/cutils.h" 24797ac58cSKevin Wolf 25797ac58cSKevin Wolf #define CMD_NOFILE_OK 0x01 26797ac58cSKevin Wolf 27f9883880SStefan Weil bool qemuio_misalign; 28797ac58cSKevin Wolf 29c2cdf5c5SKevin Wolf static cmdinfo_t *cmdtab; 30c2cdf5c5SKevin Wolf static int ncmds; 31c2cdf5c5SKevin Wolf 32c2cdf5c5SKevin Wolf static int compare_cmdname(const void *a, const void *b) 33c2cdf5c5SKevin Wolf { 34c2cdf5c5SKevin Wolf return strcmp(((const cmdinfo_t *)a)->name, 35c2cdf5c5SKevin Wolf ((const cmdinfo_t *)b)->name); 36c2cdf5c5SKevin Wolf } 37c2cdf5c5SKevin Wolf 38c2cdf5c5SKevin Wolf void qemuio_add_command(const cmdinfo_t *ci) 39c2cdf5c5SKevin Wolf { 406aabeb58SPeter Maydell /* ci->perm assumes a file is open, but the GLOBAL and NOFILE_OK 416aabeb58SPeter Maydell * flags allow it not to be, so that combination is invalid. 426aabeb58SPeter Maydell * Catch it now rather than letting it manifest as a crash if a 436aabeb58SPeter Maydell * particular set of command line options are used. 446aabeb58SPeter Maydell */ 456aabeb58SPeter Maydell assert(ci->perm == 0 || 466aabeb58SPeter Maydell (ci->flags & (CMD_FLAG_GLOBAL | CMD_NOFILE_OK)) == 0); 4702c4f26bSMarkus Armbruster cmdtab = g_renew(cmdinfo_t, cmdtab, ++ncmds); 48c2cdf5c5SKevin Wolf cmdtab[ncmds - 1] = *ci; 49c2cdf5c5SKevin Wolf qsort(cmdtab, ncmds, sizeof(*cmdtab), compare_cmdname); 50c2cdf5c5SKevin Wolf } 51c2cdf5c5SKevin Wolf 52b444d0e9SMax Reitz void qemuio_command_usage(const cmdinfo_t *ci) 53c2cdf5c5SKevin Wolf { 54c2cdf5c5SKevin Wolf printf("%s %s -- %s\n", ci->name, ci->args, ci->oneline); 55c2cdf5c5SKevin Wolf } 56c2cdf5c5SKevin Wolf 574c7b7e9bSMax Reitz static int init_check_command(BlockBackend *blk, const cmdinfo_t *ct) 58c2cdf5c5SKevin Wolf { 59c2cdf5c5SKevin Wolf if (ct->flags & CMD_FLAG_GLOBAL) { 60c2cdf5c5SKevin Wolf return 1; 61c2cdf5c5SKevin Wolf } 624c7b7e9bSMax Reitz if (!(ct->flags & CMD_NOFILE_OK) && !blk) { 63c2cdf5c5SKevin Wolf fprintf(stderr, "no file open, try 'help open'\n"); 64c2cdf5c5SKevin Wolf return 0; 65c2cdf5c5SKevin Wolf } 66c2cdf5c5SKevin Wolf return 1; 67c2cdf5c5SKevin Wolf } 68c2cdf5c5SKevin Wolf 69b32d7a39SMax Reitz static int command(BlockBackend *blk, const cmdinfo_t *ct, int argc, 703d21994fSKevin Wolf char **argv) 71c2cdf5c5SKevin Wolf { 72c2cdf5c5SKevin Wolf char *cmd = argv[0]; 73c2cdf5c5SKevin Wolf 744c7b7e9bSMax Reitz if (!init_check_command(blk, ct)) { 75b32d7a39SMax Reitz return -EINVAL; 76c2cdf5c5SKevin Wolf } 77c2cdf5c5SKevin Wolf 78c2cdf5c5SKevin Wolf if (argc - 1 < ct->argmin || (ct->argmax != -1 && argc - 1 > ct->argmax)) { 79c2cdf5c5SKevin Wolf if (ct->argmax == -1) { 80c2cdf5c5SKevin Wolf fprintf(stderr, 81c2cdf5c5SKevin Wolf "bad argument count %d to %s, expected at least %d arguments\n", 82c2cdf5c5SKevin Wolf argc-1, cmd, ct->argmin); 83c2cdf5c5SKevin Wolf } else if (ct->argmin == ct->argmax) { 84c2cdf5c5SKevin Wolf fprintf(stderr, 85c2cdf5c5SKevin Wolf "bad argument count %d to %s, expected %d arguments\n", 86c2cdf5c5SKevin Wolf argc-1, cmd, ct->argmin); 87c2cdf5c5SKevin Wolf } else { 88c2cdf5c5SKevin Wolf fprintf(stderr, 89c2cdf5c5SKevin Wolf "bad argument count %d to %s, expected between %d and %d arguments\n", 90c2cdf5c5SKevin Wolf argc-1, cmd, ct->argmin, ct->argmax); 91c2cdf5c5SKevin Wolf } 92b32d7a39SMax Reitz return -EINVAL; 93c2cdf5c5SKevin Wolf } 94887354bdSKevin Wolf 95887354bdSKevin Wolf /* Request additional permissions if necessary for this command. The caller 96887354bdSKevin Wolf * is responsible for restoring the original permissions afterwards if this 97887354bdSKevin Wolf * is what it wants. */ 98887354bdSKevin Wolf if (ct->perm && blk_is_available(blk)) { 99887354bdSKevin Wolf uint64_t orig_perm, orig_shared_perm; 100887354bdSKevin Wolf blk_get_perm(blk, &orig_perm, &orig_shared_perm); 101887354bdSKevin Wolf 102887354bdSKevin Wolf if (ct->perm & ~orig_perm) { 103887354bdSKevin Wolf uint64_t new_perm; 104887354bdSKevin Wolf Error *local_err = NULL; 105887354bdSKevin Wolf int ret; 106887354bdSKevin Wolf 107887354bdSKevin Wolf new_perm = orig_perm | ct->perm; 108887354bdSKevin Wolf 109887354bdSKevin Wolf ret = blk_set_perm(blk, new_perm, orig_shared_perm, &local_err); 110887354bdSKevin Wolf if (ret < 0) { 111887354bdSKevin Wolf error_report_err(local_err); 112b32d7a39SMax Reitz return ret; 113887354bdSKevin Wolf } 114887354bdSKevin Wolf } 115887354bdSKevin Wolf } 116887354bdSKevin Wolf 117d339d766SRichard W.M. Jones qemu_reset_optind(); 118b32d7a39SMax Reitz return ct->cfunc(blk, argc, argv); 119c2cdf5c5SKevin Wolf } 120c2cdf5c5SKevin Wolf 121c2cdf5c5SKevin Wolf static const cmdinfo_t *find_command(const char *cmd) 122c2cdf5c5SKevin Wolf { 123c2cdf5c5SKevin Wolf cmdinfo_t *ct; 124c2cdf5c5SKevin Wolf 125c2cdf5c5SKevin Wolf for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { 126c2cdf5c5SKevin Wolf if (strcmp(ct->name, cmd) == 0 || 127c2cdf5c5SKevin Wolf (ct->altname && strcmp(ct->altname, cmd) == 0)) 128c2cdf5c5SKevin Wolf { 129c2cdf5c5SKevin Wolf return (const cmdinfo_t *)ct; 130c2cdf5c5SKevin Wolf } 131c2cdf5c5SKevin Wolf } 132c2cdf5c5SKevin Wolf return NULL; 133c2cdf5c5SKevin Wolf } 134c2cdf5c5SKevin Wolf 1354694020dSStefan Hajnoczi /* Invoke fn() for commands with a matching prefix */ 1364694020dSStefan Hajnoczi void qemuio_complete_command(const char *input, 1374694020dSStefan Hajnoczi void (*fn)(const char *cmd, void *opaque), 1384694020dSStefan Hajnoczi void *opaque) 1394694020dSStefan Hajnoczi { 1404694020dSStefan Hajnoczi cmdinfo_t *ct; 1414694020dSStefan Hajnoczi size_t input_len = strlen(input); 1424694020dSStefan Hajnoczi 1434694020dSStefan Hajnoczi for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { 1444694020dSStefan Hajnoczi if (strncmp(input, ct->name, input_len) == 0) { 1454694020dSStefan Hajnoczi fn(ct->name, opaque); 1464694020dSStefan Hajnoczi } 1474694020dSStefan Hajnoczi } 1484694020dSStefan Hajnoczi } 1494694020dSStefan Hajnoczi 150c2cdf5c5SKevin Wolf static char **breakline(char *input, int *count) 151c2cdf5c5SKevin Wolf { 152c2cdf5c5SKevin Wolf int c = 0; 153c2cdf5c5SKevin Wolf char *p; 1545839e53bSMarkus Armbruster char **rval = g_new0(char *, 1); 155c2cdf5c5SKevin Wolf 156c2cdf5c5SKevin Wolf while (rval && (p = qemu_strsep(&input, " ")) != NULL) { 157c2cdf5c5SKevin Wolf if (!*p) { 158c2cdf5c5SKevin Wolf continue; 159c2cdf5c5SKevin Wolf } 160c2cdf5c5SKevin Wolf c++; 16108193dd5SMarkus Armbruster rval = g_renew(char *, rval, (c + 1)); 162c2cdf5c5SKevin Wolf rval[c - 1] = p; 163c2cdf5c5SKevin Wolf rval[c] = NULL; 164c2cdf5c5SKevin Wolf } 165c2cdf5c5SKevin Wolf *count = c; 166c2cdf5c5SKevin Wolf return rval; 167c2cdf5c5SKevin Wolf } 168c2cdf5c5SKevin Wolf 169797ac58cSKevin Wolf static int64_t cvtnum(const char *s) 170797ac58cSKevin Wolf { 171f17fd4fdSMarkus Armbruster int err; 172f46bfdbfSMarkus Armbruster uint64_t value; 173ef5a7885SJohn Snow 174f17fd4fdSMarkus Armbruster err = qemu_strtosz(s, NULL, &value); 175f17fd4fdSMarkus Armbruster if (err < 0) { 176f17fd4fdSMarkus Armbruster return err; 177f17fd4fdSMarkus Armbruster } 178f46bfdbfSMarkus Armbruster if (value > INT64_MAX) { 179f46bfdbfSMarkus Armbruster return -ERANGE; 180f46bfdbfSMarkus Armbruster } 181f17fd4fdSMarkus Armbruster return value; 182797ac58cSKevin Wolf } 183797ac58cSKevin Wolf 184a9ecfa00SJohn Snow static void print_cvtnum_err(int64_t rc, const char *arg) 185a9ecfa00SJohn Snow { 186a9ecfa00SJohn Snow switch (rc) { 187a9ecfa00SJohn Snow case -EINVAL: 188a9ecfa00SJohn Snow printf("Parsing error: non-numeric argument," 189a9ecfa00SJohn Snow " or extraneous/unrecognized suffix -- %s\n", arg); 190a9ecfa00SJohn Snow break; 191a9ecfa00SJohn Snow case -ERANGE: 192a9ecfa00SJohn Snow printf("Parsing error: argument too large -- %s\n", arg); 193a9ecfa00SJohn Snow break; 194a9ecfa00SJohn Snow default: 195a9ecfa00SJohn Snow printf("Parsing error: %s\n", arg); 196a9ecfa00SJohn Snow } 197a9ecfa00SJohn Snow } 198a9ecfa00SJohn Snow 1990b613881SKevin Wolf #define EXABYTES(x) ((long long)(x) << 60) 2000b613881SKevin Wolf #define PETABYTES(x) ((long long)(x) << 50) 2010b613881SKevin Wolf #define TERABYTES(x) ((long long)(x) << 40) 2020b613881SKevin Wolf #define GIGABYTES(x) ((long long)(x) << 30) 2030b613881SKevin Wolf #define MEGABYTES(x) ((long long)(x) << 20) 2040b613881SKevin Wolf #define KILOBYTES(x) ((long long)(x) << 10) 2050b613881SKevin Wolf 2060b613881SKevin Wolf #define TO_EXABYTES(x) ((x) / EXABYTES(1)) 2070b613881SKevin Wolf #define TO_PETABYTES(x) ((x) / PETABYTES(1)) 2080b613881SKevin Wolf #define TO_TERABYTES(x) ((x) / TERABYTES(1)) 2090b613881SKevin Wolf #define TO_GIGABYTES(x) ((x) / GIGABYTES(1)) 2100b613881SKevin Wolf #define TO_MEGABYTES(x) ((x) / MEGABYTES(1)) 2110b613881SKevin Wolf #define TO_KILOBYTES(x) ((x) / KILOBYTES(1)) 2120b613881SKevin Wolf 2130b613881SKevin Wolf static void cvtstr(double value, char *str, size_t size) 2140b613881SKevin Wolf { 2150b613881SKevin Wolf char *trim; 2160b613881SKevin Wolf const char *suffix; 2170b613881SKevin Wolf 2180b613881SKevin Wolf if (value >= EXABYTES(1)) { 2190b613881SKevin Wolf suffix = " EiB"; 2200b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_EXABYTES(value)); 2210b613881SKevin Wolf } else if (value >= PETABYTES(1)) { 2220b613881SKevin Wolf suffix = " PiB"; 2230b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_PETABYTES(value)); 2240b613881SKevin Wolf } else if (value >= TERABYTES(1)) { 2250b613881SKevin Wolf suffix = " TiB"; 2260b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_TERABYTES(value)); 2270b613881SKevin Wolf } else if (value >= GIGABYTES(1)) { 2280b613881SKevin Wolf suffix = " GiB"; 2290b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_GIGABYTES(value)); 2300b613881SKevin Wolf } else if (value >= MEGABYTES(1)) { 2310b613881SKevin Wolf suffix = " MiB"; 2320b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_MEGABYTES(value)); 2330b613881SKevin Wolf } else if (value >= KILOBYTES(1)) { 2340b613881SKevin Wolf suffix = " KiB"; 2350b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_KILOBYTES(value)); 2360b613881SKevin Wolf } else { 2370b613881SKevin Wolf suffix = " bytes"; 2380b613881SKevin Wolf snprintf(str, size - 6, "%f", value); 2390b613881SKevin Wolf } 2400b613881SKevin Wolf 2410b613881SKevin Wolf trim = strstr(str, ".000"); 2420b613881SKevin Wolf if (trim) { 2430b613881SKevin Wolf strcpy(trim, suffix); 2440b613881SKevin Wolf } else { 2450b613881SKevin Wolf strcat(str, suffix); 2460b613881SKevin Wolf } 2470b613881SKevin Wolf } 2480b613881SKevin Wolf 2490b613881SKevin Wolf 2500b613881SKevin Wolf 25150290c00SAlex Bennée static struct timespec tsub(struct timespec t1, struct timespec t2) 2520b613881SKevin Wolf { 25350290c00SAlex Bennée t1.tv_nsec -= t2.tv_nsec; 25450290c00SAlex Bennée if (t1.tv_nsec < 0) { 25550290c00SAlex Bennée t1.tv_nsec += NANOSECONDS_PER_SECOND; 2560b613881SKevin Wolf t1.tv_sec--; 2570b613881SKevin Wolf } 2580b613881SKevin Wolf t1.tv_sec -= t2.tv_sec; 2590b613881SKevin Wolf return t1; 2600b613881SKevin Wolf } 2610b613881SKevin Wolf 26250290c00SAlex Bennée static double tdiv(double value, struct timespec tv) 2630b613881SKevin Wolf { 26450290c00SAlex Bennée double seconds = tv.tv_sec + (tv.tv_nsec / 1e9); 26550290c00SAlex Bennée return value / seconds; 2660b613881SKevin Wolf } 2670b613881SKevin Wolf 2680b613881SKevin Wolf #define HOURS(sec) ((sec) / (60 * 60)) 2690b613881SKevin Wolf #define MINUTES(sec) (((sec) % (60 * 60)) / 60) 2700b613881SKevin Wolf #define SECONDS(sec) ((sec) % 60) 2710b613881SKevin Wolf 2720b613881SKevin Wolf enum { 2730b613881SKevin Wolf DEFAULT_TIME = 0x0, 2740b613881SKevin Wolf TERSE_FIXED_TIME = 0x1, 2750b613881SKevin Wolf VERBOSE_FIXED_TIME = 0x2, 2760b613881SKevin Wolf }; 2770b613881SKevin Wolf 27850290c00SAlex Bennée static void timestr(struct timespec *tv, char *ts, size_t size, int format) 2790b613881SKevin Wolf { 28050290c00SAlex Bennée double frac_sec = tv->tv_nsec / 1e9; 2810b613881SKevin Wolf 2820b613881SKevin Wolf if (format & TERSE_FIXED_TIME) { 2830b613881SKevin Wolf if (!HOURS(tv->tv_sec)) { 28450290c00SAlex Bennée snprintf(ts, size, "%u:%05.2f", 2850b613881SKevin Wolf (unsigned int) MINUTES(tv->tv_sec), 28650290c00SAlex Bennée SECONDS(tv->tv_sec) + frac_sec); 2870b613881SKevin Wolf return; 2880b613881SKevin Wolf } 2890b613881SKevin Wolf format |= VERBOSE_FIXED_TIME; /* fallback if hours needed */ 2900b613881SKevin Wolf } 2910b613881SKevin Wolf 2920b613881SKevin Wolf if ((format & VERBOSE_FIXED_TIME) || tv->tv_sec) { 29350290c00SAlex Bennée snprintf(ts, size, "%u:%02u:%05.2f", 2940b613881SKevin Wolf (unsigned int) HOURS(tv->tv_sec), 2950b613881SKevin Wolf (unsigned int) MINUTES(tv->tv_sec), 29650290c00SAlex Bennée SECONDS(tv->tv_sec) + frac_sec); 2970b613881SKevin Wolf } else { 29850290c00SAlex Bennée snprintf(ts, size, "%05.2f sec", frac_sec); 2990b613881SKevin Wolf } 3000b613881SKevin Wolf } 3010b613881SKevin Wolf 302797ac58cSKevin Wolf /* 303797ac58cSKevin Wolf * Parse the pattern argument to various sub-commands. 304797ac58cSKevin Wolf * 305797ac58cSKevin Wolf * Because the pattern is used as an argument to memset it must evaluate 306797ac58cSKevin Wolf * to an unsigned integer that fits into a single byte. 307797ac58cSKevin Wolf */ 308797ac58cSKevin Wolf static int parse_pattern(const char *arg) 309797ac58cSKevin Wolf { 310797ac58cSKevin Wolf char *endptr = NULL; 311797ac58cSKevin Wolf long pattern; 312797ac58cSKevin Wolf 313797ac58cSKevin Wolf pattern = strtol(arg, &endptr, 0); 314797ac58cSKevin Wolf if (pattern < 0 || pattern > UCHAR_MAX || *endptr != '\0') { 315797ac58cSKevin Wolf printf("%s is not a valid pattern byte\n", arg); 316797ac58cSKevin Wolf return -1; 317797ac58cSKevin Wolf } 318797ac58cSKevin Wolf 319797ac58cSKevin Wolf return pattern; 320797ac58cSKevin Wolf } 321797ac58cSKevin Wolf 322797ac58cSKevin Wolf /* 323797ac58cSKevin Wolf * Memory allocation helpers. 324797ac58cSKevin Wolf * 325797ac58cSKevin Wolf * Make sure memory is aligned by default, or purposefully misaligned if 326797ac58cSKevin Wolf * that is specified on the command line. 327797ac58cSKevin Wolf */ 328797ac58cSKevin Wolf 329797ac58cSKevin Wolf #define MISALIGN_OFFSET 16 3304c7b7e9bSMax Reitz static void *qemu_io_alloc(BlockBackend *blk, size_t len, int pattern) 331797ac58cSKevin Wolf { 332797ac58cSKevin Wolf void *buf; 333797ac58cSKevin Wolf 334797ac58cSKevin Wolf if (qemuio_misalign) { 335797ac58cSKevin Wolf len += MISALIGN_OFFSET; 336797ac58cSKevin Wolf } 3374c7b7e9bSMax Reitz buf = blk_blockalign(blk, len); 338797ac58cSKevin Wolf memset(buf, pattern, len); 339797ac58cSKevin Wolf if (qemuio_misalign) { 340797ac58cSKevin Wolf buf += MISALIGN_OFFSET; 341797ac58cSKevin Wolf } 342797ac58cSKevin Wolf return buf; 343797ac58cSKevin Wolf } 344797ac58cSKevin Wolf 345797ac58cSKevin Wolf static void qemu_io_free(void *p) 346797ac58cSKevin Wolf { 347797ac58cSKevin Wolf if (qemuio_misalign) { 348797ac58cSKevin Wolf p -= MISALIGN_OFFSET; 349797ac58cSKevin Wolf } 350797ac58cSKevin Wolf qemu_vfree(p); 351797ac58cSKevin Wolf } 352797ac58cSKevin Wolf 353*4d731510SDenis Plotnikov /* 354*4d731510SDenis Plotnikov * qemu_io_alloc_from_file() 355*4d731510SDenis Plotnikov * 356*4d731510SDenis Plotnikov * Allocates the buffer and populates it with the content of the given file 357*4d731510SDenis Plotnikov * up to @len bytes. If the file length is less than @len, then the buffer 358*4d731510SDenis Plotnikov * is populated with the file content cyclically. 359*4d731510SDenis Plotnikov * 360*4d731510SDenis Plotnikov * @blk - the block backend where the buffer content is going to be written to 361*4d731510SDenis Plotnikov * @len - the buffer length 362*4d731510SDenis Plotnikov * @file_name - the file to read the content from 363*4d731510SDenis Plotnikov * 364*4d731510SDenis Plotnikov * Returns: the buffer pointer on success 365*4d731510SDenis Plotnikov * NULL on error 366*4d731510SDenis Plotnikov */ 367*4d731510SDenis Plotnikov static void *qemu_io_alloc_from_file(BlockBackend *blk, size_t len, 368*4d731510SDenis Plotnikov const char *file_name) 369*4d731510SDenis Plotnikov { 370*4d731510SDenis Plotnikov char *buf, *buf_origin; 371*4d731510SDenis Plotnikov FILE *f = fopen(file_name, "r"); 372*4d731510SDenis Plotnikov int pattern_len; 373*4d731510SDenis Plotnikov 374*4d731510SDenis Plotnikov if (!f) { 375*4d731510SDenis Plotnikov perror(file_name); 376*4d731510SDenis Plotnikov return NULL; 377*4d731510SDenis Plotnikov } 378*4d731510SDenis Plotnikov 379*4d731510SDenis Plotnikov if (qemuio_misalign) { 380*4d731510SDenis Plotnikov len += MISALIGN_OFFSET; 381*4d731510SDenis Plotnikov } 382*4d731510SDenis Plotnikov 383*4d731510SDenis Plotnikov buf_origin = buf = blk_blockalign(blk, len); 384*4d731510SDenis Plotnikov 385*4d731510SDenis Plotnikov if (qemuio_misalign) { 386*4d731510SDenis Plotnikov buf_origin += MISALIGN_OFFSET; 387*4d731510SDenis Plotnikov buf += MISALIGN_OFFSET; 388*4d731510SDenis Plotnikov len -= MISALIGN_OFFSET; 389*4d731510SDenis Plotnikov } 390*4d731510SDenis Plotnikov 391*4d731510SDenis Plotnikov pattern_len = fread(buf_origin, 1, len, f); 392*4d731510SDenis Plotnikov 393*4d731510SDenis Plotnikov if (ferror(f)) { 394*4d731510SDenis Plotnikov perror(file_name); 395*4d731510SDenis Plotnikov goto error; 396*4d731510SDenis Plotnikov } 397*4d731510SDenis Plotnikov 398*4d731510SDenis Plotnikov if (pattern_len == 0) { 399*4d731510SDenis Plotnikov fprintf(stderr, "%s: file is empty\n", file_name); 400*4d731510SDenis Plotnikov goto error; 401*4d731510SDenis Plotnikov } 402*4d731510SDenis Plotnikov 403*4d731510SDenis Plotnikov fclose(f); 404*4d731510SDenis Plotnikov 405*4d731510SDenis Plotnikov if (len > pattern_len) { 406*4d731510SDenis Plotnikov len -= pattern_len; 407*4d731510SDenis Plotnikov buf += pattern_len; 408*4d731510SDenis Plotnikov 409*4d731510SDenis Plotnikov while (len > 0) { 410*4d731510SDenis Plotnikov size_t len_to_copy = MIN(pattern_len, len); 411*4d731510SDenis Plotnikov 412*4d731510SDenis Plotnikov memcpy(buf, buf_origin, len_to_copy); 413*4d731510SDenis Plotnikov 414*4d731510SDenis Plotnikov len -= len_to_copy; 415*4d731510SDenis Plotnikov buf += len_to_copy; 416*4d731510SDenis Plotnikov } 417*4d731510SDenis Plotnikov } 418*4d731510SDenis Plotnikov 419*4d731510SDenis Plotnikov return buf_origin; 420*4d731510SDenis Plotnikov 421*4d731510SDenis Plotnikov error: 422*4d731510SDenis Plotnikov qemu_io_free(buf_origin); 423*4d731510SDenis Plotnikov return NULL; 424*4d731510SDenis Plotnikov } 425*4d731510SDenis Plotnikov 4269b0beaf3SJohn Snow static void dump_buffer(const void *buffer, int64_t offset, int64_t len) 427797ac58cSKevin Wolf { 4289b0beaf3SJohn Snow uint64_t i; 4299b0beaf3SJohn Snow int j; 430797ac58cSKevin Wolf const uint8_t *p; 431797ac58cSKevin Wolf 432797ac58cSKevin Wolf for (i = 0, p = buffer; i < len; i += 16) { 433797ac58cSKevin Wolf const uint8_t *s = p; 434797ac58cSKevin Wolf 435797ac58cSKevin Wolf printf("%08" PRIx64 ": ", offset + i); 436797ac58cSKevin Wolf for (j = 0; j < 16 && i + j < len; j++, p++) { 437797ac58cSKevin Wolf printf("%02x ", *p); 438797ac58cSKevin Wolf } 439797ac58cSKevin Wolf printf(" "); 440797ac58cSKevin Wolf for (j = 0; j < 16 && i + j < len; j++, s++) { 441797ac58cSKevin Wolf if (isalnum(*s)) { 442797ac58cSKevin Wolf printf("%c", *s); 443797ac58cSKevin Wolf } else { 444797ac58cSKevin Wolf printf("."); 445797ac58cSKevin Wolf } 446797ac58cSKevin Wolf } 447797ac58cSKevin Wolf printf("\n"); 448797ac58cSKevin Wolf } 449797ac58cSKevin Wolf } 450797ac58cSKevin Wolf 45150290c00SAlex Bennée static void print_report(const char *op, struct timespec *t, int64_t offset, 452dc38852aSEric Blake int64_t count, int64_t total, int cnt, bool Cflag) 453797ac58cSKevin Wolf { 454797ac58cSKevin Wolf char s1[64], s2[64], ts[64]; 455797ac58cSKevin Wolf 456797ac58cSKevin Wolf timestr(t, ts, sizeof(ts), Cflag ? VERBOSE_FIXED_TIME : 0); 457797ac58cSKevin Wolf if (!Cflag) { 458797ac58cSKevin Wolf cvtstr((double)total, s1, sizeof(s1)); 459797ac58cSKevin Wolf cvtstr(tdiv((double)total, *t), s2, sizeof(s2)); 4609b0beaf3SJohn Snow printf("%s %"PRId64"/%"PRId64" bytes at offset %" PRId64 "\n", 461797ac58cSKevin Wolf op, total, count, offset); 462797ac58cSKevin Wolf printf("%s, %d ops; %s (%s/sec and %.4f ops/sec)\n", 463797ac58cSKevin Wolf s1, cnt, ts, s2, tdiv((double)cnt, *t)); 464797ac58cSKevin Wolf } else {/* bytes,ops,time,bytes/sec,ops/sec */ 4659b0beaf3SJohn Snow printf("%"PRId64",%d,%s,%.3f,%.3f\n", 466797ac58cSKevin Wolf total, cnt, ts, 467797ac58cSKevin Wolf tdiv((double)total, *t), 468797ac58cSKevin Wolf tdiv((double)cnt, *t)); 469797ac58cSKevin Wolf } 470797ac58cSKevin Wolf } 471797ac58cSKevin Wolf 472797ac58cSKevin Wolf /* 473797ac58cSKevin Wolf * Parse multiple length statements for vectored I/O, and construct an I/O 474797ac58cSKevin Wolf * vector matching it. 475797ac58cSKevin Wolf */ 476797ac58cSKevin Wolf static void * 4774c7b7e9bSMax Reitz create_iovec(BlockBackend *blk, QEMUIOVector *qiov, char **argv, int nr_iov, 478797ac58cSKevin Wolf int pattern) 479797ac58cSKevin Wolf { 480797ac58cSKevin Wolf size_t *sizes = g_new0(size_t, nr_iov); 481797ac58cSKevin Wolf size_t count = 0; 482797ac58cSKevin Wolf void *buf = NULL; 483797ac58cSKevin Wolf void *p; 484797ac58cSKevin Wolf int i; 485797ac58cSKevin Wolf 486797ac58cSKevin Wolf for (i = 0; i < nr_iov; i++) { 487797ac58cSKevin Wolf char *arg = argv[i]; 488797ac58cSKevin Wolf int64_t len; 489797ac58cSKevin Wolf 490797ac58cSKevin Wolf len = cvtnum(arg); 491797ac58cSKevin Wolf if (len < 0) { 492a9ecfa00SJohn Snow print_cvtnum_err(len, arg); 493797ac58cSKevin Wolf goto fail; 494797ac58cSKevin Wolf } 495797ac58cSKevin Wolf 4963026c468SAlberto Garcia if (len > BDRV_REQUEST_MAX_BYTES) { 4973026c468SAlberto Garcia printf("Argument '%s' exceeds maximum size %" PRIu64 "\n", arg, 4983026c468SAlberto Garcia (uint64_t)BDRV_REQUEST_MAX_BYTES); 4993026c468SAlberto Garcia goto fail; 5003026c468SAlberto Garcia } 5013026c468SAlberto Garcia 5023026c468SAlberto Garcia if (count > BDRV_REQUEST_MAX_BYTES - len) { 5033026c468SAlberto Garcia printf("The total number of bytes exceed the maximum size %" PRIu64 5043026c468SAlberto Garcia "\n", (uint64_t)BDRV_REQUEST_MAX_BYTES); 505797ac58cSKevin Wolf goto fail; 506797ac58cSKevin Wolf } 507797ac58cSKevin Wolf 508797ac58cSKevin Wolf sizes[i] = len; 509797ac58cSKevin Wolf count += len; 510797ac58cSKevin Wolf } 511797ac58cSKevin Wolf 512797ac58cSKevin Wolf qemu_iovec_init(qiov, nr_iov); 513797ac58cSKevin Wolf 5144c7b7e9bSMax Reitz buf = p = qemu_io_alloc(blk, count, pattern); 515797ac58cSKevin Wolf 516797ac58cSKevin Wolf for (i = 0; i < nr_iov; i++) { 517797ac58cSKevin Wolf qemu_iovec_add(qiov, p, sizes[i]); 518797ac58cSKevin Wolf p += sizes[i]; 519797ac58cSKevin Wolf } 520797ac58cSKevin Wolf 521797ac58cSKevin Wolf fail: 522797ac58cSKevin Wolf g_free(sizes); 523797ac58cSKevin Wolf return buf; 524797ac58cSKevin Wolf } 525797ac58cSKevin Wolf 5269b0beaf3SJohn Snow static int do_pread(BlockBackend *blk, char *buf, int64_t offset, 527f5a5ca79SManos Pitsidianakis int64_t bytes, int64_t *total) 528797ac58cSKevin Wolf { 529f5a5ca79SManos Pitsidianakis if (bytes > INT_MAX) { 5309b0beaf3SJohn Snow return -ERANGE; 5319b0beaf3SJohn Snow } 5329b0beaf3SJohn Snow 533f5a5ca79SManos Pitsidianakis *total = blk_pread(blk, offset, (uint8_t *)buf, bytes); 534797ac58cSKevin Wolf if (*total < 0) { 535797ac58cSKevin Wolf return *total; 536797ac58cSKevin Wolf } 537797ac58cSKevin Wolf return 1; 538797ac58cSKevin Wolf } 539797ac58cSKevin Wolf 5409b0beaf3SJohn Snow static int do_pwrite(BlockBackend *blk, char *buf, int64_t offset, 541f5a5ca79SManos Pitsidianakis int64_t bytes, int flags, int64_t *total) 542797ac58cSKevin Wolf { 543f5a5ca79SManos Pitsidianakis if (bytes > INT_MAX) { 5449b0beaf3SJohn Snow return -ERANGE; 5459b0beaf3SJohn Snow } 5469b0beaf3SJohn Snow 547f5a5ca79SManos Pitsidianakis *total = blk_pwrite(blk, offset, (uint8_t *)buf, bytes, flags); 548797ac58cSKevin Wolf if (*total < 0) { 549797ac58cSKevin Wolf return *total; 550797ac58cSKevin Wolf } 551797ac58cSKevin Wolf return 1; 552797ac58cSKevin Wolf } 553797ac58cSKevin Wolf 554797ac58cSKevin Wolf typedef struct { 5554c7b7e9bSMax Reitz BlockBackend *blk; 556797ac58cSKevin Wolf int64_t offset; 557f5a5ca79SManos Pitsidianakis int64_t bytes; 5589b0beaf3SJohn Snow int64_t *total; 559770e0e0eSEric Blake int flags; 560797ac58cSKevin Wolf int ret; 561797ac58cSKevin Wolf bool done; 562797ac58cSKevin Wolf } CoWriteZeroes; 563797ac58cSKevin Wolf 564d004bd52SEric Blake static void coroutine_fn co_pwrite_zeroes_entry(void *opaque) 565797ac58cSKevin Wolf { 566797ac58cSKevin Wolf CoWriteZeroes *data = opaque; 567797ac58cSKevin Wolf 568f5a5ca79SManos Pitsidianakis data->ret = blk_co_pwrite_zeroes(data->blk, data->offset, data->bytes, 569770e0e0eSEric Blake data->flags); 570797ac58cSKevin Wolf data->done = true; 571797ac58cSKevin Wolf if (data->ret < 0) { 572797ac58cSKevin Wolf *data->total = data->ret; 573797ac58cSKevin Wolf return; 574797ac58cSKevin Wolf } 575797ac58cSKevin Wolf 576f5a5ca79SManos Pitsidianakis *data->total = data->bytes; 577797ac58cSKevin Wolf } 578797ac58cSKevin Wolf 579d004bd52SEric Blake static int do_co_pwrite_zeroes(BlockBackend *blk, int64_t offset, 580f5a5ca79SManos Pitsidianakis int64_t bytes, int flags, int64_t *total) 581797ac58cSKevin Wolf { 582797ac58cSKevin Wolf Coroutine *co; 583797ac58cSKevin Wolf CoWriteZeroes data = { 5844c7b7e9bSMax Reitz .blk = blk, 585797ac58cSKevin Wolf .offset = offset, 586f5a5ca79SManos Pitsidianakis .bytes = bytes, 587797ac58cSKevin Wolf .total = total, 588770e0e0eSEric Blake .flags = flags, 589797ac58cSKevin Wolf .done = false, 590797ac58cSKevin Wolf }; 591797ac58cSKevin Wolf 592f5a5ca79SManos Pitsidianakis if (bytes > INT_MAX) { 5939b0beaf3SJohn Snow return -ERANGE; 5949b0beaf3SJohn Snow } 5959b0beaf3SJohn Snow 5960b8b8753SPaolo Bonzini co = qemu_coroutine_create(co_pwrite_zeroes_entry, &data); 597324ec3e4SFam Zheng bdrv_coroutine_enter(blk_bs(blk), co); 598797ac58cSKevin Wolf while (!data.done) { 5994c7b7e9bSMax Reitz aio_poll(blk_get_aio_context(blk), true); 600797ac58cSKevin Wolf } 601797ac58cSKevin Wolf if (data.ret < 0) { 602797ac58cSKevin Wolf return data.ret; 603797ac58cSKevin Wolf } else { 604797ac58cSKevin Wolf return 1; 605797ac58cSKevin Wolf } 606797ac58cSKevin Wolf } 607797ac58cSKevin Wolf 6084c7b7e9bSMax Reitz static int do_write_compressed(BlockBackend *blk, char *buf, int64_t offset, 609f5a5ca79SManos Pitsidianakis int64_t bytes, int64_t *total) 610797ac58cSKevin Wolf { 611797ac58cSKevin Wolf int ret; 612797ac58cSKevin Wolf 61341ae31e3SAlberto Garcia if (bytes > BDRV_REQUEST_MAX_BYTES) { 6149b0beaf3SJohn Snow return -ERANGE; 6159b0beaf3SJohn Snow } 6169b0beaf3SJohn Snow 617f5a5ca79SManos Pitsidianakis ret = blk_pwrite_compressed(blk, offset, buf, bytes); 618797ac58cSKevin Wolf if (ret < 0) { 619797ac58cSKevin Wolf return ret; 620797ac58cSKevin Wolf } 621f5a5ca79SManos Pitsidianakis *total = bytes; 622797ac58cSKevin Wolf return 1; 623797ac58cSKevin Wolf } 624797ac58cSKevin Wolf 6254c7b7e9bSMax Reitz static int do_load_vmstate(BlockBackend *blk, char *buf, int64_t offset, 6269b0beaf3SJohn Snow int64_t count, int64_t *total) 627797ac58cSKevin Wolf { 6289b0beaf3SJohn Snow if (count > INT_MAX) { 6299b0beaf3SJohn Snow return -ERANGE; 6309b0beaf3SJohn Snow } 6319b0beaf3SJohn Snow 6324c7b7e9bSMax Reitz *total = blk_load_vmstate(blk, (uint8_t *)buf, offset, count); 633797ac58cSKevin Wolf if (*total < 0) { 634797ac58cSKevin Wolf return *total; 635797ac58cSKevin Wolf } 636797ac58cSKevin Wolf return 1; 637797ac58cSKevin Wolf } 638797ac58cSKevin Wolf 6394c7b7e9bSMax Reitz static int do_save_vmstate(BlockBackend *blk, char *buf, int64_t offset, 6409b0beaf3SJohn Snow int64_t count, int64_t *total) 641797ac58cSKevin Wolf { 6429b0beaf3SJohn Snow if (count > INT_MAX) { 6439b0beaf3SJohn Snow return -ERANGE; 6449b0beaf3SJohn Snow } 6459b0beaf3SJohn Snow 6464c7b7e9bSMax Reitz *total = blk_save_vmstate(blk, (uint8_t *)buf, offset, count); 647797ac58cSKevin Wolf if (*total < 0) { 648797ac58cSKevin Wolf return *total; 649797ac58cSKevin Wolf } 650797ac58cSKevin Wolf return 1; 651797ac58cSKevin Wolf } 652797ac58cSKevin Wolf 653797ac58cSKevin Wolf #define NOT_DONE 0x7fffffff 654797ac58cSKevin Wolf static void aio_rw_done(void *opaque, int ret) 655797ac58cSKevin Wolf { 656797ac58cSKevin Wolf *(int *)opaque = ret; 657797ac58cSKevin Wolf } 658797ac58cSKevin Wolf 6594c7b7e9bSMax Reitz static int do_aio_readv(BlockBackend *blk, QEMUIOVector *qiov, 660797ac58cSKevin Wolf int64_t offset, int *total) 661797ac58cSKevin Wolf { 662797ac58cSKevin Wolf int async_ret = NOT_DONE; 663797ac58cSKevin Wolf 6647b3f9712SEric Blake blk_aio_preadv(blk, offset, qiov, 0, aio_rw_done, &async_ret); 665797ac58cSKevin Wolf while (async_ret == NOT_DONE) { 666797ac58cSKevin Wolf main_loop_wait(false); 667797ac58cSKevin Wolf } 668797ac58cSKevin Wolf 669797ac58cSKevin Wolf *total = qiov->size; 670797ac58cSKevin Wolf return async_ret < 0 ? async_ret : 1; 671797ac58cSKevin Wolf } 672797ac58cSKevin Wolf 6734c7b7e9bSMax Reitz static int do_aio_writev(BlockBackend *blk, QEMUIOVector *qiov, 674770e0e0eSEric Blake int64_t offset, int flags, int *total) 675797ac58cSKevin Wolf { 676797ac58cSKevin Wolf int async_ret = NOT_DONE; 677797ac58cSKevin Wolf 678770e0e0eSEric Blake blk_aio_pwritev(blk, offset, qiov, flags, aio_rw_done, &async_ret); 679797ac58cSKevin Wolf while (async_ret == NOT_DONE) { 680797ac58cSKevin Wolf main_loop_wait(false); 681797ac58cSKevin Wolf } 682797ac58cSKevin Wolf 683797ac58cSKevin Wolf *total = qiov->size; 684797ac58cSKevin Wolf return async_ret < 0 ? async_ret : 1; 685797ac58cSKevin Wolf } 686797ac58cSKevin Wolf 687797ac58cSKevin Wolf static void read_help(void) 688797ac58cSKevin Wolf { 689797ac58cSKevin Wolf printf( 690797ac58cSKevin Wolf "\n" 691797ac58cSKevin Wolf " reads a range of bytes from the given offset\n" 692797ac58cSKevin Wolf "\n" 693797ac58cSKevin Wolf " Example:\n" 694797ac58cSKevin Wolf " 'read -v 512 1k' - dumps 1 kilobyte read from 512 bytes into the file\n" 695797ac58cSKevin Wolf "\n" 696797ac58cSKevin Wolf " Reads a segment of the currently open file, optionally dumping it to the\n" 697797ac58cSKevin Wolf " standard output stream (with -v option) for subsequent inspection.\n" 698797ac58cSKevin Wolf " -b, -- read from the VM state rather than the virtual disk\n" 699797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 700797ac58cSKevin Wolf " -l, -- length for pattern verification (only with -P)\n" 701093ea232SEric Blake " -p, -- ignored for backwards compatibility\n" 702797ac58cSKevin Wolf " -P, -- use a pattern to verify read data\n" 703797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 704797ac58cSKevin Wolf " -s, -- start offset for pattern verification (only with -P)\n" 705797ac58cSKevin Wolf " -v, -- dump buffer to standard output\n" 706797ac58cSKevin Wolf "\n"); 707797ac58cSKevin Wolf } 708797ac58cSKevin Wolf 709b32d7a39SMax Reitz static int read_f(BlockBackend *blk, int argc, char **argv); 710797ac58cSKevin Wolf 711797ac58cSKevin Wolf static const cmdinfo_t read_cmd = { 712797ac58cSKevin Wolf .name = "read", 713797ac58cSKevin Wolf .altname = "r", 714797ac58cSKevin Wolf .cfunc = read_f, 715797ac58cSKevin Wolf .argmin = 2, 716797ac58cSKevin Wolf .argmax = -1, 717093ea232SEric Blake .args = "[-abCqv] [-P pattern [-s off] [-l len]] off len", 718797ac58cSKevin Wolf .oneline = "reads a number of bytes at a specified offset", 719797ac58cSKevin Wolf .help = read_help, 720797ac58cSKevin Wolf }; 721797ac58cSKevin Wolf 722b32d7a39SMax Reitz static int read_f(BlockBackend *blk, int argc, char **argv) 723797ac58cSKevin Wolf { 72450290c00SAlex Bennée struct timespec t1, t2; 725093ea232SEric Blake bool Cflag = false, qflag = false, vflag = false; 726dc38852aSEric Blake bool Pflag = false, sflag = false, lflag = false, bflag = false; 727b32d7a39SMax Reitz int c, cnt, ret; 728797ac58cSKevin Wolf char *buf; 729797ac58cSKevin Wolf int64_t offset; 7309b0beaf3SJohn Snow int64_t count; 731797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 7329b0beaf3SJohn Snow int64_t total = 0; 7339b0beaf3SJohn Snow int pattern = 0; 7349b0beaf3SJohn Snow int64_t pattern_offset = 0, pattern_count = 0; 735797ac58cSKevin Wolf 736b062ad86SEric Blake while ((c = getopt(argc, argv, "bCl:pP:qs:v")) != -1) { 737797ac58cSKevin Wolf switch (c) { 738797ac58cSKevin Wolf case 'b': 739dc38852aSEric Blake bflag = true; 740797ac58cSKevin Wolf break; 741797ac58cSKevin Wolf case 'C': 742dc38852aSEric Blake Cflag = true; 743797ac58cSKevin Wolf break; 744797ac58cSKevin Wolf case 'l': 745dc38852aSEric Blake lflag = true; 746797ac58cSKevin Wolf pattern_count = cvtnum(optarg); 747797ac58cSKevin Wolf if (pattern_count < 0) { 748a9ecfa00SJohn Snow print_cvtnum_err(pattern_count, optarg); 749b32d7a39SMax Reitz return pattern_count; 750797ac58cSKevin Wolf } 751797ac58cSKevin Wolf break; 752797ac58cSKevin Wolf case 'p': 753093ea232SEric Blake /* Ignored for backwards compatibility */ 754797ac58cSKevin Wolf break; 755797ac58cSKevin Wolf case 'P': 756dc38852aSEric Blake Pflag = true; 757797ac58cSKevin Wolf pattern = parse_pattern(optarg); 758797ac58cSKevin Wolf if (pattern < 0) { 759b32d7a39SMax Reitz return -EINVAL; 760797ac58cSKevin Wolf } 761797ac58cSKevin Wolf break; 762797ac58cSKevin Wolf case 'q': 763dc38852aSEric Blake qflag = true; 764797ac58cSKevin Wolf break; 765797ac58cSKevin Wolf case 's': 766dc38852aSEric Blake sflag = true; 767797ac58cSKevin Wolf pattern_offset = cvtnum(optarg); 768797ac58cSKevin Wolf if (pattern_offset < 0) { 769a9ecfa00SJohn Snow print_cvtnum_err(pattern_offset, optarg); 770b32d7a39SMax Reitz return pattern_offset; 771797ac58cSKevin Wolf } 772797ac58cSKevin Wolf break; 773797ac58cSKevin Wolf case 'v': 774dc38852aSEric Blake vflag = true; 775797ac58cSKevin Wolf break; 776797ac58cSKevin Wolf default: 777b444d0e9SMax Reitz qemuio_command_usage(&read_cmd); 778b32d7a39SMax Reitz return -EINVAL; 779797ac58cSKevin Wolf } 780797ac58cSKevin Wolf } 781797ac58cSKevin Wolf 782797ac58cSKevin Wolf if (optind != argc - 2) { 783b444d0e9SMax Reitz qemuio_command_usage(&read_cmd); 784b32d7a39SMax Reitz return -EINVAL; 785797ac58cSKevin Wolf } 786797ac58cSKevin Wolf 787797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 788797ac58cSKevin Wolf if (offset < 0) { 789a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 790b32d7a39SMax Reitz return offset; 791797ac58cSKevin Wolf } 792797ac58cSKevin Wolf 793797ac58cSKevin Wolf optind++; 794797ac58cSKevin Wolf count = cvtnum(argv[optind]); 795797ac58cSKevin Wolf if (count < 0) { 796a9ecfa00SJohn Snow print_cvtnum_err(count, argv[optind]); 797b32d7a39SMax Reitz return count; 7983026c468SAlberto Garcia } else if (count > BDRV_REQUEST_MAX_BYTES) { 7999b0beaf3SJohn Snow printf("length cannot exceed %" PRIu64 ", given %s\n", 8003026c468SAlberto Garcia (uint64_t)BDRV_REQUEST_MAX_BYTES, argv[optind]); 801b32d7a39SMax Reitz return -EINVAL; 802797ac58cSKevin Wolf } 803797ac58cSKevin Wolf 804797ac58cSKevin Wolf if (!Pflag && (lflag || sflag)) { 805b444d0e9SMax Reitz qemuio_command_usage(&read_cmd); 806b32d7a39SMax Reitz return -EINVAL; 807797ac58cSKevin Wolf } 808797ac58cSKevin Wolf 809797ac58cSKevin Wolf if (!lflag) { 810797ac58cSKevin Wolf pattern_count = count - pattern_offset; 811797ac58cSKevin Wolf } 812797ac58cSKevin Wolf 813797ac58cSKevin Wolf if ((pattern_count < 0) || (pattern_count + pattern_offset > count)) { 814797ac58cSKevin Wolf printf("pattern verification range exceeds end of read data\n"); 815b32d7a39SMax Reitz return -EINVAL; 816797ac58cSKevin Wolf } 817797ac58cSKevin Wolf 818093ea232SEric Blake if (bflag) { 8191bce6b4cSEric Blake if (!QEMU_IS_ALIGNED(offset, BDRV_SECTOR_SIZE)) { 8201bce6b4cSEric Blake printf("%" PRId64 " is not a sector-aligned value for 'offset'\n", 821797ac58cSKevin Wolf offset); 822b32d7a39SMax Reitz return -EINVAL; 823797ac58cSKevin Wolf } 8241bce6b4cSEric Blake if (!QEMU_IS_ALIGNED(count, BDRV_SECTOR_SIZE)) { 8251bce6b4cSEric Blake printf("%"PRId64" is not a sector-aligned value for 'count'\n", 826797ac58cSKevin Wolf count); 827b32d7a39SMax Reitz return -EINVAL; 828797ac58cSKevin Wolf } 829797ac58cSKevin Wolf } 830797ac58cSKevin Wolf 8314c7b7e9bSMax Reitz buf = qemu_io_alloc(blk, count, 0xab); 832797ac58cSKevin Wolf 83350290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t1); 8347b3f9712SEric Blake if (bflag) { 835b32d7a39SMax Reitz ret = do_load_vmstate(blk, buf, offset, count, &total); 836797ac58cSKevin Wolf } else { 837b32d7a39SMax Reitz ret = do_pread(blk, buf, offset, count, &total); 838797ac58cSKevin Wolf } 83950290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 840797ac58cSKevin Wolf 841b32d7a39SMax Reitz if (ret < 0) { 842b32d7a39SMax Reitz printf("read failed: %s\n", strerror(-ret)); 843797ac58cSKevin Wolf goto out; 844797ac58cSKevin Wolf } 845b32d7a39SMax Reitz cnt = ret; 846b32d7a39SMax Reitz 847b32d7a39SMax Reitz ret = 0; 848797ac58cSKevin Wolf 849797ac58cSKevin Wolf if (Pflag) { 850797ac58cSKevin Wolf void *cmp_buf = g_malloc(pattern_count); 851797ac58cSKevin Wolf memset(cmp_buf, pattern, pattern_count); 852797ac58cSKevin Wolf if (memcmp(buf + pattern_offset, cmp_buf, pattern_count)) { 853797ac58cSKevin Wolf printf("Pattern verification failed at offset %" 8549b0beaf3SJohn Snow PRId64 ", %"PRId64" bytes\n", 855797ac58cSKevin Wolf offset + pattern_offset, pattern_count); 856b32d7a39SMax Reitz ret = -EINVAL; 857797ac58cSKevin Wolf } 858797ac58cSKevin Wolf g_free(cmp_buf); 859797ac58cSKevin Wolf } 860797ac58cSKevin Wolf 861797ac58cSKevin Wolf if (qflag) { 862797ac58cSKevin Wolf goto out; 863797ac58cSKevin Wolf } 864797ac58cSKevin Wolf 865797ac58cSKevin Wolf if (vflag) { 866797ac58cSKevin Wolf dump_buffer(buf, offset, count); 867797ac58cSKevin Wolf } 868797ac58cSKevin Wolf 869797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 870797ac58cSKevin Wolf t2 = tsub(t2, t1); 871797ac58cSKevin Wolf print_report("read", &t2, offset, count, total, cnt, Cflag); 872797ac58cSKevin Wolf 873797ac58cSKevin Wolf out: 874797ac58cSKevin Wolf qemu_io_free(buf); 875b32d7a39SMax Reitz return ret; 876797ac58cSKevin Wolf } 877797ac58cSKevin Wolf 878797ac58cSKevin Wolf static void readv_help(void) 879797ac58cSKevin Wolf { 880797ac58cSKevin Wolf printf( 881797ac58cSKevin Wolf "\n" 882797ac58cSKevin Wolf " reads a range of bytes from the given offset into multiple buffers\n" 883797ac58cSKevin Wolf "\n" 884797ac58cSKevin Wolf " Example:\n" 885797ac58cSKevin Wolf " 'readv -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n" 886797ac58cSKevin Wolf "\n" 887797ac58cSKevin Wolf " Reads a segment of the currently open file, optionally dumping it to the\n" 888797ac58cSKevin Wolf " standard output stream (with -v option) for subsequent inspection.\n" 889797ac58cSKevin Wolf " Uses multiple iovec buffers if more than one byte range is specified.\n" 890797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 891797ac58cSKevin Wolf " -P, -- use a pattern to verify read data\n" 892797ac58cSKevin Wolf " -v, -- dump buffer to standard output\n" 893797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 894797ac58cSKevin Wolf "\n"); 895797ac58cSKevin Wolf } 896797ac58cSKevin Wolf 897b32d7a39SMax Reitz static int readv_f(BlockBackend *blk, int argc, char **argv); 898797ac58cSKevin Wolf 899797ac58cSKevin Wolf static const cmdinfo_t readv_cmd = { 900797ac58cSKevin Wolf .name = "readv", 901797ac58cSKevin Wolf .cfunc = readv_f, 902797ac58cSKevin Wolf .argmin = 2, 903797ac58cSKevin Wolf .argmax = -1, 904797ac58cSKevin Wolf .args = "[-Cqv] [-P pattern] off len [len..]", 905797ac58cSKevin Wolf .oneline = "reads a number of bytes at a specified offset", 906797ac58cSKevin Wolf .help = readv_help, 907797ac58cSKevin Wolf }; 908797ac58cSKevin Wolf 909b32d7a39SMax Reitz static int readv_f(BlockBackend *blk, int argc, char **argv) 910797ac58cSKevin Wolf { 91150290c00SAlex Bennée struct timespec t1, t2; 912dc38852aSEric Blake bool Cflag = false, qflag = false, vflag = false; 913b32d7a39SMax Reitz int c, cnt, ret; 914797ac58cSKevin Wolf char *buf; 915797ac58cSKevin Wolf int64_t offset; 916797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 917797ac58cSKevin Wolf int total = 0; 918797ac58cSKevin Wolf int nr_iov; 919797ac58cSKevin Wolf QEMUIOVector qiov; 920797ac58cSKevin Wolf int pattern = 0; 921dc38852aSEric Blake bool Pflag = false; 922797ac58cSKevin Wolf 923b062ad86SEric Blake while ((c = getopt(argc, argv, "CP:qv")) != -1) { 924797ac58cSKevin Wolf switch (c) { 925797ac58cSKevin Wolf case 'C': 926dc38852aSEric Blake Cflag = true; 927797ac58cSKevin Wolf break; 928797ac58cSKevin Wolf case 'P': 929dc38852aSEric Blake Pflag = true; 930797ac58cSKevin Wolf pattern = parse_pattern(optarg); 931797ac58cSKevin Wolf if (pattern < 0) { 932b32d7a39SMax Reitz return -EINVAL; 933797ac58cSKevin Wolf } 934797ac58cSKevin Wolf break; 935797ac58cSKevin Wolf case 'q': 936dc38852aSEric Blake qflag = true; 937797ac58cSKevin Wolf break; 938797ac58cSKevin Wolf case 'v': 939dc38852aSEric Blake vflag = true; 940797ac58cSKevin Wolf break; 941797ac58cSKevin Wolf default: 942b444d0e9SMax Reitz qemuio_command_usage(&readv_cmd); 943b32d7a39SMax Reitz return -EINVAL; 944797ac58cSKevin Wolf } 945797ac58cSKevin Wolf } 946797ac58cSKevin Wolf 947797ac58cSKevin Wolf if (optind > argc - 2) { 948b444d0e9SMax Reitz qemuio_command_usage(&readv_cmd); 949b32d7a39SMax Reitz return -EINVAL; 950797ac58cSKevin Wolf } 951797ac58cSKevin Wolf 952797ac58cSKevin Wolf 953797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 954797ac58cSKevin Wolf if (offset < 0) { 955a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 956b32d7a39SMax Reitz return offset; 957797ac58cSKevin Wolf } 958797ac58cSKevin Wolf optind++; 959797ac58cSKevin Wolf 960797ac58cSKevin Wolf nr_iov = argc - optind; 9614c7b7e9bSMax Reitz buf = create_iovec(blk, &qiov, &argv[optind], nr_iov, 0xab); 962797ac58cSKevin Wolf if (buf == NULL) { 963b32d7a39SMax Reitz return -EINVAL; 964797ac58cSKevin Wolf } 965797ac58cSKevin Wolf 96650290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t1); 967b32d7a39SMax Reitz ret = do_aio_readv(blk, &qiov, offset, &total); 96850290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 969797ac58cSKevin Wolf 970b32d7a39SMax Reitz if (ret < 0) { 971b32d7a39SMax Reitz printf("readv failed: %s\n", strerror(-ret)); 972797ac58cSKevin Wolf goto out; 973797ac58cSKevin Wolf } 974b32d7a39SMax Reitz cnt = ret; 975b32d7a39SMax Reitz 976b32d7a39SMax Reitz ret = 0; 977797ac58cSKevin Wolf 978797ac58cSKevin Wolf if (Pflag) { 979797ac58cSKevin Wolf void *cmp_buf = g_malloc(qiov.size); 980797ac58cSKevin Wolf memset(cmp_buf, pattern, qiov.size); 981797ac58cSKevin Wolf if (memcmp(buf, cmp_buf, qiov.size)) { 982797ac58cSKevin Wolf printf("Pattern verification failed at offset %" 983cf67b692SStefan Weil PRId64 ", %zu bytes\n", offset, qiov.size); 984b32d7a39SMax Reitz ret = -EINVAL; 985797ac58cSKevin Wolf } 986797ac58cSKevin Wolf g_free(cmp_buf); 987797ac58cSKevin Wolf } 988797ac58cSKevin Wolf 989797ac58cSKevin Wolf if (qflag) { 990797ac58cSKevin Wolf goto out; 991797ac58cSKevin Wolf } 992797ac58cSKevin Wolf 993797ac58cSKevin Wolf if (vflag) { 994797ac58cSKevin Wolf dump_buffer(buf, offset, qiov.size); 995797ac58cSKevin Wolf } 996797ac58cSKevin Wolf 997797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 998797ac58cSKevin Wolf t2 = tsub(t2, t1); 999797ac58cSKevin Wolf print_report("read", &t2, offset, qiov.size, total, cnt, Cflag); 1000797ac58cSKevin Wolf 1001797ac58cSKevin Wolf out: 1002797ac58cSKevin Wolf qemu_iovec_destroy(&qiov); 1003797ac58cSKevin Wolf qemu_io_free(buf); 1004b32d7a39SMax Reitz return ret; 1005797ac58cSKevin Wolf } 1006797ac58cSKevin Wolf 1007797ac58cSKevin Wolf static void write_help(void) 1008797ac58cSKevin Wolf { 1009797ac58cSKevin Wolf printf( 1010797ac58cSKevin Wolf "\n" 1011797ac58cSKevin Wolf " writes a range of bytes from the given offset\n" 1012797ac58cSKevin Wolf "\n" 1013797ac58cSKevin Wolf " Example:\n" 1014797ac58cSKevin Wolf " 'write 512 1k' - writes 1 kilobyte at 512 bytes into the open file\n" 1015797ac58cSKevin Wolf "\n" 1016797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 1017797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd).\n" 1018797ac58cSKevin Wolf " -b, -- write to the VM state rather than the virtual disk\n" 10194c7b7e9bSMax Reitz " -c, -- write compressed data with blk_write_compressed\n" 1020770e0e0eSEric Blake " -f, -- use Force Unit Access semantics\n" 1021c6e3f520SKevin Wolf " -n, -- with -z, don't allow slow fallback\n" 1022093ea232SEric Blake " -p, -- ignored for backwards compatibility\n" 1023797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 1024*4d731510SDenis Plotnikov " -s, -- use a pattern file to fill the write buffer\n" 1025797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1026797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1027c2e001ccSEric Blake " -u, -- with -z, allow unmapping\n" 1028d004bd52SEric Blake " -z, -- write zeroes using blk_co_pwrite_zeroes\n" 1029797ac58cSKevin Wolf "\n"); 1030797ac58cSKevin Wolf } 1031797ac58cSKevin Wolf 1032b32d7a39SMax Reitz static int write_f(BlockBackend *blk, int argc, char **argv); 1033797ac58cSKevin Wolf 1034797ac58cSKevin Wolf static const cmdinfo_t write_cmd = { 1035797ac58cSKevin Wolf .name = "write", 1036797ac58cSKevin Wolf .altname = "w", 1037797ac58cSKevin Wolf .cfunc = write_f, 1038887354bdSKevin Wolf .perm = BLK_PERM_WRITE, 1039797ac58cSKevin Wolf .argmin = 2, 1040797ac58cSKevin Wolf .argmax = -1, 1041*4d731510SDenis Plotnikov .args = "[-bcCfnquz] [-P pattern | -s source_file] off len", 1042797ac58cSKevin Wolf .oneline = "writes a number of bytes at a specified offset", 1043797ac58cSKevin Wolf .help = write_help, 1044797ac58cSKevin Wolf }; 1045797ac58cSKevin Wolf 1046b32d7a39SMax Reitz static int write_f(BlockBackend *blk, int argc, char **argv) 1047797ac58cSKevin Wolf { 104850290c00SAlex Bennée struct timespec t1, t2; 1049093ea232SEric Blake bool Cflag = false, qflag = false, bflag = false; 1050*4d731510SDenis Plotnikov bool Pflag = false, zflag = false, cflag = false, sflag = false; 1051770e0e0eSEric Blake int flags = 0; 1052b32d7a39SMax Reitz int c, cnt, ret; 1053797ac58cSKevin Wolf char *buf = NULL; 1054797ac58cSKevin Wolf int64_t offset; 10559b0beaf3SJohn Snow int64_t count; 1056797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 10579b0beaf3SJohn Snow int64_t total = 0; 1058797ac58cSKevin Wolf int pattern = 0xcd; 1059*4d731510SDenis Plotnikov const char *file_name = NULL; 1060797ac58cSKevin Wolf 1061*4d731510SDenis Plotnikov while ((c = getopt(argc, argv, "bcCfnpP:qs:uz")) != -1) { 1062797ac58cSKevin Wolf switch (c) { 1063797ac58cSKevin Wolf case 'b': 1064dc38852aSEric Blake bflag = true; 1065797ac58cSKevin Wolf break; 1066797ac58cSKevin Wolf case 'c': 1067dc38852aSEric Blake cflag = true; 1068797ac58cSKevin Wolf break; 1069797ac58cSKevin Wolf case 'C': 1070dc38852aSEric Blake Cflag = true; 1071797ac58cSKevin Wolf break; 1072770e0e0eSEric Blake case 'f': 1073770e0e0eSEric Blake flags |= BDRV_REQ_FUA; 1074770e0e0eSEric Blake break; 1075c6e3f520SKevin Wolf case 'n': 1076c6e3f520SKevin Wolf flags |= BDRV_REQ_NO_FALLBACK; 1077c6e3f520SKevin Wolf break; 1078797ac58cSKevin Wolf case 'p': 1079093ea232SEric Blake /* Ignored for backwards compatibility */ 1080797ac58cSKevin Wolf break; 1081797ac58cSKevin Wolf case 'P': 1082dc38852aSEric Blake Pflag = true; 1083797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1084797ac58cSKevin Wolf if (pattern < 0) { 1085b32d7a39SMax Reitz return -EINVAL; 1086797ac58cSKevin Wolf } 1087797ac58cSKevin Wolf break; 1088797ac58cSKevin Wolf case 'q': 1089dc38852aSEric Blake qflag = true; 1090797ac58cSKevin Wolf break; 1091*4d731510SDenis Plotnikov case 's': 1092*4d731510SDenis Plotnikov sflag = true; 1093*4d731510SDenis Plotnikov file_name = optarg; 1094*4d731510SDenis Plotnikov break; 1095c2e001ccSEric Blake case 'u': 1096c2e001ccSEric Blake flags |= BDRV_REQ_MAY_UNMAP; 1097c2e001ccSEric Blake break; 1098797ac58cSKevin Wolf case 'z': 1099dc38852aSEric Blake zflag = true; 1100797ac58cSKevin Wolf break; 1101797ac58cSKevin Wolf default: 1102b444d0e9SMax Reitz qemuio_command_usage(&write_cmd); 1103b32d7a39SMax Reitz return -EINVAL; 1104797ac58cSKevin Wolf } 1105797ac58cSKevin Wolf } 1106797ac58cSKevin Wolf 1107797ac58cSKevin Wolf if (optind != argc - 2) { 1108b444d0e9SMax Reitz qemuio_command_usage(&write_cmd); 1109b32d7a39SMax Reitz return -EINVAL; 1110797ac58cSKevin Wolf } 1111797ac58cSKevin Wolf 1112093ea232SEric Blake if (bflag && zflag) { 1113093ea232SEric Blake printf("-b and -z cannot be specified at the same time\n"); 1114b32d7a39SMax Reitz return -EINVAL; 1115797ac58cSKevin Wolf } 1116797ac58cSKevin Wolf 1117770e0e0eSEric Blake if ((flags & BDRV_REQ_FUA) && (bflag || cflag)) { 1118770e0e0eSEric Blake printf("-f and -b or -c cannot be specified at the same time\n"); 1119b32d7a39SMax Reitz return -EINVAL; 1120770e0e0eSEric Blake } 1121770e0e0eSEric Blake 1122c6e3f520SKevin Wolf if ((flags & BDRV_REQ_NO_FALLBACK) && !zflag) { 1123c6e3f520SKevin Wolf printf("-n requires -z to be specified\n"); 1124c6e3f520SKevin Wolf return -EINVAL; 1125c6e3f520SKevin Wolf } 1126c6e3f520SKevin Wolf 1127c2e001ccSEric Blake if ((flags & BDRV_REQ_MAY_UNMAP) && !zflag) { 1128c2e001ccSEric Blake printf("-u requires -z to be specified\n"); 1129b32d7a39SMax Reitz return -EINVAL; 1130c2e001ccSEric Blake } 1131c2e001ccSEric Blake 1132*4d731510SDenis Plotnikov if (zflag + Pflag + sflag > 1) { 1133*4d731510SDenis Plotnikov printf("Only one of -z, -P, and -s " 1134*4d731510SDenis Plotnikov "can be specified at the same time\n"); 1135b32d7a39SMax Reitz return -EINVAL; 1136797ac58cSKevin Wolf } 1137797ac58cSKevin Wolf 1138797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1139797ac58cSKevin Wolf if (offset < 0) { 1140a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 1141b32d7a39SMax Reitz return offset; 1142797ac58cSKevin Wolf } 1143797ac58cSKevin Wolf 1144797ac58cSKevin Wolf optind++; 1145797ac58cSKevin Wolf count = cvtnum(argv[optind]); 1146797ac58cSKevin Wolf if (count < 0) { 1147a9ecfa00SJohn Snow print_cvtnum_err(count, argv[optind]); 1148b32d7a39SMax Reitz return count; 11493026c468SAlberto Garcia } else if (count > BDRV_REQUEST_MAX_BYTES) { 11509b0beaf3SJohn Snow printf("length cannot exceed %" PRIu64 ", given %s\n", 11513026c468SAlberto Garcia (uint64_t)BDRV_REQUEST_MAX_BYTES, argv[optind]); 1152b32d7a39SMax Reitz return -EINVAL; 1153797ac58cSKevin Wolf } 1154797ac58cSKevin Wolf 1155093ea232SEric Blake if (bflag || cflag) { 11561bce6b4cSEric Blake if (!QEMU_IS_ALIGNED(offset, BDRV_SECTOR_SIZE)) { 11571bce6b4cSEric Blake printf("%" PRId64 " is not a sector-aligned value for 'offset'\n", 1158797ac58cSKevin Wolf offset); 1159b32d7a39SMax Reitz return -EINVAL; 1160797ac58cSKevin Wolf } 1161797ac58cSKevin Wolf 11621bce6b4cSEric Blake if (!QEMU_IS_ALIGNED(count, BDRV_SECTOR_SIZE)) { 11631bce6b4cSEric Blake printf("%"PRId64" is not a sector-aligned value for 'count'\n", 1164797ac58cSKevin Wolf count); 1165b32d7a39SMax Reitz return -EINVAL; 1166797ac58cSKevin Wolf } 1167797ac58cSKevin Wolf } 1168797ac58cSKevin Wolf 1169797ac58cSKevin Wolf if (!zflag) { 1170*4d731510SDenis Plotnikov if (sflag) { 1171*4d731510SDenis Plotnikov buf = qemu_io_alloc_from_file(blk, count, file_name); 1172*4d731510SDenis Plotnikov if (!buf) { 1173*4d731510SDenis Plotnikov return -EINVAL; 1174*4d731510SDenis Plotnikov } 1175*4d731510SDenis Plotnikov } else { 11764c7b7e9bSMax Reitz buf = qemu_io_alloc(blk, count, pattern); 1177797ac58cSKevin Wolf } 1178*4d731510SDenis Plotnikov } 1179797ac58cSKevin Wolf 118050290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t1); 11817b3f9712SEric Blake if (bflag) { 1182b32d7a39SMax Reitz ret = do_save_vmstate(blk, buf, offset, count, &total); 1183797ac58cSKevin Wolf } else if (zflag) { 1184b32d7a39SMax Reitz ret = do_co_pwrite_zeroes(blk, offset, count, flags, &total); 1185797ac58cSKevin Wolf } else if (cflag) { 1186b32d7a39SMax Reitz ret = do_write_compressed(blk, buf, offset, count, &total); 1187797ac58cSKevin Wolf } else { 1188b32d7a39SMax Reitz ret = do_pwrite(blk, buf, offset, count, flags, &total); 1189797ac58cSKevin Wolf } 119050290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 1191797ac58cSKevin Wolf 1192b32d7a39SMax Reitz if (ret < 0) { 1193b32d7a39SMax Reitz printf("write failed: %s\n", strerror(-ret)); 1194797ac58cSKevin Wolf goto out; 1195797ac58cSKevin Wolf } 1196b32d7a39SMax Reitz cnt = ret; 1197b32d7a39SMax Reitz 1198b32d7a39SMax Reitz ret = 0; 1199797ac58cSKevin Wolf 1200797ac58cSKevin Wolf if (qflag) { 1201797ac58cSKevin Wolf goto out; 1202797ac58cSKevin Wolf } 1203797ac58cSKevin Wolf 1204797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1205797ac58cSKevin Wolf t2 = tsub(t2, t1); 1206797ac58cSKevin Wolf print_report("wrote", &t2, offset, count, total, cnt, Cflag); 1207797ac58cSKevin Wolf 1208797ac58cSKevin Wolf out: 1209797ac58cSKevin Wolf if (!zflag) { 1210797ac58cSKevin Wolf qemu_io_free(buf); 1211797ac58cSKevin Wolf } 1212b32d7a39SMax Reitz return ret; 1213797ac58cSKevin Wolf } 1214797ac58cSKevin Wolf 1215797ac58cSKevin Wolf static void 1216797ac58cSKevin Wolf writev_help(void) 1217797ac58cSKevin Wolf { 1218797ac58cSKevin Wolf printf( 1219797ac58cSKevin Wolf "\n" 1220797ac58cSKevin Wolf " writes a range of bytes from the given offset source from multiple buffers\n" 1221797ac58cSKevin Wolf "\n" 1222797ac58cSKevin Wolf " Example:\n" 12236e6507c0SMaria Kustova " 'writev 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n" 1224797ac58cSKevin Wolf "\n" 1225797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 1226797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd).\n" 1227797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 1228797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1229770e0e0eSEric Blake " -f, -- use Force Unit Access semantics\n" 1230797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1231797ac58cSKevin Wolf "\n"); 1232797ac58cSKevin Wolf } 1233797ac58cSKevin Wolf 1234b32d7a39SMax Reitz static int writev_f(BlockBackend *blk, int argc, char **argv); 1235797ac58cSKevin Wolf 1236797ac58cSKevin Wolf static const cmdinfo_t writev_cmd = { 1237797ac58cSKevin Wolf .name = "writev", 1238797ac58cSKevin Wolf .cfunc = writev_f, 1239887354bdSKevin Wolf .perm = BLK_PERM_WRITE, 1240797ac58cSKevin Wolf .argmin = 2, 1241797ac58cSKevin Wolf .argmax = -1, 1242770e0e0eSEric Blake .args = "[-Cfq] [-P pattern] off len [len..]", 1243797ac58cSKevin Wolf .oneline = "writes a number of bytes at a specified offset", 1244797ac58cSKevin Wolf .help = writev_help, 1245797ac58cSKevin Wolf }; 1246797ac58cSKevin Wolf 1247b32d7a39SMax Reitz static int writev_f(BlockBackend *blk, int argc, char **argv) 1248797ac58cSKevin Wolf { 124950290c00SAlex Bennée struct timespec t1, t2; 1250dc38852aSEric Blake bool Cflag = false, qflag = false; 1251770e0e0eSEric Blake int flags = 0; 1252b32d7a39SMax Reitz int c, cnt, ret; 1253797ac58cSKevin Wolf char *buf; 1254797ac58cSKevin Wolf int64_t offset; 1255797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 1256797ac58cSKevin Wolf int total = 0; 1257797ac58cSKevin Wolf int nr_iov; 1258797ac58cSKevin Wolf int pattern = 0xcd; 1259797ac58cSKevin Wolf QEMUIOVector qiov; 1260797ac58cSKevin Wolf 12614ca1d340SEric Blake while ((c = getopt(argc, argv, "CfqP:")) != -1) { 1262797ac58cSKevin Wolf switch (c) { 1263797ac58cSKevin Wolf case 'C': 1264dc38852aSEric Blake Cflag = true; 1265797ac58cSKevin Wolf break; 1266770e0e0eSEric Blake case 'f': 1267770e0e0eSEric Blake flags |= BDRV_REQ_FUA; 1268770e0e0eSEric Blake break; 1269797ac58cSKevin Wolf case 'q': 1270dc38852aSEric Blake qflag = true; 1271797ac58cSKevin Wolf break; 1272797ac58cSKevin Wolf case 'P': 1273797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1274797ac58cSKevin Wolf if (pattern < 0) { 1275b32d7a39SMax Reitz return -EINVAL; 1276797ac58cSKevin Wolf } 1277797ac58cSKevin Wolf break; 1278797ac58cSKevin Wolf default: 1279b444d0e9SMax Reitz qemuio_command_usage(&writev_cmd); 1280b32d7a39SMax Reitz return -EINVAL; 1281797ac58cSKevin Wolf } 1282797ac58cSKevin Wolf } 1283797ac58cSKevin Wolf 1284797ac58cSKevin Wolf if (optind > argc - 2) { 1285b444d0e9SMax Reitz qemuio_command_usage(&writev_cmd); 1286b32d7a39SMax Reitz return -EINVAL; 1287797ac58cSKevin Wolf } 1288797ac58cSKevin Wolf 1289797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1290797ac58cSKevin Wolf if (offset < 0) { 1291a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 1292b32d7a39SMax Reitz return offset; 1293797ac58cSKevin Wolf } 1294797ac58cSKevin Wolf optind++; 1295797ac58cSKevin Wolf 1296797ac58cSKevin Wolf nr_iov = argc - optind; 12974c7b7e9bSMax Reitz buf = create_iovec(blk, &qiov, &argv[optind], nr_iov, pattern); 1298797ac58cSKevin Wolf if (buf == NULL) { 1299b32d7a39SMax Reitz return -EINVAL; 1300797ac58cSKevin Wolf } 1301797ac58cSKevin Wolf 130250290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t1); 1303b32d7a39SMax Reitz ret = do_aio_writev(blk, &qiov, offset, flags, &total); 130450290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 1305797ac58cSKevin Wolf 1306b32d7a39SMax Reitz if (ret < 0) { 1307b32d7a39SMax Reitz printf("writev failed: %s\n", strerror(-ret)); 1308797ac58cSKevin Wolf goto out; 1309797ac58cSKevin Wolf } 1310b32d7a39SMax Reitz cnt = ret; 1311b32d7a39SMax Reitz 1312b32d7a39SMax Reitz ret = 0; 1313797ac58cSKevin Wolf 1314797ac58cSKevin Wolf if (qflag) { 1315797ac58cSKevin Wolf goto out; 1316797ac58cSKevin Wolf } 1317797ac58cSKevin Wolf 1318797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1319797ac58cSKevin Wolf t2 = tsub(t2, t1); 1320797ac58cSKevin Wolf print_report("wrote", &t2, offset, qiov.size, total, cnt, Cflag); 1321797ac58cSKevin Wolf out: 1322797ac58cSKevin Wolf qemu_iovec_destroy(&qiov); 1323797ac58cSKevin Wolf qemu_io_free(buf); 1324b32d7a39SMax Reitz return ret; 1325797ac58cSKevin Wolf } 1326797ac58cSKevin Wolf 1327797ac58cSKevin Wolf struct aio_ctx { 13284c7b7e9bSMax Reitz BlockBackend *blk; 1329797ac58cSKevin Wolf QEMUIOVector qiov; 1330797ac58cSKevin Wolf int64_t offset; 1331797ac58cSKevin Wolf char *buf; 1332dc38852aSEric Blake bool qflag; 1333dc38852aSEric Blake bool vflag; 1334dc38852aSEric Blake bool Cflag; 1335dc38852aSEric Blake bool Pflag; 1336dc38852aSEric Blake bool zflag; 1337a91f9584SFam Zheng BlockAcctCookie acct; 1338797ac58cSKevin Wolf int pattern; 133950290c00SAlex Bennée struct timespec t1; 1340797ac58cSKevin Wolf }; 1341797ac58cSKevin Wolf 1342797ac58cSKevin Wolf static void aio_write_done(void *opaque, int ret) 1343797ac58cSKevin Wolf { 1344797ac58cSKevin Wolf struct aio_ctx *ctx = opaque; 134550290c00SAlex Bennée struct timespec t2; 1346797ac58cSKevin Wolf 134750290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 1348797ac58cSKevin Wolf 1349797ac58cSKevin Wolf 1350797ac58cSKevin Wolf if (ret < 0) { 1351797ac58cSKevin Wolf printf("aio_write failed: %s\n", strerror(-ret)); 1352556c2b60SAlberto Garcia block_acct_failed(blk_get_stats(ctx->blk), &ctx->acct); 1353797ac58cSKevin Wolf goto out; 1354797ac58cSKevin Wolf } 1355797ac58cSKevin Wolf 13564c7b7e9bSMax Reitz block_acct_done(blk_get_stats(ctx->blk), &ctx->acct); 1357a91f9584SFam Zheng 1358797ac58cSKevin Wolf if (ctx->qflag) { 1359797ac58cSKevin Wolf goto out; 1360797ac58cSKevin Wolf } 1361797ac58cSKevin Wolf 1362797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1363797ac58cSKevin Wolf t2 = tsub(t2, ctx->t1); 1364797ac58cSKevin Wolf print_report("wrote", &t2, ctx->offset, ctx->qiov.size, 1365797ac58cSKevin Wolf ctx->qiov.size, 1, ctx->Cflag); 1366797ac58cSKevin Wolf out: 13675ceb7765SKevin Wolf if (!ctx->zflag) { 1368797ac58cSKevin Wolf qemu_io_free(ctx->buf); 1369797ac58cSKevin Wolf qemu_iovec_destroy(&ctx->qiov); 13705ceb7765SKevin Wolf } 1371797ac58cSKevin Wolf g_free(ctx); 1372797ac58cSKevin Wolf } 1373797ac58cSKevin Wolf 1374797ac58cSKevin Wolf static void aio_read_done(void *opaque, int ret) 1375797ac58cSKevin Wolf { 1376797ac58cSKevin Wolf struct aio_ctx *ctx = opaque; 137750290c00SAlex Bennée struct timespec t2; 1378797ac58cSKevin Wolf 137950290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 1380797ac58cSKevin Wolf 1381797ac58cSKevin Wolf if (ret < 0) { 1382797ac58cSKevin Wolf printf("readv failed: %s\n", strerror(-ret)); 1383556c2b60SAlberto Garcia block_acct_failed(blk_get_stats(ctx->blk), &ctx->acct); 1384797ac58cSKevin Wolf goto out; 1385797ac58cSKevin Wolf } 1386797ac58cSKevin Wolf 1387797ac58cSKevin Wolf if (ctx->Pflag) { 1388797ac58cSKevin Wolf void *cmp_buf = g_malloc(ctx->qiov.size); 1389797ac58cSKevin Wolf 1390797ac58cSKevin Wolf memset(cmp_buf, ctx->pattern, ctx->qiov.size); 1391797ac58cSKevin Wolf if (memcmp(ctx->buf, cmp_buf, ctx->qiov.size)) { 1392797ac58cSKevin Wolf printf("Pattern verification failed at offset %" 1393cf67b692SStefan Weil PRId64 ", %zu bytes\n", ctx->offset, ctx->qiov.size); 1394797ac58cSKevin Wolf } 1395797ac58cSKevin Wolf g_free(cmp_buf); 1396797ac58cSKevin Wolf } 1397797ac58cSKevin Wolf 13984c7b7e9bSMax Reitz block_acct_done(blk_get_stats(ctx->blk), &ctx->acct); 1399a91f9584SFam Zheng 1400797ac58cSKevin Wolf if (ctx->qflag) { 1401797ac58cSKevin Wolf goto out; 1402797ac58cSKevin Wolf } 1403797ac58cSKevin Wolf 1404797ac58cSKevin Wolf if (ctx->vflag) { 1405797ac58cSKevin Wolf dump_buffer(ctx->buf, ctx->offset, ctx->qiov.size); 1406797ac58cSKevin Wolf } 1407797ac58cSKevin Wolf 1408797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1409797ac58cSKevin Wolf t2 = tsub(t2, ctx->t1); 1410797ac58cSKevin Wolf print_report("read", &t2, ctx->offset, ctx->qiov.size, 1411797ac58cSKevin Wolf ctx->qiov.size, 1, ctx->Cflag); 1412797ac58cSKevin Wolf out: 1413797ac58cSKevin Wolf qemu_io_free(ctx->buf); 1414797ac58cSKevin Wolf qemu_iovec_destroy(&ctx->qiov); 1415797ac58cSKevin Wolf g_free(ctx); 1416797ac58cSKevin Wolf } 1417797ac58cSKevin Wolf 1418797ac58cSKevin Wolf static void aio_read_help(void) 1419797ac58cSKevin Wolf { 1420797ac58cSKevin Wolf printf( 1421797ac58cSKevin Wolf "\n" 1422797ac58cSKevin Wolf " asynchronously reads a range of bytes from the given offset\n" 1423797ac58cSKevin Wolf "\n" 1424797ac58cSKevin Wolf " Example:\n" 1425797ac58cSKevin Wolf " 'aio_read -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n" 1426797ac58cSKevin Wolf "\n" 1427797ac58cSKevin Wolf " Reads a segment of the currently open file, optionally dumping it to the\n" 1428797ac58cSKevin Wolf " standard output stream (with -v option) for subsequent inspection.\n" 1429797ac58cSKevin Wolf " The read is performed asynchronously and the aio_flush command must be\n" 1430797ac58cSKevin Wolf " used to ensure all outstanding aio requests have been completed.\n" 1431b32d7a39SMax Reitz " Note that due to its asynchronous nature, this command will be\n" 1432b32d7a39SMax Reitz " considered successful once the request is submitted, independently\n" 1433b32d7a39SMax Reitz " of potential I/O errors or pattern mismatches.\n" 1434797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1435797ac58cSKevin Wolf " -P, -- use a pattern to verify read data\n" 143637546ff2SEric Blake " -i, -- treat request as invalid, for exercising stats\n" 1437797ac58cSKevin Wolf " -v, -- dump buffer to standard output\n" 1438797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1439797ac58cSKevin Wolf "\n"); 1440797ac58cSKevin Wolf } 1441797ac58cSKevin Wolf 1442b32d7a39SMax Reitz static int aio_read_f(BlockBackend *blk, int argc, char **argv); 1443797ac58cSKevin Wolf 1444797ac58cSKevin Wolf static const cmdinfo_t aio_read_cmd = { 1445797ac58cSKevin Wolf .name = "aio_read", 1446797ac58cSKevin Wolf .cfunc = aio_read_f, 1447797ac58cSKevin Wolf .argmin = 2, 1448797ac58cSKevin Wolf .argmax = -1, 144937546ff2SEric Blake .args = "[-Ciqv] [-P pattern] off len [len..]", 1450797ac58cSKevin Wolf .oneline = "asynchronously reads a number of bytes", 1451797ac58cSKevin Wolf .help = aio_read_help, 1452797ac58cSKevin Wolf }; 1453797ac58cSKevin Wolf 1454b32d7a39SMax Reitz static int aio_read_f(BlockBackend *blk, int argc, char **argv) 1455797ac58cSKevin Wolf { 1456797ac58cSKevin Wolf int nr_iov, c; 1457797ac58cSKevin Wolf struct aio_ctx *ctx = g_new0(struct aio_ctx, 1); 1458797ac58cSKevin Wolf 14594c7b7e9bSMax Reitz ctx->blk = blk; 146037546ff2SEric Blake while ((c = getopt(argc, argv, "CP:iqv")) != -1) { 1461797ac58cSKevin Wolf switch (c) { 1462797ac58cSKevin Wolf case 'C': 1463dc38852aSEric Blake ctx->Cflag = true; 1464797ac58cSKevin Wolf break; 1465797ac58cSKevin Wolf case 'P': 1466dc38852aSEric Blake ctx->Pflag = true; 1467797ac58cSKevin Wolf ctx->pattern = parse_pattern(optarg); 1468797ac58cSKevin Wolf if (ctx->pattern < 0) { 1469797ac58cSKevin Wolf g_free(ctx); 1470b32d7a39SMax Reitz return -EINVAL; 1471797ac58cSKevin Wolf } 1472797ac58cSKevin Wolf break; 147337546ff2SEric Blake case 'i': 147437546ff2SEric Blake printf("injecting invalid read request\n"); 147537546ff2SEric Blake block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_READ); 147637546ff2SEric Blake g_free(ctx); 1477b32d7a39SMax Reitz return 0; 1478797ac58cSKevin Wolf case 'q': 1479dc38852aSEric Blake ctx->qflag = true; 1480797ac58cSKevin Wolf break; 1481797ac58cSKevin Wolf case 'v': 1482dc38852aSEric Blake ctx->vflag = true; 1483797ac58cSKevin Wolf break; 1484797ac58cSKevin Wolf default: 1485797ac58cSKevin Wolf g_free(ctx); 1486b444d0e9SMax Reitz qemuio_command_usage(&aio_read_cmd); 1487b32d7a39SMax Reitz return -EINVAL; 1488797ac58cSKevin Wolf } 1489797ac58cSKevin Wolf } 1490797ac58cSKevin Wolf 1491797ac58cSKevin Wolf if (optind > argc - 2) { 1492797ac58cSKevin Wolf g_free(ctx); 1493b444d0e9SMax Reitz qemuio_command_usage(&aio_read_cmd); 1494b32d7a39SMax Reitz return -EINVAL; 1495797ac58cSKevin Wolf } 1496797ac58cSKevin Wolf 1497797ac58cSKevin Wolf ctx->offset = cvtnum(argv[optind]); 1498797ac58cSKevin Wolf if (ctx->offset < 0) { 1499b32d7a39SMax Reitz int ret = ctx->offset; 1500b32d7a39SMax Reitz print_cvtnum_err(ret, argv[optind]); 1501797ac58cSKevin Wolf g_free(ctx); 1502b32d7a39SMax Reitz return ret; 1503797ac58cSKevin Wolf } 1504797ac58cSKevin Wolf optind++; 1505797ac58cSKevin Wolf 1506797ac58cSKevin Wolf nr_iov = argc - optind; 15074c7b7e9bSMax Reitz ctx->buf = create_iovec(blk, &ctx->qiov, &argv[optind], nr_iov, 0xab); 1508797ac58cSKevin Wolf if (ctx->buf == NULL) { 1509556c2b60SAlberto Garcia block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_READ); 1510797ac58cSKevin Wolf g_free(ctx); 1511b32d7a39SMax Reitz return -EINVAL; 1512797ac58cSKevin Wolf } 1513797ac58cSKevin Wolf 151450290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &ctx->t1); 15154c7b7e9bSMax Reitz block_acct_start(blk_get_stats(blk), &ctx->acct, ctx->qiov.size, 15164c7b7e9bSMax Reitz BLOCK_ACCT_READ); 15177b3f9712SEric Blake blk_aio_preadv(blk, ctx->offset, &ctx->qiov, 0, aio_read_done, ctx); 1518b32d7a39SMax Reitz return 0; 1519797ac58cSKevin Wolf } 1520797ac58cSKevin Wolf 1521797ac58cSKevin Wolf static void aio_write_help(void) 1522797ac58cSKevin Wolf { 1523797ac58cSKevin Wolf printf( 1524797ac58cSKevin Wolf "\n" 1525797ac58cSKevin Wolf " asynchronously writes a range of bytes from the given offset source\n" 1526797ac58cSKevin Wolf " from multiple buffers\n" 1527797ac58cSKevin Wolf "\n" 1528797ac58cSKevin Wolf " Example:\n" 1529797ac58cSKevin Wolf " 'aio_write 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n" 1530797ac58cSKevin Wolf "\n" 1531797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 1532797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd).\n" 1533797ac58cSKevin Wolf " The write is performed asynchronously and the aio_flush command must be\n" 1534797ac58cSKevin Wolf " used to ensure all outstanding aio requests have been completed.\n" 1535b32d7a39SMax Reitz " Note that due to its asynchronous nature, this command will be\n" 1536b32d7a39SMax Reitz " considered successful once the request is submitted, independently\n" 1537b32d7a39SMax Reitz " of potential I/O errors or pattern mismatches.\n" 1538797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 1539797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1540770e0e0eSEric Blake " -f, -- use Force Unit Access semantics\n" 154137546ff2SEric Blake " -i, -- treat request as invalid, for exercising stats\n" 1542797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1543c2e001ccSEric Blake " -u, -- with -z, allow unmapping\n" 1544d004bd52SEric Blake " -z, -- write zeroes using blk_aio_pwrite_zeroes\n" 1545797ac58cSKevin Wolf "\n"); 1546797ac58cSKevin Wolf } 1547797ac58cSKevin Wolf 1548b32d7a39SMax Reitz static int aio_write_f(BlockBackend *blk, int argc, char **argv); 1549797ac58cSKevin Wolf 1550797ac58cSKevin Wolf static const cmdinfo_t aio_write_cmd = { 1551797ac58cSKevin Wolf .name = "aio_write", 1552797ac58cSKevin Wolf .cfunc = aio_write_f, 1553887354bdSKevin Wolf .perm = BLK_PERM_WRITE, 1554797ac58cSKevin Wolf .argmin = 2, 1555797ac58cSKevin Wolf .argmax = -1, 155637546ff2SEric Blake .args = "[-Cfiquz] [-P pattern] off len [len..]", 1557797ac58cSKevin Wolf .oneline = "asynchronously writes a number of bytes", 1558797ac58cSKevin Wolf .help = aio_write_help, 1559797ac58cSKevin Wolf }; 1560797ac58cSKevin Wolf 1561b32d7a39SMax Reitz static int aio_write_f(BlockBackend *blk, int argc, char **argv) 1562797ac58cSKevin Wolf { 1563797ac58cSKevin Wolf int nr_iov, c; 1564797ac58cSKevin Wolf int pattern = 0xcd; 1565797ac58cSKevin Wolf struct aio_ctx *ctx = g_new0(struct aio_ctx, 1); 1566770e0e0eSEric Blake int flags = 0; 1567797ac58cSKevin Wolf 15684c7b7e9bSMax Reitz ctx->blk = blk; 156937546ff2SEric Blake while ((c = getopt(argc, argv, "CfiqP:uz")) != -1) { 1570797ac58cSKevin Wolf switch (c) { 1571797ac58cSKevin Wolf case 'C': 1572dc38852aSEric Blake ctx->Cflag = true; 1573797ac58cSKevin Wolf break; 1574770e0e0eSEric Blake case 'f': 1575770e0e0eSEric Blake flags |= BDRV_REQ_FUA; 1576770e0e0eSEric Blake break; 1577797ac58cSKevin Wolf case 'q': 1578dc38852aSEric Blake ctx->qflag = true; 1579797ac58cSKevin Wolf break; 1580c2e001ccSEric Blake case 'u': 1581c2e001ccSEric Blake flags |= BDRV_REQ_MAY_UNMAP; 1582c2e001ccSEric Blake break; 1583797ac58cSKevin Wolf case 'P': 1584797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1585797ac58cSKevin Wolf if (pattern < 0) { 1586797ac58cSKevin Wolf g_free(ctx); 1587b32d7a39SMax Reitz return -EINVAL; 1588797ac58cSKevin Wolf } 1589797ac58cSKevin Wolf break; 159037546ff2SEric Blake case 'i': 159137546ff2SEric Blake printf("injecting invalid write request\n"); 159237546ff2SEric Blake block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_WRITE); 159337546ff2SEric Blake g_free(ctx); 1594b32d7a39SMax Reitz return 0; 15955ceb7765SKevin Wolf case 'z': 1596dc38852aSEric Blake ctx->zflag = true; 15975ceb7765SKevin Wolf break; 1598797ac58cSKevin Wolf default: 1599797ac58cSKevin Wolf g_free(ctx); 1600b444d0e9SMax Reitz qemuio_command_usage(&aio_write_cmd); 1601b32d7a39SMax Reitz return -EINVAL; 1602797ac58cSKevin Wolf } 1603797ac58cSKevin Wolf } 1604797ac58cSKevin Wolf 1605797ac58cSKevin Wolf if (optind > argc - 2) { 1606797ac58cSKevin Wolf g_free(ctx); 1607b444d0e9SMax Reitz qemuio_command_usage(&aio_write_cmd); 1608b32d7a39SMax Reitz return -EINVAL; 1609797ac58cSKevin Wolf } 1610797ac58cSKevin Wolf 16115ceb7765SKevin Wolf if (ctx->zflag && optind != argc - 2) { 16125ceb7765SKevin Wolf printf("-z supports only a single length parameter\n"); 16135ceb7765SKevin Wolf g_free(ctx); 1614b32d7a39SMax Reitz return -EINVAL; 16155ceb7765SKevin Wolf } 16165ceb7765SKevin Wolf 1617c2e001ccSEric Blake if ((flags & BDRV_REQ_MAY_UNMAP) && !ctx->zflag) { 1618c2e001ccSEric Blake printf("-u requires -z to be specified\n"); 16194ca1d340SEric Blake g_free(ctx); 1620b32d7a39SMax Reitz return -EINVAL; 1621c2e001ccSEric Blake } 1622c2e001ccSEric Blake 16235ceb7765SKevin Wolf if (ctx->zflag && ctx->Pflag) { 16245ceb7765SKevin Wolf printf("-z and -P cannot be specified at the same time\n"); 16255ceb7765SKevin Wolf g_free(ctx); 1626b32d7a39SMax Reitz return -EINVAL; 16275ceb7765SKevin Wolf } 16285ceb7765SKevin Wolf 1629797ac58cSKevin Wolf ctx->offset = cvtnum(argv[optind]); 1630797ac58cSKevin Wolf if (ctx->offset < 0) { 1631b32d7a39SMax Reitz int ret = ctx->offset; 1632b32d7a39SMax Reitz print_cvtnum_err(ret, argv[optind]); 1633797ac58cSKevin Wolf g_free(ctx); 1634b32d7a39SMax Reitz return ret; 1635797ac58cSKevin Wolf } 1636797ac58cSKevin Wolf optind++; 1637797ac58cSKevin Wolf 16385ceb7765SKevin Wolf if (ctx->zflag) { 16395ceb7765SKevin Wolf int64_t count = cvtnum(argv[optind]); 16405ceb7765SKevin Wolf if (count < 0) { 16415ceb7765SKevin Wolf print_cvtnum_err(count, argv[optind]); 16420e01b76eSKevin Wolf g_free(ctx); 1643b32d7a39SMax Reitz return count; 16445ceb7765SKevin Wolf } 16455ceb7765SKevin Wolf 16465ceb7765SKevin Wolf ctx->qiov.size = count; 1647d004bd52SEric Blake blk_aio_pwrite_zeroes(blk, ctx->offset, count, flags, aio_write_done, 1648770e0e0eSEric Blake ctx); 16495ceb7765SKevin Wolf } else { 1650797ac58cSKevin Wolf nr_iov = argc - optind; 16515ceb7765SKevin Wolf ctx->buf = create_iovec(blk, &ctx->qiov, &argv[optind], nr_iov, 16525ceb7765SKevin Wolf pattern); 1653797ac58cSKevin Wolf if (ctx->buf == NULL) { 1654556c2b60SAlberto Garcia block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_WRITE); 1655797ac58cSKevin Wolf g_free(ctx); 1656b32d7a39SMax Reitz return -EINVAL; 1657797ac58cSKevin Wolf } 1658797ac58cSKevin Wolf 165950290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &ctx->t1); 16604c7b7e9bSMax Reitz block_acct_start(blk_get_stats(blk), &ctx->acct, ctx->qiov.size, 16614c7b7e9bSMax Reitz BLOCK_ACCT_WRITE); 16625ceb7765SKevin Wolf 1663770e0e0eSEric Blake blk_aio_pwritev(blk, ctx->offset, &ctx->qiov, flags, aio_write_done, 1664770e0e0eSEric Blake ctx); 16655ceb7765SKevin Wolf } 1666b32d7a39SMax Reitz 1667b32d7a39SMax Reitz return 0; 1668797ac58cSKevin Wolf } 1669797ac58cSKevin Wolf 1670b32d7a39SMax Reitz static int aio_flush_f(BlockBackend *blk, int argc, char **argv) 1671797ac58cSKevin Wolf { 1672556c2b60SAlberto Garcia BlockAcctCookie cookie; 1673556c2b60SAlberto Garcia block_acct_start(blk_get_stats(blk), &cookie, 0, BLOCK_ACCT_FLUSH); 16744c7b7e9bSMax Reitz blk_drain_all(); 1675556c2b60SAlberto Garcia block_acct_done(blk_get_stats(blk), &cookie); 1676b32d7a39SMax Reitz return 0; 1677797ac58cSKevin Wolf } 1678797ac58cSKevin Wolf 1679797ac58cSKevin Wolf static const cmdinfo_t aio_flush_cmd = { 1680797ac58cSKevin Wolf .name = "aio_flush", 1681797ac58cSKevin Wolf .cfunc = aio_flush_f, 1682797ac58cSKevin Wolf .oneline = "completes all outstanding aio requests" 1683797ac58cSKevin Wolf }; 1684797ac58cSKevin Wolf 1685b32d7a39SMax Reitz static int flush_f(BlockBackend *blk, int argc, char **argv) 1686797ac58cSKevin Wolf { 1687b32d7a39SMax Reitz return blk_flush(blk); 1688797ac58cSKevin Wolf } 1689797ac58cSKevin Wolf 1690797ac58cSKevin Wolf static const cmdinfo_t flush_cmd = { 1691797ac58cSKevin Wolf .name = "flush", 1692797ac58cSKevin Wolf .altname = "f", 1693797ac58cSKevin Wolf .cfunc = flush_f, 1694797ac58cSKevin Wolf .oneline = "flush all in-core file state to disk", 1695797ac58cSKevin Wolf }; 1696797ac58cSKevin Wolf 1697b32d7a39SMax Reitz static int truncate_f(BlockBackend *blk, int argc, char **argv) 1698797ac58cSKevin Wolf { 1699ed3d2ec9SMax Reitz Error *local_err = NULL; 1700797ac58cSKevin Wolf int64_t offset; 1701797ac58cSKevin Wolf int ret; 1702797ac58cSKevin Wolf 1703797ac58cSKevin Wolf offset = cvtnum(argv[1]); 1704797ac58cSKevin Wolf if (offset < 0) { 1705a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[1]); 1706b32d7a39SMax Reitz return offset; 1707797ac58cSKevin Wolf } 1708797ac58cSKevin Wolf 17093a691c50SMax Reitz ret = blk_truncate(blk, offset, PREALLOC_MODE_OFF, &local_err); 1710797ac58cSKevin Wolf if (ret < 0) { 1711ed3d2ec9SMax Reitz error_report_err(local_err); 1712b32d7a39SMax Reitz return ret; 1713797ac58cSKevin Wolf } 1714b32d7a39SMax Reitz 1715b32d7a39SMax Reitz return 0; 1716797ac58cSKevin Wolf } 1717797ac58cSKevin Wolf 1718797ac58cSKevin Wolf static const cmdinfo_t truncate_cmd = { 1719797ac58cSKevin Wolf .name = "truncate", 1720797ac58cSKevin Wolf .altname = "t", 1721797ac58cSKevin Wolf .cfunc = truncate_f, 1722887354bdSKevin Wolf .perm = BLK_PERM_WRITE | BLK_PERM_RESIZE, 1723797ac58cSKevin Wolf .argmin = 1, 1724797ac58cSKevin Wolf .argmax = 1, 1725797ac58cSKevin Wolf .args = "off", 1726797ac58cSKevin Wolf .oneline = "truncates the current file at the given offset", 1727797ac58cSKevin Wolf }; 1728797ac58cSKevin Wolf 1729b32d7a39SMax Reitz static int length_f(BlockBackend *blk, int argc, char **argv) 1730797ac58cSKevin Wolf { 1731797ac58cSKevin Wolf int64_t size; 1732797ac58cSKevin Wolf char s1[64]; 1733797ac58cSKevin Wolf 17344c7b7e9bSMax Reitz size = blk_getlength(blk); 1735797ac58cSKevin Wolf if (size < 0) { 1736797ac58cSKevin Wolf printf("getlength: %s\n", strerror(-size)); 1737b32d7a39SMax Reitz return size; 1738797ac58cSKevin Wolf } 1739797ac58cSKevin Wolf 1740797ac58cSKevin Wolf cvtstr(size, s1, sizeof(s1)); 1741797ac58cSKevin Wolf printf("%s\n", s1); 1742b32d7a39SMax Reitz return 0; 1743797ac58cSKevin Wolf } 1744797ac58cSKevin Wolf 1745797ac58cSKevin Wolf 1746797ac58cSKevin Wolf static const cmdinfo_t length_cmd = { 1747797ac58cSKevin Wolf .name = "length", 1748797ac58cSKevin Wolf .altname = "l", 1749797ac58cSKevin Wolf .cfunc = length_f, 1750797ac58cSKevin Wolf .oneline = "gets the length of the current file", 1751797ac58cSKevin Wolf }; 1752797ac58cSKevin Wolf 1753797ac58cSKevin Wolf 1754b32d7a39SMax Reitz static int info_f(BlockBackend *blk, int argc, char **argv) 1755797ac58cSKevin Wolf { 17564c7b7e9bSMax Reitz BlockDriverState *bs = blk_bs(blk); 1757797ac58cSKevin Wolf BlockDriverInfo bdi; 1758a8d8ecb7SMax Reitz ImageInfoSpecific *spec_info; 17591bf6e9caSAndrey Shinkevich Error *local_err = NULL; 1760797ac58cSKevin Wolf char s1[64], s2[64]; 1761797ac58cSKevin Wolf int ret; 1762797ac58cSKevin Wolf 1763797ac58cSKevin Wolf if (bs->drv && bs->drv->format_name) { 1764797ac58cSKevin Wolf printf("format name: %s\n", bs->drv->format_name); 1765797ac58cSKevin Wolf } 1766797ac58cSKevin Wolf if (bs->drv && bs->drv->protocol_name) { 1767797ac58cSKevin Wolf printf("format name: %s\n", bs->drv->protocol_name); 1768797ac58cSKevin Wolf } 1769797ac58cSKevin Wolf 1770797ac58cSKevin Wolf ret = bdrv_get_info(bs, &bdi); 1771797ac58cSKevin Wolf if (ret) { 1772b32d7a39SMax Reitz return ret; 1773797ac58cSKevin Wolf } 1774797ac58cSKevin Wolf 1775797ac58cSKevin Wolf cvtstr(bdi.cluster_size, s1, sizeof(s1)); 1776797ac58cSKevin Wolf cvtstr(bdi.vm_state_offset, s2, sizeof(s2)); 1777797ac58cSKevin Wolf 1778797ac58cSKevin Wolf printf("cluster size: %s\n", s1); 1779797ac58cSKevin Wolf printf("vm state offset: %s\n", s2); 1780797ac58cSKevin Wolf 17811bf6e9caSAndrey Shinkevich spec_info = bdrv_get_specific_info(bs, &local_err); 17821bf6e9caSAndrey Shinkevich if (local_err) { 17831bf6e9caSAndrey Shinkevich error_report_err(local_err); 17841bf6e9caSAndrey Shinkevich return -EIO; 17851bf6e9caSAndrey Shinkevich } 1786a8d8ecb7SMax Reitz if (spec_info) { 1787a8d8ecb7SMax Reitz printf("Format specific information:\n"); 1788e1ce7d74SMarkus Armbruster bdrv_image_info_specific_dump(spec_info); 1789a8d8ecb7SMax Reitz qapi_free_ImageInfoSpecific(spec_info); 1790a8d8ecb7SMax Reitz } 1791b32d7a39SMax Reitz 1792b32d7a39SMax Reitz return 0; 1793797ac58cSKevin Wolf } 1794797ac58cSKevin Wolf 1795797ac58cSKevin Wolf 1796797ac58cSKevin Wolf 1797797ac58cSKevin Wolf static const cmdinfo_t info_cmd = { 1798797ac58cSKevin Wolf .name = "info", 1799797ac58cSKevin Wolf .altname = "i", 1800797ac58cSKevin Wolf .cfunc = info_f, 1801797ac58cSKevin Wolf .oneline = "prints information about the current file", 1802797ac58cSKevin Wolf }; 1803797ac58cSKevin Wolf 1804797ac58cSKevin Wolf static void discard_help(void) 1805797ac58cSKevin Wolf { 1806797ac58cSKevin Wolf printf( 1807797ac58cSKevin Wolf "\n" 1808797ac58cSKevin Wolf " discards a range of bytes from the given offset\n" 1809797ac58cSKevin Wolf "\n" 1810797ac58cSKevin Wolf " Example:\n" 1811797ac58cSKevin Wolf " 'discard 512 1k' - discards 1 kilobyte from 512 bytes into the file\n" 1812797ac58cSKevin Wolf "\n" 1813797ac58cSKevin Wolf " Discards a segment of the currently open file.\n" 1814797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1815797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1816797ac58cSKevin Wolf "\n"); 1817797ac58cSKevin Wolf } 1818797ac58cSKevin Wolf 1819b32d7a39SMax Reitz static int discard_f(BlockBackend *blk, int argc, char **argv); 1820797ac58cSKevin Wolf 1821797ac58cSKevin Wolf static const cmdinfo_t discard_cmd = { 1822797ac58cSKevin Wolf .name = "discard", 1823797ac58cSKevin Wolf .altname = "d", 1824797ac58cSKevin Wolf .cfunc = discard_f, 1825887354bdSKevin Wolf .perm = BLK_PERM_WRITE, 1826797ac58cSKevin Wolf .argmin = 2, 1827797ac58cSKevin Wolf .argmax = -1, 1828797ac58cSKevin Wolf .args = "[-Cq] off len", 1829797ac58cSKevin Wolf .oneline = "discards a number of bytes at a specified offset", 1830797ac58cSKevin Wolf .help = discard_help, 1831797ac58cSKevin Wolf }; 1832797ac58cSKevin Wolf 1833b32d7a39SMax Reitz static int discard_f(BlockBackend *blk, int argc, char **argv) 1834797ac58cSKevin Wolf { 183550290c00SAlex Bennée struct timespec t1, t2; 1836dc38852aSEric Blake bool Cflag = false, qflag = false; 1837797ac58cSKevin Wolf int c, ret; 1838f5a5ca79SManos Pitsidianakis int64_t offset, bytes; 1839797ac58cSKevin Wolf 1840b062ad86SEric Blake while ((c = getopt(argc, argv, "Cq")) != -1) { 1841797ac58cSKevin Wolf switch (c) { 1842797ac58cSKevin Wolf case 'C': 1843dc38852aSEric Blake Cflag = true; 1844797ac58cSKevin Wolf break; 1845797ac58cSKevin Wolf case 'q': 1846dc38852aSEric Blake qflag = true; 1847797ac58cSKevin Wolf break; 1848797ac58cSKevin Wolf default: 1849b444d0e9SMax Reitz qemuio_command_usage(&discard_cmd); 1850b32d7a39SMax Reitz return -EINVAL; 1851797ac58cSKevin Wolf } 1852797ac58cSKevin Wolf } 1853797ac58cSKevin Wolf 1854797ac58cSKevin Wolf if (optind != argc - 2) { 1855b444d0e9SMax Reitz qemuio_command_usage(&discard_cmd); 1856b32d7a39SMax Reitz return -EINVAL; 1857797ac58cSKevin Wolf } 1858797ac58cSKevin Wolf 1859797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1860797ac58cSKevin Wolf if (offset < 0) { 1861a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 1862b32d7a39SMax Reitz return offset; 1863797ac58cSKevin Wolf } 1864797ac58cSKevin Wolf 1865797ac58cSKevin Wolf optind++; 1866f5a5ca79SManos Pitsidianakis bytes = cvtnum(argv[optind]); 1867f5a5ca79SManos Pitsidianakis if (bytes < 0) { 1868f5a5ca79SManos Pitsidianakis print_cvtnum_err(bytes, argv[optind]); 1869b32d7a39SMax Reitz return bytes; 187041ae31e3SAlberto Garcia } else if (bytes > BDRV_REQUEST_MAX_BYTES) { 18719b0beaf3SJohn Snow printf("length cannot exceed %"PRIu64", given %s\n", 187241ae31e3SAlberto Garcia (uint64_t)BDRV_REQUEST_MAX_BYTES, argv[optind]); 1873b32d7a39SMax Reitz return -EINVAL; 1874797ac58cSKevin Wolf } 1875797ac58cSKevin Wolf 187650290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t1); 1877f5a5ca79SManos Pitsidianakis ret = blk_pdiscard(blk, offset, bytes); 187850290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 1879797ac58cSKevin Wolf 1880797ac58cSKevin Wolf if (ret < 0) { 1881797ac58cSKevin Wolf printf("discard failed: %s\n", strerror(-ret)); 1882b32d7a39SMax Reitz return ret; 1883797ac58cSKevin Wolf } 1884797ac58cSKevin Wolf 1885797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1886797ac58cSKevin Wolf if (!qflag) { 1887797ac58cSKevin Wolf t2 = tsub(t2, t1); 1888f5a5ca79SManos Pitsidianakis print_report("discard", &t2, offset, bytes, bytes, 1, Cflag); 1889797ac58cSKevin Wolf } 1890b32d7a39SMax Reitz 1891b32d7a39SMax Reitz return 0; 1892797ac58cSKevin Wolf } 1893797ac58cSKevin Wolf 1894b32d7a39SMax Reitz static int alloc_f(BlockBackend *blk, int argc, char **argv) 1895797ac58cSKevin Wolf { 18964c7b7e9bSMax Reitz BlockDriverState *bs = blk_bs(blk); 1897d6a644bbSEric Blake int64_t offset, start, remaining, count; 1898797ac58cSKevin Wolf char s1[64]; 1899d6a644bbSEric Blake int ret; 1900d6a644bbSEric Blake int64_t num, sum_alloc; 1901797ac58cSKevin Wolf 1902d6a644bbSEric Blake start = offset = cvtnum(argv[1]); 1903797ac58cSKevin Wolf if (offset < 0) { 1904a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[1]); 1905b32d7a39SMax Reitz return offset; 1906797ac58cSKevin Wolf } 1907797ac58cSKevin Wolf 1908797ac58cSKevin Wolf if (argc == 3) { 19094401fdc7SEric Blake count = cvtnum(argv[2]); 19104401fdc7SEric Blake if (count < 0) { 19114401fdc7SEric Blake print_cvtnum_err(count, argv[2]); 1912b32d7a39SMax Reitz return count; 1913797ac58cSKevin Wolf } 1914797ac58cSKevin Wolf } else { 19154401fdc7SEric Blake count = BDRV_SECTOR_SIZE; 1916797ac58cSKevin Wolf } 1917797ac58cSKevin Wolf 1918d6a644bbSEric Blake remaining = count; 1919797ac58cSKevin Wolf sum_alloc = 0; 1920797ac58cSKevin Wolf while (remaining) { 1921d6a644bbSEric Blake ret = bdrv_is_allocated(bs, offset, remaining, &num); 1922d663640cSPaolo Bonzini if (ret < 0) { 1923d663640cSPaolo Bonzini printf("is_allocated failed: %s\n", strerror(-ret)); 1924b32d7a39SMax Reitz return ret; 1925d663640cSPaolo Bonzini } 1926d6a644bbSEric Blake offset += num; 1927797ac58cSKevin Wolf remaining -= num; 1928797ac58cSKevin Wolf if (ret) { 1929797ac58cSKevin Wolf sum_alloc += num; 1930797ac58cSKevin Wolf } 1931797ac58cSKevin Wolf if (num == 0) { 1932d6a644bbSEric Blake count -= remaining; 1933797ac58cSKevin Wolf remaining = 0; 1934797ac58cSKevin Wolf } 1935797ac58cSKevin Wolf } 1936797ac58cSKevin Wolf 1937d6a644bbSEric Blake cvtstr(start, s1, sizeof(s1)); 1938797ac58cSKevin Wolf 19394401fdc7SEric Blake printf("%"PRId64"/%"PRId64" bytes allocated at offset %s\n", 1940d6a644bbSEric Blake sum_alloc, count, s1); 1941b32d7a39SMax Reitz return 0; 1942797ac58cSKevin Wolf } 1943797ac58cSKevin Wolf 1944797ac58cSKevin Wolf static const cmdinfo_t alloc_cmd = { 1945797ac58cSKevin Wolf .name = "alloc", 1946797ac58cSKevin Wolf .altname = "a", 1947797ac58cSKevin Wolf .argmin = 1, 1948797ac58cSKevin Wolf .argmax = 2, 1949797ac58cSKevin Wolf .cfunc = alloc_f, 19504401fdc7SEric Blake .args = "offset [count]", 19514401fdc7SEric Blake .oneline = "checks if offset is allocated in the file", 1952797ac58cSKevin Wolf }; 1953797ac58cSKevin Wolf 1954797ac58cSKevin Wolf 1955d6a644bbSEric Blake static int map_is_allocated(BlockDriverState *bs, int64_t offset, 1956d6a644bbSEric Blake int64_t bytes, int64_t *pnum) 1957797ac58cSKevin Wolf { 1958d6a644bbSEric Blake int64_t num; 1959d6a644bbSEric Blake int num_checked; 1960797ac58cSKevin Wolf int ret, firstret; 1961797ac58cSKevin Wolf 1962d6a644bbSEric Blake num_checked = MIN(bytes, BDRV_REQUEST_MAX_BYTES); 1963d6a644bbSEric Blake ret = bdrv_is_allocated(bs, offset, num_checked, &num); 1964797ac58cSKevin Wolf if (ret < 0) { 1965797ac58cSKevin Wolf return ret; 1966797ac58cSKevin Wolf } 1967797ac58cSKevin Wolf 1968797ac58cSKevin Wolf firstret = ret; 1969797ac58cSKevin Wolf *pnum = num; 1970797ac58cSKevin Wolf 1971d6a644bbSEric Blake while (bytes > 0 && ret == firstret) { 1972d6a644bbSEric Blake offset += num; 1973d6a644bbSEric Blake bytes -= num; 1974797ac58cSKevin Wolf 1975d6a644bbSEric Blake num_checked = MIN(bytes, BDRV_REQUEST_MAX_BYTES); 1976d6a644bbSEric Blake ret = bdrv_is_allocated(bs, offset, num_checked, &num); 19774b25bbc4SMax Reitz if (ret == firstret && num) { 1978797ac58cSKevin Wolf *pnum += num; 1979797ac58cSKevin Wolf } else { 1980797ac58cSKevin Wolf break; 1981797ac58cSKevin Wolf } 1982797ac58cSKevin Wolf } 1983797ac58cSKevin Wolf 1984797ac58cSKevin Wolf return firstret; 1985797ac58cSKevin Wolf } 1986797ac58cSKevin Wolf 1987b32d7a39SMax Reitz static int map_f(BlockBackend *blk, int argc, char **argv) 1988797ac58cSKevin Wolf { 1989d6a644bbSEric Blake int64_t offset, bytes; 19906f3c90afSEric Blake char s1[64], s2[64]; 1991797ac58cSKevin Wolf int64_t num; 1992797ac58cSKevin Wolf int ret; 1993797ac58cSKevin Wolf const char *retstr; 1994797ac58cSKevin Wolf 1995797ac58cSKevin Wolf offset = 0; 1996d6a644bbSEric Blake bytes = blk_getlength(blk); 1997d6a644bbSEric Blake if (bytes < 0) { 1998d6a644bbSEric Blake error_report("Failed to query image length: %s", strerror(-bytes)); 1999b32d7a39SMax Reitz return bytes; 20004c7b7e9bSMax Reitz } 20014c7b7e9bSMax Reitz 2002d6a644bbSEric Blake while (bytes) { 2003d6a644bbSEric Blake ret = map_is_allocated(blk_bs(blk), offset, bytes, &num); 2004797ac58cSKevin Wolf if (ret < 0) { 2005797ac58cSKevin Wolf error_report("Failed to get allocation status: %s", strerror(-ret)); 2006b32d7a39SMax Reitz return ret; 20074b25bbc4SMax Reitz } else if (!num) { 20084b25bbc4SMax Reitz error_report("Unexpected end of image"); 2009b32d7a39SMax Reitz return -EIO; 2010797ac58cSKevin Wolf } 2011797ac58cSKevin Wolf 2012797ac58cSKevin Wolf retstr = ret ? " allocated" : "not allocated"; 2013d6a644bbSEric Blake cvtstr(num, s1, sizeof(s1)); 2014d6a644bbSEric Blake cvtstr(offset, s2, sizeof(s2)); 20156f3c90afSEric Blake printf("%s (0x%" PRIx64 ") bytes %s at offset %s (0x%" PRIx64 ")\n", 2016d6a644bbSEric Blake s1, num, retstr, s2, offset); 2017797ac58cSKevin Wolf 2018797ac58cSKevin Wolf offset += num; 2019d6a644bbSEric Blake bytes -= num; 2020d6a644bbSEric Blake } 2021b32d7a39SMax Reitz 2022b32d7a39SMax Reitz return 0; 2023797ac58cSKevin Wolf } 2024797ac58cSKevin Wolf 2025797ac58cSKevin Wolf static const cmdinfo_t map_cmd = { 2026797ac58cSKevin Wolf .name = "map", 2027797ac58cSKevin Wolf .argmin = 0, 2028797ac58cSKevin Wolf .argmax = 0, 2029797ac58cSKevin Wolf .cfunc = map_f, 2030797ac58cSKevin Wolf .args = "", 2031797ac58cSKevin Wolf .oneline = "prints the allocated areas of a file", 2032797ac58cSKevin Wolf }; 2033797ac58cSKevin Wolf 20345bbd2e59SKevin Wolf static void reopen_help(void) 20355bbd2e59SKevin Wolf { 20365bbd2e59SKevin Wolf printf( 20375bbd2e59SKevin Wolf "\n" 20385bbd2e59SKevin Wolf " Changes the open options of an already opened image\n" 20395bbd2e59SKevin Wolf "\n" 20405bbd2e59SKevin Wolf " Example:\n" 20415bbd2e59SKevin Wolf " 'reopen -o lazy-refcounts=on' - activates lazy refcount writeback on a qcow2 image\n" 20425bbd2e59SKevin Wolf "\n" 20435bbd2e59SKevin Wolf " -r, -- Reopen the image read-only\n" 2044ea92203cSKevin Wolf " -w, -- Reopen the image read-write\n" 20455bbd2e59SKevin Wolf " -c, -- Change the cache mode to the given value\n" 20465bbd2e59SKevin Wolf " -o, -- Changes block driver options (cf. 'open' command)\n" 20475bbd2e59SKevin Wolf "\n"); 20485bbd2e59SKevin Wolf } 20495bbd2e59SKevin Wolf 2050b32d7a39SMax Reitz static int reopen_f(BlockBackend *blk, int argc, char **argv); 20515bbd2e59SKevin Wolf 20525bbd2e59SKevin Wolf static QemuOptsList reopen_opts = { 20535bbd2e59SKevin Wolf .name = "reopen", 20545bbd2e59SKevin Wolf .merge_lists = true, 20555bbd2e59SKevin Wolf .head = QTAILQ_HEAD_INITIALIZER(reopen_opts.head), 20565bbd2e59SKevin Wolf .desc = { 20575bbd2e59SKevin Wolf /* no elements => accept any params */ 20585bbd2e59SKevin Wolf { /* end of list */ } 20595bbd2e59SKevin Wolf }, 20605bbd2e59SKevin Wolf }; 20615bbd2e59SKevin Wolf 20625bbd2e59SKevin Wolf static const cmdinfo_t reopen_cmd = { 20635bbd2e59SKevin Wolf .name = "reopen", 20645bbd2e59SKevin Wolf .argmin = 0, 20655bbd2e59SKevin Wolf .argmax = -1, 20665bbd2e59SKevin Wolf .cfunc = reopen_f, 2067ea92203cSKevin Wolf .args = "[(-r|-w)] [-c cache] [-o options]", 20685bbd2e59SKevin Wolf .oneline = "reopens an image with new options", 20695bbd2e59SKevin Wolf .help = reopen_help, 20705bbd2e59SKevin Wolf }; 20715bbd2e59SKevin Wolf 2072b32d7a39SMax Reitz static int reopen_f(BlockBackend *blk, int argc, char **argv) 20735bbd2e59SKevin Wolf { 20745bbd2e59SKevin Wolf BlockDriverState *bs = blk_bs(blk); 20755bbd2e59SKevin Wolf QemuOpts *qopts; 20765bbd2e59SKevin Wolf QDict *opts; 20775bbd2e59SKevin Wolf int c; 20785bbd2e59SKevin Wolf int flags = bs->open_flags; 207919dbecdcSKevin Wolf bool writethrough = !blk_enable_write_cache(blk); 2080ea92203cSKevin Wolf bool has_rw_option = false; 2081dc900c35SAlberto Garcia bool has_cache_option = false; 20825bbd2e59SKevin Wolf 20835bbd2e59SKevin Wolf BlockReopenQueue *brq; 20845bbd2e59SKevin Wolf Error *local_err = NULL; 20855bbd2e59SKevin Wolf 2086ea92203cSKevin Wolf while ((c = getopt(argc, argv, "c:o:rw")) != -1) { 20875bbd2e59SKevin Wolf switch (c) { 20885bbd2e59SKevin Wolf case 'c': 208919dbecdcSKevin Wolf if (bdrv_parse_cache_mode(optarg, &flags, &writethrough) < 0) { 20905bbd2e59SKevin Wolf error_report("Invalid cache option: %s", optarg); 2091b32d7a39SMax Reitz return -EINVAL; 20925bbd2e59SKevin Wolf } 2093dc900c35SAlberto Garcia has_cache_option = true; 20945bbd2e59SKevin Wolf break; 20955bbd2e59SKevin Wolf case 'o': 20965bbd2e59SKevin Wolf if (!qemu_opts_parse_noisily(&reopen_opts, optarg, 0)) { 20975bbd2e59SKevin Wolf qemu_opts_reset(&reopen_opts); 2098b32d7a39SMax Reitz return -EINVAL; 20995bbd2e59SKevin Wolf } 21005bbd2e59SKevin Wolf break; 21015bbd2e59SKevin Wolf case 'r': 2102ea92203cSKevin Wolf if (has_rw_option) { 2103ea92203cSKevin Wolf error_report("Only one -r/-w option may be given"); 2104b32d7a39SMax Reitz return -EINVAL; 2105ea92203cSKevin Wolf } 21065bbd2e59SKevin Wolf flags &= ~BDRV_O_RDWR; 2107ea92203cSKevin Wolf has_rw_option = true; 2108ea92203cSKevin Wolf break; 2109ea92203cSKevin Wolf case 'w': 2110ea92203cSKevin Wolf if (has_rw_option) { 2111ea92203cSKevin Wolf error_report("Only one -r/-w option may be given"); 2112b32d7a39SMax Reitz return -EINVAL; 2113ea92203cSKevin Wolf } 2114ea92203cSKevin Wolf flags |= BDRV_O_RDWR; 2115ea92203cSKevin Wolf has_rw_option = true; 21165bbd2e59SKevin Wolf break; 21175bbd2e59SKevin Wolf default: 21185bbd2e59SKevin Wolf qemu_opts_reset(&reopen_opts); 2119b444d0e9SMax Reitz qemuio_command_usage(&reopen_cmd); 2120b32d7a39SMax Reitz return -EINVAL; 21215bbd2e59SKevin Wolf } 21225bbd2e59SKevin Wolf } 21235bbd2e59SKevin Wolf 21245bbd2e59SKevin Wolf if (optind != argc) { 21255bbd2e59SKevin Wolf qemu_opts_reset(&reopen_opts); 2126b444d0e9SMax Reitz qemuio_command_usage(&reopen_cmd); 2127b32d7a39SMax Reitz return -EINVAL; 21285bbd2e59SKevin Wolf } 21295bbd2e59SKevin Wolf 2130a8003ec4SAlberto Garcia if (!writethrough != blk_enable_write_cache(blk) && 213119dbecdcSKevin Wolf blk_get_attached_dev(blk)) 213219dbecdcSKevin Wolf { 213319dbecdcSKevin Wolf error_report("Cannot change cache.writeback: Device attached"); 213419dbecdcSKevin Wolf qemu_opts_reset(&reopen_opts); 2135b32d7a39SMax Reitz return -EBUSY; 213619dbecdcSKevin Wolf } 213719dbecdcSKevin Wolf 2138f3adefb2SKevin Wolf if (!(flags & BDRV_O_RDWR)) { 2139f3adefb2SKevin Wolf uint64_t orig_perm, orig_shared_perm; 2140f3adefb2SKevin Wolf 2141f3adefb2SKevin Wolf bdrv_drain(bs); 2142f3adefb2SKevin Wolf 2143f3adefb2SKevin Wolf blk_get_perm(blk, &orig_perm, &orig_shared_perm); 2144f3adefb2SKevin Wolf blk_set_perm(blk, 2145f3adefb2SKevin Wolf orig_perm & ~(BLK_PERM_WRITE | BLK_PERM_WRITE_UNCHANGED), 2146f3adefb2SKevin Wolf orig_shared_perm, 2147f3adefb2SKevin Wolf &error_abort); 2148f3adefb2SKevin Wolf } 2149f3adefb2SKevin Wolf 21505bbd2e59SKevin Wolf qopts = qemu_opts_find(&reopen_opts, NULL); 2151dc900c35SAlberto Garcia opts = qopts ? qemu_opts_to_qdict(qopts, NULL) : qdict_new(); 21525bbd2e59SKevin Wolf qemu_opts_reset(&reopen_opts); 21535bbd2e59SKevin Wolf 2154dc900c35SAlberto Garcia if (qdict_haskey(opts, BDRV_OPT_READ_ONLY)) { 2155dc900c35SAlberto Garcia if (has_rw_option) { 2156dc900c35SAlberto Garcia error_report("Cannot set both -r/-w and '" BDRV_OPT_READ_ONLY "'"); 2157dc900c35SAlberto Garcia qobject_unref(opts); 2158dc900c35SAlberto Garcia return -EINVAL; 2159dc900c35SAlberto Garcia } 2160dc900c35SAlberto Garcia } else { 2161dc900c35SAlberto Garcia qdict_put_bool(opts, BDRV_OPT_READ_ONLY, !(flags & BDRV_O_RDWR)); 2162dc900c35SAlberto Garcia } 2163dc900c35SAlberto Garcia 2164dc900c35SAlberto Garcia if (qdict_haskey(opts, BDRV_OPT_CACHE_DIRECT) || 2165dc900c35SAlberto Garcia qdict_haskey(opts, BDRV_OPT_CACHE_NO_FLUSH)) { 2166dc900c35SAlberto Garcia if (has_cache_option) { 2167dc900c35SAlberto Garcia error_report("Cannot set both -c and the cache options"); 2168dc900c35SAlberto Garcia qobject_unref(opts); 2169dc900c35SAlberto Garcia return -EINVAL; 2170dc900c35SAlberto Garcia } 2171dc900c35SAlberto Garcia } else { 2172dc900c35SAlberto Garcia qdict_put_bool(opts, BDRV_OPT_CACHE_DIRECT, flags & BDRV_O_NOCACHE); 2173dc900c35SAlberto Garcia qdict_put_bool(opts, BDRV_OPT_CACHE_NO_FLUSH, flags & BDRV_O_NO_FLUSH); 2174dc900c35SAlberto Garcia } 2175dc900c35SAlberto Garcia 21761a63a907SKevin Wolf bdrv_subtree_drained_begin(bs); 2177077e8e20SAlberto Garcia brq = bdrv_reopen_queue(NULL, bs, opts, true); 21785019aeceSAlberto Garcia bdrv_reopen_multiple(brq, &local_err); 21791a63a907SKevin Wolf bdrv_subtree_drained_end(bs); 21801a63a907SKevin Wolf 21815bbd2e59SKevin Wolf if (local_err) { 21825bbd2e59SKevin Wolf error_report_err(local_err); 2183b32d7a39SMax Reitz return -EINVAL; 21845bbd2e59SKevin Wolf } 21855bbd2e59SKevin Wolf 2186b32d7a39SMax Reitz blk_set_enable_write_cache(blk, !writethrough); 2187b32d7a39SMax Reitz return 0; 2188b32d7a39SMax Reitz } 2189b32d7a39SMax Reitz 2190b32d7a39SMax Reitz static int break_f(BlockBackend *blk, int argc, char **argv) 2191797ac58cSKevin Wolf { 2192797ac58cSKevin Wolf int ret; 2193797ac58cSKevin Wolf 21944c7b7e9bSMax Reitz ret = bdrv_debug_breakpoint(blk_bs(blk), argv[1], argv[2]); 2195797ac58cSKevin Wolf if (ret < 0) { 2196797ac58cSKevin Wolf printf("Could not set breakpoint: %s\n", strerror(-ret)); 2197b32d7a39SMax Reitz return ret; 2198797ac58cSKevin Wolf } 2199797ac58cSKevin Wolf 2200b32d7a39SMax Reitz return 0; 2201b32d7a39SMax Reitz } 2202b32d7a39SMax Reitz 2203b32d7a39SMax Reitz static int remove_break_f(BlockBackend *blk, int argc, char **argv) 22044cc70e93SFam Zheng { 22054cc70e93SFam Zheng int ret; 22064cc70e93SFam Zheng 22074c7b7e9bSMax Reitz ret = bdrv_debug_remove_breakpoint(blk_bs(blk), argv[1]); 22084cc70e93SFam Zheng if (ret < 0) { 22094cc70e93SFam Zheng printf("Could not remove breakpoint %s: %s\n", argv[1], strerror(-ret)); 2210b32d7a39SMax Reitz return ret; 22114cc70e93SFam Zheng } 2212b32d7a39SMax Reitz 2213b32d7a39SMax Reitz return 0; 22144cc70e93SFam Zheng } 22154cc70e93SFam Zheng 2216797ac58cSKevin Wolf static const cmdinfo_t break_cmd = { 2217797ac58cSKevin Wolf .name = "break", 2218797ac58cSKevin Wolf .argmin = 2, 2219797ac58cSKevin Wolf .argmax = 2, 2220797ac58cSKevin Wolf .cfunc = break_f, 2221797ac58cSKevin Wolf .args = "event tag", 2222797ac58cSKevin Wolf .oneline = "sets a breakpoint on event and tags the stopped " 2223797ac58cSKevin Wolf "request as tag", 2224797ac58cSKevin Wolf }; 2225797ac58cSKevin Wolf 22264cc70e93SFam Zheng static const cmdinfo_t remove_break_cmd = { 22274cc70e93SFam Zheng .name = "remove_break", 22284cc70e93SFam Zheng .argmin = 1, 22294cc70e93SFam Zheng .argmax = 1, 22304cc70e93SFam Zheng .cfunc = remove_break_f, 22314cc70e93SFam Zheng .args = "tag", 22324cc70e93SFam Zheng .oneline = "remove a breakpoint by tag", 22334cc70e93SFam Zheng }; 22344cc70e93SFam Zheng 2235b32d7a39SMax Reitz static int resume_f(BlockBackend *blk, int argc, char **argv) 2236797ac58cSKevin Wolf { 2237797ac58cSKevin Wolf int ret; 2238797ac58cSKevin Wolf 22394c7b7e9bSMax Reitz ret = bdrv_debug_resume(blk_bs(blk), argv[1]); 2240797ac58cSKevin Wolf if (ret < 0) { 2241797ac58cSKevin Wolf printf("Could not resume request: %s\n", strerror(-ret)); 2242b32d7a39SMax Reitz return ret; 2243797ac58cSKevin Wolf } 2244b32d7a39SMax Reitz 2245b32d7a39SMax Reitz return 0; 2246797ac58cSKevin Wolf } 2247797ac58cSKevin Wolf 2248797ac58cSKevin Wolf static const cmdinfo_t resume_cmd = { 2249797ac58cSKevin Wolf .name = "resume", 2250797ac58cSKevin Wolf .argmin = 1, 2251797ac58cSKevin Wolf .argmax = 1, 2252797ac58cSKevin Wolf .cfunc = resume_f, 2253797ac58cSKevin Wolf .args = "tag", 2254797ac58cSKevin Wolf .oneline = "resumes the request tagged as tag", 2255797ac58cSKevin Wolf }; 2256797ac58cSKevin Wolf 2257b32d7a39SMax Reitz static int wait_break_f(BlockBackend *blk, int argc, char **argv) 2258797ac58cSKevin Wolf { 22594c7b7e9bSMax Reitz while (!bdrv_debug_is_suspended(blk_bs(blk), argv[1])) { 22604c7b7e9bSMax Reitz aio_poll(blk_get_aio_context(blk), true); 2261797ac58cSKevin Wolf } 2262b32d7a39SMax Reitz return 0; 2263797ac58cSKevin Wolf } 2264797ac58cSKevin Wolf 2265797ac58cSKevin Wolf static const cmdinfo_t wait_break_cmd = { 2266797ac58cSKevin Wolf .name = "wait_break", 2267797ac58cSKevin Wolf .argmin = 1, 2268797ac58cSKevin Wolf .argmax = 1, 2269797ac58cSKevin Wolf .cfunc = wait_break_f, 2270797ac58cSKevin Wolf .args = "tag", 2271797ac58cSKevin Wolf .oneline = "waits for the suspension of a request", 2272797ac58cSKevin Wolf }; 2273797ac58cSKevin Wolf 2274b32d7a39SMax Reitz static int abort_f(BlockBackend *blk, int argc, char **argv) 2275797ac58cSKevin Wolf { 2276797ac58cSKevin Wolf abort(); 2277797ac58cSKevin Wolf } 2278797ac58cSKevin Wolf 2279797ac58cSKevin Wolf static const cmdinfo_t abort_cmd = { 2280797ac58cSKevin Wolf .name = "abort", 2281797ac58cSKevin Wolf .cfunc = abort_f, 2282797ac58cSKevin Wolf .flags = CMD_NOFILE_OK, 2283797ac58cSKevin Wolf .oneline = "simulate a program crash using abort(3)", 2284797ac58cSKevin Wolf }; 2285797ac58cSKevin Wolf 22860e82dc7bSMax Reitz static void sigraise_help(void) 22870e82dc7bSMax Reitz { 22880e82dc7bSMax Reitz printf( 22890e82dc7bSMax Reitz "\n" 22900e82dc7bSMax Reitz " raises the given signal\n" 22910e82dc7bSMax Reitz "\n" 22920e82dc7bSMax Reitz " Example:\n" 22930e82dc7bSMax Reitz " 'sigraise %i' - raises SIGTERM\n" 22940e82dc7bSMax Reitz "\n" 22950e82dc7bSMax Reitz " Invokes raise(signal), where \"signal\" is the mandatory integer argument\n" 22960e82dc7bSMax Reitz " given to sigraise.\n" 22970e82dc7bSMax Reitz "\n", SIGTERM); 22980e82dc7bSMax Reitz } 22990e82dc7bSMax Reitz 2300b32d7a39SMax Reitz static int sigraise_f(BlockBackend *blk, int argc, char **argv); 23010e82dc7bSMax Reitz 23020e82dc7bSMax Reitz static const cmdinfo_t sigraise_cmd = { 23030e82dc7bSMax Reitz .name = "sigraise", 23040e82dc7bSMax Reitz .cfunc = sigraise_f, 23050e82dc7bSMax Reitz .argmin = 1, 23060e82dc7bSMax Reitz .argmax = 1, 23070e82dc7bSMax Reitz .flags = CMD_NOFILE_OK, 23080e82dc7bSMax Reitz .args = "signal", 23090e82dc7bSMax Reitz .oneline = "raises a signal", 23100e82dc7bSMax Reitz .help = sigraise_help, 23110e82dc7bSMax Reitz }; 23120e82dc7bSMax Reitz 2313b32d7a39SMax Reitz static int sigraise_f(BlockBackend *blk, int argc, char **argv) 23140e82dc7bSMax Reitz { 23159b0beaf3SJohn Snow int64_t sig = cvtnum(argv[1]); 23160e82dc7bSMax Reitz if (sig < 0) { 2317a9ecfa00SJohn Snow print_cvtnum_err(sig, argv[1]); 2318b32d7a39SMax Reitz return sig; 23199b0beaf3SJohn Snow } else if (sig > NSIG) { 23209b0beaf3SJohn Snow printf("signal argument '%s' is too large to be a valid signal\n", 23219b0beaf3SJohn Snow argv[1]); 2322b32d7a39SMax Reitz return -EINVAL; 23230e82dc7bSMax Reitz } 23240e82dc7bSMax Reitz 23250e82dc7bSMax Reitz /* Using raise() to kill this process does not necessarily flush all open 23260e82dc7bSMax Reitz * streams. At least stdout and stderr (although the latter should be 23270e82dc7bSMax Reitz * non-buffered anyway) should be flushed, though. */ 23280e82dc7bSMax Reitz fflush(stdout); 23290e82dc7bSMax Reitz fflush(stderr); 23300e82dc7bSMax Reitz 23310e82dc7bSMax Reitz raise(sig); 2332b32d7a39SMax Reitz 2333b32d7a39SMax Reitz return 0; 23340e82dc7bSMax Reitz } 23350e82dc7bSMax Reitz 2336cd33d02aSKevin Wolf static void sleep_cb(void *opaque) 2337cd33d02aSKevin Wolf { 2338cd33d02aSKevin Wolf bool *expired = opaque; 2339cd33d02aSKevin Wolf *expired = true; 2340cd33d02aSKevin Wolf } 2341cd33d02aSKevin Wolf 2342b32d7a39SMax Reitz static int sleep_f(BlockBackend *blk, int argc, char **argv) 2343cd33d02aSKevin Wolf { 2344cd33d02aSKevin Wolf char *endptr; 2345cd33d02aSKevin Wolf long ms; 2346cd33d02aSKevin Wolf struct QEMUTimer *timer; 2347cd33d02aSKevin Wolf bool expired = false; 2348cd33d02aSKevin Wolf 2349cd33d02aSKevin Wolf ms = strtol(argv[1], &endptr, 0); 2350cd33d02aSKevin Wolf if (ms < 0 || *endptr != '\0') { 2351cd33d02aSKevin Wolf printf("%s is not a valid number\n", argv[1]); 2352b32d7a39SMax Reitz return -EINVAL; 2353cd33d02aSKevin Wolf } 2354cd33d02aSKevin Wolf 2355cd33d02aSKevin Wolf timer = timer_new_ns(QEMU_CLOCK_HOST, sleep_cb, &expired); 2356cd33d02aSKevin Wolf timer_mod(timer, qemu_clock_get_ns(QEMU_CLOCK_HOST) + SCALE_MS * ms); 2357cd33d02aSKevin Wolf 2358cd33d02aSKevin Wolf while (!expired) { 2359cd33d02aSKevin Wolf main_loop_wait(false); 2360cd33d02aSKevin Wolf } 2361cd33d02aSKevin Wolf 2362cd33d02aSKevin Wolf timer_free(timer); 2363b32d7a39SMax Reitz return 0; 2364cd33d02aSKevin Wolf } 2365cd33d02aSKevin Wolf 2366cd33d02aSKevin Wolf static const cmdinfo_t sleep_cmd = { 2367cd33d02aSKevin Wolf .name = "sleep", 2368cd33d02aSKevin Wolf .argmin = 1, 2369cd33d02aSKevin Wolf .argmax = 1, 2370cd33d02aSKevin Wolf .cfunc = sleep_f, 2371cd33d02aSKevin Wolf .flags = CMD_NOFILE_OK, 2372cd33d02aSKevin Wolf .oneline = "waits for the given value in milliseconds", 2373cd33d02aSKevin Wolf }; 2374cd33d02aSKevin Wolf 2375f18a834aSKevin Wolf static void help_oneline(const char *cmd, const cmdinfo_t *ct) 2376f18a834aSKevin Wolf { 2377f18a834aSKevin Wolf if (cmd) { 2378f18a834aSKevin Wolf printf("%s ", cmd); 2379f18a834aSKevin Wolf } else { 2380f18a834aSKevin Wolf printf("%s ", ct->name); 2381f18a834aSKevin Wolf if (ct->altname) { 2382f18a834aSKevin Wolf printf("(or %s) ", ct->altname); 2383f18a834aSKevin Wolf } 2384f18a834aSKevin Wolf } 2385f18a834aSKevin Wolf 2386f18a834aSKevin Wolf if (ct->args) { 2387f18a834aSKevin Wolf printf("%s ", ct->args); 2388f18a834aSKevin Wolf } 2389f18a834aSKevin Wolf printf("-- %s\n", ct->oneline); 2390f18a834aSKevin Wolf } 2391f18a834aSKevin Wolf 2392f18a834aSKevin Wolf static void help_onecmd(const char *cmd, const cmdinfo_t *ct) 2393f18a834aSKevin Wolf { 2394f18a834aSKevin Wolf help_oneline(cmd, ct); 2395f18a834aSKevin Wolf if (ct->help) { 2396f18a834aSKevin Wolf ct->help(); 2397f18a834aSKevin Wolf } 2398f18a834aSKevin Wolf } 2399f18a834aSKevin Wolf 2400f18a834aSKevin Wolf static void help_all(void) 2401f18a834aSKevin Wolf { 2402f18a834aSKevin Wolf const cmdinfo_t *ct; 2403f18a834aSKevin Wolf 2404f18a834aSKevin Wolf for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { 2405f18a834aSKevin Wolf help_oneline(ct->name, ct); 2406f18a834aSKevin Wolf } 2407f18a834aSKevin Wolf printf("\nUse 'help commandname' for extended help.\n"); 2408f18a834aSKevin Wolf } 2409f18a834aSKevin Wolf 2410b32d7a39SMax Reitz static int help_f(BlockBackend *blk, int argc, char **argv) 2411f18a834aSKevin Wolf { 2412f18a834aSKevin Wolf const cmdinfo_t *ct; 2413f18a834aSKevin Wolf 2414f18a834aSKevin Wolf if (argc == 1) { 2415f18a834aSKevin Wolf help_all(); 2416b32d7a39SMax Reitz return 0; 2417f18a834aSKevin Wolf } 2418f18a834aSKevin Wolf 2419f18a834aSKevin Wolf ct = find_command(argv[1]); 2420f18a834aSKevin Wolf if (ct == NULL) { 2421f18a834aSKevin Wolf printf("command %s not found\n", argv[1]); 2422b32d7a39SMax Reitz return -EINVAL; 2423f18a834aSKevin Wolf } 2424f18a834aSKevin Wolf 2425f18a834aSKevin Wolf help_onecmd(argv[1], ct); 2426b32d7a39SMax Reitz return 0; 2427f18a834aSKevin Wolf } 2428f18a834aSKevin Wolf 2429f18a834aSKevin Wolf static const cmdinfo_t help_cmd = { 2430f18a834aSKevin Wolf .name = "help", 2431f18a834aSKevin Wolf .altname = "?", 2432f18a834aSKevin Wolf .cfunc = help_f, 2433f18a834aSKevin Wolf .argmin = 0, 2434f18a834aSKevin Wolf .argmax = 1, 2435f18a834aSKevin Wolf .flags = CMD_FLAG_GLOBAL, 2436f18a834aSKevin Wolf .args = "[command]", 2437f18a834aSKevin Wolf .oneline = "help for one or all commands", 2438f18a834aSKevin Wolf }; 2439f18a834aSKevin Wolf 2440b32d7a39SMax Reitz int qemuio_command(BlockBackend *blk, const char *cmd) 2441dd583296SKevin Wolf { 244215afd94aSPaolo Bonzini AioContext *ctx; 2443dd583296SKevin Wolf char *input; 2444dd583296SKevin Wolf const cmdinfo_t *ct; 2445dd583296SKevin Wolf char **v; 2446dd583296SKevin Wolf int c; 2447b32d7a39SMax Reitz int ret = 0; 2448dd583296SKevin Wolf 2449dd583296SKevin Wolf input = g_strdup(cmd); 2450dd583296SKevin Wolf v = breakline(input, &c); 2451dd583296SKevin Wolf if (c) { 2452dd583296SKevin Wolf ct = find_command(v[0]); 2453dd583296SKevin Wolf if (ct) { 245415afd94aSPaolo Bonzini ctx = blk ? blk_get_aio_context(blk) : qemu_get_aio_context(); 245515afd94aSPaolo Bonzini aio_context_acquire(ctx); 2456b32d7a39SMax Reitz ret = command(blk, ct, c, v); 245715afd94aSPaolo Bonzini aio_context_release(ctx); 2458dd583296SKevin Wolf } else { 2459dd583296SKevin Wolf fprintf(stderr, "command \"%s\" not found\n", v[0]); 2460b32d7a39SMax Reitz ret = -EINVAL; 2461dd583296SKevin Wolf } 2462dd583296SKevin Wolf } 2463dd583296SKevin Wolf g_free(input); 2464dd583296SKevin Wolf g_free(v); 2465b32d7a39SMax Reitz 2466b32d7a39SMax Reitz return ret; 2467dd583296SKevin Wolf } 2468dd583296SKevin Wolf 2469797ac58cSKevin Wolf static void __attribute((constructor)) init_qemuio_commands(void) 2470797ac58cSKevin Wolf { 2471797ac58cSKevin Wolf /* initialize commands */ 2472c2cdf5c5SKevin Wolf qemuio_add_command(&help_cmd); 2473c2cdf5c5SKevin Wolf qemuio_add_command(&read_cmd); 2474c2cdf5c5SKevin Wolf qemuio_add_command(&readv_cmd); 2475c2cdf5c5SKevin Wolf qemuio_add_command(&write_cmd); 2476c2cdf5c5SKevin Wolf qemuio_add_command(&writev_cmd); 2477c2cdf5c5SKevin Wolf qemuio_add_command(&aio_read_cmd); 2478c2cdf5c5SKevin Wolf qemuio_add_command(&aio_write_cmd); 2479c2cdf5c5SKevin Wolf qemuio_add_command(&aio_flush_cmd); 2480c2cdf5c5SKevin Wolf qemuio_add_command(&flush_cmd); 2481c2cdf5c5SKevin Wolf qemuio_add_command(&truncate_cmd); 2482c2cdf5c5SKevin Wolf qemuio_add_command(&length_cmd); 2483c2cdf5c5SKevin Wolf qemuio_add_command(&info_cmd); 2484c2cdf5c5SKevin Wolf qemuio_add_command(&discard_cmd); 2485c2cdf5c5SKevin Wolf qemuio_add_command(&alloc_cmd); 2486c2cdf5c5SKevin Wolf qemuio_add_command(&map_cmd); 24875bbd2e59SKevin Wolf qemuio_add_command(&reopen_cmd); 2488c2cdf5c5SKevin Wolf qemuio_add_command(&break_cmd); 24894cc70e93SFam Zheng qemuio_add_command(&remove_break_cmd); 2490c2cdf5c5SKevin Wolf qemuio_add_command(&resume_cmd); 2491c2cdf5c5SKevin Wolf qemuio_add_command(&wait_break_cmd); 2492c2cdf5c5SKevin Wolf qemuio_add_command(&abort_cmd); 2493cd33d02aSKevin Wolf qemuio_add_command(&sleep_cmd); 24940e82dc7bSMax Reitz qemuio_add_command(&sigraise_cmd); 2495797ac58cSKevin Wolf } 2496