diff options
author | Kevin Wolf <kwolf@redhat.com> | 2020-10-05 17:58:53 +0200 |
---|---|---|
committer | Markus Armbruster <armbru@redhat.com> | 2020-10-09 07:08:20 +0200 |
commit | e336fd4c4b2fa04e5d6c7f8ee524bfd2d9e9e8f1 (patch) | |
tree | 5e067cd1cc99e744e2eea9eac4cdb58904e564aa /block.c | |
parent | 26b0b698c00bd9176f86c539aeb680481fa19473 (diff) | |
download | qemu-e336fd4c4b2fa04e5d6c7f8ee524bfd2d9e9e8f1.zip qemu-e336fd4c4b2fa04e5d6c7f8ee524bfd2d9e9e8f1.tar.gz qemu-e336fd4c4b2fa04e5d6c7f8ee524bfd2d9e9e8f1.tar.bz2 |
block: Add bdrv_co_enter()/leave()
Add a pair of functions to temporarily move the current coroutine to the
AioContext of a given BlockDriverState.
Signed-off-by: Kevin Wolf <kwolf@redhat.com>
Message-Id: <20201005155855.256490-13-kwolf@redhat.com>
Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com>
Signed-off-by: Markus Armbruster <armbru@redhat.com>
Diffstat (limited to 'block.c')
-rw-r--r-- | block.c | 23 |
1 files changed, 23 insertions, 0 deletions
@@ -6303,6 +6303,29 @@ AioContext *bdrv_get_aio_context(BlockDriverState *bs) return bs ? bs->aio_context : qemu_get_aio_context(); } +AioContext *coroutine_fn bdrv_co_enter(BlockDriverState *bs) +{ + Coroutine *self = qemu_coroutine_self(); + AioContext *old_ctx = qemu_coroutine_get_aio_context(self); + AioContext *new_ctx; + + /* + * Increase bs->in_flight to ensure that this operation is completed before + * moving the node to a different AioContext. Read new_ctx only afterwards. + */ + bdrv_inc_in_flight(bs); + + new_ctx = bdrv_get_aio_context(bs); + aio_co_reschedule_self(new_ctx); + return old_ctx; +} + +void coroutine_fn bdrv_co_leave(BlockDriverState *bs, AioContext *old_ctx) +{ + aio_co_reschedule_self(old_ctx); + bdrv_dec_in_flight(bs); +} + void bdrv_coroutine_enter(BlockDriverState *bs, Coroutine *co) { aio_co_enter(bdrv_get_aio_context(bs), co); |