Mercurial > prosody-modules

comparison mod_bidi/mod_bidi.lua @ 927:a9dfa7232d88

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge
author Matthew Wild <mwild1@gmail.com>
date Tue, 12 Mar 2013 12:10:25 +0000
parents 4cd018025a26
children 4e235e565693
comparison
equal deleted inserted replaced
926:f88381a39c56 927:a9dfa7232d88
1 -- Bidirectional Server-to-Server Connections
2 -- http://xmpp.org/extensions/xep-0288.html
3 -- Copyright (C) 2013 Kim Alvefur
4 --
5 -- This file is MIT/X11 licensed.
6 --
7 local s2smanager = require"core.s2smanager";
8 local add_filter = require "util.filters".add_filter;
9 local st = require "util.stanza";
10 local jid_split = require"util.jid".prepped_split;
11
12 local xmlns_bidi_feature = "urn:xmpp:features:bidi"
13 local xmlns_bidi = "urn:xmpp:bidi";
14 local noop = function () end
15 local core_process_stanza = prosody.core_process_stanza or core_process_stanza;
16 local traceback = debug.traceback;
17
18 local function handleerr(err) log("error", "Traceback[s2s]: %s: %s", tostring(err), traceback()); end
19 local function handlestanza(session, stanza)
20 if stanza.attr.xmlns == "jabber:client" then --COMPAT: Prosody pre-0.6.2 may send jabber:client
21 stanza.attr.xmlns = nil;
22 end
23 -- stanza = session.filter("stanzas/in", stanza);
24 if stanza then
25 return xpcall(function () return core_process_stanza(session, stanza) end, handleerr);
26 end
27 end
28
29 local function new_bidi(origin)
30 local bidi_session, remote_host;
31 origin.log("debug", "Creating bidirectional session wrapper");
32 if origin.direction == "incoming" then -- then we create an "outgoing" bidirectional session
33 local conflicting_session = hosts[origin.to_host].s2sout[origin.from_host]
34 if conflicting_session then
35 conflicting_session.log("info", "We already have an outgoing connection to %s, closing it...", origin.from_host);
36 conflicting_session:close{ condition = "conflict", text = "Replaced by bidirectional stream" }
37 s2smanager.destroy_session(conflicting_session);
38 end
39 remote_host = origin.from_host;
40 bidi_session = s2smanager.new_outgoing(origin.to_host, origin.from_host)
41 else -- outgoing -- then we create an "incoming" bidirectional session
42 remote_host = origin.to_host;
43 bidi_session = s2smanager.new_incoming(origin.conn)
44 bidi_session.to_host = origin.from_host;
45 bidi_session.from_host = origin.to_host;
46 add_filter(origin, "stanzas/in", function(stanza)
47 if stanza.attr.xmlns ~= nil then return stanza end
48 local _, host = jid_split(stanza.attr.from);
49 if host ~= remote_host then return stanza end
50 handlestanza(bidi_session, stanza);
51 end, 1);
52 end
53 origin.bidi_session = bidi_session;
54 bidi_session.sends2s = origin.sends2s;
55 bidi_session.bounce_sendq = noop;
56 bidi_session.notopen = nil;
57 bidi_session.is_bidi = true;
58 bidi_session.bidi_session = false;
59 bidi_session.orig_session = origin;
60 bidi_session.secure = origin.secure;
61 bidi_session.cert_identity_status = origin.cert_identity_status;
62 bidi_session.cert_chain_status = origin.cert_chain_status;
63 bidi_session.close = function(...)
64 return origin.close(...);
65 end
66
67 bidi_session.log("info", "Bidirectional session established");
68 s2smanager.make_authenticated(bidi_session, remote_host);
69 return bidi_session;
70 end
71
72 -- Incoming s2s
73 module:hook("s2s-stream-features", function(event)
74 local origin, features = event.origin, event.features;
75 if not origin.is_bidi and not hosts[module.host].s2sout[origin.from_host] then
76 module:log("debug", "Announcing support for bidirectional streams");
77 features:tag("bidi", { xmlns = xmlns_bidi_feature }):up();
78 end
79 end);
80
81 module:hook("stanza/urn:xmpp:bidi:bidi", function(event)
82 local origin = event.session or event.origin;
83 if not origin.is_bidi and not origin.bidi_session then
84 module:log("debug", "%s requested bidirectional stream", origin.from_host);
85 origin.do_bidi = true;
86 return true;
87 end
88 end);
89
90 -- Outgoing s2s
91 module:hook("stanza/http://etherx.jabber.org/streams:features", function(event)
92 local origin = event.session or event.origin;
93 if not ( origin.bidi_session or origin.is_bidi or origin.do_bidi)
94 and event.stanza:get_child("bidi", xmlns_bidi_feature) then
95 module:log("debug", "%s supports bidirectional streams", origin.to_host);
96 origin.sends2s(st.stanza("bidi", { xmlns = xmlns_bidi }));
97 origin.do_bidi = true;
98 end
99 end, 160);
100
101 function enable_bidi(event)
102 local session = event.session;
103 if session.do_bidi and not ( session.is_bidi or session.bidi_session ) then
104 session.do_bidi = nil;
105 new_bidi(session);
106 end
107 end
108
109 module:hook("s2sin-established", enable_bidi);
110 module:hook("s2sout-established", enable_bidi);
111
112 function disable_bidi(event)
113 local session = event.session;
114 if session.bidi_session then
115 local bidi_session = session.bidi_session;
116 session.bidi_session, bidi_session.orig_session = nil, nil;
117 session.log("debug", "Tearing down bidirectional stream");
118 s2smanager.destroy_session(bidi_session, event.reason);
119 elseif session.orig_session then
120 local orig_session = session.orig_session;
121 orig_session.bidi_session, session.orig_session = nil, nil;
122 orig_session.log("debug", "Tearing down bidirectional stream");
123 s2smanager.destroy_session(orig_session, event.reason);
124 end
125 end
126
127 module:hook("s2sin-destroyed", disable_bidi);
128 module:hook("s2sout-destroyed", disable_bidi);
129