trAvis - MANAGER

Edit File: http_headers.lua

--[[ Copyright (c) 2022, Vsevolod Stakhov <vsevolod@rspamd.com> 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. ]]-- local logger = require "rspamd_logger" local ucl = require "ucl" local spf_symbols = { symbol_allow = 'R_SPF_ALLOW', symbol_deny = 'R_SPF_FAIL', symbol_softfail = 'R_SPF_SOFTFAIL', symbol_neutral = 'R_SPF_NEUTRAL', symbol_tempfail = 'R_SPF_DNSFAIL', symbol_na = 'R_SPF_NA', symbol_permfail = 'R_SPF_PERMFAIL', } local dkim_symbols = { symbol_allow = 'R_DKIM_ALLOW', symbol_deny = 'R_DKIM_REJECT', symbol_tempfail = 'R_DKIM_TEMPFAIL', symbol_na = 'R_DKIM_NA', symbol_permfail = 'R_DKIM_PERMFAIL', symbol_trace = 'DKIM_TRACE', } local dkim_trace = { pass = '+', fail = '-', temperror = '?', permerror = '~', } local dmarc_symbols = { allow = 'DMARC_POLICY_ALLOW', badpolicy = 'DMARC_BAD_POLICY', dnsfail = 'DMARC_DNSFAIL', na = 'DMARC_NA', reject = 'DMARC_POLICY_REJECT', softfail = 'DMARC_POLICY_SOFTFAIL', quarantine = 'DMARC_POLICY_QUARANTINE', } local opts = rspamd_config:get_all_opt('dmarc') if opts and opts['symbols'] then for k, _ in pairs(dmarc_symbols) do if opts['symbols'][k] then dmarc_symbols[k] = opts['symbols'][k] end end end opts = rspamd_config:get_all_opt('dkim') if opts then for k, _ in pairs(dkim_symbols) do if opts[k] then dkim_symbols[k] = opts[k] end end end opts = rspamd_config:get_all_opt('spf') if opts then for k, _ in pairs(spf_symbols) do if opts[k] then spf_symbols[k] = opts[k] end end end -- Disable DKIM checks if passed via HTTP headers rspamd_config:add_condition("DKIM_CHECK", function(task) local hdr = task:get_request_header('DKIM') if hdr then local parser = ucl.parser() local res, err = parser:parse_string(tostring(hdr)) if not res then logger.infox(task, "cannot parse DKIM header: %1", err) return true end local p_obj = parser:get_object() local results = p_obj['results'] if not results and p_obj['result'] then results = { { result = p_obj['result'], domain = 'unknown' } } end if results then for _, obj in ipairs(results) do local dkim_domain = obj['domain'] or 'unknown' if obj['result'] == 'pass' or obj['result'] == 'allow' then task:insert_result(dkim_symbols['symbol_allow'], 1.0, 'http header') task:insert_result(dkim_symbols['symbol_trace'], 1.0, string.format('%s:%s', dkim_domain, dkim_trace.pass)) elseif obj['result'] == 'fail' or obj['result'] == 'reject' then task:insert_result(dkim_symbols['symbol_deny'], 1.0, 'http header') task:insert_result(dkim_symbols['symbol_trace'], 1.0, string.format('%s:%s', dkim_domain, dkim_trace.fail)) elseif obj['result'] == 'tempfail' or obj['result'] == 'softfail' then task:insert_result(dkim_symbols['symbol_tempfail'], 1.0, 'http header') task:insert_result(dkim_symbols['symbol_trace'], 1.0, string.format('%s:%s', dkim_domain, dkim_trace.temperror)) elseif obj['result'] == 'permfail' then task:insert_result(dkim_symbols['symbol_permfail'], 1.0, 'http header') task:insert_result(dkim_symbols['symbol_trace'], 1.0, string.format('%s:%s', dkim_domain, dkim_trace.permerror)) elseif obj['result'] == 'na' then task:insert_result(dkim_symbols['symbol_na'], 1.0, 'http header') end end end end return false end) -- Disable SPF checks if passed via HTTP headers rspamd_config:add_condition("SPF_CHECK", function(task) local hdr = task:get_request_header('SPF') if hdr then local parser = ucl.parser() local res, err = parser:parse_string(tostring(hdr)) if not res then logger.infox(task, "cannot parse SPF header: %1", err) return true end local obj = parser:get_object() if obj['result'] then if obj['result'] == 'pass' or obj['result'] == 'allow' then task:insert_result(spf_symbols['symbol_allow'], 1.0, 'http header') elseif obj['result'] == 'fail' or obj['result'] == 'reject' then task:insert_result(spf_symbols['symbol_deny'], 1.0, 'http header') elseif obj['result'] == 'neutral' then task:insert_result(spf_symbols['symbol_neutral'], 1.0, 'http header') elseif obj['result'] == 'softfail' then task:insert_result(spf_symbols['symbol_softfail'], 1.0, 'http header') elseif obj['result'] == 'permfail' then task:insert_result(spf_symbols['symbol_permfail'], 1.0, 'http header') elseif obj['result'] == 'na' then task:insert_result(spf_symbols['symbol_na'], 1.0, 'http header') end end end return false end) rspamd_config:add_condition("DMARC_CALLBACK", function(task) local hdr = task:get_request_header('DMARC') if hdr then local parser = ucl.parser() local res, err = parser:parse_string(tostring(hdr)) if not res then logger.infox(task, "cannot parse DMARC header: %1", err) return true end local obj = parser:get_object() if obj['result'] then if obj['result'] == 'pass' or obj['result'] == 'allow' then task:insert_result(dmarc_symbols['allow'], 1.0, 'http header') elseif obj['result'] == 'fail' or obj['result'] == 'reject' then task:insert_result(dmarc_symbols['reject'], 1.0, 'http header') elseif obj['result'] == 'quarantine' then task:insert_result(dmarc_symbols['quarantine'], 1.0, 'http header') elseif obj['result'] == 'tempfail' then task:insert_result(dmarc_symbols['dnsfail'], 1.0, 'http header') elseif obj['result'] == 'softfail' or obj['result'] == 'none' then task:insert_result(dmarc_symbols['softfail'], 1.0, 'http header') elseif obj['result'] == 'permfail' or obj['result'] == 'badpolicy' then task:insert_result(dmarc_symbols['badpolicy'], 1.0, 'http header') elseif obj['result'] == 'na' then task:insert_result(dmarc_symbols['na'], 1.0, 'http header') end end end return false end)