|
@@ -1,387 +0,0 @@
|
|
|
---[[
|
|
|
-Copyright (c) 2016, Andrew Lewis <nerf@judo.za.org>
|
|
|
-Copyright (c) 2016, Vsevolod Stakhov <vsevolod@highsecure.ru>
|
|
|
-
|
|
|
-Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
-you may not use this file except in compliance with the License.
|
|
|
-You may obtain a copy of the License at
|
|
|
-
|
|
|
- http://www.apache.org/licenses/LICENSE-2.0
|
|
|
-
|
|
|
-Unless required by applicable law or agreed to in writing, software
|
|
|
-distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
-See the License for the specific language governing permissions and
|
|
|
-limitations under the License.
|
|
|
-]]--
|
|
|
-
|
|
|
-if confighelp then
|
|
|
- return
|
|
|
-end
|
|
|
-
|
|
|
--- A plugin that provides common header manipulations
|
|
|
-
|
|
|
-local logger = require "rspamd_logger"
|
|
|
-local util = require "rspamd_util"
|
|
|
-local N = 'milter_headers'
|
|
|
-local E = {}
|
|
|
-
|
|
|
-local HOSTNAME = util.get_hostname()
|
|
|
-
|
|
|
-local settings = {
|
|
|
- skip_local = false,
|
|
|
- skip_authenticated = false,
|
|
|
- routines = {
|
|
|
- ['x-spamd-result'] = {
|
|
|
- header = 'X-Spamd-Result',
|
|
|
- remove = 1,
|
|
|
- },
|
|
|
- ['x-rspamd-server'] = {
|
|
|
- header = 'X-Rspamd-Server',
|
|
|
- remove = 1,
|
|
|
- },
|
|
|
- ['x-rspamd-queue-id'] = {
|
|
|
- header = 'X-Rspamd-Queue-Id',
|
|
|
- remove = 1,
|
|
|
- },
|
|
|
- ['spam-header'] = {
|
|
|
- header = 'Deliver-To',
|
|
|
- value = 'Junk',
|
|
|
- remove = 1,
|
|
|
- },
|
|
|
- ['x-virus'] = {
|
|
|
- header = 'X-Virus',
|
|
|
- remove = 1,
|
|
|
- symbols = {}, -- needs config
|
|
|
- },
|
|
|
- ['x-spamd-bar'] = {
|
|
|
- header = 'X-Spamd-Bar',
|
|
|
- positive = '+',
|
|
|
- negative = '-',
|
|
|
- neutral = '/',
|
|
|
- remove = 1,
|
|
|
- },
|
|
|
- ['x-spam-level'] = {
|
|
|
- header = 'X-Spam-Level',
|
|
|
- char = '*',
|
|
|
- remove = 1,
|
|
|
- },
|
|
|
- ['x-spam-status'] = {
|
|
|
- header = 'X-Spam-Status',
|
|
|
- remove = 1,
|
|
|
- },
|
|
|
- ['authentication-results'] = {
|
|
|
- header = 'Authentication-Results',
|
|
|
- remove = 1,
|
|
|
- spf_symbols = {
|
|
|
- pass = 'R_SPF_ALLOW',
|
|
|
- fail = 'R_SPF_FAIL',
|
|
|
- softfail = 'R_SPF_SOFTFAIL',
|
|
|
- neutral = 'R_SPF_NEUTRAL',
|
|
|
- temperror = 'R_SPF_DNSFAIL',
|
|
|
- none = 'R_SPF_NA',
|
|
|
- permerror = 'R_SPF_PERMFAIL',
|
|
|
- },
|
|
|
- dkim_symbols = {
|
|
|
- pass = 'R_DKIM_ALLOW',
|
|
|
- fail = 'R_DKIM_REJECT',
|
|
|
- temperror = 'R_DKIM_TEMPFAIL',
|
|
|
- none = 'R_DKIM_NA',
|
|
|
- permerror = 'R_DKIM_PERMFAIL',
|
|
|
- },
|
|
|
- dmarc_symbols = {
|
|
|
- pass = 'DMARC_POLICY_ALLOW',
|
|
|
- permerror = 'DMARC_BAD_POLICY',
|
|
|
- temperror = 'DMARC_DNSFAIL',
|
|
|
- none = 'DMARC_NA',
|
|
|
- reject = 'DMARC_POLICY_REJECT',
|
|
|
- softfail = 'DMARC_POLICY_SOFTFAIL',
|
|
|
- quarantine = 'DMARC_POLICY_QUARANTINE',
|
|
|
- },
|
|
|
- },
|
|
|
- },
|
|
|
-}
|
|
|
-
|
|
|
-local active_routines = {}
|
|
|
-local custom_routines = {}
|
|
|
-
|
|
|
-local function milter_headers(task)
|
|
|
-
|
|
|
- if settings.skip_local then
|
|
|
- local ip = task:get_ip()
|
|
|
- if (ip and ip:is_local()) then return end
|
|
|
- end
|
|
|
-
|
|
|
- if settings.skip_authenticated then
|
|
|
- if task:get_user() ~= nil then return end
|
|
|
- end
|
|
|
-
|
|
|
- local routines, common, add, remove = {}, {}, {}, {}
|
|
|
-
|
|
|
- routines['x-spamd-result'] = function()
|
|
|
- if not common.symbols then
|
|
|
- common.symbols = task:get_symbols_all()
|
|
|
- common['metric_score'] = task:get_metric_score('default')
|
|
|
- common['metric_action'] = task:get_metric_action('default')
|
|
|
- end
|
|
|
- if settings.routines['x-spamd-result'].remove then
|
|
|
- remove[settings.routines['x-spamd-result'].header] = settings.routines['x-spamd-result'].remove
|
|
|
- end
|
|
|
- local buf = {}
|
|
|
- table.insert(buf, table.concat({
|
|
|
- 'default: ', (common['metric_action'] == 'reject') and 'True' or 'False', ' [',
|
|
|
- common['metric_score'][1], ' / ', common['metric_score'][2], ']'
|
|
|
- }))
|
|
|
- for _, s in ipairs(common.symbols) do
|
|
|
- if not s.options then s.options = {} end
|
|
|
- table.insert(buf, table.concat({
|
|
|
- ' ', s.name, ' (', s.score, ') [', table.concat(s.options, ','), ']',
|
|
|
- }))
|
|
|
- end
|
|
|
- add[settings.routines['x-spamd-result'].header] = table.concat(buf, '\n')
|
|
|
- end
|
|
|
-
|
|
|
- routines['x-rspamd-queue-id'] = function()
|
|
|
- if common.queue_id ~= false then
|
|
|
- common.queue_id = task:get_queue_id()
|
|
|
- if not common.queue_id then
|
|
|
- common.queue_id = false
|
|
|
- end
|
|
|
- end
|
|
|
- if settings.routines['x-rspamd-queue-id'].remove then
|
|
|
- remove[settings.routines['x-rspamd-queue-id'].header] = settings.routines['x-rspamd-queue-id'].remove
|
|
|
- end
|
|
|
- if common.queue_id then
|
|
|
- add[settings.routines['x-rspamd-queue-id'].header] = common.queue_id
|
|
|
- end
|
|
|
- end
|
|
|
-
|
|
|
- routines['x-rspamd-server'] = function()
|
|
|
- if settings.routines['x-rspamd-server'].remove then
|
|
|
- remove[settings.routines['x-rspamd-server'].header] = settings.routines['x-rspamd-server'].remove
|
|
|
- end
|
|
|
- add[settings.routines['x-rspamd-server'].header] = HOSTNAME
|
|
|
- end
|
|
|
-
|
|
|
- routines['x-spamd-bar'] = function()
|
|
|
- if not common['metric_score'] then
|
|
|
- common['metric_score'] = task:get_metric_score('default')
|
|
|
- end
|
|
|
- local score = common['metric_score'][1]
|
|
|
- local spambar
|
|
|
- if score <= -1 then
|
|
|
- spambar = string.rep(settings.routines['x-spamd-bar'].negative, score*-1)
|
|
|
- elseif score >= 1 then
|
|
|
- spambar = string.rep(settings.routines['x-spamd-bar'].positive, score)
|
|
|
- else
|
|
|
- spambar = settings.routines['x-spamd-bar'].neutral
|
|
|
- end
|
|
|
- if settings.routines['x-spamd-bar'].remove then
|
|
|
- remove[settings.routines['x-spamd-bar'].header] = settings.routines['x-spamd-bar'].remove
|
|
|
- end
|
|
|
- if spambar ~= '' then
|
|
|
- add[settings.routines['x-spamd-bar'].header] = spambar
|
|
|
- end
|
|
|
- end
|
|
|
-
|
|
|
- routines['x-spam-level'] = function()
|
|
|
- if not common['metric_score'] then
|
|
|
- common['metric_score'] = task:get_metric_score('default')
|
|
|
- end
|
|
|
- local score = common['metric_score'][1]
|
|
|
- if score < 1 then
|
|
|
- return nil, {}, {}
|
|
|
- end
|
|
|
- if settings.routines['x-spam-level'].remove then
|
|
|
- remove[settings.routines['x-spam-level'].header] = settings.routines['x-spam-level'].remove
|
|
|
- end
|
|
|
- add[settings.routines['x-spam-level'].header] = string.rep(settings.routines['x-spam-level'].char, score)
|
|
|
- end
|
|
|
-
|
|
|
- routines['spam-header'] = function()
|
|
|
- if not common['metric_action'] then
|
|
|
- common['metric_action'] = task:get_metric_action('default')
|
|
|
- end
|
|
|
- if settings.routines['spam-header'].remove then
|
|
|
- remove[settings.routines['spam-header'].header] = settings.routines['spam-header'].remove
|
|
|
- end
|
|
|
- local action = common['metric_action']
|
|
|
- if action ~= 'no action' and action ~= 'greylist' then
|
|
|
- add[settings.routines['spam-header'].header] = settings.routines['spam-header'].value
|
|
|
- end
|
|
|
- end
|
|
|
-
|
|
|
- routines['x-virus'] = function()
|
|
|
- if not common.symbols then
|
|
|
- common.symbols = {}
|
|
|
- end
|
|
|
- if settings.routines['x-virus'].remove then
|
|
|
- remove[settings.routines['x-virus'].header] = settings.routines['x-virus'].remove
|
|
|
- end
|
|
|
- local virii = {}
|
|
|
- for _, sym in ipairs(settings.routines['x-virus'].symbols) do
|
|
|
- if not (common.symbols[sym] == false) then
|
|
|
- local s = task:get_symbol(sym)
|
|
|
- if not s then
|
|
|
- common.symbols[sym] = false
|
|
|
- else
|
|
|
- common.symbols[sym] = s
|
|
|
- if (((s or E)[1] or E).options or E)[1] then
|
|
|
- table.insert(virii, s[1].options[1])
|
|
|
- else
|
|
|
- table.insert(virii, 'unknown')
|
|
|
- end
|
|
|
- end
|
|
|
- end
|
|
|
- end
|
|
|
- if #virii > 0 then
|
|
|
- add[settings.routines['x-virus'].header] = table.concat(virii, ',')
|
|
|
- end
|
|
|
- end
|
|
|
-
|
|
|
- routines['x-spam-status'] = function()
|
|
|
- if not common['metric_score'] then
|
|
|
- common['metric_score'] = task:get_metric_score('default')
|
|
|
- end
|
|
|
- if not common['metric_action'] then
|
|
|
- common['metric_action'] = task:get_metric_action('default')
|
|
|
- end
|
|
|
- local score = common['metric_score'][1]
|
|
|
- local action = common['metric_action']
|
|
|
- local is_spam
|
|
|
- local spamstatus
|
|
|
- if action ~= 'no action' and action ~= 'greylist' then
|
|
|
- is_spam = 'Yes'
|
|
|
- else
|
|
|
- is_spam = 'No'
|
|
|
- end
|
|
|
- spamstatus = is_spam .. ', score=' .. string.format('%.2f', score)
|
|
|
- if settings.routines['x-spam-status'].remove then
|
|
|
- remove[settings.routines['x-spam-status'].header] = settings.routines['x-spam-status'].remove
|
|
|
- end
|
|
|
- add[settings.routines['x-spam-status'].header] = spamstatus
|
|
|
- end
|
|
|
-
|
|
|
- routines['authentication-results'] = function()
|
|
|
- local ar = require "auth_results"
|
|
|
-
|
|
|
- if settings.routines['authentication-results'].remove then
|
|
|
- remove[settings.routines['authentication-results'].header] =
|
|
|
- settings.routines['authentication-results'].remove
|
|
|
- end
|
|
|
-
|
|
|
- local res = ar.gen_auth_results(task,
|
|
|
- settings.routines['authentication-results'])
|
|
|
-
|
|
|
- if res then
|
|
|
- add[settings.routines['authentication-results'].header] = res
|
|
|
- end
|
|
|
- end
|
|
|
-
|
|
|
- for _, n in ipairs(active_routines) do
|
|
|
- local ok, err
|
|
|
- if custom_routines[n] then
|
|
|
- local to_add, to_remove, common_in
|
|
|
- ok, err, to_add, to_remove, common_in = pcall(custom_routines[n], task, common)
|
|
|
- if ok then
|
|
|
- for k, v in pairs(to_add) do
|
|
|
- add[k] = v
|
|
|
- end
|
|
|
- for k, v in pairs(to_remove) do
|
|
|
- add[k] = v
|
|
|
- end
|
|
|
- for k, v in pairs(common_in) do
|
|
|
- if type(v) == 'table' then
|
|
|
- if not common[k] then
|
|
|
- common[k] = {}
|
|
|
- end
|
|
|
- for kk, vv in pairs(v) do
|
|
|
- common[k][kk] = vv
|
|
|
- end
|
|
|
- else
|
|
|
- common[k] = v
|
|
|
- end
|
|
|
- end
|
|
|
- end
|
|
|
- else
|
|
|
- ok, err = pcall(routines[n])
|
|
|
- end
|
|
|
- if not ok then
|
|
|
- logger.errx(task, 'call to %s failed: %s', n, err)
|
|
|
- end
|
|
|
- end
|
|
|
-
|
|
|
- if not next(add) then add = nil end
|
|
|
- if not next(remove) then remove = nil end
|
|
|
- if add or remove then
|
|
|
- task:set_milter_reply({
|
|
|
- add_headers = add,
|
|
|
- remove_headers = remove
|
|
|
- })
|
|
|
- end
|
|
|
-end
|
|
|
-
|
|
|
-local opts = rspamd_config:get_all_opt(N) or rspamd_config:get_all_opt('rmilter_headers')
|
|
|
-if not opts then return end
|
|
|
-
|
|
|
-if type(opts['use']) == 'string' then
|
|
|
- opts['use'] = {opts['use']}
|
|
|
-elseif (type(opts['use']) == 'table' and not opts['use'][1]) then
|
|
|
- logger.debugm(N, rspamd_config, 'no functions are enabled')
|
|
|
- return
|
|
|
-end
|
|
|
-if type(opts['use']) ~= 'table' then
|
|
|
- logger.errx(rspamd_config, 'unexpected type for "use" option: %s', type(opts['use']))
|
|
|
- return
|
|
|
-end
|
|
|
-if type(opts['custom']) == 'table' then
|
|
|
- for k, v in pairs(opts['custom']) do
|
|
|
- local f, err = load(v)
|
|
|
- if not f then
|
|
|
- logger.errx(rspamd_config, 'could not load "%s": %s', k, err)
|
|
|
- else
|
|
|
- custom_routines[k] = f()
|
|
|
- end
|
|
|
- end
|
|
|
-end
|
|
|
-local have_routine = {}
|
|
|
-local function activate_routine(s)
|
|
|
- if settings.routines[s] or custom_routines[s] then
|
|
|
- have_routine[s] = true
|
|
|
- table.insert(active_routines, s)
|
|
|
- if (opts.routines and opts.routines[s]) then
|
|
|
- for k, v in pairs(opts.routines[s]) do
|
|
|
- settings.routines[s][k] = v
|
|
|
- end
|
|
|
- end
|
|
|
- else
|
|
|
- logger.errx(rspamd_config, 'routine "%s" does not exist', s)
|
|
|
- end
|
|
|
-end
|
|
|
-if opts['extended_spam_headers'] then
|
|
|
- activate_routine('x-spamd-result')
|
|
|
- activate_routine('x-rspamd-server')
|
|
|
- activate_routine('x-rspamd-queue-id')
|
|
|
-end
|
|
|
-if opts['skip_local'] then
|
|
|
- settings.skip_local = true
|
|
|
-end
|
|
|
-if opts['skip_authenticated'] then
|
|
|
- settings.skip_authenticated = true
|
|
|
-end
|
|
|
-for _, s in ipairs(opts['use']) do
|
|
|
- if not have_routine[s] then
|
|
|
- activate_routine(s)
|
|
|
- end
|
|
|
-end
|
|
|
-if (#active_routines < 1) then
|
|
|
- logger.errx(rspamd_config, 'no active routines')
|
|
|
- return
|
|
|
-end
|
|
|
-logger.infox(rspamd_config, 'active routines [%s]', table.concat(active_routines, ','))
|
|
|
-rspamd_config:register_symbol({
|
|
|
- name = 'MILTER_HEADERS',
|
|
|
- type = 'postfilter',
|
|
|
- callback = milter_headers,
|
|
|
- priority = 10
|
|
|
-})
|