fix: Fix stack too deep for unoptimized compile of PoolConfigurator

aave · Oct 6, 2021 · ef14f03 · ef14f03
1 parent 10300d4
commit ef14f03
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 2 deletions.
diff --git a/contracts/interfaces/IPoolDataProvider.sol b/contracts/interfaces/IPoolDataProvider.sol
@@ -35,4 +35,11 @@ interface IPoolDataProvider {
       uint256 variableBorrowIndex,
       uint40 lastUpdateTimestamp
     );
+
+  /**
+   * @notice Returns the total supply of aTokens for a given asset
+   * @param asset The address of the underlying asset of the reserve
+   * @return The total supply of the aToken
+   **/
+  function getATokenTotalSupply(address asset) external view returns (uint256);
 }
diff --git a/contracts/misc/AaveProtocolDataProvider.sol b/contracts/misc/AaveProtocolDataProvider.sol
@@ -229,6 +229,18 @@ contract AaveProtocolDataProvider is IPoolDataProvider {
     );
   }
 
+  /**
+   * @notice Returns the total supply of aTokens for a given asset
+   * @param asset The address of the underlying asset of the reserve
+   * @return The total supply of the aToken
+   **/
+  function getATokenTotalSupply(address asset) external view override returns (uint256) {
+    DataTypes.ReserveData memory reserve = IPool(ADDRESSES_PROVIDER.getPool()).getReserveData(
+      asset
+    );
+    return IERC20Detailed(reserve.aTokenAddress).totalSupply();
+  }
+
   /**
    * @notice Returns the user data in a reserve
    * @param asset The address of the underlying asset of the reserve

diff --git a/contracts/protocol/pool/PoolConfigurator.sol b/contracts/protocol/pool/PoolConfigurator.sol
@@ -409,9 +409,11 @@ contract PoolConfigurator is VersionedInitializable, IPoolConfigurator {
   }
 
   function _checkNoLiquidity(address asset) internal view {
-    (, , uint256 totalATokens, , , , , , , , , ) = IPoolDataProvider(
+    uint256 totalATokens = IPoolDataProvider(_addressesProvider.getPoolDataProvider())
+      .getATokenTotalSupply(asset);
+    /*(, , uint256 totalATokens, , , , , , , , , ) = IPoolDataProvider(
       _addressesProvider.getPoolDataProvider()
-    ).getReserveData(asset);
+    ).getReserveData(asset);*/
 
     require(totalATokens == 0, Errors.PC_RESERVE_LIQUIDITY_NOT_0);
   }