1/* 2 * Block driver for the VMDK format 3 * 4 * Copyright (c) 2004 Fabrice Bellard 5 * Copyright (c) 2005 Filip Navara 6 * 7 * Permission is hereby granted, free of charge, to any person obtaining a copy 8 * of this software and associated documentation files (the "Software"), to deal 9 * in the Software without restriction, including without limitation the rights 10 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 11 * copies of the Software, and to permit persons to whom the Software is 12 * furnished to do so, subject to the following conditions: 13 * 14 * The above copyright notice and this permission notice shall be included in 15 * all copies or substantial portions of the Software. 16 * 17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 18 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 19 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 20 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 21 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 22 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 23 * THE SOFTWARE. 24 */ 25 26#include "qemu-common.h" 27#include "block_int.h" 28#include "module.h" 29 30#define VMDK3_MAGIC (('C' << 24) | ('O' << 16) | ('W' << 8) | 'D') 31#define VMDK4_MAGIC (('K' << 24) | ('D' << 16) | ('M' << 8) | 'V') 32 33typedef struct { 34 uint32_t version; 35 uint32_t flags; 36 uint32_t disk_sectors; 37 uint32_t granularity; 38 uint32_t l1dir_offset; 39 uint32_t l1dir_size; 40 uint32_t file_sectors; 41 uint32_t cylinders; 42 uint32_t heads; 43 uint32_t sectors_per_track; 44} VMDK3Header; 45 46typedef struct { 47 uint32_t version; 48 uint32_t flags; 49 int64_t capacity; 50 int64_t granularity; 51 int64_t desc_offset; 52 int64_t desc_size; 53 int32_t num_gtes_per_gte; 54 int64_t rgd_offset; 55 int64_t gd_offset; 56 int64_t grain_offset; 57 char filler[1]; 58 char check_bytes[4]; 59} __attribute__((packed)) VMDK4Header; 60 61#define L2_CACHE_SIZE 16 62 63typedef struct BDRVVmdkState { 64 BlockDriverState *hd; 65 int64_t l1_table_offset; 66 int64_t l1_backup_table_offset; 67 uint32_t *l1_table; 68 uint32_t *l1_backup_table; 69 unsigned int l1_size; 70 uint32_t l1_entry_sectors; 71 72 unsigned int l2_size; 73 uint32_t *l2_cache; 74 uint32_t l2_cache_offsets[L2_CACHE_SIZE]; 75 uint32_t l2_cache_counts[L2_CACHE_SIZE]; 76 77 unsigned int cluster_sectors; 78 uint32_t parent_cid; 79} BDRVVmdkState; 80 81typedef struct VmdkMetaData { 82 uint32_t offset; 83 unsigned int l1_index; 84 unsigned int l2_index; 85 unsigned int l2_offset; 86 int valid; 87} VmdkMetaData; 88 89static int vmdk_probe(const uint8_t *buf, int buf_size, const char *filename) 90{ 91 uint32_t magic; 92 93 if (buf_size < 4) 94 return 0; 95 magic = be32_to_cpu(*(uint32_t *)buf); 96 if (magic == VMDK3_MAGIC || 97 magic == VMDK4_MAGIC) 98 return 100; 99 else 100 return 0; 101} 102 103#define CHECK_CID 1 104 105#define SECTOR_SIZE 512 106#define DESC_SIZE 20*SECTOR_SIZE // 20 sectors of 512 bytes each 107#define HEADER_SIZE 512 // first sector of 512 bytes 108 109static uint32_t vmdk_read_cid(BlockDriverState *bs, int parent) 110{ 111 char desc[DESC_SIZE]; 112 uint32_t cid; 113 const char *p_name, *cid_str; 114 size_t cid_str_size; 115 116 /* the descriptor offset = 0x200 */ 117 if (bdrv_pread(bs->file, 0x200, desc, DESC_SIZE) != DESC_SIZE) 118 return 0; 119 120 if (parent) { 121 cid_str = "parentCID"; 122 cid_str_size = sizeof("parentCID"); 123 } else { 124 cid_str = "CID"; 125 cid_str_size = sizeof("CID"); 126 } 127 128 if ((p_name = strstr(desc,cid_str)) != NULL) { 129 p_name += cid_str_size; 130 sscanf(p_name,"%x",&cid); 131 } 132 133 return cid; 134} 135 136static int vmdk_write_cid(BlockDriverState *bs, uint32_t cid) 137{ 138 char desc[DESC_SIZE], tmp_desc[DESC_SIZE]; 139 char *p_name, *tmp_str; 140 141 /* the descriptor offset = 0x200 */ 142 if (bdrv_pread(bs->file, 0x200, desc, DESC_SIZE) != DESC_SIZE) 143 return -1; 144 145 tmp_str = strstr(desc,"parentCID"); 146 pstrcpy(tmp_desc, sizeof(tmp_desc), tmp_str); 147 if ((p_name = strstr(desc,"CID")) != NULL) { 148 p_name += sizeof("CID"); 149 snprintf(p_name, sizeof(desc) - (p_name - desc), "%x\n", cid); 150 pstrcat(desc, sizeof(desc), tmp_desc); 151 } 152 153 if (bdrv_pwrite_sync(bs->file, 0x200, desc, DESC_SIZE) < 0) 154 return -1; 155 return 0; 156} 157 158static int vmdk_is_cid_valid(BlockDriverState *bs) 159{ 160#ifdef CHECK_CID 161 BDRVVmdkState *s = bs->opaque; 162 BlockDriverState *p_bs = bs->backing_hd; 163 uint32_t cur_pcid; 164 165 if (p_bs) { 166 cur_pcid = vmdk_read_cid(p_bs,0); 167 if (s->parent_cid != cur_pcid) 168 // CID not valid 169 return 0; 170 } 171#endif 172 // CID valid 173 return 1; 174} 175 176static int vmdk_snapshot_create(const char *filename, const char *backing_file) 177{ 178 int snp_fd, p_fd; 179 int ret; 180 uint32_t p_cid; 181 char *p_name, *gd_buf, *rgd_buf; 182 const char *real_filename, *temp_str; 183 VMDK4Header header; 184 uint32_t gde_entries, gd_size; 185 int64_t gd_offset, rgd_offset, capacity, gt_size; 186 char p_desc[DESC_SIZE], s_desc[DESC_SIZE], hdr[HEADER_SIZE]; 187 static const char desc_template[] = 188 "# Disk DescriptorFile\n" 189 "version=1\n" 190 "CID=%x\n" 191 "parentCID=%x\n" 192 "createType=\"monolithicSparse\"\n" 193 "parentFileNameHint=\"%s\"\n" 194 "\n" 195 "# Extent description\n" 196 "RW %u SPARSE \"%s\"\n" 197 "\n" 198 "# The Disk Data Base \n" 199 "#DDB\n" 200 "\n"; 201 202 snp_fd = open(filename, O_RDWR | O_CREAT | O_TRUNC | O_BINARY | O_LARGEFILE, 0644); 203 if (snp_fd < 0) 204 return -errno; 205 p_fd = open(backing_file, O_RDONLY | O_BINARY | O_LARGEFILE); 206 if (p_fd < 0) { 207 close(snp_fd); 208 return -errno; 209 } 210 211 /* read the header */ 212 if (lseek(p_fd, 0x0, SEEK_SET) == -1) { 213 ret = -errno; 214 goto fail; 215 } 216 if (read(p_fd, hdr, HEADER_SIZE) != HEADER_SIZE) { 217 ret = -errno; 218 goto fail; 219 } 220 221 /* write the header */ 222 if (lseek(snp_fd, 0x0, SEEK_SET) == -1) { 223 ret = -errno; 224 goto fail; 225 } 226 if (write(snp_fd, hdr, HEADER_SIZE) == -1) { 227 ret = -errno; 228 goto fail; 229 } 230 231 memset(&header, 0, sizeof(header)); 232 memcpy(&header,&hdr[4], sizeof(header)); // skip the VMDK4_MAGIC 233 234 if (ftruncate(snp_fd, header.grain_offset << 9)) { 235 ret = -errno; 236 goto fail; 237 } 238 /* the descriptor offset = 0x200 */ 239 if (lseek(p_fd, 0x200, SEEK_SET) == -1) { 240 ret = -errno; 241 goto fail; 242 } 243 if (read(p_fd, p_desc, DESC_SIZE) != DESC_SIZE) { 244 ret = -errno; 245 goto fail; 246 } 247 248 if ((p_name = strstr(p_desc,"CID")) != NULL) { 249 p_name += sizeof("CID"); 250 sscanf(p_name,"%x",&p_cid); 251 } 252 253 real_filename = filename; 254 if ((temp_str = strrchr(real_filename, '\\')) != NULL) 255 real_filename = temp_str + 1; 256 if ((temp_str = strrchr(real_filename, '/')) != NULL) 257 real_filename = temp_str + 1; 258 if ((temp_str = strrchr(real_filename, ':')) != NULL) 259 real_filename = temp_str + 1; 260 261 snprintf(s_desc, sizeof(s_desc), desc_template, p_cid, p_cid, backing_file, 262 (uint32_t)header.capacity, real_filename); 263 264 /* write the descriptor */ 265 if (lseek(snp_fd, 0x200, SEEK_SET) == -1) { 266 ret = -errno; 267 goto fail; 268 } 269 if (write(snp_fd, s_desc, strlen(s_desc)) == -1) { 270 ret = -errno; 271 goto fail; 272 } 273 274 gd_offset = header.gd_offset * SECTOR_SIZE; // offset of GD table 275 rgd_offset = header.rgd_offset * SECTOR_SIZE; // offset of RGD table 276 capacity = header.capacity * SECTOR_SIZE; // Extent size 277 /* 278 * Each GDE span 32M disk, means: 279 * 512 GTE per GT, each GTE points to grain 280 */ 281 gt_size = (int64_t)header.num_gtes_per_gte * header.granularity * SECTOR_SIZE; 282 if (!gt_size) { 283 ret = -EINVAL; 284 goto fail; 285 } 286 gde_entries = (uint32_t)(capacity / gt_size); // number of gde/rgde 287 gd_size = gde_entries * sizeof(uint32_t); 288 289 /* write RGD */ 290 rgd_buf = qemu_malloc(gd_size); 291 if (lseek(p_fd, rgd_offset, SEEK_SET) == -1) { 292 ret = -errno; 293 goto fail_rgd; 294 } 295 if (read(p_fd, rgd_buf, gd_size) != gd_size) { 296 ret = -errno; 297 goto fail_rgd; 298 } 299 if (lseek(snp_fd, rgd_offset, SEEK_SET) == -1) { 300 ret = -errno; 301 goto fail_rgd; 302 } 303 if (write(snp_fd, rgd_buf, gd_size) == -1) { 304 ret = -errno; 305 goto fail_rgd; 306 } 307 308 /* write GD */ 309 gd_buf = qemu_malloc(gd_size); 310 if (lseek(p_fd, gd_offset, SEEK_SET) == -1) { 311 ret = -errno; 312 goto fail_gd; 313 } 314 if (read(p_fd, gd_buf, gd_size) != gd_size) { 315 ret = -errno; 316 goto fail_gd; 317 } 318 if (lseek(snp_fd, gd_offset, SEEK_SET) == -1) { 319 ret = -errno; 320 goto fail_gd; 321 } 322 if (write(snp_fd, gd_buf, gd_size) == -1) { 323 ret = -errno; 324 goto fail_gd; 325 } 326 ret = 0; 327 328fail_gd: 329 qemu_free(gd_buf); 330fail_rgd: 331 qemu_free(rgd_buf); 332fail: 333 close(p_fd); 334 close(snp_fd); 335 return ret; 336} 337 338static int vmdk_parent_open(BlockDriverState *bs) 339{ 340 char *p_name; 341 char desc[DESC_SIZE]; 342 343 /* the descriptor offset = 0x200 */ 344 if (bdrv_pread(bs->file, 0x200, desc, DESC_SIZE) != DESC_SIZE) 345 return -1; 346 347 if ((p_name = strstr(desc,"parentFileNameHint")) != NULL) { 348 char *end_name; 349 350 p_name += sizeof("parentFileNameHint") + 1; 351 if ((end_name = strchr(p_name,'\"')) == NULL) 352 return -1; 353 if ((end_name - p_name) > sizeof (bs->backing_file) - 1) 354 return -1; 355 356 pstrcpy(bs->backing_file, end_name - p_name + 1, p_name); 357 } 358 359 return 0; 360} 361 362static int vmdk_open(BlockDriverState *bs, int flags) 363{ 364 BDRVVmdkState *s = bs->opaque; 365 uint32_t magic; 366 int l1_size, i; 367 368 if (bdrv_pread(bs->file, 0, &magic, sizeof(magic)) != sizeof(magic)) 369 goto fail; 370 371 magic = be32_to_cpu(magic); 372 if (magic == VMDK3_MAGIC) { 373 VMDK3Header header; 374 375 if (bdrv_pread(bs->file, sizeof(magic), &header, sizeof(header)) != sizeof(header)) 376 goto fail; 377 s->cluster_sectors = le32_to_cpu(header.granularity); 378 s->l2_size = 1 << 9; 379 s->l1_size = 1 << 6; 380 bs->total_sectors = le32_to_cpu(header.disk_sectors); 381 s->l1_table_offset = le32_to_cpu(header.l1dir_offset) << 9; 382 s->l1_backup_table_offset = 0; 383 s->l1_entry_sectors = s->l2_size * s->cluster_sectors; 384 } else if (magic == VMDK4_MAGIC) { 385 VMDK4Header header; 386 387 if (bdrv_pread(bs->file, sizeof(magic), &header, sizeof(header)) != sizeof(header)) 388 goto fail; 389 bs->total_sectors = le64_to_cpu(header.capacity); 390 s->cluster_sectors = le64_to_cpu(header.granularity); 391 s->l2_size = le32_to_cpu(header.num_gtes_per_gte); 392 s->l1_entry_sectors = s->l2_size * s->cluster_sectors; 393 if (s->l1_entry_sectors <= 0) 394 goto fail; 395 s->l1_size = (bs->total_sectors + s->l1_entry_sectors - 1) 396 / s->l1_entry_sectors; 397 s->l1_table_offset = le64_to_cpu(header.rgd_offset) << 9; 398 s->l1_backup_table_offset = le64_to_cpu(header.gd_offset) << 9; 399 400 // try to open parent images, if exist 401 if (vmdk_parent_open(bs) != 0) 402 goto fail; 403 // write the CID once after the image creation 404 s->parent_cid = vmdk_read_cid(bs,1); 405 } else { 406 goto fail; 407 } 408 409 /* read the L1 table */ 410 l1_size = s->l1_size * sizeof(uint32_t); 411 s->l1_table = qemu_malloc(l1_size); 412 if (bdrv_pread(bs->file, s->l1_table_offset, s->l1_table, l1_size) != l1_size) 413 goto fail; 414 for(i = 0; i < s->l1_size; i++) { 415 le32_to_cpus(&s->l1_table[i]); 416 } 417 418 if (s->l1_backup_table_offset) { 419 s->l1_backup_table = qemu_malloc(l1_size); 420 if (bdrv_pread(bs->file, s->l1_backup_table_offset, s->l1_backup_table, l1_size) != l1_size) 421 goto fail; 422 for(i = 0; i < s->l1_size; i++) { 423 le32_to_cpus(&s->l1_backup_table[i]); 424 } 425 } 426 427 s->l2_cache = qemu_malloc(s->l2_size * L2_CACHE_SIZE * sizeof(uint32_t)); 428 return 0; 429 fail: 430 qemu_free(s->l1_backup_table); 431 qemu_free(s->l1_table); 432 qemu_free(s->l2_cache); 433 return -1; 434} 435 436static uint64_t get_cluster_offset(BlockDriverState *bs, VmdkMetaData *m_data, 437 uint64_t offset, int allocate); 438 439static int get_whole_cluster(BlockDriverState *bs, uint64_t cluster_offset, 440 uint64_t offset, int allocate) 441{ 442 BDRVVmdkState *s = bs->opaque; 443 uint8_t whole_grain[s->cluster_sectors*512]; // 128 sectors * 512 bytes each = grain size 64KB 444 445 // we will be here if it's first write on non-exist grain(cluster). 446 // try to read from parent image, if exist 447 if (bs->backing_hd) { 448 int ret; 449 450 if (!vmdk_is_cid_valid(bs)) 451 return -1; 452 453 ret = bdrv_read(bs->backing_hd, offset >> 9, whole_grain, 454 s->cluster_sectors); 455 if (ret < 0) { 456 return -1; 457 } 458 459 //Write grain only into the active image 460 ret = bdrv_write(bs->file, cluster_offset, whole_grain, 461 s->cluster_sectors); 462 if (ret < 0) { 463 return -1; 464 } 465 } 466 return 0; 467} 468 469static int vmdk_L2update(BlockDriverState *bs, VmdkMetaData *m_data) 470{ 471 BDRVVmdkState *s = bs->opaque; 472 473 /* update L2 table */ 474 if (bdrv_pwrite_sync(bs->file, ((int64_t)m_data->l2_offset * 512) + (m_data->l2_index * sizeof(m_data->offset)), 475 &(m_data->offset), sizeof(m_data->offset)) < 0) 476 return -1; 477 /* update backup L2 table */ 478 if (s->l1_backup_table_offset != 0) { 479 m_data->l2_offset = s->l1_backup_table[m_data->l1_index]; 480 if (bdrv_pwrite_sync(bs->file, ((int64_t)m_data->l2_offset * 512) + (m_data->l2_index * sizeof(m_data->offset)), 481 &(m_data->offset), sizeof(m_data->offset)) < 0) 482 return -1; 483 } 484 485 return 0; 486} 487 488static uint64_t get_cluster_offset(BlockDriverState *bs, VmdkMetaData *m_data, 489 uint64_t offset, int allocate) 490{ 491 BDRVVmdkState *s = bs->opaque; 492 unsigned int l1_index, l2_offset, l2_index; 493 int min_index, i, j; 494 uint32_t min_count, *l2_table, tmp = 0; 495 uint64_t cluster_offset; 496 497 if (m_data) 498 m_data->valid = 0; 499 500 l1_index = (offset >> 9) / s->l1_entry_sectors; 501 if (l1_index >= s->l1_size) 502 return 0; 503 l2_offset = s->l1_table[l1_index]; 504 if (!l2_offset) 505 return 0; 506 for(i = 0; i < L2_CACHE_SIZE; i++) { 507 if (l2_offset == s->l2_cache_offsets[i]) { 508 /* increment the hit count */ 509 if (++s->l2_cache_counts[i] == 0xffffffff) { 510 for(j = 0; j < L2_CACHE_SIZE; j++) { 511 s->l2_cache_counts[j] >>= 1; 512 } 513 } 514 l2_table = s->l2_cache + (i * s->l2_size); 515 goto found; 516 } 517 } 518 /* not found: load a new entry in the least used one */ 519 min_index = 0; 520 min_count = 0xffffffff; 521 for(i = 0; i < L2_CACHE_SIZE; i++) { 522 if (s->l2_cache_counts[i] < min_count) { 523 min_count = s->l2_cache_counts[i]; 524 min_index = i; 525 } 526 } 527 l2_table = s->l2_cache + (min_index * s->l2_size); 528 if (bdrv_pread(bs->file, (int64_t)l2_offset * 512, l2_table, s->l2_size * sizeof(uint32_t)) != 529 s->l2_size * sizeof(uint32_t)) 530 return 0; 531 532 s->l2_cache_offsets[min_index] = l2_offset; 533 s->l2_cache_counts[min_index] = 1; 534 found: 535 l2_index = ((offset >> 9) / s->cluster_sectors) % s->l2_size; 536 cluster_offset = le32_to_cpu(l2_table[l2_index]); 537 538 if (!cluster_offset) { 539 if (!allocate) 540 return 0; 541 542 // Avoid the L2 tables update for the images that have snapshots. 543 cluster_offset = bdrv_getlength(bs->file); 544 bdrv_truncate(bs->file, cluster_offset + (s->cluster_sectors << 9)); 545 546 cluster_offset >>= 9; 547 tmp = cpu_to_le32(cluster_offset); 548 l2_table[l2_index] = tmp; 549 550 /* First of all we write grain itself, to avoid race condition 551 * that may to corrupt the image. 552 * This problem may occur because of insufficient space on host disk 553 * or inappropriate VM shutdown. 554 */ 555 if (get_whole_cluster(bs, cluster_offset, offset, allocate) == -1) 556 return 0; 557 558 if (m_data) { 559 m_data->offset = tmp; 560 m_data->l1_index = l1_index; 561 m_data->l2_index = l2_index; 562 m_data->l2_offset = l2_offset; 563 m_data->valid = 1; 564 } 565 } 566 cluster_offset <<= 9; 567 return cluster_offset; 568} 569 570static int vmdk_is_allocated(BlockDriverState *bs, int64_t sector_num, 571 int nb_sectors, int *pnum) 572{ 573 BDRVVmdkState *s = bs->opaque; 574 int index_in_cluster, n; 575 uint64_t cluster_offset; 576 577 cluster_offset = get_cluster_offset(bs, NULL, sector_num << 9, 0); 578 index_in_cluster = sector_num % s->cluster_sectors; 579 n = s->cluster_sectors - index_in_cluster; 580 if (n > nb_sectors) 581 n = nb_sectors; 582 *pnum = n; 583 return (cluster_offset != 0); 584} 585 586static int vmdk_read(BlockDriverState *bs, int64_t sector_num, 587 uint8_t *buf, int nb_sectors) 588{ 589 BDRVVmdkState *s = bs->opaque; 590 int index_in_cluster, n, ret; 591 uint64_t cluster_offset; 592 593 while (nb_sectors > 0) { 594 cluster_offset = get_cluster_offset(bs, NULL, sector_num << 9, 0); 595 index_in_cluster = sector_num % s->cluster_sectors; 596 n = s->cluster_sectors - index_in_cluster; 597 if (n > nb_sectors) 598 n = nb_sectors; 599 if (!cluster_offset) { 600 // try to read from parent image, if exist 601 if (bs->backing_hd) { 602 if (!vmdk_is_cid_valid(bs)) 603 return -1; 604 ret = bdrv_read(bs->backing_hd, sector_num, buf, n); 605 if (ret < 0) 606 return -1; 607 } else { 608 memset(buf, 0, 512 * n); 609 } 610 } else { 611 if(bdrv_pread(bs->file, cluster_offset + index_in_cluster * 512, buf, n * 512) != n * 512) 612 return -1; 613 } 614 nb_sectors -= n; 615 sector_num += n; 616 buf += n * 512; 617 } 618 return 0; 619} 620 621static int vmdk_write(BlockDriverState *bs, int64_t sector_num, 622 const uint8_t *buf, int nb_sectors) 623{ 624 BDRVVmdkState *s = bs->opaque; 625 VmdkMetaData m_data; 626 int index_in_cluster, n; 627 uint64_t cluster_offset; 628 static int cid_update = 0; 629 630 if (sector_num > bs->total_sectors) { 631 fprintf(stderr, 632 "(VMDK) Wrong offset: sector_num=0x%" PRIx64 633 " total_sectors=0x%" PRIx64 "\n", 634 sector_num, bs->total_sectors); 635 return -1; 636 } 637 638 while (nb_sectors > 0) { 639 index_in_cluster = sector_num & (s->cluster_sectors - 1); 640 n = s->cluster_sectors - index_in_cluster; 641 if (n > nb_sectors) 642 n = nb_sectors; 643 cluster_offset = get_cluster_offset(bs, &m_data, sector_num << 9, 1); 644 if (!cluster_offset) 645 return -1; 646 647 if (bdrv_pwrite(bs->file, cluster_offset + index_in_cluster * 512, buf, n * 512) != n * 512) 648 return -1; 649 if (m_data.valid) { 650 /* update L2 tables */ 651 if (vmdk_L2update(bs, &m_data) == -1) 652 return -1; 653 } 654 nb_sectors -= n; 655 sector_num += n; 656 buf += n * 512; 657 658 // update CID on the first write every time the virtual disk is opened 659 if (!cid_update) { 660 vmdk_write_cid(bs, time(NULL)); 661 cid_update++; 662 } 663 } 664 return 0; 665} 666 667static int vmdk_create(const char *filename, QEMUOptionParameter *options) 668{ 669 int fd, i; 670 VMDK4Header header; 671 uint32_t tmp, magic, grains, gd_size, gt_size, gt_count; 672 static const char desc_template[] = 673 "# Disk DescriptorFile\n" 674 "version=1\n" 675 "CID=%x\n" 676 "parentCID=ffffffff\n" 677 "createType=\"monolithicSparse\"\n" 678 "\n" 679 "# Extent description\n" 680 "RW %" PRId64 " SPARSE \"%s\"\n" 681 "\n" 682 "# The Disk Data Base \n" 683 "#DDB\n" 684 "\n" 685 "ddb.virtualHWVersion = \"%d\"\n" 686 "ddb.geometry.cylinders = \"%" PRId64 "\"\n" 687 "ddb.geometry.heads = \"16\"\n" 688 "ddb.geometry.sectors = \"63\"\n" 689 "ddb.adapterType = \"ide\"\n"; 690 char desc[1024]; 691 const char *real_filename, *temp_str; 692 int64_t total_size = 0; 693 const char *backing_file = NULL; 694 int flags = 0; 695 int ret; 696 697 // Read out options 698 while (options && options->name) { 699 if (!strcmp(options->name, BLOCK_OPT_SIZE)) { 700 total_size = options->value.n / 512; 701 } else if (!strcmp(options->name, BLOCK_OPT_BACKING_FILE)) { 702 backing_file = options->value.s; 703 } else if (!strcmp(options->name, BLOCK_OPT_COMPAT6)) { 704 flags |= options->value.n ? BLOCK_FLAG_COMPAT6: 0; 705 } 706 options++; 707 } 708 709 /* XXX: add support for backing file */ 710 if (backing_file) { 711 return vmdk_snapshot_create(filename, backing_file); 712 } 713 714 fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY | O_LARGEFILE, 715 0644); 716 if (fd < 0) 717 return -errno; 718 magic = cpu_to_be32(VMDK4_MAGIC); 719 memset(&header, 0, sizeof(header)); 720 header.version = cpu_to_le32(1); 721 header.flags = cpu_to_le32(3); /* ?? */ 722 header.capacity = cpu_to_le64(total_size); 723 header.granularity = cpu_to_le64(128); 724 header.num_gtes_per_gte = cpu_to_le32(512); 725 726 grains = (total_size + header.granularity - 1) / header.granularity; 727 gt_size = ((header.num_gtes_per_gte * sizeof(uint32_t)) + 511) >> 9; 728 gt_count = (grains + header.num_gtes_per_gte - 1) / header.num_gtes_per_gte; 729 gd_size = (gt_count * sizeof(uint32_t) + 511) >> 9; 730 731 header.desc_offset = 1; 732 header.desc_size = 20; 733 header.rgd_offset = header.desc_offset + header.desc_size; 734 header.gd_offset = header.rgd_offset + gd_size + (gt_size * gt_count); 735 header.grain_offset = 736 ((header.gd_offset + gd_size + (gt_size * gt_count) + 737 header.granularity - 1) / header.granularity) * 738 header.granularity; 739 740 header.desc_offset = cpu_to_le64(header.desc_offset); 741 header.desc_size = cpu_to_le64(header.desc_size); 742 header.rgd_offset = cpu_to_le64(header.rgd_offset); 743 header.gd_offset = cpu_to_le64(header.gd_offset); 744 header.grain_offset = cpu_to_le64(header.grain_offset); 745 746 header.check_bytes[0] = 0xa; 747 header.check_bytes[1] = 0x20; 748 header.check_bytes[2] = 0xd; 749 header.check_bytes[3] = 0xa; 750 751 /* write all the data */ 752 ret = qemu_write_full(fd, &magic, sizeof(magic)); 753 if (ret != sizeof(magic)) { 754 ret = -errno; 755 goto exit; 756 } 757 ret = qemu_write_full(fd, &header, sizeof(header)); 758 if (ret != sizeof(header)) { 759 ret = -errno; 760 goto exit; 761 } 762 763 ret = ftruncate(fd, header.grain_offset << 9); 764 if (ret < 0) { 765 ret = -errno; 766 goto exit; 767 } 768 769 /* write grain directory */ 770 lseek(fd, le64_to_cpu(header.rgd_offset) << 9, SEEK_SET); 771 for (i = 0, tmp = header.rgd_offset + gd_size; 772 i < gt_count; i++, tmp += gt_size) { 773 ret = qemu_write_full(fd, &tmp, sizeof(tmp)); 774 if (ret != sizeof(tmp)) { 775 ret = -errno; 776 goto exit; 777 } 778 } 779 780 /* write backup grain directory */ 781 lseek(fd, le64_to_cpu(header.gd_offset) << 9, SEEK_SET); 782 for (i = 0, tmp = header.gd_offset + gd_size; 783 i < gt_count; i++, tmp += gt_size) { 784 ret = qemu_write_full(fd, &tmp, sizeof(tmp)); 785 if (ret != sizeof(tmp)) { 786 ret = -errno; 787 goto exit; 788 } 789 } 790 791 /* compose the descriptor */ 792 real_filename = filename; 793 if ((temp_str = strrchr(real_filename, '\\')) != NULL) 794 real_filename = temp_str + 1; 795 if ((temp_str = strrchr(real_filename, '/')) != NULL) 796 real_filename = temp_str + 1; 797 if ((temp_str = strrchr(real_filename, ':')) != NULL) 798 real_filename = temp_str + 1; 799 snprintf(desc, sizeof(desc), desc_template, (unsigned int)time(NULL), 800 total_size, real_filename, 801 (flags & BLOCK_FLAG_COMPAT6 ? 6 : 4), 802 total_size / (int64_t)(63 * 16)); 803 804 /* write the descriptor */ 805 lseek(fd, le64_to_cpu(header.desc_offset) << 9, SEEK_SET); 806 ret = qemu_write_full(fd, desc, strlen(desc)); 807 if (ret != strlen(desc)) { 808 ret = -errno; 809 goto exit; 810 } 811 812 ret = 0; 813exit: 814 close(fd); 815 return ret; 816} 817 818static void vmdk_close(BlockDriverState *bs) 819{ 820 BDRVVmdkState *s = bs->opaque; 821 822 qemu_free(s->l1_table); 823 qemu_free(s->l2_cache); 824} 825 826static void vmdk_flush(BlockDriverState *bs) 827{ 828 bdrv_flush(bs->file); 829} 830 831 832static QEMUOptionParameter vmdk_create_options[] = { 833 { 834 .name = BLOCK_OPT_SIZE, 835 .type = OPT_SIZE, 836 .help = "Virtual disk size" 837 }, 838 { 839 .name = BLOCK_OPT_BACKING_FILE, 840 .type = OPT_STRING, 841 .help = "File name of a base image" 842 }, 843 { 844 .name = BLOCK_OPT_COMPAT6, 845 .type = OPT_FLAG, 846 .help = "VMDK version 6 image" 847 }, 848 { NULL } 849}; 850 851static BlockDriver bdrv_vmdk = { 852 .format_name = "vmdk", 853 .instance_size = sizeof(BDRVVmdkState), 854 .bdrv_probe = vmdk_probe, 855 .bdrv_open = vmdk_open, 856 .bdrv_read = vmdk_read, 857 .bdrv_write = vmdk_write, 858 .bdrv_close = vmdk_close, 859 .bdrv_create = vmdk_create, 860 .bdrv_flush = vmdk_flush, 861 .bdrv_is_allocated = vmdk_is_allocated, 862 863 .create_options = vmdk_create_options, 864}; 865 866static void bdrv_vmdk_init(void) 867{ 868 bdrv_register(&bdrv_vmdk); 869} 870 871block_init(bdrv_vmdk_init); 872