1893f7ebaSPaolo Bonzini /* 2893f7ebaSPaolo Bonzini * Image mirroring 3893f7ebaSPaolo Bonzini * 4893f7ebaSPaolo Bonzini * Copyright Red Hat, Inc. 2012 5893f7ebaSPaolo Bonzini * 6893f7ebaSPaolo Bonzini * Authors: 7893f7ebaSPaolo Bonzini * Paolo Bonzini <pbonzini@redhat.com> 8893f7ebaSPaolo Bonzini * 9893f7ebaSPaolo Bonzini * This work is licensed under the terms of the GNU LGPL, version 2 or later. 10893f7ebaSPaolo Bonzini * See the COPYING.LIB file in the top-level directory. 11893f7ebaSPaolo Bonzini * 12893f7ebaSPaolo Bonzini */ 13893f7ebaSPaolo Bonzini 14893f7ebaSPaolo Bonzini #include "trace.h" 15737e150eSPaolo Bonzini #include "block/blockjob.h" 16737e150eSPaolo Bonzini #include "block/block_int.h" 17cc7a8ea7SMarkus Armbruster #include "qapi/qmp/qerror.h" 18893f7ebaSPaolo Bonzini #include "qemu/ratelimit.h" 19b812f671SPaolo Bonzini #include "qemu/bitmap.h" 20893f7ebaSPaolo Bonzini 21893f7ebaSPaolo Bonzini #define SLICE_TIME 100000000ULL /* ns */ 22402a4741SPaolo Bonzini #define MAX_IN_FLIGHT 16 2348ac0a4dSWen Congyang #define DEFAULT_MIRROR_BUF_SIZE (10 << 20) 24402a4741SPaolo Bonzini 25402a4741SPaolo Bonzini /* The mirroring buffer is a list of granularity-sized chunks. 26402a4741SPaolo Bonzini * Free chunks are organized in a list. 27402a4741SPaolo Bonzini */ 28402a4741SPaolo Bonzini typedef struct MirrorBuffer { 29402a4741SPaolo Bonzini QSIMPLEQ_ENTRY(MirrorBuffer) next; 30402a4741SPaolo Bonzini } MirrorBuffer; 31893f7ebaSPaolo Bonzini 32893f7ebaSPaolo Bonzini typedef struct MirrorBlockJob { 33893f7ebaSPaolo Bonzini BlockJob common; 34893f7ebaSPaolo Bonzini RateLimit limit; 35893f7ebaSPaolo Bonzini BlockDriverState *target; 365bc361b8SFam Zheng BlockDriverState *base; 3709158f00SBenoît Canet /* The name of the graph node to replace */ 3809158f00SBenoît Canet char *replaces; 3909158f00SBenoît Canet /* The BDS to replace */ 4009158f00SBenoît Canet BlockDriverState *to_replace; 4109158f00SBenoît Canet /* Used to block operations on the drive-mirror-replace target */ 4209158f00SBenoît Canet Error *replace_blocker; 4303544a6eSFam Zheng bool is_none_mode; 44b952b558SPaolo Bonzini BlockdevOnError on_source_error, on_target_error; 45d63ffd87SPaolo Bonzini bool synced; 46d63ffd87SPaolo Bonzini bool should_complete; 47893f7ebaSPaolo Bonzini int64_t sector_num; 48eee13dfeSPaolo Bonzini int64_t granularity; 49b812f671SPaolo Bonzini size_t buf_size; 50b21c7652SMax Reitz int64_t bdev_length; 51b812f671SPaolo Bonzini unsigned long *cow_bitmap; 52e4654d2dSFam Zheng BdrvDirtyBitmap *dirty_bitmap; 538f0720ecSPaolo Bonzini HBitmapIter hbi; 54893f7ebaSPaolo Bonzini uint8_t *buf; 55402a4741SPaolo Bonzini QSIMPLEQ_HEAD(, MirrorBuffer) buf_free; 56402a4741SPaolo Bonzini int buf_free_count; 57bd48bde8SPaolo Bonzini 58402a4741SPaolo Bonzini unsigned long *in_flight_bitmap; 59bd48bde8SPaolo Bonzini int in_flight; 60b21c7652SMax Reitz int sectors_in_flight; 61bd48bde8SPaolo Bonzini int ret; 620fc9f8eaSFam Zheng bool unmap; 63e424aff5SKevin Wolf bool waiting_for_io; 64893f7ebaSPaolo Bonzini } MirrorBlockJob; 65893f7ebaSPaolo Bonzini 66bd48bde8SPaolo Bonzini typedef struct MirrorOp { 67bd48bde8SPaolo Bonzini MirrorBlockJob *s; 68bd48bde8SPaolo Bonzini QEMUIOVector qiov; 69bd48bde8SPaolo Bonzini int64_t sector_num; 70bd48bde8SPaolo Bonzini int nb_sectors; 71bd48bde8SPaolo Bonzini } MirrorOp; 72bd48bde8SPaolo Bonzini 73b952b558SPaolo Bonzini static BlockErrorAction mirror_error_action(MirrorBlockJob *s, bool read, 74b952b558SPaolo Bonzini int error) 75b952b558SPaolo Bonzini { 76b952b558SPaolo Bonzini s->synced = false; 77b952b558SPaolo Bonzini if (read) { 78b952b558SPaolo Bonzini return block_job_error_action(&s->common, s->common.bs, 79b952b558SPaolo Bonzini s->on_source_error, true, error); 80b952b558SPaolo Bonzini } else { 81b952b558SPaolo Bonzini return block_job_error_action(&s->common, s->target, 82b952b558SPaolo Bonzini s->on_target_error, false, error); 83b952b558SPaolo Bonzini } 84b952b558SPaolo Bonzini } 85b952b558SPaolo Bonzini 86bd48bde8SPaolo Bonzini static void mirror_iteration_done(MirrorOp *op, int ret) 87bd48bde8SPaolo Bonzini { 88bd48bde8SPaolo Bonzini MirrorBlockJob *s = op->s; 89402a4741SPaolo Bonzini struct iovec *iov; 90bd48bde8SPaolo Bonzini int64_t chunk_num; 91402a4741SPaolo Bonzini int i, nb_chunks, sectors_per_chunk; 92bd48bde8SPaolo Bonzini 93bd48bde8SPaolo Bonzini trace_mirror_iteration_done(s, op->sector_num, op->nb_sectors, ret); 94bd48bde8SPaolo Bonzini 95bd48bde8SPaolo Bonzini s->in_flight--; 96b21c7652SMax Reitz s->sectors_in_flight -= op->nb_sectors; 97402a4741SPaolo Bonzini iov = op->qiov.iov; 98402a4741SPaolo Bonzini for (i = 0; i < op->qiov.niov; i++) { 99402a4741SPaolo Bonzini MirrorBuffer *buf = (MirrorBuffer *) iov[i].iov_base; 100402a4741SPaolo Bonzini QSIMPLEQ_INSERT_TAIL(&s->buf_free, buf, next); 101402a4741SPaolo Bonzini s->buf_free_count++; 102402a4741SPaolo Bonzini } 103402a4741SPaolo Bonzini 104bd48bde8SPaolo Bonzini sectors_per_chunk = s->granularity >> BDRV_SECTOR_BITS; 105bd48bde8SPaolo Bonzini chunk_num = op->sector_num / sectors_per_chunk; 106bd48bde8SPaolo Bonzini nb_chunks = op->nb_sectors / sectors_per_chunk; 107402a4741SPaolo Bonzini bitmap_clear(s->in_flight_bitmap, chunk_num, nb_chunks); 108b21c7652SMax Reitz if (ret >= 0) { 109b21c7652SMax Reitz if (s->cow_bitmap) { 110bd48bde8SPaolo Bonzini bitmap_set(s->cow_bitmap, chunk_num, nb_chunks); 111bd48bde8SPaolo Bonzini } 112b21c7652SMax Reitz s->common.offset += (uint64_t)op->nb_sectors * BDRV_SECTOR_SIZE; 113b21c7652SMax Reitz } 114bd48bde8SPaolo Bonzini 1156df3bf8eSZhang Min qemu_iovec_destroy(&op->qiov); 116c84b3192SPaolo Bonzini g_free(op); 1177b770c72SStefan Hajnoczi 118e424aff5SKevin Wolf if (s->waiting_for_io) { 119bd48bde8SPaolo Bonzini qemu_coroutine_enter(s->common.co, NULL); 120bd48bde8SPaolo Bonzini } 1217b770c72SStefan Hajnoczi } 122bd48bde8SPaolo Bonzini 123bd48bde8SPaolo Bonzini static void mirror_write_complete(void *opaque, int ret) 124bd48bde8SPaolo Bonzini { 125bd48bde8SPaolo Bonzini MirrorOp *op = opaque; 126bd48bde8SPaolo Bonzini MirrorBlockJob *s = op->s; 127bd48bde8SPaolo Bonzini if (ret < 0) { 128bd48bde8SPaolo Bonzini BlockErrorAction action; 129bd48bde8SPaolo Bonzini 13020dca810SJohn Snow bdrv_set_dirty_bitmap(s->dirty_bitmap, op->sector_num, op->nb_sectors); 131bd48bde8SPaolo Bonzini action = mirror_error_action(s, false, -ret); 132a589569fSWenchao Xia if (action == BLOCK_ERROR_ACTION_REPORT && s->ret >= 0) { 133bd48bde8SPaolo Bonzini s->ret = ret; 134bd48bde8SPaolo Bonzini } 135bd48bde8SPaolo Bonzini } 136bd48bde8SPaolo Bonzini mirror_iteration_done(op, ret); 137bd48bde8SPaolo Bonzini } 138bd48bde8SPaolo Bonzini 139bd48bde8SPaolo Bonzini static void mirror_read_complete(void *opaque, int ret) 140bd48bde8SPaolo Bonzini { 141bd48bde8SPaolo Bonzini MirrorOp *op = opaque; 142bd48bde8SPaolo Bonzini MirrorBlockJob *s = op->s; 143bd48bde8SPaolo Bonzini if (ret < 0) { 144bd48bde8SPaolo Bonzini BlockErrorAction action; 145bd48bde8SPaolo Bonzini 14620dca810SJohn Snow bdrv_set_dirty_bitmap(s->dirty_bitmap, op->sector_num, op->nb_sectors); 147bd48bde8SPaolo Bonzini action = mirror_error_action(s, true, -ret); 148a589569fSWenchao Xia if (action == BLOCK_ERROR_ACTION_REPORT && s->ret >= 0) { 149bd48bde8SPaolo Bonzini s->ret = ret; 150bd48bde8SPaolo Bonzini } 151bd48bde8SPaolo Bonzini 152bd48bde8SPaolo Bonzini mirror_iteration_done(op, ret); 153bd48bde8SPaolo Bonzini return; 154bd48bde8SPaolo Bonzini } 155bd48bde8SPaolo Bonzini bdrv_aio_writev(s->target, op->sector_num, &op->qiov, op->nb_sectors, 156bd48bde8SPaolo Bonzini mirror_write_complete, op); 157bd48bde8SPaolo Bonzini } 158bd48bde8SPaolo Bonzini 159cc8c9d6cSPaolo Bonzini static uint64_t coroutine_fn mirror_iteration(MirrorBlockJob *s) 160893f7ebaSPaolo Bonzini { 161893f7ebaSPaolo Bonzini BlockDriverState *source = s->common.bs; 162402a4741SPaolo Bonzini int nb_sectors, sectors_per_chunk, nb_chunks; 163884fea4eSPaolo Bonzini int64_t end, sector_num, next_chunk, next_sector, hbitmap_next_sector; 1646d0de8ebSStefan Hajnoczi uint64_t delay_ns = 0; 165bd48bde8SPaolo Bonzini MirrorOp *op; 166dcfb3bebSFam Zheng int pnum; 167dcfb3bebSFam Zheng int64_t ret; 168893f7ebaSPaolo Bonzini 1698f0720ecSPaolo Bonzini s->sector_num = hbitmap_iter_next(&s->hbi); 1708f0720ecSPaolo Bonzini if (s->sector_num < 0) { 17120dca810SJohn Snow bdrv_dirty_iter_init(s->dirty_bitmap, &s->hbi); 1728f0720ecSPaolo Bonzini s->sector_num = hbitmap_iter_next(&s->hbi); 17320dca810SJohn Snow trace_mirror_restart_iter(s, bdrv_get_dirty_count(s->dirty_bitmap)); 1748f0720ecSPaolo Bonzini assert(s->sector_num >= 0); 1758f0720ecSPaolo Bonzini } 1768f0720ecSPaolo Bonzini 177402a4741SPaolo Bonzini hbitmap_next_sector = s->sector_num; 178b812f671SPaolo Bonzini sector_num = s->sector_num; 179884fea4eSPaolo Bonzini sectors_per_chunk = s->granularity >> BDRV_SECTOR_BITS; 180b21c7652SMax Reitz end = s->bdev_length / BDRV_SECTOR_SIZE; 181402a4741SPaolo Bonzini 182884fea4eSPaolo Bonzini /* Extend the QEMUIOVector to include all adjacent blocks that will 183884fea4eSPaolo Bonzini * be copied in this operation. 184884fea4eSPaolo Bonzini * 185884fea4eSPaolo Bonzini * We have to do this if we have no backing file yet in the destination, 186884fea4eSPaolo Bonzini * and the cluster size is very large. Then we need to do COW ourselves. 187884fea4eSPaolo Bonzini * The first time a cluster is copied, copy it entirely. Note that, 188884fea4eSPaolo Bonzini * because both the granularity and the cluster size are powers of two, 189884fea4eSPaolo Bonzini * the number of sectors to copy cannot exceed one cluster. 190884fea4eSPaolo Bonzini * 191884fea4eSPaolo Bonzini * We also want to extend the QEMUIOVector to include more adjacent 192884fea4eSPaolo Bonzini * dirty blocks if possible, to limit the number of I/O operations and 193884fea4eSPaolo Bonzini * run efficiently even with a small granularity. 194402a4741SPaolo Bonzini */ 195884fea4eSPaolo Bonzini nb_chunks = 0; 196884fea4eSPaolo Bonzini nb_sectors = 0; 197884fea4eSPaolo Bonzini next_sector = sector_num; 198884fea4eSPaolo Bonzini next_chunk = sector_num / sectors_per_chunk; 199402a4741SPaolo Bonzini 200402a4741SPaolo Bonzini /* Wait for I/O to this cluster (from a previous iteration) to be done. */ 201884fea4eSPaolo Bonzini while (test_bit(next_chunk, s->in_flight_bitmap)) { 202402a4741SPaolo Bonzini trace_mirror_yield_in_flight(s, sector_num, s->in_flight); 203e424aff5SKevin Wolf s->waiting_for_io = true; 204402a4741SPaolo Bonzini qemu_coroutine_yield(); 205e424aff5SKevin Wolf s->waiting_for_io = false; 206b812f671SPaolo Bonzini } 207b812f671SPaolo Bonzini 208884fea4eSPaolo Bonzini do { 209884fea4eSPaolo Bonzini int added_sectors, added_chunks; 210884fea4eSPaolo Bonzini 211e4654d2dSFam Zheng if (!bdrv_get_dirty(source, s->dirty_bitmap, next_sector) || 212884fea4eSPaolo Bonzini test_bit(next_chunk, s->in_flight_bitmap)) { 213884fea4eSPaolo Bonzini assert(nb_sectors > 0); 214884fea4eSPaolo Bonzini break; 215884fea4eSPaolo Bonzini } 216884fea4eSPaolo Bonzini 217884fea4eSPaolo Bonzini added_sectors = sectors_per_chunk; 218884fea4eSPaolo Bonzini if (s->cow_bitmap && !test_bit(next_chunk, s->cow_bitmap)) { 219884fea4eSPaolo Bonzini bdrv_round_to_clusters(s->target, 220884fea4eSPaolo Bonzini next_sector, added_sectors, 221884fea4eSPaolo Bonzini &next_sector, &added_sectors); 222884fea4eSPaolo Bonzini 223884fea4eSPaolo Bonzini /* On the first iteration, the rounding may make us copy 224884fea4eSPaolo Bonzini * sectors before the first dirty one. 225884fea4eSPaolo Bonzini */ 226884fea4eSPaolo Bonzini if (next_sector < sector_num) { 227884fea4eSPaolo Bonzini assert(nb_sectors == 0); 228884fea4eSPaolo Bonzini sector_num = next_sector; 229884fea4eSPaolo Bonzini next_chunk = next_sector / sectors_per_chunk; 230884fea4eSPaolo Bonzini } 231884fea4eSPaolo Bonzini } 232884fea4eSPaolo Bonzini 233884fea4eSPaolo Bonzini added_sectors = MIN(added_sectors, end - (sector_num + nb_sectors)); 234884fea4eSPaolo Bonzini added_chunks = (added_sectors + sectors_per_chunk - 1) / sectors_per_chunk; 235884fea4eSPaolo Bonzini 236884fea4eSPaolo Bonzini /* When doing COW, it may happen that there is not enough space for 237884fea4eSPaolo Bonzini * a full cluster. Wait if that is the case. 238884fea4eSPaolo Bonzini */ 239884fea4eSPaolo Bonzini while (nb_chunks == 0 && s->buf_free_count < added_chunks) { 240402a4741SPaolo Bonzini trace_mirror_yield_buf_busy(s, nb_chunks, s->in_flight); 241e424aff5SKevin Wolf s->waiting_for_io = true; 242402a4741SPaolo Bonzini qemu_coroutine_yield(); 243e424aff5SKevin Wolf s->waiting_for_io = false; 244402a4741SPaolo Bonzini } 245884fea4eSPaolo Bonzini if (s->buf_free_count < nb_chunks + added_chunks) { 246884fea4eSPaolo Bonzini trace_mirror_break_buf_busy(s, nb_chunks, s->in_flight); 247884fea4eSPaolo Bonzini break; 248884fea4eSPaolo Bonzini } 249cae98cb8SStefan Hajnoczi if (IOV_MAX < nb_chunks + added_chunks) { 250cae98cb8SStefan Hajnoczi trace_mirror_break_iov_max(s, nb_chunks, added_chunks); 251cae98cb8SStefan Hajnoczi break; 252cae98cb8SStefan Hajnoczi } 253402a4741SPaolo Bonzini 254402a4741SPaolo Bonzini /* We have enough free space to copy these sectors. */ 255884fea4eSPaolo Bonzini bitmap_set(s->in_flight_bitmap, next_chunk, added_chunks); 256884fea4eSPaolo Bonzini 257884fea4eSPaolo Bonzini nb_sectors += added_sectors; 258884fea4eSPaolo Bonzini nb_chunks += added_chunks; 259884fea4eSPaolo Bonzini next_sector += added_sectors; 260884fea4eSPaolo Bonzini next_chunk += added_chunks; 261cc8c9d6cSPaolo Bonzini if (!s->synced && s->common.speed) { 262cc8c9d6cSPaolo Bonzini delay_ns = ratelimit_calculate_delay(&s->limit, added_sectors); 263cc8c9d6cSPaolo Bonzini } 264cc8c9d6cSPaolo Bonzini } while (delay_ns == 0 && next_sector < end); 265bd48bde8SPaolo Bonzini 266bd48bde8SPaolo Bonzini /* Allocate a MirrorOp that is used as an AIO callback. */ 267c84b3192SPaolo Bonzini op = g_new(MirrorOp, 1); 268bd48bde8SPaolo Bonzini op->s = s; 269bd48bde8SPaolo Bonzini op->sector_num = sector_num; 270bd48bde8SPaolo Bonzini op->nb_sectors = nb_sectors; 271402a4741SPaolo Bonzini 272402a4741SPaolo Bonzini /* Now make a QEMUIOVector taking enough granularity-sized chunks 273402a4741SPaolo Bonzini * from s->buf_free. 274402a4741SPaolo Bonzini */ 275402a4741SPaolo Bonzini qemu_iovec_init(&op->qiov, nb_chunks); 276402a4741SPaolo Bonzini next_sector = sector_num; 277402a4741SPaolo Bonzini while (nb_chunks-- > 0) { 278402a4741SPaolo Bonzini MirrorBuffer *buf = QSIMPLEQ_FIRST(&s->buf_free); 2795a0f6fd5SKevin Wolf size_t remaining = (nb_sectors * BDRV_SECTOR_SIZE) - op->qiov.size; 2805a0f6fd5SKevin Wolf 281402a4741SPaolo Bonzini QSIMPLEQ_REMOVE_HEAD(&s->buf_free, next); 282402a4741SPaolo Bonzini s->buf_free_count--; 2835a0f6fd5SKevin Wolf qemu_iovec_add(&op->qiov, buf, MIN(s->granularity, remaining)); 284402a4741SPaolo Bonzini 285402a4741SPaolo Bonzini /* Advance the HBitmapIter in parallel, so that we do not examine 286402a4741SPaolo Bonzini * the same sector twice. 287402a4741SPaolo Bonzini */ 288e4654d2dSFam Zheng if (next_sector > hbitmap_next_sector 289e4654d2dSFam Zheng && bdrv_get_dirty(source, s->dirty_bitmap, next_sector)) { 290402a4741SPaolo Bonzini hbitmap_next_sector = hbitmap_iter_next(&s->hbi); 291402a4741SPaolo Bonzini } 292402a4741SPaolo Bonzini 293402a4741SPaolo Bonzini next_sector += sectors_per_chunk; 294402a4741SPaolo Bonzini } 295bd48bde8SPaolo Bonzini 29620dca810SJohn Snow bdrv_reset_dirty_bitmap(s->dirty_bitmap, sector_num, nb_sectors); 297893f7ebaSPaolo Bonzini 298893f7ebaSPaolo Bonzini /* Copy the dirty cluster. */ 299bd48bde8SPaolo Bonzini s->in_flight++; 300b21c7652SMax Reitz s->sectors_in_flight += nb_sectors; 301b812f671SPaolo Bonzini trace_mirror_one_iteration(s, sector_num, nb_sectors); 302dcfb3bebSFam Zheng 303dcfb3bebSFam Zheng ret = bdrv_get_block_status_above(source, NULL, sector_num, 304dcfb3bebSFam Zheng nb_sectors, &pnum); 305dcfb3bebSFam Zheng if (ret < 0 || pnum < nb_sectors || 306dcfb3bebSFam Zheng (ret & BDRV_BLOCK_DATA && !(ret & BDRV_BLOCK_ZERO))) { 307bd48bde8SPaolo Bonzini bdrv_aio_readv(source, sector_num, &op->qiov, nb_sectors, 308bd48bde8SPaolo Bonzini mirror_read_complete, op); 309dcfb3bebSFam Zheng } else if (ret & BDRV_BLOCK_ZERO) { 310dcfb3bebSFam Zheng bdrv_aio_write_zeroes(s->target, sector_num, op->nb_sectors, 311dcfb3bebSFam Zheng s->unmap ? BDRV_REQ_MAY_UNMAP : 0, 312dcfb3bebSFam Zheng mirror_write_complete, op); 313dcfb3bebSFam Zheng } else { 314dcfb3bebSFam Zheng assert(!(ret & BDRV_BLOCK_DATA)); 315dcfb3bebSFam Zheng bdrv_aio_discard(s->target, sector_num, op->nb_sectors, 316dcfb3bebSFam Zheng mirror_write_complete, op); 317dcfb3bebSFam Zheng } 318cc8c9d6cSPaolo Bonzini return delay_ns; 319893f7ebaSPaolo Bonzini } 320b952b558SPaolo Bonzini 321402a4741SPaolo Bonzini static void mirror_free_init(MirrorBlockJob *s) 322402a4741SPaolo Bonzini { 323402a4741SPaolo Bonzini int granularity = s->granularity; 324402a4741SPaolo Bonzini size_t buf_size = s->buf_size; 325402a4741SPaolo Bonzini uint8_t *buf = s->buf; 326402a4741SPaolo Bonzini 327402a4741SPaolo Bonzini assert(s->buf_free_count == 0); 328402a4741SPaolo Bonzini QSIMPLEQ_INIT(&s->buf_free); 329402a4741SPaolo Bonzini while (buf_size != 0) { 330402a4741SPaolo Bonzini MirrorBuffer *cur = (MirrorBuffer *)buf; 331402a4741SPaolo Bonzini QSIMPLEQ_INSERT_TAIL(&s->buf_free, cur, next); 332402a4741SPaolo Bonzini s->buf_free_count++; 333402a4741SPaolo Bonzini buf_size -= granularity; 334402a4741SPaolo Bonzini buf += granularity; 335402a4741SPaolo Bonzini } 336402a4741SPaolo Bonzini } 337402a4741SPaolo Bonzini 338bd48bde8SPaolo Bonzini static void mirror_drain(MirrorBlockJob *s) 339bd48bde8SPaolo Bonzini { 340bd48bde8SPaolo Bonzini while (s->in_flight > 0) { 341e424aff5SKevin Wolf s->waiting_for_io = true; 342bd48bde8SPaolo Bonzini qemu_coroutine_yield(); 343e424aff5SKevin Wolf s->waiting_for_io = false; 344bd48bde8SPaolo Bonzini } 345893f7ebaSPaolo Bonzini } 346893f7ebaSPaolo Bonzini 3475a7e7a0bSStefan Hajnoczi typedef struct { 3485a7e7a0bSStefan Hajnoczi int ret; 3495a7e7a0bSStefan Hajnoczi } MirrorExitData; 3505a7e7a0bSStefan Hajnoczi 3515a7e7a0bSStefan Hajnoczi static void mirror_exit(BlockJob *job, void *opaque) 3525a7e7a0bSStefan Hajnoczi { 3535a7e7a0bSStefan Hajnoczi MirrorBlockJob *s = container_of(job, MirrorBlockJob, common); 3545a7e7a0bSStefan Hajnoczi MirrorExitData *data = opaque; 3555a7e7a0bSStefan Hajnoczi AioContext *replace_aio_context = NULL; 3565a7e7a0bSStefan Hajnoczi 3575a7e7a0bSStefan Hajnoczi if (s->to_replace) { 3585a7e7a0bSStefan Hajnoczi replace_aio_context = bdrv_get_aio_context(s->to_replace); 3595a7e7a0bSStefan Hajnoczi aio_context_acquire(replace_aio_context); 3605a7e7a0bSStefan Hajnoczi } 3615a7e7a0bSStefan Hajnoczi 3625a7e7a0bSStefan Hajnoczi if (s->should_complete && data->ret == 0) { 3635a7e7a0bSStefan Hajnoczi BlockDriverState *to_replace = s->common.bs; 3645a7e7a0bSStefan Hajnoczi if (s->to_replace) { 3655a7e7a0bSStefan Hajnoczi to_replace = s->to_replace; 3665a7e7a0bSStefan Hajnoczi } 3675a7e7a0bSStefan Hajnoczi if (bdrv_get_flags(s->target) != bdrv_get_flags(to_replace)) { 3685a7e7a0bSStefan Hajnoczi bdrv_reopen(s->target, bdrv_get_flags(to_replace), NULL); 3695a7e7a0bSStefan Hajnoczi } 3705a7e7a0bSStefan Hajnoczi bdrv_swap(s->target, to_replace); 3715a7e7a0bSStefan Hajnoczi if (s->common.driver->job_type == BLOCK_JOB_TYPE_COMMIT) { 3725a7e7a0bSStefan Hajnoczi /* drop the bs loop chain formed by the swap: break the loop then 3735db15a57SKevin Wolf * trigger the unref */ 3745db15a57SKevin Wolf /* FIXME This duplicates bdrv_set_backing_hd(), except for the 3755db15a57SKevin Wolf * actual detach/unref so that the loop can be broken. When 3765db15a57SKevin Wolf * bdrv_swap() gets replaced, this will become sane again. */ 3775db15a57SKevin Wolf BlockDriverState *backing = s->base->backing->bs; 3785db15a57SKevin Wolf assert(s->base->backing_blocker); 3795db15a57SKevin Wolf bdrv_op_unblock_all(backing, s->base->backing_blocker); 3805db15a57SKevin Wolf error_free(s->base->backing_blocker); 3815db15a57SKevin Wolf s->base->backing_blocker = NULL; 3825db15a57SKevin Wolf bdrv_detach_child(s->base->backing); 3835db15a57SKevin Wolf s->base->backing = NULL; 3845db15a57SKevin Wolf bdrv_unref(backing); 3855a7e7a0bSStefan Hajnoczi } 3865a7e7a0bSStefan Hajnoczi } 3875a7e7a0bSStefan Hajnoczi if (s->to_replace) { 3885a7e7a0bSStefan Hajnoczi bdrv_op_unblock_all(s->to_replace, s->replace_blocker); 3895a7e7a0bSStefan Hajnoczi error_free(s->replace_blocker); 3905a7e7a0bSStefan Hajnoczi bdrv_unref(s->to_replace); 3915a7e7a0bSStefan Hajnoczi } 3925a7e7a0bSStefan Hajnoczi if (replace_aio_context) { 3935a7e7a0bSStefan Hajnoczi aio_context_release(replace_aio_context); 3945a7e7a0bSStefan Hajnoczi } 3955a7e7a0bSStefan Hajnoczi g_free(s->replaces); 3965a7e7a0bSStefan Hajnoczi bdrv_unref(s->target); 3975a7e7a0bSStefan Hajnoczi block_job_completed(&s->common, data->ret); 3985a7e7a0bSStefan Hajnoczi g_free(data); 3995a7e7a0bSStefan Hajnoczi } 4005a7e7a0bSStefan Hajnoczi 401893f7ebaSPaolo Bonzini static void coroutine_fn mirror_run(void *opaque) 402893f7ebaSPaolo Bonzini { 403893f7ebaSPaolo Bonzini MirrorBlockJob *s = opaque; 4045a7e7a0bSStefan Hajnoczi MirrorExitData *data; 405893f7ebaSPaolo Bonzini BlockDriverState *bs = s->common.bs; 40699900697SFam Zheng int64_t sector_num, end, length; 407bd48bde8SPaolo Bonzini uint64_t last_pause_ns; 408b812f671SPaolo Bonzini BlockDriverInfo bdi; 4091d33936eSJeff Cody char backing_filename[2]; /* we only need 2 characters because we are only 4101d33936eSJeff Cody checking for a NULL string */ 411893f7ebaSPaolo Bonzini int ret = 0; 412893f7ebaSPaolo Bonzini int n; 413893f7ebaSPaolo Bonzini 414893f7ebaSPaolo Bonzini if (block_job_is_cancelled(&s->common)) { 415893f7ebaSPaolo Bonzini goto immediate_exit; 416893f7ebaSPaolo Bonzini } 417893f7ebaSPaolo Bonzini 418b21c7652SMax Reitz s->bdev_length = bdrv_getlength(bs); 419b21c7652SMax Reitz if (s->bdev_length < 0) { 420b21c7652SMax Reitz ret = s->bdev_length; 421373df5b1SFam Zheng goto immediate_exit; 422b21c7652SMax Reitz } else if (s->bdev_length == 0) { 4239e48b025SFam Zheng /* Report BLOCK_JOB_READY and wait for complete. */ 4249e48b025SFam Zheng block_job_event_ready(&s->common); 4259e48b025SFam Zheng s->synced = true; 4269e48b025SFam Zheng while (!block_job_is_cancelled(&s->common) && !s->should_complete) { 4279e48b025SFam Zheng block_job_yield(&s->common); 4289e48b025SFam Zheng } 4299e48b025SFam Zheng s->common.cancelled = false; 4309e48b025SFam Zheng goto immediate_exit; 431893f7ebaSPaolo Bonzini } 432893f7ebaSPaolo Bonzini 433b21c7652SMax Reitz length = DIV_ROUND_UP(s->bdev_length, s->granularity); 434402a4741SPaolo Bonzini s->in_flight_bitmap = bitmap_new(length); 435402a4741SPaolo Bonzini 436b812f671SPaolo Bonzini /* If we have no backing file yet in the destination, we cannot let 437b812f671SPaolo Bonzini * the destination do COW. Instead, we copy sectors around the 438b812f671SPaolo Bonzini * dirty data if needed. We need a bitmap to do that. 439b812f671SPaolo Bonzini */ 440b812f671SPaolo Bonzini bdrv_get_backing_filename(s->target, backing_filename, 441b812f671SPaolo Bonzini sizeof(backing_filename)); 442760e0063SKevin Wolf if (backing_filename[0] && !s->target->backing) { 443c3cc95bdSFam Zheng ret = bdrv_get_info(s->target, &bdi); 444c3cc95bdSFam Zheng if (ret < 0) { 445c3cc95bdSFam Zheng goto immediate_exit; 446c3cc95bdSFam Zheng } 447eee13dfeSPaolo Bonzini if (s->granularity < bdi.cluster_size) { 44808e4ed6cSPaolo Bonzini s->buf_size = MAX(s->buf_size, bdi.cluster_size); 449b812f671SPaolo Bonzini s->cow_bitmap = bitmap_new(length); 450b812f671SPaolo Bonzini } 451b812f671SPaolo Bonzini } 452b812f671SPaolo Bonzini 453b21c7652SMax Reitz end = s->bdev_length / BDRV_SECTOR_SIZE; 4547504edf4SKevin Wolf s->buf = qemu_try_blockalign(bs, s->buf_size); 4557504edf4SKevin Wolf if (s->buf == NULL) { 4567504edf4SKevin Wolf ret = -ENOMEM; 4577504edf4SKevin Wolf goto immediate_exit; 4587504edf4SKevin Wolf } 4597504edf4SKevin Wolf 460402a4741SPaolo Bonzini mirror_free_init(s); 461893f7ebaSPaolo Bonzini 4624c0cbd6fSFam Zheng last_pause_ns = qemu_clock_get_ns(QEMU_CLOCK_REALTIME); 46303544a6eSFam Zheng if (!s->is_none_mode) { 464893f7ebaSPaolo Bonzini /* First part, loop on the sectors and initialize the dirty bitmap. */ 4655bc361b8SFam Zheng BlockDriverState *base = s->base; 4665279efebSJeff Cody bool mark_all_dirty = s->base == NULL && !bdrv_has_zero_init(s->target); 4675279efebSJeff Cody 468893f7ebaSPaolo Bonzini for (sector_num = 0; sector_num < end; ) { 46999900697SFam Zheng /* Just to make sure we are not exceeding int limit. */ 47099900697SFam Zheng int nb_sectors = MIN(INT_MAX >> BDRV_SECTOR_BITS, 47199900697SFam Zheng end - sector_num); 4724c0cbd6fSFam Zheng int64_t now = qemu_clock_get_ns(QEMU_CLOCK_REALTIME); 4734c0cbd6fSFam Zheng 4744c0cbd6fSFam Zheng if (now - last_pause_ns > SLICE_TIME) { 4754c0cbd6fSFam Zheng last_pause_ns = now; 4764c0cbd6fSFam Zheng block_job_sleep_ns(&s->common, QEMU_CLOCK_REALTIME, 0); 4774c0cbd6fSFam Zheng } 4784c0cbd6fSFam Zheng 4794c0cbd6fSFam Zheng if (block_job_is_cancelled(&s->common)) { 4804c0cbd6fSFam Zheng goto immediate_exit; 4814c0cbd6fSFam Zheng } 4824c0cbd6fSFam Zheng 48399900697SFam Zheng ret = bdrv_is_allocated_above(bs, base, sector_num, nb_sectors, &n); 484893f7ebaSPaolo Bonzini 485893f7ebaSPaolo Bonzini if (ret < 0) { 486893f7ebaSPaolo Bonzini goto immediate_exit; 487893f7ebaSPaolo Bonzini } 488893f7ebaSPaolo Bonzini 489893f7ebaSPaolo Bonzini assert(n > 0); 4905279efebSJeff Cody if (ret == 1 || mark_all_dirty) { 49120dca810SJohn Snow bdrv_set_dirty_bitmap(s->dirty_bitmap, sector_num, n); 492893f7ebaSPaolo Bonzini } 49399900697SFam Zheng sector_num += n; 494893f7ebaSPaolo Bonzini } 495893f7ebaSPaolo Bonzini } 496893f7ebaSPaolo Bonzini 49720dca810SJohn Snow bdrv_dirty_iter_init(s->dirty_bitmap, &s->hbi); 498893f7ebaSPaolo Bonzini for (;;) { 499cc8c9d6cSPaolo Bonzini uint64_t delay_ns = 0; 500893f7ebaSPaolo Bonzini int64_t cnt; 501893f7ebaSPaolo Bonzini bool should_complete; 502893f7ebaSPaolo Bonzini 503bd48bde8SPaolo Bonzini if (s->ret < 0) { 504bd48bde8SPaolo Bonzini ret = s->ret; 505893f7ebaSPaolo Bonzini goto immediate_exit; 506893f7ebaSPaolo Bonzini } 507bd48bde8SPaolo Bonzini 50820dca810SJohn Snow cnt = bdrv_get_dirty_count(s->dirty_bitmap); 509b21c7652SMax Reitz /* s->common.offset contains the number of bytes already processed so 510b21c7652SMax Reitz * far, cnt is the number of dirty sectors remaining and 511b21c7652SMax Reitz * s->sectors_in_flight is the number of sectors currently being 512b21c7652SMax Reitz * processed; together those are the current total operation length */ 513b21c7652SMax Reitz s->common.len = s->common.offset + 514b21c7652SMax Reitz (cnt + s->sectors_in_flight) * BDRV_SECTOR_SIZE; 515bd48bde8SPaolo Bonzini 516bd48bde8SPaolo Bonzini /* Note that even when no rate limit is applied we need to yield 517a7282330SFam Zheng * periodically with no pending I/O so that bdrv_drain_all() returns. 518bd48bde8SPaolo Bonzini * We do so every SLICE_TIME nanoseconds, or when there is an error, 519bd48bde8SPaolo Bonzini * or when the source is clean, whichever comes first. 520bd48bde8SPaolo Bonzini */ 521bc72ad67SAlex Bligh if (qemu_clock_get_ns(QEMU_CLOCK_REALTIME) - last_pause_ns < SLICE_TIME && 522bd48bde8SPaolo Bonzini s->common.iostatus == BLOCK_DEVICE_IO_STATUS_OK) { 523402a4741SPaolo Bonzini if (s->in_flight == MAX_IN_FLIGHT || s->buf_free_count == 0 || 524402a4741SPaolo Bonzini (cnt == 0 && s->in_flight > 0)) { 525402a4741SPaolo Bonzini trace_mirror_yield(s, s->in_flight, s->buf_free_count, cnt); 526e424aff5SKevin Wolf s->waiting_for_io = true; 527bd48bde8SPaolo Bonzini qemu_coroutine_yield(); 528e424aff5SKevin Wolf s->waiting_for_io = false; 529bd48bde8SPaolo Bonzini continue; 530bd48bde8SPaolo Bonzini } else if (cnt != 0) { 531cc8c9d6cSPaolo Bonzini delay_ns = mirror_iteration(s); 532893f7ebaSPaolo Bonzini } 533cc8c9d6cSPaolo Bonzini } 534893f7ebaSPaolo Bonzini 535893f7ebaSPaolo Bonzini should_complete = false; 536bd48bde8SPaolo Bonzini if (s->in_flight == 0 && cnt == 0) { 537893f7ebaSPaolo Bonzini trace_mirror_before_flush(s); 538893f7ebaSPaolo Bonzini ret = bdrv_flush(s->target); 539893f7ebaSPaolo Bonzini if (ret < 0) { 540a589569fSWenchao Xia if (mirror_error_action(s, false, -ret) == 541a589569fSWenchao Xia BLOCK_ERROR_ACTION_REPORT) { 542893f7ebaSPaolo Bonzini goto immediate_exit; 543893f7ebaSPaolo Bonzini } 544b952b558SPaolo Bonzini } else { 545893f7ebaSPaolo Bonzini /* We're out of the streaming phase. From now on, if the job 546893f7ebaSPaolo Bonzini * is cancelled we will actually complete all pending I/O and 547893f7ebaSPaolo Bonzini * report completion. This way, block-job-cancel will leave 548893f7ebaSPaolo Bonzini * the target in a consistent state. 549893f7ebaSPaolo Bonzini */ 550d63ffd87SPaolo Bonzini if (!s->synced) { 551bcada37bSWenchao Xia block_job_event_ready(&s->common); 552d63ffd87SPaolo Bonzini s->synced = true; 553d63ffd87SPaolo Bonzini } 554d63ffd87SPaolo Bonzini 555d63ffd87SPaolo Bonzini should_complete = s->should_complete || 556d63ffd87SPaolo Bonzini block_job_is_cancelled(&s->common); 55720dca810SJohn Snow cnt = bdrv_get_dirty_count(s->dirty_bitmap); 558893f7ebaSPaolo Bonzini } 559b952b558SPaolo Bonzini } 560893f7ebaSPaolo Bonzini 561893f7ebaSPaolo Bonzini if (cnt == 0 && should_complete) { 562893f7ebaSPaolo Bonzini /* The dirty bitmap is not updated while operations are pending. 563893f7ebaSPaolo Bonzini * If we're about to exit, wait for pending operations before 564893f7ebaSPaolo Bonzini * calling bdrv_get_dirty_count(bs), or we may exit while the 565893f7ebaSPaolo Bonzini * source has dirty data to copy! 566893f7ebaSPaolo Bonzini * 567893f7ebaSPaolo Bonzini * Note that I/O can be submitted by the guest while 568893f7ebaSPaolo Bonzini * mirror_populate runs. 569893f7ebaSPaolo Bonzini */ 570893f7ebaSPaolo Bonzini trace_mirror_before_drain(s, cnt); 5715a7e7a0bSStefan Hajnoczi bdrv_drain(bs); 57220dca810SJohn Snow cnt = bdrv_get_dirty_count(s->dirty_bitmap); 573893f7ebaSPaolo Bonzini } 574893f7ebaSPaolo Bonzini 575893f7ebaSPaolo Bonzini ret = 0; 576cc8c9d6cSPaolo Bonzini trace_mirror_before_sleep(s, cnt, s->synced, delay_ns); 577d63ffd87SPaolo Bonzini if (!s->synced) { 5787483d1e5SAlex Bligh block_job_sleep_ns(&s->common, QEMU_CLOCK_REALTIME, delay_ns); 579893f7ebaSPaolo Bonzini if (block_job_is_cancelled(&s->common)) { 580893f7ebaSPaolo Bonzini break; 581893f7ebaSPaolo Bonzini } 582893f7ebaSPaolo Bonzini } else if (!should_complete) { 583bd48bde8SPaolo Bonzini delay_ns = (s->in_flight == 0 && cnt == 0 ? SLICE_TIME : 0); 5847483d1e5SAlex Bligh block_job_sleep_ns(&s->common, QEMU_CLOCK_REALTIME, delay_ns); 585893f7ebaSPaolo Bonzini } else if (cnt == 0) { 586893f7ebaSPaolo Bonzini /* The two disks are in sync. Exit and report successful 587893f7ebaSPaolo Bonzini * completion. 588893f7ebaSPaolo Bonzini */ 589893f7ebaSPaolo Bonzini assert(QLIST_EMPTY(&bs->tracked_requests)); 590893f7ebaSPaolo Bonzini s->common.cancelled = false; 591893f7ebaSPaolo Bonzini break; 592893f7ebaSPaolo Bonzini } 593bc72ad67SAlex Bligh last_pause_ns = qemu_clock_get_ns(QEMU_CLOCK_REALTIME); 594893f7ebaSPaolo Bonzini } 595893f7ebaSPaolo Bonzini 596893f7ebaSPaolo Bonzini immediate_exit: 597bd48bde8SPaolo Bonzini if (s->in_flight > 0) { 598bd48bde8SPaolo Bonzini /* We get here only if something went wrong. Either the job failed, 599bd48bde8SPaolo Bonzini * or it was cancelled prematurely so that we do not guarantee that 600bd48bde8SPaolo Bonzini * the target is a copy of the source. 601bd48bde8SPaolo Bonzini */ 602bd48bde8SPaolo Bonzini assert(ret < 0 || (!s->synced && block_job_is_cancelled(&s->common))); 603bd48bde8SPaolo Bonzini mirror_drain(s); 604bd48bde8SPaolo Bonzini } 605bd48bde8SPaolo Bonzini 606bd48bde8SPaolo Bonzini assert(s->in_flight == 0); 6077191bf31SMarkus Armbruster qemu_vfree(s->buf); 608b812f671SPaolo Bonzini g_free(s->cow_bitmap); 609402a4741SPaolo Bonzini g_free(s->in_flight_bitmap); 610e4654d2dSFam Zheng bdrv_release_dirty_bitmap(bs, s->dirty_bitmap); 611b952b558SPaolo Bonzini bdrv_iostatus_disable(s->target); 6125a7e7a0bSStefan Hajnoczi 6135a7e7a0bSStefan Hajnoczi data = g_malloc(sizeof(*data)); 6145a7e7a0bSStefan Hajnoczi data->ret = ret; 6155a7e7a0bSStefan Hajnoczi block_job_defer_to_main_loop(&s->common, mirror_exit, data); 616893f7ebaSPaolo Bonzini } 617893f7ebaSPaolo Bonzini 618893f7ebaSPaolo Bonzini static void mirror_set_speed(BlockJob *job, int64_t speed, Error **errp) 619893f7ebaSPaolo Bonzini { 620893f7ebaSPaolo Bonzini MirrorBlockJob *s = container_of(job, MirrorBlockJob, common); 621893f7ebaSPaolo Bonzini 622893f7ebaSPaolo Bonzini if (speed < 0) { 623c6bd8c70SMarkus Armbruster error_setg(errp, QERR_INVALID_PARAMETER, "speed"); 624893f7ebaSPaolo Bonzini return; 625893f7ebaSPaolo Bonzini } 626893f7ebaSPaolo Bonzini ratelimit_set_speed(&s->limit, speed / BDRV_SECTOR_SIZE, SLICE_TIME); 627893f7ebaSPaolo Bonzini } 628893f7ebaSPaolo Bonzini 629b952b558SPaolo Bonzini static void mirror_iostatus_reset(BlockJob *job) 630b952b558SPaolo Bonzini { 631b952b558SPaolo Bonzini MirrorBlockJob *s = container_of(job, MirrorBlockJob, common); 632b952b558SPaolo Bonzini 633b952b558SPaolo Bonzini bdrv_iostatus_reset(s->target); 634b952b558SPaolo Bonzini } 635b952b558SPaolo Bonzini 636d63ffd87SPaolo Bonzini static void mirror_complete(BlockJob *job, Error **errp) 637d63ffd87SPaolo Bonzini { 638d63ffd87SPaolo Bonzini MirrorBlockJob *s = container_of(job, MirrorBlockJob, common); 63934b5d2c6SMax Reitz Error *local_err = NULL; 640d63ffd87SPaolo Bonzini int ret; 641d63ffd87SPaolo Bonzini 64234b5d2c6SMax Reitz ret = bdrv_open_backing_file(s->target, NULL, &local_err); 643d63ffd87SPaolo Bonzini if (ret < 0) { 64434b5d2c6SMax Reitz error_propagate(errp, local_err); 645d63ffd87SPaolo Bonzini return; 646d63ffd87SPaolo Bonzini } 647d63ffd87SPaolo Bonzini if (!s->synced) { 648*8ccb9569SKevin Wolf error_setg(errp, QERR_BLOCK_JOB_NOT_READY, job->id); 649d63ffd87SPaolo Bonzini return; 650d63ffd87SPaolo Bonzini } 651d63ffd87SPaolo Bonzini 65209158f00SBenoît Canet /* check the target bs is not blocked and block all operations on it */ 65309158f00SBenoît Canet if (s->replaces) { 6545a7e7a0bSStefan Hajnoczi AioContext *replace_aio_context; 6555a7e7a0bSStefan Hajnoczi 656e12f3784SWen Congyang s->to_replace = bdrv_find_node(s->replaces); 65709158f00SBenoît Canet if (!s->to_replace) { 658e12f3784SWen Congyang error_setg(errp, "Node name '%s' not found", s->replaces); 65909158f00SBenoît Canet return; 66009158f00SBenoît Canet } 66109158f00SBenoît Canet 6625a7e7a0bSStefan Hajnoczi replace_aio_context = bdrv_get_aio_context(s->to_replace); 6635a7e7a0bSStefan Hajnoczi aio_context_acquire(replace_aio_context); 6645a7e7a0bSStefan Hajnoczi 66509158f00SBenoît Canet error_setg(&s->replace_blocker, 66609158f00SBenoît Canet "block device is in use by block-job-complete"); 66709158f00SBenoît Canet bdrv_op_block_all(s->to_replace, s->replace_blocker); 66809158f00SBenoît Canet bdrv_ref(s->to_replace); 6695a7e7a0bSStefan Hajnoczi 6705a7e7a0bSStefan Hajnoczi aio_context_release(replace_aio_context); 67109158f00SBenoît Canet } 67209158f00SBenoît Canet 673d63ffd87SPaolo Bonzini s->should_complete = true; 674751ebd76SFam Zheng block_job_enter(&s->common); 675d63ffd87SPaolo Bonzini } 676d63ffd87SPaolo Bonzini 6773fc4b10aSFam Zheng static const BlockJobDriver mirror_job_driver = { 678893f7ebaSPaolo Bonzini .instance_size = sizeof(MirrorBlockJob), 67979e14bf7SFam Zheng .job_type = BLOCK_JOB_TYPE_MIRROR, 680893f7ebaSPaolo Bonzini .set_speed = mirror_set_speed, 681b952b558SPaolo Bonzini .iostatus_reset= mirror_iostatus_reset, 682d63ffd87SPaolo Bonzini .complete = mirror_complete, 683893f7ebaSPaolo Bonzini }; 684893f7ebaSPaolo Bonzini 68503544a6eSFam Zheng static const BlockJobDriver commit_active_job_driver = { 68603544a6eSFam Zheng .instance_size = sizeof(MirrorBlockJob), 68703544a6eSFam Zheng .job_type = BLOCK_JOB_TYPE_COMMIT, 68803544a6eSFam Zheng .set_speed = mirror_set_speed, 68903544a6eSFam Zheng .iostatus_reset 69003544a6eSFam Zheng = mirror_iostatus_reset, 69103544a6eSFam Zheng .complete = mirror_complete, 69203544a6eSFam Zheng }; 69303544a6eSFam Zheng 69403544a6eSFam Zheng static void mirror_start_job(BlockDriverState *bs, BlockDriverState *target, 69509158f00SBenoît Canet const char *replaces, 6965fba6c0eSJohn Snow int64_t speed, uint32_t granularity, 69703544a6eSFam Zheng int64_t buf_size, 69803544a6eSFam Zheng BlockdevOnError on_source_error, 699b952b558SPaolo Bonzini BlockdevOnError on_target_error, 7000fc9f8eaSFam Zheng bool unmap, 701097310b5SMarkus Armbruster BlockCompletionFunc *cb, 70203544a6eSFam Zheng void *opaque, Error **errp, 70303544a6eSFam Zheng const BlockJobDriver *driver, 70403544a6eSFam Zheng bool is_none_mode, BlockDriverState *base) 705893f7ebaSPaolo Bonzini { 706893f7ebaSPaolo Bonzini MirrorBlockJob *s; 707893f7ebaSPaolo Bonzini 708eee13dfeSPaolo Bonzini if (granularity == 0) { 709341ebc2fSJohn Snow granularity = bdrv_get_default_bitmap_granularity(target); 710eee13dfeSPaolo Bonzini } 711eee13dfeSPaolo Bonzini 712eee13dfeSPaolo Bonzini assert ((granularity & (granularity - 1)) == 0); 713eee13dfeSPaolo Bonzini 714b952b558SPaolo Bonzini if ((on_source_error == BLOCKDEV_ON_ERROR_STOP || 715b952b558SPaolo Bonzini on_source_error == BLOCKDEV_ON_ERROR_ENOSPC) && 716b952b558SPaolo Bonzini !bdrv_iostatus_is_enabled(bs)) { 717c6bd8c70SMarkus Armbruster error_setg(errp, QERR_INVALID_PARAMETER, "on-source-error"); 718b952b558SPaolo Bonzini return; 719b952b558SPaolo Bonzini } 720b952b558SPaolo Bonzini 72148ac0a4dSWen Congyang if (buf_size < 0) { 72248ac0a4dSWen Congyang error_setg(errp, "Invalid parameter 'buf-size'"); 72348ac0a4dSWen Congyang return; 72448ac0a4dSWen Congyang } 72548ac0a4dSWen Congyang 72648ac0a4dSWen Congyang if (buf_size == 0) { 72748ac0a4dSWen Congyang buf_size = DEFAULT_MIRROR_BUF_SIZE; 72848ac0a4dSWen Congyang } 7295bc361b8SFam Zheng 73003544a6eSFam Zheng s = block_job_create(driver, bs, speed, cb, opaque, errp); 731893f7ebaSPaolo Bonzini if (!s) { 732893f7ebaSPaolo Bonzini return; 733893f7ebaSPaolo Bonzini } 734893f7ebaSPaolo Bonzini 73509158f00SBenoît Canet s->replaces = g_strdup(replaces); 736b952b558SPaolo Bonzini s->on_source_error = on_source_error; 737b952b558SPaolo Bonzini s->on_target_error = on_target_error; 738893f7ebaSPaolo Bonzini s->target = target; 73903544a6eSFam Zheng s->is_none_mode = is_none_mode; 7405bc361b8SFam Zheng s->base = base; 741eee13dfeSPaolo Bonzini s->granularity = granularity; 74248ac0a4dSWen Congyang s->buf_size = ROUND_UP(buf_size, granularity); 7430fc9f8eaSFam Zheng s->unmap = unmap; 744b812f671SPaolo Bonzini 7450db6e54aSFam Zheng s->dirty_bitmap = bdrv_create_dirty_bitmap(bs, granularity, NULL, errp); 746b8afb520SFam Zheng if (!s->dirty_bitmap) { 74797031164STing Wang g_free(s->replaces); 74897031164STing Wang block_job_release(bs); 749b8afb520SFam Zheng return; 750b8afb520SFam Zheng } 751893f7ebaSPaolo Bonzini bdrv_set_enable_write_cache(s->target, true); 752b952b558SPaolo Bonzini bdrv_set_on_error(s->target, on_target_error, on_target_error); 753b952b558SPaolo Bonzini bdrv_iostatus_enable(s->target); 754893f7ebaSPaolo Bonzini s->common.co = qemu_coroutine_create(mirror_run); 755893f7ebaSPaolo Bonzini trace_mirror_start(bs, s, s->common.co, opaque); 756893f7ebaSPaolo Bonzini qemu_coroutine_enter(s->common.co, s); 757893f7ebaSPaolo Bonzini } 75803544a6eSFam Zheng 75903544a6eSFam Zheng void mirror_start(BlockDriverState *bs, BlockDriverState *target, 76009158f00SBenoît Canet const char *replaces, 7615fba6c0eSJohn Snow int64_t speed, uint32_t granularity, int64_t buf_size, 76203544a6eSFam Zheng MirrorSyncMode mode, BlockdevOnError on_source_error, 76303544a6eSFam Zheng BlockdevOnError on_target_error, 7640fc9f8eaSFam Zheng bool unmap, 765097310b5SMarkus Armbruster BlockCompletionFunc *cb, 76603544a6eSFam Zheng void *opaque, Error **errp) 76703544a6eSFam Zheng { 76803544a6eSFam Zheng bool is_none_mode; 76903544a6eSFam Zheng BlockDriverState *base; 77003544a6eSFam Zheng 7714b80ab2bSJohn Snow if (mode == MIRROR_SYNC_MODE_INCREMENTAL) { 7724b80ab2bSJohn Snow error_setg(errp, "Sync mode 'incremental' not supported"); 773d58d8453SJohn Snow return; 774d58d8453SJohn Snow } 77503544a6eSFam Zheng is_none_mode = mode == MIRROR_SYNC_MODE_NONE; 776760e0063SKevin Wolf base = mode == MIRROR_SYNC_MODE_TOP ? backing_bs(bs) : NULL; 77709158f00SBenoît Canet mirror_start_job(bs, target, replaces, 77809158f00SBenoît Canet speed, granularity, buf_size, 7790fc9f8eaSFam Zheng on_source_error, on_target_error, unmap, cb, opaque, errp, 78003544a6eSFam Zheng &mirror_job_driver, is_none_mode, base); 78103544a6eSFam Zheng } 78203544a6eSFam Zheng 78303544a6eSFam Zheng void commit_active_start(BlockDriverState *bs, BlockDriverState *base, 78403544a6eSFam Zheng int64_t speed, 78503544a6eSFam Zheng BlockdevOnError on_error, 786097310b5SMarkus Armbruster BlockCompletionFunc *cb, 78703544a6eSFam Zheng void *opaque, Error **errp) 78803544a6eSFam Zheng { 7894da83585SJeff Cody int64_t length, base_length; 7904da83585SJeff Cody int orig_base_flags; 79139a611a3SJeff Cody int ret; 792cc67f4d1SJeff Cody Error *local_err = NULL; 7934da83585SJeff Cody 7944da83585SJeff Cody orig_base_flags = bdrv_get_flags(base); 7954da83585SJeff Cody 79620a63d2cSFam Zheng if (bdrv_reopen(base, bs->open_flags, errp)) { 79720a63d2cSFam Zheng return; 79820a63d2cSFam Zheng } 7994da83585SJeff Cody 8004da83585SJeff Cody length = bdrv_getlength(bs); 8014da83585SJeff Cody if (length < 0) { 80239a611a3SJeff Cody error_setg_errno(errp, -length, 80339a611a3SJeff Cody "Unable to determine length of %s", bs->filename); 8044da83585SJeff Cody goto error_restore_flags; 8054da83585SJeff Cody } 8064da83585SJeff Cody 8074da83585SJeff Cody base_length = bdrv_getlength(base); 8084da83585SJeff Cody if (base_length < 0) { 80939a611a3SJeff Cody error_setg_errno(errp, -base_length, 81039a611a3SJeff Cody "Unable to determine length of %s", base->filename); 8114da83585SJeff Cody goto error_restore_flags; 8124da83585SJeff Cody } 8134da83585SJeff Cody 8144da83585SJeff Cody if (length > base_length) { 81539a611a3SJeff Cody ret = bdrv_truncate(base, length); 81639a611a3SJeff Cody if (ret < 0) { 81739a611a3SJeff Cody error_setg_errno(errp, -ret, 81839a611a3SJeff Cody "Top image %s is larger than base image %s, and " 8194da83585SJeff Cody "resize of base image failed", 8204da83585SJeff Cody bs->filename, base->filename); 8214da83585SJeff Cody goto error_restore_flags; 8224da83585SJeff Cody } 8234da83585SJeff Cody } 8244da83585SJeff Cody 82520a63d2cSFam Zheng bdrv_ref(base); 82609158f00SBenoît Canet mirror_start_job(bs, base, NULL, speed, 0, 0, 8270fc9f8eaSFam Zheng on_error, on_error, false, cb, opaque, &local_err, 82803544a6eSFam Zheng &commit_active_job_driver, false, base); 8290fb6395cSMarkus Armbruster if (local_err) { 830cc67f4d1SJeff Cody error_propagate(errp, local_err); 8314da83585SJeff Cody goto error_restore_flags; 8324da83585SJeff Cody } 8334da83585SJeff Cody 8344da83585SJeff Cody return; 8354da83585SJeff Cody 8364da83585SJeff Cody error_restore_flags: 8374da83585SJeff Cody /* ignore error and errp for bdrv_reopen, because we want to propagate 8384da83585SJeff Cody * the original error */ 8394da83585SJeff Cody bdrv_reopen(base, orig_base_flags, NULL); 8404da83585SJeff Cody return; 84103544a6eSFam Zheng } 842