Merge branch 'main' into feature/gas-optimization

2025-04-07 19:59:55 -04:00
parent 56d3eee618 f7cdc6f537
commit db89a99826
7 changed files with 531 additions and 4 deletions
--- a/foundry/src/executors/CurveExecutor.sol
+++ b/foundry/src/executors/CurveExecutor.sol
@@ -0,0 +1,133 @@
+// SPDX-License-Identifier: BUSL-1.1
+pragma solidity ^0.8.26;
+
+import "@interfaces/IExecutor.sol";
+import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
+
+error CurveExecutor__InvalidAddresses();
+
+interface CryptoPool {
+    // slither-disable-next-line naming-convention
+    function exchange(uint256 i, uint256 j, uint256 dx, uint256 min_dy)
+        external
+        payable;
+}
+
+interface StablePool {
+    // slither-disable-next-line naming-convention
+    function exchange(int128 i, int128 j, uint256 dx, uint256 min_dy)
+        external
+        payable;
+}
+
+interface CryptoPoolETH {
+    // slither-disable-start naming-convention
+    function exchange(
+        uint256 i,
+        uint256 j,
+        uint256 dx,
+        uint256 min_dy,
+        bool use_eth
+    ) external payable;
+    // slither-disable-end naming-convention
+}
+
+contract CurveExecutor is IExecutor {
+    using SafeERC20 for IERC20;
+
+    address public immutable nativeToken;
+
+    constructor(address _nativeToken) {
+        if (_nativeToken == address(0)) {
+            revert CurveExecutor__InvalidAddresses();
+        }
+        nativeToken = _nativeToken;
+    }
+
+    // slither-disable-next-line locked-ether
+    function swap(uint256 amountIn, bytes calldata data)
+        external
+        payable
+        returns (uint256)
+    {
+        (
+            address tokenIn,
+            address tokenOut,
+            address pool,
+            uint8 poolType,
+            int128 i,
+            int128 j,
+            bool tokenApprovalNeeded
+        ) = _decodeData(data);
+
+        if (tokenApprovalNeeded && tokenIn != nativeToken) {
+            // slither-disable-next-line unused-return
+            IERC20(tokenIn).approve(address(pool), type(uint256).max);
+        }
+
+        /// Inspired by Curve's router contract: https://github.com/curvefi/curve-router-ng/blob/9ab006ca848fc7f1995b6fbbecfecc1e0eb29e2a/contracts/Router.vy#L44
+        uint256 balanceBefore = _balanceOf(tokenOut);
+
+        uint256 ethAmount = 0;
+        if (tokenIn == nativeToken) {
+            ethAmount = amountIn;
+        }
+
+        if (poolType == 1 || poolType == 10) {
+            // stable and stable_ng
+            // slither-disable-next-line arbitrary-send-eth
+            StablePool(pool).exchange{value: ethAmount}(i, j, amountIn, 0);
+        } else {
+            // crypto or llamma
+            if (tokenIn == nativeToken || tokenOut == nativeToken) {
+                // slither-disable-next-line arbitrary-send-eth
+                CryptoPoolETH(pool).exchange{value: ethAmount}(
+                    uint256(int256(i)), uint256(int256(j)), amountIn, 0, true
+                );
+            } else {
+                CryptoPool(pool).exchange(
+                    uint256(int256(i)), uint256(int256(j)), amountIn, 0
+                );
+            }
+        }
+
+        uint256 balanceAfter = _balanceOf(tokenOut);
+        return balanceAfter - balanceBefore;
+    }
+
+    function _decodeData(bytes calldata data)
+        internal
+        pure
+        returns (
+            address tokenIn,
+            address tokenOut,
+            address pool,
+            uint8 poolType,
+            int128 i,
+            int128 j,
+            bool tokenApprovalNeeded
+        )
+    {
+        tokenIn = address(bytes20(data[0:20]));
+        tokenOut = address(bytes20(data[20:40]));
+        pool = address(bytes20(data[40:60]));
+        poolType = uint8(data[60]);
+        i = int128(uint128(uint8(data[61])));
+        j = int128(uint128(uint8(data[62])));
+        tokenApprovalNeeded = data[63] != 0;
+    }
+
+    receive() external payable {
+        require(msg.sender.code.length != 0);
+    }
+
+    function _balanceOf(address token)
+        internal
+        view
+        returns (uint256 balance)
+    {
+        balance = token == nativeToken
+            ? address(this).balance
+            : IERC20(token).balanceOf(address(this));
+    }
+}