koffi 0.9.6 → 0.9.7

package/src/call_x64_win.cc

@@ -1,243 +1,240 @@
-// This program is free software: you can redistribute it and/or modify
-// it under the terms of the GNU Affero General Public License as published by
-// the Free Software Foundation, either version 3 of the License, or
-// (at your option) any later version.
-//
-// This program is distributed in the hope that it will be useful,
-// but WITHOUT ANY WARRANTY; without even the implied warranty of
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-// GNU Affero General Public License for more details.
-//
-// You should have received a copy of the GNU Affero General Public License
-// along with this program. If not, see https://www.gnu.org/licenses/.
-
-#ifdef _WIN64
-
-#include "vendor/libcc/libcc.hh"
-#include "ffi.hh"
-#include "call.hh"
-#include "util.hh"
-
-#include <napi.h>
-
-namespace RG {
-
-extern "C" uint64_t ForwardCallG(const void *func, uint8_t *sp);
-extern "C" float ForwardCallF(const void *func, uint8_t *sp);
-extern "C" double ForwardCallD(const void *func, uint8_t *sp);
-extern "C" uint64_t ForwardCallXG(const void *func, uint8_t *sp);
-extern "C" float ForwardCallXF(const void *func, uint8_t *sp);
-extern "C" double ForwardCallXD(const void *func, uint8_t *sp);
-
-static inline bool IsRegular(Size size)
-{
-    bool regular = (size <= 8 && !(size & (size - 1)));
-    return regular;
-}
-
-bool AnalyseFunction(FunctionInfo *func)
-{
-    func->ret.regular = IsRegular(func->ret.type->size);
-
-    for (ParameterInfo &param: func->parameters) {
-        param.regular = IsRegular(param.type->size);
-
-        func->forward_fp |= (param.type->primitive == PrimitiveKind::Float32 ||
-                             param.type->primitive == PrimitiveKind::Float64);
-    }
-
-    return true;
-}
-
-Napi::Value TranslateCall(const Napi::CallbackInfo &info)
-{
-    Napi::Env env = info.Env();
-    InstanceData *instance = env.GetInstanceData<InstanceData>();
-
-    FunctionInfo *func = (FunctionInfo *)info.Data();
-    LibraryData *lib = func->lib.get();
-
-    RG_DEFER { lib->tmp_alloc.ReleaseAll(); };
-
-    // Sanity checks
-    if (info.Length() < (uint32_t)func->parameters.len) {
-        ThrowError<Napi::TypeError>(env, "Expected %1 arguments, got %2", func->parameters.len, info.Length());
-        return env.Null();
-    }
-
-    // Stack pointer and register
-    uint8_t *top_ptr = lib->stack.end();
-    uint8_t *scratch_ptr = top_ptr - func->scratch_size;
-    uint8_t *return_ptr = nullptr;
-    uint8_t *args_ptr = nullptr;
-
-    // Reserve space for return value if needed
-    if (func->ret.regular) {
-        args_ptr = scratch_ptr - AlignLen(8 * std::max((Size)4, func->parameters.len), 16);
-    } else {
-        return_ptr = scratch_ptr - AlignLen(func->ret.type->size, 16);
-
-        args_ptr = return_ptr - AlignLen(8 * std::max((Size)4, func->parameters.len + 1), 16);
-        *(uint64_t *)args_ptr = (uint64_t)return_ptr;
-    }
-
-    RG_ASSERT(AlignUp(lib->stack.ptr, 16) == lib->stack.ptr);
-    RG_ASSERT(AlignUp(lib->stack.end(), 16) == lib->stack.end());
-    RG_ASSERT(AlignUp(args_ptr, 16) == args_ptr);
-
-    // Push arguments
-    for (Size i = 0, j = return_ptr ? 8 : 0; i < func->parameters.len; i++, j += 8) {
-        const ParameterInfo &param = func->parameters[i];
-        Napi::Value value = info[i];
-
-        switch (param.type->primitive) {
-            case PrimitiveKind::Void: { RG_UNREACHABLE(); } break;
-
-            case PrimitiveKind::Bool: {
-                if (RG_UNLIKELY(!value.IsBoolean())) {
-                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected boolean", GetValueType(instance, value), i + 1);
-                    return env.Null();
-                }
-
-                bool b = value.As<Napi::Boolean>();
-                *(bool *)(args_ptr + j) = b;
-            } break;
-            case PrimitiveKind::Int8:
-            case PrimitiveKind::UInt8:
-            case PrimitiveKind::Int16:
-            case PrimitiveKind::UInt16:
-            case PrimitiveKind::Int32:
-            case PrimitiveKind::UInt32:
-            case PrimitiveKind::Int64:
-            case PrimitiveKind::UInt64: {
-                if (RG_UNLIKELY(!value.IsNumber() && !value.IsBigInt())) {
-                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected number", GetValueType(instance, value), i + 1);
-                    return env.Null();
-                }
-
-                int64_t v = CopyNodeNumber<int64_t>(value);
-                *(uint64_t *)(args_ptr + j) = (uint64_t)v;
-            } break;
-            case PrimitiveKind::Float32: {
-                if (RG_UNLIKELY(!value.IsNumber() && !value.IsBigInt())) {
-                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected number", GetValueType(instance, value), i + 1);
-                    return env.Null();
-                }
-
-                float f = CopyNodeNumber<float>(value);
-                *(float *)(args_ptr + j) = f;
-            } break;
-            case PrimitiveKind::Float64: {
-                if (RG_UNLIKELY(!value.IsNumber() && !value.IsBigInt())) {
-                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected number", GetValueType(instance, value), i + 1);
-                    return env.Null();
-                }
-
-                double d = CopyNodeNumber<double>(value);
-                *(double *)(args_ptr + j) = d;
-            } break;
-            case PrimitiveKind::String: {
-                if (RG_UNLIKELY(!value.IsString())) {
-                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected string", GetValueType(instance, value), i + 1);
-                    return env.Null();
-                }
-
-                const char *str = CopyNodeString(value, &lib->tmp_alloc);
-                *(const char **)(args_ptr + j) = str;
-            } break;
-            case PrimitiveKind::Pointer: {
-                if (RG_UNLIKELY(!CheckValueTag(instance, value, param.type))) {
-                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected %3", GetValueType(instance, value), i + 1, param.type->name);
-                    return env.Null();
-                }
-
-                void *ptr = value.As<Napi::External<void>>();
-                *(void **)(args_ptr + j) = ptr;
-            } break;
-
-            case PrimitiveKind::Record: {
-                if (RG_UNLIKELY(!value.IsObject())) {
-                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected object", GetValueType(instance, value), i + 1);
-                    return env.Null();
-                }
-
-                uint8_t *ptr;
-                if (param.regular) {
-                    ptr = (uint8_t *)(args_ptr + j);
-                } else {
-                    ptr = scratch_ptr;
-                    *(uint8_t **)(args_ptr + j) = ptr;
-
-                    scratch_ptr = AlignUp(scratch_ptr + param.type->size, 16);
-                }
-
-                Napi::Object obj = value.As<Napi::Object>();
-                if (!PushObject(obj, param.type, &lib->tmp_alloc, ptr))
-                    return env.Null();
-            } break;
-        }
-    }
-
-    // DumpStack(func, MakeSpan(args_ptr, top_ptr - args_ptr));
-
-#define PERFORM_CALL(Suffix) \
-        (func->forward_fp ? ForwardCallX ## Suffix(func->func, args_ptr) \
-                          : ForwardCall ## Suffix(func->func, args_ptr))
-
-    // Execute and convert return value
-    switch (func->ret.type->primitive) {
-        case PrimitiveKind::Float32: {
-            float f = PERFORM_CALL(F);
-
-            return Napi::Number::New(env, (double)f);
-        } break;
-
-        case PrimitiveKind::Float64: {
-            double d = PERFORM_CALL(D);
-
-            return Napi::Number::New(env, d);
-        } break;
-
-        default: {
-            uint64_t rax = PERFORM_CALL(G);
-
-            switch (func->ret.type->primitive) {
-                case PrimitiveKind::Void: return env.Null();
-                case PrimitiveKind::Bool: return Napi::Boolean::New(env, rax);
-                case PrimitiveKind::Int8: return Napi::Number::New(env, (double)rax);
-                case PrimitiveKind::UInt8: return Napi::Number::New(env, (double)rax);
-                case PrimitiveKind::Int16: return Napi::Number::New(env, (double)rax);
-                case PrimitiveKind::UInt16: return Napi::Number::New(env, (double)rax);
-                case PrimitiveKind::Int32: return Napi::Number::New(env, (double)rax);
-                case PrimitiveKind::UInt32: return Napi::Number::New(env, (double)rax);
-                case PrimitiveKind::Int64: return Napi::BigInt::New(env, (int64_t)rax);
-                case PrimitiveKind::UInt64: return Napi::BigInt::New(env, rax);
-                case PrimitiveKind::Float32: { RG_UNREACHABLE(); } break;
-                case PrimitiveKind::Float64: { RG_UNREACHABLE(); } break;
-                case PrimitiveKind::String: return Napi::String::New(env, (const char *)rax);
-                case PrimitiveKind::Pointer: {
-                    void *ptr = (void *)rax;
-
-                    Napi::External<void> external = Napi::External<void>::New(env, ptr);
-                    SetValueTag(instance, external, func->ret.type);
-
-                    return external;
-                } break;
-
-                case PrimitiveKind::Record: {
-                    const uint8_t *ptr = return_ptr ? return_ptr : (const uint8_t *)&rax;
-                    Napi::Object obj = PopObject(env, ptr, func->ret.type);
-                    return obj;
-                } break;
-            }
-        } break;
-    }
-
-#undef PERFORM_CALL
-
-    RG_UNREACHABLE();
-}
-
-}
-
-#endif
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see https://www.gnu.org/licenses/.
+
+#ifdef _WIN64
+
+#include "vendor/libcc/libcc.hh"
+#include "ffi.hh"
+#include "call.hh"
+#include "util.hh"
+
+#include <napi.h>
+
+namespace RG {
+
+extern "C" uint64_t ForwardCallG(const void *func, uint8_t *sp);
+extern "C" float ForwardCallF(const void *func, uint8_t *sp);
+extern "C" double ForwardCallD(const void *func, uint8_t *sp);
+extern "C" uint64_t ForwardCallXG(const void *func, uint8_t *sp);
+extern "C" float ForwardCallXF(const void *func, uint8_t *sp);
+extern "C" double ForwardCallXD(const void *func, uint8_t *sp);
+
+static inline bool IsRegular(Size size)
+{
+    bool regular = (size <= 8 && !(size & (size - 1)));
+    return regular;
+}
+
+bool AnalyseFunction(InstanceData *, FunctionInfo *func)
+{
+    func->ret.regular = IsRegular(func->ret.type->size);
+
+    for (ParameterInfo &param: func->parameters) {
+        param.regular = IsRegular(param.type->size);
+
+        func->forward_fp |= (param.type->primitive == PrimitiveKind::Float32 ||
+                             param.type->primitive == PrimitiveKind::Float64);
+    }
+
+    func->args_size = AlignLen(8 * std::max((Size)4, func->parameters.len + !func->ret.regular), 16);
+
+    return true;
+}
+
+Napi::Value TranslateCall(const Napi::CallbackInfo &info)
+{
+    Napi::Env env = info.Env();
+    InstanceData *instance = env.GetInstanceData<InstanceData>();
+    FunctionInfo *func = (FunctionInfo *)info.Data();
+
+    CallData call(env, instance, func);
+
+    // Sanity checks
+    if (info.Length() < (uint32_t)func->parameters.len) {
+        ThrowError<Napi::TypeError>(env, "Expected %1 arguments, got %2", func->parameters.len, info.Length());
+        return env.Null();
+    }
+
+    uint8_t *return_ptr = nullptr;
+    uint64_t *args_ptr = nullptr;
+
+    // Pass return value in register or through memory
+    if (RG_UNLIKELY(!call.AllocStack(func->args_size, 16, &args_ptr)))
+        return env.Null();
+    if (!func->ret.regular) {
+        if (RG_UNLIKELY(!call.AllocHeap(func->ret.type->size, 16, &return_ptr)))
+            return env.Null();
+        *(uint8_t **)(args_ptr++) = return_ptr;
+    }
+
+    // Push arguments
+    for (Size i = 0; i < func->parameters.len; i++) {
+        const ParameterInfo &param = func->parameters[i];
+        Napi::Value value = info[i];
+
+        switch (param.type->primitive) {
+            case PrimitiveKind::Void: { RG_UNREACHABLE(); } break;
+
+            case PrimitiveKind::Bool: {
+                if (RG_UNLIKELY(!value.IsBoolean())) {
+                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected boolean", GetValueType(instance, value), i + 1);
+                    return env.Null();
+                }
+
+                bool b = value.As<Napi::Boolean>();
+
+                *(bool *)(args_ptr++) = b;
+            } break;
+            case PrimitiveKind::Int8:
+            case PrimitiveKind::UInt8:
+            case PrimitiveKind::Int16:
+            case PrimitiveKind::UInt16:
+            case PrimitiveKind::Int32:
+            case PrimitiveKind::UInt32:
+            case PrimitiveKind::Int64:
+            case PrimitiveKind::UInt64: {
+                if (RG_UNLIKELY(!value.IsNumber() && !value.IsBigInt())) {
+                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected number", GetValueType(instance, value), i + 1);
+                    return env.Null();
+                }
+
+                int64_t v = CopyNumber<int64_t>(value);
+                *(args_ptr++) = (uint64_t)v;
+            } break;
+            case PrimitiveKind::Float32: {
+                if (RG_UNLIKELY(!value.IsNumber() && !value.IsBigInt())) {
+                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected number", GetValueType(instance, value), i + 1);
+                    return env.Null();
+                }
+
+                float f = CopyNumber<float>(value);
+                *(float *)(args_ptr++) = f;
+            } break;
+            case PrimitiveKind::Float64: {
+                if (RG_UNLIKELY(!value.IsNumber() && !value.IsBigInt())) {
+                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected number", GetValueType(instance, value), i + 1);
+                    return env.Null();
+                }
+
+                double d = CopyNumber<double>(value);
+                *(double *)(args_ptr++) = d;
+            } break;
+            case PrimitiveKind::String: {
+                if (RG_UNLIKELY(!value.IsString())) {
+                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected string", GetValueType(instance, value), i + 1);
+                    return env.Null();
+                }
+
+                const char *str = call.PushString(value);
+                if (RG_UNLIKELY(!str))
+                    return env.Null();
+                *(const char **)(args_ptr++) = str;
+            } break;
+
+            case PrimitiveKind::Pointer: {
+                if (RG_UNLIKELY(!CheckValueTag(instance, value, param.type))) {
+                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected %3", GetValueType(instance, value), i + 1, param.type->name);
+                    return env.Null();
+                }
+
+                void *ptr = value.As<Napi::External<void>>();
+                *(void **)(args_ptr++) = ptr;
+            } break;
+
+            case PrimitiveKind::Record: {
+                if (RG_UNLIKELY(!value.IsObject())) {
+                    ThrowError<Napi::TypeError>(env, "Unexpected %1 value for argument %2, expected object", GetValueType(instance, value), i + 1);
+                    return env.Null();
+                }
+
+                uint8_t *ptr;
+                if (param.regular) {
+                    ptr = (uint8_t *)(args_ptr++);
+                } else {
+                    if (RG_UNLIKELY(!call.AllocHeap(param.type->size, 16, &ptr)))
+                        return env.Null();
+                    *(uint8_t **)(args_ptr++) = ptr;
+                }
+
+                Napi::Object obj = value.As<Napi::Object>();
+                if (!call.PushObject(obj, param.type, ptr))
+                    return env.Null();
+            } break;
+        }
+    }
+
+    if (instance->debug) {
+        call.DumpDebug();
+    }
+
+#define PERFORM_CALL(Suffix) \
+        (func->forward_fp ? ForwardCallX ## Suffix(func->func, call.GetSP()) \
+                          : ForwardCall ## Suffix(func->func, call.GetSP()))
+
+    // Execute and convert return value
+    switch (func->ret.type->primitive) {
+        case PrimitiveKind::Float32: {
+            float f = PERFORM_CALL(F);
+
+            return Napi::Number::New(env, (double)f);
+        } break;
+
+        case PrimitiveKind::Float64: {
+            double d = PERFORM_CALL(D);
+
+            return Napi::Number::New(env, d);
+        } break;
+
+        default: {
+            uint64_t rax = PERFORM_CALL(G);
+
+            switch (func->ret.type->primitive) {
+                case PrimitiveKind::Void: return env.Null();
+                case PrimitiveKind::Bool: return Napi::Boolean::New(env, rax);
+                case PrimitiveKind::Int8: return Napi::Number::New(env, (double)rax);
+                case PrimitiveKind::UInt8: return Napi::Number::New(env, (double)rax);
+                case PrimitiveKind::Int16: return Napi::Number::New(env, (double)rax);
+                case PrimitiveKind::UInt16: return Napi::Number::New(env, (double)rax);
+                case PrimitiveKind::Int32: return Napi::Number::New(env, (double)rax);
+                case PrimitiveKind::UInt32: return Napi::Number::New(env, (double)rax);
+                case PrimitiveKind::Int64: return Napi::BigInt::New(env, (int64_t)rax);
+                case PrimitiveKind::UInt64: return Napi::BigInt::New(env, rax);
+                case PrimitiveKind::Float32: { RG_UNREACHABLE(); } break;
+                case PrimitiveKind::Float64: { RG_UNREACHABLE(); } break;
+                case PrimitiveKind::String: return Napi::String::New(env, (const char *)rax);
+                case PrimitiveKind::Pointer: {
+                    void *ptr = (void *)rax;
+
+                    Napi::External<void> external = Napi::External<void>::New(env, ptr);
+                    SetValueTag(instance, external, func->ret.type);
+
+                    return external;
+                } break;
+
+                case PrimitiveKind::Record: {
+                    const uint8_t *ptr = return_ptr ? return_ptr : (const uint8_t *)&rax;
+                    Napi::Object obj = PopObject(env, ptr, func->ret.type);
+                    return obj;
+                } break;
+            }
+        } break;
+    }
+
+#undef PERFORM_CALL
+
+    RG_UNREACHABLE();
+}
+
+}
+
+#endif