* Add ability to add maps from lua scripts and access theirs elements

* Add whitelist module for whitelisting score for some ip/from addresses

1 files changed, 73 insertions, 0 deletions

diff --git a/src/plugins/lua/whitelist.lua b/src/plugins/lua/whitelist.lua
new file mode 100644
index 000000000..e95aca6c7
--- /dev/null
+++ b/src/plugins/lua/whitelist.lua
@@ -0,0 +1,73 @@
+-- Module that add symbols to those hosts or from domains that are contained in whitelist
+
+local metric = 'default'
+local symbol_ip = nil
+local symbol_from = nil
+
+local r = nil
+local h = nil  -- radix tree and hash table
+
+function check_whitelist (task)
+	if symbol_ip then
+		-- check client's ip
+		local ipn = task:get_from_ip_num()
+		if ipn then
+			local key = r:get_key(ipn)
+			if key then
+				task:insert_result(metric, symbol_ip, 1)
+			end
+		end
+	end
+
+	if symbol_from then
+		-- check client's from domain
+		local from = task:get_from()
+		if from then
+			local _,_,domain = string.find(from, '@(.+)>?$')
+			local key = h:get_key(domain)
+			if key then
+				task:insert_result(metric, symbol_from, 1)
+			end
+		end
+	end
+
+end
+
+
+-- Configuration
+local opts =  rspamd_config:get_all_opt('whitelist')
+if opts then
+    if opts['symbol_ip'] or opts['symbol_from'] then
+        symbol_ip = opts['symbol_ip']
+        symbol_from = opts['symbol_from']
+		
+		if symbol_ip then
+			if opts['ip_whitelist'] then
+				r = rspamd_config:add_radix_map (opts['ip_whitelist'])
+			else
+				-- No whitelist defined
+				symbol_ip = nil
+			end
+		end
+		if symbol_from then
+			if opts['from_whitelist'] then
+				h = rspamd_config:add_host_map (opts['from_whitelist'])
+			else
+				-- No whitelist defined
+				symbol_from = nil
+			end
+		end
+
+		if opts['metric'] then
+			metric = opts['metric']
+		end
+
+		-- Register symbol's callback
+		local m = rspamd_config:get_metric(metric)
+		if symbol_ip then
+			m:register_symbol(symbol_ip, 1.0, 'check_whitelist')
+		elseif symbol_from then
+			m:register_symbol(symbol_from, 1.0, 'check_whitelist')
+		end
+	end
+end