make elf loader work with the mips test files, test by generating jso…

…n files

o1vm/src/cannon.rs

@@ -209,7 +209,7 @@ pub struct HostProgram {
 pub struct VmConfiguration {
     pub input_state_file: String,
     pub output_state_file: String,
-    pub metadata_file: String,
+    pub metadata_file: Option<String>,
     pub proof_at: StepFrequency,
     pub stop_at: StepFrequency,
     pub snapshot_state_at: StepFrequency,

o1vm/src/cli/cannon.rs

@@ -19,13 +19,8 @@ pub struct MipsVmConfigurationArgs {
     )]
     output: String,
 
-    #[arg(
-        long,
-        value_name = "FILE",
-        default_value = "meta.json",
-        help = "metadata file"
-    )]
-    meta: String,
+    #[arg(long, value_name = "FILE", help = "metadata file")]
+    meta: Option<String>,
 
     #[arg(
         long = "proof-at",
@@ -104,9 +99,25 @@ pub struct RunArgs {
     pub vm_cfg: MipsVmConfigurationArgs,
 }
 
+#[derive(Parser, Debug, Clone)]
+pub struct GenStateJsonArgs {
+    #[arg(short = 'i', long, value_name = "FILE", help = "input ELF file")]
+    pub input: String,
+    #[arg(
+        short = 'o',
+        long,
+        value_name = "FILE",
+        default_value = "state.json",
+        help = "output state.json file"
+    )]
+    pub output: String,
+}
+
 #[derive(Subcommand, Clone, Debug)]
 pub enum Cannon {
     Run(RunArgs),
     #[command(name = "test-optimism-preimage-read")]
     TestPreimageRead(RunArgs),
+    #[command(name = "gen-state-json")]
+    GenStateJson(GenStateJsonArgs),
 }

o1vm/src/elf_loader.rs

@@ -1,24 +1,19 @@
 use crate::cannon::{Page, State, PAGE_SIZE};
-use elf::{endian::LittleEndian, section::SectionHeader, ElfBytes};
+use elf::{
+    endian::{BigEndian, EndianParse, LittleEndian},
+    section::SectionHeader,
+    ElfBytes,
+};
 use log::debug;
 use std::{collections::HashMap, path::Path};
 
-/// Parse an ELF file and return the parsed data as a structure that is expected
-/// by the o1vm RISC-V 32 bits edition.
-// FIXME: parametrize by an architecture. We should return a state depending on the
-// architecture. In the meantime, we can have parse_riscv32i and parse_mips.
-// FIXME: for now, we return a State structure, either for RISC-V 32i or MIPS.
-// We should return a structure specifically built for the o1vm, and not tight
-// to Cannon. It will be done in a future PR to avoid breaking the current code
-// and have a huge diff.
-pub fn parse_riscv32(path: &Path) -> Result<State, String> {
-    debug!("Start parsing the ELF file to load a RISC-V 32i compatible state");
-    let file_data = std::fs::read(path).expect("Could not read file.");
-    let slice = file_data.as_slice();
-    let file = ElfBytes::<LittleEndian>::minimal_parse(slice).expect("Open ELF file failed.");
+pub enum Architecture {
+    Mips,
+    RiscV,
+}
 
+pub fn make_state<T: EndianParse>(file: ElfBytes<T>) -> Result<State, String> {
     // Checking it is RISC-V
-    assert_eq!(file.ehdr.e_machine, 243);
 
     let (shdrs_opt, strtab_opt) = file
         .section_headers_with_strtab()
@@ -51,9 +46,11 @@ pub fn parse_riscv32(path: &Path) -> Result<State, String> {
         .section_data(text_section)
         .expect("Failed to read data from .text section");
 
+    // address of starting instruction in code section
     let code_section_starting_address = text_section.sh_addr as usize;
     let code_section_size = text_section.sh_size as usize;
-    let code_section_end_address = code_section_starting_address + code_section_size;
+    // address of last instruction in code section
+    let code_section_end_address = code_section_starting_address + code_section_size - 1;
     debug!(
         "The executable code starts at address {}, has size {} bytes, and ends at address {}.",
         code_section_starting_address, code_section_size, code_section_end_address
@@ -64,36 +61,48 @@ pub fn parse_riscv32(path: &Path) -> Result<State, String> {
     let page_size_usize: usize = PAGE_SIZE.try_into().unwrap();
     // Padding to get the right number of pages. We suppose that the memory
     // index starts at 0.
+
+    // the address that the first page starts on
     let start_page_address: usize =
         (code_section_starting_address / page_size_usize) * page_size_usize;
-    let end_page_address =
-        (code_section_end_address / (page_size_usize - 1)) * page_size_usize + page_size_usize;
+
+    // the address that the last page starts on
+    let end_page_address = (code_section_end_address / page_size_usize) * page_size_usize;
 
     let first_page_index = start_page_address / page_size_usize;
-    let last_page_index = (end_page_address - 1) / page_size_usize;
+
+    let last_page_index = end_page_address / page_size_usize;
+
     let mut data_offset = 0;
     (first_page_index..=last_page_index).for_each(|page_index| {
         let mut data = vec![0; page_size_usize];
-        // Special case of only one page
+        // Special case where all code fits in one page
         if first_page_index == last_page_index {
             let data_length = code_section_end_address - code_section_starting_address;
             let page_offset = code_section_starting_address - start_page_address;
             data[page_offset..page_offset + data_length]
                 .copy_from_slice(&text_section_data[0..data_length]);
             data_offset += data_length;
         } else {
-            let data_length = if page_index == last_page_index {
-                code_section_end_address - (page_index * page_size_usize)
-            } else {
-                page_size_usize
-            };
             let page_offset = if page_index == first_page_index {
                 code_section_starting_address - start_page_address
             } else {
                 0
             };
-            data[page_offset..]
+            let data_length = if page_index == last_page_index {
+                let data_length = code_section_end_address - end_page_address;
+                // for the last page, we might need to pad with zeros if the text_section_data is not
+                // a multiple of the page size.
+                if page_size_usize > data_length {
+                    data[page_offset + data_length..page_offset + page_size_usize].fill(0);
+                }
+                data_length
+            } else {
+                page_size_usize
+            };
+            data[page_offset..page_offset + data_length]
                 .copy_from_slice(&text_section_data[data_offset..data_offset + data_length]);
+
             data_offset += data_length;
         }
         let page = Page {
@@ -115,7 +124,7 @@ pub fn parse_riscv32(path: &Path) -> Result<State, String> {
 
     // Entry point of the program
     let pc: u32 = file.ehdr.e_entry as u32;
-    assert!(pc != 0, "Entry point is 0. The documentation of the ELF library says that it means the ELF doesn't have an entry point. This is not supported. This can happen if the binary given is an object file and not an executable file. You might need to call the linker (ld) before running the binary.");
+    assert!(pc != 0, "Entry point is 0. The documentation of the ELF library says that it means the ELF doesn't have an entry point. This is not supported.");
     let next_pc: u32 = pc + 4u32;
 
     let state = State {
@@ -143,3 +152,22 @@ pub fn parse_riscv32(path: &Path) -> Result<State, String> {
 
     Ok(state)
 }
+
+pub fn parse_elf(arch: Architecture, path: &Path) -> Result<State, String> {
+    debug!("Start parsing the ELF file to load a compatible state");
+    let file_data = std::fs::read(path).expect("Could not read file.");
+    let slice = file_data.as_slice();
+    match arch {
+        Architecture::Mips => {
+            let file = ElfBytes::<BigEndian>::minimal_parse(slice).expect("Open ELF file failed.");
+            assert_eq!(file.ehdr.e_machine, 8);
+            make_state(file)
+        }
+        Architecture::RiscV => {
+            let file =
+                ElfBytes::<LittleEndian>::minimal_parse(slice).expect("Open ELF file failed.");
+            assert_eq!(file.ehdr.e_machine, 243);
+            make_state(file)
+        }
+    }
+}

o1vm/src/interpreters/mips/witness.rs

@@ -1143,7 +1143,7 @@ impl<Fp: Field, PreImageOracle: PreImageOracleT> Env<Fp, PreImageOracle> {
     pub fn step(
         &mut self,
         config: &VmConfiguration,
-        metadata: &Meta,
+        metadata: &Option<Meta>,
         start: &Start,
     ) -> Instruction {
         self.reset_scratch_state();
@@ -1267,7 +1267,7 @@ impl<Fp: Field, PreImageOracle: PreImageOracleT> Env<Fp, PreImageOracle> {
         }
     }
 
-    fn pp_info(&mut self, at: &StepFrequency, meta: &Meta, start: &Start) {
+    fn pp_info(&mut self, at: &StepFrequency, meta: &Option<Meta>, start: &Start) {
         if self.should_trigger_at(at) {
             let elapsed = start.time.elapsed();
             // Compute the step number removing the MAX_ACC factor
@@ -1286,8 +1286,9 @@ impl<Fp: Field, PreImageOracle: PreImageOracleT> Env<Fp, PreImageOracle> {
 
             let mem = self.memory_usage();
             let name = meta
-                .find_address_symbol(pc)
-                .unwrap_or_else(|| "n/a".to_string());
+                .as_ref()
+                .and_then(|m| m.find_address_symbol(pc))
+                .unwrap_or("n/a".to_string());
 
             info!(
                 "processing step={} pc={:010x} insn={:010x} ips={:.2} pages={} mem={} name={}",

o1vm/src/legacy/main.rs

@@ -5,8 +5,8 @@ use kimchi::o1_utils;
 use kimchi_msm::{proof::ProofInputs, prover::prove, verifier::verify, witness::Witness};
 use log::debug;
 use o1vm::{
-    cannon::{self, Meta, Start, State},
-    cli,
+    cannon::{self, Start, State},
+    cli, elf_loader,
     interpreters::{
         keccak::{
             column::{Steps, N_ZKVM_KECCAK_COLS, N_ZKVM_KECCAK_REL_COLS, N_ZKVM_KECCAK_SEL_COLS},
@@ -33,7 +33,9 @@ use o1vm::{
     test_preimage_read,
 };
 use poly_commitment::SRS as _;
-use std::{cmp::Ordering, collections::HashMap, fs::File, io::BufReader, process::ExitCode};
+use std::{
+    cmp::Ordering, collections::HashMap, fs::File, io::BufReader, path::Path, process::ExitCode,
+};
 use strum::IntoEnumIterator;
 
 /// Domain size shared by the Keccak evaluations, MIPS evaluation and main
@@ -50,18 +52,11 @@ pub fn cannon_main(args: cli::cannon::RunArgs) {
     // Read the JSON contents of the file as an instance of `State`.
     let state: State = serde_json::from_reader(reader).expect("Error reading input state file");
 
-    let meta_file = File::open(&configuration.metadata_file).unwrap_or_else(|_| {
-        panic!(
-            "Could not open metadata file {}",
-            &configuration.metadata_file
-        )
-    });
-
-    let meta: Meta = serde_json::from_reader(BufReader::new(meta_file)).unwrap_or_else(|_| {
-        panic!(
-            "Error deserializing metadata file {}",
-            &configuration.metadata_file
-        )
+    let meta = &configuration.metadata_file.as_ref().map(|f| {
+        let meta_file =
+            File::open(f).unwrap_or_else(|_| panic!("Could not open metadata file {}", f));
+        serde_json::from_reader(BufReader::new(meta_file))
+            .unwrap_or_else(|_| panic!("Error deserializing metadata file {}", f))
     });
 
     let mut po = PreImageOracle::create(&configuration.host);
@@ -124,7 +119,7 @@ pub fn cannon_main(args: cli::cannon::RunArgs) {
     }
 
     while !mips_wit_env.halt {
-        let instr = mips_wit_env.step(&configuration, &meta, &start);
+        let instr = mips_wit_env.step(&configuration, meta, &start);
 
         if let Some(ref mut keccak_env) = mips_wit_env.keccak_env {
             // Run all steps of hash
@@ -327,6 +322,14 @@ pub fn cannon_main(args: cli::cannon::RunArgs) {
     // TODO: Logic
 }
 
+fn gen_state_json(arg: cli::cannon::GenStateJsonArgs) -> Result<(), String> {
+    let path = Path::new(&arg.input);
+    let state = elf_loader::parse_elf(elf_loader::Architecture::Mips, path)?;
+    let file = File::create(&arg.output).expect("Error creating output state file");
+    serde_json::to_writer_pretty(file, &state).expect("Error writing output state file");
+    Ok(())
+}
+
 pub fn main() -> ExitCode {
     env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info")).init();
     let args = cli::Commands::parse();
@@ -338,6 +341,9 @@ pub fn main() -> ExitCode {
             cli::cannon::Cannon::TestPreimageRead(args) => {
                 test_preimage_read::main(args);
             }
+            cli::cannon::Cannon::GenStateJson(args) => {
+                gen_state_json(args).expect("Error generating state.json");
+            }
         },
     }
     ExitCode::SUCCESS

o1vm/src/pickles/main.rs

@@ -9,8 +9,8 @@ use mina_poseidon::{
     sponge::{DefaultFqSponge, DefaultFrSponge},
 };
 use o1vm::{
-    cannon::{self, Meta, Start, State},
-    cli,
+    cannon::{self, Start, State},
+    cli, elf_loader,
     interpreters::mips::{
         column::N_MIPS_REL_COLS,
         constraints as mips_constraints,
@@ -23,7 +23,7 @@ use o1vm::{
     test_preimage_read,
 };
 use poly_commitment::{ipa::SRS, SRS as _};
-use std::{fs::File, io::BufReader, process::ExitCode, time::Instant};
+use std::{fs::File, io::BufReader, path::Path, process::ExitCode, time::Instant};
 use strum::IntoEnumIterator;
 
 pub const DOMAIN_SIZE: usize = 1 << 15;
@@ -40,18 +40,11 @@ pub fn cannon_main(args: cli::cannon::RunArgs) {
     // Read the JSON contents of the file as an instance of `State`.
     let state: State = serde_json::from_reader(reader).expect("Error reading input state file");
 
-    let meta_file = File::open(&configuration.metadata_file).unwrap_or_else(|_| {
-        panic!(
-            "Could not open metadata file {}",
-            &configuration.metadata_file
-        )
-    });
-
-    let meta: Meta = serde_json::from_reader(BufReader::new(meta_file)).unwrap_or_else(|_| {
-        panic!(
-            "Error deserializing metadata file {}",
-            &configuration.metadata_file
-        )
+    let meta = &configuration.metadata_file.as_ref().map(|f| {
+        let meta_file =
+            File::open(f).unwrap_or_else(|_| panic!("Could not open metadata file {}", f));
+        serde_json::from_reader(BufReader::new(meta_file))
+            .unwrap_or_else(|_| panic!("Error deserializing metadata file {}", f))
     });
 
     let mut po = PreImageOracle::create(&configuration.host);
@@ -93,7 +86,7 @@ pub fn cannon_main(args: cli::cannon::RunArgs) {
 
     let mut curr_proof_inputs: ProofInputs<Vesta> = ProofInputs::new(DOMAIN_SIZE);
     while !mips_wit_env.halt {
-        let _instr: Instruction = mips_wit_env.step(&configuration, &meta, &start);
+        let _instr: Instruction = mips_wit_env.step(&configuration, meta, &start);
         for (scratch, scratch_chunk) in mips_wit_env
             .scratch_state
             .iter()
@@ -156,6 +149,14 @@ pub fn cannon_main(args: cli::cannon::RunArgs) {
     }
 }
 
+fn gen_state_json(arg: cli::cannon::GenStateJsonArgs) -> Result<(), String> {
+    let path = Path::new(&arg.input);
+    let state = elf_loader::parse_elf(elf_loader::Architecture::Mips, path)?;
+    let file = File::create(&arg.output).expect("Error creating output state file");
+    serde_json::to_writer_pretty(file, &state).expect("Error writing output state file");
+    Ok(())
+}
+
 pub fn main() -> ExitCode {
     env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info")).init();
     let args = cli::Commands::parse();
@@ -167,6 +168,9 @@ pub fn main() -> ExitCode {
             cli::cannon::Cannon::TestPreimageRead(args) => {
                 test_preimage_read::main(args);
             }
+            cli::cannon::Cannon::GenStateJson(args) => {
+                gen_state_json(args).expect("Error generating state.json");
+            }
         },
     }
     ExitCode::SUCCESS

o1vm/test-gen-state-json.sh

@@ -0,0 +1,14 @@
+#!/bin/bash
+
+BIN_DIR=${1:-${BIN_DIR:-o1vm/resources/programs/mips/bin}}
+
+# Ensure the directory exists
+if [[ ! -d "$BIN_DIR" ]]; then
+  echo "Error: Directory '$BIN_DIR' does not exist."
+  exit 1
+fi
+
+find "$BIN_DIR" -type f ! -name "*.o" | while read -r file; do
+  echo "Processing: $file"
+  cargo run --bin pickles_o1vm -- cannon gen-state-json -i "$file"
+done