system/freebsd-src
system/freebsd-src
1
0
Fork 0
mirror of https://github.com/freebsd/freebsd-src synced 2024-10-07 00:50:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

whois(1): Fix potential double-close and logic mistakes

Browse source 
Close the fd the poll error was detected on, rather than the last opened fd, to
fix the double-close.

Use -1 to make it explict which int variables no longer own socket file
descriptors.

Actually shrink, rather than grow, the poll timeout to match comment.

Reported by:	Coverity
CID:		1304860, 1305616
Sponsored by:	EMC / Isilon Storage Division
This commit is contained in:
Conrad Meyer 2016-05-11 18:03:51 +00:00
parent 5a333f6f5f
commit c24375a7de
Notes: svn2git 2020-12-20 02:59:44 +00:00 
svn path=/head/; revision=299476
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
usr.bin/whois/whois.c
View file

@ -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);