util/xmppstream.lua

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
-- 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.
--

local tblinsert = table.insert

local lex = require "util.xmllex";
local index = lex.index
local tagindex_to_tree  = lex.tagindex_to_tree;
local get_name, get_attr = lex.get_name, lex.get_attr;

local default_log = require "util.logger".init("xmppstream");

local xmlns_streams = "http://etherx.jabber.org/streams";

local function new(session, stream_callbacks)
	local partial
	local openindexes = { }
	local lastindex = 0
	
	return {
		reset = function ( )
			partial = nil
			openindexes = { }
			lastindex = 0
		end ;
		feed = function ( stream , data )
			--print("FFEED", data)
			partial = index ( data , partial )
			local tmp = lastindex
			lastindex = #partial - 1
			for i = tmp + 1 , #partial - 1 do
				local v = partial [ i ]
				
				if v.finish then
					if v.type == "open" or v.type == "selfclosing" then
						tblinsert ( openindexes , i )
						if #openindexes == 1 then
							local vstr = tostring(v);
							local name = get_name ( vstr )
							if name == "stream:stream" then
								local attr = get_attr ( vstr );
								stream_callbacks.streamopened ( session , attr )
							end
						end
					end
					if v.type == "close" or v.type == "selfclosing" then
						if #openindexes == 2 then -- If closing level 2...
							local attr = get_attr ( tostring(partial[openindexes[1]]) )
							local stanza = tagindex_to_tree ( partial , openindexes [ #openindexes ] , i , {
								attr = attr,
							} )
							if stanza[1].attr.xmlns == attr.xmlns then
								stanza[1].attr.xmlns = false;
							end
							stream_callbacks.handlestanza ( session , stanza[1] )
						elseif #openindexes == 1 then
							local prefix, name = get_name ( tostring(v) )
							if name == "stream" then
								stream_callbacks.streamclosed ( session )
							end
						end
						openindexes [ #openindexes ] = nil
					end
				end
			end
			return true
		end ;
		set_session = function ( newsession ) session = newsession end
	};
end

return {
	new = new ;
}