Merge pull request #25 from privacy-ethereum/feat/v1-spartan2

Add Spartan2 Fork Integration

Author: 0xVikasRushi

.vscode/settings.json

@@ -1,6 +1,5 @@
 {
     "rust-analyzer.linkedProjects": [
         "./wallet-unit-poc/ecdsa-spartan2/Cargo.toml",
-        "./wallet-unit-poc/forks/Spartan2/Cargo.toml"
     ]
 }

wallet-unit-poc/ecdsa-spartan2/Cargo.lock

@@ -5,7 +5,7 @@ edition = "2021"
 
 [dependencies]
 circom-scotia = "0.2.0"
-spartan2 = { path = "../forks/Spartan2", default-features = false }
+spartan2 = { git = "https://github.com/therealyingtong/Spartan2.git", branch = "zk", default-features = false }
 bellpepper = "0.4.0"
 bellpepper-core = "0.4.0"
 ff = { version = "0.13.0", features = ["derive"] }
@@ -15,3 +15,7 @@ tracing = "0.1"
 tracing-subscriber = { version = "0.3.18", features = ["env-filter", "time"] }
 sha2 = "0.10.7"
 bincode = "1.3.3"
+memmap2 = "0.9.8"
+
+[profile.release]
+debug = true

wallet-unit-poc/ecdsa-spartan2/Cargo.toml

@@ -0,0 +1,71 @@
+use std::time::Instant;
+
+use crate::{
+    ecdsa_circuit::ECDSACircuit,
+    jwt_circuit::JWTCircuit,
+    setup::{load_keys, load_proving_key},
+    E,
+};
+
+use spartan2::{traits::snark::R1CSSNARKTrait, zk_spartan::R1CSSNARK};
+use tracing::info;
+
+pub fn prove_ecdsa() {
+    let circuit = ECDSACircuit;
+    let pk_path = "keys/ecdsa_proving.key";
+    let vk_path = "keys/ecdsa_verifying.key";
+
+    let (pk, _vk) = match load_keys(pk_path, vk_path) {
+        Ok(keys) => keys,
+        Err(e) => {
+            eprintln!("Failed to load keys: {}", e);
+            panic!("Could not load keys: {}", e);
+        }
+    };
+
+    let t0 = Instant::now();
+    let mut prep_snark =
+        R1CSSNARK::<E>::prep_prove(&pk, circuit.clone(), false).expect("prep_prove failed");
+    let prep_ms = t0.elapsed().as_millis();
+    info!("ECDSA prep_prove: {} ms", prep_ms);
+
+    let t0 = Instant::now();
+    R1CSSNARK::<E>::prove(&pk, circuit.clone(), &mut prep_snark, false).expect("prove failed");
+    let sumcheck_ms = t0.elapsed().as_millis();
+
+    info!("ECDSA prove: {} ms", sumcheck_ms);
+
+    let total_ms = prep_ms + sumcheck_ms;
+    info!(
+        "ECDSA ZK-Spartan prove TOTAL: {} ms (~{:.1}s)",
+        total_ms,
+        total_ms as f64 / 1000.0
+    );
+}
+
+pub fn prove_jwt() {
+    let circuit = JWTCircuit;
+    let pk_path = "keys/jwt_proving.key";
+
+    // load_proving_chunked_key also can be used here
+    let pk = load_proving_key(pk_path).expect("load proving key failed");
+
+    let t0 = Instant::now();
+    let mut prep_snark =
+        R1CSSNARK::<E>::prep_prove(&pk, circuit.clone(), false).expect("prep_prove failed");
+    let prep_ms = t0.elapsed().as_millis();
+    info!("JWT ZK-Spartan prep_prove: {} ms", prep_ms);
+
+    let t0 = Instant::now();
+    R1CSSNARK::<E>::prove(&pk, circuit.clone(), &mut prep_snark, false).expect("prove failed");
+    let sumcheck_ms = t0.elapsed().as_millis();
+
+    info!("JWT ZK-Spartan prove: {} ms", sumcheck_ms);
+
+    let total_ms = prep_ms + sumcheck_ms;
+    info!(
+        "JWT ZK-Spartan prove TOTAL: {} ms (~{:.1}s)",
+        total_ms,
+        total_ms as f64 / 1000.0
+    );
+}