Optimize

Vectorized · Jan 26, 2024 · b3b0a88 · b3b0a88
1 parent dfa6dad
commit b3b0a88
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/utils/ReentrancyGuard.sol b/src/utils/ReentrancyGuard.sol
@@ -31,11 +31,11 @@ abstract contract ReentrancyGuard {
     modifier nonReentrant() virtual {
         /// @solidity memory-safe-assembly
         assembly {
-            if eq(tload(_REENTRANCY_GUARD_SLOT), 1) {
+            if tload(_REENTRANCY_GUARD_SLOT) {
                 mstore(0x00, 0xab143c06) // `Reentrancy()`.
                 revert(0x1c, 0x04)
             }
-            tstore(_REENTRANCY_GUARD_SLOT, 1)
+            tstore(_REENTRANCY_GUARD_SLOT, address())
         }
         _;
         /// @solidity memory-safe-assembly
@@ -48,7 +48,7 @@ abstract contract ReentrancyGuard {
     modifier nonReadReentrant() virtual {
         /// @solidity memory-safe-assembly
         assembly {
-            if eq(tload(_REENTRANCY_GUARD_SLOT), 1) {
+            if tload(_REENTRANCY_GUARD_SLOT) {
                 mstore(0x00, 0xab143c06) // `Reentrancy()`.
                 revert(0x1c, 0x04)
             }

diff --git a/test/utils/mocks/MockReentrancyGuard.sol b/test/utils/mocks/MockReentrancyGuard.sol
@@ -16,7 +16,7 @@ contract MockReentrancyGuard is ReentrancyGuard {
     function isReentrancyGuardLocked() public view returns (bool locked) {
         /// @solidity memory-safe-assembly
         assembly {
-            if eq(tload(_REENTRANCY_GUARD_SLOT), 1) { locked := true }
+            if tload(_REENTRANCY_GUARD_SLOT) { locked := true }
         }
     }