|
| 1 | +#include <node_api.h> |
| 2 | +#include "../../js-native-api/common.h" |
| 3 | +#include "stdlib.h" |
| 4 | + |
| 5 | +#define NODE_API_ASSERT_STATUS(env, assertion, message) \ |
| 6 | + NODE_API_ASSERT_BASE(env, assertion, message, napi_generic_failure) |
| 7 | + |
| 8 | +#define NODE_API_CHECK_STATUS(env, the_call) \ |
| 9 | + do { \ |
| 10 | + napi_status status = (the_call); \ |
| 11 | + if (status != napi_ok) { \ |
| 12 | + return status; \ |
| 13 | + } \ |
| 14 | + } while (0) |
| 15 | + |
| 16 | +static uint32_t finalizeCount = 0; |
| 17 | + |
| 18 | +static void FreeData(napi_env env, void* data, void* hint) { |
| 19 | + NODE_API_ASSERT_RETURN_VOID(env, data != NULL, "Expects non-NULL data."); |
| 20 | + free(data); |
| 21 | +} |
| 22 | + |
| 23 | +static void Finalize(napi_env env, void* data, void* hint) { |
| 24 | + ++finalizeCount; |
| 25 | +} |
| 26 | + |
| 27 | +static napi_status GetArgValue(napi_env env, |
| 28 | + napi_callback_info info, |
| 29 | + napi_value* argValue) { |
| 30 | + size_t argc = 1; |
| 31 | + NODE_API_CHECK_STATUS( |
| 32 | + env, napi_get_cb_info(env, info, &argc, argValue, NULL, NULL)); |
| 33 | + |
| 34 | + NODE_API_ASSERT_STATUS(env, argc == 1, "Expects one arg."); |
| 35 | + return napi_ok; |
| 36 | +} |
| 37 | + |
| 38 | +static napi_status GetArgValueAsIndex(napi_env env, |
| 39 | + napi_callback_info info, |
| 40 | + uint32_t* index) { |
| 41 | + napi_value argValue; |
| 42 | + NODE_API_CHECK_STATUS(env, GetArgValue(env, info, &argValue)); |
| 43 | + |
| 44 | + napi_valuetype valueType; |
| 45 | + NODE_API_CHECK_STATUS(env, napi_typeof(env, argValue, &valueType)); |
| 46 | + NODE_API_ASSERT_STATUS( |
| 47 | + env, valueType == napi_number, "Argument must be a number."); |
| 48 | + |
| 49 | + return napi_get_value_uint32(env, argValue, index); |
| 50 | +} |
| 51 | + |
| 52 | +static napi_status GetRef(napi_env env, |
| 53 | + napi_callback_info info, |
| 54 | + napi_ref* ref) { |
| 55 | + uint32_t index; |
| 56 | + NODE_API_CHECK_STATUS(env, GetArgValueAsIndex(env, info, &index)); |
| 57 | + |
| 58 | + napi_ref* refValues; |
| 59 | + NODE_API_CHECK_STATUS(env, napi_get_instance_data(env, (void**)&refValues)); |
| 60 | + NODE_API_ASSERT_STATUS(env, refValues != NULL, "Cannot get instance data."); |
| 61 | + |
| 62 | + *ref = refValues[index]; |
| 63 | + return napi_ok; |
| 64 | +} |
| 65 | + |
| 66 | +static napi_value ToUInt32Value(napi_env env, uint32_t value) { |
| 67 | + napi_value result; |
| 68 | + NODE_API_CALL(env, napi_create_uint32(env, value, &result)); |
| 69 | + return result; |
| 70 | +} |
| 71 | + |
| 72 | +static napi_status InitRefArray(napi_env env) { |
| 73 | + // valueRefs array has one entry per napi_valuetype |
| 74 | + napi_ref* valueRefs = malloc(sizeof(napi_ref) * ((int)napi_bigint + 1)); |
| 75 | + return napi_set_instance_data(env, valueRefs, &FreeData, NULL); |
| 76 | +} |
| 77 | + |
| 78 | +static napi_value CreateExternal(napi_env env, napi_callback_info info) { |
| 79 | + napi_value result; |
| 80 | + int* data = (int*)malloc(sizeof(int)); |
| 81 | + *data = 42; |
| 82 | + NODE_API_CALL(env, napi_create_external(env, data, &FreeData, NULL, &result)); |
| 83 | + return result; |
| 84 | +} |
| 85 | + |
| 86 | +static napi_value CreateRef(napi_env env, napi_callback_info info) { |
| 87 | + napi_value argValue; |
| 88 | + NODE_API_CALL(env, GetArgValue(env, info, &argValue)); |
| 89 | + |
| 90 | + napi_valuetype valueType; |
| 91 | + NODE_API_CALL(env, napi_typeof(env, argValue, &valueType)); |
| 92 | + uint32_t index = (uint32_t)valueType; |
| 93 | + |
| 94 | + napi_ref* valueRefs; |
| 95 | + NODE_API_CALL(env, napi_get_instance_data(env, (void**)&valueRefs)); |
| 96 | + NODE_API_CALL(env, |
| 97 | + napi_create_reference(env, argValue, 1, valueRefs + index)); |
| 98 | + |
| 99 | + return ToUInt32Value(env, index); |
| 100 | +} |
| 101 | + |
| 102 | +static napi_value GetRefValue(napi_env env, napi_callback_info info) { |
| 103 | + napi_ref refValue; |
| 104 | + NODE_API_CALL(env, GetRef(env, info, &refValue)); |
| 105 | + napi_value value; |
| 106 | + NODE_API_CALL(env, napi_get_reference_value(env, refValue, &value)); |
| 107 | + return value; |
| 108 | +} |
| 109 | + |
| 110 | +static napi_value Ref(napi_env env, napi_callback_info info) { |
| 111 | + napi_ref refValue; |
| 112 | + NODE_API_CALL(env, GetRef(env, info, &refValue)); |
| 113 | + uint32_t refCount; |
| 114 | + NODE_API_CALL(env, napi_reference_ref(env, refValue, &refCount)); |
| 115 | + return ToUInt32Value(env, refCount); |
| 116 | +} |
| 117 | + |
| 118 | +static napi_value Unref(napi_env env, napi_callback_info info) { |
| 119 | + napi_ref refValue; |
| 120 | + NODE_API_CALL(env, GetRef(env, info, &refValue)); |
| 121 | + uint32_t refCount; |
| 122 | + NODE_API_CALL(env, napi_reference_unref(env, refValue, &refCount)); |
| 123 | + return ToUInt32Value(env, refCount); |
| 124 | +} |
| 125 | + |
| 126 | +static napi_value DeleteRef(napi_env env, napi_callback_info info) { |
| 127 | + napi_ref refValue; |
| 128 | + NODE_API_CALL(env, GetRef(env, info, &refValue)); |
| 129 | + NODE_API_CALL(env, napi_delete_reference(env, refValue)); |
| 130 | + return NULL; |
| 131 | +} |
| 132 | + |
| 133 | +static napi_value AddFinalizer(napi_env env, napi_callback_info info) { |
| 134 | + napi_value obj; |
| 135 | + NODE_API_CALL(env, GetArgValue(env, info, &obj)); |
| 136 | + |
| 137 | + napi_valuetype valueType; |
| 138 | + NODE_API_CALL(env, napi_typeof(env, obj, &valueType)); |
| 139 | + NODE_API_ASSERT(env, valueType == napi_object, "Argument must be an object."); |
| 140 | + |
| 141 | + NODE_API_CALL(env, napi_add_finalizer(env, obj, NULL, &Finalize, NULL, NULL)); |
| 142 | + return NULL; |
| 143 | +} |
| 144 | + |
| 145 | +static napi_value GetFinalizeCount(napi_env env, napi_callback_info info) { |
| 146 | + return ToUInt32Value(env, finalizeCount); |
| 147 | +} |
| 148 | + |
| 149 | +static napi_value InitFinalizeCount(napi_env env, napi_callback_info info) { |
| 150 | + finalizeCount = 0; |
| 151 | + return NULL; |
| 152 | +} |
| 153 | + |
| 154 | +EXTERN_C_START |
| 155 | + |
| 156 | +NAPI_MODULE_INIT() { |
| 157 | + finalizeCount = 0; |
| 158 | + NODE_API_CALL(env, InitRefArray(env)); |
| 159 | + |
| 160 | + napi_property_descriptor properties[] = { |
| 161 | + DECLARE_NODE_API_PROPERTY("createExternal", CreateExternal), |
| 162 | + DECLARE_NODE_API_PROPERTY("createRef", CreateRef), |
| 163 | + DECLARE_NODE_API_PROPERTY("getRefValue", GetRefValue), |
| 164 | + DECLARE_NODE_API_PROPERTY("ref", Ref), |
| 165 | + DECLARE_NODE_API_PROPERTY("unref", Unref), |
| 166 | + DECLARE_NODE_API_PROPERTY("deleteRef", DeleteRef), |
| 167 | + DECLARE_NODE_API_PROPERTY("addFinalizer", AddFinalizer), |
| 168 | + DECLARE_NODE_API_PROPERTY("getFinalizeCount", GetFinalizeCount), |
| 169 | + DECLARE_NODE_API_PROPERTY("initFinalizeCount", InitFinalizeCount), |
| 170 | + }; |
| 171 | + |
| 172 | + NODE_API_CALL( |
| 173 | + env, |
| 174 | + napi_define_properties( |
| 175 | + env, exports, sizeof(properties) / sizeof(*properties), properties)); |
| 176 | + |
| 177 | + return exports; |
| 178 | +} |
| 179 | + |
| 180 | +EXTERN_C_END |
0 commit comments