Adam Tkac 2de2c4b
diff -up bind-9.5.0b2/lib/isc/unix/socket.c.recv bind-9.5.0b2/lib/isc/unix/socket.c
Adam Tkac 2de2c4b
--- bind-9.5.0b2/lib/isc/unix/socket.c.recv	2008-04-10 16:45:33.000000000 +0200
Adam Tkac 2de2c4b
+++ bind-9.5.0b2/lib/isc/unix/socket.c	2008-04-10 17:00:46.000000000 +0200
Adam Tkac 2de2c4b
@@ -261,10 +261,10 @@ static isc_result_t allocate_socket(isc_
Adam Tkac 2de2c4b
 static void destroy(isc_socket_t **);
Adam Tkac 2de2c4b
 static void internal_accept(isc_task_t *, isc_event_t *);
Adam Tkac 2de2c4b
 static void internal_connect(isc_task_t *, isc_event_t *);
Adam Tkac 2de2c4b
-static void internal_recv(isc_task_t *, isc_event_t *);
Adam Tkac 2de2c4b
+static void internal_recv(isc_event_t *);
Adam Tkac 2de2c4b
 static void internal_send(isc_task_t *, isc_event_t *);
Adam Tkac 2de2c4b
 static void internal_fdwatch_write(isc_task_t *, isc_event_t *);
Adam Tkac 2de2c4b
-static void internal_fdwatch_read(isc_task_t *, isc_event_t *);
Adam Tkac 2de2c4b
+static void internal_fdwatch_read(isc_event_t *);
Adam Tkac 2de2c4b
 static void process_cmsg(isc_socket_t *, struct msghdr *, isc_socketevent_t *);
Adam Tkac 2de2c4b
 static void build_msghdr_send(isc_socket_t *, isc_socketevent_t *,
Adam Tkac 2de2c4b
 			      struct msghdr *, struct iovec *, size_t *);
Adam Tkac 2de2c4b
@@ -1830,7 +1830,7 @@ isc_socket_detach(isc_socket_t **socketp
Adam Tkac 2de2c4b
  *
Adam Tkac 2de2c4b
  * The socket and manager must be locked before calling this function.
Adam Tkac 2de2c4b
  */
Adam Tkac 2de2c4b
-static void
Adam Tkac 2de2c4b
+static isc_boolean_t
Adam Tkac 2de2c4b
 dispatch_recv(isc_socket_t *sock) {
Adam Tkac 2de2c4b
 	intev_t *iev;
Adam Tkac 2de2c4b
 	isc_socketevent_t *ev;
Adam Tkac 2de2c4b
@@ -1841,7 +1841,7 @@ dispatch_recv(isc_socket_t *sock) {
Adam Tkac 2de2c4b
 	if (sock->type != isc_sockettype_fdwatch) {
Adam Tkac 2de2c4b
 		ev = ISC_LIST_HEAD(sock->recv_list);
Adam Tkac 2de2c4b
 		if (ev == NULL)
Adam Tkac 2de2c4b
-			return;
Adam Tkac 2de2c4b
+			return ISC_FALSE;
Adam Tkac 2de2c4b
 		socket_log(sock, NULL, EVENT, NULL, 0, 0,
Adam Tkac 2de2c4b
 			   "dispatch_recv:  event %p -> task %p",
Adam Tkac 2de2c4b
 			   ev, ev->ev_sender);
Adam Tkac 2de2c4b
@@ -1855,13 +1855,16 @@ dispatch_recv(isc_socket_t *sock) {
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 	sock->references++;
Adam Tkac 2de2c4b
 	iev->ev_sender = sock;
Adam Tkac 2de2c4b
+	iev->ev_arg = sock;
Adam Tkac 2de2c4b
 	if (sock->type == isc_sockettype_fdwatch)
Adam Tkac 2de2c4b
-		iev->ev_action = internal_fdwatch_read;
Adam Tkac 2de2c4b
+		internal_fdwatch_read (iev);
Adam Tkac 2de2c4b
 	else
Adam Tkac 2de2c4b
-		iev->ev_action = internal_recv;
Adam Tkac 2de2c4b
-	iev->ev_arg = sock;
Adam Tkac 2de2c4b
+		internal_recv (iev);
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
-	isc_task_send(sender, (isc_event_t **)&iev;;
Adam Tkac 2de2c4b
+	if (sock->references == 0)
Adam Tkac 2de2c4b
+		return ISC_TRUE;
Adam Tkac 2de2c4b
+
Adam Tkac 2de2c4b
+	return ISC_FALSE;
Adam Tkac 2de2c4b
 }
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 static void
Adam Tkac 2de2c4b
@@ -2228,7 +2231,7 @@ internal_accept(isc_task_t *me, isc_even
Adam Tkac 2de2c4b
 }
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 static void
Adam Tkac 2de2c4b
-internal_recv(isc_task_t *me, isc_event_t *ev) {
Adam Tkac 2de2c4b
+internal_recv(isc_event_t *ev) {
Adam Tkac 2de2c4b
 	isc_socketevent_t *dev;
Adam Tkac 2de2c4b
 	isc_socket_t *sock;
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
@@ -2237,21 +2240,13 @@ internal_recv(isc_task_t *me, isc_event_
Adam Tkac 2de2c4b
 	sock = ev->ev_sender;
Adam Tkac 2de2c4b
 	INSIST(VALID_SOCKET(sock));
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
-	LOCK(&sock->lock);
Adam Tkac 2de2c4b
-	socket_log(sock, NULL, IOEVENT,
Adam Tkac 2de2c4b
-		   isc_msgcat, ISC_MSGSET_SOCKET, ISC_MSG_INTERNALRECV,
Adam Tkac 2de2c4b
-		   "internal_recv: task %p got event %p", me, ev);
Adam Tkac 2de2c4b
-
Adam Tkac 2de2c4b
 	INSIST(sock->pending_recv == 1);
Adam Tkac 2de2c4b
 	sock->pending_recv = 0;
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 	INSIST(sock->references > 0);
Adam Tkac 2de2c4b
 	sock->references--;  /* the internal event is done with this socket */
Adam Tkac 2de2c4b
-	if (sock->references == 0) {
Adam Tkac 2de2c4b
-		UNLOCK(&sock->lock);
Adam Tkac 2de2c4b
-		destroy(&sock);
Adam Tkac 2de2c4b
+	if (sock->references == 0)
Adam Tkac 2de2c4b
 		return;
Adam Tkac 2de2c4b
-	}
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 	/*
Adam Tkac 2de2c4b
 	 * Try to do as much I/O as possible on this socket.  There are no
Adam Tkac 2de2c4b
@@ -2289,7 +2284,6 @@ internal_recv(isc_task_t *me, isc_event_
Adam Tkac 2de2c4b
 	if (!ISC_LIST_EMPTY(sock->recv_list))
Adam Tkac 2de2c4b
 		select_poke(sock->manager, sock->fd, SELECT_POKE_READ);
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
-	UNLOCK(&sock->lock);
Adam Tkac 2de2c4b
 }
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 static void
Adam Tkac 2de2c4b
@@ -2388,7 +2382,7 @@ internal_fdwatch_write(isc_task_t *me, i
Adam Tkac 2de2c4b
 }
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 static void
Adam Tkac 2de2c4b
-internal_fdwatch_read(isc_task_t *me, isc_event_t *ev) {
Adam Tkac 2de2c4b
+internal_fdwatch_read(isc_event_t *ev) {
Adam Tkac 2de2c4b
 	isc_socket_t *sock;
Adam Tkac 2de2c4b
 	int more_data;
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
@@ -2400,31 +2394,19 @@ internal_fdwatch_read(isc_task_t *me, is
Adam Tkac 2de2c4b
 	sock = (isc_socket_t *)ev->ev_sender;
Adam Tkac 2de2c4b
 	INSIST(VALID_SOCKET(sock));
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
-	LOCK(&sock->lock);
Adam Tkac 2de2c4b
-	socket_log(sock, NULL, IOEVENT,
Adam Tkac 2de2c4b
-		   isc_msgcat, ISC_MSGSET_SOCKET, ISC_MSG_INTERNALRECV,
Adam Tkac 2de2c4b
-		   "internal_fdwatch_read: task %p got event %p", me, ev);
Adam Tkac 2de2c4b
-
Adam Tkac 2de2c4b
 	INSIST(sock->pending_recv == 1);
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
-	UNLOCK(&sock->lock);
Adam Tkac 2de2c4b
-	more_data = (sock->fdwatchcb)(me, sock, sock->fdwatcharg);
Adam Tkac 2de2c4b
-	LOCK(&sock->lock);
Adam Tkac 2de2c4b
+	INSIST(0); /* We should not be here */
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 	sock->pending_recv = 0;
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 	INSIST(sock->references > 0);
Adam Tkac 2de2c4b
 	sock->references--;  /* the internal event is done with this socket */
Adam Tkac 2de2c4b
-	if (sock->references == 0) {
Adam Tkac 2de2c4b
-		UNLOCK(&sock->lock);
Adam Tkac 2de2c4b
-		destroy(&sock);
Adam Tkac 2de2c4b
+	if (sock->references == 0)
Adam Tkac 2de2c4b
 		return;
Adam Tkac 2de2c4b
-	}
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 	if (more_data)
Adam Tkac 2de2c4b
 		select_poke(sock->manager, sock->fd, SELECT_POKE_READ);
Adam Tkac 2de2c4b
-
Adam Tkac 2de2c4b
-	UNLOCK(&sock->lock);
Adam Tkac 2de2c4b
 }
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 static void
Adam Tkac 2de2c4b
@@ -2434,6 +2416,7 @@ process_fds(isc_socketmgr_t *manager, in
Adam Tkac 2de2c4b
 	int i;
Adam Tkac 2de2c4b
 	isc_socket_t *sock;
Adam Tkac 2de2c4b
 	isc_boolean_t unlock_sock;
Adam Tkac 2de2c4b
+	isc_boolean_t destroy_sock;
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 	REQUIRE(maxfd <= (int)FD_SETSIZE);
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
@@ -2462,6 +2445,7 @@ process_fds(isc_socketmgr_t *manager, in
Adam Tkac 2de2c4b
 
Adam Tkac 2de2c4b
 		sock = manager->fds[i];
Adam Tkac 2de2c4b
 		unlock_sock = ISC_FALSE;
Adam Tkac 2de2c4b
+		destroy_sock = ISC_FALSE;
Adam Tkac 2de2c4b
 		if (FD_ISSET(i, readfds)) {
Adam Tkac 2de2c4b
 			if (sock == NULL) {
Adam Tkac 2de2c4b
 				FD_CLR(i, &manager->read_fds);
Adam Tkac 2de2c4b
@@ -2473,7 +2457,7 @@ process_fds(isc_socketmgr_t *manager, in
Adam Tkac 2de2c4b
 				if (sock->listener)
Adam Tkac 2de2c4b
 					dispatch_accept(sock);
Adam Tkac 2de2c4b
 				else
Adam Tkac 2de2c4b
-					dispatch_recv(sock);
Adam Tkac 2de2c4b
+					destroy_sock = dispatch_recv(sock);
Adam Tkac 2de2c4b
 			}
Adam Tkac 2de2c4b
 			FD_CLR(i, &manager->read_fds);
Adam Tkac 2de2c4b
 		}
Adam Tkac 2de2c4b
@@ -2497,6 +2481,8 @@ process_fds(isc_socketmgr_t *manager, in
Adam Tkac 2de2c4b
 		}
Adam Tkac 2de2c4b
 		if (unlock_sock)
Adam Tkac 2de2c4b
 			UNLOCK(&sock->lock);
Adam Tkac 2de2c4b
+		if (destroy_sock)
Adam Tkac 2de2c4b
+			destroy(&sock);
Adam Tkac 2de2c4b
 	}
Adam Tkac 2de2c4b
 }
Adam Tkac 2de2c4b