47 lines
966 B
Solidity
47 lines
966 B
Solidity
pragma solidity ^0.4.24;
|
|
|
|
import "../ReentrancyGuard.sol";
|
|
import "./ReentrancyAttack.sol";
|
|
|
|
|
|
contract ReentrancyMock is ReentrancyGuard {
|
|
|
|
uint256 public counter;
|
|
|
|
constructor() public {
|
|
counter = 0;
|
|
}
|
|
|
|
function callback() external nonReentrant {
|
|
count();
|
|
}
|
|
|
|
function countLocalRecursive(uint256 n) public nonReentrant {
|
|
if (n > 0) {
|
|
count();
|
|
countLocalRecursive(n - 1);
|
|
}
|
|
}
|
|
|
|
function countThisRecursive(uint256 n) public nonReentrant {
|
|
bytes4 func = bytes4(keccak256("countThisRecursive(uint256)"));
|
|
if (n > 0) {
|
|
count();
|
|
// solium-disable-next-line security/no-low-level-calls
|
|
bool result = address(this).call(func, n - 1);
|
|
require(result == true);
|
|
}
|
|
}
|
|
|
|
function countAndCall(ReentrancyAttack attacker) public nonReentrant {
|
|
count();
|
|
bytes4 func = bytes4(keccak256("callback()"));
|
|
attacker.callSender(func);
|
|
}
|
|
|
|
function count() private {
|
|
counter += 1;
|
|
}
|
|
|
|
}
|