feat(blockifier): add support for sha256_process_block syscall

starkware-libs · Nov 11, 2024 · 097ae6a · 097ae6a
1 parent 4276a78
commit 097ae6a
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 6 deletions.
diff --git a/crates/blockifier/cairo_native b/crates/blockifier/cairo_native
diff --git a/crates/blockifier/src/execution/native/syscall_handler.rs b/crates/blockifier/src/execution/native/syscall_handler.rs
@@ -367,10 +367,25 @@ impl<'state> StarknetSyscallHandler for &mut NativeSyscallHandler<'state> {
 
     fn sha256_process_block(
         &mut self,
-        _prev_state: &mut [u32; 8],
-        _current_block: &[u32; 16],
-        _remaining_gas: &mut u128,
+        prev_state: &mut [u32; 8],
+        current_block: &[u32; 16],
+        remaining_gas: &mut u128,
     ) -> SyscallResult<()> {
-        todo!("Implement sha256_process_block syscall.");
+        self.pre_execute_syscall(
+            remaining_gas,
+            SyscallSelector::Sha256ProcessBlock,
+            self.context.gas_costs().sha256_process_block_gas_cost,
+        )?;
+
+        let data_as_bytes = sha2::digest::generic_array::GenericArray::from_exact_iter(
+            current_block.iter().flat_map(|x| x.to_be_bytes()),
+        )
+        .expect(
+            "u32.to_be_bytes() returns 4 bytes, and data.len() == 16. So data contains 64 bytes.",
+        );
+
+        sha2::compress256(prev_state, &[data_as_bytes]);
+
+        Ok(())
     }
 }
diff --git a/crates/blockifier/src/execution/syscalls/syscall_tests/sha256.rs b/crates/blockifier/src/execution/syscalls/syscall_tests/sha256.rs
@@ -10,6 +10,10 @@ use crate::test_utils::contracts::FeatureContract;
 use crate::test_utils::initial_test_state::test_state;
 use crate::test_utils::{trivial_external_entry_point_new, CairoVersion, BALANCE};
 
+#[cfg_attr(
+    feature = "cairo_native",
+    test_case(FeatureContract::TestContract(CairoVersion::Native), 891625; "Native")
+)]
 #[test_case(FeatureContract::TestContract(CairoVersion::Cairo1), 881425; "VM")]
 fn test_sha256(test_contract: FeatureContract, gas_consumed: u64) {
     let chain_info = &ChainInfo::create_for_testing();
@@ -22,7 +26,7 @@ fn test_sha256(test_contract: FeatureContract, gas_consumed: u64) {
         ..trivial_external_entry_point_new(test_contract)
     };
 
-    assert_eq!(
+    pretty_assertions::assert_eq!(
         entry_point_call.execute_directly(&mut state).unwrap().execution,
         CallExecution { gas_consumed, ..CallExecution::from_retdata(retdata![]) }
     );
+0 −36		.github/actions/install-linux-deps/action.yml
+0 −95		.github/workflows/daily.yml
+137 −169		Cargo.lock
+18 −18		Cargo.toml
+1 −1		Makefile
+1 −11		docs/sierra.md
+0 −8		examples/erc20.rs
+0 −8		examples/starknet.rs
+2 −2		runtime/Cargo.toml
+0 −48		scripts/cmp_state_dumps.sh
+7 −11		src/bin/cairo-native-test.rs
+0 −1		src/bin/utils/mod.rs
+0 −6		src/debug.rs
+0 −2		src/executor.rs
+0 −4		src/libfuncs.rs
+2 −9		src/libfuncs/felt252.rs
+0 −181		src/libfuncs/int_range.rs
+0 −190		src/libfuncs/starknet.rs
+2 −43		src/starknet.rs
+0 −8		src/starknet_stub.rs
+1 −22		src/types.rs
+0 −46		src/types/int_range.rs
+0 −63		src/values.rs
+7 −19		tests/common.rs
+1 −6		tests/tests/starknet/programs/syscalls.cairo
+0 −8		tests/tests/starknet/secp256.rs
+0 −28		tests/tests/starknet/syscalls.rs