ccan/io: have io_plan mappable back to io_conn.

[ccan] / ccan / io / io.c

diff --git a/ccan/io/io.c b/ccan/io/io.c
index c0dd9b838e84af6ec8ef6b62dec8fe3695adf94c..c5c9bee0daf0329f433eec930058bfd20905f96c 100644 (file)
--- a/ccan/io/io.c
+++ b/ccan/io/io.c
@@ -61,10 +61,7 @@ static bool next_plan(struct io_conn *conn, struct io_plan *plan)
        if (plan == &io_conn_freed)
                return false;
 
-       /* It should have set a plan inside this conn (or duplex) */
-       assert(plan == &conn->plan[IO_IN]
-              || plan == &conn->plan[IO_OUT]
-              || plan == &conn->plan[2]);
+       assert(plan == &conn->plan[plan->dir]);
        assert(conn->plan[IO_IN].status != IO_UNSET
               || conn->plan[IO_OUT].status != IO_UNSET);
 
@@ -108,6 +105,10 @@ struct io_conn *io_new_conn_(const tal_t *ctx, int fd,
        /* Keep our I/O async. */
        io_fd_block(fd, false);
 
+       /* So we can get back from plan -> conn later */
+       conn->plan[IO_OUT].dir = IO_OUT;
+       conn->plan[IO_IN].dir = IO_IN;
+
        /* We start with out doing nothing, and in doing our init. */
        conn->plan[IO_OUT].status = IO_UNSET;
 
@@ -202,8 +203,12 @@ struct io_plan *io_write_(struct io_conn *conn, const void *data, size_t len,
 static int do_read(int fd, struct io_plan_arg *arg)
 {
        ssize_t ret = read(fd, arg->u1.cp, arg->u2.s);
-       if (ret <= 0)
+       if (ret <= 0) {
+               /* Errno isn't set if we hit EOF, so set it to distinct value */
+               if (ret == 0)
+                       errno = 0;
                return -1;
+       }
 
        arg->u1.cp += ret;
        arg->u2.s -= ret;
@@ -230,8 +235,12 @@ struct io_plan *io_read_(struct io_conn *conn,
 static int do_read_partial(int fd, struct io_plan_arg *arg)
 {
        ssize_t ret = read(fd, arg->u1.cp, *(size_t *)arg->u2.vp);
-       if (ret <= 0)
+       if (ret <= 0) {
+               /* Errno isn't set if we hit EOF, so set it to distinct value */
+               if (ret == 0)
+                       errno = 0;
                return -1;
+       }
 
        *(size_t *)arg->u2.vp = ret;
        return 1;
@@ -480,7 +489,7 @@ struct io_plan *io_duplex(struct io_conn *conn,
        assert(conn == container_of(in_plan, struct io_conn, plan[IO_IN]));
        /* in_plan must be conn->plan[IO_IN], out_plan must be [IO_OUT] */
        assert(out_plan == in_plan + 1);
-       return out_plan + 1;
+       return in_plan;
 }
 
 struct io_plan *io_halfclose(struct io_conn *conn)