Fix issues with connection closing.

diff --git a/src/controller.c b/src/controller.c
index d27a2d1a1fe4ae40c1b3e1cf21b1dca8307de1d5..e135ecdea2f0d9cf25127894a0139ecc9924c4e7 100644 (file)
--- a/src/controller.c
+++ b/src/controller.c
@@ -843,6 +843,7 @@ rspamd_controller_handle_learn_common (
        session->task = task;
        session->cl = cl;
        session->is_spam = is_spam;
+       check_session_pending (task->s);
 
        return 0;
 }
@@ -930,6 +931,7 @@ rspamd_controller_handle_scan (struct rspamd_http_connection_entry *conn_ent,
        }
 
        session->task = task;
+       check_session_pending (task->s);
 
        return 0;
 }

diff --git a/src/libserver/task.c b/src/libserver/task.c
index 8eb0c7932306d85a4dfda91a3007eb3ecf7fb236..e03790f7b1a2bc5c12ed2e54a2bbfb117d9228b9 100644 (file)
--- a/src/libserver/task.c
+++ b/src/libserver/task.c
@@ -344,8 +344,6 @@ rspamd_task_process (struct rspamd_task *task,
                task->state = WAIT_PRE_FILTER;
        }
 
-       check_session_pending (task->s);
-
        return TRUE;
 }