add contract (#8)

sdk/Cargo.toml

@@ -27,6 +27,7 @@ plonky2_maybe_rayon = { git = "https://github.com/zkMIPS/plonky2.git", branch =
 
 tonic = "0.8.1"
 prost = "0.11.0"
+reqwest = { version = "0.11", features = ["rustls-tls"] }
 tokio = { version = "1.21.0", features = ["macros", "rt-multi-thread", "signal"] }
 ethers = "2.0.14"
 

sdk/src/network/prover.rs

@@ -79,6 +79,12 @@ impl NetworkProver {
         let signature = self.wallet.sign_message(sign_data).await.unwrap();
         request.signature = signature.to_string();
     }
+
+    pub async fn download_file(url: &str) -> anyhow::Result<Vec<u8>> {
+        let response = reqwest::get(url).await?;
+        let content = response.bytes().await?;
+        Ok(content.to_vec())
+    }
 }
 
 #[async_trait]
@@ -121,12 +127,20 @@ impl Prover for NetworkProver {
 
             match Status::from_i32(get_status_response.status as i32) {
                 Some(Status::Computing) => {
+                    log::debug!("generate_proof step: {}", get_status_response.step);
                     sleep(Duration::from_secs(2)).await;
                 }
                 Some(Status::Success) => {
+                    let stark_proof =
+                        NetworkProver::download_file(&get_status_response.stark_proof_url).await?;
+                    let solidity_verifier =
+                        NetworkProver::download_file(&get_status_response.solidity_verifier_url)
+                            .await?;
                     let proof_result = ProverResult {
                         output_stream: get_status_response.output_stream,
                         proof_with_public_inputs: get_status_response.proof_with_public_inputs,
+                        stark_proof,
+                        solidity_verifier,
                     };
                     return Ok(Some(proof_result));
                 }

sdk/src/proto/stage.proto

@@ -20,6 +20,16 @@ enum Status {
     FINAL_ERROR = 8;
 }
 
+enum Step {
+    Init = 0;
+    InSplit = 1;
+    InProve = 2;
+    InAgg = 3;
+    InAggAll = 4;
+    InFinal = 5;
+    End = 6;
+}
+
 message BlockFileItem {
     string file_name = 1;
     bytes file_content = 2;
@@ -60,4 +70,5 @@ message GetStatusResponse {
     string stark_proof_url = 5;
     string solidity_verifier_url = 6;
     bytes output_stream = 7;
+    int32 step = 8; // Step
 }

sdk/src/prover.rs

@@ -15,8 +15,9 @@ pub struct ProverInput {
 #[derive(Debug, Default, Deserialize, Serialize)]
 pub struct ProverResult {
     pub output_stream: Vec<u8>,
-    // pub stark_proof: Vec<u8>,
     pub proof_with_public_inputs: Vec<u8>,
+    pub stark_proof: Vec<u8>,
+    pub solidity_verifier: Vec<u8>,
 }
 
 #[async_trait]