From bada3312a8c1c6cdca474e99c74babb759e65823 Mon Sep 17 00:00:00 2001
From: Vsevolod Stakhov <vsevolod@rambler-co.ru>
Date: Mon, 18 Jul 2011 17:55:27 +0400
Subject: [PATCH] Fix statfiles class determination euristic. Fix call of
 classifier pre-callback.

---
 src/cfg_utils.c          |  6 +++---
 src/lua/lua_classifier.c | 18 +++++++++---------
 2 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/src/cfg_utils.c b/src/cfg_utils.c
index a34084cce..da1fe34a1 100644
--- a/src/cfg_utils.c
+++ b/src/cfg_utils.c
@@ -1051,7 +1051,7 @@ gboolean
 check_classifier_statfiles (struct classifier_config *cf)
 {
 	struct statfile                *st;
-	gboolean                        has_other = FALSE, cur_class;
+	gboolean                        has_other = FALSE, res = FALSE, cur_class;
 	GList                          *cur;
 
 	/* First check classes directly */
@@ -1093,14 +1093,14 @@ check_classifier_statfiles (struct classifier_config *cf)
 		}
 		else {
 			if (cur_class != st->is_spam) {
-				return TRUE;
+				res = TRUE;
 			}
 		}
 
 		cur = g_list_next (cur);
 	}
 
-	return FALSE;
+	return res;
 }
 
 /*
diff --git a/src/lua/lua_classifier.c b/src/lua/lua_classifier.c
index 901d6cff5..4e868b7f7 100644
--- a/src/lua/lua_classifier.c
+++ b/src/lua/lua_classifier.c
@@ -81,8 +81,7 @@ call_classifier_pre_callback (struct classifier_config *ccf, struct worker_task
 {
 	struct classifier_config      **pccf;
 	struct worker_task            **ptask;
-	struct statfile                *st;
-	gint                            i, len;
+	struct statfile               **pst;
 	GList                          *res = NULL;
 
 	pccf = lua_newuserdata (L, sizeof (struct classifier_config *));
@@ -100,15 +99,16 @@ call_classifier_pre_callback (struct classifier_config *ccf, struct worker_task
 		msg_warn ("error running pre classifier callback %s", lua_tostring (L, -1));
 	}
 	else {
-		if (lua_istable (L, 1)) {
-			len = lua_objlen (L, 1);
-			for (i = 1; i <= len; i ++) {
-				lua_rawgeti (L, 1, i);
-				st = lua_check_statfile (L);
-				if (st) {
-					res = g_list_prepend (res, st);
+		if (lua_istable (L, -1)) {
+			lua_pushnil (L);
+			while(lua_next (L, -2)) {
+				pst = luaL_checkudata (L, -1, "rspamd{statfile}");
+				if (pst) {
+					res = g_list_prepend (res, *pst);
 				}
+				lua_pop (L, 1);
 			}
+			lua_pop (L, 1);
 		}
 	}
 
-- 
2.39.5