1 | The following changes since commit 96662996eda78c48aadddd4e76d8615c7eb72d80: | 1 | The following changes since commit 56f9e46b841c7be478ca038d8d4085d776ab4b0d: |
---|---|---|---|
2 | 2 | ||
3 | Merge remote-tracking branch 'remotes/dgilbert/tags/pull-migration-20210513a' into staging (2021-05-14 12:03:47 +0100) | 3 | Merge remote-tracking branch 'remotes/armbru/tags/pull-qapi-2017-02-20' into staging (2017-02-20 17:42:47 +0000) |
4 | 4 | ||
5 | are available in the Git repository at: | 5 | are available in the git repository at: |
6 | 6 | ||
7 | https://github.com/XanClic/qemu.git tags/pull-block-2021-05-14 | 7 | git://github.com/stefanha/qemu.git tags/block-pull-request |
8 | 8 | ||
9 | for you to fetch changes up to c61ebf362d0abf288ce266845519d5a550a1d89f: | 9 | for you to fetch changes up to a7b91d35bab97a2d3e779d0c64c9b837b52a6cf7: |
10 | 10 | ||
11 | write-threshold: deal with includes (2021-05-14 16:14:10 +0200) | 11 | coroutine-lock: make CoRwlock thread-safe and fair (2017-02-21 11:39:40 +0000) |
12 | 12 | ||
13 | ---------------------------------------------------------------- | 13 | ---------------------------------------------------------------- |
14 | Block patches: | 14 | Pull request |
15 | - drop block/io write notifiers | 15 | |
16 | - qemu-iotests enhancements to make debugging easier | 16 | v2: |
17 | - rbd parsing fix | 17 | * Rebased to resolve scsi conflicts |
18 | - HMP qemu-io fix (for iothreads) | ||
19 | - mirror job cancel relaxation (do not cancel in-flight requests when a | ||
20 | READY mirror job is canceled with force=false) | ||
21 | - document qcow2's data_file and data_file_raw features | ||
22 | - fix iotest 297 for pylint 2.8 | ||
23 | - block/copy-on-read refactoring | ||
24 | 18 | ||
25 | ---------------------------------------------------------------- | 19 | ---------------------------------------------------------------- |
26 | Connor Kuehl (3): | ||
27 | iotests/231: Update expected deprecation message | ||
28 | block/rbd: Add an escape-aware strchr helper | ||
29 | Document qemu-img options data_file and data_file_raw | ||
30 | 20 | ||
31 | Emanuele Giuseppe Esposito (1): | 21 | Paolo Bonzini (24): |
32 | qemu-iotests: fix pylint 2.8 consider-using-with error | 22 | block: move AioContext, QEMUTimer, main-loop to libqemuutil |
23 | aio: introduce aio_co_schedule and aio_co_wake | ||
24 | block-backend: allow blk_prw from coroutine context | ||
25 | test-thread-pool: use generic AioContext infrastructure | ||
26 | io: add methods to set I/O handlers on AioContext | ||
27 | io: make qio_channel_yield aware of AioContexts | ||
28 | nbd: convert to use qio_channel_yield | ||
29 | coroutine-lock: reschedule coroutine on the AioContext it was running | ||
30 | on | ||
31 | blkdebug: reschedule coroutine on the AioContext it is running on | ||
32 | qed: introduce qed_aio_start_io and qed_aio_next_io_cb | ||
33 | aio: push aio_context_acquire/release down to dispatching | ||
34 | block: explicitly acquire aiocontext in timers that need it | ||
35 | block: explicitly acquire aiocontext in callbacks that need it | ||
36 | block: explicitly acquire aiocontext in bottom halves that need it | ||
37 | block: explicitly acquire aiocontext in aio callbacks that need it | ||
38 | aio-posix: partially inline aio_dispatch into aio_poll | ||
39 | async: remove unnecessary inc/dec pairs | ||
40 | block: document fields protected by AioContext lock | ||
41 | coroutine-lock: make CoMutex thread-safe | ||
42 | coroutine-lock: add limited spinning to CoMutex | ||
43 | test-aio-multithread: add performance comparison with thread-based | ||
44 | mutexes | ||
45 | coroutine-lock: place CoMutex before CoQueue in header | ||
46 | coroutine-lock: add mutex argument to CoQueue APIs | ||
47 | coroutine-lock: make CoRwlock thread-safe and fair | ||
33 | 48 | ||
34 | Paolo Bonzini (5): | 49 | Makefile.objs | 4 - |
35 | qemu-iotests: do not buffer the test output | 50 | stubs/Makefile.objs | 1 + |
36 | qemu-iotests: allow passing unittest.main arguments to the test | 51 | tests/Makefile.include | 19 +- |
37 | scripts | 52 | util/Makefile.objs | 6 +- |
38 | qemu-iotests: move command line and environment handling from | 53 | block/nbd-client.h | 2 +- |
39 | TestRunner to TestEnv | 54 | block/qed.h | 3 + |
40 | qemu-iotests: let "check" spawn an arbitrary test command | 55 | include/block/aio.h | 38 ++- |
41 | qemu-iotests: fix case of SOCK_DIR already in the environment | 56 | include/block/block_int.h | 64 +++-- |
42 | 57 | include/io/channel.h | 72 +++++- | |
43 | Vladimir Sementsov-Ogievskiy (10): | 58 | include/qemu/coroutine.h | 84 ++++--- |
44 | monitor: hmp_qemu_io: acquire aio contex, fix crash | 59 | include/qemu/coroutine_int.h | 11 +- |
45 | mirror: stop cancelling in-flight requests on non-force cancel in | 60 | include/sysemu/block-backend.h | 14 +- |
46 | READY | 61 | tests/iothread.h | 25 ++ |
47 | block/copy-on-read: use bdrv_drop_filter() and drop s->active | 62 | block/backup.c | 2 +- |
48 | block/write-threshold: don't use write notifiers | 63 | block/blkdebug.c | 9 +- |
49 | block: drop write notifiers | 64 | block/blkreplay.c | 2 +- |
50 | test-write-threshold: rewrite test_threshold_(not_)trigger tests | 65 | block/block-backend.c | 13 +- |
51 | block/write-threshold: drop extra APIs | 66 | block/curl.c | 44 +++- |
52 | test-write-threshold: drop extra tests | 67 | block/gluster.c | 9 +- |
53 | test-write-threshold: drop extra TestStruct structure | 68 | block/io.c | 42 +--- |
54 | write-threshold: deal with includes | 69 | block/iscsi.c | 15 +- |
55 | 70 | block/linux-aio.c | 10 +- | |
56 | docs/tools/qemu-img.rst | 31 +++++++ | 71 | block/mirror.c | 12 +- |
57 | include/block/block_int.h | 15 +--- | 72 | block/nbd-client.c | 119 +++++---- |
58 | include/block/write-threshold.h | 27 ++---- | 73 | block/nfs.c | 9 +- |
59 | include/qemu/job.h | 2 +- | 74 | block/qcow2-cluster.c | 4 +- |
60 | block.c | 1 - | 75 | block/qed-cluster.c | 2 + |
61 | block/backup.c | 2 +- | 76 | block/qed-table.c | 12 +- |
62 | block/copy-on-read.c | 33 +------ | 77 | block/qed.c | 58 +++-- |
63 | block/io.c | 11 +-- | 78 | block/sheepdog.c | 31 +-- |
64 | block/mirror.c | 6 +- | 79 | block/ssh.c | 29 +-- |
65 | block/monitor/block-hmp-cmds.c | 31 ++++--- | 80 | block/throttle-groups.c | 4 +- |
66 | block/rbd.c | 32 ++++--- | 81 | block/win32-aio.c | 9 +- |
67 | block/write-threshold.c | 91 ++++--------------- | 82 | dma-helpers.c | 2 + |
68 | job.c | 2 +- | 83 | hw/9pfs/9p.c | 2 +- |
69 | qemu-io-cmds.c | 8 +- | 84 | hw/block/virtio-blk.c | 19 +- |
70 | qemu-io.c | 17 +++- | 85 | hw/scsi/scsi-bus.c | 2 + |
71 | tests/unit/test-write-threshold.c | 90 ++----------------- | 86 | hw/scsi/scsi-disk.c | 15 ++ |
72 | tests/qemu-iotests/231 | 4 + | 87 | hw/scsi/scsi-generic.c | 20 +- |
73 | tests/qemu-iotests/231.out | 7 +- | 88 | hw/scsi/virtio-scsi.c | 7 + |
74 | tests/qemu-iotests/240.out | 8 +- | 89 | io/channel-command.c | 13 + |
75 | tests/qemu-iotests/245.out | 8 +- | 90 | io/channel-file.c | 11 + |
76 | tests/qemu-iotests/264 | 2 +- | 91 | io/channel-socket.c | 16 +- |
77 | tests/qemu-iotests/295.out | 6 +- | 92 | io/channel-tls.c | 12 + |
78 | tests/qemu-iotests/296.out | 8 +- | 93 | io/channel-watch.c | 6 + |
79 | tests/qemu-iotests/check | 19 +++- | 94 | io/channel.c | 97 ++++++-- |
80 | tests/qemu-iotests/iotests.py | 145 +++++++++++++++++------------- | 95 | nbd/client.c | 2 +- |
81 | tests/qemu-iotests/pylintrc | 3 + | 96 | nbd/common.c | 9 +- |
82 | tests/qemu-iotests/testenv.py | 22 ++++- | 97 | nbd/server.c | 94 +++----- |
83 | tests/qemu-iotests/testrunner.py | 37 +++----- | 98 | stubs/linux-aio.c | 32 +++ |
84 | 28 files changed, 289 insertions(+), 379 deletions(-) | 99 | stubs/set-fd-handler.c | 11 - |
100 | tests/iothread.c | 91 +++++++ | ||
101 | tests/test-aio-multithread.c | 463 ++++++++++++++++++++++++++++++++++++ | ||
102 | tests/test-thread-pool.c | 12 +- | ||
103 | aio-posix.c => util/aio-posix.c | 62 ++--- | ||
104 | aio-win32.c => util/aio-win32.c | 30 +-- | ||
105 | util/aiocb.c | 55 +++++ | ||
106 | async.c => util/async.c | 84 ++++++- | ||
107 | iohandler.c => util/iohandler.c | 0 | ||
108 | main-loop.c => util/main-loop.c | 0 | ||
109 | util/qemu-coroutine-lock.c | 254 ++++++++++++++++++-- | ||
110 | util/qemu-coroutine-sleep.c | 2 +- | ||
111 | util/qemu-coroutine.c | 8 + | ||
112 | qemu-timer.c => util/qemu-timer.c | 0 | ||
113 | thread-pool.c => util/thread-pool.c | 8 +- | ||
114 | trace-events | 11 - | ||
115 | util/trace-events | 17 +- | ||
116 | 67 files changed, 1712 insertions(+), 533 deletions(-) | ||
117 | create mode 100644 tests/iothread.h | ||
118 | create mode 100644 stubs/linux-aio.c | ||
119 | create mode 100644 tests/iothread.c | ||
120 | create mode 100644 tests/test-aio-multithread.c | ||
121 | rename aio-posix.c => util/aio-posix.c (94%) | ||
122 | rename aio-win32.c => util/aio-win32.c (95%) | ||
123 | create mode 100644 util/aiocb.c | ||
124 | rename async.c => util/async.c (82%) | ||
125 | rename iohandler.c => util/iohandler.c (100%) | ||
126 | rename main-loop.c => util/main-loop.c (100%) | ||
127 | rename qemu-timer.c => util/qemu-timer.c (100%) | ||
128 | rename thread-pool.c => util/thread-pool.c (97%) | ||
85 | 129 | ||
86 | -- | 130 | -- |
87 | 2.31.1 | 131 | 2.9.3 |
88 | 132 | ||
89 | 133 | diff view generated by jsdifflib |
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | write-notifiers are used only for write-threshold. New code for such | 3 | AioContext is fairly self contained, the only dependency is QEMUTimer but |
4 | purpose should create filters. | 4 | that in turn doesn't need anything else. So move them out of block-obj-y |
5 | 5 | to avoid introducing a dependency from io/ to block-obj-y. | |
6 | Let's better special-case write-threshold and drop write notifiers at | 6 | |
7 | all. (Actually, write-threshold is special-cased anyway, as the only | 7 | main-loop and its dependency iohandler also need to be moved, because |
8 | user of write-notifiers) | 8 | later in this series io/ will call iohandler_get_aio_context. |
9 | 9 | ||
10 | So, create a new direct interface for bdrv_co_write_req_prepare() and | 10 | [Changed copyright "the QEMU team" to "other QEMU contributors" as |
11 | drop all write-notifier related logic from write-threshold.c. | 11 | suggested by Daniel Berrange and agreed by Paolo. |
12 | 12 | --Stefan] | |
13 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 13 | |
14 | Reviewed-by: Max Reitz <mreitz@redhat.com> | 14 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
15 | Message-Id: <20210506090621.11848-2-vsementsov@virtuozzo.com> | 15 | Reviewed-by: Fam Zheng <famz@redhat.com> |
16 | Reviewed-by: Eric Blake <eblake@redhat.com> | 16 | Message-id: 20170213135235.12274-2-pbonzini@redhat.com |
17 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | 17 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
18 | [mreitz: Adjusted comment as per Eric's suggestion] | ||
19 | Signed-off-by: Max Reitz <mreitz@redhat.com> | ||
20 | --- | 18 | --- |
21 | include/block/block_int.h | 1 - | 19 | Makefile.objs | 4 --- |
22 | include/block/write-threshold.h | 9 +++++ | 20 | stubs/Makefile.objs | 1 + |
23 | block/io.c | 5 ++- | 21 | tests/Makefile.include | 11 ++++---- |
24 | block/write-threshold.c | 70 +++++++-------------------------- | 22 | util/Makefile.objs | 6 +++- |
25 | 4 files changed, 27 insertions(+), 58 deletions(-) | 23 | block/io.c | 29 ------------------- |
26 | 24 | stubs/linux-aio.c | 32 +++++++++++++++++++++ | |
27 | diff --git a/include/block/block_int.h b/include/block/block_int.h | 25 | stubs/set-fd-handler.c | 11 -------- |
28 | index XXXXXXX..XXXXXXX 100644 | 26 | aio-posix.c => util/aio-posix.c | 2 +- |
29 | --- a/include/block/block_int.h | 27 | aio-win32.c => util/aio-win32.c | 0 |
30 | +++ b/include/block/block_int.h | 28 | util/aiocb.c | 55 +++++++++++++++++++++++++++++++++++++ |
31 | @@ -XXX,XX +XXX,XX @@ struct BlockDriverState { | 29 | async.c => util/async.c | 3 +- |
32 | 30 | iohandler.c => util/iohandler.c | 0 | |
33 | /* threshold limit for writes, in bytes. "High water mark". */ | 31 | main-loop.c => util/main-loop.c | 0 |
34 | uint64_t write_threshold_offset; | 32 | qemu-timer.c => util/qemu-timer.c | 0 |
35 | - NotifierWithReturn write_threshold_notifier; | 33 | thread-pool.c => util/thread-pool.c | 2 +- |
36 | 34 | trace-events | 11 -------- | |
37 | /* Writing to the list requires the BQL _and_ the dirty_bitmap_mutex. | 35 | util/trace-events | 11 ++++++++ |
38 | * Reading from the list can be done with either the BQL or the | 36 | 17 files changed, 114 insertions(+), 64 deletions(-) |
39 | diff --git a/include/block/write-threshold.h b/include/block/write-threshold.h | 37 | create mode 100644 stubs/linux-aio.c |
40 | index XXXXXXX..XXXXXXX 100644 | 38 | rename aio-posix.c => util/aio-posix.c (99%) |
41 | --- a/include/block/write-threshold.h | 39 | rename aio-win32.c => util/aio-win32.c (100%) |
42 | +++ b/include/block/write-threshold.h | 40 | create mode 100644 util/aiocb.c |
43 | @@ -XXX,XX +XXX,XX @@ bool bdrv_write_threshold_is_set(const BlockDriverState *bs); | 41 | rename async.c => util/async.c (99%) |
44 | uint64_t bdrv_write_threshold_exceeded(const BlockDriverState *bs, | 42 | rename iohandler.c => util/iohandler.c (100%) |
45 | const BdrvTrackedRequest *req); | 43 | rename main-loop.c => util/main-loop.c (100%) |
46 | 44 | rename qemu-timer.c => util/qemu-timer.c (100%) | |
47 | +/* | 45 | rename thread-pool.c => util/thread-pool.c (99%) |
48 | + * bdrv_write_threshold_check_write | 46 | |
49 | + * | 47 | diff --git a/Makefile.objs b/Makefile.objs |
50 | + * Check whether the specified request exceeds the write threshold. | 48 | index XXXXXXX..XXXXXXX 100644 |
51 | + * If so, send a corresponding event and disable write threshold checking. | 49 | --- a/Makefile.objs |
52 | + */ | 50 | +++ b/Makefile.objs |
53 | +void bdrv_write_threshold_check_write(BlockDriverState *bs, int64_t offset, | 51 | @@ -XXX,XX +XXX,XX @@ chardev-obj-y = chardev/ |
54 | + int64_t bytes); | 52 | ####################################################################### |
55 | + | 53 | # block-obj-y is code used by both qemu system emulation and qemu-img |
56 | #endif | 54 | |
55 | -block-obj-y = async.o thread-pool.o | ||
56 | block-obj-y += nbd/ | ||
57 | block-obj-y += block.o blockjob.o | ||
58 | -block-obj-y += main-loop.o iohandler.o qemu-timer.o | ||
59 | -block-obj-$(CONFIG_POSIX) += aio-posix.o | ||
60 | -block-obj-$(CONFIG_WIN32) += aio-win32.o | ||
61 | block-obj-y += block/ | ||
62 | block-obj-y += qemu-io-cmds.o | ||
63 | block-obj-$(CONFIG_REPLICATION) += replication.o | ||
64 | diff --git a/stubs/Makefile.objs b/stubs/Makefile.objs | ||
65 | index XXXXXXX..XXXXXXX 100644 | ||
66 | --- a/stubs/Makefile.objs | ||
67 | +++ b/stubs/Makefile.objs | ||
68 | @@ -XXX,XX +XXX,XX @@ stub-obj-y += get-vm-name.o | ||
69 | stub-obj-y += iothread.o | ||
70 | stub-obj-y += iothread-lock.o | ||
71 | stub-obj-y += is-daemonized.o | ||
72 | +stub-obj-$(CONFIG_LINUX_AIO) += linux-aio.o | ||
73 | stub-obj-y += machine-init-done.o | ||
74 | stub-obj-y += migr-blocker.o | ||
75 | stub-obj-y += monitor.o | ||
76 | diff --git a/tests/Makefile.include b/tests/Makefile.include | ||
77 | index XXXXXXX..XXXXXXX 100644 | ||
78 | --- a/tests/Makefile.include | ||
79 | +++ b/tests/Makefile.include | ||
80 | @@ -XXX,XX +XXX,XX @@ check-unit-y += tests/test-visitor-serialization$(EXESUF) | ||
81 | check-unit-y += tests/test-iov$(EXESUF) | ||
82 | gcov-files-test-iov-y = util/iov.c | ||
83 | check-unit-y += tests/test-aio$(EXESUF) | ||
84 | +gcov-files-test-aio-y = util/async.c util/qemu-timer.o | ||
85 | +gcov-files-test-aio-$(CONFIG_WIN32) += util/aio-win32.c | ||
86 | +gcov-files-test-aio-$(CONFIG_POSIX) += util/aio-posix.c | ||
87 | check-unit-y += tests/test-throttle$(EXESUF) | ||
88 | gcov-files-test-aio-$(CONFIG_WIN32) = aio-win32.c | ||
89 | gcov-files-test-aio-$(CONFIG_POSIX) = aio-posix.c | ||
90 | @@ -XXX,XX +XXX,XX @@ tests/check-qjson$(EXESUF): tests/check-qjson.o $(test-util-obj-y) | ||
91 | tests/check-qom-interface$(EXESUF): tests/check-qom-interface.o $(test-qom-obj-y) | ||
92 | tests/check-qom-proplist$(EXESUF): tests/check-qom-proplist.o $(test-qom-obj-y) | ||
93 | |||
94 | -tests/test-char$(EXESUF): tests/test-char.o qemu-timer.o \ | ||
95 | - $(test-util-obj-y) $(qtest-obj-y) $(test-block-obj-y) $(chardev-obj-y) | ||
96 | +tests/test-char$(EXESUF): tests/test-char.o $(test-util-obj-y) $(qtest-obj-y) $(test-io-obj-y) $(chardev-obj-y) | ||
97 | tests/test-coroutine$(EXESUF): tests/test-coroutine.o $(test-block-obj-y) | ||
98 | tests/test-aio$(EXESUF): tests/test-aio.o $(test-block-obj-y) | ||
99 | tests/test-throttle$(EXESUF): tests/test-throttle.o $(test-block-obj-y) | ||
100 | @@ -XXX,XX +XXX,XX @@ tests/test-vmstate$(EXESUF): tests/test-vmstate.o \ | ||
101 | migration/vmstate.o migration/qemu-file.o \ | ||
102 | migration/qemu-file-channel.o migration/qjson.o \ | ||
103 | $(test-io-obj-y) | ||
104 | -tests/test-timed-average$(EXESUF): tests/test-timed-average.o qemu-timer.o \ | ||
105 | - $(test-util-obj-y) | ||
106 | +tests/test-timed-average$(EXESUF): tests/test-timed-average.o $(test-util-obj-y) | ||
107 | tests/test-base64$(EXESUF): tests/test-base64.o \ | ||
108 | libqemuutil.a libqemustub.a | ||
109 | tests/ptimer-test$(EXESUF): tests/ptimer-test.o tests/ptimer-test-stubs.o hw/core/ptimer.o libqemustub.a | ||
110 | @@ -XXX,XX +XXX,XX @@ tests/usb-hcd-ehci-test$(EXESUF): tests/usb-hcd-ehci-test.o $(libqos-usb-obj-y) | ||
111 | tests/usb-hcd-xhci-test$(EXESUF): tests/usb-hcd-xhci-test.o $(libqos-usb-obj-y) | ||
112 | tests/pc-cpu-test$(EXESUF): tests/pc-cpu-test.o | ||
113 | tests/postcopy-test$(EXESUF): tests/postcopy-test.o | ||
114 | -tests/vhost-user-test$(EXESUF): tests/vhost-user-test.o qemu-timer.o \ | ||
115 | +tests/vhost-user-test$(EXESUF): tests/vhost-user-test.o $(test-util-obj-y) \ | ||
116 | $(qtest-obj-y) $(test-io-obj-y) $(libqos-virtio-obj-y) $(libqos-pc-obj-y) \ | ||
117 | $(chardev-obj-y) | ||
118 | tests/qemu-iotests/socket_scm_helper$(EXESUF): tests/qemu-iotests/socket_scm_helper.o | ||
119 | diff --git a/util/Makefile.objs b/util/Makefile.objs | ||
120 | index XXXXXXX..XXXXXXX 100644 | ||
121 | --- a/util/Makefile.objs | ||
122 | +++ b/util/Makefile.objs | ||
123 | @@ -XXX,XX +XXX,XX @@ | ||
124 | util-obj-y = osdep.o cutils.o unicode.o qemu-timer-common.o | ||
125 | util-obj-y += bufferiszero.o | ||
126 | util-obj-y += lockcnt.o | ||
127 | +util-obj-y += aiocb.o async.o thread-pool.o qemu-timer.o | ||
128 | +util-obj-y += main-loop.o iohandler.o | ||
129 | +util-obj-$(CONFIG_POSIX) += aio-posix.o | ||
130 | util-obj-$(CONFIG_POSIX) += compatfd.o | ||
131 | util-obj-$(CONFIG_POSIX) += event_notifier-posix.o | ||
132 | util-obj-$(CONFIG_POSIX) += mmap-alloc.o | ||
133 | util-obj-$(CONFIG_POSIX) += oslib-posix.o | ||
134 | util-obj-$(CONFIG_POSIX) += qemu-openpty.o | ||
135 | util-obj-$(CONFIG_POSIX) += qemu-thread-posix.o | ||
136 | -util-obj-$(CONFIG_WIN32) += event_notifier-win32.o | ||
137 | util-obj-$(CONFIG_POSIX) += memfd.o | ||
138 | +util-obj-$(CONFIG_WIN32) += aio-win32.o | ||
139 | +util-obj-$(CONFIG_WIN32) += event_notifier-win32.o | ||
140 | util-obj-$(CONFIG_WIN32) += oslib-win32.o | ||
141 | util-obj-$(CONFIG_WIN32) += qemu-thread-win32.o | ||
142 | util-obj-y += envlist.o path.o module.o | ||
57 | diff --git a/block/io.c b/block/io.c | 143 | diff --git a/block/io.c b/block/io.c |
58 | index XXXXXXX..XXXXXXX 100644 | 144 | index XXXXXXX..XXXXXXX 100644 |
59 | --- a/block/io.c | 145 | --- a/block/io.c |
60 | +++ b/block/io.c | 146 | +++ b/block/io.c |
61 | @@ -XXX,XX +XXX,XX @@ | 147 | @@ -XXX,XX +XXX,XX @@ BlockAIOCB *bdrv_aio_flush(BlockDriverState *bs, |
62 | #include "block/blockjob_int.h" | 148 | return &acb->common; |
63 | #include "block/block_int.h" | ||
64 | #include "block/coroutines.h" | ||
65 | +#include "block/write-threshold.h" | ||
66 | #include "qemu/cutils.h" | ||
67 | #include "qapi/error.h" | ||
68 | #include "qemu/error-report.h" | ||
69 | @@ -XXX,XX +XXX,XX @@ bdrv_co_write_req_prepare(BdrvChild *child, int64_t offset, int64_t bytes, | ||
70 | } else { | ||
71 | assert(child->perm & BLK_PERM_WRITE); | ||
72 | } | ||
73 | - return notifier_with_return_list_notify(&bs->before_write_notifiers, | ||
74 | - req); | ||
75 | + bdrv_write_threshold_check_write(bs, offset, bytes); | ||
76 | + return 0; | ||
77 | case BDRV_TRACKED_TRUNCATE: | ||
78 | assert(child->perm & BLK_PERM_RESIZE); | ||
79 | return 0; | ||
80 | diff --git a/block/write-threshold.c b/block/write-threshold.c | ||
81 | index XXXXXXX..XXXXXXX 100644 | ||
82 | --- a/block/write-threshold.c | ||
83 | +++ b/block/write-threshold.c | ||
84 | @@ -XXX,XX +XXX,XX @@ bool bdrv_write_threshold_is_set(const BlockDriverState *bs) | ||
85 | return bs->write_threshold_offset > 0; | ||
86 | } | 149 | } |
87 | 150 | ||
88 | -static void write_threshold_disable(BlockDriverState *bs) | 151 | -void *qemu_aio_get(const AIOCBInfo *aiocb_info, BlockDriverState *bs, |
152 | - BlockCompletionFunc *cb, void *opaque) | ||
89 | -{ | 153 | -{ |
90 | - if (bdrv_write_threshold_is_set(bs)) { | 154 | - BlockAIOCB *acb; |
91 | - notifier_with_return_remove(&bs->write_threshold_notifier); | 155 | - |
92 | - bs->write_threshold_offset = 0; | 156 | - acb = g_malloc(aiocb_info->aiocb_size); |
157 | - acb->aiocb_info = aiocb_info; | ||
158 | - acb->bs = bs; | ||
159 | - acb->cb = cb; | ||
160 | - acb->opaque = opaque; | ||
161 | - acb->refcnt = 1; | ||
162 | - return acb; | ||
163 | -} | ||
164 | - | ||
165 | -void qemu_aio_ref(void *p) | ||
166 | -{ | ||
167 | - BlockAIOCB *acb = p; | ||
168 | - acb->refcnt++; | ||
169 | -} | ||
170 | - | ||
171 | -void qemu_aio_unref(void *p) | ||
172 | -{ | ||
173 | - BlockAIOCB *acb = p; | ||
174 | - assert(acb->refcnt > 0); | ||
175 | - if (--acb->refcnt == 0) { | ||
176 | - g_free(acb); | ||
93 | - } | 177 | - } |
94 | -} | 178 | -} |
95 | - | 179 | - |
96 | uint64_t bdrv_write_threshold_exceeded(const BlockDriverState *bs, | 180 | /**************************************************************/ |
97 | const BdrvTrackedRequest *req) | 181 | /* Coroutine block device emulation */ |
182 | |||
183 | diff --git a/stubs/linux-aio.c b/stubs/linux-aio.c | ||
184 | new file mode 100644 | ||
185 | index XXXXXXX..XXXXXXX | ||
186 | --- /dev/null | ||
187 | +++ b/stubs/linux-aio.c | ||
188 | @@ -XXX,XX +XXX,XX @@ | ||
189 | +/* | ||
190 | + * Linux native AIO support. | ||
191 | + * | ||
192 | + * Copyright (C) 2009 IBM, Corp. | ||
193 | + * Copyright (C) 2009 Red Hat, Inc. | ||
194 | + * | ||
195 | + * This work is licensed under the terms of the GNU GPL, version 2 or later. | ||
196 | + * See the COPYING file in the top-level directory. | ||
197 | + */ | ||
198 | +#include "qemu/osdep.h" | ||
199 | +#include "block/aio.h" | ||
200 | +#include "block/raw-aio.h" | ||
201 | + | ||
202 | +void laio_detach_aio_context(LinuxAioState *s, AioContext *old_context) | ||
203 | +{ | ||
204 | + abort(); | ||
205 | +} | ||
206 | + | ||
207 | +void laio_attach_aio_context(LinuxAioState *s, AioContext *new_context) | ||
208 | +{ | ||
209 | + abort(); | ||
210 | +} | ||
211 | + | ||
212 | +LinuxAioState *laio_init(void) | ||
213 | +{ | ||
214 | + abort(); | ||
215 | +} | ||
216 | + | ||
217 | +void laio_cleanup(LinuxAioState *s) | ||
218 | +{ | ||
219 | + abort(); | ||
220 | +} | ||
221 | diff --git a/stubs/set-fd-handler.c b/stubs/set-fd-handler.c | ||
222 | index XXXXXXX..XXXXXXX 100644 | ||
223 | --- a/stubs/set-fd-handler.c | ||
224 | +++ b/stubs/set-fd-handler.c | ||
225 | @@ -XXX,XX +XXX,XX @@ void qemu_set_fd_handler(int fd, | ||
98 | { | 226 | { |
99 | @@ -XXX,XX +XXX,XX @@ uint64_t bdrv_write_threshold_exceeded(const BlockDriverState *bs, | 227 | abort(); |
100 | return 0; | ||
101 | } | 228 | } |
102 | 229 | - | |
103 | -static int coroutine_fn before_write_notify(NotifierWithReturn *notifier, | 230 | -void aio_set_fd_handler(AioContext *ctx, |
104 | - void *opaque) | 231 | - int fd, |
232 | - bool is_external, | ||
233 | - IOHandler *io_read, | ||
234 | - IOHandler *io_write, | ||
235 | - AioPollFn *io_poll, | ||
236 | - void *opaque) | ||
105 | -{ | 237 | -{ |
106 | - BdrvTrackedRequest *req = opaque; | 238 | - abort(); |
107 | - BlockDriverState *bs = req->bs; | ||
108 | - uint64_t amount = 0; | ||
109 | - | ||
110 | - amount = bdrv_write_threshold_exceeded(bs, req); | ||
111 | - if (amount > 0) { | ||
112 | - qapi_event_send_block_write_threshold( | ||
113 | - bs->node_name, | ||
114 | - amount, | ||
115 | - bs->write_threshold_offset); | ||
116 | - | ||
117 | - /* autodisable to avoid flooding the monitor */ | ||
118 | - write_threshold_disable(bs); | ||
119 | - } | ||
120 | - | ||
121 | - return 0; /* should always let other notifiers run */ | ||
122 | -} | 239 | -} |
123 | - | 240 | diff --git a/aio-posix.c b/util/aio-posix.c |
124 | -static void write_threshold_register_notifier(BlockDriverState *bs) | 241 | similarity index 99% |
125 | -{ | 242 | rename from aio-posix.c |
126 | - bs->write_threshold_notifier.notify = before_write_notify; | 243 | rename to util/aio-posix.c |
127 | - bdrv_add_before_write_notifier(bs, &bs->write_threshold_notifier); | 244 | index XXXXXXX..XXXXXXX 100644 |
128 | -} | 245 | --- a/aio-posix.c |
129 | - | 246 | +++ b/util/aio-posix.c |
130 | -static void write_threshold_update(BlockDriverState *bs, | 247 | @@ -XXX,XX +XXX,XX @@ |
131 | - int64_t threshold_bytes) | 248 | #include "qemu/rcu_queue.h" |
132 | -{ | 249 | #include "qemu/sockets.h" |
133 | - bs->write_threshold_offset = threshold_bytes; | 250 | #include "qemu/cutils.h" |
134 | -} | 251 | -#include "trace-root.h" |
135 | - | 252 | +#include "trace.h" |
136 | void bdrv_write_threshold_set(BlockDriverState *bs, uint64_t threshold_bytes) | 253 | #ifdef CONFIG_EPOLL_CREATE1 |
137 | { | 254 | #include <sys/epoll.h> |
138 | - if (bdrv_write_threshold_is_set(bs)) { | 255 | #endif |
139 | - if (threshold_bytes > 0) { | 256 | diff --git a/aio-win32.c b/util/aio-win32.c |
140 | - write_threshold_update(bs, threshold_bytes); | 257 | similarity index 100% |
141 | - } else { | 258 | rename from aio-win32.c |
142 | - write_threshold_disable(bs); | 259 | rename to util/aio-win32.c |
143 | - } | 260 | diff --git a/util/aiocb.c b/util/aiocb.c |
144 | - } else { | 261 | new file mode 100644 |
145 | - if (threshold_bytes > 0) { | 262 | index XXXXXXX..XXXXXXX |
146 | - /* avoid multiple registration */ | 263 | --- /dev/null |
147 | - write_threshold_register_notifier(bs); | 264 | +++ b/util/aiocb.c |
148 | - write_threshold_update(bs, threshold_bytes); | 265 | @@ -XXX,XX +XXX,XX @@ |
149 | - } | 266 | +/* |
150 | - /* discard bogus disable request */ | 267 | + * BlockAIOCB allocation |
151 | - } | 268 | + * |
152 | + bs->write_threshold_offset = threshold_bytes; | 269 | + * Copyright (c) 2003-2017 Fabrice Bellard and other QEMU contributors |
153 | } | 270 | + * |
154 | 271 | + * Permission is hereby granted, free of charge, to any person obtaining a copy | |
155 | void qmp_block_set_write_threshold(const char *node_name, | 272 | + * of this software and associated documentation files (the "Software"), to deal |
156 | @@ -XXX,XX +XXX,XX @@ void qmp_block_set_write_threshold(const char *node_name, | 273 | + * in the Software without restriction, including without limitation the rights |
157 | 274 | + * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
158 | aio_context_release(aio_context); | 275 | + * copies of the Software, and to permit persons to whom the Software is |
159 | } | 276 | + * furnished to do so, subject to the following conditions: |
160 | + | 277 | + * |
161 | +void bdrv_write_threshold_check_write(BlockDriverState *bs, int64_t offset, | 278 | + * The above copyright notice and this permission notice shall be included in |
162 | + int64_t bytes) | 279 | + * all copies or substantial portions of the Software. |
163 | +{ | 280 | + * |
164 | + int64_t end = offset + bytes; | 281 | + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
165 | + uint64_t wtr = bs->write_threshold_offset; | 282 | + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
166 | + | 283 | + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL |
167 | + if (wtr > 0 && end > wtr) { | 284 | + * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
168 | + qapi_event_send_block_write_threshold(bs->node_name, end - wtr, wtr); | 285 | + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
169 | + | 286 | + * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
170 | + /* autodisable to avoid flooding the monitor */ | 287 | + * THE SOFTWARE. |
171 | + bdrv_write_threshold_set(bs, 0); | 288 | + */ |
289 | + | ||
290 | +#include "qemu/osdep.h" | ||
291 | +#include "block/aio.h" | ||
292 | + | ||
293 | +void *qemu_aio_get(const AIOCBInfo *aiocb_info, BlockDriverState *bs, | ||
294 | + BlockCompletionFunc *cb, void *opaque) | ||
295 | +{ | ||
296 | + BlockAIOCB *acb; | ||
297 | + | ||
298 | + acb = g_malloc(aiocb_info->aiocb_size); | ||
299 | + acb->aiocb_info = aiocb_info; | ||
300 | + acb->bs = bs; | ||
301 | + acb->cb = cb; | ||
302 | + acb->opaque = opaque; | ||
303 | + acb->refcnt = 1; | ||
304 | + return acb; | ||
305 | +} | ||
306 | + | ||
307 | +void qemu_aio_ref(void *p) | ||
308 | +{ | ||
309 | + BlockAIOCB *acb = p; | ||
310 | + acb->refcnt++; | ||
311 | +} | ||
312 | + | ||
313 | +void qemu_aio_unref(void *p) | ||
314 | +{ | ||
315 | + BlockAIOCB *acb = p; | ||
316 | + assert(acb->refcnt > 0); | ||
317 | + if (--acb->refcnt == 0) { | ||
318 | + g_free(acb); | ||
172 | + } | 319 | + } |
173 | +} | 320 | +} |
321 | diff --git a/async.c b/util/async.c | ||
322 | similarity index 99% | ||
323 | rename from async.c | ||
324 | rename to util/async.c | ||
325 | index XXXXXXX..XXXXXXX 100644 | ||
326 | --- a/async.c | ||
327 | +++ b/util/async.c | ||
328 | @@ -XXX,XX +XXX,XX @@ | ||
329 | /* | ||
330 | - * QEMU System Emulator | ||
331 | + * Data plane event loop | ||
332 | * | ||
333 | * Copyright (c) 2003-2008 Fabrice Bellard | ||
334 | + * Copyright (c) 2009-2017 QEMU contributors | ||
335 | * | ||
336 | * Permission is hereby granted, free of charge, to any person obtaining a copy | ||
337 | * of this software and associated documentation files (the "Software"), to deal | ||
338 | diff --git a/iohandler.c b/util/iohandler.c | ||
339 | similarity index 100% | ||
340 | rename from iohandler.c | ||
341 | rename to util/iohandler.c | ||
342 | diff --git a/main-loop.c b/util/main-loop.c | ||
343 | similarity index 100% | ||
344 | rename from main-loop.c | ||
345 | rename to util/main-loop.c | ||
346 | diff --git a/qemu-timer.c b/util/qemu-timer.c | ||
347 | similarity index 100% | ||
348 | rename from qemu-timer.c | ||
349 | rename to util/qemu-timer.c | ||
350 | diff --git a/thread-pool.c b/util/thread-pool.c | ||
351 | similarity index 99% | ||
352 | rename from thread-pool.c | ||
353 | rename to util/thread-pool.c | ||
354 | index XXXXXXX..XXXXXXX 100644 | ||
355 | --- a/thread-pool.c | ||
356 | +++ b/util/thread-pool.c | ||
357 | @@ -XXX,XX +XXX,XX @@ | ||
358 | #include "qemu/queue.h" | ||
359 | #include "qemu/thread.h" | ||
360 | #include "qemu/coroutine.h" | ||
361 | -#include "trace-root.h" | ||
362 | +#include "trace.h" | ||
363 | #include "block/thread-pool.h" | ||
364 | #include "qemu/main-loop.h" | ||
365 | |||
366 | diff --git a/trace-events b/trace-events | ||
367 | index XXXXXXX..XXXXXXX 100644 | ||
368 | --- a/trace-events | ||
369 | +++ b/trace-events | ||
370 | @@ -XXX,XX +XXX,XX @@ | ||
371 | # | ||
372 | # The <format-string> should be a sprintf()-compatible format string. | ||
373 | |||
374 | -# aio-posix.c | ||
375 | -run_poll_handlers_begin(void *ctx, int64_t max_ns) "ctx %p max_ns %"PRId64 | ||
376 | -run_poll_handlers_end(void *ctx, bool progress) "ctx %p progress %d" | ||
377 | -poll_shrink(void *ctx, int64_t old, int64_t new) "ctx %p old %"PRId64" new %"PRId64 | ||
378 | -poll_grow(void *ctx, int64_t old, int64_t new) "ctx %p old %"PRId64" new %"PRId64 | ||
379 | - | ||
380 | -# thread-pool.c | ||
381 | -thread_pool_submit(void *pool, void *req, void *opaque) "pool %p req %p opaque %p" | ||
382 | -thread_pool_complete(void *pool, void *req, void *opaque, int ret) "pool %p req %p opaque %p ret %d" | ||
383 | -thread_pool_cancel(void *req, void *opaque) "req %p opaque %p" | ||
384 | - | ||
385 | # ioport.c | ||
386 | cpu_in(unsigned int addr, char size, unsigned int val) "addr %#x(%c) value %u" | ||
387 | cpu_out(unsigned int addr, char size, unsigned int val) "addr %#x(%c) value %u" | ||
388 | diff --git a/util/trace-events b/util/trace-events | ||
389 | index XXXXXXX..XXXXXXX 100644 | ||
390 | --- a/util/trace-events | ||
391 | +++ b/util/trace-events | ||
392 | @@ -XXX,XX +XXX,XX @@ | ||
393 | # See docs/tracing.txt for syntax documentation. | ||
394 | |||
395 | +# util/aio-posix.c | ||
396 | +run_poll_handlers_begin(void *ctx, int64_t max_ns) "ctx %p max_ns %"PRId64 | ||
397 | +run_poll_handlers_end(void *ctx, bool progress) "ctx %p progress %d" | ||
398 | +poll_shrink(void *ctx, int64_t old, int64_t new) "ctx %p old %"PRId64" new %"PRId64 | ||
399 | +poll_grow(void *ctx, int64_t old, int64_t new) "ctx %p old %"PRId64" new %"PRId64 | ||
400 | + | ||
401 | +# util/thread-pool.c | ||
402 | +thread_pool_submit(void *pool, void *req, void *opaque) "pool %p req %p opaque %p" | ||
403 | +thread_pool_complete(void *pool, void *req, void *opaque, int ret) "pool %p req %p opaque %p ret %d" | ||
404 | +thread_pool_cancel(void *req, void *opaque) "req %p opaque %p" | ||
405 | + | ||
406 | # util/buffer.c | ||
407 | buffer_resize(const char *buf, size_t olen, size_t len) "%s: old %zd, new %zd" | ||
408 | buffer_move_empty(const char *buf, size_t len, const char *from) "%s: %zd bytes from %s" | ||
174 | -- | 409 | -- |
175 | 2.31.1 | 410 | 2.9.3 |
176 | 411 | ||
177 | 412 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | 1 | From: Paolo Bonzini <pbonzini@redhat.com> | |
2 | |||
3 | aio_co_wake provides the infrastructure to start a coroutine on a "home" | ||
4 | AioContext. It will be used by CoMutex and CoQueue, so that coroutines | ||
5 | don't jump from one context to another when they go to sleep on a | ||
6 | mutex or waitqueue. However, it can also be used as a more efficient | ||
7 | alternative to one-shot bottom halves, and saves the effort of tracking | ||
8 | which AioContext a coroutine is running on. | ||
9 | |||
10 | aio_co_schedule is the part of aio_co_wake that starts a coroutine | ||
11 | on a remove AioContext, but it is also useful to implement e.g. | ||
12 | bdrv_set_aio_context callbacks. | ||
13 | |||
14 | The implementation of aio_co_schedule is based on a lock-free | ||
15 | multiple-producer, single-consumer queue. The multiple producers use | ||
16 | cmpxchg to add to a LIFO stack. The consumer (a per-AioContext bottom | ||
17 | half) grabs all items added so far, inverts the list to make it FIFO, | ||
18 | and goes through it one item at a time until it's empty. The data | ||
19 | structure was inspired by OSv, which uses it in the very code we'll | ||
20 | "port" to QEMU for the thread-safe CoMutex. | ||
21 | |||
22 | Most of the new code is really tests. | ||
23 | |||
24 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> | ||
25 | Reviewed-by: Fam Zheng <famz@redhat.com> | ||
26 | Message-id: 20170213135235.12274-3-pbonzini@redhat.com | ||
27 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
28 | --- | ||
29 | tests/Makefile.include | 8 +- | ||
30 | include/block/aio.h | 32 +++++++ | ||
31 | include/qemu/coroutine_int.h | 11 ++- | ||
32 | tests/iothread.h | 25 +++++ | ||
33 | tests/iothread.c | 91 ++++++++++++++++++ | ||
34 | tests/test-aio-multithread.c | 213 +++++++++++++++++++++++++++++++++++++++++++ | ||
35 | util/async.c | 65 +++++++++++++ | ||
36 | util/qemu-coroutine.c | 8 ++ | ||
37 | util/trace-events | 4 + | ||
38 | 9 files changed, 453 insertions(+), 4 deletions(-) | ||
39 | create mode 100644 tests/iothread.h | ||
40 | create mode 100644 tests/iothread.c | ||
41 | create mode 100644 tests/test-aio-multithread.c | ||
42 | |||
43 | diff --git a/tests/Makefile.include b/tests/Makefile.include | ||
44 | index XXXXXXX..XXXXXXX 100644 | ||
45 | --- a/tests/Makefile.include | ||
46 | +++ b/tests/Makefile.include | ||
47 | @@ -XXX,XX +XXX,XX @@ check-unit-y += tests/test-aio$(EXESUF) | ||
48 | gcov-files-test-aio-y = util/async.c util/qemu-timer.o | ||
49 | gcov-files-test-aio-$(CONFIG_WIN32) += util/aio-win32.c | ||
50 | gcov-files-test-aio-$(CONFIG_POSIX) += util/aio-posix.c | ||
51 | +check-unit-y += tests/test-aio-multithread$(EXESUF) | ||
52 | +gcov-files-test-aio-multithread-y = $(gcov-files-test-aio-y) | ||
53 | +gcov-files-test-aio-multithread-y += util/qemu-coroutine.c tests/iothread.c | ||
54 | check-unit-y += tests/test-throttle$(EXESUF) | ||
55 | -gcov-files-test-aio-$(CONFIG_WIN32) = aio-win32.c | ||
56 | -gcov-files-test-aio-$(CONFIG_POSIX) = aio-posix.c | ||
57 | check-unit-y += tests/test-thread-pool$(EXESUF) | ||
58 | gcov-files-test-thread-pool-y = thread-pool.c | ||
59 | gcov-files-test-hbitmap-y = util/hbitmap.c | ||
60 | @@ -XXX,XX +XXX,XX @@ test-qapi-obj-y = tests/test-qapi-visit.o tests/test-qapi-types.o \ | ||
61 | $(test-qom-obj-y) | ||
62 | test-crypto-obj-y = $(crypto-obj-y) $(test-qom-obj-y) | ||
63 | test-io-obj-y = $(io-obj-y) $(test-crypto-obj-y) | ||
64 | -test-block-obj-y = $(block-obj-y) $(test-io-obj-y) | ||
65 | +test-block-obj-y = $(block-obj-y) $(test-io-obj-y) tests/iothread.o | ||
66 | |||
67 | tests/check-qint$(EXESUF): tests/check-qint.o $(test-util-obj-y) | ||
68 | tests/check-qstring$(EXESUF): tests/check-qstring.o $(test-util-obj-y) | ||
69 | @@ -XXX,XX +XXX,XX @@ tests/check-qom-proplist$(EXESUF): tests/check-qom-proplist.o $(test-qom-obj-y) | ||
70 | tests/test-char$(EXESUF): tests/test-char.o $(test-util-obj-y) $(qtest-obj-y) $(test-io-obj-y) $(chardev-obj-y) | ||
71 | tests/test-coroutine$(EXESUF): tests/test-coroutine.o $(test-block-obj-y) | ||
72 | tests/test-aio$(EXESUF): tests/test-aio.o $(test-block-obj-y) | ||
73 | +tests/test-aio-multithread$(EXESUF): tests/test-aio-multithread.o $(test-block-obj-y) | ||
74 | tests/test-throttle$(EXESUF): tests/test-throttle.o $(test-block-obj-y) | ||
75 | tests/test-blockjob$(EXESUF): tests/test-blockjob.o $(test-block-obj-y) $(test-util-obj-y) | ||
76 | tests/test-blockjob-txn$(EXESUF): tests/test-blockjob-txn.o $(test-block-obj-y) $(test-util-obj-y) | ||
77 | diff --git a/include/block/aio.h b/include/block/aio.h | ||
78 | index XXXXXXX..XXXXXXX 100644 | ||
79 | --- a/include/block/aio.h | ||
80 | +++ b/include/block/aio.h | ||
81 | @@ -XXX,XX +XXX,XX @@ typedef void QEMUBHFunc(void *opaque); | ||
82 | typedef bool AioPollFn(void *opaque); | ||
83 | typedef void IOHandler(void *opaque); | ||
84 | |||
85 | +struct Coroutine; | ||
86 | struct ThreadPool; | ||
87 | struct LinuxAioState; | ||
88 | |||
89 | @@ -XXX,XX +XXX,XX @@ struct AioContext { | ||
90 | bool notified; | ||
91 | EventNotifier notifier; | ||
92 | |||
93 | + QSLIST_HEAD(, Coroutine) scheduled_coroutines; | ||
94 | + QEMUBH *co_schedule_bh; | ||
95 | + | ||
96 | /* Thread pool for performing work and receiving completion callbacks. | ||
97 | * Has its own locking. | ||
98 | */ | ||
99 | @@ -XXX,XX +XXX,XX @@ static inline bool aio_node_check(AioContext *ctx, bool is_external) | ||
100 | } | ||
101 | |||
102 | /** | ||
103 | + * aio_co_schedule: | ||
104 | + * @ctx: the aio context | ||
105 | + * @co: the coroutine | ||
106 | + * | ||
107 | + * Start a coroutine on a remote AioContext. | ||
108 | + * | ||
109 | + * The coroutine must not be entered by anyone else while aio_co_schedule() | ||
110 | + * is active. In addition the coroutine must have yielded unless ctx | ||
111 | + * is the context in which the coroutine is running (i.e. the value of | ||
112 | + * qemu_get_current_aio_context() from the coroutine itself). | ||
113 | + */ | ||
114 | +void aio_co_schedule(AioContext *ctx, struct Coroutine *co); | ||
115 | + | ||
116 | +/** | ||
117 | + * aio_co_wake: | ||
118 | + * @co: the coroutine | ||
119 | + * | ||
120 | + * Restart a coroutine on the AioContext where it was running last, thus | ||
121 | + * preventing coroutines from jumping from one context to another when they | ||
122 | + * go to sleep. | ||
123 | + * | ||
124 | + * aio_co_wake may be executed either in coroutine or non-coroutine | ||
125 | + * context. The coroutine must not be entered by anyone else while | ||
126 | + * aio_co_wake() is active. | ||
127 | + */ | ||
128 | +void aio_co_wake(struct Coroutine *co); | ||
129 | + | ||
130 | +/** | ||
131 | * Return the AioContext whose event loop runs in the current thread. | ||
132 | * | ||
133 | * If called from an IOThread this will be the IOThread's AioContext. If | ||
134 | diff --git a/include/qemu/coroutine_int.h b/include/qemu/coroutine_int.h | ||
135 | index XXXXXXX..XXXXXXX 100644 | ||
136 | --- a/include/qemu/coroutine_int.h | ||
137 | +++ b/include/qemu/coroutine_int.h | ||
138 | @@ -XXX,XX +XXX,XX @@ struct Coroutine { | ||
139 | CoroutineEntry *entry; | ||
140 | void *entry_arg; | ||
141 | Coroutine *caller; | ||
142 | + | ||
143 | + /* Only used when the coroutine has terminated. */ | ||
144 | QSLIST_ENTRY(Coroutine) pool_next; | ||
145 | + | ||
146 | size_t locks_held; | ||
147 | |||
148 | - /* Coroutines that should be woken up when we yield or terminate */ | ||
149 | + /* Coroutines that should be woken up when we yield or terminate. | ||
150 | + * Only used when the coroutine is running. | ||
151 | + */ | ||
152 | QSIMPLEQ_HEAD(, Coroutine) co_queue_wakeup; | ||
153 | + | ||
154 | + /* Only used when the coroutine has yielded. */ | ||
155 | + AioContext *ctx; | ||
156 | QSIMPLEQ_ENTRY(Coroutine) co_queue_next; | ||
157 | + QSLIST_ENTRY(Coroutine) co_scheduled_next; | ||
158 | }; | ||
159 | |||
160 | Coroutine *qemu_coroutine_new(void); | ||
161 | diff --git a/tests/iothread.h b/tests/iothread.h | ||
162 | new file mode 100644 | ||
163 | index XXXXXXX..XXXXXXX | ||
164 | --- /dev/null | ||
165 | +++ b/tests/iothread.h | ||
166 | @@ -XXX,XX +XXX,XX @@ | ||
167 | +/* | ||
168 | + * Event loop thread implementation for unit tests | ||
169 | + * | ||
170 | + * Copyright Red Hat Inc., 2013, 2016 | ||
171 | + * | ||
172 | + * Authors: | ||
173 | + * Stefan Hajnoczi <stefanha@redhat.com> | ||
174 | + * Paolo Bonzini <pbonzini@redhat.com> | ||
175 | + * | ||
176 | + * This work is licensed under the terms of the GNU GPL, version 2 or later. | ||
177 | + * See the COPYING file in the top-level directory. | ||
178 | + */ | ||
179 | +#ifndef TEST_IOTHREAD_H | ||
180 | +#define TEST_IOTHREAD_H | ||
181 | + | ||
182 | +#include "block/aio.h" | ||
183 | +#include "qemu/thread.h" | ||
184 | + | ||
185 | +typedef struct IOThread IOThread; | ||
186 | + | ||
187 | +IOThread *iothread_new(void); | ||
188 | +void iothread_join(IOThread *iothread); | ||
189 | +AioContext *iothread_get_aio_context(IOThread *iothread); | ||
190 | + | ||
191 | +#endif | ||
192 | diff --git a/tests/iothread.c b/tests/iothread.c | ||
193 | new file mode 100644 | ||
194 | index XXXXXXX..XXXXXXX | ||
195 | --- /dev/null | ||
196 | +++ b/tests/iothread.c | ||
197 | @@ -XXX,XX +XXX,XX @@ | ||
198 | +/* | ||
199 | + * Event loop thread implementation for unit tests | ||
200 | + * | ||
201 | + * Copyright Red Hat Inc., 2013, 2016 | ||
202 | + * | ||
203 | + * Authors: | ||
204 | + * Stefan Hajnoczi <stefanha@redhat.com> | ||
205 | + * Paolo Bonzini <pbonzini@redhat.com> | ||
206 | + * | ||
207 | + * This work is licensed under the terms of the GNU GPL, version 2 or later. | ||
208 | + * See the COPYING file in the top-level directory. | ||
209 | + * | ||
210 | + */ | ||
211 | + | ||
212 | +#include "qemu/osdep.h" | ||
213 | +#include "qapi/error.h" | ||
214 | +#include "block/aio.h" | ||
215 | +#include "qemu/main-loop.h" | ||
216 | +#include "qemu/rcu.h" | ||
217 | +#include "iothread.h" | ||
218 | + | ||
219 | +struct IOThread { | ||
220 | + AioContext *ctx; | ||
221 | + | ||
222 | + QemuThread thread; | ||
223 | + QemuMutex init_done_lock; | ||
224 | + QemuCond init_done_cond; /* is thread initialization done? */ | ||
225 | + bool stopping; | ||
226 | +}; | ||
227 | + | ||
228 | +static __thread IOThread *my_iothread; | ||
229 | + | ||
230 | +AioContext *qemu_get_current_aio_context(void) | ||
231 | +{ | ||
232 | + return my_iothread ? my_iothread->ctx : qemu_get_aio_context(); | ||
233 | +} | ||
234 | + | ||
235 | +static void *iothread_run(void *opaque) | ||
236 | +{ | ||
237 | + IOThread *iothread = opaque; | ||
238 | + | ||
239 | + rcu_register_thread(); | ||
240 | + | ||
241 | + my_iothread = iothread; | ||
242 | + qemu_mutex_lock(&iothread->init_done_lock); | ||
243 | + iothread->ctx = aio_context_new(&error_abort); | ||
244 | + qemu_cond_signal(&iothread->init_done_cond); | ||
245 | + qemu_mutex_unlock(&iothread->init_done_lock); | ||
246 | + | ||
247 | + while (!atomic_read(&iothread->stopping)) { | ||
248 | + aio_poll(iothread->ctx, true); | ||
249 | + } | ||
250 | + | ||
251 | + rcu_unregister_thread(); | ||
252 | + return NULL; | ||
253 | +} | ||
254 | + | ||
255 | +void iothread_join(IOThread *iothread) | ||
256 | +{ | ||
257 | + iothread->stopping = true; | ||
258 | + aio_notify(iothread->ctx); | ||
259 | + qemu_thread_join(&iothread->thread); | ||
260 | + qemu_cond_destroy(&iothread->init_done_cond); | ||
261 | + qemu_mutex_destroy(&iothread->init_done_lock); | ||
262 | + aio_context_unref(iothread->ctx); | ||
263 | + g_free(iothread); | ||
264 | +} | ||
265 | + | ||
266 | +IOThread *iothread_new(void) | ||
267 | +{ | ||
268 | + IOThread *iothread = g_new0(IOThread, 1); | ||
269 | + | ||
270 | + qemu_mutex_init(&iothread->init_done_lock); | ||
271 | + qemu_cond_init(&iothread->init_done_cond); | ||
272 | + qemu_thread_create(&iothread->thread, NULL, iothread_run, | ||
273 | + iothread, QEMU_THREAD_JOINABLE); | ||
274 | + | ||
275 | + /* Wait for initialization to complete */ | ||
276 | + qemu_mutex_lock(&iothread->init_done_lock); | ||
277 | + while (iothread->ctx == NULL) { | ||
278 | + qemu_cond_wait(&iothread->init_done_cond, | ||
279 | + &iothread->init_done_lock); | ||
280 | + } | ||
281 | + qemu_mutex_unlock(&iothread->init_done_lock); | ||
282 | + return iothread; | ||
283 | +} | ||
284 | + | ||
285 | +AioContext *iothread_get_aio_context(IOThread *iothread) | ||
286 | +{ | ||
287 | + return iothread->ctx; | ||
288 | +} | ||
289 | diff --git a/tests/test-aio-multithread.c b/tests/test-aio-multithread.c | ||
290 | new file mode 100644 | ||
291 | index XXXXXXX..XXXXXXX | ||
292 | --- /dev/null | ||
293 | +++ b/tests/test-aio-multithread.c | ||
294 | @@ -XXX,XX +XXX,XX @@ | ||
295 | +/* | ||
296 | + * AioContext multithreading tests | ||
297 | + * | ||
298 | + * Copyright Red Hat, Inc. 2016 | ||
299 | + * | ||
300 | + * Authors: | ||
301 | + * Paolo Bonzini <pbonzini@redhat.com> | ||
302 | + * | ||
303 | + * This work is licensed under the terms of the GNU LGPL, version 2 or later. | ||
304 | + * See the COPYING.LIB file in the top-level directory. | ||
305 | + */ | ||
306 | + | ||
307 | +#include "qemu/osdep.h" | ||
308 | +#include <glib.h> | ||
309 | +#include "block/aio.h" | ||
310 | +#include "qapi/error.h" | ||
311 | +#include "qemu/coroutine.h" | ||
312 | +#include "qemu/thread.h" | ||
313 | +#include "qemu/error-report.h" | ||
314 | +#include "iothread.h" | ||
315 | + | ||
316 | +/* AioContext management */ | ||
317 | + | ||
318 | +#define NUM_CONTEXTS 5 | ||
319 | + | ||
320 | +static IOThread *threads[NUM_CONTEXTS]; | ||
321 | +static AioContext *ctx[NUM_CONTEXTS]; | ||
322 | +static __thread int id = -1; | ||
323 | + | ||
324 | +static QemuEvent done_event; | ||
325 | + | ||
326 | +/* Run a function synchronously on a remote iothread. */ | ||
327 | + | ||
328 | +typedef struct CtxRunData { | ||
329 | + QEMUBHFunc *cb; | ||
330 | + void *arg; | ||
331 | +} CtxRunData; | ||
332 | + | ||
333 | +static void ctx_run_bh_cb(void *opaque) | ||
334 | +{ | ||
335 | + CtxRunData *data = opaque; | ||
336 | + | ||
337 | + data->cb(data->arg); | ||
338 | + qemu_event_set(&done_event); | ||
339 | +} | ||
340 | + | ||
341 | +static void ctx_run(int i, QEMUBHFunc *cb, void *opaque) | ||
342 | +{ | ||
343 | + CtxRunData data = { | ||
344 | + .cb = cb, | ||
345 | + .arg = opaque | ||
346 | + }; | ||
347 | + | ||
348 | + qemu_event_reset(&done_event); | ||
349 | + aio_bh_schedule_oneshot(ctx[i], ctx_run_bh_cb, &data); | ||
350 | + qemu_event_wait(&done_event); | ||
351 | +} | ||
352 | + | ||
353 | +/* Starting the iothreads. */ | ||
354 | + | ||
355 | +static void set_id_cb(void *opaque) | ||
356 | +{ | ||
357 | + int *i = opaque; | ||
358 | + | ||
359 | + id = *i; | ||
360 | +} | ||
361 | + | ||
362 | +static void create_aio_contexts(void) | ||
363 | +{ | ||
364 | + int i; | ||
365 | + | ||
366 | + for (i = 0; i < NUM_CONTEXTS; i++) { | ||
367 | + threads[i] = iothread_new(); | ||
368 | + ctx[i] = iothread_get_aio_context(threads[i]); | ||
369 | + } | ||
370 | + | ||
371 | + qemu_event_init(&done_event, false); | ||
372 | + for (i = 0; i < NUM_CONTEXTS; i++) { | ||
373 | + ctx_run(i, set_id_cb, &i); | ||
374 | + } | ||
375 | +} | ||
376 | + | ||
377 | +/* Stopping the iothreads. */ | ||
378 | + | ||
379 | +static void join_aio_contexts(void) | ||
380 | +{ | ||
381 | + int i; | ||
382 | + | ||
383 | + for (i = 0; i < NUM_CONTEXTS; i++) { | ||
384 | + aio_context_ref(ctx[i]); | ||
385 | + } | ||
386 | + for (i = 0; i < NUM_CONTEXTS; i++) { | ||
387 | + iothread_join(threads[i]); | ||
388 | + } | ||
389 | + for (i = 0; i < NUM_CONTEXTS; i++) { | ||
390 | + aio_context_unref(ctx[i]); | ||
391 | + } | ||
392 | + qemu_event_destroy(&done_event); | ||
393 | +} | ||
394 | + | ||
395 | +/* Basic test for the stuff above. */ | ||
396 | + | ||
397 | +static void test_lifecycle(void) | ||
398 | +{ | ||
399 | + create_aio_contexts(); | ||
400 | + join_aio_contexts(); | ||
401 | +} | ||
402 | + | ||
403 | +/* aio_co_schedule test. */ | ||
404 | + | ||
405 | +static Coroutine *to_schedule[NUM_CONTEXTS]; | ||
406 | + | ||
407 | +static bool now_stopping; | ||
408 | + | ||
409 | +static int count_retry; | ||
410 | +static int count_here; | ||
411 | +static int count_other; | ||
412 | + | ||
413 | +static bool schedule_next(int n) | ||
414 | +{ | ||
415 | + Coroutine *co; | ||
416 | + | ||
417 | + co = atomic_xchg(&to_schedule[n], NULL); | ||
418 | + if (!co) { | ||
419 | + atomic_inc(&count_retry); | ||
420 | + return false; | ||
421 | + } | ||
422 | + | ||
423 | + if (n == id) { | ||
424 | + atomic_inc(&count_here); | ||
425 | + } else { | ||
426 | + atomic_inc(&count_other); | ||
427 | + } | ||
428 | + | ||
429 | + aio_co_schedule(ctx[n], co); | ||
430 | + return true; | ||
431 | +} | ||
432 | + | ||
433 | +static void finish_cb(void *opaque) | ||
434 | +{ | ||
435 | + schedule_next(id); | ||
436 | +} | ||
437 | + | ||
438 | +static coroutine_fn void test_multi_co_schedule_entry(void *opaque) | ||
439 | +{ | ||
440 | + g_assert(to_schedule[id] == NULL); | ||
441 | + atomic_mb_set(&to_schedule[id], qemu_coroutine_self()); | ||
442 | + | ||
443 | + while (!atomic_mb_read(&now_stopping)) { | ||
444 | + int n; | ||
445 | + | ||
446 | + n = g_test_rand_int_range(0, NUM_CONTEXTS); | ||
447 | + schedule_next(n); | ||
448 | + qemu_coroutine_yield(); | ||
449 | + | ||
450 | + g_assert(to_schedule[id] == NULL); | ||
451 | + atomic_mb_set(&to_schedule[id], qemu_coroutine_self()); | ||
452 | + } | ||
453 | +} | ||
454 | + | ||
455 | + | ||
456 | +static void test_multi_co_schedule(int seconds) | ||
457 | +{ | ||
458 | + int i; | ||
459 | + | ||
460 | + count_here = count_other = count_retry = 0; | ||
461 | + now_stopping = false; | ||
462 | + | ||
463 | + create_aio_contexts(); | ||
464 | + for (i = 0; i < NUM_CONTEXTS; i++) { | ||
465 | + Coroutine *co1 = qemu_coroutine_create(test_multi_co_schedule_entry, NULL); | ||
466 | + aio_co_schedule(ctx[i], co1); | ||
467 | + } | ||
468 | + | ||
469 | + g_usleep(seconds * 1000000); | ||
470 | + | ||
471 | + atomic_mb_set(&now_stopping, true); | ||
472 | + for (i = 0; i < NUM_CONTEXTS; i++) { | ||
473 | + ctx_run(i, finish_cb, NULL); | ||
474 | + to_schedule[i] = NULL; | ||
475 | + } | ||
476 | + | ||
477 | + join_aio_contexts(); | ||
478 | + g_test_message("scheduled %d, queued %d, retry %d, total %d\n", | ||
479 | + count_other, count_here, count_retry, | ||
480 | + count_here + count_other + count_retry); | ||
481 | +} | ||
482 | + | ||
483 | +static void test_multi_co_schedule_1(void) | ||
484 | +{ | ||
485 | + test_multi_co_schedule(1); | ||
486 | +} | ||
487 | + | ||
488 | +static void test_multi_co_schedule_10(void) | ||
489 | +{ | ||
490 | + test_multi_co_schedule(10); | ||
491 | +} | ||
492 | + | ||
493 | +/* End of tests. */ | ||
494 | + | ||
495 | +int main(int argc, char **argv) | ||
496 | +{ | ||
497 | + init_clocks(); | ||
498 | + | ||
499 | + g_test_init(&argc, &argv, NULL); | ||
500 | + g_test_add_func("/aio/multi/lifecycle", test_lifecycle); | ||
501 | + if (g_test_quick()) { | ||
502 | + g_test_add_func("/aio/multi/schedule", test_multi_co_schedule_1); | ||
503 | + } else { | ||
504 | + g_test_add_func("/aio/multi/schedule", test_multi_co_schedule_10); | ||
505 | + } | ||
506 | + return g_test_run(); | ||
507 | +} | ||
508 | diff --git a/util/async.c b/util/async.c | ||
509 | index XXXXXXX..XXXXXXX 100644 | ||
510 | --- a/util/async.c | ||
511 | +++ b/util/async.c | ||
512 | @@ -XXX,XX +XXX,XX @@ | ||
513 | #include "qemu/main-loop.h" | ||
514 | #include "qemu/atomic.h" | ||
515 | #include "block/raw-aio.h" | ||
516 | +#include "qemu/coroutine_int.h" | ||
517 | +#include "trace.h" | ||
518 | |||
519 | /***********************************************************/ | ||
520 | /* bottom halves (can be seen as timers which expire ASAP) */ | ||
521 | @@ -XXX,XX +XXX,XX @@ aio_ctx_finalize(GSource *source) | ||
522 | } | ||
523 | #endif | ||
524 | |||
525 | + assert(QSLIST_EMPTY(&ctx->scheduled_coroutines)); | ||
526 | + qemu_bh_delete(ctx->co_schedule_bh); | ||
527 | + | ||
528 | qemu_lockcnt_lock(&ctx->list_lock); | ||
529 | assert(!qemu_lockcnt_count(&ctx->list_lock)); | ||
530 | while (ctx->first_bh) { | ||
531 | @@ -XXX,XX +XXX,XX @@ static bool event_notifier_poll(void *opaque) | ||
532 | return atomic_read(&ctx->notified); | ||
533 | } | ||
534 | |||
535 | +static void co_schedule_bh_cb(void *opaque) | ||
536 | +{ | ||
537 | + AioContext *ctx = opaque; | ||
538 | + QSLIST_HEAD(, Coroutine) straight, reversed; | ||
539 | + | ||
540 | + QSLIST_MOVE_ATOMIC(&reversed, &ctx->scheduled_coroutines); | ||
541 | + QSLIST_INIT(&straight); | ||
542 | + | ||
543 | + while (!QSLIST_EMPTY(&reversed)) { | ||
544 | + Coroutine *co = QSLIST_FIRST(&reversed); | ||
545 | + QSLIST_REMOVE_HEAD(&reversed, co_scheduled_next); | ||
546 | + QSLIST_INSERT_HEAD(&straight, co, co_scheduled_next); | ||
547 | + } | ||
548 | + | ||
549 | + while (!QSLIST_EMPTY(&straight)) { | ||
550 | + Coroutine *co = QSLIST_FIRST(&straight); | ||
551 | + QSLIST_REMOVE_HEAD(&straight, co_scheduled_next); | ||
552 | + trace_aio_co_schedule_bh_cb(ctx, co); | ||
553 | + qemu_coroutine_enter(co); | ||
554 | + } | ||
555 | +} | ||
556 | + | ||
557 | AioContext *aio_context_new(Error **errp) | ||
558 | { | ||
559 | int ret; | ||
560 | @@ -XXX,XX +XXX,XX @@ AioContext *aio_context_new(Error **errp) | ||
561 | } | ||
562 | g_source_set_can_recurse(&ctx->source, true); | ||
563 | qemu_lockcnt_init(&ctx->list_lock); | ||
564 | + | ||
565 | + ctx->co_schedule_bh = aio_bh_new(ctx, co_schedule_bh_cb, ctx); | ||
566 | + QSLIST_INIT(&ctx->scheduled_coroutines); | ||
567 | + | ||
568 | aio_set_event_notifier(ctx, &ctx->notifier, | ||
569 | false, | ||
570 | (EventNotifierHandler *) | ||
571 | @@ -XXX,XX +XXX,XX @@ fail: | ||
572 | return NULL; | ||
573 | } | ||
574 | |||
575 | +void aio_co_schedule(AioContext *ctx, Coroutine *co) | ||
576 | +{ | ||
577 | + trace_aio_co_schedule(ctx, co); | ||
578 | + QSLIST_INSERT_HEAD_ATOMIC(&ctx->scheduled_coroutines, | ||
579 | + co, co_scheduled_next); | ||
580 | + qemu_bh_schedule(ctx->co_schedule_bh); | ||
581 | +} | ||
582 | + | ||
583 | +void aio_co_wake(struct Coroutine *co) | ||
584 | +{ | ||
585 | + AioContext *ctx; | ||
586 | + | ||
587 | + /* Read coroutine before co->ctx. Matches smp_wmb in | ||
588 | + * qemu_coroutine_enter. | ||
589 | + */ | ||
590 | + smp_read_barrier_depends(); | ||
591 | + ctx = atomic_read(&co->ctx); | ||
592 | + | ||
593 | + if (ctx != qemu_get_current_aio_context()) { | ||
594 | + aio_co_schedule(ctx, co); | ||
595 | + return; | ||
596 | + } | ||
597 | + | ||
598 | + if (qemu_in_coroutine()) { | ||
599 | + Coroutine *self = qemu_coroutine_self(); | ||
600 | + assert(self != co); | ||
601 | + QSIMPLEQ_INSERT_TAIL(&self->co_queue_wakeup, co, co_queue_next); | ||
602 | + } else { | ||
603 | + aio_context_acquire(ctx); | ||
604 | + qemu_coroutine_enter(co); | ||
605 | + aio_context_release(ctx); | ||
606 | + } | ||
607 | +} | ||
608 | + | ||
609 | void aio_context_ref(AioContext *ctx) | ||
610 | { | ||
611 | g_source_ref(&ctx->source); | ||
612 | diff --git a/util/qemu-coroutine.c b/util/qemu-coroutine.c | ||
613 | index XXXXXXX..XXXXXXX 100644 | ||
614 | --- a/util/qemu-coroutine.c | ||
615 | +++ b/util/qemu-coroutine.c | ||
616 | @@ -XXX,XX +XXX,XX @@ | ||
617 | #include "qemu/atomic.h" | ||
618 | #include "qemu/coroutine.h" | ||
619 | #include "qemu/coroutine_int.h" | ||
620 | +#include "block/aio.h" | ||
621 | |||
622 | enum { | ||
623 | POOL_BATCH_SIZE = 64, | ||
624 | @@ -XXX,XX +XXX,XX @@ void qemu_coroutine_enter(Coroutine *co) | ||
625 | } | ||
626 | |||
627 | co->caller = self; | ||
628 | + co->ctx = qemu_get_current_aio_context(); | ||
629 | + | ||
630 | + /* Store co->ctx before anything that stores co. Matches | ||
631 | + * barrier in aio_co_wake. | ||
632 | + */ | ||
633 | + smp_wmb(); | ||
634 | + | ||
635 | ret = qemu_coroutine_switch(self, co, COROUTINE_ENTER); | ||
636 | |||
637 | qemu_co_queue_run_restart(co); | ||
638 | diff --git a/util/trace-events b/util/trace-events | ||
639 | index XXXXXXX..XXXXXXX 100644 | ||
640 | --- a/util/trace-events | ||
641 | +++ b/util/trace-events | ||
642 | @@ -XXX,XX +XXX,XX @@ run_poll_handlers_end(void *ctx, bool progress) "ctx %p progress %d" | ||
643 | poll_shrink(void *ctx, int64_t old, int64_t new) "ctx %p old %"PRId64" new %"PRId64 | ||
644 | poll_grow(void *ctx, int64_t old, int64_t new) "ctx %p old %"PRId64" new %"PRId64 | ||
645 | |||
646 | +# util/async.c | ||
647 | +aio_co_schedule(void *ctx, void *co) "ctx %p co %p" | ||
648 | +aio_co_schedule_bh_cb(void *ctx, void *co) "ctx %p co %p" | ||
649 | + | ||
650 | # util/thread-pool.c | ||
651 | thread_pool_submit(void *pool, void *req, void *opaque) "pool %p req %p opaque %p" | ||
652 | thread_pool_complete(void *pool, void *req, void *opaque, int ret) "pool %p req %p opaque %p ret %d" | ||
653 | -- | ||
654 | 2.9.3 | ||
655 | |||
656 | diff view generated by jsdifflib |
1 | From: Emanuele Giuseppe Esposito <eesposit@redhat.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | pylint 2.8 introduces consider-using-with error, suggesting | 3 | qcow2_create2 calls this. Do not run a nested event loop, as that |
4 | to use the 'with' block statement when possible. | 4 | breaks when aio_co_wake tries to queue the coroutine on the co_queue_wakeup |
5 | list of the currently running one. | ||
5 | 6 | ||
6 | Modify all subprocess.Popen call to use the 'with' statement, | 7 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
7 | except the one in __init__ of QemuIoInteractive class, since | 8 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
8 | it is assigned to a class field and used in other methods. | 9 | Reviewed-by: Fam Zheng <famz@redhat.com> |
10 | Message-id: 20170213135235.12274-4-pbonzini@redhat.com | ||
11 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
12 | --- | ||
13 | block/block-backend.c | 12 ++++++++---- | ||
14 | 1 file changed, 8 insertions(+), 4 deletions(-) | ||
9 | 15 | ||
10 | Signed-off-by: Emanuele Giuseppe Esposito <eesposit@redhat.com> | 16 | diff --git a/block/block-backend.c b/block/block-backend.c |
11 | Message-Id: <20210510190449.65948-1-eesposit@redhat.com> | ||
12 | [mreitz: Disable bad-option-value warning in the iotests' pylintrc, so | ||
13 | that disabling consider-using-with in QemuIoInteractive will | ||
14 | not produce a warning in pre-2.8 pylint versions] | ||
15 | Signed-off-by: Max Reitz <mreitz@redhat.com> | ||
16 | --- | ||
17 | tests/qemu-iotests/iotests.py | 65 ++++++++++++++++---------------- | ||
18 | tests/qemu-iotests/pylintrc | 3 ++ | ||
19 | tests/qemu-iotests/testrunner.py | 22 +++++------ | ||
20 | 3 files changed, 47 insertions(+), 43 deletions(-) | ||
21 | |||
22 | diff --git a/tests/qemu-iotests/iotests.py b/tests/qemu-iotests/iotests.py | ||
23 | index XXXXXXX..XXXXXXX 100644 | 17 | index XXXXXXX..XXXXXXX 100644 |
24 | --- a/tests/qemu-iotests/iotests.py | 18 | --- a/block/block-backend.c |
25 | +++ b/tests/qemu-iotests/iotests.py | 19 | +++ b/block/block-backend.c |
26 | @@ -XXX,XX +XXX,XX @@ def qemu_tool_pipe_and_status(tool: str, args: Sequence[str], | 20 | @@ -XXX,XX +XXX,XX @@ static int blk_prw(BlockBackend *blk, int64_t offset, uint8_t *buf, |
27 | Run a tool and return both its output and its exit code | 21 | { |
28 | """ | 22 | QEMUIOVector qiov; |
29 | stderr = subprocess.STDOUT if connect_stderr else None | 23 | struct iovec iov; |
30 | - subp = subprocess.Popen(args, | 24 | - Coroutine *co; |
31 | - stdout=subprocess.PIPE, | 25 | BlkRwCo rwco; |
32 | - stderr=stderr, | 26 | |
33 | - universal_newlines=True) | 27 | iov = (struct iovec) { |
34 | - output = subp.communicate()[0] | 28 | @@ -XXX,XX +XXX,XX @@ static int blk_prw(BlockBackend *blk, int64_t offset, uint8_t *buf, |
35 | - if subp.returncode < 0: | 29 | .ret = NOT_DONE, |
36 | - cmd = ' '.join(args) | 30 | }; |
37 | - sys.stderr.write(f'{tool} received signal {-subp.returncode}: {cmd}\n') | 31 | |
38 | - return (output, subp.returncode) | 32 | - co = qemu_coroutine_create(co_entry, &rwco); |
39 | + with subprocess.Popen(args, stdout=subprocess.PIPE, | 33 | - qemu_coroutine_enter(co); |
40 | + stderr=stderr, universal_newlines=True) as subp: | 34 | - BDRV_POLL_WHILE(blk_bs(blk), rwco.ret == NOT_DONE); |
41 | + output = subp.communicate()[0] | 35 | + if (qemu_in_coroutine()) { |
42 | + if subp.returncode < 0: | 36 | + /* Fast-path if already in coroutine context */ |
43 | + cmd = ' '.join(args) | 37 | + co_entry(&rwco); |
44 | + sys.stderr.write(f'{tool} received signal \ | 38 | + } else { |
45 | + {-subp.returncode}: {cmd}\n') | 39 | + Coroutine *co = qemu_coroutine_create(co_entry, &rwco); |
46 | + return (output, subp.returncode) | 40 | + qemu_coroutine_enter(co); |
47 | 41 | + BDRV_POLL_WHILE(blk_bs(blk), rwco.ret == NOT_DONE); | |
48 | def qemu_img_pipe_and_status(*args: str) -> Tuple[str, int]: | 42 | + } |
49 | """ | 43 | |
50 | @@ -XXX,XX +XXX,XX @@ def qemu_io_silent_check(*args): | 44 | return rwco.ret; |
51 | class QemuIoInteractive: | 45 | } |
52 | def __init__(self, *args): | ||
53 | self.args = qemu_io_args_no_fmt + list(args) | ||
54 | + # We need to keep the Popen objext around, and not | ||
55 | + # close it immediately. Therefore, disable the pylint check: | ||
56 | + # pylint: disable=consider-using-with | ||
57 | self._p = subprocess.Popen(self.args, stdin=subprocess.PIPE, | ||
58 | stdout=subprocess.PIPE, | ||
59 | stderr=subprocess.STDOUT, | ||
60 | @@ -XXX,XX +XXX,XX @@ def qemu_nbd_popen(*args): | ||
61 | cmd.extend(args) | ||
62 | |||
63 | log('Start NBD server') | ||
64 | - p = subprocess.Popen(cmd) | ||
65 | - try: | ||
66 | - while not os.path.exists(pid_file): | ||
67 | - if p.poll() is not None: | ||
68 | - raise RuntimeError( | ||
69 | - "qemu-nbd terminated with exit code {}: {}" | ||
70 | - .format(p.returncode, ' '.join(cmd))) | ||
71 | - | ||
72 | - time.sleep(0.01) | ||
73 | - yield | ||
74 | - finally: | ||
75 | - if os.path.exists(pid_file): | ||
76 | - os.remove(pid_file) | ||
77 | - log('Kill NBD server') | ||
78 | - p.kill() | ||
79 | - p.wait() | ||
80 | + with subprocess.Popen(cmd) as p: | ||
81 | + try: | ||
82 | + while not os.path.exists(pid_file): | ||
83 | + if p.poll() is not None: | ||
84 | + raise RuntimeError( | ||
85 | + "qemu-nbd terminated with exit code {}: {}" | ||
86 | + .format(p.returncode, ' '.join(cmd))) | ||
87 | + | ||
88 | + time.sleep(0.01) | ||
89 | + yield | ||
90 | + finally: | ||
91 | + if os.path.exists(pid_file): | ||
92 | + os.remove(pid_file) | ||
93 | + log('Kill NBD server') | ||
94 | + p.kill() | ||
95 | + p.wait() | ||
96 | |||
97 | def compare_images(img1, img2, fmt1=imgfmt, fmt2=imgfmt): | ||
98 | '''Return True if two image files are identical''' | ||
99 | @@ -XXX,XX +XXX,XX @@ def compare_images(img1, img2, fmt1=imgfmt, fmt2=imgfmt): | ||
100 | |||
101 | def create_image(name, size): | ||
102 | '''Create a fully-allocated raw image with sector markers''' | ||
103 | - file = open(name, 'wb') | ||
104 | - i = 0 | ||
105 | - while i < size: | ||
106 | - sector = struct.pack('>l504xl', i // 512, i // 512) | ||
107 | - file.write(sector) | ||
108 | - i = i + 512 | ||
109 | - file.close() | ||
110 | + with open(name, 'wb') as file: | ||
111 | + i = 0 | ||
112 | + while i < size: | ||
113 | + sector = struct.pack('>l504xl', i // 512, i // 512) | ||
114 | + file.write(sector) | ||
115 | + i = i + 512 | ||
116 | |||
117 | def image_size(img): | ||
118 | '''Return image's virtual size''' | ||
119 | diff --git a/tests/qemu-iotests/pylintrc b/tests/qemu-iotests/pylintrc | ||
120 | index XXXXXXX..XXXXXXX 100644 | ||
121 | --- a/tests/qemu-iotests/pylintrc | ||
122 | +++ b/tests/qemu-iotests/pylintrc | ||
123 | @@ -XXX,XX +XXX,XX @@ disable=invalid-name, | ||
124 | too-many-public-methods, | ||
125 | # pylint warns about Optional[] etc. as unsubscriptable in 3.9 | ||
126 | unsubscriptable-object, | ||
127 | + # Sometimes we need to disable a newly introduced pylint warning. | ||
128 | + # Doing so should not produce a warning in older versions of pylint. | ||
129 | + bad-option-value, | ||
130 | # These are temporary, and should be removed: | ||
131 | missing-docstring, | ||
132 | too-many-return-statements, | ||
133 | diff --git a/tests/qemu-iotests/testrunner.py b/tests/qemu-iotests/testrunner.py | ||
134 | index XXXXXXX..XXXXXXX 100644 | ||
135 | --- a/tests/qemu-iotests/testrunner.py | ||
136 | +++ b/tests/qemu-iotests/testrunner.py | ||
137 | @@ -XXX,XX +XXX,XX @@ def do_run_test(self, test: str) -> TestResult: | ||
138 | |||
139 | t0 = time.time() | ||
140 | with f_bad.open('w', encoding="utf-8") as f: | ||
141 | - proc = subprocess.Popen(args, cwd=str(f_test.parent), env=env, | ||
142 | - stdout=f, stderr=subprocess.STDOUT) | ||
143 | - try: | ||
144 | - proc.wait() | ||
145 | - except KeyboardInterrupt: | ||
146 | - proc.terminate() | ||
147 | - proc.wait() | ||
148 | - return TestResult(status='not run', | ||
149 | - description='Interrupted by user', | ||
150 | - interrupted=True) | ||
151 | - ret = proc.returncode | ||
152 | + with subprocess.Popen(args, cwd=str(f_test.parent), env=env, | ||
153 | + stdout=f, stderr=subprocess.STDOUT) as proc: | ||
154 | + try: | ||
155 | + proc.wait() | ||
156 | + except KeyboardInterrupt: | ||
157 | + proc.terminate() | ||
158 | + proc.wait() | ||
159 | + return TestResult(status='not run', | ||
160 | + description='Interrupted by user', | ||
161 | + interrupted=True) | ||
162 | + ret = proc.returncode | ||
163 | |||
164 | elapsed = round(time.time() - t0, 1) | ||
165 | |||
166 | -- | 46 | -- |
167 | 2.31.1 | 47 | 2.9.3 |
168 | 48 | ||
169 | 49 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Paolo Bonzini <pbonzini@redhat.com> | ||
1 | 2 | ||
3 | Once the thread pool starts using aio_co_wake, it will also need | ||
4 | qemu_get_current_aio_context(). Make test-thread-pool create | ||
5 | an AioContext with qemu_init_main_loop, so that stubs/iothread.c | ||
6 | and tests/iothread.c can provide the rest. | ||
7 | |||
8 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
9 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> | ||
10 | Reviewed-by: Fam Zheng <famz@redhat.com> | ||
11 | Message-id: 20170213135235.12274-5-pbonzini@redhat.com | ||
12 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
13 | --- | ||
14 | tests/test-thread-pool.c | 12 +++--------- | ||
15 | 1 file changed, 3 insertions(+), 9 deletions(-) | ||
16 | |||
17 | diff --git a/tests/test-thread-pool.c b/tests/test-thread-pool.c | ||
18 | index XXXXXXX..XXXXXXX 100644 | ||
19 | --- a/tests/test-thread-pool.c | ||
20 | +++ b/tests/test-thread-pool.c | ||
21 | @@ -XXX,XX +XXX,XX @@ | ||
22 | #include "qapi/error.h" | ||
23 | #include "qemu/timer.h" | ||
24 | #include "qemu/error-report.h" | ||
25 | +#include "qemu/main-loop.h" | ||
26 | |||
27 | static AioContext *ctx; | ||
28 | static ThreadPool *pool; | ||
29 | @@ -XXX,XX +XXX,XX @@ static void test_cancel_async(void) | ||
30 | int main(int argc, char **argv) | ||
31 | { | ||
32 | int ret; | ||
33 | - Error *local_error = NULL; | ||
34 | |||
35 | - init_clocks(); | ||
36 | - | ||
37 | - ctx = aio_context_new(&local_error); | ||
38 | - if (!ctx) { | ||
39 | - error_reportf_err(local_error, "Failed to create AIO Context: "); | ||
40 | - exit(1); | ||
41 | - } | ||
42 | + qemu_init_main_loop(&error_abort); | ||
43 | + ctx = qemu_get_current_aio_context(); | ||
44 | pool = aio_get_thread_pool(ctx); | ||
45 | |||
46 | g_test_init(&argc, &argv, NULL); | ||
47 | @@ -XXX,XX +XXX,XX @@ int main(int argc, char **argv) | ||
48 | |||
49 | ret = g_test_run(); | ||
50 | |||
51 | - aio_context_unref(ctx); | ||
52 | return ret; | ||
53 | } | ||
54 | -- | ||
55 | 2.9.3 | ||
56 | |||
57 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | 1 | From: Paolo Bonzini <pbonzini@redhat.com> | |
2 | |||
3 | This is in preparation for making qio_channel_yield work on | ||
4 | AioContexts other than the main one. | ||
5 | |||
6 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> | ||
7 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
8 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> | ||
9 | Reviewed-by: Fam Zheng <famz@redhat.com> | ||
10 | Message-id: 20170213135235.12274-6-pbonzini@redhat.com | ||
11 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
12 | --- | ||
13 | include/io/channel.h | 25 +++++++++++++++++++++++++ | ||
14 | io/channel-command.c | 13 +++++++++++++ | ||
15 | io/channel-file.c | 11 +++++++++++ | ||
16 | io/channel-socket.c | 16 +++++++++++----- | ||
17 | io/channel-tls.c | 12 ++++++++++++ | ||
18 | io/channel-watch.c | 6 ++++++ | ||
19 | io/channel.c | 11 +++++++++++ | ||
20 | 7 files changed, 89 insertions(+), 5 deletions(-) | ||
21 | |||
22 | diff --git a/include/io/channel.h b/include/io/channel.h | ||
23 | index XXXXXXX..XXXXXXX 100644 | ||
24 | --- a/include/io/channel.h | ||
25 | +++ b/include/io/channel.h | ||
26 | @@ -XXX,XX +XXX,XX @@ | ||
27 | |||
28 | #include "qemu-common.h" | ||
29 | #include "qom/object.h" | ||
30 | +#include "block/aio.h" | ||
31 | |||
32 | #define TYPE_QIO_CHANNEL "qio-channel" | ||
33 | #define QIO_CHANNEL(obj) \ | ||
34 | @@ -XXX,XX +XXX,XX @@ struct QIOChannelClass { | ||
35 | off_t offset, | ||
36 | int whence, | ||
37 | Error **errp); | ||
38 | + void (*io_set_aio_fd_handler)(QIOChannel *ioc, | ||
39 | + AioContext *ctx, | ||
40 | + IOHandler *io_read, | ||
41 | + IOHandler *io_write, | ||
42 | + void *opaque); | ||
43 | }; | ||
44 | |||
45 | /* General I/O handling functions */ | ||
46 | @@ -XXX,XX +XXX,XX @@ void qio_channel_yield(QIOChannel *ioc, | ||
47 | void qio_channel_wait(QIOChannel *ioc, | ||
48 | GIOCondition condition); | ||
49 | |||
50 | +/** | ||
51 | + * qio_channel_set_aio_fd_handler: | ||
52 | + * @ioc: the channel object | ||
53 | + * @ctx: the AioContext to set the handlers on | ||
54 | + * @io_read: the read handler | ||
55 | + * @io_write: the write handler | ||
56 | + * @opaque: the opaque value passed to the handler | ||
57 | + * | ||
58 | + * This is used internally by qio_channel_yield(). It can | ||
59 | + * be used by channel implementations to forward the handlers | ||
60 | + * to another channel (e.g. from #QIOChannelTLS to the | ||
61 | + * underlying socket). | ||
62 | + */ | ||
63 | +void qio_channel_set_aio_fd_handler(QIOChannel *ioc, | ||
64 | + AioContext *ctx, | ||
65 | + IOHandler *io_read, | ||
66 | + IOHandler *io_write, | ||
67 | + void *opaque); | ||
68 | + | ||
69 | #endif /* QIO_CHANNEL_H */ | ||
70 | diff --git a/io/channel-command.c b/io/channel-command.c | ||
71 | index XXXXXXX..XXXXXXX 100644 | ||
72 | --- a/io/channel-command.c | ||
73 | +++ b/io/channel-command.c | ||
74 | @@ -XXX,XX +XXX,XX @@ static int qio_channel_command_close(QIOChannel *ioc, | ||
75 | } | ||
76 | |||
77 | |||
78 | +static void qio_channel_command_set_aio_fd_handler(QIOChannel *ioc, | ||
79 | + AioContext *ctx, | ||
80 | + IOHandler *io_read, | ||
81 | + IOHandler *io_write, | ||
82 | + void *opaque) | ||
83 | +{ | ||
84 | + QIOChannelCommand *cioc = QIO_CHANNEL_COMMAND(ioc); | ||
85 | + aio_set_fd_handler(ctx, cioc->readfd, false, io_read, NULL, NULL, opaque); | ||
86 | + aio_set_fd_handler(ctx, cioc->writefd, false, NULL, io_write, NULL, opaque); | ||
87 | +} | ||
88 | + | ||
89 | + | ||
90 | static GSource *qio_channel_command_create_watch(QIOChannel *ioc, | ||
91 | GIOCondition condition) | ||
92 | { | ||
93 | @@ -XXX,XX +XXX,XX @@ static void qio_channel_command_class_init(ObjectClass *klass, | ||
94 | ioc_klass->io_set_blocking = qio_channel_command_set_blocking; | ||
95 | ioc_klass->io_close = qio_channel_command_close; | ||
96 | ioc_klass->io_create_watch = qio_channel_command_create_watch; | ||
97 | + ioc_klass->io_set_aio_fd_handler = qio_channel_command_set_aio_fd_handler; | ||
98 | } | ||
99 | |||
100 | static const TypeInfo qio_channel_command_info = { | ||
101 | diff --git a/io/channel-file.c b/io/channel-file.c | ||
102 | index XXXXXXX..XXXXXXX 100644 | ||
103 | --- a/io/channel-file.c | ||
104 | +++ b/io/channel-file.c | ||
105 | @@ -XXX,XX +XXX,XX @@ static int qio_channel_file_close(QIOChannel *ioc, | ||
106 | } | ||
107 | |||
108 | |||
109 | +static void qio_channel_file_set_aio_fd_handler(QIOChannel *ioc, | ||
110 | + AioContext *ctx, | ||
111 | + IOHandler *io_read, | ||
112 | + IOHandler *io_write, | ||
113 | + void *opaque) | ||
114 | +{ | ||
115 | + QIOChannelFile *fioc = QIO_CHANNEL_FILE(ioc); | ||
116 | + aio_set_fd_handler(ctx, fioc->fd, false, io_read, io_write, NULL, opaque); | ||
117 | +} | ||
118 | + | ||
119 | static GSource *qio_channel_file_create_watch(QIOChannel *ioc, | ||
120 | GIOCondition condition) | ||
121 | { | ||
122 | @@ -XXX,XX +XXX,XX @@ static void qio_channel_file_class_init(ObjectClass *klass, | ||
123 | ioc_klass->io_seek = qio_channel_file_seek; | ||
124 | ioc_klass->io_close = qio_channel_file_close; | ||
125 | ioc_klass->io_create_watch = qio_channel_file_create_watch; | ||
126 | + ioc_klass->io_set_aio_fd_handler = qio_channel_file_set_aio_fd_handler; | ||
127 | } | ||
128 | |||
129 | static const TypeInfo qio_channel_file_info = { | ||
130 | diff --git a/io/channel-socket.c b/io/channel-socket.c | ||
131 | index XXXXXXX..XXXXXXX 100644 | ||
132 | --- a/io/channel-socket.c | ||
133 | +++ b/io/channel-socket.c | ||
134 | @@ -XXX,XX +XXX,XX @@ qio_channel_socket_set_blocking(QIOChannel *ioc, | ||
135 | qemu_set_block(sioc->fd); | ||
136 | } else { | ||
137 | qemu_set_nonblock(sioc->fd); | ||
138 | -#ifdef WIN32 | ||
139 | - WSAEventSelect(sioc->fd, ioc->event, | ||
140 | - FD_READ | FD_ACCEPT | FD_CLOSE | | ||
141 | - FD_CONNECT | FD_WRITE | FD_OOB); | ||
142 | -#endif | ||
143 | } | ||
144 | return 0; | ||
145 | } | ||
146 | @@ -XXX,XX +XXX,XX @@ qio_channel_socket_shutdown(QIOChannel *ioc, | ||
147 | return 0; | ||
148 | } | ||
149 | |||
150 | +static void qio_channel_socket_set_aio_fd_handler(QIOChannel *ioc, | ||
151 | + AioContext *ctx, | ||
152 | + IOHandler *io_read, | ||
153 | + IOHandler *io_write, | ||
154 | + void *opaque) | ||
155 | +{ | ||
156 | + QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc); | ||
157 | + aio_set_fd_handler(ctx, sioc->fd, false, io_read, io_write, NULL, opaque); | ||
158 | +} | ||
159 | + | ||
160 | static GSource *qio_channel_socket_create_watch(QIOChannel *ioc, | ||
161 | GIOCondition condition) | ||
162 | { | ||
163 | @@ -XXX,XX +XXX,XX @@ static void qio_channel_socket_class_init(ObjectClass *klass, | ||
164 | ioc_klass->io_set_cork = qio_channel_socket_set_cork; | ||
165 | ioc_klass->io_set_delay = qio_channel_socket_set_delay; | ||
166 | ioc_klass->io_create_watch = qio_channel_socket_create_watch; | ||
167 | + ioc_klass->io_set_aio_fd_handler = qio_channel_socket_set_aio_fd_handler; | ||
168 | } | ||
169 | |||
170 | static const TypeInfo qio_channel_socket_info = { | ||
171 | diff --git a/io/channel-tls.c b/io/channel-tls.c | ||
172 | index XXXXXXX..XXXXXXX 100644 | ||
173 | --- a/io/channel-tls.c | ||
174 | +++ b/io/channel-tls.c | ||
175 | @@ -XXX,XX +XXX,XX @@ static int qio_channel_tls_close(QIOChannel *ioc, | ||
176 | return qio_channel_close(tioc->master, errp); | ||
177 | } | ||
178 | |||
179 | +static void qio_channel_tls_set_aio_fd_handler(QIOChannel *ioc, | ||
180 | + AioContext *ctx, | ||
181 | + IOHandler *io_read, | ||
182 | + IOHandler *io_write, | ||
183 | + void *opaque) | ||
184 | +{ | ||
185 | + QIOChannelTLS *tioc = QIO_CHANNEL_TLS(ioc); | ||
186 | + | ||
187 | + qio_channel_set_aio_fd_handler(tioc->master, ctx, io_read, io_write, opaque); | ||
188 | +} | ||
189 | + | ||
190 | static GSource *qio_channel_tls_create_watch(QIOChannel *ioc, | ||
191 | GIOCondition condition) | ||
192 | { | ||
193 | @@ -XXX,XX +XXX,XX @@ static void qio_channel_tls_class_init(ObjectClass *klass, | ||
194 | ioc_klass->io_close = qio_channel_tls_close; | ||
195 | ioc_klass->io_shutdown = qio_channel_tls_shutdown; | ||
196 | ioc_klass->io_create_watch = qio_channel_tls_create_watch; | ||
197 | + ioc_klass->io_set_aio_fd_handler = qio_channel_tls_set_aio_fd_handler; | ||
198 | } | ||
199 | |||
200 | static const TypeInfo qio_channel_tls_info = { | ||
201 | diff --git a/io/channel-watch.c b/io/channel-watch.c | ||
202 | index XXXXXXX..XXXXXXX 100644 | ||
203 | --- a/io/channel-watch.c | ||
204 | +++ b/io/channel-watch.c | ||
205 | @@ -XXX,XX +XXX,XX @@ GSource *qio_channel_create_socket_watch(QIOChannel *ioc, | ||
206 | GSource *source; | ||
207 | QIOChannelSocketSource *ssource; | ||
208 | |||
209 | +#ifdef WIN32 | ||
210 | + WSAEventSelect(socket, ioc->event, | ||
211 | + FD_READ | FD_ACCEPT | FD_CLOSE | | ||
212 | + FD_CONNECT | FD_WRITE | FD_OOB); | ||
213 | +#endif | ||
214 | + | ||
215 | source = g_source_new(&qio_channel_socket_source_funcs, | ||
216 | sizeof(QIOChannelSocketSource)); | ||
217 | ssource = (QIOChannelSocketSource *)source; | ||
218 | diff --git a/io/channel.c b/io/channel.c | ||
219 | index XXXXXXX..XXXXXXX 100644 | ||
220 | --- a/io/channel.c | ||
221 | +++ b/io/channel.c | ||
222 | @@ -XXX,XX +XXX,XX @@ GSource *qio_channel_create_watch(QIOChannel *ioc, | ||
223 | } | ||
224 | |||
225 | |||
226 | +void qio_channel_set_aio_fd_handler(QIOChannel *ioc, | ||
227 | + AioContext *ctx, | ||
228 | + IOHandler *io_read, | ||
229 | + IOHandler *io_write, | ||
230 | + void *opaque) | ||
231 | +{ | ||
232 | + QIOChannelClass *klass = QIO_CHANNEL_GET_CLASS(ioc); | ||
233 | + | ||
234 | + klass->io_set_aio_fd_handler(ioc, ctx, io_read, io_write, opaque); | ||
235 | +} | ||
236 | + | ||
237 | guint qio_channel_add_watch(QIOChannel *ioc, | ||
238 | GIOCondition condition, | ||
239 | QIOChannelFunc func, | ||
240 | -- | ||
241 | 2.9.3 | ||
242 | |||
243 | diff view generated by jsdifflib |
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | Now, after huge update of block graph permission update algorithm, we | 3 | Support separate coroutines for reading and writing, and place the |
4 | don't need this workaround with active state of the filter. Drop it and | 4 | read/write handlers on the AioContext that the QIOChannel is registered |
5 | use new smart bdrv_drop_filter() function. | 5 | with. |
6 | 6 | ||
7 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 7 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> |
8 | Message-Id: <20210506194143.394141-1-vsementsov@virtuozzo.com> | 8 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
9 | Signed-off-by: Max Reitz <mreitz@redhat.com> | 9 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
10 | Reviewed-by: Fam Zheng <famz@redhat.com> | ||
11 | Message-id: 20170213135235.12274-7-pbonzini@redhat.com | ||
12 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
10 | --- | 13 | --- |
11 | block/copy-on-read.c | 33 +-------------------------------- | 14 | include/io/channel.h | 47 ++++++++++++++++++++++++++-- |
12 | 1 file changed, 1 insertion(+), 32 deletions(-) | 15 | io/channel.c | 86 +++++++++++++++++++++++++++++++++++++++------------- |
13 | 16 | 2 files changed, 109 insertions(+), 24 deletions(-) | |
14 | diff --git a/block/copy-on-read.c b/block/copy-on-read.c | 17 | |
18 | diff --git a/include/io/channel.h b/include/io/channel.h | ||
15 | index XXXXXXX..XXXXXXX 100644 | 19 | index XXXXXXX..XXXXXXX 100644 |
16 | --- a/block/copy-on-read.c | 20 | --- a/include/io/channel.h |
17 | +++ b/block/copy-on-read.c | 21 | +++ b/include/io/channel.h |
18 | @@ -XXX,XX +XXX,XX @@ | 22 | @@ -XXX,XX +XXX,XX @@ |
19 | 23 | ||
20 | 24 | #include "qemu-common.h" | |
21 | typedef struct BDRVStateCOR { | 25 | #include "qom/object.h" |
22 | - bool active; | 26 | +#include "qemu/coroutine.h" |
23 | BlockDriverState *bottom_bs; | 27 | #include "block/aio.h" |
24 | bool chain_frozen; | 28 | |
25 | } BDRVStateCOR; | 29 | #define TYPE_QIO_CHANNEL "qio-channel" |
26 | @@ -XXX,XX +XXX,XX @@ static int cor_open(BlockDriverState *bs, QDict *options, int flags, | 30 | @@ -XXX,XX +XXX,XX @@ struct QIOChannel { |
27 | */ | 31 | Object parent; |
28 | bdrv_ref(bottom_bs); | 32 | unsigned int features; /* bitmask of QIOChannelFeatures */ |
29 | } | 33 | char *name; |
30 | - state->active = true; | 34 | + AioContext *ctx; |
31 | state->bottom_bs = bottom_bs; | 35 | + Coroutine *read_coroutine; |
32 | 36 | + Coroutine *write_coroutine; | |
33 | /* | 37 | #ifdef _WIN32 |
34 | @@ -XXX,XX +XXX,XX @@ static void cor_child_perm(BlockDriverState *bs, BdrvChild *c, | 38 | HANDLE event; /* For use with GSource on Win32 */ |
35 | uint64_t perm, uint64_t shared, | 39 | #endif |
36 | uint64_t *nperm, uint64_t *nshared) | 40 | @@ -XXX,XX +XXX,XX @@ guint qio_channel_add_watch(QIOChannel *ioc, |
41 | |||
42 | |||
43 | /** | ||
44 | + * qio_channel_attach_aio_context: | ||
45 | + * @ioc: the channel object | ||
46 | + * @ctx: the #AioContext to set the handlers on | ||
47 | + * | ||
48 | + * Request that qio_channel_yield() sets I/O handlers on | ||
49 | + * the given #AioContext. If @ctx is %NULL, qio_channel_yield() | ||
50 | + * uses QEMU's main thread event loop. | ||
51 | + * | ||
52 | + * You can move a #QIOChannel from one #AioContext to another even if | ||
53 | + * I/O handlers are set for a coroutine. However, #QIOChannel provides | ||
54 | + * no synchronization between the calls to qio_channel_yield() and | ||
55 | + * qio_channel_attach_aio_context(). | ||
56 | + * | ||
57 | + * Therefore you should first call qio_channel_detach_aio_context() | ||
58 | + * to ensure that the coroutine is not entered concurrently. Then, | ||
59 | + * while the coroutine has yielded, call qio_channel_attach_aio_context(), | ||
60 | + * and then aio_co_schedule() to place the coroutine on the new | ||
61 | + * #AioContext. The calls to qio_channel_detach_aio_context() | ||
62 | + * and qio_channel_attach_aio_context() should be protected with | ||
63 | + * aio_context_acquire() and aio_context_release(). | ||
64 | + */ | ||
65 | +void qio_channel_attach_aio_context(QIOChannel *ioc, | ||
66 | + AioContext *ctx); | ||
67 | + | ||
68 | +/** | ||
69 | + * qio_channel_detach_aio_context: | ||
70 | + * @ioc: the channel object | ||
71 | + * | ||
72 | + * Disable any I/O handlers set by qio_channel_yield(). With the | ||
73 | + * help of aio_co_schedule(), this allows moving a coroutine that was | ||
74 | + * paused by qio_channel_yield() to another context. | ||
75 | + */ | ||
76 | +void qio_channel_detach_aio_context(QIOChannel *ioc); | ||
77 | + | ||
78 | +/** | ||
79 | * qio_channel_yield: | ||
80 | * @ioc: the channel object | ||
81 | * @condition: the I/O condition to wait for | ||
82 | * | ||
83 | - * Yields execution from the current coroutine until | ||
84 | - * the condition indicated by @condition becomes | ||
85 | - * available. | ||
86 | + * Yields execution from the current coroutine until the condition | ||
87 | + * indicated by @condition becomes available. @condition must | ||
88 | + * be either %G_IO_IN or %G_IO_OUT; it cannot contain both. In | ||
89 | + * addition, no two coroutine can be waiting on the same condition | ||
90 | + * and channel at the same time. | ||
91 | * | ||
92 | * This must only be called from coroutine context | ||
93 | */ | ||
94 | diff --git a/io/channel.c b/io/channel.c | ||
95 | index XXXXXXX..XXXXXXX 100644 | ||
96 | --- a/io/channel.c | ||
97 | +++ b/io/channel.c | ||
98 | @@ -XXX,XX +XXX,XX @@ | ||
99 | #include "qemu/osdep.h" | ||
100 | #include "io/channel.h" | ||
101 | #include "qapi/error.h" | ||
102 | -#include "qemu/coroutine.h" | ||
103 | +#include "qemu/main-loop.h" | ||
104 | |||
105 | bool qio_channel_has_feature(QIOChannel *ioc, | ||
106 | QIOChannelFeature feature) | ||
107 | @@ -XXX,XX +XXX,XX @@ off_t qio_channel_io_seek(QIOChannel *ioc, | ||
108 | } | ||
109 | |||
110 | |||
111 | -typedef struct QIOChannelYieldData QIOChannelYieldData; | ||
112 | -struct QIOChannelYieldData { | ||
113 | - QIOChannel *ioc; | ||
114 | - Coroutine *co; | ||
115 | -}; | ||
116 | +static void qio_channel_set_aio_fd_handlers(QIOChannel *ioc); | ||
117 | |||
118 | +static void qio_channel_restart_read(void *opaque) | ||
119 | +{ | ||
120 | + QIOChannel *ioc = opaque; | ||
121 | + Coroutine *co = ioc->read_coroutine; | ||
122 | + | ||
123 | + ioc->read_coroutine = NULL; | ||
124 | + qio_channel_set_aio_fd_handlers(ioc); | ||
125 | + aio_co_wake(co); | ||
126 | +} | ||
127 | |||
128 | -static gboolean qio_channel_yield_enter(QIOChannel *ioc, | ||
129 | - GIOCondition condition, | ||
130 | - gpointer opaque) | ||
131 | +static void qio_channel_restart_write(void *opaque) | ||
37 | { | 132 | { |
38 | - BDRVStateCOR *s = bs->opaque; | 133 | - QIOChannelYieldData *data = opaque; |
134 | - qemu_coroutine_enter(data->co); | ||
135 | - return FALSE; | ||
136 | + QIOChannel *ioc = opaque; | ||
137 | + Coroutine *co = ioc->write_coroutine; | ||
138 | + | ||
139 | + ioc->write_coroutine = NULL; | ||
140 | + qio_channel_set_aio_fd_handlers(ioc); | ||
141 | + aio_co_wake(co); | ||
142 | } | ||
143 | |||
144 | +static void qio_channel_set_aio_fd_handlers(QIOChannel *ioc) | ||
145 | +{ | ||
146 | + IOHandler *rd_handler = NULL, *wr_handler = NULL; | ||
147 | + AioContext *ctx; | ||
148 | + | ||
149 | + if (ioc->read_coroutine) { | ||
150 | + rd_handler = qio_channel_restart_read; | ||
151 | + } | ||
152 | + if (ioc->write_coroutine) { | ||
153 | + wr_handler = qio_channel_restart_write; | ||
154 | + } | ||
155 | + | ||
156 | + ctx = ioc->ctx ? ioc->ctx : iohandler_get_aio_context(); | ||
157 | + qio_channel_set_aio_fd_handler(ioc, ctx, rd_handler, wr_handler, ioc); | ||
158 | +} | ||
159 | + | ||
160 | +void qio_channel_attach_aio_context(QIOChannel *ioc, | ||
161 | + AioContext *ctx) | ||
162 | +{ | ||
163 | + AioContext *old_ctx; | ||
164 | + if (ioc->ctx == ctx) { | ||
165 | + return; | ||
166 | + } | ||
167 | + | ||
168 | + old_ctx = ioc->ctx ? ioc->ctx : iohandler_get_aio_context(); | ||
169 | + qio_channel_set_aio_fd_handler(ioc, old_ctx, NULL, NULL, NULL); | ||
170 | + ioc->ctx = ctx; | ||
171 | + qio_channel_set_aio_fd_handlers(ioc); | ||
172 | +} | ||
173 | + | ||
174 | +void qio_channel_detach_aio_context(QIOChannel *ioc) | ||
175 | +{ | ||
176 | + ioc->read_coroutine = NULL; | ||
177 | + ioc->write_coroutine = NULL; | ||
178 | + qio_channel_set_aio_fd_handlers(ioc); | ||
179 | + ioc->ctx = NULL; | ||
180 | +} | ||
181 | |||
182 | void coroutine_fn qio_channel_yield(QIOChannel *ioc, | ||
183 | GIOCondition condition) | ||
184 | { | ||
185 | - QIOChannelYieldData data; | ||
39 | - | 186 | - |
40 | - if (!s->active) { | 187 | assert(qemu_in_coroutine()); |
41 | - /* | 188 | - data.ioc = ioc; |
42 | - * While the filter is being removed | 189 | - data.co = qemu_coroutine_self(); |
43 | - */ | 190 | - qio_channel_add_watch(ioc, |
44 | - *nperm = 0; | 191 | - condition, |
45 | - *nshared = BLK_PERM_ALL; | 192 | - qio_channel_yield_enter, |
46 | - return; | 193 | - &data, |
47 | - } | 194 | - NULL); |
48 | - | 195 | + if (condition == G_IO_IN) { |
49 | *nperm = perm & PERM_PASSTHROUGH; | 196 | + assert(!ioc->read_coroutine); |
50 | *nshared = (shared & PERM_PASSTHROUGH) | PERM_UNCHANGED; | 197 | + ioc->read_coroutine = qemu_coroutine_self(); |
51 | 198 | + } else if (condition == G_IO_OUT) { | |
52 | @@ -XXX,XX +XXX,XX @@ static BlockDriver bdrv_copy_on_read = { | 199 | + assert(!ioc->write_coroutine); |
53 | 200 | + ioc->write_coroutine = qemu_coroutine_self(); | |
54 | void bdrv_cor_filter_drop(BlockDriverState *cor_filter_bs) | 201 | + } else { |
55 | { | 202 | + abort(); |
56 | - BdrvChild *child; | 203 | + } |
57 | - BlockDriverState *bs; | 204 | + qio_channel_set_aio_fd_handlers(ioc); |
58 | BDRVStateCOR *s = cor_filter_bs->opaque; | 205 | qemu_coroutine_yield(); |
59 | |||
60 | - child = bdrv_filter_child(cor_filter_bs); | ||
61 | - if (!child) { | ||
62 | - return; | ||
63 | - } | ||
64 | - bs = child->bs; | ||
65 | - | ||
66 | - /* Retain the BDS until we complete the graph change. */ | ||
67 | - bdrv_ref(bs); | ||
68 | - /* Hold a guest back from writing while permissions are being reset. */ | ||
69 | - bdrv_drained_begin(bs); | ||
70 | - /* Drop permissions before the graph change. */ | ||
71 | - s->active = false; | ||
72 | /* unfreeze, as otherwise bdrv_replace_node() will fail */ | ||
73 | if (s->chain_frozen) { | ||
74 | s->chain_frozen = false; | ||
75 | bdrv_unfreeze_backing_chain(cor_filter_bs, s->bottom_bs); | ||
76 | } | ||
77 | - bdrv_child_refresh_perms(cor_filter_bs, child, &error_abort); | ||
78 | - bdrv_replace_node(cor_filter_bs, bs, &error_abort); | ||
79 | - | ||
80 | - bdrv_drained_end(bs); | ||
81 | - bdrv_unref(bs); | ||
82 | + bdrv_drop_filter(cor_filter_bs, &error_abort); | ||
83 | bdrv_unref(cor_filter_bs); | ||
84 | } | 206 | } |
85 | 207 | ||
86 | -- | 208 | -- |
87 | 2.31.1 | 209 | 2.9.3 |
88 | 210 | ||
89 | 211 | diff view generated by jsdifflib |
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | Testing set/get of one 64bit variable doesn't seem necessary. We have a | 3 | In the client, read the reply headers from a coroutine, switching the |
4 | lot of such variables. Also remaining tests do test set/get anyway. | 4 | read side between the "read header" coroutine and the I/O coroutine that |
5 | reads the body of the reply. | ||
5 | 6 | ||
6 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 7 | In the server, if the server can read more requests it will create a new |
7 | Reviewed-by: Max Reitz <mreitz@redhat.com> | 8 | "read request" coroutine as soon as a request has been read. Otherwise, |
8 | Message-Id: <20210506090621.11848-7-vsementsov@virtuozzo.com> | 9 | the new coroutine is created in nbd_request_put. |
10 | |||
9 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | 11 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
10 | Signed-off-by: Max Reitz <mreitz@redhat.com> | 12 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
13 | Reviewed-by: Fam Zheng <famz@redhat.com> | ||
14 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> | ||
15 | Message-id: 20170213135235.12274-8-pbonzini@redhat.com | ||
16 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
11 | --- | 17 | --- |
12 | tests/unit/test-write-threshold.c | 43 ------------------------------- | 18 | block/nbd-client.h | 2 +- |
13 | 1 file changed, 43 deletions(-) | 19 | block/nbd-client.c | 117 ++++++++++++++++++++++++----------------------------- |
20 | nbd/client.c | 2 +- | ||
21 | nbd/common.c | 9 +---- | ||
22 | nbd/server.c | 94 +++++++++++++----------------------------- | ||
23 | 5 files changed, 83 insertions(+), 141 deletions(-) | ||
14 | 24 | ||
15 | diff --git a/tests/unit/test-write-threshold.c b/tests/unit/test-write-threshold.c | 25 | diff --git a/block/nbd-client.h b/block/nbd-client.h |
16 | index XXXXXXX..XXXXXXX 100644 | 26 | index XXXXXXX..XXXXXXX 100644 |
17 | --- a/tests/unit/test-write-threshold.c | 27 | --- a/block/nbd-client.h |
18 | +++ b/tests/unit/test-write-threshold.c | 28 | +++ b/block/nbd-client.h |
29 | @@ -XXX,XX +XXX,XX @@ typedef struct NBDClientSession { | ||
30 | |||
31 | CoMutex send_mutex; | ||
32 | CoQueue free_sema; | ||
33 | - Coroutine *send_coroutine; | ||
34 | + Coroutine *read_reply_co; | ||
35 | int in_flight; | ||
36 | |||
37 | Coroutine *recv_coroutine[MAX_NBD_REQUESTS]; | ||
38 | diff --git a/block/nbd-client.c b/block/nbd-client.c | ||
39 | index XXXXXXX..XXXXXXX 100644 | ||
40 | --- a/block/nbd-client.c | ||
41 | +++ b/block/nbd-client.c | ||
19 | @@ -XXX,XX +XXX,XX @@ | 42 | @@ -XXX,XX +XXX,XX @@ |
20 | #include "block/write-threshold.h" | 43 | #define HANDLE_TO_INDEX(bs, handle) ((handle) ^ ((uint64_t)(intptr_t)bs)) |
21 | 44 | #define INDEX_TO_HANDLE(bs, index) ((index) ^ ((uint64_t)(intptr_t)bs)) | |
22 | 45 | ||
23 | -static void test_threshold_not_set_on_init(void) | 46 | -static void nbd_recv_coroutines_enter_all(NBDClientSession *s) |
47 | +static void nbd_recv_coroutines_enter_all(BlockDriverState *bs) | ||
48 | { | ||
49 | + NBDClientSession *s = nbd_get_client_session(bs); | ||
50 | int i; | ||
51 | |||
52 | for (i = 0; i < MAX_NBD_REQUESTS; i++) { | ||
53 | @@ -XXX,XX +XXX,XX @@ static void nbd_recv_coroutines_enter_all(NBDClientSession *s) | ||
54 | qemu_coroutine_enter(s->recv_coroutine[i]); | ||
55 | } | ||
56 | } | ||
57 | + BDRV_POLL_WHILE(bs, s->read_reply_co); | ||
58 | } | ||
59 | |||
60 | static void nbd_teardown_connection(BlockDriverState *bs) | ||
61 | @@ -XXX,XX +XXX,XX @@ static void nbd_teardown_connection(BlockDriverState *bs) | ||
62 | qio_channel_shutdown(client->ioc, | ||
63 | QIO_CHANNEL_SHUTDOWN_BOTH, | ||
64 | NULL); | ||
65 | - nbd_recv_coroutines_enter_all(client); | ||
66 | + nbd_recv_coroutines_enter_all(bs); | ||
67 | |||
68 | nbd_client_detach_aio_context(bs); | ||
69 | object_unref(OBJECT(client->sioc)); | ||
70 | @@ -XXX,XX +XXX,XX @@ static void nbd_teardown_connection(BlockDriverState *bs) | ||
71 | client->ioc = NULL; | ||
72 | } | ||
73 | |||
74 | -static void nbd_reply_ready(void *opaque) | ||
75 | +static coroutine_fn void nbd_read_reply_entry(void *opaque) | ||
76 | { | ||
77 | - BlockDriverState *bs = opaque; | ||
78 | - NBDClientSession *s = nbd_get_client_session(bs); | ||
79 | + NBDClientSession *s = opaque; | ||
80 | uint64_t i; | ||
81 | int ret; | ||
82 | |||
83 | - if (!s->ioc) { /* Already closed */ | ||
84 | - return; | ||
85 | - } | ||
86 | - | ||
87 | - if (s->reply.handle == 0) { | ||
88 | - /* No reply already in flight. Fetch a header. It is possible | ||
89 | - * that another thread has done the same thing in parallel, so | ||
90 | - * the socket is not readable anymore. | ||
91 | - */ | ||
92 | + for (;;) { | ||
93 | + assert(s->reply.handle == 0); | ||
94 | ret = nbd_receive_reply(s->ioc, &s->reply); | ||
95 | - if (ret == -EAGAIN) { | ||
96 | - return; | ||
97 | - } | ||
98 | if (ret < 0) { | ||
99 | - s->reply.handle = 0; | ||
100 | - goto fail; | ||
101 | + break; | ||
102 | } | ||
103 | - } | ||
104 | |||
105 | - /* There's no need for a mutex on the receive side, because the | ||
106 | - * handler acts as a synchronization point and ensures that only | ||
107 | - * one coroutine is called until the reply finishes. */ | ||
108 | - i = HANDLE_TO_INDEX(s, s->reply.handle); | ||
109 | - if (i >= MAX_NBD_REQUESTS) { | ||
110 | - goto fail; | ||
111 | - } | ||
112 | + /* There's no need for a mutex on the receive side, because the | ||
113 | + * handler acts as a synchronization point and ensures that only | ||
114 | + * one coroutine is called until the reply finishes. | ||
115 | + */ | ||
116 | + i = HANDLE_TO_INDEX(s, s->reply.handle); | ||
117 | + if (i >= MAX_NBD_REQUESTS || !s->recv_coroutine[i]) { | ||
118 | + break; | ||
119 | + } | ||
120 | |||
121 | - if (s->recv_coroutine[i]) { | ||
122 | - qemu_coroutine_enter(s->recv_coroutine[i]); | ||
123 | - return; | ||
124 | + /* We're woken up by the recv_coroutine itself. Note that there | ||
125 | + * is no race between yielding and reentering read_reply_co. This | ||
126 | + * is because: | ||
127 | + * | ||
128 | + * - if recv_coroutine[i] runs on the same AioContext, it is only | ||
129 | + * entered after we yield | ||
130 | + * | ||
131 | + * - if recv_coroutine[i] runs on a different AioContext, reentering | ||
132 | + * read_reply_co happens through a bottom half, which can only | ||
133 | + * run after we yield. | ||
134 | + */ | ||
135 | + aio_co_wake(s->recv_coroutine[i]); | ||
136 | + qemu_coroutine_yield(); | ||
137 | } | ||
138 | - | ||
139 | -fail: | ||
140 | - nbd_teardown_connection(bs); | ||
141 | -} | ||
142 | - | ||
143 | -static void nbd_restart_write(void *opaque) | ||
24 | -{ | 144 | -{ |
25 | - uint64_t res; | 145 | - BlockDriverState *bs = opaque; |
26 | - BlockDriverState bs; | 146 | - |
27 | - memset(&bs, 0, sizeof(bs)); | 147 | - qemu_coroutine_enter(nbd_get_client_session(bs)->send_coroutine); |
28 | - | 148 | + s->read_reply_co = NULL; |
29 | - res = bdrv_write_threshold_get(&bs); | 149 | } |
30 | - g_assert_cmpint(res, ==, 0); | 150 | |
151 | static int nbd_co_send_request(BlockDriverState *bs, | ||
152 | @@ -XXX,XX +XXX,XX @@ static int nbd_co_send_request(BlockDriverState *bs, | ||
153 | QEMUIOVector *qiov) | ||
154 | { | ||
155 | NBDClientSession *s = nbd_get_client_session(bs); | ||
156 | - AioContext *aio_context; | ||
157 | int rc, ret, i; | ||
158 | |||
159 | qemu_co_mutex_lock(&s->send_mutex); | ||
160 | @@ -XXX,XX +XXX,XX @@ static int nbd_co_send_request(BlockDriverState *bs, | ||
161 | return -EPIPE; | ||
162 | } | ||
163 | |||
164 | - s->send_coroutine = qemu_coroutine_self(); | ||
165 | - aio_context = bdrv_get_aio_context(bs); | ||
166 | - | ||
167 | - aio_set_fd_handler(aio_context, s->sioc->fd, false, | ||
168 | - nbd_reply_ready, nbd_restart_write, NULL, bs); | ||
169 | if (qiov) { | ||
170 | qio_channel_set_cork(s->ioc, true); | ||
171 | rc = nbd_send_request(s->ioc, request); | ||
172 | @@ -XXX,XX +XXX,XX @@ static int nbd_co_send_request(BlockDriverState *bs, | ||
173 | } else { | ||
174 | rc = nbd_send_request(s->ioc, request); | ||
175 | } | ||
176 | - aio_set_fd_handler(aio_context, s->sioc->fd, false, | ||
177 | - nbd_reply_ready, NULL, NULL, bs); | ||
178 | - s->send_coroutine = NULL; | ||
179 | qemu_co_mutex_unlock(&s->send_mutex); | ||
180 | return rc; | ||
181 | } | ||
182 | @@ -XXX,XX +XXX,XX @@ static void nbd_co_receive_reply(NBDClientSession *s, | ||
183 | { | ||
184 | int ret; | ||
185 | |||
186 | - /* Wait until we're woken up by the read handler. TODO: perhaps | ||
187 | - * peek at the next reply and avoid yielding if it's ours? */ | ||
188 | + /* Wait until we're woken up by nbd_read_reply_entry. */ | ||
189 | qemu_coroutine_yield(); | ||
190 | *reply = s->reply; | ||
191 | if (reply->handle != request->handle || | ||
192 | @@ -XXX,XX +XXX,XX @@ static void nbd_coroutine_start(NBDClientSession *s, | ||
193 | /* s->recv_coroutine[i] is set as soon as we get the send_lock. */ | ||
194 | } | ||
195 | |||
196 | -static void nbd_coroutine_end(NBDClientSession *s, | ||
197 | +static void nbd_coroutine_end(BlockDriverState *bs, | ||
198 | NBDRequest *request) | ||
199 | { | ||
200 | + NBDClientSession *s = nbd_get_client_session(bs); | ||
201 | int i = HANDLE_TO_INDEX(s, request->handle); | ||
202 | + | ||
203 | s->recv_coroutine[i] = NULL; | ||
204 | - if (s->in_flight-- == MAX_NBD_REQUESTS) { | ||
205 | - qemu_co_queue_next(&s->free_sema); | ||
206 | + s->in_flight--; | ||
207 | + qemu_co_queue_next(&s->free_sema); | ||
208 | + | ||
209 | + /* Kick the read_reply_co to get the next reply. */ | ||
210 | + if (s->read_reply_co) { | ||
211 | + aio_co_wake(s->read_reply_co); | ||
212 | } | ||
213 | } | ||
214 | |||
215 | @@ -XXX,XX +XXX,XX @@ int nbd_client_co_preadv(BlockDriverState *bs, uint64_t offset, | ||
216 | } else { | ||
217 | nbd_co_receive_reply(client, &request, &reply, qiov); | ||
218 | } | ||
219 | - nbd_coroutine_end(client, &request); | ||
220 | + nbd_coroutine_end(bs, &request); | ||
221 | return -reply.error; | ||
222 | } | ||
223 | |||
224 | @@ -XXX,XX +XXX,XX @@ int nbd_client_co_pwritev(BlockDriverState *bs, uint64_t offset, | ||
225 | } else { | ||
226 | nbd_co_receive_reply(client, &request, &reply, NULL); | ||
227 | } | ||
228 | - nbd_coroutine_end(client, &request); | ||
229 | + nbd_coroutine_end(bs, &request); | ||
230 | return -reply.error; | ||
231 | } | ||
232 | |||
233 | @@ -XXX,XX +XXX,XX @@ int nbd_client_co_pwrite_zeroes(BlockDriverState *bs, int64_t offset, | ||
234 | } else { | ||
235 | nbd_co_receive_reply(client, &request, &reply, NULL); | ||
236 | } | ||
237 | - nbd_coroutine_end(client, &request); | ||
238 | + nbd_coroutine_end(bs, &request); | ||
239 | return -reply.error; | ||
240 | } | ||
241 | |||
242 | @@ -XXX,XX +XXX,XX @@ int nbd_client_co_flush(BlockDriverState *bs) | ||
243 | } else { | ||
244 | nbd_co_receive_reply(client, &request, &reply, NULL); | ||
245 | } | ||
246 | - nbd_coroutine_end(client, &request); | ||
247 | + nbd_coroutine_end(bs, &request); | ||
248 | return -reply.error; | ||
249 | } | ||
250 | |||
251 | @@ -XXX,XX +XXX,XX @@ int nbd_client_co_pdiscard(BlockDriverState *bs, int64_t offset, int count) | ||
252 | } else { | ||
253 | nbd_co_receive_reply(client, &request, &reply, NULL); | ||
254 | } | ||
255 | - nbd_coroutine_end(client, &request); | ||
256 | + nbd_coroutine_end(bs, &request); | ||
257 | return -reply.error; | ||
258 | |||
259 | } | ||
260 | |||
261 | void nbd_client_detach_aio_context(BlockDriverState *bs) | ||
262 | { | ||
263 | - aio_set_fd_handler(bdrv_get_aio_context(bs), | ||
264 | - nbd_get_client_session(bs)->sioc->fd, | ||
265 | - false, NULL, NULL, NULL, NULL); | ||
266 | + NBDClientSession *client = nbd_get_client_session(bs); | ||
267 | + qio_channel_detach_aio_context(QIO_CHANNEL(client->sioc)); | ||
268 | } | ||
269 | |||
270 | void nbd_client_attach_aio_context(BlockDriverState *bs, | ||
271 | AioContext *new_context) | ||
272 | { | ||
273 | - aio_set_fd_handler(new_context, nbd_get_client_session(bs)->sioc->fd, | ||
274 | - false, nbd_reply_ready, NULL, NULL, bs); | ||
275 | + NBDClientSession *client = nbd_get_client_session(bs); | ||
276 | + qio_channel_attach_aio_context(QIO_CHANNEL(client->sioc), new_context); | ||
277 | + aio_co_schedule(new_context, client->read_reply_co); | ||
278 | } | ||
279 | |||
280 | void nbd_client_close(BlockDriverState *bs) | ||
281 | @@ -XXX,XX +XXX,XX @@ int nbd_client_init(BlockDriverState *bs, | ||
282 | /* Now that we're connected, set the socket to be non-blocking and | ||
283 | * kick the reply mechanism. */ | ||
284 | qio_channel_set_blocking(QIO_CHANNEL(sioc), false, NULL); | ||
285 | - | ||
286 | + client->read_reply_co = qemu_coroutine_create(nbd_read_reply_entry, client); | ||
287 | nbd_client_attach_aio_context(bs, bdrv_get_aio_context(bs)); | ||
288 | |||
289 | logout("Established connection with NBD server\n"); | ||
290 | diff --git a/nbd/client.c b/nbd/client.c | ||
291 | index XXXXXXX..XXXXXXX 100644 | ||
292 | --- a/nbd/client.c | ||
293 | +++ b/nbd/client.c | ||
294 | @@ -XXX,XX +XXX,XX @@ ssize_t nbd_receive_reply(QIOChannel *ioc, NBDReply *reply) | ||
295 | ssize_t ret; | ||
296 | |||
297 | ret = read_sync(ioc, buf, sizeof(buf)); | ||
298 | - if (ret < 0) { | ||
299 | + if (ret <= 0) { | ||
300 | return ret; | ||
301 | } | ||
302 | |||
303 | diff --git a/nbd/common.c b/nbd/common.c | ||
304 | index XXXXXXX..XXXXXXX 100644 | ||
305 | --- a/nbd/common.c | ||
306 | +++ b/nbd/common.c | ||
307 | @@ -XXX,XX +XXX,XX @@ ssize_t nbd_wr_syncv(QIOChannel *ioc, | ||
308 | } | ||
309 | if (len == QIO_CHANNEL_ERR_BLOCK) { | ||
310 | if (qemu_in_coroutine()) { | ||
311 | - /* XXX figure out if we can create a variant on | ||
312 | - * qio_channel_yield() that works with AIO contexts | ||
313 | - * and consider using that in this branch */ | ||
314 | - qemu_coroutine_yield(); | ||
315 | - } else if (done) { | ||
316 | - /* XXX this is needed by nbd_reply_ready. */ | ||
317 | - qio_channel_wait(ioc, | ||
318 | - do_read ? G_IO_IN : G_IO_OUT); | ||
319 | + qio_channel_yield(ioc, do_read ? G_IO_IN : G_IO_OUT); | ||
320 | } else { | ||
321 | return -EAGAIN; | ||
322 | } | ||
323 | diff --git a/nbd/server.c b/nbd/server.c | ||
324 | index XXXXXXX..XXXXXXX 100644 | ||
325 | --- a/nbd/server.c | ||
326 | +++ b/nbd/server.c | ||
327 | @@ -XXX,XX +XXX,XX @@ struct NBDClient { | ||
328 | CoMutex send_lock; | ||
329 | Coroutine *send_coroutine; | ||
330 | |||
331 | - bool can_read; | ||
332 | - | ||
333 | QTAILQ_ENTRY(NBDClient) next; | ||
334 | int nb_requests; | ||
335 | bool closing; | ||
336 | @@ -XXX,XX +XXX,XX @@ struct NBDClient { | ||
337 | |||
338 | /* That's all folks */ | ||
339 | |||
340 | -static void nbd_set_handlers(NBDClient *client); | ||
341 | -static void nbd_unset_handlers(NBDClient *client); | ||
342 | -static void nbd_update_can_read(NBDClient *client); | ||
343 | +static void nbd_client_receive_next_request(NBDClient *client); | ||
344 | |||
345 | static gboolean nbd_negotiate_continue(QIOChannel *ioc, | ||
346 | GIOCondition condition, | ||
347 | @@ -XXX,XX +XXX,XX @@ void nbd_client_put(NBDClient *client) | ||
348 | */ | ||
349 | assert(client->closing); | ||
350 | |||
351 | - nbd_unset_handlers(client); | ||
352 | + qio_channel_detach_aio_context(client->ioc); | ||
353 | object_unref(OBJECT(client->sioc)); | ||
354 | object_unref(OBJECT(client->ioc)); | ||
355 | if (client->tlscreds) { | ||
356 | @@ -XXX,XX +XXX,XX @@ static NBDRequestData *nbd_request_get(NBDClient *client) | ||
357 | |||
358 | assert(client->nb_requests <= MAX_NBD_REQUESTS - 1); | ||
359 | client->nb_requests++; | ||
360 | - nbd_update_can_read(client); | ||
361 | |||
362 | req = g_new0(NBDRequestData, 1); | ||
363 | nbd_client_get(client); | ||
364 | @@ -XXX,XX +XXX,XX @@ static void nbd_request_put(NBDRequestData *req) | ||
365 | g_free(req); | ||
366 | |||
367 | client->nb_requests--; | ||
368 | - nbd_update_can_read(client); | ||
369 | + nbd_client_receive_next_request(client); | ||
370 | + | ||
371 | nbd_client_put(client); | ||
372 | } | ||
373 | |||
374 | @@ -XXX,XX +XXX,XX @@ static void blk_aio_attached(AioContext *ctx, void *opaque) | ||
375 | exp->ctx = ctx; | ||
376 | |||
377 | QTAILQ_FOREACH(client, &exp->clients, next) { | ||
378 | - nbd_set_handlers(client); | ||
379 | + qio_channel_attach_aio_context(client->ioc, ctx); | ||
380 | + if (client->recv_coroutine) { | ||
381 | + aio_co_schedule(ctx, client->recv_coroutine); | ||
382 | + } | ||
383 | + if (client->send_coroutine) { | ||
384 | + aio_co_schedule(ctx, client->send_coroutine); | ||
385 | + } | ||
386 | } | ||
387 | } | ||
388 | |||
389 | @@ -XXX,XX +XXX,XX @@ static void blk_aio_detach(void *opaque) | ||
390 | TRACE("Export %s: Detaching clients from AIO context %p\n", exp->name, exp->ctx); | ||
391 | |||
392 | QTAILQ_FOREACH(client, &exp->clients, next) { | ||
393 | - nbd_unset_handlers(client); | ||
394 | + qio_channel_detach_aio_context(client->ioc); | ||
395 | } | ||
396 | |||
397 | exp->ctx = NULL; | ||
398 | @@ -XXX,XX +XXX,XX @@ static ssize_t nbd_co_send_reply(NBDRequestData *req, NBDReply *reply, | ||
399 | g_assert(qemu_in_coroutine()); | ||
400 | qemu_co_mutex_lock(&client->send_lock); | ||
401 | client->send_coroutine = qemu_coroutine_self(); | ||
402 | - nbd_set_handlers(client); | ||
403 | |||
404 | if (!len) { | ||
405 | rc = nbd_send_reply(client->ioc, reply); | ||
406 | @@ -XXX,XX +XXX,XX @@ static ssize_t nbd_co_send_reply(NBDRequestData *req, NBDReply *reply, | ||
407 | } | ||
408 | |||
409 | client->send_coroutine = NULL; | ||
410 | - nbd_set_handlers(client); | ||
411 | qemu_co_mutex_unlock(&client->send_lock); | ||
412 | return rc; | ||
413 | } | ||
414 | @@ -XXX,XX +XXX,XX @@ static ssize_t nbd_co_receive_request(NBDRequestData *req, | ||
415 | ssize_t rc; | ||
416 | |||
417 | g_assert(qemu_in_coroutine()); | ||
418 | - client->recv_coroutine = qemu_coroutine_self(); | ||
419 | - nbd_update_can_read(client); | ||
420 | - | ||
421 | + assert(client->recv_coroutine == qemu_coroutine_self()); | ||
422 | rc = nbd_receive_request(client->ioc, request); | ||
423 | if (rc < 0) { | ||
424 | if (rc != -EAGAIN) { | ||
425 | @@ -XXX,XX +XXX,XX @@ static ssize_t nbd_co_receive_request(NBDRequestData *req, | ||
426 | |||
427 | out: | ||
428 | client->recv_coroutine = NULL; | ||
429 | - nbd_update_can_read(client); | ||
430 | + nbd_client_receive_next_request(client); | ||
431 | |||
432 | return rc; | ||
433 | } | ||
434 | |||
435 | -static void nbd_trip(void *opaque) | ||
436 | +/* Owns a reference to the NBDClient passed as opaque. */ | ||
437 | +static coroutine_fn void nbd_trip(void *opaque) | ||
438 | { | ||
439 | NBDClient *client = opaque; | ||
440 | NBDExport *exp = client->exp; | ||
441 | NBDRequestData *req; | ||
442 | - NBDRequest request; | ||
443 | + NBDRequest request = { 0 }; /* GCC thinks it can be used uninitialized */ | ||
444 | NBDReply reply; | ||
445 | ssize_t ret; | ||
446 | int flags; | ||
447 | |||
448 | TRACE("Reading request."); | ||
449 | if (client->closing) { | ||
450 | + nbd_client_put(client); | ||
451 | return; | ||
452 | } | ||
453 | |||
454 | @@ -XXX,XX +XXX,XX @@ static void nbd_trip(void *opaque) | ||
455 | |||
456 | done: | ||
457 | nbd_request_put(req); | ||
458 | + nbd_client_put(client); | ||
459 | return; | ||
460 | |||
461 | out: | ||
462 | nbd_request_put(req); | ||
463 | client_close(client); | ||
464 | + nbd_client_put(client); | ||
465 | } | ||
466 | |||
467 | -static void nbd_read(void *opaque) | ||
468 | +static void nbd_client_receive_next_request(NBDClient *client) | ||
469 | { | ||
470 | - NBDClient *client = opaque; | ||
471 | - | ||
472 | - if (client->recv_coroutine) { | ||
473 | - qemu_coroutine_enter(client->recv_coroutine); | ||
474 | - } else { | ||
475 | - qemu_coroutine_enter(qemu_coroutine_create(nbd_trip, client)); | ||
476 | - } | ||
31 | -} | 477 | -} |
32 | - | 478 | - |
33 | -static void test_threshold_set_get(void) | 479 | -static void nbd_restart_write(void *opaque) |
34 | -{ | 480 | -{ |
35 | - uint64_t threshold = 4 * 1024 * 1024; | 481 | - NBDClient *client = opaque; |
36 | - uint64_t res; | 482 | - |
37 | - BlockDriverState bs; | 483 | - qemu_coroutine_enter(client->send_coroutine); |
38 | - memset(&bs, 0, sizeof(bs)); | ||
39 | - | ||
40 | - bdrv_write_threshold_set(&bs, threshold); | ||
41 | - | ||
42 | - res = bdrv_write_threshold_get(&bs); | ||
43 | - g_assert_cmpint(res, ==, threshold); | ||
44 | -} | 484 | -} |
45 | - | 485 | - |
46 | -static void test_threshold_multi_set_get(void) | 486 | -static void nbd_set_handlers(NBDClient *client) |
47 | -{ | 487 | -{ |
48 | - uint64_t threshold1 = 4 * 1024 * 1024; | 488 | - if (client->exp && client->exp->ctx) { |
49 | - uint64_t threshold2 = 15 * 1024 * 1024; | 489 | - aio_set_fd_handler(client->exp->ctx, client->sioc->fd, true, |
50 | - uint64_t res; | 490 | - client->can_read ? nbd_read : NULL, |
51 | - BlockDriverState bs; | 491 | - client->send_coroutine ? nbd_restart_write : NULL, |
52 | - memset(&bs, 0, sizeof(bs)); | 492 | - NULL, client); |
53 | - | 493 | - } |
54 | - bdrv_write_threshold_set(&bs, threshold1); | ||
55 | - bdrv_write_threshold_set(&bs, threshold2); | ||
56 | - res = bdrv_write_threshold_get(&bs); | ||
57 | - g_assert_cmpint(res, ==, threshold2); | ||
58 | -} | 494 | -} |
59 | - | 495 | - |
60 | static void test_threshold_not_trigger(void) | 496 | -static void nbd_unset_handlers(NBDClient *client) |
61 | { | 497 | -{ |
62 | uint64_t threshold = 4 * 1024 * 1024; | 498 | - if (client->exp && client->exp->ctx) { |
63 | @@ -XXX,XX +XXX,XX @@ int main(int argc, char **argv) | 499 | - aio_set_fd_handler(client->exp->ctx, client->sioc->fd, true, NULL, |
64 | { | 500 | - NULL, NULL, NULL); |
65 | size_t i; | 501 | - } |
66 | TestStruct tests[] = { | 502 | -} |
67 | - { "/write-threshold/not-set-on-init", | 503 | - |
68 | - test_threshold_not_set_on_init }, | 504 | -static void nbd_update_can_read(NBDClient *client) |
69 | - { "/write-threshold/set-get", | 505 | -{ |
70 | - test_threshold_set_get }, | 506 | - bool can_read = client->recv_coroutine || |
71 | - { "/write-threshold/multi-set-get", | 507 | - client->nb_requests < MAX_NBD_REQUESTS; |
72 | - test_threshold_multi_set_get }, | 508 | - |
73 | { "/write-threshold/not-trigger", | 509 | - if (can_read != client->can_read) { |
74 | test_threshold_not_trigger }, | 510 | - client->can_read = can_read; |
75 | { "/write-threshold/trigger", | 511 | - nbd_set_handlers(client); |
512 | - | ||
513 | - /* There is no need to invoke aio_notify(), since aio_set_fd_handler() | ||
514 | - * in nbd_set_handlers() will have taken care of that */ | ||
515 | + if (!client->recv_coroutine && client->nb_requests < MAX_NBD_REQUESTS) { | ||
516 | + nbd_client_get(client); | ||
517 | + client->recv_coroutine = qemu_coroutine_create(nbd_trip, client); | ||
518 | + aio_co_schedule(client->exp->ctx, client->recv_coroutine); | ||
519 | } | ||
520 | } | ||
521 | |||
522 | @@ -XXX,XX +XXX,XX @@ static coroutine_fn void nbd_co_client_start(void *opaque) | ||
523 | goto out; | ||
524 | } | ||
525 | qemu_co_mutex_init(&client->send_lock); | ||
526 | - nbd_set_handlers(client); | ||
527 | |||
528 | if (exp) { | ||
529 | QTAILQ_INSERT_TAIL(&exp->clients, client, next); | ||
530 | } | ||
531 | + | ||
532 | + nbd_client_receive_next_request(client); | ||
533 | + | ||
534 | out: | ||
535 | g_free(data); | ||
536 | } | ||
537 | @@ -XXX,XX +XXX,XX @@ void nbd_client_new(NBDExport *exp, | ||
538 | object_ref(OBJECT(client->sioc)); | ||
539 | client->ioc = QIO_CHANNEL(sioc); | ||
540 | object_ref(OBJECT(client->ioc)); | ||
541 | - client->can_read = true; | ||
542 | client->close = close_fn; | ||
543 | |||
544 | data->client = client; | ||
76 | -- | 545 | -- |
77 | 2.31.1 | 546 | 2.9.3 |
78 | 547 | ||
79 | 548 | diff view generated by jsdifflib |
1 | From: Connor Kuehl <ckuehl@redhat.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | The contents of this patch were initially developed and posted by Han | 3 | As a small step towards the introduction of multiqueue, we want |
4 | Han[1], however, it appears the original patch was not applied. Since | 4 | coroutines to remain on the same AioContext that started them, |
5 | then, the relevant documentation has been moved and adapted to a new | 5 | unless they are moved explicitly with e.g. aio_co_schedule. This patch |
6 | format. | 6 | avoids that coroutines switch AioContext when they use a CoMutex. |
7 | For now it does not make much of a difference, because the CoMutex | ||
8 | is not thread-safe and the AioContext itself is used to protect the | ||
9 | CoMutex from concurrent access. However, this is going to change. | ||
7 | 10 | ||
8 | I've taken most of the original wording and tweaked it according to | 11 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
9 | some of the feedback from the original patch submission. I've also | 12 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
10 | adapted it to restructured text, which is the format the documentation | 13 | Reviewed-by: Fam Zheng <famz@redhat.com> |
11 | currently uses. | 14 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> |
15 | Message-id: 20170213135235.12274-9-pbonzini@redhat.com | ||
16 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
17 | --- | ||
18 | util/qemu-coroutine-lock.c | 5 ++--- | ||
19 | util/trace-events | 1 - | ||
20 | 2 files changed, 2 insertions(+), 4 deletions(-) | ||
12 | 21 | ||
13 | [1] https://lists.nongnu.org/archive/html/qemu-block/2019-10/msg01253.html | 22 | diff --git a/util/qemu-coroutine-lock.c b/util/qemu-coroutine-lock.c |
14 | |||
15 | Fixes: https://bugzilla.redhat.com/1763105 | ||
16 | Signed-off-by: Han Han <hhan@redhat.com> | ||
17 | Suggested-by: Max Reitz <mreitz@redhat.com> | ||
18 | [ Max: provided description of data_file_raw behavior ] | ||
19 | Signed-off-by: Connor Kuehl <ckuehl@redhat.com> | ||
20 | Message-Id: <20210505195512.391128-1-ckuehl@redhat.com> | ||
21 | Signed-off-by: Max Reitz <mreitz@redhat.com> | ||
22 | --- | ||
23 | docs/tools/qemu-img.rst | 31 +++++++++++++++++++++++++++++++ | ||
24 | 1 file changed, 31 insertions(+) | ||
25 | |||
26 | diff --git a/docs/tools/qemu-img.rst b/docs/tools/qemu-img.rst | ||
27 | index XXXXXXX..XXXXXXX 100644 | 23 | index XXXXXXX..XXXXXXX 100644 |
28 | --- a/docs/tools/qemu-img.rst | 24 | --- a/util/qemu-coroutine-lock.c |
29 | +++ b/docs/tools/qemu-img.rst | 25 | +++ b/util/qemu-coroutine-lock.c |
30 | @@ -XXX,XX +XXX,XX @@ Supported image file formats: | 26 | @@ -XXX,XX +XXX,XX @@ |
31 | issue ``lsattr filename`` to check if the NOCOW flag is set or not | 27 | #include "qemu/coroutine.h" |
32 | (Capital 'C' is NOCOW flag). | 28 | #include "qemu/coroutine_int.h" |
33 | 29 | #include "qemu/queue.h" | |
34 | + ``data_file`` | 30 | +#include "block/aio.h" |
35 | + Filename where all guest data will be stored. If this option is used, | 31 | #include "trace.h" |
36 | + the qcow2 file will only contain the image's metadata. | 32 | |
37 | + | 33 | void qemu_co_queue_init(CoQueue *queue) |
38 | + Note: Data loss will occur if the given filename already exists when | 34 | @@ -XXX,XX +XXX,XX @@ void qemu_co_queue_run_restart(Coroutine *co) |
39 | + using this option with ``qemu-img create`` since ``qemu-img`` will create | 35 | |
40 | + the data file anew, overwriting the file's original contents. To simply | 36 | static bool qemu_co_queue_do_restart(CoQueue *queue, bool single) |
41 | + update the reference to point to the given pre-existing file, use | 37 | { |
42 | + ``qemu-img amend``. | 38 | - Coroutine *self = qemu_coroutine_self(); |
43 | + | 39 | Coroutine *next; |
44 | + ``data_file_raw`` | 40 | |
45 | + If this option is set to ``on``, QEMU will always keep the external data | 41 | if (QSIMPLEQ_EMPTY(&queue->entries)) { |
46 | + file consistent as a standalone read-only raw image. | 42 | @@ -XXX,XX +XXX,XX @@ static bool qemu_co_queue_do_restart(CoQueue *queue, bool single) |
47 | + | 43 | |
48 | + It does this by forwarding all write accesses to the qcow2 file through to | 44 | while ((next = QSIMPLEQ_FIRST(&queue->entries)) != NULL) { |
49 | + the raw data file, including their offsets. Therefore, data that is visible | 45 | QSIMPLEQ_REMOVE_HEAD(&queue->entries, co_queue_next); |
50 | + on the qcow2 node (i.e., to the guest) at some offset is visible at the same | 46 | - QSIMPLEQ_INSERT_TAIL(&self->co_queue_wakeup, next, co_queue_next); |
51 | + offset in the raw data file. This results in a read-only raw image. Writes | 47 | - trace_qemu_co_queue_next(next); |
52 | + that bypass the qcow2 metadata may corrupt the qcow2 metadata because the | 48 | + aio_co_wake(next); |
53 | + out-of-band writes may result in the metadata falling out of sync with the | 49 | if (single) { |
54 | + raw image. | 50 | break; |
55 | + | 51 | } |
56 | + If this option is ``off``, QEMU will use the data file to store data in an | 52 | diff --git a/util/trace-events b/util/trace-events |
57 | + arbitrary manner. The file’s content will not make sense without the | 53 | index XXXXXXX..XXXXXXX 100644 |
58 | + accompanying qcow2 metadata. Where data is written will have no relation to | 54 | --- a/util/trace-events |
59 | + its offset as seen by the guest, and some writes (specifically zero writes) | 55 | +++ b/util/trace-events |
60 | + may not be forwarded to the data file at all, but will only be handled by | 56 | @@ -XXX,XX +XXX,XX @@ qemu_coroutine_terminate(void *co) "self %p" |
61 | + modifying qcow2 metadata. | 57 | |
62 | + | 58 | # util/qemu-coroutine-lock.c |
63 | + This option can only be enabled if ``data_file`` is set. | 59 | qemu_co_queue_run_restart(void *co) "co %p" |
64 | + | 60 | -qemu_co_queue_next(void *nxt) "next %p" |
65 | ``Other`` | 61 | qemu_co_mutex_lock_entry(void *mutex, void *self) "mutex %p self %p" |
66 | 62 | qemu_co_mutex_lock_return(void *mutex, void *self) "mutex %p self %p" | |
67 | QEMU also supports various other image file formats for | 63 | qemu_co_mutex_unlock_entry(void *mutex, void *self) "mutex %p self %p" |
68 | -- | 64 | -- |
69 | 2.31.1 | 65 | 2.9.3 |
70 | 66 | ||
71 | 67 | diff view generated by jsdifflib |
1 | From: Paolo Bonzini <pbonzini@redhat.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | Right now there is no easy way for "check" to print a reproducer command. | 3 | Keep the coroutine on the same AioContext. Without this change, |
4 | Because such a reproducer command line would be huge, we can instead teach | 4 | there would be a race between yielding the coroutine and reentering it. |
5 | check to start a command of our choice. This can be for example a Python | 5 | While the race cannot happen now, because the code only runs from a single |
6 | unit test with arguments to only run a specific subtest. | 6 | AioContext, this will change with multiqueue support in the block layer. |
7 | 7 | ||
8 | Move the trailing empty line to print_env(), since it always looks better | 8 | While doing the change, replace custom bottom half with aio_co_schedule. |
9 | and one caller was not adding it. | ||
10 | 9 | ||
11 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> | 10 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
12 | Reviewed-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 11 | Reviewed-by: Fam Zheng <famz@redhat.com> |
13 | Tested-by: Emanuele Giuseppe Esposito <eesposit@redhat.com> | 12 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
14 | Message-Id: <20210323181928.311862-5-pbonzini@redhat.com> | 13 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> |
15 | Message-Id: <20210503110110.476887-5-pbonzini@redhat.com> | 14 | Message-id: 20170213135235.12274-10-pbonzini@redhat.com |
16 | Signed-off-by: Max Reitz <mreitz@redhat.com> | 15 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
17 | --- | 16 | --- |
18 | tests/qemu-iotests/check | 19 ++++++++++++++++++- | 17 | block/blkdebug.c | 9 +-------- |
19 | tests/qemu-iotests/testenv.py | 3 ++- | 18 | 1 file changed, 1 insertion(+), 8 deletions(-) |
20 | tests/qemu-iotests/testrunner.py | 1 - | ||
21 | 3 files changed, 20 insertions(+), 3 deletions(-) | ||
22 | 19 | ||
23 | diff --git a/tests/qemu-iotests/check b/tests/qemu-iotests/check | 20 | diff --git a/block/blkdebug.c b/block/blkdebug.c |
24 | index XXXXXXX..XXXXXXX 100755 | ||
25 | --- a/tests/qemu-iotests/check | ||
26 | +++ b/tests/qemu-iotests/check | ||
27 | @@ -XXX,XX +XXX,XX @@ | ||
28 | import os | ||
29 | import sys | ||
30 | import argparse | ||
31 | +import shutil | ||
32 | +from pathlib import Path | ||
33 | + | ||
34 | from findtests import TestFinder | ||
35 | from testenv import TestEnv | ||
36 | from testrunner import TestRunner | ||
37 | @@ -XXX,XX +XXX,XX @@ def make_argparser() -> argparse.ArgumentParser: | ||
38 | 'rerun failed ./check command, starting from the ' | ||
39 | 'middle of the process.') | ||
40 | g_sel.add_argument('tests', metavar='TEST_FILES', nargs='*', | ||
41 | - help='tests to run') | ||
42 | + help='tests to run, or "--" followed by a command') | ||
43 | |||
44 | return p | ||
45 | |||
46 | @@ -XXX,XX +XXX,XX @@ if __name__ == '__main__': | ||
47 | imgopts=args.imgopts, misalign=args.misalign, | ||
48 | debug=args.debug, valgrind=args.valgrind) | ||
49 | |||
50 | + if len(sys.argv) > 1 and sys.argv[-len(args.tests)-1] == '--': | ||
51 | + if not args.tests: | ||
52 | + sys.exit("missing command after '--'") | ||
53 | + cmd = args.tests | ||
54 | + env.print_env() | ||
55 | + exec_pathstr = shutil.which(cmd[0]) | ||
56 | + if exec_pathstr is None: | ||
57 | + sys.exit('command not found: ' + cmd[0]) | ||
58 | + exec_path = Path(exec_pathstr).resolve() | ||
59 | + cmd[0] = str(exec_path) | ||
60 | + full_env = env.prepare_subprocess(cmd) | ||
61 | + os.chdir(exec_path.parent) | ||
62 | + os.execve(cmd[0], cmd, full_env) | ||
63 | + | ||
64 | testfinder = TestFinder(test_dir=env.source_iotests) | ||
65 | |||
66 | groups = args.groups.split(',') if args.groups else None | ||
67 | diff --git a/tests/qemu-iotests/testenv.py b/tests/qemu-iotests/testenv.py | ||
68 | index XXXXXXX..XXXXXXX 100644 | 21 | index XXXXXXX..XXXXXXX 100644 |
69 | --- a/tests/qemu-iotests/testenv.py | 22 | --- a/block/blkdebug.c |
70 | +++ b/tests/qemu-iotests/testenv.py | 23 | +++ b/block/blkdebug.c |
71 | @@ -XXX,XX +XXX,XX @@ def print_env(self) -> None: | 24 | @@ -XXX,XX +XXX,XX @@ out: |
72 | PLATFORM -- {platform} | 25 | return ret; |
73 | TEST_DIR -- {TEST_DIR} | 26 | } |
74 | SOCK_DIR -- {SOCK_DIR} | 27 | |
75 | -SOCKET_SCM_HELPER -- {SOCKET_SCM_HELPER}""" | 28 | -static void error_callback_bh(void *opaque) |
76 | +SOCKET_SCM_HELPER -- {SOCKET_SCM_HELPER} | 29 | -{ |
77 | +""" | 30 | - Coroutine *co = opaque; |
78 | 31 | - qemu_coroutine_enter(co); | |
79 | args = collections.defaultdict(str, self.get_env()) | 32 | -} |
80 | 33 | - | |
81 | diff --git a/tests/qemu-iotests/testrunner.py b/tests/qemu-iotests/testrunner.py | 34 | static int inject_error(BlockDriverState *bs, BlkdebugRule *rule) |
82 | index XXXXXXX..XXXXXXX 100644 | 35 | { |
83 | --- a/tests/qemu-iotests/testrunner.py | 36 | BDRVBlkdebugState *s = bs->opaque; |
84 | +++ b/tests/qemu-iotests/testrunner.py | 37 | @@ -XXX,XX +XXX,XX @@ static int inject_error(BlockDriverState *bs, BlkdebugRule *rule) |
85 | @@ -XXX,XX +XXX,XX @@ def run_tests(self, tests: List[str]) -> bool: | 38 | } |
86 | 39 | ||
87 | if not self.makecheck: | 40 | if (!immediately) { |
88 | self.env.print_env() | 41 | - aio_bh_schedule_oneshot(bdrv_get_aio_context(bs), error_callback_bh, |
89 | - print() | 42 | - qemu_coroutine_self()); |
90 | 43 | + aio_co_schedule(qemu_get_current_aio_context(), qemu_coroutine_self()); | |
91 | test_field_width = max(len(os.path.basename(t)) for t in tests) + 2 | 44 | qemu_coroutine_yield(); |
45 | } | ||
92 | 46 | ||
93 | -- | 47 | -- |
94 | 2.31.1 | 48 | 2.9.3 |
95 | 49 | ||
96 | 50 | diff view generated by jsdifflib |
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | These tests use bdrv_write_threshold_exceeded() API, which is used only | 3 | qed_aio_start_io and qed_aio_next_io will not have to acquire/release |
4 | for test (since pre-previous commit). Better is testing real API, which | 4 | the AioContext, while qed_aio_next_io_cb will. Split the functionality |
5 | is used in block.c as well. | 5 | and gain a little type-safety in the process. |
6 | 6 | ||
7 | So, let's call bdrv_write_threshold_check_write(), and check is | 7 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
8 | bs->write_threshold_offset cleared or not (it's cleared iff threshold | 8 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
9 | triggered). | 9 | Reviewed-by: Fam Zheng <famz@redhat.com> |
10 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> | ||
11 | Message-id: 20170213135235.12274-11-pbonzini@redhat.com | ||
12 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
13 | --- | ||
14 | block/qed.c | 39 +++++++++++++++++++++++++-------------- | ||
15 | 1 file changed, 25 insertions(+), 14 deletions(-) | ||
10 | 16 | ||
11 | Also we get rid of BdrvTrackedRequest use here. Note, that paranoiac | 17 | diff --git a/block/qed.c b/block/qed.c |
12 | bdrv_check_request() calls were added in 8b1170012b1 to protect | ||
13 | BdrvTrackedRequest. Drop them now. | ||
14 | |||
15 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
16 | Reviewed-by: Max Reitz <mreitz@redhat.com> | ||
17 | Message-Id: <20210506090621.11848-4-vsementsov@virtuozzo.com> | ||
18 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
19 | Signed-off-by: Max Reitz <mreitz@redhat.com> | ||
20 | --- | ||
21 | tests/unit/test-write-threshold.c | 22 ++++------------------ | ||
22 | 1 file changed, 4 insertions(+), 18 deletions(-) | ||
23 | |||
24 | diff --git a/tests/unit/test-write-threshold.c b/tests/unit/test-write-threshold.c | ||
25 | index XXXXXXX..XXXXXXX 100644 | 18 | index XXXXXXX..XXXXXXX 100644 |
26 | --- a/tests/unit/test-write-threshold.c | 19 | --- a/block/qed.c |
27 | +++ b/tests/unit/test-write-threshold.c | 20 | +++ b/block/qed.c |
28 | @@ -XXX,XX +XXX,XX @@ static void test_threshold_multi_set_get(void) | 21 | @@ -XXX,XX +XXX,XX @@ static CachedL2Table *qed_new_l2_table(BDRVQEDState *s) |
29 | 22 | return l2_table; | |
30 | static void test_threshold_not_trigger(void) | 23 | } |
24 | |||
25 | -static void qed_aio_next_io(void *opaque, int ret); | ||
26 | +static void qed_aio_next_io(QEDAIOCB *acb, int ret); | ||
27 | + | ||
28 | +static void qed_aio_start_io(QEDAIOCB *acb) | ||
29 | +{ | ||
30 | + qed_aio_next_io(acb, 0); | ||
31 | +} | ||
32 | + | ||
33 | +static void qed_aio_next_io_cb(void *opaque, int ret) | ||
34 | +{ | ||
35 | + QEDAIOCB *acb = opaque; | ||
36 | + | ||
37 | + qed_aio_next_io(acb, ret); | ||
38 | +} | ||
39 | |||
40 | static void qed_plug_allocating_write_reqs(BDRVQEDState *s) | ||
31 | { | 41 | { |
32 | - uint64_t amount = 0; | 42 | @@ -XXX,XX +XXX,XX @@ static void qed_unplug_allocating_write_reqs(BDRVQEDState *s) |
33 | uint64_t threshold = 4 * 1024 * 1024; | 43 | |
34 | BlockDriverState bs; | 44 | acb = QSIMPLEQ_FIRST(&s->allocating_write_reqs); |
35 | - BdrvTrackedRequest req; | 45 | if (acb) { |
36 | 46 | - qed_aio_next_io(acb, 0); | |
37 | memset(&bs, 0, sizeof(bs)); | 47 | + qed_aio_start_io(acb); |
38 | - memset(&req, 0, sizeof(req)); | 48 | } |
39 | - req.offset = 1024; | ||
40 | - req.bytes = 1024; | ||
41 | - | ||
42 | - bdrv_check_request(req.offset, req.bytes, &error_abort); | ||
43 | |||
44 | bdrv_write_threshold_set(&bs, threshold); | ||
45 | - amount = bdrv_write_threshold_exceeded(&bs, &req); | ||
46 | - g_assert_cmpuint(amount, ==, 0); | ||
47 | + bdrv_write_threshold_check_write(&bs, 1024, 1024); | ||
48 | + g_assert_cmpuint(bdrv_write_threshold_get(&bs), ==, threshold); | ||
49 | } | 49 | } |
50 | 50 | ||
51 | 51 | @@ -XXX,XX +XXX,XX @@ static void qed_aio_complete(QEDAIOCB *acb, int ret) | |
52 | static void test_threshold_trigger(void) | 52 | QSIMPLEQ_REMOVE_HEAD(&s->allocating_write_reqs, next); |
53 | acb = QSIMPLEQ_FIRST(&s->allocating_write_reqs); | ||
54 | if (acb) { | ||
55 | - qed_aio_next_io(acb, 0); | ||
56 | + qed_aio_start_io(acb); | ||
57 | } else if (s->header.features & QED_F_NEED_CHECK) { | ||
58 | qed_start_need_check_timer(s); | ||
59 | } | ||
60 | @@ -XXX,XX +XXX,XX @@ static void qed_commit_l2_update(void *opaque, int ret) | ||
61 | acb->request.l2_table = qed_find_l2_cache_entry(&s->l2_cache, l2_offset); | ||
62 | assert(acb->request.l2_table != NULL); | ||
63 | |||
64 | - qed_aio_next_io(opaque, ret); | ||
65 | + qed_aio_next_io(acb, ret); | ||
66 | } | ||
67 | |||
68 | /** | ||
69 | @@ -XXX,XX +XXX,XX @@ static void qed_aio_write_l2_update(QEDAIOCB *acb, int ret, uint64_t offset) | ||
70 | if (need_alloc) { | ||
71 | /* Write out the whole new L2 table */ | ||
72 | qed_write_l2_table(s, &acb->request, 0, s->table_nelems, true, | ||
73 | - qed_aio_write_l1_update, acb); | ||
74 | + qed_aio_write_l1_update, acb); | ||
75 | } else { | ||
76 | /* Write out only the updated part of the L2 table */ | ||
77 | qed_write_l2_table(s, &acb->request, index, acb->cur_nclusters, false, | ||
78 | - qed_aio_next_io, acb); | ||
79 | + qed_aio_next_io_cb, acb); | ||
80 | } | ||
81 | return; | ||
82 | |||
83 | @@ -XXX,XX +XXX,XX @@ static void qed_aio_write_main(void *opaque, int ret) | ||
84 | } | ||
85 | |||
86 | if (acb->find_cluster_ret == QED_CLUSTER_FOUND) { | ||
87 | - next_fn = qed_aio_next_io; | ||
88 | + next_fn = qed_aio_next_io_cb; | ||
89 | } else { | ||
90 | if (s->bs->backing) { | ||
91 | next_fn = qed_aio_write_flush_before_l2_update; | ||
92 | @@ -XXX,XX +XXX,XX @@ static void qed_aio_write_alloc(QEDAIOCB *acb, size_t len) | ||
93 | if (acb->flags & QED_AIOCB_ZERO) { | ||
94 | /* Skip ahead if the clusters are already zero */ | ||
95 | if (acb->find_cluster_ret == QED_CLUSTER_ZERO) { | ||
96 | - qed_aio_next_io(acb, 0); | ||
97 | + qed_aio_start_io(acb); | ||
98 | return; | ||
99 | } | ||
100 | |||
101 | @@ -XXX,XX +XXX,XX @@ static void qed_aio_read_data(void *opaque, int ret, | ||
102 | /* Handle zero cluster and backing file reads */ | ||
103 | if (ret == QED_CLUSTER_ZERO) { | ||
104 | qemu_iovec_memset(&acb->cur_qiov, 0, 0, acb->cur_qiov.size); | ||
105 | - qed_aio_next_io(acb, 0); | ||
106 | + qed_aio_start_io(acb); | ||
107 | return; | ||
108 | } else if (ret != QED_CLUSTER_FOUND) { | ||
109 | qed_read_backing_file(s, acb->cur_pos, &acb->cur_qiov, | ||
110 | - &acb->backing_qiov, qed_aio_next_io, acb); | ||
111 | + &acb->backing_qiov, qed_aio_next_io_cb, acb); | ||
112 | return; | ||
113 | } | ||
114 | |||
115 | BLKDBG_EVENT(bs->file, BLKDBG_READ_AIO); | ||
116 | bdrv_aio_readv(bs->file, offset / BDRV_SECTOR_SIZE, | ||
117 | &acb->cur_qiov, acb->cur_qiov.size / BDRV_SECTOR_SIZE, | ||
118 | - qed_aio_next_io, acb); | ||
119 | + qed_aio_next_io_cb, acb); | ||
120 | return; | ||
121 | |||
122 | err: | ||
123 | @@ -XXX,XX +XXX,XX @@ err: | ||
124 | /** | ||
125 | * Begin next I/O or complete the request | ||
126 | */ | ||
127 | -static void qed_aio_next_io(void *opaque, int ret) | ||
128 | +static void qed_aio_next_io(QEDAIOCB *acb, int ret) | ||
53 | { | 129 | { |
54 | - uint64_t amount = 0; | 130 | - QEDAIOCB *acb = opaque; |
55 | uint64_t threshold = 4 * 1024 * 1024; | 131 | BDRVQEDState *s = acb_to_s(acb); |
56 | BlockDriverState bs; | 132 | QEDFindClusterFunc *io_fn = (acb->flags & QED_AIOCB_WRITE) ? |
57 | - BdrvTrackedRequest req; | 133 | qed_aio_write_data : qed_aio_read_data; |
58 | 134 | @@ -XXX,XX +XXX,XX @@ static BlockAIOCB *qed_aio_setup(BlockDriverState *bs, | |
59 | memset(&bs, 0, sizeof(bs)); | 135 | qemu_iovec_init(&acb->cur_qiov, qiov->niov); |
60 | - memset(&req, 0, sizeof(req)); | 136 | |
61 | - req.offset = (4 * 1024 * 1024) - 1024; | 137 | /* Start request */ |
62 | - req.bytes = 2 * 1024; | 138 | - qed_aio_next_io(acb, 0); |
63 | - | 139 | + qed_aio_start_io(acb); |
64 | - bdrv_check_request(req.offset, req.bytes, &error_abort); | 140 | return &acb->common; |
65 | |||
66 | bdrv_write_threshold_set(&bs, threshold); | ||
67 | - amount = bdrv_write_threshold_exceeded(&bs, &req); | ||
68 | - g_assert_cmpuint(amount, >=, 1024); | ||
69 | + bdrv_write_threshold_check_write(&bs, threshold - 1024, 2 * 1024); | ||
70 | + g_assert_cmpuint(bdrv_write_threshold_get(&bs), ==, 0); | ||
71 | } | 141 | } |
72 | 142 | ||
73 | typedef struct TestStruct { | ||
74 | -- | 143 | -- |
75 | 2.31.1 | 144 | 2.9.3 |
76 | 145 | ||
77 | 146 | diff view generated by jsdifflib |
1 | From: Paolo Bonzini <pbonzini@redhat.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | Due to a typo, in this case the SOCK_DIR was not being created. | 3 | The AioContext data structures are now protected by list_lock and/or |
4 | they are walked with FOREACH_RCU primitives. There is no need anymore | ||
5 | to acquire the AioContext for the entire duration of aio_dispatch. | ||
6 | Instead, just acquire it before and after invoking the callbacks. | ||
7 | The next step is then to push it further down. | ||
4 | 8 | ||
5 | Reviewed-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 9 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
6 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> | 10 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
7 | Tested-by: Emanuele Giuseppe Esposito <eesposit@redhat.com> | 11 | Reviewed-by: Fam Zheng <famz@redhat.com> |
8 | Message-Id: <20210323181928.311862-6-pbonzini@redhat.com> | 12 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> |
9 | Message-Id: <20210503110110.476887-6-pbonzini@redhat.com> | 13 | Message-id: 20170213135235.12274-12-pbonzini@redhat.com |
10 | Signed-off-by: Max Reitz <mreitz@redhat.com> | 14 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
11 | --- | 15 | --- |
12 | tests/qemu-iotests/testenv.py | 2 +- | 16 | util/aio-posix.c | 25 +++++++++++-------------- |
13 | 1 file changed, 1 insertion(+), 1 deletion(-) | 17 | util/aio-win32.c | 15 +++++++-------- |
18 | util/async.c | 2 ++ | ||
19 | 3 files changed, 20 insertions(+), 22 deletions(-) | ||
14 | 20 | ||
15 | diff --git a/tests/qemu-iotests/testenv.py b/tests/qemu-iotests/testenv.py | 21 | diff --git a/util/aio-posix.c b/util/aio-posix.c |
16 | index XXXXXXX..XXXXXXX 100644 | 22 | index XXXXXXX..XXXXXXX 100644 |
17 | --- a/tests/qemu-iotests/testenv.py | 23 | --- a/util/aio-posix.c |
18 | +++ b/tests/qemu-iotests/testenv.py | 24 | +++ b/util/aio-posix.c |
19 | @@ -XXX,XX +XXX,XX @@ def init_directories(self) -> None: | 25 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx) |
20 | try: | 26 | (revents & (G_IO_IN | G_IO_HUP | G_IO_ERR)) && |
21 | self.sock_dir = os.environ['SOCK_DIR'] | 27 | aio_node_check(ctx, node->is_external) && |
22 | self.tmp_sock_dir = False | 28 | node->io_read) { |
23 | - Path(self.test_dir).mkdir(parents=True, exist_ok=True) | 29 | + aio_context_acquire(ctx); |
24 | + Path(self.sock_dir).mkdir(parents=True, exist_ok=True) | 30 | node->io_read(node->opaque); |
25 | except KeyError: | 31 | + aio_context_release(ctx); |
26 | self.sock_dir = tempfile.mkdtemp() | 32 | |
27 | self.tmp_sock_dir = True | 33 | /* aio_notify() does not count as progress */ |
34 | if (node->opaque != &ctx->notifier) { | ||
35 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx) | ||
36 | (revents & (G_IO_OUT | G_IO_ERR)) && | ||
37 | aio_node_check(ctx, node->is_external) && | ||
38 | node->io_write) { | ||
39 | + aio_context_acquire(ctx); | ||
40 | node->io_write(node->opaque); | ||
41 | + aio_context_release(ctx); | ||
42 | progress = true; | ||
43 | } | ||
44 | |||
45 | @@ -XXX,XX +XXX,XX @@ bool aio_dispatch(AioContext *ctx, bool dispatch_fds) | ||
46 | } | ||
47 | |||
48 | /* Run our timers */ | ||
49 | + aio_context_acquire(ctx); | ||
50 | progress |= timerlistgroup_run_timers(&ctx->tlg); | ||
51 | + aio_context_release(ctx); | ||
52 | |||
53 | return progress; | ||
54 | } | ||
55 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
56 | int64_t timeout; | ||
57 | int64_t start = 0; | ||
58 | |||
59 | - aio_context_acquire(ctx); | ||
60 | - progress = false; | ||
61 | - | ||
62 | /* aio_notify can avoid the expensive event_notifier_set if | ||
63 | * everything (file descriptors, bottom halves, timers) will | ||
64 | * be re-evaluated before the next blocking poll(). This is | ||
65 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
66 | start = qemu_clock_get_ns(QEMU_CLOCK_REALTIME); | ||
67 | } | ||
68 | |||
69 | - if (try_poll_mode(ctx, blocking)) { | ||
70 | - progress = true; | ||
71 | - } else { | ||
72 | + aio_context_acquire(ctx); | ||
73 | + progress = try_poll_mode(ctx, blocking); | ||
74 | + aio_context_release(ctx); | ||
75 | + | ||
76 | + if (!progress) { | ||
77 | assert(npfd == 0); | ||
78 | |||
79 | /* fill pollfds */ | ||
80 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
81 | timeout = blocking ? aio_compute_timeout(ctx) : 0; | ||
82 | |||
83 | /* wait until next event */ | ||
84 | - if (timeout) { | ||
85 | - aio_context_release(ctx); | ||
86 | - } | ||
87 | if (aio_epoll_check_poll(ctx, pollfds, npfd, timeout)) { | ||
88 | AioHandler epoll_handler; | ||
89 | |||
90 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
91 | } else { | ||
92 | ret = qemu_poll_ns(pollfds, npfd, timeout); | ||
93 | } | ||
94 | - if (timeout) { | ||
95 | - aio_context_acquire(ctx); | ||
96 | - } | ||
97 | } | ||
98 | |||
99 | if (blocking) { | ||
100 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
101 | progress = true; | ||
102 | } | ||
103 | |||
104 | - aio_context_release(ctx); | ||
105 | - | ||
106 | return progress; | ||
107 | } | ||
108 | |||
109 | diff --git a/util/aio-win32.c b/util/aio-win32.c | ||
110 | index XXXXXXX..XXXXXXX 100644 | ||
111 | --- a/util/aio-win32.c | ||
112 | +++ b/util/aio-win32.c | ||
113 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx, HANDLE event) | ||
114 | (revents || event_notifier_get_handle(node->e) == event) && | ||
115 | node->io_notify) { | ||
116 | node->pfd.revents = 0; | ||
117 | + aio_context_acquire(ctx); | ||
118 | node->io_notify(node->e); | ||
119 | + aio_context_release(ctx); | ||
120 | |||
121 | /* aio_notify() does not count as progress */ | ||
122 | if (node->e != &ctx->notifier) { | ||
123 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx, HANDLE event) | ||
124 | (node->io_read || node->io_write)) { | ||
125 | node->pfd.revents = 0; | ||
126 | if ((revents & G_IO_IN) && node->io_read) { | ||
127 | + aio_context_acquire(ctx); | ||
128 | node->io_read(node->opaque); | ||
129 | + aio_context_release(ctx); | ||
130 | progress = true; | ||
131 | } | ||
132 | if ((revents & G_IO_OUT) && node->io_write) { | ||
133 | + aio_context_acquire(ctx); | ||
134 | node->io_write(node->opaque); | ||
135 | + aio_context_release(ctx); | ||
136 | progress = true; | ||
137 | } | ||
138 | |||
139 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
140 | int count; | ||
141 | int timeout; | ||
142 | |||
143 | - aio_context_acquire(ctx); | ||
144 | progress = false; | ||
145 | |||
146 | /* aio_notify can avoid the expensive event_notifier_set if | ||
147 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
148 | |||
149 | timeout = blocking && !have_select_revents | ||
150 | ? qemu_timeout_ns_to_ms(aio_compute_timeout(ctx)) : 0; | ||
151 | - if (timeout) { | ||
152 | - aio_context_release(ctx); | ||
153 | - } | ||
154 | ret = WaitForMultipleObjects(count, events, FALSE, timeout); | ||
155 | if (blocking) { | ||
156 | assert(first); | ||
157 | atomic_sub(&ctx->notify_me, 2); | ||
158 | } | ||
159 | - if (timeout) { | ||
160 | - aio_context_acquire(ctx); | ||
161 | - } | ||
162 | |||
163 | if (first) { | ||
164 | aio_notify_accept(ctx); | ||
165 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
166 | progress |= aio_dispatch_handlers(ctx, event); | ||
167 | } while (count > 0); | ||
168 | |||
169 | + aio_context_acquire(ctx); | ||
170 | progress |= timerlistgroup_run_timers(&ctx->tlg); | ||
171 | - | ||
172 | aio_context_release(ctx); | ||
173 | return progress; | ||
174 | } | ||
175 | diff --git a/util/async.c b/util/async.c | ||
176 | index XXXXXXX..XXXXXXX 100644 | ||
177 | --- a/util/async.c | ||
178 | +++ b/util/async.c | ||
179 | @@ -XXX,XX +XXX,XX @@ int aio_bh_poll(AioContext *ctx) | ||
180 | ret = 1; | ||
181 | } | ||
182 | bh->idle = 0; | ||
183 | + aio_context_acquire(ctx); | ||
184 | aio_bh_call(bh); | ||
185 | + aio_context_release(ctx); | ||
186 | } | ||
187 | if (bh->deleted) { | ||
188 | deleted = true; | ||
28 | -- | 189 | -- |
29 | 2.31.1 | 190 | 2.9.3 |
30 | 191 | ||
31 | 192 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | 1 | From: Paolo Bonzini <pbonzini@redhat.com> | |
2 | |||
3 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
4 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> | ||
5 | Reviewed-by: Fam Zheng <famz@redhat.com> | ||
6 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> | ||
7 | Message-id: 20170213135235.12274-13-pbonzini@redhat.com | ||
8 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
9 | --- | ||
10 | block/qed.h | 3 +++ | ||
11 | block/curl.c | 2 ++ | ||
12 | block/io.c | 5 +++++ | ||
13 | block/iscsi.c | 8 ++++++-- | ||
14 | block/null.c | 4 ++++ | ||
15 | block/qed.c | 12 ++++++++++++ | ||
16 | block/throttle-groups.c | 2 ++ | ||
17 | util/aio-posix.c | 2 -- | ||
18 | util/aio-win32.c | 2 -- | ||
19 | util/qemu-coroutine-sleep.c | 2 +- | ||
20 | 10 files changed, 35 insertions(+), 7 deletions(-) | ||
21 | |||
22 | diff --git a/block/qed.h b/block/qed.h | ||
23 | index XXXXXXX..XXXXXXX 100644 | ||
24 | --- a/block/qed.h | ||
25 | +++ b/block/qed.h | ||
26 | @@ -XXX,XX +XXX,XX @@ enum { | ||
27 | */ | ||
28 | typedef void QEDFindClusterFunc(void *opaque, int ret, uint64_t offset, size_t len); | ||
29 | |||
30 | +void qed_acquire(BDRVQEDState *s); | ||
31 | +void qed_release(BDRVQEDState *s); | ||
32 | + | ||
33 | /** | ||
34 | * Generic callback for chaining async callbacks | ||
35 | */ | ||
36 | diff --git a/block/curl.c b/block/curl.c | ||
37 | index XXXXXXX..XXXXXXX 100644 | ||
38 | --- a/block/curl.c | ||
39 | +++ b/block/curl.c | ||
40 | @@ -XXX,XX +XXX,XX @@ static void curl_multi_timeout_do(void *arg) | ||
41 | return; | ||
42 | } | ||
43 | |||
44 | + aio_context_acquire(s->aio_context); | ||
45 | curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running); | ||
46 | |||
47 | curl_multi_check_completion(s); | ||
48 | + aio_context_release(s->aio_context); | ||
49 | #else | ||
50 | abort(); | ||
51 | #endif | ||
52 | diff --git a/block/io.c b/block/io.c | ||
53 | index XXXXXXX..XXXXXXX 100644 | ||
54 | --- a/block/io.c | ||
55 | +++ b/block/io.c | ||
56 | @@ -XXX,XX +XXX,XX @@ void bdrv_aio_cancel(BlockAIOCB *acb) | ||
57 | if (acb->aiocb_info->get_aio_context) { | ||
58 | aio_poll(acb->aiocb_info->get_aio_context(acb), true); | ||
59 | } else if (acb->bs) { | ||
60 | + /* qemu_aio_ref and qemu_aio_unref are not thread-safe, so | ||
61 | + * assert that we're not using an I/O thread. Thread-safe | ||
62 | + * code should use bdrv_aio_cancel_async exclusively. | ||
63 | + */ | ||
64 | + assert(bdrv_get_aio_context(acb->bs) == qemu_get_aio_context()); | ||
65 | aio_poll(bdrv_get_aio_context(acb->bs), true); | ||
66 | } else { | ||
67 | abort(); | ||
68 | diff --git a/block/iscsi.c b/block/iscsi.c | ||
69 | index XXXXXXX..XXXXXXX 100644 | ||
70 | --- a/block/iscsi.c | ||
71 | +++ b/block/iscsi.c | ||
72 | @@ -XXX,XX +XXX,XX @@ static void iscsi_retry_timer_expired(void *opaque) | ||
73 | struct IscsiTask *iTask = opaque; | ||
74 | iTask->complete = 1; | ||
75 | if (iTask->co) { | ||
76 | - qemu_coroutine_enter(iTask->co); | ||
77 | + aio_co_wake(iTask->co); | ||
78 | } | ||
79 | } | ||
80 | |||
81 | @@ -XXX,XX +XXX,XX @@ static void iscsi_nop_timed_event(void *opaque) | ||
82 | { | ||
83 | IscsiLun *iscsilun = opaque; | ||
84 | |||
85 | + aio_context_acquire(iscsilun->aio_context); | ||
86 | if (iscsi_get_nops_in_flight(iscsilun->iscsi) >= MAX_NOP_FAILURES) { | ||
87 | error_report("iSCSI: NOP timeout. Reconnecting..."); | ||
88 | iscsilun->request_timed_out = true; | ||
89 | } else if (iscsi_nop_out_async(iscsilun->iscsi, NULL, NULL, 0, NULL) != 0) { | ||
90 | error_report("iSCSI: failed to sent NOP-Out. Disabling NOP messages."); | ||
91 | - return; | ||
92 | + goto out; | ||
93 | } | ||
94 | |||
95 | timer_mod(iscsilun->nop_timer, qemu_clock_get_ms(QEMU_CLOCK_REALTIME) + NOP_INTERVAL); | ||
96 | iscsi_set_events(iscsilun); | ||
97 | + | ||
98 | +out: | ||
99 | + aio_context_release(iscsilun->aio_context); | ||
100 | } | ||
101 | |||
102 | static void iscsi_readcapacity_sync(IscsiLun *iscsilun, Error **errp) | ||
103 | diff --git a/block/null.c b/block/null.c | ||
104 | index XXXXXXX..XXXXXXX 100644 | ||
105 | --- a/block/null.c | ||
106 | +++ b/block/null.c | ||
107 | @@ -XXX,XX +XXX,XX @@ static void null_bh_cb(void *opaque) | ||
108 | static void null_timer_cb(void *opaque) | ||
109 | { | ||
110 | NullAIOCB *acb = opaque; | ||
111 | + AioContext *ctx = bdrv_get_aio_context(acb->common.bs); | ||
112 | + | ||
113 | + aio_context_acquire(ctx); | ||
114 | acb->common.cb(acb->common.opaque, 0); | ||
115 | + aio_context_release(ctx); | ||
116 | timer_deinit(&acb->timer); | ||
117 | qemu_aio_unref(acb); | ||
118 | } | ||
119 | diff --git a/block/qed.c b/block/qed.c | ||
120 | index XXXXXXX..XXXXXXX 100644 | ||
121 | --- a/block/qed.c | ||
122 | +++ b/block/qed.c | ||
123 | @@ -XXX,XX +XXX,XX @@ static void qed_need_check_timer_cb(void *opaque) | ||
124 | |||
125 | trace_qed_need_check_timer_cb(s); | ||
126 | |||
127 | + qed_acquire(s); | ||
128 | qed_plug_allocating_write_reqs(s); | ||
129 | |||
130 | /* Ensure writes are on disk before clearing flag */ | ||
131 | bdrv_aio_flush(s->bs->file->bs, qed_clear_need_check, s); | ||
132 | + qed_release(s); | ||
133 | +} | ||
134 | + | ||
135 | +void qed_acquire(BDRVQEDState *s) | ||
136 | +{ | ||
137 | + aio_context_acquire(bdrv_get_aio_context(s->bs)); | ||
138 | +} | ||
139 | + | ||
140 | +void qed_release(BDRVQEDState *s) | ||
141 | +{ | ||
142 | + aio_context_release(bdrv_get_aio_context(s->bs)); | ||
143 | } | ||
144 | |||
145 | static void qed_start_need_check_timer(BDRVQEDState *s) | ||
146 | diff --git a/block/throttle-groups.c b/block/throttle-groups.c | ||
147 | index XXXXXXX..XXXXXXX 100644 | ||
148 | --- a/block/throttle-groups.c | ||
149 | +++ b/block/throttle-groups.c | ||
150 | @@ -XXX,XX +XXX,XX @@ static void timer_cb(BlockBackend *blk, bool is_write) | ||
151 | qemu_mutex_unlock(&tg->lock); | ||
152 | |||
153 | /* Run the request that was waiting for this timer */ | ||
154 | + aio_context_acquire(blk_get_aio_context(blk)); | ||
155 | empty_queue = !qemu_co_enter_next(&blkp->throttled_reqs[is_write]); | ||
156 | + aio_context_release(blk_get_aio_context(blk)); | ||
157 | |||
158 | /* If the request queue was empty then we have to take care of | ||
159 | * scheduling the next one */ | ||
160 | diff --git a/util/aio-posix.c b/util/aio-posix.c | ||
161 | index XXXXXXX..XXXXXXX 100644 | ||
162 | --- a/util/aio-posix.c | ||
163 | +++ b/util/aio-posix.c | ||
164 | @@ -XXX,XX +XXX,XX @@ bool aio_dispatch(AioContext *ctx, bool dispatch_fds) | ||
165 | } | ||
166 | |||
167 | /* Run our timers */ | ||
168 | - aio_context_acquire(ctx); | ||
169 | progress |= timerlistgroup_run_timers(&ctx->tlg); | ||
170 | - aio_context_release(ctx); | ||
171 | |||
172 | return progress; | ||
173 | } | ||
174 | diff --git a/util/aio-win32.c b/util/aio-win32.c | ||
175 | index XXXXXXX..XXXXXXX 100644 | ||
176 | --- a/util/aio-win32.c | ||
177 | +++ b/util/aio-win32.c | ||
178 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
179 | progress |= aio_dispatch_handlers(ctx, event); | ||
180 | } while (count > 0); | ||
181 | |||
182 | - aio_context_acquire(ctx); | ||
183 | progress |= timerlistgroup_run_timers(&ctx->tlg); | ||
184 | - aio_context_release(ctx); | ||
185 | return progress; | ||
186 | } | ||
187 | |||
188 | diff --git a/util/qemu-coroutine-sleep.c b/util/qemu-coroutine-sleep.c | ||
189 | index XXXXXXX..XXXXXXX 100644 | ||
190 | --- a/util/qemu-coroutine-sleep.c | ||
191 | +++ b/util/qemu-coroutine-sleep.c | ||
192 | @@ -XXX,XX +XXX,XX @@ static void co_sleep_cb(void *opaque) | ||
193 | { | ||
194 | CoSleepCB *sleep_cb = opaque; | ||
195 | |||
196 | - qemu_coroutine_enter(sleep_cb->co); | ||
197 | + aio_co_wake(sleep_cb->co); | ||
198 | } | ||
199 | |||
200 | void coroutine_fn co_aio_sleep_ns(AioContext *ctx, QEMUClockType type, | ||
201 | -- | ||
202 | 2.9.3 | ||
203 | |||
204 | diff view generated by jsdifflib |
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | "qemu/typedefs.h" is enough for include/block/write-threshold.h header | 3 | This covers both file descriptor callbacks and polling callbacks, |
4 | with forward declaration of BlockDriverState. Also drop extra includes | 4 | since they execute related code. |
5 | from block/write-threshold.c and tests/unit/test-write-threshold.c | ||
6 | 5 | ||
7 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
8 | Message-Id: <20210506090621.11848-9-vsementsov@virtuozzo.com> | ||
9 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | 6 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
10 | Signed-off-by: Max Reitz <mreitz@redhat.com> | 7 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
8 | Reviewed-by: Fam Zheng <famz@redhat.com> | ||
9 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> | ||
10 | Message-id: 20170213135235.12274-14-pbonzini@redhat.com | ||
11 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
11 | --- | 12 | --- |
12 | include/block/write-threshold.h | 2 +- | 13 | block/curl.c | 16 +++++++++++++--- |
13 | block/write-threshold.c | 2 -- | 14 | block/iscsi.c | 4 ++++ |
14 | tests/unit/test-write-threshold.c | 1 - | 15 | block/linux-aio.c | 4 ++++ |
15 | 3 files changed, 1 insertion(+), 4 deletions(-) | 16 | block/nfs.c | 6 ++++++ |
17 | block/sheepdog.c | 29 +++++++++++++++-------------- | ||
18 | block/ssh.c | 29 +++++++++-------------------- | ||
19 | block/win32-aio.c | 10 ++++++---- | ||
20 | hw/block/virtio-blk.c | 5 ++++- | ||
21 | hw/scsi/virtio-scsi.c | 7 +++++++ | ||
22 | util/aio-posix.c | 7 ------- | ||
23 | util/aio-win32.c | 6 ------ | ||
24 | 11 files changed, 68 insertions(+), 55 deletions(-) | ||
16 | 25 | ||
17 | diff --git a/include/block/write-threshold.h b/include/block/write-threshold.h | 26 | diff --git a/block/curl.c b/block/curl.c |
18 | index XXXXXXX..XXXXXXX 100644 | 27 | index XXXXXXX..XXXXXXX 100644 |
19 | --- a/include/block/write-threshold.h | 28 | --- a/block/curl.c |
20 | +++ b/include/block/write-threshold.h | 29 | +++ b/block/curl.c |
21 | @@ -XXX,XX +XXX,XX @@ | 30 | @@ -XXX,XX +XXX,XX @@ static void curl_multi_check_completion(BDRVCURLState *s) |
22 | #ifndef BLOCK_WRITE_THRESHOLD_H | 31 | } |
23 | #define BLOCK_WRITE_THRESHOLD_H | 32 | } |
24 | 33 | ||
25 | -#include "block/block_int.h" | 34 | -static void curl_multi_do(void *arg) |
26 | +#include "qemu/typedefs.h" | 35 | +static void curl_multi_do_locked(CURLState *s) |
36 | { | ||
37 | - CURLState *s = (CURLState *)arg; | ||
38 | CURLSocket *socket, *next_socket; | ||
39 | int running; | ||
40 | int r; | ||
41 | @@ -XXX,XX +XXX,XX @@ static void curl_multi_do(void *arg) | ||
42 | } | ||
43 | } | ||
44 | |||
45 | +static void curl_multi_do(void *arg) | ||
46 | +{ | ||
47 | + CURLState *s = (CURLState *)arg; | ||
48 | + | ||
49 | + aio_context_acquire(s->s->aio_context); | ||
50 | + curl_multi_do_locked(s); | ||
51 | + aio_context_release(s->s->aio_context); | ||
52 | +} | ||
53 | + | ||
54 | static void curl_multi_read(void *arg) | ||
55 | { | ||
56 | CURLState *s = (CURLState *)arg; | ||
57 | |||
58 | - curl_multi_do(arg); | ||
59 | + aio_context_acquire(s->s->aio_context); | ||
60 | + curl_multi_do_locked(s); | ||
61 | curl_multi_check_completion(s->s); | ||
62 | + aio_context_release(s->s->aio_context); | ||
63 | } | ||
64 | |||
65 | static void curl_multi_timeout_do(void *arg) | ||
66 | diff --git a/block/iscsi.c b/block/iscsi.c | ||
67 | index XXXXXXX..XXXXXXX 100644 | ||
68 | --- a/block/iscsi.c | ||
69 | +++ b/block/iscsi.c | ||
70 | @@ -XXX,XX +XXX,XX @@ iscsi_process_read(void *arg) | ||
71 | IscsiLun *iscsilun = arg; | ||
72 | struct iscsi_context *iscsi = iscsilun->iscsi; | ||
73 | |||
74 | + aio_context_acquire(iscsilun->aio_context); | ||
75 | iscsi_service(iscsi, POLLIN); | ||
76 | iscsi_set_events(iscsilun); | ||
77 | + aio_context_release(iscsilun->aio_context); | ||
78 | } | ||
79 | |||
80 | static void | ||
81 | @@ -XXX,XX +XXX,XX @@ iscsi_process_write(void *arg) | ||
82 | IscsiLun *iscsilun = arg; | ||
83 | struct iscsi_context *iscsi = iscsilun->iscsi; | ||
84 | |||
85 | + aio_context_acquire(iscsilun->aio_context); | ||
86 | iscsi_service(iscsi, POLLOUT); | ||
87 | iscsi_set_events(iscsilun); | ||
88 | + aio_context_release(iscsilun->aio_context); | ||
89 | } | ||
90 | |||
91 | static int64_t sector_lun2qemu(int64_t sector, IscsiLun *iscsilun) | ||
92 | diff --git a/block/linux-aio.c b/block/linux-aio.c | ||
93 | index XXXXXXX..XXXXXXX 100644 | ||
94 | --- a/block/linux-aio.c | ||
95 | +++ b/block/linux-aio.c | ||
96 | @@ -XXX,XX +XXX,XX @@ static void qemu_laio_completion_cb(EventNotifier *e) | ||
97 | LinuxAioState *s = container_of(e, LinuxAioState, e); | ||
98 | |||
99 | if (event_notifier_test_and_clear(&s->e)) { | ||
100 | + aio_context_acquire(s->aio_context); | ||
101 | qemu_laio_process_completions_and_submit(s); | ||
102 | + aio_context_release(s->aio_context); | ||
103 | } | ||
104 | } | ||
105 | |||
106 | @@ -XXX,XX +XXX,XX @@ static bool qemu_laio_poll_cb(void *opaque) | ||
107 | return false; | ||
108 | } | ||
109 | |||
110 | + aio_context_acquire(s->aio_context); | ||
111 | qemu_laio_process_completions_and_submit(s); | ||
112 | + aio_context_release(s->aio_context); | ||
113 | return true; | ||
114 | } | ||
115 | |||
116 | diff --git a/block/nfs.c b/block/nfs.c | ||
117 | index XXXXXXX..XXXXXXX 100644 | ||
118 | --- a/block/nfs.c | ||
119 | +++ b/block/nfs.c | ||
120 | @@ -XXX,XX +XXX,XX @@ static void nfs_set_events(NFSClient *client) | ||
121 | static void nfs_process_read(void *arg) | ||
122 | { | ||
123 | NFSClient *client = arg; | ||
124 | + | ||
125 | + aio_context_acquire(client->aio_context); | ||
126 | nfs_service(client->context, POLLIN); | ||
127 | nfs_set_events(client); | ||
128 | + aio_context_release(client->aio_context); | ||
129 | } | ||
130 | |||
131 | static void nfs_process_write(void *arg) | ||
132 | { | ||
133 | NFSClient *client = arg; | ||
134 | + | ||
135 | + aio_context_acquire(client->aio_context); | ||
136 | nfs_service(client->context, POLLOUT); | ||
137 | nfs_set_events(client); | ||
138 | + aio_context_release(client->aio_context); | ||
139 | } | ||
140 | |||
141 | static void nfs_co_init_task(BlockDriverState *bs, NFSRPC *task) | ||
142 | diff --git a/block/sheepdog.c b/block/sheepdog.c | ||
143 | index XXXXXXX..XXXXXXX 100644 | ||
144 | --- a/block/sheepdog.c | ||
145 | +++ b/block/sheepdog.c | ||
146 | @@ -XXX,XX +XXX,XX @@ static coroutine_fn int send_co_req(int sockfd, SheepdogReq *hdr, void *data, | ||
147 | return ret; | ||
148 | } | ||
149 | |||
150 | -static void restart_co_req(void *opaque) | ||
151 | -{ | ||
152 | - Coroutine *co = opaque; | ||
153 | - | ||
154 | - qemu_coroutine_enter(co); | ||
155 | -} | ||
156 | - | ||
157 | typedef struct SheepdogReqCo { | ||
158 | int sockfd; | ||
159 | BlockDriverState *bs; | ||
160 | @@ -XXX,XX +XXX,XX @@ typedef struct SheepdogReqCo { | ||
161 | unsigned int *rlen; | ||
162 | int ret; | ||
163 | bool finished; | ||
164 | + Coroutine *co; | ||
165 | } SheepdogReqCo; | ||
166 | |||
167 | +static void restart_co_req(void *opaque) | ||
168 | +{ | ||
169 | + SheepdogReqCo *srco = opaque; | ||
170 | + | ||
171 | + aio_co_wake(srco->co); | ||
172 | +} | ||
173 | + | ||
174 | static coroutine_fn void do_co_req(void *opaque) | ||
175 | { | ||
176 | int ret; | ||
177 | - Coroutine *co; | ||
178 | SheepdogReqCo *srco = opaque; | ||
179 | int sockfd = srco->sockfd; | ||
180 | SheepdogReq *hdr = srco->hdr; | ||
181 | @@ -XXX,XX +XXX,XX @@ static coroutine_fn void do_co_req(void *opaque) | ||
182 | unsigned int *wlen = srco->wlen; | ||
183 | unsigned int *rlen = srco->rlen; | ||
184 | |||
185 | - co = qemu_coroutine_self(); | ||
186 | + srco->co = qemu_coroutine_self(); | ||
187 | aio_set_fd_handler(srco->aio_context, sockfd, false, | ||
188 | - NULL, restart_co_req, NULL, co); | ||
189 | + NULL, restart_co_req, NULL, srco); | ||
190 | |||
191 | ret = send_co_req(sockfd, hdr, data, wlen); | ||
192 | if (ret < 0) { | ||
193 | @@ -XXX,XX +XXX,XX @@ static coroutine_fn void do_co_req(void *opaque) | ||
194 | } | ||
195 | |||
196 | aio_set_fd_handler(srco->aio_context, sockfd, false, | ||
197 | - restart_co_req, NULL, NULL, co); | ||
198 | + restart_co_req, NULL, NULL, srco); | ||
199 | |||
200 | ret = qemu_co_recv(sockfd, hdr, sizeof(*hdr)); | ||
201 | if (ret != sizeof(*hdr)) { | ||
202 | @@ -XXX,XX +XXX,XX @@ out: | ||
203 | aio_set_fd_handler(srco->aio_context, sockfd, false, | ||
204 | NULL, NULL, NULL, NULL); | ||
205 | |||
206 | + srco->co = NULL; | ||
207 | srco->ret = ret; | ||
208 | srco->finished = true; | ||
209 | if (srco->bs) { | ||
210 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn aio_read_response(void *opaque) | ||
211 | * We've finished all requests which belong to the AIOCB, so | ||
212 | * we can switch back to sd_co_readv/writev now. | ||
213 | */ | ||
214 | - qemu_coroutine_enter(acb->coroutine); | ||
215 | + aio_co_wake(acb->coroutine); | ||
216 | } | ||
217 | |||
218 | return; | ||
219 | @@ -XXX,XX +XXX,XX @@ static void co_read_response(void *opaque) | ||
220 | s->co_recv = qemu_coroutine_create(aio_read_response, opaque); | ||
221 | } | ||
222 | |||
223 | - qemu_coroutine_enter(s->co_recv); | ||
224 | + aio_co_wake(s->co_recv); | ||
225 | } | ||
226 | |||
227 | static void co_write_request(void *opaque) | ||
228 | { | ||
229 | BDRVSheepdogState *s = opaque; | ||
230 | |||
231 | - qemu_coroutine_enter(s->co_send); | ||
232 | + aio_co_wake(s->co_send); | ||
233 | } | ||
27 | 234 | ||
28 | /* | 235 | /* |
29 | * bdrv_write_threshold_set: | 236 | diff --git a/block/ssh.c b/block/ssh.c |
30 | diff --git a/block/write-threshold.c b/block/write-threshold.c | 237 | index XXXXXXX..XXXXXXX 100644 |
31 | index XXXXXXX..XXXXXXX 100644 | 238 | --- a/block/ssh.c |
32 | --- a/block/write-threshold.c | 239 | +++ b/block/ssh.c |
33 | +++ b/block/write-threshold.c | 240 | @@ -XXX,XX +XXX,XX @@ static void restart_coroutine(void *opaque) |
34 | @@ -XXX,XX +XXX,XX @@ | 241 | |
35 | 242 | DPRINTF("co=%p", co); | |
36 | #include "qemu/osdep.h" | 243 | |
37 | #include "block/block_int.h" | 244 | - qemu_coroutine_enter(co); |
38 | -#include "qemu/coroutine.h" | 245 | + aio_co_wake(co); |
39 | #include "block/write-threshold.h" | 246 | } |
40 | -#include "qemu/notify.h" | 247 | |
41 | #include "qapi/error.h" | 248 | -static coroutine_fn void set_fd_handler(BDRVSSHState *s, BlockDriverState *bs) |
42 | #include "qapi/qapi-commands-block-core.h" | 249 | +/* A non-blocking call returned EAGAIN, so yield, ensuring the |
43 | #include "qapi/qapi-events-block-core.h" | 250 | + * handlers are set up so that we'll be rescheduled when there is an |
44 | diff --git a/tests/unit/test-write-threshold.c b/tests/unit/test-write-threshold.c | 251 | + * interesting event on the socket. |
45 | index XXXXXXX..XXXXXXX 100644 | 252 | + */ |
46 | --- a/tests/unit/test-write-threshold.c | 253 | +static coroutine_fn void co_yield(BDRVSSHState *s, BlockDriverState *bs) |
47 | +++ b/tests/unit/test-write-threshold.c | 254 | { |
48 | @@ -XXX,XX +XXX,XX @@ | 255 | int r; |
49 | */ | 256 | IOHandler *rd_handler = NULL, *wr_handler = NULL; |
50 | 257 | @@ -XXX,XX +XXX,XX @@ static coroutine_fn void set_fd_handler(BDRVSSHState *s, BlockDriverState *bs) | |
51 | #include "qemu/osdep.h" | 258 | |
52 | -#include "qapi/error.h" | 259 | aio_set_fd_handler(bdrv_get_aio_context(bs), s->sock, |
53 | #include "block/block_int.h" | 260 | false, rd_handler, wr_handler, NULL, co); |
54 | #include "block/write-threshold.h" | 261 | -} |
262 | - | ||
263 | -static coroutine_fn void clear_fd_handler(BDRVSSHState *s, | ||
264 | - BlockDriverState *bs) | ||
265 | -{ | ||
266 | - DPRINTF("s->sock=%d", s->sock); | ||
267 | - aio_set_fd_handler(bdrv_get_aio_context(bs), s->sock, | ||
268 | - false, NULL, NULL, NULL, NULL); | ||
269 | -} | ||
270 | - | ||
271 | -/* A non-blocking call returned EAGAIN, so yield, ensuring the | ||
272 | - * handlers are set up so that we'll be rescheduled when there is an | ||
273 | - * interesting event on the socket. | ||
274 | - */ | ||
275 | -static coroutine_fn void co_yield(BDRVSSHState *s, BlockDriverState *bs) | ||
276 | -{ | ||
277 | - set_fd_handler(s, bs); | ||
278 | qemu_coroutine_yield(); | ||
279 | - clear_fd_handler(s, bs); | ||
280 | + DPRINTF("s->sock=%d - back", s->sock); | ||
281 | + aio_set_fd_handler(bdrv_get_aio_context(bs), s->sock, false, | ||
282 | + NULL, NULL, NULL, NULL); | ||
283 | } | ||
284 | |||
285 | /* SFTP has a function `libssh2_sftp_seek64' which seeks to a position | ||
286 | diff --git a/block/win32-aio.c b/block/win32-aio.c | ||
287 | index XXXXXXX..XXXXXXX 100644 | ||
288 | --- a/block/win32-aio.c | ||
289 | +++ b/block/win32-aio.c | ||
290 | @@ -XXX,XX +XXX,XX @@ struct QEMUWin32AIOState { | ||
291 | HANDLE hIOCP; | ||
292 | EventNotifier e; | ||
293 | int count; | ||
294 | - bool is_aio_context_attached; | ||
295 | + AioContext *aio_ctx; | ||
296 | }; | ||
297 | |||
298 | typedef struct QEMUWin32AIOCB { | ||
299 | @@ -XXX,XX +XXX,XX @@ static void win32_aio_process_completion(QEMUWin32AIOState *s, | ||
300 | } | ||
301 | |||
302 | |||
303 | + aio_context_acquire(s->aio_ctx); | ||
304 | waiocb->common.cb(waiocb->common.opaque, ret); | ||
305 | + aio_context_release(s->aio_ctx); | ||
306 | qemu_aio_unref(waiocb); | ||
307 | } | ||
308 | |||
309 | @@ -XXX,XX +XXX,XX @@ void win32_aio_detach_aio_context(QEMUWin32AIOState *aio, | ||
310 | AioContext *old_context) | ||
311 | { | ||
312 | aio_set_event_notifier(old_context, &aio->e, false, NULL, NULL); | ||
313 | - aio->is_aio_context_attached = false; | ||
314 | + aio->aio_ctx = NULL; | ||
315 | } | ||
316 | |||
317 | void win32_aio_attach_aio_context(QEMUWin32AIOState *aio, | ||
318 | AioContext *new_context) | ||
319 | { | ||
320 | - aio->is_aio_context_attached = true; | ||
321 | + aio->aio_ctx = new_context; | ||
322 | aio_set_event_notifier(new_context, &aio->e, false, | ||
323 | win32_aio_completion_cb, NULL); | ||
324 | } | ||
325 | @@ -XXX,XX +XXX,XX @@ out_free_state: | ||
326 | |||
327 | void win32_aio_cleanup(QEMUWin32AIOState *aio) | ||
328 | { | ||
329 | - assert(!aio->is_aio_context_attached); | ||
330 | + assert(!aio->aio_ctx); | ||
331 | CloseHandle(aio->hIOCP); | ||
332 | event_notifier_cleanup(&aio->e); | ||
333 | g_free(aio); | ||
334 | diff --git a/hw/block/virtio-blk.c b/hw/block/virtio-blk.c | ||
335 | index XXXXXXX..XXXXXXX 100644 | ||
336 | --- a/hw/block/virtio-blk.c | ||
337 | +++ b/hw/block/virtio-blk.c | ||
338 | @@ -XXX,XX +XXX,XX @@ static void virtio_blk_ioctl_complete(void *opaque, int status) | ||
339 | { | ||
340 | VirtIOBlockIoctlReq *ioctl_req = opaque; | ||
341 | VirtIOBlockReq *req = ioctl_req->req; | ||
342 | - VirtIODevice *vdev = VIRTIO_DEVICE(req->dev); | ||
343 | + VirtIOBlock *s = req->dev; | ||
344 | + VirtIODevice *vdev = VIRTIO_DEVICE(s); | ||
345 | struct virtio_scsi_inhdr *scsi; | ||
346 | struct sg_io_hdr *hdr; | ||
347 | |||
348 | @@ -XXX,XX +XXX,XX @@ bool virtio_blk_handle_vq(VirtIOBlock *s, VirtQueue *vq) | ||
349 | MultiReqBuffer mrb = {}; | ||
350 | bool progress = false; | ||
351 | |||
352 | + aio_context_acquire(blk_get_aio_context(s->blk)); | ||
353 | blk_io_plug(s->blk); | ||
354 | |||
355 | do { | ||
356 | @@ -XXX,XX +XXX,XX @@ bool virtio_blk_handle_vq(VirtIOBlock *s, VirtQueue *vq) | ||
357 | } | ||
358 | |||
359 | blk_io_unplug(s->blk); | ||
360 | + aio_context_release(blk_get_aio_context(s->blk)); | ||
361 | return progress; | ||
362 | } | ||
363 | |||
364 | diff --git a/hw/scsi/virtio-scsi.c b/hw/scsi/virtio-scsi.c | ||
365 | index XXXXXXX..XXXXXXX 100644 | ||
366 | --- a/hw/scsi/virtio-scsi.c | ||
367 | +++ b/hw/scsi/virtio-scsi.c | ||
368 | @@ -XXX,XX +XXX,XX @@ bool virtio_scsi_handle_ctrl_vq(VirtIOSCSI *s, VirtQueue *vq) | ||
369 | VirtIOSCSIReq *req; | ||
370 | bool progress = false; | ||
371 | |||
372 | + virtio_scsi_acquire(s); | ||
373 | while ((req = virtio_scsi_pop_req(s, vq))) { | ||
374 | progress = true; | ||
375 | virtio_scsi_handle_ctrl_req(s, req); | ||
376 | } | ||
377 | + virtio_scsi_release(s); | ||
378 | return progress; | ||
379 | } | ||
380 | |||
381 | @@ -XXX,XX +XXX,XX @@ bool virtio_scsi_handle_cmd_vq(VirtIOSCSI *s, VirtQueue *vq) | ||
382 | |||
383 | QTAILQ_HEAD(, VirtIOSCSIReq) reqs = QTAILQ_HEAD_INITIALIZER(reqs); | ||
384 | |||
385 | + virtio_scsi_acquire(s); | ||
386 | do { | ||
387 | virtio_queue_set_notification(vq, 0); | ||
388 | |||
389 | @@ -XXX,XX +XXX,XX @@ bool virtio_scsi_handle_cmd_vq(VirtIOSCSI *s, VirtQueue *vq) | ||
390 | QTAILQ_FOREACH_SAFE(req, &reqs, next, next) { | ||
391 | virtio_scsi_handle_cmd_req_submit(s, req); | ||
392 | } | ||
393 | + virtio_scsi_release(s); | ||
394 | return progress; | ||
395 | } | ||
396 | |||
397 | @@ -XXX,XX +XXX,XX @@ out: | ||
398 | |||
399 | bool virtio_scsi_handle_event_vq(VirtIOSCSI *s, VirtQueue *vq) | ||
400 | { | ||
401 | + virtio_scsi_acquire(s); | ||
402 | if (s->events_dropped) { | ||
403 | virtio_scsi_push_event(s, NULL, VIRTIO_SCSI_T_NO_EVENT, 0); | ||
404 | + virtio_scsi_release(s); | ||
405 | return true; | ||
406 | } | ||
407 | + virtio_scsi_release(s); | ||
408 | return false; | ||
409 | } | ||
410 | |||
411 | diff --git a/util/aio-posix.c b/util/aio-posix.c | ||
412 | index XXXXXXX..XXXXXXX 100644 | ||
413 | --- a/util/aio-posix.c | ||
414 | +++ b/util/aio-posix.c | ||
415 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx) | ||
416 | (revents & (G_IO_IN | G_IO_HUP | G_IO_ERR)) && | ||
417 | aio_node_check(ctx, node->is_external) && | ||
418 | node->io_read) { | ||
419 | - aio_context_acquire(ctx); | ||
420 | node->io_read(node->opaque); | ||
421 | - aio_context_release(ctx); | ||
422 | |||
423 | /* aio_notify() does not count as progress */ | ||
424 | if (node->opaque != &ctx->notifier) { | ||
425 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx) | ||
426 | (revents & (G_IO_OUT | G_IO_ERR)) && | ||
427 | aio_node_check(ctx, node->is_external) && | ||
428 | node->io_write) { | ||
429 | - aio_context_acquire(ctx); | ||
430 | node->io_write(node->opaque); | ||
431 | - aio_context_release(ctx); | ||
432 | progress = true; | ||
433 | } | ||
434 | |||
435 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
436 | start = qemu_clock_get_ns(QEMU_CLOCK_REALTIME); | ||
437 | } | ||
438 | |||
439 | - aio_context_acquire(ctx); | ||
440 | progress = try_poll_mode(ctx, blocking); | ||
441 | - aio_context_release(ctx); | ||
442 | - | ||
443 | if (!progress) { | ||
444 | assert(npfd == 0); | ||
445 | |||
446 | diff --git a/util/aio-win32.c b/util/aio-win32.c | ||
447 | index XXXXXXX..XXXXXXX 100644 | ||
448 | --- a/util/aio-win32.c | ||
449 | +++ b/util/aio-win32.c | ||
450 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx, HANDLE event) | ||
451 | (revents || event_notifier_get_handle(node->e) == event) && | ||
452 | node->io_notify) { | ||
453 | node->pfd.revents = 0; | ||
454 | - aio_context_acquire(ctx); | ||
455 | node->io_notify(node->e); | ||
456 | - aio_context_release(ctx); | ||
457 | |||
458 | /* aio_notify() does not count as progress */ | ||
459 | if (node->e != &ctx->notifier) { | ||
460 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx, HANDLE event) | ||
461 | (node->io_read || node->io_write)) { | ||
462 | node->pfd.revents = 0; | ||
463 | if ((revents & G_IO_IN) && node->io_read) { | ||
464 | - aio_context_acquire(ctx); | ||
465 | node->io_read(node->opaque); | ||
466 | - aio_context_release(ctx); | ||
467 | progress = true; | ||
468 | } | ||
469 | if ((revents & G_IO_OUT) && node->io_write) { | ||
470 | - aio_context_acquire(ctx); | ||
471 | node->io_write(node->opaque); | ||
472 | - aio_context_release(ctx); | ||
473 | progress = true; | ||
474 | } | ||
55 | 475 | ||
56 | -- | 476 | -- |
57 | 2.31.1 | 477 | 2.9.3 |
58 | 478 | ||
59 | 479 | diff view generated by jsdifflib |
1 | From: Paolo Bonzini <pbonzini@redhat.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | In the next patch, "check" will learn how to execute a test script without | 3 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
4 | going through TestRunner. To enable this, keep only the text output | 4 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
5 | and subprocess handling in the TestRunner; move into TestEnv the logic | 5 | Reviewed-by: Fam Zheng <famz@redhat.com> |
6 | to prepare for running a subprocess. | 6 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> |
7 | Message-id: 20170213135235.12274-15-pbonzini@redhat.com | ||
8 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
9 | --- | ||
10 | block/archipelago.c | 3 +++ | ||
11 | block/blkreplay.c | 2 +- | ||
12 | block/block-backend.c | 6 ++++++ | ||
13 | block/curl.c | 26 ++++++++++++++++++-------- | ||
14 | block/gluster.c | 9 +-------- | ||
15 | block/io.c | 6 +++++- | ||
16 | block/iscsi.c | 6 +++++- | ||
17 | block/linux-aio.c | 15 +++++++++------ | ||
18 | block/nfs.c | 3 ++- | ||
19 | block/null.c | 4 ++++ | ||
20 | block/qed.c | 3 +++ | ||
21 | block/rbd.c | 4 ++++ | ||
22 | dma-helpers.c | 2 ++ | ||
23 | hw/block/virtio-blk.c | 2 ++ | ||
24 | hw/scsi/scsi-bus.c | 2 ++ | ||
25 | util/async.c | 4 ++-- | ||
26 | util/thread-pool.c | 2 ++ | ||
27 | 17 files changed, 71 insertions(+), 28 deletions(-) | ||
7 | 28 | ||
8 | Reviewed-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 29 | diff --git a/block/archipelago.c b/block/archipelago.c |
9 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> | 30 | index XXXXXXX..XXXXXXX 100644 |
10 | Tested-by: Emanuele Giuseppe Esposito <eesposit@redhat.com> | 31 | --- a/block/archipelago.c |
11 | Message-Id: <20210323181928.311862-4-pbonzini@redhat.com> | 32 | +++ b/block/archipelago.c |
12 | Message-Id: <20210503110110.476887-4-pbonzini@redhat.com> | 33 | @@ -XXX,XX +XXX,XX @@ static void qemu_archipelago_complete_aio(void *opaque) |
13 | Signed-off-by: Max Reitz <mreitz@redhat.com> | 34 | { |
14 | --- | 35 | AIORequestData *reqdata = (AIORequestData *) opaque; |
15 | tests/qemu-iotests/testenv.py | 17 ++++++++++++++++- | 36 | ArchipelagoAIOCB *aio_cb = (ArchipelagoAIOCB *) reqdata->aio_cb; |
16 | tests/qemu-iotests/testrunner.py | 14 +------------- | 37 | + AioContext *ctx = bdrv_get_aio_context(aio_cb->common.bs); |
17 | 2 files changed, 17 insertions(+), 14 deletions(-) | 38 | |
18 | 39 | + aio_context_acquire(ctx); | |
19 | diff --git a/tests/qemu-iotests/testenv.py b/tests/qemu-iotests/testenv.py | 40 | aio_cb->common.cb(aio_cb->common.opaque, aio_cb->ret); |
20 | index XXXXXXX..XXXXXXX 100644 | 41 | + aio_context_release(ctx); |
21 | --- a/tests/qemu-iotests/testenv.py | 42 | aio_cb->status = 0; |
22 | +++ b/tests/qemu-iotests/testenv.py | 43 | |
23 | @@ -XXX,XX +XXX,XX @@ | 44 | qemu_aio_unref(aio_cb); |
24 | import random | 45 | diff --git a/block/blkreplay.c b/block/blkreplay.c |
25 | import subprocess | 46 | index XXXXXXX..XXXXXXX 100755 |
26 | import glob | 47 | --- a/block/blkreplay.c |
27 | -from typing import Dict, Any, Optional, ContextManager | 48 | +++ b/block/blkreplay.c |
28 | +from typing import List, Dict, Any, Optional, ContextManager | 49 | @@ -XXX,XX +XXX,XX @@ static int64_t blkreplay_getlength(BlockDriverState *bs) |
29 | 50 | static void blkreplay_bh_cb(void *opaque) | |
30 | 51 | { | |
31 | def isxfile(path: str) -> bool: | 52 | Request *req = opaque; |
32 | @@ -XXX,XX +XXX,XX @@ class TestEnv(ContextManager['TestEnv']): | 53 | - qemu_coroutine_enter(req->co); |
33 | 'CACHEMODE_IS_DEFAULT', 'IMGFMT_GENERIC', 'IMGOPTSSYNTAX', | 54 | + aio_co_wake(req->co); |
34 | 'IMGKEYSECRET', 'QEMU_DEFAULT_MACHINE', 'MALLOC_PERTURB_'] | 55 | qemu_bh_delete(req->bh); |
35 | 56 | g_free(req); | |
36 | + def prepare_subprocess(self, args: List[str]) -> Dict[str, str]: | 57 | } |
37 | + if self.debug: | 58 | diff --git a/block/block-backend.c b/block/block-backend.c |
38 | + args.append('-d') | 59 | index XXXXXXX..XXXXXXX 100644 |
39 | + | 60 | --- a/block/block-backend.c |
40 | + with open(args[0], encoding="utf-8") as f: | 61 | +++ b/block/block-backend.c |
41 | + try: | 62 | @@ -XXX,XX +XXX,XX @@ int blk_make_zero(BlockBackend *blk, BdrvRequestFlags flags) |
42 | + if f.readline().rstrip() == '#!/usr/bin/env python3': | 63 | static void error_callback_bh(void *opaque) |
43 | + args.insert(0, self.python) | 64 | { |
44 | + except UnicodeDecodeError: # binary test? for future. | 65 | struct BlockBackendAIOCB *acb = opaque; |
45 | + pass | 66 | + AioContext *ctx = bdrv_get_aio_context(acb->common.bs); |
46 | + | 67 | |
47 | + os_env = os.environ.copy() | 68 | bdrv_dec_in_flight(acb->common.bs); |
48 | + os_env.update(self.get_env()) | 69 | + aio_context_acquire(ctx); |
49 | + return os_env | 70 | acb->common.cb(acb->common.opaque, acb->ret); |
50 | + | 71 | + aio_context_release(ctx); |
51 | def get_env(self) -> Dict[str, str]: | 72 | qemu_aio_unref(acb); |
52 | env = {} | 73 | } |
53 | for v in self.env_variables: | 74 | |
54 | diff --git a/tests/qemu-iotests/testrunner.py b/tests/qemu-iotests/testrunner.py | 75 | @@ -XXX,XX +XXX,XX @@ static void blk_aio_complete(BlkAioEmAIOCB *acb) |
55 | index XXXXXXX..XXXXXXX 100644 | 76 | static void blk_aio_complete_bh(void *opaque) |
56 | --- a/tests/qemu-iotests/testrunner.py | 77 | { |
57 | +++ b/tests/qemu-iotests/testrunner.py | 78 | BlkAioEmAIOCB *acb = opaque; |
58 | @@ -XXX,XX +XXX,XX @@ class TestRunner(ContextManager['TestRunner']): | 79 | + AioContext *ctx = bdrv_get_aio_context(acb->common.bs); |
59 | def __init__(self, env: TestEnv, makecheck: bool = False, | 80 | |
60 | color: str = 'auto') -> None: | 81 | assert(acb->has_returned); |
61 | self.env = env | 82 | + aio_context_acquire(ctx); |
62 | - self.test_run_env = self.env.get_env() | 83 | blk_aio_complete(acb); |
63 | self.makecheck = makecheck | 84 | + aio_context_release(ctx); |
64 | self.last_elapsed = LastElapsedTime('.last-elapsed-cache', env) | 85 | } |
65 | 86 | ||
66 | @@ -XXX,XX +XXX,XX @@ def do_run_test(self, test: str) -> TestResult: | 87 | static BlockAIOCB *blk_aio_prwv(BlockBackend *blk, int64_t offset, int bytes, |
67 | silent_unlink(p) | 88 | diff --git a/block/curl.c b/block/curl.c |
68 | 89 | index XXXXXXX..XXXXXXX 100644 | |
69 | args = [str(f_test.resolve())] | 90 | --- a/block/curl.c |
70 | - if self.env.debug: | 91 | +++ b/block/curl.c |
71 | - args.append('-d') | 92 | @@ -XXX,XX +XXX,XX @@ static void curl_readv_bh_cb(void *p) |
93 | { | ||
94 | CURLState *state; | ||
95 | int running; | ||
96 | + int ret = -EINPROGRESS; | ||
97 | |||
98 | CURLAIOCB *acb = p; | ||
99 | - BDRVCURLState *s = acb->common.bs->opaque; | ||
100 | + BlockDriverState *bs = acb->common.bs; | ||
101 | + BDRVCURLState *s = bs->opaque; | ||
102 | + AioContext *ctx = bdrv_get_aio_context(bs); | ||
103 | |||
104 | size_t start = acb->sector_num * BDRV_SECTOR_SIZE; | ||
105 | size_t end; | ||
106 | |||
107 | + aio_context_acquire(ctx); | ||
108 | + | ||
109 | // In case we have the requested data already (e.g. read-ahead), | ||
110 | // we can just call the callback and be done. | ||
111 | switch (curl_find_buf(s, start, acb->nb_sectors * BDRV_SECTOR_SIZE, acb)) { | ||
112 | @@ -XXX,XX +XXX,XX @@ static void curl_readv_bh_cb(void *p) | ||
113 | qemu_aio_unref(acb); | ||
114 | // fall through | ||
115 | case FIND_RET_WAIT: | ||
116 | - return; | ||
117 | + goto out; | ||
118 | default: | ||
119 | break; | ||
120 | } | ||
121 | @@ -XXX,XX +XXX,XX @@ static void curl_readv_bh_cb(void *p) | ||
122 | // No cache found, so let's start a new request | ||
123 | state = curl_init_state(acb->common.bs, s); | ||
124 | if (!state) { | ||
125 | - acb->common.cb(acb->common.opaque, -EIO); | ||
126 | - qemu_aio_unref(acb); | ||
127 | - return; | ||
128 | + ret = -EIO; | ||
129 | + goto out; | ||
130 | } | ||
131 | |||
132 | acb->start = 0; | ||
133 | @@ -XXX,XX +XXX,XX @@ static void curl_readv_bh_cb(void *p) | ||
134 | state->orig_buf = g_try_malloc(state->buf_len); | ||
135 | if (state->buf_len && state->orig_buf == NULL) { | ||
136 | curl_clean_state(state); | ||
137 | - acb->common.cb(acb->common.opaque, -ENOMEM); | ||
138 | - qemu_aio_unref(acb); | ||
139 | - return; | ||
140 | + ret = -ENOMEM; | ||
141 | + goto out; | ||
142 | } | ||
143 | state->acb[0] = acb; | ||
144 | |||
145 | @@ -XXX,XX +XXX,XX @@ static void curl_readv_bh_cb(void *p) | ||
146 | |||
147 | /* Tell curl it needs to kick things off */ | ||
148 | curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running); | ||
149 | + | ||
150 | +out: | ||
151 | + if (ret != -EINPROGRESS) { | ||
152 | + acb->common.cb(acb->common.opaque, ret); | ||
153 | + qemu_aio_unref(acb); | ||
154 | + } | ||
155 | + aio_context_release(ctx); | ||
156 | } | ||
157 | |||
158 | static BlockAIOCB *curl_aio_readv(BlockDriverState *bs, | ||
159 | diff --git a/block/gluster.c b/block/gluster.c | ||
160 | index XXXXXXX..XXXXXXX 100644 | ||
161 | --- a/block/gluster.c | ||
162 | +++ b/block/gluster.c | ||
163 | @@ -XXX,XX +XXX,XX @@ static struct glfs *qemu_gluster_init(BlockdevOptionsGluster *gconf, | ||
164 | return qemu_gluster_glfs_init(gconf, errp); | ||
165 | } | ||
166 | |||
167 | -static void qemu_gluster_complete_aio(void *opaque) | ||
168 | -{ | ||
169 | - GlusterAIOCB *acb = (GlusterAIOCB *)opaque; | ||
72 | - | 170 | - |
73 | - with f_test.open(encoding="utf-8") as f: | 171 | - qemu_coroutine_enter(acb->coroutine); |
74 | - try: | 172 | -} |
75 | - if f.readline().rstrip() == '#!/usr/bin/env python3': | ||
76 | - args.insert(0, self.env.python) | ||
77 | - except UnicodeDecodeError: # binary test? for future. | ||
78 | - pass | ||
79 | - | 173 | - |
80 | - env = os.environ.copy() | 174 | /* |
81 | - env.update(self.test_run_env) | 175 | * AIO callback routine called from GlusterFS thread. |
82 | + env = self.env.prepare_subprocess(args) | 176 | */ |
83 | 177 | @@ -XXX,XX +XXX,XX @@ static void gluster_finish_aiocb(struct glfs_fd *fd, ssize_t ret, void *arg) | |
84 | t0 = time.time() | 178 | acb->ret = -EIO; /* Partial read/write - fail it */ |
85 | with f_bad.open('w', encoding="utf-8") as f: | 179 | } |
180 | |||
181 | - aio_bh_schedule_oneshot(acb->aio_context, qemu_gluster_complete_aio, acb); | ||
182 | + aio_co_schedule(acb->aio_context, acb->coroutine); | ||
183 | } | ||
184 | |||
185 | static void qemu_gluster_parse_flags(int bdrv_flags, int *open_flags) | ||
186 | diff --git a/block/io.c b/block/io.c | ||
187 | index XXXXXXX..XXXXXXX 100644 | ||
188 | --- a/block/io.c | ||
189 | +++ b/block/io.c | ||
190 | @@ -XXX,XX +XXX,XX @@ static void bdrv_co_drain_bh_cb(void *opaque) | ||
191 | bdrv_dec_in_flight(bs); | ||
192 | bdrv_drained_begin(bs); | ||
193 | data->done = true; | ||
194 | - qemu_coroutine_enter(co); | ||
195 | + aio_co_wake(co); | ||
196 | } | ||
197 | |||
198 | static void coroutine_fn bdrv_co_yield_to_drain(BlockDriverState *bs) | ||
199 | @@ -XXX,XX +XXX,XX @@ static void bdrv_co_complete(BlockAIOCBCoroutine *acb) | ||
200 | static void bdrv_co_em_bh(void *opaque) | ||
201 | { | ||
202 | BlockAIOCBCoroutine *acb = opaque; | ||
203 | + BlockDriverState *bs = acb->common.bs; | ||
204 | + AioContext *ctx = bdrv_get_aio_context(bs); | ||
205 | |||
206 | assert(!acb->need_bh); | ||
207 | + aio_context_acquire(ctx); | ||
208 | bdrv_co_complete(acb); | ||
209 | + aio_context_release(ctx); | ||
210 | } | ||
211 | |||
212 | static void bdrv_co_maybe_schedule_bh(BlockAIOCBCoroutine *acb) | ||
213 | diff --git a/block/iscsi.c b/block/iscsi.c | ||
214 | index XXXXXXX..XXXXXXX 100644 | ||
215 | --- a/block/iscsi.c | ||
216 | +++ b/block/iscsi.c | ||
217 | @@ -XXX,XX +XXX,XX @@ static void | ||
218 | iscsi_bh_cb(void *p) | ||
219 | { | ||
220 | IscsiAIOCB *acb = p; | ||
221 | + AioContext *ctx = bdrv_get_aio_context(acb->common.bs); | ||
222 | |||
223 | qemu_bh_delete(acb->bh); | ||
224 | |||
225 | g_free(acb->buf); | ||
226 | acb->buf = NULL; | ||
227 | |||
228 | + aio_context_acquire(ctx); | ||
229 | acb->common.cb(acb->common.opaque, acb->status); | ||
230 | + aio_context_release(ctx); | ||
231 | |||
232 | if (acb->task != NULL) { | ||
233 | scsi_free_scsi_task(acb->task); | ||
234 | @@ -XXX,XX +XXX,XX @@ iscsi_schedule_bh(IscsiAIOCB *acb) | ||
235 | static void iscsi_co_generic_bh_cb(void *opaque) | ||
236 | { | ||
237 | struct IscsiTask *iTask = opaque; | ||
238 | + | ||
239 | iTask->complete = 1; | ||
240 | - qemu_coroutine_enter(iTask->co); | ||
241 | + aio_co_wake(iTask->co); | ||
242 | } | ||
243 | |||
244 | static void iscsi_retry_timer_expired(void *opaque) | ||
245 | diff --git a/block/linux-aio.c b/block/linux-aio.c | ||
246 | index XXXXXXX..XXXXXXX 100644 | ||
247 | --- a/block/linux-aio.c | ||
248 | +++ b/block/linux-aio.c | ||
249 | @@ -XXX,XX +XXX,XX @@ struct LinuxAioState { | ||
250 | io_context_t ctx; | ||
251 | EventNotifier e; | ||
252 | |||
253 | - /* io queue for submit at batch */ | ||
254 | + /* io queue for submit at batch. Protected by AioContext lock. */ | ||
255 | LaioQueue io_q; | ||
256 | |||
257 | - /* I/O completion processing */ | ||
258 | + /* I/O completion processing. Only runs in I/O thread. */ | ||
259 | QEMUBH *completion_bh; | ||
260 | int event_idx; | ||
261 | int event_max; | ||
262 | @@ -XXX,XX +XXX,XX @@ static inline ssize_t io_event_ret(struct io_event *ev) | ||
263 | */ | ||
264 | static void qemu_laio_process_completion(struct qemu_laiocb *laiocb) | ||
265 | { | ||
266 | + LinuxAioState *s = laiocb->ctx; | ||
267 | int ret; | ||
268 | |||
269 | ret = laiocb->ret; | ||
270 | @@ -XXX,XX +XXX,XX @@ static void qemu_laio_process_completion(struct qemu_laiocb *laiocb) | ||
271 | } | ||
272 | |||
273 | laiocb->ret = ret; | ||
274 | + aio_context_acquire(s->aio_context); | ||
275 | if (laiocb->co) { | ||
276 | /* If the coroutine is already entered it must be in ioq_submit() and | ||
277 | * will notice laio->ret has been filled in when it eventually runs | ||
278 | @@ -XXX,XX +XXX,XX @@ static void qemu_laio_process_completion(struct qemu_laiocb *laiocb) | ||
279 | laiocb->common.cb(laiocb->common.opaque, ret); | ||
280 | qemu_aio_unref(laiocb); | ||
281 | } | ||
282 | + aio_context_release(s->aio_context); | ||
283 | } | ||
284 | |||
285 | /** | ||
286 | @@ -XXX,XX +XXX,XX @@ static void qemu_laio_process_completions(LinuxAioState *s) | ||
287 | static void qemu_laio_process_completions_and_submit(LinuxAioState *s) | ||
288 | { | ||
289 | qemu_laio_process_completions(s); | ||
290 | + | ||
291 | + aio_context_acquire(s->aio_context); | ||
292 | if (!s->io_q.plugged && !QSIMPLEQ_EMPTY(&s->io_q.pending)) { | ||
293 | ioq_submit(s); | ||
294 | } | ||
295 | + aio_context_release(s->aio_context); | ||
296 | } | ||
297 | |||
298 | static void qemu_laio_completion_bh(void *opaque) | ||
299 | @@ -XXX,XX +XXX,XX @@ static void qemu_laio_completion_cb(EventNotifier *e) | ||
300 | LinuxAioState *s = container_of(e, LinuxAioState, e); | ||
301 | |||
302 | if (event_notifier_test_and_clear(&s->e)) { | ||
303 | - aio_context_acquire(s->aio_context); | ||
304 | qemu_laio_process_completions_and_submit(s); | ||
305 | - aio_context_release(s->aio_context); | ||
306 | } | ||
307 | } | ||
308 | |||
309 | @@ -XXX,XX +XXX,XX @@ static bool qemu_laio_poll_cb(void *opaque) | ||
310 | return false; | ||
311 | } | ||
312 | |||
313 | - aio_context_acquire(s->aio_context); | ||
314 | qemu_laio_process_completions_and_submit(s); | ||
315 | - aio_context_release(s->aio_context); | ||
316 | return true; | ||
317 | } | ||
318 | |||
319 | @@ -XXX,XX +XXX,XX @@ void laio_detach_aio_context(LinuxAioState *s, AioContext *old_context) | ||
320 | { | ||
321 | aio_set_event_notifier(old_context, &s->e, false, NULL, NULL); | ||
322 | qemu_bh_delete(s->completion_bh); | ||
323 | + s->aio_context = NULL; | ||
324 | } | ||
325 | |||
326 | void laio_attach_aio_context(LinuxAioState *s, AioContext *new_context) | ||
327 | diff --git a/block/nfs.c b/block/nfs.c | ||
328 | index XXXXXXX..XXXXXXX 100644 | ||
329 | --- a/block/nfs.c | ||
330 | +++ b/block/nfs.c | ||
331 | @@ -XXX,XX +XXX,XX @@ static void nfs_co_init_task(BlockDriverState *bs, NFSRPC *task) | ||
332 | static void nfs_co_generic_bh_cb(void *opaque) | ||
333 | { | ||
334 | NFSRPC *task = opaque; | ||
335 | + | ||
336 | task->complete = 1; | ||
337 | - qemu_coroutine_enter(task->co); | ||
338 | + aio_co_wake(task->co); | ||
339 | } | ||
340 | |||
341 | static void | ||
342 | diff --git a/block/null.c b/block/null.c | ||
343 | index XXXXXXX..XXXXXXX 100644 | ||
344 | --- a/block/null.c | ||
345 | +++ b/block/null.c | ||
346 | @@ -XXX,XX +XXX,XX @@ static const AIOCBInfo null_aiocb_info = { | ||
347 | static void null_bh_cb(void *opaque) | ||
348 | { | ||
349 | NullAIOCB *acb = opaque; | ||
350 | + AioContext *ctx = bdrv_get_aio_context(acb->common.bs); | ||
351 | + | ||
352 | + aio_context_acquire(ctx); | ||
353 | acb->common.cb(acb->common.opaque, 0); | ||
354 | + aio_context_release(ctx); | ||
355 | qemu_aio_unref(acb); | ||
356 | } | ||
357 | |||
358 | diff --git a/block/qed.c b/block/qed.c | ||
359 | index XXXXXXX..XXXXXXX 100644 | ||
360 | --- a/block/qed.c | ||
361 | +++ b/block/qed.c | ||
362 | @@ -XXX,XX +XXX,XX @@ static void qed_update_l2_table(BDRVQEDState *s, QEDTable *table, int index, | ||
363 | static void qed_aio_complete_bh(void *opaque) | ||
364 | { | ||
365 | QEDAIOCB *acb = opaque; | ||
366 | + BDRVQEDState *s = acb_to_s(acb); | ||
367 | BlockCompletionFunc *cb = acb->common.cb; | ||
368 | void *user_opaque = acb->common.opaque; | ||
369 | int ret = acb->bh_ret; | ||
370 | @@ -XXX,XX +XXX,XX @@ static void qed_aio_complete_bh(void *opaque) | ||
371 | qemu_aio_unref(acb); | ||
372 | |||
373 | /* Invoke callback */ | ||
374 | + qed_acquire(s); | ||
375 | cb(user_opaque, ret); | ||
376 | + qed_release(s); | ||
377 | } | ||
378 | |||
379 | static void qed_aio_complete(QEDAIOCB *acb, int ret) | ||
380 | diff --git a/block/rbd.c b/block/rbd.c | ||
381 | index XXXXXXX..XXXXXXX 100644 | ||
382 | --- a/block/rbd.c | ||
383 | +++ b/block/rbd.c | ||
384 | @@ -XXX,XX +XXX,XX @@ shutdown: | ||
385 | static void qemu_rbd_complete_aio(RADOSCB *rcb) | ||
386 | { | ||
387 | RBDAIOCB *acb = rcb->acb; | ||
388 | + AioContext *ctx = bdrv_get_aio_context(acb->common.bs); | ||
389 | int64_t r; | ||
390 | |||
391 | r = rcb->ret; | ||
392 | @@ -XXX,XX +XXX,XX @@ static void qemu_rbd_complete_aio(RADOSCB *rcb) | ||
393 | qemu_iovec_from_buf(acb->qiov, 0, acb->bounce, acb->qiov->size); | ||
394 | } | ||
395 | qemu_vfree(acb->bounce); | ||
396 | + | ||
397 | + aio_context_acquire(ctx); | ||
398 | acb->common.cb(acb->common.opaque, (acb->ret > 0 ? 0 : acb->ret)); | ||
399 | + aio_context_release(ctx); | ||
400 | |||
401 | qemu_aio_unref(acb); | ||
402 | } | ||
403 | diff --git a/dma-helpers.c b/dma-helpers.c | ||
404 | index XXXXXXX..XXXXXXX 100644 | ||
405 | --- a/dma-helpers.c | ||
406 | +++ b/dma-helpers.c | ||
407 | @@ -XXX,XX +XXX,XX @@ static void dma_blk_cb(void *opaque, int ret) | ||
408 | QEMU_ALIGN_DOWN(dbs->iov.size, dbs->align)); | ||
409 | } | ||
410 | |||
411 | + aio_context_acquire(dbs->ctx); | ||
412 | dbs->acb = dbs->io_func(dbs->offset, &dbs->iov, | ||
413 | dma_blk_cb, dbs, dbs->io_func_opaque); | ||
414 | + aio_context_release(dbs->ctx); | ||
415 | assert(dbs->acb); | ||
416 | } | ||
417 | |||
418 | diff --git a/hw/block/virtio-blk.c b/hw/block/virtio-blk.c | ||
419 | index XXXXXXX..XXXXXXX 100644 | ||
420 | --- a/hw/block/virtio-blk.c | ||
421 | +++ b/hw/block/virtio-blk.c | ||
422 | @@ -XXX,XX +XXX,XX @@ static void virtio_blk_dma_restart_bh(void *opaque) | ||
423 | |||
424 | s->rq = NULL; | ||
425 | |||
426 | + aio_context_acquire(blk_get_aio_context(s->conf.conf.blk)); | ||
427 | while (req) { | ||
428 | VirtIOBlockReq *next = req->next; | ||
429 | if (virtio_blk_handle_request(req, &mrb)) { | ||
430 | @@ -XXX,XX +XXX,XX @@ static void virtio_blk_dma_restart_bh(void *opaque) | ||
431 | if (mrb.num_reqs) { | ||
432 | virtio_blk_submit_multireq(s->blk, &mrb); | ||
433 | } | ||
434 | + aio_context_release(blk_get_aio_context(s->conf.conf.blk)); | ||
435 | } | ||
436 | |||
437 | static void virtio_blk_dma_restart_cb(void *opaque, int running, | ||
438 | diff --git a/hw/scsi/scsi-bus.c b/hw/scsi/scsi-bus.c | ||
439 | index XXXXXXX..XXXXXXX 100644 | ||
440 | --- a/hw/scsi/scsi-bus.c | ||
441 | +++ b/hw/scsi/scsi-bus.c | ||
442 | @@ -XXX,XX +XXX,XX @@ static void scsi_dma_restart_bh(void *opaque) | ||
443 | qemu_bh_delete(s->bh); | ||
444 | s->bh = NULL; | ||
445 | |||
446 | + aio_context_acquire(blk_get_aio_context(s->conf.blk)); | ||
447 | QTAILQ_FOREACH_SAFE(req, &s->requests, next, next) { | ||
448 | scsi_req_ref(req); | ||
449 | if (req->retry) { | ||
450 | @@ -XXX,XX +XXX,XX @@ static void scsi_dma_restart_bh(void *opaque) | ||
451 | } | ||
452 | scsi_req_unref(req); | ||
453 | } | ||
454 | + aio_context_release(blk_get_aio_context(s->conf.blk)); | ||
455 | } | ||
456 | |||
457 | void scsi_req_retry(SCSIRequest *req) | ||
458 | diff --git a/util/async.c b/util/async.c | ||
459 | index XXXXXXX..XXXXXXX 100644 | ||
460 | --- a/util/async.c | ||
461 | +++ b/util/async.c | ||
462 | @@ -XXX,XX +XXX,XX @@ int aio_bh_poll(AioContext *ctx) | ||
463 | ret = 1; | ||
464 | } | ||
465 | bh->idle = 0; | ||
466 | - aio_context_acquire(ctx); | ||
467 | aio_bh_call(bh); | ||
468 | - aio_context_release(ctx); | ||
469 | } | ||
470 | if (bh->deleted) { | ||
471 | deleted = true; | ||
472 | @@ -XXX,XX +XXX,XX @@ static void co_schedule_bh_cb(void *opaque) | ||
473 | Coroutine *co = QSLIST_FIRST(&straight); | ||
474 | QSLIST_REMOVE_HEAD(&straight, co_scheduled_next); | ||
475 | trace_aio_co_schedule_bh_cb(ctx, co); | ||
476 | + aio_context_acquire(ctx); | ||
477 | qemu_coroutine_enter(co); | ||
478 | + aio_context_release(ctx); | ||
479 | } | ||
480 | } | ||
481 | |||
482 | diff --git a/util/thread-pool.c b/util/thread-pool.c | ||
483 | index XXXXXXX..XXXXXXX 100644 | ||
484 | --- a/util/thread-pool.c | ||
485 | +++ b/util/thread-pool.c | ||
486 | @@ -XXX,XX +XXX,XX @@ static void thread_pool_completion_bh(void *opaque) | ||
487 | ThreadPool *pool = opaque; | ||
488 | ThreadPoolElement *elem, *next; | ||
489 | |||
490 | + aio_context_acquire(pool->ctx); | ||
491 | restart: | ||
492 | QLIST_FOREACH_SAFE(elem, &pool->head, all, next) { | ||
493 | if (elem->state != THREAD_DONE) { | ||
494 | @@ -XXX,XX +XXX,XX @@ restart: | ||
495 | qemu_aio_unref(elem); | ||
496 | } | ||
497 | } | ||
498 | + aio_context_release(pool->ctx); | ||
499 | } | ||
500 | |||
501 | static void thread_pool_cancel(BlockAIOCB *acb) | ||
86 | -- | 502 | -- |
87 | 2.31.1 | 503 | 2.9.3 |
88 | 504 | ||
89 | 505 | diff view generated by jsdifflib |
1 | From: Connor Kuehl <ckuehl@redhat.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | Sometimes the parser needs to further split a token it has collected | 3 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
4 | from the token input stream. Right now, it does a cursory check to see | 4 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
5 | if the relevant characters appear in the token to determine if it should | 5 | Reviewed-by: Fam Zheng <famz@redhat.com> |
6 | break it down further. | 6 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> |
7 | Message-id: 20170213135235.12274-16-pbonzini@redhat.com | ||
8 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
9 | --- | ||
10 | block/archipelago.c | 3 --- | ||
11 | block/block-backend.c | 7 ------- | ||
12 | block/curl.c | 2 +- | ||
13 | block/io.c | 6 +----- | ||
14 | block/iscsi.c | 3 --- | ||
15 | block/linux-aio.c | 5 +---- | ||
16 | block/mirror.c | 12 +++++++++--- | ||
17 | block/null.c | 8 -------- | ||
18 | block/qed-cluster.c | 2 ++ | ||
19 | block/qed-table.c | 12 ++++++++++-- | ||
20 | block/qed.c | 4 ++-- | ||
21 | block/rbd.c | 4 ---- | ||
22 | block/win32-aio.c | 3 --- | ||
23 | hw/block/virtio-blk.c | 12 +++++++++++- | ||
24 | hw/scsi/scsi-disk.c | 15 +++++++++++++++ | ||
25 | hw/scsi/scsi-generic.c | 20 +++++++++++++++++--- | ||
26 | util/thread-pool.c | 4 +++- | ||
27 | 17 files changed, 72 insertions(+), 50 deletions(-) | ||
7 | 28 | ||
8 | However, qemu_rbd_next_tok() will escape characters as it removes tokens | 29 | diff --git a/block/archipelago.c b/block/archipelago.c |
9 | from the token stream and plain strchr() won't. This can make the | 30 | index XXXXXXX..XXXXXXX 100644 |
10 | initial strchr() check slightly misleading since it implies | 31 | --- a/block/archipelago.c |
11 | qemu_rbd_next_tok() will find the token and split on it, except the | 32 | +++ b/block/archipelago.c |
12 | reality is that qemu_rbd_next_tok() will pass over it if it is escaped. | 33 | @@ -XXX,XX +XXX,XX @@ static void qemu_archipelago_complete_aio(void *opaque) |
13 | 34 | { | |
14 | Use a custom strchr to avoid mixing escaped and unescaped string | 35 | AIORequestData *reqdata = (AIORequestData *) opaque; |
15 | operations. Furthermore, this code is identical to how | 36 | ArchipelagoAIOCB *aio_cb = (ArchipelagoAIOCB *) reqdata->aio_cb; |
16 | qemu_rbd_next_tok() seeks its next token, so incorporate this custom | 37 | - AioContext *ctx = bdrv_get_aio_context(aio_cb->common.bs); |
17 | strchr into the body of that function to reduce duplication. | 38 | |
18 | 39 | - aio_context_acquire(ctx); | |
19 | Reported-by: Han Han <hhan@redhat.com> | 40 | aio_cb->common.cb(aio_cb->common.opaque, aio_cb->ret); |
20 | Fixes: https://bugzilla.redhat.com/1873913 | 41 | - aio_context_release(ctx); |
21 | Signed-off-by: Connor Kuehl <ckuehl@redhat.com> | 42 | aio_cb->status = 0; |
22 | Message-Id: <20210421212343.85524-3-ckuehl@redhat.com> | 43 | |
23 | Reviewed-by: Stefano Garzarella <sgarzare@redhat.com> | 44 | qemu_aio_unref(aio_cb); |
24 | Signed-off-by: Max Reitz <mreitz@redhat.com> | 45 | diff --git a/block/block-backend.c b/block/block-backend.c |
25 | --- | 46 | index XXXXXXX..XXXXXXX 100644 |
26 | block/rbd.c | 32 +++++++++++++++++++++----------- | 47 | --- a/block/block-backend.c |
27 | tests/qemu-iotests/231 | 4 ++++ | 48 | +++ b/block/block-backend.c |
28 | tests/qemu-iotests/231.out | 3 +++ | 49 | @@ -XXX,XX +XXX,XX @@ int blk_make_zero(BlockBackend *blk, BdrvRequestFlags flags) |
29 | 3 files changed, 28 insertions(+), 11 deletions(-) | 50 | static void error_callback_bh(void *opaque) |
30 | 51 | { | |
52 | struct BlockBackendAIOCB *acb = opaque; | ||
53 | - AioContext *ctx = bdrv_get_aio_context(acb->common.bs); | ||
54 | |||
55 | bdrv_dec_in_flight(acb->common.bs); | ||
56 | - aio_context_acquire(ctx); | ||
57 | acb->common.cb(acb->common.opaque, acb->ret); | ||
58 | - aio_context_release(ctx); | ||
59 | qemu_aio_unref(acb); | ||
60 | } | ||
61 | |||
62 | @@ -XXX,XX +XXX,XX @@ static void blk_aio_complete(BlkAioEmAIOCB *acb) | ||
63 | static void blk_aio_complete_bh(void *opaque) | ||
64 | { | ||
65 | BlkAioEmAIOCB *acb = opaque; | ||
66 | - AioContext *ctx = bdrv_get_aio_context(acb->common.bs); | ||
67 | - | ||
68 | assert(acb->has_returned); | ||
69 | - aio_context_acquire(ctx); | ||
70 | blk_aio_complete(acb); | ||
71 | - aio_context_release(ctx); | ||
72 | } | ||
73 | |||
74 | static BlockAIOCB *blk_aio_prwv(BlockBackend *blk, int64_t offset, int bytes, | ||
75 | diff --git a/block/curl.c b/block/curl.c | ||
76 | index XXXXXXX..XXXXXXX 100644 | ||
77 | --- a/block/curl.c | ||
78 | +++ b/block/curl.c | ||
79 | @@ -XXX,XX +XXX,XX @@ static void curl_readv_bh_cb(void *p) | ||
80 | curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running); | ||
81 | |||
82 | out: | ||
83 | + aio_context_release(ctx); | ||
84 | if (ret != -EINPROGRESS) { | ||
85 | acb->common.cb(acb->common.opaque, ret); | ||
86 | qemu_aio_unref(acb); | ||
87 | } | ||
88 | - aio_context_release(ctx); | ||
89 | } | ||
90 | |||
91 | static BlockAIOCB *curl_aio_readv(BlockDriverState *bs, | ||
92 | diff --git a/block/io.c b/block/io.c | ||
93 | index XXXXXXX..XXXXXXX 100644 | ||
94 | --- a/block/io.c | ||
95 | +++ b/block/io.c | ||
96 | @@ -XXX,XX +XXX,XX @@ static void bdrv_co_io_em_complete(void *opaque, int ret) | ||
97 | CoroutineIOCompletion *co = opaque; | ||
98 | |||
99 | co->ret = ret; | ||
100 | - qemu_coroutine_enter(co->coroutine); | ||
101 | + aio_co_wake(co->coroutine); | ||
102 | } | ||
103 | |||
104 | static int coroutine_fn bdrv_driver_preadv(BlockDriverState *bs, | ||
105 | @@ -XXX,XX +XXX,XX @@ static void bdrv_co_complete(BlockAIOCBCoroutine *acb) | ||
106 | static void bdrv_co_em_bh(void *opaque) | ||
107 | { | ||
108 | BlockAIOCBCoroutine *acb = opaque; | ||
109 | - BlockDriverState *bs = acb->common.bs; | ||
110 | - AioContext *ctx = bdrv_get_aio_context(bs); | ||
111 | |||
112 | assert(!acb->need_bh); | ||
113 | - aio_context_acquire(ctx); | ||
114 | bdrv_co_complete(acb); | ||
115 | - aio_context_release(ctx); | ||
116 | } | ||
117 | |||
118 | static void bdrv_co_maybe_schedule_bh(BlockAIOCBCoroutine *acb) | ||
119 | diff --git a/block/iscsi.c b/block/iscsi.c | ||
120 | index XXXXXXX..XXXXXXX 100644 | ||
121 | --- a/block/iscsi.c | ||
122 | +++ b/block/iscsi.c | ||
123 | @@ -XXX,XX +XXX,XX @@ static void | ||
124 | iscsi_bh_cb(void *p) | ||
125 | { | ||
126 | IscsiAIOCB *acb = p; | ||
127 | - AioContext *ctx = bdrv_get_aio_context(acb->common.bs); | ||
128 | |||
129 | qemu_bh_delete(acb->bh); | ||
130 | |||
131 | g_free(acb->buf); | ||
132 | acb->buf = NULL; | ||
133 | |||
134 | - aio_context_acquire(ctx); | ||
135 | acb->common.cb(acb->common.opaque, acb->status); | ||
136 | - aio_context_release(ctx); | ||
137 | |||
138 | if (acb->task != NULL) { | ||
139 | scsi_free_scsi_task(acb->task); | ||
140 | diff --git a/block/linux-aio.c b/block/linux-aio.c | ||
141 | index XXXXXXX..XXXXXXX 100644 | ||
142 | --- a/block/linux-aio.c | ||
143 | +++ b/block/linux-aio.c | ||
144 | @@ -XXX,XX +XXX,XX @@ static inline ssize_t io_event_ret(struct io_event *ev) | ||
145 | */ | ||
146 | static void qemu_laio_process_completion(struct qemu_laiocb *laiocb) | ||
147 | { | ||
148 | - LinuxAioState *s = laiocb->ctx; | ||
149 | int ret; | ||
150 | |||
151 | ret = laiocb->ret; | ||
152 | @@ -XXX,XX +XXX,XX @@ static void qemu_laio_process_completion(struct qemu_laiocb *laiocb) | ||
153 | } | ||
154 | |||
155 | laiocb->ret = ret; | ||
156 | - aio_context_acquire(s->aio_context); | ||
157 | if (laiocb->co) { | ||
158 | /* If the coroutine is already entered it must be in ioq_submit() and | ||
159 | * will notice laio->ret has been filled in when it eventually runs | ||
160 | @@ -XXX,XX +XXX,XX @@ static void qemu_laio_process_completion(struct qemu_laiocb *laiocb) | ||
161 | * that! | ||
162 | */ | ||
163 | if (!qemu_coroutine_entered(laiocb->co)) { | ||
164 | - qemu_coroutine_enter(laiocb->co); | ||
165 | + aio_co_wake(laiocb->co); | ||
166 | } | ||
167 | } else { | ||
168 | laiocb->common.cb(laiocb->common.opaque, ret); | ||
169 | qemu_aio_unref(laiocb); | ||
170 | } | ||
171 | - aio_context_release(s->aio_context); | ||
172 | } | ||
173 | |||
174 | /** | ||
175 | diff --git a/block/mirror.c b/block/mirror.c | ||
176 | index XXXXXXX..XXXXXXX 100644 | ||
177 | --- a/block/mirror.c | ||
178 | +++ b/block/mirror.c | ||
179 | @@ -XXX,XX +XXX,XX @@ static void mirror_write_complete(void *opaque, int ret) | ||
180 | { | ||
181 | MirrorOp *op = opaque; | ||
182 | MirrorBlockJob *s = op->s; | ||
183 | + | ||
184 | + aio_context_acquire(blk_get_aio_context(s->common.blk)); | ||
185 | if (ret < 0) { | ||
186 | BlockErrorAction action; | ||
187 | |||
188 | @@ -XXX,XX +XXX,XX @@ static void mirror_write_complete(void *opaque, int ret) | ||
189 | } | ||
190 | } | ||
191 | mirror_iteration_done(op, ret); | ||
192 | + aio_context_release(blk_get_aio_context(s->common.blk)); | ||
193 | } | ||
194 | |||
195 | static void mirror_read_complete(void *opaque, int ret) | ||
196 | { | ||
197 | MirrorOp *op = opaque; | ||
198 | MirrorBlockJob *s = op->s; | ||
199 | + | ||
200 | + aio_context_acquire(blk_get_aio_context(s->common.blk)); | ||
201 | if (ret < 0) { | ||
202 | BlockErrorAction action; | ||
203 | |||
204 | @@ -XXX,XX +XXX,XX @@ static void mirror_read_complete(void *opaque, int ret) | ||
205 | } | ||
206 | |||
207 | mirror_iteration_done(op, ret); | ||
208 | - return; | ||
209 | + } else { | ||
210 | + blk_aio_pwritev(s->target, op->sector_num * BDRV_SECTOR_SIZE, &op->qiov, | ||
211 | + 0, mirror_write_complete, op); | ||
212 | } | ||
213 | - blk_aio_pwritev(s->target, op->sector_num * BDRV_SECTOR_SIZE, &op->qiov, | ||
214 | - 0, mirror_write_complete, op); | ||
215 | + aio_context_release(blk_get_aio_context(s->common.blk)); | ||
216 | } | ||
217 | |||
218 | static inline void mirror_clip_sectors(MirrorBlockJob *s, | ||
219 | diff --git a/block/null.c b/block/null.c | ||
220 | index XXXXXXX..XXXXXXX 100644 | ||
221 | --- a/block/null.c | ||
222 | +++ b/block/null.c | ||
223 | @@ -XXX,XX +XXX,XX @@ static const AIOCBInfo null_aiocb_info = { | ||
224 | static void null_bh_cb(void *opaque) | ||
225 | { | ||
226 | NullAIOCB *acb = opaque; | ||
227 | - AioContext *ctx = bdrv_get_aio_context(acb->common.bs); | ||
228 | - | ||
229 | - aio_context_acquire(ctx); | ||
230 | acb->common.cb(acb->common.opaque, 0); | ||
231 | - aio_context_release(ctx); | ||
232 | qemu_aio_unref(acb); | ||
233 | } | ||
234 | |||
235 | static void null_timer_cb(void *opaque) | ||
236 | { | ||
237 | NullAIOCB *acb = opaque; | ||
238 | - AioContext *ctx = bdrv_get_aio_context(acb->common.bs); | ||
239 | - | ||
240 | - aio_context_acquire(ctx); | ||
241 | acb->common.cb(acb->common.opaque, 0); | ||
242 | - aio_context_release(ctx); | ||
243 | timer_deinit(&acb->timer); | ||
244 | qemu_aio_unref(acb); | ||
245 | } | ||
246 | diff --git a/block/qed-cluster.c b/block/qed-cluster.c | ||
247 | index XXXXXXX..XXXXXXX 100644 | ||
248 | --- a/block/qed-cluster.c | ||
249 | +++ b/block/qed-cluster.c | ||
250 | @@ -XXX,XX +XXX,XX @@ static void qed_find_cluster_cb(void *opaque, int ret) | ||
251 | unsigned int index; | ||
252 | unsigned int n; | ||
253 | |||
254 | + qed_acquire(s); | ||
255 | if (ret) { | ||
256 | goto out; | ||
257 | } | ||
258 | @@ -XXX,XX +XXX,XX @@ static void qed_find_cluster_cb(void *opaque, int ret) | ||
259 | |||
260 | out: | ||
261 | find_cluster_cb->cb(find_cluster_cb->opaque, ret, offset, len); | ||
262 | + qed_release(s); | ||
263 | g_free(find_cluster_cb); | ||
264 | } | ||
265 | |||
266 | diff --git a/block/qed-table.c b/block/qed-table.c | ||
267 | index XXXXXXX..XXXXXXX 100644 | ||
268 | --- a/block/qed-table.c | ||
269 | +++ b/block/qed-table.c | ||
270 | @@ -XXX,XX +XXX,XX @@ static void qed_read_table_cb(void *opaque, int ret) | ||
271 | { | ||
272 | QEDReadTableCB *read_table_cb = opaque; | ||
273 | QEDTable *table = read_table_cb->table; | ||
274 | + BDRVQEDState *s = read_table_cb->s; | ||
275 | int noffsets = read_table_cb->qiov.size / sizeof(uint64_t); | ||
276 | int i; | ||
277 | |||
278 | @@ -XXX,XX +XXX,XX @@ static void qed_read_table_cb(void *opaque, int ret) | ||
279 | } | ||
280 | |||
281 | /* Byteswap offsets */ | ||
282 | + qed_acquire(s); | ||
283 | for (i = 0; i < noffsets; i++) { | ||
284 | table->offsets[i] = le64_to_cpu(table->offsets[i]); | ||
285 | } | ||
286 | + qed_release(s); | ||
287 | |||
288 | out: | ||
289 | /* Completion */ | ||
290 | - trace_qed_read_table_cb(read_table_cb->s, read_table_cb->table, ret); | ||
291 | + trace_qed_read_table_cb(s, read_table_cb->table, ret); | ||
292 | gencb_complete(&read_table_cb->gencb, ret); | ||
293 | } | ||
294 | |||
295 | @@ -XXX,XX +XXX,XX @@ typedef struct { | ||
296 | static void qed_write_table_cb(void *opaque, int ret) | ||
297 | { | ||
298 | QEDWriteTableCB *write_table_cb = opaque; | ||
299 | + BDRVQEDState *s = write_table_cb->s; | ||
300 | |||
301 | - trace_qed_write_table_cb(write_table_cb->s, | ||
302 | + trace_qed_write_table_cb(s, | ||
303 | write_table_cb->orig_table, | ||
304 | write_table_cb->flush, | ||
305 | ret); | ||
306 | @@ -XXX,XX +XXX,XX @@ static void qed_write_table_cb(void *opaque, int ret) | ||
307 | if (write_table_cb->flush) { | ||
308 | /* We still need to flush first */ | ||
309 | write_table_cb->flush = false; | ||
310 | + qed_acquire(s); | ||
311 | bdrv_aio_flush(write_table_cb->s->bs, qed_write_table_cb, | ||
312 | write_table_cb); | ||
313 | + qed_release(s); | ||
314 | return; | ||
315 | } | ||
316 | |||
317 | @@ -XXX,XX +XXX,XX @@ static void qed_read_l2_table_cb(void *opaque, int ret) | ||
318 | CachedL2Table *l2_table = request->l2_table; | ||
319 | uint64_t l2_offset = read_l2_table_cb->l2_offset; | ||
320 | |||
321 | + qed_acquire(s); | ||
322 | if (ret) { | ||
323 | /* can't trust loaded L2 table anymore */ | ||
324 | qed_unref_l2_cache_entry(l2_table); | ||
325 | @@ -XXX,XX +XXX,XX @@ static void qed_read_l2_table_cb(void *opaque, int ret) | ||
326 | request->l2_table = qed_find_l2_cache_entry(&s->l2_cache, l2_offset); | ||
327 | assert(request->l2_table != NULL); | ||
328 | } | ||
329 | + qed_release(s); | ||
330 | |||
331 | gencb_complete(&read_l2_table_cb->gencb, ret); | ||
332 | } | ||
333 | diff --git a/block/qed.c b/block/qed.c | ||
334 | index XXXXXXX..XXXXXXX 100644 | ||
335 | --- a/block/qed.c | ||
336 | +++ b/block/qed.c | ||
337 | @@ -XXX,XX +XXX,XX @@ static void qed_is_allocated_cb(void *opaque, int ret, uint64_t offset, size_t l | ||
338 | } | ||
339 | |||
340 | if (cb->co) { | ||
341 | - qemu_coroutine_enter(cb->co); | ||
342 | + aio_co_wake(cb->co); | ||
343 | } | ||
344 | } | ||
345 | |||
346 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn qed_co_pwrite_zeroes_cb(void *opaque, int ret) | ||
347 | cb->done = true; | ||
348 | cb->ret = ret; | ||
349 | if (cb->co) { | ||
350 | - qemu_coroutine_enter(cb->co); | ||
351 | + aio_co_wake(cb->co); | ||
352 | } | ||
353 | } | ||
354 | |||
31 | diff --git a/block/rbd.c b/block/rbd.c | 355 | diff --git a/block/rbd.c b/block/rbd.c |
32 | index XXXXXXX..XXXXXXX 100644 | 356 | index XXXXXXX..XXXXXXX 100644 |
33 | --- a/block/rbd.c | 357 | --- a/block/rbd.c |
34 | +++ b/block/rbd.c | 358 | +++ b/block/rbd.c |
35 | @@ -XXX,XX +XXX,XX @@ static int qemu_rbd_connect(rados_t *cluster, rados_ioctx_t *io_ctx, | 359 | @@ -XXX,XX +XXX,XX @@ shutdown: |
36 | const char *keypairs, const char *secretid, | 360 | static void qemu_rbd_complete_aio(RADOSCB *rcb) |
37 | Error **errp); | 361 | { |
38 | 362 | RBDAIOCB *acb = rcb->acb; | |
39 | +static char *qemu_rbd_strchr(char *src, char delim) | 363 | - AioContext *ctx = bdrv_get_aio_context(acb->common.bs); |
40 | +{ | 364 | int64_t r; |
41 | + char *p; | 365 | |
42 | + | 366 | r = rcb->ret; |
43 | + for (p = src; *p; ++p) { | 367 | @@ -XXX,XX +XXX,XX @@ static void qemu_rbd_complete_aio(RADOSCB *rcb) |
44 | + if (*p == delim) { | 368 | qemu_iovec_from_buf(acb->qiov, 0, acb->bounce, acb->qiov->size); |
45 | + return p; | 369 | } |
46 | + } | 370 | qemu_vfree(acb->bounce); |
47 | + if (*p == '\\' && p[1] != '\0') { | 371 | - |
48 | + ++p; | 372 | - aio_context_acquire(ctx); |
49 | + } | 373 | acb->common.cb(acb->common.opaque, (acb->ret > 0 ? 0 : acb->ret)); |
50 | + } | 374 | - aio_context_release(ctx); |
51 | + | 375 | |
52 | + return NULL; | 376 | qemu_aio_unref(acb); |
53 | +} | 377 | } |
54 | + | 378 | diff --git a/block/win32-aio.c b/block/win32-aio.c |
55 | + | 379 | index XXXXXXX..XXXXXXX 100644 |
56 | static char *qemu_rbd_next_tok(char *src, char delim, char **p) | 380 | --- a/block/win32-aio.c |
57 | { | 381 | +++ b/block/win32-aio.c |
58 | char *end; | 382 | @@ -XXX,XX +XXX,XX @@ static void win32_aio_process_completion(QEMUWin32AIOState *s, |
59 | 383 | qemu_vfree(waiocb->buf); | |
60 | *p = NULL; | 384 | } |
61 | 385 | ||
62 | - for (end = src; *end; ++end) { | 386 | - |
63 | - if (*end == delim) { | 387 | - aio_context_acquire(s->aio_ctx); |
64 | - break; | 388 | waiocb->common.cb(waiocb->common.opaque, ret); |
65 | - } | 389 | - aio_context_release(s->aio_ctx); |
66 | - if (*end == '\\' && end[1] != '\0') { | 390 | qemu_aio_unref(waiocb); |
67 | - end++; | 391 | } |
68 | - } | 392 | |
69 | - } | 393 | diff --git a/hw/block/virtio-blk.c b/hw/block/virtio-blk.c |
70 | - if (*end == delim) { | 394 | index XXXXXXX..XXXXXXX 100644 |
71 | + end = qemu_rbd_strchr(src, delim); | 395 | --- a/hw/block/virtio-blk.c |
72 | + if (end) { | 396 | +++ b/hw/block/virtio-blk.c |
73 | *p = end + 1; | 397 | @@ -XXX,XX +XXX,XX @@ static int virtio_blk_handle_rw_error(VirtIOBlockReq *req, int error, |
74 | *end = '\0'; | 398 | static void virtio_blk_rw_complete(void *opaque, int ret) |
75 | } | 399 | { |
76 | @@ -XXX,XX +XXX,XX @@ static void qemu_rbd_parse_filename(const char *filename, QDict *options, | 400 | VirtIOBlockReq *next = opaque; |
77 | qemu_rbd_unescape(found_str); | 401 | + VirtIOBlock *s = next->dev; |
78 | qdict_put_str(options, "pool", found_str); | 402 | |
79 | 403 | + aio_context_acquire(blk_get_aio_context(s->conf.conf.blk)); | |
80 | - if (strchr(p, '@')) { | 404 | while (next) { |
81 | + if (qemu_rbd_strchr(p, '@')) { | 405 | VirtIOBlockReq *req = next; |
82 | image_name = qemu_rbd_next_tok(p, '@', &p); | 406 | next = req->mr_next; |
83 | 407 | @@ -XXX,XX +XXX,XX @@ static void virtio_blk_rw_complete(void *opaque, int ret) | |
84 | found_str = qemu_rbd_next_tok(p, ':', &p); | 408 | block_acct_done(blk_get_stats(req->dev->blk), &req->acct); |
85 | @@ -XXX,XX +XXX,XX @@ static void qemu_rbd_parse_filename(const char *filename, QDict *options, | 409 | virtio_blk_free_request(req); |
86 | image_name = qemu_rbd_next_tok(p, ':', &p); | 410 | } |
87 | } | 411 | + aio_context_release(blk_get_aio_context(s->conf.conf.blk)); |
88 | /* Check for namespace in the image_name */ | 412 | } |
89 | - if (strchr(image_name, '/')) { | 413 | |
90 | + if (qemu_rbd_strchr(image_name, '/')) { | 414 | static void virtio_blk_flush_complete(void *opaque, int ret) |
91 | found_str = qemu_rbd_next_tok(image_name, '/', &image_name); | 415 | { |
92 | qemu_rbd_unescape(found_str); | 416 | VirtIOBlockReq *req = opaque; |
93 | qdict_put_str(options, "namespace", found_str); | 417 | + VirtIOBlock *s = req->dev; |
94 | diff --git a/tests/qemu-iotests/231 b/tests/qemu-iotests/231 | 418 | |
95 | index XXXXXXX..XXXXXXX 100755 | 419 | + aio_context_acquire(blk_get_aio_context(s->conf.conf.blk)); |
96 | --- a/tests/qemu-iotests/231 | 420 | if (ret) { |
97 | +++ b/tests/qemu-iotests/231 | 421 | if (virtio_blk_handle_rw_error(req, -ret, 0)) { |
98 | @@ -XXX,XX +XXX,XX @@ _filter_conf() | 422 | - return; |
99 | $QEMU_IMG info "json:{'file.driver':'rbd','file.filename':'rbd:rbd/bogus:conf=${BOGUS_CONF}'}" 2>&1 | _filter_conf | 423 | + goto out; |
100 | $QEMU_IMG info "json:{'file.driver':'rbd','file.pool':'rbd','file.image':'bogus','file.conf':'${BOGUS_CONF}'}" 2>&1 | _filter_conf | 424 | } |
101 | 425 | } | |
102 | +# Regression test: the qemu-img invocation is expected to fail, but it should | 426 | |
103 | +# not seg fault the parser. | 427 | virtio_blk_req_complete(req, VIRTIO_BLK_S_OK); |
104 | +$QEMU_IMG create "rbd:rbd/aa\/bb:conf=${BOGUS_CONF}" 1M 2>&1 | _filter_conf | 428 | block_acct_done(blk_get_stats(req->dev->blk), &req->acct); |
105 | + | 429 | virtio_blk_free_request(req); |
106 | # success, all done | 430 | + |
107 | echo "*** done" | 431 | +out: |
108 | rm -f $seq.full | 432 | + aio_context_release(blk_get_aio_context(s->conf.conf.blk)); |
109 | diff --git a/tests/qemu-iotests/231.out b/tests/qemu-iotests/231.out | 433 | } |
110 | index XXXXXXX..XXXXXXX 100644 | 434 | |
111 | --- a/tests/qemu-iotests/231.out | 435 | #ifdef __linux__ |
112 | +++ b/tests/qemu-iotests/231.out | 436 | @@ -XXX,XX +XXX,XX @@ static void virtio_blk_ioctl_complete(void *opaque, int status) |
113 | @@ -XXX,XX +XXX,XX @@ unable to get monitor info from DNS SRV with service name: ceph-mon | 437 | virtio_stl_p(vdev, &scsi->data_len, hdr->dxfer_len); |
114 | qemu-img: Could not open 'json:{'file.driver':'rbd','file.filename':'rbd:rbd/bogus:conf=BOGUS_CONF'}': error connecting: No such file or directory | 438 | |
115 | unable to get monitor info from DNS SRV with service name: ceph-mon | 439 | out: |
116 | qemu-img: Could not open 'json:{'file.driver':'rbd','file.pool':'rbd','file.image':'bogus','file.conf':'BOGUS_CONF'}': error connecting: No such file or directory | 440 | + aio_context_acquire(blk_get_aio_context(s->conf.conf.blk)); |
117 | +Formatting 'rbd:rbd/aa\/bb:conf=BOGUS_CONF', fmt=raw size=1048576 | 441 | virtio_blk_req_complete(req, status); |
118 | +unable to get monitor info from DNS SRV with service name: ceph-mon | 442 | virtio_blk_free_request(req); |
119 | +qemu-img: rbd:rbd/aa\/bb:conf=BOGUS_CONF: error connecting: No such file or directory | 443 | + aio_context_release(blk_get_aio_context(s->conf.conf.blk)); |
120 | *** done | 444 | g_free(ioctl_req); |
445 | } | ||
446 | |||
447 | diff --git a/hw/scsi/scsi-disk.c b/hw/scsi/scsi-disk.c | ||
448 | index XXXXXXX..XXXXXXX 100644 | ||
449 | --- a/hw/scsi/scsi-disk.c | ||
450 | +++ b/hw/scsi/scsi-disk.c | ||
451 | @@ -XXX,XX +XXX,XX @@ static void scsi_aio_complete(void *opaque, int ret) | ||
452 | |||
453 | assert(r->req.aiocb != NULL); | ||
454 | r->req.aiocb = NULL; | ||
455 | + aio_context_acquire(blk_get_aio_context(s->qdev.conf.blk)); | ||
456 | if (scsi_disk_req_check_error(r, ret, true)) { | ||
457 | goto done; | ||
458 | } | ||
459 | @@ -XXX,XX +XXX,XX @@ static void scsi_aio_complete(void *opaque, int ret) | ||
460 | scsi_req_complete(&r->req, GOOD); | ||
461 | |||
462 | done: | ||
463 | + aio_context_release(blk_get_aio_context(s->qdev.conf.blk)); | ||
464 | scsi_req_unref(&r->req); | ||
465 | } | ||
466 | |||
467 | @@ -XXX,XX +XXX,XX @@ static void scsi_dma_complete(void *opaque, int ret) | ||
468 | assert(r->req.aiocb != NULL); | ||
469 | r->req.aiocb = NULL; | ||
470 | |||
471 | + aio_context_acquire(blk_get_aio_context(s->qdev.conf.blk)); | ||
472 | if (ret < 0) { | ||
473 | block_acct_failed(blk_get_stats(s->qdev.conf.blk), &r->acct); | ||
474 | } else { | ||
475 | block_acct_done(blk_get_stats(s->qdev.conf.blk), &r->acct); | ||
476 | } | ||
477 | scsi_dma_complete_noio(r, ret); | ||
478 | + aio_context_release(blk_get_aio_context(s->qdev.conf.blk)); | ||
479 | } | ||
480 | |||
481 | static void scsi_read_complete(void * opaque, int ret) | ||
482 | @@ -XXX,XX +XXX,XX @@ static void scsi_read_complete(void * opaque, int ret) | ||
483 | |||
484 | assert(r->req.aiocb != NULL); | ||
485 | r->req.aiocb = NULL; | ||
486 | + aio_context_acquire(blk_get_aio_context(s->qdev.conf.blk)); | ||
487 | if (scsi_disk_req_check_error(r, ret, true)) { | ||
488 | goto done; | ||
489 | } | ||
490 | @@ -XXX,XX +XXX,XX @@ static void scsi_read_complete(void * opaque, int ret) | ||
491 | |||
492 | done: | ||
493 | scsi_req_unref(&r->req); | ||
494 | + aio_context_release(blk_get_aio_context(s->qdev.conf.blk)); | ||
495 | } | ||
496 | |||
497 | /* Actually issue a read to the block device. */ | ||
498 | @@ -XXX,XX +XXX,XX @@ static void scsi_do_read_cb(void *opaque, int ret) | ||
499 | assert (r->req.aiocb != NULL); | ||
500 | r->req.aiocb = NULL; | ||
501 | |||
502 | + aio_context_acquire(blk_get_aio_context(s->qdev.conf.blk)); | ||
503 | if (ret < 0) { | ||
504 | block_acct_failed(blk_get_stats(s->qdev.conf.blk), &r->acct); | ||
505 | } else { | ||
506 | block_acct_done(blk_get_stats(s->qdev.conf.blk), &r->acct); | ||
507 | } | ||
508 | scsi_do_read(opaque, ret); | ||
509 | + aio_context_release(blk_get_aio_context(s->qdev.conf.blk)); | ||
510 | } | ||
511 | |||
512 | /* Read more data from scsi device into buffer. */ | ||
513 | @@ -XXX,XX +XXX,XX @@ static void scsi_write_complete(void * opaque, int ret) | ||
514 | assert (r->req.aiocb != NULL); | ||
515 | r->req.aiocb = NULL; | ||
516 | |||
517 | + aio_context_acquire(blk_get_aio_context(s->qdev.conf.blk)); | ||
518 | if (ret < 0) { | ||
519 | block_acct_failed(blk_get_stats(s->qdev.conf.blk), &r->acct); | ||
520 | } else { | ||
521 | block_acct_done(blk_get_stats(s->qdev.conf.blk), &r->acct); | ||
522 | } | ||
523 | scsi_write_complete_noio(r, ret); | ||
524 | + aio_context_release(blk_get_aio_context(s->qdev.conf.blk)); | ||
525 | } | ||
526 | |||
527 | static void scsi_write_data(SCSIRequest *req) | ||
528 | @@ -XXX,XX +XXX,XX @@ static void scsi_unmap_complete(void *opaque, int ret) | ||
529 | { | ||
530 | UnmapCBData *data = opaque; | ||
531 | SCSIDiskReq *r = data->r; | ||
532 | + SCSIDiskState *s = DO_UPCAST(SCSIDiskState, qdev, r->req.dev); | ||
533 | |||
534 | assert(r->req.aiocb != NULL); | ||
535 | r->req.aiocb = NULL; | ||
536 | |||
537 | + aio_context_acquire(blk_get_aio_context(s->qdev.conf.blk)); | ||
538 | scsi_unmap_complete_noio(data, ret); | ||
539 | + aio_context_release(blk_get_aio_context(s->qdev.conf.blk)); | ||
540 | } | ||
541 | |||
542 | static void scsi_disk_emulate_unmap(SCSIDiskReq *r, uint8_t *inbuf) | ||
543 | @@ -XXX,XX +XXX,XX @@ static void scsi_write_same_complete(void *opaque, int ret) | ||
544 | |||
545 | assert(r->req.aiocb != NULL); | ||
546 | r->req.aiocb = NULL; | ||
547 | + aio_context_acquire(blk_get_aio_context(s->qdev.conf.blk)); | ||
548 | if (scsi_disk_req_check_error(r, ret, true)) { | ||
549 | goto done; | ||
550 | } | ||
551 | @@ -XXX,XX +XXX,XX @@ done: | ||
552 | scsi_req_unref(&r->req); | ||
553 | qemu_vfree(data->iov.iov_base); | ||
554 | g_free(data); | ||
555 | + aio_context_release(blk_get_aio_context(s->qdev.conf.blk)); | ||
556 | } | ||
557 | |||
558 | static void scsi_disk_emulate_write_same(SCSIDiskReq *r, uint8_t *inbuf) | ||
559 | diff --git a/hw/scsi/scsi-generic.c b/hw/scsi/scsi-generic.c | ||
560 | index XXXXXXX..XXXXXXX 100644 | ||
561 | --- a/hw/scsi/scsi-generic.c | ||
562 | +++ b/hw/scsi/scsi-generic.c | ||
563 | @@ -XXX,XX +XXX,XX @@ done: | ||
564 | static void scsi_command_complete(void *opaque, int ret) | ||
565 | { | ||
566 | SCSIGenericReq *r = (SCSIGenericReq *)opaque; | ||
567 | + SCSIDevice *s = r->req.dev; | ||
568 | |||
569 | assert(r->req.aiocb != NULL); | ||
570 | r->req.aiocb = NULL; | ||
571 | + | ||
572 | + aio_context_acquire(blk_get_aio_context(s->conf.blk)); | ||
573 | scsi_command_complete_noio(r, ret); | ||
574 | + aio_context_release(blk_get_aio_context(s->conf.blk)); | ||
575 | } | ||
576 | |||
577 | static int execute_command(BlockBackend *blk, | ||
578 | @@ -XXX,XX +XXX,XX @@ static void scsi_read_complete(void * opaque, int ret) | ||
579 | assert(r->req.aiocb != NULL); | ||
580 | r->req.aiocb = NULL; | ||
581 | |||
582 | + aio_context_acquire(blk_get_aio_context(s->conf.blk)); | ||
583 | + | ||
584 | if (ret || r->req.io_canceled) { | ||
585 | scsi_command_complete_noio(r, ret); | ||
586 | - return; | ||
587 | + goto done; | ||
588 | } | ||
589 | |||
590 | len = r->io_header.dxfer_len - r->io_header.resid; | ||
591 | @@ -XXX,XX +XXX,XX @@ static void scsi_read_complete(void * opaque, int ret) | ||
592 | r->len = -1; | ||
593 | if (len == 0) { | ||
594 | scsi_command_complete_noio(r, 0); | ||
595 | - return; | ||
596 | + goto done; | ||
597 | } | ||
598 | |||
599 | /* Snoop READ CAPACITY output to set the blocksize. */ | ||
600 | @@ -XXX,XX +XXX,XX @@ static void scsi_read_complete(void * opaque, int ret) | ||
601 | } | ||
602 | scsi_req_data(&r->req, len); | ||
603 | scsi_req_unref(&r->req); | ||
604 | + | ||
605 | +done: | ||
606 | + aio_context_release(blk_get_aio_context(s->conf.blk)); | ||
607 | } | ||
608 | |||
609 | /* Read more data from scsi device into buffer. */ | ||
610 | @@ -XXX,XX +XXX,XX @@ static void scsi_write_complete(void * opaque, int ret) | ||
611 | assert(r->req.aiocb != NULL); | ||
612 | r->req.aiocb = NULL; | ||
613 | |||
614 | + aio_context_acquire(blk_get_aio_context(s->conf.blk)); | ||
615 | + | ||
616 | if (ret || r->req.io_canceled) { | ||
617 | scsi_command_complete_noio(r, ret); | ||
618 | - return; | ||
619 | + goto done; | ||
620 | } | ||
621 | |||
622 | if (r->req.cmd.buf[0] == MODE_SELECT && r->req.cmd.buf[4] == 12 && | ||
623 | @@ -XXX,XX +XXX,XX @@ static void scsi_write_complete(void * opaque, int ret) | ||
624 | } | ||
625 | |||
626 | scsi_command_complete_noio(r, ret); | ||
627 | + | ||
628 | +done: | ||
629 | + aio_context_release(blk_get_aio_context(s->conf.blk)); | ||
630 | } | ||
631 | |||
632 | /* Write data to a scsi device. Returns nonzero on failure. | ||
633 | diff --git a/util/thread-pool.c b/util/thread-pool.c | ||
634 | index XXXXXXX..XXXXXXX 100644 | ||
635 | --- a/util/thread-pool.c | ||
636 | +++ b/util/thread-pool.c | ||
637 | @@ -XXX,XX +XXX,XX @@ restart: | ||
638 | */ | ||
639 | qemu_bh_schedule(pool->completion_bh); | ||
640 | |||
641 | + aio_context_release(pool->ctx); | ||
642 | elem->common.cb(elem->common.opaque, elem->ret); | ||
643 | + aio_context_acquire(pool->ctx); | ||
644 | qemu_aio_unref(elem); | ||
645 | goto restart; | ||
646 | } else { | ||
647 | @@ -XXX,XX +XXX,XX @@ static void thread_pool_co_cb(void *opaque, int ret) | ||
648 | ThreadPoolCo *co = opaque; | ||
649 | |||
650 | co->ret = ret; | ||
651 | - qemu_coroutine_enter(co->co); | ||
652 | + aio_co_wake(co->co); | ||
653 | } | ||
654 | |||
655 | int coroutine_fn thread_pool_submit_co(ThreadPool *pool, ThreadPoolFunc *func, | ||
121 | -- | 656 | -- |
122 | 2.31.1 | 657 | 2.9.3 |
123 | 658 | ||
124 | 659 | diff view generated by jsdifflib |
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | Max reported the following bug: | 3 | This patch prepares for the removal of unnecessary lockcnt inc/dec pairs. |
4 | Extract the dispatching loop for file descriptor handlers into a new | ||
5 | function aio_dispatch_handlers, and then inline aio_dispatch into | ||
6 | aio_poll. | ||
4 | 7 | ||
5 | $ ./qemu-img create -f raw src.img 1G | 8 | aio_dispatch can now become void. |
6 | $ ./qemu-img create -f raw dst.img 1G | ||
7 | 9 | ||
8 | $ (echo ' | 10 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
9 | {"execute":"qmp_capabilities"} | 11 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
10 | {"execute":"blockdev-mirror", | 12 | Reviewed-by: Fam Zheng <famz@redhat.com> |
11 | "arguments":{"job-id":"mirror", | 13 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> |
12 | "device":"source", | 14 | Message-id: 20170213135235.12274-17-pbonzini@redhat.com |
13 | "target":"target", | 15 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
14 | "sync":"full", | 16 | --- |
15 | "filter-node-name":"mirror-top"}} | 17 | include/block/aio.h | 6 +----- |
16 | '; sleep 3; echo ' | 18 | util/aio-posix.c | 44 ++++++++++++++------------------------------ |
17 | {"execute":"human-monitor-command", | 19 | util/aio-win32.c | 13 ++++--------- |
18 | "arguments":{"command-line": | 20 | util/async.c | 2 +- |
19 | "qemu-io mirror-top \"write 0 1G\""}}') \ | 21 | 4 files changed, 20 insertions(+), 45 deletions(-) |
20 | | x86_64-softmmu/qemu-system-x86_64 \ | ||
21 | -qmp stdio \ | ||
22 | -blockdev file,node-name=source,filename=src.img \ | ||
23 | -blockdev file,node-name=target,filename=dst.img \ | ||
24 | -object iothread,id=iothr0 \ | ||
25 | -device virtio-blk,drive=source,iothread=iothr0 | ||
26 | 22 | ||
27 | crashes: | 23 | diff --git a/include/block/aio.h b/include/block/aio.h |
28 | |||
29 | 0 raise () at /usr/lib/libc.so.6 | ||
30 | 1 abort () at /usr/lib/libc.so.6 | ||
31 | 2 error_exit | ||
32 | (err=<optimized out>, | ||
33 | msg=msg@entry=0x55fbb1634790 <__func__.27> "qemu_mutex_unlock_impl") | ||
34 | at ../util/qemu-thread-posix.c:37 | ||
35 | 3 qemu_mutex_unlock_impl | ||
36 | (mutex=mutex@entry=0x55fbb25ab6e0, | ||
37 | file=file@entry=0x55fbb1636957 "../util/async.c", | ||
38 | line=line@entry=650) | ||
39 | at ../util/qemu-thread-posix.c:109 | ||
40 | 4 aio_context_release (ctx=ctx@entry=0x55fbb25ab680) at ../util/async.c:650 | ||
41 | 5 bdrv_do_drained_begin | ||
42 | (bs=bs@entry=0x55fbb3a87000, recursive=recursive@entry=false, | ||
43 | parent=parent@entry=0x0, | ||
44 | ignore_bds_parents=ignore_bds_parents@entry=false, | ||
45 | poll=poll@entry=true) at ../block/io.c:441 | ||
46 | 6 bdrv_do_drained_begin | ||
47 | (poll=true, ignore_bds_parents=false, parent=0x0, recursive=false, | ||
48 | bs=0x55fbb3a87000) at ../block/io.c:448 | ||
49 | 7 blk_drain (blk=0x55fbb26c5a00) at ../block/block-backend.c:1718 | ||
50 | 8 blk_unref (blk=0x55fbb26c5a00) at ../block/block-backend.c:498 | ||
51 | 9 blk_unref (blk=0x55fbb26c5a00) at ../block/block-backend.c:491 | ||
52 | 10 hmp_qemu_io (mon=0x7fffaf3fc7d0, qdict=<optimized out>) | ||
53 | at ../block/monitor/block-hmp-cmds.c:628 | ||
54 | |||
55 | man pthread_mutex_unlock | ||
56 | ... | ||
57 | EPERM The mutex type is PTHREAD_MUTEX_ERRORCHECK or | ||
58 | PTHREAD_MUTEX_RECURSIVE, or the mutex is a robust mutex, and the | ||
59 | current thread does not own the mutex. | ||
60 | |||
61 | So, thread doesn't own the mutex. And we have iothread here. | ||
62 | |||
63 | Next, note that AIO_WAIT_WHILE() documents that ctx must be acquired | ||
64 | exactly once by caller. But where is it acquired in the call stack? | ||
65 | Seems nowhere. | ||
66 | |||
67 | qemuio_command do acquire aio context.. But we need context acquired | ||
68 | around blk_unref() as well and actually around blk_insert_bs() too. | ||
69 | |||
70 | Let's refactor qemuio_command so that it doesn't acquire aio context | ||
71 | but callers do that instead. This way we can cleanly acquire aio | ||
72 | context in hmp_qemu_io() around all three calls. | ||
73 | |||
74 | Reported-by: Max Reitz <mreitz@redhat.com> | ||
75 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
76 | Message-Id: <20210423134233.51495-1-vsementsov@virtuozzo.com> | ||
77 | [mreitz: Fixed comment] | ||
78 | Signed-off-by: Max Reitz <mreitz@redhat.com> | ||
79 | --- | ||
80 | block/monitor/block-hmp-cmds.c | 31 +++++++++++++++++++++---------- | ||
81 | qemu-io-cmds.c | 8 ++++---- | ||
82 | qemu-io.c | 17 +++++++++++++++-- | ||
83 | 3 files changed, 40 insertions(+), 16 deletions(-) | ||
84 | |||
85 | diff --git a/block/monitor/block-hmp-cmds.c b/block/monitor/block-hmp-cmds.c | ||
86 | index XXXXXXX..XXXXXXX 100644 | 24 | index XXXXXXX..XXXXXXX 100644 |
87 | --- a/block/monitor/block-hmp-cmds.c | 25 | --- a/include/block/aio.h |
88 | +++ b/block/monitor/block-hmp-cmds.c | 26 | +++ b/include/block/aio.h |
89 | @@ -XXX,XX +XXX,XX @@ void hmp_eject(Monitor *mon, const QDict *qdict) | 27 | @@ -XXX,XX +XXX,XX @@ bool aio_pending(AioContext *ctx); |
90 | 28 | /* Dispatch any pending callbacks from the GSource attached to the AioContext. | |
91 | void hmp_qemu_io(Monitor *mon, const QDict *qdict) | 29 | * |
92 | { | 30 | * This is used internally in the implementation of the GSource. |
93 | - BlockBackend *blk; | 31 | - * |
94 | + BlockBackend *blk = NULL; | 32 | - * @dispatch_fds: true to process fds, false to skip them |
95 | + BlockDriverState *bs = NULL; | 33 | - * (can be used as an optimization by callers that know there |
96 | BlockBackend *local_blk = NULL; | 34 | - * are no fds ready) |
97 | + AioContext *ctx = NULL; | 35 | */ |
98 | bool qdev = qdict_get_try_bool(qdict, "qdev", false); | 36 | -bool aio_dispatch(AioContext *ctx, bool dispatch_fds); |
99 | const char *device = qdict_get_str(qdict, "device"); | 37 | +void aio_dispatch(AioContext *ctx); |
100 | const char *command = qdict_get_str(qdict, "command"); | 38 | |
101 | @@ -XXX,XX +XXX,XX @@ void hmp_qemu_io(Monitor *mon, const QDict *qdict) | 39 | /* Progress in completing AIO work to occur. This can issue new pending |
102 | } else { | 40 | * aio as a result of executing I/O completion or bh callbacks. |
103 | blk = blk_by_name(device); | 41 | diff --git a/util/aio-posix.c b/util/aio-posix.c |
104 | if (!blk) { | 42 | index XXXXXXX..XXXXXXX 100644 |
105 | - BlockDriverState *bs = bdrv_lookup_bs(NULL, device, &err); | 43 | --- a/util/aio-posix.c |
106 | - if (bs) { | 44 | +++ b/util/aio-posix.c |
107 | - blk = local_blk = blk_new(bdrv_get_aio_context(bs), | 45 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx) |
108 | - 0, BLK_PERM_ALL); | 46 | AioHandler *node, *tmp; |
109 | - ret = blk_insert_bs(blk, bs, &err); | 47 | bool progress = false; |
110 | - if (ret < 0) { | 48 | |
111 | - goto fail; | 49 | - /* |
112 | - } | 50 | - * We have to walk very carefully in case aio_set_fd_handler is |
113 | - } else { | 51 | - * called while we're walking. |
114 | + bs = bdrv_lookup_bs(NULL, device, &err); | 52 | - */ |
115 | + if (!bs) { | 53 | - qemu_lockcnt_inc(&ctx->list_lock); |
116 | goto fail; | 54 | - |
117 | } | 55 | QLIST_FOREACH_SAFE_RCU(node, &ctx->aio_handlers, node, tmp) { |
56 | int revents; | ||
57 | |||
58 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx) | ||
118 | } | 59 | } |
119 | } | 60 | } |
120 | 61 | ||
121 | + ctx = blk ? blk_get_aio_context(blk) : bdrv_get_aio_context(bs); | 62 | - qemu_lockcnt_dec(&ctx->list_lock); |
122 | + aio_context_acquire(ctx); | 63 | return progress; |
64 | } | ||
65 | |||
66 | -/* | ||
67 | - * Note that dispatch_fds == false has the side-effect of post-poning the | ||
68 | - * freeing of deleted handlers. | ||
69 | - */ | ||
70 | -bool aio_dispatch(AioContext *ctx, bool dispatch_fds) | ||
71 | +void aio_dispatch(AioContext *ctx) | ||
72 | { | ||
73 | - bool progress; | ||
74 | + aio_bh_poll(ctx); | ||
75 | |||
76 | - /* | ||
77 | - * If there are callbacks left that have been queued, we need to call them. | ||
78 | - * Do not call select in this case, because it is possible that the caller | ||
79 | - * does not need a complete flush (as is the case for aio_poll loops). | ||
80 | - */ | ||
81 | - progress = aio_bh_poll(ctx); | ||
82 | + qemu_lockcnt_inc(&ctx->list_lock); | ||
83 | + aio_dispatch_handlers(ctx); | ||
84 | + qemu_lockcnt_dec(&ctx->list_lock); | ||
85 | |||
86 | - if (dispatch_fds) { | ||
87 | - progress |= aio_dispatch_handlers(ctx); | ||
88 | - } | ||
89 | - | ||
90 | - /* Run our timers */ | ||
91 | - progress |= timerlistgroup_run_timers(&ctx->tlg); | ||
92 | - | ||
93 | - return progress; | ||
94 | + timerlistgroup_run_timers(&ctx->tlg); | ||
95 | } | ||
96 | |||
97 | /* These thread-local variables are used only in a small part of aio_poll | ||
98 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
99 | npfd = 0; | ||
100 | qemu_lockcnt_dec(&ctx->list_lock); | ||
101 | |||
102 | - /* Run dispatch even if there were no readable fds to run timers */ | ||
103 | - if (aio_dispatch(ctx, ret > 0)) { | ||
104 | - progress = true; | ||
105 | + progress |= aio_bh_poll(ctx); | ||
123 | + | 106 | + |
124 | + if (bs) { | 107 | + if (ret > 0) { |
125 | + blk = local_blk = blk_new(bdrv_get_aio_context(bs), 0, BLK_PERM_ALL); | 108 | + qemu_lockcnt_inc(&ctx->list_lock); |
126 | + ret = blk_insert_bs(blk, bs, &err); | 109 | + progress |= aio_dispatch_handlers(ctx); |
127 | + if (ret < 0) { | 110 | + qemu_lockcnt_dec(&ctx->list_lock); |
128 | + goto fail; | 111 | } |
129 | + } | 112 | |
130 | + } | 113 | + progress |= timerlistgroup_run_timers(&ctx->tlg); |
131 | + | 114 | + |
132 | /* | 115 | return progress; |
133 | * Notably absent: Proper permission management. This is sad, but it seems | ||
134 | * almost impossible to achieve without changing the semantics and thereby | ||
135 | @@ -XXX,XX +XXX,XX @@ void hmp_qemu_io(Monitor *mon, const QDict *qdict) | ||
136 | |||
137 | fail: | ||
138 | blk_unref(local_blk); | ||
139 | + | ||
140 | + if (ctx) { | ||
141 | + aio_context_release(ctx); | ||
142 | + } | ||
143 | + | ||
144 | hmp_handle_error(mon, err); | ||
145 | } | 116 | } |
146 | 117 | ||
147 | diff --git a/qemu-io-cmds.c b/qemu-io-cmds.c | 118 | diff --git a/util/aio-win32.c b/util/aio-win32.c |
148 | index XXXXXXX..XXXXXXX 100644 | 119 | index XXXXXXX..XXXXXXX 100644 |
149 | --- a/qemu-io-cmds.c | 120 | --- a/util/aio-win32.c |
150 | +++ b/qemu-io-cmds.c | 121 | +++ b/util/aio-win32.c |
151 | @@ -XXX,XX +XXX,XX @@ static const cmdinfo_t help_cmd = { | 122 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx, HANDLE event) |
152 | .oneline = "help for one or all commands", | 123 | return progress; |
153 | }; | 124 | } |
154 | 125 | ||
155 | +/* | 126 | -bool aio_dispatch(AioContext *ctx, bool dispatch_fds) |
156 | + * Called with aio context of blk acquired. Or with qemu_get_aio_context() | 127 | +void aio_dispatch(AioContext *ctx) |
157 | + * context acquired if blk is NULL. | ||
158 | + */ | ||
159 | int qemuio_command(BlockBackend *blk, const char *cmd) | ||
160 | { | 128 | { |
161 | - AioContext *ctx; | 129 | - bool progress; |
162 | char *input; | 130 | - |
163 | const cmdinfo_t *ct; | 131 | - progress = aio_bh_poll(ctx); |
164 | char **v; | 132 | - if (dispatch_fds) { |
165 | @@ -XXX,XX +XXX,XX @@ int qemuio_command(BlockBackend *blk, const char *cmd) | 133 | - progress |= aio_dispatch_handlers(ctx, INVALID_HANDLE_VALUE); |
166 | if (c) { | 134 | - } |
167 | ct = find_command(v[0]); | 135 | - progress |= timerlistgroup_run_timers(&ctx->tlg); |
168 | if (ct) { | 136 | - return progress; |
169 | - ctx = blk ? blk_get_aio_context(blk) : qemu_get_aio_context(); | 137 | + aio_bh_poll(ctx); |
170 | - aio_context_acquire(ctx); | 138 | + aio_dispatch_handlers(ctx, INVALID_HANDLE_VALUE); |
171 | ret = command(blk, ct, c, v); | 139 | + timerlistgroup_run_timers(&ctx->tlg); |
172 | - aio_context_release(ctx); | 140 | } |
173 | } else { | 141 | |
174 | fprintf(stderr, "command \"%s\" not found\n", v[0]); | 142 | bool aio_poll(AioContext *ctx, bool blocking) |
175 | ret = -EINVAL; | 143 | diff --git a/util/async.c b/util/async.c |
176 | diff --git a/qemu-io.c b/qemu-io.c | ||
177 | index XXXXXXX..XXXXXXX 100644 | 144 | index XXXXXXX..XXXXXXX 100644 |
178 | --- a/qemu-io.c | 145 | --- a/util/async.c |
179 | +++ b/qemu-io.c | 146 | +++ b/util/async.c |
180 | @@ -XXX,XX +XXX,XX @@ static void prep_fetchline(void *opaque) | 147 | @@ -XXX,XX +XXX,XX @@ aio_ctx_dispatch(GSource *source, |
181 | *fetchable= 1; | 148 | AioContext *ctx = (AioContext *) source; |
149 | |||
150 | assert(callback == NULL); | ||
151 | - aio_dispatch(ctx, true); | ||
152 | + aio_dispatch(ctx); | ||
153 | return true; | ||
182 | } | 154 | } |
183 | 155 | ||
184 | +static int do_qemuio_command(const char *cmd) | ||
185 | +{ | ||
186 | + int ret; | ||
187 | + AioContext *ctx = | ||
188 | + qemuio_blk ? blk_get_aio_context(qemuio_blk) : qemu_get_aio_context(); | ||
189 | + | ||
190 | + aio_context_acquire(ctx); | ||
191 | + ret = qemuio_command(qemuio_blk, cmd); | ||
192 | + aio_context_release(ctx); | ||
193 | + | ||
194 | + return ret; | ||
195 | +} | ||
196 | + | ||
197 | static int command_loop(void) | ||
198 | { | ||
199 | int i, fetchable = 0, prompted = 0; | ||
200 | @@ -XXX,XX +XXX,XX @@ static int command_loop(void) | ||
201 | char *input; | ||
202 | |||
203 | for (i = 0; !quit_qemu_io && i < ncmdline; i++) { | ||
204 | - ret = qemuio_command(qemuio_blk, cmdline[i]); | ||
205 | + ret = do_qemuio_command(cmdline[i]); | ||
206 | if (ret < 0) { | ||
207 | last_error = ret; | ||
208 | } | ||
209 | @@ -XXX,XX +XXX,XX @@ static int command_loop(void) | ||
210 | if (input == NULL) { | ||
211 | break; | ||
212 | } | ||
213 | - ret = qemuio_command(qemuio_blk, input); | ||
214 | + ret = do_qemuio_command(input); | ||
215 | g_free(input); | ||
216 | |||
217 | if (ret < 0) { | ||
218 | -- | 156 | -- |
219 | 2.31.1 | 157 | 2.9.3 |
220 | 158 | ||
221 | 159 | diff view generated by jsdifflib |
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | bdrv_write_threshold_exceeded() is unused. | 3 | Pull the increment/decrement pair out of aio_bh_poll and into the |
4 | callers. | ||
4 | 5 | ||
5 | bdrv_write_threshold_is_set() is used only to double check the value of | 6 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
6 | bs->write_threshold_offset in tests. No real sense in it (both tests do | 7 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
7 | check real value with help of bdrv_write_threshold_get()) | 8 | Reviewed-by: Fam Zheng <famz@redhat.com> |
9 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> | ||
10 | Message-id: 20170213135235.12274-18-pbonzini@redhat.com | ||
11 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
12 | --- | ||
13 | util/aio-posix.c | 8 +++----- | ||
14 | util/aio-win32.c | 8 ++++---- | ||
15 | util/async.c | 12 ++++++------ | ||
16 | 3 files changed, 13 insertions(+), 15 deletions(-) | ||
8 | 17 | ||
9 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 18 | diff --git a/util/aio-posix.c b/util/aio-posix.c |
10 | Reviewed-by: Max Reitz <mreitz@redhat.com> | ||
11 | Message-Id: <20210506090621.11848-5-vsementsov@virtuozzo.com> | ||
12 | Reviewed-by: Eric Blake <eblake@redhat.com> | ||
13 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
14 | [mreitz: Adjusted commit message as per Eric's suggestion] | ||
15 | Signed-off-by: Max Reitz <mreitz@redhat.com> | ||
16 | --- | ||
17 | include/block/write-threshold.h | 24 ------------------------ | ||
18 | block/write-threshold.c | 19 ------------------- | ||
19 | tests/unit/test-write-threshold.c | 4 ---- | ||
20 | 3 files changed, 47 deletions(-) | ||
21 | |||
22 | diff --git a/include/block/write-threshold.h b/include/block/write-threshold.h | ||
23 | index XXXXXXX..XXXXXXX 100644 | 19 | index XXXXXXX..XXXXXXX 100644 |
24 | --- a/include/block/write-threshold.h | 20 | --- a/util/aio-posix.c |
25 | +++ b/include/block/write-threshold.h | 21 | +++ b/util/aio-posix.c |
26 | @@ -XXX,XX +XXX,XX @@ void bdrv_write_threshold_set(BlockDriverState *bs, uint64_t threshold_bytes); | 22 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx) |
27 | */ | 23 | |
28 | uint64_t bdrv_write_threshold_get(const BlockDriverState *bs); | 24 | void aio_dispatch(AioContext *ctx) |
29 | 25 | { | |
30 | -/* | 26 | + qemu_lockcnt_inc(&ctx->list_lock); |
31 | - * bdrv_write_threshold_is_set | 27 | aio_bh_poll(ctx); |
32 | - * | ||
33 | - * Tell if a write threshold is set for a given BDS. | ||
34 | - */ | ||
35 | -bool bdrv_write_threshold_is_set(const BlockDriverState *bs); | ||
36 | - | 28 | - |
37 | -/* | 29 | - qemu_lockcnt_inc(&ctx->list_lock); |
38 | - * bdrv_write_threshold_exceeded | 30 | aio_dispatch_handlers(ctx); |
39 | - * | 31 | qemu_lockcnt_dec(&ctx->list_lock); |
40 | - * Return the extent of a write request that exceeded the threshold, | 32 | |
41 | - * or zero if the request is below the threshold. | 33 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) |
42 | - * Return zero also if the threshold was not set. | 34 | } |
43 | - * | 35 | |
44 | - * NOTE: here we assume the following holds for each request this code | 36 | npfd = 0; |
45 | - * deals with: | 37 | - qemu_lockcnt_dec(&ctx->list_lock); |
46 | - * | 38 | |
47 | - * assert((req->offset + req->bytes) <= UINT64_MAX) | 39 | progress |= aio_bh_poll(ctx); |
48 | - * | 40 | |
49 | - * Please not there is *not* an actual C assert(). | 41 | if (ret > 0) { |
50 | - */ | 42 | - qemu_lockcnt_inc(&ctx->list_lock); |
51 | -uint64_t bdrv_write_threshold_exceeded(const BlockDriverState *bs, | 43 | progress |= aio_dispatch_handlers(ctx); |
52 | - const BdrvTrackedRequest *req); | 44 | - qemu_lockcnt_dec(&ctx->list_lock); |
45 | } | ||
46 | |||
47 | + qemu_lockcnt_dec(&ctx->list_lock); | ||
48 | + | ||
49 | progress |= timerlistgroup_run_timers(&ctx->tlg); | ||
50 | |||
51 | return progress; | ||
52 | diff --git a/util/aio-win32.c b/util/aio-win32.c | ||
53 | index XXXXXXX..XXXXXXX 100644 | ||
54 | --- a/util/aio-win32.c | ||
55 | +++ b/util/aio-win32.c | ||
56 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx, HANDLE event) | ||
57 | bool progress = false; | ||
58 | AioHandler *tmp; | ||
59 | |||
60 | - qemu_lockcnt_inc(&ctx->list_lock); | ||
53 | - | 61 | - |
54 | /* | 62 | /* |
55 | * bdrv_write_threshold_check_write | 63 | * We have to walk very carefully in case aio_set_fd_handler is |
56 | * | 64 | * called while we're walking. |
57 | diff --git a/block/write-threshold.c b/block/write-threshold.c | 65 | @@ -XXX,XX +XXX,XX @@ static bool aio_dispatch_handlers(AioContext *ctx, HANDLE event) |
66 | } | ||
67 | } | ||
68 | |||
69 | - qemu_lockcnt_dec(&ctx->list_lock); | ||
70 | return progress; | ||
71 | } | ||
72 | |||
73 | void aio_dispatch(AioContext *ctx) | ||
74 | { | ||
75 | + qemu_lockcnt_inc(&ctx->list_lock); | ||
76 | aio_bh_poll(ctx); | ||
77 | aio_dispatch_handlers(ctx, INVALID_HANDLE_VALUE); | ||
78 | + qemu_lockcnt_dec(&ctx->list_lock); | ||
79 | timerlistgroup_run_timers(&ctx->tlg); | ||
80 | } | ||
81 | |||
82 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
83 | } | ||
84 | } | ||
85 | |||
86 | - qemu_lockcnt_dec(&ctx->list_lock); | ||
87 | first = true; | ||
88 | |||
89 | /* ctx->notifier is always registered. */ | ||
90 | @@ -XXX,XX +XXX,XX @@ bool aio_poll(AioContext *ctx, bool blocking) | ||
91 | progress |= aio_dispatch_handlers(ctx, event); | ||
92 | } while (count > 0); | ||
93 | |||
94 | + qemu_lockcnt_dec(&ctx->list_lock); | ||
95 | + | ||
96 | progress |= timerlistgroup_run_timers(&ctx->tlg); | ||
97 | return progress; | ||
98 | } | ||
99 | diff --git a/util/async.c b/util/async.c | ||
58 | index XXXXXXX..XXXXXXX 100644 | 100 | index XXXXXXX..XXXXXXX 100644 |
59 | --- a/block/write-threshold.c | 101 | --- a/util/async.c |
60 | +++ b/block/write-threshold.c | 102 | +++ b/util/async.c |
61 | @@ -XXX,XX +XXX,XX @@ uint64_t bdrv_write_threshold_get(const BlockDriverState *bs) | 103 | @@ -XXX,XX +XXX,XX @@ void aio_bh_call(QEMUBH *bh) |
62 | return bs->write_threshold_offset; | 104 | bh->cb(bh->opaque); |
63 | } | 105 | } |
64 | 106 | ||
65 | -bool bdrv_write_threshold_is_set(const BlockDriverState *bs) | 107 | -/* Multiple occurrences of aio_bh_poll cannot be called concurrently */ |
66 | -{ | 108 | +/* Multiple occurrences of aio_bh_poll cannot be called concurrently. |
67 | - return bs->write_threshold_offset > 0; | 109 | + * The count in ctx->list_lock is incremented before the call, and is |
68 | -} | 110 | + * not affected by the call. |
111 | + */ | ||
112 | int aio_bh_poll(AioContext *ctx) | ||
113 | { | ||
114 | QEMUBH *bh, **bhp, *next; | ||
115 | int ret; | ||
116 | bool deleted = false; | ||
117 | |||
118 | - qemu_lockcnt_inc(&ctx->list_lock); | ||
69 | - | 119 | - |
70 | -uint64_t bdrv_write_threshold_exceeded(const BlockDriverState *bs, | 120 | ret = 0; |
71 | - const BdrvTrackedRequest *req) | 121 | for (bh = atomic_rcu_read(&ctx->first_bh); bh; bh = next) { |
72 | -{ | 122 | next = atomic_rcu_read(&bh->next); |
73 | - if (bdrv_write_threshold_is_set(bs)) { | 123 | @@ -XXX,XX +XXX,XX @@ int aio_bh_poll(AioContext *ctx) |
74 | - if (req->offset > bs->write_threshold_offset) { | 124 | |
75 | - return (req->offset - bs->write_threshold_offset) + req->bytes; | 125 | /* remove deleted bhs */ |
76 | - } | 126 | if (!deleted) { |
77 | - if ((req->offset + req->bytes) > bs->write_threshold_offset) { | 127 | - qemu_lockcnt_dec(&ctx->list_lock); |
78 | - return (req->offset + req->bytes) - bs->write_threshold_offset; | 128 | return ret; |
79 | - } | 129 | } |
80 | - } | 130 | |
81 | - return 0; | 131 | - if (qemu_lockcnt_dec_and_lock(&ctx->list_lock)) { |
82 | -} | 132 | + if (qemu_lockcnt_dec_if_lock(&ctx->list_lock)) { |
83 | - | 133 | bhp = &ctx->first_bh; |
84 | void bdrv_write_threshold_set(BlockDriverState *bs, uint64_t threshold_bytes) | 134 | while (*bhp) { |
85 | { | 135 | bh = *bhp; |
86 | bs->write_threshold_offset = threshold_bytes; | 136 | @@ -XXX,XX +XXX,XX @@ int aio_bh_poll(AioContext *ctx) |
87 | diff --git a/tests/unit/test-write-threshold.c b/tests/unit/test-write-threshold.c | 137 | bhp = &bh->next; |
88 | index XXXXXXX..XXXXXXX 100644 | 138 | } |
89 | --- a/tests/unit/test-write-threshold.c | 139 | } |
90 | +++ b/tests/unit/test-write-threshold.c | 140 | - qemu_lockcnt_unlock(&ctx->list_lock); |
91 | @@ -XXX,XX +XXX,XX @@ static void test_threshold_not_set_on_init(void) | 141 | + qemu_lockcnt_inc_and_unlock(&ctx->list_lock); |
92 | BlockDriverState bs; | 142 | } |
93 | memset(&bs, 0, sizeof(bs)); | 143 | return ret; |
94 | |||
95 | - g_assert(!bdrv_write_threshold_is_set(&bs)); | ||
96 | - | ||
97 | res = bdrv_write_threshold_get(&bs); | ||
98 | g_assert_cmpint(res, ==, 0); | ||
99 | } | ||
100 | @@ -XXX,XX +XXX,XX @@ static void test_threshold_set_get(void) | ||
101 | |||
102 | bdrv_write_threshold_set(&bs, threshold); | ||
103 | |||
104 | - g_assert(bdrv_write_threshold_is_set(&bs)); | ||
105 | - | ||
106 | res = bdrv_write_threshold_get(&bs); | ||
107 | g_assert_cmpint(res, ==, threshold); | ||
108 | } | 144 | } |
109 | -- | 145 | -- |
110 | 2.31.1 | 146 | 2.9.3 |
111 | 147 | ||
112 | 148 | diff view generated by jsdifflib |
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | They are unused now. | ||
4 | |||
5 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
6 | Reviewed-by: Max Reitz <mreitz@redhat.com> | ||
7 | Message-Id: <20210506090621.11848-3-vsementsov@virtuozzo.com> | ||
8 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | 3 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> |
9 | Signed-off-by: Max Reitz <mreitz@redhat.com> | 4 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
5 | Reviewed-by: Fam Zheng <famz@redhat.com> | ||
6 | Reviewed-by: Daniel P. Berrange <berrange@redhat.com> | ||
7 | Message-id: 20170213135235.12274-19-pbonzini@redhat.com | ||
8 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
10 | --- | 9 | --- |
11 | include/block/block_int.h | 12 ------------ | 10 | include/block/block_int.h | 64 +++++++++++++++++++++++++----------------- |
12 | block.c | 1 - | 11 | include/sysemu/block-backend.h | 14 ++++++--- |
13 | block/io.c | 6 ------ | 12 | 2 files changed, 49 insertions(+), 29 deletions(-) |
14 | 3 files changed, 19 deletions(-) | ||
15 | 13 | ||
16 | diff --git a/include/block/block_int.h b/include/block/block_int.h | 14 | diff --git a/include/block/block_int.h b/include/block/block_int.h |
17 | index XXXXXXX..XXXXXXX 100644 | 15 | index XXXXXXX..XXXXXXX 100644 |
18 | --- a/include/block/block_int.h | 16 | --- a/include/block/block_int.h |
19 | +++ b/include/block/block_int.h | 17 | +++ b/include/block/block_int.h |
18 | @@ -XXX,XX +XXX,XX @@ struct BdrvChild { | ||
19 | * copied as well. | ||
20 | */ | ||
21 | struct BlockDriverState { | ||
22 | - int64_t total_sectors; /* if we are reading a disk image, give its | ||
23 | - size in sectors */ | ||
24 | + /* Protected by big QEMU lock or read-only after opening. No special | ||
25 | + * locking needed during I/O... | ||
26 | + */ | ||
27 | int open_flags; /* flags used to open the file, re-used for re-open */ | ||
28 | bool read_only; /* if true, the media is read only */ | ||
29 | bool encrypted; /* if true, the media is encrypted */ | ||
20 | @@ -XXX,XX +XXX,XX @@ struct BlockDriverState { | 30 | @@ -XXX,XX +XXX,XX @@ struct BlockDriverState { |
21 | */ | 31 | bool sg; /* if true, the device is a /dev/sg* */ |
22 | int64_t total_sectors; | 32 | bool probed; /* if true, format was probed rather than specified */ |
33 | |||
34 | - int copy_on_read; /* if nonzero, copy read backing sectors into image. | ||
35 | - note this is a reference count */ | ||
36 | - | ||
37 | - CoQueue flush_queue; /* Serializing flush queue */ | ||
38 | - bool active_flush_req; /* Flush request in flight? */ | ||
39 | - unsigned int write_gen; /* Current data generation */ | ||
40 | - unsigned int flushed_gen; /* Flushed write generation */ | ||
41 | - | ||
42 | BlockDriver *drv; /* NULL means no media */ | ||
43 | void *opaque; | ||
44 | |||
45 | @@ -XXX,XX +XXX,XX @@ struct BlockDriverState { | ||
46 | BdrvChild *backing; | ||
47 | BdrvChild *file; | ||
23 | 48 | ||
24 | - /* Callback before write request is processed */ | 49 | - /* Callback before write request is processed */ |
25 | - NotifierWithReturnList before_write_notifiers; | 50 | - NotifierWithReturnList before_write_notifiers; |
26 | - | 51 | - |
52 | - /* number of in-flight requests; overall and serialising */ | ||
53 | - unsigned int in_flight; | ||
54 | - unsigned int serialising_in_flight; | ||
55 | - | ||
56 | - bool wakeup; | ||
57 | - | ||
58 | - /* Offset after the highest byte written to */ | ||
59 | - uint64_t wr_highest_offset; | ||
60 | - | ||
61 | /* I/O Limits */ | ||
62 | BlockLimits bl; | ||
63 | |||
64 | @@ -XXX,XX +XXX,XX @@ struct BlockDriverState { | ||
65 | QTAILQ_ENTRY(BlockDriverState) bs_list; | ||
66 | /* element of the list of monitor-owned BDS */ | ||
67 | QTAILQ_ENTRY(BlockDriverState) monitor_list; | ||
68 | - QLIST_HEAD(, BdrvDirtyBitmap) dirty_bitmaps; | ||
69 | int refcnt; | ||
70 | |||
71 | - QLIST_HEAD(, BdrvTrackedRequest) tracked_requests; | ||
72 | - | ||
73 | /* operation blockers */ | ||
74 | QLIST_HEAD(, BdrvOpBlocker) op_blockers[BLOCK_OP_TYPE_MAX]; | ||
75 | |||
76 | @@ -XXX,XX +XXX,XX @@ struct BlockDriverState { | ||
77 | /* The error object in use for blocking operations on backing_hd */ | ||
78 | Error *backing_blocker; | ||
79 | |||
80 | + /* Protected by AioContext lock */ | ||
81 | + | ||
82 | + /* If true, copy read backing sectors into image. Can be >1 if more | ||
83 | + * than one client has requested copy-on-read. | ||
84 | + */ | ||
85 | + int copy_on_read; | ||
86 | + | ||
87 | + /* If we are reading a disk image, give its size in sectors. | ||
88 | + * Generally read-only; it is written to by load_vmstate and save_vmstate, | ||
89 | + * but the block layer is quiescent during those. | ||
90 | + */ | ||
91 | + int64_t total_sectors; | ||
92 | + | ||
93 | + /* Callback before write request is processed */ | ||
94 | + NotifierWithReturnList before_write_notifiers; | ||
95 | + | ||
96 | + /* number of in-flight requests; overall and serialising */ | ||
97 | + unsigned int in_flight; | ||
98 | + unsigned int serialising_in_flight; | ||
99 | + | ||
100 | + bool wakeup; | ||
101 | + | ||
102 | + /* Offset after the highest byte written to */ | ||
103 | + uint64_t wr_highest_offset; | ||
104 | + | ||
27 | /* threshold limit for writes, in bytes. "High water mark". */ | 105 | /* threshold limit for writes, in bytes. "High water mark". */ |
28 | uint64_t write_threshold_offset; | 106 | uint64_t write_threshold_offset; |
29 | 107 | NotifierWithReturn write_threshold_notifier; | |
30 | @@ -XXX,XX +XXX,XX @@ void bdrv_parse_filename_strip_prefix(const char *filename, const char *prefix, | 108 | @@ -XXX,XX +XXX,XX @@ struct BlockDriverState { |
31 | bool bdrv_backing_overridden(BlockDriverState *bs); | 109 | /* counter for nested bdrv_io_plug */ |
32 | 110 | unsigned io_plugged; | |
33 | 111 | ||
34 | -/** | 112 | + QLIST_HEAD(, BdrvTrackedRequest) tracked_requests; |
35 | - * bdrv_add_before_write_notifier: | 113 | + CoQueue flush_queue; /* Serializing flush queue */ |
36 | - * | 114 | + bool active_flush_req; /* Flush request in flight? */ |
37 | - * Register a callback that is invoked before write requests are processed but | 115 | + unsigned int write_gen; /* Current data generation */ |
38 | - * after any throttling or waiting for overlapping requests. | 116 | + unsigned int flushed_gen; /* Flushed write generation */ |
39 | - */ | 117 | + |
40 | -void bdrv_add_before_write_notifier(BlockDriverState *bs, | 118 | + QLIST_HEAD(, BdrvDirtyBitmap) dirty_bitmaps; |
41 | - NotifierWithReturn *notifier); | 119 | + |
42 | - | 120 | + /* do we need to tell the quest if we have a volatile write cache? */ |
43 | /** | 121 | + int enable_write_cache; |
44 | * bdrv_add_aio_context_notifier: | 122 | + |
45 | * | 123 | int quiesce_counter; |
46 | diff --git a/block.c b/block.c | 124 | }; |
125 | |||
126 | diff --git a/include/sysemu/block-backend.h b/include/sysemu/block-backend.h | ||
47 | index XXXXXXX..XXXXXXX 100644 | 127 | index XXXXXXX..XXXXXXX 100644 |
48 | --- a/block.c | 128 | --- a/include/sysemu/block-backend.h |
49 | +++ b/block.c | 129 | +++ b/include/sysemu/block-backend.h |
50 | @@ -XXX,XX +XXX,XX @@ BlockDriverState *bdrv_new(void) | 130 | @@ -XXX,XX +XXX,XX @@ typedef struct BlockDevOps { |
51 | for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) { | 131 | * fields that must be public. This is in particular for QLIST_ENTRY() and |
52 | QLIST_INIT(&bs->op_blockers[i]); | 132 | * friends so that BlockBackends can be kept in lists outside block-backend.c */ |
53 | } | 133 | typedef struct BlockBackendPublic { |
54 | - notifier_with_return_list_init(&bs->before_write_notifiers); | 134 | - /* I/O throttling. |
55 | qemu_co_mutex_init(&bs->reqs_lock); | 135 | - * throttle_state tells us if this BlockBackend has I/O limits configured. |
56 | qemu_mutex_init(&bs->dirty_bitmap_mutex); | 136 | - * io_limits_disabled tells us if they are currently being enforced */ |
57 | bs->refcnt = 1; | 137 | + /* I/O throttling has its own locking, but also some fields are |
58 | diff --git a/block/io.c b/block/io.c | 138 | + * protected by the AioContext lock. |
59 | index XXXXXXX..XXXXXXX 100644 | 139 | + */ |
60 | --- a/block/io.c | 140 | + |
61 | +++ b/block/io.c | 141 | + /* Protected by AioContext lock. */ |
62 | @@ -XXX,XX +XXX,XX @@ bool bdrv_qiov_is_aligned(BlockDriverState *bs, QEMUIOVector *qiov) | 142 | CoQueue throttled_reqs[2]; |
63 | return true; | 143 | + |
64 | } | 144 | + /* Nonzero if the I/O limits are currently being ignored; generally |
65 | 145 | + * it is zero. */ | |
66 | -void bdrv_add_before_write_notifier(BlockDriverState *bs, | 146 | unsigned int io_limits_disabled; |
67 | - NotifierWithReturn *notifier) | 147 | |
68 | -{ | 148 | /* The following fields are protected by the ThrottleGroup lock. |
69 | - notifier_with_return_list_add(&bs->before_write_notifiers, notifier); | 149 | - * See the ThrottleGroup documentation for details. */ |
70 | -} | 150 | + * See the ThrottleGroup documentation for details. |
71 | - | 151 | + * throttle_state tells us if I/O limits are configured. */ |
72 | void bdrv_io_plug(BlockDriverState *bs) | 152 | ThrottleState *throttle_state; |
73 | { | 153 | ThrottleTimers throttle_timers; |
74 | BdrvChild *child; | 154 | unsigned pending_reqs[2]; |
75 | -- | 155 | -- |
76 | 2.31.1 | 156 | 2.9.3 |
77 | 157 | ||
78 | 158 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | 1 | From: Paolo Bonzini <pbonzini@redhat.com> | |
2 | |||
3 | This uses the lock-free mutex described in the paper '"Blocking without | ||
4 | Locking", or LFTHREADS: A lock-free thread library' by Gidenstam and | ||
5 | Papatriantafilou. The same technique is used in OSv, and in fact | ||
6 | the code is essentially a conversion to C of OSv's code. | ||
7 | |||
8 | [Added missing coroutine_fn in tests/test-aio-multithread.c. | ||
9 | --Stefan] | ||
10 | |||
11 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> | ||
12 | Reviewed-by: Fam Zheng <famz@redhat.com> | ||
13 | Message-id: 20170213181244.16297-2-pbonzini@redhat.com | ||
14 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
15 | --- | ||
16 | include/qemu/coroutine.h | 17 ++++- | ||
17 | tests/test-aio-multithread.c | 86 ++++++++++++++++++++++++ | ||
18 | util/qemu-coroutine-lock.c | 155 ++++++++++++++++++++++++++++++++++++++++--- | ||
19 | util/trace-events | 1 + | ||
20 | 4 files changed, 246 insertions(+), 13 deletions(-) | ||
21 | |||
22 | diff --git a/include/qemu/coroutine.h b/include/qemu/coroutine.h | ||
23 | index XXXXXXX..XXXXXXX 100644 | ||
24 | --- a/include/qemu/coroutine.h | ||
25 | +++ b/include/qemu/coroutine.h | ||
26 | @@ -XXX,XX +XXX,XX @@ bool qemu_co_queue_empty(CoQueue *queue); | ||
27 | /** | ||
28 | * Provides a mutex that can be used to synchronise coroutines | ||
29 | */ | ||
30 | +struct CoWaitRecord; | ||
31 | typedef struct CoMutex { | ||
32 | - bool locked; | ||
33 | + /* Count of pending lockers; 0 for a free mutex, 1 for an | ||
34 | + * uncontended mutex. | ||
35 | + */ | ||
36 | + unsigned locked; | ||
37 | + | ||
38 | + /* A queue of waiters. Elements are added atomically in front of | ||
39 | + * from_push. to_pop is only populated, and popped from, by whoever | ||
40 | + * is in charge of the next wakeup. This can be an unlocker or, | ||
41 | + * through the handoff protocol, a locker that is about to go to sleep. | ||
42 | + */ | ||
43 | + QSLIST_HEAD(, CoWaitRecord) from_push, to_pop; | ||
44 | + | ||
45 | + unsigned handoff, sequence; | ||
46 | + | ||
47 | Coroutine *holder; | ||
48 | - CoQueue queue; | ||
49 | } CoMutex; | ||
50 | |||
51 | /** | ||
52 | diff --git a/tests/test-aio-multithread.c b/tests/test-aio-multithread.c | ||
53 | index XXXXXXX..XXXXXXX 100644 | ||
54 | --- a/tests/test-aio-multithread.c | ||
55 | +++ b/tests/test-aio-multithread.c | ||
56 | @@ -XXX,XX +XXX,XX @@ static void test_multi_co_schedule_10(void) | ||
57 | test_multi_co_schedule(10); | ||
58 | } | ||
59 | |||
60 | +/* CoMutex thread-safety. */ | ||
61 | + | ||
62 | +static uint32_t atomic_counter; | ||
63 | +static uint32_t running; | ||
64 | +static uint32_t counter; | ||
65 | +static CoMutex comutex; | ||
66 | + | ||
67 | +static void coroutine_fn test_multi_co_mutex_entry(void *opaque) | ||
68 | +{ | ||
69 | + while (!atomic_mb_read(&now_stopping)) { | ||
70 | + qemu_co_mutex_lock(&comutex); | ||
71 | + counter++; | ||
72 | + qemu_co_mutex_unlock(&comutex); | ||
73 | + | ||
74 | + /* Increase atomic_counter *after* releasing the mutex. Otherwise | ||
75 | + * there is a chance (it happens about 1 in 3 runs) that the iothread | ||
76 | + * exits before the coroutine is woken up, causing a spurious | ||
77 | + * assertion failure. | ||
78 | + */ | ||
79 | + atomic_inc(&atomic_counter); | ||
80 | + } | ||
81 | + atomic_dec(&running); | ||
82 | +} | ||
83 | + | ||
84 | +static void test_multi_co_mutex(int threads, int seconds) | ||
85 | +{ | ||
86 | + int i; | ||
87 | + | ||
88 | + qemu_co_mutex_init(&comutex); | ||
89 | + counter = 0; | ||
90 | + atomic_counter = 0; | ||
91 | + now_stopping = false; | ||
92 | + | ||
93 | + create_aio_contexts(); | ||
94 | + assert(threads <= NUM_CONTEXTS); | ||
95 | + running = threads; | ||
96 | + for (i = 0; i < threads; i++) { | ||
97 | + Coroutine *co1 = qemu_coroutine_create(test_multi_co_mutex_entry, NULL); | ||
98 | + aio_co_schedule(ctx[i], co1); | ||
99 | + } | ||
100 | + | ||
101 | + g_usleep(seconds * 1000000); | ||
102 | + | ||
103 | + atomic_mb_set(&now_stopping, true); | ||
104 | + while (running > 0) { | ||
105 | + g_usleep(100000); | ||
106 | + } | ||
107 | + | ||
108 | + join_aio_contexts(); | ||
109 | + g_test_message("%d iterations/second\n", counter / seconds); | ||
110 | + g_assert_cmpint(counter, ==, atomic_counter); | ||
111 | +} | ||
112 | + | ||
113 | +/* Testing with NUM_CONTEXTS threads focuses on the queue. The mutex however | ||
114 | + * is too contended (and the threads spend too much time in aio_poll) | ||
115 | + * to actually stress the handoff protocol. | ||
116 | + */ | ||
117 | +static void test_multi_co_mutex_1(void) | ||
118 | +{ | ||
119 | + test_multi_co_mutex(NUM_CONTEXTS, 1); | ||
120 | +} | ||
121 | + | ||
122 | +static void test_multi_co_mutex_10(void) | ||
123 | +{ | ||
124 | + test_multi_co_mutex(NUM_CONTEXTS, 10); | ||
125 | +} | ||
126 | + | ||
127 | +/* Testing with fewer threads stresses the handoff protocol too. Still, the | ||
128 | + * case where the locker _can_ pick up a handoff is very rare, happening | ||
129 | + * about 10 times in 1 million, so increase the runtime a bit compared to | ||
130 | + * other "quick" testcases that only run for 1 second. | ||
131 | + */ | ||
132 | +static void test_multi_co_mutex_2_3(void) | ||
133 | +{ | ||
134 | + test_multi_co_mutex(2, 3); | ||
135 | +} | ||
136 | + | ||
137 | +static void test_multi_co_mutex_2_30(void) | ||
138 | +{ | ||
139 | + test_multi_co_mutex(2, 30); | ||
140 | +} | ||
141 | + | ||
142 | /* End of tests. */ | ||
143 | |||
144 | int main(int argc, char **argv) | ||
145 | @@ -XXX,XX +XXX,XX @@ int main(int argc, char **argv) | ||
146 | g_test_add_func("/aio/multi/lifecycle", test_lifecycle); | ||
147 | if (g_test_quick()) { | ||
148 | g_test_add_func("/aio/multi/schedule", test_multi_co_schedule_1); | ||
149 | + g_test_add_func("/aio/multi/mutex/contended", test_multi_co_mutex_1); | ||
150 | + g_test_add_func("/aio/multi/mutex/handoff", test_multi_co_mutex_2_3); | ||
151 | } else { | ||
152 | g_test_add_func("/aio/multi/schedule", test_multi_co_schedule_10); | ||
153 | + g_test_add_func("/aio/multi/mutex/contended", test_multi_co_mutex_10); | ||
154 | + g_test_add_func("/aio/multi/mutex/handoff", test_multi_co_mutex_2_30); | ||
155 | } | ||
156 | return g_test_run(); | ||
157 | } | ||
158 | diff --git a/util/qemu-coroutine-lock.c b/util/qemu-coroutine-lock.c | ||
159 | index XXXXXXX..XXXXXXX 100644 | ||
160 | --- a/util/qemu-coroutine-lock.c | ||
161 | +++ b/util/qemu-coroutine-lock.c | ||
162 | @@ -XXX,XX +XXX,XX @@ | ||
163 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | ||
164 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | ||
165 | * THE SOFTWARE. | ||
166 | + * | ||
167 | + * The lock-free mutex implementation is based on OSv | ||
168 | + * (core/lfmutex.cc, include/lockfree/mutex.hh). | ||
169 | + * Copyright (C) 2013 Cloudius Systems, Ltd. | ||
170 | */ | ||
171 | |||
172 | #include "qemu/osdep.h" | ||
173 | @@ -XXX,XX +XXX,XX @@ bool qemu_co_queue_empty(CoQueue *queue) | ||
174 | return QSIMPLEQ_FIRST(&queue->entries) == NULL; | ||
175 | } | ||
176 | |||
177 | +/* The wait records are handled with a multiple-producer, single-consumer | ||
178 | + * lock-free queue. There cannot be two concurrent pop_waiter() calls | ||
179 | + * because pop_waiter() can only be called while mutex->handoff is zero. | ||
180 | + * This can happen in three cases: | ||
181 | + * - in qemu_co_mutex_unlock, before the hand-off protocol has started. | ||
182 | + * In this case, qemu_co_mutex_lock will see mutex->handoff == 0 and | ||
183 | + * not take part in the handoff. | ||
184 | + * - in qemu_co_mutex_lock, if it steals the hand-off responsibility from | ||
185 | + * qemu_co_mutex_unlock. In this case, qemu_co_mutex_unlock will fail | ||
186 | + * the cmpxchg (it will see either 0 or the next sequence value) and | ||
187 | + * exit. The next hand-off cannot begin until qemu_co_mutex_lock has | ||
188 | + * woken up someone. | ||
189 | + * - in qemu_co_mutex_unlock, if it takes the hand-off token itself. | ||
190 | + * In this case another iteration starts with mutex->handoff == 0; | ||
191 | + * a concurrent qemu_co_mutex_lock will fail the cmpxchg, and | ||
192 | + * qemu_co_mutex_unlock will go back to case (1). | ||
193 | + * | ||
194 | + * The following functions manage this queue. | ||
195 | + */ | ||
196 | +typedef struct CoWaitRecord { | ||
197 | + Coroutine *co; | ||
198 | + QSLIST_ENTRY(CoWaitRecord) next; | ||
199 | +} CoWaitRecord; | ||
200 | + | ||
201 | +static void push_waiter(CoMutex *mutex, CoWaitRecord *w) | ||
202 | +{ | ||
203 | + w->co = qemu_coroutine_self(); | ||
204 | + QSLIST_INSERT_HEAD_ATOMIC(&mutex->from_push, w, next); | ||
205 | +} | ||
206 | + | ||
207 | +static void move_waiters(CoMutex *mutex) | ||
208 | +{ | ||
209 | + QSLIST_HEAD(, CoWaitRecord) reversed; | ||
210 | + QSLIST_MOVE_ATOMIC(&reversed, &mutex->from_push); | ||
211 | + while (!QSLIST_EMPTY(&reversed)) { | ||
212 | + CoWaitRecord *w = QSLIST_FIRST(&reversed); | ||
213 | + QSLIST_REMOVE_HEAD(&reversed, next); | ||
214 | + QSLIST_INSERT_HEAD(&mutex->to_pop, w, next); | ||
215 | + } | ||
216 | +} | ||
217 | + | ||
218 | +static CoWaitRecord *pop_waiter(CoMutex *mutex) | ||
219 | +{ | ||
220 | + CoWaitRecord *w; | ||
221 | + | ||
222 | + if (QSLIST_EMPTY(&mutex->to_pop)) { | ||
223 | + move_waiters(mutex); | ||
224 | + if (QSLIST_EMPTY(&mutex->to_pop)) { | ||
225 | + return NULL; | ||
226 | + } | ||
227 | + } | ||
228 | + w = QSLIST_FIRST(&mutex->to_pop); | ||
229 | + QSLIST_REMOVE_HEAD(&mutex->to_pop, next); | ||
230 | + return w; | ||
231 | +} | ||
232 | + | ||
233 | +static bool has_waiters(CoMutex *mutex) | ||
234 | +{ | ||
235 | + return QSLIST_EMPTY(&mutex->to_pop) || QSLIST_EMPTY(&mutex->from_push); | ||
236 | +} | ||
237 | + | ||
238 | void qemu_co_mutex_init(CoMutex *mutex) | ||
239 | { | ||
240 | memset(mutex, 0, sizeof(*mutex)); | ||
241 | - qemu_co_queue_init(&mutex->queue); | ||
242 | } | ||
243 | |||
244 | -void coroutine_fn qemu_co_mutex_lock(CoMutex *mutex) | ||
245 | +static void coroutine_fn qemu_co_mutex_lock_slowpath(CoMutex *mutex) | ||
246 | { | ||
247 | Coroutine *self = qemu_coroutine_self(); | ||
248 | + CoWaitRecord w; | ||
249 | + unsigned old_handoff; | ||
250 | |||
251 | trace_qemu_co_mutex_lock_entry(mutex, self); | ||
252 | + w.co = self; | ||
253 | + push_waiter(mutex, &w); | ||
254 | |||
255 | - while (mutex->locked) { | ||
256 | - qemu_co_queue_wait(&mutex->queue); | ||
257 | + /* This is the "Responsibility Hand-Off" protocol; a lock() picks from | ||
258 | + * a concurrent unlock() the responsibility of waking somebody up. | ||
259 | + */ | ||
260 | + old_handoff = atomic_mb_read(&mutex->handoff); | ||
261 | + if (old_handoff && | ||
262 | + has_waiters(mutex) && | ||
263 | + atomic_cmpxchg(&mutex->handoff, old_handoff, 0) == old_handoff) { | ||
264 | + /* There can be no concurrent pops, because there can be only | ||
265 | + * one active handoff at a time. | ||
266 | + */ | ||
267 | + CoWaitRecord *to_wake = pop_waiter(mutex); | ||
268 | + Coroutine *co = to_wake->co; | ||
269 | + if (co == self) { | ||
270 | + /* We got the lock ourselves! */ | ||
271 | + assert(to_wake == &w); | ||
272 | + return; | ||
273 | + } | ||
274 | + | ||
275 | + aio_co_wake(co); | ||
276 | } | ||
277 | |||
278 | - mutex->locked = true; | ||
279 | - mutex->holder = self; | ||
280 | - self->locks_held++; | ||
281 | - | ||
282 | + qemu_coroutine_yield(); | ||
283 | trace_qemu_co_mutex_lock_return(mutex, self); | ||
284 | } | ||
285 | |||
286 | +void coroutine_fn qemu_co_mutex_lock(CoMutex *mutex) | ||
287 | +{ | ||
288 | + Coroutine *self = qemu_coroutine_self(); | ||
289 | + | ||
290 | + if (atomic_fetch_inc(&mutex->locked) == 0) { | ||
291 | + /* Uncontended. */ | ||
292 | + trace_qemu_co_mutex_lock_uncontended(mutex, self); | ||
293 | + } else { | ||
294 | + qemu_co_mutex_lock_slowpath(mutex); | ||
295 | + } | ||
296 | + mutex->holder = self; | ||
297 | + self->locks_held++; | ||
298 | +} | ||
299 | + | ||
300 | void coroutine_fn qemu_co_mutex_unlock(CoMutex *mutex) | ||
301 | { | ||
302 | Coroutine *self = qemu_coroutine_self(); | ||
303 | |||
304 | trace_qemu_co_mutex_unlock_entry(mutex, self); | ||
305 | |||
306 | - assert(mutex->locked == true); | ||
307 | + assert(mutex->locked); | ||
308 | assert(mutex->holder == self); | ||
309 | assert(qemu_in_coroutine()); | ||
310 | |||
311 | - mutex->locked = false; | ||
312 | mutex->holder = NULL; | ||
313 | self->locks_held--; | ||
314 | - qemu_co_queue_next(&mutex->queue); | ||
315 | + if (atomic_fetch_dec(&mutex->locked) == 1) { | ||
316 | + /* No waiting qemu_co_mutex_lock(). Pfew, that was easy! */ | ||
317 | + return; | ||
318 | + } | ||
319 | + | ||
320 | + for (;;) { | ||
321 | + CoWaitRecord *to_wake = pop_waiter(mutex); | ||
322 | + unsigned our_handoff; | ||
323 | + | ||
324 | + if (to_wake) { | ||
325 | + Coroutine *co = to_wake->co; | ||
326 | + aio_co_wake(co); | ||
327 | + break; | ||
328 | + } | ||
329 | + | ||
330 | + /* Some concurrent lock() is in progress (we know this because | ||
331 | + * mutex->locked was >1) but it hasn't yet put itself on the wait | ||
332 | + * queue. Pick a sequence number for the handoff protocol (not 0). | ||
333 | + */ | ||
334 | + if (++mutex->sequence == 0) { | ||
335 | + mutex->sequence = 1; | ||
336 | + } | ||
337 | + | ||
338 | + our_handoff = mutex->sequence; | ||
339 | + atomic_mb_set(&mutex->handoff, our_handoff); | ||
340 | + if (!has_waiters(mutex)) { | ||
341 | + /* The concurrent lock has not added itself yet, so it | ||
342 | + * will be able to pick our handoff. | ||
343 | + */ | ||
344 | + break; | ||
345 | + } | ||
346 | + | ||
347 | + /* Try to do the handoff protocol ourselves; if somebody else has | ||
348 | + * already taken it, however, we're done and they're responsible. | ||
349 | + */ | ||
350 | + if (atomic_cmpxchg(&mutex->handoff, our_handoff, 0) != our_handoff) { | ||
351 | + break; | ||
352 | + } | ||
353 | + } | ||
354 | |||
355 | trace_qemu_co_mutex_unlock_return(mutex, self); | ||
356 | } | ||
357 | diff --git a/util/trace-events b/util/trace-events | ||
358 | index XXXXXXX..XXXXXXX 100644 | ||
359 | --- a/util/trace-events | ||
360 | +++ b/util/trace-events | ||
361 | @@ -XXX,XX +XXX,XX @@ qemu_coroutine_terminate(void *co) "self %p" | ||
362 | |||
363 | # util/qemu-coroutine-lock.c | ||
364 | qemu_co_queue_run_restart(void *co) "co %p" | ||
365 | +qemu_co_mutex_lock_uncontended(void *mutex, void *self) "mutex %p self %p" | ||
366 | qemu_co_mutex_lock_entry(void *mutex, void *self) "mutex %p self %p" | ||
367 | qemu_co_mutex_lock_return(void *mutex, void *self) "mutex %p self %p" | ||
368 | qemu_co_mutex_unlock_entry(void *mutex, void *self) "mutex %p self %p" | ||
369 | -- | ||
370 | 2.9.3 | ||
371 | |||
372 | diff view generated by jsdifflib |
1 | From: Paolo Bonzini <pbonzini@redhat.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | Python test scripts that use unittest consist of multiple tests. | 3 | Running a very small critical section on pthread_mutex_t and CoMutex |
4 | unittest.main allows selecting which tests to run, but currently this | 4 | shows that pthread_mutex_t is much faster because it doesn't actually |
5 | is not possible because the iotests wrapper ignores sys.argv. | 5 | go to sleep. What happens is that the critical section is shorter |
6 | than the latency of entering the kernel and thus FUTEX_WAIT always | ||
7 | fails. With CoMutex there is no such latency but you still want to | ||
8 | avoid wait and wakeup. So introduce it artificially. | ||
6 | 9 | ||
7 | unittest.main command line options also allow the user to pick the | 10 | This only works with one waiters; because CoMutex is fair, it will |
8 | desired options for verbosity, failfast mode, etc. While "-d" is | 11 | always have more waits and wakeups than a pthread_mutex_t. |
9 | currently translated to "-v", it also enables extra debug output, | ||
10 | and other options are not available at all. | ||
11 | |||
12 | These command line options only work if the unittest.main testRunner | ||
13 | argument is a type, rather than a TestRunner instance. Therefore, pass | ||
14 | the class name and "verbosity" argument to unittest.main, and adjust for | ||
15 | the different default warnings between TextTestRunner and unittest.main. | ||
16 | 12 | ||
17 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> | 13 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
18 | Reviewed-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 14 | Reviewed-by: Fam Zheng <famz@redhat.com> |
19 | Tested-by: Emanuele Giuseppe Esposito <eesposit@redhat.com> | 15 | Message-id: 20170213181244.16297-3-pbonzini@redhat.com |
20 | Message-Id: <20210323181928.311862-3-pbonzini@redhat.com> | 16 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
21 | Message-Id: <20210503110110.476887-3-pbonzini@redhat.com> | ||
22 | Signed-off-by: Max Reitz <mreitz@redhat.com> | ||
23 | --- | 17 | --- |
24 | tests/qemu-iotests/iotests.py | 14 +++++++++----- | 18 | include/qemu/coroutine.h | 5 +++++ |
25 | 1 file changed, 9 insertions(+), 5 deletions(-) | 19 | util/qemu-coroutine-lock.c | 51 ++++++++++++++++++++++++++++++++++++++++------ |
20 | util/qemu-coroutine.c | 2 +- | ||
21 | 3 files changed, 51 insertions(+), 7 deletions(-) | ||
26 | 22 | ||
27 | diff --git a/tests/qemu-iotests/iotests.py b/tests/qemu-iotests/iotests.py | 23 | diff --git a/include/qemu/coroutine.h b/include/qemu/coroutine.h |
28 | index XXXXXXX..XXXXXXX 100644 | 24 | index XXXXXXX..XXXXXXX 100644 |
29 | --- a/tests/qemu-iotests/iotests.py | 25 | --- a/include/qemu/coroutine.h |
30 | +++ b/tests/qemu-iotests/iotests.py | 26 | +++ b/include/qemu/coroutine.h |
31 | @@ -XXX,XX +XXX,XX @@ def __init__(self, stream: Optional[TextIO] = None, | 27 | @@ -XXX,XX +XXX,XX @@ typedef struct CoMutex { |
32 | resultclass=resultclass, | 28 | */ |
33 | **kwargs) | 29 | unsigned locked; |
34 | 30 | ||
35 | -def execute_unittest(debug=False): | 31 | + /* Context that is holding the lock. Useful to avoid spinning |
36 | +def execute_unittest(argv: List[str], debug: bool = False) -> None: | 32 | + * when two coroutines on the same AioContext try to get the lock. :) |
37 | """Executes unittests within the calling module.""" | 33 | + */ |
38 | 34 | + AioContext *ctx; | |
39 | - verbosity = 2 if debug else 1 | 35 | + |
40 | - runner = ReproducibleTestRunner(verbosity=verbosity) | 36 | /* A queue of waiters. Elements are added atomically in front of |
41 | - unittest.main(testRunner=runner) | 37 | * from_push. to_pop is only populated, and popped from, by whoever |
42 | + # Some tests have warnings, especially ResourceWarnings for unclosed | 38 | * is in charge of the next wakeup. This can be an unlocker or, |
43 | + # files and sockets. Ignore them for now to ensure reproducibility of | 39 | diff --git a/util/qemu-coroutine-lock.c b/util/qemu-coroutine-lock.c |
44 | + # the test output. | 40 | index XXXXXXX..XXXXXXX 100644 |
45 | + unittest.main(argv=argv, | 41 | --- a/util/qemu-coroutine-lock.c |
46 | + testRunner=ReproducibleTestRunner, | 42 | +++ b/util/qemu-coroutine-lock.c |
47 | + verbosity=2 if debug else 1, | 43 | @@ -XXX,XX +XXX,XX @@ |
48 | + warnings=None if sys.warnoptions else 'ignore') | 44 | #include "qemu-common.h" |
49 | 45 | #include "qemu/coroutine.h" | |
50 | def execute_setup_common(supported_fmts: Sequence[str] = (), | 46 | #include "qemu/coroutine_int.h" |
51 | supported_platforms: Sequence[str] = (), | 47 | +#include "qemu/processor.h" |
52 | @@ -XXX,XX +XXX,XX @@ def execute_test(*args, test_function=None, **kwargs): | 48 | #include "qemu/queue.h" |
53 | 49 | #include "block/aio.h" | |
54 | debug = execute_setup_common(*args, **kwargs) | 50 | #include "trace.h" |
55 | if not test_function: | 51 | @@ -XXX,XX +XXX,XX @@ void qemu_co_mutex_init(CoMutex *mutex) |
56 | - execute_unittest(debug) | 52 | memset(mutex, 0, sizeof(*mutex)); |
57 | + execute_unittest(sys.argv, debug) | 53 | } |
58 | else: | 54 | |
59 | test_function() | 55 | -static void coroutine_fn qemu_co_mutex_lock_slowpath(CoMutex *mutex) |
56 | +static void coroutine_fn qemu_co_mutex_wake(CoMutex *mutex, Coroutine *co) | ||
57 | +{ | ||
58 | + /* Read co before co->ctx; pairs with smp_wmb() in | ||
59 | + * qemu_coroutine_enter(). | ||
60 | + */ | ||
61 | + smp_read_barrier_depends(); | ||
62 | + mutex->ctx = co->ctx; | ||
63 | + aio_co_wake(co); | ||
64 | +} | ||
65 | + | ||
66 | +static void coroutine_fn qemu_co_mutex_lock_slowpath(AioContext *ctx, | ||
67 | + CoMutex *mutex) | ||
68 | { | ||
69 | Coroutine *self = qemu_coroutine_self(); | ||
70 | CoWaitRecord w; | ||
71 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn qemu_co_mutex_lock_slowpath(CoMutex *mutex) | ||
72 | if (co == self) { | ||
73 | /* We got the lock ourselves! */ | ||
74 | assert(to_wake == &w); | ||
75 | + mutex->ctx = ctx; | ||
76 | return; | ||
77 | } | ||
78 | |||
79 | - aio_co_wake(co); | ||
80 | + qemu_co_mutex_wake(mutex, co); | ||
81 | } | ||
82 | |||
83 | qemu_coroutine_yield(); | ||
84 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn qemu_co_mutex_lock_slowpath(CoMutex *mutex) | ||
85 | |||
86 | void coroutine_fn qemu_co_mutex_lock(CoMutex *mutex) | ||
87 | { | ||
88 | + AioContext *ctx = qemu_get_current_aio_context(); | ||
89 | Coroutine *self = qemu_coroutine_self(); | ||
90 | + int waiters, i; | ||
91 | |||
92 | - if (atomic_fetch_inc(&mutex->locked) == 0) { | ||
93 | + /* Running a very small critical section on pthread_mutex_t and CoMutex | ||
94 | + * shows that pthread_mutex_t is much faster because it doesn't actually | ||
95 | + * go to sleep. What happens is that the critical section is shorter | ||
96 | + * than the latency of entering the kernel and thus FUTEX_WAIT always | ||
97 | + * fails. With CoMutex there is no such latency but you still want to | ||
98 | + * avoid wait and wakeup. So introduce it artificially. | ||
99 | + */ | ||
100 | + i = 0; | ||
101 | +retry_fast_path: | ||
102 | + waiters = atomic_cmpxchg(&mutex->locked, 0, 1); | ||
103 | + if (waiters != 0) { | ||
104 | + while (waiters == 1 && ++i < 1000) { | ||
105 | + if (atomic_read(&mutex->ctx) == ctx) { | ||
106 | + break; | ||
107 | + } | ||
108 | + if (atomic_read(&mutex->locked) == 0) { | ||
109 | + goto retry_fast_path; | ||
110 | + } | ||
111 | + cpu_relax(); | ||
112 | + } | ||
113 | + waiters = atomic_fetch_inc(&mutex->locked); | ||
114 | + } | ||
115 | + | ||
116 | + if (waiters == 0) { | ||
117 | /* Uncontended. */ | ||
118 | trace_qemu_co_mutex_lock_uncontended(mutex, self); | ||
119 | + mutex->ctx = ctx; | ||
120 | } else { | ||
121 | - qemu_co_mutex_lock_slowpath(mutex); | ||
122 | + qemu_co_mutex_lock_slowpath(ctx, mutex); | ||
123 | } | ||
124 | mutex->holder = self; | ||
125 | self->locks_held++; | ||
126 | @@ -XXX,XX +XXX,XX @@ void coroutine_fn qemu_co_mutex_unlock(CoMutex *mutex) | ||
127 | assert(mutex->holder == self); | ||
128 | assert(qemu_in_coroutine()); | ||
129 | |||
130 | + mutex->ctx = NULL; | ||
131 | mutex->holder = NULL; | ||
132 | self->locks_held--; | ||
133 | if (atomic_fetch_dec(&mutex->locked) == 1) { | ||
134 | @@ -XXX,XX +XXX,XX @@ void coroutine_fn qemu_co_mutex_unlock(CoMutex *mutex) | ||
135 | unsigned our_handoff; | ||
136 | |||
137 | if (to_wake) { | ||
138 | - Coroutine *co = to_wake->co; | ||
139 | - aio_co_wake(co); | ||
140 | + qemu_co_mutex_wake(mutex, to_wake->co); | ||
141 | break; | ||
142 | } | ||
143 | |||
144 | diff --git a/util/qemu-coroutine.c b/util/qemu-coroutine.c | ||
145 | index XXXXXXX..XXXXXXX 100644 | ||
146 | --- a/util/qemu-coroutine.c | ||
147 | +++ b/util/qemu-coroutine.c | ||
148 | @@ -XXX,XX +XXX,XX @@ void qemu_coroutine_enter(Coroutine *co) | ||
149 | co->ctx = qemu_get_current_aio_context(); | ||
150 | |||
151 | /* Store co->ctx before anything that stores co. Matches | ||
152 | - * barrier in aio_co_wake. | ||
153 | + * barrier in aio_co_wake and qemu_co_mutex_wake. | ||
154 | */ | ||
155 | smp_wmb(); | ||
60 | 156 | ||
61 | -- | 157 | -- |
62 | 2.31.1 | 158 | 2.9.3 |
63 | 159 | ||
64 | 160 | diff view generated by jsdifflib |
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | We don't need this extra logic: it doesn't make code simpler. | 3 | Add two implementations of the same benchmark as the previous patch, |
4 | 4 | but using pthreads. One uses a normal QemuMutex, the other is Linux | |
5 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 5 | only and implements a fair mutex based on MCS locks and futexes. |
6 | Reviewed-by: Max Reitz <mreitz@redhat.com> | 6 | This shows that the slower performance of the 5-thread case is due to |
7 | Message-Id: <20210506090621.11848-8-vsementsov@virtuozzo.com> | 7 | the fairness of CoMutex, rather than to coroutines. If fairness does |
8 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | 8 | not matter, as is the case with two threads, CoMutex can actually be |
9 | Signed-off-by: Max Reitz <mreitz@redhat.com> | 9 | faster than pthreads. |
10 | |||
11 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> | ||
12 | Reviewed-by: Fam Zheng <famz@redhat.com> | ||
13 | Message-id: 20170213181244.16297-4-pbonzini@redhat.com | ||
14 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
10 | --- | 15 | --- |
11 | tests/unit/test-write-threshold.c | 20 +++----------------- | 16 | tests/test-aio-multithread.c | 164 +++++++++++++++++++++++++++++++++++++++++++ |
12 | 1 file changed, 3 insertions(+), 17 deletions(-) | 17 | 1 file changed, 164 insertions(+) |
13 | 18 | ||
14 | diff --git a/tests/unit/test-write-threshold.c b/tests/unit/test-write-threshold.c | 19 | diff --git a/tests/test-aio-multithread.c b/tests/test-aio-multithread.c |
15 | index XXXXXXX..XXXXXXX 100644 | 20 | index XXXXXXX..XXXXXXX 100644 |
16 | --- a/tests/unit/test-write-threshold.c | 21 | --- a/tests/test-aio-multithread.c |
17 | +++ b/tests/unit/test-write-threshold.c | 22 | +++ b/tests/test-aio-multithread.c |
18 | @@ -XXX,XX +XXX,XX @@ static void test_threshold_trigger(void) | 23 | @@ -XXX,XX +XXX,XX @@ static void test_multi_co_mutex_2_30(void) |
19 | g_assert_cmpuint(bdrv_write_threshold_get(&bs), ==, 0); | 24 | test_multi_co_mutex(2, 30); |
20 | } | 25 | } |
21 | 26 | ||
22 | -typedef struct TestStruct { | 27 | +/* Same test with fair mutexes, for performance comparison. */ |
23 | - const char *name; | 28 | + |
24 | - void (*func)(void); | 29 | +#ifdef CONFIG_LINUX |
25 | -} TestStruct; | 30 | +#include "qemu/futex.h" |
26 | - | 31 | + |
32 | +/* The nodes for the mutex reside in this structure (on which we try to avoid | ||
33 | + * false sharing). The head of the mutex is in the "mutex_head" variable. | ||
34 | + */ | ||
35 | +static struct { | ||
36 | + int next, locked; | ||
37 | + int padding[14]; | ||
38 | +} nodes[NUM_CONTEXTS] __attribute__((__aligned__(64))); | ||
39 | + | ||
40 | +static int mutex_head = -1; | ||
41 | + | ||
42 | +static void mcs_mutex_lock(void) | ||
43 | +{ | ||
44 | + int prev; | ||
45 | + | ||
46 | + nodes[id].next = -1; | ||
47 | + nodes[id].locked = 1; | ||
48 | + prev = atomic_xchg(&mutex_head, id); | ||
49 | + if (prev != -1) { | ||
50 | + atomic_set(&nodes[prev].next, id); | ||
51 | + qemu_futex_wait(&nodes[id].locked, 1); | ||
52 | + } | ||
53 | +} | ||
54 | + | ||
55 | +static void mcs_mutex_unlock(void) | ||
56 | +{ | ||
57 | + int next; | ||
58 | + if (nodes[id].next == -1) { | ||
59 | + if (atomic_read(&mutex_head) == id && | ||
60 | + atomic_cmpxchg(&mutex_head, id, -1) == id) { | ||
61 | + /* Last item in the list, exit. */ | ||
62 | + return; | ||
63 | + } | ||
64 | + while (atomic_read(&nodes[id].next) == -1) { | ||
65 | + /* mcs_mutex_lock did the xchg, but has not updated | ||
66 | + * nodes[prev].next yet. | ||
67 | + */ | ||
68 | + } | ||
69 | + } | ||
70 | + | ||
71 | + /* Wake up the next in line. */ | ||
72 | + next = nodes[id].next; | ||
73 | + nodes[next].locked = 0; | ||
74 | + qemu_futex_wake(&nodes[next].locked, 1); | ||
75 | +} | ||
76 | + | ||
77 | +static void test_multi_fair_mutex_entry(void *opaque) | ||
78 | +{ | ||
79 | + while (!atomic_mb_read(&now_stopping)) { | ||
80 | + mcs_mutex_lock(); | ||
81 | + counter++; | ||
82 | + mcs_mutex_unlock(); | ||
83 | + atomic_inc(&atomic_counter); | ||
84 | + } | ||
85 | + atomic_dec(&running); | ||
86 | +} | ||
87 | + | ||
88 | +static void test_multi_fair_mutex(int threads, int seconds) | ||
89 | +{ | ||
90 | + int i; | ||
91 | + | ||
92 | + assert(mutex_head == -1); | ||
93 | + counter = 0; | ||
94 | + atomic_counter = 0; | ||
95 | + now_stopping = false; | ||
96 | + | ||
97 | + create_aio_contexts(); | ||
98 | + assert(threads <= NUM_CONTEXTS); | ||
99 | + running = threads; | ||
100 | + for (i = 0; i < threads; i++) { | ||
101 | + Coroutine *co1 = qemu_coroutine_create(test_multi_fair_mutex_entry, NULL); | ||
102 | + aio_co_schedule(ctx[i], co1); | ||
103 | + } | ||
104 | + | ||
105 | + g_usleep(seconds * 1000000); | ||
106 | + | ||
107 | + atomic_mb_set(&now_stopping, true); | ||
108 | + while (running > 0) { | ||
109 | + g_usleep(100000); | ||
110 | + } | ||
111 | + | ||
112 | + join_aio_contexts(); | ||
113 | + g_test_message("%d iterations/second\n", counter / seconds); | ||
114 | + g_assert_cmpint(counter, ==, atomic_counter); | ||
115 | +} | ||
116 | + | ||
117 | +static void test_multi_fair_mutex_1(void) | ||
118 | +{ | ||
119 | + test_multi_fair_mutex(NUM_CONTEXTS, 1); | ||
120 | +} | ||
121 | + | ||
122 | +static void test_multi_fair_mutex_10(void) | ||
123 | +{ | ||
124 | + test_multi_fair_mutex(NUM_CONTEXTS, 10); | ||
125 | +} | ||
126 | +#endif | ||
127 | + | ||
128 | +/* Same test with pthread mutexes, for performance comparison and | ||
129 | + * portability. */ | ||
130 | + | ||
131 | +static QemuMutex mutex; | ||
132 | + | ||
133 | +static void test_multi_mutex_entry(void *opaque) | ||
134 | +{ | ||
135 | + while (!atomic_mb_read(&now_stopping)) { | ||
136 | + qemu_mutex_lock(&mutex); | ||
137 | + counter++; | ||
138 | + qemu_mutex_unlock(&mutex); | ||
139 | + atomic_inc(&atomic_counter); | ||
140 | + } | ||
141 | + atomic_dec(&running); | ||
142 | +} | ||
143 | + | ||
144 | +static void test_multi_mutex(int threads, int seconds) | ||
145 | +{ | ||
146 | + int i; | ||
147 | + | ||
148 | + qemu_mutex_init(&mutex); | ||
149 | + counter = 0; | ||
150 | + atomic_counter = 0; | ||
151 | + now_stopping = false; | ||
152 | + | ||
153 | + create_aio_contexts(); | ||
154 | + assert(threads <= NUM_CONTEXTS); | ||
155 | + running = threads; | ||
156 | + for (i = 0; i < threads; i++) { | ||
157 | + Coroutine *co1 = qemu_coroutine_create(test_multi_mutex_entry, NULL); | ||
158 | + aio_co_schedule(ctx[i], co1); | ||
159 | + } | ||
160 | + | ||
161 | + g_usleep(seconds * 1000000); | ||
162 | + | ||
163 | + atomic_mb_set(&now_stopping, true); | ||
164 | + while (running > 0) { | ||
165 | + g_usleep(100000); | ||
166 | + } | ||
167 | + | ||
168 | + join_aio_contexts(); | ||
169 | + g_test_message("%d iterations/second\n", counter / seconds); | ||
170 | + g_assert_cmpint(counter, ==, atomic_counter); | ||
171 | +} | ||
172 | + | ||
173 | +static void test_multi_mutex_1(void) | ||
174 | +{ | ||
175 | + test_multi_mutex(NUM_CONTEXTS, 1); | ||
176 | +} | ||
177 | + | ||
178 | +static void test_multi_mutex_10(void) | ||
179 | +{ | ||
180 | + test_multi_mutex(NUM_CONTEXTS, 10); | ||
181 | +} | ||
182 | + | ||
183 | /* End of tests. */ | ||
27 | 184 | ||
28 | int main(int argc, char **argv) | 185 | int main(int argc, char **argv) |
29 | { | 186 | @@ -XXX,XX +XXX,XX @@ int main(int argc, char **argv) |
30 | - size_t i; | 187 | g_test_add_func("/aio/multi/schedule", test_multi_co_schedule_1); |
31 | - TestStruct tests[] = { | 188 | g_test_add_func("/aio/multi/mutex/contended", test_multi_co_mutex_1); |
32 | - { "/write-threshold/not-trigger", | 189 | g_test_add_func("/aio/multi/mutex/handoff", test_multi_co_mutex_2_3); |
33 | - test_threshold_not_trigger }, | 190 | +#ifdef CONFIG_LINUX |
34 | - { "/write-threshold/trigger", | 191 | + g_test_add_func("/aio/multi/mutex/mcs", test_multi_fair_mutex_1); |
35 | - test_threshold_trigger }, | 192 | +#endif |
36 | - { NULL, NULL } | 193 | + g_test_add_func("/aio/multi/mutex/pthread", test_multi_mutex_1); |
37 | - }; | 194 | } else { |
38 | - | 195 | g_test_add_func("/aio/multi/schedule", test_multi_co_schedule_10); |
39 | g_test_init(&argc, &argv, NULL); | 196 | g_test_add_func("/aio/multi/mutex/contended", test_multi_co_mutex_10); |
40 | - for (i = 0; tests[i].name != NULL; i++) { | 197 | g_test_add_func("/aio/multi/mutex/handoff", test_multi_co_mutex_2_30); |
41 | - g_test_add_func(tests[i].name, tests[i].func); | 198 | +#ifdef CONFIG_LINUX |
42 | - } | 199 | + g_test_add_func("/aio/multi/mutex/mcs", test_multi_fair_mutex_10); |
43 | + g_test_add_func("/write-threshold/not-trigger", test_threshold_not_trigger); | 200 | +#endif |
44 | + g_test_add_func("/write-threshold/trigger", test_threshold_trigger); | 201 | + g_test_add_func("/aio/multi/mutex/pthread", test_multi_mutex_10); |
45 | + | 202 | } |
46 | return g_test_run(); | 203 | return g_test_run(); |
47 | } | 204 | } |
48 | -- | 205 | -- |
49 | 2.31.1 | 206 | 2.9.3 |
50 | 207 | ||
51 | 208 | diff view generated by jsdifflib |
1 | From: Connor Kuehl <ckuehl@redhat.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | The deprecation message in the expected output has technically been | 3 | This will avoid forward references in the next patch. It is also |
4 | wrong since the wrong version of a patch was applied to it. Because of | 4 | more logical because CoQueue is not anymore the basic primitive. |
5 | this, the test fails. Correct the expected output so that it passes. | ||
6 | 5 | ||
7 | Signed-off-by: Connor Kuehl <ckuehl@redhat.com> | 6 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
8 | Reviewed-by: Max Reitz <mreitz@redhat.com> | 7 | Reviewed-by: Fam Zheng <famz@redhat.com> |
9 | Reviewed-by: Stefano Garzarella <sgarzare@redhat.com> | 8 | Message-id: 20170213181244.16297-5-pbonzini@redhat.com |
10 | Message-Id: <20210421212343.85524-2-ckuehl@redhat.com> | 9 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
11 | Signed-off-by: Max Reitz <mreitz@redhat.com> | ||
12 | --- | 10 | --- |
13 | tests/qemu-iotests/231.out | 4 +--- | 11 | include/qemu/coroutine.h | 89 ++++++++++++++++++++++++------------------------ |
14 | 1 file changed, 1 insertion(+), 3 deletions(-) | 12 | 1 file changed, 44 insertions(+), 45 deletions(-) |
15 | 13 | ||
16 | diff --git a/tests/qemu-iotests/231.out b/tests/qemu-iotests/231.out | 14 | diff --git a/include/qemu/coroutine.h b/include/qemu/coroutine.h |
17 | index XXXXXXX..XXXXXXX 100644 | 15 | index XXXXXXX..XXXXXXX 100644 |
18 | --- a/tests/qemu-iotests/231.out | 16 | --- a/include/qemu/coroutine.h |
19 | +++ b/tests/qemu-iotests/231.out | 17 | +++ b/include/qemu/coroutine.h |
20 | @@ -XXX,XX +XXX,XX @@ | 18 | @@ -XXX,XX +XXX,XX @@ bool qemu_in_coroutine(void); |
21 | QA output created by 231 | 19 | */ |
22 | -qemu-img: RBD options encoded in the filename as keyvalue pairs is deprecated. Future versions may cease to parse these options in the future. | 20 | bool qemu_coroutine_entered(Coroutine *co); |
23 | +qemu-img: warning: RBD options encoded in the filename as keyvalue pairs is deprecated | 21 | |
24 | unable to get monitor info from DNS SRV with service name: ceph-mon | 22 | - |
25 | -no monitors specified to connect to. | 23 | -/** |
26 | qemu-img: Could not open 'json:{'file.driver':'rbd','file.filename':'rbd:rbd/bogus:conf=BOGUS_CONF'}': error connecting: No such file or directory | 24 | - * CoQueues are a mechanism to queue coroutines in order to continue executing |
27 | unable to get monitor info from DNS SRV with service name: ceph-mon | 25 | - * them later. They provide the fundamental primitives on which coroutine locks |
28 | -no monitors specified to connect to. | 26 | - * are built. |
29 | qemu-img: Could not open 'json:{'file.driver':'rbd','file.pool':'rbd','file.image':'bogus','file.conf':'BOGUS_CONF'}': error connecting: No such file or directory | 27 | - */ |
30 | *** done | 28 | -typedef struct CoQueue { |
29 | - QSIMPLEQ_HEAD(, Coroutine) entries; | ||
30 | -} CoQueue; | ||
31 | - | ||
32 | -/** | ||
33 | - * Initialise a CoQueue. This must be called before any other operation is used | ||
34 | - * on the CoQueue. | ||
35 | - */ | ||
36 | -void qemu_co_queue_init(CoQueue *queue); | ||
37 | - | ||
38 | -/** | ||
39 | - * Adds the current coroutine to the CoQueue and transfers control to the | ||
40 | - * caller of the coroutine. | ||
41 | - */ | ||
42 | -void coroutine_fn qemu_co_queue_wait(CoQueue *queue); | ||
43 | - | ||
44 | -/** | ||
45 | - * Restarts the next coroutine in the CoQueue and removes it from the queue. | ||
46 | - * | ||
47 | - * Returns true if a coroutine was restarted, false if the queue is empty. | ||
48 | - */ | ||
49 | -bool coroutine_fn qemu_co_queue_next(CoQueue *queue); | ||
50 | - | ||
51 | -/** | ||
52 | - * Restarts all coroutines in the CoQueue and leaves the queue empty. | ||
53 | - */ | ||
54 | -void coroutine_fn qemu_co_queue_restart_all(CoQueue *queue); | ||
55 | - | ||
56 | -/** | ||
57 | - * Enter the next coroutine in the queue | ||
58 | - */ | ||
59 | -bool qemu_co_enter_next(CoQueue *queue); | ||
60 | - | ||
61 | -/** | ||
62 | - * Checks if the CoQueue is empty. | ||
63 | - */ | ||
64 | -bool qemu_co_queue_empty(CoQueue *queue); | ||
65 | - | ||
66 | - | ||
67 | /** | ||
68 | * Provides a mutex that can be used to synchronise coroutines | ||
69 | */ | ||
70 | @@ -XXX,XX +XXX,XX @@ void coroutine_fn qemu_co_mutex_lock(CoMutex *mutex); | ||
71 | */ | ||
72 | void coroutine_fn qemu_co_mutex_unlock(CoMutex *mutex); | ||
73 | |||
74 | + | ||
75 | +/** | ||
76 | + * CoQueues are a mechanism to queue coroutines in order to continue executing | ||
77 | + * them later. | ||
78 | + */ | ||
79 | +typedef struct CoQueue { | ||
80 | + QSIMPLEQ_HEAD(, Coroutine) entries; | ||
81 | +} CoQueue; | ||
82 | + | ||
83 | +/** | ||
84 | + * Initialise a CoQueue. This must be called before any other operation is used | ||
85 | + * on the CoQueue. | ||
86 | + */ | ||
87 | +void qemu_co_queue_init(CoQueue *queue); | ||
88 | + | ||
89 | +/** | ||
90 | + * Adds the current coroutine to the CoQueue and transfers control to the | ||
91 | + * caller of the coroutine. | ||
92 | + */ | ||
93 | +void coroutine_fn qemu_co_queue_wait(CoQueue *queue); | ||
94 | + | ||
95 | +/** | ||
96 | + * Restarts the next coroutine in the CoQueue and removes it from the queue. | ||
97 | + * | ||
98 | + * Returns true if a coroutine was restarted, false if the queue is empty. | ||
99 | + */ | ||
100 | +bool coroutine_fn qemu_co_queue_next(CoQueue *queue); | ||
101 | + | ||
102 | +/** | ||
103 | + * Restarts all coroutines in the CoQueue and leaves the queue empty. | ||
104 | + */ | ||
105 | +void coroutine_fn qemu_co_queue_restart_all(CoQueue *queue); | ||
106 | + | ||
107 | +/** | ||
108 | + * Enter the next coroutine in the queue | ||
109 | + */ | ||
110 | +bool qemu_co_enter_next(CoQueue *queue); | ||
111 | + | ||
112 | +/** | ||
113 | + * Checks if the CoQueue is empty. | ||
114 | + */ | ||
115 | +bool qemu_co_queue_empty(CoQueue *queue); | ||
116 | + | ||
117 | + | ||
118 | typedef struct CoRwlock { | ||
119 | bool writer; | ||
120 | int reader; | ||
31 | -- | 121 | -- |
32 | 2.31.1 | 122 | 2.9.3 |
33 | 123 | ||
34 | 124 | diff view generated by jsdifflib |
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | If mirror is READY than cancel operation is not discarding the whole | 3 | All that CoQueue needs in order to become thread-safe is help |
4 | result of the operation, but instead it's a documented way get a | 4 | from an external mutex. Add this to the API. |
5 | point-in-time snapshot of source disk. | 5 | |
6 | 6 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> | |
7 | So, we should not cancel any requests if mirror is READ and | 7 | Reviewed-by: Fam Zheng <famz@redhat.com> |
8 | force=false. Let's fix that case. | 8 | Message-id: 20170213181244.16297-6-pbonzini@redhat.com |
9 | 9 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | |
10 | Note, that bug that we have before this commit is not critical, as the | ||
11 | only .bdrv_cancel_in_flight implementation is nbd_cancel_in_flight() | ||
12 | and it cancels only requests waiting for reconnection, so it should be | ||
13 | rare case. | ||
14 | |||
15 | Fixes: 521ff8b779b11c394dbdc43f02e158dd99df308a | ||
16 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
17 | Message-Id: <20210421075858.40197-1-vsementsov@virtuozzo.com> | ||
18 | Signed-off-by: Max Reitz <mreitz@redhat.com> | ||
19 | --- | 10 | --- |
20 | include/block/block_int.h | 2 +- | 11 | include/qemu/coroutine.h | 8 +++++--- |
21 | include/qemu/job.h | 2 +- | 12 | block/backup.c | 2 +- |
22 | block/backup.c | 2 +- | 13 | block/io.c | 4 ++-- |
23 | block/mirror.c | 6 ++++-- | 14 | block/nbd-client.c | 2 +- |
24 | job.c | 2 +- | 15 | block/qcow2-cluster.c | 4 +--- |
25 | tests/qemu-iotests/264 | 2 +- | 16 | block/sheepdog.c | 2 +- |
26 | 6 files changed, 9 insertions(+), 7 deletions(-) | 17 | block/throttle-groups.c | 2 +- |
27 | 18 | hw/9pfs/9p.c | 2 +- | |
28 | diff --git a/include/block/block_int.h b/include/block/block_int.h | 19 | util/qemu-coroutine-lock.c | 24 +++++++++++++++++++++--- |
29 | index XXXXXXX..XXXXXXX 100644 | 20 | 9 files changed, 34 insertions(+), 16 deletions(-) |
30 | --- a/include/block/block_int.h | 21 | |
31 | +++ b/include/block/block_int.h | 22 | diff --git a/include/qemu/coroutine.h b/include/qemu/coroutine.h |
32 | @@ -XXX,XX +XXX,XX @@ struct BlockDriver { | 23 | index XXXXXXX..XXXXXXX 100644 |
33 | * of in-flight requests, so don't waste the time if possible. | 24 | --- a/include/qemu/coroutine.h |
34 | * | 25 | +++ b/include/qemu/coroutine.h |
35 | * One example usage is to avoid waiting for an nbd target node reconnect | 26 | @@ -XXX,XX +XXX,XX @@ void coroutine_fn qemu_co_mutex_unlock(CoMutex *mutex); |
36 | - * timeout during job-cancel. | 27 | |
37 | + * timeout during job-cancel with force=true. | 28 | /** |
38 | */ | 29 | * CoQueues are a mechanism to queue coroutines in order to continue executing |
39 | void (*bdrv_cancel_in_flight)(BlockDriverState *bs); | 30 | - * them later. |
40 | 31 | + * them later. They are similar to condition variables, but they need help | |
41 | diff --git a/include/qemu/job.h b/include/qemu/job.h | 32 | + * from an external mutex in order to maintain thread-safety. |
42 | index XXXXXXX..XXXXXXX 100644 | 33 | */ |
43 | --- a/include/qemu/job.h | 34 | typedef struct CoQueue { |
44 | +++ b/include/qemu/job.h | 35 | QSIMPLEQ_HEAD(, Coroutine) entries; |
45 | @@ -XXX,XX +XXX,XX @@ struct JobDriver { | 36 | @@ -XXX,XX +XXX,XX @@ void qemu_co_queue_init(CoQueue *queue); |
46 | /** | 37 | |
47 | * If the callback is not NULL, it will be invoked in job_cancel_async | 38 | /** |
48 | */ | 39 | * Adds the current coroutine to the CoQueue and transfers control to the |
49 | - void (*cancel)(Job *job); | 40 | - * caller of the coroutine. |
50 | + void (*cancel)(Job *job, bool force); | 41 | + * caller of the coroutine. The mutex is unlocked during the wait and |
51 | 42 | + * locked again afterwards. | |
52 | 43 | */ | |
53 | /** Called when the job is freed */ | 44 | -void coroutine_fn qemu_co_queue_wait(CoQueue *queue); |
45 | +void coroutine_fn qemu_co_queue_wait(CoQueue *queue, CoMutex *mutex); | ||
46 | |||
47 | /** | ||
48 | * Restarts the next coroutine in the CoQueue and removes it from the queue. | ||
54 | diff --git a/block/backup.c b/block/backup.c | 49 | diff --git a/block/backup.c b/block/backup.c |
55 | index XXXXXXX..XXXXXXX 100644 | 50 | index XXXXXXX..XXXXXXX 100644 |
56 | --- a/block/backup.c | 51 | --- a/block/backup.c |
57 | +++ b/block/backup.c | 52 | +++ b/block/backup.c |
58 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn backup_set_speed(BlockJob *job, int64_t speed) | 53 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn wait_for_overlapping_requests(BackupBlockJob *job, |
59 | } | 54 | retry = false; |
55 | QLIST_FOREACH(req, &job->inflight_reqs, list) { | ||
56 | if (end > req->start && start < req->end) { | ||
57 | - qemu_co_queue_wait(&req->wait_queue); | ||
58 | + qemu_co_queue_wait(&req->wait_queue, NULL); | ||
59 | retry = true; | ||
60 | break; | ||
61 | } | ||
62 | diff --git a/block/io.c b/block/io.c | ||
63 | index XXXXXXX..XXXXXXX 100644 | ||
64 | --- a/block/io.c | ||
65 | +++ b/block/io.c | ||
66 | @@ -XXX,XX +XXX,XX @@ static bool coroutine_fn wait_serialising_requests(BdrvTrackedRequest *self) | ||
67 | * (instead of producing a deadlock in the former case). */ | ||
68 | if (!req->waiting_for) { | ||
69 | self->waiting_for = req; | ||
70 | - qemu_co_queue_wait(&req->wait_queue); | ||
71 | + qemu_co_queue_wait(&req->wait_queue, NULL); | ||
72 | self->waiting_for = NULL; | ||
73 | retry = true; | ||
74 | waited = true; | ||
75 | @@ -XXX,XX +XXX,XX @@ int coroutine_fn bdrv_co_flush(BlockDriverState *bs) | ||
76 | |||
77 | /* Wait until any previous flushes are completed */ | ||
78 | while (bs->active_flush_req) { | ||
79 | - qemu_co_queue_wait(&bs->flush_queue); | ||
80 | + qemu_co_queue_wait(&bs->flush_queue, NULL); | ||
81 | } | ||
82 | |||
83 | bs->active_flush_req = true; | ||
84 | diff --git a/block/nbd-client.c b/block/nbd-client.c | ||
85 | index XXXXXXX..XXXXXXX 100644 | ||
86 | --- a/block/nbd-client.c | ||
87 | +++ b/block/nbd-client.c | ||
88 | @@ -XXX,XX +XXX,XX @@ static void nbd_coroutine_start(NBDClientSession *s, | ||
89 | /* Poor man semaphore. The free_sema is locked when no other request | ||
90 | * can be accepted, and unlocked after receiving one reply. */ | ||
91 | if (s->in_flight == MAX_NBD_REQUESTS) { | ||
92 | - qemu_co_queue_wait(&s->free_sema); | ||
93 | + qemu_co_queue_wait(&s->free_sema, NULL); | ||
94 | assert(s->in_flight < MAX_NBD_REQUESTS); | ||
95 | } | ||
96 | s->in_flight++; | ||
97 | diff --git a/block/qcow2-cluster.c b/block/qcow2-cluster.c | ||
98 | index XXXXXXX..XXXXXXX 100644 | ||
99 | --- a/block/qcow2-cluster.c | ||
100 | +++ b/block/qcow2-cluster.c | ||
101 | @@ -XXX,XX +XXX,XX @@ static int handle_dependencies(BlockDriverState *bs, uint64_t guest_offset, | ||
102 | if (bytes == 0) { | ||
103 | /* Wait for the dependency to complete. We need to recheck | ||
104 | * the free/allocated clusters when we continue. */ | ||
105 | - qemu_co_mutex_unlock(&s->lock); | ||
106 | - qemu_co_queue_wait(&old_alloc->dependent_requests); | ||
107 | - qemu_co_mutex_lock(&s->lock); | ||
108 | + qemu_co_queue_wait(&old_alloc->dependent_requests, &s->lock); | ||
109 | return -EAGAIN; | ||
110 | } | ||
111 | } | ||
112 | diff --git a/block/sheepdog.c b/block/sheepdog.c | ||
113 | index XXXXXXX..XXXXXXX 100644 | ||
114 | --- a/block/sheepdog.c | ||
115 | +++ b/block/sheepdog.c | ||
116 | @@ -XXX,XX +XXX,XX @@ static void wait_for_overlapping_aiocb(BDRVSheepdogState *s, SheepdogAIOCB *acb) | ||
117 | retry: | ||
118 | QLIST_FOREACH(cb, &s->inflight_aiocb_head, aiocb_siblings) { | ||
119 | if (AIOCBOverlapping(acb, cb)) { | ||
120 | - qemu_co_queue_wait(&s->overlapping_queue); | ||
121 | + qemu_co_queue_wait(&s->overlapping_queue, NULL); | ||
122 | goto retry; | ||
123 | } | ||
124 | } | ||
125 | diff --git a/block/throttle-groups.c b/block/throttle-groups.c | ||
126 | index XXXXXXX..XXXXXXX 100644 | ||
127 | --- a/block/throttle-groups.c | ||
128 | +++ b/block/throttle-groups.c | ||
129 | @@ -XXX,XX +XXX,XX @@ void coroutine_fn throttle_group_co_io_limits_intercept(BlockBackend *blk, | ||
130 | if (must_wait || blkp->pending_reqs[is_write]) { | ||
131 | blkp->pending_reqs[is_write]++; | ||
132 | qemu_mutex_unlock(&tg->lock); | ||
133 | - qemu_co_queue_wait(&blkp->throttled_reqs[is_write]); | ||
134 | + qemu_co_queue_wait(&blkp->throttled_reqs[is_write], NULL); | ||
135 | qemu_mutex_lock(&tg->lock); | ||
136 | blkp->pending_reqs[is_write]--; | ||
137 | } | ||
138 | diff --git a/hw/9pfs/9p.c b/hw/9pfs/9p.c | ||
139 | index XXXXXXX..XXXXXXX 100644 | ||
140 | --- a/hw/9pfs/9p.c | ||
141 | +++ b/hw/9pfs/9p.c | ||
142 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn v9fs_flush(void *opaque) | ||
143 | /* | ||
144 | * Wait for pdu to complete. | ||
145 | */ | ||
146 | - qemu_co_queue_wait(&cancel_pdu->complete); | ||
147 | + qemu_co_queue_wait(&cancel_pdu->complete, NULL); | ||
148 | cancel_pdu->cancelled = 0; | ||
149 | pdu_free(cancel_pdu); | ||
150 | } | ||
151 | diff --git a/util/qemu-coroutine-lock.c b/util/qemu-coroutine-lock.c | ||
152 | index XXXXXXX..XXXXXXX 100644 | ||
153 | --- a/util/qemu-coroutine-lock.c | ||
154 | +++ b/util/qemu-coroutine-lock.c | ||
155 | @@ -XXX,XX +XXX,XX @@ void qemu_co_queue_init(CoQueue *queue) | ||
156 | QSIMPLEQ_INIT(&queue->entries); | ||
60 | } | 157 | } |
61 | 158 | ||
62 | -static void backup_cancel(Job *job) | 159 | -void coroutine_fn qemu_co_queue_wait(CoQueue *queue) |
63 | +static void backup_cancel(Job *job, bool force) | 160 | +void coroutine_fn qemu_co_queue_wait(CoQueue *queue, CoMutex *mutex) |
64 | { | 161 | { |
65 | BackupBlockJob *s = container_of(job, BackupBlockJob, common.job); | 162 | Coroutine *self = qemu_coroutine_self(); |
66 | 163 | QSIMPLEQ_INSERT_TAIL(&queue->entries, self, co_queue_next); | |
67 | diff --git a/block/mirror.c b/block/mirror.c | 164 | + |
68 | index XXXXXXX..XXXXXXX 100644 | 165 | + if (mutex) { |
69 | --- a/block/mirror.c | 166 | + qemu_co_mutex_unlock(mutex); |
70 | +++ b/block/mirror.c | 167 | + } |
71 | @@ -XXX,XX +XXX,XX @@ static bool mirror_drained_poll(BlockJob *job) | 168 | + |
72 | return !!s->in_flight; | 169 | + /* There is no race condition here. Other threads will call |
73 | } | 170 | + * aio_co_schedule on our AioContext, which can reenter this |
74 | 171 | + * coroutine but only after this yield and after the main loop | |
75 | -static void mirror_cancel(Job *job) | 172 | + * has gone through the next iteration. |
76 | +static void mirror_cancel(Job *job, bool force) | 173 | + */ |
77 | { | 174 | qemu_coroutine_yield(); |
78 | MirrorBlockJob *s = container_of(job, MirrorBlockJob, common.job); | 175 | assert(qemu_in_coroutine()); |
79 | BlockDriverState *target = blk_bs(s->target); | 176 | + |
80 | 177 | + /* TODO: OSv implements wait morphing here, where the wakeup | |
81 | - bdrv_cancel_in_flight(target); | 178 | + * primitive automatically places the woken coroutine on the |
82 | + if (force || !job_is_ready(job)) { | 179 | + * mutex's queue. This avoids the thundering herd effect. |
83 | + bdrv_cancel_in_flight(target); | 180 | + */ |
181 | + if (mutex) { | ||
182 | + qemu_co_mutex_lock(mutex); | ||
84 | + } | 183 | + } |
85 | } | 184 | } |
86 | 185 | ||
87 | static const BlockJobDriver mirror_job_driver = { | 186 | /** |
88 | diff --git a/job.c b/job.c | 187 | @@ -XXX,XX +XXX,XX @@ void qemu_co_rwlock_rdlock(CoRwlock *lock) |
89 | index XXXXXXX..XXXXXXX 100644 | 188 | Coroutine *self = qemu_coroutine_self(); |
90 | --- a/job.c | 189 | |
91 | +++ b/job.c | 190 | while (lock->writer) { |
92 | @@ -XXX,XX +XXX,XX @@ static int job_finalize_single(Job *job) | 191 | - qemu_co_queue_wait(&lock->queue); |
93 | static void job_cancel_async(Job *job, bool force) | 192 | + qemu_co_queue_wait(&lock->queue, NULL); |
94 | { | 193 | } |
95 | if (job->driver->cancel) { | 194 | lock->reader++; |
96 | - job->driver->cancel(job); | 195 | self->locks_held++; |
97 | + job->driver->cancel(job, force); | 196 | @@ -XXX,XX +XXX,XX @@ void qemu_co_rwlock_wrlock(CoRwlock *lock) |
98 | } | 197 | Coroutine *self = qemu_coroutine_self(); |
99 | if (job->user_paused) { | 198 | |
100 | /* Do not call job_enter here, the caller will handle it. */ | 199 | while (lock->writer || lock->reader) { |
101 | diff --git a/tests/qemu-iotests/264 b/tests/qemu-iotests/264 | 200 | - qemu_co_queue_wait(&lock->queue); |
102 | index XXXXXXX..XXXXXXX 100755 | 201 | + qemu_co_queue_wait(&lock->queue, NULL); |
103 | --- a/tests/qemu-iotests/264 | 202 | } |
104 | +++ b/tests/qemu-iotests/264 | 203 | lock->writer = true; |
105 | @@ -XXX,XX +XXX,XX @@ class TestNbdReconnect(iotests.QMPTestCase): | 204 | self->locks_held++; |
106 | self.assert_qmp(result, 'return', {}) | ||
107 | |||
108 | def cancel_job(self): | ||
109 | - result = self.vm.qmp('block-job-cancel', device='drive0') | ||
110 | + result = self.vm.qmp('block-job-cancel', device='drive0', force=True) | ||
111 | self.assert_qmp(result, 'return', {}) | ||
112 | |||
113 | start_t = time.time() | ||
114 | -- | 205 | -- |
115 | 2.31.1 | 206 | 2.9.3 |
116 | 207 | ||
117 | 208 | diff view generated by jsdifflib |
1 | From: Paolo Bonzini <pbonzini@redhat.com> | 1 | From: Paolo Bonzini <pbonzini@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | Instead of buffering the test output into a StringIO, patch it on | 3 | This adds a CoMutex around the existing CoQueue. Because the write-side |
4 | the fly by wrapping sys.stdout's write method. This can be | 4 | can just take CoMutex, the old "writer" field is not necessary anymore. |
5 | done unconditionally, even if using -d, which makes execute_unittest | 5 | Instead of removing it altogether, count the number of pending writers |
6 | a bit simpler. | 6 | during a read-side critical section and forbid further readers from |
7 | entering. | ||
7 | 8 | ||
8 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> | 9 | Signed-off-by: Paolo Bonzini <pbonzini@redhat.com> |
9 | Reviewed-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | 10 | Reviewed-by: Fam Zheng <famz@redhat.com> |
10 | Tested-by: Emanuele Giuseppe Esposito <eesposit@redhat.com> | 11 | Message-id: 20170213181244.16297-7-pbonzini@redhat.com |
11 | Message-Id: <20210323181928.311862-2-pbonzini@redhat.com> | 12 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
12 | Message-Id: <20210503110110.476887-2-pbonzini@redhat.com> | ||
13 | Signed-off-by: Max Reitz <mreitz@redhat.com> | ||
14 | --- | 13 | --- |
15 | tests/qemu-iotests/240.out | 8 ++-- | 14 | include/qemu/coroutine.h | 3 ++- |
16 | tests/qemu-iotests/245.out | 8 ++-- | 15 | util/qemu-coroutine-lock.c | 35 ++++++++++++++++++++++++----------- |
17 | tests/qemu-iotests/295.out | 6 +-- | 16 | 2 files changed, 26 insertions(+), 12 deletions(-) |
18 | tests/qemu-iotests/296.out | 8 ++-- | ||
19 | tests/qemu-iotests/iotests.py | 70 ++++++++++++++++++++--------------- | ||
20 | 5 files changed, 56 insertions(+), 44 deletions(-) | ||
21 | 17 | ||
22 | diff --git a/tests/qemu-iotests/240.out b/tests/qemu-iotests/240.out | 18 | diff --git a/include/qemu/coroutine.h b/include/qemu/coroutine.h |
23 | index XXXXXXX..XXXXXXX 100644 | 19 | index XXXXXXX..XXXXXXX 100644 |
24 | --- a/tests/qemu-iotests/240.out | 20 | --- a/include/qemu/coroutine.h |
25 | +++ b/tests/qemu-iotests/240.out | 21 | +++ b/include/qemu/coroutine.h |
26 | @@ -XXX,XX +XXX,XX @@ | 22 | @@ -XXX,XX +XXX,XX @@ bool qemu_co_queue_empty(CoQueue *queue); |
27 | {"return": {}} | 23 | |
28 | {"execute": "blockdev-del", "arguments": {"node-name": "hd0"}} | 24 | |
29 | {"return": {}} | 25 | typedef struct CoRwlock { |
30 | -==Attach two SCSI disks using the same block device and the same iothread== | 26 | - bool writer; |
31 | +.==Attach two SCSI disks using the same block device and the same iothread== | 27 | + int pending_writer; |
32 | {"execute": "blockdev-add", "arguments": {"driver": "null-co", "node-name": "hd0", "read-only": true, "read-zeroes": true}} | 28 | int reader; |
33 | {"return": {}} | 29 | + CoMutex mutex; |
34 | {"execute": "object-add", "arguments": {"id": "iothread0", "qom-type": "iothread"}} | 30 | CoQueue queue; |
35 | @@ -XXX,XX +XXX,XX @@ | 31 | } CoRwlock; |
36 | {"return": {}} | 32 | |
37 | {"execute": "blockdev-del", "arguments": {"node-name": "hd0"}} | 33 | diff --git a/util/qemu-coroutine-lock.c b/util/qemu-coroutine-lock.c |
38 | {"return": {}} | ||
39 | -==Attach two SCSI disks using the same block device but different iothreads== | ||
40 | +.==Attach two SCSI disks using the same block device but different iothreads== | ||
41 | {"execute": "blockdev-add", "arguments": {"driver": "null-co", "node-name": "hd0", "read-only": true, "read-zeroes": true}} | ||
42 | {"return": {}} | ||
43 | {"execute": "object-add", "arguments": {"id": "iothread0", "qom-type": "iothread"}} | ||
44 | @@ -XXX,XX +XXX,XX @@ | ||
45 | {"return": {}} | ||
46 | {"execute": "blockdev-del", "arguments": {"node-name": "hd0"}} | ||
47 | {"return": {}} | ||
48 | -==Attach a SCSI disks using the same block device as a NBD server== | ||
49 | +.==Attach a SCSI disks using the same block device as a NBD server== | ||
50 | {"execute": "blockdev-add", "arguments": {"driver": "null-co", "node-name": "hd0", "read-only": true, "read-zeroes": true}} | ||
51 | {"return": {}} | ||
52 | {"execute": "nbd-server-start", "arguments": {"addr": {"data": {"path": "SOCK_DIR/PID-nbd.sock"}, "type": "unix"}}} | ||
53 | @@ -XXX,XX +XXX,XX @@ | ||
54 | {"return": {}} | ||
55 | {"execute": "device_add", "arguments": {"drive": "hd0", "driver": "scsi-hd", "id": "scsi-hd0"}} | ||
56 | {"return": {}} | ||
57 | -.... | ||
58 | +. | ||
59 | ---------------------------------------------------------------------- | ||
60 | Ran 4 tests | ||
61 | |||
62 | diff --git a/tests/qemu-iotests/245.out b/tests/qemu-iotests/245.out | ||
63 | index XXXXXXX..XXXXXXX 100644 | 34 | index XXXXXXX..XXXXXXX 100644 |
64 | --- a/tests/qemu-iotests/245.out | 35 | --- a/util/qemu-coroutine-lock.c |
65 | +++ b/tests/qemu-iotests/245.out | 36 | +++ b/util/qemu-coroutine-lock.c |
66 | @@ -XXX,XX +XXX,XX @@ | 37 | @@ -XXX,XX +XXX,XX @@ void qemu_co_rwlock_init(CoRwlock *lock) |
67 | -{"execute": "job-finalize", "arguments": {"id": "commit0"}} | 38 | { |
68 | +..{"execute": "job-finalize", "arguments": {"id": "commit0"}} | 39 | memset(lock, 0, sizeof(*lock)); |
69 | {"return": {}} | 40 | qemu_co_queue_init(&lock->queue); |
70 | {"data": {"id": "commit0", "type": "commit"}, "event": "BLOCK_JOB_PENDING", "timestamp": {"microseconds": "USECS", "seconds": "SECS"}} | 41 | + qemu_co_mutex_init(&lock->mutex); |
71 | {"data": {"device": "commit0", "len": 3145728, "offset": 3145728, "speed": 0, "type": "commit"}, "event": "BLOCK_JOB_COMPLETED", "timestamp": {"microseconds": "USECS", "seconds": "SECS"}} | 42 | } |
72 | -{"execute": "job-finalize", "arguments": {"id": "stream0"}} | 43 | |
73 | +...{"execute": "job-finalize", "arguments": {"id": "stream0"}} | 44 | void qemu_co_rwlock_rdlock(CoRwlock *lock) |
74 | {"return": {}} | 45 | { |
75 | {"data": {"id": "stream0", "type": "stream"}, "event": "BLOCK_JOB_PENDING", "timestamp": {"microseconds": "USECS", "seconds": "SECS"}} | 46 | Coroutine *self = qemu_coroutine_self(); |
76 | {"data": {"device": "stream0", "len": 3145728, "offset": 3145728, "speed": 0, "type": "stream"}, "event": "BLOCK_JOB_COMPLETED", "timestamp": {"microseconds": "USECS", "seconds": "SECS"}} | 47 | |
77 | -{"execute": "job-finalize", "arguments": {"id": "stream0"}} | 48 | - while (lock->writer) { |
78 | +.{"execute": "job-finalize", "arguments": {"id": "stream0"}} | 49 | - qemu_co_queue_wait(&lock->queue, NULL); |
79 | {"return": {}} | 50 | + qemu_co_mutex_lock(&lock->mutex); |
80 | {"data": {"id": "stream0", "type": "stream"}, "event": "BLOCK_JOB_PENDING", "timestamp": {"microseconds": "USECS", "seconds": "SECS"}} | 51 | + /* For fairness, wait if a writer is in line. */ |
81 | {"data": {"device": "stream0", "len": 3145728, "offset": 3145728, "speed": 0, "type": "stream"}, "event": "BLOCK_JOB_COMPLETED", "timestamp": {"microseconds": "USECS", "seconds": "SECS"}} | 52 | + while (lock->pending_writer) { |
82 | -..................... | 53 | + qemu_co_queue_wait(&lock->queue, &lock->mutex); |
83 | +............... | 54 | } |
84 | ---------------------------------------------------------------------- | 55 | lock->reader++; |
85 | Ran 21 tests | 56 | + qemu_co_mutex_unlock(&lock->mutex); |
86 | |||
87 | diff --git a/tests/qemu-iotests/295.out b/tests/qemu-iotests/295.out | ||
88 | index XXXXXXX..XXXXXXX 100644 | ||
89 | --- a/tests/qemu-iotests/295.out | ||
90 | +++ b/tests/qemu-iotests/295.out | ||
91 | @@ -XXX,XX +XXX,XX @@ | ||
92 | {"return": {}} | ||
93 | {"execute": "job-dismiss", "arguments": {"id": "job_erase_key"}} | ||
94 | {"return": {}} | ||
95 | -{"execute": "job-dismiss", "arguments": {"id": "job_add_key"}} | ||
96 | +.{"execute": "job-dismiss", "arguments": {"id": "job_add_key"}} | ||
97 | {"return": {}} | ||
98 | {"execute": "job-dismiss", "arguments": {"id": "job_erase_key"}} | ||
99 | {"return": {}} | ||
100 | @@ -XXX,XX +XXX,XX @@ Job failed: Invalid password, cannot unlock any keyslot | ||
101 | {"return": {}} | ||
102 | {"execute": "job-dismiss", "arguments": {"id": "job_add_key"}} | ||
103 | {"return": {}} | ||
104 | -{"execute": "job-dismiss", "arguments": {"id": "job_add_key"}} | ||
105 | +.{"execute": "job-dismiss", "arguments": {"id": "job_add_key"}} | ||
106 | {"return": {}} | ||
107 | {"execute": "job-dismiss", "arguments": {"id": "job_add_key"}} | ||
108 | {"return": {}} | ||
109 | @@ -XXX,XX +XXX,XX @@ Job failed: All the active keyslots match the (old) password that was given and | ||
110 | {"return": {}} | ||
111 | {"execute": "job-dismiss", "arguments": {"id": "job_erase_key"}} | ||
112 | {"return": {}} | ||
113 | -... | ||
114 | +. | ||
115 | ---------------------------------------------------------------------- | ||
116 | Ran 3 tests | ||
117 | |||
118 | diff --git a/tests/qemu-iotests/296.out b/tests/qemu-iotests/296.out | ||
119 | index XXXXXXX..XXXXXXX 100644 | ||
120 | --- a/tests/qemu-iotests/296.out | ||
121 | +++ b/tests/qemu-iotests/296.out | ||
122 | @@ -XXX,XX +XXX,XX @@ Job failed: Failed to get shared "consistent read" lock | ||
123 | qemu-img: Failed to get shared "consistent read" lock | ||
124 | Is another process using the image [TEST_DIR/test.img]? | ||
125 | |||
126 | -Formatting 'TEST_DIR/test.img', fmt=luks size=1048576 key-secret=keysec0 iter-time=10 | ||
127 | +.Formatting 'TEST_DIR/test.img', fmt=luks size=1048576 key-secret=keysec0 iter-time=10 | ||
128 | |||
129 | Job failed: Block node is read-only | ||
130 | {"execute": "job-dismiss", "arguments": {"id": "job0"}} | ||
131 | @@ -XXX,XX +XXX,XX @@ Job failed: Failed to get shared "consistent read" lock | ||
132 | {"return": {}} | ||
133 | {"execute": "job-dismiss", "arguments": {"id": "job0"}} | ||
134 | {"return": {}} | ||
135 | -Formatting 'TEST_DIR/test.img', fmt=luks size=1048576 key-secret=keysec0 iter-time=10 | ||
136 | +.Formatting 'TEST_DIR/test.img', fmt=luks size=1048576 key-secret=keysec0 iter-time=10 | ||
137 | |||
138 | {"return": {}} | ||
139 | {"error": {"class": "GenericError", "desc": "Failed to get \"write\" lock"}} | ||
140 | -Formatting 'TEST_DIR/test.img', fmt=luks size=1048576 key-secret=keysec0 iter-time=10 | ||
141 | +.Formatting 'TEST_DIR/test.img', fmt=luks size=1048576 key-secret=keysec0 iter-time=10 | ||
142 | |||
143 | {"return": {}} | ||
144 | {"return": {}} | ||
145 | -.... | ||
146 | +. | ||
147 | ---------------------------------------------------------------------- | ||
148 | Ran 4 tests | ||
149 | |||
150 | diff --git a/tests/qemu-iotests/iotests.py b/tests/qemu-iotests/iotests.py | ||
151 | index XXXXXXX..XXXXXXX 100644 | ||
152 | --- a/tests/qemu-iotests/iotests.py | ||
153 | +++ b/tests/qemu-iotests/iotests.py | ||
154 | @@ -XXX,XX +XXX,XX @@ | ||
155 | import bz2 | ||
156 | from collections import OrderedDict | ||
157 | import faulthandler | ||
158 | -import io | ||
159 | import json | ||
160 | import logging | ||
161 | import os | ||
162 | @@ -XXX,XX +XXX,XX @@ | ||
163 | import sys | ||
164 | import time | ||
165 | from typing import (Any, Callable, Dict, Iterable, | ||
166 | - List, Optional, Sequence, Tuple, TypeVar) | ||
167 | + List, Optional, Sequence, TextIO, Tuple, Type, TypeVar) | ||
168 | import unittest | ||
169 | |||
170 | from contextlib import contextmanager | ||
171 | @@ -XXX,XX +XXX,XX @@ def func_wrapper(*args, **kwargs): | ||
172 | return func(*args, **kwargs) | ||
173 | return func_wrapper | ||
174 | |||
175 | +# We need to filter out the time taken from the output so that | ||
176 | +# qemu-iotest can reliably diff the results against master output, | ||
177 | +# and hide skipped tests from the reference output. | ||
178 | + | 57 | + |
179 | +class ReproducibleTestResult(unittest.TextTestResult): | 58 | + /* The rest of the read-side critical section is run without the mutex. */ |
180 | + def addSkip(self, test, reason): | 59 | self->locks_held++; |
181 | + # Same as TextTestResult, but print dot instead of "s" | 60 | } |
182 | + unittest.TestResult.addSkip(self, test, reason) | 61 | |
183 | + if self.showAll: | 62 | @@ -XXX,XX +XXX,XX @@ void qemu_co_rwlock_unlock(CoRwlock *lock) |
184 | + self.stream.writeln("skipped {0!r}".format(reason)) | 63 | Coroutine *self = qemu_coroutine_self(); |
185 | + elif self.dots: | 64 | |
186 | + self.stream.write(".") | 65 | assert(qemu_in_coroutine()); |
187 | + self.stream.flush() | 66 | - if (lock->writer) { |
67 | - lock->writer = false; | ||
68 | + if (!lock->reader) { | ||
69 | + /* The critical section started in qemu_co_rwlock_wrlock. */ | ||
70 | qemu_co_queue_restart_all(&lock->queue); | ||
71 | } else { | ||
72 | + self->locks_held--; | ||
188 | + | 73 | + |
189 | +class ReproducibleStreamWrapper: | 74 | + qemu_co_mutex_lock(&lock->mutex); |
190 | + def __init__(self, stream: TextIO): | 75 | lock->reader--; |
191 | + self.stream = stream | 76 | assert(lock->reader >= 0); |
77 | /* Wakeup only one waiting writer */ | ||
78 | @@ -XXX,XX +XXX,XX @@ void qemu_co_rwlock_unlock(CoRwlock *lock) | ||
79 | qemu_co_queue_next(&lock->queue); | ||
80 | } | ||
81 | } | ||
82 | - self->locks_held--; | ||
83 | + qemu_co_mutex_unlock(&lock->mutex); | ||
84 | } | ||
85 | |||
86 | void qemu_co_rwlock_wrlock(CoRwlock *lock) | ||
87 | { | ||
88 | - Coroutine *self = qemu_coroutine_self(); | ||
89 | - | ||
90 | - while (lock->writer || lock->reader) { | ||
91 | - qemu_co_queue_wait(&lock->queue, NULL); | ||
92 | + qemu_co_mutex_lock(&lock->mutex); | ||
93 | + lock->pending_writer++; | ||
94 | + while (lock->reader) { | ||
95 | + qemu_co_queue_wait(&lock->queue, &lock->mutex); | ||
96 | } | ||
97 | - lock->writer = true; | ||
98 | - self->locks_held++; | ||
99 | + lock->pending_writer--; | ||
192 | + | 100 | + |
193 | + def __getattr__(self, attr): | 101 | + /* The rest of the write-side critical section is run with |
194 | + if attr in ('stream', '__getstate__'): | 102 | + * the mutex taken, so that lock->reader remains zero. |
195 | + raise AttributeError(attr) | 103 | + * There is no need to update self->locks_held. |
196 | + return getattr(self.stream, attr) | 104 | + */ |
197 | + | 105 | } |
198 | + def write(self, arg=None): | ||
199 | + arg = re.sub(r'Ran (\d+) tests? in [\d.]+s', r'Ran \1 tests', arg) | ||
200 | + arg = re.sub(r' \(skipped=\d+\)', r'', arg) | ||
201 | + self.stream.write(arg) | ||
202 | + | ||
203 | +class ReproducibleTestRunner(unittest.TextTestRunner): | ||
204 | + def __init__(self, stream: Optional[TextIO] = None, | ||
205 | + resultclass: Type[unittest.TestResult] = ReproducibleTestResult, | ||
206 | + **kwargs: Any) -> None: | ||
207 | + rstream = ReproducibleStreamWrapper(stream or sys.stdout) | ||
208 | + super().__init__(stream=rstream, # type: ignore | ||
209 | + descriptions=True, | ||
210 | + resultclass=resultclass, | ||
211 | + **kwargs) | ||
212 | + | ||
213 | def execute_unittest(debug=False): | ||
214 | """Executes unittests within the calling module.""" | ||
215 | |||
216 | verbosity = 2 if debug else 1 | ||
217 | - | ||
218 | - if debug: | ||
219 | - output = sys.stdout | ||
220 | - else: | ||
221 | - # We need to filter out the time taken from the output so that | ||
222 | - # qemu-iotest can reliably diff the results against master output. | ||
223 | - output = io.StringIO() | ||
224 | - | ||
225 | - runner = unittest.TextTestRunner(stream=output, descriptions=True, | ||
226 | - verbosity=verbosity) | ||
227 | - try: | ||
228 | - # unittest.main() will use sys.exit(); so expect a SystemExit | ||
229 | - # exception | ||
230 | - unittest.main(testRunner=runner) | ||
231 | - finally: | ||
232 | - # We need to filter out the time taken from the output so that | ||
233 | - # qemu-iotest can reliably diff the results against master output. | ||
234 | - if not debug: | ||
235 | - out = output.getvalue() | ||
236 | - out = re.sub(r'Ran (\d+) tests? in [\d.]+s', r'Ran \1 tests', out) | ||
237 | - | ||
238 | - # Hide skipped tests from the reference output | ||
239 | - out = re.sub(r'OK \(skipped=\d+\)', 'OK', out) | ||
240 | - out_first_line, out_rest = out.split('\n', 1) | ||
241 | - out = out_first_line.replace('s', '.') + '\n' + out_rest | ||
242 | - | ||
243 | - sys.stderr.write(out) | ||
244 | + runner = ReproducibleTestRunner(verbosity=verbosity) | ||
245 | + unittest.main(testRunner=runner) | ||
246 | |||
247 | def execute_setup_common(supported_fmts: Sequence[str] = (), | ||
248 | supported_platforms: Sequence[str] = (), | ||
249 | -- | 106 | -- |
250 | 2.31.1 | 107 | 2.9.3 |
251 | 108 | ||
252 | 109 | diff view generated by jsdifflib |