From 820021f67c80d21ba9ebb332ee550a7d346153dd Mon Sep 17 00:00:00 2001
From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Sat, 17 Nov 2018 13:30:26 +0000
Subject: [PATCH] [Minor] Add received tokens

---
 lualib/lua_stat.lua | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/lualib/lua_stat.lua b/lualib/lua_stat.lua
index 2fb633cd3..21870b27d 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
 
-- 
2.39.5