NoBey/openzeppelin-contracts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use SafeERC20#safeTransfer

Browse Source
This commit is contained in:
Francisco Giordano
2017-09-07 20:04:48 -03:00
parent dcdc453a55
commit 20187f2970
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
contracts/ownership/CanReclaimToken.sol
View File

@ -2,6 +2,7 @@ pragma solidity ^0.4.11;
import "./Ownable.sol";
import "../token/ERC20Basic.sol";
import "../token/SafeERC20.sol";
/**
* @title Contracts that should be able to recover tokens
@ -10,6 +11,7 @@ import "../token/ERC20Basic.sol";
* This will prevent any accidental loss of tokens.
*/
contract CanReclaimToken is Ownable {
using SafeERC20 for ERC20Basic;
/**
* @dev Reclaim all ERC20Basic compatible tokens
@ -17,7 +19,7 @@ contract CanReclaimToken is Ownable {
*/
function reclaimToken(ERC20Basic token) external onlyOwner {
uint256 balance = token.balanceOf(this);
token.transfer(owner, balance);
token.safeTransfer(owner, balance);
}
}
}

4
contracts/token/TokenTimelock.sol
View File

@ -2,6 +2,7 @@ pragma solidity ^0.4.11;
import './ERC20Basic.sol';
import "../token/SafeERC20.sol";
/**
* @title TokenTimelock
@ -9,6 +10,7 @@ import './ERC20Basic.sol';
* beneficiary to extract the tokens after a given release time
*/
contract TokenTimelock {
using SafeERC20 for ERC20Basic;
// ERC20 basic token contract being held
ERC20Basic token;
@ -44,6 +46,6 @@ contract TokenTimelock {
uint256 amount = token.balanceOf(this);
require(amount > 0);
token.transfer(beneficiary, amount);
token.safeTransfer(beneficiary, amount);
}
}