fix: allow revoke keys via paymaster (#212)

matter-labs · Dec 4, 2024 · f10f345 · f10f345
1 parent 3ae42a7
commit f10f345
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -13,6 +13,7 @@
     "prepare": "husky",
     "fmt": "pnpm prettier --write .",
     "lint": "pnpm run lint:markdown && pnpm run lint:spelling && pnpm run lint:prettier",
+    "lint:fix": "pnpm run lint:prettier --write",
     "lint:markdown": "markdownlint-cli2 '**/*.md'",
     "lint:spelling": "cspell .",
     "lint:prettier": "prettier --check .",

diff --git a/src/test/ExampleAuthServerPaymaster.sol b/src/test/ExampleAuthServerPaymaster.sol
@@ -17,7 +17,9 @@ contract ExampleAuthServerPaymaster is IPaymaster, Ownable {
   address public immutable AA_FACTORY_CONTRACT_ADDRESS;
   address public immutable SESSION_KEY_VALIDATOR_CONTRACT_ADDRESS;
   bytes4 constant DEPLOY_ACCOUNT_SELECTOR = AAFactory.deployProxySsoAccount.selector;
-  bytes4 constant CREATE_SESSION_SELECTOR = SessionKeyValidator.createSession.selector;
+  bytes4 constant SESSION_CREATE_SELECTOR = SessionKeyValidator.createSession.selector;
+  bytes4 constant SESSION_REVOKE_KEY_SELECTOR = SessionKeyValidator.revokeKey.selector;
+  bytes4 constant SESSION_REVOKE_KEYS_SELECTOR = SessionKeyValidator.revokeKeys.selector;
 
   modifier onlyBootloader() {
     require(msg.sender == BOOTLOADER_FORMAL_ADDRESS, "Only bootloader can call this method");
@@ -53,7 +55,12 @@ contract ExampleAuthServerPaymaster is IPaymaster, Ownable {
       require(methodSelector == DEPLOY_ACCOUNT_SELECTOR, "Unsupported method");
     }
     if (to == SESSION_KEY_VALIDATOR_CONTRACT_ADDRESS) {
-      require(methodSelector == CREATE_SESSION_SELECTOR, "Unsupported method");
+      require(
+        methodSelector == SESSION_CREATE_SELECTOR ||
+          methodSelector == SESSION_REVOKE_KEY_SELECTOR ||
+          methodSelector == SESSION_REVOKE_KEYS_SELECTOR,
+        "Unsupported method"
+      );
     }
 
     bytes4 paymasterInputSelector = bytes4(_transaction.paymasterInput[0:4]);