]>
Commit | Line | Data |
---|---|---|
3c56521b | 1 | /* |
585d0ed9 | 2 | * QEMU Block driver for CLOOP images |
5fafdf24 | 3 | * |
3c56521b | 4 | * Copyright (c) 2004 Johannes E. Schindelin |
5fafdf24 | 5 | * |
3c56521b FB |
6 | * Permission is hereby granted, free of charge, to any person obtaining a copy |
7 | * of this software and associated documentation files (the "Software"), to deal | |
8 | * in the Software without restriction, including without limitation the rights | |
9 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
10 | * copies of the Software, and to permit persons to whom the Software is | |
11 | * furnished to do so, subject to the following conditions: | |
12 | * | |
13 | * The above copyright notice and this permission notice shall be included in | |
14 | * all copies or substantial portions of the Software. | |
15 | * | |
16 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
17 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
18 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL | |
19 | * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
20 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
21 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | |
22 | * THE SOFTWARE. | |
23 | */ | |
80c71a24 | 24 | #include "qemu/osdep.h" |
da34e65c | 25 | #include "qapi/error.h" |
398e6ad0 | 26 | #include "qemu/error-report.h" |
e2c1c34f | 27 | #include "block/block-io.h" |
737e150e | 28 | #include "block/block_int.h" |
1de7afc9 | 29 | #include "qemu/module.h" |
58369e22 | 30 | #include "qemu/bswap.h" |
3c56521b FB |
31 | #include <zlib.h> |
32 | ||
d65f97a8 SH |
33 | /* Maximum compressed block size */ |
34 | #define MAX_BLOCK_SIZE (64 * 1024 * 1024) | |
35 | ||
3c56521b | 36 | typedef struct BDRVCloopState { |
848c66e8 | 37 | CoMutex lock; |
3c56521b FB |
38 | uint32_t block_size; |
39 | uint32_t n_blocks; | |
5b47b7c3 | 40 | uint64_t *offsets; |
3c56521b FB |
41 | uint32_t sectors_per_block; |
42 | uint32_t current_block; | |
28a5c9c8 FB |
43 | uint8_t *compressed_block; |
44 | uint8_t *uncompressed_block; | |
3c56521b FB |
45 | z_stream zstream; |
46 | } BDRVCloopState; | |
47 | ||
48 | static int cloop_probe(const uint8_t *buf, int buf_size, const char *filename) | |
49 | { | |
5b47b7c3 DXW |
50 | const char *magic_version_2_0 = "#!/bin/sh\n" |
51 | "#V2.0 Format\n" | |
52 | "modprobe cloop file=$0 && mount -r -t iso9660 /dev/cloop $1\n"; | |
53 | int length = strlen(magic_version_2_0); | |
54 | if (length > buf_size) { | |
55 | length = buf_size; | |
56 | } | |
57 | if (!memcmp(magic_version_2_0, buf, length)) { | |
58 | return 2; | |
59 | } | |
3c56521b FB |
60 | return 0; |
61 | } | |
62 | ||
015a1036 HR |
63 | static int cloop_open(BlockDriverState *bs, QDict *options, int flags, |
64 | Error **errp) | |
3c56521b FB |
65 | { |
66 | BDRVCloopState *s = bs->opaque; | |
5b47b7c3 | 67 | uint32_t offsets_size, max_compressed_block_size = 1, i; |
1a60657f | 68 | int ret; |
3c56521b | 69 | |
a4b740db KW |
70 | GLOBAL_STATE_CODE(); |
71 | ||
018f9dea | 72 | bdrv_graph_rdlock_main_loop(); |
eaa2410f | 73 | ret = bdrv_apply_auto_read_only(bs, NULL, errp); |
018f9dea | 74 | bdrv_graph_rdunlock_main_loop(); |
eaa2410f KW |
75 | if (ret < 0) { |
76 | return ret; | |
77 | } | |
78 | ||
83930780 VSO |
79 | ret = bdrv_open_file_child(NULL, options, "file", bs, errp); |
80 | if (ret < 0) { | |
81 | return ret; | |
4e4bf5c4 KW |
82 | } |
83 | ||
a4b740db KW |
84 | GRAPH_RDLOCK_GUARD_MAINLOOP(); |
85 | ||
3c56521b | 86 | /* read header */ |
32cc71de | 87 | ret = bdrv_pread(bs->file, 128, 4, &s->block_size, 0); |
1a60657f KW |
88 | if (ret < 0) { |
89 | return ret; | |
3c56521b | 90 | } |
c94304be | 91 | s->block_size = be32_to_cpu(s->block_size); |
d65f97a8 | 92 | if (s->block_size % 512) { |
370e6816 | 93 | error_setg(errp, "block_size %" PRIu32 " must be a multiple of 512", |
d65f97a8 SH |
94 | s->block_size); |
95 | return -EINVAL; | |
96 | } | |
97 | if (s->block_size == 0) { | |
98 | error_setg(errp, "block_size cannot be zero"); | |
99 | return -EINVAL; | |
100 | } | |
101 | ||
102 | /* cloop's create_compressed_fs.c warns about block sizes beyond 256 KB but | |
103 | * we can accept more. Prevent ridiculous values like 4 GB - 1 since we | |
104 | * need a buffer this big. | |
105 | */ | |
106 | if (s->block_size > MAX_BLOCK_SIZE) { | |
370e6816 | 107 | error_setg(errp, "block_size %" PRIu32 " must be %u MB or less", |
d65f97a8 SH |
108 | s->block_size, |
109 | MAX_BLOCK_SIZE / (1024 * 1024)); | |
110 | return -EINVAL; | |
111 | } | |
c94304be | 112 | |
32cc71de | 113 | ret = bdrv_pread(bs->file, 128 + 4, 4, &s->n_blocks, 0); |
1a60657f KW |
114 | if (ret < 0) { |
115 | return ret; | |
c94304be CH |
116 | } |
117 | s->n_blocks = be32_to_cpu(s->n_blocks); | |
3c56521b FB |
118 | |
119 | /* read offsets */ | |
42d43d35 | 120 | if (s->n_blocks > (UINT32_MAX - 1) / sizeof(uint64_t)) { |
509a41ba | 121 | /* Prevent integer overflow */ |
370e6816 | 122 | error_setg(errp, "n_blocks %" PRIu32 " must be %zu or less", |
509a41ba | 123 | s->n_blocks, |
42d43d35 | 124 | (UINT32_MAX - 1) / sizeof(uint64_t)); |
509a41ba SH |
125 | return -EINVAL; |
126 | } | |
42d43d35 | 127 | offsets_size = (s->n_blocks + 1) * sizeof(uint64_t); |
7b103b36 SH |
128 | if (offsets_size > 512 * 1024 * 1024) { |
129 | /* Prevent ridiculous offsets_size which causes memory allocation to | |
130 | * fail or overflows bdrv_pread() size. In practice the 512 MB | |
131 | * offsets[] limit supports 16 TB images at 256 KB block size. | |
132 | */ | |
133 | error_setg(errp, "image requires too many offsets, " | |
134 | "try increasing block size"); | |
135 | return -EINVAL; | |
136 | } | |
4ae7a52e KW |
137 | |
138 | s->offsets = g_try_malloc(offsets_size); | |
139 | if (s->offsets == NULL) { | |
140 | error_setg(errp, "Could not allocate offsets table"); | |
141 | return -ENOMEM; | |
142 | } | |
1a60657f | 143 | |
32cc71de | 144 | ret = bdrv_pread(bs->file, 128 + 4 + 4, offsets_size, s->offsets, 0); |
1a60657f KW |
145 | if (ret < 0) { |
146 | goto fail; | |
c94304be | 147 | } |
1a60657f | 148 | |
42d43d35 | 149 | for (i = 0; i < s->n_blocks + 1; i++) { |
f56b9bc3 SH |
150 | uint64_t size; |
151 | ||
5b47b7c3 | 152 | s->offsets[i] = be64_to_cpu(s->offsets[i]); |
f56b9bc3 SH |
153 | if (i == 0) { |
154 | continue; | |
155 | } | |
156 | ||
157 | if (s->offsets[i] < s->offsets[i - 1]) { | |
158 | error_setg(errp, "offsets not monotonically increasing at " | |
370e6816 | 159 | "index %" PRIu32 ", image file is corrupt", i); |
f56b9bc3 SH |
160 | ret = -EINVAL; |
161 | goto fail; | |
162 | } | |
163 | ||
164 | size = s->offsets[i] - s->offsets[i - 1]; | |
165 | ||
166 | /* Compressed blocks should be smaller than the uncompressed block size | |
167 | * but maybe compression performed poorly so the compressed block is | |
168 | * actually bigger. Clamp down on unrealistic values to prevent | |
169 | * ridiculous s->compressed_block allocation. | |
170 | */ | |
171 | if (size > 2 * MAX_BLOCK_SIZE) { | |
370e6816 SH |
172 | error_setg(errp, "invalid compressed block size at index %" PRIu32 |
173 | ", image file is corrupt", i); | |
f56b9bc3 SH |
174 | ret = -EINVAL; |
175 | goto fail; | |
176 | } | |
177 | ||
178 | if (size > max_compressed_block_size) { | |
179 | max_compressed_block_size = size; | |
5b47b7c3 | 180 | } |
3c56521b FB |
181 | } |
182 | ||
183 | /* initialize zlib engine */ | |
4ae7a52e KW |
184 | s->compressed_block = g_try_malloc(max_compressed_block_size + 1); |
185 | if (s->compressed_block == NULL) { | |
186 | error_setg(errp, "Could not allocate compressed_block"); | |
187 | ret = -ENOMEM; | |
188 | goto fail; | |
189 | } | |
190 | ||
191 | s->uncompressed_block = g_try_malloc(s->block_size); | |
192 | if (s->uncompressed_block == NULL) { | |
193 | error_setg(errp, "Could not allocate uncompressed_block"); | |
194 | ret = -ENOMEM; | |
195 | goto fail; | |
196 | } | |
197 | ||
5b47b7c3 | 198 | if (inflateInit(&s->zstream) != Z_OK) { |
1a60657f KW |
199 | ret = -EINVAL; |
200 | goto fail; | |
5b47b7c3 DXW |
201 | } |
202 | s->current_block = s->n_blocks; | |
3b46e624 | 203 | |
3c56521b | 204 | s->sectors_per_block = s->block_size/512; |
5b47b7c3 | 205 | bs->total_sectors = s->n_blocks * s->sectors_per_block; |
848c66e8 | 206 | qemu_co_mutex_init(&s->lock); |
3c56521b | 207 | return 0; |
c94304be | 208 | |
1a60657f KW |
209 | fail: |
210 | g_free(s->offsets); | |
211 | g_free(s->compressed_block); | |
212 | g_free(s->uncompressed_block); | |
213 | return ret; | |
3c56521b FB |
214 | } |
215 | ||
a6506481 EB |
216 | static void cloop_refresh_limits(BlockDriverState *bs, Error **errp) |
217 | { | |
a5b8dd2c | 218 | bs->bl.request_alignment = BDRV_SECTOR_SIZE; /* No sub-sector I/O */ |
a6506481 EB |
219 | } |
220 | ||
cf8d4c58 PB |
221 | static int coroutine_fn GRAPH_RDLOCK |
222 | cloop_read_block(BlockDriverState *bs, int block_num) | |
3c56521b | 223 | { |
20be49e4 CH |
224 | BDRVCloopState *s = bs->opaque; |
225 | ||
5b47b7c3 DXW |
226 | if (s->current_block != block_num) { |
227 | int ret; | |
228 | uint32_t bytes = s->offsets[block_num + 1] - s->offsets[block_num]; | |
3b46e624 | 229 | |
cf8d4c58 PB |
230 | ret = bdrv_co_pread(bs->file, s->offsets[block_num], bytes, |
231 | s->compressed_block, 0); | |
353a5d84 | 232 | if (ret < 0) { |
3c56521b | 233 | return -1; |
5b47b7c3 DXW |
234 | } |
235 | ||
236 | s->zstream.next_in = s->compressed_block; | |
237 | s->zstream.avail_in = bytes; | |
238 | s->zstream.next_out = s->uncompressed_block; | |
239 | s->zstream.avail_out = s->block_size; | |
240 | ret = inflateReset(&s->zstream); | |
241 | if (ret != Z_OK) { | |
242 | return -1; | |
243 | } | |
244 | ret = inflate(&s->zstream, Z_FINISH); | |
245 | if (ret != Z_STREAM_END || s->zstream.total_out != s->block_size) { | |
246 | return -1; | |
247 | } | |
5fafdf24 | 248 | |
5b47b7c3 | 249 | s->current_block = block_num; |
3c56521b FB |
250 | } |
251 | return 0; | |
252 | } | |
253 | ||
cf8d4c58 | 254 | static int coroutine_fn GRAPH_RDLOCK |
f7ef38dd VSO |
255 | cloop_co_preadv(BlockDriverState *bs, int64_t offset, int64_t bytes, |
256 | QEMUIOVector *qiov, BdrvRequestFlags flags) | |
3c56521b FB |
257 | { |
258 | BDRVCloopState *s = bs->opaque; | |
5cd23081 KW |
259 | uint64_t sector_num = offset >> BDRV_SECTOR_BITS; |
260 | int nb_sectors = bytes >> BDRV_SECTOR_BITS; | |
261 | int ret, i; | |
262 | ||
1bbbf32d NS |
263 | assert(QEMU_IS_ALIGNED(offset, BDRV_SECTOR_SIZE)); |
264 | assert(QEMU_IS_ALIGNED(bytes, BDRV_SECTOR_SIZE)); | |
5cd23081 KW |
265 | |
266 | qemu_co_mutex_lock(&s->lock); | |
3c56521b | 267 | |
5b47b7c3 | 268 | for (i = 0; i < nb_sectors; i++) { |
5cd23081 | 269 | void *data; |
5b47b7c3 DXW |
270 | uint32_t sector_offset_in_block = |
271 | ((sector_num + i) % s->sectors_per_block), | |
272 | block_num = (sector_num + i) / s->sectors_per_block; | |
273 | if (cloop_read_block(bs, block_num) != 0) { | |
5cd23081 KW |
274 | ret = -EIO; |
275 | goto fail; | |
5b47b7c3 | 276 | } |
5cd23081 KW |
277 | |
278 | data = s->uncompressed_block + sector_offset_in_block * 512; | |
279 | qemu_iovec_from_buf(qiov, i * 512, data, 512); | |
3c56521b | 280 | } |
3c56521b | 281 | |
5cd23081 KW |
282 | ret = 0; |
283 | fail: | |
2914caa0 | 284 | qemu_co_mutex_unlock(&s->lock); |
5cd23081 | 285 | |
2914caa0 PB |
286 | return ret; |
287 | } | |
288 | ||
3c56521b FB |
289 | static void cloop_close(BlockDriverState *bs) |
290 | { | |
291 | BDRVCloopState *s = bs->opaque; | |
42d43d35 | 292 | g_free(s->offsets); |
756f51e4 DXW |
293 | g_free(s->compressed_block); |
294 | g_free(s->uncompressed_block); | |
3c56521b FB |
295 | inflateEnd(&s->zstream); |
296 | } | |
297 | ||
5efa9d5a | 298 | static BlockDriver bdrv_cloop = { |
5b47b7c3 DXW |
299 | .format_name = "cloop", |
300 | .instance_size = sizeof(BDRVCloopState), | |
301 | .bdrv_probe = cloop_probe, | |
302 | .bdrv_open = cloop_open, | |
69dca43d | 303 | .bdrv_child_perm = bdrv_default_perms, |
a6506481 | 304 | .bdrv_refresh_limits = cloop_refresh_limits, |
5cd23081 | 305 | .bdrv_co_preadv = cloop_co_preadv, |
5b47b7c3 | 306 | .bdrv_close = cloop_close, |
d67066d8 | 307 | .is_format = true, |
3c56521b | 308 | }; |
5efa9d5a AL |
309 | |
310 | static void bdrv_cloop_init(void) | |
311 | { | |
312 | bdrv_register(&bdrv_cloop); | |
313 | } | |
314 | ||
315 | block_init(bdrv_cloop_init); |