diff --git a/contracts/token/BiconomyTokenPaymaster.sol b/contracts/token/BiconomyTokenPaymaster.sol index 1642099..53e7cad 100644 --- a/contracts/token/BiconomyTokenPaymaster.sol +++ b/contracts/token/BiconomyTokenPaymaster.sol @@ -5,6 +5,7 @@ import { ReentrancyGuardTransient } from "@openzeppelin/contracts/utils/Reentran import { IEntryPoint } from "account-abstraction/interfaces/IEntryPoint.sol"; import { PackedUserOperation, UserOperationLib } from "account-abstraction/core/UserOperationLib.sol"; import { IERC20 } from "@openzeppelin/contracts/token/ERC20/ERC20.sol"; +import { SafeERC20 } from "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol"; import { IERC20Metadata } from "@openzeppelin/contracts/token/ERC20/extensions/IERC20Metadata.sol"; import { SafeTransferLib } from "solady/utils/SafeTransferLib.sol"; import { BasePaymaster } from "../base/BasePaymaster.sol"; @@ -129,7 +130,7 @@ contract BiconomyTokenPaymaster is // Approve swappable tokens for max amount uint256 length = swappableTokens.length; for (uint256 i; i < length; i++) { - IERC20(swappableTokens[i]).approve(address(uniswapRouterArg), type(uint256).max); + SafeERC20.forceApprove(IERC20(swappableTokens[i]), address(uniswapRouterArg), type(uint256).max); } } diff --git a/contracts/token/swaps/Uniswapper.sol b/contracts/token/swaps/Uniswapper.sol index 2994478..9671ec8 100644 --- a/contracts/token/swaps/Uniswapper.sol +++ b/contracts/token/swaps/Uniswapper.sol @@ -2,6 +2,7 @@ pragma solidity ^0.8.27; import "@openzeppelin/contracts/token/ERC20/IERC20.sol"; +import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol"; import "@uniswap/v3-periphery/contracts/interfaces/ISwapRouter.sol"; import "@uniswap/v3-periphery/contracts/interfaces/IPeripheryPayments.sol"; @@ -48,7 +49,7 @@ abstract contract Uniswapper { } function _setTokenPool(address token, uint24 poolFeeTier) internal { - IERC20(token).approve(address(uniswapRouter), type(uint256).max); // one time max approval + SafeERC20.forceApprove(IERC20(token), address(uniswapRouter), type(uint256).max); // one time max approval tokenToPools[token] = poolFeeTier; // set mapping of token to uniswap pool to use for swap } diff --git a/package.json b/package.json index fd27a7d..96bcd82 100644 --- a/package.json +++ b/package.json @@ -7,7 +7,7 @@ "url": "https://github.com/bcnmy" }, "dependencies": { - "@openzeppelin/contracts": "5.0.2", + "@openzeppelin/contracts": "5.1.0", "@rhinestone/modulekit": "^0.4.10", "@uniswap/v3-core": "https://github.com/Uniswap/v3-core#0.8", "@uniswap/v3-periphery": "https://github.com/Uniswap/v3-periphery#0.8",