55 lines
1.3 KiB
Rust
55 lines
1.3 KiB
Rust
use once_cell::sync::Lazy;
|
|
use std::collections::HashMap;
|
|
use std::sync::Mutex;
|
|
|
|
static ENABLED: Lazy<bool> =
|
|
Lazy::new(|| std::env::var("NYASH_LEAK_LOG").unwrap_or_default() == "1");
|
|
static LEAKS: Lazy<Mutex<HashMap<(String, u32), &'static str>>> =
|
|
Lazy::new(|| Mutex::new(HashMap::new()));
|
|
|
|
pub fn init() {
|
|
let _ = &*REPORTER;
|
|
}
|
|
|
|
pub fn register_plugin(box_type: &str, instance_id: u32) {
|
|
if !*ENABLED {
|
|
return;
|
|
}
|
|
LEAKS
|
|
.lock()
|
|
.unwrap()
|
|
.insert((box_type.to_string(), instance_id), "plugin");
|
|
}
|
|
|
|
pub fn finalize_plugin(box_type: &str, instance_id: u32) {
|
|
if !*ENABLED {
|
|
return;
|
|
}
|
|
LEAKS
|
|
.lock()
|
|
.unwrap()
|
|
.remove(&(box_type.to_string(), instance_id));
|
|
}
|
|
|
|
struct Reporter;
|
|
impl Drop for Reporter {
|
|
fn drop(&mut self) {
|
|
if !*ENABLED {
|
|
return;
|
|
}
|
|
let m = LEAKS.lock().unwrap();
|
|
if m.is_empty() {
|
|
return;
|
|
}
|
|
eprintln!("[leak] Detected {} non-finalized plugin boxes:", m.len());
|
|
for ((ty, id), _) in m.iter() {
|
|
eprintln!(
|
|
" - {}(id={}) not finalized (missing fini or scope)",
|
|
ty, id
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
static REPORTER: Lazy<Reporter> = Lazy::new(|| Reporter);
|