xref: /qemu/block/vdi.c (revision 3a691c50)
19aebd98aSStefan Weil /*
29aebd98aSStefan Weil  * Block driver for the Virtual Disk Image (VDI) format
39aebd98aSStefan Weil  *
4641543b7SStefan Weil  * Copyright (c) 2009, 2012 Stefan Weil
59aebd98aSStefan Weil  *
69aebd98aSStefan Weil  * This program is free software: you can redistribute it and/or modify
79aebd98aSStefan Weil  * it under the terms of the GNU General Public License as published by
89aebd98aSStefan Weil  * the Free Software Foundation, either version 2 of the License, or
99aebd98aSStefan Weil  * (at your option) version 3 or any later version.
109aebd98aSStefan Weil  *
119aebd98aSStefan Weil  * This program is distributed in the hope that it will be useful,
129aebd98aSStefan Weil  * but WITHOUT ANY WARRANTY; without even the implied warranty of
139aebd98aSStefan Weil  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
149aebd98aSStefan Weil  * GNU General Public License for more details.
159aebd98aSStefan Weil  *
169aebd98aSStefan Weil  * You should have received a copy of the GNU General Public License
179aebd98aSStefan Weil  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
189aebd98aSStefan Weil  *
199aebd98aSStefan Weil  * Reference:
209aebd98aSStefan Weil  * http://forums.virtualbox.org/viewtopic.php?t=8046
219aebd98aSStefan Weil  *
229aebd98aSStefan Weil  * This driver supports create / read / write operations on VDI images.
239aebd98aSStefan Weil  *
249aebd98aSStefan Weil  * Todo (see also TODO in code):
259aebd98aSStefan Weil  *
269aebd98aSStefan Weil  * Some features like snapshots are still missing.
279aebd98aSStefan Weil  *
289aebd98aSStefan Weil  * Deallocation of zero-filled blocks and shrinking images are missing, too
299aebd98aSStefan Weil  * (might be added to common block layer).
309aebd98aSStefan Weil  *
319aebd98aSStefan Weil  * Allocation of blocks could be optimized (less writes to block map and
329aebd98aSStefan Weil  * header).
339aebd98aSStefan Weil  *
34dc6fb73dSDeepak Kathayat  * Read and write of adjacent blocks could be done in one operation
359aebd98aSStefan Weil  * (current code uses one operation per block (1 MiB).
369aebd98aSStefan Weil  *
379aebd98aSStefan Weil  * The code is not thread safe (missing locks for changes in header and
389aebd98aSStefan Weil  * block table, no problem with current QEMU).
399aebd98aSStefan Weil  *
409aebd98aSStefan Weil  * Hints:
419aebd98aSStefan Weil  *
429aebd98aSStefan Weil  * Blocks (VDI documentation) correspond to clusters (QEMU).
439aebd98aSStefan Weil  * QEMU's backing files could be implemented using VDI snapshot files (TODO).
449aebd98aSStefan Weil  * VDI snapshot files may also contain the complete machine state.
459aebd98aSStefan Weil  * Maybe this machine state can be converted to QEMU PC machine snapshot data.
469aebd98aSStefan Weil  *
479aebd98aSStefan Weil  * The driver keeps a block cache (little endian entries) in memory.
489aebd98aSStefan Weil  * For the standard block size (1 MiB), a 1 TiB disk will use 4 MiB RAM,
499aebd98aSStefan Weil  * so this seems to be reasonable.
509aebd98aSStefan Weil  */
519aebd98aSStefan Weil 
5280c71a24SPeter Maydell #include "qemu/osdep.h"
53da34e65cSMarkus Armbruster #include "qapi/error.h"
54737e150eSPaolo Bonzini #include "block/block_int.h"
55a08f0c3bSKevin Wolf #include "sysemu/block-backend.h"
561de7afc9SPaolo Bonzini #include "qemu/module.h"
5758369e22SPaolo Bonzini #include "qemu/bswap.h"
58795c40b8SJuan Quintela #include "migration/blocker.h"
5910817bf0SDaniel P. Berrange #include "qemu/coroutine.h"
60f348b6d1SVeronia Bahaa #include "qemu/cutils.h"
617c6f55b6SFam Zheng #include "qemu/uuid.h"
629aebd98aSStefan Weil 
639aebd98aSStefan Weil /* Code configuration options. */
649aebd98aSStefan Weil 
659aebd98aSStefan Weil /* Enable debug messages. */
669aebd98aSStefan Weil //~ #define CONFIG_VDI_DEBUG
679aebd98aSStefan Weil 
689aebd98aSStefan Weil /* Support write operations on VDI images. */
699aebd98aSStefan Weil #define CONFIG_VDI_WRITE
709aebd98aSStefan Weil 
719aebd98aSStefan Weil /* Support non-standard block (cluster) size. This is untested.
729aebd98aSStefan Weil  * Maybe it will be needed for very large images.
739aebd98aSStefan Weil  */
749aebd98aSStefan Weil //~ #define CONFIG_VDI_BLOCK_SIZE
759aebd98aSStefan Weil 
769aebd98aSStefan Weil /* Support static (fixed, pre-allocated) images. */
779aebd98aSStefan Weil #define CONFIG_VDI_STATIC_IMAGE
789aebd98aSStefan Weil 
799aebd98aSStefan Weil /* Command line option for static images. */
809aebd98aSStefan Weil #define BLOCK_OPT_STATIC "static"
819aebd98aSStefan Weil 
829aebd98aSStefan Weil #define KiB     1024
839aebd98aSStefan Weil #define MiB     (KiB * KiB)
849aebd98aSStefan Weil 
859aebd98aSStefan Weil #define SECTOR_SIZE 512
8699cce9faSKevin Wolf #define DEFAULT_CLUSTER_SIZE (1 * MiB)
879aebd98aSStefan Weil 
889aebd98aSStefan Weil #if defined(CONFIG_VDI_DEBUG)
899aebd98aSStefan Weil #define logout(fmt, ...) \
909aebd98aSStefan Weil                 fprintf(stderr, "vdi\t%-24s" fmt, __func__, ##__VA_ARGS__)
919aebd98aSStefan Weil #else
929aebd98aSStefan Weil #define logout(fmt, ...) ((void)0)
939aebd98aSStefan Weil #endif
949aebd98aSStefan Weil 
959aebd98aSStefan Weil /* Image signature. */
969aebd98aSStefan Weil #define VDI_SIGNATURE 0xbeda107f
979aebd98aSStefan Weil 
989aebd98aSStefan Weil /* Image version. */
999aebd98aSStefan Weil #define VDI_VERSION_1_1 0x00010001
1009aebd98aSStefan Weil 
1019aebd98aSStefan Weil /* Image type. */
1029aebd98aSStefan Weil #define VDI_TYPE_DYNAMIC 1
1039aebd98aSStefan Weil #define VDI_TYPE_STATIC  2
1049aebd98aSStefan Weil 
1059aebd98aSStefan Weil /* Innotek / SUN images use these strings in header.text:
1069aebd98aSStefan Weil  * "<<< innotek VirtualBox Disk Image >>>\n"
1079aebd98aSStefan Weil  * "<<< Sun xVM VirtualBox Disk Image >>>\n"
1089aebd98aSStefan Weil  * "<<< Sun VirtualBox Disk Image >>>\n"
1099aebd98aSStefan Weil  * The value does not matter, so QEMU created images use a different text.
1109aebd98aSStefan Weil  */
1119aebd98aSStefan Weil #define VDI_TEXT "<<< QEMU VM Virtual Disk Image >>>\n"
1129aebd98aSStefan Weil 
113c794b4e0SEric Sunshine /* A never-allocated block; semantically arbitrary content. */
114c794b4e0SEric Sunshine #define VDI_UNALLOCATED 0xffffffffU
115c794b4e0SEric Sunshine 
116c794b4e0SEric Sunshine /* A discarded (no longer allocated) block; semantically zero-filled. */
117c794b4e0SEric Sunshine #define VDI_DISCARDED   0xfffffffeU
118c794b4e0SEric Sunshine 
119c794b4e0SEric Sunshine #define VDI_IS_ALLOCATED(X) ((X) < VDI_DISCARDED)
1209aebd98aSStefan Weil 
121d20418eeSMax Reitz /* The bmap will take up VDI_BLOCKS_IN_IMAGE_MAX * sizeof(uint32_t) bytes; since
122d20418eeSMax Reitz  * the bmap is read and written in a single operation, its size needs to be
123d20418eeSMax Reitz  * limited to INT_MAX; furthermore, when opening an image, the bmap size is
124d20418eeSMax Reitz  * rounded up to be aligned on BDRV_SECTOR_SIZE.
125d20418eeSMax Reitz  * Therefore this should satisfy the following:
126d20418eeSMax Reitz  * VDI_BLOCKS_IN_IMAGE_MAX * sizeof(uint32_t) + BDRV_SECTOR_SIZE == INT_MAX + 1
127d20418eeSMax Reitz  * (INT_MAX + 1 is the first value not representable as an int)
128d20418eeSMax Reitz  * This guarantees that any value below or equal to the constant will, when
129d20418eeSMax Reitz  * multiplied by sizeof(uint32_t) and rounded up to a BDRV_SECTOR_SIZE boundary,
130d20418eeSMax Reitz  * still be below or equal to INT_MAX. */
131d20418eeSMax Reitz #define VDI_BLOCKS_IN_IMAGE_MAX \
132d20418eeSMax Reitz     ((unsigned)((INT_MAX + 1u - BDRV_SECTOR_SIZE) / sizeof(uint32_t)))
13363fa06dcSJeff Cody #define VDI_DISK_SIZE_MAX        ((uint64_t)VDI_BLOCKS_IN_IMAGE_MAX * \
13463fa06dcSJeff Cody                                   (uint64_t)DEFAULT_CLUSTER_SIZE)
13563fa06dcSJeff Cody 
1369aebd98aSStefan Weil typedef struct {
1379aebd98aSStefan Weil     char text[0x40];
1389aebd98aSStefan Weil     uint32_t signature;
1399aebd98aSStefan Weil     uint32_t version;
1409aebd98aSStefan Weil     uint32_t header_size;
1419aebd98aSStefan Weil     uint32_t image_type;
1429aebd98aSStefan Weil     uint32_t image_flags;
1439aebd98aSStefan Weil     char description[256];
1449aebd98aSStefan Weil     uint32_t offset_bmap;
1459aebd98aSStefan Weil     uint32_t offset_data;
1469aebd98aSStefan Weil     uint32_t cylinders;         /* disk geometry, unused here */
1479aebd98aSStefan Weil     uint32_t heads;             /* disk geometry, unused here */
1489aebd98aSStefan Weil     uint32_t sectors;           /* disk geometry, unused here */
1499aebd98aSStefan Weil     uint32_t sector_size;
1509aebd98aSStefan Weil     uint32_t unused1;
1519aebd98aSStefan Weil     uint64_t disk_size;
1529aebd98aSStefan Weil     uint32_t block_size;
1539aebd98aSStefan Weil     uint32_t block_extra;       /* unused here */
1549aebd98aSStefan Weil     uint32_t blocks_in_image;
1559aebd98aSStefan Weil     uint32_t blocks_allocated;
1567c6f55b6SFam Zheng     QemuUUID uuid_image;
1577c6f55b6SFam Zheng     QemuUUID uuid_last_snap;
1587c6f55b6SFam Zheng     QemuUUID uuid_link;
1597c6f55b6SFam Zheng     QemuUUID uuid_parent;
1609aebd98aSStefan Weil     uint64_t unused2[7];
1618368febdSJeff Cody } QEMU_PACKED VdiHeader;
1629aebd98aSStefan Weil 
1639aebd98aSStefan Weil typedef struct {
1649aebd98aSStefan Weil     /* The block map entries are little endian (even in memory). */
1659aebd98aSStefan Weil     uint32_t *bmap;
1669aebd98aSStefan Weil     /* Size of block (bytes). */
1679aebd98aSStefan Weil     uint32_t block_size;
1689aebd98aSStefan Weil     /* Size of block (sectors). */
1699aebd98aSStefan Weil     uint32_t block_sectors;
1709aebd98aSStefan Weil     /* First sector of block map. */
1719aebd98aSStefan Weil     uint32_t bmap_sector;
1724ff9786cSStefan Weil     /* VDI header (converted to host endianness). */
1739aebd98aSStefan Weil     VdiHeader header;
174fc9d106cSKevin Wolf 
175f0ab6f10SMax Reitz     CoMutex write_lock;
176f0ab6f10SMax Reitz 
177fc9d106cSKevin Wolf     Error *migration_blocker;
1789aebd98aSStefan Weil } BDRVVdiState;
1799aebd98aSStefan Weil 
1809aebd98aSStefan Weil static void vdi_header_to_cpu(VdiHeader *header)
1819aebd98aSStefan Weil {
1829aebd98aSStefan Weil     le32_to_cpus(&header->signature);
1839aebd98aSStefan Weil     le32_to_cpus(&header->version);
1849aebd98aSStefan Weil     le32_to_cpus(&header->header_size);
1859aebd98aSStefan Weil     le32_to_cpus(&header->image_type);
1869aebd98aSStefan Weil     le32_to_cpus(&header->image_flags);
1879aebd98aSStefan Weil     le32_to_cpus(&header->offset_bmap);
1889aebd98aSStefan Weil     le32_to_cpus(&header->offset_data);
1899aebd98aSStefan Weil     le32_to_cpus(&header->cylinders);
1909aebd98aSStefan Weil     le32_to_cpus(&header->heads);
1919aebd98aSStefan Weil     le32_to_cpus(&header->sectors);
1929aebd98aSStefan Weil     le32_to_cpus(&header->sector_size);
1939aebd98aSStefan Weil     le64_to_cpus(&header->disk_size);
1949aebd98aSStefan Weil     le32_to_cpus(&header->block_size);
1959aebd98aSStefan Weil     le32_to_cpus(&header->block_extra);
1969aebd98aSStefan Weil     le32_to_cpus(&header->blocks_in_image);
1979aebd98aSStefan Weil     le32_to_cpus(&header->blocks_allocated);
1987c6f55b6SFam Zheng     qemu_uuid_bswap(&header->uuid_image);
1997c6f55b6SFam Zheng     qemu_uuid_bswap(&header->uuid_last_snap);
2007c6f55b6SFam Zheng     qemu_uuid_bswap(&header->uuid_link);
2017c6f55b6SFam Zheng     qemu_uuid_bswap(&header->uuid_parent);
2029aebd98aSStefan Weil }
2039aebd98aSStefan Weil 
2049aebd98aSStefan Weil static void vdi_header_to_le(VdiHeader *header)
2059aebd98aSStefan Weil {
2069aebd98aSStefan Weil     cpu_to_le32s(&header->signature);
2079aebd98aSStefan Weil     cpu_to_le32s(&header->version);
2089aebd98aSStefan Weil     cpu_to_le32s(&header->header_size);
2099aebd98aSStefan Weil     cpu_to_le32s(&header->image_type);
2109aebd98aSStefan Weil     cpu_to_le32s(&header->image_flags);
2119aebd98aSStefan Weil     cpu_to_le32s(&header->offset_bmap);
2129aebd98aSStefan Weil     cpu_to_le32s(&header->offset_data);
2139aebd98aSStefan Weil     cpu_to_le32s(&header->cylinders);
2149aebd98aSStefan Weil     cpu_to_le32s(&header->heads);
2159aebd98aSStefan Weil     cpu_to_le32s(&header->sectors);
2169aebd98aSStefan Weil     cpu_to_le32s(&header->sector_size);
2179aebd98aSStefan Weil     cpu_to_le64s(&header->disk_size);
2189aebd98aSStefan Weil     cpu_to_le32s(&header->block_size);
2199aebd98aSStefan Weil     cpu_to_le32s(&header->block_extra);
2209aebd98aSStefan Weil     cpu_to_le32s(&header->blocks_in_image);
2219aebd98aSStefan Weil     cpu_to_le32s(&header->blocks_allocated);
2227c6f55b6SFam Zheng     qemu_uuid_bswap(&header->uuid_image);
2237c6f55b6SFam Zheng     qemu_uuid_bswap(&header->uuid_last_snap);
2247c6f55b6SFam Zheng     qemu_uuid_bswap(&header->uuid_link);
2257c6f55b6SFam Zheng     qemu_uuid_bswap(&header->uuid_parent);
2269aebd98aSStefan Weil }
2279aebd98aSStefan Weil 
2289aebd98aSStefan Weil #if defined(CONFIG_VDI_DEBUG)
2299aebd98aSStefan Weil static void vdi_header_print(VdiHeader *header)
2309aebd98aSStefan Weil {
2319aebd98aSStefan Weil     char uuid[37];
2329aebd98aSStefan Weil     logout("text        %s", header->text);
2339f0470bbSStefan Weil     logout("signature   0x%08x\n", header->signature);
2349aebd98aSStefan Weil     logout("header size 0x%04x\n", header->header_size);
2359aebd98aSStefan Weil     logout("image type  0x%04x\n", header->image_type);
2369aebd98aSStefan Weil     logout("image flags 0x%04x\n", header->image_flags);
2379aebd98aSStefan Weil     logout("description %s\n", header->description);
2389aebd98aSStefan Weil     logout("offset bmap 0x%04x\n", header->offset_bmap);
2399aebd98aSStefan Weil     logout("offset data 0x%04x\n", header->offset_data);
2409aebd98aSStefan Weil     logout("cylinders   0x%04x\n", header->cylinders);
2419aebd98aSStefan Weil     logout("heads       0x%04x\n", header->heads);
2429aebd98aSStefan Weil     logout("sectors     0x%04x\n", header->sectors);
2439aebd98aSStefan Weil     logout("sector size 0x%04x\n", header->sector_size);
2449aebd98aSStefan Weil     logout("image size  0x%" PRIx64 " B (%" PRIu64 " MiB)\n",
2459aebd98aSStefan Weil            header->disk_size, header->disk_size / MiB);
2469aebd98aSStefan Weil     logout("block size  0x%04x\n", header->block_size);
2479aebd98aSStefan Weil     logout("block extra 0x%04x\n", header->block_extra);
2489aebd98aSStefan Weil     logout("blocks tot. 0x%04x\n", header->blocks_in_image);
2499aebd98aSStefan Weil     logout("blocks all. 0x%04x\n", header->blocks_allocated);
2509aebd98aSStefan Weil     uuid_unparse(header->uuid_image, uuid);
2519aebd98aSStefan Weil     logout("uuid image  %s\n", uuid);
2529aebd98aSStefan Weil     uuid_unparse(header->uuid_last_snap, uuid);
2539aebd98aSStefan Weil     logout("uuid snap   %s\n", uuid);
2549aebd98aSStefan Weil     uuid_unparse(header->uuid_link, uuid);
2559aebd98aSStefan Weil     logout("uuid link   %s\n", uuid);
2569aebd98aSStefan Weil     uuid_unparse(header->uuid_parent, uuid);
2579aebd98aSStefan Weil     logout("uuid parent %s\n", uuid);
2589aebd98aSStefan Weil }
2599aebd98aSStefan Weil #endif
2609aebd98aSStefan Weil 
2614534ff54SKevin Wolf static int vdi_check(BlockDriverState *bs, BdrvCheckResult *res,
2624534ff54SKevin Wolf                      BdrvCheckMode fix)
2639aebd98aSStefan Weil {
2649aebd98aSStefan Weil     /* TODO: additional checks possible. */
2659aebd98aSStefan Weil     BDRVVdiState *s = (BDRVVdiState *)bs->opaque;
2669aebd98aSStefan Weil     uint32_t blocks_allocated = 0;
2679aebd98aSStefan Weil     uint32_t block;
2689aebd98aSStefan Weil     uint32_t *bmap;
2699aebd98aSStefan Weil     logout("\n");
2709aebd98aSStefan Weil 
2714534ff54SKevin Wolf     if (fix) {
2724534ff54SKevin Wolf         return -ENOTSUP;
2734534ff54SKevin Wolf     }
2744534ff54SKevin Wolf 
2755839e53bSMarkus Armbruster     bmap = g_try_new(uint32_t, s->header.blocks_in_image);
27617cce735SKevin Wolf     if (s->header.blocks_in_image && bmap == NULL) {
27717cce735SKevin Wolf         res->check_errors++;
27817cce735SKevin Wolf         return -ENOMEM;
27917cce735SKevin Wolf     }
28017cce735SKevin Wolf 
2819aebd98aSStefan Weil     memset(bmap, 0xff, s->header.blocks_in_image * sizeof(uint32_t));
2829aebd98aSStefan Weil 
2839aebd98aSStefan Weil     /* Check block map and value of blocks_allocated. */
2849aebd98aSStefan Weil     for (block = 0; block < s->header.blocks_in_image; block++) {
2859aebd98aSStefan Weil         uint32_t bmap_entry = le32_to_cpu(s->bmap[block]);
286c794b4e0SEric Sunshine         if (VDI_IS_ALLOCATED(bmap_entry)) {
2879aebd98aSStefan Weil             if (bmap_entry < s->header.blocks_in_image) {
2889aebd98aSStefan Weil                 blocks_allocated++;
289c794b4e0SEric Sunshine                 if (!VDI_IS_ALLOCATED(bmap[bmap_entry])) {
2909aebd98aSStefan Weil                     bmap[bmap_entry] = bmap_entry;
2919aebd98aSStefan Weil                 } else {
2929aebd98aSStefan Weil                     fprintf(stderr, "ERROR: block index %" PRIu32
2939aebd98aSStefan Weil                             " also used by %" PRIu32 "\n", bmap[bmap_entry], bmap_entry);
2949ac228e0SKevin Wolf                     res->corruptions++;
2959aebd98aSStefan Weil                 }
2969aebd98aSStefan Weil             } else {
2979aebd98aSStefan Weil                 fprintf(stderr, "ERROR: block index %" PRIu32
2989aebd98aSStefan Weil                         " too large, is %" PRIu32 "\n", block, bmap_entry);
2999ac228e0SKevin Wolf                 res->corruptions++;
3009aebd98aSStefan Weil             }
3019aebd98aSStefan Weil         }
3029aebd98aSStefan Weil     }
3039aebd98aSStefan Weil     if (blocks_allocated != s->header.blocks_allocated) {
3049aebd98aSStefan Weil         fprintf(stderr, "ERROR: allocated blocks mismatch, is %" PRIu32
3059aebd98aSStefan Weil                ", should be %" PRIu32 "\n",
3069aebd98aSStefan Weil                blocks_allocated, s->header.blocks_allocated);
3079ac228e0SKevin Wolf         res->corruptions++;
3089aebd98aSStefan Weil     }
3099aebd98aSStefan Weil 
3107267c094SAnthony Liguori     g_free(bmap);
3119aebd98aSStefan Weil 
3129ac228e0SKevin Wolf     return 0;
3139aebd98aSStefan Weil }
3149aebd98aSStefan Weil 
3159aebd98aSStefan Weil static int vdi_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
3169aebd98aSStefan Weil {
3179aebd98aSStefan Weil     /* TODO: vdi_get_info would be needed for machine snapshots.
3189aebd98aSStefan Weil        vm_state_offset is still missing. */
3199aebd98aSStefan Weil     BDRVVdiState *s = (BDRVVdiState *)bs->opaque;
3209aebd98aSStefan Weil     logout("\n");
3219aebd98aSStefan Weil     bdi->cluster_size = s->block_size;
3229aebd98aSStefan Weil     bdi->vm_state_offset = 0;
32395de6d70SPaolo Bonzini     bdi->unallocated_blocks_are_zero = true;
3249aebd98aSStefan Weil     return 0;
3259aebd98aSStefan Weil }
3269aebd98aSStefan Weil 
3279aebd98aSStefan Weil static int vdi_make_empty(BlockDriverState *bs)
3289aebd98aSStefan Weil {
3299aebd98aSStefan Weil     /* TODO: missing code. */
3309aebd98aSStefan Weil     logout("\n");
3319aebd98aSStefan Weil     /* The return value for missing code must be 0, see block.c. */
3329aebd98aSStefan Weil     return 0;
3339aebd98aSStefan Weil }
3349aebd98aSStefan Weil 
3359aebd98aSStefan Weil static int vdi_probe(const uint8_t *buf, int buf_size, const char *filename)
3369aebd98aSStefan Weil {
3379aebd98aSStefan Weil     const VdiHeader *header = (const VdiHeader *)buf;
338dddc7750SJeff Cody     int ret = 0;
3399aebd98aSStefan Weil 
3409aebd98aSStefan Weil     logout("\n");
3419aebd98aSStefan Weil 
3429aebd98aSStefan Weil     if (buf_size < sizeof(*header)) {
3439aebd98aSStefan Weil         /* Header too small, no VDI. */
3449aebd98aSStefan Weil     } else if (le32_to_cpu(header->signature) == VDI_SIGNATURE) {
345dddc7750SJeff Cody         ret = 100;
3469aebd98aSStefan Weil     }
3479aebd98aSStefan Weil 
348dddc7750SJeff Cody     if (ret == 0) {
3499aebd98aSStefan Weil         logout("no vdi image\n");
3509aebd98aSStefan Weil     } else {
3519aebd98aSStefan Weil         logout("%s", header->text);
3529aebd98aSStefan Weil     }
3539aebd98aSStefan Weil 
354dddc7750SJeff Cody     return ret;
3559aebd98aSStefan Weil }
3569aebd98aSStefan Weil 
357015a1036SMax Reitz static int vdi_open(BlockDriverState *bs, QDict *options, int flags,
358015a1036SMax Reitz                     Error **errp)
3599aebd98aSStefan Weil {
3609aebd98aSStefan Weil     BDRVVdiState *s = bs->opaque;
3619aebd98aSStefan Weil     VdiHeader header;
3629aebd98aSStefan Weil     size_t bmap_size;
3638937f822SStefan Weil     int ret;
364fe44dc91SAshijeet Acharya     Error *local_err = NULL;
3659aebd98aSStefan Weil 
3664e4bf5c4SKevin Wolf     bs->file = bdrv_open_child(NULL, options, "file", bs, &child_file,
3674e4bf5c4SKevin Wolf                                false, errp);
3684e4bf5c4SKevin Wolf     if (!bs->file) {
3694e4bf5c4SKevin Wolf         return -EINVAL;
3704e4bf5c4SKevin Wolf     }
3714e4bf5c4SKevin Wolf 
3729aebd98aSStefan Weil     logout("\n");
3739aebd98aSStefan Weil 
374fbcbbf4eSKevin Wolf     ret = bdrv_read(bs->file, 0, (uint8_t *)&header, 1);
3758937f822SStefan Weil     if (ret < 0) {
3769aebd98aSStefan Weil         goto fail;
3779aebd98aSStefan Weil     }
3789aebd98aSStefan Weil 
3799aebd98aSStefan Weil     vdi_header_to_cpu(&header);
3809aebd98aSStefan Weil #if defined(CONFIG_VDI_DEBUG)
3819aebd98aSStefan Weil     vdi_header_print(&header);
3829aebd98aSStefan Weil #endif
3839aebd98aSStefan Weil 
38463fa06dcSJeff Cody     if (header.disk_size > VDI_DISK_SIZE_MAX) {
38563fa06dcSJeff Cody         error_setg(errp, "Unsupported VDI image size (size is 0x%" PRIx64
38663fa06dcSJeff Cody                           ", max supported is 0x%" PRIx64 ")",
38763fa06dcSJeff Cody                           header.disk_size, VDI_DISK_SIZE_MAX);
38863fa06dcSJeff Cody         ret = -ENOTSUP;
38963fa06dcSJeff Cody         goto fail;
39063fa06dcSJeff Cody     }
39163fa06dcSJeff Cody 
392f21dc3a4SStefan Weil     if (header.disk_size % SECTOR_SIZE != 0) {
393f21dc3a4SStefan Weil         /* 'VBoxManage convertfromraw' can create images with odd disk sizes.
394f21dc3a4SStefan Weil            We accept them but round the disk size to the next multiple of
395f21dc3a4SStefan Weil            SECTOR_SIZE. */
396f21dc3a4SStefan Weil         logout("odd disk size %" PRIu64 " B, round up\n", header.disk_size);
397e9082e47SMax Reitz         header.disk_size = ROUND_UP(header.disk_size, SECTOR_SIZE);
398f21dc3a4SStefan Weil     }
399f21dc3a4SStefan Weil 
4000e87ba2cSStefan Weil     if (header.signature != VDI_SIGNATURE) {
401521b2b5dSMax Reitz         error_setg(errp, "Image not in VDI format (bad signature %08" PRIx32
402521b2b5dSMax Reitz                    ")", header.signature);
40376abe407SPaolo Bonzini         ret = -EINVAL;
4040e87ba2cSStefan Weil         goto fail;
4050e87ba2cSStefan Weil     } else if (header.version != VDI_VERSION_1_1) {
406521b2b5dSMax Reitz         error_setg(errp, "unsupported VDI image (version %" PRIu32 ".%" PRIu32
407521b2b5dSMax Reitz                    ")", header.version >> 16, header.version & 0xffff);
4088937f822SStefan Weil         ret = -ENOTSUP;
4099aebd98aSStefan Weil         goto fail;
4109aebd98aSStefan Weil     } else if (header.offset_bmap % SECTOR_SIZE != 0) {
4119aebd98aSStefan Weil         /* We only support block maps which start on a sector boundary. */
4125b7aa9b5SPaolo Bonzini         error_setg(errp, "unsupported VDI image (unaligned block map offset "
413521b2b5dSMax Reitz                    "0x%" PRIx32 ")", header.offset_bmap);
4148937f822SStefan Weil         ret = -ENOTSUP;
4159aebd98aSStefan Weil         goto fail;
4169aebd98aSStefan Weil     } else if (header.offset_data % SECTOR_SIZE != 0) {
4179aebd98aSStefan Weil         /* We only support data blocks which start on a sector boundary. */
418521b2b5dSMax Reitz         error_setg(errp, "unsupported VDI image (unaligned data offset 0x%"
419521b2b5dSMax Reitz                    PRIx32 ")", header.offset_data);
4208937f822SStefan Weil         ret = -ENOTSUP;
4219aebd98aSStefan Weil         goto fail;
4229aebd98aSStefan Weil     } else if (header.sector_size != SECTOR_SIZE) {
423521b2b5dSMax Reitz         error_setg(errp, "unsupported VDI image (sector size %" PRIu32
424521b2b5dSMax Reitz                    " is not %u)", header.sector_size, SECTOR_SIZE);
4258937f822SStefan Weil         ret = -ENOTSUP;
4269aebd98aSStefan Weil         goto fail;
42763fa06dcSJeff Cody     } else if (header.block_size != DEFAULT_CLUSTER_SIZE) {
428521b2b5dSMax Reitz         error_setg(errp, "unsupported VDI image (block size %" PRIu32
429521b2b5dSMax Reitz                    " is not %u)", header.block_size, DEFAULT_CLUSTER_SIZE);
4308937f822SStefan Weil         ret = -ENOTSUP;
4319aebd98aSStefan Weil         goto fail;
432f21dc3a4SStefan Weil     } else if (header.disk_size >
433f21dc3a4SStefan Weil                (uint64_t)header.blocks_in_image * header.block_size) {
4345b7aa9b5SPaolo Bonzini         error_setg(errp, "unsupported VDI image (disk size %" PRIu64 ", "
4355b7aa9b5SPaolo Bonzini                    "image bitmap has room for %" PRIu64 ")",
4365b7aa9b5SPaolo Bonzini                    header.disk_size,
4375b7aa9b5SPaolo Bonzini                    (uint64_t)header.blocks_in_image * header.block_size);
4388937f822SStefan Weil         ret = -ENOTSUP;
4399aebd98aSStefan Weil         goto fail;
4407c6f55b6SFam Zheng     } else if (!qemu_uuid_is_null(&header.uuid_link)) {
4415b7aa9b5SPaolo Bonzini         error_setg(errp, "unsupported VDI image (non-NULL link UUID)");
4428937f822SStefan Weil         ret = -ENOTSUP;
4439aebd98aSStefan Weil         goto fail;
4447c6f55b6SFam Zheng     } else if (!qemu_uuid_is_null(&header.uuid_parent)) {
4455b7aa9b5SPaolo Bonzini         error_setg(errp, "unsupported VDI image (non-NULL parent UUID)");
4468937f822SStefan Weil         ret = -ENOTSUP;
4479aebd98aSStefan Weil         goto fail;
44863fa06dcSJeff Cody     } else if (header.blocks_in_image > VDI_BLOCKS_IN_IMAGE_MAX) {
44963fa06dcSJeff Cody         error_setg(errp, "unsupported VDI image "
45063fa06dcSJeff Cody                          "(too many blocks %u, max is %u)",
45163fa06dcSJeff Cody                           header.blocks_in_image, VDI_BLOCKS_IN_IMAGE_MAX);
45263fa06dcSJeff Cody         ret = -ENOTSUP;
45363fa06dcSJeff Cody         goto fail;
4549aebd98aSStefan Weil     }
4559aebd98aSStefan Weil 
4569aebd98aSStefan Weil     bs->total_sectors = header.disk_size / SECTOR_SIZE;
4579aebd98aSStefan Weil 
4589aebd98aSStefan Weil     s->block_size = header.block_size;
4599aebd98aSStefan Weil     s->block_sectors = header.block_size / SECTOR_SIZE;
4609aebd98aSStefan Weil     s->bmap_sector = header.offset_bmap / SECTOR_SIZE;
4619aebd98aSStefan Weil     s->header = header;
4629aebd98aSStefan Weil 
4639aebd98aSStefan Weil     bmap_size = header.blocks_in_image * sizeof(uint32_t);
464e9082e47SMax Reitz     bmap_size = DIV_ROUND_UP(bmap_size, SECTOR_SIZE);
4659a4f4c31SKevin Wolf     s->bmap = qemu_try_blockalign(bs->file->bs, bmap_size * SECTOR_SIZE);
46617cce735SKevin Wolf     if (s->bmap == NULL) {
46717cce735SKevin Wolf         ret = -ENOMEM;
46817cce735SKevin Wolf         goto fail;
46917cce735SKevin Wolf     }
47017cce735SKevin Wolf 
471fbcbbf4eSKevin Wolf     ret = bdrv_read(bs->file, s->bmap_sector, (uint8_t *)s->bmap,
4729a4f4c31SKevin Wolf                     bmap_size);
4738937f822SStefan Weil     if (ret < 0) {
4749aebd98aSStefan Weil         goto fail_free_bmap;
4759aebd98aSStefan Weil     }
4769aebd98aSStefan Weil 
477fc9d106cSKevin Wolf     /* Disable migration when vdi images are used */
47881e5f78aSAlberto Garcia     error_setg(&s->migration_blocker, "The vdi format used by node '%s' "
47981e5f78aSAlberto Garcia                "does not support live migration",
48081e5f78aSAlberto Garcia                bdrv_get_device_or_node_name(bs));
481fe44dc91SAshijeet Acharya     ret = migrate_add_blocker(s->migration_blocker, &local_err);
482fe44dc91SAshijeet Acharya     if (local_err) {
483fe44dc91SAshijeet Acharya         error_propagate(errp, local_err);
484fe44dc91SAshijeet Acharya         error_free(s->migration_blocker);
485fe44dc91SAshijeet Acharya         goto fail_free_bmap;
486fe44dc91SAshijeet Acharya     }
487fc9d106cSKevin Wolf 
488f0ab6f10SMax Reitz     qemu_co_mutex_init(&s->write_lock);
489f0ab6f10SMax Reitz 
4909aebd98aSStefan Weil     return 0;
4919aebd98aSStefan Weil 
4929aebd98aSStefan Weil  fail_free_bmap:
49317cce735SKevin Wolf     qemu_vfree(s->bmap);
4949aebd98aSStefan Weil 
4959aebd98aSStefan Weil  fail:
4968937f822SStefan Weil     return ret;
4979aebd98aSStefan Weil }
4989aebd98aSStefan Weil 
499ecfe2bbaSJeff Cody static int vdi_reopen_prepare(BDRVReopenState *state,
500ecfe2bbaSJeff Cody                               BlockReopenQueue *queue, Error **errp)
501ecfe2bbaSJeff Cody {
502ecfe2bbaSJeff Cody     return 0;
503ecfe2bbaSJeff Cody }
504ecfe2bbaSJeff Cody 
505b6b8a333SPaolo Bonzini static int64_t coroutine_fn vdi_co_get_block_status(BlockDriverState *bs,
50667a0fd2aSFam Zheng         int64_t sector_num, int nb_sectors, int *pnum, BlockDriverState **file)
5079aebd98aSStefan Weil {
5089aebd98aSStefan Weil     /* TODO: Check for too large sector_num (in bdrv_is_allocated or here). */
5099aebd98aSStefan Weil     BDRVVdiState *s = (BDRVVdiState *)bs->opaque;
5109aebd98aSStefan Weil     size_t bmap_index = sector_num / s->block_sectors;
5119aebd98aSStefan Weil     size_t sector_in_block = sector_num % s->block_sectors;
5129aebd98aSStefan Weil     int n_sectors = s->block_sectors - sector_in_block;
5139aebd98aSStefan Weil     uint32_t bmap_entry = le32_to_cpu(s->bmap[bmap_index]);
5144bc74be9SPaolo Bonzini     uint64_t offset;
5154bc74be9SPaolo Bonzini     int result;
5164bc74be9SPaolo Bonzini 
5179aebd98aSStefan Weil     logout("%p, %" PRId64 ", %d, %p\n", bs, sector_num, nb_sectors, pnum);
5189aebd98aSStefan Weil     if (n_sectors > nb_sectors) {
5199aebd98aSStefan Weil         n_sectors = nb_sectors;
5209aebd98aSStefan Weil     }
5219aebd98aSStefan Weil     *pnum = n_sectors;
5224bc74be9SPaolo Bonzini     result = VDI_IS_ALLOCATED(bmap_entry);
5234bc74be9SPaolo Bonzini     if (!result) {
5244bc74be9SPaolo Bonzini         return 0;
5254bc74be9SPaolo Bonzini     }
5264bc74be9SPaolo Bonzini 
5274bc74be9SPaolo Bonzini     offset = s->header.offset_data +
5284bc74be9SPaolo Bonzini                               (uint64_t)bmap_entry * s->block_size +
5294bc74be9SPaolo Bonzini                               sector_in_block * SECTOR_SIZE;
5308bfb1371SFam Zheng     *file = bs->file->bs;
5314bc74be9SPaolo Bonzini     return BDRV_BLOCK_DATA | BDRV_BLOCK_OFFSET_VALID | offset;
5329aebd98aSStefan Weil }
5339aebd98aSStefan Weil 
5340865bb6fSKevin Wolf static int coroutine_fn
5350865bb6fSKevin Wolf vdi_co_preadv(BlockDriverState *bs, uint64_t offset, uint64_t bytes,
5360865bb6fSKevin Wolf               QEMUIOVector *qiov, int flags)
5379aebd98aSStefan Weil {
5389aebd98aSStefan Weil     BDRVVdiState *s = bs->opaque;
5390865bb6fSKevin Wolf     QEMUIOVector local_qiov;
5409aebd98aSStefan Weil     uint32_t bmap_entry;
5419aebd98aSStefan Weil     uint32_t block_index;
5420865bb6fSKevin Wolf     uint32_t offset_in_block;
5430865bb6fSKevin Wolf     uint32_t n_bytes;
5440865bb6fSKevin Wolf     uint64_t bytes_done = 0;
545eb9566d1SPaolo Bonzini     int ret = 0;
5464de659e8SPaolo Bonzini 
5474de659e8SPaolo Bonzini     logout("\n");
5489aebd98aSStefan Weil 
5490865bb6fSKevin Wolf     qemu_iovec_init(&local_qiov, qiov->niov);
5509aebd98aSStefan Weil 
5510865bb6fSKevin Wolf     while (ret >= 0 && bytes > 0) {
5520865bb6fSKevin Wolf         block_index = offset / s->block_size;
5530865bb6fSKevin Wolf         offset_in_block = offset % s->block_size;
5540865bb6fSKevin Wolf         n_bytes = MIN(bytes, s->block_size - offset_in_block);
5550865bb6fSKevin Wolf 
5560865bb6fSKevin Wolf         logout("will read %u bytes starting at offset %" PRIu64 "\n",
5570865bb6fSKevin Wolf                n_bytes, offset);
5589aebd98aSStefan Weil 
5599aebd98aSStefan Weil         /* prepare next AIO request */
5609aebd98aSStefan Weil         bmap_entry = le32_to_cpu(s->bmap[block_index]);
561c794b4e0SEric Sunshine         if (!VDI_IS_ALLOCATED(bmap_entry)) {
5629aebd98aSStefan Weil             /* Block not allocated, return zeros, no need to wait. */
5630865bb6fSKevin Wolf             qemu_iovec_memset(qiov, bytes_done, 0, n_bytes);
5643d46a75aSPaolo Bonzini             ret = 0;
5659aebd98aSStefan Weil         } else {
5660865bb6fSKevin Wolf             uint64_t data_offset = s->header.offset_data +
5670865bb6fSKevin Wolf                                    (uint64_t)bmap_entry * s->block_size +
5680865bb6fSKevin Wolf                                    offset_in_block;
5690c7bfc32SPaolo Bonzini 
5700865bb6fSKevin Wolf             qemu_iovec_reset(&local_qiov);
5710865bb6fSKevin Wolf             qemu_iovec_concat(&local_qiov, qiov, bytes_done, n_bytes);
5720865bb6fSKevin Wolf 
573a03ef88fSKevin Wolf             ret = bdrv_co_preadv(bs->file, data_offset, n_bytes,
5740865bb6fSKevin Wolf                                  &local_qiov, 0);
5753d46a75aSPaolo Bonzini         }
5760865bb6fSKevin Wolf         logout("%u bytes read\n", n_bytes);
5770865bb6fSKevin Wolf 
5780865bb6fSKevin Wolf         bytes -= n_bytes;
5790865bb6fSKevin Wolf         offset += n_bytes;
5800865bb6fSKevin Wolf         bytes_done += n_bytes;
5810865bb6fSKevin Wolf     }
5820865bb6fSKevin Wolf 
5830865bb6fSKevin Wolf     qemu_iovec_destroy(&local_qiov);
5843d46a75aSPaolo Bonzini 
5853d46a75aSPaolo Bonzini     return ret;
5869aebd98aSStefan Weil }
5879aebd98aSStefan Weil 
588fde9d56fSKevin Wolf static int coroutine_fn
589fde9d56fSKevin Wolf vdi_co_pwritev(BlockDriverState *bs, uint64_t offset, uint64_t bytes,
590fde9d56fSKevin Wolf                QEMUIOVector *qiov, int flags)
5919aebd98aSStefan Weil {
5929aebd98aSStefan Weil     BDRVVdiState *s = bs->opaque;
593fde9d56fSKevin Wolf     QEMUIOVector local_qiov;
5949aebd98aSStefan Weil     uint32_t bmap_entry;
5959aebd98aSStefan Weil     uint32_t block_index;
596fde9d56fSKevin Wolf     uint32_t offset_in_block;
597fde9d56fSKevin Wolf     uint32_t n_bytes;
598bfc45fc1SPaolo Bonzini     uint32_t bmap_first = VDI_UNALLOCATED;
599bfc45fc1SPaolo Bonzini     uint32_t bmap_last = VDI_UNALLOCATED;
600bfc45fc1SPaolo Bonzini     uint8_t *block = NULL;
601fde9d56fSKevin Wolf     uint64_t bytes_done = 0;
602eb9566d1SPaolo Bonzini     int ret = 0;
6034de659e8SPaolo Bonzini 
6044de659e8SPaolo Bonzini     logout("\n");
6059aebd98aSStefan Weil 
606fde9d56fSKevin Wolf     qemu_iovec_init(&local_qiov, qiov->niov);
6079aebd98aSStefan Weil 
608fde9d56fSKevin Wolf     while (ret >= 0 && bytes > 0) {
609fde9d56fSKevin Wolf         block_index = offset / s->block_size;
610fde9d56fSKevin Wolf         offset_in_block = offset % s->block_size;
611fde9d56fSKevin Wolf         n_bytes = MIN(bytes, s->block_size - offset_in_block);
612fde9d56fSKevin Wolf 
613fde9d56fSKevin Wolf         logout("will write %u bytes starting at offset %" PRIu64 "\n",
614fde9d56fSKevin Wolf                n_bytes, offset);
6159aebd98aSStefan Weil 
6169aebd98aSStefan Weil         /* prepare next AIO request */
6179aebd98aSStefan Weil         bmap_entry = le32_to_cpu(s->bmap[block_index]);
618c794b4e0SEric Sunshine         if (!VDI_IS_ALLOCATED(bmap_entry)) {
6199aebd98aSStefan Weil             /* Allocate new block and write to it. */
620fde9d56fSKevin Wolf             uint64_t data_offset;
6219aebd98aSStefan Weil             bmap_entry = s->header.blocks_allocated;
6229aebd98aSStefan Weil             s->bmap[block_index] = cpu_to_le32(bmap_entry);
6239aebd98aSStefan Weil             s->header.blocks_allocated++;
624fde9d56fSKevin Wolf             data_offset = s->header.offset_data +
625fde9d56fSKevin Wolf                           (uint64_t)bmap_entry * s->block_size;
6269aebd98aSStefan Weil             if (block == NULL) {
627641543b7SStefan Weil                 block = g_malloc(s->block_size);
628bfc45fc1SPaolo Bonzini                 bmap_first = block_index;
6299aebd98aSStefan Weil             }
630bfc45fc1SPaolo Bonzini             bmap_last = block_index;
631641543b7SStefan Weil             /* Copy data to be written to new block and zero unused parts. */
632fde9d56fSKevin Wolf             memset(block, 0, offset_in_block);
633fde9d56fSKevin Wolf             qemu_iovec_to_buf(qiov, bytes_done, block + offset_in_block,
634fde9d56fSKevin Wolf                               n_bytes);
635fde9d56fSKevin Wolf             memset(block + offset_in_block + n_bytes, 0,
636fde9d56fSKevin Wolf                    s->block_size - n_bytes - offset_in_block);
637f0ab6f10SMax Reitz 
638f0ab6f10SMax Reitz             /* Note that this coroutine does not yield anywhere from reading the
639f0ab6f10SMax Reitz              * bmap entry until here, so in regards to all the coroutines trying
640f0ab6f10SMax Reitz              * to write to this cluster, the one doing the allocation will
641f0ab6f10SMax Reitz              * always be the first to try to acquire the lock.
642f0ab6f10SMax Reitz              * Therefore, it is also the first that will actually be able to
643f0ab6f10SMax Reitz              * acquire the lock and thus the padded cluster is written before
644f0ab6f10SMax Reitz              * the other coroutines can write to the affected area. */
645f0ab6f10SMax Reitz             qemu_co_mutex_lock(&s->write_lock);
646d9ca2ea2SKevin Wolf             ret = bdrv_pwrite(bs->file, data_offset, block, s->block_size);
647f0ab6f10SMax Reitz             qemu_co_mutex_unlock(&s->write_lock);
6489aebd98aSStefan Weil         } else {
649fde9d56fSKevin Wolf             uint64_t data_offset = s->header.offset_data +
650fde9d56fSKevin Wolf                                    (uint64_t)bmap_entry * s->block_size +
651fde9d56fSKevin Wolf                                    offset_in_block;
652f0ab6f10SMax Reitz             qemu_co_mutex_lock(&s->write_lock);
653f0ab6f10SMax Reitz             /* This lock is only used to make sure the following write operation
654f0ab6f10SMax Reitz              * is executed after the write issued by the coroutine allocating
655f0ab6f10SMax Reitz              * this cluster, therefore we do not need to keep it locked.
656f0ab6f10SMax Reitz              * As stated above, the allocating coroutine will always try to lock
657f0ab6f10SMax Reitz              * the mutex before all the other concurrent accesses to that
658f0ab6f10SMax Reitz              * cluster, therefore at this point we can be absolutely certain
659f0ab6f10SMax Reitz              * that that write operation has returned (there may be other writes
660f0ab6f10SMax Reitz              * in flight, but they do not concern this very operation). */
661f0ab6f10SMax Reitz             qemu_co_mutex_unlock(&s->write_lock);
662fde9d56fSKevin Wolf 
663fde9d56fSKevin Wolf             qemu_iovec_reset(&local_qiov);
664fde9d56fSKevin Wolf             qemu_iovec_concat(&local_qiov, qiov, bytes_done, n_bytes);
665fde9d56fSKevin Wolf 
666a03ef88fSKevin Wolf             ret = bdrv_co_pwritev(bs->file, data_offset, n_bytes,
667fde9d56fSKevin Wolf                                   &local_qiov, 0);
6689aebd98aSStefan Weil         }
6690c7bfc32SPaolo Bonzini 
670fde9d56fSKevin Wolf         bytes -= n_bytes;
671fde9d56fSKevin Wolf         offset += n_bytes;
672fde9d56fSKevin Wolf         bytes_done += n_bytes;
6730c7bfc32SPaolo Bonzini 
674fde9d56fSKevin Wolf         logout("%u bytes written\n", n_bytes);
6753d46a75aSPaolo Bonzini     }
6769aebd98aSStefan Weil 
677fde9d56fSKevin Wolf     qemu_iovec_destroy(&local_qiov);
678fde9d56fSKevin Wolf 
6790c7bfc32SPaolo Bonzini     logout("finished data write\n");
6804eea78e6SPaolo Bonzini     if (ret < 0) {
6814eea78e6SPaolo Bonzini         return ret;
6824eea78e6SPaolo Bonzini     }
6834eea78e6SPaolo Bonzini 
684bfc45fc1SPaolo Bonzini     if (block) {
6854eea78e6SPaolo Bonzini         /* One or more new blocks were allocated. */
686bfc45fc1SPaolo Bonzini         VdiHeader *header = (VdiHeader *) block;
6874eea78e6SPaolo Bonzini         uint8_t *base;
6884eea78e6SPaolo Bonzini         uint64_t offset;
689fde9d56fSKevin Wolf         uint32_t n_sectors;
6904eea78e6SPaolo Bonzini 
6910c7bfc32SPaolo Bonzini         logout("now writing modified header\n");
692bfc45fc1SPaolo Bonzini         assert(VDI_IS_ALLOCATED(bmap_first));
6930c7bfc32SPaolo Bonzini         *header = s->header;
6940c7bfc32SPaolo Bonzini         vdi_header_to_le(header);
69518d51c4bSKevin Wolf         ret = bdrv_write(bs->file, 0, block, 1);
696bfc45fc1SPaolo Bonzini         g_free(block);
697bfc45fc1SPaolo Bonzini         block = NULL;
6984eea78e6SPaolo Bonzini 
6994eea78e6SPaolo Bonzini         if (ret < 0) {
7004eea78e6SPaolo Bonzini             return ret;
7014eea78e6SPaolo Bonzini         }
7024eea78e6SPaolo Bonzini 
7030c7bfc32SPaolo Bonzini         logout("now writing modified block map entry %u...%u\n",
7040c7bfc32SPaolo Bonzini                bmap_first, bmap_last);
7050c7bfc32SPaolo Bonzini         /* Write modified sectors from block map. */
7060c7bfc32SPaolo Bonzini         bmap_first /= (SECTOR_SIZE / sizeof(uint32_t));
7070c7bfc32SPaolo Bonzini         bmap_last /= (SECTOR_SIZE / sizeof(uint32_t));
7080c7bfc32SPaolo Bonzini         n_sectors = bmap_last - bmap_first + 1;
7090c7bfc32SPaolo Bonzini         offset = s->bmap_sector + bmap_first;
7104eea78e6SPaolo Bonzini         base = ((uint8_t *)&s->bmap[0]) + bmap_first * SECTOR_SIZE;
7110c7bfc32SPaolo Bonzini         logout("will write %u block map sectors starting from entry %u\n",
7120c7bfc32SPaolo Bonzini                n_sectors, bmap_first);
71318d51c4bSKevin Wolf         ret = bdrv_write(bs->file, offset, base, n_sectors);
7140c7bfc32SPaolo Bonzini     }
7150c7bfc32SPaolo Bonzini 
7163d46a75aSPaolo Bonzini     return ret;
7179aebd98aSStefan Weil }
7189aebd98aSStefan Weil 
719004b7f25SChunyan Liu static int vdi_create(const char *filename, QemuOpts *opts, Error **errp)
7209aebd98aSStefan Weil {
721dddc7750SJeff Cody     int ret = 0;
7229aebd98aSStefan Weil     uint64_t bytes = 0;
7239aebd98aSStefan Weil     uint32_t blocks;
72499cce9faSKevin Wolf     size_t block_size = DEFAULT_CLUSTER_SIZE;
7259aebd98aSStefan Weil     uint32_t image_type = VDI_TYPE_DYNAMIC;
7269aebd98aSStefan Weil     VdiHeader header;
7279aebd98aSStefan Weil     size_t i;
7289aebd98aSStefan Weil     size_t bmap_size;
72970747862SJeff Cody     int64_t offset = 0;
73070747862SJeff Cody     Error *local_err = NULL;
731a08f0c3bSKevin Wolf     BlockBackend *blk = NULL;
73270747862SJeff Cody     uint32_t *bmap = NULL;
7339aebd98aSStefan Weil 
7349aebd98aSStefan Weil     logout("\n");
7359aebd98aSStefan Weil 
7369aebd98aSStefan Weil     /* Read out options. */
737c2eb918eSHu Tao     bytes = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
738c2eb918eSHu Tao                      BDRV_SECTOR_SIZE);
7399aebd98aSStefan Weil #if defined(CONFIG_VDI_BLOCK_SIZE)
7409aebd98aSStefan Weil     /* TODO: Additional checks (SECTOR_SIZE * 2^n, ...). */
741004b7f25SChunyan Liu     block_size = qemu_opt_get_size_del(opts,
742004b7f25SChunyan Liu                                        BLOCK_OPT_CLUSTER_SIZE,
743004b7f25SChunyan Liu                                        DEFAULT_CLUSTER_SIZE);
7449aebd98aSStefan Weil #endif
7459aebd98aSStefan Weil #if defined(CONFIG_VDI_STATIC_IMAGE)
746004b7f25SChunyan Liu     if (qemu_opt_get_bool_del(opts, BLOCK_OPT_STATIC, false)) {
7479aebd98aSStefan Weil         image_type = VDI_TYPE_STATIC;
7486eea90ebSStefan Weil     }
7499aebd98aSStefan Weil #endif
7509aebd98aSStefan Weil 
75163fa06dcSJeff Cody     if (bytes > VDI_DISK_SIZE_MAX) {
752dddc7750SJeff Cody         ret = -ENOTSUP;
75363fa06dcSJeff Cody         error_setg(errp, "Unsupported VDI image size (size is 0x%" PRIx64
75463fa06dcSJeff Cody                           ", max supported is 0x%" PRIx64 ")",
75563fa06dcSJeff Cody                           bytes, VDI_DISK_SIZE_MAX);
75663fa06dcSJeff Cody         goto exit;
75763fa06dcSJeff Cody     }
75863fa06dcSJeff Cody 
759dddc7750SJeff Cody     ret = bdrv_create_file(filename, opts, &local_err);
760dddc7750SJeff Cody     if (ret < 0) {
76170747862SJeff Cody         error_propagate(errp, local_err);
76263fa06dcSJeff Cody         goto exit;
7639aebd98aSStefan Weil     }
764a08f0c3bSKevin Wolf 
765efaa7c4eSMax Reitz     blk = blk_new_open(filename, NULL, NULL,
76655880601SKevin Wolf                        BDRV_O_RDWR | BDRV_O_RESIZE | BDRV_O_PROTOCOL,
76755880601SKevin Wolf                        &local_err);
768a08f0c3bSKevin Wolf     if (blk == NULL) {
76970747862SJeff Cody         error_propagate(errp, local_err);
770a08f0c3bSKevin Wolf         ret = -EIO;
77170747862SJeff Cody         goto exit;
7724ab15590SChunyan Liu     }
7734ab15590SChunyan Liu 
774a08f0c3bSKevin Wolf     blk_set_allow_write_beyond_eof(blk, true);
775a08f0c3bSKevin Wolf 
776f21dc3a4SStefan Weil     /* We need enough blocks to store the given disk size,
777f21dc3a4SStefan Weil        so always round up. */
778e9082e47SMax Reitz     blocks = DIV_ROUND_UP(bytes, block_size);
779f21dc3a4SStefan Weil 
7809aebd98aSStefan Weil     bmap_size = blocks * sizeof(uint32_t);
781e9082e47SMax Reitz     bmap_size = ROUND_UP(bmap_size, SECTOR_SIZE);
7829aebd98aSStefan Weil 
7839aebd98aSStefan Weil     memset(&header, 0, sizeof(header));
7841786dc15SBlue Swirl     pstrcpy(header.text, sizeof(header.text), VDI_TEXT);
7859aebd98aSStefan Weil     header.signature = VDI_SIGNATURE;
7869aebd98aSStefan Weil     header.version = VDI_VERSION_1_1;
7879aebd98aSStefan Weil     header.header_size = 0x180;
7889aebd98aSStefan Weil     header.image_type = image_type;
7899aebd98aSStefan Weil     header.offset_bmap = 0x200;
7909aebd98aSStefan Weil     header.offset_data = 0x200 + bmap_size;
7919aebd98aSStefan Weil     header.sector_size = SECTOR_SIZE;
7929aebd98aSStefan Weil     header.disk_size = bytes;
7939aebd98aSStefan Weil     header.block_size = block_size;
7949aebd98aSStefan Weil     header.blocks_in_image = blocks;
7956eea90ebSStefan Weil     if (image_type == VDI_TYPE_STATIC) {
7966eea90ebSStefan Weil         header.blocks_allocated = blocks;
7976eea90ebSStefan Weil     }
7987c6f55b6SFam Zheng     qemu_uuid_generate(&header.uuid_image);
7997c6f55b6SFam Zheng     qemu_uuid_generate(&header.uuid_last_snap);
8009aebd98aSStefan Weil     /* There is no need to set header.uuid_link or header.uuid_parent here. */
8019aebd98aSStefan Weil #if defined(CONFIG_VDI_DEBUG)
8029aebd98aSStefan Weil     vdi_header_print(&header);
8039aebd98aSStefan Weil #endif
8049aebd98aSStefan Weil     vdi_header_to_le(&header);
8058341f00dSEric Blake     ret = blk_pwrite(blk, offset, &header, sizeof(header), 0);
806dddc7750SJeff Cody     if (ret < 0) {
80770747862SJeff Cody         error_setg(errp, "Error writing header to %s", filename);
80870747862SJeff Cody         goto exit;
8099aebd98aSStefan Weil     }
81070747862SJeff Cody     offset += sizeof(header);
8119aebd98aSStefan Weil 
812b76b6e95SStefan Weil     if (bmap_size > 0) {
81317cce735SKevin Wolf         bmap = g_try_malloc0(bmap_size);
81417cce735SKevin Wolf         if (bmap == NULL) {
81517cce735SKevin Wolf             ret = -ENOMEM;
81617cce735SKevin Wolf             error_setg(errp, "Could not allocate bmap");
81717cce735SKevin Wolf             goto exit;
81817cce735SKevin Wolf         }
8199aebd98aSStefan Weil         for (i = 0; i < blocks; i++) {
8209aebd98aSStefan Weil             if (image_type == VDI_TYPE_STATIC) {
8219aebd98aSStefan Weil                 bmap[i] = i;
8229aebd98aSStefan Weil             } else {
8239aebd98aSStefan Weil                 bmap[i] = VDI_UNALLOCATED;
8249aebd98aSStefan Weil             }
8259aebd98aSStefan Weil         }
8268341f00dSEric Blake         ret = blk_pwrite(blk, offset, bmap, bmap_size, 0);
827dddc7750SJeff Cody         if (ret < 0) {
82870747862SJeff Cody             error_setg(errp, "Error writing bmap to %s", filename);
82970747862SJeff Cody             goto exit;
8309aebd98aSStefan Weil         }
83170747862SJeff Cody         offset += bmap_size;
832514f21a5SStefan Weil     }
833514f21a5SStefan Weil 
8349aebd98aSStefan Weil     if (image_type == VDI_TYPE_STATIC) {
835*3a691c50SMax Reitz         ret = blk_truncate(blk, offset + blocks * block_size,
836*3a691c50SMax Reitz                            PREALLOC_MODE_OFF, errp);
837dddc7750SJeff Cody         if (ret < 0) {
838ed3d2ec9SMax Reitz             error_prepend(errp, "Failed to statically allocate %s", filename);
83970747862SJeff Cody             goto exit;
8409aebd98aSStefan Weil         }
8419aebd98aSStefan Weil     }
8429aebd98aSStefan Weil 
84363fa06dcSJeff Cody exit:
844a08f0c3bSKevin Wolf     blk_unref(blk);
84570747862SJeff Cody     g_free(bmap);
846dddc7750SJeff Cody     return ret;
8479aebd98aSStefan Weil }
8489aebd98aSStefan Weil 
8499aebd98aSStefan Weil static void vdi_close(BlockDriverState *bs)
8509aebd98aSStefan Weil {
851fc9d106cSKevin Wolf     BDRVVdiState *s = bs->opaque;
8526ac5f388SKevin Wolf 
85317cce735SKevin Wolf     qemu_vfree(s->bmap);
8546ac5f388SKevin Wolf 
855fc9d106cSKevin Wolf     migrate_del_blocker(s->migration_blocker);
856fc9d106cSKevin Wolf     error_free(s->migration_blocker);
8579aebd98aSStefan Weil }
8589aebd98aSStefan Weil 
859004b7f25SChunyan Liu static QemuOptsList vdi_create_opts = {
860004b7f25SChunyan Liu     .name = "vdi-create-opts",
861004b7f25SChunyan Liu     .head = QTAILQ_HEAD_INITIALIZER(vdi_create_opts.head),
862004b7f25SChunyan Liu     .desc = {
8639aebd98aSStefan Weil         {
8649aebd98aSStefan Weil             .name = BLOCK_OPT_SIZE,
865004b7f25SChunyan Liu             .type = QEMU_OPT_SIZE,
8669aebd98aSStefan Weil             .help = "Virtual disk size"
8679aebd98aSStefan Weil         },
8689aebd98aSStefan Weil #if defined(CONFIG_VDI_BLOCK_SIZE)
8699aebd98aSStefan Weil         {
8709aebd98aSStefan Weil             .name = BLOCK_OPT_CLUSTER_SIZE,
871004b7f25SChunyan Liu             .type = QEMU_OPT_SIZE,
87299cce9faSKevin Wolf             .help = "VDI cluster (block) size",
873004b7f25SChunyan Liu             .def_value_str = stringify(DEFAULT_CLUSTER_SIZE)
8749aebd98aSStefan Weil         },
8759aebd98aSStefan Weil #endif
8769aebd98aSStefan Weil #if defined(CONFIG_VDI_STATIC_IMAGE)
8779aebd98aSStefan Weil         {
8789aebd98aSStefan Weil             .name = BLOCK_OPT_STATIC,
879004b7f25SChunyan Liu             .type = QEMU_OPT_BOOL,
880004b7f25SChunyan Liu             .help = "VDI static (pre-allocated) image",
881004b7f25SChunyan Liu             .def_value_str = "off"
8829aebd98aSStefan Weil         },
8839aebd98aSStefan Weil #endif
8849aebd98aSStefan Weil         /* TODO: An additional option to set UUID values might be useful. */
885004b7f25SChunyan Liu         { /* end of list */ }
886004b7f25SChunyan Liu     }
8879aebd98aSStefan Weil };
8889aebd98aSStefan Weil 
8899aebd98aSStefan Weil static BlockDriver bdrv_vdi = {
8909aebd98aSStefan Weil     .format_name = "vdi",
8919aebd98aSStefan Weil     .instance_size = sizeof(BDRVVdiState),
8929aebd98aSStefan Weil     .bdrv_probe = vdi_probe,
8939aebd98aSStefan Weil     .bdrv_open = vdi_open,
8949aebd98aSStefan Weil     .bdrv_close = vdi_close,
895ecfe2bbaSJeff Cody     .bdrv_reopen_prepare = vdi_reopen_prepare,
896862f215fSKevin Wolf     .bdrv_child_perm          = bdrv_format_default_perms,
897c282e1fdSChunyan Liu     .bdrv_create = vdi_create,
8983ac21627SPeter Lieven     .bdrv_has_zero_init = bdrv_has_zero_init_1,
899b6b8a333SPaolo Bonzini     .bdrv_co_get_block_status = vdi_co_get_block_status,
9009aebd98aSStefan Weil     .bdrv_make_empty = vdi_make_empty,
9019aebd98aSStefan Weil 
9020865bb6fSKevin Wolf     .bdrv_co_preadv     = vdi_co_preadv,
9039aebd98aSStefan Weil #if defined(CONFIG_VDI_WRITE)
904fde9d56fSKevin Wolf     .bdrv_co_pwritev    = vdi_co_pwritev,
9059aebd98aSStefan Weil #endif
9069aebd98aSStefan Weil 
9079aebd98aSStefan Weil     .bdrv_get_info = vdi_get_info,
9089aebd98aSStefan Weil 
909004b7f25SChunyan Liu     .create_opts = &vdi_create_opts,
9109aebd98aSStefan Weil     .bdrv_check = vdi_check,
9119aebd98aSStefan Weil };
9129aebd98aSStefan Weil 
9139aebd98aSStefan Weil static void bdrv_vdi_init(void)
9149aebd98aSStefan Weil {
9159aebd98aSStefan Weil     logout("\n");
9169aebd98aSStefan Weil     bdrv_register(&bdrv_vdi);
9179aebd98aSStefan Weil }
9189aebd98aSStefan Weil 
9199aebd98aSStefan Weil block_init(bdrv_vdi_init);
920