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

added more require statements

Browse Source
This commit is contained in:
Thrilok Kumar 2020-08-25 19:51:21 +05:30
parent 525d2ab403
commit 79779c9269
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
contracts/registry.sol
View File

@ -24,8 +24,8 @@ contract Registry {
mapping (address => bool) public chief; mapping (address => bool) public chief;
mapping (address => bool) public signer; mapping (address => bool) public signer;
mapping (address => address) public poolToken;
mapping (address => bool) public isPool; mapping (address => bool) public isPool;
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 insureFee;
@ -89,8 +89,9 @@ contract Registry {
* @param token ERC20 token address * @param token ERC20 token address
* @param pool pool address * @param pool pool address
*/ */
function addPool(address token, address pool) external isMaster { // TODO: all good? function addPool(address token, address pool) external isMaster {
require(token != address(0) && pool != address(0), "invalid-address"); require(token != address(0) && pool != address(0), "invalid-address");
require(poolToken[token] == address(0), "pool-already-added");
poolToken[token] = pool; poolToken[token] = pool;
emit LogAddPool(token, pool); emit LogAddPool(token, pool);
} }
@ -99,8 +100,9 @@ contract Registry {
* @dev Remove Pool * @dev Remove Pool
* @param token ERC20 token address * @param token ERC20 token address
*/ */
function removePool(address token) external isMaster { // TODO: all good? function removePool(address token) external isMaster {
require(token != address(0), "invalid-address"); require(token != address(0), "invalid-address");
require(poolToken[token] != address(0), "pool-already-removed");
address poolAddr = poolToken[token]; address poolAddr = poolToken[token];
delete poolToken[token]; delete poolToken[token];
emit LogRemovePool(token, poolAddr); emit LogRemovePool(token, poolAddr);