SmtLib updates

contracts/interfaces/IHasher.sol

@@ -0,0 +1,21 @@
+// SPDX-License-Identifier: GPL-3.0
+pragma solidity 0.8.27;
+
+/**
+ * @dev IHasher. Interface for generating hashes. Specifically used for Merkle Tree hashing.
+ */
+interface IHasher {
+    /**
+     * @dev hash2. hashes two uint256 parameters and returns the resulting hash as uint256.
+     * @param params The parameters array of size 2 to be hashed.
+     * @return The resulting hash as uint256.
+     */
+    function hash2(uint256[2] memory params) external pure returns (uint256);
+
+    /**
+     * @dev hash3. hashes three uint256 parameters and returns the resulting hash as uint256.
+     * @param params The parameters array of size 3 to be hashed.
+     * @return The resulting hash as uint256.
+     */
+    function hash3(uint256[3] memory params) external pure returns (uint256);
+}

contracts/lib/IdentityBase.sol

@@ -6,6 +6,7 @@ import {IOnchainCredentialStatusResolver} from "../interfaces/IOnchainCredential
 import {IdentityLib} from "../lib/IdentityLib.sol";
 import {SmtLib} from "../lib/SmtLib.sol";
 import {ERC165} from "@openzeppelin/contracts/utils/introspection/ERC165.sol";
+import {IHasher} from "../interfaces/IHasher.sol";
 
 error IdentityIdMismatch();
 
@@ -46,12 +47,13 @@ abstract contract IdentityBase is IIdentifiable, IOnchainCredentialStatusResolve
      * @dev Initialization of IdentityLib library
      * @param _stateContractAddr - address of the State contract
      */
-    function initialize(address _stateContractAddr, bytes2 idType) public virtual {
+    function initialize(address _stateContractAddr, bytes2 idType, IHasher hasher) public virtual {
         _getIdentityBaseStorage().identity.initialize(
             _stateContractAddr,
             address(this),
             getSmtDepth(),
-            idType
+            idType,
+            hasher
         );
     }
 

contracts/lib/IdentityLib.sol

@@ -5,6 +5,7 @@ import {IState} from "../interfaces/IState.sol";
 import {SmtLib} from "../lib/SmtLib.sol";
 import {PoseidonUnit3L, PoseidonUnit4L} from "../lib/Poseidon.sol";
 import {GenesisUtils} from "../lib/GenesisUtils.sol";
+import {IHasher} from "../interfaces/IHasher.sol";
 
 error SMTDepthIsGreaterThanMaxAllowed();
 error IdTypeNotSupported();
@@ -86,7 +87,8 @@ library IdentityLib {
         address _stateContractAddr,
         address _identityAddr,
         uint256 depth,
-        bytes2 idType
+        bytes2 idType,
+        IHasher hasher
     ) external {
         if (depth > IDENTITY_MAX_SMT_DEPTH) {
             revert SMTDepthIsGreaterThanMaxAllowed();
@@ -96,9 +98,9 @@ library IdentityLib {
             revert IdTypeNotSupported();
         }
         self.isOldStateGenesis = true;
-        self.trees.claimsTree.initialize(depth);
-        self.trees.revocationsTree.initialize(depth);
-        self.trees.rootsTree.initialize(depth);
+        self.trees.claimsTree.initialize(depth, hasher);
+        self.trees.revocationsTree.initialize(depth, hasher);
+        self.trees.rootsTree.initialize(depth, hasher);
         self.id = GenesisUtils.calcIdFromEthAddress(idType, _identityAddr);
     }