commit da57529: [Minor] Set protocol flags in self-scan mode

[Vsevolod Stakhov]

Author: Vsevolod Stakhov
Date: 2019-08-03 18:14:29 +0100
URL: https://github.com/rspamd/rspamd/commit/da57529be06ceb91fada005a3458a3be604ef271 (HEAD -> master)

[Minor] Set protocol flags in self-scan mode

---
 src/rspamd_proxy.c | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/rspamd_proxy.c b/src/rspamd_proxy.c
index 70e393ddd..086dfd2c1 100644
--- a/src/rspamd_proxy.c
+++ b/src/rspamd_proxy.c
@@ -1728,6 +1728,12 @@ rspamd_proxy_self_scan (struct rspamd_proxy_session *session)
 			session->pool, session->ctx->lang_det,
 			session->ctx->event_loop);
 	task->flags |= RSPAMD_TASK_FLAG_MIME;
+
+	if (session->ctx->milter) {
+		task->protocol_flags |= RSPAMD_TASK_PROTOCOL_FLAG_MILTER|
+				RSPAMD_TASK_PROTOCOL_FLAG_BODY_BLOCK;
+	}
+
 	task->sock = -1;
 
 	if (session->client_milter_conn) {