175411d23SStefan Hajnoczi /* 275411d23SStefan Hajnoczi * QEMU Enhanced Disk Format 375411d23SStefan Hajnoczi * 475411d23SStefan Hajnoczi * Copyright IBM, Corp. 2010 575411d23SStefan Hajnoczi * 675411d23SStefan Hajnoczi * Authors: 775411d23SStefan Hajnoczi * Stefan Hajnoczi <stefanha@linux.vnet.ibm.com> 875411d23SStefan Hajnoczi * Anthony Liguori <aliguori@us.ibm.com> 975411d23SStefan Hajnoczi * 1075411d23SStefan Hajnoczi * This work is licensed under the terms of the GNU LGPL, version 2 or later. 1175411d23SStefan Hajnoczi * See the COPYING.LIB file in the top-level directory. 1275411d23SStefan Hajnoczi * 1375411d23SStefan Hajnoczi */ 1475411d23SStefan Hajnoczi 1580c71a24SPeter Maydell #include "qemu/osdep.h" 16da34e65cSMarkus Armbruster #include "qapi/error.h" 171de7afc9SPaolo Bonzini #include "qemu/timer.h" 1858369e22SPaolo Bonzini #include "qemu/bswap.h" 19eabba580SStefan Hajnoczi #include "trace.h" 2075411d23SStefan Hajnoczi #include "qed.h" 217b1b5d19SPaolo Bonzini #include "qapi/qmp/qerror.h" 228a56fdadSKevin Wolf #include "sysemu/block-backend.h" 2375411d23SStefan Hajnoczi 2475411d23SStefan Hajnoczi static int bdrv_qed_probe(const uint8_t *buf, int buf_size, 2575411d23SStefan Hajnoczi const char *filename) 2675411d23SStefan Hajnoczi { 2775411d23SStefan Hajnoczi const QEDHeader *header = (const QEDHeader *)buf; 2875411d23SStefan Hajnoczi 2975411d23SStefan Hajnoczi if (buf_size < sizeof(*header)) { 3075411d23SStefan Hajnoczi return 0; 3175411d23SStefan Hajnoczi } 3275411d23SStefan Hajnoczi if (le32_to_cpu(header->magic) != QED_MAGIC) { 3375411d23SStefan Hajnoczi return 0; 3475411d23SStefan Hajnoczi } 3575411d23SStefan Hajnoczi return 100; 3675411d23SStefan Hajnoczi } 3775411d23SStefan Hajnoczi 3875411d23SStefan Hajnoczi /** 3975411d23SStefan Hajnoczi * Check whether an image format is raw 4075411d23SStefan Hajnoczi * 4175411d23SStefan Hajnoczi * @fmt: Backing file format, may be NULL 4275411d23SStefan Hajnoczi */ 4375411d23SStefan Hajnoczi static bool qed_fmt_is_raw(const char *fmt) 4475411d23SStefan Hajnoczi { 4575411d23SStefan Hajnoczi return fmt && strcmp(fmt, "raw") == 0; 4675411d23SStefan Hajnoczi } 4775411d23SStefan Hajnoczi 4875411d23SStefan Hajnoczi static void qed_header_le_to_cpu(const QEDHeader *le, QEDHeader *cpu) 4975411d23SStefan Hajnoczi { 5075411d23SStefan Hajnoczi cpu->magic = le32_to_cpu(le->magic); 5175411d23SStefan Hajnoczi cpu->cluster_size = le32_to_cpu(le->cluster_size); 5275411d23SStefan Hajnoczi cpu->table_size = le32_to_cpu(le->table_size); 5375411d23SStefan Hajnoczi cpu->header_size = le32_to_cpu(le->header_size); 5475411d23SStefan Hajnoczi cpu->features = le64_to_cpu(le->features); 5575411d23SStefan Hajnoczi cpu->compat_features = le64_to_cpu(le->compat_features); 5675411d23SStefan Hajnoczi cpu->autoclear_features = le64_to_cpu(le->autoclear_features); 5775411d23SStefan Hajnoczi cpu->l1_table_offset = le64_to_cpu(le->l1_table_offset); 5875411d23SStefan Hajnoczi cpu->image_size = le64_to_cpu(le->image_size); 5975411d23SStefan Hajnoczi cpu->backing_filename_offset = le32_to_cpu(le->backing_filename_offset); 6075411d23SStefan Hajnoczi cpu->backing_filename_size = le32_to_cpu(le->backing_filename_size); 6175411d23SStefan Hajnoczi } 6275411d23SStefan Hajnoczi 6375411d23SStefan Hajnoczi static void qed_header_cpu_to_le(const QEDHeader *cpu, QEDHeader *le) 6475411d23SStefan Hajnoczi { 6575411d23SStefan Hajnoczi le->magic = cpu_to_le32(cpu->magic); 6675411d23SStefan Hajnoczi le->cluster_size = cpu_to_le32(cpu->cluster_size); 6775411d23SStefan Hajnoczi le->table_size = cpu_to_le32(cpu->table_size); 6875411d23SStefan Hajnoczi le->header_size = cpu_to_le32(cpu->header_size); 6975411d23SStefan Hajnoczi le->features = cpu_to_le64(cpu->features); 7075411d23SStefan Hajnoczi le->compat_features = cpu_to_le64(cpu->compat_features); 7175411d23SStefan Hajnoczi le->autoclear_features = cpu_to_le64(cpu->autoclear_features); 7275411d23SStefan Hajnoczi le->l1_table_offset = cpu_to_le64(cpu->l1_table_offset); 7375411d23SStefan Hajnoczi le->image_size = cpu_to_le64(cpu->image_size); 7475411d23SStefan Hajnoczi le->backing_filename_offset = cpu_to_le32(cpu->backing_filename_offset); 7575411d23SStefan Hajnoczi le->backing_filename_size = cpu_to_le32(cpu->backing_filename_size); 7675411d23SStefan Hajnoczi } 7775411d23SStefan Hajnoczi 78b10170acSStefan Hajnoczi int qed_write_header_sync(BDRVQEDState *s) 7975411d23SStefan Hajnoczi { 8075411d23SStefan Hajnoczi QEDHeader le; 8175411d23SStefan Hajnoczi int ret; 8275411d23SStefan Hajnoczi 8375411d23SStefan Hajnoczi qed_header_cpu_to_le(&s->header, &le); 84d9ca2ea2SKevin Wolf ret = bdrv_pwrite(s->bs->file, 0, &le, sizeof(le)); 8575411d23SStefan Hajnoczi if (ret != sizeof(le)) { 8675411d23SStefan Hajnoczi return ret; 8775411d23SStefan Hajnoczi } 8875411d23SStefan Hajnoczi return 0; 8975411d23SStefan Hajnoczi } 9075411d23SStefan Hajnoczi 9101979a98SStefan Hajnoczi /** 9201979a98SStefan Hajnoczi * Update header in-place (does not rewrite backing filename or other strings) 9301979a98SStefan Hajnoczi * 9401979a98SStefan Hajnoczi * This function only updates known header fields in-place and does not affect 9501979a98SStefan Hajnoczi * extra data after the QED header. 9601979a98SStefan Hajnoczi */ 9787f0d882SKevin Wolf static int coroutine_fn qed_write_header(BDRVQEDState *s) 9801979a98SStefan Hajnoczi { 9901979a98SStefan Hajnoczi /* We must write full sectors for O_DIRECT but cannot necessarily generate 10001979a98SStefan Hajnoczi * the data following the header if an unrecognized compat feature is 10101979a98SStefan Hajnoczi * active. Therefore, first read the sectors containing the header, update 10201979a98SStefan Hajnoczi * them, and write back. 10301979a98SStefan Hajnoczi */ 10401979a98SStefan Hajnoczi 105c41a73ffSLaurent Vivier int nsectors = DIV_ROUND_UP(sizeof(QEDHeader), BDRV_SECTOR_SIZE); 10601979a98SStefan Hajnoczi size_t len = nsectors * BDRV_SECTOR_SIZE; 1077076309aSKevin Wolf uint8_t *buf; 1087076309aSKevin Wolf struct iovec iov; 1097076309aSKevin Wolf QEMUIOVector qiov; 1107076309aSKevin Wolf int ret; 11101979a98SStefan Hajnoczi 1127076309aSKevin Wolf buf = qemu_blockalign(s->bs, len); 1137076309aSKevin Wolf iov = (struct iovec) { 1147076309aSKevin Wolf .iov_base = buf, 1157076309aSKevin Wolf .iov_len = len, 1167076309aSKevin Wolf }; 1177076309aSKevin Wolf qemu_iovec_init_external(&qiov, &iov, 1); 11801979a98SStefan Hajnoczi 1190f714ec7SKevin Wolf ret = bdrv_co_preadv(s->bs->file, 0, qiov.size, &qiov, 0); 1207076309aSKevin Wolf if (ret < 0) { 1217076309aSKevin Wolf goto out; 1227076309aSKevin Wolf } 1237076309aSKevin Wolf 1247076309aSKevin Wolf /* Update header */ 1257076309aSKevin Wolf qed_header_cpu_to_le(&s->header, (QEDHeader *) buf); 1267076309aSKevin Wolf 1270f714ec7SKevin Wolf ret = bdrv_co_pwritev(s->bs->file, 0, qiov.size, &qiov, 0); 1287076309aSKevin Wolf if (ret < 0) { 1297076309aSKevin Wolf goto out; 1307076309aSKevin Wolf } 1317076309aSKevin Wolf 1327076309aSKevin Wolf ret = 0; 1337076309aSKevin Wolf out: 1347076309aSKevin Wolf qemu_vfree(buf); 135f13d712bSKevin Wolf return ret; 13601979a98SStefan Hajnoczi } 13701979a98SStefan Hajnoczi 13875411d23SStefan Hajnoczi static uint64_t qed_max_image_size(uint32_t cluster_size, uint32_t table_size) 13975411d23SStefan Hajnoczi { 14075411d23SStefan Hajnoczi uint64_t table_entries; 14175411d23SStefan Hajnoczi uint64_t l2_size; 14275411d23SStefan Hajnoczi 14375411d23SStefan Hajnoczi table_entries = (table_size * cluster_size) / sizeof(uint64_t); 14475411d23SStefan Hajnoczi l2_size = table_entries * cluster_size; 14575411d23SStefan Hajnoczi 14675411d23SStefan Hajnoczi return l2_size * table_entries; 14775411d23SStefan Hajnoczi } 14875411d23SStefan Hajnoczi 14975411d23SStefan Hajnoczi static bool qed_is_cluster_size_valid(uint32_t cluster_size) 15075411d23SStefan Hajnoczi { 15175411d23SStefan Hajnoczi if (cluster_size < QED_MIN_CLUSTER_SIZE || 15275411d23SStefan Hajnoczi cluster_size > QED_MAX_CLUSTER_SIZE) { 15375411d23SStefan Hajnoczi return false; 15475411d23SStefan Hajnoczi } 15575411d23SStefan Hajnoczi if (cluster_size & (cluster_size - 1)) { 15675411d23SStefan Hajnoczi return false; /* not power of 2 */ 15775411d23SStefan Hajnoczi } 15875411d23SStefan Hajnoczi return true; 15975411d23SStefan Hajnoczi } 16075411d23SStefan Hajnoczi 16175411d23SStefan Hajnoczi static bool qed_is_table_size_valid(uint32_t table_size) 16275411d23SStefan Hajnoczi { 16375411d23SStefan Hajnoczi if (table_size < QED_MIN_TABLE_SIZE || 16475411d23SStefan Hajnoczi table_size > QED_MAX_TABLE_SIZE) { 16575411d23SStefan Hajnoczi return false; 16675411d23SStefan Hajnoczi } 16775411d23SStefan Hajnoczi if (table_size & (table_size - 1)) { 16875411d23SStefan Hajnoczi return false; /* not power of 2 */ 16975411d23SStefan Hajnoczi } 17075411d23SStefan Hajnoczi return true; 17175411d23SStefan Hajnoczi } 17275411d23SStefan Hajnoczi 17375411d23SStefan Hajnoczi static bool qed_is_image_size_valid(uint64_t image_size, uint32_t cluster_size, 17475411d23SStefan Hajnoczi uint32_t table_size) 17575411d23SStefan Hajnoczi { 17675411d23SStefan Hajnoczi if (image_size % BDRV_SECTOR_SIZE != 0) { 17775411d23SStefan Hajnoczi return false; /* not multiple of sector size */ 17875411d23SStefan Hajnoczi } 17975411d23SStefan Hajnoczi if (image_size > qed_max_image_size(cluster_size, table_size)) { 18075411d23SStefan Hajnoczi return false; /* image is too large */ 18175411d23SStefan Hajnoczi } 18275411d23SStefan Hajnoczi return true; 18375411d23SStefan Hajnoczi } 18475411d23SStefan Hajnoczi 18575411d23SStefan Hajnoczi /** 18675411d23SStefan Hajnoczi * Read a string of known length from the image file 18775411d23SStefan Hajnoczi * 18875411d23SStefan Hajnoczi * @file: Image file 18975411d23SStefan Hajnoczi * @offset: File offset to start of string, in bytes 19075411d23SStefan Hajnoczi * @n: String length in bytes 19175411d23SStefan Hajnoczi * @buf: Destination buffer 19275411d23SStefan Hajnoczi * @buflen: Destination buffer length in bytes 19375411d23SStefan Hajnoczi * @ret: 0 on success, -errno on failure 19475411d23SStefan Hajnoczi * 19575411d23SStefan Hajnoczi * The string is NUL-terminated. 19675411d23SStefan Hajnoczi */ 197cf2ab8fcSKevin Wolf static int qed_read_string(BdrvChild *file, uint64_t offset, size_t n, 19875411d23SStefan Hajnoczi char *buf, size_t buflen) 19975411d23SStefan Hajnoczi { 20075411d23SStefan Hajnoczi int ret; 20175411d23SStefan Hajnoczi if (n >= buflen) { 20275411d23SStefan Hajnoczi return -EINVAL; 20375411d23SStefan Hajnoczi } 20475411d23SStefan Hajnoczi ret = bdrv_pread(file, offset, buf, n); 20575411d23SStefan Hajnoczi if (ret < 0) { 20675411d23SStefan Hajnoczi return ret; 20775411d23SStefan Hajnoczi } 20875411d23SStefan Hajnoczi buf[n] = '\0'; 20975411d23SStefan Hajnoczi return 0; 21075411d23SStefan Hajnoczi } 21175411d23SStefan Hajnoczi 212eabba580SStefan Hajnoczi /** 213eabba580SStefan Hajnoczi * Allocate new clusters 214eabba580SStefan Hajnoczi * 215eabba580SStefan Hajnoczi * @s: QED state 216eabba580SStefan Hajnoczi * @n: Number of contiguous clusters to allocate 217eabba580SStefan Hajnoczi * @ret: Offset of first allocated cluster 218eabba580SStefan Hajnoczi * 219eabba580SStefan Hajnoczi * This function only produces the offset where the new clusters should be 220eabba580SStefan Hajnoczi * written. It updates BDRVQEDState but does not make any changes to the image 221eabba580SStefan Hajnoczi * file. 222eabba580SStefan Hajnoczi */ 223eabba580SStefan Hajnoczi static uint64_t qed_alloc_clusters(BDRVQEDState *s, unsigned int n) 224eabba580SStefan Hajnoczi { 225eabba580SStefan Hajnoczi uint64_t offset = s->file_size; 226eabba580SStefan Hajnoczi s->file_size += n * s->header.cluster_size; 227eabba580SStefan Hajnoczi return offset; 228eabba580SStefan Hajnoczi } 229eabba580SStefan Hajnoczi 230298800caSStefan Hajnoczi QEDTable *qed_alloc_table(BDRVQEDState *s) 231298800caSStefan Hajnoczi { 232298800caSStefan Hajnoczi /* Honor O_DIRECT memory alignment requirements */ 233298800caSStefan Hajnoczi return qemu_blockalign(s->bs, 234298800caSStefan Hajnoczi s->header.cluster_size * s->header.table_size); 235298800caSStefan Hajnoczi } 236298800caSStefan Hajnoczi 237eabba580SStefan Hajnoczi /** 238eabba580SStefan Hajnoczi * Allocate a new zeroed L2 table 239eabba580SStefan Hajnoczi */ 240eabba580SStefan Hajnoczi static CachedL2Table *qed_new_l2_table(BDRVQEDState *s) 241eabba580SStefan Hajnoczi { 242eabba580SStefan Hajnoczi CachedL2Table *l2_table = qed_alloc_l2_cache_entry(&s->l2_cache); 243eabba580SStefan Hajnoczi 244eabba580SStefan Hajnoczi l2_table->table = qed_alloc_table(s); 245eabba580SStefan Hajnoczi l2_table->offset = qed_alloc_clusters(s, s->header.table_size); 246eabba580SStefan Hajnoczi 247eabba580SStefan Hajnoczi memset(l2_table->table->offsets, 0, 248eabba580SStefan Hajnoczi s->header.cluster_size * s->header.table_size); 249eabba580SStefan Hajnoczi return l2_table; 250eabba580SStefan Hajnoczi } 251eabba580SStefan Hajnoczi 2526f321e93SStefan Hajnoczi static void qed_plug_allocating_write_reqs(BDRVQEDState *s) 2536f321e93SStefan Hajnoczi { 2546f321e93SStefan Hajnoczi assert(!s->allocating_write_reqs_plugged); 2556f321e93SStefan Hajnoczi 2566f321e93SStefan Hajnoczi s->allocating_write_reqs_plugged = true; 2576f321e93SStefan Hajnoczi } 2586f321e93SStefan Hajnoczi 2596f321e93SStefan Hajnoczi static void qed_unplug_allocating_write_reqs(BDRVQEDState *s) 2606f321e93SStefan Hajnoczi { 2616f321e93SStefan Hajnoczi assert(s->allocating_write_reqs_plugged); 2626f321e93SStefan Hajnoczi 2636f321e93SStefan Hajnoczi s->allocating_write_reqs_plugged = false; 2640806c3b5SKevin Wolf qemu_co_enter_next(&s->allocating_write_reqs); 2656f321e93SStefan Hajnoczi } 2666f321e93SStefan Hajnoczi 26787f0d882SKevin Wolf static void coroutine_fn qed_need_check_timer_entry(void *opaque) 2686f321e93SStefan Hajnoczi { 2696f321e93SStefan Hajnoczi BDRVQEDState *s = opaque; 270c0e8f989SKevin Wolf int ret; 2716f321e93SStefan Hajnoczi 2726f321e93SStefan Hajnoczi /* The timer should only fire when allocating writes have drained */ 2730806c3b5SKevin Wolf assert(!s->allocating_acb); 2746f321e93SStefan Hajnoczi 2756f321e93SStefan Hajnoczi trace_qed_need_check_timer_cb(s); 2766f321e93SStefan Hajnoczi 2772f47da5fSPaolo Bonzini qed_acquire(s); 2786f321e93SStefan Hajnoczi qed_plug_allocating_write_reqs(s); 2796f321e93SStefan Hajnoczi 2806f321e93SStefan Hajnoczi /* Ensure writes are on disk before clearing flag */ 281c0e8f989SKevin Wolf ret = bdrv_co_flush(s->bs->file->bs); 2822f47da5fSPaolo Bonzini qed_release(s); 283c0e8f989SKevin Wolf if (ret < 0) { 284c0e8f989SKevin Wolf qed_unplug_allocating_write_reqs(s); 285c0e8f989SKevin Wolf return; 286c0e8f989SKevin Wolf } 287c0e8f989SKevin Wolf 288c0e8f989SKevin Wolf s->header.features &= ~QED_F_NEED_CHECK; 289c0e8f989SKevin Wolf ret = qed_write_header(s); 290c0e8f989SKevin Wolf (void) ret; 291c0e8f989SKevin Wolf 292c0e8f989SKevin Wolf qed_unplug_allocating_write_reqs(s); 293c0e8f989SKevin Wolf 294c0e8f989SKevin Wolf ret = bdrv_co_flush(s->bs); 295c0e8f989SKevin Wolf (void) ret; 296c0e8f989SKevin Wolf } 297c0e8f989SKevin Wolf 298c0e8f989SKevin Wolf static void qed_need_check_timer_cb(void *opaque) 299c0e8f989SKevin Wolf { 300c0e8f989SKevin Wolf Coroutine *co = qemu_coroutine_create(qed_need_check_timer_entry, opaque); 301c0e8f989SKevin Wolf qemu_coroutine_enter(co); 3022f47da5fSPaolo Bonzini } 3032f47da5fSPaolo Bonzini 3042f47da5fSPaolo Bonzini void qed_acquire(BDRVQEDState *s) 3052f47da5fSPaolo Bonzini { 3062f47da5fSPaolo Bonzini aio_context_acquire(bdrv_get_aio_context(s->bs)); 3072f47da5fSPaolo Bonzini } 3082f47da5fSPaolo Bonzini 3092f47da5fSPaolo Bonzini void qed_release(BDRVQEDState *s) 3102f47da5fSPaolo Bonzini { 3112f47da5fSPaolo Bonzini aio_context_release(bdrv_get_aio_context(s->bs)); 3126f321e93SStefan Hajnoczi } 3136f321e93SStefan Hajnoczi 3146f321e93SStefan Hajnoczi static void qed_start_need_check_timer(BDRVQEDState *s) 3156f321e93SStefan Hajnoczi { 3166f321e93SStefan Hajnoczi trace_qed_start_need_check_timer(s); 3176f321e93SStefan Hajnoczi 318bc72ad67SAlex Bligh /* Use QEMU_CLOCK_VIRTUAL so we don't alter the image file while suspended for 3196f321e93SStefan Hajnoczi * migration. 3206f321e93SStefan Hajnoczi */ 321bc72ad67SAlex Bligh timer_mod(s->need_check_timer, qemu_clock_get_ns(QEMU_CLOCK_VIRTUAL) + 32273bcb24dSRutuja Shah NANOSECONDS_PER_SECOND * QED_NEED_CHECK_TIMEOUT); 3236f321e93SStefan Hajnoczi } 3246f321e93SStefan Hajnoczi 3256f321e93SStefan Hajnoczi /* It's okay to call this multiple times or when no timer is started */ 3266f321e93SStefan Hajnoczi static void qed_cancel_need_check_timer(BDRVQEDState *s) 3276f321e93SStefan Hajnoczi { 3286f321e93SStefan Hajnoczi trace_qed_cancel_need_check_timer(s); 329bc72ad67SAlex Bligh timer_del(s->need_check_timer); 3306f321e93SStefan Hajnoczi } 3316f321e93SStefan Hajnoczi 332a8c868c3SStefan Hajnoczi static void bdrv_qed_detach_aio_context(BlockDriverState *bs) 333a8c868c3SStefan Hajnoczi { 334a8c868c3SStefan Hajnoczi BDRVQEDState *s = bs->opaque; 335a8c868c3SStefan Hajnoczi 336a8c868c3SStefan Hajnoczi qed_cancel_need_check_timer(s); 337a8c868c3SStefan Hajnoczi timer_free(s->need_check_timer); 338a8c868c3SStefan Hajnoczi } 339a8c868c3SStefan Hajnoczi 340a8c868c3SStefan Hajnoczi static void bdrv_qed_attach_aio_context(BlockDriverState *bs, 341a8c868c3SStefan Hajnoczi AioContext *new_context) 342a8c868c3SStefan Hajnoczi { 343a8c868c3SStefan Hajnoczi BDRVQEDState *s = bs->opaque; 344a8c868c3SStefan Hajnoczi 345a8c868c3SStefan Hajnoczi s->need_check_timer = aio_timer_new(new_context, 346a8c868c3SStefan Hajnoczi QEMU_CLOCK_VIRTUAL, SCALE_NS, 347a8c868c3SStefan Hajnoczi qed_need_check_timer_cb, s); 348a8c868c3SStefan Hajnoczi if (s->header.features & QED_F_NEED_CHECK) { 349a8c868c3SStefan Hajnoczi qed_start_need_check_timer(s); 350a8c868c3SStefan Hajnoczi } 351a8c868c3SStefan Hajnoczi } 352a8c868c3SStefan Hajnoczi 35361124f03SPaolo Bonzini static void coroutine_fn bdrv_qed_co_drain(BlockDriverState *bs) 3546653a73dSFam Zheng { 3556653a73dSFam Zheng BDRVQEDState *s = bs->opaque; 3566653a73dSFam Zheng 3576653a73dSFam Zheng /* Fire the timer immediately in order to start doing I/O as soon as the 3586653a73dSFam Zheng * header is flushed. 3596653a73dSFam Zheng */ 3606653a73dSFam Zheng if (s->need_check_timer && timer_pending(s->need_check_timer)) { 3616653a73dSFam Zheng qed_cancel_need_check_timer(s); 36261124f03SPaolo Bonzini qed_need_check_timer_entry(s); 3636653a73dSFam Zheng } 3646653a73dSFam Zheng } 3656653a73dSFam Zheng 366*61c7887eSPaolo Bonzini static void bdrv_qed_init_state(BlockDriverState *bs) 367*61c7887eSPaolo Bonzini { 368*61c7887eSPaolo Bonzini BDRVQEDState *s = bs->opaque; 369*61c7887eSPaolo Bonzini 370*61c7887eSPaolo Bonzini memset(s, 0, sizeof(BDRVQEDState)); 371*61c7887eSPaolo Bonzini s->bs = bs; 372*61c7887eSPaolo Bonzini qemu_co_queue_init(&s->allocating_write_reqs); 373*61c7887eSPaolo Bonzini } 374*61c7887eSPaolo Bonzini 3754e4bf5c4SKevin Wolf static int bdrv_qed_do_open(BlockDriverState *bs, QDict *options, int flags, 376015a1036SMax Reitz Error **errp) 37775411d23SStefan Hajnoczi { 37875411d23SStefan Hajnoczi BDRVQEDState *s = bs->opaque; 37975411d23SStefan Hajnoczi QEDHeader le_header; 38075411d23SStefan Hajnoczi int64_t file_size; 38175411d23SStefan Hajnoczi int ret; 38275411d23SStefan Hajnoczi 383cf2ab8fcSKevin Wolf ret = bdrv_pread(bs->file, 0, &le_header, sizeof(le_header)); 38475411d23SStefan Hajnoczi if (ret < 0) { 38575411d23SStefan Hajnoczi return ret; 38675411d23SStefan Hajnoczi } 38775411d23SStefan Hajnoczi qed_header_le_to_cpu(&le_header, &s->header); 38875411d23SStefan Hajnoczi 38975411d23SStefan Hajnoczi if (s->header.magic != QED_MAGIC) { 39076abe407SPaolo Bonzini error_setg(errp, "Image not in QED format"); 39176abe407SPaolo Bonzini return -EINVAL; 39275411d23SStefan Hajnoczi } 39375411d23SStefan Hajnoczi if (s->header.features & ~QED_FEATURE_MASK) { 39410b758e8SKevin Wolf /* image uses unsupported feature bits */ 395a55448b3SMax Reitz error_setg(errp, "Unsupported QED features: %" PRIx64, 39610b758e8SKevin Wolf s->header.features & ~QED_FEATURE_MASK); 39710b758e8SKevin Wolf return -ENOTSUP; 39875411d23SStefan Hajnoczi } 39975411d23SStefan Hajnoczi if (!qed_is_cluster_size_valid(s->header.cluster_size)) { 40075411d23SStefan Hajnoczi return -EINVAL; 40175411d23SStefan Hajnoczi } 40275411d23SStefan Hajnoczi 40375411d23SStefan Hajnoczi /* Round down file size to the last cluster */ 4049a4f4c31SKevin Wolf file_size = bdrv_getlength(bs->file->bs); 40575411d23SStefan Hajnoczi if (file_size < 0) { 40675411d23SStefan Hajnoczi return file_size; 40775411d23SStefan Hajnoczi } 40875411d23SStefan Hajnoczi s->file_size = qed_start_of_cluster(s, file_size); 40975411d23SStefan Hajnoczi 41075411d23SStefan Hajnoczi if (!qed_is_table_size_valid(s->header.table_size)) { 41175411d23SStefan Hajnoczi return -EINVAL; 41275411d23SStefan Hajnoczi } 41375411d23SStefan Hajnoczi if (!qed_is_image_size_valid(s->header.image_size, 41475411d23SStefan Hajnoczi s->header.cluster_size, 41575411d23SStefan Hajnoczi s->header.table_size)) { 41675411d23SStefan Hajnoczi return -EINVAL; 41775411d23SStefan Hajnoczi } 41875411d23SStefan Hajnoczi if (!qed_check_table_offset(s, s->header.l1_table_offset)) { 41975411d23SStefan Hajnoczi return -EINVAL; 42075411d23SStefan Hajnoczi } 42175411d23SStefan Hajnoczi 42275411d23SStefan Hajnoczi s->table_nelems = (s->header.cluster_size * s->header.table_size) / 42375411d23SStefan Hajnoczi sizeof(uint64_t); 424786a4ea8SStefan Hajnoczi s->l2_shift = ctz32(s->header.cluster_size); 42575411d23SStefan Hajnoczi s->l2_mask = s->table_nelems - 1; 426786a4ea8SStefan Hajnoczi s->l1_shift = s->l2_shift + ctz32(s->table_nelems); 42775411d23SStefan Hajnoczi 4280adfa1edSStefan Hajnoczi /* Header size calculation must not overflow uint32_t */ 4290adfa1edSStefan Hajnoczi if (s->header.header_size > UINT32_MAX / s->header.cluster_size) { 4300adfa1edSStefan Hajnoczi return -EINVAL; 4310adfa1edSStefan Hajnoczi } 4320adfa1edSStefan Hajnoczi 43375411d23SStefan Hajnoczi if ((s->header.features & QED_F_BACKING_FILE)) { 43475411d23SStefan Hajnoczi if ((uint64_t)s->header.backing_filename_offset + 43575411d23SStefan Hajnoczi s->header.backing_filename_size > 43675411d23SStefan Hajnoczi s->header.cluster_size * s->header.header_size) { 43775411d23SStefan Hajnoczi return -EINVAL; 43875411d23SStefan Hajnoczi } 43975411d23SStefan Hajnoczi 440cf2ab8fcSKevin Wolf ret = qed_read_string(bs->file, s->header.backing_filename_offset, 44175411d23SStefan Hajnoczi s->header.backing_filename_size, bs->backing_file, 44275411d23SStefan Hajnoczi sizeof(bs->backing_file)); 44375411d23SStefan Hajnoczi if (ret < 0) { 44475411d23SStefan Hajnoczi return ret; 44575411d23SStefan Hajnoczi } 44675411d23SStefan Hajnoczi 44775411d23SStefan Hajnoczi if (s->header.features & QED_F_BACKING_FORMAT_NO_PROBE) { 44875411d23SStefan Hajnoczi pstrcpy(bs->backing_format, sizeof(bs->backing_format), "raw"); 44975411d23SStefan Hajnoczi } 45075411d23SStefan Hajnoczi } 45175411d23SStefan Hajnoczi 45275411d23SStefan Hajnoczi /* Reset unknown autoclear feature bits. This is a backwards 45375411d23SStefan Hajnoczi * compatibility mechanism that allows images to be opened by older 45475411d23SStefan Hajnoczi * programs, which "knock out" unknown feature bits. When an image is 45575411d23SStefan Hajnoczi * opened by a newer program again it can detect that the autoclear 45675411d23SStefan Hajnoczi * feature is no longer valid. 45775411d23SStefan Hajnoczi */ 45875411d23SStefan Hajnoczi if ((s->header.autoclear_features & ~QED_AUTOCLEAR_FEATURE_MASK) != 0 && 45904c01a5cSKevin Wolf !bdrv_is_read_only(bs->file->bs) && !(flags & BDRV_O_INACTIVE)) { 46075411d23SStefan Hajnoczi s->header.autoclear_features &= QED_AUTOCLEAR_FEATURE_MASK; 46175411d23SStefan Hajnoczi 46275411d23SStefan Hajnoczi ret = qed_write_header_sync(s); 46375411d23SStefan Hajnoczi if (ret) { 46475411d23SStefan Hajnoczi return ret; 46575411d23SStefan Hajnoczi } 46675411d23SStefan Hajnoczi 46775411d23SStefan Hajnoczi /* From here on only known autoclear feature bits are valid */ 4689a4f4c31SKevin Wolf bdrv_flush(bs->file->bs); 46975411d23SStefan Hajnoczi } 47075411d23SStefan Hajnoczi 471298800caSStefan Hajnoczi s->l1_table = qed_alloc_table(s); 472298800caSStefan Hajnoczi qed_init_l2_cache(&s->l2_cache); 473298800caSStefan Hajnoczi 474298800caSStefan Hajnoczi ret = qed_read_l1_table_sync(s); 475298800caSStefan Hajnoczi if (ret) { 47601979a98SStefan Hajnoczi goto out; 47701979a98SStefan Hajnoczi } 47801979a98SStefan Hajnoczi 47901979a98SStefan Hajnoczi /* If image was not closed cleanly, check consistency */ 480058f8f16SStefan Hajnoczi if (!(flags & BDRV_O_CHECK) && (s->header.features & QED_F_NEED_CHECK)) { 48101979a98SStefan Hajnoczi /* Read-only images cannot be fixed. There is no risk of corruption 48201979a98SStefan Hajnoczi * since write operations are not possible. Therefore, allow 48301979a98SStefan Hajnoczi * potentially inconsistent images to be opened read-only. This can 48401979a98SStefan Hajnoczi * aid data recovery from an otherwise inconsistent image. 48501979a98SStefan Hajnoczi */ 4869a4f4c31SKevin Wolf if (!bdrv_is_read_only(bs->file->bs) && 48704c01a5cSKevin Wolf !(flags & BDRV_O_INACTIVE)) { 48801979a98SStefan Hajnoczi BdrvCheckResult result = {0}; 48901979a98SStefan Hajnoczi 49001979a98SStefan Hajnoczi ret = qed_check(s, &result, true); 4916f321e93SStefan Hajnoczi if (ret) { 4926f321e93SStefan Hajnoczi goto out; 4936f321e93SStefan Hajnoczi } 49401979a98SStefan Hajnoczi } 49501979a98SStefan Hajnoczi } 49601979a98SStefan Hajnoczi 497a8c868c3SStefan Hajnoczi bdrv_qed_attach_aio_context(bs, bdrv_get_aio_context(bs)); 4986f321e93SStefan Hajnoczi 49901979a98SStefan Hajnoczi out: 50001979a98SStefan Hajnoczi if (ret) { 501298800caSStefan Hajnoczi qed_free_l2_cache(&s->l2_cache); 502298800caSStefan Hajnoczi qemu_vfree(s->l1_table); 503298800caSStefan Hajnoczi } 50475411d23SStefan Hajnoczi return ret; 50575411d23SStefan Hajnoczi } 50675411d23SStefan Hajnoczi 5074e4bf5c4SKevin Wolf static int bdrv_qed_open(BlockDriverState *bs, QDict *options, int flags, 5084e4bf5c4SKevin Wolf Error **errp) 5094e4bf5c4SKevin Wolf { 5104e4bf5c4SKevin Wolf bs->file = bdrv_open_child(NULL, options, "file", bs, &child_file, 5114e4bf5c4SKevin Wolf false, errp); 5124e4bf5c4SKevin Wolf if (!bs->file) { 5134e4bf5c4SKevin Wolf return -EINVAL; 5144e4bf5c4SKevin Wolf } 5154e4bf5c4SKevin Wolf 516*61c7887eSPaolo Bonzini bdrv_qed_init_state(bs); 5174e4bf5c4SKevin Wolf return bdrv_qed_do_open(bs, options, flags, errp); 5184e4bf5c4SKevin Wolf } 5194e4bf5c4SKevin Wolf 5203baca891SKevin Wolf static void bdrv_qed_refresh_limits(BlockDriverState *bs, Error **errp) 521d34682cdSKevin Wolf { 522d34682cdSKevin Wolf BDRVQEDState *s = bs->opaque; 523d34682cdSKevin Wolf 524cf081fcaSEric Blake bs->bl.pwrite_zeroes_alignment = s->header.cluster_size; 525d34682cdSKevin Wolf } 526d34682cdSKevin Wolf 527f9cb20f1SJeff Cody /* We have nothing to do for QED reopen, stubs just return 528f9cb20f1SJeff Cody * success */ 529f9cb20f1SJeff Cody static int bdrv_qed_reopen_prepare(BDRVReopenState *state, 530f9cb20f1SJeff Cody BlockReopenQueue *queue, Error **errp) 531f9cb20f1SJeff Cody { 532f9cb20f1SJeff Cody return 0; 533f9cb20f1SJeff Cody } 534f9cb20f1SJeff Cody 53575411d23SStefan Hajnoczi static void bdrv_qed_close(BlockDriverState *bs) 53675411d23SStefan Hajnoczi { 537298800caSStefan Hajnoczi BDRVQEDState *s = bs->opaque; 538298800caSStefan Hajnoczi 539a8c868c3SStefan Hajnoczi bdrv_qed_detach_aio_context(bs); 5406f321e93SStefan Hajnoczi 54101979a98SStefan Hajnoczi /* Ensure writes reach stable storage */ 5429a4f4c31SKevin Wolf bdrv_flush(bs->file->bs); 54301979a98SStefan Hajnoczi 54401979a98SStefan Hajnoczi /* Clean shutdown, no check required on next open */ 54501979a98SStefan Hajnoczi if (s->header.features & QED_F_NEED_CHECK) { 54601979a98SStefan Hajnoczi s->header.features &= ~QED_F_NEED_CHECK; 54701979a98SStefan Hajnoczi qed_write_header_sync(s); 54801979a98SStefan Hajnoczi } 54901979a98SStefan Hajnoczi 550298800caSStefan Hajnoczi qed_free_l2_cache(&s->l2_cache); 551298800caSStefan Hajnoczi qemu_vfree(s->l1_table); 55275411d23SStefan Hajnoczi } 55375411d23SStefan Hajnoczi 55475411d23SStefan Hajnoczi static int qed_create(const char *filename, uint32_t cluster_size, 55575411d23SStefan Hajnoczi uint64_t image_size, uint32_t table_size, 5560fea6b79SPaolo Bonzini const char *backing_file, const char *backing_fmt, 5574ab15590SChunyan Liu QemuOpts *opts, Error **errp) 55875411d23SStefan Hajnoczi { 55975411d23SStefan Hajnoczi QEDHeader header = { 56075411d23SStefan Hajnoczi .magic = QED_MAGIC, 56175411d23SStefan Hajnoczi .cluster_size = cluster_size, 56275411d23SStefan Hajnoczi .table_size = table_size, 56375411d23SStefan Hajnoczi .header_size = 1, 56475411d23SStefan Hajnoczi .features = 0, 56575411d23SStefan Hajnoczi .compat_features = 0, 56675411d23SStefan Hajnoczi .l1_table_offset = cluster_size, 56775411d23SStefan Hajnoczi .image_size = image_size, 56875411d23SStefan Hajnoczi }; 56975411d23SStefan Hajnoczi QEDHeader le_header; 57075411d23SStefan Hajnoczi uint8_t *l1_table = NULL; 57175411d23SStefan Hajnoczi size_t l1_size = header.cluster_size * header.table_size; 57234b5d2c6SMax Reitz Error *local_err = NULL; 57375411d23SStefan Hajnoczi int ret = 0; 5748a56fdadSKevin Wolf BlockBackend *blk; 57575411d23SStefan Hajnoczi 5764ab15590SChunyan Liu ret = bdrv_create_file(filename, opts, &local_err); 57775411d23SStefan Hajnoczi if (ret < 0) { 5780fea6b79SPaolo Bonzini error_propagate(errp, local_err); 57975411d23SStefan Hajnoczi return ret; 58075411d23SStefan Hajnoczi } 58175411d23SStefan Hajnoczi 582efaa7c4eSMax Reitz blk = blk_new_open(filename, NULL, NULL, 58355880601SKevin Wolf BDRV_O_RDWR | BDRV_O_RESIZE | BDRV_O_PROTOCOL, 58455880601SKevin Wolf &local_err); 5858a56fdadSKevin Wolf if (blk == NULL) { 5860fea6b79SPaolo Bonzini error_propagate(errp, local_err); 5878a56fdadSKevin Wolf return -EIO; 58875411d23SStefan Hajnoczi } 58975411d23SStefan Hajnoczi 5908a56fdadSKevin Wolf blk_set_allow_write_beyond_eof(blk, true); 5918a56fdadSKevin Wolf 592c743849bSStefan Hajnoczi /* File must start empty and grow, check truncate is supported */ 5933a691c50SMax Reitz ret = blk_truncate(blk, 0, PREALLOC_MODE_OFF, errp); 594c743849bSStefan Hajnoczi if (ret < 0) { 595c743849bSStefan Hajnoczi goto out; 596c743849bSStefan Hajnoczi } 597c743849bSStefan Hajnoczi 59875411d23SStefan Hajnoczi if (backing_file) { 59975411d23SStefan Hajnoczi header.features |= QED_F_BACKING_FILE; 60075411d23SStefan Hajnoczi header.backing_filename_offset = sizeof(le_header); 60175411d23SStefan Hajnoczi header.backing_filename_size = strlen(backing_file); 60275411d23SStefan Hajnoczi 60375411d23SStefan Hajnoczi if (qed_fmt_is_raw(backing_fmt)) { 60475411d23SStefan Hajnoczi header.features |= QED_F_BACKING_FORMAT_NO_PROBE; 60575411d23SStefan Hajnoczi } 60675411d23SStefan Hajnoczi } 60775411d23SStefan Hajnoczi 60875411d23SStefan Hajnoczi qed_header_cpu_to_le(&header, &le_header); 6098341f00dSEric Blake ret = blk_pwrite(blk, 0, &le_header, sizeof(le_header), 0); 61075411d23SStefan Hajnoczi if (ret < 0) { 61175411d23SStefan Hajnoczi goto out; 61275411d23SStefan Hajnoczi } 6138a56fdadSKevin Wolf ret = blk_pwrite(blk, sizeof(le_header), backing_file, 6148341f00dSEric Blake header.backing_filename_size, 0); 61575411d23SStefan Hajnoczi if (ret < 0) { 61675411d23SStefan Hajnoczi goto out; 61775411d23SStefan Hajnoczi } 61875411d23SStefan Hajnoczi 6197267c094SAnthony Liguori l1_table = g_malloc0(l1_size); 6208341f00dSEric Blake ret = blk_pwrite(blk, header.l1_table_offset, l1_table, l1_size, 0); 62175411d23SStefan Hajnoczi if (ret < 0) { 62275411d23SStefan Hajnoczi goto out; 62375411d23SStefan Hajnoczi } 62475411d23SStefan Hajnoczi 62575411d23SStefan Hajnoczi ret = 0; /* success */ 62675411d23SStefan Hajnoczi out: 6277267c094SAnthony Liguori g_free(l1_table); 6288a56fdadSKevin Wolf blk_unref(blk); 62975411d23SStefan Hajnoczi return ret; 63075411d23SStefan Hajnoczi } 63175411d23SStefan Hajnoczi 6327ab74849SChunyan Liu static int bdrv_qed_create(const char *filename, QemuOpts *opts, Error **errp) 63375411d23SStefan Hajnoczi { 63475411d23SStefan Hajnoczi uint64_t image_size = 0; 63575411d23SStefan Hajnoczi uint32_t cluster_size = QED_DEFAULT_CLUSTER_SIZE; 63675411d23SStefan Hajnoczi uint32_t table_size = QED_DEFAULT_TABLE_SIZE; 6377ab74849SChunyan Liu char *backing_file = NULL; 6387ab74849SChunyan Liu char *backing_fmt = NULL; 6397ab74849SChunyan Liu int ret; 64075411d23SStefan Hajnoczi 641c2eb918eSHu Tao image_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0), 642c2eb918eSHu Tao BDRV_SECTOR_SIZE); 6437ab74849SChunyan Liu backing_file = qemu_opt_get_del(opts, BLOCK_OPT_BACKING_FILE); 6447ab74849SChunyan Liu backing_fmt = qemu_opt_get_del(opts, BLOCK_OPT_BACKING_FMT); 6457ab74849SChunyan Liu cluster_size = qemu_opt_get_size_del(opts, 6467ab74849SChunyan Liu BLOCK_OPT_CLUSTER_SIZE, 6477ab74849SChunyan Liu QED_DEFAULT_CLUSTER_SIZE); 6487ab74849SChunyan Liu table_size = qemu_opt_get_size_del(opts, BLOCK_OPT_TABLE_SIZE, 6497ab74849SChunyan Liu QED_DEFAULT_TABLE_SIZE); 65075411d23SStefan Hajnoczi 65175411d23SStefan Hajnoczi if (!qed_is_cluster_size_valid(cluster_size)) { 6525ff679b4SAakriti Gupta error_setg(errp, "QED cluster size must be within range [%u, %u] " 6535ff679b4SAakriti Gupta "and power of 2", 65475411d23SStefan Hajnoczi QED_MIN_CLUSTER_SIZE, QED_MAX_CLUSTER_SIZE); 6557ab74849SChunyan Liu ret = -EINVAL; 6567ab74849SChunyan Liu goto finish; 65775411d23SStefan Hajnoczi } 65875411d23SStefan Hajnoczi if (!qed_is_table_size_valid(table_size)) { 6595ff679b4SAakriti Gupta error_setg(errp, "QED table size must be within range [%u, %u] " 6605ff679b4SAakriti Gupta "and power of 2", 66175411d23SStefan Hajnoczi QED_MIN_TABLE_SIZE, QED_MAX_TABLE_SIZE); 6627ab74849SChunyan Liu ret = -EINVAL; 6637ab74849SChunyan Liu goto finish; 66475411d23SStefan Hajnoczi } 66575411d23SStefan Hajnoczi if (!qed_is_image_size_valid(image_size, cluster_size, table_size)) { 6665ff679b4SAakriti Gupta error_setg(errp, "QED image size must be a non-zero multiple of " 6675ff679b4SAakriti Gupta "cluster size and less than %" PRIu64 " bytes", 66875411d23SStefan Hajnoczi qed_max_image_size(cluster_size, table_size)); 6697ab74849SChunyan Liu ret = -EINVAL; 6707ab74849SChunyan Liu goto finish; 67175411d23SStefan Hajnoczi } 67275411d23SStefan Hajnoczi 6737ab74849SChunyan Liu ret = qed_create(filename, cluster_size, image_size, table_size, 6744ab15590SChunyan Liu backing_file, backing_fmt, opts, errp); 6757ab74849SChunyan Liu 6767ab74849SChunyan Liu finish: 6777ab74849SChunyan Liu g_free(backing_file); 6787ab74849SChunyan Liu g_free(backing_fmt); 6797ab74849SChunyan Liu return ret; 68075411d23SStefan Hajnoczi } 68175411d23SStefan Hajnoczi 682298800caSStefan Hajnoczi typedef struct { 6834bc74be9SPaolo Bonzini BlockDriverState *bs; 684b7d5a5b8SStefan Hajnoczi Coroutine *co; 6854bc74be9SPaolo Bonzini uint64_t pos; 6864bc74be9SPaolo Bonzini int64_t status; 687298800caSStefan Hajnoczi int *pnum; 68853f1dfd1SFam Zheng BlockDriverState **file; 689298800caSStefan Hajnoczi } QEDIsAllocatedCB; 690298800caSStefan Hajnoczi 691298800caSStefan Hajnoczi static void qed_is_allocated_cb(void *opaque, int ret, uint64_t offset, size_t len) 692298800caSStefan Hajnoczi { 693298800caSStefan Hajnoczi QEDIsAllocatedCB *cb = opaque; 6944bc74be9SPaolo Bonzini BDRVQEDState *s = cb->bs->opaque; 695298800caSStefan Hajnoczi *cb->pnum = len / BDRV_SECTOR_SIZE; 6964bc74be9SPaolo Bonzini switch (ret) { 6974bc74be9SPaolo Bonzini case QED_CLUSTER_FOUND: 6984bc74be9SPaolo Bonzini offset |= qed_offset_into_cluster(s, cb->pos); 6994bc74be9SPaolo Bonzini cb->status = BDRV_BLOCK_DATA | BDRV_BLOCK_OFFSET_VALID | offset; 70053f1dfd1SFam Zheng *cb->file = cb->bs->file->bs; 7014bc74be9SPaolo Bonzini break; 7024bc74be9SPaolo Bonzini case QED_CLUSTER_ZERO: 7034bc74be9SPaolo Bonzini cb->status = BDRV_BLOCK_ZERO; 7044bc74be9SPaolo Bonzini break; 7054bc74be9SPaolo Bonzini case QED_CLUSTER_L2: 7064bc74be9SPaolo Bonzini case QED_CLUSTER_L1: 7074bc74be9SPaolo Bonzini cb->status = 0; 7084bc74be9SPaolo Bonzini break; 7094bc74be9SPaolo Bonzini default: 7104bc74be9SPaolo Bonzini assert(ret < 0); 7114bc74be9SPaolo Bonzini cb->status = ret; 7124bc74be9SPaolo Bonzini break; 7134bc74be9SPaolo Bonzini } 7144bc74be9SPaolo Bonzini 715b7d5a5b8SStefan Hajnoczi if (cb->co) { 716b9e413ddSPaolo Bonzini aio_co_wake(cb->co); 717b7d5a5b8SStefan Hajnoczi } 718298800caSStefan Hajnoczi } 719298800caSStefan Hajnoczi 720b6b8a333SPaolo Bonzini static int64_t coroutine_fn bdrv_qed_co_get_block_status(BlockDriverState *bs, 721b7d5a5b8SStefan Hajnoczi int64_t sector_num, 72267a0fd2aSFam Zheng int nb_sectors, int *pnum, 72367a0fd2aSFam Zheng BlockDriverState **file) 72475411d23SStefan Hajnoczi { 725298800caSStefan Hajnoczi BDRVQEDState *s = bs->opaque; 726298800caSStefan Hajnoczi size_t len = (size_t)nb_sectors * BDRV_SECTOR_SIZE; 727298800caSStefan Hajnoczi QEDIsAllocatedCB cb = { 7284bc74be9SPaolo Bonzini .bs = bs, 7294bc74be9SPaolo Bonzini .pos = (uint64_t)sector_num * BDRV_SECTOR_SIZE, 7304bc74be9SPaolo Bonzini .status = BDRV_BLOCK_OFFSET_MASK, 731298800caSStefan Hajnoczi .pnum = pnum, 73253f1dfd1SFam Zheng .file = file, 733298800caSStefan Hajnoczi }; 734298800caSStefan Hajnoczi QEDRequest request = { .l2_table = NULL }; 7350f21b7a1SKevin Wolf uint64_t offset; 7360f21b7a1SKevin Wolf int ret; 737298800caSStefan Hajnoczi 7380f21b7a1SKevin Wolf ret = qed_find_cluster(s, &request, cb.pos, &len, &offset); 7390f21b7a1SKevin Wolf qed_is_allocated_cb(&cb, ret, offset, len); 740298800caSStefan Hajnoczi 7410f21b7a1SKevin Wolf /* The callback was invoked immediately */ 7420f21b7a1SKevin Wolf assert(cb.status != BDRV_BLOCK_OFFSET_MASK); 743298800caSStefan Hajnoczi 744298800caSStefan Hajnoczi qed_unref_l2_cache_entry(request.l2_table); 745298800caSStefan Hajnoczi 7464bc74be9SPaolo Bonzini return cb.status; 74775411d23SStefan Hajnoczi } 74875411d23SStefan Hajnoczi 749eabba580SStefan Hajnoczi static BDRVQEDState *acb_to_s(QEDAIOCB *acb) 750eabba580SStefan Hajnoczi { 75148cc565eSKevin Wolf return acb->bs->opaque; 752eabba580SStefan Hajnoczi } 753eabba580SStefan Hajnoczi 754eabba580SStefan Hajnoczi /** 755eabba580SStefan Hajnoczi * Read from the backing file or zero-fill if no backing file 756eabba580SStefan Hajnoczi * 757eabba580SStefan Hajnoczi * @s: QED state 758eabba580SStefan Hajnoczi * @pos: Byte position in device 759eabba580SStefan Hajnoczi * @qiov: Destination I/O vector 760f06ee3d4SKevin Wolf * @backing_qiov: Possibly shortened copy of qiov, to be allocated here 761eabba580SStefan Hajnoczi * @cb: Completion function 762eabba580SStefan Hajnoczi * @opaque: User data for completion function 763eabba580SStefan Hajnoczi * 764eabba580SStefan Hajnoczi * This function reads qiov->size bytes starting at pos from the backing file. 765eabba580SStefan Hajnoczi * If there is no backing file then zeroes are read. 766eabba580SStefan Hajnoczi */ 76787f0d882SKevin Wolf static int coroutine_fn qed_read_backing_file(BDRVQEDState *s, uint64_t pos, 768eabba580SStefan Hajnoczi QEMUIOVector *qiov, 769e85c5281SKevin Wolf QEMUIOVector **backing_qiov) 770eabba580SStefan Hajnoczi { 771eabba580SStefan Hajnoczi uint64_t backing_length = 0; 772eabba580SStefan Hajnoczi size_t size; 773e85c5281SKevin Wolf int ret; 774eabba580SStefan Hajnoczi 775eabba580SStefan Hajnoczi /* If there is a backing file, get its length. Treat the absence of a 776eabba580SStefan Hajnoczi * backing file like a zero length backing file. 777eabba580SStefan Hajnoczi */ 778760e0063SKevin Wolf if (s->bs->backing) { 779760e0063SKevin Wolf int64_t l = bdrv_getlength(s->bs->backing->bs); 780eabba580SStefan Hajnoczi if (l < 0) { 781e85c5281SKevin Wolf return l; 782eabba580SStefan Hajnoczi } 783eabba580SStefan Hajnoczi backing_length = l; 784eabba580SStefan Hajnoczi } 785eabba580SStefan Hajnoczi 786eabba580SStefan Hajnoczi /* Zero all sectors if reading beyond the end of the backing file */ 787eabba580SStefan Hajnoczi if (pos >= backing_length || 788eabba580SStefan Hajnoczi pos + qiov->size > backing_length) { 7893d9b4925SMichael Tokarev qemu_iovec_memset(qiov, 0, 0, qiov->size); 790eabba580SStefan Hajnoczi } 791eabba580SStefan Hajnoczi 792eabba580SStefan Hajnoczi /* Complete now if there are no backing file sectors to read */ 793eabba580SStefan Hajnoczi if (pos >= backing_length) { 794e85c5281SKevin Wolf return 0; 795eabba580SStefan Hajnoczi } 796eabba580SStefan Hajnoczi 797eabba580SStefan Hajnoczi /* If the read straddles the end of the backing file, shorten it */ 798eabba580SStefan Hajnoczi size = MIN((uint64_t)backing_length - pos, qiov->size); 799eabba580SStefan Hajnoczi 800f06ee3d4SKevin Wolf assert(*backing_qiov == NULL); 801f06ee3d4SKevin Wolf *backing_qiov = g_new(QEMUIOVector, 1); 802f06ee3d4SKevin Wolf qemu_iovec_init(*backing_qiov, qiov->niov); 803f06ee3d4SKevin Wolf qemu_iovec_concat(*backing_qiov, qiov, 0, size); 804f06ee3d4SKevin Wolf 805820100fdSPaolo Bonzini BLKDBG_EVENT(s->bs->file, BLKDBG_READ_BACKING_AIO); 8060f714ec7SKevin Wolf ret = bdrv_co_preadv(s->bs->backing, pos, size, *backing_qiov, 0); 807e85c5281SKevin Wolf if (ret < 0) { 808e85c5281SKevin Wolf return ret; 809e85c5281SKevin Wolf } 810e85c5281SKevin Wolf return 0; 811eabba580SStefan Hajnoczi } 812eabba580SStefan Hajnoczi 813eabba580SStefan Hajnoczi /** 814eabba580SStefan Hajnoczi * Copy data from backing file into the image 815eabba580SStefan Hajnoczi * 816eabba580SStefan Hajnoczi * @s: QED state 817eabba580SStefan Hajnoczi * @pos: Byte position in device 818eabba580SStefan Hajnoczi * @len: Number of bytes 819eabba580SStefan Hajnoczi * @offset: Byte offset in image file 820eabba580SStefan Hajnoczi */ 82187f0d882SKevin Wolf static int coroutine_fn qed_copy_from_backing_file(BDRVQEDState *s, 82287f0d882SKevin Wolf uint64_t pos, uint64_t len, 82387f0d882SKevin Wolf uint64_t offset) 824eabba580SStefan Hajnoczi { 8250f7aa24dSKevin Wolf QEMUIOVector qiov; 8260f7aa24dSKevin Wolf QEMUIOVector *backing_qiov = NULL; 8270f7aa24dSKevin Wolf struct iovec iov; 828e85c5281SKevin Wolf int ret; 829eabba580SStefan Hajnoczi 830eabba580SStefan Hajnoczi /* Skip copy entirely if there is no work to do */ 831eabba580SStefan Hajnoczi if (len == 0) { 832b4ac32f3SKevin Wolf return 0; 833eabba580SStefan Hajnoczi } 834eabba580SStefan Hajnoczi 8350f7aa24dSKevin Wolf iov = (struct iovec) { 8360f7aa24dSKevin Wolf .iov_base = qemu_blockalign(s->bs, len), 8370f7aa24dSKevin Wolf .iov_len = len, 8380f7aa24dSKevin Wolf }; 8390f7aa24dSKevin Wolf qemu_iovec_init_external(&qiov, &iov, 1); 840eabba580SStefan Hajnoczi 8410f7aa24dSKevin Wolf ret = qed_read_backing_file(s, pos, &qiov, &backing_qiov); 8420f7aa24dSKevin Wolf 8430f7aa24dSKevin Wolf if (backing_qiov) { 8440f7aa24dSKevin Wolf qemu_iovec_destroy(backing_qiov); 8450f7aa24dSKevin Wolf g_free(backing_qiov); 8460f7aa24dSKevin Wolf backing_qiov = NULL; 8470f7aa24dSKevin Wolf } 8480f7aa24dSKevin Wolf 8490f7aa24dSKevin Wolf if (ret) { 8500f7aa24dSKevin Wolf goto out; 8510f7aa24dSKevin Wolf } 8520f7aa24dSKevin Wolf 8530f7aa24dSKevin Wolf BLKDBG_EVENT(s->bs->file, BLKDBG_COW_WRITE); 8540f714ec7SKevin Wolf ret = bdrv_co_pwritev(s->bs->file, offset, qiov.size, &qiov, 0); 8550f7aa24dSKevin Wolf if (ret < 0) { 8560f7aa24dSKevin Wolf goto out; 8570f7aa24dSKevin Wolf } 8580f7aa24dSKevin Wolf ret = 0; 8590f7aa24dSKevin Wolf out: 8600f7aa24dSKevin Wolf qemu_vfree(iov.iov_base); 861b4ac32f3SKevin Wolf return ret; 862eabba580SStefan Hajnoczi } 863eabba580SStefan Hajnoczi 864eabba580SStefan Hajnoczi /** 865eabba580SStefan Hajnoczi * Link one or more contiguous clusters into a table 866eabba580SStefan Hajnoczi * 867eabba580SStefan Hajnoczi * @s: QED state 868eabba580SStefan Hajnoczi * @table: L2 table 869eabba580SStefan Hajnoczi * @index: First cluster index 870eabba580SStefan Hajnoczi * @n: Number of contiguous clusters 87121df65b6SAnthony Liguori * @cluster: First cluster offset 87221df65b6SAnthony Liguori * 87321df65b6SAnthony Liguori * The cluster offset may be an allocated byte offset in the image file, the 87421df65b6SAnthony Liguori * zero cluster marker, or the unallocated cluster marker. 875eabba580SStefan Hajnoczi */ 87687f0d882SKevin Wolf static void coroutine_fn qed_update_l2_table(BDRVQEDState *s, QEDTable *table, 87787f0d882SKevin Wolf int index, unsigned int n, 87887f0d882SKevin Wolf uint64_t cluster) 879eabba580SStefan Hajnoczi { 880eabba580SStefan Hajnoczi int i; 881eabba580SStefan Hajnoczi for (i = index; i < index + n; i++) { 882eabba580SStefan Hajnoczi table->offsets[i] = cluster; 88321df65b6SAnthony Liguori if (!qed_offset_is_unalloc_cluster(cluster) && 88421df65b6SAnthony Liguori !qed_offset_is_zero_cluster(cluster)) { 885eabba580SStefan Hajnoczi cluster += s->header.cluster_size; 886eabba580SStefan Hajnoczi } 887eabba580SStefan Hajnoczi } 88821df65b6SAnthony Liguori } 889eabba580SStefan Hajnoczi 89087f0d882SKevin Wolf static void coroutine_fn qed_aio_complete(QEDAIOCB *acb) 891eabba580SStefan Hajnoczi { 892eabba580SStefan Hajnoczi BDRVQEDState *s = acb_to_s(acb); 893eabba580SStefan Hajnoczi 894eabba580SStefan Hajnoczi /* Free resources */ 895eabba580SStefan Hajnoczi qemu_iovec_destroy(&acb->cur_qiov); 896eabba580SStefan Hajnoczi qed_unref_l2_cache_entry(acb->request.l2_table); 897eabba580SStefan Hajnoczi 8980e71be19SStefan Hajnoczi /* Free the buffer we may have allocated for zero writes */ 8990e71be19SStefan Hajnoczi if (acb->flags & QED_AIOCB_ZERO) { 9000e71be19SStefan Hajnoczi qemu_vfree(acb->qiov->iov[0].iov_base); 9010e71be19SStefan Hajnoczi acb->qiov->iov[0].iov_base = NULL; 9020e71be19SStefan Hajnoczi } 9030e71be19SStefan Hajnoczi 904eabba580SStefan Hajnoczi /* Start next allocating write request waiting behind this one. Note that 905eabba580SStefan Hajnoczi * requests enqueue themselves when they first hit an unallocated cluster 906eabba580SStefan Hajnoczi * but they wait until the entire request is finished before waking up the 907eabba580SStefan Hajnoczi * next request in the queue. This ensures that we don't cycle through 908eabba580SStefan Hajnoczi * requests multiple times but rather finish one at a time completely. 909eabba580SStefan Hajnoczi */ 9100806c3b5SKevin Wolf if (acb == s->allocating_acb) { 9110806c3b5SKevin Wolf s->allocating_acb = NULL; 9120806c3b5SKevin Wolf if (!qemu_co_queue_empty(&s->allocating_write_reqs)) { 9130806c3b5SKevin Wolf qemu_co_enter_next(&s->allocating_write_reqs); 9146f321e93SStefan Hajnoczi } else if (s->header.features & QED_F_NEED_CHECK) { 9156f321e93SStefan Hajnoczi qed_start_need_check_timer(s); 916eabba580SStefan Hajnoczi } 917eabba580SStefan Hajnoczi } 918eabba580SStefan Hajnoczi } 919eabba580SStefan Hajnoczi 920eabba580SStefan Hajnoczi /** 921fae25ac7SKevin Wolf * Update L1 table with new L2 table offset and write it out 922eabba580SStefan Hajnoczi */ 92387f0d882SKevin Wolf static int coroutine_fn qed_aio_write_l1_update(QEDAIOCB *acb) 924eabba580SStefan Hajnoczi { 925eabba580SStefan Hajnoczi BDRVQEDState *s = acb_to_s(acb); 926eabba580SStefan Hajnoczi CachedL2Table *l2_table = acb->request.l2_table; 927e4fc8781SStefan Hajnoczi uint64_t l2_offset = l2_table->offset; 928fb18de21SKevin Wolf int index, ret; 929fae25ac7SKevin Wolf 930fae25ac7SKevin Wolf index = qed_l1_index(s, acb->cur_pos); 931fae25ac7SKevin Wolf s->l1_table->offsets[index] = l2_table->offset; 932fae25ac7SKevin Wolf 933fae25ac7SKevin Wolf ret = qed_write_l1_table(s, index, 1); 934fae25ac7SKevin Wolf 935fae25ac7SKevin Wolf /* Commit the current L2 table to the cache */ 936eabba580SStefan Hajnoczi qed_commit_l2_cache_entry(&s->l2_cache, l2_table); 937eabba580SStefan Hajnoczi 938eabba580SStefan Hajnoczi /* This is guaranteed to succeed because we just committed the entry to the 939eabba580SStefan Hajnoczi * cache. 940eabba580SStefan Hajnoczi */ 941e4fc8781SStefan Hajnoczi acb->request.l2_table = qed_find_l2_cache_entry(&s->l2_cache, l2_offset); 942eabba580SStefan Hajnoczi assert(acb->request.l2_table != NULL); 943eabba580SStefan Hajnoczi 944fb18de21SKevin Wolf return ret; 945eabba580SStefan Hajnoczi } 946eabba580SStefan Hajnoczi 947eabba580SStefan Hajnoczi 948eabba580SStefan Hajnoczi /** 949eabba580SStefan Hajnoczi * Update L2 table with new cluster offsets and write them out 950eabba580SStefan Hajnoczi */ 95187f0d882SKevin Wolf static int coroutine_fn qed_aio_write_l2_update(QEDAIOCB *acb, uint64_t offset) 952eabba580SStefan Hajnoczi { 953eabba580SStefan Hajnoczi BDRVQEDState *s = acb_to_s(acb); 954eabba580SStefan Hajnoczi bool need_alloc = acb->find_cluster_ret == QED_CLUSTER_L1; 95588d2dd72SKevin Wolf int index, ret; 956eabba580SStefan Hajnoczi 957eabba580SStefan Hajnoczi if (need_alloc) { 958eabba580SStefan Hajnoczi qed_unref_l2_cache_entry(acb->request.l2_table); 959eabba580SStefan Hajnoczi acb->request.l2_table = qed_new_l2_table(s); 960eabba580SStefan Hajnoczi } 961eabba580SStefan Hajnoczi 962eabba580SStefan Hajnoczi index = qed_l2_index(s, acb->cur_pos); 963eabba580SStefan Hajnoczi qed_update_l2_table(s, acb->request.l2_table->table, index, acb->cur_nclusters, 9640e71be19SStefan Hajnoczi offset); 965eabba580SStefan Hajnoczi 966eabba580SStefan Hajnoczi if (need_alloc) { 967eabba580SStefan Hajnoczi /* Write out the whole new L2 table */ 968453e53e2SKevin Wolf ret = qed_write_l2_table(s, &acb->request, 0, s->table_nelems, true); 969fb18de21SKevin Wolf if (ret) { 97088d2dd72SKevin Wolf return ret; 971fb18de21SKevin Wolf } 97288d2dd72SKevin Wolf return qed_aio_write_l1_update(acb); 973eabba580SStefan Hajnoczi } else { 974eabba580SStefan Hajnoczi /* Write out only the updated part of the L2 table */ 975453e53e2SKevin Wolf ret = qed_write_l2_table(s, &acb->request, index, acb->cur_nclusters, 976453e53e2SKevin Wolf false); 97788d2dd72SKevin Wolf if (ret) { 97888d2dd72SKevin Wolf return ret; 979eabba580SStefan Hajnoczi } 98088d2dd72SKevin Wolf } 98188d2dd72SKevin Wolf return 0; 982eabba580SStefan Hajnoczi } 983eabba580SStefan Hajnoczi 984eabba580SStefan Hajnoczi /** 985eabba580SStefan Hajnoczi * Write data to the image file 986eabba580SStefan Hajnoczi */ 98787f0d882SKevin Wolf static int coroutine_fn qed_aio_write_main(QEDAIOCB *acb) 988eabba580SStefan Hajnoczi { 989eabba580SStefan Hajnoczi BDRVQEDState *s = acb_to_s(acb); 990eabba580SStefan Hajnoczi uint64_t offset = acb->cur_cluster + 991eabba580SStefan Hajnoczi qed_offset_into_cluster(s, acb->cur_pos); 992eabba580SStefan Hajnoczi 993eaf0bc56SKevin Wolf trace_qed_aio_write_main(s, acb, 0, offset, acb->cur_qiov.size); 994eabba580SStefan Hajnoczi 995a4d8f1aeSKevin Wolf BLKDBG_EVENT(s->bs->file, BLKDBG_WRITE_AIO); 996e7569c18SPaolo Bonzini return bdrv_co_pwritev(s->bs->file, offset, acb->cur_qiov.size, 9970f714ec7SKevin Wolf &acb->cur_qiov, 0); 998eabba580SStefan Hajnoczi } 999eabba580SStefan Hajnoczi 1000eabba580SStefan Hajnoczi /** 1001b4ac32f3SKevin Wolf * Populate untouched regions of new data cluster 1002eabba580SStefan Hajnoczi */ 100387f0d882SKevin Wolf static int coroutine_fn qed_aio_write_cow(QEDAIOCB *acb) 1004eabba580SStefan Hajnoczi { 1005eabba580SStefan Hajnoczi BDRVQEDState *s = acb_to_s(acb); 1006b4ac32f3SKevin Wolf uint64_t start, len, offset; 1007a101341aSKevin Wolf int ret; 1008eabba580SStefan Hajnoczi 1009b4ac32f3SKevin Wolf /* Populate front untouched region of new data cluster */ 1010b4ac32f3SKevin Wolf start = qed_start_of_cluster(s, acb->cur_pos); 1011b4ac32f3SKevin Wolf len = qed_offset_into_cluster(s, acb->cur_pos); 1012b4ac32f3SKevin Wolf 1013b4ac32f3SKevin Wolf trace_qed_aio_write_prefill(s, acb, start, len, acb->cur_cluster); 1014b4ac32f3SKevin Wolf ret = qed_copy_from_backing_file(s, start, len, acb->cur_cluster); 1015a101341aSKevin Wolf if (ret < 0) { 1016a101341aSKevin Wolf return ret; 1017eabba580SStefan Hajnoczi } 1018eabba580SStefan Hajnoczi 1019b4ac32f3SKevin Wolf /* Populate back untouched region of new data cluster */ 1020b4ac32f3SKevin Wolf start = acb->cur_pos + acb->cur_qiov.size; 1021b4ac32f3SKevin Wolf len = qed_start_of_cluster(s, start + s->header.cluster_size - 1) - start; 1022b4ac32f3SKevin Wolf offset = acb->cur_cluster + 1023b4ac32f3SKevin Wolf qed_offset_into_cluster(s, acb->cur_pos) + 1024b4ac32f3SKevin Wolf acb->cur_qiov.size; 1025b4ac32f3SKevin Wolf 1026eabba580SStefan Hajnoczi trace_qed_aio_write_postfill(s, acb, start, len, offset); 1027b4ac32f3SKevin Wolf ret = qed_copy_from_backing_file(s, start, len, offset); 1028a101341aSKevin Wolf if (ret < 0) { 1029a101341aSKevin Wolf return ret; 1030eaf0bc56SKevin Wolf } 1031eabba580SStefan Hajnoczi 1032e7569c18SPaolo Bonzini ret = qed_aio_write_main(acb); 1033e7569c18SPaolo Bonzini if (ret < 0) { 1034e7569c18SPaolo Bonzini return ret; 1035e7569c18SPaolo Bonzini } 1036e7569c18SPaolo Bonzini 1037e7569c18SPaolo Bonzini if (s->bs->backing) { 1038e7569c18SPaolo Bonzini /* 1039e7569c18SPaolo Bonzini * Flush new data clusters before updating the L2 table 1040e7569c18SPaolo Bonzini * 1041e7569c18SPaolo Bonzini * This flush is necessary when a backing file is in use. A crash 1042e7569c18SPaolo Bonzini * during an allocating write could result in empty clusters in the 1043e7569c18SPaolo Bonzini * image. If the write only touched a subregion of the cluster, 1044e7569c18SPaolo Bonzini * then backing image sectors have been lost in the untouched 1045e7569c18SPaolo Bonzini * region. The solution is to flush after writing a new data 1046e7569c18SPaolo Bonzini * cluster and before updating the L2 table. 1047e7569c18SPaolo Bonzini */ 1048e7569c18SPaolo Bonzini ret = bdrv_co_flush(s->bs->file->bs); 1049e7569c18SPaolo Bonzini if (ret < 0) { 1050e7569c18SPaolo Bonzini return ret; 1051e7569c18SPaolo Bonzini } 1052e7569c18SPaolo Bonzini } 1053e7569c18SPaolo Bonzini 1054e7569c18SPaolo Bonzini return 0; 1055eabba580SStefan Hajnoczi } 1056eabba580SStefan Hajnoczi 1057eabba580SStefan Hajnoczi /** 10580d09c797SStefan Hajnoczi * Check if the QED_F_NEED_CHECK bit should be set during allocating write 10590d09c797SStefan Hajnoczi */ 10600d09c797SStefan Hajnoczi static bool qed_should_set_need_check(BDRVQEDState *s) 10610d09c797SStefan Hajnoczi { 10620d09c797SStefan Hajnoczi /* The flush before L2 update path ensures consistency */ 1063760e0063SKevin Wolf if (s->bs->backing) { 10640d09c797SStefan Hajnoczi return false; 10650d09c797SStefan Hajnoczi } 10660d09c797SStefan Hajnoczi 10670d09c797SStefan Hajnoczi return !(s->header.features & QED_F_NEED_CHECK); 10680d09c797SStefan Hajnoczi } 10690d09c797SStefan Hajnoczi 10700d09c797SStefan Hajnoczi /** 1071eabba580SStefan Hajnoczi * Write new data cluster 1072eabba580SStefan Hajnoczi * 1073eabba580SStefan Hajnoczi * @acb: Write request 1074eabba580SStefan Hajnoczi * @len: Length in bytes 1075eabba580SStefan Hajnoczi * 1076eabba580SStefan Hajnoczi * This path is taken when writing to previously unallocated clusters. 1077eabba580SStefan Hajnoczi */ 107887f0d882SKevin Wolf static int coroutine_fn qed_aio_write_alloc(QEDAIOCB *acb, size_t len) 1079eabba580SStefan Hajnoczi { 1080eabba580SStefan Hajnoczi BDRVQEDState *s = acb_to_s(acb); 1081f13d712bSKevin Wolf int ret; 1082eabba580SStefan Hajnoczi 10836f321e93SStefan Hajnoczi /* Cancel timer when the first allocating request comes in */ 10840806c3b5SKevin Wolf if (s->allocating_acb == NULL) { 10856f321e93SStefan Hajnoczi qed_cancel_need_check_timer(s); 10866f321e93SStefan Hajnoczi } 10876f321e93SStefan Hajnoczi 1088eabba580SStefan Hajnoczi /* Freeze this request if another allocating write is in progress */ 10890806c3b5SKevin Wolf if (s->allocating_acb != acb || s->allocating_write_reqs_plugged) { 10900806c3b5SKevin Wolf if (s->allocating_acb != NULL) { 10910806c3b5SKevin Wolf qemu_co_queue_wait(&s->allocating_write_reqs, NULL); 10920806c3b5SKevin Wolf assert(s->allocating_acb == NULL); 1093eabba580SStefan Hajnoczi } 10940806c3b5SKevin Wolf s->allocating_acb = acb; 10950806c3b5SKevin Wolf return -EAGAIN; /* start over with looking up table entries */ 1096eabba580SStefan Hajnoczi } 1097eabba580SStefan Hajnoczi 1098eabba580SStefan Hajnoczi acb->cur_nclusters = qed_bytes_to_clusters(s, 1099eabba580SStefan Hajnoczi qed_offset_into_cluster(s, acb->cur_pos) + len); 11001b093c48SMichael Tokarev qemu_iovec_concat(&acb->cur_qiov, acb->qiov, acb->qiov_offset, len); 1101eabba580SStefan Hajnoczi 11020e71be19SStefan Hajnoczi if (acb->flags & QED_AIOCB_ZERO) { 11030e71be19SStefan Hajnoczi /* Skip ahead if the clusters are already zero */ 11040e71be19SStefan Hajnoczi if (acb->find_cluster_ret == QED_CLUSTER_ZERO) { 1105d6daddcdSKevin Wolf return 0; 11060e71be19SStefan Hajnoczi } 1107e7569c18SPaolo Bonzini acb->cur_cluster = 1; 11080e71be19SStefan Hajnoczi } else { 11090e71be19SStefan Hajnoczi acb->cur_cluster = qed_alloc_clusters(s, acb->cur_nclusters); 11100e71be19SStefan Hajnoczi } 11110e71be19SStefan Hajnoczi 11120d09c797SStefan Hajnoczi if (qed_should_set_need_check(s)) { 111301979a98SStefan Hajnoczi s->header.features |= QED_F_NEED_CHECK; 1114f13d712bSKevin Wolf ret = qed_write_header(s); 1115a101341aSKevin Wolf if (ret < 0) { 1116d6daddcdSKevin Wolf return ret; 11170d09c797SStefan Hajnoczi } 1118eabba580SStefan Hajnoczi } 1119eabba580SStefan Hajnoczi 1120e7569c18SPaolo Bonzini if (!(acb->flags & QED_AIOCB_ZERO)) { 1121a101341aSKevin Wolf ret = qed_aio_write_cow(acb); 1122a101341aSKevin Wolf if (ret < 0) { 1123d6daddcdSKevin Wolf return ret; 1124a101341aSKevin Wolf } 1125e7569c18SPaolo Bonzini } 1126e7569c18SPaolo Bonzini 1127e7569c18SPaolo Bonzini return qed_aio_write_l2_update(acb, acb->cur_cluster); 1128a101341aSKevin Wolf } 1129a101341aSKevin Wolf 1130eabba580SStefan Hajnoczi /** 1131eabba580SStefan Hajnoczi * Write data cluster in place 1132eabba580SStefan Hajnoczi * 1133eabba580SStefan Hajnoczi * @acb: Write request 1134eabba580SStefan Hajnoczi * @offset: Cluster offset in bytes 1135eabba580SStefan Hajnoczi * @len: Length in bytes 1136eabba580SStefan Hajnoczi * 1137eabba580SStefan Hajnoczi * This path is taken when writing to already allocated clusters. 1138eabba580SStefan Hajnoczi */ 113987f0d882SKevin Wolf static int coroutine_fn qed_aio_write_inplace(QEDAIOCB *acb, uint64_t offset, 114087f0d882SKevin Wolf size_t len) 1141eabba580SStefan Hajnoczi { 11420e71be19SStefan Hajnoczi /* Allocate buffer for zero writes */ 11430e71be19SStefan Hajnoczi if (acb->flags & QED_AIOCB_ZERO) { 11440e71be19SStefan Hajnoczi struct iovec *iov = acb->qiov->iov; 11450e71be19SStefan Hajnoczi 11460e71be19SStefan Hajnoczi if (!iov->iov_base) { 114748cc565eSKevin Wolf iov->iov_base = qemu_try_blockalign(acb->bs, iov->iov_len); 11484f4896dbSKevin Wolf if (iov->iov_base == NULL) { 1149d6daddcdSKevin Wolf return -ENOMEM; 11504f4896dbSKevin Wolf } 11510e71be19SStefan Hajnoczi memset(iov->iov_base, 0, iov->iov_len); 11520e71be19SStefan Hajnoczi } 11530e71be19SStefan Hajnoczi } 11540e71be19SStefan Hajnoczi 1155eabba580SStefan Hajnoczi /* Calculate the I/O vector */ 1156eabba580SStefan Hajnoczi acb->cur_cluster = offset; 11571b093c48SMichael Tokarev qemu_iovec_concat(&acb->cur_qiov, acb->qiov, acb->qiov_offset, len); 1158eabba580SStefan Hajnoczi 1159eabba580SStefan Hajnoczi /* Do the actual write */ 1160d6daddcdSKevin Wolf return qed_aio_write_main(acb); 1161eabba580SStefan Hajnoczi } 1162eabba580SStefan Hajnoczi 1163eabba580SStefan Hajnoczi /** 1164eabba580SStefan Hajnoczi * Write data cluster 1165eabba580SStefan Hajnoczi * 1166eabba580SStefan Hajnoczi * @opaque: Write request 11670596be7eSKevin Wolf * @ret: QED_CLUSTER_FOUND, QED_CLUSTER_L2 or QED_CLUSTER_L1 1168eabba580SStefan Hajnoczi * @offset: Cluster offset in bytes 1169eabba580SStefan Hajnoczi * @len: Length in bytes 1170eabba580SStefan Hajnoczi */ 117187f0d882SKevin Wolf static int coroutine_fn qed_aio_write_data(void *opaque, int ret, 1172eabba580SStefan Hajnoczi uint64_t offset, size_t len) 1173eabba580SStefan Hajnoczi { 1174eabba580SStefan Hajnoczi QEDAIOCB *acb = opaque; 1175eabba580SStefan Hajnoczi 1176eabba580SStefan Hajnoczi trace_qed_aio_write_data(acb_to_s(acb), acb, ret, offset, len); 1177eabba580SStefan Hajnoczi 1178eabba580SStefan Hajnoczi acb->find_cluster_ret = ret; 1179eabba580SStefan Hajnoczi 1180eabba580SStefan Hajnoczi switch (ret) { 1181eabba580SStefan Hajnoczi case QED_CLUSTER_FOUND: 11820596be7eSKevin Wolf return qed_aio_write_inplace(acb, offset, len); 1183eabba580SStefan Hajnoczi 1184eabba580SStefan Hajnoczi case QED_CLUSTER_L2: 1185eabba580SStefan Hajnoczi case QED_CLUSTER_L1: 118621df65b6SAnthony Liguori case QED_CLUSTER_ZERO: 11870596be7eSKevin Wolf return qed_aio_write_alloc(acb, len); 1188eabba580SStefan Hajnoczi 1189eabba580SStefan Hajnoczi default: 11900596be7eSKevin Wolf g_assert_not_reached(); 1191eabba580SStefan Hajnoczi } 1192eabba580SStefan Hajnoczi } 1193eabba580SStefan Hajnoczi 1194eabba580SStefan Hajnoczi /** 1195eabba580SStefan Hajnoczi * Read data cluster 1196eabba580SStefan Hajnoczi * 1197eabba580SStefan Hajnoczi * @opaque: Read request 11980596be7eSKevin Wolf * @ret: QED_CLUSTER_FOUND, QED_CLUSTER_L2 or QED_CLUSTER_L1 1199eabba580SStefan Hajnoczi * @offset: Cluster offset in bytes 1200eabba580SStefan Hajnoczi * @len: Length in bytes 1201eabba580SStefan Hajnoczi */ 120287f0d882SKevin Wolf static int coroutine_fn qed_aio_read_data(void *opaque, int ret, 120387f0d882SKevin Wolf uint64_t offset, size_t len) 1204eabba580SStefan Hajnoczi { 1205eabba580SStefan Hajnoczi QEDAIOCB *acb = opaque; 1206eabba580SStefan Hajnoczi BDRVQEDState *s = acb_to_s(acb); 120748cc565eSKevin Wolf BlockDriverState *bs = acb->bs; 1208eabba580SStefan Hajnoczi 1209eabba580SStefan Hajnoczi /* Adjust offset into cluster */ 1210eabba580SStefan Hajnoczi offset += qed_offset_into_cluster(s, acb->cur_pos); 1211eabba580SStefan Hajnoczi 1212eabba580SStefan Hajnoczi trace_qed_aio_read_data(s, acb, ret, offset, len); 1213eabba580SStefan Hajnoczi 12141b093c48SMichael Tokarev qemu_iovec_concat(&acb->cur_qiov, acb->qiov, acb->qiov_offset, len); 1215eabba580SStefan Hajnoczi 121621df65b6SAnthony Liguori /* Handle zero cluster and backing file reads */ 121721df65b6SAnthony Liguori if (ret == QED_CLUSTER_ZERO) { 12183d9b4925SMichael Tokarev qemu_iovec_memset(&acb->cur_qiov, 0, 0, acb->cur_qiov.size); 12190596be7eSKevin Wolf return 0; 122021df65b6SAnthony Liguori } else if (ret != QED_CLUSTER_FOUND) { 12210596be7eSKevin Wolf return qed_read_backing_file(s, acb->cur_pos, &acb->cur_qiov, 1222e85c5281SKevin Wolf &acb->backing_qiov); 1223eabba580SStefan Hajnoczi } 1224eabba580SStefan Hajnoczi 1225eabba580SStefan Hajnoczi BLKDBG_EVENT(bs->file, BLKDBG_READ_AIO); 12260f714ec7SKevin Wolf ret = bdrv_co_preadv(bs->file, offset, acb->cur_qiov.size, 12270f714ec7SKevin Wolf &acb->cur_qiov, 0); 12283e248cdcSKevin Wolf if (ret < 0) { 12290596be7eSKevin Wolf return ret; 12303e248cdcSKevin Wolf } 12310596be7eSKevin Wolf return 0; 1232eabba580SStefan Hajnoczi } 1233eabba580SStefan Hajnoczi 1234eabba580SStefan Hajnoczi /** 1235eabba580SStefan Hajnoczi * Begin next I/O or complete the request 1236eabba580SStefan Hajnoczi */ 123787f0d882SKevin Wolf static int coroutine_fn qed_aio_next_io(QEDAIOCB *acb) 1238eabba580SStefan Hajnoczi { 1239eabba580SStefan Hajnoczi BDRVQEDState *s = acb_to_s(acb); 12400f21b7a1SKevin Wolf uint64_t offset; 12410f21b7a1SKevin Wolf size_t len; 1242dddf8db1SKevin Wolf int ret; 1243eabba580SStefan Hajnoczi 124401859874SKevin Wolf while (1) { 1245dddf8db1SKevin Wolf trace_qed_aio_next_io(s, acb, 0, acb->cur_pos + acb->cur_qiov.size); 1246eabba580SStefan Hajnoczi 1247f06ee3d4SKevin Wolf if (acb->backing_qiov) { 1248f06ee3d4SKevin Wolf qemu_iovec_destroy(acb->backing_qiov); 1249f06ee3d4SKevin Wolf g_free(acb->backing_qiov); 1250f06ee3d4SKevin Wolf acb->backing_qiov = NULL; 1251f06ee3d4SKevin Wolf } 1252f06ee3d4SKevin Wolf 1253eabba580SStefan Hajnoczi acb->qiov_offset += acb->cur_qiov.size; 1254eabba580SStefan Hajnoczi acb->cur_pos += acb->cur_qiov.size; 1255eabba580SStefan Hajnoczi qemu_iovec_reset(&acb->cur_qiov); 1256eabba580SStefan Hajnoczi 1257eabba580SStefan Hajnoczi /* Complete request */ 1258eabba580SStefan Hajnoczi if (acb->cur_pos >= acb->end_pos) { 125948cc565eSKevin Wolf ret = 0; 126048cc565eSKevin Wolf break; 1261eabba580SStefan Hajnoczi } 1262eabba580SStefan Hajnoczi 1263eabba580SStefan Hajnoczi /* Find next cluster and start I/O */ 12640f21b7a1SKevin Wolf len = acb->end_pos - acb->cur_pos; 12650f21b7a1SKevin Wolf ret = qed_find_cluster(s, &acb->request, acb->cur_pos, &len, &offset); 12660596be7eSKevin Wolf if (ret < 0) { 126748cc565eSKevin Wolf break; 12680596be7eSKevin Wolf } 12690596be7eSKevin Wolf 12700596be7eSKevin Wolf if (acb->flags & QED_AIOCB_WRITE) { 12710596be7eSKevin Wolf ret = qed_aio_write_data(acb, ret, offset, len); 12720596be7eSKevin Wolf } else { 12730596be7eSKevin Wolf ret = qed_aio_read_data(acb, ret, offset, len); 12740596be7eSKevin Wolf } 12750596be7eSKevin Wolf 12760806c3b5SKevin Wolf if (ret < 0 && ret != -EAGAIN) { 127748cc565eSKevin Wolf break; 127801859874SKevin Wolf } 1279eabba580SStefan Hajnoczi } 1280eabba580SStefan Hajnoczi 128148cc565eSKevin Wolf trace_qed_aio_complete(s, acb, ret); 128248cc565eSKevin Wolf qed_aio_complete(acb); 128348cc565eSKevin Wolf return ret; 128489f89709SKevin Wolf } 128589f89709SKevin Wolf 128689f89709SKevin Wolf static int coroutine_fn qed_co_request(BlockDriverState *bs, int64_t sector_num, 128789f89709SKevin Wolf QEMUIOVector *qiov, int nb_sectors, 128889f89709SKevin Wolf int flags) 128989f89709SKevin Wolf { 129048cc565eSKevin Wolf QEDAIOCB acb = { 129148cc565eSKevin Wolf .bs = bs, 129248cc565eSKevin Wolf .cur_pos = (uint64_t) sector_num * BDRV_SECTOR_SIZE, 129348cc565eSKevin Wolf .end_pos = (sector_num + nb_sectors) * BDRV_SECTOR_SIZE, 129448cc565eSKevin Wolf .qiov = qiov, 129548cc565eSKevin Wolf .flags = flags, 129689f89709SKevin Wolf }; 129748cc565eSKevin Wolf qemu_iovec_init(&acb.cur_qiov, qiov->niov); 129889f89709SKevin Wolf 129948cc565eSKevin Wolf trace_qed_aio_setup(bs->opaque, &acb, sector_num, nb_sectors, NULL, flags); 1300eabba580SStefan Hajnoczi 1301eabba580SStefan Hajnoczi /* Start request */ 130248cc565eSKevin Wolf return qed_aio_next_io(&acb); 130389f89709SKevin Wolf } 130489f89709SKevin Wolf 130589f89709SKevin Wolf static int coroutine_fn bdrv_qed_co_readv(BlockDriverState *bs, 130689f89709SKevin Wolf int64_t sector_num, int nb_sectors, 130789f89709SKevin Wolf QEMUIOVector *qiov) 130875411d23SStefan Hajnoczi { 130989f89709SKevin Wolf return qed_co_request(bs, sector_num, qiov, nb_sectors, 0); 131075411d23SStefan Hajnoczi } 131175411d23SStefan Hajnoczi 131289f89709SKevin Wolf static int coroutine_fn bdrv_qed_co_writev(BlockDriverState *bs, 131389f89709SKevin Wolf int64_t sector_num, int nb_sectors, 131489f89709SKevin Wolf QEMUIOVector *qiov) 131575411d23SStefan Hajnoczi { 131689f89709SKevin Wolf return qed_co_request(bs, sector_num, qiov, nb_sectors, QED_AIOCB_WRITE); 13170e71be19SStefan Hajnoczi } 13180e71be19SStefan Hajnoczi 131949a2e483SEric Blake static int coroutine_fn bdrv_qed_co_pwrite_zeroes(BlockDriverState *bs, 132049a2e483SEric Blake int64_t offset, 1321f5a5ca79SManos Pitsidianakis int bytes, 1322aa7bfbffSPeter Lieven BdrvRequestFlags flags) 13230e71be19SStefan Hajnoczi { 1324ef72f76eSStefan Hajnoczi BDRVQEDState *s = bs->opaque; 13250e71be19SStefan Hajnoczi QEMUIOVector qiov; 13260e71be19SStefan Hajnoczi struct iovec iov; 13270e71be19SStefan Hajnoczi 132849a2e483SEric Blake /* Fall back if the request is not aligned */ 132949a2e483SEric Blake if (qed_offset_into_cluster(s, offset) || 1330f5a5ca79SManos Pitsidianakis qed_offset_into_cluster(s, bytes)) { 1331ef72f76eSStefan Hajnoczi return -ENOTSUP; 1332ef72f76eSStefan Hajnoczi } 1333ef72f76eSStefan Hajnoczi 13340e71be19SStefan Hajnoczi /* Zero writes start without an I/O buffer. If a buffer becomes necessary 13350e71be19SStefan Hajnoczi * then it will be allocated during request processing. 13360e71be19SStefan Hajnoczi */ 133749a2e483SEric Blake iov.iov_base = NULL; 1338f5a5ca79SManos Pitsidianakis iov.iov_len = bytes; 13390e71be19SStefan Hajnoczi 13400e71be19SStefan Hajnoczi qemu_iovec_init_external(&qiov, &iov, 1); 134189f89709SKevin Wolf return qed_co_request(bs, offset >> BDRV_SECTOR_BITS, &qiov, 1342f5a5ca79SManos Pitsidianakis bytes >> BDRV_SECTOR_BITS, 13430e71be19SStefan Hajnoczi QED_AIOCB_WRITE | QED_AIOCB_ZERO); 13440e71be19SStefan Hajnoczi } 13450e71be19SStefan Hajnoczi 13468243ccb7SMax Reitz static int bdrv_qed_truncate(BlockDriverState *bs, int64_t offset, 13478243ccb7SMax Reitz PreallocMode prealloc, Error **errp) 134875411d23SStefan Hajnoczi { 134977a5a000SStefan Hajnoczi BDRVQEDState *s = bs->opaque; 135077a5a000SStefan Hajnoczi uint64_t old_image_size; 135177a5a000SStefan Hajnoczi int ret; 135277a5a000SStefan Hajnoczi 13538243ccb7SMax Reitz if (prealloc != PREALLOC_MODE_OFF) { 13548243ccb7SMax Reitz error_setg(errp, "Unsupported preallocation mode '%s'", 13558243ccb7SMax Reitz PreallocMode_lookup[prealloc]); 13568243ccb7SMax Reitz return -ENOTSUP; 13578243ccb7SMax Reitz } 13588243ccb7SMax Reitz 135977a5a000SStefan Hajnoczi if (!qed_is_image_size_valid(offset, s->header.cluster_size, 136077a5a000SStefan Hajnoczi s->header.table_size)) { 1361f59adb32SMax Reitz error_setg(errp, "Invalid image size specified"); 136277a5a000SStefan Hajnoczi return -EINVAL; 136377a5a000SStefan Hajnoczi } 136477a5a000SStefan Hajnoczi 136577a5a000SStefan Hajnoczi if ((uint64_t)offset < s->header.image_size) { 1366f59adb32SMax Reitz error_setg(errp, "Shrinking images is currently not supported"); 136775411d23SStefan Hajnoczi return -ENOTSUP; 136875411d23SStefan Hajnoczi } 136975411d23SStefan Hajnoczi 137077a5a000SStefan Hajnoczi old_image_size = s->header.image_size; 137177a5a000SStefan Hajnoczi s->header.image_size = offset; 137277a5a000SStefan Hajnoczi ret = qed_write_header_sync(s); 137377a5a000SStefan Hajnoczi if (ret < 0) { 137477a5a000SStefan Hajnoczi s->header.image_size = old_image_size; 1375f59adb32SMax Reitz error_setg_errno(errp, -ret, "Failed to update the image size"); 137677a5a000SStefan Hajnoczi } 137777a5a000SStefan Hajnoczi return ret; 137877a5a000SStefan Hajnoczi } 137977a5a000SStefan Hajnoczi 138075411d23SStefan Hajnoczi static int64_t bdrv_qed_getlength(BlockDriverState *bs) 138175411d23SStefan Hajnoczi { 138275411d23SStefan Hajnoczi BDRVQEDState *s = bs->opaque; 138375411d23SStefan Hajnoczi return s->header.image_size; 138475411d23SStefan Hajnoczi } 138575411d23SStefan Hajnoczi 138675411d23SStefan Hajnoczi static int bdrv_qed_get_info(BlockDriverState *bs, BlockDriverInfo *bdi) 138775411d23SStefan Hajnoczi { 138875411d23SStefan Hajnoczi BDRVQEDState *s = bs->opaque; 138975411d23SStefan Hajnoczi 139075411d23SStefan Hajnoczi memset(bdi, 0, sizeof(*bdi)); 139175411d23SStefan Hajnoczi bdi->cluster_size = s->header.cluster_size; 1392d68dbee8SDong Xu Wang bdi->is_dirty = s->header.features & QED_F_NEED_CHECK; 139395de6d70SPaolo Bonzini bdi->unallocated_blocks_are_zero = true; 139495de6d70SPaolo Bonzini bdi->can_write_zeroes_with_unmap = true; 139575411d23SStefan Hajnoczi return 0; 139675411d23SStefan Hajnoczi } 139775411d23SStefan Hajnoczi 139875411d23SStefan Hajnoczi static int bdrv_qed_change_backing_file(BlockDriverState *bs, 139975411d23SStefan Hajnoczi const char *backing_file, 140075411d23SStefan Hajnoczi const char *backing_fmt) 140175411d23SStefan Hajnoczi { 140275411d23SStefan Hajnoczi BDRVQEDState *s = bs->opaque; 140375411d23SStefan Hajnoczi QEDHeader new_header, le_header; 140475411d23SStefan Hajnoczi void *buffer; 140575411d23SStefan Hajnoczi size_t buffer_len, backing_file_len; 140675411d23SStefan Hajnoczi int ret; 140775411d23SStefan Hajnoczi 140875411d23SStefan Hajnoczi /* Refuse to set backing filename if unknown compat feature bits are 140975411d23SStefan Hajnoczi * active. If the image uses an unknown compat feature then we may not 141075411d23SStefan Hajnoczi * know the layout of data following the header structure and cannot safely 141175411d23SStefan Hajnoczi * add a new string. 141275411d23SStefan Hajnoczi */ 141375411d23SStefan Hajnoczi if (backing_file && (s->header.compat_features & 141475411d23SStefan Hajnoczi ~QED_COMPAT_FEATURE_MASK)) { 141575411d23SStefan Hajnoczi return -ENOTSUP; 141675411d23SStefan Hajnoczi } 141775411d23SStefan Hajnoczi 141875411d23SStefan Hajnoczi memcpy(&new_header, &s->header, sizeof(new_header)); 141975411d23SStefan Hajnoczi 142075411d23SStefan Hajnoczi new_header.features &= ~(QED_F_BACKING_FILE | 142175411d23SStefan Hajnoczi QED_F_BACKING_FORMAT_NO_PROBE); 142275411d23SStefan Hajnoczi 142375411d23SStefan Hajnoczi /* Adjust feature flags */ 142475411d23SStefan Hajnoczi if (backing_file) { 142575411d23SStefan Hajnoczi new_header.features |= QED_F_BACKING_FILE; 142675411d23SStefan Hajnoczi 142775411d23SStefan Hajnoczi if (qed_fmt_is_raw(backing_fmt)) { 142875411d23SStefan Hajnoczi new_header.features |= QED_F_BACKING_FORMAT_NO_PROBE; 142975411d23SStefan Hajnoczi } 143075411d23SStefan Hajnoczi } 143175411d23SStefan Hajnoczi 143275411d23SStefan Hajnoczi /* Calculate new header size */ 143375411d23SStefan Hajnoczi backing_file_len = 0; 143475411d23SStefan Hajnoczi 143575411d23SStefan Hajnoczi if (backing_file) { 143675411d23SStefan Hajnoczi backing_file_len = strlen(backing_file); 143775411d23SStefan Hajnoczi } 143875411d23SStefan Hajnoczi 143975411d23SStefan Hajnoczi buffer_len = sizeof(new_header); 144075411d23SStefan Hajnoczi new_header.backing_filename_offset = buffer_len; 144175411d23SStefan Hajnoczi new_header.backing_filename_size = backing_file_len; 144275411d23SStefan Hajnoczi buffer_len += backing_file_len; 144375411d23SStefan Hajnoczi 144475411d23SStefan Hajnoczi /* Make sure we can rewrite header without failing */ 144575411d23SStefan Hajnoczi if (buffer_len > new_header.header_size * new_header.cluster_size) { 144675411d23SStefan Hajnoczi return -ENOSPC; 144775411d23SStefan Hajnoczi } 144875411d23SStefan Hajnoczi 144975411d23SStefan Hajnoczi /* Prepare new header */ 14507267c094SAnthony Liguori buffer = g_malloc(buffer_len); 145175411d23SStefan Hajnoczi 145275411d23SStefan Hajnoczi qed_header_cpu_to_le(&new_header, &le_header); 145375411d23SStefan Hajnoczi memcpy(buffer, &le_header, sizeof(le_header)); 145475411d23SStefan Hajnoczi buffer_len = sizeof(le_header); 145575411d23SStefan Hajnoczi 1456feba23b1SPavel Borzenkov if (backing_file) { 145775411d23SStefan Hajnoczi memcpy(buffer + buffer_len, backing_file, backing_file_len); 145875411d23SStefan Hajnoczi buffer_len += backing_file_len; 1459feba23b1SPavel Borzenkov } 146075411d23SStefan Hajnoczi 146175411d23SStefan Hajnoczi /* Write new header */ 1462d9ca2ea2SKevin Wolf ret = bdrv_pwrite_sync(bs->file, 0, buffer, buffer_len); 14637267c094SAnthony Liguori g_free(buffer); 146475411d23SStefan Hajnoczi if (ret == 0) { 146575411d23SStefan Hajnoczi memcpy(&s->header, &new_header, sizeof(new_header)); 146675411d23SStefan Hajnoczi } 146775411d23SStefan Hajnoczi return ret; 146875411d23SStefan Hajnoczi } 146975411d23SStefan Hajnoczi 14705a8a30dbSKevin Wolf static void bdrv_qed_invalidate_cache(BlockDriverState *bs, Error **errp) 1471c82954e5SBenoît Canet { 14725a8a30dbSKevin Wolf Error *local_err = NULL; 14735a8a30dbSKevin Wolf int ret; 1474c82954e5SBenoît Canet 1475c82954e5SBenoît Canet bdrv_qed_close(bs); 14763456a8d1SKevin Wolf 1477*61c7887eSPaolo Bonzini bdrv_qed_init_state(bs); 14784e4bf5c4SKevin Wolf ret = bdrv_qed_do_open(bs, NULL, bs->open_flags, &local_err); 14795a8a30dbSKevin Wolf if (local_err) { 1480e43bfd9cSMarkus Armbruster error_propagate(errp, local_err); 1481e43bfd9cSMarkus Armbruster error_prepend(errp, "Could not reopen qed layer: "); 14825a8a30dbSKevin Wolf return; 14835a8a30dbSKevin Wolf } else if (ret < 0) { 14845a8a30dbSKevin Wolf error_setg_errno(errp, -ret, "Could not reopen qed layer"); 14855a8a30dbSKevin Wolf return; 14865a8a30dbSKevin Wolf } 1487c82954e5SBenoît Canet } 1488c82954e5SBenoît Canet 14894534ff54SKevin Wolf static int bdrv_qed_check(BlockDriverState *bs, BdrvCheckResult *result, 14904534ff54SKevin Wolf BdrvCheckMode fix) 149175411d23SStefan Hajnoczi { 149201979a98SStefan Hajnoczi BDRVQEDState *s = bs->opaque; 149301979a98SStefan Hajnoczi 14944534ff54SKevin Wolf return qed_check(s, result, !!fix); 149575411d23SStefan Hajnoczi } 149675411d23SStefan Hajnoczi 14977ab74849SChunyan Liu static QemuOptsList qed_create_opts = { 14987ab74849SChunyan Liu .name = "qed-create-opts", 14997ab74849SChunyan Liu .head = QTAILQ_HEAD_INITIALIZER(qed_create_opts.head), 15007ab74849SChunyan Liu .desc = { 150175411d23SStefan Hajnoczi { 150275411d23SStefan Hajnoczi .name = BLOCK_OPT_SIZE, 15037ab74849SChunyan Liu .type = QEMU_OPT_SIZE, 15047ab74849SChunyan Liu .help = "Virtual disk size" 15057ab74849SChunyan Liu }, 15067ab74849SChunyan Liu { 150775411d23SStefan Hajnoczi .name = BLOCK_OPT_BACKING_FILE, 15087ab74849SChunyan Liu .type = QEMU_OPT_STRING, 150975411d23SStefan Hajnoczi .help = "File name of a base image" 15107ab74849SChunyan Liu }, 15117ab74849SChunyan Liu { 151275411d23SStefan Hajnoczi .name = BLOCK_OPT_BACKING_FMT, 15137ab74849SChunyan Liu .type = QEMU_OPT_STRING, 151475411d23SStefan Hajnoczi .help = "Image format of the base image" 15157ab74849SChunyan Liu }, 15167ab74849SChunyan Liu { 151775411d23SStefan Hajnoczi .name = BLOCK_OPT_CLUSTER_SIZE, 15187ab74849SChunyan Liu .type = QEMU_OPT_SIZE, 151999cce9faSKevin Wolf .help = "Cluster size (in bytes)", 15207ab74849SChunyan Liu .def_value_str = stringify(QED_DEFAULT_CLUSTER_SIZE) 15217ab74849SChunyan Liu }, 15227ab74849SChunyan Liu { 152375411d23SStefan Hajnoczi .name = BLOCK_OPT_TABLE_SIZE, 15247ab74849SChunyan Liu .type = QEMU_OPT_SIZE, 152575411d23SStefan Hajnoczi .help = "L1/L2 table size (in clusters)" 152675411d23SStefan Hajnoczi }, 152775411d23SStefan Hajnoczi { /* end of list */ } 15287ab74849SChunyan Liu } 152975411d23SStefan Hajnoczi }; 153075411d23SStefan Hajnoczi 153175411d23SStefan Hajnoczi static BlockDriver bdrv_qed = { 153275411d23SStefan Hajnoczi .format_name = "qed", 153375411d23SStefan Hajnoczi .instance_size = sizeof(BDRVQEDState), 15347ab74849SChunyan Liu .create_opts = &qed_create_opts, 15358ee79e70SKevin Wolf .supports_backing = true, 153675411d23SStefan Hajnoczi 153775411d23SStefan Hajnoczi .bdrv_probe = bdrv_qed_probe, 153875411d23SStefan Hajnoczi .bdrv_open = bdrv_qed_open, 153975411d23SStefan Hajnoczi .bdrv_close = bdrv_qed_close, 1540f9cb20f1SJeff Cody .bdrv_reopen_prepare = bdrv_qed_reopen_prepare, 1541862f215fSKevin Wolf .bdrv_child_perm = bdrv_format_default_perms, 1542c282e1fdSChunyan Liu .bdrv_create = bdrv_qed_create, 15433ac21627SPeter Lieven .bdrv_has_zero_init = bdrv_has_zero_init_1, 1544b6b8a333SPaolo Bonzini .bdrv_co_get_block_status = bdrv_qed_co_get_block_status, 154589f89709SKevin Wolf .bdrv_co_readv = bdrv_qed_co_readv, 154689f89709SKevin Wolf .bdrv_co_writev = bdrv_qed_co_writev, 154749a2e483SEric Blake .bdrv_co_pwrite_zeroes = bdrv_qed_co_pwrite_zeroes, 154875411d23SStefan Hajnoczi .bdrv_truncate = bdrv_qed_truncate, 154975411d23SStefan Hajnoczi .bdrv_getlength = bdrv_qed_getlength, 155075411d23SStefan Hajnoczi .bdrv_get_info = bdrv_qed_get_info, 1551d34682cdSKevin Wolf .bdrv_refresh_limits = bdrv_qed_refresh_limits, 155275411d23SStefan Hajnoczi .bdrv_change_backing_file = bdrv_qed_change_backing_file, 1553c82954e5SBenoît Canet .bdrv_invalidate_cache = bdrv_qed_invalidate_cache, 155475411d23SStefan Hajnoczi .bdrv_check = bdrv_qed_check, 1555a8c868c3SStefan Hajnoczi .bdrv_detach_aio_context = bdrv_qed_detach_aio_context, 1556a8c868c3SStefan Hajnoczi .bdrv_attach_aio_context = bdrv_qed_attach_aio_context, 155761124f03SPaolo Bonzini .bdrv_co_drain = bdrv_qed_co_drain, 155875411d23SStefan Hajnoczi }; 155975411d23SStefan Hajnoczi 156075411d23SStefan Hajnoczi static void bdrv_qed_init(void) 156175411d23SStefan Hajnoczi { 156275411d23SStefan Hajnoczi bdrv_register(&bdrv_qed); 156375411d23SStefan Hajnoczi } 156475411d23SStefan Hajnoczi 156575411d23SStefan Hajnoczi block_init(bdrv_qed_init); 1566