T0b1/wasmtime
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

catch panic in libcalls for memory and table grow

Browse Source
This commit is contained in:
Pat Hickey
2021-10-21 12:15:00 -07:00
parent d3deaae99d
commit 5aef8f47c8
1 changed files with 33 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
crates/runtime/src/libcalls.rs
View File

@@ -59,7 +59,7 @@
use crate::externref::VMExternRef; use crate::externref::VMExternRef;
use crate::instance::Instance; use crate::instance::Instance;
use crate::table::{Table, TableElementType}; use crate::table::{Table, TableElementType};
use crate::traphandlers::{raise_lib_trap, Trap}; use crate::traphandlers::{raise_lib_trap, resume_panic, Trap};
use crate::vmcontext::{VMCallerCheckedAnyfunc, VMContext}; use crate::vmcontext::{VMCallerCheckedAnyfunc, VMContext};
use backtrace::Backtrace; use backtrace::Backtrace;
use std::mem; use std::mem;
@@ -190,6 +190,7 @@ pub unsafe extern "C" fn wasmtime_memory32_grow(
delta: u64, delta: u64,
memory_index: u32, memory_index: u32,
) -> usize { ) -> usize {
std::panic::catch_unwind(|| {
let instance = (*vmctx).instance_mut(); let instance = (*vmctx).instance_mut();
let memory_index = MemoryIndex::from_u32(memory_index); let memory_index = MemoryIndex::from_u32(memory_index);
match instance.memory_grow(memory_index, delta) { match instance.memory_grow(memory_index, delta) {
@@ -197,6 +198,9 @@ pub unsafe extern "C" fn wasmtime_memory32_grow(
Ok(None) => usize::max_value(), Ok(None) => usize::max_value(),
Err(err) => crate::traphandlers::raise_user_trap(err), Err(err) => crate::traphandlers::raise_user_trap(err),
} }
})
.map_err(|panic| resume_panic(panic))
.unwrap()
} }
/// Implementation of `table.grow`. /// Implementation of `table.grow`.
@@ -208,6 +212,7 @@ pub unsafe extern "C" fn wasmtime_table_grow(
// or is a `VMExternRef` until we look at the table type. // or is a `VMExternRef` until we look at the table type.
init_value: *mut u8, init_value: *mut u8,
) -> u32 { ) -> u32 {
std::panic::catch_unwind(|| {
let instance = (*vmctx).instance_mut(); let instance = (*vmctx).instance_mut();
let table_index = TableIndex::from_u32(table_index); let table_index = TableIndex::from_u32(table_index);
let element = match instance.table_element_type(table_index) { let element = match instance.table_element_type(table_index) {
@@ -226,6 +231,9 @@ pub unsafe extern "C" fn wasmtime_table_grow(
Ok(None) => -1_i32 as u32, Ok(None) => -1_i32 as u32,
Err(err) => crate::traphandlers::raise_user_trap(err), Err(err) => crate::traphandlers::raise_user_trap(err),
} }
})
.map_err(|panic| resume_panic(panic))
.unwrap()
} }
/// Implementation of `table.fill`. /// Implementation of `table.fill`.