From: Michael Kuron <m.kuron@gmx.de>
Date: Fri, 1 Sep 2017 17:44:48 +0000 (+0200)
Subject: User settings: add header rules
X-Git-Tag: 1.7.0~681^2
X-Git-Url: https://source.dussan.org/?a=commitdiff_plain;h=1b320f794510d741771de2c5dea1dcd35d5ff835;p=rspamd.git

User settings: add header rules
---

diff --git a/src/plugins/lua/settings.lua b/src/plugins/lua/settings.lua
index 8f75d15c3..7d28f2545 100644
--- a/src/plugins/lua/settings.lua
+++ b/src/plugins/lua/settings.lua
@@ -250,6 +250,19 @@ local function check_settings(task)
       end
     end
 
+    if rule['header'] then
+      for k, v in pairs(rule['header']) do
+        local h = task:get_header(k)
+        res = (h and v:match(h))
+        if res then
+          break
+        end
+      end
+      if not res then
+        return nil
+      end
+    end
+
     if res then
       if rule['whitelist'] then
         rule['apply'] = {whitelist = true}
@@ -483,6 +496,19 @@ local function process_settings_table(tbl)
       end
       out['request_header'] = rho
     end
+    if elt['header'] then
+      local rho = {}
+      for k, v in pairs(elt['header']) do
+        local re = rspamd_regexp.get_cached(v)
+        if not re then
+          re = rspamd_regexp.create_cached(v)
+        end
+        if re then
+          rho[k] = re
+        end
+      end
+      out['header'] = rho
+    end
 
     -- Now we must process actions
     if elt['symbols'] then out['symbols'] = elt['symbols'] end