feature: range check added

vacp2p · Aug 28, 2023 · 5493bd8 · 5493bd8
1 parent ea1ac53
commit 5493bd8
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/rln/src/protocol.rs b/rln/src/protocol.rs
@@ -626,6 +626,11 @@ pub fn generate_proof_with_witness(
     Ok(proof)
 }
 
+/// Formats inputs for witness calculation
+///
+/// # Errors
+///
+/// Returns an error if `rln_witness.message_id` is not within `rln_witness.user_message_limit`.
 pub fn inputs_for_witness_calculation(
     rln_witness: &RLNWitnessInput,
 ) -> Result<[(&str, Vec<BigInt>); 8]> {
@@ -643,6 +648,10 @@ pub fn inputs_for_witness_calculation(
         .iter()
         .for_each(|v| identity_path_index.push(BigInt::from(*v)));
 
+    if rln_witness.message_id > rln_witness.user_message_limit {
+        return Err(color_eyre::Report::msg("message_id is not within user_message_limit"))
+    }
+
     Ok([
         (
             "identity_secret",