fix: Add check for already initialized guardian

jpgonzalezra · May 20, 2024 · b8e2cbd · b8e2cbd
1 parent 89967b4
commit b8e2cbd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/SimpleGuardianModule.sol b/src/SimpleGuardianModule.sol
@@ -18,6 +18,7 @@ abstract contract SimpleGuardianModule {
  error InvalidGuardianSignature();
  error NotGuardian();
  error InvalidNonce();
+ error GuardianAlreadyInitialized();
 
  address public guardian;
  mapping(address => uint256) private _nonces;
@@ -37,7 +38,9 @@ abstract contract SimpleGuardianModule {
  }
 
  function initGuardian(address newGuardian) external {
- require(guardian == address(0));
+ if (guardian != address(0)) {
+ revert GuardianAlreadyInitialized();
+ }
  _updateGuardian(newGuardian);
  }