1 | The following changes since commit 136c67e07869227b21b3f627316e03679ce7b738: | 1 | The following changes since commit ac793156f650ae2d77834932d72224175ee69086: |
---|---|---|---|
2 | 2 | ||
3 | Merge remote-tracking branch 'remotes/bkoppelmann/tags/pull-tricore-2018-03-02' into staging (2018-03-02 16:56:20 +0000) | 3 | Merge remote-tracking branch 'remotes/pmaydell/tags/pull-target-arm-20201020-1' into staging (2020-10-20 21:11:35 +0100) |
4 | 4 | ||
5 | are available in the Git repository at: | 5 | are available in the Git repository at: |
6 | 6 | ||
7 | git://github.com/stefanha/qemu.git tags/block-pull-request | 7 | https://gitlab.com/stefanha/qemu.git tags/block-pull-request |
8 | 8 | ||
9 | for you to fetch changes up to 23500c6a9409efc80d696aede0629bfbe7556a90: | 9 | for you to fetch changes up to 32a3fd65e7e3551337fd26bfc0e2f899d70c028c: |
10 | 10 | ||
11 | README: Document 'git-publish' workflow (2018-03-05 09:03:17 +0000) | 11 | iotests: add commit top->base cases to 274 (2020-10-22 09:55:39 +0100) |
12 | 12 | ||
13 | ---------------------------------------------------------------- | 13 | ---------------------------------------------------------------- |
14 | Pull request | 14 | Pull request |
15 | 15 | ||
16 | Mostly patches that are only indirectly related to the block layer, but I've | 16 | v2: |
17 | reviewed them and there is no maintainer. | 17 | * Fix format string issues on 32-bit hosts [Peter] |
18 | * Fix qemu-nbd.c CONFIG_POSIX ifdef issue [Eric] | ||
19 | * Fix missing eventfd.h header on macOS [Peter] | ||
20 | * Drop unreliable vhost-user-blk test (will send a new patch when ready) [Peter] | ||
21 | |||
22 | This pull request contains the vhost-user-blk server by Coiby Xu along with my | ||
23 | additions, block/nvme.c alignment and hardware error statistics by Philippe | ||
24 | Mathieu-Daudé, and bdrv_co_block_status_above() fixes by Vladimir | ||
25 | Sementsov-Ogievskiy. | ||
18 | 26 | ||
19 | ---------------------------------------------------------------- | 27 | ---------------------------------------------------------------- |
20 | 28 | ||
21 | Fam Zheng (2): | 29 | Coiby Xu (6): |
22 | Add a git-publish configuration file | 30 | libvhost-user: Allow vu_message_read to be replaced |
23 | README: Document 'git-publish' workflow | 31 | libvhost-user: remove watch for kick_fd when de-initialize vu-dev |
32 | util/vhost-user-server: generic vhost user server | ||
33 | block: move logical block size check function to a common utility | ||
34 | function | ||
35 | block/export: vhost-user block device backend server | ||
36 | MAINTAINERS: Add vhost-user block device backend server maintainer | ||
24 | 37 | ||
25 | Su Hang (3): | 38 | Philippe Mathieu-Daudé (1): |
26 | util/uri.c: Coding style check, Only whitespace involved | 39 | block/nvme: Add driver statistics for access alignment and hw errors |
27 | util/uri.c: remove brackets that wrap `return` statement's content. | ||
28 | util/uri.c: wrap single statement blocks with braces {} | ||
29 | 40 | ||
30 | Thomas Huth (1): | 41 | Stefan Hajnoczi (16): |
31 | tests/libqos: Check for valid dev pointer when looking for PCI devices | 42 | util/vhost-user-server: s/fileds/fields/ typo fix |
43 | util/vhost-user-server: drop unnecessary QOM cast | ||
44 | util/vhost-user-server: drop unnecessary watch deletion | ||
45 | block/export: consolidate request structs into VuBlockReq | ||
46 | util/vhost-user-server: drop unused DevicePanicNotifier | ||
47 | util/vhost-user-server: fix memory leak in vu_message_read() | ||
48 | util/vhost-user-server: check EOF when reading payload | ||
49 | util/vhost-user-server: rework vu_client_trip() coroutine lifecycle | ||
50 | block/export: report flush errors | ||
51 | block/export: convert vhost-user-blk server to block export API | ||
52 | util/vhost-user-server: move header to include/ | ||
53 | util/vhost-user-server: use static library in meson.build | ||
54 | qemu-storage-daemon: avoid compiling blockdev_ss twice | ||
55 | block: move block exports to libblockdev | ||
56 | block/export: add iothread and fixed-iothread options | ||
57 | block/export: add vhost-user-blk multi-queue support | ||
32 | 58 | ||
33 | tests/libqos/virtio-pci.c | 4 +- | 59 | Vladimir Sementsov-Ogievskiy (5): |
34 | util/uri.c | 1733 ++++++++++++++++++++++++--------------------- | 60 | block/io: fix bdrv_co_block_status_above |
35 | .gitpublish | 51 ++ | 61 | block/io: bdrv_common_block_status_above: support include_base |
36 | README | 31 +- | 62 | block/io: bdrv_common_block_status_above: support bs == base |
37 | 4 files changed, 1014 insertions(+), 805 deletions(-) | 63 | block/io: fix bdrv_is_allocated_above |
38 | create mode 100644 .gitpublish | 64 | iotests: add commit top->base cases to 274 |
65 | |||
66 | MAINTAINERS | 9 + | ||
67 | qapi/block-core.json | 24 +- | ||
68 | qapi/block-export.json | 36 +- | ||
69 | block/coroutines.h | 2 + | ||
70 | block/export/vhost-user-blk-server.h | 19 + | ||
71 | contrib/libvhost-user/libvhost-user.h | 21 + | ||
72 | include/qemu/vhost-user-server.h | 65 +++ | ||
73 | util/block-helpers.h | 19 + | ||
74 | block/export/export.c | 37 +- | ||
75 | block/export/vhost-user-blk-server.c | 431 ++++++++++++++++++++ | ||
76 | block/io.c | 132 +++--- | ||
77 | block/nvme.c | 27 ++ | ||
78 | block/qcow2.c | 16 +- | ||
79 | contrib/libvhost-user/libvhost-user-glib.c | 2 +- | ||
80 | contrib/libvhost-user/libvhost-user.c | 15 +- | ||
81 | hw/core/qdev-properties-system.c | 31 +- | ||
82 | nbd/server.c | 2 - | ||
83 | qemu-nbd.c | 21 +- | ||
84 | softmmu/vl.c | 4 + | ||
85 | stubs/blk-exp-close-all.c | 7 + | ||
86 | tests/vhost-user-bridge.c | 2 + | ||
87 | tools/virtiofsd/fuse_virtio.c | 4 +- | ||
88 | util/block-helpers.c | 46 +++ | ||
89 | util/vhost-user-server.c | 446 +++++++++++++++++++++ | ||
90 | block/export/meson.build | 3 +- | ||
91 | contrib/libvhost-user/meson.build | 1 + | ||
92 | meson.build | 22 +- | ||
93 | nbd/meson.build | 2 + | ||
94 | storage-daemon/meson.build | 3 +- | ||
95 | stubs/meson.build | 1 + | ||
96 | tests/qemu-iotests/274 | 20 + | ||
97 | tests/qemu-iotests/274.out | 68 ++++ | ||
98 | util/meson.build | 4 + | ||
99 | 33 files changed, 1420 insertions(+), 122 deletions(-) | ||
100 | create mode 100644 block/export/vhost-user-blk-server.h | ||
101 | create mode 100644 include/qemu/vhost-user-server.h | ||
102 | create mode 100644 util/block-helpers.h | ||
103 | create mode 100644 block/export/vhost-user-blk-server.c | ||
104 | create mode 100644 stubs/blk-exp-close-all.c | ||
105 | create mode 100644 util/block-helpers.c | ||
106 | create mode 100644 util/vhost-user-server.c | ||
39 | 107 | ||
40 | -- | 108 | -- |
41 | 2.14.3 | 109 | 2.26.2 |
42 | 110 | ||
43 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Philippe Mathieu-Daudé <philmd@redhat.com> | ||
1 | 2 | ||
3 | Keep statistics of some hardware errors, and number of | ||
4 | aligned/unaligned I/O accesses. | ||
5 | |||
6 | QMP example booting a full RHEL 8.3 aarch64 guest: | ||
7 | |||
8 | { "execute": "query-blockstats" } | ||
9 | { | ||
10 | "return": [ | ||
11 | { | ||
12 | "device": "", | ||
13 | "node-name": "drive0", | ||
14 | "stats": { | ||
15 | "flush_total_time_ns": 6026948, | ||
16 | "wr_highest_offset": 3383991230464, | ||
17 | "wr_total_time_ns": 807450995, | ||
18 | "failed_wr_operations": 0, | ||
19 | "failed_rd_operations": 0, | ||
20 | "wr_merged": 3, | ||
21 | "wr_bytes": 50133504, | ||
22 | "failed_unmap_operations": 0, | ||
23 | "failed_flush_operations": 0, | ||
24 | "account_invalid": false, | ||
25 | "rd_total_time_ns": 1846979900, | ||
26 | "flush_operations": 130, | ||
27 | "wr_operations": 659, | ||
28 | "rd_merged": 1192, | ||
29 | "rd_bytes": 218244096, | ||
30 | "account_failed": false, | ||
31 | "idle_time_ns": 2678641497, | ||
32 | "rd_operations": 7406, | ||
33 | }, | ||
34 | "driver-specific": { | ||
35 | "driver": "nvme", | ||
36 | "completion-errors": 0, | ||
37 | "unaligned-accesses": 2959, | ||
38 | "aligned-accesses": 4477 | ||
39 | }, | ||
40 | "qdev": "/machine/peripheral-anon/device[0]/virtio-backend" | ||
41 | } | ||
42 | ] | ||
43 | } | ||
44 | |||
45 | Suggested-by: Stefan Hajnoczi <stefanha@gmail.com> | ||
46 | Signed-off-by: Philippe Mathieu-Daudé <philmd@redhat.com> | ||
47 | Acked-by: Markus Armbruster <armbru@redhat.com> | ||
48 | Message-id: 20201001162939.1567915-1-philmd@redhat.com | ||
49 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
50 | --- | ||
51 | qapi/block-core.json | 24 +++++++++++++++++++++++- | ||
52 | block/nvme.c | 27 +++++++++++++++++++++++++++ | ||
53 | 2 files changed, 50 insertions(+), 1 deletion(-) | ||
54 | |||
55 | diff --git a/qapi/block-core.json b/qapi/block-core.json | ||
56 | index XXXXXXX..XXXXXXX 100644 | ||
57 | --- a/qapi/block-core.json | ||
58 | +++ b/qapi/block-core.json | ||
59 | @@ -XXX,XX +XXX,XX @@ | ||
60 | 'discard-nb-failed': 'uint64', | ||
61 | 'discard-bytes-ok': 'uint64' } } | ||
62 | |||
63 | +## | ||
64 | +# @BlockStatsSpecificNvme: | ||
65 | +# | ||
66 | +# NVMe driver statistics | ||
67 | +# | ||
68 | +# @completion-errors: The number of completion errors. | ||
69 | +# | ||
70 | +# @aligned-accesses: The number of aligned accesses performed by | ||
71 | +# the driver. | ||
72 | +# | ||
73 | +# @unaligned-accesses: The number of unaligned accesses performed by | ||
74 | +# the driver. | ||
75 | +# | ||
76 | +# Since: 5.2 | ||
77 | +## | ||
78 | +{ 'struct': 'BlockStatsSpecificNvme', | ||
79 | + 'data': { | ||
80 | + 'completion-errors': 'uint64', | ||
81 | + 'aligned-accesses': 'uint64', | ||
82 | + 'unaligned-accesses': 'uint64' } } | ||
83 | + | ||
84 | ## | ||
85 | # @BlockStatsSpecific: | ||
86 | # | ||
87 | @@ -XXX,XX +XXX,XX @@ | ||
88 | 'discriminator': 'driver', | ||
89 | 'data': { | ||
90 | 'file': 'BlockStatsSpecificFile', | ||
91 | - 'host_device': 'BlockStatsSpecificFile' } } | ||
92 | + 'host_device': 'BlockStatsSpecificFile', | ||
93 | + 'nvme': 'BlockStatsSpecificNvme' } } | ||
94 | |||
95 | ## | ||
96 | # @BlockStats: | ||
97 | diff --git a/block/nvme.c b/block/nvme.c | ||
98 | index XXXXXXX..XXXXXXX 100644 | ||
99 | --- a/block/nvme.c | ||
100 | +++ b/block/nvme.c | ||
101 | @@ -XXX,XX +XXX,XX @@ struct BDRVNVMeState { | ||
102 | |||
103 | /* PCI address (required for nvme_refresh_filename()) */ | ||
104 | char *device; | ||
105 | + | ||
106 | + struct { | ||
107 | + uint64_t completion_errors; | ||
108 | + uint64_t aligned_accesses; | ||
109 | + uint64_t unaligned_accesses; | ||
110 | + } stats; | ||
111 | }; | ||
112 | |||
113 | #define NVME_BLOCK_OPT_DEVICE "device" | ||
114 | @@ -XXX,XX +XXX,XX @@ static bool nvme_process_completion(NVMeQueuePair *q) | ||
115 | break; | ||
116 | } | ||
117 | ret = nvme_translate_error(c); | ||
118 | + if (ret) { | ||
119 | + s->stats.completion_errors++; | ||
120 | + } | ||
121 | q->cq.head = (q->cq.head + 1) % NVME_QUEUE_SIZE; | ||
122 | if (!q->cq.head) { | ||
123 | q->cq_phase = !q->cq_phase; | ||
124 | @@ -XXX,XX +XXX,XX @@ static int nvme_co_prw(BlockDriverState *bs, uint64_t offset, uint64_t bytes, | ||
125 | assert(QEMU_IS_ALIGNED(bytes, s->page_size)); | ||
126 | assert(bytes <= s->max_transfer); | ||
127 | if (nvme_qiov_aligned(bs, qiov)) { | ||
128 | + s->stats.aligned_accesses++; | ||
129 | return nvme_co_prw_aligned(bs, offset, bytes, qiov, is_write, flags); | ||
130 | } | ||
131 | + s->stats.unaligned_accesses++; | ||
132 | trace_nvme_prw_buffered(s, offset, bytes, qiov->niov, is_write); | ||
133 | buf = qemu_try_memalign(s->page_size, bytes); | ||
134 | |||
135 | @@ -XXX,XX +XXX,XX @@ static void nvme_unregister_buf(BlockDriverState *bs, void *host) | ||
136 | qemu_vfio_dma_unmap(s->vfio, host); | ||
137 | } | ||
138 | |||
139 | +static BlockStatsSpecific *nvme_get_specific_stats(BlockDriverState *bs) | ||
140 | +{ | ||
141 | + BlockStatsSpecific *stats = g_new(BlockStatsSpecific, 1); | ||
142 | + BDRVNVMeState *s = bs->opaque; | ||
143 | + | ||
144 | + stats->driver = BLOCKDEV_DRIVER_NVME; | ||
145 | + stats->u.nvme = (BlockStatsSpecificNvme) { | ||
146 | + .completion_errors = s->stats.completion_errors, | ||
147 | + .aligned_accesses = s->stats.aligned_accesses, | ||
148 | + .unaligned_accesses = s->stats.unaligned_accesses, | ||
149 | + }; | ||
150 | + | ||
151 | + return stats; | ||
152 | +} | ||
153 | + | ||
154 | static const char *const nvme_strong_runtime_opts[] = { | ||
155 | NVME_BLOCK_OPT_DEVICE, | ||
156 | NVME_BLOCK_OPT_NAMESPACE, | ||
157 | @@ -XXX,XX +XXX,XX @@ static BlockDriver bdrv_nvme = { | ||
158 | .bdrv_refresh_filename = nvme_refresh_filename, | ||
159 | .bdrv_refresh_limits = nvme_refresh_limits, | ||
160 | .strong_runtime_opts = nvme_strong_runtime_opts, | ||
161 | + .bdrv_get_specific_stats = nvme_get_specific_stats, | ||
162 | |||
163 | .bdrv_detach_aio_context = nvme_detach_aio_context, | ||
164 | .bdrv_attach_aio_context = nvme_attach_aio_context, | ||
165 | -- | ||
166 | 2.26.2 | ||
167 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Coiby Xu <coiby.xu@gmail.com> | ||
1 | 2 | ||
3 | Allow vu_message_read to be replaced by one which will make use of the | ||
4 | QIOChannel functions. Thus reading vhost-user message won't stall the | ||
5 | guest. For slave channel, we still use the default vu_message_read. | ||
6 | |||
7 | Reviewed-by: Marc-André Lureau <marcandre.lureau@redhat.com> | ||
8 | Signed-off-by: Coiby Xu <coiby.xu@gmail.com> | ||
9 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
10 | Message-id: 20200918080912.321299-2-coiby.xu@gmail.com | ||
11 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
12 | --- | ||
13 | contrib/libvhost-user/libvhost-user.h | 21 +++++++++++++++++++++ | ||
14 | contrib/libvhost-user/libvhost-user-glib.c | 2 +- | ||
15 | contrib/libvhost-user/libvhost-user.c | 14 +++++++------- | ||
16 | tests/vhost-user-bridge.c | 2 ++ | ||
17 | tools/virtiofsd/fuse_virtio.c | 4 ++-- | ||
18 | 5 files changed, 33 insertions(+), 10 deletions(-) | ||
19 | |||
20 | diff --git a/contrib/libvhost-user/libvhost-user.h b/contrib/libvhost-user/libvhost-user.h | ||
21 | index XXXXXXX..XXXXXXX 100644 | ||
22 | --- a/contrib/libvhost-user/libvhost-user.h | ||
23 | +++ b/contrib/libvhost-user/libvhost-user.h | ||
24 | @@ -XXX,XX +XXX,XX @@ | ||
25 | */ | ||
26 | #define VHOST_USER_MAX_RAM_SLOTS 32 | ||
27 | |||
28 | +#define VHOST_USER_HDR_SIZE offsetof(VhostUserMsg, payload.u64) | ||
29 | + | ||
30 | typedef enum VhostSetConfigType { | ||
31 | VHOST_SET_CONFIG_TYPE_MASTER = 0, | ||
32 | VHOST_SET_CONFIG_TYPE_MIGRATION = 1, | ||
33 | @@ -XXX,XX +XXX,XX @@ typedef uint64_t (*vu_get_features_cb) (VuDev *dev); | ||
34 | typedef void (*vu_set_features_cb) (VuDev *dev, uint64_t features); | ||
35 | typedef int (*vu_process_msg_cb) (VuDev *dev, VhostUserMsg *vmsg, | ||
36 | int *do_reply); | ||
37 | +typedef bool (*vu_read_msg_cb) (VuDev *dev, int sock, VhostUserMsg *vmsg); | ||
38 | typedef void (*vu_queue_set_started_cb) (VuDev *dev, int qidx, bool started); | ||
39 | typedef bool (*vu_queue_is_processed_in_order_cb) (VuDev *dev, int qidx); | ||
40 | typedef int (*vu_get_config_cb) (VuDev *dev, uint8_t *config, uint32_t len); | ||
41 | @@ -XXX,XX +XXX,XX @@ struct VuDev { | ||
42 | bool broken; | ||
43 | uint16_t max_queues; | ||
44 | |||
45 | + /* @read_msg: custom method to read vhost-user message | ||
46 | + * | ||
47 | + * Read data from vhost_user socket fd and fill up | ||
48 | + * the passed VhostUserMsg *vmsg struct. | ||
49 | + * | ||
50 | + * If reading fails, it should close the received set of file | ||
51 | + * descriptors as socket message's auxiliary data. | ||
52 | + * | ||
53 | + * For the details, please refer to vu_message_read in libvhost-user.c | ||
54 | + * which will be used by default if not custom method is provided when | ||
55 | + * calling vu_init | ||
56 | + * | ||
57 | + * Returns: true if vhost-user message successfully received, | ||
58 | + * otherwise return false. | ||
59 | + * | ||
60 | + */ | ||
61 | + vu_read_msg_cb read_msg; | ||
62 | /* @set_watch: add or update the given fd to the watch set, | ||
63 | * call cb when condition is met */ | ||
64 | vu_set_watch_cb set_watch; | ||
65 | @@ -XXX,XX +XXX,XX @@ bool vu_init(VuDev *dev, | ||
66 | uint16_t max_queues, | ||
67 | int socket, | ||
68 | vu_panic_cb panic, | ||
69 | + vu_read_msg_cb read_msg, | ||
70 | vu_set_watch_cb set_watch, | ||
71 | vu_remove_watch_cb remove_watch, | ||
72 | const VuDevIface *iface); | ||
73 | diff --git a/contrib/libvhost-user/libvhost-user-glib.c b/contrib/libvhost-user/libvhost-user-glib.c | ||
74 | index XXXXXXX..XXXXXXX 100644 | ||
75 | --- a/contrib/libvhost-user/libvhost-user-glib.c | ||
76 | +++ b/contrib/libvhost-user/libvhost-user-glib.c | ||
77 | @@ -XXX,XX +XXX,XX @@ vug_init(VugDev *dev, uint16_t max_queues, int socket, | ||
78 | g_assert(dev); | ||
79 | g_assert(iface); | ||
80 | |||
81 | - if (!vu_init(&dev->parent, max_queues, socket, panic, set_watch, | ||
82 | + if (!vu_init(&dev->parent, max_queues, socket, panic, NULL, set_watch, | ||
83 | remove_watch, iface)) { | ||
84 | return false; | ||
85 | } | ||
86 | diff --git a/contrib/libvhost-user/libvhost-user.c b/contrib/libvhost-user/libvhost-user.c | ||
87 | index XXXXXXX..XXXXXXX 100644 | ||
88 | --- a/contrib/libvhost-user/libvhost-user.c | ||
89 | +++ b/contrib/libvhost-user/libvhost-user.c | ||
90 | @@ -XXX,XX +XXX,XX @@ | ||
91 | /* The version of inflight buffer */ | ||
92 | #define INFLIGHT_VERSION 1 | ||
93 | |||
94 | -#define VHOST_USER_HDR_SIZE offsetof(VhostUserMsg, payload.u64) | ||
95 | - | ||
96 | /* The version of the protocol we support */ | ||
97 | #define VHOST_USER_VERSION 1 | ||
98 | #define LIBVHOST_USER_DEBUG 0 | ||
99 | @@ -XXX,XX +XXX,XX @@ have_userfault(void) | ||
100 | } | ||
101 | |||
102 | static bool | ||
103 | -vu_message_read(VuDev *dev, int conn_fd, VhostUserMsg *vmsg) | ||
104 | +vu_message_read_default(VuDev *dev, int conn_fd, VhostUserMsg *vmsg) | ||
105 | { | ||
106 | char control[CMSG_SPACE(VHOST_MEMORY_BASELINE_NREGIONS * sizeof(int))] = {}; | ||
107 | struct iovec iov = { | ||
108 | @@ -XXX,XX +XXX,XX @@ vu_process_message_reply(VuDev *dev, const VhostUserMsg *vmsg) | ||
109 | goto out; | ||
110 | } | ||
111 | |||
112 | - if (!vu_message_read(dev, dev->slave_fd, &msg_reply)) { | ||
113 | + if (!vu_message_read_default(dev, dev->slave_fd, &msg_reply)) { | ||
114 | goto out; | ||
115 | } | ||
116 | |||
117 | @@ -XXX,XX +XXX,XX @@ vu_set_mem_table_exec_postcopy(VuDev *dev, VhostUserMsg *vmsg) | ||
118 | /* Wait for QEMU to confirm that it's registered the handler for the | ||
119 | * faults. | ||
120 | */ | ||
121 | - if (!vu_message_read(dev, dev->sock, vmsg) || | ||
122 | + if (!dev->read_msg(dev, dev->sock, vmsg) || | ||
123 | vmsg->size != sizeof(vmsg->payload.u64) || | ||
124 | vmsg->payload.u64 != 0) { | ||
125 | vu_panic(dev, "failed to receive valid ack for postcopy set-mem-table"); | ||
126 | @@ -XXX,XX +XXX,XX @@ vu_dispatch(VuDev *dev) | ||
127 | int reply_requested; | ||
128 | bool need_reply, success = false; | ||
129 | |||
130 | - if (!vu_message_read(dev, dev->sock, &vmsg)) { | ||
131 | + if (!dev->read_msg(dev, dev->sock, &vmsg)) { | ||
132 | goto end; | ||
133 | } | ||
134 | |||
135 | @@ -XXX,XX +XXX,XX @@ vu_init(VuDev *dev, | ||
136 | uint16_t max_queues, | ||
137 | int socket, | ||
138 | vu_panic_cb panic, | ||
139 | + vu_read_msg_cb read_msg, | ||
140 | vu_set_watch_cb set_watch, | ||
141 | vu_remove_watch_cb remove_watch, | ||
142 | const VuDevIface *iface) | ||
143 | @@ -XXX,XX +XXX,XX @@ vu_init(VuDev *dev, | ||
144 | |||
145 | dev->sock = socket; | ||
146 | dev->panic = panic; | ||
147 | + dev->read_msg = read_msg ? read_msg : vu_message_read_default; | ||
148 | dev->set_watch = set_watch; | ||
149 | dev->remove_watch = remove_watch; | ||
150 | dev->iface = iface; | ||
151 | @@ -XXX,XX +XXX,XX @@ static void _vu_queue_notify(VuDev *dev, VuVirtq *vq, bool sync) | ||
152 | |||
153 | vu_message_write(dev, dev->slave_fd, &vmsg); | ||
154 | if (ack) { | ||
155 | - vu_message_read(dev, dev->slave_fd, &vmsg); | ||
156 | + vu_message_read_default(dev, dev->slave_fd, &vmsg); | ||
157 | } | ||
158 | return; | ||
159 | } | ||
160 | diff --git a/tests/vhost-user-bridge.c b/tests/vhost-user-bridge.c | ||
161 | index XXXXXXX..XXXXXXX 100644 | ||
162 | --- a/tests/vhost-user-bridge.c | ||
163 | +++ b/tests/vhost-user-bridge.c | ||
164 | @@ -XXX,XX +XXX,XX @@ vubr_accept_cb(int sock, void *ctx) | ||
165 | VHOST_USER_BRIDGE_MAX_QUEUES, | ||
166 | conn_fd, | ||
167 | vubr_panic, | ||
168 | + NULL, | ||
169 | vubr_set_watch, | ||
170 | vubr_remove_watch, | ||
171 | &vuiface)) { | ||
172 | @@ -XXX,XX +XXX,XX @@ vubr_new(const char *path, bool client) | ||
173 | VHOST_USER_BRIDGE_MAX_QUEUES, | ||
174 | dev->sock, | ||
175 | vubr_panic, | ||
176 | + NULL, | ||
177 | vubr_set_watch, | ||
178 | vubr_remove_watch, | ||
179 | &vuiface)) { | ||
180 | diff --git a/tools/virtiofsd/fuse_virtio.c b/tools/virtiofsd/fuse_virtio.c | ||
181 | index XXXXXXX..XXXXXXX 100644 | ||
182 | --- a/tools/virtiofsd/fuse_virtio.c | ||
183 | +++ b/tools/virtiofsd/fuse_virtio.c | ||
184 | @@ -XXX,XX +XXX,XX @@ int virtio_session_mount(struct fuse_session *se) | ||
185 | se->vu_socketfd = data_sock; | ||
186 | se->virtio_dev->se = se; | ||
187 | pthread_rwlock_init(&se->virtio_dev->vu_dispatch_rwlock, NULL); | ||
188 | - vu_init(&se->virtio_dev->dev, 2, se->vu_socketfd, fv_panic, fv_set_watch, | ||
189 | - fv_remove_watch, &fv_iface); | ||
190 | + vu_init(&se->virtio_dev->dev, 2, se->vu_socketfd, fv_panic, NULL, | ||
191 | + fv_set_watch, fv_remove_watch, &fv_iface); | ||
192 | |||
193 | return 0; | ||
194 | } | ||
195 | -- | ||
196 | 2.26.2 | ||
197 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Coiby Xu <coiby.xu@gmail.com> | ||
1 | 2 | ||
3 | When the client is running in gdb and quit command is run in gdb, | ||
4 | QEMU will still dispatch the event which will cause segment fault in | ||
5 | the callback function. | ||
6 | |||
7 | Signed-off-by: Coiby Xu <coiby.xu@gmail.com> | ||
8 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
9 | Reviewed-by: Marc-André Lureau <marcandre.lureau@redhat.com> | ||
10 | Message-id: 20200918080912.321299-3-coiby.xu@gmail.com | ||
11 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
12 | --- | ||
13 | contrib/libvhost-user/libvhost-user.c | 1 + | ||
14 | 1 file changed, 1 insertion(+) | ||
15 | |||
16 | diff --git a/contrib/libvhost-user/libvhost-user.c b/contrib/libvhost-user/libvhost-user.c | ||
17 | index XXXXXXX..XXXXXXX 100644 | ||
18 | --- a/contrib/libvhost-user/libvhost-user.c | ||
19 | +++ b/contrib/libvhost-user/libvhost-user.c | ||
20 | @@ -XXX,XX +XXX,XX @@ vu_deinit(VuDev *dev) | ||
21 | } | ||
22 | |||
23 | if (vq->kick_fd != -1) { | ||
24 | + dev->remove_watch(dev, vq->kick_fd); | ||
25 | close(vq->kick_fd); | ||
26 | vq->kick_fd = -1; | ||
27 | } | ||
28 | -- | ||
29 | 2.26.2 | ||
30 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Coiby Xu <coiby.xu@gmail.com> | ||
1 | 2 | ||
3 | Sharing QEMU devices via vhost-user protocol. | ||
4 | |||
5 | Only one vhost-user client can connect to the server one time. | ||
6 | |||
7 | Suggested-by: Kevin Wolf <kwolf@redhat.com> | ||
8 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
9 | Signed-off-by: Coiby Xu <coiby.xu@gmail.com> | ||
10 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
11 | Reviewed-by: Marc-André Lureau <marcandre.lureau@redhat.com> | ||
12 | Message-id: 20200918080912.321299-4-coiby.xu@gmail.com | ||
13 | [Fixed size_t %lu -> %zu format string compiler error. | ||
14 | --Stefan] | ||
15 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
16 | --- | ||
17 | util/vhost-user-server.h | 65 ++++++ | ||
18 | util/vhost-user-server.c | 428 +++++++++++++++++++++++++++++++++++++++ | ||
19 | util/meson.build | 1 + | ||
20 | 3 files changed, 494 insertions(+) | ||
21 | create mode 100644 util/vhost-user-server.h | ||
22 | create mode 100644 util/vhost-user-server.c | ||
23 | |||
24 | diff --git a/util/vhost-user-server.h b/util/vhost-user-server.h | ||
25 | new file mode 100644 | ||
26 | index XXXXXXX..XXXXXXX | ||
27 | --- /dev/null | ||
28 | +++ b/util/vhost-user-server.h | ||
29 | @@ -XXX,XX +XXX,XX @@ | ||
30 | +/* | ||
31 | + * Sharing QEMU devices via vhost-user protocol | ||
32 | + * | ||
33 | + * Copyright (c) Coiby Xu <coiby.xu@gmail.com>. | ||
34 | + * Copyright (c) 2020 Red Hat, Inc. | ||
35 | + * | ||
36 | + * This work is licensed under the terms of the GNU GPL, version 2 or | ||
37 | + * later. See the COPYING file in the top-level directory. | ||
38 | + */ | ||
39 | + | ||
40 | +#ifndef VHOST_USER_SERVER_H | ||
41 | +#define VHOST_USER_SERVER_H | ||
42 | + | ||
43 | +#include "contrib/libvhost-user/libvhost-user.h" | ||
44 | +#include "io/channel-socket.h" | ||
45 | +#include "io/channel-file.h" | ||
46 | +#include "io/net-listener.h" | ||
47 | +#include "qemu/error-report.h" | ||
48 | +#include "qapi/error.h" | ||
49 | +#include "standard-headers/linux/virtio_blk.h" | ||
50 | + | ||
51 | +typedef struct VuFdWatch { | ||
52 | + VuDev *vu_dev; | ||
53 | + int fd; /*kick fd*/ | ||
54 | + void *pvt; | ||
55 | + vu_watch_cb cb; | ||
56 | + bool processing; | ||
57 | + QTAILQ_ENTRY(VuFdWatch) next; | ||
58 | +} VuFdWatch; | ||
59 | + | ||
60 | +typedef struct VuServer VuServer; | ||
61 | +typedef void DevicePanicNotifierFn(VuServer *server); | ||
62 | + | ||
63 | +struct VuServer { | ||
64 | + QIONetListener *listener; | ||
65 | + AioContext *ctx; | ||
66 | + DevicePanicNotifierFn *device_panic_notifier; | ||
67 | + int max_queues; | ||
68 | + const VuDevIface *vu_iface; | ||
69 | + VuDev vu_dev; | ||
70 | + QIOChannel *ioc; /* The I/O channel with the client */ | ||
71 | + QIOChannelSocket *sioc; /* The underlying data channel with the client */ | ||
72 | + /* IOChannel for fd provided via VHOST_USER_SET_SLAVE_REQ_FD */ | ||
73 | + QIOChannel *ioc_slave; | ||
74 | + QIOChannelSocket *sioc_slave; | ||
75 | + Coroutine *co_trip; /* coroutine for processing VhostUserMsg */ | ||
76 | + QTAILQ_HEAD(, VuFdWatch) vu_fd_watches; | ||
77 | + /* restart coroutine co_trip if AIOContext is changed */ | ||
78 | + bool aio_context_changed; | ||
79 | + bool processing_msg; | ||
80 | +}; | ||
81 | + | ||
82 | +bool vhost_user_server_start(VuServer *server, | ||
83 | + SocketAddress *unix_socket, | ||
84 | + AioContext *ctx, | ||
85 | + uint16_t max_queues, | ||
86 | + DevicePanicNotifierFn *device_panic_notifier, | ||
87 | + const VuDevIface *vu_iface, | ||
88 | + Error **errp); | ||
89 | + | ||
90 | +void vhost_user_server_stop(VuServer *server); | ||
91 | + | ||
92 | +void vhost_user_server_set_aio_context(VuServer *server, AioContext *ctx); | ||
93 | + | ||
94 | +#endif /* VHOST_USER_SERVER_H */ | ||
95 | diff --git a/util/vhost-user-server.c b/util/vhost-user-server.c | ||
96 | new file mode 100644 | ||
97 | index XXXXXXX..XXXXXXX | ||
98 | --- /dev/null | ||
99 | +++ b/util/vhost-user-server.c | ||
100 | @@ -XXX,XX +XXX,XX @@ | ||
101 | +/* | ||
102 | + * Sharing QEMU devices via vhost-user protocol | ||
103 | + * | ||
104 | + * Copyright (c) Coiby Xu <coiby.xu@gmail.com>. | ||
105 | + * Copyright (c) 2020 Red Hat, Inc. | ||
106 | + * | ||
107 | + * This work is licensed under the terms of the GNU GPL, version 2 or | ||
108 | + * later. See the COPYING file in the top-level directory. | ||
109 | + */ | ||
110 | +#include "qemu/osdep.h" | ||
111 | +#include "qemu/main-loop.h" | ||
112 | +#include "vhost-user-server.h" | ||
113 | + | ||
114 | +static void vmsg_close_fds(VhostUserMsg *vmsg) | ||
115 | +{ | ||
116 | + int i; | ||
117 | + for (i = 0; i < vmsg->fd_num; i++) { | ||
118 | + close(vmsg->fds[i]); | ||
119 | + } | ||
120 | +} | ||
121 | + | ||
122 | +static void vmsg_unblock_fds(VhostUserMsg *vmsg) | ||
123 | +{ | ||
124 | + int i; | ||
125 | + for (i = 0; i < vmsg->fd_num; i++) { | ||
126 | + qemu_set_nonblock(vmsg->fds[i]); | ||
127 | + } | ||
128 | +} | ||
129 | + | ||
130 | +static void vu_accept(QIONetListener *listener, QIOChannelSocket *sioc, | ||
131 | + gpointer opaque); | ||
132 | + | ||
133 | +static void close_client(VuServer *server) | ||
134 | +{ | ||
135 | + /* | ||
136 | + * Before closing the client | ||
137 | + * | ||
138 | + * 1. Let vu_client_trip stop processing new vhost-user msg | ||
139 | + * | ||
140 | + * 2. remove kick_handler | ||
141 | + * | ||
142 | + * 3. wait for the kick handler to be finished | ||
143 | + * | ||
144 | + * 4. wait for the current vhost-user msg to be finished processing | ||
145 | + */ | ||
146 | + | ||
147 | + QIOChannelSocket *sioc = server->sioc; | ||
148 | + /* When this is set vu_client_trip will stop new processing vhost-user message */ | ||
149 | + server->sioc = NULL; | ||
150 | + | ||
151 | + VuFdWatch *vu_fd_watch, *next; | ||
152 | + QTAILQ_FOREACH_SAFE(vu_fd_watch, &server->vu_fd_watches, next, next) { | ||
153 | + aio_set_fd_handler(server->ioc->ctx, vu_fd_watch->fd, true, NULL, | ||
154 | + NULL, NULL, NULL); | ||
155 | + } | ||
156 | + | ||
157 | + while (!QTAILQ_EMPTY(&server->vu_fd_watches)) { | ||
158 | + QTAILQ_FOREACH_SAFE(vu_fd_watch, &server->vu_fd_watches, next, next) { | ||
159 | + if (!vu_fd_watch->processing) { | ||
160 | + QTAILQ_REMOVE(&server->vu_fd_watches, vu_fd_watch, next); | ||
161 | + g_free(vu_fd_watch); | ||
162 | + } | ||
163 | + } | ||
164 | + } | ||
165 | + | ||
166 | + while (server->processing_msg) { | ||
167 | + if (server->ioc->read_coroutine) { | ||
168 | + server->ioc->read_coroutine = NULL; | ||
169 | + qio_channel_set_aio_fd_handler(server->ioc, server->ioc->ctx, NULL, | ||
170 | + NULL, server->ioc); | ||
171 | + server->processing_msg = false; | ||
172 | + } | ||
173 | + } | ||
174 | + | ||
175 | + vu_deinit(&server->vu_dev); | ||
176 | + object_unref(OBJECT(sioc)); | ||
177 | + object_unref(OBJECT(server->ioc)); | ||
178 | +} | ||
179 | + | ||
180 | +static void panic_cb(VuDev *vu_dev, const char *buf) | ||
181 | +{ | ||
182 | + VuServer *server = container_of(vu_dev, VuServer, vu_dev); | ||
183 | + | ||
184 | + /* avoid while loop in close_client */ | ||
185 | + server->processing_msg = false; | ||
186 | + | ||
187 | + if (buf) { | ||
188 | + error_report("vu_panic: %s", buf); | ||
189 | + } | ||
190 | + | ||
191 | + if (server->sioc) { | ||
192 | + close_client(server); | ||
193 | + } | ||
194 | + | ||
195 | + if (server->device_panic_notifier) { | ||
196 | + server->device_panic_notifier(server); | ||
197 | + } | ||
198 | + | ||
199 | + /* | ||
200 | + * Set the callback function for network listener so another | ||
201 | + * vhost-user client can connect to this server | ||
202 | + */ | ||
203 | + qio_net_listener_set_client_func(server->listener, | ||
204 | + vu_accept, | ||
205 | + server, | ||
206 | + NULL); | ||
207 | +} | ||
208 | + | ||
209 | +static bool coroutine_fn | ||
210 | +vu_message_read(VuDev *vu_dev, int conn_fd, VhostUserMsg *vmsg) | ||
211 | +{ | ||
212 | + struct iovec iov = { | ||
213 | + .iov_base = (char *)vmsg, | ||
214 | + .iov_len = VHOST_USER_HDR_SIZE, | ||
215 | + }; | ||
216 | + int rc, read_bytes = 0; | ||
217 | + Error *local_err = NULL; | ||
218 | + /* | ||
219 | + * Store fds/nfds returned from qio_channel_readv_full into | ||
220 | + * temporary variables. | ||
221 | + * | ||
222 | + * VhostUserMsg is a packed structure, gcc will complain about passing | ||
223 | + * pointer to a packed structure member if we pass &VhostUserMsg.fd_num | ||
224 | + * and &VhostUserMsg.fds directly when calling qio_channel_readv_full, | ||
225 | + * thus two temporary variables nfds and fds are used here. | ||
226 | + */ | ||
227 | + size_t nfds = 0, nfds_t = 0; | ||
228 | + const size_t max_fds = G_N_ELEMENTS(vmsg->fds); | ||
229 | + int *fds_t = NULL; | ||
230 | + VuServer *server = container_of(vu_dev, VuServer, vu_dev); | ||
231 | + QIOChannel *ioc = server->ioc; | ||
232 | + | ||
233 | + if (!ioc) { | ||
234 | + error_report_err(local_err); | ||
235 | + goto fail; | ||
236 | + } | ||
237 | + | ||
238 | + assert(qemu_in_coroutine()); | ||
239 | + do { | ||
240 | + /* | ||
241 | + * qio_channel_readv_full may have short reads, keeping calling it | ||
242 | + * until getting VHOST_USER_HDR_SIZE or 0 bytes in total | ||
243 | + */ | ||
244 | + rc = qio_channel_readv_full(ioc, &iov, 1, &fds_t, &nfds_t, &local_err); | ||
245 | + if (rc < 0) { | ||
246 | + if (rc == QIO_CHANNEL_ERR_BLOCK) { | ||
247 | + qio_channel_yield(ioc, G_IO_IN); | ||
248 | + continue; | ||
249 | + } else { | ||
250 | + error_report_err(local_err); | ||
251 | + return false; | ||
252 | + } | ||
253 | + } | ||
254 | + read_bytes += rc; | ||
255 | + if (nfds_t > 0) { | ||
256 | + if (nfds + nfds_t > max_fds) { | ||
257 | + error_report("A maximum of %zu fds are allowed, " | ||
258 | + "however got %zu fds now", | ||
259 | + max_fds, nfds + nfds_t); | ||
260 | + goto fail; | ||
261 | + } | ||
262 | + memcpy(vmsg->fds + nfds, fds_t, | ||
263 | + nfds_t *sizeof(vmsg->fds[0])); | ||
264 | + nfds += nfds_t; | ||
265 | + g_free(fds_t); | ||
266 | + } | ||
267 | + if (read_bytes == VHOST_USER_HDR_SIZE || rc == 0) { | ||
268 | + break; | ||
269 | + } | ||
270 | + iov.iov_base = (char *)vmsg + read_bytes; | ||
271 | + iov.iov_len = VHOST_USER_HDR_SIZE - read_bytes; | ||
272 | + } while (true); | ||
273 | + | ||
274 | + vmsg->fd_num = nfds; | ||
275 | + /* qio_channel_readv_full will make socket fds blocking, unblock them */ | ||
276 | + vmsg_unblock_fds(vmsg); | ||
277 | + if (vmsg->size > sizeof(vmsg->payload)) { | ||
278 | + error_report("Error: too big message request: %d, " | ||
279 | + "size: vmsg->size: %u, " | ||
280 | + "while sizeof(vmsg->payload) = %zu", | ||
281 | + vmsg->request, vmsg->size, sizeof(vmsg->payload)); | ||
282 | + goto fail; | ||
283 | + } | ||
284 | + | ||
285 | + struct iovec iov_payload = { | ||
286 | + .iov_base = (char *)&vmsg->payload, | ||
287 | + .iov_len = vmsg->size, | ||
288 | + }; | ||
289 | + if (vmsg->size) { | ||
290 | + rc = qio_channel_readv_all_eof(ioc, &iov_payload, 1, &local_err); | ||
291 | + if (rc == -1) { | ||
292 | + error_report_err(local_err); | ||
293 | + goto fail; | ||
294 | + } | ||
295 | + } | ||
296 | + | ||
297 | + return true; | ||
298 | + | ||
299 | +fail: | ||
300 | + vmsg_close_fds(vmsg); | ||
301 | + | ||
302 | + return false; | ||
303 | +} | ||
304 | + | ||
305 | + | ||
306 | +static void vu_client_start(VuServer *server); | ||
307 | +static coroutine_fn void vu_client_trip(void *opaque) | ||
308 | +{ | ||
309 | + VuServer *server = opaque; | ||
310 | + | ||
311 | + while (!server->aio_context_changed && server->sioc) { | ||
312 | + server->processing_msg = true; | ||
313 | + vu_dispatch(&server->vu_dev); | ||
314 | + server->processing_msg = false; | ||
315 | + } | ||
316 | + | ||
317 | + if (server->aio_context_changed && server->sioc) { | ||
318 | + server->aio_context_changed = false; | ||
319 | + vu_client_start(server); | ||
320 | + } | ||
321 | +} | ||
322 | + | ||
323 | +static void vu_client_start(VuServer *server) | ||
324 | +{ | ||
325 | + server->co_trip = qemu_coroutine_create(vu_client_trip, server); | ||
326 | + aio_co_enter(server->ctx, server->co_trip); | ||
327 | +} | ||
328 | + | ||
329 | +/* | ||
330 | + * a wrapper for vu_kick_cb | ||
331 | + * | ||
332 | + * since aio_dispatch can only pass one user data pointer to the | ||
333 | + * callback function, pack VuDev and pvt into a struct. Then unpack it | ||
334 | + * and pass them to vu_kick_cb | ||
335 | + */ | ||
336 | +static void kick_handler(void *opaque) | ||
337 | +{ | ||
338 | + VuFdWatch *vu_fd_watch = opaque; | ||
339 | + vu_fd_watch->processing = true; | ||
340 | + vu_fd_watch->cb(vu_fd_watch->vu_dev, 0, vu_fd_watch->pvt); | ||
341 | + vu_fd_watch->processing = false; | ||
342 | +} | ||
343 | + | ||
344 | + | ||
345 | +static VuFdWatch *find_vu_fd_watch(VuServer *server, int fd) | ||
346 | +{ | ||
347 | + | ||
348 | + VuFdWatch *vu_fd_watch, *next; | ||
349 | + QTAILQ_FOREACH_SAFE(vu_fd_watch, &server->vu_fd_watches, next, next) { | ||
350 | + if (vu_fd_watch->fd == fd) { | ||
351 | + return vu_fd_watch; | ||
352 | + } | ||
353 | + } | ||
354 | + return NULL; | ||
355 | +} | ||
356 | + | ||
357 | +static void | ||
358 | +set_watch(VuDev *vu_dev, int fd, int vu_evt, | ||
359 | + vu_watch_cb cb, void *pvt) | ||
360 | +{ | ||
361 | + | ||
362 | + VuServer *server = container_of(vu_dev, VuServer, vu_dev); | ||
363 | + g_assert(vu_dev); | ||
364 | + g_assert(fd >= 0); | ||
365 | + g_assert(cb); | ||
366 | + | ||
367 | + VuFdWatch *vu_fd_watch = find_vu_fd_watch(server, fd); | ||
368 | + | ||
369 | + if (!vu_fd_watch) { | ||
370 | + VuFdWatch *vu_fd_watch = g_new0(VuFdWatch, 1); | ||
371 | + | ||
372 | + QTAILQ_INSERT_TAIL(&server->vu_fd_watches, vu_fd_watch, next); | ||
373 | + | ||
374 | + vu_fd_watch->fd = fd; | ||
375 | + vu_fd_watch->cb = cb; | ||
376 | + qemu_set_nonblock(fd); | ||
377 | + aio_set_fd_handler(server->ioc->ctx, fd, true, kick_handler, | ||
378 | + NULL, NULL, vu_fd_watch); | ||
379 | + vu_fd_watch->vu_dev = vu_dev; | ||
380 | + vu_fd_watch->pvt = pvt; | ||
381 | + } | ||
382 | +} | ||
383 | + | ||
384 | + | ||
385 | +static void remove_watch(VuDev *vu_dev, int fd) | ||
386 | +{ | ||
387 | + VuServer *server; | ||
388 | + g_assert(vu_dev); | ||
389 | + g_assert(fd >= 0); | ||
390 | + | ||
391 | + server = container_of(vu_dev, VuServer, vu_dev); | ||
392 | + | ||
393 | + VuFdWatch *vu_fd_watch = find_vu_fd_watch(server, fd); | ||
394 | + | ||
395 | + if (!vu_fd_watch) { | ||
396 | + return; | ||
397 | + } | ||
398 | + aio_set_fd_handler(server->ioc->ctx, fd, true, NULL, NULL, NULL, NULL); | ||
399 | + | ||
400 | + QTAILQ_REMOVE(&server->vu_fd_watches, vu_fd_watch, next); | ||
401 | + g_free(vu_fd_watch); | ||
402 | +} | ||
403 | + | ||
404 | + | ||
405 | +static void vu_accept(QIONetListener *listener, QIOChannelSocket *sioc, | ||
406 | + gpointer opaque) | ||
407 | +{ | ||
408 | + VuServer *server = opaque; | ||
409 | + | ||
410 | + if (server->sioc) { | ||
411 | + warn_report("Only one vhost-user client is allowed to " | ||
412 | + "connect the server one time"); | ||
413 | + return; | ||
414 | + } | ||
415 | + | ||
416 | + if (!vu_init(&server->vu_dev, server->max_queues, sioc->fd, panic_cb, | ||
417 | + vu_message_read, set_watch, remove_watch, server->vu_iface)) { | ||
418 | + error_report("Failed to initialize libvhost-user"); | ||
419 | + return; | ||
420 | + } | ||
421 | + | ||
422 | + /* | ||
423 | + * Unset the callback function for network listener to make another | ||
424 | + * vhost-user client keeping waiting until this client disconnects | ||
425 | + */ | ||
426 | + qio_net_listener_set_client_func(server->listener, | ||
427 | + NULL, | ||
428 | + NULL, | ||
429 | + NULL); | ||
430 | + server->sioc = sioc; | ||
431 | + /* | ||
432 | + * Increase the object reference, so sioc will not freed by | ||
433 | + * qio_net_listener_channel_func which will call object_unref(OBJECT(sioc)) | ||
434 | + */ | ||
435 | + object_ref(OBJECT(server->sioc)); | ||
436 | + qio_channel_set_name(QIO_CHANNEL(sioc), "vhost-user client"); | ||
437 | + server->ioc = QIO_CHANNEL(sioc); | ||
438 | + object_ref(OBJECT(server->ioc)); | ||
439 | + qio_channel_attach_aio_context(server->ioc, server->ctx); | ||
440 | + qio_channel_set_blocking(QIO_CHANNEL(server->sioc), false, NULL); | ||
441 | + vu_client_start(server); | ||
442 | +} | ||
443 | + | ||
444 | + | ||
445 | +void vhost_user_server_stop(VuServer *server) | ||
446 | +{ | ||
447 | + if (server->sioc) { | ||
448 | + close_client(server); | ||
449 | + } | ||
450 | + | ||
451 | + if (server->listener) { | ||
452 | + qio_net_listener_disconnect(server->listener); | ||
453 | + object_unref(OBJECT(server->listener)); | ||
454 | + } | ||
455 | + | ||
456 | +} | ||
457 | + | ||
458 | +void vhost_user_server_set_aio_context(VuServer *server, AioContext *ctx) | ||
459 | +{ | ||
460 | + VuFdWatch *vu_fd_watch, *next; | ||
461 | + void *opaque = NULL; | ||
462 | + IOHandler *io_read = NULL; | ||
463 | + bool attach; | ||
464 | + | ||
465 | + server->ctx = ctx ? ctx : qemu_get_aio_context(); | ||
466 | + | ||
467 | + if (!server->sioc) { | ||
468 | + /* not yet serving any client*/ | ||
469 | + return; | ||
470 | + } | ||
471 | + | ||
472 | + if (ctx) { | ||
473 | + qio_channel_attach_aio_context(server->ioc, ctx); | ||
474 | + server->aio_context_changed = true; | ||
475 | + io_read = kick_handler; | ||
476 | + attach = true; | ||
477 | + } else { | ||
478 | + qio_channel_detach_aio_context(server->ioc); | ||
479 | + /* server->ioc->ctx keeps the old AioConext */ | ||
480 | + ctx = server->ioc->ctx; | ||
481 | + attach = false; | ||
482 | + } | ||
483 | + | ||
484 | + QTAILQ_FOREACH_SAFE(vu_fd_watch, &server->vu_fd_watches, next, next) { | ||
485 | + if (vu_fd_watch->cb) { | ||
486 | + opaque = attach ? vu_fd_watch : NULL; | ||
487 | + aio_set_fd_handler(ctx, vu_fd_watch->fd, true, | ||
488 | + io_read, NULL, NULL, | ||
489 | + opaque); | ||
490 | + } | ||
491 | + } | ||
492 | +} | ||
493 | + | ||
494 | + | ||
495 | +bool vhost_user_server_start(VuServer *server, | ||
496 | + SocketAddress *socket_addr, | ||
497 | + AioContext *ctx, | ||
498 | + uint16_t max_queues, | ||
499 | + DevicePanicNotifierFn *device_panic_notifier, | ||
500 | + const VuDevIface *vu_iface, | ||
501 | + Error **errp) | ||
502 | +{ | ||
503 | + QIONetListener *listener = qio_net_listener_new(); | ||
504 | + if (qio_net_listener_open_sync(listener, socket_addr, 1, | ||
505 | + errp) < 0) { | ||
506 | + object_unref(OBJECT(listener)); | ||
507 | + return false; | ||
508 | + } | ||
509 | + | ||
510 | + /* zero out unspecified fileds */ | ||
511 | + *server = (VuServer) { | ||
512 | + .listener = listener, | ||
513 | + .vu_iface = vu_iface, | ||
514 | + .max_queues = max_queues, | ||
515 | + .ctx = ctx, | ||
516 | + .device_panic_notifier = device_panic_notifier, | ||
517 | + }; | ||
518 | + | ||
519 | + qio_net_listener_set_name(server->listener, "vhost-user-backend-listener"); | ||
520 | + | ||
521 | + qio_net_listener_set_client_func(server->listener, | ||
522 | + vu_accept, | ||
523 | + server, | ||
524 | + NULL); | ||
525 | + | ||
526 | + QTAILQ_INIT(&server->vu_fd_watches); | ||
527 | + return true; | ||
528 | +} | ||
529 | diff --git a/util/meson.build b/util/meson.build | ||
530 | index XXXXXXX..XXXXXXX 100644 | ||
531 | --- a/util/meson.build | ||
532 | +++ b/util/meson.build | ||
533 | @@ -XXX,XX +XXX,XX @@ if have_block | ||
534 | util_ss.add(files('main-loop.c')) | ||
535 | util_ss.add(files('nvdimm-utils.c')) | ||
536 | util_ss.add(files('qemu-coroutine.c', 'qemu-coroutine-lock.c', 'qemu-coroutine-io.c')) | ||
537 | + util_ss.add(when: 'CONFIG_LINUX', if_true: files('vhost-user-server.c')) | ||
538 | util_ss.add(files('qemu-coroutine-sleep.c')) | ||
539 | util_ss.add(files('qemu-co-shared-resource.c')) | ||
540 | util_ss.add(files('thread-pool.c', 'qemu-timer.c')) | ||
541 | -- | ||
542 | 2.26.2 | ||
543 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Coiby Xu <coiby.xu@gmail.com> | ||
1 | 2 | ||
3 | Move the constants from hw/core/qdev-properties.c to | ||
4 | util/block-helpers.h so that knowledge of the min/max values is | ||
5 | |||
6 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
7 | Signed-off-by: Coiby Xu <coiby.xu@gmail.com> | ||
8 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
9 | Reviewed-by: Marc-André Lureau <marcandre.lureau@redhat.com> | ||
10 | Acked-by: Eduardo Habkost <ehabkost@redhat.com> | ||
11 | Message-id: 20200918080912.321299-5-coiby.xu@gmail.com | ||
12 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
13 | --- | ||
14 | util/block-helpers.h | 19 +++++++++++++ | ||
15 | hw/core/qdev-properties-system.c | 31 ++++----------------- | ||
16 | util/block-helpers.c | 46 ++++++++++++++++++++++++++++++++ | ||
17 | util/meson.build | 1 + | ||
18 | 4 files changed, 71 insertions(+), 26 deletions(-) | ||
19 | create mode 100644 util/block-helpers.h | ||
20 | create mode 100644 util/block-helpers.c | ||
21 | |||
22 | diff --git a/util/block-helpers.h b/util/block-helpers.h | ||
23 | new file mode 100644 | ||
24 | index XXXXXXX..XXXXXXX | ||
25 | --- /dev/null | ||
26 | +++ b/util/block-helpers.h | ||
27 | @@ -XXX,XX +XXX,XX @@ | ||
28 | +#ifndef BLOCK_HELPERS_H | ||
29 | +#define BLOCK_HELPERS_H | ||
30 | + | ||
31 | +#include "qemu/units.h" | ||
32 | + | ||
33 | +/* lower limit is sector size */ | ||
34 | +#define MIN_BLOCK_SIZE INT64_C(512) | ||
35 | +#define MIN_BLOCK_SIZE_STR "512 B" | ||
36 | +/* | ||
37 | + * upper limit is arbitrary, 2 MiB looks sufficient for all sensible uses, and | ||
38 | + * matches qcow2 cluster size limit | ||
39 | + */ | ||
40 | +#define MAX_BLOCK_SIZE (2 * MiB) | ||
41 | +#define MAX_BLOCK_SIZE_STR "2 MiB" | ||
42 | + | ||
43 | +void check_block_size(const char *id, const char *name, int64_t value, | ||
44 | + Error **errp); | ||
45 | + | ||
46 | +#endif /* BLOCK_HELPERS_H */ | ||
47 | diff --git a/hw/core/qdev-properties-system.c b/hw/core/qdev-properties-system.c | ||
48 | index XXXXXXX..XXXXXXX 100644 | ||
49 | --- a/hw/core/qdev-properties-system.c | ||
50 | +++ b/hw/core/qdev-properties-system.c | ||
51 | @@ -XXX,XX +XXX,XX @@ | ||
52 | #include "sysemu/blockdev.h" | ||
53 | #include "net/net.h" | ||
54 | #include "hw/pci/pci.h" | ||
55 | +#include "util/block-helpers.h" | ||
56 | |||
57 | static bool check_prop_still_unset(DeviceState *dev, const char *name, | ||
58 | const void *old_val, const char *new_val, | ||
59 | @@ -XXX,XX +XXX,XX @@ const PropertyInfo qdev_prop_losttickpolicy = { | ||
60 | |||
61 | /* --- blocksize --- */ | ||
62 | |||
63 | -/* lower limit is sector size */ | ||
64 | -#define MIN_BLOCK_SIZE 512 | ||
65 | -#define MIN_BLOCK_SIZE_STR "512 B" | ||
66 | -/* | ||
67 | - * upper limit is arbitrary, 2 MiB looks sufficient for all sensible uses, and | ||
68 | - * matches qcow2 cluster size limit | ||
69 | - */ | ||
70 | -#define MAX_BLOCK_SIZE (2 * MiB) | ||
71 | -#define MAX_BLOCK_SIZE_STR "2 MiB" | ||
72 | - | ||
73 | static void set_blocksize(Object *obj, Visitor *v, const char *name, | ||
74 | void *opaque, Error **errp) | ||
75 | { | ||
76 | @@ -XXX,XX +XXX,XX @@ static void set_blocksize(Object *obj, Visitor *v, const char *name, | ||
77 | Property *prop = opaque; | ||
78 | uint32_t *ptr = qdev_get_prop_ptr(dev, prop); | ||
79 | uint64_t value; | ||
80 | + Error *local_err = NULL; | ||
81 | |||
82 | if (dev->realized) { | ||
83 | qdev_prop_set_after_realize(dev, name, errp); | ||
84 | @@ -XXX,XX +XXX,XX @@ static void set_blocksize(Object *obj, Visitor *v, const char *name, | ||
85 | if (!visit_type_size(v, name, &value, errp)) { | ||
86 | return; | ||
87 | } | ||
88 | - /* value of 0 means "unset" */ | ||
89 | - if (value && (value < MIN_BLOCK_SIZE || value > MAX_BLOCK_SIZE)) { | ||
90 | - error_setg(errp, | ||
91 | - "Property %s.%s doesn't take value %" PRIu64 | ||
92 | - " (minimum: " MIN_BLOCK_SIZE_STR | ||
93 | - ", maximum: " MAX_BLOCK_SIZE_STR ")", | ||
94 | - dev->id ? : "", name, value); | ||
95 | + check_block_size(dev->id ? : "", name, value, &local_err); | ||
96 | + if (local_err) { | ||
97 | + error_propagate(errp, local_err); | ||
98 | return; | ||
99 | } | ||
100 | - | ||
101 | - /* We rely on power-of-2 blocksizes for bitmasks */ | ||
102 | - if ((value & (value - 1)) != 0) { | ||
103 | - error_setg(errp, | ||
104 | - "Property %s.%s doesn't take value '%" PRId64 "', " | ||
105 | - "it's not a power of 2", dev->id ?: "", name, (int64_t)value); | ||
106 | - return; | ||
107 | - } | ||
108 | - | ||
109 | *ptr = value; | ||
110 | } | ||
111 | |||
112 | diff --git a/util/block-helpers.c b/util/block-helpers.c | ||
113 | new file mode 100644 | ||
114 | index XXXXXXX..XXXXXXX | ||
115 | --- /dev/null | ||
116 | +++ b/util/block-helpers.c | ||
117 | @@ -XXX,XX +XXX,XX @@ | ||
118 | +/* | ||
119 | + * Block utility functions | ||
120 | + * | ||
121 | + * Copyright IBM, Corp. 2011 | ||
122 | + * Copyright (c) 2020 Coiby Xu <coiby.xu@gmail.com> | ||
123 | + * | ||
124 | + * This work is licensed under the terms of the GNU GPL, version 2 or later. | ||
125 | + * See the COPYING file in the top-level directory. | ||
126 | + */ | ||
127 | + | ||
128 | +#include "qemu/osdep.h" | ||
129 | +#include "qapi/error.h" | ||
130 | +#include "qapi/qmp/qerror.h" | ||
131 | +#include "block-helpers.h" | ||
132 | + | ||
133 | +/** | ||
134 | + * check_block_size: | ||
135 | + * @id: The unique ID of the object | ||
136 | + * @name: The name of the property being validated | ||
137 | + * @value: The block size in bytes | ||
138 | + * @errp: A pointer to an area to store an error | ||
139 | + * | ||
140 | + * This function checks that the block size meets the following conditions: | ||
141 | + * 1. At least MIN_BLOCK_SIZE | ||
142 | + * 2. No larger than MAX_BLOCK_SIZE | ||
143 | + * 3. A power of 2 | ||
144 | + */ | ||
145 | +void check_block_size(const char *id, const char *name, int64_t value, | ||
146 | + Error **errp) | ||
147 | +{ | ||
148 | + /* value of 0 means "unset" */ | ||
149 | + if (value && (value < MIN_BLOCK_SIZE || value > MAX_BLOCK_SIZE)) { | ||
150 | + error_setg(errp, QERR_PROPERTY_VALUE_OUT_OF_RANGE, | ||
151 | + id, name, value, MIN_BLOCK_SIZE, MAX_BLOCK_SIZE); | ||
152 | + return; | ||
153 | + } | ||
154 | + | ||
155 | + /* We rely on power-of-2 blocksizes for bitmasks */ | ||
156 | + if ((value & (value - 1)) != 0) { | ||
157 | + error_setg(errp, | ||
158 | + "Property %s.%s doesn't take value '%" PRId64 | ||
159 | + "', it's not a power of 2", | ||
160 | + id, name, value); | ||
161 | + return; | ||
162 | + } | ||
163 | +} | ||
164 | diff --git a/util/meson.build b/util/meson.build | ||
165 | index XXXXXXX..XXXXXXX 100644 | ||
166 | --- a/util/meson.build | ||
167 | +++ b/util/meson.build | ||
168 | @@ -XXX,XX +XXX,XX @@ if have_block | ||
169 | util_ss.add(files('nvdimm-utils.c')) | ||
170 | util_ss.add(files('qemu-coroutine.c', 'qemu-coroutine-lock.c', 'qemu-coroutine-io.c')) | ||
171 | util_ss.add(when: 'CONFIG_LINUX', if_true: files('vhost-user-server.c')) | ||
172 | + util_ss.add(files('block-helpers.c')) | ||
173 | util_ss.add(files('qemu-coroutine-sleep.c')) | ||
174 | util_ss.add(files('qemu-co-shared-resource.c')) | ||
175 | util_ss.add(files('thread-pool.c', 'qemu-timer.c')) | ||
176 | -- | ||
177 | 2.26.2 | ||
178 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Coiby Xu <coiby.xu@gmail.com> | ||
1 | 2 | ||
3 | By making use of libvhost-user, block device drive can be shared to | ||
4 | the connected vhost-user client. Only one client can connect to the | ||
5 | server one time. | ||
6 | |||
7 | Since vhost-user-server needs a block drive to be created first, delay | ||
8 | the creation of this object. | ||
9 | |||
10 | Suggested-by: Kevin Wolf <kwolf@redhat.com> | ||
11 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
12 | Signed-off-by: Coiby Xu <coiby.xu@gmail.com> | ||
13 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
14 | Reviewed-by: Marc-André Lureau <marcandre.lureau@redhat.com> | ||
15 | Message-id: 20200918080912.321299-6-coiby.xu@gmail.com | ||
16 | [Shorten "vhost_user_blk_server" string to "vhost_user_blk" to avoid the | ||
17 | following compiler warning: | ||
18 | ../block/export/vhost-user-blk-server.c:178:50: error: ‘%s’ directive output truncated writing 21 bytes into a region of size 20 [-Werror=format-truncation=] | ||
19 | and fix "Invalid size %ld ..." ssize_t format string arguments for | ||
20 | 32-bit hosts. | ||
21 | --Stefan] | ||
22 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
23 | --- | ||
24 | block/export/vhost-user-blk-server.h | 36 ++ | ||
25 | block/export/vhost-user-blk-server.c | 661 +++++++++++++++++++++++++++ | ||
26 | softmmu/vl.c | 4 + | ||
27 | block/meson.build | 1 + | ||
28 | 4 files changed, 702 insertions(+) | ||
29 | create mode 100644 block/export/vhost-user-blk-server.h | ||
30 | create mode 100644 block/export/vhost-user-blk-server.c | ||
31 | |||
32 | diff --git a/block/export/vhost-user-blk-server.h b/block/export/vhost-user-blk-server.h | ||
33 | new file mode 100644 | ||
34 | index XXXXXXX..XXXXXXX | ||
35 | --- /dev/null | ||
36 | +++ b/block/export/vhost-user-blk-server.h | ||
37 | @@ -XXX,XX +XXX,XX @@ | ||
38 | +/* | ||
39 | + * Sharing QEMU block devices via vhost-user protocal | ||
40 | + * | ||
41 | + * Copyright (c) Coiby Xu <coiby.xu@gmail.com>. | ||
42 | + * Copyright (c) 2020 Red Hat, Inc. | ||
43 | + * | ||
44 | + * This work is licensed under the terms of the GNU GPL, version 2 or | ||
45 | + * later. See the COPYING file in the top-level directory. | ||
46 | + */ | ||
47 | + | ||
48 | +#ifndef VHOST_USER_BLK_SERVER_H | ||
49 | +#define VHOST_USER_BLK_SERVER_H | ||
50 | +#include "util/vhost-user-server.h" | ||
51 | + | ||
52 | +typedef struct VuBlockDev VuBlockDev; | ||
53 | +#define TYPE_VHOST_USER_BLK_SERVER "vhost-user-blk-server" | ||
54 | +#define VHOST_USER_BLK_SERVER(obj) \ | ||
55 | + OBJECT_CHECK(VuBlockDev, obj, TYPE_VHOST_USER_BLK_SERVER) | ||
56 | + | ||
57 | +/* vhost user block device */ | ||
58 | +struct VuBlockDev { | ||
59 | + Object parent_obj; | ||
60 | + char *node_name; | ||
61 | + SocketAddress *addr; | ||
62 | + AioContext *ctx; | ||
63 | + VuServer vu_server; | ||
64 | + bool running; | ||
65 | + uint32_t blk_size; | ||
66 | + BlockBackend *backend; | ||
67 | + QIOChannelSocket *sioc; | ||
68 | + QTAILQ_ENTRY(VuBlockDev) next; | ||
69 | + struct virtio_blk_config blkcfg; | ||
70 | + bool writable; | ||
71 | +}; | ||
72 | + | ||
73 | +#endif /* VHOST_USER_BLK_SERVER_H */ | ||
74 | diff --git a/block/export/vhost-user-blk-server.c b/block/export/vhost-user-blk-server.c | ||
75 | new file mode 100644 | ||
76 | index XXXXXXX..XXXXXXX | ||
77 | --- /dev/null | ||
78 | +++ b/block/export/vhost-user-blk-server.c | ||
79 | @@ -XXX,XX +XXX,XX @@ | ||
80 | +/* | ||
81 | + * Sharing QEMU block devices via vhost-user protocal | ||
82 | + * | ||
83 | + * Parts of the code based on nbd/server.c. | ||
84 | + * | ||
85 | + * Copyright (c) Coiby Xu <coiby.xu@gmail.com>. | ||
86 | + * Copyright (c) 2020 Red Hat, Inc. | ||
87 | + * | ||
88 | + * This work is licensed under the terms of the GNU GPL, version 2 or | ||
89 | + * later. See the COPYING file in the top-level directory. | ||
90 | + */ | ||
91 | +#include "qemu/osdep.h" | ||
92 | +#include "block/block.h" | ||
93 | +#include "vhost-user-blk-server.h" | ||
94 | +#include "qapi/error.h" | ||
95 | +#include "qom/object_interfaces.h" | ||
96 | +#include "sysemu/block-backend.h" | ||
97 | +#include "util/block-helpers.h" | ||
98 | + | ||
99 | +enum { | ||
100 | + VHOST_USER_BLK_MAX_QUEUES = 1, | ||
101 | +}; | ||
102 | +struct virtio_blk_inhdr { | ||
103 | + unsigned char status; | ||
104 | +}; | ||
105 | + | ||
106 | +typedef struct VuBlockReq { | ||
107 | + VuVirtqElement *elem; | ||
108 | + int64_t sector_num; | ||
109 | + size_t size; | ||
110 | + struct virtio_blk_inhdr *in; | ||
111 | + struct virtio_blk_outhdr out; | ||
112 | + VuServer *server; | ||
113 | + struct VuVirtq *vq; | ||
114 | +} VuBlockReq; | ||
115 | + | ||
116 | +static void vu_block_req_complete(VuBlockReq *req) | ||
117 | +{ | ||
118 | + VuDev *vu_dev = &req->server->vu_dev; | ||
119 | + | ||
120 | + /* IO size with 1 extra status byte */ | ||
121 | + vu_queue_push(vu_dev, req->vq, req->elem, req->size + 1); | ||
122 | + vu_queue_notify(vu_dev, req->vq); | ||
123 | + | ||
124 | + if (req->elem) { | ||
125 | + free(req->elem); | ||
126 | + } | ||
127 | + | ||
128 | + g_free(req); | ||
129 | +} | ||
130 | + | ||
131 | +static VuBlockDev *get_vu_block_device_by_server(VuServer *server) | ||
132 | +{ | ||
133 | + return container_of(server, VuBlockDev, vu_server); | ||
134 | +} | ||
135 | + | ||
136 | +static int coroutine_fn | ||
137 | +vu_block_discard_write_zeroes(VuBlockReq *req, struct iovec *iov, | ||
138 | + uint32_t iovcnt, uint32_t type) | ||
139 | +{ | ||
140 | + struct virtio_blk_discard_write_zeroes desc; | ||
141 | + ssize_t size = iov_to_buf(iov, iovcnt, 0, &desc, sizeof(desc)); | ||
142 | + if (unlikely(size != sizeof(desc))) { | ||
143 | + error_report("Invalid size %zd, expect %zu", size, sizeof(desc)); | ||
144 | + return -EINVAL; | ||
145 | + } | ||
146 | + | ||
147 | + VuBlockDev *vdev_blk = get_vu_block_device_by_server(req->server); | ||
148 | + uint64_t range[2] = { le64_to_cpu(desc.sector) << 9, | ||
149 | + le32_to_cpu(desc.num_sectors) << 9 }; | ||
150 | + if (type == VIRTIO_BLK_T_DISCARD) { | ||
151 | + if (blk_co_pdiscard(vdev_blk->backend, range[0], range[1]) == 0) { | ||
152 | + return 0; | ||
153 | + } | ||
154 | + } else if (type == VIRTIO_BLK_T_WRITE_ZEROES) { | ||
155 | + if (blk_co_pwrite_zeroes(vdev_blk->backend, | ||
156 | + range[0], range[1], 0) == 0) { | ||
157 | + return 0; | ||
158 | + } | ||
159 | + } | ||
160 | + | ||
161 | + return -EINVAL; | ||
162 | +} | ||
163 | + | ||
164 | +static void coroutine_fn vu_block_flush(VuBlockReq *req) | ||
165 | +{ | ||
166 | + VuBlockDev *vdev_blk = get_vu_block_device_by_server(req->server); | ||
167 | + BlockBackend *backend = vdev_blk->backend; | ||
168 | + blk_co_flush(backend); | ||
169 | +} | ||
170 | + | ||
171 | +struct req_data { | ||
172 | + VuServer *server; | ||
173 | + VuVirtq *vq; | ||
174 | + VuVirtqElement *elem; | ||
175 | +}; | ||
176 | + | ||
177 | +static void coroutine_fn vu_block_virtio_process_req(void *opaque) | ||
178 | +{ | ||
179 | + struct req_data *data = opaque; | ||
180 | + VuServer *server = data->server; | ||
181 | + VuVirtq *vq = data->vq; | ||
182 | + VuVirtqElement *elem = data->elem; | ||
183 | + uint32_t type; | ||
184 | + VuBlockReq *req; | ||
185 | + | ||
186 | + VuBlockDev *vdev_blk = get_vu_block_device_by_server(server); | ||
187 | + BlockBackend *backend = vdev_blk->backend; | ||
188 | + | ||
189 | + struct iovec *in_iov = elem->in_sg; | ||
190 | + struct iovec *out_iov = elem->out_sg; | ||
191 | + unsigned in_num = elem->in_num; | ||
192 | + unsigned out_num = elem->out_num; | ||
193 | + /* refer to hw/block/virtio_blk.c */ | ||
194 | + if (elem->out_num < 1 || elem->in_num < 1) { | ||
195 | + error_report("virtio-blk request missing headers"); | ||
196 | + free(elem); | ||
197 | + return; | ||
198 | + } | ||
199 | + | ||
200 | + req = g_new0(VuBlockReq, 1); | ||
201 | + req->server = server; | ||
202 | + req->vq = vq; | ||
203 | + req->elem = elem; | ||
204 | + | ||
205 | + if (unlikely(iov_to_buf(out_iov, out_num, 0, &req->out, | ||
206 | + sizeof(req->out)) != sizeof(req->out))) { | ||
207 | + error_report("virtio-blk request outhdr too short"); | ||
208 | + goto err; | ||
209 | + } | ||
210 | + | ||
211 | + iov_discard_front(&out_iov, &out_num, sizeof(req->out)); | ||
212 | + | ||
213 | + if (in_iov[in_num - 1].iov_len < sizeof(struct virtio_blk_inhdr)) { | ||
214 | + error_report("virtio-blk request inhdr too short"); | ||
215 | + goto err; | ||
216 | + } | ||
217 | + | ||
218 | + /* We always touch the last byte, so just see how big in_iov is. */ | ||
219 | + req->in = (void *)in_iov[in_num - 1].iov_base | ||
220 | + + in_iov[in_num - 1].iov_len | ||
221 | + - sizeof(struct virtio_blk_inhdr); | ||
222 | + iov_discard_back(in_iov, &in_num, sizeof(struct virtio_blk_inhdr)); | ||
223 | + | ||
224 | + type = le32_to_cpu(req->out.type); | ||
225 | + switch (type & ~VIRTIO_BLK_T_BARRIER) { | ||
226 | + case VIRTIO_BLK_T_IN: | ||
227 | + case VIRTIO_BLK_T_OUT: { | ||
228 | + ssize_t ret = 0; | ||
229 | + bool is_write = type & VIRTIO_BLK_T_OUT; | ||
230 | + req->sector_num = le64_to_cpu(req->out.sector); | ||
231 | + | ||
232 | + int64_t offset = req->sector_num * vdev_blk->blk_size; | ||
233 | + QEMUIOVector qiov; | ||
234 | + if (is_write) { | ||
235 | + qemu_iovec_init_external(&qiov, out_iov, out_num); | ||
236 | + ret = blk_co_pwritev(backend, offset, qiov.size, | ||
237 | + &qiov, 0); | ||
238 | + } else { | ||
239 | + qemu_iovec_init_external(&qiov, in_iov, in_num); | ||
240 | + ret = blk_co_preadv(backend, offset, qiov.size, | ||
241 | + &qiov, 0); | ||
242 | + } | ||
243 | + if (ret >= 0) { | ||
244 | + req->in->status = VIRTIO_BLK_S_OK; | ||
245 | + } else { | ||
246 | + req->in->status = VIRTIO_BLK_S_IOERR; | ||
247 | + } | ||
248 | + break; | ||
249 | + } | ||
250 | + case VIRTIO_BLK_T_FLUSH: | ||
251 | + vu_block_flush(req); | ||
252 | + req->in->status = VIRTIO_BLK_S_OK; | ||
253 | + break; | ||
254 | + case VIRTIO_BLK_T_GET_ID: { | ||
255 | + size_t size = MIN(iov_size(&elem->in_sg[0], in_num), | ||
256 | + VIRTIO_BLK_ID_BYTES); | ||
257 | + snprintf(elem->in_sg[0].iov_base, size, "%s", "vhost_user_blk"); | ||
258 | + req->in->status = VIRTIO_BLK_S_OK; | ||
259 | + req->size = elem->in_sg[0].iov_len; | ||
260 | + break; | ||
261 | + } | ||
262 | + case VIRTIO_BLK_T_DISCARD: | ||
263 | + case VIRTIO_BLK_T_WRITE_ZEROES: { | ||
264 | + int rc; | ||
265 | + rc = vu_block_discard_write_zeroes(req, &elem->out_sg[1], | ||
266 | + out_num, type); | ||
267 | + if (rc == 0) { | ||
268 | + req->in->status = VIRTIO_BLK_S_OK; | ||
269 | + } else { | ||
270 | + req->in->status = VIRTIO_BLK_S_IOERR; | ||
271 | + } | ||
272 | + break; | ||
273 | + } | ||
274 | + default: | ||
275 | + req->in->status = VIRTIO_BLK_S_UNSUPP; | ||
276 | + break; | ||
277 | + } | ||
278 | + | ||
279 | + vu_block_req_complete(req); | ||
280 | + return; | ||
281 | + | ||
282 | +err: | ||
283 | + free(elem); | ||
284 | + g_free(req); | ||
285 | + return; | ||
286 | +} | ||
287 | + | ||
288 | +static void vu_block_process_vq(VuDev *vu_dev, int idx) | ||
289 | +{ | ||
290 | + VuServer *server; | ||
291 | + VuVirtq *vq; | ||
292 | + struct req_data *req_data; | ||
293 | + | ||
294 | + server = container_of(vu_dev, VuServer, vu_dev); | ||
295 | + assert(server); | ||
296 | + | ||
297 | + vq = vu_get_queue(vu_dev, idx); | ||
298 | + assert(vq); | ||
299 | + VuVirtqElement *elem; | ||
300 | + while (1) { | ||
301 | + elem = vu_queue_pop(vu_dev, vq, sizeof(VuVirtqElement) + | ||
302 | + sizeof(VuBlockReq)); | ||
303 | + if (elem) { | ||
304 | + req_data = g_new0(struct req_data, 1); | ||
305 | + req_data->server = server; | ||
306 | + req_data->vq = vq; | ||
307 | + req_data->elem = elem; | ||
308 | + Coroutine *co = qemu_coroutine_create(vu_block_virtio_process_req, | ||
309 | + req_data); | ||
310 | + aio_co_enter(server->ioc->ctx, co); | ||
311 | + } else { | ||
312 | + break; | ||
313 | + } | ||
314 | + } | ||
315 | +} | ||
316 | + | ||
317 | +static void vu_block_queue_set_started(VuDev *vu_dev, int idx, bool started) | ||
318 | +{ | ||
319 | + VuVirtq *vq; | ||
320 | + | ||
321 | + assert(vu_dev); | ||
322 | + | ||
323 | + vq = vu_get_queue(vu_dev, idx); | ||
324 | + vu_set_queue_handler(vu_dev, vq, started ? vu_block_process_vq : NULL); | ||
325 | +} | ||
326 | + | ||
327 | +static uint64_t vu_block_get_features(VuDev *dev) | ||
328 | +{ | ||
329 | + uint64_t features; | ||
330 | + VuServer *server = container_of(dev, VuServer, vu_dev); | ||
331 | + VuBlockDev *vdev_blk = get_vu_block_device_by_server(server); | ||
332 | + features = 1ull << VIRTIO_BLK_F_SIZE_MAX | | ||
333 | + 1ull << VIRTIO_BLK_F_SEG_MAX | | ||
334 | + 1ull << VIRTIO_BLK_F_TOPOLOGY | | ||
335 | + 1ull << VIRTIO_BLK_F_BLK_SIZE | | ||
336 | + 1ull << VIRTIO_BLK_F_FLUSH | | ||
337 | + 1ull << VIRTIO_BLK_F_DISCARD | | ||
338 | + 1ull << VIRTIO_BLK_F_WRITE_ZEROES | | ||
339 | + 1ull << VIRTIO_BLK_F_CONFIG_WCE | | ||
340 | + 1ull << VIRTIO_F_VERSION_1 | | ||
341 | + 1ull << VIRTIO_RING_F_INDIRECT_DESC | | ||
342 | + 1ull << VIRTIO_RING_F_EVENT_IDX | | ||
343 | + 1ull << VHOST_USER_F_PROTOCOL_FEATURES; | ||
344 | + | ||
345 | + if (!vdev_blk->writable) { | ||
346 | + features |= 1ull << VIRTIO_BLK_F_RO; | ||
347 | + } | ||
348 | + | ||
349 | + return features; | ||
350 | +} | ||
351 | + | ||
352 | +static uint64_t vu_block_get_protocol_features(VuDev *dev) | ||
353 | +{ | ||
354 | + return 1ull << VHOST_USER_PROTOCOL_F_CONFIG | | ||
355 | + 1ull << VHOST_USER_PROTOCOL_F_INFLIGHT_SHMFD; | ||
356 | +} | ||
357 | + | ||
358 | +static int | ||
359 | +vu_block_get_config(VuDev *vu_dev, uint8_t *config, uint32_t len) | ||
360 | +{ | ||
361 | + VuServer *server = container_of(vu_dev, VuServer, vu_dev); | ||
362 | + VuBlockDev *vdev_blk = get_vu_block_device_by_server(server); | ||
363 | + memcpy(config, &vdev_blk->blkcfg, len); | ||
364 | + | ||
365 | + return 0; | ||
366 | +} | ||
367 | + | ||
368 | +static int | ||
369 | +vu_block_set_config(VuDev *vu_dev, const uint8_t *data, | ||
370 | + uint32_t offset, uint32_t size, uint32_t flags) | ||
371 | +{ | ||
372 | + VuServer *server = container_of(vu_dev, VuServer, vu_dev); | ||
373 | + VuBlockDev *vdev_blk = get_vu_block_device_by_server(server); | ||
374 | + uint8_t wce; | ||
375 | + | ||
376 | + /* don't support live migration */ | ||
377 | + if (flags != VHOST_SET_CONFIG_TYPE_MASTER) { | ||
378 | + return -EINVAL; | ||
379 | + } | ||
380 | + | ||
381 | + if (offset != offsetof(struct virtio_blk_config, wce) || | ||
382 | + size != 1) { | ||
383 | + return -EINVAL; | ||
384 | + } | ||
385 | + | ||
386 | + wce = *data; | ||
387 | + vdev_blk->blkcfg.wce = wce; | ||
388 | + blk_set_enable_write_cache(vdev_blk->backend, wce); | ||
389 | + return 0; | ||
390 | +} | ||
391 | + | ||
392 | +/* | ||
393 | + * When the client disconnects, it sends a VHOST_USER_NONE request | ||
394 | + * and vu_process_message will simple call exit which cause the VM | ||
395 | + * to exit abruptly. | ||
396 | + * To avoid this issue, process VHOST_USER_NONE request ahead | ||
397 | + * of vu_process_message. | ||
398 | + * | ||
399 | + */ | ||
400 | +static int vu_block_process_msg(VuDev *dev, VhostUserMsg *vmsg, int *do_reply) | ||
401 | +{ | ||
402 | + if (vmsg->request == VHOST_USER_NONE) { | ||
403 | + dev->panic(dev, "disconnect"); | ||
404 | + return true; | ||
405 | + } | ||
406 | + return false; | ||
407 | +} | ||
408 | + | ||
409 | +static const VuDevIface vu_block_iface = { | ||
410 | + .get_features = vu_block_get_features, | ||
411 | + .queue_set_started = vu_block_queue_set_started, | ||
412 | + .get_protocol_features = vu_block_get_protocol_features, | ||
413 | + .get_config = vu_block_get_config, | ||
414 | + .set_config = vu_block_set_config, | ||
415 | + .process_msg = vu_block_process_msg, | ||
416 | +}; | ||
417 | + | ||
418 | +static void blk_aio_attached(AioContext *ctx, void *opaque) | ||
419 | +{ | ||
420 | + VuBlockDev *vub_dev = opaque; | ||
421 | + aio_context_acquire(ctx); | ||
422 | + vhost_user_server_set_aio_context(&vub_dev->vu_server, ctx); | ||
423 | + aio_context_release(ctx); | ||
424 | +} | ||
425 | + | ||
426 | +static void blk_aio_detach(void *opaque) | ||
427 | +{ | ||
428 | + VuBlockDev *vub_dev = opaque; | ||
429 | + AioContext *ctx = vub_dev->vu_server.ctx; | ||
430 | + aio_context_acquire(ctx); | ||
431 | + vhost_user_server_set_aio_context(&vub_dev->vu_server, NULL); | ||
432 | + aio_context_release(ctx); | ||
433 | +} | ||
434 | + | ||
435 | +static void | ||
436 | +vu_block_initialize_config(BlockDriverState *bs, | ||
437 | + struct virtio_blk_config *config, uint32_t blk_size) | ||
438 | +{ | ||
439 | + config->capacity = bdrv_getlength(bs) >> BDRV_SECTOR_BITS; | ||
440 | + config->blk_size = blk_size; | ||
441 | + config->size_max = 0; | ||
442 | + config->seg_max = 128 - 2; | ||
443 | + config->min_io_size = 1; | ||
444 | + config->opt_io_size = 1; | ||
445 | + config->num_queues = VHOST_USER_BLK_MAX_QUEUES; | ||
446 | + config->max_discard_sectors = 32768; | ||
447 | + config->max_discard_seg = 1; | ||
448 | + config->discard_sector_alignment = config->blk_size >> 9; | ||
449 | + config->max_write_zeroes_sectors = 32768; | ||
450 | + config->max_write_zeroes_seg = 1; | ||
451 | +} | ||
452 | + | ||
453 | +static VuBlockDev *vu_block_init(VuBlockDev *vu_block_device, Error **errp) | ||
454 | +{ | ||
455 | + | ||
456 | + BlockBackend *blk; | ||
457 | + Error *local_error = NULL; | ||
458 | + const char *node_name = vu_block_device->node_name; | ||
459 | + bool writable = vu_block_device->writable; | ||
460 | + uint64_t perm = BLK_PERM_CONSISTENT_READ; | ||
461 | + int ret; | ||
462 | + | ||
463 | + AioContext *ctx; | ||
464 | + | ||
465 | + BlockDriverState *bs = bdrv_lookup_bs(node_name, node_name, &local_error); | ||
466 | + | ||
467 | + if (!bs) { | ||
468 | + error_propagate(errp, local_error); | ||
469 | + return NULL; | ||
470 | + } | ||
471 | + | ||
472 | + if (bdrv_is_read_only(bs)) { | ||
473 | + writable = false; | ||
474 | + } | ||
475 | + | ||
476 | + if (writable) { | ||
477 | + perm |= BLK_PERM_WRITE; | ||
478 | + } | ||
479 | + | ||
480 | + ctx = bdrv_get_aio_context(bs); | ||
481 | + aio_context_acquire(ctx); | ||
482 | + bdrv_invalidate_cache(bs, NULL); | ||
483 | + aio_context_release(ctx); | ||
484 | + | ||
485 | + /* | ||
486 | + * Don't allow resize while the vhost user server is running, | ||
487 | + * otherwise we don't care what happens with the node. | ||
488 | + */ | ||
489 | + blk = blk_new(bdrv_get_aio_context(bs), perm, | ||
490 | + BLK_PERM_CONSISTENT_READ | BLK_PERM_WRITE_UNCHANGED | | ||
491 | + BLK_PERM_WRITE | BLK_PERM_GRAPH_MOD); | ||
492 | + ret = blk_insert_bs(blk, bs, errp); | ||
493 | + | ||
494 | + if (ret < 0) { | ||
495 | + goto fail; | ||
496 | + } | ||
497 | + | ||
498 | + blk_set_enable_write_cache(blk, false); | ||
499 | + | ||
500 | + blk_set_allow_aio_context_change(blk, true); | ||
501 | + | ||
502 | + vu_block_device->blkcfg.wce = 0; | ||
503 | + vu_block_device->backend = blk; | ||
504 | + if (!vu_block_device->blk_size) { | ||
505 | + vu_block_device->blk_size = BDRV_SECTOR_SIZE; | ||
506 | + } | ||
507 | + vu_block_device->blkcfg.blk_size = vu_block_device->blk_size; | ||
508 | + blk_set_guest_block_size(blk, vu_block_device->blk_size); | ||
509 | + vu_block_initialize_config(bs, &vu_block_device->blkcfg, | ||
510 | + vu_block_device->blk_size); | ||
511 | + return vu_block_device; | ||
512 | + | ||
513 | +fail: | ||
514 | + blk_unref(blk); | ||
515 | + return NULL; | ||
516 | +} | ||
517 | + | ||
518 | +static void vu_block_deinit(VuBlockDev *vu_block_device) | ||
519 | +{ | ||
520 | + if (vu_block_device->backend) { | ||
521 | + blk_remove_aio_context_notifier(vu_block_device->backend, blk_aio_attached, | ||
522 | + blk_aio_detach, vu_block_device); | ||
523 | + } | ||
524 | + | ||
525 | + blk_unref(vu_block_device->backend); | ||
526 | +} | ||
527 | + | ||
528 | +static void vhost_user_blk_server_stop(VuBlockDev *vu_block_device) | ||
529 | +{ | ||
530 | + vhost_user_server_stop(&vu_block_device->vu_server); | ||
531 | + vu_block_deinit(vu_block_device); | ||
532 | +} | ||
533 | + | ||
534 | +static void vhost_user_blk_server_start(VuBlockDev *vu_block_device, | ||
535 | + Error **errp) | ||
536 | +{ | ||
537 | + AioContext *ctx; | ||
538 | + SocketAddress *addr = vu_block_device->addr; | ||
539 | + | ||
540 | + if (!vu_block_init(vu_block_device, errp)) { | ||
541 | + return; | ||
542 | + } | ||
543 | + | ||
544 | + ctx = bdrv_get_aio_context(blk_bs(vu_block_device->backend)); | ||
545 | + | ||
546 | + if (!vhost_user_server_start(&vu_block_device->vu_server, addr, ctx, | ||
547 | + VHOST_USER_BLK_MAX_QUEUES, | ||
548 | + NULL, &vu_block_iface, | ||
549 | + errp)) { | ||
550 | + goto error; | ||
551 | + } | ||
552 | + | ||
553 | + blk_add_aio_context_notifier(vu_block_device->backend, blk_aio_attached, | ||
554 | + blk_aio_detach, vu_block_device); | ||
555 | + vu_block_device->running = true; | ||
556 | + return; | ||
557 | + | ||
558 | + error: | ||
559 | + vu_block_deinit(vu_block_device); | ||
560 | +} | ||
561 | + | ||
562 | +static bool vu_prop_modifiable(VuBlockDev *vus, Error **errp) | ||
563 | +{ | ||
564 | + if (vus->running) { | ||
565 | + error_setg(errp, "The property can't be modified " | ||
566 | + "while the server is running"); | ||
567 | + return false; | ||
568 | + } | ||
569 | + return true; | ||
570 | +} | ||
571 | + | ||
572 | +static void vu_set_node_name(Object *obj, const char *value, Error **errp) | ||
573 | +{ | ||
574 | + VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
575 | + | ||
576 | + if (!vu_prop_modifiable(vus, errp)) { | ||
577 | + return; | ||
578 | + } | ||
579 | + | ||
580 | + if (vus->node_name) { | ||
581 | + g_free(vus->node_name); | ||
582 | + } | ||
583 | + | ||
584 | + vus->node_name = g_strdup(value); | ||
585 | +} | ||
586 | + | ||
587 | +static char *vu_get_node_name(Object *obj, Error **errp) | ||
588 | +{ | ||
589 | + VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
590 | + return g_strdup(vus->node_name); | ||
591 | +} | ||
592 | + | ||
593 | +static void free_socket_addr(SocketAddress *addr) | ||
594 | +{ | ||
595 | + g_free(addr->u.q_unix.path); | ||
596 | + g_free(addr); | ||
597 | +} | ||
598 | + | ||
599 | +static void vu_set_unix_socket(Object *obj, const char *value, | ||
600 | + Error **errp) | ||
601 | +{ | ||
602 | + VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
603 | + | ||
604 | + if (!vu_prop_modifiable(vus, errp)) { | ||
605 | + return; | ||
606 | + } | ||
607 | + | ||
608 | + if (vus->addr) { | ||
609 | + free_socket_addr(vus->addr); | ||
610 | + } | ||
611 | + | ||
612 | + SocketAddress *addr = g_new0(SocketAddress, 1); | ||
613 | + addr->type = SOCKET_ADDRESS_TYPE_UNIX; | ||
614 | + addr->u.q_unix.path = g_strdup(value); | ||
615 | + vus->addr = addr; | ||
616 | +} | ||
617 | + | ||
618 | +static char *vu_get_unix_socket(Object *obj, Error **errp) | ||
619 | +{ | ||
620 | + VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
621 | + return g_strdup(vus->addr->u.q_unix.path); | ||
622 | +} | ||
623 | + | ||
624 | +static bool vu_get_block_writable(Object *obj, Error **errp) | ||
625 | +{ | ||
626 | + VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
627 | + return vus->writable; | ||
628 | +} | ||
629 | + | ||
630 | +static void vu_set_block_writable(Object *obj, bool value, Error **errp) | ||
631 | +{ | ||
632 | + VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
633 | + | ||
634 | + if (!vu_prop_modifiable(vus, errp)) { | ||
635 | + return; | ||
636 | + } | ||
637 | + | ||
638 | + vus->writable = value; | ||
639 | +} | ||
640 | + | ||
641 | +static void vu_get_blk_size(Object *obj, Visitor *v, const char *name, | ||
642 | + void *opaque, Error **errp) | ||
643 | +{ | ||
644 | + VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
645 | + uint32_t value = vus->blk_size; | ||
646 | + | ||
647 | + visit_type_uint32(v, name, &value, errp); | ||
648 | +} | ||
649 | + | ||
650 | +static void vu_set_blk_size(Object *obj, Visitor *v, const char *name, | ||
651 | + void *opaque, Error **errp) | ||
652 | +{ | ||
653 | + VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
654 | + | ||
655 | + Error *local_err = NULL; | ||
656 | + uint32_t value; | ||
657 | + | ||
658 | + if (!vu_prop_modifiable(vus, errp)) { | ||
659 | + return; | ||
660 | + } | ||
661 | + | ||
662 | + visit_type_uint32(v, name, &value, &local_err); | ||
663 | + if (local_err) { | ||
664 | + goto out; | ||
665 | + } | ||
666 | + | ||
667 | + check_block_size(object_get_typename(obj), name, value, &local_err); | ||
668 | + if (local_err) { | ||
669 | + goto out; | ||
670 | + } | ||
671 | + | ||
672 | + vus->blk_size = value; | ||
673 | + | ||
674 | +out: | ||
675 | + error_propagate(errp, local_err); | ||
676 | +} | ||
677 | + | ||
678 | +static void vhost_user_blk_server_instance_finalize(Object *obj) | ||
679 | +{ | ||
680 | + VuBlockDev *vub = VHOST_USER_BLK_SERVER(obj); | ||
681 | + | ||
682 | + vhost_user_blk_server_stop(vub); | ||
683 | + | ||
684 | + /* | ||
685 | + * Unlike object_property_add_str, object_class_property_add_str | ||
686 | + * doesn't have a release method. Thus manual memory freeing is | ||
687 | + * needed. | ||
688 | + */ | ||
689 | + free_socket_addr(vub->addr); | ||
690 | + g_free(vub->node_name); | ||
691 | +} | ||
692 | + | ||
693 | +static void vhost_user_blk_server_complete(UserCreatable *obj, Error **errp) | ||
694 | +{ | ||
695 | + VuBlockDev *vub = VHOST_USER_BLK_SERVER(obj); | ||
696 | + | ||
697 | + vhost_user_blk_server_start(vub, errp); | ||
698 | +} | ||
699 | + | ||
700 | +static void vhost_user_blk_server_class_init(ObjectClass *klass, | ||
701 | + void *class_data) | ||
702 | +{ | ||
703 | + UserCreatableClass *ucc = USER_CREATABLE_CLASS(klass); | ||
704 | + ucc->complete = vhost_user_blk_server_complete; | ||
705 | + | ||
706 | + object_class_property_add_bool(klass, "writable", | ||
707 | + vu_get_block_writable, | ||
708 | + vu_set_block_writable); | ||
709 | + | ||
710 | + object_class_property_add_str(klass, "node-name", | ||
711 | + vu_get_node_name, | ||
712 | + vu_set_node_name); | ||
713 | + | ||
714 | + object_class_property_add_str(klass, "unix-socket", | ||
715 | + vu_get_unix_socket, | ||
716 | + vu_set_unix_socket); | ||
717 | + | ||
718 | + object_class_property_add(klass, "logical-block-size", "uint32", | ||
719 | + vu_get_blk_size, vu_set_blk_size, | ||
720 | + NULL, NULL); | ||
721 | +} | ||
722 | + | ||
723 | +static const TypeInfo vhost_user_blk_server_info = { | ||
724 | + .name = TYPE_VHOST_USER_BLK_SERVER, | ||
725 | + .parent = TYPE_OBJECT, | ||
726 | + .instance_size = sizeof(VuBlockDev), | ||
727 | + .instance_finalize = vhost_user_blk_server_instance_finalize, | ||
728 | + .class_init = vhost_user_blk_server_class_init, | ||
729 | + .interfaces = (InterfaceInfo[]) { | ||
730 | + {TYPE_USER_CREATABLE}, | ||
731 | + {} | ||
732 | + }, | ||
733 | +}; | ||
734 | + | ||
735 | +static void vhost_user_blk_server_register_types(void) | ||
736 | +{ | ||
737 | + type_register_static(&vhost_user_blk_server_info); | ||
738 | +} | ||
739 | + | ||
740 | +type_init(vhost_user_blk_server_register_types) | ||
741 | diff --git a/softmmu/vl.c b/softmmu/vl.c | ||
742 | index XXXXXXX..XXXXXXX 100644 | ||
743 | --- a/softmmu/vl.c | ||
744 | +++ b/softmmu/vl.c | ||
745 | @@ -XXX,XX +XXX,XX @@ static bool object_create_initial(const char *type, QemuOpts *opts) | ||
746 | } | ||
747 | #endif | ||
748 | |||
749 | + /* Reason: vhost-user-blk-server property "node-name" */ | ||
750 | + if (g_str_equal(type, "vhost-user-blk-server")) { | ||
751 | + return false; | ||
752 | + } | ||
753 | /* | ||
754 | * Reason: filter-* property "netdev" etc. | ||
755 | */ | ||
756 | diff --git a/block/meson.build b/block/meson.build | ||
757 | index XXXXXXX..XXXXXXX 100644 | ||
758 | --- a/block/meson.build | ||
759 | +++ b/block/meson.build | ||
760 | @@ -XXX,XX +XXX,XX @@ block_ss.add(when: 'CONFIG_WIN32', if_true: files('file-win32.c', 'win32-aio.c') | ||
761 | block_ss.add(when: 'CONFIG_POSIX', if_true: [files('file-posix.c'), coref, iokit]) | ||
762 | block_ss.add(when: 'CONFIG_LIBISCSI', if_true: files('iscsi-opts.c')) | ||
763 | block_ss.add(when: 'CONFIG_LINUX', if_true: files('nvme.c')) | ||
764 | +block_ss.add(when: 'CONFIG_LINUX', if_true: files('export/vhost-user-blk-server.c', '../contrib/libvhost-user/libvhost-user.c')) | ||
765 | block_ss.add(when: 'CONFIG_REPLICATION', if_true: files('replication.c')) | ||
766 | block_ss.add(when: 'CONFIG_SHEEPDOG', if_true: files('sheepdog.c')) | ||
767 | block_ss.add(when: ['CONFIG_LINUX_AIO', libaio], if_true: files('linux-aio.c')) | ||
768 | -- | ||
769 | 2.26.2 | ||
770 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Coiby Xu <coiby.xu@gmail.com> | ||
1 | 2 | ||
3 | Suggested-by: Stefano Garzarella <sgarzare@redhat.com> | ||
4 | Signed-off-by: Coiby Xu <coiby.xu@gmail.com> | ||
5 | Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
6 | Reviewed-by: Marc-André Lureau <marcandre.lureau@redhat.com> | ||
7 | Message-id: 20200918080912.321299-8-coiby.xu@gmail.com | ||
8 | [Removed reference to vhost-user-blk-test.c, it will be sent in a | ||
9 | separate pull request. | ||
10 | --Stefan] | ||
11 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
12 | --- | ||
13 | MAINTAINERS | 7 +++++++ | ||
14 | 1 file changed, 7 insertions(+) | ||
15 | |||
16 | diff --git a/MAINTAINERS b/MAINTAINERS | ||
17 | index XXXXXXX..XXXXXXX 100644 | ||
18 | --- a/MAINTAINERS | ||
19 | +++ b/MAINTAINERS | ||
20 | @@ -XXX,XX +XXX,XX @@ L: qemu-block@nongnu.org | ||
21 | S: Supported | ||
22 | F: tests/image-fuzzer/ | ||
23 | |||
24 | +Vhost-user block device backend server | ||
25 | +M: Coiby Xu <Coiby.Xu@gmail.com> | ||
26 | +S: Maintained | ||
27 | +F: block/export/vhost-user-blk-server.c | ||
28 | +F: util/vhost-user-server.c | ||
29 | +F: tests/qtest/libqos/vhost-user-blk.c | ||
30 | + | ||
31 | Replication | ||
32 | M: Wen Congyang <wencongyang2@huawei.com> | ||
33 | M: Xie Changlong <xiechanglong.d@gmail.com> | ||
34 | -- | ||
35 | 2.26.2 | ||
36 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
2 | Message-id: 20200924151549.913737-3-stefanha@redhat.com | ||
3 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
4 | --- | ||
5 | util/vhost-user-server.c | 2 +- | ||
6 | 1 file changed, 1 insertion(+), 1 deletion(-) | ||
1 | 7 | ||
8 | diff --git a/util/vhost-user-server.c b/util/vhost-user-server.c | ||
9 | index XXXXXXX..XXXXXXX 100644 | ||
10 | --- a/util/vhost-user-server.c | ||
11 | +++ b/util/vhost-user-server.c | ||
12 | @@ -XXX,XX +XXX,XX @@ bool vhost_user_server_start(VuServer *server, | ||
13 | return false; | ||
14 | } | ||
15 | |||
16 | - /* zero out unspecified fileds */ | ||
17 | + /* zero out unspecified fields */ | ||
18 | *server = (VuServer) { | ||
19 | .listener = listener, | ||
20 | .vu_iface = vu_iface, | ||
21 | -- | ||
22 | 2.26.2 | ||
23 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | We already have access to the value with the correct type (ioc and sioc | ||
2 | are the same QIOChannel). | ||
1 | 3 | ||
4 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
5 | Message-id: 20200924151549.913737-4-stefanha@redhat.com | ||
6 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
7 | --- | ||
8 | util/vhost-user-server.c | 2 +- | ||
9 | 1 file changed, 1 insertion(+), 1 deletion(-) | ||
10 | |||
11 | diff --git a/util/vhost-user-server.c b/util/vhost-user-server.c | ||
12 | index XXXXXXX..XXXXXXX 100644 | ||
13 | --- a/util/vhost-user-server.c | ||
14 | +++ b/util/vhost-user-server.c | ||
15 | @@ -XXX,XX +XXX,XX @@ static void vu_accept(QIONetListener *listener, QIOChannelSocket *sioc, | ||
16 | server->ioc = QIO_CHANNEL(sioc); | ||
17 | object_ref(OBJECT(server->ioc)); | ||
18 | qio_channel_attach_aio_context(server->ioc, server->ctx); | ||
19 | - qio_channel_set_blocking(QIO_CHANNEL(server->sioc), false, NULL); | ||
20 | + qio_channel_set_blocking(server->ioc, false, NULL); | ||
21 | vu_client_start(server); | ||
22 | } | ||
23 | |||
24 | -- | ||
25 | 2.26.2 | ||
26 | diff view generated by jsdifflib |
1 | From: Thomas Huth <thuth@redhat.com> | 1 | Explicitly deleting watches is not necessary since libvhost-user calls |
---|---|---|---|
2 | remove_watch() during vu_deinit(). Add an assertion to check this | ||
3 | though. | ||
2 | 4 | ||
3 | dev could be NULL if the PCI device can not be found due to some | 5 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
4 | reasons, so we must not dereference the pointer in this case. | 6 | Message-id: 20200924151549.913737-5-stefanha@redhat.com |
5 | |||
6 | Signed-off-by: Thomas Huth <thuth@redhat.com> | ||
7 | Message-id: 1519713884-2346-1-git-send-email-thuth@redhat.com | ||
8 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 7 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
9 | --- | 8 | --- |
10 | tests/libqos/virtio-pci.c | 4 +++- | 9 | util/vhost-user-server.c | 19 ++++--------------- |
11 | 1 file changed, 3 insertions(+), 1 deletion(-) | 10 | 1 file changed, 4 insertions(+), 15 deletions(-) |
12 | 11 | ||
13 | diff --git a/tests/libqos/virtio-pci.c b/tests/libqos/virtio-pci.c | 12 | diff --git a/util/vhost-user-server.c b/util/vhost-user-server.c |
14 | index XXXXXXX..XXXXXXX 100644 | 13 | index XXXXXXX..XXXXXXX 100644 |
15 | --- a/tests/libqos/virtio-pci.c | 14 | --- a/util/vhost-user-server.c |
16 | +++ b/tests/libqos/virtio-pci.c | 15 | +++ b/util/vhost-user-server.c |
17 | @@ -XXX,XX +XXX,XX @@ QVirtioPCIDevice *qvirtio_pci_device_find(QPCIBus *bus, uint16_t device_type) | 16 | @@ -XXX,XX +XXX,XX @@ static void close_client(VuServer *server) |
18 | qvirtio_pci_foreach(bus, device_type, false, 0, | 17 | /* When this is set vu_client_trip will stop new processing vhost-user message */ |
19 | qvirtio_pci_assign_device, &dev); | 18 | server->sioc = NULL; |
20 | 19 | ||
21 | - dev->vdev.bus = &qvirtio_pci; | 20 | - VuFdWatch *vu_fd_watch, *next; |
22 | + if (dev) { | 21 | - QTAILQ_FOREACH_SAFE(vu_fd_watch, &server->vu_fd_watches, next, next) { |
23 | + dev->vdev.bus = &qvirtio_pci; | 22 | - aio_set_fd_handler(server->ioc->ctx, vu_fd_watch->fd, true, NULL, |
24 | + } | 23 | - NULL, NULL, NULL); |
25 | 24 | - } | |
26 | return dev; | 25 | - |
26 | - while (!QTAILQ_EMPTY(&server->vu_fd_watches)) { | ||
27 | - QTAILQ_FOREACH_SAFE(vu_fd_watch, &server->vu_fd_watches, next, next) { | ||
28 | - if (!vu_fd_watch->processing) { | ||
29 | - QTAILQ_REMOVE(&server->vu_fd_watches, vu_fd_watch, next); | ||
30 | - g_free(vu_fd_watch); | ||
31 | - } | ||
32 | - } | ||
33 | - } | ||
34 | - | ||
35 | while (server->processing_msg) { | ||
36 | if (server->ioc->read_coroutine) { | ||
37 | server->ioc->read_coroutine = NULL; | ||
38 | @@ -XXX,XX +XXX,XX @@ static void close_client(VuServer *server) | ||
39 | } | ||
40 | |||
41 | vu_deinit(&server->vu_dev); | ||
42 | + | ||
43 | + /* vu_deinit() should have called remove_watch() */ | ||
44 | + assert(QTAILQ_EMPTY(&server->vu_fd_watches)); | ||
45 | + | ||
46 | object_unref(OBJECT(sioc)); | ||
47 | object_unref(OBJECT(server->ioc)); | ||
27 | } | 48 | } |
28 | -- | 49 | -- |
29 | 2.14.3 | 50 | 2.26.2 |
30 | 51 | ||
31 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | Only one struct is needed per request. Drop req_data and the separate | ||
2 | VuBlockReq instance. Instead let vu_queue_pop() allocate everything at | ||
3 | once. | ||
1 | 4 | ||
5 | This fixes the req_data memory leak in vu_block_virtio_process_req(). | ||
6 | |||
7 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
8 | Message-id: 20200924151549.913737-6-stefanha@redhat.com | ||
9 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
10 | --- | ||
11 | block/export/vhost-user-blk-server.c | 68 +++++++++------------------- | ||
12 | 1 file changed, 21 insertions(+), 47 deletions(-) | ||
13 | |||
14 | diff --git a/block/export/vhost-user-blk-server.c b/block/export/vhost-user-blk-server.c | ||
15 | index XXXXXXX..XXXXXXX 100644 | ||
16 | --- a/block/export/vhost-user-blk-server.c | ||
17 | +++ b/block/export/vhost-user-blk-server.c | ||
18 | @@ -XXX,XX +XXX,XX @@ struct virtio_blk_inhdr { | ||
19 | }; | ||
20 | |||
21 | typedef struct VuBlockReq { | ||
22 | - VuVirtqElement *elem; | ||
23 | + VuVirtqElement elem; | ||
24 | int64_t sector_num; | ||
25 | size_t size; | ||
26 | struct virtio_blk_inhdr *in; | ||
27 | @@ -XXX,XX +XXX,XX @@ static void vu_block_req_complete(VuBlockReq *req) | ||
28 | VuDev *vu_dev = &req->server->vu_dev; | ||
29 | |||
30 | /* IO size with 1 extra status byte */ | ||
31 | - vu_queue_push(vu_dev, req->vq, req->elem, req->size + 1); | ||
32 | + vu_queue_push(vu_dev, req->vq, &req->elem, req->size + 1); | ||
33 | vu_queue_notify(vu_dev, req->vq); | ||
34 | |||
35 | - if (req->elem) { | ||
36 | - free(req->elem); | ||
37 | - } | ||
38 | - | ||
39 | - g_free(req); | ||
40 | + free(req); | ||
41 | } | ||
42 | |||
43 | static VuBlockDev *get_vu_block_device_by_server(VuServer *server) | ||
44 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn vu_block_flush(VuBlockReq *req) | ||
45 | blk_co_flush(backend); | ||
46 | } | ||
47 | |||
48 | -struct req_data { | ||
49 | - VuServer *server; | ||
50 | - VuVirtq *vq; | ||
51 | - VuVirtqElement *elem; | ||
52 | -}; | ||
53 | - | ||
54 | static void coroutine_fn vu_block_virtio_process_req(void *opaque) | ||
55 | { | ||
56 | - struct req_data *data = opaque; | ||
57 | - VuServer *server = data->server; | ||
58 | - VuVirtq *vq = data->vq; | ||
59 | - VuVirtqElement *elem = data->elem; | ||
60 | + VuBlockReq *req = opaque; | ||
61 | + VuServer *server = req->server; | ||
62 | + VuVirtqElement *elem = &req->elem; | ||
63 | uint32_t type; | ||
64 | - VuBlockReq *req; | ||
65 | |||
66 | VuBlockDev *vdev_blk = get_vu_block_device_by_server(server); | ||
67 | BlockBackend *backend = vdev_blk->backend; | ||
68 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn vu_block_virtio_process_req(void *opaque) | ||
69 | struct iovec *out_iov = elem->out_sg; | ||
70 | unsigned in_num = elem->in_num; | ||
71 | unsigned out_num = elem->out_num; | ||
72 | + | ||
73 | /* refer to hw/block/virtio_blk.c */ | ||
74 | if (elem->out_num < 1 || elem->in_num < 1) { | ||
75 | error_report("virtio-blk request missing headers"); | ||
76 | - free(elem); | ||
77 | - return; | ||
78 | + goto err; | ||
79 | } | ||
80 | |||
81 | - req = g_new0(VuBlockReq, 1); | ||
82 | - req->server = server; | ||
83 | - req->vq = vq; | ||
84 | - req->elem = elem; | ||
85 | - | ||
86 | if (unlikely(iov_to_buf(out_iov, out_num, 0, &req->out, | ||
87 | sizeof(req->out)) != sizeof(req->out))) { | ||
88 | error_report("virtio-blk request outhdr too short"); | ||
89 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn vu_block_virtio_process_req(void *opaque) | ||
90 | |||
91 | err: | ||
92 | free(elem); | ||
93 | - g_free(req); | ||
94 | - return; | ||
95 | } | ||
96 | |||
97 | static void vu_block_process_vq(VuDev *vu_dev, int idx) | ||
98 | { | ||
99 | - VuServer *server; | ||
100 | - VuVirtq *vq; | ||
101 | - struct req_data *req_data; | ||
102 | + VuServer *server = container_of(vu_dev, VuServer, vu_dev); | ||
103 | + VuVirtq *vq = vu_get_queue(vu_dev, idx); | ||
104 | |||
105 | - server = container_of(vu_dev, VuServer, vu_dev); | ||
106 | - assert(server); | ||
107 | - | ||
108 | - vq = vu_get_queue(vu_dev, idx); | ||
109 | - assert(vq); | ||
110 | - VuVirtqElement *elem; | ||
111 | while (1) { | ||
112 | - elem = vu_queue_pop(vu_dev, vq, sizeof(VuVirtqElement) + | ||
113 | - sizeof(VuBlockReq)); | ||
114 | - if (elem) { | ||
115 | - req_data = g_new0(struct req_data, 1); | ||
116 | - req_data->server = server; | ||
117 | - req_data->vq = vq; | ||
118 | - req_data->elem = elem; | ||
119 | - Coroutine *co = qemu_coroutine_create(vu_block_virtio_process_req, | ||
120 | - req_data); | ||
121 | - aio_co_enter(server->ioc->ctx, co); | ||
122 | - } else { | ||
123 | + VuBlockReq *req; | ||
124 | + | ||
125 | + req = vu_queue_pop(vu_dev, vq, sizeof(VuBlockReq)); | ||
126 | + if (!req) { | ||
127 | break; | ||
128 | } | ||
129 | + | ||
130 | + req->server = server; | ||
131 | + req->vq = vq; | ||
132 | + | ||
133 | + Coroutine *co = | ||
134 | + qemu_coroutine_create(vu_block_virtio_process_req, req); | ||
135 | + qemu_coroutine_enter(co); | ||
136 | } | ||
137 | } | ||
138 | |||
139 | -- | ||
140 | 2.26.2 | ||
141 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | The device panic notifier callback is not used. Drop it. | ||
1 | 2 | ||
3 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
4 | Message-id: 20200924151549.913737-7-stefanha@redhat.com | ||
5 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
6 | --- | ||
7 | util/vhost-user-server.h | 3 --- | ||
8 | block/export/vhost-user-blk-server.c | 3 +-- | ||
9 | util/vhost-user-server.c | 6 ------ | ||
10 | 3 files changed, 1 insertion(+), 11 deletions(-) | ||
11 | |||
12 | diff --git a/util/vhost-user-server.h b/util/vhost-user-server.h | ||
13 | index XXXXXXX..XXXXXXX 100644 | ||
14 | --- a/util/vhost-user-server.h | ||
15 | +++ b/util/vhost-user-server.h | ||
16 | @@ -XXX,XX +XXX,XX @@ typedef struct VuFdWatch { | ||
17 | } VuFdWatch; | ||
18 | |||
19 | typedef struct VuServer VuServer; | ||
20 | -typedef void DevicePanicNotifierFn(VuServer *server); | ||
21 | |||
22 | struct VuServer { | ||
23 | QIONetListener *listener; | ||
24 | AioContext *ctx; | ||
25 | - DevicePanicNotifierFn *device_panic_notifier; | ||
26 | int max_queues; | ||
27 | const VuDevIface *vu_iface; | ||
28 | VuDev vu_dev; | ||
29 | @@ -XXX,XX +XXX,XX @@ bool vhost_user_server_start(VuServer *server, | ||
30 | SocketAddress *unix_socket, | ||
31 | AioContext *ctx, | ||
32 | uint16_t max_queues, | ||
33 | - DevicePanicNotifierFn *device_panic_notifier, | ||
34 | const VuDevIface *vu_iface, | ||
35 | Error **errp); | ||
36 | |||
37 | diff --git a/block/export/vhost-user-blk-server.c b/block/export/vhost-user-blk-server.c | ||
38 | index XXXXXXX..XXXXXXX 100644 | ||
39 | --- a/block/export/vhost-user-blk-server.c | ||
40 | +++ b/block/export/vhost-user-blk-server.c | ||
41 | @@ -XXX,XX +XXX,XX @@ static void vhost_user_blk_server_start(VuBlockDev *vu_block_device, | ||
42 | ctx = bdrv_get_aio_context(blk_bs(vu_block_device->backend)); | ||
43 | |||
44 | if (!vhost_user_server_start(&vu_block_device->vu_server, addr, ctx, | ||
45 | - VHOST_USER_BLK_MAX_QUEUES, | ||
46 | - NULL, &vu_block_iface, | ||
47 | + VHOST_USER_BLK_MAX_QUEUES, &vu_block_iface, | ||
48 | errp)) { | ||
49 | goto error; | ||
50 | } | ||
51 | diff --git a/util/vhost-user-server.c b/util/vhost-user-server.c | ||
52 | index XXXXXXX..XXXXXXX 100644 | ||
53 | --- a/util/vhost-user-server.c | ||
54 | +++ b/util/vhost-user-server.c | ||
55 | @@ -XXX,XX +XXX,XX @@ static void panic_cb(VuDev *vu_dev, const char *buf) | ||
56 | close_client(server); | ||
57 | } | ||
58 | |||
59 | - if (server->device_panic_notifier) { | ||
60 | - server->device_panic_notifier(server); | ||
61 | - } | ||
62 | - | ||
63 | /* | ||
64 | * Set the callback function for network listener so another | ||
65 | * vhost-user client can connect to this server | ||
66 | @@ -XXX,XX +XXX,XX @@ bool vhost_user_server_start(VuServer *server, | ||
67 | SocketAddress *socket_addr, | ||
68 | AioContext *ctx, | ||
69 | uint16_t max_queues, | ||
70 | - DevicePanicNotifierFn *device_panic_notifier, | ||
71 | const VuDevIface *vu_iface, | ||
72 | Error **errp) | ||
73 | { | ||
74 | @@ -XXX,XX +XXX,XX @@ bool vhost_user_server_start(VuServer *server, | ||
75 | .vu_iface = vu_iface, | ||
76 | .max_queues = max_queues, | ||
77 | .ctx = ctx, | ||
78 | - .device_panic_notifier = device_panic_notifier, | ||
79 | }; | ||
80 | |||
81 | qio_net_listener_set_name(server->listener, "vhost-user-backend-listener"); | ||
82 | -- | ||
83 | 2.26.2 | ||
84 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | fds[] is leaked when qio_channel_readv_full() fails. | ||
1 | 2 | ||
3 | Use vmsg->fds[] instead of keeping a local fds[] array. Then we can | ||
4 | reuse goto fail to clean up fds. vmsg->fd_num must be zeroed before the | ||
5 | loop to make this safe. | ||
6 | |||
7 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
8 | Message-id: 20200924151549.913737-8-stefanha@redhat.com | ||
9 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
10 | --- | ||
11 | util/vhost-user-server.c | 50 ++++++++++++++++++---------------------- | ||
12 | 1 file changed, 23 insertions(+), 27 deletions(-) | ||
13 | |||
14 | diff --git a/util/vhost-user-server.c b/util/vhost-user-server.c | ||
15 | index XXXXXXX..XXXXXXX 100644 | ||
16 | --- a/util/vhost-user-server.c | ||
17 | +++ b/util/vhost-user-server.c | ||
18 | @@ -XXX,XX +XXX,XX @@ vu_message_read(VuDev *vu_dev, int conn_fd, VhostUserMsg *vmsg) | ||
19 | }; | ||
20 | int rc, read_bytes = 0; | ||
21 | Error *local_err = NULL; | ||
22 | - /* | ||
23 | - * Store fds/nfds returned from qio_channel_readv_full into | ||
24 | - * temporary variables. | ||
25 | - * | ||
26 | - * VhostUserMsg is a packed structure, gcc will complain about passing | ||
27 | - * pointer to a packed structure member if we pass &VhostUserMsg.fd_num | ||
28 | - * and &VhostUserMsg.fds directly when calling qio_channel_readv_full, | ||
29 | - * thus two temporary variables nfds and fds are used here. | ||
30 | - */ | ||
31 | - size_t nfds = 0, nfds_t = 0; | ||
32 | const size_t max_fds = G_N_ELEMENTS(vmsg->fds); | ||
33 | - int *fds_t = NULL; | ||
34 | VuServer *server = container_of(vu_dev, VuServer, vu_dev); | ||
35 | QIOChannel *ioc = server->ioc; | ||
36 | |||
37 | + vmsg->fd_num = 0; | ||
38 | if (!ioc) { | ||
39 | error_report_err(local_err); | ||
40 | goto fail; | ||
41 | @@ -XXX,XX +XXX,XX @@ vu_message_read(VuDev *vu_dev, int conn_fd, VhostUserMsg *vmsg) | ||
42 | |||
43 | assert(qemu_in_coroutine()); | ||
44 | do { | ||
45 | + size_t nfds = 0; | ||
46 | + int *fds = NULL; | ||
47 | + | ||
48 | /* | ||
49 | * qio_channel_readv_full may have short reads, keeping calling it | ||
50 | * until getting VHOST_USER_HDR_SIZE or 0 bytes in total | ||
51 | */ | ||
52 | - rc = qio_channel_readv_full(ioc, &iov, 1, &fds_t, &nfds_t, &local_err); | ||
53 | + rc = qio_channel_readv_full(ioc, &iov, 1, &fds, &nfds, &local_err); | ||
54 | if (rc < 0) { | ||
55 | if (rc == QIO_CHANNEL_ERR_BLOCK) { | ||
56 | + assert(local_err == NULL); | ||
57 | qio_channel_yield(ioc, G_IO_IN); | ||
58 | continue; | ||
59 | } else { | ||
60 | error_report_err(local_err); | ||
61 | - return false; | ||
62 | + goto fail; | ||
63 | } | ||
64 | } | ||
65 | - read_bytes += rc; | ||
66 | - if (nfds_t > 0) { | ||
67 | - if (nfds + nfds_t > max_fds) { | ||
68 | + | ||
69 | + if (nfds > 0) { | ||
70 | + if (vmsg->fd_num + nfds > max_fds) { | ||
71 | error_report("A maximum of %zu fds are allowed, " | ||
72 | "however got %zu fds now", | ||
73 | - max_fds, nfds + nfds_t); | ||
74 | + max_fds, vmsg->fd_num + nfds); | ||
75 | + g_free(fds); | ||
76 | goto fail; | ||
77 | } | ||
78 | - memcpy(vmsg->fds + nfds, fds_t, | ||
79 | - nfds_t *sizeof(vmsg->fds[0])); | ||
80 | - nfds += nfds_t; | ||
81 | - g_free(fds_t); | ||
82 | + memcpy(vmsg->fds + vmsg->fd_num, fds, nfds * sizeof(vmsg->fds[0])); | ||
83 | + vmsg->fd_num += nfds; | ||
84 | + g_free(fds); | ||
85 | } | ||
86 | - if (read_bytes == VHOST_USER_HDR_SIZE || rc == 0) { | ||
87 | - break; | ||
88 | + | ||
89 | + if (rc == 0) { /* socket closed */ | ||
90 | + goto fail; | ||
91 | } | ||
92 | - iov.iov_base = (char *)vmsg + read_bytes; | ||
93 | - iov.iov_len = VHOST_USER_HDR_SIZE - read_bytes; | ||
94 | - } while (true); | ||
95 | |||
96 | - vmsg->fd_num = nfds; | ||
97 | + iov.iov_base += rc; | ||
98 | + iov.iov_len -= rc; | ||
99 | + read_bytes += rc; | ||
100 | + } while (read_bytes != VHOST_USER_HDR_SIZE); | ||
101 | + | ||
102 | /* qio_channel_readv_full will make socket fds blocking, unblock them */ | ||
103 | vmsg_unblock_fds(vmsg); | ||
104 | if (vmsg->size > sizeof(vmsg->payload)) { | ||
105 | -- | ||
106 | 2.26.2 | ||
107 | diff view generated by jsdifflib |
1 | From: Su Hang <suhang16@mails.ucas.ac.cn> | 1 | Unexpected EOF is an error that must be reported. |
---|---|---|---|
2 | |||
3 | For this patch, using curly braces to wrap `if` `while` `else` statements, | ||
4 | which only hold single statement. For example: | ||
5 | ''' | ||
6 | if (cond) | ||
7 | statement; | ||
8 | ''' | ||
9 | to | ||
10 | ''' | ||
11 | if (cond) { | ||
12 | statement; | ||
13 | } | ||
14 | ''' | ||
15 | |||
16 | And using tricks that compare the disassemblies before and after | ||
17 | code changes, to make sure code logic isn't changed: | ||
18 | ''' | ||
19 | git checkout master | ||
20 | make util/uri.o | ||
21 | strip util/uri.o | ||
22 | objdump -Drx util/uri.o > /tmp/uri-master.txt | ||
23 | git checkout cleanupbranch | ||
24 | make util/uri.o | ||
25 | strip util/uri.o | ||
26 | objdump -Drx util/uri.o > /tmp/uri-cleanup.txt | ||
27 | 2 | ||
28 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 3 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
4 | Message-id: 20200924151549.913737-9-stefanha@redhat.com | ||
5 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
29 | --- | 6 | --- |
30 | util/uri.c | 463 +++++++++++++++++++++++++++++++++++++++---------------------- | 7 | util/vhost-user-server.c | 6 ++++-- |
31 | 1 file changed, 294 insertions(+), 169 deletions(-) | 8 | 1 file changed, 4 insertions(+), 2 deletions(-) |
32 | 9 | ||
33 | diff --git a/util/uri.c b/util/uri.c | 10 | diff --git a/util/vhost-user-server.c b/util/vhost-user-server.c |
34 | index XXXXXXX..XXXXXXX 100644 | 11 | index XXXXXXX..XXXXXXX 100644 |
35 | --- a/util/uri.c | 12 | --- a/util/vhost-user-server.c |
36 | +++ b/util/uri.c | 13 | +++ b/util/vhost-user-server.c |
37 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_scheme(URI *uri, const char **str) | 14 | @@ -XXX,XX +XXX,XX @@ vu_message_read(VuDev *vu_dev, int conn_fd, VhostUserMsg *vmsg) |
38 | { | 15 | }; |
39 | const char *cur; | 16 | if (vmsg->size) { |
40 | 17 | rc = qio_channel_readv_all_eof(ioc, &iov_payload, 1, &local_err); | |
41 | - if (str == NULL) | 18 | - if (rc == -1) { |
42 | + if (str == NULL) { | 19 | - error_report_err(local_err); |
43 | return -1; | 20 | + if (rc != 1) { |
44 | + } | 21 | + if (local_err) { |
45 | 22 | + error_report_err(local_err); | |
46 | cur = *str; | ||
47 | - if (!ISA_ALPHA(cur)) | ||
48 | + if (!ISA_ALPHA(cur)) { | ||
49 | return 2; | ||
50 | + } | ||
51 | cur++; | ||
52 | while (ISA_ALPHA(cur) || ISA_DIGIT(cur) || (*cur == '+') || (*cur == '-') || | ||
53 | - (*cur == '.')) | ||
54 | + (*cur == '.')) { | ||
55 | cur++; | ||
56 | + } | ||
57 | if (uri != NULL) { | ||
58 | g_free(uri->scheme); | ||
59 | uri->scheme = g_strndup(*str, cur - *str); | ||
60 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_fragment(URI *uri, const char **str) | ||
61 | { | ||
62 | const char *cur; | ||
63 | |||
64 | - if (str == NULL) | ||
65 | + if (str == NULL) { | ||
66 | return -1; | ||
67 | + } | ||
68 | |||
69 | cur = *str; | ||
70 | |||
71 | while ((ISA_PCHAR(cur)) || (*cur == '/') || (*cur == '?') || | ||
72 | (*cur == '[') || (*cur == ']') || | ||
73 | - ((uri != NULL) && (uri->cleanup & 1) && (IS_UNWISE(cur)))) | ||
74 | + ((uri != NULL) && (uri->cleanup & 1) && (IS_UNWISE(cur)))) { | ||
75 | NEXT(cur); | ||
76 | + } | ||
77 | if (uri != NULL) { | ||
78 | g_free(uri->fragment); | ||
79 | - if (uri->cleanup & 2) | ||
80 | + if (uri->cleanup & 2) { | ||
81 | uri->fragment = g_strndup(*str, cur - *str); | ||
82 | - else | ||
83 | + } else { | ||
84 | uri->fragment = uri_string_unescape(*str, cur - *str, NULL); | ||
85 | + } | ||
86 | } | ||
87 | *str = cur; | ||
88 | return 0; | ||
89 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_query(URI *uri, const char **str) | ||
90 | { | ||
91 | const char *cur; | ||
92 | |||
93 | - if (str == NULL) | ||
94 | + if (str == NULL) { | ||
95 | return -1; | ||
96 | + } | ||
97 | |||
98 | cur = *str; | ||
99 | |||
100 | while ((ISA_PCHAR(cur)) || (*cur == '/') || (*cur == '?') || | ||
101 | - ((uri != NULL) && (uri->cleanup & 1) && (IS_UNWISE(cur)))) | ||
102 | + ((uri != NULL) && (uri->cleanup & 1) && (IS_UNWISE(cur)))) { | ||
103 | NEXT(cur); | ||
104 | + } | ||
105 | if (uri != NULL) { | ||
106 | g_free(uri->query); | ||
107 | uri->query = g_strndup(*str, cur - *str); | ||
108 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_user_info(URI *uri, const char **str) | ||
109 | |||
110 | cur = *str; | ||
111 | while (ISA_UNRESERVED(cur) || ISA_PCT_ENCODED(cur) || ISA_SUB_DELIM(cur) || | ||
112 | - (*cur == ':')) | ||
113 | + (*cur == ':')) { | ||
114 | NEXT(cur); | ||
115 | + } | ||
116 | if (*cur == '@') { | ||
117 | if (uri != NULL) { | ||
118 | g_free(uri->user); | ||
119 | - if (uri->cleanup & 2) | ||
120 | + if (uri->cleanup & 2) { | ||
121 | uri->user = g_strndup(*str, cur - *str); | ||
122 | - else | ||
123 | + } else { | ||
124 | uri->user = uri_string_unescape(*str, cur - *str, NULL); | ||
125 | + } | 23 | + } |
126 | } | 24 | goto fail; |
127 | *str = cur; | ||
128 | return 0; | ||
129 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_dec_octet(const char **str) | ||
130 | { | ||
131 | const char *cur = *str; | ||
132 | |||
133 | - if (!(ISA_DIGIT(cur))) | ||
134 | + if (!(ISA_DIGIT(cur))) { | ||
135 | return 1; | ||
136 | - if (!ISA_DIGIT(cur + 1)) | ||
137 | + } | ||
138 | + if (!ISA_DIGIT(cur + 1)) { | ||
139 | cur++; | ||
140 | - else if ((*cur != '0') && (ISA_DIGIT(cur + 1)) && (!ISA_DIGIT(cur + 2))) | ||
141 | + } else if ((*cur != '0') && (ISA_DIGIT(cur + 1)) && (!ISA_DIGIT(cur + 2))) { | ||
142 | cur += 2; | ||
143 | - else if ((*cur == '1') && (ISA_DIGIT(cur + 1)) && (ISA_DIGIT(cur + 2))) | ||
144 | + } else if ((*cur == '1') && (ISA_DIGIT(cur + 1)) && (ISA_DIGIT(cur + 2))) { | ||
145 | cur += 3; | ||
146 | - else if ((*cur == '2') && (*(cur + 1) >= '0') && (*(cur + 1) <= '4') && | ||
147 | - (ISA_DIGIT(cur + 2))) | ||
148 | + } else if ((*cur == '2') && (*(cur + 1) >= '0') && (*(cur + 1) <= '4') && | ||
149 | + (ISA_DIGIT(cur + 2))) { | ||
150 | cur += 3; | ||
151 | - else if ((*cur == '2') && (*(cur + 1) == '5') && (*(cur + 2) >= '0') && | ||
152 | - (*(cur + 1) <= '5')) | ||
153 | + } else if ((*cur == '2') && (*(cur + 1) == '5') && (*(cur + 2) >= '0') && | ||
154 | + (*(cur + 1) <= '5')) { | ||
155 | cur += 3; | ||
156 | - else | ||
157 | + } else { | ||
158 | return 1; | ||
159 | + } | ||
160 | *str = cur; | ||
161 | return 0; | ||
162 | } | ||
163 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_host(URI *uri, const char **str) | ||
164 | */ | ||
165 | if (*cur == '[') { | ||
166 | cur++; | ||
167 | - while ((*cur != ']') && (*cur != 0)) | ||
168 | + while ((*cur != ']') && (*cur != 0)) { | ||
169 | cur++; | ||
170 | - if (*cur != ']') | ||
171 | + } | ||
172 | + if (*cur != ']') { | ||
173 | return 1; | ||
174 | + } | ||
175 | cur++; | ||
176 | goto found; | ||
177 | } | ||
178 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_host(URI *uri, const char **str) | ||
179 | * try to parse an IPv4 | ||
180 | */ | ||
181 | if (ISA_DIGIT(cur)) { | ||
182 | - if (rfc3986_parse_dec_octet(&cur) != 0) | ||
183 | + if (rfc3986_parse_dec_octet(&cur) != 0) { | ||
184 | goto not_ipv4; | ||
185 | - if (*cur != '.') | ||
186 | + } | ||
187 | + if (*cur != '.') { | ||
188 | goto not_ipv4; | ||
189 | + } | ||
190 | cur++; | ||
191 | - if (rfc3986_parse_dec_octet(&cur) != 0) | ||
192 | + if (rfc3986_parse_dec_octet(&cur) != 0) { | ||
193 | goto not_ipv4; | ||
194 | - if (*cur != '.') | ||
195 | + } | ||
196 | + if (*cur != '.') { | ||
197 | goto not_ipv4; | ||
198 | - if (rfc3986_parse_dec_octet(&cur) != 0) | ||
199 | + } | ||
200 | + if (rfc3986_parse_dec_octet(&cur) != 0) { | ||
201 | goto not_ipv4; | ||
202 | - if (*cur != '.') | ||
203 | + } | ||
204 | + if (*cur != '.') { | ||
205 | goto not_ipv4; | ||
206 | - if (rfc3986_parse_dec_octet(&cur) != 0) | ||
207 | + } | ||
208 | + if (rfc3986_parse_dec_octet(&cur) != 0) { | ||
209 | goto not_ipv4; | ||
210 | + } | ||
211 | goto found; | ||
212 | not_ipv4: | ||
213 | cur = *str; | ||
214 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_host(URI *uri, const char **str) | ||
215 | /* | ||
216 | * then this should be a hostname which can be empty | ||
217 | */ | ||
218 | - while (ISA_UNRESERVED(cur) || ISA_PCT_ENCODED(cur) || ISA_SUB_DELIM(cur)) | ||
219 | + while (ISA_UNRESERVED(cur) || ISA_PCT_ENCODED(cur) || ISA_SUB_DELIM(cur)) { | ||
220 | NEXT(cur); | ||
221 | + } | ||
222 | found: | ||
223 | if (uri != NULL) { | ||
224 | g_free(uri->authority); | ||
225 | uri->authority = NULL; | ||
226 | g_free(uri->server); | ||
227 | if (cur != host) { | ||
228 | - if (uri->cleanup & 2) | ||
229 | + if (uri->cleanup & 2) { | ||
230 | uri->server = g_strndup(host, cur - host); | ||
231 | - else | ||
232 | + } else { | ||
233 | uri->server = uri_string_unescape(host, cur - host, NULL); | ||
234 | - } else | ||
235 | + } | ||
236 | + } else { | ||
237 | uri->server = NULL; | ||
238 | + } | ||
239 | } | ||
240 | *str = cur; | ||
241 | return 0; | ||
242 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_authority(URI *uri, const char **str) | ||
243 | * try to parse a userinfo and check for the trailing @ | ||
244 | */ | ||
245 | ret = rfc3986_parse_user_info(uri, &cur); | ||
246 | - if ((ret != 0) || (*cur != '@')) | ||
247 | + if ((ret != 0) || (*cur != '@')) { | ||
248 | cur = *str; | ||
249 | - else | ||
250 | + } else { | ||
251 | cur++; | ||
252 | + } | ||
253 | ret = rfc3986_parse_host(uri, &cur); | ||
254 | - if (ret != 0) | ||
255 | + if (ret != 0) { | ||
256 | return ret; | ||
257 | + } | ||
258 | if (*cur == ':') { | ||
259 | cur++; | ||
260 | ret = rfc3986_parse_port(uri, &cur); | ||
261 | - if (ret != 0) | ||
262 | + if (ret != 0) { | ||
263 | return ret; | ||
264 | + } | ||
265 | } | ||
266 | *str = cur; | ||
267 | return 0; | ||
268 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_segment(const char **str, char forbid, int empty) | ||
269 | |||
270 | cur = *str; | ||
271 | if (!ISA_PCHAR(cur)) { | ||
272 | - if (empty) | ||
273 | + if (empty) { | ||
274 | return 0; | ||
275 | + } | ||
276 | return 1; | ||
277 | } | ||
278 | - while (ISA_PCHAR(cur) && (*cur != forbid)) | ||
279 | + while (ISA_PCHAR(cur) && (*cur != forbid)) { | ||
280 | NEXT(cur); | ||
281 | + } | ||
282 | *str = cur; | ||
283 | return 0; | ||
284 | } | ||
285 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
286 | while (*cur == '/') { | ||
287 | cur++; | ||
288 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
289 | - if (ret != 0) | ||
290 | + if (ret != 0) { | ||
291 | return ret; | ||
292 | + } | ||
293 | } | ||
294 | if (uri != NULL) { | ||
295 | g_free(uri->path); | ||
296 | if (*str != cur) { | ||
297 | - if (uri->cleanup & 2) | ||
298 | + if (uri->cleanup & 2) { | ||
299 | uri->path = g_strndup(*str, cur - *str); | ||
300 | - else | ||
301 | + } else { | ||
302 | uri->path = uri_string_unescape(*str, cur - *str, NULL); | ||
303 | + } | ||
304 | } else { | ||
305 | uri->path = NULL; | ||
306 | } | ||
307 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
308 | |||
309 | cur = *str; | ||
310 | |||
311 | - if (*cur != '/') | ||
312 | + if (*cur != '/') { | ||
313 | return 1; | ||
314 | + } | ||
315 | cur++; | ||
316 | ret = rfc3986_parse_segment(&cur, 0, 0); | ||
317 | if (ret == 0) { | ||
318 | while (*cur == '/') { | ||
319 | cur++; | ||
320 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
321 | - if (ret != 0) | ||
322 | + if (ret != 0) { | ||
323 | return ret; | ||
324 | + } | ||
325 | } | 25 | } |
326 | } | 26 | } |
327 | if (uri != NULL) { | ||
328 | g_free(uri->path); | ||
329 | if (cur != *str) { | ||
330 | - if (uri->cleanup & 2) | ||
331 | + if (uri->cleanup & 2) { | ||
332 | uri->path = g_strndup(*str, cur - *str); | ||
333 | - else | ||
334 | + } else { | ||
335 | uri->path = uri_string_unescape(*str, cur - *str, NULL); | ||
336 | + } | ||
337 | } else { | ||
338 | uri->path = NULL; | ||
339 | } | ||
340 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_rootless(URI *uri, const char **str) | ||
341 | cur = *str; | ||
342 | |||
343 | ret = rfc3986_parse_segment(&cur, 0, 0); | ||
344 | - if (ret != 0) | ||
345 | + if (ret != 0) { | ||
346 | return ret; | ||
347 | + } | ||
348 | while (*cur == '/') { | ||
349 | cur++; | ||
350 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
351 | - if (ret != 0) | ||
352 | + if (ret != 0) { | ||
353 | return ret; | ||
354 | + } | ||
355 | } | ||
356 | if (uri != NULL) { | ||
357 | g_free(uri->path); | ||
358 | if (cur != *str) { | ||
359 | - if (uri->cleanup & 2) | ||
360 | + if (uri->cleanup & 2) { | ||
361 | uri->path = g_strndup(*str, cur - *str); | ||
362 | - else | ||
363 | + } else { | ||
364 | uri->path = uri_string_unescape(*str, cur - *str, NULL); | ||
365 | + } | ||
366 | } else { | ||
367 | uri->path = NULL; | ||
368 | } | ||
369 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_no_scheme(URI *uri, const char **str) | ||
370 | cur = *str; | ||
371 | |||
372 | ret = rfc3986_parse_segment(&cur, ':', 0); | ||
373 | - if (ret != 0) | ||
374 | + if (ret != 0) { | ||
375 | return ret; | ||
376 | + } | ||
377 | while (*cur == '/') { | ||
378 | cur++; | ||
379 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
380 | - if (ret != 0) | ||
381 | + if (ret != 0) { | ||
382 | return ret; | ||
383 | + } | ||
384 | } | ||
385 | if (uri != NULL) { | ||
386 | g_free(uri->path); | ||
387 | if (cur != *str) { | ||
388 | - if (uri->cleanup & 2) | ||
389 | + if (uri->cleanup & 2) { | ||
390 | uri->path = g_strndup(*str, cur - *str); | ||
391 | - else | ||
392 | + } else { | ||
393 | uri->path = uri_string_unescape(*str, cur - *str, NULL); | ||
394 | + } | ||
395 | } else { | ||
396 | uri->path = NULL; | ||
397 | } | ||
398 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_hier_part(URI *uri, const char **str) | ||
399 | if ((*cur == '/') && (*(cur + 1) == '/')) { | ||
400 | cur += 2; | ||
401 | ret = rfc3986_parse_authority(uri, &cur); | ||
402 | - if (ret != 0) | ||
403 | + if (ret != 0) { | ||
404 | return ret; | ||
405 | + } | ||
406 | ret = rfc3986_parse_path_ab_empty(uri, &cur); | ||
407 | - if (ret != 0) | ||
408 | + if (ret != 0) { | ||
409 | return ret; | ||
410 | + } | ||
411 | *str = cur; | ||
412 | return 0; | ||
413 | } else if (*cur == '/') { | ||
414 | ret = rfc3986_parse_path_absolute(uri, &cur); | ||
415 | - if (ret != 0) | ||
416 | + if (ret != 0) { | ||
417 | return ret; | ||
418 | + } | ||
419 | } else if (ISA_PCHAR(cur)) { | ||
420 | ret = rfc3986_parse_path_rootless(uri, &cur); | ||
421 | - if (ret != 0) | ||
422 | + if (ret != 0) { | ||
423 | return ret; | ||
424 | + } | ||
425 | } else { | ||
426 | /* path-empty is effectively empty */ | ||
427 | if (uri != NULL) { | ||
428 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_relative_ref(URI *uri, const char *str) | ||
429 | if ((*str == '/') && (*(str + 1) == '/')) { | ||
430 | str += 2; | ||
431 | ret = rfc3986_parse_authority(uri, &str); | ||
432 | - if (ret != 0) | ||
433 | + if (ret != 0) { | ||
434 | return ret; | ||
435 | + } | ||
436 | ret = rfc3986_parse_path_ab_empty(uri, &str); | ||
437 | - if (ret != 0) | ||
438 | + if (ret != 0) { | ||
439 | return ret; | ||
440 | + } | ||
441 | } else if (*str == '/') { | ||
442 | ret = rfc3986_parse_path_absolute(uri, &str); | ||
443 | - if (ret != 0) | ||
444 | + if (ret != 0) { | ||
445 | return ret; | ||
446 | + } | ||
447 | } else if (ISA_PCHAR(str)) { | ||
448 | ret = rfc3986_parse_path_no_scheme(uri, &str); | ||
449 | - if (ret != 0) | ||
450 | + if (ret != 0) { | ||
451 | return ret; | ||
452 | + } | ||
453 | } else { | ||
454 | /* path-empty is effectively empty */ | ||
455 | if (uri != NULL) { | ||
456 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_relative_ref(URI *uri, const char *str) | ||
457 | if (*str == '?') { | ||
458 | str++; | ||
459 | ret = rfc3986_parse_query(uri, &str); | ||
460 | - if (ret != 0) | ||
461 | + if (ret != 0) { | ||
462 | return ret; | ||
463 | + } | ||
464 | } | ||
465 | if (*str == '#') { | ||
466 | str++; | ||
467 | ret = rfc3986_parse_fragment(uri, &str); | ||
468 | - if (ret != 0) | ||
469 | + if (ret != 0) { | ||
470 | return ret; | ||
471 | + } | ||
472 | } | ||
473 | if (*str != 0) { | ||
474 | uri_clean(uri); | ||
475 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse(URI *uri, const char *str) | ||
476 | int ret; | ||
477 | |||
478 | ret = rfc3986_parse_scheme(uri, &str); | ||
479 | - if (ret != 0) | ||
480 | + if (ret != 0) { | ||
481 | return ret; | ||
482 | + } | ||
483 | if (*str != ':') { | ||
484 | return 1; | ||
485 | } | ||
486 | str++; | ||
487 | ret = rfc3986_parse_hier_part(uri, &str); | ||
488 | - if (ret != 0) | ||
489 | + if (ret != 0) { | ||
490 | return ret; | ||
491 | + } | ||
492 | if (*str == '?') { | ||
493 | str++; | ||
494 | ret = rfc3986_parse_query(uri, &str); | ||
495 | - if (ret != 0) | ||
496 | + if (ret != 0) { | ||
497 | return ret; | ||
498 | + } | ||
499 | } | ||
500 | if (*str == '#') { | ||
501 | str++; | ||
502 | ret = rfc3986_parse_fragment(uri, &str); | ||
503 | - if (ret != 0) | ||
504 | + if (ret != 0) { | ||
505 | return ret; | ||
506 | + } | ||
507 | } | ||
508 | if (*str != 0) { | ||
509 | uri_clean(uri); | ||
510 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_uri_reference(URI *uri, const char *str) | ||
511 | { | ||
512 | int ret; | ||
513 | |||
514 | - if (str == NULL) | ||
515 | + if (str == NULL) { | ||
516 | return -1; | ||
517 | + } | ||
518 | uri_clean(uri); | ||
519 | |||
520 | /* | ||
521 | @@ -XXX,XX +XXX,XX @@ URI *uri_parse(const char *str) | ||
522 | URI *uri; | ||
523 | int ret; | ||
524 | |||
525 | - if (str == NULL) | ||
526 | + if (str == NULL) { | ||
527 | return NULL; | ||
528 | + } | ||
529 | uri = uri_new(); | ||
530 | ret = rfc3986_parse_uri_reference(uri, str); | ||
531 | if (ret) { | ||
532 | @@ -XXX,XX +XXX,XX @@ URI *uri_parse_raw(const char *str, int raw) | ||
533 | URI *uri; | ||
534 | int ret; | ||
535 | |||
536 | - if (str == NULL) | ||
537 | + if (str == NULL) { | ||
538 | return NULL; | ||
539 | + } | ||
540 | uri = uri_new(); | ||
541 | if (raw) { | ||
542 | uri->cleanup |= 2; | ||
543 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
544 | int len; | ||
545 | int max; | ||
546 | |||
547 | - if (uri == NULL) | ||
548 | + if (uri == NULL) { | ||
549 | return NULL; | ||
550 | + } | ||
551 | |||
552 | max = 80; | ||
553 | ret = g_malloc(max + 1); | ||
554 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
555 | temp = realloc2n(ret, &max); | ||
556 | ret = temp; | ||
557 | } | ||
558 | - if (IS_RESERVED(*(p)) || IS_UNRESERVED(*(p))) | ||
559 | + if (IS_RESERVED(*(p)) || IS_UNRESERVED(*(p))) { | ||
560 | ret[len++] = *p++; | ||
561 | - else { | ||
562 | + } else { | ||
563 | int val = *(unsigned char *)p++; | ||
564 | int hi = val / 0x10, lo = val % 0x10; | ||
565 | ret[len++] = '%'; | ||
566 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
567 | } | ||
568 | if ((IS_UNRESERVED(*(p))) || ((*(p) == ';')) || | ||
569 | ((*(p) == ':')) || ((*(p) == '&')) || ((*(p) == '=')) || | ||
570 | - ((*(p) == '+')) || ((*(p) == '$')) || ((*(p) == ','))) | ||
571 | + ((*(p) == '+')) || ((*(p) == '$')) || ((*(p) == ','))) { | ||
572 | ret[len++] = *p++; | ||
573 | - else { | ||
574 | + } else { | ||
575 | int val = *(unsigned char *)p++; | ||
576 | int hi = val / 0x10, lo = val % 0x10; | ||
577 | ret[len++] = '%'; | ||
578 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
579 | if ((IS_UNRESERVED(*(p))) || ((*(p) == '$')) || | ||
580 | ((*(p) == ',')) || ((*(p) == ';')) || ((*(p) == ':')) || | ||
581 | ((*(p) == '@')) || ((*(p) == '&')) || ((*(p) == '=')) || | ||
582 | - ((*(p) == '+'))) | ||
583 | + ((*(p) == '+'))) { | ||
584 | ret[len++] = *p++; | ||
585 | - else { | ||
586 | + } else { | ||
587 | int val = *(unsigned char *)p++; | ||
588 | int hi = val / 0x10, lo = val % 0x10; | ||
589 | ret[len++] = '%'; | ||
590 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
591 | if ((IS_UNRESERVED(*(p))) || ((*(p) == '/')) || | ||
592 | ((*(p) == ';')) || ((*(p) == '@')) || ((*(p) == '&')) || | ||
593 | ((*(p) == '=')) || ((*(p) == '+')) || ((*(p) == '$')) || | ||
594 | - ((*(p) == ','))) | ||
595 | + ((*(p) == ','))) { | ||
596 | ret[len++] = *p++; | ||
597 | - else { | ||
598 | + } else { | ||
599 | int val = *(unsigned char *)p++; | ||
600 | int hi = val / 0x10, lo = val % 0x10; | ||
601 | ret[len++] = '%'; | ||
602 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
603 | temp = realloc2n(ret, &max); | ||
604 | ret = temp; | ||
605 | } | ||
606 | - if ((IS_UNRESERVED(*(p))) || (IS_RESERVED(*(p)))) | ||
607 | + if ((IS_UNRESERVED(*(p))) || (IS_RESERVED(*(p)))) { | ||
608 | ret[len++] = *p++; | ||
609 | - else { | ||
610 | + } else { | ||
611 | int val = *(unsigned char *)p++; | ||
612 | int hi = val / 0x10, lo = val % 0x10; | ||
613 | ret[len++] = '%'; | ||
614 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
615 | */ | ||
616 | static void uri_clean(URI *uri) | ||
617 | { | ||
618 | - if (uri == NULL) | ||
619 | + if (uri == NULL) { | ||
620 | return; | ||
621 | + } | ||
622 | |||
623 | g_free(uri->scheme); | ||
624 | uri->scheme = NULL; | ||
625 | @@ -XXX,XX +XXX,XX @@ static int normalize_uri_path(char *path) | ||
626 | { | ||
627 | char *cur, *out; | ||
628 | |||
629 | - if (path == NULL) | ||
630 | + if (path == NULL) { | ||
631 | return -1; | ||
632 | + } | ||
633 | |||
634 | /* Skip all initial "/" chars. We want to get to the beginning of the | ||
635 | * first non-empty segment. | ||
636 | */ | ||
637 | cur = path; | ||
638 | - while (cur[0] == '/') | ||
639 | + while (cur[0] == '/') { | ||
640 | ++cur; | ||
641 | - if (cur[0] == '\0') | ||
642 | + } | ||
643 | + if (cur[0] == '\0') { | ||
644 | return 0; | ||
645 | + } | ||
646 | |||
647 | /* Keep everything we've seen so far. */ | ||
648 | out = cur; | ||
649 | @@ -XXX,XX +XXX,XX @@ static int normalize_uri_path(char *path) | ||
650 | if ((cur[0] == '.') && (cur[1] == '/')) { | ||
651 | cur += 2; | ||
652 | /* '//' normalization should be done at this point too */ | ||
653 | - while (cur[0] == '/') | ||
654 | + while (cur[0] == '/') { | ||
655 | cur++; | ||
656 | + } | ||
657 | continue; | ||
658 | } | ||
659 | |||
660 | @@ -XXX,XX +XXX,XX @@ static int normalize_uri_path(char *path) | ||
661 | * d) If the buffer string ends with "." as a complete path segment, | ||
662 | * that "." is removed. | ||
663 | */ | ||
664 | - if ((cur[0] == '.') && (cur[1] == '\0')) | ||
665 | + if ((cur[0] == '.') && (cur[1] == '\0')) { | ||
666 | break; | ||
667 | + } | ||
668 | |||
669 | /* Otherwise keep the segment. */ | ||
670 | while (cur[0] != '/') { | ||
671 | - if (cur[0] == '\0') | ||
672 | + if (cur[0] == '\0') { | ||
673 | goto done_cd; | ||
674 | + } | ||
675 | (out++)[0] = (cur++)[0]; | ||
676 | } | ||
677 | /* nomalize // */ | ||
678 | - while ((cur[0] == '/') && (cur[1] == '/')) | ||
679 | + while ((cur[0] == '/') && (cur[1] == '/')) { | ||
680 | cur++; | ||
681 | + } | ||
682 | |||
683 | (out++)[0] = (cur++)[0]; | ||
684 | } | ||
685 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
686 | |||
687 | /* Reset to the beginning of the first segment for the next sequence. */ | ||
688 | cur = path; | ||
689 | - while (cur[0] == '/') | ||
690 | + while (cur[0] == '/') { | ||
691 | ++cur; | ||
692 | - if (cur[0] == '\0') | ||
693 | + } | ||
694 | + if (cur[0] == '\0') { | ||
695 | return 0; | ||
696 | + } | ||
697 | |||
698 | /* | ||
699 | * Analyze each segment in sequence for cases (e) and (f). | ||
700 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
701 | |||
702 | /* Find the end of the current segment. */ | ||
703 | segp = cur; | ||
704 | - while ((segp[0] != '/') && (segp[0] != '\0')) | ||
705 | + while ((segp[0] != '/') && (segp[0] != '\0')) { | ||
706 | ++segp; | ||
707 | + } | ||
708 | |||
709 | /* If this is the last segment, we're done (we need at least two | ||
710 | * segments to meet the criteria for the (e) and (f) cases). | ||
711 | */ | ||
712 | - if (segp[0] == '\0') | ||
713 | + if (segp[0] == '\0') { | ||
714 | break; | ||
715 | + } | ||
716 | |||
717 | /* If the first segment is "..", or if the next segment _isn't_ "..", | ||
718 | * keep this segment and try the next one. | ||
719 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
720 | /* string will overlap, do not use strcpy */ | ||
721 | tmp = cur; | ||
722 | segp += 3; | ||
723 | - while ((*tmp++ = *segp++) != 0) | ||
724 | - ; | ||
725 | + while ((*tmp++ = *segp++) != 0) { | ||
726 | + /* No further work */ | ||
727 | + } | ||
728 | |||
729 | /* If there are no previous segments, then keep going from here. */ | ||
730 | segp = cur; | ||
731 | - while ((segp > path) && ((--segp)[0] == '/')) | ||
732 | - ; | ||
733 | - if (segp == path) | ||
734 | + while ((segp > path) && ((--segp)[0] == '/')) { | ||
735 | + /* No further work */ | ||
736 | + } | ||
737 | + if (segp == path) { | ||
738 | continue; | ||
739 | + } | ||
740 | |||
741 | /* "segp" is pointing to the end of a previous segment; find it's | ||
742 | * start. We need to back up to the previous segment and start | ||
743 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
744 | * remove the "foo/..". | ||
745 | */ | ||
746 | cur = segp; | ||
747 | - while ((cur > path) && (cur[-1] != '/')) | ||
748 | + while ((cur > path) && (cur[-1] != '/')) { | ||
749 | --cur; | ||
750 | + } | ||
751 | } | ||
752 | out[0] = '\0'; | ||
753 | |||
754 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
755 | if (path[0] == '/') { | ||
756 | cur = path; | ||
757 | while ((cur[0] == '/') && (cur[1] == '.') && (cur[2] == '.') && | ||
758 | - ((cur[3] == '/') || (cur[3] == '\0'))) | ||
759 | + ((cur[3] == '/') || (cur[3] == '\0'))) { | ||
760 | cur += 3; | ||
761 | + } | ||
762 | |||
763 | if (cur != path) { | ||
764 | out = path; | ||
765 | - while (cur[0] != '\0') | ||
766 | + while (cur[0] != '\0') { | ||
767 | (out++)[0] = (cur++)[0]; | ||
768 | + } | ||
769 | out[0] = 0; | ||
770 | } | ||
771 | } | ||
772 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
773 | static int is_hex(char c) | ||
774 | { | ||
775 | if (((c >= '0') && (c <= '9')) || ((c >= 'a') && (c <= 'f')) || | ||
776 | - ((c >= 'A') && (c <= 'F'))) | ||
777 | + ((c >= 'A') && (c <= 'F'))) { | ||
778 | return 1; | ||
779 | + } | ||
780 | return 0; | ||
781 | } | ||
782 | |||
783 | @@ -XXX,XX +XXX,XX @@ char *uri_string_unescape(const char *str, int len, char *target) | ||
784 | char *ret, *out; | ||
785 | const char *in; | ||
786 | |||
787 | - if (str == NULL) | ||
788 | + if (str == NULL) { | ||
789 | return NULL; | ||
790 | - if (len <= 0) | ||
791 | + } | ||
792 | + if (len <= 0) { | ||
793 | len = strlen(str); | ||
794 | - if (len < 0) | ||
795 | + } | ||
796 | + if (len < 0) { | ||
797 | return NULL; | ||
798 | + } | ||
799 | |||
800 | if (target == NULL) { | ||
801 | ret = g_malloc(len + 1); | ||
802 | - } else | ||
803 | + } else { | ||
804 | ret = target; | ||
805 | + } | ||
806 | in = str; | ||
807 | out = ret; | ||
808 | while (len > 0) { | ||
809 | if ((len > 2) && (*in == '%') && (is_hex(in[1])) && (is_hex(in[2]))) { | ||
810 | in++; | ||
811 | - if ((*in >= '0') && (*in <= '9')) | ||
812 | + if ((*in >= '0') && (*in <= '9')) { | ||
813 | *out = (*in - '0'); | ||
814 | - else if ((*in >= 'a') && (*in <= 'f')) | ||
815 | + } else if ((*in >= 'a') && (*in <= 'f')) { | ||
816 | *out = (*in - 'a') + 10; | ||
817 | - else if ((*in >= 'A') && (*in <= 'F')) | ||
818 | + } else if ((*in >= 'A') && (*in <= 'F')) { | ||
819 | *out = (*in - 'A') + 10; | ||
820 | + } | ||
821 | in++; | ||
822 | - if ((*in >= '0') && (*in <= '9')) | ||
823 | + if ((*in >= '0') && (*in <= '9')) { | ||
824 | *out = *out * 16 + (*in - '0'); | ||
825 | - else if ((*in >= 'a') && (*in <= 'f')) | ||
826 | + } else if ((*in >= 'a') && (*in <= 'f')) { | ||
827 | *out = *out * 16 + (*in - 'a') + 10; | ||
828 | - else if ((*in >= 'A') && (*in <= 'F')) | ||
829 | + } else if ((*in >= 'A') && (*in <= 'F')) { | ||
830 | *out = *out * 16 + (*in - 'A') + 10; | ||
831 | + } | ||
832 | in++; | ||
833 | len -= 3; | ||
834 | out++; | ||
835 | @@ -XXX,XX +XXX,XX @@ char *uri_string_escape(const char *str, const char *list) | ||
836 | const char *in; | ||
837 | int len, out; | ||
838 | |||
839 | - if (str == NULL) | ||
840 | + if (str == NULL) { | ||
841 | return NULL; | ||
842 | - if (str[0] == 0) | ||
843 | + } | ||
844 | + if (str[0] == 0) { | ||
845 | return g_strdup(str); | ||
846 | + } | ||
847 | len = strlen(str); | ||
848 | - if (!(len > 0)) | ||
849 | + if (!(len > 0)) { | ||
850 | return NULL; | ||
851 | + } | ||
852 | |||
853 | len += 20; | ||
854 | ret = g_malloc(len); | ||
855 | @@ -XXX,XX +XXX,XX @@ char *uri_string_escape(const char *str, const char *list) | ||
856 | unsigned char val; | ||
857 | ret[out++] = '%'; | ||
858 | val = ch >> 4; | ||
859 | - if (val <= 9) | ||
860 | + if (val <= 9) { | ||
861 | ret[out++] = '0' + val; | ||
862 | - else | ||
863 | + } else { | ||
864 | ret[out++] = 'A' + val - 0xA; | ||
865 | + } | ||
866 | val = ch & 0xF; | ||
867 | - if (val <= 9) | ||
868 | + if (val <= 9) { | ||
869 | ret[out++] = '0' + val; | ||
870 | - else | ||
871 | + } else { | ||
872 | ret[out++] = 'A' + val - 0xA; | ||
873 | + } | ||
874 | in++; | ||
875 | } else { | ||
876 | ret[out++] = *in++; | ||
877 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
878 | * as a reference to "." rather than as a synonym for the current | ||
879 | * URI. Should we do that here? | ||
880 | */ | ||
881 | - if (uri == NULL) | ||
882 | + if (uri == NULL) { | ||
883 | ret = -1; | ||
884 | - else { | ||
885 | + } else { | ||
886 | if (*uri) { | ||
887 | ref = uri_new(); | ||
888 | ret = uri_parse_into(ref, uri); | ||
889 | - } else | ||
890 | + } else { | ||
891 | ret = 0; | ||
892 | + } | ||
893 | } | ||
894 | - if (ret != 0) | ||
895 | + if (ret != 0) { | ||
896 | goto done; | ||
897 | + } | ||
898 | if ((ref != NULL) && (ref->scheme != NULL)) { | ||
899 | /* | ||
900 | * The URI is absolute don't modify. | ||
901 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
902 | val = g_strdup(uri); | ||
903 | goto done; | ||
904 | } | ||
905 | - if (base == NULL) | ||
906 | + if (base == NULL) { | ||
907 | ret = -1; | ||
908 | - else { | ||
909 | + } else { | ||
910 | bas = uri_new(); | ||
911 | ret = uri_parse_into(bas, base); | ||
912 | } | ||
913 | if (ret != 0) { | ||
914 | - if (ref) | ||
915 | + if (ref) { | ||
916 | val = uri_to_string(ref); | ||
917 | + } | ||
918 | goto done; | ||
919 | } | ||
920 | if (ref == NULL) { | ||
921 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
922 | if ((ref->scheme == NULL) && (ref->path == NULL) && | ||
923 | ((ref->authority == NULL) && (ref->server == NULL))) { | ||
924 | res->scheme = g_strdup(bas->scheme); | ||
925 | - if (bas->authority != NULL) | ||
926 | + if (bas->authority != NULL) { | ||
927 | res->authority = g_strdup(bas->authority); | ||
928 | - else if (bas->server != NULL) { | ||
929 | + } else if (bas->server != NULL) { | ||
930 | res->server = g_strdup(bas->server); | ||
931 | res->user = g_strdup(bas->user); | ||
932 | res->port = bas->port; | ||
933 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
934 | * use an authority component. | ||
935 | */ | ||
936 | if ((ref->authority != NULL) || (ref->server != NULL)) { | ||
937 | - if (ref->authority != NULL) | ||
938 | + if (ref->authority != NULL) { | ||
939 | res->authority = g_strdup(ref->authority); | ||
940 | - else { | ||
941 | + } else { | ||
942 | res->server = g_strdup(ref->server); | ||
943 | res->user = g_strdup(ref->user); | ||
944 | res->port = ref->port; | ||
945 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
946 | res->path = g_strdup(ref->path); | ||
947 | goto step_7; | ||
948 | } | ||
949 | - if (bas->authority != NULL) | ||
950 | + if (bas->authority != NULL) { | ||
951 | res->authority = g_strdup(bas->authority); | ||
952 | - else if (bas->server != NULL) { | ||
953 | + } else if (bas->server != NULL) { | ||
954 | res->server = g_strdup(bas->server); | ||
955 | res->user = g_strdup(bas->user); | ||
956 | res->port = bas->port; | ||
957 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
958 | * Allocate a buffer large enough for the result string. | ||
959 | */ | ||
960 | len = 2; /* extra / and 0 */ | ||
961 | - if (ref->path != NULL) | ||
962 | + if (ref->path != NULL) { | ||
963 | len += strlen(ref->path); | ||
964 | - if (bas->path != NULL) | ||
965 | + } | ||
966 | + if (bas->path != NULL) { | ||
967 | len += strlen(bas->path); | ||
968 | + } | ||
969 | res->path = g_malloc(len); | ||
970 | res->path[0] = 0; | ||
971 | |||
972 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
973 | out = 0; | ||
974 | if (bas->path != NULL) { | ||
975 | while (bas->path[cur] != 0) { | ||
976 | - while ((bas->path[cur] != 0) && (bas->path[cur] != '/')) | ||
977 | + while ((bas->path[cur] != 0) && (bas->path[cur] != '/')) { | ||
978 | cur++; | ||
979 | - if (bas->path[cur] == 0) | ||
980 | + } | ||
981 | + if (bas->path[cur] == 0) { | ||
982 | break; | ||
983 | + } | ||
984 | |||
985 | cur++; | ||
986 | while (out < cur) { | ||
987 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
988 | /* | ||
989 | * Ensure the path includes a '/' | ||
990 | */ | ||
991 | - if ((out == 0) && (bas->server != NULL)) | ||
992 | + if ((out == 0) && (bas->server != NULL)) { | ||
993 | res->path[out++] = '/'; | ||
994 | + } | ||
995 | while (ref->path[indx] != 0) { | ||
996 | res->path[out++] = ref->path[indx++]; | ||
997 | } | ||
998 | @@ -XXX,XX +XXX,XX @@ step_7: | ||
999 | val = uri_to_string(res); | ||
1000 | |||
1001 | done: | ||
1002 | - if (ref != NULL) | ||
1003 | + if (ref != NULL) { | ||
1004 | uri_free(ref); | ||
1005 | - if (bas != NULL) | ||
1006 | + } | ||
1007 | + if (bas != NULL) { | ||
1008 | uri_free(bas); | ||
1009 | - if (res != NULL) | ||
1010 | + } | ||
1011 | + if (res != NULL) { | ||
1012 | uri_free(res); | ||
1013 | + } | ||
1014 | return val; | ||
1015 | } | ||
1016 | |||
1017 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1018 | char *bptr, *uptr, *vptr; | ||
1019 | int remove_path = 0; | ||
1020 | |||
1021 | - if ((uri == NULL) || (*uri == 0)) | ||
1022 | + if ((uri == NULL) || (*uri == 0)) { | ||
1023 | return NULL; | ||
1024 | + } | ||
1025 | |||
1026 | /* | ||
1027 | * First parse URI into a standard form | ||
1028 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1029 | /* If URI not already in "relative" form */ | ||
1030 | if (uri[0] != '.') { | ||
1031 | ret = uri_parse_into(ref, uri); | ||
1032 | - if (ret != 0) | ||
1033 | + if (ret != 0) { | ||
1034 | goto done; /* Error in URI, return NULL */ | ||
1035 | - } else | ||
1036 | + } | ||
1037 | + } else { | ||
1038 | ref->path = g_strdup(uri); | ||
1039 | + } | ||
1040 | |||
1041 | /* | ||
1042 | * Next parse base into the same standard form | ||
1043 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1044 | bas = uri_new(); | ||
1045 | if (base[0] != '.') { | ||
1046 | ret = uri_parse_into(bas, base); | ||
1047 | - if (ret != 0) | ||
1048 | + if (ret != 0) { | ||
1049 | goto done; /* Error in base, return NULL */ | ||
1050 | - } else | ||
1051 | + } | ||
1052 | + } else { | ||
1053 | bas->path = g_strdup(base); | ||
1054 | + } | ||
1055 | |||
1056 | /* | ||
1057 | * If the scheme / server on the URI differs from the base, | ||
1058 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1059 | if (bas->path == NULL) { | ||
1060 | if (ref->path != NULL) { | ||
1061 | uptr = ref->path; | ||
1062 | - if (*uptr == '/') | ||
1063 | + if (*uptr == '/') { | ||
1064 | uptr++; | ||
1065 | + } | ||
1066 | /* exception characters from uri_to_string */ | ||
1067 | val = uri_string_escape(uptr, "/;&=+$,"); | ||
1068 | } | ||
1069 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1070 | bptr = bas->path; | ||
1071 | if (ref->path == NULL) { | ||
1072 | for (ix = 0; bptr[ix] != 0; ix++) { | ||
1073 | - if (bptr[ix] == '/') | ||
1074 | + if (bptr[ix] == '/') { | ||
1075 | nbslash++; | ||
1076 | + } | ||
1077 | } | ||
1078 | uptr = NULL; | ||
1079 | len = 1; /* this is for a string terminator only */ | ||
1080 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1081 | /* | ||
1082 | * Next we compare the two strings and find where they first differ | ||
1083 | */ | ||
1084 | - if ((ref->path[pos] == '.') && (ref->path[pos + 1] == '/')) | ||
1085 | + if ((ref->path[pos] == '.') && (ref->path[pos + 1] == '/')) { | ||
1086 | pos += 2; | ||
1087 | - if ((*bptr == '.') && (bptr[1] == '/')) | ||
1088 | + } | ||
1089 | + if ((*bptr == '.') && (bptr[1] == '/')) { | ||
1090 | bptr += 2; | ||
1091 | - else if ((*bptr == '/') && (ref->path[pos] != '/')) | ||
1092 | + } else if ((*bptr == '/') && (ref->path[pos] != '/')) { | ||
1093 | bptr++; | ||
1094 | - while ((bptr[pos] == ref->path[pos]) && (bptr[pos] != 0)) | ||
1095 | + } | ||
1096 | + while ((bptr[pos] == ref->path[pos]) && (bptr[pos] != 0)) { | ||
1097 | pos++; | ||
1098 | + } | ||
1099 | |||
1100 | if (bptr[pos] == ref->path[pos]) { | ||
1101 | val = g_strdup(""); | ||
1102 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1103 | * beginning of the "unique" suffix of URI | ||
1104 | */ | ||
1105 | ix = pos; | ||
1106 | - if ((ref->path[ix] == '/') && (ix > 0)) | ||
1107 | + if ((ref->path[ix] == '/') && (ix > 0)) { | ||
1108 | ix--; | ||
1109 | - else if ((ref->path[ix] == 0) && (ix > 1) && (ref->path[ix - 1] == '/')) | ||
1110 | + } else if ((ref->path[ix] == 0) && (ix > 1) | ||
1111 | + && (ref->path[ix - 1] == '/')) { | ||
1112 | ix -= 2; | ||
1113 | + } | ||
1114 | for (; ix > 0; ix--) { | ||
1115 | - if (ref->path[ix] == '/') | ||
1116 | + if (ref->path[ix] == '/') { | ||
1117 | break; | ||
1118 | + } | ||
1119 | } | ||
1120 | if (ix == 0) { | ||
1121 | uptr = ref->path; | ||
1122 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1123 | */ | ||
1124 | if (bptr[pos] != ref->path[pos]) { /* check for trivial URI == base */ | ||
1125 | for (; bptr[ix] != 0; ix++) { | ||
1126 | - if (bptr[ix] == '/') | ||
1127 | + if (bptr[ix] == '/') { | ||
1128 | nbslash++; | ||
1129 | + } | ||
1130 | } | ||
1131 | } | ||
1132 | len = strlen(uptr) + 1; | ||
1133 | } | ||
1134 | |||
1135 | if (nbslash == 0) { | ||
1136 | - if (uptr != NULL) | ||
1137 | + if (uptr != NULL) { | ||
1138 | /* exception characters from uri_to_string */ | ||
1139 | val = uri_string_escape(uptr, "/;&=+$,"); | ||
1140 | + } | ||
1141 | goto done; | ||
1142 | } | ||
1143 | |||
1144 | @@ -XXX,XX +XXX,XX @@ done: | ||
1145 | /* | ||
1146 | * Free the working variables | ||
1147 | */ | ||
1148 | - if (remove_path != 0) | ||
1149 | + if (remove_path != 0) { | ||
1150 | ref->path = NULL; | ||
1151 | - if (ref != NULL) | ||
1152 | + } | ||
1153 | + if (ref != NULL) { | ||
1154 | uri_free(ref); | ||
1155 | - if (bas != NULL) | ||
1156 | + } | ||
1157 | + if (bas != NULL) { | ||
1158 | uri_free(bas); | ||
1159 | + } | ||
1160 | |||
1161 | return val; | ||
1162 | } | ||
1163 | @@ -XXX,XX +XXX,XX @@ struct QueryParams *query_params_new(int init_alloc) | ||
1164 | { | ||
1165 | struct QueryParams *ps; | ||
1166 | |||
1167 | - if (init_alloc <= 0) | ||
1168 | + if (init_alloc <= 0) { | ||
1169 | init_alloc = 1; | ||
1170 | + } | ||
1171 | |||
1172 | ps = g_new(QueryParams, 1); | ||
1173 | ps->n = 0; | ||
1174 | @@ -XXX,XX +XXX,XX @@ struct QueryParams *query_params_parse(const char *query) | ||
1175 | const char *end, *eq; | ||
1176 | |||
1177 | ps = query_params_new(0); | ||
1178 | - if (!query || query[0] == '\0') | ||
1179 | + if (!query || query[0] == '\0') { | ||
1180 | return ps; | ||
1181 | + } | ||
1182 | |||
1183 | while (*query) { | ||
1184 | char *name = NULL, *value = NULL; | ||
1185 | |||
1186 | /* Find the next separator, or end of the string. */ | ||
1187 | end = strchr(query, '&'); | ||
1188 | - if (!end) | ||
1189 | + if (!end) { | ||
1190 | end = strchr(query, ';'); | ||
1191 | - if (!end) | ||
1192 | + } | ||
1193 | + if (!end) { | ||
1194 | end = query + strlen(query); | ||
1195 | + } | ||
1196 | |||
1197 | /* Find the first '=' character between here and end. */ | ||
1198 | eq = strchr(query, '='); | ||
1199 | - if (eq && eq >= end) | ||
1200 | + if (eq && eq >= end) { | ||
1201 | eq = NULL; | ||
1202 | + } | ||
1203 | |||
1204 | /* Empty section (eg. "&&"). */ | ||
1205 | - if (end == query) | ||
1206 | + if (end == query) { | ||
1207 | goto next; | ||
1208 | + } | ||
1209 | |||
1210 | /* If there is no '=' character, then we have just "name" | ||
1211 | * and consistent with CGI.pm we assume value is "". | ||
1212 | @@ -XXX,XX +XXX,XX @@ struct QueryParams *query_params_parse(const char *query) | ||
1213 | /* If the '=' character is at the beginning then we have | ||
1214 | * "=value" and consistent with CGI.pm we _ignore_ this. | ||
1215 | */ | ||
1216 | - else if (query == eq) | ||
1217 | + else if (query == eq) { | ||
1218 | goto next; | ||
1219 | + } | ||
1220 | |||
1221 | /* Otherwise it's "name=value". */ | ||
1222 | else { | ||
1223 | @@ -XXX,XX +XXX,XX @@ struct QueryParams *query_params_parse(const char *query) | ||
1224 | |||
1225 | next: | ||
1226 | query = end; | ||
1227 | - if (*query) | ||
1228 | + if (*query) { | ||
1229 | query++; /* skip '&' separator */ | ||
1230 | + } | ||
1231 | } | ||
1232 | |||
1233 | return ps; | ||
1234 | -- | 27 | -- |
1235 | 2.14.3 | 28 | 2.26.2 |
1236 | 29 | ||
1237 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | The vu_client_trip() coroutine is leaked during AioContext switching. It | ||
2 | is also unsafe to destroy the vu_dev in panic_cb() since its callers | ||
3 | still access it in some cases. | ||
1 | 4 | ||
5 | Rework the lifecycle to solve these safety issues. | ||
6 | |||
7 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
8 | Message-id: 20200924151549.913737-10-stefanha@redhat.com | ||
9 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
10 | --- | ||
11 | util/vhost-user-server.h | 29 ++-- | ||
12 | block/export/vhost-user-blk-server.c | 9 +- | ||
13 | util/vhost-user-server.c | 245 +++++++++++++++------------ | ||
14 | 3 files changed, 155 insertions(+), 128 deletions(-) | ||
15 | |||
16 | diff --git a/util/vhost-user-server.h b/util/vhost-user-server.h | ||
17 | index XXXXXXX..XXXXXXX 100644 | ||
18 | --- a/util/vhost-user-server.h | ||
19 | +++ b/util/vhost-user-server.h | ||
20 | @@ -XXX,XX +XXX,XX @@ | ||
21 | #include "qapi/error.h" | ||
22 | #include "standard-headers/linux/virtio_blk.h" | ||
23 | |||
24 | +/* A kick fd that we monitor on behalf of libvhost-user */ | ||
25 | typedef struct VuFdWatch { | ||
26 | VuDev *vu_dev; | ||
27 | int fd; /*kick fd*/ | ||
28 | void *pvt; | ||
29 | vu_watch_cb cb; | ||
30 | - bool processing; | ||
31 | QTAILQ_ENTRY(VuFdWatch) next; | ||
32 | } VuFdWatch; | ||
33 | |||
34 | -typedef struct VuServer VuServer; | ||
35 | - | ||
36 | -struct VuServer { | ||
37 | +/** | ||
38 | + * VuServer: | ||
39 | + * A vhost-user server instance with user-defined VuDevIface callbacks. | ||
40 | + * Vhost-user device backends can be implemented using VuServer. VuDevIface | ||
41 | + * callbacks and virtqueue kicks run in the given AioContext. | ||
42 | + */ | ||
43 | +typedef struct { | ||
44 | QIONetListener *listener; | ||
45 | + QEMUBH *restart_listener_bh; | ||
46 | AioContext *ctx; | ||
47 | int max_queues; | ||
48 | const VuDevIface *vu_iface; | ||
49 | + | ||
50 | + /* Protected by ctx lock */ | ||
51 | VuDev vu_dev; | ||
52 | QIOChannel *ioc; /* The I/O channel with the client */ | ||
53 | QIOChannelSocket *sioc; /* The underlying data channel with the client */ | ||
54 | - /* IOChannel for fd provided via VHOST_USER_SET_SLAVE_REQ_FD */ | ||
55 | - QIOChannel *ioc_slave; | ||
56 | - QIOChannelSocket *sioc_slave; | ||
57 | - Coroutine *co_trip; /* coroutine for processing VhostUserMsg */ | ||
58 | QTAILQ_HEAD(, VuFdWatch) vu_fd_watches; | ||
59 | - /* restart coroutine co_trip if AIOContext is changed */ | ||
60 | - bool aio_context_changed; | ||
61 | - bool processing_msg; | ||
62 | -}; | ||
63 | + | ||
64 | + Coroutine *co_trip; /* coroutine for processing VhostUserMsg */ | ||
65 | +} VuServer; | ||
66 | |||
67 | bool vhost_user_server_start(VuServer *server, | ||
68 | SocketAddress *unix_socket, | ||
69 | @@ -XXX,XX +XXX,XX @@ bool vhost_user_server_start(VuServer *server, | ||
70 | |||
71 | void vhost_user_server_stop(VuServer *server); | ||
72 | |||
73 | -void vhost_user_server_set_aio_context(VuServer *server, AioContext *ctx); | ||
74 | +void vhost_user_server_attach_aio_context(VuServer *server, AioContext *ctx); | ||
75 | +void vhost_user_server_detach_aio_context(VuServer *server); | ||
76 | |||
77 | #endif /* VHOST_USER_SERVER_H */ | ||
78 | diff --git a/block/export/vhost-user-blk-server.c b/block/export/vhost-user-blk-server.c | ||
79 | index XXXXXXX..XXXXXXX 100644 | ||
80 | --- a/block/export/vhost-user-blk-server.c | ||
81 | +++ b/block/export/vhost-user-blk-server.c | ||
82 | @@ -XXX,XX +XXX,XX @@ static const VuDevIface vu_block_iface = { | ||
83 | static void blk_aio_attached(AioContext *ctx, void *opaque) | ||
84 | { | ||
85 | VuBlockDev *vub_dev = opaque; | ||
86 | - aio_context_acquire(ctx); | ||
87 | - vhost_user_server_set_aio_context(&vub_dev->vu_server, ctx); | ||
88 | - aio_context_release(ctx); | ||
89 | + vhost_user_server_attach_aio_context(&vub_dev->vu_server, ctx); | ||
90 | } | ||
91 | |||
92 | static void blk_aio_detach(void *opaque) | ||
93 | { | ||
94 | VuBlockDev *vub_dev = opaque; | ||
95 | - AioContext *ctx = vub_dev->vu_server.ctx; | ||
96 | - aio_context_acquire(ctx); | ||
97 | - vhost_user_server_set_aio_context(&vub_dev->vu_server, NULL); | ||
98 | - aio_context_release(ctx); | ||
99 | + vhost_user_server_detach_aio_context(&vub_dev->vu_server); | ||
100 | } | ||
101 | |||
102 | static void | ||
103 | diff --git a/util/vhost-user-server.c b/util/vhost-user-server.c | ||
104 | index XXXXXXX..XXXXXXX 100644 | ||
105 | --- a/util/vhost-user-server.c | ||
106 | +++ b/util/vhost-user-server.c | ||
107 | @@ -XXX,XX +XXX,XX @@ | ||
108 | */ | ||
109 | #include "qemu/osdep.h" | ||
110 | #include "qemu/main-loop.h" | ||
111 | +#include "block/aio-wait.h" | ||
112 | #include "vhost-user-server.h" | ||
113 | |||
114 | +/* | ||
115 | + * Theory of operation: | ||
116 | + * | ||
117 | + * VuServer is started and stopped by vhost_user_server_start() and | ||
118 | + * vhost_user_server_stop() from the main loop thread. Starting the server | ||
119 | + * opens a vhost-user UNIX domain socket and listens for incoming connections. | ||
120 | + * Only one connection is allowed at a time. | ||
121 | + * | ||
122 | + * The connection is handled by the vu_client_trip() coroutine in the | ||
123 | + * VuServer->ctx AioContext. The coroutine consists of a vu_dispatch() loop | ||
124 | + * where libvhost-user calls vu_message_read() to receive the next vhost-user | ||
125 | + * protocol messages over the UNIX domain socket. | ||
126 | + * | ||
127 | + * When virtqueues are set up libvhost-user calls set_watch() to monitor kick | ||
128 | + * fds. These fds are also handled in the VuServer->ctx AioContext. | ||
129 | + * | ||
130 | + * Both vu_client_trip() and kick fd monitoring can be stopped by shutting down | ||
131 | + * the socket connection. Shutting down the socket connection causes | ||
132 | + * vu_message_read() to fail since no more data can be received from the socket. | ||
133 | + * After vu_dispatch() fails, vu_client_trip() calls vu_deinit() to stop | ||
134 | + * libvhost-user before terminating the coroutine. vu_deinit() calls | ||
135 | + * remove_watch() to stop monitoring kick fds and this stops virtqueue | ||
136 | + * processing. | ||
137 | + * | ||
138 | + * When vu_client_trip() has finished cleaning up it schedules a BH in the main | ||
139 | + * loop thread to accept the next client connection. | ||
140 | + * | ||
141 | + * When libvhost-user detects an error it calls panic_cb() and sets the | ||
142 | + * dev->broken flag. Both vu_client_trip() and kick fd processing stop when | ||
143 | + * the dev->broken flag is set. | ||
144 | + * | ||
145 | + * It is possible to switch AioContexts using | ||
146 | + * vhost_user_server_detach_aio_context() and | ||
147 | + * vhost_user_server_attach_aio_context(). They stop monitoring fds in the old | ||
148 | + * AioContext and resume monitoring in the new AioContext. The vu_client_trip() | ||
149 | + * coroutine remains in a yielded state during the switch. This is made | ||
150 | + * possible by QIOChannel's support for spurious coroutine re-entry in | ||
151 | + * qio_channel_yield(). The coroutine will restart I/O when re-entered from the | ||
152 | + * new AioContext. | ||
153 | + */ | ||
154 | + | ||
155 | static void vmsg_close_fds(VhostUserMsg *vmsg) | ||
156 | { | ||
157 | int i; | ||
158 | @@ -XXX,XX +XXX,XX @@ static void vmsg_unblock_fds(VhostUserMsg *vmsg) | ||
159 | } | ||
160 | } | ||
161 | |||
162 | -static void vu_accept(QIONetListener *listener, QIOChannelSocket *sioc, | ||
163 | - gpointer opaque); | ||
164 | - | ||
165 | -static void close_client(VuServer *server) | ||
166 | -{ | ||
167 | - /* | ||
168 | - * Before closing the client | ||
169 | - * | ||
170 | - * 1. Let vu_client_trip stop processing new vhost-user msg | ||
171 | - * | ||
172 | - * 2. remove kick_handler | ||
173 | - * | ||
174 | - * 3. wait for the kick handler to be finished | ||
175 | - * | ||
176 | - * 4. wait for the current vhost-user msg to be finished processing | ||
177 | - */ | ||
178 | - | ||
179 | - QIOChannelSocket *sioc = server->sioc; | ||
180 | - /* When this is set vu_client_trip will stop new processing vhost-user message */ | ||
181 | - server->sioc = NULL; | ||
182 | - | ||
183 | - while (server->processing_msg) { | ||
184 | - if (server->ioc->read_coroutine) { | ||
185 | - server->ioc->read_coroutine = NULL; | ||
186 | - qio_channel_set_aio_fd_handler(server->ioc, server->ioc->ctx, NULL, | ||
187 | - NULL, server->ioc); | ||
188 | - server->processing_msg = false; | ||
189 | - } | ||
190 | - } | ||
191 | - | ||
192 | - vu_deinit(&server->vu_dev); | ||
193 | - | ||
194 | - /* vu_deinit() should have called remove_watch() */ | ||
195 | - assert(QTAILQ_EMPTY(&server->vu_fd_watches)); | ||
196 | - | ||
197 | - object_unref(OBJECT(sioc)); | ||
198 | - object_unref(OBJECT(server->ioc)); | ||
199 | -} | ||
200 | - | ||
201 | static void panic_cb(VuDev *vu_dev, const char *buf) | ||
202 | { | ||
203 | - VuServer *server = container_of(vu_dev, VuServer, vu_dev); | ||
204 | - | ||
205 | - /* avoid while loop in close_client */ | ||
206 | - server->processing_msg = false; | ||
207 | - | ||
208 | - if (buf) { | ||
209 | - error_report("vu_panic: %s", buf); | ||
210 | - } | ||
211 | - | ||
212 | - if (server->sioc) { | ||
213 | - close_client(server); | ||
214 | - } | ||
215 | - | ||
216 | - /* | ||
217 | - * Set the callback function for network listener so another | ||
218 | - * vhost-user client can connect to this server | ||
219 | - */ | ||
220 | - qio_net_listener_set_client_func(server->listener, | ||
221 | - vu_accept, | ||
222 | - server, | ||
223 | - NULL); | ||
224 | + error_report("vu_panic: %s", buf); | ||
225 | } | ||
226 | |||
227 | static bool coroutine_fn | ||
228 | @@ -XXX,XX +XXX,XX @@ fail: | ||
229 | return false; | ||
230 | } | ||
231 | |||
232 | - | ||
233 | -static void vu_client_start(VuServer *server); | ||
234 | static coroutine_fn void vu_client_trip(void *opaque) | ||
235 | { | ||
236 | VuServer *server = opaque; | ||
237 | + VuDev *vu_dev = &server->vu_dev; | ||
238 | |||
239 | - while (!server->aio_context_changed && server->sioc) { | ||
240 | - server->processing_msg = true; | ||
241 | - vu_dispatch(&server->vu_dev); | ||
242 | - server->processing_msg = false; | ||
243 | + while (!vu_dev->broken && vu_dispatch(vu_dev)) { | ||
244 | + /* Keep running */ | ||
245 | } | ||
246 | |||
247 | - if (server->aio_context_changed && server->sioc) { | ||
248 | - server->aio_context_changed = false; | ||
249 | - vu_client_start(server); | ||
250 | - } | ||
251 | -} | ||
252 | + vu_deinit(vu_dev); | ||
253 | + | ||
254 | + /* vu_deinit() should have called remove_watch() */ | ||
255 | + assert(QTAILQ_EMPTY(&server->vu_fd_watches)); | ||
256 | + | ||
257 | + object_unref(OBJECT(server->sioc)); | ||
258 | + server->sioc = NULL; | ||
259 | |||
260 | -static void vu_client_start(VuServer *server) | ||
261 | -{ | ||
262 | - server->co_trip = qemu_coroutine_create(vu_client_trip, server); | ||
263 | - aio_co_enter(server->ctx, server->co_trip); | ||
264 | + object_unref(OBJECT(server->ioc)); | ||
265 | + server->ioc = NULL; | ||
266 | + | ||
267 | + server->co_trip = NULL; | ||
268 | + if (server->restart_listener_bh) { | ||
269 | + qemu_bh_schedule(server->restart_listener_bh); | ||
270 | + } | ||
271 | + aio_wait_kick(); | ||
272 | } | ||
273 | |||
274 | /* | ||
275 | @@ -XXX,XX +XXX,XX @@ static void vu_client_start(VuServer *server) | ||
276 | static void kick_handler(void *opaque) | ||
277 | { | ||
278 | VuFdWatch *vu_fd_watch = opaque; | ||
279 | - vu_fd_watch->processing = true; | ||
280 | - vu_fd_watch->cb(vu_fd_watch->vu_dev, 0, vu_fd_watch->pvt); | ||
281 | - vu_fd_watch->processing = false; | ||
282 | + VuDev *vu_dev = vu_fd_watch->vu_dev; | ||
283 | + | ||
284 | + vu_fd_watch->cb(vu_dev, 0, vu_fd_watch->pvt); | ||
285 | + | ||
286 | + /* Stop vu_client_trip() if an error occurred in vu_fd_watch->cb() */ | ||
287 | + if (vu_dev->broken) { | ||
288 | + VuServer *server = container_of(vu_dev, VuServer, vu_dev); | ||
289 | + | ||
290 | + qio_channel_shutdown(server->ioc, QIO_CHANNEL_SHUTDOWN_BOTH, NULL); | ||
291 | + } | ||
292 | } | ||
293 | |||
294 | - | ||
295 | static VuFdWatch *find_vu_fd_watch(VuServer *server, int fd) | ||
296 | { | ||
297 | |||
298 | @@ -XXX,XX +XXX,XX @@ static void vu_accept(QIONetListener *listener, QIOChannelSocket *sioc, | ||
299 | qio_channel_set_name(QIO_CHANNEL(sioc), "vhost-user client"); | ||
300 | server->ioc = QIO_CHANNEL(sioc); | ||
301 | object_ref(OBJECT(server->ioc)); | ||
302 | - qio_channel_attach_aio_context(server->ioc, server->ctx); | ||
303 | + | ||
304 | + /* TODO vu_message_write() spins if non-blocking! */ | ||
305 | qio_channel_set_blocking(server->ioc, false, NULL); | ||
306 | - vu_client_start(server); | ||
307 | + | ||
308 | + server->co_trip = qemu_coroutine_create(vu_client_trip, server); | ||
309 | + | ||
310 | + aio_context_acquire(server->ctx); | ||
311 | + vhost_user_server_attach_aio_context(server, server->ctx); | ||
312 | + aio_context_release(server->ctx); | ||
313 | } | ||
314 | |||
315 | - | ||
316 | void vhost_user_server_stop(VuServer *server) | ||
317 | { | ||
318 | + aio_context_acquire(server->ctx); | ||
319 | + | ||
320 | + qemu_bh_delete(server->restart_listener_bh); | ||
321 | + server->restart_listener_bh = NULL; | ||
322 | + | ||
323 | if (server->sioc) { | ||
324 | - close_client(server); | ||
325 | + VuFdWatch *vu_fd_watch; | ||
326 | + | ||
327 | + QTAILQ_FOREACH(vu_fd_watch, &server->vu_fd_watches, next) { | ||
328 | + aio_set_fd_handler(server->ctx, vu_fd_watch->fd, true, | ||
329 | + NULL, NULL, NULL, vu_fd_watch); | ||
330 | + } | ||
331 | + | ||
332 | + qio_channel_shutdown(server->ioc, QIO_CHANNEL_SHUTDOWN_BOTH, NULL); | ||
333 | + | ||
334 | + AIO_WAIT_WHILE(server->ctx, server->co_trip); | ||
335 | } | ||
336 | |||
337 | + aio_context_release(server->ctx); | ||
338 | + | ||
339 | if (server->listener) { | ||
340 | qio_net_listener_disconnect(server->listener); | ||
341 | object_unref(OBJECT(server->listener)); | ||
342 | } | ||
343 | +} | ||
344 | + | ||
345 | +/* | ||
346 | + * Allow the next client to connect to the server. Called from a BH in the main | ||
347 | + * loop. | ||
348 | + */ | ||
349 | +static void restart_listener_bh(void *opaque) | ||
350 | +{ | ||
351 | + VuServer *server = opaque; | ||
352 | |||
353 | + qio_net_listener_set_client_func(server->listener, vu_accept, server, | ||
354 | + NULL); | ||
355 | } | ||
356 | |||
357 | -void vhost_user_server_set_aio_context(VuServer *server, AioContext *ctx) | ||
358 | +/* Called with ctx acquired */ | ||
359 | +void vhost_user_server_attach_aio_context(VuServer *server, AioContext *ctx) | ||
360 | { | ||
361 | - VuFdWatch *vu_fd_watch, *next; | ||
362 | - void *opaque = NULL; | ||
363 | - IOHandler *io_read = NULL; | ||
364 | - bool attach; | ||
365 | + VuFdWatch *vu_fd_watch; | ||
366 | |||
367 | - server->ctx = ctx ? ctx : qemu_get_aio_context(); | ||
368 | + server->ctx = ctx; | ||
369 | |||
370 | if (!server->sioc) { | ||
371 | - /* not yet serving any client*/ | ||
372 | return; | ||
373 | } | ||
374 | |||
375 | - if (ctx) { | ||
376 | - qio_channel_attach_aio_context(server->ioc, ctx); | ||
377 | - server->aio_context_changed = true; | ||
378 | - io_read = kick_handler; | ||
379 | - attach = true; | ||
380 | - } else { | ||
381 | + qio_channel_attach_aio_context(server->ioc, ctx); | ||
382 | + | ||
383 | + QTAILQ_FOREACH(vu_fd_watch, &server->vu_fd_watches, next) { | ||
384 | + aio_set_fd_handler(ctx, vu_fd_watch->fd, true, kick_handler, NULL, | ||
385 | + NULL, vu_fd_watch); | ||
386 | + } | ||
387 | + | ||
388 | + aio_co_schedule(ctx, server->co_trip); | ||
389 | +} | ||
390 | + | ||
391 | +/* Called with server->ctx acquired */ | ||
392 | +void vhost_user_server_detach_aio_context(VuServer *server) | ||
393 | +{ | ||
394 | + if (server->sioc) { | ||
395 | + VuFdWatch *vu_fd_watch; | ||
396 | + | ||
397 | + QTAILQ_FOREACH(vu_fd_watch, &server->vu_fd_watches, next) { | ||
398 | + aio_set_fd_handler(server->ctx, vu_fd_watch->fd, true, | ||
399 | + NULL, NULL, NULL, vu_fd_watch); | ||
400 | + } | ||
401 | + | ||
402 | qio_channel_detach_aio_context(server->ioc); | ||
403 | - /* server->ioc->ctx keeps the old AioConext */ | ||
404 | - ctx = server->ioc->ctx; | ||
405 | - attach = false; | ||
406 | } | ||
407 | |||
408 | - QTAILQ_FOREACH_SAFE(vu_fd_watch, &server->vu_fd_watches, next, next) { | ||
409 | - if (vu_fd_watch->cb) { | ||
410 | - opaque = attach ? vu_fd_watch : NULL; | ||
411 | - aio_set_fd_handler(ctx, vu_fd_watch->fd, true, | ||
412 | - io_read, NULL, NULL, | ||
413 | - opaque); | ||
414 | - } | ||
415 | - } | ||
416 | + server->ctx = NULL; | ||
417 | } | ||
418 | |||
419 | - | ||
420 | bool vhost_user_server_start(VuServer *server, | ||
421 | SocketAddress *socket_addr, | ||
422 | AioContext *ctx, | ||
423 | @@ -XXX,XX +XXX,XX @@ bool vhost_user_server_start(VuServer *server, | ||
424 | const VuDevIface *vu_iface, | ||
425 | Error **errp) | ||
426 | { | ||
427 | + QEMUBH *bh; | ||
428 | QIONetListener *listener = qio_net_listener_new(); | ||
429 | if (qio_net_listener_open_sync(listener, socket_addr, 1, | ||
430 | errp) < 0) { | ||
431 | @@ -XXX,XX +XXX,XX @@ bool vhost_user_server_start(VuServer *server, | ||
432 | return false; | ||
433 | } | ||
434 | |||
435 | + bh = qemu_bh_new(restart_listener_bh, server); | ||
436 | + | ||
437 | /* zero out unspecified fields */ | ||
438 | *server = (VuServer) { | ||
439 | .listener = listener, | ||
440 | + .restart_listener_bh = bh, | ||
441 | .vu_iface = vu_iface, | ||
442 | .max_queues = max_queues, | ||
443 | .ctx = ctx, | ||
444 | -- | ||
445 | 2.26.2 | ||
446 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | Propagate the flush return value since errors are possible. | ||
1 | 2 | ||
3 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
4 | Message-id: 20200924151549.913737-11-stefanha@redhat.com | ||
5 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
6 | --- | ||
7 | block/export/vhost-user-blk-server.c | 11 +++++++---- | ||
8 | 1 file changed, 7 insertions(+), 4 deletions(-) | ||
9 | |||
10 | diff --git a/block/export/vhost-user-blk-server.c b/block/export/vhost-user-blk-server.c | ||
11 | index XXXXXXX..XXXXXXX 100644 | ||
12 | --- a/block/export/vhost-user-blk-server.c | ||
13 | +++ b/block/export/vhost-user-blk-server.c | ||
14 | @@ -XXX,XX +XXX,XX @@ vu_block_discard_write_zeroes(VuBlockReq *req, struct iovec *iov, | ||
15 | return -EINVAL; | ||
16 | } | ||
17 | |||
18 | -static void coroutine_fn vu_block_flush(VuBlockReq *req) | ||
19 | +static int coroutine_fn vu_block_flush(VuBlockReq *req) | ||
20 | { | ||
21 | VuBlockDev *vdev_blk = get_vu_block_device_by_server(req->server); | ||
22 | BlockBackend *backend = vdev_blk->backend; | ||
23 | - blk_co_flush(backend); | ||
24 | + return blk_co_flush(backend); | ||
25 | } | ||
26 | |||
27 | static void coroutine_fn vu_block_virtio_process_req(void *opaque) | ||
28 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn vu_block_virtio_process_req(void *opaque) | ||
29 | break; | ||
30 | } | ||
31 | case VIRTIO_BLK_T_FLUSH: | ||
32 | - vu_block_flush(req); | ||
33 | - req->in->status = VIRTIO_BLK_S_OK; | ||
34 | + if (vu_block_flush(req) == 0) { | ||
35 | + req->in->status = VIRTIO_BLK_S_OK; | ||
36 | + } else { | ||
37 | + req->in->status = VIRTIO_BLK_S_IOERR; | ||
38 | + } | ||
39 | break; | ||
40 | case VIRTIO_BLK_T_GET_ID: { | ||
41 | size_t size = MIN(iov_size(&elem->in_sg[0], in_num), | ||
42 | -- | ||
43 | 2.26.2 | ||
44 | diff view generated by jsdifflib |
1 | From: Su Hang <suhang16@mails.ucas.ac.cn> | 1 | Use the new QAPI block exports API instead of defining our own QOM |
---|---|---|---|
2 | 2 | objects. | |
3 | only remove brackets that wrap `return` statements' content. | 3 | |
4 | 4 | This is a large change because the lifecycle of VuBlockDev needs to | |
5 | use `perl -pi -e "s/return \((.*?)\);/return \1;/g" util/uri.c` | 5 | follow BlockExportDriver. QOM properties are replaced by QAPI options |
6 | to remove pattern like this: "return (1);" | 6 | objects. |
7 | 7 | ||
8 | Signed-off-by: Su Hang <suhang16@mails.ucas.ac.cn> | 8 | VuBlockDev is renamed VuBlkExport and contains a BlockExport field. |
9 | Reviewed-by: Thomas Huth <thuth@redhat.com> | 9 | Several fields can be dropped since BlockExport already has equivalents. |
10 | Message-id: 1519533358-13759-3-git-send-email-suhang16@mails.ucas.ac.cn | 10 | |
11 | The file names and meson build integration will be adjusted in a future | ||
12 | patch. libvhost-user should probably be built as a static library that | ||
13 | is linked into QEMU instead of as a .c file that results in duplicate | ||
14 | compilation. | ||
15 | |||
16 | The new command-line syntax is: | ||
17 | |||
18 | $ qemu-storage-daemon \ | ||
19 | --blockdev file,node-name=drive0,filename=test.img \ | ||
20 | --export vhost-user-blk,node-name=drive0,id=export0,unix-socket=/tmp/vhost-user-blk.sock | ||
21 | |||
22 | Note that unix-socket is optional because we may wish to accept chardevs | ||
23 | too in the future. | ||
24 | |||
25 | Markus noted that supported address families are not explicit in the | ||
26 | QAPI schema. It is unlikely that support for more address families will | ||
27 | be added since file descriptor passing is required and few address | ||
28 | families support it. If a new address family needs to be added, then the | ||
29 | QAPI 'features' syntax can be used to advertize them. | ||
30 | |||
31 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
32 | Acked-by: Markus Armbruster <armbru@redhat.com> | ||
33 | Message-id: 20200924151549.913737-12-stefanha@redhat.com | ||
34 | [Skip test on big-endian host architectures because this device doesn't | ||
35 | support them yet (as already mentioned in a code comment). | ||
36 | --Stefan] | ||
11 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 37 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
12 | --- | 38 | --- |
13 | util/uri.c | 160 ++++++++++++++++++++++++++++++------------------------------- | 39 | qapi/block-export.json | 21 +- |
14 | 1 file changed, 80 insertions(+), 80 deletions(-) | 40 | block/export/vhost-user-blk-server.h | 23 +- |
15 | 41 | block/export/export.c | 6 + | |
16 | diff --git a/util/uri.c b/util/uri.c | 42 | block/export/vhost-user-blk-server.c | 452 +++++++-------------------- |
43 | util/vhost-user-server.c | 10 +- | ||
44 | block/export/meson.build | 1 + | ||
45 | block/meson.build | 1 - | ||
46 | 7 files changed, 156 insertions(+), 358 deletions(-) | ||
47 | |||
48 | diff --git a/qapi/block-export.json b/qapi/block-export.json | ||
17 | index XXXXXXX..XXXXXXX 100644 | 49 | index XXXXXXX..XXXXXXX 100644 |
18 | --- a/util/uri.c | 50 | --- a/qapi/block-export.json |
19 | +++ b/util/uri.c | 51 | +++ b/qapi/block-export.json |
20 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_scheme(URI *uri, const char **str) | 52 | @@ -XXX,XX +XXX,XX @@ |
21 | const char *cur; | 53 | 'data': { '*name': 'str', '*description': 'str', |
22 | 54 | '*bitmap': 'str' } } | |
23 | if (str == NULL) | 55 | |
24 | - return (-1); | 56 | +## |
25 | + return -1; | 57 | +# @BlockExportOptionsVhostUserBlk: |
26 | 58 | +# | |
27 | cur = *str; | 59 | +# A vhost-user-blk block export. |
28 | if (!ISA_ALPHA(cur)) | 60 | +# |
29 | - return (2); | 61 | +# @addr: The vhost-user socket on which to listen. Both 'unix' and 'fd' |
30 | + return 2; | 62 | +# SocketAddress types are supported. Passed fds must be UNIX domain |
31 | cur++; | 63 | +# sockets. |
32 | while (ISA_ALPHA(cur) || ISA_DIGIT(cur) || (*cur == '+') || (*cur == '-') || | 64 | +# @logical-block-size: Logical block size in bytes. Defaults to 512 bytes. |
33 | (*cur == '.')) | 65 | +# |
34 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_scheme(URI *uri, const char **str) | 66 | +# Since: 5.2 |
35 | uri->scheme = g_strndup(*str, cur - *str); | 67 | +## |
68 | +{ 'struct': 'BlockExportOptionsVhostUserBlk', | ||
69 | + 'data': { 'addr': 'SocketAddress', '*logical-block-size': 'size' } } | ||
70 | + | ||
71 | ## | ||
72 | # @NbdServerAddOptions: | ||
73 | # | ||
74 | @@ -XXX,XX +XXX,XX @@ | ||
75 | # An enumeration of block export types | ||
76 | # | ||
77 | # @nbd: NBD export | ||
78 | +# @vhost-user-blk: vhost-user-blk export (since 5.2) | ||
79 | # | ||
80 | # Since: 4.2 | ||
81 | ## | ||
82 | { 'enum': 'BlockExportType', | ||
83 | - 'data': [ 'nbd' ] } | ||
84 | + 'data': [ 'nbd', 'vhost-user-blk' ] } | ||
85 | |||
86 | ## | ||
87 | # @BlockExportOptions: | ||
88 | @@ -XXX,XX +XXX,XX @@ | ||
89 | '*writethrough': 'bool' }, | ||
90 | 'discriminator': 'type', | ||
91 | 'data': { | ||
92 | - 'nbd': 'BlockExportOptionsNbd' | ||
93 | + 'nbd': 'BlockExportOptionsNbd', | ||
94 | + 'vhost-user-blk': 'BlockExportOptionsVhostUserBlk' | ||
95 | } } | ||
96 | |||
97 | ## | ||
98 | diff --git a/block/export/vhost-user-blk-server.h b/block/export/vhost-user-blk-server.h | ||
99 | index XXXXXXX..XXXXXXX 100644 | ||
100 | --- a/block/export/vhost-user-blk-server.h | ||
101 | +++ b/block/export/vhost-user-blk-server.h | ||
102 | @@ -XXX,XX +XXX,XX @@ | ||
103 | |||
104 | #ifndef VHOST_USER_BLK_SERVER_H | ||
105 | #define VHOST_USER_BLK_SERVER_H | ||
106 | -#include "util/vhost-user-server.h" | ||
107 | |||
108 | -typedef struct VuBlockDev VuBlockDev; | ||
109 | -#define TYPE_VHOST_USER_BLK_SERVER "vhost-user-blk-server" | ||
110 | -#define VHOST_USER_BLK_SERVER(obj) \ | ||
111 | - OBJECT_CHECK(VuBlockDev, obj, TYPE_VHOST_USER_BLK_SERVER) | ||
112 | +#include "block/export.h" | ||
113 | |||
114 | -/* vhost user block device */ | ||
115 | -struct VuBlockDev { | ||
116 | - Object parent_obj; | ||
117 | - char *node_name; | ||
118 | - SocketAddress *addr; | ||
119 | - AioContext *ctx; | ||
120 | - VuServer vu_server; | ||
121 | - bool running; | ||
122 | - uint32_t blk_size; | ||
123 | - BlockBackend *backend; | ||
124 | - QIOChannelSocket *sioc; | ||
125 | - QTAILQ_ENTRY(VuBlockDev) next; | ||
126 | - struct virtio_blk_config blkcfg; | ||
127 | - bool writable; | ||
128 | -}; | ||
129 | +/* For block/export/export.c */ | ||
130 | +extern const BlockExportDriver blk_exp_vhost_user_blk; | ||
131 | |||
132 | #endif /* VHOST_USER_BLK_SERVER_H */ | ||
133 | diff --git a/block/export/export.c b/block/export/export.c | ||
134 | index XXXXXXX..XXXXXXX 100644 | ||
135 | --- a/block/export/export.c | ||
136 | +++ b/block/export/export.c | ||
137 | @@ -XXX,XX +XXX,XX @@ | ||
138 | #include "sysemu/block-backend.h" | ||
139 | #include "block/export.h" | ||
140 | #include "block/nbd.h" | ||
141 | +#if CONFIG_LINUX | ||
142 | +#include "block/export/vhost-user-blk-server.h" | ||
143 | +#endif | ||
144 | #include "qapi/error.h" | ||
145 | #include "qapi/qapi-commands-block-export.h" | ||
146 | #include "qapi/qapi-events-block-export.h" | ||
147 | @@ -XXX,XX +XXX,XX @@ | ||
148 | |||
149 | static const BlockExportDriver *blk_exp_drivers[] = { | ||
150 | &blk_exp_nbd, | ||
151 | +#if CONFIG_LINUX | ||
152 | + &blk_exp_vhost_user_blk, | ||
153 | +#endif | ||
154 | }; | ||
155 | |||
156 | /* Only accessed from the main thread */ | ||
157 | diff --git a/block/export/vhost-user-blk-server.c b/block/export/vhost-user-blk-server.c | ||
158 | index XXXXXXX..XXXXXXX 100644 | ||
159 | --- a/block/export/vhost-user-blk-server.c | ||
160 | +++ b/block/export/vhost-user-blk-server.c | ||
161 | @@ -XXX,XX +XXX,XX @@ | ||
162 | */ | ||
163 | #include "qemu/osdep.h" | ||
164 | #include "block/block.h" | ||
165 | +#include "contrib/libvhost-user/libvhost-user.h" | ||
166 | +#include "standard-headers/linux/virtio_blk.h" | ||
167 | +#include "util/vhost-user-server.h" | ||
168 | #include "vhost-user-blk-server.h" | ||
169 | #include "qapi/error.h" | ||
170 | #include "qom/object_interfaces.h" | ||
171 | @@ -XXX,XX +XXX,XX @@ struct virtio_blk_inhdr { | ||
172 | unsigned char status; | ||
173 | }; | ||
174 | |||
175 | -typedef struct VuBlockReq { | ||
176 | +typedef struct VuBlkReq { | ||
177 | VuVirtqElement elem; | ||
178 | int64_t sector_num; | ||
179 | size_t size; | ||
180 | @@ -XXX,XX +XXX,XX @@ typedef struct VuBlockReq { | ||
181 | struct virtio_blk_outhdr out; | ||
182 | VuServer *server; | ||
183 | struct VuVirtq *vq; | ||
184 | -} VuBlockReq; | ||
185 | +} VuBlkReq; | ||
186 | |||
187 | -static void vu_block_req_complete(VuBlockReq *req) | ||
188 | +/* vhost user block device */ | ||
189 | +typedef struct { | ||
190 | + BlockExport export; | ||
191 | + VuServer vu_server; | ||
192 | + uint32_t blk_size; | ||
193 | + QIOChannelSocket *sioc; | ||
194 | + struct virtio_blk_config blkcfg; | ||
195 | + bool writable; | ||
196 | +} VuBlkExport; | ||
197 | + | ||
198 | +static void vu_blk_req_complete(VuBlkReq *req) | ||
199 | { | ||
200 | VuDev *vu_dev = &req->server->vu_dev; | ||
201 | |||
202 | @@ -XXX,XX +XXX,XX @@ static void vu_block_req_complete(VuBlockReq *req) | ||
203 | free(req); | ||
204 | } | ||
205 | |||
206 | -static VuBlockDev *get_vu_block_device_by_server(VuServer *server) | ||
207 | -{ | ||
208 | - return container_of(server, VuBlockDev, vu_server); | ||
209 | -} | ||
210 | - | ||
211 | static int coroutine_fn | ||
212 | -vu_block_discard_write_zeroes(VuBlockReq *req, struct iovec *iov, | ||
213 | - uint32_t iovcnt, uint32_t type) | ||
214 | +vu_blk_discard_write_zeroes(BlockBackend *blk, struct iovec *iov, | ||
215 | + uint32_t iovcnt, uint32_t type) | ||
216 | { | ||
217 | struct virtio_blk_discard_write_zeroes desc; | ||
218 | ssize_t size = iov_to_buf(iov, iovcnt, 0, &desc, sizeof(desc)); | ||
219 | @@ -XXX,XX +XXX,XX @@ vu_block_discard_write_zeroes(VuBlockReq *req, struct iovec *iov, | ||
220 | return -EINVAL; | ||
36 | } | 221 | } |
37 | *str = cur; | 222 | |
38 | - return (0); | 223 | - VuBlockDev *vdev_blk = get_vu_block_device_by_server(req->server); |
39 | + return 0; | 224 | uint64_t range[2] = { le64_to_cpu(desc.sector) << 9, |
40 | } | 225 | le32_to_cpu(desc.num_sectors) << 9 }; |
41 | 226 | if (type == VIRTIO_BLK_T_DISCARD) { | |
42 | /** | 227 | - if (blk_co_pdiscard(vdev_blk->backend, range[0], range[1]) == 0) { |
43 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_fragment(URI *uri, const char **str) | 228 | + if (blk_co_pdiscard(blk, range[0], range[1]) == 0) { |
44 | const char *cur; | 229 | return 0; |
45 | |||
46 | if (str == NULL) | ||
47 | - return (-1); | ||
48 | + return -1; | ||
49 | |||
50 | cur = *str; | ||
51 | |||
52 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_fragment(URI *uri, const char **str) | ||
53 | uri->fragment = uri_string_unescape(*str, cur - *str, NULL); | ||
54 | } | ||
55 | *str = cur; | ||
56 | - return (0); | ||
57 | + return 0; | ||
58 | } | ||
59 | |||
60 | /** | ||
61 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_query(URI *uri, const char **str) | ||
62 | const char *cur; | ||
63 | |||
64 | if (str == NULL) | ||
65 | - return (-1); | ||
66 | + return -1; | ||
67 | |||
68 | cur = *str; | ||
69 | |||
70 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_query(URI *uri, const char **str) | ||
71 | uri->query = g_strndup(*str, cur - *str); | ||
72 | } | ||
73 | *str = cur; | ||
74 | - return (0); | ||
75 | + return 0; | ||
76 | } | ||
77 | |||
78 | /** | ||
79 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_user_info(URI *uri, const char **str) | ||
80 | uri->user = uri_string_unescape(*str, cur - *str, NULL); | ||
81 | } | 230 | } |
82 | *str = cur; | 231 | } else if (type == VIRTIO_BLK_T_WRITE_ZEROES) { |
83 | - return (0); | 232 | - if (blk_co_pwrite_zeroes(vdev_blk->backend, |
84 | + return 0; | 233 | - range[0], range[1], 0) == 0) { |
85 | } | 234 | + if (blk_co_pwrite_zeroes(blk, range[0], range[1], 0) == 0) { |
86 | - return (1); | 235 | return 0; |
87 | + return 1; | ||
88 | } | ||
89 | |||
90 | /** | ||
91 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_dec_octet(const char **str) | ||
92 | const char *cur = *str; | ||
93 | |||
94 | if (!(ISA_DIGIT(cur))) | ||
95 | - return (1); | ||
96 | + return 1; | ||
97 | if (!ISA_DIGIT(cur + 1)) | ||
98 | cur++; | ||
99 | else if ((*cur != '0') && (ISA_DIGIT(cur + 1)) && (!ISA_DIGIT(cur + 2))) | ||
100 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_dec_octet(const char **str) | ||
101 | (*(cur + 1) <= '5')) | ||
102 | cur += 3; | ||
103 | else | ||
104 | - return (1); | ||
105 | + return 1; | ||
106 | *str = cur; | ||
107 | - return (0); | ||
108 | + return 0; | ||
109 | } | ||
110 | /** | ||
111 | * rfc3986_parse_host: | ||
112 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_host(URI *uri, const char **str) | ||
113 | while ((*cur != ']') && (*cur != 0)) | ||
114 | cur++; | ||
115 | if (*cur != ']') | ||
116 | - return (1); | ||
117 | + return 1; | ||
118 | cur++; | ||
119 | goto found; | ||
120 | } | ||
121 | @@ -XXX,XX +XXX,XX @@ found: | ||
122 | uri->server = NULL; | ||
123 | } | ||
124 | *str = cur; | ||
125 | - return (0); | ||
126 | + return 0; | ||
127 | } | ||
128 | |||
129 | /** | ||
130 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_authority(URI *uri, const char **str) | ||
131 | cur++; | ||
132 | ret = rfc3986_parse_host(uri, &cur); | ||
133 | if (ret != 0) | ||
134 | - return (ret); | ||
135 | + return ret; | ||
136 | if (*cur == ':') { | ||
137 | cur++; | ||
138 | ret = rfc3986_parse_port(uri, &cur); | ||
139 | if (ret != 0) | ||
140 | - return (ret); | ||
141 | + return ret; | ||
142 | } | ||
143 | *str = cur; | ||
144 | - return (0); | ||
145 | + return 0; | ||
146 | } | ||
147 | |||
148 | /** | ||
149 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_segment(const char **str, char forbid, int empty) | ||
150 | cur = *str; | ||
151 | if (!ISA_PCHAR(cur)) { | ||
152 | if (empty) | ||
153 | - return (0); | ||
154 | - return (1); | ||
155 | + return 0; | ||
156 | + return 1; | ||
157 | } | ||
158 | while (ISA_PCHAR(cur) && (*cur != forbid)) | ||
159 | NEXT(cur); | ||
160 | *str = cur; | ||
161 | - return (0); | ||
162 | + return 0; | ||
163 | } | ||
164 | |||
165 | /** | ||
166 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
167 | cur++; | ||
168 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
169 | if (ret != 0) | ||
170 | - return (ret); | ||
171 | + return ret; | ||
172 | } | ||
173 | if (uri != NULL) { | ||
174 | g_free(uri->path); | ||
175 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
176 | } | 236 | } |
177 | } | 237 | } |
178 | *str = cur; | 238 | @@ -XXX,XX +XXX,XX @@ vu_block_discard_write_zeroes(VuBlockReq *req, struct iovec *iov, |
179 | - return (0); | 239 | return -EINVAL; |
240 | } | ||
241 | |||
242 | -static int coroutine_fn vu_block_flush(VuBlockReq *req) | ||
243 | +static void coroutine_fn vu_blk_virtio_process_req(void *opaque) | ||
244 | { | ||
245 | - VuBlockDev *vdev_blk = get_vu_block_device_by_server(req->server); | ||
246 | - BlockBackend *backend = vdev_blk->backend; | ||
247 | - return blk_co_flush(backend); | ||
248 | -} | ||
249 | - | ||
250 | -static void coroutine_fn vu_block_virtio_process_req(void *opaque) | ||
251 | -{ | ||
252 | - VuBlockReq *req = opaque; | ||
253 | + VuBlkReq *req = opaque; | ||
254 | VuServer *server = req->server; | ||
255 | VuVirtqElement *elem = &req->elem; | ||
256 | uint32_t type; | ||
257 | |||
258 | - VuBlockDev *vdev_blk = get_vu_block_device_by_server(server); | ||
259 | - BlockBackend *backend = vdev_blk->backend; | ||
260 | + VuBlkExport *vexp = container_of(server, VuBlkExport, vu_server); | ||
261 | + BlockBackend *blk = vexp->export.blk; | ||
262 | |||
263 | struct iovec *in_iov = elem->in_sg; | ||
264 | struct iovec *out_iov = elem->out_sg; | ||
265 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn vu_block_virtio_process_req(void *opaque) | ||
266 | bool is_write = type & VIRTIO_BLK_T_OUT; | ||
267 | req->sector_num = le64_to_cpu(req->out.sector); | ||
268 | |||
269 | - int64_t offset = req->sector_num * vdev_blk->blk_size; | ||
270 | + if (is_write && !vexp->writable) { | ||
271 | + req->in->status = VIRTIO_BLK_S_IOERR; | ||
272 | + break; | ||
273 | + } | ||
274 | + | ||
275 | + int64_t offset = req->sector_num * vexp->blk_size; | ||
276 | QEMUIOVector qiov; | ||
277 | if (is_write) { | ||
278 | qemu_iovec_init_external(&qiov, out_iov, out_num); | ||
279 | - ret = blk_co_pwritev(backend, offset, qiov.size, | ||
280 | - &qiov, 0); | ||
281 | + ret = blk_co_pwritev(blk, offset, qiov.size, &qiov, 0); | ||
282 | } else { | ||
283 | qemu_iovec_init_external(&qiov, in_iov, in_num); | ||
284 | - ret = blk_co_preadv(backend, offset, qiov.size, | ||
285 | - &qiov, 0); | ||
286 | + ret = blk_co_preadv(blk, offset, qiov.size, &qiov, 0); | ||
287 | } | ||
288 | if (ret >= 0) { | ||
289 | req->in->status = VIRTIO_BLK_S_OK; | ||
290 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn vu_block_virtio_process_req(void *opaque) | ||
291 | break; | ||
292 | } | ||
293 | case VIRTIO_BLK_T_FLUSH: | ||
294 | - if (vu_block_flush(req) == 0) { | ||
295 | + if (blk_co_flush(blk) == 0) { | ||
296 | req->in->status = VIRTIO_BLK_S_OK; | ||
297 | } else { | ||
298 | req->in->status = VIRTIO_BLK_S_IOERR; | ||
299 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn vu_block_virtio_process_req(void *opaque) | ||
300 | case VIRTIO_BLK_T_DISCARD: | ||
301 | case VIRTIO_BLK_T_WRITE_ZEROES: { | ||
302 | int rc; | ||
303 | - rc = vu_block_discard_write_zeroes(req, &elem->out_sg[1], | ||
304 | - out_num, type); | ||
305 | + | ||
306 | + if (!vexp->writable) { | ||
307 | + req->in->status = VIRTIO_BLK_S_IOERR; | ||
308 | + break; | ||
309 | + } | ||
310 | + | ||
311 | + rc = vu_blk_discard_write_zeroes(blk, &elem->out_sg[1], out_num, type); | ||
312 | if (rc == 0) { | ||
313 | req->in->status = VIRTIO_BLK_S_OK; | ||
314 | } else { | ||
315 | @@ -XXX,XX +XXX,XX @@ static void coroutine_fn vu_block_virtio_process_req(void *opaque) | ||
316 | break; | ||
317 | } | ||
318 | |||
319 | - vu_block_req_complete(req); | ||
320 | + vu_blk_req_complete(req); | ||
321 | return; | ||
322 | |||
323 | err: | ||
324 | - free(elem); | ||
325 | + free(req); | ||
326 | } | ||
327 | |||
328 | -static void vu_block_process_vq(VuDev *vu_dev, int idx) | ||
329 | +static void vu_blk_process_vq(VuDev *vu_dev, int idx) | ||
330 | { | ||
331 | VuServer *server = container_of(vu_dev, VuServer, vu_dev); | ||
332 | VuVirtq *vq = vu_get_queue(vu_dev, idx); | ||
333 | |||
334 | while (1) { | ||
335 | - VuBlockReq *req; | ||
336 | + VuBlkReq *req; | ||
337 | |||
338 | - req = vu_queue_pop(vu_dev, vq, sizeof(VuBlockReq)); | ||
339 | + req = vu_queue_pop(vu_dev, vq, sizeof(VuBlkReq)); | ||
340 | if (!req) { | ||
341 | break; | ||
342 | } | ||
343 | @@ -XXX,XX +XXX,XX @@ static void vu_block_process_vq(VuDev *vu_dev, int idx) | ||
344 | req->vq = vq; | ||
345 | |||
346 | Coroutine *co = | ||
347 | - qemu_coroutine_create(vu_block_virtio_process_req, req); | ||
348 | + qemu_coroutine_create(vu_blk_virtio_process_req, req); | ||
349 | qemu_coroutine_enter(co); | ||
350 | } | ||
351 | } | ||
352 | |||
353 | -static void vu_block_queue_set_started(VuDev *vu_dev, int idx, bool started) | ||
354 | +static void vu_blk_queue_set_started(VuDev *vu_dev, int idx, bool started) | ||
355 | { | ||
356 | VuVirtq *vq; | ||
357 | |||
358 | assert(vu_dev); | ||
359 | |||
360 | vq = vu_get_queue(vu_dev, idx); | ||
361 | - vu_set_queue_handler(vu_dev, vq, started ? vu_block_process_vq : NULL); | ||
362 | + vu_set_queue_handler(vu_dev, vq, started ? vu_blk_process_vq : NULL); | ||
363 | } | ||
364 | |||
365 | -static uint64_t vu_block_get_features(VuDev *dev) | ||
366 | +static uint64_t vu_blk_get_features(VuDev *dev) | ||
367 | { | ||
368 | uint64_t features; | ||
369 | VuServer *server = container_of(dev, VuServer, vu_dev); | ||
370 | - VuBlockDev *vdev_blk = get_vu_block_device_by_server(server); | ||
371 | + VuBlkExport *vexp = container_of(server, VuBlkExport, vu_server); | ||
372 | features = 1ull << VIRTIO_BLK_F_SIZE_MAX | | ||
373 | 1ull << VIRTIO_BLK_F_SEG_MAX | | ||
374 | 1ull << VIRTIO_BLK_F_TOPOLOGY | | ||
375 | @@ -XXX,XX +XXX,XX @@ static uint64_t vu_block_get_features(VuDev *dev) | ||
376 | 1ull << VIRTIO_RING_F_EVENT_IDX | | ||
377 | 1ull << VHOST_USER_F_PROTOCOL_FEATURES; | ||
378 | |||
379 | - if (!vdev_blk->writable) { | ||
380 | + if (!vexp->writable) { | ||
381 | features |= 1ull << VIRTIO_BLK_F_RO; | ||
382 | } | ||
383 | |||
384 | return features; | ||
385 | } | ||
386 | |||
387 | -static uint64_t vu_block_get_protocol_features(VuDev *dev) | ||
388 | +static uint64_t vu_blk_get_protocol_features(VuDev *dev) | ||
389 | { | ||
390 | return 1ull << VHOST_USER_PROTOCOL_F_CONFIG | | ||
391 | 1ull << VHOST_USER_PROTOCOL_F_INFLIGHT_SHMFD; | ||
392 | } | ||
393 | |||
394 | static int | ||
395 | -vu_block_get_config(VuDev *vu_dev, uint8_t *config, uint32_t len) | ||
396 | +vu_blk_get_config(VuDev *vu_dev, uint8_t *config, uint32_t len) | ||
397 | { | ||
398 | + /* TODO blkcfg must be little-endian for VIRTIO 1.0 */ | ||
399 | VuServer *server = container_of(vu_dev, VuServer, vu_dev); | ||
400 | - VuBlockDev *vdev_blk = get_vu_block_device_by_server(server); | ||
401 | - memcpy(config, &vdev_blk->blkcfg, len); | ||
402 | - | ||
403 | + VuBlkExport *vexp = container_of(server, VuBlkExport, vu_server); | ||
404 | + memcpy(config, &vexp->blkcfg, len); | ||
405 | return 0; | ||
406 | } | ||
407 | |||
408 | static int | ||
409 | -vu_block_set_config(VuDev *vu_dev, const uint8_t *data, | ||
410 | +vu_blk_set_config(VuDev *vu_dev, const uint8_t *data, | ||
411 | uint32_t offset, uint32_t size, uint32_t flags) | ||
412 | { | ||
413 | VuServer *server = container_of(vu_dev, VuServer, vu_dev); | ||
414 | - VuBlockDev *vdev_blk = get_vu_block_device_by_server(server); | ||
415 | + VuBlkExport *vexp = container_of(server, VuBlkExport, vu_server); | ||
416 | uint8_t wce; | ||
417 | |||
418 | /* don't support live migration */ | ||
419 | @@ -XXX,XX +XXX,XX @@ vu_block_set_config(VuDev *vu_dev, const uint8_t *data, | ||
420 | } | ||
421 | |||
422 | wce = *data; | ||
423 | - vdev_blk->blkcfg.wce = wce; | ||
424 | - blk_set_enable_write_cache(vdev_blk->backend, wce); | ||
425 | + vexp->blkcfg.wce = wce; | ||
426 | + blk_set_enable_write_cache(vexp->export.blk, wce); | ||
427 | return 0; | ||
428 | } | ||
429 | |||
430 | @@ -XXX,XX +XXX,XX @@ vu_block_set_config(VuDev *vu_dev, const uint8_t *data, | ||
431 | * of vu_process_message. | ||
432 | * | ||
433 | */ | ||
434 | -static int vu_block_process_msg(VuDev *dev, VhostUserMsg *vmsg, int *do_reply) | ||
435 | +static int vu_blk_process_msg(VuDev *dev, VhostUserMsg *vmsg, int *do_reply) | ||
436 | { | ||
437 | if (vmsg->request == VHOST_USER_NONE) { | ||
438 | dev->panic(dev, "disconnect"); | ||
439 | @@ -XXX,XX +XXX,XX @@ static int vu_block_process_msg(VuDev *dev, VhostUserMsg *vmsg, int *do_reply) | ||
440 | return false; | ||
441 | } | ||
442 | |||
443 | -static const VuDevIface vu_block_iface = { | ||
444 | - .get_features = vu_block_get_features, | ||
445 | - .queue_set_started = vu_block_queue_set_started, | ||
446 | - .get_protocol_features = vu_block_get_protocol_features, | ||
447 | - .get_config = vu_block_get_config, | ||
448 | - .set_config = vu_block_set_config, | ||
449 | - .process_msg = vu_block_process_msg, | ||
450 | +static const VuDevIface vu_blk_iface = { | ||
451 | + .get_features = vu_blk_get_features, | ||
452 | + .queue_set_started = vu_blk_queue_set_started, | ||
453 | + .get_protocol_features = vu_blk_get_protocol_features, | ||
454 | + .get_config = vu_blk_get_config, | ||
455 | + .set_config = vu_blk_set_config, | ||
456 | + .process_msg = vu_blk_process_msg, | ||
457 | }; | ||
458 | |||
459 | static void blk_aio_attached(AioContext *ctx, void *opaque) | ||
460 | { | ||
461 | - VuBlockDev *vub_dev = opaque; | ||
462 | - vhost_user_server_attach_aio_context(&vub_dev->vu_server, ctx); | ||
463 | + VuBlkExport *vexp = opaque; | ||
464 | + vhost_user_server_attach_aio_context(&vexp->vu_server, ctx); | ||
465 | } | ||
466 | |||
467 | static void blk_aio_detach(void *opaque) | ||
468 | { | ||
469 | - VuBlockDev *vub_dev = opaque; | ||
470 | - vhost_user_server_detach_aio_context(&vub_dev->vu_server); | ||
471 | + VuBlkExport *vexp = opaque; | ||
472 | + vhost_user_server_detach_aio_context(&vexp->vu_server); | ||
473 | } | ||
474 | |||
475 | static void | ||
476 | -vu_block_initialize_config(BlockDriverState *bs, | ||
477 | +vu_blk_initialize_config(BlockDriverState *bs, | ||
478 | struct virtio_blk_config *config, uint32_t blk_size) | ||
479 | { | ||
480 | config->capacity = bdrv_getlength(bs) >> BDRV_SECTOR_BITS; | ||
481 | @@ -XXX,XX +XXX,XX @@ vu_block_initialize_config(BlockDriverState *bs, | ||
482 | config->max_write_zeroes_seg = 1; | ||
483 | } | ||
484 | |||
485 | -static VuBlockDev *vu_block_init(VuBlockDev *vu_block_device, Error **errp) | ||
486 | +static void vu_blk_exp_request_shutdown(BlockExport *exp) | ||
487 | { | ||
488 | + VuBlkExport *vexp = container_of(exp, VuBlkExport, export); | ||
489 | |||
490 | - BlockBackend *blk; | ||
491 | - Error *local_error = NULL; | ||
492 | - const char *node_name = vu_block_device->node_name; | ||
493 | - bool writable = vu_block_device->writable; | ||
494 | - uint64_t perm = BLK_PERM_CONSISTENT_READ; | ||
495 | - int ret; | ||
496 | - | ||
497 | - AioContext *ctx; | ||
498 | - | ||
499 | - BlockDriverState *bs = bdrv_lookup_bs(node_name, node_name, &local_error); | ||
500 | - | ||
501 | - if (!bs) { | ||
502 | - error_propagate(errp, local_error); | ||
503 | - return NULL; | ||
504 | - } | ||
505 | - | ||
506 | - if (bdrv_is_read_only(bs)) { | ||
507 | - writable = false; | ||
508 | - } | ||
509 | - | ||
510 | - if (writable) { | ||
511 | - perm |= BLK_PERM_WRITE; | ||
512 | - } | ||
513 | - | ||
514 | - ctx = bdrv_get_aio_context(bs); | ||
515 | - aio_context_acquire(ctx); | ||
516 | - bdrv_invalidate_cache(bs, NULL); | ||
517 | - aio_context_release(ctx); | ||
518 | - | ||
519 | - /* | ||
520 | - * Don't allow resize while the vhost user server is running, | ||
521 | - * otherwise we don't care what happens with the node. | ||
522 | - */ | ||
523 | - blk = blk_new(bdrv_get_aio_context(bs), perm, | ||
524 | - BLK_PERM_CONSISTENT_READ | BLK_PERM_WRITE_UNCHANGED | | ||
525 | - BLK_PERM_WRITE | BLK_PERM_GRAPH_MOD); | ||
526 | - ret = blk_insert_bs(blk, bs, errp); | ||
527 | - | ||
528 | - if (ret < 0) { | ||
529 | - goto fail; | ||
530 | - } | ||
531 | - | ||
532 | - blk_set_enable_write_cache(blk, false); | ||
533 | - | ||
534 | - blk_set_allow_aio_context_change(blk, true); | ||
535 | - | ||
536 | - vu_block_device->blkcfg.wce = 0; | ||
537 | - vu_block_device->backend = blk; | ||
538 | - if (!vu_block_device->blk_size) { | ||
539 | - vu_block_device->blk_size = BDRV_SECTOR_SIZE; | ||
540 | - } | ||
541 | - vu_block_device->blkcfg.blk_size = vu_block_device->blk_size; | ||
542 | - blk_set_guest_block_size(blk, vu_block_device->blk_size); | ||
543 | - vu_block_initialize_config(bs, &vu_block_device->blkcfg, | ||
544 | - vu_block_device->blk_size); | ||
545 | - return vu_block_device; | ||
546 | - | ||
547 | -fail: | ||
548 | - blk_unref(blk); | ||
549 | - return NULL; | ||
550 | -} | ||
551 | - | ||
552 | -static void vu_block_deinit(VuBlockDev *vu_block_device) | ||
553 | -{ | ||
554 | - if (vu_block_device->backend) { | ||
555 | - blk_remove_aio_context_notifier(vu_block_device->backend, blk_aio_attached, | ||
556 | - blk_aio_detach, vu_block_device); | ||
557 | - } | ||
558 | - | ||
559 | - blk_unref(vu_block_device->backend); | ||
560 | -} | ||
561 | - | ||
562 | -static void vhost_user_blk_server_stop(VuBlockDev *vu_block_device) | ||
563 | -{ | ||
564 | - vhost_user_server_stop(&vu_block_device->vu_server); | ||
565 | - vu_block_deinit(vu_block_device); | ||
566 | -} | ||
567 | - | ||
568 | -static void vhost_user_blk_server_start(VuBlockDev *vu_block_device, | ||
569 | - Error **errp) | ||
570 | -{ | ||
571 | - AioContext *ctx; | ||
572 | - SocketAddress *addr = vu_block_device->addr; | ||
573 | - | ||
574 | - if (!vu_block_init(vu_block_device, errp)) { | ||
575 | - return; | ||
576 | - } | ||
577 | - | ||
578 | - ctx = bdrv_get_aio_context(blk_bs(vu_block_device->backend)); | ||
579 | - | ||
580 | - if (!vhost_user_server_start(&vu_block_device->vu_server, addr, ctx, | ||
581 | - VHOST_USER_BLK_MAX_QUEUES, &vu_block_iface, | ||
582 | - errp)) { | ||
583 | - goto error; | ||
584 | - } | ||
585 | - | ||
586 | - blk_add_aio_context_notifier(vu_block_device->backend, blk_aio_attached, | ||
587 | - blk_aio_detach, vu_block_device); | ||
588 | - vu_block_device->running = true; | ||
589 | - return; | ||
590 | - | ||
591 | - error: | ||
592 | - vu_block_deinit(vu_block_device); | ||
593 | -} | ||
594 | - | ||
595 | -static bool vu_prop_modifiable(VuBlockDev *vus, Error **errp) | ||
596 | -{ | ||
597 | - if (vus->running) { | ||
598 | - error_setg(errp, "The property can't be modified " | ||
599 | - "while the server is running"); | ||
600 | - return false; | ||
601 | - } | ||
602 | - return true; | ||
603 | -} | ||
604 | - | ||
605 | -static void vu_set_node_name(Object *obj, const char *value, Error **errp) | ||
606 | -{ | ||
607 | - VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
608 | - | ||
609 | - if (!vu_prop_modifiable(vus, errp)) { | ||
610 | - return; | ||
611 | - } | ||
612 | - | ||
613 | - if (vus->node_name) { | ||
614 | - g_free(vus->node_name); | ||
615 | - } | ||
616 | - | ||
617 | - vus->node_name = g_strdup(value); | ||
618 | -} | ||
619 | - | ||
620 | -static char *vu_get_node_name(Object *obj, Error **errp) | ||
621 | -{ | ||
622 | - VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
623 | - return g_strdup(vus->node_name); | ||
624 | -} | ||
625 | - | ||
626 | -static void free_socket_addr(SocketAddress *addr) | ||
627 | -{ | ||
628 | - g_free(addr->u.q_unix.path); | ||
629 | - g_free(addr); | ||
630 | -} | ||
631 | - | ||
632 | -static void vu_set_unix_socket(Object *obj, const char *value, | ||
633 | - Error **errp) | ||
634 | -{ | ||
635 | - VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
636 | - | ||
637 | - if (!vu_prop_modifiable(vus, errp)) { | ||
638 | - return; | ||
639 | - } | ||
640 | - | ||
641 | - if (vus->addr) { | ||
642 | - free_socket_addr(vus->addr); | ||
643 | - } | ||
644 | - | ||
645 | - SocketAddress *addr = g_new0(SocketAddress, 1); | ||
646 | - addr->type = SOCKET_ADDRESS_TYPE_UNIX; | ||
647 | - addr->u.q_unix.path = g_strdup(value); | ||
648 | - vus->addr = addr; | ||
649 | + vhost_user_server_stop(&vexp->vu_server); | ||
650 | } | ||
651 | |||
652 | -static char *vu_get_unix_socket(Object *obj, Error **errp) | ||
653 | +static int vu_blk_exp_create(BlockExport *exp, BlockExportOptions *opts, | ||
654 | + Error **errp) | ||
655 | { | ||
656 | - VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
657 | - return g_strdup(vus->addr->u.q_unix.path); | ||
658 | -} | ||
659 | - | ||
660 | -static bool vu_get_block_writable(Object *obj, Error **errp) | ||
661 | -{ | ||
662 | - VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
663 | - return vus->writable; | ||
664 | -} | ||
665 | - | ||
666 | -static void vu_set_block_writable(Object *obj, bool value, Error **errp) | ||
667 | -{ | ||
668 | - VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
669 | - | ||
670 | - if (!vu_prop_modifiable(vus, errp)) { | ||
671 | - return; | ||
672 | - } | ||
673 | - | ||
674 | - vus->writable = value; | ||
675 | -} | ||
676 | - | ||
677 | -static void vu_get_blk_size(Object *obj, Visitor *v, const char *name, | ||
678 | - void *opaque, Error **errp) | ||
679 | -{ | ||
680 | - VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
681 | - uint32_t value = vus->blk_size; | ||
682 | - | ||
683 | - visit_type_uint32(v, name, &value, errp); | ||
684 | -} | ||
685 | - | ||
686 | -static void vu_set_blk_size(Object *obj, Visitor *v, const char *name, | ||
687 | - void *opaque, Error **errp) | ||
688 | -{ | ||
689 | - VuBlockDev *vus = VHOST_USER_BLK_SERVER(obj); | ||
690 | - | ||
691 | + VuBlkExport *vexp = container_of(exp, VuBlkExport, export); | ||
692 | + BlockExportOptionsVhostUserBlk *vu_opts = &opts->u.vhost_user_blk; | ||
693 | Error *local_err = NULL; | ||
694 | - uint32_t value; | ||
695 | + uint64_t logical_block_size; | ||
696 | |||
697 | - if (!vu_prop_modifiable(vus, errp)) { | ||
698 | - return; | ||
699 | - } | ||
700 | + vexp->writable = opts->writable; | ||
701 | + vexp->blkcfg.wce = 0; | ||
702 | |||
703 | - visit_type_uint32(v, name, &value, &local_err); | ||
704 | - if (local_err) { | ||
705 | - goto out; | ||
706 | + if (vu_opts->has_logical_block_size) { | ||
707 | + logical_block_size = vu_opts->logical_block_size; | ||
708 | + } else { | ||
709 | + logical_block_size = BDRV_SECTOR_SIZE; | ||
710 | } | ||
711 | - | ||
712 | - check_block_size(object_get_typename(obj), name, value, &local_err); | ||
713 | + check_block_size(exp->id, "logical-block-size", logical_block_size, | ||
714 | + &local_err); | ||
715 | if (local_err) { | ||
716 | - goto out; | ||
717 | + error_propagate(errp, local_err); | ||
718 | + return -EINVAL; | ||
719 | + } | ||
720 | + vexp->blk_size = logical_block_size; | ||
721 | + blk_set_guest_block_size(exp->blk, logical_block_size); | ||
722 | + vu_blk_initialize_config(blk_bs(exp->blk), &vexp->blkcfg, | ||
723 | + logical_block_size); | ||
724 | + | ||
725 | + blk_set_allow_aio_context_change(exp->blk, true); | ||
726 | + blk_add_aio_context_notifier(exp->blk, blk_aio_attached, blk_aio_detach, | ||
727 | + vexp); | ||
728 | + | ||
729 | + if (!vhost_user_server_start(&vexp->vu_server, vu_opts->addr, exp->ctx, | ||
730 | + VHOST_USER_BLK_MAX_QUEUES, &vu_blk_iface, | ||
731 | + errp)) { | ||
732 | + blk_remove_aio_context_notifier(exp->blk, blk_aio_attached, | ||
733 | + blk_aio_detach, vexp); | ||
734 | + return -EADDRNOTAVAIL; | ||
735 | } | ||
736 | |||
737 | - vus->blk_size = value; | ||
738 | - | ||
739 | -out: | ||
740 | - error_propagate(errp, local_err); | ||
741 | -} | ||
742 | - | ||
743 | -static void vhost_user_blk_server_instance_finalize(Object *obj) | ||
744 | -{ | ||
745 | - VuBlockDev *vub = VHOST_USER_BLK_SERVER(obj); | ||
746 | - | ||
747 | - vhost_user_blk_server_stop(vub); | ||
748 | - | ||
749 | - /* | ||
750 | - * Unlike object_property_add_str, object_class_property_add_str | ||
751 | - * doesn't have a release method. Thus manual memory freeing is | ||
752 | - * needed. | ||
753 | - */ | ||
754 | - free_socket_addr(vub->addr); | ||
755 | - g_free(vub->node_name); | ||
756 | -} | ||
757 | - | ||
758 | -static void vhost_user_blk_server_complete(UserCreatable *obj, Error **errp) | ||
759 | -{ | ||
760 | - VuBlockDev *vub = VHOST_USER_BLK_SERVER(obj); | ||
761 | - | ||
762 | - vhost_user_blk_server_start(vub, errp); | ||
180 | + return 0; | 763 | + return 0; |
181 | } | 764 | } |
182 | 765 | ||
183 | /** | 766 | -static void vhost_user_blk_server_class_init(ObjectClass *klass, |
184 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_absolute(URI *uri, const char **str) | 767 | - void *class_data) |
185 | cur = *str; | 768 | +static void vu_blk_exp_delete(BlockExport *exp) |
186 | 769 | { | |
187 | if (*cur != '/') | 770 | - UserCreatableClass *ucc = USER_CREATABLE_CLASS(klass); |
188 | - return (1); | 771 | - ucc->complete = vhost_user_blk_server_complete; |
189 | + return 1; | 772 | - |
190 | cur++; | 773 | - object_class_property_add_bool(klass, "writable", |
191 | ret = rfc3986_parse_segment(&cur, 0, 0); | 774 | - vu_get_block_writable, |
192 | if (ret == 0) { | 775 | - vu_set_block_writable); |
193 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_absolute(URI *uri, const char **str) | 776 | - |
194 | cur++; | 777 | - object_class_property_add_str(klass, "node-name", |
195 | ret = rfc3986_parse_segment(&cur, 0, 1); | 778 | - vu_get_node_name, |
196 | if (ret != 0) | 779 | - vu_set_node_name); |
197 | - return (ret); | 780 | - |
198 | + return ret; | 781 | - object_class_property_add_str(klass, "unix-socket", |
199 | } | 782 | - vu_get_unix_socket, |
200 | } | 783 | - vu_set_unix_socket); |
201 | if (uri != NULL) { | 784 | + VuBlkExport *vexp = container_of(exp, VuBlkExport, export); |
202 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_absolute(URI *uri, const char **str) | 785 | |
203 | } | 786 | - object_class_property_add(klass, "logical-block-size", "uint32", |
204 | } | 787 | - vu_get_blk_size, vu_set_blk_size, |
205 | *str = cur; | 788 | - NULL, NULL); |
206 | - return (0); | 789 | + blk_remove_aio_context_notifier(exp->blk, blk_aio_attached, blk_aio_detach, |
207 | + return 0; | 790 | + vexp); |
208 | } | 791 | } |
209 | 792 | ||
210 | /** | 793 | -static const TypeInfo vhost_user_blk_server_info = { |
211 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_rootless(URI *uri, const char **str) | 794 | - .name = TYPE_VHOST_USER_BLK_SERVER, |
212 | 795 | - .parent = TYPE_OBJECT, | |
213 | ret = rfc3986_parse_segment(&cur, 0, 0); | 796 | - .instance_size = sizeof(VuBlockDev), |
214 | if (ret != 0) | 797 | - .instance_finalize = vhost_user_blk_server_instance_finalize, |
215 | - return (ret); | 798 | - .class_init = vhost_user_blk_server_class_init, |
216 | + return ret; | 799 | - .interfaces = (InterfaceInfo[]) { |
217 | while (*cur == '/') { | 800 | - {TYPE_USER_CREATABLE}, |
218 | cur++; | 801 | - {} |
219 | ret = rfc3986_parse_segment(&cur, 0, 1); | 802 | - }, |
220 | if (ret != 0) | 803 | +const BlockExportDriver blk_exp_vhost_user_blk = { |
221 | - return (ret); | 804 | + .type = BLOCK_EXPORT_TYPE_VHOST_USER_BLK, |
222 | + return ret; | 805 | + .instance_size = sizeof(VuBlkExport), |
223 | } | 806 | + .create = vu_blk_exp_create, |
224 | if (uri != NULL) { | 807 | + .delete = vu_blk_exp_delete, |
225 | g_free(uri->path); | 808 | + .request_shutdown = vu_blk_exp_request_shutdown, |
226 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_rootless(URI *uri, const char **str) | 809 | }; |
227 | } | 810 | - |
228 | } | 811 | -static void vhost_user_blk_server_register_types(void) |
229 | *str = cur; | 812 | -{ |
230 | - return (0); | 813 | - type_register_static(&vhost_user_blk_server_info); |
231 | + return 0; | 814 | -} |
232 | } | 815 | - |
233 | 816 | -type_init(vhost_user_blk_server_register_types) | |
234 | /** | 817 | diff --git a/util/vhost-user-server.c b/util/vhost-user-server.c |
235 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_no_scheme(URI *uri, const char **str) | 818 | index XXXXXXX..XXXXXXX 100644 |
236 | 819 | --- a/util/vhost-user-server.c | |
237 | ret = rfc3986_parse_segment(&cur, ':', 0); | 820 | +++ b/util/vhost-user-server.c |
238 | if (ret != 0) | 821 | @@ -XXX,XX +XXX,XX @@ bool vhost_user_server_start(VuServer *server, |
239 | - return (ret); | 822 | Error **errp) |
240 | + return ret; | 823 | { |
241 | while (*cur == '/') { | 824 | QEMUBH *bh; |
242 | cur++; | 825 | - QIONetListener *listener = qio_net_listener_new(); |
243 | ret = rfc3986_parse_segment(&cur, 0, 1); | 826 | + QIONetListener *listener; |
244 | if (ret != 0) | 827 | + |
245 | - return (ret); | 828 | + if (socket_addr->type != SOCKET_ADDRESS_TYPE_UNIX && |
246 | + return ret; | 829 | + socket_addr->type != SOCKET_ADDRESS_TYPE_FD) { |
247 | } | 830 | + error_setg(errp, "Only socket address types 'unix' and 'fd' are supported"); |
248 | if (uri != NULL) { | 831 | + return false; |
249 | g_free(uri->path); | 832 | + } |
250 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_no_scheme(URI *uri, const char **str) | 833 | + |
251 | } | 834 | + listener = qio_net_listener_new(); |
252 | } | 835 | if (qio_net_listener_open_sync(listener, socket_addr, 1, |
253 | *str = cur; | 836 | errp) < 0) { |
254 | - return (0); | 837 | object_unref(OBJECT(listener)); |
255 | + return 0; | 838 | diff --git a/block/export/meson.build b/block/export/meson.build |
256 | } | 839 | index XXXXXXX..XXXXXXX 100644 |
257 | 840 | --- a/block/export/meson.build | |
258 | /** | 841 | +++ b/block/export/meson.build |
259 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_hier_part(URI *uri, const char **str) | 842 | @@ -1 +1,2 @@ |
260 | cur += 2; | 843 | block_ss.add(files('export.c')) |
261 | ret = rfc3986_parse_authority(uri, &cur); | 844 | +block_ss.add(when: 'CONFIG_LINUX', if_true: files('vhost-user-blk-server.c', '../../contrib/libvhost-user/libvhost-user.c')) |
262 | if (ret != 0) | 845 | diff --git a/block/meson.build b/block/meson.build |
263 | - return (ret); | 846 | index XXXXXXX..XXXXXXX 100644 |
264 | + return ret; | 847 | --- a/block/meson.build |
265 | ret = rfc3986_parse_path_ab_empty(uri, &cur); | 848 | +++ b/block/meson.build |
266 | if (ret != 0) | 849 | @@ -XXX,XX +XXX,XX @@ block_ss.add(when: 'CONFIG_WIN32', if_true: files('file-win32.c', 'win32-aio.c') |
267 | - return (ret); | 850 | block_ss.add(when: 'CONFIG_POSIX', if_true: [files('file-posix.c'), coref, iokit]) |
268 | + return ret; | 851 | block_ss.add(when: 'CONFIG_LIBISCSI', if_true: files('iscsi-opts.c')) |
269 | *str = cur; | 852 | block_ss.add(when: 'CONFIG_LINUX', if_true: files('nvme.c')) |
270 | - return (0); | 853 | -block_ss.add(when: 'CONFIG_LINUX', if_true: files('export/vhost-user-blk-server.c', '../contrib/libvhost-user/libvhost-user.c')) |
271 | + return 0; | 854 | block_ss.add(when: 'CONFIG_REPLICATION', if_true: files('replication.c')) |
272 | } else if (*cur == '/') { | 855 | block_ss.add(when: 'CONFIG_SHEEPDOG', if_true: files('sheepdog.c')) |
273 | ret = rfc3986_parse_path_absolute(uri, &cur); | 856 | block_ss.add(when: ['CONFIG_LINUX_AIO', libaio], if_true: files('linux-aio.c')) |
274 | if (ret != 0) | ||
275 | - return (ret); | ||
276 | + return ret; | ||
277 | } else if (ISA_PCHAR(cur)) { | ||
278 | ret = rfc3986_parse_path_rootless(uri, &cur); | ||
279 | if (ret != 0) | ||
280 | - return (ret); | ||
281 | + return ret; | ||
282 | } else { | ||
283 | /* path-empty is effectively empty */ | ||
284 | if (uri != NULL) { | ||
285 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_hier_part(URI *uri, const char **str) | ||
286 | } | ||
287 | } | ||
288 | *str = cur; | ||
289 | - return (0); | ||
290 | + return 0; | ||
291 | } | ||
292 | |||
293 | /** | ||
294 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_relative_ref(URI *uri, const char *str) | ||
295 | str += 2; | ||
296 | ret = rfc3986_parse_authority(uri, &str); | ||
297 | if (ret != 0) | ||
298 | - return (ret); | ||
299 | + return ret; | ||
300 | ret = rfc3986_parse_path_ab_empty(uri, &str); | ||
301 | if (ret != 0) | ||
302 | - return (ret); | ||
303 | + return ret; | ||
304 | } else if (*str == '/') { | ||
305 | ret = rfc3986_parse_path_absolute(uri, &str); | ||
306 | if (ret != 0) | ||
307 | - return (ret); | ||
308 | + return ret; | ||
309 | } else if (ISA_PCHAR(str)) { | ||
310 | ret = rfc3986_parse_path_no_scheme(uri, &str); | ||
311 | if (ret != 0) | ||
312 | - return (ret); | ||
313 | + return ret; | ||
314 | } else { | ||
315 | /* path-empty is effectively empty */ | ||
316 | if (uri != NULL) { | ||
317 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_relative_ref(URI *uri, const char *str) | ||
318 | str++; | ||
319 | ret = rfc3986_parse_query(uri, &str); | ||
320 | if (ret != 0) | ||
321 | - return (ret); | ||
322 | + return ret; | ||
323 | } | ||
324 | if (*str == '#') { | ||
325 | str++; | ||
326 | ret = rfc3986_parse_fragment(uri, &str); | ||
327 | if (ret != 0) | ||
328 | - return (ret); | ||
329 | + return ret; | ||
330 | } | ||
331 | if (*str != 0) { | ||
332 | uri_clean(uri); | ||
333 | - return (1); | ||
334 | + return 1; | ||
335 | } | ||
336 | - return (0); | ||
337 | + return 0; | ||
338 | } | ||
339 | |||
340 | /** | ||
341 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse(URI *uri, const char *str) | ||
342 | |||
343 | ret = rfc3986_parse_scheme(uri, &str); | ||
344 | if (ret != 0) | ||
345 | - return (ret); | ||
346 | + return ret; | ||
347 | if (*str != ':') { | ||
348 | - return (1); | ||
349 | + return 1; | ||
350 | } | ||
351 | str++; | ||
352 | ret = rfc3986_parse_hier_part(uri, &str); | ||
353 | if (ret != 0) | ||
354 | - return (ret); | ||
355 | + return ret; | ||
356 | if (*str == '?') { | ||
357 | str++; | ||
358 | ret = rfc3986_parse_query(uri, &str); | ||
359 | if (ret != 0) | ||
360 | - return (ret); | ||
361 | + return ret; | ||
362 | } | ||
363 | if (*str == '#') { | ||
364 | str++; | ||
365 | ret = rfc3986_parse_fragment(uri, &str); | ||
366 | if (ret != 0) | ||
367 | - return (ret); | ||
368 | + return ret; | ||
369 | } | ||
370 | if (*str != 0) { | ||
371 | uri_clean(uri); | ||
372 | - return (1); | ||
373 | + return 1; | ||
374 | } | ||
375 | - return (0); | ||
376 | + return 0; | ||
377 | } | ||
378 | |||
379 | /** | ||
380 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_uri_reference(URI *uri, const char *str) | ||
381 | int ret; | ||
382 | |||
383 | if (str == NULL) | ||
384 | - return (-1); | ||
385 | + return -1; | ||
386 | uri_clean(uri); | ||
387 | |||
388 | /* | ||
389 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_uri_reference(URI *uri, const char *str) | ||
390 | ret = rfc3986_parse_relative_ref(uri, str); | ||
391 | if (ret != 0) { | ||
392 | uri_clean(uri); | ||
393 | - return (ret); | ||
394 | + return ret; | ||
395 | } | ||
396 | } | ||
397 | - return (0); | ||
398 | + return 0; | ||
399 | } | ||
400 | |||
401 | /** | ||
402 | @@ -XXX,XX +XXX,XX @@ URI *uri_parse(const char *str) | ||
403 | int ret; | ||
404 | |||
405 | if (str == NULL) | ||
406 | - return (NULL); | ||
407 | + return NULL; | ||
408 | uri = uri_new(); | ||
409 | ret = rfc3986_parse_uri_reference(uri, str); | ||
410 | if (ret) { | ||
411 | uri_free(uri); | ||
412 | - return (NULL); | ||
413 | + return NULL; | ||
414 | } | ||
415 | - return (uri); | ||
416 | + return uri; | ||
417 | } | ||
418 | |||
419 | /** | ||
420 | @@ -XXX,XX +XXX,XX @@ URI *uri_parse(const char *str) | ||
421 | */ | ||
422 | int uri_parse_into(URI *uri, const char *str) | ||
423 | { | ||
424 | - return (rfc3986_parse_uri_reference(uri, str)); | ||
425 | + return rfc3986_parse_uri_reference(uri, str); | ||
426 | } | ||
427 | |||
428 | /** | ||
429 | @@ -XXX,XX +XXX,XX @@ URI *uri_parse_raw(const char *str, int raw) | ||
430 | int ret; | ||
431 | |||
432 | if (str == NULL) | ||
433 | - return (NULL); | ||
434 | + return NULL; | ||
435 | uri = uri_new(); | ||
436 | if (raw) { | ||
437 | uri->cleanup |= 2; | ||
438 | @@ -XXX,XX +XXX,XX @@ URI *uri_parse_raw(const char *str, int raw) | ||
439 | ret = uri_parse_into(uri, str); | ||
440 | if (ret) { | ||
441 | uri_free(uri); | ||
442 | - return (NULL); | ||
443 | + return NULL; | ||
444 | } | ||
445 | - return (uri); | ||
446 | + return uri; | ||
447 | } | ||
448 | |||
449 | /************************************************************************ | ||
450 | @@ -XXX,XX +XXX,XX @@ URI *uri_new(void) | ||
451 | URI *ret; | ||
452 | |||
453 | ret = g_new0(URI, 1); | ||
454 | - return (ret); | ||
455 | + return ret; | ||
456 | } | ||
457 | |||
458 | /** | ||
459 | @@ -XXX,XX +XXX,XX @@ static char *realloc2n(char *ret, int *max) | ||
460 | tmp = *max * 2; | ||
461 | temp = g_realloc(ret, (tmp + 1)); | ||
462 | *max = tmp; | ||
463 | - return (temp); | ||
464 | + return temp; | ||
465 | } | ||
466 | |||
467 | /** | ||
468 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
469 | int max; | ||
470 | |||
471 | if (uri == NULL) | ||
472 | - return (NULL); | ||
473 | + return NULL; | ||
474 | |||
475 | max = 80; | ||
476 | ret = g_malloc(max + 1); | ||
477 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
478 | ret = temp; | ||
479 | } | ||
480 | ret[len] = 0; | ||
481 | - return (ret); | ||
482 | + return ret; | ||
483 | } | ||
484 | |||
485 | /** | ||
486 | @@ -XXX,XX +XXX,XX @@ static int normalize_uri_path(char *path) | ||
487 | char *cur, *out; | ||
488 | |||
489 | if (path == NULL) | ||
490 | - return (-1); | ||
491 | + return -1; | ||
492 | |||
493 | /* Skip all initial "/" chars. We want to get to the beginning of the | ||
494 | * first non-empty segment. | ||
495 | @@ -XXX,XX +XXX,XX @@ static int normalize_uri_path(char *path) | ||
496 | while (cur[0] == '/') | ||
497 | ++cur; | ||
498 | if (cur[0] == '\0') | ||
499 | - return (0); | ||
500 | + return 0; | ||
501 | |||
502 | /* Keep everything we've seen so far. */ | ||
503 | out = cur; | ||
504 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
505 | while (cur[0] == '/') | ||
506 | ++cur; | ||
507 | if (cur[0] == '\0') | ||
508 | - return (0); | ||
509 | + return 0; | ||
510 | |||
511 | /* | ||
512 | * Analyze each segment in sequence for cases (e) and (f). | ||
513 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
514 | } | ||
515 | } | ||
516 | |||
517 | - return (0); | ||
518 | + return 0; | ||
519 | } | ||
520 | |||
521 | static int is_hex(char c) | ||
522 | { | ||
523 | if (((c >= '0') && (c <= '9')) || ((c >= 'a') && (c <= 'f')) || | ||
524 | ((c >= 'A') && (c <= 'F'))) | ||
525 | - return (1); | ||
526 | - return (0); | ||
527 | + return 1; | ||
528 | + return 0; | ||
529 | } | ||
530 | |||
531 | /** | ||
532 | @@ -XXX,XX +XXX,XX @@ char *uri_string_unescape(const char *str, int len, char *target) | ||
533 | const char *in; | ||
534 | |||
535 | if (str == NULL) | ||
536 | - return (NULL); | ||
537 | + return NULL; | ||
538 | if (len <= 0) | ||
539 | len = strlen(str); | ||
540 | if (len < 0) | ||
541 | - return (NULL); | ||
542 | + return NULL; | ||
543 | |||
544 | if (target == NULL) { | ||
545 | ret = g_malloc(len + 1); | ||
546 | @@ -XXX,XX +XXX,XX @@ char *uri_string_unescape(const char *str, int len, char *target) | ||
547 | } | ||
548 | } | ||
549 | *out = 0; | ||
550 | - return (ret); | ||
551 | + return ret; | ||
552 | } | ||
553 | |||
554 | /** | ||
555 | @@ -XXX,XX +XXX,XX @@ char *uri_string_escape(const char *str, const char *list) | ||
556 | int len, out; | ||
557 | |||
558 | if (str == NULL) | ||
559 | - return (NULL); | ||
560 | + return NULL; | ||
561 | if (str[0] == 0) | ||
562 | - return (g_strdup(str)); | ||
563 | + return g_strdup(str); | ||
564 | len = strlen(str); | ||
565 | if (!(len > 0)) | ||
566 | - return (NULL); | ||
567 | + return NULL; | ||
568 | |||
569 | len += 20; | ||
570 | ret = g_malloc(len); | ||
571 | @@ -XXX,XX +XXX,XX @@ char *uri_string_escape(const char *str, const char *list) | ||
572 | } | ||
573 | } | ||
574 | ret[out] = 0; | ||
575 | - return (ret); | ||
576 | + return ret; | ||
577 | } | ||
578 | |||
579 | /************************************************************************ | ||
580 | @@ -XXX,XX +XXX,XX @@ done: | ||
581 | uri_free(bas); | ||
582 | if (res != NULL) | ||
583 | uri_free(res); | ||
584 | - return (val); | ||
585 | + return val; | ||
586 | } | ||
587 | |||
588 | /** | ||
589 | -- | 857 | -- |
590 | 2.14.3 | 858 | 2.26.2 |
591 | 859 | ||
592 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | Headers used by other subsystems are located in include/. Also add the | ||
2 | vhost-user-server and vhost-user-blk-server headers to MAINTAINERS. | ||
1 | 3 | ||
4 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
5 | Message-id: 20200924151549.913737-13-stefanha@redhat.com | ||
6 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
7 | --- | ||
8 | MAINTAINERS | 4 +++- | ||
9 | {util => include/qemu}/vhost-user-server.h | 0 | ||
10 | block/export/vhost-user-blk-server.c | 2 +- | ||
11 | util/vhost-user-server.c | 2 +- | ||
12 | 4 files changed, 5 insertions(+), 3 deletions(-) | ||
13 | rename {util => include/qemu}/vhost-user-server.h (100%) | ||
14 | |||
15 | diff --git a/MAINTAINERS b/MAINTAINERS | ||
16 | index XXXXXXX..XXXXXXX 100644 | ||
17 | --- a/MAINTAINERS | ||
18 | +++ b/MAINTAINERS | ||
19 | @@ -XXX,XX +XXX,XX @@ Vhost-user block device backend server | ||
20 | M: Coiby Xu <Coiby.Xu@gmail.com> | ||
21 | S: Maintained | ||
22 | F: block/export/vhost-user-blk-server.c | ||
23 | -F: util/vhost-user-server.c | ||
24 | +F: block/export/vhost-user-blk-server.h | ||
25 | +F: include/qemu/vhost-user-server.h | ||
26 | F: tests/qtest/libqos/vhost-user-blk.c | ||
27 | +F: util/vhost-user-server.c | ||
28 | |||
29 | Replication | ||
30 | M: Wen Congyang <wencongyang2@huawei.com> | ||
31 | diff --git a/util/vhost-user-server.h b/include/qemu/vhost-user-server.h | ||
32 | similarity index 100% | ||
33 | rename from util/vhost-user-server.h | ||
34 | rename to include/qemu/vhost-user-server.h | ||
35 | diff --git a/block/export/vhost-user-blk-server.c b/block/export/vhost-user-blk-server.c | ||
36 | index XXXXXXX..XXXXXXX 100644 | ||
37 | --- a/block/export/vhost-user-blk-server.c | ||
38 | +++ b/block/export/vhost-user-blk-server.c | ||
39 | @@ -XXX,XX +XXX,XX @@ | ||
40 | #include "block/block.h" | ||
41 | #include "contrib/libvhost-user/libvhost-user.h" | ||
42 | #include "standard-headers/linux/virtio_blk.h" | ||
43 | -#include "util/vhost-user-server.h" | ||
44 | +#include "qemu/vhost-user-server.h" | ||
45 | #include "vhost-user-blk-server.h" | ||
46 | #include "qapi/error.h" | ||
47 | #include "qom/object_interfaces.h" | ||
48 | diff --git a/util/vhost-user-server.c b/util/vhost-user-server.c | ||
49 | index XXXXXXX..XXXXXXX 100644 | ||
50 | --- a/util/vhost-user-server.c | ||
51 | +++ b/util/vhost-user-server.c | ||
52 | @@ -XXX,XX +XXX,XX @@ | ||
53 | */ | ||
54 | #include "qemu/osdep.h" | ||
55 | #include "qemu/main-loop.h" | ||
56 | +#include "qemu/vhost-user-server.h" | ||
57 | #include "block/aio-wait.h" | ||
58 | -#include "vhost-user-server.h" | ||
59 | |||
60 | /* | ||
61 | * Theory of operation: | ||
62 | -- | ||
63 | 2.26.2 | ||
64 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | Don't compile contrib/libvhost-user/libvhost-user.c again. Instead build | ||
2 | the static library once and then reuse it throughout QEMU. | ||
1 | 3 | ||
4 | Also switch from CONFIG_LINUX to CONFIG_VHOST_USER, which is what the | ||
5 | vhost-user tools (vhost-user-gpu, etc) do. | ||
6 | |||
7 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
8 | Message-id: 20200924151549.913737-14-stefanha@redhat.com | ||
9 | [Added CONFIG_LINUX again because libvhost-user doesn't build on macOS. | ||
10 | --Stefan] | ||
11 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
12 | --- | ||
13 | block/export/export.c | 8 ++++---- | ||
14 | block/export/meson.build | 2 +- | ||
15 | contrib/libvhost-user/meson.build | 1 + | ||
16 | meson.build | 6 +++++- | ||
17 | util/meson.build | 4 +++- | ||
18 | 5 files changed, 14 insertions(+), 7 deletions(-) | ||
19 | |||
20 | diff --git a/block/export/export.c b/block/export/export.c | ||
21 | index XXXXXXX..XXXXXXX 100644 | ||
22 | --- a/block/export/export.c | ||
23 | +++ b/block/export/export.c | ||
24 | @@ -XXX,XX +XXX,XX @@ | ||
25 | #include "sysemu/block-backend.h" | ||
26 | #include "block/export.h" | ||
27 | #include "block/nbd.h" | ||
28 | -#if CONFIG_LINUX | ||
29 | -#include "block/export/vhost-user-blk-server.h" | ||
30 | -#endif | ||
31 | #include "qapi/error.h" | ||
32 | #include "qapi/qapi-commands-block-export.h" | ||
33 | #include "qapi/qapi-events-block-export.h" | ||
34 | #include "qemu/id.h" | ||
35 | +#ifdef CONFIG_VHOST_USER | ||
36 | +#include "vhost-user-blk-server.h" | ||
37 | +#endif | ||
38 | |||
39 | static const BlockExportDriver *blk_exp_drivers[] = { | ||
40 | &blk_exp_nbd, | ||
41 | -#if CONFIG_LINUX | ||
42 | +#ifdef CONFIG_VHOST_USER | ||
43 | &blk_exp_vhost_user_blk, | ||
44 | #endif | ||
45 | }; | ||
46 | diff --git a/block/export/meson.build b/block/export/meson.build | ||
47 | index XXXXXXX..XXXXXXX 100644 | ||
48 | --- a/block/export/meson.build | ||
49 | +++ b/block/export/meson.build | ||
50 | @@ -XXX,XX +XXX,XX @@ | ||
51 | block_ss.add(files('export.c')) | ||
52 | -block_ss.add(when: 'CONFIG_LINUX', if_true: files('vhost-user-blk-server.c', '../../contrib/libvhost-user/libvhost-user.c')) | ||
53 | +block_ss.add(when: ['CONFIG_LINUX', 'CONFIG_VHOST_USER'], if_true: files('vhost-user-blk-server.c')) | ||
54 | diff --git a/contrib/libvhost-user/meson.build b/contrib/libvhost-user/meson.build | ||
55 | index XXXXXXX..XXXXXXX 100644 | ||
56 | --- a/contrib/libvhost-user/meson.build | ||
57 | +++ b/contrib/libvhost-user/meson.build | ||
58 | @@ -XXX,XX +XXX,XX @@ | ||
59 | libvhost_user = static_library('vhost-user', | ||
60 | files('libvhost-user.c', 'libvhost-user-glib.c'), | ||
61 | build_by_default: false) | ||
62 | +vhost_user = declare_dependency(link_with: libvhost_user) | ||
63 | diff --git a/meson.build b/meson.build | ||
64 | index XXXXXXX..XXXXXXX 100644 | ||
65 | --- a/meson.build | ||
66 | +++ b/meson.build | ||
67 | @@ -XXX,XX +XXX,XX @@ trace_events_subdirs += [ | ||
68 | 'util', | ||
69 | ] | ||
70 | |||
71 | +vhost_user = not_found | ||
72 | +if 'CONFIG_VHOST_USER' in config_host | ||
73 | + subdir('contrib/libvhost-user') | ||
74 | +endif | ||
75 | + | ||
76 | subdir('qapi') | ||
77 | subdir('qobject') | ||
78 | subdir('stubs') | ||
79 | @@ -XXX,XX +XXX,XX @@ if have_tools | ||
80 | install: true) | ||
81 | |||
82 | if 'CONFIG_VHOST_USER' in config_host | ||
83 | - subdir('contrib/libvhost-user') | ||
84 | subdir('contrib/vhost-user-blk') | ||
85 | subdir('contrib/vhost-user-gpu') | ||
86 | subdir('contrib/vhost-user-input') | ||
87 | diff --git a/util/meson.build b/util/meson.build | ||
88 | index XXXXXXX..XXXXXXX 100644 | ||
89 | --- a/util/meson.build | ||
90 | +++ b/util/meson.build | ||
91 | @@ -XXX,XX +XXX,XX @@ if have_block | ||
92 | util_ss.add(files('main-loop.c')) | ||
93 | util_ss.add(files('nvdimm-utils.c')) | ||
94 | util_ss.add(files('qemu-coroutine.c', 'qemu-coroutine-lock.c', 'qemu-coroutine-io.c')) | ||
95 | - util_ss.add(when: 'CONFIG_LINUX', if_true: files('vhost-user-server.c')) | ||
96 | + util_ss.add(when: ['CONFIG_LINUX', 'CONFIG_VHOST_USER'], if_true: [ | ||
97 | + files('vhost-user-server.c'), vhost_user | ||
98 | + ]) | ||
99 | util_ss.add(files('block-helpers.c')) | ||
100 | util_ss.add(files('qemu-coroutine-sleep.c')) | ||
101 | util_ss.add(files('qemu-co-shared-resource.c')) | ||
102 | -- | ||
103 | 2.26.2 | ||
104 | diff view generated by jsdifflib |
1 | From: Fam Zheng <famz@redhat.com> | 1 | Introduce libblkdev.fa to avoid recompiling blockdev_ss twice. |
---|---|---|---|
2 | 2 | ||
3 | Signed-off-by: Fam Zheng <famz@redhat.com> | 3 | Suggested-by: Paolo Bonzini <pbonzini@redhat.com> |
4 | Message-id: 20180226030326.20219-3-famz@redhat.com | 4 | Reviewed-by: Paolo Bonzini <pbonzini@redhat.com> |
5 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
6 | Message-id: 20200929125516.186715-3-stefanha@redhat.com | ||
5 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 7 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
6 | --- | 8 | --- |
7 | README | 31 ++++++++++++++++++++++++++++++- | 9 | meson.build | 12 ++++++++++-- |
8 | 1 file changed, 30 insertions(+), 1 deletion(-) | 10 | storage-daemon/meson.build | 3 +-- |
11 | 2 files changed, 11 insertions(+), 4 deletions(-) | ||
9 | 12 | ||
10 | diff --git a/README b/README | 13 | diff --git a/meson.build b/meson.build |
11 | index XXXXXXX..XXXXXXX 100644 | 14 | index XXXXXXX..XXXXXXX 100644 |
12 | --- a/README | 15 | --- a/meson.build |
13 | +++ b/README | 16 | +++ b/meson.build |
14 | @@ -XXX,XX +XXX,XX @@ The QEMU source code is maintained under the GIT version control system. | 17 | @@ -XXX,XX +XXX,XX @@ blockdev_ss.add(files( |
15 | 18 | # os-win32.c does not | |
16 | git clone git://git.qemu.org/qemu.git | 19 | blockdev_ss.add(when: 'CONFIG_POSIX', if_true: files('os-posix.c')) |
17 | 20 | softmmu_ss.add(when: 'CONFIG_WIN32', if_true: [files('os-win32.c')]) | |
18 | -When submitting patches, the preferred approach is to use 'git | 21 | -softmmu_ss.add_all(blockdev_ss) |
19 | +When submitting patches, one common approach is to use 'git | 22 | |
20 | format-patch' and/or 'git send-email' to format & send the mail to the | 23 | common_ss.add(files('cpus-common.c')) |
21 | qemu-devel@nongnu.org mailing list. All patches submitted must contain | 24 | |
22 | a 'Signed-off-by' line from the author. Patches should follow the | 25 | @@ -XXX,XX +XXX,XX @@ block = declare_dependency(link_whole: [libblock], |
23 | @@ -XXX,XX +XXX,XX @@ The QEMU website is also maintained under source control. | 26 | link_args: '@block.syms', |
24 | git clone git://git.qemu.org/qemu-web.git | 27 | dependencies: [crypto, io]) |
25 | https://www.qemu.org/2017/02/04/the-new-qemu-website-is-up/ | 28 | |
26 | 29 | +blockdev_ss = blockdev_ss.apply(config_host, strict: false) | |
27 | +A 'git-profile' utility was created to make above process less | 30 | +libblockdev = static_library('blockdev', blockdev_ss.sources() + genh, |
28 | +cumbersome, and is highly recommended for making regular contributions, | 31 | + dependencies: blockdev_ss.dependencies(), |
29 | +or even just for sending consecutive patch series revisions. It also | 32 | + name_suffix: 'fa', |
30 | +requires a working 'git send-email' setup, and by default doesn't | 33 | + build_by_default: false) |
31 | +automate everything, so you may want to go through the above steps | ||
32 | +manually for once. | ||
33 | + | 34 | + |
34 | +For installation instructions, please go to | 35 | +blockdev = declare_dependency(link_whole: [libblockdev], |
36 | + dependencies: [block]) | ||
35 | + | 37 | + |
36 | + https://github.com/stefanha/git-publish | 38 | qmp_ss = qmp_ss.apply(config_host, strict: false) |
37 | + | 39 | libqmp = static_library('qmp', qmp_ss.sources() + genh, |
38 | +The workflow with 'git-publish' is: | 40 | dependencies: qmp_ss.dependencies(), |
39 | + | 41 | @@ -XXX,XX +XXX,XX @@ foreach m : block_mods + softmmu_mods |
40 | + $ git checkout master -b my-feature | 42 | install_dir: config_host['qemu_moddir']) |
41 | + $ # work on new commits, add your 'Signed-off-by' lines to each | 43 | endforeach |
42 | + $ git publish | 44 | |
43 | + | 45 | -softmmu_ss.add(authz, block, chardev, crypto, io, qmp) |
44 | +Your patch series will be sent and tagged as my-feature-v1 if you need to refer | 46 | +softmmu_ss.add(authz, blockdev, chardev, crypto, io, qmp) |
45 | +back to it in the future. | 47 | common_ss.add(qom, qemuutil) |
46 | + | 48 | |
47 | +Sending v2: | 49 | common_ss.add_all(when: 'CONFIG_SOFTMMU', if_true: [softmmu_ss]) |
48 | + | 50 | diff --git a/storage-daemon/meson.build b/storage-daemon/meson.build |
49 | + $ git checkout my-feature # same topic branch | 51 | index XXXXXXX..XXXXXXX 100644 |
50 | + $ # making changes to the commits (using 'git rebase', for example) | 52 | --- a/storage-daemon/meson.build |
51 | + $ git publish | 53 | +++ b/storage-daemon/meson.build |
52 | + | 54 | @@ -XXX,XX +XXX,XX @@ |
53 | +Your patch series will be sent with 'v2' tag in the subject and the git tip | 55 | qsd_ss = ss.source_set() |
54 | +will be tagged as my-feature-v2. | 56 | qsd_ss.add(files('qemu-storage-daemon.c')) |
55 | + | 57 | -qsd_ss.add(block, chardev, qmp, qom, qemuutil) |
56 | Bug reporting | 58 | -qsd_ss.add_all(blockdev_ss) |
57 | ============= | 59 | +qsd_ss.add(blockdev, chardev, qmp, qom, qemuutil) |
60 | |||
61 | subdir('qapi') | ||
58 | 62 | ||
59 | -- | 63 | -- |
60 | 2.14.3 | 64 | 2.26.2 |
61 | 65 | ||
62 | diff view generated by jsdifflib |
1 | From: Fam Zheng <famz@redhat.com> | 1 | Block exports are used by softmmu, qemu-storage-daemon, and qemu-nbd. |
---|---|---|---|
2 | They are not used by other programs and are not otherwise needed in | ||
3 | libblock. | ||
2 | 4 | ||
3 | git-publish [1] is a convenient tool to send patches and has been | 5 | Undo the recent move of blockdev-nbd.c from blockdev_ss into block_ss. |
4 | popular among QEMU developers. Recently it has been made available in | 6 | Since bdrv_close_all() (libblock) calls blk_exp_close_all() |
5 | Fedora/Debian official repo. | 7 | (libblockdev) a stub function is required.. |
6 | 8 | ||
7 | One nice feature of the tool is a per-project configuration with | 9 | Make qemu-nbd.c use signal handling utility functions instead of |
8 | profiles, especially in which the cccmd option is a handy method to | 10 | duplicating the code. This helps because os-posix.c is in libblockdev |
9 | create the Cc list. | 11 | and it depends on a qemu_system_killed() symbol that qemu-nbd.c lacks. |
12 | Once we use the signal handling utility functions we also end up | ||
13 | providing the necessary symbol. | ||
10 | 14 | ||
11 | [1]: https://github.com/stefanha/git-publish | 15 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
12 | 16 | Reviewed-by: Paolo Bonzini <pbonzini@redhat.com> | |
13 | Signed-off-by: Fam Zheng <famz@redhat.com> | 17 | Reviewed-by: Eric Blake <eblake@redhat.com> |
14 | Message-id: 20180226030326.20219-2-famz@redhat.com | 18 | Message-id: 20200929125516.186715-4-stefanha@redhat.com |
19 | [Fixed s/ndb/nbd/ typo in commit description as suggested by Eric Blake | ||
20 | --Stefan] | ||
15 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 21 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
16 | --- | 22 | --- |
17 | .gitpublish | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++ | 23 | qemu-nbd.c | 21 ++++++++------------- |
18 | 1 file changed, 51 insertions(+) | 24 | stubs/blk-exp-close-all.c | 7 +++++++ |
19 | create mode 100644 .gitpublish | 25 | block/export/meson.build | 4 ++-- |
26 | meson.build | 4 ++-- | ||
27 | nbd/meson.build | 2 ++ | ||
28 | stubs/meson.build | 1 + | ||
29 | 6 files changed, 22 insertions(+), 17 deletions(-) | ||
30 | create mode 100644 stubs/blk-exp-close-all.c | ||
20 | 31 | ||
21 | diff --git a/.gitpublish b/.gitpublish | 32 | diff --git a/qemu-nbd.c b/qemu-nbd.c |
33 | index XXXXXXX..XXXXXXX 100644 | ||
34 | --- a/qemu-nbd.c | ||
35 | +++ b/qemu-nbd.c | ||
36 | @@ -XXX,XX +XXX,XX @@ | ||
37 | #include "qapi/error.h" | ||
38 | #include "qemu/cutils.h" | ||
39 | #include "sysemu/block-backend.h" | ||
40 | +#include "sysemu/runstate.h" /* for qemu_system_killed() prototype */ | ||
41 | #include "block/block_int.h" | ||
42 | #include "block/nbd.h" | ||
43 | #include "qemu/main-loop.h" | ||
44 | @@ -XXX,XX +XXX,XX @@ QEMU_COPYRIGHT "\n" | ||
45 | } | ||
46 | |||
47 | #ifdef CONFIG_POSIX | ||
48 | -static void termsig_handler(int signum) | ||
49 | +/* | ||
50 | + * The client thread uses SIGTERM to interrupt the server. A signal | ||
51 | + * handler ensures that "qemu-nbd -v -c" exits with a nice status code. | ||
52 | + */ | ||
53 | +void qemu_system_killed(int signum, pid_t pid) | ||
54 | { | ||
55 | qatomic_cmpxchg(&state, RUNNING, TERMINATE); | ||
56 | qemu_notify_event(); | ||
57 | @@ -XXX,XX +XXX,XX @@ int main(int argc, char **argv) | ||
58 | BlockExportOptions *export_opts; | ||
59 | |||
60 | #ifdef CONFIG_POSIX | ||
61 | - /* | ||
62 | - * Exit gracefully on various signals, which includes SIGTERM used | ||
63 | - * by 'qemu-nbd -v -c'. | ||
64 | - */ | ||
65 | - struct sigaction sa_sigterm; | ||
66 | - memset(&sa_sigterm, 0, sizeof(sa_sigterm)); | ||
67 | - sa_sigterm.sa_handler = termsig_handler; | ||
68 | - sigaction(SIGTERM, &sa_sigterm, NULL); | ||
69 | - sigaction(SIGINT, &sa_sigterm, NULL); | ||
70 | - sigaction(SIGHUP, &sa_sigterm, NULL); | ||
71 | - | ||
72 | - signal(SIGPIPE, SIG_IGN); | ||
73 | + os_setup_early_signal_handling(); | ||
74 | + os_setup_signal_handling(); | ||
75 | #endif | ||
76 | |||
77 | socket_init(); | ||
78 | diff --git a/stubs/blk-exp-close-all.c b/stubs/blk-exp-close-all.c | ||
22 | new file mode 100644 | 79 | new file mode 100644 |
23 | index XXXXXXX..XXXXXXX | 80 | index XXXXXXX..XXXXXXX |
24 | --- /dev/null | 81 | --- /dev/null |
25 | +++ b/.gitpublish | 82 | +++ b/stubs/blk-exp-close-all.c |
26 | @@ -XXX,XX +XXX,XX @@ | 83 | @@ -XXX,XX +XXX,XX @@ |
27 | +# | 84 | +#include "qemu/osdep.h" |
28 | +# Common git-publish profiles that can be used to send patches to QEMU upstream. | 85 | +#include "block/export.h" |
29 | +# | ||
30 | +# See https://github.com/stefanha/git-publish for more information | ||
31 | +# | ||
32 | +[gitpublishprofile "default"] | ||
33 | +base = master | ||
34 | +to = qemu-devel@nongnu.org | ||
35 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | ||
36 | + | 86 | + |
37 | +[gitpublishprofile "rfc"] | 87 | +/* Only used in programs that support block exports (libblockdev.fa) */ |
38 | +base = master | 88 | +void blk_exp_close_all(void) |
39 | +prefix = RFC PATCH | 89 | +{ |
40 | +to = qemu-devel@nongnu.org | 90 | +} |
41 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | 91 | diff --git a/block/export/meson.build b/block/export/meson.build |
42 | + | 92 | index XXXXXXX..XXXXXXX 100644 |
43 | +[gitpublishprofile "stable"] | 93 | --- a/block/export/meson.build |
44 | +base = master | 94 | +++ b/block/export/meson.build |
45 | +to = qemu-devel@nongnu.org | 95 | @@ -XXX,XX +XXX,XX @@ |
46 | +cc = qemu-stable@nongnu.org | 96 | -block_ss.add(files('export.c')) |
47 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | 97 | -block_ss.add(when: ['CONFIG_LINUX', 'CONFIG_VHOST_USER'], if_true: files('vhost-user-blk-server.c')) |
48 | + | 98 | +blockdev_ss.add(files('export.c')) |
49 | +[gitpublishprofile "trivial"] | 99 | +blockdev_ss.add(when: ['CONFIG_LINUX', 'CONFIG_VHOST_USER'], if_true: files('vhost-user-blk-server.c')) |
50 | +base = master | 100 | diff --git a/meson.build b/meson.build |
51 | +to = qemu-devel@nongnu.org | 101 | index XXXXXXX..XXXXXXX 100644 |
52 | +cc = qemu-trivial@nongnu.org | 102 | --- a/meson.build |
53 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | 103 | +++ b/meson.build |
54 | + | 104 | @@ -XXX,XX +XXX,XX @@ subdir('dump') |
55 | +[gitpublishprofile "block"] | 105 | |
56 | +base = master | 106 | block_ss.add(files( |
57 | +to = qemu-devel@nongnu.org | 107 | 'block.c', |
58 | +cc = qemu-block@nongnu.org | 108 | - 'blockdev-nbd.c', |
59 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | 109 | 'blockjob.c', |
60 | + | 110 | 'job.c', |
61 | +[gitpublishprofile "arm"] | 111 | 'qemu-io-cmds.c', |
62 | +base = master | 112 | @@ -XXX,XX +XXX,XX @@ subdir('block') |
63 | +to = qemu-devel@nongnu.org | 113 | |
64 | +cc = qemu-arm@nongnu.org | 114 | blockdev_ss.add(files( |
65 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | 115 | 'blockdev.c', |
66 | + | 116 | + 'blockdev-nbd.c', |
67 | +[gitpublishprofile "s390"] | 117 | 'iothread.c', |
68 | +base = master | 118 | 'job-qmp.c', |
69 | +to = qemu-devel@nongnu.org | 119 | )) |
70 | +cc = qemu-s390@nongnu.org | 120 | @@ -XXX,XX +XXX,XX @@ if have_tools |
71 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | 121 | qemu_io = executable('qemu-io', files('qemu-io.c'), |
72 | + | 122 | dependencies: [block, qemuutil], install: true) |
73 | +[gitpublishprofile "ppc"] | 123 | qemu_nbd = executable('qemu-nbd', files('qemu-nbd.c'), |
74 | +base = master | 124 | - dependencies: [block, qemuutil], install: true) |
75 | +to = qemu-devel@nongnu.org | 125 | + dependencies: [blockdev, qemuutil], install: true) |
76 | +cc = qemu-ppc@nongnu.org | 126 | |
77 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | 127 | subdir('storage-daemon') |
128 | subdir('contrib/rdmacm-mux') | ||
129 | diff --git a/nbd/meson.build b/nbd/meson.build | ||
130 | index XXXXXXX..XXXXXXX 100644 | ||
131 | --- a/nbd/meson.build | ||
132 | +++ b/nbd/meson.build | ||
133 | @@ -XXX,XX +XXX,XX @@ | ||
134 | block_ss.add(files( | ||
135 | 'client.c', | ||
136 | 'common.c', | ||
137 | +)) | ||
138 | +blockdev_ss.add(files( | ||
139 | 'server.c', | ||
140 | )) | ||
141 | diff --git a/stubs/meson.build b/stubs/meson.build | ||
142 | index XXXXXXX..XXXXXXX 100644 | ||
143 | --- a/stubs/meson.build | ||
144 | +++ b/stubs/meson.build | ||
145 | @@ -XXX,XX +XXX,XX @@ | ||
146 | stub_ss.add(files('arch_type.c')) | ||
147 | stub_ss.add(files('bdrv-next-monitor-owned.c')) | ||
148 | stub_ss.add(files('blk-commit-all.c')) | ||
149 | +stub_ss.add(files('blk-exp-close-all.c')) | ||
150 | stub_ss.add(files('blockdev-close-all-bdrv-states.c')) | ||
151 | stub_ss.add(files('change-state-handler.c')) | ||
152 | stub_ss.add(files('cmos.c')) | ||
78 | -- | 153 | -- |
79 | 2.14.3 | 154 | 2.26.2 |
80 | 155 | ||
81 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | Make it possible to specify the iothread where the export will run. By | ||
2 | default the block node can be moved to other AioContexts later and the | ||
3 | export will follow. The fixed-iothread option forces strict behavior | ||
4 | that prevents changing AioContext while the export is active. See the | ||
5 | QAPI docs for details. | ||
1 | 6 | ||
7 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
8 | Message-id: 20200929125516.186715-5-stefanha@redhat.com | ||
9 | [Fix stray '#' character in block-export.json and add missing "(since: | ||
10 | 5.2)" as suggested by Eric Blake. | ||
11 | --Stefan] | ||
12 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
13 | --- | ||
14 | qapi/block-export.json | 11 ++++++++++ | ||
15 | block/export/export.c | 31 +++++++++++++++++++++++++++- | ||
16 | block/export/vhost-user-blk-server.c | 5 ++++- | ||
17 | nbd/server.c | 2 -- | ||
18 | 4 files changed, 45 insertions(+), 4 deletions(-) | ||
19 | |||
20 | diff --git a/qapi/block-export.json b/qapi/block-export.json | ||
21 | index XXXXXXX..XXXXXXX 100644 | ||
22 | --- a/qapi/block-export.json | ||
23 | +++ b/qapi/block-export.json | ||
24 | @@ -XXX,XX +XXX,XX @@ | ||
25 | # export before completion is signalled. (since: 5.2; | ||
26 | # default: false) | ||
27 | # | ||
28 | +# @iothread: The name of the iothread object where the export will run. The | ||
29 | +# default is to use the thread currently associated with the | ||
30 | +# block node. (since: 5.2) | ||
31 | +# | ||
32 | +# @fixed-iothread: True prevents the block node from being moved to another | ||
33 | +# thread while the export is active. If true and @iothread is | ||
34 | +# given, export creation fails if the block node cannot be | ||
35 | +# moved to the iothread. The default is false. (since: 5.2) | ||
36 | +# | ||
37 | # Since: 4.2 | ||
38 | ## | ||
39 | { 'union': 'BlockExportOptions', | ||
40 | 'base': { 'type': 'BlockExportType', | ||
41 | 'id': 'str', | ||
42 | + '*fixed-iothread': 'bool', | ||
43 | + '*iothread': 'str', | ||
44 | 'node-name': 'str', | ||
45 | '*writable': 'bool', | ||
46 | '*writethrough': 'bool' }, | ||
47 | diff --git a/block/export/export.c b/block/export/export.c | ||
48 | index XXXXXXX..XXXXXXX 100644 | ||
49 | --- a/block/export/export.c | ||
50 | +++ b/block/export/export.c | ||
51 | @@ -XXX,XX +XXX,XX @@ | ||
52 | |||
53 | #include "block/block.h" | ||
54 | #include "sysemu/block-backend.h" | ||
55 | +#include "sysemu/iothread.h" | ||
56 | #include "block/export.h" | ||
57 | #include "block/nbd.h" | ||
58 | #include "qapi/error.h" | ||
59 | @@ -XXX,XX +XXX,XX @@ static const BlockExportDriver *blk_exp_find_driver(BlockExportType type) | ||
60 | |||
61 | BlockExport *blk_exp_add(BlockExportOptions *export, Error **errp) | ||
62 | { | ||
63 | + bool fixed_iothread = export->has_fixed_iothread && export->fixed_iothread; | ||
64 | const BlockExportDriver *drv; | ||
65 | BlockExport *exp = NULL; | ||
66 | BlockDriverState *bs; | ||
67 | - BlockBackend *blk; | ||
68 | + BlockBackend *blk = NULL; | ||
69 | AioContext *ctx; | ||
70 | uint64_t perm; | ||
71 | int ret; | ||
72 | @@ -XXX,XX +XXX,XX @@ BlockExport *blk_exp_add(BlockExportOptions *export, Error **errp) | ||
73 | ctx = bdrv_get_aio_context(bs); | ||
74 | aio_context_acquire(ctx); | ||
75 | |||
76 | + if (export->has_iothread) { | ||
77 | + IOThread *iothread; | ||
78 | + AioContext *new_ctx; | ||
79 | + | ||
80 | + iothread = iothread_by_id(export->iothread); | ||
81 | + if (!iothread) { | ||
82 | + error_setg(errp, "iothread \"%s\" not found", export->iothread); | ||
83 | + goto fail; | ||
84 | + } | ||
85 | + | ||
86 | + new_ctx = iothread_get_aio_context(iothread); | ||
87 | + | ||
88 | + ret = bdrv_try_set_aio_context(bs, new_ctx, errp); | ||
89 | + if (ret == 0) { | ||
90 | + aio_context_release(ctx); | ||
91 | + aio_context_acquire(new_ctx); | ||
92 | + ctx = new_ctx; | ||
93 | + } else if (fixed_iothread) { | ||
94 | + goto fail; | ||
95 | + } | ||
96 | + } | ||
97 | + | ||
98 | /* | ||
99 | * Block exports are used for non-shared storage migration. Make sure | ||
100 | * that BDRV_O_INACTIVE is cleared and the image is ready for write | ||
101 | @@ -XXX,XX +XXX,XX @@ BlockExport *blk_exp_add(BlockExportOptions *export, Error **errp) | ||
102 | } | ||
103 | |||
104 | blk = blk_new(ctx, perm, BLK_PERM_ALL); | ||
105 | + | ||
106 | + if (!fixed_iothread) { | ||
107 | + blk_set_allow_aio_context_change(blk, true); | ||
108 | + } | ||
109 | + | ||
110 | ret = blk_insert_bs(blk, bs, errp); | ||
111 | if (ret < 0) { | ||
112 | goto fail; | ||
113 | diff --git a/block/export/vhost-user-blk-server.c b/block/export/vhost-user-blk-server.c | ||
114 | index XXXXXXX..XXXXXXX 100644 | ||
115 | --- a/block/export/vhost-user-blk-server.c | ||
116 | +++ b/block/export/vhost-user-blk-server.c | ||
117 | @@ -XXX,XX +XXX,XX @@ static const VuDevIface vu_blk_iface = { | ||
118 | static void blk_aio_attached(AioContext *ctx, void *opaque) | ||
119 | { | ||
120 | VuBlkExport *vexp = opaque; | ||
121 | + | ||
122 | + vexp->export.ctx = ctx; | ||
123 | vhost_user_server_attach_aio_context(&vexp->vu_server, ctx); | ||
124 | } | ||
125 | |||
126 | static void blk_aio_detach(void *opaque) | ||
127 | { | ||
128 | VuBlkExport *vexp = opaque; | ||
129 | + | ||
130 | vhost_user_server_detach_aio_context(&vexp->vu_server); | ||
131 | + vexp->export.ctx = NULL; | ||
132 | } | ||
133 | |||
134 | static void | ||
135 | @@ -XXX,XX +XXX,XX @@ static int vu_blk_exp_create(BlockExport *exp, BlockExportOptions *opts, | ||
136 | vu_blk_initialize_config(blk_bs(exp->blk), &vexp->blkcfg, | ||
137 | logical_block_size); | ||
138 | |||
139 | - blk_set_allow_aio_context_change(exp->blk, true); | ||
140 | blk_add_aio_context_notifier(exp->blk, blk_aio_attached, blk_aio_detach, | ||
141 | vexp); | ||
142 | |||
143 | diff --git a/nbd/server.c b/nbd/server.c | ||
144 | index XXXXXXX..XXXXXXX 100644 | ||
145 | --- a/nbd/server.c | ||
146 | +++ b/nbd/server.c | ||
147 | @@ -XXX,XX +XXX,XX @@ static int nbd_export_create(BlockExport *blk_exp, BlockExportOptions *exp_args, | ||
148 | return ret; | ||
149 | } | ||
150 | |||
151 | - blk_set_allow_aio_context_change(blk, true); | ||
152 | - | ||
153 | QTAILQ_INIT(&exp->clients); | ||
154 | exp->name = g_strdup(arg->name); | ||
155 | exp->description = g_strdup(arg->description); | ||
156 | -- | ||
157 | 2.26.2 | ||
158 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | Allow the number of queues to be configured using --export | ||
2 | vhost-user-blk,num-queues=N. This setting should match the QEMU --device | ||
3 | vhost-user-blk-pci,num-queues=N setting but QEMU vhost-user-blk.c lowers | ||
4 | its own value if the vhost-user-blk backend offers fewer queues than | ||
5 | QEMU. | ||
1 | 6 | ||
7 | The vhost-user-blk-server.c code is already capable of multi-queue. All | ||
8 | virtqueue processing runs in the same AioContext. No new locking is | ||
9 | needed. | ||
10 | |||
11 | Add the num-queues=N option and set the VIRTIO_BLK_F_MQ feature bit. | ||
12 | Note that the feature bit only announces the presence of the num_queues | ||
13 | configuration space field. It does not promise that there is more than 1 | ||
14 | virtqueue, so we can set it unconditionally. | ||
15 | |||
16 | I tested multi-queue by running a random read fio test with numjobs=4 on | ||
17 | an -smp 4 guest. After the benchmark finished the guest /proc/interrupts | ||
18 | file showed activity on all 4 virtio-blk MSI-X. The /sys/block/vda/mq/ | ||
19 | directory shows that Linux blk-mq has 4 queues configured. | ||
20 | |||
21 | An automated test is included in the next commit. | ||
22 | |||
23 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
24 | Acked-by: Markus Armbruster <armbru@redhat.com> | ||
25 | Message-id: 20201001144604.559733-2-stefanha@redhat.com | ||
26 | [Fixed accidental tab characters as suggested by Markus Armbruster | ||
27 | --Stefan] | ||
28 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
29 | --- | ||
30 | qapi/block-export.json | 10 +++++++--- | ||
31 | block/export/vhost-user-blk-server.c | 24 ++++++++++++++++++------ | ||
32 | 2 files changed, 25 insertions(+), 9 deletions(-) | ||
33 | |||
34 | diff --git a/qapi/block-export.json b/qapi/block-export.json | ||
35 | index XXXXXXX..XXXXXXX 100644 | ||
36 | --- a/qapi/block-export.json | ||
37 | +++ b/qapi/block-export.json | ||
38 | @@ -XXX,XX +XXX,XX @@ | ||
39 | # SocketAddress types are supported. Passed fds must be UNIX domain | ||
40 | # sockets. | ||
41 | # @logical-block-size: Logical block size in bytes. Defaults to 512 bytes. | ||
42 | +# @num-queues: Number of request virtqueues. Must be greater than 0. Defaults | ||
43 | +# to 1. | ||
44 | # | ||
45 | # Since: 5.2 | ||
46 | ## | ||
47 | { 'struct': 'BlockExportOptionsVhostUserBlk', | ||
48 | - 'data': { 'addr': 'SocketAddress', '*logical-block-size': 'size' } } | ||
49 | + 'data': { 'addr': 'SocketAddress', | ||
50 | + '*logical-block-size': 'size', | ||
51 | + '*num-queues': 'uint16'} } | ||
52 | |||
53 | ## | ||
54 | # @NbdServerAddOptions: | ||
55 | @@ -XXX,XX +XXX,XX @@ | ||
56 | { 'union': 'BlockExportOptions', | ||
57 | 'base': { 'type': 'BlockExportType', | ||
58 | 'id': 'str', | ||
59 | - '*fixed-iothread': 'bool', | ||
60 | - '*iothread': 'str', | ||
61 | + '*fixed-iothread': 'bool', | ||
62 | + '*iothread': 'str', | ||
63 | 'node-name': 'str', | ||
64 | '*writable': 'bool', | ||
65 | '*writethrough': 'bool' }, | ||
66 | diff --git a/block/export/vhost-user-blk-server.c b/block/export/vhost-user-blk-server.c | ||
67 | index XXXXXXX..XXXXXXX 100644 | ||
68 | --- a/block/export/vhost-user-blk-server.c | ||
69 | +++ b/block/export/vhost-user-blk-server.c | ||
70 | @@ -XXX,XX +XXX,XX @@ | ||
71 | #include "util/block-helpers.h" | ||
72 | |||
73 | enum { | ||
74 | - VHOST_USER_BLK_MAX_QUEUES = 1, | ||
75 | + VHOST_USER_BLK_NUM_QUEUES_DEFAULT = 1, | ||
76 | }; | ||
77 | struct virtio_blk_inhdr { | ||
78 | unsigned char status; | ||
79 | @@ -XXX,XX +XXX,XX @@ static uint64_t vu_blk_get_features(VuDev *dev) | ||
80 | 1ull << VIRTIO_BLK_F_DISCARD | | ||
81 | 1ull << VIRTIO_BLK_F_WRITE_ZEROES | | ||
82 | 1ull << VIRTIO_BLK_F_CONFIG_WCE | | ||
83 | + 1ull << VIRTIO_BLK_F_MQ | | ||
84 | 1ull << VIRTIO_F_VERSION_1 | | ||
85 | 1ull << VIRTIO_RING_F_INDIRECT_DESC | | ||
86 | 1ull << VIRTIO_RING_F_EVENT_IDX | | ||
87 | @@ -XXX,XX +XXX,XX @@ static void blk_aio_detach(void *opaque) | ||
88 | |||
89 | static void | ||
90 | vu_blk_initialize_config(BlockDriverState *bs, | ||
91 | - struct virtio_blk_config *config, uint32_t blk_size) | ||
92 | + struct virtio_blk_config *config, | ||
93 | + uint32_t blk_size, | ||
94 | + uint16_t num_queues) | ||
95 | { | ||
96 | config->capacity = bdrv_getlength(bs) >> BDRV_SECTOR_BITS; | ||
97 | config->blk_size = blk_size; | ||
98 | @@ -XXX,XX +XXX,XX @@ vu_blk_initialize_config(BlockDriverState *bs, | ||
99 | config->seg_max = 128 - 2; | ||
100 | config->min_io_size = 1; | ||
101 | config->opt_io_size = 1; | ||
102 | - config->num_queues = VHOST_USER_BLK_MAX_QUEUES; | ||
103 | + config->num_queues = num_queues; | ||
104 | config->max_discard_sectors = 32768; | ||
105 | config->max_discard_seg = 1; | ||
106 | config->discard_sector_alignment = config->blk_size >> 9; | ||
107 | @@ -XXX,XX +XXX,XX @@ static int vu_blk_exp_create(BlockExport *exp, BlockExportOptions *opts, | ||
108 | BlockExportOptionsVhostUserBlk *vu_opts = &opts->u.vhost_user_blk; | ||
109 | Error *local_err = NULL; | ||
110 | uint64_t logical_block_size; | ||
111 | + uint16_t num_queues = VHOST_USER_BLK_NUM_QUEUES_DEFAULT; | ||
112 | |||
113 | vexp->writable = opts->writable; | ||
114 | vexp->blkcfg.wce = 0; | ||
115 | @@ -XXX,XX +XXX,XX @@ static int vu_blk_exp_create(BlockExport *exp, BlockExportOptions *opts, | ||
116 | } | ||
117 | vexp->blk_size = logical_block_size; | ||
118 | blk_set_guest_block_size(exp->blk, logical_block_size); | ||
119 | + | ||
120 | + if (vu_opts->has_num_queues) { | ||
121 | + num_queues = vu_opts->num_queues; | ||
122 | + } | ||
123 | + if (num_queues == 0) { | ||
124 | + error_setg(errp, "num-queues must be greater than 0"); | ||
125 | + return -EINVAL; | ||
126 | + } | ||
127 | + | ||
128 | vu_blk_initialize_config(blk_bs(exp->blk), &vexp->blkcfg, | ||
129 | - logical_block_size); | ||
130 | + logical_block_size, num_queues); | ||
131 | |||
132 | blk_add_aio_context_notifier(exp->blk, blk_aio_attached, blk_aio_detach, | ||
133 | vexp); | ||
134 | |||
135 | if (!vhost_user_server_start(&vexp->vu_server, vu_opts->addr, exp->ctx, | ||
136 | - VHOST_USER_BLK_MAX_QUEUES, &vu_blk_iface, | ||
137 | - errp)) { | ||
138 | + num_queues, &vu_blk_iface, errp)) { | ||
139 | blk_remove_aio_context_notifier(exp->blk, blk_aio_attached, | ||
140 | blk_aio_detach, vexp); | ||
141 | return -EADDRNOTAVAIL; | ||
142 | -- | ||
143 | 2.26.2 | ||
144 | diff view generated by jsdifflib |
1 | From: Su Hang <suhang16@mails.ucas.ac.cn> | 1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> |
---|---|---|---|
2 | 2 | ||
3 | Using `clang-format -i util/uri.c` first, then change back few code | 3 | bdrv_co_block_status_above has several design problems with handling |
4 | manually, to make sure only whitespace involved. | 4 | short backing files: |
5 | 5 | ||
6 | Signed-off-by: Su Hang <suhang16@mails.ucas.ac.cn> | 6 | 1. With want_zeros=true, it may return ret with BDRV_BLOCK_ZERO but |
7 | Reviewed-by: Thomas Huth <thuth@redhat.com> | 7 | without BDRV_BLOCK_ALLOCATED flag, when actually short backing file |
8 | Message-id: 1519533358-13759-2-git-send-email-suhang16@mails.ucas.ac.cn | 8 | which produces these after-EOF zeros is inside requested backing |
9 | sequence. | ||
10 | |||
11 | 2. With want_zero=false, it may return pnum=0 prior to actual EOF, | ||
12 | because of EOF of short backing file. | ||
13 | |||
14 | Fix these things, making logic about short backing files clearer. | ||
15 | |||
16 | With fixed bdrv_block_status_above we also have to improve is_zero in | ||
17 | qcow2 code, otherwise iotest 154 will fail, because with this patch we | ||
18 | stop to merge zeros of different types (produced by fully unallocated | ||
19 | in the whole backing chain regions vs produced by short backing files). | ||
20 | |||
21 | Note also, that this patch leaves for another day the general problem | ||
22 | around block-status: misuse of BDRV_BLOCK_ALLOCATED as is-fs-allocated | ||
23 | vs go-to-backing. | ||
24 | |||
25 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
26 | Reviewed-by: Alberto Garcia <berto@igalia.com> | ||
27 | Reviewed-by: Eric Blake <eblake@redhat.com> | ||
28 | Message-id: 20200924194003.22080-2-vsementsov@virtuozzo.com | ||
29 | [Fix s/comes/come/ as suggested by Eric Blake | ||
30 | --Stefan] | ||
9 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 31 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
10 | --- | 32 | --- |
11 | util/uri.c | 1450 ++++++++++++++++++++++++++++++------------------------------ | 33 | block/io.c | 68 ++++++++++++++++++++++++++++++++++++++++----------- |
12 | 1 file changed, 726 insertions(+), 724 deletions(-) | 34 | block/qcow2.c | 16 ++++++++++-- |
35 | 2 files changed, 68 insertions(+), 16 deletions(-) | ||
13 | 36 | ||
14 | diff --git a/util/uri.c b/util/uri.c | 37 | diff --git a/block/io.c b/block/io.c |
15 | index XXXXXXX..XXXXXXX 100644 | 38 | index XXXXXXX..XXXXXXX 100644 |
16 | --- a/util/uri.c | 39 | --- a/block/io.c |
17 | +++ b/util/uri.c | 40 | +++ b/block/io.c |
18 | @@ -XXX,XX +XXX,XX @@ static void uri_clean(URI *uri); | 41 | @@ -XXX,XX +XXX,XX @@ bdrv_co_common_block_status_above(BlockDriverState *bs, |
19 | */ | 42 | int64_t *map, |
20 | #define IS_ALPHA(x) (IS_LOWALPHA(x) || IS_UPALPHA(x)) | 43 | BlockDriverState **file) |
21 | |||
22 | - | ||
23 | /* | ||
24 | * lowalpha = "a" | "b" | "c" | "d" | "e" | "f" | "g" | "h" | "i" | "j" | | ||
25 | * "k" | "l" | "m" | "n" | "o" | "p" | "q" | "r" | "s" | "t" | | ||
26 | @@ -XXX,XX +XXX,XX @@ static void uri_clean(URI *uri); | ||
27 | * mark = "-" | "_" | "." | "!" | "~" | "*" | "'" | "(" | ")" | ||
28 | */ | ||
29 | |||
30 | -#define IS_MARK(x) (((x) == '-') || ((x) == '_') || ((x) == '.') || \ | ||
31 | - ((x) == '!') || ((x) == '~') || ((x) == '*') || ((x) == '\'') || \ | ||
32 | +#define IS_MARK(x) (((x) == '-') || ((x) == '_') || ((x) == '.') || \ | ||
33 | + ((x) == '!') || ((x) == '~') || ((x) == '*') || ((x) == '\'') || \ | ||
34 | ((x) == '(') || ((x) == ')')) | ||
35 | |||
36 | /* | ||
37 | * unwise = "{" | "}" | "|" | "\" | "^" | "`" | ||
38 | */ | ||
39 | |||
40 | -#define IS_UNWISE(p) \ | ||
41 | - (((*(p) == '{')) || ((*(p) == '}')) || ((*(p) == '|')) || \ | ||
42 | - ((*(p) == '\\')) || ((*(p) == '^')) || ((*(p) == '[')) || \ | ||
43 | - ((*(p) == ']')) || ((*(p) == '`'))) | ||
44 | +#define IS_UNWISE(p) \ | ||
45 | + (((*(p) == '{')) || ((*(p) == '}')) || ((*(p) == '|')) || \ | ||
46 | + ((*(p) == '\\')) || ((*(p) == '^')) || ((*(p) == '[')) || \ | ||
47 | + ((*(p) == ']')) || ((*(p) == '`'))) | ||
48 | /* | ||
49 | * reserved = ";" | "/" | "?" | ":" | "@" | "&" | "=" | "+" | "$" | "," | | ||
50 | * "[" | "]" | ||
51 | */ | ||
52 | |||
53 | -#define IS_RESERVED(x) (((x) == ';') || ((x) == '/') || ((x) == '?') || \ | ||
54 | - ((x) == ':') || ((x) == '@') || ((x) == '&') || ((x) == '=') || \ | ||
55 | - ((x) == '+') || ((x) == '$') || ((x) == ',') || ((x) == '[') || \ | ||
56 | - ((x) == ']')) | ||
57 | +#define IS_RESERVED(x) (((x) == ';') || ((x) == '/') || ((x) == '?') || \ | ||
58 | + ((x) == ':') || ((x) == '@') || ((x) == '&') || ((x) == '=') || \ | ||
59 | + ((x) == '+') || ((x) == '$') || ((x) == ',') || ((x) == '[') || \ | ||
60 | + ((x) == ']')) | ||
61 | |||
62 | /* | ||
63 | * unreserved = alphanum | mark | ||
64 | @@ -XXX,XX +XXX,XX @@ static void uri_clean(URI *uri); | ||
65 | * Skip to next pointer char, handle escaped sequences | ||
66 | */ | ||
67 | |||
68 | -#define NEXT(p) ((*p == '%')? p += 3 : p++) | ||
69 | +#define NEXT(p) ((*p == '%') ? p += 3 : p++) | ||
70 | |||
71 | /* | ||
72 | * Productions from the spec. | ||
73 | @@ -XXX,XX +XXX,XX @@ static void uri_clean(URI *uri); | ||
74 | * path = [ abs_path | opaque_part ] | ||
75 | */ | ||
76 | |||
77 | - | ||
78 | /************************************************************************ | ||
79 | - * * | ||
80 | - * RFC 3986 parser * | ||
81 | - * * | ||
82 | + * * | ||
83 | + * RFC 3986 parser * | ||
84 | + * * | ||
85 | ************************************************************************/ | ||
86 | |||
87 | #define ISA_DIGIT(p) ((*(p) >= '0') && (*(p) <= '9')) | ||
88 | -#define ISA_ALPHA(p) (((*(p) >= 'a') && (*(p) <= 'z')) || \ | ||
89 | +#define ISA_ALPHA(p) (((*(p) >= 'a') && (*(p) <= 'z')) || \ | ||
90 | ((*(p) >= 'A') && (*(p) <= 'Z'))) | ||
91 | -#define ISA_HEXDIG(p) \ | ||
92 | - (ISA_DIGIT(p) || ((*(p) >= 'a') && (*(p) <= 'f')) || \ | ||
93 | - ((*(p) >= 'A') && (*(p) <= 'F'))) | ||
94 | +#define ISA_HEXDIG(p) \ | ||
95 | + (ISA_DIGIT(p) || ((*(p) >= 'a') && (*(p) <= 'f')) || \ | ||
96 | + ((*(p) >= 'A') && (*(p) <= 'F'))) | ||
97 | |||
98 | /* | ||
99 | * sub-delims = "!" / "$" / "&" / "'" / "(" / ")" | ||
100 | * / "*" / "+" / "," / ";" / "=" | ||
101 | */ | ||
102 | -#define ISA_SUB_DELIM(p) \ | ||
103 | - (((*(p) == '!')) || ((*(p) == '$')) || ((*(p) == '&')) || \ | ||
104 | - ((*(p) == '(')) || ((*(p) == ')')) || ((*(p) == '*')) || \ | ||
105 | - ((*(p) == '+')) || ((*(p) == ',')) || ((*(p) == ';')) || \ | ||
106 | - ((*(p) == '=')) || ((*(p) == '\''))) | ||
107 | +#define ISA_SUB_DELIM(p) \ | ||
108 | + (((*(p) == '!')) || ((*(p) == '$')) || ((*(p) == '&')) || \ | ||
109 | + ((*(p) == '(')) || ((*(p) == ')')) || ((*(p) == '*')) || \ | ||
110 | + ((*(p) == '+')) || ((*(p) == ',')) || ((*(p) == ';')) || \ | ||
111 | + ((*(p) == '=')) || ((*(p) == '\''))) | ||
112 | |||
113 | /* | ||
114 | * gen-delims = ":" / "/" / "?" / "#" / "[" / "]" / "@" | ||
115 | */ | ||
116 | -#define ISA_GEN_DELIM(p) \ | ||
117 | - (((*(p) == ':')) || ((*(p) == '/')) || ((*(p) == '?')) || \ | ||
118 | - ((*(p) == '#')) || ((*(p) == '[')) || ((*(p) == ']')) || \ | ||
119 | - ((*(p) == '@'))) | ||
120 | +#define ISA_GEN_DELIM(p) \ | ||
121 | + (((*(p) == ':')) || ((*(p) == '/')) || ((*(p) == '?')) || \ | ||
122 | + ((*(p) == '#')) || ((*(p) == '[')) || ((*(p) == ']')) || \ | ||
123 | + ((*(p) == '@'))) | ||
124 | |||
125 | /* | ||
126 | * reserved = gen-delims / sub-delims | ||
127 | @@ -XXX,XX +XXX,XX @@ static void uri_clean(URI *uri); | ||
128 | /* | ||
129 | * unreserved = ALPHA / DIGIT / "-" / "." / "_" / "~" | ||
130 | */ | ||
131 | -#define ISA_UNRESERVED(p) \ | ||
132 | - ((ISA_ALPHA(p)) || (ISA_DIGIT(p)) || ((*(p) == '-')) || \ | ||
133 | - ((*(p) == '.')) || ((*(p) == '_')) || ((*(p) == '~'))) | ||
134 | +#define ISA_UNRESERVED(p) \ | ||
135 | + ((ISA_ALPHA(p)) || (ISA_DIGIT(p)) || ((*(p) == '-')) || \ | ||
136 | + ((*(p) == '.')) || ((*(p) == '_')) || ((*(p) == '~'))) | ||
137 | |||
138 | /* | ||
139 | * pct-encoded = "%" HEXDIG HEXDIG | ||
140 | */ | ||
141 | -#define ISA_PCT_ENCODED(p) \ | ||
142 | - ((*(p) == '%') && (ISA_HEXDIG(p + 1)) && (ISA_HEXDIG(p + 2))) | ||
143 | +#define ISA_PCT_ENCODED(p) \ | ||
144 | + ((*(p) == '%') && (ISA_HEXDIG(p + 1)) && (ISA_HEXDIG(p + 2))) | ||
145 | |||
146 | /* | ||
147 | * pchar = unreserved / pct-encoded / sub-delims / ":" / "@" | ||
148 | */ | ||
149 | -#define ISA_PCHAR(p) \ | ||
150 | - (ISA_UNRESERVED(p) || ISA_PCT_ENCODED(p) || ISA_SUB_DELIM(p) || \ | ||
151 | - ((*(p) == ':')) || ((*(p) == '@'))) | ||
152 | +#define ISA_PCHAR(p) \ | ||
153 | + (ISA_UNRESERVED(p) || ISA_PCT_ENCODED(p) || ISA_SUB_DELIM(p) || \ | ||
154 | + ((*(p) == ':')) || ((*(p) == '@'))) | ||
155 | |||
156 | /** | ||
157 | * rfc3986_parse_scheme: | ||
158 | @@ -XXX,XX +XXX,XX @@ static void uri_clean(URI *uri); | ||
159 | * | ||
160 | * Returns 0 or the error code | ||
161 | */ | ||
162 | -static int | ||
163 | -rfc3986_parse_scheme(URI *uri, const char **str) { | ||
164 | +static int rfc3986_parse_scheme(URI *uri, const char **str) | ||
165 | +{ | ||
166 | const char *cur; | ||
167 | |||
168 | if (str == NULL) | ||
169 | - return(-1); | ||
170 | + return (-1); | ||
171 | |||
172 | cur = *str; | ||
173 | if (!ISA_ALPHA(cur)) | ||
174 | - return(2); | ||
175 | + return (2); | ||
176 | cur++; | ||
177 | - while (ISA_ALPHA(cur) || ISA_DIGIT(cur) || | ||
178 | - (*cur == '+') || (*cur == '-') || (*cur == '.')) cur++; | ||
179 | + while (ISA_ALPHA(cur) || ISA_DIGIT(cur) || (*cur == '+') || (*cur == '-') || | ||
180 | + (*cur == '.')) | ||
181 | + cur++; | ||
182 | if (uri != NULL) { | ||
183 | g_free(uri->scheme); | ||
184 | - uri->scheme = g_strndup(*str, cur - *str); | ||
185 | + uri->scheme = g_strndup(*str, cur - *str); | ||
186 | } | ||
187 | *str = cur; | ||
188 | - return(0); | ||
189 | + return (0); | ||
190 | } | ||
191 | |||
192 | /** | ||
193 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_scheme(URI *uri, const char **str) { | ||
194 | * | ||
195 | * Returns 0 or the error code | ||
196 | */ | ||
197 | -static int | ||
198 | -rfc3986_parse_fragment(URI *uri, const char **str) | ||
199 | +static int rfc3986_parse_fragment(URI *uri, const char **str) | ||
200 | { | 44 | { |
201 | const char *cur; | 45 | + int ret; |
202 | 46 | BlockDriverState *p; | |
203 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_fragment(URI *uri, const char **str) | 47 | - int ret = 0; |
204 | NEXT(cur); | 48 | - bool first = true; |
205 | if (uri != NULL) { | 49 | + int64_t eof = 0; |
206 | g_free(uri->fragment); | 50 | |
207 | - if (uri->cleanup & 2) | 51 | assert(bs != base); |
208 | - uri->fragment = g_strndup(*str, cur - *str); | 52 | - for (p = bs; p != base; p = bdrv_filter_or_cow_bs(p)) { |
209 | - else | 53 | + |
210 | - uri->fragment = uri_string_unescape(*str, cur - *str, NULL); | 54 | + ret = bdrv_co_block_status(bs, want_zero, offset, bytes, pnum, map, file); |
211 | + if (uri->cleanup & 2) | 55 | + if (ret < 0 || *pnum == 0 || ret & BDRV_BLOCK_ALLOCATED) { |
212 | + uri->fragment = g_strndup(*str, cur - *str); | 56 | + return ret; |
213 | + else | 57 | + } |
214 | + uri->fragment = uri_string_unescape(*str, cur - *str, NULL); | 58 | + |
215 | } | 59 | + if (ret & BDRV_BLOCK_EOF) { |
216 | *str = cur; | 60 | + eof = offset + *pnum; |
217 | return (0); | 61 | + } |
218 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_fragment(URI *uri, const char **str) | 62 | + |
219 | * | 63 | + assert(*pnum <= bytes); |
220 | * Returns 0 or the error code | 64 | + bytes = *pnum; |
221 | */ | 65 | + |
222 | -static int | 66 | + for (p = bdrv_filter_or_cow_bs(bs); p != base; |
223 | -rfc3986_parse_query(URI *uri, const char **str) | 67 | + p = bdrv_filter_or_cow_bs(p)) |
224 | +static int rfc3986_parse_query(URI *uri, const char **str) | 68 | + { |
225 | { | 69 | ret = bdrv_co_block_status(p, want_zero, offset, bytes, pnum, map, |
226 | const char *cur; | 70 | file); |
227 | 71 | if (ret < 0) { | |
228 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_query(URI *uri, const char **str) | 72 | - break; |
229 | NEXT(cur); | 73 | + return ret; |
230 | if (uri != NULL) { | 74 | } |
231 | g_free(uri->query); | 75 | - if (ret & BDRV_BLOCK_ZERO && ret & BDRV_BLOCK_EOF && !first) { |
232 | - uri->query = g_strndup (*str, cur - *str); | 76 | + if (*pnum == 0) { |
233 | + uri->query = g_strndup(*str, cur - *str); | 77 | /* |
234 | } | 78 | - * Reading beyond the end of the file continues to read |
235 | *str = cur; | 79 | - * zeroes, but we can only widen the result to the |
236 | return (0); | 80 | - * unallocated length we learned from an earlier |
237 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_query(URI *uri, const char **str) | 81 | - * iteration. |
238 | * | 82 | + * The top layer deferred to this layer, and because this layer is |
239 | * Returns 0 or the error code | 83 | + * short, any zeroes that we synthesize beyond EOF behave as if they |
240 | */ | 84 | + * were allocated at this layer. |
241 | -static int | 85 | + * |
242 | -rfc3986_parse_port(URI *uri, const char **str) | 86 | + * We don't include BDRV_BLOCK_EOF into ret, as upper layer may be |
243 | +static int rfc3986_parse_port(URI *uri, const char **str) | 87 | + * larger. We'll add BDRV_BLOCK_EOF if needed at function end, see |
244 | { | 88 | + * below. |
245 | const char *cur = *str; | 89 | */ |
246 | int port = 0; | 90 | + assert(ret & BDRV_BLOCK_EOF); |
247 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_port(URI *uri, const char **str) | 91 | *pnum = bytes; |
248 | * | 92 | + if (file) { |
249 | * Returns 0 or the error code | 93 | + *file = p; |
250 | */ | ||
251 | -static int | ||
252 | -rfc3986_parse_user_info(URI *uri, const char **str) | ||
253 | +static int rfc3986_parse_user_info(URI *uri, const char **str) | ||
254 | { | ||
255 | const char *cur; | ||
256 | |||
257 | cur = *str; | ||
258 | - while (ISA_UNRESERVED(cur) || ISA_PCT_ENCODED(cur) || | ||
259 | - ISA_SUB_DELIM(cur) || (*cur == ':')) | ||
260 | - NEXT(cur); | ||
261 | + while (ISA_UNRESERVED(cur) || ISA_PCT_ENCODED(cur) || ISA_SUB_DELIM(cur) || | ||
262 | + (*cur == ':')) | ||
263 | + NEXT(cur); | ||
264 | if (*cur == '@') { | ||
265 | - if (uri != NULL) { | ||
266 | + if (uri != NULL) { | ||
267 | g_free(uri->user); | ||
268 | - if (uri->cleanup & 2) | ||
269 | - uri->user = g_strndup(*str, cur - *str); | ||
270 | - else | ||
271 | - uri->user = uri_string_unescape(*str, cur - *str, NULL); | ||
272 | - } | ||
273 | - *str = cur; | ||
274 | - return(0); | ||
275 | + if (uri->cleanup & 2) | ||
276 | + uri->user = g_strndup(*str, cur - *str); | ||
277 | + else | ||
278 | + uri->user = uri_string_unescape(*str, cur - *str, NULL); | ||
279 | + } | ||
280 | + *str = cur; | ||
281 | + return (0); | ||
282 | } | ||
283 | - return(1); | ||
284 | + return (1); | ||
285 | } | ||
286 | |||
287 | /** | ||
288 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_user_info(URI *uri, const char **str) | ||
289 | * | ||
290 | * Returns 0 if found and skipped, 1 otherwise | ||
291 | */ | ||
292 | -static int | ||
293 | -rfc3986_parse_dec_octet(const char **str) { | ||
294 | +static int rfc3986_parse_dec_octet(const char **str) | ||
295 | +{ | ||
296 | const char *cur = *str; | ||
297 | |||
298 | if (!(ISA_DIGIT(cur))) | ||
299 | - return(1); | ||
300 | - if (!ISA_DIGIT(cur+1)) | ||
301 | - cur++; | ||
302 | - else if ((*cur != '0') && (ISA_DIGIT(cur + 1)) && (!ISA_DIGIT(cur+2))) | ||
303 | - cur += 2; | ||
304 | + return (1); | ||
305 | + if (!ISA_DIGIT(cur + 1)) | ||
306 | + cur++; | ||
307 | + else if ((*cur != '0') && (ISA_DIGIT(cur + 1)) && (!ISA_DIGIT(cur + 2))) | ||
308 | + cur += 2; | ||
309 | else if ((*cur == '1') && (ISA_DIGIT(cur + 1)) && (ISA_DIGIT(cur + 2))) | ||
310 | - cur += 3; | ||
311 | - else if ((*cur == '2') && (*(cur + 1) >= '0') && | ||
312 | - (*(cur + 1) <= '4') && (ISA_DIGIT(cur + 2))) | ||
313 | - cur += 3; | ||
314 | - else if ((*cur == '2') && (*(cur + 1) == '5') && | ||
315 | - (*(cur + 2) >= '0') && (*(cur + 1) <= '5')) | ||
316 | - cur += 3; | ||
317 | + cur += 3; | ||
318 | + else if ((*cur == '2') && (*(cur + 1) >= '0') && (*(cur + 1) <= '4') && | ||
319 | + (ISA_DIGIT(cur + 2))) | ||
320 | + cur += 3; | ||
321 | + else if ((*cur == '2') && (*(cur + 1) == '5') && (*(cur + 2) >= '0') && | ||
322 | + (*(cur + 1) <= '5')) | ||
323 | + cur += 3; | ||
324 | else | ||
325 | - return(1); | ||
326 | + return (1); | ||
327 | *str = cur; | ||
328 | - return(0); | ||
329 | + return (0); | ||
330 | } | ||
331 | /** | ||
332 | * rfc3986_parse_host: | ||
333 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_dec_octet(const char **str) { | ||
334 | * | ||
335 | * Returns 0 or the error code | ||
336 | */ | ||
337 | -static int | ||
338 | -rfc3986_parse_host(URI *uri, const char **str) | ||
339 | +static int rfc3986_parse_host(URI *uri, const char **str) | ||
340 | { | ||
341 | const char *cur = *str; | ||
342 | const char *host; | ||
343 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_host(URI *uri, const char **str) | ||
344 | */ | ||
345 | if (*cur == '[') { | ||
346 | cur++; | ||
347 | - while ((*cur != ']') && (*cur != 0)) | ||
348 | - cur++; | ||
349 | - if (*cur != ']') | ||
350 | - return(1); | ||
351 | - cur++; | ||
352 | - goto found; | ||
353 | + while ((*cur != ']') && (*cur != 0)) | ||
354 | + cur++; | ||
355 | + if (*cur != ']') | ||
356 | + return (1); | ||
357 | + cur++; | ||
358 | + goto found; | ||
359 | } | ||
360 | /* | ||
361 | * try to parse an IPv4 | ||
362 | */ | ||
363 | if (ISA_DIGIT(cur)) { | ||
364 | if (rfc3986_parse_dec_octet(&cur) != 0) | ||
365 | - goto not_ipv4; | ||
366 | - if (*cur != '.') | ||
367 | - goto not_ipv4; | ||
368 | - cur++; | ||
369 | + goto not_ipv4; | ||
370 | + if (*cur != '.') | ||
371 | + goto not_ipv4; | ||
372 | + cur++; | ||
373 | if (rfc3986_parse_dec_octet(&cur) != 0) | ||
374 | - goto not_ipv4; | ||
375 | - if (*cur != '.') | ||
376 | - goto not_ipv4; | ||
377 | + goto not_ipv4; | ||
378 | + if (*cur != '.') | ||
379 | + goto not_ipv4; | ||
380 | if (rfc3986_parse_dec_octet(&cur) != 0) | ||
381 | - goto not_ipv4; | ||
382 | - if (*cur != '.') | ||
383 | - goto not_ipv4; | ||
384 | + goto not_ipv4; | ||
385 | + if (*cur != '.') | ||
386 | + goto not_ipv4; | ||
387 | if (rfc3986_parse_dec_octet(&cur) != 0) | ||
388 | - goto not_ipv4; | ||
389 | - goto found; | ||
390 | -not_ipv4: | ||
391 | + goto not_ipv4; | ||
392 | + goto found; | ||
393 | + not_ipv4: | ||
394 | cur = *str; | ||
395 | } | ||
396 | /* | ||
397 | @@ -XXX,XX +XXX,XX @@ not_ipv4: | ||
398 | found: | ||
399 | if (uri != NULL) { | ||
400 | g_free(uri->authority); | ||
401 | - uri->authority = NULL; | ||
402 | + uri->authority = NULL; | ||
403 | g_free(uri->server); | ||
404 | - if (cur != host) { | ||
405 | - if (uri->cleanup & 2) | ||
406 | - uri->server = g_strndup(host, cur - host); | ||
407 | - else | ||
408 | - uri->server = uri_string_unescape(host, cur - host, NULL); | ||
409 | - } else | ||
410 | - uri->server = NULL; | ||
411 | + if (cur != host) { | ||
412 | + if (uri->cleanup & 2) | ||
413 | + uri->server = g_strndup(host, cur - host); | ||
414 | + else | ||
415 | + uri->server = uri_string_unescape(host, cur - host, NULL); | ||
416 | + } else | ||
417 | + uri->server = NULL; | ||
418 | } | ||
419 | *str = cur; | ||
420 | - return(0); | ||
421 | + return (0); | ||
422 | } | ||
423 | |||
424 | /** | ||
425 | @@ -XXX,XX +XXX,XX @@ found: | ||
426 | * | ||
427 | * Returns 0 or the error code | ||
428 | */ | ||
429 | -static int | ||
430 | -rfc3986_parse_authority(URI *uri, const char **str) | ||
431 | +static int rfc3986_parse_authority(URI *uri, const char **str) | ||
432 | { | ||
433 | const char *cur; | ||
434 | int ret; | ||
435 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_authority(URI *uri, const char **str) | ||
436 | else | ||
437 | cur++; | ||
438 | ret = rfc3986_parse_host(uri, &cur); | ||
439 | - if (ret != 0) return(ret); | ||
440 | + if (ret != 0) | ||
441 | + return (ret); | ||
442 | if (*cur == ':') { | ||
443 | cur++; | ||
444 | ret = rfc3986_parse_port(uri, &cur); | ||
445 | - if (ret != 0) return(ret); | ||
446 | + if (ret != 0) | ||
447 | + return (ret); | ||
448 | } | ||
449 | *str = cur; | ||
450 | - return(0); | ||
451 | + return (0); | ||
452 | } | ||
453 | |||
454 | /** | ||
455 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_authority(URI *uri, const char **str) | ||
456 | * | ||
457 | * Returns 0 or the error code | ||
458 | */ | ||
459 | -static int | ||
460 | -rfc3986_parse_segment(const char **str, char forbid, int empty) | ||
461 | +static int rfc3986_parse_segment(const char **str, char forbid, int empty) | ||
462 | { | ||
463 | const char *cur; | ||
464 | |||
465 | cur = *str; | ||
466 | if (!ISA_PCHAR(cur)) { | ||
467 | if (empty) | ||
468 | - return(0); | ||
469 | - return(1); | ||
470 | + return (0); | ||
471 | + return (1); | ||
472 | } | ||
473 | while (ISA_PCHAR(cur) && (*cur != forbid)) | ||
474 | NEXT(cur); | ||
475 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_segment(const char **str, char forbid, int empty) | ||
476 | * | ||
477 | * Returns 0 or the error code | ||
478 | */ | ||
479 | -static int | ||
480 | -rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
481 | +static int rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
482 | { | ||
483 | const char *cur; | ||
484 | int ret; | ||
485 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
486 | |||
487 | while (*cur == '/') { | ||
488 | cur++; | ||
489 | - ret = rfc3986_parse_segment(&cur, 0, 1); | ||
490 | - if (ret != 0) return(ret); | ||
491 | + ret = rfc3986_parse_segment(&cur, 0, 1); | ||
492 | + if (ret != 0) | ||
493 | + return (ret); | ||
494 | } | ||
495 | if (uri != NULL) { | ||
496 | g_free(uri->path); | ||
497 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
498 | * | ||
499 | * Returns 0 or the error code | ||
500 | */ | ||
501 | -static int | ||
502 | -rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
503 | +static int rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
504 | { | ||
505 | const char *cur; | ||
506 | int ret; | ||
507 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
508 | cur = *str; | ||
509 | |||
510 | if (*cur != '/') | ||
511 | - return(1); | ||
512 | + return (1); | ||
513 | cur++; | ||
514 | ret = rfc3986_parse_segment(&cur, 0, 0); | ||
515 | if (ret == 0) { | ||
516 | - while (*cur == '/') { | ||
517 | - cur++; | ||
518 | - ret = rfc3986_parse_segment(&cur, 0, 1); | ||
519 | - if (ret != 0) return(ret); | ||
520 | - } | ||
521 | + while (*cur == '/') { | ||
522 | + cur++; | ||
523 | + ret = rfc3986_parse_segment(&cur, 0, 1); | ||
524 | + if (ret != 0) | ||
525 | + return (ret); | ||
526 | + } | ||
527 | } | ||
528 | if (uri != NULL) { | ||
529 | g_free(uri->path); | ||
530 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
531 | * | ||
532 | * Returns 0 or the error code | ||
533 | */ | ||
534 | -static int | ||
535 | -rfc3986_parse_path_rootless(URI *uri, const char **str) | ||
536 | +static int rfc3986_parse_path_rootless(URI *uri, const char **str) | ||
537 | { | ||
538 | const char *cur; | ||
539 | int ret; | ||
540 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_rootless(URI *uri, const char **str) | ||
541 | cur = *str; | ||
542 | |||
543 | ret = rfc3986_parse_segment(&cur, 0, 0); | ||
544 | - if (ret != 0) return(ret); | ||
545 | + if (ret != 0) | ||
546 | + return (ret); | ||
547 | while (*cur == '/') { | ||
548 | cur++; | ||
549 | - ret = rfc3986_parse_segment(&cur, 0, 1); | ||
550 | - if (ret != 0) return(ret); | ||
551 | + ret = rfc3986_parse_segment(&cur, 0, 1); | ||
552 | + if (ret != 0) | ||
553 | + return (ret); | ||
554 | } | ||
555 | if (uri != NULL) { | ||
556 | g_free(uri->path); | ||
557 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_rootless(URI *uri, const char **str) | ||
558 | * | ||
559 | * Returns 0 or the error code | ||
560 | */ | ||
561 | -static int | ||
562 | -rfc3986_parse_path_no_scheme(URI *uri, const char **str) | ||
563 | +static int rfc3986_parse_path_no_scheme(URI *uri, const char **str) | ||
564 | { | ||
565 | const char *cur; | ||
566 | int ret; | ||
567 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_no_scheme(URI *uri, const char **str) | ||
568 | cur = *str; | ||
569 | |||
570 | ret = rfc3986_parse_segment(&cur, ':', 0); | ||
571 | - if (ret != 0) return(ret); | ||
572 | + if (ret != 0) | ||
573 | + return (ret); | ||
574 | while (*cur == '/') { | ||
575 | cur++; | ||
576 | - ret = rfc3986_parse_segment(&cur, 0, 1); | ||
577 | - if (ret != 0) return(ret); | ||
578 | + ret = rfc3986_parse_segment(&cur, 0, 1); | ||
579 | + if (ret != 0) | ||
580 | + return (ret); | ||
581 | } | ||
582 | if (uri != NULL) { | ||
583 | g_free(uri->path); | ||
584 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_no_scheme(URI *uri, const char **str) | ||
585 | * | ||
586 | * Returns 0 or the error code | ||
587 | */ | ||
588 | -static int | ||
589 | -rfc3986_parse_hier_part(URI *uri, const char **str) | ||
590 | +static int rfc3986_parse_hier_part(URI *uri, const char **str) | ||
591 | { | ||
592 | const char *cur; | ||
593 | int ret; | ||
594 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_hier_part(URI *uri, const char **str) | ||
595 | |||
596 | if ((*cur == '/') && (*(cur + 1) == '/')) { | ||
597 | cur += 2; | ||
598 | - ret = rfc3986_parse_authority(uri, &cur); | ||
599 | - if (ret != 0) return(ret); | ||
600 | - ret = rfc3986_parse_path_ab_empty(uri, &cur); | ||
601 | - if (ret != 0) return(ret); | ||
602 | - *str = cur; | ||
603 | - return(0); | ||
604 | + ret = rfc3986_parse_authority(uri, &cur); | ||
605 | + if (ret != 0) | ||
606 | + return (ret); | ||
607 | + ret = rfc3986_parse_path_ab_empty(uri, &cur); | ||
608 | + if (ret != 0) | ||
609 | + return (ret); | ||
610 | + *str = cur; | ||
611 | + return (0); | ||
612 | } else if (*cur == '/') { | ||
613 | ret = rfc3986_parse_path_absolute(uri, &cur); | ||
614 | - if (ret != 0) return(ret); | ||
615 | + if (ret != 0) | ||
616 | + return (ret); | ||
617 | } else if (ISA_PCHAR(cur)) { | ||
618 | ret = rfc3986_parse_path_rootless(uri, &cur); | ||
619 | - if (ret != 0) return(ret); | ||
620 | + if (ret != 0) | ||
621 | + return (ret); | ||
622 | } else { | ||
623 | - /* path-empty is effectively empty */ | ||
624 | - if (uri != NULL) { | ||
625 | + /* path-empty is effectively empty */ | ||
626 | + if (uri != NULL) { | ||
627 | g_free(uri->path); | ||
628 | - uri->path = NULL; | ||
629 | - } | ||
630 | + uri->path = NULL; | ||
631 | + } | ||
632 | } | ||
633 | *str = cur; | ||
634 | return (0); | ||
635 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_hier_part(URI *uri, const char **str) | ||
636 | * | ||
637 | * Returns 0 or the error code | ||
638 | */ | ||
639 | -static int | ||
640 | -rfc3986_parse_relative_ref(URI *uri, const char *str) { | ||
641 | +static int rfc3986_parse_relative_ref(URI *uri, const char *str) | ||
642 | +{ | ||
643 | int ret; | ||
644 | |||
645 | if ((*str == '/') && (*(str + 1) == '/')) { | ||
646 | str += 2; | ||
647 | - ret = rfc3986_parse_authority(uri, &str); | ||
648 | - if (ret != 0) return(ret); | ||
649 | - ret = rfc3986_parse_path_ab_empty(uri, &str); | ||
650 | - if (ret != 0) return(ret); | ||
651 | + ret = rfc3986_parse_authority(uri, &str); | ||
652 | + if (ret != 0) | ||
653 | + return (ret); | ||
654 | + ret = rfc3986_parse_path_ab_empty(uri, &str); | ||
655 | + if (ret != 0) | ||
656 | + return (ret); | ||
657 | } else if (*str == '/') { | ||
658 | - ret = rfc3986_parse_path_absolute(uri, &str); | ||
659 | - if (ret != 0) return(ret); | ||
660 | + ret = rfc3986_parse_path_absolute(uri, &str); | ||
661 | + if (ret != 0) | ||
662 | + return (ret); | ||
663 | } else if (ISA_PCHAR(str)) { | ||
664 | ret = rfc3986_parse_path_no_scheme(uri, &str); | ||
665 | - if (ret != 0) return(ret); | ||
666 | + if (ret != 0) | ||
667 | + return (ret); | ||
668 | } else { | ||
669 | - /* path-empty is effectively empty */ | ||
670 | - if (uri != NULL) { | ||
671 | + /* path-empty is effectively empty */ | ||
672 | + if (uri != NULL) { | ||
673 | g_free(uri->path); | ||
674 | - uri->path = NULL; | ||
675 | - } | ||
676 | + uri->path = NULL; | ||
677 | + } | ||
678 | } | ||
679 | |||
680 | if (*str == '?') { | ||
681 | - str++; | ||
682 | - ret = rfc3986_parse_query(uri, &str); | ||
683 | - if (ret != 0) return(ret); | ||
684 | + str++; | ||
685 | + ret = rfc3986_parse_query(uri, &str); | ||
686 | + if (ret != 0) | ||
687 | + return (ret); | ||
688 | } | ||
689 | if (*str == '#') { | ||
690 | - str++; | ||
691 | - ret = rfc3986_parse_fragment(uri, &str); | ||
692 | - if (ret != 0) return(ret); | ||
693 | + str++; | ||
694 | + ret = rfc3986_parse_fragment(uri, &str); | ||
695 | + if (ret != 0) | ||
696 | + return (ret); | ||
697 | } | ||
698 | if (*str != 0) { | ||
699 | - uri_clean(uri); | ||
700 | - return(1); | ||
701 | + uri_clean(uri); | ||
702 | + return (1); | ||
703 | } | ||
704 | - return(0); | ||
705 | + return (0); | ||
706 | } | ||
707 | |||
708 | - | ||
709 | /** | ||
710 | * rfc3986_parse: | ||
711 | * @uri: pointer to an URI structure | ||
712 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_relative_ref(URI *uri, const char *str) { | ||
713 | * | ||
714 | * Returns 0 or the error code | ||
715 | */ | ||
716 | -static int | ||
717 | -rfc3986_parse(URI *uri, const char *str) { | ||
718 | +static int rfc3986_parse(URI *uri, const char *str) | ||
719 | +{ | ||
720 | int ret; | ||
721 | |||
722 | ret = rfc3986_parse_scheme(uri, &str); | ||
723 | - if (ret != 0) return(ret); | ||
724 | + if (ret != 0) | ||
725 | + return (ret); | ||
726 | if (*str != ':') { | ||
727 | - return(1); | ||
728 | + return (1); | ||
729 | } | ||
730 | str++; | ||
731 | ret = rfc3986_parse_hier_part(uri, &str); | ||
732 | - if (ret != 0) return(ret); | ||
733 | + if (ret != 0) | ||
734 | + return (ret); | ||
735 | if (*str == '?') { | ||
736 | - str++; | ||
737 | - ret = rfc3986_parse_query(uri, &str); | ||
738 | - if (ret != 0) return(ret); | ||
739 | + str++; | ||
740 | + ret = rfc3986_parse_query(uri, &str); | ||
741 | + if (ret != 0) | ||
742 | + return (ret); | ||
743 | } | ||
744 | if (*str == '#') { | ||
745 | - str++; | ||
746 | - ret = rfc3986_parse_fragment(uri, &str); | ||
747 | - if (ret != 0) return(ret); | ||
748 | + str++; | ||
749 | + ret = rfc3986_parse_fragment(uri, &str); | ||
750 | + if (ret != 0) | ||
751 | + return (ret); | ||
752 | } | ||
753 | if (*str != 0) { | ||
754 | - uri_clean(uri); | ||
755 | - return(1); | ||
756 | + uri_clean(uri); | ||
757 | + return (1); | ||
758 | } | ||
759 | - return(0); | ||
760 | + return (0); | ||
761 | } | ||
762 | |||
763 | /** | ||
764 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse(URI *uri, const char *str) { | ||
765 | * | ||
766 | * Returns 0 or the error code | ||
767 | */ | ||
768 | -static int | ||
769 | -rfc3986_parse_uri_reference(URI *uri, const char *str) { | ||
770 | +static int rfc3986_parse_uri_reference(URI *uri, const char *str) | ||
771 | +{ | ||
772 | int ret; | ||
773 | |||
774 | if (str == NULL) | ||
775 | - return(-1); | ||
776 | + return (-1); | ||
777 | uri_clean(uri); | ||
778 | |||
779 | /* | ||
780 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_uri_reference(URI *uri, const char *str) { | ||
781 | */ | ||
782 | ret = rfc3986_parse(uri, str); | ||
783 | if (ret != 0) { | ||
784 | - uri_clean(uri); | ||
785 | + uri_clean(uri); | ||
786 | ret = rfc3986_parse_relative_ref(uri, str); | ||
787 | - if (ret != 0) { | ||
788 | - uri_clean(uri); | ||
789 | - return(ret); | ||
790 | - } | ||
791 | + if (ret != 0) { | ||
792 | + uri_clean(uri); | ||
793 | + return (ret); | ||
794 | + } | ||
795 | } | ||
796 | - return(0); | ||
797 | + return (0); | ||
798 | } | ||
799 | |||
800 | /** | ||
801 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_uri_reference(URI *uri, const char *str) { | ||
802 | * | ||
803 | * Returns a newly built URI or NULL in case of error | ||
804 | */ | ||
805 | -URI * | ||
806 | -uri_parse(const char *str) { | ||
807 | +URI *uri_parse(const char *str) | ||
808 | +{ | ||
809 | URI *uri; | ||
810 | int ret; | ||
811 | |||
812 | if (str == NULL) | ||
813 | - return(NULL); | ||
814 | + return (NULL); | ||
815 | uri = uri_new(); | ||
816 | ret = rfc3986_parse_uri_reference(uri, str); | ||
817 | if (ret) { | ||
818 | uri_free(uri); | ||
819 | - return(NULL); | ||
820 | + return (NULL); | ||
821 | } | ||
822 | - return(uri); | ||
823 | + return (uri); | ||
824 | } | ||
825 | |||
826 | /** | ||
827 | @@ -XXX,XX +XXX,XX @@ uri_parse(const char *str) { | ||
828 | * | ||
829 | * Returns 0 or the error code | ||
830 | */ | ||
831 | -int | ||
832 | -uri_parse_into(URI *uri, const char *str) { | ||
833 | - return(rfc3986_parse_uri_reference(uri, str)); | ||
834 | +int uri_parse_into(URI *uri, const char *str) | ||
835 | +{ | ||
836 | + return (rfc3986_parse_uri_reference(uri, str)); | ||
837 | } | ||
838 | |||
839 | /** | ||
840 | @@ -XXX,XX +XXX,XX @@ uri_parse_into(URI *uri, const char *str) { | ||
841 | * | ||
842 | * Returns a newly built URI or NULL in case of error | ||
843 | */ | ||
844 | -URI * | ||
845 | -uri_parse_raw(const char *str, int raw) { | ||
846 | +URI *uri_parse_raw(const char *str, int raw) | ||
847 | +{ | ||
848 | URI *uri; | ||
849 | int ret; | ||
850 | |||
851 | if (str == NULL) | ||
852 | - return(NULL); | ||
853 | + return (NULL); | ||
854 | uri = uri_new(); | ||
855 | if (raw) { | ||
856 | uri->cleanup |= 2; | ||
857 | @@ -XXX,XX +XXX,XX @@ uri_parse_raw(const char *str, int raw) { | ||
858 | ret = uri_parse_into(uri, str); | ||
859 | if (ret) { | ||
860 | uri_free(uri); | ||
861 | - return(NULL); | ||
862 | + return (NULL); | ||
863 | } | ||
864 | - return(uri); | ||
865 | + return (uri); | ||
866 | } | ||
867 | |||
868 | /************************************************************************ | ||
869 | - * * | ||
870 | - * Generic URI structure functions * | ||
871 | - * * | ||
872 | + * * | ||
873 | + * Generic URI structure functions * | ||
874 | + * * | ||
875 | ************************************************************************/ | ||
876 | |||
877 | /** | ||
878 | @@ -XXX,XX +XXX,XX @@ uri_parse_raw(const char *str, int raw) { | ||
879 | * | ||
880 | * Returns the new structure or NULL in case of error | ||
881 | */ | ||
882 | -URI * | ||
883 | -uri_new(void) { | ||
884 | +URI *uri_new(void) | ||
885 | +{ | ||
886 | URI *ret; | ||
887 | |||
888 | ret = g_new0(URI, 1); | ||
889 | - return(ret); | ||
890 | + return (ret); | ||
891 | } | ||
892 | |||
893 | /** | ||
894 | @@ -XXX,XX +XXX,XX @@ uri_new(void) { | ||
895 | * Function to handle properly a reallocation when saving an URI | ||
896 | * Also imposes some limit on the length of an URI string output | ||
897 | */ | ||
898 | -static char * | ||
899 | -realloc2n(char *ret, int *max) { | ||
900 | +static char *realloc2n(char *ret, int *max) | ||
901 | +{ | ||
902 | char *temp; | ||
903 | int tmp; | ||
904 | |||
905 | tmp = *max * 2; | ||
906 | temp = g_realloc(ret, (tmp + 1)); | ||
907 | *max = tmp; | ||
908 | - return(temp); | ||
909 | + return (temp); | ||
910 | } | ||
911 | |||
912 | /** | ||
913 | @@ -XXX,XX +XXX,XX @@ realloc2n(char *ret, int *max) { | ||
914 | * | ||
915 | * Returns a new string (to be deallocated by caller) | ||
916 | */ | ||
917 | -char * | ||
918 | -uri_to_string(URI *uri) { | ||
919 | +char *uri_to_string(URI *uri) | ||
920 | +{ | ||
921 | char *ret = NULL; | ||
922 | char *temp; | ||
923 | const char *p; | ||
924 | int len; | ||
925 | int max; | ||
926 | |||
927 | - if (uri == NULL) return(NULL); | ||
928 | - | ||
929 | + if (uri == NULL) | ||
930 | + return (NULL); | ||
931 | |||
932 | max = 80; | ||
933 | ret = g_malloc(max + 1); | ||
934 | len = 0; | ||
935 | |||
936 | if (uri->scheme != NULL) { | ||
937 | - p = uri->scheme; | ||
938 | - while (*p != 0) { | ||
939 | - if (len >= max) { | ||
940 | + p = uri->scheme; | ||
941 | + while (*p != 0) { | ||
942 | + if (len >= max) { | ||
943 | temp = realloc2n(ret, &max); | ||
944 | - ret = temp; | ||
945 | - } | ||
946 | - ret[len++] = *p++; | ||
947 | - } | ||
948 | - if (len >= max) { | ||
949 | + ret = temp; | ||
950 | + } | 94 | + } |
951 | + ret[len++] = *p++; | 95 | + ret = BDRV_BLOCK_ZERO | BDRV_BLOCK_ALLOCATED; |
952 | + } | ||
953 | + if (len >= max) { | ||
954 | temp = realloc2n(ret, &max); | ||
955 | ret = temp; | ||
956 | - } | ||
957 | - ret[len++] = ':'; | ||
958 | + } | ||
959 | + ret[len++] = ':'; | ||
960 | } | ||
961 | if (uri->opaque != NULL) { | ||
962 | - p = uri->opaque; | ||
963 | - while (*p != 0) { | ||
964 | - if (len + 3 >= max) { | ||
965 | + p = uri->opaque; | ||
966 | + while (*p != 0) { | ||
967 | + if (len + 3 >= max) { | ||
968 | temp = realloc2n(ret, &max); | ||
969 | ret = temp; | ||
970 | - } | ||
971 | - if (IS_RESERVED(*(p)) || IS_UNRESERVED(*(p))) | ||
972 | - ret[len++] = *p++; | ||
973 | - else { | ||
974 | - int val = *(unsigned char *)p++; | ||
975 | - int hi = val / 0x10, lo = val % 0x10; | ||
976 | - ret[len++] = '%'; | ||
977 | - ret[len++] = hi + (hi > 9? 'A'-10 : '0'); | ||
978 | - ret[len++] = lo + (lo > 9? 'A'-10 : '0'); | ||
979 | - } | ||
980 | - } | ||
981 | + } | ||
982 | + if (IS_RESERVED(*(p)) || IS_UNRESERVED(*(p))) | ||
983 | + ret[len++] = *p++; | ||
984 | + else { | ||
985 | + int val = *(unsigned char *)p++; | ||
986 | + int hi = val / 0x10, lo = val % 0x10; | ||
987 | + ret[len++] = '%'; | ||
988 | + ret[len++] = hi + (hi > 9 ? 'A' - 10 : '0'); | ||
989 | + ret[len++] = lo + (lo > 9 ? 'A' - 10 : '0'); | ||
990 | + } | ||
991 | + } | ||
992 | } else { | ||
993 | - if (uri->server != NULL) { | ||
994 | - if (len + 3 >= max) { | ||
995 | + if (uri->server != NULL) { | ||
996 | + if (len + 3 >= max) { | ||
997 | temp = realloc2n(ret, &max); | ||
998 | ret = temp; | ||
999 | - } | ||
1000 | - ret[len++] = '/'; | ||
1001 | - ret[len++] = '/'; | ||
1002 | - if (uri->user != NULL) { | ||
1003 | - p = uri->user; | ||
1004 | - while (*p != 0) { | ||
1005 | - if (len + 3 >= max) { | ||
1006 | + } | ||
1007 | + ret[len++] = '/'; | ||
1008 | + ret[len++] = '/'; | ||
1009 | + if (uri->user != NULL) { | ||
1010 | + p = uri->user; | ||
1011 | + while (*p != 0) { | ||
1012 | + if (len + 3 >= max) { | ||
1013 | temp = realloc2n(ret, &max); | ||
1014 | ret = temp; | ||
1015 | - } | ||
1016 | - if ((IS_UNRESERVED(*(p))) || | ||
1017 | - ((*(p) == ';')) || ((*(p) == ':')) || | ||
1018 | - ((*(p) == '&')) || ((*(p) == '=')) || | ||
1019 | - ((*(p) == '+')) || ((*(p) == '$')) || | ||
1020 | - ((*(p) == ','))) | ||
1021 | - ret[len++] = *p++; | ||
1022 | - else { | ||
1023 | - int val = *(unsigned char *)p++; | ||
1024 | - int hi = val / 0x10, lo = val % 0x10; | ||
1025 | - ret[len++] = '%'; | ||
1026 | - ret[len++] = hi + (hi > 9? 'A'-10 : '0'); | ||
1027 | - ret[len++] = lo + (lo > 9? 'A'-10 : '0'); | ||
1028 | - } | ||
1029 | - } | ||
1030 | - if (len + 3 >= max) { | ||
1031 | + } | ||
1032 | + if ((IS_UNRESERVED(*(p))) || ((*(p) == ';')) || | ||
1033 | + ((*(p) == ':')) || ((*(p) == '&')) || ((*(p) == '=')) || | ||
1034 | + ((*(p) == '+')) || ((*(p) == '$')) || ((*(p) == ','))) | ||
1035 | + ret[len++] = *p++; | ||
1036 | + else { | ||
1037 | + int val = *(unsigned char *)p++; | ||
1038 | + int hi = val / 0x10, lo = val % 0x10; | ||
1039 | + ret[len++] = '%'; | ||
1040 | + ret[len++] = hi + (hi > 9 ? 'A' - 10 : '0'); | ||
1041 | + ret[len++] = lo + (lo > 9 ? 'A' - 10 : '0'); | ||
1042 | + } | ||
1043 | + } | ||
1044 | + if (len + 3 >= max) { | ||
1045 | temp = realloc2n(ret, &max); | ||
1046 | ret = temp; | ||
1047 | - } | ||
1048 | - ret[len++] = '@'; | ||
1049 | - } | ||
1050 | - p = uri->server; | ||
1051 | - while (*p != 0) { | ||
1052 | - if (len >= max) { | ||
1053 | + } | ||
1054 | + ret[len++] = '@'; | ||
1055 | + } | ||
1056 | + p = uri->server; | ||
1057 | + while (*p != 0) { | ||
1058 | + if (len >= max) { | ||
1059 | temp = realloc2n(ret, &max); | ||
1060 | ret = temp; | ||
1061 | - } | ||
1062 | - ret[len++] = *p++; | ||
1063 | - } | ||
1064 | - if (uri->port > 0) { | ||
1065 | - if (len + 10 >= max) { | ||
1066 | + } | ||
1067 | + ret[len++] = *p++; | ||
1068 | + } | ||
1069 | + if (uri->port > 0) { | ||
1070 | + if (len + 10 >= max) { | ||
1071 | temp = realloc2n(ret, &max); | ||
1072 | ret = temp; | ||
1073 | - } | ||
1074 | - len += snprintf(&ret[len], max - len, ":%d", uri->port); | ||
1075 | - } | ||
1076 | - } else if (uri->authority != NULL) { | ||
1077 | - if (len + 3 >= max) { | ||
1078 | + } | ||
1079 | + len += snprintf(&ret[len], max - len, ":%d", uri->port); | ||
1080 | + } | ||
1081 | + } else if (uri->authority != NULL) { | ||
1082 | + if (len + 3 >= max) { | ||
1083 | temp = realloc2n(ret, &max); | ||
1084 | ret = temp; | ||
1085 | - } | ||
1086 | - ret[len++] = '/'; | ||
1087 | - ret[len++] = '/'; | ||
1088 | - p = uri->authority; | ||
1089 | - while (*p != 0) { | ||
1090 | - if (len + 3 >= max) { | ||
1091 | + } | ||
1092 | + ret[len++] = '/'; | ||
1093 | + ret[len++] = '/'; | ||
1094 | + p = uri->authority; | ||
1095 | + while (*p != 0) { | ||
1096 | + if (len + 3 >= max) { | ||
1097 | temp = realloc2n(ret, &max); | ||
1098 | ret = temp; | ||
1099 | - } | ||
1100 | - if ((IS_UNRESERVED(*(p))) || | ||
1101 | - ((*(p) == '$')) || ((*(p) == ',')) || ((*(p) == ';')) || | ||
1102 | - ((*(p) == ':')) || ((*(p) == '@')) || ((*(p) == '&')) || | ||
1103 | - ((*(p) == '=')) || ((*(p) == '+'))) | ||
1104 | - ret[len++] = *p++; | ||
1105 | - else { | ||
1106 | - int val = *(unsigned char *)p++; | ||
1107 | - int hi = val / 0x10, lo = val % 0x10; | ||
1108 | - ret[len++] = '%'; | ||
1109 | - ret[len++] = hi + (hi > 9? 'A'-10 : '0'); | ||
1110 | - ret[len++] = lo + (lo > 9? 'A'-10 : '0'); | ||
1111 | - } | ||
1112 | - } | ||
1113 | - } else if (uri->scheme != NULL) { | ||
1114 | - if (len + 3 >= max) { | ||
1115 | + } | ||
1116 | + if ((IS_UNRESERVED(*(p))) || ((*(p) == '$')) || | ||
1117 | + ((*(p) == ',')) || ((*(p) == ';')) || ((*(p) == ':')) || | ||
1118 | + ((*(p) == '@')) || ((*(p) == '&')) || ((*(p) == '=')) || | ||
1119 | + ((*(p) == '+'))) | ||
1120 | + ret[len++] = *p++; | ||
1121 | + else { | ||
1122 | + int val = *(unsigned char *)p++; | ||
1123 | + int hi = val / 0x10, lo = val % 0x10; | ||
1124 | + ret[len++] = '%'; | ||
1125 | + ret[len++] = hi + (hi > 9 ? 'A' - 10 : '0'); | ||
1126 | + ret[len++] = lo + (lo > 9 ? 'A' - 10 : '0'); | ||
1127 | + } | ||
1128 | + } | ||
1129 | + } else if (uri->scheme != NULL) { | ||
1130 | + if (len + 3 >= max) { | ||
1131 | temp = realloc2n(ret, &max); | ||
1132 | ret = temp; | ||
1133 | - } | ||
1134 | - ret[len++] = '/'; | ||
1135 | - ret[len++] = '/'; | ||
1136 | - } | ||
1137 | - if (uri->path != NULL) { | ||
1138 | - p = uri->path; | ||
1139 | - /* | ||
1140 | - * the colon in file:///d: should not be escaped or | ||
1141 | - * Windows accesses fail later. | ||
1142 | - */ | ||
1143 | - if ((uri->scheme != NULL) && | ||
1144 | - (p[0] == '/') && | ||
1145 | - (((p[1] >= 'a') && (p[1] <= 'z')) || | ||
1146 | - ((p[1] >= 'A') && (p[1] <= 'Z'))) && | ||
1147 | - (p[2] == ':') && | ||
1148 | - (!strcmp(uri->scheme, "file"))) { | ||
1149 | - if (len + 3 >= max) { | ||
1150 | + } | ||
1151 | + ret[len++] = '/'; | ||
1152 | + ret[len++] = '/'; | ||
1153 | + } | ||
1154 | + if (uri->path != NULL) { | ||
1155 | + p = uri->path; | ||
1156 | + /* | ||
1157 | + * the colon in file:///d: should not be escaped or | ||
1158 | + * Windows accesses fail later. | ||
1159 | + */ | ||
1160 | + if ((uri->scheme != NULL) && (p[0] == '/') && | ||
1161 | + (((p[1] >= 'a') && (p[1] <= 'z')) || | ||
1162 | + ((p[1] >= 'A') && (p[1] <= 'Z'))) && | ||
1163 | + (p[2] == ':') && (!strcmp(uri->scheme, "file"))) { | ||
1164 | + if (len + 3 >= max) { | ||
1165 | temp = realloc2n(ret, &max); | ||
1166 | ret = temp; | ||
1167 | - } | ||
1168 | - ret[len++] = *p++; | ||
1169 | - ret[len++] = *p++; | ||
1170 | - ret[len++] = *p++; | ||
1171 | - } | ||
1172 | - while (*p != 0) { | ||
1173 | - if (len + 3 >= max) { | ||
1174 | + } | ||
1175 | + ret[len++] = *p++; | ||
1176 | + ret[len++] = *p++; | ||
1177 | + ret[len++] = *p++; | ||
1178 | + } | ||
1179 | + while (*p != 0) { | ||
1180 | + if (len + 3 >= max) { | ||
1181 | temp = realloc2n(ret, &max); | ||
1182 | ret = temp; | ||
1183 | - } | ||
1184 | - if ((IS_UNRESERVED(*(p))) || ((*(p) == '/')) || | ||
1185 | + } | ||
1186 | + if ((IS_UNRESERVED(*(p))) || ((*(p) == '/')) || | ||
1187 | ((*(p) == ';')) || ((*(p) == '@')) || ((*(p) == '&')) || | ||
1188 | - ((*(p) == '=')) || ((*(p) == '+')) || ((*(p) == '$')) || | ||
1189 | - ((*(p) == ','))) | ||
1190 | - ret[len++] = *p++; | ||
1191 | - else { | ||
1192 | - int val = *(unsigned char *)p++; | ||
1193 | - int hi = val / 0x10, lo = val % 0x10; | ||
1194 | - ret[len++] = '%'; | ||
1195 | - ret[len++] = hi + (hi > 9? 'A'-10 : '0'); | ||
1196 | - ret[len++] = lo + (lo > 9? 'A'-10 : '0'); | ||
1197 | - } | ||
1198 | - } | ||
1199 | - } | ||
1200 | - if (uri->query != NULL) { | ||
1201 | - if (len + 1 >= max) { | ||
1202 | + ((*(p) == '=')) || ((*(p) == '+')) || ((*(p) == '$')) || | ||
1203 | + ((*(p) == ','))) | ||
1204 | + ret[len++] = *p++; | ||
1205 | + else { | ||
1206 | + int val = *(unsigned char *)p++; | ||
1207 | + int hi = val / 0x10, lo = val % 0x10; | ||
1208 | + ret[len++] = '%'; | ||
1209 | + ret[len++] = hi + (hi > 9 ? 'A' - 10 : '0'); | ||
1210 | + ret[len++] = lo + (lo > 9 ? 'A' - 10 : '0'); | ||
1211 | + } | ||
1212 | + } | ||
1213 | + } | ||
1214 | + if (uri->query != NULL) { | ||
1215 | + if (len + 1 >= max) { | ||
1216 | temp = realloc2n(ret, &max); | ||
1217 | ret = temp; | ||
1218 | - } | ||
1219 | - ret[len++] = '?'; | ||
1220 | - p = uri->query; | ||
1221 | - while (*p != 0) { | ||
1222 | - if (len + 1 >= max) { | ||
1223 | + } | ||
1224 | + ret[len++] = '?'; | ||
1225 | + p = uri->query; | ||
1226 | + while (*p != 0) { | ||
1227 | + if (len + 1 >= max) { | ||
1228 | temp = realloc2n(ret, &max); | ||
1229 | ret = temp; | ||
1230 | - } | ||
1231 | - ret[len++] = *p++; | ||
1232 | - } | ||
1233 | - } | ||
1234 | + } | ||
1235 | + ret[len++] = *p++; | ||
1236 | + } | ||
1237 | + } | ||
1238 | } | ||
1239 | if (uri->fragment != NULL) { | ||
1240 | - if (len + 3 >= max) { | ||
1241 | + if (len + 3 >= max) { | ||
1242 | temp = realloc2n(ret, &max); | ||
1243 | ret = temp; | ||
1244 | - } | ||
1245 | - ret[len++] = '#'; | ||
1246 | - p = uri->fragment; | ||
1247 | - while (*p != 0) { | ||
1248 | - if (len + 3 >= max) { | ||
1249 | + } | ||
1250 | + ret[len++] = '#'; | ||
1251 | + p = uri->fragment; | ||
1252 | + while (*p != 0) { | ||
1253 | + if (len + 3 >= max) { | ||
1254 | temp = realloc2n(ret, &max); | ||
1255 | ret = temp; | ||
1256 | - } | ||
1257 | - if ((IS_UNRESERVED(*(p))) || (IS_RESERVED(*(p)))) | ||
1258 | - ret[len++] = *p++; | ||
1259 | - else { | ||
1260 | - int val = *(unsigned char *)p++; | ||
1261 | - int hi = val / 0x10, lo = val % 0x10; | ||
1262 | - ret[len++] = '%'; | ||
1263 | - ret[len++] = hi + (hi > 9? 'A'-10 : '0'); | ||
1264 | - ret[len++] = lo + (lo > 9? 'A'-10 : '0'); | ||
1265 | - } | ||
1266 | - } | ||
1267 | + } | ||
1268 | + if ((IS_UNRESERVED(*(p))) || (IS_RESERVED(*(p)))) | ||
1269 | + ret[len++] = *p++; | ||
1270 | + else { | ||
1271 | + int val = *(unsigned char *)p++; | ||
1272 | + int hi = val / 0x10, lo = val % 0x10; | ||
1273 | + ret[len++] = '%'; | ||
1274 | + ret[len++] = hi + (hi > 9 ? 'A' - 10 : '0'); | ||
1275 | + ret[len++] = lo + (lo > 9 ? 'A' - 10 : '0'); | ||
1276 | + } | ||
1277 | + } | ||
1278 | } | ||
1279 | if (len >= max) { | ||
1280 | temp = realloc2n(ret, &max); | ||
1281 | ret = temp; | ||
1282 | } | ||
1283 | ret[len] = 0; | ||
1284 | - return(ret); | ||
1285 | + return (ret); | ||
1286 | } | ||
1287 | |||
1288 | /** | ||
1289 | @@ -XXX,XX +XXX,XX @@ uri_to_string(URI *uri) { | ||
1290 | * | ||
1291 | * Make sure the URI struct is free of content | ||
1292 | */ | ||
1293 | -static void | ||
1294 | -uri_clean(URI *uri) { | ||
1295 | - if (uri == NULL) return; | ||
1296 | +static void uri_clean(URI *uri) | ||
1297 | +{ | ||
1298 | + if (uri == NULL) | ||
1299 | + return; | ||
1300 | |||
1301 | g_free(uri->scheme); | ||
1302 | uri->scheme = NULL; | ||
1303 | @@ -XXX,XX +XXX,XX @@ uri_clean(URI *uri) { | ||
1304 | * | ||
1305 | * Free up the URI struct | ||
1306 | */ | ||
1307 | -void | ||
1308 | -uri_free(URI *uri) { | ||
1309 | +void uri_free(URI *uri) | ||
1310 | +{ | ||
1311 | uri_clean(uri); | ||
1312 | g_free(uri); | ||
1313 | } | ||
1314 | |||
1315 | /************************************************************************ | ||
1316 | - * * | ||
1317 | - * Helper functions * | ||
1318 | - * * | ||
1319 | + * * | ||
1320 | + * Helper functions * | ||
1321 | + * * | ||
1322 | ************************************************************************/ | ||
1323 | |||
1324 | /** | ||
1325 | @@ -XXX,XX +XXX,XX @@ uri_free(URI *uri) { | ||
1326 | * | ||
1327 | * Returns 0 or an error code | ||
1328 | */ | ||
1329 | -static int | ||
1330 | -normalize_uri_path(char *path) { | ||
1331 | +static int normalize_uri_path(char *path) | ||
1332 | +{ | ||
1333 | char *cur, *out; | ||
1334 | |||
1335 | if (path == NULL) | ||
1336 | - return(-1); | ||
1337 | + return (-1); | ||
1338 | |||
1339 | /* Skip all initial "/" chars. We want to get to the beginning of the | ||
1340 | * first non-empty segment. | ||
1341 | */ | ||
1342 | cur = path; | ||
1343 | while (cur[0] == '/') | ||
1344 | - ++cur; | ||
1345 | + ++cur; | ||
1346 | if (cur[0] == '\0') | ||
1347 | - return(0); | ||
1348 | + return (0); | ||
1349 | |||
1350 | /* Keep everything we've seen so far. */ | ||
1351 | out = cur; | ||
1352 | @@ -XXX,XX +XXX,XX @@ normalize_uri_path(char *path) { | ||
1353 | * Analyze each segment in sequence for cases (c) and (d). | ||
1354 | */ | ||
1355 | while (cur[0] != '\0') { | ||
1356 | - /* | ||
1357 | - * c) All occurrences of "./", where "." is a complete path segment, | ||
1358 | - * are removed from the buffer string. | ||
1359 | - */ | ||
1360 | - if ((cur[0] == '.') && (cur[1] == '/')) { | ||
1361 | - cur += 2; | ||
1362 | - /* '//' normalization should be done at this point too */ | ||
1363 | - while (cur[0] == '/') | ||
1364 | - cur++; | ||
1365 | - continue; | ||
1366 | - } | ||
1367 | + /* | ||
1368 | + * c) All occurrences of "./", where "." is a complete path segment, | ||
1369 | + * are removed from the buffer string. | ||
1370 | + */ | ||
1371 | + if ((cur[0] == '.') && (cur[1] == '/')) { | ||
1372 | + cur += 2; | ||
1373 | + /* '//' normalization should be done at this point too */ | ||
1374 | + while (cur[0] == '/') | ||
1375 | + cur++; | ||
1376 | + continue; | ||
1377 | + } | ||
1378 | |||
1379 | - /* | ||
1380 | - * d) If the buffer string ends with "." as a complete path segment, | ||
1381 | - * that "." is removed. | ||
1382 | - */ | ||
1383 | - if ((cur[0] == '.') && (cur[1] == '\0')) | ||
1384 | - break; | ||
1385 | + /* | ||
1386 | + * d) If the buffer string ends with "." as a complete path segment, | ||
1387 | + * that "." is removed. | ||
1388 | + */ | ||
1389 | + if ((cur[0] == '.') && (cur[1] == '\0')) | ||
1390 | + break; | ||
1391 | |||
1392 | - /* Otherwise keep the segment. */ | ||
1393 | - while (cur[0] != '/') { | ||
1394 | + /* Otherwise keep the segment. */ | ||
1395 | + while (cur[0] != '/') { | ||
1396 | if (cur[0] == '\0') | ||
1397 | - goto done_cd; | ||
1398 | - (out++)[0] = (cur++)[0]; | ||
1399 | - } | ||
1400 | - /* nomalize // */ | ||
1401 | - while ((cur[0] == '/') && (cur[1] == '/')) | ||
1402 | - cur++; | ||
1403 | + goto done_cd; | ||
1404 | + (out++)[0] = (cur++)[0]; | ||
1405 | + } | ||
1406 | + /* nomalize // */ | ||
1407 | + while ((cur[0] == '/') && (cur[1] == '/')) | ||
1408 | + cur++; | ||
1409 | |||
1410 | (out++)[0] = (cur++)[0]; | ||
1411 | } | ||
1412 | - done_cd: | ||
1413 | +done_cd: | ||
1414 | out[0] = '\0'; | ||
1415 | |||
1416 | /* Reset to the beginning of the first segment for the next sequence. */ | ||
1417 | cur = path; | ||
1418 | while (cur[0] == '/') | ||
1419 | - ++cur; | ||
1420 | + ++cur; | ||
1421 | if (cur[0] == '\0') | ||
1422 | - return(0); | ||
1423 | + return (0); | ||
1424 | |||
1425 | /* | ||
1426 | * Analyze each segment in sequence for cases (e) and (f). | ||
1427 | @@ -XXX,XX +XXX,XX @@ normalize_uri_path(char *path) { | ||
1428 | /* Find the end of the current segment. */ | ||
1429 | segp = cur; | ||
1430 | while ((segp[0] != '/') && (segp[0] != '\0')) | ||
1431 | - ++segp; | ||
1432 | + ++segp; | ||
1433 | |||
1434 | /* If this is the last segment, we're done (we need at least two | ||
1435 | * segments to meet the criteria for the (e) and (f) cases). | ||
1436 | */ | ||
1437 | if (segp[0] == '\0') | ||
1438 | - break; | ||
1439 | + break; | ||
1440 | |||
1441 | /* If the first segment is "..", or if the next segment _isn't_ "..", | ||
1442 | * keep this segment and try the next one. | ||
1443 | */ | ||
1444 | ++segp; | ||
1445 | - if (((cur[0] == '.') && (cur[1] == '.') && (segp == cur+3)) | ||
1446 | - || ((segp[0] != '.') || (segp[1] != '.') | ||
1447 | - || ((segp[2] != '/') && (segp[2] != '\0')))) { | ||
1448 | - cur = segp; | ||
1449 | - continue; | ||
1450 | + if (((cur[0] == '.') && (cur[1] == '.') && (segp == cur + 3)) || | ||
1451 | + ((segp[0] != '.') || (segp[1] != '.') || | ||
1452 | + ((segp[2] != '/') && (segp[2] != '\0')))) { | ||
1453 | + cur = segp; | ||
1454 | + continue; | ||
1455 | } | ||
1456 | |||
1457 | /* If we get here, remove this segment and the next one and back up | ||
1458 | @@ -XXX,XX +XXX,XX @@ normalize_uri_path(char *path) { | ||
1459 | |||
1460 | /* If this is the end of the buffer, we're done. */ | ||
1461 | if (segp[2] == '\0') { | ||
1462 | - cur[0] = '\0'; | ||
1463 | - break; | ||
1464 | + cur[0] = '\0'; | ||
1465 | + break; | 96 | + break; |
1466 | } | 97 | } |
1467 | /* Valgrind complained, strcpy(cur, segp + 3); */ | 98 | - if (ret & (BDRV_BLOCK_ZERO | BDRV_BLOCK_DATA)) { |
1468 | /* string will overlap, do not use strcpy */ | 99 | + if (ret & BDRV_BLOCK_ALLOCATED) { |
1469 | tmp = cur; | 100 | + /* |
1470 | segp += 3; | 101 | + * We've found the node and the status, we must break. |
1471 | while ((*tmp++ = *segp++) != 0) | 102 | + * |
1472 | - ; | 103 | + * Drop BDRV_BLOCK_EOF, as it's not for upper layer, which may be |
1473 | + ; | 104 | + * larger. We'll add BDRV_BLOCK_EOF if needed at function end, see |
1474 | 105 | + * below. | |
1475 | /* If there are no previous segments, then keep going from here. */ | 106 | + */ |
1476 | segp = cur; | 107 | + ret &= ~BDRV_BLOCK_EOF; |
1477 | while ((segp > path) && ((--segp)[0] == '/')) | 108 | break; |
1478 | - ; | 109 | } |
1479 | + ; | 110 | - /* [offset, pnum] unallocated on this layer, which could be only |
1480 | if (segp == path) | 111 | - * the first part of [offset, bytes]. */ |
1481 | - continue; | 112 | - bytes = MIN(bytes, *pnum); |
1482 | + continue; | 113 | - first = false; |
1483 | 114 | + | |
1484 | /* "segp" is pointing to the end of a previous segment; find it's | 115 | + /* |
1485 | * start. We need to back up to the previous segment and start | 116 | + * OK, [offset, offset + *pnum) region is unallocated on this layer, |
1486 | @@ -XXX,XX +XXX,XX @@ normalize_uri_path(char *path) { | 117 | + * let's continue the diving. |
1487 | */ | 118 | + */ |
1488 | cur = segp; | 119 | + assert(*pnum <= bytes); |
1489 | while ((cur > path) && (cur[-1] != '/')) | 120 | + bytes = *pnum; |
1490 | - --cur; | 121 | + } |
1491 | + --cur; | 122 | + |
123 | + if (offset + *pnum == eof) { | ||
124 | + ret |= BDRV_BLOCK_EOF; | ||
1492 | } | 125 | } |
1493 | out[0] = '\0'; | 126 | + |
1494 | 127 | return ret; | |
1495 | @@ -XXX,XX +XXX,XX @@ normalize_uri_path(char *path) { | 128 | } |
1496 | * We discard them from the final path. | 129 | |
1497 | */ | 130 | diff --git a/block/qcow2.c b/block/qcow2.c |
1498 | if (path[0] == '/') { | 131 | index XXXXXXX..XXXXXXX 100644 |
1499 | - cur = path; | 132 | --- a/block/qcow2.c |
1500 | - while ((cur[0] == '/') && (cur[1] == '.') && (cur[2] == '.') | 133 | +++ b/block/qcow2.c |
1501 | - && ((cur[3] == '/') || (cur[3] == '\0'))) | 134 | @@ -XXX,XX +XXX,XX @@ static bool is_zero(BlockDriverState *bs, int64_t offset, int64_t bytes) |
1502 | - cur += 3; | 135 | if (!bytes) { |
1503 | + cur = path; | 136 | return true; |
1504 | + while ((cur[0] == '/') && (cur[1] == '.') && (cur[2] == '.') && | ||
1505 | + ((cur[3] == '/') || (cur[3] == '\0'))) | ||
1506 | + cur += 3; | ||
1507 | |||
1508 | - if (cur != path) { | ||
1509 | - out = path; | ||
1510 | - while (cur[0] != '\0') | ||
1511 | - (out++)[0] = (cur++)[0]; | ||
1512 | - out[0] = 0; | ||
1513 | - } | ||
1514 | + if (cur != path) { | ||
1515 | + out = path; | ||
1516 | + while (cur[0] != '\0') | ||
1517 | + (out++)[0] = (cur++)[0]; | ||
1518 | + out[0] = 0; | ||
1519 | + } | ||
1520 | } | 137 | } |
1521 | 138 | - res = bdrv_block_status_above(bs, NULL, offset, bytes, &nr, NULL, NULL); | |
1522 | - return(0); | 139 | - return res >= 0 && (res & BDRV_BLOCK_ZERO) && nr == bytes; |
1523 | + return (0); | 140 | + |
141 | + /* | ||
142 | + * bdrv_block_status_above doesn't merge different types of zeros, for | ||
143 | + * example, zeros which come from the region which is unallocated in | ||
144 | + * the whole backing chain, and zeros which come because of a short | ||
145 | + * backing file. So, we need a loop. | ||
146 | + */ | ||
147 | + do { | ||
148 | + res = bdrv_block_status_above(bs, NULL, offset, bytes, &nr, NULL, NULL); | ||
149 | + offset += nr; | ||
150 | + bytes -= nr; | ||
151 | + } while (res >= 0 && (res & BDRV_BLOCK_ZERO) && nr && bytes); | ||
152 | + | ||
153 | + return res >= 0 && (res & BDRV_BLOCK_ZERO) && bytes == 0; | ||
1524 | } | 154 | } |
1525 | 155 | ||
1526 | -static int is_hex(char c) { | 156 | static coroutine_fn int qcow2_co_pwrite_zeroes(BlockDriverState *bs, |
1527 | - if (((c >= '0') && (c <= '9')) || | ||
1528 | - ((c >= 'a') && (c <= 'f')) || | ||
1529 | +static int is_hex(char c) | ||
1530 | +{ | ||
1531 | + if (((c >= '0') && (c <= '9')) || ((c >= 'a') && (c <= 'f')) || | ||
1532 | ((c >= 'A') && (c <= 'F'))) | ||
1533 | - return(1); | ||
1534 | - return(0); | ||
1535 | + return (1); | ||
1536 | + return (0); | ||
1537 | } | ||
1538 | |||
1539 | - | ||
1540 | /** | ||
1541 | * uri_string_unescape: | ||
1542 | * @str: the string to unescape | ||
1543 | @@ -XXX,XX +XXX,XX @@ static int is_hex(char c) { | ||
1544 | * Returns a copy of the string, but unescaped, will return NULL only in case | ||
1545 | * of error | ||
1546 | */ | ||
1547 | -char * | ||
1548 | -uri_string_unescape(const char *str, int len, char *target) { | ||
1549 | +char *uri_string_unescape(const char *str, int len, char *target) | ||
1550 | +{ | ||
1551 | char *ret, *out; | ||
1552 | const char *in; | ||
1553 | |||
1554 | if (str == NULL) | ||
1555 | - return(NULL); | ||
1556 | - if (len <= 0) len = strlen(str); | ||
1557 | - if (len < 0) return(NULL); | ||
1558 | + return (NULL); | ||
1559 | + if (len <= 0) | ||
1560 | + len = strlen(str); | ||
1561 | + if (len < 0) | ||
1562 | + return (NULL); | ||
1563 | |||
1564 | if (target == NULL) { | ||
1565 | - ret = g_malloc(len + 1); | ||
1566 | + ret = g_malloc(len + 1); | ||
1567 | } else | ||
1568 | - ret = target; | ||
1569 | + ret = target; | ||
1570 | in = str; | ||
1571 | out = ret; | ||
1572 | - while(len > 0) { | ||
1573 | - if ((len > 2) && (*in == '%') && (is_hex(in[1])) && (is_hex(in[2]))) { | ||
1574 | - in++; | ||
1575 | - if ((*in >= '0') && (*in <= '9')) | ||
1576 | - *out = (*in - '0'); | ||
1577 | - else if ((*in >= 'a') && (*in <= 'f')) | ||
1578 | - *out = (*in - 'a') + 10; | ||
1579 | - else if ((*in >= 'A') && (*in <= 'F')) | ||
1580 | - *out = (*in - 'A') + 10; | ||
1581 | - in++; | ||
1582 | - if ((*in >= '0') && (*in <= '9')) | ||
1583 | - *out = *out * 16 + (*in - '0'); | ||
1584 | - else if ((*in >= 'a') && (*in <= 'f')) | ||
1585 | - *out = *out * 16 + (*in - 'a') + 10; | ||
1586 | - else if ((*in >= 'A') && (*in <= 'F')) | ||
1587 | - *out = *out * 16 + (*in - 'A') + 10; | ||
1588 | - in++; | ||
1589 | - len -= 3; | ||
1590 | - out++; | ||
1591 | - } else { | ||
1592 | - *out++ = *in++; | ||
1593 | - len--; | ||
1594 | - } | ||
1595 | + while (len > 0) { | ||
1596 | + if ((len > 2) && (*in == '%') && (is_hex(in[1])) && (is_hex(in[2]))) { | ||
1597 | + in++; | ||
1598 | + if ((*in >= '0') && (*in <= '9')) | ||
1599 | + *out = (*in - '0'); | ||
1600 | + else if ((*in >= 'a') && (*in <= 'f')) | ||
1601 | + *out = (*in - 'a') + 10; | ||
1602 | + else if ((*in >= 'A') && (*in <= 'F')) | ||
1603 | + *out = (*in - 'A') + 10; | ||
1604 | + in++; | ||
1605 | + if ((*in >= '0') && (*in <= '9')) | ||
1606 | + *out = *out * 16 + (*in - '0'); | ||
1607 | + else if ((*in >= 'a') && (*in <= 'f')) | ||
1608 | + *out = *out * 16 + (*in - 'a') + 10; | ||
1609 | + else if ((*in >= 'A') && (*in <= 'F')) | ||
1610 | + *out = *out * 16 + (*in - 'A') + 10; | ||
1611 | + in++; | ||
1612 | + len -= 3; | ||
1613 | + out++; | ||
1614 | + } else { | ||
1615 | + *out++ = *in++; | ||
1616 | + len--; | ||
1617 | + } | ||
1618 | } | ||
1619 | *out = 0; | ||
1620 | - return(ret); | ||
1621 | + return (ret); | ||
1622 | } | ||
1623 | |||
1624 | /** | ||
1625 | @@ -XXX,XX +XXX,XX @@ uri_string_unescape(const char *str, int len, char *target) { | ||
1626 | * | ||
1627 | * Returns a new escaped string or NULL in case of error. | ||
1628 | */ | ||
1629 | -char * | ||
1630 | -uri_string_escape(const char *str, const char *list) { | ||
1631 | +char *uri_string_escape(const char *str, const char *list) | ||
1632 | +{ | ||
1633 | char *ret, ch; | ||
1634 | char *temp; | ||
1635 | const char *in; | ||
1636 | int len, out; | ||
1637 | |||
1638 | if (str == NULL) | ||
1639 | - return(NULL); | ||
1640 | + return (NULL); | ||
1641 | if (str[0] == 0) | ||
1642 | - return(g_strdup(str)); | ||
1643 | + return (g_strdup(str)); | ||
1644 | len = strlen(str); | ||
1645 | - if (!(len > 0)) return(NULL); | ||
1646 | + if (!(len > 0)) | ||
1647 | + return (NULL); | ||
1648 | |||
1649 | len += 20; | ||
1650 | ret = g_malloc(len); | ||
1651 | in = str; | ||
1652 | out = 0; | ||
1653 | - while(*in != 0) { | ||
1654 | - if (len - out <= 3) { | ||
1655 | + while (*in != 0) { | ||
1656 | + if (len - out <= 3) { | ||
1657 | temp = realloc2n(ret, &len); | ||
1658 | - ret = temp; | ||
1659 | - } | ||
1660 | + ret = temp; | ||
1661 | + } | ||
1662 | |||
1663 | - ch = *in; | ||
1664 | - | ||
1665 | - if ((ch != '@') && (!IS_UNRESERVED(ch)) && (!strchr(list, ch))) { | ||
1666 | - unsigned char val; | ||
1667 | - ret[out++] = '%'; | ||
1668 | - val = ch >> 4; | ||
1669 | - if (val <= 9) | ||
1670 | - ret[out++] = '0' + val; | ||
1671 | - else | ||
1672 | - ret[out++] = 'A' + val - 0xA; | ||
1673 | - val = ch & 0xF; | ||
1674 | - if (val <= 9) | ||
1675 | - ret[out++] = '0' + val; | ||
1676 | - else | ||
1677 | - ret[out++] = 'A' + val - 0xA; | ||
1678 | - in++; | ||
1679 | - } else { | ||
1680 | - ret[out++] = *in++; | ||
1681 | - } | ||
1682 | + ch = *in; | ||
1683 | |||
1684 | + if ((ch != '@') && (!IS_UNRESERVED(ch)) && (!strchr(list, ch))) { | ||
1685 | + unsigned char val; | ||
1686 | + ret[out++] = '%'; | ||
1687 | + val = ch >> 4; | ||
1688 | + if (val <= 9) | ||
1689 | + ret[out++] = '0' + val; | ||
1690 | + else | ||
1691 | + ret[out++] = 'A' + val - 0xA; | ||
1692 | + val = ch & 0xF; | ||
1693 | + if (val <= 9) | ||
1694 | + ret[out++] = '0' + val; | ||
1695 | + else | ||
1696 | + ret[out++] = 'A' + val - 0xA; | ||
1697 | + in++; | ||
1698 | + } else { | ||
1699 | + ret[out++] = *in++; | ||
1700 | + } | ||
1701 | } | ||
1702 | ret[out] = 0; | ||
1703 | - return(ret); | ||
1704 | + return (ret); | ||
1705 | } | ||
1706 | |||
1707 | /************************************************************************ | ||
1708 | - * * | ||
1709 | - * Public functions * | ||
1710 | - * * | ||
1711 | + * * | ||
1712 | + * Public functions * | ||
1713 | + * * | ||
1714 | ************************************************************************/ | ||
1715 | |||
1716 | /** | ||
1717 | @@ -XXX,XX +XXX,XX @@ uri_string_escape(const char *str, const char *list) { | ||
1718 | * Returns a new URI string (to be freed by the caller) or NULL in case | ||
1719 | * of error. | ||
1720 | */ | ||
1721 | -char * | ||
1722 | -uri_resolve(const char *uri, const char *base) { | ||
1723 | +char *uri_resolve(const char *uri, const char *base) | ||
1724 | +{ | ||
1725 | char *val = NULL; | ||
1726 | int ret, len, indx, cur, out; | ||
1727 | URI *ref = NULL; | ||
1728 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1729 | * URI. Should we do that here? | ||
1730 | */ | ||
1731 | if (uri == NULL) | ||
1732 | - ret = -1; | ||
1733 | + ret = -1; | ||
1734 | else { | ||
1735 | - if (*uri) { | ||
1736 | - ref = uri_new(); | ||
1737 | - ret = uri_parse_into(ref, uri); | ||
1738 | - } | ||
1739 | - else | ||
1740 | - ret = 0; | ||
1741 | + if (*uri) { | ||
1742 | + ref = uri_new(); | ||
1743 | + ret = uri_parse_into(ref, uri); | ||
1744 | + } else | ||
1745 | + ret = 0; | ||
1746 | } | ||
1747 | if (ret != 0) | ||
1748 | - goto done; | ||
1749 | + goto done; | ||
1750 | if ((ref != NULL) && (ref->scheme != NULL)) { | ||
1751 | - /* | ||
1752 | - * The URI is absolute don't modify. | ||
1753 | - */ | ||
1754 | - val = g_strdup(uri); | ||
1755 | - goto done; | ||
1756 | + /* | ||
1757 | + * The URI is absolute don't modify. | ||
1758 | + */ | ||
1759 | + val = g_strdup(uri); | ||
1760 | + goto done; | ||
1761 | } | ||
1762 | if (base == NULL) | ||
1763 | - ret = -1; | ||
1764 | + ret = -1; | ||
1765 | else { | ||
1766 | - bas = uri_new(); | ||
1767 | - ret = uri_parse_into(bas, base); | ||
1768 | + bas = uri_new(); | ||
1769 | + ret = uri_parse_into(bas, base); | ||
1770 | } | ||
1771 | if (ret != 0) { | ||
1772 | - if (ref) | ||
1773 | - val = uri_to_string(ref); | ||
1774 | - goto done; | ||
1775 | + if (ref) | ||
1776 | + val = uri_to_string(ref); | ||
1777 | + goto done; | ||
1778 | } | ||
1779 | if (ref == NULL) { | ||
1780 | - /* | ||
1781 | - * the base fragment must be ignored | ||
1782 | - */ | ||
1783 | + /* | ||
1784 | + * the base fragment must be ignored | ||
1785 | + */ | ||
1786 | g_free(bas->fragment); | ||
1787 | bas->fragment = NULL; | ||
1788 | - val = uri_to_string(bas); | ||
1789 | - goto done; | ||
1790 | + val = uri_to_string(bas); | ||
1791 | + goto done; | ||
1792 | } | ||
1793 | |||
1794 | /* | ||
1795 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1796 | */ | ||
1797 | res = uri_new(); | ||
1798 | if ((ref->scheme == NULL) && (ref->path == NULL) && | ||
1799 | - ((ref->authority == NULL) && (ref->server == NULL))) { | ||
1800 | + ((ref->authority == NULL) && (ref->server == NULL))) { | ||
1801 | res->scheme = g_strdup(bas->scheme); | ||
1802 | - if (bas->authority != NULL) | ||
1803 | - res->authority = g_strdup(bas->authority); | ||
1804 | - else if (bas->server != NULL) { | ||
1805 | + if (bas->authority != NULL) | ||
1806 | + res->authority = g_strdup(bas->authority); | ||
1807 | + else if (bas->server != NULL) { | ||
1808 | res->server = g_strdup(bas->server); | ||
1809 | res->user = g_strdup(bas->user); | ||
1810 | res->port = bas->port; | ||
1811 | - } | ||
1812 | + } | ||
1813 | res->path = g_strdup(bas->path); | ||
1814 | if (ref->query != NULL) { | ||
1815 | - res->query = g_strdup (ref->query); | ||
1816 | + res->query = g_strdup(ref->query); | ||
1817 | } else { | ||
1818 | res->query = g_strdup(bas->query); | ||
1819 | } | ||
1820 | res->fragment = g_strdup(ref->fragment); | ||
1821 | - goto step_7; | ||
1822 | + goto step_7; | ||
1823 | } | ||
1824 | |||
1825 | /* | ||
1826 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1827 | * scheme is inherited from the base URI's scheme component. | ||
1828 | */ | ||
1829 | if (ref->scheme != NULL) { | ||
1830 | - val = uri_to_string(ref); | ||
1831 | - goto done; | ||
1832 | + val = uri_to_string(ref); | ||
1833 | + goto done; | ||
1834 | } | ||
1835 | res->scheme = g_strdup(bas->scheme); | ||
1836 | |||
1837 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1838 | * use an authority component. | ||
1839 | */ | ||
1840 | if ((ref->authority != NULL) || (ref->server != NULL)) { | ||
1841 | - if (ref->authority != NULL) | ||
1842 | - res->authority = g_strdup(ref->authority); | ||
1843 | - else { | ||
1844 | - res->server = g_strdup(ref->server); | ||
1845 | + if (ref->authority != NULL) | ||
1846 | + res->authority = g_strdup(ref->authority); | ||
1847 | + else { | ||
1848 | + res->server = g_strdup(ref->server); | ||
1849 | res->user = g_strdup(ref->user); | ||
1850 | res->port = ref->port; | ||
1851 | - } | ||
1852 | + } | ||
1853 | res->path = g_strdup(ref->path); | ||
1854 | - goto step_7; | ||
1855 | + goto step_7; | ||
1856 | } | ||
1857 | if (bas->authority != NULL) | ||
1858 | - res->authority = g_strdup(bas->authority); | ||
1859 | + res->authority = g_strdup(bas->authority); | ||
1860 | else if (bas->server != NULL) { | ||
1861 | res->server = g_strdup(bas->server); | ||
1862 | res->user = g_strdup(bas->user); | ||
1863 | - res->port = bas->port; | ||
1864 | + res->port = bas->port; | ||
1865 | } | ||
1866 | |||
1867 | /* | ||
1868 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1869 | * the reference is an absolute-path and we skip to step 7. | ||
1870 | */ | ||
1871 | if ((ref->path != NULL) && (ref->path[0] == '/')) { | ||
1872 | - res->path = g_strdup(ref->path); | ||
1873 | - goto step_7; | ||
1874 | + res->path = g_strdup(ref->path); | ||
1875 | + goto step_7; | ||
1876 | } | ||
1877 | |||
1878 | - | ||
1879 | /* | ||
1880 | * 6) If this step is reached, then we are resolving a relative-path | ||
1881 | * reference. The relative path needs to be merged with the base | ||
1882 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1883 | */ | ||
1884 | len = 2; /* extra / and 0 */ | ||
1885 | if (ref->path != NULL) | ||
1886 | - len += strlen(ref->path); | ||
1887 | + len += strlen(ref->path); | ||
1888 | if (bas->path != NULL) | ||
1889 | - len += strlen(bas->path); | ||
1890 | + len += strlen(bas->path); | ||
1891 | res->path = g_malloc(len); | ||
1892 | res->path[0] = 0; | ||
1893 | |||
1894 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1895 | cur = 0; | ||
1896 | out = 0; | ||
1897 | if (bas->path != NULL) { | ||
1898 | - while (bas->path[cur] != 0) { | ||
1899 | - while ((bas->path[cur] != 0) && (bas->path[cur] != '/')) | ||
1900 | - cur++; | ||
1901 | - if (bas->path[cur] == 0) | ||
1902 | - break; | ||
1903 | + while (bas->path[cur] != 0) { | ||
1904 | + while ((bas->path[cur] != 0) && (bas->path[cur] != '/')) | ||
1905 | + cur++; | ||
1906 | + if (bas->path[cur] == 0) | ||
1907 | + break; | ||
1908 | |||
1909 | - cur++; | ||
1910 | - while (out < cur) { | ||
1911 | - res->path[out] = bas->path[out]; | ||
1912 | - out++; | ||
1913 | - } | ||
1914 | - } | ||
1915 | + cur++; | ||
1916 | + while (out < cur) { | ||
1917 | + res->path[out] = bas->path[out]; | ||
1918 | + out++; | ||
1919 | + } | ||
1920 | + } | ||
1921 | } | ||
1922 | res->path[out] = 0; | ||
1923 | |||
1924 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1925 | * string. | ||
1926 | */ | ||
1927 | if (ref->path != NULL && ref->path[0] != 0) { | ||
1928 | - indx = 0; | ||
1929 | - /* | ||
1930 | - * Ensure the path includes a '/' | ||
1931 | - */ | ||
1932 | - if ((out == 0) && (bas->server != NULL)) | ||
1933 | - res->path[out++] = '/'; | ||
1934 | - while (ref->path[indx] != 0) { | ||
1935 | - res->path[out++] = ref->path[indx++]; | ||
1936 | - } | ||
1937 | + indx = 0; | ||
1938 | + /* | ||
1939 | + * Ensure the path includes a '/' | ||
1940 | + */ | ||
1941 | + if ((out == 0) && (bas->server != NULL)) | ||
1942 | + res->path[out++] = '/'; | ||
1943 | + while (ref->path[indx] != 0) { | ||
1944 | + res->path[out++] = ref->path[indx++]; | ||
1945 | + } | ||
1946 | } | ||
1947 | res->path[out] = 0; | ||
1948 | |||
1949 | @@ -XXX,XX +XXX,XX @@ step_7: | ||
1950 | |||
1951 | done: | ||
1952 | if (ref != NULL) | ||
1953 | - uri_free(ref); | ||
1954 | + uri_free(ref); | ||
1955 | if (bas != NULL) | ||
1956 | - uri_free(bas); | ||
1957 | + uri_free(bas); | ||
1958 | if (res != NULL) | ||
1959 | - uri_free(res); | ||
1960 | - return(val); | ||
1961 | + uri_free(res); | ||
1962 | + return (val); | ||
1963 | } | ||
1964 | |||
1965 | /** | ||
1966 | @@ -XXX,XX +XXX,XX @@ done: | ||
1967 | * Returns a new URI string (to be freed by the caller) or NULL in case | ||
1968 | * error. | ||
1969 | */ | ||
1970 | -char * | ||
1971 | -uri_resolve_relative (const char *uri, const char * base) | ||
1972 | +char *uri_resolve_relative(const char *uri, const char *base) | ||
1973 | { | ||
1974 | char *val = NULL; | ||
1975 | int ret; | ||
1976 | @@ -XXX,XX +XXX,XX @@ uri_resolve_relative (const char *uri, const char * base) | ||
1977 | int remove_path = 0; | ||
1978 | |||
1979 | if ((uri == NULL) || (*uri == 0)) | ||
1980 | - return NULL; | ||
1981 | + return NULL; | ||
1982 | |||
1983 | /* | ||
1984 | * First parse URI into a standard form | ||
1985 | */ | ||
1986 | - ref = uri_new (); | ||
1987 | + ref = uri_new(); | ||
1988 | /* If URI not already in "relative" form */ | ||
1989 | if (uri[0] != '.') { | ||
1990 | - ret = uri_parse_into (ref, uri); | ||
1991 | - if (ret != 0) | ||
1992 | - goto done; /* Error in URI, return NULL */ | ||
1993 | + ret = uri_parse_into(ref, uri); | ||
1994 | + if (ret != 0) | ||
1995 | + goto done; /* Error in URI, return NULL */ | ||
1996 | } else | ||
1997 | - ref->path = g_strdup(uri); | ||
1998 | + ref->path = g_strdup(uri); | ||
1999 | |||
2000 | /* | ||
2001 | * Next parse base into the same standard form | ||
2002 | */ | ||
2003 | if ((base == NULL) || (*base == 0)) { | ||
2004 | - val = g_strdup (uri); | ||
2005 | - goto done; | ||
2006 | + val = g_strdup(uri); | ||
2007 | + goto done; | ||
2008 | } | ||
2009 | - bas = uri_new (); | ||
2010 | + bas = uri_new(); | ||
2011 | if (base[0] != '.') { | ||
2012 | - ret = uri_parse_into (bas, base); | ||
2013 | - if (ret != 0) | ||
2014 | - goto done; /* Error in base, return NULL */ | ||
2015 | + ret = uri_parse_into(bas, base); | ||
2016 | + if (ret != 0) | ||
2017 | + goto done; /* Error in base, return NULL */ | ||
2018 | } else | ||
2019 | - bas->path = g_strdup(base); | ||
2020 | + bas->path = g_strdup(base); | ||
2021 | |||
2022 | /* | ||
2023 | * If the scheme / server on the URI differs from the base, | ||
2024 | * just return the URI | ||
2025 | */ | ||
2026 | if ((ref->scheme != NULL) && | ||
2027 | - ((bas->scheme == NULL) || | ||
2028 | - (strcmp (bas->scheme, ref->scheme)) || | ||
2029 | - (strcmp (bas->server, ref->server)))) { | ||
2030 | - val = g_strdup (uri); | ||
2031 | - goto done; | ||
2032 | + ((bas->scheme == NULL) || (strcmp(bas->scheme, ref->scheme)) || | ||
2033 | + (strcmp(bas->server, ref->server)))) { | ||
2034 | + val = g_strdup(uri); | ||
2035 | + goto done; | ||
2036 | } | ||
2037 | if (bas->path == ref->path || | ||
2038 | (bas->path && ref->path && !strcmp(bas->path, ref->path))) { | ||
2039 | - val = g_strdup(""); | ||
2040 | - goto done; | ||
2041 | + val = g_strdup(""); | ||
2042 | + goto done; | ||
2043 | } | ||
2044 | if (bas->path == NULL) { | ||
2045 | - val = g_strdup(ref->path); | ||
2046 | - goto done; | ||
2047 | + val = g_strdup(ref->path); | ||
2048 | + goto done; | ||
2049 | } | ||
2050 | if (ref->path == NULL) { | ||
2051 | - ref->path = (char *) "/"; | ||
2052 | - remove_path = 1; | ||
2053 | + ref->path = (char *)"/"; | ||
2054 | + remove_path = 1; | ||
2055 | } | ||
2056 | |||
2057 | /* | ||
2058 | @@ -XXX,XX +XXX,XX @@ uri_resolve_relative (const char *uri, const char * base) | ||
2059 | * two path components may be missing (bug 316224) | ||
2060 | */ | ||
2061 | if (bas->path == NULL) { | ||
2062 | - if (ref->path != NULL) { | ||
2063 | - uptr = ref->path; | ||
2064 | - if (*uptr == '/') | ||
2065 | - uptr++; | ||
2066 | - /* exception characters from uri_to_string */ | ||
2067 | - val = uri_string_escape(uptr, "/;&=+$,"); | ||
2068 | - } | ||
2069 | - goto done; | ||
2070 | + if (ref->path != NULL) { | ||
2071 | + uptr = ref->path; | ||
2072 | + if (*uptr == '/') | ||
2073 | + uptr++; | ||
2074 | + /* exception characters from uri_to_string */ | ||
2075 | + val = uri_string_escape(uptr, "/;&=+$,"); | ||
2076 | + } | ||
2077 | + goto done; | ||
2078 | } | ||
2079 | bptr = bas->path; | ||
2080 | if (ref->path == NULL) { | ||
2081 | - for (ix = 0; bptr[ix] != 0; ix++) { | ||
2082 | - if (bptr[ix] == '/') | ||
2083 | - nbslash++; | ||
2084 | - } | ||
2085 | - uptr = NULL; | ||
2086 | - len = 1; /* this is for a string terminator only */ | ||
2087 | + for (ix = 0; bptr[ix] != 0; ix++) { | ||
2088 | + if (bptr[ix] == '/') | ||
2089 | + nbslash++; | ||
2090 | + } | ||
2091 | + uptr = NULL; | ||
2092 | + len = 1; /* this is for a string terminator only */ | ||
2093 | } else { | ||
2094 | - /* | ||
2095 | - * Next we compare the two strings and find where they first differ | ||
2096 | - */ | ||
2097 | - if ((ref->path[pos] == '.') && (ref->path[pos+1] == '/')) | ||
2098 | + /* | ||
2099 | + * Next we compare the two strings and find where they first differ | ||
2100 | + */ | ||
2101 | + if ((ref->path[pos] == '.') && (ref->path[pos + 1] == '/')) | ||
2102 | pos += 2; | ||
2103 | - if ((*bptr == '.') && (bptr[1] == '/')) | ||
2104 | + if ((*bptr == '.') && (bptr[1] == '/')) | ||
2105 | bptr += 2; | ||
2106 | - else if ((*bptr == '/') && (ref->path[pos] != '/')) | ||
2107 | - bptr++; | ||
2108 | - while ((bptr[pos] == ref->path[pos]) && (bptr[pos] != 0)) | ||
2109 | - pos++; | ||
2110 | + else if ((*bptr == '/') && (ref->path[pos] != '/')) | ||
2111 | + bptr++; | ||
2112 | + while ((bptr[pos] == ref->path[pos]) && (bptr[pos] != 0)) | ||
2113 | + pos++; | ||
2114 | |||
2115 | - if (bptr[pos] == ref->path[pos]) { | ||
2116 | - val = g_strdup(""); | ||
2117 | - goto done; /* (I can't imagine why anyone would do this) */ | ||
2118 | - } | ||
2119 | + if (bptr[pos] == ref->path[pos]) { | ||
2120 | + val = g_strdup(""); | ||
2121 | + goto done; /* (I can't imagine why anyone would do this) */ | ||
2122 | + } | ||
2123 | |||
2124 | - /* | ||
2125 | - * In URI, "back up" to the last '/' encountered. This will be the | ||
2126 | - * beginning of the "unique" suffix of URI | ||
2127 | - */ | ||
2128 | - ix = pos; | ||
2129 | - if ((ref->path[ix] == '/') && (ix > 0)) | ||
2130 | - ix--; | ||
2131 | - else if ((ref->path[ix] == 0) && (ix > 1) && (ref->path[ix - 1] == '/')) | ||
2132 | - ix -= 2; | ||
2133 | - for (; ix > 0; ix--) { | ||
2134 | - if (ref->path[ix] == '/') | ||
2135 | - break; | ||
2136 | - } | ||
2137 | - if (ix == 0) { | ||
2138 | - uptr = ref->path; | ||
2139 | - } else { | ||
2140 | - ix++; | ||
2141 | - uptr = &ref->path[ix]; | ||
2142 | - } | ||
2143 | + /* | ||
2144 | + * In URI, "back up" to the last '/' encountered. This will be the | ||
2145 | + * beginning of the "unique" suffix of URI | ||
2146 | + */ | ||
2147 | + ix = pos; | ||
2148 | + if ((ref->path[ix] == '/') && (ix > 0)) | ||
2149 | + ix--; | ||
2150 | + else if ((ref->path[ix] == 0) && (ix > 1) && (ref->path[ix - 1] == '/')) | ||
2151 | + ix -= 2; | ||
2152 | + for (; ix > 0; ix--) { | ||
2153 | + if (ref->path[ix] == '/') | ||
2154 | + break; | ||
2155 | + } | ||
2156 | + if (ix == 0) { | ||
2157 | + uptr = ref->path; | ||
2158 | + } else { | ||
2159 | + ix++; | ||
2160 | + uptr = &ref->path[ix]; | ||
2161 | + } | ||
2162 | |||
2163 | - /* | ||
2164 | - * In base, count the number of '/' from the differing point | ||
2165 | - */ | ||
2166 | - if (bptr[pos] != ref->path[pos]) {/* check for trivial URI == base */ | ||
2167 | - for (; bptr[ix] != 0; ix++) { | ||
2168 | - if (bptr[ix] == '/') | ||
2169 | - nbslash++; | ||
2170 | - } | ||
2171 | - } | ||
2172 | - len = strlen (uptr) + 1; | ||
2173 | + /* | ||
2174 | + * In base, count the number of '/' from the differing point | ||
2175 | + */ | ||
2176 | + if (bptr[pos] != ref->path[pos]) { /* check for trivial URI == base */ | ||
2177 | + for (; bptr[ix] != 0; ix++) { | ||
2178 | + if (bptr[ix] == '/') | ||
2179 | + nbslash++; | ||
2180 | + } | ||
2181 | + } | ||
2182 | + len = strlen(uptr) + 1; | ||
2183 | } | ||
2184 | |||
2185 | if (nbslash == 0) { | ||
2186 | - if (uptr != NULL) | ||
2187 | - /* exception characters from uri_to_string */ | ||
2188 | - val = uri_string_escape(uptr, "/;&=+$,"); | ||
2189 | - goto done; | ||
2190 | + if (uptr != NULL) | ||
2191 | + /* exception characters from uri_to_string */ | ||
2192 | + val = uri_string_escape(uptr, "/;&=+$,"); | ||
2193 | + goto done; | ||
2194 | } | ||
2195 | |||
2196 | /* | ||
2197 | @@ -XXX,XX +XXX,XX @@ uri_resolve_relative (const char *uri, const char * base) | ||
2198 | * length of the remainder of the URI, plus enough space | ||
2199 | * for the "../" groups, plus one for the terminator | ||
2200 | */ | ||
2201 | - val = g_malloc (len + 3 * nbslash); | ||
2202 | + val = g_malloc(len + 3 * nbslash); | ||
2203 | vptr = val; | ||
2204 | /* | ||
2205 | * Put in as many "../" as needed | ||
2206 | */ | ||
2207 | - for (; nbslash>0; nbslash--) { | ||
2208 | - *vptr++ = '.'; | ||
2209 | - *vptr++ = '.'; | ||
2210 | - *vptr++ = '/'; | ||
2211 | + for (; nbslash > 0; nbslash--) { | ||
2212 | + *vptr++ = '.'; | ||
2213 | + *vptr++ = '.'; | ||
2214 | + *vptr++ = '/'; | ||
2215 | } | ||
2216 | /* | ||
2217 | * Finish up with the end of the URI | ||
2218 | */ | ||
2219 | if (uptr != NULL) { | ||
2220 | - if ((vptr > val) && (len > 0) && | ||
2221 | - (uptr[0] == '/') && (vptr[-1] == '/')) { | ||
2222 | - memcpy (vptr, uptr + 1, len - 1); | ||
2223 | - vptr[len - 2] = 0; | ||
2224 | - } else { | ||
2225 | - memcpy (vptr, uptr, len); | ||
2226 | - vptr[len - 1] = 0; | ||
2227 | - } | ||
2228 | + if ((vptr > val) && (len > 0) && (uptr[0] == '/') && | ||
2229 | + (vptr[-1] == '/')) { | ||
2230 | + memcpy(vptr, uptr + 1, len - 1); | ||
2231 | + vptr[len - 2] = 0; | ||
2232 | + } else { | ||
2233 | + memcpy(vptr, uptr, len); | ||
2234 | + vptr[len - 1] = 0; | ||
2235 | + } | ||
2236 | } else { | ||
2237 | - vptr[len - 1] = 0; | ||
2238 | + vptr[len - 1] = 0; | ||
2239 | } | ||
2240 | |||
2241 | /* escape the freshly-built path */ | ||
2242 | vptr = val; | ||
2243 | - /* exception characters from uri_to_string */ | ||
2244 | + /* exception characters from uri_to_string */ | ||
2245 | val = uri_string_escape(vptr, "/;&=+$,"); | ||
2246 | g_free(vptr); | ||
2247 | |||
2248 | @@ -XXX,XX +XXX,XX @@ done: | ||
2249 | if (remove_path != 0) | ||
2250 | ref->path = NULL; | ||
2251 | if (ref != NULL) | ||
2252 | - uri_free (ref); | ||
2253 | + uri_free(ref); | ||
2254 | if (bas != NULL) | ||
2255 | - uri_free (bas); | ||
2256 | + uri_free(bas); | ||
2257 | |||
2258 | return val; | ||
2259 | } | ||
2260 | @@ -XXX,XX +XXX,XX @@ done: | ||
2261 | * Utility functions to help parse and assemble query strings. | ||
2262 | */ | ||
2263 | |||
2264 | -struct QueryParams * | ||
2265 | -query_params_new (int init_alloc) | ||
2266 | +struct QueryParams *query_params_new(int init_alloc) | ||
2267 | { | ||
2268 | struct QueryParams *ps; | ||
2269 | |||
2270 | - if (init_alloc <= 0) init_alloc = 1; | ||
2271 | + if (init_alloc <= 0) | ||
2272 | + init_alloc = 1; | ||
2273 | |||
2274 | ps = g_new(QueryParams, 1); | ||
2275 | ps->n = 0; | ||
2276 | @@ -XXX,XX +XXX,XX @@ query_params_new (int init_alloc) | ||
2277 | /* Ensure there is space to store at least one more parameter | ||
2278 | * at the end of the set. | ||
2279 | */ | ||
2280 | -static int | ||
2281 | -query_params_append (struct QueryParams *ps, | ||
2282 | - const char *name, const char *value) | ||
2283 | +static int query_params_append(struct QueryParams *ps, const char *name, | ||
2284 | + const char *value) | ||
2285 | { | ||
2286 | if (ps->n >= ps->alloc) { | ||
2287 | ps->p = g_renew(QueryParam, ps->p, ps->alloc * 2); | ||
2288 | @@ -XXX,XX +XXX,XX @@ query_params_append (struct QueryParams *ps, | ||
2289 | return 0; | ||
2290 | } | ||
2291 | |||
2292 | -void | ||
2293 | -query_params_free (struct QueryParams *ps) | ||
2294 | +void query_params_free(struct QueryParams *ps) | ||
2295 | { | ||
2296 | int i; | ||
2297 | |||
2298 | for (i = 0; i < ps->n; ++i) { | ||
2299 | - g_free (ps->p[i].name); | ||
2300 | - g_free (ps->p[i].value); | ||
2301 | + g_free(ps->p[i].name); | ||
2302 | + g_free(ps->p[i].value); | ||
2303 | } | ||
2304 | - g_free (ps->p); | ||
2305 | - g_free (ps); | ||
2306 | + g_free(ps->p); | ||
2307 | + g_free(ps); | ||
2308 | } | ||
2309 | |||
2310 | -struct QueryParams * | ||
2311 | -query_params_parse (const char *query) | ||
2312 | +struct QueryParams *query_params_parse(const char *query) | ||
2313 | { | ||
2314 | struct QueryParams *ps; | ||
2315 | const char *end, *eq; | ||
2316 | |||
2317 | - ps = query_params_new (0); | ||
2318 | - if (!query || query[0] == '\0') return ps; | ||
2319 | + ps = query_params_new(0); | ||
2320 | + if (!query || query[0] == '\0') | ||
2321 | + return ps; | ||
2322 | |||
2323 | while (*query) { | ||
2324 | char *name = NULL, *value = NULL; | ||
2325 | |||
2326 | /* Find the next separator, or end of the string. */ | ||
2327 | - end = strchr (query, '&'); | ||
2328 | + end = strchr(query, '&'); | ||
2329 | if (!end) | ||
2330 | - end = strchr (query, ';'); | ||
2331 | + end = strchr(query, ';'); | ||
2332 | if (!end) | ||
2333 | - end = query + strlen (query); | ||
2334 | + end = query + strlen(query); | ||
2335 | |||
2336 | /* Find the first '=' character between here and end. */ | ||
2337 | - eq = strchr (query, '='); | ||
2338 | - if (eq && eq >= end) eq = NULL; | ||
2339 | + eq = strchr(query, '='); | ||
2340 | + if (eq && eq >= end) | ||
2341 | + eq = NULL; | ||
2342 | |||
2343 | /* Empty section (eg. "&&"). */ | ||
2344 | if (end == query) | ||
2345 | @@ -XXX,XX +XXX,XX @@ query_params_parse (const char *query) | ||
2346 | * and consistent with CGI.pm we assume value is "". | ||
2347 | */ | ||
2348 | else if (!eq) { | ||
2349 | - name = uri_string_unescape (query, end - query, NULL); | ||
2350 | + name = uri_string_unescape(query, end - query, NULL); | ||
2351 | value = NULL; | ||
2352 | } | ||
2353 | /* Or if we have "name=" here (works around annoying | ||
2354 | * problem when calling uri_string_unescape with len = 0). | ||
2355 | */ | ||
2356 | - else if (eq+1 == end) { | ||
2357 | - name = uri_string_unescape (query, eq - query, NULL); | ||
2358 | + else if (eq + 1 == end) { | ||
2359 | + name = uri_string_unescape(query, eq - query, NULL); | ||
2360 | value = g_new0(char, 1); | ||
2361 | } | ||
2362 | /* If the '=' character is at the beginning then we have | ||
2363 | @@ -XXX,XX +XXX,XX @@ query_params_parse (const char *query) | ||
2364 | |||
2365 | /* Otherwise it's "name=value". */ | ||
2366 | else { | ||
2367 | - name = uri_string_unescape (query, eq - query, NULL); | ||
2368 | - value = uri_string_unescape (eq+1, end - (eq+1), NULL); | ||
2369 | + name = uri_string_unescape(query, eq - query, NULL); | ||
2370 | + value = uri_string_unescape(eq + 1, end - (eq + 1), NULL); | ||
2371 | } | ||
2372 | |||
2373 | /* Append to the parameter set. */ | ||
2374 | - query_params_append (ps, name, value); | ||
2375 | + query_params_append(ps, name, value); | ||
2376 | g_free(name); | ||
2377 | g_free(value); | ||
2378 | |||
2379 | next: | ||
2380 | query = end; | ||
2381 | - if (*query) query ++; /* skip '&' separator */ | ||
2382 | + if (*query) | ||
2383 | + query++; /* skip '&' separator */ | ||
2384 | } | ||
2385 | |||
2386 | return ps; | ||
2387 | -- | 157 | -- |
2388 | 2.14.3 | 158 | 2.26.2 |
2389 | 159 | ||
2390 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
1 | 2 | ||
3 | In order to reuse bdrv_common_block_status_above in | ||
4 | bdrv_is_allocated_above, let's support include_base parameter. | ||
5 | |||
6 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
7 | Reviewed-by: Alberto Garcia <berto@igalia.com> | ||
8 | Reviewed-by: Eric Blake <eblake@redhat.com> | ||
9 | Message-id: 20200924194003.22080-3-vsementsov@virtuozzo.com | ||
10 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
11 | --- | ||
12 | block/coroutines.h | 2 ++ | ||
13 | block/io.c | 21 ++++++++++++++------- | ||
14 | 2 files changed, 16 insertions(+), 7 deletions(-) | ||
15 | |||
16 | diff --git a/block/coroutines.h b/block/coroutines.h | ||
17 | index XXXXXXX..XXXXXXX 100644 | ||
18 | --- a/block/coroutines.h | ||
19 | +++ b/block/coroutines.h | ||
20 | @@ -XXX,XX +XXX,XX @@ bdrv_pwritev(BdrvChild *child, int64_t offset, unsigned int bytes, | ||
21 | int coroutine_fn | ||
22 | bdrv_co_common_block_status_above(BlockDriverState *bs, | ||
23 | BlockDriverState *base, | ||
24 | + bool include_base, | ||
25 | bool want_zero, | ||
26 | int64_t offset, | ||
27 | int64_t bytes, | ||
28 | @@ -XXX,XX +XXX,XX @@ bdrv_co_common_block_status_above(BlockDriverState *bs, | ||
29 | int generated_co_wrapper | ||
30 | bdrv_common_block_status_above(BlockDriverState *bs, | ||
31 | BlockDriverState *base, | ||
32 | + bool include_base, | ||
33 | bool want_zero, | ||
34 | int64_t offset, | ||
35 | int64_t bytes, | ||
36 | diff --git a/block/io.c b/block/io.c | ||
37 | index XXXXXXX..XXXXXXX 100644 | ||
38 | --- a/block/io.c | ||
39 | +++ b/block/io.c | ||
40 | @@ -XXX,XX +XXX,XX @@ early_out: | ||
41 | int coroutine_fn | ||
42 | bdrv_co_common_block_status_above(BlockDriverState *bs, | ||
43 | BlockDriverState *base, | ||
44 | + bool include_base, | ||
45 | bool want_zero, | ||
46 | int64_t offset, | ||
47 | int64_t bytes, | ||
48 | @@ -XXX,XX +XXX,XX @@ bdrv_co_common_block_status_above(BlockDriverState *bs, | ||
49 | BlockDriverState *p; | ||
50 | int64_t eof = 0; | ||
51 | |||
52 | - assert(bs != base); | ||
53 | + assert(include_base || bs != base); | ||
54 | + assert(!include_base || base); /* Can't include NULL base */ | ||
55 | |||
56 | ret = bdrv_co_block_status(bs, want_zero, offset, bytes, pnum, map, file); | ||
57 | - if (ret < 0 || *pnum == 0 || ret & BDRV_BLOCK_ALLOCATED) { | ||
58 | + if (ret < 0 || *pnum == 0 || ret & BDRV_BLOCK_ALLOCATED || bs == base) { | ||
59 | return ret; | ||
60 | } | ||
61 | |||
62 | @@ -XXX,XX +XXX,XX @@ bdrv_co_common_block_status_above(BlockDriverState *bs, | ||
63 | assert(*pnum <= bytes); | ||
64 | bytes = *pnum; | ||
65 | |||
66 | - for (p = bdrv_filter_or_cow_bs(bs); p != base; | ||
67 | + for (p = bdrv_filter_or_cow_bs(bs); include_base || p != base; | ||
68 | p = bdrv_filter_or_cow_bs(p)) | ||
69 | { | ||
70 | ret = bdrv_co_block_status(p, want_zero, offset, bytes, pnum, map, | ||
71 | @@ -XXX,XX +XXX,XX @@ bdrv_co_common_block_status_above(BlockDriverState *bs, | ||
72 | break; | ||
73 | } | ||
74 | |||
75 | + if (p == base) { | ||
76 | + assert(include_base); | ||
77 | + break; | ||
78 | + } | ||
79 | + | ||
80 | /* | ||
81 | * OK, [offset, offset + *pnum) region is unallocated on this layer, | ||
82 | * let's continue the diving. | ||
83 | @@ -XXX,XX +XXX,XX @@ int bdrv_block_status_above(BlockDriverState *bs, BlockDriverState *base, | ||
84 | int64_t offset, int64_t bytes, int64_t *pnum, | ||
85 | int64_t *map, BlockDriverState **file) | ||
86 | { | ||
87 | - return bdrv_common_block_status_above(bs, base, true, offset, bytes, | ||
88 | + return bdrv_common_block_status_above(bs, base, false, true, offset, bytes, | ||
89 | pnum, map, file); | ||
90 | } | ||
91 | |||
92 | @@ -XXX,XX +XXX,XX @@ int coroutine_fn bdrv_is_allocated(BlockDriverState *bs, int64_t offset, | ||
93 | int ret; | ||
94 | int64_t dummy; | ||
95 | |||
96 | - ret = bdrv_common_block_status_above(bs, bdrv_filter_or_cow_bs(bs), false, | ||
97 | - offset, bytes, pnum ? pnum : &dummy, | ||
98 | - NULL, NULL); | ||
99 | + ret = bdrv_common_block_status_above(bs, bs, true, false, offset, | ||
100 | + bytes, pnum ? pnum : &dummy, NULL, | ||
101 | + NULL); | ||
102 | if (ret < 0) { | ||
103 | return ret; | ||
104 | } | ||
105 | -- | ||
106 | 2.26.2 | ||
107 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
1 | 2 | ||
3 | We are going to reuse bdrv_common_block_status_above in | ||
4 | bdrv_is_allocated_above. bdrv_is_allocated_above may be called with | ||
5 | include_base == false and still bs == base (for ex. from img_rebase()). | ||
6 | |||
7 | So, support this corner case. | ||
8 | |||
9 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
10 | Reviewed-by: Kevin Wolf <kwolf@redhat.com> | ||
11 | Reviewed-by: Eric Blake <eblake@redhat.com> | ||
12 | Reviewed-by: Alberto Garcia <berto@igalia.com> | ||
13 | Message-id: 20200924194003.22080-4-vsementsov@virtuozzo.com | ||
14 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
15 | --- | ||
16 | block/io.c | 6 +++++- | ||
17 | 1 file changed, 5 insertions(+), 1 deletion(-) | ||
18 | |||
19 | diff --git a/block/io.c b/block/io.c | ||
20 | index XXXXXXX..XXXXXXX 100644 | ||
21 | --- a/block/io.c | ||
22 | +++ b/block/io.c | ||
23 | @@ -XXX,XX +XXX,XX @@ bdrv_co_common_block_status_above(BlockDriverState *bs, | ||
24 | BlockDriverState *p; | ||
25 | int64_t eof = 0; | ||
26 | |||
27 | - assert(include_base || bs != base); | ||
28 | assert(!include_base || base); /* Can't include NULL base */ | ||
29 | |||
30 | + if (!include_base && bs == base) { | ||
31 | + *pnum = bytes; | ||
32 | + return 0; | ||
33 | + } | ||
34 | + | ||
35 | ret = bdrv_co_block_status(bs, want_zero, offset, bytes, pnum, map, file); | ||
36 | if (ret < 0 || *pnum == 0 || ret & BDRV_BLOCK_ALLOCATED || bs == base) { | ||
37 | return ret; | ||
38 | -- | ||
39 | 2.26.2 | ||
40 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
1 | 2 | ||
3 | bdrv_is_allocated_above wrongly handles short backing files: it reports | ||
4 | after-EOF space as UNALLOCATED which is wrong, as on read the data is | ||
5 | generated on the level of short backing file (if all overlays have | ||
6 | unallocated areas at that place). | ||
7 | |||
8 | Reusing bdrv_common_block_status_above fixes the issue and unifies code | ||
9 | path. | ||
10 | |||
11 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
12 | Reviewed-by: Eric Blake <eblake@redhat.com> | ||
13 | Reviewed-by: Alberto Garcia <berto@igalia.com> | ||
14 | Message-id: 20200924194003.22080-5-vsementsov@virtuozzo.com | ||
15 | [Fix s/has/have/ as suggested by Eric Blake. Fix s/area/areas/. | ||
16 | --Stefan] | ||
17 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
18 | --- | ||
19 | block/io.c | 43 +++++-------------------------------------- | ||
20 | 1 file changed, 5 insertions(+), 38 deletions(-) | ||
21 | |||
22 | diff --git a/block/io.c b/block/io.c | ||
23 | index XXXXXXX..XXXXXXX 100644 | ||
24 | --- a/block/io.c | ||
25 | +++ b/block/io.c | ||
26 | @@ -XXX,XX +XXX,XX @@ int coroutine_fn bdrv_is_allocated(BlockDriverState *bs, int64_t offset, | ||
27 | * at 'offset + *pnum' may return the same allocation status (in other | ||
28 | * words, the result is not necessarily the maximum possible range); | ||
29 | * but 'pnum' will only be 0 when end of file is reached. | ||
30 | - * | ||
31 | */ | ||
32 | int bdrv_is_allocated_above(BlockDriverState *top, | ||
33 | BlockDriverState *base, | ||
34 | bool include_base, int64_t offset, | ||
35 | int64_t bytes, int64_t *pnum) | ||
36 | { | ||
37 | - BlockDriverState *intermediate; | ||
38 | - int ret; | ||
39 | - int64_t n = bytes; | ||
40 | - | ||
41 | - assert(base || !include_base); | ||
42 | - | ||
43 | - intermediate = top; | ||
44 | - while (include_base || intermediate != base) { | ||
45 | - int64_t pnum_inter; | ||
46 | - int64_t size_inter; | ||
47 | - | ||
48 | - assert(intermediate); | ||
49 | - ret = bdrv_is_allocated(intermediate, offset, bytes, &pnum_inter); | ||
50 | - if (ret < 0) { | ||
51 | - return ret; | ||
52 | - } | ||
53 | - if (ret) { | ||
54 | - *pnum = pnum_inter; | ||
55 | - return 1; | ||
56 | - } | ||
57 | - | ||
58 | - size_inter = bdrv_getlength(intermediate); | ||
59 | - if (size_inter < 0) { | ||
60 | - return size_inter; | ||
61 | - } | ||
62 | - if (n > pnum_inter && | ||
63 | - (intermediate == top || offset + pnum_inter < size_inter)) { | ||
64 | - n = pnum_inter; | ||
65 | - } | ||
66 | - | ||
67 | - if (intermediate == base) { | ||
68 | - break; | ||
69 | - } | ||
70 | - | ||
71 | - intermediate = bdrv_filter_or_cow_bs(intermediate); | ||
72 | + int ret = bdrv_common_block_status_above(top, base, include_base, false, | ||
73 | + offset, bytes, pnum, NULL, NULL); | ||
74 | + if (ret < 0) { | ||
75 | + return ret; | ||
76 | } | ||
77 | |||
78 | - *pnum = n; | ||
79 | - return 0; | ||
80 | + return !!(ret & BDRV_BLOCK_ALLOCATED); | ||
81 | } | ||
82 | |||
83 | int coroutine_fn | ||
84 | -- | ||
85 | 2.26.2 | ||
86 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
1 | 2 | ||
3 | These cases are fixed by previous patches around block_status and | ||
4 | is_allocated. | ||
5 | |||
6 | Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> | ||
7 | Reviewed-by: Eric Blake <eblake@redhat.com> | ||
8 | Reviewed-by: Alberto Garcia <berto@igalia.com> | ||
9 | Message-id: 20200924194003.22080-6-vsementsov@virtuozzo.com | ||
10 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
11 | --- | ||
12 | tests/qemu-iotests/274 | 20 +++++++++++ | ||
13 | tests/qemu-iotests/274.out | 68 ++++++++++++++++++++++++++++++++++++++ | ||
14 | 2 files changed, 88 insertions(+) | ||
15 | |||
16 | diff --git a/tests/qemu-iotests/274 b/tests/qemu-iotests/274 | ||
17 | index XXXXXXX..XXXXXXX 100755 | ||
18 | --- a/tests/qemu-iotests/274 | ||
19 | +++ b/tests/qemu-iotests/274 | ||
20 | @@ -XXX,XX +XXX,XX @@ with iotests.FilePath('base') as base, \ | ||
21 | iotests.qemu_io_log('-c', 'read -P 1 0 %d' % size_short, mid) | ||
22 | iotests.qemu_io_log('-c', 'read -P 0 %d %d' % (size_short, size_diff), mid) | ||
23 | |||
24 | + iotests.log('=== Testing qemu-img commit (top -> base) ===') | ||
25 | + | ||
26 | + create_chain() | ||
27 | + iotests.qemu_img_log('commit', '-b', base, top) | ||
28 | + iotests.img_info_log(base) | ||
29 | + iotests.qemu_io_log('-c', 'read -P 1 0 %d' % size_short, base) | ||
30 | + iotests.qemu_io_log('-c', 'read -P 0 %d %d' % (size_short, size_diff), base) | ||
31 | + | ||
32 | + iotests.log('=== Testing QMP active commit (top -> base) ===') | ||
33 | + | ||
34 | + create_chain() | ||
35 | + with create_vm() as vm: | ||
36 | + vm.launch() | ||
37 | + vm.qmp_log('block-commit', device='top', base_node='base', | ||
38 | + job_id='job0', auto_dismiss=False) | ||
39 | + vm.run_job('job0', wait=5) | ||
40 | + | ||
41 | + iotests.img_info_log(mid) | ||
42 | + iotests.qemu_io_log('-c', 'read -P 1 0 %d' % size_short, base) | ||
43 | + iotests.qemu_io_log('-c', 'read -P 0 %d %d' % (size_short, size_diff), base) | ||
44 | |||
45 | iotests.log('== Resize tests ==') | ||
46 | |||
47 | diff --git a/tests/qemu-iotests/274.out b/tests/qemu-iotests/274.out | ||
48 | index XXXXXXX..XXXXXXX 100644 | ||
49 | --- a/tests/qemu-iotests/274.out | ||
50 | +++ b/tests/qemu-iotests/274.out | ||
51 | @@ -XXX,XX +XXX,XX @@ read 1048576/1048576 bytes at offset 0 | ||
52 | read 1048576/1048576 bytes at offset 1048576 | ||
53 | 1 MiB, X ops; XX:XX:XX.X (XXX YYY/sec and XXX ops/sec) | ||
54 | |||
55 | +=== Testing qemu-img commit (top -> base) === | ||
56 | +Formatting 'TEST_DIR/PID-base', fmt=qcow2 cluster_size=65536 extended_l2=off compression_type=zlib size=2097152 lazy_refcounts=off refcount_bits=16 | ||
57 | + | ||
58 | +Formatting 'TEST_DIR/PID-mid', fmt=qcow2 cluster_size=65536 extended_l2=off compression_type=zlib size=1048576 backing_file=TEST_DIR/PID-base backing_fmt=qcow2 lazy_refcounts=off refcount_bits=16 | ||
59 | + | ||
60 | +Formatting 'TEST_DIR/PID-top', fmt=qcow2 cluster_size=65536 extended_l2=off compression_type=zlib size=2097152 backing_file=TEST_DIR/PID-mid backing_fmt=qcow2 lazy_refcounts=off refcount_bits=16 | ||
61 | + | ||
62 | +wrote 2097152/2097152 bytes at offset 0 | ||
63 | +2 MiB, X ops; XX:XX:XX.X (XXX YYY/sec and XXX ops/sec) | ||
64 | + | ||
65 | +Image committed. | ||
66 | + | ||
67 | +image: TEST_IMG | ||
68 | +file format: IMGFMT | ||
69 | +virtual size: 2 MiB (2097152 bytes) | ||
70 | +cluster_size: 65536 | ||
71 | +Format specific information: | ||
72 | + compat: 1.1 | ||
73 | + compression type: zlib | ||
74 | + lazy refcounts: false | ||
75 | + refcount bits: 16 | ||
76 | + corrupt: false | ||
77 | + extended l2: false | ||
78 | + | ||
79 | +read 1048576/1048576 bytes at offset 0 | ||
80 | +1 MiB, X ops; XX:XX:XX.X (XXX YYY/sec and XXX ops/sec) | ||
81 | + | ||
82 | +read 1048576/1048576 bytes at offset 1048576 | ||
83 | +1 MiB, X ops; XX:XX:XX.X (XXX YYY/sec and XXX ops/sec) | ||
84 | + | ||
85 | +=== Testing QMP active commit (top -> base) === | ||
86 | +Formatting 'TEST_DIR/PID-base', fmt=qcow2 cluster_size=65536 extended_l2=off compression_type=zlib size=2097152 lazy_refcounts=off refcount_bits=16 | ||
87 | + | ||
88 | +Formatting 'TEST_DIR/PID-mid', fmt=qcow2 cluster_size=65536 extended_l2=off compression_type=zlib size=1048576 backing_file=TEST_DIR/PID-base backing_fmt=qcow2 lazy_refcounts=off refcount_bits=16 | ||
89 | + | ||
90 | +Formatting 'TEST_DIR/PID-top', fmt=qcow2 cluster_size=65536 extended_l2=off compression_type=zlib size=2097152 backing_file=TEST_DIR/PID-mid backing_fmt=qcow2 lazy_refcounts=off refcount_bits=16 | ||
91 | + | ||
92 | +wrote 2097152/2097152 bytes at offset 0 | ||
93 | +2 MiB, X ops; XX:XX:XX.X (XXX YYY/sec and XXX ops/sec) | ||
94 | + | ||
95 | +{"execute": "block-commit", "arguments": {"auto-dismiss": false, "base-node": "base", "device": "top", "job-id": "job0"}} | ||
96 | +{"return": {}} | ||
97 | +{"execute": "job-complete", "arguments": {"id": "job0"}} | ||
98 | +{"return": {}} | ||
99 | +{"data": {"device": "job0", "len": 1048576, "offset": 1048576, "speed": 0, "type": "commit"}, "event": "BLOCK_JOB_READY", "timestamp": {"microseconds": "USECS", "seconds": "SECS"}} | ||
100 | +{"data": {"device": "job0", "len": 1048576, "offset": 1048576, "speed": 0, "type": "commit"}, "event": "BLOCK_JOB_COMPLETED", "timestamp": {"microseconds": "USECS", "seconds": "SECS"}} | ||
101 | +{"execute": "job-dismiss", "arguments": {"id": "job0"}} | ||
102 | +{"return": {}} | ||
103 | +image: TEST_IMG | ||
104 | +file format: IMGFMT | ||
105 | +virtual size: 1 MiB (1048576 bytes) | ||
106 | +cluster_size: 65536 | ||
107 | +backing file: TEST_DIR/PID-base | ||
108 | +backing file format: IMGFMT | ||
109 | +Format specific information: | ||
110 | + compat: 1.1 | ||
111 | + compression type: zlib | ||
112 | + lazy refcounts: false | ||
113 | + refcount bits: 16 | ||
114 | + corrupt: false | ||
115 | + extended l2: false | ||
116 | + | ||
117 | +read 1048576/1048576 bytes at offset 0 | ||
118 | +1 MiB, X ops; XX:XX:XX.X (XXX YYY/sec and XXX ops/sec) | ||
119 | + | ||
120 | +read 1048576/1048576 bytes at offset 1048576 | ||
121 | +1 MiB, X ops; XX:XX:XX.X (XXX YYY/sec and XXX ops/sec) | ||
122 | + | ||
123 | == Resize tests == | ||
124 | === preallocation=off === | ||
125 | Formatting 'TEST_DIR/PID-base', fmt=qcow2 cluster_size=65536 extended_l2=off compression_type=zlib size=6442450944 lazy_refcounts=off refcount_bits=16 | ||
126 | -- | ||
127 | 2.26.2 | ||
128 | diff view generated by jsdifflib |