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}