throw error if function doesn't exist when making arbitrary call

zeta-chain · Nov 12, 2024 · e47cc0e · e47cc0e
1 parent 0b47bea
commit e47cc0e
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/packages/localnet/src/handleOnZEVMCalled.ts b/packages/localnet/src/handleOnZEVMCalled.ts
@@ -43,6 +43,16 @@ export const handleOnZEVMCalled = async ({
     };
     log(chainID, `Calling ${receiver} with message ${message}`);
 
+    if (isArbitraryCall) {
+      const selector = message.slice(0, 10);
+      const code = await provider.getCode(receiver);
+      if (!code.includes(selector.slice(2))) {
+        throw new Error(
+          `Receiver contract does not contain function with selector ${selector}`
+        );
+      }
+    }
+
     const executeTx = await evmContracts[chainID].gatewayEVM
       .connect(tss)
       .execute(messageContext, receiver, message, deployOpts);

diff --git a/packages/localnet/src/handleOnZEVMWithdrawnAndCalled.ts b/packages/localnet/src/handleOnZEVMWithdrawnAndCalled.ts
@@ -56,7 +56,17 @@ export const handleOnZEVMWithdrawnAndCalled = async ({
     const zrc20Contract = new ethers.Contract(zrc20, ZRC20.abi, deployer);
     const coinType = await zrc20Contract.COIN_TYPE();
     const isGasToken = coinType === 1n;
-    const isERC20orZETA = coinType === 2n;
+
+    if (isArbitraryCall) {
+      const selector = message.slice(0, 10);
+      const code = await provider.getCode(receiver);
+      if (!code.includes(selector.slice(2))) {
+        throw new Error(
+          `Receiver contract does not contain function with selector ${selector}`
+        );
+      }
+    }
+
     log(chainID, `Calling ${receiver} with message ${message}`);
     if (isGasToken) {
       const executeTx = await evmContracts[chainID].gatewayEVM