changeRootValidator (#114)

zerodevapp · May 8, 2024 · 51e935d · 51e935d
1 parent 10c0997
commit 51e935d
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/src/Kernel.sol b/src/Kernel.sol
@@ -110,6 +110,28 @@ contract Kernel is IAccount, IAccountExecute, IERC7579Account, ValidationManager
         _installValidation(_rootValidator, config, validatorData, hookData);
     }
 
+    function changeRootValidator(
+        ValidationId _rootValidator,
+        IHook hook,
+        bytes calldata validatorData,
+        bytes calldata hookData
+    ) external payable onlyEntryPointOrSelfOrRoot {
+        ValidationStorage storage vs = _validationStorage();
+        if (ValidationId.unwrap(_rootValidator) == bytes21(0)) {
+            revert InvalidValidator();
+        }
+        ValidationType vType = ValidatorLib.getType(_rootValidator);
+        if (vType != VALIDATION_TYPE_VALIDATOR && vType != VALIDATION_TYPE_PERMISSION) {
+            revert InvalidValidationType();
+        }
+        _setRootValidator(_rootValidator);
+        if (_validationStorage().validationConfig[_rootValidator].hook == IHook(address(0))) {
+            // when new rootValidator is not installed yet
+            ValidationConfig memory config = ValidationConfig({nonce: uint32(vs.currentNonce), hook: hook});
+            _installValidation(_rootValidator, config, validatorData, hookData);
+        }
+    }
+
     function upgradeTo(address _newImplementation) external payable onlyEntryPointOrSelfOrRoot {
         assembly {
             sstore(ERC1967_IMPLEMENTATION_SLOT, _newImplementation)