Instadapp/yield-contract
1
0
Fork 0
mirror of https://github.com/Instadapp/yield-contract.git synced 2024-07-29 21:47:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Set max if amount higher than max withdraw amount

Browse Source
This commit is contained in:
Thrilok Kumar 2020-08-31 12:54:45 +05:30
parent d43a1509ee
commit ee9e1b39d0
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
contracts/pools/erc20.sol
View File

@ -169,11 +169,10 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable {
uint _bal = balanceOf(msg.sender); uint _bal = balanceOf(msg.sender);
uint _tknBal = wdiv(_bal, exchangeRate); uint _tknBal = wdiv(_bal, exchangeRate);
uint _burnAmt; uint _burnAmt;
if (tknAmt == uint(-1)) { if (tknAmt >= _tknBal) {
_burnAmt = _bal; _burnAmt = _bal;
_tknAmt = _tknBal; _tknAmt = _tknBal;
} else { } else {
require(tknAmt <= _tknBal, "balance-exceeded");
_burnAmt = wmul(tknAmt, exchangeRate); _burnAmt = wmul(tknAmt, exchangeRate);
_tknAmt = tknAmt; _tknAmt = tknAmt;
} }

3
contracts/pools/eth.sol
View File

@ -165,11 +165,10 @@ contract PoolToken is ReentrancyGuard, ERC20Pausable, DSMath {
uint _bal = balanceOf(msg.sender); uint _bal = balanceOf(msg.sender);
uint _tknBal = wdiv(_bal, exchangeRate); uint _tknBal = wdiv(_bal, exchangeRate);
uint _burnAmt; uint _burnAmt;
if (tknAmt == uint(-1)) { if (tknAmt >= _tknBal) {
_burnAmt = _bal; _burnAmt = _bal;
_tknAmt = _tknBal; _tknAmt = _tknBal;
} else { } else {
require(tknAmt <= _tknBal, "balance-exceeded");
_burnAmt = wmul(tknAmt, exchangeRate); _burnAmt = wmul(tknAmt, exchangeRate);
_tknAmt = tknAmt; _tknAmt = tknAmt;
} }