Software / code / prosody
Annotate
core/sessionmanager.lua @ 41:68297fef35ff
An oops in sessionmanager stream:features code :)
| author | Matthew Wild <mwild1@gmail.com> |
|---|---|
| date | Fri, 03 Oct 2008 22:21:01 +0100 |
| parent | 40:2c0147bbd81a |
| child | 44:80d2ade0fd69 |
| rev | line source |
|---|---|
| 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 |
|
41
68297fef35ff
An oops in sessionmanager stream:features code :)
Matthew Wild <mwild1@gmail.com>
parents:
40
diff
changeset
|
78 send_to_session(session, tostring(feature)); |
|
40
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; |