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