d-bis/dodo-contractV2
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add check

Browse Source
This commit is contained in:
owen05
2021-04-02 15:28:00 +08:00
parent 150b5fdacd
commit dc1a83cf68
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
contracts/DODOToken/DODOMineV2/BaseMine.sol
View File

@@ -49,6 +49,7 @@ contract BaseMine is InitializableOwnable {
// ============ View ============ // ============ View ============
function getPendingReward(address user, uint256 i) public view returns (uint256) { function getPendingReward(address user, uint256 i) public view returns (uint256) {
require(i<rewardTokenInfos.length, "DODOMineV2: REWARD_ID_NOT_FOUND");
RewardTokenInfo storage rt = rewardTokenInfos[i]; RewardTokenInfo storage rt = rewardTokenInfos[i];
uint256 accRewardPerShare = rt.accRewardPerShare; uint256 accRewardPerShare = rt.accRewardPerShare;
if (rt.lastRewardBlock != block.number) { if (rt.lastRewardBlock != block.number) {
@@ -92,6 +93,7 @@ contract BaseMine is InitializableOwnable {
// ============ Claim ============ // ============ Claim ============
function claimReward(uint256 i) public { function claimReward(uint256 i) public {
require(i<rewardTokenInfos.length, "DODOMineV2: REWARD_ID_NOT_FOUND");
_updateReward(msg.sender, i); _updateReward(msg.sender, i);
RewardTokenInfo storage rt = rewardTokenInfos[i]; RewardTokenInfo storage rt = rewardTokenInfos[i];
uint256 reward = rt.userRewards[msg.sender]; uint256 reward = rt.userRewards[msg.sender];