From b0c254add44e56d10f203eae30301c861fd9d8ff Mon Sep 17 00:00:00 2001 From: Diana Carvalho Date: Wed, 14 May 2025 09:02:55 +0100 Subject: [PATCH] fix: Revert if the TransferType is not valid Took 8 minutes --- foundry/src/executors/TokenTransfer.sol | 3 +++ 1 file changed, 3 insertions(+) diff --git a/foundry/src/executors/TokenTransfer.sol b/foundry/src/executors/TokenTransfer.sol index 05e296a..6eac1c0 100644 --- a/foundry/src/executors/TokenTransfer.sol +++ b/foundry/src/executors/TokenTransfer.sol @@ -6,6 +6,7 @@ import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol"; import "@permit2/src/interfaces/IAllowanceTransfer.sol"; error TokenTransfer__AddressZero(); +error TokenTransfer__InvalidTransferType(); contract TokenTransfer { using SafeERC20 for IERC20; @@ -67,6 +68,8 @@ contract TokenTransfer { permit2.transferFrom( sender, address(this), uint160(amount), tokenIn ); + } else { + revert TokenTransfer__InvalidTransferType(); } } }