Add scarb verify (#1922)

## Stack
- #1924
- #1900
- #1922 


## Issues
Closes #1815

---------

Signed-off-by: Maksim Zdobnikau <[email protected]>
Co-authored-by: FroyaTheHen <[email protected]>
Co-authored-by: maciektr <[email protected]>

Author: 3 people

.github/workflows/ci.yml

@@ -29,7 +29,7 @@ jobs:
       - uses: Swatinem/rust-cache@v2
       - uses: taiki-e/install-action@nextest
       - name: nextest archive
-        run: cargo nextest archive --workspace --all-features --cargo-profile ci --archive-file 'nextest-archive-${{ matrix.platform.os }}.tar.zst' --exclude scarb-prove
+        run: cargo nextest archive --workspace --all-features --cargo-profile ci --archive-file 'nextest-archive-${{ matrix.platform.os }}.tar.zst' --exclude scarb-prove --exclude scarb-verify
       - uses: actions/upload-artifact@v4
         with:
           name: nextest-archive-${{ matrix.platform.os }}
@@ -82,8 +82,8 @@ jobs:
       - name: run tests
         run: cargo test -p scarb-metadata
 
-  test-scarb-prove:
-    name: test scarb-prove ${{ matrix.platform.name }}
+  test-nightly-crates:
+    name: test nightly crates ${{ matrix.platform.name }}
     runs-on: ${{ matrix.platform.os }}
     env:
       # TODO(#1915): Use stable toolchain once stwo is stable.
@@ -104,9 +104,9 @@ jobs:
           toolchain: ${{ env.RUST_TOOLCHAIN }}
       - uses: Swatinem/rust-cache@v2
       - name: Build
-        run: cargo +stable build --workspace --all-features --exclude scarb-prove
-      - name: Run scarb-prove tests
-        run: cargo +${{ env.RUST_TOOLCHAIN }} test -p scarb-prove
+        run: cargo +stable build --workspace --all-features --exclude scarb-prove --exclude scarb-verify
+      - name: Run scarb-prove and scarb-verify tests
+        run: cargo +${{ env.RUST_TOOLCHAIN }} test -p scarb-prove -p scarb-verify
 
   test-prebuilt-plugins:
     name: test prebuilt plugins ${{ matrix.platform.name }}
@@ -140,17 +140,17 @@ jobs:
       - uses: Swatinem/rust-cache@v2
       - run: cargo fmt --check
       # TODO(#1915): Build all crates with stable toolchain once stwo is stable.
-      - run: cargo clippy --all-targets --all-features --workspace --exclude scarb-prove -- --no-deps
+      - run: cargo clippy --all-targets --all-features --workspace --exclude scarb-prove --exclude scarb-verify -- --no-deps
         env:
           # Make sure CI fails on all warnings, including Clippy lints.
           RUSTFLAGS: "-Dwarnings"
-      - run: cargo doc --all-features --no-deps --workspace --exclude scarb-prove
+      - run: cargo doc --all-features --no-deps --workspace --exclude scarb-prove --exclude scarb-verify
         env:
           # Make sure CI fails on all warnings, including Clippy lints.
           RUSTDOCFLAGS: "-Dwarnings"
 
-  check-rust-scarb-prove:
-    name: check-rust (scarb-prove)
+  check-rust-nightly:
+    name: check-rust (nightly)
     runs-on: ubuntu-latest
     env:
       # TODO(#1915): Use stable toolchain once stwo is stable.
@@ -162,14 +162,14 @@ jobs:
           toolchain: ${{ env.RUST_TOOLCHAIN }}
           components: rustfmt, clippy
       - uses: Swatinem/rust-cache@v2
-      - run: cargo +${{ env.RUST_TOOLCHAIN }} clippy --all-targets --all-features -p scarb-prove -- --no-deps
+      - run: cargo +${{ env.RUST_TOOLCHAIN }} clippy --all-targets --all-features -p scarb-prove -p scarb-verify -- --no-deps
         env:
           # Make sure CI fails on all warnings, including Clippy lints.
           RUSTFLAGS: "-Dwarnings"
-      - run: cargo +${{ env.RUST_TOOLCHAIN }} doc --all-features --no-deps -p scarb-prove
+      - run: cargo +${{ env.RUST_TOOLCHAIN }} doc --all-features --no-deps -p scarb-prove -p scarb-verify
         env:
           # Make sure CI fails on all warnings, including Clippy lints.
-          RUSTDOCFLAGS: "-Dwarnings"      
+          RUSTDOCFLAGS: "-Dwarnings"
 
   check-website:
     runs-on: ubuntu-latest

Cargo.lock

@@ -5,6 +5,7 @@ members = [
     "extensions/scarb-doc",
     "extensions/scarb-execute",
     "extensions/scarb-prove",
+    "extensions/scarb-verify",
     "extensions/scarb-cairo-language-server",
     "extensions/scarb-cairo-run",
     "extensions/scarb-cairo-test",
@@ -76,6 +77,8 @@ cairo-lang-utils = { version = "*", features = ["env_logger"] }
 cairo-language-server = "*"
 cairo-lint-core = "*"
 cairo-vm = "1.0.1"
+stwo_cairo_prover = { git = "https://github.com/starkware-libs/stwo-cairo", rev = "a27287c21e8b7e677b07cc40988c8c145533c55d" }
+stwo-prover = { git = "https://github.com/starkware-libs/stwo", rev = "af5475cb", features = ["parallel"] }
 camino = { version = "1", features = ["serde1"] }
 cargo_metadata = ">=0.18"
 clap = { version = "4", features = ["derive", "env", "string"] }

Cargo.toml

@@ -20,15 +20,12 @@ serde.workspace = true
 camino.workspace = true
 serde_json.workspace = true
 create-output-dir = { path = "../../utils/create-output-dir" }
-stwo_cairo_prover = { git = "https://github.com/starkware-libs/stwo-cairo", rev = "a27287c21e8b7e677b07cc40988c8c145533c55d" }
-stwo-prover = { git = "https://github.com/starkware-libs/stwo", rev = "af5475cb", features = [
-    "parallel",
-] }
+stwo_cairo_prover.workspace = true
+stwo-prover.workspace = true
 scarb-execute = { path = "../scarb-execute" }
 
 [dev-dependencies]
 assert_fs.workspace = true
 scarb-test-support = { path = "../../utils/scarb-test-support" }
 snapbox.workspace = true
 predicates.workspace = true
-

extensions/scarb-prove/Cargo.toml

@@ -0,0 +1,28 @@
+[package]
+name = "scarb-verify"
+publish = false
+
+authors.workspace = true
+edition.workspace = true
+homepage.workspace = true
+license.workspace = true
+readme.workspace = true
+repository.workspace = true
+version.workspace = true
+
+[dependencies]
+anyhow.workspace = true
+indoc.workspace = true
+scarb-metadata = { path = "../../scarb-metadata" }
+scarb-ui = { path = "../../utils/scarb-ui" }
+clap.workspace = true
+serde.workspace = true
+camino.workspace = true
+serde_json.workspace = true
+stwo_cairo_prover.workspace = true
+stwo-prover.workspace = true
+
+[dev-dependencies]
+assert_fs.workspace = true
+scarb-test-support = { path = "../../utils/scarb-test-support" }
+snapbox.workspace = true

extensions/scarb-verify/Cargo.toml

@@ -0,0 +1,119 @@
+use anyhow::{ensure, Context, Result};
+use camino::{Utf8Path, Utf8PathBuf};
+use clap::Parser;
+use indoc::formatdoc;
+use scarb_metadata::{MetadataCommand, PackageMetadata};
+use scarb_ui::args::{PackagesFilter, VerbositySpec};
+use scarb_ui::components::Status;
+use scarb_ui::{OutputFormat, Ui};
+use std::env;
+use std::fs;
+use std::process::ExitCode;
+use stwo_cairo_prover::cairo_air::air::CairoProof;
+use stwo_cairo_prover::cairo_air::verify_cairo;
+use stwo_prover::core::vcs::blake2_merkle::{Blake2sMerkleChannel, Blake2sMerkleHasher};
+
+/// Verifies `scarb prove` output using Stwo verifier.
+#[derive(Parser, Clone, Debug)]
+#[clap(version, verbatim_doc_comment)]
+struct Args {
+    /// Name of the package.
+    #[command(flatten)]
+    packages_filter: PackagesFilter,
+
+    /// ID of `scarb execute` output for given package, for which proof was generated using `scarb prove`.
+    #[arg(long)]
+    execution_id: Option<u32>,
+
+    /// Proof file path.
+    #[arg(
+        long,
+        required_unless_present = "execution_id",
+        conflicts_with = "execution_id"
+    )]
+    proof_file: Option<Utf8PathBuf>,
+
+    /// Logging verbosity.
+    #[command(flatten)]
+    pub verbose: VerbositySpec,
+}
+
+fn main() -> ExitCode {
+    let args = Args::parse();
+    let ui = Ui::new(args.verbose.clone().into(), OutputFormat::Text);
+
+    match main_inner(args, ui.clone()) {
+        Ok(()) => ExitCode::SUCCESS,
+        Err(error) => {
+            ui.error(format!("{error:#}"));
+            ExitCode::FAILURE
+        }
+    }
+}
+
+fn main_inner(args: Args, ui: Ui) -> Result<()> {
+    let scarb_target_dir = Utf8PathBuf::from(env::var("SCARB_TARGET_DIR")?);
+
+    let metadata = MetadataCommand::new().inherit_stderr().exec()?;
+    let package = args.packages_filter.match_one(&metadata)?;
+
+    let proof_path = if let Some(execution_id) = args.execution_id {
+        ui.print(Status::new("Verifying", &package.name));
+        resolve_proof_path_from_package(&scarb_target_dir, &package, execution_id)?
+    } else {
+        ui.print(Status::new("Verifying", "proof"));
+        args.proof_file.unwrap()
+    };
+
+    let proof = load_proof(&proof_path)?;
+
+    verify_cairo::<Blake2sMerkleChannel>(proof).with_context(|| "failed to verify proof")?;
+
+    ui.print(Status::new("Verified", "proof successfully"));
+
+    Ok(())
+}
+
+fn load_proof(path: &Utf8Path) -> Result<CairoProof<Blake2sMerkleHasher>> {
+    ensure!(
+        path.exists(),
+        format!("proof file does not exist at path: {path}")
+    );
+
+    let proof_contents =
+        fs::read_to_string(path).with_context(|| format!("failed to read proof file: {path}"))?;
+    let proof = serde_json::from_str(&proof_contents)
+        .with_context(|| format!("failed to deserialize proof file: {path}"))?;
+    Ok(proof)
+}
+
+fn resolve_proof_path_from_package(
+    scarb_target_dir: &Utf8Path,
+    package: &PackageMetadata,
+    execution_id: u32,
+) -> Result<Utf8PathBuf> {
+    let execution_dir = scarb_target_dir
+        .join("execute")
+        .join(&package.name)
+        .join(format!("execution{execution_id}"));
+
+    ensure!(
+        execution_dir.exists(),
+        formatdoc! {r#"
+            execution directory does not exist at path: {execution_dir}
+            help: make sure to run `scarb prove --execute` first
+            and that the execution ID is correct
+        "#}
+    );
+
+    let proof_path = execution_dir.join("proof").join("proof.json");
+    ensure!(
+        proof_path.exists(),
+        formatdoc! {r#"
+            proof file does not exist at path: {proof_path}
+            help: run `scarb prove --execution-id={execution_id}` first
+        "#}
+    );
+
+    Ok(proof_path)
+}