mirror of
https://github.com/bjc/prosody.git
synced 2025-04-04 21:57:45 +03:00
140 lines
3.6 KiB
Lua
140 lines
3.6 KiB
Lua
-- Prosody IM
|
|
-- Copyright (C) 2008-2010 Matthew Wild
|
|
-- Copyright (C) 2008-2010 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 pairs = pairs;
|
|
local ipairs = ipairs;
|
|
local t_insert = table.insert;
|
|
local t_concat = table.concat;
|
|
|
|
local default_log = require "util.logger".init("xmlhandlers");
|
|
|
|
local error = error;
|
|
|
|
module "xmlhandlers"
|
|
|
|
local ns_prefixes = {
|
|
["http://www.w3.org/XML/1998/namespace"] = "xml";
|
|
}
|
|
|
|
function init_xmlhandlers(session, stream_callbacks)
|
|
local ns_stack = { "" };
|
|
local curr_tag;
|
|
local chardata = {};
|
|
local xml_handlers = {};
|
|
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 stream_tag = stream_callbacks.stream_tag;
|
|
local stream_default_ns = stream_callbacks.default_ns;
|
|
|
|
local stanza
|
|
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
|
|
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 tagname == stream_tag then
|
|
if cb_streamopened then
|
|
cb_streamopened(session, attr);
|
|
end
|
|
else
|
|
-- Garbage before stream?
|
|
cb_error(session, "no-stream");
|
|
end
|
|
return;
|
|
end
|
|
if curr_ns == "jabber:client" and name ~= "iq" and name ~= "presence" and name ~= "message" then
|
|
cb_error(session, "invalid-top-level-element");
|
|
end
|
|
|
|
stanza = st.stanza(name, attr);
|
|
curr_tag = stanza;
|
|
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
|
|
function xml_handlers:CharacterData(data)
|
|
if stanza then
|
|
t_insert(chardata, data);
|
|
end
|
|
end
|
|
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
|
|
cb_error(session, "parse-error", "unexpected-element-close", name);
|
|
end
|
|
stanza, chardata = nil, {};
|
|
return;
|
|
end
|
|
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
|
|
cb_handlestanza(session, stanza);
|
|
stanza = nil;
|
|
else
|
|
stanza:up();
|
|
end
|
|
end
|
|
return xml_handlers;
|
|
end
|
|
|
|
return init_xmlhandlers;
|