use commonware_cryptography::{bls12381, Signer as _, Verifier as _};
use commonware_math::algebra::Random;
use criterion::{criterion_group, BatchSize, Criterion};
use rand::{thread_rng, Rng};
use std::hint::black_box;
fn bench_signature_verification(c: &mut Criterion) {
let namespace = b"namespace";
let mut msg = [0u8; 32];
thread_rng().fill(&mut msg);
c.bench_function(
&format!(
"{}/ns_len={} msg_len={}",
module_path!(),
namespace.len(),
msg.len()
),
|b| {
b.iter_batched(
|| {
let private_key = bls12381::PrivateKey::random(&mut thread_rng());
let public_key = private_key.public_key();
let signature = private_key.sign(namespace, &msg);
(public_key, signature)
},
|(public_key, signature)| black_box(public_key.verify(namespace, &msg, &signature)),
BatchSize::SmallInput,
);
},
);
}
criterion_group!(benches, bench_signature_verification);