2983660
From: Paolo Bonzini <pbonzini@redhat.com>
2983660
Date: Fri, 20 Sep 2013 16:57:54 +0200
2983660
Subject: [PATCH] virtio-blk: switch exit callback to VirtioDeviceClass
2983660
2983660
This ensures hot-unplug is handled properly by the proxy.
2983660
2983660
Cc: qemu-stable@nongnu.org
2983660
Signed-off-by: Paolo Bonzini <pbonzini@redhat.com>
2983660
---
2983660
 hw/block/virtio-blk.c | 10 ++++------
2983660
 1 file changed, 4 insertions(+), 6 deletions(-)
2983660
2983660
diff --git a/hw/block/virtio-blk.c b/hw/block/virtio-blk.c
2983660
index 49a23c3..aa37cc9 100644
2983660
--- a/hw/block/virtio-blk.c
2983660
+++ b/hw/block/virtio-blk.c
2983660
@@ -729,20 +729,18 @@ static int virtio_blk_device_init(VirtIODevice *vdev)
2983660
     return 0;
2983660
 }
2983660
 
2983660
-static int virtio_blk_device_exit(DeviceState *dev)
2983660
+static void virtio_blk_device_exit(VirtIODevice *vdev)
2983660
 {
2983660
-    VirtIODevice *vdev = VIRTIO_DEVICE(dev);
2983660
-    VirtIOBlock *s = VIRTIO_BLK(dev);
2983660
+    VirtIOBlock *s = VIRTIO_BLK(vdev);
2983660
 #ifdef CONFIG_VIRTIO_BLK_DATA_PLANE
2983660
     remove_migration_state_change_notifier(&s->migration_state_notifier);
2983660
     virtio_blk_data_plane_destroy(s->dataplane);
2983660
     s->dataplane = NULL;
2983660
 #endif
2983660
     qemu_del_vm_change_state_handler(s->change);
2983660
-    unregister_savevm(dev, "virtio-blk", s);
2983660
+    unregister_savevm(DEVICE(vdev), "virtio-blk", s);
2983660
     blockdev_mark_auto_del(s->bs);
2983660
     virtio_cleanup(vdev);
2983660
-    return 0;
2983660
 }
2983660
 
2983660
 static Property virtio_blk_properties[] = {
2983660
@@ -754,10 +752,10 @@ static void virtio_blk_class_init(ObjectClass *klass, void *data)
2983660
 {
2983660
     DeviceClass *dc = DEVICE_CLASS(klass);
2983660
     VirtioDeviceClass *vdc = VIRTIO_DEVICE_CLASS(klass);
2983660
-    dc->exit = virtio_blk_device_exit;
2983660
     dc->props = virtio_blk_properties;
2983660
     set_bit(DEVICE_CATEGORY_STORAGE, dc->categories);
2983660
     vdc->init = virtio_blk_device_init;
2983660
+    vdc->exit = virtio_blk_device_exit;
2983660
     vdc->get_config = virtio_blk_update_config;
2983660
     vdc->set_config = virtio_blk_set_config;
2983660
     vdc->get_features = virtio_blk_get_features;