1 files changed, 64 insertions, 0 deletions
diff --git a/zkvms_guest_io/src/main.rs b/zkvms_guest_io/src/main.rs
new file mode 100644
index 0000000..f4187b3
--- /dev/null
+++ b/zkvms_guest_io/src/main.rs
@@ -0,0 +1,64 @@
+use clap::Parser;
+use std::process::{Command, Stdio};
+use std::thread;
+use std::time::Duration;
+
+/// A CLI tool for running and benchmarking a guest program inside all
+/// supported zkVMs.
+/// This binary has been built with a single guest program in mind.
+/// If you want to run or benchmark your own guest program inside a zkVM,
+/// head on over to https://github.com/blocksense-network/zkVMs-benchmarks
+#[derive(Parser, Debug)]
+#[command(about, long_about = None)]
+struct Cli {
+    /// Ignored zkVMs. Values are substrings of names.
+    #[arg(short, long, value_delimiter = ',', num_args = 1..)]
+    ignore: Option<Vec<String>>,
+
+    /// Arguments which are passed to each tool for a single guest and single zkVM
+    #[arg(trailing_var_arg = true, allow_hyphen_values = true)]
+    zkvm_args: Vec<String>,
+}
+
+fn main() {
+    let cli = Cli::parse();
+
+    let guests: Vec<&str> = env!("PROGRAMS")
+        .split(',')
+        .filter(|x| !x.is_empty())
+        .collect();
+    let mut threads = Vec::new();
+    let ignored = cli.ignore.unwrap_or(Vec::new());
+
+    for guest in guests.into_iter() {
+        if ignored.iter().any(|i| guest.contains(i)) {
+            continue;
+        }
+
+        let args = cli.zkvm_args.clone();
+        threads.push(
+            thread::Builder::new()
+                .name(format!(r#"Running "{}""#, guest))
+                .spawn(move || {
+                    let output = Command::new(guest)
+                        .args(args)
+                        .stdout(Stdio::piped())
+                        .output()
+                        .expect("error occured!");
+
+                    let mut stdout = String::from_utf8(output.stdout).unwrap();
+                    if !output.stderr.is_empty() {
+                        stdout.push('\n');
+                        stdout += &String::from_utf8(output.stderr).unwrap();
+                    }
+
+                    print!("== Executing {} ==\n{}", guest, stdout);
+                })
+                .unwrap(),
+        );
+    }
+
+    while threads.iter().any(|t| !t.is_finished()) {
+        thread::sleep(Duration::from_millis(200));
+    }
+}