core/s2smanager.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2010 Matthew Wild
   3 -- Copyright (C) 2008-2010 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9
  10
  11 local hosts = prosody.hosts;
  12 local tostring, pairs, setmetatable
  13     = tostring, pairs, setmetatable;
  14
  15 local logger_init = require "util.logger".init;
  16
  17 local log = logger_init("s2smanager");
  18
  19 local prosody = _G.prosody;
  20 incoming_s2s = {};
  21 prosody.incoming_s2s = incoming_s2s;
  22 local incoming_s2s = incoming_s2s;
  23 local fire_event = prosody.events.fire_event;
  24
  25 module "s2smanager"
  26
  27 function new_incoming(conn)
  28         local session = { conn = conn, type = "s2sin_unauthed", direction = "incoming", hosts = {} };
  29         session.log = logger_init("s2sin"..tostring(session):match("[a-f0-9]+$"));
  30         incoming_s2s[session] = true;
  31         return session;
  32 end
  33
  34 function new_outgoing(from_host, to_host)
  35         local host_session = { to_host = to_host, from_host = from_host, host = from_host,
  36                                notopen = true, type = "s2sout_unauthed", direction = "outgoing" };
  37         hosts[from_host].s2sout[to_host] = host_session;
  38         local conn_name = "s2sout"..tostring(host_session):match("[a-f0-9]*$");
  39         host_session.log = logger_init(conn_name);
  40         return host_session;
  41 end
  42
  43 local resting_session = { -- Resting, not dead
  44                 destroyed = true;
  45                 type = "s2s_destroyed";
  46                 open_stream = function (session)
  47                         session.log("debug", "Attempt to open stream on resting session");
  48                 end;
  49                 close = function (session)
  50                         session.log("debug", "Attempt to close already-closed session");
  51                 end;
  52                 filter = function (type, data) return data; end;
  53         }; resting_session.__index = resting_session;
  54
  55 function retire_session(session, reason)
  56         local log = session.log or log;
  57         for k in pairs(session) do
  58                 if k ~= "log" and k ~= "id" and k ~= "conn" then
  59                         session[k] = nil;
  60                 end
  61         end
  62
  63         session.destruction_reason = reason;
  64
  65         function session.send(data) log("debug", "Discarding data sent to resting session: %s", tostring(data)); end
  66         function session.data(data) log("debug", "Discarding data received from resting session: %s", tostring(data)); end
  67         return setmetatable(session, resting_session);
  68 end
  69
  70 function destroy_session(session, reason)
  71         if session.destroyed then return; end
  72         (session.log or log)("debug", "Destroying "..tostring(session.direction).." session "..tostring(session.from_host).."->"..tostring(session.to_host)..(reason and (": "..reason) or ""));
  73
  74         if session.direction == "outgoing" then
  75                 hosts[session.from_host].s2sout[session.to_host] = nil;
  76                 session:bounce_sendq(reason);
  77         elseif session.direction == "incoming" then
  78                 incoming_s2s[session] = nil;
  79         end
  80
  81         local event_data = { session = session, reason = reason };
  82         if session.type == "s2sout" then
  83                 fire_event("s2sout-destroyed", event_data);
  84                 if hosts[session.from_host] then
  85                         hosts[session.from_host].events.fire_event("s2sout-destroyed", event_data);
  86                 end
  87         elseif session.type == "s2sin" then
  88                 fire_event("s2sin-destroyed", event_data);
  89                 if hosts[session.to_host] then
  90                         hosts[session.to_host].events.fire_event("s2sin-destroyed", event_data);
  91                 end
  92         end
  93
  94         retire_session(session, reason); -- Clean session until it is GC'd
  95         return true;
  96 end
  97
  98 return _M;