From: ventulis <72085617+ventulis@users.noreply.github.com>
Date: Tue, 24 Nov 2020 16:00:53 +0000 (+0100)
Subject: [Minor] Add task uid to ICAP User Agent
X-Git-Tag: 2.7~117^2
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=refs%2Fpull%2F3550%2Fhead;p=thirdparty%2Frspamd.git

[Minor] Add task uid to ICAP User Agent
---

diff --git a/lualib/lua_scanners/icap.lua b/lualib/lua_scanners/icap.lua
index 5fb332dc04..9efe1c14ce 100644
--- a/lualib/lua_scanners/icap.lua
+++ b/lualib/lua_scanners/icap.lua
@@ -88,10 +88,6 @@ local function icap_config(opts)
     icap_conf.servers,
     icap_conf.default_port)
 
-  if icap_conf.user_agent == "extended" then
-    icap_conf.user_agent = string.format("Rspamd/%s-%s (%s)", rspamd_version('main'), rspamd_version('id'), rspamd_util.get_hostname())
-  end
-
   if icap_conf.upstreams then
     lua_util.add_debug_alias('external_services', icap_conf.name)
     return icap_conf
@@ -109,6 +105,11 @@ local function icap_check(task, content, digest, rule)
     local retransmits = rule.retransmits
     local respond_headers = {}
 
+    -- Build extended User Agent
+    if rule.user_agent == "extended" then
+      rule.user_agent = string.format("Rspamd/%s-%s (%s/%s)", rspamd_version('main'), rspamd_version('id'), rspamd_util.get_hostname(), string.sub(task:get_uid(), 1,6))
+    end
+
     -- Build the icap queries
     local options_request = {
       string.format("OPTIONS icap://%s:%s/%s ICAP/1.0\r\n", addr:to_string(), addr:get_port(), rule.scheme),