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;
 5
 6use flate2::{write::ZlibEncoder, Compression};
 7use libspartan::{Instance, SNARKGens, SNARK};
 8use merlin::Transcript;
 9
10fn print(msg: &str) {
11  let star = "* ";
12  println!("{:indent$}{}{}", "", star, msg, indent = 2);
13}
14
15pub fn main() {
16  // the list of number of variables (and constraints) in an R1CS instance
17  let inst_sizes = vec![10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20];
18
19  println!("Profiler:: SNARK");
20  for &s in inst_sizes.iter() {
21    let num_vars = (2_usize).pow(s as u32);
22    let num_cons = num_vars;
23    let num_inputs = 10;
24
25    // produce a synthetic R1CSInstance
26    let (inst, vars, inputs) = Instance::produce_synthetic_r1cs(num_cons, num_vars, num_inputs);
27
28    // produce public generators
29    let gens = SNARKGens::new(num_cons, num_vars, num_inputs, num_cons);
30
31    // create a commitment to R1CSInstance
32    let (comm, decomm) = SNARK::encode(&inst, &gens);
33
34    // produce a proof of satisfiability
35    let mut prover_transcript = Transcript::new(b"snark_example");
36    let proof = SNARK::prove(
37      &inst,
38      &comm,
39      &decomm,
40      vars,
41      &inputs,
42      &gens,
43      &mut prover_transcript,
44    );
45
46    let mut encoder = ZlibEncoder::new(Vec::new(), Compression::default());
47    bincode::serialize_into(&mut encoder, &proof).unwrap();
48    let proof_encoded = encoder.finish().unwrap();
49    let msg_proof_len = format!("SNARK::proof_compressed_len {:?}", proof_encoded.len());
50    print(&msg_proof_len);
51
52    // verify the proof of satisfiability
53    let mut verifier_transcript = Transcript::new(b"snark_example");
54    assert!(proof
55      .verify(&comm, &inputs, &mut verifier_transcript, &gens)
56      .is_ok());
57
58    println!();
59  }
60}