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 "qemu/module.h"
29
#include "migration/migration.h"
S
Stefan Weil 已提交
30
#include <zlib.h>
F
Fam Zheng 已提交
31
#include <glib.h>
B
bellard 已提交
32 33 34

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

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

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

53 54
#define BLOCK_OPT_ZEROED_GRAIN "zeroed_grain"

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

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

#define L2_CACHE_SIZE 16

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

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

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

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

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

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

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

F
Fam Zheng 已提交
153
    if (buf_size < 4) {
B
bellard 已提交
154
        return 0;
F
Fam Zheng 已提交
155
    }
B
bellard 已提交
156 157
    magic = be32_to_cpu(*(uint32_t *)buf);
    if (magic == VMDK3_MAGIC ||
158
        magic == VMDK4_MAGIC) {
B
bellard 已提交
159
        return 100;
160 161 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
    } 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 已提交
201
        return 0;
202
    }
B
bellard 已提交
203 204
}

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

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

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

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

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

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

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

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

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

    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;
276
    BDRVVmdkState *s = bs->opaque;
K
Kevin Wolf 已提交
277
    int ret;
278

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

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

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

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

303 304 305 306 307 308
    return 0;
}

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

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

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

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

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

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

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

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

    return 0;
}

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

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

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

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

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

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

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

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

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

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

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

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

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

P
Paolo Bonzini 已提交
524 525 526 527 528 529 530 531 532 533 534 535 536
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;
    }

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

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

    return buf;
}

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

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

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

595 596 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
    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) {
625
            error_setg_errno(errp, -ret, "Failed to read footer");
626 627 628 629 630 631 632 633 634 635 636
            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)
        {
637
            error_setg(errp, "Invalid footer");
638 639 640 641 642 643
            return -EINVAL;
        }

        header = footer.header;
    }

644
    if (le32_to_cpu(header.version) > 3) {
645
        char buf[64];
646
        snprintf(buf, sizeof(buf), "VMDK version %" PRId32,
647
                 le32_to_cpu(header.version));
P
Paolo Bonzini 已提交
648
        error_set(errp, QERR_UNKNOWN_BLOCK_FORMAT_FEATURE,
649
                  bdrv_get_device_or_node_name(bs), "vmdk", buf);
650
        return -ENOTSUP;
651 652 653 654 655 656 657
    } 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;
658 659
    }

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

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

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

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

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

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

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

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

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

816 817 818 819 820 821 822 823
        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;
        }

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

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

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

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

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

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

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

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

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

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

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

    /* Disable migration when VMDK images are used */
949 950 951
    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 已提交
952
    migrate_add_blocker(s->migration_blocker);
953
    g_free(buf);
K
Kevin Wolf 已提交
954
    return 0;
P
Paolo Bonzini 已提交
955 956

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

964

965
static void vmdk_refresh_limits(BlockDriverState *bs, Error **errp)
966 967 968 969 970 971 972 973 974 975 976 977 978
{
    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 已提交
979 980 981 982 983 984 985 986 987 988
/**
 * 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 已提交
989
static int get_whole_cluster(BlockDriverState *bs,
F
Fam Zheng 已提交
990 991 992 993 994
                             VmdkExtent *extent,
                             uint64_t cluster_sector_num,
                             uint64_t sector_num,
                             uint64_t skip_start_sector,
                             uint64_t skip_end_sector)
995
{
996
    int ret = VMDK_OK;
F
Fam Zheng 已提交
997 998 999 1000 1001 1002 1003
    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);
1004

F
Fam Zheng 已提交
1005 1006 1007 1008 1009 1010 1011
    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);
1012 1013
    /* 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 已提交
1014 1015 1016 1017
    if (bs->backing_hd && !vmdk_is_cid_valid(bs)) {
        ret = VMDK_ERROR;
        goto exit;
    }
1018

F
Fam Zheng 已提交
1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030
    /* 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 已提交
1031
        if (ret < 0) {
1032 1033
            ret = VMDK_ERROR;
            goto exit;
K
Kevin Wolf 已提交
1034
        }
F
Fam Zheng 已提交
1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049
    }
    /* 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 已提交
1050
        if (ret < 0) {
1051 1052
            ret = VMDK_ERROR;
            goto exit;
1053 1054
        }
    }
F
Fam Zheng 已提交
1055

1056 1057 1058
exit:
    qemu_vfree(whole_grain);
    return ret;
1059 1060
}

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

F
Fam Zheng 已提交
1088
    return VMDK_OK;
1089 1090
}

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

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

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

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

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

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

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

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

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

1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244
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;
}

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

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

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

        break;
    }
1281

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

1291 1292 1293 1294 1295
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 已提交
1296 1297
    VmdkGrainMarker *data = NULL;
    uLongf buf_len;
1298 1299
    const uint8_t *write_buf = buf;
    int write_len = nb_sectors * 512;
1300 1301
    int64_t write_offset;
    int64_t write_end_sector;
1302

F
Fam Zheng 已提交
1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319
    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);
    }
1320 1321 1322 1323 1324 1325 1326 1327
    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);

1328 1329 1330 1331 1332 1333
    if (ret != write_len) {
        ret = ret < 0 ? ret : -EIO;
        goto out;
    }
    ret = 0;
 out:
F
Fam Zheng 已提交
1334
    g_free(data);
1335 1336 1337 1338 1339 1340 1341 1342
    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 已提交
1343 1344 1345 1346 1347 1348 1349
    int cluster_bytes, buf_bytes;
    uint8_t *cluster_buf, *compressed_data;
    uint8_t *uncomp_buf;
    uint32_t data_len;
    VmdkGrainMarker *marker;
    uLongf buf_len;

1350

F
Fam Zheng 已提交
1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365
    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);
1366
    ret = bdrv_pread(extent->file,
F
Fam Zheng 已提交
1367 1368 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
                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;
1394
    }
F
Fam Zheng 已提交
1395 1396 1397 1398 1399 1400 1401
    memcpy(buf, uncomp_buf + offset_in_cluster, nb_sectors * 512);
    ret = 0;

 out:
    g_free(uncomp_buf);
    g_free(cluster_buf);
    return ret;
1402 1403
}

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

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

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

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

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

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

1571 1572 1573 1574 1575 1576
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 已提交
1577 1578 1579 1580 1581
    ret = vmdk_write(bs, sector_num, buf, nb_sectors, false, false);
    qemu_co_mutex_unlock(&s->lock);
    return ret;
}

1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594
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 已提交
1595 1596
static int coroutine_fn vmdk_co_write_zeroes(BlockDriverState *bs,
                                             int64_t sector_num,
1597 1598
                                             int nb_sectors,
                                             BdrvRequestFlags flags)
F
Fam Zheng 已提交
1599 1600 1601 1602
{
    int ret;
    BDRVVmdkState *s = bs->opaque;
    qemu_co_mutex_lock(&s->lock);
1603 1604
    /* write zeroes could fail if sectors not aligned to cluster, test it with
     * dry_run == true before really updating image */
F
Fam Zheng 已提交
1605 1606 1607 1608
    ret = vmdk_write(bs, sector_num, NULL, nb_sectors, true, true);
    if (!ret) {
        ret = vmdk_write(bs, sector_num, NULL, nb_sectors, true, false);
    }
1609 1610 1611 1612
    qemu_co_mutex_unlock(&s->lock);
    return ret;
}

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

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

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

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

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

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

    header.check_bytes[0] = 0xa;
    header.check_bytes[1] = 0x20;
    header.check_bytes[2] = 0xd;
    header.check_bytes[3] = 0xa;
1687 1688

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

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

    /* write grain directory */
1707 1708 1709
    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;
1710
         i < gt_count; i++, tmp += gt_size) {
1711 1712 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) {
        error_set(errp, QERR_IO_ERROR);
        goto exit;
1718
    }
1719

1720
    /* write backup grain directory */
1721
    for (i = 0, tmp = le64_to_cpu(header.gd_offset) + gd_sectors;
1722
         i < gt_count; i++, tmp += gt_size) {
1723 1724 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) {
        error_set(errp, QERR_IO_ERROR);
        goto exit;
1730
    }
1731

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

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

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

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

1828 1829
    ext_desc_lines = g_string_new(NULL);

F
Fam Zheng 已提交
1830
    if (filename_decompose(filename, path, prefix, postfix, PATH_MAX, errp)) {
1831 1832
        ret = -EINVAL;
        goto exit;
F
Fam Zheng 已提交
1833 1834
    }
    /* Read out options */
1835 1836
    total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
                          BDRV_SECTOR_SIZE);
1837 1838 1839 1840 1841 1842 1843 1844
    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 已提交
1845
    }
1846

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

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

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

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

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

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

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

    migrate_del_blocker(s->migration_blocker);
    error_free(s->migration_blocker);
B
bellard 已提交
2024 2025
}

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

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

2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063
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;
}
2064

F
Fam Zheng 已提交
2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081
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 已提交
2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098
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;
}

2099 2100 2101 2102 2103 2104
static int vmdk_check(BlockDriverState *bs, BdrvCheckResult *result,
                      BdrvCheckMode fix)
{
    BDRVVmdkState *s = bs->opaque;
    VmdkExtent *extent = NULL;
    int64_t sector_num = 0;
2105
    int64_t total_sectors = bdrv_nb_sectors(bs);
2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125
    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 已提交
2126
                                 false, &cluster_offset, 0, 0);
2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145
        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 已提交
2146 2147 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
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;
}

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

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

2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223
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);
    }
}

2224 2225 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
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 */ }
    }
2265 2266
};

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

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

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

block_init(bdrv_vmdk_init);