forked from AbleOS/ableos
119 lines
3.8 KiB
Rust
119 lines
3.8 KiB
Rust
use alloc::string::String;
|
|
use wasmi::TrapKind;
|
|
|
|
use {
|
|
alloc::format,
|
|
wasm_sys::SysCall,
|
|
wasmi::{
|
|
Error, Externals, FuncInstance, FuncRef, ImportsBuilder, ModuleImportResolver,
|
|
ModuleInstance, RuntimeArgs, RuntimeValue, Signature, Trap, ValueType,
|
|
},
|
|
};
|
|
|
|
mod wasm_sys;
|
|
struct HostFunctions;
|
|
impl HostFunctions {
|
|
fn check_signature(&self, index: usize, signature: &Signature) -> bool {
|
|
let (params, ret_ty): (&[ValueType], Option<ValueType>) = match index.into() {
|
|
SysCall::KILL => (&[], None),
|
|
SysCall::EMPTY => (&[], None),
|
|
_ => return false,
|
|
};
|
|
signature.params() == params && signature.return_type() == ret_ty
|
|
}
|
|
}
|
|
impl Externals for HostFunctions {
|
|
fn invoke_index(
|
|
&mut self,
|
|
index: usize,
|
|
args: RuntimeArgs,
|
|
) -> Result<Option<RuntimeValue>, Trap> {
|
|
match index.into() {
|
|
// Take in one arg discard the rest
|
|
SysCall::KILL => {
|
|
info!("Program run at runtime called a system call");
|
|
debug!("Runtime arguments: {:?}", args);
|
|
Ok(None)
|
|
}
|
|
// Do nothing
|
|
SysCall::EMPTY => Ok(None),
|
|
SysCall::EXIT => Ok(None),
|
|
SysCall::CONSOLE_RESET => Ok(None),
|
|
SysCall::CONSOLE_IN => {
|
|
info!("In");
|
|
Ok(None)
|
|
}
|
|
|
|
SysCall::CONSOLE_OUT => {
|
|
// Eventually change this to 2- ptr and len
|
|
if args.len() != 1 {
|
|
return Err(Trap::new(TrapKind::UnexpectedSignature));
|
|
}
|
|
let arg: u64 = args.nth(0);
|
|
let buf = unsafe { String::from_utf8_unchecked(arg.to_le_bytes().to_vec()) };
|
|
println!["{}", buf];
|
|
Ok(None)
|
|
}
|
|
SysCall::CONSOLE_GET_TITLE => Ok(None),
|
|
SysCall::CONSOLE_SET_TITLE => Ok(None),
|
|
_ => panic!("Unimplemented function at {}", index),
|
|
}
|
|
}
|
|
}
|
|
impl ModuleImportResolver for HostFunctions {
|
|
fn resolve_func(&self, field_name: &str, signature: &Signature) -> Result<FuncRef, Error> {
|
|
let index = match field_name {
|
|
"kill" => SysCall::KILL as usize,
|
|
"empty" => SysCall::EMPTY as usize,
|
|
"exit" => SysCall::EXIT as usize,
|
|
"console_out" => SysCall::CONSOLE_OUT as usize,
|
|
_ => {
|
|
return Err(Error::Instantiation(format!(
|
|
"Export {} not found",
|
|
field_name
|
|
)))
|
|
}
|
|
};
|
|
|
|
if !self.check_signature(index, signature) {
|
|
return Err(Error::Instantiation(format!(
|
|
"Export {} has a bad signature",
|
|
field_name
|
|
)));
|
|
}
|
|
|
|
Ok(FuncInstance::alloc_host(
|
|
Signature::new(&[][..], None),
|
|
index,
|
|
))
|
|
}
|
|
}
|
|
|
|
pub fn evaluate() {
|
|
// let wasm_binary = include_bytes!("bin/console_out_test.wasm");
|
|
let wasm_binary = include_bytes!("bin/rust.wasm");
|
|
|
|
// Load wasm binary and prepare it for instantiation.
|
|
let module = wasmi::Module::from_buffer(&wasm_binary).expect("failed to load wasm");
|
|
|
|
let imports = ImportsBuilder::new().with_resolver("env", &HostFunctions);
|
|
|
|
// Instantiate a module with empty imports and
|
|
// assert that there is no `start` function.
|
|
let instance = ModuleInstance::new(&module, &imports)
|
|
// .expect("failed to instantiate wasm module")
|
|
.unwrap()
|
|
.assert_no_start();
|
|
|
|
// Finally, invoke the exported function "test" with no parameters
|
|
// and empty external function executor.
|
|
let result: i32 = instance
|
|
.invoke_export("_start", &[], &mut HostFunctions)
|
|
.expect("failed to execute export")
|
|
.unwrap()
|
|
.try_into()
|
|
.unwrap();
|
|
|
|
println!("{:?}", result);
|
|
}
|