]> git.proxmox.com Git - mirror_qemu.git/blob - block/block-backend.c
Merge remote-tracking branch 'remotes/kevin/tags/for-upstream' into staging
[mirror_qemu.git] / block / block-backend.c
1 /*
2 * QEMU Block backends
3 *
4 * Copyright (C) 2014-2016 Red Hat, Inc.
5 *
6 * Authors:
7 * Markus Armbruster <armbru@redhat.com>,
8 *
9 * This work is licensed under the terms of the GNU LGPL, version 2.1
10 * or later. See the COPYING.LIB file in the top-level directory.
11 */
12
13 #include "qemu/osdep.h"
14 #include "sysemu/block-backend.h"
15 #include "block/block_int.h"
16 #include "block/blockjob.h"
17 #include "block/throttle-groups.h"
18 #include "sysemu/blockdev.h"
19 #include "sysemu/sysemu.h"
20 #include "qapi/error.h"
21 #include "qapi/qapi-events-block.h"
22 #include "qemu/id.h"
23 #include "qemu/option.h"
24 #include "trace.h"
25 #include "migration/misc.h"
26
27 /* Number of coroutines to reserve per attached device model */
28 #define COROUTINE_POOL_RESERVATION 64
29
30 #define NOT_DONE 0x7fffffff /* used while emulated sync operation in progress */
31
32 static AioContext *blk_aiocb_get_aio_context(BlockAIOCB *acb);
33
34 typedef struct BlockBackendAioNotifier {
35 void (*attached_aio_context)(AioContext *new_context, void *opaque);
36 void (*detach_aio_context)(void *opaque);
37 void *opaque;
38 QLIST_ENTRY(BlockBackendAioNotifier) list;
39 } BlockBackendAioNotifier;
40
41 struct BlockBackend {
42 char *name;
43 int refcnt;
44 BdrvChild *root;
45 AioContext *ctx;
46 DriveInfo *legacy_dinfo; /* null unless created by drive_new() */
47 QTAILQ_ENTRY(BlockBackend) link; /* for block_backends */
48 QTAILQ_ENTRY(BlockBackend) monitor_link; /* for monitor_block_backends */
49 BlockBackendPublic public;
50
51 DeviceState *dev; /* attached device model, if any */
52 const BlockDevOps *dev_ops;
53 void *dev_opaque;
54
55 /* the block size for which the guest device expects atomicity */
56 int guest_block_size;
57
58 /* If the BDS tree is removed, some of its options are stored here (which
59 * can be used to restore those options in the new BDS on insert) */
60 BlockBackendRootState root_state;
61
62 bool enable_write_cache;
63
64 /* I/O stats (display with "info blockstats"). */
65 BlockAcctStats stats;
66
67 BlockdevOnError on_read_error, on_write_error;
68 bool iostatus_enabled;
69 BlockDeviceIoStatus iostatus;
70
71 uint64_t perm;
72 uint64_t shared_perm;
73 bool disable_perm;
74
75 bool allow_aio_context_change;
76 bool allow_write_beyond_eof;
77
78 NotifierList remove_bs_notifiers, insert_bs_notifiers;
79 QLIST_HEAD(, BlockBackendAioNotifier) aio_notifiers;
80
81 int quiesce_counter;
82 VMChangeStateEntry *vmsh;
83 bool force_allow_inactivate;
84
85 /* Number of in-flight aio requests. BlockDriverState also counts
86 * in-flight requests but aio requests can exist even when blk->root is
87 * NULL, so we cannot rely on its counter for that case.
88 * Accessed with atomic ops.
89 */
90 unsigned int in_flight;
91 };
92
93 typedef struct BlockBackendAIOCB {
94 BlockAIOCB common;
95 BlockBackend *blk;
96 int ret;
97 } BlockBackendAIOCB;
98
99 static const AIOCBInfo block_backend_aiocb_info = {
100 .get_aio_context = blk_aiocb_get_aio_context,
101 .aiocb_size = sizeof(BlockBackendAIOCB),
102 };
103
104 static void drive_info_del(DriveInfo *dinfo);
105 static BlockBackend *bdrv_first_blk(BlockDriverState *bs);
106
107 /* All BlockBackends */
108 static QTAILQ_HEAD(, BlockBackend) block_backends =
109 QTAILQ_HEAD_INITIALIZER(block_backends);
110
111 /* All BlockBackends referenced by the monitor and which are iterated through by
112 * blk_next() */
113 static QTAILQ_HEAD(, BlockBackend) monitor_block_backends =
114 QTAILQ_HEAD_INITIALIZER(monitor_block_backends);
115
116 static void blk_root_inherit_options(int *child_flags, QDict *child_options,
117 int parent_flags, QDict *parent_options)
118 {
119 /* We're not supposed to call this function for root nodes */
120 abort();
121 }
122 static void blk_root_drained_begin(BdrvChild *child);
123 static bool blk_root_drained_poll(BdrvChild *child);
124 static void blk_root_drained_end(BdrvChild *child);
125
126 static void blk_root_change_media(BdrvChild *child, bool load);
127 static void blk_root_resize(BdrvChild *child);
128
129 static bool blk_root_can_set_aio_ctx(BdrvChild *child, AioContext *ctx,
130 GSList **ignore, Error **errp);
131 static void blk_root_set_aio_ctx(BdrvChild *child, AioContext *ctx,
132 GSList **ignore);
133
134 static char *blk_root_get_parent_desc(BdrvChild *child)
135 {
136 BlockBackend *blk = child->opaque;
137 char *dev_id;
138
139 if (blk->name) {
140 return g_strdup(blk->name);
141 }
142
143 dev_id = blk_get_attached_dev_id(blk);
144 if (*dev_id) {
145 return dev_id;
146 } else {
147 /* TODO Callback into the BB owner for something more detailed */
148 g_free(dev_id);
149 return g_strdup("a block device");
150 }
151 }
152
153 static const char *blk_root_get_name(BdrvChild *child)
154 {
155 return blk_name(child->opaque);
156 }
157
158 static void blk_vm_state_changed(void *opaque, int running, RunState state)
159 {
160 Error *local_err = NULL;
161 BlockBackend *blk = opaque;
162
163 if (state == RUN_STATE_INMIGRATE) {
164 return;
165 }
166
167 qemu_del_vm_change_state_handler(blk->vmsh);
168 blk->vmsh = NULL;
169 blk_set_perm(blk, blk->perm, blk->shared_perm, &local_err);
170 if (local_err) {
171 error_report_err(local_err);
172 }
173 }
174
175 /*
176 * Notifies the user of the BlockBackend that migration has completed. qdev
177 * devices can tighten their permissions in response (specifically revoke
178 * shared write permissions that we needed for storage migration).
179 *
180 * If an error is returned, the VM cannot be allowed to be resumed.
181 */
182 static void blk_root_activate(BdrvChild *child, Error **errp)
183 {
184 BlockBackend *blk = child->opaque;
185 Error *local_err = NULL;
186
187 if (!blk->disable_perm) {
188 return;
189 }
190
191 blk->disable_perm = false;
192
193 blk_set_perm(blk, blk->perm, BLK_PERM_ALL, &local_err);
194 if (local_err) {
195 error_propagate(errp, local_err);
196 blk->disable_perm = true;
197 return;
198 }
199
200 if (runstate_check(RUN_STATE_INMIGRATE)) {
201 /* Activation can happen when migration process is still active, for
202 * example when nbd_server_add is called during non-shared storage
203 * migration. Defer the shared_perm update to migration completion. */
204 if (!blk->vmsh) {
205 blk->vmsh = qemu_add_vm_change_state_handler(blk_vm_state_changed,
206 blk);
207 }
208 return;
209 }
210
211 blk_set_perm(blk, blk->perm, blk->shared_perm, &local_err);
212 if (local_err) {
213 error_propagate(errp, local_err);
214 blk->disable_perm = true;
215 return;
216 }
217 }
218
219 void blk_set_force_allow_inactivate(BlockBackend *blk)
220 {
221 blk->force_allow_inactivate = true;
222 }
223
224 static bool blk_can_inactivate(BlockBackend *blk)
225 {
226 /* If it is a guest device, inactivate is ok. */
227 if (blk->dev || blk_name(blk)[0]) {
228 return true;
229 }
230
231 /* Inactivating means no more writes to the image can be done,
232 * even if those writes would be changes invisible to the
233 * guest. For block job BBs that satisfy this, we can just allow
234 * it. This is the case for mirror job source, which is required
235 * by libvirt non-shared block migration. */
236 if (!(blk->perm & (BLK_PERM_WRITE | BLK_PERM_WRITE_UNCHANGED))) {
237 return true;
238 }
239
240 return blk->force_allow_inactivate;
241 }
242
243 static int blk_root_inactivate(BdrvChild *child)
244 {
245 BlockBackend *blk = child->opaque;
246
247 if (blk->disable_perm) {
248 return 0;
249 }
250
251 if (!blk_can_inactivate(blk)) {
252 return -EPERM;
253 }
254
255 blk->disable_perm = true;
256 if (blk->root) {
257 bdrv_child_try_set_perm(blk->root, 0, BLK_PERM_ALL, &error_abort);
258 }
259
260 return 0;
261 }
262
263 static void blk_root_attach(BdrvChild *child)
264 {
265 BlockBackend *blk = child->opaque;
266 BlockBackendAioNotifier *notifier;
267
268 trace_blk_root_attach(child, blk, child->bs);
269
270 QLIST_FOREACH(notifier, &blk->aio_notifiers, list) {
271 bdrv_add_aio_context_notifier(child->bs,
272 notifier->attached_aio_context,
273 notifier->detach_aio_context,
274 notifier->opaque);
275 }
276 }
277
278 static void blk_root_detach(BdrvChild *child)
279 {
280 BlockBackend *blk = child->opaque;
281 BlockBackendAioNotifier *notifier;
282
283 trace_blk_root_detach(child, blk, child->bs);
284
285 QLIST_FOREACH(notifier, &blk->aio_notifiers, list) {
286 bdrv_remove_aio_context_notifier(child->bs,
287 notifier->attached_aio_context,
288 notifier->detach_aio_context,
289 notifier->opaque);
290 }
291 }
292
293 static const BdrvChildRole child_root = {
294 .inherit_options = blk_root_inherit_options,
295
296 .change_media = blk_root_change_media,
297 .resize = blk_root_resize,
298 .get_name = blk_root_get_name,
299 .get_parent_desc = blk_root_get_parent_desc,
300
301 .drained_begin = blk_root_drained_begin,
302 .drained_poll = blk_root_drained_poll,
303 .drained_end = blk_root_drained_end,
304
305 .activate = blk_root_activate,
306 .inactivate = blk_root_inactivate,
307
308 .attach = blk_root_attach,
309 .detach = blk_root_detach,
310
311 .can_set_aio_ctx = blk_root_can_set_aio_ctx,
312 .set_aio_ctx = blk_root_set_aio_ctx,
313 };
314
315 /*
316 * Create a new BlockBackend with a reference count of one.
317 *
318 * @perm is a bitmasks of BLK_PERM_* constants which describes the permissions
319 * to request for a block driver node that is attached to this BlockBackend.
320 * @shared_perm is a bitmask which describes which permissions may be granted
321 * to other users of the attached node.
322 * Both sets of permissions can be changed later using blk_set_perm().
323 *
324 * Return the new BlockBackend on success, null on failure.
325 */
326 BlockBackend *blk_new(AioContext *ctx, uint64_t perm, uint64_t shared_perm)
327 {
328 BlockBackend *blk;
329
330 blk = g_new0(BlockBackend, 1);
331 blk->refcnt = 1;
332 blk->ctx = ctx;
333 blk->perm = perm;
334 blk->shared_perm = shared_perm;
335 blk_set_enable_write_cache(blk, true);
336
337 blk->on_read_error = BLOCKDEV_ON_ERROR_REPORT;
338 blk->on_write_error = BLOCKDEV_ON_ERROR_ENOSPC;
339
340 block_acct_init(&blk->stats);
341
342 notifier_list_init(&blk->remove_bs_notifiers);
343 notifier_list_init(&blk->insert_bs_notifiers);
344 QLIST_INIT(&blk->aio_notifiers);
345
346 QTAILQ_INSERT_TAIL(&block_backends, blk, link);
347 return blk;
348 }
349
350 /*
351 * Creates a new BlockBackend, opens a new BlockDriverState, and connects both.
352 * The new BlockBackend is in the main AioContext.
353 *
354 * Just as with bdrv_open(), after having called this function the reference to
355 * @options belongs to the block layer (even on failure).
356 *
357 * TODO: Remove @filename and @flags; it should be possible to specify a whole
358 * BDS tree just by specifying the @options QDict (or @reference,
359 * alternatively). At the time of adding this function, this is not possible,
360 * though, so callers of this function have to be able to specify @filename and
361 * @flags.
362 */
363 BlockBackend *blk_new_open(const char *filename, const char *reference,
364 QDict *options, int flags, Error **errp)
365 {
366 BlockBackend *blk;
367 BlockDriverState *bs;
368 uint64_t perm = 0;
369
370 /* blk_new_open() is mainly used in .bdrv_create implementations and the
371 * tools where sharing isn't a concern because the BDS stays private, so we
372 * just request permission according to the flags.
373 *
374 * The exceptions are xen_disk and blockdev_init(); in these cases, the
375 * caller of blk_new_open() doesn't make use of the permissions, but they
376 * shouldn't hurt either. We can still share everything here because the
377 * guest devices will add their own blockers if they can't share. */
378 if ((flags & BDRV_O_NO_IO) == 0) {
379 perm |= BLK_PERM_CONSISTENT_READ;
380 if (flags & BDRV_O_RDWR) {
381 perm |= BLK_PERM_WRITE;
382 }
383 }
384 if (flags & BDRV_O_RESIZE) {
385 perm |= BLK_PERM_RESIZE;
386 }
387
388 blk = blk_new(qemu_get_aio_context(), perm, BLK_PERM_ALL);
389 bs = bdrv_open(filename, reference, options, flags, errp);
390 if (!bs) {
391 blk_unref(blk);
392 return NULL;
393 }
394
395 blk->root = bdrv_root_attach_child(bs, "root", &child_root, blk->ctx,
396 perm, BLK_PERM_ALL, blk, errp);
397 if (!blk->root) {
398 blk_unref(blk);
399 return NULL;
400 }
401
402 return blk;
403 }
404
405 static void blk_delete(BlockBackend *blk)
406 {
407 assert(!blk->refcnt);
408 assert(!blk->name);
409 assert(!blk->dev);
410 if (blk->public.throttle_group_member.throttle_state) {
411 blk_io_limits_disable(blk);
412 }
413 if (blk->root) {
414 blk_remove_bs(blk);
415 }
416 if (blk->vmsh) {
417 qemu_del_vm_change_state_handler(blk->vmsh);
418 blk->vmsh = NULL;
419 }
420 assert(QLIST_EMPTY(&blk->remove_bs_notifiers.notifiers));
421 assert(QLIST_EMPTY(&blk->insert_bs_notifiers.notifiers));
422 assert(QLIST_EMPTY(&blk->aio_notifiers));
423 QTAILQ_REMOVE(&block_backends, blk, link);
424 drive_info_del(blk->legacy_dinfo);
425 block_acct_cleanup(&blk->stats);
426 g_free(blk);
427 }
428
429 static void drive_info_del(DriveInfo *dinfo)
430 {
431 if (!dinfo) {
432 return;
433 }
434 qemu_opts_del(dinfo->opts);
435 g_free(dinfo);
436 }
437
438 int blk_get_refcnt(BlockBackend *blk)
439 {
440 return blk ? blk->refcnt : 0;
441 }
442
443 /*
444 * Increment @blk's reference count.
445 * @blk must not be null.
446 */
447 void blk_ref(BlockBackend *blk)
448 {
449 assert(blk->refcnt > 0);
450 blk->refcnt++;
451 }
452
453 /*
454 * Decrement @blk's reference count.
455 * If this drops it to zero, destroy @blk.
456 * For convenience, do nothing if @blk is null.
457 */
458 void blk_unref(BlockBackend *blk)
459 {
460 if (blk) {
461 assert(blk->refcnt > 0);
462 if (blk->refcnt > 1) {
463 blk->refcnt--;
464 } else {
465 blk_drain(blk);
466 /* blk_drain() cannot resurrect blk, nobody held a reference */
467 assert(blk->refcnt == 1);
468 blk->refcnt = 0;
469 blk_delete(blk);
470 }
471 }
472 }
473
474 /*
475 * Behaves similarly to blk_next() but iterates over all BlockBackends, even the
476 * ones which are hidden (i.e. are not referenced by the monitor).
477 */
478 BlockBackend *blk_all_next(BlockBackend *blk)
479 {
480 return blk ? QTAILQ_NEXT(blk, link)
481 : QTAILQ_FIRST(&block_backends);
482 }
483
484 void blk_remove_all_bs(void)
485 {
486 BlockBackend *blk = NULL;
487
488 while ((blk = blk_all_next(blk)) != NULL) {
489 AioContext *ctx = blk_get_aio_context(blk);
490
491 aio_context_acquire(ctx);
492 if (blk->root) {
493 blk_remove_bs(blk);
494 }
495 aio_context_release(ctx);
496 }
497 }
498
499 /*
500 * Return the monitor-owned BlockBackend after @blk.
501 * If @blk is null, return the first one.
502 * Else, return @blk's next sibling, which may be null.
503 *
504 * To iterate over all BlockBackends, do
505 * for (blk = blk_next(NULL); blk; blk = blk_next(blk)) {
506 * ...
507 * }
508 */
509 BlockBackend *blk_next(BlockBackend *blk)
510 {
511 return blk ? QTAILQ_NEXT(blk, monitor_link)
512 : QTAILQ_FIRST(&monitor_block_backends);
513 }
514
515 /* Iterates over all top-level BlockDriverStates, i.e. BDSs that are owned by
516 * the monitor or attached to a BlockBackend */
517 BlockDriverState *bdrv_next(BdrvNextIterator *it)
518 {
519 BlockDriverState *bs, *old_bs;
520
521 /* Must be called from the main loop */
522 assert(qemu_get_current_aio_context() == qemu_get_aio_context());
523
524 /* First, return all root nodes of BlockBackends. In order to avoid
525 * returning a BDS twice when multiple BBs refer to it, we only return it
526 * if the BB is the first one in the parent list of the BDS. */
527 if (it->phase == BDRV_NEXT_BACKEND_ROOTS) {
528 BlockBackend *old_blk = it->blk;
529
530 old_bs = old_blk ? blk_bs(old_blk) : NULL;
531
532 do {
533 it->blk = blk_all_next(it->blk);
534 bs = it->blk ? blk_bs(it->blk) : NULL;
535 } while (it->blk && (bs == NULL || bdrv_first_blk(bs) != it->blk));
536
537 if (it->blk) {
538 blk_ref(it->blk);
539 }
540 blk_unref(old_blk);
541
542 if (bs) {
543 bdrv_ref(bs);
544 bdrv_unref(old_bs);
545 return bs;
546 }
547 it->phase = BDRV_NEXT_MONITOR_OWNED;
548 } else {
549 old_bs = it->bs;
550 }
551
552 /* Then return the monitor-owned BDSes without a BB attached. Ignore all
553 * BDSes that are attached to a BlockBackend here; they have been handled
554 * by the above block already */
555 do {
556 it->bs = bdrv_next_monitor_owned(it->bs);
557 bs = it->bs;
558 } while (bs && bdrv_has_blk(bs));
559
560 if (bs) {
561 bdrv_ref(bs);
562 }
563 bdrv_unref(old_bs);
564
565 return bs;
566 }
567
568 static void bdrv_next_reset(BdrvNextIterator *it)
569 {
570 *it = (BdrvNextIterator) {
571 .phase = BDRV_NEXT_BACKEND_ROOTS,
572 };
573 }
574
575 BlockDriverState *bdrv_first(BdrvNextIterator *it)
576 {
577 bdrv_next_reset(it);
578 return bdrv_next(it);
579 }
580
581 /* Must be called when aborting a bdrv_next() iteration before
582 * bdrv_next() returns NULL */
583 void bdrv_next_cleanup(BdrvNextIterator *it)
584 {
585 /* Must be called from the main loop */
586 assert(qemu_get_current_aio_context() == qemu_get_aio_context());
587
588 if (it->phase == BDRV_NEXT_BACKEND_ROOTS) {
589 if (it->blk) {
590 bdrv_unref(blk_bs(it->blk));
591 blk_unref(it->blk);
592 }
593 } else {
594 bdrv_unref(it->bs);
595 }
596
597 bdrv_next_reset(it);
598 }
599
600 /*
601 * Add a BlockBackend into the list of backends referenced by the monitor, with
602 * the given @name acting as the handle for the monitor.
603 * Strictly for use by blockdev.c.
604 *
605 * @name must not be null or empty.
606 *
607 * Returns true on success and false on failure. In the latter case, an Error
608 * object is returned through @errp.
609 */
610 bool monitor_add_blk(BlockBackend *blk, const char *name, Error **errp)
611 {
612 assert(!blk->name);
613 assert(name && name[0]);
614
615 if (!id_wellformed(name)) {
616 error_setg(errp, "Invalid device name");
617 return false;
618 }
619 if (blk_by_name(name)) {
620 error_setg(errp, "Device with id '%s' already exists", name);
621 return false;
622 }
623 if (bdrv_find_node(name)) {
624 error_setg(errp,
625 "Device name '%s' conflicts with an existing node name",
626 name);
627 return false;
628 }
629
630 blk->name = g_strdup(name);
631 QTAILQ_INSERT_TAIL(&monitor_block_backends, blk, monitor_link);
632 return true;
633 }
634
635 /*
636 * Remove a BlockBackend from the list of backends referenced by the monitor.
637 * Strictly for use by blockdev.c.
638 */
639 void monitor_remove_blk(BlockBackend *blk)
640 {
641 if (!blk->name) {
642 return;
643 }
644
645 QTAILQ_REMOVE(&monitor_block_backends, blk, monitor_link);
646 g_free(blk->name);
647 blk->name = NULL;
648 }
649
650 /*
651 * Return @blk's name, a non-null string.
652 * Returns an empty string iff @blk is not referenced by the monitor.
653 */
654 const char *blk_name(const BlockBackend *blk)
655 {
656 return blk->name ?: "";
657 }
658
659 /*
660 * Return the BlockBackend with name @name if it exists, else null.
661 * @name must not be null.
662 */
663 BlockBackend *blk_by_name(const char *name)
664 {
665 BlockBackend *blk = NULL;
666
667 assert(name);
668 while ((blk = blk_next(blk)) != NULL) {
669 if (!strcmp(name, blk->name)) {
670 return blk;
671 }
672 }
673 return NULL;
674 }
675
676 /*
677 * Return the BlockDriverState attached to @blk if any, else null.
678 */
679 BlockDriverState *blk_bs(BlockBackend *blk)
680 {
681 return blk->root ? blk->root->bs : NULL;
682 }
683
684 static BlockBackend *bdrv_first_blk(BlockDriverState *bs)
685 {
686 BdrvChild *child;
687 QLIST_FOREACH(child, &bs->parents, next_parent) {
688 if (child->role == &child_root) {
689 return child->opaque;
690 }
691 }
692
693 return NULL;
694 }
695
696 /*
697 * Returns true if @bs has an associated BlockBackend.
698 */
699 bool bdrv_has_blk(BlockDriverState *bs)
700 {
701 return bdrv_first_blk(bs) != NULL;
702 }
703
704 /*
705 * Returns true if @bs has only BlockBackends as parents.
706 */
707 bool bdrv_is_root_node(BlockDriverState *bs)
708 {
709 BdrvChild *c;
710
711 QLIST_FOREACH(c, &bs->parents, next_parent) {
712 if (c->role != &child_root) {
713 return false;
714 }
715 }
716
717 return true;
718 }
719
720 /*
721 * Return @blk's DriveInfo if any, else null.
722 */
723 DriveInfo *blk_legacy_dinfo(BlockBackend *blk)
724 {
725 return blk->legacy_dinfo;
726 }
727
728 /*
729 * Set @blk's DriveInfo to @dinfo, and return it.
730 * @blk must not have a DriveInfo set already.
731 * No other BlockBackend may have the same DriveInfo set.
732 */
733 DriveInfo *blk_set_legacy_dinfo(BlockBackend *blk, DriveInfo *dinfo)
734 {
735 assert(!blk->legacy_dinfo);
736 return blk->legacy_dinfo = dinfo;
737 }
738
739 /*
740 * Return the BlockBackend with DriveInfo @dinfo.
741 * It must exist.
742 */
743 BlockBackend *blk_by_legacy_dinfo(DriveInfo *dinfo)
744 {
745 BlockBackend *blk = NULL;
746
747 while ((blk = blk_next(blk)) != NULL) {
748 if (blk->legacy_dinfo == dinfo) {
749 return blk;
750 }
751 }
752 abort();
753 }
754
755 /*
756 * Returns a pointer to the publicly accessible fields of @blk.
757 */
758 BlockBackendPublic *blk_get_public(BlockBackend *blk)
759 {
760 return &blk->public;
761 }
762
763 /*
764 * Returns a BlockBackend given the associated @public fields.
765 */
766 BlockBackend *blk_by_public(BlockBackendPublic *public)
767 {
768 return container_of(public, BlockBackend, public);
769 }
770
771 /*
772 * Disassociates the currently associated BlockDriverState from @blk.
773 */
774 void blk_remove_bs(BlockBackend *blk)
775 {
776 ThrottleGroupMember *tgm = &blk->public.throttle_group_member;
777 BlockDriverState *bs;
778
779 notifier_list_notify(&blk->remove_bs_notifiers, blk);
780 if (tgm->throttle_state) {
781 bs = blk_bs(blk);
782 bdrv_drained_begin(bs);
783 throttle_group_detach_aio_context(tgm);
784 throttle_group_attach_aio_context(tgm, qemu_get_aio_context());
785 bdrv_drained_end(bs);
786 }
787
788 blk_update_root_state(blk);
789
790 /* bdrv_root_unref_child() will cause blk->root to become stale and may
791 * switch to a completion coroutine later on. Let's drain all I/O here
792 * to avoid that and a potential QEMU crash.
793 */
794 blk_drain(blk);
795 bdrv_root_unref_child(blk->root);
796 blk->root = NULL;
797 }
798
799 /*
800 * Associates a new BlockDriverState with @blk.
801 */
802 int blk_insert_bs(BlockBackend *blk, BlockDriverState *bs, Error **errp)
803 {
804 ThrottleGroupMember *tgm = &blk->public.throttle_group_member;
805 bdrv_ref(bs);
806 blk->root = bdrv_root_attach_child(bs, "root", &child_root, blk->ctx,
807 blk->perm, blk->shared_perm, blk, errp);
808 if (blk->root == NULL) {
809 return -EPERM;
810 }
811
812 notifier_list_notify(&blk->insert_bs_notifiers, blk);
813 if (tgm->throttle_state) {
814 throttle_group_detach_aio_context(tgm);
815 throttle_group_attach_aio_context(tgm, bdrv_get_aio_context(bs));
816 }
817
818 return 0;
819 }
820
821 /*
822 * Sets the permission bitmasks that the user of the BlockBackend needs.
823 */
824 int blk_set_perm(BlockBackend *blk, uint64_t perm, uint64_t shared_perm,
825 Error **errp)
826 {
827 int ret;
828
829 if (blk->root && !blk->disable_perm) {
830 ret = bdrv_child_try_set_perm(blk->root, perm, shared_perm, errp);
831 if (ret < 0) {
832 return ret;
833 }
834 }
835
836 blk->perm = perm;
837 blk->shared_perm = shared_perm;
838
839 return 0;
840 }
841
842 void blk_get_perm(BlockBackend *blk, uint64_t *perm, uint64_t *shared_perm)
843 {
844 *perm = blk->perm;
845 *shared_perm = blk->shared_perm;
846 }
847
848 /*
849 * Attach device model @dev to @blk.
850 * Return 0 on success, -EBUSY when a device model is attached already.
851 */
852 int blk_attach_dev(BlockBackend *blk, DeviceState *dev)
853 {
854 if (blk->dev) {
855 return -EBUSY;
856 }
857
858 /* While migration is still incoming, we don't need to apply the
859 * permissions of guest device BlockBackends. We might still have a block
860 * job or NBD server writing to the image for storage migration. */
861 if (runstate_check(RUN_STATE_INMIGRATE)) {
862 blk->disable_perm = true;
863 }
864
865 blk_ref(blk);
866 blk->dev = dev;
867 blk_iostatus_reset(blk);
868
869 return 0;
870 }
871
872 /*
873 * Detach device model @dev from @blk.
874 * @dev must be currently attached to @blk.
875 */
876 void blk_detach_dev(BlockBackend *blk, DeviceState *dev)
877 {
878 assert(blk->dev == dev);
879 blk->dev = NULL;
880 blk->dev_ops = NULL;
881 blk->dev_opaque = NULL;
882 blk->guest_block_size = 512;
883 blk_set_perm(blk, 0, BLK_PERM_ALL, &error_abort);
884 blk_unref(blk);
885 }
886
887 /*
888 * Return the device model attached to @blk if any, else null.
889 */
890 DeviceState *blk_get_attached_dev(BlockBackend *blk)
891 {
892 return blk->dev;
893 }
894
895 /* Return the qdev ID, or if no ID is assigned the QOM path, of the block
896 * device attached to the BlockBackend. */
897 char *blk_get_attached_dev_id(BlockBackend *blk)
898 {
899 DeviceState *dev = blk->dev;
900
901 if (!dev) {
902 return g_strdup("");
903 } else if (dev->id) {
904 return g_strdup(dev->id);
905 }
906
907 return object_get_canonical_path(OBJECT(dev)) ?: g_strdup("");
908 }
909
910 /*
911 * Return the BlockBackend which has the device model @dev attached if it
912 * exists, else null.
913 *
914 * @dev must not be null.
915 */
916 BlockBackend *blk_by_dev(void *dev)
917 {
918 BlockBackend *blk = NULL;
919
920 assert(dev != NULL);
921 while ((blk = blk_all_next(blk)) != NULL) {
922 if (blk->dev == dev) {
923 return blk;
924 }
925 }
926 return NULL;
927 }
928
929 /*
930 * Set @blk's device model callbacks to @ops.
931 * @opaque is the opaque argument to pass to the callbacks.
932 * This is for use by device models.
933 */
934 void blk_set_dev_ops(BlockBackend *blk, const BlockDevOps *ops,
935 void *opaque)
936 {
937 blk->dev_ops = ops;
938 blk->dev_opaque = opaque;
939
940 /* Are we currently quiesced? Should we enforce this right now? */
941 if (blk->quiesce_counter && ops->drained_begin) {
942 ops->drained_begin(opaque);
943 }
944 }
945
946 /*
947 * Notify @blk's attached device model of media change.
948 *
949 * If @load is true, notify of media load. This action can fail, meaning that
950 * the medium cannot be loaded. @errp is set then.
951 *
952 * If @load is false, notify of media eject. This can never fail.
953 *
954 * Also send DEVICE_TRAY_MOVED events as appropriate.
955 */
956 void blk_dev_change_media_cb(BlockBackend *blk, bool load, Error **errp)
957 {
958 if (blk->dev_ops && blk->dev_ops->change_media_cb) {
959 bool tray_was_open, tray_is_open;
960 Error *local_err = NULL;
961
962 tray_was_open = blk_dev_is_tray_open(blk);
963 blk->dev_ops->change_media_cb(blk->dev_opaque, load, &local_err);
964 if (local_err) {
965 assert(load == true);
966 error_propagate(errp, local_err);
967 return;
968 }
969 tray_is_open = blk_dev_is_tray_open(blk);
970
971 if (tray_was_open != tray_is_open) {
972 char *id = blk_get_attached_dev_id(blk);
973 qapi_event_send_device_tray_moved(blk_name(blk), id, tray_is_open);
974 g_free(id);
975 }
976 }
977 }
978
979 static void blk_root_change_media(BdrvChild *child, bool load)
980 {
981 blk_dev_change_media_cb(child->opaque, load, NULL);
982 }
983
984 /*
985 * Does @blk's attached device model have removable media?
986 * %true if no device model is attached.
987 */
988 bool blk_dev_has_removable_media(BlockBackend *blk)
989 {
990 return !blk->dev || (blk->dev_ops && blk->dev_ops->change_media_cb);
991 }
992
993 /*
994 * Does @blk's attached device model have a tray?
995 */
996 bool blk_dev_has_tray(BlockBackend *blk)
997 {
998 return blk->dev_ops && blk->dev_ops->is_tray_open;
999 }
1000
1001 /*
1002 * Notify @blk's attached device model of a media eject request.
1003 * If @force is true, the medium is about to be yanked out forcefully.
1004 */
1005 void blk_dev_eject_request(BlockBackend *blk, bool force)
1006 {
1007 if (blk->dev_ops && blk->dev_ops->eject_request_cb) {
1008 blk->dev_ops->eject_request_cb(blk->dev_opaque, force);
1009 }
1010 }
1011
1012 /*
1013 * Does @blk's attached device model have a tray, and is it open?
1014 */
1015 bool blk_dev_is_tray_open(BlockBackend *blk)
1016 {
1017 if (blk_dev_has_tray(blk)) {
1018 return blk->dev_ops->is_tray_open(blk->dev_opaque);
1019 }
1020 return false;
1021 }
1022
1023 /*
1024 * Does @blk's attached device model have the medium locked?
1025 * %false if the device model has no such lock.
1026 */
1027 bool blk_dev_is_medium_locked(BlockBackend *blk)
1028 {
1029 if (blk->dev_ops && blk->dev_ops->is_medium_locked) {
1030 return blk->dev_ops->is_medium_locked(blk->dev_opaque);
1031 }
1032 return false;
1033 }
1034
1035 /*
1036 * Notify @blk's attached device model of a backend size change.
1037 */
1038 static void blk_root_resize(BdrvChild *child)
1039 {
1040 BlockBackend *blk = child->opaque;
1041
1042 if (blk->dev_ops && blk->dev_ops->resize_cb) {
1043 blk->dev_ops->resize_cb(blk->dev_opaque);
1044 }
1045 }
1046
1047 void blk_iostatus_enable(BlockBackend *blk)
1048 {
1049 blk->iostatus_enabled = true;
1050 blk->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
1051 }
1052
1053 /* The I/O status is only enabled if the drive explicitly
1054 * enables it _and_ the VM is configured to stop on errors */
1055 bool blk_iostatus_is_enabled(const BlockBackend *blk)
1056 {
1057 return (blk->iostatus_enabled &&
1058 (blk->on_write_error == BLOCKDEV_ON_ERROR_ENOSPC ||
1059 blk->on_write_error == BLOCKDEV_ON_ERROR_STOP ||
1060 blk->on_read_error == BLOCKDEV_ON_ERROR_STOP));
1061 }
1062
1063 BlockDeviceIoStatus blk_iostatus(const BlockBackend *blk)
1064 {
1065 return blk->iostatus;
1066 }
1067
1068 void blk_iostatus_disable(BlockBackend *blk)
1069 {
1070 blk->iostatus_enabled = false;
1071 }
1072
1073 void blk_iostatus_reset(BlockBackend *blk)
1074 {
1075 if (blk_iostatus_is_enabled(blk)) {
1076 BlockDriverState *bs = blk_bs(blk);
1077 blk->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
1078 if (bs && bs->job) {
1079 block_job_iostatus_reset(bs->job);
1080 }
1081 }
1082 }
1083
1084 void blk_iostatus_set_err(BlockBackend *blk, int error)
1085 {
1086 assert(blk_iostatus_is_enabled(blk));
1087 if (blk->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
1088 blk->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
1089 BLOCK_DEVICE_IO_STATUS_FAILED;
1090 }
1091 }
1092
1093 void blk_set_allow_write_beyond_eof(BlockBackend *blk, bool allow)
1094 {
1095 blk->allow_write_beyond_eof = allow;
1096 }
1097
1098 void blk_set_allow_aio_context_change(BlockBackend *blk, bool allow)
1099 {
1100 blk->allow_aio_context_change = allow;
1101 }
1102
1103 static int blk_check_byte_request(BlockBackend *blk, int64_t offset,
1104 size_t size)
1105 {
1106 int64_t len;
1107
1108 if (size > INT_MAX) {
1109 return -EIO;
1110 }
1111
1112 if (!blk_is_available(blk)) {
1113 return -ENOMEDIUM;
1114 }
1115
1116 if (offset < 0) {
1117 return -EIO;
1118 }
1119
1120 if (!blk->allow_write_beyond_eof) {
1121 len = blk_getlength(blk);
1122 if (len < 0) {
1123 return len;
1124 }
1125
1126 if (offset > len || len - offset < size) {
1127 return -EIO;
1128 }
1129 }
1130
1131 return 0;
1132 }
1133
1134 int coroutine_fn blk_co_preadv(BlockBackend *blk, int64_t offset,
1135 unsigned int bytes, QEMUIOVector *qiov,
1136 BdrvRequestFlags flags)
1137 {
1138 int ret;
1139 BlockDriverState *bs = blk_bs(blk);
1140
1141 trace_blk_co_preadv(blk, bs, offset, bytes, flags);
1142
1143 ret = blk_check_byte_request(blk, offset, bytes);
1144 if (ret < 0) {
1145 return ret;
1146 }
1147
1148 bdrv_inc_in_flight(bs);
1149
1150 /* throttling disk I/O */
1151 if (blk->public.throttle_group_member.throttle_state) {
1152 throttle_group_co_io_limits_intercept(&blk->public.throttle_group_member,
1153 bytes, false);
1154 }
1155
1156 ret = bdrv_co_preadv(blk->root, offset, bytes, qiov, flags);
1157 bdrv_dec_in_flight(bs);
1158 return ret;
1159 }
1160
1161 int coroutine_fn blk_co_pwritev(BlockBackend *blk, int64_t offset,
1162 unsigned int bytes, QEMUIOVector *qiov,
1163 BdrvRequestFlags flags)
1164 {
1165 int ret;
1166 BlockDriverState *bs = blk_bs(blk);
1167
1168 trace_blk_co_pwritev(blk, bs, offset, bytes, flags);
1169
1170 ret = blk_check_byte_request(blk, offset, bytes);
1171 if (ret < 0) {
1172 return ret;
1173 }
1174
1175 bdrv_inc_in_flight(bs);
1176 /* throttling disk I/O */
1177 if (blk->public.throttle_group_member.throttle_state) {
1178 throttle_group_co_io_limits_intercept(&blk->public.throttle_group_member,
1179 bytes, true);
1180 }
1181
1182 if (!blk->enable_write_cache) {
1183 flags |= BDRV_REQ_FUA;
1184 }
1185
1186 ret = bdrv_co_pwritev(blk->root, offset, bytes, qiov, flags);
1187 bdrv_dec_in_flight(bs);
1188 return ret;
1189 }
1190
1191 typedef struct BlkRwCo {
1192 BlockBackend *blk;
1193 int64_t offset;
1194 void *iobuf;
1195 int ret;
1196 BdrvRequestFlags flags;
1197 } BlkRwCo;
1198
1199 static void blk_read_entry(void *opaque)
1200 {
1201 BlkRwCo *rwco = opaque;
1202 QEMUIOVector *qiov = rwco->iobuf;
1203
1204 rwco->ret = blk_co_preadv(rwco->blk, rwco->offset, qiov->size,
1205 qiov, rwco->flags);
1206 aio_wait_kick();
1207 }
1208
1209 static void blk_write_entry(void *opaque)
1210 {
1211 BlkRwCo *rwco = opaque;
1212 QEMUIOVector *qiov = rwco->iobuf;
1213
1214 rwco->ret = blk_co_pwritev(rwco->blk, rwco->offset, qiov->size,
1215 qiov, rwco->flags);
1216 aio_wait_kick();
1217 }
1218
1219 static int blk_prw(BlockBackend *blk, int64_t offset, uint8_t *buf,
1220 int64_t bytes, CoroutineEntry co_entry,
1221 BdrvRequestFlags flags)
1222 {
1223 QEMUIOVector qiov = QEMU_IOVEC_INIT_BUF(qiov, buf, bytes);
1224 BlkRwCo rwco = {
1225 .blk = blk,
1226 .offset = offset,
1227 .iobuf = &qiov,
1228 .flags = flags,
1229 .ret = NOT_DONE,
1230 };
1231
1232 if (qemu_in_coroutine()) {
1233 /* Fast-path if already in coroutine context */
1234 co_entry(&rwco);
1235 } else {
1236 Coroutine *co = qemu_coroutine_create(co_entry, &rwco);
1237 bdrv_coroutine_enter(blk_bs(blk), co);
1238 BDRV_POLL_WHILE(blk_bs(blk), rwco.ret == NOT_DONE);
1239 }
1240
1241 return rwco.ret;
1242 }
1243
1244 int blk_pread_unthrottled(BlockBackend *blk, int64_t offset, uint8_t *buf,
1245 int count)
1246 {
1247 int ret;
1248
1249 ret = blk_check_byte_request(blk, offset, count);
1250 if (ret < 0) {
1251 return ret;
1252 }
1253
1254 blk_root_drained_begin(blk->root);
1255 ret = blk_pread(blk, offset, buf, count);
1256 blk_root_drained_end(blk->root);
1257 return ret;
1258 }
1259
1260 int blk_pwrite_zeroes(BlockBackend *blk, int64_t offset,
1261 int bytes, BdrvRequestFlags flags)
1262 {
1263 return blk_prw(blk, offset, NULL, bytes, blk_write_entry,
1264 flags | BDRV_REQ_ZERO_WRITE);
1265 }
1266
1267 int blk_make_zero(BlockBackend *blk, BdrvRequestFlags flags)
1268 {
1269 return bdrv_make_zero(blk->root, flags);
1270 }
1271
1272 void blk_inc_in_flight(BlockBackend *blk)
1273 {
1274 atomic_inc(&blk->in_flight);
1275 }
1276
1277 void blk_dec_in_flight(BlockBackend *blk)
1278 {
1279 atomic_dec(&blk->in_flight);
1280 aio_wait_kick();
1281 }
1282
1283 static void error_callback_bh(void *opaque)
1284 {
1285 struct BlockBackendAIOCB *acb = opaque;
1286
1287 blk_dec_in_flight(acb->blk);
1288 acb->common.cb(acb->common.opaque, acb->ret);
1289 qemu_aio_unref(acb);
1290 }
1291
1292 BlockAIOCB *blk_abort_aio_request(BlockBackend *blk,
1293 BlockCompletionFunc *cb,
1294 void *opaque, int ret)
1295 {
1296 struct BlockBackendAIOCB *acb;
1297
1298 blk_inc_in_flight(blk);
1299 acb = blk_aio_get(&block_backend_aiocb_info, blk, cb, opaque);
1300 acb->blk = blk;
1301 acb->ret = ret;
1302
1303 aio_bh_schedule_oneshot(blk_get_aio_context(blk), error_callback_bh, acb);
1304 return &acb->common;
1305 }
1306
1307 typedef struct BlkAioEmAIOCB {
1308 BlockAIOCB common;
1309 BlkRwCo rwco;
1310 int bytes;
1311 bool has_returned;
1312 } BlkAioEmAIOCB;
1313
1314 static const AIOCBInfo blk_aio_em_aiocb_info = {
1315 .aiocb_size = sizeof(BlkAioEmAIOCB),
1316 };
1317
1318 static void blk_aio_complete(BlkAioEmAIOCB *acb)
1319 {
1320 if (acb->has_returned) {
1321 acb->common.cb(acb->common.opaque, acb->rwco.ret);
1322 blk_dec_in_flight(acb->rwco.blk);
1323 qemu_aio_unref(acb);
1324 }
1325 }
1326
1327 static void blk_aio_complete_bh(void *opaque)
1328 {
1329 BlkAioEmAIOCB *acb = opaque;
1330 assert(acb->has_returned);
1331 blk_aio_complete(acb);
1332 }
1333
1334 static BlockAIOCB *blk_aio_prwv(BlockBackend *blk, int64_t offset, int bytes,
1335 void *iobuf, CoroutineEntry co_entry,
1336 BdrvRequestFlags flags,
1337 BlockCompletionFunc *cb, void *opaque)
1338 {
1339 BlkAioEmAIOCB *acb;
1340 Coroutine *co;
1341
1342 blk_inc_in_flight(blk);
1343 acb = blk_aio_get(&blk_aio_em_aiocb_info, blk, cb, opaque);
1344 acb->rwco = (BlkRwCo) {
1345 .blk = blk,
1346 .offset = offset,
1347 .iobuf = iobuf,
1348 .flags = flags,
1349 .ret = NOT_DONE,
1350 };
1351 acb->bytes = bytes;
1352 acb->has_returned = false;
1353
1354 co = qemu_coroutine_create(co_entry, acb);
1355 bdrv_coroutine_enter(blk_bs(blk), co);
1356
1357 acb->has_returned = true;
1358 if (acb->rwco.ret != NOT_DONE) {
1359 aio_bh_schedule_oneshot(blk_get_aio_context(blk),
1360 blk_aio_complete_bh, acb);
1361 }
1362
1363 return &acb->common;
1364 }
1365
1366 static void blk_aio_read_entry(void *opaque)
1367 {
1368 BlkAioEmAIOCB *acb = opaque;
1369 BlkRwCo *rwco = &acb->rwco;
1370 QEMUIOVector *qiov = rwco->iobuf;
1371
1372 assert(qiov->size == acb->bytes);
1373 rwco->ret = blk_co_preadv(rwco->blk, rwco->offset, acb->bytes,
1374 qiov, rwco->flags);
1375 blk_aio_complete(acb);
1376 }
1377
1378 static void blk_aio_write_entry(void *opaque)
1379 {
1380 BlkAioEmAIOCB *acb = opaque;
1381 BlkRwCo *rwco = &acb->rwco;
1382 QEMUIOVector *qiov = rwco->iobuf;
1383
1384 assert(!qiov || qiov->size == acb->bytes);
1385 rwco->ret = blk_co_pwritev(rwco->blk, rwco->offset, acb->bytes,
1386 qiov, rwco->flags);
1387 blk_aio_complete(acb);
1388 }
1389
1390 BlockAIOCB *blk_aio_pwrite_zeroes(BlockBackend *blk, int64_t offset,
1391 int count, BdrvRequestFlags flags,
1392 BlockCompletionFunc *cb, void *opaque)
1393 {
1394 return blk_aio_prwv(blk, offset, count, NULL, blk_aio_write_entry,
1395 flags | BDRV_REQ_ZERO_WRITE, cb, opaque);
1396 }
1397
1398 int blk_pread(BlockBackend *blk, int64_t offset, void *buf, int count)
1399 {
1400 int ret = blk_prw(blk, offset, buf, count, blk_read_entry, 0);
1401 if (ret < 0) {
1402 return ret;
1403 }
1404 return count;
1405 }
1406
1407 int blk_pwrite(BlockBackend *blk, int64_t offset, const void *buf, int count,
1408 BdrvRequestFlags flags)
1409 {
1410 int ret = blk_prw(blk, offset, (void *) buf, count, blk_write_entry,
1411 flags);
1412 if (ret < 0) {
1413 return ret;
1414 }
1415 return count;
1416 }
1417
1418 int64_t blk_getlength(BlockBackend *blk)
1419 {
1420 if (!blk_is_available(blk)) {
1421 return -ENOMEDIUM;
1422 }
1423
1424 return bdrv_getlength(blk_bs(blk));
1425 }
1426
1427 void blk_get_geometry(BlockBackend *blk, uint64_t *nb_sectors_ptr)
1428 {
1429 if (!blk_bs(blk)) {
1430 *nb_sectors_ptr = 0;
1431 } else {
1432 bdrv_get_geometry(blk_bs(blk), nb_sectors_ptr);
1433 }
1434 }
1435
1436 int64_t blk_nb_sectors(BlockBackend *blk)
1437 {
1438 if (!blk_is_available(blk)) {
1439 return -ENOMEDIUM;
1440 }
1441
1442 return bdrv_nb_sectors(blk_bs(blk));
1443 }
1444
1445 BlockAIOCB *blk_aio_preadv(BlockBackend *blk, int64_t offset,
1446 QEMUIOVector *qiov, BdrvRequestFlags flags,
1447 BlockCompletionFunc *cb, void *opaque)
1448 {
1449 return blk_aio_prwv(blk, offset, qiov->size, qiov,
1450 blk_aio_read_entry, flags, cb, opaque);
1451 }
1452
1453 BlockAIOCB *blk_aio_pwritev(BlockBackend *blk, int64_t offset,
1454 QEMUIOVector *qiov, BdrvRequestFlags flags,
1455 BlockCompletionFunc *cb, void *opaque)
1456 {
1457 return blk_aio_prwv(blk, offset, qiov->size, qiov,
1458 blk_aio_write_entry, flags, cb, opaque);
1459 }
1460
1461 static void blk_aio_flush_entry(void *opaque)
1462 {
1463 BlkAioEmAIOCB *acb = opaque;
1464 BlkRwCo *rwco = &acb->rwco;
1465
1466 rwco->ret = blk_co_flush(rwco->blk);
1467 blk_aio_complete(acb);
1468 }
1469
1470 BlockAIOCB *blk_aio_flush(BlockBackend *blk,
1471 BlockCompletionFunc *cb, void *opaque)
1472 {
1473 return blk_aio_prwv(blk, 0, 0, NULL, blk_aio_flush_entry, 0, cb, opaque);
1474 }
1475
1476 static void blk_aio_pdiscard_entry(void *opaque)
1477 {
1478 BlkAioEmAIOCB *acb = opaque;
1479 BlkRwCo *rwco = &acb->rwco;
1480
1481 rwco->ret = blk_co_pdiscard(rwco->blk, rwco->offset, acb->bytes);
1482 blk_aio_complete(acb);
1483 }
1484
1485 BlockAIOCB *blk_aio_pdiscard(BlockBackend *blk,
1486 int64_t offset, int bytes,
1487 BlockCompletionFunc *cb, void *opaque)
1488 {
1489 return blk_aio_prwv(blk, offset, bytes, NULL, blk_aio_pdiscard_entry, 0,
1490 cb, opaque);
1491 }
1492
1493 void blk_aio_cancel(BlockAIOCB *acb)
1494 {
1495 bdrv_aio_cancel(acb);
1496 }
1497
1498 void blk_aio_cancel_async(BlockAIOCB *acb)
1499 {
1500 bdrv_aio_cancel_async(acb);
1501 }
1502
1503 int blk_co_ioctl(BlockBackend *blk, unsigned long int req, void *buf)
1504 {
1505 if (!blk_is_available(blk)) {
1506 return -ENOMEDIUM;
1507 }
1508
1509 return bdrv_co_ioctl(blk_bs(blk), req, buf);
1510 }
1511
1512 static void blk_ioctl_entry(void *opaque)
1513 {
1514 BlkRwCo *rwco = opaque;
1515 QEMUIOVector *qiov = rwco->iobuf;
1516
1517 rwco->ret = blk_co_ioctl(rwco->blk, rwco->offset,
1518 qiov->iov[0].iov_base);
1519 aio_wait_kick();
1520 }
1521
1522 int blk_ioctl(BlockBackend *blk, unsigned long int req, void *buf)
1523 {
1524 return blk_prw(blk, req, buf, 0, blk_ioctl_entry, 0);
1525 }
1526
1527 static void blk_aio_ioctl_entry(void *opaque)
1528 {
1529 BlkAioEmAIOCB *acb = opaque;
1530 BlkRwCo *rwco = &acb->rwco;
1531
1532 rwco->ret = blk_co_ioctl(rwco->blk, rwco->offset, rwco->iobuf);
1533
1534 blk_aio_complete(acb);
1535 }
1536
1537 BlockAIOCB *blk_aio_ioctl(BlockBackend *blk, unsigned long int req, void *buf,
1538 BlockCompletionFunc *cb, void *opaque)
1539 {
1540 return blk_aio_prwv(blk, req, 0, buf, blk_aio_ioctl_entry, 0, cb, opaque);
1541 }
1542
1543 int blk_co_pdiscard(BlockBackend *blk, int64_t offset, int bytes)
1544 {
1545 int ret = blk_check_byte_request(blk, offset, bytes);
1546 if (ret < 0) {
1547 return ret;
1548 }
1549
1550 return bdrv_co_pdiscard(blk->root, offset, bytes);
1551 }
1552
1553 int blk_co_flush(BlockBackend *blk)
1554 {
1555 if (!blk_is_available(blk)) {
1556 return -ENOMEDIUM;
1557 }
1558
1559 return bdrv_co_flush(blk_bs(blk));
1560 }
1561
1562 static void blk_flush_entry(void *opaque)
1563 {
1564 BlkRwCo *rwco = opaque;
1565 rwco->ret = blk_co_flush(rwco->blk);
1566 aio_wait_kick();
1567 }
1568
1569 int blk_flush(BlockBackend *blk)
1570 {
1571 return blk_prw(blk, 0, NULL, 0, blk_flush_entry, 0);
1572 }
1573
1574 void blk_drain(BlockBackend *blk)
1575 {
1576 BlockDriverState *bs = blk_bs(blk);
1577
1578 if (bs) {
1579 bdrv_drained_begin(bs);
1580 }
1581
1582 /* We may have -ENOMEDIUM completions in flight */
1583 AIO_WAIT_WHILE(blk_get_aio_context(blk),
1584 atomic_mb_read(&blk->in_flight) > 0);
1585
1586 if (bs) {
1587 bdrv_drained_end(bs);
1588 }
1589 }
1590
1591 void blk_drain_all(void)
1592 {
1593 BlockBackend *blk = NULL;
1594
1595 bdrv_drain_all_begin();
1596
1597 while ((blk = blk_all_next(blk)) != NULL) {
1598 AioContext *ctx = blk_get_aio_context(blk);
1599
1600 aio_context_acquire(ctx);
1601
1602 /* We may have -ENOMEDIUM completions in flight */
1603 AIO_WAIT_WHILE(ctx, atomic_mb_read(&blk->in_flight) > 0);
1604
1605 aio_context_release(ctx);
1606 }
1607
1608 bdrv_drain_all_end();
1609 }
1610
1611 void blk_set_on_error(BlockBackend *blk, BlockdevOnError on_read_error,
1612 BlockdevOnError on_write_error)
1613 {
1614 blk->on_read_error = on_read_error;
1615 blk->on_write_error = on_write_error;
1616 }
1617
1618 BlockdevOnError blk_get_on_error(BlockBackend *blk, bool is_read)
1619 {
1620 return is_read ? blk->on_read_error : blk->on_write_error;
1621 }
1622
1623 BlockErrorAction blk_get_error_action(BlockBackend *blk, bool is_read,
1624 int error)
1625 {
1626 BlockdevOnError on_err = blk_get_on_error(blk, is_read);
1627
1628 switch (on_err) {
1629 case BLOCKDEV_ON_ERROR_ENOSPC:
1630 return (error == ENOSPC) ?
1631 BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT;
1632 case BLOCKDEV_ON_ERROR_STOP:
1633 return BLOCK_ERROR_ACTION_STOP;
1634 case BLOCKDEV_ON_ERROR_REPORT:
1635 return BLOCK_ERROR_ACTION_REPORT;
1636 case BLOCKDEV_ON_ERROR_IGNORE:
1637 return BLOCK_ERROR_ACTION_IGNORE;
1638 case BLOCKDEV_ON_ERROR_AUTO:
1639 default:
1640 abort();
1641 }
1642 }
1643
1644 static void send_qmp_error_event(BlockBackend *blk,
1645 BlockErrorAction action,
1646 bool is_read, int error)
1647 {
1648 IoOperationType optype;
1649 BlockDriverState *bs = blk_bs(blk);
1650
1651 optype = is_read ? IO_OPERATION_TYPE_READ : IO_OPERATION_TYPE_WRITE;
1652 qapi_event_send_block_io_error(blk_name(blk), !!bs,
1653 bs ? bdrv_get_node_name(bs) : NULL, optype,
1654 action, blk_iostatus_is_enabled(blk),
1655 error == ENOSPC, strerror(error));
1656 }
1657
1658 /* This is done by device models because, while the block layer knows
1659 * about the error, it does not know whether an operation comes from
1660 * the device or the block layer (from a job, for example).
1661 */
1662 void blk_error_action(BlockBackend *blk, BlockErrorAction action,
1663 bool is_read, int error)
1664 {
1665 assert(error >= 0);
1666
1667 if (action == BLOCK_ERROR_ACTION_STOP) {
1668 /* First set the iostatus, so that "info block" returns an iostatus
1669 * that matches the events raised so far (an additional error iostatus
1670 * is fine, but not a lost one).
1671 */
1672 blk_iostatus_set_err(blk, error);
1673
1674 /* Then raise the request to stop the VM and the event.
1675 * qemu_system_vmstop_request_prepare has two effects. First,
1676 * it ensures that the STOP event always comes after the
1677 * BLOCK_IO_ERROR event. Second, it ensures that even if management
1678 * can observe the STOP event and do a "cont" before the STOP
1679 * event is issued, the VM will not stop. In this case, vm_start()
1680 * also ensures that the STOP/RESUME pair of events is emitted.
1681 */
1682 qemu_system_vmstop_request_prepare();
1683 send_qmp_error_event(blk, action, is_read, error);
1684 qemu_system_vmstop_request(RUN_STATE_IO_ERROR);
1685 } else {
1686 send_qmp_error_event(blk, action, is_read, error);
1687 }
1688 }
1689
1690 bool blk_is_read_only(BlockBackend *blk)
1691 {
1692 BlockDriverState *bs = blk_bs(blk);
1693
1694 if (bs) {
1695 return bdrv_is_read_only(bs);
1696 } else {
1697 return blk->root_state.read_only;
1698 }
1699 }
1700
1701 bool blk_is_sg(BlockBackend *blk)
1702 {
1703 BlockDriverState *bs = blk_bs(blk);
1704
1705 if (!bs) {
1706 return false;
1707 }
1708
1709 return bdrv_is_sg(bs);
1710 }
1711
1712 bool blk_enable_write_cache(BlockBackend *blk)
1713 {
1714 return blk->enable_write_cache;
1715 }
1716
1717 void blk_set_enable_write_cache(BlockBackend *blk, bool wce)
1718 {
1719 blk->enable_write_cache = wce;
1720 }
1721
1722 void blk_invalidate_cache(BlockBackend *blk, Error **errp)
1723 {
1724 BlockDriverState *bs = blk_bs(blk);
1725
1726 if (!bs) {
1727 error_setg(errp, "Device '%s' has no medium", blk->name);
1728 return;
1729 }
1730
1731 bdrv_invalidate_cache(bs, errp);
1732 }
1733
1734 bool blk_is_inserted(BlockBackend *blk)
1735 {
1736 BlockDriverState *bs = blk_bs(blk);
1737
1738 return bs && bdrv_is_inserted(bs);
1739 }
1740
1741 bool blk_is_available(BlockBackend *blk)
1742 {
1743 return blk_is_inserted(blk) && !blk_dev_is_tray_open(blk);
1744 }
1745
1746 void blk_lock_medium(BlockBackend *blk, bool locked)
1747 {
1748 BlockDriverState *bs = blk_bs(blk);
1749
1750 if (bs) {
1751 bdrv_lock_medium(bs, locked);
1752 }
1753 }
1754
1755 void blk_eject(BlockBackend *blk, bool eject_flag)
1756 {
1757 BlockDriverState *bs = blk_bs(blk);
1758 char *id;
1759
1760 if (bs) {
1761 bdrv_eject(bs, eject_flag);
1762 }
1763
1764 /* Whether or not we ejected on the backend,
1765 * the frontend experienced a tray event. */
1766 id = blk_get_attached_dev_id(blk);
1767 qapi_event_send_device_tray_moved(blk_name(blk), id,
1768 eject_flag);
1769 g_free(id);
1770 }
1771
1772 int blk_get_flags(BlockBackend *blk)
1773 {
1774 BlockDriverState *bs = blk_bs(blk);
1775
1776 if (bs) {
1777 return bdrv_get_flags(bs);
1778 } else {
1779 return blk->root_state.open_flags;
1780 }
1781 }
1782
1783 /* Returns the minimum request alignment, in bytes; guaranteed nonzero */
1784 uint32_t blk_get_request_alignment(BlockBackend *blk)
1785 {
1786 BlockDriverState *bs = blk_bs(blk);
1787 return bs ? bs->bl.request_alignment : BDRV_SECTOR_SIZE;
1788 }
1789
1790 /* Returns the maximum transfer length, in bytes; guaranteed nonzero */
1791 uint32_t blk_get_max_transfer(BlockBackend *blk)
1792 {
1793 BlockDriverState *bs = blk_bs(blk);
1794 uint32_t max = 0;
1795
1796 if (bs) {
1797 max = bs->bl.max_transfer;
1798 }
1799 return MIN_NON_ZERO(max, INT_MAX);
1800 }
1801
1802 int blk_get_max_iov(BlockBackend *blk)
1803 {
1804 return blk->root->bs->bl.max_iov;
1805 }
1806
1807 void blk_set_guest_block_size(BlockBackend *blk, int align)
1808 {
1809 blk->guest_block_size = align;
1810 }
1811
1812 void *blk_try_blockalign(BlockBackend *blk, size_t size)
1813 {
1814 return qemu_try_blockalign(blk ? blk_bs(blk) : NULL, size);
1815 }
1816
1817 void *blk_blockalign(BlockBackend *blk, size_t size)
1818 {
1819 return qemu_blockalign(blk ? blk_bs(blk) : NULL, size);
1820 }
1821
1822 bool blk_op_is_blocked(BlockBackend *blk, BlockOpType op, Error **errp)
1823 {
1824 BlockDriverState *bs = blk_bs(blk);
1825
1826 if (!bs) {
1827 return false;
1828 }
1829
1830 return bdrv_op_is_blocked(bs, op, errp);
1831 }
1832
1833 void blk_op_unblock(BlockBackend *blk, BlockOpType op, Error *reason)
1834 {
1835 BlockDriverState *bs = blk_bs(blk);
1836
1837 if (bs) {
1838 bdrv_op_unblock(bs, op, reason);
1839 }
1840 }
1841
1842 void blk_op_block_all(BlockBackend *blk, Error *reason)
1843 {
1844 BlockDriverState *bs = blk_bs(blk);
1845
1846 if (bs) {
1847 bdrv_op_block_all(bs, reason);
1848 }
1849 }
1850
1851 void blk_op_unblock_all(BlockBackend *blk, Error *reason)
1852 {
1853 BlockDriverState *bs = blk_bs(blk);
1854
1855 if (bs) {
1856 bdrv_op_unblock_all(bs, reason);
1857 }
1858 }
1859
1860 AioContext *blk_get_aio_context(BlockBackend *blk)
1861 {
1862 BlockDriverState *bs = blk_bs(blk);
1863
1864 if (bs) {
1865 AioContext *ctx = bdrv_get_aio_context(blk_bs(blk));
1866 assert(ctx == blk->ctx);
1867 }
1868
1869 return blk->ctx;
1870 }
1871
1872 static AioContext *blk_aiocb_get_aio_context(BlockAIOCB *acb)
1873 {
1874 BlockBackendAIOCB *blk_acb = DO_UPCAST(BlockBackendAIOCB, common, acb);
1875 return blk_get_aio_context(blk_acb->blk);
1876 }
1877
1878 static int blk_do_set_aio_context(BlockBackend *blk, AioContext *new_context,
1879 bool update_root_node, Error **errp)
1880 {
1881 BlockDriverState *bs = blk_bs(blk);
1882 ThrottleGroupMember *tgm = &blk->public.throttle_group_member;
1883 int ret;
1884
1885 if (bs) {
1886 if (update_root_node) {
1887 ret = bdrv_child_try_set_aio_context(bs, new_context, blk->root,
1888 errp);
1889 if (ret < 0) {
1890 return ret;
1891 }
1892 }
1893 if (tgm->throttle_state) {
1894 bdrv_drained_begin(bs);
1895 throttle_group_detach_aio_context(tgm);
1896 throttle_group_attach_aio_context(tgm, new_context);
1897 bdrv_drained_end(bs);
1898 }
1899 }
1900
1901 blk->ctx = new_context;
1902 return 0;
1903 }
1904
1905 int blk_set_aio_context(BlockBackend *blk, AioContext *new_context,
1906 Error **errp)
1907 {
1908 return blk_do_set_aio_context(blk, new_context, true, errp);
1909 }
1910
1911 static bool blk_root_can_set_aio_ctx(BdrvChild *child, AioContext *ctx,
1912 GSList **ignore, Error **errp)
1913 {
1914 BlockBackend *blk = child->opaque;
1915
1916 if (blk->allow_aio_context_change) {
1917 return true;
1918 }
1919
1920 /* Only manually created BlockBackends that are not attached to anything
1921 * can change their AioContext without updating their user. */
1922 if (!blk->name || blk->dev) {
1923 /* TODO Add BB name/QOM path */
1924 error_setg(errp, "Cannot change iothread of active block backend");
1925 return false;
1926 }
1927
1928 return true;
1929 }
1930
1931 static void blk_root_set_aio_ctx(BdrvChild *child, AioContext *ctx,
1932 GSList **ignore)
1933 {
1934 BlockBackend *blk = child->opaque;
1935 blk_do_set_aio_context(blk, ctx, false, &error_abort);
1936 }
1937
1938 void blk_add_aio_context_notifier(BlockBackend *blk,
1939 void (*attached_aio_context)(AioContext *new_context, void *opaque),
1940 void (*detach_aio_context)(void *opaque), void *opaque)
1941 {
1942 BlockBackendAioNotifier *notifier;
1943 BlockDriverState *bs = blk_bs(blk);
1944
1945 notifier = g_new(BlockBackendAioNotifier, 1);
1946 notifier->attached_aio_context = attached_aio_context;
1947 notifier->detach_aio_context = detach_aio_context;
1948 notifier->opaque = opaque;
1949 QLIST_INSERT_HEAD(&blk->aio_notifiers, notifier, list);
1950
1951 if (bs) {
1952 bdrv_add_aio_context_notifier(bs, attached_aio_context,
1953 detach_aio_context, opaque);
1954 }
1955 }
1956
1957 void blk_remove_aio_context_notifier(BlockBackend *blk,
1958 void (*attached_aio_context)(AioContext *,
1959 void *),
1960 void (*detach_aio_context)(void *),
1961 void *opaque)
1962 {
1963 BlockBackendAioNotifier *notifier;
1964 BlockDriverState *bs = blk_bs(blk);
1965
1966 if (bs) {
1967 bdrv_remove_aio_context_notifier(bs, attached_aio_context,
1968 detach_aio_context, opaque);
1969 }
1970
1971 QLIST_FOREACH(notifier, &blk->aio_notifiers, list) {
1972 if (notifier->attached_aio_context == attached_aio_context &&
1973 notifier->detach_aio_context == detach_aio_context &&
1974 notifier->opaque == opaque) {
1975 QLIST_REMOVE(notifier, list);
1976 g_free(notifier);
1977 return;
1978 }
1979 }
1980
1981 abort();
1982 }
1983
1984 void blk_add_remove_bs_notifier(BlockBackend *blk, Notifier *notify)
1985 {
1986 notifier_list_add(&blk->remove_bs_notifiers, notify);
1987 }
1988
1989 void blk_add_insert_bs_notifier(BlockBackend *blk, Notifier *notify)
1990 {
1991 notifier_list_add(&blk->insert_bs_notifiers, notify);
1992 }
1993
1994 void blk_io_plug(BlockBackend *blk)
1995 {
1996 BlockDriverState *bs = blk_bs(blk);
1997
1998 if (bs) {
1999 bdrv_io_plug(bs);
2000 }
2001 }
2002
2003 void blk_io_unplug(BlockBackend *blk)
2004 {
2005 BlockDriverState *bs = blk_bs(blk);
2006
2007 if (bs) {
2008 bdrv_io_unplug(bs);
2009 }
2010 }
2011
2012 BlockAcctStats *blk_get_stats(BlockBackend *blk)
2013 {
2014 return &blk->stats;
2015 }
2016
2017 void *blk_aio_get(const AIOCBInfo *aiocb_info, BlockBackend *blk,
2018 BlockCompletionFunc *cb, void *opaque)
2019 {
2020 return qemu_aio_get(aiocb_info, blk_bs(blk), cb, opaque);
2021 }
2022
2023 int coroutine_fn blk_co_pwrite_zeroes(BlockBackend *blk, int64_t offset,
2024 int bytes, BdrvRequestFlags flags)
2025 {
2026 return blk_co_pwritev(blk, offset, bytes, NULL,
2027 flags | BDRV_REQ_ZERO_WRITE);
2028 }
2029
2030 int blk_pwrite_compressed(BlockBackend *blk, int64_t offset, const void *buf,
2031 int count)
2032 {
2033 return blk_prw(blk, offset, (void *) buf, count, blk_write_entry,
2034 BDRV_REQ_WRITE_COMPRESSED);
2035 }
2036
2037 int blk_truncate(BlockBackend *blk, int64_t offset, PreallocMode prealloc,
2038 Error **errp)
2039 {
2040 if (!blk_is_available(blk)) {
2041 error_setg(errp, "No medium inserted");
2042 return -ENOMEDIUM;
2043 }
2044
2045 return bdrv_truncate(blk->root, offset, prealloc, errp);
2046 }
2047
2048 static void blk_pdiscard_entry(void *opaque)
2049 {
2050 BlkRwCo *rwco = opaque;
2051 QEMUIOVector *qiov = rwco->iobuf;
2052
2053 rwco->ret = blk_co_pdiscard(rwco->blk, rwco->offset, qiov->size);
2054 aio_wait_kick();
2055 }
2056
2057 int blk_pdiscard(BlockBackend *blk, int64_t offset, int bytes)
2058 {
2059 return blk_prw(blk, offset, NULL, bytes, blk_pdiscard_entry, 0);
2060 }
2061
2062 int blk_save_vmstate(BlockBackend *blk, const uint8_t *buf,
2063 int64_t pos, int size)
2064 {
2065 int ret;
2066
2067 if (!blk_is_available(blk)) {
2068 return -ENOMEDIUM;
2069 }
2070
2071 ret = bdrv_save_vmstate(blk_bs(blk), buf, pos, size);
2072 if (ret < 0) {
2073 return ret;
2074 }
2075
2076 if (ret == size && !blk->enable_write_cache) {
2077 ret = bdrv_flush(blk_bs(blk));
2078 }
2079
2080 return ret < 0 ? ret : size;
2081 }
2082
2083 int blk_load_vmstate(BlockBackend *blk, uint8_t *buf, int64_t pos, int size)
2084 {
2085 if (!blk_is_available(blk)) {
2086 return -ENOMEDIUM;
2087 }
2088
2089 return bdrv_load_vmstate(blk_bs(blk), buf, pos, size);
2090 }
2091
2092 int blk_probe_blocksizes(BlockBackend *blk, BlockSizes *bsz)
2093 {
2094 if (!blk_is_available(blk)) {
2095 return -ENOMEDIUM;
2096 }
2097
2098 return bdrv_probe_blocksizes(blk_bs(blk), bsz);
2099 }
2100
2101 int blk_probe_geometry(BlockBackend *blk, HDGeometry *geo)
2102 {
2103 if (!blk_is_available(blk)) {
2104 return -ENOMEDIUM;
2105 }
2106
2107 return bdrv_probe_geometry(blk_bs(blk), geo);
2108 }
2109
2110 /*
2111 * Updates the BlockBackendRootState object with data from the currently
2112 * attached BlockDriverState.
2113 */
2114 void blk_update_root_state(BlockBackend *blk)
2115 {
2116 assert(blk->root);
2117
2118 blk->root_state.open_flags = blk->root->bs->open_flags;
2119 blk->root_state.read_only = blk->root->bs->read_only;
2120 blk->root_state.detect_zeroes = blk->root->bs->detect_zeroes;
2121 }
2122
2123 /*
2124 * Returns the detect-zeroes setting to be used for bdrv_open() of a
2125 * BlockDriverState which is supposed to inherit the root state.
2126 */
2127 bool blk_get_detect_zeroes_from_root_state(BlockBackend *blk)
2128 {
2129 return blk->root_state.detect_zeroes;
2130 }
2131
2132 /*
2133 * Returns the flags to be used for bdrv_open() of a BlockDriverState which is
2134 * supposed to inherit the root state.
2135 */
2136 int blk_get_open_flags_from_root_state(BlockBackend *blk)
2137 {
2138 int bs_flags;
2139
2140 bs_flags = blk->root_state.read_only ? 0 : BDRV_O_RDWR;
2141 bs_flags |= blk->root_state.open_flags & ~BDRV_O_RDWR;
2142
2143 return bs_flags;
2144 }
2145
2146 BlockBackendRootState *blk_get_root_state(BlockBackend *blk)
2147 {
2148 return &blk->root_state;
2149 }
2150
2151 int blk_commit_all(void)
2152 {
2153 BlockBackend *blk = NULL;
2154
2155 while ((blk = blk_all_next(blk)) != NULL) {
2156 AioContext *aio_context = blk_get_aio_context(blk);
2157
2158 aio_context_acquire(aio_context);
2159 if (blk_is_inserted(blk) && blk->root->bs->backing) {
2160 int ret = bdrv_commit(blk->root->bs);
2161 if (ret < 0) {
2162 aio_context_release(aio_context);
2163 return ret;
2164 }
2165 }
2166 aio_context_release(aio_context);
2167 }
2168 return 0;
2169 }
2170
2171
2172 /* throttling disk I/O limits */
2173 void blk_set_io_limits(BlockBackend *blk, ThrottleConfig *cfg)
2174 {
2175 throttle_group_config(&blk->public.throttle_group_member, cfg);
2176 }
2177
2178 void blk_io_limits_disable(BlockBackend *blk)
2179 {
2180 BlockDriverState *bs = blk_bs(blk);
2181 ThrottleGroupMember *tgm = &blk->public.throttle_group_member;
2182 assert(tgm->throttle_state);
2183 if (bs) {
2184 bdrv_drained_begin(bs);
2185 }
2186 throttle_group_unregister_tgm(tgm);
2187 if (bs) {
2188 bdrv_drained_end(bs);
2189 }
2190 }
2191
2192 /* should be called before blk_set_io_limits if a limit is set */
2193 void blk_io_limits_enable(BlockBackend *blk, const char *group)
2194 {
2195 assert(!blk->public.throttle_group_member.throttle_state);
2196 throttle_group_register_tgm(&blk->public.throttle_group_member,
2197 group, blk_get_aio_context(blk));
2198 }
2199
2200 void blk_io_limits_update_group(BlockBackend *blk, const char *group)
2201 {
2202 /* this BB is not part of any group */
2203 if (!blk->public.throttle_group_member.throttle_state) {
2204 return;
2205 }
2206
2207 /* this BB is a part of the same group than the one we want */
2208 if (!g_strcmp0(throttle_group_get_name(&blk->public.throttle_group_member),
2209 group)) {
2210 return;
2211 }
2212
2213 /* need to change the group this bs belong to */
2214 blk_io_limits_disable(blk);
2215 blk_io_limits_enable(blk, group);
2216 }
2217
2218 static void blk_root_drained_begin(BdrvChild *child)
2219 {
2220 BlockBackend *blk = child->opaque;
2221
2222 if (++blk->quiesce_counter == 1) {
2223 if (blk->dev_ops && blk->dev_ops->drained_begin) {
2224 blk->dev_ops->drained_begin(blk->dev_opaque);
2225 }
2226 }
2227
2228 /* Note that blk->root may not be accessible here yet if we are just
2229 * attaching to a BlockDriverState that is drained. Use child instead. */
2230
2231 if (atomic_fetch_inc(&blk->public.throttle_group_member.io_limits_disabled) == 0) {
2232 throttle_group_restart_tgm(&blk->public.throttle_group_member);
2233 }
2234 }
2235
2236 static bool blk_root_drained_poll(BdrvChild *child)
2237 {
2238 BlockBackend *blk = child->opaque;
2239 assert(blk->quiesce_counter);
2240 return !!blk->in_flight;
2241 }
2242
2243 static void blk_root_drained_end(BdrvChild *child)
2244 {
2245 BlockBackend *blk = child->opaque;
2246 assert(blk->quiesce_counter);
2247
2248 assert(blk->public.throttle_group_member.io_limits_disabled);
2249 atomic_dec(&blk->public.throttle_group_member.io_limits_disabled);
2250
2251 if (--blk->quiesce_counter == 0) {
2252 if (blk->dev_ops && blk->dev_ops->drained_end) {
2253 blk->dev_ops->drained_end(blk->dev_opaque);
2254 }
2255 }
2256 }
2257
2258 void blk_register_buf(BlockBackend *blk, void *host, size_t size)
2259 {
2260 bdrv_register_buf(blk_bs(blk), host, size);
2261 }
2262
2263 void blk_unregister_buf(BlockBackend *blk, void *host)
2264 {
2265 bdrv_unregister_buf(blk_bs(blk), host);
2266 }
2267
2268 int coroutine_fn blk_co_copy_range(BlockBackend *blk_in, int64_t off_in,
2269 BlockBackend *blk_out, int64_t off_out,
2270 int bytes, BdrvRequestFlags read_flags,
2271 BdrvRequestFlags write_flags)
2272 {
2273 int r;
2274 r = blk_check_byte_request(blk_in, off_in, bytes);
2275 if (r) {
2276 return r;
2277 }
2278 r = blk_check_byte_request(blk_out, off_out, bytes);
2279 if (r) {
2280 return r;
2281 }
2282 return bdrv_co_copy_range(blk_in->root, off_in,
2283 blk_out->root, off_out,
2284 bytes, read_flags, write_flags);
2285 }
2286
2287 const BdrvChild *blk_root(BlockBackend *blk)
2288 {
2289 return blk->root;
2290 }