feat(sdk): make scenario_end returns proof

kroma-network · Aug 16, 2024 · 4582ea2 · 4582ea2
1 parent 6814964
commit 4582ea2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/sdk/src/multi_prover/scenario/compress_prove.rs b/sdk/src/multi_prover/scenario/compress_prove.rs
@@ -98,7 +98,7 @@ pub fn scenario_end<T: Serialize + DeserializeOwned>(
  args: &ProveArgs<T>,
  core_proof: &Vec<u8>,
  compress_proof: &Vec<u8>,
-) {
+) -> Result<SP1ProofWithPublicValues> {
  let compress_proof_obj: SP1ReduceProof<BabyBearPoseidon2> =
  bincode::deserialize(compress_proof).unwrap();
 
@@ -114,4 +114,6 @@ pub fn scenario_end<T: Serialize + DeserializeOwned>(
 
  client.verify(&proof, &vk).unwrap();
  tracing::info!("Successfully verified compress proof");
+
+ Ok(proof)
 }
diff --git a/sdk/src/multi_prover/scenario/plonk_prove.rs b/sdk/src/multi_prover/scenario/plonk_prove.rs
@@ -4,7 +4,7 @@ use crate::multi_prover::{
  operator::{operator_prove_plonk, operator_prove_shrink},
 };
 use crate::{PlonkBn254Proof, SP1Proof, SP1ProofWithPublicValues};
-use anyhow::Result;
+use anyhow::{Ok, Result};
 use serde::{de::DeserializeOwned, Serialize};
 use sp1_prover::SP1CoreProof;
 use tracing::info_span;
@@ -34,7 +34,7 @@ pub fn scenario_end<T: Serialize + DeserializeOwned>(
  args: &ProveArgs<T>,
  core_proof: &Vec<u8>,
  plonk_proof: &Vec<u8>,
-) {
+) -> Result<SP1ProofWithPublicValues> {
  let plonk_proof: PlonkBn254Proof = bincode::deserialize(plonk_proof).unwrap();
 
  let (client, _, _, vk) = common::init_client(args);
@@ -49,4 +49,6 @@ pub fn scenario_end<T: Serialize + DeserializeOwned>(
 
  client.verify(&proof, &vk).unwrap();
  tracing::info!("Successfully verified compress proof");
+
+ Ok(proof)
 }