Local<Function> fn = env->async_hooks_destroy_function();

TryCatchScope try_catch(env,

TryCatchScope::CatchMode::kFatalRethrowStackOverflow);

do {

std::vector<double> destroy_async_id_list;

destroy_async_id_list.swap(*env->destroy_async_id_list());

if (!env->can_call_into_js()) return;

for (auto async_id : destroy_async_id_list) {

// Want each callback to be cleaned up after itself, instead of cleaning

// them all up after the while() loop completes.

HandleScope scope(env->isolate());

Local<Value> async_id_value = Number::New(env->isolate(), async_id);

MaybeLocal<Value> ret = fn->Call(

env->context(), Undefined(env->isolate()), 1, &async_id_value);

if (ret.IsEmpty())

return;

}

} while (!env->destroy_async_id_list()->empty());

Local<Function> fn) {

AsyncHooks* async_hooks = env->async_hooks();

if (async_hooks->fields()[type] == 0 || !env->can_call_into_js())

return;

HandleScope handle_scope(env->isolate());

Local<Value> async_id_value = Number::New(env->isolate(), async_id);

TryCatchScope try_catch(env,

TryCatchScope::CatchMode::kFatalRethrowStackOverflow);

USE(fn->Call(env->context(), Undefined(env->isolate()), 1, &async_id_value));

if (*TRACE_EVENT_API_GET_CATEGORY_GROUP_ENABLED(

TRACING_CATEGORY_NODE1(async_hooks))) {

tracing::AsyncWrapArgs data(env()->execution_async_id(),

get_trigger_async_id());

TRACE_EVENT_NESTABLE_ASYNC_BEGIN1(TRACING_CATEGORY_NODE1(async_hooks),

provider_names[provider_type()],

static_cast<int64_t>(get_async_id()),

"data",

tracing::CastTracedValue(data));

}

switch (provider_type()) {

#define V(PROVIDER) \

case PROVIDER_ ## PROVIDER: \

TRACE_EVENT_NESTABLE_ASYNC_BEGIN0( \

TRACING_CATEGORY_NODE1(async_hooks), \

#PROVIDER "_CALLBACK", static_cast<int64_t>(get_async_id())); \

break;

NODE_ASYNC_PROVIDER_TYPES(V)

#undef V

default:

UNREACHABLE();

}

switch (type) {

#define V(PROVIDER) \

case PROVIDER_ ## PROVIDER: \

TRACE_EVENT_NESTABLE_ASYNC_END0( \

TRACING_CATEGORY_NODE1(async_hooks), \

#PROVIDER "_CALLBACK", static_cast<int64_t>(async_id)); \

break;

NODE_ASYNC_PROVIDER_TYPES(V)

#undef V

default:

UNREACHABLE();

}

// If the user's callback failed then the after() hooks will be called at the

// end of _fatalException().

Emit(env, async_id, AsyncHooks::kAfter,

env->async_hooks_after_function());

Environment* env = Environment::GetCurrent(args);

CHECK(args[0]->IsObject());

// All of init, before, after, destroy, and promise_resolve are supplied by

// async_hooks internally, so this should only ever be called once. At which

// time all the functions should be set. Detect this by checking if

// init !IsEmpty().

CHECK(env->async_hooks_init_function().IsEmpty());

Local<Object> fn_obj = args[0].As<Object>();

#define SET_HOOK_FN(name) \

do { \

Local<Value> v = \

fn_obj->Get(env->context(), \

FIXED_ONE_BYTE_STRING(env->isolate(), #name)) \

.ToLocalChecked(); \

CHECK(v->IsFunction()); \

env->set_async_hooks_##name##_function(v.As<Function>()); \

} while (0)

SET_HOOK_FN(init);

SET_HOOK_FN(before);

SET_HOOK_FN(after);

SET_HOOK_FN(destroy);

SET_HOOK_FN(promise_resolve);

#undef SET_HOOK_FN

Environment* env = Environment::GetCurrent(args);

env->ResetPromiseHooks(

args[0]->IsFunction() ? args[0].As<Function>() : Local<Function>(),

args[1]->IsFunction() ? args[1].As<Function>() : Local<Function>(),

args[2]->IsFunction() ? args[2].As<Function>() : Local<Function>(),

args[3]->IsFunction() ? args[3].As<Function>() : Local<Function>());

Environment* env = Environment::GetCurrent(args);

args.GetReturnValue().Set(

env->async_hooks()->GetPromiseHooks(args.GetIsolate()));

public:

double asyncId;

Environment* env;

Global<Object> target;

Global<Object> propBag;

HandleScope scope(info.GetIsolate());

std::unique_ptr<DestroyParam> p{info.GetParameter()};

Local<Object> prop_bag = PersistentToLocal::Default(info.GetIsolate(),

p->propBag);

Local<Value> val;

p->env->RemoveCleanupHook(DestroyParamCleanupHook, p.get());

if (!prop_bag.IsEmpty() &&

!prop_bag->Get(p->env->context(), p->env->destroyed_string())

.ToLocal(&val)) {

return;

}

if (val.IsEmpty() || val->IsFalse()) {

AsyncWrap::EmitDestroy(p->env, p->asyncId);

}

// unique_ptr goes out of scope here and pointer is deleted.

CHECK(args[0]->IsObject());

CHECK(args[1]->IsNumber());

CHECK(args.Length() == 2 || args[2]->IsObject());

Isolate* isolate = args.GetIsolate();

DestroyParam* p = new DestroyParam();

p->asyncId = args[1].As<Number>()->Value();

p->env = Environment::GetCurrent(args);

p->target.Reset(isolate, args[0].As<Object>());

if (args.Length() > 2) {

p->propBag.Reset(isolate, args[2].As<Object>());

}

p->target.SetWeak(p, AsyncWrap::WeakCallback, WeakCallbackType::kParameter);

p->env->AddCleanupHook(DestroyParamCleanupHook, p);

AsyncWrap* wrap;

args.GetReturnValue().Set(kInvalidAsyncId);

ASSIGN_OR_RETURN_UNWRAP(&wrap, args.This());

args.GetReturnValue().Set(wrap->get_async_id());

Environment* env = Environment::GetCurrent(args);

// No need for CHECK(IsNumber()) on args because if FromJust() doesn't fail

// then the checks in push_async_ids() and pop_async_id() will.

double async_id = args[0]->NumberValue(env->context()).FromJust();

double trigger_async_id = args[1]->NumberValue(env->context()).FromJust();

env->async_hooks()->push_async_context(async_id, trigger_async_id, {});

Environment* env = Environment::GetCurrent(args);

double async_id = args[0]->NumberValue(env->context()).FromJust();

args.GetReturnValue().Set(env->async_hooks()->pop_async_context(async_id));

const FunctionCallbackInfo<Value>& args) {

Environment* env = Environment::GetCurrent(args);

uint32_t index;

if (!args[0]->Uint32Value(env->context()).To(&index)) return;

args.GetReturnValue().Set(

env->async_hooks()->native_execution_async_resource(index));

CHECK(args[0]->IsObject());

AsyncWrap* wrap;

ASSIGN_OR_RETURN_UNWRAP(&wrap, args.This());

Local<Object> resource = args[0].As<Object>();

double execution_async_id =

args[1]->IsNumber() ? args[1].As<Number>()->Value() : kInvalidAsyncId;

wrap->AsyncReset(resource, execution_async_id);

AsyncWrap* wrap;

ASSIGN_OR_RETURN_UNWRAP(&wrap, args.This());

args.GetReturnValue().Set(wrap->context_frame());

AsyncWrap* wrap;

args.GetReturnValue().Set(AsyncWrap::PROVIDER_NONE);

ASSIGN_OR_RETURN_UNWRAP(&wrap, args.This());

args.GetReturnValue().Set(wrap->provider_type());

AsyncWrap::EmitDestroy(env(), async_id_);

// Ensure no double destroy is emitted via AsyncReset().

async_id_ = kInvalidAsyncId;

if (!from_gc) {

if (!persistent().IsEmpty()) {

HandleScope handle_scope(env()->isolate());

USE(object()->Set(env()->context(), env()->resource_symbol(), object()));

}

set_context_frame({});

}

CHECK(args[0]->IsNumber());

AsyncWrap::EmitDestroy(

Environment::GetCurrent(args),

args[0].As<Number>()->Value());

Environment* env = Environment::GetCurrent(args);

if (args[0]->IsFunction()) {

env->set_async_hooks_callback_trampoline(args[0].As<Function>());

} else {

env->set_async_hooks_callback_trampoline(Local<Function>());

}

IsolateData* isolate_data) {

Local<FunctionTemplate> tmpl = isolate_data->async_wrap_ctor_template();

if (tmpl.IsEmpty()) {

Isolate* isolate = isolate_data->isolate();

tmpl = NewFunctionTemplate(isolate, nullptr);

tmpl->SetClassName(

FIXED_ONE_BYTE_STRING(isolate_data->isolate(), "AsyncWrap"));

SetProtoMethod(isolate, tmpl, "getAsyncId", AsyncWrap::GetAsyncId);

SetProtoMethod(isolate, tmpl, "asyncReset", AsyncWrap::AsyncReset);

SetProtoMethod(isolate,

tmpl,

"getAsyncContextFrameForDebuggingOnly",

AsyncWrap::GetAsyncContextFrame);

SetProtoMethod(

isolate, tmpl, "getProviderType", AsyncWrap::GetProviderType);

isolate_data->set_async_wrap_ctor_template(tmpl);

}

return tmpl;

Local<ObjectTemplate> target) {

Isolate* isolate = isolate_data->isolate();

SetMethod(isolate, target, "setupHooks", SetupHooks);

SetMethod(isolate, target, "setCallbackTrampoline", SetCallbackTrampoline);

SetMethod(isolate, target, "pushAsyncContext", PushAsyncContext);

SetMethod(isolate, target, "popAsyncContext", PopAsyncContext);

SetMethod(isolate, target, "executionAsyncResource", ExecutionAsyncResource);

SetMethod(isolate, target, "clearAsyncIdStack", ClearAsyncIdStack);

SetMethod(isolate, target, "queueDestroyAsyncId", QueueDestroyAsyncId);

SetMethod(isolate, target, "setPromiseHooks", SetPromiseHooks);

SetMethod(isolate, target, "getPromiseHooks", GetPromiseHooks);

SetMethod(isolate, target, "registerDestroyHook", RegisterDestroyHook);

AsyncWrap::GetConstructorTemplate(isolate_data);

Local<Value> unused,

Local<Context> context,

void* priv) {

Realm* realm = Realm::GetCurrent(context);

Environment* env = realm->env();

Isolate* isolate = realm->isolate();

HandleScope scope(isolate);

PropertyAttribute ReadOnlyDontDelete =

static_cast<PropertyAttribute>(ReadOnly | DontDelete);

#define FORCE_SET_TARGET_FIELD(obj, str, field) \

(obj)->DefineOwnProperty(context, \

FIXED_ONE_BYTE_STRING(isolate, str), \

field, \

ReadOnlyDontDelete).FromJust()

// Attach the uint32_t[] where each slot contains the count of the number of

// callbacks waiting to be called on a particular event. It can then be

// incremented/decremented from JS quickly to communicate to C++ if there are

// any callbacks waiting to be called.

FORCE_SET_TARGET_FIELD(target,

"async_hook_fields",

env->async_hooks()->fields().GetJSArray());

// The following v8::Float64Array has 5 fields. These fields are shared in

// this way to allow JS and C++ to read/write each value as quickly as

// possible. The fields are represented as follows:

//

// kAsyncIdCounter: Maintains the state of the next unique id to be assigned.

//

// kDefaultTriggerAsyncId: Write the id of the resource responsible for a

// handle's creation just before calling the new handle's constructor.

// After the new handle is constructed kDefaultTriggerAsyncId is set back

// to kInvalidAsyncId.

FORCE_SET_TARGET_FIELD(target,

"async_id_fields",

env->async_hooks()->async_id_fields().GetJSArray());

FORCE_SET_TARGET_FIELD(target,

"execution_async_resources",

env->async_hooks()->js_execution_async_resources());

target->Set(context,

env->async_ids_stack_string(),

env->async_hooks()->async_ids_stack().GetJSArray()).Check();

Local<Object> constants = Object::New(isolate);

#define SET_HOOKS_CONSTANT(name) \

FORCE_SET_TARGET_FIELD( \

constants, #name, Integer::New(isolate, AsyncHooks::name))

SET_HOOKS_CONSTANT(kInit);

SET_HOOKS_CONSTANT(kBefore);

SET_HOOKS_CONSTANT(kAfter);

SET_HOOKS_CONSTANT(kDestroy);

SET_HOOKS_CONSTANT(kPromiseResolve);

SET_HOOKS_CONSTANT(kTotals);

SET_HOOKS_CONSTANT(kCheck);

SET_HOOKS_CONSTANT(kExecutionAsyncId);

SET_HOOKS_CONSTANT(kTriggerAsyncId);

SET_HOOKS_CONSTANT(kAsyncIdCounter);

SET_HOOKS_CONSTANT(kDefaultTriggerAsyncId);

SET_HOOKS_CONSTANT(kUsesExecutionAsyncResource);

SET_HOOKS_CONSTANT(kStackLength);

#undef SET_HOOKS_CONSTANT

FORCE_SET_TARGET_FIELD(target, "constants", constants);

Local<Object> async_providers = Object::New(isolate);

#define V(p) \

FORCE_SET_TARGET_FIELD( \

async_providers, #p, Integer::New(isolate, AsyncWrap::PROVIDER_ ## p));

NODE_ASYNC_PROVIDER_TYPES(V)

#undef V

FORCE_SET_TARGET_FIELD(target, "Providers", async_providers);

#undef FORCE_SET_TARGET_FIELD

// TODO(legendecas): async hook functions are not realm-aware yet.

// This simply avoid overriding principal realm's functions when a

// ShadowRealm initializes the binding.

realm->set_async_hooks_init_function(Local<Function>());

realm->set_async_hooks_before_function(Local<Function>());

realm->set_async_hooks_after_function(Local<Function>());

realm->set_async_hooks_destroy_function(Local<Function>());

realm->set_async_hooks_promise_resolve_function(Local<Function>());

realm->set_async_hooks_callback_trampoline(Local<Function>());

realm->set_async_hooks_binding(target);

ExternalReferenceRegistry* registry) {

registry->Register(SetupHooks);

registry->Register(SetCallbackTrampoline);

registry->Register(PushAsyncContext);

registry->Register(PopAsyncContext);

registry->Register(ExecutionAsyncResource);

registry->Register(ClearAsyncIdStack);

registry->Register(QueueDestroyAsyncId);

registry->Register(SetPromiseHooks);

registry->Register(GetPromiseHooks);

registry->Register(RegisterDestroyHook);

registry->Register(AsyncWrap::GetAsyncId);

registry->Register(AsyncWrap::AsyncReset);

registry->Register(AsyncWrap::GetAsyncContextFrame);

registry->Register(AsyncWrap::GetProviderType);

Local<Object> object,

ProviderType provider,

double execution_async_id)

: AsyncWrap(env, object) {

CHECK_NE(provider, PROVIDER_NONE);

provider_type_ = provider;

// Use AsyncReset() call to execute the init() callbacks.

AsyncReset(object, execution_async_id);

init_hook_ran_ = true;

switch (provider_type()) {

#define V(PROVIDER) \

case PROVIDER_ ## PROVIDER: \

TRACE_EVENT_NESTABLE_ASYNC_END0( \

TRACING_CATEGORY_NODE1(async_hooks), \

#PROVIDER, static_cast<int64_t>(get_async_id())); \

break;

NODE_ASYNC_PROVIDER_TYPES(V)

#undef V

default:

UNREACHABLE();

}

if (env->async_hooks()->fields()[AsyncHooks::kDestroy] == 0 ||

!env->can_call_into_js()) {

return;

}

if (env->destroy_async_id_list()->empty()) {

env->SetImmediate(&DestroyAsyncIdsCallback, CallbackFlags::kUnrefed);

}

// If the list gets very large empty it faster using a Microtask.

// Microtasks can't be added in GC context therefore we use an

// interrupt to get this Microtask scheduled as fast as possible.

if (env->destroy_async_id_list()->size() == 16384) {

env->RequestInterrupt([](Environment* env) {

env->context()->GetMicrotaskQueue()->EnqueueMicrotask(

env->isolate(),

[](void* arg) {

DestroyAsyncIdsCallback(static_cast<Environment*>(arg));

}, env);

});

}

env->destroy_async_id_list()->push_back(async_id);

CHECK_NE(provider_type(), PROVIDER_NONE);

if (async_id_ != kInvalidAsyncId) {

// This instance was in use before, we have already emitted an init with

// its previous async_id and need to emit a matching destroy for that

// before generating a new async_id.

EmitDestroy();

}

// Now we can assign a new async_id_ to this instance.

async_id_ = execution_async_id == kInvalidAsyncId ? env()->new_async_id()

: execution_async_id;

trigger_async_id_ = env()->get_default_trigger_async_id();

Isolate* isolate = env()->isolate();

{

HandleScope handle_scope(isolate);

Local<Object> obj = object();

CHECK(!obj.IsEmpty());

if (resource != obj) {

USE(obj->Set(env()->context(), env()->resource_symbol(), resource));

}

}

EmitTraceAsyncStart();

set_context_frame(async_context_frame::current(isolate));

EmitAsyncInit(env(), resource,

env()->async_hooks()->provider_string(provider_type()),

async_id_, trigger_async_id_);

Local<Object> object,

Local<String> type,

double async_id,

double trigger_async_id) {

CHECK(!object.IsEmpty());

CHECK(!type.IsEmpty());

AsyncHooks* async_hooks = env->async_hooks();

// Nothing to execute, so can continue normally.

if (async_hooks->fields()[AsyncHooks::kInit] == 0) {

return;

}

HandleScope scope(env->isolate());

Local<Function> init_fn = env->async_hooks_init_function();

Local<Value> argv[] = {

Number::New(env->isolate(), async_id),

type,

Number::New(env->isolate(), trigger_async_id),

object,

};

TryCatchScope try_catch(env,

TryCatchScope::CatchMode::kFatalRethrowStackOverflow);

USE(init_fn->Call(env->context(), object, arraysize(argv), argv));

int argc,

Local<Value>* argv) {

EmitTraceEventBefore();

// If this check fails it indicates an use after-free.

DCHECK(!context_frame().IsEmpty());

ProviderType provider = provider_type();

async_context context { get_async_id(), get_trigger_async_id() };

MaybeLocal<Value> ret = InternalMakeCallback(

env(), object(), object(), cb, argc, argv, context, context_frame());

// This is a static call with cached values because the `this` object may

// no longer be alive at this point.

EmitTraceEventAfter(provider, context.async_id);

return ret;

char buf[64];

snprintf(buf,

sizeof(buf),

"%s(%" PRIu64 ":%.0f)",

MemoryInfoName(),

env()->thread_id(),

async_id_);

return buf;

EscapableHandleScope handle_scope(env->isolate());

CHECK(!obj.IsEmpty());

TryCatchScope ignore_exceptions(env);

while (true) {

Local<Value> owner;

if (!obj->Get(env->context(),

env->owner_symbol()).ToLocal(&owner) ||

!owner->IsObject()) {

return handle_scope.Escape(obj);

}

obj = owner.As<Object>();

}