diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/libserver/cfg_rcl.c | 59 | ||||
-rw-r--r-- | src/libserver/cfg_rcl.h | 13 |
2 files changed, 54 insertions, 18 deletions
diff --git a/src/libserver/cfg_rcl.c b/src/libserver/cfg_rcl.c index d549b2579..b6ca6705c 100644 --- a/src/libserver/cfg_rcl.c +++ b/src/libserver/cfg_rcl.c @@ -1121,10 +1121,10 @@ rspamd_rcl_add_section (struct rspamd_rcl_section **top, struct rspamd_rcl_default_handler_data * rspamd_rcl_add_default_handler (struct rspamd_rcl_section *section, - const gchar *name, - rspamd_rcl_default_handler_t handler, - gsize offset, - gint flags) + const gchar *name, + rspamd_rcl_default_handler_t handler, + goffset offset, + gint flags) { struct rspamd_rcl_default_handler_data *new; @@ -2115,6 +2115,35 @@ rspamd_rcl_parse_struct_keypair (rspamd_mempool_t *pool, return FALSE; } +static void +rspamd_rcl_insert_string_list_item (gpointer *target, rspamd_mempool_t *pool, + const gchar *src, gboolean is_hash) +{ + union { + GHashTable *hv; + GList *lv; + gpointer p; + } d; + gchar *val; + + d.p = *target; + + if (is_hash) { + if (d.hv == NULL) { + d.hv = g_hash_table_new (rspamd_str_hash, rspamd_str_equal); + } + + val = rspamd_mempool_strdup (pool, src); + g_hash_table_insert (d.hv, val, val); + } + else { + val = rspamd_mempool_strdup (pool, src); + d.lv = g_list_prepend (d.lv, val); + } + + *target = d.p; +} + gboolean rspamd_rcl_parse_struct_string_list (rspamd_mempool_t *pool, const ucl_object_t *obj, @@ -2123,13 +2152,16 @@ rspamd_rcl_parse_struct_string_list (rspamd_mempool_t *pool, GError **err) { struct rspamd_rcl_struct_parser *pd = ud; - GList **target; + gpointer *target; gchar *val, **strvec, **cvec; const ucl_object_t *cur; const gsize num_str_len = 32; ucl_object_iter_t iter = NULL; + gboolean is_hash; - target = (GList **)(((gchar *)pd->user_struct) + pd->offset); + + is_hash = pd->flags & RSPAMD_CL_FLAG_STRING_LIST_HASH; + target = (gpointer *)(((gchar *)pd->user_struct) + pd->offset); iter = ucl_object_iterate_new (obj); @@ -2140,8 +2172,7 @@ rspamd_rcl_parse_struct_string_list (rspamd_mempool_t *pool, cvec = strvec; while (*cvec) { - *target = g_list_prepend (*target, - rspamd_mempool_strdup (pool, *cvec)); + rspamd_rcl_insert_string_list_item (target, pool, *cvec, is_hash); cvec ++; } @@ -2167,7 +2198,8 @@ rspamd_rcl_parse_struct_string_list (rspamd_mempool_t *pool, "cannot convert an object or array to string"); return FALSE; } - *target = g_list_prepend (*target, val); + + rspamd_rcl_insert_string_list_item (target, pool, val, is_hash); } if (*target == NULL) { @@ -2179,9 +2211,12 @@ rspamd_rcl_parse_struct_string_list (rspamd_mempool_t *pool, } /* Add a destructor */ - rspamd_mempool_add_destructor (pool, - (rspamd_mempool_destruct_t)g_list_free, - *target); + + if (!is_hash) { + rspamd_mempool_add_destructor (pool, + (rspamd_mempool_destruct_t) g_list_free, + *target); + } return TRUE; } diff --git a/src/libserver/cfg_rcl.h b/src/libserver/cfg_rcl.h index ca9ffe83f..b18867cd1 100644 --- a/src/libserver/cfg_rcl.h +++ b/src/libserver/cfg_rcl.h @@ -54,7 +54,8 @@ struct rspamd_rcl_struct_parser { RSPAMD_CL_FLAG_UINT = 0x1 << 8, RSPAMD_CL_FLAG_INT_SIZE = 0x1 << 9, RSPAMD_CL_FLAG_STRING_PATH = 0x1 << 10, - RSPAMD_CL_FLAG_BOOLEAN_INVERSE = 0x1 << 11 + RSPAMD_CL_FLAG_BOOLEAN_INVERSE = 0x1 << 11, + RSPAMD_CL_FLAG_STRING_LIST_HASH = 0x1 << 12 } flags; }; @@ -96,11 +97,11 @@ typedef void (*rspamd_rcl_section_fin_t)(rspamd_mempool_t *pool, gpointer ud); * @return newly created structure */ struct rspamd_rcl_default_handler_data * rspamd_rcl_add_default_handler ( - struct rspamd_rcl_section *section, - const gchar *name, - rspamd_rcl_default_handler_t handler, - gsize offset, - gint flags); + struct rspamd_rcl_section *section, + const gchar *name, + rspamd_rcl_default_handler_t handler, + goffset offset, + gint flags); /** * Add new section to the configuration |