extra checks

multiversx · Apr 1, 2024 · d720e83 · d720e83
1 parent c792082
commit d720e83
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/locked-asset/proxy_dex/src/lib.rs b/locked-asset/proxy_dex/src/lib.rs
@@ -60,6 +60,12 @@ pub trait ProxyDexImpl:
 
     #[endpoint]
     fn upgrade(&self, old_locked_token_id: TokenIdentifier, old_factory_address: ManagedAddress) {
+        require!(
+            old_locked_token_id.is_valid_esdt_identifier(),
+            "Invalid token id"
+        );
+        self.require_sc_address(&old_factory_address);
+
         self.old_locked_token_id()
             .set_if_empty(&old_locked_token_id);
         self.old_factory_address()