fix: Revert if the TransferType is not valid

Took 8 minutes
This commit is contained in:
Diana Carvalho
2025-05-14 09:02:55 +01:00
parent 65bd0d0749
commit b0c254add4

View File

@@ -6,6 +6,7 @@ import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
import "@permit2/src/interfaces/IAllowanceTransfer.sol"; import "@permit2/src/interfaces/IAllowanceTransfer.sol";
error TokenTransfer__AddressZero(); error TokenTransfer__AddressZero();
error TokenTransfer__InvalidTransferType();
contract TokenTransfer { contract TokenTransfer {
using SafeERC20 for IERC20; using SafeERC20 for IERC20;
@@ -67,6 +68,8 @@ contract TokenTransfer {
permit2.transferFrom( permit2.transferFrom(
sender, address(this), uint160(amount), tokenIn sender, address(this), uint160(amount), tokenIn
); );
} else {
revert TokenTransfer__InvalidTransferType();
} }
} }
} }