Windows fix

Signed-off-by: Simon Davies <[email protected]>

Author: simongdavies

src/hyperlight_host/src/hypervisor/hyperv_windows.rs

@@ -610,24 +610,7 @@ impl Hypervisor for HypervWindowsDriver {
     fn get_and_clear_dirty_pages(&mut self) -> Result<Vec<u64>> {
         // For now we just mark all pages dirty which is the equivalent of taking a full snapshot
         let total_size = self.mem_regions.iter().map(|r| r.guest_region.len()).sum();
-        let bitmap = new_page_bitmap(total_size, true)?;
-
-        // Count the number of set bits in the bitmap
-        let set_bit_count = crate::mem::bitmap::bit_index_iterator(&bitmap).count();
-        let expected_count = total_size.div_ceil(hyperlight_common::mem::PAGE_SIZE_USIZE);
-
-        if set_bit_count * hyperlight_common::mem::PAGE_SIZE_USIZE != total_size {
-            println!(
-                "Dirty page bitmap size mismatch: {} set bits * {} page size = {} bytes, but expected {} bytes and {} set bits",
-                set_bit_count,
-                hyperlight_common::mem::PAGE_SIZE_USIZE,
-                set_bit_count * hyperlight_common::mem::PAGE_SIZE_USIZE,
-                total_size,
-                expected_count
-            );
-        }
-
-        Ok(bitmap)
+        new_page_bitmap(total_size, true)
     }
 
     #[instrument(err(Debug), skip_all, parent = Span::current(), level = "Trace")]

src/hyperlight_host/src/mem/windows_dirty_page_tracker.rs

@@ -62,6 +62,6 @@ impl DirtyPageTracking for WindowsDirtyPageTracker {
     /// until we implement actual dirty page tracking
     fn get_dirty_pages(self) -> Vec<usize> {
         // Return all page indices from 0 to num_pages-1
-        (0..self.num_pages).collect()
+        (0..self.num_pages - 2).collect() // exclude the guard pages
     }
 }