Software `/` code `/` prosody

Annotate

util/adminstream.lua @ 12905:8473a516004f

core.usermanager: Add methods for enabling and disabling users Calling into the auth module, where available.

author	Kim Alvefur <zash@zash.se>
date	Thu, 23 Feb 2023 16:24:41 +0100
parent	12852:c35afa353f8f
child	12975:d10957394a3c

rev	line source
10855 70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	1 local st = require "util.stanza";
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	2 local new_xmpp_stream = require "util.xmppstream".new;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	3 local sessionlib = require "util.session";
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	4 local gettime = require "util.time".now;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	5 local runner = require "util.async".runner;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	6 local add_task = require "util.timer".add_task;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	7 local events = require "util.events";
10876 c01c39a2c7a2 util.adminstream: Import net.server [luacheck] Matthew Wild <mwild1@gmail.com> parents: 10875 diff changeset	8 local server = require "net.server";
10855 70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	9
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	10 local stream_close_timeout = 5;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	11
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	12 local log = require "util.logger".init("adminstream");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	13
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	14 local xmlns_xmpp_streams = "urn:ietf:params:xml:ns:xmpp-streams";
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	15
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	16 local stream_callbacks = { default_ns = "xmpp:prosody.im/admin" };
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	17
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	18 function stream_callbacks.streamopened(session, attr)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	19 -- run _streamopened in async context
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	20 session.thread:run({ stream = "opened", attr = attr });
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	21 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	22
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	23 function stream_callbacks._streamopened(session, attr) --luacheck: ignore 212/attr
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	24 if session.type ~= "client" then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	25 session:open_stream();
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	26 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	27 session.notopen = nil;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	28 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	29
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	30 function stream_callbacks.streamclosed(session, attr)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	31 -- run _streamclosed in async context
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	32 session.thread:run({ stream = "closed", attr = attr });
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	33 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	34
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	35 function stream_callbacks._streamclosed(session)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	36 session.log("debug", "Received </stream:stream>");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	37 session:close(false);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	38 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	39
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	40 function stream_callbacks.error(session, error, data)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	41 if error == "no-stream" then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	42 session.log("debug", "Invalid opening stream header (%s)", (data:gsub("^([^\1]+)\1", "{%1}")));
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	43 session:close("invalid-namespace");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	44 elseif error == "parse-error" then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	45 session.log("debug", "Client XML parse error: %s", data);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	46 session:close("not-well-formed");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	47 elseif error == "stream-error" then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	48 local condition, text = "undefined-condition";
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	49 for child in data:childtags(nil, xmlns_xmpp_streams) do
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	50 if child.name ~= "text" then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	51 condition = child.name;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	52 else
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	53 text = child:get_text();
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	54 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	55 if condition ~= "undefined-condition" and text then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	56 break;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	57 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	58 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	59 text = condition .. (text and (" ("..text..")") or "");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	60 session.log("info", "Session closed by remote with error: %s", text);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	61 session:close(nil, text);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	62 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	63 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	64
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	65 function stream_callbacks.handlestanza(session, stanza)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	66 session.thread:run(stanza);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	67 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	68
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	69 local runner_callbacks = {};
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	70
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	71 function runner_callbacks:error(err)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	72 self.data.log("error", "Traceback[c2s]: %s", err);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	73 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	74
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	75 local stream_xmlns_attr = {xmlns='urn:ietf:params:xml:ns:xmpp-streams'};
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	76
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	77 local function destroy_session(session, reason)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	78 if session.destroyed then return; end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	79 session.destroyed = true;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	80 session.log("debug", "Destroying session: %s", reason or "unknown reason");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	81 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	82
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	83 local function session_close(session, reason)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	84 local log = session.log or log;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	85 if session.conn then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	86 if session.notopen then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	87 session:open_stream();
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	88 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	89 if reason then -- nil == no err, initiated by us, false == initiated by client
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	90 local stream_error = st.stanza("stream:error");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	91 if type(reason) == "string" then -- assume stream error
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	92 stream_error:tag(reason, {xmlns = 'urn:ietf:params:xml:ns:xmpp-streams' });
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	93 elseif type(reason) == "table" then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	94 if reason.condition then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	95 stream_error:tag(reason.condition, stream_xmlns_attr):up();
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	96 if reason.text then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	97 stream_error:tag("text", stream_xmlns_attr):text(reason.text):up();
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	98 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	99 if reason.extra then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	100 stream_error:add_child(reason.extra);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	101 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	102 elseif reason.name then -- a stanza
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	103 stream_error = reason;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	104 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	105 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	106 stream_error = tostring(stream_error);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	107 log("debug", "Disconnecting client, <stream:error> is: %s", stream_error);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	108 session.send(stream_error);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	109 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	110
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	111 session.send("</stream:stream>");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	112 function session.send() return false; end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	113
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	114 local reason_text = (reason and (reason.name or reason.text or reason.condition)) or reason;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	115 session.log("debug", "c2s stream for %s closed: %s", session.full_jid or session.ip or "<unknown>", reason_text or "session closed");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	116
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	117 -- Authenticated incoming stream may still be sending us stanzas, so wait for </stream:stream> from remote
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	118 local conn = session.conn;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	119 if reason_text == nil and not session.notopen and session.type == "c2s" then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	120 -- Grace time to process data from authenticated cleanly-closed stream
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	121 add_task(stream_close_timeout, function ()
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	122 if not session.destroyed then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	123 session.log("warn", "Failed to receive a stream close response, closing connection anyway...");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	124 destroy_session(session);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	125 conn:close();
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	126 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	127 end);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	128 else
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	129 destroy_session(session, reason_text);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	130 conn:close();
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	131 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	132 else
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	133 local reason_text = (reason and (reason.name or reason.text or reason.condition)) or reason;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	134 destroy_session(session, reason_text);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	135 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	136 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	137
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	138 --- Public methods
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	139
10875 09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	140 local function new_connection(socket_path, listeners)
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	141 local have_unix, unix = pcall(require, "socket.unix");
12392 5373724e08a5 mod_admin_socket: Compat for luasocket prior to unix datagram support Kim Alvefur <zash@zash.se> parents: 12094 diff changeset	142 if have_unix and type(unix) == "function" then
12393 6966026262f4 mod_admin_socket: Comment on LuaSocket UNIX compat code Kim Alvefur <zash@zash.se> parents: 12392 diff changeset	143 -- COMPAT #1717
6966026262f4 mod_admin_socket: Comment on LuaSocket UNIX compat code Kim Alvefur <zash@zash.se> parents: 12392 diff changeset	144 -- Before the introduction of datagram support, only the stream socket
6966026262f4 mod_admin_socket: Comment on LuaSocket UNIX compat code Kim Alvefur <zash@zash.se> parents: 12392 diff changeset	145 -- constructor was exported instead of a module table. Due to the lack of a
6966026262f4 mod_admin_socket: Comment on LuaSocket UNIX compat code Kim Alvefur <zash@zash.se> parents: 12392 diff changeset	146 -- proper release of LuaSocket, distros have settled on shipping either the
6966026262f4 mod_admin_socket: Comment on LuaSocket UNIX compat code Kim Alvefur <zash@zash.se> parents: 12392 diff changeset	147 -- last RC tag or some commit since then.
12852 c35afa353f8f mod_admin_socket: Fix typo in comments Kim Alvefur <zash@zash.se> parents: 12393 diff changeset	148 -- Here we accommodate both variants.
12392 5373724e08a5 mod_admin_socket: Compat for luasocket prior to unix datagram support Kim Alvefur <zash@zash.se> parents: 12094 diff changeset	149 unix = { stream = unix };
5373724e08a5 mod_admin_socket: Compat for luasocket prior to unix datagram support Kim Alvefur <zash@zash.se> parents: 12094 diff changeset	150 end
10875 09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	151 if type(unix) ~= "table" then
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	152 have_unix = false;
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	153 end
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	154 local conn, sock;
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	155
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	156 return {
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	157 connect = function ()
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	158 if not have_unix then
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	159 return nil, "no unix socket support";
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	160 end
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	161 if sock or conn then
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	162 return nil, "already connected";
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	163 end
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	164 sock = unix.stream();
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	165 sock:settimeout(0);
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	166 local ok, err = sock:connect(socket_path);
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	167 if not ok then
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	168 return nil, err;
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	169 end
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	170 conn = server.wrapclient(sock, nil, nil, listeners, "*a");
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	171 return true;
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	172 end;
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	173 disconnect = function ()
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	174 if conn then
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	175 conn:close();
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	176 conn = nil;
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	177 end
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	178 if sock then
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	179 sock:close();
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	180 sock = nil;
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	181 end
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	182 return true;
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	183 end;
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	184 };
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	185 end
09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	186
10855 70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	187 local function new_server(sessions, stanza_handler)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	188 local listeners = {};
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	189
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	190 function listeners.onconnect(conn)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	191 log("debug", "New connection");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	192 local session = sessionlib.new("admin");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	193 sessionlib.set_id(session);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	194 sessionlib.set_logger(session);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	195 sessionlib.set_conn(session, conn);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	196
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	197 session.conntime = gettime();
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	198 session.type = "admin";
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	199
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	200 local stream = new_xmpp_stream(session, stream_callbacks);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	201 session.stream = stream;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	202 session.notopen = true;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	203
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	204 session.thread = runner(function (stanza)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	205 if st.is_stanza(stanza) then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	206 stanza_handler(session, stanza);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	207 elseif stanza.stream == "opened" then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	208 stream_callbacks._streamopened(session, stanza.attr);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	209 elseif stanza.stream == "closed" then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	210 stream_callbacks._streamclosed(session, stanza.attr);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	211 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	212 end, runner_callbacks, session);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	213
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	214 function session.data(data)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	215 -- Parse the data, which will store stanzas in session.pending_stanzas
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	216 if data then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	217 local ok, err = stream:feed(data);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	218 if not ok then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	219 session.log("debug", "Received invalid XML (%s) %d bytes: %q", err, #data, data:sub(1, 300));
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	220 session:close("not-well-formed");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	221 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	222 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	223 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	224
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	225 session.close = session_close;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	226
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	227 session.send = function (t)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	228 session.log("debug", "Sending[%s]: %s", session.type, t.top_tag and t:top_tag() or t:match("^[^>]*>?"));
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	229 return session.rawsend(tostring(t));
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	230 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	231
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	232 function session.rawsend(t)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	233 local ret, err = conn:write(t);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	234 if not ret then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	235 session.log("debug", "Error writing to connection: %s", err);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	236 return false, err;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	237 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	238 return true;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	239 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	240
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	241 sessions[conn] = session;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	242 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	243
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	244 function listeners.onincoming(conn, data)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	245 local session = sessions[conn];
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	246 if session then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	247 session.data(data);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	248 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	249 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	250
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	251 function listeners.ondisconnect(conn, err)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	252 local session = sessions[conn];
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	253 if session then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	254 session.log("info", "Admin client disconnected: %s", err or "connection closed");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	255 session.conn = nil;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	256 sessions[conn] = nil;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	257 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	258 end
10900 9e6d979dd603 util.adminstream: Set a read timeout handler Kim Alvefur <zash@zash.se> parents: 10892 diff changeset	259
9e6d979dd603 util.adminstream: Set a read timeout handler Kim Alvefur <zash@zash.se> parents: 10892 diff changeset	260 function listeners.onreadtimeout(conn)
10940 18e4e446a76c util.adminstream: Prevent closure on read timeout Kim Alvefur <zash@zash.se> parents: 10900 diff changeset	261 return conn:send(" ");
10900 9e6d979dd603 util.adminstream: Set a read timeout handler Kim Alvefur <zash@zash.se> parents: 10892 diff changeset	262 end
9e6d979dd603 util.adminstream: Set a read timeout handler Kim Alvefur <zash@zash.se> parents: 10892 diff changeset	263
10855 70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	264 return {
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	265 listeners = listeners;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	266 };
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	267 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	268
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	269 local function new_client()
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	270 local client = {
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	271 type = "client";
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	272 events = events.new();
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	273 log = log;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	274 };
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	275
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	276 local listeners = {};
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	277
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	278 function listeners.onconnect(conn)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	279 log("debug", "Connected");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	280 client.conn = conn;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	281
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	282 local stream = new_xmpp_stream(client, stream_callbacks);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	283 client.stream = stream;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	284 client.notopen = true;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	285
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	286 client.thread = runner(function (stanza)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	287 if st.is_stanza(stanza) then
10892 b9ff7178787c util.adminstream: Fire event based on stanza name too for convenience Matthew Wild <mwild1@gmail.com> parents: 10876 diff changeset	288 if not client.events.fire_event("received", stanza) and not stanza.attr.xmlns then
b9ff7178787c util.adminstream: Fire event based on stanza name too for convenience Matthew Wild <mwild1@gmail.com> parents: 10876 diff changeset	289 client.events.fire_event("received/"..stanza.name, stanza);
b9ff7178787c util.adminstream: Fire event based on stanza name too for convenience Matthew Wild <mwild1@gmail.com> parents: 10876 diff changeset	290 end
10855 70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	291 elseif stanza.stream == "opened" then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	292 stream_callbacks._streamopened(client, stanza.attr);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	293 client.events.fire_event("connected");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	294 elseif stanza.stream == "closed" then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	295 client.events.fire_event("disconnected");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	296 stream_callbacks._streamclosed(client, stanza.attr);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	297 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	298 end, runner_callbacks, client);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	299
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	300 client.close = session_close;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	301
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	302 function client.send(t)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	303 client.log("debug", "Sending: %s", t.top_tag and t:top_tag() or t:match("^[^>]*>?"));
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	304 return client.rawsend(tostring(t));
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	305 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	306
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	307 function client.rawsend(t)
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	308 local ret, err = conn:write(t);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	309 if not ret then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	310 client.log("debug", "Error writing to connection: %s", err);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	311 return false, err;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	312 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	313 return true;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	314 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	315 client.log("debug", "Opening stream...");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	316 client:open_stream();
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	317 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	318
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	319 function listeners.onincoming(conn, data) --luacheck: ignore 212/conn
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	320 local ok, err = client.stream:feed(data);
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	321 if not ok then
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	322 client.log("debug", "Received invalid XML (%s) %d bytes: %q", err, #data, data:sub(1, 300));
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	323 client:close("not-well-formed");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	324 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	325 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	326
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	327 function listeners.ondisconnect(conn, err) --luacheck: ignore 212/conn
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	328 client.log("info", "Admin client disconnected: %s", err or "connection closed");
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	329 client.conn = nil;
12094 84fd6a79cda7 adminstream: fire disconnected event on lower layer disconnect Jonas Schäfer <jonas@wielicki.name> parents: 10940 diff changeset	330 client.events.fire_event("disconnected");
10855 70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	331 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	332
10900 9e6d979dd603 util.adminstream: Set a read timeout handler Kim Alvefur <zash@zash.se> parents: 10892 diff changeset	333 function listeners.onreadtimeout(conn)
9e6d979dd603 util.adminstream: Set a read timeout handler Kim Alvefur <zash@zash.se> parents: 10892 diff changeset	334 conn:send(" ");
9e6d979dd603 util.adminstream: Set a read timeout handler Kim Alvefur <zash@zash.se> parents: 10892 diff changeset	335 end
9e6d979dd603 util.adminstream: Set a read timeout handler Kim Alvefur <zash@zash.se> parents: 10892 diff changeset	336
10855 70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	337 client.listeners = listeners;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	338
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	339 return client;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	340 end
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	341
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	342 return {
10875 09674bbb833f util.prosodyctl.shell, util.adminstream: Move connection logic into adminstream for easier reuse Matthew Wild <mwild1@gmail.com> parents: 10855 diff changeset	343 connection = new_connection;
10855 70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	344 server = new_server;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	345 client = new_client;
70ac7d23673d mod_admin_socket, util.adminstream: New module to manage a local unix domain socket for admin functionality Matthew Wild <mwild1@gmail.com> parents: diff changeset	346 };

prosody

8473a516004f

util/adminstream.lua

Software / code / prosody

Annotate

util/adminstream.lua @ 12905:8473a516004f

Software `/` code `/` prosody