pci: vfio: Release memory slots upon unmap

This prevents starvation of the limited set of memory slots in the kernel. Signed-off-by: Rob Bradford <[email protected]>
UiPath · Nov 24, 2024 · 0d6cef4 · 0d6cef4
1 parent 81f8a27
commit 0d6cef4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/pci/src/vfio.rs b/pci/src/vfio.rs
@@ -1699,6 +1699,9 @@ impl VfioPciDevice {
                     error!("Could not remove the userspace memory region: {}", e);
                 }
 
+                self.memory_slot_allocator
+                    .free_memory_slot(user_memory_region.slot);
+
                 // SAFETY: FFI call with correct arguments
                 let ret = unsafe {
                     libc::munmap(

diff --git a/pci/src/vfio_user.rs b/pci/src/vfio_user.rs
@@ -222,6 +222,9 @@ impl VfioUserPciDevice {
                     error!("Could not remove the userspace memory region: {}", e);
                 }
 
+                self.memory_slot_allocator
+                    .free_memory_slot(user_memory_region.slot);
+
                 // Remove mmaps
                 // SAFETY: FFI call with correct arguments
                 let ret = unsafe {