| /* | 
 |  * Command line utility to exercise the QEMU I/O path. | 
 |  * | 
 |  * Copyright (C) 2009-2016 Red Hat, Inc. | 
 |  * Copyright (c) 2003-2005 Silicon Graphics, Inc. | 
 |  * | 
 |  * This work is licensed under the terms of the GNU GPL, version 2 or later. | 
 |  * See the COPYING file in the top-level directory. | 
 |  */ | 
 |  | 
 | #include "qemu/osdep.h" | 
 | #include "qapi/error.h" | 
 | #include "qapi/qmp/qdict.h" | 
 | #include "qemu-io.h" | 
 | #include "sysemu/block-backend.h" | 
 | #include "block/block.h" | 
 | #include "block/block_int.h" /* for info_f() */ | 
 | #include "block/qapi.h" | 
 | #include "qemu/error-report.h" | 
 | #include "qemu/main-loop.h" | 
 | #include "qemu/option.h" | 
 | #include "qemu/timer.h" | 
 | #include "qemu/cutils.h" | 
 | #include "qemu/memalign.h" | 
 |  | 
 | #define CMD_NOFILE_OK   0x01 | 
 |  | 
 | bool qemuio_misalign; | 
 |  | 
 | static cmdinfo_t *cmdtab; | 
 | static int ncmds; | 
 |  | 
 | static int compare_cmdname(const void *a, const void *b) | 
 | { | 
 |     return strcmp(((const cmdinfo_t *)a)->name, | 
 |                   ((const cmdinfo_t *)b)->name); | 
 | } | 
 |  | 
 | void qemuio_add_command(const cmdinfo_t *ci) | 
 | { | 
 |     /* ci->perm assumes a file is open, but the GLOBAL and NOFILE_OK | 
 |      * flags allow it not to be, so that combination is invalid. | 
 |      * Catch it now rather than letting it manifest as a crash if a | 
 |      * particular set of command line options are used. | 
 |      */ | 
 |     assert(ci->perm == 0 || | 
 |            (ci->flags & (CMD_FLAG_GLOBAL | CMD_NOFILE_OK)) == 0); | 
 |     cmdtab = g_renew(cmdinfo_t, cmdtab, ++ncmds); | 
 |     cmdtab[ncmds - 1] = *ci; | 
 |     qsort(cmdtab, ncmds, sizeof(*cmdtab), compare_cmdname); | 
 | } | 
 |  | 
 | void qemuio_command_usage(const cmdinfo_t *ci) | 
 | { | 
 |     printf("%s %s -- %s\n", ci->name, ci->args, ci->oneline); | 
 | } | 
 |  | 
 | static int init_check_command(BlockBackend *blk, const cmdinfo_t *ct) | 
 | { | 
 |     if (ct->flags & CMD_FLAG_GLOBAL) { | 
 |         return 1; | 
 |     } | 
 |     if (!(ct->flags & CMD_NOFILE_OK) && !blk) { | 
 |         fprintf(stderr, "no file open, try 'help open'\n"); | 
 |         return 0; | 
 |     } | 
 |     return 1; | 
 | } | 
 |  | 
 | static int command(BlockBackend *blk, const cmdinfo_t *ct, int argc, | 
 |                    char **argv) | 
 | { | 
 |     char *cmd = argv[0]; | 
 |  | 
 |     if (!init_check_command(blk, ct)) { | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if (argc - 1 < ct->argmin || (ct->argmax != -1 && argc - 1 > ct->argmax)) { | 
 |         if (ct->argmax == -1) { | 
 |             fprintf(stderr, | 
 |                     "bad argument count %d to %s, expected at least %d arguments\n", | 
 |                     argc-1, cmd, ct->argmin); | 
 |         } else if (ct->argmin == ct->argmax) { | 
 |             fprintf(stderr, | 
 |                     "bad argument count %d to %s, expected %d arguments\n", | 
 |                     argc-1, cmd, ct->argmin); | 
 |         } else { | 
 |             fprintf(stderr, | 
 |                     "bad argument count %d to %s, expected between %d and %d arguments\n", | 
 |                     argc-1, cmd, ct->argmin, ct->argmax); | 
 |         } | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     /* | 
 |      * Request additional permissions if necessary for this command. The caller | 
 |      * is responsible for restoring the original permissions afterwards if this | 
 |      * is what it wants. | 
 |      * | 
 |      * Coverity thinks that blk may be NULL in the following if condition. It's | 
 |      * not so: in init_check_command() we fail if blk is NULL for command with | 
 |      * both CMD_FLAG_GLOBAL and CMD_NOFILE_OK flags unset. And in | 
 |      * qemuio_add_command() we assert that command with non-zero .perm field | 
 |      * doesn't set this flags. So, the following assertion is to silence | 
 |      * Coverity: | 
 |      */ | 
 |     assert(blk || !ct->perm); | 
 |     if (ct->perm && blk_is_available(blk)) { | 
 |         uint64_t orig_perm, orig_shared_perm; | 
 |         blk_get_perm(blk, &orig_perm, &orig_shared_perm); | 
 |  | 
 |         if (ct->perm & ~orig_perm) { | 
 |             uint64_t new_perm; | 
 |             Error *local_err = NULL; | 
 |             int ret; | 
 |  | 
 |             new_perm = orig_perm | ct->perm; | 
 |  | 
 |             ret = blk_set_perm(blk, new_perm, orig_shared_perm, &local_err); | 
 |             if (ret < 0) { | 
 |                 error_report_err(local_err); | 
 |                 return ret; | 
 |             } | 
 |         } | 
 |     } | 
 |  | 
 |     qemu_reset_optind(); | 
 |     return ct->cfunc(blk, argc, argv); | 
 | } | 
 |  | 
 | static const cmdinfo_t *find_command(const char *cmd) | 
 | { | 
 |     cmdinfo_t *ct; | 
 |  | 
 |     for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { | 
 |         if (strcmp(ct->name, cmd) == 0 || | 
 |             (ct->altname && strcmp(ct->altname, cmd) == 0)) | 
 |         { | 
 |             return (const cmdinfo_t *)ct; | 
 |         } | 
 |     } | 
 |     return NULL; | 
 | } | 
 |  | 
 | /* Invoke fn() for commands with a matching prefix */ | 
 | void qemuio_complete_command(const char *input, | 
 |                              void (*fn)(const char *cmd, void *opaque), | 
 |                              void *opaque) | 
 | { | 
 |     cmdinfo_t *ct; | 
 |     size_t input_len = strlen(input); | 
 |  | 
 |     for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { | 
 |         if (strncmp(input, ct->name, input_len) == 0) { | 
 |             fn(ct->name, opaque); | 
 |         } | 
 |     } | 
 | } | 
 |  | 
 | static char **breakline(char *input, int *count) | 
 | { | 
 |     int c = 0; | 
 |     char *p; | 
 |     char **rval = g_new0(char *, 1); | 
 |  | 
 |     while (rval && (p = qemu_strsep(&input, " ")) != NULL) { | 
 |         if (!*p) { | 
 |             continue; | 
 |         } | 
 |         c++; | 
 |         rval = g_renew(char *, rval, (c + 1)); | 
 |         rval[c - 1] = p; | 
 |         rval[c] = NULL; | 
 |     } | 
 |     *count = c; | 
 |     return rval; | 
 | } | 
 |  | 
 | static int64_t cvtnum(const char *s) | 
 | { | 
 |     int err; | 
 |     uint64_t value; | 
 |  | 
 |     err = qemu_strtosz(s, NULL, &value); | 
 |     if (err < 0) { | 
 |         return err; | 
 |     } | 
 |     if (value > INT64_MAX) { | 
 |         return -ERANGE; | 
 |     } | 
 |     return value; | 
 | } | 
 |  | 
 | static void print_cvtnum_err(int64_t rc, const char *arg) | 
 | { | 
 |     switch (rc) { | 
 |     case -EINVAL: | 
 |         printf("Parsing error: non-numeric argument," | 
 |                " or extraneous/unrecognized suffix -- %s\n", arg); | 
 |         break; | 
 |     case -ERANGE: | 
 |         printf("Parsing error: argument too large -- %s\n", arg); | 
 |         break; | 
 |     default: | 
 |         printf("Parsing error: %s\n", arg); | 
 |     } | 
 | } | 
 |  | 
 | #define EXABYTES(x)     ((long long)(x) << 60) | 
 | #define PETABYTES(x)    ((long long)(x) << 50) | 
 | #define TERABYTES(x)    ((long long)(x) << 40) | 
 | #define GIGABYTES(x)    ((long long)(x) << 30) | 
 | #define MEGABYTES(x)    ((long long)(x) << 20) | 
 | #define KILOBYTES(x)    ((long long)(x) << 10) | 
 |  | 
 | #define TO_EXABYTES(x)  ((x) / EXABYTES(1)) | 
 | #define TO_PETABYTES(x) ((x) / PETABYTES(1)) | 
 | #define TO_TERABYTES(x) ((x) / TERABYTES(1)) | 
 | #define TO_GIGABYTES(x) ((x) / GIGABYTES(1)) | 
 | #define TO_MEGABYTES(x) ((x) / MEGABYTES(1)) | 
 | #define TO_KILOBYTES(x) ((x) / KILOBYTES(1)) | 
 |  | 
 | static void cvtstr(double value, char *str, size_t size) | 
 | { | 
 |     char *trim; | 
 |     const char *suffix; | 
 |  | 
 |     if (value >= EXABYTES(1)) { | 
 |         suffix = " EiB"; | 
 |         snprintf(str, size - 4, "%.3f", TO_EXABYTES(value)); | 
 |     } else if (value >= PETABYTES(1)) { | 
 |         suffix = " PiB"; | 
 |         snprintf(str, size - 4, "%.3f", TO_PETABYTES(value)); | 
 |     } else if (value >= TERABYTES(1)) { | 
 |         suffix = " TiB"; | 
 |         snprintf(str, size - 4, "%.3f", TO_TERABYTES(value)); | 
 |     } else if (value >= GIGABYTES(1)) { | 
 |         suffix = " GiB"; | 
 |         snprintf(str, size - 4, "%.3f", TO_GIGABYTES(value)); | 
 |     } else if (value >= MEGABYTES(1)) { | 
 |         suffix = " MiB"; | 
 |         snprintf(str, size - 4, "%.3f", TO_MEGABYTES(value)); | 
 |     } else if (value >= KILOBYTES(1)) { | 
 |         suffix = " KiB"; | 
 |         snprintf(str, size - 4, "%.3f", TO_KILOBYTES(value)); | 
 |     } else { | 
 |         suffix = " bytes"; | 
 |         snprintf(str, size - 6, "%f", value); | 
 |     } | 
 |  | 
 |     trim = strstr(str, ".000"); | 
 |     if (trim) { | 
 |         strcpy(trim, suffix); | 
 |     } else { | 
 |         strcat(str, suffix); | 
 |     } | 
 | } | 
 |  | 
 |  | 
 |  | 
 | static struct timespec tsub(struct timespec t1, struct timespec t2) | 
 | { | 
 |     t1.tv_nsec -= t2.tv_nsec; | 
 |     if (t1.tv_nsec < 0) { | 
 |         t1.tv_nsec += NANOSECONDS_PER_SECOND; | 
 |         t1.tv_sec--; | 
 |     } | 
 |     t1.tv_sec -= t2.tv_sec; | 
 |     return t1; | 
 | } | 
 |  | 
 | static double tdiv(double value, struct timespec tv) | 
 | { | 
 |     double seconds = tv.tv_sec + (tv.tv_nsec / 1e9); | 
 |     return value / seconds; | 
 | } | 
 |  | 
 | #define HOURS(sec)      ((sec) / (60 * 60)) | 
 | #define MINUTES(sec)    (((sec) % (60 * 60)) / 60) | 
 | #define SECONDS(sec)    ((sec) % 60) | 
 |  | 
 | enum { | 
 |     DEFAULT_TIME        = 0x0, | 
 |     TERSE_FIXED_TIME    = 0x1, | 
 |     VERBOSE_FIXED_TIME  = 0x2, | 
 | }; | 
 |  | 
 | static void timestr(struct timespec *tv, char *ts, size_t size, int format) | 
 | { | 
 |     double frac_sec = tv->tv_nsec / 1e9; | 
 |  | 
 |     if (format & TERSE_FIXED_TIME) { | 
 |         if (!HOURS(tv->tv_sec)) { | 
 |             snprintf(ts, size, "%u:%05.2f", | 
 |                      (unsigned int) MINUTES(tv->tv_sec), | 
 |                      SECONDS(tv->tv_sec) + frac_sec); | 
 |             return; | 
 |         } | 
 |         format |= VERBOSE_FIXED_TIME; /* fallback if hours needed */ | 
 |     } | 
 |  | 
 |     if ((format & VERBOSE_FIXED_TIME) || tv->tv_sec) { | 
 |         snprintf(ts, size, "%u:%02u:%05.2f", | 
 |                 (unsigned int) HOURS(tv->tv_sec), | 
 |                 (unsigned int) MINUTES(tv->tv_sec), | 
 |                  SECONDS(tv->tv_sec) + frac_sec); | 
 |     } else { | 
 |         snprintf(ts, size, "%05.2f sec", frac_sec); | 
 |     } | 
 | } | 
 |  | 
 | /* | 
 |  * Parse the pattern argument to various sub-commands. | 
 |  * | 
 |  * Because the pattern is used as an argument to memset it must evaluate | 
 |  * to an unsigned integer that fits into a single byte. | 
 |  */ | 
 | static int parse_pattern(const char *arg) | 
 | { | 
 |     char *endptr = NULL; | 
 |     long pattern; | 
 |  | 
 |     pattern = strtol(arg, &endptr, 0); | 
 |     if (pattern < 0 || pattern > UCHAR_MAX || *endptr != '\0') { | 
 |         printf("%s is not a valid pattern byte\n", arg); | 
 |         return -1; | 
 |     } | 
 |  | 
 |     return pattern; | 
 | } | 
 |  | 
 | /* | 
 |  * Memory allocation helpers. | 
 |  * | 
 |  * Make sure memory is aligned by default, or purposefully misaligned if | 
 |  * that is specified on the command line. | 
 |  */ | 
 |  | 
 | #define MISALIGN_OFFSET     16 | 
 | static void *qemu_io_alloc(BlockBackend *blk, size_t len, int pattern, | 
 |                            bool register_buf) | 
 | { | 
 |     void *buf; | 
 |  | 
 |     if (qemuio_misalign) { | 
 |         len += MISALIGN_OFFSET; | 
 |     } | 
 |     buf = blk_blockalign(blk, len); | 
 |     memset(buf, pattern, len); | 
 |     if (register_buf) { | 
 |         blk_register_buf(blk, buf, len, &error_abort); | 
 |     } | 
 |     if (qemuio_misalign) { | 
 |         buf += MISALIGN_OFFSET; | 
 |     } | 
 |     return buf; | 
 | } | 
 |  | 
 | static void qemu_io_free(BlockBackend *blk, void *p, size_t len, | 
 |                          bool unregister_buf) | 
 | { | 
 |     if (qemuio_misalign) { | 
 |         p -= MISALIGN_OFFSET; | 
 |         len += MISALIGN_OFFSET; | 
 |     } | 
 |     if (unregister_buf) { | 
 |         blk_unregister_buf(blk, p, len); | 
 |     } | 
 |     qemu_vfree(p); | 
 | } | 
 |  | 
 | /* | 
 |  * qemu_io_alloc_from_file() | 
 |  * | 
 |  * Allocates the buffer and populates it with the content of the given file | 
 |  * up to @len bytes. If the file length is less than @len, then the buffer | 
 |  * is populated with the file content cyclically. | 
 |  * | 
 |  * @blk - the block backend where the buffer content is going to be written to | 
 |  * @len - the buffer length | 
 |  * @file_name - the file to read the content from | 
 |  * @register_buf - call blk_register_buf() | 
 |  * | 
 |  * Returns: the buffer pointer on success | 
 |  *          NULL on error | 
 |  */ | 
 | static void *qemu_io_alloc_from_file(BlockBackend *blk, size_t len, | 
 |                                      const char *file_name, bool register_buf) | 
 | { | 
 |     size_t alloc_len = len + (qemuio_misalign ? MISALIGN_OFFSET : 0); | 
 |     char *alloc_buf, *buf, *end; | 
 |     FILE *f = fopen(file_name, "r"); | 
 |     int pattern_len; | 
 |  | 
 |     if (!f) { | 
 |         perror(file_name); | 
 |         return NULL; | 
 |     } | 
 |  | 
 |     alloc_buf = buf = blk_blockalign(blk, alloc_len); | 
 |  | 
 |     if (qemuio_misalign) { | 
 |         buf += MISALIGN_OFFSET; | 
 |     } | 
 |  | 
 |     pattern_len = fread(buf, 1, len, f); | 
 |  | 
 |     if (ferror(f)) { | 
 |         perror(file_name); | 
 |         goto error; | 
 |     } | 
 |  | 
 |     if (pattern_len == 0) { | 
 |         fprintf(stderr, "%s: file is empty\n", file_name); | 
 |         goto error; | 
 |     } | 
 |  | 
 |     fclose(f); | 
 |     f = NULL; | 
 |  | 
 |     if (register_buf) { | 
 |         blk_register_buf(blk, alloc_buf, alloc_len, &error_abort); | 
 |     } | 
 |  | 
 |     end = buf + len; | 
 |     for (char *p = buf + pattern_len; p < end; p += pattern_len) { | 
 |         memcpy(p, buf, MIN(pattern_len, end - p)); | 
 |     } | 
 |  | 
 |     return buf; | 
 |  | 
 | error: | 
 |     /* | 
 |      * This code path is only taken before blk_register_buf() is called, so | 
 |      * hardcode the qemu_io_free() unregister_buf argument to false. | 
 |      */ | 
 |     qemu_io_free(blk, alloc_buf, alloc_len, false); | 
 |     if (f) { | 
 |         fclose(f); | 
 |     } | 
 |     return NULL; | 
 | } | 
 |  | 
 | static void dump_buffer(const void *buffer, int64_t offset, int64_t len) | 
 | { | 
 |     uint64_t i; | 
 |     int j; | 
 |     const uint8_t *p; | 
 |  | 
 |     for (i = 0, p = buffer; i < len; i += 16) { | 
 |         const uint8_t *s = p; | 
 |  | 
 |         printf("%08" PRIx64 ":  ", offset + i); | 
 |         for (j = 0; j < 16 && i + j < len; j++, p++) { | 
 |             printf("%02x ", *p); | 
 |         } | 
 |         printf(" "); | 
 |         for (j = 0; j < 16 && i + j < len; j++, s++) { | 
 |             if (isalnum(*s)) { | 
 |                 printf("%c", *s); | 
 |             } else { | 
 |                 printf("."); | 
 |             } | 
 |         } | 
 |         printf("\n"); | 
 |     } | 
 | } | 
 |  | 
 | static void print_report(const char *op, struct timespec *t, int64_t offset, | 
 |                          int64_t count, int64_t total, int cnt, bool Cflag) | 
 | { | 
 |     char s1[64], s2[64], ts[64]; | 
 |  | 
 |     timestr(t, ts, sizeof(ts), Cflag ? VERBOSE_FIXED_TIME : 0); | 
 |     if (!Cflag) { | 
 |         cvtstr((double)total, s1, sizeof(s1)); | 
 |         cvtstr(tdiv((double)total, *t), s2, sizeof(s2)); | 
 |         printf("%s %"PRId64"/%"PRId64" bytes at offset %" PRId64 "\n", | 
 |                op, total, count, offset); | 
 |         printf("%s, %d ops; %s (%s/sec and %.4f ops/sec)\n", | 
 |                s1, cnt, ts, s2, tdiv((double)cnt, *t)); | 
 |     } else {/* bytes,ops,time,bytes/sec,ops/sec */ | 
 |         printf("%"PRId64",%d,%s,%.3f,%.3f\n", | 
 |             total, cnt, ts, | 
 |             tdiv((double)total, *t), | 
 |             tdiv((double)cnt, *t)); | 
 |     } | 
 | } | 
 |  | 
 | /* | 
 |  * Parse multiple length statements for vectored I/O, and construct an I/O | 
 |  * vector matching it. | 
 |  */ | 
 | static void * | 
 | create_iovec(BlockBackend *blk, QEMUIOVector *qiov, char **argv, int nr_iov, | 
 |              int pattern, bool register_buf) | 
 | { | 
 |     size_t *sizes = g_new0(size_t, nr_iov); | 
 |     size_t count = 0; | 
 |     void *buf = NULL; | 
 |     void *p; | 
 |     int i; | 
 |  | 
 |     for (i = 0; i < nr_iov; i++) { | 
 |         char *arg = argv[i]; | 
 |         int64_t len; | 
 |  | 
 |         len = cvtnum(arg); | 
 |         if (len < 0) { | 
 |             print_cvtnum_err(len, arg); | 
 |             goto fail; | 
 |         } | 
 |  | 
 |         if (len > BDRV_REQUEST_MAX_BYTES) { | 
 |             printf("Argument '%s' exceeds maximum size %" PRIu64 "\n", arg, | 
 |                    (uint64_t)BDRV_REQUEST_MAX_BYTES); | 
 |             goto fail; | 
 |         } | 
 |  | 
 |         if (count > BDRV_REQUEST_MAX_BYTES - len) { | 
 |             printf("The total number of bytes exceed the maximum size %" PRIu64 | 
 |                    "\n", (uint64_t)BDRV_REQUEST_MAX_BYTES); | 
 |             goto fail; | 
 |         } | 
 |  | 
 |         sizes[i] = len; | 
 |         count += len; | 
 |     } | 
 |  | 
 |     qemu_iovec_init(qiov, nr_iov); | 
 |  | 
 |     buf = p = qemu_io_alloc(blk, count, pattern, register_buf); | 
 |  | 
 |     for (i = 0; i < nr_iov; i++) { | 
 |         qemu_iovec_add(qiov, p, sizes[i]); | 
 |         p += sizes[i]; | 
 |     } | 
 |  | 
 | fail: | 
 |     g_free(sizes); | 
 |     return buf; | 
 | } | 
 |  | 
 | static int do_pread(BlockBackend *blk, char *buf, int64_t offset, | 
 |                     int64_t bytes, BdrvRequestFlags flags, int64_t *total) | 
 | { | 
 |     int ret; | 
 |  | 
 |     if (bytes > INT_MAX) { | 
 |         return -ERANGE; | 
 |     } | 
 |  | 
 |     ret = blk_pread(blk, offset, bytes, (uint8_t *)buf, flags); | 
 |     if (ret < 0) { | 
 |         return ret; | 
 |     } | 
 |     *total = bytes; | 
 |     return 1; | 
 | } | 
 |  | 
 | static int do_pwrite(BlockBackend *blk, char *buf, int64_t offset, | 
 |                      int64_t bytes, BdrvRequestFlags flags, int64_t *total) | 
 | { | 
 |     int ret; | 
 |  | 
 |     if (bytes > INT_MAX) { | 
 |         return -ERANGE; | 
 |     } | 
 |  | 
 |     ret = blk_pwrite(blk, offset, bytes, (uint8_t *)buf, flags); | 
 |     if (ret < 0) { | 
 |         return ret; | 
 |     } | 
 |     *total = bytes; | 
 |     return 1; | 
 | } | 
 |  | 
 | static int do_pwrite_zeroes(BlockBackend *blk, int64_t offset, | 
 |                                int64_t bytes, BdrvRequestFlags flags, | 
 |                                int64_t *total) | 
 | { | 
 |     int ret = blk_pwrite_zeroes(blk, offset, bytes, | 
 |                                 flags | BDRV_REQ_ZERO_WRITE); | 
 |  | 
 |     if (ret < 0) { | 
 |         return ret; | 
 |     } | 
 |     *total = bytes; | 
 |     return 1; | 
 | } | 
 |  | 
 | static int do_write_compressed(BlockBackend *blk, char *buf, int64_t offset, | 
 |                                int64_t bytes, int64_t *total) | 
 | { | 
 |     int ret; | 
 |  | 
 |     if (bytes > BDRV_REQUEST_MAX_BYTES) { | 
 |         return -ERANGE; | 
 |     } | 
 |  | 
 |     ret = blk_pwrite_compressed(blk, offset, bytes, buf); | 
 |     if (ret < 0) { | 
 |         return ret; | 
 |     } | 
 |     *total = bytes; | 
 |     return 1; | 
 | } | 
 |  | 
 | static int do_load_vmstate(BlockBackend *blk, char *buf, int64_t offset, | 
 |                            int64_t count, int64_t *total) | 
 | { | 
 |     if (count > INT_MAX) { | 
 |         return -ERANGE; | 
 |     } | 
 |  | 
 |     *total = blk_load_vmstate(blk, (uint8_t *)buf, offset, count); | 
 |     if (*total < 0) { | 
 |         return *total; | 
 |     } | 
 |     return 1; | 
 | } | 
 |  | 
 | static int do_save_vmstate(BlockBackend *blk, char *buf, int64_t offset, | 
 |                            int64_t count, int64_t *total) | 
 | { | 
 |     if (count > INT_MAX) { | 
 |         return -ERANGE; | 
 |     } | 
 |  | 
 |     *total = blk_save_vmstate(blk, (uint8_t *)buf, offset, count); | 
 |     if (*total < 0) { | 
 |         return *total; | 
 |     } | 
 |     return 1; | 
 | } | 
 |  | 
 | #define NOT_DONE 0x7fffffff | 
 | static void aio_rw_done(void *opaque, int ret) | 
 | { | 
 |     *(int *)opaque = ret; | 
 | } | 
 |  | 
 | static int do_aio_readv(BlockBackend *blk, QEMUIOVector *qiov, | 
 |                         int64_t offset, BdrvRequestFlags flags, int *total) | 
 | { | 
 |     int async_ret = NOT_DONE; | 
 |  | 
 |     blk_aio_preadv(blk, offset, qiov, flags, aio_rw_done, &async_ret); | 
 |     while (async_ret == NOT_DONE) { | 
 |         main_loop_wait(false); | 
 |     } | 
 |  | 
 |     *total = qiov->size; | 
 |     return async_ret < 0 ? async_ret : 1; | 
 | } | 
 |  | 
 | static int do_aio_writev(BlockBackend *blk, QEMUIOVector *qiov, | 
 |                          int64_t offset, BdrvRequestFlags flags, int *total) | 
 | { | 
 |     int async_ret = NOT_DONE; | 
 |  | 
 |     blk_aio_pwritev(blk, offset, qiov, flags, aio_rw_done, &async_ret); | 
 |     while (async_ret == NOT_DONE) { | 
 |         main_loop_wait(false); | 
 |     } | 
 |  | 
 |     *total = qiov->size; | 
 |     return async_ret < 0 ? async_ret : 1; | 
 | } | 
 |  | 
 | static void read_help(void) | 
 | { | 
 |     printf( | 
 | "\n" | 
 | " reads a range of bytes from the given offset\n" | 
 | "\n" | 
 | " Example:\n" | 
 | " 'read -v 512 1k' - dumps 1 kilobyte read from 512 bytes into the file\n" | 
 | "\n" | 
 | " Reads a segment of the currently open file, optionally dumping it to the\n" | 
 | " standard output stream (with -v option) for subsequent inspection.\n" | 
 | " -b, -- read from the VM state rather than the virtual disk\n" | 
 | " -C, -- report statistics in a machine parsable format\n" | 
 | " -l, -- length for pattern verification (only with -P)\n" | 
 | " -p, -- ignored for backwards compatibility\n" | 
 | " -P, -- use a pattern to verify read data\n" | 
 | " -q, -- quiet mode, do not show I/O statistics\n" | 
 | " -r, -- register I/O buffer\n" | 
 | " -s, -- start offset for pattern verification (only with -P)\n" | 
 | " -v, -- dump buffer to standard output\n" | 
 | "\n"); | 
 | } | 
 |  | 
 | static int read_f(BlockBackend *blk, int argc, char **argv); | 
 |  | 
 | static const cmdinfo_t read_cmd = { | 
 |     .name       = "read", | 
 |     .altname    = "r", | 
 |     .cfunc      = read_f, | 
 |     .argmin     = 2, | 
 |     .argmax     = -1, | 
 |     .args       = "[-abCqrv] [-P pattern [-s off] [-l len]] off len", | 
 |     .oneline    = "reads a number of bytes at a specified offset", | 
 |     .help       = read_help, | 
 | }; | 
 |  | 
 | static int read_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     struct timespec t1, t2; | 
 |     bool Cflag = false, qflag = false, vflag = false; | 
 |     bool Pflag = false, sflag = false, lflag = false, bflag = false; | 
 |     int c, cnt, ret; | 
 |     char *buf; | 
 |     int64_t offset; | 
 |     int64_t count; | 
 |     /* Some compilers get confused and warn if this is not initialized.  */ | 
 |     int64_t total = 0; | 
 |     int pattern = 0; | 
 |     int64_t pattern_offset = 0, pattern_count = 0; | 
 |     BdrvRequestFlags flags = 0; | 
 |  | 
 |     while ((c = getopt(argc, argv, "bCl:pP:qrs:v")) != -1) { | 
 |         switch (c) { | 
 |         case 'b': | 
 |             bflag = true; | 
 |             break; | 
 |         case 'C': | 
 |             Cflag = true; | 
 |             break; | 
 |         case 'l': | 
 |             lflag = true; | 
 |             pattern_count = cvtnum(optarg); | 
 |             if (pattern_count < 0) { | 
 |                 print_cvtnum_err(pattern_count, optarg); | 
 |                 return pattern_count; | 
 |             } | 
 |             break; | 
 |         case 'p': | 
 |             /* Ignored for backwards compatibility */ | 
 |             break; | 
 |         case 'P': | 
 |             Pflag = true; | 
 |             pattern = parse_pattern(optarg); | 
 |             if (pattern < 0) { | 
 |                 return -EINVAL; | 
 |             } | 
 |             break; | 
 |         case 'q': | 
 |             qflag = true; | 
 |             break; | 
 |         case 'r': | 
 |             flags |= BDRV_REQ_REGISTERED_BUF; | 
 |             break; | 
 |         case 's': | 
 |             sflag = true; | 
 |             pattern_offset = cvtnum(optarg); | 
 |             if (pattern_offset < 0) { | 
 |                 print_cvtnum_err(pattern_offset, optarg); | 
 |                 return pattern_offset; | 
 |             } | 
 |             break; | 
 |         case 'v': | 
 |             vflag = true; | 
 |             break; | 
 |         default: | 
 |             qemuio_command_usage(&read_cmd); | 
 |             return -EINVAL; | 
 |         } | 
 |     } | 
 |  | 
 |     if (optind != argc - 2) { | 
 |         qemuio_command_usage(&read_cmd); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     offset = cvtnum(argv[optind]); | 
 |     if (offset < 0) { | 
 |         print_cvtnum_err(offset, argv[optind]); | 
 |         return offset; | 
 |     } | 
 |  | 
 |     optind++; | 
 |     count = cvtnum(argv[optind]); | 
 |     if (count < 0) { | 
 |         print_cvtnum_err(count, argv[optind]); | 
 |         return count; | 
 |     } else if (count > BDRV_REQUEST_MAX_BYTES) { | 
 |         printf("length cannot exceed %" PRIu64 ", given %s\n", | 
 |                (uint64_t)BDRV_REQUEST_MAX_BYTES, argv[optind]); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if (!Pflag && (lflag || sflag)) { | 
 |         qemuio_command_usage(&read_cmd); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if (!lflag) { | 
 |         pattern_count = count - pattern_offset; | 
 |     } | 
 |  | 
 |     if ((pattern_count < 0) || (pattern_count + pattern_offset > count))  { | 
 |         printf("pattern verification range exceeds end of read data\n"); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if (bflag) { | 
 |         if (!QEMU_IS_ALIGNED(offset, BDRV_SECTOR_SIZE)) { | 
 |             printf("%" PRId64 " is not a sector-aligned value for 'offset'\n", | 
 |                    offset); | 
 |             return -EINVAL; | 
 |         } | 
 |         if (!QEMU_IS_ALIGNED(count, BDRV_SECTOR_SIZE)) { | 
 |             printf("%"PRId64" is not a sector-aligned value for 'count'\n", | 
 |                    count); | 
 |             return -EINVAL; | 
 |         } | 
 |         if (flags & BDRV_REQ_REGISTERED_BUF) { | 
 |             printf("I/O buffer registration is not supported when reading " | 
 |                     "from vmstate\n"); | 
 |             return -EINVAL; | 
 |         } | 
 |     } | 
 |  | 
 |     buf = qemu_io_alloc(blk, count, 0xab, flags & BDRV_REQ_REGISTERED_BUF); | 
 |  | 
 |     clock_gettime(CLOCK_MONOTONIC, &t1); | 
 |     if (bflag) { | 
 |         ret = do_load_vmstate(blk, buf, offset, count, &total); | 
 |     } else { | 
 |         ret = do_pread(blk, buf, offset, count, flags, &total); | 
 |     } | 
 |     clock_gettime(CLOCK_MONOTONIC, &t2); | 
 |  | 
 |     if (ret < 0) { | 
 |         printf("read failed: %s\n", strerror(-ret)); | 
 |         goto out; | 
 |     } | 
 |     cnt = ret; | 
 |  | 
 |     ret = 0; | 
 |  | 
 |     if (Pflag) { | 
 |         void *cmp_buf = g_malloc(pattern_count); | 
 |         memset(cmp_buf, pattern, pattern_count); | 
 |         if (memcmp(buf + pattern_offset, cmp_buf, pattern_count)) { | 
 |             printf("Pattern verification failed at offset %" | 
 |                    PRId64 ", %"PRId64" bytes\n", | 
 |                    offset + pattern_offset, pattern_count); | 
 |             ret = -EINVAL; | 
 |         } | 
 |         g_free(cmp_buf); | 
 |     } | 
 |  | 
 |     if (qflag) { | 
 |         goto out; | 
 |     } | 
 |  | 
 |     if (vflag) { | 
 |         dump_buffer(buf, offset, count); | 
 |     } | 
 |  | 
 |     /* Finally, report back -- -C gives a parsable format */ | 
 |     t2 = tsub(t2, t1); | 
 |     print_report("read", &t2, offset, count, total, cnt, Cflag); | 
 |  | 
 | out: | 
 |     qemu_io_free(blk, buf, count, flags & BDRV_REQ_REGISTERED_BUF); | 
 |     return ret; | 
 | } | 
 |  | 
 | static void readv_help(void) | 
 | { | 
 |     printf( | 
 | "\n" | 
 | " reads a range of bytes from the given offset into multiple buffers\n" | 
 | "\n" | 
 | " Example:\n" | 
 | " 'readv -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n" | 
 | "\n" | 
 | " Reads a segment of the currently open file, optionally dumping it to the\n" | 
 | " standard output stream (with -v option) for subsequent inspection.\n" | 
 | " Uses multiple iovec buffers if more than one byte range is specified.\n" | 
 | " -C, -- report statistics in a machine parsable format\n" | 
 | " -P, -- use a pattern to verify read data\n" | 
 | " -q, -- quiet mode, do not show I/O statistics\n" | 
 | " -r, -- register I/O buffer\n" | 
 | " -v, -- dump buffer to standard output\n" | 
 | "\n"); | 
 | } | 
 |  | 
 | static int readv_f(BlockBackend *blk, int argc, char **argv); | 
 |  | 
 | static const cmdinfo_t readv_cmd = { | 
 |     .name       = "readv", | 
 |     .cfunc      = readv_f, | 
 |     .argmin     = 2, | 
 |     .argmax     = -1, | 
 |     .args       = "[-Cqrv] [-P pattern] off len [len..]", | 
 |     .oneline    = "reads a number of bytes at a specified offset", | 
 |     .help       = readv_help, | 
 | }; | 
 |  | 
 | static int readv_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     struct timespec t1, t2; | 
 |     bool Cflag = false, qflag = false, vflag = false; | 
 |     int c, cnt, ret; | 
 |     char *buf; | 
 |     int64_t offset; | 
 |     /* Some compilers get confused and warn if this is not initialized.  */ | 
 |     int total = 0; | 
 |     int nr_iov; | 
 |     QEMUIOVector qiov; | 
 |     int pattern = 0; | 
 |     bool Pflag = false; | 
 |     BdrvRequestFlags flags = 0; | 
 |  | 
 |     while ((c = getopt(argc, argv, "CP:qrv")) != -1) { | 
 |         switch (c) { | 
 |         case 'C': | 
 |             Cflag = true; | 
 |             break; | 
 |         case 'P': | 
 |             Pflag = true; | 
 |             pattern = parse_pattern(optarg); | 
 |             if (pattern < 0) { | 
 |                 return -EINVAL; | 
 |             } | 
 |             break; | 
 |         case 'q': | 
 |             qflag = true; | 
 |             break; | 
 |         case 'r': | 
 |             flags |= BDRV_REQ_REGISTERED_BUF; | 
 |             break; | 
 |         case 'v': | 
 |             vflag = true; | 
 |             break; | 
 |         default: | 
 |             qemuio_command_usage(&readv_cmd); | 
 |             return -EINVAL; | 
 |         } | 
 |     } | 
 |  | 
 |     if (optind > argc - 2) { | 
 |         qemuio_command_usage(&readv_cmd); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |  | 
 |     offset = cvtnum(argv[optind]); | 
 |     if (offset < 0) { | 
 |         print_cvtnum_err(offset, argv[optind]); | 
 |         return offset; | 
 |     } | 
 |     optind++; | 
 |  | 
 |     nr_iov = argc - optind; | 
 |     buf = create_iovec(blk, &qiov, &argv[optind], nr_iov, 0xab, | 
 |                        flags & BDRV_REQ_REGISTERED_BUF); | 
 |     if (buf == NULL) { | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     clock_gettime(CLOCK_MONOTONIC, &t1); | 
 |     ret = do_aio_readv(blk, &qiov, offset, flags, &total); | 
 |     clock_gettime(CLOCK_MONOTONIC, &t2); | 
 |  | 
 |     if (ret < 0) { | 
 |         printf("readv failed: %s\n", strerror(-ret)); | 
 |         goto out; | 
 |     } | 
 |     cnt = ret; | 
 |  | 
 |     ret = 0; | 
 |  | 
 |     if (Pflag) { | 
 |         void *cmp_buf = g_malloc(qiov.size); | 
 |         memset(cmp_buf, pattern, qiov.size); | 
 |         if (memcmp(buf, cmp_buf, qiov.size)) { | 
 |             printf("Pattern verification failed at offset %" | 
 |                    PRId64 ", %zu bytes\n", offset, qiov.size); | 
 |             ret = -EINVAL; | 
 |         } | 
 |         g_free(cmp_buf); | 
 |     } | 
 |  | 
 |     if (qflag) { | 
 |         goto out; | 
 |     } | 
 |  | 
 |     if (vflag) { | 
 |         dump_buffer(buf, offset, qiov.size); | 
 |     } | 
 |  | 
 |     /* Finally, report back -- -C gives a parsable format */ | 
 |     t2 = tsub(t2, t1); | 
 |     print_report("read", &t2, offset, qiov.size, total, cnt, Cflag); | 
 |  | 
 | out: | 
 |     qemu_io_free(blk, buf, qiov.size, flags & BDRV_REQ_REGISTERED_BUF); | 
 |     qemu_iovec_destroy(&qiov); | 
 |     return ret; | 
 | } | 
 |  | 
 | static void write_help(void) | 
 | { | 
 |     printf( | 
 | "\n" | 
 | " writes a range of bytes from the given offset\n" | 
 | "\n" | 
 | " Example:\n" | 
 | " 'write 512 1k' - writes 1 kilobyte at 512 bytes into the open file\n" | 
 | "\n" | 
 | " Writes into a segment of the currently open file, using a buffer\n" | 
 | " filled with a set pattern (0xcdcdcdcd).\n" | 
 | " -b, -- write to the VM state rather than the virtual disk\n" | 
 | " -c, -- write compressed data with blk_write_compressed\n" | 
 | " -C, -- report statistics in a machine parsable format\n" | 
 | " -f, -- use Force Unit Access semantics\n" | 
 | " -n, -- with -z, don't allow slow fallback\n" | 
 | " -p, -- ignored for backwards compatibility\n" | 
 | " -P, -- use different pattern to fill file\n" | 
 | " -q, -- quiet mode, do not show I/O statistics\n" | 
 | " -r, -- register I/O buffer\n" | 
 | " -s, -- use a pattern file to fill the write buffer\n" | 
 | " -u, -- with -z, allow unmapping\n" | 
 | " -z, -- write zeroes using blk_pwrite_zeroes\n" | 
 | "\n"); | 
 | } | 
 |  | 
 | static int write_f(BlockBackend *blk, int argc, char **argv); | 
 |  | 
 | static const cmdinfo_t write_cmd = { | 
 |     .name       = "write", | 
 |     .altname    = "w", | 
 |     .cfunc      = write_f, | 
 |     .perm       = BLK_PERM_WRITE, | 
 |     .argmin     = 2, | 
 |     .argmax     = -1, | 
 |     .args       = "[-bcCfnqruz] [-P pattern | -s source_file] off len", | 
 |     .oneline    = "writes a number of bytes at a specified offset", | 
 |     .help       = write_help, | 
 | }; | 
 |  | 
 | static int write_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     struct timespec t1, t2; | 
 |     bool Cflag = false, qflag = false, bflag = false; | 
 |     bool Pflag = false, zflag = false, cflag = false, sflag = false; | 
 |     BdrvRequestFlags flags = 0; | 
 |     int c, cnt, ret; | 
 |     char *buf = NULL; | 
 |     int64_t offset; | 
 |     int64_t count; | 
 |     /* Some compilers get confused and warn if this is not initialized.  */ | 
 |     int64_t total = 0; | 
 |     int pattern = 0xcd; | 
 |     const char *file_name = NULL; | 
 |  | 
 |     while ((c = getopt(argc, argv, "bcCfnpP:qrs:uz")) != -1) { | 
 |         switch (c) { | 
 |         case 'b': | 
 |             bflag = true; | 
 |             break; | 
 |         case 'c': | 
 |             cflag = true; | 
 |             break; | 
 |         case 'C': | 
 |             Cflag = true; | 
 |             break; | 
 |         case 'f': | 
 |             flags |= BDRV_REQ_FUA; | 
 |             break; | 
 |         case 'n': | 
 |             flags |= BDRV_REQ_NO_FALLBACK; | 
 |             break; | 
 |         case 'p': | 
 |             /* Ignored for backwards compatibility */ | 
 |             break; | 
 |         case 'P': | 
 |             Pflag = true; | 
 |             pattern = parse_pattern(optarg); | 
 |             if (pattern < 0) { | 
 |                 return -EINVAL; | 
 |             } | 
 |             break; | 
 |         case 'q': | 
 |             qflag = true; | 
 |             break; | 
 |         case 'r': | 
 |             flags |= BDRV_REQ_REGISTERED_BUF; | 
 |             break; | 
 |         case 's': | 
 |             sflag = true; | 
 |             file_name = optarg; | 
 |             break; | 
 |         case 'u': | 
 |             flags |= BDRV_REQ_MAY_UNMAP; | 
 |             break; | 
 |         case 'z': | 
 |             zflag = true; | 
 |             break; | 
 |         default: | 
 |             qemuio_command_usage(&write_cmd); | 
 |             return -EINVAL; | 
 |         } | 
 |     } | 
 |  | 
 |     if (optind != argc - 2) { | 
 |         qemuio_command_usage(&write_cmd); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if (bflag && zflag) { | 
 |         printf("-b and -z cannot be specified at the same time\n"); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if ((flags & BDRV_REQ_FUA) && (bflag || cflag)) { | 
 |         printf("-f and -b or -c cannot be specified at the same time\n"); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if ((flags & BDRV_REQ_NO_FALLBACK) && !zflag) { | 
 |         printf("-n requires -z to be specified\n"); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if ((flags & BDRV_REQ_MAY_UNMAP) && !zflag) { | 
 |         printf("-u requires -z to be specified\n"); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if (zflag + Pflag + sflag > 1) { | 
 |         printf("Only one of -z, -P, and -s " | 
 |                "can be specified at the same time\n"); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     offset = cvtnum(argv[optind]); | 
 |     if (offset < 0) { | 
 |         print_cvtnum_err(offset, argv[optind]); | 
 |         return offset; | 
 |     } | 
 |  | 
 |     optind++; | 
 |     count = cvtnum(argv[optind]); | 
 |     if (count < 0) { | 
 |         print_cvtnum_err(count, argv[optind]); | 
 |         return count; | 
 |     } else if (count > BDRV_REQUEST_MAX_BYTES && | 
 |                !(flags & BDRV_REQ_NO_FALLBACK)) { | 
 |         printf("length cannot exceed %" PRIu64 " without -n, given %s\n", | 
 |                (uint64_t)BDRV_REQUEST_MAX_BYTES, argv[optind]); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if (bflag || cflag) { | 
 |         if (!QEMU_IS_ALIGNED(offset, BDRV_SECTOR_SIZE)) { | 
 |             printf("%" PRId64 " is not a sector-aligned value for 'offset'\n", | 
 |                    offset); | 
 |             return -EINVAL; | 
 |         } | 
 |  | 
 |         if (!QEMU_IS_ALIGNED(count, BDRV_SECTOR_SIZE)) { | 
 |             printf("%"PRId64" is not a sector-aligned value for 'count'\n", | 
 |                    count); | 
 |             return -EINVAL; | 
 |         } | 
 |     } | 
 |  | 
 |     if (zflag) { | 
 |         if (flags & BDRV_REQ_REGISTERED_BUF) { | 
 |             printf("cannot combine zero write with registered I/O buffer\n"); | 
 |             return -EINVAL; | 
 |         } | 
 |     } else { | 
 |         if (sflag) { | 
 |             buf = qemu_io_alloc_from_file(blk, count, file_name, | 
 |                                           flags & BDRV_REQ_REGISTERED_BUF); | 
 |             if (!buf) { | 
 |                 return -EINVAL; | 
 |             } | 
 |         } else { | 
 |             buf = qemu_io_alloc(blk, count, pattern, | 
 |                                 flags & BDRV_REQ_REGISTERED_BUF); | 
 |         } | 
 |     } | 
 |  | 
 |     clock_gettime(CLOCK_MONOTONIC, &t1); | 
 |     if (bflag) { | 
 |         ret = do_save_vmstate(blk, buf, offset, count, &total); | 
 |     } else if (zflag) { | 
 |         ret = do_pwrite_zeroes(blk, offset, count, flags, &total); | 
 |     } else if (cflag) { | 
 |         ret = do_write_compressed(blk, buf, offset, count, &total); | 
 |     } else { | 
 |         ret = do_pwrite(blk, buf, offset, count, flags, &total); | 
 |     } | 
 |     clock_gettime(CLOCK_MONOTONIC, &t2); | 
 |  | 
 |     if (ret < 0) { | 
 |         printf("write failed: %s\n", strerror(-ret)); | 
 |         goto out; | 
 |     } | 
 |     cnt = ret; | 
 |  | 
 |     ret = 0; | 
 |  | 
 |     if (qflag) { | 
 |         goto out; | 
 |     } | 
 |  | 
 |     /* Finally, report back -- -C gives a parsable format */ | 
 |     t2 = tsub(t2, t1); | 
 |     print_report("wrote", &t2, offset, count, total, cnt, Cflag); | 
 |  | 
 | out: | 
 |     if (!zflag) { | 
 |         qemu_io_free(blk, buf, count, flags & BDRV_REQ_REGISTERED_BUF); | 
 |     } | 
 |     return ret; | 
 | } | 
 |  | 
 | static void | 
 | writev_help(void) | 
 | { | 
 |     printf( | 
 | "\n" | 
 | " writes a range of bytes from the given offset source from multiple buffers\n" | 
 | "\n" | 
 | " Example:\n" | 
 | " 'writev 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n" | 
 | "\n" | 
 | " Writes into a segment of the currently open file, using a buffer\n" | 
 | " filled with a set pattern (0xcdcdcdcd).\n" | 
 | " -C, -- report statistics in a machine parsable format\n" | 
 | " -f, -- use Force Unit Access semantics\n" | 
 | " -P, -- use different pattern to fill file\n" | 
 | " -q, -- quiet mode, do not show I/O statistics\n" | 
 | " -r, -- register I/O buffer\n" | 
 | "\n"); | 
 | } | 
 |  | 
 | static int writev_f(BlockBackend *blk, int argc, char **argv); | 
 |  | 
 | static const cmdinfo_t writev_cmd = { | 
 |     .name       = "writev", | 
 |     .cfunc      = writev_f, | 
 |     .perm       = BLK_PERM_WRITE, | 
 |     .argmin     = 2, | 
 |     .argmax     = -1, | 
 |     .args       = "[-Cfqr] [-P pattern] off len [len..]", | 
 |     .oneline    = "writes a number of bytes at a specified offset", | 
 |     .help       = writev_help, | 
 | }; | 
 |  | 
 | static int writev_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     struct timespec t1, t2; | 
 |     bool Cflag = false, qflag = false; | 
 |     BdrvRequestFlags flags = 0; | 
 |     int c, cnt, ret; | 
 |     char *buf; | 
 |     int64_t offset; | 
 |     /* Some compilers get confused and warn if this is not initialized.  */ | 
 |     int total = 0; | 
 |     int nr_iov; | 
 |     int pattern = 0xcd; | 
 |     QEMUIOVector qiov; | 
 |  | 
 |     while ((c = getopt(argc, argv, "CfP:qr")) != -1) { | 
 |         switch (c) { | 
 |         case 'C': | 
 |             Cflag = true; | 
 |             break; | 
 |         case 'f': | 
 |             flags |= BDRV_REQ_FUA; | 
 |             break; | 
 |         case 'q': | 
 |             qflag = true; | 
 |             break; | 
 |         case 'r': | 
 |             flags |= BDRV_REQ_REGISTERED_BUF; | 
 |             break; | 
 |         case 'P': | 
 |             pattern = parse_pattern(optarg); | 
 |             if (pattern < 0) { | 
 |                 return -EINVAL; | 
 |             } | 
 |             break; | 
 |         default: | 
 |             qemuio_command_usage(&writev_cmd); | 
 |             return -EINVAL; | 
 |         } | 
 |     } | 
 |  | 
 |     if (optind > argc - 2) { | 
 |         qemuio_command_usage(&writev_cmd); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     offset = cvtnum(argv[optind]); | 
 |     if (offset < 0) { | 
 |         print_cvtnum_err(offset, argv[optind]); | 
 |         return offset; | 
 |     } | 
 |     optind++; | 
 |  | 
 |     nr_iov = argc - optind; | 
 |     buf = create_iovec(blk, &qiov, &argv[optind], nr_iov, pattern, | 
 |                        flags & BDRV_REQ_REGISTERED_BUF); | 
 |     if (buf == NULL) { | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     clock_gettime(CLOCK_MONOTONIC, &t1); | 
 |     ret = do_aio_writev(blk, &qiov, offset, flags, &total); | 
 |     clock_gettime(CLOCK_MONOTONIC, &t2); | 
 |  | 
 |     if (ret < 0) { | 
 |         printf("writev failed: %s\n", strerror(-ret)); | 
 |         goto out; | 
 |     } | 
 |     cnt = ret; | 
 |  | 
 |     ret = 0; | 
 |  | 
 |     if (qflag) { | 
 |         goto out; | 
 |     } | 
 |  | 
 |     /* Finally, report back -- -C gives a parsable format */ | 
 |     t2 = tsub(t2, t1); | 
 |     print_report("wrote", &t2, offset, qiov.size, total, cnt, Cflag); | 
 | out: | 
 |     qemu_io_free(blk, buf, qiov.size, flags & BDRV_REQ_REGISTERED_BUF); | 
 |     qemu_iovec_destroy(&qiov); | 
 |     return ret; | 
 | } | 
 |  | 
 | struct aio_ctx { | 
 |     BlockBackend *blk; | 
 |     QEMUIOVector qiov; | 
 |     int64_t offset; | 
 |     char *buf; | 
 |     bool qflag; | 
 |     bool vflag; | 
 |     bool Cflag; | 
 |     bool Pflag; | 
 |     bool zflag; | 
 |     BlockAcctCookie acct; | 
 |     int pattern; | 
 |     BdrvRequestFlags flags; | 
 |     struct timespec t1; | 
 | }; | 
 |  | 
 | static void aio_write_done(void *opaque, int ret) | 
 | { | 
 |     struct aio_ctx *ctx = opaque; | 
 |     struct timespec t2; | 
 |  | 
 |     clock_gettime(CLOCK_MONOTONIC, &t2); | 
 |  | 
 |  | 
 |     if (ret < 0) { | 
 |         printf("aio_write failed: %s\n", strerror(-ret)); | 
 |         block_acct_failed(blk_get_stats(ctx->blk), &ctx->acct); | 
 |         goto out; | 
 |     } | 
 |  | 
 |     block_acct_done(blk_get_stats(ctx->blk), &ctx->acct); | 
 |  | 
 |     if (ctx->qflag) { | 
 |         goto out; | 
 |     } | 
 |  | 
 |     /* Finally, report back -- -C gives a parsable format */ | 
 |     t2 = tsub(t2, ctx->t1); | 
 |     print_report("wrote", &t2, ctx->offset, ctx->qiov.size, | 
 |                  ctx->qiov.size, 1, ctx->Cflag); | 
 | out: | 
 |     if (!ctx->zflag) { | 
 |         qemu_io_free(ctx->blk, ctx->buf, ctx->qiov.size, | 
 |                      ctx->flags & BDRV_REQ_REGISTERED_BUF); | 
 |         qemu_iovec_destroy(&ctx->qiov); | 
 |     } | 
 |     g_free(ctx); | 
 | } | 
 |  | 
 | static void aio_read_done(void *opaque, int ret) | 
 | { | 
 |     struct aio_ctx *ctx = opaque; | 
 |     struct timespec t2; | 
 |  | 
 |     clock_gettime(CLOCK_MONOTONIC, &t2); | 
 |  | 
 |     if (ret < 0) { | 
 |         printf("readv failed: %s\n", strerror(-ret)); | 
 |         block_acct_failed(blk_get_stats(ctx->blk), &ctx->acct); | 
 |         goto out; | 
 |     } | 
 |  | 
 |     if (ctx->Pflag) { | 
 |         void *cmp_buf = g_malloc(ctx->qiov.size); | 
 |  | 
 |         memset(cmp_buf, ctx->pattern, ctx->qiov.size); | 
 |         if (memcmp(ctx->buf, cmp_buf, ctx->qiov.size)) { | 
 |             printf("Pattern verification failed at offset %" | 
 |                    PRId64 ", %zu bytes\n", ctx->offset, ctx->qiov.size); | 
 |         } | 
 |         g_free(cmp_buf); | 
 |     } | 
 |  | 
 |     block_acct_done(blk_get_stats(ctx->blk), &ctx->acct); | 
 |  | 
 |     if (ctx->qflag) { | 
 |         goto out; | 
 |     } | 
 |  | 
 |     if (ctx->vflag) { | 
 |         dump_buffer(ctx->buf, ctx->offset, ctx->qiov.size); | 
 |     } | 
 |  | 
 |     /* Finally, report back -- -C gives a parsable format */ | 
 |     t2 = tsub(t2, ctx->t1); | 
 |     print_report("read", &t2, ctx->offset, ctx->qiov.size, | 
 |                  ctx->qiov.size, 1, ctx->Cflag); | 
 | out: | 
 |     qemu_io_free(ctx->blk, ctx->buf, ctx->qiov.size, | 
 |                  ctx->flags & BDRV_REQ_REGISTERED_BUF); | 
 |     qemu_iovec_destroy(&ctx->qiov); | 
 |     g_free(ctx); | 
 | } | 
 |  | 
 | static void aio_read_help(void) | 
 | { | 
 |     printf( | 
 | "\n" | 
 | " asynchronously reads a range of bytes from the given offset\n" | 
 | "\n" | 
 | " Example:\n" | 
 | " 'aio_read -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n" | 
 | "\n" | 
 | " Reads a segment of the currently open file, optionally dumping it to the\n" | 
 | " standard output stream (with -v option) for subsequent inspection.\n" | 
 | " The read is performed asynchronously and the aio_flush command must be\n" | 
 | " used to ensure all outstanding aio requests have been completed.\n" | 
 | " Note that due to its asynchronous nature, this command will be\n" | 
 | " considered successful once the request is submitted, independently\n" | 
 | " of potential I/O errors or pattern mismatches.\n" | 
 | " -C, -- report statistics in a machine parsable format\n" | 
 | " -i, -- treat request as invalid, for exercising stats\n" | 
 | " -P, -- use a pattern to verify read data\n" | 
 | " -q, -- quiet mode, do not show I/O statistics\n" | 
 | " -r, -- register I/O buffer\n" | 
 | " -v, -- dump buffer to standard output\n" | 
 | "\n"); | 
 | } | 
 |  | 
 | static int aio_read_f(BlockBackend *blk, int argc, char **argv); | 
 |  | 
 | static const cmdinfo_t aio_read_cmd = { | 
 |     .name       = "aio_read", | 
 |     .cfunc      = aio_read_f, | 
 |     .argmin     = 2, | 
 |     .argmax     = -1, | 
 |     .args       = "[-Ciqrv] [-P pattern] off len [len..]", | 
 |     .oneline    = "asynchronously reads a number of bytes", | 
 |     .help       = aio_read_help, | 
 | }; | 
 |  | 
 | static int aio_read_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int nr_iov, c; | 
 |     struct aio_ctx *ctx = g_new0(struct aio_ctx, 1); | 
 |  | 
 |     ctx->blk = blk; | 
 |     while ((c = getopt(argc, argv, "CiP:qrv")) != -1) { | 
 |         switch (c) { | 
 |         case 'C': | 
 |             ctx->Cflag = true; | 
 |             break; | 
 |         case 'P': | 
 |             ctx->Pflag = true; | 
 |             ctx->pattern = parse_pattern(optarg); | 
 |             if (ctx->pattern < 0) { | 
 |                 g_free(ctx); | 
 |                 return -EINVAL; | 
 |             } | 
 |             break; | 
 |         case 'i': | 
 |             printf("injecting invalid read request\n"); | 
 |             block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_READ); | 
 |             g_free(ctx); | 
 |             return 0; | 
 |         case 'q': | 
 |             ctx->qflag = true; | 
 |             break; | 
 |         case 'r': | 
 |             ctx->flags |= BDRV_REQ_REGISTERED_BUF; | 
 |             break; | 
 |         case 'v': | 
 |             ctx->vflag = true; | 
 |             break; | 
 |         default: | 
 |             g_free(ctx); | 
 |             qemuio_command_usage(&aio_read_cmd); | 
 |             return -EINVAL; | 
 |         } | 
 |     } | 
 |  | 
 |     if (optind > argc - 2) { | 
 |         g_free(ctx); | 
 |         qemuio_command_usage(&aio_read_cmd); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     ctx->offset = cvtnum(argv[optind]); | 
 |     if (ctx->offset < 0) { | 
 |         int ret = ctx->offset; | 
 |         print_cvtnum_err(ret, argv[optind]); | 
 |         g_free(ctx); | 
 |         return ret; | 
 |     } | 
 |     optind++; | 
 |  | 
 |     nr_iov = argc - optind; | 
 |     ctx->buf = create_iovec(blk, &ctx->qiov, &argv[optind], nr_iov, 0xab, | 
 |                             ctx->flags & BDRV_REQ_REGISTERED_BUF); | 
 |     if (ctx->buf == NULL) { | 
 |         block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_READ); | 
 |         g_free(ctx); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     clock_gettime(CLOCK_MONOTONIC, &ctx->t1); | 
 |     block_acct_start(blk_get_stats(blk), &ctx->acct, ctx->qiov.size, | 
 |                      BLOCK_ACCT_READ); | 
 |     blk_aio_preadv(blk, ctx->offset, &ctx->qiov, ctx->flags, aio_read_done, | 
 |                    ctx); | 
 |     return 0; | 
 | } | 
 |  | 
 | static void aio_write_help(void) | 
 | { | 
 |     printf( | 
 | "\n" | 
 | " asynchronously writes a range of bytes from the given offset source\n" | 
 | " from multiple buffers\n" | 
 | "\n" | 
 | " Example:\n" | 
 | " 'aio_write 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n" | 
 | "\n" | 
 | " Writes into a segment of the currently open file, using a buffer\n" | 
 | " filled with a set pattern (0xcdcdcdcd).\n" | 
 | " The write is performed asynchronously and the aio_flush command must be\n" | 
 | " used to ensure all outstanding aio requests have been completed.\n" | 
 | " Note that due to its asynchronous nature, this command will be\n" | 
 | " considered successful once the request is submitted, independently\n" | 
 | " of potential I/O errors or pattern mismatches.\n" | 
 | " -C, -- report statistics in a machine parsable format\n" | 
 | " -f, -- use Force Unit Access semantics\n" | 
 | " -i, -- treat request as invalid, for exercising stats\n" | 
 | " -P, -- use different pattern to fill file\n" | 
 | " -q, -- quiet mode, do not show I/O statistics\n" | 
 | " -r, -- register I/O buffer\n" | 
 | " -u, -- with -z, allow unmapping\n" | 
 | " -z, -- write zeroes using blk_aio_pwrite_zeroes\n" | 
 | "\n"); | 
 | } | 
 |  | 
 | static int aio_write_f(BlockBackend *blk, int argc, char **argv); | 
 |  | 
 | static const cmdinfo_t aio_write_cmd = { | 
 |     .name       = "aio_write", | 
 |     .cfunc      = aio_write_f, | 
 |     .perm       = BLK_PERM_WRITE, | 
 |     .argmin     = 2, | 
 |     .argmax     = -1, | 
 |     .args       = "[-Cfiqruz] [-P pattern] off len [len..]", | 
 |     .oneline    = "asynchronously writes a number of bytes", | 
 |     .help       = aio_write_help, | 
 | }; | 
 |  | 
 | static int aio_write_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int nr_iov, c; | 
 |     int pattern = 0xcd; | 
 |     struct aio_ctx *ctx = g_new0(struct aio_ctx, 1); | 
 |  | 
 |     ctx->blk = blk; | 
 |     while ((c = getopt(argc, argv, "CfiP:qruz")) != -1) { | 
 |         switch (c) { | 
 |         case 'C': | 
 |             ctx->Cflag = true; | 
 |             break; | 
 |         case 'f': | 
 |             ctx->flags |= BDRV_REQ_FUA; | 
 |             break; | 
 |         case 'q': | 
 |             ctx->qflag = true; | 
 |             break; | 
 |         case 'r': | 
 |             ctx->flags |= BDRV_REQ_REGISTERED_BUF; | 
 |             break; | 
 |         case 'u': | 
 |             ctx->flags |= BDRV_REQ_MAY_UNMAP; | 
 |             break; | 
 |         case 'P': | 
 |             pattern = parse_pattern(optarg); | 
 |             if (pattern < 0) { | 
 |                 g_free(ctx); | 
 |                 return -EINVAL; | 
 |             } | 
 |             break; | 
 |         case 'i': | 
 |             printf("injecting invalid write request\n"); | 
 |             block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_WRITE); | 
 |             g_free(ctx); | 
 |             return 0; | 
 |         case 'z': | 
 |             ctx->zflag = true; | 
 |             break; | 
 |         default: | 
 |             g_free(ctx); | 
 |             qemuio_command_usage(&aio_write_cmd); | 
 |             return -EINVAL; | 
 |         } | 
 |     } | 
 |  | 
 |     if (optind > argc - 2) { | 
 |         g_free(ctx); | 
 |         qemuio_command_usage(&aio_write_cmd); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if (ctx->zflag && optind != argc - 2) { | 
 |         printf("-z supports only a single length parameter\n"); | 
 |         g_free(ctx); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if ((ctx->flags & BDRV_REQ_MAY_UNMAP) && !ctx->zflag) { | 
 |         printf("-u requires -z to be specified\n"); | 
 |         g_free(ctx); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if (ctx->zflag && ctx->Pflag) { | 
 |         printf("-z and -P cannot be specified at the same time\n"); | 
 |         g_free(ctx); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if (ctx->zflag && (ctx->flags & BDRV_REQ_REGISTERED_BUF)) { | 
 |         printf("cannot combine zero write with registered I/O buffer\n"); | 
 |         g_free(ctx); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     ctx->offset = cvtnum(argv[optind]); | 
 |     if (ctx->offset < 0) { | 
 |         int ret = ctx->offset; | 
 |         print_cvtnum_err(ret, argv[optind]); | 
 |         g_free(ctx); | 
 |         return ret; | 
 |     } | 
 |     optind++; | 
 |  | 
 |     if (ctx->zflag) { | 
 |         int64_t count = cvtnum(argv[optind]); | 
 |         if (count < 0) { | 
 |             print_cvtnum_err(count, argv[optind]); | 
 |             g_free(ctx); | 
 |             return count; | 
 |         } | 
 |  | 
 |         ctx->qiov.size = count; | 
 |         blk_aio_pwrite_zeroes(blk, ctx->offset, count, ctx->flags, | 
 |                               aio_write_done, ctx); | 
 |     } else { | 
 |         nr_iov = argc - optind; | 
 |         ctx->buf = create_iovec(blk, &ctx->qiov, &argv[optind], nr_iov, | 
 |                                 pattern, ctx->flags & BDRV_REQ_REGISTERED_BUF); | 
 |         if (ctx->buf == NULL) { | 
 |             block_acct_invalid(blk_get_stats(blk), BLOCK_ACCT_WRITE); | 
 |             g_free(ctx); | 
 |             return -EINVAL; | 
 |         } | 
 |  | 
 |         clock_gettime(CLOCK_MONOTONIC, &ctx->t1); | 
 |         block_acct_start(blk_get_stats(blk), &ctx->acct, ctx->qiov.size, | 
 |                          BLOCK_ACCT_WRITE); | 
 |  | 
 |         blk_aio_pwritev(blk, ctx->offset, &ctx->qiov, ctx->flags, | 
 |                         aio_write_done, ctx); | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | static int aio_flush_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     BlockAcctCookie cookie; | 
 |     block_acct_start(blk_get_stats(blk), &cookie, 0, BLOCK_ACCT_FLUSH); | 
 |     blk_drain_all(); | 
 |     block_acct_done(blk_get_stats(blk), &cookie); | 
 |     return 0; | 
 | } | 
 |  | 
 | static const cmdinfo_t aio_flush_cmd = { | 
 |     .name       = "aio_flush", | 
 |     .cfunc      = aio_flush_f, | 
 |     .oneline    = "completes all outstanding aio requests" | 
 | }; | 
 |  | 
 | static int flush_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     return blk_flush(blk); | 
 | } | 
 |  | 
 | static const cmdinfo_t flush_cmd = { | 
 |     .name       = "flush", | 
 |     .altname    = "f", | 
 |     .cfunc      = flush_f, | 
 |     .oneline    = "flush all in-core file state to disk", | 
 | }; | 
 |  | 
 | static inline int64_t tosector(int64_t bytes) | 
 | { | 
 |     return bytes >> BDRV_SECTOR_BITS; | 
 | } | 
 |  | 
 | static int zone_report_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int ret; | 
 |     int64_t offset; | 
 |     unsigned int nr_zones; | 
 |  | 
 |     ++optind; | 
 |     offset = cvtnum(argv[optind]); | 
 |     ++optind; | 
 |     nr_zones = cvtnum(argv[optind]); | 
 |  | 
 |     g_autofree BlockZoneDescriptor *zones = NULL; | 
 |     zones = g_new(BlockZoneDescriptor, nr_zones); | 
 |     ret = blk_zone_report(blk, offset, &nr_zones, zones); | 
 |     if (ret < 0) { | 
 |         printf("zone report failed: %s\n", strerror(-ret)); | 
 |     } else { | 
 |         for (int i = 0; i < nr_zones; ++i) { | 
 |             printf("start: 0x%" PRIx64 ", len 0x%" PRIx64 ", " | 
 |                    "cap"" 0x%" PRIx64 ", wptr 0x%" PRIx64 ", " | 
 |                    "zcond:%u, [type: %u]\n", | 
 |                     tosector(zones[i].start), tosector(zones[i].length), | 
 |                     tosector(zones[i].cap), tosector(zones[i].wp), | 
 |                     zones[i].state, zones[i].type); | 
 |         } | 
 |     } | 
 |     return ret; | 
 | } | 
 |  | 
 | static const cmdinfo_t zone_report_cmd = { | 
 |     .name = "zone_report", | 
 |     .altname = "zrp", | 
 |     .cfunc = zone_report_f, | 
 |     .argmin = 2, | 
 |     .argmax = 2, | 
 |     .args = "offset number", | 
 |     .oneline = "report zone information", | 
 | }; | 
 |  | 
 | static int zone_open_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int ret; | 
 |     int64_t offset, len; | 
 |     ++optind; | 
 |     offset = cvtnum(argv[optind]); | 
 |     ++optind; | 
 |     len = cvtnum(argv[optind]); | 
 |     ret = blk_zone_mgmt(blk, BLK_ZO_OPEN, offset, len); | 
 |     if (ret < 0) { | 
 |         printf("zone open failed: %s\n", strerror(-ret)); | 
 |     } | 
 |     return ret; | 
 | } | 
 |  | 
 | static const cmdinfo_t zone_open_cmd = { | 
 |     .name = "zone_open", | 
 |     .altname = "zo", | 
 |     .cfunc = zone_open_f, | 
 |     .argmin = 2, | 
 |     .argmax = 2, | 
 |     .args = "offset len", | 
 |     .oneline = "explicit open a range of zones in zone block device", | 
 | }; | 
 |  | 
 | static int zone_close_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int ret; | 
 |     int64_t offset, len; | 
 |     ++optind; | 
 |     offset = cvtnum(argv[optind]); | 
 |     ++optind; | 
 |     len = cvtnum(argv[optind]); | 
 |     ret = blk_zone_mgmt(blk, BLK_ZO_CLOSE, offset, len); | 
 |     if (ret < 0) { | 
 |         printf("zone close failed: %s\n", strerror(-ret)); | 
 |     } | 
 |     return ret; | 
 | } | 
 |  | 
 | static const cmdinfo_t zone_close_cmd = { | 
 |     .name = "zone_close", | 
 |     .altname = "zc", | 
 |     .cfunc = zone_close_f, | 
 |     .argmin = 2, | 
 |     .argmax = 2, | 
 |     .args = "offset len", | 
 |     .oneline = "close a range of zones in zone block device", | 
 | }; | 
 |  | 
 | static int zone_finish_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int ret; | 
 |     int64_t offset, len; | 
 |     ++optind; | 
 |     offset = cvtnum(argv[optind]); | 
 |     ++optind; | 
 |     len = cvtnum(argv[optind]); | 
 |     ret = blk_zone_mgmt(blk, BLK_ZO_FINISH, offset, len); | 
 |     if (ret < 0) { | 
 |         printf("zone finish failed: %s\n", strerror(-ret)); | 
 |     } | 
 |     return ret; | 
 | } | 
 |  | 
 | static const cmdinfo_t zone_finish_cmd = { | 
 |     .name = "zone_finish", | 
 |     .altname = "zf", | 
 |     .cfunc = zone_finish_f, | 
 |     .argmin = 2, | 
 |     .argmax = 2, | 
 |     .args = "offset len", | 
 |     .oneline = "finish a range of zones in zone block device", | 
 | }; | 
 |  | 
 | static int zone_reset_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int ret; | 
 |     int64_t offset, len; | 
 |     ++optind; | 
 |     offset = cvtnum(argv[optind]); | 
 |     ++optind; | 
 |     len = cvtnum(argv[optind]); | 
 |     ret = blk_zone_mgmt(blk, BLK_ZO_RESET, offset, len); | 
 |     if (ret < 0) { | 
 |         printf("zone reset failed: %s\n", strerror(-ret)); | 
 |     } | 
 |     return ret; | 
 | } | 
 |  | 
 | static const cmdinfo_t zone_reset_cmd = { | 
 |     .name = "zone_reset", | 
 |     .altname = "zrs", | 
 |     .cfunc = zone_reset_f, | 
 |     .argmin = 2, | 
 |     .argmax = 2, | 
 |     .args = "offset len", | 
 |     .oneline = "reset a zone write pointer in zone block device", | 
 | }; | 
 |  | 
 | static int do_aio_zone_append(BlockBackend *blk, QEMUIOVector *qiov, | 
 |                               int64_t *offset, int flags, int *total) | 
 | { | 
 |     int async_ret = NOT_DONE; | 
 |  | 
 |     blk_aio_zone_append(blk, offset, qiov, flags, aio_rw_done, &async_ret); | 
 |     while (async_ret == NOT_DONE) { | 
 |         main_loop_wait(false); | 
 |     } | 
 |  | 
 |     *total = qiov->size; | 
 |     return async_ret < 0 ? async_ret : 1; | 
 | } | 
 |  | 
 | static int zone_append_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int ret; | 
 |     bool pflag = false; | 
 |     int flags = 0; | 
 |     int total = 0; | 
 |     int64_t offset; | 
 |     char *buf; | 
 |     int c, nr_iov; | 
 |     int pattern = 0xcd; | 
 |     QEMUIOVector qiov; | 
 |  | 
 |     if (optind > argc - 3) { | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if ((c = getopt(argc, argv, "p")) != -1) { | 
 |         pflag = true; | 
 |     } | 
 |  | 
 |     offset = cvtnum(argv[optind]); | 
 |     if (offset < 0) { | 
 |         print_cvtnum_err(offset, argv[optind]); | 
 |         return offset; | 
 |     } | 
 |     optind++; | 
 |     nr_iov = argc - optind; | 
 |     buf = create_iovec(blk, &qiov, &argv[optind], nr_iov, pattern, | 
 |                        flags & BDRV_REQ_REGISTERED_BUF); | 
 |     if (buf == NULL) { | 
 |         return -EINVAL; | 
 |     } | 
 |     ret = do_aio_zone_append(blk, &qiov, &offset, flags, &total); | 
 |     if (ret < 0) { | 
 |         printf("zone append failed: %s\n", strerror(-ret)); | 
 |         goto out; | 
 |     } | 
 |  | 
 |     if (pflag) { | 
 |         printf("After zap done, the append sector is 0x%" PRIx64 "\n", | 
 |                tosector(offset)); | 
 |     } | 
 |  | 
 | out: | 
 |     qemu_io_free(blk, buf, qiov.size, | 
 |                  flags & BDRV_REQ_REGISTERED_BUF); | 
 |     qemu_iovec_destroy(&qiov); | 
 |     return ret; | 
 | } | 
 |  | 
 | static const cmdinfo_t zone_append_cmd = { | 
 |     .name = "zone_append", | 
 |     .altname = "zap", | 
 |     .cfunc = zone_append_f, | 
 |     .argmin = 3, | 
 |     .argmax = 4, | 
 |     .args = "offset len [len..]", | 
 |     .oneline = "append write a number of bytes at a specified offset", | 
 | }; | 
 |  | 
 | static int truncate_f(BlockBackend *blk, int argc, char **argv); | 
 | static const cmdinfo_t truncate_cmd = { | 
 |     .name       = "truncate", | 
 |     .altname    = "t", | 
 |     .cfunc      = truncate_f, | 
 |     .perm       = BLK_PERM_WRITE | BLK_PERM_RESIZE, | 
 |     .argmin     = 1, | 
 |     .argmax     = 3, | 
 |     .args       = "[-m prealloc_mode] off", | 
 |     .oneline    = "truncates the current file at the given offset", | 
 | }; | 
 |  | 
 | static int truncate_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     Error *local_err = NULL; | 
 |     int64_t offset; | 
 |     int c, ret; | 
 |     PreallocMode prealloc = PREALLOC_MODE_OFF; | 
 |  | 
 |     while ((c = getopt(argc, argv, "m:")) != -1) { | 
 |         switch (c) { | 
 |         case 'm': | 
 |             prealloc = qapi_enum_parse(&PreallocMode_lookup, optarg, | 
 |                                        PREALLOC_MODE__MAX, NULL); | 
 |             if (prealloc == PREALLOC_MODE__MAX) { | 
 |                 error_report("Invalid preallocation mode '%s'", optarg); | 
 |                 return -EINVAL; | 
 |             } | 
 |             break; | 
 |         default: | 
 |             qemuio_command_usage(&truncate_cmd); | 
 |             return -EINVAL; | 
 |         } | 
 |     } | 
 |  | 
 |     offset = cvtnum(argv[optind]); | 
 |     if (offset < 0) { | 
 |         print_cvtnum_err(offset, argv[1]); | 
 |         return offset; | 
 |     } | 
 |  | 
 |     /* | 
 |      * qemu-io is a debugging tool, so let us be strict here and pass | 
 |      * exact=true.  It is better to err on the "emit more errors" side | 
 |      * than to be overly permissive. | 
 |      */ | 
 |     ret = blk_truncate(blk, offset, false, prealloc, 0, &local_err); | 
 |     if (ret < 0) { | 
 |         error_report_err(local_err); | 
 |         return ret; | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | static int length_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int64_t size; | 
 |     char s1[64]; | 
 |  | 
 |     size = blk_getlength(blk); | 
 |     if (size < 0) { | 
 |         printf("getlength: %s\n", strerror(-size)); | 
 |         return size; | 
 |     } | 
 |  | 
 |     cvtstr(size, s1, sizeof(s1)); | 
 |     printf("%s\n", s1); | 
 |     return 0; | 
 | } | 
 |  | 
 |  | 
 | static const cmdinfo_t length_cmd = { | 
 |     .name   = "length", | 
 |     .altname    = "l", | 
 |     .cfunc      = length_f, | 
 |     .oneline    = "gets the length of the current file", | 
 | }; | 
 |  | 
 |  | 
 | static int info_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     BlockDriverState *bs = blk_bs(blk); | 
 |     BlockDriverInfo bdi; | 
 |     ImageInfoSpecific *spec_info; | 
 |     Error *local_err = NULL; | 
 |     char s1[64], s2[64]; | 
 |     int ret; | 
 |  | 
 |     GLOBAL_STATE_CODE(); | 
 |     GRAPH_RDLOCK_GUARD_MAINLOOP(); | 
 |  | 
 |     if (bs->drv && bs->drv->format_name) { | 
 |         printf("format name: %s\n", bs->drv->format_name); | 
 |     } | 
 |     if (bs->drv && bs->drv->protocol_name) { | 
 |         printf("format name: %s\n", bs->drv->protocol_name); | 
 |     } | 
 |  | 
 |     ret = bdrv_get_info(bs, &bdi); | 
 |     if (ret) { | 
 |         return ret; | 
 |     } | 
 |  | 
 |     cvtstr(bdi.cluster_size, s1, sizeof(s1)); | 
 |     cvtstr(bdi.vm_state_offset, s2, sizeof(s2)); | 
 |  | 
 |     printf("cluster size: %s\n", s1); | 
 |     printf("vm state offset: %s\n", s2); | 
 |  | 
 |     spec_info = bdrv_get_specific_info(bs, &local_err); | 
 |     if (local_err) { | 
 |         error_report_err(local_err); | 
 |         return -EIO; | 
 |     } | 
 |     if (spec_info) { | 
 |         bdrv_image_info_specific_dump(spec_info, | 
 |                                       "Format specific information:\n", | 
 |                                       0); | 
 |         qapi_free_ImageInfoSpecific(spec_info); | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 |  | 
 |  | 
 | static const cmdinfo_t info_cmd = { | 
 |     .name       = "info", | 
 |     .altname    = "i", | 
 |     .cfunc      = info_f, | 
 |     .oneline    = "prints information about the current file", | 
 | }; | 
 |  | 
 | static void discard_help(void) | 
 | { | 
 |     printf( | 
 | "\n" | 
 | " discards a range of bytes from the given offset\n" | 
 | "\n" | 
 | " Example:\n" | 
 | " 'discard 512 1k' - discards 1 kilobyte from 512 bytes into the file\n" | 
 | "\n" | 
 | " Discards a segment of the currently open file.\n" | 
 | " -C, -- report statistics in a machine parsable format\n" | 
 | " -q, -- quiet mode, do not show I/O statistics\n" | 
 | "\n"); | 
 | } | 
 |  | 
 | static int discard_f(BlockBackend *blk, int argc, char **argv); | 
 |  | 
 | static const cmdinfo_t discard_cmd = { | 
 |     .name       = "discard", | 
 |     .altname    = "d", | 
 |     .cfunc      = discard_f, | 
 |     .perm       = BLK_PERM_WRITE, | 
 |     .argmin     = 2, | 
 |     .argmax     = -1, | 
 |     .args       = "[-Cq] off len", | 
 |     .oneline    = "discards a number of bytes at a specified offset", | 
 |     .help       = discard_help, | 
 | }; | 
 |  | 
 | static int discard_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     struct timespec t1, t2; | 
 |     bool Cflag = false, qflag = false; | 
 |     int c, ret; | 
 |     int64_t offset, bytes; | 
 |  | 
 |     while ((c = getopt(argc, argv, "Cq")) != -1) { | 
 |         switch (c) { | 
 |         case 'C': | 
 |             Cflag = true; | 
 |             break; | 
 |         case 'q': | 
 |             qflag = true; | 
 |             break; | 
 |         default: | 
 |             qemuio_command_usage(&discard_cmd); | 
 |             return -EINVAL; | 
 |         } | 
 |     } | 
 |  | 
 |     if (optind != argc - 2) { | 
 |         qemuio_command_usage(&discard_cmd); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     offset = cvtnum(argv[optind]); | 
 |     if (offset < 0) { | 
 |         print_cvtnum_err(offset, argv[optind]); | 
 |         return offset; | 
 |     } | 
 |  | 
 |     optind++; | 
 |     bytes = cvtnum(argv[optind]); | 
 |     if (bytes < 0) { | 
 |         print_cvtnum_err(bytes, argv[optind]); | 
 |         return bytes; | 
 |     } else if (bytes > BDRV_REQUEST_MAX_BYTES) { | 
 |         printf("length cannot exceed %"PRIu64", given %s\n", | 
 |                (uint64_t)BDRV_REQUEST_MAX_BYTES, argv[optind]); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     clock_gettime(CLOCK_MONOTONIC, &t1); | 
 |     ret = blk_pdiscard(blk, offset, bytes); | 
 |     clock_gettime(CLOCK_MONOTONIC, &t2); | 
 |  | 
 |     if (ret < 0) { | 
 |         printf("discard failed: %s\n", strerror(-ret)); | 
 |         return ret; | 
 |     } | 
 |  | 
 |     /* Finally, report back -- -C gives a parsable format */ | 
 |     if (!qflag) { | 
 |         t2 = tsub(t2, t1); | 
 |         print_report("discard", &t2, offset, bytes, bytes, 1, Cflag); | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | static int alloc_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     BlockDriverState *bs = blk_bs(blk); | 
 |     int64_t offset, start, remaining, count; | 
 |     char s1[64]; | 
 |     int ret; | 
 |     int64_t num, sum_alloc; | 
 |  | 
 |     start = offset = cvtnum(argv[1]); | 
 |     if (offset < 0) { | 
 |         print_cvtnum_err(offset, argv[1]); | 
 |         return offset; | 
 |     } | 
 |  | 
 |     if (argc == 3) { | 
 |         count = cvtnum(argv[2]); | 
 |         if (count < 0) { | 
 |             print_cvtnum_err(count, argv[2]); | 
 |             return count; | 
 |         } | 
 |     } else { | 
 |         count = BDRV_SECTOR_SIZE; | 
 |     } | 
 |  | 
 |     remaining = count; | 
 |     sum_alloc = 0; | 
 |     while (remaining) { | 
 |         ret = bdrv_is_allocated(bs, offset, remaining, &num); | 
 |         if (ret < 0) { | 
 |             printf("is_allocated failed: %s\n", strerror(-ret)); | 
 |             return ret; | 
 |         } | 
 |         offset += num; | 
 |         remaining -= num; | 
 |         if (ret) { | 
 |             sum_alloc += num; | 
 |         } | 
 |         if (num == 0) { | 
 |             count -= remaining; | 
 |             remaining = 0; | 
 |         } | 
 |     } | 
 |  | 
 |     cvtstr(start, s1, sizeof(s1)); | 
 |  | 
 |     printf("%"PRId64"/%"PRId64" bytes allocated at offset %s\n", | 
 |            sum_alloc, count, s1); | 
 |     return 0; | 
 | } | 
 |  | 
 | static const cmdinfo_t alloc_cmd = { | 
 |     .name       = "alloc", | 
 |     .altname    = "a", | 
 |     .argmin     = 1, | 
 |     .argmax     = 2, | 
 |     .cfunc      = alloc_f, | 
 |     .args       = "offset [count]", | 
 |     .oneline    = "checks if offset is allocated in the file", | 
 | }; | 
 |  | 
 |  | 
 | static int map_is_allocated(BlockDriverState *bs, int64_t offset, | 
 |                             int64_t bytes, int64_t *pnum) | 
 | { | 
 |     int64_t num; | 
 |     int ret, firstret; | 
 |  | 
 |     ret = bdrv_is_allocated(bs, offset, bytes, &num); | 
 |     if (ret < 0) { | 
 |         return ret; | 
 |     } | 
 |  | 
 |     firstret = ret; | 
 |     *pnum = num; | 
 |  | 
 |     while (bytes > 0 && ret == firstret) { | 
 |         offset += num; | 
 |         bytes -= num; | 
 |  | 
 |         ret = bdrv_is_allocated(bs, offset, bytes, &num); | 
 |         if (ret == firstret && num) { | 
 |             *pnum += num; | 
 |         } else { | 
 |             break; | 
 |         } | 
 |     } | 
 |  | 
 |     return firstret; | 
 | } | 
 |  | 
 | static int map_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int64_t offset, bytes; | 
 |     char s1[64], s2[64]; | 
 |     int64_t num; | 
 |     int ret; | 
 |     const char *retstr; | 
 |  | 
 |     offset = 0; | 
 |     bytes = blk_getlength(blk); | 
 |     if (bytes < 0) { | 
 |         error_report("Failed to query image length: %s", strerror(-bytes)); | 
 |         return bytes; | 
 |     } | 
 |  | 
 |     while (bytes) { | 
 |         ret = map_is_allocated(blk_bs(blk), offset, bytes, &num); | 
 |         if (ret < 0) { | 
 |             error_report("Failed to get allocation status: %s", strerror(-ret)); | 
 |             return ret; | 
 |         } else if (!num) { | 
 |             error_report("Unexpected end of image"); | 
 |             return -EIO; | 
 |         } | 
 |  | 
 |         retstr = ret ? "    allocated" : "not allocated"; | 
 |         cvtstr(num, s1, sizeof(s1)); | 
 |         cvtstr(offset, s2, sizeof(s2)); | 
 |         printf("%s (0x%" PRIx64 ") bytes %s at offset %s (0x%" PRIx64 ")\n", | 
 |                s1, num, retstr, s2, offset); | 
 |  | 
 |         offset += num; | 
 |         bytes -= num; | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | static const cmdinfo_t map_cmd = { | 
 |        .name           = "map", | 
 |        .argmin         = 0, | 
 |        .argmax         = 0, | 
 |        .cfunc          = map_f, | 
 |        .args           = "", | 
 |        .oneline        = "prints the allocated areas of a file", | 
 | }; | 
 |  | 
 | static void reopen_help(void) | 
 | { | 
 |     printf( | 
 | "\n" | 
 | " Changes the open options of an already opened image\n" | 
 | "\n" | 
 | " Example:\n" | 
 | " 'reopen -o lazy-refcounts=on' - activates lazy refcount writeback on a qcow2 image\n" | 
 | "\n" | 
 | " -r, -- Reopen the image read-only\n" | 
 | " -w, -- Reopen the image read-write\n" | 
 | " -c, -- Change the cache mode to the given value\n" | 
 | " -o, -- Changes block driver options (cf. 'open' command)\n" | 
 | "\n"); | 
 | } | 
 |  | 
 | static int reopen_f(BlockBackend *blk, int argc, char **argv); | 
 |  | 
 | static QemuOptsList reopen_opts = { | 
 |     .name = "reopen", | 
 |     .merge_lists = true, | 
 |     .head = QTAILQ_HEAD_INITIALIZER(reopen_opts.head), | 
 |     .desc = { | 
 |         /* no elements => accept any params */ | 
 |         { /* end of list */ } | 
 |     }, | 
 | }; | 
 |  | 
 | static const cmdinfo_t reopen_cmd = { | 
 |        .name           = "reopen", | 
 |        .argmin         = 0, | 
 |        .argmax         = -1, | 
 |        .cfunc          = reopen_f, | 
 |        .args           = "[(-r|-w)] [-c cache] [-o options]", | 
 |        .oneline        = "reopens an image with new options", | 
 |        .help           = reopen_help, | 
 | }; | 
 |  | 
 | static int reopen_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     BlockDriverState *bs = blk_bs(blk); | 
 |     QemuOpts *qopts; | 
 |     QDict *opts; | 
 |     int c; | 
 |     int flags = bs->open_flags; | 
 |     bool writethrough = !blk_enable_write_cache(blk); | 
 |     bool has_rw_option = false; | 
 |     bool has_cache_option = false; | 
 |     Error *local_err = NULL; | 
 |  | 
 |     while ((c = getopt(argc, argv, "c:o:rw")) != -1) { | 
 |         switch (c) { | 
 |         case 'c': | 
 |             if (bdrv_parse_cache_mode(optarg, &flags, &writethrough) < 0) { | 
 |                 error_report("Invalid cache option: %s", optarg); | 
 |                 return -EINVAL; | 
 |             } | 
 |             has_cache_option = true; | 
 |             break; | 
 |         case 'o': | 
 |             if (!qemu_opts_parse_noisily(&reopen_opts, optarg, 0)) { | 
 |                 qemu_opts_reset(&reopen_opts); | 
 |                 return -EINVAL; | 
 |             } | 
 |             break; | 
 |         case 'r': | 
 |             if (has_rw_option) { | 
 |                 error_report("Only one -r/-w option may be given"); | 
 |                 return -EINVAL; | 
 |             } | 
 |             flags &= ~BDRV_O_RDWR; | 
 |             has_rw_option = true; | 
 |             break; | 
 |         case 'w': | 
 |             if (has_rw_option) { | 
 |                 error_report("Only one -r/-w option may be given"); | 
 |                 return -EINVAL; | 
 |             } | 
 |             flags |= BDRV_O_RDWR; | 
 |             has_rw_option = true; | 
 |             break; | 
 |         default: | 
 |             qemu_opts_reset(&reopen_opts); | 
 |             qemuio_command_usage(&reopen_cmd); | 
 |             return -EINVAL; | 
 |         } | 
 |     } | 
 |  | 
 |     if (optind != argc) { | 
 |         qemu_opts_reset(&reopen_opts); | 
 |         qemuio_command_usage(&reopen_cmd); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     if (!writethrough != blk_enable_write_cache(blk) && | 
 |         blk_get_attached_dev(blk)) | 
 |     { | 
 |         error_report("Cannot change cache.writeback: Device attached"); | 
 |         qemu_opts_reset(&reopen_opts); | 
 |         return -EBUSY; | 
 |     } | 
 |  | 
 |     if (!(flags & BDRV_O_RDWR)) { | 
 |         uint64_t orig_perm, orig_shared_perm; | 
 |  | 
 |         bdrv_drain(bs); | 
 |  | 
 |         blk_get_perm(blk, &orig_perm, &orig_shared_perm); | 
 |         blk_set_perm(blk, | 
 |                      orig_perm & ~(BLK_PERM_WRITE | BLK_PERM_WRITE_UNCHANGED), | 
 |                      orig_shared_perm, | 
 |                      &error_abort); | 
 |     } | 
 |  | 
 |     qopts = qemu_opts_find(&reopen_opts, NULL); | 
 |     opts = qopts ? qemu_opts_to_qdict(qopts, NULL) : qdict_new(); | 
 |     qemu_opts_reset(&reopen_opts); | 
 |  | 
 |     if (qdict_haskey(opts, BDRV_OPT_READ_ONLY)) { | 
 |         if (has_rw_option) { | 
 |             error_report("Cannot set both -r/-w and '" BDRV_OPT_READ_ONLY "'"); | 
 |             qobject_unref(opts); | 
 |             return -EINVAL; | 
 |         } | 
 |     } else { | 
 |         qdict_put_bool(opts, BDRV_OPT_READ_ONLY, !(flags & BDRV_O_RDWR)); | 
 |     } | 
 |  | 
 |     if (qdict_haskey(opts, BDRV_OPT_CACHE_DIRECT) || | 
 |         qdict_haskey(opts, BDRV_OPT_CACHE_NO_FLUSH)) { | 
 |         if (has_cache_option) { | 
 |             error_report("Cannot set both -c and the cache options"); | 
 |             qobject_unref(opts); | 
 |             return -EINVAL; | 
 |         } | 
 |     } else { | 
 |         qdict_put_bool(opts, BDRV_OPT_CACHE_DIRECT, flags & BDRV_O_NOCACHE); | 
 |         qdict_put_bool(opts, BDRV_OPT_CACHE_NO_FLUSH, flags & BDRV_O_NO_FLUSH); | 
 |     } | 
 |  | 
 |     bdrv_reopen(bs, opts, true, &local_err); | 
 |  | 
 |     if (local_err) { | 
 |         error_report_err(local_err); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     blk_set_enable_write_cache(blk, !writethrough); | 
 |     return 0; | 
 | } | 
 |  | 
 | static int break_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int ret; | 
 |  | 
 |     ret = bdrv_debug_breakpoint(blk_bs(blk), argv[1], argv[2]); | 
 |     if (ret < 0) { | 
 |         printf("Could not set breakpoint: %s\n", strerror(-ret)); | 
 |         return ret; | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | static int remove_break_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int ret; | 
 |  | 
 |     ret = bdrv_debug_remove_breakpoint(blk_bs(blk), argv[1]); | 
 |     if (ret < 0) { | 
 |         printf("Could not remove breakpoint %s: %s\n", argv[1], strerror(-ret)); | 
 |         return ret; | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | static const cmdinfo_t break_cmd = { | 
 |        .name           = "break", | 
 |        .argmin         = 2, | 
 |        .argmax         = 2, | 
 |        .cfunc          = break_f, | 
 |        .args           = "event tag", | 
 |        .oneline        = "sets a breakpoint on event and tags the stopped " | 
 |                          "request as tag", | 
 | }; | 
 |  | 
 | static const cmdinfo_t remove_break_cmd = { | 
 |        .name           = "remove_break", | 
 |        .argmin         = 1, | 
 |        .argmax         = 1, | 
 |        .cfunc          = remove_break_f, | 
 |        .args           = "tag", | 
 |        .oneline        = "remove a breakpoint by tag", | 
 | }; | 
 |  | 
 | static int resume_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int ret; | 
 |  | 
 |     ret = bdrv_debug_resume(blk_bs(blk), argv[1]); | 
 |     if (ret < 0) { | 
 |         printf("Could not resume request: %s\n", strerror(-ret)); | 
 |         return ret; | 
 |     } | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | static const cmdinfo_t resume_cmd = { | 
 |        .name           = "resume", | 
 |        .argmin         = 1, | 
 |        .argmax         = 1, | 
 |        .cfunc          = resume_f, | 
 |        .args           = "tag", | 
 |        .oneline        = "resumes the request tagged as tag", | 
 | }; | 
 |  | 
 | static int wait_break_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     while (!bdrv_debug_is_suspended(blk_bs(blk), argv[1])) { | 
 |         aio_poll(blk_get_aio_context(blk), true); | 
 |     } | 
 |     return 0; | 
 | } | 
 |  | 
 | static const cmdinfo_t wait_break_cmd = { | 
 |        .name           = "wait_break", | 
 |        .argmin         = 1, | 
 |        .argmax         = 1, | 
 |        .cfunc          = wait_break_f, | 
 |        .args           = "tag", | 
 |        .oneline        = "waits for the suspension of a request", | 
 | }; | 
 |  | 
 | static int abort_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     abort(); | 
 | } | 
 |  | 
 | static const cmdinfo_t abort_cmd = { | 
 |        .name           = "abort", | 
 |        .cfunc          = abort_f, | 
 |        .flags          = CMD_NOFILE_OK, | 
 |        .oneline        = "simulate a program crash using abort(3)", | 
 | }; | 
 |  | 
 | static void sigraise_help(void) | 
 | { | 
 |     printf( | 
 | "\n" | 
 | " raises the given signal\n" | 
 | "\n" | 
 | " Example:\n" | 
 | " 'sigraise %i' - raises SIGTERM\n" | 
 | "\n" | 
 | " Invokes raise(signal), where \"signal\" is the mandatory integer argument\n" | 
 | " given to sigraise.\n" | 
 | "\n", SIGTERM); | 
 | } | 
 |  | 
 | static int sigraise_f(BlockBackend *blk, int argc, char **argv); | 
 |  | 
 | static const cmdinfo_t sigraise_cmd = { | 
 |     .name       = "sigraise", | 
 |     .cfunc      = sigraise_f, | 
 |     .argmin     = 1, | 
 |     .argmax     = 1, | 
 |     .flags      = CMD_NOFILE_OK, | 
 |     .args       = "signal", | 
 |     .oneline    = "raises a signal", | 
 |     .help       = sigraise_help, | 
 | }; | 
 |  | 
 | static int sigraise_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     int64_t sig = cvtnum(argv[1]); | 
 |     if (sig < 0) { | 
 |         print_cvtnum_err(sig, argv[1]); | 
 |         return sig; | 
 |     } else if (sig > NSIG) { | 
 |         printf("signal argument '%s' is too large to be a valid signal\n", | 
 |                argv[1]); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     /* Using raise() to kill this process does not necessarily flush all open | 
 |      * streams. At least stdout and stderr (although the latter should be | 
 |      * non-buffered anyway) should be flushed, though. */ | 
 |     fflush(stdout); | 
 |     fflush(stderr); | 
 |  | 
 |     raise(sig); | 
 |  | 
 |     return 0; | 
 | } | 
 |  | 
 | static void sleep_cb(void *opaque) | 
 | { | 
 |     bool *expired = opaque; | 
 |     *expired = true; | 
 | } | 
 |  | 
 | static int sleep_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     char *endptr; | 
 |     long ms; | 
 |     struct QEMUTimer *timer; | 
 |     bool expired = false; | 
 |  | 
 |     ms = strtol(argv[1], &endptr, 0); | 
 |     if (ms < 0 || *endptr != '\0') { | 
 |         printf("%s is not a valid number\n", argv[1]); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     timer = timer_new_ns(QEMU_CLOCK_HOST, sleep_cb, &expired); | 
 |     timer_mod(timer, qemu_clock_get_ns(QEMU_CLOCK_HOST) + SCALE_MS * ms); | 
 |  | 
 |     while (!expired) { | 
 |         main_loop_wait(false); | 
 |     } | 
 |  | 
 |     timer_free(timer); | 
 |     return 0; | 
 | } | 
 |  | 
 | static const cmdinfo_t sleep_cmd = { | 
 |        .name           = "sleep", | 
 |        .argmin         = 1, | 
 |        .argmax         = 1, | 
 |        .cfunc          = sleep_f, | 
 |        .flags          = CMD_NOFILE_OK, | 
 |        .oneline        = "waits for the given value in milliseconds", | 
 | }; | 
 |  | 
 | static void help_oneline(const char *cmd, const cmdinfo_t *ct) | 
 | { | 
 |     printf("%s ", cmd); | 
 |  | 
 |     if (ct->args) { | 
 |         printf("%s ", ct->args); | 
 |     } | 
 |     printf("-- %s\n", ct->oneline); | 
 | } | 
 |  | 
 | static void help_onecmd(const char *cmd, const cmdinfo_t *ct) | 
 | { | 
 |     help_oneline(cmd, ct); | 
 |     if (ct->help) { | 
 |         ct->help(); | 
 |     } | 
 | } | 
 |  | 
 | static void help_all(void) | 
 | { | 
 |     const cmdinfo_t *ct; | 
 |  | 
 |     for (ct = cmdtab; ct < &cmdtab[ncmds]; ct++) { | 
 |         help_oneline(ct->name, ct); | 
 |     } | 
 |     printf("\nUse 'help commandname' for extended help.\n"); | 
 | } | 
 |  | 
 | static int help_f(BlockBackend *blk, int argc, char **argv) | 
 | { | 
 |     const cmdinfo_t *ct; | 
 |  | 
 |     if (argc < 2) { | 
 |         help_all(); | 
 |         return 0; | 
 |     } | 
 |  | 
 |     ct = find_command(argv[1]); | 
 |     if (ct == NULL) { | 
 |         printf("command %s not found\n", argv[1]); | 
 |         return -EINVAL; | 
 |     } | 
 |  | 
 |     help_onecmd(argv[1], ct); | 
 |     return 0; | 
 | } | 
 |  | 
 | static const cmdinfo_t help_cmd = { | 
 |     .name       = "help", | 
 |     .altname    = "?", | 
 |     .cfunc      = help_f, | 
 |     .argmin     = 0, | 
 |     .argmax     = 1, | 
 |     .flags      = CMD_FLAG_GLOBAL, | 
 |     .args       = "[command]", | 
 |     .oneline    = "help for one or all commands", | 
 | }; | 
 |  | 
 | /* | 
 |  * Called with aio context of blk acquired. Or with qemu_get_aio_context() | 
 |  * context acquired if blk is NULL. | 
 |  */ | 
 | int qemuio_command(BlockBackend *blk, const char *cmd) | 
 | { | 
 |     char *input; | 
 |     const cmdinfo_t *ct; | 
 |     char **v; | 
 |     int c; | 
 |     int ret = 0; | 
 |  | 
 |     input = g_strdup(cmd); | 
 |     v = breakline(input, &c); | 
 |     if (c) { | 
 |         ct = find_command(v[0]); | 
 |         if (ct) { | 
 |             ret = command(blk, ct, c, v); | 
 |         } else { | 
 |             fprintf(stderr, "command \"%s\" not found\n", v[0]); | 
 |             ret = -EINVAL; | 
 |         } | 
 |     } | 
 |     g_free(input); | 
 |     g_free(v); | 
 |  | 
 |     return ret; | 
 | } | 
 |  | 
 | static void __attribute((constructor)) init_qemuio_commands(void) | 
 | { | 
 |     /* initialize commands */ | 
 |     qemuio_add_command(&help_cmd); | 
 |     qemuio_add_command(&read_cmd); | 
 |     qemuio_add_command(&readv_cmd); | 
 |     qemuio_add_command(&write_cmd); | 
 |     qemuio_add_command(&writev_cmd); | 
 |     qemuio_add_command(&aio_read_cmd); | 
 |     qemuio_add_command(&aio_write_cmd); | 
 |     qemuio_add_command(&aio_flush_cmd); | 
 |     qemuio_add_command(&flush_cmd); | 
 |     qemuio_add_command(&zone_report_cmd); | 
 |     qemuio_add_command(&zone_open_cmd); | 
 |     qemuio_add_command(&zone_close_cmd); | 
 |     qemuio_add_command(&zone_finish_cmd); | 
 |     qemuio_add_command(&zone_reset_cmd); | 
 |     qemuio_add_command(&zone_append_cmd); | 
 |     qemuio_add_command(&truncate_cmd); | 
 |     qemuio_add_command(&length_cmd); | 
 |     qemuio_add_command(&info_cmd); | 
 |     qemuio_add_command(&discard_cmd); | 
 |     qemuio_add_command(&alloc_cmd); | 
 |     qemuio_add_command(&map_cmd); | 
 |     qemuio_add_command(&reopen_cmd); | 
 |     qemuio_add_command(&break_cmd); | 
 |     qemuio_add_command(&remove_break_cmd); | 
 |     qemuio_add_command(&resume_cmd); | 
 |     qemuio_add_command(&wait_break_cmd); | 
 |     qemuio_add_command(&abort_cmd); | 
 |     qemuio_add_command(&sleep_cmd); | 
 |     qemuio_add_command(&sigraise_cmd); | 
 | } |