]>
git.proxmox.com Git - mirror_qemu.git/blob - block/dmg.c
2 * QEMU Block driver for DMG images
4 * Copyright (c) 2004 Johannes E. Schindelin
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:
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
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
24 #include "qemu-common.h"
25 #include "block/block_int.h"
26 #include "qemu/bswap.h"
27 #include "qemu/module.h"
31 /* Limit chunk sizes to prevent unreasonable amounts of memory being used
32 * or truncating when converting to 32-bit types
34 DMG_LENGTHS_MAX
= 64 * 1024 * 1024, /* 64 MB */
35 DMG_SECTORCOUNTS_MAX
= DMG_LENGTHS_MAX
/ 512,
38 typedef struct BDRVDMGState
{
40 /* each chunk contains a certain number of sectors,
41 * offsets[i] is the offset in the .dmg file,
42 * lengths[i] is the length of the compressed chunk,
43 * sectors[i] is the sector beginning at offsets[i],
44 * sectorcounts[i] is the number of sectors in that chunk,
45 * the sectors array is ordered
53 uint64_t* sectorcounts
;
54 uint32_t current_chunk
;
55 uint8_t *compressed_chunk
;
56 uint8_t *uncompressed_chunk
;
60 static int dmg_probe(const uint8_t *buf
, int buf_size
, const char *filename
)
68 len
= strlen(filename
);
69 if (len
> 4 && !strcmp(filename
+ len
- 4, ".dmg")) {
75 static int read_uint64(BlockDriverState
*bs
, int64_t offset
, uint64_t *result
)
80 ret
= bdrv_pread(bs
->file
, offset
, &buffer
, 8);
85 *result
= be64_to_cpu(buffer
);
89 static int read_uint32(BlockDriverState
*bs
, int64_t offset
, uint32_t *result
)
94 ret
= bdrv_pread(bs
->file
, offset
, &buffer
, 4);
99 *result
= be32_to_cpu(buffer
);
103 static int dmg_open(BlockDriverState
*bs
, QDict
*options
, int flags
,
106 BDRVDMGState
*s
= bs
->opaque
;
107 uint64_t info_begin
, info_end
, last_in_offset
, last_out_offset
;
109 uint32_t max_compressed_size
= 1, max_sectors_per_chunk
= 1, i
;
115 s
->offsets
= s
->lengths
= s
->sectors
= s
->sectorcounts
= NULL
;
117 /* read offset of info blocks */
118 offset
= bdrv_getlength(bs
->file
);
125 ret
= read_uint64(bs
, offset
, &info_begin
);
128 } else if (info_begin
== 0) {
133 ret
= read_uint32(bs
, info_begin
, &tmp
);
136 } else if (tmp
!= 0x100) {
141 ret
= read_uint32(bs
, info_begin
+ 4, &count
);
144 } else if (count
== 0) {
148 info_end
= info_begin
+ count
;
150 offset
= info_begin
+ 0x100;
153 last_in_offset
= last_out_offset
= 0;
154 while (offset
< info_end
) {
157 ret
= read_uint32(bs
, offset
, &count
);
160 } else if (count
== 0) {
166 ret
= read_uint32(bs
, offset
, &type
);
171 if (type
== 0x6d697368 && count
>= 244) {
173 uint32_t chunk_count
;
178 chunk_count
= (count
- 204) / 40;
179 new_size
= sizeof(uint64_t) * (s
->n_chunks
+ chunk_count
);
180 s
->types
= g_realloc(s
->types
, new_size
/ 2);
181 s
->offsets
= g_realloc(s
->offsets
, new_size
);
182 s
->lengths
= g_realloc(s
->lengths
, new_size
);
183 s
->sectors
= g_realloc(s
->sectors
, new_size
);
184 s
->sectorcounts
= g_realloc(s
->sectorcounts
, new_size
);
186 for (i
= s
->n_chunks
; i
< s
->n_chunks
+ chunk_count
; i
++) {
187 ret
= read_uint32(bs
, offset
, &s
->types
[i
]);
192 if (s
->types
[i
] != 0x80000005 && s
->types
[i
] != 1 &&
194 if (s
->types
[i
] == 0xffffffff && i
> 0) {
195 last_in_offset
= s
->offsets
[i
- 1] + s
->lengths
[i
- 1];
196 last_out_offset
= s
->sectors
[i
- 1] +
197 s
->sectorcounts
[i
- 1];
206 ret
= read_uint64(bs
, offset
, &s
->sectors
[i
]);
210 s
->sectors
[i
] += last_out_offset
;
213 ret
= read_uint64(bs
, offset
, &s
->sectorcounts
[i
]);
219 if (s
->sectorcounts
[i
] > DMG_SECTORCOUNTS_MAX
) {
220 error_report("sector count %" PRIu64
" for chunk %u is "
221 "larger than max (%u)",
222 s
->sectorcounts
[i
], i
, DMG_SECTORCOUNTS_MAX
);
227 ret
= read_uint64(bs
, offset
, &s
->offsets
[i
]);
231 s
->offsets
[i
] += last_in_offset
;
234 ret
= read_uint64(bs
, offset
, &s
->lengths
[i
]);
240 if (s
->lengths
[i
] > DMG_LENGTHS_MAX
) {
241 error_report("length %" PRIu64
" for chunk %u is larger "
243 s
->lengths
[i
], i
, DMG_LENGTHS_MAX
);
248 if (s
->lengths
[i
] > max_compressed_size
) {
249 max_compressed_size
= s
->lengths
[i
];
251 if (s
->sectorcounts
[i
] > max_sectors_per_chunk
) {
252 max_sectors_per_chunk
= s
->sectorcounts
[i
];
255 s
->n_chunks
+= chunk_count
;
259 /* initialize zlib engine */
260 s
->compressed_chunk
= g_malloc(max_compressed_size
+ 1);
261 s
->uncompressed_chunk
= g_malloc(512 * max_sectors_per_chunk
);
262 if (inflateInit(&s
->zstream
) != Z_OK
) {
267 s
->current_chunk
= s
->n_chunks
;
269 qemu_co_mutex_init(&s
->lock
);
277 g_free(s
->sectorcounts
);
278 g_free(s
->compressed_chunk
);
279 g_free(s
->uncompressed_chunk
);
283 static inline int is_sector_in_chunk(BDRVDMGState
* s
,
284 uint32_t chunk_num
, int sector_num
)
286 if (chunk_num
>= s
->n_chunks
|| s
->sectors
[chunk_num
] > sector_num
||
287 s
->sectors
[chunk_num
] + s
->sectorcounts
[chunk_num
] <= sector_num
) {
294 static inline uint32_t search_chunk(BDRVDMGState
*s
, int sector_num
)
297 uint32_t chunk1
= 0, chunk2
= s
->n_chunks
, chunk3
;
298 while (chunk1
!= chunk2
) {
299 chunk3
= (chunk1
+ chunk2
) / 2;
300 if (s
->sectors
[chunk3
] > sector_num
) {
302 } else if (s
->sectors
[chunk3
] + s
->sectorcounts
[chunk3
] > sector_num
) {
308 return s
->n_chunks
; /* error */
311 static inline int dmg_read_chunk(BlockDriverState
*bs
, int sector_num
)
313 BDRVDMGState
*s
= bs
->opaque
;
315 if (!is_sector_in_chunk(s
, s
->current_chunk
, sector_num
)) {
317 uint32_t chunk
= search_chunk(s
, sector_num
);
319 if (chunk
>= s
->n_chunks
) {
323 s
->current_chunk
= s
->n_chunks
;
324 switch (s
->types
[chunk
]) {
325 case 0x80000005: { /* zlib compressed */
326 /* we need to buffer, because only the chunk as whole can be
328 ret
= bdrv_pread(bs
->file
, s
->offsets
[chunk
],
329 s
->compressed_chunk
, s
->lengths
[chunk
]);
330 if (ret
!= s
->lengths
[chunk
]) {
334 s
->zstream
.next_in
= s
->compressed_chunk
;
335 s
->zstream
.avail_in
= s
->lengths
[chunk
];
336 s
->zstream
.next_out
= s
->uncompressed_chunk
;
337 s
->zstream
.avail_out
= 512 * s
->sectorcounts
[chunk
];
338 ret
= inflateReset(&s
->zstream
);
342 ret
= inflate(&s
->zstream
, Z_FINISH
);
343 if (ret
!= Z_STREAM_END
||
344 s
->zstream
.total_out
!= 512 * s
->sectorcounts
[chunk
]) {
349 ret
= bdrv_pread(bs
->file
, s
->offsets
[chunk
],
350 s
->uncompressed_chunk
, s
->lengths
[chunk
]);
351 if (ret
!= s
->lengths
[chunk
]) {
356 memset(s
->uncompressed_chunk
, 0, 512 * s
->sectorcounts
[chunk
]);
359 s
->current_chunk
= chunk
;
364 static int dmg_read(BlockDriverState
*bs
, int64_t sector_num
,
365 uint8_t *buf
, int nb_sectors
)
367 BDRVDMGState
*s
= bs
->opaque
;
370 for (i
= 0; i
< nb_sectors
; i
++) {
371 uint32_t sector_offset_in_chunk
;
372 if (dmg_read_chunk(bs
, sector_num
+ i
) != 0) {
375 sector_offset_in_chunk
= sector_num
+ i
- s
->sectors
[s
->current_chunk
];
376 memcpy(buf
+ i
* 512,
377 s
->uncompressed_chunk
+ sector_offset_in_chunk
* 512, 512);
382 static coroutine_fn
int dmg_co_read(BlockDriverState
*bs
, int64_t sector_num
,
383 uint8_t *buf
, int nb_sectors
)
386 BDRVDMGState
*s
= bs
->opaque
;
387 qemu_co_mutex_lock(&s
->lock
);
388 ret
= dmg_read(bs
, sector_num
, buf
, nb_sectors
);
389 qemu_co_mutex_unlock(&s
->lock
);
393 static void dmg_close(BlockDriverState
*bs
)
395 BDRVDMGState
*s
= bs
->opaque
;
401 g_free(s
->sectorcounts
);
402 g_free(s
->compressed_chunk
);
403 g_free(s
->uncompressed_chunk
);
405 inflateEnd(&s
->zstream
);
408 static BlockDriver bdrv_dmg
= {
409 .format_name
= "dmg",
410 .instance_size
= sizeof(BDRVDMGState
),
411 .bdrv_probe
= dmg_probe
,
412 .bdrv_open
= dmg_open
,
413 .bdrv_read
= dmg_co_read
,
414 .bdrv_close
= dmg_close
,
417 static void bdrv_dmg_init(void)
419 bdrv_register(&bdrv_dmg
);
422 block_init(bdrv_dmg_init
);