emnapi/packages/test/sharedarraybuffer/binding.c at 437f9a8ba7bb53ae69b158552c24a4f5a6e53499 · toyobayashi/emnapi · GitHub

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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
#define NAPI_EXPERIMENTAL
#include <js_native_api.h>
#include "../common.h"
#include "../entry_point.h"

#ifdef __wasm__
#include "emnapi.h"
#endif

static napi_value TestIsSharedArrayBuffer(napi_env env,
                                          napi_callback_info info) {
  size_t argc = 1;
  napi_value args[1];
  NODE_API_CALL(env, napi_get_cb_info(env, info, &argc, args, NULL, NULL));

  NODE_API_ASSERT(env, argc >= 1, "Wrong number of arguments");

  bool is_sharedarraybuffer;
  NODE_API_CALL(
      env, node_api_is_sharedarraybuffer(env, args[0], &is_sharedarraybuffer));

  napi_value ret;
  NODE_API_CALL(env, napi_get_boolean(env, is_sharedarraybuffer, &ret));

  return ret;
}

static napi_value TestCreateSharedArrayBuffer(napi_env env,
                                              napi_callback_info info) {
  size_t argc = 1;
  napi_value args[1];
  NODE_API_CALL(env, napi_get_cb_info(env, info, &argc, args, NULL, NULL));

  NODE_API_ASSERT(env, argc >= 1, "Wrong number of arguments");

  napi_valuetype valuetype0;
  NODE_API_CALL(env, napi_typeof(env, args[0], &valuetype0));

  NODE_API_ASSERT(
      env,
      valuetype0 == napi_number,
      "Wrong type of arguments. Expects a number as first argument.");

  int32_t byte_length;
  NODE_API_CALL(env, napi_get_value_int32(env, args[0], &byte_length));

  NODE_API_ASSERT(env,
                  byte_length >= 0,
                  "Invalid byte length. Expects a non-negative integer.");

  napi_value ret;
  void* data;
  NODE_API_CALL(
      env, node_api_create_sharedarraybuffer(env, byte_length, &data, &ret));

  return ret;
}

static napi_value TestGetSharedArrayBufferInfo(napi_env env,
                                               napi_callback_info info) {
  size_t argc = 1;
  napi_value args[1];
  NODE_API_CALL(env, napi_get_cb_info(env, info, &argc, args, NULL, NULL));

  NODE_API_ASSERT(env, argc >= 1, "Wrong number of arguments");

  void* data;
  size_t byte_length;
  NODE_API_CALL(env,
                napi_get_arraybuffer_info(env, args[0], &data, &byte_length));

  napi_value ret;
  NODE_API_CALL(env, napi_create_uint32(env, byte_length, &ret));

  return ret;
}

static void WriteTestDataToBuffer(void* data, size_t byte_length) {
  if (byte_length > 0 && data != NULL) {
    uint8_t* bytes = (uint8_t*)data;
    for (size_t i = 0; i < byte_length; i++) {
      bytes[i] = i % 256;
    }
  }
}

static napi_value TestSharedArrayBufferData(napi_env env,
                                            napi_callback_info info) {
  size_t argc = 1;
  napi_value args[1];
  NODE_API_CALL(env, napi_get_cb_info(env, info, &argc, args, NULL, NULL));

  NODE_API_ASSERT(env, argc >= 1, "Wrong number of arguments");

  void* data;
  size_t byte_length;
  NODE_API_CALL(env,
                napi_get_arraybuffer_info(env, args[0], &data, &byte_length));

  WriteTestDataToBuffer(data, byte_length);
#ifdef __wasm__
  emnapi_sync_memory(env, false, &args[0], 0, NAPI_AUTO_LENGTH);
#endif

  // Return the same data pointer validity
  bool data_valid = (data != NULL) && (byte_length > 0);

  napi_value ret;
  NODE_API_CALL(env, napi_get_boolean(env, data_valid, &ret));

  return ret;
}

EXTERN_C_START
napi_value Init(napi_env env, napi_value exports) {
  napi_property_descriptor descriptors[] = {
      DECLARE_NODE_API_PROPERTY("TestIsSharedArrayBuffer",
                                TestIsSharedArrayBuffer),
      DECLARE_NODE_API_PROPERTY("TestCreateSharedArrayBuffer",
                                TestCreateSharedArrayBuffer),
      DECLARE_NODE_API_PROPERTY("TestGetSharedArrayBufferInfo",
                                TestGetSharedArrayBufferInfo),
      DECLARE_NODE_API_PROPERTY("TestSharedArrayBufferData",
                                TestSharedArrayBufferData),
  };

  NODE_API_CALL(
      env,
      napi_define_properties(env,
                             exports,
                             sizeof(descriptors) / sizeof(*descriptors),
                             descriptors));

  return exports;
}
EXTERN_C_END