🦺 market: prevent zero floating borrows

exactly · May 24, 2024 · 277120a · 277120a
1 parent 25d0fe7
commit 277120a
Show file tree

Hide file tree

Showing 6 changed files with 173 additions and 162 deletions.
diff --git a/.gas-snapshot b/.gas-snapshot
diff --git a/contracts/Market.sol b/contracts/Market.sol
@@ -142,6 +142,7 @@ contract Market is Initializable, AccessControlUpgradeable, PausableUpgradeable,
     address receiver,
     address borrower
   ) external whenNotPaused whenNotFrozen returns (uint256 borrowShares) {
+    if (assets == 0) revert ZeroBorrow();
     spendAllowance(borrower, assets);
 
     RewardsController memRewardsController = rewardsController;

diff --git a/test/InterestRateModel.t.sol b/test/InterestRateModel.t.sol
@@ -234,7 +234,9 @@ contract InterestRateModelTest is Test {
     asset.approve(address(market), type(uint128).max);
     if (floatingAssets != 0) market.deposit(floatingAssets, address(this));
     vm.warp(timestamp);
-    market.borrow(floatingDebt, address(this), address(this));
+    if (floatingDebt != 0) {
+      market.borrow(floatingDebt, address(this), address(this));
+    }
 
     Vars memory v;
     v.backupBorrowed = 0;

diff --git a/test/Market.t.sol b/test/Market.t.sol
@@ -227,6 +227,11 @@ contract MarketTest is Test {
     market.borrowAtMaturity(FixedLib.INTERVAL, 0, 0, address(this), address(this));
   }
 
+  function testBorrowWithZeroAssets() external {
+    vm.expectRevert(ZeroBorrow.selector);
+    market.borrow(0, address(this), BOB);
+  }
+
   function testDepositAtMaturityWithZeroAssets() external {
     vm.expectRevert(ZeroDeposit.selector);
     market.depositAtMaturity(FixedLib.INTERVAL, 0, 0, address(this));

diff --git a/test/Previewer.t.sol b/test/Previewer.t.sol
@@ -1051,7 +1051,7 @@ contract PreviewerTest is Test {
     market.deposit(10_000 ether, address(this));
     market.borrow(2_000 ether, address(this), address(this));
     vm.warp(block.timestamp + 1 weeks);
-    market.borrow(0, address(this), address(this));
+    market.borrow(1, address(this), address(this));
     Previewer.MarketAccount[] memory data = previewer.exactly(address(this));
     assertEq(data[0].rewardRates.length, 1);
     assertEq(data[0].rewardRates[0].maturities.length, 12);

diff --git a/test/Protocol.t.sol b/test/Protocol.t.sol
@@ -410,7 +410,9 @@ contract ProtocolTest is Test {
     uint256 expectedShares = market.previewBorrow(assets);
     (uint256 collateral, uint256 debt) = previewAccountLiquidity(account, market, assets, expectedShares);
 
-    if (
+    if (assets == 0) {
+      vm.expectRevert(ZeroBorrow.selector);
+    } else if (
       market.floatingBackupBorrowed() + market.totalFloatingBorrowAssets() + assets >
       (market.floatingAssets() + previewNewFloatingDebt(market)).mulWadDown(1e18 - RESERVE_FACTOR)
     ) {