]> source.dussan.org Git - rspamd.git/commitdiff
[Minor] Support keepalive and authorization 4806/head
authorVsevolod Stakhov <vsevolod@rspamd.com>
Thu, 1 Feb 2024 13:25:11 +0000 (13:25 +0000)
committerVsevolod Stakhov <vsevolod@rspamd.com>
Thu, 1 Feb 2024 13:25:11 +0000 (13:25 +0000)
lualib/lua_scanners/kaspersky_se.lua

index ec9589ac62604090b1635cddb6dce476a684ab5e..10a3e0f475cd60c5104de4d3492a560d26314981 100644 (file)
@@ -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)