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::math::Math;
 9use libspartan::r1csinstance::{R1CSCommitmentGens, R1CSInstance};
10use libspartan::r1csproof::R1CSGens;
11use libspartan::spartan::{NIZKGens, SNARKGens, NIZK, SNARK};
12use libspartan::timer::Timer;
13use merlin::Transcript;
14
15pub fn main() {
16  // the list of number of variables (and constraints) in an R1CS instance
17  let inst_sizes = vec![12, 16, 20];
18
19  println!("Profiler:: SNARK");
20  for &s in inst_sizes.iter() {
21    let num_vars = (s as usize).pow2();
22    let num_cons = num_vars;
23    let num_inputs = 10;
24    let (inst, vars, input) = R1CSInstance::produce_synthetic_r1cs(num_cons, num_vars, num_inputs);
25
26    Timer::print(&format!("number_of_constraints {}", num_cons));
27
28    let r1cs_size = inst.size();
29    let gens_r1cs_eval = R1CSCommitmentGens::new(&r1cs_size, b"gens_r1cs_eval");
30
31    // create a commitment to R1CSInstance
32    let timer_encode = Timer::new("SNARK::encode");
33    let (comm, decomm) = SNARK::encode(&inst, &gens_r1cs_eval);
34    timer_encode.stop();
35
36    let gens_r1cs_sat = R1CSGens::new(num_cons, num_vars, b"gens_r1cs_sat");
37    let gens = SNARKGens::new(gens_r1cs_sat, gens_r1cs_eval);
38
39    // produce a proof of satisfiability
40    let timer_prove = Timer::new("SNARK::prove");
41    let mut prover_transcript = Transcript::new(b"example");
42    let proof = SNARK::prove(&inst, &decomm, vars, &input, &gens, &mut prover_transcript);
43    timer_prove.stop();
44
45    let mut encoder = ZlibEncoder::new(Vec::new(), Compression::default());
46    bincode::serialize_into(&mut encoder, &proof).unwrap();
47    let proof_encoded = encoder.finish().unwrap();
48    let msg_proof_len = format!("SNARK::proof_compressed_len {:?}", proof_encoded.len());
49    Timer::print(&msg_proof_len);
50
51    // verify the proof of satisfiability
52    let timer_verify = Timer::new("SNARK::verify");
53    let mut verifier_transcript = Transcript::new(b"example");
54    assert!(proof
55      .verify(&comm, &input, &mut verifier_transcript, &gens)
56      .is_ok());
57    timer_verify.stop();
58
59    println!();
60  }
61
62  println!("Profiler:: NIZK");
63  for &s in inst_sizes.iter() {
64    let num_vars = (s as usize).pow2();
65    let num_cons = num_vars;
66    let num_inputs = 10;
67    let (inst, vars, input) = R1CSInstance::produce_synthetic_r1cs(num_cons, num_vars, num_inputs);
68
69    Timer::print(&format!("number_of_constraints {}", num_cons));
70
71    let gens_r1cs_sat = R1CSGens::new(num_cons, num_vars, b"gens_r1cs_sat");
72    let gens = NIZKGens::new(gens_r1cs_sat);
73
74    // produce a proof of satisfiability
75    let timer_prove = Timer::new("NIZK::prove");
76    let mut prover_transcript = Transcript::new(b"example");
77    let proof = NIZK::prove(&inst, vars, &input, &gens, &mut prover_transcript);
78    timer_prove.stop();
79
80    let mut encoder = ZlibEncoder::new(Vec::new(), Compression::default());
81    bincode::serialize_into(&mut encoder, &proof).unwrap();
82    let proof_encoded = encoder.finish().unwrap();
83    let msg_proof_len = format!("NIZK::proof_compressed_len {:?}", proof_encoded.len());
84    Timer::print(&msg_proof_len);
85
86    // verify the proof of satisfiability
87    let timer_verify = Timer::new("NIZK::verify");
88    let mut verifier_transcript = Transcript::new(b"example");
89    assert!(proof
90      .verify(&inst, &input, &mut verifier_transcript, &gens)
91      .is_ok());
92    timer_verify.stop();
93
94    println!();
95  }
96}