prosody
09c3845ed442
core/stanza_dispatch.lua

Find changesets by keywords by author, files, the commit message, revision number or hash, or revset expression.

Software / code / prosody

Annotate

core/stanza_dispatch.lua @ 4:09c3845ed442

Presence unavailable on disconnect
author matthew
date Sun, 24 Aug 2008 14:52:02 +0000
parent 2:9bb397205f26
child 6:7ad47ce20394
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
1
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
2 require "util.stanza"
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
3
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
4 local st = stanza;
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
5
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
6 local t_concat = table.concat;
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
7 local format = string.format;
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
8
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
9 function init_stanza_dispatcher(session)
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
10 local iq_handlers = {};
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
11
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
12 local session_log = session.log;
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
13 local log = function (type, msg) session_log(type, "stanza_dispatcher", msg); end
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
14 local send = session.send;
1
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
15 local send_to;
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
16 do
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
17 local _send_to = session.send_to;
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
18 send_to = function (...) _send_to(session, ...); end
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
19 end
0
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
20
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
21 iq_handlers["jabber:iq:auth"] =
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
22 function (stanza)
1
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
23 local username = stanza.tags[1]:child_with_name("username");
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
24 local password = stanza.tags[1]:child_with_name("password");
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
25 local resource = stanza.tags[1]:child_with_name("resource");
0
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
26 if not (username and password and resource) then
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
27 local reply = st.reply(stanza);
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
28 send(reply:query("jabber:iq:auth")
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
29 :tag("username"):up()
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
30 :tag("password"):up()
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
31 :tag("resource"):up());
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
32 return true;
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
33 else
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
34 username, password, resource = t_concat(username), t_concat(password), t_concat(resource);
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
35 print(username, password, resource)
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
36 local reply = st.reply(stanza);
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
37 require "core.usermanager"
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
38 if usermanager.validate_credentials(session.host, username, password) then
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
39 -- Authentication successful!
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
40 session.username = username;
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
41 session.resource = resource;
2
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
42 session.full_jid = username.."@"..session.host.."/"..session.resource;
0
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
43 if not hosts[session.host].sessions[username] then
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
44 hosts[session.host].sessions[username] = { sessions = {} };
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
45 end
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
46 hosts[session.host].sessions[username].sessions[resource] = session;
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
47 send(st.reply(stanza));
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
48 return true;
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
49 else
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
50 local reply = st.reply(stanza);
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
51 reply.attr.type = "error";
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
52 reply:tag("error", { code = "401", type = "auth" })
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
53 :tag("not-authorized", { xmlns = "urn:ietf:params:xml:ns:xmpp-stanzas" });
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
54 send(reply);
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
55 return true;
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
56 end
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
57 end
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
58
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
59 end
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
60
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
61 iq_handlers["jabber:iq:roster"] =
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
62 function (stanza)
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
63 if stanza.attr.type == "get" then
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
64 session.roster = session.roster or rostermanager.getroster(session.username, session.host);
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
65 if session.roster == false then
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
66 send(st.reply(stanza)
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
67 :tag("error", { type = "wait" })
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
68 :tag("internal-server-error", { xmlns = "urn:ietf:params:xml:ns:xmpp-stanzas"}));
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
69 return true;
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
70 else session.roster = session.roster or {};
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
71 end
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
72 local roster = st.reply(stanza)
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
73 :query("jabber:iq:roster");
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
74 for jid in pairs(session.roster) do
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
75 roster:tag("item", { jid = jid, subscription = "none" }):up();
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
76 end
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
77 send(roster);
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
78 return true;
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
79 end
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
80 end
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
81
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
82
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
83 return function (stanza)
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
84 log("info", "--> "..tostring(stanza));
1
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
85 if (not stanza.attr.to) or (hosts[stanza.attr.to] and hosts[stanza.attr.to].type == "local") then
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
86 if stanza.name == "iq" then
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
87 if not stanza.tags[1] then log("warn", "<iq> without child is invalid"); return; end
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
88 if not stanza.attr.id then log("warn", "<iq> without id attribute is invalid"); end
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
89 local xmlns = (stanza.tags[1].attr and stanza.tags[1].attr.xmlns) or nil;
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
90 if not xmlns then log("warn", "Child of <iq> has no xmlns - invalid"); return; end
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
91 if (((not stanza.attr.to) or stanza.attr.to == session.host or stanza.attr.to:match("@[^/]+$")) and (stanza.attr.type == "get" or stanza.attr.type == "set")) then -- Stanza sent to us
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
92 if iq_handlers[xmlns] then
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
93 if iq_handlers[xmlns](stanza) then return; end;
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
94 end
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
95 log("warn", "Unhandled namespace: "..xmlns);
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
96 send(format("<iq type='error' id='%s'><error type='cancel'><service-unavailable/></error></iq>", stanza.attr.id));
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
97 return;
0
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
98 end
1
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
99 elseif stanza.name == "presence" then
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
100 if session.roster then
2
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
101 local initial_presence = not session.last_presence;
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
102 session.last_presence = stanza;
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
103
1
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
104 -- Broadcast presence and probes
2
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
105 local broadcast = st.presence({ from = session.full_jid, type = stanza.attr.type });
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
106 --local probe = st.presence { from = broadcast.attr.from, type = "probe" };
1
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
107
2
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
108 for child in stanza:childtags() do
4
09c3845ed442 Presence unavailable on disconnect
matthew
parents: 2
diff changeset
109 broadcast:add_child(child);
1
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
110 end
2
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
111 for contact_jid in pairs(session.roster) do
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
112 broadcast.attr.to = contact_jid;
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
113 send_to(contact_jid, broadcast);
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
114 if initial_presence then
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
115 print("Initital presence");
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
116 local node, host = jid.split(contact_jid);
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
117 if hosts[host] and hosts[host].type == "local" then
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
118 local contact = hosts[host].sessions[node]
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
119 if contact then
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
120 local pres = st.presence { to = session.full_jid };
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
121 for resource, contact_session in pairs(contact.sessions) do
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
122 if contact_session.last_presence then
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
123 pres.tags = contact_session.last_presence.tags;
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
124 pres.attr.from = contact_session.full_jid;
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
125 send(pres);
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
126 end
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
127 end
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
128 end
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
129 --FIXME: Do we send unavailable if they are offline?
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
130 else
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
131 probe.attr.to = contact;
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
132 send_to(contact, probe);
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
133 end
9bb397205f26 Working presence!
matthew
parents: 1
diff changeset
134 end
1
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
135 end
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
136
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
137 -- Probe for our contacts' presence
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
138 end
0
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
139 end
1
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
140 else
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
141 --end
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
142 --if stanza.attr.to and ((not hosts[stanza.attr.to]) or hosts[stanza.attr.to].type ~= "local") then
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
143 -- Need to route stanza
0
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
144 stanza.attr.from = session.username.."@"..session.host;
1
b8787e859fd2 Switched to new connection framework, courtesy of the luadch project
matthew
parents: 0
diff changeset
145 session:send_to(stanza.attr.to, stanza);
0
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
146 end
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
147 end
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
148
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
149 end
3e3171b59028 First commit, where do you want to go tomorrow?
matthew
parents:
diff changeset
150