diff options
Diffstat (limited to 'src/plugins/lua/once_received.lua')
| -rw-r--r-- | src/plugins/lua/once_received.lua | 21 |
1 files changed, 9 insertions, 12 deletions
diff --git a/src/plugins/lua/once_received.lua b/src/plugins/lua/once_received.lua index 80ef8cf29..7cbdacf50 100644 --- a/src/plugins/lua/once_received.lua +++ b/src/plugins/lua/once_received.lua @@ -124,31 +124,28 @@ end -- Configuration local opts = rspamd_config:get_all_opt('once_received') if opts then - if opts['symbol'] then - local symbol = opts['symbol'] + if opts['symbol'] then + local symbol = opts['symbol'] - for n,v in pairs(opts) do + local id = rspamd_config:register_symbol(symbol, 1.0, check_quantity_received) + + for n,v in pairs(opts) do if n == 'symbol_strict' then symbol_strict = v - if type(rspamd_config.get_api_version) ~= 'nil' then - rspamd_config:register_virtual_symbol(symbol_strict, 1.0) - end + rspamd_config:register_virtual_symbol(symbol_strict, 1.0, id) elseif n == 'bad_host' then if type(v) == 'string' then - bad_hosts[1] = v + bad_hosts[1] = v else bad_hosts = v end elseif n == 'good_host' then if type(v) == 'string' then - good_hosts[1] = v + good_hosts[1] = v else good_hosts = v end - end end - - -- Register symbol's callback - rspamd_config:register_symbol(symbol, 1.0, check_quantity_received) + end end end |