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" 24*5df022cfSPeter Maydell #include "qemu/memalign.h" 25797ac58cSKevin Wolf 26797ac58cSKevin Wolf #define CMD_NOFILE_OK 0x01 27797ac58cSKevin Wolf 28f9883880SStefan Weil bool qemuio_misalign; 29797ac58cSKevin Wolf 30c2cdf5c5SKevin Wolf static cmdinfo_t *cmdtab; 31c2cdf5c5SKevin Wolf static int ncmds; 32c2cdf5c5SKevin Wolf 33c2cdf5c5SKevin Wolf static int compare_cmdname(const void *a, const void *b) 34c2cdf5c5SKevin Wolf { 35c2cdf5c5SKevin Wolf return strcmp(((const cmdinfo_t *)a)->name, 36c2cdf5c5SKevin Wolf ((const cmdinfo_t *)b)->name); 37c2cdf5c5SKevin Wolf } 38c2cdf5c5SKevin Wolf 39c2cdf5c5SKevin Wolf void qemuio_add_command(const cmdinfo_t *ci) 40c2cdf5c5SKevin Wolf { 416aabeb58SPeter Maydell /* ci->perm assumes a file is open, but the GLOBAL and NOFILE_OK 426aabeb58SPeter Maydell * flags allow it not to be, so that combination is invalid. 436aabeb58SPeter Maydell * Catch it now rather than letting it manifest as a crash if a 446aabeb58SPeter Maydell * particular set of command line options are used. 456aabeb58SPeter Maydell */ 466aabeb58SPeter Maydell assert(ci->perm == 0 || 476aabeb58SPeter Maydell (ci->flags & (CMD_FLAG_GLOBAL | CMD_NOFILE_OK)) == 0); 4802c4f26bSMarkus Armbruster cmdtab = g_renew(cmdinfo_t, cmdtab, ++ncmds); 49c2cdf5c5SKevin Wolf cmdtab[ncmds - 1] = *ci; 50c2cdf5c5SKevin Wolf qsort(cmdtab, ncmds, sizeof(*cmdtab), compare_cmdname); 51c2cdf5c5SKevin Wolf } 52c2cdf5c5SKevin Wolf 53b444d0e9SMax Reitz void qemuio_command_usage(const cmdinfo_t *ci) 54c2cdf5c5SKevin Wolf { 55c2cdf5c5SKevin Wolf printf("%s %s -- %s\n", ci->name, ci->args, ci->oneline); 56c2cdf5c5SKevin Wolf } 57c2cdf5c5SKevin Wolf 584c7b7e9bSMax Reitz static int init_check_command(BlockBackend *blk, const cmdinfo_t *ct) 59c2cdf5c5SKevin Wolf { 60c2cdf5c5SKevin Wolf if (ct->flags & CMD_FLAG_GLOBAL) { 61c2cdf5c5SKevin Wolf return 1; 62c2cdf5c5SKevin Wolf } 634c7b7e9bSMax Reitz if (!(ct->flags & CMD_NOFILE_OK) && !blk) { 64c2cdf5c5SKevin Wolf fprintf(stderr, "no file open, try 'help open'\n"); 65c2cdf5c5SKevin Wolf return 0; 66c2cdf5c5SKevin Wolf } 67c2cdf5c5SKevin Wolf return 1; 68c2cdf5c5SKevin Wolf } 69c2cdf5c5SKevin Wolf 70b32d7a39SMax Reitz static int command(BlockBackend *blk, const cmdinfo_t *ct, int argc, 713d21994fSKevin Wolf char **argv) 72c2cdf5c5SKevin Wolf { 73c2cdf5c5SKevin Wolf char *cmd = argv[0]; 74c2cdf5c5SKevin Wolf 754c7b7e9bSMax Reitz if (!init_check_command(blk, ct)) { 76b32d7a39SMax Reitz return -EINVAL; 77c2cdf5c5SKevin Wolf } 78c2cdf5c5SKevin Wolf 79c2cdf5c5SKevin Wolf if (argc - 1 < ct->argmin || (ct->argmax != -1 && argc - 1 > ct->argmax)) { 80c2cdf5c5SKevin Wolf if (ct->argmax == -1) { 81c2cdf5c5SKevin Wolf fprintf(stderr, 82c2cdf5c5SKevin Wolf "bad argument count %d to %s, expected at least %d arguments\n", 83c2cdf5c5SKevin Wolf argc-1, cmd, ct->argmin); 84c2cdf5c5SKevin Wolf } else if (ct->argmin == ct->argmax) { 85c2cdf5c5SKevin Wolf fprintf(stderr, 86c2cdf5c5SKevin Wolf "bad argument count %d to %s, expected %d arguments\n", 87c2cdf5c5SKevin Wolf argc-1, cmd, ct->argmin); 88c2cdf5c5SKevin Wolf } else { 89c2cdf5c5SKevin Wolf fprintf(stderr, 90c2cdf5c5SKevin Wolf "bad argument count %d to %s, expected between %d and %d arguments\n", 91c2cdf5c5SKevin Wolf argc-1, cmd, ct->argmin, ct->argmax); 92c2cdf5c5SKevin Wolf } 93b32d7a39SMax Reitz return -EINVAL; 94c2cdf5c5SKevin Wolf } 95887354bdSKevin Wolf 968eaf1018SVladimir Sementsov-Ogievskiy /* 978eaf1018SVladimir Sementsov-Ogievskiy * Request additional permissions if necessary for this command. The caller 98887354bdSKevin Wolf * is responsible for restoring the original permissions afterwards if this 998eaf1018SVladimir Sementsov-Ogievskiy * is what it wants. 1008eaf1018SVladimir Sementsov-Ogievskiy * 1018eaf1018SVladimir Sementsov-Ogievskiy * Coverity thinks that blk may be NULL in the following if condition. It's 1028eaf1018SVladimir Sementsov-Ogievskiy * not so: in init_check_command() we fail if blk is NULL for command with 1038eaf1018SVladimir Sementsov-Ogievskiy * both CMD_FLAG_GLOBAL and CMD_NOFILE_OK flags unset. And in 1048eaf1018SVladimir Sementsov-Ogievskiy * qemuio_add_command() we assert that command with non-zero .perm field 1058eaf1018SVladimir Sementsov-Ogievskiy * doesn't set this flags. So, the following assertion is to silence 1068eaf1018SVladimir Sementsov-Ogievskiy * Coverity: 1078eaf1018SVladimir Sementsov-Ogievskiy */ 1088eaf1018SVladimir Sementsov-Ogievskiy assert(blk || !ct->perm); 109887354bdSKevin Wolf if (ct->perm && blk_is_available(blk)) { 110887354bdSKevin Wolf uint64_t orig_perm, orig_shared_perm; 111887354bdSKevin Wolf blk_get_perm(blk, &orig_perm, &orig_shared_perm); 112887354bdSKevin Wolf 113887354bdSKevin Wolf if (ct->perm & ~orig_perm) { 114887354bdSKevin Wolf uint64_t new_perm; 115887354bdSKevin Wolf Error *local_err = NULL; 116887354bdSKevin Wolf int ret; 117887354bdSKevin Wolf 118887354bdSKevin Wolf new_perm = orig_perm | ct->perm; 119887354bdSKevin Wolf 120887354bdSKevin Wolf ret = blk_set_perm(blk, new_perm, orig_shared_perm, &local_err); 121887354bdSKevin Wolf if (ret < 0) { 122887354bdSKevin Wolf error_report_err(local_err); 123b32d7a39SMax Reitz return ret; 124887354bdSKevin Wolf } 125887354bdSKevin Wolf } 126887354bdSKevin Wolf } 127887354bdSKevin Wolf 128d339d766SRichard W.M. Jones qemu_reset_optind(); 129b32d7a39SMax Reitz return ct->cfunc(blk, argc, argv); 130c2cdf5c5SKevin Wolf } 131c2cdf5c5SKevin Wolf 132c2cdf5c5SKevin Wolf static const cmdinfo_t *find_command(const char *cmd) 133c2cdf5c5SKevin Wolf { 134c2cdf5c5SKevin Wolf cmdinfo_t *ct; 135c2cdf5c5SKevin Wolf 136c2cdf5c5SKevin Wolf for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { 137c2cdf5c5SKevin Wolf if (strcmp(ct->name, cmd) == 0 || 138c2cdf5c5SKevin Wolf (ct->altname && strcmp(ct->altname, cmd) == 0)) 139c2cdf5c5SKevin Wolf { 140c2cdf5c5SKevin Wolf return (const cmdinfo_t *)ct; 141c2cdf5c5SKevin Wolf } 142c2cdf5c5SKevin Wolf } 143c2cdf5c5SKevin Wolf return NULL; 144c2cdf5c5SKevin Wolf } 145c2cdf5c5SKevin Wolf 1464694020dSStefan Hajnoczi /* Invoke fn() for commands with a matching prefix */ 1474694020dSStefan Hajnoczi void qemuio_complete_command(const char *input, 1484694020dSStefan Hajnoczi void (*fn)(const char *cmd, void *opaque), 1494694020dSStefan Hajnoczi void *opaque) 1504694020dSStefan Hajnoczi { 1514694020dSStefan Hajnoczi cmdinfo_t *ct; 1524694020dSStefan Hajnoczi size_t input_len = strlen(input); 1534694020dSStefan Hajnoczi 1544694020dSStefan Hajnoczi for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { 1554694020dSStefan Hajnoczi if (strncmp(input, ct->name, input_len) == 0) { 1564694020dSStefan Hajnoczi fn(ct->name, opaque); 1574694020dSStefan Hajnoczi } 1584694020dSStefan Hajnoczi } 1594694020dSStefan Hajnoczi } 1604694020dSStefan Hajnoczi 161c2cdf5c5SKevin Wolf static char **breakline(char *input, int *count) 162c2cdf5c5SKevin Wolf { 163c2cdf5c5SKevin Wolf int c = 0; 164c2cdf5c5SKevin Wolf char *p; 1655839e53bSMarkus Armbruster char **rval = g_new0(char *, 1); 166c2cdf5c5SKevin Wolf 167c2cdf5c5SKevin Wolf while (rval && (p = qemu_strsep(&input, " ")) != NULL) { 168c2cdf5c5SKevin Wolf if (!*p) { 169c2cdf5c5SKevin Wolf continue; 170c2cdf5c5SKevin Wolf } 171c2cdf5c5SKevin Wolf c++; 17208193dd5SMarkus Armbruster rval = g_renew(char *, rval, (c + 1)); 173c2cdf5c5SKevin Wolf rval[c - 1] = p; 174c2cdf5c5SKevin Wolf rval[c] = NULL; 175c2cdf5c5SKevin Wolf } 176c2cdf5c5SKevin Wolf *count = c; 177c2cdf5c5SKevin Wolf return rval; 178c2cdf5c5SKevin Wolf } 179c2cdf5c5SKevin Wolf 180797ac58cSKevin Wolf static int64_t cvtnum(const char *s) 181797ac58cSKevin Wolf { 182f17fd4fdSMarkus Armbruster int err; 183f46bfdbfSMarkus Armbruster uint64_t value; 184ef5a7885SJohn Snow 185f17fd4fdSMarkus Armbruster err = qemu_strtosz(s, NULL, &value); 186f17fd4fdSMarkus Armbruster if (err < 0) { 187f17fd4fdSMarkus Armbruster return err; 188f17fd4fdSMarkus Armbruster } 189f46bfdbfSMarkus Armbruster if (value > INT64_MAX) { 190f46bfdbfSMarkus Armbruster return -ERANGE; 191f46bfdbfSMarkus Armbruster } 192f17fd4fdSMarkus Armbruster return value; 193797ac58cSKevin Wolf } 194797ac58cSKevin Wolf 195a9ecfa00SJohn Snow static void print_cvtnum_err(int64_t rc, const char *arg) 196a9ecfa00SJohn Snow { 197a9ecfa00SJohn Snow switch (rc) { 198a9ecfa00SJohn Snow case -EINVAL: 199a9ecfa00SJohn Snow printf("Parsing error: non-numeric argument," 200a9ecfa00SJohn Snow " or extraneous/unrecognized suffix -- %s\n", arg); 201a9ecfa00SJohn Snow break; 202a9ecfa00SJohn Snow case -ERANGE: 203a9ecfa00SJohn Snow printf("Parsing error: argument too large -- %s\n", arg); 204a9ecfa00SJohn Snow break; 205a9ecfa00SJohn Snow default: 206a9ecfa00SJohn Snow printf("Parsing error: %s\n", arg); 207a9ecfa00SJohn Snow } 208a9ecfa00SJohn Snow } 209a9ecfa00SJohn Snow 2100b613881SKevin Wolf #define EXABYTES(x) ((long long)(x) << 60) 2110b613881SKevin Wolf #define PETABYTES(x) ((long long)(x) << 50) 2120b613881SKevin Wolf #define TERABYTES(x) ((long long)(x) << 40) 2130b613881SKevin Wolf #define GIGABYTES(x) ((long long)(x) << 30) 2140b613881SKevin Wolf #define MEGABYTES(x) ((long long)(x) << 20) 2150b613881SKevin Wolf #define KILOBYTES(x) ((long long)(x) << 10) 2160b613881SKevin Wolf 2170b613881SKevin Wolf #define TO_EXABYTES(x) ((x) / EXABYTES(1)) 2180b613881SKevin Wolf #define TO_PETABYTES(x) ((x) / PETABYTES(1)) 2190b613881SKevin Wolf #define TO_TERABYTES(x) ((x) / TERABYTES(1)) 2200b613881SKevin Wolf #define TO_GIGABYTES(x) ((x) / GIGABYTES(1)) 2210b613881SKevin Wolf #define TO_MEGABYTES(x) ((x) / MEGABYTES(1)) 2220b613881SKevin Wolf #define TO_KILOBYTES(x) ((x) / KILOBYTES(1)) 2230b613881SKevin Wolf 2240b613881SKevin Wolf static void cvtstr(double value, char *str, size_t size) 2250b613881SKevin Wolf { 2260b613881SKevin Wolf char *trim; 2270b613881SKevin Wolf const char *suffix; 2280b613881SKevin Wolf 2290b613881SKevin Wolf if (value >= EXABYTES(1)) { 2300b613881SKevin Wolf suffix = " EiB"; 2310b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_EXABYTES(value)); 2320b613881SKevin Wolf } else if (value >= PETABYTES(1)) { 2330b613881SKevin Wolf suffix = " PiB"; 2340b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_PETABYTES(value)); 2350b613881SKevin Wolf } else if (value >= TERABYTES(1)) { 2360b613881SKevin Wolf suffix = " TiB"; 2370b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_TERABYTES(value)); 2380b613881SKevin Wolf } else if (value >= GIGABYTES(1)) { 2390b613881SKevin Wolf suffix = " GiB"; 2400b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_GIGABYTES(value)); 2410b613881SKevin Wolf } else if (value >= MEGABYTES(1)) { 2420b613881SKevin Wolf suffix = " MiB"; 2430b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_MEGABYTES(value)); 2440b613881SKevin Wolf } else if (value >= KILOBYTES(1)) { 2450b613881SKevin Wolf suffix = " KiB"; 2460b613881SKevin Wolf snprintf(str, size - 4, "%.3f", TO_KILOBYTES(value)); 2470b613881SKevin Wolf } else { 2480b613881SKevin Wolf suffix = " bytes"; 2490b613881SKevin Wolf snprintf(str, size - 6, "%f", value); 2500b613881SKevin Wolf } 2510b613881SKevin Wolf 2520b613881SKevin Wolf trim = strstr(str, ".000"); 2530b613881SKevin Wolf if (trim) { 2540b613881SKevin Wolf strcpy(trim, suffix); 2550b613881SKevin Wolf } else { 2560b613881SKevin Wolf strcat(str, suffix); 2570b613881SKevin Wolf } 2580b613881SKevin Wolf } 2590b613881SKevin Wolf 2600b613881SKevin Wolf 2610b613881SKevin Wolf 26250290c00SAlex Bennée static struct timespec tsub(struct timespec t1, struct timespec t2) 2630b613881SKevin Wolf { 26450290c00SAlex Bennée t1.tv_nsec -= t2.tv_nsec; 26550290c00SAlex Bennée if (t1.tv_nsec < 0) { 26650290c00SAlex Bennée t1.tv_nsec += NANOSECONDS_PER_SECOND; 2670b613881SKevin Wolf t1.tv_sec--; 2680b613881SKevin Wolf } 2690b613881SKevin Wolf t1.tv_sec -= t2.tv_sec; 2700b613881SKevin Wolf return t1; 2710b613881SKevin Wolf } 2720b613881SKevin Wolf 27350290c00SAlex Bennée static double tdiv(double value, struct timespec tv) 2740b613881SKevin Wolf { 27550290c00SAlex Bennée double seconds = tv.tv_sec + (tv.tv_nsec / 1e9); 27650290c00SAlex Bennée return value / seconds; 2770b613881SKevin Wolf } 2780b613881SKevin Wolf 2790b613881SKevin Wolf #define HOURS(sec) ((sec) / (60 * 60)) 2800b613881SKevin Wolf #define MINUTES(sec) (((sec) % (60 * 60)) / 60) 2810b613881SKevin Wolf #define SECONDS(sec) ((sec) % 60) 2820b613881SKevin Wolf 2830b613881SKevin Wolf enum { 2840b613881SKevin Wolf DEFAULT_TIME = 0x0, 2850b613881SKevin Wolf TERSE_FIXED_TIME = 0x1, 2860b613881SKevin Wolf VERBOSE_FIXED_TIME = 0x2, 2870b613881SKevin Wolf }; 2880b613881SKevin Wolf 28950290c00SAlex Bennée static void timestr(struct timespec *tv, char *ts, size_t size, int format) 2900b613881SKevin Wolf { 29150290c00SAlex Bennée double frac_sec = tv->tv_nsec / 1e9; 2920b613881SKevin Wolf 2930b613881SKevin Wolf if (format & TERSE_FIXED_TIME) { 2940b613881SKevin Wolf if (!HOURS(tv->tv_sec)) { 29550290c00SAlex Bennée snprintf(ts, size, "%u:%05.2f", 2960b613881SKevin Wolf (unsigned int) MINUTES(tv->tv_sec), 29750290c00SAlex Bennée SECONDS(tv->tv_sec) + frac_sec); 2980b613881SKevin Wolf return; 2990b613881SKevin Wolf } 3000b613881SKevin Wolf format |= VERBOSE_FIXED_TIME; /* fallback if hours needed */ 3010b613881SKevin Wolf } 3020b613881SKevin Wolf 3030b613881SKevin Wolf if ((format & VERBOSE_FIXED_TIME) || tv->tv_sec) { 30450290c00SAlex Bennée snprintf(ts, size, "%u:%02u:%05.2f", 3050b613881SKevin Wolf (unsigned int) HOURS(tv->tv_sec), 3060b613881SKevin Wolf (unsigned int) MINUTES(tv->tv_sec), 30750290c00SAlex Bennée SECONDS(tv->tv_sec) + frac_sec); 3080b613881SKevin Wolf } else { 30950290c00SAlex Bennée snprintf(ts, size, "%05.2f sec", frac_sec); 3100b613881SKevin Wolf } 3110b613881SKevin Wolf } 3120b613881SKevin Wolf 313797ac58cSKevin Wolf /* 314797ac58cSKevin Wolf * Parse the pattern argument to various sub-commands. 315797ac58cSKevin Wolf * 316797ac58cSKevin Wolf * Because the pattern is used as an argument to memset it must evaluate 317797ac58cSKevin Wolf * to an unsigned integer that fits into a single byte. 318797ac58cSKevin Wolf */ 319797ac58cSKevin Wolf static int parse_pattern(const char *arg) 320797ac58cSKevin Wolf { 321797ac58cSKevin Wolf char *endptr = NULL; 322797ac58cSKevin Wolf long pattern; 323797ac58cSKevin Wolf 324797ac58cSKevin Wolf pattern = strtol(arg, &endptr, 0); 325797ac58cSKevin Wolf if (pattern < 0 || pattern > UCHAR_MAX || *endptr != '\0') { 326797ac58cSKevin Wolf printf("%s is not a valid pattern byte\n", arg); 327797ac58cSKevin Wolf return -1; 328797ac58cSKevin Wolf } 329797ac58cSKevin Wolf 330797ac58cSKevin Wolf return pattern; 331797ac58cSKevin Wolf } 332797ac58cSKevin Wolf 333797ac58cSKevin Wolf /* 334797ac58cSKevin Wolf * Memory allocation helpers. 335797ac58cSKevin Wolf * 336797ac58cSKevin Wolf * Make sure memory is aligned by default, or purposefully misaligned if 337797ac58cSKevin Wolf * that is specified on the command line. 338797ac58cSKevin Wolf */ 339797ac58cSKevin Wolf 340797ac58cSKevin Wolf #define MISALIGN_OFFSET 16 3414c7b7e9bSMax Reitz static void *qemu_io_alloc(BlockBackend *blk, size_t len, int pattern) 342797ac58cSKevin Wolf { 343797ac58cSKevin Wolf void *buf; 344797ac58cSKevin Wolf 345797ac58cSKevin Wolf if (qemuio_misalign) { 346797ac58cSKevin Wolf len += MISALIGN_OFFSET; 347797ac58cSKevin Wolf } 3484c7b7e9bSMax Reitz buf = blk_blockalign(blk, len); 349797ac58cSKevin Wolf memset(buf, pattern, len); 350797ac58cSKevin Wolf if (qemuio_misalign) { 351797ac58cSKevin Wolf buf += MISALIGN_OFFSET; 352797ac58cSKevin Wolf } 353797ac58cSKevin Wolf return buf; 354797ac58cSKevin Wolf } 355797ac58cSKevin Wolf 356797ac58cSKevin Wolf static void qemu_io_free(void *p) 357797ac58cSKevin Wolf { 358797ac58cSKevin Wolf if (qemuio_misalign) { 359797ac58cSKevin Wolf p -= MISALIGN_OFFSET; 360797ac58cSKevin Wolf } 361797ac58cSKevin Wolf qemu_vfree(p); 362797ac58cSKevin Wolf } 363797ac58cSKevin Wolf 3644d731510SDenis Plotnikov /* 3654d731510SDenis Plotnikov * qemu_io_alloc_from_file() 3664d731510SDenis Plotnikov * 3674d731510SDenis Plotnikov * Allocates the buffer and populates it with the content of the given file 3684d731510SDenis Plotnikov * up to @len bytes. If the file length is less than @len, then the buffer 3694d731510SDenis Plotnikov * is populated with the file content cyclically. 3704d731510SDenis Plotnikov * 3714d731510SDenis Plotnikov * @blk - the block backend where the buffer content is going to be written to 3724d731510SDenis Plotnikov * @len - the buffer length 3734d731510SDenis Plotnikov * @file_name - the file to read the content from 3744d731510SDenis Plotnikov * 3754d731510SDenis Plotnikov * Returns: the buffer pointer on success 3764d731510SDenis Plotnikov * NULL on error 3774d731510SDenis Plotnikov */ 3784d731510SDenis Plotnikov static void *qemu_io_alloc_from_file(BlockBackend *blk, size_t len, 3794d731510SDenis Plotnikov const char *file_name) 3804d731510SDenis Plotnikov { 3814d731510SDenis Plotnikov char *buf, *buf_origin; 3824d731510SDenis Plotnikov FILE *f = fopen(file_name, "r"); 3834d731510SDenis Plotnikov int pattern_len; 3844d731510SDenis Plotnikov 3854d731510SDenis Plotnikov if (!f) { 3864d731510SDenis Plotnikov perror(file_name); 3874d731510SDenis Plotnikov return NULL; 3884d731510SDenis Plotnikov } 3894d731510SDenis Plotnikov 3904d731510SDenis Plotnikov if (qemuio_misalign) { 3914d731510SDenis Plotnikov len += MISALIGN_OFFSET; 3924d731510SDenis Plotnikov } 3934d731510SDenis Plotnikov 3944d731510SDenis Plotnikov buf_origin = buf = blk_blockalign(blk, len); 3954d731510SDenis Plotnikov 3964d731510SDenis Plotnikov if (qemuio_misalign) { 3974d731510SDenis Plotnikov buf_origin += MISALIGN_OFFSET; 3984d731510SDenis Plotnikov buf += MISALIGN_OFFSET; 3994d731510SDenis Plotnikov len -= MISALIGN_OFFSET; 4004d731510SDenis Plotnikov } 4014d731510SDenis Plotnikov 4024d731510SDenis Plotnikov pattern_len = fread(buf_origin, 1, len, f); 4034d731510SDenis Plotnikov 4044d731510SDenis Plotnikov if (ferror(f)) { 4054d731510SDenis Plotnikov perror(file_name); 4064d731510SDenis Plotnikov goto error; 4074d731510SDenis Plotnikov } 4084d731510SDenis Plotnikov 4094d731510SDenis Plotnikov if (pattern_len == 0) { 4104d731510SDenis Plotnikov fprintf(stderr, "%s: file is empty\n", file_name); 4114d731510SDenis Plotnikov goto error; 4124d731510SDenis Plotnikov } 4134d731510SDenis Plotnikov 4144d731510SDenis Plotnikov fclose(f); 415c8e68b43SKevin Wolf f = NULL; 4164d731510SDenis Plotnikov 4174d731510SDenis Plotnikov if (len > pattern_len) { 4184d731510SDenis Plotnikov len -= pattern_len; 4194d731510SDenis Plotnikov buf += pattern_len; 4204d731510SDenis Plotnikov 4214d731510SDenis Plotnikov while (len > 0) { 4224d731510SDenis Plotnikov size_t len_to_copy = MIN(pattern_len, len); 4234d731510SDenis Plotnikov 4244d731510SDenis Plotnikov memcpy(buf, buf_origin, len_to_copy); 4254d731510SDenis Plotnikov 4264d731510SDenis Plotnikov len -= len_to_copy; 4274d731510SDenis Plotnikov buf += len_to_copy; 4284d731510SDenis Plotnikov } 4294d731510SDenis Plotnikov } 4304d731510SDenis Plotnikov 4314d731510SDenis Plotnikov return buf_origin; 4324d731510SDenis Plotnikov 4334d731510SDenis Plotnikov error: 4344d731510SDenis Plotnikov qemu_io_free(buf_origin); 435c8e68b43SKevin Wolf if (f) { 436c8e68b43SKevin Wolf fclose(f); 437c8e68b43SKevin Wolf } 4384d731510SDenis Plotnikov return NULL; 4394d731510SDenis Plotnikov } 4404d731510SDenis Plotnikov 4419b0beaf3SJohn Snow static void dump_buffer(const void *buffer, int64_t offset, int64_t len) 442797ac58cSKevin Wolf { 4439b0beaf3SJohn Snow uint64_t i; 4449b0beaf3SJohn Snow int j; 445797ac58cSKevin Wolf const uint8_t *p; 446797ac58cSKevin Wolf 447797ac58cSKevin Wolf for (i = 0, p = buffer; i < len; i += 16) { 448797ac58cSKevin Wolf const uint8_t *s = p; 449797ac58cSKevin Wolf 450797ac58cSKevin Wolf printf("%08" PRIx64 ": ", offset + i); 451797ac58cSKevin Wolf for (j = 0; j < 16 && i + j < len; j++, p++) { 452797ac58cSKevin Wolf printf("%02x ", *p); 453797ac58cSKevin Wolf } 454797ac58cSKevin Wolf printf(" "); 455797ac58cSKevin Wolf for (j = 0; j < 16 && i + j < len; j++, s++) { 456797ac58cSKevin Wolf if (isalnum(*s)) { 457797ac58cSKevin Wolf printf("%c", *s); 458797ac58cSKevin Wolf } else { 459797ac58cSKevin Wolf printf("."); 460797ac58cSKevin Wolf } 461797ac58cSKevin Wolf } 462797ac58cSKevin Wolf printf("\n"); 463797ac58cSKevin Wolf } 464797ac58cSKevin Wolf } 465797ac58cSKevin Wolf 46650290c00SAlex Bennée static void print_report(const char *op, struct timespec *t, int64_t offset, 467dc38852aSEric Blake int64_t count, int64_t total, int cnt, bool Cflag) 468797ac58cSKevin Wolf { 469797ac58cSKevin Wolf char s1[64], s2[64], ts[64]; 470797ac58cSKevin Wolf 471797ac58cSKevin Wolf timestr(t, ts, sizeof(ts), Cflag ? VERBOSE_FIXED_TIME : 0); 472797ac58cSKevin Wolf if (!Cflag) { 473797ac58cSKevin Wolf cvtstr((double)total, s1, sizeof(s1)); 474797ac58cSKevin Wolf cvtstr(tdiv((double)total, *t), s2, sizeof(s2)); 4759b0beaf3SJohn Snow printf("%s %"PRId64"/%"PRId64" bytes at offset %" PRId64 "\n", 476797ac58cSKevin Wolf op, total, count, offset); 477797ac58cSKevin Wolf printf("%s, %d ops; %s (%s/sec and %.4f ops/sec)\n", 478797ac58cSKevin Wolf s1, cnt, ts, s2, tdiv((double)cnt, *t)); 479797ac58cSKevin Wolf } else {/* bytes,ops,time,bytes/sec,ops/sec */ 4809b0beaf3SJohn Snow printf("%"PRId64",%d,%s,%.3f,%.3f\n", 481797ac58cSKevin Wolf total, cnt, ts, 482797ac58cSKevin Wolf tdiv((double)total, *t), 483797ac58cSKevin Wolf tdiv((double)cnt, *t)); 484797ac58cSKevin Wolf } 485797ac58cSKevin Wolf } 486797ac58cSKevin Wolf 487797ac58cSKevin Wolf /* 488797ac58cSKevin Wolf * Parse multiple length statements for vectored I/O, and construct an I/O 489797ac58cSKevin Wolf * vector matching it. 490797ac58cSKevin Wolf */ 491797ac58cSKevin Wolf static void * 4924c7b7e9bSMax Reitz create_iovec(BlockBackend *blk, QEMUIOVector *qiov, char **argv, int nr_iov, 493797ac58cSKevin Wolf int pattern) 494797ac58cSKevin Wolf { 495797ac58cSKevin Wolf size_t *sizes = g_new0(size_t, nr_iov); 496797ac58cSKevin Wolf size_t count = 0; 497797ac58cSKevin Wolf void *buf = NULL; 498797ac58cSKevin Wolf void *p; 499797ac58cSKevin Wolf int i; 500797ac58cSKevin Wolf 501797ac58cSKevin Wolf for (i = 0; i < nr_iov; i++) { 502797ac58cSKevin Wolf char *arg = argv[i]; 503797ac58cSKevin Wolf int64_t len; 504797ac58cSKevin Wolf 505797ac58cSKevin Wolf len = cvtnum(arg); 506797ac58cSKevin Wolf if (len < 0) { 507a9ecfa00SJohn Snow print_cvtnum_err(len, arg); 508797ac58cSKevin Wolf goto fail; 509797ac58cSKevin Wolf } 510797ac58cSKevin Wolf 5113026c468SAlberto Garcia if (len > BDRV_REQUEST_MAX_BYTES) { 5123026c468SAlberto Garcia printf("Argument '%s' exceeds maximum size %" PRIu64 "\n", arg, 5133026c468SAlberto Garcia (uint64_t)BDRV_REQUEST_MAX_BYTES); 5143026c468SAlberto Garcia goto fail; 5153026c468SAlberto Garcia } 5163026c468SAlberto Garcia 5173026c468SAlberto Garcia if (count > BDRV_REQUEST_MAX_BYTES - len) { 5183026c468SAlberto Garcia printf("The total number of bytes exceed the maximum size %" PRIu64 5193026c468SAlberto Garcia "\n", (uint64_t)BDRV_REQUEST_MAX_BYTES); 520797ac58cSKevin Wolf goto fail; 521797ac58cSKevin Wolf } 522797ac58cSKevin Wolf 523797ac58cSKevin Wolf sizes[i] = len; 524797ac58cSKevin Wolf count += len; 525797ac58cSKevin Wolf } 526797ac58cSKevin Wolf 527797ac58cSKevin Wolf qemu_iovec_init(qiov, nr_iov); 528797ac58cSKevin Wolf 5294c7b7e9bSMax Reitz buf = p = qemu_io_alloc(blk, count, pattern); 530797ac58cSKevin Wolf 531797ac58cSKevin Wolf for (i = 0; i < nr_iov; i++) { 532797ac58cSKevin Wolf qemu_iovec_add(qiov, p, sizes[i]); 533797ac58cSKevin Wolf p += sizes[i]; 534797ac58cSKevin Wolf } 535797ac58cSKevin Wolf 536797ac58cSKevin Wolf fail: 537797ac58cSKevin Wolf g_free(sizes); 538797ac58cSKevin Wolf return buf; 539797ac58cSKevin Wolf } 540797ac58cSKevin Wolf 5419b0beaf3SJohn Snow static int do_pread(BlockBackend *blk, char *buf, int64_t offset, 542f5a5ca79SManos Pitsidianakis int64_t bytes, int64_t *total) 543797ac58cSKevin Wolf { 544f5a5ca79SManos Pitsidianakis if (bytes > INT_MAX) { 5459b0beaf3SJohn Snow return -ERANGE; 5469b0beaf3SJohn Snow } 5479b0beaf3SJohn Snow 548f5a5ca79SManos Pitsidianakis *total = blk_pread(blk, offset, (uint8_t *)buf, bytes); 549797ac58cSKevin Wolf if (*total < 0) { 550797ac58cSKevin Wolf return *total; 551797ac58cSKevin Wolf } 552797ac58cSKevin Wolf return 1; 553797ac58cSKevin Wolf } 554797ac58cSKevin Wolf 5559b0beaf3SJohn Snow static int do_pwrite(BlockBackend *blk, char *buf, int64_t offset, 556f5a5ca79SManos Pitsidianakis int64_t bytes, int flags, int64_t *total) 557797ac58cSKevin Wolf { 558f5a5ca79SManos Pitsidianakis if (bytes > INT_MAX) { 5599b0beaf3SJohn Snow return -ERANGE; 5609b0beaf3SJohn Snow } 5619b0beaf3SJohn Snow 562f5a5ca79SManos Pitsidianakis *total = blk_pwrite(blk, offset, (uint8_t *)buf, bytes, flags); 563797ac58cSKevin Wolf if (*total < 0) { 564797ac58cSKevin Wolf return *total; 565797ac58cSKevin Wolf } 566797ac58cSKevin Wolf return 1; 567797ac58cSKevin Wolf } 568797ac58cSKevin Wolf 569797ac58cSKevin Wolf typedef struct { 5704c7b7e9bSMax Reitz BlockBackend *blk; 571797ac58cSKevin Wolf int64_t offset; 572f5a5ca79SManos Pitsidianakis int64_t bytes; 5739b0beaf3SJohn Snow int64_t *total; 574770e0e0eSEric Blake int flags; 575797ac58cSKevin Wolf int ret; 576797ac58cSKevin Wolf bool done; 577797ac58cSKevin Wolf } CoWriteZeroes; 578797ac58cSKevin Wolf 579d004bd52SEric Blake static void coroutine_fn co_pwrite_zeroes_entry(void *opaque) 580797ac58cSKevin Wolf { 581797ac58cSKevin Wolf CoWriteZeroes *data = opaque; 582797ac58cSKevin Wolf 583f5a5ca79SManos Pitsidianakis data->ret = blk_co_pwrite_zeroes(data->blk, data->offset, data->bytes, 584770e0e0eSEric Blake data->flags); 585797ac58cSKevin Wolf data->done = true; 586797ac58cSKevin Wolf if (data->ret < 0) { 587797ac58cSKevin Wolf *data->total = data->ret; 588797ac58cSKevin Wolf return; 589797ac58cSKevin Wolf } 590797ac58cSKevin Wolf 591f5a5ca79SManos Pitsidianakis *data->total = data->bytes; 592797ac58cSKevin Wolf } 593797ac58cSKevin Wolf 594d004bd52SEric Blake static int do_co_pwrite_zeroes(BlockBackend *blk, int64_t offset, 595f5a5ca79SManos Pitsidianakis int64_t bytes, int flags, int64_t *total) 596797ac58cSKevin Wolf { 597797ac58cSKevin Wolf Coroutine *co; 598797ac58cSKevin Wolf CoWriteZeroes data = { 5994c7b7e9bSMax Reitz .blk = blk, 600797ac58cSKevin Wolf .offset = offset, 601f5a5ca79SManos Pitsidianakis .bytes = bytes, 602797ac58cSKevin Wolf .total = total, 603770e0e0eSEric Blake .flags = flags, 604797ac58cSKevin Wolf .done = false, 605797ac58cSKevin Wolf }; 606797ac58cSKevin Wolf 607f5a5ca79SManos Pitsidianakis if (bytes > INT_MAX) { 6089b0beaf3SJohn Snow return -ERANGE; 6099b0beaf3SJohn Snow } 6109b0beaf3SJohn Snow 6110b8b8753SPaolo Bonzini co = qemu_coroutine_create(co_pwrite_zeroes_entry, &data); 612324ec3e4SFam Zheng bdrv_coroutine_enter(blk_bs(blk), co); 613797ac58cSKevin Wolf while (!data.done) { 6144c7b7e9bSMax Reitz aio_poll(blk_get_aio_context(blk), true); 615797ac58cSKevin Wolf } 616797ac58cSKevin Wolf if (data.ret < 0) { 617797ac58cSKevin Wolf return data.ret; 618797ac58cSKevin Wolf } else { 619797ac58cSKevin Wolf return 1; 620797ac58cSKevin Wolf } 621797ac58cSKevin Wolf } 622797ac58cSKevin Wolf 6234c7b7e9bSMax Reitz static int do_write_compressed(BlockBackend *blk, char *buf, int64_t offset, 624f5a5ca79SManos Pitsidianakis int64_t bytes, int64_t *total) 625797ac58cSKevin Wolf { 626797ac58cSKevin Wolf int ret; 627797ac58cSKevin Wolf 62841ae31e3SAlberto Garcia if (bytes > BDRV_REQUEST_MAX_BYTES) { 6299b0beaf3SJohn Snow return -ERANGE; 6309b0beaf3SJohn Snow } 6319b0beaf3SJohn Snow 632f5a5ca79SManos Pitsidianakis ret = blk_pwrite_compressed(blk, offset, buf, bytes); 633797ac58cSKevin Wolf if (ret < 0) { 634797ac58cSKevin Wolf return ret; 635797ac58cSKevin Wolf } 636f5a5ca79SManos Pitsidianakis *total = bytes; 637797ac58cSKevin Wolf return 1; 638797ac58cSKevin Wolf } 639797ac58cSKevin Wolf 6404c7b7e9bSMax Reitz static int do_load_vmstate(BlockBackend *blk, char *buf, int64_t offset, 6419b0beaf3SJohn Snow int64_t count, int64_t *total) 642797ac58cSKevin Wolf { 6439b0beaf3SJohn Snow if (count > INT_MAX) { 6449b0beaf3SJohn Snow return -ERANGE; 6459b0beaf3SJohn Snow } 6469b0beaf3SJohn Snow 6474c7b7e9bSMax Reitz *total = blk_load_vmstate(blk, (uint8_t *)buf, offset, count); 648797ac58cSKevin Wolf if (*total < 0) { 649797ac58cSKevin Wolf return *total; 650797ac58cSKevin Wolf } 651797ac58cSKevin Wolf return 1; 652797ac58cSKevin Wolf } 653797ac58cSKevin Wolf 6544c7b7e9bSMax Reitz static int do_save_vmstate(BlockBackend *blk, char *buf, int64_t offset, 6559b0beaf3SJohn Snow int64_t count, int64_t *total) 656797ac58cSKevin Wolf { 6579b0beaf3SJohn Snow if (count > INT_MAX) { 6589b0beaf3SJohn Snow return -ERANGE; 6599b0beaf3SJohn Snow } 6609b0beaf3SJohn Snow 6614c7b7e9bSMax Reitz *total = blk_save_vmstate(blk, (uint8_t *)buf, offset, count); 662797ac58cSKevin Wolf if (*total < 0) { 663797ac58cSKevin Wolf return *total; 664797ac58cSKevin Wolf } 665797ac58cSKevin Wolf return 1; 666797ac58cSKevin Wolf } 667797ac58cSKevin Wolf 668797ac58cSKevin Wolf #define NOT_DONE 0x7fffffff 669797ac58cSKevin Wolf static void aio_rw_done(void *opaque, int ret) 670797ac58cSKevin Wolf { 671797ac58cSKevin Wolf *(int *)opaque = ret; 672797ac58cSKevin Wolf } 673797ac58cSKevin Wolf 6744c7b7e9bSMax Reitz static int do_aio_readv(BlockBackend *blk, QEMUIOVector *qiov, 675797ac58cSKevin Wolf int64_t offset, int *total) 676797ac58cSKevin Wolf { 677797ac58cSKevin Wolf int async_ret = NOT_DONE; 678797ac58cSKevin Wolf 6797b3f9712SEric Blake blk_aio_preadv(blk, offset, qiov, 0, aio_rw_done, &async_ret); 680797ac58cSKevin Wolf while (async_ret == NOT_DONE) { 681797ac58cSKevin Wolf main_loop_wait(false); 682797ac58cSKevin Wolf } 683797ac58cSKevin Wolf 684797ac58cSKevin Wolf *total = qiov->size; 685797ac58cSKevin Wolf return async_ret < 0 ? async_ret : 1; 686797ac58cSKevin Wolf } 687797ac58cSKevin Wolf 6884c7b7e9bSMax Reitz static int do_aio_writev(BlockBackend *blk, QEMUIOVector *qiov, 689770e0e0eSEric Blake int64_t offset, int flags, int *total) 690797ac58cSKevin Wolf { 691797ac58cSKevin Wolf int async_ret = NOT_DONE; 692797ac58cSKevin Wolf 693770e0e0eSEric Blake blk_aio_pwritev(blk, offset, qiov, flags, aio_rw_done, &async_ret); 694797ac58cSKevin Wolf while (async_ret == NOT_DONE) { 695797ac58cSKevin Wolf main_loop_wait(false); 696797ac58cSKevin Wolf } 697797ac58cSKevin Wolf 698797ac58cSKevin Wolf *total = qiov->size; 699797ac58cSKevin Wolf return async_ret < 0 ? async_ret : 1; 700797ac58cSKevin Wolf } 701797ac58cSKevin Wolf 702797ac58cSKevin Wolf static void read_help(void) 703797ac58cSKevin Wolf { 704797ac58cSKevin Wolf printf( 705797ac58cSKevin Wolf "\n" 706797ac58cSKevin Wolf " reads a range of bytes from the given offset\n" 707797ac58cSKevin Wolf "\n" 708797ac58cSKevin Wolf " Example:\n" 709797ac58cSKevin Wolf " 'read -v 512 1k' - dumps 1 kilobyte read from 512 bytes into the file\n" 710797ac58cSKevin Wolf "\n" 711797ac58cSKevin Wolf " Reads a segment of the currently open file, optionally dumping it to the\n" 712797ac58cSKevin Wolf " standard output stream (with -v option) for subsequent inspection.\n" 713797ac58cSKevin Wolf " -b, -- read from the VM state rather than the virtual disk\n" 714797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 715797ac58cSKevin Wolf " -l, -- length for pattern verification (only with -P)\n" 716093ea232SEric Blake " -p, -- ignored for backwards compatibility\n" 717797ac58cSKevin Wolf " -P, -- use a pattern to verify read data\n" 718797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 719797ac58cSKevin Wolf " -s, -- start offset for pattern verification (only with -P)\n" 720797ac58cSKevin Wolf " -v, -- dump buffer to standard output\n" 721797ac58cSKevin Wolf "\n"); 722797ac58cSKevin Wolf } 723797ac58cSKevin Wolf 724b32d7a39SMax Reitz static int read_f(BlockBackend *blk, int argc, char **argv); 725797ac58cSKevin Wolf 726797ac58cSKevin Wolf static const cmdinfo_t read_cmd = { 727797ac58cSKevin Wolf .name = "read", 728797ac58cSKevin Wolf .altname = "r", 729797ac58cSKevin Wolf .cfunc = read_f, 730797ac58cSKevin Wolf .argmin = 2, 731797ac58cSKevin Wolf .argmax = -1, 732093ea232SEric Blake .args = "[-abCqv] [-P pattern [-s off] [-l len]] off len", 733797ac58cSKevin Wolf .oneline = "reads a number of bytes at a specified offset", 734797ac58cSKevin Wolf .help = read_help, 735797ac58cSKevin Wolf }; 736797ac58cSKevin Wolf 737b32d7a39SMax Reitz static int read_f(BlockBackend *blk, int argc, char **argv) 738797ac58cSKevin Wolf { 73950290c00SAlex Bennée struct timespec t1, t2; 740093ea232SEric Blake bool Cflag = false, qflag = false, vflag = false; 741dc38852aSEric Blake bool Pflag = false, sflag = false, lflag = false, bflag = false; 742b32d7a39SMax Reitz int c, cnt, ret; 743797ac58cSKevin Wolf char *buf; 744797ac58cSKevin Wolf int64_t offset; 7459b0beaf3SJohn Snow int64_t count; 746797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 7479b0beaf3SJohn Snow int64_t total = 0; 7489b0beaf3SJohn Snow int pattern = 0; 7499b0beaf3SJohn Snow int64_t pattern_offset = 0, pattern_count = 0; 750797ac58cSKevin Wolf 751b062ad86SEric Blake while ((c = getopt(argc, argv, "bCl:pP:qs:v")) != -1) { 752797ac58cSKevin Wolf switch (c) { 753797ac58cSKevin Wolf case 'b': 754dc38852aSEric Blake bflag = true; 755797ac58cSKevin Wolf break; 756797ac58cSKevin Wolf case 'C': 757dc38852aSEric Blake Cflag = true; 758797ac58cSKevin Wolf break; 759797ac58cSKevin Wolf case 'l': 760dc38852aSEric Blake lflag = true; 761797ac58cSKevin Wolf pattern_count = cvtnum(optarg); 762797ac58cSKevin Wolf if (pattern_count < 0) { 763a9ecfa00SJohn Snow print_cvtnum_err(pattern_count, optarg); 764b32d7a39SMax Reitz return pattern_count; 765797ac58cSKevin Wolf } 766797ac58cSKevin Wolf break; 767797ac58cSKevin Wolf case 'p': 768093ea232SEric Blake /* Ignored for backwards compatibility */ 769797ac58cSKevin Wolf break; 770797ac58cSKevin Wolf case 'P': 771dc38852aSEric Blake Pflag = true; 772797ac58cSKevin Wolf pattern = parse_pattern(optarg); 773797ac58cSKevin Wolf if (pattern < 0) { 774b32d7a39SMax Reitz return -EINVAL; 775797ac58cSKevin Wolf } 776797ac58cSKevin Wolf break; 777797ac58cSKevin Wolf case 'q': 778dc38852aSEric Blake qflag = true; 779797ac58cSKevin Wolf break; 780797ac58cSKevin Wolf case 's': 781dc38852aSEric Blake sflag = true; 782797ac58cSKevin Wolf pattern_offset = cvtnum(optarg); 783797ac58cSKevin Wolf if (pattern_offset < 0) { 784a9ecfa00SJohn Snow print_cvtnum_err(pattern_offset, optarg); 785b32d7a39SMax Reitz return pattern_offset; 786797ac58cSKevin Wolf } 787797ac58cSKevin Wolf break; 788797ac58cSKevin Wolf case 'v': 789dc38852aSEric Blake vflag = true; 790797ac58cSKevin Wolf break; 791797ac58cSKevin Wolf default: 792b444d0e9SMax Reitz qemuio_command_usage(&read_cmd); 793b32d7a39SMax Reitz return -EINVAL; 794797ac58cSKevin Wolf } 795797ac58cSKevin Wolf } 796797ac58cSKevin Wolf 797797ac58cSKevin Wolf if (optind != argc - 2) { 798b444d0e9SMax Reitz qemuio_command_usage(&read_cmd); 799b32d7a39SMax Reitz return -EINVAL; 800797ac58cSKevin Wolf } 801797ac58cSKevin Wolf 802797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 803797ac58cSKevin Wolf if (offset < 0) { 804a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 805b32d7a39SMax Reitz return offset; 806797ac58cSKevin Wolf } 807797ac58cSKevin Wolf 808797ac58cSKevin Wolf optind++; 809797ac58cSKevin Wolf count = cvtnum(argv[optind]); 810797ac58cSKevin Wolf if (count < 0) { 811a9ecfa00SJohn Snow print_cvtnum_err(count, argv[optind]); 812b32d7a39SMax Reitz return count; 8133026c468SAlberto Garcia } else if (count > BDRV_REQUEST_MAX_BYTES) { 8149b0beaf3SJohn Snow printf("length cannot exceed %" PRIu64 ", given %s\n", 8153026c468SAlberto Garcia (uint64_t)BDRV_REQUEST_MAX_BYTES, argv[optind]); 816b32d7a39SMax Reitz return -EINVAL; 817797ac58cSKevin Wolf } 818797ac58cSKevin Wolf 819797ac58cSKevin Wolf if (!Pflag && (lflag || sflag)) { 820b444d0e9SMax Reitz qemuio_command_usage(&read_cmd); 821b32d7a39SMax Reitz return -EINVAL; 822797ac58cSKevin Wolf } 823797ac58cSKevin Wolf 824797ac58cSKevin Wolf if (!lflag) { 825797ac58cSKevin Wolf pattern_count = count - pattern_offset; 826797ac58cSKevin Wolf } 827797ac58cSKevin Wolf 828797ac58cSKevin Wolf if ((pattern_count < 0) || (pattern_count + pattern_offset > count)) { 829797ac58cSKevin Wolf printf("pattern verification range exceeds end of read data\n"); 830b32d7a39SMax Reitz return -EINVAL; 831797ac58cSKevin Wolf } 832797ac58cSKevin Wolf 833093ea232SEric Blake if (bflag) { 8341bce6b4cSEric Blake if (!QEMU_IS_ALIGNED(offset, BDRV_SECTOR_SIZE)) { 8351bce6b4cSEric Blake printf("%" PRId64 " is not a sector-aligned value for 'offset'\n", 836797ac58cSKevin Wolf offset); 837b32d7a39SMax Reitz return -EINVAL; 838797ac58cSKevin Wolf } 8391bce6b4cSEric Blake if (!QEMU_IS_ALIGNED(count, BDRV_SECTOR_SIZE)) { 8401bce6b4cSEric Blake printf("%"PRId64" is not a sector-aligned value for 'count'\n", 841797ac58cSKevin Wolf count); 842b32d7a39SMax Reitz return -EINVAL; 843797ac58cSKevin Wolf } 844797ac58cSKevin Wolf } 845797ac58cSKevin Wolf 8464c7b7e9bSMax Reitz buf = qemu_io_alloc(blk, count, 0xab); 847797ac58cSKevin Wolf 84850290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t1); 8497b3f9712SEric Blake if (bflag) { 850b32d7a39SMax Reitz ret = do_load_vmstate(blk, buf, offset, count, &total); 851797ac58cSKevin Wolf } else { 852b32d7a39SMax Reitz ret = do_pread(blk, buf, offset, count, &total); 853797ac58cSKevin Wolf } 85450290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 855797ac58cSKevin Wolf 856b32d7a39SMax Reitz if (ret < 0) { 857b32d7a39SMax Reitz printf("read failed: %s\n", strerror(-ret)); 858797ac58cSKevin Wolf goto out; 859797ac58cSKevin Wolf } 860b32d7a39SMax Reitz cnt = ret; 861b32d7a39SMax Reitz 862b32d7a39SMax Reitz ret = 0; 863797ac58cSKevin Wolf 864797ac58cSKevin Wolf if (Pflag) { 865797ac58cSKevin Wolf void *cmp_buf = g_malloc(pattern_count); 866797ac58cSKevin Wolf memset(cmp_buf, pattern, pattern_count); 867797ac58cSKevin Wolf if (memcmp(buf + pattern_offset, cmp_buf, pattern_count)) { 868797ac58cSKevin Wolf printf("Pattern verification failed at offset %" 8699b0beaf3SJohn Snow PRId64 ", %"PRId64" bytes\n", 870797ac58cSKevin Wolf offset + pattern_offset, pattern_count); 871b32d7a39SMax Reitz ret = -EINVAL; 872797ac58cSKevin Wolf } 873797ac58cSKevin Wolf g_free(cmp_buf); 874797ac58cSKevin Wolf } 875797ac58cSKevin Wolf 876797ac58cSKevin Wolf if (qflag) { 877797ac58cSKevin Wolf goto out; 878797ac58cSKevin Wolf } 879797ac58cSKevin Wolf 880797ac58cSKevin Wolf if (vflag) { 881797ac58cSKevin Wolf dump_buffer(buf, offset, count); 882797ac58cSKevin Wolf } 883797ac58cSKevin Wolf 884797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 885797ac58cSKevin Wolf t2 = tsub(t2, t1); 886797ac58cSKevin Wolf print_report("read", &t2, offset, count, total, cnt, Cflag); 887797ac58cSKevin Wolf 888797ac58cSKevin Wolf out: 889797ac58cSKevin Wolf qemu_io_free(buf); 890b32d7a39SMax Reitz return ret; 891797ac58cSKevin Wolf } 892797ac58cSKevin Wolf 893797ac58cSKevin Wolf static void readv_help(void) 894797ac58cSKevin Wolf { 895797ac58cSKevin Wolf printf( 896797ac58cSKevin Wolf "\n" 897797ac58cSKevin Wolf " reads a range of bytes from the given offset into multiple buffers\n" 898797ac58cSKevin Wolf "\n" 899797ac58cSKevin Wolf " Example:\n" 900797ac58cSKevin Wolf " 'readv -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n" 901797ac58cSKevin Wolf "\n" 902797ac58cSKevin Wolf " Reads a segment of the currently open file, optionally dumping it to the\n" 903797ac58cSKevin Wolf " standard output stream (with -v option) for subsequent inspection.\n" 904797ac58cSKevin Wolf " Uses multiple iovec buffers if more than one byte range is specified.\n" 905797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 906797ac58cSKevin Wolf " -P, -- use a pattern to verify read data\n" 907797ac58cSKevin Wolf " -v, -- dump buffer to standard output\n" 908797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 909797ac58cSKevin Wolf "\n"); 910797ac58cSKevin Wolf } 911797ac58cSKevin Wolf 912b32d7a39SMax Reitz static int readv_f(BlockBackend *blk, int argc, char **argv); 913797ac58cSKevin Wolf 914797ac58cSKevin Wolf static const cmdinfo_t readv_cmd = { 915797ac58cSKevin Wolf .name = "readv", 916797ac58cSKevin Wolf .cfunc = readv_f, 917797ac58cSKevin Wolf .argmin = 2, 918797ac58cSKevin Wolf .argmax = -1, 919797ac58cSKevin Wolf .args = "[-Cqv] [-P pattern] off len [len..]", 920797ac58cSKevin Wolf .oneline = "reads a number of bytes at a specified offset", 921797ac58cSKevin Wolf .help = readv_help, 922797ac58cSKevin Wolf }; 923797ac58cSKevin Wolf 924b32d7a39SMax Reitz static int readv_f(BlockBackend *blk, int argc, char **argv) 925797ac58cSKevin Wolf { 92650290c00SAlex Bennée struct timespec t1, t2; 927dc38852aSEric Blake bool Cflag = false, qflag = false, vflag = false; 928b32d7a39SMax Reitz int c, cnt, ret; 929797ac58cSKevin Wolf char *buf; 930797ac58cSKevin Wolf int64_t offset; 931797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 932797ac58cSKevin Wolf int total = 0; 933797ac58cSKevin Wolf int nr_iov; 934797ac58cSKevin Wolf QEMUIOVector qiov; 935797ac58cSKevin Wolf int pattern = 0; 936dc38852aSEric Blake bool Pflag = false; 937797ac58cSKevin Wolf 938b062ad86SEric Blake while ((c = getopt(argc, argv, "CP:qv")) != -1) { 939797ac58cSKevin Wolf switch (c) { 940797ac58cSKevin Wolf case 'C': 941dc38852aSEric Blake Cflag = true; 942797ac58cSKevin Wolf break; 943797ac58cSKevin Wolf case 'P': 944dc38852aSEric Blake Pflag = true; 945797ac58cSKevin Wolf pattern = parse_pattern(optarg); 946797ac58cSKevin Wolf if (pattern < 0) { 947b32d7a39SMax Reitz return -EINVAL; 948797ac58cSKevin Wolf } 949797ac58cSKevin Wolf break; 950797ac58cSKevin Wolf case 'q': 951dc38852aSEric Blake qflag = true; 952797ac58cSKevin Wolf break; 953797ac58cSKevin Wolf case 'v': 954dc38852aSEric Blake vflag = true; 955797ac58cSKevin Wolf break; 956797ac58cSKevin Wolf default: 957b444d0e9SMax Reitz qemuio_command_usage(&readv_cmd); 958b32d7a39SMax Reitz return -EINVAL; 959797ac58cSKevin Wolf } 960797ac58cSKevin Wolf } 961797ac58cSKevin Wolf 962797ac58cSKevin Wolf if (optind > argc - 2) { 963b444d0e9SMax Reitz qemuio_command_usage(&readv_cmd); 964b32d7a39SMax Reitz return -EINVAL; 965797ac58cSKevin Wolf } 966797ac58cSKevin Wolf 967797ac58cSKevin Wolf 968797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 969797ac58cSKevin Wolf if (offset < 0) { 970a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 971b32d7a39SMax Reitz return offset; 972797ac58cSKevin Wolf } 973797ac58cSKevin Wolf optind++; 974797ac58cSKevin Wolf 975797ac58cSKevin Wolf nr_iov = argc - optind; 9764c7b7e9bSMax Reitz buf = create_iovec(blk, &qiov, &argv[optind], nr_iov, 0xab); 977797ac58cSKevin Wolf if (buf == NULL) { 978b32d7a39SMax Reitz return -EINVAL; 979797ac58cSKevin Wolf } 980797ac58cSKevin Wolf 98150290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t1); 982b32d7a39SMax Reitz ret = do_aio_readv(blk, &qiov, offset, &total); 98350290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 984797ac58cSKevin Wolf 985b32d7a39SMax Reitz if (ret < 0) { 986b32d7a39SMax Reitz printf("readv failed: %s\n", strerror(-ret)); 987797ac58cSKevin Wolf goto out; 988797ac58cSKevin Wolf } 989b32d7a39SMax Reitz cnt = ret; 990b32d7a39SMax Reitz 991b32d7a39SMax Reitz ret = 0; 992797ac58cSKevin Wolf 993797ac58cSKevin Wolf if (Pflag) { 994797ac58cSKevin Wolf void *cmp_buf = g_malloc(qiov.size); 995797ac58cSKevin Wolf memset(cmp_buf, pattern, qiov.size); 996797ac58cSKevin Wolf if (memcmp(buf, cmp_buf, qiov.size)) { 997797ac58cSKevin Wolf printf("Pattern verification failed at offset %" 998cf67b692SStefan Weil PRId64 ", %zu bytes\n", offset, qiov.size); 999b32d7a39SMax Reitz ret = -EINVAL; 1000797ac58cSKevin Wolf } 1001797ac58cSKevin Wolf g_free(cmp_buf); 1002797ac58cSKevin Wolf } 1003797ac58cSKevin Wolf 1004797ac58cSKevin Wolf if (qflag) { 1005797ac58cSKevin Wolf goto out; 1006797ac58cSKevin Wolf } 1007797ac58cSKevin Wolf 1008797ac58cSKevin Wolf if (vflag) { 1009797ac58cSKevin Wolf dump_buffer(buf, offset, qiov.size); 1010797ac58cSKevin Wolf } 1011797ac58cSKevin Wolf 1012797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1013797ac58cSKevin Wolf t2 = tsub(t2, t1); 1014797ac58cSKevin Wolf print_report("read", &t2, offset, qiov.size, total, cnt, Cflag); 1015797ac58cSKevin Wolf 1016797ac58cSKevin Wolf out: 1017797ac58cSKevin Wolf qemu_iovec_destroy(&qiov); 1018797ac58cSKevin Wolf qemu_io_free(buf); 1019b32d7a39SMax Reitz return ret; 1020797ac58cSKevin Wolf } 1021797ac58cSKevin Wolf 1022797ac58cSKevin Wolf static void write_help(void) 1023797ac58cSKevin Wolf { 1024797ac58cSKevin Wolf printf( 1025797ac58cSKevin Wolf "\n" 1026797ac58cSKevin Wolf " writes a range of bytes from the given offset\n" 1027797ac58cSKevin Wolf "\n" 1028797ac58cSKevin Wolf " Example:\n" 1029797ac58cSKevin Wolf " 'write 512 1k' - writes 1 kilobyte at 512 bytes into the open file\n" 1030797ac58cSKevin Wolf "\n" 1031797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 1032797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd).\n" 1033797ac58cSKevin Wolf " -b, -- write to the VM state rather than the virtual disk\n" 10344c7b7e9bSMax Reitz " -c, -- write compressed data with blk_write_compressed\n" 1035770e0e0eSEric Blake " -f, -- use Force Unit Access semantics\n" 1036c6e3f520SKevin Wolf " -n, -- with -z, don't allow slow fallback\n" 1037093ea232SEric Blake " -p, -- ignored for backwards compatibility\n" 1038797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 10394d731510SDenis Plotnikov " -s, -- use a pattern file to fill the write buffer\n" 1040797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1041797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1042c2e001ccSEric Blake " -u, -- with -z, allow unmapping\n" 1043d004bd52SEric Blake " -z, -- write zeroes using blk_co_pwrite_zeroes\n" 1044797ac58cSKevin Wolf "\n"); 1045797ac58cSKevin Wolf } 1046797ac58cSKevin Wolf 1047b32d7a39SMax Reitz static int write_f(BlockBackend *blk, int argc, char **argv); 1048797ac58cSKevin Wolf 1049797ac58cSKevin Wolf static const cmdinfo_t write_cmd = { 1050797ac58cSKevin Wolf .name = "write", 1051797ac58cSKevin Wolf .altname = "w", 1052797ac58cSKevin Wolf .cfunc = write_f, 1053887354bdSKevin Wolf .perm = BLK_PERM_WRITE, 1054797ac58cSKevin Wolf .argmin = 2, 1055797ac58cSKevin Wolf .argmax = -1, 10564d731510SDenis Plotnikov .args = "[-bcCfnquz] [-P pattern | -s source_file] off len", 1057797ac58cSKevin Wolf .oneline = "writes a number of bytes at a specified offset", 1058797ac58cSKevin Wolf .help = write_help, 1059797ac58cSKevin Wolf }; 1060797ac58cSKevin Wolf 1061b32d7a39SMax Reitz static int write_f(BlockBackend *blk, int argc, char **argv) 1062797ac58cSKevin Wolf { 106350290c00SAlex Bennée struct timespec t1, t2; 1064093ea232SEric Blake bool Cflag = false, qflag = false, bflag = false; 10654d731510SDenis Plotnikov bool Pflag = false, zflag = false, cflag = false, sflag = false; 1066770e0e0eSEric Blake int flags = 0; 1067b32d7a39SMax Reitz int c, cnt, ret; 1068797ac58cSKevin Wolf char *buf = NULL; 1069797ac58cSKevin Wolf int64_t offset; 10709b0beaf3SJohn Snow int64_t count; 1071797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 10729b0beaf3SJohn Snow int64_t total = 0; 1073797ac58cSKevin Wolf int pattern = 0xcd; 10744d731510SDenis Plotnikov const char *file_name = NULL; 1075797ac58cSKevin Wolf 10764d731510SDenis Plotnikov while ((c = getopt(argc, argv, "bcCfnpP:qs:uz")) != -1) { 1077797ac58cSKevin Wolf switch (c) { 1078797ac58cSKevin Wolf case 'b': 1079dc38852aSEric Blake bflag = true; 1080797ac58cSKevin Wolf break; 1081797ac58cSKevin Wolf case 'c': 1082dc38852aSEric Blake cflag = true; 1083797ac58cSKevin Wolf break; 1084797ac58cSKevin Wolf case 'C': 1085dc38852aSEric Blake Cflag = true; 1086797ac58cSKevin Wolf break; 1087770e0e0eSEric Blake case 'f': 1088770e0e0eSEric Blake flags |= BDRV_REQ_FUA; 1089770e0e0eSEric Blake break; 1090c6e3f520SKevin Wolf case 'n': 1091c6e3f520SKevin Wolf flags |= BDRV_REQ_NO_FALLBACK; 1092c6e3f520SKevin Wolf break; 1093797ac58cSKevin Wolf case 'p': 1094093ea232SEric Blake /* Ignored for backwards compatibility */ 1095797ac58cSKevin Wolf break; 1096797ac58cSKevin Wolf case 'P': 1097dc38852aSEric Blake Pflag = true; 1098797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1099797ac58cSKevin Wolf if (pattern < 0) { 1100b32d7a39SMax Reitz return -EINVAL; 1101797ac58cSKevin Wolf } 1102797ac58cSKevin Wolf break; 1103797ac58cSKevin Wolf case 'q': 1104dc38852aSEric Blake qflag = true; 1105797ac58cSKevin Wolf break; 11064d731510SDenis Plotnikov case 's': 11074d731510SDenis Plotnikov sflag = true; 11084d731510SDenis Plotnikov file_name = optarg; 11094d731510SDenis Plotnikov break; 1110c2e001ccSEric Blake case 'u': 1111c2e001ccSEric Blake flags |= BDRV_REQ_MAY_UNMAP; 1112c2e001ccSEric Blake break; 1113797ac58cSKevin Wolf case 'z': 1114dc38852aSEric Blake zflag = true; 1115797ac58cSKevin Wolf break; 1116797ac58cSKevin Wolf default: 1117b444d0e9SMax Reitz qemuio_command_usage(&write_cmd); 1118b32d7a39SMax Reitz return -EINVAL; 1119797ac58cSKevin Wolf } 1120797ac58cSKevin Wolf } 1121797ac58cSKevin Wolf 1122797ac58cSKevin Wolf if (optind != argc - 2) { 1123b444d0e9SMax Reitz qemuio_command_usage(&write_cmd); 1124b32d7a39SMax Reitz return -EINVAL; 1125797ac58cSKevin Wolf } 1126797ac58cSKevin Wolf 1127093ea232SEric Blake if (bflag && zflag) { 1128093ea232SEric Blake printf("-b and -z cannot be specified at the same time\n"); 1129b32d7a39SMax Reitz return -EINVAL; 1130797ac58cSKevin Wolf } 1131797ac58cSKevin Wolf 1132770e0e0eSEric Blake if ((flags & BDRV_REQ_FUA) && (bflag || cflag)) { 1133770e0e0eSEric Blake printf("-f and -b or -c cannot be specified at the same time\n"); 1134b32d7a39SMax Reitz return -EINVAL; 1135770e0e0eSEric Blake } 1136770e0e0eSEric Blake 1137c6e3f520SKevin Wolf if ((flags & BDRV_REQ_NO_FALLBACK) && !zflag) { 1138c6e3f520SKevin Wolf printf("-n requires -z to be specified\n"); 1139c6e3f520SKevin Wolf return -EINVAL; 1140c6e3f520SKevin Wolf } 1141c6e3f520SKevin Wolf 1142c2e001ccSEric Blake if ((flags & BDRV_REQ_MAY_UNMAP) && !zflag) { 1143c2e001ccSEric Blake printf("-u requires -z to be specified\n"); 1144b32d7a39SMax Reitz return -EINVAL; 1145c2e001ccSEric Blake } 1146c2e001ccSEric Blake 11474d731510SDenis Plotnikov if (zflag + Pflag + sflag > 1) { 11484d731510SDenis Plotnikov printf("Only one of -z, -P, and -s " 11494d731510SDenis Plotnikov "can be specified at the same time\n"); 1150b32d7a39SMax Reitz return -EINVAL; 1151797ac58cSKevin Wolf } 1152797ac58cSKevin Wolf 1153797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1154797ac58cSKevin Wolf if (offset < 0) { 1155a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 1156b32d7a39SMax Reitz return offset; 1157797ac58cSKevin Wolf } 1158797ac58cSKevin Wolf 1159797ac58cSKevin Wolf optind++; 1160797ac58cSKevin Wolf count = cvtnum(argv[optind]); 1161797ac58cSKevin Wolf if (count < 0) { 1162a9ecfa00SJohn Snow print_cvtnum_err(count, argv[optind]); 1163b32d7a39SMax Reitz return count; 11643026c468SAlberto Garcia } else if (count > BDRV_REQUEST_MAX_BYTES) { 11659b0beaf3SJohn Snow printf("length cannot exceed %" PRIu64 ", given %s\n", 11663026c468SAlberto Garcia (uint64_t)BDRV_REQUEST_MAX_BYTES, argv[optind]); 1167b32d7a39SMax Reitz return -EINVAL; 1168797ac58cSKevin Wolf } 1169797ac58cSKevin Wolf 1170093ea232SEric Blake if (bflag || cflag) { 11711bce6b4cSEric Blake if (!QEMU_IS_ALIGNED(offset, BDRV_SECTOR_SIZE)) { 11721bce6b4cSEric Blake printf("%" PRId64 " is not a sector-aligned value for 'offset'\n", 1173797ac58cSKevin Wolf offset); 1174b32d7a39SMax Reitz return -EINVAL; 1175797ac58cSKevin Wolf } 1176797ac58cSKevin Wolf 11771bce6b4cSEric Blake if (!QEMU_IS_ALIGNED(count, BDRV_SECTOR_SIZE)) { 11781bce6b4cSEric Blake printf("%"PRId64" is not a sector-aligned value for 'count'\n", 1179797ac58cSKevin Wolf count); 1180b32d7a39SMax Reitz return -EINVAL; 1181797ac58cSKevin Wolf } 1182797ac58cSKevin Wolf } 1183797ac58cSKevin Wolf 1184797ac58cSKevin Wolf if (!zflag) { 11854d731510SDenis Plotnikov if (sflag) { 11864d731510SDenis Plotnikov buf = qemu_io_alloc_from_file(blk, count, file_name); 11874d731510SDenis Plotnikov if (!buf) { 11884d731510SDenis Plotnikov return -EINVAL; 11894d731510SDenis Plotnikov } 11904d731510SDenis Plotnikov } else { 11914c7b7e9bSMax Reitz buf = qemu_io_alloc(blk, count, pattern); 1192797ac58cSKevin Wolf } 11934d731510SDenis Plotnikov } 1194797ac58cSKevin Wolf 119550290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t1); 11967b3f9712SEric Blake if (bflag) { 1197b32d7a39SMax Reitz ret = do_save_vmstate(blk, buf, offset, count, &total); 1198797ac58cSKevin Wolf } else if (zflag) { 1199b32d7a39SMax Reitz ret = do_co_pwrite_zeroes(blk, offset, count, flags, &total); 1200797ac58cSKevin Wolf } else if (cflag) { 1201b32d7a39SMax Reitz ret = do_write_compressed(blk, buf, offset, count, &total); 1202797ac58cSKevin Wolf } else { 1203b32d7a39SMax Reitz ret = do_pwrite(blk, buf, offset, count, flags, &total); 1204797ac58cSKevin Wolf } 120550290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 1206797ac58cSKevin Wolf 1207b32d7a39SMax Reitz if (ret < 0) { 1208b32d7a39SMax Reitz printf("write failed: %s\n", strerror(-ret)); 1209797ac58cSKevin Wolf goto out; 1210797ac58cSKevin Wolf } 1211b32d7a39SMax Reitz cnt = ret; 1212b32d7a39SMax Reitz 1213b32d7a39SMax Reitz ret = 0; 1214797ac58cSKevin Wolf 1215797ac58cSKevin Wolf if (qflag) { 1216797ac58cSKevin Wolf goto out; 1217797ac58cSKevin Wolf } 1218797ac58cSKevin Wolf 1219797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1220797ac58cSKevin Wolf t2 = tsub(t2, t1); 1221797ac58cSKevin Wolf print_report("wrote", &t2, offset, count, total, cnt, Cflag); 1222797ac58cSKevin Wolf 1223797ac58cSKevin Wolf out: 1224797ac58cSKevin Wolf if (!zflag) { 1225797ac58cSKevin Wolf qemu_io_free(buf); 1226797ac58cSKevin Wolf } 1227b32d7a39SMax Reitz return ret; 1228797ac58cSKevin Wolf } 1229797ac58cSKevin Wolf 1230797ac58cSKevin Wolf static void 1231797ac58cSKevin Wolf writev_help(void) 1232797ac58cSKevin Wolf { 1233797ac58cSKevin Wolf printf( 1234797ac58cSKevin Wolf "\n" 1235797ac58cSKevin Wolf " writes a range of bytes from the given offset source from multiple buffers\n" 1236797ac58cSKevin Wolf "\n" 1237797ac58cSKevin Wolf " Example:\n" 12386e6507c0SMaria Kustova " 'writev 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n" 1239797ac58cSKevin Wolf "\n" 1240797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 1241797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd).\n" 1242797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 1243797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1244770e0e0eSEric Blake " -f, -- use Force Unit Access semantics\n" 1245797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1246797ac58cSKevin Wolf "\n"); 1247797ac58cSKevin Wolf } 1248797ac58cSKevin Wolf 1249b32d7a39SMax Reitz static int writev_f(BlockBackend *blk, int argc, char **argv); 1250797ac58cSKevin Wolf 1251797ac58cSKevin Wolf static const cmdinfo_t writev_cmd = { 1252797ac58cSKevin Wolf .name = "writev", 1253797ac58cSKevin Wolf .cfunc = writev_f, 1254887354bdSKevin Wolf .perm = BLK_PERM_WRITE, 1255797ac58cSKevin Wolf .argmin = 2, 1256797ac58cSKevin Wolf .argmax = -1, 1257770e0e0eSEric Blake .args = "[-Cfq] [-P pattern] off len [len..]", 1258797ac58cSKevin Wolf .oneline = "writes a number of bytes at a specified offset", 1259797ac58cSKevin Wolf .help = writev_help, 1260797ac58cSKevin Wolf }; 1261797ac58cSKevin Wolf 1262b32d7a39SMax Reitz static int writev_f(BlockBackend *blk, int argc, char **argv) 1263797ac58cSKevin Wolf { 126450290c00SAlex Bennée struct timespec t1, t2; 1265dc38852aSEric Blake bool Cflag = false, qflag = false; 1266770e0e0eSEric Blake int flags = 0; 1267b32d7a39SMax Reitz int c, cnt, ret; 1268797ac58cSKevin Wolf char *buf; 1269797ac58cSKevin Wolf int64_t offset; 1270797ac58cSKevin Wolf /* Some compilers get confused and warn if this is not initialized. */ 1271797ac58cSKevin Wolf int total = 0; 1272797ac58cSKevin Wolf int nr_iov; 1273797ac58cSKevin Wolf int pattern = 0xcd; 1274797ac58cSKevin Wolf QEMUIOVector qiov; 1275797ac58cSKevin Wolf 12764ca1d340SEric Blake while ((c = getopt(argc, argv, "CfqP:")) != -1) { 1277797ac58cSKevin Wolf switch (c) { 1278797ac58cSKevin Wolf case 'C': 1279dc38852aSEric Blake Cflag = true; 1280797ac58cSKevin Wolf break; 1281770e0e0eSEric Blake case 'f': 1282770e0e0eSEric Blake flags |= BDRV_REQ_FUA; 1283770e0e0eSEric Blake break; 1284797ac58cSKevin Wolf case 'q': 1285dc38852aSEric Blake qflag = true; 1286797ac58cSKevin Wolf break; 1287797ac58cSKevin Wolf case 'P': 1288797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1289797ac58cSKevin Wolf if (pattern < 0) { 1290b32d7a39SMax Reitz return -EINVAL; 1291797ac58cSKevin Wolf } 1292797ac58cSKevin Wolf break; 1293797ac58cSKevin Wolf default: 1294b444d0e9SMax Reitz qemuio_command_usage(&writev_cmd); 1295b32d7a39SMax Reitz return -EINVAL; 1296797ac58cSKevin Wolf } 1297797ac58cSKevin Wolf } 1298797ac58cSKevin Wolf 1299797ac58cSKevin Wolf if (optind > argc - 2) { 1300b444d0e9SMax Reitz qemuio_command_usage(&writev_cmd); 1301b32d7a39SMax Reitz return -EINVAL; 1302797ac58cSKevin Wolf } 1303797ac58cSKevin Wolf 1304797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1305797ac58cSKevin Wolf if (offset < 0) { 1306a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 1307b32d7a39SMax Reitz return offset; 1308797ac58cSKevin Wolf } 1309797ac58cSKevin Wolf optind++; 1310797ac58cSKevin Wolf 1311797ac58cSKevin Wolf nr_iov = argc - optind; 13124c7b7e9bSMax Reitz buf = create_iovec(blk, &qiov, &argv[optind], nr_iov, pattern); 1313797ac58cSKevin Wolf if (buf == NULL) { 1314b32d7a39SMax Reitz return -EINVAL; 1315797ac58cSKevin Wolf } 1316797ac58cSKevin Wolf 131750290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t1); 1318b32d7a39SMax Reitz ret = do_aio_writev(blk, &qiov, offset, flags, &total); 131950290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 1320797ac58cSKevin Wolf 1321b32d7a39SMax Reitz if (ret < 0) { 1322b32d7a39SMax Reitz printf("writev failed: %s\n", strerror(-ret)); 1323797ac58cSKevin Wolf goto out; 1324797ac58cSKevin Wolf } 1325b32d7a39SMax Reitz cnt = ret; 1326b32d7a39SMax Reitz 1327b32d7a39SMax Reitz ret = 0; 1328797ac58cSKevin Wolf 1329797ac58cSKevin Wolf if (qflag) { 1330797ac58cSKevin Wolf goto out; 1331797ac58cSKevin Wolf } 1332797ac58cSKevin Wolf 1333797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1334797ac58cSKevin Wolf t2 = tsub(t2, t1); 1335797ac58cSKevin Wolf print_report("wrote", &t2, offset, qiov.size, total, cnt, Cflag); 1336797ac58cSKevin Wolf out: 1337797ac58cSKevin Wolf qemu_iovec_destroy(&qiov); 1338797ac58cSKevin Wolf qemu_io_free(buf); 1339b32d7a39SMax Reitz return ret; 1340797ac58cSKevin Wolf } 1341797ac58cSKevin Wolf 1342797ac58cSKevin Wolf struct aio_ctx { 13434c7b7e9bSMax Reitz BlockBackend *blk; 1344797ac58cSKevin Wolf QEMUIOVector qiov; 1345797ac58cSKevin Wolf int64_t offset; 1346797ac58cSKevin Wolf char *buf; 1347dc38852aSEric Blake bool qflag; 1348dc38852aSEric Blake bool vflag; 1349dc38852aSEric Blake bool Cflag; 1350dc38852aSEric Blake bool Pflag; 1351dc38852aSEric Blake bool zflag; 1352a91f9584SFam Zheng BlockAcctCookie acct; 1353797ac58cSKevin Wolf int pattern; 135450290c00SAlex Bennée struct timespec t1; 1355797ac58cSKevin Wolf }; 1356797ac58cSKevin Wolf 1357797ac58cSKevin Wolf static void aio_write_done(void *opaque, int ret) 1358797ac58cSKevin Wolf { 1359797ac58cSKevin Wolf struct aio_ctx *ctx = opaque; 136050290c00SAlex Bennée struct timespec t2; 1361797ac58cSKevin Wolf 136250290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 1363797ac58cSKevin Wolf 1364797ac58cSKevin Wolf 1365797ac58cSKevin Wolf if (ret < 0) { 1366797ac58cSKevin Wolf printf("aio_write failed: %s\n", strerror(-ret)); 1367556c2b60SAlberto Garcia block_acct_failed(blk_get_stats(ctx->blk), &ctx->acct); 1368797ac58cSKevin Wolf goto out; 1369797ac58cSKevin Wolf } 1370797ac58cSKevin Wolf 13714c7b7e9bSMax Reitz block_acct_done(blk_get_stats(ctx->blk), &ctx->acct); 1372a91f9584SFam Zheng 1373797ac58cSKevin Wolf if (ctx->qflag) { 1374797ac58cSKevin Wolf goto out; 1375797ac58cSKevin Wolf } 1376797ac58cSKevin Wolf 1377797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1378797ac58cSKevin Wolf t2 = tsub(t2, ctx->t1); 1379797ac58cSKevin Wolf print_report("wrote", &t2, ctx->offset, ctx->qiov.size, 1380797ac58cSKevin Wolf ctx->qiov.size, 1, ctx->Cflag); 1381797ac58cSKevin Wolf out: 13825ceb7765SKevin Wolf if (!ctx->zflag) { 1383797ac58cSKevin Wolf qemu_io_free(ctx->buf); 1384797ac58cSKevin Wolf qemu_iovec_destroy(&ctx->qiov); 13855ceb7765SKevin Wolf } 1386797ac58cSKevin Wolf g_free(ctx); 1387797ac58cSKevin Wolf } 1388797ac58cSKevin Wolf 1389797ac58cSKevin Wolf static void aio_read_done(void *opaque, int ret) 1390797ac58cSKevin Wolf { 1391797ac58cSKevin Wolf struct aio_ctx *ctx = opaque; 139250290c00SAlex Bennée struct timespec t2; 1393797ac58cSKevin Wolf 139450290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 1395797ac58cSKevin Wolf 1396797ac58cSKevin Wolf if (ret < 0) { 1397797ac58cSKevin Wolf printf("readv failed: %s\n", strerror(-ret)); 1398556c2b60SAlberto Garcia block_acct_failed(blk_get_stats(ctx->blk), &ctx->acct); 1399797ac58cSKevin Wolf goto out; 1400797ac58cSKevin Wolf } 1401797ac58cSKevin Wolf 1402797ac58cSKevin Wolf if (ctx->Pflag) { 1403797ac58cSKevin Wolf void *cmp_buf = g_malloc(ctx->qiov.size); 1404797ac58cSKevin Wolf 1405797ac58cSKevin Wolf memset(cmp_buf, ctx->pattern, ctx->qiov.size); 1406797ac58cSKevin Wolf if (memcmp(ctx->buf, cmp_buf, ctx->qiov.size)) { 1407797ac58cSKevin Wolf printf("Pattern verification failed at offset %" 1408cf67b692SStefan Weil PRId64 ", %zu bytes\n", ctx->offset, ctx->qiov.size); 1409797ac58cSKevin Wolf } 1410797ac58cSKevin Wolf g_free(cmp_buf); 1411797ac58cSKevin Wolf } 1412797ac58cSKevin Wolf 14134c7b7e9bSMax Reitz block_acct_done(blk_get_stats(ctx->blk), &ctx->acct); 1414a91f9584SFam Zheng 1415797ac58cSKevin Wolf if (ctx->qflag) { 1416797ac58cSKevin Wolf goto out; 1417797ac58cSKevin Wolf } 1418797ac58cSKevin Wolf 1419797ac58cSKevin Wolf if (ctx->vflag) { 1420797ac58cSKevin Wolf dump_buffer(ctx->buf, ctx->offset, ctx->qiov.size); 1421797ac58cSKevin Wolf } 1422797ac58cSKevin Wolf 1423797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1424797ac58cSKevin Wolf t2 = tsub(t2, ctx->t1); 1425797ac58cSKevin Wolf print_report("read", &t2, ctx->offset, ctx->qiov.size, 1426797ac58cSKevin Wolf ctx->qiov.size, 1, ctx->Cflag); 1427797ac58cSKevin Wolf out: 1428797ac58cSKevin Wolf qemu_io_free(ctx->buf); 1429797ac58cSKevin Wolf qemu_iovec_destroy(&ctx->qiov); 1430797ac58cSKevin Wolf g_free(ctx); 1431797ac58cSKevin Wolf } 1432797ac58cSKevin Wolf 1433797ac58cSKevin Wolf static void aio_read_help(void) 1434797ac58cSKevin Wolf { 1435797ac58cSKevin Wolf printf( 1436797ac58cSKevin Wolf "\n" 1437797ac58cSKevin Wolf " asynchronously reads a range of bytes from the given offset\n" 1438797ac58cSKevin Wolf "\n" 1439797ac58cSKevin Wolf " Example:\n" 1440797ac58cSKevin Wolf " 'aio_read -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n" 1441797ac58cSKevin Wolf "\n" 1442797ac58cSKevin Wolf " Reads a segment of the currently open file, optionally dumping it to the\n" 1443797ac58cSKevin Wolf " standard output stream (with -v option) for subsequent inspection.\n" 1444797ac58cSKevin Wolf " The read is performed asynchronously and the aio_flush command must be\n" 1445797ac58cSKevin Wolf " used to ensure all outstanding aio requests have been completed.\n" 1446b32d7a39SMax Reitz " Note that due to its asynchronous nature, this command will be\n" 1447b32d7a39SMax Reitz " considered successful once the request is submitted, independently\n" 1448b32d7a39SMax Reitz " of potential I/O errors or pattern mismatches.\n" 1449797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1450797ac58cSKevin Wolf " -P, -- use a pattern to verify read data\n" 145137546ff2SEric Blake " -i, -- treat request as invalid, for exercising stats\n" 1452797ac58cSKevin Wolf " -v, -- dump buffer to standard output\n" 1453797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1454797ac58cSKevin Wolf "\n"); 1455797ac58cSKevin Wolf } 1456797ac58cSKevin Wolf 1457b32d7a39SMax Reitz static int aio_read_f(BlockBackend *blk, int argc, char **argv); 1458797ac58cSKevin Wolf 1459797ac58cSKevin Wolf static const cmdinfo_t aio_read_cmd = { 1460797ac58cSKevin Wolf .name = "aio_read", 1461797ac58cSKevin Wolf .cfunc = aio_read_f, 1462797ac58cSKevin Wolf .argmin = 2, 1463797ac58cSKevin Wolf .argmax = -1, 146437546ff2SEric Blake .args = "[-Ciqv] [-P pattern] off len [len..]", 1465797ac58cSKevin Wolf .oneline = "asynchronously reads a number of bytes", 1466797ac58cSKevin Wolf .help = aio_read_help, 1467797ac58cSKevin Wolf }; 1468797ac58cSKevin Wolf 1469b32d7a39SMax Reitz static int aio_read_f(BlockBackend *blk, int argc, char **argv) 1470797ac58cSKevin Wolf { 1471797ac58cSKevin Wolf int nr_iov, c; 1472797ac58cSKevin Wolf struct aio_ctx *ctx = g_new0(struct aio_ctx, 1); 1473797ac58cSKevin Wolf 14744c7b7e9bSMax Reitz ctx->blk = blk; 147537546ff2SEric Blake while ((c = getopt(argc, argv, "CP:iqv")) != -1) { 1476797ac58cSKevin Wolf switch (c) { 1477797ac58cSKevin Wolf case 'C': 1478dc38852aSEric Blake ctx->Cflag = true; 1479797ac58cSKevin Wolf break; 1480797ac58cSKevin Wolf case 'P': 1481dc38852aSEric Blake ctx->Pflag = true; 1482797ac58cSKevin Wolf ctx->pattern = parse_pattern(optarg); 1483797ac58cSKevin Wolf if (ctx->pattern < 0) { 1484797ac58cSKevin Wolf g_free(ctx); 1485b32d7a39SMax Reitz return -EINVAL; 1486797ac58cSKevin Wolf } 1487797ac58cSKevin Wolf break; 148837546ff2SEric Blake case 'i': 148937546ff2SEric Blake printf("injecting invalid read request\n"); 149037546ff2SEric Blake block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_READ); 149137546ff2SEric Blake g_free(ctx); 1492b32d7a39SMax Reitz return 0; 1493797ac58cSKevin Wolf case 'q': 1494dc38852aSEric Blake ctx->qflag = true; 1495797ac58cSKevin Wolf break; 1496797ac58cSKevin Wolf case 'v': 1497dc38852aSEric Blake ctx->vflag = true; 1498797ac58cSKevin Wolf break; 1499797ac58cSKevin Wolf default: 1500797ac58cSKevin Wolf g_free(ctx); 1501b444d0e9SMax Reitz qemuio_command_usage(&aio_read_cmd); 1502b32d7a39SMax Reitz return -EINVAL; 1503797ac58cSKevin Wolf } 1504797ac58cSKevin Wolf } 1505797ac58cSKevin Wolf 1506797ac58cSKevin Wolf if (optind > argc - 2) { 1507797ac58cSKevin Wolf g_free(ctx); 1508b444d0e9SMax Reitz qemuio_command_usage(&aio_read_cmd); 1509b32d7a39SMax Reitz return -EINVAL; 1510797ac58cSKevin Wolf } 1511797ac58cSKevin Wolf 1512797ac58cSKevin Wolf ctx->offset = cvtnum(argv[optind]); 1513797ac58cSKevin Wolf if (ctx->offset < 0) { 1514b32d7a39SMax Reitz int ret = ctx->offset; 1515b32d7a39SMax Reitz print_cvtnum_err(ret, argv[optind]); 1516797ac58cSKevin Wolf g_free(ctx); 1517b32d7a39SMax Reitz return ret; 1518797ac58cSKevin Wolf } 1519797ac58cSKevin Wolf optind++; 1520797ac58cSKevin Wolf 1521797ac58cSKevin Wolf nr_iov = argc - optind; 15224c7b7e9bSMax Reitz ctx->buf = create_iovec(blk, &ctx->qiov, &argv[optind], nr_iov, 0xab); 1523797ac58cSKevin Wolf if (ctx->buf == NULL) { 1524556c2b60SAlberto Garcia block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_READ); 1525797ac58cSKevin Wolf g_free(ctx); 1526b32d7a39SMax Reitz return -EINVAL; 1527797ac58cSKevin Wolf } 1528797ac58cSKevin Wolf 152950290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &ctx->t1); 15304c7b7e9bSMax Reitz block_acct_start(blk_get_stats(blk), &ctx->acct, ctx->qiov.size, 15314c7b7e9bSMax Reitz BLOCK_ACCT_READ); 15327b3f9712SEric Blake blk_aio_preadv(blk, ctx->offset, &ctx->qiov, 0, aio_read_done, ctx); 1533b32d7a39SMax Reitz return 0; 1534797ac58cSKevin Wolf } 1535797ac58cSKevin Wolf 1536797ac58cSKevin Wolf static void aio_write_help(void) 1537797ac58cSKevin Wolf { 1538797ac58cSKevin Wolf printf( 1539797ac58cSKevin Wolf "\n" 1540797ac58cSKevin Wolf " asynchronously writes a range of bytes from the given offset source\n" 1541797ac58cSKevin Wolf " from multiple buffers\n" 1542797ac58cSKevin Wolf "\n" 1543797ac58cSKevin Wolf " Example:\n" 1544797ac58cSKevin Wolf " 'aio_write 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n" 1545797ac58cSKevin Wolf "\n" 1546797ac58cSKevin Wolf " Writes into a segment of the currently open file, using a buffer\n" 1547797ac58cSKevin Wolf " filled with a set pattern (0xcdcdcdcd).\n" 1548797ac58cSKevin Wolf " The write is performed asynchronously and the aio_flush command must be\n" 1549797ac58cSKevin Wolf " used to ensure all outstanding aio requests have been completed.\n" 1550b32d7a39SMax Reitz " Note that due to its asynchronous nature, this command will be\n" 1551b32d7a39SMax Reitz " considered successful once the request is submitted, independently\n" 1552b32d7a39SMax Reitz " of potential I/O errors or pattern mismatches.\n" 1553797ac58cSKevin Wolf " -P, -- use different pattern to fill file\n" 1554797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1555770e0e0eSEric Blake " -f, -- use Force Unit Access semantics\n" 155637546ff2SEric Blake " -i, -- treat request as invalid, for exercising stats\n" 1557797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1558c2e001ccSEric Blake " -u, -- with -z, allow unmapping\n" 1559d004bd52SEric Blake " -z, -- write zeroes using blk_aio_pwrite_zeroes\n" 1560797ac58cSKevin Wolf "\n"); 1561797ac58cSKevin Wolf } 1562797ac58cSKevin Wolf 1563b32d7a39SMax Reitz static int aio_write_f(BlockBackend *blk, int argc, char **argv); 1564797ac58cSKevin Wolf 1565797ac58cSKevin Wolf static const cmdinfo_t aio_write_cmd = { 1566797ac58cSKevin Wolf .name = "aio_write", 1567797ac58cSKevin Wolf .cfunc = aio_write_f, 1568887354bdSKevin Wolf .perm = BLK_PERM_WRITE, 1569797ac58cSKevin Wolf .argmin = 2, 1570797ac58cSKevin Wolf .argmax = -1, 157137546ff2SEric Blake .args = "[-Cfiquz] [-P pattern] off len [len..]", 1572797ac58cSKevin Wolf .oneline = "asynchronously writes a number of bytes", 1573797ac58cSKevin Wolf .help = aio_write_help, 1574797ac58cSKevin Wolf }; 1575797ac58cSKevin Wolf 1576b32d7a39SMax Reitz static int aio_write_f(BlockBackend *blk, int argc, char **argv) 1577797ac58cSKevin Wolf { 1578797ac58cSKevin Wolf int nr_iov, c; 1579797ac58cSKevin Wolf int pattern = 0xcd; 1580797ac58cSKevin Wolf struct aio_ctx *ctx = g_new0(struct aio_ctx, 1); 1581770e0e0eSEric Blake int flags = 0; 1582797ac58cSKevin Wolf 15834c7b7e9bSMax Reitz ctx->blk = blk; 158437546ff2SEric Blake while ((c = getopt(argc, argv, "CfiqP:uz")) != -1) { 1585797ac58cSKevin Wolf switch (c) { 1586797ac58cSKevin Wolf case 'C': 1587dc38852aSEric Blake ctx->Cflag = true; 1588797ac58cSKevin Wolf break; 1589770e0e0eSEric Blake case 'f': 1590770e0e0eSEric Blake flags |= BDRV_REQ_FUA; 1591770e0e0eSEric Blake break; 1592797ac58cSKevin Wolf case 'q': 1593dc38852aSEric Blake ctx->qflag = true; 1594797ac58cSKevin Wolf break; 1595c2e001ccSEric Blake case 'u': 1596c2e001ccSEric Blake flags |= BDRV_REQ_MAY_UNMAP; 1597c2e001ccSEric Blake break; 1598797ac58cSKevin Wolf case 'P': 1599797ac58cSKevin Wolf pattern = parse_pattern(optarg); 1600797ac58cSKevin Wolf if (pattern < 0) { 1601797ac58cSKevin Wolf g_free(ctx); 1602b32d7a39SMax Reitz return -EINVAL; 1603797ac58cSKevin Wolf } 1604797ac58cSKevin Wolf break; 160537546ff2SEric Blake case 'i': 160637546ff2SEric Blake printf("injecting invalid write request\n"); 160737546ff2SEric Blake block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_WRITE); 160837546ff2SEric Blake g_free(ctx); 1609b32d7a39SMax Reitz return 0; 16105ceb7765SKevin Wolf case 'z': 1611dc38852aSEric Blake ctx->zflag = true; 16125ceb7765SKevin Wolf break; 1613797ac58cSKevin Wolf default: 1614797ac58cSKevin Wolf g_free(ctx); 1615b444d0e9SMax Reitz qemuio_command_usage(&aio_write_cmd); 1616b32d7a39SMax Reitz return -EINVAL; 1617797ac58cSKevin Wolf } 1618797ac58cSKevin Wolf } 1619797ac58cSKevin Wolf 1620797ac58cSKevin Wolf if (optind > argc - 2) { 1621797ac58cSKevin Wolf g_free(ctx); 1622b444d0e9SMax Reitz qemuio_command_usage(&aio_write_cmd); 1623b32d7a39SMax Reitz return -EINVAL; 1624797ac58cSKevin Wolf } 1625797ac58cSKevin Wolf 16265ceb7765SKevin Wolf if (ctx->zflag && optind != argc - 2) { 16275ceb7765SKevin Wolf printf("-z supports only a single length parameter\n"); 16285ceb7765SKevin Wolf g_free(ctx); 1629b32d7a39SMax Reitz return -EINVAL; 16305ceb7765SKevin Wolf } 16315ceb7765SKevin Wolf 1632c2e001ccSEric Blake if ((flags & BDRV_REQ_MAY_UNMAP) && !ctx->zflag) { 1633c2e001ccSEric Blake printf("-u requires -z to be specified\n"); 16344ca1d340SEric Blake g_free(ctx); 1635b32d7a39SMax Reitz return -EINVAL; 1636c2e001ccSEric Blake } 1637c2e001ccSEric Blake 16385ceb7765SKevin Wolf if (ctx->zflag && ctx->Pflag) { 16395ceb7765SKevin Wolf printf("-z and -P cannot be specified at the same time\n"); 16405ceb7765SKevin Wolf g_free(ctx); 1641b32d7a39SMax Reitz return -EINVAL; 16425ceb7765SKevin Wolf } 16435ceb7765SKevin Wolf 1644797ac58cSKevin Wolf ctx->offset = cvtnum(argv[optind]); 1645797ac58cSKevin Wolf if (ctx->offset < 0) { 1646b32d7a39SMax Reitz int ret = ctx->offset; 1647b32d7a39SMax Reitz print_cvtnum_err(ret, argv[optind]); 1648797ac58cSKevin Wolf g_free(ctx); 1649b32d7a39SMax Reitz return ret; 1650797ac58cSKevin Wolf } 1651797ac58cSKevin Wolf optind++; 1652797ac58cSKevin Wolf 16535ceb7765SKevin Wolf if (ctx->zflag) { 16545ceb7765SKevin Wolf int64_t count = cvtnum(argv[optind]); 16555ceb7765SKevin Wolf if (count < 0) { 16565ceb7765SKevin Wolf print_cvtnum_err(count, argv[optind]); 16570e01b76eSKevin Wolf g_free(ctx); 1658b32d7a39SMax Reitz return count; 16595ceb7765SKevin Wolf } 16605ceb7765SKevin Wolf 16615ceb7765SKevin Wolf ctx->qiov.size = count; 1662d004bd52SEric Blake blk_aio_pwrite_zeroes(blk, ctx->offset, count, flags, aio_write_done, 1663770e0e0eSEric Blake ctx); 16645ceb7765SKevin Wolf } else { 1665797ac58cSKevin Wolf nr_iov = argc - optind; 16665ceb7765SKevin Wolf ctx->buf = create_iovec(blk, &ctx->qiov, &argv[optind], nr_iov, 16675ceb7765SKevin Wolf pattern); 1668797ac58cSKevin Wolf if (ctx->buf == NULL) { 1669556c2b60SAlberto Garcia block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_WRITE); 1670797ac58cSKevin Wolf g_free(ctx); 1671b32d7a39SMax Reitz return -EINVAL; 1672797ac58cSKevin Wolf } 1673797ac58cSKevin Wolf 167450290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &ctx->t1); 16754c7b7e9bSMax Reitz block_acct_start(blk_get_stats(blk), &ctx->acct, ctx->qiov.size, 16764c7b7e9bSMax Reitz BLOCK_ACCT_WRITE); 16775ceb7765SKevin Wolf 1678770e0e0eSEric Blake blk_aio_pwritev(blk, ctx->offset, &ctx->qiov, flags, aio_write_done, 1679770e0e0eSEric Blake ctx); 16805ceb7765SKevin Wolf } 1681b32d7a39SMax Reitz 1682b32d7a39SMax Reitz return 0; 1683797ac58cSKevin Wolf } 1684797ac58cSKevin Wolf 1685b32d7a39SMax Reitz static int aio_flush_f(BlockBackend *blk, int argc, char **argv) 1686797ac58cSKevin Wolf { 1687556c2b60SAlberto Garcia BlockAcctCookie cookie; 1688556c2b60SAlberto Garcia block_acct_start(blk_get_stats(blk), &cookie, 0, BLOCK_ACCT_FLUSH); 16894c7b7e9bSMax Reitz blk_drain_all(); 1690556c2b60SAlberto Garcia block_acct_done(blk_get_stats(blk), &cookie); 1691b32d7a39SMax Reitz return 0; 1692797ac58cSKevin Wolf } 1693797ac58cSKevin Wolf 1694797ac58cSKevin Wolf static const cmdinfo_t aio_flush_cmd = { 1695797ac58cSKevin Wolf .name = "aio_flush", 1696797ac58cSKevin Wolf .cfunc = aio_flush_f, 1697797ac58cSKevin Wolf .oneline = "completes all outstanding aio requests" 1698797ac58cSKevin Wolf }; 1699797ac58cSKevin Wolf 1700b32d7a39SMax Reitz static int flush_f(BlockBackend *blk, int argc, char **argv) 1701797ac58cSKevin Wolf { 1702b32d7a39SMax Reitz return blk_flush(blk); 1703797ac58cSKevin Wolf } 1704797ac58cSKevin Wolf 1705797ac58cSKevin Wolf static const cmdinfo_t flush_cmd = { 1706797ac58cSKevin Wolf .name = "flush", 1707797ac58cSKevin Wolf .altname = "f", 1708797ac58cSKevin Wolf .cfunc = flush_f, 1709797ac58cSKevin Wolf .oneline = "flush all in-core file state to disk", 1710797ac58cSKevin Wolf }; 1711797ac58cSKevin Wolf 171242ba0225SVladimir Sementsov-Ogievskiy static int truncate_f(BlockBackend *blk, int argc, char **argv); 171342ba0225SVladimir Sementsov-Ogievskiy static const cmdinfo_t truncate_cmd = { 171442ba0225SVladimir Sementsov-Ogievskiy .name = "truncate", 171542ba0225SVladimir Sementsov-Ogievskiy .altname = "t", 171642ba0225SVladimir Sementsov-Ogievskiy .cfunc = truncate_f, 171742ba0225SVladimir Sementsov-Ogievskiy .perm = BLK_PERM_WRITE | BLK_PERM_RESIZE, 171842ba0225SVladimir Sementsov-Ogievskiy .argmin = 1, 171942ba0225SVladimir Sementsov-Ogievskiy .argmax = 3, 172042ba0225SVladimir Sementsov-Ogievskiy .args = "[-m prealloc_mode] off", 172142ba0225SVladimir Sementsov-Ogievskiy .oneline = "truncates the current file at the given offset", 172242ba0225SVladimir Sementsov-Ogievskiy }; 172342ba0225SVladimir Sementsov-Ogievskiy 1724b32d7a39SMax Reitz static int truncate_f(BlockBackend *blk, int argc, char **argv) 1725797ac58cSKevin Wolf { 1726ed3d2ec9SMax Reitz Error *local_err = NULL; 1727797ac58cSKevin Wolf int64_t offset; 172842ba0225SVladimir Sementsov-Ogievskiy int c, ret; 172942ba0225SVladimir Sementsov-Ogievskiy PreallocMode prealloc = PREALLOC_MODE_OFF; 1730797ac58cSKevin Wolf 173142ba0225SVladimir Sementsov-Ogievskiy while ((c = getopt(argc, argv, "m:")) != -1) { 173242ba0225SVladimir Sementsov-Ogievskiy switch (c) { 173342ba0225SVladimir Sementsov-Ogievskiy case 'm': 173442ba0225SVladimir Sementsov-Ogievskiy prealloc = qapi_enum_parse(&PreallocMode_lookup, optarg, 173542ba0225SVladimir Sementsov-Ogievskiy PREALLOC_MODE__MAX, NULL); 173642ba0225SVladimir Sementsov-Ogievskiy if (prealloc == PREALLOC_MODE__MAX) { 173742ba0225SVladimir Sementsov-Ogievskiy error_report("Invalid preallocation mode '%s'", optarg); 173842ba0225SVladimir Sementsov-Ogievskiy return -EINVAL; 173942ba0225SVladimir Sementsov-Ogievskiy } 174042ba0225SVladimir Sementsov-Ogievskiy break; 174142ba0225SVladimir Sementsov-Ogievskiy default: 174242ba0225SVladimir Sementsov-Ogievskiy qemuio_command_usage(&truncate_cmd); 174342ba0225SVladimir Sementsov-Ogievskiy return -EINVAL; 174442ba0225SVladimir Sementsov-Ogievskiy } 174542ba0225SVladimir Sementsov-Ogievskiy } 174642ba0225SVladimir Sementsov-Ogievskiy 174742ba0225SVladimir Sementsov-Ogievskiy offset = cvtnum(argv[optind]); 1748797ac58cSKevin Wolf if (offset < 0) { 1749a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[1]); 1750b32d7a39SMax Reitz return offset; 1751797ac58cSKevin Wolf } 1752797ac58cSKevin Wolf 1753e8d04f92SMax Reitz /* 1754e8d04f92SMax Reitz * qemu-io is a debugging tool, so let us be strict here and pass 1755e8d04f92SMax Reitz * exact=true. It is better to err on the "emit more errors" side 1756e8d04f92SMax Reitz * than to be overly permissive. 1757e8d04f92SMax Reitz */ 175842ba0225SVladimir Sementsov-Ogievskiy ret = blk_truncate(blk, offset, false, prealloc, 0, &local_err); 1759797ac58cSKevin Wolf if (ret < 0) { 1760ed3d2ec9SMax Reitz error_report_err(local_err); 1761b32d7a39SMax Reitz return ret; 1762797ac58cSKevin Wolf } 1763b32d7a39SMax Reitz 1764b32d7a39SMax Reitz return 0; 1765797ac58cSKevin Wolf } 1766797ac58cSKevin Wolf 1767b32d7a39SMax Reitz static int length_f(BlockBackend *blk, int argc, char **argv) 1768797ac58cSKevin Wolf { 1769797ac58cSKevin Wolf int64_t size; 1770797ac58cSKevin Wolf char s1[64]; 1771797ac58cSKevin Wolf 17724c7b7e9bSMax Reitz size = blk_getlength(blk); 1773797ac58cSKevin Wolf if (size < 0) { 1774797ac58cSKevin Wolf printf("getlength: %s\n", strerror(-size)); 1775b32d7a39SMax Reitz return size; 1776797ac58cSKevin Wolf } 1777797ac58cSKevin Wolf 1778797ac58cSKevin Wolf cvtstr(size, s1, sizeof(s1)); 1779797ac58cSKevin Wolf printf("%s\n", s1); 1780b32d7a39SMax Reitz return 0; 1781797ac58cSKevin Wolf } 1782797ac58cSKevin Wolf 1783797ac58cSKevin Wolf 1784797ac58cSKevin Wolf static const cmdinfo_t length_cmd = { 1785797ac58cSKevin Wolf .name = "length", 1786797ac58cSKevin Wolf .altname = "l", 1787797ac58cSKevin Wolf .cfunc = length_f, 1788797ac58cSKevin Wolf .oneline = "gets the length of the current file", 1789797ac58cSKevin Wolf }; 1790797ac58cSKevin Wolf 1791797ac58cSKevin Wolf 1792b32d7a39SMax Reitz static int info_f(BlockBackend *blk, int argc, char **argv) 1793797ac58cSKevin Wolf { 17944c7b7e9bSMax Reitz BlockDriverState *bs = blk_bs(blk); 1795797ac58cSKevin Wolf BlockDriverInfo bdi; 1796a8d8ecb7SMax Reitz ImageInfoSpecific *spec_info; 17971bf6e9caSAndrey Shinkevich Error *local_err = NULL; 1798797ac58cSKevin Wolf char s1[64], s2[64]; 1799797ac58cSKevin Wolf int ret; 1800797ac58cSKevin Wolf 1801797ac58cSKevin Wolf if (bs->drv && bs->drv->format_name) { 1802797ac58cSKevin Wolf printf("format name: %s\n", bs->drv->format_name); 1803797ac58cSKevin Wolf } 1804797ac58cSKevin Wolf if (bs->drv && bs->drv->protocol_name) { 1805797ac58cSKevin Wolf printf("format name: %s\n", bs->drv->protocol_name); 1806797ac58cSKevin Wolf } 1807797ac58cSKevin Wolf 1808797ac58cSKevin Wolf ret = bdrv_get_info(bs, &bdi); 1809797ac58cSKevin Wolf if (ret) { 1810b32d7a39SMax Reitz return ret; 1811797ac58cSKevin Wolf } 1812797ac58cSKevin Wolf 1813797ac58cSKevin Wolf cvtstr(bdi.cluster_size, s1, sizeof(s1)); 1814797ac58cSKevin Wolf cvtstr(bdi.vm_state_offset, s2, sizeof(s2)); 1815797ac58cSKevin Wolf 1816797ac58cSKevin Wolf printf("cluster size: %s\n", s1); 1817797ac58cSKevin Wolf printf("vm state offset: %s\n", s2); 1818797ac58cSKevin Wolf 18191bf6e9caSAndrey Shinkevich spec_info = bdrv_get_specific_info(bs, &local_err); 18201bf6e9caSAndrey Shinkevich if (local_err) { 18211bf6e9caSAndrey Shinkevich error_report_err(local_err); 18221bf6e9caSAndrey Shinkevich return -EIO; 18231bf6e9caSAndrey Shinkevich } 1824a8d8ecb7SMax Reitz if (spec_info) { 1825a8d8ecb7SMax Reitz printf("Format specific information:\n"); 1826e1ce7d74SMarkus Armbruster bdrv_image_info_specific_dump(spec_info); 1827a8d8ecb7SMax Reitz qapi_free_ImageInfoSpecific(spec_info); 1828a8d8ecb7SMax Reitz } 1829b32d7a39SMax Reitz 1830b32d7a39SMax Reitz return 0; 1831797ac58cSKevin Wolf } 1832797ac58cSKevin Wolf 1833797ac58cSKevin Wolf 1834797ac58cSKevin Wolf 1835797ac58cSKevin Wolf static const cmdinfo_t info_cmd = { 1836797ac58cSKevin Wolf .name = "info", 1837797ac58cSKevin Wolf .altname = "i", 1838797ac58cSKevin Wolf .cfunc = info_f, 1839797ac58cSKevin Wolf .oneline = "prints information about the current file", 1840797ac58cSKevin Wolf }; 1841797ac58cSKevin Wolf 1842797ac58cSKevin Wolf static void discard_help(void) 1843797ac58cSKevin Wolf { 1844797ac58cSKevin Wolf printf( 1845797ac58cSKevin Wolf "\n" 1846797ac58cSKevin Wolf " discards a range of bytes from the given offset\n" 1847797ac58cSKevin Wolf "\n" 1848797ac58cSKevin Wolf " Example:\n" 1849797ac58cSKevin Wolf " 'discard 512 1k' - discards 1 kilobyte from 512 bytes into the file\n" 1850797ac58cSKevin Wolf "\n" 1851797ac58cSKevin Wolf " Discards a segment of the currently open file.\n" 1852797ac58cSKevin Wolf " -C, -- report statistics in a machine parsable format\n" 1853797ac58cSKevin Wolf " -q, -- quiet mode, do not show I/O statistics\n" 1854797ac58cSKevin Wolf "\n"); 1855797ac58cSKevin Wolf } 1856797ac58cSKevin Wolf 1857b32d7a39SMax Reitz static int discard_f(BlockBackend *blk, int argc, char **argv); 1858797ac58cSKevin Wolf 1859797ac58cSKevin Wolf static const cmdinfo_t discard_cmd = { 1860797ac58cSKevin Wolf .name = "discard", 1861797ac58cSKevin Wolf .altname = "d", 1862797ac58cSKevin Wolf .cfunc = discard_f, 1863887354bdSKevin Wolf .perm = BLK_PERM_WRITE, 1864797ac58cSKevin Wolf .argmin = 2, 1865797ac58cSKevin Wolf .argmax = -1, 1866797ac58cSKevin Wolf .args = "[-Cq] off len", 1867797ac58cSKevin Wolf .oneline = "discards a number of bytes at a specified offset", 1868797ac58cSKevin Wolf .help = discard_help, 1869797ac58cSKevin Wolf }; 1870797ac58cSKevin Wolf 1871b32d7a39SMax Reitz static int discard_f(BlockBackend *blk, int argc, char **argv) 1872797ac58cSKevin Wolf { 187350290c00SAlex Bennée struct timespec t1, t2; 1874dc38852aSEric Blake bool Cflag = false, qflag = false; 1875797ac58cSKevin Wolf int c, ret; 1876f5a5ca79SManos Pitsidianakis int64_t offset, bytes; 1877797ac58cSKevin Wolf 1878b062ad86SEric Blake while ((c = getopt(argc, argv, "Cq")) != -1) { 1879797ac58cSKevin Wolf switch (c) { 1880797ac58cSKevin Wolf case 'C': 1881dc38852aSEric Blake Cflag = true; 1882797ac58cSKevin Wolf break; 1883797ac58cSKevin Wolf case 'q': 1884dc38852aSEric Blake qflag = true; 1885797ac58cSKevin Wolf break; 1886797ac58cSKevin Wolf default: 1887b444d0e9SMax Reitz qemuio_command_usage(&discard_cmd); 1888b32d7a39SMax Reitz return -EINVAL; 1889797ac58cSKevin Wolf } 1890797ac58cSKevin Wolf } 1891797ac58cSKevin Wolf 1892797ac58cSKevin Wolf if (optind != argc - 2) { 1893b444d0e9SMax Reitz qemuio_command_usage(&discard_cmd); 1894b32d7a39SMax Reitz return -EINVAL; 1895797ac58cSKevin Wolf } 1896797ac58cSKevin Wolf 1897797ac58cSKevin Wolf offset = cvtnum(argv[optind]); 1898797ac58cSKevin Wolf if (offset < 0) { 1899a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[optind]); 1900b32d7a39SMax Reitz return offset; 1901797ac58cSKevin Wolf } 1902797ac58cSKevin Wolf 1903797ac58cSKevin Wolf optind++; 1904f5a5ca79SManos Pitsidianakis bytes = cvtnum(argv[optind]); 1905f5a5ca79SManos Pitsidianakis if (bytes < 0) { 1906f5a5ca79SManos Pitsidianakis print_cvtnum_err(bytes, argv[optind]); 1907b32d7a39SMax Reitz return bytes; 190841ae31e3SAlberto Garcia } else if (bytes > BDRV_REQUEST_MAX_BYTES) { 19099b0beaf3SJohn Snow printf("length cannot exceed %"PRIu64", given %s\n", 191041ae31e3SAlberto Garcia (uint64_t)BDRV_REQUEST_MAX_BYTES, argv[optind]); 1911b32d7a39SMax Reitz return -EINVAL; 1912797ac58cSKevin Wolf } 1913797ac58cSKevin Wolf 191450290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t1); 1915f5a5ca79SManos Pitsidianakis ret = blk_pdiscard(blk, offset, bytes); 191650290c00SAlex Bennée clock_gettime(CLOCK_MONOTONIC, &t2); 1917797ac58cSKevin Wolf 1918797ac58cSKevin Wolf if (ret < 0) { 1919797ac58cSKevin Wolf printf("discard failed: %s\n", strerror(-ret)); 1920b32d7a39SMax Reitz return ret; 1921797ac58cSKevin Wolf } 1922797ac58cSKevin Wolf 1923797ac58cSKevin Wolf /* Finally, report back -- -C gives a parsable format */ 1924797ac58cSKevin Wolf if (!qflag) { 1925797ac58cSKevin Wolf t2 = tsub(t2, t1); 1926f5a5ca79SManos Pitsidianakis print_report("discard", &t2, offset, bytes, bytes, 1, Cflag); 1927797ac58cSKevin Wolf } 1928b32d7a39SMax Reitz 1929b32d7a39SMax Reitz return 0; 1930797ac58cSKevin Wolf } 1931797ac58cSKevin Wolf 1932b32d7a39SMax Reitz static int alloc_f(BlockBackend *blk, int argc, char **argv) 1933797ac58cSKevin Wolf { 19344c7b7e9bSMax Reitz BlockDriverState *bs = blk_bs(blk); 1935d6a644bbSEric Blake int64_t offset, start, remaining, count; 1936797ac58cSKevin Wolf char s1[64]; 1937d6a644bbSEric Blake int ret; 1938d6a644bbSEric Blake int64_t num, sum_alloc; 1939797ac58cSKevin Wolf 1940d6a644bbSEric Blake start = offset = cvtnum(argv[1]); 1941797ac58cSKevin Wolf if (offset < 0) { 1942a9ecfa00SJohn Snow print_cvtnum_err(offset, argv[1]); 1943b32d7a39SMax Reitz return offset; 1944797ac58cSKevin Wolf } 1945797ac58cSKevin Wolf 1946797ac58cSKevin Wolf if (argc == 3) { 19474401fdc7SEric Blake count = cvtnum(argv[2]); 19484401fdc7SEric Blake if (count < 0) { 19494401fdc7SEric Blake print_cvtnum_err(count, argv[2]); 1950b32d7a39SMax Reitz return count; 1951797ac58cSKevin Wolf } 1952797ac58cSKevin Wolf } else { 19534401fdc7SEric Blake count = BDRV_SECTOR_SIZE; 1954797ac58cSKevin Wolf } 1955797ac58cSKevin Wolf 1956d6a644bbSEric Blake remaining = count; 1957797ac58cSKevin Wolf sum_alloc = 0; 1958797ac58cSKevin Wolf while (remaining) { 1959d6a644bbSEric Blake ret = bdrv_is_allocated(bs, offset, remaining, &num); 1960d663640cSPaolo Bonzini if (ret < 0) { 1961d663640cSPaolo Bonzini printf("is_allocated failed: %s\n", strerror(-ret)); 1962b32d7a39SMax Reitz return ret; 1963d663640cSPaolo Bonzini } 1964d6a644bbSEric Blake offset += num; 1965797ac58cSKevin Wolf remaining -= num; 1966797ac58cSKevin Wolf if (ret) { 1967797ac58cSKevin Wolf sum_alloc += num; 1968797ac58cSKevin Wolf } 1969797ac58cSKevin Wolf if (num == 0) { 1970d6a644bbSEric Blake count -= remaining; 1971797ac58cSKevin Wolf remaining = 0; 1972797ac58cSKevin Wolf } 1973797ac58cSKevin Wolf } 1974797ac58cSKevin Wolf 1975d6a644bbSEric Blake cvtstr(start, s1, sizeof(s1)); 1976797ac58cSKevin Wolf 19774401fdc7SEric Blake printf("%"PRId64"/%"PRId64" bytes allocated at offset %s\n", 1978d6a644bbSEric Blake sum_alloc, count, s1); 1979b32d7a39SMax Reitz return 0; 1980797ac58cSKevin Wolf } 1981797ac58cSKevin Wolf 1982797ac58cSKevin Wolf static const cmdinfo_t alloc_cmd = { 1983797ac58cSKevin Wolf .name = "alloc", 1984797ac58cSKevin Wolf .altname = "a", 1985797ac58cSKevin Wolf .argmin = 1, 1986797ac58cSKevin Wolf .argmax = 2, 1987797ac58cSKevin Wolf .cfunc = alloc_f, 19884401fdc7SEric Blake .args = "offset [count]", 19894401fdc7SEric Blake .oneline = "checks if offset is allocated in the file", 1990797ac58cSKevin Wolf }; 1991797ac58cSKevin Wolf 1992797ac58cSKevin Wolf 1993d6a644bbSEric Blake static int map_is_allocated(BlockDriverState *bs, int64_t offset, 1994d6a644bbSEric Blake int64_t bytes, int64_t *pnum) 1995797ac58cSKevin Wolf { 1996d6a644bbSEric Blake int64_t num; 1997d6a644bbSEric Blake int num_checked; 1998797ac58cSKevin Wolf int ret, firstret; 1999797ac58cSKevin Wolf 2000d6a644bbSEric Blake num_checked = MIN(bytes, BDRV_REQUEST_MAX_BYTES); 2001d6a644bbSEric Blake ret = bdrv_is_allocated(bs, offset, num_checked, &num); 2002797ac58cSKevin Wolf if (ret < 0) { 2003797ac58cSKevin Wolf return ret; 2004797ac58cSKevin Wolf } 2005797ac58cSKevin Wolf 2006797ac58cSKevin Wolf firstret = ret; 2007797ac58cSKevin Wolf *pnum = num; 2008797ac58cSKevin Wolf 2009d6a644bbSEric Blake while (bytes > 0 && ret == firstret) { 2010d6a644bbSEric Blake offset += num; 2011d6a644bbSEric Blake bytes -= num; 2012797ac58cSKevin Wolf 2013d6a644bbSEric Blake num_checked = MIN(bytes, BDRV_REQUEST_MAX_BYTES); 2014d6a644bbSEric Blake ret = bdrv_is_allocated(bs, offset, num_checked, &num); 20154b25bbc4SMax Reitz if (ret == firstret && num) { 2016797ac58cSKevin Wolf *pnum += num; 2017797ac58cSKevin Wolf } else { 2018797ac58cSKevin Wolf break; 2019797ac58cSKevin Wolf } 2020797ac58cSKevin Wolf } 2021797ac58cSKevin Wolf 2022797ac58cSKevin Wolf return firstret; 2023797ac58cSKevin Wolf } 2024797ac58cSKevin Wolf 2025b32d7a39SMax Reitz static int map_f(BlockBackend *blk, int argc, char **argv) 2026797ac58cSKevin Wolf { 2027d6a644bbSEric Blake int64_t offset, bytes; 20286f3c90afSEric Blake char s1[64], s2[64]; 2029797ac58cSKevin Wolf int64_t num; 2030797ac58cSKevin Wolf int ret; 2031797ac58cSKevin Wolf const char *retstr; 2032797ac58cSKevin Wolf 2033797ac58cSKevin Wolf offset = 0; 2034d6a644bbSEric Blake bytes = blk_getlength(blk); 2035d6a644bbSEric Blake if (bytes < 0) { 2036d6a644bbSEric Blake error_report("Failed to query image length: %s", strerror(-bytes)); 2037b32d7a39SMax Reitz return bytes; 20384c7b7e9bSMax Reitz } 20394c7b7e9bSMax Reitz 2040d6a644bbSEric Blake while (bytes) { 2041d6a644bbSEric Blake ret = map_is_allocated(blk_bs(blk), offset, bytes, &num); 2042797ac58cSKevin Wolf if (ret < 0) { 2043797ac58cSKevin Wolf error_report("Failed to get allocation status: %s", strerror(-ret)); 2044b32d7a39SMax Reitz return ret; 20454b25bbc4SMax Reitz } else if (!num) { 20464b25bbc4SMax Reitz error_report("Unexpected end of image"); 2047b32d7a39SMax Reitz return -EIO; 2048797ac58cSKevin Wolf } 2049797ac58cSKevin Wolf 2050797ac58cSKevin Wolf retstr = ret ? " allocated" : "not allocated"; 2051d6a644bbSEric Blake cvtstr(num, s1, sizeof(s1)); 2052d6a644bbSEric Blake cvtstr(offset, s2, sizeof(s2)); 20536f3c90afSEric Blake printf("%s (0x%" PRIx64 ") bytes %s at offset %s (0x%" PRIx64 ")\n", 2054d6a644bbSEric Blake s1, num, retstr, s2, offset); 2055797ac58cSKevin Wolf 2056797ac58cSKevin Wolf offset += num; 2057d6a644bbSEric Blake bytes -= num; 2058d6a644bbSEric Blake } 2059b32d7a39SMax Reitz 2060b32d7a39SMax Reitz return 0; 2061797ac58cSKevin Wolf } 2062797ac58cSKevin Wolf 2063797ac58cSKevin Wolf static const cmdinfo_t map_cmd = { 2064797ac58cSKevin Wolf .name = "map", 2065797ac58cSKevin Wolf .argmin = 0, 2066797ac58cSKevin Wolf .argmax = 0, 2067797ac58cSKevin Wolf .cfunc = map_f, 2068797ac58cSKevin Wolf .args = "", 2069797ac58cSKevin Wolf .oneline = "prints the allocated areas of a file", 2070797ac58cSKevin Wolf }; 2071797ac58cSKevin Wolf 20725bbd2e59SKevin Wolf static void reopen_help(void) 20735bbd2e59SKevin Wolf { 20745bbd2e59SKevin Wolf printf( 20755bbd2e59SKevin Wolf "\n" 20765bbd2e59SKevin Wolf " Changes the open options of an already opened image\n" 20775bbd2e59SKevin Wolf "\n" 20785bbd2e59SKevin Wolf " Example:\n" 20795bbd2e59SKevin Wolf " 'reopen -o lazy-refcounts=on' - activates lazy refcount writeback on a qcow2 image\n" 20805bbd2e59SKevin Wolf "\n" 20815bbd2e59SKevin Wolf " -r, -- Reopen the image read-only\n" 2082ea92203cSKevin Wolf " -w, -- Reopen the image read-write\n" 20835bbd2e59SKevin Wolf " -c, -- Change the cache mode to the given value\n" 20845bbd2e59SKevin Wolf " -o, -- Changes block driver options (cf. 'open' command)\n" 20855bbd2e59SKevin Wolf "\n"); 20865bbd2e59SKevin Wolf } 20875bbd2e59SKevin Wolf 2088b32d7a39SMax Reitz static int reopen_f(BlockBackend *blk, int argc, char **argv); 20895bbd2e59SKevin Wolf 20905bbd2e59SKevin Wolf static QemuOptsList reopen_opts = { 20915bbd2e59SKevin Wolf .name = "reopen", 20925bbd2e59SKevin Wolf .merge_lists = true, 20935bbd2e59SKevin Wolf .head = QTAILQ_HEAD_INITIALIZER(reopen_opts.head), 20945bbd2e59SKevin Wolf .desc = { 20955bbd2e59SKevin Wolf /* no elements => accept any params */ 20965bbd2e59SKevin Wolf { /* end of list */ } 20975bbd2e59SKevin Wolf }, 20985bbd2e59SKevin Wolf }; 20995bbd2e59SKevin Wolf 21005bbd2e59SKevin Wolf static const cmdinfo_t reopen_cmd = { 21015bbd2e59SKevin Wolf .name = "reopen", 21025bbd2e59SKevin Wolf .argmin = 0, 21035bbd2e59SKevin Wolf .argmax = -1, 21045bbd2e59SKevin Wolf .cfunc = reopen_f, 2105ea92203cSKevin Wolf .args = "[(-r|-w)] [-c cache] [-o options]", 21065bbd2e59SKevin Wolf .oneline = "reopens an image with new options", 21075bbd2e59SKevin Wolf .help = reopen_help, 21085bbd2e59SKevin Wolf }; 21095bbd2e59SKevin Wolf 2110b32d7a39SMax Reitz static int reopen_f(BlockBackend *blk, int argc, char **argv) 21115bbd2e59SKevin Wolf { 21125bbd2e59SKevin Wolf BlockDriverState *bs = blk_bs(blk); 21135bbd2e59SKevin Wolf QemuOpts *qopts; 21145bbd2e59SKevin Wolf QDict *opts; 21155bbd2e59SKevin Wolf int c; 21165bbd2e59SKevin Wolf int flags = bs->open_flags; 211719dbecdcSKevin Wolf bool writethrough = !blk_enable_write_cache(blk); 2118ea92203cSKevin Wolf bool has_rw_option = false; 2119dc900c35SAlberto Garcia bool has_cache_option = false; 21205bbd2e59SKevin Wolf Error *local_err = NULL; 21215bbd2e59SKevin Wolf 2122ea92203cSKevin Wolf while ((c = getopt(argc, argv, "c:o:rw")) != -1) { 21235bbd2e59SKevin Wolf switch (c) { 21245bbd2e59SKevin Wolf case 'c': 212519dbecdcSKevin Wolf if (bdrv_parse_cache_mode(optarg, &flags, &writethrough) < 0) { 21265bbd2e59SKevin Wolf error_report("Invalid cache option: %s", optarg); 2127b32d7a39SMax Reitz return -EINVAL; 21285bbd2e59SKevin Wolf } 2129dc900c35SAlberto Garcia has_cache_option = true; 21305bbd2e59SKevin Wolf break; 21315bbd2e59SKevin Wolf case 'o': 21325bbd2e59SKevin Wolf if (!qemu_opts_parse_noisily(&reopen_opts, optarg, 0)) { 21335bbd2e59SKevin Wolf qemu_opts_reset(&reopen_opts); 2134b32d7a39SMax Reitz return -EINVAL; 21355bbd2e59SKevin Wolf } 21365bbd2e59SKevin Wolf break; 21375bbd2e59SKevin Wolf case 'r': 2138ea92203cSKevin Wolf if (has_rw_option) { 2139ea92203cSKevin Wolf error_report("Only one -r/-w option may be given"); 2140b32d7a39SMax Reitz return -EINVAL; 2141ea92203cSKevin Wolf } 21425bbd2e59SKevin Wolf flags &= ~BDRV_O_RDWR; 2143ea92203cSKevin Wolf has_rw_option = true; 2144ea92203cSKevin Wolf break; 2145ea92203cSKevin Wolf case 'w': 2146ea92203cSKevin Wolf if (has_rw_option) { 2147ea92203cSKevin Wolf error_report("Only one -r/-w option may be given"); 2148b32d7a39SMax Reitz return -EINVAL; 2149ea92203cSKevin Wolf } 2150ea92203cSKevin Wolf flags |= BDRV_O_RDWR; 2151ea92203cSKevin Wolf has_rw_option = true; 21525bbd2e59SKevin Wolf break; 21535bbd2e59SKevin Wolf default: 21545bbd2e59SKevin Wolf qemu_opts_reset(&reopen_opts); 2155b444d0e9SMax Reitz qemuio_command_usage(&reopen_cmd); 2156b32d7a39SMax Reitz return -EINVAL; 21575bbd2e59SKevin Wolf } 21585bbd2e59SKevin Wolf } 21595bbd2e59SKevin Wolf 21605bbd2e59SKevin Wolf if (optind != argc) { 21615bbd2e59SKevin Wolf qemu_opts_reset(&reopen_opts); 2162b444d0e9SMax Reitz qemuio_command_usage(&reopen_cmd); 2163b32d7a39SMax Reitz return -EINVAL; 21645bbd2e59SKevin Wolf } 21655bbd2e59SKevin Wolf 2166a8003ec4SAlberto Garcia if (!writethrough != blk_enable_write_cache(blk) && 216719dbecdcSKevin Wolf blk_get_attached_dev(blk)) 216819dbecdcSKevin Wolf { 216919dbecdcSKevin Wolf error_report("Cannot change cache.writeback: Device attached"); 217019dbecdcSKevin Wolf qemu_opts_reset(&reopen_opts); 2171b32d7a39SMax Reitz return -EBUSY; 217219dbecdcSKevin Wolf } 217319dbecdcSKevin Wolf 2174f3adefb2SKevin Wolf if (!(flags & BDRV_O_RDWR)) { 2175f3adefb2SKevin Wolf uint64_t orig_perm, orig_shared_perm; 2176f3adefb2SKevin Wolf 2177f3adefb2SKevin Wolf bdrv_drain(bs); 2178f3adefb2SKevin Wolf 2179f3adefb2SKevin Wolf blk_get_perm(blk, &orig_perm, &orig_shared_perm); 2180f3adefb2SKevin Wolf blk_set_perm(blk, 2181f3adefb2SKevin Wolf orig_perm & ~(BLK_PERM_WRITE | BLK_PERM_WRITE_UNCHANGED), 2182f3adefb2SKevin Wolf orig_shared_perm, 2183f3adefb2SKevin Wolf &error_abort); 2184f3adefb2SKevin Wolf } 2185f3adefb2SKevin Wolf 21865bbd2e59SKevin Wolf qopts = qemu_opts_find(&reopen_opts, NULL); 2187dc900c35SAlberto Garcia opts = qopts ? qemu_opts_to_qdict(qopts, NULL) : qdict_new(); 21885bbd2e59SKevin Wolf qemu_opts_reset(&reopen_opts); 21895bbd2e59SKevin Wolf 2190dc900c35SAlberto Garcia if (qdict_haskey(opts, BDRV_OPT_READ_ONLY)) { 2191dc900c35SAlberto Garcia if (has_rw_option) { 2192dc900c35SAlberto Garcia error_report("Cannot set both -r/-w and '" BDRV_OPT_READ_ONLY "'"); 2193dc900c35SAlberto Garcia qobject_unref(opts); 2194dc900c35SAlberto Garcia return -EINVAL; 2195dc900c35SAlberto Garcia } 2196dc900c35SAlberto Garcia } else { 2197dc900c35SAlberto Garcia qdict_put_bool(opts, BDRV_OPT_READ_ONLY, !(flags & BDRV_O_RDWR)); 2198dc900c35SAlberto Garcia } 2199dc900c35SAlberto Garcia 2200dc900c35SAlberto Garcia if (qdict_haskey(opts, BDRV_OPT_CACHE_DIRECT) || 2201dc900c35SAlberto Garcia qdict_haskey(opts, BDRV_OPT_CACHE_NO_FLUSH)) { 2202dc900c35SAlberto Garcia if (has_cache_option) { 2203dc900c35SAlberto Garcia error_report("Cannot set both -c and the cache options"); 2204dc900c35SAlberto Garcia qobject_unref(opts); 2205dc900c35SAlberto Garcia return -EINVAL; 2206dc900c35SAlberto Garcia } 2207dc900c35SAlberto Garcia } else { 2208dc900c35SAlberto Garcia qdict_put_bool(opts, BDRV_OPT_CACHE_DIRECT, flags & BDRV_O_NOCACHE); 2209dc900c35SAlberto Garcia qdict_put_bool(opts, BDRV_OPT_CACHE_NO_FLUSH, flags & BDRV_O_NO_FLUSH); 2210dc900c35SAlberto Garcia } 2211dc900c35SAlberto Garcia 22126cf42ca2SKevin Wolf bdrv_reopen(bs, opts, true, &local_err); 22131a63a907SKevin Wolf 22145bbd2e59SKevin Wolf if (local_err) { 22155bbd2e59SKevin Wolf error_report_err(local_err); 2216b32d7a39SMax Reitz return -EINVAL; 22175bbd2e59SKevin Wolf } 22185bbd2e59SKevin Wolf 2219b32d7a39SMax Reitz blk_set_enable_write_cache(blk, !writethrough); 2220b32d7a39SMax Reitz return 0; 2221b32d7a39SMax Reitz } 2222b32d7a39SMax Reitz 2223b32d7a39SMax Reitz static int break_f(BlockBackend *blk, int argc, char **argv) 2224797ac58cSKevin Wolf { 2225797ac58cSKevin Wolf int ret; 2226797ac58cSKevin Wolf 22274c7b7e9bSMax Reitz ret = bdrv_debug_breakpoint(blk_bs(blk), argv[1], argv[2]); 2228797ac58cSKevin Wolf if (ret < 0) { 2229797ac58cSKevin Wolf printf("Could not set breakpoint: %s\n", strerror(-ret)); 2230b32d7a39SMax Reitz return ret; 2231797ac58cSKevin Wolf } 2232797ac58cSKevin Wolf 2233b32d7a39SMax Reitz return 0; 2234b32d7a39SMax Reitz } 2235b32d7a39SMax Reitz 2236b32d7a39SMax Reitz static int remove_break_f(BlockBackend *blk, int argc, char **argv) 22374cc70e93SFam Zheng { 22384cc70e93SFam Zheng int ret; 22394cc70e93SFam Zheng 22404c7b7e9bSMax Reitz ret = bdrv_debug_remove_breakpoint(blk_bs(blk), argv[1]); 22414cc70e93SFam Zheng if (ret < 0) { 22424cc70e93SFam Zheng printf("Could not remove breakpoint %s: %s\n", argv[1], strerror(-ret)); 2243b32d7a39SMax Reitz return ret; 22444cc70e93SFam Zheng } 2245b32d7a39SMax Reitz 2246b32d7a39SMax Reitz return 0; 22474cc70e93SFam Zheng } 22484cc70e93SFam Zheng 2249797ac58cSKevin Wolf static const cmdinfo_t break_cmd = { 2250797ac58cSKevin Wolf .name = "break", 2251797ac58cSKevin Wolf .argmin = 2, 2252797ac58cSKevin Wolf .argmax = 2, 2253797ac58cSKevin Wolf .cfunc = break_f, 2254797ac58cSKevin Wolf .args = "event tag", 2255797ac58cSKevin Wolf .oneline = "sets a breakpoint on event and tags the stopped " 2256797ac58cSKevin Wolf "request as tag", 2257797ac58cSKevin Wolf }; 2258797ac58cSKevin Wolf 22594cc70e93SFam Zheng static const cmdinfo_t remove_break_cmd = { 22604cc70e93SFam Zheng .name = "remove_break", 22614cc70e93SFam Zheng .argmin = 1, 22624cc70e93SFam Zheng .argmax = 1, 22634cc70e93SFam Zheng .cfunc = remove_break_f, 22644cc70e93SFam Zheng .args = "tag", 22654cc70e93SFam Zheng .oneline = "remove a breakpoint by tag", 22664cc70e93SFam Zheng }; 22674cc70e93SFam Zheng 2268b32d7a39SMax Reitz static int resume_f(BlockBackend *blk, int argc, char **argv) 2269797ac58cSKevin Wolf { 2270797ac58cSKevin Wolf int ret; 2271797ac58cSKevin Wolf 22724c7b7e9bSMax Reitz ret = bdrv_debug_resume(blk_bs(blk), argv[1]); 2273797ac58cSKevin Wolf if (ret < 0) { 2274797ac58cSKevin Wolf printf("Could not resume request: %s\n", strerror(-ret)); 2275b32d7a39SMax Reitz return ret; 2276797ac58cSKevin Wolf } 2277b32d7a39SMax Reitz 2278b32d7a39SMax Reitz return 0; 2279797ac58cSKevin Wolf } 2280797ac58cSKevin Wolf 2281797ac58cSKevin Wolf static const cmdinfo_t resume_cmd = { 2282797ac58cSKevin Wolf .name = "resume", 2283797ac58cSKevin Wolf .argmin = 1, 2284797ac58cSKevin Wolf .argmax = 1, 2285797ac58cSKevin Wolf .cfunc = resume_f, 2286797ac58cSKevin Wolf .args = "tag", 2287797ac58cSKevin Wolf .oneline = "resumes the request tagged as tag", 2288797ac58cSKevin Wolf }; 2289797ac58cSKevin Wolf 2290b32d7a39SMax Reitz static int wait_break_f(BlockBackend *blk, int argc, char **argv) 2291797ac58cSKevin Wolf { 22924c7b7e9bSMax Reitz while (!bdrv_debug_is_suspended(blk_bs(blk), argv[1])) { 22934c7b7e9bSMax Reitz aio_poll(blk_get_aio_context(blk), true); 2294797ac58cSKevin Wolf } 2295b32d7a39SMax Reitz return 0; 2296797ac58cSKevin Wolf } 2297797ac58cSKevin Wolf 2298797ac58cSKevin Wolf static const cmdinfo_t wait_break_cmd = { 2299797ac58cSKevin Wolf .name = "wait_break", 2300797ac58cSKevin Wolf .argmin = 1, 2301797ac58cSKevin Wolf .argmax = 1, 2302797ac58cSKevin Wolf .cfunc = wait_break_f, 2303797ac58cSKevin Wolf .args = "tag", 2304797ac58cSKevin Wolf .oneline = "waits for the suspension of a request", 2305797ac58cSKevin Wolf }; 2306797ac58cSKevin Wolf 2307b32d7a39SMax Reitz static int abort_f(BlockBackend *blk, int argc, char **argv) 2308797ac58cSKevin Wolf { 2309797ac58cSKevin Wolf abort(); 2310797ac58cSKevin Wolf } 2311797ac58cSKevin Wolf 2312797ac58cSKevin Wolf static const cmdinfo_t abort_cmd = { 2313797ac58cSKevin Wolf .name = "abort", 2314797ac58cSKevin Wolf .cfunc = abort_f, 2315797ac58cSKevin Wolf .flags = CMD_NOFILE_OK, 2316797ac58cSKevin Wolf .oneline = "simulate a program crash using abort(3)", 2317797ac58cSKevin Wolf }; 2318797ac58cSKevin Wolf 23190e82dc7bSMax Reitz static void sigraise_help(void) 23200e82dc7bSMax Reitz { 23210e82dc7bSMax Reitz printf( 23220e82dc7bSMax Reitz "\n" 23230e82dc7bSMax Reitz " raises the given signal\n" 23240e82dc7bSMax Reitz "\n" 23250e82dc7bSMax Reitz " Example:\n" 23260e82dc7bSMax Reitz " 'sigraise %i' - raises SIGTERM\n" 23270e82dc7bSMax Reitz "\n" 23280e82dc7bSMax Reitz " Invokes raise(signal), where \"signal\" is the mandatory integer argument\n" 23290e82dc7bSMax Reitz " given to sigraise.\n" 23300e82dc7bSMax Reitz "\n", SIGTERM); 23310e82dc7bSMax Reitz } 23320e82dc7bSMax Reitz 2333b32d7a39SMax Reitz static int sigraise_f(BlockBackend *blk, int argc, char **argv); 23340e82dc7bSMax Reitz 23350e82dc7bSMax Reitz static const cmdinfo_t sigraise_cmd = { 23360e82dc7bSMax Reitz .name = "sigraise", 23370e82dc7bSMax Reitz .cfunc = sigraise_f, 23380e82dc7bSMax Reitz .argmin = 1, 23390e82dc7bSMax Reitz .argmax = 1, 23400e82dc7bSMax Reitz .flags = CMD_NOFILE_OK, 23410e82dc7bSMax Reitz .args = "signal", 23420e82dc7bSMax Reitz .oneline = "raises a signal", 23430e82dc7bSMax Reitz .help = sigraise_help, 23440e82dc7bSMax Reitz }; 23450e82dc7bSMax Reitz 2346b32d7a39SMax Reitz static int sigraise_f(BlockBackend *blk, int argc, char **argv) 23470e82dc7bSMax Reitz { 23489b0beaf3SJohn Snow int64_t sig = cvtnum(argv[1]); 23490e82dc7bSMax Reitz if (sig < 0) { 2350a9ecfa00SJohn Snow print_cvtnum_err(sig, argv[1]); 2351b32d7a39SMax Reitz return sig; 23529b0beaf3SJohn Snow } else if (sig > NSIG) { 23539b0beaf3SJohn Snow printf("signal argument '%s' is too large to be a valid signal\n", 23549b0beaf3SJohn Snow argv[1]); 2355b32d7a39SMax Reitz return -EINVAL; 23560e82dc7bSMax Reitz } 23570e82dc7bSMax Reitz 23580e82dc7bSMax Reitz /* Using raise() to kill this process does not necessarily flush all open 23590e82dc7bSMax Reitz * streams. At least stdout and stderr (although the latter should be 23600e82dc7bSMax Reitz * non-buffered anyway) should be flushed, though. */ 23610e82dc7bSMax Reitz fflush(stdout); 23620e82dc7bSMax Reitz fflush(stderr); 23630e82dc7bSMax Reitz 23640e82dc7bSMax Reitz raise(sig); 2365b32d7a39SMax Reitz 2366b32d7a39SMax Reitz return 0; 23670e82dc7bSMax Reitz } 23680e82dc7bSMax Reitz 2369cd33d02aSKevin Wolf static void sleep_cb(void *opaque) 2370cd33d02aSKevin Wolf { 2371cd33d02aSKevin Wolf bool *expired = opaque; 2372cd33d02aSKevin Wolf *expired = true; 2373cd33d02aSKevin Wolf } 2374cd33d02aSKevin Wolf 2375b32d7a39SMax Reitz static int sleep_f(BlockBackend *blk, int argc, char **argv) 2376cd33d02aSKevin Wolf { 2377cd33d02aSKevin Wolf char *endptr; 2378cd33d02aSKevin Wolf long ms; 2379cd33d02aSKevin Wolf struct QEMUTimer *timer; 2380cd33d02aSKevin Wolf bool expired = false; 2381cd33d02aSKevin Wolf 2382cd33d02aSKevin Wolf ms = strtol(argv[1], &endptr, 0); 2383cd33d02aSKevin Wolf if (ms < 0 || *endptr != '\0') { 2384cd33d02aSKevin Wolf printf("%s is not a valid number\n", argv[1]); 2385b32d7a39SMax Reitz return -EINVAL; 2386cd33d02aSKevin Wolf } 2387cd33d02aSKevin Wolf 2388cd33d02aSKevin Wolf timer = timer_new_ns(QEMU_CLOCK_HOST, sleep_cb, &expired); 2389cd33d02aSKevin Wolf timer_mod(timer, qemu_clock_get_ns(QEMU_CLOCK_HOST) + SCALE_MS * ms); 2390cd33d02aSKevin Wolf 2391cd33d02aSKevin Wolf while (!expired) { 2392cd33d02aSKevin Wolf main_loop_wait(false); 2393cd33d02aSKevin Wolf } 2394cd33d02aSKevin Wolf 2395cd33d02aSKevin Wolf timer_free(timer); 2396b32d7a39SMax Reitz return 0; 2397cd33d02aSKevin Wolf } 2398cd33d02aSKevin Wolf 2399cd33d02aSKevin Wolf static const cmdinfo_t sleep_cmd = { 2400cd33d02aSKevin Wolf .name = "sleep", 2401cd33d02aSKevin Wolf .argmin = 1, 2402cd33d02aSKevin Wolf .argmax = 1, 2403cd33d02aSKevin Wolf .cfunc = sleep_f, 2404cd33d02aSKevin Wolf .flags = CMD_NOFILE_OK, 2405cd33d02aSKevin Wolf .oneline = "waits for the given value in milliseconds", 2406cd33d02aSKevin Wolf }; 2407cd33d02aSKevin Wolf 2408f18a834aSKevin Wolf static void help_oneline(const char *cmd, const cmdinfo_t *ct) 2409f18a834aSKevin Wolf { 2410f18a834aSKevin Wolf printf("%s ", cmd); 2411f18a834aSKevin Wolf 2412f18a834aSKevin Wolf if (ct->args) { 2413f18a834aSKevin Wolf printf("%s ", ct->args); 2414f18a834aSKevin Wolf } 2415f18a834aSKevin Wolf printf("-- %s\n", ct->oneline); 2416f18a834aSKevin Wolf } 2417f18a834aSKevin Wolf 2418f18a834aSKevin Wolf static void help_onecmd(const char *cmd, const cmdinfo_t *ct) 2419f18a834aSKevin Wolf { 2420f18a834aSKevin Wolf help_oneline(cmd, ct); 2421f18a834aSKevin Wolf if (ct->help) { 2422f18a834aSKevin Wolf ct->help(); 2423f18a834aSKevin Wolf } 2424f18a834aSKevin Wolf } 2425f18a834aSKevin Wolf 2426f18a834aSKevin Wolf static void help_all(void) 2427f18a834aSKevin Wolf { 2428f18a834aSKevin Wolf const cmdinfo_t *ct; 2429f18a834aSKevin Wolf 2430f18a834aSKevin Wolf for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { 2431f18a834aSKevin Wolf help_oneline(ct->name, ct); 2432f18a834aSKevin Wolf } 2433f18a834aSKevin Wolf printf("\nUse 'help commandname' for extended help.\n"); 2434f18a834aSKevin Wolf } 2435f18a834aSKevin Wolf 2436b32d7a39SMax Reitz static int help_f(BlockBackend *blk, int argc, char **argv) 2437f18a834aSKevin Wolf { 2438f18a834aSKevin Wolf const cmdinfo_t *ct; 2439f18a834aSKevin Wolf 2440da16f4b8SDr. David Alan Gilbert if (argc < 2) { 2441f18a834aSKevin Wolf help_all(); 2442b32d7a39SMax Reitz return 0; 2443f18a834aSKevin Wolf } 2444f18a834aSKevin Wolf 2445f18a834aSKevin Wolf ct = find_command(argv[1]); 2446f18a834aSKevin Wolf if (ct == NULL) { 2447f18a834aSKevin Wolf printf("command %s not found\n", argv[1]); 2448b32d7a39SMax Reitz return -EINVAL; 2449f18a834aSKevin Wolf } 2450f18a834aSKevin Wolf 2451f18a834aSKevin Wolf help_onecmd(argv[1], ct); 2452b32d7a39SMax Reitz return 0; 2453f18a834aSKevin Wolf } 2454f18a834aSKevin Wolf 2455f18a834aSKevin Wolf static const cmdinfo_t help_cmd = { 2456f18a834aSKevin Wolf .name = "help", 2457f18a834aSKevin Wolf .altname = "?", 2458f18a834aSKevin Wolf .cfunc = help_f, 2459f18a834aSKevin Wolf .argmin = 0, 2460f18a834aSKevin Wolf .argmax = 1, 2461f18a834aSKevin Wolf .flags = CMD_FLAG_GLOBAL, 2462f18a834aSKevin Wolf .args = "[command]", 2463f18a834aSKevin Wolf .oneline = "help for one or all commands", 2464f18a834aSKevin Wolf }; 2465f18a834aSKevin Wolf 246678632a3dSVladimir Sementsov-Ogievskiy /* 246778632a3dSVladimir Sementsov-Ogievskiy * Called with aio context of blk acquired. Or with qemu_get_aio_context() 246878632a3dSVladimir Sementsov-Ogievskiy * context acquired if blk is NULL. 246978632a3dSVladimir Sementsov-Ogievskiy */ 2470b32d7a39SMax Reitz int qemuio_command(BlockBackend *blk, const char *cmd) 2471dd583296SKevin Wolf { 2472dd583296SKevin Wolf char *input; 2473dd583296SKevin Wolf const cmdinfo_t *ct; 2474dd583296SKevin Wolf char **v; 2475dd583296SKevin Wolf int c; 2476b32d7a39SMax Reitz int ret = 0; 2477dd583296SKevin Wolf 2478dd583296SKevin Wolf input = g_strdup(cmd); 2479dd583296SKevin Wolf v = breakline(input, &c); 2480dd583296SKevin Wolf if (c) { 2481dd583296SKevin Wolf ct = find_command(v[0]); 2482dd583296SKevin Wolf if (ct) { 2483b32d7a39SMax Reitz ret = command(blk, ct, c, v); 2484dd583296SKevin Wolf } else { 2485dd583296SKevin Wolf fprintf(stderr, "command \"%s\" not found\n", v[0]); 2486b32d7a39SMax Reitz ret = -EINVAL; 2487dd583296SKevin Wolf } 2488dd583296SKevin Wolf } 2489dd583296SKevin Wolf g_free(input); 2490dd583296SKevin Wolf g_free(v); 2491b32d7a39SMax Reitz 2492b32d7a39SMax Reitz return ret; 2493dd583296SKevin Wolf } 2494dd583296SKevin Wolf 2495797ac58cSKevin Wolf static void __attribute((constructor)) init_qemuio_commands(void) 2496797ac58cSKevin Wolf { 2497797ac58cSKevin Wolf /* initialize commands */ 2498c2cdf5c5SKevin Wolf qemuio_add_command(&help_cmd); 2499c2cdf5c5SKevin Wolf qemuio_add_command(&read_cmd); 2500c2cdf5c5SKevin Wolf qemuio_add_command(&readv_cmd); 2501c2cdf5c5SKevin Wolf qemuio_add_command(&write_cmd); 2502c2cdf5c5SKevin Wolf qemuio_add_command(&writev_cmd); 2503c2cdf5c5SKevin Wolf qemuio_add_command(&aio_read_cmd); 2504c2cdf5c5SKevin Wolf qemuio_add_command(&aio_write_cmd); 2505c2cdf5c5SKevin Wolf qemuio_add_command(&aio_flush_cmd); 2506c2cdf5c5SKevin Wolf qemuio_add_command(&flush_cmd); 2507c2cdf5c5SKevin Wolf qemuio_add_command(&truncate_cmd); 2508c2cdf5c5SKevin Wolf qemuio_add_command(&length_cmd); 2509c2cdf5c5SKevin Wolf qemuio_add_command(&info_cmd); 2510c2cdf5c5SKevin Wolf qemuio_add_command(&discard_cmd); 2511c2cdf5c5SKevin Wolf qemuio_add_command(&alloc_cmd); 2512c2cdf5c5SKevin Wolf qemuio_add_command(&map_cmd); 25135bbd2e59SKevin Wolf qemuio_add_command(&reopen_cmd); 2514c2cdf5c5SKevin Wolf qemuio_add_command(&break_cmd); 25154cc70e93SFam Zheng qemuio_add_command(&remove_break_cmd); 2516c2cdf5c5SKevin Wolf qemuio_add_command(&resume_cmd); 2517c2cdf5c5SKevin Wolf qemuio_add_command(&wait_break_cmd); 2518c2cdf5c5SKevin Wolf qemuio_add_command(&abort_cmd); 2519cd33d02aSKevin Wolf qemuio_add_command(&sleep_cmd); 25200e82dc7bSMax Reitz qemuio_add_command(&sigraise_cmd); 2521797ac58cSKevin Wolf } 2522