feat: suggest passing argument from cli (#87)

Author: weilzkm

examples/README.md

@@ -16,7 +16,7 @@ go build hello.go
 * Split the ELF hello into segments. Note that the flag `BLOCK_NO` is only necessary for minigeth.
 
 ```
-BASEDIR=test-vectors RUST_LOG=info ELF_PATH=test-vectors/hello BLOCK_NO=13284491 SEG_OUTPUT=/tmp/output SEG_SIZE=262144 \
+BASEDIR=test-vectors RUST_LOG=info ELF_PATH=test-vectors/hello BLOCK_NO=13284491 SEG_OUTPUT=/tmp/output SEG_SIZE=262144 ARGS="" \
     cargo run --release --example zkmips split
 ```
 

examples/zkmips.rs

@@ -33,13 +33,14 @@ fn split_elf_into_segs() {
     let seg_path = env::var("SEG_OUTPUT").expect("Segment output path is missing");
     let seg_size = env::var("SEG_SIZE").unwrap_or(format!("{SEGMENT_STEPS}"));
     let seg_size = seg_size.parse::<_>().unwrap_or(SEGMENT_STEPS);
+    let args = env::var("ARGS").unwrap_or("".to_string());
 
     let data = fs::read(elf_path).expect("could not read file");
     let file =
         ElfBytes::<AnyEndian>::minimal_parse(data.as_slice()).expect("opening elf file failed");
     let (mut state, _) = State::load_elf(&file);
     state.patch_go(&file);
-    state.patch_stack("");
+    state.patch_stack(&args);
 
     let block_path = get_block_path(&basedir, &block_no, "");
     state.load_input(&block_path);

src/mips_emulator/tests.rs

@@ -133,4 +133,103 @@ mod tests {
 
         instrumented_state.split_segment(true, OUTPUT);
     }
+
+    #[test]
+    fn test_execute_app() {
+        let path = PathBuf::from("./test-vectors/test_local_app");
+        let data = fs::read(path).expect("could not read file");
+        let file =
+            ElfBytes::<AnyEndian>::minimal_parse(data.as_slice()).expect("opening elf file failed");
+        let (mut state, _) = State::load_elf(&file);
+
+        state.patch_go(&file);
+        state.patch_stack("");
+
+        let mut instrumented_state = InstrumentedState::new(state, String::from(""));
+
+        let mut step_number = 0;
+        loop {
+            if instrumented_state.state.exited {
+                break;
+            }
+            instrumented_state.step();
+            step_number += 1;
+        }
+        println!("total steps {}", step_number);
+    }
+
+    #[test]
+    fn test_execute_nested_fib() {
+        let path = PathBuf::from("./test-vectors/fib");
+        let data = fs::read(path).expect("could not read file");
+        let file =
+            ElfBytes::<AnyEndian>::minimal_parse(data.as_slice()).expect("opening elf file failed");
+        let (mut state, _) = State::load_elf(&file);
+
+        state.patch_go(&file);
+        state.patch_stack("");
+
+        let mut instrumented_state = InstrumentedState::new(state, String::from(""));
+
+        let mut step_number = 0usize;
+        loop {
+            if instrumented_state.state.exited {
+                break;
+            }
+            instrumented_state.step();
+            step_number += 1;
+        }
+
+        println!("total steps {}", step_number);
+    }
+
+    #[test]
+    fn test_execute_fib() {
+        let path = PathBuf::from("./test-vectors/fib2");
+        let data = fs::read(path).expect("could not read file");
+        let file =
+            ElfBytes::<AnyEndian>::minimal_parse(data.as_slice()).expect("opening elf file failed");
+        let (mut state, _) = State::load_elf(&file);
+
+        state.patch_go(&file);
+        state.patch_stack("");
+
+        let mut instrumented_state = InstrumentedState::new(state, String::from(""));
+
+        let mut step_number = 0usize;
+        loop {
+            if instrumented_state.state.exited {
+                break;
+            }
+            instrumented_state.step();
+            step_number += 1;
+        }
+
+        println!("total steps {}", step_number);
+    }
+
+    #[test]
+    fn test_execute_simple_add() {
+        let path = PathBuf::from("./test-vectors/simpleArith");
+        let data = fs::read(path).expect("could not read file");
+        let file =
+            ElfBytes::<AnyEndian>::minimal_parse(data.as_slice()).expect("opening elf file failed");
+        let (mut state, _) = State::load_elf(&file);
+
+        state.patch_go(&file);
+        state.patch_stack("");
+
+        let mut instrumented_state = InstrumentedState::new(state, String::from(""));
+
+        let mut step_number = 0usize;
+        loop {
+            if instrumented_state.state.exited {
+                break;
+            }
+            instrumented_state.step();
+            step_number += 1;
+        }
+
+        println!("total steps {}", step_number);
+    }
 }