root / qemu-io.c @ d7585251
History | View | Annotate | Download (20.4 kB)
1 | e3aff4f6 | aliguori | /*
|
---|---|---|---|
2 | e3aff4f6 | aliguori | * Command line utility to exercise the QEMU I/O path.
|
3 | e3aff4f6 | aliguori | *
|
4 | e3aff4f6 | aliguori | * Copyright (C) 2009 Red Hat, Inc.
|
5 | e3aff4f6 | aliguori | * Copyright (c) 2003-2005 Silicon Graphics, Inc.
|
6 | e3aff4f6 | aliguori | *
|
7 | e3aff4f6 | aliguori | * This work is licensed under the terms of the GNU GPL, version 2 or later.
|
8 | e3aff4f6 | aliguori | * See the COPYING file in the top-level directory.
|
9 | e3aff4f6 | aliguori | */
|
10 | e3aff4f6 | aliguori | #include <sys/types.h> |
11 | e3aff4f6 | aliguori | #include <stdarg.h> |
12 | e3aff4f6 | aliguori | #include <stdio.h> |
13 | e3aff4f6 | aliguori | #include <getopt.h> |
14 | e3aff4f6 | aliguori | |
15 | e3aff4f6 | aliguori | #include "qemu-common.h" |
16 | e3aff4f6 | aliguori | #include "block_int.h" |
17 | e3aff4f6 | aliguori | #include "cmd.h" |
18 | e3aff4f6 | aliguori | |
19 | e3aff4f6 | aliguori | #define VERSION "0.0.1" |
20 | e3aff4f6 | aliguori | |
21 | e3aff4f6 | aliguori | #define CMD_NOFILE_OK 0x01 |
22 | e3aff4f6 | aliguori | |
23 | e3aff4f6 | aliguori | char *progname;
|
24 | e3aff4f6 | aliguori | static BlockDriverState *bs;
|
25 | e3aff4f6 | aliguori | |
26 | e3aff4f6 | aliguori | static int misalign; |
27 | e3aff4f6 | aliguori | |
28 | e3aff4f6 | aliguori | /*
|
29 | e3aff4f6 | aliguori | * Memory allocation helpers.
|
30 | e3aff4f6 | aliguori | *
|
31 | e3aff4f6 | aliguori | * Make sure memory is aligned by default, or purposefully misaligned if
|
32 | e3aff4f6 | aliguori | * that is specified on the command line.
|
33 | e3aff4f6 | aliguori | */
|
34 | e3aff4f6 | aliguori | |
35 | e3aff4f6 | aliguori | #define MISALIGN_OFFSET 16 |
36 | e3aff4f6 | aliguori | static void *qemu_io_alloc(size_t len, int pattern) |
37 | e3aff4f6 | aliguori | { |
38 | e3aff4f6 | aliguori | void *buf;
|
39 | e3aff4f6 | aliguori | |
40 | e3aff4f6 | aliguori | if (misalign)
|
41 | e3aff4f6 | aliguori | len += MISALIGN_OFFSET; |
42 | e3aff4f6 | aliguori | buf = qemu_memalign(512, len);
|
43 | e3aff4f6 | aliguori | memset(buf, pattern, len); |
44 | e3aff4f6 | aliguori | if (misalign)
|
45 | e3aff4f6 | aliguori | buf += MISALIGN_OFFSET; |
46 | e3aff4f6 | aliguori | return buf;
|
47 | e3aff4f6 | aliguori | } |
48 | e3aff4f6 | aliguori | |
49 | e3aff4f6 | aliguori | static void qemu_io_free(void *p) |
50 | e3aff4f6 | aliguori | { |
51 | e3aff4f6 | aliguori | if (misalign)
|
52 | e3aff4f6 | aliguori | p -= MISALIGN_OFFSET; |
53 | e3aff4f6 | aliguori | qemu_vfree(p); |
54 | e3aff4f6 | aliguori | } |
55 | e3aff4f6 | aliguori | |
56 | e3aff4f6 | aliguori | static void |
57 | e3aff4f6 | aliguori | dump_buffer(char *buffer, int64_t offset, int len) |
58 | e3aff4f6 | aliguori | { |
59 | e3aff4f6 | aliguori | int i, j;
|
60 | e3aff4f6 | aliguori | char *p;
|
61 | e3aff4f6 | aliguori | |
62 | e3aff4f6 | aliguori | for (i = 0, p = buffer; i < len; i += 16) { |
63 | e3aff4f6 | aliguori | char *s = p;
|
64 | e3aff4f6 | aliguori | |
65 | e3aff4f6 | aliguori | printf("%08llx: ", (unsigned long long)offset + i); |
66 | e3aff4f6 | aliguori | for (j = 0; j < 16 && i + j < len; j++, p++) |
67 | e3aff4f6 | aliguori | printf("%02x ", *p);
|
68 | e3aff4f6 | aliguori | printf(" ");
|
69 | e3aff4f6 | aliguori | for (j = 0; j < 16 && i + j < len; j++, s++) { |
70 | e3aff4f6 | aliguori | if (isalnum((int)*s)) |
71 | e3aff4f6 | aliguori | printf("%c", *s);
|
72 | e3aff4f6 | aliguori | else
|
73 | e3aff4f6 | aliguori | printf(".");
|
74 | e3aff4f6 | aliguori | } |
75 | e3aff4f6 | aliguori | printf("\n");
|
76 | e3aff4f6 | aliguori | } |
77 | e3aff4f6 | aliguori | } |
78 | e3aff4f6 | aliguori | |
79 | e3aff4f6 | aliguori | static void |
80 | e3aff4f6 | aliguori | print_report(const char *op, struct timeval *t, int64_t offset, |
81 | e3aff4f6 | aliguori | int count, int total, int cnt, int Cflag) |
82 | e3aff4f6 | aliguori | { |
83 | e3aff4f6 | aliguori | char s1[64], s2[64], ts[64]; |
84 | e3aff4f6 | aliguori | |
85 | e3aff4f6 | aliguori | timestr(t, ts, sizeof(ts), Cflag ? VERBOSE_FIXED_TIME : 0); |
86 | e3aff4f6 | aliguori | if (!Cflag) {
|
87 | e3aff4f6 | aliguori | cvtstr((double)total, s1, sizeof(s1)); |
88 | e3aff4f6 | aliguori | cvtstr(tdiv((double)total, *t), s2, sizeof(s2)); |
89 | e3aff4f6 | aliguori | printf("%s %d/%d bytes at offset %lld\n",
|
90 | e3aff4f6 | aliguori | op, total, count, (long long)offset); |
91 | e3aff4f6 | aliguori | printf("%s, %d ops; %s (%s/sec and %.4f ops/sec)\n",
|
92 | e3aff4f6 | aliguori | s1, cnt, ts, s2, tdiv((double)cnt, *t));
|
93 | e3aff4f6 | aliguori | } else {/* bytes,ops,time,bytes/sec,ops/sec */ |
94 | e3aff4f6 | aliguori | printf("%d,%d,%s,%.3f,%.3f\n",
|
95 | e3aff4f6 | aliguori | total, cnt, ts, |
96 | e3aff4f6 | aliguori | tdiv((double)total, *t),
|
97 | e3aff4f6 | aliguori | tdiv((double)cnt, *t));
|
98 | e3aff4f6 | aliguori | } |
99 | e3aff4f6 | aliguori | } |
100 | e3aff4f6 | aliguori | |
101 | e3aff4f6 | aliguori | static int do_read(char *buf, int64_t offset, int count, int *total) |
102 | e3aff4f6 | aliguori | { |
103 | e3aff4f6 | aliguori | int ret;
|
104 | e3aff4f6 | aliguori | |
105 | e3aff4f6 | aliguori | ret = bdrv_read(bs, offset >> 9, (uint8_t *)buf, count >> 9); |
106 | e3aff4f6 | aliguori | if (ret < 0) |
107 | e3aff4f6 | aliguori | return ret;
|
108 | e3aff4f6 | aliguori | *total = count; |
109 | e3aff4f6 | aliguori | return 1; |
110 | e3aff4f6 | aliguori | } |
111 | e3aff4f6 | aliguori | |
112 | e3aff4f6 | aliguori | static int do_write(char *buf, int64_t offset, int count, int *total) |
113 | e3aff4f6 | aliguori | { |
114 | e3aff4f6 | aliguori | int ret;
|
115 | e3aff4f6 | aliguori | |
116 | e3aff4f6 | aliguori | ret = bdrv_write(bs, offset >> 9, (uint8_t *)buf, count >> 9); |
117 | e3aff4f6 | aliguori | if (ret < 0) |
118 | e3aff4f6 | aliguori | return ret;
|
119 | e3aff4f6 | aliguori | *total = count; |
120 | e3aff4f6 | aliguori | return 1; |
121 | e3aff4f6 | aliguori | } |
122 | e3aff4f6 | aliguori | |
123 | e3aff4f6 | aliguori | static int do_pread(char *buf, int64_t offset, int count, int *total) |
124 | e3aff4f6 | aliguori | { |
125 | e3aff4f6 | aliguori | *total = bdrv_pread(bs, offset, (uint8_t *)buf, count); |
126 | e3aff4f6 | aliguori | if (*total < 0) |
127 | e3aff4f6 | aliguori | return *total;
|
128 | e3aff4f6 | aliguori | return 1; |
129 | e3aff4f6 | aliguori | } |
130 | e3aff4f6 | aliguori | |
131 | e3aff4f6 | aliguori | static int do_pwrite(char *buf, int64_t offset, int count, int *total) |
132 | e3aff4f6 | aliguori | { |
133 | e3aff4f6 | aliguori | *total = bdrv_pwrite(bs, offset, (uint8_t *)buf, count); |
134 | e3aff4f6 | aliguori | if (*total < 0) |
135 | e3aff4f6 | aliguori | return *total;
|
136 | e3aff4f6 | aliguori | return 1; |
137 | e3aff4f6 | aliguori | } |
138 | e3aff4f6 | aliguori | |
139 | e3aff4f6 | aliguori | #define NOT_DONE 0x7fffffff |
140 | e3aff4f6 | aliguori | static void aio_rw_done(void *opaque, int ret) |
141 | e3aff4f6 | aliguori | { |
142 | e3aff4f6 | aliguori | *(int *)opaque = ret;
|
143 | e3aff4f6 | aliguori | } |
144 | e3aff4f6 | aliguori | |
145 | e3aff4f6 | aliguori | static int do_aio_readv(QEMUIOVector *qiov, int64_t offset, int *total) |
146 | e3aff4f6 | aliguori | { |
147 | e3aff4f6 | aliguori | BlockDriverAIOCB *acb; |
148 | e3aff4f6 | aliguori | int async_ret = NOT_DONE;
|
149 | e3aff4f6 | aliguori | |
150 | e3aff4f6 | aliguori | acb = bdrv_aio_readv(bs, offset >> 9, qiov, qiov->size >> 9, |
151 | e3aff4f6 | aliguori | aio_rw_done, &async_ret); |
152 | e3aff4f6 | aliguori | if (!acb)
|
153 | e3aff4f6 | aliguori | return -EIO;
|
154 | e3aff4f6 | aliguori | |
155 | e3aff4f6 | aliguori | while (async_ret == NOT_DONE)
|
156 | e3aff4f6 | aliguori | qemu_aio_wait(); |
157 | e3aff4f6 | aliguori | |
158 | e3aff4f6 | aliguori | *total = qiov->size; |
159 | e3aff4f6 | aliguori | return async_ret < 0 ? async_ret : 1; |
160 | e3aff4f6 | aliguori | } |
161 | e3aff4f6 | aliguori | |
162 | e3aff4f6 | aliguori | static int do_aio_writev(QEMUIOVector *qiov, int64_t offset, int *total) |
163 | e3aff4f6 | aliguori | { |
164 | e3aff4f6 | aliguori | BlockDriverAIOCB *acb; |
165 | e3aff4f6 | aliguori | int async_ret = NOT_DONE;
|
166 | e3aff4f6 | aliguori | |
167 | e3aff4f6 | aliguori | acb = bdrv_aio_writev(bs, offset >> 9, qiov, qiov->size >> 9, |
168 | e3aff4f6 | aliguori | aio_rw_done, &async_ret); |
169 | e3aff4f6 | aliguori | if (!acb)
|
170 | e3aff4f6 | aliguori | return -EIO;
|
171 | e3aff4f6 | aliguori | |
172 | e3aff4f6 | aliguori | while (async_ret == NOT_DONE)
|
173 | e3aff4f6 | aliguori | qemu_aio_wait(); |
174 | e3aff4f6 | aliguori | |
175 | e3aff4f6 | aliguori | *total = qiov->size >> 9;
|
176 | e3aff4f6 | aliguori | return async_ret < 0 ? async_ret : 1; |
177 | e3aff4f6 | aliguori | } |
178 | e3aff4f6 | aliguori | |
179 | e3aff4f6 | aliguori | |
180 | e3aff4f6 | aliguori | static const cmdinfo_t read_cmd; |
181 | e3aff4f6 | aliguori | |
182 | e3aff4f6 | aliguori | static void |
183 | e3aff4f6 | aliguori | read_help(void)
|
184 | e3aff4f6 | aliguori | { |
185 | e3aff4f6 | aliguori | printf( |
186 | e3aff4f6 | aliguori | "\n"
|
187 | e3aff4f6 | aliguori | " reads a range of bytes from the given offset\n"
|
188 | e3aff4f6 | aliguori | "\n"
|
189 | e3aff4f6 | aliguori | " Example:\n"
|
190 | e3aff4f6 | aliguori | " 'read -v 512 1k' - dumps 1 kilobyte read from 512 bytes into the file\n"
|
191 | e3aff4f6 | aliguori | "\n"
|
192 | e3aff4f6 | aliguori | " Reads a segment of the currently open file, optionally dumping it to the\n"
|
193 | e3aff4f6 | aliguori | " standard output stream (with -v option) for subsequent inspection.\n"
|
194 | e3aff4f6 | aliguori | " -p, -- use bdrv_pread to read the file\n"
|
195 | e3aff4f6 | aliguori | " -C, -- report statistics in a machine parsable format\n"
|
196 | e3aff4f6 | aliguori | " -v, -- dump buffer to standard output\n"
|
197 | e3aff4f6 | aliguori | " -q, -- quite mode, do not show I/O statistics\n"
|
198 | e3aff4f6 | aliguori | "\n");
|
199 | e3aff4f6 | aliguori | } |
200 | e3aff4f6 | aliguori | |
201 | e3aff4f6 | aliguori | static int |
202 | e3aff4f6 | aliguori | read_f(int argc, char **argv) |
203 | e3aff4f6 | aliguori | { |
204 | e3aff4f6 | aliguori | struct timeval t1, t2;
|
205 | e3aff4f6 | aliguori | int Cflag = 0, pflag = 0, qflag = 0, vflag = 0; |
206 | e3aff4f6 | aliguori | int c, cnt;
|
207 | e3aff4f6 | aliguori | char *buf;
|
208 | e3aff4f6 | aliguori | int64_t offset; |
209 | e3aff4f6 | aliguori | int count, total;
|
210 | e3aff4f6 | aliguori | |
211 | e3aff4f6 | aliguori | while ((c = getopt(argc, argv, "Cpqv")) != EOF) { |
212 | e3aff4f6 | aliguori | switch (c) {
|
213 | e3aff4f6 | aliguori | case 'C': |
214 | e3aff4f6 | aliguori | Cflag = 1;
|
215 | e3aff4f6 | aliguori | break;
|
216 | e3aff4f6 | aliguori | case 'p': |
217 | e3aff4f6 | aliguori | pflag = 1;
|
218 | e3aff4f6 | aliguori | break;
|
219 | e3aff4f6 | aliguori | case 'q': |
220 | e3aff4f6 | aliguori | qflag = 1;
|
221 | e3aff4f6 | aliguori | break;
|
222 | e3aff4f6 | aliguori | case 'v': |
223 | e3aff4f6 | aliguori | vflag = 1;
|
224 | e3aff4f6 | aliguori | break;
|
225 | e3aff4f6 | aliguori | default:
|
226 | e3aff4f6 | aliguori | return command_usage(&read_cmd);
|
227 | e3aff4f6 | aliguori | } |
228 | e3aff4f6 | aliguori | } |
229 | e3aff4f6 | aliguori | |
230 | e3aff4f6 | aliguori | if (optind != argc - 2) |
231 | e3aff4f6 | aliguori | return command_usage(&read_cmd);
|
232 | e3aff4f6 | aliguori | |
233 | e3aff4f6 | aliguori | offset = cvtnum(argv[optind]); |
234 | e3aff4f6 | aliguori | if (offset < 0) { |
235 | e3aff4f6 | aliguori | printf("non-numeric length argument -- %s\n", argv[optind]);
|
236 | e3aff4f6 | aliguori | return 0; |
237 | e3aff4f6 | aliguori | } |
238 | e3aff4f6 | aliguori | |
239 | e3aff4f6 | aliguori | optind++; |
240 | e3aff4f6 | aliguori | count = cvtnum(argv[optind]); |
241 | e3aff4f6 | aliguori | if (count < 0) { |
242 | e3aff4f6 | aliguori | printf("non-numeric length argument -- %s\n", argv[optind]);
|
243 | e3aff4f6 | aliguori | return 0; |
244 | e3aff4f6 | aliguori | } |
245 | e3aff4f6 | aliguori | |
246 | e3aff4f6 | aliguori | if (!pflag)
|
247 | e3aff4f6 | aliguori | if (offset & 0x1ff) { |
248 | e3aff4f6 | aliguori | printf("offset %lld is not sector aligned\n",
|
249 | e3aff4f6 | aliguori | (long long)offset); |
250 | e3aff4f6 | aliguori | return 0; |
251 | e3aff4f6 | aliguori | |
252 | e3aff4f6 | aliguori | if (count & 0x1ff) { |
253 | e3aff4f6 | aliguori | printf("count %d is not sector aligned\n",
|
254 | e3aff4f6 | aliguori | count); |
255 | e3aff4f6 | aliguori | return 0; |
256 | e3aff4f6 | aliguori | } |
257 | e3aff4f6 | aliguori | } |
258 | e3aff4f6 | aliguori | |
259 | e3aff4f6 | aliguori | buf = qemu_io_alloc(count, 0xab);
|
260 | e3aff4f6 | aliguori | |
261 | e3aff4f6 | aliguori | gettimeofday(&t1, NULL);
|
262 | e3aff4f6 | aliguori | if (pflag)
|
263 | e3aff4f6 | aliguori | cnt = do_pread(buf, offset, count, &total); |
264 | e3aff4f6 | aliguori | else
|
265 | e3aff4f6 | aliguori | cnt = do_read(buf, offset, count, &total); |
266 | e3aff4f6 | aliguori | gettimeofday(&t2, NULL);
|
267 | e3aff4f6 | aliguori | |
268 | e3aff4f6 | aliguori | if (cnt < 0) { |
269 | e3aff4f6 | aliguori | printf("read failed: %s\n", strerror(-cnt));
|
270 | e3aff4f6 | aliguori | return 0; |
271 | e3aff4f6 | aliguori | } |
272 | e3aff4f6 | aliguori | |
273 | e3aff4f6 | aliguori | if (qflag)
|
274 | e3aff4f6 | aliguori | return 0; |
275 | e3aff4f6 | aliguori | |
276 | e3aff4f6 | aliguori | if (vflag)
|
277 | e3aff4f6 | aliguori | dump_buffer(buf, offset, count); |
278 | e3aff4f6 | aliguori | |
279 | e3aff4f6 | aliguori | /* Finally, report back -- -C gives a parsable format */
|
280 | e3aff4f6 | aliguori | t2 = tsub(t2, t1); |
281 | e3aff4f6 | aliguori | print_report("read", &t2, offset, count, total, cnt, Cflag);
|
282 | e3aff4f6 | aliguori | |
283 | e3aff4f6 | aliguori | qemu_io_free(buf); |
284 | e3aff4f6 | aliguori | |
285 | e3aff4f6 | aliguori | return 0; |
286 | e3aff4f6 | aliguori | } |
287 | e3aff4f6 | aliguori | |
288 | e3aff4f6 | aliguori | static const cmdinfo_t read_cmd = { |
289 | e3aff4f6 | aliguori | .name = "read",
|
290 | e3aff4f6 | aliguori | .altname = "r",
|
291 | e3aff4f6 | aliguori | .cfunc = read_f, |
292 | e3aff4f6 | aliguori | .argmin = 2,
|
293 | e3aff4f6 | aliguori | .argmax = -1,
|
294 | e3aff4f6 | aliguori | .args = "[-aCpqv] off len",
|
295 | e3aff4f6 | aliguori | .oneline = "reads a number of bytes at a specified offset",
|
296 | e3aff4f6 | aliguori | .help = read_help, |
297 | e3aff4f6 | aliguori | }; |
298 | e3aff4f6 | aliguori | |
299 | e3aff4f6 | aliguori | static const cmdinfo_t readv_cmd; |
300 | e3aff4f6 | aliguori | |
301 | e3aff4f6 | aliguori | static void |
302 | e3aff4f6 | aliguori | readv_help(void)
|
303 | e3aff4f6 | aliguori | { |
304 | e3aff4f6 | aliguori | printf( |
305 | e3aff4f6 | aliguori | "\n"
|
306 | e3aff4f6 | aliguori | " reads a range of bytes from the given offset into multiple buffers\n"
|
307 | e3aff4f6 | aliguori | "\n"
|
308 | e3aff4f6 | aliguori | " Example:\n"
|
309 | e3aff4f6 | aliguori | " 'readv -v 512 1k 1k ' - dumps 2 kilobytes read from 512 bytes into the file\n"
|
310 | e3aff4f6 | aliguori | "\n"
|
311 | e3aff4f6 | aliguori | " Reads a segment of the currently open file, optionally dumping it to the\n"
|
312 | e3aff4f6 | aliguori | " standard output stream (with -v option) for subsequent inspection.\n"
|
313 | e3aff4f6 | aliguori | " Uses multiple iovec buffers if more than one byte range is specified.\n"
|
314 | e3aff4f6 | aliguori | " -C, -- report statistics in a machine parsable format\n"
|
315 | e3aff4f6 | aliguori | " -v, -- dump buffer to standard output\n"
|
316 | e3aff4f6 | aliguori | " -q, -- quite mode, do not show I/O statistics\n"
|
317 | e3aff4f6 | aliguori | "\n");
|
318 | e3aff4f6 | aliguori | } |
319 | e3aff4f6 | aliguori | |
320 | e3aff4f6 | aliguori | static int |
321 | e3aff4f6 | aliguori | readv_f(int argc, char **argv) |
322 | e3aff4f6 | aliguori | { |
323 | e3aff4f6 | aliguori | struct timeval t1, t2;
|
324 | e3aff4f6 | aliguori | int Cflag = 0, qflag = 0, vflag = 0; |
325 | e3aff4f6 | aliguori | int c, cnt;
|
326 | e3aff4f6 | aliguori | char *buf, *p;
|
327 | e3aff4f6 | aliguori | int64_t offset; |
328 | e3aff4f6 | aliguori | int count = 0, total; |
329 | e3aff4f6 | aliguori | int nr_iov, i;
|
330 | e3aff4f6 | aliguori | QEMUIOVector qiov; |
331 | e3aff4f6 | aliguori | |
332 | e3aff4f6 | aliguori | while ((c = getopt(argc, argv, "Cqv")) != EOF) { |
333 | e3aff4f6 | aliguori | switch (c) {
|
334 | e3aff4f6 | aliguori | case 'C': |
335 | e3aff4f6 | aliguori | Cflag = 1;
|
336 | e3aff4f6 | aliguori | break;
|
337 | e3aff4f6 | aliguori | case 'q': |
338 | e3aff4f6 | aliguori | qflag = 1;
|
339 | e3aff4f6 | aliguori | break;
|
340 | e3aff4f6 | aliguori | case 'v': |
341 | e3aff4f6 | aliguori | vflag = 1;
|
342 | e3aff4f6 | aliguori | break;
|
343 | e3aff4f6 | aliguori | default:
|
344 | e3aff4f6 | aliguori | return command_usage(&readv_cmd);
|
345 | e3aff4f6 | aliguori | } |
346 | e3aff4f6 | aliguori | } |
347 | e3aff4f6 | aliguori | |
348 | e3aff4f6 | aliguori | if (optind > argc - 2) |
349 | e3aff4f6 | aliguori | return command_usage(&readv_cmd);
|
350 | e3aff4f6 | aliguori | |
351 | e3aff4f6 | aliguori | |
352 | e3aff4f6 | aliguori | offset = cvtnum(argv[optind]); |
353 | e3aff4f6 | aliguori | if (offset < 0) { |
354 | e3aff4f6 | aliguori | printf("non-numeric length argument -- %s\n", argv[optind]);
|
355 | e3aff4f6 | aliguori | return 0; |
356 | e3aff4f6 | aliguori | } |
357 | e3aff4f6 | aliguori | optind++; |
358 | e3aff4f6 | aliguori | |
359 | e3aff4f6 | aliguori | if (offset & 0x1ff) { |
360 | e3aff4f6 | aliguori | printf("offset %lld is not sector aligned\n",
|
361 | e3aff4f6 | aliguori | (long long)offset); |
362 | e3aff4f6 | aliguori | return 0; |
363 | e3aff4f6 | aliguori | } |
364 | e3aff4f6 | aliguori | |
365 | e3aff4f6 | aliguori | if (count & 0x1ff) { |
366 | e3aff4f6 | aliguori | printf("count %d is not sector aligned\n",
|
367 | e3aff4f6 | aliguori | count); |
368 | e3aff4f6 | aliguori | return 0; |
369 | e3aff4f6 | aliguori | } |
370 | e3aff4f6 | aliguori | |
371 | e3aff4f6 | aliguori | for (i = optind; i < argc; i++) {
|
372 | e3aff4f6 | aliguori | size_t len; |
373 | e3aff4f6 | aliguori | |
374 | e3aff4f6 | aliguori | len = cvtnum(argv[i]); |
375 | e3aff4f6 | aliguori | if (len < 0) { |
376 | e3aff4f6 | aliguori | printf("non-numeric length argument -- %s\n", argv[i]);
|
377 | e3aff4f6 | aliguori | return 0; |
378 | e3aff4f6 | aliguori | } |
379 | e3aff4f6 | aliguori | count += len; |
380 | e3aff4f6 | aliguori | } |
381 | e3aff4f6 | aliguori | |
382 | e3aff4f6 | aliguori | nr_iov = argc - optind; |
383 | e3aff4f6 | aliguori | qemu_iovec_init(&qiov, nr_iov); |
384 | e3aff4f6 | aliguori | buf = p = qemu_io_alloc(count, 0xab);
|
385 | e3aff4f6 | aliguori | for (i = 0; i < nr_iov; i++) { |
386 | e3aff4f6 | aliguori | size_t len; |
387 | e3aff4f6 | aliguori | |
388 | e3aff4f6 | aliguori | len = cvtnum(argv[optind]); |
389 | e3aff4f6 | aliguori | if (len < 0) { |
390 | e3aff4f6 | aliguori | printf("non-numeric length argument -- %s\n",
|
391 | e3aff4f6 | aliguori | argv[optind]); |
392 | e3aff4f6 | aliguori | return 0; |
393 | e3aff4f6 | aliguori | } |
394 | e3aff4f6 | aliguori | |
395 | e3aff4f6 | aliguori | qemu_iovec_add(&qiov, p, len); |
396 | e3aff4f6 | aliguori | p += len; |
397 | e3aff4f6 | aliguori | optind++; |
398 | e3aff4f6 | aliguori | } |
399 | e3aff4f6 | aliguori | |
400 | e3aff4f6 | aliguori | gettimeofday(&t1, NULL);
|
401 | e3aff4f6 | aliguori | cnt = do_aio_readv(&qiov, offset, &total); |
402 | e3aff4f6 | aliguori | gettimeofday(&t2, NULL);
|
403 | e3aff4f6 | aliguori | |
404 | e3aff4f6 | aliguori | if (cnt < 0) { |
405 | e3aff4f6 | aliguori | printf("readv failed: %s\n", strerror(-cnt));
|
406 | e3aff4f6 | aliguori | return 0; |
407 | e3aff4f6 | aliguori | } |
408 | e3aff4f6 | aliguori | |
409 | e3aff4f6 | aliguori | if (qflag)
|
410 | e3aff4f6 | aliguori | return 0; |
411 | e3aff4f6 | aliguori | |
412 | e3aff4f6 | aliguori | if (vflag)
|
413 | e3aff4f6 | aliguori | dump_buffer(buf, offset, qiov.size); |
414 | e3aff4f6 | aliguori | |
415 | e3aff4f6 | aliguori | /* Finally, report back -- -C gives a parsable format */
|
416 | e3aff4f6 | aliguori | t2 = tsub(t2, t1); |
417 | e3aff4f6 | aliguori | print_report("read", &t2, offset, qiov.size, total, cnt, Cflag);
|
418 | e3aff4f6 | aliguori | |
419 | e3aff4f6 | aliguori | qemu_io_free(buf); |
420 | e3aff4f6 | aliguori | |
421 | e3aff4f6 | aliguori | return 0; |
422 | e3aff4f6 | aliguori | } |
423 | e3aff4f6 | aliguori | |
424 | e3aff4f6 | aliguori | static const cmdinfo_t readv_cmd = { |
425 | e3aff4f6 | aliguori | .name = "readv",
|
426 | e3aff4f6 | aliguori | .cfunc = readv_f, |
427 | e3aff4f6 | aliguori | .argmin = 2,
|
428 | e3aff4f6 | aliguori | .argmax = -1,
|
429 | e3aff4f6 | aliguori | .args = "[-Cqv] off len [len..]",
|
430 | e3aff4f6 | aliguori | .oneline = "reads a number of bytes at a specified offset",
|
431 | e3aff4f6 | aliguori | .help = readv_help, |
432 | e3aff4f6 | aliguori | }; |
433 | e3aff4f6 | aliguori | |
434 | e3aff4f6 | aliguori | static const cmdinfo_t write_cmd; |
435 | e3aff4f6 | aliguori | |
436 | e3aff4f6 | aliguori | static void |
437 | e3aff4f6 | aliguori | write_help(void)
|
438 | e3aff4f6 | aliguori | { |
439 | e3aff4f6 | aliguori | printf( |
440 | e3aff4f6 | aliguori | "\n"
|
441 | e3aff4f6 | aliguori | " writes a range of bytes from the given offset\n"
|
442 | e3aff4f6 | aliguori | "\n"
|
443 | e3aff4f6 | aliguori | " Example:\n"
|
444 | e3aff4f6 | aliguori | " 'write 512 1k' - writes 1 kilobyte at 512 bytes into the open file\n"
|
445 | e3aff4f6 | aliguori | "\n"
|
446 | e3aff4f6 | aliguori | " Writes into a segment of the currently open file, using a buffer\n"
|
447 | e3aff4f6 | aliguori | " filled with a set pattern (0xcdcdcdcd).\n"
|
448 | e3aff4f6 | aliguori | " -p, -- use bdrv_pwrite to write the file\n"
|
449 | e3aff4f6 | aliguori | " -P, -- use different pattern to fill file\n"
|
450 | e3aff4f6 | aliguori | " -C, -- report statistics in a machine parsable format\n"
|
451 | e3aff4f6 | aliguori | " -q, -- quite mode, do not show I/O statistics\n"
|
452 | e3aff4f6 | aliguori | "\n");
|
453 | e3aff4f6 | aliguori | } |
454 | e3aff4f6 | aliguori | |
455 | e3aff4f6 | aliguori | static int |
456 | e3aff4f6 | aliguori | write_f(int argc, char **argv) |
457 | e3aff4f6 | aliguori | { |
458 | e3aff4f6 | aliguori | struct timeval t1, t2;
|
459 | e3aff4f6 | aliguori | int Cflag = 0, pflag = 0, qflag = 0; |
460 | e3aff4f6 | aliguori | int c, cnt;
|
461 | e3aff4f6 | aliguori | char *buf;
|
462 | e3aff4f6 | aliguori | int64_t offset; |
463 | e3aff4f6 | aliguori | int count, total;
|
464 | e3aff4f6 | aliguori | int pattern = 0xcd; |
465 | e3aff4f6 | aliguori | |
466 | e3aff4f6 | aliguori | while ((c = getopt(argc, argv, "CpP:q")) != EOF) { |
467 | e3aff4f6 | aliguori | switch (c) {
|
468 | e3aff4f6 | aliguori | case 'C': |
469 | e3aff4f6 | aliguori | Cflag = 1;
|
470 | e3aff4f6 | aliguori | break;
|
471 | e3aff4f6 | aliguori | case 'p': |
472 | e3aff4f6 | aliguori | pflag = 1;
|
473 | e3aff4f6 | aliguori | break;
|
474 | e3aff4f6 | aliguori | case 'P': |
475 | e3aff4f6 | aliguori | pattern = atoi(optarg); |
476 | e3aff4f6 | aliguori | break;
|
477 | e3aff4f6 | aliguori | case 'q': |
478 | e3aff4f6 | aliguori | qflag = 1;
|
479 | e3aff4f6 | aliguori | break;
|
480 | e3aff4f6 | aliguori | default:
|
481 | e3aff4f6 | aliguori | return command_usage(&write_cmd);
|
482 | e3aff4f6 | aliguori | } |
483 | e3aff4f6 | aliguori | } |
484 | e3aff4f6 | aliguori | |
485 | e3aff4f6 | aliguori | if (optind != argc - 2) |
486 | e3aff4f6 | aliguori | return command_usage(&write_cmd);
|
487 | e3aff4f6 | aliguori | |
488 | e3aff4f6 | aliguori | offset = cvtnum(argv[optind]); |
489 | e3aff4f6 | aliguori | if (offset < 0) { |
490 | e3aff4f6 | aliguori | printf("non-numeric length argument -- %s\n", argv[optind]);
|
491 | e3aff4f6 | aliguori | return 0; |
492 | e3aff4f6 | aliguori | } |
493 | e3aff4f6 | aliguori | |
494 | e3aff4f6 | aliguori | optind++; |
495 | e3aff4f6 | aliguori | count = cvtnum(argv[optind]); |
496 | e3aff4f6 | aliguori | if (count < 0) { |
497 | e3aff4f6 | aliguori | printf("non-numeric length argument -- %s\n", argv[optind]);
|
498 | e3aff4f6 | aliguori | return 0; |
499 | e3aff4f6 | aliguori | } |
500 | e3aff4f6 | aliguori | |
501 | e3aff4f6 | aliguori | if (!pflag) {
|
502 | e3aff4f6 | aliguori | if (offset & 0x1ff) { |
503 | e3aff4f6 | aliguori | printf("offset %lld is not sector aligned\n",
|
504 | e3aff4f6 | aliguori | (long long)offset); |
505 | e3aff4f6 | aliguori | return 0; |
506 | e3aff4f6 | aliguori | } |
507 | e3aff4f6 | aliguori | |
508 | e3aff4f6 | aliguori | if (count & 0x1ff) { |
509 | e3aff4f6 | aliguori | printf("count %d is not sector aligned\n",
|
510 | e3aff4f6 | aliguori | count); |
511 | e3aff4f6 | aliguori | return 0; |
512 | e3aff4f6 | aliguori | } |
513 | e3aff4f6 | aliguori | } |
514 | e3aff4f6 | aliguori | |
515 | e3aff4f6 | aliguori | buf = qemu_io_alloc(count, pattern); |
516 | e3aff4f6 | aliguori | |
517 | e3aff4f6 | aliguori | gettimeofday(&t1, NULL);
|
518 | e3aff4f6 | aliguori | if (pflag)
|
519 | e3aff4f6 | aliguori | cnt = do_pwrite(buf, offset, count, &total); |
520 | e3aff4f6 | aliguori | else
|
521 | e3aff4f6 | aliguori | cnt = do_write(buf, offset, count, &total); |
522 | e3aff4f6 | aliguori | gettimeofday(&t2, NULL);
|
523 | e3aff4f6 | aliguori | |
524 | e3aff4f6 | aliguori | if (cnt < 0) { |
525 | e3aff4f6 | aliguori | printf("write failed: %s\n", strerror(-cnt));
|
526 | e3aff4f6 | aliguori | return 0; |
527 | e3aff4f6 | aliguori | } |
528 | e3aff4f6 | aliguori | |
529 | e3aff4f6 | aliguori | if (qflag)
|
530 | e3aff4f6 | aliguori | return 0; |
531 | e3aff4f6 | aliguori | |
532 | e3aff4f6 | aliguori | /* Finally, report back -- -C gives a parsable format */
|
533 | e3aff4f6 | aliguori | t2 = tsub(t2, t1); |
534 | e3aff4f6 | aliguori | print_report("wrote", &t2, offset, count, total, cnt, Cflag);
|
535 | e3aff4f6 | aliguori | |
536 | e3aff4f6 | aliguori | qemu_io_free(buf); |
537 | e3aff4f6 | aliguori | |
538 | e3aff4f6 | aliguori | return 0; |
539 | e3aff4f6 | aliguori | } |
540 | e3aff4f6 | aliguori | |
541 | e3aff4f6 | aliguori | static const cmdinfo_t write_cmd = { |
542 | e3aff4f6 | aliguori | .name = "write",
|
543 | e3aff4f6 | aliguori | .altname = "w",
|
544 | e3aff4f6 | aliguori | .cfunc = write_f, |
545 | e3aff4f6 | aliguori | .argmin = 2,
|
546 | e3aff4f6 | aliguori | .argmax = -1,
|
547 | e3aff4f6 | aliguori | .args = "[-aCpq] [-P pattern ] off len",
|
548 | e3aff4f6 | aliguori | .oneline = "writes a number of bytes at a specified offset",
|
549 | e3aff4f6 | aliguori | .help = write_help, |
550 | e3aff4f6 | aliguori | }; |
551 | e3aff4f6 | aliguori | |
552 | e3aff4f6 | aliguori | static const cmdinfo_t writev_cmd; |
553 | e3aff4f6 | aliguori | |
554 | e3aff4f6 | aliguori | static void |
555 | e3aff4f6 | aliguori | writev_help(void)
|
556 | e3aff4f6 | aliguori | { |
557 | e3aff4f6 | aliguori | printf( |
558 | e3aff4f6 | aliguori | "\n"
|
559 | e3aff4f6 | aliguori | " writes a range of bytes from the given offset source from multiple buffers\n"
|
560 | e3aff4f6 | aliguori | "\n"
|
561 | e3aff4f6 | aliguori | " Example:\n"
|
562 | e3aff4f6 | aliguori | " 'write 512 1k 1k' - writes 2 kilobytes at 512 bytes into the open file\n"
|
563 | e3aff4f6 | aliguori | "\n"
|
564 | e3aff4f6 | aliguori | " Writes into a segment of the currently open file, using a buffer\n"
|
565 | e3aff4f6 | aliguori | " filled with a set pattern (0xcdcdcdcd).\n"
|
566 | e3aff4f6 | aliguori | " -P, -- use different pattern to fill file\n"
|
567 | e3aff4f6 | aliguori | " -C, -- report statistics in a machine parsable format\n"
|
568 | e3aff4f6 | aliguori | " -q, -- quite mode, do not show I/O statistics\n"
|
569 | e3aff4f6 | aliguori | "\n");
|
570 | e3aff4f6 | aliguori | } |
571 | e3aff4f6 | aliguori | |
572 | e3aff4f6 | aliguori | static int |
573 | e3aff4f6 | aliguori | writev_f(int argc, char **argv) |
574 | e3aff4f6 | aliguori | { |
575 | e3aff4f6 | aliguori | struct timeval t1, t2;
|
576 | e3aff4f6 | aliguori | int Cflag = 0, qflag = 0; |
577 | e3aff4f6 | aliguori | int c, cnt;
|
578 | e3aff4f6 | aliguori | char *buf, *p;
|
579 | e3aff4f6 | aliguori | int64_t offset; |
580 | e3aff4f6 | aliguori | int count = 0, total; |
581 | e3aff4f6 | aliguori | int nr_iov, i;
|
582 | e3aff4f6 | aliguori | int pattern = 0xcd; |
583 | e3aff4f6 | aliguori | QEMUIOVector qiov; |
584 | e3aff4f6 | aliguori | |
585 | e3aff4f6 | aliguori | while ((c = getopt(argc, argv, "CqP:")) != EOF) { |
586 | e3aff4f6 | aliguori | switch (c) {
|
587 | e3aff4f6 | aliguori | case 'C': |
588 | e3aff4f6 | aliguori | Cflag = 1;
|
589 | e3aff4f6 | aliguori | break;
|
590 | e3aff4f6 | aliguori | case 'q': |
591 | e3aff4f6 | aliguori | qflag = 1;
|
592 | e3aff4f6 | aliguori | break;
|
593 | e3aff4f6 | aliguori | case 'P': |
594 | e3aff4f6 | aliguori | pattern = atoi(optarg); |
595 | e3aff4f6 | aliguori | break;
|
596 | e3aff4f6 | aliguori | default:
|
597 | e3aff4f6 | aliguori | return command_usage(&writev_cmd);
|
598 | e3aff4f6 | aliguori | } |
599 | e3aff4f6 | aliguori | } |
600 | e3aff4f6 | aliguori | |
601 | e3aff4f6 | aliguori | if (optind > argc - 2) |
602 | e3aff4f6 | aliguori | return command_usage(&writev_cmd);
|
603 | e3aff4f6 | aliguori | |
604 | e3aff4f6 | aliguori | offset = cvtnum(argv[optind]); |
605 | e3aff4f6 | aliguori | if (offset < 0) { |
606 | e3aff4f6 | aliguori | printf("non-numeric length argument -- %s\n", argv[optind]);
|
607 | e3aff4f6 | aliguori | return 0; |
608 | e3aff4f6 | aliguori | } |
609 | e3aff4f6 | aliguori | optind++; |
610 | e3aff4f6 | aliguori | |
611 | e3aff4f6 | aliguori | if (offset & 0x1ff) { |
612 | e3aff4f6 | aliguori | printf("offset %lld is not sector aligned\n",
|
613 | e3aff4f6 | aliguori | (long long)offset); |
614 | e3aff4f6 | aliguori | return 0; |
615 | e3aff4f6 | aliguori | } |
616 | e3aff4f6 | aliguori | |
617 | e3aff4f6 | aliguori | if (count & 0x1ff) { |
618 | e3aff4f6 | aliguori | printf("count %d is not sector aligned\n",
|
619 | e3aff4f6 | aliguori | count); |
620 | e3aff4f6 | aliguori | return 0; |
621 | e3aff4f6 | aliguori | } |
622 | e3aff4f6 | aliguori | |
623 | e3aff4f6 | aliguori | |
624 | e3aff4f6 | aliguori | for (i = optind; i < argc; i++) {
|
625 | e3aff4f6 | aliguori | size_t len; |
626 | e3aff4f6 | aliguori | |
627 | e3aff4f6 | aliguori | len = cvtnum(argv[optind]); |
628 | e3aff4f6 | aliguori | if (len < 0) { |
629 | e3aff4f6 | aliguori | printf("non-numeric length argument -- %s\n", argv[i]);
|
630 | e3aff4f6 | aliguori | return 0; |
631 | e3aff4f6 | aliguori | } |
632 | e3aff4f6 | aliguori | count += len; |
633 | e3aff4f6 | aliguori | } |
634 | e3aff4f6 | aliguori | |
635 | e3aff4f6 | aliguori | nr_iov = argc - optind; |
636 | e3aff4f6 | aliguori | qemu_iovec_init(&qiov, nr_iov); |
637 | e3aff4f6 | aliguori | buf = p = qemu_io_alloc(count, 0xab);
|
638 | e3aff4f6 | aliguori | for (i = 0; i < nr_iov; i++) { |
639 | e3aff4f6 | aliguori | size_t len; |
640 | e3aff4f6 | aliguori | |
641 | e3aff4f6 | aliguori | len = cvtnum(argv[optind]); |
642 | e3aff4f6 | aliguori | if (len < 0) { |
643 | e3aff4f6 | aliguori | printf("non-numeric length argument -- %s\n",
|
644 | e3aff4f6 | aliguori | argv[optind]); |
645 | e3aff4f6 | aliguori | return 0; |
646 | e3aff4f6 | aliguori | } |
647 | e3aff4f6 | aliguori | |
648 | e3aff4f6 | aliguori | qemu_iovec_add(&qiov, p, len); |
649 | e3aff4f6 | aliguori | p += len; |
650 | e3aff4f6 | aliguori | optind++; |
651 | e3aff4f6 | aliguori | } |
652 | e3aff4f6 | aliguori | |
653 | e3aff4f6 | aliguori | gettimeofday(&t1, NULL);
|
654 | e3aff4f6 | aliguori | cnt = do_aio_writev(&qiov, offset, &total); |
655 | e3aff4f6 | aliguori | gettimeofday(&t2, NULL);
|
656 | e3aff4f6 | aliguori | |
657 | e3aff4f6 | aliguori | if (cnt < 0) { |
658 | e3aff4f6 | aliguori | printf("writev failed: %s\n", strerror(-cnt));
|
659 | e3aff4f6 | aliguori | return 0; |
660 | e3aff4f6 | aliguori | } |
661 | e3aff4f6 | aliguori | |
662 | e3aff4f6 | aliguori | if (qflag)
|
663 | e3aff4f6 | aliguori | return 0; |
664 | e3aff4f6 | aliguori | |
665 | e3aff4f6 | aliguori | /* Finally, report back -- -C gives a parsable format */
|
666 | e3aff4f6 | aliguori | t2 = tsub(t2, t1); |
667 | e3aff4f6 | aliguori | print_report("wrote", &t2, offset, qiov.size, total, cnt, Cflag);
|
668 | e3aff4f6 | aliguori | |
669 | e3aff4f6 | aliguori | qemu_io_free(buf); |
670 | e3aff4f6 | aliguori | |
671 | e3aff4f6 | aliguori | return 0; |
672 | e3aff4f6 | aliguori | } |
673 | e3aff4f6 | aliguori | |
674 | e3aff4f6 | aliguori | static const cmdinfo_t writev_cmd = { |
675 | e3aff4f6 | aliguori | .name = "writev",
|
676 | e3aff4f6 | aliguori | .cfunc = writev_f, |
677 | e3aff4f6 | aliguori | .argmin = 2,
|
678 | e3aff4f6 | aliguori | .argmax = -1,
|
679 | e3aff4f6 | aliguori | .args = "[-Cq] [-P pattern ] off len [len..]",
|
680 | e3aff4f6 | aliguori | .oneline = "writes a number of bytes at a specified offset",
|
681 | e3aff4f6 | aliguori | .help = writev_help, |
682 | e3aff4f6 | aliguori | }; |
683 | e3aff4f6 | aliguori | |
684 | e3aff4f6 | aliguori | static int |
685 | e3aff4f6 | aliguori | flush_f(int argc, char **argv) |
686 | e3aff4f6 | aliguori | { |
687 | e3aff4f6 | aliguori | bdrv_flush(bs); |
688 | e3aff4f6 | aliguori | return 0; |
689 | e3aff4f6 | aliguori | } |
690 | e3aff4f6 | aliguori | |
691 | e3aff4f6 | aliguori | static const cmdinfo_t flush_cmd = { |
692 | e3aff4f6 | aliguori | .name = "flush",
|
693 | e3aff4f6 | aliguori | .altname = "f",
|
694 | e3aff4f6 | aliguori | .cfunc = flush_f, |
695 | e3aff4f6 | aliguori | .oneline = "flush all in-core file state to disk",
|
696 | e3aff4f6 | aliguori | }; |
697 | e3aff4f6 | aliguori | |
698 | e3aff4f6 | aliguori | static int |
699 | e3aff4f6 | aliguori | truncate_f(int argc, char **argv) |
700 | e3aff4f6 | aliguori | { |
701 | e3aff4f6 | aliguori | int64_t offset; |
702 | e3aff4f6 | aliguori | int ret;
|
703 | e3aff4f6 | aliguori | |
704 | e3aff4f6 | aliguori | offset = cvtnum(argv[1]);
|
705 | e3aff4f6 | aliguori | if (offset < 0) { |
706 | e3aff4f6 | aliguori | printf("non-numeric truncate argument -- %s\n", argv[1]); |
707 | e3aff4f6 | aliguori | return 0; |
708 | e3aff4f6 | aliguori | } |
709 | e3aff4f6 | aliguori | |
710 | e3aff4f6 | aliguori | ret = bdrv_truncate(bs, offset); |
711 | e3aff4f6 | aliguori | if (ret < 0) { |
712 | e3aff4f6 | aliguori | printf("truncate: %s", strerror(ret));
|
713 | e3aff4f6 | aliguori | return 0; |
714 | e3aff4f6 | aliguori | } |
715 | e3aff4f6 | aliguori | |
716 | e3aff4f6 | aliguori | return 0; |
717 | e3aff4f6 | aliguori | } |
718 | e3aff4f6 | aliguori | |
719 | e3aff4f6 | aliguori | static const cmdinfo_t truncate_cmd = { |
720 | e3aff4f6 | aliguori | .name = "truncate",
|
721 | e3aff4f6 | aliguori | .altname = "t",
|
722 | e3aff4f6 | aliguori | .cfunc = truncate_f, |
723 | e3aff4f6 | aliguori | .argmin = 1,
|
724 | e3aff4f6 | aliguori | .argmax = 1,
|
725 | e3aff4f6 | aliguori | .args = "off",
|
726 | e3aff4f6 | aliguori | .oneline = "truncates the current file at the given offset",
|
727 | e3aff4f6 | aliguori | }; |
728 | e3aff4f6 | aliguori | |
729 | e3aff4f6 | aliguori | static int |
730 | e3aff4f6 | aliguori | length_f(int argc, char **argv) |
731 | e3aff4f6 | aliguori | { |
732 | e3aff4f6 | aliguori | int64_t size; |
733 | e3aff4f6 | aliguori | char s1[64]; |
734 | e3aff4f6 | aliguori | |
735 | e3aff4f6 | aliguori | size = bdrv_getlength(bs); |
736 | e3aff4f6 | aliguori | if (size < 0) { |
737 | e3aff4f6 | aliguori | printf("getlength: %s", strerror(size));
|
738 | e3aff4f6 | aliguori | return 0; |
739 | e3aff4f6 | aliguori | } |
740 | e3aff4f6 | aliguori | |
741 | e3aff4f6 | aliguori | cvtstr(size, s1, sizeof(s1));
|
742 | e3aff4f6 | aliguori | printf("%s\n", s1);
|
743 | e3aff4f6 | aliguori | return 0; |
744 | e3aff4f6 | aliguori | } |
745 | e3aff4f6 | aliguori | |
746 | e3aff4f6 | aliguori | |
747 | e3aff4f6 | aliguori | static const cmdinfo_t length_cmd = { |
748 | e3aff4f6 | aliguori | .name = "length",
|
749 | e3aff4f6 | aliguori | .altname = "l",
|
750 | e3aff4f6 | aliguori | .cfunc = length_f, |
751 | e3aff4f6 | aliguori | .oneline = "gets the length of the current file",
|
752 | e3aff4f6 | aliguori | }; |
753 | e3aff4f6 | aliguori | |
754 | e3aff4f6 | aliguori | |
755 | e3aff4f6 | aliguori | static int |
756 | e3aff4f6 | aliguori | info_f(int argc, char **argv) |
757 | e3aff4f6 | aliguori | { |
758 | e3aff4f6 | aliguori | BlockDriverInfo bdi; |
759 | e3aff4f6 | aliguori | char s1[64], s2[64]; |
760 | e3aff4f6 | aliguori | int ret;
|
761 | e3aff4f6 | aliguori | |
762 | e3aff4f6 | aliguori | if (bs->drv && bs->drv->format_name)
|
763 | e3aff4f6 | aliguori | printf("format name: %s\n", bs->drv->format_name);
|
764 | e3aff4f6 | aliguori | if (bs->drv && bs->drv->protocol_name)
|
765 | e3aff4f6 | aliguori | printf("format name: %s\n", bs->drv->protocol_name);
|
766 | e3aff4f6 | aliguori | |
767 | e3aff4f6 | aliguori | ret = bdrv_get_info(bs, &bdi); |
768 | e3aff4f6 | aliguori | if (ret)
|
769 | e3aff4f6 | aliguori | return 0; |
770 | e3aff4f6 | aliguori | |
771 | e3aff4f6 | aliguori | cvtstr(bdi.cluster_size, s1, sizeof(s1));
|
772 | e3aff4f6 | aliguori | cvtstr(bdi.vm_state_offset, s2, sizeof(s2));
|
773 | e3aff4f6 | aliguori | |
774 | e3aff4f6 | aliguori | printf("cluster size: %s\n", s1);
|
775 | e3aff4f6 | aliguori | printf("vm state offset: %s\n", s2);
|
776 | e3aff4f6 | aliguori | |
777 | e3aff4f6 | aliguori | return 0; |
778 | e3aff4f6 | aliguori | } |
779 | e3aff4f6 | aliguori | |
780 | e3aff4f6 | aliguori | |
781 | e3aff4f6 | aliguori | |
782 | e3aff4f6 | aliguori | static const cmdinfo_t info_cmd = { |
783 | e3aff4f6 | aliguori | .name = "info",
|
784 | e3aff4f6 | aliguori | .altname = "i",
|
785 | e3aff4f6 | aliguori | .cfunc = info_f, |
786 | e3aff4f6 | aliguori | .oneline = "prints information about the current file",
|
787 | e3aff4f6 | aliguori | }; |
788 | e3aff4f6 | aliguori | |
789 | e3aff4f6 | aliguori | static int |
790 | e3aff4f6 | aliguori | alloc_f(int argc, char **argv) |
791 | e3aff4f6 | aliguori | { |
792 | e3aff4f6 | aliguori | int64_t offset; |
793 | e3aff4f6 | aliguori | int nb_sectors;
|
794 | e3aff4f6 | aliguori | char s1[64]; |
795 | e3aff4f6 | aliguori | int num;
|
796 | e3aff4f6 | aliguori | int ret;
|
797 | e3aff4f6 | aliguori | |
798 | e3aff4f6 | aliguori | offset = cvtnum(argv[1]);
|
799 | e3aff4f6 | aliguori | if (offset & 0x1ff) { |
800 | e3aff4f6 | aliguori | printf("offset %lld is not sector aligned\n",
|
801 | e3aff4f6 | aliguori | (long long)offset); |
802 | e3aff4f6 | aliguori | return 0; |
803 | e3aff4f6 | aliguori | } |
804 | e3aff4f6 | aliguori | |
805 | e3aff4f6 | aliguori | if (argc == 3) |
806 | e3aff4f6 | aliguori | nb_sectors = cvtnum(argv[2]);
|
807 | e3aff4f6 | aliguori | else
|
808 | e3aff4f6 | aliguori | nb_sectors = 1;
|
809 | e3aff4f6 | aliguori | |
810 | e3aff4f6 | aliguori | ret = bdrv_is_allocated(bs, offset >> 9, nb_sectors, &num);
|
811 | e3aff4f6 | aliguori | if (ret) {
|
812 | e3aff4f6 | aliguori | printf("is_allocated: %s", strerror(ret));
|
813 | e3aff4f6 | aliguori | return 0; |
814 | e3aff4f6 | aliguori | } |
815 | e3aff4f6 | aliguori | |
816 | e3aff4f6 | aliguori | cvtstr(offset, s1, sizeof(s1));
|
817 | e3aff4f6 | aliguori | |
818 | e3aff4f6 | aliguori | if (nb_sectors == 1) |
819 | e3aff4f6 | aliguori | printf("sector allocated at offset %s\n", s1);
|
820 | e3aff4f6 | aliguori | else
|
821 | e3aff4f6 | aliguori | printf("%d/%d sectors allocated at offset %s\n",
|
822 | e3aff4f6 | aliguori | num, nb_sectors, s1); |
823 | e3aff4f6 | aliguori | return 0; |
824 | e3aff4f6 | aliguori | } |
825 | e3aff4f6 | aliguori | |
826 | e3aff4f6 | aliguori | static const cmdinfo_t alloc_cmd = { |
827 | e3aff4f6 | aliguori | .name = "alloc",
|
828 | e3aff4f6 | aliguori | .altname = "a",
|
829 | e3aff4f6 | aliguori | .argmin = 1,
|
830 | e3aff4f6 | aliguori | .argmax = 2,
|
831 | e3aff4f6 | aliguori | .cfunc = alloc_f, |
832 | e3aff4f6 | aliguori | .args = "off [sectors]",
|
833 | e3aff4f6 | aliguori | .oneline = "checks if a sector is present in the file",
|
834 | e3aff4f6 | aliguori | }; |
835 | e3aff4f6 | aliguori | |
836 | e3aff4f6 | aliguori | static int |
837 | e3aff4f6 | aliguori | close_f(int argc, char **argv) |
838 | e3aff4f6 | aliguori | { |
839 | e3aff4f6 | aliguori | bdrv_close(bs); |
840 | e3aff4f6 | aliguori | bs = NULL;
|
841 | e3aff4f6 | aliguori | return 0; |
842 | e3aff4f6 | aliguori | } |
843 | e3aff4f6 | aliguori | |
844 | e3aff4f6 | aliguori | static const cmdinfo_t close_cmd = { |
845 | e3aff4f6 | aliguori | .name = "close",
|
846 | e3aff4f6 | aliguori | .altname = "c",
|
847 | e3aff4f6 | aliguori | .cfunc = close_f, |
848 | e3aff4f6 | aliguori | .oneline = "close the current open file",
|
849 | e3aff4f6 | aliguori | }; |
850 | e3aff4f6 | aliguori | |
851 | e3aff4f6 | aliguori | static int openfile(char *name, int flags) |
852 | e3aff4f6 | aliguori | { |
853 | e3aff4f6 | aliguori | if (bs) {
|
854 | e3aff4f6 | aliguori | fprintf(stderr, "file open already, try 'help close'\n");
|
855 | e3aff4f6 | aliguori | return 1; |
856 | e3aff4f6 | aliguori | } |
857 | e3aff4f6 | aliguori | |
858 | e3aff4f6 | aliguori | bs = bdrv_new("hda");
|
859 | e3aff4f6 | aliguori | if (!bs)
|
860 | e3aff4f6 | aliguori | return 1; |
861 | e3aff4f6 | aliguori | |
862 | e3aff4f6 | aliguori | if (bdrv_open(bs, name, flags) == -1) { |
863 | e3aff4f6 | aliguori | fprintf(stderr, "%s: can't open device %s\n", progname, name);
|
864 | e3aff4f6 | aliguori | bs = NULL;
|
865 | e3aff4f6 | aliguori | return 1; |
866 | e3aff4f6 | aliguori | } |
867 | e3aff4f6 | aliguori | |
868 | e3aff4f6 | aliguori | return 0; |
869 | e3aff4f6 | aliguori | } |
870 | e3aff4f6 | aliguori | |
871 | e3aff4f6 | aliguori | static void |
872 | e3aff4f6 | aliguori | open_help(void)
|
873 | e3aff4f6 | aliguori | { |
874 | e3aff4f6 | aliguori | printf( |
875 | e3aff4f6 | aliguori | "\n"
|
876 | e3aff4f6 | aliguori | " opens a new file in the requested mode\n"
|
877 | e3aff4f6 | aliguori | "\n"
|
878 | e3aff4f6 | aliguori | " Example:\n"
|
879 | e3aff4f6 | aliguori | " 'open -Cn /tmp/data' - creates/opens data file read-write and uncached\n"
|
880 | e3aff4f6 | aliguori | "\n"
|
881 | e3aff4f6 | aliguori | " Opens a file for subsequent use by all of the other qemu-io commands.\n"
|
882 | e3aff4f6 | aliguori | " -C, -- create new file if it doesn't exist\n"
|
883 | e3aff4f6 | aliguori | " -r, -- open file read-only\n"
|
884 | e3aff4f6 | aliguori | " -s, -- use snapshot file\n"
|
885 | e3aff4f6 | aliguori | " -n, -- disable host cache\n"
|
886 | e3aff4f6 | aliguori | "\n");
|
887 | e3aff4f6 | aliguori | } |
888 | e3aff4f6 | aliguori | |
889 | e3aff4f6 | aliguori | static const cmdinfo_t open_cmd; |
890 | e3aff4f6 | aliguori | |
891 | e3aff4f6 | aliguori | static int |
892 | e3aff4f6 | aliguori | open_f(int argc, char **argv) |
893 | e3aff4f6 | aliguori | { |
894 | e3aff4f6 | aliguori | int flags = 0; |
895 | e3aff4f6 | aliguori | int readonly = 0; |
896 | e3aff4f6 | aliguori | int c;
|
897 | e3aff4f6 | aliguori | |
898 | e3aff4f6 | aliguori | while ((c = getopt(argc, argv, "snCr")) != EOF) { |
899 | e3aff4f6 | aliguori | switch (c) {
|
900 | e3aff4f6 | aliguori | case 's': |
901 | e3aff4f6 | aliguori | flags |= BDRV_O_SNAPSHOT; |
902 | e3aff4f6 | aliguori | break;
|
903 | e3aff4f6 | aliguori | case 'n': |
904 | e3aff4f6 | aliguori | flags |= BDRV_O_NOCACHE; |
905 | e3aff4f6 | aliguori | break;
|
906 | e3aff4f6 | aliguori | case 'C': |
907 | e3aff4f6 | aliguori | flags |= BDRV_O_CREAT; |
908 | e3aff4f6 | aliguori | break;
|
909 | e3aff4f6 | aliguori | case 'r': |
910 | e3aff4f6 | aliguori | readonly = 1;
|
911 | e3aff4f6 | aliguori | break;
|
912 | e3aff4f6 | aliguori | default:
|
913 | e3aff4f6 | aliguori | return command_usage(&open_cmd);
|
914 | e3aff4f6 | aliguori | } |
915 | e3aff4f6 | aliguori | } |
916 | e3aff4f6 | aliguori | |
917 | e3aff4f6 | aliguori | if (readonly)
|
918 | e3aff4f6 | aliguori | flags |= BDRV_O_RDONLY; |
919 | e3aff4f6 | aliguori | else
|
920 | e3aff4f6 | aliguori | flags |= BDRV_O_RDWR; |
921 | e3aff4f6 | aliguori | |
922 | e3aff4f6 | aliguori | if (optind != argc - 1) |
923 | e3aff4f6 | aliguori | return command_usage(&open_cmd);
|
924 | e3aff4f6 | aliguori | |
925 | e3aff4f6 | aliguori | return openfile(argv[optind], flags);
|
926 | e3aff4f6 | aliguori | } |
927 | e3aff4f6 | aliguori | |
928 | e3aff4f6 | aliguori | static const cmdinfo_t open_cmd = { |
929 | e3aff4f6 | aliguori | .name = "open",
|
930 | e3aff4f6 | aliguori | .altname = "o",
|
931 | e3aff4f6 | aliguori | .cfunc = open_f, |
932 | e3aff4f6 | aliguori | .argmin = 1,
|
933 | e3aff4f6 | aliguori | .argmax = -1,
|
934 | e3aff4f6 | aliguori | .flags = CMD_NOFILE_OK, |
935 | e3aff4f6 | aliguori | .args = "[-Crsn] [path]",
|
936 | e3aff4f6 | aliguori | .oneline = "open the file specified by path",
|
937 | e3aff4f6 | aliguori | .help = open_help, |
938 | e3aff4f6 | aliguori | }; |
939 | e3aff4f6 | aliguori | |
940 | e3aff4f6 | aliguori | static int |
941 | e3aff4f6 | aliguori | init_args_command( |
942 | e3aff4f6 | aliguori | int index)
|
943 | e3aff4f6 | aliguori | { |
944 | e3aff4f6 | aliguori | /* only one device allowed so far */
|
945 | e3aff4f6 | aliguori | if (index >= 1) |
946 | e3aff4f6 | aliguori | return 0; |
947 | e3aff4f6 | aliguori | return ++index;
|
948 | e3aff4f6 | aliguori | } |
949 | e3aff4f6 | aliguori | |
950 | e3aff4f6 | aliguori | static int |
951 | e3aff4f6 | aliguori | init_check_command( |
952 | e3aff4f6 | aliguori | const cmdinfo_t *ct)
|
953 | e3aff4f6 | aliguori | { |
954 | e3aff4f6 | aliguori | if (ct->flags & CMD_FLAG_GLOBAL)
|
955 | e3aff4f6 | aliguori | return 1; |
956 | e3aff4f6 | aliguori | if (!(ct->flags & CMD_NOFILE_OK) && !bs) {
|
957 | e3aff4f6 | aliguori | fprintf(stderr, "no file open, try 'help open'\n");
|
958 | e3aff4f6 | aliguori | return 0; |
959 | e3aff4f6 | aliguori | } |
960 | e3aff4f6 | aliguori | return 1; |
961 | e3aff4f6 | aliguori | } |
962 | e3aff4f6 | aliguori | |
963 | e3aff4f6 | aliguori | static void usage(const char *name) |
964 | e3aff4f6 | aliguori | { |
965 | e3aff4f6 | aliguori | printf( |
966 | e3aff4f6 | aliguori | "Usage: %s [-h] [-V] [-Crsnm] [-c cmd] ... [file]\n"
|
967 | e3aff4f6 | aliguori | "QEMU Disk excerciser\n"
|
968 | e3aff4f6 | aliguori | "\n"
|
969 | e3aff4f6 | aliguori | " -C, --create create new file if it doesn't exist\n"
|
970 | e3aff4f6 | aliguori | " -c, --cmd command to execute\n"
|
971 | e3aff4f6 | aliguori | " -r, --read-only export read-only\n"
|
972 | e3aff4f6 | aliguori | " -s, --snapshot use snapshot file\n"
|
973 | e3aff4f6 | aliguori | " -n, --nocache disable host cache\n"
|
974 | e3aff4f6 | aliguori | " -m, --misalign misalign allocations for O_DIRECT\n"
|
975 | e3aff4f6 | aliguori | " -h, --help display this help and exit\n"
|
976 | e3aff4f6 | aliguori | " -V, --version output version information and exit\n"
|
977 | e3aff4f6 | aliguori | "\n",
|
978 | e3aff4f6 | aliguori | name); |
979 | e3aff4f6 | aliguori | } |
980 | e3aff4f6 | aliguori | |
981 | e3aff4f6 | aliguori | |
982 | e3aff4f6 | aliguori | int main(int argc, char **argv) |
983 | e3aff4f6 | aliguori | { |
984 | e3aff4f6 | aliguori | int readonly = 0; |
985 | e3aff4f6 | aliguori | const char *sopt = "hVc:Crsnm"; |
986 | e3aff4f6 | aliguori | struct option lopt[] = {
|
987 | e3aff4f6 | aliguori | { "help", 0, 0, 'h' }, |
988 | e3aff4f6 | aliguori | { "version", 0, 0, 'V' }, |
989 | e3aff4f6 | aliguori | { "offset", 1, 0, 'o' }, |
990 | e3aff4f6 | aliguori | { "cmd", 1, 0, 'c' }, |
991 | e3aff4f6 | aliguori | { "create", 0, 0, 'C' }, |
992 | e3aff4f6 | aliguori | { "read-only", 0, 0, 'r' }, |
993 | e3aff4f6 | aliguori | { "snapshot", 0, 0, 's' }, |
994 | e3aff4f6 | aliguori | { "nocache", 0, 0, 'n' }, |
995 | e3aff4f6 | aliguori | { "misalign", 0, 0, 'm' }, |
996 | e3aff4f6 | aliguori | { NULL, 0, 0, 0 } |
997 | e3aff4f6 | aliguori | }; |
998 | e3aff4f6 | aliguori | int c;
|
999 | e3aff4f6 | aliguori | int opt_index = 0; |
1000 | e3aff4f6 | aliguori | int flags = 0; |
1001 | e3aff4f6 | aliguori | |
1002 | e3aff4f6 | aliguori | progname = basename(argv[0]);
|
1003 | e3aff4f6 | aliguori | |
1004 | e3aff4f6 | aliguori | while ((c = getopt_long(argc, argv, sopt, lopt, &opt_index)) != -1) { |
1005 | e3aff4f6 | aliguori | switch (c) {
|
1006 | e3aff4f6 | aliguori | case 's': |
1007 | e3aff4f6 | aliguori | flags |= BDRV_O_SNAPSHOT; |
1008 | e3aff4f6 | aliguori | break;
|
1009 | e3aff4f6 | aliguori | case 'n': |
1010 | e3aff4f6 | aliguori | flags |= BDRV_O_NOCACHE; |
1011 | e3aff4f6 | aliguori | break;
|
1012 | e3aff4f6 | aliguori | case 'c': |
1013 | e3aff4f6 | aliguori | add_user_command(optarg); |
1014 | e3aff4f6 | aliguori | break;
|
1015 | e3aff4f6 | aliguori | case 'C': |
1016 | e3aff4f6 | aliguori | flags |= BDRV_O_CREAT; |
1017 | e3aff4f6 | aliguori | break;
|
1018 | e3aff4f6 | aliguori | case 'r': |
1019 | e3aff4f6 | aliguori | readonly = 1;
|
1020 | e3aff4f6 | aliguori | break;
|
1021 | e3aff4f6 | aliguori | case 'm': |
1022 | e3aff4f6 | aliguori | misalign = 1;
|
1023 | e3aff4f6 | aliguori | break;
|
1024 | e3aff4f6 | aliguori | case 'V': |
1025 | e3aff4f6 | aliguori | printf("%s version %s\n", progname, VERSION);
|
1026 | e3aff4f6 | aliguori | exit(0);
|
1027 | e3aff4f6 | aliguori | case 'h': |
1028 | e3aff4f6 | aliguori | usage(progname); |
1029 | e3aff4f6 | aliguori | exit(0);
|
1030 | e3aff4f6 | aliguori | default:
|
1031 | e3aff4f6 | aliguori | usage(progname); |
1032 | e3aff4f6 | aliguori | exit(1);
|
1033 | e3aff4f6 | aliguori | } |
1034 | e3aff4f6 | aliguori | } |
1035 | e3aff4f6 | aliguori | |
1036 | e3aff4f6 | aliguori | if ((argc - optind) > 1) { |
1037 | e3aff4f6 | aliguori | usage(progname); |
1038 | e3aff4f6 | aliguori | exit(1);
|
1039 | e3aff4f6 | aliguori | } |
1040 | e3aff4f6 | aliguori | |
1041 | e3aff4f6 | aliguori | bdrv_init(); |
1042 | e3aff4f6 | aliguori | |
1043 | e3aff4f6 | aliguori | /* initialize commands */
|
1044 | e3aff4f6 | aliguori | quit_init(); |
1045 | e3aff4f6 | aliguori | help_init(); |
1046 | e3aff4f6 | aliguori | add_command(&open_cmd); |
1047 | e3aff4f6 | aliguori | add_command(&close_cmd); |
1048 | e3aff4f6 | aliguori | add_command(&read_cmd); |
1049 | e3aff4f6 | aliguori | add_command(&readv_cmd); |
1050 | e3aff4f6 | aliguori | add_command(&write_cmd); |
1051 | e3aff4f6 | aliguori | add_command(&writev_cmd); |
1052 | e3aff4f6 | aliguori | add_command(&flush_cmd); |
1053 | e3aff4f6 | aliguori | add_command(&truncate_cmd); |
1054 | e3aff4f6 | aliguori | add_command(&length_cmd); |
1055 | e3aff4f6 | aliguori | add_command(&info_cmd); |
1056 | e3aff4f6 | aliguori | add_command(&alloc_cmd); |
1057 | e3aff4f6 | aliguori | |
1058 | e3aff4f6 | aliguori | add_args_command(init_args_command); |
1059 | e3aff4f6 | aliguori | add_check_command(init_check_command); |
1060 | e3aff4f6 | aliguori | |
1061 | e3aff4f6 | aliguori | /* open the device */
|
1062 | e3aff4f6 | aliguori | if (readonly)
|
1063 | e3aff4f6 | aliguori | flags |= BDRV_O_RDONLY; |
1064 | e3aff4f6 | aliguori | else
|
1065 | e3aff4f6 | aliguori | flags |= BDRV_O_RDWR; |
1066 | e3aff4f6 | aliguori | |
1067 | e3aff4f6 | aliguori | if ((argc - optind) == 1) |
1068 | e3aff4f6 | aliguori | openfile(argv[optind], flags); |
1069 | e3aff4f6 | aliguori | command_loop(); |
1070 | e3aff4f6 | aliguori | |
1071 | e3aff4f6 | aliguori | if (bs)
|
1072 | e3aff4f6 | aliguori | bdrv_close(bs); |
1073 | e3aff4f6 | aliguori | return 0; |
1074 | e3aff4f6 | aliguori | } |