[Minor] Set protocol flags in self-scan mode

diff --git a/src/rspamd_proxy.c b/src/rspamd_proxy.c
index 70e393dddc91666e8a5de4265d6955c7506d57ce..086dfd2c190ac7429b34d53b8b767182078912b0 100644 (file)
--- 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) {