From 1c7e14482b1fd0ee30aa0522446aad8c1319e712 Mon Sep 17 00:00:00 2001
From: rymnc <43716372+rymnc@users.noreply.github.com>
Date: Wed, 24 Jan 2024 11:38:28 +0530
Subject: [PATCH] fix(rln): incl userMessageLimit in validateRegistration

---
 src/Rln.sol     | 2 +-
 src/RlnBase.sol | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/Rln.sol b/src/Rln.sol
index 31d7978..9a988e2 100644
--- 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
index 57c045b..4f71e25 100644
--- 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