throw error

zeta-chain · Nov 22, 2024 · e111f44 · e111f44
1 parent a279ac1
commit e111f44
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/examples/swap/contracts/SwapToAnyToken.sol b/examples/swap/contracts/SwapToAnyToken.sol
@@ -92,12 +92,21 @@ contract SwapToAnyToken is UniversalContract {
         bytes memory recipient,
         bool withdrawFlag
     ) public {
-        IZRC20(inputToken).transferFrom(msg.sender, address(this), amount);
+        bool success = IZRC20(inputToken).transferFrom(
+            msg.sender,
+            address(this),
+            amount
+        );
+        if (!success) {
+            revert TransferFailed();
+        }
+
         (uint256 out, address gasZRC20, uint256 gasFee) = handleGasAndSwap(
             inputToken,
             amount,
             targetToken
         );
+
         withdraw(
             Params({
                 target: targetToken,