From e47feeffd618310f570917ac63e6f4eb7d5e8895 Mon Sep 17 00:00:00 2001
From: andryyy <andre.peters@debinux.de>
Date: Fri, 18 Aug 2017 22:17:01 +0200
Subject: [PATCH] [Rspamd] Add custom directory for own files
---
data/conf/rspamd/custom/.empty | 1 +
data/conf/rspamd/custom/ratelimit.lua | 66 +++++++++++++++++++++++++++
2 files changed, 67 insertions(+)
create mode 100644 data/conf/rspamd/custom/.empty
create mode 100644 data/conf/rspamd/custom/ratelimit.lua
diff --git a/data/conf/rspamd/custom/.empty b/data/conf/rspamd/custom/.empty
new file mode 100644
index 00000000..d00491fd
--- /dev/null
+++ b/data/conf/rspamd/custom/.empty
@@ -0,0 +1 @@
+1
diff --git a/data/conf/rspamd/custom/ratelimit.lua b/data/conf/rspamd/custom/ratelimit.lua
new file mode 100644
index 00000000..ed6b1bd6
--- /dev/null
+++ b/data/conf/rspamd/custom/ratelimit.lua
@@ -0,0 +1,66 @@
+local custom_keywords = {
+ ['customrl'] = {},
+}
+function custom_keywords.customrl.get_value(task)
+ local rspamd_logger = require "rspamd_logger"
+ local rspamd_redis = require "rspamd_redis"
+ local rspamd_regexp = require "rspamd_regexp"
+ local re = rspamd_regexp.create('/^\\s*$/i')
+ local envfrom = task:get_from(1)
+ local env_from_addr = envfrom[1].addr:lower() -- get smtp from addr in lower case
+ local env_from_domain = envfrom[1].domain:lower() -- get smtp from domain in lower case
+
+ local function rlo(object) -- get ratelimited object
+ local rlobj = string.format('%s', object)
+
+ local rl_ret, rl_obj = rspamd_redis.make_request_sync({host="172.22.1.249:6379", cmd='HGET', args={'RL_OBJECT', rlobj}, timeout=2.0})
+
+ if rl_ret and rl_obj then
+ return rl_obj
+ else
+ return false
+ end
+ end
+
+ rl_addr = rlo(env_from_addr)
+ rl_domain = rlo(env_from_domain)
+ if type(rl_addr) == 'string' and not re:match(rl_addr) then
+ rspamd_logger.infox(rspamd_config, "returning ratelimit object for %s", env_from_addr)
+ return rl_addr
+ elseif type(rl_domain) == 'string' and not re:match(rl_domain) then
+ rspamd_logger.infox(rspamd_config, "returning ratelimit object for %s", env_from_domain)
+ return rl_domain
+ end
+end
+function custom_keywords.customrl.get_limit(task)
+ local rspamd_logger = require "rspamd_logger"
+ local rspamd_redis = require "rspamd_redis"
+ local rspamd_regexp = require "rspamd_regexp"
+ local re = rspamd_regexp.create('/^\\s*$/i')
+ local envfrom = task:get_from(1)
+ local env_from_addr = envfrom[1].addr:lower() -- get smtp from addr in lower case
+ local env_from_domain = envfrom[1].domain:lower() -- get smtp from domain in lower case
+
+ local function rlv(object) -- get ratelimited object
+ local rlobj = string.format('%s', object)
+
+ local rl_ret, rl_value = rspamd_redis.make_request_sync({host="172.22.1.249:6379", cmd='HGET', args={'RL_VALUE', rlobj}, timeout=2.0})
+
+ if rl_ret and rl_value then
+ return rl_value
+ else
+ return false
+ end
+ end
+
+ rl_addr = rlv(env_from_addr)
+ rl_domain = rlv(env_from_domain)
+ if type(rl_addr) == 'string' and not re:match(rl_addr) then
+ rspamd_logger.infox(rspamd_config, "returning ratelimit %s for %s", rl_addr, env_from_addr)
+ return rl_addr
+ elseif type(rl_domain) == 'string' and not re:match(rl_domain) then
+ rspamd_logger.infox(rspamd_config, "returning ratelimit %s for %s", rl_domain, env_from_domain)
+ return rl_domain
+ end
+end
+return custom_keywords