Skip to content

Commit

Permalink
Fix tests for the Cranelift backend
Browse files Browse the repository at this point in the history
  • Loading branch information
Mark McCaskey committed Apr 24, 2020
1 parent b9ec8f9 commit 197f2f8
Show file tree
Hide file tree
Showing 9 changed files with 61 additions and 42 deletions.
20 changes: 10 additions & 10 deletions lib/llvm-backend/cpp/object_loader.hh
Original file line number Diff line number Diff line change
Expand Up @@ -50,8 +50,8 @@ typedef struct {
} callbacks_t;

typedef struct {
size_t data, vtable;
} box_any_t;
size_t data;
} runtime_error_t;

enum WasmTrapType {
Unreachable = 0,
Expand Down Expand Up @@ -121,7 +121,7 @@ private:

struct WasmErrorSink {
WasmTrapType *trap_out;
box_any_t *user_error;
runtime_error_t *user_error;
};

struct WasmException : std::exception {
Expand Down Expand Up @@ -149,14 +149,14 @@ public:

struct UserException : UncatchableException {
public:
UserException(size_t data, size_t vtable) : error_data({data, vtable}) {}
UserException(size_t data) : error_data({data}) {}

virtual std::string description() const noexcept override {
return "user exception";
}

// The parts of a `Box<dyn Any>`.
box_any_t error_data;
// The pointer to `RuntimeError`.
runtime_error_t error_data;

virtual void write_error(WasmErrorSink &out) const noexcept override {
*out.user_error = error_data;
Expand Down Expand Up @@ -274,10 +274,10 @@ result_t module_load(const uint8_t *mem_ptr, size_t mem_size,

void module_delete(WasmModule *module) { delete module; }

// Throw a fat pointer that's assumed to be `*mut dyn Any` on the rust
// Throw a pointer that's assumed to be `*mut RuntimeError` on the rust
// side.
[[noreturn]] void throw_any(size_t data, size_t vtable) {
unsafe_unwind(new UserException(data, vtable));
[[noreturn]] void throw_runtime_error(size_t data) {
unsafe_unwind(new UserException(data));
}

// Throw a pointer that's assumed to be codegen::BreakpointHandler on the
Expand All @@ -288,7 +288,7 @@ void module_delete(WasmModule *module) { delete module; }

bool cxx_invoke_trampoline(trampoline_t trampoline, void *ctx, void *func,
void *params, void *results, WasmTrapType *trap_out,
box_any_t *user_error, void *invoke_env) noexcept {
runtime_error_t *user_error, void *invoke_env) noexcept {
try {
catch_unwind([trampoline, ctx, func, params, results]() {
trampoline(ctx, func, params, results);
Expand Down
6 changes: 3 additions & 3 deletions lib/llvm-backend/src/backend.rs
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,8 @@ extern "C" {
/// but this is cleaner, I think?
#[cfg_attr(nightly, unwind(allowed))]
#[allow(improper_ctypes)]
fn throw_runtime_error(data: RuntimeError) -> !;
// POINTER!!!!! TODOARINO
fn throw_runtime_error(data: *mut RuntimeError) -> !;

#[allow(improper_ctypes)]
fn cxx_invoke_trampoline(
Expand Down Expand Up @@ -475,8 +476,7 @@ impl RunnableModule for LLVMBackend {
}

unsafe fn do_early_trap(&self, data: RuntimeError) -> ! {
// maybe need to box leak it?
throw_runtime_error(data)
throw_runtime_error(Box::into_raw(Box::new(data)))
}
}

Expand Down
5 changes: 3 additions & 2 deletions lib/llvm-backend/src/code.rs
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@ use std::{
use wasmer_runtime_core::{
backend::{CacheGen, CompilerConfig, Token},
cache::{Artifact, Error as CacheError},
error::RuntimeError,
codegen::*,
memory::MemoryType,
module::{ModuleInfo, ModuleInner},
Expand Down Expand Up @@ -940,11 +941,11 @@ pub struct CodegenError {
// prevents unused function elimination.
#[no_mangle]
pub unsafe extern "C" fn callback_trampoline(
b: *mut Option<Box<dyn std::any::Any>>,
b: *mut Option<RuntimeError>,
callback: *mut BreakpointHandler,
) {
let callback = Box::from_raw(callback);
let result: Result<(), Box<dyn std::any::Any + Send>> =
let result: Result<(), RuntimeError> =
callback(BreakpointInfo { fault: None });
match result {
Ok(()) => *b = None,
Expand Down
17 changes: 16 additions & 1 deletion lib/runtime-core/src/error.rs
Original file line number Diff line number Diff line change
Expand Up @@ -265,10 +265,25 @@ pub enum InvokeError {
/// extra TODO: investigate if this can be a `Box<InvokeError>` instead (looks like probably no)
/// TODO:
EarlyTrap(Box<RuntimeError>),
/// Indicates an error that ocurred related to breakpoints. (currently Singlepass only)
/// Indicates an error that ocurred related to breakpoints.
Breakpoint(Box<RuntimeError>),
}

impl From<InvokeError> for RuntimeError {
fn from(other: InvokeError) -> RuntimeError {
match other {
InvokeError::EarlyTrap(re) | InvokeError::Breakpoint(re) => *re,
_ => RuntimeError::InvokeError(other),
}
}
}

impl PartialEq for RuntimeError {
fn eq(&self, _other: &RuntimeError) -> bool {
false
}
}

//impl std::error::Error for InvokeError {}

impl std::error::Error for RuntimeError {}
Expand Down
24 changes: 11 additions & 13 deletions lib/runtime-core/src/instance.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
use crate::{
backend::RunnableModule,
backing::{ImportBacking, LocalBacking},
error::{CallResult, InvokeError, ResolveError, ResolveResult, Result, RuntimeError},
error::{CallResult, ResolveError, ResolveResult, Result, RuntimeError, InvokeError},
export::{Context, Export, ExportIter, Exportable, FuncPointer},
global::Global,
import::{ImportObject, LikeNamespace},
Expand Down Expand Up @@ -587,23 +587,21 @@ pub(crate) fn call_func_with_index_inner(
let run_wasm = |result_space: *mut u64| -> CallResult<()> {
let mut error_out = None;

let success = unsafe {
invoke(
trampoline,
ctx_ptr,
func_ptr,
raw_args.as_ptr(),
result_space,
&mut error_out,
invoke_env,
)
};
let success = unsafe { invoke(
trampoline,
ctx_ptr,
func_ptr,
raw_args.as_ptr(),
result_space,
&mut error_out,
invoke_env,
)};

if success {
Ok(())
} else {
let error: RuntimeError = error_out
.map(RuntimeError::InvokeError)
.map(Into::into)
.unwrap_or_else(|| RuntimeError::InvokeError(InvokeError::FailedWithNoError));
Err(error.into())
}
Expand Down
2 changes: 1 addition & 1 deletion lib/runtime-core/src/typed_func.rs
Original file line number Diff line number Diff line change
Expand Up @@ -590,7 +590,7 @@ macro_rules! impl_traits {
) {
Ok(Rets::from_ret_array(rets))
} else {
Err(error_out.map_or_else(|| RuntimeError::InvokeError(InvokeError::FailedWithNoError), RuntimeError::InvokeError))
Err(error_out.map_or_else(|| RuntimeError::InvokeError(InvokeError::FailedWithNoError), Into::into))
}
}
}
Expand Down
2 changes: 1 addition & 1 deletion tests/high_level_api.rs
Original file line number Diff line number Diff line change
Expand Up @@ -268,7 +268,7 @@ wasmer_backends! {

let result = foo.call();

if let Err(RuntimeError(e)) = result {
if let Err(RuntimeError::User(e)) = result {
let exit_code = e.downcast::<ExitCode>().unwrap();
assert_eq!(exit_code.code, 42);
} else {
Expand Down
14 changes: 7 additions & 7 deletions tests/imports.rs
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ macro_rules! call_and_assert {
expected_value,
concat!("Expected right when calling `", stringify!($function), "`.")
),
(Err(RuntimeError(data)), Err(RuntimeError(expected_data))) => {
(Err(RuntimeError::User(data)), Err(RuntimeError::User(expected_data))) => {
if let (Some(data), Some(expected_data)) = (
data.downcast_ref::<&str>(),
expected_data.downcast_ref::<&str>(),
Expand Down Expand Up @@ -406,7 +406,7 @@ wasmer_backends! {
test!( test_fn, function_fn(i32) -> i32, (1) == Ok(2));
test!( test_closure, function_closure(i32) -> i32, (1) == Ok(2));
test!( test_fn_dynamic, function_fn_dynamic(i32) -> i32, (1) == Ok(2));
test!( test_fn_dynamic_panic, function_fn_dynamic_panic(i32) -> i32, (1) == Err(RuntimeError(Box::new("test"))));
test!( test_fn_dynamic_panic, function_fn_dynamic_panic(i32) -> i32, (1) == Err(RuntimeError::User(Box::new("test"))));
test!(

test_closure_dynamic_0,
Expand Down Expand Up @@ -460,31 +460,31 @@ wasmer_backends! {

test_fn_trap,
function_fn_trap(i32) -> i32,
(1) == Err(RuntimeError(Box::new(format!("foo {}", 2))))
(1) == Err(RuntimeError::User(Box::new(format!("foo {}", 2))))
);
test!(

test_closure_trap,
function_closure_trap(i32) -> i32,
(1) == Err(RuntimeError(Box::new(format!("bar {}", 2))))
(1) == Err(RuntimeError::User(Box::new(format!("bar {}", 2))))
);
test!(

test_fn_trap_with_vmctx,
function_fn_trap_with_vmctx(i32) -> i32,
(1) == Err(RuntimeError(Box::new(format!("baz {}", 2 + SHIFT))))
(1) == Err(RuntimeError::User(Box::new(format!("baz {}", 2 + SHIFT))))
);
test!(

test_closure_trap_with_vmctx,
function_closure_trap_with_vmctx(i32) -> i32,
(1) == Err(RuntimeError(Box::new(format!("qux {}", 2 + SHIFT))))
(1) == Err(RuntimeError::User(Box::new(format!("qux {}", 2 + SHIFT))))
);
test!(

test_closure_trap_with_vmctx_and_env,
function_closure_trap_with_vmctx_and_env(i32) -> i32,
(1) == Err(RuntimeError(Box::new(format!("! {}", 2 + shift + SHIFT))))
(1) == Err(RuntimeError::User(Box::new(format!("! {}", 2 + shift + SHIFT))))
);

#[test]
Expand Down
13 changes: 9 additions & 4 deletions tests/middleware_common.rs
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ use wasmer_middleware_common::metering::*;
use wasmer_runtime_core::codegen::ModuleCodeGenerator;
use wasmer_runtime_core::codegen::{MiddlewareChain, StreamingCompiler};
use wasmer_runtime_core::fault::{pop_code_version, push_code_version};
use wasmer_runtime_core::error::RuntimeError;
use wasmer_runtime_core::state::CodeVersion;

// Assemblyscript
Expand Down Expand Up @@ -179,10 +180,14 @@ fn middleware_test_traps_after_costly_call(backend: &'static str, compiler: impl
}

let err = result.unwrap_err();
assert!(err
.0
.downcast_ref::<ExecutionLimitExceededError>()
.is_some());
if let RuntimeError::Metering(metering_err) = err {
assert!(metering_err
.downcast_ref::<ExecutionLimitExceededError>()
.is_some());
} else {
assert!(false, "metering error not found");
}


// verify it used the correct number of points
assert_eq!(get_points_used(&instance), 109); // Used points will be slightly more than `limit` because of the way we do gas checking.
Expand Down

0 comments on commit 197f2f8

Please sign in to comment.