mirror of
https://github.com/bjc/prosody.git
synced 2025-04-03 05:07:42 +03:00
109 lines
3.5 KiB
Lua
109 lines
3.5 KiB
Lua
-- Prosody IM
|
|
-- Copyright (C) 2008-2010 Matthew Wild
|
|
-- Copyright (C) 2008-2010 Waqas Hussain
|
|
--
|
|
-- This project is MIT/X11 licensed. Please see the
|
|
-- COPYING file in the source package for more information.
|
|
--
|
|
|
|
|
|
|
|
local hosts = prosody.hosts;
|
|
local tostring, pairs, setmetatable
|
|
= tostring, pairs, setmetatable;
|
|
|
|
local logger_init = require "util.logger".init;
|
|
|
|
local log = logger_init("s2smanager");
|
|
|
|
local prosody = _G.prosody;
|
|
incoming_s2s = {};
|
|
prosody.incoming_s2s = incoming_s2s;
|
|
local incoming_s2s = incoming_s2s;
|
|
local fire_event = prosody.events.fire_event;
|
|
|
|
local _ENV = nil;
|
|
-- luacheck: std none
|
|
|
|
local function new_incoming(conn)
|
|
local session = { conn = conn, type = "s2sin_unauthed", direction = "incoming", hosts = {} };
|
|
session.log = logger_init("s2sin"..tostring(session):match("[a-f0-9]+$"));
|
|
incoming_s2s[session] = true;
|
|
return session;
|
|
end
|
|
|
|
local function new_outgoing(from_host, to_host)
|
|
local host_session = { to_host = to_host, from_host = from_host, host = from_host,
|
|
notopen = true, type = "s2sout_unauthed", direction = "outgoing" };
|
|
hosts[from_host].s2sout[to_host] = host_session;
|
|
local conn_name = "s2sout"..tostring(host_session):match("[a-f0-9]*$");
|
|
host_session.log = logger_init(conn_name);
|
|
return host_session;
|
|
end
|
|
|
|
local resting_session = { -- Resting, not dead
|
|
destroyed = true;
|
|
type = "s2s_destroyed";
|
|
open_stream = function (session)
|
|
session.log("debug", "Attempt to open stream on resting session");
|
|
end;
|
|
close = function (session)
|
|
session.log("debug", "Attempt to close already-closed session");
|
|
end;
|
|
filter = function (type, data) return data; end; --luacheck: ignore 212/type
|
|
}; resting_session.__index = resting_session;
|
|
|
|
local function retire_session(session, reason)
|
|
local log = session.log or log; --luacheck: ignore 431/log
|
|
for k in pairs(session) do
|
|
if k ~= "log" and k ~= "id" and k ~= "conn" then
|
|
session[k] = nil;
|
|
end
|
|
end
|
|
|
|
session.destruction_reason = reason;
|
|
|
|
function session.send(data) log("debug", "Discarding data sent to resting session: %s", tostring(data)); end
|
|
function session.data(data) log("debug", "Discarding data received from resting session: %s", tostring(data)); end
|
|
session.thread = { run = function (_, data) return session.data(data) end };
|
|
session.sends2s = session.send;
|
|
return setmetatable(session, resting_session);
|
|
end
|
|
|
|
local function destroy_session(session, reason)
|
|
if session.destroyed then return; end
|
|
(session.log or log)("debug", "Destroying "..tostring(session.direction)
|
|
.." session "..tostring(session.from_host).."->"..tostring(session.to_host)
|
|
..(reason and (": "..reason) or ""));
|
|
|
|
if session.direction == "outgoing" then
|
|
hosts[session.from_host].s2sout[session.to_host] = nil;
|
|
session:bounce_sendq(reason);
|
|
elseif session.direction == "incoming" then
|
|
incoming_s2s[session] = nil;
|
|
end
|
|
|
|
local event_data = { session = session, reason = reason };
|
|
if session.type == "s2sout" then
|
|
fire_event("s2sout-destroyed", event_data);
|
|
if hosts[session.from_host] then
|
|
hosts[session.from_host].events.fire_event("s2sout-destroyed", event_data);
|
|
end
|
|
elseif session.type == "s2sin" then
|
|
fire_event("s2sin-destroyed", event_data);
|
|
if hosts[session.to_host] then
|
|
hosts[session.to_host].events.fire_event("s2sin-destroyed", event_data);
|
|
end
|
|
end
|
|
|
|
retire_session(session, reason); -- Clean session until it is GC'd
|
|
return true;
|
|
end
|
|
|
|
return {
|
|
incoming_s2s = incoming_s2s;
|
|
new_incoming = new_incoming;
|
|
new_outgoing = new_outgoing;
|
|
retire_session = retire_session;
|
|
destroy_session = destroy_session;
|
|
};
|