block/parallels.c

   1 /*
   2  * Block driver for Parallels disk image format
   3  *
   4  * Copyright (c) 2007 Alex Beregszaszi
   5  * Copyright (c) 2015 Denis V. Lunev <den@openvz.org>
   6  *
   7  * This code was originally based on comparing different disk images created
   8  * by Parallels. Currently it is based on opened OpenVZ sources
   9  * available at
  10  *     http://git.openvz.org/?p=ploop;a=summary
  11  *
  12  * Permission is hereby granted, free of charge, to any person obtaining a copy
  13  * of this software and associated documentation files (the "Software"), to deal
  14  * in the Software without restriction, including without limitation the rights
  15  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  16  * copies of the Software, and to permit persons to whom the Software is
  17  * furnished to do so, subject to the following conditions:
  18  *
  19  * The above copyright notice and this permission notice shall be included in
  20  * all copies or substantial portions of the Software.
  21  *
  22  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  23  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  24  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  25  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  26  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  27  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  28  * THE SOFTWARE.
  29  */
  30 #include "qemu/osdep.h"
  31 #include "qapi/error.h"
  32 #include "qemu-common.h"
  33 #include "block/block_int.h"
  34 #include "sysemu/block-backend.h"
  35 #include "qemu/module.h"
  36 #include "qemu/bswap.h"
  37 #include "qemu/bitmap.h"
  38 #include "qapi/util.h"
  39
  40 /**************************************************************/
  41
  42 #define HEADER_MAGIC "WithoutFreeSpace"
  43 #define HEADER_MAGIC2 "WithouFreSpacExt"
  44 #define HEADER_VERSION 2
  45 #define HEADER_INUSE_MAGIC  (0x746F6E59)
  46 #define MAX_PARALLELS_IMAGE_FACTOR (1ull << 32)
  47
  48 #define DEFAULT_CLUSTER_SIZE 1048576        /* 1 MiB */
  49
  50
  51 // always little-endian
  52 typedef struct ParallelsHeader {
  53     char magic[16]; // "WithoutFreeSpace"
  54     uint32_t version;
  55     uint32_t heads;
  56     uint32_t cylinders;
  57     uint32_t tracks;
  58     uint32_t bat_entries;
  59     uint64_t nb_sectors;
  60     uint32_t inuse;
  61     uint32_t data_off;
  62     char padding[12];
  63 } QEMU_PACKED ParallelsHeader;
  64
  65
  66 typedef enum ParallelsPreallocMode {
  67     PRL_PREALLOC_MODE_FALLOCATE = 0,
  68     PRL_PREALLOC_MODE_TRUNCATE = 1,
  69     PRL_PREALLOC_MODE__MAX = 2,
  70 } ParallelsPreallocMode;
  71
  72 static const char *prealloc_mode_lookup[] = {
  73     "falloc",
  74     "truncate",
  75     NULL,
  76 };
  77
  78
  79 typedef struct BDRVParallelsState {
  80     /** Locking is conservative, the lock protects
  81      *   - image file extending (truncate, fallocate)
  82      *   - any access to block allocation table
  83      */
  84     CoMutex lock;
  85
  86     ParallelsHeader *header;
  87     uint32_t header_size;
  88     bool header_unclean;
  89
  90     unsigned long *bat_dirty_bmap;
  91     unsigned int  bat_dirty_block;
  92
  93     uint32_t *bat_bitmap;
  94     unsigned int bat_size;
  95
  96     int64_t  data_end;
  97     uint64_t prealloc_size;
  98     ParallelsPreallocMode prealloc_mode;
  99
 100     unsigned int tracks;
 101
 102     unsigned int off_multiplier;
 103 } BDRVParallelsState;
 104
 105
 106 #define PARALLELS_OPT_PREALLOC_MODE     "prealloc-mode"
 107 #define PARALLELS_OPT_PREALLOC_SIZE     "prealloc-size"
 108
 109 static QemuOptsList parallels_runtime_opts = {
 110     .name = "parallels",
 111     .head = QTAILQ_HEAD_INITIALIZER(parallels_runtime_opts.head),
 112     .desc = {
 113         {
 114             .name = PARALLELS_OPT_PREALLOC_SIZE,
 115             .type = QEMU_OPT_SIZE,
 116             .help = "Preallocation size on image expansion",
 117             .def_value_str = "128M",
 118         },
 119         {
 120             .name = PARALLELS_OPT_PREALLOC_MODE,
 121             .type = QEMU_OPT_STRING,
 122             .help = "Preallocation mode on image expansion "
 123                     "(allowed values: falloc, truncate)",
 124             .def_value_str = "falloc",
 125         },
 126         { /* end of list */ },
 127     },
 128 };
 129
 130
 131 static int64_t bat2sect(BDRVParallelsState *s, uint32_t idx)
 132 {
 133     return (uint64_t)le32_to_cpu(s->bat_bitmap[idx]) * s->off_multiplier;
 134 }
 135
 136 static uint32_t bat_entry_off(uint32_t idx)
 137 {
 138     return sizeof(ParallelsHeader) + sizeof(uint32_t) * idx;
 139 }
 140
 141 static int64_t seek_to_sector(BDRVParallelsState *s, int64_t sector_num)
 142 {
 143     uint32_t index, offset;
 144
 145     index = sector_num / s->tracks;
 146     offset = sector_num % s->tracks;
 147
 148     /* not allocated */
 149     if ((index >= s->bat_size) || (s->bat_bitmap[index] == 0)) {
 150         return -1;
 151     }
 152     return bat2sect(s, index) + offset;
 153 }
 154
 155 static int cluster_remainder(BDRVParallelsState *s, int64_t sector_num,
 156         int nb_sectors)
 157 {
 158     int ret = s->tracks - sector_num % s->tracks;
 159     return MIN(nb_sectors, ret);
 160 }
 161
 162 static int64_t block_status(BDRVParallelsState *s, int64_t sector_num,
 163                             int nb_sectors, int *pnum)
 164 {
 165     int64_t start_off = -2, prev_end_off = -2;
 166
 167     *pnum = 0;
 168     while (nb_sectors > 0 || start_off == -2) {
 169         int64_t offset = seek_to_sector(s, sector_num);
 170         int to_end;
 171
 172         if (start_off == -2) {
 173             start_off = offset;
 174             prev_end_off = offset;
 175         } else if (offset != prev_end_off) {
 176             break;
 177         }
 178
 179         to_end = cluster_remainder(s, sector_num, nb_sectors);
 180         nb_sectors -= to_end;
 181         sector_num += to_end;
 182         *pnum += to_end;
 183
 184         if (offset > 0) {
 185             prev_end_off += to_end;
 186         }
 187     }
 188     return start_off;
 189 }
 190
 191 static int64_t allocate_clusters(BlockDriverState *bs, int64_t sector_num,
 192                                  int nb_sectors, int *pnum)
 193 {
 194     BDRVParallelsState *s = bs->opaque;
 195     int64_t pos, space, idx, to_allocate, i;
 196
 197     pos = block_status(s, sector_num, nb_sectors, pnum);
 198     if (pos > 0) {
 199         return pos;
 200     }
 201
 202     idx = sector_num / s->tracks;
 203     to_allocate = DIV_ROUND_UP(sector_num + *pnum, s->tracks) - idx;
 204
 205     /* This function is called only by parallels_co_writev(), which will never
 206      * pass a sector_num at or beyond the end of the image (because the block
 207      * layer never passes such a sector_num to that function). Therefore, idx
 208      * is always below s->bat_size.
 209      * block_status() will limit *pnum so that sector_num + *pnum will not
 210      * exceed the image end. Therefore, idx + to_allocate cannot exceed
 211      * s->bat_size.
 212      * Note that s->bat_size is an unsigned int, therefore idx + to_allocate
 213      * will always fit into a uint32_t. */
 214     assert(idx < s->bat_size && idx + to_allocate <= s->bat_size);
 215
 216     space = to_allocate * s->tracks;
 217     if (s->data_end + space > bdrv_getlength(bs->file->bs) >> BDRV_SECTOR_BITS) {
 218         int ret;
 219         space += s->prealloc_size;
 220         if (s->prealloc_mode == PRL_PREALLOC_MODE_FALLOCATE) {
 221             ret = bdrv_pwrite_zeroes(bs->file,
 222                                      s->data_end << BDRV_SECTOR_BITS,
 223                                      space << BDRV_SECTOR_BITS, 0);
 224         } else {
 225             ret = bdrv_truncate(bs->file,
 226                                 (s->data_end + space) << BDRV_SECTOR_BITS,
 227                                 NULL);
 228         }
 229         if (ret < 0) {
 230             return ret;
 231         }
 232     }
 233
 234     for (i = 0; i < to_allocate; i++) {
 235         s->bat_bitmap[idx + i] = cpu_to_le32(s->data_end / s->off_multiplier);
 236         s->data_end += s->tracks;
 237         bitmap_set(s->bat_dirty_bmap,
 238                    bat_entry_off(idx + i) / s->bat_dirty_block, 1);
 239     }
 240
 241     return bat2sect(s, idx) + sector_num % s->tracks;
 242 }
 243
 244
 245 static coroutine_fn int parallels_co_flush_to_os(BlockDriverState *bs)
 246 {
 247     BDRVParallelsState *s = bs->opaque;
 248     unsigned long size = DIV_ROUND_UP(s->header_size, s->bat_dirty_block);
 249     unsigned long bit;
 250
 251     qemu_co_mutex_lock(&s->lock);
 252
 253     bit = find_first_bit(s->bat_dirty_bmap, size);
 254     while (bit < size) {
 255         uint32_t off = bit * s->bat_dirty_block;
 256         uint32_t to_write = s->bat_dirty_block;
 257         int ret;
 258
 259         if (off + to_write > s->header_size) {
 260             to_write = s->header_size - off;
 261         }
 262         ret = bdrv_pwrite(bs->file, off, (uint8_t *)s->header + off,
 263                           to_write);
 264         if (ret < 0) {
 265             qemu_co_mutex_unlock(&s->lock);
 266             return ret;
 267         }
 268         bit = find_next_bit(s->bat_dirty_bmap, size, bit + 1);
 269     }
 270     bitmap_zero(s->bat_dirty_bmap, size);
 271
 272     qemu_co_mutex_unlock(&s->lock);
 273     return 0;
 274 }
 275
 276
 277 static int64_t coroutine_fn parallels_co_get_block_status(BlockDriverState *bs,
 278         int64_t sector_num, int nb_sectors, int *pnum, BlockDriverState **file)
 279 {
 280     BDRVParallelsState *s = bs->opaque;
 281     int64_t offset;
 282
 283     qemu_co_mutex_lock(&s->lock);
 284     offset = block_status(s, sector_num, nb_sectors, pnum);
 285     qemu_co_mutex_unlock(&s->lock);
 286
 287     if (offset < 0) {
 288         return 0;
 289     }
 290
 291     *file = bs->file->bs;
 292     return (offset << BDRV_SECTOR_BITS) |
 293         BDRV_BLOCK_DATA | BDRV_BLOCK_OFFSET_VALID;
 294 }
 295
 296 static coroutine_fn int parallels_co_writev(BlockDriverState *bs,
 297         int64_t sector_num, int nb_sectors, QEMUIOVector *qiov)
 298 {
 299     BDRVParallelsState *s = bs->opaque;
 300     uint64_t bytes_done = 0;
 301     QEMUIOVector hd_qiov;
 302     int ret = 0;
 303
 304     qemu_iovec_init(&hd_qiov, qiov->niov);
 305
 306     while (nb_sectors > 0) {
 307         int64_t position;
 308         int n, nbytes;
 309
 310         qemu_co_mutex_lock(&s->lock);
 311         position = allocate_clusters(bs, sector_num, nb_sectors, &n);
 312         qemu_co_mutex_unlock(&s->lock);
 313         if (position < 0) {
 314             ret = (int)position;
 315             break;
 316         }
 317
 318         nbytes = n << BDRV_SECTOR_BITS;
 319
 320         qemu_iovec_reset(&hd_qiov);
 321         qemu_iovec_concat(&hd_qiov, qiov, bytes_done, nbytes);
 322
 323         ret = bdrv_co_writev(bs->file, position, n, &hd_qiov);
 324         if (ret < 0) {
 325             break;
 326         }
 327
 328         nb_sectors -= n;
 329         sector_num += n;
 330         bytes_done += nbytes;
 331     }
 332
 333     qemu_iovec_destroy(&hd_qiov);
 334     return ret;
 335 }
 336
 337 static coroutine_fn int parallels_co_readv(BlockDriverState *bs,
 338         int64_t sector_num, int nb_sectors, QEMUIOVector *qiov)
 339 {
 340     BDRVParallelsState *s = bs->opaque;
 341     uint64_t bytes_done = 0;
 342     QEMUIOVector hd_qiov;
 343     int ret = 0;
 344
 345     qemu_iovec_init(&hd_qiov, qiov->niov);
 346
 347     while (nb_sectors > 0) {
 348         int64_t position;
 349         int n, nbytes;
 350
 351         qemu_co_mutex_lock(&s->lock);
 352         position = block_status(s, sector_num, nb_sectors, &n);
 353         qemu_co_mutex_unlock(&s->lock);
 354
 355         nbytes = n << BDRV_SECTOR_BITS;
 356
 357         if (position < 0) {
 358             qemu_iovec_memset(qiov, bytes_done, 0, nbytes);
 359         } else {
 360             qemu_iovec_reset(&hd_qiov);
 361             qemu_iovec_concat(&hd_qiov, qiov, bytes_done, nbytes);
 362
 363             ret = bdrv_co_readv(bs->file, position, n, &hd_qiov);
 364             if (ret < 0) {
 365                 break;
 366             }
 367         }
 368
 369         nb_sectors -= n;
 370         sector_num += n;
 371         bytes_done += nbytes;
 372     }
 373
 374     qemu_iovec_destroy(&hd_qiov);
 375     return ret;
 376 }
 377
 378
 379 static int parallels_check(BlockDriverState *bs, BdrvCheckResult *res,
 380                            BdrvCheckMode fix)
 381 {
 382     BDRVParallelsState *s = bs->opaque;
 383     int64_t size, prev_off, high_off;
 384     int ret;
 385     uint32_t i;
 386     bool flush_bat = false;
 387     int cluster_size = s->tracks << BDRV_SECTOR_BITS;
 388
 389     size = bdrv_getlength(bs->file->bs);
 390     if (size < 0) {
 391         res->check_errors++;
 392         return size;
 393     }
 394
 395     if (s->header_unclean) {
 396         fprintf(stderr, "%s image was not closed correctly\n",
 397                 fix & BDRV_FIX_ERRORS ? "Repairing" : "ERROR");
 398         res->corruptions++;
 399         if (fix & BDRV_FIX_ERRORS) {
 400             /* parallels_close will do the job right */
 401             res->corruptions_fixed++;
 402             s->header_unclean = false;
 403         }
 404     }
 405
 406     res->bfi.total_clusters = s->bat_size;
 407     res->bfi.compressed_clusters = 0; /* compression is not supported */
 408
 409     high_off = 0;
 410     prev_off = 0;
 411     for (i = 0; i < s->bat_size; i++) {
 412         int64_t off = bat2sect(s, i) << BDRV_SECTOR_BITS;
 413         if (off == 0) {
 414             prev_off = 0;
 415             continue;
 416         }
 417
 418         /* cluster outside the image */
 419         if (off > size) {
 420             fprintf(stderr, "%s cluster %u is outside image\n",
 421                     fix & BDRV_FIX_ERRORS ? "Repairing" : "ERROR", i);
 422             res->corruptions++;
 423             if (fix & BDRV_FIX_ERRORS) {
 424                 prev_off = 0;
 425                 s->bat_bitmap[i] = 0;
 426                 res->corruptions_fixed++;
 427                 flush_bat = true;
 428                 continue;
 429             }
 430         }
 431
 432         res->bfi.allocated_clusters++;
 433         if (off > high_off) {
 434             high_off = off;
 435         }
 436
 437         if (prev_off != 0 && (prev_off + cluster_size) != off) {
 438             res->bfi.fragmented_clusters++;
 439         }
 440         prev_off = off;
 441     }
 442
 443     if (flush_bat) {
 444         ret = bdrv_pwrite_sync(bs->file, 0, s->header, s->header_size);
 445         if (ret < 0) {
 446             res->check_errors++;
 447             return ret;
 448         }
 449     }
 450
 451     res->image_end_offset = high_off + cluster_size;
 452     if (size > res->image_end_offset) {
 453         int64_t count;
 454         count = DIV_ROUND_UP(size - res->image_end_offset, cluster_size);
 455         fprintf(stderr, "%s space leaked at the end of the image %" PRId64 "\n",
 456                 fix & BDRV_FIX_LEAKS ? "Repairing" : "ERROR",
 457                 size - res->image_end_offset);
 458         res->leaks += count;
 459         if (fix & BDRV_FIX_LEAKS) {
 460             Error *local_err = NULL;
 461             ret = bdrv_truncate(bs->file, res->image_end_offset, &local_err);
 462             if (ret < 0) {
 463                 error_report_err(local_err);
 464                 res->check_errors++;
 465                 return ret;
 466             }
 467             res->leaks_fixed += count;
 468         }
 469     }
 470
 471     return 0;
 472 }
 473
 474
 475 static int parallels_create(const char *filename, QemuOpts *opts, Error **errp)
 476 {
 477     int64_t total_size, cl_size;
 478     uint8_t tmp[BDRV_SECTOR_SIZE];
 479     Error *local_err = NULL;
 480     BlockBackend *file;
 481     uint32_t bat_entries, bat_sectors;
 482     ParallelsHeader header;
 483     int ret;
 484
 485     total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
 486                           BDRV_SECTOR_SIZE);
 487     cl_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_CLUSTER_SIZE,
 488                           DEFAULT_CLUSTER_SIZE), BDRV_SECTOR_SIZE);
 489     if (total_size >= MAX_PARALLELS_IMAGE_FACTOR * cl_size) {
 490         error_propagate(errp, local_err);
 491         return -E2BIG;
 492     }
 493
 494     ret = bdrv_create_file(filename, opts, &local_err);
 495     if (ret < 0) {
 496         error_propagate(errp, local_err);
 497         return ret;
 498     }
 499
 500     file = blk_new_open(filename, NULL, NULL,
 501                         BDRV_O_RDWR | BDRV_O_RESIZE | BDRV_O_PROTOCOL,
 502                         &local_err);
 503     if (file == NULL) {
 504         error_propagate(errp, local_err);
 505         return -EIO;
 506     }
 507
 508     blk_set_allow_write_beyond_eof(file, true);
 509
 510     ret = blk_truncate(file, 0, errp);
 511     if (ret < 0) {
 512         goto exit;
 513     }
 514
 515     bat_entries = DIV_ROUND_UP(total_size, cl_size);
 516     bat_sectors = DIV_ROUND_UP(bat_entry_off(bat_entries), cl_size);
 517     bat_sectors = (bat_sectors *  cl_size) >> BDRV_SECTOR_BITS;
 518
 519     memset(&header, 0, sizeof(header));
 520     memcpy(header.magic, HEADER_MAGIC2, sizeof(header.magic));
 521     header.version = cpu_to_le32(HEADER_VERSION);
 522     /* don't care much about geometry, it is not used on image level */
 523     header.heads = cpu_to_le32(16);
 524     header.cylinders = cpu_to_le32(total_size / BDRV_SECTOR_SIZE / 16 / 32);
 525     header.tracks = cpu_to_le32(cl_size >> BDRV_SECTOR_BITS);
 526     header.bat_entries = cpu_to_le32(bat_entries);
 527     header.nb_sectors = cpu_to_le64(DIV_ROUND_UP(total_size, BDRV_SECTOR_SIZE));
 528     header.data_off = cpu_to_le32(bat_sectors);
 529
 530     /* write all the data */
 531     memset(tmp, 0, sizeof(tmp));
 532     memcpy(tmp, &header, sizeof(header));
 533
 534     ret = blk_pwrite(file, 0, tmp, BDRV_SECTOR_SIZE, 0);
 535     if (ret < 0) {
 536         goto exit;
 537     }
 538     ret = blk_pwrite_zeroes(file, BDRV_SECTOR_SIZE,
 539                             (bat_sectors - 1) << BDRV_SECTOR_BITS, 0);
 540     if (ret < 0) {
 541         goto exit;
 542     }
 543     ret = 0;
 544
 545 done:
 546     blk_unref(file);
 547     return ret;
 548
 549 exit:
 550     error_setg_errno(errp, -ret, "Failed to create Parallels image");
 551     goto done;
 552 }
 553
 554
 555 static int parallels_probe(const uint8_t *buf, int buf_size,
 556                            const char *filename)
 557 {
 558     const ParallelsHeader *ph = (const void *)buf;
 559
 560     if (buf_size < sizeof(ParallelsHeader)) {
 561         return 0;
 562     }
 563
 564     if ((!memcmp(ph->magic, HEADER_MAGIC, 16) ||
 565            !memcmp(ph->magic, HEADER_MAGIC2, 16)) &&
 566            (le32_to_cpu(ph->version) == HEADER_VERSION)) {
 567         return 100;
 568     }
 569
 570     return 0;
 571 }
 572
 573 static int parallels_update_header(BlockDriverState *bs)
 574 {
 575     BDRVParallelsState *s = bs->opaque;
 576     unsigned size = MAX(bdrv_opt_mem_align(bs->file->bs),
 577                         sizeof(ParallelsHeader));
 578
 579     if (size > s->header_size) {
 580         size = s->header_size;
 581     }
 582     return bdrv_pwrite_sync(bs->file, 0, s->header, size);
 583 }
 584
 585 static int parallels_open(BlockDriverState *bs, QDict *options, int flags,
 586                           Error **errp)
 587 {
 588     BDRVParallelsState *s = bs->opaque;
 589     ParallelsHeader ph;
 590     int ret, size, i;
 591     QemuOpts *opts = NULL;
 592     Error *local_err = NULL;
 593     char *buf;
 594
 595     bs->file = bdrv_open_child(NULL, options, "file", bs, &child_file,
 596                                false, errp);
 597     if (!bs->file) {
 598         return -EINVAL;
 599     }
 600
 601     ret = bdrv_pread(bs->file, 0, &ph, sizeof(ph));
 602     if (ret < 0) {
 603         goto fail;
 604     }
 605
 606     bs->total_sectors = le64_to_cpu(ph.nb_sectors);
 607
 608     if (le32_to_cpu(ph.version) != HEADER_VERSION) {
 609         goto fail_format;
 610     }
 611     if (!memcmp(ph.magic, HEADER_MAGIC, 16)) {
 612         s->off_multiplier = 1;
 613         bs->total_sectors = 0xffffffff & bs->total_sectors;
 614     } else if (!memcmp(ph.magic, HEADER_MAGIC2, 16)) {
 615         s->off_multiplier = le32_to_cpu(ph.tracks);
 616     } else {
 617         goto fail_format;
 618     }
 619
 620     s->tracks = le32_to_cpu(ph.tracks);
 621     if (s->tracks == 0) {
 622         error_setg(errp, "Invalid image: Zero sectors per track");
 623         ret = -EINVAL;
 624         goto fail;
 625     }
 626     if (s->tracks > INT32_MAX/513) {
 627         error_setg(errp, "Invalid image: Too big cluster");
 628         ret = -EFBIG;
 629         goto fail;
 630     }
 631
 632     s->bat_size = le32_to_cpu(ph.bat_entries);
 633     if (s->bat_size > INT_MAX / sizeof(uint32_t)) {
 634         error_setg(errp, "Catalog too large");
 635         ret = -EFBIG;
 636         goto fail;
 637     }
 638
 639     size = bat_entry_off(s->bat_size);
 640     s->header_size = ROUND_UP(size, bdrv_opt_mem_align(bs->file->bs));
 641     s->header = qemu_try_blockalign(bs->file->bs, s->header_size);
 642     if (s->header == NULL) {
 643         ret = -ENOMEM;
 644         goto fail;
 645     }
 646     s->data_end = le32_to_cpu(ph.data_off);
 647     if (s->data_end == 0) {
 648         s->data_end = ROUND_UP(bat_entry_off(s->bat_size), BDRV_SECTOR_SIZE);
 649     }
 650     if (s->data_end < s->header_size) {
 651         /* there is not enough unused space to fit to block align between BAT
 652            and actual data. We can't avoid read-modify-write... */
 653         s->header_size = size;
 654     }
 655
 656     ret = bdrv_pread(bs->file, 0, s->header, s->header_size);
 657     if (ret < 0) {
 658         goto fail;
 659     }
 660     s->bat_bitmap = (uint32_t *)(s->header + 1);
 661
 662     for (i = 0; i < s->bat_size; i++) {
 663         int64_t off = bat2sect(s, i);
 664         if (off >= s->data_end) {
 665             s->data_end = off + s->tracks;
 666         }
 667     }
 668
 669     if (le32_to_cpu(ph.inuse) == HEADER_INUSE_MAGIC) {
 670         /* Image was not closed correctly. The check is mandatory */
 671         s->header_unclean = true;
 672         if ((flags & BDRV_O_RDWR) && !(flags & BDRV_O_CHECK)) {
 673             error_setg(errp, "parallels: Image was not closed correctly; "
 674                        "cannot be opened read/write");
 675             ret = -EACCES;
 676             goto fail;
 677         }
 678     }
 679
 680     opts = qemu_opts_create(&parallels_runtime_opts, NULL, 0, &local_err);
 681     if (local_err != NULL) {
 682         goto fail_options;
 683     }
 684
 685     qemu_opts_absorb_qdict(opts, options, &local_err);
 686     if (local_err != NULL) {
 687         goto fail_options;
 688     }
 689
 690     s->prealloc_size =
 691         qemu_opt_get_size_del(opts, PARALLELS_OPT_PREALLOC_SIZE, 0);
 692     s->prealloc_size = MAX(s->tracks, s->prealloc_size >> BDRV_SECTOR_BITS);
 693     buf = qemu_opt_get_del(opts, PARALLELS_OPT_PREALLOC_MODE);
 694     s->prealloc_mode = qapi_enum_parse(prealloc_mode_lookup, buf,
 695             PRL_PREALLOC_MODE__MAX, PRL_PREALLOC_MODE_FALLOCATE, &local_err);
 696     g_free(buf);
 697     if (local_err != NULL) {
 698         goto fail_options;
 699     }
 700
 701     if (!(flags & BDRV_O_RESIZE) || !bdrv_has_zero_init(bs->file->bs) ||
 702             bdrv_truncate(bs->file, bdrv_getlength(bs->file->bs), NULL) != 0) {
 703         s->prealloc_mode = PRL_PREALLOC_MODE_FALLOCATE;
 704     }
 705
 706     if (flags & BDRV_O_RDWR) {
 707         s->header->inuse = cpu_to_le32(HEADER_INUSE_MAGIC);
 708         ret = parallels_update_header(bs);
 709         if (ret < 0) {
 710             goto fail;
 711         }
 712     }
 713
 714     s->bat_dirty_block = 4 * getpagesize();
 715     s->bat_dirty_bmap =
 716         bitmap_new(DIV_ROUND_UP(s->header_size, s->bat_dirty_block));
 717
 718     qemu_co_mutex_init(&s->lock);
 719     return 0;
 720
 721 fail_format:
 722     error_setg(errp, "Image not in Parallels format");
 723     ret = -EINVAL;
 724 fail:
 725     qemu_vfree(s->header);
 726     return ret;
 727
 728 fail_options:
 729     error_propagate(errp, local_err);
 730     ret = -EINVAL;
 731     goto fail;
 732 }
 733
 734
 735 static void parallels_close(BlockDriverState *bs)
 736 {
 737     BDRVParallelsState *s = bs->opaque;
 738
 739     if (bs->open_flags & BDRV_O_RDWR) {
 740         s->header->inuse = 0;
 741         parallels_update_header(bs);
 742     }
 743
 744     if (bs->open_flags & BDRV_O_RDWR) {
 745         bdrv_truncate(bs->file, s->data_end << BDRV_SECTOR_BITS, NULL);
 746     }
 747
 748     g_free(s->bat_dirty_bmap);
 749     qemu_vfree(s->header);
 750 }
 751
 752 static QemuOptsList parallels_create_opts = {
 753     .name = "parallels-create-opts",
 754     .head = QTAILQ_HEAD_INITIALIZER(parallels_create_opts.head),
 755     .desc = {
 756         {
 757             .name = BLOCK_OPT_SIZE,
 758             .type = QEMU_OPT_SIZE,
 759             .help = "Virtual disk size",
 760         },
 761         {
 762             .name = BLOCK_OPT_CLUSTER_SIZE,
 763             .type = QEMU_OPT_SIZE,
 764             .help = "Parallels image cluster size",
 765             .def_value_str = stringify(DEFAULT_CLUSTER_SIZE),
 766         },
 767         { /* end of list */ }
 768     }
 769 };
 770
 771 static BlockDriver bdrv_parallels = {
 772     .format_name        = "parallels",
 773     .instance_size      = sizeof(BDRVParallelsState),
 774     .bdrv_probe         = parallels_probe,
 775     .bdrv_open          = parallels_open,
 776     .bdrv_close         = parallels_close,
 777     .bdrv_child_perm          = bdrv_format_default_perms,
 778     .bdrv_co_get_block_status = parallels_co_get_block_status,
 779     .bdrv_has_zero_init       = bdrv_has_zero_init_1,
 780     .bdrv_co_flush_to_os      = parallels_co_flush_to_os,
 781     .bdrv_co_readv  = parallels_co_readv,
 782     .bdrv_co_writev = parallels_co_writev,
 783
 784     .bdrv_create    = parallels_create,
 785     .bdrv_check     = parallels_check,
 786     .create_opts    = &parallels_create_opts,
 787 };
 788
 789 static void bdrv_parallels_init(void)
 790 {
 791     bdrv_register(&bdrv_parallels);
 792 }
 793
 794 block_init(bdrv_parallels_init);