2018-10-23 23:15:20 +00:00
|
|
|
#![feature(test, libc, core_intrinsics)]
|
2018-10-16 11:27:26 +00:00
|
|
|
|
|
|
|
extern crate test;
|
2018-10-11 19:29:36 +00:00
|
|
|
#[macro_use]
|
|
|
|
extern crate error_chain;
|
|
|
|
extern crate cranelift_codegen;
|
2018-10-14 21:48:59 +00:00
|
|
|
extern crate cranelift_entity;
|
2018-10-11 19:29:36 +00:00
|
|
|
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;
|
2018-11-06 10:15:40 +00:00
|
|
|
extern crate nix;
|
2018-10-14 11:59:11 +00:00
|
|
|
extern crate spin;
|
2018-10-11 19:29:36 +00:00
|
|
|
|
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;
|
2018-10-11 19:29:36 +00:00
|
|
|
use std::fs::File;
|
|
|
|
use std::io;
|
|
|
|
use std::io::Read;
|
2018-10-14 21:48:59 +00:00
|
|
|
use std::path::PathBuf;
|
2018-10-11 19:29:36 +00:00
|
|
|
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;
|
2018-11-06 10:15:40 +00:00
|
|
|
pub mod sighandler;
|
2018-10-24 00:01:16 +00:00
|
|
|
#[cfg(test)]
|
|
|
|
mod spectests;
|
2018-10-24 00:01:46 +00:00
|
|
|
pub mod webassembly;
|
2018-10-11 19:29:36 +00:00
|
|
|
|
|
|
|
#[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 {
|
2018-10-11 19:29:36 +00:00
|
|
|
#[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
|
2018-10-11 19:29:36 +00:00
|
|
|
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()))?;
|
2018-10-11 19:29:36 +00:00
|
|
|
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-11 19:29:36 +00:00
|
|
|
}
|
|
|
|
|
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);
|
|
|
|
|
2018-10-17 22:09:04 +00:00
|
|
|
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-10-17 22:09:04 +00:00
|
|
|
});
|
2018-11-14 01:21:03 +00:00
|
|
|
let main: fn(&webassembly::Instance) = get_instance_function!(instance, func_index);
|
|
|
|
main(&instance);
|
2018-10-11 19:29:36 +00:00
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
2018-10-14 21:47:35 +00:00
|
|
|
fn run(options: Run) {
|
|
|
|
match execute_wasm(options.path.clone()) {
|
2018-10-11 19:29:36 +00:00
|
|
|
Ok(()) => {}
|
|
|
|
Err(message) => {
|
2018-10-14 21:47:35 +00:00
|
|
|
let name = options.path.as_os_str().to_string_lossy();
|
2018-10-11 19:29:36 +00:00
|
|
|
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),
|
|
|
|
}
|
|
|
|
}
|