Fix CALLDATACOPY non-compliance

lambdaclass · Nov 25, 2024 · 43093d5 · 43093d5
1 parent 08f817d
commit 43093d5
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/crates/vm/levm/src/opcode_handlers/environment.rs b/crates/vm/levm/src/opcode_handlers/environment.rs
@@ -167,7 +167,7 @@ impl VM {
             return Ok(OpcodeSuccess::Continue);
         }
 
-        let mut data = [0u8; 32];
+        let mut data = vec![0u8; size];
         for (i, byte) in current_call_frame
             .calldata
             .iter()

diff --git a/crates/vm/levm/tests/edge_case_tests.rs b/crates/vm/levm/tests/edge_case_tests.rs
@@ -101,3 +101,15 @@ fn test_is_negative() {
     let mut current_call_frame = vm.call_frames.pop().unwrap();
     vm.execute(&mut current_call_frame);
 }
+
+#[test]
+fn test_non_compliance_calldatacopy_memory_resize() {
+    let mut vm =
+        new_vm_with_bytecode(Bytes::copy_from_slice(&[0x60, 34, 0x5f, 0x5f, 55, 89])).unwrap();
+    let mut current_call_frame = vm.call_frames.pop().unwrap();
+    vm.execute(&mut current_call_frame);
+    assert_eq!(
+        *current_call_frame.stack.stack.first().unwrap(),
+        U256::from(64)
+    );
+}