fix(rln): incl userMessageLimit in validateRegistration

vacp2p · Jan 24, 2024 · 1c7e144 · 1c7e144
1 parent 5d9108a
commit 1c7e144
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/Rln.sol b/src/Rln.sol
@@ -13,7 +13,7 @@ contract Rln is RlnBase {
         RlnBase(membershipDeposit, depth, maxMessageLimit, _verifier)
     { }
 
-    function _validateRegistration(uint256 idCommitment) internal pure override { }
+    function _validateRegistration(uint256 idCommitment, uint256 userMessageLimit) internal pure override { }
 
     function _validateSlash(
         uint256 idCommitment,

diff --git a/src/RlnBase.sol b/src/RlnBase.sol
@@ -147,7 +147,7 @@ abstract contract RlnBase {
         if (msg.value != requiredDeposit) {
             revert InsufficientDeposit(MEMBERSHIP_DEPOSIT, msg.value);
         }
-        _validateRegistration(idCommitment);
+        _validateRegistration(idCommitment, userMessageLimit);
         _register(idCommitment, userMessageLimit, msg.value);
     }
 
@@ -170,7 +170,7 @@ abstract contract RlnBase {
     }
 
     /// @dev Inheriting contracts MUST override this function
-    function _validateRegistration(uint256 idCommitment) internal view virtual;
+    function _validateRegistration(uint256 idCommitment, uint256 userMessageLimit) internal view virtual;
 
     /// @dev Allows a user to slash a member
     /// @param idCommitment The idCommitment of the member