xref: /openbmc/qemu/block/block-backend.c (revision 5540cb97f711d191bae6cde89a03a8a9c306b638)
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             g_free(id);
985         }
986     }
987 }
988 
989 static void blk_root_change_media(BdrvChild *child, bool load)
990 {
991     blk_dev_change_media_cb(child->opaque, load, NULL);
992 }
993 
994 /*
995  * Does @blk's attached device model have removable media?
996  * %true if no device model is attached.
997  */
998 bool blk_dev_has_removable_media(BlockBackend *blk)
999 {
1000     return !blk->dev || (blk->dev_ops && blk->dev_ops->change_media_cb);
1001 }
1002 
1003 /*
1004  * Does @blk's attached device model have a tray?
1005  */
1006 bool blk_dev_has_tray(BlockBackend *blk)
1007 {
1008     return blk->dev_ops && blk->dev_ops->is_tray_open;
1009 }
1010 
1011 /*
1012  * Notify @blk's attached device model of a media eject request.
1013  * If @force is true, the medium is about to be yanked out forcefully.
1014  */
1015 void blk_dev_eject_request(BlockBackend *blk, bool force)
1016 {
1017     if (blk->dev_ops && blk->dev_ops->eject_request_cb) {
1018         blk->dev_ops->eject_request_cb(blk->dev_opaque, force);
1019     }
1020 }
1021 
1022 /*
1023  * Does @blk's attached device model have a tray, and is it open?
1024  */
1025 bool blk_dev_is_tray_open(BlockBackend *blk)
1026 {
1027     if (blk_dev_has_tray(blk)) {
1028         return blk->dev_ops->is_tray_open(blk->dev_opaque);
1029     }
1030     return false;
1031 }
1032 
1033 /*
1034  * Does @blk's attached device model have the medium locked?
1035  * %false if the device model has no such lock.
1036  */
1037 bool blk_dev_is_medium_locked(BlockBackend *blk)
1038 {
1039     if (blk->dev_ops && blk->dev_ops->is_medium_locked) {
1040         return blk->dev_ops->is_medium_locked(blk->dev_opaque);
1041     }
1042     return false;
1043 }
1044 
1045 /*
1046  * Notify @blk's attached device model of a backend size change.
1047  */
1048 static void blk_root_resize(BdrvChild *child)
1049 {
1050     BlockBackend *blk = child->opaque;
1051 
1052     if (blk->dev_ops && blk->dev_ops->resize_cb) {
1053         blk->dev_ops->resize_cb(blk->dev_opaque);
1054     }
1055 }
1056 
1057 void blk_iostatus_enable(BlockBackend *blk)
1058 {
1059     blk->iostatus_enabled = true;
1060     blk->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
1061 }
1062 
1063 /* The I/O status is only enabled if the drive explicitly
1064  * enables it _and_ the VM is configured to stop on errors */
1065 bool blk_iostatus_is_enabled(const BlockBackend *blk)
1066 {
1067     return (blk->iostatus_enabled &&
1068            (blk->on_write_error == BLOCKDEV_ON_ERROR_ENOSPC ||
1069             blk->on_write_error == BLOCKDEV_ON_ERROR_STOP   ||
1070             blk->on_read_error == BLOCKDEV_ON_ERROR_STOP));
1071 }
1072 
1073 BlockDeviceIoStatus blk_iostatus(const BlockBackend *blk)
1074 {
1075     return blk->iostatus;
1076 }
1077 
1078 void blk_iostatus_disable(BlockBackend *blk)
1079 {
1080     blk->iostatus_enabled = false;
1081 }
1082 
1083 void blk_iostatus_reset(BlockBackend *blk)
1084 {
1085     if (blk_iostatus_is_enabled(blk)) {
1086         BlockDriverState *bs = blk_bs(blk);
1087         blk->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
1088         if (bs && bs->job) {
1089             block_job_iostatus_reset(bs->job);
1090         }
1091     }
1092 }
1093 
1094 void blk_iostatus_set_err(BlockBackend *blk, int error)
1095 {
1096     assert(blk_iostatus_is_enabled(blk));
1097     if (blk->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
1098         blk->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
1099                                           BLOCK_DEVICE_IO_STATUS_FAILED;
1100     }
1101 }
1102 
1103 void blk_set_allow_write_beyond_eof(BlockBackend *blk, bool allow)
1104 {
1105     blk->allow_write_beyond_eof = allow;
1106 }
1107 
1108 static int blk_check_byte_request(BlockBackend *blk, int64_t offset,
1109                                   size_t size)
1110 {
1111     int64_t len;
1112 
1113     if (size > INT_MAX) {
1114         return -EIO;
1115     }
1116 
1117     if (!blk_is_available(blk)) {
1118         return -ENOMEDIUM;
1119     }
1120 
1121     if (offset < 0) {
1122         return -EIO;
1123     }
1124 
1125     if (!blk->allow_write_beyond_eof) {
1126         len = blk_getlength(blk);
1127         if (len < 0) {
1128             return len;
1129         }
1130 
1131         if (offset > len || len - offset < size) {
1132             return -EIO;
1133         }
1134     }
1135 
1136     return 0;
1137 }
1138 
1139 int coroutine_fn blk_co_preadv(BlockBackend *blk, int64_t offset,
1140                                unsigned int bytes, QEMUIOVector *qiov,
1141                                BdrvRequestFlags flags)
1142 {
1143     int ret;
1144     BlockDriverState *bs = blk_bs(blk);
1145 
1146     trace_blk_co_preadv(blk, bs, offset, bytes, flags);
1147 
1148     ret = blk_check_byte_request(blk, offset, bytes);
1149     if (ret < 0) {
1150         return ret;
1151     }
1152 
1153     bdrv_inc_in_flight(bs);
1154 
1155     /* throttling disk I/O */
1156     if (blk->public.throttle_group_member.throttle_state) {
1157         throttle_group_co_io_limits_intercept(&blk->public.throttle_group_member,
1158                 bytes, false);
1159     }
1160 
1161     ret = bdrv_co_preadv(blk->root, offset, bytes, qiov, flags);
1162     bdrv_dec_in_flight(bs);
1163     return ret;
1164 }
1165 
1166 int coroutine_fn blk_co_pwritev(BlockBackend *blk, int64_t offset,
1167                                 unsigned int bytes, QEMUIOVector *qiov,
1168                                 BdrvRequestFlags flags)
1169 {
1170     int ret;
1171     BlockDriverState *bs = blk_bs(blk);
1172 
1173     trace_blk_co_pwritev(blk, bs, offset, bytes, flags);
1174 
1175     ret = blk_check_byte_request(blk, offset, bytes);
1176     if (ret < 0) {
1177         return ret;
1178     }
1179 
1180     bdrv_inc_in_flight(bs);
1181     /* throttling disk I/O */
1182     if (blk->public.throttle_group_member.throttle_state) {
1183         throttle_group_co_io_limits_intercept(&blk->public.throttle_group_member,
1184                 bytes, true);
1185     }
1186 
1187     if (!blk->enable_write_cache) {
1188         flags |= BDRV_REQ_FUA;
1189     }
1190 
1191     ret = bdrv_co_pwritev(blk->root, offset, bytes, qiov, flags);
1192     bdrv_dec_in_flight(bs);
1193     return ret;
1194 }
1195 
1196 typedef struct BlkRwCo {
1197     BlockBackend *blk;
1198     int64_t offset;
1199     void *iobuf;
1200     int ret;
1201     BdrvRequestFlags flags;
1202 } BlkRwCo;
1203 
1204 static void blk_read_entry(void *opaque)
1205 {
1206     BlkRwCo *rwco = opaque;
1207     QEMUIOVector *qiov = rwco->iobuf;
1208 
1209     rwco->ret = blk_co_preadv(rwco->blk, rwco->offset, qiov->size,
1210                               qiov, rwco->flags);
1211 }
1212 
1213 static void blk_write_entry(void *opaque)
1214 {
1215     BlkRwCo *rwco = opaque;
1216     QEMUIOVector *qiov = rwco->iobuf;
1217 
1218     rwco->ret = blk_co_pwritev(rwco->blk, rwco->offset, qiov->size,
1219                                qiov, rwco->flags);
1220 }
1221 
1222 static int blk_prw(BlockBackend *blk, int64_t offset, uint8_t *buf,
1223                    int64_t bytes, CoroutineEntry co_entry,
1224                    BdrvRequestFlags flags)
1225 {
1226     QEMUIOVector qiov;
1227     struct iovec iov;
1228     BlkRwCo rwco;
1229 
1230     iov = (struct iovec) {
1231         .iov_base = buf,
1232         .iov_len = bytes,
1233     };
1234     qemu_iovec_init_external(&qiov, &iov, 1);
1235 
1236     rwco = (BlkRwCo) {
1237         .blk    = blk,
1238         .offset = offset,
1239         .iobuf  = &qiov,
1240         .flags  = flags,
1241         .ret    = NOT_DONE,
1242     };
1243 
1244     if (qemu_in_coroutine()) {
1245         /* Fast-path if already in coroutine context */
1246         co_entry(&rwco);
1247     } else {
1248         Coroutine *co = qemu_coroutine_create(co_entry, &rwco);
1249         bdrv_coroutine_enter(blk_bs(blk), co);
1250         BDRV_POLL_WHILE(blk_bs(blk), rwco.ret == NOT_DONE);
1251     }
1252 
1253     return rwco.ret;
1254 }
1255 
1256 int blk_pread_unthrottled(BlockBackend *blk, int64_t offset, uint8_t *buf,
1257                           int count)
1258 {
1259     int ret;
1260 
1261     ret = blk_check_byte_request(blk, offset, count);
1262     if (ret < 0) {
1263         return ret;
1264     }
1265 
1266     blk_root_drained_begin(blk->root);
1267     ret = blk_pread(blk, offset, buf, count);
1268     blk_root_drained_end(blk->root);
1269     return ret;
1270 }
1271 
1272 int blk_pwrite_zeroes(BlockBackend *blk, int64_t offset,
1273                       int bytes, BdrvRequestFlags flags)
1274 {
1275     return blk_prw(blk, offset, NULL, bytes, blk_write_entry,
1276                    flags | BDRV_REQ_ZERO_WRITE);
1277 }
1278 
1279 int blk_make_zero(BlockBackend *blk, BdrvRequestFlags flags)
1280 {
1281     return bdrv_make_zero(blk->root, flags);
1282 }
1283 
1284 static void blk_inc_in_flight(BlockBackend *blk)
1285 {
1286     atomic_inc(&blk->in_flight);
1287 }
1288 
1289 static void blk_dec_in_flight(BlockBackend *blk)
1290 {
1291     atomic_dec(&blk->in_flight);
1292     aio_wait_kick(&blk->wait);
1293 }
1294 
1295 static void error_callback_bh(void *opaque)
1296 {
1297     struct BlockBackendAIOCB *acb = opaque;
1298 
1299     blk_dec_in_flight(acb->blk);
1300     acb->common.cb(acb->common.opaque, acb->ret);
1301     qemu_aio_unref(acb);
1302 }
1303 
1304 BlockAIOCB *blk_abort_aio_request(BlockBackend *blk,
1305                                   BlockCompletionFunc *cb,
1306                                   void *opaque, int ret)
1307 {
1308     struct BlockBackendAIOCB *acb;
1309 
1310     blk_inc_in_flight(blk);
1311     acb = blk_aio_get(&block_backend_aiocb_info, blk, cb, opaque);
1312     acb->blk = blk;
1313     acb->ret = ret;
1314 
1315     aio_bh_schedule_oneshot(blk_get_aio_context(blk), error_callback_bh, acb);
1316     return &acb->common;
1317 }
1318 
1319 typedef struct BlkAioEmAIOCB {
1320     BlockAIOCB common;
1321     BlkRwCo rwco;
1322     int bytes;
1323     bool has_returned;
1324 } BlkAioEmAIOCB;
1325 
1326 static const AIOCBInfo blk_aio_em_aiocb_info = {
1327     .aiocb_size         = sizeof(BlkAioEmAIOCB),
1328 };
1329 
1330 static void blk_aio_complete(BlkAioEmAIOCB *acb)
1331 {
1332     if (acb->has_returned) {
1333         blk_dec_in_flight(acb->rwco.blk);
1334         acb->common.cb(acb->common.opaque, acb->rwco.ret);
1335         qemu_aio_unref(acb);
1336     }
1337 }
1338 
1339 static void blk_aio_complete_bh(void *opaque)
1340 {
1341     BlkAioEmAIOCB *acb = opaque;
1342     assert(acb->has_returned);
1343     blk_aio_complete(acb);
1344 }
1345 
1346 static BlockAIOCB *blk_aio_prwv(BlockBackend *blk, int64_t offset, int bytes,
1347                                 void *iobuf, CoroutineEntry co_entry,
1348                                 BdrvRequestFlags flags,
1349                                 BlockCompletionFunc *cb, void *opaque)
1350 {
1351     BlkAioEmAIOCB *acb;
1352     Coroutine *co;
1353 
1354     blk_inc_in_flight(blk);
1355     acb = blk_aio_get(&blk_aio_em_aiocb_info, blk, cb, opaque);
1356     acb->rwco = (BlkRwCo) {
1357         .blk    = blk,
1358         .offset = offset,
1359         .iobuf  = iobuf,
1360         .flags  = flags,
1361         .ret    = NOT_DONE,
1362     };
1363     acb->bytes = bytes;
1364     acb->has_returned = false;
1365 
1366     co = qemu_coroutine_create(co_entry, acb);
1367     bdrv_coroutine_enter(blk_bs(blk), co);
1368 
1369     acb->has_returned = true;
1370     if (acb->rwco.ret != NOT_DONE) {
1371         aio_bh_schedule_oneshot(blk_get_aio_context(blk),
1372                                 blk_aio_complete_bh, acb);
1373     }
1374 
1375     return &acb->common;
1376 }
1377 
1378 static void blk_aio_read_entry(void *opaque)
1379 {
1380     BlkAioEmAIOCB *acb = opaque;
1381     BlkRwCo *rwco = &acb->rwco;
1382     QEMUIOVector *qiov = rwco->iobuf;
1383 
1384     assert(qiov->size == acb->bytes);
1385     rwco->ret = blk_co_preadv(rwco->blk, rwco->offset, acb->bytes,
1386                               qiov, rwco->flags);
1387     blk_aio_complete(acb);
1388 }
1389 
1390 static void blk_aio_write_entry(void *opaque)
1391 {
1392     BlkAioEmAIOCB *acb = opaque;
1393     BlkRwCo *rwco = &acb->rwco;
1394     QEMUIOVector *qiov = rwco->iobuf;
1395 
1396     assert(!qiov || qiov->size == acb->bytes);
1397     rwco->ret = blk_co_pwritev(rwco->blk, rwco->offset, acb->bytes,
1398                                qiov, rwco->flags);
1399     blk_aio_complete(acb);
1400 }
1401 
1402 BlockAIOCB *blk_aio_pwrite_zeroes(BlockBackend *blk, int64_t offset,
1403                                   int count, BdrvRequestFlags flags,
1404                                   BlockCompletionFunc *cb, void *opaque)
1405 {
1406     return blk_aio_prwv(blk, offset, count, NULL, blk_aio_write_entry,
1407                         flags | BDRV_REQ_ZERO_WRITE, cb, opaque);
1408 }
1409 
1410 int blk_pread(BlockBackend *blk, int64_t offset, void *buf, int count)
1411 {
1412     int ret = blk_prw(blk, offset, buf, count, blk_read_entry, 0);
1413     if (ret < 0) {
1414         return ret;
1415     }
1416     return count;
1417 }
1418 
1419 int blk_pwrite(BlockBackend *blk, int64_t offset, const void *buf, int count,
1420                BdrvRequestFlags flags)
1421 {
1422     int ret = blk_prw(blk, offset, (void *) buf, count, blk_write_entry,
1423                       flags);
1424     if (ret < 0) {
1425         return ret;
1426     }
1427     return count;
1428 }
1429 
1430 int64_t blk_getlength(BlockBackend *blk)
1431 {
1432     if (!blk_is_available(blk)) {
1433         return -ENOMEDIUM;
1434     }
1435 
1436     return bdrv_getlength(blk_bs(blk));
1437 }
1438 
1439 void blk_get_geometry(BlockBackend *blk, uint64_t *nb_sectors_ptr)
1440 {
1441     if (!blk_bs(blk)) {
1442         *nb_sectors_ptr = 0;
1443     } else {
1444         bdrv_get_geometry(blk_bs(blk), nb_sectors_ptr);
1445     }
1446 }
1447 
1448 int64_t blk_nb_sectors(BlockBackend *blk)
1449 {
1450     if (!blk_is_available(blk)) {
1451         return -ENOMEDIUM;
1452     }
1453 
1454     return bdrv_nb_sectors(blk_bs(blk));
1455 }
1456 
1457 BlockAIOCB *blk_aio_preadv(BlockBackend *blk, int64_t offset,
1458                            QEMUIOVector *qiov, BdrvRequestFlags flags,
1459                            BlockCompletionFunc *cb, void *opaque)
1460 {
1461     return blk_aio_prwv(blk, offset, qiov->size, qiov,
1462                         blk_aio_read_entry, flags, cb, opaque);
1463 }
1464 
1465 BlockAIOCB *blk_aio_pwritev(BlockBackend *blk, int64_t offset,
1466                             QEMUIOVector *qiov, BdrvRequestFlags flags,
1467                             BlockCompletionFunc *cb, void *opaque)
1468 {
1469     return blk_aio_prwv(blk, offset, qiov->size, qiov,
1470                         blk_aio_write_entry, flags, cb, opaque);
1471 }
1472 
1473 static void blk_aio_flush_entry(void *opaque)
1474 {
1475     BlkAioEmAIOCB *acb = opaque;
1476     BlkRwCo *rwco = &acb->rwco;
1477 
1478     rwco->ret = blk_co_flush(rwco->blk);
1479     blk_aio_complete(acb);
1480 }
1481 
1482 BlockAIOCB *blk_aio_flush(BlockBackend *blk,
1483                           BlockCompletionFunc *cb, void *opaque)
1484 {
1485     return blk_aio_prwv(blk, 0, 0, NULL, blk_aio_flush_entry, 0, cb, opaque);
1486 }
1487 
1488 static void blk_aio_pdiscard_entry(void *opaque)
1489 {
1490     BlkAioEmAIOCB *acb = opaque;
1491     BlkRwCo *rwco = &acb->rwco;
1492 
1493     rwco->ret = blk_co_pdiscard(rwco->blk, rwco->offset, acb->bytes);
1494     blk_aio_complete(acb);
1495 }
1496 
1497 BlockAIOCB *blk_aio_pdiscard(BlockBackend *blk,
1498                              int64_t offset, int bytes,
1499                              BlockCompletionFunc *cb, void *opaque)
1500 {
1501     return blk_aio_prwv(blk, offset, bytes, NULL, blk_aio_pdiscard_entry, 0,
1502                         cb, opaque);
1503 }
1504 
1505 void blk_aio_cancel(BlockAIOCB *acb)
1506 {
1507     bdrv_aio_cancel(acb);
1508 }
1509 
1510 void blk_aio_cancel_async(BlockAIOCB *acb)
1511 {
1512     bdrv_aio_cancel_async(acb);
1513 }
1514 
1515 int blk_co_ioctl(BlockBackend *blk, unsigned long int req, void *buf)
1516 {
1517     if (!blk_is_available(blk)) {
1518         return -ENOMEDIUM;
1519     }
1520 
1521     return bdrv_co_ioctl(blk_bs(blk), req, buf);
1522 }
1523 
1524 static void blk_ioctl_entry(void *opaque)
1525 {
1526     BlkRwCo *rwco = opaque;
1527     QEMUIOVector *qiov = rwco->iobuf;
1528 
1529     rwco->ret = blk_co_ioctl(rwco->blk, rwco->offset,
1530                              qiov->iov[0].iov_base);
1531 }
1532 
1533 int blk_ioctl(BlockBackend *blk, unsigned long int req, void *buf)
1534 {
1535     return blk_prw(blk, req, buf, 0, blk_ioctl_entry, 0);
1536 }
1537 
1538 static void blk_aio_ioctl_entry(void *opaque)
1539 {
1540     BlkAioEmAIOCB *acb = opaque;
1541     BlkRwCo *rwco = &acb->rwco;
1542 
1543     rwco->ret = blk_co_ioctl(rwco->blk, rwco->offset, rwco->iobuf);
1544 
1545     blk_aio_complete(acb);
1546 }
1547 
1548 BlockAIOCB *blk_aio_ioctl(BlockBackend *blk, unsigned long int req, void *buf,
1549                           BlockCompletionFunc *cb, void *opaque)
1550 {
1551     return blk_aio_prwv(blk, req, 0, buf, blk_aio_ioctl_entry, 0, cb, opaque);
1552 }
1553 
1554 int blk_co_pdiscard(BlockBackend *blk, int64_t offset, int bytes)
1555 {
1556     int ret = blk_check_byte_request(blk, offset, bytes);
1557     if (ret < 0) {
1558         return ret;
1559     }
1560 
1561     return bdrv_co_pdiscard(blk->root, offset, bytes);
1562 }
1563 
1564 int blk_co_flush(BlockBackend *blk)
1565 {
1566     if (!blk_is_available(blk)) {
1567         return -ENOMEDIUM;
1568     }
1569 
1570     return bdrv_co_flush(blk_bs(blk));
1571 }
1572 
1573 static void blk_flush_entry(void *opaque)
1574 {
1575     BlkRwCo *rwco = opaque;
1576     rwco->ret = blk_co_flush(rwco->blk);
1577 }
1578 
1579 int blk_flush(BlockBackend *blk)
1580 {
1581     return blk_prw(blk, 0, NULL, 0, blk_flush_entry, 0);
1582 }
1583 
1584 void blk_drain(BlockBackend *blk)
1585 {
1586     BlockDriverState *bs = blk_bs(blk);
1587 
1588     if (bs) {
1589         bdrv_drained_begin(bs);
1590     }
1591 
1592     /* We may have -ENOMEDIUM completions in flight */
1593     AIO_WAIT_WHILE(&blk->wait,
1594             blk_get_aio_context(blk),
1595             atomic_mb_read(&blk->in_flight) > 0);
1596 
1597     if (bs) {
1598         bdrv_drained_end(bs);
1599     }
1600 }
1601 
1602 void blk_drain_all(void)
1603 {
1604     BlockBackend *blk = NULL;
1605 
1606     bdrv_drain_all_begin();
1607 
1608     while ((blk = blk_all_next(blk)) != NULL) {
1609         AioContext *ctx = blk_get_aio_context(blk);
1610 
1611         aio_context_acquire(ctx);
1612 
1613         /* We may have -ENOMEDIUM completions in flight */
1614         AIO_WAIT_WHILE(&blk->wait, ctx,
1615                 atomic_mb_read(&blk->in_flight) > 0);
1616 
1617         aio_context_release(ctx);
1618     }
1619 
1620     bdrv_drain_all_end();
1621 }
1622 
1623 void blk_set_on_error(BlockBackend *blk, BlockdevOnError on_read_error,
1624                       BlockdevOnError on_write_error)
1625 {
1626     blk->on_read_error = on_read_error;
1627     blk->on_write_error = on_write_error;
1628 }
1629 
1630 BlockdevOnError blk_get_on_error(BlockBackend *blk, bool is_read)
1631 {
1632     return is_read ? blk->on_read_error : blk->on_write_error;
1633 }
1634 
1635 BlockErrorAction blk_get_error_action(BlockBackend *blk, bool is_read,
1636                                       int error)
1637 {
1638     BlockdevOnError on_err = blk_get_on_error(blk, is_read);
1639 
1640     switch (on_err) {
1641     case BLOCKDEV_ON_ERROR_ENOSPC:
1642         return (error == ENOSPC) ?
1643                BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT;
1644     case BLOCKDEV_ON_ERROR_STOP:
1645         return BLOCK_ERROR_ACTION_STOP;
1646     case BLOCKDEV_ON_ERROR_REPORT:
1647         return BLOCK_ERROR_ACTION_REPORT;
1648     case BLOCKDEV_ON_ERROR_IGNORE:
1649         return BLOCK_ERROR_ACTION_IGNORE;
1650     case BLOCKDEV_ON_ERROR_AUTO:
1651     default:
1652         abort();
1653     }
1654 }
1655 
1656 static void send_qmp_error_event(BlockBackend *blk,
1657                                  BlockErrorAction action,
1658                                  bool is_read, int error)
1659 {
1660     IoOperationType optype;
1661     BlockDriverState *bs = blk_bs(blk);
1662 
1663     optype = is_read ? IO_OPERATION_TYPE_READ : IO_OPERATION_TYPE_WRITE;
1664     qapi_event_send_block_io_error(blk_name(blk), !!bs,
1665                                    bs ? bdrv_get_node_name(bs) : NULL, optype,
1666                                    action, blk_iostatus_is_enabled(blk),
1667                                    error == ENOSPC, strerror(error));
1668 }
1669 
1670 /* This is done by device models because, while the block layer knows
1671  * about the error, it does not know whether an operation comes from
1672  * the device or the block layer (from a job, for example).
1673  */
1674 void blk_error_action(BlockBackend *blk, BlockErrorAction action,
1675                       bool is_read, int error)
1676 {
1677     assert(error >= 0);
1678 
1679     if (action == BLOCK_ERROR_ACTION_STOP) {
1680         /* First set the iostatus, so that "info block" returns an iostatus
1681          * that matches the events raised so far (an additional error iostatus
1682          * is fine, but not a lost one).
1683          */
1684         blk_iostatus_set_err(blk, error);
1685 
1686         /* Then raise the request to stop the VM and the event.
1687          * qemu_system_vmstop_request_prepare has two effects.  First,
1688          * it ensures that the STOP event always comes after the
1689          * BLOCK_IO_ERROR event.  Second, it ensures that even if management
1690          * can observe the STOP event and do a "cont" before the STOP
1691          * event is issued, the VM will not stop.  In this case, vm_start()
1692          * also ensures that the STOP/RESUME pair of events is emitted.
1693          */
1694         qemu_system_vmstop_request_prepare();
1695         send_qmp_error_event(blk, action, is_read, error);
1696         qemu_system_vmstop_request(RUN_STATE_IO_ERROR);
1697     } else {
1698         send_qmp_error_event(blk, action, is_read, error);
1699     }
1700 }
1701 
1702 int blk_is_read_only(BlockBackend *blk)
1703 {
1704     BlockDriverState *bs = blk_bs(blk);
1705 
1706     if (bs) {
1707         return bdrv_is_read_only(bs);
1708     } else {
1709         return blk->root_state.read_only;
1710     }
1711 }
1712 
1713 int blk_is_sg(BlockBackend *blk)
1714 {
1715     BlockDriverState *bs = blk_bs(blk);
1716 
1717     if (!bs) {
1718         return 0;
1719     }
1720 
1721     return bdrv_is_sg(bs);
1722 }
1723 
1724 int blk_enable_write_cache(BlockBackend *blk)
1725 {
1726     return blk->enable_write_cache;
1727 }
1728 
1729 void blk_set_enable_write_cache(BlockBackend *blk, bool wce)
1730 {
1731     blk->enable_write_cache = wce;
1732 }
1733 
1734 void blk_invalidate_cache(BlockBackend *blk, Error **errp)
1735 {
1736     BlockDriverState *bs = blk_bs(blk);
1737 
1738     if (!bs) {
1739         error_setg(errp, "Device '%s' has no medium", blk->name);
1740         return;
1741     }
1742 
1743     bdrv_invalidate_cache(bs, errp);
1744 }
1745 
1746 bool blk_is_inserted(BlockBackend *blk)
1747 {
1748     BlockDriverState *bs = blk_bs(blk);
1749 
1750     return bs && bdrv_is_inserted(bs);
1751 }
1752 
1753 bool blk_is_available(BlockBackend *blk)
1754 {
1755     return blk_is_inserted(blk) && !blk_dev_is_tray_open(blk);
1756 }
1757 
1758 void blk_lock_medium(BlockBackend *blk, bool locked)
1759 {
1760     BlockDriverState *bs = blk_bs(blk);
1761 
1762     if (bs) {
1763         bdrv_lock_medium(bs, locked);
1764     }
1765 }
1766 
1767 void blk_eject(BlockBackend *blk, bool eject_flag)
1768 {
1769     BlockDriverState *bs = blk_bs(blk);
1770     char *id;
1771 
1772     /* blk_eject is only called by qdevified devices */
1773     assert(!blk->legacy_dev);
1774 
1775     if (bs) {
1776         bdrv_eject(bs, eject_flag);
1777     }
1778 
1779     /* Whether or not we ejected on the backend,
1780      * the frontend experienced a tray event. */
1781     id = blk_get_attached_dev_id(blk);
1782     qapi_event_send_device_tray_moved(blk_name(blk), id,
1783                                       eject_flag);
1784     g_free(id);
1785 }
1786 
1787 int blk_get_flags(BlockBackend *blk)
1788 {
1789     BlockDriverState *bs = blk_bs(blk);
1790 
1791     if (bs) {
1792         return bdrv_get_flags(bs);
1793     } else {
1794         return blk->root_state.open_flags;
1795     }
1796 }
1797 
1798 /* Returns the maximum transfer length, in bytes; guaranteed nonzero */
1799 uint32_t blk_get_max_transfer(BlockBackend *blk)
1800 {
1801     BlockDriverState *bs = blk_bs(blk);
1802     uint32_t max = 0;
1803 
1804     if (bs) {
1805         max = bs->bl.max_transfer;
1806     }
1807     return MIN_NON_ZERO(max, INT_MAX);
1808 }
1809 
1810 int blk_get_max_iov(BlockBackend *blk)
1811 {
1812     return blk->root->bs->bl.max_iov;
1813 }
1814 
1815 void blk_set_guest_block_size(BlockBackend *blk, int align)
1816 {
1817     blk->guest_block_size = align;
1818 }
1819 
1820 void *blk_try_blockalign(BlockBackend *blk, size_t size)
1821 {
1822     return qemu_try_blockalign(blk ? blk_bs(blk) : NULL, size);
1823 }
1824 
1825 void *blk_blockalign(BlockBackend *blk, size_t size)
1826 {
1827     return qemu_blockalign(blk ? blk_bs(blk) : NULL, size);
1828 }
1829 
1830 bool blk_op_is_blocked(BlockBackend *blk, BlockOpType op, Error **errp)
1831 {
1832     BlockDriverState *bs = blk_bs(blk);
1833 
1834     if (!bs) {
1835         return false;
1836     }
1837 
1838     return bdrv_op_is_blocked(bs, op, errp);
1839 }
1840 
1841 void blk_op_unblock(BlockBackend *blk, BlockOpType op, Error *reason)
1842 {
1843     BlockDriverState *bs = blk_bs(blk);
1844 
1845     if (bs) {
1846         bdrv_op_unblock(bs, op, reason);
1847     }
1848 }
1849 
1850 void blk_op_block_all(BlockBackend *blk, Error *reason)
1851 {
1852     BlockDriverState *bs = blk_bs(blk);
1853 
1854     if (bs) {
1855         bdrv_op_block_all(bs, reason);
1856     }
1857 }
1858 
1859 void blk_op_unblock_all(BlockBackend *blk, Error *reason)
1860 {
1861     BlockDriverState *bs = blk_bs(blk);
1862 
1863     if (bs) {
1864         bdrv_op_unblock_all(bs, reason);
1865     }
1866 }
1867 
1868 AioContext *blk_get_aio_context(BlockBackend *blk)
1869 {
1870     return bdrv_get_aio_context(blk_bs(blk));
1871 }
1872 
1873 static AioContext *blk_aiocb_get_aio_context(BlockAIOCB *acb)
1874 {
1875     BlockBackendAIOCB *blk_acb = DO_UPCAST(BlockBackendAIOCB, common, acb);
1876     return blk_get_aio_context(blk_acb->blk);
1877 }
1878 
1879 void blk_set_aio_context(BlockBackend *blk, AioContext *new_context)
1880 {
1881     BlockDriverState *bs = blk_bs(blk);
1882     ThrottleGroupMember *tgm = &blk->public.throttle_group_member;
1883 
1884     if (bs) {
1885         if (tgm->throttle_state) {
1886             bdrv_drained_begin(bs);
1887             throttle_group_detach_aio_context(tgm);
1888             throttle_group_attach_aio_context(tgm, new_context);
1889             bdrv_drained_end(bs);
1890         }
1891         bdrv_set_aio_context(bs, new_context);
1892     }
1893 }
1894 
1895 void blk_add_aio_context_notifier(BlockBackend *blk,
1896         void (*attached_aio_context)(AioContext *new_context, void *opaque),
1897         void (*detach_aio_context)(void *opaque), void *opaque)
1898 {
1899     BlockBackendAioNotifier *notifier;
1900     BlockDriverState *bs = blk_bs(blk);
1901 
1902     notifier = g_new(BlockBackendAioNotifier, 1);
1903     notifier->attached_aio_context = attached_aio_context;
1904     notifier->detach_aio_context = detach_aio_context;
1905     notifier->opaque = opaque;
1906     QLIST_INSERT_HEAD(&blk->aio_notifiers, notifier, list);
1907 
1908     if (bs) {
1909         bdrv_add_aio_context_notifier(bs, attached_aio_context,
1910                                       detach_aio_context, opaque);
1911     }
1912 }
1913 
1914 void blk_remove_aio_context_notifier(BlockBackend *blk,
1915                                      void (*attached_aio_context)(AioContext *,
1916                                                                   void *),
1917                                      void (*detach_aio_context)(void *),
1918                                      void *opaque)
1919 {
1920     BlockBackendAioNotifier *notifier;
1921     BlockDriverState *bs = blk_bs(blk);
1922 
1923     if (bs) {
1924         bdrv_remove_aio_context_notifier(bs, attached_aio_context,
1925                                          detach_aio_context, opaque);
1926     }
1927 
1928     QLIST_FOREACH(notifier, &blk->aio_notifiers, list) {
1929         if (notifier->attached_aio_context == attached_aio_context &&
1930             notifier->detach_aio_context == detach_aio_context &&
1931             notifier->opaque == opaque) {
1932             QLIST_REMOVE(notifier, list);
1933             g_free(notifier);
1934             return;
1935         }
1936     }
1937 
1938     abort();
1939 }
1940 
1941 void blk_add_remove_bs_notifier(BlockBackend *blk, Notifier *notify)
1942 {
1943     notifier_list_add(&blk->remove_bs_notifiers, notify);
1944 }
1945 
1946 void blk_add_insert_bs_notifier(BlockBackend *blk, Notifier *notify)
1947 {
1948     notifier_list_add(&blk->insert_bs_notifiers, notify);
1949 }
1950 
1951 void blk_io_plug(BlockBackend *blk)
1952 {
1953     BlockDriverState *bs = blk_bs(blk);
1954 
1955     if (bs) {
1956         bdrv_io_plug(bs);
1957     }
1958 }
1959 
1960 void blk_io_unplug(BlockBackend *blk)
1961 {
1962     BlockDriverState *bs = blk_bs(blk);
1963 
1964     if (bs) {
1965         bdrv_io_unplug(bs);
1966     }
1967 }
1968 
1969 BlockAcctStats *blk_get_stats(BlockBackend *blk)
1970 {
1971     return &blk->stats;
1972 }
1973 
1974 void *blk_aio_get(const AIOCBInfo *aiocb_info, BlockBackend *blk,
1975                   BlockCompletionFunc *cb, void *opaque)
1976 {
1977     return qemu_aio_get(aiocb_info, blk_bs(blk), cb, opaque);
1978 }
1979 
1980 int coroutine_fn blk_co_pwrite_zeroes(BlockBackend *blk, int64_t offset,
1981                                       int bytes, BdrvRequestFlags flags)
1982 {
1983     return blk_co_pwritev(blk, offset, bytes, NULL,
1984                           flags | BDRV_REQ_ZERO_WRITE);
1985 }
1986 
1987 int blk_pwrite_compressed(BlockBackend *blk, int64_t offset, const void *buf,
1988                           int count)
1989 {
1990     return blk_prw(blk, offset, (void *) buf, count, blk_write_entry,
1991                    BDRV_REQ_WRITE_COMPRESSED);
1992 }
1993 
1994 int blk_truncate(BlockBackend *blk, int64_t offset, PreallocMode prealloc,
1995                  Error **errp)
1996 {
1997     if (!blk_is_available(blk)) {
1998         error_setg(errp, "No medium inserted");
1999         return -ENOMEDIUM;
2000     }
2001 
2002     return bdrv_truncate(blk->root, offset, prealloc, errp);
2003 }
2004 
2005 static void blk_pdiscard_entry(void *opaque)
2006 {
2007     BlkRwCo *rwco = opaque;
2008     QEMUIOVector *qiov = rwco->iobuf;
2009 
2010     rwco->ret = blk_co_pdiscard(rwco->blk, rwco->offset, qiov->size);
2011 }
2012 
2013 int blk_pdiscard(BlockBackend *blk, int64_t offset, int bytes)
2014 {
2015     return blk_prw(blk, offset, NULL, bytes, blk_pdiscard_entry, 0);
2016 }
2017 
2018 int blk_save_vmstate(BlockBackend *blk, const uint8_t *buf,
2019                      int64_t pos, int size)
2020 {
2021     int ret;
2022 
2023     if (!blk_is_available(blk)) {
2024         return -ENOMEDIUM;
2025     }
2026 
2027     ret = bdrv_save_vmstate(blk_bs(blk), buf, pos, size);
2028     if (ret < 0) {
2029         return ret;
2030     }
2031 
2032     if (ret == size && !blk->enable_write_cache) {
2033         ret = bdrv_flush(blk_bs(blk));
2034     }
2035 
2036     return ret < 0 ? ret : size;
2037 }
2038 
2039 int blk_load_vmstate(BlockBackend *blk, uint8_t *buf, int64_t pos, int size)
2040 {
2041     if (!blk_is_available(blk)) {
2042         return -ENOMEDIUM;
2043     }
2044 
2045     return bdrv_load_vmstate(blk_bs(blk), buf, pos, size);
2046 }
2047 
2048 int blk_probe_blocksizes(BlockBackend *blk, BlockSizes *bsz)
2049 {
2050     if (!blk_is_available(blk)) {
2051         return -ENOMEDIUM;
2052     }
2053 
2054     return bdrv_probe_blocksizes(blk_bs(blk), bsz);
2055 }
2056 
2057 int blk_probe_geometry(BlockBackend *blk, HDGeometry *geo)
2058 {
2059     if (!blk_is_available(blk)) {
2060         return -ENOMEDIUM;
2061     }
2062 
2063     return bdrv_probe_geometry(blk_bs(blk), geo);
2064 }
2065 
2066 /*
2067  * Updates the BlockBackendRootState object with data from the currently
2068  * attached BlockDriverState.
2069  */
2070 void blk_update_root_state(BlockBackend *blk)
2071 {
2072     assert(blk->root);
2073 
2074     blk->root_state.open_flags    = blk->root->bs->open_flags;
2075     blk->root_state.read_only     = blk->root->bs->read_only;
2076     blk->root_state.detect_zeroes = blk->root->bs->detect_zeroes;
2077 }
2078 
2079 /*
2080  * Returns the detect-zeroes setting to be used for bdrv_open() of a
2081  * BlockDriverState which is supposed to inherit the root state.
2082  */
2083 bool blk_get_detect_zeroes_from_root_state(BlockBackend *blk)
2084 {
2085     return blk->root_state.detect_zeroes;
2086 }
2087 
2088 /*
2089  * Returns the flags to be used for bdrv_open() of a BlockDriverState which is
2090  * supposed to inherit the root state.
2091  */
2092 int blk_get_open_flags_from_root_state(BlockBackend *blk)
2093 {
2094     int bs_flags;
2095 
2096     bs_flags = blk->root_state.read_only ? 0 : BDRV_O_RDWR;
2097     bs_flags |= blk->root_state.open_flags & ~BDRV_O_RDWR;
2098 
2099     return bs_flags;
2100 }
2101 
2102 BlockBackendRootState *blk_get_root_state(BlockBackend *blk)
2103 {
2104     return &blk->root_state;
2105 }
2106 
2107 int blk_commit_all(void)
2108 {
2109     BlockBackend *blk = NULL;
2110 
2111     while ((blk = blk_all_next(blk)) != NULL) {
2112         AioContext *aio_context = blk_get_aio_context(blk);
2113 
2114         aio_context_acquire(aio_context);
2115         if (blk_is_inserted(blk) && blk->root->bs->backing) {
2116             int ret = bdrv_commit(blk->root->bs);
2117             if (ret < 0) {
2118                 aio_context_release(aio_context);
2119                 return ret;
2120             }
2121         }
2122         aio_context_release(aio_context);
2123     }
2124     return 0;
2125 }
2126 
2127 
2128 /* throttling disk I/O limits */
2129 void blk_set_io_limits(BlockBackend *blk, ThrottleConfig *cfg)
2130 {
2131     throttle_group_config(&blk->public.throttle_group_member, cfg);
2132 }
2133 
2134 void blk_io_limits_disable(BlockBackend *blk)
2135 {
2136     BlockDriverState *bs = blk_bs(blk);
2137     ThrottleGroupMember *tgm = &blk->public.throttle_group_member;
2138     assert(tgm->throttle_state);
2139     if (bs) {
2140         bdrv_drained_begin(bs);
2141     }
2142     throttle_group_unregister_tgm(tgm);
2143     if (bs) {
2144         bdrv_drained_end(bs);
2145     }
2146 }
2147 
2148 /* should be called before blk_set_io_limits if a limit is set */
2149 void blk_io_limits_enable(BlockBackend *blk, const char *group)
2150 {
2151     assert(!blk->public.throttle_group_member.throttle_state);
2152     throttle_group_register_tgm(&blk->public.throttle_group_member,
2153                                 group, blk_get_aio_context(blk));
2154 }
2155 
2156 void blk_io_limits_update_group(BlockBackend *blk, const char *group)
2157 {
2158     /* this BB is not part of any group */
2159     if (!blk->public.throttle_group_member.throttle_state) {
2160         return;
2161     }
2162 
2163     /* this BB is a part of the same group than the one we want */
2164     if (!g_strcmp0(throttle_group_get_name(&blk->public.throttle_group_member),
2165                 group)) {
2166         return;
2167     }
2168 
2169     /* need to change the group this bs belong to */
2170     blk_io_limits_disable(blk);
2171     blk_io_limits_enable(blk, group);
2172 }
2173 
2174 static void blk_root_drained_begin(BdrvChild *child)
2175 {
2176     BlockBackend *blk = child->opaque;
2177 
2178     if (++blk->quiesce_counter == 1) {
2179         if (blk->dev_ops && blk->dev_ops->drained_begin) {
2180             blk->dev_ops->drained_begin(blk->dev_opaque);
2181         }
2182     }
2183 
2184     /* Note that blk->root may not be accessible here yet if we are just
2185      * attaching to a BlockDriverState that is drained. Use child instead. */
2186 
2187     if (atomic_fetch_inc(&blk->public.throttle_group_member.io_limits_disabled) == 0) {
2188         throttle_group_restart_tgm(&blk->public.throttle_group_member);
2189     }
2190 }
2191 
2192 static void blk_root_drained_end(BdrvChild *child)
2193 {
2194     BlockBackend *blk = child->opaque;
2195     assert(blk->quiesce_counter);
2196 
2197     assert(blk->public.throttle_group_member.io_limits_disabled);
2198     atomic_dec(&blk->public.throttle_group_member.io_limits_disabled);
2199 
2200     if (--blk->quiesce_counter == 0) {
2201         if (blk->dev_ops && blk->dev_ops->drained_end) {
2202             blk->dev_ops->drained_end(blk->dev_opaque);
2203         }
2204     }
2205 }
2206 
2207 void blk_register_buf(BlockBackend *blk, void *host, size_t size)
2208 {
2209     bdrv_register_buf(blk_bs(blk), host, size);
2210 }
2211 
2212 void blk_unregister_buf(BlockBackend *blk, void *host)
2213 {
2214     bdrv_unregister_buf(blk_bs(blk), host);
2215 }
2216 
2217 int coroutine_fn blk_co_copy_range(BlockBackend *blk_in, int64_t off_in,
2218                                    BlockBackend *blk_out, int64_t off_out,
2219                                    int bytes, BdrvRequestFlags read_flags,
2220                                    BdrvRequestFlags write_flags)
2221 {
2222     int r;
2223     r = blk_check_byte_request(blk_in, off_in, bytes);
2224     if (r) {
2225         return r;
2226     }
2227     r = blk_check_byte_request(blk_out, off_out, bytes);
2228     if (r) {
2229         return r;
2230     }
2231     return bdrv_co_copy_range(blk_in->root, off_in,
2232                               blk_out->root, off_out,
2233                               bytes, read_flags, write_flags);
2234 }
2235