add rentrancy guard

karnotxyz · Jul 29, 2024 · fb4be14 · fb4be14
1 parent 1015603
commit fb4be14
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/bridge/token_bridge.cairo b/src/bridge/token_bridge.cairo
@@ -515,6 +515,7 @@ pub mod TokenBridge {
             appchain_recipient: ContractAddress,
             message: Span<felt252>
         ) {
+            self.reentrancy_guard.start();
             self.accept_deposit(token, amount);
             let nonce = self
                 .send_deposit_message(
@@ -534,6 +535,7 @@ pub mod TokenBridge {
                 );
             // Piggy-back the deposit tx to check and update the status of token bridge deployment.
             self.check_deployment_status(token);
+            self.reentrancy_guard.end();
         }
 
         //