Gecko [ www.jdcpatala.org ]

Name	Size	Permission
antivirus.lua	11.24 KB	-rw-r--r--
arc.lua	26.72 KB	-rw-r--r--
asn.lua	4.78 KB	-rw-r--r--
aws_s3.lua	7.93 KB	-rw-r--r--
bayes_expiry.lua	14.21 KB	-rw-r--r--
bimi.lua	11.5 KB	-rw-r--r--
clickhouse.lua	49.6 KB	-rw-r--r--
clustering.lua	9.16 KB	-rw-r--r--
dcc.lua	3.19 KB	-rw-r--r--
dkim_signing.lua	5.25 KB	-rw-r--r--
dmarc.lua	21.89 KB	-rw-r--r--
dynamic_conf.lua	8.7 KB	-rw-r--r--
elastic.lua	54.28 KB	-rw-r--r--
emails.lua	128 B	-rw-r--r--
external_relay.lua	7.89 KB	-rw-r--r--
external_services.lua	12.3 KB	-rw-r--r--
force_actions.lua	7.67 KB	-rw-r--r--
forged_recipients.lua	5.32 KB	-rw-r--r--
fuzzy_collect.lua	5.46 KB	-rw-r--r--
gpt.lua	11.32 KB	-rw-r--r--
greylist.lua	16.36 KB	-rw-r--r--
hfilter.lua	16.15 KB	-rw-r--r--
history_redis.lua	9.31 KB	-rw-r--r--
http_headers.lua	6.46 KB	-rw-r--r--
ip_score.lua	130 B	-rw-r--r--
known_senders.lua	13.2 KB	-rw-r--r--
maillist.lua	6.55 KB	-rw-r--r--
maps_stats.lua	3.7 KB	-rw-r--r--
metadata_exporter.lua	20.19 KB	-rw-r--r--
metric_exporter.lua	6.4 KB	-rw-r--r--
mid.lua	3.58 KB	-rw-r--r--
milter_headers.lua	21 KB	-rw-r--r--
mime_types.lua	20.09 KB	-rw-r--r--
multimap.lua	40.08 KB	-rw-r--r--
mx_check.lua	11.42 KB	-rw-r--r--
neural.lua	32.82 KB	-rw-r--r--
once_received.lua	4.76 KB	-rw-r--r--
p0f.lua	2.88 KB	-rw-r--r--
phishing.lua	19.33 KB	-rw-r--r--
ratelimit.lua	22.77 KB	-rw-r--r--
rbl.lua	40.09 KB	-rw-r--r--
replies.lua	13.69 KB	-rw-r--r--
reputation.lua	39.69 KB	-rw-r--r--
rspamd_update.lua	4.07 KB	-rw-r--r--
settings.lua	40.17 KB	-rw-r--r--
spamassassin.lua	48.89 KB	-rw-r--r--
spamtrap.lua	5.87 KB	-rw-r--r--
spf.lua	7.49 KB	-rw-r--r--
trie.lua	4.89 KB	-rw-r--r--
url_redirector.lua	13.42 KB	-rw-r--r--
whitelist.lua	11.54 KB	-rw-r--r--

Code Editor : asn.lua

--[[
Copyright (c) 2022, Vsevolod Stakhov <vsevolod@rspamd.com>
Copyright (c) 2016, Andrew Lewis <nerf@judo.za.org>

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 rspamd_logger = require "rspamd_logger"
local rspamd_regexp = require "rspamd_regexp"
local lua_util = require "lua_util"
local N = "asn"

if confighelp then
  return
end

local options = {
  provider_type = 'rspamd',
  provider_info = {
    ip4 = 'asn.rspamd.com',
    ip6 = 'asn6.rspamd.com',
  },
  symbol = 'ASN',
  check_local = false,
}

local rspamd_re = rspamd_regexp.create_cached("[\\|\\s]")

local function asn_check(task)

local function asn_set(asn, ipnet, country)
    local descr_t = {}
    local mempool = task:get_mempool()
    if asn then
      if tonumber(asn) ~= nil then
        mempool:set_variable("asn", asn)
        table.insert(descr_t, "asn:" .. asn)
      else
        rspamd_logger.errx(task, 'malformed ASN "%s" for ip %s', asn, task:get_from_ip())
      end
    end
    if ipnet then
      mempool:set_variable("ipnet", ipnet)
      table.insert(descr_t, "ipnet:" .. ipnet)
    end
    if country then
      mempool:set_variable("country", country)
      table.insert(descr_t, "country:" .. country)
    end
    if options['symbol'] then
      task:insert_result(options['symbol'], 0.0, table.concat(descr_t, ', '))
    end
  end

local asn_check_func = {}
  asn_check_func.rspamd = function(ip)
    local dnsbl = options['provider_info']['ip' .. ip:get_version()]
    local req_name = string.format("%s.%s",
        table.concat(ip:inversed_str_octets(), '.'), dnsbl)
    local function rspamd_dns_cb(_, _, results, dns_err, _, _, serv)
      if dns_err and (dns_err ~= 'requested record is not found' and dns_err ~= 'no records with this name') then
        rspamd_logger.errx(task, 'error querying dns "%s" on %s: %s',
            req_name, serv, dns_err)
        task:insert_result(options['symbol_fail'], 0, string.format('%s:%s', req_name, dns_err))
        return
      end
      if not results or not results[1] then
        rspamd_logger.infox(task, 'no ASN information is available for the IP address "%s" on %s',
            req_name, serv)
        return
      end

lua_util.debugm(N, task, 'got reply from %s when requesting %s: %s',
          serv, req_name, results[1])

local parts = rspamd_re:split(results[1])
      -- "15169 | 8.8.8.0/24 | US | arin |" for 8.8.8.8
      asn_set(parts[1], parts[2], parts[3])
    end

task:get_resolver():resolve_txt({
      task = task,
      name = req_name,
      callback = rspamd_dns_cb
    })
  end

local ip = task:get_from_ip()
  if not (ip and ip:is_valid()) or
      (not options.check_local and ip:is_local()) then
    return
  end

asn_check_func[options['provider_type']](ip)
end

-- Configuration options
local configure_asn_module = function()
  local opts = rspamd_config:get_all_opt('asn')
  if opts then
    for k, v in pairs(opts) do
      options[k] = v
    end
  end

local auth_and_local_conf = lua_util.config_check_local_or_authed(rspamd_config, N,
      false, true)
  options.check_local = auth_and_local_conf[1]
  options.check_authed = auth_and_local_conf[2]

if options['provider_type'] == 'rspamd' then
    if not options['provider_info'] and options['provider_info']['ip4'] and
        options['provider_info']['ip6'] then
      rspamd_logger.errx("Missing required provider_info for rspamd")
      return false
    end
  else
    rspamd_logger.errx("Unknown provider_type: %s", options['provider_type'])
    return false
  end

if options['symbol'] then
    options['symbol_fail'] = options['symbol'] .. '_FAIL'
  else
    options['symbol_fail'] = 'ASN_FAIL'
  end

return true
end

if configure_asn_module() then
  local id = rspamd_config:register_symbol({
    name = 'ASN_CHECK',
    type = 'prefilter',
    callback = asn_check,
    priority = lua_util.symbols_priorities.high,
    flags = 'empty,nostat',
    augmentations = { lua_util.dns_timeout_augmentation(rspamd_config) },
  })
  if options['symbol'] then
    rspamd_config:register_symbol({
      name = options['symbol'],
      parent = id,
      type = 'virtual',
      flags = 'empty,nostat',
      score = 0,
    })
  end
  rspamd_config:register_symbol {
    name = options['symbol_fail'],
    parent = id,
    type = 'virtual',
    flags = 'empty,nostat',
    score = 0,
  }
else
  lua_util.disable_module(N, 'config')
end

${this.title}

Code Editor : asn.lua