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