fix: Revert if the TransferType is not valid
Took 8 minutes
This commit is contained in:
@@ -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();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user