vmdk.c 70.3 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
pbrook 已提交
26
#include "qemu-common.h"
27
#include "block/block_int.h"
28
#include "qapi/qmp/qerror.h"
29
#include "qemu/error-report.h"
30
#include "qemu/module.h"
31
#include "migration/migration.h"
S
Stefan Weil 已提交
32
#include <zlib.h>
F
Fam Zheng 已提交
33
#include <glib.h>
B
bellard 已提交
34 35 36

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

46
#define VMDK_GTE_ZEROED 0x1
F
Fam Zheng 已提交
47 48 49 50 51 52 53 54

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

55 56
#define BLOCK_OPT_ZEROED_GRAIN "zeroed_grain"

B
bellard 已提交
57 58 59 60 61 62 63 64 65 66 67
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;
68
} QEMU_PACKED VMDK3Header;
B
bellard 已提交
69 70 71 72

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

#define L2_CACHE_SIZE 16

F
Fam Zheng 已提交
89 90 91
typedef struct VmdkExtent {
    BlockDriverState *file;
    bool flat;
F
Fam Zheng 已提交
92 93
    bool compressed;
    bool has_marker;
94 95
    bool has_zero_grain;
    int version;
F
Fam Zheng 已提交
96 97
    int64_t sectors;
    int64_t end_sector;
98
    int64_t flat_start_offset;
B
bellard 已提交
99
    int64_t l1_table_offset;
100
    int64_t l1_backup_table_offset;
B
bellard 已提交
101
    uint32_t *l1_table;
102
    uint32_t *l1_backup_table;
B
bellard 已提交
103 104 105 106 107 108 109 110
    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];

111
    int64_t cluster_sectors;
F
Fam Zheng 已提交
112
    int64_t next_cluster_sector;
F
Fam Zheng 已提交
113
    char *type;
F
Fam Zheng 已提交
114 115 116
} VmdkExtent;

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

130 131 132 133 134
typedef struct VmdkMetaData {
    unsigned int l1_index;
    unsigned int l2_index;
    unsigned int l2_offset;
    int valid;
F
Fam Zheng 已提交
135
    uint32_t *l2_cache_entry;
136 137
} VmdkMetaData;

F
Fam Zheng 已提交
138 139 140 141
typedef struct VmdkGrainMarker {
    uint64_t lba;
    uint32_t size;
    uint8_t  data[0];
142
} QEMU_PACKED VmdkGrainMarker;
F
Fam Zheng 已提交
143

144 145 146 147 148 149 150
enum {
    MARKER_END_OF_STREAM    = 0,
    MARKER_GRAIN_TABLE      = 1,
    MARKER_GRAIN_DIRECTORY  = 2,
    MARKER_FOOTER           = 3,
};

B
bellard 已提交
151 152 153 154
static int vmdk_probe(const uint8_t *buf, int buf_size, const char *filename)
{
    uint32_t magic;

F
Fam Zheng 已提交
155
    if (buf_size < 4) {
B
bellard 已提交
156
        return 0;
F
Fam Zheng 已提交
157
    }
B
bellard 已提交
158 159
    magic = be32_to_cpu(*(uint32_t *)buf);
    if (magic == VMDK3_MAGIC ||
160
        magic == VMDK4_MAGIC) {
B
bellard 已提交
161
        return 100;
162 163 164 165 166 167 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
    } 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 已提交
203
        return 0;
204
    }
B
bellard 已提交
205 206
}

207
#define SECTOR_SIZE 512
F
Fam Zheng 已提交
208 209 210
#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 */
211

F
Fam Zheng 已提交
212 213 214 215
static void vmdk_free_extents(BlockDriverState *bs)
{
    int i;
    BDRVVmdkState *s = bs->opaque;
F
Fam Zheng 已提交
216
    VmdkExtent *e;
F
Fam Zheng 已提交
217 218

    for (i = 0; i < s->num_extents; i++) {
F
Fam Zheng 已提交
219 220 221 222
        e = &s->extents[i];
        g_free(e->l1_table);
        g_free(e->l2_cache);
        g_free(e->l1_backup_table);
F
Fam Zheng 已提交
223
        g_free(e->type);
F
Fam Zheng 已提交
224
        if (e->file != bs->file) {
F
Fam Zheng 已提交
225
            bdrv_unref(e->file);
F
Fam Zheng 已提交
226
        }
F
Fam Zheng 已提交
227
    }
228
    g_free(s->extents);
F
Fam Zheng 已提交
229 230
}

231 232 233 234 235 236 237 238
static void vmdk_free_last_extent(BlockDriverState *bs)
{
    BDRVVmdkState *s = bs->opaque;

    if (s->num_extents == 0) {
        return;
    }
    s->num_extents--;
239
    s->extents = g_renew(VmdkExtent, s->extents, s->num_extents);
240 241
}

242
static uint32_t vmdk_read_cid(BlockDriverState *bs, int parent)
B
bellard 已提交
243
{
244
    char desc[DESC_SIZE];
245
    uint32_t cid = 0xffffffff;
246
    const char *p_name, *cid_str;
247
    size_t cid_str_size;
248
    BDRVVmdkState *s = bs->opaque;
K
Kevin Wolf 已提交
249
    int ret;
250

K
Kevin Wolf 已提交
251 252
    ret = bdrv_pread(bs->file, s->desc_offset, desc, DESC_SIZE);
    if (ret < 0) {
253
        return 0;
254
    }
255 256 257 258 259 260 261 262 263

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

K
Kevin Wolf 已提交
264
    desc[DESC_SIZE - 1] = '\0';
F
Fam Zheng 已提交
265 266
    p_name = strstr(desc, cid_str);
    if (p_name != NULL) {
267
        p_name += cid_str_size;
268
        sscanf(p_name, "%" SCNx32, &cid);
269 270 271 272 273 274 275 276 277
    }

    return cid;
}

static int vmdk_write_cid(BlockDriverState *bs, uint32_t cid)
{
    char desc[DESC_SIZE], tmp_desc[DESC_SIZE];
    char *p_name, *tmp_str;
278
    BDRVVmdkState *s = bs->opaque;
K
Kevin Wolf 已提交
279
    int ret;
280

K
Kevin Wolf 已提交
281 282 283
    ret = bdrv_pread(bs->file, s->desc_offset, desc, DESC_SIZE);
    if (ret < 0) {
        return ret;
284
    }
285

K
Kevin Wolf 已提交
286
    desc[DESC_SIZE - 1] = '\0';
F
Fam Zheng 已提交
287
    tmp_str = strstr(desc, "parentCID");
K
Kevin Wolf 已提交
288 289 290 291
    if (tmp_str == NULL) {
        return -EINVAL;
    }

B
blueswir1 已提交
292
    pstrcpy(tmp_desc, sizeof(tmp_desc), tmp_str);
F
Fam Zheng 已提交
293 294
    p_name = strstr(desc, "CID");
    if (p_name != NULL) {
295
        p_name += sizeof("CID");
296
        snprintf(p_name, sizeof(desc) - (p_name - desc), "%" PRIx32 "\n", cid);
B
blueswir1 已提交
297
        pstrcat(desc, sizeof(desc), tmp_desc);
298 299
    }

K
Kevin Wolf 已提交
300 301 302
    ret = bdrv_pwrite_sync(bs->file, s->desc_offset, desc, DESC_SIZE);
    if (ret < 0) {
        return ret;
303
    }
K
Kevin Wolf 已提交
304

305 306 307 308 309 310
    return 0;
}

static int vmdk_is_cid_valid(BlockDriverState *bs)
{
    BDRVVmdkState *s = bs->opaque;
K
Kevin Wolf 已提交
311
    BlockDriverState *p_bs = bs->backing_hd;
312 313
    uint32_t cur_pcid;

314
    if (!s->cid_checked && p_bs) {
F
Fam Zheng 已提交
315 316 317
        cur_pcid = vmdk_read_cid(p_bs, 0);
        if (s->parent_cid != cur_pcid) {
            /* CID not valid */
318
            return 0;
F
Fam Zheng 已提交
319
        }
320
    }
321
    s->cid_checked = true;
F
Fam Zheng 已提交
322
    /* CID valid */
323 324 325
    return 1;
}

K
Kevin Wolf 已提交
326
/* We have nothing to do for VMDK reopen, stubs just return success */
J
Jeff Cody 已提交
327 328 329 330 331
static int vmdk_reopen_prepare(BDRVReopenState *state,
                               BlockReopenQueue *queue, Error **errp)
{
    assert(state != NULL);
    assert(state->bs != NULL);
K
Kevin Wolf 已提交
332
    return 0;
J
Jeff Cody 已提交
333 334
}

335
static int vmdk_parent_open(BlockDriverState *bs)
336
{
337
    char *p_name;
338
    char desc[DESC_SIZE + 1];
339
    BDRVVmdkState *s = bs->opaque;
340
    int ret;
341

342
    desc[DESC_SIZE] = '\0';
343 344 345
    ret = bdrv_pread(bs->file, s->desc_offset, desc, DESC_SIZE);
    if (ret < 0) {
        return ret;
346
    }
347

F
Fam Zheng 已提交
348 349
    p_name = strstr(desc, "parentFileNameHint");
    if (p_name != NULL) {
350 351 352
        char *end_name;

        p_name += sizeof("parentFileNameHint") + 1;
F
Fam Zheng 已提交
353 354
        end_name = strchr(p_name, '\"');
        if (end_name == NULL) {
355
            return -EINVAL;
F
Fam Zheng 已提交
356 357
        }
        if ((end_name - p_name) > sizeof(bs->backing_file) - 1) {
358
            return -EINVAL;
F
Fam Zheng 已提交
359
        }
360

K
Kevin Wolf 已提交
361
        pstrcpy(bs->backing_file, end_name - p_name + 1, p_name);
362
    }
363 364 365 366

    return 0;
}

F
Fam Zheng 已提交
367 368
/* Create and append extent to the extent array. Return the added VmdkExtent
 * address. return NULL if allocation failed. */
369
static int vmdk_add_extent(BlockDriverState *bs,
F
Fam Zheng 已提交
370 371 372
                           BlockDriverState *file, bool flat, int64_t sectors,
                           int64_t l1_offset, int64_t l1_backup_offset,
                           uint32_t l1_size,
373
                           int l2_size, uint64_t cluster_sectors,
F
Fam Zheng 已提交
374 375
                           VmdkExtent **new_extent,
                           Error **errp)
F
Fam Zheng 已提交
376 377 378
{
    VmdkExtent *extent;
    BDRVVmdkState *s = bs->opaque;
379
    int64_t nb_sectors;
F
Fam Zheng 已提交
380

381 382
    if (cluster_sectors > 0x200000) {
        /* 0x200000 * 512Bytes = 1GB for one cluster is unrealistic */
F
Fam Zheng 已提交
383 384
        error_setg(errp, "Invalid granularity, image may be corrupt");
        return -EFBIG;
385
    }
386 387 388 389 390
    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 已提交
391
        error_setg(errp, "L1 size too big");
392 393
        return -EFBIG;
    }
394

395 396 397
    nb_sectors = bdrv_nb_sectors(file);
    if (nb_sectors < 0) {
        return nb_sectors;
F
Fam Zheng 已提交
398 399
    }

400
    s->extents = g_renew(VmdkExtent, s->extents, s->num_extents + 1);
F
Fam Zheng 已提交
401 402 403 404 405 406 407 408 409 410 411 412
    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;
413
    extent->cluster_sectors = flat ? sectors : cluster_sectors;
414
    extent->next_cluster_sector = ROUND_UP(nb_sectors, cluster_sectors);
F
Fam Zheng 已提交
415 416 417 418 419 420 421

    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;
422 423 424 425
    if (new_extent) {
        *new_extent = extent;
    }
    return 0;
F
Fam Zheng 已提交
426 427
}

F
Fam Zheng 已提交
428 429
static int vmdk_init_tables(BlockDriverState *bs, VmdkExtent *extent,
                            Error **errp)
430
{
431
    int ret;
432 433
    size_t l1_size;
    int i;
434

B
bellard 已提交
435
    /* read the L1 table */
F
Fam Zheng 已提交
436
    l1_size = extent->l1_size * sizeof(uint32_t);
437 438 439 440 441
    extent->l1_table = g_try_malloc(l1_size);
    if (l1_size && extent->l1_table == NULL) {
        return -ENOMEM;
    }

442
    ret = bdrv_pread(extent->file,
F
Fam Zheng 已提交
443 444 445
                     extent->l1_table_offset,
                     extent->l1_table,
                     l1_size);
446
    if (ret < 0) {
F
Fam Zheng 已提交
447 448 449
        error_setg_errno(errp, -ret,
                         "Could not read l1 table from extent '%s'",
                         extent->file->filename);
450
        goto fail_l1;
F
Fam Zheng 已提交
451 452 453
    }
    for (i = 0; i < extent->l1_size; i++) {
        le32_to_cpus(&extent->l1_table[i]);
B
bellard 已提交
454 455
    }

F
Fam Zheng 已提交
456
    if (extent->l1_backup_table_offset) {
457 458 459 460 461
        extent->l1_backup_table = g_try_malloc(l1_size);
        if (l1_size && extent->l1_backup_table == NULL) {
            ret = -ENOMEM;
            goto fail_l1;
        }
462
        ret = bdrv_pread(extent->file,
F
Fam Zheng 已提交
463 464 465
                         extent->l1_backup_table_offset,
                         extent->l1_backup_table,
                         l1_size);
466
        if (ret < 0) {
F
Fam Zheng 已提交
467 468 469
            error_setg_errno(errp, -ret,
                             "Could not read l1 backup table from extent '%s'",
                             extent->file->filename);
470
            goto fail_l1b;
F
Fam Zheng 已提交
471 472 473
        }
        for (i = 0; i < extent->l1_size; i++) {
            le32_to_cpus(&extent->l1_backup_table[i]);
474 475 476
        }
    }

F
Fam Zheng 已提交
477
    extent->l2_cache =
478
        g_new(uint32_t, extent->l2_size * L2_CACHE_SIZE);
B
bellard 已提交
479
    return 0;
480
 fail_l1b:
481
    g_free(extent->l1_backup_table);
482
 fail_l1:
483
    g_free(extent->l1_table);
484 485 486
    return ret;
}

F
Fam Zheng 已提交
487 488
static int vmdk_open_vmfs_sparse(BlockDriverState *bs,
                                 BlockDriverState *file,
F
Fam Zheng 已提交
489
                                 int flags, Error **errp)
490 491 492 493 494 495
{
    int ret;
    uint32_t magic;
    VMDK3Header header;
    VmdkExtent *extent;

496
    ret = bdrv_pread(file, sizeof(magic), &header, sizeof(header));
497
    if (ret < 0) {
F
Fam Zheng 已提交
498 499 500
        error_setg_errno(errp, -ret,
                         "Could not read header from file '%s'",
                         file->filename);
501
        return ret;
502
    }
503 504
    ret = vmdk_add_extent(bs, file, false,
                          le32_to_cpu(header.disk_sectors),
505
                          (int64_t)le32_to_cpu(header.l1dir_offset) << 9,
506 507 508 509
                          0,
                          le32_to_cpu(header.l1dir_size),
                          4096,
                          le32_to_cpu(header.granularity),
F
Fam Zheng 已提交
510 511
                          &extent,
                          errp);
512 513 514
    if (ret < 0) {
        return ret;
    }
F
Fam Zheng 已提交
515
    ret = vmdk_init_tables(bs, extent, errp);
516
    if (ret) {
517 518
        /* free extent allocated by vmdk_add_extent */
        vmdk_free_last_extent(bs);
519 520 521 522
    }
    return ret;
}

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

P
Paolo Bonzini 已提交
526 527 528 529 530 531 532 533 534 535 536 537 538
static char *vmdk_read_desc(BlockDriverState *file, uint64_t desc_offset,
                            Error **errp)
{
    int64_t size;
    char *buf;
    int ret;

    size = bdrv_getlength(file);
    if (size < 0) {
        error_setg_errno(errp, -size, "Could not access file");
        return NULL;
    }

539 540 541 542 543 544 545 546
    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 已提交
547 548
    size = MIN(size, (1 << 20) - 1);  /* avoid unbounded allocation */
    buf = g_malloc(size + 1);
P
Paolo Bonzini 已提交
549 550 551 552 553 554 555

    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 已提交
556
    buf[ret] = 0;
P
Paolo Bonzini 已提交
557 558 559 560

    return buf;
}

561 562
static int vmdk_open_vmdk4(BlockDriverState *bs,
                           BlockDriverState *file,
K
Kevin Wolf 已提交
563
                           int flags, QDict *options, Error **errp)
564 565 566 567 568 569
{
    int ret;
    uint32_t magic;
    uint32_t l1_size, l1_entry_sectors;
    VMDK4Header header;
    VmdkExtent *extent;
F
Fam Zheng 已提交
570
    BDRVVmdkState *s = bs->opaque;
571
    int64_t l1_backup_offset = 0;
572

573
    ret = bdrv_pread(file, sizeof(magic), &header, sizeof(header));
574
    if (ret < 0) {
F
Fam Zheng 已提交
575 576 577
        error_setg_errno(errp, -ret,
                         "Could not read header from file '%s'",
                         file->filename);
P
Paolo Bonzini 已提交
578
        return -EINVAL;
579
    }
580
    if (header.capacity == 0) {
581
        uint64_t desc_offset = le64_to_cpu(header.desc_offset);
582
        if (desc_offset) {
583 584 585 586
            char *buf = vmdk_read_desc(file, desc_offset << 9, errp);
            if (!buf) {
                return -EINVAL;
            }
K
Kevin Wolf 已提交
587
            ret = vmdk_open_desc_file(bs, flags, buf, options, errp);
588 589
            g_free(buf);
            return ret;
590
        }
F
Fam Zheng 已提交
591
    }
592

F
Fam Zheng 已提交
593 594 595 596
    if (!s->create_type) {
        s->create_type = g_strdup("monolithicSparse");
    }

597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626
    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;

        ret = bdrv_pread(file,
            bs->file->total_sectors * 512 - 1536,
            &footer, sizeof(footer));
        if (ret < 0) {
627
            error_setg_errno(errp, -ret, "Failed to read footer");
628 629 630 631 632 633 634 635 636 637 638
            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)
        {
639
            error_setg(errp, "Invalid footer");
640 641 642 643 644 645
            return -EINVAL;
        }

        header = footer.header;
    }

646
    if (le32_to_cpu(header.version) > 3) {
647
        char buf[64];
648
        snprintf(buf, sizeof(buf), "VMDK version %" PRId32,
649
                 le32_to_cpu(header.version));
650 651
        error_setg(errp, QERR_UNKNOWN_BLOCK_FORMAT_FEATURE,
                   bdrv_get_device_or_node_name(bs), "vmdk", buf);
652
        return -ENOTSUP;
653 654 655 656 657 658 659
    } else if (le32_to_cpu(header.version) == 3 && (flags & BDRV_O_RDWR)) {
        /* 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;
660 661
    }

662
    if (le32_to_cpu(header.num_gtes_per_gt) > 512) {
P
Paolo Bonzini 已提交
663
        error_setg(errp, "L2 table size too big");
664 665 666
        return -EINVAL;
    }

667
    l1_entry_sectors = le32_to_cpu(header.num_gtes_per_gt)
668
                        * le64_to_cpu(header.granularity);
669
    if (l1_entry_sectors == 0) {
670
        error_setg(errp, "L1 entry size is invalid");
671 672
        return -EINVAL;
    }
673 674
    l1_size = (le64_to_cpu(header.capacity) + l1_entry_sectors - 1)
                / l1_entry_sectors;
675 676 677
    if (le32_to_cpu(header.flags) & VMDK4_FLAG_RGD) {
        l1_backup_offset = le64_to_cpu(header.rgd_offset) << 9;
    }
678
    if (bdrv_nb_sectors(file) < le64_to_cpu(header.grain_offset)) {
679 680 681
        error_setg(errp, "File truncated, expecting at least %" PRId64 " bytes",
                   (int64_t)(le64_to_cpu(header.grain_offset)
                             * BDRV_SECTOR_SIZE));
682 683 684
        return -EINVAL;
    }

685
    ret = vmdk_add_extent(bs, file, false,
686 687
                          le64_to_cpu(header.capacity),
                          le64_to_cpu(header.gd_offset) << 9,
688
                          l1_backup_offset,
689
                          l1_size,
690
                          le32_to_cpu(header.num_gtes_per_gt),
691
                          le64_to_cpu(header.granularity),
F
Fam Zheng 已提交
692 693
                          &extent,
                          errp);
694 695 696
    if (ret < 0) {
        return ret;
    }
F
Fam Zheng 已提交
697 698
    extent->compressed =
        le16_to_cpu(header.compressAlgorithm) == VMDK4_COMPRESSION_DEFLATE;
699 700 701 702
    if (extent->compressed) {
        g_free(s->create_type);
        s->create_type = g_strdup("streamOptimized");
    }
F
Fam Zheng 已提交
703
    extent->has_marker = le32_to_cpu(header.flags) & VMDK4_FLAG_MARKER;
704 705
    extent->version = le32_to_cpu(header.version);
    extent->has_zero_grain = le32_to_cpu(header.flags) & VMDK4_FLAG_ZERO_GRAIN;
F
Fam Zheng 已提交
706
    ret = vmdk_init_tables(bs, extent, errp);
707
    if (ret) {
708 709
        /* free extent allocated by vmdk_add_extent */
        vmdk_free_last_extent(bs);
710 711 712 713
    }
    return ret;
}

714 715 716 717 718 719 720 721 722
/* 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 已提交
723
        return VMDK_ERROR;
724 725 726 727
    }
    /* Skip "=\"" following opt_name */
    opt_pos += strlen(opt_name) + 2;
    if (opt_pos >= end) {
F
Fam Zheng 已提交
728
        return VMDK_ERROR;
729 730 731 732 733 734
    }
    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 已提交
735
        return VMDK_ERROR;
736 737
    }
    pstrcpy(buf, opt_end - opt_pos + 1, opt_pos);
F
Fam Zheng 已提交
738
    return VMDK_OK;
739 740
}

741 742
/* Open an extent file and append to bs array */
static int vmdk_open_sparse(BlockDriverState *bs,
743
                            BlockDriverState *file, int flags,
K
Kevin Wolf 已提交
744
                            char *buf, QDict *options, Error **errp)
745 746 747
{
    uint32_t magic;

748
    magic = ldl_be_p(buf);
749 750
    switch (magic) {
        case VMDK3_MAGIC:
F
Fam Zheng 已提交
751
            return vmdk_open_vmfs_sparse(bs, file, flags, errp);
752 753
            break;
        case VMDK4_MAGIC:
K
Kevin Wolf 已提交
754
            return vmdk_open_vmdk4(bs, file, flags, options, errp);
755 756
            break;
        default:
P
Paolo Bonzini 已提交
757 758
            error_setg(errp, "Image not in VMDK format");
            return -EINVAL;
759 760 761 762
            break;
    }
}

763
static int vmdk_parse_extents(const char *desc, BlockDriverState *bs,
K
Kevin Wolf 已提交
764 765
                              const char *desc_file_path, QDict *options,
                              Error **errp)
766 767
{
    int ret;
768
    int matches;
769 770 771 772 773 774
    char access[11];
    char type[11];
    char fname[512];
    const char *p = desc;
    int64_t sectors = 0;
    int64_t flat_offset;
775
    char *extent_path;
776
    BlockDriverState *extent_file;
F
Fam Zheng 已提交
777 778
    BDRVVmdkState *s = bs->opaque;
    VmdkExtent *extent;
K
Kevin Wolf 已提交
779
    char extent_opt_prefix[32];
780 781

    while (*p) {
782 783
        /* parse extent line in one of below formats:
         *
784 785
         * RW [size in sectors] FLAT "file-name.vmdk" OFFSET
         * RW [size in sectors] SPARSE "file-name.vmdk"
786 787
         * RW [size in sectors] VMFS "file-name.vmdk"
         * RW [size in sectors] VMFSSPARSE "file-name.vmdk"
788 789
         */
        flat_offset = -1;
790 791 792
        matches = sscanf(p, "%10s %" SCNd64 " %10s \"%511[^\n\r\"]\" %" SCNd64,
                         access, &sectors, type, fname, &flat_offset);
        if (matches < 4 || strcmp(access, "RW")) {
793 794
            goto next_line;
        } else if (!strcmp(type, "FLAT")) {
795
            if (matches != 5 || flat_offset < 0) {
F
Fam Zheng 已提交
796
                error_setg(errp, "Invalid extent lines: \n%s", p);
797 798
                return -EINVAL;
            }
F
Fam Zheng 已提交
799
        } else if (!strcmp(type, "VMFS")) {
800
            if (matches == 4) {
801 802 803 804 805
                flat_offset = 0;
            } else {
                error_setg(errp, "Invalid extent lines:\n%s", p);
                return -EINVAL;
            }
806
        } else if (matches != 4) {
807
            error_setg(errp, "Invalid extent lines:\n%s", p);
808 809 810 811
            return -EINVAL;
        }

        if (sectors <= 0 ||
F
Fam Zheng 已提交
812
            (strcmp(type, "FLAT") && strcmp(type, "SPARSE") &&
P
Paolo Bonzini 已提交
813
             strcmp(type, "VMFS") && strcmp(type, "VMFSSPARSE")) ||
814 815 816 817
            (strcmp(access, "RW"))) {
            goto next_line;
        }

818 819 820 821 822 823 824 825
        if (!path_is_absolute(fname) && !path_has_protocol(fname) &&
            !desc_file_path[0])
        {
            error_setg(errp, "Cannot use relative extent paths with VMDK "
                       "descriptor file '%s'", bs->file->filename);
            return -EINVAL;
        }

826
        extent_path = g_malloc0(PATH_MAX);
J
Jeff Cody 已提交
827
        path_combine(extent_path, PATH_MAX, desc_file_path, fname);
M
Max Reitz 已提交
828
        extent_file = NULL;
K
Kevin Wolf 已提交
829 830 831 832

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

833 834
        ret = bdrv_open_image(&extent_file, extent_path, options,
                              extent_opt_prefix, bs, &child_file, false, errp);
835
        g_free(extent_path);
836 837 838 839
        if (ret) {
            return ret;
        }

840
        /* save to extents array */
P
Paolo Bonzini 已提交
841
        if (!strcmp(type, "FLAT") || !strcmp(type, "VMFS")) {
842 843
            /* FLAT extent */

844
            ret = vmdk_add_extent(bs, extent_file, true, sectors,
F
Fam Zheng 已提交
845
                            0, 0, 0, 0, 0, &extent, errp);
846
            if (ret < 0) {
847
                bdrv_unref(extent_file);
848 849
                return ret;
            }
F
Fam Zheng 已提交
850
            extent->flat_start_offset = flat_offset << 9;
F
Fam Zheng 已提交
851 852
        } else if (!strcmp(type, "SPARSE") || !strcmp(type, "VMFSSPARSE")) {
            /* SPARSE extent and VMFSSPARSE extent are both "COWD" sparse file*/
853 854 855 856
            char *buf = vmdk_read_desc(extent_file, 0, errp);
            if (!buf) {
                ret = -EINVAL;
            } else {
K
Kevin Wolf 已提交
857 858
                ret = vmdk_open_sparse(bs, extent_file, bs->open_flags, buf,
                                       options, errp);
859
            }
860
            g_free(buf);
861
            if (ret) {
F
Fam Zheng 已提交
862
                bdrv_unref(extent_file);
863 864
                return ret;
            }
F
Fam Zheng 已提交
865
            extent = &s->extents[s->num_extents - 1];
866
        } else {
F
Fam Zheng 已提交
867
            error_setg(errp, "Unsupported extent type '%s'", type);
868
            bdrv_unref(extent_file);
869 870
            return -ENOTSUP;
        }
F
Fam Zheng 已提交
871
        extent->type = g_strdup(type);
872 873
next_line:
        /* move to next line */
F
Fam Zheng 已提交
874 875 876 877 878
        while (*p) {
            if (*p == '\n') {
                p++;
                break;
            }
879 880 881 882 883 884
            p++;
        }
    }
    return 0;
}

885
static int vmdk_open_desc_file(BlockDriverState *bs, int flags, char *buf,
K
Kevin Wolf 已提交
886
                               QDict *options, Error **errp)
887 888 889 890 891 892
{
    int ret;
    char ct[128];
    BDRVVmdkState *s = bs->opaque;

    if (vmdk_parse_description(buf, "createType", ct, sizeof(ct))) {
P
Paolo Bonzini 已提交
893 894
        error_setg(errp, "invalid VMDK image descriptor");
        ret = -EINVAL;
895
        goto exit;
896
    }
F
Fam Zheng 已提交
897
    if (strcmp(ct, "monolithicFlat") &&
P
Paolo Bonzini 已提交
898
        strcmp(ct, "vmfs") &&
F
Fam Zheng 已提交
899
        strcmp(ct, "vmfsSparse") &&
900
        strcmp(ct, "twoGbMaxExtentSparse") &&
F
Fam Zheng 已提交
901
        strcmp(ct, "twoGbMaxExtentFlat")) {
F
Fam Zheng 已提交
902
        error_setg(errp, "Unsupported image type '%s'", ct);
903 904
        ret = -ENOTSUP;
        goto exit;
905
    }
F
Fam Zheng 已提交
906
    s->create_type = g_strdup(ct);
907
    s->desc_offset = 0;
K
Kevin Wolf 已提交
908
    ret = vmdk_parse_extents(buf, bs, bs->file->exact_filename, options, errp);
909 910
exit:
    return ret;
911 912
}

M
Max Reitz 已提交
913 914
static int vmdk_open(BlockDriverState *bs, QDict *options, int flags,
                     Error **errp)
915
{
916
    char *buf;
917 918
    int ret;
    BDRVVmdkState *s = bs->opaque;
919
    uint32_t magic;
920

921 922 923 924 925
    buf = vmdk_read_desc(bs->file, 0, errp);
    if (!buf) {
        return -EINVAL;
    }

926 927 928 929
    magic = ldl_be_p(buf);
    switch (magic) {
        case VMDK3_MAGIC:
        case VMDK4_MAGIC:
K
Kevin Wolf 已提交
930
            ret = vmdk_open_sparse(bs, bs->file, flags, buf, options, errp);
931 932 933
            s->desc_offset = 0x200;
            break;
        default:
K
Kevin Wolf 已提交
934
            ret = vmdk_open_desc_file(bs, flags, buf, options, errp);
935
            break;
936
    }
937 938 939 940
    if (ret) {
        goto fail;
    }

P
Paolo Bonzini 已提交
941 942 943 944 945
    /* try to open parent images, if exist */
    ret = vmdk_parent_open(bs);
    if (ret) {
        goto fail;
    }
F
Fam Zheng 已提交
946
    s->cid = vmdk_read_cid(bs, 0);
P
Paolo Bonzini 已提交
947
    s->parent_cid = vmdk_read_cid(bs, 1);
948
    qemu_co_mutex_init(&s->lock);
K
Kevin Wolf 已提交
949 950

    /* Disable migration when VMDK images are used */
951 952 953
    error_setg(&s->migration_blocker, "The vmdk format used by node '%s' "
               "does not support live migration",
               bdrv_get_device_or_node_name(bs));
K
Kevin Wolf 已提交
954
    migrate_add_blocker(s->migration_blocker);
955
    g_free(buf);
K
Kevin Wolf 已提交
956
    return 0;
P
Paolo Bonzini 已提交
957 958

fail:
959
    g_free(buf);
F
Fam Zheng 已提交
960 961
    g_free(s->create_type);
    s->create_type = NULL;
P
Paolo Bonzini 已提交
962 963
    vmdk_free_extents(bs);
    return ret;
B
bellard 已提交
964 965
}

966

967
static void vmdk_refresh_limits(BlockDriverState *bs, Error **errp)
968 969 970 971 972 973 974 975 976 977 978 979 980
{
    BDRVVmdkState *s = bs->opaque;
    int i;

    for (i = 0; i < s->num_extents; i++) {
        if (!s->extents[i].flat) {
            bs->bl.write_zeroes_alignment =
                MAX(bs->bl.write_zeroes_alignment,
                    s->extents[i].cluster_sectors);
        }
    }
}

F
Fam Zheng 已提交
981 982 983 984 985 986 987 988 989 990
/**
 * 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 已提交
991
static int get_whole_cluster(BlockDriverState *bs,
F
Fam Zheng 已提交
992 993 994 995 996
                             VmdkExtent *extent,
                             uint64_t cluster_sector_num,
                             uint64_t sector_num,
                             uint64_t skip_start_sector,
                             uint64_t skip_end_sector)
997
{
998
    int ret = VMDK_OK;
F
Fam Zheng 已提交
999 1000 1001 1002 1003 1004 1005
    int64_t cluster_bytes;
    uint8_t *whole_grain;

    /* For COW, align request sector_num to cluster start */
    sector_num = QEMU_ALIGN_DOWN(sector_num, extent->cluster_sectors);
    cluster_bytes = extent->cluster_sectors << BDRV_SECTOR_BITS;
    whole_grain = qemu_blockalign(bs, cluster_bytes);
1006

F
Fam Zheng 已提交
1007 1008 1009 1010 1011 1012 1013
    if (!bs->backing_hd) {
        memset(whole_grain, 0,  skip_start_sector << BDRV_SECTOR_BITS);
        memset(whole_grain + (skip_end_sector << BDRV_SECTOR_BITS), 0,
               cluster_bytes - (skip_end_sector << BDRV_SECTOR_BITS));
    }

    assert(skip_end_sector <= extent->cluster_sectors);
1014 1015
    /* we will be here if it's first write on non-exist grain(cluster).
     * try to read from parent image, if exist */
F
Fam Zheng 已提交
1016 1017 1018 1019
    if (bs->backing_hd && !vmdk_is_cid_valid(bs)) {
        ret = VMDK_ERROR;
        goto exit;
    }
1020

F
Fam Zheng 已提交
1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032
    /* Read backing data before skip range */
    if (skip_start_sector > 0) {
        if (bs->backing_hd) {
            ret = bdrv_read(bs->backing_hd, sector_num,
                            whole_grain, skip_start_sector);
            if (ret < 0) {
                ret = VMDK_ERROR;
                goto exit;
            }
        }
        ret = bdrv_write(extent->file, cluster_sector_num, whole_grain,
                         skip_start_sector);
K
Kevin Wolf 已提交
1033
        if (ret < 0) {
1034 1035
            ret = VMDK_ERROR;
            goto exit;
K
Kevin Wolf 已提交
1036
        }
F
Fam Zheng 已提交
1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051
    }
    /* Read backing data after skip range */
    if (skip_end_sector < extent->cluster_sectors) {
        if (bs->backing_hd) {
            ret = bdrv_read(bs->backing_hd, sector_num + skip_end_sector,
                            whole_grain + (skip_end_sector << BDRV_SECTOR_BITS),
                            extent->cluster_sectors - skip_end_sector);
            if (ret < 0) {
                ret = VMDK_ERROR;
                goto exit;
            }
        }
        ret = bdrv_write(extent->file, cluster_sector_num + skip_end_sector,
                         whole_grain + (skip_end_sector << BDRV_SECTOR_BITS),
                         extent->cluster_sectors - skip_end_sector);
K
Kevin Wolf 已提交
1052
        if (ret < 0) {
1053 1054
            ret = VMDK_ERROR;
            goto exit;
1055 1056
        }
    }
F
Fam Zheng 已提交
1057

1058 1059 1060
exit:
    qemu_vfree(whole_grain);
    return ret;
1061 1062
}

F
Fam Zheng 已提交
1063 1064
static int vmdk_L2update(VmdkExtent *extent, VmdkMetaData *m_data,
                         uint32_t offset)
1065
{
F
Fam Zheng 已提交
1066
    offset = cpu_to_le32(offset);
1067
    /* update L2 table */
F
Fam Zheng 已提交
1068 1069 1070
    if (bdrv_pwrite_sync(
                extent->file,
                ((int64_t)m_data->l2_offset * 512)
F
Fam Zheng 已提交
1071
                    + (m_data->l2_index * sizeof(offset)),
1072
                &offset, sizeof(offset)) < 0) {
F
Fam Zheng 已提交
1073
        return VMDK_ERROR;
F
Fam Zheng 已提交
1074
    }
1075
    /* update backup L2 table */
F
Fam Zheng 已提交
1076 1077 1078 1079 1080
    if (extent->l1_backup_table_offset != 0) {
        m_data->l2_offset = extent->l1_backup_table[m_data->l1_index];
        if (bdrv_pwrite_sync(
                    extent->file,
                    ((int64_t)m_data->l2_offset * 512)
F
Fam Zheng 已提交
1081
                        + (m_data->l2_index * sizeof(offset)),
1082
                    &offset, sizeof(offset)) < 0) {
F
Fam Zheng 已提交
1083
            return VMDK_ERROR;
F
Fam Zheng 已提交
1084
        }
1085
    }
F
Fam Zheng 已提交
1086 1087 1088
    if (m_data->l2_cache_entry) {
        *m_data->l2_cache_entry = offset;
    }
1089

F
Fam Zheng 已提交
1090
    return VMDK_OK;
1091 1092
}

F
Fam Zheng 已提交
1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112
/**
 * 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.
 */
1113
static int get_cluster_offset(BlockDriverState *bs,
F
Fam Zheng 已提交
1114 1115 1116 1117 1118 1119 1120
                              VmdkExtent *extent,
                              VmdkMetaData *m_data,
                              uint64_t offset,
                              bool allocate,
                              uint64_t *cluster_offset,
                              uint64_t skip_start_sector,
                              uint64_t skip_end_sector)
B
bellard 已提交
1121 1122 1123
{
    unsigned int l1_index, l2_offset, l2_index;
    int min_index, i, j;
1124
    uint32_t min_count, *l2_table;
1125
    bool zeroed = false;
F
Fam Zheng 已提交
1126
    int64_t ret;
1127
    int64_t cluster_sector;
1128

F
Fam Zheng 已提交
1129
    if (m_data) {
1130
        m_data->valid = 0;
F
Fam Zheng 已提交
1131
    }
1132
    if (extent->flat) {
1133
        *cluster_offset = extent->flat_start_offset;
F
Fam Zheng 已提交
1134
        return VMDK_OK;
1135
    }
1136

F
Fam Zheng 已提交
1137
    offset -= (extent->end_sector - extent->sectors) * SECTOR_SIZE;
F
Fam Zheng 已提交
1138 1139
    l1_index = (offset >> 9) / extent->l1_entry_sectors;
    if (l1_index >= extent->l1_size) {
F
Fam Zheng 已提交
1140
        return VMDK_ERROR;
F
Fam Zheng 已提交
1141 1142 1143
    }
    l2_offset = extent->l1_table[l1_index];
    if (!l2_offset) {
F
Fam Zheng 已提交
1144
        return VMDK_UNALLOC;
F
Fam Zheng 已提交
1145
    }
1146
    for (i = 0; i < L2_CACHE_SIZE; i++) {
F
Fam Zheng 已提交
1147
        if (l2_offset == extent->l2_cache_offsets[i]) {
B
bellard 已提交
1148
            /* increment the hit count */
F
Fam Zheng 已提交
1149
            if (++extent->l2_cache_counts[i] == 0xffffffff) {
1150
                for (j = 0; j < L2_CACHE_SIZE; j++) {
F
Fam Zheng 已提交
1151
                    extent->l2_cache_counts[j] >>= 1;
B
bellard 已提交
1152 1153
                }
            }
F
Fam Zheng 已提交
1154
            l2_table = extent->l2_cache + (i * extent->l2_size);
B
bellard 已提交
1155 1156 1157 1158 1159 1160
            goto found;
        }
    }
    /* not found: load a new entry in the least used one */
    min_index = 0;
    min_count = 0xffffffff;
1161
    for (i = 0; i < L2_CACHE_SIZE; i++) {
F
Fam Zheng 已提交
1162 1163
        if (extent->l2_cache_counts[i] < min_count) {
            min_count = extent->l2_cache_counts[i];
B
bellard 已提交
1164 1165 1166
            min_index = i;
        }
    }
F
Fam Zheng 已提交
1167 1168 1169 1170 1171 1172 1173
    l2_table = extent->l2_cache + (min_index * extent->l2_size);
    if (bdrv_pread(
                extent->file,
                (int64_t)l2_offset * 512,
                l2_table,
                extent->l2_size * sizeof(uint32_t)
            ) != extent->l2_size * sizeof(uint32_t)) {
F
Fam Zheng 已提交
1174
        return VMDK_ERROR;
F
Fam Zheng 已提交
1175
    }
1176

F
Fam Zheng 已提交
1177 1178
    extent->l2_cache_offsets[min_index] = l2_offset;
    extent->l2_cache_counts[min_index] = 1;
B
bellard 已提交
1179
 found:
F
Fam Zheng 已提交
1180
    l2_index = ((offset >> 9) / extent->cluster_sectors) % extent->l2_size;
F
Fam Zheng 已提交
1181
    cluster_sector = le32_to_cpu(l2_table[l2_index]);
1182

F
Fam Zheng 已提交
1183 1184 1185 1186 1187 1188 1189
    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];
    }
F
Fam Zheng 已提交
1190
    if (extent->has_zero_grain && cluster_sector == VMDK_GTE_ZEROED) {
1191 1192 1193
        zeroed = true;
    }

F
Fam Zheng 已提交
1194
    if (!cluster_sector || zeroed) {
1195
        if (!allocate) {
1196
            return zeroed ? VMDK_ZEROED : VMDK_UNALLOC;
1197
        }
1198

F
Fam Zheng 已提交
1199 1200
        cluster_sector = extent->next_cluster_sector;
        extent->next_cluster_sector += extent->cluster_sectors;
1201 1202 1203 1204 1205 1206

        /* 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.
         */
F
Fam Zheng 已提交
1207 1208 1209 1210 1211 1212
        ret = get_whole_cluster(bs, extent,
                                cluster_sector,
                                offset >> BDRV_SECTOR_BITS,
                                skip_start_sector, skip_end_sector);
        if (ret) {
            return ret;
1213
        }
1214
    }
F
Fam Zheng 已提交
1215
    *cluster_offset = cluster_sector << BDRV_SECTOR_BITS;
F
Fam Zheng 已提交
1216
    return VMDK_OK;
B
bellard 已提交
1217 1218
}

F
Fam Zheng 已提交
1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235
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;
}

1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246
static inline uint64_t vmdk_find_index_in_cluster(VmdkExtent *extent,
                                                  int64_t sector_num)
{
    uint64_t index_in_cluster, extent_begin_sector, extent_relative_sector_num;

    extent_begin_sector = extent->end_sector - extent->sectors;
    extent_relative_sector_num = sector_num - extent_begin_sector;
    index_in_cluster = extent_relative_sector_num % extent->cluster_sectors;
    return index_in_cluster;
}

1247
static int64_t coroutine_fn vmdk_co_get_block_status(BlockDriverState *bs,
1248
        int64_t sector_num, int nb_sectors, int *pnum)
B
bellard 已提交
1249 1250
{
    BDRVVmdkState *s = bs->opaque;
F
Fam Zheng 已提交
1251 1252 1253 1254 1255 1256 1257 1258
    int64_t index_in_cluster, n, ret;
    uint64_t offset;
    VmdkExtent *extent;

    extent = find_extent(s, sector_num, NULL);
    if (!extent) {
        return 0;
    }
1259
    qemu_co_mutex_lock(&s->lock);
1260
    ret = get_cluster_offset(bs, extent, NULL,
F
Fam Zheng 已提交
1261 1262
                             sector_num * 512, false, &offset,
                             0, 0);
1263
    qemu_co_mutex_unlock(&s->lock);
1264

1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276
    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;
1277
        if (extent->file == bs->file && !extent->compressed) {
1278 1279 1280 1281 1282
            ret |= BDRV_BLOCK_OFFSET_VALID | offset;
        }

        break;
    }
1283

1284
    index_in_cluster = vmdk_find_index_in_cluster(extent, sector_num);
1285
    n = extent->cluster_sectors - index_in_cluster;
F
Fam Zheng 已提交
1286
    if (n > nb_sectors) {
B
bellard 已提交
1287
        n = nb_sectors;
F
Fam Zheng 已提交
1288
    }
B
bellard 已提交
1289
    *pnum = n;
F
Fam Zheng 已提交
1290
    return ret;
B
bellard 已提交
1291 1292
}

1293 1294 1295 1296 1297
static int vmdk_write_extent(VmdkExtent *extent, int64_t cluster_offset,
                            int64_t offset_in_cluster, const uint8_t *buf,
                            int nb_sectors, int64_t sector_num)
{
    int ret;
F
Fam Zheng 已提交
1298 1299
    VmdkGrainMarker *data = NULL;
    uLongf buf_len;
1300 1301
    const uint8_t *write_buf = buf;
    int write_len = nb_sectors * 512;
1302 1303
    int64_t write_offset;
    int64_t write_end_sector;
1304

F
Fam Zheng 已提交
1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321
    if (extent->compressed) {
        if (!extent->has_marker) {
            ret = -EINVAL;
            goto out;
        }
        buf_len = (extent->cluster_sectors << 9) * 2;
        data = g_malloc(buf_len + sizeof(VmdkGrainMarker));
        if (compress(data->data, &buf_len, buf, nb_sectors << 9) != Z_OK ||
                buf_len == 0) {
            ret = -EINVAL;
            goto out;
        }
        data->lba = sector_num;
        data->size = buf_len;
        write_buf = (uint8_t *)data;
        write_len = buf_len + sizeof(VmdkGrainMarker);
    }
1322 1323 1324 1325 1326 1327 1328 1329
    write_offset = cluster_offset + offset_in_cluster,
    ret = bdrv_pwrite(extent->file, write_offset, write_buf, write_len);

    write_end_sector = DIV_ROUND_UP(write_offset + write_len, BDRV_SECTOR_SIZE);

    extent->next_cluster_sector = MAX(extent->next_cluster_sector,
                                      write_end_sector);

1330 1331 1332 1333 1334 1335
    if (ret != write_len) {
        ret = ret < 0 ? ret : -EIO;
        goto out;
    }
    ret = 0;
 out:
F
Fam Zheng 已提交
1336
    g_free(data);
1337 1338 1339 1340 1341 1342 1343 1344
    return ret;
}

static int vmdk_read_extent(VmdkExtent *extent, int64_t cluster_offset,
                            int64_t offset_in_cluster, uint8_t *buf,
                            int nb_sectors)
{
    int ret;
F
Fam Zheng 已提交
1345 1346 1347 1348 1349 1350 1351
    int cluster_bytes, buf_bytes;
    uint8_t *cluster_buf, *compressed_data;
    uint8_t *uncomp_buf;
    uint32_t data_len;
    VmdkGrainMarker *marker;
    uLongf buf_len;

1352

F
Fam Zheng 已提交
1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367
    if (!extent->compressed) {
        ret = bdrv_pread(extent->file,
                          cluster_offset + offset_in_cluster,
                          buf, nb_sectors * 512);
        if (ret == nb_sectors * 512) {
            return 0;
        } else {
            return -EIO;
        }
    }
    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);
1368
    ret = bdrv_pread(extent->file,
F
Fam Zheng 已提交
1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395
                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 ||
            offset_in_cluster + nb_sectors * 512 > buf_len) {
        ret = -EINVAL;
        goto out;
1396
    }
F
Fam Zheng 已提交
1397 1398 1399 1400 1401 1402 1403
    memcpy(buf, uncomp_buf + offset_in_cluster, nb_sectors * 512);
    ret = 0;

 out:
    g_free(uncomp_buf);
    g_free(cluster_buf);
    return ret;
1404 1405
}

1406
static int vmdk_read(BlockDriverState *bs, int64_t sector_num,
B
bellard 已提交
1407 1408 1409
                    uint8_t *buf, int nb_sectors)
{
    BDRVVmdkState *s = bs->opaque;
F
Fam Zheng 已提交
1410 1411 1412
    int ret;
    uint64_t n, index_in_cluster;
    VmdkExtent *extent = NULL;
B
bellard 已提交
1413
    uint64_t cluster_offset;
1414

B
bellard 已提交
1415
    while (nb_sectors > 0) {
F
Fam Zheng 已提交
1416 1417 1418 1419
        extent = find_extent(s, sector_num, extent);
        if (!extent) {
            return -EIO;
        }
F
Fam Zheng 已提交
1420 1421 1422
        ret = get_cluster_offset(bs, extent, NULL,
                                 sector_num << 9, false, &cluster_offset,
                                 0, 0);
1423
        index_in_cluster = vmdk_find_index_in_cluster(extent, sector_num);
F
Fam Zheng 已提交
1424
        n = extent->cluster_sectors - index_in_cluster;
F
Fam Zheng 已提交
1425
        if (n > nb_sectors) {
B
bellard 已提交
1426
            n = nb_sectors;
F
Fam Zheng 已提交
1427
        }
1428
        if (ret != VMDK_OK) {
1429
            /* if not allocated, try to read from parent image, if exist */
1430
            if (bs->backing_hd && ret != VMDK_ZEROED) {
F
Fam Zheng 已提交
1431
                if (!vmdk_is_cid_valid(bs)) {
1432
                    return -EINVAL;
F
Fam Zheng 已提交
1433
                }
K
Kevin Wolf 已提交
1434
                ret = bdrv_read(bs->backing_hd, sector_num, buf, n);
F
Fam Zheng 已提交
1435
                if (ret < 0) {
1436
                    return ret;
F
Fam Zheng 已提交
1437
                }
1438 1439 1440
            } else {
                memset(buf, 0, 512 * n);
            }
B
bellard 已提交
1441
        } else {
1442 1443 1444 1445
            ret = vmdk_read_extent(extent,
                            cluster_offset, index_in_cluster * 512,
                            buf, n);
            if (ret) {
1446 1447
                return ret;
            }
B
bellard 已提交
1448 1449 1450 1451 1452 1453 1454 1455
        }
        nb_sectors -= n;
        sector_num += n;
        buf += n * 512;
    }
    return 0;
}

1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466
static coroutine_fn int vmdk_co_read(BlockDriverState *bs, int64_t sector_num,
                                     uint8_t *buf, int nb_sectors)
{
    int ret;
    BDRVVmdkState *s = bs->opaque;
    qemu_co_mutex_lock(&s->lock);
    ret = vmdk_read(bs, sector_num, buf, nb_sectors);
    qemu_co_mutex_unlock(&s->lock);
    return ret;
}

F
Fam Zheng 已提交
1467 1468 1469
/**
 * vmdk_write:
 * @zeroed:       buf is ignored (data is zero), use zeroed_grain GTE feature
1470 1471 1472 1473
 *                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 已提交
1474 1475 1476
 *
 * Returns: error code with 0 for success.
 */
1477
static int vmdk_write(BlockDriverState *bs, int64_t sector_num,
F
Fam Zheng 已提交
1478 1479
                      const uint8_t *buf, int nb_sectors,
                      bool zeroed, bool zero_dry_run)
B
bellard 已提交
1480
{
1481
    BDRVVmdkState *s = bs->opaque;
F
Fam Zheng 已提交
1482
    VmdkExtent *extent = NULL;
F
Fam Zheng 已提交
1483 1484
    int ret;
    int64_t index_in_cluster, n;
1485
    uint64_t cluster_offset;
F
Fam Zheng 已提交
1486
    VmdkMetaData m_data;
1487

1488
    if (sector_num > bs->total_sectors) {
F
Fam Zheng 已提交
1489
        error_report("Wrong offset: sector_num=0x%" PRIx64
1490
                " total_sectors=0x%" PRIx64 "\n",
1491
                sector_num, bs->total_sectors);
1492
        return -EIO;
1493 1494
    }

1495
    while (nb_sectors > 0) {
F
Fam Zheng 已提交
1496 1497 1498 1499
        extent = find_extent(s, sector_num, extent);
        if (!extent) {
            return -EIO;
        }
1500
        index_in_cluster = vmdk_find_index_in_cluster(extent, sector_num);
F
Fam Zheng 已提交
1501 1502 1503 1504 1505 1506 1507 1508
        n = extent->cluster_sectors - index_in_cluster;
        if (n > nb_sectors) {
            n = nb_sectors;
        }
        ret = get_cluster_offset(bs, extent, &m_data, sector_num << 9,
                                 !(extent->compressed || zeroed),
                                 &cluster_offset,
                                 index_in_cluster, index_in_cluster + n);
F
Fam Zheng 已提交
1509
        if (extent->compressed) {
F
Fam Zheng 已提交
1510
            if (ret == VMDK_OK) {
F
Fam Zheng 已提交
1511
                /* Refuse write to allocated cluster for streamOptimized */
F
Fam Zheng 已提交
1512 1513
                error_report("Could not write to allocated cluster"
                              " for streamOptimized");
F
Fam Zheng 已提交
1514 1515 1516
                return -EIO;
            } else {
                /* allocate */
F
Fam Zheng 已提交
1517 1518
                ret = get_cluster_offset(bs, extent, &m_data, sector_num << 9,
                                         true, &cluster_offset, 0, 0);
F
Fam Zheng 已提交
1519 1520
            }
        }
F
Fam Zheng 已提交
1521
        if (ret == VMDK_ERROR) {
1522
            return -EINVAL;
F
Fam Zheng 已提交
1523
        }
F
Fam Zheng 已提交
1524 1525 1526 1527 1528 1529 1530 1531
        if (zeroed) {
            /* Do zeroed write, buf is ignored */
            if (extent->has_zero_grain &&
                    index_in_cluster == 0 &&
                    n >= extent->cluster_sectors) {
                n = extent->cluster_sectors;
                if (!zero_dry_run) {
                    /* update L2 tables */
F
Fam Zheng 已提交
1532 1533
                    if (vmdk_L2update(extent, &m_data, VMDK_GTE_ZEROED)
                            != VMDK_OK) {
F
Fam Zheng 已提交
1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548
                        return -EIO;
                    }
                }
            } else {
                return -ENOTSUP;
            }
        } else {
            ret = vmdk_write_extent(extent,
                            cluster_offset, index_in_cluster * 512,
                            buf, n, sector_num);
            if (ret) {
                return ret;
            }
            if (m_data.valid) {
                /* update L2 tables */
F
Fam Zheng 已提交
1549 1550 1551
                if (vmdk_L2update(extent, &m_data,
                                  cluster_offset >> BDRV_SECTOR_BITS)
                        != VMDK_OK) {
F
Fam Zheng 已提交
1552 1553
                    return -EIO;
                }
F
Fam Zheng 已提交
1554
            }
1555
        }
1556 1557 1558
        nb_sectors -= n;
        sector_num += n;
        buf += n * 512;
1559

F
Fam Zheng 已提交
1560 1561
        /* update CID on the first write every time the virtual disk is
         * opened */
1562
        if (!s->cid_updated) {
F
Fam Zheng 已提交
1563
            ret = vmdk_write_cid(bs, g_random_int());
K
Kevin Wolf 已提交
1564 1565 1566
            if (ret < 0) {
                return ret;
            }
1567
            s->cid_updated = true;
1568
        }
1569 1570
    }
    return 0;
B
bellard 已提交
1571 1572
}

1573 1574 1575 1576 1577 1578
static coroutine_fn int vmdk_co_write(BlockDriverState *bs, int64_t sector_num,
                                      const uint8_t *buf, int nb_sectors)
{
    int ret;
    BDRVVmdkState *s = bs->opaque;
    qemu_co_mutex_lock(&s->lock);
F
Fam Zheng 已提交
1579 1580 1581 1582 1583
    ret = vmdk_write(bs, sector_num, buf, nb_sectors, false, false);
    qemu_co_mutex_unlock(&s->lock);
    return ret;
}

1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596
static int vmdk_write_compressed(BlockDriverState *bs,
                                 int64_t sector_num,
                                 const uint8_t *buf,
                                 int nb_sectors)
{
    BDRVVmdkState *s = bs->opaque;
    if (s->num_extents == 1 && s->extents[0].compressed) {
        return vmdk_write(bs, sector_num, buf, nb_sectors, false, false);
    } else {
        return -ENOTSUP;
    }
}

F
Fam Zheng 已提交
1597 1598
static int coroutine_fn vmdk_co_write_zeroes(BlockDriverState *bs,
                                             int64_t sector_num,
1599 1600
                                             int nb_sectors,
                                             BdrvRequestFlags flags)
F
Fam Zheng 已提交
1601 1602 1603 1604
{
    int ret;
    BDRVVmdkState *s = bs->opaque;
    qemu_co_mutex_lock(&s->lock);
1605 1606
    /* write zeroes could fail if sectors not aligned to cluster, test it with
     * dry_run == true before really updating image */
F
Fam Zheng 已提交
1607 1608 1609 1610
    ret = vmdk_write(bs, sector_num, NULL, nb_sectors, true, true);
    if (!ret) {
        ret = vmdk_write(bs, sector_num, NULL, nb_sectors, true, false);
    }
1611 1612 1613 1614
    qemu_co_mutex_unlock(&s->lock);
    return ret;
}

1615
static int vmdk_create_extent(const char *filename, int64_t filesize,
1616
                              bool flat, bool compress, bool zeroed_grain,
1617
                              QemuOpts *opts, Error **errp)
1618
{
F
Fam Zheng 已提交
1619
    int ret, i;
1620
    BlockDriverState *bs = NULL;
1621
    VMDK4Header header;
F
Fam Zheng 已提交
1622
    Error *local_err = NULL;
1623 1624 1625
    uint32_t tmp, magic, grains, gd_sectors, gt_size, gt_count;
    uint32_t *gd_buf = NULL;
    int gd_buf_size;
1626

1627
    ret = bdrv_create_file(filename, opts, &local_err);
1628 1629 1630
    if (ret < 0) {
        error_propagate(errp, local_err);
        goto exit;
1631
    }
1632

M
Max Reitz 已提交
1633 1634 1635
    assert(bs == NULL);
    ret = bdrv_open(&bs, filename, NULL, NULL, BDRV_O_RDWR | BDRV_O_PROTOCOL,
                    NULL, &local_err);
1636 1637 1638 1639 1640
    if (ret < 0) {
        error_propagate(errp, local_err);
        goto exit;
    }

F
Fam Zheng 已提交
1641
    if (flat) {
1642
        ret = bdrv_truncate(bs, filesize);
F
Fam Zheng 已提交
1643
        if (ret < 0) {
1644
            error_setg_errno(errp, -ret, "Could not truncate file");
F
Fam Zheng 已提交
1645 1646
        }
        goto exit;
1647
    }
1648 1649
    magic = cpu_to_be32(VMDK4_MAGIC);
    memset(&header, 0, sizeof(header));
1650
    header.version = zeroed_grain ? 2 : 1;
F
Fam Zheng 已提交
1651
    header.flags = VMDK4_FLAG_RGD | VMDK4_FLAG_NL_DETECT
1652 1653
                   | (compress ? VMDK4_FLAG_COMPRESS | VMDK4_FLAG_MARKER : 0)
                   | (zeroed_grain ? VMDK4_FLAG_ZERO_GRAIN : 0);
1654
    header.compressAlgorithm = compress ? VMDK4_COMPRESSION_DEFLATE : 0;
1655
    header.capacity = filesize / BDRV_SECTOR_SIZE;
A
Alexander Graf 已提交
1656
    header.granularity = 128;
1657
    header.num_gtes_per_gt = BDRV_SECTOR_SIZE;
1658

1659 1660 1661 1662 1663
    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);
1664 1665 1666 1667

    header.desc_offset = 1;
    header.desc_size = 20;
    header.rgd_offset = header.desc_offset + header.desc_size;
1668
    header.gd_offset = header.rgd_offset + gd_sectors + (gt_size * gt_count);
1669
    header.grain_offset =
1670 1671
        ROUND_UP(header.gd_offset + gd_sectors + (gt_size * gt_count),
                 header.granularity);
A
Alexander Graf 已提交
1672 1673 1674 1675 1676
    /* 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);
1677
    header.num_gtes_per_gt = cpu_to_le32(header.num_gtes_per_gt);
1678 1679 1680 1681 1682
    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);
1683
    header.compressAlgorithm = cpu_to_le16(header.compressAlgorithm);
1684 1685 1686 1687 1688

    header.check_bytes[0] = 0xa;
    header.check_bytes[1] = 0x20;
    header.check_bytes[2] = 0xd;
    header.check_bytes[3] = 0xa;
1689 1690

    /* write all the data */
1691 1692
    ret = bdrv_pwrite(bs, 0, &magic, sizeof(magic));
    if (ret < 0) {
1693
        error_setg(errp, QERR_IO_ERROR);
1694 1695
        goto exit;
    }
1696 1697
    ret = bdrv_pwrite(bs, sizeof(magic), &header, sizeof(header));
    if (ret < 0) {
1698
        error_setg(errp, QERR_IO_ERROR);
1699 1700
        goto exit;
    }
1701

1702
    ret = bdrv_truncate(bs, le64_to_cpu(header.grain_offset) << 9);
1703
    if (ret < 0) {
1704
        error_setg_errno(errp, -ret, "Could not truncate file");
1705 1706
        goto exit;
    }
1707 1708

    /* write grain directory */
1709 1710 1711
    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;
1712
         i < gt_count; i++, tmp += gt_size) {
1713 1714 1715 1716 1717
        gd_buf[i] = cpu_to_le32(tmp);
    }
    ret = bdrv_pwrite(bs, le64_to_cpu(header.rgd_offset) * BDRV_SECTOR_SIZE,
                      gd_buf, gd_buf_size);
    if (ret < 0) {
1718
        error_setg(errp, QERR_IO_ERROR);
1719
        goto exit;
1720
    }
1721

1722
    /* write backup grain directory */
1723
    for (i = 0, tmp = le64_to_cpu(header.gd_offset) + gd_sectors;
1724
         i < gt_count; i++, tmp += gt_size) {
1725 1726 1727 1728 1729
        gd_buf[i] = cpu_to_le32(tmp);
    }
    ret = bdrv_pwrite(bs, le64_to_cpu(header.gd_offset) * BDRV_SECTOR_SIZE,
                      gd_buf, gd_buf_size);
    if (ret < 0) {
1730
        error_setg(errp, QERR_IO_ERROR);
1731
        goto exit;
1732
    }
1733

F
Fam Zheng 已提交
1734
    ret = 0;
1735 1736 1737 1738 1739
exit:
    if (bs) {
        bdrv_unref(bs);
    }
    g_free(gd_buf);
F
Fam Zheng 已提交
1740 1741 1742 1743
    return ret;
}

static int filename_decompose(const char *filename, char *path, char *prefix,
F
Fam Zheng 已提交
1744
                              char *postfix, size_t buf_len, Error **errp)
F
Fam Zheng 已提交
1745 1746 1747 1748
{
    const char *p, *q;

    if (filename == NULL || !strlen(filename)) {
F
Fam Zheng 已提交
1749
        error_setg(errp, "No filename provided");
F
Fam Zheng 已提交
1750
        return VMDK_ERROR;
F
Fam Zheng 已提交
1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761
    }
    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 已提交
1762
            return VMDK_ERROR;
F
Fam Zheng 已提交
1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774
        }
        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 已提交
1775
            return VMDK_ERROR;
F
Fam Zheng 已提交
1776 1777 1778 1779
        }
        pstrcpy(prefix, q - p + 1, p);
        pstrcpy(postfix, buf_len, q);
    }
F
Fam Zheng 已提交
1780
    return VMDK_OK;
F
Fam Zheng 已提交
1781 1782
}

1783
static int vmdk_create(const char *filename, QemuOpts *opts, Error **errp)
F
Fam Zheng 已提交
1784
{
1785 1786
    int idx = 0;
    BlockDriverState *new_bs = NULL;
F
Fam Zheng 已提交
1787
    Error *local_err = NULL;
1788
    char *desc = NULL;
F
Fam Zheng 已提交
1789
    int64_t total_size = 0, filesize;
1790 1791 1792
    char *adapter_type = NULL;
    char *backing_file = NULL;
    char *fmt = NULL;
F
Fam Zheng 已提交
1793 1794
    int flags = 0;
    int ret = 0;
1795
    bool flat, split, compress;
1796
    GString *ext_desc_lines;
1797 1798 1799 1800 1801 1802
    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);
F
Fam Zheng 已提交
1803 1804
    const int64_t split_size = 0x80000000;  /* VMDK has constant split size */
    const char *desc_extent_line;
1805
    char *parent_desc_line = g_malloc0(BUF_SIZE);
F
Fam Zheng 已提交
1806
    uint32_t parent_cid = 0xffffffff;
1807
    uint32_t number_heads = 16;
1808
    bool zeroed_grain = false;
1809
    uint32_t desc_offset = 0, desc_len;
F
Fam Zheng 已提交
1810 1811 1812
    const char desc_template[] =
        "# Disk DescriptorFile\n"
        "version=1\n"
1813 1814
        "CID=%" PRIx32 "\n"
        "parentCID=%" PRIx32 "\n"
F
Fam Zheng 已提交
1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825
        "createType=\"%s\"\n"
        "%s"
        "\n"
        "# Extent description\n"
        "%s"
        "\n"
        "# The Disk Data Base\n"
        "#DDB\n"
        "\n"
        "ddb.virtualHWVersion = \"%d\"\n"
        "ddb.geometry.cylinders = \"%" PRId64 "\"\n"
1826
        "ddb.geometry.heads = \"%" PRIu32 "\"\n"
F
Fam Zheng 已提交
1827
        "ddb.geometry.sectors = \"63\"\n"
1828
        "ddb.adapterType = \"%s\"\n";
F
Fam Zheng 已提交
1829

1830 1831
    ext_desc_lines = g_string_new(NULL);

F
Fam Zheng 已提交
1832
    if (filename_decompose(filename, path, prefix, postfix, PATH_MAX, errp)) {
1833 1834
        ret = -EINVAL;
        goto exit;
F
Fam Zheng 已提交
1835 1836
    }
    /* Read out options */
1837 1838
    total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
                          BDRV_SECTOR_SIZE);
1839 1840 1841 1842 1843 1844 1845 1846
    adapter_type = qemu_opt_get_del(opts, BLOCK_OPT_ADAPTER_TYPE);
    backing_file = qemu_opt_get_del(opts, BLOCK_OPT_BACKING_FILE);
    if (qemu_opt_get_bool_del(opts, BLOCK_OPT_COMPAT6, false)) {
        flags |= BLOCK_FLAG_COMPAT6;
    }
    fmt = qemu_opt_get_del(opts, BLOCK_OPT_SUBFMT);
    if (qemu_opt_get_bool_del(opts, BLOCK_OPT_ZEROED_GRAIN, false)) {
        zeroed_grain = true;
F
Fam Zheng 已提交
1847
    }
1848

1849
    if (!adapter_type) {
1850
        adapter_type = g_strdup("ide");
1851 1852 1853 1854
    } else if (strcmp(adapter_type, "ide") &&
               strcmp(adapter_type, "buslogic") &&
               strcmp(adapter_type, "lsilogic") &&
               strcmp(adapter_type, "legacyESX")) {
F
Fam Zheng 已提交
1855
        error_setg(errp, "Unknown adapter type: '%s'", adapter_type);
1856 1857
        ret = -EINVAL;
        goto exit;
1858 1859 1860 1861 1862 1863
    }
    if (strcmp(adapter_type, "ide") != 0) {
        /* 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;
    }
F
Fam Zheng 已提交
1864 1865
    if (!fmt) {
        /* Default format to monolithicSparse */
1866
        fmt = g_strdup("monolithicSparse");
F
Fam Zheng 已提交
1867 1868 1869
    } else if (strcmp(fmt, "monolithicFlat") &&
               strcmp(fmt, "monolithicSparse") &&
               strcmp(fmt, "twoGbMaxExtentSparse") &&
1870 1871
               strcmp(fmt, "twoGbMaxExtentFlat") &&
               strcmp(fmt, "streamOptimized")) {
F
Fam Zheng 已提交
1872
        error_setg(errp, "Unknown subformat: '%s'", fmt);
1873 1874
        ret = -EINVAL;
        goto exit;
F
Fam Zheng 已提交
1875 1876 1877 1878 1879
    }
    split = !(strcmp(fmt, "twoGbMaxExtentFlat") &&
              strcmp(fmt, "twoGbMaxExtentSparse"));
    flat = !(strcmp(fmt, "monolithicFlat") &&
             strcmp(fmt, "twoGbMaxExtentFlat"));
1880
    compress = !strcmp(fmt, "streamOptimized");
F
Fam Zheng 已提交
1881
    if (flat) {
1882
        desc_extent_line = "RW %" PRId64 " FLAT \"%s\" 0\n";
F
Fam Zheng 已提交
1883
    } else {
1884
        desc_extent_line = "RW %" PRId64 " SPARSE \"%s\"\n";
F
Fam Zheng 已提交
1885 1886
    }
    if (flat && backing_file) {
F
Fam Zheng 已提交
1887
        error_setg(errp, "Flat image can't have backing file");
1888 1889
        ret = -ENOTSUP;
        goto exit;
F
Fam Zheng 已提交
1890
    }
1891 1892
    if (flat && zeroed_grain) {
        error_setg(errp, "Flat image can't enable zeroed grain");
1893 1894
        ret = -ENOTSUP;
        goto exit;
1895
    }
F
Fam Zheng 已提交
1896
    if (backing_file) {
1897
        BlockDriverState *bs = NULL;
1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908
        char *full_backing = g_new0(char, PATH_MAX);
        bdrv_get_full_backing_filename_from_filename(filename, backing_file,
                                                     full_backing, PATH_MAX,
                                                     &local_err);
        if (local_err) {
            g_free(full_backing);
            error_propagate(errp, local_err);
            ret = -ENOENT;
            goto exit;
        }
        ret = bdrv_open(&bs, full_backing, NULL, NULL, BDRV_O_NO_BACKING, NULL,
1909
                        errp);
1910
        g_free(full_backing);
F
Fam Zheng 已提交
1911
        if (ret != 0) {
1912
            goto exit;
F
Fam Zheng 已提交
1913 1914
        }
        if (strcmp(bs->drv->format_name, "vmdk")) {
F
Fam Zheng 已提交
1915
            bdrv_unref(bs);
1916 1917
            ret = -EINVAL;
            goto exit;
F
Fam Zheng 已提交
1918 1919
        }
        parent_cid = vmdk_read_cid(bs, 0);
F
Fam Zheng 已提交
1920
        bdrv_unref(bs);
1921
        snprintf(parent_desc_line, BUF_SIZE,
1922
                "parentFileNameHint=\"%s\"", backing_file);
F
Fam Zheng 已提交
1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933
    }

    /* Create extents */
    filesize = total_size;
    while (filesize > 0) {
        int64_t size = filesize;

        if (split && size > split_size) {
            size = split_size;
        }
        if (split) {
1934
            snprintf(desc_filename, PATH_MAX, "%s-%c%03d%s",
F
Fam Zheng 已提交
1935 1936
                    prefix, flat ? 'f' : 's', ++idx, postfix);
        } else if (flat) {
1937
            snprintf(desc_filename, PATH_MAX, "%s-flat%s", prefix, postfix);
F
Fam Zheng 已提交
1938
        } else {
1939
            snprintf(desc_filename, PATH_MAX, "%s%s", prefix, postfix);
F
Fam Zheng 已提交
1940
        }
1941
        snprintf(ext_filename, PATH_MAX, "%s%s", path, desc_filename);
F
Fam Zheng 已提交
1942

1943
        if (vmdk_create_extent(ext_filename, size,
1944
                               flat, compress, zeroed_grain, opts, errp)) {
1945 1946
            ret = -EINVAL;
            goto exit;
F
Fam Zheng 已提交
1947 1948 1949 1950
        }
        filesize -= size;

        /* Format description line */
1951
        snprintf(desc_line, BUF_SIZE,
1952
                    desc_extent_line, size / BDRV_SECTOR_SIZE, desc_filename);
1953
        g_string_append(ext_desc_lines, desc_line);
F
Fam Zheng 已提交
1954 1955
    }
    /* generate descriptor file */
1956
    desc = g_strdup_printf(desc_template,
F
Fam Zheng 已提交
1957
                           g_random_int(),
1958 1959 1960 1961 1962
                           parent_cid,
                           fmt,
                           parent_desc_line,
                           ext_desc_lines->str,
                           (flags & BLOCK_FLAG_COMPAT6 ? 6 : 4),
1963 1964
                           total_size /
                               (int64_t)(63 * number_heads * BDRV_SECTOR_SIZE),
1965 1966
                           number_heads,
                           adapter_type);
1967 1968 1969 1970
    desc_len = strlen(desc);
    /* the descriptor offset = 0x200 */
    if (!split && !flat) {
        desc_offset = 0x200;
F
Fam Zheng 已提交
1971
    } else {
C
Chunyan Liu 已提交
1972
        ret = bdrv_create_file(filename, opts, &local_err);
1973
        if (ret < 0) {
F
Fam Zheng 已提交
1974
            error_propagate(errp, local_err);
1975 1976
            goto exit;
        }
F
Fam Zheng 已提交
1977
    }
M
Max Reitz 已提交
1978 1979 1980
    assert(new_bs == NULL);
    ret = bdrv_open(&new_bs, filename, NULL, NULL,
                    BDRV_O_RDWR | BDRV_O_PROTOCOL, NULL, &local_err);
1981
    if (ret < 0) {
F
Fam Zheng 已提交
1982
        error_propagate(errp, local_err);
1983
        goto exit;
F
Fam Zheng 已提交
1984
    }
1985 1986 1987 1988
    ret = bdrv_pwrite(new_bs, desc_offset, desc, desc_len);
    if (ret < 0) {
        error_setg_errno(errp, -ret, "Could not write description");
        goto exit;
F
Fam Zheng 已提交
1989
    }
1990 1991 1992 1993 1994
    /* bdrv_pwrite write padding zeros to align to sector, we don't need that
     * for description file */
    if (desc_offset == 0) {
        ret = bdrv_truncate(new_bs, desc_len);
        if (ret < 0) {
1995
            error_setg_errno(errp, -ret, "Could not truncate file");
1996
        }
1997
    }
1998
exit:
1999 2000 2001
    if (new_bs) {
        bdrv_unref(new_bs);
    }
2002 2003 2004
    g_free(adapter_type);
    g_free(backing_file);
    g_free(fmt);
2005
    g_free(desc);
2006 2007 2008 2009 2010 2011 2012
    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);
2013
    g_string_free(ext_desc_lines, true);
2014
    return ret;
2015 2016
}

B
bellard 已提交
2017
static void vmdk_close(BlockDriverState *bs)
B
bellard 已提交
2018
{
K
Kevin Wolf 已提交
2019 2020
    BDRVVmdkState *s = bs->opaque;

F
Fam Zheng 已提交
2021
    vmdk_free_extents(bs);
F
Fam Zheng 已提交
2022
    g_free(s->create_type);
K
Kevin Wolf 已提交
2023 2024 2025

    migrate_del_blocker(s->migration_blocker);
    error_free(s->migration_blocker);
B
bellard 已提交
2026 2027
}

P
Paolo Bonzini 已提交
2028
static coroutine_fn int vmdk_co_flush(BlockDriverState *bs)
P
pbrook 已提交
2029
{
F
Fam Zheng 已提交
2030
    BDRVVmdkState *s = bs->opaque;
2031 2032
    int i, err;
    int ret = 0;
F
Fam Zheng 已提交
2033 2034

    for (i = 0; i < s->num_extents; i++) {
P
Paolo Bonzini 已提交
2035
        err = bdrv_co_flush(s->extents[i].file);
F
Fam Zheng 已提交
2036 2037 2038 2039 2040
        if (err < 0) {
            ret = err;
        }
    }
    return ret;
P
pbrook 已提交
2041 2042
}

2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065
static int64_t vmdk_get_allocated_file_size(BlockDriverState *bs)
{
    int i;
    int64_t ret = 0;
    int64_t r;
    BDRVVmdkState *s = bs->opaque;

    ret = bdrv_get_allocated_file_size(bs->file);
    if (ret < 0) {
        return ret;
    }
    for (i = 0; i < s->num_extents; i++) {
        if (s->extents[i].file == bs->file) {
            continue;
        }
        r = bdrv_get_allocated_file_size(s->extents[i].file);
        if (r < 0) {
            return r;
        }
        ret += r;
    }
    return ret;
}
2066

F
Fam Zheng 已提交
2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083
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) {
            if (!bdrv_has_zero_init(s->extents[i].file)) {
                return 0;
            }
        }
    }
    return 1;
}

F
Fam Zheng 已提交
2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100
static ImageInfo *vmdk_get_extent_info(VmdkExtent *extent)
{
    ImageInfo *info = g_new0(ImageInfo, 1);

    *info = (ImageInfo){
        .filename         = g_strdup(extent->file->filename),
        .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;
}

2101 2102 2103 2104 2105 2106
static int vmdk_check(BlockDriverState *bs, BdrvCheckResult *result,
                      BdrvCheckMode fix)
{
    BDRVVmdkState *s = bs->opaque;
    VmdkExtent *extent = NULL;
    int64_t sector_num = 0;
2107
    int64_t total_sectors = bdrv_nb_sectors(bs);
2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127
    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);
            break;
        }
        ret = get_cluster_offset(bs, extent, NULL,
                                 sector_num << BDRV_SECTOR_BITS,
F
Fam Zheng 已提交
2128
                                 false, &cluster_offset, 0, 0);
2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147
        if (ret == VMDK_ERROR) {
            fprintf(stderr,
                    "ERROR: could not get cluster_offset for sector %"
                    PRId64 "\n", sector_num);
            break;
        }
        if (ret == VMDK_OK && cluster_offset >= bdrv_getlength(extent->file)) {
            fprintf(stderr,
                    "ERROR: cluster offset for sector %"
                    PRId64 " points after EOF\n", sector_num);
            break;
        }
        sector_num += extent->cluster_sectors;
    }

    result->corruptions++;
    return 0;
}

F
Fam Zheng 已提交
2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178
static ImageInfoSpecific *vmdk_get_specific_info(BlockDriverState *bs)
{
    int i;
    BDRVVmdkState *s = bs->opaque;
    ImageInfoSpecific *spec_info = g_new0(ImageInfoSpecific, 1);
    ImageInfoList **next;

    *spec_info = (ImageInfoSpecific){
        .kind = IMAGE_INFO_SPECIFIC_KIND_VMDK,
        {
            .vmdk = g_new0(ImageInfoSpecificVmdk, 1),
        },
    };

    *spec_info->vmdk = (ImageInfoSpecificVmdk) {
        .create_type = g_strdup(s->create_type),
        .cid = s->cid,
        .parent_cid = s->parent_cid,
    };

    next = &spec_info->vmdk->extents;
    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;
}

2179 2180 2181 2182 2183 2184 2185
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 已提交
2186 2187 2188 2189 2190
static int vmdk_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
{
    int i;
    BDRVVmdkState *s = bs->opaque;
    assert(s->num_extents);
2191

F
Fam Zheng 已提交
2192 2193
    /* See if we have multiple extents but they have different cases */
    for (i = 1; i < s->num_extents; i++) {
2194
        if (!vmdk_extents_type_eq(&s->extents[0], &s->extents[i])) {
F
Fam Zheng 已提交
2195 2196 2197
            return -ENOTSUP;
        }
    }
2198 2199 2200 2201
    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 已提交
2202 2203 2204
    return 0;
}

2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225
static void vmdk_detach_aio_context(BlockDriverState *bs)
{
    BDRVVmdkState *s = bs->opaque;
    int i;

    for (i = 0; i < s->num_extents; i++) {
        bdrv_detach_aio_context(s->extents[i].file);
    }
}

static void vmdk_attach_aio_context(BlockDriverState *bs,
                                    AioContext *new_context)
{
    BDRVVmdkState *s = bs->opaque;
    int i;

    for (i = 0; i < s->num_extents; i++) {
        bdrv_attach_aio_context(s->extents[i].file, new_context);
    }
}

2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266
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"
        },
        {
            .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 */ }
    }
2267 2268
};

2269
static BlockDriver bdrv_vmdk = {
F
Fam Zheng 已提交
2270 2271 2272 2273
    .format_name                  = "vmdk",
    .instance_size                = sizeof(BDRVVmdkState),
    .bdrv_probe                   = vmdk_probe,
    .bdrv_open                    = vmdk_open,
2274
    .bdrv_check                   = vmdk_check,
F
Fam Zheng 已提交
2275 2276 2277
    .bdrv_reopen_prepare          = vmdk_reopen_prepare,
    .bdrv_read                    = vmdk_co_read,
    .bdrv_write                   = vmdk_co_write,
2278
    .bdrv_write_compressed        = vmdk_write_compressed,
F
Fam Zheng 已提交
2279 2280
    .bdrv_co_write_zeroes         = vmdk_co_write_zeroes,
    .bdrv_close                   = vmdk_close,
C
Chunyan Liu 已提交
2281
    .bdrv_create                  = vmdk_create,
F
Fam Zheng 已提交
2282
    .bdrv_co_flush_to_disk        = vmdk_co_flush,
2283
    .bdrv_co_get_block_status     = vmdk_co_get_block_status,
F
Fam Zheng 已提交
2284 2285
    .bdrv_get_allocated_file_size = vmdk_get_allocated_file_size,
    .bdrv_has_zero_init           = vmdk_has_zero_init,
F
Fam Zheng 已提交
2286
    .bdrv_get_specific_info       = vmdk_get_specific_info,
2287
    .bdrv_refresh_limits          = vmdk_refresh_limits,
F
Fam Zheng 已提交
2288
    .bdrv_get_info                = vmdk_get_info,
2289 2290
    .bdrv_detach_aio_context      = vmdk_detach_aio_context,
    .bdrv_attach_aio_context      = vmdk_attach_aio_context,
F
Fam Zheng 已提交
2291

2292
    .supports_backing             = true,
2293
    .create_opts                  = &vmdk_create_opts,
B
bellard 已提交
2294
};
2295 2296 2297 2298 2299 2300 2301

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

block_init(bdrv_vmdk_init);