vmdk.c 82.1 KB
Newer Older
B
bellard 已提交
1 2
/*
 * Block driver for the VMDK format
3
 *
B
bellard 已提交
4
 * Copyright (c) 2004 Fabrice Bellard
5
 * Copyright (c) 2005 Filip Navara
6
 *
B
bellard 已提交
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
25

P
Peter Maydell 已提交
26
#include "qemu/osdep.h"
27
#include "qapi/error.h"
28
#include "block/block_int.h"
29
#include "sysemu/block-backend.h"
30
#include "qapi/qmp/qerror.h"
31
#include "qemu/error-report.h"
32
#include "qemu/module.h"
33
#include "qemu/option.h"
34
#include "qemu/bswap.h"
35
#include "migration/blocker.h"
36
#include "qemu/cutils.h"
S
Stefan Weil 已提交
37
#include <zlib.h>
B
bellard 已提交
38 39 40

#define VMDK3_MAGIC (('C' << 24) | ('O' << 16) | ('W' << 8) | 'D')
#define VMDK4_MAGIC (('K' << 24) | ('D' << 16) | ('M' << 8) | 'V')
F
Fam Zheng 已提交
41
#define VMDK4_COMPRESSION_DEFLATE 1
F
Fam Zheng 已提交
42
#define VMDK4_FLAG_NL_DETECT (1 << 0)
43
#define VMDK4_FLAG_RGD (1 << 1)
44 45
/* Zeroed-grain enable bit */
#define VMDK4_FLAG_ZERO_GRAIN   (1 << 2)
F
Fam Zheng 已提交
46 47
#define VMDK4_FLAG_COMPRESS (1 << 16)
#define VMDK4_FLAG_MARKER (1 << 17)
48
#define VMDK4_GD_AT_END 0xffffffffffffffffULL
B
bellard 已提交
49

50 51
#define VMDK_EXTENT_MAX_SECTORS (1ULL << 32)

52
#define VMDK_GTE_ZEROED 0x1
F
Fam Zheng 已提交
53 54 55 56 57 58 59 60

/* VMDK internal error codes */
#define VMDK_OK      0
#define VMDK_ERROR   (-1)
/* Cluster not allocated */
#define VMDK_UNALLOC (-2)
#define VMDK_ZEROED  (-3)

61 62
#define BLOCK_OPT_ZEROED_GRAIN "zeroed_grain"

B
bellard 已提交
63 64 65 66 67 68 69 70 71 72 73
typedef struct {
    uint32_t version;
    uint32_t flags;
    uint32_t disk_sectors;
    uint32_t granularity;
    uint32_t l1dir_offset;
    uint32_t l1dir_size;
    uint32_t file_sectors;
    uint32_t cylinders;
    uint32_t heads;
    uint32_t sectors_per_track;
74
} QEMU_PACKED VMDK3Header;
B
bellard 已提交
75 76 77 78

typedef struct {
    uint32_t version;
    uint32_t flags;
79 80 81 82
    uint64_t capacity;
    uint64_t granularity;
    uint64_t desc_offset;
    uint64_t desc_size;
83 84
    /* Number of GrainTableEntries per GrainTable */
    uint32_t num_gtes_per_gt;
85 86 87
    uint64_t rgd_offset;
    uint64_t gd_offset;
    uint64_t grain_offset;
B
bellard 已提交
88 89
    char filler[1];
    char check_bytes[4];
F
Fam Zheng 已提交
90
    uint16_t compressAlgorithm;
91
} QEMU_PACKED VMDK4Header;
B
bellard 已提交
92 93 94

#define L2_CACHE_SIZE 16

F
Fam Zheng 已提交
95
typedef struct VmdkExtent {
96
    BdrvChild *file;
F
Fam Zheng 已提交
97
    bool flat;
F
Fam Zheng 已提交
98 99
    bool compressed;
    bool has_marker;
100 101
    bool has_zero_grain;
    int version;
F
Fam Zheng 已提交
102 103
    int64_t sectors;
    int64_t end_sector;
104
    int64_t flat_start_offset;
B
bellard 已提交
105
    int64_t l1_table_offset;
106
    int64_t l1_backup_table_offset;
B
bellard 已提交
107
    uint32_t *l1_table;
108
    uint32_t *l1_backup_table;
B
bellard 已提交
109 110 111 112 113 114 115 116
    unsigned int l1_size;
    uint32_t l1_entry_sectors;

    unsigned int l2_size;
    uint32_t *l2_cache;
    uint32_t l2_cache_offsets[L2_CACHE_SIZE];
    uint32_t l2_cache_counts[L2_CACHE_SIZE];

117
    int64_t cluster_sectors;
F
Fam Zheng 已提交
118
    int64_t next_cluster_sector;
F
Fam Zheng 已提交
119
    char *type;
F
Fam Zheng 已提交
120 121 122
} VmdkExtent;

typedef struct BDRVVmdkState {
123
    CoMutex lock;
124
    uint64_t desc_offset;
125
    bool cid_updated;
126
    bool cid_checked;
F
Fam Zheng 已提交
127
    uint32_t cid;
128
    uint32_t parent_cid;
F
Fam Zheng 已提交
129 130 131
    int num_extents;
    /* Extent array with num_extents entries, ascend ordered by address */
    VmdkExtent *extents;
K
Kevin Wolf 已提交
132
    Error *migration_blocker;
F
Fam Zheng 已提交
133
    char *create_type;
B
bellard 已提交
134 135
} BDRVVmdkState;

136 137 138 139 140
typedef struct VmdkMetaData {
    unsigned int l1_index;
    unsigned int l2_index;
    unsigned int l2_offset;
    int valid;
F
Fam Zheng 已提交
141
    uint32_t *l2_cache_entry;
142 143
} VmdkMetaData;

F
Fam Zheng 已提交
144 145 146 147
typedef struct VmdkGrainMarker {
    uint64_t lba;
    uint32_t size;
    uint8_t  data[0];
148
} QEMU_PACKED VmdkGrainMarker;
F
Fam Zheng 已提交
149

150 151 152 153 154 155 156
enum {
    MARKER_END_OF_STREAM    = 0,
    MARKER_GRAIN_TABLE      = 1,
    MARKER_GRAIN_DIRECTORY  = 2,
    MARKER_FOOTER           = 3,
};

B
bellard 已提交
157 158 159 160
static int vmdk_probe(const uint8_t *buf, int buf_size, const char *filename)
{
    uint32_t magic;

F
Fam Zheng 已提交
161
    if (buf_size < 4) {
B
bellard 已提交
162
        return 0;
F
Fam Zheng 已提交
163
    }
B
bellard 已提交
164 165
    magic = be32_to_cpu(*(uint32_t *)buf);
    if (magic == VMDK3_MAGIC ||
166
        magic == VMDK4_MAGIC) {
B
bellard 已提交
167
        return 100;
168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208
    } else {
        const char *p = (const char *)buf;
        const char *end = p + buf_size;
        while (p < end) {
            if (*p == '#') {
                /* skip comment line */
                while (p < end && *p != '\n') {
                    p++;
                }
                p++;
                continue;
            }
            if (*p == ' ') {
                while (p < end && *p == ' ') {
                    p++;
                }
                /* skip '\r' if windows line endings used. */
                if (p < end && *p == '\r') {
                    p++;
                }
                /* only accept blank lines before 'version=' line */
                if (p == end || *p != '\n') {
                    return 0;
                }
                p++;
                continue;
            }
            if (end - p >= strlen("version=X\n")) {
                if (strncmp("version=1\n", p, strlen("version=1\n")) == 0 ||
                    strncmp("version=2\n", p, strlen("version=2\n")) == 0) {
                    return 100;
                }
            }
            if (end - p >= strlen("version=X\r\n")) {
                if (strncmp("version=1\r\n", p, strlen("version=1\r\n")) == 0 ||
                    strncmp("version=2\r\n", p, strlen("version=2\r\n")) == 0) {
                    return 100;
                }
            }
            return 0;
        }
B
bellard 已提交
209
        return 0;
210
    }
B
bellard 已提交
211 212
}

213
#define SECTOR_SIZE 512
F
Fam Zheng 已提交
214 215 216
#define DESC_SIZE (20 * SECTOR_SIZE)    /* 20 sectors of 512 bytes each */
#define BUF_SIZE 4096
#define HEADER_SIZE 512                 /* first sector of 512 bytes */
217

F
Fam Zheng 已提交
218 219 220 221
static void vmdk_free_extents(BlockDriverState *bs)
{
    int i;
    BDRVVmdkState *s = bs->opaque;
F
Fam Zheng 已提交
222
    VmdkExtent *e;
F
Fam Zheng 已提交
223 224

    for (i = 0; i < s->num_extents; i++) {
F
Fam Zheng 已提交
225 226 227 228
        e = &s->extents[i];
        g_free(e->l1_table);
        g_free(e->l2_cache);
        g_free(e->l1_backup_table);
F
Fam Zheng 已提交
229
        g_free(e->type);
K
Kevin Wolf 已提交
230
        if (e->file != bs->file) {
231
            bdrv_unref_child(bs, e->file);
F
Fam Zheng 已提交
232
        }
F
Fam Zheng 已提交
233
    }
234
    g_free(s->extents);
F
Fam Zheng 已提交
235 236
}

237 238 239 240 241 242 243 244
static void vmdk_free_last_extent(BlockDriverState *bs)
{
    BDRVVmdkState *s = bs->opaque;

    if (s->num_extents == 0) {
        return;
    }
    s->num_extents--;
245
    s->extents = g_renew(VmdkExtent, s->extents, s->num_extents);
246 247
}

248 249
/* Return -ve errno, or 0 on success and write CID into *pcid. */
static int vmdk_read_cid(BlockDriverState *bs, int parent, uint32_t *pcid)
B
bellard 已提交
250
{
251
    char *desc;
252
    uint32_t cid;
253
    const char *p_name, *cid_str;
254
    size_t cid_str_size;
255
    BDRVVmdkState *s = bs->opaque;
K
Kevin Wolf 已提交
256
    int ret;
257

258
    desc = g_malloc0(DESC_SIZE);
259
    ret = bdrv_pread(bs->file, s->desc_offset, desc, DESC_SIZE);
K
Kevin Wolf 已提交
260
    if (ret < 0) {
261
        goto out;
262
    }
263 264 265 266 267 268 269 270 271

    if (parent) {
        cid_str = "parentCID";
        cid_str_size = sizeof("parentCID");
    } else {
        cid_str = "CID";
        cid_str_size = sizeof("CID");
    }

K
Kevin Wolf 已提交
272
    desc[DESC_SIZE - 1] = '\0';
F
Fam Zheng 已提交
273
    p_name = strstr(desc, cid_str);
274 275 276
    if (p_name == NULL) {
        ret = -EINVAL;
        goto out;
277
    }
278 279 280 281 282 283 284
    p_name += cid_str_size;
    if (sscanf(p_name, "%" SCNx32, &cid) != 1) {
        ret = -EINVAL;
        goto out;
    }
    *pcid = cid;
    ret = 0;
285

286
out:
287
    g_free(desc);
288
    return ret;
289 290 291 292
}

static int vmdk_write_cid(BlockDriverState *bs, uint32_t cid)
{
293
    char *desc, *tmp_desc;
294
    char *p_name, *tmp_str;
295
    BDRVVmdkState *s = bs->opaque;
296
    int ret = 0;
297

298 299
    desc = g_malloc0(DESC_SIZE);
    tmp_desc = g_malloc0(DESC_SIZE);
300
    ret = bdrv_pread(bs->file, s->desc_offset, desc, DESC_SIZE);
K
Kevin Wolf 已提交
301
    if (ret < 0) {
302
        goto out;
303
    }
304

K
Kevin Wolf 已提交
305
    desc[DESC_SIZE - 1] = '\0';
F
Fam Zheng 已提交
306
    tmp_str = strstr(desc, "parentCID");
K
Kevin Wolf 已提交
307
    if (tmp_str == NULL) {
308 309
        ret = -EINVAL;
        goto out;
K
Kevin Wolf 已提交
310 311
    }

312
    pstrcpy(tmp_desc, DESC_SIZE, tmp_str);
F
Fam Zheng 已提交
313 314
    p_name = strstr(desc, "CID");
    if (p_name != NULL) {
315
        p_name += sizeof("CID");
316 317
        snprintf(p_name, DESC_SIZE - (p_name - desc), "%" PRIx32 "\n", cid);
        pstrcat(desc, DESC_SIZE, tmp_desc);
318 319
    }

320
    ret = bdrv_pwrite_sync(bs->file, s->desc_offset, desc, DESC_SIZE);
K
Kevin Wolf 已提交
321

322 323 324 325
out:
    g_free(desc);
    g_free(tmp_desc);
    return ret;
326 327 328 329 330 331 332
}

static int vmdk_is_cid_valid(BlockDriverState *bs)
{
    BDRVVmdkState *s = bs->opaque;
    uint32_t cur_pcid;

333 334 335
    if (!s->cid_checked && bs->backing) {
        BlockDriverState *p_bs = bs->backing->bs;

336 337 338 339 340 341
        if (strcmp(p_bs->drv->format_name, "vmdk")) {
            /* Backing file is not in vmdk format, so it does not have
             * a CID, which makes the overlay's parent CID invalid */
            return 0;
        }

342 343 344 345
        if (vmdk_read_cid(p_bs, 0, &cur_pcid) != 0) {
            /* read failure: report as not valid */
            return 0;
        }
F
Fam Zheng 已提交
346 347
        if (s->parent_cid != cur_pcid) {
            /* CID not valid */
348
            return 0;
F
Fam Zheng 已提交
349
        }
350
    }
351
    s->cid_checked = true;
F
Fam Zheng 已提交
352
    /* CID valid */
353 354 355
    return 1;
}

K
Kevin Wolf 已提交
356
/* We have nothing to do for VMDK reopen, stubs just return success */
J
Jeff Cody 已提交
357 358 359 360 361
static int vmdk_reopen_prepare(BDRVReopenState *state,
                               BlockReopenQueue *queue, Error **errp)
{
    assert(state != NULL);
    assert(state->bs != NULL);
K
Kevin Wolf 已提交
362
    return 0;
J
Jeff Cody 已提交
363 364
}

365
static int vmdk_parent_open(BlockDriverState *bs)
366
{
367
    char *p_name;
368
    char *desc;
369
    BDRVVmdkState *s = bs->opaque;
370
    int ret;
371

372
    desc = g_malloc0(DESC_SIZE + 1);
373
    ret = bdrv_pread(bs->file, s->desc_offset, desc, DESC_SIZE);
374
    if (ret < 0) {
375
        goto out;
376
    }
377
    ret = 0;
378

F
Fam Zheng 已提交
379 380
    p_name = strstr(desc, "parentFileNameHint");
    if (p_name != NULL) {
381 382 383
        char *end_name;

        p_name += sizeof("parentFileNameHint") + 1;
F
Fam Zheng 已提交
384 385
        end_name = strchr(p_name, '\"');
        if (end_name == NULL) {
386 387
            ret = -EINVAL;
            goto out;
F
Fam Zheng 已提交
388 389
        }
        if ((end_name - p_name) > sizeof(bs->backing_file) - 1) {
390 391
            ret = -EINVAL;
            goto out;
F
Fam Zheng 已提交
392
        }
393

K
Kevin Wolf 已提交
394
        pstrcpy(bs->backing_file, end_name - p_name + 1, p_name);
395
    }
396

397 398 399
out:
    g_free(desc);
    return ret;
400 401
}

F
Fam Zheng 已提交
402 403
/* Create and append extent to the extent array. Return the added VmdkExtent
 * address. return NULL if allocation failed. */
404
static int vmdk_add_extent(BlockDriverState *bs,
405
                           BdrvChild *file, bool flat, int64_t sectors,
F
Fam Zheng 已提交
406 407
                           int64_t l1_offset, int64_t l1_backup_offset,
                           uint32_t l1_size,
408
                           int l2_size, uint64_t cluster_sectors,
F
Fam Zheng 已提交
409 410
                           VmdkExtent **new_extent,
                           Error **errp)
F
Fam Zheng 已提交
411 412 413
{
    VmdkExtent *extent;
    BDRVVmdkState *s = bs->opaque;
414
    int64_t nb_sectors;
F
Fam Zheng 已提交
415

416 417
    if (cluster_sectors > 0x200000) {
        /* 0x200000 * 512Bytes = 1GB for one cluster is unrealistic */
F
Fam Zheng 已提交
418 419
        error_setg(errp, "Invalid granularity, image may be corrupt");
        return -EFBIG;
420
    }
421 422 423 424 425
    if (l1_size > 512 * 1024 * 1024) {
        /* Although with big capacity and small l1_entry_sectors, we can get a
         * big l1_size, we don't want unbounded value to allocate the table.
         * Limit it to 512M, which is 16PB for default cluster and L2 table
         * size */
F
Fam Zheng 已提交
426
        error_setg(errp, "L1 size too big");
427 428
        return -EFBIG;
    }
429

430
    nb_sectors = bdrv_nb_sectors(file->bs);
431 432
    if (nb_sectors < 0) {
        return nb_sectors;
F
Fam Zheng 已提交
433 434
    }

435
    s->extents = g_renew(VmdkExtent, s->extents, s->num_extents + 1);
F
Fam Zheng 已提交
436 437 438 439 440 441 442 443 444 445 446 447
    extent = &s->extents[s->num_extents];
    s->num_extents++;

    memset(extent, 0, sizeof(VmdkExtent));
    extent->file = file;
    extent->flat = flat;
    extent->sectors = sectors;
    extent->l1_table_offset = l1_offset;
    extent->l1_backup_table_offset = l1_backup_offset;
    extent->l1_size = l1_size;
    extent->l1_entry_sectors = l2_size * cluster_sectors;
    extent->l2_size = l2_size;
448
    extent->cluster_sectors = flat ? sectors : cluster_sectors;
449
    extent->next_cluster_sector = ROUND_UP(nb_sectors, cluster_sectors);
F
Fam Zheng 已提交
450 451 452 453 454 455 456

    if (s->num_extents > 1) {
        extent->end_sector = (*(extent - 1)).end_sector + extent->sectors;
    } else {
        extent->end_sector = extent->sectors;
    }
    bs->total_sectors = extent->end_sector;
457 458 459 460
    if (new_extent) {
        *new_extent = extent;
    }
    return 0;
F
Fam Zheng 已提交
461 462
}

F
Fam Zheng 已提交
463 464
static int vmdk_init_tables(BlockDriverState *bs, VmdkExtent *extent,
                            Error **errp)
465
{
466
    int ret;
467 468
    size_t l1_size;
    int i;
469

B
bellard 已提交
470
    /* read the L1 table */
F
Fam Zheng 已提交
471
    l1_size = extent->l1_size * sizeof(uint32_t);
472 473 474 475 476
    extent->l1_table = g_try_malloc(l1_size);
    if (l1_size && extent->l1_table == NULL) {
        return -ENOMEM;
    }

477
    ret = bdrv_pread(extent->file,
F
Fam Zheng 已提交
478 479 480
                     extent->l1_table_offset,
                     extent->l1_table,
                     l1_size);
481
    if (ret < 0) {
F
Fam Zheng 已提交
482 483
        error_setg_errno(errp, -ret,
                         "Could not read l1 table from extent '%s'",
484
                         extent->file->bs->filename);
485
        goto fail_l1;
F
Fam Zheng 已提交
486 487 488
    }
    for (i = 0; i < extent->l1_size; i++) {
        le32_to_cpus(&extent->l1_table[i]);
B
bellard 已提交
489 490
    }

F
Fam Zheng 已提交
491
    if (extent->l1_backup_table_offset) {
492 493 494 495 496
        extent->l1_backup_table = g_try_malloc(l1_size);
        if (l1_size && extent->l1_backup_table == NULL) {
            ret = -ENOMEM;
            goto fail_l1;
        }
497
        ret = bdrv_pread(extent->file,
F
Fam Zheng 已提交
498 499 500
                         extent->l1_backup_table_offset,
                         extent->l1_backup_table,
                         l1_size);
501
        if (ret < 0) {
F
Fam Zheng 已提交
502 503
            error_setg_errno(errp, -ret,
                             "Could not read l1 backup table from extent '%s'",
504
                             extent->file->bs->filename);
505
            goto fail_l1b;
F
Fam Zheng 已提交
506 507 508
        }
        for (i = 0; i < extent->l1_size; i++) {
            le32_to_cpus(&extent->l1_backup_table[i]);
509 510 511
        }
    }

F
Fam Zheng 已提交
512
    extent->l2_cache =
513
        g_new(uint32_t, extent->l2_size * L2_CACHE_SIZE);
B
bellard 已提交
514
    return 0;
515
 fail_l1b:
516
    g_free(extent->l1_backup_table);
517
 fail_l1:
518
    g_free(extent->l1_table);
519 520 521
    return ret;
}

F
Fam Zheng 已提交
522
static int vmdk_open_vmfs_sparse(BlockDriverState *bs,
523
                                 BdrvChild *file,
F
Fam Zheng 已提交
524
                                 int flags, Error **errp)
525 526 527 528 529 530
{
    int ret;
    uint32_t magic;
    VMDK3Header header;
    VmdkExtent *extent;

531
    ret = bdrv_pread(file, sizeof(magic), &header, sizeof(header));
532
    if (ret < 0) {
F
Fam Zheng 已提交
533 534
        error_setg_errno(errp, -ret,
                         "Could not read header from file '%s'",
535
                         file->bs->filename);
536
        return ret;
537
    }
538 539
    ret = vmdk_add_extent(bs, file, false,
                          le32_to_cpu(header.disk_sectors),
540
                          (int64_t)le32_to_cpu(header.l1dir_offset) << 9,
541 542 543 544
                          0,
                          le32_to_cpu(header.l1dir_size),
                          4096,
                          le32_to_cpu(header.granularity),
F
Fam Zheng 已提交
545 546
                          &extent,
                          errp);
547 548 549
    if (ret < 0) {
        return ret;
    }
F
Fam Zheng 已提交
550
    ret = vmdk_init_tables(bs, extent, errp);
551
    if (ret) {
552 553
        /* free extent allocated by vmdk_add_extent */
        vmdk_free_last_extent(bs);
554 555 556 557
    }
    return ret;
}

558
static int vmdk_open_desc_file(BlockDriverState *bs, int flags, char *buf,
K
Kevin Wolf 已提交
559
                               QDict *options, Error **errp);
F
Fam Zheng 已提交
560

561
static char *vmdk_read_desc(BdrvChild *file, uint64_t desc_offset, Error **errp)
P
Paolo Bonzini 已提交
562 563 564 565 566
{
    int64_t size;
    char *buf;
    int ret;

567
    size = bdrv_getlength(file->bs);
P
Paolo Bonzini 已提交
568 569 570 571 572
    if (size < 0) {
        error_setg_errno(errp, -size, "Could not access file");
        return NULL;
    }

573 574 575 576 577 578 579 580
    if (size < 4) {
        /* Both descriptor file and sparse image must be much larger than 4
         * bytes, also callers of vmdk_read_desc want to compare the first 4
         * bytes with VMDK4_MAGIC, let's error out if less is read. */
        error_setg(errp, "File is too small, not a valid image");
        return NULL;
    }

F
Fam Zheng 已提交
581 582
    size = MIN(size, (1 << 20) - 1);  /* avoid unbounded allocation */
    buf = g_malloc(size + 1);
P
Paolo Bonzini 已提交
583 584 585 586 587 588 589

    ret = bdrv_pread(file, desc_offset, buf, size);
    if (ret < 0) {
        error_setg_errno(errp, -ret, "Could not read from file");
        g_free(buf);
        return NULL;
    }
F
Fam Zheng 已提交
590
    buf[ret] = 0;
P
Paolo Bonzini 已提交
591 592 593 594

    return buf;
}

595
static int vmdk_open_vmdk4(BlockDriverState *bs,
596
                           BdrvChild *file,
K
Kevin Wolf 已提交
597
                           int flags, QDict *options, Error **errp)
598 599 600 601 602 603
{
    int ret;
    uint32_t magic;
    uint32_t l1_size, l1_entry_sectors;
    VMDK4Header header;
    VmdkExtent *extent;
F
Fam Zheng 已提交
604
    BDRVVmdkState *s = bs->opaque;
605
    int64_t l1_backup_offset = 0;
606
    bool compressed;
607

608
    ret = bdrv_pread(file, sizeof(magic), &header, sizeof(header));
609
    if (ret < 0) {
F
Fam Zheng 已提交
610 611
        error_setg_errno(errp, -ret,
                         "Could not read header from file '%s'",
612
                         file->bs->filename);
P
Paolo Bonzini 已提交
613
        return -EINVAL;
614
    }
615
    if (header.capacity == 0) {
616
        uint64_t desc_offset = le64_to_cpu(header.desc_offset);
617
        if (desc_offset) {
618
            char *buf = vmdk_read_desc(file, desc_offset << 9, errp);
619 620 621
            if (!buf) {
                return -EINVAL;
            }
K
Kevin Wolf 已提交
622
            ret = vmdk_open_desc_file(bs, flags, buf, options, errp);
623 624
            g_free(buf);
            return ret;
625
        }
F
Fam Zheng 已提交
626
    }
627

F
Fam Zheng 已提交
628 629 630 631
    if (!s->create_type) {
        s->create_type = g_strdup("monolithicSparse");
    }

632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657
    if (le64_to_cpu(header.gd_offset) == VMDK4_GD_AT_END) {
        /*
         * The footer takes precedence over the header, so read it in. The
         * footer starts at offset -1024 from the end: One sector for the
         * footer, and another one for the end-of-stream marker.
         */
        struct {
            struct {
                uint64_t val;
                uint32_t size;
                uint32_t type;
                uint8_t pad[512 - 16];
            } QEMU_PACKED footer_marker;

            uint32_t magic;
            VMDK4Header header;
            uint8_t pad[512 - 4 - sizeof(VMDK4Header)];

            struct {
                uint64_t val;
                uint32_t size;
                uint32_t type;
                uint8_t pad[512 - 16];
            } QEMU_PACKED eos_marker;
        } QEMU_PACKED footer;

658
        ret = bdrv_pread(file,
K
Kevin Wolf 已提交
659
            bs->file->bs->total_sectors * 512 - 1536,
660 661
            &footer, sizeof(footer));
        if (ret < 0) {
662
            error_setg_errno(errp, -ret, "Failed to read footer");
663 664 665 666 667 668 669 670 671 672 673
            return ret;
        }

        /* Some sanity checks for the footer */
        if (be32_to_cpu(footer.magic) != VMDK4_MAGIC ||
            le32_to_cpu(footer.footer_marker.size) != 0  ||
            le32_to_cpu(footer.footer_marker.type) != MARKER_FOOTER ||
            le64_to_cpu(footer.eos_marker.val) != 0  ||
            le32_to_cpu(footer.eos_marker.size) != 0  ||
            le32_to_cpu(footer.eos_marker.type) != MARKER_END_OF_STREAM)
        {
674
            error_setg(errp, "Invalid footer");
675 676 677 678 679 680
            return -EINVAL;
        }

        header = footer.header;
    }

681 682
    compressed =
        le16_to_cpu(header.compressAlgorithm) == VMDK4_COMPRESSION_DEFLATE;
683
    if (le32_to_cpu(header.version) > 3) {
684 685
        error_setg(errp, "Unsupported VMDK version %" PRIu32,
                   le32_to_cpu(header.version));
686
        return -ENOTSUP;
687 688
    } else if (le32_to_cpu(header.version) == 3 && (flags & BDRV_O_RDWR) &&
               !compressed) {
689 690 691 692 693 694
        /* VMware KB 2064959 explains that version 3 added support for
         * persistent changed block tracking (CBT), and backup software can
         * read it as version=1 if it doesn't care about the changed area
         * information. So we are safe to enable read only. */
        error_setg(errp, "VMDK version 3 must be read only");
        return -EINVAL;
695 696
    }

697
    if (le32_to_cpu(header.num_gtes_per_gt) > 512) {
P
Paolo Bonzini 已提交
698
        error_setg(errp, "L2 table size too big");
699 700 701
        return -EINVAL;
    }

702
    l1_entry_sectors = le32_to_cpu(header.num_gtes_per_gt)
703
                        * le64_to_cpu(header.granularity);
704
    if (l1_entry_sectors == 0) {
705
        error_setg(errp, "L1 entry size is invalid");
706 707
        return -EINVAL;
    }
708 709
    l1_size = (le64_to_cpu(header.capacity) + l1_entry_sectors - 1)
                / l1_entry_sectors;
710 711 712
    if (le32_to_cpu(header.flags) & VMDK4_FLAG_RGD) {
        l1_backup_offset = le64_to_cpu(header.rgd_offset) << 9;
    }
713
    if (bdrv_nb_sectors(file->bs) < le64_to_cpu(header.grain_offset)) {
714 715 716
        error_setg(errp, "File truncated, expecting at least %" PRId64 " bytes",
                   (int64_t)(le64_to_cpu(header.grain_offset)
                             * BDRV_SECTOR_SIZE));
717 718 719
        return -EINVAL;
    }

720
    ret = vmdk_add_extent(bs, file, false,
721 722
                          le64_to_cpu(header.capacity),
                          le64_to_cpu(header.gd_offset) << 9,
723
                          l1_backup_offset,
724
                          l1_size,
725
                          le32_to_cpu(header.num_gtes_per_gt),
726
                          le64_to_cpu(header.granularity),
F
Fam Zheng 已提交
727 728
                          &extent,
                          errp);
729 730 731
    if (ret < 0) {
        return ret;
    }
F
Fam Zheng 已提交
732 733
    extent->compressed =
        le16_to_cpu(header.compressAlgorithm) == VMDK4_COMPRESSION_DEFLATE;
734 735 736 737
    if (extent->compressed) {
        g_free(s->create_type);
        s->create_type = g_strdup("streamOptimized");
    }
F
Fam Zheng 已提交
738
    extent->has_marker = le32_to_cpu(header.flags) & VMDK4_FLAG_MARKER;
739 740
    extent->version = le32_to_cpu(header.version);
    extent->has_zero_grain = le32_to_cpu(header.flags) & VMDK4_FLAG_ZERO_GRAIN;
F
Fam Zheng 已提交
741
    ret = vmdk_init_tables(bs, extent, errp);
742
    if (ret) {
743 744
        /* free extent allocated by vmdk_add_extent */
        vmdk_free_last_extent(bs);
745 746 747 748
    }
    return ret;
}

749 750 751 752 753 754 755 756 757
/* find an option value out of descriptor file */
static int vmdk_parse_description(const char *desc, const char *opt_name,
        char *buf, int buf_size)
{
    char *opt_pos, *opt_end;
    const char *end = desc + strlen(desc);

    opt_pos = strstr(desc, opt_name);
    if (!opt_pos) {
F
Fam Zheng 已提交
758
        return VMDK_ERROR;
759 760 761 762
    }
    /* Skip "=\"" following opt_name */
    opt_pos += strlen(opt_name) + 2;
    if (opt_pos >= end) {
F
Fam Zheng 已提交
763
        return VMDK_ERROR;
764 765 766 767 768 769
    }
    opt_end = opt_pos;
    while (opt_end < end && *opt_end != '"') {
        opt_end++;
    }
    if (opt_end == end || buf_size < opt_end - opt_pos + 1) {
F
Fam Zheng 已提交
770
        return VMDK_ERROR;
771 772
    }
    pstrcpy(buf, opt_end - opt_pos + 1, opt_pos);
F
Fam Zheng 已提交
773
    return VMDK_OK;
774 775
}

776
/* Open an extent file and append to bs array */
777
static int vmdk_open_sparse(BlockDriverState *bs, BdrvChild *file, int flags,
K
Kevin Wolf 已提交
778
                            char *buf, QDict *options, Error **errp)
779 780 781
{
    uint32_t magic;

782
    magic = ldl_be_p(buf);
783 784
    switch (magic) {
        case VMDK3_MAGIC:
F
Fam Zheng 已提交
785
            return vmdk_open_vmfs_sparse(bs, file, flags, errp);
786 787
            break;
        case VMDK4_MAGIC:
K
Kevin Wolf 已提交
788
            return vmdk_open_vmdk4(bs, file, flags, options, errp);
789 790
            break;
        default:
P
Paolo Bonzini 已提交
791 792
            error_setg(errp, "Image not in VMDK format");
            return -EINVAL;
793 794 795 796
            break;
    }
}

797 798 799 800 801 802 803 804 805 806 807
static const char *next_line(const char *s)
{
    while (*s) {
        if (*s == '\n') {
            return s + 1;
        }
        s++;
    }
    return s;
}

808
static int vmdk_parse_extents(const char *desc, BlockDriverState *bs,
K
Kevin Wolf 已提交
809 810
                              const char *desc_file_path, QDict *options,
                              Error **errp)
811 812
{
    int ret;
813
    int matches;
814 815 816
    char access[11];
    char type[11];
    char fname[512];
817
    const char *p, *np;
818 819
    int64_t sectors = 0;
    int64_t flat_offset;
820
    char *extent_path;
821
    BdrvChild *extent_file;
F
Fam Zheng 已提交
822 823
    BDRVVmdkState *s = bs->opaque;
    VmdkExtent *extent;
K
Kevin Wolf 已提交
824
    char extent_opt_prefix[32];
825
    Error *local_err = NULL;
826

827
    for (p = desc; *p; p = next_line(p)) {
828 829
        /* parse extent line in one of below formats:
         *
830 831
         * RW [size in sectors] FLAT "file-name.vmdk" OFFSET
         * RW [size in sectors] SPARSE "file-name.vmdk"
832 833
         * RW [size in sectors] VMFS "file-name.vmdk"
         * RW [size in sectors] VMFSSPARSE "file-name.vmdk"
834 835
         */
        flat_offset = -1;
836 837 838
        matches = sscanf(p, "%10s %" SCNd64 " %10s \"%511[^\n\r\"]\" %" SCNd64,
                         access, &sectors, type, fname, &flat_offset);
        if (matches < 4 || strcmp(access, "RW")) {
839
            continue;
840
        } else if (!strcmp(type, "FLAT")) {
841
            if (matches != 5 || flat_offset < 0) {
842
                goto invalid;
843
            }
F
Fam Zheng 已提交
844
        } else if (!strcmp(type, "VMFS")) {
845
            if (matches == 4) {
846 847
                flat_offset = 0;
            } else {
848
                goto invalid;
849
            }
850
        } else if (matches != 4) {
851
            goto invalid;
852 853 854
        }

        if (sectors <= 0 ||
F
Fam Zheng 已提交
855
            (strcmp(type, "FLAT") && strcmp(type, "SPARSE") &&
P
Paolo Bonzini 已提交
856
             strcmp(type, "VMFS") && strcmp(type, "VMFSSPARSE")) ||
857
            (strcmp(access, "RW"))) {
858
            continue;
859 860
        }

861 862 863 864
        if (!path_is_absolute(fname) && !path_has_protocol(fname) &&
            !desc_file_path[0])
        {
            error_setg(errp, "Cannot use relative extent paths with VMDK "
K
Kevin Wolf 已提交
865
                       "descriptor file '%s'", bs->file->bs->filename);
866 867 868
            return -EINVAL;
        }

869
        extent_path = g_malloc0(PATH_MAX);
J
Jeff Cody 已提交
870
        path_combine(extent_path, PATH_MAX, desc_file_path, fname);
K
Kevin Wolf 已提交
871 872 873 874

        ret = snprintf(extent_opt_prefix, 32, "extents.%d", s->num_extents);
        assert(ret < 32);

875 876
        extent_file = bdrv_open_child(extent_path, options, extent_opt_prefix,
                                      bs, &child_file, false, &local_err);
877
        g_free(extent_path);
878 879 880
        if (local_err) {
            error_propagate(errp, local_err);
            return -EINVAL;
881 882
        }

883
        /* save to extents array */
P
Paolo Bonzini 已提交
884
        if (!strcmp(type, "FLAT") || !strcmp(type, "VMFS")) {
885 886
            /* FLAT extent */

887
            ret = vmdk_add_extent(bs, extent_file, true, sectors,
F
Fam Zheng 已提交
888
                            0, 0, 0, 0, 0, &extent, errp);
889
            if (ret < 0) {
890
                bdrv_unref_child(bs, extent_file);
891 892
                return ret;
            }
F
Fam Zheng 已提交
893
            extent->flat_start_offset = flat_offset << 9;
F
Fam Zheng 已提交
894 895
        } else if (!strcmp(type, "SPARSE") || !strcmp(type, "VMFSSPARSE")) {
            /* SPARSE extent and VMFSSPARSE extent are both "COWD" sparse file*/
896
            char *buf = vmdk_read_desc(extent_file, 0, errp);
897 898 899
            if (!buf) {
                ret = -EINVAL;
            } else {
K
Kevin Wolf 已提交
900 901
                ret = vmdk_open_sparse(bs, extent_file, bs->open_flags, buf,
                                       options, errp);
902
            }
903
            g_free(buf);
904
            if (ret) {
905
                bdrv_unref_child(bs, extent_file);
906 907
                return ret;
            }
F
Fam Zheng 已提交
908
            extent = &s->extents[s->num_extents - 1];
909
        } else {
F
Fam Zheng 已提交
910
            error_setg(errp, "Unsupported extent type '%s'", type);
911
            bdrv_unref_child(bs, extent_file);
912 913
            return -ENOTSUP;
        }
F
Fam Zheng 已提交
914
        extent->type = g_strdup(type);
915 916
    }
    return 0;
917 918 919 920 921 922 923 924 925

invalid:
    np = next_line(p);
    assert(np != p);
    if (np[-1] == '\n') {
        np--;
    }
    error_setg(errp, "Invalid extent line: %.*s", (int)(np - p), p);
    return -EINVAL;
926 927
}

928
static int vmdk_open_desc_file(BlockDriverState *bs, int flags, char *buf,
K
Kevin Wolf 已提交
929
                               QDict *options, Error **errp)
930 931 932 933 934 935
{
    int ret;
    char ct[128];
    BDRVVmdkState *s = bs->opaque;

    if (vmdk_parse_description(buf, "createType", ct, sizeof(ct))) {
P
Paolo Bonzini 已提交
936 937
        error_setg(errp, "invalid VMDK image descriptor");
        ret = -EINVAL;
938
        goto exit;
939
    }
F
Fam Zheng 已提交
940
    if (strcmp(ct, "monolithicFlat") &&
P
Paolo Bonzini 已提交
941
        strcmp(ct, "vmfs") &&
F
Fam Zheng 已提交
942
        strcmp(ct, "vmfsSparse") &&
943
        strcmp(ct, "twoGbMaxExtentSparse") &&
F
Fam Zheng 已提交
944
        strcmp(ct, "twoGbMaxExtentFlat")) {
F
Fam Zheng 已提交
945
        error_setg(errp, "Unsupported image type '%s'", ct);
946 947
        ret = -ENOTSUP;
        goto exit;
948
    }
F
Fam Zheng 已提交
949
    s->create_type = g_strdup(ct);
950
    s->desc_offset = 0;
K
Kevin Wolf 已提交
951 952
    ret = vmdk_parse_extents(buf, bs, bs->file->bs->exact_filename, options,
                             errp);
953 954
exit:
    return ret;
955 956
}

M
Max Reitz 已提交
957 958
static int vmdk_open(BlockDriverState *bs, QDict *options, int flags,
                     Error **errp)
959
{
960
    char *buf;
961 962
    int ret;
    BDRVVmdkState *s = bs->opaque;
963
    uint32_t magic;
964
    Error *local_err = NULL;
965

966 967 968 969 970 971
    bs->file = bdrv_open_child(NULL, options, "file", bs, &child_file,
                               false, errp);
    if (!bs->file) {
        return -EINVAL;
    }

972
    buf = vmdk_read_desc(bs->file, 0, errp);
973 974 975 976
    if (!buf) {
        return -EINVAL;
    }

977 978 979 980
    magic = ldl_be_p(buf);
    switch (magic) {
        case VMDK3_MAGIC:
        case VMDK4_MAGIC:
K
Kevin Wolf 已提交
981
            ret = vmdk_open_sparse(bs, bs->file, flags, buf, options,
982
                                   errp);
983 984 985
            s->desc_offset = 0x200;
            break;
        default:
K
Kevin Wolf 已提交
986
            ret = vmdk_open_desc_file(bs, flags, buf, options, errp);
987
            break;
988
    }
989 990 991 992
    if (ret) {
        goto fail;
    }

P
Paolo Bonzini 已提交
993 994 995 996 997
    /* try to open parent images, if exist */
    ret = vmdk_parent_open(bs);
    if (ret) {
        goto fail;
    }
998 999 1000 1001 1002 1003 1004 1005
    ret = vmdk_read_cid(bs, 0, &s->cid);
    if (ret) {
        goto fail;
    }
    ret = vmdk_read_cid(bs, 1, &s->parent_cid);
    if (ret) {
        goto fail;
    }
1006
    qemu_co_mutex_init(&s->lock);
K
Kevin Wolf 已提交
1007 1008

    /* Disable migration when VMDK images are used */
1009 1010 1011
    error_setg(&s->migration_blocker, "The vmdk format used by node '%s' "
               "does not support live migration",
               bdrv_get_device_or_node_name(bs));
1012 1013 1014 1015 1016 1017 1018
    ret = migrate_add_blocker(s->migration_blocker, &local_err);
    if (local_err) {
        error_propagate(errp, local_err);
        error_free(s->migration_blocker);
        goto fail;
    }

1019
    g_free(buf);
K
Kevin Wolf 已提交
1020
    return 0;
P
Paolo Bonzini 已提交
1021 1022

fail:
1023
    g_free(buf);
F
Fam Zheng 已提交
1024 1025
    g_free(s->create_type);
    s->create_type = NULL;
P
Paolo Bonzini 已提交
1026 1027
    vmdk_free_extents(bs);
    return ret;
B
bellard 已提交
1028 1029
}

1030

1031
static void vmdk_refresh_limits(BlockDriverState *bs, Error **errp)
1032 1033 1034 1035 1036 1037
{
    BDRVVmdkState *s = bs->opaque;
    int i;

    for (i = 0; i < s->num_extents; i++) {
        if (!s->extents[i].flat) {
1038 1039 1040
            bs->bl.pwrite_zeroes_alignment =
                MAX(bs->bl.pwrite_zeroes_alignment,
                    s->extents[i].cluster_sectors << BDRV_SECTOR_BITS);
1041 1042 1043 1044
        }
    }
}

F
Fam Zheng 已提交
1045 1046 1047 1048 1049 1050 1051 1052 1053 1054
/**
 * get_whole_cluster
 *
 * Copy backing file's cluster that covers @sector_num, otherwise write zero,
 * to the cluster at @cluster_sector_num.
 *
 * If @skip_start_sector < @skip_end_sector, the relative range
 * [@skip_start_sector, @skip_end_sector) is not copied or written, and leave
 * it for call to write user data in the request.
 */
F
Fam Zheng 已提交
1055
static int get_whole_cluster(BlockDriverState *bs,
F
Fam Zheng 已提交
1056
                             VmdkExtent *extent,
1057 1058 1059 1060
                             uint64_t cluster_offset,
                             uint64_t offset,
                             uint64_t skip_start_bytes,
                             uint64_t skip_end_bytes)
1061
{
1062
    int ret = VMDK_OK;
F
Fam Zheng 已提交
1063 1064 1065 1066 1067
    int64_t cluster_bytes;
    uint8_t *whole_grain;

    /* For COW, align request sector_num to cluster start */
    cluster_bytes = extent->cluster_sectors << BDRV_SECTOR_BITS;
1068
    offset = QEMU_ALIGN_DOWN(offset, cluster_bytes);
F
Fam Zheng 已提交
1069
    whole_grain = qemu_blockalign(bs, cluster_bytes);
1070

1071
    if (!bs->backing) {
1072 1073
        memset(whole_grain, 0, skip_start_bytes);
        memset(whole_grain + skip_end_bytes, 0, cluster_bytes - skip_end_bytes);
F
Fam Zheng 已提交
1074 1075
    }

1076
    assert(skip_end_bytes <= cluster_bytes);
1077 1078
    /* we will be here if it's first write on non-exist grain(cluster).
     * try to read from parent image, if exist */
1079
    if (bs->backing && !vmdk_is_cid_valid(bs)) {
F
Fam Zheng 已提交
1080 1081 1082
        ret = VMDK_ERROR;
        goto exit;
    }
1083

F
Fam Zheng 已提交
1084
    /* Read backing data before skip range */
1085
    if (skip_start_bytes > 0) {
1086
        if (bs->backing) {
M
Max Reitz 已提交
1087 1088
            /* qcow2 emits this on bs->file instead of bs->backing */
            BLKDBG_EVENT(extent->file, BLKDBG_COW_READ);
1089
            ret = bdrv_pread(bs->backing, offset, whole_grain,
1090
                             skip_start_bytes);
F
Fam Zheng 已提交
1091 1092 1093 1094 1095
            if (ret < 0) {
                ret = VMDK_ERROR;
                goto exit;
            }
        }
M
Max Reitz 已提交
1096
        BLKDBG_EVENT(extent->file, BLKDBG_COW_WRITE);
1097
        ret = bdrv_pwrite(extent->file, cluster_offset, whole_grain,
1098
                          skip_start_bytes);
K
Kevin Wolf 已提交
1099
        if (ret < 0) {
1100 1101
            ret = VMDK_ERROR;
            goto exit;
K
Kevin Wolf 已提交
1102
        }
F
Fam Zheng 已提交
1103 1104
    }
    /* Read backing data after skip range */
1105
    if (skip_end_bytes < cluster_bytes) {
1106
        if (bs->backing) {
M
Max Reitz 已提交
1107 1108
            /* qcow2 emits this on bs->file instead of bs->backing */
            BLKDBG_EVENT(extent->file, BLKDBG_COW_READ);
1109
            ret = bdrv_pread(bs->backing, offset + skip_end_bytes,
1110 1111
                             whole_grain + skip_end_bytes,
                             cluster_bytes - skip_end_bytes);
F
Fam Zheng 已提交
1112 1113 1114 1115 1116
            if (ret < 0) {
                ret = VMDK_ERROR;
                goto exit;
            }
        }
M
Max Reitz 已提交
1117
        BLKDBG_EVENT(extent->file, BLKDBG_COW_WRITE);
1118
        ret = bdrv_pwrite(extent->file, cluster_offset + skip_end_bytes,
1119 1120
                          whole_grain + skip_end_bytes,
                          cluster_bytes - skip_end_bytes);
K
Kevin Wolf 已提交
1121
        if (ret < 0) {
1122 1123
            ret = VMDK_ERROR;
            goto exit;
1124 1125
        }
    }
F
Fam Zheng 已提交
1126

1127
    ret = VMDK_OK;
1128 1129 1130
exit:
    qemu_vfree(whole_grain);
    return ret;
1131 1132
}

F
Fam Zheng 已提交
1133 1134
static int vmdk_L2update(VmdkExtent *extent, VmdkMetaData *m_data,
                         uint32_t offset)
1135
{
F
Fam Zheng 已提交
1136
    offset = cpu_to_le32(offset);
1137
    /* update L2 table */
M
Max Reitz 已提交
1138
    BLKDBG_EVENT(extent->file, BLKDBG_L2_UPDATE);
1139
    if (bdrv_pwrite_sync(extent->file,
F
Fam Zheng 已提交
1140
                ((int64_t)m_data->l2_offset * 512)
F
Fam Zheng 已提交
1141
                    + (m_data->l2_index * sizeof(offset)),
1142
                &offset, sizeof(offset)) < 0) {
F
Fam Zheng 已提交
1143
        return VMDK_ERROR;
F
Fam Zheng 已提交
1144
    }
1145
    /* update backup L2 table */
F
Fam Zheng 已提交
1146 1147
    if (extent->l1_backup_table_offset != 0) {
        m_data->l2_offset = extent->l1_backup_table[m_data->l1_index];
1148
        if (bdrv_pwrite_sync(extent->file,
F
Fam Zheng 已提交
1149
                    ((int64_t)m_data->l2_offset * 512)
F
Fam Zheng 已提交
1150
                        + (m_data->l2_index * sizeof(offset)),
1151
                    &offset, sizeof(offset)) < 0) {
F
Fam Zheng 已提交
1152
            return VMDK_ERROR;
F
Fam Zheng 已提交
1153
        }
1154
    }
F
Fam Zheng 已提交
1155 1156 1157
    if (m_data->l2_cache_entry) {
        *m_data->l2_cache_entry = offset;
    }
1158

F
Fam Zheng 已提交
1159
    return VMDK_OK;
1160 1161
}

F
Fam Zheng 已提交
1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181
/**
 * get_cluster_offset
 *
 * Look up cluster offset in extent file by sector number, and store in
 * @cluster_offset.
 *
 * For flat extents, the start offset as parsed from the description file is
 * returned.
 *
 * For sparse extents, look up in L1, L2 table. If allocate is true, return an
 * offset for a new cluster and update L2 cache. If there is a backing file,
 * COW is done before returning; otherwise, zeroes are written to the allocated
 * cluster. Both COW and zero writing skips the sector range
 * [@skip_start_sector, @skip_end_sector) passed in by caller, because caller
 * has new data to write there.
 *
 * Returns: VMDK_OK if cluster exists and mapped in the image.
 *          VMDK_UNALLOC if cluster is not mapped and @allocate is false.
 *          VMDK_ERROR if failed.
 */
1182
static int get_cluster_offset(BlockDriverState *bs,
F
Fam Zheng 已提交
1183 1184 1185 1186 1187
                              VmdkExtent *extent,
                              VmdkMetaData *m_data,
                              uint64_t offset,
                              bool allocate,
                              uint64_t *cluster_offset,
1188 1189
                              uint64_t skip_start_bytes,
                              uint64_t skip_end_bytes)
B
bellard 已提交
1190 1191 1192
{
    unsigned int l1_index, l2_offset, l2_index;
    int min_index, i, j;
1193
    uint32_t min_count, *l2_table;
1194
    bool zeroed = false;
F
Fam Zheng 已提交
1195
    int64_t ret;
1196
    int64_t cluster_sector;
1197

F
Fam Zheng 已提交
1198
    if (m_data) {
1199
        m_data->valid = 0;
F
Fam Zheng 已提交
1200
    }
1201
    if (extent->flat) {
1202
        *cluster_offset = extent->flat_start_offset;
F
Fam Zheng 已提交
1203
        return VMDK_OK;
1204
    }
1205

F
Fam Zheng 已提交
1206
    offset -= (extent->end_sector - extent->sectors) * SECTOR_SIZE;
F
Fam Zheng 已提交
1207 1208
    l1_index = (offset >> 9) / extent->l1_entry_sectors;
    if (l1_index >= extent->l1_size) {
F
Fam Zheng 已提交
1209
        return VMDK_ERROR;
F
Fam Zheng 已提交
1210 1211 1212
    }
    l2_offset = extent->l1_table[l1_index];
    if (!l2_offset) {
F
Fam Zheng 已提交
1213
        return VMDK_UNALLOC;
F
Fam Zheng 已提交
1214
    }
1215
    for (i = 0; i < L2_CACHE_SIZE; i++) {
F
Fam Zheng 已提交
1216
        if (l2_offset == extent->l2_cache_offsets[i]) {
B
bellard 已提交
1217
            /* increment the hit count */
F
Fam Zheng 已提交
1218
            if (++extent->l2_cache_counts[i] == 0xffffffff) {
1219
                for (j = 0; j < L2_CACHE_SIZE; j++) {
F
Fam Zheng 已提交
1220
                    extent->l2_cache_counts[j] >>= 1;
B
bellard 已提交
1221 1222
                }
            }
F
Fam Zheng 已提交
1223
            l2_table = extent->l2_cache + (i * extent->l2_size);
B
bellard 已提交
1224 1225 1226 1227 1228 1229
            goto found;
        }
    }
    /* not found: load a new entry in the least used one */
    min_index = 0;
    min_count = 0xffffffff;
1230
    for (i = 0; i < L2_CACHE_SIZE; i++) {
F
Fam Zheng 已提交
1231 1232
        if (extent->l2_cache_counts[i] < min_count) {
            min_count = extent->l2_cache_counts[i];
B
bellard 已提交
1233 1234 1235
            min_index = i;
        }
    }
F
Fam Zheng 已提交
1236
    l2_table = extent->l2_cache + (min_index * extent->l2_size);
M
Max Reitz 已提交
1237
    BLKDBG_EVENT(extent->file, BLKDBG_L2_LOAD);
1238
    if (bdrv_pread(extent->file,
F
Fam Zheng 已提交
1239 1240 1241 1242
                (int64_t)l2_offset * 512,
                l2_table,
                extent->l2_size * sizeof(uint32_t)
            ) != extent->l2_size * sizeof(uint32_t)) {
F
Fam Zheng 已提交
1243
        return VMDK_ERROR;
F
Fam Zheng 已提交
1244
    }
1245

F
Fam Zheng 已提交
1246 1247
    extent->l2_cache_offsets[min_index] = l2_offset;
    extent->l2_cache_counts[min_index] = 1;
B
bellard 已提交
1248
 found:
F
Fam Zheng 已提交
1249
    l2_index = ((offset >> 9) / extent->cluster_sectors) % extent->l2_size;
F
Fam Zheng 已提交
1250
    cluster_sector = le32_to_cpu(l2_table[l2_index]);
1251

F
Fam Zheng 已提交
1252
    if (extent->has_zero_grain && cluster_sector == VMDK_GTE_ZEROED) {
1253 1254 1255
        zeroed = true;
    }

F
Fam Zheng 已提交
1256
    if (!cluster_sector || zeroed) {
1257
        if (!allocate) {
1258
            return zeroed ? VMDK_ZEROED : VMDK_UNALLOC;
1259
        }
1260

1261 1262 1263 1264
        if (extent->next_cluster_sector >= VMDK_EXTENT_MAX_SECTORS) {
            return VMDK_ERROR;
        }

F
Fam Zheng 已提交
1265 1266
        cluster_sector = extent->next_cluster_sector;
        extent->next_cluster_sector += extent->cluster_sectors;
1267 1268 1269 1270 1271 1272

        /* First of all we write grain itself, to avoid race condition
         * that may to corrupt the image.
         * This problem may occur because of insufficient space on host disk
         * or inappropriate VM shutdown.
         */
1273 1274
        ret = get_whole_cluster(bs, extent, cluster_sector * BDRV_SECTOR_SIZE,
                                offset, skip_start_bytes, skip_end_bytes);
F
Fam Zheng 已提交
1275 1276
        if (ret) {
            return ret;
1277
        }
1278 1279 1280 1281 1282 1283 1284
        if (m_data) {
            m_data->valid = 1;
            m_data->l1_index = l1_index;
            m_data->l2_index = l2_index;
            m_data->l2_offset = l2_offset;
            m_data->l2_cache_entry = &l2_table[l2_index];
        }
1285
    }
F
Fam Zheng 已提交
1286
    *cluster_offset = cluster_sector << BDRV_SECTOR_BITS;
F
Fam Zheng 已提交
1287
    return VMDK_OK;
B
bellard 已提交
1288 1289
}

F
Fam Zheng 已提交
1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306
static VmdkExtent *find_extent(BDRVVmdkState *s,
                                int64_t sector_num, VmdkExtent *start_hint)
{
    VmdkExtent *extent = start_hint;

    if (!extent) {
        extent = &s->extents[0];
    }
    while (extent < &s->extents[s->num_extents]) {
        if (sector_num < extent->end_sector) {
            return extent;
        }
        extent++;
    }
    return NULL;
}

1307 1308 1309
static inline uint64_t vmdk_find_offset_in_cluster(VmdkExtent *extent,
                                                   int64_t offset)
{
1310
    uint64_t extent_begin_offset, extent_relative_offset;
1311 1312 1313 1314 1315
    uint64_t cluster_size = extent->cluster_sectors * BDRV_SECTOR_SIZE;

    extent_begin_offset =
        (extent->end_sector - extent->sectors) * BDRV_SECTOR_SIZE;
    extent_relative_offset = offset - extent_begin_offset;
1316
    return extent_relative_offset % cluster_size;
1317 1318
}

1319 1320 1321 1322 1323
static int coroutine_fn vmdk_co_block_status(BlockDriverState *bs,
                                             bool want_zero,
                                             int64_t offset, int64_t bytes,
                                             int64_t *pnum, int64_t *map,
                                             BlockDriverState **file)
B
bellard 已提交
1324 1325
{
    BDRVVmdkState *s = bs->opaque;
F
Fam Zheng 已提交
1326
    int64_t index_in_cluster, n, ret;
1327
    uint64_t cluster_offset;
F
Fam Zheng 已提交
1328 1329
    VmdkExtent *extent;

1330
    extent = find_extent(s, offset >> BDRV_SECTOR_BITS, NULL);
F
Fam Zheng 已提交
1331
    if (!extent) {
1332
        return -EIO;
F
Fam Zheng 已提交
1333
    }
1334
    qemu_co_mutex_lock(&s->lock);
1335
    ret = get_cluster_offset(bs, extent, NULL, offset, false, &cluster_offset,
F
Fam Zheng 已提交
1336
                             0, 0);
1337
    qemu_co_mutex_unlock(&s->lock);
1338

1339
    index_in_cluster = vmdk_find_offset_in_cluster(extent, offset);
1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351
    switch (ret) {
    case VMDK_ERROR:
        ret = -EIO;
        break;
    case VMDK_UNALLOC:
        ret = 0;
        break;
    case VMDK_ZEROED:
        ret = BDRV_BLOCK_ZERO;
        break;
    case VMDK_OK:
        ret = BDRV_BLOCK_DATA;
1352
        if (!extent->compressed) {
1353
            ret |= BDRV_BLOCK_OFFSET_VALID;
1354
            *map = cluster_offset + index_in_cluster;
1355
        }
1356
        *file = extent->file->bs;
1357 1358
        break;
    }
1359

1360 1361
    n = extent->cluster_sectors * BDRV_SECTOR_SIZE - index_in_cluster;
    *pnum = MIN(n, bytes);
F
Fam Zheng 已提交
1362
    return ret;
B
bellard 已提交
1363 1364
}

1365
static int vmdk_write_extent(VmdkExtent *extent, int64_t cluster_offset,
1366 1367 1368
                            int64_t offset_in_cluster, QEMUIOVector *qiov,
                            uint64_t qiov_offset, uint64_t n_bytes,
                            uint64_t offset)
1369 1370
{
    int ret;
F
Fam Zheng 已提交
1371 1372
    VmdkGrainMarker *data = NULL;
    uLongf buf_len;
1373 1374
    QEMUIOVector local_qiov;
    struct iovec iov;
1375 1376
    int64_t write_offset;
    int64_t write_end_sector;
1377

F
Fam Zheng 已提交
1378
    if (extent->compressed) {
1379 1380
        void *compressed_data;

F
Fam Zheng 已提交
1381 1382 1383 1384 1385 1386
        if (!extent->has_marker) {
            ret = -EINVAL;
            goto out;
        }
        buf_len = (extent->cluster_sectors << 9) * 2;
        data = g_malloc(buf_len + sizeof(VmdkGrainMarker));
1387 1388 1389 1390 1391 1392 1393

        compressed_data = g_malloc(n_bytes);
        qemu_iovec_to_buf(qiov, qiov_offset, compressed_data, n_bytes);
        ret = compress(data->data, &buf_len, compressed_data, n_bytes);
        g_free(compressed_data);

        if (ret != Z_OK || buf_len == 0) {
F
Fam Zheng 已提交
1394 1395 1396
            ret = -EINVAL;
            goto out;
        }
1397

1398 1399
        data->lba = cpu_to_le64(offset >> BDRV_SECTOR_BITS);
        data->size = cpu_to_le32(buf_len);
1400 1401 1402 1403 1404 1405 1406

        n_bytes = buf_len + sizeof(VmdkGrainMarker);
        iov = (struct iovec) {
            .iov_base   = data,
            .iov_len    = n_bytes,
        };
        qemu_iovec_init_external(&local_qiov, &iov, 1);
M
Max Reitz 已提交
1407 1408

        BLKDBG_EVENT(extent->file, BLKDBG_WRITE_COMPRESSED);
1409 1410 1411
    } else {
        qemu_iovec_init(&local_qiov, qiov->niov);
        qemu_iovec_concat(&local_qiov, qiov, qiov_offset, n_bytes);
M
Max Reitz 已提交
1412 1413

        BLKDBG_EVENT(extent->file, BLKDBG_WRITE_AIO);
F
Fam Zheng 已提交
1414
    }
1415

1416
    write_offset = cluster_offset + offset_in_cluster;
1417
    ret = bdrv_co_pwritev(extent->file, write_offset, n_bytes,
1418
                          &local_qiov, 0);
1419

1420
    write_end_sector = DIV_ROUND_UP(write_offset + n_bytes, BDRV_SECTOR_SIZE);
1421

1422 1423 1424 1425 1426 1427
    if (extent->compressed) {
        extent->next_cluster_sector = write_end_sector;
    } else {
        extent->next_cluster_sector = MAX(extent->next_cluster_sector,
                                          write_end_sector);
    }
1428

1429
    if (ret < 0) {
1430 1431 1432 1433
        goto out;
    }
    ret = 0;
 out:
F
Fam Zheng 已提交
1434
    g_free(data);
1435 1436 1437
    if (!extent->compressed) {
        qemu_iovec_destroy(&local_qiov);
    }
1438 1439 1440 1441
    return ret;
}

static int vmdk_read_extent(VmdkExtent *extent, int64_t cluster_offset,
1442 1443
                            int64_t offset_in_cluster, QEMUIOVector *qiov,
                            int bytes)
1444 1445
{
    int ret;
F
Fam Zheng 已提交
1446 1447 1448 1449 1450 1451 1452
    int cluster_bytes, buf_bytes;
    uint8_t *cluster_buf, *compressed_data;
    uint8_t *uncomp_buf;
    uint32_t data_len;
    VmdkGrainMarker *marker;
    uLongf buf_len;

1453

F
Fam Zheng 已提交
1454
    if (!extent->compressed) {
M
Max Reitz 已提交
1455
        BLKDBG_EVENT(extent->file, BLKDBG_READ_AIO);
1456
        ret = bdrv_co_preadv(extent->file,
1457 1458 1459 1460
                             cluster_offset + offset_in_cluster, bytes,
                             qiov, 0);
        if (ret < 0) {
            return ret;
F
Fam Zheng 已提交
1461
        }
1462
        return 0;
F
Fam Zheng 已提交
1463 1464 1465 1466 1467 1468
    }
    cluster_bytes = extent->cluster_sectors * 512;
    /* Read two clusters in case GrainMarker + compressed data > one cluster */
    buf_bytes = cluster_bytes * 2;
    cluster_buf = g_malloc(buf_bytes);
    uncomp_buf = g_malloc(cluster_bytes);
M
Max Reitz 已提交
1469
    BLKDBG_EVENT(extent->file, BLKDBG_READ_COMPRESSED);
1470
    ret = bdrv_pread(extent->file,
F
Fam Zheng 已提交
1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494
                cluster_offset,
                cluster_buf, buf_bytes);
    if (ret < 0) {
        goto out;
    }
    compressed_data = cluster_buf;
    buf_len = cluster_bytes;
    data_len = cluster_bytes;
    if (extent->has_marker) {
        marker = (VmdkGrainMarker *)cluster_buf;
        compressed_data = marker->data;
        data_len = le32_to_cpu(marker->size);
    }
    if (!data_len || data_len > buf_bytes) {
        ret = -EINVAL;
        goto out;
    }
    ret = uncompress(uncomp_buf, &buf_len, compressed_data, data_len);
    if (ret != Z_OK) {
        ret = -EINVAL;
        goto out;

    }
    if (offset_in_cluster < 0 ||
1495
            offset_in_cluster + bytes > buf_len) {
F
Fam Zheng 已提交
1496 1497
        ret = -EINVAL;
        goto out;
1498
    }
1499
    qemu_iovec_from_buf(qiov, 0, uncomp_buf + offset_in_cluster, bytes);
F
Fam Zheng 已提交
1500 1501 1502 1503 1504 1505
    ret = 0;

 out:
    g_free(uncomp_buf);
    g_free(cluster_buf);
    return ret;
1506 1507
}

1508 1509 1510
static int coroutine_fn
vmdk_co_preadv(BlockDriverState *bs, uint64_t offset, uint64_t bytes,
               QEMUIOVector *qiov, int flags)
B
bellard 已提交
1511 1512
{
    BDRVVmdkState *s = bs->opaque;
F
Fam Zheng 已提交
1513
    int ret;
1514
    uint64_t n_bytes, offset_in_cluster;
F
Fam Zheng 已提交
1515
    VmdkExtent *extent = NULL;
1516
    QEMUIOVector local_qiov;
B
bellard 已提交
1517
    uint64_t cluster_offset;
1518
    uint64_t bytes_done = 0;
1519

1520 1521 1522 1523 1524
    qemu_iovec_init(&local_qiov, qiov->niov);
    qemu_co_mutex_lock(&s->lock);

    while (bytes > 0) {
        extent = find_extent(s, offset >> BDRV_SECTOR_BITS, extent);
F
Fam Zheng 已提交
1525
        if (!extent) {
1526 1527
            ret = -EIO;
            goto fail;
F
Fam Zheng 已提交
1528
        }
F
Fam Zheng 已提交
1529
        ret = get_cluster_offset(bs, extent, NULL,
1530 1531 1532 1533 1534 1535
                                 offset, false, &cluster_offset, 0, 0);
        offset_in_cluster = vmdk_find_offset_in_cluster(extent, offset);

        n_bytes = MIN(bytes, extent->cluster_sectors * BDRV_SECTOR_SIZE
                             - offset_in_cluster);

1536
        if (ret != VMDK_OK) {
1537
            /* if not allocated, try to read from parent image, if exist */
1538
            if (bs->backing && ret != VMDK_ZEROED) {
F
Fam Zheng 已提交
1539
                if (!vmdk_is_cid_valid(bs)) {
1540 1541
                    ret = -EINVAL;
                    goto fail;
F
Fam Zheng 已提交
1542
                }
1543 1544 1545 1546

                qemu_iovec_reset(&local_qiov);
                qemu_iovec_concat(&local_qiov, qiov, bytes_done, n_bytes);

M
Max Reitz 已提交
1547 1548
                /* qcow2 emits this on bs->file instead of bs->backing */
                BLKDBG_EVENT(bs->file, BLKDBG_READ_BACKING_AIO);
1549
                ret = bdrv_co_preadv(bs->backing, offset, n_bytes,
1550
                                     &local_qiov, 0);
F
Fam Zheng 已提交
1551
                if (ret < 0) {
1552
                    goto fail;
F
Fam Zheng 已提交
1553
                }
1554
            } else {
1555
                qemu_iovec_memset(qiov, bytes_done, 0, n_bytes);
1556
            }
B
bellard 已提交
1557
        } else {
1558 1559 1560 1561 1562
            qemu_iovec_reset(&local_qiov);
            qemu_iovec_concat(&local_qiov, qiov, bytes_done, n_bytes);

            ret = vmdk_read_extent(extent, cluster_offset, offset_in_cluster,
                                   &local_qiov, n_bytes);
1563
            if (ret) {
1564
                goto fail;
1565
            }
B
bellard 已提交
1566
        }
1567 1568 1569
        bytes -= n_bytes;
        offset += n_bytes;
        bytes_done += n_bytes;
B
bellard 已提交
1570 1571
    }

1572 1573
    ret = 0;
fail:
1574
    qemu_co_mutex_unlock(&s->lock);
1575 1576
    qemu_iovec_destroy(&local_qiov);

1577 1578 1579
    return ret;
}

F
Fam Zheng 已提交
1580 1581 1582
/**
 * vmdk_write:
 * @zeroed:       buf is ignored (data is zero), use zeroed_grain GTE feature
1583 1584 1585 1586
 *                if possible, otherwise return -ENOTSUP.
 * @zero_dry_run: used for zeroed == true only, don't update L2 table, just try
 *                with each cluster. By dry run we can find if the zero write
 *                is possible without modifying image data.
F
Fam Zheng 已提交
1587 1588 1589
 *
 * Returns: error code with 0 for success.
 */
1590 1591 1592
static int vmdk_pwritev(BlockDriverState *bs, uint64_t offset,
                       uint64_t bytes, QEMUIOVector *qiov,
                       bool zeroed, bool zero_dry_run)
B
bellard 已提交
1593
{
1594
    BDRVVmdkState *s = bs->opaque;
F
Fam Zheng 已提交
1595
    VmdkExtent *extent = NULL;
F
Fam Zheng 已提交
1596
    int ret;
1597
    int64_t offset_in_cluster, n_bytes;
1598
    uint64_t cluster_offset;
1599
    uint64_t bytes_done = 0;
F
Fam Zheng 已提交
1600
    VmdkMetaData m_data;
1601

1602 1603
    if (DIV_ROUND_UP(offset, BDRV_SECTOR_SIZE) > bs->total_sectors) {
        error_report("Wrong offset: offset=0x%" PRIx64
1604
                     " total_sectors=0x%" PRIx64,
1605
                     offset, bs->total_sectors);
1606
        return -EIO;
1607 1608
    }

1609 1610
    while (bytes > 0) {
        extent = find_extent(s, offset >> BDRV_SECTOR_BITS, extent);
F
Fam Zheng 已提交
1611 1612 1613
        if (!extent) {
            return -EIO;
        }
1614 1615 1616 1617 1618
        offset_in_cluster = vmdk_find_offset_in_cluster(extent, offset);
        n_bytes = MIN(bytes, extent->cluster_sectors * BDRV_SECTOR_SIZE
                             - offset_in_cluster);

        ret = get_cluster_offset(bs, extent, &m_data, offset,
F
Fam Zheng 已提交
1619
                                 !(extent->compressed || zeroed),
1620 1621
                                 &cluster_offset, offset_in_cluster,
                                 offset_in_cluster + n_bytes);
F
Fam Zheng 已提交
1622
        if (extent->compressed) {
F
Fam Zheng 已提交
1623
            if (ret == VMDK_OK) {
F
Fam Zheng 已提交
1624
                /* Refuse write to allocated cluster for streamOptimized */
F
Fam Zheng 已提交
1625 1626
                error_report("Could not write to allocated cluster"
                              " for streamOptimized");
F
Fam Zheng 已提交
1627 1628 1629
                return -EIO;
            } else {
                /* allocate */
1630
                ret = get_cluster_offset(bs, extent, &m_data, offset,
F
Fam Zheng 已提交
1631
                                         true, &cluster_offset, 0, 0);
F
Fam Zheng 已提交
1632 1633
            }
        }
F
Fam Zheng 已提交
1634
        if (ret == VMDK_ERROR) {
1635
            return -EINVAL;
F
Fam Zheng 已提交
1636
        }
F
Fam Zheng 已提交
1637 1638 1639
        if (zeroed) {
            /* Do zeroed write, buf is ignored */
            if (extent->has_zero_grain &&
1640 1641 1642
                    offset_in_cluster == 0 &&
                    n_bytes >= extent->cluster_sectors * BDRV_SECTOR_SIZE) {
                n_bytes = extent->cluster_sectors * BDRV_SECTOR_SIZE;
F
Fam Zheng 已提交
1643 1644
                if (!zero_dry_run) {
                    /* update L2 tables */
F
Fam Zheng 已提交
1645 1646
                    if (vmdk_L2update(extent, &m_data, VMDK_GTE_ZEROED)
                            != VMDK_OK) {
F
Fam Zheng 已提交
1647 1648 1649 1650 1651 1652 1653
                        return -EIO;
                    }
                }
            } else {
                return -ENOTSUP;
            }
        } else {
1654 1655
            ret = vmdk_write_extent(extent, cluster_offset, offset_in_cluster,
                                    qiov, bytes_done, n_bytes, offset);
F
Fam Zheng 已提交
1656 1657 1658 1659 1660
            if (ret) {
                return ret;
            }
            if (m_data.valid) {
                /* update L2 tables */
F
Fam Zheng 已提交
1661 1662 1663
                if (vmdk_L2update(extent, &m_data,
                                  cluster_offset >> BDRV_SECTOR_BITS)
                        != VMDK_OK) {
F
Fam Zheng 已提交
1664 1665
                    return -EIO;
                }
F
Fam Zheng 已提交
1666
            }
1667
        }
1668 1669 1670
        bytes -= n_bytes;
        offset += n_bytes;
        bytes_done += n_bytes;
1671

F
Fam Zheng 已提交
1672 1673
        /* update CID on the first write every time the virtual disk is
         * opened */
1674
        if (!s->cid_updated) {
F
Fam Zheng 已提交
1675
            ret = vmdk_write_cid(bs, g_random_int());
K
Kevin Wolf 已提交
1676 1677 1678
            if (ret < 0) {
                return ret;
            }
1679
            s->cid_updated = true;
1680
        }
1681 1682
    }
    return 0;
B
bellard 已提交
1683 1684
}

1685 1686 1687
static int coroutine_fn
vmdk_co_pwritev(BlockDriverState *bs, uint64_t offset, uint64_t bytes,
                QEMUIOVector *qiov, int flags)
1688 1689 1690 1691
{
    int ret;
    BDRVVmdkState *s = bs->opaque;
    qemu_co_mutex_lock(&s->lock);
1692
    ret = vmdk_pwritev(bs, offset, bytes, qiov, false, false);
F
Fam Zheng 已提交
1693 1694 1695 1696
    qemu_co_mutex_unlock(&s->lock);
    return ret;
}

1697 1698 1699
static int coroutine_fn
vmdk_co_pwritev_compressed(BlockDriverState *bs, uint64_t offset,
                           uint64_t bytes, QEMUIOVector *qiov)
1700
{
1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721
    if (bytes == 0) {
        /* The caller will write bytes 0 to signal EOF.
         * When receive it, we align EOF to a sector boundary. */
        BDRVVmdkState *s = bs->opaque;
        int i, ret;
        int64_t length;

        for (i = 0; i < s->num_extents; i++) {
            length = bdrv_getlength(s->extents[i].file->bs);
            if (length < 0) {
                return length;
            }
            length = QEMU_ALIGN_UP(length, BDRV_SECTOR_SIZE);
            ret = bdrv_truncate(s->extents[i].file, length,
                                PREALLOC_MODE_OFF, NULL);
            if (ret < 0) {
                return ret;
            }
        }
        return 0;
    }
1722
    return vmdk_co_pwritev(bs, offset, bytes, qiov, 0);
1723 1724
}

1725 1726 1727 1728
static int coroutine_fn vmdk_co_pwrite_zeroes(BlockDriverState *bs,
                                              int64_t offset,
                                              int bytes,
                                              BdrvRequestFlags flags)
F
Fam Zheng 已提交
1729 1730 1731
{
    int ret;
    BDRVVmdkState *s = bs->opaque;
1732

F
Fam Zheng 已提交
1733
    qemu_co_mutex_lock(&s->lock);
1734 1735
    /* write zeroes could fail if sectors not aligned to cluster, test it with
     * dry_run == true before really updating image */
1736
    ret = vmdk_pwritev(bs, offset, bytes, NULL, true, true);
F
Fam Zheng 已提交
1737
    if (!ret) {
1738
        ret = vmdk_pwritev(bs, offset, bytes, NULL, true, false);
F
Fam Zheng 已提交
1739
    }
1740 1741 1742 1743
    qemu_co_mutex_unlock(&s->lock);
    return ret;
}

F
Fam Zheng 已提交
1744 1745 1746 1747
static int vmdk_init_extent(BlockBackend *blk,
                            int64_t filesize, bool flat,
                            bool compress, bool zeroed_grain,
                            Error **errp)
1748
{
F
Fam Zheng 已提交
1749
    int ret, i;
1750
    VMDK4Header header;
1751 1752 1753
    uint32_t tmp, magic, grains, gd_sectors, gt_size, gt_count;
    uint32_t *gd_buf = NULL;
    int gd_buf_size;
1754

F
Fam Zheng 已提交
1755
    if (flat) {
1756
        ret = blk_truncate(blk, filesize, PREALLOC_MODE_OFF, errp);
F
Fam Zheng 已提交
1757
        goto exit;
1758
    }
1759 1760
    magic = cpu_to_be32(VMDK4_MAGIC);
    memset(&header, 0, sizeof(header));
1761 1762 1763 1764 1765 1766 1767
    if (compress) {
        header.version = 3;
    } else if (zeroed_grain) {
        header.version = 2;
    } else {
        header.version = 1;
    }
F
Fam Zheng 已提交
1768
    header.flags = VMDK4_FLAG_RGD | VMDK4_FLAG_NL_DETECT
1769 1770
                   | (compress ? VMDK4_FLAG_COMPRESS | VMDK4_FLAG_MARKER : 0)
                   | (zeroed_grain ? VMDK4_FLAG_ZERO_GRAIN : 0);
1771
    header.compressAlgorithm = compress ? VMDK4_COMPRESSION_DEFLATE : 0;
1772
    header.capacity = filesize / BDRV_SECTOR_SIZE;
A
Alexander Graf 已提交
1773
    header.granularity = 128;
1774
    header.num_gtes_per_gt = BDRV_SECTOR_SIZE;
1775

1776 1777 1778 1779 1780
    grains = DIV_ROUND_UP(filesize / BDRV_SECTOR_SIZE, header.granularity);
    gt_size = DIV_ROUND_UP(header.num_gtes_per_gt * sizeof(uint32_t),
                           BDRV_SECTOR_SIZE);
    gt_count = DIV_ROUND_UP(grains, header.num_gtes_per_gt);
    gd_sectors = DIV_ROUND_UP(gt_count * sizeof(uint32_t), BDRV_SECTOR_SIZE);
1781 1782 1783 1784

    header.desc_offset = 1;
    header.desc_size = 20;
    header.rgd_offset = header.desc_offset + header.desc_size;
1785
    header.gd_offset = header.rgd_offset + gd_sectors + (gt_size * gt_count);
1786
    header.grain_offset =
1787 1788
        ROUND_UP(header.gd_offset + gd_sectors + (gt_size * gt_count),
                 header.granularity);
A
Alexander Graf 已提交
1789 1790 1791 1792 1793
    /* swap endianness for all header fields */
    header.version = cpu_to_le32(header.version);
    header.flags = cpu_to_le32(header.flags);
    header.capacity = cpu_to_le64(header.capacity);
    header.granularity = cpu_to_le64(header.granularity);
1794
    header.num_gtes_per_gt = cpu_to_le32(header.num_gtes_per_gt);
1795 1796 1797 1798 1799
    header.desc_offset = cpu_to_le64(header.desc_offset);
    header.desc_size = cpu_to_le64(header.desc_size);
    header.rgd_offset = cpu_to_le64(header.rgd_offset);
    header.gd_offset = cpu_to_le64(header.gd_offset);
    header.grain_offset = cpu_to_le64(header.grain_offset);
1800
    header.compressAlgorithm = cpu_to_le16(header.compressAlgorithm);
1801 1802 1803 1804 1805

    header.check_bytes[0] = 0xa;
    header.check_bytes[1] = 0x20;
    header.check_bytes[2] = 0xd;
    header.check_bytes[3] = 0xa;
1806 1807

    /* write all the data */
1808
    ret = blk_pwrite(blk, 0, &magic, sizeof(magic), 0);
1809
    if (ret < 0) {
1810
        error_setg(errp, QERR_IO_ERROR);
1811 1812
        goto exit;
    }
1813
    ret = blk_pwrite(blk, sizeof(magic), &header, sizeof(header), 0);
1814
    if (ret < 0) {
1815
        error_setg(errp, QERR_IO_ERROR);
1816 1817
        goto exit;
    }
1818

1819 1820
    ret = blk_truncate(blk, le64_to_cpu(header.grain_offset) << 9,
                       PREALLOC_MODE_OFF, errp);
1821 1822 1823
    if (ret < 0) {
        goto exit;
    }
1824 1825

    /* write grain directory */
1826 1827 1828
    gd_buf_size = gd_sectors * BDRV_SECTOR_SIZE;
    gd_buf = g_malloc0(gd_buf_size);
    for (i = 0, tmp = le64_to_cpu(header.rgd_offset) + gd_sectors;
1829
         i < gt_count; i++, tmp += gt_size) {
1830 1831
        gd_buf[i] = cpu_to_le32(tmp);
    }
1832
    ret = blk_pwrite(blk, le64_to_cpu(header.rgd_offset) * BDRV_SECTOR_SIZE,
1833
                     gd_buf, gd_buf_size, 0);
1834
    if (ret < 0) {
1835
        error_setg(errp, QERR_IO_ERROR);
1836
        goto exit;
1837
    }
1838

1839
    /* write backup grain directory */
1840
    for (i = 0, tmp = le64_to_cpu(header.gd_offset) + gd_sectors;
1841
         i < gt_count; i++, tmp += gt_size) {
1842 1843
        gd_buf[i] = cpu_to_le32(tmp);
    }
1844
    ret = blk_pwrite(blk, le64_to_cpu(header.gd_offset) * BDRV_SECTOR_SIZE,
1845
                     gd_buf, gd_buf_size, 0);
1846
    if (ret < 0) {
1847
        error_setg(errp, QERR_IO_ERROR);
1848
    }
1849

F
Fam Zheng 已提交
1850
    ret = 0;
F
Fam Zheng 已提交
1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882
exit:
    g_free(gd_buf);
    return ret;
}

static int vmdk_create_extent(const char *filename, int64_t filesize,
                              bool flat, bool compress, bool zeroed_grain,
                              BlockBackend **pbb,
                              QemuOpts *opts, Error **errp)
{
    int ret;
    BlockBackend *blk = NULL;
    Error *local_err = NULL;

    ret = bdrv_create_file(filename, opts, &local_err);
    if (ret < 0) {
        error_propagate(errp, local_err);
        goto exit;
    }

    blk = blk_new_open(filename, NULL, NULL,
                       BDRV_O_RDWR | BDRV_O_RESIZE | BDRV_O_PROTOCOL,
                       &local_err);
    if (blk == NULL) {
        error_propagate(errp, local_err);
        ret = -EIO;
        goto exit;
    }

    blk_set_allow_write_beyond_eof(blk, true);

    ret = vmdk_init_extent(blk, filesize, flat, compress, zeroed_grain, errp);
1883
exit:
1884
    if (blk) {
F
Fam Zheng 已提交
1885 1886 1887 1888 1889 1890
        if (pbb) {
            *pbb = blk;
        } else {
            blk_unref(blk);
            blk = NULL;
        }
1891
    }
F
Fam Zheng 已提交
1892 1893 1894 1895
    return ret;
}

static int filename_decompose(const char *filename, char *path, char *prefix,
F
Fam Zheng 已提交
1896
                              char *postfix, size_t buf_len, Error **errp)
F
Fam Zheng 已提交
1897 1898 1899 1900
{
    const char *p, *q;

    if (filename == NULL || !strlen(filename)) {
F
Fam Zheng 已提交
1901
        error_setg(errp, "No filename provided");
F
Fam Zheng 已提交
1902
        return VMDK_ERROR;
F
Fam Zheng 已提交
1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913
    }
    p = strrchr(filename, '/');
    if (p == NULL) {
        p = strrchr(filename, '\\');
    }
    if (p == NULL) {
        p = strrchr(filename, ':');
    }
    if (p != NULL) {
        p++;
        if (p - filename >= buf_len) {
F
Fam Zheng 已提交
1914
            return VMDK_ERROR;
F
Fam Zheng 已提交
1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926
        }
        pstrcpy(path, p - filename + 1, filename);
    } else {
        p = filename;
        path[0] = '\0';
    }
    q = strrchr(p, '.');
    if (q == NULL) {
        pstrcpy(prefix, buf_len, p);
        postfix[0] = '\0';
    } else {
        if (q - p >= buf_len) {
F
Fam Zheng 已提交
1927
            return VMDK_ERROR;
F
Fam Zheng 已提交
1928 1929 1930 1931
        }
        pstrcpy(prefix, q - p + 1, p);
        pstrcpy(postfix, buf_len, q);
    }
F
Fam Zheng 已提交
1932
    return VMDK_OK;
F
Fam Zheng 已提交
1933 1934
}

1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969
/*
 * idx == 0: get or create the descriptor file (also the image file if in a
 *           non-split format.
 * idx >= 1: get the n-th extent if in a split subformat
 */
typedef BlockBackend *(*vmdk_create_extent_fn)(int64_t size,
                                               int idx,
                                               bool flat,
                                               bool split,
                                               bool compress,
                                               bool zeroed_grain,
                                               void *opaque,
                                               Error **errp);

static void vmdk_desc_add_extent(GString *desc,
                                 const char *extent_line_fmt,
                                 int64_t size, const char *filename)
{
    char *basename = g_path_get_basename(filename);

    g_string_append_printf(desc, extent_line_fmt,
                           DIV_ROUND_UP(size, BDRV_SECTOR_SIZE), basename);
    g_free(basename);
}

static int coroutine_fn vmdk_co_do_create(int64_t size,
                                          BlockdevVmdkSubformat subformat,
                                          BlockdevVmdkAdapterType adapter_type,
                                          const char *backing_file,
                                          const char *hw_version,
                                          bool compat6,
                                          bool zeroed_grain,
                                          vmdk_create_extent_fn extent_fn,
                                          void *opaque,
                                          Error **errp)
F
Fam Zheng 已提交
1970
{
1971 1972
    int extent_idx;
    BlockBackend *blk = NULL;
1973
    BlockBackend *extent_blk;
F
Fam Zheng 已提交
1974
    Error *local_err = NULL;
1975
    char *desc = NULL;
F
Fam Zheng 已提交
1976
    int ret = 0;
1977
    bool flat, split, compress;
1978
    GString *ext_desc_lines;
F
Fam Zheng 已提交
1979
    const int64_t split_size = 0x80000000;  /* VMDK has constant split size */
1980 1981 1982
    int64_t extent_size;
    int64_t created_size = 0;
    const char *extent_line_fmt;
1983
    char *parent_desc_line = g_malloc0(BUF_SIZE);
F
Fam Zheng 已提交
1984
    uint32_t parent_cid = 0xffffffff;
1985
    uint32_t number_heads = 16;
1986
    uint32_t desc_offset = 0, desc_len;
F
Fam Zheng 已提交
1987 1988 1989
    const char desc_template[] =
        "# Disk DescriptorFile\n"
        "version=1\n"
1990 1991
        "CID=%" PRIx32 "\n"
        "parentCID=%" PRIx32 "\n"
F
Fam Zheng 已提交
1992 1993 1994 1995 1996 1997 1998 1999 2000
        "createType=\"%s\"\n"
        "%s"
        "\n"
        "# Extent description\n"
        "%s"
        "\n"
        "# The Disk Data Base\n"
        "#DDB\n"
        "\n"
2001
        "ddb.virtualHWVersion = \"%s\"\n"
F
Fam Zheng 已提交
2002
        "ddb.geometry.cylinders = \"%" PRId64 "\"\n"
2003
        "ddb.geometry.heads = \"%" PRIu32 "\"\n"
F
Fam Zheng 已提交
2004
        "ddb.geometry.sectors = \"63\"\n"
2005
        "ddb.adapterType = \"%s\"\n";
F
Fam Zheng 已提交
2006

2007 2008
    ext_desc_lines = g_string_new(NULL);

F
Fam Zheng 已提交
2009
    /* Read out options */
2010 2011
    if (compat6) {
        if (hw_version) {
2012 2013 2014 2015 2016
            error_setg(errp,
                       "compat6 cannot be enabled with hwversion set");
            ret = -EINVAL;
            goto exit;
        }
2017
        hw_version = "6";
2018
    }
2019 2020
    if (!hw_version) {
        hw_version = "4";
F
Fam Zheng 已提交
2021
    }
2022

2023
    if (adapter_type != BLOCKDEV_VMDK_ADAPTER_TYPE_IDE) {
2024 2025 2026 2027
        /* that's the number of heads with which vmware operates when
           creating, exporting, etc. vmdk files with a non-ide adapter type */
        number_heads = 255;
    }
2028 2029 2030 2031 2032 2033
    split = (subformat == BLOCKDEV_VMDK_SUBFORMAT_TWOGBMAXEXTENTFLAT) ||
            (subformat == BLOCKDEV_VMDK_SUBFORMAT_TWOGBMAXEXTENTSPARSE);
    flat = (subformat == BLOCKDEV_VMDK_SUBFORMAT_MONOLITHICFLAT) ||
           (subformat == BLOCKDEV_VMDK_SUBFORMAT_TWOGBMAXEXTENTFLAT);
    compress = subformat == BLOCKDEV_VMDK_SUBFORMAT_STREAMOPTIMIZED;

F
Fam Zheng 已提交
2034
    if (flat) {
2035
        extent_line_fmt = "RW %" PRId64 " FLAT \"%s\" 0\n";
F
Fam Zheng 已提交
2036
    } else {
2037
        extent_line_fmt = "RW %" PRId64 " SPARSE \"%s\"\n";
F
Fam Zheng 已提交
2038 2039
    }
    if (flat && backing_file) {
F
Fam Zheng 已提交
2040
        error_setg(errp, "Flat image can't have backing file");
2041 2042
        ret = -ENOTSUP;
        goto exit;
F
Fam Zheng 已提交
2043
    }
2044 2045
    if (flat && zeroed_grain) {
        error_setg(errp, "Flat image can't enable zeroed grain");
2046 2047
        ret = -ENOTSUP;
        goto exit;
2048
    }
2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072

    /* Create extents */
    if (split) {
        extent_size = split_size;
    } else {
        extent_size = size;
    }
    if (!split && !flat) {
        created_size = extent_size;
    } else {
        created_size = 0;
    }
    /* Get the descriptor file BDS */
    blk = extent_fn(created_size, 0, flat, split, compress, zeroed_grain,
                    opaque, errp);
    if (!blk) {
        ret = -EIO;
        goto exit;
    }
    if (!split && !flat) {
        vmdk_desc_add_extent(ext_desc_lines, extent_line_fmt, created_size,
                             blk_bs(blk)->filename);
    }

F
Fam Zheng 已提交
2073
    if (backing_file) {
2074
        BlockBackend *backing;
2075
        char *full_backing = g_new0(char, PATH_MAX);
2076
        bdrv_get_full_backing_filename_from_filename(blk_bs(blk)->filename, backing_file,
2077 2078 2079 2080 2081 2082 2083 2084
                                                     full_backing, PATH_MAX,
                                                     &local_err);
        if (local_err) {
            g_free(full_backing);
            error_propagate(errp, local_err);
            ret = -ENOENT;
            goto exit;
        }
2085

2086 2087
        backing = blk_new_open(full_backing, NULL, NULL,
                               BDRV_O_NO_BACKING, errp);
2088
        g_free(full_backing);
2089
        if (backing == NULL) {
2090
            ret = -EIO;
2091
            goto exit;
F
Fam Zheng 已提交
2092
        }
2093 2094 2095 2096
        if (strcmp(blk_bs(backing)->drv->format_name, "vmdk")) {
            error_setg(errp, "Invalid backing file format: %s. Must be vmdk",
                       blk_bs(backing)->drv->format_name);
            blk_unref(backing);
2097 2098
            ret = -EINVAL;
            goto exit;
F
Fam Zheng 已提交
2099
        }
2100 2101
        ret = vmdk_read_cid(blk_bs(backing), 0, &parent_cid);
        blk_unref(backing);
2102
        if (ret) {
2103
            error_setg(errp, "Failed to read parent CID");
2104 2105
            goto exit;
        }
2106
        snprintf(parent_desc_line, BUF_SIZE,
2107
                "parentFileNameHint=\"%s\"", backing_file);
F
Fam Zheng 已提交
2108
    }
2109 2110 2111 2112 2113 2114
    extent_idx = 1;
    while (created_size < size) {
        int64_t cur_size = MIN(size - created_size, extent_size);
        extent_blk = extent_fn(cur_size, extent_idx, flat, split, compress,
                               zeroed_grain, opaque, errp);
        if (!extent_blk) {
2115 2116
            ret = -EINVAL;
            goto exit;
F
Fam Zheng 已提交
2117
        }
2118 2119 2120 2121 2122
        vmdk_desc_add_extent(ext_desc_lines, extent_line_fmt, cur_size,
                             blk_bs(extent_blk)->filename);
        created_size += cur_size;
        extent_idx++;
        blk_unref(extent_blk);
F
Fam Zheng 已提交
2123
    }
2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134

    /* Check whether we got excess extents */
    extent_blk = extent_fn(-1, extent_idx, flat, split, compress, zeroed_grain,
                           opaque, NULL);
    if (extent_blk) {
        blk_unref(extent_blk);
        error_setg(errp, "List of extents contains unused extents");
        ret = -EINVAL;
        goto exit;
    }

F
Fam Zheng 已提交
2135
    /* generate descriptor file */
2136
    desc = g_strdup_printf(desc_template,
F
Fam Zheng 已提交
2137
                           g_random_int(),
2138
                           parent_cid,
2139
                           BlockdevVmdkSubformat_str(subformat),
2140 2141
                           parent_desc_line,
                           ext_desc_lines->str,
2142
                           hw_version,
2143
                           size /
2144
                               (int64_t)(63 * number_heads * BDRV_SECTOR_SIZE),
2145
                           number_heads,
2146
                           BlockdevVmdkAdapterType_str(adapter_type));
2147 2148 2149 2150
    desc_len = strlen(desc);
    /* the descriptor offset = 0x200 */
    if (!split && !flat) {
        desc_offset = 0x200;
2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161
    }

    ret = blk_pwrite(blk, desc_offset, desc, desc_len, 0);
    if (ret < 0) {
        error_setg_errno(errp, -ret, "Could not write description");
        goto exit;
    }
    /* bdrv_pwrite write padding zeros to align to sector, we don't need that
     * for description file */
    if (desc_offset == 0) {
        ret = blk_truncate(blk, desc_len, PREALLOC_MODE_OFF, errp);
2162 2163 2164
        if (ret < 0) {
            goto exit;
        }
F
Fam Zheng 已提交
2165
    }
2166 2167 2168 2169 2170 2171 2172 2173 2174 2175
    ret = 0;
exit:
    if (blk) {
        blk_unref(blk);
    }
    g_free(desc);
    g_free(parent_desc_line);
    g_string_free(ext_desc_lines, true);
    return ret;
}
2176

2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194
typedef struct {
    char *path;
    char *prefix;
    char *postfix;
    QemuOpts *opts;
} VMDKCreateOptsData;

static BlockBackend *vmdk_co_create_opts_cb(int64_t size, int idx,
                                            bool flat, bool split, bool compress,
                                            bool zeroed_grain, void *opaque,
                                            Error **errp)
{
    BlockBackend *blk = NULL;
    BlockDriverState *bs = NULL;
    VMDKCreateOptsData *data = opaque;
    char *ext_filename = NULL;
    char *rel_filename = NULL;

2195 2196 2197 2198 2199 2200
    /* We're done, don't create excess extents. */
    if (size == -1) {
        assert(errp == NULL);
        return NULL;
    }

2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217
    if (idx == 0) {
        rel_filename = g_strdup_printf("%s%s", data->prefix, data->postfix);
    } else if (split) {
        rel_filename = g_strdup_printf("%s-%c%03d%s",
                                       data->prefix,
                                       flat ? 'f' : 's', idx, data->postfix);
    } else {
        assert(idx == 1);
        rel_filename = g_strdup_printf("%s-flat%s", data->prefix, data->postfix);
    }

    ext_filename = g_strdup_printf("%s%s", data->path, rel_filename);
    g_free(rel_filename);

    if (vmdk_create_extent(ext_filename, size,
                           flat, compress, zeroed_grain, &blk, data->opts,
                           errp)) {
2218
        goto exit;
F
Fam Zheng 已提交
2219
    }
2220 2221 2222 2223 2224
    bdrv_unref(bs);
exit:
    g_free(ext_filename);
    return blk;
}
2225

2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248
static int coroutine_fn vmdk_co_create_opts(const char *filename, QemuOpts *opts,
                                            Error **errp)
{
    Error *local_err = NULL;
    char *desc = NULL;
    int64_t total_size = 0;
    char *adapter_type = NULL;
    BlockdevVmdkAdapterType adapter_type_enum;
    char *backing_file = NULL;
    char *hw_version = NULL;
    char *fmt = NULL;
    BlockdevVmdkSubformat subformat;
    int ret = 0;
    char *path = g_malloc0(PATH_MAX);
    char *prefix = g_malloc0(PATH_MAX);
    char *postfix = g_malloc0(PATH_MAX);
    char *desc_line = g_malloc0(BUF_SIZE);
    char *ext_filename = g_malloc0(PATH_MAX);
    char *desc_filename = g_malloc0(PATH_MAX);
    char *parent_desc_line = g_malloc0(BUF_SIZE);
    bool zeroed_grain;
    bool compat6;
    VMDKCreateOptsData data;
2249

2250 2251
    if (filename_decompose(filename, path, prefix, postfix, PATH_MAX, errp)) {
        ret = -EINVAL;
2252
        goto exit;
F
Fam Zheng 已提交
2253
    }
2254 2255 2256 2257 2258 2259 2260 2261 2262 2263
    /* Read out options */
    total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
                          BDRV_SECTOR_SIZE);
    adapter_type = qemu_opt_get_del(opts, BLOCK_OPT_ADAPTER_TYPE);
    backing_file = qemu_opt_get_del(opts, BLOCK_OPT_BACKING_FILE);
    hw_version = qemu_opt_get_del(opts, BLOCK_OPT_HWVERSION);
    compat6 = qemu_opt_get_bool_del(opts, BLOCK_OPT_COMPAT6, false);
    if (strcmp(hw_version, "undefined") == 0) {
        g_free(hw_version);
        hw_version = g_strdup("4");
2264
    }
2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279
    fmt = qemu_opt_get_del(opts, BLOCK_OPT_SUBFMT);
    zeroed_grain = qemu_opt_get_bool_del(opts, BLOCK_OPT_ZEROED_GRAIN, false);

    if (adapter_type) {
        adapter_type_enum = qapi_enum_parse(&BlockdevVmdkAdapterType_lookup,
                                            adapter_type,
                                            BLOCKDEV_VMDK_ADAPTER_TYPE_IDE,
                                            &local_err);
        if (local_err) {
            error_propagate(errp, local_err);
            ret = -EINVAL;
            goto exit;
        }
    } else {
        adapter_type_enum = BLOCKDEV_VMDK_ADAPTER_TYPE_IDE;
2280
    }
2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306

    if (!fmt) {
        /* Default format to monolithicSparse */
        subformat = BLOCKDEV_VMDK_SUBFORMAT_MONOLITHICSPARSE;
    } else {
        subformat = qapi_enum_parse(&BlockdevVmdkSubformat_lookup,
                                    fmt,
                                    BLOCKDEV_VMDK_SUBFORMAT_MONOLITHICSPARSE,
                                    &local_err);
        if (local_err) {
            error_propagate(errp, local_err);
            ret = -EINVAL;
            goto exit;
        }
    }
    data = (VMDKCreateOptsData){
        .prefix = prefix,
        .postfix = postfix,
        .path = path,
        .opts = opts,
    };
    ret = vmdk_co_do_create(total_size, subformat, adapter_type_enum,
                            backing_file, hw_version, compat6, zeroed_grain,
                            vmdk_co_create_opts_cb, &data, errp);

exit:
2307 2308
    g_free(adapter_type);
    g_free(backing_file);
2309
    g_free(hw_version);
2310
    g_free(fmt);
2311
    g_free(desc);
2312 2313 2314 2315 2316 2317 2318
    g_free(path);
    g_free(prefix);
    g_free(postfix);
    g_free(desc_line);
    g_free(ext_filename);
    g_free(desc_filename);
    g_free(parent_desc_line);
2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361
    return ret;
}

static BlockBackend *vmdk_co_create_cb(int64_t size, int idx,
                                       bool flat, bool split, bool compress,
                                       bool zeroed_grain, void *opaque,
                                       Error **errp)
{
    int ret;
    BlockDriverState *bs;
    BlockBackend *blk;
    BlockdevCreateOptionsVmdk *opts = opaque;

    if (idx == 0) {
        bs = bdrv_open_blockdev_ref(opts->file, errp);
    } else {
        int i;
        BlockdevRefList *list = opts->extents;
        for (i = 1; i < idx; i++) {
            if (!list || !list->next) {
                error_setg(errp, "Extent [%d] not specified", i);
                return NULL;
            }
            list = list->next;
        }
        if (!list) {
            error_setg(errp, "Extent [%d] not specified", idx - 1);
            return NULL;
        }
        bs = bdrv_open_blockdev_ref(list->value, errp);
    }
    if (!bs) {
        return NULL;
    }
    blk = blk_new(BLK_PERM_CONSISTENT_READ | BLK_PERM_WRITE | BLK_PERM_RESIZE,
                  BLK_PERM_ALL);
    if (blk_insert_bs(blk, bs, errp)) {
        bdrv_unref(bs);
        return NULL;
    }
    blk_set_allow_write_beyond_eof(blk, true);
    bdrv_unref(bs);

2362 2363 2364 2365 2366 2367
    if (size != -1) {
        ret = vmdk_init_extent(blk, size, flat, compress, zeroed_grain, errp);
        if (ret) {
            blk_unref(blk);
            blk = NULL;
        }
2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398
    }
    return blk;
}

static int coroutine_fn vmdk_co_create(BlockdevCreateOptions *create_options,
                                       Error **errp)
{
    int ret;
    BlockdevCreateOptionsVmdk *opts;

    opts = &create_options->u.vmdk;

    /* Validate options */
    if (!QEMU_IS_ALIGNED(opts->size, BDRV_SECTOR_SIZE)) {
        error_setg(errp, "Image size must be a multiple of 512 bytes");
        ret = -EINVAL;
        goto out;
    }

    ret = vmdk_co_do_create(opts->size,
                            opts->subformat,
                            opts->adapter_type,
                            opts->backing_file,
                            opts->hwversion,
                            false,
                            opts->zeroed_grain,
                            vmdk_co_create_cb,
                            opts, errp);
    return ret;

out:
2399
    return ret;
2400 2401
}

B
bellard 已提交
2402
static void vmdk_close(BlockDriverState *bs)
B
bellard 已提交
2403
{
K
Kevin Wolf 已提交
2404 2405
    BDRVVmdkState *s = bs->opaque;

F
Fam Zheng 已提交
2406
    vmdk_free_extents(bs);
F
Fam Zheng 已提交
2407
    g_free(s->create_type);
K
Kevin Wolf 已提交
2408 2409 2410

    migrate_del_blocker(s->migration_blocker);
    error_free(s->migration_blocker);
B
bellard 已提交
2411 2412
}

P
Paolo Bonzini 已提交
2413
static coroutine_fn int vmdk_co_flush(BlockDriverState *bs)
P
pbrook 已提交
2414
{
F
Fam Zheng 已提交
2415
    BDRVVmdkState *s = bs->opaque;
2416 2417
    int i, err;
    int ret = 0;
F
Fam Zheng 已提交
2418 2419

    for (i = 0; i < s->num_extents; i++) {
2420
        err = bdrv_co_flush(s->extents[i].file->bs);
F
Fam Zheng 已提交
2421 2422 2423 2424 2425
        if (err < 0) {
            ret = err;
        }
    }
    return ret;
P
pbrook 已提交
2426 2427
}

2428 2429 2430 2431 2432 2433 2434
static int64_t vmdk_get_allocated_file_size(BlockDriverState *bs)
{
    int i;
    int64_t ret = 0;
    int64_t r;
    BDRVVmdkState *s = bs->opaque;

K
Kevin Wolf 已提交
2435
    ret = bdrv_get_allocated_file_size(bs->file->bs);
2436 2437 2438 2439
    if (ret < 0) {
        return ret;
    }
    for (i = 0; i < s->num_extents; i++) {
K
Kevin Wolf 已提交
2440
        if (s->extents[i].file == bs->file) {
2441 2442
            continue;
        }
2443
        r = bdrv_get_allocated_file_size(s->extents[i].file->bs);
2444 2445 2446 2447 2448 2449 2450
        if (r < 0) {
            return r;
        }
        ret += r;
    }
    return ret;
}
2451

F
Fam Zheng 已提交
2452 2453 2454 2455 2456 2457 2458 2459 2460
static int vmdk_has_zero_init(BlockDriverState *bs)
{
    int i;
    BDRVVmdkState *s = bs->opaque;

    /* If has a flat extent and its underlying storage doesn't have zero init,
     * return 0. */
    for (i = 0; i < s->num_extents; i++) {
        if (s->extents[i].flat) {
2461
            if (!bdrv_has_zero_init(s->extents[i].file->bs)) {
F
Fam Zheng 已提交
2462 2463 2464 2465 2466 2467 2468
                return 0;
            }
        }
    }
    return 1;
}

F
Fam Zheng 已提交
2469 2470 2471 2472 2473
static ImageInfo *vmdk_get_extent_info(VmdkExtent *extent)
{
    ImageInfo *info = g_new0(ImageInfo, 1);

    *info = (ImageInfo){
2474
        .filename         = g_strdup(extent->file->bs->filename),
F
Fam Zheng 已提交
2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485
        .format           = g_strdup(extent->type),
        .virtual_size     = extent->sectors * BDRV_SECTOR_SIZE,
        .compressed       = extent->compressed,
        .has_compressed   = extent->compressed,
        .cluster_size     = extent->cluster_sectors * BDRV_SECTOR_SIZE,
        .has_cluster_size = !extent->flat,
    };

    return info;
}

2486 2487 2488
static int coroutine_fn vmdk_co_check(BlockDriverState *bs,
                                      BdrvCheckResult *result,
                                      BdrvCheckMode fix)
2489 2490 2491 2492
{
    BDRVVmdkState *s = bs->opaque;
    VmdkExtent *extent = NULL;
    int64_t sector_num = 0;
2493
    int64_t total_sectors = bdrv_nb_sectors(bs);
2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509
    int ret;
    uint64_t cluster_offset;

    if (fix) {
        return -ENOTSUP;
    }

    for (;;) {
        if (sector_num >= total_sectors) {
            return 0;
        }
        extent = find_extent(s, sector_num, extent);
        if (!extent) {
            fprintf(stderr,
                    "ERROR: could not find extent for sector %" PRId64 "\n",
                    sector_num);
2510
            ret = -EINVAL;
2511 2512 2513 2514
            break;
        }
        ret = get_cluster_offset(bs, extent, NULL,
                                 sector_num << BDRV_SECTOR_BITS,
F
Fam Zheng 已提交
2515
                                 false, &cluster_offset, 0, 0);
2516 2517 2518 2519 2520 2521
        if (ret == VMDK_ERROR) {
            fprintf(stderr,
                    "ERROR: could not get cluster_offset for sector %"
                    PRId64 "\n", sector_num);
            break;
        }
2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537
        if (ret == VMDK_OK) {
            int64_t extent_len = bdrv_getlength(extent->file->bs);
            if (extent_len < 0) {
                fprintf(stderr,
                        "ERROR: could not get extent file length for sector %"
                        PRId64 "\n", sector_num);
                ret = extent_len;
                break;
            }
            if (cluster_offset >= extent_len) {
                fprintf(stderr,
                        "ERROR: cluster offset for sector %"
                        PRId64 " points after EOF\n", sector_num);
                ret = -EINVAL;
                break;
            }
2538 2539 2540 2541 2542
        }
        sector_num += extent->cluster_sectors;
    }

    result->corruptions++;
2543
    return ret;
2544 2545
}

2546 2547
static ImageInfoSpecific *vmdk_get_specific_info(BlockDriverState *bs,
                                                 Error **errp)
F
Fam Zheng 已提交
2548 2549 2550 2551 2552 2553 2554
{
    int i;
    BDRVVmdkState *s = bs->opaque;
    ImageInfoSpecific *spec_info = g_new0(ImageInfoSpecific, 1);
    ImageInfoList **next;

    *spec_info = (ImageInfoSpecific){
2555
        .type = IMAGE_INFO_SPECIFIC_KIND_VMDK,
2556 2557
        .u = {
            .vmdk.data = g_new0(ImageInfoSpecificVmdk, 1),
F
Fam Zheng 已提交
2558 2559 2560
        },
    };

2561
    *spec_info->u.vmdk.data = (ImageInfoSpecificVmdk) {
F
Fam Zheng 已提交
2562 2563 2564 2565 2566
        .create_type = g_strdup(s->create_type),
        .cid = s->cid,
        .parent_cid = s->parent_cid,
    };

2567
    next = &spec_info->u.vmdk.data->extents;
F
Fam Zheng 已提交
2568 2569 2570 2571 2572 2573 2574 2575 2576 2577
    for (i = 0; i < s->num_extents; i++) {
        *next = g_new0(ImageInfoList, 1);
        (*next)->value = vmdk_get_extent_info(&s->extents[i]);
        (*next)->next = NULL;
        next = &(*next)->next;
    }

    return spec_info;
}

2578 2579 2580 2581 2582 2583 2584
static bool vmdk_extents_type_eq(const VmdkExtent *a, const VmdkExtent *b)
{
    return a->flat == b->flat &&
           a->compressed == b->compressed &&
           (a->flat || a->cluster_sectors == b->cluster_sectors);
}

F
Fam Zheng 已提交
2585 2586 2587 2588 2589
static int vmdk_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
{
    int i;
    BDRVVmdkState *s = bs->opaque;
    assert(s->num_extents);
2590

F
Fam Zheng 已提交
2591 2592
    /* See if we have multiple extents but they have different cases */
    for (i = 1; i < s->num_extents; i++) {
2593
        if (!vmdk_extents_type_eq(&s->extents[0], &s->extents[i])) {
F
Fam Zheng 已提交
2594 2595 2596
            return -ENOTSUP;
        }
    }
2597 2598 2599 2600
    bdi->needs_compressed_writes = s->extents[0].compressed;
    if (!s->extents[0].flat) {
        bdi->cluster_size = s->extents[0].cluster_sectors << BDRV_SECTOR_BITS;
    }
F
Fam Zheng 已提交
2601 2602 2603
    return 0;
}

2604 2605 2606 2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629
static QemuOptsList vmdk_create_opts = {
    .name = "vmdk-create-opts",
    .head = QTAILQ_HEAD_INITIALIZER(vmdk_create_opts.head),
    .desc = {
        {
            .name = BLOCK_OPT_SIZE,
            .type = QEMU_OPT_SIZE,
            .help = "Virtual disk size"
        },
        {
            .name = BLOCK_OPT_ADAPTER_TYPE,
            .type = QEMU_OPT_STRING,
            .help = "Virtual adapter type, can be one of "
                    "ide (default), lsilogic, buslogic or legacyESX"
        },
        {
            .name = BLOCK_OPT_BACKING_FILE,
            .type = QEMU_OPT_STRING,
            .help = "File name of a base image"
        },
        {
            .name = BLOCK_OPT_COMPAT6,
            .type = QEMU_OPT_BOOL,
            .help = "VMDK version 6 image",
            .def_value_str = "off"
        },
2630 2631 2632 2633 2634 2635
        {
            .name = BLOCK_OPT_HWVERSION,
            .type = QEMU_OPT_STRING,
            .help = "VMDK hardware version",
            .def_value_str = "undefined"
        },
2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650
        {
            .name = BLOCK_OPT_SUBFMT,
            .type = QEMU_OPT_STRING,
            .help =
                "VMDK flat extent format, can be one of "
                "{monolithicSparse (default) | monolithicFlat | twoGbMaxExtentSparse | twoGbMaxExtentFlat | streamOptimized} "
        },
        {
            .name = BLOCK_OPT_ZEROED_GRAIN,
            .type = QEMU_OPT_BOOL,
            .help = "Enable efficient zero writes "
                    "using the zeroed-grain GTE feature"
        },
        { /* end of list */ }
    }
2651 2652
};

2653
static BlockDriver bdrv_vmdk = {
F
Fam Zheng 已提交
2654 2655 2656 2657
    .format_name                  = "vmdk",
    .instance_size                = sizeof(BDRVVmdkState),
    .bdrv_probe                   = vmdk_probe,
    .bdrv_open                    = vmdk_open,
2658
    .bdrv_co_check                = vmdk_co_check,
F
Fam Zheng 已提交
2659
    .bdrv_reopen_prepare          = vmdk_reopen_prepare,
2660
    .bdrv_child_perm              = bdrv_format_default_perms,
2661
    .bdrv_co_preadv               = vmdk_co_preadv,
2662
    .bdrv_co_pwritev              = vmdk_co_pwritev,
2663
    .bdrv_co_pwritev_compressed   = vmdk_co_pwritev_compressed,
2664
    .bdrv_co_pwrite_zeroes        = vmdk_co_pwrite_zeroes,
F
Fam Zheng 已提交
2665
    .bdrv_close                   = vmdk_close,
2666
    .bdrv_co_create_opts          = vmdk_co_create_opts,
2667
    .bdrv_co_create               = vmdk_co_create,
F
Fam Zheng 已提交
2668
    .bdrv_co_flush_to_disk        = vmdk_co_flush,
2669
    .bdrv_co_block_status         = vmdk_co_block_status,
F
Fam Zheng 已提交
2670 2671
    .bdrv_get_allocated_file_size = vmdk_get_allocated_file_size,
    .bdrv_has_zero_init           = vmdk_has_zero_init,
F
Fam Zheng 已提交
2672
    .bdrv_get_specific_info       = vmdk_get_specific_info,
2673
    .bdrv_refresh_limits          = vmdk_refresh_limits,
F
Fam Zheng 已提交
2674
    .bdrv_get_info                = vmdk_get_info,
F
Fam Zheng 已提交
2675

2676
    .supports_backing             = true,
2677
    .create_opts                  = &vmdk_create_opts,
B
bellard 已提交
2678
};
2679 2680 2681 2682 2683 2684 2685

static void bdrv_vmdk_init(void)
{
    bdrv_register(&bdrv_vmdk);
}

block_init(bdrv_vmdk_init);