add revert for repay

JOJOexchange · Feb 26, 2024 · 6b76d16 · 6b76d16
1 parent aaab69e
commit 6b76d16
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/JUSDRepayHelper.sol b/src/JUSDRepayHelper.sol
@@ -54,6 +54,8 @@ contract JUSDRepayHelper is Ownable {
             IJUSDExchange(JUSDExchange).buyJUSD(USDCBalance, address(this));
         }
         uint256 balance = IERC20(JUSD).balanceOf(address(this));
+        uint256 borrowed = IJUSDBank(JusdBank).getBorrowBalance(to);
+        require(balance <= borrowed, "repayed jusd too much");
         IJUSDBank(JusdBank).repay(balance, to);
         emit HelpToTransfer(from, to, balance);
     }