expose cycles statistics

matter-labs · Aug 21, 2024 · a65bbf7 · a65bbf7
1 parent 085a91e
commit a65bbf7
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 4 deletions.
diff --git a/src/instruction_handlers/precompiles.rs b/src/instruction_handlers/precompiles.rs
@@ -66,16 +66,18 @@ fn precompile_call<T: Tracer>(
         let heaps = &mut vm.state.heaps;
         match address_low {
             KECCAK256_ROUND_FUNCTION_PRECOMPILE_ADDRESS => {
-                keccak256_rounds_function::<_, false>(0, query, heaps);
+                vm.state.keccak256_cycles +=
+                    keccak256_rounds_function::<_, false>(0, query, heaps).0;
             }
             SHA256_ROUND_FUNCTION_PRECOMPILE_ADDRESS => {
-                sha256_rounds_function::<_, false>(0, query, heaps);
+                vm.state.sha256_cycles += sha256_rounds_function::<_, false>(0, query, heaps).0;
             }
             ECRECOVER_INNER_FUNCTION_PRECOMPILE_ADDRESS => {
-                ecrecover_function::<_, false>(0, query, heaps);
+                vm.state.ecrecover_cycles += ecrecover_function::<_, false>(0, query, heaps).0;
             }
             SECP256R1_VERIFY_PRECOMPILE_ADDRESS => {
-                secp256r1_verify_function::<_, false>(0, query, heaps);
+                vm.state.secp256v1_verify_cycles +=
+                    secp256r1_verify_function::<_, false>(0, query, heaps).0;
             }
             _ => {
                 // A precompile call may be used just to burn gas

diff --git a/src/single_instruction_test/vm.rs b/src/single_instruction_test/vm.rs
@@ -91,6 +91,10 @@ impl<'a, T: Tracer> Arbitrary<'a> for VirtualMachine<T> {
                 heaps,
                 transaction_number: u.arbitrary()?,
                 context_u128: u.arbitrary()?,
+                keccak256_cycles: 0,
+                ecrecover_cycles: 0,
+                sha256_cycles: 0,
+                secp256v1_verify_cycles: 0,
             },
             settings: u.arbitrary()?,
             world_diff: Default::default(),

diff --git a/src/state.rs b/src/state.rs
@@ -27,6 +27,14 @@ pub struct State<T> {
 
     pub transaction_number: u16,
 
+    pub keccak256_cycles: usize,
+
+    pub ecrecover_cycles: usize,
+
+    pub sha256_cycles: usize,
+
+    pub secp256v1_verify_cycles: usize,
+
     pub(crate) context_u128: u128,
 }
 
@@ -75,6 +83,10 @@ impl<T> State<T> {
 
             transaction_number: 0,
             context_u128: 0,
+            keccak256_cycles: 0,
+            ecrecover_cycles: 0,
+            sha256_cycles: 0,
+            secp256v1_verify_cycles: 0,
         }
     }