Fri, 03 Oct 2008 22:18:08 +0100
Move stream opening handling from xmlhandlers to sessionmanager
30 | 1 | |
40
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
2 | local tonumber, tostring = tonumber, tostring; |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
3 | local ipairs = ipairs; |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
4 | |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
5 | local m_random = math.random; |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
6 | local format = string.format; |
30 | 7 | |
38 | 8 | local print = print; |
9 | ||
10 | local hosts = hosts; | |
11 | ||
40
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
12 | local modulemanager = require "core.modulemanager"; |
30 | 13 | local log = require "util.logger".init("sessionmanager"); |
40
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
14 | local error = error; |
30 | 15 | |
16 | module "sessionmanager" | |
17 | ||
18 | function new_session(conn) | |
19 | local session = { conn = conn, notopen = true, priority = 0, type = "c2s_unauthed" }; | |
20 | local w = conn.write; | |
21 | session.send = function (t) w(tostring(t)); end | |
22 | return session; | |
23 | end | |
24 | ||
38 | 25 | function destroy_session(session) |
26 | end | |
27 | ||
30 | 28 | function send_to_session(session, data) |
38 | 29 | log("debug", "Sending: %s", tostring(data)); |
30 | 30 | session.conn.write(tostring(data)); |
31 | end | |
32 | ||
38 | 33 | function make_authenticated(session, username) |
34 | session.username = username; | |
35 | session.resource = resource; | |
36 | if session.type == "c2s_unauthed" then | |
37 | session.type = "c2s"; | |
38 | end | |
39 | end | |
40 | ||
41 | function bind_resource(session, resource) | |
42 | if not session.username then return false, "auth"; end | |
43 | if session.resource then return false, "constraint"; end -- We don't support binding multiple resources | |
44 | resource = resource or math.random(100000, 99999999); -- FIXME: Clearly we have issues :) | |
45 | --FIXME: Randomly-generated resources must be unique per-user, and never conflict with existing | |
46 | ||
47 | if not hosts[session.host].sessions[session.username] then | |
48 | hosts[session.host].sessions[session.username] = { sessions = {} }; | |
49 | else | |
50 | if hosts[session.host].sessions[session.username].sessions[resource] then | |
51 | -- Resource conflict | |
52 | return false, "conflict"; | |
53 | end | |
54 | end | |
55 | ||
56 | session.resource = resource; | |
57 | session.full_jid = session.username .. '@' .. session.host .. '/' .. resource; | |
58 | hosts[session.host].sessions[session.username].sessions[resource] = session; | |
59 | ||
60 | return true; | |
61 | end | |
62 | ||
40
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
63 | function streamopened(session, attr) |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
64 | local send = session.send; |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
65 | session.host = attr.to or error("Client failed to specify destination hostname"); |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
66 | session.version = tonumber(attr.version) or 0; |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
67 | session.streamid = m_random(1000000, 99999999); |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
68 | print(session, session.host, "Client opened stream"); |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
69 | send("<?xml version='1.0'?>"); |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
70 | send(format("<stream:stream xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' id='%s' from='%s' version='1.0'>", session.streamid, session.host)); |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
71 | |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
72 | local features = {}; |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
73 | modulemanager.fire_event("stream-features", session, features); |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
74 | |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
75 | send("<stream:features>"); |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
76 | |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
77 | for _, feature in ipairs(features) do |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
78 | send_to_session(session, tostring(features)); |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
79 | end |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
80 | |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
81 | send("</stream:features>"); |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
82 | log("info", "core", "Stream opened successfully"); |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
83 | session.notopen = nil; |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
84 | end |
2c0147bbd81a
Move stream opening handling from xmlhandlers to sessionmanager
Matthew Wild <mwild1@gmail.com>
parents:
38
diff
changeset
|
85 | |
30 | 86 | return _M; |