qemu/vhost-implement-migration-state-notifier-for-vdpa-de.patch

From ce5fa02db01263ef5188b3bb3a1367c806ddb7ce Mon Sep 17 00:00:00 2001
From: fangyi <eric.fangyi@huawei.com>
Date: Mon, 4 Dec 2023 15:55:53 +0800
Subject: [PATCH] vhost: implement migration state notifier for vdpa device

Signed-off-by: libai <libai12@huawei.com>
Signed-off-by: jiangdongxu <jiangdongxu1@huawei.com>
Signed-off-by: fangyi <eric.fangyi@huawei.com>
---
 hw/virtio/vdpa-dev-mig.c     | 28 ++++++++++++++++++++++++++++
 include/hw/virtio/vdpa-dev.h |  1 +
 2 files changed, 29 insertions(+)

diff --git a/hw/virtio/vdpa-dev-mig.c b/hw/virtio/vdpa-dev-mig.c
index c0bcbda7d4..a36517b147 100644
--- a/hw/virtio/vdpa-dev-mig.c
+++ b/hw/virtio/vdpa-dev-mig.c
@@ -344,6 +344,31 @@ static SaveVMHandlers savevm_vdpa_handlers = {
     .load_setup = vdpa_load_setup,
 };
 
+static void vdpa_migration_state_notifier(Notifier *notifier, void *data)
+{
+    MigrationState *s = data;
+    VhostVdpaDevice *vdev = container_of(notifier,
+                                         VhostVdpaDevice,
+                                         migration_state);
+    struct vhost_dev *hdev = &vdev->dev;
+    int ret;
+
+    switch (s->state) {
+    case MIGRATION_STATUS_CANCELLING:
+    case MIGRATION_STATUS_CANCELLED:
+    case MIGRATION_STATUS_FAILED:
+        ret = vhost_vdpa_set_mig_state(hdev, VDPA_DEVICE_CANCEL);
+        if (ret) {
+            error_report("Failed to set state CANCEL\n");
+        }
+
+        break;
+    case MIGRATION_STATUS_COMPLETED:
+    default:
+        break;
+    }
+}
+
 void vdpa_migration_register(VhostVdpaDevice *vdev)
 {
     vdev->vmstate = qdev_add_vm_change_state_handler(DEVICE(vdev),
@@ -351,10 +376,13 @@ void vdpa_migration_register(VhostVdpaDevice *vdev)
                                                      DEVICE(vdev));
     register_savevm_live("vdpa", -1, 1,
                          &savevm_vdpa_handlers, DEVICE(vdev));
+    vdev->migration_state.notify = vdpa_migration_state_notifier;
+    add_migration_state_change_notifier(&vdev->migration_state);
 }
 
 void vdpa_migration_unregister(VhostVdpaDevice *vdev)
 {
+    remove_migration_state_change_notifier(&vdev->migration_state);
     unregister_savevm(VMSTATE_IF(&vdev->parent_obj.parent_obj), "vdpa", DEVICE(vdev));
     qemu_del_vm_change_state_handler(vdev->vmstate);
 }
diff --git a/include/hw/virtio/vdpa-dev.h b/include/hw/virtio/vdpa-dev.h
index 43cbcef81b..20f50c76c6 100644
--- a/include/hw/virtio/vdpa-dev.h
+++ b/include/hw/virtio/vdpa-dev.h
@@ -39,6 +39,7 @@ struct VhostVdpaDevice {
     bool started;
     int (*post_init)(VhostVdpaDevice *v, Error **errp);
     VMChangeStateEntry *vmstate;
+    Notifier migration_state;
 };
 
 #endif
-- 
2.27.0