xref: /third_party/node/src/async_wrap.h (revision 1cb0ef41)
1// Copyright Joyent, Inc. and other Node contributors.
2//
3// Permission is hereby granted, free of charge, to any person obtaining a
4// copy of this software and associated documentation files (the
5// "Software"), to deal in the Software without restriction, including
6// without limitation the rights to use, copy, modify, merge, publish,
7// distribute, sublicense, and/or sell copies of the Software, and to permit
8// persons to whom the Software is furnished to do so, subject to the
9// following conditions:
10//
11// The above copyright notice and this permission notice shall be included
12// in all copies or substantial portions of the Software.
13//
14// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
15// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
16// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
17// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
18// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
19// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
20// USE OR OTHER DEALINGS IN THE SOFTWARE.
21
22#ifndef SRC_ASYNC_WRAP_H_
23#define SRC_ASYNC_WRAP_H_
24
25#if defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS
26
27#include "base_object.h"
28#include "v8.h"
29
30#include <cstdint>
31
32namespace node {
33
34#define NODE_ASYNC_NON_CRYPTO_PROVIDER_TYPES(V)                               \
35  V(NONE)                                                                     \
36  V(DIRHANDLE)                                                                \
37  V(DNSCHANNEL)                                                               \
38  V(ELDHISTOGRAM)                                                             \
39  V(FILEHANDLE)                                                               \
40  V(FILEHANDLECLOSEREQ)                                                       \
41  V(FIXEDSIZEBLOBCOPY)                                                        \
42  V(FSEVENTWRAP)                                                              \
43  V(FSREQCALLBACK)                                                            \
44  V(FSREQPROMISE)                                                             \
45  V(GETADDRINFOREQWRAP)                                                       \
46  V(GETNAMEINFOREQWRAP)                                                       \
47  V(HEAPSNAPSHOT)                                                             \
48  V(HTTP2SESSION)                                                             \
49  V(HTTP2STREAM)                                                              \
50  V(HTTP2PING)                                                                \
51  V(HTTP2SETTINGS)                                                            \
52  V(HTTPINCOMINGMESSAGE)                                                      \
53  V(HTTPCLIENTREQUEST)                                                        \
54  V(JSSTREAM)                                                                 \
55  V(JSUDPWRAP)                                                                \
56  V(MESSAGEPORT)                                                              \
57  V(PIPECONNECTWRAP)                                                          \
58  V(PIPESERVERWRAP)                                                           \
59  V(PIPEWRAP)                                                                 \
60  V(PROCESSWRAP)                                                              \
61  V(PROMISE)                                                                  \
62  V(QUERYWRAP)                                                                \
63  V(QUIC_LOGSTREAM)                                                           \
64  V(QUIC_PACKET)                                                              \
65  V(SHUTDOWNWRAP)                                                             \
66  V(SIGNALWRAP)                                                               \
67  V(STATWATCHER)                                                              \
68  V(STREAMPIPE)                                                               \
69  V(TCPCONNECTWRAP)                                                           \
70  V(TCPSERVERWRAP)                                                            \
71  V(TCPWRAP)                                                                  \
72  V(TTYWRAP)                                                                  \
73  V(UDPSENDWRAP)                                                              \
74  V(UDPWRAP)                                                                  \
75  V(SIGINTWATCHDOG)                                                           \
76  V(WORKER)                                                                   \
77  V(WORKERHEAPSNAPSHOT)                                                       \
78  V(WRITEWRAP)                                                                \
79  V(ZLIB)
80
81#if HAVE_OPENSSL
82#define NODE_ASYNC_CRYPTO_PROVIDER_TYPES(V)                                   \
83  V(CHECKPRIMEREQUEST)                                                        \
84  V(PBKDF2REQUEST)                                                            \
85  V(KEYPAIRGENREQUEST)                                                        \
86  V(KEYGENREQUEST)                                                            \
87  V(KEYEXPORTREQUEST)                                                         \
88  V(CIPHERREQUEST)                                                            \
89  V(DERIVEBITSREQUEST)                                                        \
90  V(HASHREQUEST)                                                              \
91  V(RANDOMBYTESREQUEST)                                                       \
92  V(RANDOMPRIMEREQUEST)                                                       \
93  V(SCRYPTREQUEST)                                                            \
94  V(SIGNREQUEST)                                                              \
95  V(TLSWRAP)                                                                  \
96  V(VERIFYREQUEST)
97#else
98#define NODE_ASYNC_CRYPTO_PROVIDER_TYPES(V)
99#endif  // HAVE_OPENSSL
100
101#if HAVE_INSPECTOR
102#define NODE_ASYNC_INSPECTOR_PROVIDER_TYPES(V)                                \
103  V(INSPECTORJSBINDING)
104#else
105#define NODE_ASYNC_INSPECTOR_PROVIDER_TYPES(V)
106#endif  // HAVE_INSPECTOR
107
108#define NODE_ASYNC_PROVIDER_TYPES(V)                                          \
109  NODE_ASYNC_NON_CRYPTO_PROVIDER_TYPES(V)                                     \
110  NODE_ASYNC_CRYPTO_PROVIDER_TYPES(V)                                         \
111  NODE_ASYNC_INSPECTOR_PROVIDER_TYPES(V)
112
113class Environment;
114class DestroyParam;
115class ExternalReferenceRegistry;
116
117class AsyncWrap : public BaseObject {
118 public:
119  enum ProviderType {
120#define V(PROVIDER)                                                           \
121    PROVIDER_ ## PROVIDER,
122    NODE_ASYNC_PROVIDER_TYPES(V)
123#undef V
124    PROVIDERS_LENGTH,
125  };
126
127  AsyncWrap(Environment* env,
128            v8::Local<v8::Object> object,
129            ProviderType provider,
130            double execution_async_id = kInvalidAsyncId);
131
132  // This constructor creates a reusable instance where user is responsible
133  // to call set_provider_type() and AsyncReset() before use.
134  AsyncWrap(Environment* env, v8::Local<v8::Object> object);
135
136  ~AsyncWrap() override;
137
138  AsyncWrap() = delete;
139
140  static constexpr double kInvalidAsyncId = -1;
141
142  static v8::Local<v8::FunctionTemplate> GetConstructorTemplate(
143      Environment* env);
144
145  static void RegisterExternalReferences(ExternalReferenceRegistry* registry);
146  static void Initialize(v8::Local<v8::Object> target,
147                         v8::Local<v8::Value> unused,
148                         v8::Local<v8::Context> context,
149                         void* priv);
150
151  static void GetAsyncId(const v8::FunctionCallbackInfo<v8::Value>& args);
152  static void PushAsyncContext(const v8::FunctionCallbackInfo<v8::Value>& args);
153  static void PopAsyncContext(const v8::FunctionCallbackInfo<v8::Value>& args);
154  static void ExecutionAsyncResource(
155      const v8::FunctionCallbackInfo<v8::Value>& args);
156  static void ClearAsyncIdStack(
157      const v8::FunctionCallbackInfo<v8::Value>& args);
158  static void AsyncReset(const v8::FunctionCallbackInfo<v8::Value>& args);
159  static void GetProviderType(const v8::FunctionCallbackInfo<v8::Value>& args);
160  static void QueueDestroyAsyncId(
161    const v8::FunctionCallbackInfo<v8::Value>& args);
162  static void SetCallbackTrampoline(
163    const v8::FunctionCallbackInfo<v8::Value>& args);
164
165  static void EmitAsyncInit(Environment* env,
166                            v8::Local<v8::Object> object,
167                            v8::Local<v8::String> type,
168                            double async_id,
169                            double trigger_async_id);
170
171  static void EmitDestroy(Environment* env, double async_id);
172  static void EmitBefore(Environment* env, double async_id);
173  static void EmitAfter(Environment* env, double async_id);
174  static void EmitPromiseResolve(Environment* env, double async_id);
175
176  void EmitDestroy(bool from_gc = false);
177
178  void EmitTraceEventBefore();
179  static void EmitTraceEventAfter(ProviderType type, double async_id);
180  void EmitTraceEventDestroy();
181
182  static void DestroyAsyncIdsCallback(Environment* env);
183
184  inline ProviderType provider_type() const;
185  inline ProviderType set_provider_type(ProviderType provider);
186
187  inline double get_async_id() const;
188  inline double get_trigger_async_id() const;
189
190  void AsyncReset(v8::Local<v8::Object> resource,
191                  double execution_async_id = kInvalidAsyncId,
192                  bool silent = false);
193
194  // Only call these within a valid HandleScope.
195  v8::MaybeLocal<v8::Value> MakeCallback(const v8::Local<v8::Function> cb,
196                                         int argc,
197                                         v8::Local<v8::Value>* argv);
198  inline v8::MaybeLocal<v8::Value> MakeCallback(
199      const v8::Local<v8::Symbol> symbol,
200      int argc,
201      v8::Local<v8::Value>* argv);
202  inline v8::MaybeLocal<v8::Value> MakeCallback(
203      const v8::Local<v8::String> symbol,
204      int argc,
205      v8::Local<v8::Value>* argv);
206  inline v8::MaybeLocal<v8::Value> MakeCallback(
207      const v8::Local<v8::Name> symbol,
208      int argc,
209      v8::Local<v8::Value>* argv);
210
211  virtual std::string diagnostic_name() const;
212  const char* MemoryInfoName() const override;
213
214  static void WeakCallback(const v8::WeakCallbackInfo<DestroyParam> &info);
215
216  // Returns the object that 'owns' an async wrap. For example, for a
217  // TCP connection handle, this is the corresponding net.Socket.
218  v8::Local<v8::Object> GetOwner();
219  static v8::Local<v8::Object> GetOwner(Environment* env,
220                                        v8::Local<v8::Object> obj);
221
222  bool IsDoneInitializing() const override;
223
224 private:
225  friend class PromiseWrap;
226
227  AsyncWrap(Environment* env,
228            v8::Local<v8::Object> promise,
229            ProviderType provider,
230            double execution_async_id,
231            bool silent);
232  AsyncWrap(Environment* env,
233            v8::Local<v8::Object> promise,
234            ProviderType provider,
235            double execution_async_id,
236            double trigger_async_id);
237  ProviderType provider_type_ = PROVIDER_NONE;
238  bool init_hook_ran_ = false;
239  // Because the values may be Reset(), cannot be made const.
240  double async_id_ = kInvalidAsyncId;
241  double trigger_async_id_ = kInvalidAsyncId;
242};
243
244}  // namespace node
245
246#endif  // defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS
247
248#endif  // SRC_ASYNC_WRAP_H_
249