12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061 |
- // SPDX-License-Identifier: MIT
- // OpenZeppelin Contracts (last updated v5.3.0) (utils/ReentrancyGuardTransient.sol)
- pragma solidity ^0.8.24;
- import {TransientSlot} from "./TransientSlot.sol";
- /**
- * @dev Variant of {ReentrancyGuard} that uses transient storage.
- *
- * NOTE: This variant only works on networks where EIP-1153 is available.
- *
- * _Available since v5.1._
- */
- abstract contract ReentrancyGuardTransient {
- using TransientSlot for *;
- // keccak256(abi.encode(uint256(keccak256("openzeppelin.storage.ReentrancyGuard")) - 1)) & ~bytes32(uint256(0xff))
- bytes32 private constant REENTRANCY_GUARD_STORAGE =
- 0x9b779b17422d0df92223018b32b4d1fa46e071723d6817e2486d003becc55f00;
- /**
- * @dev Unauthorized reentrant call.
- */
- error ReentrancyGuardReentrantCall();
- /**
- * @dev Prevents a contract from calling itself, directly or indirectly.
- * Calling a `nonReentrant` function from another `nonReentrant`
- * function is not supported. It is possible to prevent this from happening
- * by making the `nonReentrant` function external, and making it call a
- * `private` function that does the actual work.
- */
- modifier nonReentrant() {
- _nonReentrantBefore();
- _;
- _nonReentrantAfter();
- }
- function _nonReentrantBefore() private {
- // On the first call to nonReentrant, REENTRANCY_GUARD_STORAGE.asBoolean().tload() will be false
- if (_reentrancyGuardEntered()) {
- revert ReentrancyGuardReentrantCall();
- }
- // Any calls to nonReentrant after this point will fail
- REENTRANCY_GUARD_STORAGE.asBoolean().tstore(true);
- }
- function _nonReentrantAfter() private {
- REENTRANCY_GUARD_STORAGE.asBoolean().tstore(false);
- }
- /**
- * @dev Returns true if the reentrancy guard is currently set to "entered", which indicates there is a
- * `nonReentrant` function in the call stack.
- */
- function _reentrancyGuardEntered() internal view returns (bool) {
- return REENTRANCY_GUARD_STORAGE.asBoolean().tload();
- }
- }
|