-
Notifications
You must be signed in to change notification settings - Fork 194
/
NaiveReceiver.t.sol
64 lines (50 loc) · 2.16 KB
/
NaiveReceiver.t.sol
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
// SPDX-License-Identifier: MIT
pragma solidity >=0.8.0;
import {Utilities} from "../../utils/Utilities.sol";
import "forge-std/Test.sol";
import {FlashLoanReceiver} from "../../../src/Contracts/naive-receiver/FlashLoanReceiver.sol";
import {NaiveReceiverLenderPool} from "../../../src/Contracts/naive-receiver/NaiveReceiverLenderPool.sol";
contract NaiveReceiver is Test {
uint256 internal constant ETHER_IN_POOL = 1_000e18;
uint256 internal constant ETHER_IN_RECEIVER = 10e18;
Utilities internal utils;
NaiveReceiverLenderPool internal naiveReceiverLenderPool;
FlashLoanReceiver internal flashLoanReceiver;
address payable internal user;
address payable internal attacker;
function setUp() public {
utils = new Utilities();
address payable[] memory users = utils.createUsers(2);
user = users[0];
attacker = users[1];
vm.label(user, "User");
vm.label(attacker, "Attacker");
naiveReceiverLenderPool = new NaiveReceiverLenderPool();
vm.label(address(naiveReceiverLenderPool), "Naive Receiver Lender Pool");
vm.deal(address(naiveReceiverLenderPool), ETHER_IN_POOL);
assertEq(address(naiveReceiverLenderPool).balance, ETHER_IN_POOL);
assertEq(naiveReceiverLenderPool.fixedFee(), 1e18);
flashLoanReceiver = new FlashLoanReceiver(
payable(naiveReceiverLenderPool)
);
vm.label(address(flashLoanReceiver), "Flash Loan Receiver");
vm.deal(address(flashLoanReceiver), ETHER_IN_RECEIVER);
assertEq(address(flashLoanReceiver).balance, ETHER_IN_RECEIVER);
console.log(unicode"🧨 Let's see if you can break it... 🧨");
}
function testExploit() public {
/**
* EXPLOIT START *
*/
/**
* EXPLOIT END *
*/
validation();
console.log(unicode"\n🎉 Congratulations, you can go to the next level! 🎉");
}
function validation() internal {
// All ETH has been drained from the receiver
assertEq(address(flashLoanReceiver).balance, 0);
assertEq(address(naiveReceiverLenderPool).balance, ETHER_IN_POOL + ETHER_IN_RECEIVER);
}
}