mirror of https://github.com/bjc/prosody
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
2.8 KiB
95 lines
2.8 KiB
local server = require "net.server";
|
|
local log = require "util.logger".init("net.connect");
|
|
local new_id = require "util.id".short;
|
|
|
|
-- TODO #1246 Happy Eyeballs
|
|
-- FIXME RFC 6724
|
|
-- FIXME Error propagation from resolvers doesn't work
|
|
-- FIXME #1428 Reuse DNS resolver object between service and basic resolver
|
|
-- FIXME #1429 Close DNS resolver object when done
|
|
|
|
local pending_connection_methods = {};
|
|
local pending_connection_mt = {
|
|
__name = "pending_connection";
|
|
__index = pending_connection_methods;
|
|
__tostring = function (p)
|
|
return "<pending connection "..p.id.." to "..tostring(p.target_resolver.hostname)..">";
|
|
end;
|
|
};
|
|
|
|
function pending_connection_methods:log(level, message, ...)
|
|
log(level, "[pending connection %s] "..message, self.id, ...);
|
|
end
|
|
|
|
-- pending_connections_map[conn] = pending_connection
|
|
local pending_connections_map = {};
|
|
|
|
local pending_connection_listeners = {};
|
|
|
|
local function attempt_connection(p)
|
|
p:log("debug", "Checking for targets...");
|
|
if p.conn then
|
|
pending_connections_map[p.conn] = nil;
|
|
p.conn = nil;
|
|
end
|
|
p.target_resolver:next(function (conn_type, ip, port, extra)
|
|
if not conn_type then
|
|
-- No more targets to try
|
|
p:log("debug", "No more connection targets to try");
|
|
if p.listeners.onfail then
|
|
p.listeners.onfail(p.data, p.last_error or "unable to resolve service");
|
|
end
|
|
return;
|
|
end
|
|
p:log("debug", "Next target to try is %s:%d", ip, port);
|
|
local conn, err = server.addclient(ip, port, pending_connection_listeners, p.options.pattern or "*a", p.options.sslctx, conn_type, extra);
|
|
if not conn then
|
|
log("debug", "Connection attempt failed immediately: %s", err);
|
|
p.last_error = err or "unknown reason";
|
|
return attempt_connection(p);
|
|
end
|
|
p.conn = conn;
|
|
pending_connections_map[conn] = p;
|
|
end);
|
|
end
|
|
|
|
function pending_connection_listeners.onconnect(conn)
|
|
local p = pending_connections_map[conn];
|
|
if not p then
|
|
log("warn", "Successful connection, but unexpected! Closing.");
|
|
conn:close();
|
|
return;
|
|
end
|
|
pending_connections_map[conn] = nil;
|
|
p:log("debug", "Successfully connected");
|
|
conn:setlistener(p.listeners, p.data);
|
|
return p.listeners.onconnect(conn);
|
|
end
|
|
|
|
function pending_connection_listeners.ondisconnect(conn, reason)
|
|
local p = pending_connections_map[conn];
|
|
if not p then
|
|
log("warn", "Failed connection, but unexpected!");
|
|
return;
|
|
end
|
|
p.last_error = reason or "unknown reason";
|
|
p:log("debug", "Connection attempt failed: %s", p.last_error);
|
|
attempt_connection(p);
|
|
end
|
|
|
|
local function connect(target_resolver, listeners, options, data)
|
|
local p = setmetatable({
|
|
id = new_id();
|
|
target_resolver = target_resolver;
|
|
listeners = assert(listeners);
|
|
options = options or {};
|
|
data = data;
|
|
}, pending_connection_mt);
|
|
|
|
p:log("debug", "Starting connection process");
|
|
attempt_connection(p);
|
|
end
|
|
|
|
return {
|
|
connect = connect;
|
|
};
|
|
|