[Minor] Set processed stage when called from Lua

diff --git a/src/lua/lua_task.c b/src/lua/lua_task.c
index 4293ef1781dc5770ee616624b75d62f26a19f940..c3159f16bf9c0ae9539521a824d68729615b6768 100644 (file)
--- a/src/lua/lua_task.c
+++ b/src/lua/lua_task.c
@@ -1330,8 +1330,14 @@ lua_task_process_message (lua_State *L)
        if (task != NULL) {
                if (task->msg.len > 0) {
                        if (rspamd_message_parse (task)) {
-                               lua_pushboolean (L, TRUE);
-                               rspamd_message_process (task);
+                               if (!(task->flags & RSPAMD_TASK_FLAG_SKIP_PROCESS)) {
+                                       lua_pushboolean (L, TRUE);
+                                       rspamd_message_process (task);
+                                       task->processed_stages |= RSPAMD_TASK_STAGE_PROCESS_MESSAGE;
+                               }
+                               else {
+                                       lua_pushboolean (L, FALSE);
+                               }
                        }
                        else {
                                lua_pushboolean (L, FALSE);