util.xml: Deduplicate handlers for restricted XML 0.11 0.11.13
authorKim Alvefur <zash@zash.se>
Thu, 20 Jan 2022 10:51:46 +0100
branch0.11
changeset 12206 ebeb4d959fb3
parent 12205 e5e0ab93d7f4
child 12207 320de3e4b579
child 12214 458c5f8d5d3e
util.xml: Deduplicate handlers for restricted XML Makes the code more like util.xmppstream, allowing easier comparisons if we ever need to apply fixes in the future.
util/xml.lua
--- a/util/xml.lua	Thu Jan 20 09:57:20 2022 +0100
+++ b/util/xml.lua	Thu Jan 20 10:51:46 2022 +0100
@@ -66,23 +66,16 @@
 			stanza:up();
 		end
 		-- SECURITY: These two handlers, especially the Doctype one, are required to prevent exploits such as Billion Laughs.
-		function handler:StartDoctypeDecl()
-			if not self.stop or not self:stop() then
+		local function restricted_handler(parser)
+			if not parser.stop or not parser:stop() then
 				error("Failed to abort parsing");
 			end
 		end
-		function handler:ProcessingInstruction()
-			if not self.stop or not self:stop() then
-				error("Failed to abort parsing");
-			end
-		end
+		handler.StartDoctypeDecl = restricted_handler;
+		handler.ProcessingInstruction = restricted_handler;
 		if not options or not options.allow_comments then
 			-- NOTE: comments are generally harmless and can be useful when parsing configuration files or other data, even user-provided data
-			function handler:Comment()
-				if not self.stop or not self:stop() then
-					error("Failed to abort parsing");
-				end
-			end
+			handler.Comment = restricted_handler;
 		end
 		local parser = lxp.new(handler, ns_separator);
 		local ok, err, line, col = parser:parse(xml);