Update wasmstandalone for API changes.

This updates to the latest faerie and cretonne API changes.
This commit is contained in:
Dan Gohman
2018-02-22 09:44:02 -08:00
parent ebf2c3a17e
commit f276a021cb
9 changed files with 115 additions and 47 deletions

View File

@@ -25,9 +25,9 @@ cretonne-native = { git = "https://github.com/stoklund/cretonne.git" }
wasmstandalone_runtime = { path = "lib/runtime" } wasmstandalone_runtime = { path = "lib/runtime" }
wasmstandalone_execute = { path = "lib/execute" } wasmstandalone_execute = { path = "lib/execute" }
wasmstandalone_obj = { path = "lib/obj" } wasmstandalone_obj = { path = "lib/obj" }
docopt = "0.8.0" docopt = "0.8.3"
serde = "1.0.8" serde = "1.0.27"
serde_derive = "1.0.8" serde_derive = "1.0.27"
tempdir = "*" tempdir = "*"
faerie = { git = "https://github.com/m4b/faerie" } faerie = { git = "https://github.com/m4b/faerie" }

View File

@@ -10,5 +10,5 @@ license = "MIT/Apache-2.0"
[dependencies] [dependencies]
cretonne = { git = "https://github.com/stoklund/cretonne.git" } cretonne = { git = "https://github.com/stoklund/cretonne.git" }
cretonne-wasm = { git = "https://github.com/stoklund/cretonne.git" } cretonne-wasm = { git = "https://github.com/stoklund/cretonne.git" }
region = "0.0.8" region = "0.1.1"
wasmstandalone_runtime = { path = "../runtime" } wasmstandalone_runtime = { path = "../runtime" }

View File

@@ -38,22 +38,36 @@ fn relocate(compilation: &mut Compilation, relocations: &wasmstandalone_runtime:
// The relocations are relative to the relocation's address plus four bytes // The relocations are relative to the relocation's address plus four bytes
// TODO: Support architectures other than x64, and other reloc kinds. // TODO: Support architectures other than x64, and other reloc kinds.
for (i, function_relocs) in relocations.iter().enumerate() { for (i, function_relocs) in relocations.iter().enumerate() {
for &(_reloc, func_index, offset) in function_relocs { for ref r in function_relocs {
let target_func_address: isize = compilation.functions[func_index].as_ptr() as isize; let target_func_address: isize = compilation.functions[r.func_index].as_ptr() as isize;
let body = &mut compilation.functions[i]; let body = &mut compilation.functions[i];
unsafe { unsafe {
let reloc_address: isize = body.as_mut_ptr().offset(offset as isize + 4) as isize; let reloc_address = body.as_mut_ptr().offset(r.offset as isize + 4) as isize;
let reloc_delta_i32: i32 = (target_func_address - reloc_address) as i32; let reloc_addend = r.addend as isize;
let reloc_delta_i32 = (target_func_address - reloc_address + reloc_addend) as i32;
write_unaligned(reloc_address as *mut i32, reloc_delta_i32); write_unaligned(reloc_address as *mut i32, reloc_delta_i32);
} }
} }
} }
} }
/// Create the VmCtx data structure for the JIT'd code to use. This must
/// match the VmCtx layout in the runtime.
fn make_vmctx(instance: &mut wasmstandalone_runtime::Instance) -> Vec<*mut u8> {
let mut memories = Vec::new();
let mut vmctx = Vec::new();
vmctx.push(instance.globals.as_mut_ptr());
for mem in &mut instance.memories {
memories.push(mem.as_mut_ptr());
}
vmctx.push(memories.as_mut_ptr() as *mut u8);
vmctx
}
/// Jumps to the code region of memory and execute the start function of the module. /// Jumps to the code region of memory and execute the start function of the module.
pub fn execute( pub fn execute(
compilation: &wasmstandalone_runtime::Compilation, compilation: &wasmstandalone_runtime::Compilation,
_instance: &wasmstandalone_runtime::Instance, instance: &mut wasmstandalone_runtime::Instance,
) -> Result<(), String> { ) -> Result<(), String> {
let start_index = compilation.module.start_func.ok_or_else(|| { let start_index = compilation.module.start_func.ok_or_else(|| {
String::from("No start function defined, aborting execution") String::from("No start function defined, aborting execution")
@@ -70,17 +84,20 @@ pub fn execute(
Err(err) => { Err(err) => {
return Err(format!( return Err(format!(
"failed to give executable permission to code: {}", "failed to give executable permission to code: {}",
err.description() err
)) ))
} }
} }
let vmctx = make_vmctx(instance);
// Rather than writing inline assembly to jump to the code region, we use the fact that // Rather than writing inline assembly to jump to the code region, we use the fact that
// the Rust ABI for calling a function with no arguments and no return matches the one of // the Rust ABI for calling a function with no arguments and no return matches the one of
// the generated code.Thanks to this, we can transmute the code region into a first-class // the generated code.Thanks to this, we can transmute the code region into a first-class
// Rust function and call it. // Rust function and call it.
unsafe { unsafe {
let start_func = transmute::<_, fn()>(code_buf.as_ptr()); let start_func = transmute::<_, fn(*const *mut u8)>(code_buf.as_ptr());
start_func(); start_func(vmctx.as_ptr());
} }
Ok(()) Ok(())
} }

View File

@@ -1,9 +1,8 @@
use cretonne::ir;
use cretonne::settings; use cretonne::settings;
use cretonne::settings::Configurable; use cretonne::settings::Configurable;
use faerie::Artifact; use faerie::Artifact;
use wasmstandalone_runtime; use wasmstandalone_runtime;
use std::error::Error;
use std::str;
/// Emits a module that has been emitted with the `WasmRuntime` runtime /// Emits a module that has been emitted with the `WasmRuntime` runtime
/// implementation to a native object file. /// implementation to a native object file.
@@ -28,11 +27,19 @@ pub fn emit_module<'module>(
let body = &compilation.functions[i]; let body = &compilation.functions[i];
let external_name = let external_name =
wasmstandalone_runtime::get_func_name(compilation.module.imported_funcs.len() + i); wasmstandalone_runtime::get_func_name(compilation.module.imported_funcs.len() + i);
let string_name = str::from_utf8(external_name.as_ref()).map_err(|err| { let func_index = match external_name {
err.description().to_string() ir::ExternalName::User { namespace, index } => {
})?; debug_assert!(namespace == 0);
index
}
_ => panic!(),
};
obj.add_code(string_name, body.clone()); let string_name = format!("wasm_function[{}]", func_index);
obj.define(string_name, body.clone()).map_err(|err| {
format!("{}", err)
})?;
} }
Ok(()) Ok(())

View File

@@ -10,4 +10,4 @@ license = "Apache-2.0"
[dependencies] [dependencies]
cretonne = { git = "https://github.com/stoklund/cretonne.git" } cretonne = { git = "https://github.com/stoklund/cretonne.git" }
cretonne-wasm = { git = "https://github.com/stoklund/cretonne.git" } cretonne-wasm = { git = "https://github.com/stoklund/cretonne.git" }
wasmparser = "0.13.0" wasmparser = "0.14.1"

View File

@@ -4,6 +4,7 @@
use cretonne::ir; use cretonne::ir;
use cton_wasm::GlobalIndex; use cton_wasm::GlobalIndex;
use module::Module; use module::Module;
use DataInitializer;
const PAGE_SIZE: usize = 65536; const PAGE_SIZE: usize = 65536;
@@ -22,14 +23,14 @@ pub struct Instance {
impl Instance { impl Instance {
/// Create a new `Instance`. /// Create a new `Instance`.
pub fn new(module: &Module) -> Self { pub fn new(module: &Module, data_initializers: &[DataInitializer]) -> Self {
let mut result = Self { let mut result = Self {
tables: Vec::new(), tables: Vec::new(),
memories: Vec::new(), memories: Vec::new(),
globals: Vec::new(), globals: Vec::new(),
}; };
result.instantiate_tables(module); result.instantiate_tables(module);
result.instantiate_memories(module); result.instantiate_memories(module, data_initializers);
result.instantiate_globals(module); result.instantiate_globals(module);
result result
} }
@@ -49,7 +50,7 @@ impl Instance {
/// Allocate memory in `instance` for just the memories of the current module, /// Allocate memory in `instance` for just the memories of the current module,
/// without any initializers applied yet. /// without any initializers applied yet.
fn instantiate_memories(&mut self, module: &Module) { fn instantiate_memories(&mut self, module: &Module, data_initializers: &[DataInitializer]) {
debug_assert!(self.memories.is_empty()); debug_assert!(self.memories.is_empty());
// Allocate the underlying memory and initialize it to all zeros. // Allocate the underlying memory and initialize it to all zeros.
self.memories.reserve_exact(module.memories.len()); self.memories.reserve_exact(module.memories.len());
@@ -59,6 +60,12 @@ impl Instance {
v.resize(len, 0); v.resize(len, 0);
self.memories.push(v); self.memories.push(v);
} }
for init in data_initializers {
debug_assert!(init.base.is_none(), "globalvar base not supported yet");
let to_init = &mut self.memories[init.memory_index][init.offset..
init.offset + init.data.len()];
to_init.copy_from_slice(init.data);
}
} }
/// Allocate memory in `instance` for just the globals of the current module, /// Allocate memory in `instance` for just the globals of the current module,

View File

@@ -28,12 +28,12 @@ use cretonne::ir;
use cretonne::isa; use cretonne::isa;
use cretonne::settings; use cretonne::settings;
use cretonne::binemit; use cretonne::binemit;
use std::str::from_utf8;
use std::error::Error; use std::error::Error;
/// Compute a `ir::ExternalName` for a given wasm function index. /// Compute a `ir::ExternalName` for a given wasm function index.
pub fn get_func_name(func_index: FunctionIndex) -> cretonne::ir::ExternalName { pub fn get_func_name(func_index: FunctionIndex) -> cretonne::ir::ExternalName {
ir::ExternalName::new(format!("wasm_0x{:x}", func_index)) debug_assert!(func_index as u32 as FunctionIndex == func_index);
ir::ExternalName::user(0, func_index as u32)
} }
/// An entity to export. /// An entity to export.
@@ -48,12 +48,11 @@ pub enum Export {
Global(GlobalIndex), Global(GlobalIndex),
} }
type RelocRef = u16; /// Implementation of a relocation sink that just saves all the information for later
pub struct RelocSink<'func> {
// Implementation of a relocation sink that just saves all the information for later
struct RelocSink<'func> {
func: &'func ir::Function, func: &'func ir::Function,
pub func_relocs: Vec<(RelocRef, FunctionIndex, binemit::CodeOffset)>, /// Relocations recorded for the function.
pub func_relocs: Vec<Relocation>,
} }
impl<'func> binemit::RelocSink for RelocSink<'func> { impl<'func> binemit::RelocSink for RelocSink<'func> {
@@ -71,14 +70,21 @@ impl<'func> binemit::RelocSink for RelocSink<'func> {
offset: binemit::CodeOffset, offset: binemit::CodeOffset,
reloc: binemit::Reloc, reloc: binemit::Reloc,
name: &ExternalName, name: &ExternalName,
addend: binemit::Addend,
) { ) {
let name_bytes: &[u8] = name.as_ref();
let name = from_utf8(name_bytes).unwrap();
// See `get_func_name`; names are encoded as `wasm_0x...`, so grab the
// part after `0x...` and convert it back to an integer to get the index.
// FIXME: Handle grow_memory/current_memory. // FIXME: Handle grow_memory/current_memory.
let func_index = FunctionIndex::from_str_radix(&name[7..], 16).unwrap(); let func_index = if let ExternalName::User { namespace, index } = *name {
self.func_relocs.push((reloc.0, func_index, offset)); debug_assert!(namespace == 0);
index
} else {
panic!("unrecognized external name")
} as usize;
self.func_relocs.push(Relocation {
reloc,
func_index,
offset,
addend,
});
} }
fn reloc_jt( fn reloc_jt(
&mut self, &mut self,
@@ -100,6 +106,18 @@ impl<'func> RelocSink<'func> {
} }
} }
/// A data initializer for linear memory.
pub struct DataInitializer<'data> {
/// The index of the memory to initialize.
pub memory_index: MemoryIndex,
/// Optionally a globalvar base to initialize at.
pub base: Option<GlobalIndex>,
/// A constant offset to initialize at.
pub offset: usize,
/// The initialization data.
pub data: &'data [u8],
}
/// References to the input wasm data buffer to be decoded and processed later. /// References to the input wasm data buffer to be decoded and processed later.
/// separately from the main module translation. /// separately from the main module translation.
pub struct LazyContents<'data> { pub struct LazyContents<'data> {
@@ -107,7 +125,7 @@ pub struct LazyContents<'data> {
pub function_body_inputs: Vec<&'data [u8]>, pub function_body_inputs: Vec<&'data [u8]>,
/// References to the data initializers. /// References to the data initializers.
pub data_initializers: Vec<(MemoryIndex, Option<GlobalIndex>, usize, &'data [u8])>, pub data_initializers: Vec<DataInitializer<'data>>,
} }
impl<'data> LazyContents<'data> { impl<'data> LazyContents<'data> {
@@ -259,12 +277,13 @@ impl<'module_environment> cton_wasm::FuncEnvironment for FuncEnvironment<'module
base: memories_base, base: memories_base,
offset: Offset32::new(offset32), offset: Offset32::new(offset32),
}); });
func.create_heap(ir::HeapData { let h = func.create_heap(ir::HeapData {
base: ir::HeapBase::GlobalVar(heap_base), base: ir::HeapBase::GlobalVar(heap_base),
min_size: 0.into(), min_size: 0.into(),
guard_size: 0x8000_0000.into(), guard_size: 0x8000_0000.into(),
style: ir::HeapStyle::Static { bound: 0x1_0000_0000.into() }, style: ir::HeapStyle::Static { bound: 0x1_0000_0000.into() },
}) });
h
} }
fn make_indirect_sig(&mut self, func: &mut ir::Function, index: SignatureIndex) -> ir::SigRef { fn make_indirect_sig(&mut self, func: &mut ir::Function, index: SignatureIndex) -> ir::SigRef {
@@ -320,8 +339,9 @@ impl<'module_environment> cton_wasm::FuncEnvironment for FuncEnvironment<'module
params: vec![AbiParam::new(I32)], params: vec![AbiParam::new(I32)],
returns: vec![AbiParam::new(I32)], returns: vec![AbiParam::new(I32)],
}); });
// FIXME: Use a real ExternalName system.
pos.func.import_function(ExtFuncData { pos.func.import_function(ExtFuncData {
name: ExternalName::new("grow_memory"), name: ExternalName::testcase("grow_memory"),
signature: sig_ref, signature: sig_ref,
}) })
}); });
@@ -344,8 +364,9 @@ impl<'module_environment> cton_wasm::FuncEnvironment for FuncEnvironment<'module
params: Vec::new(), params: Vec::new(),
returns: vec![AbiParam::new(I32)], returns: vec![AbiParam::new(I32)],
}); });
// FIXME: Use a real ExternalName system.
pos.func.import_function(ExtFuncData { pos.func.import_function(ExtFuncData {
name: ExternalName::new("current_memory"), name: ExternalName::testcase("current_memory"),
signature: sig_ref, signature: sig_ref,
}) })
}); });
@@ -446,12 +467,12 @@ impl<'data, 'module> cton_wasm::ModuleEnvironment<'data> for ModuleEnvironment<'
data: &'data [u8], data: &'data [u8],
) { ) {
debug_assert!(base.is_none(), "global-value offsets not supported yet"); debug_assert!(base.is_none(), "global-value offsets not supported yet");
self.lazy.data_initializers.push(( self.lazy.data_initializers.push(DataInitializer {
memory_index, memory_index,
base, base,
offset, offset,
data, data,
)); });
} }
fn declare_func_export(&mut self, func_index: FunctionIndex, name: &str) { fn declare_func_export(&mut self, func_index: FunctionIndex, name: &str) {
@@ -493,8 +514,21 @@ impl<'data, 'module> cton_wasm::ModuleEnvironment<'data> for ModuleEnvironment<'
} }
} }
/// A record of a relocation to perform.
#[derive(Debug)]
pub struct Relocation {
/// The relocation code.
pub reloc: binemit::Reloc,
/// The function index.
pub func_index: FunctionIndex,
/// The offset where to apply the relocation.
pub offset: binemit::CodeOffset,
/// The addend to add to the relocation value.
pub addend: binemit::Addend,
}
/// Relocations to apply to function bodies. /// Relocations to apply to function bodies.
pub type Relocations = Vec<Vec<(RelocRef, FunctionIndex, binemit::CodeOffset)>>; pub type Relocations = Vec<Vec<Relocation>>;
/// The result of translating via `ModuleEnvironment`. /// The result of translating via `ModuleEnvironment`.
pub struct ModuleTranslation<'data, 'module> { pub struct ModuleTranslation<'data, 'module> {

View File

@@ -126,7 +126,8 @@ fn handle_module(args: &Args, path: PathBuf, isa: &TargetIsa) -> Result<(), Stri
let translation = environ.finish_translation(); let translation = environ.finish_translation();
let instance = match compile_module(isa, &translation) { let instance = match compile_module(isa, &translation) {
Ok(compilation) => { Ok(compilation) => {
let mut instance = Instance::new(compilation.module); let mut instance =
Instance::new(compilation.module, &translation.lazy.data_initializers);
execute(&compilation, &mut instance)?; execute(&compilation, &mut instance)?;
instance instance
} }

View File

@@ -95,12 +95,14 @@ fn handle_module(path: PathBuf, output: &str) -> Result<(), String> {
let mut environ = wasmstandalone_runtime::ModuleEnvironment::new(isa.flags(), &mut module); let mut environ = wasmstandalone_runtime::ModuleEnvironment::new(isa.flags(), &mut module);
translate_module(&data, &mut environ)?; translate_module(&data, &mut environ)?;
let mut obj = Artifact::new(faerie_target(&*isa)?, Some(String::from(output))); let mut obj = Artifact::new(faerie_target(&*isa)?, String::from(output));
// FIXME: We need to initialize memory in a way that supports alternate // FIXME: We need to initialize memory in a way that supports alternate
// memory spaces, imported base addresses, and offsets. // memory spaces, imported base addresses, and offsets.
for &(_mem_index, _base, _offset, data) in &environ.lazy.data_initializers { for init in &environ.lazy.data_initializers {
obj.add_data("memory", Vec::from(data)); obj.define("memory", Vec::from(init.data)).map_err(|err| {
format!("{}", err)
})?;
} }
let translation = environ.finish_translation(); let translation = environ.finish_translation();