# HG changeset patch # User Kim Alvefur # Date 1627582308 -7200 # Node ID c0fc4ca7404627bd8500b94e37f6648ca813218d # Parent 27699cc148df4d5ef6b25e79d320b5c5de1fb301 mod_limits: Factor out function for disabling limits allowing use from shell Also enables reuse for s2s, which we will add next. diff -r 27699cc148df -r c0fc4ca74046 plugins/mod_limits.lua --- a/plugins/mod_limits.lua Thu Jul 29 20:04:57 2021 +0200 +++ b/plugins/mod_limits.lua Thu Jul 29 20:11:48 2021 +0200 @@ -111,23 +111,27 @@ filters.remove_filter_hook(filter_hook); end +function unlimited(session) + local session_type = session.type:match("^[^_]+"); + if session.conn and session.conn.setlimit then + session.conn:setlimit(0); + -- Currently no burst support + else + local filter_set = type_filters[session_type]; + filters.remove_filter(session, "bytes/in", filter_set.bytes_in); + session.throttle = nil; + end +end + function module.add_host(module) local unlimited_jids = module:get_option_inherited_set("unlimited_jids", {}); if not unlimited_jids:empty() then module:hook("authentication-success", function (event) local session = event.session; - local session_type = session.type:match("^[^_]+"); local jid = session.username .. "@" .. session.host; if unlimited_jids:contains(jid) then - if session.conn and session.conn.setlimit then - session.conn:setlimit(0); - -- Currently no burst support - else - local filter_set = type_filters[session_type]; - filters.remove_filter(session, "bytes/in", filter_set.bytes_in); - session.throttle = nil; - end + unlimited(session); end end); end