diff --git a/contracts/common/BiconomyTokenPaymasterErrors.sol b/contracts/common/BiconomyTokenPaymasterErrors.sol index b77a63f..6188f11 100644 --- a/contracts/common/BiconomyTokenPaymasterErrors.sol +++ b/contracts/common/BiconomyTokenPaymasterErrors.sol @@ -75,4 +75,14 @@ contract BiconomyTokenPaymasterErrors { * @notice Throws when external signer's signature has invalid length */ error InvalidSignatureLength(); + + /** + * @notice Throws when ETH withdrawal fails + */ + error WithdrawalFailed(); + + /** + * @notice Emitted when ETH is withdrawn from the paymaster + */ + event EthWithdrawn(address indexed recipient, uint256 indexed amount); } diff --git a/contracts/token/BiconomyTokenPaymaster.sol b/contracts/token/BiconomyTokenPaymaster.sol index 5689715..07ce080 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"; @@ -125,6 +126,10 @@ contract BiconomyTokenPaymaster is } } + receive() external payable { + // no need to emit an event here + } + /** * @dev pull tokens out of paymaster in case they were sent to the paymaster at any point. * @param token the token deposit to withdraw @@ -135,6 +140,19 @@ contract BiconomyTokenPaymaster is _withdrawERC20(token, target, amount); } + /** + * @dev Withdraw ETH from the paymaster + * @param recipient The address to send the ETH to + * @param amount The amount of ETH to withdraw + */ + function withdrawEth(address payable recipient, uint256 amount) external payable onlyOwner nonReentrant { + (bool success,) = recipient.call{ value: amount }(""); + if (!success) { + revert WithdrawalFailed(); + } + emit EthWithdrawn(recipient, amount); + } + /** * @dev pull tokens out of paymaster in case they were sent to the paymaster at any point. * @param token the token deposit to withdraw @@ -497,7 +515,7 @@ contract BiconomyTokenPaymaster is // deduct max penalty from the token amount we pass to the postOp // so we don't refund it at postOp - context = abi.encode(userOp.sender, tokenAddress, tokenAmount-((maxPenalty*tokenPrice)/_NATIVE_TOKEN_DECIMALS), tokenPrice, externalPriceMarkup, userOpHash); + context = abi.encode(userOp.sender, tokenAddress, tokenAmount-((maxPenalty*tokenPrice*externalPriceMarkup)/(1e18*_PRICE_DENOMINATOR)), tokenPrice, externalPriceMarkup, userOpHash); validationData = _packValidationData(false, validUntil, validAfter); } else if (mode == PaymasterMode.INDEPENDENT) { // Use only oracles for the token specified in modeSpecificData @@ -525,7 +543,7 @@ contract BiconomyTokenPaymaster is SafeTransferLib.safeTransferFrom(tokenAddress, userOp.sender, address(this), tokenAmount); context = - abi.encode(userOp.sender, tokenAddress, tokenAmount-((maxPenalty*tokenPrice)/_NATIVE_TOKEN_DECIMALS), tokenPrice, independentPriceMarkup, userOpHash); + abi.encode(userOp.sender, tokenAddress, tokenAmount-((maxPenalty*tokenPrice*independentPriceMarkup)/(1e18*_PRICE_DENOMINATOR)), tokenPrice, independentPriceMarkup, userOpHash); validationData = 0; // Validation success and price is valid indefinetly } } diff --git a/contracts/token/swaps/Uniswapper.sol b/contracts/token/swaps/Uniswapper.sol index b2c1d73..2004d2e 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"; @@ -46,7 +47,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 }