aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Command line utility to exercise the QEMU I/O path. |
| 3 | * |
| 4 | * Copyright (C) 2009 Red Hat, Inc. |
| 5 | * Copyright (c) 2003-2005 Silicon Graphics, Inc. |
| 6 | * |
| 7 | * This work is licensed under the terms of the GNU GPL, version 2 or later. |
| 8 | * See the COPYING file in the top-level directory. |
| 9 | */ |
Stefan Weil | c32d766 | 2009-08-31 22:16:16 +0200 | [diff] [blame] | 10 | #include <sys/time.h> |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 11 | #include <sys/types.h> |
| 12 | #include <stdarg.h> |
| 13 | #include <stdio.h> |
| 14 | #include <getopt.h> |
Stefan Weil | c32d766 | 2009-08-31 22:16:16 +0200 | [diff] [blame] | 15 | #include <libgen.h> |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 16 | |
| 17 | #include "qemu-common.h" |
| 18 | #include "block_int.h" |
| 19 | #include "cmd.h" |
| 20 | |
| 21 | #define VERSION "0.0.1" |
| 22 | |
| 23 | #define CMD_NOFILE_OK 0x01 |
| 24 | |
| 25 | char *progname; |
| 26 | static BlockDriverState *bs; |
| 27 | |
| 28 | static int misalign; |
| 29 | |
| 30 | /* |
Christoph Hellwig | cf070d7 | 2009-07-20 01:19:25 +0200 | [diff] [blame] | 31 | * Parse the pattern argument to various sub-commands. |
| 32 | * |
| 33 | * Because the pattern is used as an argument to memset it must evaluate |
| 34 | * to an unsigned integer that fits into a single byte. |
| 35 | */ |
| 36 | static int parse_pattern(const char *arg) |
| 37 | { |
| 38 | char *endptr = NULL; |
| 39 | long pattern; |
| 40 | |
| 41 | pattern = strtol(arg, &endptr, 0); |
| 42 | if (pattern < 0 || pattern > UCHAR_MAX || *endptr != '\0') { |
| 43 | printf("%s is not a valid pattern byte\n", arg); |
| 44 | return -1; |
| 45 | } |
| 46 | |
| 47 | return pattern; |
| 48 | } |
| 49 | |
| 50 | /* |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 51 | * Memory allocation helpers. |
| 52 | * |
| 53 | * Make sure memory is aligned by default, or purposefully misaligned if |
| 54 | * that is specified on the command line. |
| 55 | */ |
| 56 | |
| 57 | #define MISALIGN_OFFSET 16 |
| 58 | static void *qemu_io_alloc(size_t len, int pattern) |
| 59 | { |
| 60 | void *buf; |
| 61 | |
| 62 | if (misalign) |
| 63 | len += MISALIGN_OFFSET; |
| 64 | buf = qemu_memalign(512, len); |
| 65 | memset(buf, pattern, len); |
| 66 | if (misalign) |
| 67 | buf += MISALIGN_OFFSET; |
| 68 | return buf; |
| 69 | } |
| 70 | |
| 71 | static void qemu_io_free(void *p) |
| 72 | { |
| 73 | if (misalign) |
| 74 | p -= MISALIGN_OFFSET; |
| 75 | qemu_vfree(p); |
| 76 | } |
| 77 | |
| 78 | static void |
Stefan Weil | 3abcdf4 | 2009-06-21 18:35:03 +0200 | [diff] [blame] | 79 | dump_buffer(const void *buffer, int64_t offset, int len) |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 80 | { |
| 81 | int i, j; |
Stefan Weil | 3abcdf4 | 2009-06-21 18:35:03 +0200 | [diff] [blame] | 82 | const uint8_t *p; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 83 | |
| 84 | for (i = 0, p = buffer; i < len; i += 16) { |
Stefan Weil | 3abcdf4 | 2009-06-21 18:35:03 +0200 | [diff] [blame] | 85 | const uint8_t *s = p; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 86 | |
| 87 | printf("%08llx: ", (unsigned long long)offset + i); |
| 88 | for (j = 0; j < 16 && i + j < len; j++, p++) |
| 89 | printf("%02x ", *p); |
| 90 | printf(" "); |
| 91 | for (j = 0; j < 16 && i + j < len; j++, s++) { |
Stefan Weil | 3abcdf4 | 2009-06-21 18:35:03 +0200 | [diff] [blame] | 92 | if (isalnum(*s)) |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 93 | printf("%c", *s); |
| 94 | else |
| 95 | printf("."); |
| 96 | } |
| 97 | printf("\n"); |
| 98 | } |
| 99 | } |
| 100 | |
| 101 | static void |
| 102 | print_report(const char *op, struct timeval *t, int64_t offset, |
| 103 | int count, int total, int cnt, int Cflag) |
| 104 | { |
| 105 | char s1[64], s2[64], ts[64]; |
| 106 | |
| 107 | timestr(t, ts, sizeof(ts), Cflag ? VERBOSE_FIXED_TIME : 0); |
| 108 | if (!Cflag) { |
| 109 | cvtstr((double)total, s1, sizeof(s1)); |
| 110 | cvtstr(tdiv((double)total, *t), s2, sizeof(s2)); |
| 111 | printf("%s %d/%d bytes at offset %lld\n", |
| 112 | op, total, count, (long long)offset); |
| 113 | printf("%s, %d ops; %s (%s/sec and %.4f ops/sec)\n", |
| 114 | s1, cnt, ts, s2, tdiv((double)cnt, *t)); |
| 115 | } else {/* bytes,ops,time,bytes/sec,ops/sec */ |
| 116 | printf("%d,%d,%s,%.3f,%.3f\n", |
| 117 | total, cnt, ts, |
| 118 | tdiv((double)total, *t), |
| 119 | tdiv((double)cnt, *t)); |
| 120 | } |
| 121 | } |
| 122 | |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 123 | /* |
| 124 | * Parse multiple length statements for vectored I/O, and construct an I/O |
| 125 | * vector matching it. |
| 126 | */ |
| 127 | static void * |
| 128 | create_iovec(QEMUIOVector *qiov, char **argv, int nr_iov, int pattern) |
| 129 | { |
| 130 | size_t *sizes = calloc(nr_iov, sizeof(size_t)); |
| 131 | size_t count = 0; |
Kevin Wolf | 40a0d7c | 2009-11-18 10:42:59 +0100 | [diff] [blame^] | 132 | void *buf = NULL; |
| 133 | void *p; |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 134 | int i; |
| 135 | |
| 136 | for (i = 0; i < nr_iov; i++) { |
| 137 | char *arg = argv[i]; |
| 138 | long long len; |
| 139 | |
| 140 | len = cvtnum(arg); |
| 141 | if (len < 0) { |
| 142 | printf("non-numeric length argument -- %s\n", arg); |
Kevin Wolf | 40a0d7c | 2009-11-18 10:42:59 +0100 | [diff] [blame^] | 143 | goto fail; |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 144 | } |
| 145 | |
| 146 | /* should be SIZE_T_MAX, but that doesn't exist */ |
| 147 | if (len > UINT_MAX) { |
| 148 | printf("too large length argument -- %s\n", arg); |
Kevin Wolf | 40a0d7c | 2009-11-18 10:42:59 +0100 | [diff] [blame^] | 149 | goto fail; |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 150 | } |
| 151 | |
| 152 | if (len & 0x1ff) { |
| 153 | printf("length argument %lld is not sector aligned\n", |
| 154 | len); |
Kevin Wolf | 40a0d7c | 2009-11-18 10:42:59 +0100 | [diff] [blame^] | 155 | goto fail; |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 156 | } |
| 157 | |
| 158 | sizes[i] = len; |
| 159 | count += len; |
| 160 | } |
| 161 | |
| 162 | qemu_iovec_init(qiov, nr_iov); |
| 163 | |
| 164 | buf = p = qemu_io_alloc(count, pattern); |
| 165 | |
| 166 | for (i = 0; i < nr_iov; i++) { |
| 167 | qemu_iovec_add(qiov, p, sizes[i]); |
| 168 | p += sizes[i]; |
| 169 | } |
| 170 | |
Kevin Wolf | 40a0d7c | 2009-11-18 10:42:59 +0100 | [diff] [blame^] | 171 | fail: |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 172 | free(sizes); |
| 173 | return buf; |
| 174 | } |
| 175 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 176 | static int do_read(char *buf, int64_t offset, int count, int *total) |
| 177 | { |
| 178 | int ret; |
| 179 | |
| 180 | ret = bdrv_read(bs, offset >> 9, (uint8_t *)buf, count >> 9); |
| 181 | if (ret < 0) |
| 182 | return ret; |
| 183 | *total = count; |
| 184 | return 1; |
| 185 | } |
| 186 | |
| 187 | static int do_write(char *buf, int64_t offset, int count, int *total) |
| 188 | { |
| 189 | int ret; |
| 190 | |
| 191 | ret = bdrv_write(bs, offset >> 9, (uint8_t *)buf, count >> 9); |
| 192 | if (ret < 0) |
| 193 | return ret; |
| 194 | *total = count; |
| 195 | return 1; |
| 196 | } |
| 197 | |
| 198 | static int do_pread(char *buf, int64_t offset, int count, int *total) |
| 199 | { |
| 200 | *total = bdrv_pread(bs, offset, (uint8_t *)buf, count); |
| 201 | if (*total < 0) |
| 202 | return *total; |
| 203 | return 1; |
| 204 | } |
| 205 | |
| 206 | static int do_pwrite(char *buf, int64_t offset, int count, int *total) |
| 207 | { |
| 208 | *total = bdrv_pwrite(bs, offset, (uint8_t *)buf, count); |
| 209 | if (*total < 0) |
| 210 | return *total; |
| 211 | return 1; |
| 212 | } |
| 213 | |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 214 | static int do_load_vmstate(char *buf, int64_t offset, int count, int *total) |
| 215 | { |
| 216 | *total = bdrv_load_vmstate(bs, (uint8_t *)buf, offset, count); |
| 217 | if (*total < 0) |
| 218 | return *total; |
| 219 | return 1; |
| 220 | } |
| 221 | |
| 222 | static int do_save_vmstate(char *buf, int64_t offset, int count, int *total) |
| 223 | { |
| 224 | *total = bdrv_save_vmstate(bs, (uint8_t *)buf, offset, count); |
| 225 | if (*total < 0) |
| 226 | return *total; |
| 227 | return 1; |
| 228 | } |
| 229 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 230 | #define NOT_DONE 0x7fffffff |
| 231 | static void aio_rw_done(void *opaque, int ret) |
| 232 | { |
| 233 | *(int *)opaque = ret; |
| 234 | } |
| 235 | |
| 236 | static int do_aio_readv(QEMUIOVector *qiov, int64_t offset, int *total) |
| 237 | { |
| 238 | BlockDriverAIOCB *acb; |
| 239 | int async_ret = NOT_DONE; |
| 240 | |
| 241 | acb = bdrv_aio_readv(bs, offset >> 9, qiov, qiov->size >> 9, |
| 242 | aio_rw_done, &async_ret); |
| 243 | if (!acb) |
| 244 | return -EIO; |
| 245 | |
| 246 | while (async_ret == NOT_DONE) |
| 247 | qemu_aio_wait(); |
| 248 | |
| 249 | *total = qiov->size; |
| 250 | return async_ret < 0 ? async_ret : 1; |
| 251 | } |
| 252 | |
| 253 | static int do_aio_writev(QEMUIOVector *qiov, int64_t offset, int *total) |
| 254 | { |
| 255 | BlockDriverAIOCB *acb; |
| 256 | int async_ret = NOT_DONE; |
| 257 | |
| 258 | acb = bdrv_aio_writev(bs, offset >> 9, qiov, qiov->size >> 9, |
| 259 | aio_rw_done, &async_ret); |
| 260 | if (!acb) |
| 261 | return -EIO; |
| 262 | |
| 263 | while (async_ret == NOT_DONE) |
| 264 | qemu_aio_wait(); |
| 265 | |
aliguori | 7e9bbc9 | 2009-04-18 15:36:06 +0000 | [diff] [blame] | 266 | *total = qiov->size; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 267 | return async_ret < 0 ? async_ret : 1; |
| 268 | } |
| 269 | |
| 270 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 271 | static void |
| 272 | read_help(void) |
| 273 | { |
| 274 | printf( |
| 275 | "\n" |
| 276 | " reads a range of bytes from the given offset\n" |
| 277 | "\n" |
| 278 | " Example:\n" |
| 279 | " 'read -v 512 1k' - dumps 1 kilobyte read from 512 bytes into the file\n" |
| 280 | "\n" |
| 281 | " Reads a segment of the currently open file, optionally dumping it to the\n" |
| 282 | " standard output stream (with -v option) for subsequent inspection.\n" |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 283 | " -b, -- read from the VM state rather than the virtual disk\n" |
Kevin Wolf | d9654a5 | 2009-04-23 15:11:13 +0200 | [diff] [blame] | 284 | " -C, -- report statistics in a machine parsable format\n" |
| 285 | " -l, -- length for pattern verification (only with -P)\n" |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 286 | " -p, -- use bdrv_pread to read the file\n" |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 287 | " -P, -- use a pattern to verify read data\n" |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 288 | " -q, -- quite mode, do not show I/O statistics\n" |
Kevin Wolf | d9654a5 | 2009-04-23 15:11:13 +0200 | [diff] [blame] | 289 | " -s, -- start offset for pattern verification (only with -P)\n" |
| 290 | " -v, -- dump buffer to standard output\n" |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 291 | "\n"); |
| 292 | } |
| 293 | |
Blue Swirl | 22a2bdc | 2009-11-21 09:06:46 +0000 | [diff] [blame] | 294 | static int read_f(int argc, char **argv); |
| 295 | |
| 296 | static const cmdinfo_t read_cmd = { |
| 297 | .name = "read", |
| 298 | .altname = "r", |
| 299 | .cfunc = read_f, |
| 300 | .argmin = 2, |
| 301 | .argmax = -1, |
| 302 | .args = "[-abCpqv] [-P pattern [-s off] [-l len]] off len", |
| 303 | .oneline = "reads a number of bytes at a specified offset", |
| 304 | .help = read_help, |
| 305 | }; |
| 306 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 307 | static int |
| 308 | read_f(int argc, char **argv) |
| 309 | { |
| 310 | struct timeval t1, t2; |
| 311 | int Cflag = 0, pflag = 0, qflag = 0, vflag = 0; |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 312 | int Pflag = 0, sflag = 0, lflag = 0, bflag = 0; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 313 | int c, cnt; |
| 314 | char *buf; |
| 315 | int64_t offset; |
Paul Brook | d4ec522 | 2009-05-09 23:21:39 +0100 | [diff] [blame] | 316 | int count; |
| 317 | /* Some compilers get confused and warn if this is not initialized. */ |
| 318 | int total = 0; |
Kevin Wolf | d9654a5 | 2009-04-23 15:11:13 +0200 | [diff] [blame] | 319 | int pattern = 0, pattern_offset = 0, pattern_count = 0; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 320 | |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 321 | while ((c = getopt(argc, argv, "bCl:pP:qs:v")) != EOF) { |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 322 | switch (c) { |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 323 | case 'b': |
| 324 | bflag = 1; |
| 325 | break; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 326 | case 'C': |
| 327 | Cflag = 1; |
| 328 | break; |
Kevin Wolf | d9654a5 | 2009-04-23 15:11:13 +0200 | [diff] [blame] | 329 | case 'l': |
| 330 | lflag = 1; |
| 331 | pattern_count = cvtnum(optarg); |
| 332 | if (pattern_count < 0) { |
| 333 | printf("non-numeric length argument -- %s\n", optarg); |
| 334 | return 0; |
| 335 | } |
| 336 | break; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 337 | case 'p': |
| 338 | pflag = 1; |
| 339 | break; |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 340 | case 'P': |
| 341 | Pflag = 1; |
Christoph Hellwig | cf070d7 | 2009-07-20 01:19:25 +0200 | [diff] [blame] | 342 | pattern = parse_pattern(optarg); |
| 343 | if (pattern < 0) |
| 344 | return 0; |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 345 | break; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 346 | case 'q': |
| 347 | qflag = 1; |
| 348 | break; |
Kevin Wolf | d9654a5 | 2009-04-23 15:11:13 +0200 | [diff] [blame] | 349 | case 's': |
| 350 | sflag = 1; |
| 351 | pattern_offset = cvtnum(optarg); |
| 352 | if (pattern_offset < 0) { |
| 353 | printf("non-numeric length argument -- %s\n", optarg); |
| 354 | return 0; |
| 355 | } |
| 356 | break; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 357 | case 'v': |
| 358 | vflag = 1; |
| 359 | break; |
| 360 | default: |
| 361 | return command_usage(&read_cmd); |
| 362 | } |
| 363 | } |
| 364 | |
| 365 | if (optind != argc - 2) |
| 366 | return command_usage(&read_cmd); |
| 367 | |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 368 | if (bflag && pflag) { |
| 369 | printf("-b and -p cannot be specified at the same time\n"); |
| 370 | return 0; |
| 371 | } |
| 372 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 373 | offset = cvtnum(argv[optind]); |
| 374 | if (offset < 0) { |
| 375 | printf("non-numeric length argument -- %s\n", argv[optind]); |
| 376 | return 0; |
| 377 | } |
| 378 | |
| 379 | optind++; |
| 380 | count = cvtnum(argv[optind]); |
| 381 | if (count < 0) { |
| 382 | printf("non-numeric length argument -- %s\n", argv[optind]); |
| 383 | return 0; |
| 384 | } |
| 385 | |
Kevin Wolf | d9654a5 | 2009-04-23 15:11:13 +0200 | [diff] [blame] | 386 | if (!Pflag && (lflag || sflag)) { |
| 387 | return command_usage(&read_cmd); |
| 388 | } |
| 389 | |
| 390 | if (!lflag) { |
| 391 | pattern_count = count - pattern_offset; |
| 392 | } |
| 393 | |
| 394 | if ((pattern_count < 0) || (pattern_count + pattern_offset > count)) { |
| 395 | printf("pattern verfication range exceeds end of read data\n"); |
| 396 | return 0; |
| 397 | } |
| 398 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 399 | if (!pflag) |
| 400 | if (offset & 0x1ff) { |
| 401 | printf("offset %lld is not sector aligned\n", |
| 402 | (long long)offset); |
| 403 | return 0; |
| 404 | |
| 405 | if (count & 0x1ff) { |
| 406 | printf("count %d is not sector aligned\n", |
| 407 | count); |
| 408 | return 0; |
| 409 | } |
| 410 | } |
| 411 | |
| 412 | buf = qemu_io_alloc(count, 0xab); |
| 413 | |
| 414 | gettimeofday(&t1, NULL); |
| 415 | if (pflag) |
| 416 | cnt = do_pread(buf, offset, count, &total); |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 417 | else if (bflag) |
| 418 | cnt = do_load_vmstate(buf, offset, count, &total); |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 419 | else |
| 420 | cnt = do_read(buf, offset, count, &total); |
| 421 | gettimeofday(&t2, NULL); |
| 422 | |
| 423 | if (cnt < 0) { |
| 424 | printf("read failed: %s\n", strerror(-cnt)); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 425 | goto out; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 426 | } |
| 427 | |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 428 | if (Pflag) { |
Kevin Wolf | d9654a5 | 2009-04-23 15:11:13 +0200 | [diff] [blame] | 429 | void* cmp_buf = malloc(pattern_count); |
| 430 | memset(cmp_buf, pattern, pattern_count); |
| 431 | if (memcmp(buf + pattern_offset, cmp_buf, pattern_count)) { |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 432 | printf("Pattern verification failed at offset %lld, " |
| 433 | "%d bytes\n", |
Kevin Wolf | d9654a5 | 2009-04-23 15:11:13 +0200 | [diff] [blame] | 434 | (long long) offset + pattern_offset, pattern_count); |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 435 | } |
| 436 | free(cmp_buf); |
| 437 | } |
| 438 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 439 | if (qflag) |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 440 | goto out; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 441 | |
| 442 | if (vflag) |
| 443 | dump_buffer(buf, offset, count); |
| 444 | |
| 445 | /* Finally, report back -- -C gives a parsable format */ |
| 446 | t2 = tsub(t2, t1); |
| 447 | print_report("read", &t2, offset, count, total, cnt, Cflag); |
| 448 | |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 449 | out: |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 450 | qemu_io_free(buf); |
| 451 | |
| 452 | return 0; |
| 453 | } |
| 454 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 455 | static void |
| 456 | readv_help(void) |
| 457 | { |
| 458 | printf( |
| 459 | "\n" |
| 460 | " reads a range of bytes from the given offset into multiple buffers\n" |
| 461 | "\n" |
| 462 | " Example:\n" |
| 463 | " 'readv -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n" |
| 464 | "\n" |
| 465 | " Reads a segment of the currently open file, optionally dumping it to the\n" |
| 466 | " standard output stream (with -v option) for subsequent inspection.\n" |
| 467 | " Uses multiple iovec buffers if more than one byte range is specified.\n" |
| 468 | " -C, -- report statistics in a machine parsable format\n" |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 469 | " -P, -- use a pattern to verify read data\n" |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 470 | " -v, -- dump buffer to standard output\n" |
| 471 | " -q, -- quite mode, do not show I/O statistics\n" |
| 472 | "\n"); |
| 473 | } |
| 474 | |
Blue Swirl | 22a2bdc | 2009-11-21 09:06:46 +0000 | [diff] [blame] | 475 | static int readv_f(int argc, char **argv); |
| 476 | |
| 477 | static const cmdinfo_t readv_cmd = { |
| 478 | .name = "readv", |
| 479 | .cfunc = readv_f, |
| 480 | .argmin = 2, |
| 481 | .argmax = -1, |
| 482 | .args = "[-Cqv] [-P pattern ] off len [len..]", |
| 483 | .oneline = "reads a number of bytes at a specified offset", |
| 484 | .help = readv_help, |
| 485 | }; |
| 486 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 487 | static int |
| 488 | readv_f(int argc, char **argv) |
| 489 | { |
| 490 | struct timeval t1, t2; |
| 491 | int Cflag = 0, qflag = 0, vflag = 0; |
| 492 | int c, cnt; |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 493 | char *buf; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 494 | int64_t offset; |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 495 | int total; |
| 496 | int nr_iov; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 497 | QEMUIOVector qiov; |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 498 | int pattern = 0; |
| 499 | int Pflag = 0; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 500 | |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 501 | while ((c = getopt(argc, argv, "CP:qv")) != EOF) { |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 502 | switch (c) { |
| 503 | case 'C': |
| 504 | Cflag = 1; |
| 505 | break; |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 506 | case 'P': |
| 507 | Pflag = 1; |
Christoph Hellwig | cf070d7 | 2009-07-20 01:19:25 +0200 | [diff] [blame] | 508 | pattern = parse_pattern(optarg); |
| 509 | if (pattern < 0) |
| 510 | return 0; |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 511 | break; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 512 | case 'q': |
| 513 | qflag = 1; |
| 514 | break; |
| 515 | case 'v': |
| 516 | vflag = 1; |
| 517 | break; |
| 518 | default: |
| 519 | return command_usage(&readv_cmd); |
| 520 | } |
| 521 | } |
| 522 | |
| 523 | if (optind > argc - 2) |
| 524 | return command_usage(&readv_cmd); |
| 525 | |
| 526 | |
| 527 | offset = cvtnum(argv[optind]); |
| 528 | if (offset < 0) { |
| 529 | printf("non-numeric length argument -- %s\n", argv[optind]); |
| 530 | return 0; |
| 531 | } |
| 532 | optind++; |
| 533 | |
| 534 | if (offset & 0x1ff) { |
| 535 | printf("offset %lld is not sector aligned\n", |
| 536 | (long long)offset); |
| 537 | return 0; |
| 538 | } |
| 539 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 540 | nr_iov = argc - optind; |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 541 | buf = create_iovec(&qiov, &argv[optind], nr_iov, 0xab); |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 542 | |
| 543 | gettimeofday(&t1, NULL); |
| 544 | cnt = do_aio_readv(&qiov, offset, &total); |
| 545 | gettimeofday(&t2, NULL); |
| 546 | |
| 547 | if (cnt < 0) { |
| 548 | printf("readv failed: %s\n", strerror(-cnt)); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 549 | goto out; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 550 | } |
| 551 | |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 552 | if (Pflag) { |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 553 | void* cmp_buf = malloc(qiov.size); |
| 554 | memset(cmp_buf, pattern, qiov.size); |
| 555 | if (memcmp(buf, cmp_buf, qiov.size)) { |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 556 | printf("Pattern verification failed at offset %lld, " |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 557 | "%zd bytes\n", |
| 558 | (long long) offset, qiov.size); |
aliguori | c48101a | 2009-04-18 15:36:23 +0000 | [diff] [blame] | 559 | } |
| 560 | free(cmp_buf); |
| 561 | } |
| 562 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 563 | if (qflag) |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 564 | goto out; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 565 | |
| 566 | if (vflag) |
| 567 | dump_buffer(buf, offset, qiov.size); |
| 568 | |
| 569 | /* Finally, report back -- -C gives a parsable format */ |
| 570 | t2 = tsub(t2, t1); |
| 571 | print_report("read", &t2, offset, qiov.size, total, cnt, Cflag); |
| 572 | |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 573 | out: |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 574 | qemu_io_free(buf); |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 575 | return 0; |
| 576 | } |
| 577 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 578 | static void |
| 579 | write_help(void) |
| 580 | { |
| 581 | printf( |
| 582 | "\n" |
| 583 | " writes a range of bytes from the given offset\n" |
| 584 | "\n" |
| 585 | " Example:\n" |
| 586 | " 'write 512 1k' - writes 1 kilobyte at 512 bytes into the open file\n" |
| 587 | "\n" |
| 588 | " Writes into a segment of the currently open file, using a buffer\n" |
| 589 | " filled with a set pattern (0xcdcdcdcd).\n" |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 590 | " -b, -- write to the VM state rather than the virtual disk\n" |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 591 | " -p, -- use bdrv_pwrite to write the file\n" |
| 592 | " -P, -- use different pattern to fill file\n" |
| 593 | " -C, -- report statistics in a machine parsable format\n" |
| 594 | " -q, -- quite mode, do not show I/O statistics\n" |
| 595 | "\n"); |
| 596 | } |
| 597 | |
Blue Swirl | 22a2bdc | 2009-11-21 09:06:46 +0000 | [diff] [blame] | 598 | static int write_f(int argc, char **argv); |
| 599 | |
| 600 | static const cmdinfo_t write_cmd = { |
| 601 | .name = "write", |
| 602 | .altname = "w", |
| 603 | .cfunc = write_f, |
| 604 | .argmin = 2, |
| 605 | .argmax = -1, |
| 606 | .args = "[-abCpq] [-P pattern ] off len", |
| 607 | .oneline = "writes a number of bytes at a specified offset", |
| 608 | .help = write_help, |
| 609 | }; |
| 610 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 611 | static int |
| 612 | write_f(int argc, char **argv) |
| 613 | { |
| 614 | struct timeval t1, t2; |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 615 | int Cflag = 0, pflag = 0, qflag = 0, bflag = 0; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 616 | int c, cnt; |
| 617 | char *buf; |
| 618 | int64_t offset; |
Paul Brook | d4ec522 | 2009-05-09 23:21:39 +0100 | [diff] [blame] | 619 | int count; |
| 620 | /* Some compilers get confused and warn if this is not initialized. */ |
| 621 | int total = 0; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 622 | int pattern = 0xcd; |
| 623 | |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 624 | while ((c = getopt(argc, argv, "bCpP:q")) != EOF) { |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 625 | switch (c) { |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 626 | case 'b': |
| 627 | bflag = 1; |
| 628 | break; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 629 | case 'C': |
| 630 | Cflag = 1; |
| 631 | break; |
| 632 | case 'p': |
| 633 | pflag = 1; |
| 634 | break; |
| 635 | case 'P': |
Christoph Hellwig | cf070d7 | 2009-07-20 01:19:25 +0200 | [diff] [blame] | 636 | pattern = parse_pattern(optarg); |
| 637 | if (pattern < 0) |
| 638 | return 0; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 639 | break; |
| 640 | case 'q': |
| 641 | qflag = 1; |
| 642 | break; |
| 643 | default: |
| 644 | return command_usage(&write_cmd); |
| 645 | } |
| 646 | } |
| 647 | |
| 648 | if (optind != argc - 2) |
| 649 | return command_usage(&write_cmd); |
| 650 | |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 651 | if (bflag && pflag) { |
| 652 | printf("-b and -p cannot be specified at the same time\n"); |
| 653 | return 0; |
| 654 | } |
| 655 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 656 | offset = cvtnum(argv[optind]); |
| 657 | if (offset < 0) { |
| 658 | printf("non-numeric length argument -- %s\n", argv[optind]); |
| 659 | return 0; |
| 660 | } |
| 661 | |
| 662 | optind++; |
| 663 | count = cvtnum(argv[optind]); |
| 664 | if (count < 0) { |
| 665 | printf("non-numeric length argument -- %s\n", argv[optind]); |
| 666 | return 0; |
| 667 | } |
| 668 | |
| 669 | if (!pflag) { |
| 670 | if (offset & 0x1ff) { |
| 671 | printf("offset %lld is not sector aligned\n", |
| 672 | (long long)offset); |
| 673 | return 0; |
| 674 | } |
| 675 | |
| 676 | if (count & 0x1ff) { |
| 677 | printf("count %d is not sector aligned\n", |
| 678 | count); |
| 679 | return 0; |
| 680 | } |
| 681 | } |
| 682 | |
| 683 | buf = qemu_io_alloc(count, pattern); |
| 684 | |
| 685 | gettimeofday(&t1, NULL); |
| 686 | if (pflag) |
| 687 | cnt = do_pwrite(buf, offset, count, &total); |
Kevin Wolf | ca94dbc | 2009-07-15 12:06:58 +0200 | [diff] [blame] | 688 | else if (bflag) |
| 689 | cnt = do_save_vmstate(buf, offset, count, &total); |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 690 | else |
| 691 | cnt = do_write(buf, offset, count, &total); |
| 692 | gettimeofday(&t2, NULL); |
| 693 | |
| 694 | if (cnt < 0) { |
| 695 | printf("write failed: %s\n", strerror(-cnt)); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 696 | goto out; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 697 | } |
| 698 | |
| 699 | if (qflag) |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 700 | goto out; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 701 | |
| 702 | /* Finally, report back -- -C gives a parsable format */ |
| 703 | t2 = tsub(t2, t1); |
| 704 | print_report("wrote", &t2, offset, count, total, cnt, Cflag); |
| 705 | |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 706 | out: |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 707 | qemu_io_free(buf); |
| 708 | |
| 709 | return 0; |
| 710 | } |
| 711 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 712 | static void |
| 713 | writev_help(void) |
| 714 | { |
| 715 | printf( |
| 716 | "\n" |
| 717 | " writes a range of bytes from the given offset source from multiple buffers\n" |
| 718 | "\n" |
| 719 | " Example:\n" |
| 720 | " 'write 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n" |
| 721 | "\n" |
| 722 | " Writes into a segment of the currently open file, using a buffer\n" |
| 723 | " filled with a set pattern (0xcdcdcdcd).\n" |
| 724 | " -P, -- use different pattern to fill file\n" |
| 725 | " -C, -- report statistics in a machine parsable format\n" |
| 726 | " -q, -- quite mode, do not show I/O statistics\n" |
| 727 | "\n"); |
| 728 | } |
| 729 | |
Blue Swirl | 22a2bdc | 2009-11-21 09:06:46 +0000 | [diff] [blame] | 730 | static int writev_f(int argc, char **argv); |
| 731 | |
| 732 | static const cmdinfo_t writev_cmd = { |
| 733 | .name = "writev", |
| 734 | .cfunc = writev_f, |
| 735 | .argmin = 2, |
| 736 | .argmax = -1, |
| 737 | .args = "[-Cq] [-P pattern ] off len [len..]", |
| 738 | .oneline = "writes a number of bytes at a specified offset", |
| 739 | .help = writev_help, |
| 740 | }; |
| 741 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 742 | static int |
| 743 | writev_f(int argc, char **argv) |
| 744 | { |
| 745 | struct timeval t1, t2; |
| 746 | int Cflag = 0, qflag = 0; |
| 747 | int c, cnt; |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 748 | char *buf; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 749 | int64_t offset; |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 750 | int total; |
| 751 | int nr_iov; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 752 | int pattern = 0xcd; |
| 753 | QEMUIOVector qiov; |
| 754 | |
| 755 | while ((c = getopt(argc, argv, "CqP:")) != EOF) { |
| 756 | switch (c) { |
| 757 | case 'C': |
| 758 | Cflag = 1; |
| 759 | break; |
| 760 | case 'q': |
| 761 | qflag = 1; |
| 762 | break; |
| 763 | case 'P': |
Christoph Hellwig | cf070d7 | 2009-07-20 01:19:25 +0200 | [diff] [blame] | 764 | pattern = parse_pattern(optarg); |
| 765 | if (pattern < 0) |
| 766 | return 0; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 767 | break; |
| 768 | default: |
| 769 | return command_usage(&writev_cmd); |
| 770 | } |
| 771 | } |
| 772 | |
| 773 | if (optind > argc - 2) |
| 774 | return command_usage(&writev_cmd); |
| 775 | |
| 776 | offset = cvtnum(argv[optind]); |
| 777 | if (offset < 0) { |
| 778 | printf("non-numeric length argument -- %s\n", argv[optind]); |
| 779 | return 0; |
| 780 | } |
| 781 | optind++; |
| 782 | |
| 783 | if (offset & 0x1ff) { |
| 784 | printf("offset %lld is not sector aligned\n", |
| 785 | (long long)offset); |
| 786 | return 0; |
| 787 | } |
| 788 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 789 | nr_iov = argc - optind; |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 790 | buf = create_iovec(&qiov, &argv[optind], nr_iov, pattern); |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 791 | |
| 792 | gettimeofday(&t1, NULL); |
| 793 | cnt = do_aio_writev(&qiov, offset, &total); |
| 794 | gettimeofday(&t2, NULL); |
| 795 | |
| 796 | if (cnt < 0) { |
| 797 | printf("writev failed: %s\n", strerror(-cnt)); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 798 | goto out; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 799 | } |
| 800 | |
| 801 | if (qflag) |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 802 | goto out; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 803 | |
| 804 | /* Finally, report back -- -C gives a parsable format */ |
| 805 | t2 = tsub(t2, t1); |
| 806 | print_report("wrote", &t2, offset, qiov.size, total, cnt, Cflag); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 807 | out: |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 808 | qemu_io_free(buf); |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 809 | return 0; |
| 810 | } |
| 811 | |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 812 | struct aio_ctx { |
| 813 | QEMUIOVector qiov; |
| 814 | int64_t offset; |
| 815 | char *buf; |
| 816 | int qflag; |
| 817 | int vflag; |
| 818 | int Cflag; |
| 819 | int Pflag; |
| 820 | int pattern; |
| 821 | struct timeval t1; |
| 822 | }; |
| 823 | |
| 824 | static void |
| 825 | aio_write_done(void *opaque, int ret) |
| 826 | { |
| 827 | struct aio_ctx *ctx = opaque; |
| 828 | struct timeval t2; |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 829 | |
| 830 | gettimeofday(&t2, NULL); |
| 831 | |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 832 | |
| 833 | if (ret < 0) { |
| 834 | printf("aio_write failed: %s\n", strerror(-ret)); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 835 | goto out; |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 836 | } |
| 837 | |
Christoph Hellwig | 230d4fa | 2009-07-10 13:33:38 +0200 | [diff] [blame] | 838 | if (ctx->qflag) { |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 839 | goto out; |
Christoph Hellwig | 230d4fa | 2009-07-10 13:33:38 +0200 | [diff] [blame] | 840 | } |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 841 | |
| 842 | /* Finally, report back -- -C gives a parsable format */ |
| 843 | t2 = tsub(t2, ctx->t1); |
Christoph Hellwig | 230d4fa | 2009-07-10 13:33:38 +0200 | [diff] [blame] | 844 | print_report("wrote", &t2, ctx->offset, ctx->qiov.size, |
| 845 | ctx->qiov.size, 1, ctx->Cflag); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 846 | out: |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 847 | qemu_io_free(ctx->buf); |
| 848 | free(ctx); |
| 849 | } |
| 850 | |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 851 | static void |
| 852 | aio_read_done(void *opaque, int ret) |
| 853 | { |
| 854 | struct aio_ctx *ctx = opaque; |
| 855 | struct timeval t2; |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 856 | |
| 857 | gettimeofday(&t2, NULL); |
| 858 | |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 859 | if (ret < 0) { |
| 860 | printf("readv failed: %s\n", strerror(-ret)); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 861 | goto out; |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 862 | } |
| 863 | |
| 864 | if (ctx->Pflag) { |
Christoph Hellwig | 230d4fa | 2009-07-10 13:33:38 +0200 | [diff] [blame] | 865 | void *cmp_buf = malloc(ctx->qiov.size); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 866 | |
Christoph Hellwig | 230d4fa | 2009-07-10 13:33:38 +0200 | [diff] [blame] | 867 | memset(cmp_buf, ctx->pattern, ctx->qiov.size); |
| 868 | if (memcmp(ctx->buf, cmp_buf, ctx->qiov.size)) { |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 869 | printf("Pattern verification failed at offset %lld, " |
Christoph Hellwig | 230d4fa | 2009-07-10 13:33:38 +0200 | [diff] [blame] | 870 | "%zd bytes\n", |
| 871 | (long long) ctx->offset, ctx->qiov.size); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 872 | } |
| 873 | free(cmp_buf); |
| 874 | } |
| 875 | |
Christoph Hellwig | 230d4fa | 2009-07-10 13:33:38 +0200 | [diff] [blame] | 876 | if (ctx->qflag) { |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 877 | goto out; |
Christoph Hellwig | 230d4fa | 2009-07-10 13:33:38 +0200 | [diff] [blame] | 878 | } |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 879 | |
Christoph Hellwig | 230d4fa | 2009-07-10 13:33:38 +0200 | [diff] [blame] | 880 | if (ctx->vflag) { |
| 881 | dump_buffer(ctx->buf, ctx->offset, ctx->qiov.size); |
| 882 | } |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 883 | |
| 884 | /* Finally, report back -- -C gives a parsable format */ |
| 885 | t2 = tsub(t2, ctx->t1); |
Christoph Hellwig | 230d4fa | 2009-07-10 13:33:38 +0200 | [diff] [blame] | 886 | print_report("read", &t2, ctx->offset, ctx->qiov.size, |
| 887 | ctx->qiov.size, 1, ctx->Cflag); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 888 | out: |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 889 | qemu_io_free(ctx->buf); |
| 890 | free(ctx); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 891 | } |
| 892 | |
| 893 | static void |
| 894 | aio_read_help(void) |
| 895 | { |
| 896 | printf( |
| 897 | "\n" |
| 898 | " asynchronously reads a range of bytes from the given offset\n" |
| 899 | "\n" |
| 900 | " Example:\n" |
| 901 | " 'aio_read -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n" |
| 902 | "\n" |
| 903 | " Reads a segment of the currently open file, optionally dumping it to the\n" |
| 904 | " standard output stream (with -v option) for subsequent inspection.\n" |
| 905 | " The read is performed asynchronously and should the aio_flush command \n" |
| 906 | " should be used to ensure all outstanding aio requests have been completed\n" |
| 907 | " -C, -- report statistics in a machine parsable format\n" |
| 908 | " -P, -- use a pattern to verify read data\n" |
| 909 | " -v, -- dump buffer to standard output\n" |
| 910 | " -q, -- quite mode, do not show I/O statistics\n" |
| 911 | "\n"); |
| 912 | } |
| 913 | |
Blue Swirl | 22a2bdc | 2009-11-21 09:06:46 +0000 | [diff] [blame] | 914 | static int aio_read_f(int argc, char **argv); |
| 915 | |
| 916 | static const cmdinfo_t aio_read_cmd = { |
| 917 | .name = "aio_read", |
| 918 | .cfunc = aio_read_f, |
| 919 | .argmin = 2, |
| 920 | .argmax = -1, |
| 921 | .args = "[-Cqv] [-P pattern ] off len [len..]", |
| 922 | .oneline = "asynchronously reads a number of bytes", |
| 923 | .help = aio_read_help, |
| 924 | }; |
| 925 | |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 926 | static int |
| 927 | aio_read_f(int argc, char **argv) |
| 928 | { |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 929 | int nr_iov, c; |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 930 | struct aio_ctx *ctx = calloc(1, sizeof(struct aio_ctx)); |
| 931 | BlockDriverAIOCB *acb; |
| 932 | |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 933 | while ((c = getopt(argc, argv, "CP:qv")) != EOF) { |
| 934 | switch (c) { |
| 935 | case 'C': |
| 936 | ctx->Cflag = 1; |
| 937 | break; |
| 938 | case 'P': |
| 939 | ctx->Pflag = 1; |
Christoph Hellwig | cf070d7 | 2009-07-20 01:19:25 +0200 | [diff] [blame] | 940 | ctx->pattern = parse_pattern(optarg); |
| 941 | if (ctx->pattern < 0) |
| 942 | return 0; |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 943 | break; |
| 944 | case 'q': |
| 945 | ctx->qflag = 1; |
| 946 | break; |
| 947 | case 'v': |
| 948 | ctx->vflag = 1; |
| 949 | break; |
| 950 | default: |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 951 | free(ctx); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 952 | return command_usage(&aio_read_cmd); |
| 953 | } |
| 954 | } |
| 955 | |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 956 | if (optind > argc - 2) { |
| 957 | free(ctx); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 958 | return command_usage(&aio_read_cmd); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 959 | } |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 960 | |
| 961 | ctx->offset = cvtnum(argv[optind]); |
| 962 | if (ctx->offset < 0) { |
| 963 | printf("non-numeric length argument -- %s\n", argv[optind]); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 964 | free(ctx); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 965 | return 0; |
| 966 | } |
| 967 | optind++; |
| 968 | |
| 969 | if (ctx->offset & 0x1ff) { |
| 970 | printf("offset %lld is not sector aligned\n", |
| 971 | (long long)ctx->offset); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 972 | free(ctx); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 973 | return 0; |
| 974 | } |
| 975 | |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 976 | nr_iov = argc - optind; |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 977 | ctx->buf = create_iovec(&ctx->qiov, &argv[optind], nr_iov, 0xab); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 978 | |
| 979 | gettimeofday(&ctx->t1, NULL); |
| 980 | acb = bdrv_aio_readv(bs, ctx->offset >> 9, &ctx->qiov, |
| 981 | ctx->qiov.size >> 9, aio_read_done, ctx); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 982 | if (!acb) { |
| 983 | free(ctx->buf); |
| 984 | free(ctx); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 985 | return -EIO; |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 986 | } |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 987 | |
| 988 | return 0; |
| 989 | } |
| 990 | |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 991 | static void |
| 992 | aio_write_help(void) |
| 993 | { |
| 994 | printf( |
| 995 | "\n" |
| 996 | " asynchronously writes a range of bytes from the given offset source \n" |
| 997 | " from multiple buffers\n" |
| 998 | "\n" |
| 999 | " Example:\n" |
| 1000 | " 'aio_write 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n" |
| 1001 | "\n" |
| 1002 | " Writes into a segment of the currently open file, using a buffer\n" |
| 1003 | " filled with a set pattern (0xcdcdcdcd).\n" |
| 1004 | " The write is performed asynchronously and should the aio_flush command \n" |
| 1005 | " should be used to ensure all outstanding aio requests have been completed\n" |
| 1006 | " -P, -- use different pattern to fill file\n" |
| 1007 | " -C, -- report statistics in a machine parsable format\n" |
| 1008 | " -q, -- quite mode, do not show I/O statistics\n" |
| 1009 | "\n"); |
| 1010 | } |
| 1011 | |
Blue Swirl | 22a2bdc | 2009-11-21 09:06:46 +0000 | [diff] [blame] | 1012 | static int aio_write_f(int argc, char **argv); |
| 1013 | |
| 1014 | static const cmdinfo_t aio_write_cmd = { |
| 1015 | .name = "aio_write", |
| 1016 | .cfunc = aio_write_f, |
| 1017 | .argmin = 2, |
| 1018 | .argmax = -1, |
| 1019 | .args = "[-Cq] [-P pattern ] off len [len..]", |
| 1020 | .oneline = "asynchronously writes a number of bytes", |
| 1021 | .help = aio_write_help, |
| 1022 | }; |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1023 | |
| 1024 | static int |
| 1025 | aio_write_f(int argc, char **argv) |
| 1026 | { |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 1027 | int nr_iov, c; |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1028 | int pattern = 0xcd; |
| 1029 | struct aio_ctx *ctx = calloc(1, sizeof(struct aio_ctx)); |
| 1030 | BlockDriverAIOCB *acb; |
| 1031 | |
| 1032 | while ((c = getopt(argc, argv, "CqP:")) != EOF) { |
| 1033 | switch (c) { |
| 1034 | case 'C': |
| 1035 | ctx->Cflag = 1; |
| 1036 | break; |
| 1037 | case 'q': |
| 1038 | ctx->qflag = 1; |
| 1039 | break; |
| 1040 | case 'P': |
Christoph Hellwig | cf070d7 | 2009-07-20 01:19:25 +0200 | [diff] [blame] | 1041 | pattern = parse_pattern(optarg); |
| 1042 | if (pattern < 0) |
| 1043 | return 0; |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1044 | break; |
| 1045 | default: |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 1046 | free(ctx); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1047 | return command_usage(&aio_write_cmd); |
| 1048 | } |
| 1049 | } |
| 1050 | |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 1051 | if (optind > argc - 2) { |
| 1052 | free(ctx); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1053 | return command_usage(&aio_write_cmd); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 1054 | } |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1055 | |
| 1056 | ctx->offset = cvtnum(argv[optind]); |
| 1057 | if (ctx->offset < 0) { |
| 1058 | printf("non-numeric length argument -- %s\n", argv[optind]); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 1059 | free(ctx); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1060 | return 0; |
| 1061 | } |
| 1062 | optind++; |
| 1063 | |
| 1064 | if (ctx->offset & 0x1ff) { |
| 1065 | printf("offset %lld is not sector aligned\n", |
| 1066 | (long long)ctx->offset); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 1067 | free(ctx); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1068 | return 0; |
| 1069 | } |
| 1070 | |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1071 | nr_iov = argc - optind; |
Christoph Hellwig | cf57298 | 2009-07-10 13:33:42 +0200 | [diff] [blame] | 1072 | ctx->buf = create_iovec(&ctx->qiov, &argv[optind], nr_iov, pattern); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1073 | |
| 1074 | gettimeofday(&ctx->t1, NULL); |
| 1075 | acb = bdrv_aio_writev(bs, ctx->offset >> 9, &ctx->qiov, |
| 1076 | ctx->qiov.size >> 9, aio_write_done, ctx); |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 1077 | if (!acb) { |
| 1078 | free(ctx->buf); |
| 1079 | free(ctx); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1080 | return -EIO; |
Kevin Wolf | 7d8abfc | 2009-07-10 13:33:52 +0200 | [diff] [blame] | 1081 | } |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1082 | |
| 1083 | return 0; |
| 1084 | } |
| 1085 | |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1086 | static int |
| 1087 | aio_flush_f(int argc, char **argv) |
| 1088 | { |
| 1089 | qemu_aio_flush(); |
| 1090 | return 0; |
| 1091 | } |
| 1092 | |
| 1093 | static const cmdinfo_t aio_flush_cmd = { |
| 1094 | .name = "aio_flush", |
| 1095 | .cfunc = aio_flush_f, |
| 1096 | .oneline = "completes all outstanding aio requets" |
| 1097 | }; |
| 1098 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1099 | static int |
| 1100 | flush_f(int argc, char **argv) |
| 1101 | { |
| 1102 | bdrv_flush(bs); |
| 1103 | return 0; |
| 1104 | } |
| 1105 | |
| 1106 | static const cmdinfo_t flush_cmd = { |
| 1107 | .name = "flush", |
| 1108 | .altname = "f", |
| 1109 | .cfunc = flush_f, |
| 1110 | .oneline = "flush all in-core file state to disk", |
| 1111 | }; |
| 1112 | |
| 1113 | static int |
| 1114 | truncate_f(int argc, char **argv) |
| 1115 | { |
| 1116 | int64_t offset; |
| 1117 | int ret; |
| 1118 | |
| 1119 | offset = cvtnum(argv[1]); |
| 1120 | if (offset < 0) { |
| 1121 | printf("non-numeric truncate argument -- %s\n", argv[1]); |
| 1122 | return 0; |
| 1123 | } |
| 1124 | |
| 1125 | ret = bdrv_truncate(bs, offset); |
| 1126 | if (ret < 0) { |
| 1127 | printf("truncate: %s", strerror(ret)); |
| 1128 | return 0; |
| 1129 | } |
| 1130 | |
| 1131 | return 0; |
| 1132 | } |
| 1133 | |
| 1134 | static const cmdinfo_t truncate_cmd = { |
| 1135 | .name = "truncate", |
| 1136 | .altname = "t", |
| 1137 | .cfunc = truncate_f, |
| 1138 | .argmin = 1, |
| 1139 | .argmax = 1, |
| 1140 | .args = "off", |
| 1141 | .oneline = "truncates the current file at the given offset", |
| 1142 | }; |
| 1143 | |
| 1144 | static int |
| 1145 | length_f(int argc, char **argv) |
| 1146 | { |
| 1147 | int64_t size; |
| 1148 | char s1[64]; |
| 1149 | |
| 1150 | size = bdrv_getlength(bs); |
| 1151 | if (size < 0) { |
| 1152 | printf("getlength: %s", strerror(size)); |
| 1153 | return 0; |
| 1154 | } |
| 1155 | |
| 1156 | cvtstr(size, s1, sizeof(s1)); |
| 1157 | printf("%s\n", s1); |
| 1158 | return 0; |
| 1159 | } |
| 1160 | |
| 1161 | |
| 1162 | static const cmdinfo_t length_cmd = { |
| 1163 | .name = "length", |
| 1164 | .altname = "l", |
| 1165 | .cfunc = length_f, |
| 1166 | .oneline = "gets the length of the current file", |
| 1167 | }; |
| 1168 | |
| 1169 | |
| 1170 | static int |
| 1171 | info_f(int argc, char **argv) |
| 1172 | { |
| 1173 | BlockDriverInfo bdi; |
| 1174 | char s1[64], s2[64]; |
| 1175 | int ret; |
| 1176 | |
| 1177 | if (bs->drv && bs->drv->format_name) |
| 1178 | printf("format name: %s\n", bs->drv->format_name); |
| 1179 | if (bs->drv && bs->drv->protocol_name) |
| 1180 | printf("format name: %s\n", bs->drv->protocol_name); |
| 1181 | |
| 1182 | ret = bdrv_get_info(bs, &bdi); |
| 1183 | if (ret) |
| 1184 | return 0; |
| 1185 | |
| 1186 | cvtstr(bdi.cluster_size, s1, sizeof(s1)); |
| 1187 | cvtstr(bdi.vm_state_offset, s2, sizeof(s2)); |
| 1188 | |
| 1189 | printf("cluster size: %s\n", s1); |
| 1190 | printf("vm state offset: %s\n", s2); |
| 1191 | |
| 1192 | return 0; |
| 1193 | } |
| 1194 | |
| 1195 | |
| 1196 | |
| 1197 | static const cmdinfo_t info_cmd = { |
| 1198 | .name = "info", |
| 1199 | .altname = "i", |
| 1200 | .cfunc = info_f, |
| 1201 | .oneline = "prints information about the current file", |
| 1202 | }; |
| 1203 | |
| 1204 | static int |
| 1205 | alloc_f(int argc, char **argv) |
| 1206 | { |
| 1207 | int64_t offset; |
Kevin Wolf | a7824a8 | 2009-07-20 16:48:43 +0200 | [diff] [blame] | 1208 | int nb_sectors, remaining; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1209 | char s1[64]; |
Kevin Wolf | a7824a8 | 2009-07-20 16:48:43 +0200 | [diff] [blame] | 1210 | int num, sum_alloc; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1211 | int ret; |
| 1212 | |
| 1213 | offset = cvtnum(argv[1]); |
| 1214 | if (offset & 0x1ff) { |
| 1215 | printf("offset %lld is not sector aligned\n", |
| 1216 | (long long)offset); |
| 1217 | return 0; |
| 1218 | } |
| 1219 | |
| 1220 | if (argc == 3) |
| 1221 | nb_sectors = cvtnum(argv[2]); |
| 1222 | else |
| 1223 | nb_sectors = 1; |
| 1224 | |
Kevin Wolf | a7824a8 | 2009-07-20 16:48:43 +0200 | [diff] [blame] | 1225 | remaining = nb_sectors; |
| 1226 | sum_alloc = 0; |
| 1227 | while (remaining) { |
| 1228 | ret = bdrv_is_allocated(bs, offset >> 9, nb_sectors, &num); |
| 1229 | remaining -= num; |
| 1230 | if (ret) { |
| 1231 | sum_alloc += num; |
| 1232 | } |
| 1233 | } |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1234 | |
| 1235 | cvtstr(offset, s1, sizeof(s1)); |
| 1236 | |
| 1237 | if (nb_sectors == 1) |
Kevin Wolf | a7824a8 | 2009-07-20 16:48:43 +0200 | [diff] [blame] | 1238 | printf("sector allocated at offset %s\n", s1); |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1239 | else |
Kevin Wolf | a7824a8 | 2009-07-20 16:48:43 +0200 | [diff] [blame] | 1240 | printf("%d/%d sectors allocated at offset %s\n", |
| 1241 | sum_alloc, nb_sectors, s1); |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1242 | return 0; |
| 1243 | } |
| 1244 | |
| 1245 | static const cmdinfo_t alloc_cmd = { |
| 1246 | .name = "alloc", |
| 1247 | .altname = "a", |
| 1248 | .argmin = 1, |
| 1249 | .argmax = 2, |
| 1250 | .cfunc = alloc_f, |
| 1251 | .args = "off [sectors]", |
| 1252 | .oneline = "checks if a sector is present in the file", |
| 1253 | }; |
| 1254 | |
| 1255 | static int |
| 1256 | close_f(int argc, char **argv) |
| 1257 | { |
| 1258 | bdrv_close(bs); |
| 1259 | bs = NULL; |
| 1260 | return 0; |
| 1261 | } |
| 1262 | |
| 1263 | static const cmdinfo_t close_cmd = { |
| 1264 | .name = "close", |
| 1265 | .altname = "c", |
| 1266 | .cfunc = close_f, |
| 1267 | .oneline = "close the current open file", |
| 1268 | }; |
| 1269 | |
Christoph Hellwig | 9c4bab2 | 2009-07-10 13:33:47 +0200 | [diff] [blame] | 1270 | static int openfile(char *name, int flags, int growable) |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1271 | { |
| 1272 | if (bs) { |
| 1273 | fprintf(stderr, "file open already, try 'help close'\n"); |
| 1274 | return 1; |
| 1275 | } |
| 1276 | |
| 1277 | bs = bdrv_new("hda"); |
| 1278 | if (!bs) |
| 1279 | return 1; |
| 1280 | |
Christoph Hellwig | 1db6947 | 2009-07-15 23:11:21 +0200 | [diff] [blame] | 1281 | if (growable) { |
| 1282 | flags |= BDRV_O_FILE; |
| 1283 | } |
| 1284 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1285 | if (bdrv_open(bs, name, flags) == -1) { |
| 1286 | fprintf(stderr, "%s: can't open device %s\n", progname, name); |
| 1287 | bs = NULL; |
| 1288 | return 1; |
| 1289 | } |
| 1290 | |
Christoph Hellwig | 9c4bab2 | 2009-07-10 13:33:47 +0200 | [diff] [blame] | 1291 | if (growable) { |
Christoph Hellwig | 9c4bab2 | 2009-07-10 13:33:47 +0200 | [diff] [blame] | 1292 | bs->growable = 1; |
| 1293 | } |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1294 | return 0; |
| 1295 | } |
| 1296 | |
| 1297 | static void |
| 1298 | open_help(void) |
| 1299 | { |
| 1300 | printf( |
| 1301 | "\n" |
| 1302 | " opens a new file in the requested mode\n" |
| 1303 | "\n" |
| 1304 | " Example:\n" |
| 1305 | " 'open -Cn /tmp/data' - creates/opens data file read-write and uncached\n" |
| 1306 | "\n" |
| 1307 | " Opens a file for subsequent use by all of the other qemu-io commands.\n" |
| 1308 | " -C, -- create new file if it doesn't exist\n" |
| 1309 | " -r, -- open file read-only\n" |
| 1310 | " -s, -- use snapshot file\n" |
| 1311 | " -n, -- disable host cache\n" |
Christoph Hellwig | 9c4bab2 | 2009-07-10 13:33:47 +0200 | [diff] [blame] | 1312 | " -g, -- allow file to grow (only applies to protocols)" |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1313 | "\n"); |
| 1314 | } |
| 1315 | |
Blue Swirl | 22a2bdc | 2009-11-21 09:06:46 +0000 | [diff] [blame] | 1316 | static int open_f(int argc, char **argv); |
| 1317 | |
| 1318 | static const cmdinfo_t open_cmd = { |
| 1319 | .name = "open", |
| 1320 | .altname = "o", |
| 1321 | .cfunc = open_f, |
| 1322 | .argmin = 1, |
| 1323 | .argmax = -1, |
| 1324 | .flags = CMD_NOFILE_OK, |
| 1325 | .args = "[-Crsn] [path]", |
| 1326 | .oneline = "open the file specified by path", |
| 1327 | .help = open_help, |
| 1328 | }; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1329 | |
| 1330 | static int |
| 1331 | open_f(int argc, char **argv) |
| 1332 | { |
| 1333 | int flags = 0; |
| 1334 | int readonly = 0; |
Christoph Hellwig | 9c4bab2 | 2009-07-10 13:33:47 +0200 | [diff] [blame] | 1335 | int growable = 0; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1336 | int c; |
| 1337 | |
Christoph Hellwig | 9c4bab2 | 2009-07-10 13:33:47 +0200 | [diff] [blame] | 1338 | while ((c = getopt(argc, argv, "snCrg")) != EOF) { |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1339 | switch (c) { |
| 1340 | case 's': |
| 1341 | flags |= BDRV_O_SNAPSHOT; |
| 1342 | break; |
| 1343 | case 'n': |
| 1344 | flags |= BDRV_O_NOCACHE; |
| 1345 | break; |
| 1346 | case 'C': |
| 1347 | flags |= BDRV_O_CREAT; |
| 1348 | break; |
| 1349 | case 'r': |
| 1350 | readonly = 1; |
| 1351 | break; |
Christoph Hellwig | 9c4bab2 | 2009-07-10 13:33:47 +0200 | [diff] [blame] | 1352 | case 'g': |
| 1353 | growable = 1; |
| 1354 | break; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1355 | default: |
| 1356 | return command_usage(&open_cmd); |
| 1357 | } |
| 1358 | } |
| 1359 | |
| 1360 | if (readonly) |
| 1361 | flags |= BDRV_O_RDONLY; |
| 1362 | else |
| 1363 | flags |= BDRV_O_RDWR; |
| 1364 | |
| 1365 | if (optind != argc - 1) |
| 1366 | return command_usage(&open_cmd); |
| 1367 | |
Christoph Hellwig | 9c4bab2 | 2009-07-10 13:33:47 +0200 | [diff] [blame] | 1368 | return openfile(argv[optind], flags, growable); |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1369 | } |
| 1370 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1371 | static int |
| 1372 | init_args_command( |
| 1373 | int index) |
| 1374 | { |
| 1375 | /* only one device allowed so far */ |
| 1376 | if (index >= 1) |
| 1377 | return 0; |
| 1378 | return ++index; |
| 1379 | } |
| 1380 | |
| 1381 | static int |
| 1382 | init_check_command( |
| 1383 | const cmdinfo_t *ct) |
| 1384 | { |
| 1385 | if (ct->flags & CMD_FLAG_GLOBAL) |
| 1386 | return 1; |
| 1387 | if (!(ct->flags & CMD_NOFILE_OK) && !bs) { |
| 1388 | fprintf(stderr, "no file open, try 'help open'\n"); |
| 1389 | return 0; |
| 1390 | } |
| 1391 | return 1; |
| 1392 | } |
| 1393 | |
| 1394 | static void usage(const char *name) |
| 1395 | { |
| 1396 | printf( |
| 1397 | "Usage: %s [-h] [-V] [-Crsnm] [-c cmd] ... [file]\n" |
Stefan Weil | 84844a2 | 2009-06-22 15:08:47 +0200 | [diff] [blame] | 1398 | "QEMU Disk exerciser\n" |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1399 | "\n" |
| 1400 | " -C, --create create new file if it doesn't exist\n" |
| 1401 | " -c, --cmd command to execute\n" |
| 1402 | " -r, --read-only export read-only\n" |
| 1403 | " -s, --snapshot use snapshot file\n" |
| 1404 | " -n, --nocache disable host cache\n" |
Christoph Hellwig | 1db6947 | 2009-07-15 23:11:21 +0200 | [diff] [blame] | 1405 | " -g, --growable allow file to grow (only applies to protocols)\n" |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1406 | " -m, --misalign misalign allocations for O_DIRECT\n" |
Christoph Hellwig | 5c6c3a6 | 2009-08-20 16:58:35 +0200 | [diff] [blame] | 1407 | " -k, --native-aio use kernel AIO implementation (on Linux only)\n" |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1408 | " -h, --help display this help and exit\n" |
| 1409 | " -V, --version output version information and exit\n" |
| 1410 | "\n", |
| 1411 | name); |
| 1412 | } |
| 1413 | |
| 1414 | |
| 1415 | int main(int argc, char **argv) |
| 1416 | { |
| 1417 | int readonly = 0; |
Christoph Hellwig | 9c4bab2 | 2009-07-10 13:33:47 +0200 | [diff] [blame] | 1418 | int growable = 0; |
Christoph Hellwig | 5c6c3a6 | 2009-08-20 16:58:35 +0200 | [diff] [blame] | 1419 | const char *sopt = "hVc:Crsnmgk"; |
Blue Swirl | b32bb95 | 2009-12-13 15:41:43 +0000 | [diff] [blame] | 1420 | const struct option lopt[] = { |
Blue Swirl | 660f11b | 2009-07-31 21:16:51 +0000 | [diff] [blame] | 1421 | { "help", 0, NULL, 'h' }, |
| 1422 | { "version", 0, NULL, 'V' }, |
| 1423 | { "offset", 1, NULL, 'o' }, |
| 1424 | { "cmd", 1, NULL, 'c' }, |
| 1425 | { "create", 0, NULL, 'C' }, |
| 1426 | { "read-only", 0, NULL, 'r' }, |
| 1427 | { "snapshot", 0, NULL, 's' }, |
| 1428 | { "nocache", 0, NULL, 'n' }, |
| 1429 | { "misalign", 0, NULL, 'm' }, |
| 1430 | { "growable", 0, NULL, 'g' }, |
Christoph Hellwig | 5c6c3a6 | 2009-08-20 16:58:35 +0200 | [diff] [blame] | 1431 | { "native-aio", 0, NULL, 'k' }, |
Blue Swirl | 660f11b | 2009-07-31 21:16:51 +0000 | [diff] [blame] | 1432 | { NULL, 0, NULL, 0 } |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1433 | }; |
| 1434 | int c; |
| 1435 | int opt_index = 0; |
| 1436 | int flags = 0; |
| 1437 | |
| 1438 | progname = basename(argv[0]); |
| 1439 | |
| 1440 | while ((c = getopt_long(argc, argv, sopt, lopt, &opt_index)) != -1) { |
| 1441 | switch (c) { |
| 1442 | case 's': |
| 1443 | flags |= BDRV_O_SNAPSHOT; |
| 1444 | break; |
| 1445 | case 'n': |
| 1446 | flags |= BDRV_O_NOCACHE; |
| 1447 | break; |
| 1448 | case 'c': |
| 1449 | add_user_command(optarg); |
| 1450 | break; |
| 1451 | case 'C': |
| 1452 | flags |= BDRV_O_CREAT; |
| 1453 | break; |
| 1454 | case 'r': |
| 1455 | readonly = 1; |
| 1456 | break; |
| 1457 | case 'm': |
| 1458 | misalign = 1; |
| 1459 | break; |
Christoph Hellwig | 9c4bab2 | 2009-07-10 13:33:47 +0200 | [diff] [blame] | 1460 | case 'g': |
| 1461 | growable = 1; |
| 1462 | break; |
Christoph Hellwig | 5c6c3a6 | 2009-08-20 16:58:35 +0200 | [diff] [blame] | 1463 | case 'k': |
| 1464 | flags |= BDRV_O_NATIVE_AIO; |
| 1465 | break; |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1466 | case 'V': |
| 1467 | printf("%s version %s\n", progname, VERSION); |
| 1468 | exit(0); |
| 1469 | case 'h': |
| 1470 | usage(progname); |
| 1471 | exit(0); |
| 1472 | default: |
| 1473 | usage(progname); |
| 1474 | exit(1); |
| 1475 | } |
| 1476 | } |
| 1477 | |
| 1478 | if ((argc - optind) > 1) { |
| 1479 | usage(progname); |
| 1480 | exit(1); |
| 1481 | } |
| 1482 | |
| 1483 | bdrv_init(); |
| 1484 | |
| 1485 | /* initialize commands */ |
| 1486 | quit_init(); |
| 1487 | help_init(); |
| 1488 | add_command(&open_cmd); |
| 1489 | add_command(&close_cmd); |
| 1490 | add_command(&read_cmd); |
| 1491 | add_command(&readv_cmd); |
| 1492 | add_command(&write_cmd); |
| 1493 | add_command(&writev_cmd); |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1494 | add_command(&aio_read_cmd); |
| 1495 | add_command(&aio_write_cmd); |
| 1496 | add_command(&aio_flush_cmd); |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1497 | add_command(&flush_cmd); |
| 1498 | add_command(&truncate_cmd); |
| 1499 | add_command(&length_cmd); |
| 1500 | add_command(&info_cmd); |
| 1501 | add_command(&alloc_cmd); |
| 1502 | |
| 1503 | add_args_command(init_args_command); |
| 1504 | add_check_command(init_check_command); |
| 1505 | |
| 1506 | /* open the device */ |
| 1507 | if (readonly) |
| 1508 | flags |= BDRV_O_RDONLY; |
| 1509 | else |
| 1510 | flags |= BDRV_O_RDWR; |
| 1511 | |
| 1512 | if ((argc - optind) == 1) |
Christoph Hellwig | 9c4bab2 | 2009-07-10 13:33:47 +0200 | [diff] [blame] | 1513 | openfile(argv[optind], flags, growable); |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1514 | command_loop(); |
| 1515 | |
Christoph Hellwig | 95533d5 | 2009-06-20 21:10:44 +0200 | [diff] [blame] | 1516 | /* |
| 1517 | * Make sure all outstanding requests get flushed the program exits. |
| 1518 | */ |
| 1519 | qemu_aio_flush(); |
| 1520 | |
aliguori | e3aff4f | 2009-04-05 19:14:04 +0000 | [diff] [blame] | 1521 | if (bs) |
| 1522 | bdrv_close(bs); |
| 1523 | return 0; |
| 1524 | } |