xref: /qemu/block/vpc.c (revision caf71f86)
1019d6b8fSAnthony Liguori /*
2cc2040f8SStefan Weil  * Block driver for Connectix / Microsoft Virtual PC images
3019d6b8fSAnthony Liguori  *
4019d6b8fSAnthony Liguori  * Copyright (c) 2005 Alex Beregszaszi
5019d6b8fSAnthony Liguori  * Copyright (c) 2009 Kevin Wolf <kwolf@suse.de>
6019d6b8fSAnthony Liguori  *
7019d6b8fSAnthony Liguori  * Permission is hereby granted, free of charge, to any person obtaining a copy
8019d6b8fSAnthony Liguori  * of this software and associated documentation files (the "Software"), to deal
9019d6b8fSAnthony Liguori  * in the Software without restriction, including without limitation the rights
10019d6b8fSAnthony Liguori  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11019d6b8fSAnthony Liguori  * copies of the Software, and to permit persons to whom the Software is
12019d6b8fSAnthony Liguori  * furnished to do so, subject to the following conditions:
13019d6b8fSAnthony Liguori  *
14019d6b8fSAnthony Liguori  * The above copyright notice and this permission notice shall be included in
15019d6b8fSAnthony Liguori  * all copies or substantial portions of the Software.
16019d6b8fSAnthony Liguori  *
17019d6b8fSAnthony Liguori  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18019d6b8fSAnthony Liguori  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19019d6b8fSAnthony Liguori  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
20019d6b8fSAnthony Liguori  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21019d6b8fSAnthony Liguori  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22019d6b8fSAnthony Liguori  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
23019d6b8fSAnthony Liguori  * THE SOFTWARE.
24019d6b8fSAnthony Liguori  */
25019d6b8fSAnthony Liguori #include "qemu-common.h"
26737e150eSPaolo Bonzini #include "block/block_int.h"
27019d6b8fSAnthony Liguori #include "module.h"
28*caf71f86SPaolo Bonzini #include "migration/migration.h"
291fe1fa51SCharles Arnold #if defined(CONFIG_UUID)
301fe1fa51SCharles Arnold #include <uuid/uuid.h>
311fe1fa51SCharles Arnold #endif
32019d6b8fSAnthony Liguori 
33019d6b8fSAnthony Liguori /**************************************************************/
34019d6b8fSAnthony Liguori 
35019d6b8fSAnthony Liguori #define HEADER_SIZE 512
36019d6b8fSAnthony Liguori 
37019d6b8fSAnthony Liguori //#define CACHE
38019d6b8fSAnthony Liguori 
39019d6b8fSAnthony Liguori enum vhd_type {
40019d6b8fSAnthony Liguori     VHD_FIXED           = 2,
41019d6b8fSAnthony Liguori     VHD_DYNAMIC         = 3,
42019d6b8fSAnthony Liguori     VHD_DIFFERENCING    = 4,
43019d6b8fSAnthony Liguori };
44019d6b8fSAnthony Liguori 
45019d6b8fSAnthony Liguori // Seconds since Jan 1, 2000 0:00:00 (UTC)
46019d6b8fSAnthony Liguori #define VHD_TIMESTAMP_BASE 946684800
47019d6b8fSAnthony Liguori 
48019d6b8fSAnthony Liguori // always big-endian
49019d6b8fSAnthony Liguori struct vhd_footer {
50019d6b8fSAnthony Liguori     char        creator[8]; // "conectix"
51019d6b8fSAnthony Liguori     uint32_t    features;
52019d6b8fSAnthony Liguori     uint32_t    version;
53019d6b8fSAnthony Liguori 
54019d6b8fSAnthony Liguori     // Offset of next header structure, 0xFFFFFFFF if none
55019d6b8fSAnthony Liguori     uint64_t    data_offset;
56019d6b8fSAnthony Liguori 
57019d6b8fSAnthony Liguori     // Seconds since Jan 1, 2000 0:00:00 (UTC)
58019d6b8fSAnthony Liguori     uint32_t    timestamp;
59019d6b8fSAnthony Liguori 
60019d6b8fSAnthony Liguori     char        creator_app[4]; // "vpc "
61019d6b8fSAnthony Liguori     uint16_t    major;
62019d6b8fSAnthony Liguori     uint16_t    minor;
63019d6b8fSAnthony Liguori     char        creator_os[4]; // "Wi2k"
64019d6b8fSAnthony Liguori 
65019d6b8fSAnthony Liguori     uint64_t    orig_size;
66019d6b8fSAnthony Liguori     uint64_t    size;
67019d6b8fSAnthony Liguori 
68019d6b8fSAnthony Liguori     uint16_t    cyls;
69019d6b8fSAnthony Liguori     uint8_t     heads;
70019d6b8fSAnthony Liguori     uint8_t     secs_per_cyl;
71019d6b8fSAnthony Liguori 
72019d6b8fSAnthony Liguori     uint32_t    type;
73019d6b8fSAnthony Liguori 
74019d6b8fSAnthony Liguori     // Checksum of the Hard Disk Footer ("one's complement of the sum of all
75019d6b8fSAnthony Liguori     // the bytes in the footer without the checksum field")
76019d6b8fSAnthony Liguori     uint32_t    checksum;
77019d6b8fSAnthony Liguori 
78019d6b8fSAnthony Liguori     // UUID used to identify a parent hard disk (backing file)
79019d6b8fSAnthony Liguori     uint8_t     uuid[16];
80019d6b8fSAnthony Liguori 
81019d6b8fSAnthony Liguori     uint8_t     in_saved_state;
82019d6b8fSAnthony Liguori };
83019d6b8fSAnthony Liguori 
84019d6b8fSAnthony Liguori struct vhd_dyndisk_header {
85019d6b8fSAnthony Liguori     char        magic[8]; // "cxsparse"
86019d6b8fSAnthony Liguori 
87019d6b8fSAnthony Liguori     // Offset of next header structure, 0xFFFFFFFF if none
88019d6b8fSAnthony Liguori     uint64_t    data_offset;
89019d6b8fSAnthony Liguori 
90019d6b8fSAnthony Liguori     // Offset of the Block Allocation Table (BAT)
91019d6b8fSAnthony Liguori     uint64_t    table_offset;
92019d6b8fSAnthony Liguori 
93019d6b8fSAnthony Liguori     uint32_t    version;
94019d6b8fSAnthony Liguori     uint32_t    max_table_entries; // 32bit/entry
95019d6b8fSAnthony Liguori 
96019d6b8fSAnthony Liguori     // 2 MB by default, must be a power of two
97019d6b8fSAnthony Liguori     uint32_t    block_size;
98019d6b8fSAnthony Liguori 
99019d6b8fSAnthony Liguori     uint32_t    checksum;
100019d6b8fSAnthony Liguori     uint8_t     parent_uuid[16];
101019d6b8fSAnthony Liguori     uint32_t    parent_timestamp;
102019d6b8fSAnthony Liguori     uint32_t    reserved;
103019d6b8fSAnthony Liguori 
104019d6b8fSAnthony Liguori     // Backing file name (in UTF-16)
105019d6b8fSAnthony Liguori     uint8_t     parent_name[512];
106019d6b8fSAnthony Liguori 
107019d6b8fSAnthony Liguori     struct {
108019d6b8fSAnthony Liguori         uint32_t    platform;
109019d6b8fSAnthony Liguori         uint32_t    data_space;
110019d6b8fSAnthony Liguori         uint32_t    data_length;
111019d6b8fSAnthony Liguori         uint32_t    reserved;
112019d6b8fSAnthony Liguori         uint64_t    data_offset;
113019d6b8fSAnthony Liguori     } parent_locator[8];
114019d6b8fSAnthony Liguori };
115019d6b8fSAnthony Liguori 
116019d6b8fSAnthony Liguori typedef struct BDRVVPCState {
117848c66e8SPaolo Bonzini     CoMutex lock;
118019d6b8fSAnthony Liguori     uint8_t footer_buf[HEADER_SIZE];
119019d6b8fSAnthony Liguori     uint64_t free_data_block_offset;
120019d6b8fSAnthony Liguori     int max_table_entries;
121019d6b8fSAnthony Liguori     uint32_t *pagetable;
122019d6b8fSAnthony Liguori     uint64_t bat_offset;
123019d6b8fSAnthony Liguori     uint64_t last_bitmap_offset;
124019d6b8fSAnthony Liguori 
125019d6b8fSAnthony Liguori     uint32_t block_size;
126019d6b8fSAnthony Liguori     uint32_t bitmap_size;
127019d6b8fSAnthony Liguori 
128019d6b8fSAnthony Liguori #ifdef CACHE
129019d6b8fSAnthony Liguori     uint8_t *pageentry_u8;
130019d6b8fSAnthony Liguori     uint32_t *pageentry_u32;
131019d6b8fSAnthony Liguori     uint16_t *pageentry_u16;
132019d6b8fSAnthony Liguori 
133019d6b8fSAnthony Liguori     uint64_t last_bitmap;
134019d6b8fSAnthony Liguori #endif
135612ff3d8SKevin Wolf 
136612ff3d8SKevin Wolf     Error *migration_blocker;
137019d6b8fSAnthony Liguori } BDRVVPCState;
138019d6b8fSAnthony Liguori 
139019d6b8fSAnthony Liguori static uint32_t vpc_checksum(uint8_t* buf, size_t size)
140019d6b8fSAnthony Liguori {
141019d6b8fSAnthony Liguori     uint32_t res = 0;
142019d6b8fSAnthony Liguori     int i;
143019d6b8fSAnthony Liguori 
144019d6b8fSAnthony Liguori     for (i = 0; i < size; i++)
145019d6b8fSAnthony Liguori         res += buf[i];
146019d6b8fSAnthony Liguori 
147019d6b8fSAnthony Liguori     return ~res;
148019d6b8fSAnthony Liguori }
149019d6b8fSAnthony Liguori 
150019d6b8fSAnthony Liguori 
151019d6b8fSAnthony Liguori static int vpc_probe(const uint8_t *buf, int buf_size, const char *filename)
152019d6b8fSAnthony Liguori {
153019d6b8fSAnthony Liguori     if (buf_size >= 8 && !strncmp((char *)buf, "conectix", 8))
154019d6b8fSAnthony Liguori 	return 100;
155019d6b8fSAnthony Liguori     return 0;
156019d6b8fSAnthony Liguori }
157019d6b8fSAnthony Liguori 
15866f82ceeSKevin Wolf static int vpc_open(BlockDriverState *bs, int flags)
159019d6b8fSAnthony Liguori {
160019d6b8fSAnthony Liguori     BDRVVPCState *s = bs->opaque;
16166f82ceeSKevin Wolf     int i;
162019d6b8fSAnthony Liguori     struct vhd_footer* footer;
163019d6b8fSAnthony Liguori     struct vhd_dyndisk_header* dyndisk_header;
164019d6b8fSAnthony Liguori     uint8_t buf[HEADER_SIZE];
165019d6b8fSAnthony Liguori     uint32_t checksum;
166efc8243dSSerge E. Hallyn     int err = -1;
16724da78dbSCharles Arnold     int disk_type = VHD_DYNAMIC;
168019d6b8fSAnthony Liguori 
16966f82ceeSKevin Wolf     if (bdrv_pread(bs->file, 0, s->footer_buf, HEADER_SIZE) != HEADER_SIZE)
170019d6b8fSAnthony Liguori         goto fail;
171019d6b8fSAnthony Liguori 
172019d6b8fSAnthony Liguori     footer = (struct vhd_footer*) s->footer_buf;
17324da78dbSCharles Arnold     if (strncmp(footer->creator, "conectix", 8)) {
17424da78dbSCharles Arnold         int64_t offset = bdrv_getlength(bs->file);
17524da78dbSCharles Arnold         if (offset < HEADER_SIZE) {
176019d6b8fSAnthony Liguori             goto fail;
17724da78dbSCharles Arnold         }
17824da78dbSCharles Arnold         /* If a fixed disk, the footer is found only at the end of the file */
17924da78dbSCharles Arnold         if (bdrv_pread(bs->file, offset-HEADER_SIZE, s->footer_buf, HEADER_SIZE)
18024da78dbSCharles Arnold                 != HEADER_SIZE) {
18124da78dbSCharles Arnold             goto fail;
18224da78dbSCharles Arnold         }
18324da78dbSCharles Arnold         if (strncmp(footer->creator, "conectix", 8)) {
18424da78dbSCharles Arnold             goto fail;
18524da78dbSCharles Arnold         }
18624da78dbSCharles Arnold         disk_type = VHD_FIXED;
18724da78dbSCharles Arnold     }
188019d6b8fSAnthony Liguori 
189019d6b8fSAnthony Liguori     checksum = be32_to_cpu(footer->checksum);
190019d6b8fSAnthony Liguori     footer->checksum = 0;
191019d6b8fSAnthony Liguori     if (vpc_checksum(s->footer_buf, HEADER_SIZE) != checksum)
192019d6b8fSAnthony Liguori         fprintf(stderr, "block-vpc: The header checksum of '%s' is "
19366f82ceeSKevin Wolf             "incorrect.\n", bs->filename);
194019d6b8fSAnthony Liguori 
195c088b691SZhang Shengju     /* Write 'checksum' back to footer, or else will leave it with zero. */
196c088b691SZhang Shengju     footer->checksum = be32_to_cpu(checksum);
197c088b691SZhang Shengju 
198019d6b8fSAnthony Liguori     // The visible size of a image in Virtual PC depends on the geometry
199019d6b8fSAnthony Liguori     // rather than on the size stored in the footer (the size in the footer
200019d6b8fSAnthony Liguori     // is too large usually)
201019d6b8fSAnthony Liguori     bs->total_sectors = (int64_t)
202019d6b8fSAnthony Liguori         be16_to_cpu(footer->cyls) * footer->heads * footer->secs_per_cyl;
203019d6b8fSAnthony Liguori 
204258d2edbSCharles Arnold     /* Allow a maximum disk size of approximately 2 TB */
205258d2edbSCharles Arnold     if (bs->total_sectors >= 65535LL * 255 * 255) {
206efc8243dSSerge E. Hallyn         err = -EFBIG;
207efc8243dSSerge E. Hallyn         goto fail;
208efc8243dSSerge E. Hallyn     }
209efc8243dSSerge E. Hallyn 
21024da78dbSCharles Arnold     if (disk_type == VHD_DYNAMIC) {
21124da78dbSCharles Arnold         if (bdrv_pread(bs->file, be64_to_cpu(footer->data_offset), buf,
21224da78dbSCharles Arnold                 HEADER_SIZE) != HEADER_SIZE) {
213019d6b8fSAnthony Liguori             goto fail;
21424da78dbSCharles Arnold         }
215019d6b8fSAnthony Liguori 
216019d6b8fSAnthony Liguori         dyndisk_header = (struct vhd_dyndisk_header *) buf;
217019d6b8fSAnthony Liguori 
21824da78dbSCharles Arnold         if (strncmp(dyndisk_header->magic, "cxsparse", 8)) {
219019d6b8fSAnthony Liguori             goto fail;
22024da78dbSCharles Arnold         }
221019d6b8fSAnthony Liguori 
222019d6b8fSAnthony Liguori         s->block_size = be32_to_cpu(dyndisk_header->block_size);
223019d6b8fSAnthony Liguori         s->bitmap_size = ((s->block_size / (8 * 512)) + 511) & ~511;
224019d6b8fSAnthony Liguori 
225019d6b8fSAnthony Liguori         s->max_table_entries = be32_to_cpu(dyndisk_header->max_table_entries);
2267267c094SAnthony Liguori         s->pagetable = g_malloc(s->max_table_entries * 4);
227019d6b8fSAnthony Liguori 
228019d6b8fSAnthony Liguori         s->bat_offset = be64_to_cpu(dyndisk_header->table_offset);
22966f82ceeSKevin Wolf         if (bdrv_pread(bs->file, s->bat_offset, s->pagetable,
23024da78dbSCharles Arnold                 s->max_table_entries * 4) != s->max_table_entries * 4) {
231019d6b8fSAnthony Liguori             goto fail;
23224da78dbSCharles Arnold         }
233019d6b8fSAnthony Liguori 
234019d6b8fSAnthony Liguori         s->free_data_block_offset =
235019d6b8fSAnthony Liguori             (s->bat_offset + (s->max_table_entries * 4) + 511) & ~511;
236019d6b8fSAnthony Liguori 
237019d6b8fSAnthony Liguori         for (i = 0; i < s->max_table_entries; i++) {
238019d6b8fSAnthony Liguori             be32_to_cpus(&s->pagetable[i]);
239019d6b8fSAnthony Liguori             if (s->pagetable[i] != 0xFFFFFFFF) {
240019d6b8fSAnthony Liguori                 int64_t next = (512 * (int64_t) s->pagetable[i]) +
241019d6b8fSAnthony Liguori                     s->bitmap_size + s->block_size;
242019d6b8fSAnthony Liguori 
24324da78dbSCharles Arnold                 if (next > s->free_data_block_offset) {
244019d6b8fSAnthony Liguori                     s->free_data_block_offset = next;
245019d6b8fSAnthony Liguori                 }
246019d6b8fSAnthony Liguori             }
24724da78dbSCharles Arnold         }
248019d6b8fSAnthony Liguori 
249019d6b8fSAnthony Liguori         s->last_bitmap_offset = (int64_t) -1;
250019d6b8fSAnthony Liguori 
251019d6b8fSAnthony Liguori #ifdef CACHE
2527267c094SAnthony Liguori         s->pageentry_u8 = g_malloc(512);
253019d6b8fSAnthony Liguori         s->pageentry_u32 = s->pageentry_u8;
254019d6b8fSAnthony Liguori         s->pageentry_u16 = s->pageentry_u8;
255019d6b8fSAnthony Liguori         s->last_pagetable = -1;
256019d6b8fSAnthony Liguori #endif
25724da78dbSCharles Arnold     }
258019d6b8fSAnthony Liguori 
259848c66e8SPaolo Bonzini     qemu_co_mutex_init(&s->lock);
260612ff3d8SKevin Wolf 
261612ff3d8SKevin Wolf     /* Disable migration when VHD images are used */
262612ff3d8SKevin Wolf     error_set(&s->migration_blocker,
263612ff3d8SKevin Wolf               QERR_BLOCK_FORMAT_FEATURE_NOT_SUPPORTED,
264612ff3d8SKevin Wolf               "vpc", bs->device_name, "live migration");
265612ff3d8SKevin Wolf     migrate_add_blocker(s->migration_blocker);
266612ff3d8SKevin Wolf 
267019d6b8fSAnthony Liguori     return 0;
268019d6b8fSAnthony Liguori  fail:
269efc8243dSSerge E. Hallyn     return err;
270019d6b8fSAnthony Liguori }
271019d6b8fSAnthony Liguori 
2723fe4b700SJeff Cody static int vpc_reopen_prepare(BDRVReopenState *state,
2733fe4b700SJeff Cody                               BlockReopenQueue *queue, Error **errp)
2743fe4b700SJeff Cody {
2753fe4b700SJeff Cody     return 0;
2763fe4b700SJeff Cody }
2773fe4b700SJeff Cody 
278019d6b8fSAnthony Liguori /*
279019d6b8fSAnthony Liguori  * Returns the absolute byte offset of the given sector in the image file.
280019d6b8fSAnthony Liguori  * If the sector is not allocated, -1 is returned instead.
281019d6b8fSAnthony Liguori  *
282019d6b8fSAnthony Liguori  * The parameter write must be 1 if the offset will be used for a write
283019d6b8fSAnthony Liguori  * operation (the block bitmaps is updated then), 0 otherwise.
284019d6b8fSAnthony Liguori  */
285019d6b8fSAnthony Liguori static inline int64_t get_sector_offset(BlockDriverState *bs,
286019d6b8fSAnthony Liguori     int64_t sector_num, int write)
287019d6b8fSAnthony Liguori {
288019d6b8fSAnthony Liguori     BDRVVPCState *s = bs->opaque;
289019d6b8fSAnthony Liguori     uint64_t offset = sector_num * 512;
290019d6b8fSAnthony Liguori     uint64_t bitmap_offset, block_offset;
291019d6b8fSAnthony Liguori     uint32_t pagetable_index, pageentry_index;
292019d6b8fSAnthony Liguori 
293019d6b8fSAnthony Liguori     pagetable_index = offset / s->block_size;
294019d6b8fSAnthony Liguori     pageentry_index = (offset % s->block_size) / 512;
295019d6b8fSAnthony Liguori 
296019d6b8fSAnthony Liguori     if (pagetable_index >= s->max_table_entries || s->pagetable[pagetable_index] == 0xffffffff)
297019d6b8fSAnthony Liguori         return -1; // not allocated
298019d6b8fSAnthony Liguori 
299019d6b8fSAnthony Liguori     bitmap_offset = 512 * (uint64_t) s->pagetable[pagetable_index];
300019d6b8fSAnthony Liguori     block_offset = bitmap_offset + s->bitmap_size + (512 * pageentry_index);
301019d6b8fSAnthony Liguori 
302019d6b8fSAnthony Liguori     // We must ensure that we don't write to any sectors which are marked as
303019d6b8fSAnthony Liguori     // unused in the bitmap. We get away with setting all bits in the block
304019d6b8fSAnthony Liguori     // bitmap each time we write to a new block. This might cause Virtual PC to
305019d6b8fSAnthony Liguori     // miss sparse read optimization, but it's not a problem in terms of
306019d6b8fSAnthony Liguori     // correctness.
307019d6b8fSAnthony Liguori     if (write && (s->last_bitmap_offset != bitmap_offset)) {
308019d6b8fSAnthony Liguori         uint8_t bitmap[s->bitmap_size];
309019d6b8fSAnthony Liguori 
310019d6b8fSAnthony Liguori         s->last_bitmap_offset = bitmap_offset;
311019d6b8fSAnthony Liguori         memset(bitmap, 0xff, s->bitmap_size);
312078a458eSKevin Wolf         bdrv_pwrite_sync(bs->file, bitmap_offset, bitmap, s->bitmap_size);
313019d6b8fSAnthony Liguori     }
314019d6b8fSAnthony Liguori 
315019d6b8fSAnthony Liguori //    printf("sector: %" PRIx64 ", index: %x, offset: %x, bioff: %" PRIx64 ", bloff: %" PRIx64 "\n",
316019d6b8fSAnthony Liguori //	sector_num, pagetable_index, pageentry_index,
317019d6b8fSAnthony Liguori //	bitmap_offset, block_offset);
318019d6b8fSAnthony Liguori 
319019d6b8fSAnthony Liguori // disabled by reason
320019d6b8fSAnthony Liguori #if 0
321019d6b8fSAnthony Liguori #ifdef CACHE
322019d6b8fSAnthony Liguori     if (bitmap_offset != s->last_bitmap)
323019d6b8fSAnthony Liguori     {
324019d6b8fSAnthony Liguori 	lseek(s->fd, bitmap_offset, SEEK_SET);
325019d6b8fSAnthony Liguori 
326019d6b8fSAnthony Liguori 	s->last_bitmap = bitmap_offset;
327019d6b8fSAnthony Liguori 
328019d6b8fSAnthony Liguori 	// Scary! Bitmap is stored as big endian 32bit entries,
329019d6b8fSAnthony Liguori 	// while we used to look it up byte by byte
330019d6b8fSAnthony Liguori 	read(s->fd, s->pageentry_u8, 512);
331019d6b8fSAnthony Liguori 	for (i = 0; i < 128; i++)
332019d6b8fSAnthony Liguori 	    be32_to_cpus(&s->pageentry_u32[i]);
333019d6b8fSAnthony Liguori     }
334019d6b8fSAnthony Liguori 
335019d6b8fSAnthony Liguori     if ((s->pageentry_u8[pageentry_index / 8] >> (pageentry_index % 8)) & 1)
336019d6b8fSAnthony Liguori 	return -1;
337019d6b8fSAnthony Liguori #else
338019d6b8fSAnthony Liguori     lseek(s->fd, bitmap_offset + (pageentry_index / 8), SEEK_SET);
339019d6b8fSAnthony Liguori 
340019d6b8fSAnthony Liguori     read(s->fd, &bitmap_entry, 1);
341019d6b8fSAnthony Liguori 
342019d6b8fSAnthony Liguori     if ((bitmap_entry >> (pageentry_index % 8)) & 1)
343019d6b8fSAnthony Liguori 	return -1; // not allocated
344019d6b8fSAnthony Liguori #endif
345019d6b8fSAnthony Liguori #endif
346019d6b8fSAnthony Liguori 
347019d6b8fSAnthony Liguori     return block_offset;
348019d6b8fSAnthony Liguori }
349019d6b8fSAnthony Liguori 
350019d6b8fSAnthony Liguori /*
351019d6b8fSAnthony Liguori  * Writes the footer to the end of the image file. This is needed when the
352019d6b8fSAnthony Liguori  * file grows as it overwrites the old footer
353019d6b8fSAnthony Liguori  *
354019d6b8fSAnthony Liguori  * Returns 0 on success and < 0 on error
355019d6b8fSAnthony Liguori  */
356019d6b8fSAnthony Liguori static int rewrite_footer(BlockDriverState* bs)
357019d6b8fSAnthony Liguori {
358019d6b8fSAnthony Liguori     int ret;
359019d6b8fSAnthony Liguori     BDRVVPCState *s = bs->opaque;
360019d6b8fSAnthony Liguori     int64_t offset = s->free_data_block_offset;
361019d6b8fSAnthony Liguori 
362078a458eSKevin Wolf     ret = bdrv_pwrite_sync(bs->file, offset, s->footer_buf, HEADER_SIZE);
363019d6b8fSAnthony Liguori     if (ret < 0)
364019d6b8fSAnthony Liguori         return ret;
365019d6b8fSAnthony Liguori 
366019d6b8fSAnthony Liguori     return 0;
367019d6b8fSAnthony Liguori }
368019d6b8fSAnthony Liguori 
369019d6b8fSAnthony Liguori /*
370019d6b8fSAnthony Liguori  * Allocates a new block. This involves writing a new footer and updating
371019d6b8fSAnthony Liguori  * the Block Allocation Table to use the space at the old end of the image
372019d6b8fSAnthony Liguori  * file (overwriting the old footer)
373019d6b8fSAnthony Liguori  *
374019d6b8fSAnthony Liguori  * Returns the sectors' offset in the image file on success and < 0 on error
375019d6b8fSAnthony Liguori  */
376019d6b8fSAnthony Liguori static int64_t alloc_block(BlockDriverState* bs, int64_t sector_num)
377019d6b8fSAnthony Liguori {
378019d6b8fSAnthony Liguori     BDRVVPCState *s = bs->opaque;
379019d6b8fSAnthony Liguori     int64_t bat_offset;
380019d6b8fSAnthony Liguori     uint32_t index, bat_value;
381019d6b8fSAnthony Liguori     int ret;
382019d6b8fSAnthony Liguori     uint8_t bitmap[s->bitmap_size];
383019d6b8fSAnthony Liguori 
384019d6b8fSAnthony Liguori     // Check if sector_num is valid
385019d6b8fSAnthony Liguori     if ((sector_num < 0) || (sector_num > bs->total_sectors))
386019d6b8fSAnthony Liguori         return -1;
387019d6b8fSAnthony Liguori 
388019d6b8fSAnthony Liguori     // Write entry into in-memory BAT
389019d6b8fSAnthony Liguori     index = (sector_num * 512) / s->block_size;
390019d6b8fSAnthony Liguori     if (s->pagetable[index] != 0xFFFFFFFF)
391019d6b8fSAnthony Liguori         return -1;
392019d6b8fSAnthony Liguori 
393019d6b8fSAnthony Liguori     s->pagetable[index] = s->free_data_block_offset / 512;
394019d6b8fSAnthony Liguori 
395019d6b8fSAnthony Liguori     // Initialize the block's bitmap
396019d6b8fSAnthony Liguori     memset(bitmap, 0xff, s->bitmap_size);
3975bb1cbacSKevin Wolf     ret = bdrv_pwrite_sync(bs->file, s->free_data_block_offset, bitmap,
398078a458eSKevin Wolf         s->bitmap_size);
3995bb1cbacSKevin Wolf     if (ret < 0) {
4005bb1cbacSKevin Wolf         return ret;
4015bb1cbacSKevin Wolf     }
402019d6b8fSAnthony Liguori 
403019d6b8fSAnthony Liguori     // Write new footer (the old one will be overwritten)
404019d6b8fSAnthony Liguori     s->free_data_block_offset += s->block_size + s->bitmap_size;
405019d6b8fSAnthony Liguori     ret = rewrite_footer(bs);
406019d6b8fSAnthony Liguori     if (ret < 0)
407019d6b8fSAnthony Liguori         goto fail;
408019d6b8fSAnthony Liguori 
409019d6b8fSAnthony Liguori     // Write BAT entry to disk
410019d6b8fSAnthony Liguori     bat_offset = s->bat_offset + (4 * index);
411019d6b8fSAnthony Liguori     bat_value = be32_to_cpu(s->pagetable[index]);
412078a458eSKevin Wolf     ret = bdrv_pwrite_sync(bs->file, bat_offset, &bat_value, 4);
413019d6b8fSAnthony Liguori     if (ret < 0)
414019d6b8fSAnthony Liguori         goto fail;
415019d6b8fSAnthony Liguori 
416019d6b8fSAnthony Liguori     return get_sector_offset(bs, sector_num, 0);
417019d6b8fSAnthony Liguori 
418019d6b8fSAnthony Liguori fail:
419019d6b8fSAnthony Liguori     s->free_data_block_offset -= (s->block_size + s->bitmap_size);
420019d6b8fSAnthony Liguori     return -1;
421019d6b8fSAnthony Liguori }
422019d6b8fSAnthony Liguori 
423019d6b8fSAnthony Liguori static int vpc_read(BlockDriverState *bs, int64_t sector_num,
424019d6b8fSAnthony Liguori                     uint8_t *buf, int nb_sectors)
425019d6b8fSAnthony Liguori {
4266c6ea921SKevin Wolf     BDRVVPCState *s = bs->opaque;
427019d6b8fSAnthony Liguori     int ret;
428019d6b8fSAnthony Liguori     int64_t offset;
4296c6ea921SKevin Wolf     int64_t sectors, sectors_per_block;
43024da78dbSCharles Arnold     struct vhd_footer *footer = (struct vhd_footer *) s->footer_buf;
431019d6b8fSAnthony Liguori 
43224da78dbSCharles Arnold     if (cpu_to_be32(footer->type) == VHD_FIXED) {
43324da78dbSCharles Arnold         return bdrv_read(bs->file, sector_num, buf, nb_sectors);
43424da78dbSCharles Arnold     }
435019d6b8fSAnthony Liguori     while (nb_sectors > 0) {
436019d6b8fSAnthony Liguori         offset = get_sector_offset(bs, sector_num, 0);
437019d6b8fSAnthony Liguori 
4386c6ea921SKevin Wolf         sectors_per_block = s->block_size >> BDRV_SECTOR_BITS;
4396c6ea921SKevin Wolf         sectors = sectors_per_block - (sector_num % sectors_per_block);
4406c6ea921SKevin Wolf         if (sectors > nb_sectors) {
4416c6ea921SKevin Wolf             sectors = nb_sectors;
442019d6b8fSAnthony Liguori         }
443019d6b8fSAnthony Liguori 
4446c6ea921SKevin Wolf         if (offset == -1) {
4456c6ea921SKevin Wolf             memset(buf, 0, sectors * BDRV_SECTOR_SIZE);
4466c6ea921SKevin Wolf         } else {
4476c6ea921SKevin Wolf             ret = bdrv_pread(bs->file, offset, buf,
4486c6ea921SKevin Wolf                 sectors * BDRV_SECTOR_SIZE);
4496c6ea921SKevin Wolf             if (ret != sectors * BDRV_SECTOR_SIZE) {
4506c6ea921SKevin Wolf                 return -1;
4516c6ea921SKevin Wolf             }
4526c6ea921SKevin Wolf         }
4536c6ea921SKevin Wolf 
4546c6ea921SKevin Wolf         nb_sectors -= sectors;
4556c6ea921SKevin Wolf         sector_num += sectors;
4566c6ea921SKevin Wolf         buf += sectors * BDRV_SECTOR_SIZE;
457019d6b8fSAnthony Liguori     }
458019d6b8fSAnthony Liguori     return 0;
459019d6b8fSAnthony Liguori }
460019d6b8fSAnthony Liguori 
4612914caa0SPaolo Bonzini static coroutine_fn int vpc_co_read(BlockDriverState *bs, int64_t sector_num,
4622914caa0SPaolo Bonzini                                     uint8_t *buf, int nb_sectors)
4632914caa0SPaolo Bonzini {
4642914caa0SPaolo Bonzini     int ret;
4652914caa0SPaolo Bonzini     BDRVVPCState *s = bs->opaque;
4662914caa0SPaolo Bonzini     qemu_co_mutex_lock(&s->lock);
4672914caa0SPaolo Bonzini     ret = vpc_read(bs, sector_num, buf, nb_sectors);
4682914caa0SPaolo Bonzini     qemu_co_mutex_unlock(&s->lock);
4692914caa0SPaolo Bonzini     return ret;
4702914caa0SPaolo Bonzini }
4712914caa0SPaolo Bonzini 
472019d6b8fSAnthony Liguori static int vpc_write(BlockDriverState *bs, int64_t sector_num,
473019d6b8fSAnthony Liguori     const uint8_t *buf, int nb_sectors)
474019d6b8fSAnthony Liguori {
4756c6ea921SKevin Wolf     BDRVVPCState *s = bs->opaque;
476019d6b8fSAnthony Liguori     int64_t offset;
4776c6ea921SKevin Wolf     int64_t sectors, sectors_per_block;
478019d6b8fSAnthony Liguori     int ret;
47924da78dbSCharles Arnold     struct vhd_footer *footer =  (struct vhd_footer *) s->footer_buf;
480019d6b8fSAnthony Liguori 
48124da78dbSCharles Arnold     if (cpu_to_be32(footer->type) == VHD_FIXED) {
48224da78dbSCharles Arnold         return bdrv_write(bs->file, sector_num, buf, nb_sectors);
48324da78dbSCharles Arnold     }
484019d6b8fSAnthony Liguori     while (nb_sectors > 0) {
485019d6b8fSAnthony Liguori         offset = get_sector_offset(bs, sector_num, 1);
486019d6b8fSAnthony Liguori 
4876c6ea921SKevin Wolf         sectors_per_block = s->block_size >> BDRV_SECTOR_BITS;
4886c6ea921SKevin Wolf         sectors = sectors_per_block - (sector_num % sectors_per_block);
4896c6ea921SKevin Wolf         if (sectors > nb_sectors) {
4906c6ea921SKevin Wolf             sectors = nb_sectors;
4916c6ea921SKevin Wolf         }
4926c6ea921SKevin Wolf 
493019d6b8fSAnthony Liguori         if (offset == -1) {
494019d6b8fSAnthony Liguori             offset = alloc_block(bs, sector_num);
495019d6b8fSAnthony Liguori             if (offset < 0)
496019d6b8fSAnthony Liguori                 return -1;
497019d6b8fSAnthony Liguori         }
498019d6b8fSAnthony Liguori 
4996c6ea921SKevin Wolf         ret = bdrv_pwrite(bs->file, offset, buf, sectors * BDRV_SECTOR_SIZE);
5006c6ea921SKevin Wolf         if (ret != sectors * BDRV_SECTOR_SIZE) {
501019d6b8fSAnthony Liguori             return -1;
5026c6ea921SKevin Wolf         }
503019d6b8fSAnthony Liguori 
5046c6ea921SKevin Wolf         nb_sectors -= sectors;
5056c6ea921SKevin Wolf         sector_num += sectors;
5066c6ea921SKevin Wolf         buf += sectors * BDRV_SECTOR_SIZE;
507019d6b8fSAnthony Liguori     }
508019d6b8fSAnthony Liguori 
509019d6b8fSAnthony Liguori     return 0;
510019d6b8fSAnthony Liguori }
511019d6b8fSAnthony Liguori 
512e183ef75SPaolo Bonzini static coroutine_fn int vpc_co_write(BlockDriverState *bs, int64_t sector_num,
513e183ef75SPaolo Bonzini                                      const uint8_t *buf, int nb_sectors)
514e183ef75SPaolo Bonzini {
515e183ef75SPaolo Bonzini     int ret;
516e183ef75SPaolo Bonzini     BDRVVPCState *s = bs->opaque;
517e183ef75SPaolo Bonzini     qemu_co_mutex_lock(&s->lock);
518e183ef75SPaolo Bonzini     ret = vpc_write(bs, sector_num, buf, nb_sectors);
519e183ef75SPaolo Bonzini     qemu_co_mutex_unlock(&s->lock);
520e183ef75SPaolo Bonzini     return ret;
521e183ef75SPaolo Bonzini }
522e183ef75SPaolo Bonzini 
523019d6b8fSAnthony Liguori /*
524019d6b8fSAnthony Liguori  * Calculates the number of cylinders, heads and sectors per cylinder
525019d6b8fSAnthony Liguori  * based on a given number of sectors. This is the algorithm described
526019d6b8fSAnthony Liguori  * in the VHD specification.
527019d6b8fSAnthony Liguori  *
528019d6b8fSAnthony Liguori  * Note that the geometry doesn't always exactly match total_sectors but
529019d6b8fSAnthony Liguori  * may round it down.
530019d6b8fSAnthony Liguori  *
531258d2edbSCharles Arnold  * Returns 0 on success, -EFBIG if the size is larger than ~2 TB. Override
532258d2edbSCharles Arnold  * the hardware EIDE and ATA-2 limit of 16 heads (max disk size of 127 GB)
533258d2edbSCharles Arnold  * and instead allow up to 255 heads.
534019d6b8fSAnthony Liguori  */
535019d6b8fSAnthony Liguori static int calculate_geometry(int64_t total_sectors, uint16_t* cyls,
536019d6b8fSAnthony Liguori     uint8_t* heads, uint8_t* secs_per_cyl)
537019d6b8fSAnthony Liguori {
538019d6b8fSAnthony Liguori     uint32_t cyls_times_heads;
539019d6b8fSAnthony Liguori 
540258d2edbSCharles Arnold     /* Allow a maximum disk size of approximately 2 TB */
541258d2edbSCharles Arnold     if (total_sectors > 65535LL * 255 * 255) {
542019d6b8fSAnthony Liguori         return -EFBIG;
543258d2edbSCharles Arnold     }
544019d6b8fSAnthony Liguori 
545019d6b8fSAnthony Liguori     if (total_sectors > 65535 * 16 * 63) {
546019d6b8fSAnthony Liguori         *secs_per_cyl = 255;
547258d2edbSCharles Arnold         if (total_sectors > 65535 * 16 * 255) {
548258d2edbSCharles Arnold             *heads = 255;
549258d2edbSCharles Arnold         } else {
550019d6b8fSAnthony Liguori             *heads = 16;
551258d2edbSCharles Arnold         }
552019d6b8fSAnthony Liguori         cyls_times_heads = total_sectors / *secs_per_cyl;
553019d6b8fSAnthony Liguori     } else {
554019d6b8fSAnthony Liguori         *secs_per_cyl = 17;
555019d6b8fSAnthony Liguori         cyls_times_heads = total_sectors / *secs_per_cyl;
556019d6b8fSAnthony Liguori         *heads = (cyls_times_heads + 1023) / 1024;
557019d6b8fSAnthony Liguori 
558019d6b8fSAnthony Liguori         if (*heads < 4)
559019d6b8fSAnthony Liguori             *heads = 4;
560019d6b8fSAnthony Liguori 
561019d6b8fSAnthony Liguori         if (cyls_times_heads >= (*heads * 1024) || *heads > 16) {
562019d6b8fSAnthony Liguori             *secs_per_cyl = 31;
563019d6b8fSAnthony Liguori             *heads = 16;
564019d6b8fSAnthony Liguori             cyls_times_heads = total_sectors / *secs_per_cyl;
565019d6b8fSAnthony Liguori         }
566019d6b8fSAnthony Liguori 
567019d6b8fSAnthony Liguori         if (cyls_times_heads >= (*heads * 1024)) {
568019d6b8fSAnthony Liguori             *secs_per_cyl = 63;
569019d6b8fSAnthony Liguori             *heads = 16;
570019d6b8fSAnthony Liguori             cyls_times_heads = total_sectors / *secs_per_cyl;
571019d6b8fSAnthony Liguori         }
572019d6b8fSAnthony Liguori     }
573019d6b8fSAnthony Liguori 
574dede4188SStefan Weil     *cyls = cyls_times_heads / *heads;
575019d6b8fSAnthony Liguori 
576019d6b8fSAnthony Liguori     return 0;
577019d6b8fSAnthony Liguori }
578019d6b8fSAnthony Liguori 
57924da78dbSCharles Arnold static int create_dynamic_disk(int fd, uint8_t *buf, int64_t total_sectors)
580019d6b8fSAnthony Liguori {
581019d6b8fSAnthony Liguori     struct vhd_dyndisk_header* dyndisk_header =
582019d6b8fSAnthony Liguori         (struct vhd_dyndisk_header*) buf;
583019d6b8fSAnthony Liguori     size_t block_size, num_bat_entries;
58424da78dbSCharles Arnold     int i;
585f0ff243aSBlue Swirl     int ret = -EIO;
586019d6b8fSAnthony Liguori 
587019d6b8fSAnthony Liguori     // Write the footer (twice: at the beginning and at the end)
588019d6b8fSAnthony Liguori     block_size = 0x200000;
589019d6b8fSAnthony Liguori     num_bat_entries = (total_sectors + block_size / 512) / (block_size / 512);
590019d6b8fSAnthony Liguori 
591f0ff243aSBlue Swirl     if (write(fd, buf, HEADER_SIZE) != HEADER_SIZE) {
592f0ff243aSBlue Swirl         goto fail;
593f0ff243aSBlue Swirl     }
594019d6b8fSAnthony Liguori 
595f0ff243aSBlue Swirl     if (lseek(fd, 1536 + ((num_bat_entries * 4 + 511) & ~511), SEEK_SET) < 0) {
596f0ff243aSBlue Swirl         goto fail;
597f0ff243aSBlue Swirl     }
598f0ff243aSBlue Swirl     if (write(fd, buf, HEADER_SIZE) != HEADER_SIZE) {
599f0ff243aSBlue Swirl         goto fail;
600f0ff243aSBlue Swirl     }
601019d6b8fSAnthony Liguori 
602019d6b8fSAnthony Liguori     // Write the initial BAT
603f0ff243aSBlue Swirl     if (lseek(fd, 3 * 512, SEEK_SET) < 0) {
604f0ff243aSBlue Swirl         goto fail;
605f0ff243aSBlue Swirl     }
606019d6b8fSAnthony Liguori 
607019d6b8fSAnthony Liguori     memset(buf, 0xFF, 512);
608f0ff243aSBlue Swirl     for (i = 0; i < (num_bat_entries * 4 + 511) / 512; i++) {
609f0ff243aSBlue Swirl         if (write(fd, buf, 512) != 512) {
610f0ff243aSBlue Swirl             goto fail;
611f0ff243aSBlue Swirl         }
612f0ff243aSBlue Swirl     }
613019d6b8fSAnthony Liguori 
614019d6b8fSAnthony Liguori     // Prepare the Dynamic Disk Header
615019d6b8fSAnthony Liguori     memset(buf, 0, 1024);
616019d6b8fSAnthony Liguori 
6175ec4d682SNathan Froyd     memcpy(dyndisk_header->magic, "cxsparse", 8);
618019d6b8fSAnthony Liguori 
61978439f6aSCharles Arnold     /*
62078439f6aSCharles Arnold      * Note: The spec is actually wrong here for data_offset, it says
62178439f6aSCharles Arnold      * 0xFFFFFFFF, but MS tools expect all 64 bits to be set.
62278439f6aSCharles Arnold      */
62378439f6aSCharles Arnold     dyndisk_header->data_offset = be64_to_cpu(0xFFFFFFFFFFFFFFFFULL);
624019d6b8fSAnthony Liguori     dyndisk_header->table_offset = be64_to_cpu(3 * 512);
625019d6b8fSAnthony Liguori     dyndisk_header->version = be32_to_cpu(0x00010000);
626019d6b8fSAnthony Liguori     dyndisk_header->block_size = be32_to_cpu(block_size);
627019d6b8fSAnthony Liguori     dyndisk_header->max_table_entries = be32_to_cpu(num_bat_entries);
628019d6b8fSAnthony Liguori 
629019d6b8fSAnthony Liguori     dyndisk_header->checksum = be32_to_cpu(vpc_checksum(buf, 1024));
630019d6b8fSAnthony Liguori 
631019d6b8fSAnthony Liguori     // Write the header
632f0ff243aSBlue Swirl     if (lseek(fd, 512, SEEK_SET) < 0) {
633f0ff243aSBlue Swirl         goto fail;
634f0ff243aSBlue Swirl     }
635019d6b8fSAnthony Liguori 
636f0ff243aSBlue Swirl     if (write(fd, buf, 1024) != 1024) {
637f0ff243aSBlue Swirl         goto fail;
638f0ff243aSBlue Swirl     }
639f0ff243aSBlue Swirl     ret = 0;
640f0ff243aSBlue Swirl 
641f0ff243aSBlue Swirl  fail:
64224da78dbSCharles Arnold     return ret;
64324da78dbSCharles Arnold }
64424da78dbSCharles Arnold 
64524da78dbSCharles Arnold static int create_fixed_disk(int fd, uint8_t *buf, int64_t total_size)
64624da78dbSCharles Arnold {
64724da78dbSCharles Arnold     int ret = -EIO;
64824da78dbSCharles Arnold 
64924da78dbSCharles Arnold     /* Add footer to total size */
65024da78dbSCharles Arnold     total_size += 512;
65124da78dbSCharles Arnold     if (ftruncate(fd, total_size) != 0) {
65224da78dbSCharles Arnold         ret = -errno;
65324da78dbSCharles Arnold         goto fail;
65424da78dbSCharles Arnold     }
65524da78dbSCharles Arnold     if (lseek(fd, -512, SEEK_END) < 0) {
65624da78dbSCharles Arnold         goto fail;
65724da78dbSCharles Arnold     }
65824da78dbSCharles Arnold     if (write(fd, buf, HEADER_SIZE) != HEADER_SIZE) {
65924da78dbSCharles Arnold         goto fail;
66024da78dbSCharles Arnold     }
66124da78dbSCharles Arnold 
66224da78dbSCharles Arnold     ret = 0;
66324da78dbSCharles Arnold 
66424da78dbSCharles Arnold  fail:
66524da78dbSCharles Arnold     return ret;
66624da78dbSCharles Arnold }
66724da78dbSCharles Arnold 
66824da78dbSCharles Arnold static int vpc_create(const char *filename, QEMUOptionParameter *options)
66924da78dbSCharles Arnold {
67024da78dbSCharles Arnold     uint8_t buf[1024];
67124da78dbSCharles Arnold     struct vhd_footer *footer = (struct vhd_footer *) buf;
67224da78dbSCharles Arnold     QEMUOptionParameter *disk_type_param;
67324da78dbSCharles Arnold     int fd, i;
67424da78dbSCharles Arnold     uint16_t cyls = 0;
67524da78dbSCharles Arnold     uint8_t heads = 0;
67624da78dbSCharles Arnold     uint8_t secs_per_cyl = 0;
67724da78dbSCharles Arnold     int64_t total_sectors;
67824da78dbSCharles Arnold     int64_t total_size;
67924da78dbSCharles Arnold     int disk_type;
68024da78dbSCharles Arnold     int ret = -EIO;
68124da78dbSCharles Arnold 
68224da78dbSCharles Arnold     /* Read out options */
68324da78dbSCharles Arnold     total_size = get_option_parameter(options, BLOCK_OPT_SIZE)->value.n;
68424da78dbSCharles Arnold 
68524da78dbSCharles Arnold     disk_type_param = get_option_parameter(options, BLOCK_OPT_SUBFMT);
68624da78dbSCharles Arnold     if (disk_type_param && disk_type_param->value.s) {
68724da78dbSCharles Arnold         if (!strcmp(disk_type_param->value.s, "dynamic")) {
68824da78dbSCharles Arnold             disk_type = VHD_DYNAMIC;
68924da78dbSCharles Arnold         } else if (!strcmp(disk_type_param->value.s, "fixed")) {
69024da78dbSCharles Arnold             disk_type = VHD_FIXED;
69124da78dbSCharles Arnold         } else {
69224da78dbSCharles Arnold             return -EINVAL;
69324da78dbSCharles Arnold         }
69424da78dbSCharles Arnold     } else {
69524da78dbSCharles Arnold         disk_type = VHD_DYNAMIC;
69624da78dbSCharles Arnold     }
69724da78dbSCharles Arnold 
69824da78dbSCharles Arnold     /* Create the file */
6996165f4d8SCorey Bryant     fd = qemu_open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0644);
70024da78dbSCharles Arnold     if (fd < 0) {
70124da78dbSCharles Arnold         return -EIO;
70224da78dbSCharles Arnold     }
70324da78dbSCharles Arnold 
704ecd880d9SKevin Wolf     /*
705ecd880d9SKevin Wolf      * Calculate matching total_size and geometry. Increase the number of
706ecd880d9SKevin Wolf      * sectors requested until we get enough (or fail). This ensures that
707ecd880d9SKevin Wolf      * qemu-img convert doesn't truncate images, but rather rounds up.
708ecd880d9SKevin Wolf      */
70924da78dbSCharles Arnold     total_sectors = total_size / BDRV_SECTOR_SIZE;
710ecd880d9SKevin Wolf     for (i = 0; total_sectors > (int64_t)cyls * heads * secs_per_cyl; i++) {
711ecd880d9SKevin Wolf         if (calculate_geometry(total_sectors + i, &cyls, &heads,
712ecd880d9SKevin Wolf                                &secs_per_cyl))
713ecd880d9SKevin Wolf         {
71424da78dbSCharles Arnold             ret = -EFBIG;
71524da78dbSCharles Arnold             goto fail;
71624da78dbSCharles Arnold         }
71724da78dbSCharles Arnold     }
718ecd880d9SKevin Wolf 
71924da78dbSCharles Arnold     total_sectors = (int64_t) cyls * heads * secs_per_cyl;
72024da78dbSCharles Arnold 
72124da78dbSCharles Arnold     /* Prepare the Hard Disk Footer */
72224da78dbSCharles Arnold     memset(buf, 0, 1024);
72324da78dbSCharles Arnold 
72424da78dbSCharles Arnold     memcpy(footer->creator, "conectix", 8);
72524da78dbSCharles Arnold     /* TODO Check if "qemu" creator_app is ok for VPC */
72624da78dbSCharles Arnold     memcpy(footer->creator_app, "qemu", 4);
72724da78dbSCharles Arnold     memcpy(footer->creator_os, "Wi2k", 4);
72824da78dbSCharles Arnold 
72924da78dbSCharles Arnold     footer->features = be32_to_cpu(0x02);
73024da78dbSCharles Arnold     footer->version = be32_to_cpu(0x00010000);
73124da78dbSCharles Arnold     if (disk_type == VHD_DYNAMIC) {
73224da78dbSCharles Arnold         footer->data_offset = be64_to_cpu(HEADER_SIZE);
73324da78dbSCharles Arnold     } else {
73424da78dbSCharles Arnold         footer->data_offset = be64_to_cpu(0xFFFFFFFFFFFFFFFFULL);
73524da78dbSCharles Arnold     }
73624da78dbSCharles Arnold     footer->timestamp = be32_to_cpu(time(NULL) - VHD_TIMESTAMP_BASE);
73724da78dbSCharles Arnold 
73824da78dbSCharles Arnold     /* Version of Virtual PC 2007 */
73924da78dbSCharles Arnold     footer->major = be16_to_cpu(0x0005);
74024da78dbSCharles Arnold     footer->minor = be16_to_cpu(0x0003);
74124da78dbSCharles Arnold     if (disk_type == VHD_DYNAMIC) {
74224da78dbSCharles Arnold         footer->orig_size = be64_to_cpu(total_sectors * 512);
74324da78dbSCharles Arnold         footer->size = be64_to_cpu(total_sectors * 512);
74424da78dbSCharles Arnold     } else {
74524da78dbSCharles Arnold         footer->orig_size = be64_to_cpu(total_size);
74624da78dbSCharles Arnold         footer->size = be64_to_cpu(total_size);
74724da78dbSCharles Arnold     }
74824da78dbSCharles Arnold     footer->cyls = be16_to_cpu(cyls);
74924da78dbSCharles Arnold     footer->heads = heads;
75024da78dbSCharles Arnold     footer->secs_per_cyl = secs_per_cyl;
75124da78dbSCharles Arnold 
75224da78dbSCharles Arnold     footer->type = be32_to_cpu(disk_type);
75324da78dbSCharles Arnold 
7541fe1fa51SCharles Arnold #if defined(CONFIG_UUID)
7551fe1fa51SCharles Arnold     uuid_generate(footer->uuid);
7561fe1fa51SCharles Arnold #endif
75724da78dbSCharles Arnold 
75824da78dbSCharles Arnold     footer->checksum = be32_to_cpu(vpc_checksum(buf, HEADER_SIZE));
75924da78dbSCharles Arnold 
76024da78dbSCharles Arnold     if (disk_type == VHD_DYNAMIC) {
76124da78dbSCharles Arnold         ret = create_dynamic_disk(fd, buf, total_sectors);
76224da78dbSCharles Arnold     } else {
76324da78dbSCharles Arnold         ret = create_fixed_disk(fd, buf, total_size);
76424da78dbSCharles Arnold     }
76524da78dbSCharles Arnold 
76624da78dbSCharles Arnold  fail:
7672e1e79daSCorey Bryant     qemu_close(fd);
768f0ff243aSBlue Swirl     return ret;
769019d6b8fSAnthony Liguori }
770019d6b8fSAnthony Liguori 
771019d6b8fSAnthony Liguori static void vpc_close(BlockDriverState *bs)
772019d6b8fSAnthony Liguori {
773019d6b8fSAnthony Liguori     BDRVVPCState *s = bs->opaque;
7747267c094SAnthony Liguori     g_free(s->pagetable);
775019d6b8fSAnthony Liguori #ifdef CACHE
7767267c094SAnthony Liguori     g_free(s->pageentry_u8);
777019d6b8fSAnthony Liguori #endif
778612ff3d8SKevin Wolf 
779612ff3d8SKevin Wolf     migrate_del_blocker(s->migration_blocker);
780612ff3d8SKevin Wolf     error_free(s->migration_blocker);
781019d6b8fSAnthony Liguori }
782019d6b8fSAnthony Liguori 
7830e7e1989SKevin Wolf static QEMUOptionParameter vpc_create_options[] = {
784db08adf5SKevin Wolf     {
785db08adf5SKevin Wolf         .name = BLOCK_OPT_SIZE,
786db08adf5SKevin Wolf         .type = OPT_SIZE,
787db08adf5SKevin Wolf         .help = "Virtual disk size"
788db08adf5SKevin Wolf     },
78924da78dbSCharles Arnold     {
79024da78dbSCharles Arnold         .name = BLOCK_OPT_SUBFMT,
79124da78dbSCharles Arnold         .type = OPT_STRING,
79224da78dbSCharles Arnold         .help =
79324da78dbSCharles Arnold             "Type of virtual hard disk format. Supported formats are "
79424da78dbSCharles Arnold             "{dynamic (default) | fixed} "
79524da78dbSCharles Arnold     },
7960e7e1989SKevin Wolf     { NULL }
7970e7e1989SKevin Wolf };
7980e7e1989SKevin Wolf 
799019d6b8fSAnthony Liguori static BlockDriver bdrv_vpc = {
800019d6b8fSAnthony Liguori     .format_name    = "vpc",
801019d6b8fSAnthony Liguori     .instance_size  = sizeof(BDRVVPCState),
802c68b89acSKevin Wolf 
803019d6b8fSAnthony Liguori     .bdrv_probe     = vpc_probe,
804019d6b8fSAnthony Liguori     .bdrv_open      = vpc_open,
805019d6b8fSAnthony Liguori     .bdrv_close     = vpc_close,
8063fe4b700SJeff Cody     .bdrv_reopen_prepare = vpc_reopen_prepare,
807019d6b8fSAnthony Liguori     .bdrv_create    = vpc_create,
8080e7e1989SKevin Wolf 
809c68b89acSKevin Wolf     .bdrv_read              = vpc_co_read,
810c68b89acSKevin Wolf     .bdrv_write             = vpc_co_write,
811c68b89acSKevin Wolf 
8120e7e1989SKevin Wolf     .create_options = vpc_create_options,
813019d6b8fSAnthony Liguori };
814019d6b8fSAnthony Liguori 
815019d6b8fSAnthony Liguori static void bdrv_vpc_init(void)
816019d6b8fSAnthony Liguori {
817019d6b8fSAnthony Liguori     bdrv_register(&bdrv_vpc);
818019d6b8fSAnthony Liguori }
819019d6b8fSAnthony Liguori 
820019d6b8fSAnthony Liguori block_init(bdrv_vpc_init);
821