feat(blockifier): add syscall counting logic for native

starkware-libs · Oct 29, 2024 · 147c1d3 · 147c1d3
1 parent a3166cf
commit 147c1d3
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/crates/blockifier/src/execution/native/entry_point_execution.rs b/crates/blockifier/src/execution/native/entry_point_execution.rs
@@ -61,6 +61,12 @@ fn create_callinfo(
 
     let gas_consumed = syscall_handler.call.initial_gas - remaining_gas;
 
+    // todo(rodrigo): execution resources for native execution are still wip until future
+    // development on both Cairo lang and the Native compiler
+    let versioned_constants = syscall_handler.context.versioned_constants();
+    *syscall_handler.resources +=
+        &versioned_constants.get_additional_os_syscall_resources(&syscall_handler.syscall_counter);
+
     Ok(CallInfo {
         call: syscall_handler.call,
         execution: CallExecution {
@@ -70,9 +76,7 @@ fn create_callinfo(
             failed: call_result.failure_flag,
             gas_consumed,
         },
-        // todo(rodrigo): execution resources rely heavily on how the VM work, therefore
-        // the dummy values
-        resources: ExecutionResources::default(),
+        resources: syscall_handler.resources.clone(),
         inner_calls: syscall_handler.inner_calls,
         storage_read_values: syscall_handler.read_values,
         accessed_storage_keys: syscall_handler.accessed_keys,

diff --git a/crates/blockifier/src/execution/native/syscall_handler.rs b/crates/blockifier/src/execution/native/syscall_handler.rs
@@ -17,7 +17,8 @@ use starknet_types_core::felt::Felt;
 use crate::execution::call_info::{CallInfo, OrderedEvent, OrderedL2ToL1Message, Retdata};
 use crate::execution::entry_point::{CallEntryPoint, EntryPointExecutionContext};
 use crate::execution::native::utils::encode_str_as_felts;
-use crate::execution::syscalls::hint_processor::OUT_OF_GAS_ERROR;
+use crate::execution::syscalls::hint_processor::{SyscallCounter, OUT_OF_GAS_ERROR};
+use crate::execution::syscalls::SyscallSelector;
 use crate::state::state_api::State;
 
 pub struct NativeSyscallHandler<'state> {
@@ -57,6 +58,12 @@ impl<'state> NativeSyscallHandler<'state> {
         }
     }
 
+    #[allow(dead_code)]
+    fn increment_syscall_count_by(&mut self, selector: &SyscallSelector, n: usize) {
+        let syscall_count = self.syscall_counter.entry(*selector).or_default();
+        *syscall_count += n
+    }
+
     #[allow(dead_code)]
     fn execute_inner_call(
         &mut self,
@@ -91,8 +98,14 @@ impl<'state> NativeSyscallHandler<'state> {
     fn substract_syscall_gas_cost(
         &mut self,
         remaining_gas: &mut u128,
+        syscall_selector: SyscallSelector,
         syscall_gas_cost: u64,
     ) -> SyscallResult<()> {
+        // Syscall count for Keccak is done differently
+        if syscall_selector != SyscallSelector::Keccak {
+            self.increment_syscall_count_by(&syscall_selector, 1);
+        }
+
         // Refund `SYSCALL_BASE_GAS_COST` as it was pre-charged.
         let required_gas =
             u128::from(syscall_gas_cost - self.context.gas_costs().syscall_base_gas_cost);