vmm: Notify Migratable objects about migration being complete

Signed-off-by: Sebastien Boeuf <sebastien.boeuf@intel.com>

vmm/src/device_manager.rs

@@ -4188,6 +4188,15 @@ impl Migratable for DeviceManager {
         }
         Ok(MemoryRangeTable::new_from_tables(tables))
     }
+
+    fn complete_migration(&mut self) -> std::result::Result<(), MigratableError> {
+        for (_, device_node) in self.device_tree.lock().unwrap().iter() {
+            if let Some(migratable) = &device_node.migratable {
+                migratable.lock().unwrap().complete_migration()?;
+            }
+        }
+        Ok(())
+    }
 }
 
 const PCIU_FIELD_OFFSET: u64 = 0;

vmm/src/lib.rs

@@ -1135,6 +1135,9 @@ impl Vmm {
             } {
                 // Stop logging dirty pages and keep the source VM paused unpon successful migration
                 Ok(()) => {
+                    // Let every Migratable object know about the migration being complete
+                    vm.complete_migration()?;
+
                     // Stop logging dirty pages
                     vm.stop_dirty_log()?;
 

vmm/src/vm.rs

@@ -2547,6 +2547,11 @@ impl Migratable for Vm {
             self.device_manager.lock().unwrap().dirty_log()?,
         ]))
     }
+
+    fn complete_migration(&mut self) -> std::result::Result<(), MigratableError> {
+        self.memory_manager.lock().unwrap().complete_migration()?;
+        self.device_manager.lock().unwrap().complete_migration()
+    }
 }
 
 #[cfg(all(feature = "kvm", target_arch = "x86_64"))]