extracted redeem validation utils

This commit is contained in:
Michele Balistreri 2020-04-21 13:39:40 +03:00
parent 3dfa0a7fb7
commit e34f1a8e99
No known key found for this signature in database
GPG Key ID: E9567DA33A4F791A
3 changed files with 98 additions and 135 deletions

View File

@ -2,6 +2,7 @@ pragma solidity ^0.6.1;
pragma experimental ABIEncoderV2;
import "./erc20/IERC20.sol";
import "./RedeemUtil.sol";
contract GiftBucket {
@ -23,17 +24,9 @@ contract GiftBucket {
mapping(address => Gift) public gifts;
struct Redeem {
uint256 blockNumber;
bytes32 blockHash;
address receiver;
bytes32 code;
}
uint256 public redeemableSupply;
bytes32 constant EIP712DOMAIN_TYPEHASH = keccak256("EIP712Domain(string name,string version,uint256 chainId,address verifyingContract)");
bytes32 constant REDEEM_TYPEHASH = keccak256("Redeem(uint256 blockNumber,bytes32 blockHash,address receiver,bytes32 code)");
bytes32 DOMAIN_SEPARATOR;
modifier onlyOwner() {
@ -48,7 +41,7 @@ contract GiftBucket {
function initialize(address _tokenAddress, uint256 _expirationTime, address _owner) public {
require(initialized == false, "already initialized");
require(_expirationTime > block.timestamp, "expiration can't be in the past");
RedeemUtil.validateExpiryDate(_expirationTime);
tokenContract = IERC20(_tokenAddress);
expirationTime = _expirationTime;
@ -58,22 +51,13 @@ contract GiftBucket {
EIP712DOMAIN_TYPEHASH,
keccak256("KeycardGift"),
keccak256("1"),
_getChainID(),
RedeemUtil.getChainID(),
address(this)
));
initialized = true;
}
function _getChainID() internal pure returns (uint256) {
uint256 id;
assembly {
id := chainid()
}
return id;
}
function totalSupply() public view returns(uint256) {
return tokenContract.balanceOf(address(this));
}
@ -102,20 +86,15 @@ contract GiftBucket {
redeemableSupply += amount;
}
function redeem(Redeem calldata _redeem, bytes calldata sig) external {
require(_redeem.blockNumber < block.number, "transaction cannot be in the future");
require(_redeem.blockNumber >= (block.number - maxTxDelayInBlocks), "transaction too old");
require(_redeem.blockHash == blockhash(_redeem.blockNumber), "invalid block hash");
function redeem(RedeemUtil.Redeem calldata _redeem, bytes calldata _sig) external {
RedeemUtil.validateRedeem(_redeem, maxTxDelayInBlocks, expirationTime, 0);
require(block.timestamp < expirationTime, "expired gift");
address recipient = recoverSigner(_redeem, sig);
address recipient = RedeemUtil.recoverSigner(DOMAIN_SEPARATOR, _redeem, _sig);
Gift storage gift = gifts[recipient];
require(gift.recipient == recipient, "not found");
bytes32 codeHash = keccak256(abi.encodePacked(_redeem.code));
require(codeHash == gift.code, "invalid code");
RedeemUtil.validateCode(_redeem, gift.code);
uint256 amount = gift.amount;
require(redeemableSupply >= amount, "not enough redeemable supply");
@ -127,49 +106,11 @@ contract GiftBucket {
}
function kill() external onlyOwner {
require(block.timestamp >= expirationTime, "not expired yet");
RedeemUtil.validateExpired(expirationTime);
bool success = tokenContract.transfer(owner, this.totalSupply());
assert(success);
selfdestruct(owner);
}
function hashRedeem(Redeem memory _redeem) internal pure returns (bytes32) {
return keccak256(abi.encode(
REDEEM_TYPEHASH,
_redeem.blockNumber,
_redeem.blockHash,
_redeem.receiver,
_redeem.code
));
}
function recoverSigner(Redeem memory _redeem, bytes memory sig) internal view returns(address) {
require(sig.length == 65, "bad signature length");
bytes32 r;
bytes32 s;
uint8 v;
assembly {
r := mload(add(sig, 32))
s := mload(add(sig, 64))
v := byte(0, mload(add(sig, 96)))
}
if (v < 27) {
v += 27;
}
require(v == 27 || v == 28, "signature version doesn't match");
bytes32 digest = keccak256(abi.encodePacked(
"\x19\x01",
DOMAIN_SEPARATOR,
hashRedeem(_redeem)
));
return ecrecover(digest, v, r, s);
}
}

View File

@ -4,9 +4,9 @@ pragma experimental ABIEncoderV2;
import "./erc721/IERC721.sol";
import "./erc721/IERC721Receiver.sol";
import "./erc721/IERC165.sol";
import "./RedeemUtil.sol";
contract NFTBucket is IERC165, IERC721Receiver {
bool initialized;
address payable public owner;
@ -25,17 +25,9 @@ contract NFTBucket is IERC165, IERC721Receiver {
mapping(address => Gift) public gifts;
struct Redeem {
uint256 blockNumber;
bytes32 blockHash;
address receiver;
bytes32 code;
}
bytes4 private constant _ERC721_RECEIVED = 0x150b7a02; //bytes4(keccak256("onERC721Received(address,address,uint256,bytes)"))
bytes32 constant EIP712DOMAIN_TYPEHASH = keccak256("EIP712Domain(string name,string version,uint256 chainId,address verifyingContract)");
bytes32 constant REDEEM_TYPEHASH = keccak256("Redeem(uint256 blockNumber,bytes32 blockHash,address receiver,bytes32 code)");
bytes32 DOMAIN_SEPARATOR;
modifier onlyOwner() {
@ -50,7 +42,7 @@ contract NFTBucket is IERC165, IERC721Receiver {
function initialize(address _tokenAddress, uint256 _expirationTime, address _owner) public {
require(initialized == false, "already initialized");
require(_expirationTime > block.timestamp, "expiration can't be in the past");
RedeemUtil.validateExpiryDate(_expirationTime);
tokenContract = IERC721(_tokenAddress);
expirationTime = _expirationTime;
@ -60,42 +52,28 @@ contract NFTBucket is IERC165, IERC721Receiver {
EIP712DOMAIN_TYPEHASH,
keccak256("KeycardNFTGift"),
keccak256("1"),
_getChainID(),
RedeemUtil.getChainID(),
address(this)
));
initialized = true;
}
function _getChainID() internal pure returns (uint256) {
uint256 id;
assembly {
id := chainid()
}
function redeem(RedeemUtil.Redeem calldata _redeem, bytes calldata _sig) external {
RedeemUtil.validateRedeem(_redeem, maxTxDelayInBlocks, expirationTime, 0);
return id;
}
function redeem(Redeem calldata _redeem, bytes calldata sig) external {
require(_redeem.blockNumber < block.number, "transaction cannot be in the future");
require(_redeem.blockNumber >= (block.number - maxTxDelayInBlocks), "transaction too old");
require(_redeem.blockHash == blockhash(_redeem.blockNumber), "invalid block hash");
require(block.timestamp < expirationTime, "expired gift");
address recipient = recoverSigner(_redeem, sig);
address recipient = RedeemUtil.recoverSigner(DOMAIN_SEPARATOR, _redeem, _sig);
Gift storage gift = gifts[recipient];
require(gift.recipient == recipient, "not found");
bytes32 codeHash = keccak256(abi.encodePacked(_redeem.code));
require(codeHash == gift.code, "invalid code");
RedeemUtil.validateCode(_redeem, gift.code);
tokenContract.safeTransferFrom(address(this), _redeem.receiver, gift.tokenID);
}
function kill() external onlyOwner {
require(block.timestamp >= expirationTime, "not expired yet");
RedeemUtil.validateExpired(expirationTime);
tokenContract.setApprovalForAll(owner, true);
assert(tokenContract.isApprovedForAll(address(this), owner));
@ -103,44 +81,6 @@ contract NFTBucket is IERC165, IERC721Receiver {
selfdestruct(owner);
}
function hashRedeem(Redeem memory _redeem) internal pure returns (bytes32) {
return keccak256(abi.encode(
REDEEM_TYPEHASH,
_redeem.blockNumber,
_redeem.blockHash,
_redeem.receiver,
_redeem.code
));
}
function recoverSigner(Redeem memory _redeem, bytes memory sig) internal view returns(address) {
require(sig.length == 65, "bad signature length");
bytes32 r;
bytes32 s;
uint8 v;
assembly {
r := mload(add(sig, 32))
s := mload(add(sig, 64))
v := byte(0, mload(add(sig, 96)))
}
if (v < 27) {
v += 27;
}
require(v == 27 || v == 28, "signature version doesn't match");
bytes32 digest = keccak256(abi.encodePacked(
"\x19\x01",
DOMAIN_SEPARATOR,
hashRedeem(_redeem)
));
return ecrecover(digest, v, r, s);
}
function supportsInterface(bytes4 interfaceID) external override(IERC165) view returns (bool) {
return interfaceID == _ERC721_RECEIVED;
}

82
contracts/RedeemUtil.sol Normal file
View File

@ -0,0 +1,82 @@
pragma solidity ^0.6.1;
pragma experimental ABIEncoderV2;
library RedeemUtil {
struct Redeem {
uint256 blockNumber;
bytes32 blockHash;
address receiver;
bytes32 code;
}
bytes32 constant REDEEM_TYPEHASH = keccak256("Redeem(uint256 blockNumber,bytes32 blockHash,address receiver,bytes32 code)");
function getChainID() internal pure returns (uint256) {
uint256 id;
assembly {
id := chainid()
}
return id;
}
function validateExpiryDate(uint256 _expirationTime) internal view {
require(_expirationTime > block.timestamp, "expiration can't be in the past");
}
function validateExpired(uint256 _expirationTime) internal view {
require(block.timestamp >= _expirationTime, "not expired yet");
}
function validateRedeem(Redeem memory _redeem, uint256 _maxTxDelayInBlocks, uint256 _expirationTime, uint256 _startTime) internal view {
require(_redeem.blockNumber < block.number, "transaction cannot be in the future");
require(_redeem.blockNumber >= (block.number - _maxTxDelayInBlocks), "transaction too old");
require(_redeem.blockHash == blockhash(_redeem.blockNumber), "invalid block hash");
require(block.timestamp < _expirationTime, "expired gift");
require(block.timestamp > _startTime, "reedeming not yet started");
}
function hashRedeem(Redeem memory _redeem) internal pure returns (bytes32) {
return keccak256(abi.encode(
REDEEM_TYPEHASH,
_redeem.blockNumber,
_redeem.blockHash,
_redeem.receiver,
_redeem.code
));
}
function recoverSigner(bytes32 _domainSeparator, Redeem memory _redeem, bytes memory _sig) internal pure returns(address) {
require(_sig.length == 65, "bad signature length");
bytes32 r;
bytes32 s;
uint8 v;
assembly {
r := mload(add(_sig, 32))
s := mload(add(_sig, 64))
v := byte(0, mload(add(_sig, 96)))
}
if (v < 27) {
v += 27;
}
require(v == 27 || v == 28, "signature version doesn't match");
bytes32 digest = keccak256(abi.encodePacked(
"\x19\x01",
_domainSeparator,
hashRedeem(_redeem)
));
return ecrecover(digest, v, r, s);
}
function validateCode(Redeem memory _redeem, bytes32 _code) internal pure {
bytes32 codeHash = keccak256(abi.encodePacked(_redeem.code));
require(codeHash == _code, "invalid code");
}
}