]> git.proxmox.com Git - mirror_qemu.git/blame - block/blkdebug.c
qapi: Change data type of the FOO_lookup generated for enum FOO
[mirror_qemu.git] / block / blkdebug.c
CommitLineData
6a143727
KW
1/*
2 * Block protocol for I/O error injection
3 *
63188c24 4 * Copyright (C) 2016-2017 Red Hat, Inc.
6a143727
KW
5 * Copyright (c) 2010 Kevin Wolf <kwolf@redhat.com>
6 *
7 * Permission is hereby granted, free of charge, to any person obtaining a copy
8 * of this software and associated documentation files (the "Software"), to deal
9 * in the Software without restriction, including without limitation the rights
10 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11 * copies of the Software, and to permit persons to whom the Software is
12 * furnished to do so, subject to the following conditions:
13 *
14 * The above copyright notice and this permission notice shall be included in
15 * all copies or substantial portions of the Software.
16 *
17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
20 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
23 * THE SOFTWARE.
24 */
25
80c71a24 26#include "qemu/osdep.h"
da34e65c 27#include "qapi/error.h"
f348b6d1 28#include "qemu/cutils.h"
1de7afc9 29#include "qemu/config-file.h"
737e150e 30#include "block/block_int.h"
1de7afc9 31#include "qemu/module.h"
2c31b04c
HR
32#include "qapi/qmp/qbool.h"
33#include "qapi/qmp/qdict.h"
2c31b04c 34#include "qapi/qmp/qstring.h"
20873526 35#include "sysemu/qtest.h"
6a143727
KW
36
37typedef struct BDRVBlkdebugState {
571cd43e 38 int state;
8f96b5be 39 int new_state;
3dc834f8 40 uint64_t align;
430b26a8
EB
41 uint64_t max_transfer;
42 uint64_t opt_write_zero;
43 uint64_t max_write_zero;
44 uint64_t opt_discard;
45 uint64_t max_discard;
3c90c65d 46
036990d7
HR
47 /* For blkdebug_refresh_filename() */
48 char *config_file;
49
7fb1cf16 50 QLIST_HEAD(, BlkdebugRule) rules[BLKDBG__MAX];
571cd43e 51 QSIMPLEQ_HEAD(, BlkdebugRule) active_rules;
3c90c65d 52 QLIST_HEAD(, BlkdebugSuspendedReq) suspended_reqs;
6a143727
KW
53} BDRVBlkdebugState;
54
b9f66d96 55typedef struct BlkdebugAIOCB {
7c84b1b8 56 BlockAIOCB common;
b9f66d96
KW
57 int ret;
58} BlkdebugAIOCB;
59
3c90c65d
KW
60typedef struct BlkdebugSuspendedReq {
61 Coroutine *co;
62 char *tag;
63 QLIST_ENTRY(BlkdebugSuspendedReq) next;
64} BlkdebugSuspendedReq;
65
8b9b0cc2
KW
66enum {
67 ACTION_INJECT_ERROR,
68 ACTION_SET_STATE,
3c90c65d 69 ACTION_SUSPEND,
8b9b0cc2
KW
70};
71
72typedef struct BlkdebugRule {
a31939e6 73 BlkdebugEvent event;
8b9b0cc2
KW
74 int action;
75 int state;
76 union {
77 struct {
78 int error;
79 int immediately;
80 int once;
7c3a9985 81 int64_t offset;
8b9b0cc2
KW
82 } inject;
83 struct {
84 int new_state;
85 } set_state;
3c90c65d
KW
86 struct {
87 char *tag;
88 } suspend;
8b9b0cc2
KW
89 } options;
90 QLIST_ENTRY(BlkdebugRule) next;
571cd43e 91 QSIMPLEQ_ENTRY(BlkdebugRule) active_next;
8b9b0cc2
KW
92} BlkdebugRule;
93
94static QemuOptsList inject_error_opts = {
95 .name = "inject-error",
96 .head = QTAILQ_HEAD_INITIALIZER(inject_error_opts.head),
97 .desc = {
98 {
99 .name = "event",
100 .type = QEMU_OPT_STRING,
101 },
102 {
103 .name = "state",
104 .type = QEMU_OPT_NUMBER,
105 },
106 {
107 .name = "errno",
108 .type = QEMU_OPT_NUMBER,
109 },
e4780db4
PB
110 {
111 .name = "sector",
112 .type = QEMU_OPT_NUMBER,
113 },
8b9b0cc2
KW
114 {
115 .name = "once",
116 .type = QEMU_OPT_BOOL,
117 },
118 {
119 .name = "immediately",
120 .type = QEMU_OPT_BOOL,
121 },
122 { /* end of list */ }
123 },
124};
125
126static QemuOptsList set_state_opts = {
127 .name = "set-state",
327cdad4 128 .head = QTAILQ_HEAD_INITIALIZER(set_state_opts.head),
8b9b0cc2
KW
129 .desc = {
130 {
131 .name = "event",
132 .type = QEMU_OPT_STRING,
133 },
134 {
135 .name = "state",
136 .type = QEMU_OPT_NUMBER,
137 },
138 {
139 .name = "new_state",
140 .type = QEMU_OPT_NUMBER,
141 },
142 { /* end of list */ }
143 },
144};
145
146static QemuOptsList *config_groups[] = {
147 &inject_error_opts,
148 &set_state_opts,
149 NULL
150};
151
8b9b0cc2
KW
152struct add_rule_data {
153 BDRVBlkdebugState *s;
154 int action;
155};
156
28d0de7a 157static int add_rule(void *opaque, QemuOpts *opts, Error **errp)
8b9b0cc2
KW
158{
159 struct add_rule_data *d = opaque;
160 BDRVBlkdebugState *s = d->s;
161 const char* event_name;
f9509d15 162 int event;
8b9b0cc2 163 struct BlkdebugRule *rule;
7c3a9985 164 int64_t sector;
8b9b0cc2
KW
165
166 /* Find the right event for the rule */
167 event_name = qemu_opt_get(opts, "event");
d4362d64 168 if (!event_name) {
8809cfc3 169 error_setg(errp, "Missing event name for rule");
d4362d64 170 return -1;
f9509d15 171 }
f7abe0ec 172 event = qapi_enum_parse(&BlkdebugEvent_lookup, event_name, -1, errp);
f9509d15 173 if (event < 0) {
8b9b0cc2
KW
174 return -1;
175 }
176
177 /* Set attributes common for all actions */
7267c094 178 rule = g_malloc0(sizeof(*rule));
8b9b0cc2
KW
179 *rule = (struct BlkdebugRule) {
180 .event = event,
181 .action = d->action,
182 .state = qemu_opt_get_number(opts, "state", 0),
183 };
184
185 /* Parse action-specific options */
186 switch (d->action) {
187 case ACTION_INJECT_ERROR:
188 rule->options.inject.error = qemu_opt_get_number(opts, "errno", EIO);
189 rule->options.inject.once = qemu_opt_get_bool(opts, "once", 0);
190 rule->options.inject.immediately =
191 qemu_opt_get_bool(opts, "immediately", 0);
7c3a9985
KW
192 sector = qemu_opt_get_number(opts, "sector", -1);
193 rule->options.inject.offset =
194 sector == -1 ? -1 : sector * BDRV_SECTOR_SIZE;
8b9b0cc2
KW
195 break;
196
197 case ACTION_SET_STATE:
198 rule->options.set_state.new_state =
199 qemu_opt_get_number(opts, "new_state", 0);
200 break;
3c90c65d
KW
201
202 case ACTION_SUSPEND:
203 rule->options.suspend.tag =
204 g_strdup(qemu_opt_get(opts, "tag"));
205 break;
8b9b0cc2
KW
206 };
207
208 /* Add the rule */
209 QLIST_INSERT_HEAD(&s->rules[event], rule, next);
210
211 return 0;
212}
213
9e35542b
KW
214static void remove_rule(BlkdebugRule *rule)
215{
216 switch (rule->action) {
217 case ACTION_INJECT_ERROR:
218 case ACTION_SET_STATE:
219 break;
3c90c65d
KW
220 case ACTION_SUSPEND:
221 g_free(rule->options.suspend.tag);
222 break;
9e35542b
KW
223 }
224
225 QLIST_REMOVE(rule, next);
226 g_free(rule);
227}
228
89f2b21e
HR
229static int read_config(BDRVBlkdebugState *s, const char *filename,
230 QDict *options, Error **errp)
8b9b0cc2 231{
85a040e5 232 FILE *f = NULL;
8b9b0cc2
KW
233 int ret;
234 struct add_rule_data d;
89f2b21e 235 Error *local_err = NULL;
8b9b0cc2 236
85a040e5
HR
237 if (filename) {
238 f = fopen(filename, "r");
239 if (f == NULL) {
240 error_setg_errno(errp, errno, "Could not read blkdebug config file");
241 return -errno;
242 }
8b9b0cc2 243
85a040e5
HR
244 ret = qemu_config_parse(f, config_groups, filename);
245 if (ret < 0) {
246 error_setg(errp, "Could not parse blkdebug config file");
247 ret = -EINVAL;
248 goto fail;
249 }
8b9b0cc2
KW
250 }
251
89f2b21e 252 qemu_config_parse_qdict(options, config_groups, &local_err);
84d18f06 253 if (local_err) {
89f2b21e
HR
254 error_propagate(errp, local_err);
255 ret = -EINVAL;
256 goto fail;
257 }
258
8b9b0cc2
KW
259 d.s = s;
260 d.action = ACTION_INJECT_ERROR;
8809cfc3 261 qemu_opts_foreach(&inject_error_opts, add_rule, &d, &local_err);
d4362d64
SH
262 if (local_err) {
263 error_propagate(errp, local_err);
264 ret = -EINVAL;
265 goto fail;
266 }
8b9b0cc2
KW
267
268 d.action = ACTION_SET_STATE;
8809cfc3 269 qemu_opts_foreach(&set_state_opts, add_rule, &d, &local_err);
d4362d64
SH
270 if (local_err) {
271 error_propagate(errp, local_err);
272 ret = -EINVAL;
273 goto fail;
274 }
8b9b0cc2
KW
275
276 ret = 0;
277fail:
698f0d52
KW
278 qemu_opts_reset(&inject_error_opts);
279 qemu_opts_reset(&set_state_opts);
85a040e5
HR
280 if (f) {
281 fclose(f);
282 }
8b9b0cc2
KW
283 return ret;
284}
285
286/* Valid blkdebug filenames look like blkdebug:path/to/config:path/to/image */
f4681212
KW
287static void blkdebug_parse_filename(const char *filename, QDict *options,
288 Error **errp)
6a143727 289{
f4681212 290 const char *c;
6a143727 291
8b9b0cc2 292 /* Parse the blkdebug: prefix */
f4681212 293 if (!strstart(filename, "blkdebug:", &filename)) {
d4881b9b
HR
294 /* There was no prefix; therefore, all options have to be already
295 present in the QDict (except for the filename) */
46f5ac20 296 qdict_put_str(options, "x-image", filename);
f4681212 297 return;
6a143727 298 }
6a143727 299
f4681212 300 /* Parse config file path */
8b9b0cc2
KW
301 c = strchr(filename, ':');
302 if (c == NULL) {
f4681212
KW
303 error_setg(errp, "blkdebug requires both config file and image path");
304 return;
8b9b0cc2
KW
305 }
306
f4681212
KW
307 if (c != filename) {
308 QString *config_path;
309 config_path = qstring_from_substr(filename, 0, c - filename - 1);
310 qdict_put(options, "config", config_path);
8b9b0cc2 311 }
f4681212
KW
312
313 /* TODO Allow multi-level nesting and set file.filename here */
8b9b0cc2 314 filename = c + 1;
46f5ac20 315 qdict_put_str(options, "x-image", filename);
f4681212
KW
316}
317
318static QemuOptsList runtime_opts = {
319 .name = "blkdebug",
320 .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head),
321 .desc = {
322 {
323 .name = "config",
324 .type = QEMU_OPT_STRING,
325 .help = "Path to the configuration file",
326 },
327 {
328 .name = "x-image",
329 .type = QEMU_OPT_STRING,
330 .help = "[internal use only, will be removed]",
331 },
b35ee7fb
KW
332 {
333 .name = "align",
334 .type = QEMU_OPT_SIZE,
335 .help = "Required alignment in bytes",
336 },
430b26a8
EB
337 {
338 .name = "max-transfer",
339 .type = QEMU_OPT_SIZE,
340 .help = "Maximum transfer size in bytes",
341 },
342 {
343 .name = "opt-write-zero",
344 .type = QEMU_OPT_SIZE,
345 .help = "Optimum write zero alignment in bytes",
346 },
347 {
348 .name = "max-write-zero",
349 .type = QEMU_OPT_SIZE,
350 .help = "Maximum write zero size in bytes",
351 },
352 {
353 .name = "opt-discard",
354 .type = QEMU_OPT_SIZE,
355 .help = "Optimum discard alignment in bytes",
356 },
357 {
358 .name = "max-discard",
359 .type = QEMU_OPT_SIZE,
360 .help = "Maximum discard size in bytes",
361 },
f4681212
KW
362 { /* end of list */ }
363 },
364};
365
015a1036
HR
366static int blkdebug_open(BlockDriverState *bs, QDict *options, int flags,
367 Error **errp)
f4681212
KW
368{
369 BDRVBlkdebugState *s = bs->opaque;
370 QemuOpts *opts;
371 Error *local_err = NULL;
f4681212 372 int ret;
430b26a8 373 uint64_t align;
f4681212 374
87ea75d5 375 opts = qemu_opts_create(&runtime_opts, NULL, 0, &error_abort);
f4681212 376 qemu_opts_absorb_qdict(opts, options, &local_err);
84d18f06 377 if (local_err) {
10ffa72f 378 error_propagate(errp, local_err);
f4681212 379 ret = -EINVAL;
eaf944a4 380 goto out;
f4681212
KW
381 }
382
89f2b21e 383 /* Read rules from config file or command line options */
036990d7
HR
384 s->config_file = g_strdup(qemu_opt_get(opts, "config"));
385 ret = read_config(s, s->config_file, options, errp);
85a040e5 386 if (ret) {
eaf944a4 387 goto out;
f4681212 388 }
8b9b0cc2 389
8db520ce 390 /* Set initial state */
571cd43e 391 s->state = 1;
8db520ce 392
6b826af7 393 /* Open the image file */
9a4f4c31
KW
394 bs->file = bdrv_open_child(qemu_opt_get(opts, "x-image"), options, "image",
395 bs, &child_file, false, &local_err);
396 if (local_err) {
397 ret = -EINVAL;
10ffa72f 398 error_propagate(errp, local_err);
eaf944a4 399 goto out;
8b9b0cc2
KW
400 }
401
63188c24
EB
402 bs->supported_write_flags = BDRV_REQ_FUA &
403 bs->file->bs->supported_write_flags;
404 bs->supported_zero_flags = (BDRV_REQ_FUA | BDRV_REQ_MAY_UNMAP) &
405 bs->file->bs->supported_zero_flags;
3dc834f8 406 ret = -EINVAL;
63188c24 407
430b26a8 408 /* Set alignment overrides */
3dc834f8
EB
409 s->align = qemu_opt_get_size(opts, "align", 0);
410 if (s->align && (s->align >= INT_MAX || !is_power_of_2(s->align))) {
411 error_setg(errp, "Cannot meet constraints with align %" PRIu64,
412 s->align);
de234897 413 goto out;
b35ee7fb 414 }
430b26a8
EB
415 align = MAX(s->align, bs->file->bs->bl.request_alignment);
416
417 s->max_transfer = qemu_opt_get_size(opts, "max-transfer", 0);
418 if (s->max_transfer &&
419 (s->max_transfer >= INT_MAX ||
420 !QEMU_IS_ALIGNED(s->max_transfer, align))) {
421 error_setg(errp, "Cannot meet constraints with max-transfer %" PRIu64,
422 s->max_transfer);
423 goto out;
424 }
425
426 s->opt_write_zero = qemu_opt_get_size(opts, "opt-write-zero", 0);
427 if (s->opt_write_zero &&
428 (s->opt_write_zero >= INT_MAX ||
429 !QEMU_IS_ALIGNED(s->opt_write_zero, align))) {
430 error_setg(errp, "Cannot meet constraints with opt-write-zero %" PRIu64,
431 s->opt_write_zero);
432 goto out;
433 }
434
435 s->max_write_zero = qemu_opt_get_size(opts, "max-write-zero", 0);
436 if (s->max_write_zero &&
437 (s->max_write_zero >= INT_MAX ||
438 !QEMU_IS_ALIGNED(s->max_write_zero,
439 MAX(s->opt_write_zero, align)))) {
440 error_setg(errp, "Cannot meet constraints with max-write-zero %" PRIu64,
441 s->max_write_zero);
442 goto out;
443 }
444
445 s->opt_discard = qemu_opt_get_size(opts, "opt-discard", 0);
446 if (s->opt_discard &&
447 (s->opt_discard >= INT_MAX ||
448 !QEMU_IS_ALIGNED(s->opt_discard, align))) {
449 error_setg(errp, "Cannot meet constraints with opt-discard %" PRIu64,
450 s->opt_discard);
451 goto out;
452 }
453
454 s->max_discard = qemu_opt_get_size(opts, "max-discard", 0);
455 if (s->max_discard &&
456 (s->max_discard >= INT_MAX ||
457 !QEMU_IS_ALIGNED(s->max_discard,
458 MAX(s->opt_discard, align)))) {
459 error_setg(errp, "Cannot meet constraints with max-discard %" PRIu64,
460 s->max_discard);
461 goto out;
462 }
b35ee7fb 463
f4681212 464 ret = 0;
eaf944a4 465out:
036990d7
HR
466 if (ret < 0) {
467 g_free(s->config_file);
468 }
f4681212
KW
469 qemu_opts_del(opts);
470 return ret;
6a143727
KW
471}
472
d157ed5f 473static int rule_check(BlockDriverState *bs, uint64_t offset, uint64_t bytes)
b9f66d96
KW
474{
475 BDRVBlkdebugState *s = bs->opaque;
d157ed5f
EB
476 BlkdebugRule *rule = NULL;
477 int error;
478 bool immediately;
479
480 QSIMPLEQ_FOREACH(rule, &s->active_rules, active_next) {
481 uint64_t inject_offset = rule->options.inject.offset;
482
483 if (inject_offset == -1 ||
484 (bytes && inject_offset >= offset &&
485 inject_offset < offset + bytes))
486 {
487 break;
488 }
489 }
490
491 if (!rule || !rule->options.inject.error) {
492 return 0;
493 }
494
495 immediately = rule->options.inject.immediately;
496 error = rule->options.inject.error;
b9f66d96 497
571cd43e 498 if (rule->options.inject.once) {
a069e2f1
JS
499 QSIMPLEQ_REMOVE(&s->active_rules, rule, BlkdebugRule, active_next);
500 remove_rule(rule);
b9f66d96
KW
501 }
502
7c3a9985 503 if (!immediately) {
e5c67ab5 504 aio_co_schedule(qemu_get_current_aio_context(), qemu_coroutine_self());
7c3a9985 505 qemu_coroutine_yield();
b9f66d96
KW
506 }
507
7c3a9985 508 return -error;
b9f66d96
KW
509}
510
7c3a9985
KW
511static int coroutine_fn
512blkdebug_co_preadv(BlockDriverState *bs, uint64_t offset, uint64_t bytes,
513 QEMUIOVector *qiov, int flags)
6a143727 514{
d157ed5f 515 int err;
e4780db4 516
e0ef4395
EB
517 /* Sanity check block layer guarantees */
518 assert(QEMU_IS_ALIGNED(offset, bs->bl.request_alignment));
519 assert(QEMU_IS_ALIGNED(bytes, bs->bl.request_alignment));
520 if (bs->bl.max_transfer) {
521 assert(bytes <= bs->bl.max_transfer);
522 }
523
d157ed5f
EB
524 err = rule_check(bs, offset, bytes);
525 if (err) {
526 return err;
b9f66d96
KW
527 }
528
7c3a9985 529 return bdrv_co_preadv(bs->file, offset, bytes, qiov, flags);
6a143727
KW
530}
531
7c3a9985
KW
532static int coroutine_fn
533blkdebug_co_pwritev(BlockDriverState *bs, uint64_t offset, uint64_t bytes,
534 QEMUIOVector *qiov, int flags)
6a143727 535{
d157ed5f 536 int err;
e4780db4 537
e0ef4395
EB
538 /* Sanity check block layer guarantees */
539 assert(QEMU_IS_ALIGNED(offset, bs->bl.request_alignment));
540 assert(QEMU_IS_ALIGNED(bytes, bs->bl.request_alignment));
541 if (bs->bl.max_transfer) {
542 assert(bytes <= bs->bl.max_transfer);
543 }
544
d157ed5f
EB
545 err = rule_check(bs, offset, bytes);
546 if (err) {
547 return err;
b9f66d96
KW
548 }
549
7c3a9985 550 return bdrv_co_pwritev(bs->file, offset, bytes, qiov, flags);
6a143727
KW
551}
552
7c3a9985 553static int blkdebug_co_flush(BlockDriverState *bs)
9e52c53b 554{
d157ed5f 555 int err = rule_check(bs, 0, 0);
9e52c53b 556
d157ed5f
EB
557 if (err) {
558 return err;
9e52c53b
PB
559 }
560
7c3a9985 561 return bdrv_co_flush(bs->file->bs);
9e52c53b
PB
562}
563
63188c24 564static int coroutine_fn blkdebug_co_pwrite_zeroes(BlockDriverState *bs,
f5a5ca79 565 int64_t offset, int bytes,
63188c24
EB
566 BdrvRequestFlags flags)
567{
568 uint32_t align = MAX(bs->bl.request_alignment,
569 bs->bl.pwrite_zeroes_alignment);
570 int err;
571
572 /* Only pass through requests that are larger than requested
573 * preferred alignment (so that we test the fallback to writes on
574 * unaligned portions), and check that the block layer never hands
575 * us anything unaligned that crosses an alignment boundary. */
f5a5ca79 576 if (bytes < align) {
63188c24 577 assert(QEMU_IS_ALIGNED(offset, align) ||
f5a5ca79 578 QEMU_IS_ALIGNED(offset + bytes, align) ||
63188c24 579 DIV_ROUND_UP(offset, align) ==
f5a5ca79 580 DIV_ROUND_UP(offset + bytes, align));
63188c24
EB
581 return -ENOTSUP;
582 }
583 assert(QEMU_IS_ALIGNED(offset, align));
f5a5ca79 584 assert(QEMU_IS_ALIGNED(bytes, align));
63188c24 585 if (bs->bl.max_pwrite_zeroes) {
f5a5ca79 586 assert(bytes <= bs->bl.max_pwrite_zeroes);
63188c24
EB
587 }
588
f5a5ca79 589 err = rule_check(bs, offset, bytes);
63188c24
EB
590 if (err) {
591 return err;
592 }
593
f5a5ca79 594 return bdrv_co_pwrite_zeroes(bs->file, offset, bytes, flags);
63188c24
EB
595}
596
597static int coroutine_fn blkdebug_co_pdiscard(BlockDriverState *bs,
f5a5ca79 598 int64_t offset, int bytes)
63188c24
EB
599{
600 uint32_t align = bs->bl.pdiscard_alignment;
601 int err;
602
603 /* Only pass through requests that are larger than requested
604 * minimum alignment, and ensure that unaligned requests do not
605 * cross optimum discard boundaries. */
f5a5ca79 606 if (bytes < bs->bl.request_alignment) {
63188c24 607 assert(QEMU_IS_ALIGNED(offset, align) ||
f5a5ca79 608 QEMU_IS_ALIGNED(offset + bytes, align) ||
63188c24 609 DIV_ROUND_UP(offset, align) ==
f5a5ca79 610 DIV_ROUND_UP(offset + bytes, align));
63188c24
EB
611 return -ENOTSUP;
612 }
613 assert(QEMU_IS_ALIGNED(offset, bs->bl.request_alignment));
f5a5ca79
MP
614 assert(QEMU_IS_ALIGNED(bytes, bs->bl.request_alignment));
615 if (align && bytes >= align) {
63188c24 616 assert(QEMU_IS_ALIGNED(offset, align));
f5a5ca79 617 assert(QEMU_IS_ALIGNED(bytes, align));
63188c24
EB
618 }
619 if (bs->bl.max_pdiscard) {
f5a5ca79 620 assert(bytes <= bs->bl.max_pdiscard);
63188c24
EB
621 }
622
f5a5ca79 623 err = rule_check(bs, offset, bytes);
63188c24
EB
624 if (err) {
625 return err;
626 }
627
f5a5ca79 628 return bdrv_co_pdiscard(bs->file->bs, offset, bytes);
63188c24 629}
3c90c65d 630
544daf66
EB
631static int64_t coroutine_fn blkdebug_co_get_block_status(
632 BlockDriverState *bs, int64_t sector_num, int nb_sectors, int *pnum,
633 BlockDriverState **file)
634{
635 *pnum = nb_sectors;
636 *file = bs->file->bs;
637 return BDRV_BLOCK_RAW | BDRV_BLOCK_OFFSET_VALID |
638 (sector_num << BDRV_SECTOR_BITS);
639}
640
6a143727
KW
641static void blkdebug_close(BlockDriverState *bs)
642{
643 BDRVBlkdebugState *s = bs->opaque;
8b9b0cc2
KW
644 BlkdebugRule *rule, *next;
645 int i;
646
7fb1cf16 647 for (i = 0; i < BLKDBG__MAX; i++) {
8b9b0cc2 648 QLIST_FOREACH_SAFE(rule, &s->rules[i], next, next) {
9e35542b 649 remove_rule(rule);
8b9b0cc2
KW
650 }
651 }
036990d7
HR
652
653 g_free(s->config_file);
6a143727
KW
654}
655
3c90c65d
KW
656static void suspend_request(BlockDriverState *bs, BlkdebugRule *rule)
657{
658 BDRVBlkdebugState *s = bs->opaque;
659 BlkdebugSuspendedReq r;
660
661 r = (BlkdebugSuspendedReq) {
662 .co = qemu_coroutine_self(),
663 .tag = g_strdup(rule->options.suspend.tag),
664 };
665
666 remove_rule(rule);
667 QLIST_INSERT_HEAD(&s->suspended_reqs, &r, next);
668
20873526
MT
669 if (!qtest_enabled()) {
670 printf("blkdebug: Suspended request '%s'\n", r.tag);
671 }
3c90c65d 672 qemu_coroutine_yield();
20873526
MT
673 if (!qtest_enabled()) {
674 printf("blkdebug: Resuming request '%s'\n", r.tag);
675 }
3c90c65d
KW
676
677 QLIST_REMOVE(&r, next);
678 g_free(r.tag);
679}
680
571cd43e 681static bool process_rule(BlockDriverState *bs, struct BlkdebugRule *rule,
8f96b5be 682 bool injected)
8b9b0cc2
KW
683{
684 BDRVBlkdebugState *s = bs->opaque;
8b9b0cc2
KW
685
686 /* Only process rules for the current state */
8f96b5be 687 if (rule->state && rule->state != s->state) {
571cd43e 688 return injected;
8b9b0cc2
KW
689 }
690
691 /* Take the action */
692 switch (rule->action) {
693 case ACTION_INJECT_ERROR:
571cd43e
PB
694 if (!injected) {
695 QSIMPLEQ_INIT(&s->active_rules);
696 injected = true;
697 }
698 QSIMPLEQ_INSERT_HEAD(&s->active_rules, rule, active_next);
8b9b0cc2
KW
699 break;
700
701 case ACTION_SET_STATE:
8f96b5be 702 s->new_state = rule->options.set_state.new_state;
8b9b0cc2 703 break;
3c90c65d
KW
704
705 case ACTION_SUSPEND:
706 suspend_request(bs, rule);
707 break;
8b9b0cc2 708 }
571cd43e 709 return injected;
8b9b0cc2
KW
710}
711
a31939e6 712static void blkdebug_debug_event(BlockDriverState *bs, BlkdebugEvent event)
8b9b0cc2
KW
713{
714 BDRVBlkdebugState *s = bs->opaque;
3c90c65d 715 struct BlkdebugRule *rule, *next;
571cd43e 716 bool injected;
8b9b0cc2 717
7fb1cf16 718 assert((int)event >= 0 && event < BLKDBG__MAX);
8b9b0cc2 719
571cd43e 720 injected = false;
8f96b5be 721 s->new_state = s->state;
3c90c65d 722 QLIST_FOREACH_SAFE(rule, &s->rules[event], next, next) {
8f96b5be 723 injected = process_rule(bs, rule, injected);
8b9b0cc2 724 }
8f96b5be 725 s->state = s->new_state;
8b9b0cc2
KW
726}
727
3c90c65d
KW
728static int blkdebug_debug_breakpoint(BlockDriverState *bs, const char *event,
729 const char *tag)
730{
731 BDRVBlkdebugState *s = bs->opaque;
732 struct BlkdebugRule *rule;
f9509d15 733 int blkdebug_event;
3c90c65d 734
f7abe0ec 735 blkdebug_event = qapi_enum_parse(&BlkdebugEvent_lookup, event, -1, NULL);
f9509d15 736 if (blkdebug_event < 0) {
3c90c65d
KW
737 return -ENOENT;
738 }
739
3c90c65d
KW
740 rule = g_malloc(sizeof(*rule));
741 *rule = (struct BlkdebugRule) {
742 .event = blkdebug_event,
743 .action = ACTION_SUSPEND,
744 .state = 0,
745 .options.suspend.tag = g_strdup(tag),
746 };
747
748 QLIST_INSERT_HEAD(&s->rules[blkdebug_event], rule, next);
749
750 return 0;
751}
752
753static int blkdebug_debug_resume(BlockDriverState *bs, const char *tag)
754{
755 BDRVBlkdebugState *s = bs->opaque;
c547e564 756 BlkdebugSuspendedReq *r, *next;
3c90c65d 757
c547e564 758 QLIST_FOREACH_SAFE(r, &s->suspended_reqs, next, next) {
3c90c65d 759 if (!strcmp(r->tag, tag)) {
0b8b8753 760 qemu_coroutine_enter(r->co);
3c90c65d
KW
761 return 0;
762 }
763 }
764 return -ENOENT;
765}
766
4cc70e93
FZ
767static int blkdebug_debug_remove_breakpoint(BlockDriverState *bs,
768 const char *tag)
769{
770 BDRVBlkdebugState *s = bs->opaque;
c547e564 771 BlkdebugSuspendedReq *r, *r_next;
4cc70e93
FZ
772 BlkdebugRule *rule, *next;
773 int i, ret = -ENOENT;
774
7fb1cf16 775 for (i = 0; i < BLKDBG__MAX; i++) {
4cc70e93
FZ
776 QLIST_FOREACH_SAFE(rule, &s->rules[i], next, next) {
777 if (rule->action == ACTION_SUSPEND &&
778 !strcmp(rule->options.suspend.tag, tag)) {
779 remove_rule(rule);
780 ret = 0;
781 }
782 }
783 }
c547e564 784 QLIST_FOREACH_SAFE(r, &s->suspended_reqs, next, r_next) {
4cc70e93 785 if (!strcmp(r->tag, tag)) {
0b8b8753 786 qemu_coroutine_enter(r->co);
4cc70e93
FZ
787 ret = 0;
788 }
789 }
790 return ret;
791}
3c90c65d
KW
792
793static bool blkdebug_debug_is_suspended(BlockDriverState *bs, const char *tag)
794{
795 BDRVBlkdebugState *s = bs->opaque;
796 BlkdebugSuspendedReq *r;
797
798 QLIST_FOREACH(r, &s->suspended_reqs, next) {
799 if (!strcmp(r->tag, tag)) {
800 return true;
801 }
802 }
803 return false;
804}
805
e1302255
PB
806static int64_t blkdebug_getlength(BlockDriverState *bs)
807{
9a4f4c31 808 return bdrv_getlength(bs->file->bs);
e1302255
PB
809}
810
8243ccb7
HR
811static int blkdebug_truncate(BlockDriverState *bs, int64_t offset,
812 PreallocMode prealloc, Error **errp)
8eedfbd4 813{
7ea37c30 814 return bdrv_truncate(bs->file, offset, prealloc, errp);
8eedfbd4
KW
815}
816
4cdd01d3 817static void blkdebug_refresh_filename(BlockDriverState *bs, QDict *options)
2c31b04c 818{
036990d7 819 BDRVBlkdebugState *s = bs->opaque;
2c31b04c 820 QDict *opts;
8779441b
HR
821 const QDictEntry *e;
822 bool force_json = false;
823
4cdd01d3 824 for (e = qdict_first(options); e; e = qdict_next(options, e)) {
8779441b 825 if (strcmp(qdict_entry_key(e), "config") &&
4cdd01d3 826 strcmp(qdict_entry_key(e), "x-image"))
8779441b
HR
827 {
828 force_json = true;
829 break;
830 }
831 }
2c31b04c 832
9a4f4c31 833 if (force_json && !bs->file->bs->full_open_options) {
2c31b04c
HR
834 /* The config file cannot be recreated, so creating a plain filename
835 * is impossible */
836 return;
837 }
838
9a4f4c31 839 if (!force_json && bs->file->bs->exact_filename[0]) {
de81d72d
HR
840 int ret = snprintf(bs->exact_filename, sizeof(bs->exact_filename),
841 "blkdebug:%s:%s", s->config_file ?: "",
842 bs->file->bs->exact_filename);
843 if (ret >= sizeof(bs->exact_filename)) {
844 /* An overflow makes the filename unusable, so do not report any */
845 bs->exact_filename[0] = 0;
846 }
8779441b
HR
847 }
848
2c31b04c 849 opts = qdict_new();
46f5ac20 850 qdict_put_str(opts, "driver", "blkdebug");
2c31b04c 851
9a4f4c31 852 QINCREF(bs->file->bs->full_open_options);
de6e7951 853 qdict_put(opts, "image", bs->file->bs->full_open_options);
2c31b04c 854
4cdd01d3
KW
855 for (e = qdict_first(options); e; e = qdict_next(options, e)) {
856 if (strcmp(qdict_entry_key(e), "x-image")) {
8779441b
HR
857 qobject_incref(qdict_entry_value(e));
858 qdict_put_obj(opts, qdict_entry_key(e), qdict_entry_value(e));
2c31b04c
HR
859 }
860 }
861
2c31b04c
HR
862 bs->full_open_options = opts;
863}
864
835db3ee
EB
865static void blkdebug_refresh_limits(BlockDriverState *bs, Error **errp)
866{
867 BDRVBlkdebugState *s = bs->opaque;
868
869 if (s->align) {
a5b8dd2c 870 bs->bl.request_alignment = s->align;
835db3ee 871 }
430b26a8
EB
872 if (s->max_transfer) {
873 bs->bl.max_transfer = s->max_transfer;
874 }
875 if (s->opt_write_zero) {
876 bs->bl.pwrite_zeroes_alignment = s->opt_write_zero;
877 }
878 if (s->max_write_zero) {
879 bs->bl.max_pwrite_zeroes = s->max_write_zero;
880 }
881 if (s->opt_discard) {
882 bs->bl.pdiscard_alignment = s->opt_discard;
883 }
884 if (s->max_discard) {
885 bs->bl.max_pdiscard = s->max_discard;
886 }
835db3ee
EB
887}
888
c5e8bfb7
KW
889static int blkdebug_reopen_prepare(BDRVReopenState *reopen_state,
890 BlockReopenQueue *queue, Error **errp)
891{
892 return 0;
893}
894
6a143727 895static BlockDriver bdrv_blkdebug = {
f4681212
KW
896 .format_name = "blkdebug",
897 .protocol_name = "blkdebug",
898 .instance_size = sizeof(BDRVBlkdebugState),
6a143727 899
f4681212
KW
900 .bdrv_parse_filename = blkdebug_parse_filename,
901 .bdrv_file_open = blkdebug_open,
902 .bdrv_close = blkdebug_close,
c5e8bfb7 903 .bdrv_reopen_prepare = blkdebug_reopen_prepare,
d7010dfb
KW
904 .bdrv_child_perm = bdrv_filter_default_perms,
905
f4681212 906 .bdrv_getlength = blkdebug_getlength,
8eedfbd4 907 .bdrv_truncate = blkdebug_truncate,
2c31b04c 908 .bdrv_refresh_filename = blkdebug_refresh_filename,
835db3ee 909 .bdrv_refresh_limits = blkdebug_refresh_limits,
6a143727 910
7c3a9985
KW
911 .bdrv_co_preadv = blkdebug_co_preadv,
912 .bdrv_co_pwritev = blkdebug_co_pwritev,
913 .bdrv_co_flush_to_disk = blkdebug_co_flush,
63188c24
EB
914 .bdrv_co_pwrite_zeroes = blkdebug_co_pwrite_zeroes,
915 .bdrv_co_pdiscard = blkdebug_co_pdiscard,
544daf66 916 .bdrv_co_get_block_status = blkdebug_co_get_block_status,
8b9b0cc2 917
3c90c65d
KW
918 .bdrv_debug_event = blkdebug_debug_event,
919 .bdrv_debug_breakpoint = blkdebug_debug_breakpoint,
4cc70e93
FZ
920 .bdrv_debug_remove_breakpoint
921 = blkdebug_debug_remove_breakpoint,
3c90c65d
KW
922 .bdrv_debug_resume = blkdebug_debug_resume,
923 .bdrv_debug_is_suspended = blkdebug_debug_is_suspended,
6a143727
KW
924};
925
926static void bdrv_blkdebug_init(void)
927{
928 bdrv_register(&bdrv_blkdebug);
929}
930
931block_init(bdrv_blkdebug_init);