Create a crate for vm_runner (#125)

Author: Tom-StarkWare

Cargo.toml

@@ -1,28 +1,33 @@
 [workspace]
 members = [
     "crates/cairo-program-runner",
-    "crates/cairo-program-runner-lib"
+    "crates/cairo-program-runner-lib",
+    "crates/vm_runner",
 ]
 resolver = "2"
 
 [workspace.dependencies]
 bincode = { version = "2.0.0-rc.3", default-features = false, features = [
     "serde",
 ] }
+cairo-lang-runner = { git = "https://github.com/starkware-libs/cairo.git", rev = "ddaec76b4364ca6e0af4d185a99904d13b93f81e" }
 cairo-vm = { version = "2.0.1", features = [
     "extensive_hints",
     "mod_builtin",
     "clap",
 ] }
-cairo-lang-runner = { git = "https://github.com/starkware-libs/cairo.git", rev = "ddaec76b4364ca6e0af4d185a99904d13b93f81e" }
 clap = { version = "4.3.10", features = ["derive"] }
+log = "0.4.21"
+num-bigint = "0.4"
 num-traits = "0.2.19"
+regex = "1.11.1"
 serde = { version = "1.0.202", features = ["derive"] }
 serde_json = "1.0.117"
 starknet-crypto = "0.6.2"
 starknet-types-core = "0.1.2"
+stwo_cairo_utils = { git = "https://github.com/starkware-libs/stwo-cairo", rev = "ef6b24df"}
+stwo-cairo-adapter = { git = "https://github.com/starkware-libs/stwo-cairo", rev = "ef6b24df"}
 tempfile = "3.10.1"
 thiserror = "1.0.61"
 thiserror-no-std = "2.0.2"
-regex = "1.11.1"
-num-bigint = "0.4"
+tracing = "0.1.40"

crates/vm_runner/Cargo.toml

@@ -0,0 +1,14 @@
+[package]
+name = "stwo_vm_runner"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+clap.workspace = true
+log.workspace = true
+num-traits.workspace = true
+serde_json.workspace = true
+stwo_cairo_utils.workspace = true
+stwo-cairo-adapter.workspace = true
+thiserror.workspace = true
+tracing.workspace = true

crates/vm_runner/src/main.rs

@@ -0,0 +1,66 @@
+use std::path::PathBuf;
+use std::process::ExitCode;
+
+use clap::Parser;
+use stwo_cairo_adapter::plain::adapt_finished_runner;
+use stwo_cairo_adapter::vm_import::VmImportError;
+use stwo_cairo_adapter::{ExecutionResources, ProverInput};
+use stwo_cairo_utils::binary_utils::run_binary;
+use stwo_cairo_utils::vm_utils::{run_vm, VmArgs, VmError};
+use thiserror::Error;
+use tracing::{span, Level};
+
+/// Command line arguments for stwo_vm_runner.
+/// Example command line (use absolute paths):
+///     ```
+///     cargo run -r --bin stwo_vm_runner -- --run_from_cairo_pie
+///     --output_path path/to/output --secure_run=true path/to/cairo/pie
+///     ```
+#[derive(Parser, Debug)]
+#[clap(author, version, about, long_about = None)]
+struct Args {
+    #[command(flatten)]
+    vm_args: VmArgs,
+    /// The file path for the output (the adapted execution resources of the VM run).
+    #[structopt(long = "output_path")]
+    output_path: PathBuf,
+}
+
+#[derive(Debug, Error)]
+enum Error {
+    #[error("Invalid arguments")]
+    Cli(#[from] clap::Error),
+    #[error("Failed to interact with the file system")]
+    IO(#[from] std::io::Error),
+    #[error("VM run failed: {0}")]
+    Vm(#[from] VmError),
+    #[error("Serialization failed: {0}")]
+    Serde(#[from] serde_json::Error),
+    #[error("VM import failed: {0}")]
+    VmImport(#[from] VmImportError),
+}
+
+fn main() -> ExitCode {
+    run_binary(run, "stwo_vm_runner")
+}
+
+fn run(args: impl Iterator<Item = String>) -> Result<ProverInput, Error> {
+    let _span = span!(Level::INFO, "run").entered();
+    let args = Args::try_parse_from(args)?;
+
+    // Usually vm_runner runs the VM in non-proof-mode, in which case we don't need
+    // `disable_trace_padding`. If it runs the VM in proof-mode, it should also disable trace
+    // padding as this is the padding relevant for Stwo.
+    let disable_trace_padding = args.vm_args.proof_mode;
+    let cairo_runner = run_vm(&args.vm_args, disable_trace_padding)?;
+    let cairo_input = adapt_finished_runner(cairo_runner)?;
+
+    let execution_resources = ExecutionResources::from_prover_input(&cairo_input);
+    log::info!("Execution resources: {:#?}", execution_resources);
+    std::fs::write(
+        args.output_path,
+        serde_json::to_string(&execution_resources)?,
+    )?;
+
+    Ok(cairo_input)
+}