From 9e18177654aa5a2f3bd2947cbd23eb3ca8dbec2d Mon Sep 17 00:00:00 2001
From: Sebastien Boeuf <sebastien.boeuf@intel.com>
Date: Thu, 26 Mar 2020 14:53:43 +0100
Subject: [PATCH] vmm: Add memory hotplug support to VFIO PCI devices

Extend the update_memory() method from DeviceManager so that VFIO PCI
devices can update their DMA mappings to the physical IOMMU, after a
memory hotplug has been performed.

Signed-off-by: Sebastien Boeuf <sebastien.boeuf@intel.com>
---
 vmm/src/device_manager.rs | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/vmm/src/device_manager.rs b/vmm/src/device_manager.rs
index c790cf928..66b888f75 100644
--- a/vmm/src/device_manager.rs
+++ b/vmm/src/device_manager.rs
@@ -270,6 +270,10 @@ pub enum DeviceManagerError {
 
     /// Cannot find a memory range for virtio-mem memory
     VirtioMemRangeAllocation,
+
+    /// Failed updating guest memory for VFIO PCI device.
+    #[cfg(feature = "pci_support")]
+    UpdateMemoryForVfioPciDevice(VfioPciError),
 }
 pub type DeviceManagerResult<T> = result::Result<T, DeviceManagerError>;
 
@@ -1886,6 +1890,22 @@ impl DeviceManager {
                 .map_err(DeviceManagerError::UpdateMemoryForVirtioDevice)?;
         }
 
+        // Take care of updating the memory for VFIO PCI devices.
+        #[cfg(feature = "pci_support")]
+        {
+            for (_, any_device) in self.pci_devices.iter() {
+                if let Ok(vfio_pci_device) =
+                    Arc::clone(any_device).downcast::<Mutex<VfioPciDevice>>()
+                {
+                    vfio_pci_device
+                        .lock()
+                        .unwrap()
+                        .update_memory(_new_region)
+                        .map_err(DeviceManagerError::UpdateMemoryForVfioPciDevice)?;
+                }
+            }
+        }
+
         Ok(())
     }