feat(vmm): add offset/gpa conversion functions

This is because vCPUs reason in GPAs while the secret-free UFFD
protocol is guest_memfd-offset-based.

TODO: add cab/sob from Patrick

Signed-off-by: Nikita Kalyazin <[email protected]>

Author: kalyazin

src/vmm/src/vstate/memory.rs

@@ -323,6 +323,12 @@ where
 
     /// Store the dirty bitmap in internal store
     fn store_dirty_bitmap(&self, dirty_bitmap: &DirtyBitmap, page_size: usize);
+
+    /// Convert guest physical address to file offset
+    fn gpa_to_offset(&self, gpa: GuestAddress) -> Option<u64>;
+
+    /// Convert file offset to guest physical address
+    fn offset_to_gpa(&self, offset: u64) -> Option<GuestAddress>;
 }
 
 /// State of a guest memory region saved to file/buffer.
@@ -473,6 +479,32 @@ impl GuestMemoryExtension for GuestMemoryMmap {
             }
         });
     }
+
+    /// Convert guest physical address to file offset
+    fn gpa_to_offset(&self, gpa: GuestAddress) -> Option<u64> {
+        self.find_region(gpa)
+            .map(|r| gpa.0 - r.start_addr().0 + r.file_offset().unwrap().start())
+    }
+
+    /// Convert file offset to guest physical address
+    fn offset_to_gpa(&self, offset: u64) -> Option<GuestAddress> {
+        self.iter().find_map(|region| {
+            if let Some(reg_offset) = region.file_offset() {
+                let region_start = reg_offset.start();
+                let region_size = region.size();
+
+                if offset >= region_start && offset < region_start + region_size as u64 {
+                    Some(GuestAddress(
+                        region.start_addr().0 + (offset - region_start),
+                    ))
+                } else {
+                    None
+                }
+            } else {
+                None
+            }
+        })
+    }
 }
 
 /// Creates a memfd of the given size and huge pages configuration