Properly restoring contract addresses on revert. (#3534)

starkware-libs · Jun 28, 2023 · f814b2b · f814b2b
1 parent 1003d5d
commit f814b2b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/crates/cairo-lang-runner/src/casm_run/mod.rs b/crates/cairo-lang-runner/src/casm_run/mod.rs
@@ -613,6 +613,9 @@ impl HintProcessor for CairoHintProcessor<'_> {
                         // Call the function.
                         let Some(entry_point) = contract_info.externals.get(&selector) else
                         {
+                            // Restore the contract addresses in the context.
+                            self.starknet_state.exec_info.caller_address = old_caller_address;
+                            self.starknet_state.exec_info.contract_address = old_contract_address;
                             fail_syscall!(b"ENTRYPOINT_NOT_FOUND");
                         };
                         let function = runner
@@ -636,6 +639,10 @@ impl HintProcessor for CairoHintProcessor<'_> {
                                 read_array_result_as_vec(&res.memory, &value)
                             }
                             RunResultValue::Panic(mut panic_data) => {
+                                // Restore the contract addresses in the context.
+                                self.starknet_state.exec_info.caller_address = old_caller_address;
+                                self.starknet_state.exec_info.contract_address =
+                                    old_contract_address;
                                 fail_syscall!(panic_data, b"ENTRYPOINT_FAILED");
                             }
                         };