new updates implemented

This commit is contained in:
Samyak Jain 2020-09-06 22:08:29 +10:00
parent 80d318bcbb
commit 53838909ec
3 changed files with 122 additions and 78 deletions

View File

@ -21,9 +21,10 @@ interface IndexInterface {
interface RegistryInterface { interface RegistryInterface {
function chief(address) external view returns (bool); function chief(address) external view returns (bool);
function poolLogic(address) external returns (address); function poolLogic(address) external returns (address);
function insureFee(address) external view returns (uint); function fee(address) external view returns (uint);
function withdrawalFee(address) external view returns (uint); function withdrawalFee(address) external view returns (uint);
function isDsa(address, address) external view returns (bool); function isDsa(address, address) external view returns (bool);
function checkSettleLogics(address, address[] calldata) external view returns (bool);
} }
interface RateInterface { interface RateInterface {
@ -38,8 +39,8 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable {
event LogSettle(uint settleBlock); event LogSettle(uint settleBlock);
event LogDeposit(address indexed user, uint depositAmt, uint poolMintAmt); event LogDeposit(address indexed user, uint depositAmt, uint poolMintAmt);
event LogWithdraw(address indexed user, uint withdrawAmt, uint poolBurnAmt, uint feeAmt); event LogWithdraw(address indexed user, uint withdrawAmt, uint poolBurnAmt, uint feeAmt);
event LogAddInsurance(uint amount); event LogAddFee(uint amount);
event LogWithdrawInsurance(uint amount); event LogWithdrawFee(uint amount);
event LogPausePool(bool); event LogPausePool(bool);
IERC20 public immutable baseToken; // Base token. Eg:- DAI, USDC, etc. IERC20 public immutable baseToken; // Base token. Eg:- DAI, USDC, etc.
@ -48,7 +49,7 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable {
uint private tokenBalance; // total token balance uint private tokenBalance; // total token balance
uint public exchangeRate; // initial 1 token = 1 uint public exchangeRate; // initial 1 token = 1
uint public insuranceAmt; // insurance amount to keep pool safe uint public feeAmt; // fee collected on profits
constructor( constructor(
address _registry, address _registry,
@ -62,7 +63,7 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable {
} }
modifier isChief() { modifier isChief() {
require(registry.chief(msg.sender) || msg.sender == instaIndex.master(), "not-chief"); require(registry.chief(msg.sender) || msg.sender == instaIndex.master() || msg.sender == address(this), "not-chief");
_; _;
} }
@ -99,43 +100,52 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable {
function setExchangeRate() public isChief { function setExchangeRate() public isChief {
uint _previousRate = exchangeRate; uint _previousRate = exchangeRate;
uint _totalToken = RateInterface(registry.poolLogic(address(this))).getTotalToken(); uint _totalToken = RateInterface(registry.poolLogic(address(this))).getTotalToken();
_totalToken = sub(_totalToken, insuranceAmt); _totalToken = sub(_totalToken, feeAmt);
uint _currentRate = getCurrentRate(_totalToken); uint _currentRate = getCurrentRate(_totalToken);
require(_currentRate != 0, "current-rate-is-zero"); require(_currentRate != 0, "current-rate-is-zero");
if (_currentRate > _previousRate) { // loss => deduct partially/fully from insurance amount if (_currentRate > _previousRate) { // loss => deduct partially/fully from insurance amount
uint _loss = sub(tokenBalance, _totalToken);
if (_loss <= insuranceAmt) {
insuranceAmt = sub(insuranceAmt, _loss);
_currentRate = _previousRate; _currentRate = _previousRate;
} else {
tokenBalance = add(_totalToken, insuranceAmt);
insuranceAmt = 0;
_currentRate = getCurrentRate(tokenBalance);
}
} else { // profit => add to insurance amount } else { // profit => add to insurance amount
uint insureFeeAmt = wmul(sub(_totalToken, tokenBalance), registry.insureFee(address(this))); uint _newFee = wmul(sub(_totalToken, tokenBalance), registry.fee(address(this)));
insuranceAmt = add(insuranceAmt, insureFeeAmt); feeAmt = add(feeAmt, _newFee);
tokenBalance = sub(_totalToken, insureFeeAmt); tokenBalance = sub(_totalToken, _newFee);
_currentRate = getCurrentRate(tokenBalance); _currentRate = getCurrentRate(tokenBalance);
} }
exchangeRate = _currentRate; exchangeRate = _currentRate;
emit LogExchangeRate(exchangeRate, tokenBalance, insuranceAmt); emit LogExchangeRate(exchangeRate, tokenBalance, feeAmt);
}
/**
* @dev Delegate the calls to Connector And this function is ran by cast().
* @param _target Target to of Connector.
* @param _data CallData of function in Connector.
*/
function spell(address _target, bytes memory _data) internal {
require(_target != address(0), "target-invalid");
assembly {
let succeeded := delegatecall(gas(), _target, add(_data, 0x20), mload(_data), 0, 0)
switch iszero(succeeded)
case 1 {
// throw if delegatecall failed
let size := returndatasize()
returndatacopy(0x00, 0x00, size)
revert(0x00, size)
}
}
} }
/** /**
* @dev Settle the assets on dsa and update exchange rate * @dev Settle the assets on dsa and update exchange rate
* @param _dsa DSA address
* @param _targets array of connector's address * @param _targets array of connector's address
* @param _datas array of connector's function calldata * @param _data array of connector's function calldata
* @param _origin origin address
*/ */
function settle(address _dsa, address[] calldata _targets, bytes[] calldata _datas, address _origin) external isChief { function settle(address[] calldata _targets, bytes[] calldata _data) external isChief {
require(registry.isDsa(address(this), _dsa), "not-autheticated-dsa"); require(_targets.length == _data.length , "array-length-invalid");
AccountInterface dsaWallet = AccountInterface(_dsa); require(registry.checkSettleLogics(address(this), _targets), "not-logic");
if (_targets.length > 0 && _datas.length > 0) { for (uint i = 0; i < _targets.length; i++) {
dsaWallet.cast(_targets, _datas, _origin); spell(_targets[i], _data[i]);
} }
require(dsaWallet.isAuth(address(this)), "token-pool-not-auth");
setExchangeRate(); setExchangeRate();
emit LogSettle(block.number); emit LogSettle(block.number);
} }
@ -185,7 +195,7 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable {
uint _feeAmt; uint _feeAmt;
if (_withdrawalFee > 0) { if (_withdrawalFee > 0) {
_feeAmt = wmul(_tknAmt, _withdrawalFee); _feeAmt = wmul(_tknAmt, _withdrawalFee);
insuranceAmt = add(insuranceAmt, _feeAmt); feeAmt = add(feeAmt, _feeAmt);
_tknAmt = sub(_tknAmt, _feeAmt); _tknAmt = sub(_tknAmt, _feeAmt);
} }
@ -198,10 +208,10 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable {
* @dev Add Insurance to the pool. * @dev Add Insurance to the pool.
* @param tknAmt insurance token amount to add * @param tknAmt insurance token amount to add
*/ */
function addInsurance(uint tknAmt) external { function addFee(uint tknAmt) external {
baseToken.safeTransferFrom(msg.sender, address(this), tknAmt); baseToken.safeTransferFrom(msg.sender, address(this), tknAmt);
insuranceAmt = add(insuranceAmt, tknAmt); feeAmt = add(feeAmt, tknAmt);
emit LogAddInsurance(tknAmt); emit LogAddFee(tknAmt);
} }
/** /**
@ -209,12 +219,12 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable {
* @notice only master can call this function. * @notice only master can call this function.
* @param tknAmt insurance token amount to remove * @param tknAmt insurance token amount to remove
*/ */
function withdrawInsurance(uint tknAmt) external { function withdrawFee(uint tknAmt) external {
require(msg.sender == instaIndex.master(), "not-master"); require(msg.sender == instaIndex.master(), "not-master");
require(tknAmt <= insuranceAmt, "not-enough-insurance"); require(tknAmt <= feeAmt, "not-enough-insurance");
baseToken.safeTransfer(msg.sender, tknAmt); baseToken.safeTransfer(msg.sender, tknAmt);
insuranceAmt = sub(insuranceAmt, tknAmt); feeAmt = sub(feeAmt, tknAmt);
emit LogWithdrawInsurance(tknAmt); emit LogWithdrawFee(tknAmt);
} }
/** /**

View File

@ -21,9 +21,10 @@ interface IndexInterface {
interface RegistryInterface { interface RegistryInterface {
function chief(address) external view returns (bool); function chief(address) external view returns (bool);
function poolLogic(address) external returns (address); function poolLogic(address) external returns (address);
function insureFee(address) external view returns (uint); function fee(address) external view returns (uint);
function withdrawalFee(address) external view returns (uint); function withdrawalFee(address) external view returns (uint);
function isDsa(address, address) external view returns (bool); function isDsa(address, address) external view returns (bool);
function checkSettleLogics(address, address[] calldata) external view returns (bool);
} }
interface RateInterface { interface RateInterface {
@ -38,17 +39,17 @@ contract PoolETH is ReentrancyGuard, ERC20Pausable, DSMath {
event LogSettle(uint settleBlock); event LogSettle(uint settleBlock);
event LogDeposit(address indexed user, uint depositAmt, uint poolMintAmt); event LogDeposit(address indexed user, uint depositAmt, uint poolMintAmt);
event LogWithdraw(address indexed user, uint withdrawAmt, uint poolBurnAmt, uint feeAmt); event LogWithdraw(address indexed user, uint withdrawAmt, uint poolBurnAmt, uint feeAmt);
event LogAddInsurance(uint amount); event LogAddFee(uint amount);
event LogWithdrawInsurance(uint amount); event LogWithdrawFee(uint amount);
event LogPausePool(bool); event LogPausePool(bool);
IERC20 public immutable baseToken; // Base token.
RegistryInterface public immutable registry; // Pool Registry RegistryInterface public immutable registry; // Pool Registry
IndexInterface public constant instaIndex = IndexInterface(0x2971AdFa57b20E5a416aE5a708A8655A9c74f723); IndexInterface public constant instaIndex = IndexInterface(0x2971AdFa57b20E5a416aE5a708A8655A9c74f723);
IERC20 public immutable baseToken; // Base token.
uint private tokenBalance; // total token balance uint private tokenBalance; // total token balance
uint public exchangeRate = 10 ** 18; // initial 1 token = 1 uint public exchangeRate = 10 ** 18; // initial 1 token = 1
uint public insuranceAmt; // insurance amount to keep pool safe uint public feeAmt; // fee collected on profits
constructor( constructor(
address _registry, address _registry,
@ -92,47 +93,56 @@ contract PoolETH is ReentrancyGuard, ERC20Pausable, DSMath {
/** /**
* @dev sets exchange rates * @dev sets exchange rates
*/ */
function setExchangeRate() public isChief { function setExchangeRate() public isChief {
uint _previousRate = exchangeRate; uint _previousRate = exchangeRate;
uint _totalToken = RateInterface(registry.poolLogic(address(this))).getTotalToken(); uint _totalToken = RateInterface(registry.poolLogic(address(this))).getTotalToken();
_totalToken = sub(_totalToken, insuranceAmt); _totalToken = sub(_totalToken, feeAmt);
uint _currentRate = getCurrentRate(_totalToken); uint _currentRate = getCurrentRate(_totalToken);
require(_currentRate != 0, "current-rate-is-zero"); require(_currentRate != 0, "current-rate-is-zero");
if (_currentRate > _previousRate) { // loss => deduct partially/fully from insurance amount if (_currentRate > _previousRate) { // loss => deduct partially/fully from insurance amount
uint _loss = sub(tokenBalance, _totalToken);
if (_loss <= insuranceAmt) {
insuranceAmt = sub(insuranceAmt, _loss);
_currentRate = _previousRate; _currentRate = _previousRate;
} else {
tokenBalance = add(_totalToken, insuranceAmt);
insuranceAmt = 0;
_currentRate = getCurrentRate(tokenBalance);
}
} else { // profit => add to insurance amount } else { // profit => add to insurance amount
uint insureFeeAmt = wmul(sub(_totalToken, tokenBalance), registry.insureFee(address(this))); uint _newFee = wmul(sub(_totalToken, tokenBalance), registry.fee(address(this)));
insuranceAmt = add(insuranceAmt, insureFeeAmt); feeAmt = add(feeAmt, _newFee);
tokenBalance = sub(_totalToken, insureFeeAmt); tokenBalance = sub(_totalToken, _newFee);
_currentRate = getCurrentRate(tokenBalance); _currentRate = getCurrentRate(tokenBalance);
} }
exchangeRate = _currentRate; exchangeRate = _currentRate;
emit LogExchangeRate(exchangeRate, tokenBalance, insuranceAmt); emit LogExchangeRate(exchangeRate, tokenBalance, feeAmt);
}
/**
* @dev Delegate the calls to Connector And this function is ran by cast().
* @param _target Target to of Connector.
* @param _data CallData of function in Connector.
*/
function spell(address _target, bytes memory _data) internal {
require(_target != address(0), "target-invalid");
assembly {
let succeeded := delegatecall(gas(), _target, add(_data, 0x20), mload(_data), 0, 0)
switch iszero(succeeded)
case 1 {
// throw if delegatecall failed
let size := returndatasize()
returndatacopy(0x00, 0x00, size)
revert(0x00, size)
}
}
} }
/** /**
* @dev Settle the assets on dsa and update exchange rate * @dev Settle the assets on dsa and update exchange rate
* @param _dsa DSA address
* @param _targets array of connector's address * @param _targets array of connector's address
* @param _datas array of connector's function calldata * @param _data array of connector's function calldata
* @param _origin origin address
*/ */
function settle(address _dsa, address[] calldata _targets, bytes[] calldata _datas, address _origin) external isChief { function settle(address[] calldata _targets, bytes[] calldata _data) external isChief {
require(registry.isDsa(address(this), _dsa), "not-autheticated-dsa"); require(_targets.length == _data.length , "array-length-invalid");
AccountInterface dsaWallet = AccountInterface(_dsa); require(registry.checkSettleLogics(address(this), _targets), "not-logic");
if (_targets.length > 0 && _datas.length > 0) { for (uint i = 0; i < _targets.length; i++) {
dsaWallet.cast(_targets, _datas, _origin); spell(_targets[i], _data[i]);
} }
require(dsaWallet.isAuth(address(this)), "token-pool-not-auth");
setExchangeRate(); setExchangeRate();
emit LogSettle(block.number); emit LogSettle(block.number);
} }
@ -181,7 +191,7 @@ contract PoolETH is ReentrancyGuard, ERC20Pausable, DSMath {
uint _feeAmt; uint _feeAmt;
if (_withdrawalFee > 0) { if (_withdrawalFee > 0) {
_feeAmt = wmul(_tknAmt, _withdrawalFee); _feeAmt = wmul(_tknAmt, _withdrawalFee);
insuranceAmt = add(insuranceAmt, _feeAmt); feeAmt = add(feeAmt, _feeAmt);
_tknAmt = sub(_tknAmt, _feeAmt); _tknAmt = sub(_tknAmt, _feeAmt);
} }
@ -194,10 +204,10 @@ contract PoolETH is ReentrancyGuard, ERC20Pausable, DSMath {
* @dev Add Insurance to the pool. * @dev Add Insurance to the pool.
* @param tknAmt insurance token amount to add * @param tknAmt insurance token amount to add
*/ */
function addInsurance(uint tknAmt) external payable { function addFee(uint tknAmt) external payable {
require(tknAmt == msg.value, "unmatched-amount"); require(tknAmt == msg.value, "unmatched-amount");
insuranceAmt = add(insuranceAmt, tknAmt); feeAmt = add(feeAmt, tknAmt);
emit LogAddInsurance(tknAmt); emit LogAddFee(tknAmt);
} }
/** /**
@ -205,12 +215,12 @@ contract PoolETH is ReentrancyGuard, ERC20Pausable, DSMath {
* @notice only master can call this function. * @notice only master can call this function.
* @param tknAmt insurance token amount to remove * @param tknAmt insurance token amount to remove
*/ */
function withdrawInsurance(uint tknAmt) external { function withdrawFee(uint tknAmt) external {
require(msg.sender == instaIndex.master(), "not-master"); require(msg.sender == instaIndex.master(), "not-master");
require(tknAmt <= insuranceAmt, "not-enough-insurance"); require(tknAmt <= feeAmt, "not-enough-insurance");
msg.sender.transfer(tknAmt); msg.sender.transfer(tknAmt);
insuranceAmt = sub(insuranceAmt, tknAmt); feeAmt = sub(feeAmt, tknAmt);
emit LogWithdrawInsurance(tknAmt); emit LogWithdrawFee(tknAmt);
} }
/** /**

View File

@ -16,12 +16,14 @@ contract Registry {
event LogRemoveSigner(address indexed signer); event LogRemoveSigner(address indexed signer);
event LogUpdatePool(address pool, bool poolState); event LogUpdatePool(address pool, bool poolState);
event LogUpdatePoolLogic(address pool, address newLogic); event LogUpdatePoolLogic(address pool, address newLogic);
event LogUpdateInsureFee(address pool, uint newFee); event LogUpdateFee(address pool, uint newFee);
event LogUpdateWithdrawalFee(address pool, uint newFee); event LogUpdateWithdrawalFee(address pool, uint newFee);
event LogAddPool(address indexed token, address indexed pool); event LogAddPool(address indexed token, address indexed pool);
event LogRemovePool(address indexed token, address indexed pool); event LogRemovePool(address indexed token, address indexed pool);
event LogNewDSA(address indexed pool, address indexed dsa); event LogNewDSA(address indexed pool, address indexed dsa);
event LogRemoveDSA(address indexed pool, address indexed dsa); event LogRemoveDSA(address indexed pool, address indexed dsa);
event LogAddSettleLogic(address indexed pool, address indexed logic);
event LogRemoveSettleLogic(address indexed pool, address indexed logic);
IndexInterface public constant instaIndex = IndexInterface(0x2971AdFa57b20E5a416aE5a708A8655A9c74f723); IndexInterface public constant instaIndex = IndexInterface(0x2971AdFa57b20E5a416aE5a708A8655A9c74f723);
@ -31,8 +33,9 @@ contract Registry {
mapping (address => address) public poolToken; mapping (address => address) public poolToken;
mapping (address => address) public poolLogic; mapping (address => address) public poolLogic;
mapping (address => uint) public poolCap; mapping (address => uint) public poolCap;
mapping (address => uint) public insureFee; mapping (address => uint) public fee;
mapping (address => uint) public withdrawalFee; mapping (address => uint) public withdrawalFee;
mapping (address => mapping(address => bool)) public settleLogic;
mapping (address => mapping(address => bool)) public isDsa; // pool => dsa => bool mapping (address => mapping(address => bool)) public isDsa; // pool => dsa => bool
mapping (address => address[]) public dsaArr; // pool => all dsa in array mapping (address => address[]) public dsaArr; // pool => all dsa in array
@ -145,12 +148,12 @@ contract Registry {
* @param _pool pool address * @param _pool pool address
* @param _newFee new fee amount * @param _newFee new fee amount
*/ */
function updateInsureFee(address _pool, uint _newFee) external isMaster { function updateFee(address _pool, uint _newFee) external isMaster {
require(isPool[_pool], "not-pool"); require(isPool[_pool], "not-pool");
require(_newFee < 10 ** 18, "insure-fee-limit-reached"); require(_newFee < 3 * 10 ** 17, "insure-fee-limit-reached");
require(insureFee[_pool] != _newFee, "same-pool-fee"); require(fee[_pool] != _newFee, "same-pool-fee");
insureFee[_pool] = _newFee; fee[_pool] = _newFee;
emit LogUpdateInsureFee(_pool, _newFee); emit LogUpdateFee(_pool, _newFee);
} }
/** /**
@ -166,6 +169,27 @@ contract Registry {
emit LogUpdateWithdrawalFee(_pool, _newFee); emit LogUpdateWithdrawalFee(_pool, _newFee);
} }
function addSettleLogic(address _pool, address _logic) external isMaster {
require(isPool[_pool], "not-pool");
settleLogic[_pool][_logic] = true;
emit LogAddSettleLogic(_pool, _logic);
}
function removeSettleLogic(address _pool, address _logic) external isMaster {
require(isPool[_pool], "not-pool");
delete settleLogic[_pool][_logic];
emit LogRemoveSettleLogic(_pool, _logic);
}
function checkSettleLogics(address _pool, address[] calldata _logics) external view returns(bool) {
for (uint i = 0; i < _logics.length; i++) {
if (!settleLogic[_pool][_logics[i]]) {
return false;
}
}
return true;
}
/** /**
* @dev add dsa for a pool * @dev add dsa for a pool
* @param _pool pool address * @param _pool pool address