X-Git-Url: https://git.proxmox.com/?a=blobdiff_plain;f=block-migration.c;h=daf9ec1eab77cfadc35cf70d8ea369934f1c0f29;hb=392a4d5b9a1860aae82429d09321dac302d70396;hp=f803f2006f4d8da00b02b77499ecdb1b717b5c41;hpb=a20bd9eec3cf3f9dd7e27d45a96d7b1b9ab09ebd;p=qemu.git

diff --git a/block-migration.c b/block-migration.c
index f803f2006..daf9ec1ea 100644
--- a/block-migration.c
+++ b/block-migration.c
@@ -336,8 +336,8 @@ 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));
         bdrv_set_in_use(bs, 1);
+        bdrv_ref(bs);
 
         block_mig_state.total_sector_sum += sectors;
 
@@ -575,7 +575,7 @@ static void blk_mig_cleanup(void)
     while ((bmds = QSIMPLEQ_FIRST(&block_mig_state.bmds_list)) != NULL) {
         QSIMPLEQ_REMOVE_HEAD(&block_mig_state.bmds_list, entry);
         bdrv_set_in_use(bmds->bs, 0);
-        drive_put_ref(drive_get_by_blockdev(bmds->bs));
+        bdrv_unref(bmds->bs);
         g_free(bmds->aio_bitmap);
         g_free(bmds);
     }