summaryrefslogtreecommitdiff
path: root/tests/napi/src/tsfn.rs
blob: a3a231cec19c123bb6f16bed23bf40df412334dc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license.

// This test performs initialization similar to napi-rs.
// https://github.com/napi-rs/napi-rs/commit/a5a04a4e545f268769cc78e2bd6c45af4336aac3

use napi_sys as sys;
use std::ffi::c_char;
use std::ffi::c_void;
use std::ptr;

macro_rules! check_status_or_panic {
  ($code:expr, $msg:expr) => {{
    let c = $code;
    match c {
      sys::Status::napi_ok => {}
      _ => panic!($msg),
    }
  }};
}

fn create_custom_gc(env: sys::napi_env) {
  let mut custom_gc_fn = ptr::null_mut();
  check_status_or_panic!(
    unsafe {
      sys::napi_create_function(
        env,
        "custom_gc".as_ptr() as *const c_char,
        9,
        Some(empty),
        ptr::null_mut(),
        &mut custom_gc_fn,
      )
    },
    "Create Custom GC Function in napi_register_module_v1 failed"
  );
  let mut async_resource_name = ptr::null_mut();
  check_status_or_panic!(
    unsafe {
      sys::napi_create_string_utf8(
        env,
        "CustomGC".as_ptr() as *const c_char,
        8,
        &mut async_resource_name,
      )
    },
    "Create async resource string in napi_register_module_v1 napi_register_module_v1"
  );
  let mut custom_gc_tsfn = ptr::null_mut();
  let context = Box::into_raw(Box::new(0)) as *mut c_void;
  check_status_or_panic!(
    unsafe {
      sys::napi_create_threadsafe_function(
        env,
        custom_gc_fn,
        ptr::null_mut(),
        async_resource_name,
        0,
        1,
        ptr::null_mut(),
        Some(custom_gc_finalize),
        context,
        Some(custom_gc),
        &mut custom_gc_tsfn,
      )
    },
    "Create Custom GC ThreadsafeFunction in napi_register_module_v1 failed"
  );
  check_status_or_panic!(
    unsafe { sys::napi_unref_threadsafe_function(env, custom_gc_tsfn) },
    "Unref Custom GC ThreadsafeFunction in napi_register_module_v1 failed"
  );
}

unsafe extern "C" fn empty(
  _env: sys::napi_env,
  _info: sys::napi_callback_info,
) -> sys::napi_value {
  ptr::null_mut()
}

unsafe extern "C" fn custom_gc_finalize(
  _env: sys::napi_env,
  _finalize_data: *mut c_void,
  finalize_hint: *mut c_void,
) {
  let _ = Box::from_raw(finalize_hint as *mut i32);
}

extern "C" fn custom_gc(
  env: sys::napi_env,
  _js_callback: sys::napi_value,
  _context: *mut c_void,
  data: *mut c_void,
) {
  let mut ref_count = 0;
  check_status_or_panic!(
    unsafe {
      sys::napi_reference_unref(env, data as sys::napi_ref, &mut ref_count)
    },
    "Failed to unref Buffer reference in Custom GC"
  );
  check_status_or_panic!(
    unsafe { sys::napi_delete_reference(env, data as sys::napi_ref) },
    "Failed to delete Buffer reference in Custom GC"
  );
}

pub fn init(env: sys::napi_env, _exports: sys::napi_value) {
  create_custom_gc(env);
}