Vsevolod Stakhov
9 years ago
1 changed files with 0 additions and 83 deletions
+ 0
- 83
src/plugins/lua/whitelist.lua
View File
| @@ -1,86 +0,0 @@ | |||
| local symbol_ip = nil | |||
| local symbol_from = nil | |||
| local r = nil | |||
| local h = nil -- radix tree and hash table | |||
| local rspamd_logger = require "rspamd_logger" | |||
| local function check_whitelist (task) | |||
| if symbol_ip then | |||
| -- check client's ip | |||
| local ipn = task:get_from_ip():to_number() | |||
| if ipn then | |||
| local key = r:get_key(ipn) | |||
| if key then | |||
| task:insert_result( symbol_ip, 1) | |||
| end | |||
| end | |||
| end | |||
| if symbol_from then | |||
| -- check client's from domain | |||
| local from = task:get_from() | |||
| if from then | |||
| local from_addr = from[1]['addr'] | |||
| if from_addr then | |||
| local _,_,domain = string.find(from_addr, '@(.+)>?$') | |||
| local key = h:get_key(domain) | |||
| if key then | |||
| task:insert_result(symbol_from, 1) | |||
| end | |||
| end | |||
| end | |||
| end | |||
| end | |||
| if type(rspamd_config.get_api_version) ~= 'nil' then | |||
| if rspamd_config:get_api_version() >= 1 then | |||
| rspamd_config:register_module_option('whitelist', 'symbol_ip', 'string') | |||
| rspamd_config:register_module_option('whitelist', 'symbol_from', 'string') | |||
| rspamd_config:register_module_option('whitelist', 'ip_whitelist', 'map') | |||
| rspamd_config:register_module_option('whitelist', 'from_whitelist', 'map') | |||
| end | |||
| end | |||
| local opts = rspamd_config:get_all_opt('whitelist') | |||
| if opts then | |||
| if opts['symbol_ip'] or opts['symbol_from'] then | |||
| local symbol_ip = opts['symbol_ip'] | |||
| local 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_hash_map (opts['from_whitelist']) | |||
| else | |||
| -- No whitelist defined | |||
| symbol_from = nil | |||
| end | |||
| end | |||
| -- Register symbol's callback | |||
| if symbol_ip or symbol_from then | |||
| rspamd_config:register_callback_symbol_priority('WHITELIST', 1.0, -1, check_whitelist) | |||
| if symbol_from then | |||
| rspamd_config:register_virtual_symbol(symbol_from, 1.0) | |||
| end | |||
| if symbol_ip then | |||
| rspamd_config:register_virtual_symbol(symbol_ip, 1.0) | |||
| end | |||
| end | |||
| end | |||
| end | |||
Loading…