From cba520a97286b3b4febdb654f028bc923c45726b Mon Sep 17 00:00:00 2001 From: Vsevolod Stakhov Date: Sat, 13 May 2017 19:03:15 +0100 Subject: [PATCH] [Feature] Use headers hash in bayes metatokens --- src/libstat/stat_process.c | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/src/libstat/stat_process.c b/src/libstat/stat_process.c index 74bf89d5f..6411580f6 100644 --- a/src/libstat/stat_process.c +++ b/src/libstat/stat_process.c @@ -21,6 +21,7 @@ #include "libmime/images.h" #include "libserver/html.h" #include "lua/lua_common.h" +#include "libserver/mempool_vars_internal.h" #include "utlist.h" #include @@ -79,6 +80,7 @@ rspamd_stat_tokenize_parts_metadata (struct rspamd_stat_ctx *st_ctx, guint i; gchar tmpbuf[128]; lua_State *L = task->cfg->lua_state; + const gchar *headers_hash; ar = g_array_sized_new (FALSE, FALSE, sizeof (elt), 16); elt.flags = RSPAMD_STAT_TOKEN_FLAG_META; @@ -171,6 +173,17 @@ rspamd_stat_tokenize_parts_metadata (struct rspamd_stat_ctx *st_ctx, cur = g_list_next (cur); } + /* Use headers order */ + headers_hash = rspamd_mempool_get_variable (task->task_pool, + RSPAMD_MEMPOOL_HEADERS_HASH); + + if (headers_hash) { + elt.begin = (gchar *)headers_hash; + elt.len = 16; + g_array_append_val (ar, elt); + } + + /* Use metatokens plugin from Lua */ lua_getglobal (L, "rspamd_plugins"); -- 2.39.5