X-Git-Url: http://git.ozlabs.org/?a=blobdiff_plain;f=ccan%2Fio%2Fpoll.c;h=1b11badd642c0d87d1270d642ac63db40f9cb24e;hb=dbbefec6973725d9a16c82747e0c7ce8e5e6c4f2;hp=3ed6321bc7f4a0309ca2c2a55f41ab17ae328511;hpb=7bfb7a1cf4f7c5eaf4081ef7128b7fc7d1cf926e;p=ccan

diff --git a/ccan/io/poll.c b/ccan/io/poll.c
index 3ed6321b..1b11badd 100644
--- a/ccan/io/poll.c
+++ b/ccan/io/poll.c
@@ -28,7 +28,7 @@ static void io_loop_exit(void)
 		while (free_later) {
 			struct io_conn *c = free_later;
 			free_later = c->finish_arg;
-			free(c);
+			io_alloc.free(c);
 		}
 	}
 }
@@ -42,7 +42,7 @@ static void free_conn(struct io_conn *conn)
 		conn->finish_arg = free_later;
 		free_later = conn;
 	} else
-		free(conn);
+		io_alloc.free(conn);
 }
 #else
 static void io_loop_enter(void)
@@ -53,7 +53,7 @@ static void io_loop_exit(void)
 }
 static void free_conn(struct io_conn *conn)
 {
-	free(conn);
+	io_alloc.free(conn);
 }
 #endif
 
@@ -64,11 +64,11 @@ static bool add_fd(struct fd *fd, short events)
 		struct fd **newfds;
 		size_t num = max_fds ? max_fds * 2 : 8;
 
-		newpollfds = realloc(pollfds, sizeof(*newpollfds) * num);
+		newpollfds = io_alloc.realloc(pollfds, sizeof(*newpollfds)*num);
 		if (!newpollfds)
 			return false;
 		pollfds = newpollfds;
-		newfds = realloc(fds, sizeof(*newfds) * num);
+		newfds = io_alloc.realloc(fds, sizeof(*newfds) * num);
 		if (!newfds)
 			return false;
 		fds = newfds;
@@ -107,8 +107,8 @@ static void del_fd(struct fd *fd)
 		fds[n]->backend_info = n;
 	} else if (num_fds == 1) {
 		/* Free everything when no more fds. */
-		free(pollfds);
-		free(fds);
+		io_alloc.free(pollfds);
+		io_alloc.free(fds);
 		pollfds = NULL;
 		fds = NULL;
 		max_fds = 0;
@@ -172,15 +172,16 @@ bool add_duplex(struct io_conn *c)
 	return true;
 }
 
-static void del_conn(struct io_conn *conn)
+void backend_del_conn(struct io_conn *conn)
 {
 	if (conn->finish) {
-		errno = conn->plan.u.close.saved_errno;
+		/* Saved by io_close */
+		errno = conn->plan.u1.s;
 		conn->finish(conn, conn->finish_arg);
 	}
 	if (timeout_active(conn))
 		backend_del_timeout(conn);
-	free(conn->timeout);
+	io_alloc.free(conn->timeout);
 	if (conn->duplex) {
 		/* In case fds[] pointed to the other one. */
 		fds[conn->fd.backend_info] = &conn->duplex->fd;
@@ -189,6 +190,7 @@ static void del_conn(struct io_conn *conn)
 	} else
 		del_fd(&conn->fd);
 	num_closing--;
+	free_conn(conn);
 }
 
 void del_listener(struct io_listener *l)
@@ -213,7 +215,7 @@ static void accept_conn(struct io_listener *l)
 }
 
 /* It's OK to miss some, as long as we make progress. */
-static void finish_conns(void)
+static bool finish_conns(struct io_conn **ready)
 {
 	unsigned int i;
 
@@ -228,12 +230,16 @@ static void finish_conns(void)
 		c = (void *)fds[i];
 		for (duplex = c->duplex; c; c = duplex, duplex = NULL) {
 			if (!c->plan.next) {
-				del_conn(c);
-				free_conn(c);
+				if (doing_debug_on(c) && ready) {
+					*ready = c;
+					return true;
+				}
+				backend_del_conn(c);
 				i--;
 			}
 		}
 	}
+	return false;
 }
 
 void backend_add_timeout(struct io_conn *conn, struct timespec duration)
@@ -253,7 +259,7 @@ void backend_del_timeout(struct io_conn *conn)
 }
 
 /* This is the main loop. */
-void *io_loop(void)
+void *do_io_loop(struct io_conn **ready)
 {
 	void *ret;
 
@@ -291,7 +297,9 @@ void *io_loop(void)
 		}
 
 		if (num_closing) {
-			finish_conns();
+			/* If this finishes a debugging con, return now. */
+			if (finish_conns(ready))
+				return NULL;
 			/* Could have started/finished more. */
 			continue;
 		}
@@ -327,16 +335,29 @@ void *io_loop(void)
 				if (c->duplex) {
 					int mask = c->duplex->plan.pollflag;
 					if (events & mask) {
+						if (doing_debug_on(c->duplex)
+							&& ready) {
+							*ready = c->duplex;
+							return NULL;
+						}
 						io_ready(c->duplex);
 						events &= ~mask;
 						/* debug can recurse;
 						 * anything can change. */
 						if (doing_debug())
 							break;
-						if (!(events&(POLLIN|POLLOUT)))
+
+						/* If no events, or it closed
+						 * the duplex, continue. */
+						if (!(events&(POLLIN|POLLOUT))
+						    || !c->plan.next)
 							continue;
 					}
 				}
+				if (doing_debug_on(c) && ready) {
+					*ready = c;
+					return NULL;
+				}
 				io_ready(c);
 				/* debug can recurse; anything can change. */
 				if (doing_debug())
@@ -354,8 +375,10 @@ void *io_loop(void)
 		}
 	}
 
-	while (num_closing)
-		finish_conns();
+	while (num_closing && !io_loop_return) {
+		if (finish_conns(ready))
+			return NULL;
+	}
 
 	ret = io_loop_return;
 	io_loop_return = NULL;
@@ -363,3 +386,8 @@ void *io_loop(void)
 	io_loop_exit();
 	return ret;
 }
+
+void *io_loop(void)
+{
+	return do_io_loop(NULL);
+}