Instadapp/dsa-connectors
1
0
Fork 0
mirror of https://github.com/Instadapp/dsa-connectors.git synced 2024-07-29 22:37:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

error

Browse Source
This commit is contained in:
Richa-iitr 2022-09-02 22:56:08 +05:30
parent fa345da4fc
commit 91bb5ea57c
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
contracts/mainnet/connectors/compound/v3/main.sol
View File

@ -226,9 +226,15 @@ abstract contract CompoundV3Resolver is Events, Helpers {
if (token_ == getBaseToken(market)) { if (token_ == getBaseToken(market)) {
uint256 balance = CometInterface(market).balanceOf(address(this)); uint256 balance = CometInterface(market).balanceOf(address(this));
//if there are supplies, ensure withdrawn amount is not greater than supplied i.e can't borrow using withdraw.
if (balance > 0) { if (balance > 0) {
require(amt_ <= balance, "withdraw-amt-greater-than-supplies"); require(amt_ <= balance, "withdraw-amt-greater-than-supplies");
} }
//if borrow balance > 0, there are no supplies so no withdraw, borrow instead.
require(
CometInterface(market).borrowBalanceOf(address(this)) == 0,
"withdraw-disabled-for-zero-supplies"
);
} }
CometInterface(market).withdraw(token_, amt_); CometInterface(market).withdraw(token_, amt_);