From: Vsevolod Stakhov <vsevolod@rspamd.com>
Date: Thu, 1 Feb 2024 13:25:11 +0000 (+0000)
Subject: [Minor] Support keepalive and authorization
X-Git-Tag: 3.8.2~5^2
X-Git-Url: https://source.dussan.org/?a=commitdiff_plain;h=refs%2Fpull%2F4806%2Fhead;p=rspamd.git

[Minor] Support keepalive and authorization
---

diff --git a/lualib/lua_scanners/kaspersky_se.lua b/lualib/lua_scanners/kaspersky_se.lua
index ec9589ac6..10a3e0f47 100644
--- a/lualib/lua_scanners/kaspersky_se.lua
+++ b/lualib/lua_scanners/kaspersky_se.lua
@@ -48,6 +48,8 @@ local function kaspersky_se_config(opts)
     scan_mime_parts = true,
     scan_text_mime = false,
     scan_image_mime = false,
+    keepalive = true,
+    auth_string = nil
   }
 
   default_conf = lua_util.override_defaults(default_conf, opts)
@@ -123,6 +125,10 @@ local function kaspersky_se_check(task, content, digest, rule, maybe_part)
       hdrs['X-KAV-HostIP'] = tostring(ip)
     end
 
+    if rule.auth_string then
+      hdrs['Authorization'] = rule.auth_string
+    end
+
     if task:has_from() then
       hdrs['X-KAV-ObjectURL'] = string.format('[from:%s]', task:get_from()[1].addr)
     end
@@ -163,6 +169,7 @@ local function kaspersky_se_check(task, content, digest, rule, maybe_part)
       body = req_body,
       headers = hdrs,
       timeout = rule.timeout,
+      keepalive = rule.keepalive,
     }
 
     local function kas_callback(http_err, code, body, headers)