X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=ftpd.c;h=757c109e677d88091e8b513e9f16124f27d1e695;hb=85c74b6283f53e8deeab46354a4d3971039b6d56;hp=4d65210591c26c260d2c864dd5913c14c7dfdf0e;hpb=4ad0559c71f4ef2f498a8b236a53eda9bcdd5c2c;p=betaftpd

diff --git a/ftpd.c b/ftpd.c
index 4d65210..757c109 100644
--- a/ftpd.c
+++ b/ftpd.c
@@ -579,11 +579,7 @@ int process_all_clients(const fd_set * const active_clients, const int num_ac)
 		c = next;
 		next = c->next_conn;
 #if HAVE_POLL
-		if (fds[c->sock].revents & (POLLERR|POLLHUP|POLLNVAL)) {
-			destroy_conn(c);
-			continue;
-		}
-		if (!fds[c->sock].revents & POLLIN) {
+		if (!fds[c->sock].revents & (POLLIN|POLLERR|POLLHUP|POLLNVAL)) {
 			continue;
 		}
 #else
@@ -603,8 +599,10 @@ int process_all_clients(const fd_set * const active_clients, const int num_ac)
 			 * client has closed the socket. If we get a return value
 			 * of -1 (error), we close the socket ourselves.
 			 *
-			 * Just to be safe, we include this code for poll() as
-			 * well.
+			 * We do the same for poll(), even though we actually have
+			 * bits that tell us what is happening (in case of new 
+			 * input AND error/hangup at the same time, we do an
+			 * explicit check at the bottom of the loop as well).
 			 */
 			destroy_conn(c);
 			continue;
@@ -619,6 +617,11 @@ int process_all_clients(const fd_set * const active_clients, const int num_ac)
 
 		c->buf_len += bytes_avail;
 		parse_command(c);
+
+		if (fds[c->sock].revents & (POLLERR|POLLHUP|POLLNVAL)) {
+                        destroy_conn(c);
+                        continue;
+                }
 	}
 	return checked_through;
 }