fix: move downscaledAmount variable around to avoid stack too deep error

primitivefinance · Apr 10, 2024 · fcf65cb · fcf65cb
1 parent 8befbd0
commit fcf65cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/DFMM.sol b/src/DFMM.sol
@@ -268,8 +268,6 @@ contract DFMM is IDFMM {
         } else {
             // Otherwise, execute the callback and assert the input amount has been paid
             // given the before and after balances of the input token.
-            uint256 downscaledAmount =
-                downscaleUp(state.amountIn, computeScalingFactor(state.tokenIn));
             uint256 preBalance = ERC20(state.tokenIn).balanceOf(address(this));
 
             ISwapCallback(msg.sender).swapCallback(
@@ -280,6 +278,8 @@ contract DFMM is IDFMM {
                 callbackData
             );
 
+            uint256 downscaledAmount =
+                downscaleUp(state.amountIn, computeScalingFactor(state.tokenIn));
             uint256 postBalance = ERC20(state.tokenIn).balanceOf(address(this));
             if (postBalance < preBalance + downscaledAmount) {
                 revert InvalidTransfer();