X-Git-Url: https://git.enpas.org/?a=blobdiff_plain;f=core%2Fxmlhandlers.lua;h=82c2d0b8d02361e95350d1f7c662225baf9d9149;hb=4341e61cc1ca8384a0ca18bad7f8f8f3f1c894b0;hp=b1af299fef6e9ea01f64c792050c7be4382ccbdc;hpb=dda60a00dfba3152f2cd7241c14911f58c598ac4;p=prosody.git

diff --git a/core/xmlhandlers.lua b/core/xmlhandlers.lua
index b1af299f..82c2d0b8 100644
--- a/core/xmlhandlers.lua
+++ b/core/xmlhandlers.lua
@@ -1,55 +1,100 @@
+-- Prosody IM
+-- Copyright (C) 2008-2009 Matthew Wild
+-- Copyright (C) 2008-2009 Waqas Hussain
+-- 
+-- This project is MIT/X11 licensed. Please see the
+-- COPYING file in the source package for more information.
+--
+
+
 
 require "util.stanza"
 
 local st = stanza;
 local tostring = tostring;
-local format = string.format;
-local m_random = math.random;
+local pairs = pairs;
+local ipairs = ipairs;
 local t_insert = table.insert;
-local t_remove = table.remove;
 local t_concat = table.concat;
-local t_concatall = function (t, sep) local tt = {}; for _, s in ipairs(t) do t_insert(tt, tostring(s)); end return t_concat(tt, sep); end
-local sm_destroy_session = import("core.sessionmanager", "destroy_session");
+
+local default_log = require "util.logger".init("xmlhandlers");
 
 local error = error;
 
 module "xmlhandlers"
 
-function init_xmlhandlers(session, streamopened)
+local ns_prefixes = {
+						["http://www.w3.org/XML/1998/namespace"] = "xml";
+				}
+
+function init_xmlhandlers(session, stream_callbacks)
 		local ns_stack = { "" };
-		local curr_ns = "";
 		local curr_tag;
 		local chardata = {};
 		local xml_handlers = {};
-		local log = session.log;
-		local print = function (...) log("info", "xmlhandlers", t_concatall({...}, "\t")); end
+		local log = session.log or default_log;
+		
+		local cb_streamopened = stream_callbacks.streamopened;
+		local cb_streamclosed = stream_callbacks.streamclosed;
+		local cb_error = stream_callbacks.error or function (session, e) error("XML stream error: "..tostring(e)); end;
+		local cb_handlestanza = stream_callbacks.handlestanza;
 		
-		local send = session.send;
+		local stream_tag = stream_callbacks.stream_tag;
+		local stream_default_ns = stream_callbacks.default_ns;
 		
 		local stanza
-		function xml_handlers:StartElement(name, attr)
+		function xml_handlers:StartElement(tagname, attr)
 			if stanza and #chardata > 0 then
 				-- We have some character data in the buffer
 				stanza:text(t_concat(chardata));
 				chardata = {};
 			end
-			curr_ns,name = name:match("^(.+):(%w+)$");
-			if not stanza then
+			local curr_ns,name = tagname:match("^([^\1]*)\1?(.*)$");
+			if name == "" then
+				curr_ns, name = "", curr_ns;
+			end
+
+			if curr_ns ~= stream_default_ns then
+				attr.xmlns = curr_ns;
+			end
+			
+			-- FIXME !!!!!
+			for i=1,#attr do
+				local k = attr[i];
+				attr[i] = nil;
+				local ns, nm = k:match("^([^\1]*)\1?(.*)$");
+				if nm ~= "" then
+					ns = ns_prefixes[ns]; 
+					if ns then 
+						attr[ns..":"..nm] = attr[k];
+						attr[k] = nil;
+					end
+				end
+			end
+			
+			if not stanza then --if we are not currently inside a stanza
 				if session.notopen then
-					if name == "stream" then
-						streamopened(session, attr);
-						return;
+					if tagname == stream_tag then
+						if cb_streamopened then
+							cb_streamopened(session, attr);
+						end
+					else
+						-- Garbage before stream?
+						cb_error(session, "no-stream");
 					end
-					error("Client failed to open stream successfully");
+					return;
 				end
 				if curr_ns == "jabber:client" and name ~= "iq" and name ~= "presence" and name ~= "message" then
-					error("Client sent invalid top-level stanza");
+					cb_error(session, "invalid-top-level-element");
 				end
-				attr.xmlns = curr_ns;
-				stanza = st.stanza(name, attr); --{ to = attr.to, type = attr.type, id = attr.id, xmlns = curr_ns });
+				
+				stanza = st.stanza(name, attr);
 				curr_tag = stanza;
-			else
-				attr.xmlns = curr_ns;
+			else -- we are inside a stanza, so add a tag
+				attr.xmlns = nil;
+				if curr_ns ~= stream_default_ns then
+					attr.xmlns = curr_ns;
+				end
 				stanza:tag(name, attr);
 			end
 		end
@@ -58,25 +103,32 @@ function init_xmlhandlers(session, streamopened)
 				t_insert(chardata, data);
 			end
 		end
-		function xml_handlers:EndElement(name)
-			curr_ns,name = name:match("^(.+):(%w+)$");
-			if (not stanza) or #stanza.last_add < 0 or (#stanza.last_add > 0 and name ~= stanza.last_add[#stanza.last_add].name) then 
-				if name == "stream" then
-					log("debug", "Stream closed");
-					sm_destroy_session(session);
-					return;
+		function xml_handlers:EndElement(tagname)
+			local curr_ns,name = tagname:match("^([^\1]*)\1?(.*)$");
+			if name == "" then
+				curr_ns, name = "", curr_ns;
+			end
+			if (not stanza) or (#stanza.last_add > 0 and name ~= stanza.last_add[#stanza.last_add].name) then 
+				if tagname == stream_tag then
+					if cb_streamclosed then
+						cb_streamclosed(session);
+					end
+				elseif name == "error" then
+					cb_error(session, "stream-error", stanza);
 				else
-					error("XML parse error in client stream");
+					cb_error(session, "parse-error", "unexpected-element-close", name);
 				end
+				stanza, chardata = nil, {};
+				return;
 			end
-			if stanza and #chardata > 0 then
+			if #chardata > 0 then
 				-- We have some character data in the buffer
 				stanza:text(t_concat(chardata));
 				chardata = {};
 			end
 			-- Complete stanza
 			if #stanza.last_add == 0 then
-				session.stanza_dispatch(stanza);
+				cb_handlestanza(session, stanza);
 				stanza = nil;
 			else
 				stanza:up();