add type assertion to onboarding Vault

onflow · Sep 4, 2024 · fa5f613 · fa5f613
1 parent ca95d45
commit fa5f613
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cadence/contracts/bridge/FlowEVMBridge.cdc b/cadence/contracts/bridge/FlowEVMBridge.cdc
@@ -99,8 +99,13 @@ contract FlowEVMBridge : IFlowEVMNFTBridge, IFlowEVMTokenBridge {
         } else if type.isSubtype(of: Type<@{FungibleToken.Vault}>()) {
             let createVaultFunction = FlowEVMBridgeUtils.getCreateEmptyVaultFunction(forType: type)
                 ?? panic("Could not retrieve createEmptyVault function for the given type")
+            let vault <-createVaultFunction(type)
+            assert(
+                vault.getType() == type,
+                message: "Requested to onboard type=".concat(type.identifier).concat( "but contract returned type=").concat(vault.getType().identifier)
+            )
             FlowEVMBridgeTokenEscrow.initializeEscrow(
-                with: <-createVaultFunction(type),
+                with: <-vault,
                 name: onboardingValues.name,
                 symbol: onboardingValues.symbol,
                 decimals: onboardingValues.decimals!,