From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Sat, 17 Nov 2018 13:30:26 +0000 (+0000)
Subject: [Minor] Add received tokens
X-Git-Tag: 1.8.2~6
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=820021f67c80d21ba9ebb332ee550a7d346153dd;p=thirdparty%2Frspamd.git

[Minor] Add received tokens
---

diff --git a/lualib/lua_stat.lua b/lualib/lua_stat.lua
index 2fb633cd38..21870b27d0 100644
--- a/lualib/lua_stat.lua
+++ b/lualib/lua_stat.lua
@@ -766,6 +766,25 @@ local function get_meta_stat_tokens(task, res, i)
     i = i + 1
   end
 
+  local rh = task:get_received_headers()
+
+  if rh and #rh > 0 then
+    local lim = math.min(5, #rh)
+    for j =1,lim do
+      local rcvd = rh[j]
+      local ip = rcvd.real_ip
+      if ip and ip:is_valid() and ip:get_version() == 4 then
+        local masked = ip:apply_mask(24)
+
+        rawset(res, i, string.format("#rcv:%s:%s", tostring(masked),
+            rcvd.proto))
+        lua_util.debugm("bayes", task, "added received token: %s",
+            res[i])
+        i = i + 1
+      end
+    end
+  end
+
   return i
 end