arnaucube
arnaucube/testudoPublic Self-hosted repository
testudo/profiler/snark.rs
profiler/snark.rsRaw
 1#![allow(non_snake_case)]
 2extern crate flate2;
 3extern crate libspartan;
 4extern crate merlin;
 5extern crate rand;
 6
 7use flate2::{write::ZlibEncoder, Compression};
 8use libspartan::r1csinstance::R1CSInstance;
 9use libspartan::spartan::{SNARKGens, SNARK};
10use libspartan::timer::Timer;
11use merlin::Transcript;
12
13pub fn main() {
14  // the list of number of variables (and constraints) in an R1CS instance
15  let inst_sizes = vec![12, 16, 20];
16
17  println!("Profiler:: SNARK");
18  for &s in inst_sizes.iter() {
19    let num_vars = (2 as usize).pow(s as u32);
20    let num_cons = num_vars;
21    let num_inputs = 10;
22    let (inst, vars, input) = R1CSInstance::produce_synthetic_r1cs(num_cons, num_vars, num_inputs);
23
24    Timer::print(&format!("number_of_constraints {}", num_cons));
25
26    // produce public generators
27    let gens = SNARKGens::new(&inst.size());
28
29    // create a commitment to R1CSInstance
30    let timer_encode = Timer::new("SNARK::encode");
31    let (comm, decomm) = SNARK::encode(&inst, &gens);
32    timer_encode.stop();
33
34    // produce a proof of satisfiability
35    let timer_prove = Timer::new("SNARK::prove");
36    let mut prover_transcript = Transcript::new(b"example");
37    let proof = SNARK::prove(&inst, &decomm, vars, &input, &gens, &mut prover_transcript);
38    timer_prove.stop();
39
40    let mut encoder = ZlibEncoder::new(Vec::new(), Compression::default());
41    bincode::serialize_into(&mut encoder, &proof).unwrap();
42    let proof_encoded = encoder.finish().unwrap();
43    let msg_proof_len = format!("SNARK::proof_compressed_len {:?}", proof_encoded.len());
44    Timer::print(&msg_proof_len);
45
46    // verify the proof of satisfiability
47    let timer_verify = Timer::new("SNARK::verify");
48    let mut verifier_transcript = Transcript::new(b"example");
49    assert!(proof
50      .verify(&comm, &input, &mut verifier_transcript, &gens)
51      .is_ok());
52    timer_verify.stop();
53
54    println!();
55  }
56}