wasmer/src/main.rs

103 lines
2.7 KiB
Rust
Raw Normal View History

#![feature(test, libc)]
extern crate test;
#[macro_use]
extern crate error_chain;
#[macro_use]
extern crate structopt;
extern crate cranelift_codegen;
2018-10-14 21:48:59 +00:00
extern crate cranelift_entity;
extern crate cranelift_native;
extern crate cranelift_wasm;
2018-10-14 21:48:59 +00:00
extern crate wabt;
2018-10-13 17:22:57 +00:00
#[macro_use]
extern crate target_lexicon;
2018-10-14 11:59:11 +00:00
extern crate spin;
2018-10-15 00:48:59 +00:00
use std::time::{Duration, Instant};
// #[macro_use] extern crate log;
use libc;
2018-10-14 21:48:59 +00:00
use std::error::Error;
use std::fs::File;
use std::io;
use std::io::Read;
2018-10-14 21:48:59 +00:00
use std::path::PathBuf;
use std::process::exit;
use structopt::StructOpt;
use wabt::wat2wasm;
2018-10-16 15:01:47 +00:00
#[macro_use]
mod macros;
2018-10-14 11:59:11 +00:00
pub mod common;
pub mod integrations;
2018-10-14 21:48:59 +00:00
pub mod spec;
pub mod webassembly;
#[derive(Debug, StructOpt)]
#[structopt(name = "wasmer", about = "WASM execution runtime.")]
2018-10-14 21:47:35 +00:00
/// The options for the wasmer Command Line Interface
enum CLIOptions {
/// Run a WebAssembly file. Formats accepted: wasm, wast
#[structopt(name = "run")]
2018-10-14 21:48:59 +00:00
Run(Run),
2018-10-14 21:47:35 +00:00
}
#[derive(Debug, StructOpt)]
struct Run {
#[structopt(short = "d", long = "debug")]
debug: bool,
/// Input file
#[structopt(parse(from_os_str))]
path: PathBuf,
}
2018-10-14 21:48:59 +00:00
/// Read the contents of a file
fn read_file_contents(path: PathBuf) -> Result<Vec<u8>, io::Error> {
let mut buffer: Vec<u8> = Vec::new();
let mut file = File::open(path)?;
file.read_to_end(&mut buffer)?;
Ok(buffer)
}
/// Execute a WASM/WAT file
2018-10-14 21:48:59 +00:00
fn execute_wasm(wasm_path: PathBuf) -> Result<(), String> {
let mut wasm_binary: Vec<u8> =
read_file_contents(wasm_path).map_err(|err| String::from(err.description()))?;
if !webassembly::utils::is_wasm_binary(&wasm_binary) {
2018-10-14 21:48:59 +00:00
wasm_binary = wat2wasm(wasm_binary).map_err(|err| String::from(err.description()))?;
}
2018-10-17 14:45:24 +00:00
let import_object = integrations::generate_libc_env();
let webassembly::ResultObject {module, instance} = webassembly::instantiate(wasm_binary, Some(import_object)).map_err(|err| String::from(err.description()))?;
let func_index = instance.start_func.unwrap_or_else(|| {
match module.info.exports.get("main") {
Some(&webassembly::Export::Function(index)) => index,
_ => panic!("Main function not found"),
}
});
let main: fn() = get_instance_function!(instance, func_index);
main();
Ok(())
}
2018-10-14 21:47:35 +00:00
fn run(options: Run) {
match execute_wasm(options.path.clone()) {
Ok(()) => {}
Err(message) => {
2018-10-14 21:47:35 +00:00
let name = options.path.as_os_str().to_string_lossy();
println!("error while executing {}: {}", name, message);
exit(1);
}
}
}
2018-10-14 21:47:35 +00:00
fn main() {
let options = CLIOptions::from_args();
match options {
CLIOptions::Run(options) => run(options),
}
}