From: Andrew Lewis <nerf@judo.za.org>
Date: Sun, 9 Jul 2017 10:21:31 +0000 (+0200)
Subject: [Feature] Milter headers: support adding/removing arbitrary headers from config
X-Git-Tag: 1.7.0~874
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=b29b6ee49c9b3f16cf5adf5cc49065aac2f72282;p=thirdparty%2Frspamd.git

[Feature] Milter headers: support adding/removing arbitrary headers from config
---

diff --git a/src/plugins/lua/milter_headers.lua b/src/plugins/lua/milter_headers.lua
index dcf8e9ffdf..52cd874a67 100644
--- a/src/plugins/lua/milter_headers.lua
+++ b/src/plugins/lua/milter_headers.lua
@@ -35,6 +35,13 @@ local settings = {
   authenticated_headers = {},
   extended_headers_rcpt = {},
   routines = {
+    ['remove-headers'] = {
+      headers = {},
+    },
+    ['add-headers'] = {
+      headers = {},
+      remove = 1,
+    },
     ['remove-header'] = {
       remove = 1,
     },
@@ -201,6 +208,21 @@ local function milter_headers(task)
     end
   end
 
+  routines['remove-headers'] = function()
+    if skip_wanted('remove-headers') then return end
+    for h, r in pairs(settings.routines['remove-headers'].headers) do
+      remove[h] = r
+    end
+  end
+
+  routines['add-headers'] = function()
+    if skip_wanted('add-headers') then return end
+    for h, r in pairs(settings.routines['add-headers'].headers) do
+      add[h] = r
+      remove[h] = settings.routines['add-headers'].remove
+    end
+  end
+
   routines['x-rspamd-server'] = function()
     if skip_wanted('x-rspamd-server') then return end
     if settings.routines['x-rspamd-server'].remove then