Backport nervosnetwork#360

libraries · Jul 14, 2023 · 7efa8f9 · 7efa8f9
1 parent 1c18cbe
commit 7efa8f9
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/snapshot.rs b/src/snapshot.rs
@@ -1,10 +1,7 @@
 use crate::instructions::Register;
 use crate::memory::Memory;
 use crate::memory::FLAG_DIRTY;
-use crate::{
-    CoreMachine, Error, RISCV_GENERAL_REGISTER_NUMBER, RISCV_PAGES, RISCV_PAGESIZE,
-    RISCV_PAGE_SHIFTS,
-};
+use crate::{CoreMachine, Error, RISCV_GENERAL_REGISTER_NUMBER, RISCV_PAGESIZE, RISCV_PAGE_SHIFTS};
 use serde::{Deserialize, Serialize};
 
 // Snapshot provides a mechanism for suspending and resuming a virtual machine.
@@ -47,7 +44,7 @@ pub fn make_snapshot<T: CoreMachine>(machine: &mut T) -> Result<Snapshot, Error>
         snap.registers[i] = v.to_u64();
     }
 
-    for i in 0..RISCV_PAGES {
+    for i in 0..(machine.memory().memory_size() >> RISCV_PAGE_SHIFTS) {
         let flag = machine.memory_mut().fetch_flag(i as u64)?;
         if flag & FLAG_DIRTY != 0 {
             let addr_from = i << RISCV_PAGE_SHIFTS;