rasp/src/main.rs

#![allow(dead_code)]

mod common;
mod vm;

use std::{convert::TryFrom, env, fs, io, process};

type Result<T> = std::result::Result<T, Box<dyn std::error::Error>>;

fn get_input_string() -> io::Result<String> {
    let args: Vec<String> = env::args().collect();
    if args.len() <= 1 {
        println!(
            "usage: {} file.asm",
            args.get(0).map(String::as_str).unwrap_or("rasp")
        );
        process::exit(1);
    } else {
        fs::read_to_string(&args[1])
    }
}

fn main() -> Result<()> {
    use vm::obj::syn::parser::SectionsParser;
    let contents = get_input_string()?;
    let ast = match SectionsParser::new().parse(&contents) {
        Ok(ast) => ast,
        Err(err) => {
            eprintln!("{}", err);
            process::exit(1);
        }
    };
    let obj = vm::obj::obj::Object::try_from(&ast)?;
    dump(&obj)?;
    let mut vm = vm::vm::Vm::new();
    vm.load_object(obj, 1024 * 1024 * 64)?; // 64mb
    let status = vm.run()?;
    println!("{}", status);
    Ok(())
}


fn dump(obj: &vm::obj::obj::Object) -> Result<()> {
    use vm::obj::obj::Section;
    use vm::visit::VisitInst;
    let mut stdout = io::stdout();
    for section in &obj.sections {
        match section {
            Section::Data { start, contents, .. } => {
                const WIDTH: usize = 4;
                println!("data section at 0x{:08x}", start);
                for (i, b) in contents.iter().enumerate() {
                    if i % WIDTH == 0 {
                        print!("{:08x} | ", ((*start as usize) + i));
                    }
                    print!("{:02x} ", b);
                    if i % WIDTH == (WIDTH - 1) {
                        println!();
                    }
                }
                println!();
            }
            Section::Code { start, contents, .. } => {
                println!("code section at 0x{:08x}", start);
                let mut disasm = vm::disassemble::Disassemble::new(&mut stdout, contents, *start);
                while !disasm.is_done() {
                    disasm.visit_inst()?;
                }
                println!();
            }
            Section::Meta { .. } => continue,
        };
    }
    Ok(())
}