Statistics
| Branch: | Revision:

root / qemu-io.c @ a6599793

History | View | Annotate | Download (37.7 kB)

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
 */
10
#include <sys/time.h>
11
#include <sys/types.h>
12
#include <stdarg.h>
13
#include <stdio.h>
14
#include <getopt.h>
15
#include <libgen.h>
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
/*
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
/*
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_blockalign(bs, 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
79
dump_buffer(const void *buffer, int64_t offset, int len)
80
{
81
        int i, j;
82
        const uint8_t *p;
83

    
84
        for (i = 0, p = buffer; i < len; i += 16) {
85
                const uint8_t *s = p;
86

    
87
                printf("%08" PRIx64 ":  ", 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++) {
92
                        if (isalnum(*s))
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 %" PRId64 "\n",
112
                       op, total, count, 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

    
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;
132
        void *buf = NULL;
133
        void *p;
134
        int i;
135

    
136
        for (i = 0; i < nr_iov; i++) {
137
                char *arg = argv[i];
138
                int64_t len;
139

    
140
                len = cvtnum(arg);
141
                if (len < 0) {
142
                        printf("non-numeric length argument -- %s\n", arg);
143
                        goto fail;
144
                }
145

    
146
                /* should be SIZE_T_MAX, but that doesn't exist */
147
                if (len > INT_MAX) {
148
                        printf("too large length argument -- %s\n", arg);
149
                        goto fail;
150
                }
151

    
152
                if (len & 0x1ff) {
153
                        printf("length argument %" PRId64
154
                               " is not sector aligned\n", len);
155
                        goto fail;
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

    
171
fail:
172
        free(sizes);
173
        return buf;
174
}
175

    
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

    
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

    
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

    
266
        *total = qiov->size;
267
        return async_ret < 0 ? async_ret : 1;
268
}
269

    
270
struct multiwrite_async_ret {
271
        int num_done;
272
        int error;
273
};
274

    
275
static void multiwrite_cb(void *opaque, int ret)
276
{
277
        struct multiwrite_async_ret *async_ret = opaque;
278

    
279
        async_ret->num_done++;
280
        if (ret < 0) {
281
                async_ret->error = ret;
282
        }
283
}
284

    
285
static int do_aio_multiwrite(BlockRequest* reqs, int num_reqs, int *total)
286
{
287
        int i, ret;
288
        struct multiwrite_async_ret async_ret = {
289
                .num_done = 0,
290
                .error = 0,
291
        };
292

    
293
        *total = 0;
294
        for (i = 0; i < num_reqs; i++) {
295
                reqs[i].cb = multiwrite_cb;
296
                reqs[i].opaque = &async_ret;
297
                *total += reqs[i].qiov->size;
298
        }
299

    
300
        ret = bdrv_aio_multiwrite(bs, reqs, num_reqs);
301
        if (ret < 0) {
302
                return ret;
303
        }
304

    
305
        while (async_ret.num_done < num_reqs) {
306
                qemu_aio_wait();
307
        }
308

    
309
        return async_ret.error < 0 ? async_ret.error : 1;
310
}
311

    
312
static void
313
read_help(void)
314
{
315
        printf(
316
"\n"
317
" reads a range of bytes from the given offset\n"
318
"\n"
319
" Example:\n"
320
" 'read -v 512 1k' - dumps 1 kilobyte read from 512 bytes into the file\n"
321
"\n"
322
" Reads a segment of the currently open file, optionally dumping it to the\n"
323
" standard output stream (with -v option) for subsequent inspection.\n"
324
" -b, -- read from the VM state rather than the virtual disk\n"
325
" -C, -- report statistics in a machine parsable format\n"
326
" -l, -- length for pattern verification (only with -P)\n"
327
" -p, -- use bdrv_pread to read the file\n"
328
" -P, -- use a pattern to verify read data\n"
329
" -q, -- quiet mode, do not show I/O statistics\n"
330
" -s, -- start offset for pattern verification (only with -P)\n"
331
" -v, -- dump buffer to standard output\n"
332
"\n");
333
}
334

    
335
static int read_f(int argc, char **argv);
336

    
337
static const cmdinfo_t read_cmd = {
338
        .name                = "read",
339
        .altname        = "r",
340
        .cfunc                = read_f,
341
        .argmin                = 2,
342
        .argmax                = -1,
343
        .args                = "[-abCpqv] [-P pattern [-s off] [-l len]] off len",
344
        .oneline        = "reads a number of bytes at a specified offset",
345
        .help                = read_help,
346
};
347

    
348
static int
349
read_f(int argc, char **argv)
350
{
351
        struct timeval t1, t2;
352
        int Cflag = 0, pflag = 0, qflag = 0, vflag = 0;
353
        int Pflag = 0, sflag = 0, lflag = 0, bflag = 0;
354
        int c, cnt;
355
        char *buf;
356
        int64_t offset;
357
        int count;
358
        /* Some compilers get confused and warn if this is not initialized.  */
359
        int total = 0;
360
        int pattern = 0, pattern_offset = 0, pattern_count = 0;
361

    
362
        while ((c = getopt(argc, argv, "bCl:pP:qs:v")) != EOF) {
363
                switch (c) {
364
                case 'b':
365
                        bflag = 1;
366
                        break;
367
                case 'C':
368
                        Cflag = 1;
369
                        break;
370
                case 'l':
371
                        lflag = 1;
372
                        pattern_count = cvtnum(optarg);
373
                        if (pattern_count < 0) {
374
                                printf("non-numeric length argument -- %s\n", optarg);
375
                                return 0;
376
                        }
377
                        break;
378
                case 'p':
379
                        pflag = 1;
380
                        break;
381
                case 'P':
382
                        Pflag = 1;
383
                        pattern = parse_pattern(optarg);
384
                        if (pattern < 0)
385
                                return 0;
386
                        break;
387
                case 'q':
388
                        qflag = 1;
389
                        break;
390
                case 's':
391
                        sflag = 1;
392
                        pattern_offset = cvtnum(optarg);
393
                        if (pattern_offset < 0) {
394
                                printf("non-numeric length argument -- %s\n", optarg);
395
                                return 0;
396
                        }
397
                        break;
398
                case 'v':
399
                        vflag = 1;
400
                        break;
401
                default:
402
                        return command_usage(&read_cmd);
403
                }
404
        }
405

    
406
        if (optind != argc - 2)
407
                return command_usage(&read_cmd);
408

    
409
        if (bflag && pflag) {
410
                printf("-b and -p cannot be specified at the same time\n");
411
                return 0;
412
        }
413

    
414
        offset = cvtnum(argv[optind]);
415
        if (offset < 0) {
416
                printf("non-numeric length argument -- %s\n", argv[optind]);
417
                return 0;
418
        }
419

    
420
        optind++;
421
        count = cvtnum(argv[optind]);
422
        if (count < 0) {
423
                printf("non-numeric length argument -- %s\n", argv[optind]);
424
                return 0;
425
        }
426

    
427
    if (!Pflag && (lflag || sflag)) {
428
        return command_usage(&read_cmd);
429
    }
430

    
431
    if (!lflag) {
432
        pattern_count = count - pattern_offset;
433
    }
434

    
435
    if ((pattern_count < 0) || (pattern_count + pattern_offset > count))  {
436
        printf("pattern verfication range exceeds end of read data\n");
437
        return 0;
438
    }
439

    
440
        if (!pflag)
441
                if (offset & 0x1ff) {
442
                        printf("offset %" PRId64 " is not sector aligned\n",
443
                               offset);
444
                        return 0;
445

    
446
                if (count & 0x1ff) {
447
                        printf("count %d is not sector aligned\n",
448
                                count);
449
                        return 0;
450
                }
451
        }
452

    
453
        buf = qemu_io_alloc(count, 0xab);
454

    
455
        gettimeofday(&t1, NULL);
456
        if (pflag)
457
                cnt = do_pread(buf, offset, count, &total);
458
        else if (bflag)
459
                cnt = do_load_vmstate(buf, offset, count, &total);
460
        else
461
                cnt = do_read(buf, offset, count, &total);
462
        gettimeofday(&t2, NULL);
463

    
464
        if (cnt < 0) {
465
                printf("read failed: %s\n", strerror(-cnt));
466
                goto out;
467
        }
468

    
469
        if (Pflag) {
470
                void* cmp_buf = malloc(pattern_count);
471
                memset(cmp_buf, pattern, pattern_count);
472
                if (memcmp(buf + pattern_offset, cmp_buf, pattern_count)) {
473
                        printf("Pattern verification failed at offset %"
474
                               PRId64 ", %d bytes\n",
475
                               offset + pattern_offset, pattern_count);
476
                }
477
                free(cmp_buf);
478
        }
479

    
480
        if (qflag)
481
                goto out;
482

    
483
        if (vflag)
484
                dump_buffer(buf, offset, count);
485

    
486
        /* Finally, report back -- -C gives a parsable format */
487
        t2 = tsub(t2, t1);
488
        print_report("read", &t2, offset, count, total, cnt, Cflag);
489

    
490
out:
491
        qemu_io_free(buf);
492

    
493
        return 0;
494
}
495

    
496
static void
497
readv_help(void)
498
{
499
        printf(
500
"\n"
501
" reads a range of bytes from the given offset into multiple buffers\n"
502
"\n"
503
" Example:\n"
504
" 'readv -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n"
505
"\n"
506
" Reads a segment of the currently open file, optionally dumping it to the\n"
507
" standard output stream (with -v option) for subsequent inspection.\n"
508
" Uses multiple iovec buffers if more than one byte range is specified.\n"
509
" -C, -- report statistics in a machine parsable format\n"
510
" -P, -- use a pattern to verify read data\n"
511
" -v, -- dump buffer to standard output\n"
512
" -q, -- quiet mode, do not show I/O statistics\n"
513
"\n");
514
}
515

    
516
static int readv_f(int argc, char **argv);
517

    
518
static const cmdinfo_t readv_cmd = {
519
        .name                = "readv",
520
        .cfunc                = readv_f,
521
        .argmin                = 2,
522
        .argmax                = -1,
523
        .args                = "[-Cqv] [-P pattern ] off len [len..]",
524
        .oneline        = "reads a number of bytes at a specified offset",
525
        .help                = readv_help,
526
};
527

    
528
static int
529
readv_f(int argc, char **argv)
530
{
531
        struct timeval t1, t2;
532
        int Cflag = 0, qflag = 0, vflag = 0;
533
        int c, cnt;
534
        char *buf;
535
        int64_t offset;
536
        /* Some compilers get confused and warn if this is not initialized.  */
537
        int total = 0;
538
        int nr_iov;
539
        QEMUIOVector qiov;
540
        int pattern = 0;
541
        int Pflag = 0;
542

    
543
        while ((c = getopt(argc, argv, "CP:qv")) != EOF) {
544
                switch (c) {
545
                case 'C':
546
                        Cflag = 1;
547
                        break;
548
                case 'P':
549
                        Pflag = 1;
550
                        pattern = parse_pattern(optarg);
551
                        if (pattern < 0)
552
                                return 0;
553
                        break;
554
                case 'q':
555
                        qflag = 1;
556
                        break;
557
                case 'v':
558
                        vflag = 1;
559
                        break;
560
                default:
561
                        return command_usage(&readv_cmd);
562
                }
563
        }
564

    
565
        if (optind > argc - 2)
566
                return command_usage(&readv_cmd);
567

    
568

    
569
        offset = cvtnum(argv[optind]);
570
        if (offset < 0) {
571
                printf("non-numeric length argument -- %s\n", argv[optind]);
572
                return 0;
573
        }
574
        optind++;
575

    
576
        if (offset & 0x1ff) {
577
                printf("offset %" PRId64 " is not sector aligned\n",
578
                       offset);
579
                return 0;
580
        }
581

    
582
        nr_iov = argc - optind;
583
        buf = create_iovec(&qiov, &argv[optind], nr_iov, 0xab);
584

    
585
        gettimeofday(&t1, NULL);
586
        cnt = do_aio_readv(&qiov, offset, &total);
587
        gettimeofday(&t2, NULL);
588

    
589
        if (cnt < 0) {
590
                printf("readv failed: %s\n", strerror(-cnt));
591
                goto out;
592
        }
593

    
594
        if (Pflag) {
595
                void* cmp_buf = malloc(qiov.size);
596
                memset(cmp_buf, pattern, qiov.size);
597
                if (memcmp(buf, cmp_buf, qiov.size)) {
598
                        printf("Pattern verification failed at offset %"
599
                               PRId64 ", %zd bytes\n",
600
                               offset, qiov.size);
601
                }
602
                free(cmp_buf);
603
        }
604

    
605
        if (qflag)
606
                goto out;
607

    
608
        if (vflag)
609
                dump_buffer(buf, offset, qiov.size);
610

    
611
        /* Finally, report back -- -C gives a parsable format */
612
        t2 = tsub(t2, t1);
613
        print_report("read", &t2, offset, qiov.size, total, cnt, Cflag);
614

    
615
out:
616
        qemu_io_free(buf);
617
        return 0;
618
}
619

    
620
static void
621
write_help(void)
622
{
623
        printf(
624
"\n"
625
" writes a range of bytes from the given offset\n"
626
"\n"
627
" Example:\n"
628
" 'write 512 1k' - writes 1 kilobyte at 512 bytes into the open file\n"
629
"\n"
630
" Writes into a segment of the currently open file, using a buffer\n"
631
" filled with a set pattern (0xcdcdcdcd).\n"
632
" -b, -- write to the VM state rather than the virtual disk\n"
633
" -p, -- use bdrv_pwrite to write the file\n"
634
" -P, -- use different pattern to fill file\n"
635
" -C, -- report statistics in a machine parsable format\n"
636
" -q, -- quiet mode, do not show I/O statistics\n"
637
"\n");
638
}
639

    
640
static int write_f(int argc, char **argv);
641

    
642
static const cmdinfo_t write_cmd = {
643
        .name                = "write",
644
        .altname        = "w",
645
        .cfunc                = write_f,
646
        .argmin                = 2,
647
        .argmax                = -1,
648
        .args                = "[-abCpq] [-P pattern ] off len",
649
        .oneline        = "writes a number of bytes at a specified offset",
650
        .help                = write_help,
651
};
652

    
653
static int
654
write_f(int argc, char **argv)
655
{
656
        struct timeval t1, t2;
657
        int Cflag = 0, pflag = 0, qflag = 0, bflag = 0;
658
        int c, cnt;
659
        char *buf;
660
        int64_t offset;
661
        int count;
662
        /* Some compilers get confused and warn if this is not initialized.  */
663
        int total = 0;
664
        int pattern = 0xcd;
665

    
666
        while ((c = getopt(argc, argv, "bCpP:q")) != EOF) {
667
                switch (c) {
668
                case 'b':
669
                        bflag = 1;
670
                        break;
671
                case 'C':
672
                        Cflag = 1;
673
                        break;
674
                case 'p':
675
                        pflag = 1;
676
                        break;
677
                case 'P':
678
                        pattern = parse_pattern(optarg);
679
                        if (pattern < 0)
680
                                return 0;
681
                        break;
682
                case 'q':
683
                        qflag = 1;
684
                        break;
685
                default:
686
                        return command_usage(&write_cmd);
687
                }
688
        }
689

    
690
        if (optind != argc - 2)
691
                return command_usage(&write_cmd);
692

    
693
        if (bflag && pflag) {
694
                printf("-b and -p cannot be specified at the same time\n");
695
                return 0;
696
        }
697

    
698
        offset = cvtnum(argv[optind]);
699
        if (offset < 0) {
700
                printf("non-numeric length argument -- %s\n", argv[optind]);
701
                return 0;
702
        }
703

    
704
        optind++;
705
        count = cvtnum(argv[optind]);
706
        if (count < 0) {
707
                printf("non-numeric length argument -- %s\n", argv[optind]);
708
                return 0;
709
        }
710

    
711
        if (!pflag) {
712
                if (offset & 0x1ff) {
713
                        printf("offset %" PRId64 " is not sector aligned\n",
714
                               offset);
715
                        return 0;
716
                }
717

    
718
                if (count & 0x1ff) {
719
                        printf("count %d is not sector aligned\n",
720
                                count);
721
                        return 0;
722
                }
723
        }
724

    
725
        buf = qemu_io_alloc(count, pattern);
726

    
727
        gettimeofday(&t1, NULL);
728
        if (pflag)
729
                cnt = do_pwrite(buf, offset, count, &total);
730
        else if (bflag)
731
                cnt = do_save_vmstate(buf, offset, count, &total);
732
        else
733
                cnt = do_write(buf, offset, count, &total);
734
        gettimeofday(&t2, NULL);
735

    
736
        if (cnt < 0) {
737
                printf("write failed: %s\n", strerror(-cnt));
738
                goto out;
739
        }
740

    
741
        if (qflag)
742
                goto out;
743

    
744
        /* Finally, report back -- -C gives a parsable format */
745
        t2 = tsub(t2, t1);
746
        print_report("wrote", &t2, offset, count, total, cnt, Cflag);
747

    
748
out:
749
        qemu_io_free(buf);
750

    
751
        return 0;
752
}
753

    
754
static void
755
writev_help(void)
756
{
757
        printf(
758
"\n"
759
" writes a range of bytes from the given offset source from multiple buffers\n"
760
"\n"
761
" Example:\n"
762
" 'write 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n"
763
"\n"
764
" Writes into a segment of the currently open file, using a buffer\n"
765
" filled with a set pattern (0xcdcdcdcd).\n"
766
" -P, -- use different pattern to fill file\n"
767
" -C, -- report statistics in a machine parsable format\n"
768
" -q, -- quiet mode, do not show I/O statistics\n"
769
"\n");
770
}
771

    
772
static int writev_f(int argc, char **argv);
773

    
774
static const cmdinfo_t writev_cmd = {
775
        .name                = "writev",
776
        .cfunc                = writev_f,
777
        .argmin                = 2,
778
        .argmax                = -1,
779
        .args                = "[-Cq] [-P pattern ] off len [len..]",
780
        .oneline        = "writes a number of bytes at a specified offset",
781
        .help                = writev_help,
782
};
783

    
784
static int
785
writev_f(int argc, char **argv)
786
{
787
        struct timeval t1, t2;
788
        int Cflag = 0, qflag = 0;
789
        int c, cnt;
790
        char *buf;
791
        int64_t offset;
792
        /* Some compilers get confused and warn if this is not initialized.  */
793
        int total = 0;
794
        int nr_iov;
795
        int pattern = 0xcd;
796
        QEMUIOVector qiov;
797

    
798
        while ((c = getopt(argc, argv, "CqP:")) != EOF) {
799
                switch (c) {
800
                case 'C':
801
                        Cflag = 1;
802
                        break;
803
                case 'q':
804
                        qflag = 1;
805
                        break;
806
                case 'P':
807
                        pattern = parse_pattern(optarg);
808
                        if (pattern < 0)
809
                                return 0;
810
                        break;
811
                default:
812
                        return command_usage(&writev_cmd);
813
                }
814
        }
815

    
816
        if (optind > argc - 2)
817
                return command_usage(&writev_cmd);
818

    
819
        offset = cvtnum(argv[optind]);
820
        if (offset < 0) {
821
                printf("non-numeric length argument -- %s\n", argv[optind]);
822
                return 0;
823
        }
824
        optind++;
825

    
826
        if (offset & 0x1ff) {
827
                printf("offset %" PRId64 " is not sector aligned\n",
828
                       offset);
829
                return 0;
830
        }
831

    
832
        nr_iov = argc - optind;
833
        buf = create_iovec(&qiov, &argv[optind], nr_iov, pattern);
834

    
835
        gettimeofday(&t1, NULL);
836
        cnt = do_aio_writev(&qiov, offset, &total);
837
        gettimeofday(&t2, NULL);
838

    
839
        if (cnt < 0) {
840
                printf("writev failed: %s\n", strerror(-cnt));
841
                goto out;
842
        }
843

    
844
        if (qflag)
845
                goto out;
846

    
847
        /* Finally, report back -- -C gives a parsable format */
848
        t2 = tsub(t2, t1);
849
        print_report("wrote", &t2, offset, qiov.size, total, cnt, Cflag);
850
out:
851
        qemu_io_free(buf);
852
        return 0;
853
}
854

    
855
static void
856
multiwrite_help(void)
857
{
858
        printf(
859
"\n"
860
" writes a range of bytes from the given offset source from multiple buffers,\n"
861
" in a batch of requests that may be merged by qemu\n"
862
"\n"
863
" Example:\n"
864
" 'multiwrite 512 1k 1k ; 4k 1k' \n"
865
"  writes 2 kB at 512 bytes and 1 kB at 4 kB into the open file\n"
866
"\n"
867
" Writes into a segment of the currently open file, using a buffer\n"
868
" filled with a set pattern (0xcdcdcdcd). The pattern byte is increased\n"
869
" by one for each request contained in the multiwrite command.\n"
870
" -P, -- use different pattern to fill file\n"
871
" -C, -- report statistics in a machine parsable format\n"
872
" -q, -- quiet mode, do not show I/O statistics\n"
873
"\n");
874
}
875

    
876
static int multiwrite_f(int argc, char **argv);
877

    
878
static const cmdinfo_t multiwrite_cmd = {
879
        .name                = "multiwrite",
880
        .cfunc                = multiwrite_f,
881
        .argmin                = 2,
882
        .argmax                = -1,
883
        .args                = "[-Cq] [-P pattern ] off len [len..] [; off len [len..]..]",
884
        .oneline        = "issues multiple write requests at once",
885
        .help                = multiwrite_help,
886
};
887

    
888
static int
889
multiwrite_f(int argc, char **argv)
890
{
891
        struct timeval t1, t2;
892
        int Cflag = 0, qflag = 0;
893
        int c, cnt;
894
        char **buf;
895
        int64_t offset, first_offset = 0;
896
        /* Some compilers get confused and warn if this is not initialized.  */
897
        int total = 0;
898
        int nr_iov;
899
        int nr_reqs;
900
        int pattern = 0xcd;
901
        QEMUIOVector *qiovs;
902
        int i;
903
        BlockRequest *reqs;
904

    
905
        while ((c = getopt(argc, argv, "CqP:")) != EOF) {
906
                switch (c) {
907
                case 'C':
908
                        Cflag = 1;
909
                        break;
910
                case 'q':
911
                        qflag = 1;
912
                        break;
913
                case 'P':
914
                        pattern = parse_pattern(optarg);
915
                        if (pattern < 0)
916
                                return 0;
917
                        break;
918
                default:
919
                        return command_usage(&writev_cmd);
920
                }
921
        }
922

    
923
        if (optind > argc - 2)
924
                return command_usage(&writev_cmd);
925

    
926
        nr_reqs = 1;
927
        for (i = optind; i < argc; i++) {
928
                if (!strcmp(argv[i], ";")) {
929
                        nr_reqs++;
930
                }
931
        }
932

    
933
        reqs = qemu_malloc(nr_reqs * sizeof(*reqs));
934
        buf = qemu_malloc(nr_reqs * sizeof(*buf));
935
        qiovs = qemu_malloc(nr_reqs * sizeof(*qiovs));
936

    
937
        for (i = 0; i < nr_reqs; i++) {
938
                int j;
939

    
940
                /* Read the offset of the request */
941
                offset = cvtnum(argv[optind]);
942
                if (offset < 0) {
943
                        printf("non-numeric offset argument -- %s\n", argv[optind]);
944
                        return 0;
945
                }
946
                optind++;
947

    
948
                if (offset & 0x1ff) {
949
                        printf("offset %lld is not sector aligned\n",
950
                                (long long)offset);
951
                        return 0;
952
                }
953

    
954
        if (i == 0) {
955
            first_offset = offset;
956
        }
957

    
958
                /* Read lengths for qiov entries */
959
                for (j = optind; j < argc; j++) {
960
                        if (!strcmp(argv[j], ";")) {
961
                                break;
962
                        }
963
                }
964

    
965
                nr_iov = j - optind;
966

    
967
                /* Build request */
968
                reqs[i].qiov = &qiovs[i];
969
                buf[i] = create_iovec(reqs[i].qiov, &argv[optind], nr_iov, pattern);
970
                reqs[i].sector = offset >> 9;
971
                reqs[i].nb_sectors = reqs[i].qiov->size >> 9;
972

    
973
                optind = j + 1;
974

    
975
                offset += reqs[i].qiov->size;
976
                pattern++;
977
        }
978

    
979
        gettimeofday(&t1, NULL);
980
        cnt = do_aio_multiwrite(reqs, nr_reqs, &total);
981
        gettimeofday(&t2, NULL);
982

    
983
        if (cnt < 0) {
984
                printf("aio_multiwrite failed: %s\n", strerror(-cnt));
985
                goto out;
986
        }
987

    
988
        if (qflag)
989
                goto out;
990

    
991
        /* Finally, report back -- -C gives a parsable format */
992
        t2 = tsub(t2, t1);
993
        print_report("wrote", &t2, first_offset, total, total, cnt, Cflag);
994
out:
995
        for (i = 0; i < nr_reqs; i++) {
996
                qemu_io_free(buf[i]);
997
                qemu_iovec_destroy(&qiovs[i]);
998
        }
999
        qemu_free(buf);
1000
        qemu_free(reqs);
1001
        qemu_free(qiovs);
1002
        return 0;
1003
}
1004

    
1005
struct aio_ctx {
1006
        QEMUIOVector qiov;
1007
        int64_t offset;
1008
        char *buf;
1009
        int qflag;
1010
        int vflag;
1011
        int Cflag;
1012
        int Pflag;
1013
        int pattern;
1014
        struct timeval t1;
1015
};
1016

    
1017
static void
1018
aio_write_done(void *opaque, int ret)
1019
{
1020
        struct aio_ctx *ctx = opaque;
1021
        struct timeval t2;
1022

    
1023
        gettimeofday(&t2, NULL);
1024

    
1025

    
1026
        if (ret < 0) {
1027
                printf("aio_write failed: %s\n", strerror(-ret));
1028
                goto out;
1029
        }
1030

    
1031
        if (ctx->qflag) {
1032
                goto out;
1033
        }
1034

    
1035
        /* Finally, report back -- -C gives a parsable format */
1036
        t2 = tsub(t2, ctx->t1);
1037
        print_report("wrote", &t2, ctx->offset, ctx->qiov.size,
1038
                     ctx->qiov.size, 1, ctx->Cflag);
1039
out:
1040
        qemu_io_free(ctx->buf);
1041
        free(ctx);
1042
}
1043

    
1044
static void
1045
aio_read_done(void *opaque, int ret)
1046
{
1047
        struct aio_ctx *ctx = opaque;
1048
        struct timeval t2;
1049

    
1050
        gettimeofday(&t2, NULL);
1051

    
1052
        if (ret < 0) {
1053
                printf("readv failed: %s\n", strerror(-ret));
1054
                goto out;
1055
        }
1056

    
1057
        if (ctx->Pflag) {
1058
                void *cmp_buf = malloc(ctx->qiov.size);
1059

    
1060
                memset(cmp_buf, ctx->pattern, ctx->qiov.size);
1061
                if (memcmp(ctx->buf, cmp_buf, ctx->qiov.size)) {
1062
                        printf("Pattern verification failed at offset %"
1063
                               PRId64 ", %zd bytes\n",
1064
                               ctx->offset, ctx->qiov.size);
1065
                }
1066
                free(cmp_buf);
1067
        }
1068

    
1069
        if (ctx->qflag) {
1070
                goto out;
1071
        }
1072

    
1073
        if (ctx->vflag) {
1074
                dump_buffer(ctx->buf, ctx->offset, ctx->qiov.size);
1075
        }
1076

    
1077
        /* Finally, report back -- -C gives a parsable format */
1078
        t2 = tsub(t2, ctx->t1);
1079
        print_report("read", &t2, ctx->offset, ctx->qiov.size,
1080
                     ctx->qiov.size, 1, ctx->Cflag);
1081
out:
1082
        qemu_io_free(ctx->buf);
1083
        free(ctx);
1084
}
1085

    
1086
static void
1087
aio_read_help(void)
1088
{
1089
        printf(
1090
"\n"
1091
" asynchronously reads a range of bytes from the given offset\n"
1092
"\n"
1093
" Example:\n"
1094
" 'aio_read -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n"
1095
"\n"
1096
" Reads a segment of the currently open file, optionally dumping it to the\n"
1097
" standard output stream (with -v option) for subsequent inspection.\n"
1098
" The read is performed asynchronously and the aio_flush command must be\n"
1099
" used to ensure all outstanding aio requests have been completed\n"
1100
" -C, -- report statistics in a machine parsable format\n"
1101
" -P, -- use a pattern to verify read data\n"
1102
" -v, -- dump buffer to standard output\n"
1103
" -q, -- quiet mode, do not show I/O statistics\n"
1104
"\n");
1105
}
1106

    
1107
static int aio_read_f(int argc, char **argv);
1108

    
1109
static const cmdinfo_t aio_read_cmd = {
1110
        .name                = "aio_read",
1111
        .cfunc                = aio_read_f,
1112
        .argmin                = 2,
1113
        .argmax                = -1,
1114
        .args                = "[-Cqv] [-P pattern ] off len [len..]",
1115
        .oneline        = "asynchronously reads a number of bytes",
1116
        .help                = aio_read_help,
1117
};
1118

    
1119
static int
1120
aio_read_f(int argc, char **argv)
1121
{
1122
        int nr_iov, c;
1123
        struct aio_ctx *ctx = calloc(1, sizeof(struct aio_ctx));
1124
        BlockDriverAIOCB *acb;
1125

    
1126
        while ((c = getopt(argc, argv, "CP:qv")) != EOF) {
1127
                switch (c) {
1128
                case 'C':
1129
                        ctx->Cflag = 1;
1130
                        break;
1131
                case 'P':
1132
                        ctx->Pflag = 1;
1133
                        ctx->pattern = parse_pattern(optarg);
1134
                        if (ctx->pattern < 0) {
1135
                                free(ctx);
1136
                                return 0;
1137
                        }
1138
                        break;
1139
                case 'q':
1140
                        ctx->qflag = 1;
1141
                        break;
1142
                case 'v':
1143
                        ctx->vflag = 1;
1144
                        break;
1145
                default:
1146
                        free(ctx);
1147
                        return command_usage(&aio_read_cmd);
1148
                }
1149
        }
1150

    
1151
        if (optind > argc - 2) {
1152
                free(ctx);
1153
                return command_usage(&aio_read_cmd);
1154
        }
1155

    
1156
        ctx->offset = cvtnum(argv[optind]);
1157
        if (ctx->offset < 0) {
1158
                printf("non-numeric length argument -- %s\n", argv[optind]);
1159
                free(ctx);
1160
                return 0;
1161
        }
1162
        optind++;
1163

    
1164
        if (ctx->offset & 0x1ff) {
1165
                printf("offset %" PRId64 " is not sector aligned\n",
1166
                       ctx->offset);
1167
                free(ctx);
1168
                return 0;
1169
        }
1170

    
1171
        nr_iov = argc - optind;
1172
        ctx->buf = create_iovec(&ctx->qiov, &argv[optind], nr_iov, 0xab);
1173

    
1174
        gettimeofday(&ctx->t1, NULL);
1175
        acb = bdrv_aio_readv(bs, ctx->offset >> 9, &ctx->qiov,
1176
                              ctx->qiov.size >> 9, aio_read_done, ctx);
1177
        if (!acb) {
1178
                free(ctx->buf);
1179
                free(ctx);
1180
                return -EIO;
1181
        }
1182

    
1183
        return 0;
1184
}
1185

    
1186
static void
1187
aio_write_help(void)
1188
{
1189
        printf(
1190
"\n"
1191
" asynchronously writes a range of bytes from the given offset source \n"
1192
" from multiple buffers\n"
1193
"\n"
1194
" Example:\n"
1195
" 'aio_write 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n"
1196
"\n"
1197
" Writes into a segment of the currently open file, using a buffer\n"
1198
" filled with a set pattern (0xcdcdcdcd).\n"
1199
" The write is performed asynchronously and the aio_flush command must be\n"
1200
" used to ensure all outstanding aio requests have been completed\n"
1201
" -P, -- use different pattern to fill file\n"
1202
" -C, -- report statistics in a machine parsable format\n"
1203
" -q, -- quiet mode, do not show I/O statistics\n"
1204
"\n");
1205
}
1206

    
1207
static int aio_write_f(int argc, char **argv);
1208

    
1209
static const cmdinfo_t aio_write_cmd = {
1210
        .name                = "aio_write",
1211
        .cfunc                = aio_write_f,
1212
        .argmin                = 2,
1213
        .argmax                = -1,
1214
        .args                = "[-Cq] [-P pattern ] off len [len..]",
1215
        .oneline        = "asynchronously writes a number of bytes",
1216
        .help                = aio_write_help,
1217
};
1218

    
1219
static int
1220
aio_write_f(int argc, char **argv)
1221
{
1222
        int nr_iov, c;
1223
        int pattern = 0xcd;
1224
        struct aio_ctx *ctx = calloc(1, sizeof(struct aio_ctx));
1225
        BlockDriverAIOCB *acb;
1226

    
1227
        while ((c = getopt(argc, argv, "CqP:")) != EOF) {
1228
                switch (c) {
1229
                case 'C':
1230
                        ctx->Cflag = 1;
1231
                        break;
1232
                case 'q':
1233
                        ctx->qflag = 1;
1234
                        break;
1235
                case 'P':
1236
                        pattern = parse_pattern(optarg);
1237
                        if (pattern < 0)
1238
                                return 0;
1239
                        break;
1240
                default:
1241
                        free(ctx);
1242
                        return command_usage(&aio_write_cmd);
1243
                }
1244
        }
1245

    
1246
        if (optind > argc - 2) {
1247
                free(ctx);
1248
                return command_usage(&aio_write_cmd);
1249
        }
1250

    
1251
        ctx->offset = cvtnum(argv[optind]);
1252
        if (ctx->offset < 0) {
1253
                printf("non-numeric length argument -- %s\n", argv[optind]);
1254
                free(ctx);
1255
                return 0;
1256
        }
1257
        optind++;
1258

    
1259
        if (ctx->offset & 0x1ff) {
1260
                printf("offset %" PRId64 " is not sector aligned\n",
1261
                       ctx->offset);
1262
                free(ctx);
1263
                return 0;
1264
        }
1265

    
1266
        nr_iov = argc - optind;
1267
        ctx->buf = create_iovec(&ctx->qiov, &argv[optind], nr_iov, pattern);
1268

    
1269
        gettimeofday(&ctx->t1, NULL);
1270
        acb = bdrv_aio_writev(bs, ctx->offset >> 9, &ctx->qiov,
1271
                              ctx->qiov.size >> 9, aio_write_done, ctx);
1272
        if (!acb) {
1273
                free(ctx->buf);
1274
                free(ctx);
1275
                return -EIO;
1276
        }
1277

    
1278
        return 0;
1279
}
1280

    
1281
static int
1282
aio_flush_f(int argc, char **argv)
1283
{
1284
        qemu_aio_flush();
1285
        return 0;
1286
}
1287

    
1288
static const cmdinfo_t aio_flush_cmd = {
1289
        .name                = "aio_flush",
1290
        .cfunc                = aio_flush_f,
1291
        .oneline        = "completes all outstanding aio requests"
1292
};
1293

    
1294
static int
1295
flush_f(int argc, char **argv)
1296
{
1297
        bdrv_flush(bs);
1298
        return 0;
1299
}
1300

    
1301
static const cmdinfo_t flush_cmd = {
1302
        .name                = "flush",
1303
        .altname        = "f",
1304
        .cfunc                = flush_f,
1305
        .oneline        = "flush all in-core file state to disk",
1306
};
1307

    
1308
static int
1309
truncate_f(int argc, char **argv)
1310
{
1311
        int64_t offset;
1312
        int ret;
1313

    
1314
        offset = cvtnum(argv[1]);
1315
        if (offset < 0) {
1316
                printf("non-numeric truncate argument -- %s\n", argv[1]);
1317
                return 0;
1318
        }
1319

    
1320
        ret = bdrv_truncate(bs, offset);
1321
        if (ret < 0) {
1322
                printf("truncate: %s\n", strerror(-ret));
1323
                return 0;
1324
        }
1325

    
1326
        return 0;
1327
}
1328

    
1329
static const cmdinfo_t truncate_cmd = {
1330
        .name                = "truncate",
1331
        .altname        = "t",
1332
        .cfunc                = truncate_f,
1333
        .argmin                = 1,
1334
        .argmax                = 1,
1335
        .args                = "off",
1336
        .oneline        = "truncates the current file at the given offset",
1337
};
1338

    
1339
static int
1340
length_f(int argc, char **argv)
1341
{
1342
        int64_t size;
1343
        char s1[64];
1344

    
1345
        size = bdrv_getlength(bs);
1346
        if (size < 0) {
1347
                printf("getlength: %s\n", strerror(-size));
1348
                return 0;
1349
        }
1350

    
1351
        cvtstr(size, s1, sizeof(s1));
1352
        printf("%s\n", s1);
1353
        return 0;
1354
}
1355

    
1356

    
1357
static const cmdinfo_t length_cmd = {
1358
        .name                = "length",
1359
        .altname        = "l",
1360
        .cfunc                = length_f,
1361
        .oneline        = "gets the length of the current file",
1362
};
1363

    
1364

    
1365
static int
1366
info_f(int argc, char **argv)
1367
{
1368
        BlockDriverInfo bdi;
1369
        char s1[64], s2[64];
1370
        int ret;
1371

    
1372
        if (bs->drv && bs->drv->format_name)
1373
                printf("format name: %s\n", bs->drv->format_name);
1374
        if (bs->drv && bs->drv->protocol_name)
1375
                printf("format name: %s\n", bs->drv->protocol_name);
1376

    
1377
        ret = bdrv_get_info(bs, &bdi);
1378
        if (ret)
1379
                return 0;
1380

    
1381
        cvtstr(bdi.cluster_size, s1, sizeof(s1));
1382
        cvtstr(bdi.vm_state_offset, s2, sizeof(s2));
1383

    
1384
        printf("cluster size: %s\n", s1);
1385
        printf("vm state offset: %s\n", s2);
1386

    
1387
        return 0;
1388
}
1389

    
1390

    
1391

    
1392
static const cmdinfo_t info_cmd = {
1393
        .name                = "info",
1394
        .altname        = "i",
1395
        .cfunc                = info_f,
1396
        .oneline        = "prints information about the current file",
1397
};
1398

    
1399
static void
1400
discard_help(void)
1401
{
1402
        printf(
1403
"\n"
1404
" discards a range of bytes from the given offset\n"
1405
"\n"
1406
" Example:\n"
1407
" 'discard 512 1k' - discards 1 kilobyte from 512 bytes into the file\n"
1408
"\n"
1409
" Discards a segment of the currently open file.\n"
1410
" -C, -- report statistics in a machine parsable format\n"
1411
" -q, -- quiet mode, do not show I/O statistics\n"
1412
"\n");
1413
}
1414

    
1415
static int discard_f(int argc, char **argv);
1416

    
1417
static const cmdinfo_t discard_cmd = {
1418
        .name                = "discard",
1419
        .altname        = "d",
1420
        .cfunc                = discard_f,
1421
        .argmin                = 2,
1422
        .argmax                = -1,
1423
        .args                = "[-Cq] off len",
1424
        .oneline        = "discards a number of bytes at a specified offset",
1425
        .help                = discard_help,
1426
};
1427

    
1428
static int
1429
discard_f(int argc, char **argv)
1430
{
1431
        struct timeval t1, t2;
1432
        int Cflag = 0, qflag = 0;
1433
        int c, ret;
1434
        int64_t offset;
1435
        int count;
1436

    
1437
        while ((c = getopt(argc, argv, "Cq")) != EOF) {
1438
                switch (c) {
1439
                case 'C':
1440
                        Cflag = 1;
1441
                        break;
1442
                case 'q':
1443
                        qflag = 1;
1444
                        break;
1445
                default:
1446
                        return command_usage(&discard_cmd);
1447
                }
1448
        }
1449

    
1450
        if (optind != argc - 2) {
1451
                return command_usage(&discard_cmd);
1452
        }
1453

    
1454
        offset = cvtnum(argv[optind]);
1455
        if (offset < 0) {
1456
                printf("non-numeric length argument -- %s\n", argv[optind]);
1457
                return 0;
1458
        }
1459

    
1460
        optind++;
1461
        count = cvtnum(argv[optind]);
1462
        if (count < 0) {
1463
                printf("non-numeric length argument -- %s\n", argv[optind]);
1464
                return 0;
1465
        }
1466

    
1467
        gettimeofday(&t1, NULL);
1468
        ret = bdrv_discard(bs, offset >> BDRV_SECTOR_BITS, count >> BDRV_SECTOR_BITS);
1469
        gettimeofday(&t2, NULL);
1470

    
1471
        if (ret < 0) {
1472
                printf("discard failed: %s\n", strerror(-ret));
1473
                goto out;
1474
        }
1475

    
1476
        /* Finally, report back -- -C gives a parsable format */
1477
        if (!qflag) {
1478
                t2 = tsub(t2, t1);
1479
                print_report("discard", &t2, offset, count, count, 1, Cflag);
1480
        }
1481

    
1482
out:
1483
        return 0;
1484
}
1485

    
1486
static int
1487
alloc_f(int argc, char **argv)
1488
{
1489
        int64_t offset;
1490
        int nb_sectors, remaining;
1491
        char s1[64];
1492
        int num, sum_alloc;
1493
        int ret;
1494

    
1495
        offset = cvtnum(argv[1]);
1496
        if (offset & 0x1ff) {
1497
                printf("offset %" PRId64 " is not sector aligned\n",
1498
                       offset);
1499
                return 0;
1500
        }
1501

    
1502
        if (argc == 3)
1503
                nb_sectors = cvtnum(argv[2]);
1504
        else
1505
                nb_sectors = 1;
1506

    
1507
        remaining = nb_sectors;
1508
        sum_alloc = 0;
1509
        while (remaining) {
1510
                ret = bdrv_is_allocated(bs, offset >> 9, nb_sectors, &num);
1511
                remaining -= num;
1512
                if (ret) {
1513
                        sum_alloc += num;
1514
                }
1515
        }
1516

    
1517
        cvtstr(offset, s1, sizeof(s1));
1518

    
1519
        printf("%d/%d sectors allocated at offset %s\n",
1520
               sum_alloc, nb_sectors, s1);
1521
        return 0;
1522
}
1523

    
1524
static const cmdinfo_t alloc_cmd = {
1525
        .name                = "alloc",
1526
        .altname        = "a",
1527
        .argmin                = 1,
1528
        .argmax                = 2,
1529
        .cfunc                = alloc_f,
1530
        .args                = "off [sectors]",
1531
        .oneline        = "checks if a sector is present in the file",
1532
};
1533

    
1534
static int
1535
map_f(int argc, char **argv)
1536
{
1537
        int64_t offset;
1538
        int64_t nb_sectors;
1539
        char s1[64];
1540
        int num, num_checked;
1541
        int ret;
1542
        const char *retstr;
1543

    
1544
        offset = 0;
1545
        nb_sectors = bs->total_sectors;
1546

    
1547
        do {
1548
                num_checked = MIN(nb_sectors, INT_MAX);
1549
                ret = bdrv_is_allocated(bs, offset, num_checked, &num);
1550
                retstr = ret ? "    allocated" : "not allocated";
1551
                cvtstr(offset << 9ULL, s1, sizeof(s1));
1552
                printf("[% 24" PRId64 "] % 8d/% 8d sectors %s at offset %s (%d)\n",
1553
                                offset << 9ULL, num, num_checked, retstr, s1, ret);
1554

    
1555
                offset += num;
1556
                nb_sectors -= num;
1557
        } while(offset < bs->total_sectors);
1558

    
1559
        return 0;
1560
}
1561

    
1562
static const cmdinfo_t map_cmd = {
1563
       .name           = "map",
1564
       .argmin         = 0,
1565
       .argmax         = 0,
1566
       .cfunc          = map_f,
1567
       .args           = "",
1568
       .oneline        = "prints the allocated areas of a file",
1569
};
1570

    
1571

    
1572
static int
1573
close_f(int argc, char **argv)
1574
{
1575
        bdrv_close(bs);
1576
        bs = NULL;
1577
        return 0;
1578
}
1579

    
1580
static const cmdinfo_t close_cmd = {
1581
        .name                = "close",
1582
        .altname        = "c",
1583
        .cfunc                = close_f,
1584
        .oneline        = "close the current open file",
1585
};
1586

    
1587
static int openfile(char *name, int flags, int growable)
1588
{
1589
        if (bs) {
1590
                fprintf(stderr, "file open already, try 'help close'\n");
1591
                return 1;
1592
        }
1593

    
1594
        if (growable) {
1595
                if (bdrv_file_open(&bs, name, flags)) {
1596
                        fprintf(stderr, "%s: can't open device %s\n", progname, name);
1597
                        return 1;
1598
                }
1599
        } else {
1600
                bs = bdrv_new("hda");
1601

    
1602
                if (bdrv_open(bs, name, flags, NULL) < 0) {
1603
                        fprintf(stderr, "%s: can't open device %s\n", progname, name);
1604
                        bs = NULL;
1605
                        return 1;
1606
                }
1607
        }
1608

    
1609
        return 0;
1610
}
1611

    
1612
static void
1613
open_help(void)
1614
{
1615
        printf(
1616
"\n"
1617
" opens a new file in the requested mode\n"
1618
"\n"
1619
" Example:\n"
1620
" 'open -Cn /tmp/data' - creates/opens data file read-write and uncached\n"
1621
"\n"
1622
" Opens a file for subsequent use by all of the other qemu-io commands.\n"
1623
" -r, -- open file read-only\n"
1624
" -s, -- use snapshot file\n"
1625
" -n, -- disable host cache\n"
1626
" -g, -- allow file to grow (only applies to protocols)"
1627
"\n");
1628
}
1629

    
1630
static int open_f(int argc, char **argv);
1631

    
1632
static const cmdinfo_t open_cmd = {
1633
        .name                = "open",
1634
        .altname        = "o",
1635
        .cfunc                = open_f,
1636
        .argmin                = 1,
1637
        .argmax                = -1,
1638
        .flags                = CMD_NOFILE_OK,
1639
        .args                = "[-Crsn] [path]",
1640
        .oneline        = "open the file specified by path",
1641
        .help                = open_help,
1642
};
1643

    
1644
static int
1645
open_f(int argc, char **argv)
1646
{
1647
        int flags = 0;
1648
        int readonly = 0;
1649
        int growable = 0;
1650
        int c;
1651

    
1652
        while ((c = getopt(argc, argv, "snrg")) != EOF) {
1653
                switch (c) {
1654
                case 's':
1655
                        flags |= BDRV_O_SNAPSHOT;
1656
                        break;
1657
                case 'n':
1658
                        flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
1659
                        break;
1660
                case 'r':
1661
                        readonly = 1;
1662
                        break;
1663
                case 'g':
1664
                        growable = 1;
1665
                        break;
1666
                default:
1667
                        return command_usage(&open_cmd);
1668
                }
1669
        }
1670

    
1671
        if (!readonly) {
1672
            flags |= BDRV_O_RDWR;
1673
        }
1674

    
1675
        if (optind != argc - 1)
1676
                return command_usage(&open_cmd);
1677

    
1678
        return openfile(argv[optind], flags, growable);
1679
}
1680

    
1681
static int
1682
init_args_command(
1683
        int     index)
1684
{
1685
        /* only one device allowed so far */
1686
        if (index >= 1)
1687
                return 0;
1688
        return ++index;
1689
}
1690

    
1691
static int
1692
init_check_command(
1693
        const cmdinfo_t *ct)
1694
{
1695
        if (ct->flags & CMD_FLAG_GLOBAL)
1696
                return 1;
1697
        if (!(ct->flags & CMD_NOFILE_OK) && !bs) {
1698
                fprintf(stderr, "no file open, try 'help open'\n");
1699
                return 0;
1700
        }
1701
        return 1;
1702
}
1703

    
1704
static void usage(const char *name)
1705
{
1706
        printf(
1707
"Usage: %s [-h] [-V] [-rsnm] [-c cmd] ... [file]\n"
1708
"QEMU Disk exerciser\n"
1709
"\n"
1710
"  -c, --cmd            command to execute\n"
1711
"  -r, --read-only      export read-only\n"
1712
"  -s, --snapshot       use snapshot file\n"
1713
"  -n, --nocache        disable host cache\n"
1714
"  -g, --growable       allow file to grow (only applies to protocols)\n"
1715
"  -m, --misalign       misalign allocations for O_DIRECT\n"
1716
"  -k, --native-aio     use kernel AIO implementation (on Linux only)\n"
1717
"  -h, --help           display this help and exit\n"
1718
"  -V, --version        output version information and exit\n"
1719
"\n",
1720
        name);
1721
}
1722

    
1723

    
1724
int main(int argc, char **argv)
1725
{
1726
        int readonly = 0;
1727
        int growable = 0;
1728
        const char *sopt = "hVc:rsnmgk";
1729
        const struct option lopt[] = {
1730
                { "help", 0, NULL, 'h' },
1731
                { "version", 0, NULL, 'V' },
1732
                { "offset", 1, NULL, 'o' },
1733
                { "cmd", 1, NULL, 'c' },
1734
                { "read-only", 0, NULL, 'r' },
1735
                { "snapshot", 0, NULL, 's' },
1736
                { "nocache", 0, NULL, 'n' },
1737
                { "misalign", 0, NULL, 'm' },
1738
                { "growable", 0, NULL, 'g' },
1739
                { "native-aio", 0, NULL, 'k' },
1740
                { NULL, 0, NULL, 0 }
1741
        };
1742
        int c;
1743
        int opt_index = 0;
1744
        int flags = 0;
1745

    
1746
        progname = basename(argv[0]);
1747

    
1748
        while ((c = getopt_long(argc, argv, sopt, lopt, &opt_index)) != -1) {
1749
                switch (c) {
1750
                case 's':
1751
                        flags |= BDRV_O_SNAPSHOT;
1752
                        break;
1753
                case 'n':
1754
                        flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
1755
                        break;
1756
                case 'c':
1757
                        add_user_command(optarg);
1758
                        break;
1759
                case 'r':
1760
                        readonly = 1;
1761
                        break;
1762
                case 'm':
1763
                        misalign = 1;
1764
                        break;
1765
                case 'g':
1766
                        growable = 1;
1767
                        break;
1768
                case 'k':
1769
                        flags |= BDRV_O_NATIVE_AIO;
1770
                        break;
1771
                case 'V':
1772
                        printf("%s version %s\n", progname, VERSION);
1773
                        exit(0);
1774
                case 'h':
1775
                        usage(progname);
1776
                        exit(0);
1777
                default:
1778
                        usage(progname);
1779
                        exit(1);
1780
                }
1781
        }
1782

    
1783
        if ((argc - optind) > 1) {
1784
                usage(progname);
1785
                exit(1);
1786
        }
1787

    
1788
        bdrv_init();
1789

    
1790
        /* initialize commands */
1791
        quit_init();
1792
        help_init();
1793
        add_command(&open_cmd);
1794
        add_command(&close_cmd);
1795
        add_command(&read_cmd);
1796
        add_command(&readv_cmd);
1797
        add_command(&write_cmd);
1798
        add_command(&writev_cmd);
1799
        add_command(&multiwrite_cmd);
1800
        add_command(&aio_read_cmd);
1801
        add_command(&aio_write_cmd);
1802
        add_command(&aio_flush_cmd);
1803
        add_command(&flush_cmd);
1804
        add_command(&truncate_cmd);
1805
        add_command(&length_cmd);
1806
        add_command(&info_cmd);
1807
        add_command(&discard_cmd);
1808
        add_command(&alloc_cmd);
1809
        add_command(&map_cmd);
1810

    
1811
        add_args_command(init_args_command);
1812
        add_check_command(init_check_command);
1813

    
1814
        /* open the device */
1815
        if (!readonly) {
1816
            flags |= BDRV_O_RDWR;
1817
        }
1818

    
1819
        if ((argc - optind) == 1)
1820
                openfile(argv[optind], flags, growable);
1821
        command_loop();
1822

    
1823
        /*
1824
         * Make sure all outstanding requests get flushed the program exits.
1825
         */
1826
        qemu_aio_flush();
1827

    
1828
        if (bs)
1829
                bdrv_close(bs);
1830
        return 0;
1831
}