use safeTransfer
This commit is contained in:
@ -1,6 +1,7 @@
|
||||
pragma solidity ^0.4.11;
|
||||
|
||||
import './ERC20Basic.sol';
|
||||
import './SafeERC20.sol';
|
||||
import '../ownership/Ownable.sol';
|
||||
import '../math/Math.sol';
|
||||
import '../math/SafeMath.sol';
|
||||
@ -13,6 +14,7 @@ import '../math/SafeMath.sol';
|
||||
*/
|
||||
contract TokenVesting is Ownable {
|
||||
using SafeMath for uint256;
|
||||
using SafeERC20 for ERC20Basic;
|
||||
|
||||
event Released(uint256 amount);
|
||||
event Revoked();
|
||||
@ -58,7 +60,7 @@ contract TokenVesting is Ownable {
|
||||
|
||||
require(vested > 0);
|
||||
|
||||
token.transfer(beneficiary, vested);
|
||||
token.safeTransfer(beneficiary, vested);
|
||||
|
||||
released[token] = released[token].add(vested);
|
||||
|
||||
@ -81,7 +83,7 @@ contract TokenVesting is Ownable {
|
||||
|
||||
revoked[token] = true;
|
||||
|
||||
token.transfer(owner, vesting);
|
||||
token.safeTransfer(owner, vesting);
|
||||
|
||||
Revoked();
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user