end_addr max value check

starkware-libs · Jan 9, 2025 · 7f52102 · 7f52102
1 parent d921ce9
commit 7f52102
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/stwo_cairo_prover/crates/prover/src/input/vm_import/mod.rs b/stwo_cairo_prover/crates/prover/src/input/vm_import/mod.rs
@@ -15,6 +15,7 @@ use super::builtin_segments::BuiltinSegments;
 use super::memory::MemoryConfig;
 use super::state_transitions::StateTransitions;
 use super::ProverInput;
+use crate::components::memory::MEMORY_ADDRESS_BOUND;
 use crate::input::memory::MemoryBuilder;
 
 #[derive(Debug, Error)]
@@ -46,7 +47,7 @@ pub fn adapt_vm_output(
         .map(|v| v.stop_ptr)
         .max()
         .ok_or(VmImportError::NoMemorySegments)?;
-    assert!(end_addr < 1 << (usize::BITS - 1));
+    assert!(end_addr < MEMORY_ADDRESS_BOUND);
 
     let memory_path = private_input_json
         .parent()