xref: /openbmc/qemu/block/mirror.c (revision fb37726d)
1 /*
2  * Image mirroring
3  *
4  * Copyright Red Hat, Inc. 2012
5  *
6  * Authors:
7  *  Paolo Bonzini  <pbonzini@redhat.com>
8  *
9  * This work is licensed under the terms of the GNU LGPL, version 2 or later.
10  * See the COPYING.LIB file in the top-level directory.
11  *
12  */
13 
14 #include "trace.h"
15 #include "block/blockjob.h"
16 #include "block/block_int.h"
17 #include "qapi/qmp/qerror.h"
18 #include "qemu/ratelimit.h"
19 #include "qemu/bitmap.h"
20 
21 #define SLICE_TIME    100000000ULL /* ns */
22 #define MAX_IN_FLIGHT 16
23 #define DEFAULT_MIRROR_BUF_SIZE   (10 << 20)
24 
25 /* The mirroring buffer is a list of granularity-sized chunks.
26  * Free chunks are organized in a list.
27  */
28 typedef struct MirrorBuffer {
29     QSIMPLEQ_ENTRY(MirrorBuffer) next;
30 } MirrorBuffer;
31 
32 typedef struct MirrorBlockJob {
33     BlockJob common;
34     RateLimit limit;
35     BlockDriverState *target;
36     BlockDriverState *base;
37     /* The name of the graph node to replace */
38     char *replaces;
39     /* The BDS to replace */
40     BlockDriverState *to_replace;
41     /* Used to block operations on the drive-mirror-replace target */
42     Error *replace_blocker;
43     bool is_none_mode;
44     BlockdevOnError on_source_error, on_target_error;
45     bool synced;
46     bool should_complete;
47     int64_t sector_num;
48     int64_t granularity;
49     size_t buf_size;
50     int64_t bdev_length;
51     unsigned long *cow_bitmap;
52     BdrvDirtyBitmap *dirty_bitmap;
53     HBitmapIter hbi;
54     uint8_t *buf;
55     QSIMPLEQ_HEAD(, MirrorBuffer) buf_free;
56     int buf_free_count;
57 
58     unsigned long *in_flight_bitmap;
59     int in_flight;
60     int sectors_in_flight;
61     int ret;
62     bool unmap;
63     bool waiting_for_io;
64 } MirrorBlockJob;
65 
66 typedef struct MirrorOp {
67     MirrorBlockJob *s;
68     QEMUIOVector qiov;
69     int64_t sector_num;
70     int nb_sectors;
71 } MirrorOp;
72 
73 static BlockErrorAction mirror_error_action(MirrorBlockJob *s, bool read,
74                                             int error)
75 {
76     s->synced = false;
77     if (read) {
78         return block_job_error_action(&s->common, s->common.bs,
79                                       s->on_source_error, true, error);
80     } else {
81         return block_job_error_action(&s->common, s->target,
82                                       s->on_target_error, false, error);
83     }
84 }
85 
86 static void mirror_iteration_done(MirrorOp *op, int ret)
87 {
88     MirrorBlockJob *s = op->s;
89     struct iovec *iov;
90     int64_t chunk_num;
91     int i, nb_chunks, sectors_per_chunk;
92 
93     trace_mirror_iteration_done(s, op->sector_num, op->nb_sectors, ret);
94 
95     s->in_flight--;
96     s->sectors_in_flight -= op->nb_sectors;
97     iov = op->qiov.iov;
98     for (i = 0; i < op->qiov.niov; i++) {
99         MirrorBuffer *buf = (MirrorBuffer *) iov[i].iov_base;
100         QSIMPLEQ_INSERT_TAIL(&s->buf_free, buf, next);
101         s->buf_free_count++;
102     }
103 
104     sectors_per_chunk = s->granularity >> BDRV_SECTOR_BITS;
105     chunk_num = op->sector_num / sectors_per_chunk;
106     nb_chunks = op->nb_sectors / sectors_per_chunk;
107     bitmap_clear(s->in_flight_bitmap, chunk_num, nb_chunks);
108     if (ret >= 0) {
109         if (s->cow_bitmap) {
110             bitmap_set(s->cow_bitmap, chunk_num, nb_chunks);
111         }
112         s->common.offset += (uint64_t)op->nb_sectors * BDRV_SECTOR_SIZE;
113     }
114 
115     qemu_iovec_destroy(&op->qiov);
116     g_slice_free(MirrorOp, op);
117 
118     if (s->waiting_for_io) {
119         qemu_coroutine_enter(s->common.co, NULL);
120     }
121 }
122 
123 static void mirror_write_complete(void *opaque, int ret)
124 {
125     MirrorOp *op = opaque;
126     MirrorBlockJob *s = op->s;
127     if (ret < 0) {
128         BlockErrorAction action;
129 
130         bdrv_set_dirty_bitmap(s->dirty_bitmap, op->sector_num, op->nb_sectors);
131         action = mirror_error_action(s, false, -ret);
132         if (action == BLOCK_ERROR_ACTION_REPORT && s->ret >= 0) {
133             s->ret = ret;
134         }
135     }
136     mirror_iteration_done(op, ret);
137 }
138 
139 static void mirror_read_complete(void *opaque, int ret)
140 {
141     MirrorOp *op = opaque;
142     MirrorBlockJob *s = op->s;
143     if (ret < 0) {
144         BlockErrorAction action;
145 
146         bdrv_set_dirty_bitmap(s->dirty_bitmap, op->sector_num, op->nb_sectors);
147         action = mirror_error_action(s, true, -ret);
148         if (action == BLOCK_ERROR_ACTION_REPORT && s->ret >= 0) {
149             s->ret = ret;
150         }
151 
152         mirror_iteration_done(op, ret);
153         return;
154     }
155     bdrv_aio_writev(s->target, op->sector_num, &op->qiov, op->nb_sectors,
156                     mirror_write_complete, op);
157 }
158 
159 static uint64_t coroutine_fn mirror_iteration(MirrorBlockJob *s)
160 {
161     BlockDriverState *source = s->common.bs;
162     int nb_sectors, sectors_per_chunk, nb_chunks;
163     int64_t end, sector_num, next_chunk, next_sector, hbitmap_next_sector;
164     uint64_t delay_ns = 0;
165     MirrorOp *op;
166     int pnum;
167     int64_t ret;
168 
169     s->sector_num = hbitmap_iter_next(&s->hbi);
170     if (s->sector_num < 0) {
171         bdrv_dirty_iter_init(s->dirty_bitmap, &s->hbi);
172         s->sector_num = hbitmap_iter_next(&s->hbi);
173         trace_mirror_restart_iter(s, bdrv_get_dirty_count(s->dirty_bitmap));
174         assert(s->sector_num >= 0);
175     }
176 
177     hbitmap_next_sector = s->sector_num;
178     sector_num = s->sector_num;
179     sectors_per_chunk = s->granularity >> BDRV_SECTOR_BITS;
180     end = s->bdev_length / BDRV_SECTOR_SIZE;
181 
182     /* Extend the QEMUIOVector to include all adjacent blocks that will
183      * be copied in this operation.
184      *
185      * We have to do this if we have no backing file yet in the destination,
186      * and the cluster size is very large.  Then we need to do COW ourselves.
187      * The first time a cluster is copied, copy it entirely.  Note that,
188      * because both the granularity and the cluster size are powers of two,
189      * the number of sectors to copy cannot exceed one cluster.
190      *
191      * We also want to extend the QEMUIOVector to include more adjacent
192      * dirty blocks if possible, to limit the number of I/O operations and
193      * run efficiently even with a small granularity.
194      */
195     nb_chunks = 0;
196     nb_sectors = 0;
197     next_sector = sector_num;
198     next_chunk = sector_num / sectors_per_chunk;
199 
200     /* Wait for I/O to this cluster (from a previous iteration) to be done.  */
201     while (test_bit(next_chunk, s->in_flight_bitmap)) {
202         trace_mirror_yield_in_flight(s, sector_num, s->in_flight);
203         s->waiting_for_io = true;
204         qemu_coroutine_yield();
205         s->waiting_for_io = false;
206     }
207 
208     do {
209         int added_sectors, added_chunks;
210 
211         if (!bdrv_get_dirty(source, s->dirty_bitmap, next_sector) ||
212             test_bit(next_chunk, s->in_flight_bitmap)) {
213             assert(nb_sectors > 0);
214             break;
215         }
216 
217         added_sectors = sectors_per_chunk;
218         if (s->cow_bitmap && !test_bit(next_chunk, s->cow_bitmap)) {
219             bdrv_round_to_clusters(s->target,
220                                    next_sector, added_sectors,
221                                    &next_sector, &added_sectors);
222 
223             /* On the first iteration, the rounding may make us copy
224              * sectors before the first dirty one.
225              */
226             if (next_sector < sector_num) {
227                 assert(nb_sectors == 0);
228                 sector_num = next_sector;
229                 next_chunk = next_sector / sectors_per_chunk;
230             }
231         }
232 
233         added_sectors = MIN(added_sectors, end - (sector_num + nb_sectors));
234         added_chunks = (added_sectors + sectors_per_chunk - 1) / sectors_per_chunk;
235 
236         /* When doing COW, it may happen that there is not enough space for
237          * a full cluster.  Wait if that is the case.
238          */
239         while (nb_chunks == 0 && s->buf_free_count < added_chunks) {
240             trace_mirror_yield_buf_busy(s, nb_chunks, s->in_flight);
241             s->waiting_for_io = true;
242             qemu_coroutine_yield();
243             s->waiting_for_io = false;
244         }
245         if (s->buf_free_count < nb_chunks + added_chunks) {
246             trace_mirror_break_buf_busy(s, nb_chunks, s->in_flight);
247             break;
248         }
249         if (IOV_MAX < nb_chunks + added_chunks) {
250             trace_mirror_break_iov_max(s, nb_chunks, added_chunks);
251             break;
252         }
253 
254         /* We have enough free space to copy these sectors.  */
255         bitmap_set(s->in_flight_bitmap, next_chunk, added_chunks);
256 
257         nb_sectors += added_sectors;
258         nb_chunks += added_chunks;
259         next_sector += added_sectors;
260         next_chunk += added_chunks;
261         if (!s->synced && s->common.speed) {
262             delay_ns = ratelimit_calculate_delay(&s->limit, added_sectors);
263         }
264     } while (delay_ns == 0 && next_sector < end);
265 
266     /* Allocate a MirrorOp that is used as an AIO callback.  */
267     op = g_slice_new(MirrorOp);
268     op->s = s;
269     op->sector_num = sector_num;
270     op->nb_sectors = nb_sectors;
271 
272     /* Now make a QEMUIOVector taking enough granularity-sized chunks
273      * from s->buf_free.
274      */
275     qemu_iovec_init(&op->qiov, nb_chunks);
276     next_sector = sector_num;
277     while (nb_chunks-- > 0) {
278         MirrorBuffer *buf = QSIMPLEQ_FIRST(&s->buf_free);
279         size_t remaining = (nb_sectors * BDRV_SECTOR_SIZE) - op->qiov.size;
280 
281         QSIMPLEQ_REMOVE_HEAD(&s->buf_free, next);
282         s->buf_free_count--;
283         qemu_iovec_add(&op->qiov, buf, MIN(s->granularity, remaining));
284 
285         /* Advance the HBitmapIter in parallel, so that we do not examine
286          * the same sector twice.
287          */
288         if (next_sector > hbitmap_next_sector
289             && bdrv_get_dirty(source, s->dirty_bitmap, next_sector)) {
290             hbitmap_next_sector = hbitmap_iter_next(&s->hbi);
291         }
292 
293         next_sector += sectors_per_chunk;
294     }
295 
296     bdrv_reset_dirty_bitmap(s->dirty_bitmap, sector_num, nb_sectors);
297 
298     /* Copy the dirty cluster.  */
299     s->in_flight++;
300     s->sectors_in_flight += nb_sectors;
301     trace_mirror_one_iteration(s, sector_num, nb_sectors);
302 
303     ret = bdrv_get_block_status_above(source, NULL, sector_num,
304                                       nb_sectors, &pnum);
305     if (ret < 0 || pnum < nb_sectors ||
306             (ret & BDRV_BLOCK_DATA && !(ret & BDRV_BLOCK_ZERO))) {
307         bdrv_aio_readv(source, sector_num, &op->qiov, nb_sectors,
308                        mirror_read_complete, op);
309     } else if (ret & BDRV_BLOCK_ZERO) {
310         bdrv_aio_write_zeroes(s->target, sector_num, op->nb_sectors,
311                               s->unmap ? BDRV_REQ_MAY_UNMAP : 0,
312                               mirror_write_complete, op);
313     } else {
314         assert(!(ret & BDRV_BLOCK_DATA));
315         bdrv_aio_discard(s->target, sector_num, op->nb_sectors,
316                          mirror_write_complete, op);
317     }
318     return delay_ns;
319 }
320 
321 static void mirror_free_init(MirrorBlockJob *s)
322 {
323     int granularity = s->granularity;
324     size_t buf_size = s->buf_size;
325     uint8_t *buf = s->buf;
326 
327     assert(s->buf_free_count == 0);
328     QSIMPLEQ_INIT(&s->buf_free);
329     while (buf_size != 0) {
330         MirrorBuffer *cur = (MirrorBuffer *)buf;
331         QSIMPLEQ_INSERT_TAIL(&s->buf_free, cur, next);
332         s->buf_free_count++;
333         buf_size -= granularity;
334         buf += granularity;
335     }
336 }
337 
338 static void mirror_drain(MirrorBlockJob *s)
339 {
340     while (s->in_flight > 0) {
341         s->waiting_for_io = true;
342         qemu_coroutine_yield();
343         s->waiting_for_io = false;
344     }
345 }
346 
347 typedef struct {
348     int ret;
349 } MirrorExitData;
350 
351 static void mirror_exit(BlockJob *job, void *opaque)
352 {
353     MirrorBlockJob *s = container_of(job, MirrorBlockJob, common);
354     MirrorExitData *data = opaque;
355     AioContext *replace_aio_context = NULL;
356 
357     if (s->to_replace) {
358         replace_aio_context = bdrv_get_aio_context(s->to_replace);
359         aio_context_acquire(replace_aio_context);
360     }
361 
362     if (s->should_complete && data->ret == 0) {
363         BlockDriverState *to_replace = s->common.bs;
364         if (s->to_replace) {
365             to_replace = s->to_replace;
366         }
367         if (bdrv_get_flags(s->target) != bdrv_get_flags(to_replace)) {
368             bdrv_reopen(s->target, bdrv_get_flags(to_replace), NULL);
369         }
370         bdrv_swap(s->target, to_replace);
371         if (s->common.driver->job_type == BLOCK_JOB_TYPE_COMMIT) {
372             /* drop the bs loop chain formed by the swap: break the loop then
373              * trigger the unref from the top one */
374             BlockDriverState *p = s->base->backing_hd;
375             bdrv_set_backing_hd(s->base, NULL);
376             bdrv_unref(p);
377         }
378     }
379     if (s->to_replace) {
380         bdrv_op_unblock_all(s->to_replace, s->replace_blocker);
381         error_free(s->replace_blocker);
382         bdrv_unref(s->to_replace);
383     }
384     if (replace_aio_context) {
385         aio_context_release(replace_aio_context);
386     }
387     g_free(s->replaces);
388     bdrv_unref(s->target);
389     block_job_completed(&s->common, data->ret);
390     g_free(data);
391 }
392 
393 static void coroutine_fn mirror_run(void *opaque)
394 {
395     MirrorBlockJob *s = opaque;
396     MirrorExitData *data;
397     BlockDriverState *bs = s->common.bs;
398     int64_t sector_num, end, length;
399     uint64_t last_pause_ns;
400     BlockDriverInfo bdi;
401     char backing_filename[2]; /* we only need 2 characters because we are only
402                                  checking for a NULL string */
403     int ret = 0;
404     int n;
405 
406     if (block_job_is_cancelled(&s->common)) {
407         goto immediate_exit;
408     }
409 
410     s->bdev_length = bdrv_getlength(bs);
411     if (s->bdev_length < 0) {
412         ret = s->bdev_length;
413         goto immediate_exit;
414     } else if (s->bdev_length == 0) {
415         /* Report BLOCK_JOB_READY and wait for complete. */
416         block_job_event_ready(&s->common);
417         s->synced = true;
418         while (!block_job_is_cancelled(&s->common) && !s->should_complete) {
419             block_job_yield(&s->common);
420         }
421         s->common.cancelled = false;
422         goto immediate_exit;
423     }
424 
425     length = DIV_ROUND_UP(s->bdev_length, s->granularity);
426     s->in_flight_bitmap = bitmap_new(length);
427 
428     /* If we have no backing file yet in the destination, we cannot let
429      * the destination do COW.  Instead, we copy sectors around the
430      * dirty data if needed.  We need a bitmap to do that.
431      */
432     bdrv_get_backing_filename(s->target, backing_filename,
433                               sizeof(backing_filename));
434     if (backing_filename[0] && !s->target->backing_hd) {
435         ret = bdrv_get_info(s->target, &bdi);
436         if (ret < 0) {
437             goto immediate_exit;
438         }
439         if (s->granularity < bdi.cluster_size) {
440             s->buf_size = MAX(s->buf_size, bdi.cluster_size);
441             s->cow_bitmap = bitmap_new(length);
442         }
443     }
444 
445     end = s->bdev_length / BDRV_SECTOR_SIZE;
446     s->buf = qemu_try_blockalign(bs, s->buf_size);
447     if (s->buf == NULL) {
448         ret = -ENOMEM;
449         goto immediate_exit;
450     }
451 
452     mirror_free_init(s);
453 
454     last_pause_ns = qemu_clock_get_ns(QEMU_CLOCK_REALTIME);
455     if (!s->is_none_mode) {
456         /* First part, loop on the sectors and initialize the dirty bitmap.  */
457         BlockDriverState *base = s->base;
458         for (sector_num = 0; sector_num < end; ) {
459             /* Just to make sure we are not exceeding int limit. */
460             int nb_sectors = MIN(INT_MAX >> BDRV_SECTOR_BITS,
461                                  end - sector_num);
462             int64_t now = qemu_clock_get_ns(QEMU_CLOCK_REALTIME);
463 
464             if (now - last_pause_ns > SLICE_TIME) {
465                 last_pause_ns = now;
466                 block_job_sleep_ns(&s->common, QEMU_CLOCK_REALTIME, 0);
467             }
468 
469             if (block_job_is_cancelled(&s->common)) {
470                 goto immediate_exit;
471             }
472 
473             ret = bdrv_is_allocated_above(bs, base, sector_num, nb_sectors, &n);
474 
475             if (ret < 0) {
476                 goto immediate_exit;
477             }
478 
479             assert(n > 0);
480             if (ret == 1) {
481                 bdrv_set_dirty_bitmap(s->dirty_bitmap, sector_num, n);
482             }
483             sector_num += n;
484         }
485     }
486 
487     bdrv_dirty_iter_init(s->dirty_bitmap, &s->hbi);
488     for (;;) {
489         uint64_t delay_ns = 0;
490         int64_t cnt;
491         bool should_complete;
492 
493         if (s->ret < 0) {
494             ret = s->ret;
495             goto immediate_exit;
496         }
497 
498         cnt = bdrv_get_dirty_count(s->dirty_bitmap);
499         /* s->common.offset contains the number of bytes already processed so
500          * far, cnt is the number of dirty sectors remaining and
501          * s->sectors_in_flight is the number of sectors currently being
502          * processed; together those are the current total operation length */
503         s->common.len = s->common.offset +
504                         (cnt + s->sectors_in_flight) * BDRV_SECTOR_SIZE;
505 
506         /* Note that even when no rate limit is applied we need to yield
507          * periodically with no pending I/O so that bdrv_drain_all() returns.
508          * We do so every SLICE_TIME nanoseconds, or when there is an error,
509          * or when the source is clean, whichever comes first.
510          */
511         if (qemu_clock_get_ns(QEMU_CLOCK_REALTIME) - last_pause_ns < SLICE_TIME &&
512             s->common.iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
513             if (s->in_flight == MAX_IN_FLIGHT || s->buf_free_count == 0 ||
514                 (cnt == 0 && s->in_flight > 0)) {
515                 trace_mirror_yield(s, s->in_flight, s->buf_free_count, cnt);
516                 s->waiting_for_io = true;
517                 qemu_coroutine_yield();
518                 s->waiting_for_io = false;
519                 continue;
520             } else if (cnt != 0) {
521                 delay_ns = mirror_iteration(s);
522             }
523         }
524 
525         should_complete = false;
526         if (s->in_flight == 0 && cnt == 0) {
527             trace_mirror_before_flush(s);
528             ret = bdrv_flush(s->target);
529             if (ret < 0) {
530                 if (mirror_error_action(s, false, -ret) ==
531                     BLOCK_ERROR_ACTION_REPORT) {
532                     goto immediate_exit;
533                 }
534             } else {
535                 /* We're out of the streaming phase.  From now on, if the job
536                  * is cancelled we will actually complete all pending I/O and
537                  * report completion.  This way, block-job-cancel will leave
538                  * the target in a consistent state.
539                  */
540                 if (!s->synced) {
541                     block_job_event_ready(&s->common);
542                     s->synced = true;
543                 }
544 
545                 should_complete = s->should_complete ||
546                     block_job_is_cancelled(&s->common);
547                 cnt = bdrv_get_dirty_count(s->dirty_bitmap);
548             }
549         }
550 
551         if (cnt == 0 && should_complete) {
552             /* The dirty bitmap is not updated while operations are pending.
553              * If we're about to exit, wait for pending operations before
554              * calling bdrv_get_dirty_count(bs), or we may exit while the
555              * source has dirty data to copy!
556              *
557              * Note that I/O can be submitted by the guest while
558              * mirror_populate runs.
559              */
560             trace_mirror_before_drain(s, cnt);
561             bdrv_drain(bs);
562             cnt = bdrv_get_dirty_count(s->dirty_bitmap);
563         }
564 
565         ret = 0;
566         trace_mirror_before_sleep(s, cnt, s->synced, delay_ns);
567         if (!s->synced) {
568             block_job_sleep_ns(&s->common, QEMU_CLOCK_REALTIME, delay_ns);
569             if (block_job_is_cancelled(&s->common)) {
570                 break;
571             }
572         } else if (!should_complete) {
573             delay_ns = (s->in_flight == 0 && cnt == 0 ? SLICE_TIME : 0);
574             block_job_sleep_ns(&s->common, QEMU_CLOCK_REALTIME, delay_ns);
575         } else if (cnt == 0) {
576             /* The two disks are in sync.  Exit and report successful
577              * completion.
578              */
579             assert(QLIST_EMPTY(&bs->tracked_requests));
580             s->common.cancelled = false;
581             break;
582         }
583         last_pause_ns = qemu_clock_get_ns(QEMU_CLOCK_REALTIME);
584     }
585 
586 immediate_exit:
587     if (s->in_flight > 0) {
588         /* We get here only if something went wrong.  Either the job failed,
589          * or it was cancelled prematurely so that we do not guarantee that
590          * the target is a copy of the source.
591          */
592         assert(ret < 0 || (!s->synced && block_job_is_cancelled(&s->common)));
593         mirror_drain(s);
594     }
595 
596     assert(s->in_flight == 0);
597     qemu_vfree(s->buf);
598     g_free(s->cow_bitmap);
599     g_free(s->in_flight_bitmap);
600     bdrv_release_dirty_bitmap(bs, s->dirty_bitmap);
601     bdrv_iostatus_disable(s->target);
602 
603     data = g_malloc(sizeof(*data));
604     data->ret = ret;
605     block_job_defer_to_main_loop(&s->common, mirror_exit, data);
606 }
607 
608 static void mirror_set_speed(BlockJob *job, int64_t speed, Error **errp)
609 {
610     MirrorBlockJob *s = container_of(job, MirrorBlockJob, common);
611 
612     if (speed < 0) {
613         error_setg(errp, QERR_INVALID_PARAMETER, "speed");
614         return;
615     }
616     ratelimit_set_speed(&s->limit, speed / BDRV_SECTOR_SIZE, SLICE_TIME);
617 }
618 
619 static void mirror_iostatus_reset(BlockJob *job)
620 {
621     MirrorBlockJob *s = container_of(job, MirrorBlockJob, common);
622 
623     bdrv_iostatus_reset(s->target);
624 }
625 
626 static void mirror_complete(BlockJob *job, Error **errp)
627 {
628     MirrorBlockJob *s = container_of(job, MirrorBlockJob, common);
629     Error *local_err = NULL;
630     int ret;
631 
632     ret = bdrv_open_backing_file(s->target, NULL, &local_err);
633     if (ret < 0) {
634         error_propagate(errp, local_err);
635         return;
636     }
637     if (!s->synced) {
638         error_setg(errp, QERR_BLOCK_JOB_NOT_READY,
639                    bdrv_get_device_name(job->bs));
640         return;
641     }
642 
643     /* check the target bs is not blocked and block all operations on it */
644     if (s->replaces) {
645         AioContext *replace_aio_context;
646 
647         s->to_replace = bdrv_find_node(s->replaces);
648         if (!s->to_replace) {
649             error_setg(errp, "Node name '%s' not found", s->replaces);
650             return;
651         }
652 
653         replace_aio_context = bdrv_get_aio_context(s->to_replace);
654         aio_context_acquire(replace_aio_context);
655 
656         error_setg(&s->replace_blocker,
657                    "block device is in use by block-job-complete");
658         bdrv_op_block_all(s->to_replace, s->replace_blocker);
659         bdrv_ref(s->to_replace);
660 
661         aio_context_release(replace_aio_context);
662     }
663 
664     s->should_complete = true;
665     block_job_enter(&s->common);
666 }
667 
668 static const BlockJobDriver mirror_job_driver = {
669     .instance_size = sizeof(MirrorBlockJob),
670     .job_type      = BLOCK_JOB_TYPE_MIRROR,
671     .set_speed     = mirror_set_speed,
672     .iostatus_reset= mirror_iostatus_reset,
673     .complete      = mirror_complete,
674 };
675 
676 static const BlockJobDriver commit_active_job_driver = {
677     .instance_size = sizeof(MirrorBlockJob),
678     .job_type      = BLOCK_JOB_TYPE_COMMIT,
679     .set_speed     = mirror_set_speed,
680     .iostatus_reset
681                    = mirror_iostatus_reset,
682     .complete      = mirror_complete,
683 };
684 
685 static void mirror_start_job(BlockDriverState *bs, BlockDriverState *target,
686                              const char *replaces,
687                              int64_t speed, uint32_t granularity,
688                              int64_t buf_size,
689                              BlockdevOnError on_source_error,
690                              BlockdevOnError on_target_error,
691                              bool unmap,
692                              BlockCompletionFunc *cb,
693                              void *opaque, Error **errp,
694                              const BlockJobDriver *driver,
695                              bool is_none_mode, BlockDriverState *base)
696 {
697     MirrorBlockJob *s;
698 
699     if (granularity == 0) {
700         granularity = bdrv_get_default_bitmap_granularity(target);
701     }
702 
703     assert ((granularity & (granularity - 1)) == 0);
704 
705     if ((on_source_error == BLOCKDEV_ON_ERROR_STOP ||
706          on_source_error == BLOCKDEV_ON_ERROR_ENOSPC) &&
707         !bdrv_iostatus_is_enabled(bs)) {
708         error_setg(errp, QERR_INVALID_PARAMETER, "on-source-error");
709         return;
710     }
711 
712     if (buf_size < 0) {
713         error_setg(errp, "Invalid parameter 'buf-size'");
714         return;
715     }
716 
717     if (buf_size == 0) {
718         buf_size = DEFAULT_MIRROR_BUF_SIZE;
719     }
720 
721     s = block_job_create(driver, bs, speed, cb, opaque, errp);
722     if (!s) {
723         return;
724     }
725 
726     s->replaces = g_strdup(replaces);
727     s->on_source_error = on_source_error;
728     s->on_target_error = on_target_error;
729     s->target = target;
730     s->is_none_mode = is_none_mode;
731     s->base = base;
732     s->granularity = granularity;
733     s->buf_size = ROUND_UP(buf_size, granularity);
734     s->unmap = unmap;
735 
736     s->dirty_bitmap = bdrv_create_dirty_bitmap(bs, granularity, NULL, errp);
737     if (!s->dirty_bitmap) {
738         g_free(s->replaces);
739         block_job_release(bs);
740         return;
741     }
742     bdrv_set_enable_write_cache(s->target, true);
743     bdrv_set_on_error(s->target, on_target_error, on_target_error);
744     bdrv_iostatus_enable(s->target);
745     s->common.co = qemu_coroutine_create(mirror_run);
746     trace_mirror_start(bs, s, s->common.co, opaque);
747     qemu_coroutine_enter(s->common.co, s);
748 }
749 
750 void mirror_start(BlockDriverState *bs, BlockDriverState *target,
751                   const char *replaces,
752                   int64_t speed, uint32_t granularity, int64_t buf_size,
753                   MirrorSyncMode mode, BlockdevOnError on_source_error,
754                   BlockdevOnError on_target_error,
755                   bool unmap,
756                   BlockCompletionFunc *cb,
757                   void *opaque, Error **errp)
758 {
759     bool is_none_mode;
760     BlockDriverState *base;
761 
762     if (mode == MIRROR_SYNC_MODE_INCREMENTAL) {
763         error_setg(errp, "Sync mode 'incremental' not supported");
764         return;
765     }
766     is_none_mode = mode == MIRROR_SYNC_MODE_NONE;
767     base = mode == MIRROR_SYNC_MODE_TOP ? bs->backing_hd : NULL;
768     mirror_start_job(bs, target, replaces,
769                      speed, granularity, buf_size,
770                      on_source_error, on_target_error, unmap, cb, opaque, errp,
771                      &mirror_job_driver, is_none_mode, base);
772 }
773 
774 void commit_active_start(BlockDriverState *bs, BlockDriverState *base,
775                          int64_t speed,
776                          BlockdevOnError on_error,
777                          BlockCompletionFunc *cb,
778                          void *opaque, Error **errp)
779 {
780     int64_t length, base_length;
781     int orig_base_flags;
782     int ret;
783     Error *local_err = NULL;
784 
785     orig_base_flags = bdrv_get_flags(base);
786 
787     if (bdrv_reopen(base, bs->open_flags, errp)) {
788         return;
789     }
790 
791     length = bdrv_getlength(bs);
792     if (length < 0) {
793         error_setg_errno(errp, -length,
794                          "Unable to determine length of %s", bs->filename);
795         goto error_restore_flags;
796     }
797 
798     base_length = bdrv_getlength(base);
799     if (base_length < 0) {
800         error_setg_errno(errp, -base_length,
801                          "Unable to determine length of %s", base->filename);
802         goto error_restore_flags;
803     }
804 
805     if (length > base_length) {
806         ret = bdrv_truncate(base, length);
807         if (ret < 0) {
808             error_setg_errno(errp, -ret,
809                             "Top image %s is larger than base image %s, and "
810                              "resize of base image failed",
811                              bs->filename, base->filename);
812             goto error_restore_flags;
813         }
814     }
815 
816     bdrv_ref(base);
817     mirror_start_job(bs, base, NULL, speed, 0, 0,
818                      on_error, on_error, false, cb, opaque, &local_err,
819                      &commit_active_job_driver, false, base);
820     if (local_err) {
821         error_propagate(errp, local_err);
822         goto error_restore_flags;
823     }
824 
825     return;
826 
827 error_restore_flags:
828     /* ignore error and errp for bdrv_reopen, because we want to propagate
829      * the original error */
830     bdrv_reopen(base, orig_base_flags, NULL);
831     return;
832 }
833