core: use mimalloc

Author: saghul

CMakeLists.txt

@@ -45,6 +45,7 @@ macro(cpr_option OPTION_NAME OPTION_TEXT OPTION_DEFAULT)
 endmacro()
 
 cpr_option(BUILD_WITH_ASAN "If ON, build with the address sanitizer enabled" OFF)
+cpr_option(BUILD_WITH_MIMALLOC "If ON, build with mimalloc (default)" ON)
 cpr_option(USE_EXTERNAL_FFI "Specify to use external ffi dependency" OFF)
 
 add_subdirectory(deps/quickjs EXCLUDE_FROM_ALL)
@@ -57,6 +58,15 @@ add_subdirectory(deps/sqlite3 EXCLUDE_FROM_ALL)
 set(BUILD_WASI "simple" CACHE STRING "WASI implementation")
 add_subdirectory(deps/wasm3 EXCLUDE_FROM_ALL)
 
+if(BUILD_WITH_MIMALLOC)
+    option(MI_OVERRIDE "" OFF)
+    option(MI_BUILD_SHARED "" OFF)
+    option(MI_BUILD_STATIC "" ON)
+    option(MI_BUILD_OBJECT "" OFF)
+    option(MI_BUILD_TESTS "" OFF)
+    add_subdirectory(deps/mimalloc EXCLUDE_FROM_ALL)
+endif()
+
 find_package(CURL REQUIRED)
 
 add_executable(tjs
@@ -141,7 +151,7 @@ string(TOLOWER ${CMAKE_SYSTEM_NAME} TJS_PLATFORM)
 target_compile_options(tjs PRIVATE ${tjs_cflags})
 target_compile_definitions(tjs PRIVATE TJS__PLATFORM="${TJS_PLATFORM}")
 target_include_directories(tjs PRIVATE ${CURL_INCLUDE_DIRS})
-target_link_libraries(tjs qjs uv_a m3 sqlite3 m pthread ${CURL_LIBRARIES})
+target_link_libraries(tjs qjs uv_a m3 sqlite3 m pthread mimalloc-static ${CURL_LIBRARIES})
 
 if (BUILD_WITH_ASAN)
     target_compile_options(tjs PRIVATE -fsanitize=address)

src/vm.c

@@ -25,11 +25,75 @@
 #include "private.h"
 #include "tjs.h"
 
+#include <mimalloc.h>
 #include <string.h>
 
 
 #define TJS__DEFAULT_STACK_SIZE 1048576
 
+static void *tjs__malloc(JSMallocState *s, size_t size)
+{
+    void *ptr;
+
+    /* Do not allocate zero bytes: behavior is platform dependent */
+    assert(size != 0);
+
+    if (unlikely(s->malloc_size + size > s->malloc_limit))
+        return NULL;
+
+    ptr = mi_malloc(size);
+    if (!ptr)
+        return NULL;
+
+    s->malloc_count++;
+    s->malloc_size += mi_malloc_usable_size(ptr);
+    return ptr;
+}
+
+static void tjs__free(JSMallocState *s, void *ptr)
+{
+    if (!ptr)
+        return;
+
+    s->malloc_count--;
+    s->malloc_size -= mi_malloc_usable_size(ptr);
+    mi_free(ptr);
+}
+
+static void *tjs__realloc(JSMallocState *s, void *ptr, size_t size)
+{
+    size_t old_size;
+
+    if (!ptr) {
+        if (size == 0)
+            return NULL;
+        return tjs__malloc(s, size);
+    }
+    old_size = mi_malloc_usable_size(ptr);
+    if (size == 0) {
+        s->malloc_count--;
+        s->malloc_size -= old_size;
+        mi_free(ptr);
+        return NULL;
+    }
+    if (s->malloc_size + size - old_size > s->malloc_limit)
+        return NULL;
+
+    ptr = mi_realloc(ptr, size);
+    if (!ptr)
+        return NULL;
+
+    s->malloc_size += mi_malloc_usable_size(ptr) - old_size;
+    return ptr;
+}
+
+static const JSMallocFunctions tjs_mf = {
+    tjs__malloc,
+    tjs__free,
+    tjs__realloc,
+    mi_malloc_usable_size
+};
+
 /* core */
 extern const uint8_t tjs__core[];
 extern const uint32_t tjs__core_size;
@@ -161,7 +225,7 @@ TJSRuntime *TJS_NewRuntimeInternal(bool is_worker, TJSRunOptions *options) {
 
     memcpy(&qrt->options, options, sizeof(*options));
 
-    qrt->rt = JS_NewRuntime();
+    qrt->rt = JS_NewRuntime2(&tjs_mf, NULL);
     CHECK_NOT_NULL(qrt->rt);
 
     qrt->ctx = JS_NewContext(qrt->rt);