Mercurial > prosody-hg

comparison core/sessionmanager.lua @ 10411:db2a06b9ff98

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge 0.11->trunk
author Kim Alvefur <zash@zash.se>
date Sat, 16 Nov 2019 16:52:31 +0100
parents 04c4750ff8d2
children de90b2f5da8c
comparison
equal deleted inserted replaced
10410:659b577f280c 10411:db2a06b9ff98
19 local rm_load_roster = require "core.rostermanager".load_roster; 19 local rm_load_roster = require "core.rostermanager".load_roster;
20 local config_get = require "core.configmanager".get; 20 local config_get = require "core.configmanager".get;
21 local resourceprep = require "util.encodings".stringprep.resourceprep; 21 local resourceprep = require "util.encodings".stringprep.resourceprep;
22 local nodeprep = require "util.encodings".stringprep.nodeprep; 22 local nodeprep = require "util.encodings".stringprep.nodeprep;
23 local generate_identifier = require "util.id".short; 23 local generate_identifier = require "util.id".short;
24 local sessionlib = require "util.session";
24 25
25 local initialize_filters = require "util.filters".initialize; 26 local initialize_filters = require "util.filters".initialize;
26 local gettime = require "socket".gettime; 27 local gettime = require "socket".gettime;
27 28
28 local _ENV = nil; 29 local _ENV = nil;
29 -- luacheck: std none 30 -- luacheck: std none
30 31
31 local function new_session(conn) 32 local function new_session(conn)
32 local session = { conn = conn, type = "c2s_unauthed", conntime = gettime() }; 33 local session = sessionlib.new("c2s");
34 sessionlib.set_id(session);
35 sessionlib.set_logger(session);
36 sessionlib.set_conn(session, conn);
37
38 session.conntime = gettime();
33 local filter = initialize_filters(session); 39 local filter = initialize_filters(session);
34 local w = conn.write; 40 local w = conn.write;
41
42 function session.rawsend(t)
43 t = filter("bytes/out", tostring(t));
44 if t then
45 local ret, err = w(conn, t);
46 if not ret then
47 session.log("debug", "Error writing to connection: %s", err);
48 return false, err;
49 end
50 end
51 return true;
52 end
53
35 session.send = function (t) 54 session.send = function (t)
36 session.log("debug", "Sending[%s]: %s", session.type, t.top_tag and t:top_tag() or t:match("^[^>]*>?")); 55 session.log("debug", "Sending[%s]: %s", session.type, t.top_tag and t:top_tag() or t:match("^[^>]*>?"));
37 if t.name then 56 if t.name then
38 t = filter("stanzas/out", t); 57 t = filter("stanzas/out", t);
39 end 58 end
40 if t then 59 if t then
41 t = filter("bytes/out", tostring(t)); 60 return session.rawsend(t);
42 if t then
43 local ret, err = w(conn, t);
44 if not ret then
45 session.log("debug", "Error writing to connection: %s", tostring(err));
46 return false, err;
47 end
48 end
49 end 61 end
50 return true; 62 return true;
51 end 63 end
52 session.ip = conn:ip(); 64 session.ip = conn:ip();
53 local conn_name = "c2s"..tostring(session):match("[a-f0-9]+$"); 65 local conn_name = "c2s"..tostring(session):match("[a-f0-9]+$");
71 if k ~= "log" and k ~= "id" then 83 if k ~= "log" and k ~= "id" then
72 session[k] = nil; 84 session[k] = nil;
73 end 85 end
74 end 86 end
75 87
76 function session.send(data) log("debug", "Discarding data sent to resting session: %s", tostring(data)); return false; end 88 function session.send(data) log("debug", "Discarding data sent to resting session: %s", data); return false; end
77 function session.data(data) log("debug", "Discarding data received from resting session: %s", tostring(data)); end 89 function session.data(data) log("debug", "Discarding data received from resting session: %s", data); end
78 session.thread = { run = function (_, data) return session.data(data) end }; 90 session.thread = { run = function (_, data) return session.data(data) end };
79 return setmetatable(session, resting_session); 91 return setmetatable(session, resting_session);
80 end 92 end
81 93
82 local function destroy_session(session, err) 94 local function destroy_session(session, err)
115 if not username or #username == 0 then return nil, "Invalid username"; end 127 if not username or #username == 0 then return nil, "Invalid username"; end
116 session.username = username; 128 session.username = username;
117 if session.type == "c2s_unauthed" then 129 if session.type == "c2s_unauthed" then
118 session.type = "c2s_unbound"; 130 session.type = "c2s_unbound";
119 end 131 end
120 session.log("info", "Authenticated as %s@%s", username or "(unknown)", session.host or "(unknown)"); 132 session.log("info", "Authenticated as %s@%s", username, session.host or "(unknown)");
121 return true; 133 return true;
122 end 134 end
123 135
124 -- returns true, nil on success 136 -- returns true, nil on success
125 -- returns nil, err_type, err, err_message on failure 137 -- returns nil, err_type, err, err_message on failure
136 else 148 else
137 -- In case a plugin wants to poke at it 149 -- In case a plugin wants to poke at it
138 resource = event_payload.resource; 150 resource = event_payload.resource;
139 end 151 end
140 152
141 resource = resourceprep(resource); 153 resource = resourceprep(resource or "", true);
142 resource = resource ~= "" and resource or generate_identifier(); 154 resource = resource ~= "" and resource or generate_identifier();
143 --FIXME: Randomly-generated resources must be unique per-user, and never conflict with existing 155 --FIXME: Randomly-generated resources must be unique per-user, and never conflict with existing
144 156
145 if not hosts[session.host].sessions[session.username] then 157 if not hosts[session.host].sessions[session.username] then
146 local sessions = { sessions = {} }; 158 local sessions = { sessions = {} };