wasmer/src/main.rs

115 lines
2.9 KiB
Rust
Raw Normal View History

2018-10-23 23:15:20 +00:00
#![feature(test, libc, core_intrinsics)]
extern crate test;
#[macro_use]
extern crate error_chain;
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-11-06 14:51:01 +00:00
extern crate structopt;
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;
extern crate nix;
2018-10-14 11:59:11 +00:00
extern crate spin;
2018-10-23 21:54:07 +00:00
// use std::alloc::System;
2018-11-06 14:51:01 +00:00
// use std::time::{Duration, Instant};
2018-10-23 09:40:17 +00:00
// #[global_allocator]
// static A: System = System;
2018-10-15 00:48:59 +00:00
2018-11-14 01:21:03 +00:00
#[macro_use]
extern crate log;
2018-10-15 00:48:59 +00:00
2018-11-06 14:51:01 +00:00
// 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;
2018-11-14 03:19:23 +00:00
pub mod linkers;
pub mod sighandler;
#[cfg(test)]
mod spectests;
2018-10-24 00:01:46 +00:00
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-11-14 03:19:23 +00:00
let import_object = linkers::generate_emscripten_env();
2018-11-06 14:51:01 +00:00
let webassembly::ResultObject { module, instance } =
webassembly::instantiate(wasm_binary, import_object)
2018-11-14 02:41:51 +00:00
.map_err(|err| format!("{}", err))?;
2018-11-07 10:47:06 +00:00
// webassembly::utils::print_instance_offsets(&instance);
let func_index = instance
.start_func
.unwrap_or_else(|| match module.info.exports.get("main") {
2018-10-17 14:45:24 +00:00
Some(&webassembly::Export::Function(index)) => index,
_ => panic!("Main function not found"),
});
2018-11-14 01:21:03 +00:00
let main: fn(&webassembly::Instance) = get_instance_function!(instance, func_index);
main(&instance);
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),
}
}