From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Fri, 16 Nov 2018 10:16:56 +0000 (+0000)
Subject: [Minor] Add sanity checks for bayes probability
X-Git-Tag: 1.8.2~21
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=d302edad6a90062424df6883b0df9a3cb9325870;p=thirdparty%2Frspamd.git

[Minor] Add sanity checks for bayes probability
---

diff --git a/src/libstat/classifiers/bayes.c b/src/libstat/classifiers/bayes.c
index 5295e018b9..ee21254571 100644
--- a/src/libstat/classifiers/bayes.c
+++ b/src/libstat/classifiers/bayes.c
@@ -430,6 +430,19 @@ bayes_classify (struct rspamd_classifier * ctx,
 				(final_prob - 0.5) * 200.);
 		final_prob = rspamd_normalize_probability (final_prob, 0.5);
 		g_assert (st != NULL);
+
+		if (final_prob > 1 || final_prob < 0) {
+			msg_err_bayes ("internal error: probability %f is outside of the "
+				  "allowed range [0..1]", final_prob);
+
+			if (final_prob > 1) {
+				final_prob = 1.0;
+			}
+			else {
+				final_prob = 0.0;
+			}
+		}
+
 		rspamd_task_insert_result (task,
 				st->stcf->symbol,
 				final_prob,