[Feature] Save bayes probability in memory pool var

Issue: #613

diff --git a/src/libstat/classifiers/bayes.c b/src/libstat/classifiers/bayes.c
index 867fe4dc649da3e55398158ae2e9764a54df4291..4e30f191e1e8a82e1da9d635dbd88536da3d79a0 100644 (file)
--- a/src/libstat/classifiers/bayes.c
+++ b/src/libstat/classifiers/bayes.c
@@ -196,7 +196,7 @@ bayes_classify (struct rspamd_classifier * ctx,
                GPtrArray *tokens,
                struct rspamd_task *task)
 {
-       double final_prob, h, s;
+       double final_prob, h, s, *pprob;
        char *sumbuf;
        struct rspamd_statfile *st = NULL;
        struct bayes_task_closure cl;
@@ -273,6 +273,9 @@ bayes_classify (struct rspamd_classifier * ctx,
                }
        }
 
+       pprob = rspamd_mempool_alloc (task->task_pool, sizeof (*pprob));
+       rspamd_mempool_set_variable (task->task_pool, "bayes_prob", pprob, NULL);
+
        if (cl.processed_tokens > 0 && fabs (final_prob - 0.5) > 0.05) {
 
                sumbuf = rspamd_mempool_alloc (task->task_pool, 32);