Feat/proving cycles (#42)

* Add total_steps * Update install_mips_rust_tool * Update revme dependencies * Update README for install toolchain * Update version
zkMIPS · Dec 20, 2024 · ebd1755 · ebd1755
1 parent 58e6aa3
commit ebd1755
Show file tree

Hide file tree

Showing 12 changed files with 18 additions and 70 deletions.
diff --git a/README.md b/README.md
@@ -11,6 +11,14 @@ There are two ways to prove the guest program:
 
 ![image](assets/temp-run-diagram.png)
 
+## Getting Started
+
+First to install zkm toolchain run the following command and follow the instructions:
+```sh
+curl --proto '=https' --tlsv1.2 -sSf https://raw.githubusercontent.com/zkMIPS/toolchain/refs/heads/main/setup.sh | sh
+source ~/.zkm-toolchain/env
+```
+
 ## Template code structure
 
 > [!NOTE]

diff --git a/guest-program/revme/Cargo.toml b/guest-program/revme/Cargo.toml
@@ -17,9 +17,6 @@ serde = { version = "1.0", default-features = false, features = ["alloc", "deriv
 serde_json = { version = "1.0", default-features = false, features = ["alloc"] }
 k256 = { version = "0.13.3", features = ["ecdsa"], default-features = false }
 
-# TODO can be removed once the powdr RISCV nightly is updated
-ahash = { version = "=0.8.6", default-features = false }
-
 libc = { version = "0.2" , features = ["extra_traits"] }
 
 [profile.dev]

diff --git a/host-program/mem-alloc-vec/Cargo.toml b/host-program/mem-alloc-vec/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "mem-alloc-vec"
-version = "0.1.0"
+version = "0.2.0"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

diff --git a/host-program/revme/Cargo.toml b/host-program/revme/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "revme"
-version = "0.1.0"
+version = "0.2.0"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

diff --git a/host-program/sha2-go/Cargo.toml b/host-program/sha2-go/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "sha2-go"
-version = "0.1.0"
+version = "0.2.0"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

diff --git a/host-program/sha2-rust/Cargo.toml b/host-program/sha2-rust/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "sha2-rust"
-version = "0.1.0"
+version = "0.2.0"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

diff --git a/install_mips_rust_tool b/install_mips_rust_tool
diff --git a/sdk/Cargo.toml b/sdk/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "zkm-sdk"
-version = "0.1.0"
+version = "0.2.0"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

diff --git a/sdk/src/local/stark.rs b/sdk/src/local/stark.rs
@@ -20,8 +20,9 @@ pub fn prove_stark(
     state.input_stream.push(input.public_inputstream.clone());
     state.input_stream.push(input.private_inputstream.clone());
 
-    let (_total_steps, seg_num, state) = split_prog_into_segs(state, &seg_path, "", seg_size);
+    let (total_steps, seg_num, state) = split_prog_into_segs(state, &seg_path, "", seg_size);
     result.output_stream = state.public_values_stream.clone();
+    result.total_steps = total_steps as u64;
     if input.execute_only {
         return Ok(false);
     }

diff --git a/sdk/src/network/prover.rs b/sdk/src/network/prover.rs
@@ -160,6 +160,7 @@ impl Prover for NetworkProver {
                         stark_proof: vec![],
                         solidity_verifier: vec![],
                         public_values: vec![],
+                        total_steps: get_status_response.total_steps,
                     };
                     if !get_status_response.stark_proof_url.is_empty() {
                         proof_result.stark_proof =

diff --git a/sdk/src/proto/stage.proto b/sdk/src/proto/stage.proto
@@ -73,4 +73,5 @@ message GetStatusResponse {
     bytes output_stream = 7;
     int32 step = 8; // Step
     string public_values_url = 9;
+    uint64 total_steps = 10;
 }
diff --git a/sdk/src/prover.rs b/sdk/src/prover.rs
@@ -28,6 +28,7 @@ pub struct ProverInput {
 
 #[derive(Debug, Default, Deserialize, Serialize, Clone)]
 pub struct ProverResult {
+    pub total_steps: u64,
     pub output_stream: Vec<u8>,
     pub proof_with_public_inputs: Vec<u8>,
     pub stark_proof: Vec<u8>,