[cryptography/bls12381] Implement BatchVerify (#915)

Author: patrick-ogrady

cryptography/src/bls12381/benches/batch_verify_multiple_messages.rs

@@ -0,0 +1,39 @@
+use commonware_cryptography::{bls12381::Bls12381Batch, BatchScheme, Bls12381, Signer};
+use criterion::{criterion_group, BatchSize, Criterion};
+use rand::{thread_rng, Rng};
+use std::hint::black_box;
+
+fn benchmark_batch_verify_multiple_messages(c: &mut Criterion) {
+    let namespace = b"namespace";
+    for n_messages in [1, 10, 100, 1000, 10000].into_iter() {
+        let mut msgs = Vec::with_capacity(n_messages);
+        for _ in 0..n_messages {
+            let mut msg = [0u8; 32];
+            thread_rng().fill(&mut msg);
+            msgs.push(msg);
+        }
+        c.bench_function(&format!("{}/msgs={}", module_path!(), n_messages), |b| {
+            b.iter_batched(
+                || {
+                    let mut batch = Bls12381Batch::new();
+                    let mut signer = Bls12381::new(&mut thread_rng());
+                    for msg in msgs.iter() {
+                        let sig = signer.sign(Some(namespace), msg);
+                        assert!(batch.add(Some(namespace), msg, &signer.public_key(), &sig));
+                    }
+                    batch
+                },
+                |batch| {
+                    black_box(batch.verify(&mut thread_rng()));
+                },
+                BatchSize::SmallInput,
+            );
+        });
+    }
+}
+
+criterion_group!(
+    name = benches;
+    config = Criterion::default().sample_size(10);
+    targets = benchmark_batch_verify_multiple_messages
+);

cryptography/src/bls12381/benches/batch_verify_multiple_public_keys.rs

@@ -0,0 +1,35 @@
+use commonware_cryptography::{bls12381::Bls12381Batch, BatchScheme, Bls12381, Signer};
+use criterion::{criterion_group, BatchSize, Criterion};
+use rand::{thread_rng, Rng};
+use std::hint::black_box;
+
+fn benchmark_batch_verify_multiple_public_keys(c: &mut Criterion) {
+    let namespace = b"namespace";
+    let mut msg = [0u8; 32];
+    thread_rng().fill(&mut msg);
+    for n_signers in [1, 10, 100, 1000, 10000].into_iter() {
+        c.bench_function(&format!("{}/pks={}", module_path!(), n_signers), |b| {
+            b.iter_batched(
+                || {
+                    let mut batch = Bls12381Batch::new();
+                    for _ in 0..n_signers {
+                        let mut signer = Bls12381::new(&mut thread_rng());
+                        let sig = signer.sign(Some(namespace), &msg);
+                        assert!(batch.add(Some(namespace), &msg, &signer.public_key(), &sig));
+                    }
+                    batch
+                },
+                |batch| {
+                    black_box(batch.verify(&mut thread_rng()));
+                },
+                BatchSize::SmallInput,
+            );
+        });
+    }
+}
+
+criterion_group!(
+    name = benches;
+    config = Criterion::default().sample_size(10);
+    targets = benchmark_batch_verify_multiple_public_keys
+);

cryptography/src/bls12381/benches/bench.rs

@@ -4,6 +4,8 @@ mod aggregate_public_keys;
 mod aggregate_signatures;
 mod aggregate_verify_multiple_messages;
 mod aggregate_verify_multiple_public_keys;
+mod batch_verify_multiple_messages;
+mod batch_verify_multiple_public_keys;
 mod dkg_recovery;
 mod dkg_reshare_recovery;
 mod signature_generation;
@@ -20,4 +22,6 @@ criterion_main!(
     signature_verification::benches,
     aggregate_verify_multiple_messages::benches,
     aggregate_verify_multiple_public_keys::benches,
+    batch_verify_multiple_messages::benches,
+    batch_verify_multiple_public_keys::benches,
 );

cryptography/src/bls12381/mod.rs

@@ -9,4 +9,4 @@
 pub mod dkg;
 pub mod primitives;
 mod scheme;
-pub use scheme::{Bls12381, PrivateKey, PublicKey, Signature};
+pub use scheme::{Bls12381, Bls12381Batch, PrivateKey, PublicKey, Signature};