This commit is contained in:
royvardhan
2025-01-22 20:54:36 +05:30
parent a1e7b552b6
commit 0c05874477

View File

@@ -6,12 +6,12 @@ import "@permit2/src/interfaces/IAllowanceTransfer.sol";
import "@openzeppelin/contracts/token/ERC20/IERC20.sol"; import "@openzeppelin/contracts/token/ERC20/IERC20.sol";
import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol"; import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
error TychoRouter__WithdrawalFailed(); error TychoRouter__WithdrawalFailed();
error TychoRouter__InvalidReceiver(); error TychoRouter__InvalidReceiver();
contract TychoRouter is AccessControl { contract TychoRouter is AccessControl {
IAllowanceTransfer public immutable permit2; IAllowanceTransfer public immutable permit2;
using SafeERC20 for IERC20; using SafeERC20 for IERC20;
//keccak256("NAME_OF_ROLE") : save gas on deployment //keccak256("NAME_OF_ROLE") : save gas on deployment
@@ -69,8 +69,7 @@ contract TychoRouter is AccessControl {
// TODO // TODO
} }
/**
/**
* @dev Allows withdrawing any ERC20 funds if funds get stuck in case of a bug. * @dev Allows withdrawing any ERC20 funds if funds get stuck in case of a bug.
*/ */
function withdraw(IERC20[] memory tokens, address receiver) function withdraw(IERC20[] memory tokens, address receiver)
@@ -79,12 +78,12 @@ contract TychoRouter is AccessControl {
{ {
if (receiver == address(0)) revert TychoRouter__InvalidReceiver(); if (receiver == address(0)) revert TychoRouter__InvalidReceiver();
for (uint256 i = 0; i < tokens.length; i++) { for (uint256 i = 0; i < tokens.length; i++) {
// slither-disable-next-line calls-loop
uint256 tokenBalance = tokens[i].balanceOf(address(this)); uint256 tokenBalance = tokens[i].balanceOf(address(this));
if (tokenBalance > 0) { if (tokenBalance > 0) {
tokens[i].safeTransfer(receiver, tokenBalance);
emit Withdrawal(address(tokens[i]), tokenBalance); emit Withdrawal(address(tokens[i]), tokenBalance);
tokens[i].safeTransfer(receiver, tokenBalance);
} }
} }
} }
@@ -101,14 +100,13 @@ contract TychoRouter is AccessControl {
uint256 amount = address(this).balance; uint256 amount = address(this).balance;
if (amount > 0) { if (amount > 0) {
// slither-disable-next-line arbitrary-send-eth emit Withdrawal(address(0), amount);
// slither-disable-next-line arbitrary-send-eth
(bool success,) = receiver.call{value: amount}(""); (bool success,) = receiver.call{value: amount}("");
if (!success) revert TychoRouter__WithdrawalFailed(); if (!success) revert TychoRouter__WithdrawalFailed();
emit Withdrawal(address(0), amount);
} }
} }
/** /**
* @dev Allows this contract to receive native token * @dev Allows this contract to receive native token
*/ */