1#![allow(non_snake_case)]
 2#![allow(clippy::assertions_on_result_states)]
 3
 4extern crate libtestudo;
 5extern crate merlin;
 6use ark_crypto_primitives::sponge::poseidon::PoseidonConfig;
 7use ark_crypto_primitives::sponge::Absorb;
 8use ark_ec::pairing::Pairing;
 9use ark_ff::PrimeField;
10use ark_serialize::*;
11use libtestudo::parameters::PoseidonConfiguration;
12use libtestudo::poseidon_transcript::PoseidonTranscript;
13use libtestudo::{
14  testudo_snark::{TestudoSnark, TestudoSnarkGens},
15  Instance,
16};
17
18fn print(msg: &str) {
19  let star = "* ";
20  println!("{:indent$}{}{}", "", star, msg, indent = 2);
21}
22
23fn main() {
24  let params = ark_bls12_377::Fr::poseidon_params();
25  profiler::<ark_bls12_377::Bls12_377>(params);
26}
27
28fn profiler<E>(params: PoseidonConfig<E::ScalarField>)
29where
30  E: Pairing,
31  E::ScalarField: PrimeField,
32  E::ScalarField: Absorb,
33{
34  // the list of number of variables (and constraints) in an R1CS instance
35  let inst_sizes = vec![10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20];
36
37  println!("Profiler:: SNARK");
38  for &s in inst_sizes.iter() {
39    let num_vars = (2_usize).pow(s as u32);
40    let num_cons = num_vars;
41    let num_inputs = 10;
42
43    // produce a synthetic R1CSInstance
44    let (inst, vars, inputs) =
45      Instance::<E::ScalarField>::produce_synthetic_r1cs(num_cons, num_vars, num_inputs);
46
47    // produce public generators
48    let gens =
49      TestudoSnarkGens::<E>::setup(num_cons, num_vars, num_inputs, num_cons, params.clone());
50
51    // create a commitment to R1CSInstance
52    let (comm, decomm) = TestudoSnark::encode(&inst, &gens);
53
54    // produce a proof of satisfiability
55    let mut prover_transcript = PoseidonTranscript::new(&params.clone());
56    let proof = TestudoSnark::prove(
57      &inst,
58      &comm,
59      &decomm,
60      vars,
61      &inputs,
62      &gens,
63      &mut prover_transcript,
64      params.clone(),
65    )
66    .unwrap();
67
68    let mut proof_encoded = Vec::new();
69    proof
70      .serialize_with_mode(&mut proof_encoded, Compress::Yes)
71      .unwrap();
72    let msg_proof_len = format!(
73      "TestudoSnark::proof_compressed_len {:?}",
74      proof_encoded.len()
75    );
76    print(&msg_proof_len);
77
78    // verify the proof of satisfiability
79    let mut verifier_transcript = PoseidonTranscript::new(&params.clone());
80    assert!(proof
81      .verify(
82        &gens,
83        &comm,
84        &inputs,
85        &mut verifier_transcript,
86        params.clone()
87      )
88      .is_ok());
89
90    println!();
91  }
92}