|
| 1 | +#include <js_native_api.h> |
| 2 | +#include <stdlib.h> |
| 3 | +#include <string.h> |
| 4 | +#include "../common.h" |
| 5 | +#include "../entry_point.h" |
| 6 | + |
| 7 | +static napi_value Multiply(napi_env env, napi_callback_info info) { |
| 8 | + size_t argc = 2; |
| 9 | + napi_value args[2]; |
| 10 | + NODE_API_CALL(env, napi_get_cb_info(env, info, &argc, args, NULL, NULL)); |
| 11 | + |
| 12 | + NODE_API_ASSERT(env, argc == 2, "Wrong number of arguments"); |
| 13 | + |
| 14 | + napi_valuetype valuetype0; |
| 15 | + NODE_API_CALL(env, napi_typeof(env, args[0], &valuetype0)); |
| 16 | + |
| 17 | + NODE_API_ASSERT(env, valuetype0 == napi_object, |
| 18 | + "Wrong type of arguments. Expects a typed array as first argument."); |
| 19 | + |
| 20 | + napi_value input_array = args[0]; |
| 21 | + bool is_typedarray; |
| 22 | + NODE_API_CALL(env, napi_is_typedarray(env, input_array, &is_typedarray)); |
| 23 | + |
| 24 | + NODE_API_ASSERT(env, is_typedarray, |
| 25 | + "Wrong type of arguments. Expects a typed array as first argument."); |
| 26 | + |
| 27 | + napi_valuetype valuetype1; |
| 28 | + NODE_API_CALL(env, napi_typeof(env, args[1], &valuetype1)); |
| 29 | + |
| 30 | + NODE_API_ASSERT(env, valuetype1 == napi_number, |
| 31 | + "Wrong type of arguments. Expects a number as second argument."); |
| 32 | + |
| 33 | + double multiplier; |
| 34 | + NODE_API_CALL(env, napi_get_value_double(env, args[1], &multiplier)); |
| 35 | + |
| 36 | + napi_typedarray_type type; |
| 37 | + napi_value input_buffer; |
| 38 | + size_t byte_offset; |
| 39 | + size_t i, length; |
| 40 | + NODE_API_CALL(env, napi_get_typedarray_info( |
| 41 | + env, input_array, &type, &length, NULL, &input_buffer, &byte_offset)); |
| 42 | + |
| 43 | + void* data; |
| 44 | + size_t byte_length; |
| 45 | + NODE_API_CALL(env, napi_get_arraybuffer_info( |
| 46 | + env, input_buffer, &data, &byte_length)); |
| 47 | + |
| 48 | + napi_value output_buffer; |
| 49 | + void* output_ptr = NULL; |
| 50 | + NODE_API_CALL(env, napi_create_arraybuffer( |
| 51 | + env, byte_length, &output_ptr, &output_buffer)); |
| 52 | + |
| 53 | + napi_value output_array; |
| 54 | + NODE_API_CALL(env, napi_create_typedarray( |
| 55 | + env, type, length, output_buffer, byte_offset, &output_array)); |
| 56 | + |
| 57 | + if (type == napi_uint8_array) { |
| 58 | + uint8_t* input_bytes = (uint8_t*)(data) + byte_offset; |
| 59 | + uint8_t* output_bytes = (uint8_t*)(output_ptr); |
| 60 | + for (i = 0; i < length; i++) { |
| 61 | + output_bytes[i] = (uint8_t)(input_bytes[i] * multiplier); |
| 62 | + } |
| 63 | + } else if (type == napi_float64_array) { |
| 64 | + double* input_doubles = (double*)((uint8_t*)(data) + byte_offset); |
| 65 | + double* output_doubles = (double*)(output_ptr); |
| 66 | + for (i = 0; i < length; i++) { |
| 67 | + output_doubles[i] = input_doubles[i] * multiplier; |
| 68 | + } |
| 69 | + } else { |
| 70 | + napi_throw_error(env, NULL, |
| 71 | + "Typed array was of a type not expected by test."); |
| 72 | + return NULL; |
| 73 | + } |
| 74 | + |
| 75 | + return output_array; |
| 76 | +} |
| 77 | + |
| 78 | +static void FinalizeCallback(node_api_basic_env env, |
| 79 | + void* finalize_data, |
| 80 | + void* finalize_hint) { |
| 81 | + free(finalize_data); |
| 82 | +} |
| 83 | + |
| 84 | +static napi_value External(napi_env env, napi_callback_info info) { |
| 85 | + const uint8_t nElem = 3; |
| 86 | + int8_t* externalData = malloc(nElem*sizeof(int8_t)); |
| 87 | + externalData[0] = 0; |
| 88 | + externalData[1] = 1; |
| 89 | + externalData[2] = 2; |
| 90 | + |
| 91 | + napi_value output_buffer; |
| 92 | + NODE_API_CALL(env, napi_create_external_arraybuffer( |
| 93 | + env, |
| 94 | + externalData, |
| 95 | + nElem*sizeof(int8_t), |
| 96 | + FinalizeCallback, |
| 97 | + NULL, // finalize_hint |
| 98 | + &output_buffer)); |
| 99 | + |
| 100 | + napi_value output_array; |
| 101 | + NODE_API_CALL(env, napi_create_typedarray(env, |
| 102 | + napi_int8_array, |
| 103 | + nElem, |
| 104 | + output_buffer, |
| 105 | + 0, |
| 106 | + &output_array)); |
| 107 | + |
| 108 | + return output_array; |
| 109 | +} |
| 110 | + |
| 111 | + |
| 112 | +static napi_value NullArrayBuffer(napi_env env, napi_callback_info info) { |
| 113 | + static void* data = NULL; |
| 114 | + napi_value arraybuffer; |
| 115 | + NODE_API_CALL(env, |
| 116 | + napi_create_external_arraybuffer(env, data, 0, NULL, NULL, &arraybuffer)); |
| 117 | + return arraybuffer; |
| 118 | +} |
| 119 | + |
| 120 | +static napi_value CreateTypedArray(napi_env env, napi_callback_info info) { |
| 121 | + size_t argc = 4; |
| 122 | + napi_value args[4]; |
| 123 | + NODE_API_CALL(env, napi_get_cb_info(env, info, &argc, args, NULL, NULL)); |
| 124 | + |
| 125 | + NODE_API_ASSERT(env, argc == 2 || argc == 4, "Wrong number of arguments"); |
| 126 | + |
| 127 | + napi_value input_array = args[0]; |
| 128 | + napi_valuetype valuetype0; |
| 129 | + NODE_API_CALL(env, napi_typeof(env, input_array, &valuetype0)); |
| 130 | + |
| 131 | + NODE_API_ASSERT(env, valuetype0 == napi_object, |
| 132 | + "Wrong type of arguments. Expects a typed array as first argument."); |
| 133 | + |
| 134 | + bool is_typedarray; |
| 135 | + NODE_API_CALL(env, napi_is_typedarray(env, input_array, &is_typedarray)); |
| 136 | + |
| 137 | + NODE_API_ASSERT(env, is_typedarray, |
| 138 | + "Wrong type of arguments. Expects a typed array as first argument."); |
| 139 | + |
| 140 | + napi_valuetype valuetype1; |
| 141 | + napi_value input_buffer = args[1]; |
| 142 | + NODE_API_CALL(env, napi_typeof(env, input_buffer, &valuetype1)); |
| 143 | + |
| 144 | + NODE_API_ASSERT(env, valuetype1 == napi_object, |
| 145 | + "Wrong type of arguments. Expects an array buffer as second argument."); |
| 146 | + |
| 147 | + bool is_arraybuffer; |
| 148 | + NODE_API_CALL(env, napi_is_arraybuffer(env, input_buffer, &is_arraybuffer)); |
| 149 | + |
| 150 | + NODE_API_ASSERT(env, is_arraybuffer, |
| 151 | + "Wrong type of arguments. Expects an array buffer as second argument."); |
| 152 | + |
| 153 | + napi_typedarray_type type; |
| 154 | + napi_value in_array_buffer; |
| 155 | + size_t byte_offset; |
| 156 | + size_t length; |
| 157 | + NODE_API_CALL(env, napi_get_typedarray_info( |
| 158 | + env, input_array, &type, &length, NULL, &in_array_buffer, &byte_offset)); |
| 159 | + |
| 160 | + if (argc == 4) { |
| 161 | + napi_valuetype valuetype2; |
| 162 | + NODE_API_CALL(env, napi_typeof(env, args[2], &valuetype2)); |
| 163 | + |
| 164 | + NODE_API_ASSERT(env, valuetype2 == napi_number, |
| 165 | + "Wrong type of arguments. Expects a number as third argument."); |
| 166 | + |
| 167 | + uint32_t uint32_length; |
| 168 | + NODE_API_CALL(env, napi_get_value_uint32(env, args[2], &uint32_length)); |
| 169 | + length = uint32_length; |
| 170 | + |
| 171 | + napi_valuetype valuetype3; |
| 172 | + NODE_API_CALL(env, napi_typeof(env, args[3], &valuetype3)); |
| 173 | + |
| 174 | + NODE_API_ASSERT(env, valuetype3 == napi_number, |
| 175 | + "Wrong type of arguments. Expects a number as third argument."); |
| 176 | + |
| 177 | + uint32_t uint32_byte_offset; |
| 178 | + NODE_API_CALL(env, napi_get_value_uint32(env, args[3], &uint32_byte_offset)); |
| 179 | + byte_offset = uint32_byte_offset; |
| 180 | + } |
| 181 | + |
| 182 | + napi_value output_array; |
| 183 | + NODE_API_CALL(env, napi_create_typedarray( |
| 184 | + env, type, length, input_buffer, byte_offset, &output_array)); |
| 185 | + |
| 186 | + return output_array; |
| 187 | +} |
| 188 | + |
| 189 | +static napi_value Detach(napi_env env, napi_callback_info info) { |
| 190 | + size_t argc = 1; |
| 191 | + napi_value args[1]; |
| 192 | + NODE_API_CALL(env, napi_get_cb_info(env, info, &argc, args, NULL, NULL)); |
| 193 | + NODE_API_ASSERT(env, argc == 1, "Wrong number of arguments."); |
| 194 | + |
| 195 | + bool is_typedarray; |
| 196 | + NODE_API_CALL(env, napi_is_typedarray(env, args[0], &is_typedarray)); |
| 197 | + NODE_API_ASSERT( |
| 198 | + env, is_typedarray, |
| 199 | + "Wrong type of arguments. Expects a typedarray as first argument."); |
| 200 | + |
| 201 | + napi_value arraybuffer; |
| 202 | + NODE_API_CALL(env, |
| 203 | + napi_get_typedarray_info( |
| 204 | + env, args[0], NULL, NULL, NULL, &arraybuffer, NULL)); |
| 205 | + NODE_API_CALL(env, napi_detach_arraybuffer(env, arraybuffer)); |
| 206 | + |
| 207 | + return NULL; |
| 208 | +} |
| 209 | + |
| 210 | +static napi_value IsDetached(napi_env env, napi_callback_info info) { |
| 211 | + size_t argc = 1; |
| 212 | + napi_value args[1]; |
| 213 | + NODE_API_CALL(env, napi_get_cb_info(env, info, &argc, args, NULL, NULL)); |
| 214 | + NODE_API_ASSERT(env, argc == 1, "Wrong number of arguments."); |
| 215 | + |
| 216 | + napi_value array_buffer = args[0]; |
| 217 | + bool is_arraybuffer; |
| 218 | + NODE_API_CALL(env, napi_is_arraybuffer(env, array_buffer, &is_arraybuffer)); |
| 219 | + NODE_API_ASSERT(env, is_arraybuffer, |
| 220 | + "Wrong type of arguments. Expects an array buffer as first argument."); |
| 221 | + |
| 222 | + bool is_detached; |
| 223 | + NODE_API_CALL(env, |
| 224 | + napi_is_detached_arraybuffer(env, array_buffer, &is_detached)); |
| 225 | + |
| 226 | + napi_value result; |
| 227 | + NODE_API_CALL(env, napi_get_boolean(env, is_detached, &result)); |
| 228 | + |
| 229 | + return result; |
| 230 | +} |
| 231 | + |
| 232 | +EXTERN_C_START |
| 233 | +napi_value Init(napi_env env, napi_value exports) { |
| 234 | + napi_property_descriptor descriptors[] = { |
| 235 | + DECLARE_NODE_API_PROPERTY("Multiply", Multiply), |
| 236 | + DECLARE_NODE_API_PROPERTY("External", External), |
| 237 | + DECLARE_NODE_API_PROPERTY("NullArrayBuffer", NullArrayBuffer), |
| 238 | + DECLARE_NODE_API_PROPERTY("CreateTypedArray", CreateTypedArray), |
| 239 | + DECLARE_NODE_API_PROPERTY("Detach", Detach), |
| 240 | + DECLARE_NODE_API_PROPERTY("IsDetached", IsDetached), |
| 241 | + }; |
| 242 | + |
| 243 | + NODE_API_CALL(env, napi_define_properties( |
| 244 | + env, exports, sizeof(descriptors) / sizeof(*descriptors), descriptors)); |
| 245 | + |
| 246 | + return exports; |
| 247 | +} |
| 248 | +EXTERN_C_END |
0 commit comments