1 #include <assert.h>
2 #include <stdlib.h>
3 #include "../../js-native-api/common.h"
4 #include "node_api.h"
5 #include "uv.h"
6
7 static int cleanup_hook_count = 0;
cleanup(void* arg)8 static void cleanup(void* arg) {
9 cleanup_hook_count++;
10 printf("cleanup(%d)\n", *(int*)(arg));
11 }
12
13 static int secret = 42;
14 static int wrong_secret = 17;
15
ObjectFinalizer(napi_env env, void* data, void* hint)16 static void ObjectFinalizer(napi_env env, void* data, void* hint) {
17 // cleanup is called once.
18 assert(cleanup_hook_count == 1);
19
20 napi_ref* ref = data;
21 NODE_API_CALL_RETURN_VOID(env, napi_delete_reference(env, *ref));
22 free(ref);
23 }
24
CreateObjectWrap(napi_env env)25 static void CreateObjectWrap(napi_env env) {
26 napi_value js_obj;
27 napi_ref* ref = malloc(sizeof(*ref));
28 NODE_API_CALL_RETURN_VOID(env, napi_create_object(env, &js_obj));
29 NODE_API_CALL_RETURN_VOID(
30 env, napi_wrap(env, js_obj, ref, ObjectFinalizer, NULL, ref));
31 // create a strong reference so that the finalizer is called at shutdown.
32 NODE_API_CALL_RETURN_VOID(env, napi_reference_ref(env, *ref, NULL));
33 }
34
Init(napi_env env, napi_value exports)35 static napi_value Init(napi_env env, napi_value exports) {
36 // Create object wrap before cleanup hooks.
37 CreateObjectWrap(env);
38
39 napi_add_env_cleanup_hook(env, cleanup, &wrong_secret);
40 napi_add_env_cleanup_hook(env, cleanup, &secret);
41 napi_remove_env_cleanup_hook(env, cleanup, &wrong_secret);
42
43 // Create object wrap after cleanup hooks.
44 CreateObjectWrap(env);
45
46 return NULL;
47 }
48
49 NAPI_MODULE(NODE_GYP_MODULE_NAME, Init)
50