Diff

mod_firewall/conditions.lib.lua @ 2386:00eed68f63bf

mod_firewall: INSPECT: support for literal substring search and expressions
author Matthew Wild <mwild1@gmail.com>
date Sat, 19 Nov 2016 15:52:32 +0000
parent 2363:12b78170b76c
child 2403:f96bdfd81eba
line wrap: on
line diff
--- a/mod_firewall/conditions.lib.lua	Sat Nov 19 15:47:41 2016 +0000
+++ b/mod_firewall/conditions.lib.lua	Sat Nov 19 15:52:32 2016 +0000
@@ -111,14 +111,23 @@
 
 function condition_handlers.INSPECT(path)
 	if path:find("=") then
-		local query, is_pattern_match, value = path:match("(.-)(~?)=(.*)");
+		local query, match_type, value = path:match("(.-)([~/$]*)=(.*)");
 		if not(query:match("#$") or query:match("@[^/]+")) then
 			error("Stanza path does not return a string (append # for text content or @name for value of named attribute)", 0);
 		end
-		if is_pattern_match ~= "" then
-			return ("(stanza:find(%q) or ''):match(%q)"):format(path:match("(.-)~=(.*)"));
+		local quoted_value = ("%q"):format(value);
+		if match_type:find("$", 1, true) then
+			match_type = match_type:gsub("%$", "");
+			quoted_value = meta(quoted_value);
+		end
+		if match_type == "~" then -- Lua pattern match
+			return ("(stanza:find(%q) or ''):match(%s)"):format(query, quoted_value);
+		elseif match_type == "/" then -- find literal substring
+			return ("(stanza:find(%q) or ''):find(%s, 1, true)"):format(query, quoted_value);
+		elseif match_type == "" then -- exact match
+			return ("stanza:find(%q) == %s"):format(query, quoted_value);
 		else
-			return ("stanza:find(%q) == %q"):format(path:match("(.-)=(.*)"));
+			error("Unrecognised comparison '"..match_type.."='", 0);
 		end
 	end
 	return ("stanza:find(%q)"):format(path);