add stop recovery method

iotexproject · Nov 2, 2023 · ed29199 · ed29199
1 parent c869550
commit ed29199
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/contracts/accounts/secp256r1/P256Account.sol b/contracts/accounts/secp256r1/P256Account.sol
@@ -48,6 +48,7 @@ contract P256Account is BaseAccount, TokenCallbackHandler, UUPSUpgradeable, Init
     event EmailGuardianRemoved();
     event AccountPendingRecovey(uint256 timestamp, bytes publicKey);
     event AccountRecovered(bytes publicKey);
+    event AccountRecoveryStopped();
     event AccountResetted(bytes publicKey);
 
     // solhint-disable-next-line no-empty-blocks
@@ -197,6 +198,14 @@ contract P256Account is BaseAccount, TokenCallbackHandler, UUPSUpgradeable, Init
         emit AccountPendingRecovey(block.timestamp, publicKey);
     }
 
+    function stopRecovery() external {
+        require(address(this) == msg.sender, "only owner");
+        require(pendingPublicKey.timestamp > 0, "no recovery");
+
+        pendingPublicKey.timestamp = 0;
+        emit AccountRecoveryStopped();
+    }
+
     function recovery() external {
         require(
             pendingPublicKey.timestamp > 0 && pendingPublicKey.timestamp + 86400 < block.timestamp,