1#![allow(non_snake_case)]
 2#![allow(clippy::assertions_on_result_states)]
 3
 4extern crate libspartan;
 5extern crate merlin;
 6
 7use ark_serialize::*;
 8use libspartan::parameters::poseidon_params;
 9use libspartan::poseidon_transcript::PoseidonTranscript;
10use libspartan::{Instance, SNARKGens, SNARK};
11
12fn print(msg: &str) {
13    let star = "* ";
14    println!("{:indent$}{}{}", "", star, msg, indent = 2);
15}
16
17pub fn main() {
18    // the list of number of variables (and constraints) in an R1CS instance
19    let inst_sizes = vec![10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20];
20
21    println!("Profiler:: SNARK");
22    for &s in inst_sizes.iter() {
23        let num_vars = (2_usize).pow(s as u32);
24        let num_cons = num_vars;
25        let num_inputs = 10;
26
27        // produce a synthetic R1CSInstance
28        let (inst, vars, inputs) = Instance::produce_synthetic_r1cs(num_cons, num_vars, num_inputs);
29
30        // produce public generators
31        let gens = SNARKGens::new(num_cons, num_vars, num_inputs, num_cons);
32
33        // create a commitment to R1CSInstance
34        let (comm, decomm) = SNARK::encode(&inst, &gens);
35
36        let params = poseidon_params();
37
38        // produce a proof of satisfiability
39        let mut prover_transcript = PoseidonTranscript::new(&params);
40        let proof = SNARK::prove(
41            &inst,
42            &comm,
43            &decomm,
44            vars,
45            &inputs,
46            &gens,
47            &mut prover_transcript,
48        );
49
50        let mut proof_encoded = Vec::new();
51        proof.serialize(&mut proof_encoded).unwrap();
52        let msg_proof_len = format!("SNARK::proof_compressed_len {:?}", proof_encoded.len());
53        print(&msg_proof_len);
54
55        // verify the proof of satisfiability
56        let mut verifier_transcript = PoseidonTranscript::new(&params);
57        assert!(proof
58            .verify(&comm, &inputs, &mut verifier_transcript, &gens)
59            .is_ok());
60
61        println!();
62    }
63}