recover module

abstraction-hq · May 18, 2024 · c604fd8 · c604fd8
1 parent 56b2be9
commit c604fd8
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/modules/Recovery.sol b/src/modules/Recovery.sol
@@ -41,6 +41,11 @@ contract Recovery is IModule {
         _;
     }
 
+    modifier onlyWallet() {
+        require(msg.sender == _wallet, "Only wallet can call");
+        _;
+    }
+
     function _setGuardian(address guardian, bool status) internal {
         _isGuardians[guardian] = status;
 
@@ -81,6 +86,7 @@ contract Recovery is IModule {
     function validateUserOp(UserOperation calldata userOp, bytes32 userOpHash)
         external
         override
+        onlyWallet()
         returns (uint256 validationData)
     {
         require(msg.sender == _wallet, "Wrong wallet");
@@ -90,6 +96,7 @@ contract Recovery is IModule {
         public
         view
         override
+        onlyWallet()
         returns (bytes4 magicValue)
     {
         return 0x0000;