archive-gh-me/napi-rs
1
0
Fork 0
mirror of https://github.com/napi-rs/napi-rs.git synced 2025-12-08 19:56:07 +00:00
Code Issues Projects Releases Wiki Activity

Mutex on Windows

Browse Source
This commit is contained in:
LongYinan 2025-10-10 10:12:11 +08:00
parent af89eacad5
commit 52284d15e2
No known key found for this signature in database
1 changed files with 32 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
crates/napi/src/bindgen_runtime/module_register.rs
View File

@ -9,6 +9,8 @@ use std::ffi::CStr;
use std::mem::MaybeUninit;
#[cfg(not(feature = "noop"))]
use std::ptr;
#[cfg(all(not(feature = "noop"), target_os = "windows"))]
use std::sync::Mutex;
#[cfg(not(feature = "noop"))]
use std::sync::{
atomic::{AtomicBool, AtomicUsize, Ordering},
@ -123,6 +125,9 @@ thread_local! {
pub(crate) static THREADS_CAN_ACCESS_ENV: Cell<bool> = const { Cell::new(false) };
}
#[cfg(all(feature = "napi4", not(feature = "noop"), target_os = "windows"))]
static CUSTOM_GC_ENV_ALIVE: LazyLock<Mutex<bool>> = LazyLock::new(|| Mutex::new(false));
type RegisteredClasses = PersistedPerInstanceHashMap<
/* export name */ String,
/* constructor */ sys::napi_ref,
@ -567,6 +572,13 @@ fn create_custom_gc(env: sys::napi_env) {
}
THREADS_CAN_ACCESS_ENV.with(|cell| cell.set(true));
#[cfg(target_os = "windows")]
{
let mut env_alive = CUSTOM_GC_ENV_ALIVE
.lock()
.expect("Lock CUSTOM_GC_ENV_ALIVE in create_custom_gc");
*env_alive = true;
}
}
#[cfg(all(
@ -610,6 +622,13 @@ unsafe extern "C" fn custom_gc_finalize(
finalize_hint: *mut std::ffi::c_void,
) {
CUSTOM_GC_TSFN_DESTROYED.store(true, Ordering::SeqCst);
#[cfg(target_os = "windows")]
{
let mut env_alive = CUSTOM_GC_ENV_ALIVE
.lock()
.expect("Lock CUSTOM_GC_ENV_ALIVE in custom_gc_finalize");
*env_alive = false;
}
}
#[cfg(all(feature = "napi4", not(feature = "noop")))]
@ -621,9 +640,21 @@ extern "C" fn custom_gc(
data: *mut std::ffi::c_void,
) {
// current thread was destroyed
if THREADS_CAN_ACCESS_ENV.with(|cell| !cell.get()) || data.is_null() {
let env_accessible = THREADS_CAN_ACCESS_ENV.with(|cell| cell.get());
#[cfg(target_os = "windows")]
let env_guard = CUSTOM_GC_ENV_ALIVE
.lock()
.expect("Lock CUSTOM_GC_ENV_ALIVE in custom_gc");
#[cfg(target_os = "windows")]
let env_alive = *env_guard;
#[cfg(not(target_os = "windows"))]
let env_alive = true;
if !env_accessible || !env_alive || data.is_null() {
return;
}
#[cfg(target_os = "windows")]
let _env_guard = env_guard;
let mut ref_count = 0;
check_status_or_throw!(
env,