aboutsummaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/lib.h5
-rw-r--r--lib/net.c28
2 files changed, 21 insertions, 12 deletions
diff --git a/lib/lib.h b/lib/lib.h
index 0b93bde9..bb8dfd7f 100644
--- a/lib/lib.h
+++ b/lib/lib.h
@@ -284,8 +284,9 @@ void tty_sigreset(int i);
// net.c
int xsocket(int domain, int type, int protocol);
void xsetsockopt(int fd, int level, int opt, void *val, socklen_t len);
-int xconnect(char *host, char *port, int family, int socktype, int protocol,
- int flags);
+struct addrinfo *xgetaddrinfo(char *host, char *port, int family, int socktype,
+ int protocol, int flags);
+int xconnect(struct addrinfo *ai_arg);
int xpoll(struct pollfd *fds, int nfds, int timeout);
int pollinate(int in1, int in2, int out1, int out2, int timeout, int shutdown_timeout);
diff --git a/lib/net.c b/lib/net.c
index df2551f7..5c6d4f7c 100644
--- a/lib/net.c
+++ b/lib/net.c
@@ -15,11 +15,11 @@ void xsetsockopt(int fd, int level, int opt, void *val, socklen_t len)
if (-1 == setsockopt(fd, level, opt, val, len)) perror_exit("setsockopt");
}
-int xconnect(char *host, char *port, int family, int socktype, int protocol,
- int flags)
+struct addrinfo *xgetaddrinfo(char *host, char *port, int family, int socktype,
+ int protocol, int flags)
{
- struct addrinfo info, *ai, *ai2;
- int fd;
+ struct addrinfo info, *ai;
+ int rc;
memset(&info, 0, sizeof(struct addrinfo));
info.ai_family = family;
@@ -27,20 +27,28 @@ int xconnect(char *host, char *port, int family, int socktype, int protocol,
info.ai_protocol = protocol;
info.ai_flags = flags;
- fd = getaddrinfo(host, port, &info, &ai);
- if (fd || !ai)
- error_exit("Connect '%s%s%s': %s", host, port ? ":" : "", port ? port : "",
- fd ? gai_strerror(fd) : "not found");
+ rc = getaddrinfo(host, port, &info, &ai);
+ if (rc || !ai)
+ error_exit("%s%s%s: %s", host, port ? ":" : "", port ? port : "",
+ rc ? gai_strerror(rc) : "not found");
+
+ return ai;
+}
+
+int xconnect(struct addrinfo *ai_arg)
+{
+ struct addrinfo *ai;
+ int fd = -1;
// Try all the returned addresses. Report errors if last entry can't connect.
- for (ai2 = ai; ai; ai = ai->ai_next) {
+ for (ai = ai_arg; ai; ai = ai->ai_next) {
fd = (ai->ai_next ? socket : xsocket)(ai->ai_family, ai->ai_socktype,
ai->ai_protocol);
if (!connect(fd, ai->ai_addr, ai->ai_addrlen)) break;
else if (!ai->ai_next) perror_exit("connect");
close(fd);
}
- freeaddrinfo(ai2);
+ freeaddrinfo(ai_arg);
return fd;
}