commit b8eb9a5: [Feature] milter_headers: support not removing headers

[Andrew Lewis]

Author: Andrew Lewis
Date: 2024-06-04 11:38:46 +0200
URL: https://github.com/rspamd/rspamd/commit/b8eb9a516b8314d71b2c7a9117a76044d068e485

[Feature] milter_headers: support not removing headers

---
 lualib/lua_mime.lua | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/lualib/lua_mime.lua b/lualib/lua_mime.lua
index 0f5aa75c0..ea6bf5125 100644
--- a/lualib/lua_mime.lua
+++ b/lualib/lua_mime.lua
@@ -582,8 +582,11 @@ exports.modify_headers = function(task, hdr_alterations, mode)
       hdr_flattened[hname].remove = {}
     end
     local remove_tbl = hdr_flattened[hname].remove
-    if type(hdr) == 'number' then
+    local t_hdr = type(hdr)
+    if t_hdr == 'number' then
       table.insert(remove_tbl, hdr)
+    elseif t_hdr == 'userdata' then
+      hdr_alterations.remove[hname] = nil
     else
       for _, num in ipairs(hdr) do
         table.insert(remove_tbl, num)