koffi 2.4.2 → 2.5.0-beta.2

package/src/koffi/src/ffi.cc

@@ -134,7 +134,7 @@ static Napi::Value GetSetConfig(const Napi::CallbackInfo &info)
         Napi::Array keys = obj.GetPropertyNames();
 
         for (uint32_t i = 0; i < keys.Length(); i++) {
-            std::string key = ((Napi::Value)keys[i]).As<Napi::String>();
+            std::string key = keys.Get(i).As<Napi::String>();
             Napi::Value value = obj[key];
 
             if (key == "sync_stack_size") {
@@ -243,7 +243,7 @@ static Napi::Value CreateStructType(const Napi::CallbackInfo &info, bool pad)
     for (uint32_t i = 0; i < keys.Length(); i++) {
         RecordMember member = {};
 
-        std::string key = ((Napi::Value)keys[i]).As<Napi::String>();
+        std::string key = keys.Get(i).As<Napi::String>();
         Napi::Value value = obj[key];
         int16_t align = 0;
 
@@ -252,12 +252,12 @@ static Napi::Value CreateStructType(const Napi::CallbackInfo &info, bool pad)
         if (value.IsArray()) {
             Napi::Array array = value.As<Napi::Array>();
 
-            if (array.Length() != 2 || !((Napi::Value)array[0u]).IsNumber()) {
+            if (array.Length() != 2 || !array.Get(0u).IsNumber()) {
                 ThrowError<Napi::Error>(env, "Member specifier array must contain alignement value and type");
                 return env.Null();
             }
 
-            int64_t align64 = ((Napi::Value)array[0u]).As<Napi::Number>().Int64Value();
+            int64_t align64 = array.Get(0u).As<Napi::Number>().Int64Value();
 
             if (!CheckAlignment(align64)) {
                 ThrowError<Napi::Error>(env, "Alignment of member '%1' must be 1, 2, 4 or 8", member.name);
@@ -371,7 +371,7 @@ static Napi::Value CreateUnionType(const Napi::CallbackInfo &info)
     for (uint32_t i = 0; i < keys.Length(); i++) {
         RecordMember member = {};
 
-        std::string key = ((Napi::Value)keys[i]).As<Napi::String>();
+        std::string key = keys.Get(i).As<Napi::String>();
         Napi::Value value = obj[key];
         int16_t align = 0;
 
@@ -380,12 +380,12 @@ static Napi::Value CreateUnionType(const Napi::CallbackInfo &info)
         if (value.IsArray()) {
             Napi::Array array = value.As<Napi::Array>();
 
-            if (array.Length() != 2 || !((Napi::Value)array[0u]).IsNumber()) {
+            if (array.Length() != 2 || !array.Get(0u).IsNumber()) {
                 ThrowError<Napi::Error>(env, "Member specifier array must contain alignement value and type");
                 return env.Null();
             }
 
-            int64_t align64 = ((Napi::Value)array[0u]).As<Napi::Number>().Int64Value();
+            int64_t align64 = array.Get(0u).As<Napi::Number>().Int64Value();
 
             if (!CheckAlignment(align64)) {
                 ThrowError<Napi::Error>(env, "Alignment of member '%1' must be 1, 2, 4 or 8", member.name);
@@ -879,7 +879,7 @@ static bool ParseClassicFunction(Napi::Env env, Napi::String name, Napi::Value r
     uint32_t parameters_len = parameters.Length();
 
     if (parameters_len) {
-        Napi::String str = ((Napi::Value)parameters[parameters_len - 1]).As<Napi::String>();
+        Napi::String str = parameters.Get(parameters_len - 1).As<Napi::String>();
 
         if (str.IsString() && str.Utf8Value() == "...") {
             func->variadic = true;
@@ -993,12 +993,15 @@ static Napi::Value CreateTypeAlias(const Napi::CallbackInfo &info)
     if (!type)
         return env.Null();
 
-    bool inserted;
-    instance->types_map.TrySet(alias, type, &inserted);
+    // Alias the type
+    {
+        bool inserted;
+        instance->types_map.TrySet(alias, type, &inserted);
 
-    if (!inserted) {
-        ThrowError<Napi::Error>(env, "Type name '%1' already exists", alias);
-        return env.Null();
+        if (!inserted) {
+            ThrowError<Napi::Error>(env, "Type name '%1' already exists", alias);
+            return env.Null();
+        }
     }
 
     return WrapType(env, instance, type);
@@ -1451,6 +1454,35 @@ Napi::Value TranslateAsyncCall(const Napi::CallbackInfo &info)
     return TranslateAsyncCall(func, func->native, info);
 }
 
+extern "C" void RelayCallback(Size idx, uint8_t *own_sp, uint8_t *caller_sp, BackRegisters *out_reg)
+{
+    if (RG_LIKELY(exec_call)) {
+        exec_call->RelaySafe(idx, own_sp, caller_sp, false, out_reg);
+    } else {
+        // This happens if the callback pointer is called from a different thread
+        // than the one that runs the FFI call (sync or async).
+
+        TrampolineInfo *trampoline = &shared.trampolines[idx];
+
+        Napi::Env env = trampoline->func.Env();
+        InstanceData *instance = env.GetInstanceData<InstanceData>();
+
+        InstanceMemory *mem = AllocateMemory(instance, instance->config.async_stack_size, instance->config.async_heap_size);
+        if (RG_UNLIKELY(!mem)) {
+            ThrowError<Napi::Error>(env, "Too many asynchronous calls are running");
+            return;
+        }
+
+        // Avoid triggering the "use callback beyond FFI" check
+        RG_DEFER_C(generation = trampoline->generation) { trampoline->generation = generation; };
+        trampoline->generation = -1;
+
+        // We set dispose_call to true so that the main thread will dispose of CallData itself
+        CallData call(env, instance, mem);
+        call.RelaySafe(idx, own_sp, caller_sp, true, out_reg);
+    }
+}
+
 static Napi::Value FindLibraryFunction(const Napi::CallbackInfo &info, CallConvention convention)
 {
     Napi::Env env = info.Env();
@@ -1725,6 +1757,116 @@ static Napi::Value UnregisterCallback(const Napi::CallbackInfo &info)
     return env.Undefined();
 }
 
+static Napi::Value CastValue(const Napi::CallbackInfo &info)
+{
+    Napi::Env env = info.Env();
+    InstanceData *instance = env.GetInstanceData<InstanceData>();
+
+    if (RG_UNLIKELY(info.Length() < 2)) {
+        ThrowError<Napi::TypeError>(env, "Expected 2 arguments, got %1", info.Length());
+        return env.Null();
+    }
+
+    Napi::Value value = info[0];
+
+    const TypeInfo *type = ResolveType(info[1]);
+    if (RG_UNLIKELY(!type))
+        return env.Null();
+    if (type->primitive != PrimitiveKind::Pointer &&
+            type->primitive != PrimitiveKind::String &&
+            type->primitive != PrimitiveKind::String16) {
+        ThrowError<Napi::TypeError>(env, "Only pointer or string types can be used for casting");
+        return env.Null();
+    }
+
+    ValueCast *cast = new ValueCast;
+
+    cast->ref.Reset(value, 1);
+    cast->type = type;
+
+    Napi::External<ValueCast> external = Napi::External<ValueCast>::New(env, cast, [](Napi::Env, ValueCast *cast) { delete cast; });
+    SetValueTag(instance, external, &CastMarker);
+
+    return external;
+}
+
+static Napi::Value DecodeValue(const Napi::CallbackInfo &info)
+{
+    Napi::Env env = info.Env();
+
+    bool has_offset = (info.Length() >= 2 && info[1].IsNumber());
+    bool has_len = (info.Length() >= 3u + has_offset && info[2u + has_offset].IsNumber());
+
+    if (RG_UNLIKELY(info.Length() < 2u + has_offset)) {
+        ThrowError<Napi::TypeError>(env, "Expected %1 to 4 arguments, got %2", 2 + has_offset, info.Length());
+        return env.Null();
+    }
+
+    const TypeInfo *type = ResolveType(info[1u + has_offset]);
+    if (RG_UNLIKELY(!type))
+        return env.Null();
+
+    Napi::Value value = info[0];
+    int64_t offset = has_offset ? info[1].As<Napi::Number>().Int64Value() : 0;
+
+    if (has_len) {
+        Size len = info[2u + has_offset].As<Napi::Number>();
+
+        Napi::Value ret = Decode(value, offset, type, &len);
+        return ret;
+    } else {
+        Napi::Value ret = Decode(value, offset, type);
+        return ret;
+    }
+}
+
+static Napi::Value GetPointerAddress(const Napi::CallbackInfo &info)
+{
+    Napi::Env env = info.Env();
+
+    if (info.Length() < 1) {
+        ThrowError<Napi::TypeError>(env, "Expected 1 argument, got %1", info.Length());
+        return env.Null();
+    }
+
+    void *ptr = nullptr;
+    if (!GetExternalPointer(env, info[0], &ptr))
+        return env.Null();
+
+    uint64_t ptr64 = (uint64_t)(uintptr_t)ptr;
+    Napi::BigInt bigint = Napi::BigInt::New(env, ptr64);
+
+    return bigint;
+}
+
+static Napi::Value CallPointerSync(const Napi::CallbackInfo &info)
+{
+    Napi::Env env = info.Env();
+    InstanceData *instance = env.GetInstanceData<InstanceData>();
+
+    if (RG_UNLIKELY(info.Length() < 2)) {
+        ThrowError<Napi::TypeError>(env, "Expected 2 or more arguments, got %1", info.Length());
+        return env.Null();
+    }
+
+    void *ptr = nullptr;
+    if (RG_UNLIKELY(!GetExternalPointer(env, info[0], &ptr)))
+        return env.Null();
+
+    const TypeInfo *type = ResolveType(info[1]);
+    if (RG_UNLIKELY(!type))
+        return env.Null();
+    if (RG_UNLIKELY(type->primitive != PrimitiveKind::Prototype)) {
+        ThrowError<Napi::TypeError>(env, "Unexpected %1 value for type, expected function type", GetValueType(instance, info[1]));
+        return env.Null();
+    }
+
+    const FunctionInfo *proto = type->ref.proto;
+
+    return proto->variadic ? TranslateVariadicCall(proto, ptr, info)
+                           : TranslateNormalCall(proto, ptr, info);
+}
+
 void LibraryHolder::Unload()
 {
 #ifdef _WIN32
@@ -1753,6 +1895,45 @@ void LibraryHolder::Unref() const
     }
 }
 
+FunctionInfo::~FunctionInfo()
+{
+    if (lib) {
+        lib->Unref();
+    }
+}
+
+const FunctionInfo *FunctionInfo::Ref() const
+{
+    refcount++;
+    return this;
+}
+
+void FunctionInfo::Unref() const
+{
+    if (!--refcount) {
+        delete this;
+    }
+}
+
+InstanceMemory::~InstanceMemory()
+{
+#ifdef _WIN32
+    if (stack.ptr) {
+        VirtualFree(stack.ptr, 0, MEM_RELEASE);
+    }
+    if (heap.ptr) {
+        VirtualFree(heap.ptr, 0, MEM_RELEASE);
+    }
+#else
+    if (stack.ptr) {
+        munmap(stack.ptr, stack.len);
+    }
+    if (heap.ptr) {
+        munmap(heap.ptr, heap.len);
+    }
+#endif
+}
+
 static void RegisterPrimitiveType(Napi::Env env, Napi::Object map, std::initializer_list<const char *> names,
                                   PrimitiveKind primitive, int32_t size, int16_t align, const char *ref = nullptr)
 {
@@ -1877,50 +2058,13 @@ static Napi::Object InitBaseTypes(Napi::Env env)
     instance->char_type = instance->types_map.FindValue("char", nullptr);
     instance->char16_type = instance->types_map.FindValue("char16", nullptr);
 
+    instance->active_symbol = Napi::Symbol::New(env, "active");
+
     types.Freeze();
 
     return types;
 }
 
-FunctionInfo::~FunctionInfo()
-{
-    if (lib) {
-        lib->Unref();
-    }
-}
-
-const FunctionInfo *FunctionInfo::Ref() const
-{
-    refcount++;
-    return this;
-}
-
-void FunctionInfo::Unref() const
-{
-    if (!--refcount) {
-        delete this;
-    }
-}
-
-InstanceMemory::~InstanceMemory()
-{
-#ifdef _WIN32
-    if (stack.ptr) {
-        VirtualFree(stack.ptr, 0, MEM_RELEASE);
-    }
-    if (heap.ptr) {
-        VirtualFree(heap.ptr, 0, MEM_RELEASE);
-    }
-#else
-    if (stack.ptr) {
-        munmap(stack.ptr, stack.len);
-    }
-    if (heap.ptr) {
-        munmap(heap.ptr, heap.len);
-    }
-#endif
-}
-
 static InstanceData *CreateInstance(Napi::Env env)
 {
     InstanceData *instance = new InstanceData();
@@ -1928,6 +2072,9 @@ static InstanceData *CreateInstance(Napi::Env env)
 
     instance->main_thread_id = std::this_thread::get_id();
 
+    instance->debug = GetDebugFlag("DUMP_CALLS");
+    FillRandomSafe(&instance->tag_lower, RG_SIZE(instance->tag_lower));
+
     if (napi_create_threadsafe_function(env, nullptr, nullptr,
                                         Napi::String::New(env, "Koffi Async Callback Broker"),
                                         0, 1, nullptr, nullptr, nullptr,
@@ -1948,214 +2095,53 @@ static InstanceData *CreateInstance(Napi::Env env)
     return instance;
 }
 
-InstanceData::~InstanceData()
-{
-    for (InstanceMemory *mem: memories) {
-        delete mem;
-    }
-
-    // Clean-up leftover trampoline references
-    {
-        std::lock_guard<std::mutex> lock(shared.mutex);
-
-        for (int16_t idx = 0; idx < MaxTrampolines; idx++) {
-            TrampolineInfo *trampoline = &shared.trampolines[idx];
-
-            if (trampoline->instance == this) {
-                trampoline->instance = nullptr;
-                trampoline->func.Reset();
-                trampoline->recv.Reset();
-            }
-        }
-    }
-
-    if (broker) {
-        napi_release_threadsafe_function(broker, napi_tsfn_abort);
-    }
-}
-
-static Napi::Value CastValue(const Napi::CallbackInfo &info)
-{
-    Napi::Env env = info.Env();
-    InstanceData *instance = env.GetInstanceData<InstanceData>();
-
-    if (RG_UNLIKELY(info.Length() < 2)) {
-        ThrowError<Napi::TypeError>(env, "Expected 2 arguments, got %1", info.Length());
-        return env.Null();
-    }
-
-    Napi::Value value = info[0];
-
-    const TypeInfo *type = ResolveType(info[1]);
-    if (RG_UNLIKELY(!type))
-        return env.Null();
-    if (type->primitive != PrimitiveKind::Pointer &&
-            type->primitive != PrimitiveKind::String &&
-            type->primitive != PrimitiveKind::String16) {
-        ThrowError<Napi::TypeError>(env, "Only pointer or string types can be used for casting");
-        return env.Null();
-    }
-
-    ValueCast *cast = new ValueCast;
-
-    cast->ref.Reset(value, 1);
-    cast->type = type;
-
-    Napi::External<ValueCast> external = Napi::External<ValueCast>::New(env, cast, [](Napi::Env, ValueCast *cast) { delete cast; });
-    SetValueTag(instance, external, &CastMarker);
-
-    return external;
-}
-
-static Napi::Value DecodeValue(const Napi::CallbackInfo &info)
-{
-    Napi::Env env = info.Env();
-
-    bool has_offset = (info.Length() >= 2 && info[1].IsNumber());
-    bool has_len = (info.Length() >= 3u + has_offset && info[2u + has_offset].IsNumber());
-
-    if (RG_UNLIKELY(info.Length() < 2u + has_offset)) {
-        ThrowError<Napi::TypeError>(env, "Expected %1 to 4 arguments, got %2", 2 + has_offset, info.Length());
-        return env.Null();
-    }
-
-    const TypeInfo *type = ResolveType(info[1u + has_offset]);
-    if (RG_UNLIKELY(!type))
-        return env.Null();
-
-    Napi::Value value = info[0];
-    int64_t offset = has_offset ? info[1].As<Napi::Number>().Int64Value() : 0;
-
-    if (has_len) {
-        Size len = info[2u + has_offset].As<Napi::Number>();
-
-        Napi::Value ret = Decode(value, offset, type, &len);
-        return ret;
-    } else {
-        Napi::Value ret = Decode(value, offset, type);
-        return ret;
-    }
-}
-
-static Napi::Value GetPointerAddress(const Napi::CallbackInfo &info)
-{
-    Napi::Env env = info.Env();
-
-    if (info.Length() < 1) {
-        ThrowError<Napi::TypeError>(env, "Expected 1 argument, got %1", info.Length());
-        return env.Null();
-    }
-
-    void *ptr = nullptr;
-    if (!GetExternalPointer(env, info[0], &ptr))
-        return env.Null();
-
-    uint64_t ptr64 = (uint64_t)(uintptr_t)ptr;
-    Napi::BigInt bigint = Napi::BigInt::New(env, ptr64);
-
-    return bigint;
-}
-
-static Napi::Value CallPointerSync(const Napi::CallbackInfo &info)
-{
-    Napi::Env env = info.Env();
-    InstanceData *instance = env.GetInstanceData<InstanceData>();
-
-    if (RG_UNLIKELY(info.Length() < 2)) {
-        ThrowError<Napi::TypeError>(env, "Expected 2 or more arguments, got %1", info.Length());
-        return env.Null();
-    }
-
-    void *ptr = nullptr;
-    if (RG_UNLIKELY(!GetExternalPointer(env, info[0], &ptr)))
-        return env.Null();
-
-    const TypeInfo *type = ResolveType(info[1]);
-    if (RG_UNLIKELY(!type))
-        return env.Null();
-    if (RG_UNLIKELY(type->primitive != PrimitiveKind::Prototype)) {
-        ThrowError<Napi::TypeError>(env, "Unexpected %1 value for type, expected function type", GetValueType(instance, info[1]));
-        return env.Null();
-    }
-
-    const FunctionInfo *proto = type->ref.proto;
-
-    return proto->variadic ? TranslateVariadicCall(proto, ptr, info)
-                           : TranslateNormalCall(proto, ptr, info);
-}
-
-extern "C" void RelayCallback(Size idx, uint8_t *own_sp, uint8_t *caller_sp, BackRegisters *out_reg)
+static Napi::Object InitModule(Napi::Env env, Napi::Object exports)
 {
-    if (RG_LIKELY(exec_call)) {
-        exec_call->RelaySafe(idx, own_sp, caller_sp, false, out_reg);
-    } else {
-        // This happens if the callback pointer is called from a different thread
-        // than the one that runs the FFI call (sync or async).
-
-        TrampolineInfo *trampoline = &shared.trampolines[idx];
-
-        Napi::Env env = trampoline->func.Env();
-        InstanceData *instance = env.GetInstanceData<InstanceData>();
-
-        InstanceMemory *mem = AllocateMemory(instance, instance->config.async_stack_size, instance->config.async_heap_size);
-        if (RG_UNLIKELY(!mem)) {
-            ThrowError<Napi::Error>(env, "Too many asynchronous calls are running");
-            return;
-        }
-
-        // Avoid triggering the "use callback beyond FFI" check
-        RG_DEFER_C(generation = trampoline->generation) { trampoline->generation = generation; };
-        trampoline->generation = -1;
+    InstanceData *instance = CreateInstance(env);
+    RG_CRITICAL(instance, "Failed to initialize Koffi");
 
-        // We set dispose_call to true so that the main thread will dispose of CallData itself
-        CallData call(env, instance, mem);
-        call.RelaySafe(idx, own_sp, caller_sp, true, out_reg);
-    }
-}
+    env.SetInstanceData(instance);
 
-template <typename Func>
-static void SetExports(Napi::Env env, Func func)
-{
-    func("config", Napi::Function::New(env, GetSetConfig));
-    func("stats", Napi::Function::New(env, GetStats));
+    exports.Set("config", Napi::Function::New(env, GetSetConfig));
+    exports.Set("stats", Napi::Function::New(env, GetStats));
 
-    func("struct", Napi::Function::New(env, CreatePaddedStructType));
-    func("pack", Napi::Function::New(env, CreatePackedStructType));
-    // func("union", Napi::Function::New(env, CreateUnionType));
-    // func("Union", Napi::Function::New(env, InstantiateUnion));
-    func("opaque", Napi::Function::New(env, CreateOpaqueType));
-    func("pointer", Napi::Function::New(env, CreatePointerType));
-    func("array", Napi::Function::New(env, CreateArrayType));
-    func("proto", Napi::Function::New(env, CreateFunctionType));
-    func("alias", Napi::Function::New(env, CreateTypeAlias));
+    exports.Set("struct", Napi::Function::New(env, CreatePaddedStructType));
+    exports.Set("pack", Napi::Function::New(env, CreatePackedStructType));
+    exports.Set("union", Napi::Function::New(env, CreateUnionType));
+    exports.Set("Union", Napi::Function::New(env, InstantiateUnion));
+    exports.Set("opaque", Napi::Function::New(env, CreateOpaqueType));
+    exports.Set("pointer", Napi::Function::New(env, CreatePointerType));
+    exports.Set("array", Napi::Function::New(env, CreateArrayType));
+    exports.Set("proto", Napi::Function::New(env, CreateFunctionType));
+    exports.Set("alias", Napi::Function::New(env, CreateTypeAlias));
 
-    func("sizeof", Napi::Function::New(env, GetTypeSize));
-    func("alignof", Napi::Function::New(env, GetTypeAlign));
-    func("offsetof", Napi::Function::New(env, GetMemberOffset));
-    func("resolve", Napi::Function::New(env, GetResolvedType));
-    func("introspect", Napi::Function::New(env, GetTypeDefinition));
+    exports.Set("sizeof", Napi::Function::New(env, GetTypeSize));
+    exports.Set("alignof", Napi::Function::New(env, GetTypeAlign));
+    exports.Set("offsetof", Napi::Function::New(env, GetMemberOffset));
+    exports.Set("resolve", Napi::Function::New(env, GetResolvedType));
+    exports.Set("introspect", Napi::Function::New(env, GetTypeDefinition));
 
-    func("load", Napi::Function::New(env, LoadSharedLibrary));
+    exports.Set("load", Napi::Function::New(env, LoadSharedLibrary));
 
-    func("in", Napi::Function::New(env, MarkIn));
-    func("out", Napi::Function::New(env, MarkOut));
-    func("inout", Napi::Function::New(env, MarkInOut));
+    exports.Set("in", Napi::Function::New(env, MarkIn));
+    exports.Set("out", Napi::Function::New(env, MarkOut));
+    exports.Set("inout", Napi::Function::New(env, MarkInOut));
 
-    func("disposable", Napi::Function::New(env, CreateDisposableType));
-    func("free", Napi::Function::New(env, CallFree));
+    exports.Set("disposable", Napi::Function::New(env, CreateDisposableType));
+    exports.Set("free", Napi::Function::New(env, CallFree));
 
-    func("register", Napi::Function::New(env, RegisterCallback));
-    func("unregister", Napi::Function::New(env, UnregisterCallback));
+    exports.Set("register", Napi::Function::New(env, RegisterCallback));
+    exports.Set("unregister", Napi::Function::New(env, UnregisterCallback));
 
-    func("as", Napi::Function::New(env, CastValue));
-    func("decode", Napi::Function::New(env, DecodeValue));
-    func("address", Napi::Function::New(env, GetPointerAddress));
-    func("call", Napi::Function::New(env, CallPointerSync));
+    exports.Set("as", Napi::Function::New(env, CastValue));
+    exports.Set("decode", Napi::Function::New(env, DecodeValue));
+    exports.Set("address", Napi::Function::New(env, GetPointerAddress));
+    exports.Set("call", Napi::Function::New(env, CallPointerSync));
 
-    func("errno", Napi::Function::New(env, GetOrSetErrNo));
+    exports.Set("errno", Napi::Function::New(env, GetOrSetErrNo));
 
     Napi::Object os = Napi::Object::New(env);
-    func("os", os);
+    exports.Set("os", os);
 
     // Init constants mapping
     {
@@ -2169,85 +2155,47 @@ static void SetExports(Napi::Env env, Func func)
     }
 
 #if defined(_WIN32)
-    func("extension", Napi::String::New(env, ".dll"));
+    exports.Set("extension", Napi::String::New(env, ".dll"));
 #elif defined(__APPLE__)
-    func("extension", Napi::String::New(env, ".dylib"));
+    exports.Set("extension", Napi::String::New(env, ".dylib"));
 #else
-    func("extension", Napi::String::New(env, ".so"));
+    exports.Set("extension", Napi::String::New(env, ".so"));
 #endif
 
     Napi::Object types = InitBaseTypes(env);
-    func("types", types);
-}
-
-}
-
-#if NODE_WANT_INTERNALS
+    exports.Set("types", types);
 
-static void SetValue(node::Environment *env, v8::Local<v8::Object> target,
-                     const char *name, Napi::Value value)
-{
-    v8::Isolate *isolate = env->isolate();
-    v8::Local<v8::Context> context = isolate->GetCurrentContext();
-
-    v8::NewStringType str_type = v8::NewStringType::kInternalized;
-    v8::Local<v8::String> str = v8::String::NewFromUtf8(isolate, name, str_type).ToLocalChecked();
+    exports.Set("internal", Napi::Boolean::New(env, false));
 
-    target->Set(context, str, v8impl::V8LocalValueFromJsValue(value)).Check();
+    return exports;
 }
 
-static void InitInternal(v8::Local<v8::Object> target, v8::Local<v8::Value>,
-                         v8::Local<v8::Context> context, void *)
+InstanceData::~InstanceData()
 {
-    using namespace RG;
-
-    node::Environment *env = node::Environment::GetCurrent(context);
-
-    // Not very clean but I don't know enough about Node and V8 to do better...
-    // ... and it seems to work okay.
-    napi_env env_napi = new napi_env__(context);
-    Napi::Env env_cxx(env_napi);
-    env->AtExit([](void *udata) {
-        napi_env env_napi = (napi_env)udata;
-        delete env_napi;
-    }, env_napi);
+    for (InstanceMemory *mem: memories) {
+        delete mem;
+    }
 
-    InstanceData *instance = CreateInstance(env_cxx);
-    RG_CRITICAL(instance, "Failed to initialize Koffi");
+    // Clean-up leftover trampoline references
+    {
+        std::lock_guard<std::mutex> lock(shared.mutex);
 
-    env_cxx.SetInstanceData(instance);
+        for (int16_t idx = 0; idx < MaxTrampolines; idx++) {
+            TrampolineInfo *trampoline = &shared.trampolines[idx];
 
-    instance->debug = GetDebugFlag("DUMP_CALLS");
-    FillRandomSafe(&instance->tag_lower, RG_SIZE(instance->tag_lower));
+            if (trampoline->instance == this) {
+                trampoline->instance = nullptr;
+                trampoline->func.Reset();
+                trampoline->recv.Reset();
+            }
+        }
+    }
 
-    SetExports(env_napi, [&](const char *name, Napi::Value value) { SetValue(env, target, name, value); });
-    SetValue(env, target, "internal", Napi::Boolean::New(env_cxx, true));
+    if (broker) {
+        napi_release_threadsafe_function(broker, napi_tsfn_abort);
+    }
 }
 
-#else
-
-static Napi::Object InitModule(Napi::Env env, Napi::Object exports)
-{
-    using namespace RG;
-
-    InstanceData *instance = CreateInstance(env);
-    RG_CRITICAL(instance, "Failed to initialize Koffi");
-
-    env.SetInstanceData(instance);
-
-    instance->debug = GetDebugFlag("DUMP_CALLS");
-    FillRandomSafe(&instance->tag_lower, RG_SIZE(instance->tag_lower));
-
-    SetExports(env, [&](const char *name, Napi::Value value) { exports.Set(name, value); });
-    exports.Set("internal", Napi::Boolean::New(env, false));
+NODE_API_MODULE(koffi, InitModule);
 
-    return exports;
 }
-
-#endif
-
-#if NODE_WANT_INTERNALS
-    NODE_MODULE_CONTEXT_AWARE_INTERNAL(koffi, InitInternal);
-#else
-    NODE_API_MODULE(koffi, InitModule);
-#endif