refactor: handle uninitialized error in the gateway

axelarnetwork · Mar 7, 2024 · c6fc62a · c6fc62a
1 parent bba0824
commit c6fc62a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/contracts/axelar-gateway/src/contract.rs b/contracts/axelar-gateway/src/contract.rs
@@ -111,7 +111,7 @@ impl AxelarGatewayInterface for AxelarGateway {
 
         let auth_module = AxelarAuthVerifierClient::new(
             &env,
-            &env.storage().instance().get(&DataKey::AuthModule).unwrap(),
+            &env.storage().instance().get(&DataKey::AuthModule).ok_or(Error::Uninitialized)?,
         );
 
         let valid = auth_module.validate_proof(&batch_hash, &proof);

diff --git a/contracts/axelar-gateway/src/error.rs b/contracts/axelar-gateway/src/error.rs
@@ -8,4 +8,5 @@ pub enum Error {
     InvalidChainId = 2,
     CommandAlreadyExecuted = 3,
     InvalidProof = 4,
+    Uninitialized = 5,
 }