From dc1a83cf68268a35fd4260cae5ad9268bd69dc94 Mon Sep 17 00:00:00 2001
From: owen05 <liuhongnan53@163.com>
Date: Fri, 2 Apr 2021 15:28:00 +0800
Subject: [PATCH] add check

---
 contracts/DODOToken/DODOMineV2/BaseMine.sol | 2 ++
 1 file changed, 2 insertions(+)

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