Fix something

openvm-org · Jan 16, 2025 · 5070149 · 5070149
1 parent 928ccf4
commit 5070149
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/crates/vm/src/system/memory/offline.rs b/crates/vm/src/system/memory/offline.rs
@@ -301,6 +301,11 @@ impl<F: PrimeField32> OfflineMemory<F> {
         query: u32,
         records: &mut Vec<AccessAdapterRecord<F>>,
     ) {
+        let lim = (self.data[(address_space - self.as_offset) as usize].memory_size()) as u32;
+        if query == lim {
+            return;
+        }
+        assert!(query < lim);
         let original_block = self.block_containing(address_space, query);
         if original_block.pointer == query {
             return;

diff --git a/crates/vm/src/system/memory/paged_vec.rs b/crates/vm/src/system/memory/paged_vec.rs
@@ -86,6 +86,10 @@ impl<T: Default + Clone, const PAGE_SIZE: usize> PagedVec<T, PAGE_SIZE> {
             result
         }
     }
+
+    pub fn memory_size(&self) -> usize {
+        self.pages.len() * PAGE_SIZE
+    }
 }
 
 impl<T, const PAGE_SIZE: usize> PagedVec<T, PAGE_SIZE> {
-Original file line number
+Diff line change
@@ Expand Up @@
                 result
             }
         }
+        pub fn memory_size(&self) -> usize {
+            self.pages.len() * PAGE_SIZE
+        }
     }
     impl<T, const PAGE_SIZE: usize> PagedVec<T, PAGE_SIZE> {
@@ Expand Down @@