Build libumf_pool_scalable on Windows

Signed-off-by: Lukasz Dorau <[email protected]>

Author: ldorau

src/pool/CMakeLists.txt

@@ -53,12 +53,18 @@ if(UMF_BUILD_LIBUMF_POOL_JEMALLOC)
 endif()
 
 # libumf_pool_scalable
+if(LINUX)
+    set(LIBS_POOL_SCALABLE dl umf_utils)
+elseif(WINDOWS)
+    set(LIBS_POOL_SCALABLE umf_utils)
+endif()
+
 if(UMF_BUILD_LIBUMF_POOL_SCALABLE)
-    if(LINUX)
+    if(LINUX OR WINDOWS)
         add_umf_library(NAME scalable_pool
                         TYPE STATIC
                         SRCS pool_scalable.c ${POOL_EXTRA_SRCS}
-                        LIBS dl umf_utils)
+                        LIBS ${LIBS_POOL_SCALABLE})
         target_compile_definitions(scalable_pool PUBLIC ${POOL_COMPILE_DEFINITIONS})
         add_library(${PROJECT_NAME}::scalable_pool ALIAS scalable_pool)
         install(TARGETS scalable_pool

src/pool/pool_scalable.c

@@ -7,9 +7,12 @@
  *
  */
 
-#include <assert.h>
+#ifndef _WIN32
 #include <dlfcn.h>
 #include <pthread.h>
+#endif
+
+#include <assert.h>
 #include <stdbool.h>
 #include <stdint.h>
 #include <stdio.h>
@@ -62,11 +65,20 @@ static struct tbb_callbacks g_tbb_ops;
 static UTIL_ONCE_FLAG tbb_is_initialized = UTIL_ONCE_FLAG_INIT;
 static bool Init_tbb_global_state_failed;
 
+#ifdef _WIN32
+#define dlsym(handle, symbol) GetProcAddress(handle, symbol)
+#define dlclose(handle) (void)(handle)
+#endif
+
 static void init_tbb_global_state(void) {
-    const char so_name[] = "libtbbmalloc.so.2";
-    void *tbb_handle = dlopen(so_name, RTLD_LAZY);
+    const char lib_name[] = "libtbbmalloc.so.2";
+#ifdef _WIN32
+    HMODULE tbb_handle = LoadLibrary(TEXT(lib_name));
+#else
+    void *tbb_handle = dlopen(lib_name, RTLD_LAZY);
+#endif
     if (!tbb_handle) {
-        fprintf(stderr, "%s not found.\n", so_name);
+        fprintf(stderr, "%s not found.\n", lib_name);
         Init_tbb_global_state_failed = true;
         return;
     }
@@ -95,7 +107,7 @@ static void init_tbb_global_state(void) {
         !tbb_ops.pool_aligned_malloc || !tbb_ops.pool_free ||
         !tbb_ops.pool_create_v1 || !tbb_ops.pool_destroy ||
         !tbb_ops.pool_identify) {
-        fprintf(stderr, "Could not find symbols in %s.\n", so_name);
+        fprintf(stderr, "Could not find symbols in %s.\n", lib_name);
         dlclose(tbb_handle);
         Init_tbb_global_state_failed = true;
         return;