zombie/src/main.rs

//#![feature(plugin)]
//#![plugin(dynasm)]
#![feature(proc_macro_hygiene)]
#[macro_use]
extern crate dynasm;

extern crate winapi;
extern crate typed_arena;

use std::io::{self, Read};
use std::fs::File;
use std::env;

use dynasm::dynasm;
use dynasmrt::{DynasmApi, DynasmLabelApi};

pub mod ir;
pub mod parser;
pub mod interpret;
pub mod optimize;
pub mod compile;
pub mod transpile_c;

use crate::ir::MutVisitor;

fn main() -> io::Result<()> {
    let args: Vec<String> = env::args().collect();
    let mut buffer = String::new();

    if args.len() > 1 {
        File::open(&args[1])?.read_to_string(&mut buffer)?;
    }
    else {
        io::stdin().read_to_string(&mut buffer)?;
    }

    let insts = parser::parse(&buffer);
    if let Ok(mut insts) = insts {
        let mut lin_loop_optimizer = optimize::LinOptimizer::new();
        lin_loop_optimizer.visit_instructions(&mut insts);
        
        for ref inst in &insts {
            //println!("{}\n", inst.to_string());
        }
        println!("{}", transpile_c::transpile_c(&insts));

        //let code = compile::compile(&insts);
    }
    else if let Err(msg) = insts {
        println!("error parsing: {}", msg);
    }

    Ok(())
}