feat: calldatasize

kkrt-labs · Jan 21, 2025 · 38947a4 · 38947a4
1 parent 3f90598
commit 38947a4
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 0 deletions.
diff --git a/cairo/ethereum/cancun/vm/instructions/environment.cairo b/cairo/ethereum/cancun/vm/instructions/environment.cairo
@@ -896,3 +896,30 @@ func calldatacopy{range_check_ptr, evm: Evm}() -> ExceptionalHalt* {
     let ok = cast(0, ExceptionalHalt*);
     return ok;
 }
+
+func calldatasize{range_check_ptr, evm: Evm}() -> ExceptionalHalt* {
+    alloc_locals;
+    // STACK
+    // No stack input
+    let stack = evm.value.stack;
+    // GAS
+    let err = charge_gas(Uint(GasConstants.GAS_BASE));
+    if (cast(err, felt) != 0) {
+        return err;
+    }
+
+    // OPERATION
+    let calldata_len = evm.value.message.value.data.value.len;
+    with stack {
+        let err = push(U256(new U256Struct(calldata_len, 0)));
+        if (cast(err, felt) != 0) {
+            return err;
+        }
+    }
+
+    // PROGRAM COUNTER
+    EvmImpl.set_pc_stack(Uint(evm.value.pc.value + 1), stack);
+
+    let ok = cast(0, ExceptionalHalt*);
+    return ok;
+}
diff --git a/cairo/tests/ethereum/cancun/vm/instructions/test_environment.py b/cairo/tests/ethereum/cancun/vm/instructions/test_environment.py
@@ -12,6 +12,7 @@
     blob_hash,
     calldatacopy,
     calldataload,
+    calldatasize,
     caller,
     callvalue,
     codecopy,
@@ -384,3 +385,17 @@ def test_calldatacopy(self, cairo_run, evm: Evm):
 
         calldatacopy(evm)
         assert evm == cairo_result
+
+    @given(
+        evm=EvmBuilder().with_stack().with_gas_left().with_message_calldata().build()
+    )
+    def test_calldatasize(self, cairo_run, evm: Evm):
+        try:
+            cairo_result = cairo_run("calldatasize", evm)
+        except ExceptionalHalt as cairo_error:
+            with strict_raises(type(cairo_error)):
+                calldatasize(evm)
+            return
+
+        calldatasize(evm)
+        assert evm == cairo_result