xref: /openbmc/qemu/block/qed.c (revision 365fed5111b06d31c1632af63c7528dfe49d62a2)
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"
16609f45eaSMax Reitz #include "block/qdict.h"
17da34e65cSMarkus Armbruster #include "qapi/error.h"
181de7afc9SPaolo Bonzini #include "qemu/timer.h"
1958369e22SPaolo Bonzini #include "qemu/bswap.h"
20db725815SMarkus Armbruster #include "qemu/main-loop.h"
210b8fa32fSMarkus Armbruster #include "qemu/module.h"
22922a01a0SMarkus Armbruster #include "qemu/option.h"
23eabba580SStefan Hajnoczi #include "trace.h"
2475411d23SStefan Hajnoczi #include "qed.h"
258a56fdadSKevin Wolf #include "sysemu/block-backend.h"
26959355a4SKevin Wolf #include "qapi/qmp/qdict.h"
27959355a4SKevin Wolf #include "qapi/qobject-input-visitor.h"
28959355a4SKevin Wolf #include "qapi/qapi-visit-block-core.h"
29959355a4SKevin Wolf 
30959355a4SKevin Wolf static QemuOptsList qed_create_opts;
3175411d23SStefan Hajnoczi 
3275411d23SStefan Hajnoczi static int bdrv_qed_probe(const uint8_t *buf, int buf_size,
3375411d23SStefan Hajnoczi                           const char *filename)
3475411d23SStefan Hajnoczi {
3575411d23SStefan Hajnoczi     const QEDHeader *header = (const QEDHeader *)buf;
3675411d23SStefan Hajnoczi 
3775411d23SStefan Hajnoczi     if (buf_size < sizeof(*header)) {
3875411d23SStefan Hajnoczi         return 0;
3975411d23SStefan Hajnoczi     }
4075411d23SStefan Hajnoczi     if (le32_to_cpu(header->magic) != QED_MAGIC) {
4175411d23SStefan Hajnoczi         return 0;
4275411d23SStefan Hajnoczi     }
4375411d23SStefan Hajnoczi     return 100;
4475411d23SStefan Hajnoczi }
4575411d23SStefan Hajnoczi 
4675411d23SStefan Hajnoczi /**
4775411d23SStefan Hajnoczi  * Check whether an image format is raw
4875411d23SStefan Hajnoczi  *
4975411d23SStefan Hajnoczi  * @fmt:    Backing file format, may be NULL
5075411d23SStefan Hajnoczi  */
5175411d23SStefan Hajnoczi static bool qed_fmt_is_raw(const char *fmt)
5275411d23SStefan Hajnoczi {
5375411d23SStefan Hajnoczi     return fmt && strcmp(fmt, "raw") == 0;
5475411d23SStefan Hajnoczi }
5575411d23SStefan Hajnoczi 
5675411d23SStefan Hajnoczi static void qed_header_le_to_cpu(const QEDHeader *le, QEDHeader *cpu)
5775411d23SStefan Hajnoczi {
5875411d23SStefan Hajnoczi     cpu->magic = le32_to_cpu(le->magic);
5975411d23SStefan Hajnoczi     cpu->cluster_size = le32_to_cpu(le->cluster_size);
6075411d23SStefan Hajnoczi     cpu->table_size = le32_to_cpu(le->table_size);
6175411d23SStefan Hajnoczi     cpu->header_size = le32_to_cpu(le->header_size);
6275411d23SStefan Hajnoczi     cpu->features = le64_to_cpu(le->features);
6375411d23SStefan Hajnoczi     cpu->compat_features = le64_to_cpu(le->compat_features);
6475411d23SStefan Hajnoczi     cpu->autoclear_features = le64_to_cpu(le->autoclear_features);
6575411d23SStefan Hajnoczi     cpu->l1_table_offset = le64_to_cpu(le->l1_table_offset);
6675411d23SStefan Hajnoczi     cpu->image_size = le64_to_cpu(le->image_size);
6775411d23SStefan Hajnoczi     cpu->backing_filename_offset = le32_to_cpu(le->backing_filename_offset);
6875411d23SStefan Hajnoczi     cpu->backing_filename_size = le32_to_cpu(le->backing_filename_size);
6975411d23SStefan Hajnoczi }
7075411d23SStefan Hajnoczi 
7175411d23SStefan Hajnoczi static void qed_header_cpu_to_le(const QEDHeader *cpu, QEDHeader *le)
7275411d23SStefan Hajnoczi {
7375411d23SStefan Hajnoczi     le->magic = cpu_to_le32(cpu->magic);
7475411d23SStefan Hajnoczi     le->cluster_size = cpu_to_le32(cpu->cluster_size);
7575411d23SStefan Hajnoczi     le->table_size = cpu_to_le32(cpu->table_size);
7675411d23SStefan Hajnoczi     le->header_size = cpu_to_le32(cpu->header_size);
7775411d23SStefan Hajnoczi     le->features = cpu_to_le64(cpu->features);
7875411d23SStefan Hajnoczi     le->compat_features = cpu_to_le64(cpu->compat_features);
7975411d23SStefan Hajnoczi     le->autoclear_features = cpu_to_le64(cpu->autoclear_features);
8075411d23SStefan Hajnoczi     le->l1_table_offset = cpu_to_le64(cpu->l1_table_offset);
8175411d23SStefan Hajnoczi     le->image_size = cpu_to_le64(cpu->image_size);
8275411d23SStefan Hajnoczi     le->backing_filename_offset = cpu_to_le32(cpu->backing_filename_offset);
8375411d23SStefan Hajnoczi     le->backing_filename_size = cpu_to_le32(cpu->backing_filename_size);
8475411d23SStefan Hajnoczi }
8575411d23SStefan Hajnoczi 
86b10170acSStefan Hajnoczi int qed_write_header_sync(BDRVQEDState *s)
8775411d23SStefan Hajnoczi {
8875411d23SStefan Hajnoczi     QEDHeader le;
8975411d23SStefan Hajnoczi     int ret;
9075411d23SStefan Hajnoczi 
9175411d23SStefan Hajnoczi     qed_header_cpu_to_le(&s->header, &le);
92d9ca2ea2SKevin Wolf     ret = bdrv_pwrite(s->bs->file, 0, &le, sizeof(le));
9375411d23SStefan Hajnoczi     if (ret != sizeof(le)) {
9475411d23SStefan Hajnoczi         return ret;
9575411d23SStefan Hajnoczi     }
9675411d23SStefan Hajnoczi     return 0;
9775411d23SStefan Hajnoczi }
9875411d23SStefan Hajnoczi 
9901979a98SStefan Hajnoczi /**
10001979a98SStefan Hajnoczi  * Update header in-place (does not rewrite backing filename or other strings)
10101979a98SStefan Hajnoczi  *
10201979a98SStefan Hajnoczi  * This function only updates known header fields in-place and does not affect
10301979a98SStefan Hajnoczi  * extra data after the QED header.
1041f01e50bSPaolo Bonzini  *
1051f01e50bSPaolo Bonzini  * No new allocating reqs can start while this function runs.
10601979a98SStefan Hajnoczi  */
10787f0d882SKevin Wolf static int coroutine_fn qed_write_header(BDRVQEDState *s)
10801979a98SStefan Hajnoczi {
10901979a98SStefan Hajnoczi     /* We must write full sectors for O_DIRECT but cannot necessarily generate
11001979a98SStefan Hajnoczi      * the data following the header if an unrecognized compat feature is
11101979a98SStefan Hajnoczi      * active.  Therefore, first read the sectors containing the header, update
11201979a98SStefan Hajnoczi      * them, and write back.
11301979a98SStefan Hajnoczi      */
11401979a98SStefan Hajnoczi 
115c41a73ffSLaurent Vivier     int nsectors = DIV_ROUND_UP(sizeof(QEDHeader), BDRV_SECTOR_SIZE);
11601979a98SStefan Hajnoczi     size_t len = nsectors * BDRV_SECTOR_SIZE;
1177076309aSKevin Wolf     uint8_t *buf;
1187076309aSKevin Wolf     int ret;
11901979a98SStefan Hajnoczi 
1201f01e50bSPaolo Bonzini     assert(s->allocating_acb || s->allocating_write_reqs_plugged);
1211f01e50bSPaolo Bonzini 
1227076309aSKevin Wolf     buf = qemu_blockalign(s->bs, len);
12301979a98SStefan Hajnoczi 
124696e8cb2SVladimir Sementsov-Ogievskiy     ret = bdrv_co_pread(s->bs->file, 0, len, buf, 0);
1257076309aSKevin Wolf     if (ret < 0) {
1267076309aSKevin Wolf         goto out;
1277076309aSKevin Wolf     }
1287076309aSKevin Wolf 
1297076309aSKevin Wolf     /* Update header */
1307076309aSKevin Wolf     qed_header_cpu_to_le(&s->header, (QEDHeader *) buf);
1317076309aSKevin Wolf 
132696e8cb2SVladimir Sementsov-Ogievskiy     ret = bdrv_co_pwrite(s->bs->file, 0, len,  buf, 0);
1337076309aSKevin Wolf     if (ret < 0) {
1347076309aSKevin Wolf         goto out;
1357076309aSKevin Wolf     }
1367076309aSKevin Wolf 
1377076309aSKevin Wolf     ret = 0;
1387076309aSKevin Wolf out:
1397076309aSKevin Wolf     qemu_vfree(buf);
140f13d712bSKevin Wolf     return ret;
14101979a98SStefan Hajnoczi }
14201979a98SStefan Hajnoczi 
14375411d23SStefan Hajnoczi static uint64_t qed_max_image_size(uint32_t cluster_size, uint32_t table_size)
14475411d23SStefan Hajnoczi {
14575411d23SStefan Hajnoczi     uint64_t table_entries;
14675411d23SStefan Hajnoczi     uint64_t l2_size;
14775411d23SStefan Hajnoczi 
14875411d23SStefan Hajnoczi     table_entries = (table_size * cluster_size) / sizeof(uint64_t);
14975411d23SStefan Hajnoczi     l2_size = table_entries * cluster_size;
15075411d23SStefan Hajnoczi 
15175411d23SStefan Hajnoczi     return l2_size * table_entries;
15275411d23SStefan Hajnoczi }
15375411d23SStefan Hajnoczi 
15475411d23SStefan Hajnoczi static bool qed_is_cluster_size_valid(uint32_t cluster_size)
15575411d23SStefan Hajnoczi {
15675411d23SStefan Hajnoczi     if (cluster_size < QED_MIN_CLUSTER_SIZE ||
15775411d23SStefan Hajnoczi         cluster_size > QED_MAX_CLUSTER_SIZE) {
15875411d23SStefan Hajnoczi         return false;
15975411d23SStefan Hajnoczi     }
16075411d23SStefan Hajnoczi     if (cluster_size & (cluster_size - 1)) {
16175411d23SStefan Hajnoczi         return false; /* not power of 2 */
16275411d23SStefan Hajnoczi     }
16375411d23SStefan Hajnoczi     return true;
16475411d23SStefan Hajnoczi }
16575411d23SStefan Hajnoczi 
16675411d23SStefan Hajnoczi static bool qed_is_table_size_valid(uint32_t table_size)
16775411d23SStefan Hajnoczi {
16875411d23SStefan Hajnoczi     if (table_size < QED_MIN_TABLE_SIZE ||
16975411d23SStefan Hajnoczi         table_size > QED_MAX_TABLE_SIZE) {
17075411d23SStefan Hajnoczi         return false;
17175411d23SStefan Hajnoczi     }
17275411d23SStefan Hajnoczi     if (table_size & (table_size - 1)) {
17375411d23SStefan Hajnoczi         return false; /* not power of 2 */
17475411d23SStefan Hajnoczi     }
17575411d23SStefan Hajnoczi     return true;
17675411d23SStefan Hajnoczi }
17775411d23SStefan Hajnoczi 
17875411d23SStefan Hajnoczi static bool qed_is_image_size_valid(uint64_t image_size, uint32_t cluster_size,
17975411d23SStefan Hajnoczi                                     uint32_t table_size)
18075411d23SStefan Hajnoczi {
18175411d23SStefan Hajnoczi     if (image_size % BDRV_SECTOR_SIZE != 0) {
18275411d23SStefan Hajnoczi         return false; /* not multiple of sector size */
18375411d23SStefan Hajnoczi     }
18475411d23SStefan Hajnoczi     if (image_size > qed_max_image_size(cluster_size, table_size)) {
18575411d23SStefan Hajnoczi         return false; /* image is too large */
18675411d23SStefan Hajnoczi     }
18775411d23SStefan Hajnoczi     return true;
18875411d23SStefan Hajnoczi }
18975411d23SStefan Hajnoczi 
19075411d23SStefan Hajnoczi /**
19175411d23SStefan Hajnoczi  * Read a string of known length from the image file
19275411d23SStefan Hajnoczi  *
19375411d23SStefan Hajnoczi  * @file:       Image file
19475411d23SStefan Hajnoczi  * @offset:     File offset to start of string, in bytes
19575411d23SStefan Hajnoczi  * @n:          String length in bytes
19675411d23SStefan Hajnoczi  * @buf:        Destination buffer
19775411d23SStefan Hajnoczi  * @buflen:     Destination buffer length in bytes
19875411d23SStefan Hajnoczi  * @ret:        0 on success, -errno on failure
19975411d23SStefan Hajnoczi  *
20075411d23SStefan Hajnoczi  * The string is NUL-terminated.
20175411d23SStefan Hajnoczi  */
202cf2ab8fcSKevin Wolf static int qed_read_string(BdrvChild *file, uint64_t offset, size_t n,
20375411d23SStefan Hajnoczi                            char *buf, size_t buflen)
20475411d23SStefan Hajnoczi {
20575411d23SStefan Hajnoczi     int ret;
20675411d23SStefan Hajnoczi     if (n >= buflen) {
20775411d23SStefan Hajnoczi         return -EINVAL;
20875411d23SStefan Hajnoczi     }
20975411d23SStefan Hajnoczi     ret = bdrv_pread(file, offset, buf, n);
21075411d23SStefan Hajnoczi     if (ret < 0) {
21175411d23SStefan Hajnoczi         return ret;
21275411d23SStefan Hajnoczi     }
21375411d23SStefan Hajnoczi     buf[n] = '\0';
21475411d23SStefan Hajnoczi     return 0;
21575411d23SStefan Hajnoczi }
21675411d23SStefan Hajnoczi 
217eabba580SStefan Hajnoczi /**
218eabba580SStefan Hajnoczi  * Allocate new clusters
219eabba580SStefan Hajnoczi  *
220eabba580SStefan Hajnoczi  * @s:          QED state
221eabba580SStefan Hajnoczi  * @n:          Number of contiguous clusters to allocate
222eabba580SStefan Hajnoczi  * @ret:        Offset of first allocated cluster
223eabba580SStefan Hajnoczi  *
224eabba580SStefan Hajnoczi  * This function only produces the offset where the new clusters should be
225eabba580SStefan Hajnoczi  * written.  It updates BDRVQEDState but does not make any changes to the image
226eabba580SStefan Hajnoczi  * file.
2271f01e50bSPaolo Bonzini  *
2281f01e50bSPaolo Bonzini  * Called with table_lock held.
229eabba580SStefan Hajnoczi  */
230eabba580SStefan Hajnoczi static uint64_t qed_alloc_clusters(BDRVQEDState *s, unsigned int n)
231eabba580SStefan Hajnoczi {
232eabba580SStefan Hajnoczi     uint64_t offset = s->file_size;
233eabba580SStefan Hajnoczi     s->file_size += n * s->header.cluster_size;
234eabba580SStefan Hajnoczi     return offset;
235eabba580SStefan Hajnoczi }
236eabba580SStefan Hajnoczi 
237298800caSStefan Hajnoczi QEDTable *qed_alloc_table(BDRVQEDState *s)
238298800caSStefan Hajnoczi {
239298800caSStefan Hajnoczi     /* Honor O_DIRECT memory alignment requirements */
240298800caSStefan Hajnoczi     return qemu_blockalign(s->bs,
241298800caSStefan Hajnoczi                            s->header.cluster_size * s->header.table_size);
242298800caSStefan Hajnoczi }
243298800caSStefan Hajnoczi 
244eabba580SStefan Hajnoczi /**
245eabba580SStefan Hajnoczi  * Allocate a new zeroed L2 table
2461f01e50bSPaolo Bonzini  *
2471f01e50bSPaolo Bonzini  * Called with table_lock held.
248eabba580SStefan Hajnoczi  */
249eabba580SStefan Hajnoczi static CachedL2Table *qed_new_l2_table(BDRVQEDState *s)
250eabba580SStefan Hajnoczi {
251eabba580SStefan Hajnoczi     CachedL2Table *l2_table = qed_alloc_l2_cache_entry(&s->l2_cache);
252eabba580SStefan Hajnoczi 
253eabba580SStefan Hajnoczi     l2_table->table = qed_alloc_table(s);
254eabba580SStefan Hajnoczi     l2_table->offset = qed_alloc_clusters(s, s->header.table_size);
255eabba580SStefan Hajnoczi 
256eabba580SStefan Hajnoczi     memset(l2_table->table->offsets, 0,
257eabba580SStefan Hajnoczi            s->header.cluster_size * s->header.table_size);
258eabba580SStefan Hajnoczi     return l2_table;
259eabba580SStefan Hajnoczi }
260eabba580SStefan Hajnoczi 
2611f01e50bSPaolo Bonzini static bool qed_plug_allocating_write_reqs(BDRVQEDState *s)
2626f321e93SStefan Hajnoczi {
2631f01e50bSPaolo Bonzini     qemu_co_mutex_lock(&s->table_lock);
2641f01e50bSPaolo Bonzini 
2651f01e50bSPaolo Bonzini     /* No reentrancy is allowed.  */
2666f321e93SStefan Hajnoczi     assert(!s->allocating_write_reqs_plugged);
2671f01e50bSPaolo Bonzini     if (s->allocating_acb != NULL) {
2681f01e50bSPaolo Bonzini         /* Another allocating write came concurrently.  This cannot happen
269f8ea8dacSManos Pitsidianakis          * from bdrv_qed_co_drain_begin, but it can happen when the timer runs.
2701f01e50bSPaolo Bonzini          */
2711f01e50bSPaolo Bonzini         qemu_co_mutex_unlock(&s->table_lock);
2721f01e50bSPaolo Bonzini         return false;
2731f01e50bSPaolo Bonzini     }
2746f321e93SStefan Hajnoczi 
2756f321e93SStefan Hajnoczi     s->allocating_write_reqs_plugged = true;
2761f01e50bSPaolo Bonzini     qemu_co_mutex_unlock(&s->table_lock);
2771f01e50bSPaolo Bonzini     return true;
2786f321e93SStefan Hajnoczi }
2796f321e93SStefan Hajnoczi 
2806f321e93SStefan Hajnoczi static void qed_unplug_allocating_write_reqs(BDRVQEDState *s)
2816f321e93SStefan Hajnoczi {
2821f01e50bSPaolo Bonzini     qemu_co_mutex_lock(&s->table_lock);
2836f321e93SStefan Hajnoczi     assert(s->allocating_write_reqs_plugged);
2846f321e93SStefan Hajnoczi     s->allocating_write_reqs_plugged = false;
2851f01e50bSPaolo Bonzini     qemu_co_queue_next(&s->allocating_write_reqs);
2861f01e50bSPaolo Bonzini     qemu_co_mutex_unlock(&s->table_lock);
2876f321e93SStefan Hajnoczi }
2886f321e93SStefan Hajnoczi 
28987f0d882SKevin Wolf static void coroutine_fn qed_need_check_timer_entry(void *opaque)
2906f321e93SStefan Hajnoczi {
2916f321e93SStefan Hajnoczi     BDRVQEDState *s = opaque;
292c0e8f989SKevin Wolf     int ret;
2936f321e93SStefan Hajnoczi 
2946f321e93SStefan Hajnoczi     trace_qed_need_check_timer_cb(s);
2956f321e93SStefan Hajnoczi 
2961f01e50bSPaolo Bonzini     if (!qed_plug_allocating_write_reqs(s)) {
2971f01e50bSPaolo Bonzini         return;
2981f01e50bSPaolo Bonzini     }
2996f321e93SStefan Hajnoczi 
3006f321e93SStefan Hajnoczi     /* Ensure writes are on disk before clearing flag */
301c0e8f989SKevin Wolf     ret = bdrv_co_flush(s->bs->file->bs);
302c0e8f989SKevin Wolf     if (ret < 0) {
303c0e8f989SKevin Wolf         qed_unplug_allocating_write_reqs(s);
304c0e8f989SKevin Wolf         return;
305c0e8f989SKevin Wolf     }
306c0e8f989SKevin Wolf 
307c0e8f989SKevin Wolf     s->header.features &= ~QED_F_NEED_CHECK;
308c0e8f989SKevin Wolf     ret = qed_write_header(s);
309c0e8f989SKevin Wolf     (void) ret;
310c0e8f989SKevin Wolf 
311c0e8f989SKevin Wolf     qed_unplug_allocating_write_reqs(s);
312c0e8f989SKevin Wolf 
313c0e8f989SKevin Wolf     ret = bdrv_co_flush(s->bs);
314c0e8f989SKevin Wolf     (void) ret;
315c0e8f989SKevin Wolf }
316c0e8f989SKevin Wolf 
317c0e8f989SKevin Wolf static void qed_need_check_timer_cb(void *opaque)
318c0e8f989SKevin Wolf {
319c0e8f989SKevin Wolf     Coroutine *co = qemu_coroutine_create(qed_need_check_timer_entry, opaque);
320c0e8f989SKevin Wolf     qemu_coroutine_enter(co);
3212f47da5fSPaolo Bonzini }
3222f47da5fSPaolo Bonzini 
3236f321e93SStefan Hajnoczi static void qed_start_need_check_timer(BDRVQEDState *s)
3246f321e93SStefan Hajnoczi {
3256f321e93SStefan Hajnoczi     trace_qed_start_need_check_timer(s);
3266f321e93SStefan Hajnoczi 
327bc72ad67SAlex Bligh     /* Use QEMU_CLOCK_VIRTUAL so we don't alter the image file while suspended for
3286f321e93SStefan Hajnoczi      * migration.
3296f321e93SStefan Hajnoczi      */
330bc72ad67SAlex Bligh     timer_mod(s->need_check_timer, qemu_clock_get_ns(QEMU_CLOCK_VIRTUAL) +
33173bcb24dSRutuja Shah                    NANOSECONDS_PER_SECOND * QED_NEED_CHECK_TIMEOUT);
3326f321e93SStefan Hajnoczi }
3336f321e93SStefan Hajnoczi 
3346f321e93SStefan Hajnoczi /* It's okay to call this multiple times or when no timer is started */
3356f321e93SStefan Hajnoczi static void qed_cancel_need_check_timer(BDRVQEDState *s)
3366f321e93SStefan Hajnoczi {
3376f321e93SStefan Hajnoczi     trace_qed_cancel_need_check_timer(s);
338bc72ad67SAlex Bligh     timer_del(s->need_check_timer);
3396f321e93SStefan Hajnoczi }
3406f321e93SStefan Hajnoczi 
341a8c868c3SStefan Hajnoczi static void bdrv_qed_detach_aio_context(BlockDriverState *bs)
342a8c868c3SStefan Hajnoczi {
343a8c868c3SStefan Hajnoczi     BDRVQEDState *s = bs->opaque;
344a8c868c3SStefan Hajnoczi 
345a8c868c3SStefan Hajnoczi     qed_cancel_need_check_timer(s);
346a8c868c3SStefan Hajnoczi     timer_free(s->need_check_timer);
347a8c868c3SStefan Hajnoczi }
348a8c868c3SStefan Hajnoczi 
349a8c868c3SStefan Hajnoczi static void bdrv_qed_attach_aio_context(BlockDriverState *bs,
350a8c868c3SStefan Hajnoczi                                         AioContext *new_context)
351a8c868c3SStefan Hajnoczi {
352a8c868c3SStefan Hajnoczi     BDRVQEDState *s = bs->opaque;
353a8c868c3SStefan Hajnoczi 
354a8c868c3SStefan Hajnoczi     s->need_check_timer = aio_timer_new(new_context,
355a8c868c3SStefan Hajnoczi                                         QEMU_CLOCK_VIRTUAL, SCALE_NS,
356a8c868c3SStefan Hajnoczi                                         qed_need_check_timer_cb, s);
357a8c868c3SStefan Hajnoczi     if (s->header.features & QED_F_NEED_CHECK) {
358a8c868c3SStefan Hajnoczi         qed_start_need_check_timer(s);
359a8c868c3SStefan Hajnoczi     }
360a8c868c3SStefan Hajnoczi }
361a8c868c3SStefan Hajnoczi 
362f8ea8dacSManos Pitsidianakis static void coroutine_fn bdrv_qed_co_drain_begin(BlockDriverState *bs)
3636653a73dSFam Zheng {
3646653a73dSFam Zheng     BDRVQEDState *s = bs->opaque;
3656653a73dSFam Zheng 
3666653a73dSFam Zheng     /* Fire the timer immediately in order to start doing I/O as soon as the
3676653a73dSFam Zheng      * header is flushed.
3686653a73dSFam Zheng      */
3696653a73dSFam Zheng     if (s->need_check_timer && timer_pending(s->need_check_timer)) {
3706653a73dSFam Zheng         qed_cancel_need_check_timer(s);
37161124f03SPaolo Bonzini         qed_need_check_timer_entry(s);
3726653a73dSFam Zheng     }
3736653a73dSFam Zheng }
3746653a73dSFam Zheng 
37561c7887eSPaolo Bonzini static void bdrv_qed_init_state(BlockDriverState *bs)
37661c7887eSPaolo Bonzini {
37761c7887eSPaolo Bonzini     BDRVQEDState *s = bs->opaque;
37861c7887eSPaolo Bonzini 
37961c7887eSPaolo Bonzini     memset(s, 0, sizeof(BDRVQEDState));
38061c7887eSPaolo Bonzini     s->bs = bs;
3811f01e50bSPaolo Bonzini     qemu_co_mutex_init(&s->table_lock);
38261c7887eSPaolo Bonzini     qemu_co_queue_init(&s->allocating_write_reqs);
38361c7887eSPaolo Bonzini }
38461c7887eSPaolo Bonzini 
3859fb4dfc5SPaolo Bonzini /* Called with table_lock held.  */
3869fb4dfc5SPaolo Bonzini static int coroutine_fn bdrv_qed_do_open(BlockDriverState *bs, QDict *options,
3879fb4dfc5SPaolo Bonzini                                          int flags, Error **errp)
38875411d23SStefan Hajnoczi {
38975411d23SStefan Hajnoczi     BDRVQEDState *s = bs->opaque;
39075411d23SStefan Hajnoczi     QEDHeader le_header;
39175411d23SStefan Hajnoczi     int64_t file_size;
39275411d23SStefan Hajnoczi     int ret;
39375411d23SStefan Hajnoczi 
394cf2ab8fcSKevin Wolf     ret = bdrv_pread(bs->file, 0, &le_header, sizeof(le_header));
39575411d23SStefan Hajnoczi     if (ret < 0) {
39675411d23SStefan Hajnoczi         return ret;
39775411d23SStefan Hajnoczi     }
39875411d23SStefan Hajnoczi     qed_header_le_to_cpu(&le_header, &s->header);
39975411d23SStefan Hajnoczi 
40075411d23SStefan Hajnoczi     if (s->header.magic != QED_MAGIC) {
40176abe407SPaolo Bonzini         error_setg(errp, "Image not in QED format");
40276abe407SPaolo Bonzini         return -EINVAL;
40375411d23SStefan Hajnoczi     }
40475411d23SStefan Hajnoczi     if (s->header.features & ~QED_FEATURE_MASK) {
40510b758e8SKevin Wolf         /* image uses unsupported feature bits */
406a55448b3SMax Reitz         error_setg(errp, "Unsupported QED features: %" PRIx64,
40710b758e8SKevin Wolf                    s->header.features & ~QED_FEATURE_MASK);
40810b758e8SKevin Wolf         return -ENOTSUP;
40975411d23SStefan Hajnoczi     }
41075411d23SStefan Hajnoczi     if (!qed_is_cluster_size_valid(s->header.cluster_size)) {
41175411d23SStefan Hajnoczi         return -EINVAL;
41275411d23SStefan Hajnoczi     }
41375411d23SStefan Hajnoczi 
41475411d23SStefan Hajnoczi     /* Round down file size to the last cluster */
4159a4f4c31SKevin Wolf     file_size = bdrv_getlength(bs->file->bs);
41675411d23SStefan Hajnoczi     if (file_size < 0) {
41775411d23SStefan Hajnoczi         return file_size;
41875411d23SStefan Hajnoczi     }
41975411d23SStefan Hajnoczi     s->file_size = qed_start_of_cluster(s, file_size);
42075411d23SStefan Hajnoczi 
42175411d23SStefan Hajnoczi     if (!qed_is_table_size_valid(s->header.table_size)) {
42275411d23SStefan Hajnoczi         return -EINVAL;
42375411d23SStefan Hajnoczi     }
42475411d23SStefan Hajnoczi     if (!qed_is_image_size_valid(s->header.image_size,
42575411d23SStefan Hajnoczi                                  s->header.cluster_size,
42675411d23SStefan Hajnoczi                                  s->header.table_size)) {
42775411d23SStefan Hajnoczi         return -EINVAL;
42875411d23SStefan Hajnoczi     }
42975411d23SStefan Hajnoczi     if (!qed_check_table_offset(s, s->header.l1_table_offset)) {
43075411d23SStefan Hajnoczi         return -EINVAL;
43175411d23SStefan Hajnoczi     }
43275411d23SStefan Hajnoczi 
43375411d23SStefan Hajnoczi     s->table_nelems = (s->header.cluster_size * s->header.table_size) /
43475411d23SStefan Hajnoczi                       sizeof(uint64_t);
435786a4ea8SStefan Hajnoczi     s->l2_shift = ctz32(s->header.cluster_size);
43675411d23SStefan Hajnoczi     s->l2_mask = s->table_nelems - 1;
437786a4ea8SStefan Hajnoczi     s->l1_shift = s->l2_shift + ctz32(s->table_nelems);
43875411d23SStefan Hajnoczi 
4390adfa1edSStefan Hajnoczi     /* Header size calculation must not overflow uint32_t */
4400adfa1edSStefan Hajnoczi     if (s->header.header_size > UINT32_MAX / s->header.cluster_size) {
4410adfa1edSStefan Hajnoczi         return -EINVAL;
4420adfa1edSStefan Hajnoczi     }
4430adfa1edSStefan Hajnoczi 
44475411d23SStefan Hajnoczi     if ((s->header.features & QED_F_BACKING_FILE)) {
44575411d23SStefan Hajnoczi         if ((uint64_t)s->header.backing_filename_offset +
44675411d23SStefan Hajnoczi             s->header.backing_filename_size >
44775411d23SStefan Hajnoczi             s->header.cluster_size * s->header.header_size) {
44875411d23SStefan Hajnoczi             return -EINVAL;
44975411d23SStefan Hajnoczi         }
45075411d23SStefan Hajnoczi 
451cf2ab8fcSKevin Wolf         ret = qed_read_string(bs->file, s->header.backing_filename_offset,
452998c2019SMax Reitz                               s->header.backing_filename_size,
453998c2019SMax Reitz                               bs->auto_backing_file,
454998c2019SMax Reitz                               sizeof(bs->auto_backing_file));
45575411d23SStefan Hajnoczi         if (ret < 0) {
45675411d23SStefan Hajnoczi             return ret;
45775411d23SStefan Hajnoczi         }
458998c2019SMax Reitz         pstrcpy(bs->backing_file, sizeof(bs->backing_file),
459998c2019SMax Reitz                 bs->auto_backing_file);
46075411d23SStefan Hajnoczi 
46175411d23SStefan Hajnoczi         if (s->header.features & QED_F_BACKING_FORMAT_NO_PROBE) {
46275411d23SStefan Hajnoczi             pstrcpy(bs->backing_format, sizeof(bs->backing_format), "raw");
46375411d23SStefan Hajnoczi         }
46475411d23SStefan Hajnoczi     }
46575411d23SStefan Hajnoczi 
46675411d23SStefan Hajnoczi     /* Reset unknown autoclear feature bits.  This is a backwards
46775411d23SStefan Hajnoczi      * compatibility mechanism that allows images to be opened by older
46875411d23SStefan Hajnoczi      * programs, which "knock out" unknown feature bits.  When an image is
46975411d23SStefan Hajnoczi      * opened by a newer program again it can detect that the autoclear
47075411d23SStefan Hajnoczi      * feature is no longer valid.
47175411d23SStefan Hajnoczi      */
47275411d23SStefan Hajnoczi     if ((s->header.autoclear_features & ~QED_AUTOCLEAR_FEATURE_MASK) != 0 &&
47304c01a5cSKevin Wolf         !bdrv_is_read_only(bs->file->bs) && !(flags & BDRV_O_INACTIVE)) {
47475411d23SStefan Hajnoczi         s->header.autoclear_features &= QED_AUTOCLEAR_FEATURE_MASK;
47575411d23SStefan Hajnoczi 
47675411d23SStefan Hajnoczi         ret = qed_write_header_sync(s);
47775411d23SStefan Hajnoczi         if (ret) {
47875411d23SStefan Hajnoczi             return ret;
47975411d23SStefan Hajnoczi         }
48075411d23SStefan Hajnoczi 
48175411d23SStefan Hajnoczi         /* From here on only known autoclear feature bits are valid */
4829a4f4c31SKevin Wolf         bdrv_flush(bs->file->bs);
48375411d23SStefan Hajnoczi     }
48475411d23SStefan Hajnoczi 
485298800caSStefan Hajnoczi     s->l1_table = qed_alloc_table(s);
486298800caSStefan Hajnoczi     qed_init_l2_cache(&s->l2_cache);
487298800caSStefan Hajnoczi 
488298800caSStefan Hajnoczi     ret = qed_read_l1_table_sync(s);
489298800caSStefan Hajnoczi     if (ret) {
49001979a98SStefan Hajnoczi         goto out;
49101979a98SStefan Hajnoczi     }
49201979a98SStefan Hajnoczi 
49301979a98SStefan Hajnoczi     /* If image was not closed cleanly, check consistency */
494058f8f16SStefan Hajnoczi     if (!(flags & BDRV_O_CHECK) && (s->header.features & QED_F_NEED_CHECK)) {
49501979a98SStefan Hajnoczi         /* Read-only images cannot be fixed.  There is no risk of corruption
49601979a98SStefan Hajnoczi          * since write operations are not possible.  Therefore, allow
49701979a98SStefan Hajnoczi          * potentially inconsistent images to be opened read-only.  This can
49801979a98SStefan Hajnoczi          * aid data recovery from an otherwise inconsistent image.
49901979a98SStefan Hajnoczi          */
5009a4f4c31SKevin Wolf         if (!bdrv_is_read_only(bs->file->bs) &&
50104c01a5cSKevin Wolf             !(flags & BDRV_O_INACTIVE)) {
50201979a98SStefan Hajnoczi             BdrvCheckResult result = {0};
50301979a98SStefan Hajnoczi 
50401979a98SStefan Hajnoczi             ret = qed_check(s, &result, true);
5056f321e93SStefan Hajnoczi             if (ret) {
5066f321e93SStefan Hajnoczi                 goto out;
5076f321e93SStefan Hajnoczi             }
50801979a98SStefan Hajnoczi         }
50901979a98SStefan Hajnoczi     }
51001979a98SStefan Hajnoczi 
511a8c868c3SStefan Hajnoczi     bdrv_qed_attach_aio_context(bs, bdrv_get_aio_context(bs));
5126f321e93SStefan Hajnoczi 
51301979a98SStefan Hajnoczi out:
51401979a98SStefan Hajnoczi     if (ret) {
515298800caSStefan Hajnoczi         qed_free_l2_cache(&s->l2_cache);
516298800caSStefan Hajnoczi         qemu_vfree(s->l1_table);
517298800caSStefan Hajnoczi     }
51875411d23SStefan Hajnoczi     return ret;
51975411d23SStefan Hajnoczi }
52075411d23SStefan Hajnoczi 
5219fb4dfc5SPaolo Bonzini typedef struct QEDOpenCo {
5229fb4dfc5SPaolo Bonzini     BlockDriverState *bs;
5239fb4dfc5SPaolo Bonzini     QDict *options;
5249fb4dfc5SPaolo Bonzini     int flags;
5259fb4dfc5SPaolo Bonzini     Error **errp;
5269fb4dfc5SPaolo Bonzini     int ret;
5279fb4dfc5SPaolo Bonzini } QEDOpenCo;
5289fb4dfc5SPaolo Bonzini 
5299fb4dfc5SPaolo Bonzini static void coroutine_fn bdrv_qed_open_entry(void *opaque)
5309fb4dfc5SPaolo Bonzini {
5319fb4dfc5SPaolo Bonzini     QEDOpenCo *qoc = opaque;
5329fb4dfc5SPaolo Bonzini     BDRVQEDState *s = qoc->bs->opaque;
5339fb4dfc5SPaolo Bonzini 
5349fb4dfc5SPaolo Bonzini     qemu_co_mutex_lock(&s->table_lock);
5359fb4dfc5SPaolo Bonzini     qoc->ret = bdrv_qed_do_open(qoc->bs, qoc->options, qoc->flags, qoc->errp);
5369fb4dfc5SPaolo Bonzini     qemu_co_mutex_unlock(&s->table_lock);
5379fb4dfc5SPaolo Bonzini }
5389fb4dfc5SPaolo Bonzini 
5394e4bf5c4SKevin Wolf static int bdrv_qed_open(BlockDriverState *bs, QDict *options, int flags,
5404e4bf5c4SKevin Wolf                          Error **errp)
5414e4bf5c4SKevin Wolf {
5429fb4dfc5SPaolo Bonzini     QEDOpenCo qoc = {
5439fb4dfc5SPaolo Bonzini         .bs = bs,
5449fb4dfc5SPaolo Bonzini         .options = options,
5459fb4dfc5SPaolo Bonzini         .flags = flags,
5469fb4dfc5SPaolo Bonzini         .errp = errp,
5479fb4dfc5SPaolo Bonzini         .ret = -EINPROGRESS
5489fb4dfc5SPaolo Bonzini     };
5499fb4dfc5SPaolo Bonzini 
5508b1869daSMax Reitz     bs->file = bdrv_open_child(NULL, options, "file", bs, &child_of_bds,
5518b1869daSMax Reitz                                BDRV_CHILD_IMAGE, false, errp);
5524e4bf5c4SKevin Wolf     if (!bs->file) {
5534e4bf5c4SKevin Wolf         return -EINVAL;
5544e4bf5c4SKevin Wolf     }
5554e4bf5c4SKevin Wolf 
55661c7887eSPaolo Bonzini     bdrv_qed_init_state(bs);
5579fb4dfc5SPaolo Bonzini     if (qemu_in_coroutine()) {
5589fb4dfc5SPaolo Bonzini         bdrv_qed_open_entry(&qoc);
5599fb4dfc5SPaolo Bonzini     } else {
5604720cbeeSKevin Wolf         assert(qemu_get_current_aio_context() == qemu_get_aio_context());
5619fb4dfc5SPaolo Bonzini         qemu_coroutine_enter(qemu_coroutine_create(bdrv_qed_open_entry, &qoc));
5629fb4dfc5SPaolo Bonzini         BDRV_POLL_WHILE(bs, qoc.ret == -EINPROGRESS);
5639fb4dfc5SPaolo Bonzini     }
5649fb4dfc5SPaolo Bonzini     BDRV_POLL_WHILE(bs, qoc.ret == -EINPROGRESS);
5659fb4dfc5SPaolo Bonzini     return qoc.ret;
5664e4bf5c4SKevin Wolf }
5674e4bf5c4SKevin Wolf 
5683baca891SKevin Wolf static void bdrv_qed_refresh_limits(BlockDriverState *bs, Error **errp)
569d34682cdSKevin Wolf {
570d34682cdSKevin Wolf     BDRVQEDState *s = bs->opaque;
571d34682cdSKevin Wolf 
572cf081fcaSEric Blake     bs->bl.pwrite_zeroes_alignment = s->header.cluster_size;
573d34682cdSKevin Wolf }
574d34682cdSKevin Wolf 
575f9cb20f1SJeff Cody /* We have nothing to do for QED reopen, stubs just return
576f9cb20f1SJeff Cody  * success */
577f9cb20f1SJeff Cody static int bdrv_qed_reopen_prepare(BDRVReopenState *state,
578f9cb20f1SJeff Cody                                    BlockReopenQueue *queue, Error **errp)
579f9cb20f1SJeff Cody {
580f9cb20f1SJeff Cody     return 0;
581f9cb20f1SJeff Cody }
582f9cb20f1SJeff Cody 
58375411d23SStefan Hajnoczi static void bdrv_qed_close(BlockDriverState *bs)
58475411d23SStefan Hajnoczi {
585298800caSStefan Hajnoczi     BDRVQEDState *s = bs->opaque;
586298800caSStefan Hajnoczi 
587a8c868c3SStefan Hajnoczi     bdrv_qed_detach_aio_context(bs);
5886f321e93SStefan Hajnoczi 
58901979a98SStefan Hajnoczi     /* Ensure writes reach stable storage */
5909a4f4c31SKevin Wolf     bdrv_flush(bs->file->bs);
59101979a98SStefan Hajnoczi 
59201979a98SStefan Hajnoczi     /* Clean shutdown, no check required on next open */
59301979a98SStefan Hajnoczi     if (s->header.features & QED_F_NEED_CHECK) {
59401979a98SStefan Hajnoczi         s->header.features &= ~QED_F_NEED_CHECK;
59501979a98SStefan Hajnoczi         qed_write_header_sync(s);
59601979a98SStefan Hajnoczi     }
59701979a98SStefan Hajnoczi 
598298800caSStefan Hajnoczi     qed_free_l2_cache(&s->l2_cache);
599298800caSStefan Hajnoczi     qemu_vfree(s->l1_table);
60075411d23SStefan Hajnoczi }
60175411d23SStefan Hajnoczi 
602959355a4SKevin Wolf static int coroutine_fn bdrv_qed_co_create(BlockdevCreateOptions *opts,
603959355a4SKevin Wolf                                            Error **errp)
60475411d23SStefan Hajnoczi {
605959355a4SKevin Wolf     BlockdevCreateOptionsQed *qed_opts;
606959355a4SKevin Wolf     BlockBackend *blk = NULL;
607959355a4SKevin Wolf     BlockDriverState *bs = NULL;
608959355a4SKevin Wolf 
609959355a4SKevin Wolf     QEDHeader header;
61075411d23SStefan Hajnoczi     QEDHeader le_header;
61175411d23SStefan Hajnoczi     uint8_t *l1_table = NULL;
612959355a4SKevin Wolf     size_t l1_size;
61375411d23SStefan Hajnoczi     int ret = 0;
61475411d23SStefan Hajnoczi 
615959355a4SKevin Wolf     assert(opts->driver == BLOCKDEV_DRIVER_QED);
616959355a4SKevin Wolf     qed_opts = &opts->u.qed;
617959355a4SKevin Wolf 
618959355a4SKevin Wolf     /* Validate options and set default values */
619959355a4SKevin Wolf     if (!qed_opts->has_cluster_size) {
620959355a4SKevin Wolf         qed_opts->cluster_size = QED_DEFAULT_CLUSTER_SIZE;
621959355a4SKevin Wolf     }
622959355a4SKevin Wolf     if (!qed_opts->has_table_size) {
623959355a4SKevin Wolf         qed_opts->table_size = QED_DEFAULT_TABLE_SIZE;
62475411d23SStefan Hajnoczi     }
62575411d23SStefan Hajnoczi 
626959355a4SKevin Wolf     if (!qed_is_cluster_size_valid(qed_opts->cluster_size)) {
627959355a4SKevin Wolf         error_setg(errp, "QED cluster size must be within range [%u, %u] "
628959355a4SKevin Wolf                          "and power of 2",
629959355a4SKevin Wolf                    QED_MIN_CLUSTER_SIZE, QED_MAX_CLUSTER_SIZE);
630959355a4SKevin Wolf         return -EINVAL;
631959355a4SKevin Wolf     }
632959355a4SKevin Wolf     if (!qed_is_table_size_valid(qed_opts->table_size)) {
633959355a4SKevin Wolf         error_setg(errp, "QED table size must be within range [%u, %u] "
634959355a4SKevin Wolf                          "and power of 2",
635959355a4SKevin Wolf                    QED_MIN_TABLE_SIZE, QED_MAX_TABLE_SIZE);
636959355a4SKevin Wolf         return -EINVAL;
637959355a4SKevin Wolf     }
638959355a4SKevin Wolf     if (!qed_is_image_size_valid(qed_opts->size, qed_opts->cluster_size,
639959355a4SKevin Wolf                                  qed_opts->table_size))
640959355a4SKevin Wolf     {
641959355a4SKevin Wolf         error_setg(errp, "QED image size must be a non-zero multiple of "
642959355a4SKevin Wolf                          "cluster size and less than %" PRIu64 " bytes",
643959355a4SKevin Wolf                    qed_max_image_size(qed_opts->cluster_size,
644959355a4SKevin Wolf                                       qed_opts->table_size));
645959355a4SKevin Wolf         return -EINVAL;
646959355a4SKevin Wolf     }
647959355a4SKevin Wolf 
648959355a4SKevin Wolf     /* Create BlockBackend to write to the image */
649959355a4SKevin Wolf     bs = bdrv_open_blockdev_ref(qed_opts->file, errp);
650959355a4SKevin Wolf     if (bs == NULL) {
6518a56fdadSKevin Wolf         return -EIO;
65275411d23SStefan Hajnoczi     }
65375411d23SStefan Hajnoczi 
654a3aeeab5SEric Blake     blk = blk_new_with_bs(bs, BLK_PERM_WRITE | BLK_PERM_RESIZE, BLK_PERM_ALL,
655a3aeeab5SEric Blake                           errp);
656a3aeeab5SEric Blake     if (!blk) {
657a3aeeab5SEric Blake         ret = -EPERM;
658959355a4SKevin Wolf         goto out;
659959355a4SKevin Wolf     }
6608a56fdadSKevin Wolf     blk_set_allow_write_beyond_eof(blk, true);
6618a56fdadSKevin Wolf 
662959355a4SKevin Wolf     /* Prepare image format */
663959355a4SKevin Wolf     header = (QEDHeader) {
664959355a4SKevin Wolf         .magic = QED_MAGIC,
665959355a4SKevin Wolf         .cluster_size = qed_opts->cluster_size,
666959355a4SKevin Wolf         .table_size = qed_opts->table_size,
667959355a4SKevin Wolf         .header_size = 1,
668959355a4SKevin Wolf         .features = 0,
669959355a4SKevin Wolf         .compat_features = 0,
670959355a4SKevin Wolf         .l1_table_offset = qed_opts->cluster_size,
671959355a4SKevin Wolf         .image_size = qed_opts->size,
672959355a4SKevin Wolf     };
673959355a4SKevin Wolf 
674959355a4SKevin Wolf     l1_size = header.cluster_size * header.table_size;
675959355a4SKevin Wolf 
676e8d04f92SMax Reitz     /*
677e8d04f92SMax Reitz      * The QED format associates file length with allocation status,
678e8d04f92SMax Reitz      * so a new file (which is empty) must have a length of 0.
679e8d04f92SMax Reitz      */
6808c6242b6SKevin Wolf     ret = blk_truncate(blk, 0, true, PREALLOC_MODE_OFF, 0, errp);
681c743849bSStefan Hajnoczi     if (ret < 0) {
682c743849bSStefan Hajnoczi         goto out;
683c743849bSStefan Hajnoczi     }
684c743849bSStefan Hajnoczi 
685959355a4SKevin Wolf     if (qed_opts->has_backing_file) {
68675411d23SStefan Hajnoczi         header.features |= QED_F_BACKING_FILE;
68775411d23SStefan Hajnoczi         header.backing_filename_offset = sizeof(le_header);
688959355a4SKevin Wolf         header.backing_filename_size = strlen(qed_opts->backing_file);
68975411d23SStefan Hajnoczi 
690959355a4SKevin Wolf         if (qed_opts->has_backing_fmt) {
691959355a4SKevin Wolf             const char *backing_fmt = BlockdevDriver_str(qed_opts->backing_fmt);
69275411d23SStefan Hajnoczi             if (qed_fmt_is_raw(backing_fmt)) {
69375411d23SStefan Hajnoczi                 header.features |= QED_F_BACKING_FORMAT_NO_PROBE;
69475411d23SStefan Hajnoczi             }
69575411d23SStefan Hajnoczi         }
696959355a4SKevin Wolf     }
69775411d23SStefan Hajnoczi 
69875411d23SStefan Hajnoczi     qed_header_cpu_to_le(&header, &le_header);
6998341f00dSEric Blake     ret = blk_pwrite(blk, 0, &le_header, sizeof(le_header), 0);
70075411d23SStefan Hajnoczi     if (ret < 0) {
70175411d23SStefan Hajnoczi         goto out;
70275411d23SStefan Hajnoczi     }
703959355a4SKevin Wolf     ret = blk_pwrite(blk, sizeof(le_header), qed_opts->backing_file,
7048341f00dSEric Blake                      header.backing_filename_size, 0);
70575411d23SStefan Hajnoczi     if (ret < 0) {
70675411d23SStefan Hajnoczi         goto out;
70775411d23SStefan Hajnoczi     }
70875411d23SStefan Hajnoczi 
7097267c094SAnthony Liguori     l1_table = g_malloc0(l1_size);
7108341f00dSEric Blake     ret = blk_pwrite(blk, header.l1_table_offset, l1_table, l1_size, 0);
71175411d23SStefan Hajnoczi     if (ret < 0) {
71275411d23SStefan Hajnoczi         goto out;
71375411d23SStefan Hajnoczi     }
71475411d23SStefan Hajnoczi 
71575411d23SStefan Hajnoczi     ret = 0; /* success */
71675411d23SStefan Hajnoczi out:
7177267c094SAnthony Liguori     g_free(l1_table);
7188a56fdadSKevin Wolf     blk_unref(blk);
719959355a4SKevin Wolf     bdrv_unref(bs);
72075411d23SStefan Hajnoczi     return ret;
72175411d23SStefan Hajnoczi }
72275411d23SStefan Hajnoczi 
723b92902dfSMaxim Levitsky static int coroutine_fn bdrv_qed_co_create_opts(BlockDriver *drv,
724b92902dfSMaxim Levitsky                                                 const char *filename,
725efc75e2aSStefan Hajnoczi                                                 QemuOpts *opts,
726efc75e2aSStefan Hajnoczi                                                 Error **errp)
72775411d23SStefan Hajnoczi {
728959355a4SKevin Wolf     BlockdevCreateOptions *create_options = NULL;
72992adf9dbSMarkus Armbruster     QDict *qdict;
730959355a4SKevin Wolf     Visitor *v;
731959355a4SKevin Wolf     BlockDriverState *bs = NULL;
732959355a4SKevin Wolf     Error *local_err = NULL;
7337ab74849SChunyan Liu     int ret;
73475411d23SStefan Hajnoczi 
735959355a4SKevin Wolf     static const QDictRenames opt_renames[] = {
736959355a4SKevin Wolf         { BLOCK_OPT_BACKING_FILE,       "backing-file" },
737959355a4SKevin Wolf         { BLOCK_OPT_BACKING_FMT,        "backing-fmt" },
738959355a4SKevin Wolf         { BLOCK_OPT_CLUSTER_SIZE,       "cluster-size" },
739959355a4SKevin Wolf         { BLOCK_OPT_TABLE_SIZE,         "table-size" },
740959355a4SKevin Wolf         { NULL, NULL },
741959355a4SKevin Wolf     };
74275411d23SStefan Hajnoczi 
743959355a4SKevin Wolf     /* Parse options and convert legacy syntax */
744959355a4SKevin Wolf     qdict = qemu_opts_to_qdict_filtered(opts, NULL, &qed_create_opts, true);
745959355a4SKevin Wolf 
746959355a4SKevin Wolf     if (!qdict_rename_keys(qdict, opt_renames, errp)) {
7477ab74849SChunyan Liu         ret = -EINVAL;
748959355a4SKevin Wolf         goto fail;
74975411d23SStefan Hajnoczi     }
75075411d23SStefan Hajnoczi 
751959355a4SKevin Wolf     /* Create and open the file (protocol layer) */
752959355a4SKevin Wolf     ret = bdrv_create_file(filename, opts, &local_err);
753959355a4SKevin Wolf     if (ret < 0) {
754959355a4SKevin Wolf         error_propagate(errp, local_err);
755959355a4SKevin Wolf         goto fail;
756959355a4SKevin Wolf     }
7577ab74849SChunyan Liu 
758959355a4SKevin Wolf     bs = bdrv_open(filename, NULL, NULL,
759959355a4SKevin Wolf                    BDRV_O_RDWR | BDRV_O_RESIZE | BDRV_O_PROTOCOL, errp);
760959355a4SKevin Wolf     if (bs == NULL) {
761959355a4SKevin Wolf         ret = -EIO;
762959355a4SKevin Wolf         goto fail;
763959355a4SKevin Wolf     }
764959355a4SKevin Wolf 
765959355a4SKevin Wolf     /* Now get the QAPI type BlockdevCreateOptions */
766959355a4SKevin Wolf     qdict_put_str(qdict, "driver", "qed");
767959355a4SKevin Wolf     qdict_put_str(qdict, "file", bs->node_name);
768959355a4SKevin Wolf 
769af91062eSMarkus Armbruster     v = qobject_input_visitor_new_flat_confused(qdict, errp);
770af91062eSMarkus Armbruster     if (!v) {
771959355a4SKevin Wolf         ret = -EINVAL;
772959355a4SKevin Wolf         goto fail;
773959355a4SKevin Wolf     }
774959355a4SKevin Wolf 
775959355a4SKevin Wolf     visit_type_BlockdevCreateOptions(v, NULL, &create_options, &local_err);
776959355a4SKevin Wolf     visit_free(v);
777959355a4SKevin Wolf 
778959355a4SKevin Wolf     if (local_err) {
779959355a4SKevin Wolf         error_propagate(errp, local_err);
780959355a4SKevin Wolf         ret = -EINVAL;
781959355a4SKevin Wolf         goto fail;
782959355a4SKevin Wolf     }
783959355a4SKevin Wolf 
784959355a4SKevin Wolf     /* Silently round up size */
785959355a4SKevin Wolf     assert(create_options->driver == BLOCKDEV_DRIVER_QED);
786959355a4SKevin Wolf     create_options->u.qed.size =
787959355a4SKevin Wolf         ROUND_UP(create_options->u.qed.size, BDRV_SECTOR_SIZE);
788959355a4SKevin Wolf 
789959355a4SKevin Wolf     /* Create the qed image (format layer) */
790959355a4SKevin Wolf     ret = bdrv_qed_co_create(create_options, errp);
791959355a4SKevin Wolf 
792959355a4SKevin Wolf fail:
793cb3e7f08SMarc-André Lureau     qobject_unref(qdict);
794959355a4SKevin Wolf     bdrv_unref(bs);
795959355a4SKevin Wolf     qapi_free_BlockdevCreateOptions(create_options);
7967ab74849SChunyan Liu     return ret;
79775411d23SStefan Hajnoczi }
79875411d23SStefan Hajnoczi 
799b8d739fdSEric Blake static int coroutine_fn bdrv_qed_co_block_status(BlockDriverState *bs,
800b8d739fdSEric Blake                                                  bool want_zero,
801b8d739fdSEric Blake                                                  int64_t pos, int64_t bytes,
802b8d739fdSEric Blake                                                  int64_t *pnum, int64_t *map,
80367a0fd2aSFam Zheng                                                  BlockDriverState **file)
80475411d23SStefan Hajnoczi {
805298800caSStefan Hajnoczi     BDRVQEDState *s = bs->opaque;
806b8d739fdSEric Blake     size_t len = MIN(bytes, SIZE_MAX);
807b8d739fdSEric Blake     int status;
808298800caSStefan Hajnoczi     QEDRequest request = { .l2_table = NULL };
8090f21b7a1SKevin Wolf     uint64_t offset;
8100f21b7a1SKevin Wolf     int ret;
811298800caSStefan Hajnoczi 
8121f01e50bSPaolo Bonzini     qemu_co_mutex_lock(&s->table_lock);
813b8d739fdSEric Blake     ret = qed_find_cluster(s, &request, pos, &len, &offset);
814298800caSStefan Hajnoczi 
815b8d739fdSEric Blake     *pnum = len;
816b8d739fdSEric Blake     switch (ret) {
817b8d739fdSEric Blake     case QED_CLUSTER_FOUND:
818b8d739fdSEric Blake         *map = offset | qed_offset_into_cluster(s, pos);
819b8d739fdSEric Blake         status = BDRV_BLOCK_DATA | BDRV_BLOCK_OFFSET_VALID;
820b8d739fdSEric Blake         *file = bs->file->bs;
821b8d739fdSEric Blake         break;
822b8d739fdSEric Blake     case QED_CLUSTER_ZERO:
823b8d739fdSEric Blake         status = BDRV_BLOCK_ZERO;
824b8d739fdSEric Blake         break;
825b8d739fdSEric Blake     case QED_CLUSTER_L2:
826b8d739fdSEric Blake     case QED_CLUSTER_L1:
827b8d739fdSEric Blake         status = 0;
828b8d739fdSEric Blake         break;
829b8d739fdSEric Blake     default:
830b8d739fdSEric Blake         assert(ret < 0);
831b8d739fdSEric Blake         status = ret;
832b8d739fdSEric Blake         break;
833b8d739fdSEric Blake     }
834298800caSStefan Hajnoczi 
835298800caSStefan Hajnoczi     qed_unref_l2_cache_entry(request.l2_table);
8361f01e50bSPaolo Bonzini     qemu_co_mutex_unlock(&s->table_lock);
837298800caSStefan Hajnoczi 
838b8d739fdSEric Blake     return status;
83975411d23SStefan Hajnoczi }
84075411d23SStefan Hajnoczi 
841eabba580SStefan Hajnoczi static BDRVQEDState *acb_to_s(QEDAIOCB *acb)
842eabba580SStefan Hajnoczi {
84348cc565eSKevin Wolf     return acb->bs->opaque;
844eabba580SStefan Hajnoczi }
845eabba580SStefan Hajnoczi 
846eabba580SStefan Hajnoczi /**
847eabba580SStefan Hajnoczi  * Read from the backing file or zero-fill if no backing file
848eabba580SStefan Hajnoczi  *
849eabba580SStefan Hajnoczi  * @s:              QED state
850eabba580SStefan Hajnoczi  * @pos:            Byte position in device
851eabba580SStefan Hajnoczi  * @qiov:           Destination I/O vector
852eabba580SStefan Hajnoczi  *
853eabba580SStefan Hajnoczi  * This function reads qiov->size bytes starting at pos from the backing file.
854eabba580SStefan Hajnoczi  * If there is no backing file then zeroes are read.
855eabba580SStefan Hajnoczi  */
85687f0d882SKevin Wolf static int coroutine_fn qed_read_backing_file(BDRVQEDState *s, uint64_t pos,
857*365fed51SEric Blake                                               QEMUIOVector *qiov)
858eabba580SStefan Hajnoczi {
859760e0063SKevin Wolf     if (s->bs->backing) {
860820100fdSPaolo Bonzini         BLKDBG_EVENT(s->bs->file, BLKDBG_READ_BACKING_AIO);
861*365fed51SEric Blake         return bdrv_co_preadv(s->bs->backing, pos, qiov->size, qiov, 0);
862e85c5281SKevin Wolf     }
863*365fed51SEric Blake     qemu_iovec_memset(qiov, 0, 0, qiov->size);
864e85c5281SKevin Wolf     return 0;
865eabba580SStefan Hajnoczi }
866eabba580SStefan Hajnoczi 
867eabba580SStefan Hajnoczi /**
868eabba580SStefan Hajnoczi  * Copy data from backing file into the image
869eabba580SStefan Hajnoczi  *
870eabba580SStefan Hajnoczi  * @s:          QED state
871eabba580SStefan Hajnoczi  * @pos:        Byte position in device
872eabba580SStefan Hajnoczi  * @len:        Number of bytes
873eabba580SStefan Hajnoczi  * @offset:     Byte offset in image file
874eabba580SStefan Hajnoczi  */
87587f0d882SKevin Wolf static int coroutine_fn qed_copy_from_backing_file(BDRVQEDState *s,
87687f0d882SKevin Wolf                                                    uint64_t pos, uint64_t len,
87787f0d882SKevin Wolf                                                    uint64_t offset)
878eabba580SStefan Hajnoczi {
8790f7aa24dSKevin Wolf     QEMUIOVector qiov;
880e85c5281SKevin Wolf     int ret;
881eabba580SStefan Hajnoczi 
882eabba580SStefan Hajnoczi     /* Skip copy entirely if there is no work to do */
883eabba580SStefan Hajnoczi     if (len == 0) {
884b4ac32f3SKevin Wolf         return 0;
885eabba580SStefan Hajnoczi     }
886eabba580SStefan Hajnoczi 
887342544f9SVladimir Sementsov-Ogievskiy     qemu_iovec_init_buf(&qiov, qemu_blockalign(s->bs, len), len);
888eabba580SStefan Hajnoczi 
889*365fed51SEric Blake     ret = qed_read_backing_file(s, pos, &qiov);
8900f7aa24dSKevin Wolf 
8910f7aa24dSKevin Wolf     if (ret) {
8920f7aa24dSKevin Wolf         goto out;
8930f7aa24dSKevin Wolf     }
8940f7aa24dSKevin Wolf 
8950f7aa24dSKevin Wolf     BLKDBG_EVENT(s->bs->file, BLKDBG_COW_WRITE);
8960f714ec7SKevin Wolf     ret = bdrv_co_pwritev(s->bs->file, offset, qiov.size, &qiov, 0);
8970f7aa24dSKevin Wolf     if (ret < 0) {
8980f7aa24dSKevin Wolf         goto out;
8990f7aa24dSKevin Wolf     }
9000f7aa24dSKevin Wolf     ret = 0;
9010f7aa24dSKevin Wolf out:
902342544f9SVladimir Sementsov-Ogievskiy     qemu_vfree(qemu_iovec_buf(&qiov));
903b4ac32f3SKevin Wolf     return ret;
904eabba580SStefan Hajnoczi }
905eabba580SStefan Hajnoczi 
906eabba580SStefan Hajnoczi /**
907eabba580SStefan Hajnoczi  * Link one or more contiguous clusters into a table
908eabba580SStefan Hajnoczi  *
909eabba580SStefan Hajnoczi  * @s:              QED state
910eabba580SStefan Hajnoczi  * @table:          L2 table
911eabba580SStefan Hajnoczi  * @index:          First cluster index
912eabba580SStefan Hajnoczi  * @n:              Number of contiguous clusters
91321df65b6SAnthony Liguori  * @cluster:        First cluster offset
91421df65b6SAnthony Liguori  *
91521df65b6SAnthony Liguori  * The cluster offset may be an allocated byte offset in the image file, the
91621df65b6SAnthony Liguori  * zero cluster marker, or the unallocated cluster marker.
9171f01e50bSPaolo Bonzini  *
9181f01e50bSPaolo Bonzini  * Called with table_lock held.
919eabba580SStefan Hajnoczi  */
92087f0d882SKevin Wolf static void coroutine_fn qed_update_l2_table(BDRVQEDState *s, QEDTable *table,
92187f0d882SKevin Wolf                                              int index, unsigned int n,
92287f0d882SKevin Wolf                                              uint64_t cluster)
923eabba580SStefan Hajnoczi {
924eabba580SStefan Hajnoczi     int i;
925eabba580SStefan Hajnoczi     for (i = index; i < index + n; i++) {
926eabba580SStefan Hajnoczi         table->offsets[i] = cluster;
92721df65b6SAnthony Liguori         if (!qed_offset_is_unalloc_cluster(cluster) &&
92821df65b6SAnthony Liguori             !qed_offset_is_zero_cluster(cluster)) {
929eabba580SStefan Hajnoczi             cluster += s->header.cluster_size;
930eabba580SStefan Hajnoczi         }
931eabba580SStefan Hajnoczi     }
93221df65b6SAnthony Liguori }
933eabba580SStefan Hajnoczi 
9341f01e50bSPaolo Bonzini /* Called with table_lock held.  */
93587f0d882SKevin Wolf static void coroutine_fn qed_aio_complete(QEDAIOCB *acb)
936eabba580SStefan Hajnoczi {
937eabba580SStefan Hajnoczi     BDRVQEDState *s = acb_to_s(acb);
938eabba580SStefan Hajnoczi 
939eabba580SStefan Hajnoczi     /* Free resources */
940eabba580SStefan Hajnoczi     qemu_iovec_destroy(&acb->cur_qiov);
941eabba580SStefan Hajnoczi     qed_unref_l2_cache_entry(acb->request.l2_table);
942eabba580SStefan Hajnoczi 
9430e71be19SStefan Hajnoczi     /* Free the buffer we may have allocated for zero writes */
9440e71be19SStefan Hajnoczi     if (acb->flags & QED_AIOCB_ZERO) {
9450e71be19SStefan Hajnoczi         qemu_vfree(acb->qiov->iov[0].iov_base);
9460e71be19SStefan Hajnoczi         acb->qiov->iov[0].iov_base = NULL;
9470e71be19SStefan Hajnoczi     }
9480e71be19SStefan Hajnoczi 
949eabba580SStefan Hajnoczi     /* Start next allocating write request waiting behind this one.  Note that
950eabba580SStefan Hajnoczi      * requests enqueue themselves when they first hit an unallocated cluster
951eabba580SStefan Hajnoczi      * but they wait until the entire request is finished before waking up the
952eabba580SStefan Hajnoczi      * next request in the queue.  This ensures that we don't cycle through
953eabba580SStefan Hajnoczi      * requests multiple times but rather finish one at a time completely.
954eabba580SStefan Hajnoczi      */
9550806c3b5SKevin Wolf     if (acb == s->allocating_acb) {
9560806c3b5SKevin Wolf         s->allocating_acb = NULL;
9570806c3b5SKevin Wolf         if (!qemu_co_queue_empty(&s->allocating_write_reqs)) {
9581f01e50bSPaolo Bonzini             qemu_co_queue_next(&s->allocating_write_reqs);
9596f321e93SStefan Hajnoczi         } else if (s->header.features & QED_F_NEED_CHECK) {
9606f321e93SStefan Hajnoczi             qed_start_need_check_timer(s);
961eabba580SStefan Hajnoczi         }
962eabba580SStefan Hajnoczi     }
963eabba580SStefan Hajnoczi }
964eabba580SStefan Hajnoczi 
965eabba580SStefan Hajnoczi /**
966fae25ac7SKevin Wolf  * Update L1 table with new L2 table offset and write it out
9671f01e50bSPaolo Bonzini  *
9681f01e50bSPaolo Bonzini  * Called with table_lock held.
969eabba580SStefan Hajnoczi  */
97087f0d882SKevin Wolf static int coroutine_fn qed_aio_write_l1_update(QEDAIOCB *acb)
971eabba580SStefan Hajnoczi {
972eabba580SStefan Hajnoczi     BDRVQEDState *s = acb_to_s(acb);
973eabba580SStefan Hajnoczi     CachedL2Table *l2_table = acb->request.l2_table;
974e4fc8781SStefan Hajnoczi     uint64_t l2_offset = l2_table->offset;
975fb18de21SKevin Wolf     int index, ret;
976fae25ac7SKevin Wolf 
977fae25ac7SKevin Wolf     index = qed_l1_index(s, acb->cur_pos);
978fae25ac7SKevin Wolf     s->l1_table->offsets[index] = l2_table->offset;
979fae25ac7SKevin Wolf 
980fae25ac7SKevin Wolf     ret = qed_write_l1_table(s, index, 1);
981fae25ac7SKevin Wolf 
982fae25ac7SKevin Wolf     /* Commit the current L2 table to the cache */
983eabba580SStefan Hajnoczi     qed_commit_l2_cache_entry(&s->l2_cache, l2_table);
984eabba580SStefan Hajnoczi 
985eabba580SStefan Hajnoczi     /* This is guaranteed to succeed because we just committed the entry to the
986eabba580SStefan Hajnoczi      * cache.
987eabba580SStefan Hajnoczi      */
988e4fc8781SStefan Hajnoczi     acb->request.l2_table = qed_find_l2_cache_entry(&s->l2_cache, l2_offset);
989eabba580SStefan Hajnoczi     assert(acb->request.l2_table != NULL);
990eabba580SStefan Hajnoczi 
991fb18de21SKevin Wolf     return ret;
992eabba580SStefan Hajnoczi }
993eabba580SStefan Hajnoczi 
994eabba580SStefan Hajnoczi 
995eabba580SStefan Hajnoczi /**
996eabba580SStefan Hajnoczi  * Update L2 table with new cluster offsets and write them out
9971f01e50bSPaolo Bonzini  *
9981f01e50bSPaolo Bonzini  * Called with table_lock held.
999eabba580SStefan Hajnoczi  */
100087f0d882SKevin Wolf static int coroutine_fn qed_aio_write_l2_update(QEDAIOCB *acb, uint64_t offset)
1001eabba580SStefan Hajnoczi {
1002eabba580SStefan Hajnoczi     BDRVQEDState *s = acb_to_s(acb);
1003eabba580SStefan Hajnoczi     bool need_alloc = acb->find_cluster_ret == QED_CLUSTER_L1;
100488d2dd72SKevin Wolf     int index, ret;
1005eabba580SStefan Hajnoczi 
1006eabba580SStefan Hajnoczi     if (need_alloc) {
1007eabba580SStefan Hajnoczi         qed_unref_l2_cache_entry(acb->request.l2_table);
1008eabba580SStefan Hajnoczi         acb->request.l2_table = qed_new_l2_table(s);
1009eabba580SStefan Hajnoczi     }
1010eabba580SStefan Hajnoczi 
1011eabba580SStefan Hajnoczi     index = qed_l2_index(s, acb->cur_pos);
1012eabba580SStefan Hajnoczi     qed_update_l2_table(s, acb->request.l2_table->table, index, acb->cur_nclusters,
10130e71be19SStefan Hajnoczi                          offset);
1014eabba580SStefan Hajnoczi 
1015eabba580SStefan Hajnoczi     if (need_alloc) {
1016eabba580SStefan Hajnoczi         /* Write out the whole new L2 table */
1017453e53e2SKevin Wolf         ret = qed_write_l2_table(s, &acb->request, 0, s->table_nelems, true);
1018fb18de21SKevin Wolf         if (ret) {
101988d2dd72SKevin Wolf             return ret;
1020fb18de21SKevin Wolf         }
102188d2dd72SKevin Wolf         return qed_aio_write_l1_update(acb);
1022eabba580SStefan Hajnoczi     } else {
1023eabba580SStefan Hajnoczi         /* Write out only the updated part of the L2 table */
1024453e53e2SKevin Wolf         ret = qed_write_l2_table(s, &acb->request, index, acb->cur_nclusters,
1025453e53e2SKevin Wolf                                  false);
102688d2dd72SKevin Wolf         if (ret) {
102788d2dd72SKevin Wolf             return ret;
1028eabba580SStefan Hajnoczi         }
102988d2dd72SKevin Wolf     }
103088d2dd72SKevin Wolf     return 0;
1031eabba580SStefan Hajnoczi }
1032eabba580SStefan Hajnoczi 
1033eabba580SStefan Hajnoczi /**
1034eabba580SStefan Hajnoczi  * Write data to the image file
10351f01e50bSPaolo Bonzini  *
10361f01e50bSPaolo Bonzini  * Called with table_lock *not* held.
1037eabba580SStefan Hajnoczi  */
103887f0d882SKevin Wolf static int coroutine_fn qed_aio_write_main(QEDAIOCB *acb)
1039eabba580SStefan Hajnoczi {
1040eabba580SStefan Hajnoczi     BDRVQEDState *s = acb_to_s(acb);
1041eabba580SStefan Hajnoczi     uint64_t offset = acb->cur_cluster +
1042eabba580SStefan Hajnoczi                       qed_offset_into_cluster(s, acb->cur_pos);
1043eabba580SStefan Hajnoczi 
1044eaf0bc56SKevin Wolf     trace_qed_aio_write_main(s, acb, 0, offset, acb->cur_qiov.size);
1045eabba580SStefan Hajnoczi 
1046a4d8f1aeSKevin Wolf     BLKDBG_EVENT(s->bs->file, BLKDBG_WRITE_AIO);
1047e7569c18SPaolo Bonzini     return bdrv_co_pwritev(s->bs->file, offset, acb->cur_qiov.size,
10480f714ec7SKevin Wolf                            &acb->cur_qiov, 0);
1049eabba580SStefan Hajnoczi }
1050eabba580SStefan Hajnoczi 
1051eabba580SStefan Hajnoczi /**
1052b4ac32f3SKevin Wolf  * Populate untouched regions of new data cluster
10531f01e50bSPaolo Bonzini  *
10541f01e50bSPaolo Bonzini  * Called with table_lock held.
1055eabba580SStefan Hajnoczi  */
105687f0d882SKevin Wolf static int coroutine_fn qed_aio_write_cow(QEDAIOCB *acb)
1057eabba580SStefan Hajnoczi {
1058eabba580SStefan Hajnoczi     BDRVQEDState *s = acb_to_s(acb);
1059b4ac32f3SKevin Wolf     uint64_t start, len, offset;
1060a101341aSKevin Wolf     int ret;
1061eabba580SStefan Hajnoczi 
10621f01e50bSPaolo Bonzini     qemu_co_mutex_unlock(&s->table_lock);
10631f01e50bSPaolo Bonzini 
1064b4ac32f3SKevin Wolf     /* Populate front untouched region of new data cluster */
1065b4ac32f3SKevin Wolf     start = qed_start_of_cluster(s, acb->cur_pos);
1066b4ac32f3SKevin Wolf     len = qed_offset_into_cluster(s, acb->cur_pos);
1067b4ac32f3SKevin Wolf 
1068b4ac32f3SKevin Wolf     trace_qed_aio_write_prefill(s, acb, start, len, acb->cur_cluster);
1069b4ac32f3SKevin Wolf     ret = qed_copy_from_backing_file(s, start, len, acb->cur_cluster);
1070a101341aSKevin Wolf     if (ret < 0) {
10711f01e50bSPaolo Bonzini         goto out;
1072eabba580SStefan Hajnoczi     }
1073eabba580SStefan Hajnoczi 
1074b4ac32f3SKevin Wolf     /* Populate back untouched region of new data cluster */
1075b4ac32f3SKevin Wolf     start = acb->cur_pos + acb->cur_qiov.size;
1076b4ac32f3SKevin Wolf     len = qed_start_of_cluster(s, start + s->header.cluster_size - 1) - start;
1077b4ac32f3SKevin Wolf     offset = acb->cur_cluster +
1078b4ac32f3SKevin Wolf              qed_offset_into_cluster(s, acb->cur_pos) +
1079b4ac32f3SKevin Wolf              acb->cur_qiov.size;
1080b4ac32f3SKevin Wolf 
1081eabba580SStefan Hajnoczi     trace_qed_aio_write_postfill(s, acb, start, len, offset);
1082b4ac32f3SKevin Wolf     ret = qed_copy_from_backing_file(s, start, len, offset);
1083a101341aSKevin Wolf     if (ret < 0) {
10841f01e50bSPaolo Bonzini         goto out;
1085eaf0bc56SKevin Wolf     }
1086eabba580SStefan Hajnoczi 
1087e7569c18SPaolo Bonzini     ret = qed_aio_write_main(acb);
1088e7569c18SPaolo Bonzini     if (ret < 0) {
10891f01e50bSPaolo Bonzini         goto out;
1090e7569c18SPaolo Bonzini     }
1091e7569c18SPaolo Bonzini 
1092e7569c18SPaolo Bonzini     if (s->bs->backing) {
1093e7569c18SPaolo Bonzini         /*
1094e7569c18SPaolo Bonzini          * Flush new data clusters before updating the L2 table
1095e7569c18SPaolo Bonzini          *
1096e7569c18SPaolo Bonzini          * This flush is necessary when a backing file is in use.  A crash
1097e7569c18SPaolo Bonzini          * during an allocating write could result in empty clusters in the
1098e7569c18SPaolo Bonzini          * image.  If the write only touched a subregion of the cluster,
1099e7569c18SPaolo Bonzini          * then backing image sectors have been lost in the untouched
1100e7569c18SPaolo Bonzini          * region.  The solution is to flush after writing a new data
1101e7569c18SPaolo Bonzini          * cluster and before updating the L2 table.
1102e7569c18SPaolo Bonzini          */
1103e7569c18SPaolo Bonzini         ret = bdrv_co_flush(s->bs->file->bs);
1104e7569c18SPaolo Bonzini     }
1105e7569c18SPaolo Bonzini 
11061f01e50bSPaolo Bonzini out:
11071f01e50bSPaolo Bonzini     qemu_co_mutex_lock(&s->table_lock);
11081f01e50bSPaolo Bonzini     return ret;
1109eabba580SStefan Hajnoczi }
1110eabba580SStefan Hajnoczi 
1111eabba580SStefan Hajnoczi /**
11120d09c797SStefan Hajnoczi  * Check if the QED_F_NEED_CHECK bit should be set during allocating write
11130d09c797SStefan Hajnoczi  */
11140d09c797SStefan Hajnoczi static bool qed_should_set_need_check(BDRVQEDState *s)
11150d09c797SStefan Hajnoczi {
11160d09c797SStefan Hajnoczi     /* The flush before L2 update path ensures consistency */
1117760e0063SKevin Wolf     if (s->bs->backing) {
11180d09c797SStefan Hajnoczi         return false;
11190d09c797SStefan Hajnoczi     }
11200d09c797SStefan Hajnoczi 
11210d09c797SStefan Hajnoczi     return !(s->header.features & QED_F_NEED_CHECK);
11220d09c797SStefan Hajnoczi }
11230d09c797SStefan Hajnoczi 
11240d09c797SStefan Hajnoczi /**
1125eabba580SStefan Hajnoczi  * Write new data cluster
1126eabba580SStefan Hajnoczi  *
1127eabba580SStefan Hajnoczi  * @acb:        Write request
1128eabba580SStefan Hajnoczi  * @len:        Length in bytes
1129eabba580SStefan Hajnoczi  *
1130eabba580SStefan Hajnoczi  * This path is taken when writing to previously unallocated clusters.
11311f01e50bSPaolo Bonzini  *
11321f01e50bSPaolo Bonzini  * Called with table_lock held.
1133eabba580SStefan Hajnoczi  */
113487f0d882SKevin Wolf static int coroutine_fn qed_aio_write_alloc(QEDAIOCB *acb, size_t len)
1135eabba580SStefan Hajnoczi {
1136eabba580SStefan Hajnoczi     BDRVQEDState *s = acb_to_s(acb);
1137f13d712bSKevin Wolf     int ret;
1138eabba580SStefan Hajnoczi 
11396f321e93SStefan Hajnoczi     /* Cancel timer when the first allocating request comes in */
11400806c3b5SKevin Wolf     if (s->allocating_acb == NULL) {
11416f321e93SStefan Hajnoczi         qed_cancel_need_check_timer(s);
11426f321e93SStefan Hajnoczi     }
11436f321e93SStefan Hajnoczi 
1144eabba580SStefan Hajnoczi     /* Freeze this request if another allocating write is in progress */
11450806c3b5SKevin Wolf     if (s->allocating_acb != acb || s->allocating_write_reqs_plugged) {
11460806c3b5SKevin Wolf         if (s->allocating_acb != NULL) {
11471f01e50bSPaolo Bonzini             qemu_co_queue_wait(&s->allocating_write_reqs, &s->table_lock);
11480806c3b5SKevin Wolf             assert(s->allocating_acb == NULL);
1149eabba580SStefan Hajnoczi         }
11500806c3b5SKevin Wolf         s->allocating_acb = acb;
11510806c3b5SKevin Wolf         return -EAGAIN; /* start over with looking up table entries */
1152eabba580SStefan Hajnoczi     }
1153eabba580SStefan Hajnoczi 
1154eabba580SStefan Hajnoczi     acb->cur_nclusters = qed_bytes_to_clusters(s,
1155eabba580SStefan Hajnoczi             qed_offset_into_cluster(s, acb->cur_pos) + len);
11561b093c48SMichael Tokarev     qemu_iovec_concat(&acb->cur_qiov, acb->qiov, acb->qiov_offset, len);
1157eabba580SStefan Hajnoczi 
11580e71be19SStefan Hajnoczi     if (acb->flags & QED_AIOCB_ZERO) {
11590e71be19SStefan Hajnoczi         /* Skip ahead if the clusters are already zero */
11600e71be19SStefan Hajnoczi         if (acb->find_cluster_ret == QED_CLUSTER_ZERO) {
1161d6daddcdSKevin Wolf             return 0;
11620e71be19SStefan Hajnoczi         }
1163e7569c18SPaolo Bonzini         acb->cur_cluster = 1;
11640e71be19SStefan Hajnoczi     } else {
11650e71be19SStefan Hajnoczi         acb->cur_cluster = qed_alloc_clusters(s, acb->cur_nclusters);
11660e71be19SStefan Hajnoczi     }
11670e71be19SStefan Hajnoczi 
11680d09c797SStefan Hajnoczi     if (qed_should_set_need_check(s)) {
116901979a98SStefan Hajnoczi         s->header.features |= QED_F_NEED_CHECK;
1170f13d712bSKevin Wolf         ret = qed_write_header(s);
1171a101341aSKevin Wolf         if (ret < 0) {
1172d6daddcdSKevin Wolf             return ret;
11730d09c797SStefan Hajnoczi         }
1174eabba580SStefan Hajnoczi     }
1175eabba580SStefan Hajnoczi 
1176e7569c18SPaolo Bonzini     if (!(acb->flags & QED_AIOCB_ZERO)) {
1177a101341aSKevin Wolf         ret = qed_aio_write_cow(acb);
1178a101341aSKevin Wolf         if (ret < 0) {
1179d6daddcdSKevin Wolf             return ret;
1180a101341aSKevin Wolf         }
1181e7569c18SPaolo Bonzini     }
1182e7569c18SPaolo Bonzini 
1183e7569c18SPaolo Bonzini     return qed_aio_write_l2_update(acb, acb->cur_cluster);
1184a101341aSKevin Wolf }
1185a101341aSKevin Wolf 
1186eabba580SStefan Hajnoczi /**
1187eabba580SStefan Hajnoczi  * Write data cluster in place
1188eabba580SStefan Hajnoczi  *
1189eabba580SStefan Hajnoczi  * @acb:        Write request
1190eabba580SStefan Hajnoczi  * @offset:     Cluster offset in bytes
1191eabba580SStefan Hajnoczi  * @len:        Length in bytes
1192eabba580SStefan Hajnoczi  *
1193eabba580SStefan Hajnoczi  * This path is taken when writing to already allocated clusters.
11941f01e50bSPaolo Bonzini  *
11951f01e50bSPaolo Bonzini  * Called with table_lock held.
1196eabba580SStefan Hajnoczi  */
119787f0d882SKevin Wolf static int coroutine_fn qed_aio_write_inplace(QEDAIOCB *acb, uint64_t offset,
119887f0d882SKevin Wolf                                               size_t len)
1199eabba580SStefan Hajnoczi {
12001f01e50bSPaolo Bonzini     BDRVQEDState *s = acb_to_s(acb);
12011f01e50bSPaolo Bonzini     int r;
12021f01e50bSPaolo Bonzini 
12031f01e50bSPaolo Bonzini     qemu_co_mutex_unlock(&s->table_lock);
12041f01e50bSPaolo Bonzini 
12050e71be19SStefan Hajnoczi     /* Allocate buffer for zero writes */
12060e71be19SStefan Hajnoczi     if (acb->flags & QED_AIOCB_ZERO) {
12070e71be19SStefan Hajnoczi         struct iovec *iov = acb->qiov->iov;
12080e71be19SStefan Hajnoczi 
12090e71be19SStefan Hajnoczi         if (!iov->iov_base) {
121048cc565eSKevin Wolf             iov->iov_base = qemu_try_blockalign(acb->bs, iov->iov_len);
12114f4896dbSKevin Wolf             if (iov->iov_base == NULL) {
12121f01e50bSPaolo Bonzini                 r = -ENOMEM;
12131f01e50bSPaolo Bonzini                 goto out;
12144f4896dbSKevin Wolf             }
12150e71be19SStefan Hajnoczi             memset(iov->iov_base, 0, iov->iov_len);
12160e71be19SStefan Hajnoczi         }
12170e71be19SStefan Hajnoczi     }
12180e71be19SStefan Hajnoczi 
1219eabba580SStefan Hajnoczi     /* Calculate the I/O vector */
1220eabba580SStefan Hajnoczi     acb->cur_cluster = offset;
12211b093c48SMichael Tokarev     qemu_iovec_concat(&acb->cur_qiov, acb->qiov, acb->qiov_offset, len);
1222eabba580SStefan Hajnoczi 
12231f01e50bSPaolo Bonzini     /* Do the actual write.  */
12241f01e50bSPaolo Bonzini     r = qed_aio_write_main(acb);
12251f01e50bSPaolo Bonzini out:
12261f01e50bSPaolo Bonzini     qemu_co_mutex_lock(&s->table_lock);
12271f01e50bSPaolo Bonzini     return r;
1228eabba580SStefan Hajnoczi }
1229eabba580SStefan Hajnoczi 
1230eabba580SStefan Hajnoczi /**
1231eabba580SStefan Hajnoczi  * Write data cluster
1232eabba580SStefan Hajnoczi  *
1233eabba580SStefan Hajnoczi  * @opaque:     Write request
12340596be7eSKevin Wolf  * @ret:        QED_CLUSTER_FOUND, QED_CLUSTER_L2 or QED_CLUSTER_L1
1235eabba580SStefan Hajnoczi  * @offset:     Cluster offset in bytes
1236eabba580SStefan Hajnoczi  * @len:        Length in bytes
12371f01e50bSPaolo Bonzini  *
12381f01e50bSPaolo Bonzini  * Called with table_lock held.
1239eabba580SStefan Hajnoczi  */
124087f0d882SKevin Wolf static int coroutine_fn qed_aio_write_data(void *opaque, int ret,
1241eabba580SStefan Hajnoczi                                            uint64_t offset, size_t len)
1242eabba580SStefan Hajnoczi {
1243eabba580SStefan Hajnoczi     QEDAIOCB *acb = opaque;
1244eabba580SStefan Hajnoczi 
1245eabba580SStefan Hajnoczi     trace_qed_aio_write_data(acb_to_s(acb), acb, ret, offset, len);
1246eabba580SStefan Hajnoczi 
1247eabba580SStefan Hajnoczi     acb->find_cluster_ret = ret;
1248eabba580SStefan Hajnoczi 
1249eabba580SStefan Hajnoczi     switch (ret) {
1250eabba580SStefan Hajnoczi     case QED_CLUSTER_FOUND:
12510596be7eSKevin Wolf         return qed_aio_write_inplace(acb, offset, len);
1252eabba580SStefan Hajnoczi 
1253eabba580SStefan Hajnoczi     case QED_CLUSTER_L2:
1254eabba580SStefan Hajnoczi     case QED_CLUSTER_L1:
125521df65b6SAnthony Liguori     case QED_CLUSTER_ZERO:
12560596be7eSKevin Wolf         return qed_aio_write_alloc(acb, len);
1257eabba580SStefan Hajnoczi 
1258eabba580SStefan Hajnoczi     default:
12590596be7eSKevin Wolf         g_assert_not_reached();
1260eabba580SStefan Hajnoczi     }
1261eabba580SStefan Hajnoczi }
1262eabba580SStefan Hajnoczi 
1263eabba580SStefan Hajnoczi /**
1264eabba580SStefan Hajnoczi  * Read data cluster
1265eabba580SStefan Hajnoczi  *
1266eabba580SStefan Hajnoczi  * @opaque:     Read request
12670596be7eSKevin Wolf  * @ret:        QED_CLUSTER_FOUND, QED_CLUSTER_L2 or QED_CLUSTER_L1
1268eabba580SStefan Hajnoczi  * @offset:     Cluster offset in bytes
1269eabba580SStefan Hajnoczi  * @len:        Length in bytes
12701f01e50bSPaolo Bonzini  *
12711f01e50bSPaolo Bonzini  * Called with table_lock held.
1272eabba580SStefan Hajnoczi  */
127387f0d882SKevin Wolf static int coroutine_fn qed_aio_read_data(void *opaque, int ret,
127487f0d882SKevin Wolf                                           uint64_t offset, size_t len)
1275eabba580SStefan Hajnoczi {
1276eabba580SStefan Hajnoczi     QEDAIOCB *acb = opaque;
1277eabba580SStefan Hajnoczi     BDRVQEDState *s = acb_to_s(acb);
127848cc565eSKevin Wolf     BlockDriverState *bs = acb->bs;
12791f01e50bSPaolo Bonzini     int r;
12801f01e50bSPaolo Bonzini 
12811f01e50bSPaolo Bonzini     qemu_co_mutex_unlock(&s->table_lock);
1282eabba580SStefan Hajnoczi 
1283eabba580SStefan Hajnoczi     /* Adjust offset into cluster */
1284eabba580SStefan Hajnoczi     offset += qed_offset_into_cluster(s, acb->cur_pos);
1285eabba580SStefan Hajnoczi 
1286eabba580SStefan Hajnoczi     trace_qed_aio_read_data(s, acb, ret, offset, len);
1287eabba580SStefan Hajnoczi 
12881b093c48SMichael Tokarev     qemu_iovec_concat(&acb->cur_qiov, acb->qiov, acb->qiov_offset, len);
1289eabba580SStefan Hajnoczi 
12901f01e50bSPaolo Bonzini     /* Handle zero cluster and backing file reads, otherwise read
12911f01e50bSPaolo Bonzini      * data cluster directly.
12921f01e50bSPaolo Bonzini      */
129321df65b6SAnthony Liguori     if (ret == QED_CLUSTER_ZERO) {
12943d9b4925SMichael Tokarev         qemu_iovec_memset(&acb->cur_qiov, 0, 0, acb->cur_qiov.size);
12951f01e50bSPaolo Bonzini         r = 0;
129621df65b6SAnthony Liguori     } else if (ret != QED_CLUSTER_FOUND) {
1297*365fed51SEric Blake         r = qed_read_backing_file(s, acb->cur_pos, &acb->cur_qiov);
12981f01e50bSPaolo Bonzini     } else {
12991f01e50bSPaolo Bonzini         BLKDBG_EVENT(bs->file, BLKDBG_READ_AIO);
13001f01e50bSPaolo Bonzini         r = bdrv_co_preadv(bs->file, offset, acb->cur_qiov.size,
13011f01e50bSPaolo Bonzini                            &acb->cur_qiov, 0);
1302eabba580SStefan Hajnoczi     }
1303eabba580SStefan Hajnoczi 
13041f01e50bSPaolo Bonzini     qemu_co_mutex_lock(&s->table_lock);
13051f01e50bSPaolo Bonzini     return r;
1306eabba580SStefan Hajnoczi }
1307eabba580SStefan Hajnoczi 
1308eabba580SStefan Hajnoczi /**
1309eabba580SStefan Hajnoczi  * Begin next I/O or complete the request
1310eabba580SStefan Hajnoczi  */
131187f0d882SKevin Wolf static int coroutine_fn qed_aio_next_io(QEDAIOCB *acb)
1312eabba580SStefan Hajnoczi {
1313eabba580SStefan Hajnoczi     BDRVQEDState *s = acb_to_s(acb);
13140f21b7a1SKevin Wolf     uint64_t offset;
13150f21b7a1SKevin Wolf     size_t len;
1316dddf8db1SKevin Wolf     int ret;
1317eabba580SStefan Hajnoczi 
13181f01e50bSPaolo Bonzini     qemu_co_mutex_lock(&s->table_lock);
131901859874SKevin Wolf     while (1) {
1320dddf8db1SKevin Wolf         trace_qed_aio_next_io(s, acb, 0, acb->cur_pos + acb->cur_qiov.size);
1321eabba580SStefan Hajnoczi 
1322eabba580SStefan Hajnoczi         acb->qiov_offset += acb->cur_qiov.size;
1323eabba580SStefan Hajnoczi         acb->cur_pos += acb->cur_qiov.size;
1324eabba580SStefan Hajnoczi         qemu_iovec_reset(&acb->cur_qiov);
1325eabba580SStefan Hajnoczi 
1326eabba580SStefan Hajnoczi         /* Complete request */
1327eabba580SStefan Hajnoczi         if (acb->cur_pos >= acb->end_pos) {
132848cc565eSKevin Wolf             ret = 0;
132948cc565eSKevin Wolf             break;
1330eabba580SStefan Hajnoczi         }
1331eabba580SStefan Hajnoczi 
1332eabba580SStefan Hajnoczi         /* Find next cluster and start I/O */
13330f21b7a1SKevin Wolf         len = acb->end_pos - acb->cur_pos;
13340f21b7a1SKevin Wolf         ret = qed_find_cluster(s, &acb->request, acb->cur_pos, &len, &offset);
13350596be7eSKevin Wolf         if (ret < 0) {
133648cc565eSKevin Wolf             break;
13370596be7eSKevin Wolf         }
13380596be7eSKevin Wolf 
13390596be7eSKevin Wolf         if (acb->flags & QED_AIOCB_WRITE) {
13400596be7eSKevin Wolf             ret = qed_aio_write_data(acb, ret, offset, len);
13410596be7eSKevin Wolf         } else {
13420596be7eSKevin Wolf             ret = qed_aio_read_data(acb, ret, offset, len);
13430596be7eSKevin Wolf         }
13440596be7eSKevin Wolf 
13450806c3b5SKevin Wolf         if (ret < 0 && ret != -EAGAIN) {
134648cc565eSKevin Wolf             break;
134701859874SKevin Wolf         }
1348eabba580SStefan Hajnoczi     }
1349eabba580SStefan Hajnoczi 
135048cc565eSKevin Wolf     trace_qed_aio_complete(s, acb, ret);
135148cc565eSKevin Wolf     qed_aio_complete(acb);
13521f01e50bSPaolo Bonzini     qemu_co_mutex_unlock(&s->table_lock);
135348cc565eSKevin Wolf     return ret;
135489f89709SKevin Wolf }
135589f89709SKevin Wolf 
135689f89709SKevin Wolf static int coroutine_fn qed_co_request(BlockDriverState *bs, int64_t sector_num,
135789f89709SKevin Wolf                                        QEMUIOVector *qiov, int nb_sectors,
135889f89709SKevin Wolf                                        int flags)
135989f89709SKevin Wolf {
136048cc565eSKevin Wolf     QEDAIOCB acb = {
136148cc565eSKevin Wolf         .bs         = bs,
136248cc565eSKevin Wolf         .cur_pos    = (uint64_t) sector_num * BDRV_SECTOR_SIZE,
136348cc565eSKevin Wolf         .end_pos    = (sector_num + nb_sectors) * BDRV_SECTOR_SIZE,
136448cc565eSKevin Wolf         .qiov       = qiov,
136548cc565eSKevin Wolf         .flags      = flags,
136689f89709SKevin Wolf     };
136748cc565eSKevin Wolf     qemu_iovec_init(&acb.cur_qiov, qiov->niov);
136889f89709SKevin Wolf 
136948cc565eSKevin Wolf     trace_qed_aio_setup(bs->opaque, &acb, sector_num, nb_sectors, NULL, flags);
1370eabba580SStefan Hajnoczi 
1371eabba580SStefan Hajnoczi     /* Start request */
137248cc565eSKevin Wolf     return qed_aio_next_io(&acb);
137389f89709SKevin Wolf }
137489f89709SKevin Wolf 
137589f89709SKevin Wolf static int coroutine_fn bdrv_qed_co_readv(BlockDriverState *bs,
137689f89709SKevin Wolf                                           int64_t sector_num, int nb_sectors,
137789f89709SKevin Wolf                                           QEMUIOVector *qiov)
137875411d23SStefan Hajnoczi {
137989f89709SKevin Wolf     return qed_co_request(bs, sector_num, qiov, nb_sectors, 0);
138075411d23SStefan Hajnoczi }
138175411d23SStefan Hajnoczi 
138289f89709SKevin Wolf static int coroutine_fn bdrv_qed_co_writev(BlockDriverState *bs,
138389f89709SKevin Wolf                                            int64_t sector_num, int nb_sectors,
1384e18a58b4SEric Blake                                            QEMUIOVector *qiov, int flags)
138575411d23SStefan Hajnoczi {
1386e18a58b4SEric Blake     assert(!flags);
138789f89709SKevin Wolf     return qed_co_request(bs, sector_num, qiov, nb_sectors, QED_AIOCB_WRITE);
13880e71be19SStefan Hajnoczi }
13890e71be19SStefan Hajnoczi 
139049a2e483SEric Blake static int coroutine_fn bdrv_qed_co_pwrite_zeroes(BlockDriverState *bs,
139149a2e483SEric Blake                                                   int64_t offset,
1392f5a5ca79SManos Pitsidianakis                                                   int bytes,
1393aa7bfbffSPeter Lieven                                                   BdrvRequestFlags flags)
13940e71be19SStefan Hajnoczi {
1395ef72f76eSStefan Hajnoczi     BDRVQEDState *s = bs->opaque;
1396342544f9SVladimir Sementsov-Ogievskiy 
1397342544f9SVladimir Sementsov-Ogievskiy     /*
1398342544f9SVladimir Sementsov-Ogievskiy      * Zero writes start without an I/O buffer.  If a buffer becomes necessary
1399342544f9SVladimir Sementsov-Ogievskiy      * then it will be allocated during request processing.
1400342544f9SVladimir Sementsov-Ogievskiy      */
1401342544f9SVladimir Sementsov-Ogievskiy     QEMUIOVector qiov = QEMU_IOVEC_INIT_BUF(qiov, NULL, bytes);
14020e71be19SStefan Hajnoczi 
140349a2e483SEric Blake     /* Fall back if the request is not aligned */
140449a2e483SEric Blake     if (qed_offset_into_cluster(s, offset) ||
1405f5a5ca79SManos Pitsidianakis         qed_offset_into_cluster(s, bytes)) {
1406ef72f76eSStefan Hajnoczi         return -ENOTSUP;
1407ef72f76eSStefan Hajnoczi     }
1408ef72f76eSStefan Hajnoczi 
140989f89709SKevin Wolf     return qed_co_request(bs, offset >> BDRV_SECTOR_BITS, &qiov,
1410f5a5ca79SManos Pitsidianakis                           bytes >> BDRV_SECTOR_BITS,
14110e71be19SStefan Hajnoczi                           QED_AIOCB_WRITE | QED_AIOCB_ZERO);
14120e71be19SStefan Hajnoczi }
14130e71be19SStefan Hajnoczi 
1414061ca8a3SKevin Wolf static int coroutine_fn bdrv_qed_co_truncate(BlockDriverState *bs,
1415061ca8a3SKevin Wolf                                              int64_t offset,
1416c80d8b06SMax Reitz                                              bool exact,
1417061ca8a3SKevin Wolf                                              PreallocMode prealloc,
141892b92799SKevin Wolf                                              BdrvRequestFlags flags,
1419061ca8a3SKevin Wolf                                              Error **errp)
142075411d23SStefan Hajnoczi {
142177a5a000SStefan Hajnoczi     BDRVQEDState *s = bs->opaque;
142277a5a000SStefan Hajnoczi     uint64_t old_image_size;
142377a5a000SStefan Hajnoczi     int ret;
142477a5a000SStefan Hajnoczi 
14258243ccb7SMax Reitz     if (prealloc != PREALLOC_MODE_OFF) {
14268243ccb7SMax Reitz         error_setg(errp, "Unsupported preallocation mode '%s'",
1427977c736fSMarkus Armbruster                    PreallocMode_str(prealloc));
14288243ccb7SMax Reitz         return -ENOTSUP;
14298243ccb7SMax Reitz     }
14308243ccb7SMax Reitz 
143177a5a000SStefan Hajnoczi     if (!qed_is_image_size_valid(offset, s->header.cluster_size,
143277a5a000SStefan Hajnoczi                                  s->header.table_size)) {
1433f59adb32SMax Reitz         error_setg(errp, "Invalid image size specified");
143477a5a000SStefan Hajnoczi         return -EINVAL;
143577a5a000SStefan Hajnoczi     }
143677a5a000SStefan Hajnoczi 
143777a5a000SStefan Hajnoczi     if ((uint64_t)offset < s->header.image_size) {
1438f59adb32SMax Reitz         error_setg(errp, "Shrinking images is currently not supported");
143975411d23SStefan Hajnoczi         return -ENOTSUP;
144075411d23SStefan Hajnoczi     }
144175411d23SStefan Hajnoczi 
144277a5a000SStefan Hajnoczi     old_image_size = s->header.image_size;
144377a5a000SStefan Hajnoczi     s->header.image_size = offset;
144477a5a000SStefan Hajnoczi     ret = qed_write_header_sync(s);
144577a5a000SStefan Hajnoczi     if (ret < 0) {
144677a5a000SStefan Hajnoczi         s->header.image_size = old_image_size;
1447f59adb32SMax Reitz         error_setg_errno(errp, -ret, "Failed to update the image size");
144877a5a000SStefan Hajnoczi     }
144977a5a000SStefan Hajnoczi     return ret;
145077a5a000SStefan Hajnoczi }
145177a5a000SStefan Hajnoczi 
145275411d23SStefan Hajnoczi static int64_t bdrv_qed_getlength(BlockDriverState *bs)
145375411d23SStefan Hajnoczi {
145475411d23SStefan Hajnoczi     BDRVQEDState *s = bs->opaque;
145575411d23SStefan Hajnoczi     return s->header.image_size;
145675411d23SStefan Hajnoczi }
145775411d23SStefan Hajnoczi 
145875411d23SStefan Hajnoczi static int bdrv_qed_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
145975411d23SStefan Hajnoczi {
146075411d23SStefan Hajnoczi     BDRVQEDState *s = bs->opaque;
146175411d23SStefan Hajnoczi 
146275411d23SStefan Hajnoczi     memset(bdi, 0, sizeof(*bdi));
146375411d23SStefan Hajnoczi     bdi->cluster_size = s->header.cluster_size;
1464d68dbee8SDong Xu Wang     bdi->is_dirty = s->header.features & QED_F_NEED_CHECK;
146575411d23SStefan Hajnoczi     return 0;
146675411d23SStefan Hajnoczi }
146775411d23SStefan Hajnoczi 
146875411d23SStefan Hajnoczi static int bdrv_qed_change_backing_file(BlockDriverState *bs,
146975411d23SStefan Hajnoczi                                         const char *backing_file,
147075411d23SStefan Hajnoczi                                         const char *backing_fmt)
147175411d23SStefan Hajnoczi {
147275411d23SStefan Hajnoczi     BDRVQEDState *s = bs->opaque;
147375411d23SStefan Hajnoczi     QEDHeader new_header, le_header;
147475411d23SStefan Hajnoczi     void *buffer;
147575411d23SStefan Hajnoczi     size_t buffer_len, backing_file_len;
147675411d23SStefan Hajnoczi     int ret;
147775411d23SStefan Hajnoczi 
147875411d23SStefan Hajnoczi     /* Refuse to set backing filename if unknown compat feature bits are
147975411d23SStefan Hajnoczi      * active.  If the image uses an unknown compat feature then we may not
148075411d23SStefan Hajnoczi      * know the layout of data following the header structure and cannot safely
148175411d23SStefan Hajnoczi      * add a new string.
148275411d23SStefan Hajnoczi      */
148375411d23SStefan Hajnoczi     if (backing_file && (s->header.compat_features &
148475411d23SStefan Hajnoczi                          ~QED_COMPAT_FEATURE_MASK)) {
148575411d23SStefan Hajnoczi         return -ENOTSUP;
148675411d23SStefan Hajnoczi     }
148775411d23SStefan Hajnoczi 
148875411d23SStefan Hajnoczi     memcpy(&new_header, &s->header, sizeof(new_header));
148975411d23SStefan Hajnoczi 
149075411d23SStefan Hajnoczi     new_header.features &= ~(QED_F_BACKING_FILE |
149175411d23SStefan Hajnoczi                              QED_F_BACKING_FORMAT_NO_PROBE);
149275411d23SStefan Hajnoczi 
149375411d23SStefan Hajnoczi     /* Adjust feature flags */
149475411d23SStefan Hajnoczi     if (backing_file) {
149575411d23SStefan Hajnoczi         new_header.features |= QED_F_BACKING_FILE;
149675411d23SStefan Hajnoczi 
149775411d23SStefan Hajnoczi         if (qed_fmt_is_raw(backing_fmt)) {
149875411d23SStefan Hajnoczi             new_header.features |= QED_F_BACKING_FORMAT_NO_PROBE;
149975411d23SStefan Hajnoczi         }
150075411d23SStefan Hajnoczi     }
150175411d23SStefan Hajnoczi 
150275411d23SStefan Hajnoczi     /* Calculate new header size */
150375411d23SStefan Hajnoczi     backing_file_len = 0;
150475411d23SStefan Hajnoczi 
150575411d23SStefan Hajnoczi     if (backing_file) {
150675411d23SStefan Hajnoczi         backing_file_len = strlen(backing_file);
150775411d23SStefan Hajnoczi     }
150875411d23SStefan Hajnoczi 
150975411d23SStefan Hajnoczi     buffer_len = sizeof(new_header);
151075411d23SStefan Hajnoczi     new_header.backing_filename_offset = buffer_len;
151175411d23SStefan Hajnoczi     new_header.backing_filename_size = backing_file_len;
151275411d23SStefan Hajnoczi     buffer_len += backing_file_len;
151375411d23SStefan Hajnoczi 
151475411d23SStefan Hajnoczi     /* Make sure we can rewrite header without failing */
151575411d23SStefan Hajnoczi     if (buffer_len > new_header.header_size * new_header.cluster_size) {
151675411d23SStefan Hajnoczi         return -ENOSPC;
151775411d23SStefan Hajnoczi     }
151875411d23SStefan Hajnoczi 
151975411d23SStefan Hajnoczi     /* Prepare new header */
15207267c094SAnthony Liguori     buffer = g_malloc(buffer_len);
152175411d23SStefan Hajnoczi 
152275411d23SStefan Hajnoczi     qed_header_cpu_to_le(&new_header, &le_header);
152375411d23SStefan Hajnoczi     memcpy(buffer, &le_header, sizeof(le_header));
152475411d23SStefan Hajnoczi     buffer_len = sizeof(le_header);
152575411d23SStefan Hajnoczi 
1526feba23b1SPavel Borzenkov     if (backing_file) {
152775411d23SStefan Hajnoczi         memcpy(buffer + buffer_len, backing_file, backing_file_len);
152875411d23SStefan Hajnoczi         buffer_len += backing_file_len;
1529feba23b1SPavel Borzenkov     }
153075411d23SStefan Hajnoczi 
153175411d23SStefan Hajnoczi     /* Write new header */
1532d9ca2ea2SKevin Wolf     ret = bdrv_pwrite_sync(bs->file, 0, buffer, buffer_len);
15337267c094SAnthony Liguori     g_free(buffer);
153475411d23SStefan Hajnoczi     if (ret == 0) {
153575411d23SStefan Hajnoczi         memcpy(&s->header, &new_header, sizeof(new_header));
153675411d23SStefan Hajnoczi     }
153775411d23SStefan Hajnoczi     return ret;
153875411d23SStefan Hajnoczi }
153975411d23SStefan Hajnoczi 
15402b148f39SPaolo Bonzini static void coroutine_fn bdrv_qed_co_invalidate_cache(BlockDriverState *bs,
15412b148f39SPaolo Bonzini                                                       Error **errp)
1542c82954e5SBenoît Canet {
15431f01e50bSPaolo Bonzini     BDRVQEDState *s = bs->opaque;
15445a8a30dbSKevin Wolf     Error *local_err = NULL;
15455a8a30dbSKevin Wolf     int ret;
1546c82954e5SBenoît Canet 
1547c82954e5SBenoît Canet     bdrv_qed_close(bs);
15483456a8d1SKevin Wolf 
154961c7887eSPaolo Bonzini     bdrv_qed_init_state(bs);
15501f01e50bSPaolo Bonzini     qemu_co_mutex_lock(&s->table_lock);
15514e4bf5c4SKevin Wolf     ret = bdrv_qed_do_open(bs, NULL, bs->open_flags, &local_err);
15521f01e50bSPaolo Bonzini     qemu_co_mutex_unlock(&s->table_lock);
15535a8a30dbSKevin Wolf     if (local_err) {
15544b576648SMarkus Armbruster         error_propagate_prepend(errp, local_err,
15554b576648SMarkus Armbruster                                 "Could not reopen qed layer: ");
15565a8a30dbSKevin Wolf         return;
15575a8a30dbSKevin Wolf     } else if (ret < 0) {
15585a8a30dbSKevin Wolf         error_setg_errno(errp, -ret, "Could not reopen qed layer");
15595a8a30dbSKevin Wolf         return;
15605a8a30dbSKevin Wolf     }
1561c82954e5SBenoît Canet }
1562c82954e5SBenoît Canet 
156354277a2aSVladimir Sementsov-Ogievskiy static int coroutine_fn bdrv_qed_co_check(BlockDriverState *bs,
156454277a2aSVladimir Sementsov-Ogievskiy                                           BdrvCheckResult *result,
15654534ff54SKevin Wolf                                           BdrvCheckMode fix)
156675411d23SStefan Hajnoczi {
156701979a98SStefan Hajnoczi     BDRVQEDState *s = bs->opaque;
15682fd61638SPaolo Bonzini     int ret;
156901979a98SStefan Hajnoczi 
15702fd61638SPaolo Bonzini     qemu_co_mutex_lock(&s->table_lock);
15712fd61638SPaolo Bonzini     ret = qed_check(s, result, !!fix);
15722fd61638SPaolo Bonzini     qemu_co_mutex_unlock(&s->table_lock);
15732fd61638SPaolo Bonzini 
15742fd61638SPaolo Bonzini     return ret;
157575411d23SStefan Hajnoczi }
157675411d23SStefan Hajnoczi 
15777ab74849SChunyan Liu static QemuOptsList qed_create_opts = {
15787ab74849SChunyan Liu     .name = "qed-create-opts",
15797ab74849SChunyan Liu     .head = QTAILQ_HEAD_INITIALIZER(qed_create_opts.head),
15807ab74849SChunyan Liu     .desc = {
158175411d23SStefan Hajnoczi         {
158275411d23SStefan Hajnoczi             .name = BLOCK_OPT_SIZE,
15837ab74849SChunyan Liu             .type = QEMU_OPT_SIZE,
15847ab74849SChunyan Liu             .help = "Virtual disk size"
15857ab74849SChunyan Liu         },
15867ab74849SChunyan Liu         {
158775411d23SStefan Hajnoczi             .name = BLOCK_OPT_BACKING_FILE,
15887ab74849SChunyan Liu             .type = QEMU_OPT_STRING,
158975411d23SStefan Hajnoczi             .help = "File name of a base image"
15907ab74849SChunyan Liu         },
15917ab74849SChunyan Liu         {
159275411d23SStefan Hajnoczi             .name = BLOCK_OPT_BACKING_FMT,
15937ab74849SChunyan Liu             .type = QEMU_OPT_STRING,
159475411d23SStefan Hajnoczi             .help = "Image format of the base image"
15957ab74849SChunyan Liu         },
15967ab74849SChunyan Liu         {
159775411d23SStefan Hajnoczi             .name = BLOCK_OPT_CLUSTER_SIZE,
15987ab74849SChunyan Liu             .type = QEMU_OPT_SIZE,
159999cce9faSKevin Wolf             .help = "Cluster size (in bytes)",
16007ab74849SChunyan Liu             .def_value_str = stringify(QED_DEFAULT_CLUSTER_SIZE)
16017ab74849SChunyan Liu         },
16027ab74849SChunyan Liu         {
160375411d23SStefan Hajnoczi             .name = BLOCK_OPT_TABLE_SIZE,
16047ab74849SChunyan Liu             .type = QEMU_OPT_SIZE,
160575411d23SStefan Hajnoczi             .help = "L1/L2 table size (in clusters)"
160675411d23SStefan Hajnoczi         },
160775411d23SStefan Hajnoczi         { /* end of list */ }
16087ab74849SChunyan Liu     }
160975411d23SStefan Hajnoczi };
161075411d23SStefan Hajnoczi 
161175411d23SStefan Hajnoczi static BlockDriver bdrv_qed = {
161275411d23SStefan Hajnoczi     .format_name              = "qed",
161375411d23SStefan Hajnoczi     .instance_size            = sizeof(BDRVQEDState),
16147ab74849SChunyan Liu     .create_opts              = &qed_create_opts,
1615d67066d8SMax Reitz     .is_format                = true,
16168ee79e70SKevin Wolf     .supports_backing         = true,
161775411d23SStefan Hajnoczi 
161875411d23SStefan Hajnoczi     .bdrv_probe               = bdrv_qed_probe,
161975411d23SStefan Hajnoczi     .bdrv_open                = bdrv_qed_open,
162075411d23SStefan Hajnoczi     .bdrv_close               = bdrv_qed_close,
1621f9cb20f1SJeff Cody     .bdrv_reopen_prepare      = bdrv_qed_reopen_prepare,
162269dca43dSMax Reitz     .bdrv_child_perm          = bdrv_default_perms,
1623959355a4SKevin Wolf     .bdrv_co_create           = bdrv_qed_co_create,
1624efc75e2aSStefan Hajnoczi     .bdrv_co_create_opts      = bdrv_qed_co_create_opts,
16253ac21627SPeter Lieven     .bdrv_has_zero_init       = bdrv_has_zero_init_1,
1626b8d739fdSEric Blake     .bdrv_co_block_status     = bdrv_qed_co_block_status,
162789f89709SKevin Wolf     .bdrv_co_readv            = bdrv_qed_co_readv,
162889f89709SKevin Wolf     .bdrv_co_writev           = bdrv_qed_co_writev,
162949a2e483SEric Blake     .bdrv_co_pwrite_zeroes    = bdrv_qed_co_pwrite_zeroes,
1630061ca8a3SKevin Wolf     .bdrv_co_truncate         = bdrv_qed_co_truncate,
163175411d23SStefan Hajnoczi     .bdrv_getlength           = bdrv_qed_getlength,
163275411d23SStefan Hajnoczi     .bdrv_get_info            = bdrv_qed_get_info,
1633d34682cdSKevin Wolf     .bdrv_refresh_limits      = bdrv_qed_refresh_limits,
163475411d23SStefan Hajnoczi     .bdrv_change_backing_file = bdrv_qed_change_backing_file,
16352b148f39SPaolo Bonzini     .bdrv_co_invalidate_cache = bdrv_qed_co_invalidate_cache,
16362fd61638SPaolo Bonzini     .bdrv_co_check            = bdrv_qed_co_check,
1637a8c868c3SStefan Hajnoczi     .bdrv_detach_aio_context  = bdrv_qed_detach_aio_context,
1638a8c868c3SStefan Hajnoczi     .bdrv_attach_aio_context  = bdrv_qed_attach_aio_context,
1639f8ea8dacSManos Pitsidianakis     .bdrv_co_drain_begin      = bdrv_qed_co_drain_begin,
164075411d23SStefan Hajnoczi };
164175411d23SStefan Hajnoczi 
164275411d23SStefan Hajnoczi static void bdrv_qed_init(void)
164375411d23SStefan Hajnoczi {
164475411d23SStefan Hajnoczi     bdrv_register(&bdrv_qed);
164575411d23SStefan Hajnoczi }
164675411d23SStefan Hajnoczi 
164775411d23SStefan Hajnoczi block_init(bdrv_qed_init);
1648