Skip to content

Commit

Permalink
feat(blockifier): add support for sha256_process_block syscall
Browse files Browse the repository at this point in the history
  • Loading branch information
varex83 committed Oct 31, 2024
1 parent 55834b2 commit baab125
Show file tree
Hide file tree
Showing 2 changed files with 27 additions and 6 deletions.
27 changes: 22 additions & 5 deletions crates/blockifier/src/execution/native/syscall_handler.rs
Original file line number Diff line number Diff line change
Expand Up @@ -313,10 +313,27 @@ 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,
) -> SyscallResult<()> {
todo!("Implement sha256_process_block syscall.");
prev_state: &[u32; 8],
current_block: &[u32; 16],
remaining_gas: &mut u128,
) -> SyscallResult<[u32; 8]> {
self.substract_syscall_gas_cost(
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.",
);

let mut state = *prev_state;

sha2::compress256(&mut state, &[data_as_bytes]);

Ok(state)
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,10 @@ use crate::test_utils::initial_test_state::test_state;
use crate::test_utils::{trivial_external_entry_point_new, CairoVersion, BALANCE};

#[test_case(FeatureContract::TestContract(CairoVersion::Cairo1), 881625; "VM")]
#[cfg_attr(
feature = "cairo_native",
test_case(FeatureContract::TestContract(CairoVersion::Native), 891625; "Native")
)]
fn test_sha256(test_contract: FeatureContract, gas_consumed: u64) {
let chain_info = &ChainInfo::create_for_testing();
let mut state = test_state(chain_info, BALANCE, &[(test_contract, 1)]);
Expand All @@ -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![]) }
);
Expand Down

0 comments on commit baab125

Please sign in to comment.