From f48905d44f670cd83227b3a8d06ae1406f0c771c Mon Sep 17 00:00:00 2001 From: Marcelo Tosatti Date: Wed, 26 Jan 2011 12:12:33 -0200 Subject: block-migration: add reference to target DriveInfo So that ejection of attached device by guest does not free data in use by block migration instance. Signed-off-by: Marcelo Tosatti CC: Markus Armbruster Signed-off-by: Kevin Wolf --- block-migration.c | 3 +++ 1 file changed, 3 insertions(+) (limited to 'block-migration.c') diff --git a/block-migration.c b/block-migration.c index 483ca7bab9..323e1e2bcd 100644 --- a/block-migration.c +++ b/block-migration.c @@ -19,6 +19,7 @@ #include "monitor.h" #include "block-migration.h" #include "migration.h" +#include "blockdev.h" #include #define BLOCK_SIZE (BDRV_SECTORS_PER_DIRTY_CHUNK << BDRV_SECTOR_BITS) @@ -299,6 +300,7 @@ static void init_blk_migration_it(void *opaque, BlockDriverState *bs) bmds->completed_sectors = 0; bmds->shared_base = block_mig_state.shared_base; alloc_aio_bitmap(bmds); + drive_get_ref(drive_get_by_blockdev(bs)); block_mig_state.total_sector_sum += sectors; @@ -537,6 +539,7 @@ static void blk_mig_cleanup(Monitor *mon) while ((bmds = QSIMPLEQ_FIRST(&block_mig_state.bmds_list)) != NULL) { QSIMPLEQ_REMOVE_HEAD(&block_mig_state.bmds_list, entry); + drive_put_ref(drive_get_by_blockdev(bmds->bs)); qemu_free(bmds->aio_bitmap); qemu_free(bmds); } -- cgit v1.2.1