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