Enhance StakingAuRa.setErc20TokenContract function

it shouldn't let set token contract in which the balance of StakingAuRa contract is already non-zero
poanetwork · Aug 16, 2019 · 2f3449b · 2f3449b
1 parent 2c3c208
commit 2f3449b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/contracts/StakingAuRa.sol b/contracts/StakingAuRa.sol
@@ -524,6 +524,7 @@ contract StakingAuRa is UpgradeableOwned, IStakingAuRa {
     /// @param _erc20TokenContract The address of the contract.
     function setErc20TokenContract(IERC20Minting _erc20TokenContract) external onlyOwner onlyInitialized {
         require(_erc20TokenContract != IERC20Minting(0));
+        require(_erc20TokenContract.balanceOf(address(this)) == 0);
         require(!erc20Restricted);
         erc20TokenContract = _erc20TokenContract;
     }

diff --git a/docs/website/sidebars.json b/docs/website/sidebars.json
@@ -5,7 +5,7 @@
       "upgradeability_BaseAdminUpgradeabilityProxy",
       "upgradeability_BaseUpgradeabilityProxy",
       "upgradeability_Proxy",
-      "upgradeability_UpgradeabilityAdminSlot",
+      "upgradeability_UpgradeabilityAdmin",
       "upgradeability_UpgradeabilityProxy",
       "upgradeability_UpgradeableOwned"
     ],

diff --git a/test/mockContracts/ERC677BridgeTokenRewardableMock.sol b/test/mockContracts/ERC677BridgeTokenRewardableMock.sol
@@ -592,6 +592,7 @@ contract ERC677BridgeTokenRewardable is ERC677BridgeToken {
 
   function setStakingContract(address _stakingContract) onlyOwner public {
     require(_stakingContract != address(0) && isContract(_stakingContract));
+    require(balanceOf(_stakingContract) == 0);
     stakingContract = _stakingContract;
   }