add check on balance

simonesestito · Jan 5, 2024 · 0f7a5be · 0f7a5be
1 parent 3d2fd66
commit 0f7a5be
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/smart_contract/Geth.sol b/smart_contract/Geth.sol
@@ -40,6 +40,7 @@ contract Geth is ERC20, Ownable {
      * @dev function to buy Ether with tokens and transfer them to the buyer.
      */
     function purchaseWei(uint256 tokenAmount) external {
+        require(balanceOf(msg.sender) >= tokenAmount);
         uint256 WeiAmount = calculateWeiAmount(tokenAmount);