add storage proof implementation to main, add proof generation to e2e…

… script

saffron/src/main.rs

@@ -1,12 +1,15 @@
 use anyhow::Result;
 use ark_poly::{EvaluationDomain, Radix2EvaluationDomain};
 use clap::Parser;
-use mina_curves::pasta::{Fp, Vesta};
-use poly_commitment::{ipa::SRS, SRS as _};
+use kimchi::groupmap::GroupMap;
+use mina_curves::pasta::{Fp, Vesta, VestaParameters};
+use mina_poseidon::{constants::PlonkSpongeConstantsKimchi, sponge::DefaultFqSponge};
+use poly_commitment::{commitment::CommitmentCurve, ipa::SRS, SRS as _};
+use rand::rngs::OsRng;
 use saffron::{
     blob::FieldBlob,
     cli::{self, HexString},
-    commitment, env, utils,
+    commitment, env, proof, utils,
 };
 use sha3::{Digest, Sha3_256};
 use std::{
@@ -102,6 +105,24 @@ pub fn compute_commitment(args: cli::ComputeCommitmentArgs) -> Result<HexString>
     Ok(HexString(hash))
 }
 
+pub fn storage_proof(args: cli::StorageProofArgs) -> Result<HexString> {
+    let file = File::open(args.input)?;
+    let blob: FieldBlob<Vesta> = rmp_serde::decode::from_read(file)?;
+    let proof =
+        {
+            let (srs, _) = get_srs(args.srs_cache);
+            let group_map = <Vesta as CommitmentCurve>::Map::setup();
+            let mut rng = OsRng;
+            let evaluation_point = utils::encode(&args.challenge.0);
+            proof::storage_proof::<
+                Vesta,
+                DefaultFqSponge<VestaParameters, PlonkSpongeConstantsKimchi>,
+            >(&srs, &group_map, blob, evaluation_point, &mut rng)
+        };
+    let bytes = rmp_serde::to_vec(&proof).unwrap();
+    Ok(HexString(bytes))
+}
+
 pub fn main() -> Result<()> {
     env::init_console_subscriber();
     let args = cli::Commands::parse();
@@ -113,5 +134,10 @@ pub fn main() -> Result<()> {
             println!("{}", commitment);
             Ok(())
         }
+        cli::Commands::StorageProof(args) => {
+            let proof = storage_proof(args)?;
+            println!("{}", proof);
+            Ok(())
+        }
     }
 }

saffron/src/utils.rs

@@ -3,7 +3,7 @@ use ark_poly::EvaluationDomain;
 
 // For injectivity, you can only use this on inputs of length at most
 // 'F::MODULUS_BIT_SIZE / 8', e.g. for Vesta this is 31.
-fn encode<Fp: PrimeField>(bytes: &[u8]) -> Fp {
+pub fn encode<Fp: PrimeField>(bytes: &[u8]) -> Fp {
     Fp::from_be_bytes_mod_order(bytes)
 }
 

saffron/test-encoding.sh

@@ -31,6 +31,19 @@ if ! cargo run --release --bin saffron encode -i "$INPUT_FILE" -o "$ENCODED_FILE
    exit 1
 fi
 
+# Generate 32-byte random challenge as hex string
+echo "Generating random challenge..."
+CHALLENGE=$(head -c 32 /dev/urandom | xxd -p -c 32)
+echo "Challenge: $CHALLENGE"
+
+# Generate storage proof and capture proof output
+echo "Generating storage proof..."
+PROOF=$(cargo run --release --bin saffron storage-proof -i "$ENCODED_FILE" --challenge "$CHALLENGE" $SRS_ARG | tee /dev/stderr | tail -n 1)
+if [ $? -ne 0 ]; then
+    echo "Storage proof generation failed"
+    exit 1
+fi
+
 # Run decode
 echo "Decoding $ENCODED_FILE to $DECODED_FILE"
 if ! cargo run --release --bin saffron decode -i "$ENCODED_FILE" -o "$DECODED_FILE" $SRS_ARG; then