2 * QEMU live block migration
4 * Copyright IBM, Corp. 2009
7 * Liran Schour <lirans@il.ibm.com>
9 * This work is licensed under the terms of the GNU GPL, version 2. See
10 * the COPYING file in the top-level directory.
12 * Contributions after 2012-01-13 are licensed under the terms of the
13 * GNU GPL, version 2 or (at your option) any later version.
16 #include "qemu/osdep.h"
17 #include "qapi/error.h"
18 #include "qemu/error-report.h"
19 #include "qemu/main-loop.h"
20 #include "qemu/cutils.h"
21 #include "qemu/queue.h"
23 #include "block/dirty-bitmap.h"
24 #include "migration/misc.h"
25 #include "migration.h"
26 #include "migration/register.h"
27 #include "qemu-file.h"
28 #include "migration/vmstate.h"
29 #include "sysemu/block-backend.h"
33 #define BLK_MIG_BLOCK_SIZE (1ULL << 20)
34 #define BDRV_SECTORS_PER_DIRTY_CHUNK (BLK_MIG_BLOCK_SIZE >> BDRV_SECTOR_BITS)
36 #define BLK_MIG_FLAG_DEVICE_BLOCK 0x01
37 #define BLK_MIG_FLAG_EOS 0x02
38 #define BLK_MIG_FLAG_PROGRESS 0x04
39 #define BLK_MIG_FLAG_ZERO_BLOCK 0x08
41 #define MAX_IS_ALLOCATED_SEARCH (65536 * BDRV_SECTOR_SIZE)
43 #define MAX_IO_BUFFERS 512
44 #define MAX_PARALLEL_IO 16
46 typedef struct BlkMigDevState
{
47 /* Written during setup phase. Can be read without a lock. */
51 int64_t total_sectors
;
52 QSIMPLEQ_ENTRY(BlkMigDevState
) entry
;
55 /* Only used by migration thread. Does not need a lock. */
60 /* Data in the aio_bitmap is protected by block migration lock.
61 * Allocation and free happen during setup and cleanup respectively.
63 unsigned long *aio_bitmap
;
65 /* Protected by block migration lock. */
66 int64_t completed_sectors
;
68 /* During migration this is protected by iothread lock / AioContext.
69 * Allocation and free happen during setup and cleanup respectively.
71 BdrvDirtyBitmap
*dirty_bitmap
;
74 typedef struct BlkMigBlock
{
75 /* Only used by migration thread. */
83 /* Protected by block migration lock. */
85 QSIMPLEQ_ENTRY(BlkMigBlock
) entry
;
88 typedef struct BlkMigState
{
89 QSIMPLEQ_HEAD(, BlkMigDevState
) bmds_list
;
90 int64_t total_sector_sum
;
93 /* Protected by lock. */
94 QSIMPLEQ_HEAD(, BlkMigBlock
) blk_list
;
98 /* Only used by migration thread. Does not need a lock. */
103 /* Lock must be taken _inside_ the iothread lock and any AioContexts. */
107 static BlkMigState block_mig_state
;
109 static void blk_mig_lock(void)
111 qemu_mutex_lock(&block_mig_state
.lock
);
114 static void blk_mig_unlock(void)
116 qemu_mutex_unlock(&block_mig_state
.lock
);
119 /* Must run outside of the iothread lock during the bulk phase,
120 * or the VM will stall.
123 static void blk_send(QEMUFile
*f
, BlkMigBlock
* blk
)
126 uint64_t flags
= BLK_MIG_FLAG_DEVICE_BLOCK
;
128 if (block_mig_state
.zero_blocks
&&
129 buffer_is_zero(blk
->buf
, BLK_MIG_BLOCK_SIZE
)) {
130 flags
|= BLK_MIG_FLAG_ZERO_BLOCK
;
133 /* sector number and flags */
134 qemu_put_be64(f
, (blk
->sector
<< BDRV_SECTOR_BITS
)
138 len
= strlen(blk
->bmds
->blk_name
);
139 qemu_put_byte(f
, len
);
140 qemu_put_buffer(f
, (uint8_t *) blk
->bmds
->blk_name
, len
);
142 /* if a block is zero we need to flush here since the network
143 * bandwidth is now a lot higher than the storage device bandwidth.
144 * thus if we queue zero blocks we slow down the migration */
145 if (flags
& BLK_MIG_FLAG_ZERO_BLOCK
) {
150 qemu_put_buffer(f
, blk
->buf
, BLK_MIG_BLOCK_SIZE
);
153 int blk_mig_active(void)
155 return !QSIMPLEQ_EMPTY(&block_mig_state
.bmds_list
);
158 int blk_mig_bulk_active(void)
160 return blk_mig_active() && !block_mig_state
.bulk_completed
;
163 uint64_t blk_mig_bytes_transferred(void)
165 BlkMigDevState
*bmds
;
169 QSIMPLEQ_FOREACH(bmds
, &block_mig_state
.bmds_list
, entry
) {
170 sum
+= bmds
->completed_sectors
;
173 return sum
<< BDRV_SECTOR_BITS
;
176 uint64_t blk_mig_bytes_remaining(void)
178 return blk_mig_bytes_total() - blk_mig_bytes_transferred();
181 uint64_t blk_mig_bytes_total(void)
183 BlkMigDevState
*bmds
;
186 QSIMPLEQ_FOREACH(bmds
, &block_mig_state
.bmds_list
, entry
) {
187 sum
+= bmds
->total_sectors
;
189 return sum
<< BDRV_SECTOR_BITS
;
193 /* Called with migration lock held. */
195 static int bmds_aio_inflight(BlkMigDevState
*bmds
, int64_t sector
)
197 int64_t chunk
= sector
/ (int64_t)BDRV_SECTORS_PER_DIRTY_CHUNK
;
199 if (sector
< bmds
->total_sectors
) {
200 return !!(bmds
->aio_bitmap
[chunk
/ (sizeof(unsigned long) * 8)] &
201 (1UL << (chunk
% (sizeof(unsigned long) * 8))));
207 /* Called with migration lock held. */
209 static void bmds_set_aio_inflight(BlkMigDevState
*bmds
, int64_t sector_num
,
210 int nb_sectors
, int set
)
213 unsigned long val
, idx
, bit
;
215 start
= sector_num
/ BDRV_SECTORS_PER_DIRTY_CHUNK
;
216 end
= (sector_num
+ nb_sectors
- 1) / BDRV_SECTORS_PER_DIRTY_CHUNK
;
218 for (; start
<= end
; start
++) {
219 idx
= start
/ (sizeof(unsigned long) * 8);
220 bit
= start
% (sizeof(unsigned long) * 8);
221 val
= bmds
->aio_bitmap
[idx
];
225 val
&= ~(1UL << bit
);
227 bmds
->aio_bitmap
[idx
] = val
;
231 static void alloc_aio_bitmap(BlkMigDevState
*bmds
)
235 bitmap_size
= bmds
->total_sectors
+ BDRV_SECTORS_PER_DIRTY_CHUNK
* 8 - 1;
236 bitmap_size
/= BDRV_SECTORS_PER_DIRTY_CHUNK
* 8;
238 bmds
->aio_bitmap
= g_malloc0(bitmap_size
);
241 /* Never hold migration lock when yielding to the main loop! */
243 static void blk_mig_read_cb(void *opaque
, int ret
)
245 BlkMigBlock
*blk
= opaque
;
250 QSIMPLEQ_INSERT_TAIL(&block_mig_state
.blk_list
, blk
, entry
);
251 bmds_set_aio_inflight(blk
->bmds
, blk
->sector
, blk
->nr_sectors
, 0);
253 block_mig_state
.submitted
--;
254 block_mig_state
.read_done
++;
255 assert(block_mig_state
.submitted
>= 0);
259 /* Called with no lock taken. */
261 static int mig_save_device_bulk(QEMUFile
*f
, BlkMigDevState
*bmds
)
263 int64_t total_sectors
= bmds
->total_sectors
;
264 int64_t cur_sector
= bmds
->cur_sector
;
265 BlockBackend
*bb
= bmds
->blk
;
270 if (bmds
->shared_base
) {
271 qemu_mutex_lock_iothread();
272 aio_context_acquire(blk_get_aio_context(bb
));
273 /* Skip unallocated sectors; intentionally treats failure or
274 * partial sector as an allocated sector */
275 while (cur_sector
< total_sectors
&&
276 !bdrv_is_allocated(blk_bs(bb
), cur_sector
* BDRV_SECTOR_SIZE
,
277 MAX_IS_ALLOCATED_SEARCH
, &count
)) {
278 if (count
< BDRV_SECTOR_SIZE
) {
281 cur_sector
+= count
>> BDRV_SECTOR_BITS
;
283 aio_context_release(blk_get_aio_context(bb
));
284 qemu_mutex_unlock_iothread();
287 if (cur_sector
>= total_sectors
) {
288 bmds
->cur_sector
= bmds
->completed_sectors
= total_sectors
;
292 bmds
->completed_sectors
= cur_sector
;
294 cur_sector
&= ~((int64_t)BDRV_SECTORS_PER_DIRTY_CHUNK
- 1);
296 /* we are going to transfer a full block even if it is not allocated */
297 nr_sectors
= BDRV_SECTORS_PER_DIRTY_CHUNK
;
299 if (total_sectors
- cur_sector
< BDRV_SECTORS_PER_DIRTY_CHUNK
) {
300 nr_sectors
= total_sectors
- cur_sector
;
303 blk
= g_new(BlkMigBlock
, 1);
304 blk
->buf
= g_malloc(BLK_MIG_BLOCK_SIZE
);
306 blk
->sector
= cur_sector
;
307 blk
->nr_sectors
= nr_sectors
;
309 qemu_iovec_init_buf(&blk
->qiov
, blk
->buf
, nr_sectors
* BDRV_SECTOR_SIZE
);
312 block_mig_state
.submitted
++;
315 /* We do not know if bs is under the main thread (and thus does
316 * not acquire the AioContext when doing AIO) or rather under
317 * dataplane. Thus acquire both the iothread mutex and the
320 * This is ugly and will disappear when we make bdrv_* thread-safe,
321 * without the need to acquire the AioContext.
323 qemu_mutex_lock_iothread();
324 aio_context_acquire(blk_get_aio_context(bmds
->blk
));
325 bdrv_reset_dirty_bitmap(bmds
->dirty_bitmap
, cur_sector
* BDRV_SECTOR_SIZE
,
326 nr_sectors
* BDRV_SECTOR_SIZE
);
327 blk
->aiocb
= blk_aio_preadv(bb
, cur_sector
* BDRV_SECTOR_SIZE
, &blk
->qiov
,
328 0, blk_mig_read_cb
, blk
);
329 aio_context_release(blk_get_aio_context(bmds
->blk
));
330 qemu_mutex_unlock_iothread();
332 bmds
->cur_sector
= cur_sector
+ nr_sectors
;
333 return (bmds
->cur_sector
>= total_sectors
);
336 /* Called with iothread lock taken. */
338 static int set_dirty_tracking(void)
340 BlkMigDevState
*bmds
;
343 QSIMPLEQ_FOREACH(bmds
, &block_mig_state
.bmds_list
, entry
) {
344 bmds
->dirty_bitmap
= bdrv_create_dirty_bitmap(blk_bs(bmds
->blk
),
347 if (!bmds
->dirty_bitmap
) {
355 QSIMPLEQ_FOREACH(bmds
, &block_mig_state
.bmds_list
, entry
) {
356 if (bmds
->dirty_bitmap
) {
357 bdrv_release_dirty_bitmap(bmds
->dirty_bitmap
);
363 /* Called with iothread lock taken. */
365 static void unset_dirty_tracking(void)
367 BlkMigDevState
*bmds
;
369 QSIMPLEQ_FOREACH(bmds
, &block_mig_state
.bmds_list
, entry
) {
370 bdrv_release_dirty_bitmap(bmds
->dirty_bitmap
);
374 static int init_blk_migration(QEMUFile
*f
)
376 BlockDriverState
*bs
;
377 BlkMigDevState
*bmds
;
382 BlkMigDevState
*bmds
;
383 BlockDriverState
*bs
;
385 Error
*local_err
= NULL
;
388 block_mig_state
.submitted
= 0;
389 block_mig_state
.read_done
= 0;
390 block_mig_state
.transferred
= 0;
391 block_mig_state
.total_sector_sum
= 0;
392 block_mig_state
.prev_progress
= -1;
393 block_mig_state
.bulk_completed
= 0;
394 block_mig_state
.zero_blocks
= migrate_zero_blocks();
396 for (bs
= bdrv_first(&it
); bs
; bs
= bdrv_next(&it
)) {
399 bmds_bs
= g_malloc0(num_bs
* sizeof(*bmds_bs
));
401 for (i
= 0, bs
= bdrv_first(&it
); bs
; bs
= bdrv_next(&it
), i
++) {
402 if (bdrv_is_read_only(bs
)) {
406 sectors
= bdrv_nb_sectors(bs
);
409 bdrv_next_cleanup(&it
);
413 bmds
= g_new0(BlkMigDevState
, 1);
414 bmds
->blk
= blk_new(qemu_get_aio_context(),
415 BLK_PERM_CONSISTENT_READ
, BLK_PERM_ALL
);
416 bmds
->blk_name
= g_strdup(bdrv_get_device_name(bs
));
417 bmds
->bulk_completed
= 0;
418 bmds
->total_sectors
= sectors
;
419 bmds
->completed_sectors
= 0;
420 bmds
->shared_base
= migrate_block_incremental();
423 bmds_bs
[i
].bmds
= bmds
;
426 block_mig_state
.total_sector_sum
+= sectors
;
428 if (bmds
->shared_base
) {
429 trace_migration_block_init_shared(bdrv_get_device_name(bs
));
431 trace_migration_block_init_full(bdrv_get_device_name(bs
));
434 QSIMPLEQ_INSERT_TAIL(&block_mig_state
.bmds_list
, bmds
, entry
);
437 /* Can only insert new BDSes now because doing so while iterating block
438 * devices may end up in a deadlock (iterating the new BDSes, too). */
439 for (i
= 0; i
< num_bs
; i
++) {
440 BlkMigDevState
*bmds
= bmds_bs
[i
].bmds
;
441 BlockDriverState
*bs
= bmds_bs
[i
].bs
;
444 ret
= blk_insert_bs(bmds
->blk
, bs
, &local_err
);
446 error_report_err(local_err
);
450 alloc_aio_bitmap(bmds
);
451 error_setg(&bmds
->blocker
, "block device is in use by migration");
452 bdrv_op_block_all(bs
, bmds
->blocker
);
462 /* Called with no lock taken. */
464 static int blk_mig_save_bulked_block(QEMUFile
*f
)
466 int64_t completed_sector_sum
= 0;
467 BlkMigDevState
*bmds
;
471 QSIMPLEQ_FOREACH(bmds
, &block_mig_state
.bmds_list
, entry
) {
472 if (bmds
->bulk_completed
== 0) {
473 if (mig_save_device_bulk(f
, bmds
) == 1) {
474 /* completed bulk section for this device */
475 bmds
->bulk_completed
= 1;
477 completed_sector_sum
+= bmds
->completed_sectors
;
481 completed_sector_sum
+= bmds
->completed_sectors
;
485 if (block_mig_state
.total_sector_sum
!= 0) {
486 progress
= completed_sector_sum
* 100 /
487 block_mig_state
.total_sector_sum
;
491 if (progress
!= block_mig_state
.prev_progress
) {
492 block_mig_state
.prev_progress
= progress
;
493 qemu_put_be64(f
, (progress
<< BDRV_SECTOR_BITS
)
494 | BLK_MIG_FLAG_PROGRESS
);
495 trace_migration_block_progression(progress
);
501 static void blk_mig_reset_dirty_cursor(void)
503 BlkMigDevState
*bmds
;
505 QSIMPLEQ_FOREACH(bmds
, &block_mig_state
.bmds_list
, entry
) {
510 /* Called with iothread lock and AioContext taken. */
512 static int mig_save_device_dirty(QEMUFile
*f
, BlkMigDevState
*bmds
,
516 int64_t total_sectors
= bmds
->total_sectors
;
521 for (sector
= bmds
->cur_dirty
; sector
< bmds
->total_sectors
;) {
523 if (bmds_aio_inflight(bmds
, sector
)) {
525 blk_drain(bmds
->blk
);
529 bdrv_dirty_bitmap_lock(bmds
->dirty_bitmap
);
530 if (bdrv_dirty_bitmap_get_locked(bmds
->dirty_bitmap
,
531 sector
* BDRV_SECTOR_SIZE
)) {
532 if (total_sectors
- sector
< BDRV_SECTORS_PER_DIRTY_CHUNK
) {
533 nr_sectors
= total_sectors
- sector
;
535 nr_sectors
= BDRV_SECTORS_PER_DIRTY_CHUNK
;
537 bdrv_reset_dirty_bitmap_locked(bmds
->dirty_bitmap
,
538 sector
* BDRV_SECTOR_SIZE
,
539 nr_sectors
* BDRV_SECTOR_SIZE
);
540 bdrv_dirty_bitmap_unlock(bmds
->dirty_bitmap
);
542 blk
= g_new(BlkMigBlock
, 1);
543 blk
->buf
= g_malloc(BLK_MIG_BLOCK_SIZE
);
545 blk
->sector
= sector
;
546 blk
->nr_sectors
= nr_sectors
;
549 qemu_iovec_init_buf(&blk
->qiov
, blk
->buf
,
550 nr_sectors
* BDRV_SECTOR_SIZE
);
552 blk
->aiocb
= blk_aio_preadv(bmds
->blk
,
553 sector
* BDRV_SECTOR_SIZE
,
554 &blk
->qiov
, 0, blk_mig_read_cb
,
558 block_mig_state
.submitted
++;
559 bmds_set_aio_inflight(bmds
, sector
, nr_sectors
, 1);
562 ret
= blk_pread(bmds
->blk
, sector
* BDRV_SECTOR_SIZE
,
563 nr_sectors
* BDRV_SECTOR_SIZE
, blk
->buf
, 0);
573 sector
+= nr_sectors
;
574 bmds
->cur_dirty
= sector
;
578 bdrv_dirty_bitmap_unlock(bmds
->dirty_bitmap
);
579 sector
+= BDRV_SECTORS_PER_DIRTY_CHUNK
;
580 bmds
->cur_dirty
= sector
;
583 return (bmds
->cur_dirty
>= bmds
->total_sectors
);
586 trace_migration_block_save_device_dirty(sector
);
592 /* Called with iothread lock taken.
595 * 0: too much data for max_downtime
596 * 1: few enough data for max_downtime
598 static int blk_mig_save_dirty_block(QEMUFile
*f
, int is_async
)
600 BlkMigDevState
*bmds
;
603 QSIMPLEQ_FOREACH(bmds
, &block_mig_state
.bmds_list
, entry
) {
604 aio_context_acquire(blk_get_aio_context(bmds
->blk
));
605 ret
= mig_save_device_dirty(f
, bmds
, is_async
);
606 aio_context_release(blk_get_aio_context(bmds
->blk
));
615 /* Called with no locks taken. */
617 static int flush_blks(QEMUFile
*f
)
622 trace_migration_block_flush_blks("Enter", block_mig_state
.submitted
,
623 block_mig_state
.read_done
,
624 block_mig_state
.transferred
);
627 while ((blk
= QSIMPLEQ_FIRST(&block_mig_state
.blk_list
)) != NULL
) {
628 if (qemu_file_rate_limit(f
)) {
636 QSIMPLEQ_REMOVE_HEAD(&block_mig_state
.blk_list
, entry
);
644 block_mig_state
.read_done
--;
645 block_mig_state
.transferred
++;
646 assert(block_mig_state
.read_done
>= 0);
650 trace_migration_block_flush_blks("Exit", block_mig_state
.submitted
,
651 block_mig_state
.read_done
,
652 block_mig_state
.transferred
);
656 /* Called with iothread lock taken. */
658 static int64_t get_remaining_dirty(void)
660 BlkMigDevState
*bmds
;
663 QSIMPLEQ_FOREACH(bmds
, &block_mig_state
.bmds_list
, entry
) {
664 aio_context_acquire(blk_get_aio_context(bmds
->blk
));
665 dirty
+= bdrv_get_dirty_count(bmds
->dirty_bitmap
);
666 aio_context_release(blk_get_aio_context(bmds
->blk
));
674 /* Called with iothread lock taken. */
675 static void block_migration_cleanup_bmds(void)
677 BlkMigDevState
*bmds
;
680 unset_dirty_tracking();
682 while ((bmds
= QSIMPLEQ_FIRST(&block_mig_state
.bmds_list
)) != NULL
) {
683 QSIMPLEQ_REMOVE_HEAD(&block_mig_state
.bmds_list
, entry
);
684 bdrv_op_unblock_all(blk_bs(bmds
->blk
), bmds
->blocker
);
685 error_free(bmds
->blocker
);
687 /* Save ctx, because bmds->blk can disappear during blk_unref. */
688 ctx
= blk_get_aio_context(bmds
->blk
);
689 aio_context_acquire(ctx
);
690 blk_unref(bmds
->blk
);
691 aio_context_release(ctx
);
693 g_free(bmds
->blk_name
);
694 g_free(bmds
->aio_bitmap
);
699 /* Called with iothread lock taken. */
700 static void block_migration_cleanup(void *opaque
)
706 block_migration_cleanup_bmds();
709 while ((blk
= QSIMPLEQ_FIRST(&block_mig_state
.blk_list
)) != NULL
) {
710 QSIMPLEQ_REMOVE_HEAD(&block_mig_state
.blk_list
, entry
);
717 static int block_save_setup(QEMUFile
*f
, void *opaque
)
721 trace_migration_block_save("setup", block_mig_state
.submitted
,
722 block_mig_state
.transferred
);
724 qemu_mutex_lock_iothread();
725 ret
= init_blk_migration(f
);
727 qemu_mutex_unlock_iothread();
731 /* start track dirty blocks */
732 ret
= set_dirty_tracking();
734 qemu_mutex_unlock_iothread();
741 blk_mig_reset_dirty_cursor();
742 qemu_put_be64(f
, BLK_MIG_FLAG_EOS
);
747 static int block_save_iterate(QEMUFile
*f
, void *opaque
)
750 uint64_t last_bytes
= qemu_file_total_transferred(f
);
752 trace_migration_block_save("iterate", block_mig_state
.submitted
,
753 block_mig_state
.transferred
);
760 blk_mig_reset_dirty_cursor();
762 /* control the rate of transfer */
764 while (block_mig_state
.read_done
* BLK_MIG_BLOCK_SIZE
<
765 qemu_file_get_rate_limit(f
) &&
766 block_mig_state
.submitted
< MAX_PARALLEL_IO
&&
767 (block_mig_state
.submitted
+ block_mig_state
.read_done
) <
770 if (block_mig_state
.bulk_completed
== 0) {
771 /* first finish the bulk phase */
772 if (blk_mig_save_bulked_block(f
) == 0) {
773 /* finished saving bulk on all devices */
774 block_mig_state
.bulk_completed
= 1;
778 /* Always called with iothread lock taken for
779 * simplicity, block_save_complete also calls it.
781 qemu_mutex_lock_iothread();
782 ret
= blk_mig_save_dirty_block(f
, 1);
783 qemu_mutex_unlock_iothread();
790 /* no more dirty blocks */
801 qemu_put_be64(f
, BLK_MIG_FLAG_EOS
);
802 uint64_t delta_bytes
= qemu_file_total_transferred(f
) - last_bytes
;
803 return (delta_bytes
> 0);
806 /* Called with iothread lock taken. */
808 static int block_save_complete(QEMUFile
*f
, void *opaque
)
812 trace_migration_block_save("complete", block_mig_state
.submitted
,
813 block_mig_state
.transferred
);
820 blk_mig_reset_dirty_cursor();
822 /* we know for sure that save bulk is completed and
823 all async read completed */
825 assert(block_mig_state
.submitted
== 0);
829 ret
= blk_mig_save_dirty_block(f
, 0);
835 /* report completion */
836 qemu_put_be64(f
, (100 << BDRV_SECTOR_BITS
) | BLK_MIG_FLAG_PROGRESS
);
838 trace_migration_block_save_complete();
840 qemu_put_be64(f
, BLK_MIG_FLAG_EOS
);
842 /* Make sure that our BlockBackends are gone, so that the block driver
843 * nodes can be inactivated. */
844 block_migration_cleanup_bmds();
849 static void block_state_pending(void *opaque
, uint64_t *must_precopy
,
850 uint64_t *can_postcopy
)
852 /* Estimate pending number of bytes to send */
855 qemu_mutex_lock_iothread();
856 pending
= get_remaining_dirty();
857 qemu_mutex_unlock_iothread();
860 pending
+= block_mig_state
.submitted
* BLK_MIG_BLOCK_SIZE
+
861 block_mig_state
.read_done
* BLK_MIG_BLOCK_SIZE
;
864 /* Report at least one block pending during bulk phase */
865 if (!pending
&& !block_mig_state
.bulk_completed
) {
866 pending
= BLK_MIG_BLOCK_SIZE
;
869 trace_migration_block_state_pending(pending
);
870 /* We don't do postcopy */
871 *must_precopy
+= pending
;
874 static int block_load(QEMUFile
*f
, void *opaque
, int version_id
)
876 static int banner_printed
;
878 char device_name
[256];
880 BlockBackend
*blk
, *blk_prev
= NULL
;
881 Error
*local_err
= NULL
;
883 int64_t total_sectors
= 0;
887 int cluster_size
= BLK_MIG_BLOCK_SIZE
;
890 addr
= qemu_get_be64(f
);
892 flags
= addr
& (BDRV_SECTOR_SIZE
- 1);
893 addr
>>= BDRV_SECTOR_BITS
;
895 if (flags
& BLK_MIG_FLAG_DEVICE_BLOCK
) {
896 /* get device name */
897 len
= qemu_get_byte(f
);
898 qemu_get_buffer(f
, (uint8_t *)device_name
, len
);
899 device_name
[len
] = '\0';
901 blk
= blk_by_name(device_name
);
903 fprintf(stderr
, "Error unknown block device %s\n",
908 if (blk
!= blk_prev
) {
910 total_sectors
= blk_nb_sectors(blk
);
911 if (total_sectors
<= 0) {
912 error_report("Error getting length of block device %s",
917 blk_activate(blk
, &local_err
);
919 error_report_err(local_err
);
923 ret
= bdrv_get_info(blk_bs(blk
), &bdi
);
924 if (ret
== 0 && bdi
.cluster_size
> 0 &&
925 bdi
.cluster_size
<= BLK_MIG_BLOCK_SIZE
&&
926 BLK_MIG_BLOCK_SIZE
% bdi
.cluster_size
== 0) {
927 cluster_size
= bdi
.cluster_size
;
929 cluster_size
= BLK_MIG_BLOCK_SIZE
;
933 if (total_sectors
- addr
< BDRV_SECTORS_PER_DIRTY_CHUNK
) {
934 nr_sectors
= total_sectors
- addr
;
936 nr_sectors
= BDRV_SECTORS_PER_DIRTY_CHUNK
;
939 if (flags
& BLK_MIG_FLAG_ZERO_BLOCK
) {
940 ret
= blk_pwrite_zeroes(blk
, addr
* BDRV_SECTOR_SIZE
,
941 nr_sectors
* BDRV_SECTOR_SIZE
,
948 buf
= g_malloc(BLK_MIG_BLOCK_SIZE
);
949 qemu_get_buffer(f
, buf
, BLK_MIG_BLOCK_SIZE
);
950 for (i
= 0; i
< BLK_MIG_BLOCK_SIZE
/ cluster_size
; i
++) {
951 cur_addr
= addr
* BDRV_SECTOR_SIZE
+ i
* cluster_size
;
952 cur_buf
= buf
+ i
* cluster_size
;
954 if ((!block_mig_state
.zero_blocks
||
955 cluster_size
< BLK_MIG_BLOCK_SIZE
) &&
956 buffer_is_zero(cur_buf
, cluster_size
)) {
957 ret
= blk_pwrite_zeroes(blk
, cur_addr
,
961 ret
= blk_pwrite(blk
, cur_addr
, cluster_size
, cur_buf
,
974 } else if (flags
& BLK_MIG_FLAG_PROGRESS
) {
975 if (!banner_printed
) {
976 printf("Receiving block device images\n");
979 printf("Completed %d %%%c", (int)addr
,
980 (addr
== 100) ? '\n' : '\r');
982 } else if (!(flags
& BLK_MIG_FLAG_EOS
)) {
983 fprintf(stderr
, "Unknown block migration flags: 0x%x\n", flags
);
986 ret
= qemu_file_get_error(f
);
990 } while (!(flags
& BLK_MIG_FLAG_EOS
));
995 static bool block_is_active(void *opaque
)
997 return migrate_block();
1000 static SaveVMHandlers savevm_block_handlers
= {
1001 .save_setup
= block_save_setup
,
1002 .save_live_iterate
= block_save_iterate
,
1003 .save_live_complete_precopy
= block_save_complete
,
1004 .state_pending_exact
= block_state_pending
,
1005 .state_pending_estimate
= block_state_pending
,
1006 .load_state
= block_load
,
1007 .save_cleanup
= block_migration_cleanup
,
1008 .is_active
= block_is_active
,
1011 void blk_mig_init(void)
1013 QSIMPLEQ_INIT(&block_mig_state
.bmds_list
);
1014 QSIMPLEQ_INIT(&block_mig_state
.blk_list
);
1015 qemu_mutex_init(&block_mig_state
.lock
);
1017 register_savevm_live("block", 0, 1, &savevm_block_handlers
,