Fix hang on processing non-mime messages.

diff --git a/src/libserver/task.c b/src/libserver/task.c
index 61e0befc7a9b4cc7d8008edb09b3ff8b9d414252..8c9f8b282a481fe5a795f89195c23ad4f835dbbd 100644 (file)
--- a/src/libserver/task.c
+++ b/src/libserver/task.c
@@ -472,6 +472,11 @@ rspamd_task_process (struct rspamd_task *task, guint stages)
        task->flags &= ~RSPAMD_TASK_FLAG_PROCESSING;
 
        if (!ret || RSPAMD_TASK_IS_PROCESSED (task)) {
+               if (!ret) {
+                       /* Set processed flags */
+                       task->processed_stages |= RSPAMD_TASK_STAGE_DONE;
+               }
+
                return ret;
        }