Software /
code /
prosody
Changeset
1791:42d0b68a1efd
s2smanager: Pass A-record lookups through adns module
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Sat, 19 Sep 2009 17:40:19 +0100 |
parents | 1787:c4dff34f3d32 |
children | 1792:407f282f559e 1793:1fc6c2822e6b |
files | core/s2smanager.lua |
diffstat | 1 files changed, 29 insertions(+), 1 deletions(-) [+] |
line wrap: on
line diff
--- a/core/s2smanager.lua Fri Sep 18 02:46:11 2009 +0100 +++ b/core/s2smanager.lua Sat Sep 19 17:40:19 2009 +0100 @@ -173,7 +173,7 @@ if not err then -- This is our first attempt log("debug", "First attempt to connect to %s, starting with SRV lookup...", to_host); host_session.connecting = true; - local answer, handle; + local handle; handle = adns.lookup(function (answer) handle = nil; host_session.connecting = nil; @@ -235,6 +235,34 @@ end function try_connect(host_session, connect_host, connect_port) + host_session.connecting = true; + local handle; + handle = adns.lookup(function (reply) + handle = nil; + host_session.connecting = nil; + if reply and reply[1] and reply[1].a then + log("debug", "DNS reply for %s gives us %s", connect_host, reply[1].a); + return make_connect(host_session, reply[1].a, connect_port); + else + 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 + end + end + end, connect_host, "A", "IN"); + + -- Set handler for DNS timeout + add_task(dns_timeout, function () + if handle then + adns.cancel(handle, true); + end + end); + + return true; +end + +function make_connect(host_session, connect_host, connect_port) host_session.log("info", "Beginning new connection attempt to %s (%s:%d)", host_session.to_host, connect_host, connect_port); -- Ok, we're going to try to connect