fixup! Track dirty pages from host side

Author: ludfjig

src/hyperlight_host/src/mem/shared_mem.rs

@@ -401,13 +401,13 @@ impl ExclusiveSharedMemory {
     }
 
     /// Gets the dirty bitmap and then clears it in self.
-    pub(crate) fn get_and_clear_dirty_pages(&mut self) -> Result<Vec<u64>> {
+    pub(crate) fn get_and_clear_dirty_pages(&self) -> Result<Vec<u64>> {
         let mut guard = self
             .dirty_page_tracker
             .try_lock()
             .map_err(|_| new_error!("Failed to acquire lock on dirty page tracker"))?;
-        let bitmap = guard.clone();
-        guard.fill(0);
+        let mut bitmap = vec![0; guard.len()];
+        core::mem::swap(&mut bitmap, &mut *guard);
         Ok(bitmap)
     }
 
@@ -744,7 +744,7 @@ impl ExclusiveSharedMemory {
     }
 
     /// Marks pages that cover bytes [offset, offset + size) as dirty
-    pub(super) fn mark_pages_dirty(&mut self, offset: usize, size: usize) -> Result<()> {
+    pub(super) fn mark_pages_dirty(&self, offset: usize, size: usize) -> Result<()> {
         bounds_check!(offset, size, self.mem_size());
         let mut bitmap = self
             .dirty_page_tracker
@@ -1080,7 +1080,7 @@ impl HostSharedMemory {
     }
 
     /// Marks pages that cover bytes [offset, offset + size) as dirty
-    pub(super) fn mark_pages_dirty(&mut self, offset: usize, size: usize) -> Result<()> {
+    pub(super) fn mark_pages_dirty(&self, offset: usize, size: usize) -> Result<()> {
         bounds_check!(offset, size, self.mem_size());
         let mut bitmap = self
             .dirty_page_tracker