Support contract withdraw

wittyCodeX · Jul 24, 2018 · 5cecaa6 · 5cecaa6
1 parent 072a3de
commit 5cecaa6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/contracts/CryptoHerosGame.sol b/contracts/CryptoHerosGame.sol
@@ -81,4 +81,10 @@ contract CryptoHerosGame is Ownable {
     return uint(sha3(nonce))%(min+max)-min;
   }
 
+  function withdraw(uint amount) public payable onlyOwner returns(bool) {
+    require(amount <= this.balance);
+    owner.transfer(amount);
+    return true;
+  }
+
 }
diff --git a/contracts/CryptoHerosToken.sol b/contracts/CryptoHerosToken.sol
@@ -89,5 +89,11 @@ contract CryptoHerosToken is ERC721Token, Ownable {
   function getHerosLength() public returns (uint) {
     return heros.length;
   }
+
+  function withdraw(uint amount) public payable onlyOwner returns(bool) {
+    require(amount <= this.balance);
+    owner.transfer(amount);
+    return true;
+  }
 
 }