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

update

Browse Source
This commit is contained in:
owen05
2021-06-28 23:51:34 +08:00
parent d02b9b0eb8
commit 7acd630e50
7 changed files with 22 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -173,7 +173,7 @@ contract BaseMine is InitializableOwnable {
rt.rewardVault = address(new RewardVault(rewardToken));
uint256 rewardAmount = rewardPerBlock.mul(endBlock.sub(startBlock));
IERC20(rewardToken).transfer(rt.rewardVault, rewardAmount);
IERC20(rewardToken).safeTransfer(rt.rewardVault, rewardAmount);
RewardVault(rt.rewardVault).syncValue();
emit NewRewardToken(len, rewardToken);

7
contracts/DODOToken/DODOMineV3/ERC20Mine.sol
View File

@@ -10,9 +10,10 @@ pragma experimental ABIEncoderV2;
import {SafeERC20} from "../../lib/SafeERC20.sol";
import {IERC20} from "../../intf/IERC20.sol";
import {SafeMath} from "../../lib/SafeMath.sol";
import {ReentrancyGuard} from "../../lib/ReentrancyGuard.sol";
import {BaseMine} from "./BaseMine.sol";
contract ERC20Mine is BaseMine {
contract ERC20Mine is ReentrancyGuard, BaseMine {
using SafeERC20 for IERC20;
using SafeMath for uint256;
@@ -32,7 +33,7 @@ contract ERC20Mine is BaseMine {
// ============ Deposit && Withdraw && Exit ============
function deposit(uint256 amount) external {
function deposit(uint256 amount) external preventReentrant {
require(amount > 0, "DODOMineV3: CANNOT_DEPOSIT_ZERO");
_updateAllReward(msg.sender);
@@ -47,7 +48,7 @@ contract ERC20Mine is BaseMine {
emit Deposit(msg.sender, actualStakeAmount);
}
function withdraw(uint256 amount) external {
function withdraw(uint256 amount) external preventReentrant {
require(amount > 0, "DODOMineV3: CANNOT_WITHDRAW_ZERO");
_updateAllReward(msg.sender);