106 lines
3.0 KiB
Rust
106 lines
3.0 KiB
Rust
#[macro_use]
|
|
extern crate criterion;
|
|
|
|
use bellman::groth16::*;
|
|
use bls12_381::Bls12;
|
|
use criterion::Criterion;
|
|
use group::ff::Field;
|
|
use rand::{Rng, RngCore, SeedableRng};
|
|
use rand_xorshift::XorShiftRng;
|
|
use sapling_crypto::{
|
|
circuit::{Spend, ValueCommitmentOpening},
|
|
keys::ExpandedSpendingKey,
|
|
value::NoteValue,
|
|
Diversifier,
|
|
};
|
|
|
|
#[cfg(unix)]
|
|
use pprof::criterion::{Output, PProfProfiler};
|
|
|
|
const TREE_DEPTH: usize = 32;
|
|
|
|
fn criterion_benchmark(c: &mut Criterion) {
|
|
let mut rng = XorShiftRng::from_seed([
|
|
0x59, 0x62, 0xbe, 0x3d, 0x76, 0x3d, 0x31, 0x8d, 0x17, 0xdb, 0x37, 0x32, 0x54, 0x06, 0xbc,
|
|
0xe5,
|
|
]);
|
|
|
|
let groth_params = generate_random_parameters::<Bls12, _, _>(
|
|
Spend {
|
|
value_commitment_opening: None,
|
|
proof_generation_key: None,
|
|
payment_address: None,
|
|
commitment_randomness: None,
|
|
ar: None,
|
|
auth_path: vec![None; TREE_DEPTH],
|
|
anchor: None,
|
|
},
|
|
&mut rng,
|
|
)
|
|
.unwrap();
|
|
|
|
c.bench_function("sapling-spend-prove", |b| {
|
|
let value_commitment = ValueCommitmentOpening {
|
|
value: NoteValue::from_raw(1),
|
|
randomness: jubjub::Fr::random(&mut rng),
|
|
};
|
|
|
|
let sk: [u8; 32] = rng.gen();
|
|
let expsk = ExpandedSpendingKey::from_spending_key(&sk);
|
|
|
|
let proof_generation_key = expsk.proof_generation_key();
|
|
|
|
let viewing_key = proof_generation_key.to_viewing_key();
|
|
|
|
let payment_address = loop {
|
|
let diversifier = {
|
|
let mut d = [0; 11];
|
|
rng.fill_bytes(&mut d);
|
|
Diversifier(d)
|
|
};
|
|
|
|
if let Some(p) = viewing_key.to_payment_address(diversifier) {
|
|
break p;
|
|
}
|
|
};
|
|
|
|
let commitment_randomness = jubjub::Fr::random(&mut rng);
|
|
let auth_path =
|
|
vec![Some((bls12_381::Scalar::random(&mut rng), rng.next_u32() % 2 != 0)); TREE_DEPTH];
|
|
let ar = jubjub::Fr::random(&mut rng);
|
|
let anchor = bls12_381::Scalar::random(&mut rng);
|
|
|
|
b.iter(|| {
|
|
create_random_proof(
|
|
Spend {
|
|
value_commitment_opening: Some(value_commitment.clone()),
|
|
proof_generation_key: Some(proof_generation_key.clone()),
|
|
payment_address: Some(payment_address),
|
|
commitment_randomness: Some(commitment_randomness),
|
|
ar: Some(ar),
|
|
auth_path: auth_path.clone(),
|
|
anchor: Some(anchor),
|
|
},
|
|
&groth_params,
|
|
&mut rng,
|
|
)
|
|
});
|
|
});
|
|
}
|
|
|
|
#[cfg(unix)]
|
|
criterion_group! {
|
|
name = benches;
|
|
config = Criterion::default()
|
|
.sample_size(10)
|
|
.with_profiler(PProfProfiler::new(100, Output::Flamegraph(None)));
|
|
targets = criterion_benchmark
|
|
}
|
|
#[cfg(windows)]
|
|
criterion_group! {
|
|
name = benches;
|
|
config = Criterion::default().sample_size(10);
|
|
targets = criterion_benchmark
|
|
}
|
|
criterion_main!(benches);
|