comparison: component.lua
component.lua
- changeset 84
- d85d2443478e
- child 150
- 728cc7f2f0c2
equal
deleted
inserted
replaced
| 83:8221f3c25fd4 | 84:d85d2443478e |
|---|---|
| 1 local verse = require "verse"; | |
| 2 local stream = verse.stream_mt; | |
| 3 | |
| 4 local jid_split = require "util.jid".split; | |
| 5 local lxp = require "lxp"; | |
| 6 local st = require "util.stanza"; | |
| 7 local sha1 = require "util.sha1".sha1; | |
| 8 | |
| 9 -- Shortcuts to save having to load util.stanza | |
| 10 verse.message, verse.presence, verse.iq, verse.stanza, verse.reply, verse.error_reply = | |
| 11 st.message, st.presence, st.iq, st.stanza, st.reply, st.error_reply; | |
| 12 | |
| 13 local init_xmlhandlers = require "core.xmlhandlers"; | |
| 14 | |
| 15 local xmlns_stream = "http://etherx.jabber.org/streams"; | |
| 16 local xmlns_component = "jabber:component:accept"; | |
| 17 | |
| 18 local stream_callbacks = { | |
| 19 stream_ns = xmlns_stream, | |
| 20 stream_tag = "stream", | |
| 21 default_ns = "jabber:client" }; | |
| 22 | |
| 23 function stream_callbacks.streamopened(stream, attr) | |
| 24 stream.stream_id = attr.id; | |
| 25 if not stream:event("opened", attr) then | |
| 26 stream.notopen = nil; | |
| 27 end | |
| 28 return true; | |
| 29 end | |
| 30 | |
| 31 function stream_callbacks.streamclosed(stream) | |
| 32 return stream:event("closed"); | |
| 33 end | |
| 34 | |
| 35 function stream_callbacks.handlestanza(stream, stanza) | |
| 36 if stanza.attr.xmlns == xmlns_stream then | |
| 37 return stream:event("stream-"..stanza.name, stanza); | |
| 38 elseif stanza.attr.xmlns then | |
| 39 return stream:event("stream/"..stanza.attr.xmlns, stanza); | |
| 40 end | |
| 41 | |
| 42 return stream:event("stanza", stanza); | |
| 43 end | |
| 44 | |
| 45 function stream:reset() | |
| 46 -- Reset stream | |
| 47 local parser = lxp.new(init_xmlhandlers(self, stream_callbacks), "\1"); | |
| 48 self.parser = parser; | |
| 49 | |
| 50 self.notopen = true; | |
| 51 | |
| 52 return true; | |
| 53 end | |
| 54 | |
| 55 function stream:connect_component(jid, pass) | |
| 56 self.jid, self.password = jid, pass; | |
| 57 self.username, self.host, self.resource = jid_split(jid); | |
| 58 | |
| 59 function self.data(conn, data) | |
| 60 local ok, err = self.parser:parse(data); | |
| 61 if ok then return; end | |
| 62 stream:debug("debug", "Received invalid XML (%s) %d bytes: %s", tostring(err), #data, data:sub(1, 300):gsub("[\r\n]+", " ")); | |
| 63 stream:close("xml-not-well-formed"); | |
| 64 end | |
| 65 | |
| 66 self:hook("incoming-raw", function (data) return self.data(self.conn, data); end); | |
| 67 | |
| 68 self.curr_id = 0; | |
| 69 | |
| 70 self.tracked_iqs = {}; | |
| 71 self:hook("stanza", function (stanza) | |
| 72 local id, type = stanza.attr.id, stanza.attr.type; | |
| 73 if id and stanza.name == "iq" and (type == "result" or type == "error") and self.tracked_iqs[id] then | |
| 74 self.tracked_iqs[id](stanza); | |
| 75 self.tracked_iqs[id] = nil; | |
| 76 return true; | |
| 77 end | |
| 78 end); | |
| 79 | |
| 80 self:hook("stanza", function (stanza) | |
| 81 if stanza.attr.xmlns == nil or stanza.attr.xmlns == "jabber:client" then | |
| 82 if stanza.name == "iq" and (stanza.attr.type == "get" or stanza.attr.type == "set") then | |
| 83 local xmlns = stanza.tags[1] and stanza.tags[1].attr.xmlns; | |
| 84 if xmlns then | |
| 85 ret = self:event("iq/"..xmlns, stanza); | |
| 86 if not ret then | |
| 87 ret = self:event("iq", stanza); | |
| 88 end | |
| 89 end | |
| 90 if ret == nil then | |
| 91 self:send(verse.error_reply(stanza, "cancel", "service-unavailable")); | |
| 92 return true; | |
| 93 end | |
| 94 else | |
| 95 ret = self:event(stanza.name, stanza); | |
| 96 end | |
| 97 end | |
| 98 return ret; | |
| 99 end, -1); | |
| 100 | |
| 101 self:hook("opened", function (attr) | |
| 102 print(self.jid, self.stream_id, attr.id); | |
| 103 local token = sha1(self.stream_id..pass, true); | |
| 104 | |
| 105 self:send(st.stanza("handshake", { xmlns = xmlns_component }):text(token)); | |
| 106 self:hook("stream/"..xmlns_component, function (stanza) | |
| 107 if stanza.name == "handshake" then | |
| 108 self:event("authentication-success"); | |
| 109 end | |
| 110 end); | |
| 111 end); | |
| 112 | |
| 113 local function stream_ready() | |
| 114 self:event("ready"); | |
| 115 end | |
| 116 self:hook("authentication-success", stream_ready, -1); | |
| 117 | |
| 118 -- Initialise connection | |
| 119 self:connect(self.connect_host or self.host, self.connect_port or 5347); | |
| 120 self:reopen(); | |
| 121 end | |
| 122 | |
| 123 function stream:reopen() | |
| 124 self:reset(); | |
| 125 self:send(st.stanza("stream:stream", { to = self.host, ["xmlns:stream"]='http://etherx.jabber.org/streams', | |
| 126 xmlns = xmlns_component, version = "1.0" }):top_tag()); | |
| 127 end | |
| 128 | |
| 129 function stream:close(reason) | |
| 130 if not self.notopen then | |
| 131 self:send("</stream:stream>"); | |
| 132 end | |
| 133 local on_disconnect = self.conn.disconnect(); | |
| 134 self.conn:close(); | |
| 135 on_disconnect(conn, reason); | |
| 136 end | |
| 137 | |
| 138 function stream:send_iq(iq, callback) | |
| 139 local id = self:new_id(); | |
| 140 self.tracked_iqs[id] = callback; | |
| 141 iq.attr.id = id; | |
| 142 self:send(iq); | |
| 143 end | |
| 144 | |
| 145 function stream:new_id() | |
| 146 self.curr_id = self.curr_id + 1; | |
| 147 return tostring(self.curr_id); | |
| 148 end |
