use crate::{
HashMap,
abi::{self},
heap::{Heap, HeapData, HeapStyle},
table::{TableData, TableSize},
translator::{EXN_REF_TYPE, LandingPad, TAG_TYPE, materialize_global_value},
};
use cranelift_codegen::{
cursor::FuncCursor,
ir::{
self, AbiParam, ArgumentPurpose, BlockArg, Endianness, ExceptionTableData,
ExceptionTableItem, ExceptionTag, Function, InstBuilder, MemFlagsData, Signature,
UserExternalName,
condcodes::IntCC,
immediates::{Offset32, Uimm64},
types::*,
},
isa::TargetFrontendConfig,
};
use cranelift_frontend::FunctionBuilder;
use smallvec::SmallVec;
use std::convert::TryFrom;
use target_lexicon::Architecture;
use wasmer_compiler::abi::ReturnAbi;
use wasmer_compiler::wasmparser::HeapType;
use wasmer_types::{
FunctionIndex, GlobalIndex, LocalFunctionIndex, MemoryIndex, MemoryStyle, ModuleInfo,
SignatureHash, SignatureIndex, TableIndex, TableStyle, TagIndex, Type as WasmerType,
VMBuiltinFunctionIndex, VMOffsets, WasmError, WasmResult,
entity::{EntityRef, PrimaryMap, SecondaryMap},
vmctx_offset,
};
fn insert_mem_flags(func: &mut Function, flags: ir::MemFlagsData) -> ir::MemFlags {
func.dfg.mem_flags.insert(flags).unwrap()
}
pub fn get_function_name(func: &mut Function, func_index: FunctionIndex) -> ir::ExternalName {
ir::ExternalName::user(
func.params
.ensure_user_func_name(UserExternalName::new(0, func_index.as_u32())),
)
}
#[allow(unused)]
pub fn type_of_vmtable_definition_current_elements(vmoffsets: &VMOffsets) -> ir::Type {
ir::Type::int(u16::from(vmoffsets.size_of_vmtable_definition_current_elements()) * 8).unwrap()
}
#[derive(Clone)]
struct ExceptionFieldLayout {
offset: u32,
ty: ir::Type,
}
#[derive(Clone)]
struct ExceptionTypeLayout {
fields: SmallVec<[ExceptionFieldLayout; 4]>,
}
#[derive(Clone, Copy)]
pub enum GlobalVariable {
#[allow(dead_code)]
Const(ir::Value),
Memory {
gv: ir::GlobalValue,
offset: Offset32,
ty: ir::Type,
},
#[allow(dead_code)]
Custom,
}
pub struct FuncEnvironment<'module_environment> {
target_config: TargetFrontendConfig,
architecture: Architecture,
return_types: Vec<WasmerType>,
module: &'module_environment ModuleInfo,
type_stack: Vec<WasmerType>,
signatures: &'module_environment PrimaryMap<SignatureIndex, ir::Signature>,
signature_hashes: &'module_environment PrimaryMap<SignatureIndex, SignatureHash>,
heaps: PrimaryMap<Heap, HeapData>,
vmctx: Option<ir::GlobalValue>,
memory32_size_sig: Option<ir::SigRef>,
table_size_sig: Option<ir::SigRef>,
memory_grow_sig: Option<ir::SigRef>,
table_grow_sig: Option<ir::SigRef>,
table_copy_sig: Option<ir::SigRef>,
table_init_sig: Option<ir::SigRef>,
elem_drop_sig: Option<ir::SigRef>,
memory_copy_sig: Option<ir::SigRef>,
memory_fill_sig: Option<ir::SigRef>,
memory_init_sig: Option<ir::SigRef>,
data_drop_sig: Option<ir::SigRef>,
table_get_sig: Option<ir::SigRef>,
table_set_sig: Option<ir::SigRef>,
func_ref_sig: Option<ir::SigRef>,
table_fill_sig: Option<ir::SigRef>,
memory32_atomic_wait32_sig: Option<ir::SigRef>,
memory32_atomic_wait64_sig: Option<ir::SigRef>,
memory32_atomic_notify_sig: Option<ir::SigRef>,
raise_trap_sig: Option<ir::SigRef>,
personality2_sig: Option<ir::SigRef>,
throw_sig: Option<ir::SigRef>,
alloc_exception_sig: Option<ir::SigRef>,
read_exception_sig: Option<ir::SigRef>,
read_exnref_sig: Option<ir::SigRef>,
exception_type_layouts: HashMap<u32, ExceptionTypeLayout>,
offsets: VMOffsets,
memory_styles: &'module_environment PrimaryMap<MemoryIndex, MemoryStyle>,
tables: SecondaryMap<TableIndex, Option<TableData>>,
table_styles: &'module_environment PrimaryMap<TableIndex, TableStyle>,
}
impl<'module_environment> FuncEnvironment<'module_environment> {
pub fn new(
target_config: TargetFrontendConfig,
architecture: Architecture,
module: &'module_environment ModuleInfo,
signatures: &'module_environment PrimaryMap<SignatureIndex, ir::Signature>,
signature_hashes: &'module_environment PrimaryMap<SignatureIndex, SignatureHash>,
memory_styles: &'module_environment PrimaryMap<MemoryIndex, MemoryStyle>,
table_styles: &'module_environment PrimaryMap<TableIndex, TableStyle>,
) -> Self {
Self {
target_config,
architecture,
return_types: Vec::new(),
module,
signatures,
signature_hashes,
type_stack: vec![],
heaps: PrimaryMap::new(),
vmctx: None,
memory32_size_sig: None,
table_size_sig: None,
memory_grow_sig: None,
table_grow_sig: None,
table_copy_sig: None,
table_init_sig: None,
elem_drop_sig: None,
memory_copy_sig: None,
memory_fill_sig: None,
memory_init_sig: None,
table_get_sig: None,
table_set_sig: None,
data_drop_sig: None,
func_ref_sig: None,
table_fill_sig: None,
memory32_atomic_wait32_sig: None,
memory32_atomic_wait64_sig: None,
memory32_atomic_notify_sig: None,
raise_trap_sig: None,
personality2_sig: None,
throw_sig: None,
alloc_exception_sig: None,
read_exception_sig: None,
read_exnref_sig: None,
exception_type_layouts: HashMap::new(),
offsets: VMOffsets::new(target_config.pointer_bytes(), module),
memory_styles,
tables: Default::default(),
table_styles,
}
}
pub(crate) fn target_config(&self) -> TargetFrontendConfig {
self.target_config
}
pub(crate) fn pointer_type(&self) -> ir::Type {
self.target_config.pointer_type()
}
pub(crate) fn reference_type(&self) -> ir::Type {
self.target_config.pointer_type()
}
fn ensure_table_exists(
&mut self,
func: &mut ir::Function,
index: TableIndex,
) -> WasmResult<()> {
if self.tables[index].is_some() {
return Ok(());
}
let pointer_type = self.pointer_type();
let table = &self.module.tables[index];
let (base_gv, table_base_offset, bound, element_size, inline_anyfunc) =
if let Some(def_index) = self.module.local_table_index(index)
&& table.is_fixed_funcref_table()
{
(
self.vmctx(func),
vmctx_offset(
self.offsets
.vmctx_fixed_funcref_table_anyfuncs(def_index)
.expect("fixed funcref table must have inline VMContext storage"),
)?,
TableSize::Static {
bound: table.minimum,
},
u32::from(self.offsets.size_of_vmcaller_checked_anyfunc()),
true,
)
} else {
let (ptr, base_offset, current_elements_offset) = {
let vmctx = self.vmctx(func);
if let Some(def_index) = self.module.local_table_index(index) {
let base_offset =
vmctx_offset(self.offsets.vmctx_vmtable_definition_base(def_index))?;
let current_elements_offset = vmctx_offset(
self.offsets
.vmctx_vmtable_definition_current_elements(def_index),
)?;
(vmctx, base_offset, current_elements_offset)
} else {
let from_offset = self.offsets.vmctx_vmtable_import(index);
let flags = insert_mem_flags(func, MemFlagsData::trusted().with_readonly());
let table = func.create_global_value(ir::GlobalValueData::Load {
base: vmctx,
offset: Offset32::new(vmctx_offset(from_offset)?),
global_type: pointer_type,
flags,
});
let base_offset = i32::from(self.offsets.vmtable_definition_base());
let current_elements_offset =
i32::from(self.offsets.vmtable_definition_current_elements());
(table, base_offset, current_elements_offset)
}
};
let flags = if Some(table.minimum) == table.maximum {
insert_mem_flags(func, MemFlagsData::trusted().with_readonly())
} else {
insert_mem_flags(func, MemFlagsData::trusted())
};
let base_gv = func.create_global_value(ir::GlobalValueData::Load {
base: ptr,
offset: Offset32::new(base_offset),
global_type: pointer_type,
flags,
});
let bound = if Some(table.minimum) == table.maximum {
TableSize::Static {
bound: table.minimum,
}
} else {
let flags = insert_mem_flags(func, MemFlagsData::trusted());
TableSize::Dynamic {
bound_gv: func.create_global_value(ir::GlobalValueData::Load {
base: ptr,
offset: Offset32::new(current_elements_offset),
global_type: ir::Type::int(
u16::from(
self.offsets.size_of_vmtable_definition_current_elements(),
) * 8,
)
.unwrap(),
flags,
}),
}
};
(base_gv, 0, bound, self.reference_type().bytes(), false)
};
self.tables[index] = Some(TableData {
base_gv,
base_offset: table_base_offset,
bound,
element_size,
inline_anyfunc,
});
Ok(())
}
fn vmctx(&mut self, func: &mut Function) -> ir::GlobalValue {
self.vmctx.unwrap_or_else(|| {
let vmctx = func.create_global_value(ir::GlobalValueData::VMContext);
self.vmctx = Some(vmctx);
vmctx
})
}
fn get_table_fill_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.table_fill_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(self.reference_type()),
AbiParam::new(I32),
],
returns: vec![],
call_conv: self.target_config.default_call_conv,
})
});
self.table_fill_sig = Some(sig);
sig
}
fn get_table_fill_func(
&mut self,
func: &mut Function,
table_index: TableIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
(
self.get_table_fill_sig(func),
table_index.index(),
VMBuiltinFunctionIndex::get_table_fill_index(),
)
}
fn get_func_ref_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.func_ref_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
],
returns: vec![AbiParam::new(self.reference_type())],
call_conv: self.target_config.default_call_conv,
})
});
self.func_ref_sig = Some(sig);
sig
}
fn get_func_ref_func(
&mut self,
func: &mut Function,
function_index: FunctionIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
(
self.get_func_ref_sig(func),
function_index.index(),
VMBuiltinFunctionIndex::get_func_ref_index(),
)
}
fn get_table_get_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.table_get_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
AbiParam::new(I32),
],
returns: vec![AbiParam::new(self.reference_type())],
call_conv: self.target_config.default_call_conv,
})
});
self.table_get_sig = Some(sig);
sig
}
fn get_table_get_func(
&mut self,
func: &mut Function,
table_index: TableIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
if self.module.is_imported_table(table_index) {
(
self.get_table_get_sig(func),
table_index.index(),
VMBuiltinFunctionIndex::get_imported_table_get_index(),
)
} else {
(
self.get_table_get_sig(func),
self.module.local_table_index(table_index).unwrap().index(),
VMBuiltinFunctionIndex::get_table_get_index(),
)
}
}
fn get_table_set_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.table_set_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(self.reference_type()),
],
returns: vec![],
call_conv: self.target_config.default_call_conv,
})
});
self.table_set_sig = Some(sig);
sig
}
fn get_table_set_func(
&mut self,
func: &mut Function,
table_index: TableIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
if self.module.is_imported_table(table_index) {
(
self.get_table_set_sig(func),
table_index.index(),
VMBuiltinFunctionIndex::get_imported_table_set_index(),
)
} else {
(
self.get_table_set_sig(func),
self.module.local_table_index(table_index).unwrap().index(),
VMBuiltinFunctionIndex::get_table_set_index(),
)
}
}
fn get_table_grow_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.table_grow_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(self.reference_type()),
AbiParam::new(I32),
AbiParam::new(I32),
],
returns: vec![AbiParam::new(I32)],
call_conv: self.target_config.default_call_conv,
})
});
self.table_grow_sig = Some(sig);
sig
}
fn get_table_grow_func(
&mut self,
func: &mut Function,
index: TableIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
if self.module.is_imported_table(index) {
(
self.get_table_grow_sig(func),
index.index(),
VMBuiltinFunctionIndex::get_imported_table_grow_index(),
)
} else {
(
self.get_table_grow_sig(func),
self.module.local_table_index(index).unwrap().index(),
VMBuiltinFunctionIndex::get_table_grow_index(),
)
}
}
fn get_memory_grow_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.memory_grow_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
AbiParam::new(I32),
],
returns: vec![AbiParam::new(I32)],
call_conv: self.target_config.default_call_conv,
})
});
self.memory_grow_sig = Some(sig);
sig
}
fn get_memory_grow_func(
&mut self,
func: &mut Function,
index: MemoryIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
if self.module.is_imported_memory(index) {
(
self.get_memory_grow_sig(func),
index.index(),
VMBuiltinFunctionIndex::get_imported_memory32_grow_index(),
)
} else {
(
self.get_memory_grow_sig(func),
self.module.local_memory_index(index).unwrap().index(),
VMBuiltinFunctionIndex::get_memory32_grow_index(),
)
}
}
fn get_table_size_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.table_size_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
],
returns: vec![AbiParam::new(I32)],
call_conv: self.target_config.default_call_conv,
})
});
self.table_size_sig = Some(sig);
sig
}
fn get_table_size_func(
&mut self,
func: &mut Function,
index: TableIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
if self.module.is_imported_table(index) {
(
self.get_table_size_sig(func),
index.index(),
VMBuiltinFunctionIndex::get_imported_table_size_index(),
)
} else {
(
self.get_table_size_sig(func),
self.module.local_table_index(index).unwrap().index(),
VMBuiltinFunctionIndex::get_table_size_index(),
)
}
}
fn get_memory32_size_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.memory32_size_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
],
returns: vec![AbiParam::new(I32)],
call_conv: self.target_config.default_call_conv,
})
});
self.memory32_size_sig = Some(sig);
sig
}
fn get_memory_size_func(
&mut self,
func: &mut Function,
index: MemoryIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
if self.module.is_imported_memory(index) {
(
self.get_memory32_size_sig(func),
index.index(),
VMBuiltinFunctionIndex::get_imported_memory32_size_index(),
)
} else {
(
self.get_memory32_size_sig(func),
self.module.local_memory_index(index).unwrap().index(),
VMBuiltinFunctionIndex::get_memory32_size_index(),
)
}
}
fn get_table_copy_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.table_copy_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
],
returns: vec![],
call_conv: self.target_config.default_call_conv,
})
});
self.table_copy_sig = Some(sig);
sig
}
fn get_table_copy_func(
&mut self,
func: &mut Function,
dst_table_index: TableIndex,
src_table_index: TableIndex,
) -> (ir::SigRef, usize, usize, VMBuiltinFunctionIndex) {
let sig = self.get_table_copy_sig(func);
(
sig,
dst_table_index.as_u32() as usize,
src_table_index.as_u32() as usize,
VMBuiltinFunctionIndex::get_table_copy_index(),
)
}
fn get_table_init_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.table_init_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
],
returns: vec![],
call_conv: self.target_config.default_call_conv,
})
});
self.table_init_sig = Some(sig);
sig
}
fn get_table_init_func(
&mut self,
func: &mut Function,
table_index: TableIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
let sig = self.get_table_init_sig(func);
let table_index = table_index.as_u32() as usize;
(
sig,
table_index,
VMBuiltinFunctionIndex::get_table_init_index(),
)
}
fn get_elem_drop_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.elem_drop_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
],
returns: vec![],
call_conv: self.target_config.default_call_conv,
})
});
self.elem_drop_sig = Some(sig);
sig
}
fn get_elem_drop_func(&mut self, func: &mut Function) -> (ir::SigRef, VMBuiltinFunctionIndex) {
let sig = self.get_elem_drop_sig(func);
(sig, VMBuiltinFunctionIndex::get_elem_drop_index())
}
fn get_memory_copy_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.memory_copy_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
],
returns: vec![],
call_conv: self.target_config.default_call_conv,
})
});
self.memory_copy_sig = Some(sig);
sig
}
fn get_memory_copy_func(
&mut self,
func: &mut Function,
) -> (ir::SigRef, VMBuiltinFunctionIndex) {
(
self.get_memory_copy_sig(func),
VMBuiltinFunctionIndex::get_memory_copy_index(),
)
}
fn get_memory_fill_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.memory_fill_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
],
returns: vec![],
call_conv: self.target_config.default_call_conv,
})
});
self.memory_fill_sig = Some(sig);
sig
}
fn get_memory_fill_func(
&mut self,
func: &mut Function,
memory_index: MemoryIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
let sig = self.get_memory_fill_sig(func);
if let Some(local_memory_index) = self.module.local_memory_index(memory_index) {
(
sig,
local_memory_index.index(),
VMBuiltinFunctionIndex::get_memory_fill_index(),
)
} else {
(
sig,
memory_index.index(),
VMBuiltinFunctionIndex::get_imported_memory_fill_index(),
)
}
}
fn get_memory_init_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.memory_init_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
],
returns: vec![],
call_conv: self.target_config.default_call_conv,
})
});
self.memory_init_sig = Some(sig);
sig
}
fn get_memory_init_func(
&mut self,
func: &mut Function,
) -> (ir::SigRef, VMBuiltinFunctionIndex) {
let sig = self.get_memory_init_sig(func);
(sig, VMBuiltinFunctionIndex::get_memory_init_index())
}
fn get_data_drop_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.data_drop_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
],
returns: vec![],
call_conv: self.target_config.default_call_conv,
})
});
self.data_drop_sig = Some(sig);
sig
}
fn get_data_drop_func(&mut self, func: &mut Function) -> (ir::SigRef, VMBuiltinFunctionIndex) {
let sig = self.get_data_drop_sig(func);
(sig, VMBuiltinFunctionIndex::get_data_drop_index())
}
fn get_memory32_atomic_wait32_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.memory32_atomic_wait32_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I64),
],
returns: vec![AbiParam::new(I32)],
call_conv: self.target_config.default_call_conv,
})
});
self.memory32_atomic_wait32_sig = Some(sig);
sig
}
fn get_memory_atomic_wait32_func(
&mut self,
func: &mut Function,
index: MemoryIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
if self.module.is_imported_memory(index) {
(
self.get_memory32_atomic_wait32_sig(func),
index.index(),
VMBuiltinFunctionIndex::get_imported_memory_atomic_wait32_index(),
)
} else {
(
self.get_memory32_atomic_wait32_sig(func),
self.module.local_memory_index(index).unwrap().index(),
VMBuiltinFunctionIndex::get_memory_atomic_wait32_index(),
)
}
}
fn get_memory32_atomic_wait64_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.memory32_atomic_wait64_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I64),
AbiParam::new(I64),
],
returns: vec![AbiParam::new(I32)],
call_conv: self.target_config.default_call_conv,
})
});
self.memory32_atomic_wait64_sig = Some(sig);
sig
}
fn get_memory_atomic_wait64_func(
&mut self,
func: &mut Function,
index: MemoryIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
if self.module.is_imported_memory(index) {
(
self.get_memory32_atomic_wait64_sig(func),
index.index(),
VMBuiltinFunctionIndex::get_imported_memory_atomic_wait64_index(),
)
} else {
(
self.get_memory32_atomic_wait64_sig(func),
self.module.local_memory_index(index).unwrap().index(),
VMBuiltinFunctionIndex::get_memory_atomic_wait64_index(),
)
}
}
fn get_memory32_atomic_notify_sig(&mut self, func: &mut Function) -> ir::SigRef {
let sig = self.memory32_atomic_notify_sig.unwrap_or_else(|| {
func.import_signature(Signature {
params: vec![
AbiParam::special(self.pointer_type(), ArgumentPurpose::VMContext),
AbiParam::new(I32),
AbiParam::new(I32),
AbiParam::new(I32),
],
returns: vec![AbiParam::new(I32)],
call_conv: self.target_config.default_call_conv,
})
});
self.memory32_atomic_notify_sig = Some(sig);
sig
}
fn get_memory_atomic_notify_func(
&mut self,
func: &mut Function,
index: MemoryIndex,
) -> (ir::SigRef, usize, VMBuiltinFunctionIndex) {
if self.module.is_imported_memory(index) {
(
self.get_memory32_atomic_notify_sig(func),
index.index(),
VMBuiltinFunctionIndex::get_imported_memory_atomic_notify_index(),
)
} else {
(
self.get_memory32_atomic_notify_sig(func),
self.module.local_memory_index(index).unwrap().index(),
VMBuiltinFunctionIndex::get_memory_atomic_notify_index(),
)
}
}
fn get_personality2_func(
&mut self,
func: &mut Function,
) -> (ir::SigRef, VMBuiltinFunctionIndex) {
let sig = self.personality2_sig.unwrap_or_else(|| {
let mut signature = Signature::new(self.target_config.default_call_conv);
signature.params.push(AbiParam::new(self.pointer_type()));
signature.params.push(AbiParam::new(self.pointer_type()));
signature.returns.push(AbiParam::new(TAG_TYPE));
let sig = func.import_signature(signature);
self.personality2_sig = Some(sig);
sig
});
(
sig,
VMBuiltinFunctionIndex::get_imported_personality2_index(),
)
}
fn get_throw_func(&mut self, func: &mut Function) -> (ir::SigRef, VMBuiltinFunctionIndex) {
let sig = self.throw_sig.unwrap_or_else(|| {
let mut signature = Signature::new(self.target_config.default_call_conv);
signature.params.push(AbiParam::special(
self.pointer_type(),
ArgumentPurpose::VMContext,
));
signature.params.push(AbiParam::new(EXN_REF_TYPE));
let sig = func.import_signature(signature);
self.throw_sig = Some(sig);
sig
});
(sig, VMBuiltinFunctionIndex::get_imported_throw_index())
}
fn get_raise_trap_func(&mut self, func: &mut Function) -> (ir::SigRef, VMBuiltinFunctionIndex) {
let sig = self.raise_trap_sig.unwrap_or_else(|| {
let mut signature = Signature::new(self.target_config.default_call_conv);
signature.params.push(AbiParam::new(I32));
let sig = func.import_signature(signature);
self.raise_trap_sig = Some(sig);
sig
});
(sig, VMBuiltinFunctionIndex::get_raise_trap_index())
}
fn get_alloc_exception_func(
&mut self,
func: &mut Function,
) -> (ir::SigRef, VMBuiltinFunctionIndex) {
let sig = self.alloc_exception_sig.unwrap_or_else(|| {
let mut signature = Signature::new(self.target_config.default_call_conv);
signature.params.push(AbiParam::special(
self.pointer_type(),
ArgumentPurpose::VMContext,
));
signature.params.push(AbiParam::new(TAG_TYPE));
signature.returns.push(AbiParam::new(EXN_REF_TYPE));
let sig = func.import_signature(signature);
self.alloc_exception_sig = Some(sig);
sig
});
(
sig,
VMBuiltinFunctionIndex::get_imported_alloc_exception_index(),
)
}
fn get_read_exnref_func(
&mut self,
func: &mut Function,
) -> (ir::SigRef, VMBuiltinFunctionIndex) {
let sig = self.read_exnref_sig.unwrap_or_else(|| {
let mut signature = Signature::new(self.target_config.default_call_conv);
signature.params.push(AbiParam::special(
self.pointer_type(),
ArgumentPurpose::VMContext,
));
signature.params.push(AbiParam::new(EXN_REF_TYPE));
signature.returns.push(AbiParam::new(self.pointer_type()));
let sig = func.import_signature(signature);
self.read_exnref_sig = Some(sig);
sig
});
(
sig,
VMBuiltinFunctionIndex::get_imported_read_exnref_index(),
)
}
fn get_read_exception_func(
&mut self,
func: &mut Function,
) -> (ir::SigRef, VMBuiltinFunctionIndex) {
let sig = self.read_exception_sig.unwrap_or_else(|| {
let mut signature = Signature::new(self.target_config.default_call_conv);
signature.params.push(AbiParam::new(self.pointer_type()));
signature.returns.push(AbiParam::new(EXN_REF_TYPE));
let sig = func.import_signature(signature);
self.read_exception_sig = Some(sig);
sig
});
(
sig,
VMBuiltinFunctionIndex::get_imported_exception_into_exnref_index(),
)
}
fn exception_type_layout(&mut self, tag_index: TagIndex) -> WasmResult<&ExceptionTypeLayout> {
let key = tag_index.as_u32();
if !self.exception_type_layouts.contains_key(&key) {
let layout = self.compute_exception_type_layout(tag_index)?;
self.exception_type_layouts.insert(key, layout);
}
Ok(self.exception_type_layouts.get(&key).unwrap())
}
fn compute_exception_type_layout(
&self,
tag_index: TagIndex,
) -> WasmResult<ExceptionTypeLayout> {
let sig_index = self.module.tags[tag_index];
let func_type = &self.module.signatures[sig_index];
let mut offset = 0u32;
let mut max_align = 1u32;
let mut fields = SmallVec::<[ExceptionFieldLayout; 4]>::new();
for wasm_ty in func_type.params() {
let ir_ty = self.map_wasmer_type_to_ir(*wasm_ty)?;
let field_size = ir_ty.bytes();
let align = field_size.max(1);
max_align = max_align.max(align);
offset = offset.next_multiple_of(align);
fields.push(ExceptionFieldLayout { offset, ty: ir_ty });
offset = offset
.checked_add(field_size)
.ok_or_else(|| WasmError::Unsupported("exception payload too large".to_string()))?;
}
Ok(ExceptionTypeLayout { fields })
}
fn map_wasmer_type_to_ir(&self, ty: WasmerType) -> WasmResult<ir::Type> {
Ok(match ty {
WasmerType::I32 => ir::types::I32,
WasmerType::I64 => ir::types::I64,
WasmerType::F32 => ir::types::F32,
WasmerType::F64 => ir::types::F64,
WasmerType::V128 => ir::types::I8X16,
WasmerType::FuncRef | WasmerType::ExternRef | WasmerType::ExceptionRef => {
self.reference_type()
}
})
}
fn call_with_handlers(
&mut self,
builder: &mut FunctionBuilder,
callee: ir::FuncRef,
args: &[ir::Value],
context: Option<ir::Value>,
landing_pad: Option<LandingPad>,
unreachable_on_return: bool,
) -> SmallVec<[ir::Value; 4]> {
let sig_ref = builder.func.dfg.ext_funcs[callee].signature;
let return_types: SmallVec<[ir::Type; 4]> = builder.func.dfg.signatures[sig_ref]
.returns
.iter()
.map(|ret| ret.value_type)
.collect();
if landing_pad.is_none() {
let inst = builder.ins().call(callee, args);
let results: SmallVec<[ir::Value; 4]> =
builder.inst_results(inst).iter().copied().collect();
if unreachable_on_return {
builder.ins().trap(crate::TRAP_UNREACHABLE);
}
return results;
}
let continuation = builder.create_block();
let mut normal_args = SmallVec::<[BlockArg; 4]>::with_capacity(return_types.len());
let mut result_values = SmallVec::<[ir::Value; 4]>::with_capacity(return_types.len());
for (i, ty) in return_types.iter().enumerate() {
let val = builder.append_block_param(continuation, *ty);
result_values.push(val);
normal_args.push(BlockArg::TryCallRet(u32::try_from(i).unwrap()));
}
let continuation_call = builder
.func
.dfg
.block_call(continuation, normal_args.iter());
let mut table_items = Vec::new();
if let Some(ctx) = context {
table_items.push(ExceptionTableItem::Context(ctx));
}
if let Some(landing_pad) = landing_pad {
for tag in landing_pad.clauses {
let block_call = builder.func.dfg.block_call(
landing_pad.block,
&[BlockArg::TryCallExn(0), BlockArg::TryCallExn(1)],
);
table_items.push(match tag.wasm_tag {
Some(tag) => ExceptionTableItem::Tag(ExceptionTag::from_u32(tag), block_call),
None => ExceptionTableItem::Default(block_call),
});
}
}
let etd = ExceptionTableData::new(sig_ref, continuation_call, table_items);
let et = builder.func.dfg.exception_tables.push(etd);
builder.ins().try_call(callee, args, et);
builder.switch_to_block(continuation);
builder.seal_block(continuation);
if unreachable_on_return {
builder.ins().trap(crate::TRAP_UNREACHABLE);
}
result_values
}
#[allow(clippy::too_many_arguments)]
fn call_indirect_with_handlers(
&mut self,
builder: &mut FunctionBuilder,
sig: ir::SigRef,
func_addr: ir::Value,
args: &[ir::Value],
context: Option<ir::Value>,
landing_pad: Option<LandingPad>,
unreachable_on_return: bool,
) -> SmallVec<[ir::Value; 4]> {
let return_types: SmallVec<[ir::Type; 4]> = builder.func.dfg.signatures[sig]
.returns
.iter()
.map(|ret| ret.value_type)
.collect();
if landing_pad.is_none() {
let inst = builder.ins().call_indirect(sig, func_addr, args);
let results: SmallVec<[ir::Value; 4]> =
builder.inst_results(inst).iter().copied().collect();
if unreachable_on_return {
builder.ins().trap(crate::TRAP_UNREACHABLE);
}
return results;
}
let continuation = builder.create_block();
let current_block = builder.current_block().expect("current block");
builder.insert_block_after(continuation, current_block);
let mut normal_args = SmallVec::<[BlockArg; 4]>::with_capacity(return_types.len());
let mut result_values = SmallVec::<[ir::Value; 4]>::with_capacity(return_types.len());
for (i, ty) in return_types.iter().enumerate() {
let val = builder.append_block_param(continuation, *ty);
result_values.push(val);
normal_args.push(BlockArg::TryCallRet(u32::try_from(i).unwrap()));
}
let continuation_call = builder
.func
.dfg
.block_call(continuation, normal_args.iter());
let mut table_items = Vec::new();
if let Some(ctx) = context {
table_items.push(ExceptionTableItem::Context(ctx));
}
if let Some(landing_pad) = landing_pad {
for tag in landing_pad.clauses {
let block_call = builder.func.dfg.block_call(
landing_pad.block,
&[BlockArg::TryCallExn(0), BlockArg::TryCallExn(1)],
);
table_items.push(match tag.wasm_tag {
Some(tag) => ExceptionTableItem::Tag(ExceptionTag::from_u32(tag), block_call),
None => ExceptionTableItem::Default(block_call),
});
}
}
let etd = ExceptionTableData::new(sig, continuation_call, table_items);
let et = builder.func.dfg.exception_tables.push(etd);
builder.ins().try_call_indirect(func_addr, args, et);
builder.switch_to_block(continuation);
builder.seal_block(continuation);
if unreachable_on_return {
builder.ins().trap(crate::TRAP_UNREACHABLE);
}
result_values
}
fn translate_load_builtin_function_address(
&mut self,
pos: &mut FuncCursor<'_>,
callee_func_idx: VMBuiltinFunctionIndex,
) -> WasmResult<(ir::Value, ir::Value)> {
let pointer_type = self.pointer_type();
let vmctx = self.vmctx(pos.func);
let base = materialize_global_value(pos, pointer_type, vmctx);
let mut mem_flags = ir::MemFlagsData::trusted();
mem_flags.set_readonly();
let body_offset = vmctx_offset(self.offsets.vmctx_builtin_function(callee_func_idx))?;
let func_addr = pos.ins().load(pointer_type, mem_flags, base, body_offset);
Ok((base, func_addr))
}
fn get_or_init_funcref_table_elem(
&mut self,
builder: &mut FunctionBuilder,
table_index: TableIndex,
index: ir::Value,
) -> WasmResult<(ir::Value, bool)> {
let pointer_type = self.pointer_type();
self.ensure_table_exists(builder.func, table_index)?;
let table_data = self.tables[table_index].as_ref().unwrap();
let (table_entry_addr, flags) =
table_data.prepare_table_addr(builder, index, pointer_type, false);
Ok(if table_data.inline_anyfunc {
(table_entry_addr, true)
} else {
(
builder.ins().load(pointer_type, flags, table_entry_addr, 0),
false,
)
})
}
}
impl FuncEnvironment<'_> {
pub(crate) fn is_wasm_parameter(&self, signature: &ir::Signature, index: usize) -> bool {
signature.params[index].purpose == ArgumentPurpose::Normal
}
pub(crate) fn translate_unreachable(
&mut self,
builder: &mut FunctionBuilder,
) -> WasmResult<()> {
let (func_sig, func_idx) = self.get_raise_trap_func(builder.func);
let mut pos = builder.cursor();
let (_, func_addr) = self.translate_load_builtin_function_address(&mut pos, func_idx)?;
let trap_code = pos
.ins()
.iconst(I32, wasmer_types::TrapCode::UnreachableCodeReached as i64);
builder
.ins()
.call_indirect(func_sig, func_addr, &[trap_code]);
builder.ins().trap(crate::TRAP_UNREACHABLE);
Ok(())
}
pub(crate) fn translate_table_grow(
&mut self,
mut pos: cranelift_codegen::cursor::FuncCursor<'_>,
table_index: TableIndex,
delta: ir::Value,
init_value: ir::Value,
) -> WasmResult<ir::Value> {
self.ensure_table_exists(pos.func, table_index)?;
let (func_sig, index_arg, func_idx) = self.get_table_grow_func(pos.func, table_index);
let table_index = pos.ins().iconst(I32, index_arg as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
let call_inst = pos.ins().call_indirect(
func_sig,
func_addr,
&[vmctx, init_value, delta, table_index],
);
Ok(*pos.func.dfg.inst_results(call_inst).first().unwrap())
}
pub(crate) fn translate_table_get(
&mut self,
builder: &mut FunctionBuilder,
table_index: TableIndex,
index: ir::Value,
) -> WasmResult<ir::Value> {
self.ensure_table_exists(builder.func, table_index)?;
let mut pos = builder.cursor();
let (func_sig, table_index_arg, func_idx) = self.get_table_get_func(pos.func, table_index);
let table_index = pos.ins().iconst(I32, table_index_arg as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
let call_inst = pos
.ins()
.call_indirect(func_sig, func_addr, &[vmctx, table_index, index]);
Ok(*pos.func.dfg.inst_results(call_inst).first().unwrap())
}
pub(crate) fn translate_table_set(
&mut self,
builder: &mut FunctionBuilder,
table_index: TableIndex,
value: ir::Value,
index: ir::Value,
) -> WasmResult<()> {
self.ensure_table_exists(builder.func, table_index)?;
let mut pos = builder.cursor();
let (func_sig, table_index_arg, func_idx) = self.get_table_set_func(pos.func, table_index);
let n_table_index = pos.ins().iconst(I32, table_index_arg as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
pos.ins()
.call_indirect(func_sig, func_addr, &[vmctx, n_table_index, index, value]);
Ok(())
}
pub(crate) fn translate_table_fill(
&mut self,
mut pos: cranelift_codegen::cursor::FuncCursor<'_>,
table_index: TableIndex,
dst: ir::Value,
val: ir::Value,
len: ir::Value,
) -> WasmResult<()> {
self.ensure_table_exists(pos.func, table_index)?;
let (func_sig, table_index_arg, func_idx) = self.get_table_fill_func(pos.func, table_index);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
let table_index_arg = pos.ins().iconst(I32, table_index_arg as i64);
pos.ins().call_indirect(
func_sig,
func_addr,
&[vmctx, table_index_arg, dst, val, len],
);
Ok(())
}
pub(crate) fn translate_ref_null(
&mut self,
mut pos: cranelift_codegen::cursor::FuncCursor,
ty: HeapType,
) -> WasmResult<ir::Value> {
Ok(match ty {
HeapType::Abstract { ty, .. } => match ty {
wasmer_compiler::wasmparser::AbstractHeapType::Func
| wasmer_compiler::wasmparser::AbstractHeapType::Extern
| wasmer_compiler::wasmparser::AbstractHeapType::Exn => pos.ins().iconst(
if matches!(ty, wasmer_compiler::wasmparser::AbstractHeapType::Exn) {
I32
} else {
self.reference_type()
},
0,
),
_ => {
return Err(WasmError::Unsupported(format!(
"`ref.null T` that is not a `funcref`, an `externref` or an `exn`: {ty:?}"
)));
}
},
HeapType::Concrete(_) => {
return Err(WasmError::Unsupported(
"`ref.null T` that is not a `funcref` or an `externref`".into(),
));
}
HeapType::Exact(_) => {
return Err(WasmError::Unsupported(
"custom-descriptors not supported yet".into(),
));
}
})
}
pub(crate) fn translate_ref_is_null(
&mut self,
mut pos: cranelift_codegen::cursor::FuncCursor,
value: ir::Value,
) -> WasmResult<ir::Value> {
let bool_is_null =
pos.ins()
.icmp_imm_s(cranelift_codegen::ir::condcodes::IntCC::Equal, value, 0);
Ok(pos.ins().uextend(ir::types::I32, bool_is_null))
}
pub(crate) fn translate_ref_func(
&mut self,
mut pos: cranelift_codegen::cursor::FuncCursor<'_>,
func_index: FunctionIndex,
) -> WasmResult<ir::Value> {
let (func_sig, func_index_arg, func_idx) = self.get_func_ref_func(pos.func, func_index);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
let func_index_arg = pos.ins().iconst(I32, func_index_arg as i64);
let call_inst = pos
.ins()
.call_indirect(func_sig, func_addr, &[vmctx, func_index_arg]);
Ok(*pos.func.dfg.inst_results(call_inst).first().unwrap())
}
pub(crate) fn translate_custom_global_get(
&mut self,
mut _pos: cranelift_codegen::cursor::FuncCursor<'_>,
_index: GlobalIndex,
) -> WasmResult<ir::Value> {
unreachable!("we don't make any custom globals")
}
pub(crate) fn translate_custom_global_set(
&mut self,
mut _pos: cranelift_codegen::cursor::FuncCursor<'_>,
_index: GlobalIndex,
_value: ir::Value,
) -> WasmResult<()> {
unreachable!("we don't make any custom globals")
}
pub(crate) fn make_heap(
&mut self,
func: &mut ir::Function,
index: MemoryIndex,
) -> WasmResult<Heap> {
let pointer_type = self.pointer_type();
let (ptr, base_offset, current_length_offset) = {
let vmctx = self.vmctx(func);
if let Some(def_index) = self.module.local_memory_index(index) {
let base_offset =
vmctx_offset(self.offsets.vmctx_vmmemory_definition_base(def_index))?;
let current_length_offset = vmctx_offset(
self.offsets
.vmctx_vmmemory_definition_current_length(def_index),
)?;
(vmctx, base_offset, current_length_offset)
} else {
let from_offset = self.offsets.vmctx_vmmemory_import_definition(index);
let flags = insert_mem_flags(func, ir::MemFlagsData::trusted().with_readonly());
let memory = func.create_global_value(ir::GlobalValueData::Load {
base: vmctx,
offset: Offset32::new(vmctx_offset(from_offset)?),
global_type: pointer_type,
flags,
});
let base_offset = i32::from(self.offsets.vmmemory_definition_base());
let current_length_offset =
i32::from(self.offsets.vmmemory_definition_current_length());
(memory, base_offset, current_length_offset)
}
};
let (offset_guard_size, heap_style, readonly_base) = match self.memory_styles[index] {
MemoryStyle::Dynamic { offset_guard_size } => {
let flags = insert_mem_flags(func, ir::MemFlagsData::trusted());
let heap_bound = func.create_global_value(ir::GlobalValueData::Load {
base: ptr,
offset: Offset32::new(current_length_offset),
global_type: pointer_type,
flags,
});
(
Uimm64::new(offset_guard_size),
HeapStyle::Dynamic {
bound_gv: heap_bound,
},
false,
)
}
MemoryStyle::Static => (
Uimm64::new(MemoryStyle::static_offset_guard_size()),
HeapStyle::Static,
true,
),
};
let flags = if readonly_base {
insert_mem_flags(func, ir::MemFlagsData::trusted().with_readonly())
} else {
insert_mem_flags(func, ir::MemFlagsData::trusted())
};
let heap_base = func.create_global_value(ir::GlobalValueData::Load {
base: ptr,
offset: Offset32::new(base_offset),
global_type: pointer_type,
flags,
});
Ok(self.heaps.push(HeapData {
base: heap_base,
min_size: 0,
max_size: None,
offset_guard_size: offset_guard_size.into(),
style: heap_style,
index_type: I32,
page_size_log2: self.target_config.page_size_align_log2,
}))
}
pub(crate) fn make_global(
&mut self,
func: &mut ir::Function,
index: GlobalIndex,
) -> WasmResult<GlobalVariable> {
let pointer_type = self.pointer_type();
let (ptr, offset) = {
let vmctx = self.vmctx(func);
if let Some(def_index) = self.module.local_global_index(index) {
let from_offset = self.offsets.vmctx_vmglobal_definition(def_index);
let global = func.create_global_value(ir::GlobalValueData::VMContext);
(global, vmctx_offset(from_offset)?)
} else {
let from_offset = self.offsets.vmctx_vmglobal_import_definition(index);
let flags = insert_mem_flags(func, MemFlagsData::trusted());
let global = func.create_global_value(ir::GlobalValueData::Load {
base: vmctx,
offset: Offset32::new(vmctx_offset(from_offset)?),
global_type: pointer_type,
flags,
});
(global, 0)
}
};
Ok(GlobalVariable::Memory {
gv: ptr,
offset: offset.into(),
ty: match self.module.globals[index].ty {
WasmerType::I32 => ir::types::I32,
WasmerType::I64 => ir::types::I64,
WasmerType::F32 => ir::types::F32,
WasmerType::F64 => ir::types::F64,
WasmerType::V128 => ir::types::I8X16,
WasmerType::FuncRef | WasmerType::ExternRef | WasmerType::ExceptionRef => {
self.reference_type()
}
},
})
}
pub(crate) fn make_indirect_sig(
&mut self,
func: &mut ir::Function,
index: SignatureIndex,
) -> WasmResult<ir::SigRef> {
Ok(func.import_signature(self.signatures[index].clone()))
}
pub(crate) fn make_direct_func(
&mut self,
func: &mut ir::Function,
index: FunctionIndex,
) -> WasmResult<ir::FuncRef> {
let sigidx = self.module.functions[index];
let signature = func.import_signature(self.signatures[sigidx].clone());
let name = get_function_name(func, index);
Ok(func.import_function(ir::ExtFuncData {
name,
signature,
colocated: true,
patchable: false,
}))
}
fn prepare_wasm_call(
&self,
builder: &mut FunctionBuilder,
result_types: &[WasmerType],
) -> (ReturnAbi, Option<(ir::Value, abi::ReturnAreaLayout)>) {
let return_abi = abi::classify_returns(self.architecture, result_types);
let sret = match &return_abi {
ReturnAbi::Sret(types) => Some(abi::allocate_return_area(
builder,
types,
self.target_config,
)),
_ => None,
};
(return_abi, sret)
}
fn finish_wasm_call(
&self,
builder: &mut FunctionBuilder,
return_abi: &ReturnAbi,
carriers: &[ir::Value],
sret: Option<(ir::Value, abi::ReturnAreaLayout)>,
) -> SmallVec<[ir::Value; 4]> {
match return_abi {
ReturnAbi::Sret(types) => {
let (ptr, layout) = sret.expect("sret call has a return area");
abi::load_sret(builder, ptr, &layout, types, self.target_config)
}
_ => abi::unpack_register_returns(builder, return_abi, carriers, self.target_config),
}
}
#[allow(clippy::too_many_arguments)]
pub(crate) fn translate_call_indirect(
&mut self,
builder: &mut FunctionBuilder,
table_index: TableIndex,
sig_index: SignatureIndex,
sig_ref: ir::SigRef,
callee: ir::Value,
call_args: &[ir::Value],
landing_pad: Option<LandingPad>,
) -> WasmResult<SmallVec<[ir::Value; 4]>> {
let pointer_type = self.pointer_type();
let (anyfunc_ptr, inline_anyfunc) =
self.get_or_init_funcref_table_elem(builder, table_index, callee)?;
let mem_flags = ir::MemFlagsData::trusted();
if !inline_anyfunc {
builder
.ins()
.trapz(anyfunc_ptr, crate::TRAP_INDIRECT_CALL_TO_NULL);
}
let func_addr = builder.ins().load(
pointer_type,
mem_flags,
anyfunc_ptr,
i32::from(self.offsets.vmcaller_checked_anyfunc_func_ptr()),
);
if inline_anyfunc {
builder
.ins()
.trapz(func_addr, crate::TRAP_INDIRECT_CALL_TO_NULL);
}
match self.table_styles[table_index] {
TableStyle::CallerChecksSignature => {
let sig_hash_type = ir::types::I32;
let expected_sig_hash = builder.ins().iconst(
sig_hash_type,
i64::from(self.signature_hashes[sig_index].as_u32()),
);
let mem_flags = ir::MemFlagsData::trusted();
let callee_sig_hash = builder.ins().load(
sig_hash_type,
mem_flags,
anyfunc_ptr,
i32::from(self.offsets.vmcaller_checked_anyfunc_signature_hash()),
);
let cmp = builder
.ins()
.icmp(IntCC::Equal, callee_sig_hash, expected_sig_hash);
builder.ins().trapz(cmp, crate::TRAP_BAD_SIGNATURE);
}
}
let (return_abi, sret) =
self.prepare_wasm_call(builder, self.module.signatures[sig_index].results());
let mut real_call_args = Vec::with_capacity(call_args.len() + 2);
if let Some((ptr, _)) = &sret {
real_call_args.push(*ptr);
}
let vmctx = builder.ins().load(
pointer_type,
mem_flags,
anyfunc_ptr,
i32::from(self.offsets.vmcaller_checked_anyfunc_vmctx()),
);
real_call_args.push(vmctx);
real_call_args.extend_from_slice(call_args);
let results = self.call_indirect_with_handlers(
builder,
sig_ref,
func_addr,
&real_call_args,
Some(vmctx),
landing_pad,
false,
);
Ok(self.finish_wasm_call(builder, &return_abi, &results, sret))
}
pub(crate) fn translate_call(
&mut self,
builder: &mut FunctionBuilder,
callee_index: FunctionIndex,
callee: ir::FuncRef,
call_args: &[ir::Value],
landing_pad: Option<LandingPad>,
) -> WasmResult<SmallVec<[ir::Value; 4]>> {
let sig_index = self.module.functions[callee_index];
let (return_abi, sret) =
self.prepare_wasm_call(builder, self.module.signatures[sig_index].results());
let mut real_call_args = Vec::with_capacity(call_args.len() + 2);
if let Some((ptr, _)) = &sret {
real_call_args.push(*ptr);
}
if !self.module.is_imported_function(callee_index) {
let caller_vmctx = builder
.func
.special_param(ArgumentPurpose::VMContext)
.unwrap();
real_call_args.push(caller_vmctx);
real_call_args.extend_from_slice(call_args);
let results = self.call_with_handlers(
builder,
callee,
&real_call_args,
Some(caller_vmctx),
landing_pad,
false,
);
return Ok(self.finish_wasm_call(builder, &return_abi, &results, sret));
}
let pointer_type = self.pointer_type();
let sig_ref = builder.func.dfg.ext_funcs[callee].signature;
let vmctx = self.vmctx(builder.func);
let base = materialize_global_value(&mut builder.cursor(), pointer_type, vmctx);
let mem_flags = ir::MemFlagsData::trusted();
let body_offset = vmctx_offset(self.offsets.vmctx_vmfunction_import_body(callee_index))?;
let func_addr = builder
.ins()
.load(pointer_type, mem_flags, base, body_offset);
let vmctx_arg_offset =
vmctx_offset(self.offsets.vmctx_vmfunction_import_vmctx(callee_index))?;
let vmctx = builder
.ins()
.load(pointer_type, mem_flags, base, vmctx_arg_offset);
real_call_args.push(vmctx);
real_call_args.extend_from_slice(call_args);
let results = self.call_indirect_with_handlers(
builder,
sig_ref,
func_addr,
&real_call_args,
Some(vmctx),
landing_pad,
false,
);
Ok(self.finish_wasm_call(builder, &return_abi, &results, sret))
}
pub(crate) fn tag_param_arity(&self, tag_index: TagIndex) -> usize {
let sig_index = self.module.tags[tag_index];
let signature = &self.module.signatures[sig_index];
signature.params().len()
}
pub(crate) fn translate_exn_pointer_to_ref(
&mut self,
builder: &mut FunctionBuilder,
exn_ptr: ir::Value,
) -> WasmResult<ir::Value> {
let (read_sig, read_idx) = self.get_read_exception_func(builder.func);
let mut pos = builder.cursor();
let (_, read_addr) = self.translate_load_builtin_function_address(&mut pos, read_idx)?;
let read_call = builder.ins().call_indirect(read_sig, read_addr, &[exn_ptr]);
Ok(builder.inst_results(read_call)[0])
}
pub(crate) fn translate_exn_unbox(
&mut self,
builder: &mut FunctionBuilder,
tag_index: TagIndex,
exnref: ir::Value,
) -> WasmResult<SmallVec<[ir::Value; 4]>> {
let layout = self.exception_type_layout(tag_index)?.clone();
let (read_exnref_sig, read_exnref_idx) = self.get_read_exnref_func(builder.func);
let mut pos = builder.cursor();
let (vmctx, read_exnref_addr) =
self.translate_load_builtin_function_address(&mut pos, read_exnref_idx)?;
let read_exnref_call =
builder
.ins()
.call_indirect(read_exnref_sig, read_exnref_addr, &[vmctx, exnref]);
let payload_ptr = builder.inst_results(read_exnref_call)[0];
let mut values = SmallVec::<[ir::Value; 4]>::with_capacity(layout.fields.len());
let data_flags = ir::MemFlagsData::trusted();
for field in &layout.fields {
let value = builder.ins().load(
field.ty,
data_flags,
payload_ptr,
Offset32::new(field.offset as i32),
);
values.push(value);
}
Ok(values)
}
pub(crate) fn translate_exn_throw(
&mut self,
builder: &mut FunctionBuilder,
tag_index: TagIndex,
args: &[ir::Value],
landing_pad: Option<LandingPad>,
) -> WasmResult<()> {
let layout = self.exception_type_layout(tag_index)?.clone();
if layout.fields.len() != args.len() {
return Err(WasmError::Generic(format!(
"exception payload arity mismatch: expected {}, got {}",
layout.fields.len(),
args.len()
)));
}
let (alloc_sig, alloc_idx) = self.get_alloc_exception_func(builder.func);
let mut pos = builder.cursor();
let (vmctx, alloc_addr) =
self.translate_load_builtin_function_address(&mut pos, alloc_idx)?;
let tag_value = builder
.ins()
.iconst(TAG_TYPE, i64::from(tag_index.as_u32()));
let alloc_call = builder
.ins()
.call_indirect(alloc_sig, alloc_addr, &[vmctx, tag_value]);
let exnref = builder.inst_results(alloc_call)[0];
let (read_exnref_sig, read_exnref_idx) = self.get_read_exnref_func(builder.func);
let mut pos = builder.cursor();
let (vmctx, read_exnref_addr) =
self.translate_load_builtin_function_address(&mut pos, read_exnref_idx)?;
let read_exnref_call =
builder
.ins()
.call_indirect(read_exnref_sig, read_exnref_addr, &[vmctx, exnref]);
let payload_ptr = builder.inst_results(read_exnref_call)[0];
let store_flags = ir::MemFlagsData::trusted();
for (field, value) in layout.fields.iter().zip(args.iter()) {
debug_assert_eq!(
builder.func.dfg.value_type(*value),
field.ty,
"exception payload type mismatch"
);
builder.ins().store(
store_flags,
*value,
payload_ptr,
Offset32::new(field.offset as i32),
);
}
let (throw_sig, throw_idx) = self.get_throw_func(builder.func);
let mut pos = builder.cursor();
let (vmctx_value, throw_addr) =
self.translate_load_builtin_function_address(&mut pos, throw_idx)?;
let call_args = [vmctx_value, exnref];
let _ = self.call_indirect_with_handlers(
builder,
throw_sig,
throw_addr,
&call_args,
Some(vmctx_value),
landing_pad,
true,
);
Ok(())
}
pub(crate) fn translate_exn_throw_ref(
&mut self,
builder: &mut FunctionBuilder,
exnref: ir::Value,
landing_pad: Option<LandingPad>,
) -> WasmResult<()> {
let (throw_sig, throw_idx) = self.get_throw_func(builder.func);
let mut pos = builder.cursor();
let (vmctx_value, throw_addr) =
self.translate_load_builtin_function_address(&mut pos, throw_idx)?;
let call_args = [vmctx_value, exnref];
let _ = self.call_indirect_with_handlers(
builder,
throw_sig,
throw_addr,
&call_args,
Some(vmctx_value),
landing_pad,
true,
);
Ok(())
}
pub(crate) fn translate_exn_personality_selector(
&mut self,
builder: &mut FunctionBuilder,
exn_ptr: ir::Value,
) -> WasmResult<ir::Value> {
let (sig, idx) = self.get_personality2_func(builder.func);
let pointer_type = self.pointer_type();
let exn_ty = builder.func.dfg.value_type(exn_ptr);
let exn_arg = if exn_ty == pointer_type {
exn_ptr
} else {
let mut flags = MemFlagsData::new();
flags.set_endianness(Endianness::Little);
builder.ins().bitcast(pointer_type, flags, exn_ptr)
};
let mut pos = builder.cursor();
let (vmctx_value, func_addr) =
self.translate_load_builtin_function_address(&mut pos, idx)?;
let call = builder
.ins()
.call_indirect(sig, func_addr, &[vmctx_value, exn_arg]);
Ok(builder.inst_results(call)[0])
}
pub(crate) fn translate_exn_reraise_unmatched(
&mut self,
builder: &mut FunctionBuilder,
exnref: ir::Value,
) -> WasmResult<()> {
let (throw_sig, throw_idx) = self.get_throw_func(builder.func);
let mut pos = builder.cursor();
let (vmctx_value, throw_addr) =
self.translate_load_builtin_function_address(&mut pos, throw_idx)?;
builder
.ins()
.call_indirect(throw_sig, throw_addr, &[vmctx_value, exnref]);
builder.ins().trap(crate::TRAP_UNREACHABLE);
Ok(())
}
pub(crate) fn translate_memory_grow(
&mut self,
mut pos: FuncCursor<'_>,
index: MemoryIndex,
_heap: Heap,
val: ir::Value,
) -> WasmResult<ir::Value> {
let (func_sig, index_arg, func_idx) = self.get_memory_grow_func(pos.func, index);
let memory_index = pos.ins().iconst(I32, index_arg as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
let call_inst = pos
.ins()
.call_indirect(func_sig, func_addr, &[vmctx, val, memory_index]);
Ok(*pos.func.dfg.inst_results(call_inst).first().unwrap())
}
pub(crate) fn translate_memory_size(
&mut self,
mut pos: FuncCursor<'_>,
index: MemoryIndex,
_heap: Heap,
) -> WasmResult<ir::Value> {
let (func_sig, index_arg, func_idx) = self.get_memory_size_func(pos.func, index);
let memory_index = pos.ins().iconst(I32, index_arg as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
let call_inst = pos
.ins()
.call_indirect(func_sig, func_addr, &[vmctx, memory_index]);
Ok(*pos.func.dfg.inst_results(call_inst).first().unwrap())
}
#[allow(clippy::too_many_arguments)]
pub(crate) fn translate_memory_copy(
&mut self,
mut pos: FuncCursor,
src_index: MemoryIndex,
_src_heap: Heap,
dst_index: MemoryIndex,
_dst_heap: Heap,
dst: ir::Value,
src: ir::Value,
len: ir::Value,
) -> WasmResult<()> {
let (func_sig, func_idx) = self.get_memory_copy_func(pos.func);
let dst_index_arg = pos.ins().iconst(I32, dst_index.index() as i64);
let src_index_arg = pos.ins().iconst(I32, src_index.index() as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
pos.ins().call_indirect(
func_sig,
func_addr,
&[vmctx, dst_index_arg, src_index_arg, dst, src, len],
);
Ok(())
}
pub(crate) fn translate_memory_fill(
&mut self,
mut pos: FuncCursor,
memory_index: MemoryIndex,
_heap: Heap,
dst: ir::Value,
val: ir::Value,
len: ir::Value,
) -> WasmResult<()> {
let (func_sig, memory_index, func_idx) = self.get_memory_fill_func(pos.func, memory_index);
let memory_index_arg = pos.ins().iconst(I32, memory_index as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
pos.ins().call_indirect(
func_sig,
func_addr,
&[vmctx, memory_index_arg, dst, val, len],
);
Ok(())
}
#[allow(clippy::too_many_arguments)]
pub(crate) fn translate_memory_init(
&mut self,
mut pos: FuncCursor,
memory_index: MemoryIndex,
_heap: Heap,
seg_index: u32,
dst: ir::Value,
src: ir::Value,
len: ir::Value,
) -> WasmResult<()> {
let (func_sig, func_idx) = self.get_memory_init_func(pos.func);
let memory_index_arg = pos.ins().iconst(I32, memory_index.index() as i64);
let seg_index_arg = pos.ins().iconst(I32, seg_index as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
pos.ins().call_indirect(
func_sig,
func_addr,
&[vmctx, memory_index_arg, seg_index_arg, dst, src, len],
);
Ok(())
}
pub(crate) fn translate_data_drop(
&mut self,
mut pos: FuncCursor,
seg_index: u32,
) -> WasmResult<()> {
let (func_sig, func_idx) = self.get_data_drop_func(pos.func);
let seg_index_arg = pos.ins().iconst(I32, seg_index as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
pos.ins()
.call_indirect(func_sig, func_addr, &[vmctx, seg_index_arg]);
Ok(())
}
pub(crate) fn translate_table_size(
&mut self,
mut pos: FuncCursor,
table_index: TableIndex,
) -> WasmResult<ir::Value> {
self.ensure_table_exists(pos.func, table_index)?;
let (func_sig, index_arg, func_idx) = self.get_table_size_func(pos.func, table_index);
let table_index = pos.ins().iconst(I32, index_arg as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
let call_inst = pos
.ins()
.call_indirect(func_sig, func_addr, &[vmctx, table_index]);
Ok(*pos.func.dfg.inst_results(call_inst).first().unwrap())
}
pub(crate) fn translate_table_copy(
&mut self,
mut pos: FuncCursor,
dst_table_index: TableIndex,
src_table_index: TableIndex,
dst: ir::Value,
src: ir::Value,
len: ir::Value,
) -> WasmResult<()> {
self.ensure_table_exists(pos.func, src_table_index)?;
self.ensure_table_exists(pos.func, dst_table_index)?;
let (func_sig, dst_table_index_arg, src_table_index_arg, func_idx) =
self.get_table_copy_func(pos.func, dst_table_index, src_table_index);
let dst_table_index_arg = pos.ins().iconst(I32, dst_table_index_arg as i64);
let src_table_index_arg = pos.ins().iconst(I32, src_table_index_arg as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
pos.ins().call_indirect(
func_sig,
func_addr,
&[
vmctx,
dst_table_index_arg,
src_table_index_arg,
dst,
src,
len,
],
);
Ok(())
}
pub(crate) fn translate_table_init(
&mut self,
mut pos: FuncCursor,
seg_index: u32,
table_index: TableIndex,
dst: ir::Value,
src: ir::Value,
len: ir::Value,
) -> WasmResult<()> {
self.ensure_table_exists(pos.func, table_index)?;
let (func_sig, table_index_arg, func_idx) = self.get_table_init_func(pos.func, table_index);
let table_index_arg = pos.ins().iconst(I32, table_index_arg as i64);
let seg_index_arg = pos.ins().iconst(I32, seg_index as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
pos.ins().call_indirect(
func_sig,
func_addr,
&[vmctx, table_index_arg, seg_index_arg, dst, src, len],
);
Ok(())
}
pub(crate) fn translate_elem_drop(
&mut self,
mut pos: FuncCursor,
elem_index: u32,
) -> WasmResult<()> {
let (func_sig, func_idx) = self.get_elem_drop_func(pos.func);
let elem_index_arg = pos.ins().iconst(I32, elem_index as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
pos.ins()
.call_indirect(func_sig, func_addr, &[vmctx, elem_index_arg]);
Ok(())
}
pub(crate) fn translate_atomic_wait(
&mut self,
mut pos: FuncCursor,
index: MemoryIndex,
_heap: Heap,
addr: ir::Value,
expected: ir::Value,
timeout: ir::Value,
) -> WasmResult<ir::Value> {
let (func_sig, index_arg, func_idx) = if pos.func.dfg.value_type(expected) == I64 {
self.get_memory_atomic_wait64_func(pos.func, index)
} else {
self.get_memory_atomic_wait32_func(pos.func, index)
};
let memory_index = pos.ins().iconst(I32, index_arg as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
let call_inst = pos.ins().call_indirect(
func_sig,
func_addr,
&[vmctx, memory_index, addr, expected, timeout],
);
Ok(*pos.func.dfg.inst_results(call_inst).first().unwrap())
}
pub(crate) fn translate_atomic_notify(
&mut self,
mut pos: FuncCursor,
index: MemoryIndex,
_heap: Heap,
addr: ir::Value,
count: ir::Value,
) -> WasmResult<ir::Value> {
let (func_sig, index_arg, func_idx) = self.get_memory_atomic_notify_func(pos.func, index);
let memory_index = pos.ins().iconst(I32, index_arg as i64);
let (vmctx, func_addr) =
self.translate_load_builtin_function_address(&mut pos, func_idx)?;
let call_inst =
pos.ins()
.call_indirect(func_sig, func_addr, &[vmctx, memory_index, addr, count]);
Ok(*pos.func.dfg.inst_results(call_inst).first().unwrap())
}
pub(crate) fn push_local_decl_on_stack(&mut self, ty: WasmerType) {
self.type_stack.push(ty);
}
pub(crate) fn push_params_on_stack(&mut self, function_index: LocalFunctionIndex) {
let func_index = self.module.func_index(function_index);
let sig_idx = self.module.functions[func_index];
let signature = &self.module.signatures[sig_idx];
self.return_types = signature.results().to_vec();
for param in signature.params() {
self.type_stack.push(*param);
}
}
pub(crate) fn return_types(&self) -> &[WasmerType] {
&self.return_types
}
pub(crate) fn emit_wasm_return(&mut self, builder: &mut FunctionBuilder, values: &[ir::Value]) {
let return_abi = abi::classify_returns(self.architecture, &self.return_types);
match &return_abi {
ReturnAbi::Sret(types) => {
let ptr = builder
.func
.special_param(ArgumentPurpose::StructReturn)
.expect("sret function has a StructReturn parameter");
let layout = abi::return_area_layout(types);
abi::store_sret(builder, ptr, &layout, values);
builder.ins().return_(&[]);
}
_ => {
let packed = abi::pack_register_returns(builder, &return_abi, values);
builder.ins().return_(&packed);
}
}
}
pub(crate) fn heap_access_spectre_mitigation(&self) -> bool {
false
}
pub(crate) fn heaps(&self) -> &PrimaryMap<Heap, HeapData> {
&self.heaps
}
}