Browse Source

Addressed unchecked return value in AllowanceCrowdsale #945 using SafeERC20 (#946)

* addressed unchecked return value in AllowanceCrowdsale #945 using SafeERC20
Doug Crescenzi 7 years ago
parent
commit
39385f9f4e
1 changed files with 4 additions and 3 deletions
  1. 4 3
      contracts/crowdsale/emission/AllowanceCrowdsale.sol

+ 4 - 3
contracts/crowdsale/emission/AllowanceCrowdsale.sol

@@ -2,15 +2,16 @@ pragma solidity ^0.4.23;
 
 import "../Crowdsale.sol";
 import "../../token/ERC20/ERC20.sol";
+import "../../token/ERC20/ERC20Basic.sol";
+import "../../token/ERC20/SafeERC20.sol";
 import "../../math/SafeMath.sol";
-
-
 /**
  * @title AllowanceCrowdsale
  * @dev Extension of Crowdsale where tokens are held by a wallet, which approves an allowance to the crowdsale.
  */
 contract AllowanceCrowdsale is Crowdsale {
   using SafeMath for uint256;
+  using SafeERC20 for ERC20;
 
   address public tokenWallet;
 
@@ -42,6 +43,6 @@ contract AllowanceCrowdsale is Crowdsale {
   )
     internal
   {
-    token.transferFrom(tokenWallet, _beneficiary, _tokenAmount);
+    token.safeTransferFrom(tokenWallet, _beneficiary, _tokenAmount);
   }
 }