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