use safeTransfer

pull/492/head
Francisco Giordano 7 years ago
parent a184013d1e
commit d5e0714faf
  1. 6
      contracts/token/TokenVesting.sol

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

Loading…
Cancel
Save