util.pposix: Replace the unwieldy module table generation with luaL_register() call...
[prosody.git] / core / s2smanager.lua
index 04e5fa5e845c7d0bee5969ec48f1739b0f39e5da..af5e91e3585705eb946b68a2ca3759a53848232c 100644 (file)
@@ -41,7 +41,6 @@ local config = require "core.configmanager";
 local connect_timeout = config.get("*", "core", "s2s_timeout") or 60;
 local dns_timeout = config.get("*", "core", "dns_timeout") or 60;
 local max_dns_depth = config.get("*", "core", "dns_max_depth") or 3;
-local dialback_secret = config.get("*", "core", "dialback_secret") or uuid_gen();
 
 incoming_s2s = {};
 _G.prosody.incoming_s2s = incoming_s2s;
@@ -142,16 +141,17 @@ function new_incoming(conn)
                        return; -- Ok, we're connect[ed|ing]
                end
                -- Not connected, need to close session and clean up
-               (session.log or log)("warn", "Destroying incomplete session %s->%s due to inactivity", 
+               (session.log or log)("warn", "Destroying incomplete session %s->%s due to inactivity",
                    session.from_host or "(unknown)", session.to_host or "(unknown)");
                session:close("connection-timeout");
        end);
        return session;
 end
 
-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" };
+function new_outgoing(from_host, to_host, connect)
+               local host_session = { to_host = to_host, from_host = from_host, host = from_host,
+                                      notopen = true, type = "s2sout_unauthed", direction = "outgoing",
+                                      open_stream = session_open_stream };
                
                hosts[from_host].s2sout[to_host] = host_session;
                
@@ -162,10 +162,12 @@ function new_outgoing(from_host, to_host)
                        host_session.log = log;
                end
                
-               -- Kick the connection attempting machine
-               attempt_connection(host_session);
+               if connect ~= false then
+                       -- Kick the connection attempting machine into life
+                       attempt_connection(host_session);
+               end
                
-               if not host_session.sends2s then                
+               if not host_session.sends2s then
                        -- A sends2s which buffers data (until the stream is opened)
                        -- note that data in this buffer will be sent before the stream is authed
                        -- and will not be ack'd in any way, successful or otherwise
@@ -224,7 +226,7 @@ function attempt_connection(host_session, err)
                        if not ok then
                                if not attempt_connection(host_session, err) then
                                        -- No more attempts will be made
-                                       destroy_session(host_session);
+                                       destroy_session(host_session, err);
                                end
                        end
                end, "_xmpp-server._tcp."..connect_host..".", "SRV");
@@ -284,7 +286,7 @@ function try_connect(host_session, connect_host, connect_port)
                        log("debug", "DNS lookup failed to get a response for %s", connect_host);
                        if not attempt_connection(host_session, "name resolution failed") then -- Retry if we can
                                log("debug", "No other records to try for %s - destroying", host_session.to_host);
-                               destroy_session(host_session); -- End of the line, we can't
+                               destroy_session(host_session, "DNS resolution failed"); -- End of the line, we can't
                        end
                end
        end, connect_host, "A", "IN");
@@ -295,7 +297,7 @@ function try_connect(host_session, connect_host, connect_port)
                        adns.cancel(handle, true);
                end
        end);
-               
+       
        return true;
 end
 
@@ -325,7 +327,8 @@ function make_connect(host_session, connect_host, connect_port)
        local w, log = conn.write, host_session.log;
        host_session.sends2s = function (t) log("debug", "sending: %s", (t.top_tag and t:top_tag()) or t:match("^[^>]*>?")); w(conn, tostring(t)); end
        
-       conn:write(format([[<stream:stream xmlns='jabber:server' xmlns:db='jabber:server:dialback' xmlns:stream='http://etherx.jabber.org/streams' from='%s' to='%s' version='1.0' xml:lang='en'>]], from_host, to_host));
+       host_session:open_stream();
+       
        log("debug", "Connection attempt in progress...");
        add_task(connect_timeout, function ()
                if host_session.conn ~= conn or
@@ -334,13 +337,20 @@ function make_connect(host_session, connect_host, connect_port)
                        return; -- Ok, we're connect[ed|ing]
                end
                -- Not connected, need to close session and clean up
-               (host_session.log or log)("warn", "Destroying incomplete session %s->%s due to inactivity", 
+               (host_session.log or log)("warn", "Destroying incomplete session %s->%s due to inactivity",
                    host_session.from_host or "(unknown)", host_session.to_host or "(unknown)");
                host_session:close("connection-timeout");
        end);
        return true;
 end
 
+function session_open_stream(session, from, to)
+       session.sends2s(st.stanza("stream:stream", {
+               xmlns='jabber:server', ["xmlns:db"]='jabber:server:dialback',
+               ["xmlns:stream"]='http://etherx.jabber.org/streams',
+               from=from, to=to, version='1.0', ["xml:lang"]='en'}):top_tag());
+end
+
 function streamopened(session, attr)
        local send = session.sends2s;
        
@@ -363,17 +373,17 @@ function streamopened(session, attr)
        
                session.streamid = uuid_gen();
                (session.log or log)("debug", "incoming s2s received <stream:stream>");
-               send("<?xml version='1.0'?>");
-               send(stanza("stream:stream", { xmlns='jabber:server', ["xmlns:db"]='jabber:server:dialback', 
-                               ["xmlns:stream"]='http://etherx.jabber.org/streams', id=session.streamid, from=session.to_host, version=(session.version > 0 and "1.0" or nil) }):top_tag());
-               session.notopen = nil;
                if session.to_host and not hosts[session.to_host] then
                        -- Attempting to connect to a host we don't serve
                        session:close({ condition = "host-unknown"; text = "This host does not serve "..session.to_host });
                        return;
                end
+               send("<?xml version='1.0'?>");
+               send(stanza("stream:stream", { xmlns='jabber:server', ["xmlns:db"]='jabber:server:dialback',
+                               ["xmlns:stream"]='http://etherx.jabber.org/streams', id=session.streamid, from=session.to_host, to=session.from_host, version=(session.version > 0 and "1.0" or nil) }):top_tag());
                if session.version >= 1.0 then
                        local features = st.stanza("stream:features");
+                       
                        if session.to_host then
                                hosts[session.to_host].events.fire_event("s2s-stream-features", { session = session, features = features });
                        else
@@ -412,6 +422,7 @@ function streamopened(session, attr)
                        end
                end
        end
+       session.notopen = nil;
 end
 
 function streamclosed(session)
@@ -430,7 +441,7 @@ function initiate_dialback(session)
 end
 
 function generate_dialback(id, to, from)
-       return sha256_hash(id..to..from..dialback_secret, true);
+       return sha256_hash(id..to..from..hosts[from].dialback_secret, true);
 end
 
 function verify_dialback(id, to, from, key)
@@ -478,6 +489,7 @@ function mark_connected(session)
                        end
                        session.sendq = nil;
                end
+               
                session.srv_hosts = nil;
        end
 end