1use std::time::Instant;
 2
 3use libspartan::{
 4  parameters::POSEIDON_PARAMETERS_FR_377, poseidon_transcript::PoseidonTranscript, Instance, NIZK,
 5};
 6use serde::Serialize;
 7
 8#[derive(Default, Clone, Serialize)]
 9struct BenchmarkResults {
10  power: usize,
11  input_constraints: usize,
12  spartan_verifier_circuit_constraints: usize,
13  r1cs_instance_generation_time: u128,
14  spartan_proving_time: u128,
15  groth16_setup_time: u128,
16  groth16_proving_time: u128,
17  groth16_verification_time: u128,
18  testudo_proving_time: u128,
19}
20
21fn main() {
22  let mut writer = csv::Writer::from_path("testudo.csv").expect("unable to open csv writer");
23  for &s in [
24    10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26,
25  ]
26  .iter()
27  {
28    println!("Running for {} inputs", s);
29    let mut br = BenchmarkResults::default();
30    let num_vars = (2_usize).pow(s as u32);
31    let num_cons = num_vars;
32    br.power = s;
33    br.input_constraints = num_cons;
34    let num_inputs = 10;
35
36    let start = Instant::now();
37    let (inst, vars, inputs) = Instance::produce_synthetic_r1cs(num_cons, num_vars, num_inputs);
38    let duration = start.elapsed().as_millis();
39    br.r1cs_instance_generation_time = duration;
40    let mut prover_transcript = PoseidonTranscript::new(&POSEIDON_PARAMETERS_FR_377);
41    let start = Instant::now();
42    let proof = NIZK::prove(&inst, vars, &inputs, &mut prover_transcript);
43    let duration = start.elapsed().as_millis();
44    println!("{:?}", duration);
45    br.spartan_proving_time = duration;
46
47    let mut verifier_transcript = PoseidonTranscript::new(&POSEIDON_PARAMETERS_FR_377);
48    let res = proof.verify(&inst, &inputs, &mut verifier_transcript);
49    assert!(res.is_ok());
50    br.spartan_verifier_circuit_constraints = res.unwrap();
51
52    let mut verifier_transcript = PoseidonTranscript::new(&POSEIDON_PARAMETERS_FR_377);
53    let res = proof.verify_groth16(&inst, &inputs, &mut verifier_transcript);
54    assert!(res.is_ok());
55
56    let (ds, dp, dv) = res.unwrap();
57    br.groth16_setup_time = ds;
58    br.groth16_proving_time = dp;
59    br.groth16_verification_time = dv;
60
61    br.testudo_proving_time = br.spartan_proving_time + br.groth16_proving_time;
62
63    writer
64      .serialize(br)
65      .expect("unable to write results to csv");
66    writer.flush().expect("wasn't able to flush");
67  }
68}