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" 26019d6b8fSAnthony Liguori #include "block_int.h" 27019d6b8fSAnthony Liguori #include "module.h" 28019d6b8fSAnthony Liguori 29019d6b8fSAnthony Liguori /**************************************************************/ 30019d6b8fSAnthony Liguori 31019d6b8fSAnthony Liguori #define HEADER_SIZE 512 32019d6b8fSAnthony Liguori 33019d6b8fSAnthony Liguori //#define CACHE 34019d6b8fSAnthony Liguori 35019d6b8fSAnthony Liguori enum vhd_type { 36019d6b8fSAnthony Liguori VHD_FIXED = 2, 37019d6b8fSAnthony Liguori VHD_DYNAMIC = 3, 38019d6b8fSAnthony Liguori VHD_DIFFERENCING = 4, 39019d6b8fSAnthony Liguori }; 40019d6b8fSAnthony Liguori 41019d6b8fSAnthony Liguori // Seconds since Jan 1, 2000 0:00:00 (UTC) 42019d6b8fSAnthony Liguori #define VHD_TIMESTAMP_BASE 946684800 43019d6b8fSAnthony Liguori 44019d6b8fSAnthony Liguori // always big-endian 45019d6b8fSAnthony Liguori struct vhd_footer { 46019d6b8fSAnthony Liguori char creator[8]; // "conectix" 47019d6b8fSAnthony Liguori uint32_t features; 48019d6b8fSAnthony Liguori uint32_t version; 49019d6b8fSAnthony Liguori 50019d6b8fSAnthony Liguori // Offset of next header structure, 0xFFFFFFFF if none 51019d6b8fSAnthony Liguori uint64_t data_offset; 52019d6b8fSAnthony Liguori 53019d6b8fSAnthony Liguori // Seconds since Jan 1, 2000 0:00:00 (UTC) 54019d6b8fSAnthony Liguori uint32_t timestamp; 55019d6b8fSAnthony Liguori 56019d6b8fSAnthony Liguori char creator_app[4]; // "vpc " 57019d6b8fSAnthony Liguori uint16_t major; 58019d6b8fSAnthony Liguori uint16_t minor; 59019d6b8fSAnthony Liguori char creator_os[4]; // "Wi2k" 60019d6b8fSAnthony Liguori 61019d6b8fSAnthony Liguori uint64_t orig_size; 62019d6b8fSAnthony Liguori uint64_t size; 63019d6b8fSAnthony Liguori 64019d6b8fSAnthony Liguori uint16_t cyls; 65019d6b8fSAnthony Liguori uint8_t heads; 66019d6b8fSAnthony Liguori uint8_t secs_per_cyl; 67019d6b8fSAnthony Liguori 68019d6b8fSAnthony Liguori uint32_t type; 69019d6b8fSAnthony Liguori 70019d6b8fSAnthony Liguori // Checksum of the Hard Disk Footer ("one's complement of the sum of all 71019d6b8fSAnthony Liguori // the bytes in the footer without the checksum field") 72019d6b8fSAnthony Liguori uint32_t checksum; 73019d6b8fSAnthony Liguori 74019d6b8fSAnthony Liguori // UUID used to identify a parent hard disk (backing file) 75019d6b8fSAnthony Liguori uint8_t uuid[16]; 76019d6b8fSAnthony Liguori 77019d6b8fSAnthony Liguori uint8_t in_saved_state; 78019d6b8fSAnthony Liguori }; 79019d6b8fSAnthony Liguori 80019d6b8fSAnthony Liguori struct vhd_dyndisk_header { 81019d6b8fSAnthony Liguori char magic[8]; // "cxsparse" 82019d6b8fSAnthony Liguori 83019d6b8fSAnthony Liguori // Offset of next header structure, 0xFFFFFFFF if none 84019d6b8fSAnthony Liguori uint64_t data_offset; 85019d6b8fSAnthony Liguori 86019d6b8fSAnthony Liguori // Offset of the Block Allocation Table (BAT) 87019d6b8fSAnthony Liguori uint64_t table_offset; 88019d6b8fSAnthony Liguori 89019d6b8fSAnthony Liguori uint32_t version; 90019d6b8fSAnthony Liguori uint32_t max_table_entries; // 32bit/entry 91019d6b8fSAnthony Liguori 92019d6b8fSAnthony Liguori // 2 MB by default, must be a power of two 93019d6b8fSAnthony Liguori uint32_t block_size; 94019d6b8fSAnthony Liguori 95019d6b8fSAnthony Liguori uint32_t checksum; 96019d6b8fSAnthony Liguori uint8_t parent_uuid[16]; 97019d6b8fSAnthony Liguori uint32_t parent_timestamp; 98019d6b8fSAnthony Liguori uint32_t reserved; 99019d6b8fSAnthony Liguori 100019d6b8fSAnthony Liguori // Backing file name (in UTF-16) 101019d6b8fSAnthony Liguori uint8_t parent_name[512]; 102019d6b8fSAnthony Liguori 103019d6b8fSAnthony Liguori struct { 104019d6b8fSAnthony Liguori uint32_t platform; 105019d6b8fSAnthony Liguori uint32_t data_space; 106019d6b8fSAnthony Liguori uint32_t data_length; 107019d6b8fSAnthony Liguori uint32_t reserved; 108019d6b8fSAnthony Liguori uint64_t data_offset; 109019d6b8fSAnthony Liguori } parent_locator[8]; 110019d6b8fSAnthony Liguori }; 111019d6b8fSAnthony Liguori 112019d6b8fSAnthony Liguori typedef struct BDRVVPCState { 113019d6b8fSAnthony Liguori BlockDriverState *hd; 114019d6b8fSAnthony Liguori 115019d6b8fSAnthony Liguori uint8_t footer_buf[HEADER_SIZE]; 116019d6b8fSAnthony Liguori uint64_t free_data_block_offset; 117019d6b8fSAnthony Liguori int max_table_entries; 118019d6b8fSAnthony Liguori uint32_t *pagetable; 119019d6b8fSAnthony Liguori uint64_t bat_offset; 120019d6b8fSAnthony Liguori uint64_t last_bitmap_offset; 121019d6b8fSAnthony Liguori 122019d6b8fSAnthony Liguori uint32_t block_size; 123019d6b8fSAnthony Liguori uint32_t bitmap_size; 124019d6b8fSAnthony Liguori 125019d6b8fSAnthony Liguori #ifdef CACHE 126019d6b8fSAnthony Liguori uint8_t *pageentry_u8; 127019d6b8fSAnthony Liguori uint32_t *pageentry_u32; 128019d6b8fSAnthony Liguori uint16_t *pageentry_u16; 129019d6b8fSAnthony Liguori 130019d6b8fSAnthony Liguori uint64_t last_bitmap; 131019d6b8fSAnthony Liguori #endif 132019d6b8fSAnthony Liguori } BDRVVPCState; 133019d6b8fSAnthony Liguori 134019d6b8fSAnthony Liguori static uint32_t vpc_checksum(uint8_t* buf, size_t size) 135019d6b8fSAnthony Liguori { 136019d6b8fSAnthony Liguori uint32_t res = 0; 137019d6b8fSAnthony Liguori int i; 138019d6b8fSAnthony Liguori 139019d6b8fSAnthony Liguori for (i = 0; i < size; i++) 140019d6b8fSAnthony Liguori res += buf[i]; 141019d6b8fSAnthony Liguori 142019d6b8fSAnthony Liguori return ~res; 143019d6b8fSAnthony Liguori } 144019d6b8fSAnthony Liguori 145019d6b8fSAnthony Liguori 146019d6b8fSAnthony Liguori static int vpc_probe(const uint8_t *buf, int buf_size, const char *filename) 147019d6b8fSAnthony Liguori { 148019d6b8fSAnthony Liguori if (buf_size >= 8 && !strncmp((char *)buf, "conectix", 8)) 149019d6b8fSAnthony Liguori return 100; 150019d6b8fSAnthony Liguori return 0; 151019d6b8fSAnthony Liguori } 152019d6b8fSAnthony Liguori 15366f82ceeSKevin Wolf static int vpc_open(BlockDriverState *bs, int flags) 154019d6b8fSAnthony Liguori { 155019d6b8fSAnthony Liguori BDRVVPCState *s = bs->opaque; 15666f82ceeSKevin Wolf int i; 157019d6b8fSAnthony Liguori struct vhd_footer* footer; 158019d6b8fSAnthony Liguori struct vhd_dyndisk_header* dyndisk_header; 159019d6b8fSAnthony Liguori uint8_t buf[HEADER_SIZE]; 160019d6b8fSAnthony Liguori uint32_t checksum; 161019d6b8fSAnthony Liguori 16266f82ceeSKevin Wolf if (bdrv_pread(bs->file, 0, s->footer_buf, HEADER_SIZE) != HEADER_SIZE) 163019d6b8fSAnthony Liguori goto fail; 164019d6b8fSAnthony Liguori 165019d6b8fSAnthony Liguori footer = (struct vhd_footer*) s->footer_buf; 166019d6b8fSAnthony Liguori if (strncmp(footer->creator, "conectix", 8)) 167019d6b8fSAnthony Liguori goto fail; 168019d6b8fSAnthony Liguori 169019d6b8fSAnthony Liguori checksum = be32_to_cpu(footer->checksum); 170019d6b8fSAnthony Liguori footer->checksum = 0; 171019d6b8fSAnthony Liguori if (vpc_checksum(s->footer_buf, HEADER_SIZE) != checksum) 172019d6b8fSAnthony Liguori fprintf(stderr, "block-vpc: The header checksum of '%s' is " 17366f82ceeSKevin Wolf "incorrect.\n", bs->filename); 174019d6b8fSAnthony Liguori 175019d6b8fSAnthony Liguori // The visible size of a image in Virtual PC depends on the geometry 176019d6b8fSAnthony Liguori // rather than on the size stored in the footer (the size in the footer 177019d6b8fSAnthony Liguori // is too large usually) 178019d6b8fSAnthony Liguori bs->total_sectors = (int64_t) 179019d6b8fSAnthony Liguori be16_to_cpu(footer->cyls) * footer->heads * footer->secs_per_cyl; 180019d6b8fSAnthony Liguori 18166f82ceeSKevin Wolf if (bdrv_pread(bs->file, be64_to_cpu(footer->data_offset), buf, HEADER_SIZE) 182019d6b8fSAnthony Liguori != HEADER_SIZE) 183019d6b8fSAnthony Liguori goto fail; 184019d6b8fSAnthony Liguori 185019d6b8fSAnthony Liguori dyndisk_header = (struct vhd_dyndisk_header*) buf; 186019d6b8fSAnthony Liguori 187019d6b8fSAnthony Liguori if (strncmp(dyndisk_header->magic, "cxsparse", 8)) 188019d6b8fSAnthony Liguori goto fail; 189019d6b8fSAnthony Liguori 190019d6b8fSAnthony Liguori 191019d6b8fSAnthony Liguori s->block_size = be32_to_cpu(dyndisk_header->block_size); 192019d6b8fSAnthony Liguori s->bitmap_size = ((s->block_size / (8 * 512)) + 511) & ~511; 193019d6b8fSAnthony Liguori 194019d6b8fSAnthony Liguori s->max_table_entries = be32_to_cpu(dyndisk_header->max_table_entries); 195019d6b8fSAnthony Liguori s->pagetable = qemu_malloc(s->max_table_entries * 4); 196019d6b8fSAnthony Liguori 197019d6b8fSAnthony Liguori s->bat_offset = be64_to_cpu(dyndisk_header->table_offset); 19866f82ceeSKevin Wolf if (bdrv_pread(bs->file, s->bat_offset, s->pagetable, 199019d6b8fSAnthony Liguori s->max_table_entries * 4) != s->max_table_entries * 4) 200019d6b8fSAnthony Liguori goto fail; 201019d6b8fSAnthony Liguori 202019d6b8fSAnthony Liguori s->free_data_block_offset = 203019d6b8fSAnthony Liguori (s->bat_offset + (s->max_table_entries * 4) + 511) & ~511; 204019d6b8fSAnthony Liguori 205019d6b8fSAnthony Liguori for (i = 0; i < s->max_table_entries; i++) { 206019d6b8fSAnthony Liguori be32_to_cpus(&s->pagetable[i]); 207019d6b8fSAnthony Liguori if (s->pagetable[i] != 0xFFFFFFFF) { 208019d6b8fSAnthony Liguori int64_t next = (512 * (int64_t) s->pagetable[i]) + 209019d6b8fSAnthony Liguori s->bitmap_size + s->block_size; 210019d6b8fSAnthony Liguori 211019d6b8fSAnthony Liguori if (next> s->free_data_block_offset) 212019d6b8fSAnthony Liguori s->free_data_block_offset = next; 213019d6b8fSAnthony Liguori } 214019d6b8fSAnthony Liguori } 215019d6b8fSAnthony Liguori 216019d6b8fSAnthony Liguori s->last_bitmap_offset = (int64_t) -1; 217019d6b8fSAnthony Liguori 218019d6b8fSAnthony Liguori #ifdef CACHE 219019d6b8fSAnthony Liguori s->pageentry_u8 = qemu_malloc(512); 220019d6b8fSAnthony Liguori s->pageentry_u32 = s->pageentry_u8; 221019d6b8fSAnthony Liguori s->pageentry_u16 = s->pageentry_u8; 222019d6b8fSAnthony Liguori s->last_pagetable = -1; 223019d6b8fSAnthony Liguori #endif 224019d6b8fSAnthony Liguori 225019d6b8fSAnthony Liguori return 0; 226019d6b8fSAnthony Liguori fail: 227019d6b8fSAnthony Liguori return -1; 228019d6b8fSAnthony Liguori } 229019d6b8fSAnthony Liguori 230019d6b8fSAnthony Liguori /* 231019d6b8fSAnthony Liguori * Returns the absolute byte offset of the given sector in the image file. 232019d6b8fSAnthony Liguori * If the sector is not allocated, -1 is returned instead. 233019d6b8fSAnthony Liguori * 234019d6b8fSAnthony Liguori * The parameter write must be 1 if the offset will be used for a write 235019d6b8fSAnthony Liguori * operation (the block bitmaps is updated then), 0 otherwise. 236019d6b8fSAnthony Liguori */ 237019d6b8fSAnthony Liguori static inline int64_t get_sector_offset(BlockDriverState *bs, 238019d6b8fSAnthony Liguori int64_t sector_num, int write) 239019d6b8fSAnthony Liguori { 240019d6b8fSAnthony Liguori BDRVVPCState *s = bs->opaque; 241019d6b8fSAnthony Liguori uint64_t offset = sector_num * 512; 242019d6b8fSAnthony Liguori uint64_t bitmap_offset, block_offset; 243019d6b8fSAnthony Liguori uint32_t pagetable_index, pageentry_index; 244019d6b8fSAnthony Liguori 245019d6b8fSAnthony Liguori pagetable_index = offset / s->block_size; 246019d6b8fSAnthony Liguori pageentry_index = (offset % s->block_size) / 512; 247019d6b8fSAnthony Liguori 248019d6b8fSAnthony Liguori if (pagetable_index >= s->max_table_entries || s->pagetable[pagetable_index] == 0xffffffff) 249019d6b8fSAnthony Liguori return -1; // not allocated 250019d6b8fSAnthony Liguori 251019d6b8fSAnthony Liguori bitmap_offset = 512 * (uint64_t) s->pagetable[pagetable_index]; 252019d6b8fSAnthony Liguori block_offset = bitmap_offset + s->bitmap_size + (512 * pageentry_index); 253019d6b8fSAnthony Liguori 254019d6b8fSAnthony Liguori // We must ensure that we don't write to any sectors which are marked as 255019d6b8fSAnthony Liguori // unused in the bitmap. We get away with setting all bits in the block 256019d6b8fSAnthony Liguori // bitmap each time we write to a new block. This might cause Virtual PC to 257019d6b8fSAnthony Liguori // miss sparse read optimization, but it's not a problem in terms of 258019d6b8fSAnthony Liguori // correctness. 259019d6b8fSAnthony Liguori if (write && (s->last_bitmap_offset != bitmap_offset)) { 260019d6b8fSAnthony Liguori uint8_t bitmap[s->bitmap_size]; 261019d6b8fSAnthony Liguori 262019d6b8fSAnthony Liguori s->last_bitmap_offset = bitmap_offset; 263019d6b8fSAnthony Liguori memset(bitmap, 0xff, s->bitmap_size); 264*078a458eSKevin Wolf bdrv_pwrite_sync(bs->file, bitmap_offset, bitmap, s->bitmap_size); 265019d6b8fSAnthony Liguori } 266019d6b8fSAnthony Liguori 267019d6b8fSAnthony Liguori // printf("sector: %" PRIx64 ", index: %x, offset: %x, bioff: %" PRIx64 ", bloff: %" PRIx64 "\n", 268019d6b8fSAnthony Liguori // sector_num, pagetable_index, pageentry_index, 269019d6b8fSAnthony Liguori // bitmap_offset, block_offset); 270019d6b8fSAnthony Liguori 271019d6b8fSAnthony Liguori // disabled by reason 272019d6b8fSAnthony Liguori #if 0 273019d6b8fSAnthony Liguori #ifdef CACHE 274019d6b8fSAnthony Liguori if (bitmap_offset != s->last_bitmap) 275019d6b8fSAnthony Liguori { 276019d6b8fSAnthony Liguori lseek(s->fd, bitmap_offset, SEEK_SET); 277019d6b8fSAnthony Liguori 278019d6b8fSAnthony Liguori s->last_bitmap = bitmap_offset; 279019d6b8fSAnthony Liguori 280019d6b8fSAnthony Liguori // Scary! Bitmap is stored as big endian 32bit entries, 281019d6b8fSAnthony Liguori // while we used to look it up byte by byte 282019d6b8fSAnthony Liguori read(s->fd, s->pageentry_u8, 512); 283019d6b8fSAnthony Liguori for (i = 0; i < 128; i++) 284019d6b8fSAnthony Liguori be32_to_cpus(&s->pageentry_u32[i]); 285019d6b8fSAnthony Liguori } 286019d6b8fSAnthony Liguori 287019d6b8fSAnthony Liguori if ((s->pageentry_u8[pageentry_index / 8] >> (pageentry_index % 8)) & 1) 288019d6b8fSAnthony Liguori return -1; 289019d6b8fSAnthony Liguori #else 290019d6b8fSAnthony Liguori lseek(s->fd, bitmap_offset + (pageentry_index / 8), SEEK_SET); 291019d6b8fSAnthony Liguori 292019d6b8fSAnthony Liguori read(s->fd, &bitmap_entry, 1); 293019d6b8fSAnthony Liguori 294019d6b8fSAnthony Liguori if ((bitmap_entry >> (pageentry_index % 8)) & 1) 295019d6b8fSAnthony Liguori return -1; // not allocated 296019d6b8fSAnthony Liguori #endif 297019d6b8fSAnthony Liguori #endif 298019d6b8fSAnthony Liguori 299019d6b8fSAnthony Liguori return block_offset; 300019d6b8fSAnthony Liguori } 301019d6b8fSAnthony Liguori 302019d6b8fSAnthony Liguori /* 303019d6b8fSAnthony Liguori * Writes the footer to the end of the image file. This is needed when the 304019d6b8fSAnthony Liguori * file grows as it overwrites the old footer 305019d6b8fSAnthony Liguori * 306019d6b8fSAnthony Liguori * Returns 0 on success and < 0 on error 307019d6b8fSAnthony Liguori */ 308019d6b8fSAnthony Liguori static int rewrite_footer(BlockDriverState* bs) 309019d6b8fSAnthony Liguori { 310019d6b8fSAnthony Liguori int ret; 311019d6b8fSAnthony Liguori BDRVVPCState *s = bs->opaque; 312019d6b8fSAnthony Liguori int64_t offset = s->free_data_block_offset; 313019d6b8fSAnthony Liguori 314*078a458eSKevin Wolf ret = bdrv_pwrite_sync(bs->file, offset, s->footer_buf, HEADER_SIZE); 315019d6b8fSAnthony Liguori if (ret < 0) 316019d6b8fSAnthony Liguori return ret; 317019d6b8fSAnthony Liguori 318019d6b8fSAnthony Liguori return 0; 319019d6b8fSAnthony Liguori } 320019d6b8fSAnthony Liguori 321019d6b8fSAnthony Liguori /* 322019d6b8fSAnthony Liguori * Allocates a new block. This involves writing a new footer and updating 323019d6b8fSAnthony Liguori * the Block Allocation Table to use the space at the old end of the image 324019d6b8fSAnthony Liguori * file (overwriting the old footer) 325019d6b8fSAnthony Liguori * 326019d6b8fSAnthony Liguori * Returns the sectors' offset in the image file on success and < 0 on error 327019d6b8fSAnthony Liguori */ 328019d6b8fSAnthony Liguori static int64_t alloc_block(BlockDriverState* bs, int64_t sector_num) 329019d6b8fSAnthony Liguori { 330019d6b8fSAnthony Liguori BDRVVPCState *s = bs->opaque; 331019d6b8fSAnthony Liguori int64_t bat_offset; 332019d6b8fSAnthony Liguori uint32_t index, bat_value; 333019d6b8fSAnthony Liguori int ret; 334019d6b8fSAnthony Liguori uint8_t bitmap[s->bitmap_size]; 335019d6b8fSAnthony Liguori 336019d6b8fSAnthony Liguori // Check if sector_num is valid 337019d6b8fSAnthony Liguori if ((sector_num < 0) || (sector_num > bs->total_sectors)) 338019d6b8fSAnthony Liguori return -1; 339019d6b8fSAnthony Liguori 340019d6b8fSAnthony Liguori // Write entry into in-memory BAT 341019d6b8fSAnthony Liguori index = (sector_num * 512) / s->block_size; 342019d6b8fSAnthony Liguori if (s->pagetable[index] != 0xFFFFFFFF) 343019d6b8fSAnthony Liguori return -1; 344019d6b8fSAnthony Liguori 345019d6b8fSAnthony Liguori s->pagetable[index] = s->free_data_block_offset / 512; 346019d6b8fSAnthony Liguori 347019d6b8fSAnthony Liguori // Initialize the block's bitmap 348019d6b8fSAnthony Liguori memset(bitmap, 0xff, s->bitmap_size); 349*078a458eSKevin Wolf bdrv_pwrite_sync(bs->file, s->free_data_block_offset, bitmap, 350*078a458eSKevin Wolf s->bitmap_size); 351019d6b8fSAnthony Liguori 352019d6b8fSAnthony Liguori // Write new footer (the old one will be overwritten) 353019d6b8fSAnthony Liguori s->free_data_block_offset += s->block_size + s->bitmap_size; 354019d6b8fSAnthony Liguori ret = rewrite_footer(bs); 355019d6b8fSAnthony Liguori if (ret < 0) 356019d6b8fSAnthony Liguori goto fail; 357019d6b8fSAnthony Liguori 358019d6b8fSAnthony Liguori // Write BAT entry to disk 359019d6b8fSAnthony Liguori bat_offset = s->bat_offset + (4 * index); 360019d6b8fSAnthony Liguori bat_value = be32_to_cpu(s->pagetable[index]); 361*078a458eSKevin Wolf ret = bdrv_pwrite_sync(bs->file, bat_offset, &bat_value, 4); 362019d6b8fSAnthony Liguori if (ret < 0) 363019d6b8fSAnthony Liguori goto fail; 364019d6b8fSAnthony Liguori 365019d6b8fSAnthony Liguori return get_sector_offset(bs, sector_num, 0); 366019d6b8fSAnthony Liguori 367019d6b8fSAnthony Liguori fail: 368019d6b8fSAnthony Liguori s->free_data_block_offset -= (s->block_size + s->bitmap_size); 369019d6b8fSAnthony Liguori return -1; 370019d6b8fSAnthony Liguori } 371019d6b8fSAnthony Liguori 372019d6b8fSAnthony Liguori static int vpc_read(BlockDriverState *bs, int64_t sector_num, 373019d6b8fSAnthony Liguori uint8_t *buf, int nb_sectors) 374019d6b8fSAnthony Liguori { 3756c6ea921SKevin Wolf BDRVVPCState *s = bs->opaque; 376019d6b8fSAnthony Liguori int ret; 377019d6b8fSAnthony Liguori int64_t offset; 3786c6ea921SKevin Wolf int64_t sectors, sectors_per_block; 379019d6b8fSAnthony Liguori 380019d6b8fSAnthony Liguori while (nb_sectors > 0) { 381019d6b8fSAnthony Liguori offset = get_sector_offset(bs, sector_num, 0); 382019d6b8fSAnthony Liguori 3836c6ea921SKevin Wolf sectors_per_block = s->block_size >> BDRV_SECTOR_BITS; 3846c6ea921SKevin Wolf sectors = sectors_per_block - (sector_num % sectors_per_block); 3856c6ea921SKevin Wolf if (sectors > nb_sectors) { 3866c6ea921SKevin Wolf sectors = nb_sectors; 387019d6b8fSAnthony Liguori } 388019d6b8fSAnthony Liguori 3896c6ea921SKevin Wolf if (offset == -1) { 3906c6ea921SKevin Wolf memset(buf, 0, sectors * BDRV_SECTOR_SIZE); 3916c6ea921SKevin Wolf } else { 3926c6ea921SKevin Wolf ret = bdrv_pread(bs->file, offset, buf, 3936c6ea921SKevin Wolf sectors * BDRV_SECTOR_SIZE); 3946c6ea921SKevin Wolf if (ret != sectors * BDRV_SECTOR_SIZE) { 3956c6ea921SKevin Wolf return -1; 3966c6ea921SKevin Wolf } 3976c6ea921SKevin Wolf } 3986c6ea921SKevin Wolf 3996c6ea921SKevin Wolf nb_sectors -= sectors; 4006c6ea921SKevin Wolf sector_num += sectors; 4016c6ea921SKevin Wolf buf += sectors * BDRV_SECTOR_SIZE; 402019d6b8fSAnthony Liguori } 403019d6b8fSAnthony Liguori return 0; 404019d6b8fSAnthony Liguori } 405019d6b8fSAnthony Liguori 406019d6b8fSAnthony Liguori static int vpc_write(BlockDriverState *bs, int64_t sector_num, 407019d6b8fSAnthony Liguori const uint8_t *buf, int nb_sectors) 408019d6b8fSAnthony Liguori { 4096c6ea921SKevin Wolf BDRVVPCState *s = bs->opaque; 410019d6b8fSAnthony Liguori int64_t offset; 4116c6ea921SKevin Wolf int64_t sectors, sectors_per_block; 412019d6b8fSAnthony Liguori int ret; 413019d6b8fSAnthony Liguori 414019d6b8fSAnthony Liguori while (nb_sectors > 0) { 415019d6b8fSAnthony Liguori offset = get_sector_offset(bs, sector_num, 1); 416019d6b8fSAnthony Liguori 4176c6ea921SKevin Wolf sectors_per_block = s->block_size >> BDRV_SECTOR_BITS; 4186c6ea921SKevin Wolf sectors = sectors_per_block - (sector_num % sectors_per_block); 4196c6ea921SKevin Wolf if (sectors > nb_sectors) { 4206c6ea921SKevin Wolf sectors = nb_sectors; 4216c6ea921SKevin Wolf } 4226c6ea921SKevin Wolf 423019d6b8fSAnthony Liguori if (offset == -1) { 424019d6b8fSAnthony Liguori offset = alloc_block(bs, sector_num); 425019d6b8fSAnthony Liguori if (offset < 0) 426019d6b8fSAnthony Liguori return -1; 427019d6b8fSAnthony Liguori } 428019d6b8fSAnthony Liguori 4296c6ea921SKevin Wolf ret = bdrv_pwrite(bs->file, offset, buf, sectors * BDRV_SECTOR_SIZE); 4306c6ea921SKevin Wolf if (ret != sectors * BDRV_SECTOR_SIZE) { 431019d6b8fSAnthony Liguori return -1; 4326c6ea921SKevin Wolf } 433019d6b8fSAnthony Liguori 4346c6ea921SKevin Wolf nb_sectors -= sectors; 4356c6ea921SKevin Wolf sector_num += sectors; 4366c6ea921SKevin Wolf buf += sectors * BDRV_SECTOR_SIZE; 437019d6b8fSAnthony Liguori } 438019d6b8fSAnthony Liguori 439019d6b8fSAnthony Liguori return 0; 440019d6b8fSAnthony Liguori } 441019d6b8fSAnthony Liguori 442019d6b8fSAnthony Liguori 443019d6b8fSAnthony Liguori /* 444019d6b8fSAnthony Liguori * Calculates the number of cylinders, heads and sectors per cylinder 445019d6b8fSAnthony Liguori * based on a given number of sectors. This is the algorithm described 446019d6b8fSAnthony Liguori * in the VHD specification. 447019d6b8fSAnthony Liguori * 448019d6b8fSAnthony Liguori * Note that the geometry doesn't always exactly match total_sectors but 449019d6b8fSAnthony Liguori * may round it down. 450019d6b8fSAnthony Liguori * 451019d6b8fSAnthony Liguori * Returns 0 on success, -EFBIG if the size is larger than 127 GB 452019d6b8fSAnthony Liguori */ 453019d6b8fSAnthony Liguori static int calculate_geometry(int64_t total_sectors, uint16_t* cyls, 454019d6b8fSAnthony Liguori uint8_t* heads, uint8_t* secs_per_cyl) 455019d6b8fSAnthony Liguori { 456019d6b8fSAnthony Liguori uint32_t cyls_times_heads; 457019d6b8fSAnthony Liguori 458019d6b8fSAnthony Liguori if (total_sectors > 65535 * 16 * 255) 459019d6b8fSAnthony Liguori return -EFBIG; 460019d6b8fSAnthony Liguori 461019d6b8fSAnthony Liguori if (total_sectors > 65535 * 16 * 63) { 462019d6b8fSAnthony Liguori *secs_per_cyl = 255; 463019d6b8fSAnthony Liguori *heads = 16; 464019d6b8fSAnthony Liguori cyls_times_heads = total_sectors / *secs_per_cyl; 465019d6b8fSAnthony Liguori } else { 466019d6b8fSAnthony Liguori *secs_per_cyl = 17; 467019d6b8fSAnthony Liguori cyls_times_heads = total_sectors / *secs_per_cyl; 468019d6b8fSAnthony Liguori *heads = (cyls_times_heads + 1023) / 1024; 469019d6b8fSAnthony Liguori 470019d6b8fSAnthony Liguori if (*heads < 4) 471019d6b8fSAnthony Liguori *heads = 4; 472019d6b8fSAnthony Liguori 473019d6b8fSAnthony Liguori if (cyls_times_heads >= (*heads * 1024) || *heads > 16) { 474019d6b8fSAnthony Liguori *secs_per_cyl = 31; 475019d6b8fSAnthony Liguori *heads = 16; 476019d6b8fSAnthony Liguori cyls_times_heads = total_sectors / *secs_per_cyl; 477019d6b8fSAnthony Liguori } 478019d6b8fSAnthony Liguori 479019d6b8fSAnthony Liguori if (cyls_times_heads >= (*heads * 1024)) { 480019d6b8fSAnthony Liguori *secs_per_cyl = 63; 481019d6b8fSAnthony Liguori *heads = 16; 482019d6b8fSAnthony Liguori cyls_times_heads = total_sectors / *secs_per_cyl; 483019d6b8fSAnthony Liguori } 484019d6b8fSAnthony Liguori } 485019d6b8fSAnthony Liguori 486dede4188SStefan Weil *cyls = cyls_times_heads / *heads; 487019d6b8fSAnthony Liguori 488019d6b8fSAnthony Liguori return 0; 489019d6b8fSAnthony Liguori } 490019d6b8fSAnthony Liguori 4910e7e1989SKevin Wolf static int vpc_create(const char *filename, QEMUOptionParameter *options) 492019d6b8fSAnthony Liguori { 493019d6b8fSAnthony Liguori uint8_t buf[1024]; 494019d6b8fSAnthony Liguori struct vhd_footer* footer = (struct vhd_footer*) buf; 495019d6b8fSAnthony Liguori struct vhd_dyndisk_header* dyndisk_header = 496019d6b8fSAnthony Liguori (struct vhd_dyndisk_header*) buf; 497019d6b8fSAnthony Liguori int fd, i; 498dede4188SStefan Weil uint16_t cyls = 0; 499dede4188SStefan Weil uint8_t heads = 0; 500dede4188SStefan Weil uint8_t secs_per_cyl = 0; 501019d6b8fSAnthony Liguori size_t block_size, num_bat_entries; 5020e7e1989SKevin Wolf int64_t total_sectors = 0; 503019d6b8fSAnthony Liguori 5040e7e1989SKevin Wolf // Read out options 5050e7e1989SKevin Wolf while (options && options->name) { 5060e7e1989SKevin Wolf if (!strcmp(options->name, "size")) { 5070e7e1989SKevin Wolf total_sectors = options->value.n / 512; 5080e7e1989SKevin Wolf } 5090e7e1989SKevin Wolf options++; 5100e7e1989SKevin Wolf } 511019d6b8fSAnthony Liguori 5120e7e1989SKevin Wolf // Create the file 513019d6b8fSAnthony Liguori fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0644); 514019d6b8fSAnthony Liguori if (fd < 0) 515019d6b8fSAnthony Liguori return -EIO; 516019d6b8fSAnthony Liguori 517dede4188SStefan Weil /* Calculate matching total_size and geometry. Increase the number of 518dede4188SStefan Weil sectors requested until we get enough (or fail). */ 519dede4188SStefan Weil for (i = 0; total_sectors > (int64_t)cyls * heads * secs_per_cyl; i++) { 520dede4188SStefan Weil if (calculate_geometry(total_sectors + i, 521dede4188SStefan Weil &cyls, &heads, &secs_per_cyl)) { 522019d6b8fSAnthony Liguori return -EFBIG; 523dede4188SStefan Weil } 524dede4188SStefan Weil } 525019d6b8fSAnthony Liguori total_sectors = (int64_t) cyls * heads * secs_per_cyl; 526019d6b8fSAnthony Liguori 527019d6b8fSAnthony Liguori // Prepare the Hard Disk Footer 528019d6b8fSAnthony Liguori memset(buf, 0, 1024); 529019d6b8fSAnthony Liguori 5305ec4d682SNathan Froyd memcpy(footer->creator, "conectix", 8); 531019d6b8fSAnthony Liguori // TODO Check if "qemu" creator_app is ok for VPC 5325ec4d682SNathan Froyd memcpy(footer->creator_app, "qemu", 4); 5335ec4d682SNathan Froyd memcpy(footer->creator_os, "Wi2k", 4); 534019d6b8fSAnthony Liguori 535019d6b8fSAnthony Liguori footer->features = be32_to_cpu(0x02); 536019d6b8fSAnthony Liguori footer->version = be32_to_cpu(0x00010000); 537019d6b8fSAnthony Liguori footer->data_offset = be64_to_cpu(HEADER_SIZE); 538019d6b8fSAnthony Liguori footer->timestamp = be32_to_cpu(time(NULL) - VHD_TIMESTAMP_BASE); 539019d6b8fSAnthony Liguori 540019d6b8fSAnthony Liguori // Version of Virtual PC 2007 541019d6b8fSAnthony Liguori footer->major = be16_to_cpu(0x0005); 542019d6b8fSAnthony Liguori footer->minor =be16_to_cpu(0x0003); 543019d6b8fSAnthony Liguori 544019d6b8fSAnthony Liguori footer->orig_size = be64_to_cpu(total_sectors * 512); 545019d6b8fSAnthony Liguori footer->size = be64_to_cpu(total_sectors * 512); 546019d6b8fSAnthony Liguori 547019d6b8fSAnthony Liguori footer->cyls = be16_to_cpu(cyls); 548019d6b8fSAnthony Liguori footer->heads = heads; 549019d6b8fSAnthony Liguori footer->secs_per_cyl = secs_per_cyl; 550019d6b8fSAnthony Liguori 551019d6b8fSAnthony Liguori footer->type = be32_to_cpu(VHD_DYNAMIC); 552019d6b8fSAnthony Liguori 553019d6b8fSAnthony Liguori // TODO uuid is missing 554019d6b8fSAnthony Liguori 555019d6b8fSAnthony Liguori footer->checksum = be32_to_cpu(vpc_checksum(buf, HEADER_SIZE)); 556019d6b8fSAnthony Liguori 557019d6b8fSAnthony Liguori // Write the footer (twice: at the beginning and at the end) 558019d6b8fSAnthony Liguori block_size = 0x200000; 559019d6b8fSAnthony Liguori num_bat_entries = (total_sectors + block_size / 512) / (block_size / 512); 560019d6b8fSAnthony Liguori 561019d6b8fSAnthony Liguori if (write(fd, buf, HEADER_SIZE) != HEADER_SIZE) 562019d6b8fSAnthony Liguori return -EIO; 563019d6b8fSAnthony Liguori 564019d6b8fSAnthony Liguori if (lseek(fd, 1536 + ((num_bat_entries * 4 + 511) & ~511), SEEK_SET) < 0) 565019d6b8fSAnthony Liguori return -EIO; 566019d6b8fSAnthony Liguori if (write(fd, buf, HEADER_SIZE) != HEADER_SIZE) 567019d6b8fSAnthony Liguori return -EIO; 568019d6b8fSAnthony Liguori 569019d6b8fSAnthony Liguori // Write the initial BAT 570019d6b8fSAnthony Liguori if (lseek(fd, 3 * 512, SEEK_SET) < 0) 571019d6b8fSAnthony Liguori return -EIO; 572019d6b8fSAnthony Liguori 573019d6b8fSAnthony Liguori memset(buf, 0xFF, 512); 574019d6b8fSAnthony Liguori for (i = 0; i < (num_bat_entries * 4 + 511) / 512; i++) 575019d6b8fSAnthony Liguori if (write(fd, buf, 512) != 512) 576019d6b8fSAnthony Liguori return -EIO; 577019d6b8fSAnthony Liguori 578019d6b8fSAnthony Liguori 579019d6b8fSAnthony Liguori // Prepare the Dynamic Disk Header 580019d6b8fSAnthony Liguori memset(buf, 0, 1024); 581019d6b8fSAnthony Liguori 5825ec4d682SNathan Froyd memcpy(dyndisk_header->magic, "cxsparse", 8); 583019d6b8fSAnthony Liguori 584019d6b8fSAnthony Liguori dyndisk_header->data_offset = be64_to_cpu(0xFFFFFFFF); 585019d6b8fSAnthony Liguori dyndisk_header->table_offset = be64_to_cpu(3 * 512); 586019d6b8fSAnthony Liguori dyndisk_header->version = be32_to_cpu(0x00010000); 587019d6b8fSAnthony Liguori dyndisk_header->block_size = be32_to_cpu(block_size); 588019d6b8fSAnthony Liguori dyndisk_header->max_table_entries = be32_to_cpu(num_bat_entries); 589019d6b8fSAnthony Liguori 590019d6b8fSAnthony Liguori dyndisk_header->checksum = be32_to_cpu(vpc_checksum(buf, 1024)); 591019d6b8fSAnthony Liguori 592019d6b8fSAnthony Liguori // Write the header 593019d6b8fSAnthony Liguori if (lseek(fd, 512, SEEK_SET) < 0) 594019d6b8fSAnthony Liguori return -EIO; 595019d6b8fSAnthony Liguori if (write(fd, buf, 1024) != 1024) 596019d6b8fSAnthony Liguori return -EIO; 597019d6b8fSAnthony Liguori 598019d6b8fSAnthony Liguori close(fd); 599019d6b8fSAnthony Liguori return 0; 600019d6b8fSAnthony Liguori } 601019d6b8fSAnthony Liguori 602019d6b8fSAnthony Liguori static void vpc_close(BlockDriverState *bs) 603019d6b8fSAnthony Liguori { 604019d6b8fSAnthony Liguori BDRVVPCState *s = bs->opaque; 605019d6b8fSAnthony Liguori qemu_free(s->pagetable); 606019d6b8fSAnthony Liguori #ifdef CACHE 607019d6b8fSAnthony Liguori qemu_free(s->pageentry_u8); 608019d6b8fSAnthony Liguori #endif 609019d6b8fSAnthony Liguori } 610019d6b8fSAnthony Liguori 6110e7e1989SKevin Wolf static QEMUOptionParameter vpc_create_options[] = { 612db08adf5SKevin Wolf { 613db08adf5SKevin Wolf .name = BLOCK_OPT_SIZE, 614db08adf5SKevin Wolf .type = OPT_SIZE, 615db08adf5SKevin Wolf .help = "Virtual disk size" 616db08adf5SKevin Wolf }, 6170e7e1989SKevin Wolf { NULL } 6180e7e1989SKevin Wolf }; 6190e7e1989SKevin Wolf 620019d6b8fSAnthony Liguori static BlockDriver bdrv_vpc = { 621019d6b8fSAnthony Liguori .format_name = "vpc", 622019d6b8fSAnthony Liguori .instance_size = sizeof(BDRVVPCState), 623019d6b8fSAnthony Liguori .bdrv_probe = vpc_probe, 624019d6b8fSAnthony Liguori .bdrv_open = vpc_open, 625019d6b8fSAnthony Liguori .bdrv_read = vpc_read, 626019d6b8fSAnthony Liguori .bdrv_write = vpc_write, 627019d6b8fSAnthony Liguori .bdrv_close = vpc_close, 628019d6b8fSAnthony Liguori .bdrv_create = vpc_create, 6290e7e1989SKevin Wolf 6300e7e1989SKevin Wolf .create_options = vpc_create_options, 631019d6b8fSAnthony Liguori }; 632019d6b8fSAnthony Liguori 633019d6b8fSAnthony Liguori static void bdrv_vpc_init(void) 634019d6b8fSAnthony Liguori { 635019d6b8fSAnthony Liguori bdrv_register(&bdrv_vpc); 636019d6b8fSAnthony Liguori } 637019d6b8fSAnthony Liguori 638019d6b8fSAnthony Liguori block_init(bdrv_vpc_init); 639