make all methods payable

1inch · Sep 4, 2024 · 4f0281f · 4f0281f
1 parent ffa73ab
commit 4f0281f
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 16 deletions.
diff --git a/contracts/BalanceManager.sol b/contracts/BalanceManager.sol
@@ -29,7 +29,7 @@ abstract contract BalanceManager is IERC1271, IBalanceManager {
     /**
      * @notice See {IBalanceManager-arbitraryCalls}.
      */
-    function arbitraryCalls(address[] calldata targets, bytes[] calldata arguments) public {
+    function arbitraryCalls(address[] calldata targets, bytes[] calldata arguments) public payable {
         uint256[] calldata values;
         // solhint-disable-next-line no-inline-assembly
         assembly ("memory-safe") {
@@ -42,7 +42,7 @@ abstract contract BalanceManager is IERC1271, IBalanceManager {
     /**
      * @notice See {IBalanceManager-arbitraryCalls}.
      */
-    function arbitraryCalls(address[] calldata targets, bytes[] calldata arguments, uint256[] calldata values) public onlyOwner {
+    function arbitraryCalls(address[] calldata targets, bytes[] calldata arguments, uint256[] calldata values) public payable onlyOwner {
         unchecked {
             uint256 length = targets.length;
             if (length != arguments.length) revert LengthMismatch();
@@ -57,7 +57,7 @@ abstract contract BalanceManager is IERC1271, IBalanceManager {
     /**
      * @notice See {IBalanceManager-arbitraryCallsWithEthCheck}.
      */
-    function arbitraryCallsWithEthCheck(address[] calldata targets, bytes[] calldata arguments, uint256 minReturn) external {
+    function arbitraryCallsWithEthCheck(address[] calldata targets, bytes[] calldata arguments, uint256 minReturn) external payable {
         uint256[] calldata values;
         // solhint-disable-next-line no-inline-assembly
         assembly ("memory-safe") {
@@ -75,7 +75,7 @@ abstract contract BalanceManager is IERC1271, IBalanceManager {
         bytes[] calldata arguments,
         uint256[] calldata values,
         uint256 minReturn
-    ) public {
+    ) public payable {
         uint256 balanceBefore = msg.sender.balance;
         arbitraryCalls(targets, arguments, values);
         if (msg.sender.balance - balanceBefore < minReturn) revert NotEnoughProfit();
@@ -89,7 +89,7 @@ abstract contract BalanceManager is IERC1271, IBalanceManager {
         bytes[] calldata arguments,
         IERC20 token,
         uint256 minReturn
-    ) external {
+    ) external payable {
         uint256[] calldata values;
         // solhint-disable-next-line no-inline-assembly
         assembly ("memory-safe") {
@@ -108,7 +108,7 @@ abstract contract BalanceManager is IERC1271, IBalanceManager {
         uint256[] calldata values,
         IERC20 token,
         uint256 minReturn
-    ) public {
+    ) public payable {
         uint256 balanceBefore = token.balanceOf(msg.sender);
         arbitraryCalls(targets, arguments, values);
         if (token.balanceOf(msg.sender) - balanceBefore < minReturn) revert NotEnoughProfit();
@@ -117,7 +117,7 @@ abstract contract BalanceManager is IERC1271, IBalanceManager {
     /**
      * @notice See {IBalanceManager-estimateArbitraryCalls}.
      */
-    function estimateArbitraryCalls(address[] calldata targets, bytes[] calldata arguments) external {
+    function estimateArbitraryCalls(address[] calldata targets, bytes[] calldata arguments) external payable {
         uint256[] calldata values;
         // solhint-disable-next-line no-inline-assembly
         assembly ("memory-safe") {
@@ -130,7 +130,7 @@ abstract contract BalanceManager is IERC1271, IBalanceManager {
     /**
      * @notice See {IBalanceManager-estimateArbitraryCalls}.
      */
-    function estimateArbitraryCalls(address[] calldata targets, bytes[] calldata arguments, uint256[] calldata values) public onlyOwner {
+    function estimateArbitraryCalls(address[] calldata targets, bytes[] calldata arguments, uint256[] calldata values) public payable onlyOwner {
         unchecked {
             uint256 length = targets.length;
             if (length != arguments.length) revert LengthMismatch();

diff --git a/contracts/interfaces/IBalanceManager.sol b/contracts/interfaces/IBalanceManager.sol
@@ -20,23 +20,23 @@ interface IBalanceManager{
      * @param targets Addresses of the contracts to call.
      * @param arguments Data to send to each contract.
      */
-    function arbitraryCalls(address[] calldata targets, bytes[] calldata arguments) external;
+    function arbitraryCalls(address[] calldata targets, bytes[] calldata arguments) external payable;
 
     /**
      * @notice Execute arbitrary calls.
      * @param targets Addresses of the contracts to call.
      * @param arguments Data to send to each contract.
      * @param values Values to send to each contract.
      */
-    function arbitraryCalls(address[] calldata targets, bytes[] calldata arguments, uint256[] calldata values) external;
+    function arbitraryCalls(address[] calldata targets, bytes[] calldata arguments, uint256[] calldata values) external payable;
 
     /**
      * @notice Execute arbitrary calls and check the ETH balance after.
      * @param targets Addresses of the contracts to call.
      * @param arguments Data to send to each contract.
      * @param minReturn Minimum amount of ETH balance after all calls.
      */
-    function arbitraryCallsWithEthCheck(address[] calldata targets, bytes[] calldata arguments, uint256 minReturn) external;
+    function arbitraryCallsWithEthCheck(address[] calldata targets, bytes[] calldata arguments, uint256 minReturn) external payable;
 
     /**
      * @notice Execute arbitrary calls and check the ETH balance after.
@@ -50,7 +50,7 @@ interface IBalanceManager{
         bytes[] calldata arguments,
         uint256[] calldata values,
         uint256 minReturn
-    ) external;
+    ) external payable;
 
     /**
      * @notice Execute arbitrary calls and check the token balance after.
@@ -64,7 +64,7 @@ interface IBalanceManager{
         bytes[] calldata arguments,
         IERC20 token,
         uint256 minReturn
-    ) external;
+    ) external payable;
 
     /**
      * @notice Execute arbitrary calls and check the token balance after.
@@ -80,15 +80,15 @@ interface IBalanceManager{
         uint256[] calldata values,
         IERC20 token,
         uint256 minReturn
-    ) external;
+    ) external payable;
 
     /**
      * @notice Estimate the results of arbitrary calls.
      * @param targets Addresses of the contracts to call.
      * @param arguments Data to send to each contract.
      * @dev This function reverts results with `EstimationResults` error.
      */
-    function estimateArbitraryCalls(address[] calldata targets, bytes[] calldata arguments) external;
+    function estimateArbitraryCalls(address[] calldata targets, bytes[] calldata arguments) external payable;
 
     /**
      * @notice Estimate the results of arbitrary calls.
@@ -97,7 +97,7 @@ interface IBalanceManager{
      * @param values Values to send to each contract.
      * @dev This function reverts results with `EstimationResults` error.
      */
-    function estimateArbitraryCalls(address[] calldata targets, bytes[] calldata arguments, uint256[] calldata values) external;
+    function estimateArbitraryCalls(address[] calldata targets, bytes[] calldata arguments, uint256[] calldata values) external payable;
 
     /**
      * @notice Approves a spender to spend an infinite amount of tokens.