Mercurial > prosody-modules
view mod_firewall/test.lib.lua @ 5536:96dec7681af8
mod_firewall: Update user marks to store instantly via map store
The original approach was to keep marks in memory only, and persist them at
shutdown. That saves I/O, at the cost of potentially losing marks on an
unclean shutdown.
This change persists marks instantly, which may have some performance overhead
but should be more "correct".
It also splits the marking/unmarking into an event which may be watched or
even fired by other modules.
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Thu, 08 Jun 2023 16:20:42 +0100 |
parents | 001c756ead7d |
children |
line wrap: on
line source
-- luacheck: globals load_unload_scripts local set = require "util.set"; local ltn12 = require "ltn12"; local xmppstream = require "util.xmppstream"; local function stderr(...) io.stderr:write("** ", table.concat({...}, "\t", 1, select("#", ...)), "\n"); end return function (arg) require "net.http".request = function (url, ex, cb) stderr("Making HTTP request to "..url); local body_table = {}; local ok, response_status, response_headers = require "ssl.https".request({ url = url; headers = ex.headers; method = ex.body and "POST" or "GET"; sink = ltn12.sink.table(body_table); source = ex.body and ltn12.source.string(ex.body) or nil; }); stderr("HTTP response "..response_status); cb(table.concat(body_table), response_status, { headers = response_headers }); return true; end; local stats_dropped, stats_passed = 0, 0; load_unload_scripts(set.new(arg)); local stream_callbacks = { default_ns = "jabber:client" }; function stream_callbacks.streamopened(session) session.notopen = nil; end function stream_callbacks.streamclosed() end function stream_callbacks.error(session, error_name, error_message) -- luacheck: ignore 212/session stderr("Fatal error parsing XML stream: "..error_name..": "..tostring(error_message)) assert(false); end function stream_callbacks.handlestanza(session, stanza) if not module:fire_event("firewall/chains/deliver", { origin = session, stanza = stanza }) then stats_passed = stats_passed + 1; print(stanza); print(""); else stats_dropped = stats_dropped + 1; end end local session = { notopen = true }; function session.send(stanza) stderr("Reply:", "\n"..tostring(stanza).."\n"); end local stream = xmppstream.new(session, stream_callbacks); stream:feed("<stream:stream xmlns:stream='http://etherx.jabber.org/streams' xmlns='jabber:client'>"); local line_count = 0; for line in io.lines() do line_count = line_count + 1; local ok, err = stream:feed(line.."\n"); if not ok then stderr("Fatal XML parse error on line "..line_count..": "..err); return 1; end end stderr("Summary"); stderr("-------"); stderr(""); stderr(stats_dropped + stats_passed, "processed"); stderr(stats_passed, "passed"); stderr(stats_dropped, "dropped"); stderr(line_count, "input lines"); stderr(""); end