Move connection dropping earlier than adding to poll lists in case server is disabled.

Author: kvap

contrib/arbiter/src/server.c

@@ -418,8 +418,15 @@ static bool server_accept(server_t server) {
 		shout("failed to accept a connection: %s\n", strerror(errno));
 		return false;
 	}
-	debug("a new connection accepted\n");
+	debug("a new connection fd=%d accepted\n", fd);
 
+	if (!server->enabled) {
+		shout("server disabled, disconnecting the accepted connection fd=%d\n", fd);
+		// FIXME: redirect instead of disconnecting
+		close(fd);
+		return false;
+	}
+
 	s = server->free_chain;
 	if (s == NULL) { 
 		s = malloc(sizeof(stream_data_t));
@@ -430,13 +437,6 @@ static bool server_accept(server_t server) {
 	s->next = server->used_chain;
 	server->used_chain = s;
 
-	if (!server->enabled) {
-		shout("server disabled, disconnecting the accepted connection\n");
-		// FIXME: redirect instead of disconnecting
-		close(fd);
-		return false;
-	}
-
 	stream_init(s, fd);
 
 	return server_add_socket(server, fd, s);