summaryrefslogtreecommitdiff
path: root/usr.bin/whois
diff options
context:
space:
mode:
authorConrad Meyer <cem@FreeBSD.org>2016-05-11 18:03:51 +0000
committerConrad Meyer <cem@FreeBSD.org>2016-05-11 18:03:51 +0000
commitc24375a7ded40ccac59c4e9811c8deb614ee4f2b (patch)
tree789183b44bad419bca19794b385f787d0504459d /usr.bin/whois
parent5a333f6f5fe2499acd20481e9cbc1425ed3145ef (diff)
downloadsrc-test2-c24375a7ded40ccac59c4e9811c8deb614ee4f2b.tar.gz
src-test2-c24375a7ded40ccac59c4e9811c8deb614ee4f2b.zip
Notes
Diffstat (limited to 'usr.bin/whois')
-rw-r--r--usr.bin/whois/whois.c12
1 files changed, 9 insertions, 3 deletions
diff --git a/usr.bin/whois/whois.c b/usr.bin/whois/whois.c
index 6519f068286e..8c775fa041ae 100644
--- a/usr.bin/whois/whois.c
+++ b/usr.bin/whois/whois.c
@@ -316,6 +316,11 @@ connect_to_any_host(struct addrinfo *hostres)
fds[i].fd = s;
fds[i].events = POLLERR | POLLHUP |
POLLIN | POLLOUT;
+ /*
+ * From here until a socket connects, the
+ * socket fd is owned by the fds[] poll array.
+ */
+ s = -1;
count++;
i++;
} else {
@@ -357,7 +362,7 @@ connect_to_any_host(struct addrinfo *hostres)
* after a new host have been added.
*/
if (timeout >= 3)
- timeout <<= 1;
+ timeout >>= 1;
break;
} else if (n < 0) {
@@ -377,7 +382,7 @@ connect_to_any_host(struct addrinfo *hostres)
fds[j].revents == 0)
continue;
if (fds[j].revents & ~(POLLIN | POLLOUT)) {
- close(s);
+ close(fds[j].fd);
fds[j].fd = -1;
fds[j].events = 0;
count--;
@@ -385,6 +390,7 @@ connect_to_any_host(struct addrinfo *hostres)
} else if (fds[j].revents & (POLLIN | POLLOUT)) {
/* Connect succeeded. */
s = fds[j].fd;
+ fds[j].fd = -1;
goto done;
}
@@ -401,7 +407,7 @@ connect_to_any_host(struct addrinfo *hostres)
done:
/* Close all watched fds except the succeeded one */
for (j = 0; j < i; j++)
- if (fds[j].fd != s && fds[j].fd != -1)
+ if (fds[j].fd != -1)
close(fds[j].fd);
free(fds);
return (s);