xref: /qemu/block/vdi.c (revision fbcbbf4e)
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"
58caf71f86SPaolo Bonzini #include "migration/migration.h"
5910817bf0SDaniel P. Berrange #include "qemu/coroutine.h"
60f348b6d1SVeronia Bahaa #include "qemu/cutils.h"
619aebd98aSStefan Weil 
62ee682d27SStefan Weil #if defined(CONFIG_UUID)
639aebd98aSStefan Weil #include <uuid/uuid.h>
649aebd98aSStefan Weil #else
659aebd98aSStefan Weil /* TODO: move uuid emulation to some central place in QEMU. */
669c17d615SPaolo Bonzini #include "sysemu/sysemu.h"     /* UUID_FMT */
679aebd98aSStefan Weil typedef unsigned char uuid_t[16];
689aebd98aSStefan Weil #endif
699aebd98aSStefan Weil 
709aebd98aSStefan Weil /* Code configuration options. */
719aebd98aSStefan Weil 
729aebd98aSStefan Weil /* Enable debug messages. */
739aebd98aSStefan Weil //~ #define CONFIG_VDI_DEBUG
749aebd98aSStefan Weil 
759aebd98aSStefan Weil /* Support write operations on VDI images. */
769aebd98aSStefan Weil #define CONFIG_VDI_WRITE
779aebd98aSStefan Weil 
789aebd98aSStefan Weil /* Support non-standard block (cluster) size. This is untested.
799aebd98aSStefan Weil  * Maybe it will be needed for very large images.
809aebd98aSStefan Weil  */
819aebd98aSStefan Weil //~ #define CONFIG_VDI_BLOCK_SIZE
829aebd98aSStefan Weil 
839aebd98aSStefan Weil /* Support static (fixed, pre-allocated) images. */
849aebd98aSStefan Weil #define CONFIG_VDI_STATIC_IMAGE
859aebd98aSStefan Weil 
869aebd98aSStefan Weil /* Command line option for static images. */
879aebd98aSStefan Weil #define BLOCK_OPT_STATIC "static"
889aebd98aSStefan Weil 
899aebd98aSStefan Weil #define KiB     1024
909aebd98aSStefan Weil #define MiB     (KiB * KiB)
919aebd98aSStefan Weil 
929aebd98aSStefan Weil #define SECTOR_SIZE 512
9399cce9faSKevin Wolf #define DEFAULT_CLUSTER_SIZE (1 * MiB)
949aebd98aSStefan Weil 
959aebd98aSStefan Weil #if defined(CONFIG_VDI_DEBUG)
969aebd98aSStefan Weil #define logout(fmt, ...) \
979aebd98aSStefan Weil                 fprintf(stderr, "vdi\t%-24s" fmt, __func__, ##__VA_ARGS__)
989aebd98aSStefan Weil #else
999aebd98aSStefan Weil #define logout(fmt, ...) ((void)0)
1009aebd98aSStefan Weil #endif
1019aebd98aSStefan Weil 
1029aebd98aSStefan Weil /* Image signature. */
1039aebd98aSStefan Weil #define VDI_SIGNATURE 0xbeda107f
1049aebd98aSStefan Weil 
1059aebd98aSStefan Weil /* Image version. */
1069aebd98aSStefan Weil #define VDI_VERSION_1_1 0x00010001
1079aebd98aSStefan Weil 
1089aebd98aSStefan Weil /* Image type. */
1099aebd98aSStefan Weil #define VDI_TYPE_DYNAMIC 1
1109aebd98aSStefan Weil #define VDI_TYPE_STATIC  2
1119aebd98aSStefan Weil 
1129aebd98aSStefan Weil /* Innotek / SUN images use these strings in header.text:
1139aebd98aSStefan Weil  * "<<< innotek VirtualBox Disk Image >>>\n"
1149aebd98aSStefan Weil  * "<<< Sun xVM VirtualBox Disk Image >>>\n"
1159aebd98aSStefan Weil  * "<<< Sun VirtualBox Disk Image >>>\n"
1169aebd98aSStefan Weil  * The value does not matter, so QEMU created images use a different text.
1179aebd98aSStefan Weil  */
1189aebd98aSStefan Weil #define VDI_TEXT "<<< QEMU VM Virtual Disk Image >>>\n"
1199aebd98aSStefan Weil 
120c794b4e0SEric Sunshine /* A never-allocated block; semantically arbitrary content. */
121c794b4e0SEric Sunshine #define VDI_UNALLOCATED 0xffffffffU
122c794b4e0SEric Sunshine 
123c794b4e0SEric Sunshine /* A discarded (no longer allocated) block; semantically zero-filled. */
124c794b4e0SEric Sunshine #define VDI_DISCARDED   0xfffffffeU
125c794b4e0SEric Sunshine 
126c794b4e0SEric Sunshine #define VDI_IS_ALLOCATED(X) ((X) < VDI_DISCARDED)
1279aebd98aSStefan Weil 
128d20418eeSMax Reitz /* The bmap will take up VDI_BLOCKS_IN_IMAGE_MAX * sizeof(uint32_t) bytes; since
129d20418eeSMax Reitz  * the bmap is read and written in a single operation, its size needs to be
130d20418eeSMax Reitz  * limited to INT_MAX; furthermore, when opening an image, the bmap size is
131d20418eeSMax Reitz  * rounded up to be aligned on BDRV_SECTOR_SIZE.
132d20418eeSMax Reitz  * Therefore this should satisfy the following:
133d20418eeSMax Reitz  * VDI_BLOCKS_IN_IMAGE_MAX * sizeof(uint32_t) + BDRV_SECTOR_SIZE == INT_MAX + 1
134d20418eeSMax Reitz  * (INT_MAX + 1 is the first value not representable as an int)
135d20418eeSMax Reitz  * This guarantees that any value below or equal to the constant will, when
136d20418eeSMax Reitz  * multiplied by sizeof(uint32_t) and rounded up to a BDRV_SECTOR_SIZE boundary,
137d20418eeSMax Reitz  * still be below or equal to INT_MAX. */
138d20418eeSMax Reitz #define VDI_BLOCKS_IN_IMAGE_MAX \
139d20418eeSMax Reitz     ((unsigned)((INT_MAX + 1u - BDRV_SECTOR_SIZE) / sizeof(uint32_t)))
14063fa06dcSJeff Cody #define VDI_DISK_SIZE_MAX        ((uint64_t)VDI_BLOCKS_IN_IMAGE_MAX * \
14163fa06dcSJeff Cody                                   (uint64_t)DEFAULT_CLUSTER_SIZE)
14263fa06dcSJeff Cody 
143ee682d27SStefan Weil #if !defined(CONFIG_UUID)
1448ba2aae3SStefan Hajnoczi static inline void uuid_generate(uuid_t out)
1459aebd98aSStefan Weil {
1464f3669eaSStefan Weil     memset(out, 0, sizeof(uuid_t));
1479aebd98aSStefan Weil }
1489aebd98aSStefan Weil 
1498ba2aae3SStefan Hajnoczi static inline int uuid_is_null(const uuid_t uu)
1509aebd98aSStefan Weil {
1519aebd98aSStefan Weil     uuid_t null_uuid = { 0 };
1524f3669eaSStefan Weil     return memcmp(uu, null_uuid, sizeof(uuid_t)) == 0;
1539aebd98aSStefan Weil }
1549aebd98aSStefan Weil 
155f18a768eSSeokYeon Hwang # if defined(CONFIG_VDI_DEBUG)
1568ba2aae3SStefan Hajnoczi static inline void uuid_unparse(const uuid_t uu, char *out)
1579aebd98aSStefan Weil {
1589aebd98aSStefan Weil     snprintf(out, 37, UUID_FMT,
1599aebd98aSStefan Weil             uu[0], uu[1], uu[2], uu[3], uu[4], uu[5], uu[6], uu[7],
1609aebd98aSStefan Weil             uu[8], uu[9], uu[10], uu[11], uu[12], uu[13], uu[14], uu[15]);
1619aebd98aSStefan Weil }
1629aebd98aSStefan Weil # endif
163f18a768eSSeokYeon Hwang #endif
1649aebd98aSStefan Weil 
1659aebd98aSStefan Weil typedef struct {
1669aebd98aSStefan Weil     char text[0x40];
1679aebd98aSStefan Weil     uint32_t signature;
1689aebd98aSStefan Weil     uint32_t version;
1699aebd98aSStefan Weil     uint32_t header_size;
1709aebd98aSStefan Weil     uint32_t image_type;
1719aebd98aSStefan Weil     uint32_t image_flags;
1729aebd98aSStefan Weil     char description[256];
1739aebd98aSStefan Weil     uint32_t offset_bmap;
1749aebd98aSStefan Weil     uint32_t offset_data;
1759aebd98aSStefan Weil     uint32_t cylinders;         /* disk geometry, unused here */
1769aebd98aSStefan Weil     uint32_t heads;             /* disk geometry, unused here */
1779aebd98aSStefan Weil     uint32_t sectors;           /* disk geometry, unused here */
1789aebd98aSStefan Weil     uint32_t sector_size;
1799aebd98aSStefan Weil     uint32_t unused1;
1809aebd98aSStefan Weil     uint64_t disk_size;
1819aebd98aSStefan Weil     uint32_t block_size;
1829aebd98aSStefan Weil     uint32_t block_extra;       /* unused here */
1839aebd98aSStefan Weil     uint32_t blocks_in_image;
1849aebd98aSStefan Weil     uint32_t blocks_allocated;
1859aebd98aSStefan Weil     uuid_t uuid_image;
1869aebd98aSStefan Weil     uuid_t uuid_last_snap;
1879aebd98aSStefan Weil     uuid_t uuid_link;
1889aebd98aSStefan Weil     uuid_t uuid_parent;
1899aebd98aSStefan Weil     uint64_t unused2[7];
1908368febdSJeff Cody } QEMU_PACKED VdiHeader;
1919aebd98aSStefan Weil 
1929aebd98aSStefan Weil typedef struct {
1939aebd98aSStefan Weil     /* The block map entries are little endian (even in memory). */
1949aebd98aSStefan Weil     uint32_t *bmap;
1959aebd98aSStefan Weil     /* Size of block (bytes). */
1969aebd98aSStefan Weil     uint32_t block_size;
1979aebd98aSStefan Weil     /* Size of block (sectors). */
1989aebd98aSStefan Weil     uint32_t block_sectors;
1999aebd98aSStefan Weil     /* First sector of block map. */
2009aebd98aSStefan Weil     uint32_t bmap_sector;
2014ff9786cSStefan Weil     /* VDI header (converted to host endianness). */
2029aebd98aSStefan Weil     VdiHeader header;
203fc9d106cSKevin Wolf 
204f0ab6f10SMax Reitz     CoMutex write_lock;
205f0ab6f10SMax Reitz 
206fc9d106cSKevin Wolf     Error *migration_blocker;
2079aebd98aSStefan Weil } BDRVVdiState;
2089aebd98aSStefan Weil 
2099aebd98aSStefan Weil /* Change UUID from little endian (IPRT = VirtualBox format) to big endian
2109aebd98aSStefan Weil  * format (network byte order, standard, see RFC 4122) and vice versa.
2119aebd98aSStefan Weil  */
2129aebd98aSStefan Weil static void uuid_convert(uuid_t uuid)
2139aebd98aSStefan Weil {
2149aebd98aSStefan Weil     bswap32s((uint32_t *)&uuid[0]);
2159aebd98aSStefan Weil     bswap16s((uint16_t *)&uuid[4]);
2169aebd98aSStefan Weil     bswap16s((uint16_t *)&uuid[6]);
2179aebd98aSStefan Weil }
2189aebd98aSStefan Weil 
2199aebd98aSStefan Weil static void vdi_header_to_cpu(VdiHeader *header)
2209aebd98aSStefan Weil {
2219aebd98aSStefan Weil     le32_to_cpus(&header->signature);
2229aebd98aSStefan Weil     le32_to_cpus(&header->version);
2239aebd98aSStefan Weil     le32_to_cpus(&header->header_size);
2249aebd98aSStefan Weil     le32_to_cpus(&header->image_type);
2259aebd98aSStefan Weil     le32_to_cpus(&header->image_flags);
2269aebd98aSStefan Weil     le32_to_cpus(&header->offset_bmap);
2279aebd98aSStefan Weil     le32_to_cpus(&header->offset_data);
2289aebd98aSStefan Weil     le32_to_cpus(&header->cylinders);
2299aebd98aSStefan Weil     le32_to_cpus(&header->heads);
2309aebd98aSStefan Weil     le32_to_cpus(&header->sectors);
2319aebd98aSStefan Weil     le32_to_cpus(&header->sector_size);
2329aebd98aSStefan Weil     le64_to_cpus(&header->disk_size);
2339aebd98aSStefan Weil     le32_to_cpus(&header->block_size);
2349aebd98aSStefan Weil     le32_to_cpus(&header->block_extra);
2359aebd98aSStefan Weil     le32_to_cpus(&header->blocks_in_image);
2369aebd98aSStefan Weil     le32_to_cpus(&header->blocks_allocated);
2379aebd98aSStefan Weil     uuid_convert(header->uuid_image);
2389aebd98aSStefan Weil     uuid_convert(header->uuid_last_snap);
2399aebd98aSStefan Weil     uuid_convert(header->uuid_link);
2409aebd98aSStefan Weil     uuid_convert(header->uuid_parent);
2419aebd98aSStefan Weil }
2429aebd98aSStefan Weil 
2439aebd98aSStefan Weil static void vdi_header_to_le(VdiHeader *header)
2449aebd98aSStefan Weil {
2459aebd98aSStefan Weil     cpu_to_le32s(&header->signature);
2469aebd98aSStefan Weil     cpu_to_le32s(&header->version);
2479aebd98aSStefan Weil     cpu_to_le32s(&header->header_size);
2489aebd98aSStefan Weil     cpu_to_le32s(&header->image_type);
2499aebd98aSStefan Weil     cpu_to_le32s(&header->image_flags);
2509aebd98aSStefan Weil     cpu_to_le32s(&header->offset_bmap);
2519aebd98aSStefan Weil     cpu_to_le32s(&header->offset_data);
2529aebd98aSStefan Weil     cpu_to_le32s(&header->cylinders);
2539aebd98aSStefan Weil     cpu_to_le32s(&header->heads);
2549aebd98aSStefan Weil     cpu_to_le32s(&header->sectors);
2559aebd98aSStefan Weil     cpu_to_le32s(&header->sector_size);
2569aebd98aSStefan Weil     cpu_to_le64s(&header->disk_size);
2579aebd98aSStefan Weil     cpu_to_le32s(&header->block_size);
2589aebd98aSStefan Weil     cpu_to_le32s(&header->block_extra);
2599aebd98aSStefan Weil     cpu_to_le32s(&header->blocks_in_image);
2609aebd98aSStefan Weil     cpu_to_le32s(&header->blocks_allocated);
2619aebd98aSStefan Weil     uuid_convert(header->uuid_image);
2629aebd98aSStefan Weil     uuid_convert(header->uuid_last_snap);
2639aebd98aSStefan Weil     uuid_convert(header->uuid_link);
2649aebd98aSStefan Weil     uuid_convert(header->uuid_parent);
2659aebd98aSStefan Weil }
2669aebd98aSStefan Weil 
2679aebd98aSStefan Weil #if defined(CONFIG_VDI_DEBUG)
2689aebd98aSStefan Weil static void vdi_header_print(VdiHeader *header)
2699aebd98aSStefan Weil {
2709aebd98aSStefan Weil     char uuid[37];
2719aebd98aSStefan Weil     logout("text        %s", header->text);
2729f0470bbSStefan Weil     logout("signature   0x%08x\n", header->signature);
2739aebd98aSStefan Weil     logout("header size 0x%04x\n", header->header_size);
2749aebd98aSStefan Weil     logout("image type  0x%04x\n", header->image_type);
2759aebd98aSStefan Weil     logout("image flags 0x%04x\n", header->image_flags);
2769aebd98aSStefan Weil     logout("description %s\n", header->description);
2779aebd98aSStefan Weil     logout("offset bmap 0x%04x\n", header->offset_bmap);
2789aebd98aSStefan Weil     logout("offset data 0x%04x\n", header->offset_data);
2799aebd98aSStefan Weil     logout("cylinders   0x%04x\n", header->cylinders);
2809aebd98aSStefan Weil     logout("heads       0x%04x\n", header->heads);
2819aebd98aSStefan Weil     logout("sectors     0x%04x\n", header->sectors);
2829aebd98aSStefan Weil     logout("sector size 0x%04x\n", header->sector_size);
2839aebd98aSStefan Weil     logout("image size  0x%" PRIx64 " B (%" PRIu64 " MiB)\n",
2849aebd98aSStefan Weil            header->disk_size, header->disk_size / MiB);
2859aebd98aSStefan Weil     logout("block size  0x%04x\n", header->block_size);
2869aebd98aSStefan Weil     logout("block extra 0x%04x\n", header->block_extra);
2879aebd98aSStefan Weil     logout("blocks tot. 0x%04x\n", header->blocks_in_image);
2889aebd98aSStefan Weil     logout("blocks all. 0x%04x\n", header->blocks_allocated);
2899aebd98aSStefan Weil     uuid_unparse(header->uuid_image, uuid);
2909aebd98aSStefan Weil     logout("uuid image  %s\n", uuid);
2919aebd98aSStefan Weil     uuid_unparse(header->uuid_last_snap, uuid);
2929aebd98aSStefan Weil     logout("uuid snap   %s\n", uuid);
2939aebd98aSStefan Weil     uuid_unparse(header->uuid_link, uuid);
2949aebd98aSStefan Weil     logout("uuid link   %s\n", uuid);
2959aebd98aSStefan Weil     uuid_unparse(header->uuid_parent, uuid);
2969aebd98aSStefan Weil     logout("uuid parent %s\n", uuid);
2979aebd98aSStefan Weil }
2989aebd98aSStefan Weil #endif
2999aebd98aSStefan Weil 
3004534ff54SKevin Wolf static int vdi_check(BlockDriverState *bs, BdrvCheckResult *res,
3014534ff54SKevin Wolf                      BdrvCheckMode fix)
3029aebd98aSStefan Weil {
3039aebd98aSStefan Weil     /* TODO: additional checks possible. */
3049aebd98aSStefan Weil     BDRVVdiState *s = (BDRVVdiState *)bs->opaque;
3059aebd98aSStefan Weil     uint32_t blocks_allocated = 0;
3069aebd98aSStefan Weil     uint32_t block;
3079aebd98aSStefan Weil     uint32_t *bmap;
3089aebd98aSStefan Weil     logout("\n");
3099aebd98aSStefan Weil 
3104534ff54SKevin Wolf     if (fix) {
3114534ff54SKevin Wolf         return -ENOTSUP;
3124534ff54SKevin Wolf     }
3134534ff54SKevin Wolf 
3145839e53bSMarkus Armbruster     bmap = g_try_new(uint32_t, s->header.blocks_in_image);
31517cce735SKevin Wolf     if (s->header.blocks_in_image && bmap == NULL) {
31617cce735SKevin Wolf         res->check_errors++;
31717cce735SKevin Wolf         return -ENOMEM;
31817cce735SKevin Wolf     }
31917cce735SKevin Wolf 
3209aebd98aSStefan Weil     memset(bmap, 0xff, s->header.blocks_in_image * sizeof(uint32_t));
3219aebd98aSStefan Weil 
3229aebd98aSStefan Weil     /* Check block map and value of blocks_allocated. */
3239aebd98aSStefan Weil     for (block = 0; block < s->header.blocks_in_image; block++) {
3249aebd98aSStefan Weil         uint32_t bmap_entry = le32_to_cpu(s->bmap[block]);
325c794b4e0SEric Sunshine         if (VDI_IS_ALLOCATED(bmap_entry)) {
3269aebd98aSStefan Weil             if (bmap_entry < s->header.blocks_in_image) {
3279aebd98aSStefan Weil                 blocks_allocated++;
328c794b4e0SEric Sunshine                 if (!VDI_IS_ALLOCATED(bmap[bmap_entry])) {
3299aebd98aSStefan Weil                     bmap[bmap_entry] = bmap_entry;
3309aebd98aSStefan Weil                 } else {
3319aebd98aSStefan Weil                     fprintf(stderr, "ERROR: block index %" PRIu32
3329aebd98aSStefan Weil                             " also used by %" PRIu32 "\n", bmap[bmap_entry], bmap_entry);
3339ac228e0SKevin Wolf                     res->corruptions++;
3349aebd98aSStefan Weil                 }
3359aebd98aSStefan Weil             } else {
3369aebd98aSStefan Weil                 fprintf(stderr, "ERROR: block index %" PRIu32
3379aebd98aSStefan Weil                         " too large, is %" PRIu32 "\n", block, bmap_entry);
3389ac228e0SKevin Wolf                 res->corruptions++;
3399aebd98aSStefan Weil             }
3409aebd98aSStefan Weil         }
3419aebd98aSStefan Weil     }
3429aebd98aSStefan Weil     if (blocks_allocated != s->header.blocks_allocated) {
3439aebd98aSStefan Weil         fprintf(stderr, "ERROR: allocated blocks mismatch, is %" PRIu32
3449aebd98aSStefan Weil                ", should be %" PRIu32 "\n",
3459aebd98aSStefan Weil                blocks_allocated, s->header.blocks_allocated);
3469ac228e0SKevin Wolf         res->corruptions++;
3479aebd98aSStefan Weil     }
3489aebd98aSStefan Weil 
3497267c094SAnthony Liguori     g_free(bmap);
3509aebd98aSStefan Weil 
3519ac228e0SKevin Wolf     return 0;
3529aebd98aSStefan Weil }
3539aebd98aSStefan Weil 
3549aebd98aSStefan Weil static int vdi_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
3559aebd98aSStefan Weil {
3569aebd98aSStefan Weil     /* TODO: vdi_get_info would be needed for machine snapshots.
3579aebd98aSStefan Weil        vm_state_offset is still missing. */
3589aebd98aSStefan Weil     BDRVVdiState *s = (BDRVVdiState *)bs->opaque;
3599aebd98aSStefan Weil     logout("\n");
3609aebd98aSStefan Weil     bdi->cluster_size = s->block_size;
3619aebd98aSStefan Weil     bdi->vm_state_offset = 0;
36295de6d70SPaolo Bonzini     bdi->unallocated_blocks_are_zero = true;
3639aebd98aSStefan Weil     return 0;
3649aebd98aSStefan Weil }
3659aebd98aSStefan Weil 
3669aebd98aSStefan Weil static int vdi_make_empty(BlockDriverState *bs)
3679aebd98aSStefan Weil {
3689aebd98aSStefan Weil     /* TODO: missing code. */
3699aebd98aSStefan Weil     logout("\n");
3709aebd98aSStefan Weil     /* The return value for missing code must be 0, see block.c. */
3719aebd98aSStefan Weil     return 0;
3729aebd98aSStefan Weil }
3739aebd98aSStefan Weil 
3749aebd98aSStefan Weil static int vdi_probe(const uint8_t *buf, int buf_size, const char *filename)
3759aebd98aSStefan Weil {
3769aebd98aSStefan Weil     const VdiHeader *header = (const VdiHeader *)buf;
377dddc7750SJeff Cody     int ret = 0;
3789aebd98aSStefan Weil 
3799aebd98aSStefan Weil     logout("\n");
3809aebd98aSStefan Weil 
3819aebd98aSStefan Weil     if (buf_size < sizeof(*header)) {
3829aebd98aSStefan Weil         /* Header too small, no VDI. */
3839aebd98aSStefan Weil     } else if (le32_to_cpu(header->signature) == VDI_SIGNATURE) {
384dddc7750SJeff Cody         ret = 100;
3859aebd98aSStefan Weil     }
3869aebd98aSStefan Weil 
387dddc7750SJeff Cody     if (ret == 0) {
3889aebd98aSStefan Weil         logout("no vdi image\n");
3899aebd98aSStefan Weil     } else {
3909aebd98aSStefan Weil         logout("%s", header->text);
3919aebd98aSStefan Weil     }
3929aebd98aSStefan Weil 
393dddc7750SJeff Cody     return ret;
3949aebd98aSStefan Weil }
3959aebd98aSStefan Weil 
396015a1036SMax Reitz static int vdi_open(BlockDriverState *bs, QDict *options, int flags,
397015a1036SMax Reitz                     Error **errp)
3989aebd98aSStefan Weil {
3999aebd98aSStefan Weil     BDRVVdiState *s = bs->opaque;
4009aebd98aSStefan Weil     VdiHeader header;
4019aebd98aSStefan Weil     size_t bmap_size;
4028937f822SStefan Weil     int ret;
4039aebd98aSStefan Weil 
4049aebd98aSStefan Weil     logout("\n");
4059aebd98aSStefan Weil 
406*fbcbbf4eSKevin Wolf     ret = bdrv_read(bs->file, 0, (uint8_t *)&header, 1);
4078937f822SStefan Weil     if (ret < 0) {
4089aebd98aSStefan Weil         goto fail;
4099aebd98aSStefan Weil     }
4109aebd98aSStefan Weil 
4119aebd98aSStefan Weil     vdi_header_to_cpu(&header);
4129aebd98aSStefan Weil #if defined(CONFIG_VDI_DEBUG)
4139aebd98aSStefan Weil     vdi_header_print(&header);
4149aebd98aSStefan Weil #endif
4159aebd98aSStefan Weil 
41663fa06dcSJeff Cody     if (header.disk_size > VDI_DISK_SIZE_MAX) {
41763fa06dcSJeff Cody         error_setg(errp, "Unsupported VDI image size (size is 0x%" PRIx64
41863fa06dcSJeff Cody                           ", max supported is 0x%" PRIx64 ")",
41963fa06dcSJeff Cody                           header.disk_size, VDI_DISK_SIZE_MAX);
42063fa06dcSJeff Cody         ret = -ENOTSUP;
42163fa06dcSJeff Cody         goto fail;
42263fa06dcSJeff Cody     }
42363fa06dcSJeff Cody 
424f21dc3a4SStefan Weil     if (header.disk_size % SECTOR_SIZE != 0) {
425f21dc3a4SStefan Weil         /* 'VBoxManage convertfromraw' can create images with odd disk sizes.
426f21dc3a4SStefan Weil            We accept them but round the disk size to the next multiple of
427f21dc3a4SStefan Weil            SECTOR_SIZE. */
428f21dc3a4SStefan Weil         logout("odd disk size %" PRIu64 " B, round up\n", header.disk_size);
429e9082e47SMax Reitz         header.disk_size = ROUND_UP(header.disk_size, SECTOR_SIZE);
430f21dc3a4SStefan Weil     }
431f21dc3a4SStefan Weil 
4320e87ba2cSStefan Weil     if (header.signature != VDI_SIGNATURE) {
433521b2b5dSMax Reitz         error_setg(errp, "Image not in VDI format (bad signature %08" PRIx32
434521b2b5dSMax Reitz                    ")", header.signature);
43576abe407SPaolo Bonzini         ret = -EINVAL;
4360e87ba2cSStefan Weil         goto fail;
4370e87ba2cSStefan Weil     } else if (header.version != VDI_VERSION_1_1) {
438521b2b5dSMax Reitz         error_setg(errp, "unsupported VDI image (version %" PRIu32 ".%" PRIu32
439521b2b5dSMax Reitz                    ")", header.version >> 16, header.version & 0xffff);
4408937f822SStefan Weil         ret = -ENOTSUP;
4419aebd98aSStefan Weil         goto fail;
4429aebd98aSStefan Weil     } else if (header.offset_bmap % SECTOR_SIZE != 0) {
4439aebd98aSStefan Weil         /* We only support block maps which start on a sector boundary. */
4445b7aa9b5SPaolo Bonzini         error_setg(errp, "unsupported VDI image (unaligned block map offset "
445521b2b5dSMax Reitz                    "0x%" PRIx32 ")", header.offset_bmap);
4468937f822SStefan Weil         ret = -ENOTSUP;
4479aebd98aSStefan Weil         goto fail;
4489aebd98aSStefan Weil     } else if (header.offset_data % SECTOR_SIZE != 0) {
4499aebd98aSStefan Weil         /* We only support data blocks which start on a sector boundary. */
450521b2b5dSMax Reitz         error_setg(errp, "unsupported VDI image (unaligned data offset 0x%"
451521b2b5dSMax Reitz                    PRIx32 ")", header.offset_data);
4528937f822SStefan Weil         ret = -ENOTSUP;
4539aebd98aSStefan Weil         goto fail;
4549aebd98aSStefan Weil     } else if (header.sector_size != SECTOR_SIZE) {
455521b2b5dSMax Reitz         error_setg(errp, "unsupported VDI image (sector size %" PRIu32
456521b2b5dSMax Reitz                    " is not %u)", header.sector_size, SECTOR_SIZE);
4578937f822SStefan Weil         ret = -ENOTSUP;
4589aebd98aSStefan Weil         goto fail;
45963fa06dcSJeff Cody     } else if (header.block_size != DEFAULT_CLUSTER_SIZE) {
460521b2b5dSMax Reitz         error_setg(errp, "unsupported VDI image (block size %" PRIu32
461521b2b5dSMax Reitz                    " is not %u)", header.block_size, DEFAULT_CLUSTER_SIZE);
4628937f822SStefan Weil         ret = -ENOTSUP;
4639aebd98aSStefan Weil         goto fail;
464f21dc3a4SStefan Weil     } else if (header.disk_size >
465f21dc3a4SStefan Weil                (uint64_t)header.blocks_in_image * header.block_size) {
4665b7aa9b5SPaolo Bonzini         error_setg(errp, "unsupported VDI image (disk size %" PRIu64 ", "
4675b7aa9b5SPaolo Bonzini                    "image bitmap has room for %" PRIu64 ")",
4685b7aa9b5SPaolo Bonzini                    header.disk_size,
4695b7aa9b5SPaolo Bonzini                    (uint64_t)header.blocks_in_image * header.block_size);
4708937f822SStefan Weil         ret = -ENOTSUP;
4719aebd98aSStefan Weil         goto fail;
4729aebd98aSStefan Weil     } else if (!uuid_is_null(header.uuid_link)) {
4735b7aa9b5SPaolo Bonzini         error_setg(errp, "unsupported VDI image (non-NULL link UUID)");
4748937f822SStefan Weil         ret = -ENOTSUP;
4759aebd98aSStefan Weil         goto fail;
4769aebd98aSStefan Weil     } else if (!uuid_is_null(header.uuid_parent)) {
4775b7aa9b5SPaolo Bonzini         error_setg(errp, "unsupported VDI image (non-NULL parent UUID)");
4788937f822SStefan Weil         ret = -ENOTSUP;
4799aebd98aSStefan Weil         goto fail;
48063fa06dcSJeff Cody     } else if (header.blocks_in_image > VDI_BLOCKS_IN_IMAGE_MAX) {
48163fa06dcSJeff Cody         error_setg(errp, "unsupported VDI image "
48263fa06dcSJeff Cody                          "(too many blocks %u, max is %u)",
48363fa06dcSJeff Cody                           header.blocks_in_image, VDI_BLOCKS_IN_IMAGE_MAX);
48463fa06dcSJeff Cody         ret = -ENOTSUP;
48563fa06dcSJeff Cody         goto fail;
4869aebd98aSStefan Weil     }
4879aebd98aSStefan Weil 
4889aebd98aSStefan Weil     bs->total_sectors = header.disk_size / SECTOR_SIZE;
4899aebd98aSStefan Weil 
4909aebd98aSStefan Weil     s->block_size = header.block_size;
4919aebd98aSStefan Weil     s->block_sectors = header.block_size / SECTOR_SIZE;
4929aebd98aSStefan Weil     s->bmap_sector = header.offset_bmap / SECTOR_SIZE;
4939aebd98aSStefan Weil     s->header = header;
4949aebd98aSStefan Weil 
4959aebd98aSStefan Weil     bmap_size = header.blocks_in_image * sizeof(uint32_t);
496e9082e47SMax Reitz     bmap_size = DIV_ROUND_UP(bmap_size, SECTOR_SIZE);
4979a4f4c31SKevin Wolf     s->bmap = qemu_try_blockalign(bs->file->bs, bmap_size * SECTOR_SIZE);
49817cce735SKevin Wolf     if (s->bmap == NULL) {
49917cce735SKevin Wolf         ret = -ENOMEM;
50017cce735SKevin Wolf         goto fail;
50117cce735SKevin Wolf     }
50217cce735SKevin Wolf 
503*fbcbbf4eSKevin Wolf     ret = bdrv_read(bs->file, s->bmap_sector, (uint8_t *)s->bmap,
5049a4f4c31SKevin Wolf                     bmap_size);
5058937f822SStefan Weil     if (ret < 0) {
5069aebd98aSStefan Weil         goto fail_free_bmap;
5079aebd98aSStefan Weil     }
5089aebd98aSStefan Weil 
509fc9d106cSKevin Wolf     /* Disable migration when vdi images are used */
51081e5f78aSAlberto Garcia     error_setg(&s->migration_blocker, "The vdi format used by node '%s' "
51181e5f78aSAlberto Garcia                "does not support live migration",
51281e5f78aSAlberto Garcia                bdrv_get_device_or_node_name(bs));
513fc9d106cSKevin Wolf     migrate_add_blocker(s->migration_blocker);
514fc9d106cSKevin Wolf 
515f0ab6f10SMax Reitz     qemu_co_mutex_init(&s->write_lock);
516f0ab6f10SMax Reitz 
5179aebd98aSStefan Weil     return 0;
5189aebd98aSStefan Weil 
5199aebd98aSStefan Weil  fail_free_bmap:
52017cce735SKevin Wolf     qemu_vfree(s->bmap);
5219aebd98aSStefan Weil 
5229aebd98aSStefan Weil  fail:
5238937f822SStefan Weil     return ret;
5249aebd98aSStefan Weil }
5259aebd98aSStefan Weil 
526ecfe2bbaSJeff Cody static int vdi_reopen_prepare(BDRVReopenState *state,
527ecfe2bbaSJeff Cody                               BlockReopenQueue *queue, Error **errp)
528ecfe2bbaSJeff Cody {
529ecfe2bbaSJeff Cody     return 0;
530ecfe2bbaSJeff Cody }
531ecfe2bbaSJeff Cody 
532b6b8a333SPaolo Bonzini static int64_t coroutine_fn vdi_co_get_block_status(BlockDriverState *bs,
53367a0fd2aSFam Zheng         int64_t sector_num, int nb_sectors, int *pnum, BlockDriverState **file)
5349aebd98aSStefan Weil {
5359aebd98aSStefan Weil     /* TODO: Check for too large sector_num (in bdrv_is_allocated or here). */
5369aebd98aSStefan Weil     BDRVVdiState *s = (BDRVVdiState *)bs->opaque;
5379aebd98aSStefan Weil     size_t bmap_index = sector_num / s->block_sectors;
5389aebd98aSStefan Weil     size_t sector_in_block = sector_num % s->block_sectors;
5399aebd98aSStefan Weil     int n_sectors = s->block_sectors - sector_in_block;
5409aebd98aSStefan Weil     uint32_t bmap_entry = le32_to_cpu(s->bmap[bmap_index]);
5414bc74be9SPaolo Bonzini     uint64_t offset;
5424bc74be9SPaolo Bonzini     int result;
5434bc74be9SPaolo Bonzini 
5449aebd98aSStefan Weil     logout("%p, %" PRId64 ", %d, %p\n", bs, sector_num, nb_sectors, pnum);
5459aebd98aSStefan Weil     if (n_sectors > nb_sectors) {
5469aebd98aSStefan Weil         n_sectors = nb_sectors;
5479aebd98aSStefan Weil     }
5489aebd98aSStefan Weil     *pnum = n_sectors;
5494bc74be9SPaolo Bonzini     result = VDI_IS_ALLOCATED(bmap_entry);
5504bc74be9SPaolo Bonzini     if (!result) {
5514bc74be9SPaolo Bonzini         return 0;
5524bc74be9SPaolo Bonzini     }
5534bc74be9SPaolo Bonzini 
5544bc74be9SPaolo Bonzini     offset = s->header.offset_data +
5554bc74be9SPaolo Bonzini                               (uint64_t)bmap_entry * s->block_size +
5564bc74be9SPaolo Bonzini                               sector_in_block * SECTOR_SIZE;
5578bfb1371SFam Zheng     *file = bs->file->bs;
5584bc74be9SPaolo Bonzini     return BDRV_BLOCK_DATA | BDRV_BLOCK_OFFSET_VALID | offset;
5599aebd98aSStefan Weil }
5609aebd98aSStefan Weil 
5610865bb6fSKevin Wolf static int coroutine_fn
5620865bb6fSKevin Wolf vdi_co_preadv(BlockDriverState *bs, uint64_t offset, uint64_t bytes,
5630865bb6fSKevin Wolf               QEMUIOVector *qiov, int flags)
5649aebd98aSStefan Weil {
5659aebd98aSStefan Weil     BDRVVdiState *s = bs->opaque;
5660865bb6fSKevin Wolf     QEMUIOVector local_qiov;
5679aebd98aSStefan Weil     uint32_t bmap_entry;
5689aebd98aSStefan Weil     uint32_t block_index;
5690865bb6fSKevin Wolf     uint32_t offset_in_block;
5700865bb6fSKevin Wolf     uint32_t n_bytes;
5710865bb6fSKevin Wolf     uint64_t bytes_done = 0;
572eb9566d1SPaolo Bonzini     int ret = 0;
5734de659e8SPaolo Bonzini 
5744de659e8SPaolo Bonzini     logout("\n");
5759aebd98aSStefan Weil 
5760865bb6fSKevin Wolf     qemu_iovec_init(&local_qiov, qiov->niov);
5779aebd98aSStefan Weil 
5780865bb6fSKevin Wolf     while (ret >= 0 && bytes > 0) {
5790865bb6fSKevin Wolf         block_index = offset / s->block_size;
5800865bb6fSKevin Wolf         offset_in_block = offset % s->block_size;
5810865bb6fSKevin Wolf         n_bytes = MIN(bytes, s->block_size - offset_in_block);
5820865bb6fSKevin Wolf 
5830865bb6fSKevin Wolf         logout("will read %u bytes starting at offset %" PRIu64 "\n",
5840865bb6fSKevin Wolf                n_bytes, offset);
5859aebd98aSStefan Weil 
5869aebd98aSStefan Weil         /* prepare next AIO request */
5879aebd98aSStefan Weil         bmap_entry = le32_to_cpu(s->bmap[block_index]);
588c794b4e0SEric Sunshine         if (!VDI_IS_ALLOCATED(bmap_entry)) {
5899aebd98aSStefan Weil             /* Block not allocated, return zeros, no need to wait. */
5900865bb6fSKevin Wolf             qemu_iovec_memset(qiov, bytes_done, 0, n_bytes);
5913d46a75aSPaolo Bonzini             ret = 0;
5929aebd98aSStefan Weil         } else {
5930865bb6fSKevin Wolf             uint64_t data_offset = s->header.offset_data +
5940865bb6fSKevin Wolf                                    (uint64_t)bmap_entry * s->block_size +
5950865bb6fSKevin Wolf                                    offset_in_block;
5960c7bfc32SPaolo Bonzini 
5970865bb6fSKevin Wolf             qemu_iovec_reset(&local_qiov);
5980865bb6fSKevin Wolf             qemu_iovec_concat(&local_qiov, qiov, bytes_done, n_bytes);
5990865bb6fSKevin Wolf 
6000865bb6fSKevin Wolf             ret = bdrv_co_preadv(bs->file->bs, data_offset, n_bytes,
6010865bb6fSKevin Wolf                                  &local_qiov, 0);
6023d46a75aSPaolo Bonzini         }
6030865bb6fSKevin Wolf         logout("%u bytes read\n", n_bytes);
6040865bb6fSKevin Wolf 
6050865bb6fSKevin Wolf         bytes -= n_bytes;
6060865bb6fSKevin Wolf         offset += n_bytes;
6070865bb6fSKevin Wolf         bytes_done += n_bytes;
6080865bb6fSKevin Wolf     }
6090865bb6fSKevin Wolf 
6100865bb6fSKevin Wolf     qemu_iovec_destroy(&local_qiov);
6113d46a75aSPaolo Bonzini 
6123d46a75aSPaolo Bonzini     return ret;
6139aebd98aSStefan Weil }
6149aebd98aSStefan Weil 
615fde9d56fSKevin Wolf static int coroutine_fn
616fde9d56fSKevin Wolf vdi_co_pwritev(BlockDriverState *bs, uint64_t offset, uint64_t bytes,
617fde9d56fSKevin Wolf                QEMUIOVector *qiov, int flags)
6189aebd98aSStefan Weil {
6199aebd98aSStefan Weil     BDRVVdiState *s = bs->opaque;
620fde9d56fSKevin Wolf     QEMUIOVector local_qiov;
6219aebd98aSStefan Weil     uint32_t bmap_entry;
6229aebd98aSStefan Weil     uint32_t block_index;
623fde9d56fSKevin Wolf     uint32_t offset_in_block;
624fde9d56fSKevin Wolf     uint32_t n_bytes;
625bfc45fc1SPaolo Bonzini     uint32_t bmap_first = VDI_UNALLOCATED;
626bfc45fc1SPaolo Bonzini     uint32_t bmap_last = VDI_UNALLOCATED;
627bfc45fc1SPaolo Bonzini     uint8_t *block = NULL;
628fde9d56fSKevin Wolf     uint64_t bytes_done = 0;
629eb9566d1SPaolo Bonzini     int ret = 0;
6304de659e8SPaolo Bonzini 
6314de659e8SPaolo Bonzini     logout("\n");
6329aebd98aSStefan Weil 
633fde9d56fSKevin Wolf     qemu_iovec_init(&local_qiov, qiov->niov);
6349aebd98aSStefan Weil 
635fde9d56fSKevin Wolf     while (ret >= 0 && bytes > 0) {
636fde9d56fSKevin Wolf         block_index = offset / s->block_size;
637fde9d56fSKevin Wolf         offset_in_block = offset % s->block_size;
638fde9d56fSKevin Wolf         n_bytes = MIN(bytes, s->block_size - offset_in_block);
639fde9d56fSKevin Wolf 
640fde9d56fSKevin Wolf         logout("will write %u bytes starting at offset %" PRIu64 "\n",
641fde9d56fSKevin Wolf                n_bytes, offset);
6429aebd98aSStefan Weil 
6439aebd98aSStefan Weil         /* prepare next AIO request */
6449aebd98aSStefan Weil         bmap_entry = le32_to_cpu(s->bmap[block_index]);
645c794b4e0SEric Sunshine         if (!VDI_IS_ALLOCATED(bmap_entry)) {
6469aebd98aSStefan Weil             /* Allocate new block and write to it. */
647fde9d56fSKevin Wolf             uint64_t data_offset;
6489aebd98aSStefan Weil             bmap_entry = s->header.blocks_allocated;
6499aebd98aSStefan Weil             s->bmap[block_index] = cpu_to_le32(bmap_entry);
6509aebd98aSStefan Weil             s->header.blocks_allocated++;
651fde9d56fSKevin Wolf             data_offset = s->header.offset_data +
652fde9d56fSKevin Wolf                           (uint64_t)bmap_entry * s->block_size;
6539aebd98aSStefan Weil             if (block == NULL) {
654641543b7SStefan Weil                 block = g_malloc(s->block_size);
655bfc45fc1SPaolo Bonzini                 bmap_first = block_index;
6569aebd98aSStefan Weil             }
657bfc45fc1SPaolo Bonzini             bmap_last = block_index;
658641543b7SStefan Weil             /* Copy data to be written to new block and zero unused parts. */
659fde9d56fSKevin Wolf             memset(block, 0, offset_in_block);
660fde9d56fSKevin Wolf             qemu_iovec_to_buf(qiov, bytes_done, block + offset_in_block,
661fde9d56fSKevin Wolf                               n_bytes);
662fde9d56fSKevin Wolf             memset(block + offset_in_block + n_bytes, 0,
663fde9d56fSKevin Wolf                    s->block_size - n_bytes - offset_in_block);
664f0ab6f10SMax Reitz 
665f0ab6f10SMax Reitz             /* Note that this coroutine does not yield anywhere from reading the
666f0ab6f10SMax Reitz              * bmap entry until here, so in regards to all the coroutines trying
667f0ab6f10SMax Reitz              * to write to this cluster, the one doing the allocation will
668f0ab6f10SMax Reitz              * always be the first to try to acquire the lock.
669f0ab6f10SMax Reitz              * Therefore, it is also the first that will actually be able to
670f0ab6f10SMax Reitz              * acquire the lock and thus the padded cluster is written before
671f0ab6f10SMax Reitz              * the other coroutines can write to the affected area. */
672f0ab6f10SMax Reitz             qemu_co_mutex_lock(&s->write_lock);
673fde9d56fSKevin Wolf             ret = bdrv_pwrite(bs->file->bs, data_offset, block, s->block_size);
674f0ab6f10SMax Reitz             qemu_co_mutex_unlock(&s->write_lock);
6759aebd98aSStefan Weil         } else {
676fde9d56fSKevin Wolf             uint64_t data_offset = s->header.offset_data +
677fde9d56fSKevin Wolf                                    (uint64_t)bmap_entry * s->block_size +
678fde9d56fSKevin Wolf                                    offset_in_block;
679f0ab6f10SMax Reitz             qemu_co_mutex_lock(&s->write_lock);
680f0ab6f10SMax Reitz             /* This lock is only used to make sure the following write operation
681f0ab6f10SMax Reitz              * is executed after the write issued by the coroutine allocating
682f0ab6f10SMax Reitz              * this cluster, therefore we do not need to keep it locked.
683f0ab6f10SMax Reitz              * As stated above, the allocating coroutine will always try to lock
684f0ab6f10SMax Reitz              * the mutex before all the other concurrent accesses to that
685f0ab6f10SMax Reitz              * cluster, therefore at this point we can be absolutely certain
686f0ab6f10SMax Reitz              * that that write operation has returned (there may be other writes
687f0ab6f10SMax Reitz              * in flight, but they do not concern this very operation). */
688f0ab6f10SMax Reitz             qemu_co_mutex_unlock(&s->write_lock);
689fde9d56fSKevin Wolf 
690fde9d56fSKevin Wolf             qemu_iovec_reset(&local_qiov);
691fde9d56fSKevin Wolf             qemu_iovec_concat(&local_qiov, qiov, bytes_done, n_bytes);
692fde9d56fSKevin Wolf 
693fde9d56fSKevin Wolf             ret = bdrv_co_pwritev(bs->file->bs, data_offset, n_bytes,
694fde9d56fSKevin Wolf                                   &local_qiov, 0);
6959aebd98aSStefan Weil         }
6960c7bfc32SPaolo Bonzini 
697fde9d56fSKevin Wolf         bytes -= n_bytes;
698fde9d56fSKevin Wolf         offset += n_bytes;
699fde9d56fSKevin Wolf         bytes_done += n_bytes;
7000c7bfc32SPaolo Bonzini 
701fde9d56fSKevin Wolf         logout("%u bytes written\n", n_bytes);
7023d46a75aSPaolo Bonzini     }
7039aebd98aSStefan Weil 
704fde9d56fSKevin Wolf     qemu_iovec_destroy(&local_qiov);
705fde9d56fSKevin Wolf 
7060c7bfc32SPaolo Bonzini     logout("finished data write\n");
7074eea78e6SPaolo Bonzini     if (ret < 0) {
7084eea78e6SPaolo Bonzini         return ret;
7094eea78e6SPaolo Bonzini     }
7104eea78e6SPaolo Bonzini 
711bfc45fc1SPaolo Bonzini     if (block) {
7124eea78e6SPaolo Bonzini         /* One or more new blocks were allocated. */
713bfc45fc1SPaolo Bonzini         VdiHeader *header = (VdiHeader *) block;
7144eea78e6SPaolo Bonzini         uint8_t *base;
7154eea78e6SPaolo Bonzini         uint64_t offset;
716fde9d56fSKevin Wolf         uint32_t n_sectors;
7174eea78e6SPaolo Bonzini 
7180c7bfc32SPaolo Bonzini         logout("now writing modified header\n");
719bfc45fc1SPaolo Bonzini         assert(VDI_IS_ALLOCATED(bmap_first));
7200c7bfc32SPaolo Bonzini         *header = s->header;
7210c7bfc32SPaolo Bonzini         vdi_header_to_le(header);
7229a4f4c31SKevin Wolf         ret = bdrv_write(bs->file->bs, 0, block, 1);
723bfc45fc1SPaolo Bonzini         g_free(block);
724bfc45fc1SPaolo Bonzini         block = NULL;
7254eea78e6SPaolo Bonzini 
7264eea78e6SPaolo Bonzini         if (ret < 0) {
7274eea78e6SPaolo Bonzini             return ret;
7284eea78e6SPaolo Bonzini         }
7294eea78e6SPaolo Bonzini 
7300c7bfc32SPaolo Bonzini         logout("now writing modified block map entry %u...%u\n",
7310c7bfc32SPaolo Bonzini                bmap_first, bmap_last);
7320c7bfc32SPaolo Bonzini         /* Write modified sectors from block map. */
7330c7bfc32SPaolo Bonzini         bmap_first /= (SECTOR_SIZE / sizeof(uint32_t));
7340c7bfc32SPaolo Bonzini         bmap_last /= (SECTOR_SIZE / sizeof(uint32_t));
7350c7bfc32SPaolo Bonzini         n_sectors = bmap_last - bmap_first + 1;
7360c7bfc32SPaolo Bonzini         offset = s->bmap_sector + bmap_first;
7374eea78e6SPaolo Bonzini         base = ((uint8_t *)&s->bmap[0]) + bmap_first * SECTOR_SIZE;
7380c7bfc32SPaolo Bonzini         logout("will write %u block map sectors starting from entry %u\n",
7390c7bfc32SPaolo Bonzini                n_sectors, bmap_first);
7409a4f4c31SKevin Wolf         ret = bdrv_write(bs->file->bs, offset, base, n_sectors);
7410c7bfc32SPaolo Bonzini     }
7420c7bfc32SPaolo Bonzini 
7433d46a75aSPaolo Bonzini     return ret;
7449aebd98aSStefan Weil }
7459aebd98aSStefan Weil 
746004b7f25SChunyan Liu static int vdi_create(const char *filename, QemuOpts *opts, Error **errp)
7479aebd98aSStefan Weil {
748dddc7750SJeff Cody     int ret = 0;
7499aebd98aSStefan Weil     uint64_t bytes = 0;
7509aebd98aSStefan Weil     uint32_t blocks;
75199cce9faSKevin Wolf     size_t block_size = DEFAULT_CLUSTER_SIZE;
7529aebd98aSStefan Weil     uint32_t image_type = VDI_TYPE_DYNAMIC;
7539aebd98aSStefan Weil     VdiHeader header;
7549aebd98aSStefan Weil     size_t i;
7559aebd98aSStefan Weil     size_t bmap_size;
75670747862SJeff Cody     int64_t offset = 0;
75770747862SJeff Cody     Error *local_err = NULL;
758a08f0c3bSKevin Wolf     BlockBackend *blk = NULL;
75970747862SJeff Cody     uint32_t *bmap = NULL;
7609aebd98aSStefan Weil 
7619aebd98aSStefan Weil     logout("\n");
7629aebd98aSStefan Weil 
7639aebd98aSStefan Weil     /* Read out options. */
764c2eb918eSHu Tao     bytes = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
765c2eb918eSHu Tao                      BDRV_SECTOR_SIZE);
7669aebd98aSStefan Weil #if defined(CONFIG_VDI_BLOCK_SIZE)
7679aebd98aSStefan Weil     /* TODO: Additional checks (SECTOR_SIZE * 2^n, ...). */
768004b7f25SChunyan Liu     block_size = qemu_opt_get_size_del(opts,
769004b7f25SChunyan Liu                                        BLOCK_OPT_CLUSTER_SIZE,
770004b7f25SChunyan Liu                                        DEFAULT_CLUSTER_SIZE);
7719aebd98aSStefan Weil #endif
7729aebd98aSStefan Weil #if defined(CONFIG_VDI_STATIC_IMAGE)
773004b7f25SChunyan Liu     if (qemu_opt_get_bool_del(opts, BLOCK_OPT_STATIC, false)) {
7749aebd98aSStefan Weil         image_type = VDI_TYPE_STATIC;
7756eea90ebSStefan Weil     }
7769aebd98aSStefan Weil #endif
7779aebd98aSStefan Weil 
77863fa06dcSJeff Cody     if (bytes > VDI_DISK_SIZE_MAX) {
779dddc7750SJeff Cody         ret = -ENOTSUP;
78063fa06dcSJeff Cody         error_setg(errp, "Unsupported VDI image size (size is 0x%" PRIx64
78163fa06dcSJeff Cody                           ", max supported is 0x%" PRIx64 ")",
78263fa06dcSJeff Cody                           bytes, VDI_DISK_SIZE_MAX);
78363fa06dcSJeff Cody         goto exit;
78463fa06dcSJeff Cody     }
78563fa06dcSJeff Cody 
786dddc7750SJeff Cody     ret = bdrv_create_file(filename, opts, &local_err);
787dddc7750SJeff Cody     if (ret < 0) {
78870747862SJeff Cody         error_propagate(errp, local_err);
78963fa06dcSJeff Cody         goto exit;
7909aebd98aSStefan Weil     }
791a08f0c3bSKevin Wolf 
792efaa7c4eSMax Reitz     blk = blk_new_open(filename, NULL, NULL,
79372e775c7SKevin Wolf                        BDRV_O_RDWR | BDRV_O_PROTOCOL, &local_err);
794a08f0c3bSKevin Wolf     if (blk == NULL) {
79570747862SJeff Cody         error_propagate(errp, local_err);
796a08f0c3bSKevin Wolf         ret = -EIO;
79770747862SJeff Cody         goto exit;
7984ab15590SChunyan Liu     }
7994ab15590SChunyan Liu 
800a08f0c3bSKevin Wolf     blk_set_allow_write_beyond_eof(blk, true);
801a08f0c3bSKevin Wolf 
802f21dc3a4SStefan Weil     /* We need enough blocks to store the given disk size,
803f21dc3a4SStefan Weil        so always round up. */
804e9082e47SMax Reitz     blocks = DIV_ROUND_UP(bytes, block_size);
805f21dc3a4SStefan Weil 
8069aebd98aSStefan Weil     bmap_size = blocks * sizeof(uint32_t);
807e9082e47SMax Reitz     bmap_size = ROUND_UP(bmap_size, SECTOR_SIZE);
8089aebd98aSStefan Weil 
8099aebd98aSStefan Weil     memset(&header, 0, sizeof(header));
8101786dc15SBlue Swirl     pstrcpy(header.text, sizeof(header.text), VDI_TEXT);
8119aebd98aSStefan Weil     header.signature = VDI_SIGNATURE;
8129aebd98aSStefan Weil     header.version = VDI_VERSION_1_1;
8139aebd98aSStefan Weil     header.header_size = 0x180;
8149aebd98aSStefan Weil     header.image_type = image_type;
8159aebd98aSStefan Weil     header.offset_bmap = 0x200;
8169aebd98aSStefan Weil     header.offset_data = 0x200 + bmap_size;
8179aebd98aSStefan Weil     header.sector_size = SECTOR_SIZE;
8189aebd98aSStefan Weil     header.disk_size = bytes;
8199aebd98aSStefan Weil     header.block_size = block_size;
8209aebd98aSStefan Weil     header.blocks_in_image = blocks;
8216eea90ebSStefan Weil     if (image_type == VDI_TYPE_STATIC) {
8226eea90ebSStefan Weil         header.blocks_allocated = blocks;
8236eea90ebSStefan Weil     }
8249aebd98aSStefan Weil     uuid_generate(header.uuid_image);
8259aebd98aSStefan Weil     uuid_generate(header.uuid_last_snap);
8269aebd98aSStefan Weil     /* There is no need to set header.uuid_link or header.uuid_parent here. */
8279aebd98aSStefan Weil #if defined(CONFIG_VDI_DEBUG)
8289aebd98aSStefan Weil     vdi_header_print(&header);
8299aebd98aSStefan Weil #endif
8309aebd98aSStefan Weil     vdi_header_to_le(&header);
8318341f00dSEric Blake     ret = blk_pwrite(blk, offset, &header, sizeof(header), 0);
832dddc7750SJeff Cody     if (ret < 0) {
83370747862SJeff Cody         error_setg(errp, "Error writing header to %s", filename);
83470747862SJeff Cody         goto exit;
8359aebd98aSStefan Weil     }
83670747862SJeff Cody     offset += sizeof(header);
8379aebd98aSStefan Weil 
838b76b6e95SStefan Weil     if (bmap_size > 0) {
83917cce735SKevin Wolf         bmap = g_try_malloc0(bmap_size);
84017cce735SKevin Wolf         if (bmap == NULL) {
84117cce735SKevin Wolf             ret = -ENOMEM;
84217cce735SKevin Wolf             error_setg(errp, "Could not allocate bmap");
84317cce735SKevin Wolf             goto exit;
84417cce735SKevin Wolf         }
8459aebd98aSStefan Weil         for (i = 0; i < blocks; i++) {
8469aebd98aSStefan Weil             if (image_type == VDI_TYPE_STATIC) {
8479aebd98aSStefan Weil                 bmap[i] = i;
8489aebd98aSStefan Weil             } else {
8499aebd98aSStefan Weil                 bmap[i] = VDI_UNALLOCATED;
8509aebd98aSStefan Weil             }
8519aebd98aSStefan Weil         }
8528341f00dSEric Blake         ret = blk_pwrite(blk, offset, bmap, bmap_size, 0);
853dddc7750SJeff Cody         if (ret < 0) {
85470747862SJeff Cody             error_setg(errp, "Error writing bmap to %s", filename);
85570747862SJeff Cody             goto exit;
8569aebd98aSStefan Weil         }
85770747862SJeff Cody         offset += bmap_size;
858514f21a5SStefan Weil     }
859514f21a5SStefan Weil 
8609aebd98aSStefan Weil     if (image_type == VDI_TYPE_STATIC) {
861a08f0c3bSKevin Wolf         ret = blk_truncate(blk, offset + blocks * block_size);
862dddc7750SJeff Cody         if (ret < 0) {
86370747862SJeff Cody             error_setg(errp, "Failed to statically allocate %s", filename);
86470747862SJeff Cody             goto exit;
8659aebd98aSStefan Weil         }
8669aebd98aSStefan Weil     }
8679aebd98aSStefan Weil 
86863fa06dcSJeff Cody exit:
869a08f0c3bSKevin Wolf     blk_unref(blk);
87070747862SJeff Cody     g_free(bmap);
871dddc7750SJeff Cody     return ret;
8729aebd98aSStefan Weil }
8739aebd98aSStefan Weil 
8749aebd98aSStefan Weil static void vdi_close(BlockDriverState *bs)
8759aebd98aSStefan Weil {
876fc9d106cSKevin Wolf     BDRVVdiState *s = bs->opaque;
8776ac5f388SKevin Wolf 
87817cce735SKevin Wolf     qemu_vfree(s->bmap);
8796ac5f388SKevin Wolf 
880fc9d106cSKevin Wolf     migrate_del_blocker(s->migration_blocker);
881fc9d106cSKevin Wolf     error_free(s->migration_blocker);
8829aebd98aSStefan Weil }
8839aebd98aSStefan Weil 
884004b7f25SChunyan Liu static QemuOptsList vdi_create_opts = {
885004b7f25SChunyan Liu     .name = "vdi-create-opts",
886004b7f25SChunyan Liu     .head = QTAILQ_HEAD_INITIALIZER(vdi_create_opts.head),
887004b7f25SChunyan Liu     .desc = {
8889aebd98aSStefan Weil         {
8899aebd98aSStefan Weil             .name = BLOCK_OPT_SIZE,
890004b7f25SChunyan Liu             .type = QEMU_OPT_SIZE,
8919aebd98aSStefan Weil             .help = "Virtual disk size"
8929aebd98aSStefan Weil         },
8939aebd98aSStefan Weil #if defined(CONFIG_VDI_BLOCK_SIZE)
8949aebd98aSStefan Weil         {
8959aebd98aSStefan Weil             .name = BLOCK_OPT_CLUSTER_SIZE,
896004b7f25SChunyan Liu             .type = QEMU_OPT_SIZE,
89799cce9faSKevin Wolf             .help = "VDI cluster (block) size",
898004b7f25SChunyan Liu             .def_value_str = stringify(DEFAULT_CLUSTER_SIZE)
8999aebd98aSStefan Weil         },
9009aebd98aSStefan Weil #endif
9019aebd98aSStefan Weil #if defined(CONFIG_VDI_STATIC_IMAGE)
9029aebd98aSStefan Weil         {
9039aebd98aSStefan Weil             .name = BLOCK_OPT_STATIC,
904004b7f25SChunyan Liu             .type = QEMU_OPT_BOOL,
905004b7f25SChunyan Liu             .help = "VDI static (pre-allocated) image",
906004b7f25SChunyan Liu             .def_value_str = "off"
9079aebd98aSStefan Weil         },
9089aebd98aSStefan Weil #endif
9099aebd98aSStefan Weil         /* TODO: An additional option to set UUID values might be useful. */
910004b7f25SChunyan Liu         { /* end of list */ }
911004b7f25SChunyan Liu     }
9129aebd98aSStefan Weil };
9139aebd98aSStefan Weil 
9149aebd98aSStefan Weil static BlockDriver bdrv_vdi = {
9159aebd98aSStefan Weil     .format_name = "vdi",
9169aebd98aSStefan Weil     .instance_size = sizeof(BDRVVdiState),
9179aebd98aSStefan Weil     .bdrv_probe = vdi_probe,
9189aebd98aSStefan Weil     .bdrv_open = vdi_open,
9199aebd98aSStefan Weil     .bdrv_close = vdi_close,
920ecfe2bbaSJeff Cody     .bdrv_reopen_prepare = vdi_reopen_prepare,
921c282e1fdSChunyan Liu     .bdrv_create = vdi_create,
9223ac21627SPeter Lieven     .bdrv_has_zero_init = bdrv_has_zero_init_1,
923b6b8a333SPaolo Bonzini     .bdrv_co_get_block_status = vdi_co_get_block_status,
9249aebd98aSStefan Weil     .bdrv_make_empty = vdi_make_empty,
9259aebd98aSStefan Weil 
9260865bb6fSKevin Wolf     .bdrv_co_preadv     = vdi_co_preadv,
9279aebd98aSStefan Weil #if defined(CONFIG_VDI_WRITE)
928fde9d56fSKevin Wolf     .bdrv_co_pwritev    = vdi_co_pwritev,
9299aebd98aSStefan Weil #endif
9309aebd98aSStefan Weil 
9319aebd98aSStefan Weil     .bdrv_get_info = vdi_get_info,
9329aebd98aSStefan Weil 
933004b7f25SChunyan Liu     .create_opts = &vdi_create_opts,
9349aebd98aSStefan Weil     .bdrv_check = vdi_check,
9359aebd98aSStefan Weil };
9369aebd98aSStefan Weil 
9379aebd98aSStefan Weil static void bdrv_vdi_init(void)
9389aebd98aSStefan Weil {
9399aebd98aSStefan Weil     logout("\n");
9409aebd98aSStefan Weil     bdrv_register(&bdrv_vdi);
9419aebd98aSStefan Weil }
9429aebd98aSStefan Weil 
9439aebd98aSStefan Weil block_init(bdrv_vdi_init);
944