]>
Commit | Line | Data |
---|---|---|
6a143727 KW |
1 | /* |
2 | * Block protocol for I/O error injection | |
3 | * | |
4 | * Copyright (c) 2010 Kevin Wolf <[email protected]> | |
5 | * | |
6 | * Permission is hereby granted, free of charge, to any person obtaining a copy | |
7 | * of this software and associated documentation files (the "Software"), to deal | |
8 | * in the Software without restriction, including without limitation the rights | |
9 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
10 | * copies of the Software, and to permit persons to whom the Software is | |
11 | * furnished to do so, subject to the following conditions: | |
12 | * | |
13 | * The above copyright notice and this permission notice shall be included in | |
14 | * all copies or substantial portions of the Software. | |
15 | * | |
16 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
17 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
18 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL | |
19 | * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
20 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
21 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | |
22 | * THE SOFTWARE. | |
23 | */ | |
24 | ||
25 | #include "qemu-common.h" | |
1de7afc9 | 26 | #include "qemu/config-file.h" |
737e150e | 27 | #include "block/block_int.h" |
1de7afc9 | 28 | #include "qemu/module.h" |
2c31b04c HR |
29 | #include "qapi/qmp/qbool.h" |
30 | #include "qapi/qmp/qdict.h" | |
31 | #include "qapi/qmp/qint.h" | |
32 | #include "qapi/qmp/qstring.h" | |
6a143727 KW |
33 | |
34 | typedef struct BDRVBlkdebugState { | |
571cd43e | 35 | int state; |
8f96b5be | 36 | int new_state; |
3c90c65d | 37 | |
571cd43e PB |
38 | QLIST_HEAD(, BlkdebugRule) rules[BLKDBG_EVENT_MAX]; |
39 | QSIMPLEQ_HEAD(, BlkdebugRule) active_rules; | |
3c90c65d | 40 | QLIST_HEAD(, BlkdebugSuspendedReq) suspended_reqs; |
6a143727 KW |
41 | } BDRVBlkdebugState; |
42 | ||
b9f66d96 | 43 | typedef struct BlkdebugAIOCB { |
7c84b1b8 | 44 | BlockAIOCB common; |
b9f66d96 KW |
45 | QEMUBH *bh; |
46 | int ret; | |
47 | } BlkdebugAIOCB; | |
48 | ||
3c90c65d KW |
49 | typedef struct BlkdebugSuspendedReq { |
50 | Coroutine *co; | |
51 | char *tag; | |
52 | QLIST_ENTRY(BlkdebugSuspendedReq) next; | |
53 | } BlkdebugSuspendedReq; | |
54 | ||
d7331bed | 55 | static const AIOCBInfo blkdebug_aiocb_info = { |
4c781717 | 56 | .aiocb_size = sizeof(BlkdebugAIOCB), |
b9f66d96 KW |
57 | }; |
58 | ||
8b9b0cc2 KW |
59 | enum { |
60 | ACTION_INJECT_ERROR, | |
61 | ACTION_SET_STATE, | |
3c90c65d | 62 | ACTION_SUSPEND, |
8b9b0cc2 KW |
63 | }; |
64 | ||
65 | typedef struct BlkdebugRule { | |
66 | BlkDebugEvent event; | |
67 | int action; | |
68 | int state; | |
69 | union { | |
70 | struct { | |
71 | int error; | |
72 | int immediately; | |
73 | int once; | |
e4780db4 | 74 | int64_t sector; |
8b9b0cc2 KW |
75 | } inject; |
76 | struct { | |
77 | int new_state; | |
78 | } set_state; | |
3c90c65d KW |
79 | struct { |
80 | char *tag; | |
81 | } suspend; | |
8b9b0cc2 KW |
82 | } options; |
83 | QLIST_ENTRY(BlkdebugRule) next; | |
571cd43e | 84 | QSIMPLEQ_ENTRY(BlkdebugRule) active_next; |
8b9b0cc2 KW |
85 | } BlkdebugRule; |
86 | ||
87 | static QemuOptsList inject_error_opts = { | |
88 | .name = "inject-error", | |
89 | .head = QTAILQ_HEAD_INITIALIZER(inject_error_opts.head), | |
90 | .desc = { | |
91 | { | |
92 | .name = "event", | |
93 | .type = QEMU_OPT_STRING, | |
94 | }, | |
95 | { | |
96 | .name = "state", | |
97 | .type = QEMU_OPT_NUMBER, | |
98 | }, | |
99 | { | |
100 | .name = "errno", | |
101 | .type = QEMU_OPT_NUMBER, | |
102 | }, | |
e4780db4 PB |
103 | { |
104 | .name = "sector", | |
105 | .type = QEMU_OPT_NUMBER, | |
106 | }, | |
8b9b0cc2 KW |
107 | { |
108 | .name = "once", | |
109 | .type = QEMU_OPT_BOOL, | |
110 | }, | |
111 | { | |
112 | .name = "immediately", | |
113 | .type = QEMU_OPT_BOOL, | |
114 | }, | |
115 | { /* end of list */ } | |
116 | }, | |
117 | }; | |
118 | ||
119 | static QemuOptsList set_state_opts = { | |
120 | .name = "set-state", | |
327cdad4 | 121 | .head = QTAILQ_HEAD_INITIALIZER(set_state_opts.head), |
8b9b0cc2 KW |
122 | .desc = { |
123 | { | |
124 | .name = "event", | |
125 | .type = QEMU_OPT_STRING, | |
126 | }, | |
127 | { | |
128 | .name = "state", | |
129 | .type = QEMU_OPT_NUMBER, | |
130 | }, | |
131 | { | |
132 | .name = "new_state", | |
133 | .type = QEMU_OPT_NUMBER, | |
134 | }, | |
135 | { /* end of list */ } | |
136 | }, | |
137 | }; | |
138 | ||
139 | static QemuOptsList *config_groups[] = { | |
140 | &inject_error_opts, | |
141 | &set_state_opts, | |
142 | NULL | |
143 | }; | |
144 | ||
145 | static const char *event_names[BLKDBG_EVENT_MAX] = { | |
8252278a KW |
146 | [BLKDBG_L1_UPDATE] = "l1_update", |
147 | [BLKDBG_L1_GROW_ALLOC_TABLE] = "l1_grow.alloc_table", | |
148 | [BLKDBG_L1_GROW_WRITE_TABLE] = "l1_grow.write_table", | |
149 | [BLKDBG_L1_GROW_ACTIVATE_TABLE] = "l1_grow.activate_table", | |
150 | ||
151 | [BLKDBG_L2_LOAD] = "l2_load", | |
152 | [BLKDBG_L2_UPDATE] = "l2_update", | |
153 | [BLKDBG_L2_UPDATE_COMPRESSED] = "l2_update_compressed", | |
154 | [BLKDBG_L2_ALLOC_COW_READ] = "l2_alloc.cow_read", | |
155 | [BLKDBG_L2_ALLOC_WRITE] = "l2_alloc.write", | |
156 | ||
8252278a | 157 | [BLKDBG_READ_AIO] = "read_aio", |
8252278a KW |
158 | [BLKDBG_READ_BACKING_AIO] = "read_backing_aio", |
159 | [BLKDBG_READ_COMPRESSED] = "read_compressed", | |
160 | ||
161 | [BLKDBG_WRITE_AIO] = "write_aio", | |
162 | [BLKDBG_WRITE_COMPRESSED] = "write_compressed", | |
163 | ||
164 | [BLKDBG_VMSTATE_LOAD] = "vmstate_load", | |
165 | [BLKDBG_VMSTATE_SAVE] = "vmstate_save", | |
166 | ||
167 | [BLKDBG_COW_READ] = "cow_read", | |
168 | [BLKDBG_COW_WRITE] = "cow_write", | |
169 | ||
170 | [BLKDBG_REFTABLE_LOAD] = "reftable_load", | |
171 | [BLKDBG_REFTABLE_GROW] = "reftable_grow", | |
afa50193 | 172 | [BLKDBG_REFTABLE_UPDATE] = "reftable_update", |
8252278a KW |
173 | |
174 | [BLKDBG_REFBLOCK_LOAD] = "refblock_load", | |
175 | [BLKDBG_REFBLOCK_UPDATE] = "refblock_update", | |
176 | [BLKDBG_REFBLOCK_UPDATE_PART] = "refblock_update_part", | |
177 | [BLKDBG_REFBLOCK_ALLOC] = "refblock_alloc", | |
178 | [BLKDBG_REFBLOCK_ALLOC_HOOKUP] = "refblock_alloc.hookup", | |
179 | [BLKDBG_REFBLOCK_ALLOC_WRITE] = "refblock_alloc.write", | |
180 | [BLKDBG_REFBLOCK_ALLOC_WRITE_BLOCKS] = "refblock_alloc.write_blocks", | |
181 | [BLKDBG_REFBLOCK_ALLOC_WRITE_TABLE] = "refblock_alloc.write_table", | |
182 | [BLKDBG_REFBLOCK_ALLOC_SWITCH_TABLE] = "refblock_alloc.switch_table", | |
183 | ||
184 | [BLKDBG_CLUSTER_ALLOC] = "cluster_alloc", | |
185 | [BLKDBG_CLUSTER_ALLOC_BYTES] = "cluster_alloc_bytes", | |
186 | [BLKDBG_CLUSTER_FREE] = "cluster_free", | |
bf736fe3 KW |
187 | |
188 | [BLKDBG_FLUSH_TO_OS] = "flush_to_os", | |
189 | [BLKDBG_FLUSH_TO_DISK] = "flush_to_disk", | |
9e1cb96d KW |
190 | |
191 | [BLKDBG_PWRITEV_RMW_HEAD] = "pwritev_rmw.head", | |
192 | [BLKDBG_PWRITEV_RMW_AFTER_HEAD] = "pwritev_rmw.after_head", | |
193 | [BLKDBG_PWRITEV_RMW_TAIL] = "pwritev_rmw.tail", | |
194 | [BLKDBG_PWRITEV_RMW_AFTER_TAIL] = "pwritev_rmw.after_tail", | |
195 | [BLKDBG_PWRITEV] = "pwritev", | |
196 | [BLKDBG_PWRITEV_ZERO] = "pwritev_zero", | |
197 | [BLKDBG_PWRITEV_DONE] = "pwritev_done", | |
94054183 HR |
198 | |
199 | [BLKDBG_EMPTY_IMAGE_PREPARE] = "empty_image_prepare", | |
8b9b0cc2 KW |
200 | }; |
201 | ||
202 | static int get_event_by_name(const char *name, BlkDebugEvent *event) | |
203 | { | |
204 | int i; | |
205 | ||
206 | for (i = 0; i < BLKDBG_EVENT_MAX; i++) { | |
207 | if (!strcmp(event_names[i], name)) { | |
208 | *event = i; | |
209 | return 0; | |
210 | } | |
211 | } | |
212 | ||
213 | return -1; | |
214 | } | |
215 | ||
216 | struct add_rule_data { | |
217 | BDRVBlkdebugState *s; | |
218 | int action; | |
d4362d64 | 219 | Error **errp; |
8b9b0cc2 KW |
220 | }; |
221 | ||
28d0de7a | 222 | static int add_rule(void *opaque, QemuOpts *opts, Error **errp) |
8b9b0cc2 KW |
223 | { |
224 | struct add_rule_data *d = opaque; | |
225 | BDRVBlkdebugState *s = d->s; | |
226 | const char* event_name; | |
227 | BlkDebugEvent event; | |
228 | struct BlkdebugRule *rule; | |
229 | ||
230 | /* Find the right event for the rule */ | |
231 | event_name = qemu_opt_get(opts, "event"); | |
d4362d64 SH |
232 | if (!event_name) { |
233 | error_setg(d->errp, "Missing event name for rule"); | |
234 | return -1; | |
235 | } else if (get_event_by_name(event_name, &event) < 0) { | |
236 | error_setg(d->errp, "Invalid event name \"%s\"", event_name); | |
8b9b0cc2 KW |
237 | return -1; |
238 | } | |
239 | ||
240 | /* Set attributes common for all actions */ | |
7267c094 | 241 | rule = g_malloc0(sizeof(*rule)); |
8b9b0cc2 KW |
242 | *rule = (struct BlkdebugRule) { |
243 | .event = event, | |
244 | .action = d->action, | |
245 | .state = qemu_opt_get_number(opts, "state", 0), | |
246 | }; | |
247 | ||
248 | /* Parse action-specific options */ | |
249 | switch (d->action) { | |
250 | case ACTION_INJECT_ERROR: | |
251 | rule->options.inject.error = qemu_opt_get_number(opts, "errno", EIO); | |
252 | rule->options.inject.once = qemu_opt_get_bool(opts, "once", 0); | |
253 | rule->options.inject.immediately = | |
254 | qemu_opt_get_bool(opts, "immediately", 0); | |
e4780db4 | 255 | rule->options.inject.sector = qemu_opt_get_number(opts, "sector", -1); |
8b9b0cc2 KW |
256 | break; |
257 | ||
258 | case ACTION_SET_STATE: | |
259 | rule->options.set_state.new_state = | |
260 | qemu_opt_get_number(opts, "new_state", 0); | |
261 | break; | |
3c90c65d KW |
262 | |
263 | case ACTION_SUSPEND: | |
264 | rule->options.suspend.tag = | |
265 | g_strdup(qemu_opt_get(opts, "tag")); | |
266 | break; | |
8b9b0cc2 KW |
267 | }; |
268 | ||
269 | /* Add the rule */ | |
270 | QLIST_INSERT_HEAD(&s->rules[event], rule, next); | |
271 | ||
272 | return 0; | |
273 | } | |
274 | ||
9e35542b KW |
275 | static void remove_rule(BlkdebugRule *rule) |
276 | { | |
277 | switch (rule->action) { | |
278 | case ACTION_INJECT_ERROR: | |
279 | case ACTION_SET_STATE: | |
280 | break; | |
3c90c65d KW |
281 | case ACTION_SUSPEND: |
282 | g_free(rule->options.suspend.tag); | |
283 | break; | |
9e35542b KW |
284 | } |
285 | ||
286 | QLIST_REMOVE(rule, next); | |
287 | g_free(rule); | |
288 | } | |
289 | ||
89f2b21e HR |
290 | static int read_config(BDRVBlkdebugState *s, const char *filename, |
291 | QDict *options, Error **errp) | |
8b9b0cc2 | 292 | { |
85a040e5 | 293 | FILE *f = NULL; |
8b9b0cc2 KW |
294 | int ret; |
295 | struct add_rule_data d; | |
89f2b21e | 296 | Error *local_err = NULL; |
8b9b0cc2 | 297 | |
85a040e5 HR |
298 | if (filename) { |
299 | f = fopen(filename, "r"); | |
300 | if (f == NULL) { | |
301 | error_setg_errno(errp, errno, "Could not read blkdebug config file"); | |
302 | return -errno; | |
303 | } | |
8b9b0cc2 | 304 | |
85a040e5 HR |
305 | ret = qemu_config_parse(f, config_groups, filename); |
306 | if (ret < 0) { | |
307 | error_setg(errp, "Could not parse blkdebug config file"); | |
308 | ret = -EINVAL; | |
309 | goto fail; | |
310 | } | |
8b9b0cc2 KW |
311 | } |
312 | ||
89f2b21e | 313 | qemu_config_parse_qdict(options, config_groups, &local_err); |
84d18f06 | 314 | if (local_err) { |
89f2b21e HR |
315 | error_propagate(errp, local_err); |
316 | ret = -EINVAL; | |
317 | goto fail; | |
318 | } | |
319 | ||
8b9b0cc2 KW |
320 | d.s = s; |
321 | d.action = ACTION_INJECT_ERROR; | |
d4362d64 | 322 | d.errp = &local_err; |
28d0de7a | 323 | qemu_opts_foreach(&inject_error_opts, add_rule, &d, NULL); |
d4362d64 SH |
324 | if (local_err) { |
325 | error_propagate(errp, local_err); | |
326 | ret = -EINVAL; | |
327 | goto fail; | |
328 | } | |
8b9b0cc2 KW |
329 | |
330 | d.action = ACTION_SET_STATE; | |
28d0de7a | 331 | qemu_opts_foreach(&set_state_opts, add_rule, &d, NULL); |
d4362d64 SH |
332 | if (local_err) { |
333 | error_propagate(errp, local_err); | |
334 | ret = -EINVAL; | |
335 | goto fail; | |
336 | } | |
8b9b0cc2 KW |
337 | |
338 | ret = 0; | |
339 | fail: | |
698f0d52 KW |
340 | qemu_opts_reset(&inject_error_opts); |
341 | qemu_opts_reset(&set_state_opts); | |
85a040e5 HR |
342 | if (f) { |
343 | fclose(f); | |
344 | } | |
8b9b0cc2 KW |
345 | return ret; |
346 | } | |
347 | ||
348 | /* Valid blkdebug filenames look like blkdebug:path/to/config:path/to/image */ | |
f4681212 KW |
349 | static void blkdebug_parse_filename(const char *filename, QDict *options, |
350 | Error **errp) | |
6a143727 | 351 | { |
f4681212 | 352 | const char *c; |
6a143727 | 353 | |
8b9b0cc2 | 354 | /* Parse the blkdebug: prefix */ |
f4681212 | 355 | if (!strstart(filename, "blkdebug:", &filename)) { |
d4881b9b HR |
356 | /* There was no prefix; therefore, all options have to be already |
357 | present in the QDict (except for the filename) */ | |
358 | qdict_put(options, "x-image", qstring_from_str(filename)); | |
f4681212 | 359 | return; |
6a143727 | 360 | } |
6a143727 | 361 | |
f4681212 | 362 | /* Parse config file path */ |
8b9b0cc2 KW |
363 | c = strchr(filename, ':'); |
364 | if (c == NULL) { | |
f4681212 KW |
365 | error_setg(errp, "blkdebug requires both config file and image path"); |
366 | return; | |
8b9b0cc2 KW |
367 | } |
368 | ||
f4681212 KW |
369 | if (c != filename) { |
370 | QString *config_path; | |
371 | config_path = qstring_from_substr(filename, 0, c - filename - 1); | |
372 | qdict_put(options, "config", config_path); | |
8b9b0cc2 | 373 | } |
f4681212 KW |
374 | |
375 | /* TODO Allow multi-level nesting and set file.filename here */ | |
8b9b0cc2 | 376 | filename = c + 1; |
f4681212 KW |
377 | qdict_put(options, "x-image", qstring_from_str(filename)); |
378 | } | |
379 | ||
380 | static QemuOptsList runtime_opts = { | |
381 | .name = "blkdebug", | |
382 | .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head), | |
383 | .desc = { | |
384 | { | |
385 | .name = "config", | |
386 | .type = QEMU_OPT_STRING, | |
387 | .help = "Path to the configuration file", | |
388 | }, | |
389 | { | |
390 | .name = "x-image", | |
391 | .type = QEMU_OPT_STRING, | |
392 | .help = "[internal use only, will be removed]", | |
393 | }, | |
b35ee7fb KW |
394 | { |
395 | .name = "align", | |
396 | .type = QEMU_OPT_SIZE, | |
397 | .help = "Required alignment in bytes", | |
398 | }, | |
f4681212 KW |
399 | { /* end of list */ } |
400 | }, | |
401 | }; | |
402 | ||
015a1036 HR |
403 | static int blkdebug_open(BlockDriverState *bs, QDict *options, int flags, |
404 | Error **errp) | |
f4681212 KW |
405 | { |
406 | BDRVBlkdebugState *s = bs->opaque; | |
407 | QemuOpts *opts; | |
408 | Error *local_err = NULL; | |
4373593d | 409 | const char *config; |
b35ee7fb | 410 | uint64_t align; |
f4681212 KW |
411 | int ret; |
412 | ||
87ea75d5 | 413 | opts = qemu_opts_create(&runtime_opts, NULL, 0, &error_abort); |
f4681212 | 414 | qemu_opts_absorb_qdict(opts, options, &local_err); |
84d18f06 | 415 | if (local_err) { |
10ffa72f | 416 | error_propagate(errp, local_err); |
f4681212 | 417 | ret = -EINVAL; |
eaf944a4 | 418 | goto out; |
f4681212 KW |
419 | } |
420 | ||
89f2b21e | 421 | /* Read rules from config file or command line options */ |
f4681212 | 422 | config = qemu_opt_get(opts, "config"); |
89f2b21e | 423 | ret = read_config(s, config, options, errp); |
85a040e5 | 424 | if (ret) { |
eaf944a4 | 425 | goto out; |
f4681212 | 426 | } |
8b9b0cc2 | 427 | |
8db520ce | 428 | /* Set initial state */ |
571cd43e | 429 | s->state = 1; |
8db520ce | 430 | |
8b9b0cc2 | 431 | /* Open the backing file */ |
f67503e5 | 432 | assert(bs->file == NULL); |
4373593d | 433 | ret = bdrv_open_image(&bs->file, qemu_opt_get(opts, "x-image"), options, "image", |
f7d9fd8c | 434 | flags | BDRV_O_PROTOCOL, false, &local_err); |
8b9b0cc2 | 435 | if (ret < 0) { |
10ffa72f | 436 | error_propagate(errp, local_err); |
eaf944a4 | 437 | goto out; |
8b9b0cc2 KW |
438 | } |
439 | ||
b35ee7fb KW |
440 | /* Set request alignment */ |
441 | align = qemu_opt_get_size(opts, "align", bs->request_alignment); | |
442 | if (align > 0 && align < INT_MAX && !(align & (align - 1))) { | |
443 | bs->request_alignment = align; | |
444 | } else { | |
445 | error_setg(errp, "Invalid alignment"); | |
446 | ret = -EINVAL; | |
eaf944a4 | 447 | goto fail_unref; |
b35ee7fb KW |
448 | } |
449 | ||
f4681212 | 450 | ret = 0; |
eaf944a4 KW |
451 | goto out; |
452 | ||
453 | fail_unref: | |
454 | bdrv_unref(bs->file); | |
455 | out: | |
f4681212 KW |
456 | qemu_opts_del(opts); |
457 | return ret; | |
6a143727 KW |
458 | } |
459 | ||
b9f66d96 KW |
460 | static void error_callback_bh(void *opaque) |
461 | { | |
462 | struct BlkdebugAIOCB *acb = opaque; | |
463 | qemu_bh_delete(acb->bh); | |
464 | acb->common.cb(acb->common.opaque, acb->ret); | |
8007429a | 465 | qemu_aio_unref(acb); |
b9f66d96 KW |
466 | } |
467 | ||
7c84b1b8 | 468 | static BlockAIOCB *inject_error(BlockDriverState *bs, |
097310b5 | 469 | BlockCompletionFunc *cb, void *opaque, BlkdebugRule *rule) |
b9f66d96 KW |
470 | { |
471 | BDRVBlkdebugState *s = bs->opaque; | |
571cd43e | 472 | int error = rule->options.inject.error; |
b9f66d96 KW |
473 | struct BlkdebugAIOCB *acb; |
474 | QEMUBH *bh; | |
a069e2f1 | 475 | bool immediately = rule->options.inject.immediately; |
b9f66d96 | 476 | |
571cd43e | 477 | if (rule->options.inject.once) { |
a069e2f1 JS |
478 | QSIMPLEQ_REMOVE(&s->active_rules, rule, BlkdebugRule, active_next); |
479 | remove_rule(rule); | |
b9f66d96 KW |
480 | } |
481 | ||
a069e2f1 | 482 | if (immediately) { |
b9f66d96 KW |
483 | return NULL; |
484 | } | |
485 | ||
d7331bed | 486 | acb = qemu_aio_get(&blkdebug_aiocb_info, bs, cb, opaque); |
b9f66d96 KW |
487 | acb->ret = -error; |
488 | ||
7e1efdf0 | 489 | bh = aio_bh_new(bdrv_get_aio_context(bs), error_callback_bh, acb); |
b9f66d96 KW |
490 | acb->bh = bh; |
491 | qemu_bh_schedule(bh); | |
492 | ||
b666d239 | 493 | return &acb->common; |
b9f66d96 KW |
494 | } |
495 | ||
7c84b1b8 | 496 | static BlockAIOCB *blkdebug_aio_readv(BlockDriverState *bs, |
6a143727 | 497 | int64_t sector_num, QEMUIOVector *qiov, int nb_sectors, |
097310b5 | 498 | BlockCompletionFunc *cb, void *opaque) |
6a143727 KW |
499 | { |
500 | BDRVBlkdebugState *s = bs->opaque; | |
e4780db4 PB |
501 | BlkdebugRule *rule = NULL; |
502 | ||
503 | QSIMPLEQ_FOREACH(rule, &s->active_rules, active_next) { | |
504 | if (rule->options.inject.sector == -1 || | |
505 | (rule->options.inject.sector >= sector_num && | |
506 | rule->options.inject.sector < sector_num + nb_sectors)) { | |
507 | break; | |
508 | } | |
509 | } | |
b9f66d96 | 510 | |
571cd43e PB |
511 | if (rule && rule->options.inject.error) { |
512 | return inject_error(bs, cb, opaque, rule); | |
b9f66d96 KW |
513 | } |
514 | ||
368e8dd1 | 515 | return bdrv_aio_readv(bs->file, sector_num, qiov, nb_sectors, cb, opaque); |
6a143727 KW |
516 | } |
517 | ||
7c84b1b8 | 518 | static BlockAIOCB *blkdebug_aio_writev(BlockDriverState *bs, |
6a143727 | 519 | int64_t sector_num, QEMUIOVector *qiov, int nb_sectors, |
097310b5 | 520 | BlockCompletionFunc *cb, void *opaque) |
6a143727 KW |
521 | { |
522 | BDRVBlkdebugState *s = bs->opaque; | |
e4780db4 PB |
523 | BlkdebugRule *rule = NULL; |
524 | ||
525 | QSIMPLEQ_FOREACH(rule, &s->active_rules, active_next) { | |
526 | if (rule->options.inject.sector == -1 || | |
527 | (rule->options.inject.sector >= sector_num && | |
528 | rule->options.inject.sector < sector_num + nb_sectors)) { | |
529 | break; | |
530 | } | |
531 | } | |
b9f66d96 | 532 | |
571cd43e PB |
533 | if (rule && rule->options.inject.error) { |
534 | return inject_error(bs, cb, opaque, rule); | |
b9f66d96 KW |
535 | } |
536 | ||
368e8dd1 | 537 | return bdrv_aio_writev(bs->file, sector_num, qiov, nb_sectors, cb, opaque); |
6a143727 KW |
538 | } |
539 | ||
7c84b1b8 | 540 | static BlockAIOCB *blkdebug_aio_flush(BlockDriverState *bs, |
097310b5 | 541 | BlockCompletionFunc *cb, void *opaque) |
9e52c53b PB |
542 | { |
543 | BDRVBlkdebugState *s = bs->opaque; | |
544 | BlkdebugRule *rule = NULL; | |
545 | ||
546 | QSIMPLEQ_FOREACH(rule, &s->active_rules, active_next) { | |
547 | if (rule->options.inject.sector == -1) { | |
548 | break; | |
549 | } | |
550 | } | |
551 | ||
552 | if (rule && rule->options.inject.error) { | |
553 | return inject_error(bs, cb, opaque, rule); | |
554 | } | |
555 | ||
556 | return bdrv_aio_flush(bs->file, cb, opaque); | |
557 | } | |
558 | ||
3c90c65d | 559 | |
6a143727 KW |
560 | static void blkdebug_close(BlockDriverState *bs) |
561 | { | |
562 | BDRVBlkdebugState *s = bs->opaque; | |
8b9b0cc2 KW |
563 | BlkdebugRule *rule, *next; |
564 | int i; | |
565 | ||
566 | for (i = 0; i < BLKDBG_EVENT_MAX; i++) { | |
567 | QLIST_FOREACH_SAFE(rule, &s->rules[i], next, next) { | |
9e35542b | 568 | remove_rule(rule); |
8b9b0cc2 KW |
569 | } |
570 | } | |
6a143727 KW |
571 | } |
572 | ||
3c90c65d KW |
573 | static void suspend_request(BlockDriverState *bs, BlkdebugRule *rule) |
574 | { | |
575 | BDRVBlkdebugState *s = bs->opaque; | |
576 | BlkdebugSuspendedReq r; | |
577 | ||
578 | r = (BlkdebugSuspendedReq) { | |
579 | .co = qemu_coroutine_self(), | |
580 | .tag = g_strdup(rule->options.suspend.tag), | |
581 | }; | |
582 | ||
583 | remove_rule(rule); | |
584 | QLIST_INSERT_HEAD(&s->suspended_reqs, &r, next); | |
585 | ||
586 | printf("blkdebug: Suspended request '%s'\n", r.tag); | |
587 | qemu_coroutine_yield(); | |
588 | printf("blkdebug: Resuming request '%s'\n", r.tag); | |
589 | ||
590 | QLIST_REMOVE(&r, next); | |
591 | g_free(r.tag); | |
592 | } | |
593 | ||
571cd43e | 594 | static bool process_rule(BlockDriverState *bs, struct BlkdebugRule *rule, |
8f96b5be | 595 | bool injected) |
8b9b0cc2 KW |
596 | { |
597 | BDRVBlkdebugState *s = bs->opaque; | |
8b9b0cc2 KW |
598 | |
599 | /* Only process rules for the current state */ | |
8f96b5be | 600 | if (rule->state && rule->state != s->state) { |
571cd43e | 601 | return injected; |
8b9b0cc2 KW |
602 | } |
603 | ||
604 | /* Take the action */ | |
605 | switch (rule->action) { | |
606 | case ACTION_INJECT_ERROR: | |
571cd43e PB |
607 | if (!injected) { |
608 | QSIMPLEQ_INIT(&s->active_rules); | |
609 | injected = true; | |
610 | } | |
611 | QSIMPLEQ_INSERT_HEAD(&s->active_rules, rule, active_next); | |
8b9b0cc2 KW |
612 | break; |
613 | ||
614 | case ACTION_SET_STATE: | |
8f96b5be | 615 | s->new_state = rule->options.set_state.new_state; |
8b9b0cc2 | 616 | break; |
3c90c65d KW |
617 | |
618 | case ACTION_SUSPEND: | |
619 | suspend_request(bs, rule); | |
620 | break; | |
8b9b0cc2 | 621 | } |
571cd43e | 622 | return injected; |
8b9b0cc2 KW |
623 | } |
624 | ||
625 | static void blkdebug_debug_event(BlockDriverState *bs, BlkDebugEvent event) | |
626 | { | |
627 | BDRVBlkdebugState *s = bs->opaque; | |
3c90c65d | 628 | struct BlkdebugRule *rule, *next; |
571cd43e | 629 | bool injected; |
8b9b0cc2 | 630 | |
95ee3914 | 631 | assert((int)event >= 0 && event < BLKDBG_EVENT_MAX); |
8b9b0cc2 | 632 | |
571cd43e | 633 | injected = false; |
8f96b5be | 634 | s->new_state = s->state; |
3c90c65d | 635 | QLIST_FOREACH_SAFE(rule, &s->rules[event], next, next) { |
8f96b5be | 636 | injected = process_rule(bs, rule, injected); |
8b9b0cc2 | 637 | } |
8f96b5be | 638 | s->state = s->new_state; |
8b9b0cc2 KW |
639 | } |
640 | ||
3c90c65d KW |
641 | static int blkdebug_debug_breakpoint(BlockDriverState *bs, const char *event, |
642 | const char *tag) | |
643 | { | |
644 | BDRVBlkdebugState *s = bs->opaque; | |
645 | struct BlkdebugRule *rule; | |
646 | BlkDebugEvent blkdebug_event; | |
647 | ||
648 | if (get_event_by_name(event, &blkdebug_event) < 0) { | |
649 | return -ENOENT; | |
650 | } | |
651 | ||
652 | ||
653 | rule = g_malloc(sizeof(*rule)); | |
654 | *rule = (struct BlkdebugRule) { | |
655 | .event = blkdebug_event, | |
656 | .action = ACTION_SUSPEND, | |
657 | .state = 0, | |
658 | .options.suspend.tag = g_strdup(tag), | |
659 | }; | |
660 | ||
661 | QLIST_INSERT_HEAD(&s->rules[blkdebug_event], rule, next); | |
662 | ||
663 | return 0; | |
664 | } | |
665 | ||
666 | static int blkdebug_debug_resume(BlockDriverState *bs, const char *tag) | |
667 | { | |
668 | BDRVBlkdebugState *s = bs->opaque; | |
c547e564 | 669 | BlkdebugSuspendedReq *r, *next; |
3c90c65d | 670 | |
c547e564 | 671 | QLIST_FOREACH_SAFE(r, &s->suspended_reqs, next, next) { |
3c90c65d KW |
672 | if (!strcmp(r->tag, tag)) { |
673 | qemu_coroutine_enter(r->co, NULL); | |
674 | return 0; | |
675 | } | |
676 | } | |
677 | return -ENOENT; | |
678 | } | |
679 | ||
4cc70e93 FZ |
680 | static int blkdebug_debug_remove_breakpoint(BlockDriverState *bs, |
681 | const char *tag) | |
682 | { | |
683 | BDRVBlkdebugState *s = bs->opaque; | |
c547e564 | 684 | BlkdebugSuspendedReq *r, *r_next; |
4cc70e93 FZ |
685 | BlkdebugRule *rule, *next; |
686 | int i, ret = -ENOENT; | |
687 | ||
688 | for (i = 0; i < BLKDBG_EVENT_MAX; i++) { | |
689 | QLIST_FOREACH_SAFE(rule, &s->rules[i], next, next) { | |
690 | if (rule->action == ACTION_SUSPEND && | |
691 | !strcmp(rule->options.suspend.tag, tag)) { | |
692 | remove_rule(rule); | |
693 | ret = 0; | |
694 | } | |
695 | } | |
696 | } | |
c547e564 | 697 | QLIST_FOREACH_SAFE(r, &s->suspended_reqs, next, r_next) { |
4cc70e93 FZ |
698 | if (!strcmp(r->tag, tag)) { |
699 | qemu_coroutine_enter(r->co, NULL); | |
700 | ret = 0; | |
701 | } | |
702 | } | |
703 | return ret; | |
704 | } | |
3c90c65d KW |
705 | |
706 | static bool blkdebug_debug_is_suspended(BlockDriverState *bs, const char *tag) | |
707 | { | |
708 | BDRVBlkdebugState *s = bs->opaque; | |
709 | BlkdebugSuspendedReq *r; | |
710 | ||
711 | QLIST_FOREACH(r, &s->suspended_reqs, next) { | |
712 | if (!strcmp(r->tag, tag)) { | |
713 | return true; | |
714 | } | |
715 | } | |
716 | return false; | |
717 | } | |
718 | ||
e1302255 PB |
719 | static int64_t blkdebug_getlength(BlockDriverState *bs) |
720 | { | |
721 | return bdrv_getlength(bs->file); | |
722 | } | |
723 | ||
8eedfbd4 KW |
724 | static int blkdebug_truncate(BlockDriverState *bs, int64_t offset) |
725 | { | |
726 | return bdrv_truncate(bs->file, offset); | |
727 | } | |
728 | ||
2c31b04c HR |
729 | static void blkdebug_refresh_filename(BlockDriverState *bs) |
730 | { | |
2c31b04c | 731 | QDict *opts; |
8779441b HR |
732 | const QDictEntry *e; |
733 | bool force_json = false; | |
734 | ||
735 | for (e = qdict_first(bs->options); e; e = qdict_next(bs->options, e)) { | |
736 | if (strcmp(qdict_entry_key(e), "config") && | |
737 | strcmp(qdict_entry_key(e), "x-image") && | |
738 | strcmp(qdict_entry_key(e), "image") && | |
739 | strncmp(qdict_entry_key(e), "image.", strlen("image."))) | |
740 | { | |
741 | force_json = true; | |
742 | break; | |
743 | } | |
744 | } | |
2c31b04c | 745 | |
8779441b | 746 | if (force_json && !bs->file->full_open_options) { |
2c31b04c HR |
747 | /* The config file cannot be recreated, so creating a plain filename |
748 | * is impossible */ | |
749 | return; | |
750 | } | |
751 | ||
8779441b HR |
752 | if (!force_json && bs->file->exact_filename[0]) { |
753 | snprintf(bs->exact_filename, sizeof(bs->exact_filename), | |
754 | "blkdebug:%s:%s", | |
755 | qdict_get_try_str(bs->options, "config") ?: "", | |
756 | bs->file->exact_filename); | |
757 | } | |
758 | ||
2c31b04c HR |
759 | opts = qdict_new(); |
760 | qdict_put_obj(opts, "driver", QOBJECT(qstring_from_str("blkdebug"))); | |
761 | ||
762 | QINCREF(bs->file->full_open_options); | |
763 | qdict_put_obj(opts, "image", QOBJECT(bs->file->full_open_options)); | |
764 | ||
8779441b HR |
765 | for (e = qdict_first(bs->options); e; e = qdict_next(bs->options, e)) { |
766 | if (strcmp(qdict_entry_key(e), "x-image") && | |
767 | strcmp(qdict_entry_key(e), "image") && | |
768 | strncmp(qdict_entry_key(e), "image.", strlen("image."))) | |
769 | { | |
770 | qobject_incref(qdict_entry_value(e)); | |
771 | qdict_put_obj(opts, qdict_entry_key(e), qdict_entry_value(e)); | |
2c31b04c HR |
772 | } |
773 | } | |
774 | ||
2c31b04c HR |
775 | bs->full_open_options = opts; |
776 | } | |
777 | ||
6a143727 | 778 | static BlockDriver bdrv_blkdebug = { |
f4681212 KW |
779 | .format_name = "blkdebug", |
780 | .protocol_name = "blkdebug", | |
781 | .instance_size = sizeof(BDRVBlkdebugState), | |
6a143727 | 782 | |
f4681212 KW |
783 | .bdrv_parse_filename = blkdebug_parse_filename, |
784 | .bdrv_file_open = blkdebug_open, | |
785 | .bdrv_close = blkdebug_close, | |
786 | .bdrv_getlength = blkdebug_getlength, | |
8eedfbd4 | 787 | .bdrv_truncate = blkdebug_truncate, |
2c31b04c | 788 | .bdrv_refresh_filename = blkdebug_refresh_filename, |
6a143727 | 789 | |
f4681212 KW |
790 | .bdrv_aio_readv = blkdebug_aio_readv, |
791 | .bdrv_aio_writev = blkdebug_aio_writev, | |
9e52c53b | 792 | .bdrv_aio_flush = blkdebug_aio_flush, |
8b9b0cc2 | 793 | |
3c90c65d KW |
794 | .bdrv_debug_event = blkdebug_debug_event, |
795 | .bdrv_debug_breakpoint = blkdebug_debug_breakpoint, | |
4cc70e93 FZ |
796 | .bdrv_debug_remove_breakpoint |
797 | = blkdebug_debug_remove_breakpoint, | |
3c90c65d KW |
798 | .bdrv_debug_resume = blkdebug_debug_resume, |
799 | .bdrv_debug_is_suspended = blkdebug_debug_is_suspended, | |
6a143727 KW |
800 | }; |
801 | ||
802 | static void bdrv_blkdebug_init(void) | |
803 | { | |
804 | bdrv_register(&bdrv_blkdebug); | |
805 | } | |
806 | ||
807 | block_init(bdrv_blkdebug_init); |