Fix Cargo.toml and add benchmark for sqrt pst

Cargo.toml

@@ -52,7 +52,7 @@ csv = "1.1.5"
 criterion = "0.3.6"
 
 [lib]
-name = "libspartan"
+name = "libtestudo"
 path = "src/lib.rs"
 
 [[bin]]
@@ -63,6 +63,10 @@ path = "profiler/testudo.rs"
 name = "testudo"
 harness = false
 
+[[bench]]
+name = "pst"
+harness = false
+
 [features]
 multicore = ["rayon"]
 profile = []
@@ -75,6 +79,7 @@ std = ["ark-ff/std", "ark-ec/std", "ark-std/std", "ark-relations/std", "ark-seri
 ark-poly-commit = {git = "https://github.com/cryptonetlab/ark-polycommit", branch="feat/variable-crs"}
 ark-groth16 = { git = "https://github.com/arkworks-rs/groth16" }
 blstrs = { git = "https://github.com/nikkolasg/blstrs", branch = "feat/arkwork" }
-ark-ec = { git = "https://github.com/vmx/algebra", branch = "affine-repr-xy-owned" }
-ark-ff = { git = "https://github.com/vmx/algebra", branch = "affine-repr-xy-owned" }
-ark-serialize = { git = "https://github.com/vmx/algebra", branch = "affine-repr-xy-owned" }
+ark-ec = { git = "https://github.com/vmx/algebra", branch="affine-repr-xy-owned" }
+ark-ff = { git = "https://github.com/vmx/algebra", branch="affine-repr-xy-owned" }
+ark-poly = { git = "https://github.com/vmx/algebra", branch="affine-repr-xy-owned" }
+ark-serialize = { git = "https://github.com/vmx/algebra", branch="affine-repr-xy-owned" }

benches/pst.rs

@@ -0,0 +1,98 @@
+use std::time::Instant;
+
+use ark_poly_commit::multilinear_pc::MultilinearPC;
+use ark_serialize::CanonicalSerialize;
+use libtestudo::{
+  parameters::PoseidonConfiguration, poseidon_transcript::PoseidonTranscript, sqrt_pst::Polynomial,
+};
+use serde::Serialize;
+type F = ark_bls12_377::Fr;
+type E = ark_bls12_377::Bls12_377;
+use ark_std::UniformRand;
+
+#[derive(Default, Clone, Serialize)]
+struct BenchmarkResults {
+  power: usize,
+  commit_time: u128,
+  opening_time: u128,
+  verification_time: u128,
+  proof_size: usize,
+  commiter_key_size: usize,
+}
+fn main() {
+  let params = ark_bls12_377::Fr::poseidon_params();
+
+  let mut writer = csv::Writer::from_path("sqrt_pst.csv").expect("unable to open csv writer");
+  for &s in [4, 5, 20, 27].iter() {
+    println!("Running for {} inputs", s);
+    let mut rng = ark_std::test_rng();
+    let mut br = BenchmarkResults::default();
+    br.power = s;
+    let num_vars = s;
+    let len = 2_usize.pow(num_vars as u32);
+    let z: Vec<F> = (0..len).into_iter().map(|_| F::rand(&mut rng)).collect();
+    let r: Vec<F> = (0..num_vars)
+      .into_iter()
+      .map(|_| F::rand(&mut rng))
+      .collect();
+
+    let setup_vars = (num_vars as f32 / 2.0).ceil() as usize;
+    let gens = MultilinearPC::<E>::setup((num_vars as f32 / 2.0).ceil() as usize, &mut rng);
+    let (ck, vk) = MultilinearPC::<E>::trim(&gens, setup_vars);
+
+    let mut cks = Vec::<u8>::new();
+    ck.serialize_with_mode(&mut cks, ark_serialize::Compress::Yes)
+      .unwrap();
+    br.commiter_key_size = cks.len();
+
+    let mut pl = Polynomial::from_evaluations(&z.clone());
+
+    let v = pl.eval(&r);
+
+    let start = Instant::now();
+    let (comm_list, t) = pl.commit(&ck);
+    let duration = start.elapsed().as_millis();
+    br.commit_time = duration;
+
+    let mut prover_transcript = PoseidonTranscript::new(&params);
+
+    let start = Instant::now();
+    let (u, pst_proof, mipp_proof) = pl.open(&mut prover_transcript, comm_list, &ck, &r, &t);
+    let duration = start.elapsed().as_millis();
+    br.opening_time = duration;
+
+    let mut p1 = Vec::<u8>::new();
+    let mut p2 = Vec::<u8>::new();
+    pst_proof
+      .serialize_with_mode(&mut p1, ark_serialize::Compress::Yes)
+      .unwrap();
+
+    mipp_proof
+      .serialize_with_mode(&mut p2, ark_serialize::Compress::Yes)
+      .unwrap();
+
+    br.proof_size = p1.len() + p2.len();
+
+    let mut verifier_transcript = PoseidonTranscript::new(&params);
+
+    let start = Instant::now();
+    let res = Polynomial::verify(
+      &mut verifier_transcript,
+      &vk,
+      &u,
+      &r,
+      v,
+      &pst_proof,
+      &mipp_proof,
+      &t,
+    );
+    let duration = start.elapsed().as_millis();
+    br.verification_time = duration;
+    assert!(res == true);
+
+    writer
+      .serialize(br)
+      .expect("unable to write results to csv");
+    writer.flush().expect("wasn't able to flush");
+  }
+}

benches/testudo.rs

@@ -5,8 +5,8 @@ use ark_crypto_primitives::sponge::Absorb;
 use ark_ec::pairing::Pairing;
 use ark_ff::PrimeField;
 use ark_serialize::*;
-use libspartan::parameters::PoseidonConfiguration;
-use libspartan::{
+use libtestudo::parameters::PoseidonConfiguration;
+use libtestudo::{
   poseidon_transcript::PoseidonTranscript,
   testudo_snark::{TestudoSnark, TestudoSnarkGens},
   Instance,

examples/cubic.rs

@@ -11,8 +11,8 @@
 use ark_ec::pairing::Pairing;
 use ark_ff::{BigInteger, PrimeField};
 use ark_std::{One, UniformRand, Zero};
-use libspartan::testudo_snark::{TestudoSnark, TestudoSnarkGens};
-use libspartan::{
+use libtestudo::testudo_snark::{TestudoSnark, TestudoSnarkGens};
+use libtestudo::{
   parameters::poseidon_params, poseidon_transcript::PoseidonTranscript, InputsAssignment, Instance,
   VarsAssignment,
 };

profiler/testudo.rs

@@ -1,16 +1,16 @@
 #![allow(non_snake_case)]
 #![allow(clippy::assertions_on_result_states)]
 
-extern crate libspartan;
+extern crate libtestudo;
 extern crate merlin;
 use ark_crypto_primitives::sponge::poseidon::PoseidonConfig;
 use ark_crypto_primitives::sponge::Absorb;
 use ark_ec::pairing::Pairing;
 use ark_ff::PrimeField;
 use ark_serialize::*;
-use libspartan::parameters::PoseidonConfiguration;
-use libspartan::poseidon_transcript::PoseidonTranscript;
-use libspartan::{
+use libtestudo::parameters::PoseidonConfiguration;
+use libtestudo::poseidon_transcript::PoseidonTranscript;
+use libtestudo::{
   testudo_snark::{TestudoSnark, TestudoSnarkGens},
   Instance,
 };

src/lib.rs

@@ -28,7 +28,7 @@ mod product_tree;
 mod r1csinstance;
 mod r1csproof;
 mod sparse_mlpoly;
-mod sqrt_pst;
+pub mod sqrt_pst;
 mod sumcheck;
 pub mod testudo_nizk;
 pub mod testudo_snark;