From fc23f700aa82df48915a117d3c6d823f5ad82332 Mon Sep 17 00:00:00 2001 From: Vsevolod Stakhov Date: Mon, 23 Apr 2018 12:41:17 +0100 Subject: [PATCH] [Minor] Allow to create radix tree from the pool --- src/libutil/radix.c | 26 ++++++++++++++++++++++---- src/libutil/radix.h | 2 ++ 2 files changed, 24 insertions(+), 4 deletions(-) diff --git a/src/libutil/radix.c b/src/libutil/radix.c index 8c2258459..3ef471174 100644 --- a/src/libutil/radix.c +++ b/src/libutil/radix.c @@ -40,8 +40,9 @@ INIT_LOG_MODULE(radix) struct radix_tree_compressed { rspamd_mempool_t *pool; - size_t size; struct btrie *tree; + size_t size; + gboolean own_pool; }; uintptr_t @@ -116,7 +117,7 @@ radix_create_compressed (void) { radix_compressed_t *tree; - tree = g_malloc0 (sizeof (*tree)); + tree = g_malloc (sizeof (*tree)); if (tree == NULL) { return NULL; } @@ -124,6 +125,21 @@ radix_create_compressed (void) tree->pool = rspamd_mempool_new (rspamd_mempool_suggest_size (), NULL); tree->size = 0; tree->tree = btrie_init (tree->pool); + tree->own_pool = TRUE; + + return tree; +} + +radix_compressed_t * +radix_create_compressed_with_pool (rspamd_mempool_t *pool) +{ + radix_compressed_t *tree; + + tree = rspamd_mempool_alloc (pool, sizeof (*tree)); + tree->pool = pool; + tree->size = 0; + tree->tree = btrie_init (tree->pool); + tree->own_pool = FALSE; return tree; } @@ -132,8 +148,10 @@ void radix_destroy_compressed (radix_compressed_t *tree) { if (tree) { - rspamd_mempool_delete (tree->pool); - g_free (tree); + if (tree->own_pool) { + rspamd_mempool_delete (tree->pool); + g_free (tree); + } } } diff --git a/src/libutil/radix.h b/src/libutil/radix.h index c04528000..be75b1b7f 100644 --- a/src/libutil/radix.h +++ b/src/libutil/radix.h @@ -71,6 +71,8 @@ void radix_destroy_compressed (radix_compressed_t *tree); */ radix_compressed_t *radix_create_compressed (void); +radix_compressed_t *radix_create_compressed_with_pool (rspamd_mempool_t *pool); + /** * Insert list of ip addresses and masks to the radix tree * @param list string line of addresses -- 2.39.5