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

Update compound addresses

Browse Source
This commit is contained in:
Mubaris NK 2021-01-11 10:55:32 +05:30
parent f32526fb8e
commit fb89682e6f
No known key found for this signature in database
GPG Key ID: 1F132294E8700320
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
contracts/connectors/refinance.sol
View File

@ -559,15 +559,17 @@ contract CompoundHelpers is Helpers {
uint rateMode uint rateMode
) internal returns (uint) { ) internal returns (uint) {
if (amt > 0) { if (amt > 0) {
address _token = address(token) == getWethAddr() ? getEthAddr() : address(token);
if (amt == uint(-1)) { if (amt == uint(-1)) {
amt = getMaxBorrow(target, address(token), rateMode); amt = getMaxBorrow(target, _token, rateMode);
} }
uint feeAmt = wmul(amt, fee); uint feeAmt = wmul(amt, fee);
uint _amt = add(amt, feeAmt); uint _amt = add(amt, feeAmt);
require(ctoken.borrow(_amt) == 0, "borrow-failed-collateral?"); require(ctoken.borrow(_amt) == 0, "borrow-failed-collateral?");
transferFees(address(token), feeAmt); transferFees(_token, feeAmt);
} }
return amt; return amt;
} }
@ -591,16 +593,18 @@ contract CompoundHelpers is Helpers {
function _compDepositOne(uint fee, CTokenInterface ctoken, TokenInterface token, uint amt) internal { function _compDepositOne(uint fee, CTokenInterface ctoken, TokenInterface token, uint amt) internal {
if (amt > 0) { if (amt > 0) {
address _token = address(token) == getWethAddr() ? getEthAddr() : address(token);
uint feeAmt = wmul(amt, fee); uint feeAmt = wmul(amt, fee);
uint _amt = sub(amt, feeAmt); uint _amt = sub(amt, feeAmt);
if (address(token) != getEthAddr()) { if (_token != getEthAddr()) {
token.approve(address(ctoken), _amt); token.approve(address(ctoken), _amt);
require(ctoken.mint(_amt) == 0, "deposit-failed"); require(ctoken.mint(_amt) == 0, "deposit-failed");
} else { } else {
CETHInterface(address(ctoken)).mint.value(_amt)(); CETHInterface(address(ctoken)).mint.value(_amt)();
} }
transferFees(address(token), feeAmt); transferFees(_token, feeAmt);
} }
} }
@ -643,7 +647,7 @@ contract CompoundHelpers is Helpers {
if (amt == uint(-1)) { if (amt == uint(-1)) {
amt = ctoken.borrowBalanceCurrent(address(this)); amt = ctoken.borrowBalanceCurrent(address(this));
} }
if (address(token) != getEthAddr()) { if (address(token) != getWethAddr()) {
token.approve(address(ctoken), amt); token.approve(address(ctoken), amt);
require(ctoken.repayBorrow(amt) == 0, "repay-failed."); require(ctoken.repayBorrow(amt) == 0, "repay-failed.");
} else { } else {