@etherisc/gif-next 0.0.2-d4eadee-930 → 0.0.2-d53bd1e-732

package/contracts/shared/AccessAdmin.sol

@@ -0,0 +1,759 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.20;
+
+import {AccessManager} from "@openzeppelin/contracts/access/manager/AccessManager.sol";
+import {AccessManagedUpgradeable} from "@openzeppelin/contracts-upgradeable/access/manager/AccessManagedUpgradeable.sol";
+import {EnumerableSet} from "@openzeppelin/contracts/utils/structs/EnumerableSet.sol";
+
+import {IAccessAdmin} from "./IAccessAdmin.sol";
+import {RoleId, RoleIdLib} from "../type/RoleId.sol";
+import {Selector, SelectorLib, SelectorSetLib} from "../type/Selector.sol";
+import {Str, StrLib} from "../type/String.sol";
+import {Timestamp, TimestampLib} from "../type/Timestamp.sol";
+
+
+/**
+ * @dev A generic access amin contract that implements role based access control based on OpenZeppelin's AccessManager contract.
+ * The contract provides read functions to query all available roles, targets and access rights.
+ * This contract works for both a constructor based deployment or a deployment based on cloning and initialization.
+ */ 
+contract AccessAdmin is
+    AccessManagedUpgradeable,
+    IAccessAdmin
+{
+    using EnumerableSet for EnumerableSet.AddressSet;
+
+    string public constant ADMIN_ROLE_NAME = "AdminRole";
+    string public constant PUBLIC_ROLE_NAME = "PublicRole";
+
+    uint64 public constant MANAGER_ROLE = type(uint64).min + 1;
+    string public constant MANAGER_ROLE_NAME = "ManagerRole";
+
+    /// @dev the OpenZeppelin access manager driving the access admin contract
+    AccessManager internal _authority;
+
+    /// @dev stores the deployer address and allows to create initializers
+    /// that are restricted to the deployer address.
+    address internal _deployer;
+
+    /// @dev required role for state changes to this contract
+    RoleId internal _managerRoleId;
+
+    /// @dev store role info per role id
+    mapping(RoleId roleId => RoleInfo info) internal _roleInfo;
+
+    /// @dev store role name info per role name
+    mapping(Str roleName => RoleNameInfo nameInfo) internal _roleForName;
+
+    /// @dev store array with all created roles
+    RoleId [] internal _roleIds;
+
+    /// @dev store set of current role members for given role
+    mapping(RoleId roleId => EnumerableSet.AddressSet roleMembers) internal _roleMembers;
+
+    /// @dev store target info per target address
+    mapping(address target => TargetInfo info) internal _targetInfo;
+
+    /// @dev store role name info per role name
+    mapping(Str targetName => address target) internal _targetForName;
+
+    /// @dev store array with all created targets
+    address [] internal _targets;
+
+    /// @dev store all managed functions per target
+    mapping(address target => SelectorSetLib.Set selectors) internal _targetFunctions;
+
+    /// @dev temporary dynamic function infos array
+    mapping(address target => mapping(Selector selector => Str functionName)) internal _functionName;
+
+    /// @dev temporary dynamic functions array
+    bytes4[] private _functions;
+
+    modifier onlyDeployer() {
+        // special case for cloned AccessAdmin contracts
+        // IMPORTANT cloning and _initializeAuthority needs to be done in a single transaction
+        if (_deployer == address(0)) {
+            _deployer = msg.sender;
+        }
+
+        if (msg.sender != _deployer) {
+            revert ErrorNotDeployer();
+        }
+        _;
+    }
+
+    modifier onlyRoleAdmin(RoleId roleId) {
+        if (!roleExists(roleId)) {
+            revert ErrorRoleUnknown(roleId);
+        }
+
+        if (!hasRole(msg.sender, _roleInfo[roleId].adminRoleId)) {
+            revert ErrorNotAdminOfRole(_roleInfo[roleId].adminRoleId);
+        }
+        _;
+    }
+
+    modifier onlyRoleMember(RoleId roleId) {
+        if (!hasRole(msg.sender, roleId)) {
+            revert ErrorNotRoleOwner(roleId);
+        }
+        _;
+    }
+
+    modifier onlyExistingRole(RoleId roleId) {
+        _checkRoleId(roleId);
+        _;
+    }
+
+    modifier onlyExistingTarget(address target) {
+        _checkTarget(target);
+        _;
+    }
+
+    constructor() {
+        _deployer = msg.sender;
+        _authority = new AccessManager(address(this));
+
+        _setAuthority(address(_authority));
+        _createInitialRoleSetup();
+
+        _disableInitializers();
+    }
+
+    //--- role management functions -----------------------------------------//
+
+    function createRole(
+        RoleId roleId, 
+        RoleId adminRoleId, 
+        string memory name,
+        uint256 maxMemberCount,
+        bool memberRemovalDisabled
+    )
+        external
+        virtual
+        restricted()
+    {
+        _createRole(roleId, adminRoleId, name, maxMemberCount, memberRemovalDisabled);
+    }
+
+    function setRoleDisabled(
+        RoleId roleId, 
+        bool disabled
+    )
+        external
+        virtual
+        restricted()
+    {
+        _setRoleDisabled(roleId, disabled);
+    }
+
+    function grantRole(
+        address account, 
+        RoleId roleId
+    )
+        external
+        virtual
+        onlyRoleAdmin(roleId)
+        restricted()
+    {
+        _grantRoleToAccount(roleId, account);
+    }
+
+    function revokeRole(
+        address account, 
+        RoleId roleId
+    )
+        external
+        virtual
+        onlyRoleAdmin(roleId)
+        restricted()
+    {
+        _revokeRoleFromAccount(roleId, account);
+    }
+
+    function renounceRole(
+        RoleId roleId
+    )
+        external
+        virtual
+        onlyRoleMember(roleId)
+        restricted()
+    {
+        _revokeRoleFromAccount(roleId, msg.sender);
+    }
+
+    //--- target management functions ---------------------------------------//
+
+    function createTarget(
+        address target, 
+        string memory name
+    )
+        external
+        virtual
+        restricted()
+    {
+        _createTarget(target, name);
+    }
+
+    function setTargetLocked(
+        address target, 
+        bool locked
+    )
+        external
+        virtual
+        onlyExistingTarget(target)
+        restricted()
+    {
+        _authority.setTargetClosed(target, locked);
+
+        // implizit logging: rely on OpenZeppelin log TargetClosed
+    }
+
+    function authorizeFunctions(
+        address target, 
+        RoleId roleId, 
+        Function[] memory functions
+    )
+        external
+        virtual
+        onlyExistingTarget(target)
+        onlyExistingRole(roleId)
+        restricted()
+    {
+        _authorizeTargetFunctions(target, roleId, functions);
+    }
+
+    function unauthorizeFunctions(
+        address target, 
+        Function[] memory functions
+    )
+        external
+        virtual
+        restricted()
+    {
+        _unauthorizeTargetFunctions(target, functions);
+    }
+
+
+    function authorizedFunctions(address target) external view returns (uint256 numberOfFunctions) {
+        return SelectorSetLib.size(_targetFunctions[target]);
+    }
+
+    function getAuthorizedFunction(
+        address target, 
+        uint256 idx
+    )
+        external 
+        view 
+        returns (
+            Function memory func, 
+            RoleId roleId
+        )
+    {
+        Selector selector = SelectorSetLib.at(_targetFunctions[target], idx);
+
+        func = Function({
+            selector: selector, 
+            name: _functionName[target][selector]});
+
+        roleId = RoleIdLib.toRoleId(
+            _authority.getTargetFunctionRole(
+                target, 
+                selector.toBytes4()));
+    }
+
+    //--- view functions ----------------------------------------------------//
+
+    function roles() external view returns (uint256 numberOfRoles) {
+        return _roleIds.length;
+    }
+
+    function getRoleId(uint256 idx) external view returns (RoleId roleId) {
+        return _roleIds[idx];
+    }
+
+    function getAdminRole() public view returns (RoleId roleId) {
+        return RoleId.wrap(_authority.ADMIN_ROLE());
+    }
+
+    function getPublicRole() public view returns (RoleId roleId) {
+        return RoleId.wrap(_authority.PUBLIC_ROLE());
+    }
+
+    function getManagerRole() public view returns (RoleId roleId) {
+        return _managerRoleId;
+    }
+
+    function roleExists(RoleId roleId) public view returns (bool exists) {
+        return _roleInfo[roleId].exists;
+    }
+
+    function isRoleDisabled(RoleId roleId) public view returns (bool isActive) {
+        return _roleInfo[roleId].disabledAt <= TimestampLib.blockTimestamp();
+    }
+
+    function getRoleInfo(RoleId roleId) external view returns (RoleInfo memory) {
+        return _roleInfo[roleId];
+    }
+
+    function getRoleForName(Str name) external view returns (RoleNameInfo memory) {
+        return _roleForName[name];
+    }
+
+    function hasRole(address account, RoleId roleId) public view returns (bool) {
+        (bool isMember, ) = _authority.hasRole(
+            RoleId.unwrap(roleId), 
+            account);
+        return isMember;
+    }
+
+    function roleMembers(RoleId roleId) external view returns (uint256 numberOfMembers) {
+        return _roleMembers[roleId].length();
+    }
+
+    function getRoleMember(RoleId roleId, uint256 idx) external view returns (address account) {
+        return _roleMembers[roleId].at(idx);
+    }
+
+    function targetExists(address target) public view returns (bool exists) {
+        return _targetInfo[target].createdAt.gtz();
+    }
+
+    function isTargetLocked(address target) public view returns (bool locked) {
+        return _authority.isTargetClosed(target);
+    }
+
+    function targets() external view returns (uint256 numberOfTargets) {
+        return _targets.length;
+    }
+
+    function getTargetAddress(uint256 idx) external view returns (address target) {
+        return _targets[idx];
+    }
+
+    function getTargetInfo(address target) external view returns (TargetInfo memory targetInfo) {
+        return _targetInfo[target];
+    }
+
+    function getTargetForName(Str name) external view returns (address target) {
+        return _targetForName[name];
+    }
+
+    function isAccessManaged(address target) public view returns (bool) {
+        if (!_isContract(target)) {
+            return false;
+        }
+
+        (bool success, ) = target.staticcall(
+            abi.encodeWithSelector(
+                AccessManagedUpgradeable.authority.selector));
+
+        return success;
+    }
+
+    function canCall(address caller, address target, Selector selector) external view returns (bool can) {
+        (can, ) = _authority.canCall(caller, target, selector.toBytes4());
+    }
+
+    function deployer() public view returns (address) {
+        return _deployer;
+    }
+
+    function toFunction(bytes4 selector, string memory name) public pure returns (Function memory) {
+            return Function({
+                selector: SelectorLib.toSelector(selector),
+                name: StrLib.toStr(name)});
+    }
+
+    //--- internal/private functions -------------------------------------------------//
+
+    function _authorizeTargetFunctions(
+        address target, 
+        RoleId roleId, 
+        Function[] memory functions
+    )
+        internal
+    {
+        if (roleId == getAdminRole()) {
+            revert ErrorAuthorizeForAdminRoleInvalid(target);
+        }
+
+        bool addFunctions = true;
+        bytes4[] memory functionSelectors = _processFunctionSelectors(target, functions, addFunctions);
+
+        // apply authz via access manager
+        _grantRoleAccessToFunctions(target, roleId, functionSelectors);
+    }
+
+    function _unauthorizeTargetFunctions(
+        address target, 
+        Function[] memory functions
+    )
+        internal
+    {
+        bool addFunctions = false;
+        bytes4[] memory functionSelectors = _processFunctionSelectors(target, functions, addFunctions);
+        _grantRoleAccessToFunctions(target, getAdminRole(), functionSelectors);
+    }
+
+    function _processFunctionSelectors(
+        address target,
+        Function[] memory functions,
+        bool addFunctions
+    )
+        internal
+        returns (
+            bytes4[] memory functionSelectors
+        )
+    {
+        uint256 n = functions.length;
+        functionSelectors = new bytes4[](n);
+        Function memory func;
+        Selector selector;
+
+        for (uint256 i = 0; i < n; i++) {
+            func = functions[i];
+            selector = func.selector;
+
+            // add function selector to target selector set if not in set
+            if (addFunctions) { SelectorSetLib.add(_targetFunctions[target], selector); } 
+            else { SelectorSetLib.remove(_targetFunctions[target], selector); }
+
+            // set function name
+            _functionName[target][selector] = func.name;
+
+            // add bytes4 selector to function selector array
+            functionSelectors[i] = selector.toBytes4();
+        }
+    }
+
+    function _initializeAuthority(
+        address authorityAddress
+    )
+        internal
+        virtual
+        onlyInitializing()
+        onlyDeployer()
+    {
+        if (authority() != address(0)) {
+            revert ErrorAuthorityAlreadySet();
+        }
+
+        _authority = AccessManager(authorityAddress);
+
+        if(!hasRole(address(this), RoleId.wrap(_authority.ADMIN_ROLE()))) {
+            revert ErrorAdminRoleMissing();
+        }
+
+        __AccessManaged_init(address(_authority));
+    }
+
+
+    function _initializeRoleSetup()
+        internal
+        virtual
+        onlyInitializing()
+    {
+        _createInitialRoleSetup();
+    }
+
+
+    function _createInitialRoleSetup()
+        internal
+    {
+        RoleId adminRoleId = RoleIdLib.toRoleId(_authority.ADMIN_ROLE());
+        Function[] memory functions;
+
+        // setup admin role
+        _createRoleUnchecked(
+            adminRoleId,
+            adminRoleId,
+            StrLib.toStr(ADMIN_ROLE_NAME),
+            1,
+            true);
+
+        // add this contract as admin role member
+        _roleMembers[adminRoleId].add(address(this));
+
+        // setup public role
+        _createRoleUnchecked(
+            RoleIdLib.toRoleId(_authority.PUBLIC_ROLE()),
+            adminRoleId,
+            StrLib.toStr(PUBLIC_ROLE_NAME),
+            type(uint256).max,
+            true);
+
+        // setup manager role
+        _managerRoleId = RoleIdLib.toRoleId(MANAGER_ROLE);
+        _createRole(
+            _managerRoleId, 
+            adminRoleId,
+            MANAGER_ROLE_NAME,
+            3, // TODO think about max member count
+            false);
+
+        // grant public role access to grant and revoke, renounce
+        functions = new Function[](3);
+        functions[0] = toFunction(IAccessAdmin.grantRole.selector, "grantRole");
+        functions[1] = toFunction(IAccessAdmin.revokeRole.selector, "revokeRole");
+        functions[2] = toFunction(IAccessAdmin.renounceRole.selector, "renounceRole");
+        _authorizeTargetFunctions(address(this), getPublicRole(), functions);
+
+        // grant manager role access to the specified functions 
+        functions = new Function[](6);
+        functions[0] = toFunction(IAccessAdmin.createRole.selector, "createRole");
+        functions[1] = toFunction(IAccessAdmin.setRoleDisabled.selector, "setRoleDisabled");
+        functions[2] = toFunction(IAccessAdmin.createTarget.selector, "createTarget");
+        functions[3] = toFunction(IAccessAdmin.setTargetLocked.selector, "setTargetLocked");
+        functions[4] = toFunction(IAccessAdmin.authorizeFunctions.selector, "authorizeFunctions");
+        functions[5] = toFunction(IAccessAdmin.unauthorizeFunctions.selector, "unauthorizeFunctions");
+        _authorizeTargetFunctions(address(this), getManagerRole(), functions);
+    }
+
+    /// @dev check if target exists and reverts if it doesn't
+    function _checkTarget(address target)
+        internal
+    {
+        if (_targetInfo[target].createdAt.eqz()) {
+            revert ErrorTargetUnknown(target);
+        }
+    }
+
+    /// @dev grant the specified role access to all functions in the provided selector list
+    function _grantRoleAccessToFunctions(
+        address target,
+        RoleId roleId, 
+        bytes4[] memory functionSelectors
+    )
+        internal
+    {
+        _authority.setTargetFunctionRole(
+            target,
+            functionSelectors,
+            RoleId.unwrap(roleId));
+
+        // implizit logging: rely on OpenZeppelin log TargetFunctionRoleUpdated
+    }
+
+    /// @dev grant the specified role to the provided account
+    function _grantRoleToAccount(RoleId roleId, address account)
+        internal
+    {
+        _checkRoleId(roleId);
+        _checkRoleIsActive(roleId);
+
+        // check max role members will not be exceeded
+        if (_roleMembers[roleId].length() >= _roleInfo[roleId].maxMemberCount) {
+            revert ErrorRoleMembersLimitReached(roleId, _roleInfo[roleId].maxMemberCount);
+        }
+
+        _roleMembers[roleId].add(account);
+        _authority.grantRole(
+            RoleId.unwrap(roleId), 
+            account, 
+            0);
+        
+        // indirect logging: rely on OpenZeppelin log RoleGranted
+    }
+
+    /// @dev revoke the specified role from the provided account
+    function _revokeRoleFromAccount(RoleId roleId, address account)
+        internal
+    {
+        _checkRoleId(roleId);
+
+        // check role removal is permitted
+        if (_roleInfo[roleId].memberRemovalDisabled) {
+            revert ErrorRoleRemovalDisabled(roleId);
+        }
+
+        _roleMembers[roleId].remove(account);
+        _authority.revokeRole(
+            RoleId.unwrap(roleId), 
+            account);
+
+        // indirect logging: rely on OpenZeppelin log RoleGranted
+    }
+
+
+    function _checkRoleId(RoleId roleId)
+        internal
+    {
+        if (!_roleInfo[roleId].exists) {
+            revert ErrorRoleUnknown(roleId);
+        }
+
+        uint64 roleIdInt = RoleId.unwrap(roleId);
+        if (roleIdInt == _authority.ADMIN_ROLE()
+            || roleIdInt == _authority.PUBLIC_ROLE())
+        {
+            revert ErrorRoleIsLocked(roleId);
+        }
+    }
+
+
+    function _checkRoleIsActive(RoleId roleId)
+        internal
+    {
+        if (isRoleDisabled(roleId)) {
+            revert ErrorRoleIsDisabled(roleId);
+        }
+    }
+
+
+    function _createRole(
+        RoleId roleId, 
+        RoleId adminRoleId, 
+        string memory roleName,
+        uint256 maxMemberCount,
+        bool memberRemovalDisabled
+    )
+        internal
+    {
+        // check role does not yet exist
+        if(roleExists(roleId)) {
+            revert ErrorRoleAlreadyCreated(
+                roleId, 
+                _roleInfo[roleId].name.toString());
+        }
+
+        // check admin role exists
+        if(!roleExists(adminRoleId)) {
+            revert ErrorRoleAdminNotExisting(adminRoleId);
+        }
+
+        // check role name is not empty
+        Str name = StrLib.toStr(roleName);
+        if(name.length() == 0) {
+            revert ErrorRoleNameEmpty(roleId);
+        }
+
+        // check role name is not used for another role
+        if(_roleForName[name].exists) {
+            revert ErrorRoleNameAlreadyExists(
+                roleId, 
+                roleName,
+                _roleForName[name].roleId);
+        }
+
+        _createRoleUnchecked(
+            roleId, adminRoleId, 
+            name, 
+            maxMemberCount, 
+            memberRemovalDisabled);
+    }
+
+
+    function _setRoleDisabled(
+        RoleId roleId, 
+        bool disabled
+    )
+        internal
+    {
+
+        _checkRoleId(roleId);
+        Timestamp disabledAtOld = _roleInfo[roleId].disabledAt;
+
+        if (disabled) {
+            _roleInfo[roleId].disabledAt = TimestampLib.blockTimestamp();
+        } else {
+            _roleInfo[roleId].disabledAt = TimestampLib.max();
+        }
+
+        emit LogRoleDisabled(roleId, disabled, disabledAtOld);
+    }
+
+
+    function _createRoleUnchecked(
+        RoleId roleId, 
+        RoleId adminRoleId, 
+        Str name,
+        uint256 maxMemberCount,
+        bool memberRemovalDisabled
+    )
+        private
+    {
+        // create role info
+        _roleInfo[roleId] = RoleInfo({
+            adminRoleId: adminRoleId,
+            name: name,
+            maxMemberCount: maxMemberCount,
+            memberRemovalDisabled: memberRemovalDisabled,
+            disabledAt: TimestampLib.max(),
+            exists: true});
+
+        // create role name info
+        _roleForName[name] = RoleNameInfo({
+            roleId: roleId,
+            exists: true});
+
+        // add role to list of roles
+        _roleIds.push(roleId);
+
+        emit LogRoleCreated(roleId, adminRoleId, name.toString());
+    }
+
+
+    function _createTarget(address target, string memory targetName)
+        internal
+    {
+        // check target does not yet exist
+        if(targetExists(target)) {
+            revert ErrorTargetAlreadyCreated(
+                target, 
+                _targetInfo[target].name.toString());
+        }
+
+        // check target name is not empty
+        Str name = StrLib.toStr(targetName);
+        if(name.length() == 0) {
+            revert ErrorTargetNameEmpty(target);
+        }
+
+        // check target name is not used for another role
+        if( _targetForName[name] != address(0)) {
+            revert ErrorTargetNameAlreadyExists(
+                target, 
+                targetName,
+                _targetForName[name]);
+        }
+
+        // check target is an access managed contract
+        if (!isAccessManaged(target)) {
+            revert ErrorTargetNotAccessManaged(target);
+        }
+
+        // check target shares authority with this contract
+        address targetAuthority = AccessManagedUpgradeable(target).authority();
+        if (targetAuthority != authority()) {
+            revert ErrorTargetAuthorityMismatch(authority(), targetAuthority);
+        }
+
+        // create target info
+        _targetInfo[target] = TargetInfo({
+            name: name,
+            createdAt: TimestampLib.blockTimestamp()
+        });
+
+        // create name to target mapping
+        _targetForName[name] = target;
+
+        // add role to list of roles
+        _targets.push(target);
+
+        emit LogTargetCreated(target, targetName);
+    }
+
+    function _isContract(address target)
+        internal
+        view 
+        returns (bool)
+    {
+        uint256 size;
+        assembly {
+            size := extcodesize(target)
+        }
+        return size > 0;
+    }
+    
+}