chore: add proof len consistency check (#1144)

Lightprotocol · Sep 2, 2024 · 51d076e · 51d076e
1 parent 1b2a327
commit 51d076e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/programs/account-compression/src/errors.rs b/programs/account-compression/src/errors.rs
@@ -54,4 +54,5 @@ pub enum AccountCompressionErrorCode {
     InvalidAccountBalance,
     UnsupportedAdditionalBytes,
     InvalidGroup,
+    ProofLengthMismatch,
 }
diff --git a/programs/account-compression/src/instructions/nullify_leaves.rs b/programs/account-compression/src/instructions/nullify_leaves.rs
@@ -62,6 +62,14 @@ pub fn process_nullify_leaves<'a, 'b, 'c: 'info, 'info>(
         msg!("only implemented for 1 nullifier update");
         return Err(AccountCompressionErrorCode::NumberOfProofsMismatch.into());
     }
+    if proofs.len() > 1 && proofs[0].len() != proofs[1].len() {
+        msg!(
+            "Proofs length mismatch {} {}",
+            proofs[0].len(),
+            proofs[1].len()
+        );
+        return Err(AccountCompressionErrorCode::ProofLengthMismatch.into());
+    }
     insert_nullifier(
         proofs,
         change_log_indices,