diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 00:47:55 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 00:47:55 +0000 |
commit | 26a029d407be480d791972afb5975cf62c9360a6 (patch) | |
tree | f435a8308119effd964b339f76abb83a57c29483 /third_party/rust/diplomat-runtime/src/wasm_glue.rs | |
parent | Initial commit. (diff) | |
download | firefox-26a029d407be480d791972afb5975cf62c9360a6.tar.xz firefox-26a029d407be480d791972afb5975cf62c9360a6.zip |
Adding upstream version 124.0.1.upstream/124.0.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/rust/diplomat-runtime/src/wasm_glue.rs')
-rw-r--r-- | third_party/rust/diplomat-runtime/src/wasm_glue.rs | 94 |
1 files changed, 94 insertions, 0 deletions
diff --git a/third_party/rust/diplomat-runtime/src/wasm_glue.rs b/third_party/rust/diplomat-runtime/src/wasm_glue.rs new file mode 100644 index 0000000000..98a46139a3 --- /dev/null +++ b/third_party/rust/diplomat-runtime/src/wasm_glue.rs @@ -0,0 +1,94 @@ +use alloc::format; +use core::panic::PanicInfo; + +#[no_mangle] +unsafe extern "C" fn diplomat_init() { + #[cfg(debug_assertions)] + std::panic::set_hook(Box::new(panic_handler)); + #[cfg(feature = "log")] + log::set_logger(&ConsoleLogger) + .map(|()| log::set_max_level(log::LevelFilter::Debug)) + .unwrap(); +} + +fn panic_handler(info: &PanicInfo) { + let msg = match info.payload().downcast_ref::<&'static str>() { + Some(&s) => s, + None => match info.payload().downcast_ref::<String>() { + Some(s) => s.as_str(), + None => "Box<Any>", + }, + }; + + let msg = match info.location() { + Some(l) => format!( + "wasm panicked at {}:{}:{}:\n{msg}", + l.file(), + l.line(), + l.column(), + ), + None => format!("wasm panicked at <unknown location>:\n{msg}"), + }; + + extern "C" { + fn diplomat_throw_error_js(ptr: *const u8, len: usize); + } + + unsafe { diplomat_throw_error_js(msg.as_ptr(), msg.len()) } +} + +#[cfg(feature = "log")] +struct ConsoleLogger; + +#[cfg(feature = "log")] +impl log::Log for ConsoleLogger { + #[inline] + fn enabled(&self, metadata: &log::Metadata) -> bool { + metadata.level() <= log::max_level() + } + + fn log(&self, record: &log::Record) { + if !self.enabled(record.metadata()) { + return; + } + + let out = match record.level() { + log::Level::Error => { + extern "C" { + fn diplomat_console_error_js(ptr: *const u8, len: usize); + } + diplomat_console_error_js + } + log::Level::Warn => { + extern "C" { + fn diplomat_console_warn_js(ptr: *const u8, len: usize); + } + diplomat_console_warn_js + } + log::Level::Info => { + extern "C" { + fn diplomat_console_info_js(ptr: *const u8, len: usize); + } + diplomat_console_info_js + } + log::Level::Debug => { + extern "C" { + fn diplomat_console_log_js(ptr: *const u8, len: usize); + } + diplomat_console_log_js + } + log::Level::Trace => { + extern "C" { + fn diplomat_console_debug_js(ptr: *const u8, len: usize); + } + diplomat_console_debug_js + } + }; + + let msg = alloc::format!("{}", record.args()); + + unsafe { out(msg.as_ptr(), msg.len()) }; + } + + fn flush(&self) {} +} |