[Feature] Implement classifying for lua classifiers

1 files changed, 47 insertions, 2 deletions

diff --git a/src/libstat/classifiers/lua_classifier.c b/src/libstat/classifiers/lua_classifier.c
index c1aee7261..cd9fc3bd1 100644
--- a/src/libstat/classifiers/lua_classifier.c
+++ b/src/libstat/classifiers/lua_classifier.c
@@ -123,15 +123,60 @@ lua_classifier_init (rspamd_mempool_t *pool,
 	return TRUE;
 }
 gboolean
-lua_classifier_classify (struct rspamd_classifier *ctx,
+lua_classifier_classify (struct rspamd_classifier *cl,
 		GPtrArray *tokens,
 		struct rspamd_task *task)
 {
+	struct rspamd_lua_classifier_ctx *ctx;
+	struct rspamd_task **ptask;
+	struct rspamd_classifier_config **pcfg;
+	lua_State *L;
+	rspamd_token_t *tok;
+	guint i;
+	guint64 v;
+
+	ctx = g_hash_table_lookup (lua_classifiers, cl->subrs->name);
+	g_assert (ctx != NULL);
+	L = task->cfg->lua_state;
+
+	lua_rawgeti (L, LUA_REGISTRYINDEX, ctx->classify_ref);
+	ptask = lua_newuserdata (L, sizeof (*ptask));
+	*ptask = task;
+	rspamd_lua_setclass (L, "rspamd{task}", -1);
+	pcfg = lua_newuserdata (L, sizeof (*pcfg));
+	*pcfg = cl->cfg;
+	rspamd_lua_setclass (L, "rspamd{classifier}", -1);
+
+	lua_createtable (L, tokens->len, 0);
+
+	for (i = 0; i < tokens->len; i ++) {
+		tok = g_ptr_array_index (tokens, i);
+		v = 0;
+		memcpy (&v, tok->data, MIN (sizeof (v), tok->datalen));
+		lua_createtable (L, 3, 0);
+		/* High word, low word, order */
+		lua_pushnumber (L, (guint32)(v >> 32));
+		lua_rawseti (L, -2, 1);
+		lua_pushnumber (L, (guint32)(v));
+		lua_rawseti (L, -2, 2);
+		lua_pushnumber (L, tok->window_idx);
+		lua_rawseti (L, -2, 3);
+		lua_rawseti (L, -2, i + 1);
+	}
+
+	if (lua_pcall (L, 3, 0, 0) != 0) {
+		msg_err_luacl ("error running classify function for %s: %s", ctx->name,
+				lua_tostring (L, -1));
+		lua_pop (L, 1);
+
+		return FALSE;
+	}
+
 	return TRUE;
 }
 
 gboolean
-lua_classifier_learn_spam (struct rspamd_classifier *ctx,
+lua_classifier_learn_spam (struct rspamd_classifier *cl,
 		GPtrArray *tokens,
 		struct rspamd_task *task,
 		gboolean is_spam,