diff options
author | fragosti <francesco.agosti93@gmail.com> | 2018-07-26 01:33:13 +0800 |
---|---|---|
committer | fragosti <francesco.agosti93@gmail.com> | 2018-07-26 01:33:13 +0800 |
commit | da4b70fcae1da2c35905a30b79707b3f27b9fa3a (patch) | |
tree | 65dbdb8f75c8e93cd788149610522f5bf5ada643 /packages/contracts/src/2.0.0/forwarder/MixinAssets.sol | |
parent | ad4c2b95e4722e6d227c26f436bedf7b06f5356a (diff) | |
parent | 05b35c0fdcbca7980d4195e96ec791c1c2d13398 (diff) | |
download | dexon-sol-tools-da4b70fcae1da2c35905a30b79707b3f27b9fa3a.tar dexon-sol-tools-da4b70fcae1da2c35905a30b79707b3f27b9fa3a.tar.gz dexon-sol-tools-da4b70fcae1da2c35905a30b79707b3f27b9fa3a.tar.bz2 dexon-sol-tools-da4b70fcae1da2c35905a30b79707b3f27b9fa3a.tar.lz dexon-sol-tools-da4b70fcae1da2c35905a30b79707b3f27b9fa3a.tar.xz dexon-sol-tools-da4b70fcae1da2c35905a30b79707b3f27b9fa3a.tar.zst dexon-sol-tools-da4b70fcae1da2c35905a30b79707b3f27b9fa3a.zip |
Merge branch 'v2-prototype' of https://github.com/0xProject/0x-monorepo into feature/web3-wrapper/unexport-marshaller
Diffstat (limited to 'packages/contracts/src/2.0.0/forwarder/MixinAssets.sol')
-rw-r--r-- | packages/contracts/src/2.0.0/forwarder/MixinAssets.sol | 147 |
1 files changed, 147 insertions, 0 deletions
diff --git a/packages/contracts/src/2.0.0/forwarder/MixinAssets.sol b/packages/contracts/src/2.0.0/forwarder/MixinAssets.sol new file mode 100644 index 000000000..5cf5f831b --- /dev/null +++ b/packages/contracts/src/2.0.0/forwarder/MixinAssets.sol @@ -0,0 +1,147 @@ +/* + + Copyright 2018 ZeroEx Intl. + + Licensed under the Apache License, Version 2.0 (the "License"); + you may not use this file except in compliance with the License. + You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + + Unless required by applicable law or agreed to in writing, software + distributed under the License is distributed on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + See the License for the specific language governing permissions and + limitations under the License. + +*/ + +pragma solidity 0.4.24; + +import "../utils/LibBytes/LibBytes.sol"; +import "../utils/Ownable/Ownable.sol"; +import "../tokens/ERC20Token/IERC20Token.sol"; +import "../tokens/ERC721Token/IERC721Token.sol"; +import "./libs/LibConstants.sol"; +import "./mixins/MAssets.sol"; + + +contract MixinAssets is + Ownable, + LibConstants, + MAssets +{ + + using LibBytes for bytes; + + bytes4 constant internal ERC20_TRANSFER_SELECTOR = bytes4(keccak256("transfer(address,uint256)")); + + /// @dev Withdraws ERC20 tokens from this contract. The contract requires a ZRX balance in order to + /// function optimally, and this function allows the ZRX to be withdrawn by owner. It may also be + /// used to withdraw tokens that were accidentally sent to this contract. + /// @param token Address of ERC20 token to withdraw. + /// @param amount Amount of ERC20 token to withdraw. + function withdrawERC20( + address token, + uint256 amount + ) + external + onlyOwner + { + require( + IERC20Token(token).transfer(msg.sender, amount), + "WITHDRAWAL_FAILED" + ); + } + + /// @dev Transfers given amount of asset to sender. + /// @param assetData Byte array encoded for the respective asset proxy. + /// @param amount Amount of asset to transfer to sender. + function transferPurchasedAssetToSender( + bytes memory assetData, + uint256 amount + ) + internal + { + bytes4 proxyId = assetData.readBytes4(0); + + if (proxyId == ERC20_DATA_ID) { + transferERC20Token(assetData, amount); + } else if (proxyId == ERC721_DATA_ID) { + transferERC721Token(assetData, amount); + } else { + revert("UNSUPPORTED_TOKEN_PROXY"); + } + } + + /// @dev Decodes ERC20 assetData and transfers given amount to sender. + /// @param assetData Byte array encoded for the respective asset proxy. + /// @param amount Amount of asset to transfer to sender. + function transferERC20Token( + bytes memory assetData, + uint256 amount + ) + internal + { + address token = assetData.readAddress(16); + + // Transfer tokens. + // We do a raw call so we can check the success separate + // from the return data. + bool success = token.call(abi.encodeWithSelector( + ERC20_TRANSFER_SELECTOR, + msg.sender, + amount + )); + require( + success, + "TRANSFER_FAILED" + ); + + // Check return data. + // If there is no return data, we assume the token incorrectly + // does not return a bool. In this case we expect it to revert + // on failure, which was handled above. + // If the token does return data, we require that it is a single + // value that evaluates to true. + assembly { + if returndatasize { + success := 0 + if eq(returndatasize, 32) { + // First 64 bytes of memory are reserved scratch space + returndatacopy(0, 0, 32) + success := mload(0) + } + } + } + require( + success, + "TRANSFER_FAILED" + ); + } + + /// @dev Decodes ERC721 assetData and transfers given amount to sender. + /// @param assetData Byte array encoded for the respective asset proxy. + /// @param amount Amount of asset to transfer to sender. + function transferERC721Token( + bytes memory assetData, + uint256 amount + ) + internal + { + require( + amount == 1, + "INVALID_AMOUNT" + ); + // Decode asset data. + address token = assetData.readAddress(16); + uint256 tokenId = assetData.readUint256(36); + + // Perform transfer. + IERC721Token(token).transferFrom( + address(this), + msg.sender, + tokenId + ); + } +} |