56 lines
2.1 KiB
Rust
56 lines
2.1 KiB
Rust
//! Support for compiling with Lightbeam.
|
|
|
|
use crate::compilation::{AddressTransforms, Compilation, CompileError, Relocations};
|
|
use crate::func_environ::FuncEnvironment;
|
|
use crate::module::Module;
|
|
use crate::module_environ::FunctionBodyData;
|
|
// TODO: Put this in `compilation`
|
|
use crate::cranelift::RelocSink;
|
|
use cranelift_codegen::isa;
|
|
use cranelift_entity::PrimaryMap;
|
|
use cranelift_wasm::DefinedFuncIndex;
|
|
use lightbeam;
|
|
|
|
/// A compiler that compiles a WebAssembly module with Lightbeam, directly translating the Wasm file.
|
|
pub struct Lightbeam;
|
|
|
|
impl crate::compilation::Compiler for Lightbeam {
|
|
/// Compile the module using Lightbeam, producing a compilation result with
|
|
/// associated relocations.
|
|
fn compile_module<'data, 'module>(
|
|
module: &'module Module,
|
|
function_body_inputs: PrimaryMap<DefinedFuncIndex, FunctionBodyData<'data>>,
|
|
isa: &dyn isa::TargetIsa,
|
|
// TODO
|
|
_generate_debug_info: bool,
|
|
) -> Result<(Compilation, Relocations, AddressTransforms), CompileError> {
|
|
let env = FuncEnvironment::new(isa.frontend_config(), module);
|
|
let mut relocations = PrimaryMap::new();
|
|
let mut codegen_session: lightbeam::CodeGenSession<_> =
|
|
lightbeam::CodeGenSession::new(function_body_inputs.len() as u32, &env);
|
|
|
|
for (i, function_body) in &function_body_inputs {
|
|
let mut reloc_sink = RelocSink::new();
|
|
|
|
lightbeam::translate_function(
|
|
&mut codegen_session,
|
|
&mut reloc_sink,
|
|
i.as_u32(),
|
|
&lightbeam::wasmparser::FunctionBody::new(0, function_body.data),
|
|
)
|
|
.expect("Failed to translate function. TODO: Stop this from panicking");
|
|
relocations.push(reloc_sink.func_relocs);
|
|
}
|
|
|
|
let code_section = codegen_session
|
|
.into_translated_code_section()
|
|
.expect("Failed to generate output code. TODO: Stop this from panicking");
|
|
|
|
Ok((
|
|
Compilation::from_buffer(code_section.buffer().to_vec(), code_section.funcs()),
|
|
relocations,
|
|
AddressTransforms::new(),
|
|
))
|
|
}
|
|
}
|