Merge pull request #212 from ldorau/Use_base_allocator_in_jemalloc_pool

Use base allocator in jemalloc pool

Author: bratpiorka

src/pool/pool_jemalloc.c

@@ -13,6 +13,7 @@
 
 #include <jemalloc/jemalloc.h>
 
+#include "base_alloc_global.h"
 #include "utils_common.h"
 #include <umf/memory_pool.h>
 #include <umf/memory_pool_ops.h>
@@ -23,6 +24,9 @@
 typedef struct jemalloc_memory_pool_t {
     umf_memory_provider_handle_t provider;
     unsigned int arena_index; // index of jemalloc arena
+
+    // saved pointer to the global base allocator
+    umf_ba_pool_t *base_allocator;
 } jemalloc_memory_pool_t;
 
 static __TLS umf_result_t TLS_last_allocation_error;
@@ -357,11 +361,18 @@ static umf_result_t je_initialize(umf_memory_provider_handle_t provider,
     size_t unsigned_size = sizeof(unsigned);
     int err;
 
-    jemalloc_memory_pool_t *pool = malloc(sizeof(jemalloc_memory_pool_t));
+    umf_ba_pool_t *base_allocator =
+        umf_ba_get_pool(sizeof(jemalloc_memory_pool_t));
+    if (!base_allocator) {
+        return UMF_RESULT_ERROR_OUT_OF_HOST_MEMORY;
+    }
+
+    jemalloc_memory_pool_t *pool = umf_ba_alloc(base_allocator);
     if (!pool) {
         return UMF_RESULT_ERROR_OUT_OF_HOST_MEMORY;
     }
 
+    pool->base_allocator = base_allocator;
     pool->provider = provider;
 
     unsigned arena_index;
@@ -392,7 +403,7 @@ static umf_result_t je_initialize(umf_memory_provider_handle_t provider,
     return UMF_RESULT_SUCCESS;
 
 err_free_pool:
-    free(pool);
+    umf_ba_free(base_allocator, pool);
     return UMF_RESULT_ERROR_MEMORY_PROVIDER_SPECIFIC;
 }
 
@@ -403,7 +414,7 @@ static void je_finalize(void *pool) {
     snprintf(cmd, sizeof(cmd), "arena.%u.destroy", je_pool->arena_index);
     mallctl(cmd, NULL, 0, NULL, 0);
     pool_by_arena_index[je_pool->arena_index] = NULL;
-    free(je_pool);
+    umf_ba_free(je_pool->base_allocator, je_pool);
 }
 
 static size_t je_malloc_usable_size(void *pool, void *ptr) {