51 lines
1.3 KiB
Solidity
51 lines
1.3 KiB
Solidity
// SPDX-License-Identifier: MIT
|
|
|
|
pragma solidity ^0.8.24;
|
|
|
|
import {ReentrancyGuardTransient} from "../utils/ReentrancyGuardTransient.sol";
|
|
import {ReentrancyAttack} from "./ReentrancyAttack.sol";
|
|
|
|
contract ReentrancyTransientMock is ReentrancyGuardTransient {
|
|
uint256 public counter;
|
|
|
|
constructor() {
|
|
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 {
|
|
if (n > 0) {
|
|
_count();
|
|
(bool success, ) = address(this).call(abi.encodeCall(this.countThisRecursive, (n - 1)));
|
|
require(success, "ReentrancyTransientMock: failed call");
|
|
}
|
|
}
|
|
|
|
function countAndCall(ReentrancyAttack attacker) public nonReentrant {
|
|
_count();
|
|
attacker.callSender(abi.encodeCall(this.callback, ()));
|
|
}
|
|
|
|
function _count() private {
|
|
counter += 1;
|
|
}
|
|
|
|
function guardedCheckEntered() public nonReentrant {
|
|
require(_reentrancyGuardEntered());
|
|
}
|
|
|
|
function unguardedCheckNotEntered() public view {
|
|
require(!_reentrancyGuardEntered());
|
|
}
|
|
}
|