diff options
| -rw-r--r-- | src/cfg_utils.c | 6 | ||||
| -rw-r--r-- | 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); } } |