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