diff options
Diffstat (limited to 'js/rust/src/panic.rs')
-rw-r--r-- | js/rust/src/panic.rs | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/js/rust/src/panic.rs b/js/rust/src/panic.rs new file mode 100644 index 0000000000..496ed9b33d --- /dev/null +++ b/js/rust/src/panic.rs @@ -0,0 +1,34 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +use std::any::Any; +use std::cell::RefCell; +use std::panic::{catch_unwind, resume_unwind, UnwindSafe}; + +thread_local!(static PANIC_RESULT: RefCell<Option<Box<Any + Send>>> = RefCell::new(None)); + +/// If there is a pending panic, resume unwinding. +pub fn maybe_resume_unwind() { + if let Some(error) = PANIC_RESULT.with(|result| result.borrow_mut().take()) { + resume_unwind(error); + } +} + +/// Generic wrapper for JS engine callbacks panic-catching +pub fn wrap_panic<F, R>(function: F, generic_return_type: R) -> R +where + F: FnOnce() -> R + UnwindSafe, +{ + let result = catch_unwind(function); + match result { + Ok(result) => result, + Err(error) => { + PANIC_RESULT.with(|result| { + assert!(result.borrow().is_none()); + *result.borrow_mut() = Some(error); + }); + generic_return_type + } + } +} |