Fix hook 7 for benchmarks

lambdaclass · Aug 20, 2024 · b09ee3a · b09ee3a
1 parent 44337c5
commit b09ee3a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/core/lib/multivm/src/versions/era_vm/hook.rs b/core/lib/multivm/src/versions/era_vm/hook.rs
@@ -1,15 +1,18 @@
 #[derive(Debug)]
 pub(crate) enum Hook {
     AccountValidationEntered,
+    PaymasterValidationEntered,
     AccountValidationExited,
     ValidationStepEnded,
     TxHasEnded,
     DebugLog,
     DebugReturnData,
+    NearCallCatch,
     AskOperatorForRefund,
     NotifyAboutRefund,
     PostResult,
     FinalBatchInfo,
+    PubdataRequested,
 }
 
 impl Hook {
@@ -18,15 +21,18 @@ impl Hook {
     pub fn from_u32(hook: u32) -> Self {
         match hook {
             0 => Hook::AccountValidationEntered,
+            1 => Hook::PaymasterValidationEntered,
             2 => Hook::AccountValidationExited,
             3 => Hook::ValidationStepEnded,
             4 => Hook::TxHasEnded,
             5 => Hook::DebugLog,
             6 => Hook::DebugReturnData,
+            7 => Hook::NearCallCatch,
             8 => Hook::AskOperatorForRefund,
             9 => Hook::NotifyAboutRefund,
             10 => Hook::PostResult,
             11 => Hook::FinalBatchInfo,
+            12 => Hook::PubdataRequested,
             _ => panic!("Unknown hook {}", hook),
         }
     }

diff --git a/core/lib/multivm/src/versions/era_vm/vm.rs b/core/lib/multivm/src/versions/era_vm/vm.rs
@@ -186,6 +186,9 @@ impl<S: ReadStorage + 'static> Vm<S> {
                 Hook::AccountValidationEntered => {
                     // println!("ACCOUNT VALIDATION ENTERED");
                 }
+                Hook::NearCallCatch => {}
+                Hook::PaymasterValidationEntered => {}
+                Hook::PubdataRequested => {}
                 Hook::ValidationStepEnded => {
                     // println!("VALIDATION STEP ENDED");
                 }