diff mod_report_tracker/mod_report_tracker.lua @ 6110:9db1529c06c2

Merge upstream
author tmolitor <thilo@eightysoft.de>
date Sun, 05 Jan 2025 17:50:02 +0100
parents 765e0235c202
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_report_tracker/mod_report_tracker.lua	Sun Jan 05 17:50:02 2025 +0100
@@ -0,0 +1,93 @@
+local um = require "core.usermanager";
+local cache = require "util.cache";
+local jid = require "util.jid";
+local trusted_reporters = module:get_option_inherited_set("trusted_reporters", {});
+
+local reports_received = module:open_store("reports_received");
+
+local xmlns_reporting = "urn:xmpp:reporting:1";
+
+local reported_users = cache.new(256);
+
+local function is_trusted_reporter(reporter_jid)
+	return trusted_reporters:contains(reporter_jid);
+end
+
+function handle_report(event)
+	local stanza = event.stanza;
+	local report = stanza:get_child("report", xmlns_reporting);
+	if not report then
+		return;
+	end
+	local reported_jid = report:get_child_text("jid", "urn:xmpp:jid:0")
+		or stanza:find("{urn:xmpp:forward:0}forwarded/{jabber:client}message@from");
+	if not reported_jid then
+		module:log("debug", "Discarding report with no JID");
+		return;
+	elseif jid.host(reported_jid) ~= module.host then
+		module:log("debug", "Discarding report about non-local user");
+		return;
+	end
+
+	local reporter_jid = stanza.attr.from;
+	if jid.node(reporter_jid) then
+		module:log("debug", "Discarding report from non-server JID");
+		return;
+	end
+
+	local reported_user = jid.node(reported_jid);
+	if not um.user_exists(reported_user, module.host) then
+		module:log("debug", "Discarding report about non-existent user");
+		return;
+	end
+
+	if is_trusted_reporter(reporter_jid) then
+		local current_reports = reports_received:get(reported_user, reporter_jid);
+
+		if not current_reports then
+			current_reports = {
+				first = os.time();
+				last = os.time();
+				count = 1;
+			};
+		else
+			current_reports.last = os.time();
+			current_reports.count = current_reports.count + 1;
+		end
+
+		reports_received:set(reported_user, reporter_jid, current_reports);
+		reported_users:set(reported_user, true);
+
+		module:log("info", "Received abuse report about <%s> from <%s>", reported_jid, reporter_jid);
+
+		module:fire_event(module.name.."/account-reported", {
+			report_from = reporter_jid;
+			reported_user = reported_user;
+			report = report;
+		});
+	else
+		module:log("warn", "Discarding abuse report about <%s> from untrusted source <%s>", reported_jid, reporter_jid);
+	end
+
+	-- Message was handled
+	return true;
+end
+
+module:hook("message/host", handle_report);
+
+module:add_item("account-trait", {
+	name = "reported-by-trusted-server";
+	prob_bad_true = 0.80;
+	prob_bad_false = 0.50;
+});
+
+module:hook("get-account-traits", function (event)
+	local username = event.username;
+	local reported = reported_users:get(username);
+	if reported == nil then
+		-- Check storage, update cache
+		reported = not not reports_received:get(username);
+		reported_users:set(username, reported);
+	end
+	event.traits["reported-by-trusted-server"] = reported;
+end);