testudo/benches/r1cs.rs

use std::time::Instant;

use libspartan::{
    parameters::POSEIDON_PARAMETERS_FR_377, poseidon_transcript::PoseidonTranscript, Instance,
    NIZKGens, NIZK,
};
use serde::Serialize;

#[derive(Default, Clone, Serialize)]
struct BenchmarkResults {
    power: usize,
    input_constraints: usize,
    spartan_verifier_circuit_constraints: usize,
    r1cs_instance_generation_time: u128,
    spartan_proving_time: u128,
    groth16_setup_time: u128,
    groth16_proving_time: u128,
    testudo_verification_time: u128,
    testudo_proving_time: u128,
}

fn main() {
    let mut writer = csv::Writer::from_path("testudo.csv").expect("unable to open csv writer");
    // for &s in [
    //   10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26,
    // ]
    // .iter()
    // For testing purposes we currently bench on very small instance to ensure
    // correctness and then on biggest one for timings.
    for &s in [4, 26].iter() {
        println!("Running for {} inputs", s);
        let mut br = BenchmarkResults::default();
        let num_vars = (2_usize).pow(s as u32);
        let num_cons = num_vars;
        br.power = s;
        br.input_constraints = num_cons;
        let num_inputs = 10;

        let start = Instant::now();
        let (inst, vars, inputs) = Instance::produce_synthetic_r1cs(num_cons, num_vars, num_inputs);
        let duration = start.elapsed().as_millis();
        br.r1cs_instance_generation_time = duration;
        let mut prover_transcript = PoseidonTranscript::new(&POSEIDON_PARAMETERS_FR_377);

        let gens = NIZKGens::new(num_cons, num_vars, num_inputs);

        let start = Instant::now();
        let proof = NIZK::prove(&inst, vars, &inputs, &gens, &mut prover_transcript);
        let duration = start.elapsed().as_millis();
        br.spartan_proving_time = duration;

        let mut verifier_transcript = PoseidonTranscript::new(&POSEIDON_PARAMETERS_FR_377);
        let res = proof.verify(&inst, &inputs, &mut verifier_transcript, &gens);
        assert!(res.is_ok());
        br.spartan_verifier_circuit_constraints = res.unwrap();

        let mut verifier_transcript = PoseidonTranscript::new(&POSEIDON_PARAMETERS_FR_377);
        let res = proof.verify_groth16(&inst, &inputs, &mut verifier_transcript, &gens);
        assert!(res.is_ok());

        let (ds, dp, dv) = res.unwrap();
        br.groth16_setup_time = ds;
        br.groth16_proving_time = dp;

        br.testudo_proving_time = br.spartan_proving_time + br.groth16_proving_time;
        br.testudo_verification_time = dv;
        writer
            .serialize(br)
            .expect("unable to write results to csv");
        writer.flush().expect("wasn't able to flush");
    }
}