diff options
Diffstat (limited to 'test/compilationTests/zeppelin/payment')
-rw-r--r-- | test/compilationTests/zeppelin/payment/PullPayment.sol | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/test/compilationTests/zeppelin/payment/PullPayment.sol b/test/compilationTests/zeppelin/payment/PullPayment.sol new file mode 100644 index 00000000..ba710b53 --- /dev/null +++ b/test/compilationTests/zeppelin/payment/PullPayment.sol @@ -0,0 +1,50 @@ +pragma solidity ^0.4.11; + + +import '../math/SafeMath.sol'; + + +/** + * @title PullPayment + * @dev Base contract supporting async send for pull payments. Inherit from this + * contract and use asyncSend instead of send. + */ +contract PullPayment { + using SafeMath for uint256; + + mapping(address => uint256) public payments; + uint256 public totalPayments; + + /** + * @dev Called by the payer to store the sent amount as credit to be pulled. + * @param dest The destination address of the funds. + * @param amount The amount to transfer. + */ + function asyncSend(address dest, uint256 amount) internal { + payments[dest] = payments[dest].add(amount); + totalPayments = totalPayments.add(amount); + } + + /** + * @dev withdraw accumulated balance, called by payee. + */ + function withdrawPayments() { + address payee = msg.sender; + uint256 payment = payments[payee]; + + if (payment == 0) { + throw; + } + + if (this.balance < payment) { + throw; + } + + totalPayments = totalPayments.sub(payment); + payments[payee] = 0; + + if (!payee.send(payment)) { + throw; + } + } +} |