fixup! Track dirty pages from host side

Author: ludfjig

src/hyperlight_host/src/mem/shared_mem.rs

@@ -420,6 +420,8 @@ impl ExclusiveSharedMemory {
     #[cfg(target_os = "windows")]
     #[instrument(skip_all, parent = Span::current(), level= "Trace")]
     pub fn new(min_size_bytes: usize) -> Result<Self> {
+        use super::bitmap::new_page_bitmap;
+
         if min_size_bytes == 0 {
             return Err(new_error!("Cannot create shared memory with size 0"));
         }
@@ -525,13 +527,11 @@ impl ExclusiveSharedMemory {
             handle,
         });
 
-        let dirty_page_tracker = Arc::new(Mutex::new(Some(DirtyPageTracker::new(Arc::clone(
-            &host_mapping,
-        ))?)));
+        let dirty_page_tracker = new_page_bitmap(min_size_bytes, false)?;
 
         Ok(Self {
             region: host_mapping,
-            signal_dirty_bitmap_tracker: dirty_page_tracker,
+            dirty_page_tracker: Arc::new(Mutex::new(dirty_page_tracker)),
         })
     }