mii/wasmer
1
0
Fork 0
mirror of https://github.com/mii443/wasmer.git synced 2025-08-25 01:39:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Initial commit for support of custom sections.

Browse Source
This commit is contained in:
Nick Lewycky
2020-05-01 15:46:45 -07:00
parent 393f17ba29
commit 1227d50e9b
2 changed files with 78 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
lib/compiler-llvm/src/compiler.rs
View File

@ -9,14 +9,16 @@ use rayon::prelude::{IntoParallelRefIterator, ParallelIterator};
use wasm_common::entity::{EntityRef, PrimaryMap, SecondaryMap};
use wasm_common::Features;
use wasm_common::{FuncIndex, FuncType, LocalFuncIndex, MemoryIndex, TableIndex};
use wasmer_compiler::FunctionBodyData;
use wasmer_compiler::TrapInformation;
use wasmer_compiler::{Compilation, CompileError, CompiledFunction, Compiler};
use wasmer_compiler::{CompilerConfig, ModuleTranslationState, Target};
use wasmer_compiler::{
Compilation, CompileError, CompiledFunction, Compiler, CompilerConfig, CustomSection,
CustomSectionProtection, FunctionBodyData, ModuleTranslationState, Relocation,
RelocationTarget, SectionIndex, Target, TrapInformation,
};
use wasmer_runtime::{MemoryPlan, Module, TablePlan, TrapCode};
use inkwell::targets::{InitializationConfig, Target as InkwellTarget};
use std::collections::HashMap;
use std::sync::{Arc, Mutex}; // TODO: remove
/// A compiler that compiles a WebAssembly module with LLVM, translating the Wasm to LLVM IR,
@ -62,7 +64,6 @@ impl Compiler for LLVMCompiler {
) -> Result<Compilation, CompileError> {
//let data = Arc::new(Mutex::new(0));
let mut func_names = SecondaryMap::new();
let custom_sections = PrimaryMap::new();
for (func_index, _) in &module.functions {
func_names[func_index] = module
@ -71,7 +72,7 @@ impl Compiler for LLVMCompiler {
.cloned()
.unwrap_or_else(|| format!("fn{}", func_index.index()));
}
let functions = function_body_inputs
let mut outputs = function_body_inputs
.into_iter()
.collect::<Vec<(LocalFuncIndex, &FunctionBodyData<'_>)>>()
.par_iter()
@ -92,6 +93,51 @@ impl Compiler for LLVMCompiler {
.into_iter()
.collect::<PrimaryMap<LocalFuncIndex, _>>();
// We're going to "link" the sections by simply appending all compatible
// sections, then building the new relocations.
// TODO: merge constants.
let mut readonly_section = CustomSection {
protection: CustomSectionProtection::Read,
bytes: Vec::new(),
};
/*
let mut readexecute_section = CustomSection {
protection: CustomSectionProtection::Read,
bytes: Vec::new(),
};
*/
for (ref mut function, ref local_relocations, ref custom_sections) in outputs.values_mut() {
for (local_idx, custom_section) in custom_sections.iter().enumerate() {
let local_idx = local_idx as u32;
// TODO: these section numbers are potentially wrong, if there's
// no Read and only a ReadExecute then ReadExecute is 0.
let (ref mut section, section_num) = match &custom_section.protection {
Read => (&mut readonly_section, SectionIndex::from_u32(0)),
//ReadExecute => (&mut readexecute_section, 1),
};
let offset = section.bytes.len() as i64;
section.bytes.extend(&custom_section.bytes);
// TODO: we're needlessly rescanning the whole list.
for local_relocation in local_relocations {
if local_relocation.local_section_index == local_idx {
function.relocations.push(Relocation {
kind: local_relocation.kind,
reloc_target: RelocationTarget::CustomSection(section_num),
offset: local_relocation.offset,
addend: local_relocation.addend + offset,
});
}
}
}
}
let functions = outputs
.iter_mut()
.map(|(i, (f, _, _))| *f)
.collect::<PrimaryMap<LocalFuncIndex, _>>();
let mut custom_sections = PrimaryMap::new();
custom_sections.push(readonly_section);
Ok(Compilation::new(functions, custom_sections))
}