Mercurial > prosody-modules

diff mod_firewall/definitions.lib.lua @ 999:197af8440ffb

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
mod_firewall: Make defining objects generic (currently zones and rate limits), so more can easily be added. Also a syntax change... definition lines must begin with %
author Matthew Wild <mwild1@gmail.com>
date Tue, 07 May 2013 10:33:49 +0100
parents
children 92602cfac751
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_firewall/definitions.lib.lua	Tue May 07 10:33:49 2013 +0100
@@ -0,0 +1,20 @@
+
+local definition_handlers = {};
+
+local new_throttle = require "util.throttle".create;
+
+function definition_handlers.ZONE(zone_name, zone_members)
+			local zone_member_list = {};
+			for member in zone_members:gmatch("[^, ]+") do
+				zone_member_list[#zone_member_list+1] = member;
+			end
+			return set.new(zone_member_list)._items;
+end
+
+function definition_handlers.RATE(name, line)
+			local rate = assert(tonumber(line:match("([%d.]+)")), "Unable to parse rate");
+			local burst = tonumber(line:match("%(%s*burst%s+([%d.]+)%s*%)")) or 1;
+			return new_throttle(rate*burst, burst);
+end
+
+return definition_handlers;