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