Commit 20187f29 by Francisco Giordano

use SafeERC20#safeTransfer

parent dcdc453a
...@@ -2,6 +2,7 @@ pragma solidity ^0.4.11; ...@@ -2,6 +2,7 @@ pragma solidity ^0.4.11;
import "./Ownable.sol"; import "./Ownable.sol";
import "../token/ERC20Basic.sol"; import "../token/ERC20Basic.sol";
import "../token/SafeERC20.sol";
/** /**
* @title Contracts that should be able to recover tokens * @title Contracts that should be able to recover tokens
...@@ -10,6 +11,7 @@ import "../token/ERC20Basic.sol"; ...@@ -10,6 +11,7 @@ import "../token/ERC20Basic.sol";
* This will prevent any accidental loss of tokens. * This will prevent any accidental loss of tokens.
*/ */
contract CanReclaimToken is Ownable { contract CanReclaimToken is Ownable {
using SafeERC20 for ERC20Basic;
/** /**
* @dev Reclaim all ERC20Basic compatible tokens * @dev Reclaim all ERC20Basic compatible tokens
...@@ -17,7 +19,7 @@ contract CanReclaimToken is Ownable { ...@@ -17,7 +19,7 @@ contract CanReclaimToken is Ownable {
*/ */
function reclaimToken(ERC20Basic token) external onlyOwner { function reclaimToken(ERC20Basic token) external onlyOwner {
uint256 balance = token.balanceOf(this); uint256 balance = token.balanceOf(this);
token.transfer(owner, balance); token.safeTransfer(owner, balance);
} }
} }
...@@ -2,6 +2,7 @@ pragma solidity ^0.4.11; ...@@ -2,6 +2,7 @@ pragma solidity ^0.4.11;
import './ERC20Basic.sol'; import './ERC20Basic.sol';
import "../token/SafeERC20.sol";
/** /**
* @title TokenTimelock * @title TokenTimelock
...@@ -9,6 +10,7 @@ import './ERC20Basic.sol'; ...@@ -9,6 +10,7 @@ import './ERC20Basic.sol';
* beneficiary to extract the tokens after a given release time * beneficiary to extract the tokens after a given release time
*/ */
contract TokenTimelock { contract TokenTimelock {
using SafeERC20 for ERC20Basic;
// ERC20 basic token contract being held // ERC20 basic token contract being held
ERC20Basic token; ERC20Basic token;
...@@ -44,6 +46,6 @@ contract TokenTimelock { ...@@ -44,6 +46,6 @@ contract TokenTimelock {
uint256 amount = token.balanceOf(this); uint256 amount = token.balanceOf(this);
require(amount > 0); require(amount > 0);
token.transfer(beneficiary, amount); token.safeTransfer(beneficiary, amount);
} }
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment