erc7579-implementation/src/core/PreValidationHookManager.sol at 9ccedd162ebf91959e25e7b5cfd6d9bafca2c17f · erc7579/erc7579-implementation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.21;

import "./ModuleManager.sol";
import "../interfaces/IERC7579Account.sol";
import "../interfaces/IERC7579Module.sol";

/**
 * @title reference implementation of PreValidationHookManager
 * @author highskore | rhinestone.wtf
 */
abstract contract PreValidationHookManager {
    event PreValidationHookUninstallFailed(address hook, bytes data);

    error InvalidHookType();

    /// @custom:storage-location erc7201:prevalidationhookmanager.storage.msa
    struct PreValidationHookManagerStorage {
        IPreValidationHookERC1271 hook1271;
        IPreValidationHookERC4337 hook4337;
    }

    // forgefmt: disable-next-line
    // keccak256(abi.encode(uint256(keccak256("prevalidationhookmanager.storage.msa")) - 1)) & ~bytes32(uint256(0xff));
    bytes32 constant PREVALIDATION_HOOKMANAGER_STORAGE_LOCATION =
        0x088e45215d3756b04bd240e41d75700a696139d5b53082481ffc3914e4840000;

    error PreValidationHookAlreadyInstalled(address currentHook);

    function _getStorage()
        internal
        pure
        returns (PreValidationHookManagerStorage storage storage_)
    {
        bytes32 slot = PREVALIDATION_HOOKMANAGER_STORAGE_LOCATION;
        assembly {
            storage_.slot := slot
        }
    }

    function _setPreValidationHook(address hook, uint256 hookType) internal virtual {
        PreValidationHookManagerStorage storage $ = _getStorage();
        if (hookType == MODULE_TYPE_PREVALIDATION_HOOK_ERC1271) {
            $.hook1271 = IPreValidationHookERC1271(hook);
        } else if (hookType == MODULE_TYPE_PREVALIDATION_HOOK_ERC4337) {
            $.hook4337 = IPreValidationHookERC4337(hook);
        } else {
            revert InvalidHookType();
        }
    }

    function _installPreValidationHook(
        address hook,
        uint256 hookType,
        bytes calldata data
    )
        internal
        virtual
    {
        PreValidationHookManagerStorage storage $ = _getStorage();
        address currentHook = _getPreValidationHook(hookType);
        if (currentHook != address(0)) {
            revert PreValidationHookAlreadyInstalled(currentHook);
        }
        _setPreValidationHook(hook, hookType);
        if (hookType == MODULE_TYPE_PREVALIDATION_HOOK_ERC1271) {
            $.hook1271.onInstall(data);
        } else if (hookType == MODULE_TYPE_PREVALIDATION_HOOK_ERC4337) {
            $.hook4337.onInstall(data);
        }
    }

    function _uninstallPreValidationHook(
        address hook,
        uint256 hookType,
        bytes calldata data
    )
        internal
        virtual
    {
        PreValidationHookManagerStorage storage $ = _getStorage();
        if (hookType == MODULE_TYPE_PREVALIDATION_HOOK_ERC1271 && address($.hook1271) == hook) {
            $.hook1271.onUninstall(data);
        } else if (
            hookType == MODULE_TYPE_PREVALIDATION_HOOK_ERC4337 && address($.hook4337) == hook
        ) {
            $.hook4337.onUninstall(data);
        } else {
            revert InvalidHookType();
        }
        _setPreValidationHook(address(0), hookType);
    }

    function _tryUninstallPreValidationHook(address hook, uint256 hookType) internal virtual {
        if (hook == address(0)) return;
        PreValidationHookManagerStorage storage $ = _getStorage();
        if (hookType == MODULE_TYPE_PREVALIDATION_HOOK_ERC1271) {
            try $.hook1271.onUninstall("") { }
            catch {
                emit PreValidationHookUninstallFailed(hook, "");
            }
            $.hook1271 = IPreValidationHookERC1271(address(0));
        } else if (hookType == MODULE_TYPE_PREVALIDATION_HOOK_ERC4337) {
            try $.hook4337.onUninstall("") { }
            catch {
                emit PreValidationHookUninstallFailed(hook, "");
            }
            $.hook4337 = IPreValidationHookERC4337(address(0));
        } else {
            revert InvalidHookType();
        }
    }

    function _getPreValidationHook(uint256 hookType) internal view returns (address _hook) {
        PreValidationHookManagerStorage storage $ = _getStorage();
        if (hookType == MODULE_TYPE_PREVALIDATION_HOOK_ERC1271) {
            return address($.hook1271);
        } else if (hookType == MODULE_TYPE_PREVALIDATION_HOOK_ERC4337) {
            return address($.hook4337);
        } else {
            revert InvalidHookType();
        }
    }

    function _isPreValidationHookInstalled(
        address module,
        uint256 hookType
    )
        internal
        view
        returns (bool)
    {
        return _getPreValidationHook(hookType) == module;
    }

    function getActiveHook(uint256 hookType) external view returns (address hook) {
        return _getPreValidationHook(hookType);
    }

    function _withPreValidationHook(
        bytes32 hash,
        bytes calldata signature
    )
        internal
        view
        virtual
        returns (bytes32 postHash, bytes memory postSig)
    {
        address preValidationHook = _getPreValidationHook(MODULE_TYPE_PREVALIDATION_HOOK_ERC1271);
        if (preValidationHook == address(0)) {
            return (hash, signature);
        } else {
            return IPreValidationHookERC1271(preValidationHook).preValidationHookERC1271(
                msg.sender, hash, signature
            );
        }
    }

    function _withPreValidationHook(
        bytes32 hash,
        PackedUserOperation memory userOp,
        uint256 missingAccountFunds
    )
        internal
        virtual
        returns (bytes32 postHash, bytes memory postSig)
    {
        address preValidationHook = _getPreValidationHook(MODULE_TYPE_PREVALIDATION_HOOK_ERC4337);
        if (preValidationHook == address(0)) {
            return (hash, userOp.signature);
        } else {
            return IPreValidationHookERC4337(preValidationHook).preValidationHookERC4337(
                userOp, missingAccountFunds, hash
            );
        }
    }
}