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