view mod_register_dnsbl_firewall_mark/mod_register_dnsbl_firewall_mark.lua @ 4980:da151f9af861

replaced 'session' with 'origin' in push_disable session is not defined in this function, trying to access it leads to an error. The correct reference seems to be 'origin'. (This may have come about by copying from the similar code in process_stanza_queue.)
author arcseconds
date Sat, 30 Jul 2022 21:07:47 +1200
parents de40686ae9c8
children
line wrap: on
line source

local adns = require "net.adns";
local rbl = module:get_option_string("registration_rbl");
local rbl_message = module:get_option_string("registration_rbl_message");
local st = require "util.stanza";


local function cleanup_ip(ip)
	if ip:sub(1,7):lower() == "::ffff:" then
		return ip:sub(8);
	end
	return ip;
end

local function reverse(ip, suffix)
	local a,b,c,d = ip:match("^(%d+).(%d+).(%d+).(%d+)$");
	if not a then return end
	return ("%d.%d.%d.%d.%s"):format(d,c,b,a, suffix);
end

module:hook("user-registered", function (event)
	local session = event.session;
	local ip = session and session.ip and cleanup_ip(session.ip);
	local rbl_ip = ip and reverse(ip, rbl);
	if rbl_ip then
		local registration_time = os.time();
		local log = session.log;
		adns.lookup(function (reply)
			if reply and reply[1] then
				log("warn", "Account %s@%s registered from IP %s found in RBL (%s)", event.username, event.host or module.host, ip, reply[1].a);
				local user = prosody.bare_sessions[event.username .. "@" .. module.host];
				if user and user.firewall_marks then
					user.firewall_marks.dnsbl_hit = registration_time;
				else
					module:open_store("firewall_marks", "map"):set(event.username, "dnsbl_hit", registration_time);
				end
				if rbl_message then
					module:log("debug", "Warning RBL registered user %s@%s", event.username, event.host);
					event.ip = ip;
					local rbl_stanza =
						st.message({ to = event.username.."@"..event.host, from = event.host },
							rbl_message:gsub("$(%w+)", event));
					module:send(rbl_stanza);
				end
			end
		end, rbl_ip);
	end
end);