#include <sys/wait.h>
#include <stdio.h>
+#ifndef PORT
+#define PORT "65007"
+#endif
+
struct data {
int state;
char buf[4];
};
-static struct io_plan plan_read(struct io_conn *conn, struct data *d)
+static struct io_plan read_done(struct io_conn *conn, struct data *d)
{
ok1(d->state == 1);
d->state++;
- return io_read(d->buf, sizeof(d->buf), io_close, d);
+ return io_close();
}
-static struct io_plan start_break(struct io_conn *conn, struct data *d)
+static void finish_ok(struct io_conn *conn, struct data *d)
{
- ok1(d->state == 0);
+ ok1(d->state == 2);
d->state++;
- return io_break(d, plan_read, d);
}
-static void finish_ok(struct io_conn *conn, struct data *d)
+static void init_conn(int fd, struct data *d)
{
- ok1(d->state == 2);
+ ok1(d->state == 0);
d->state++;
+
+ io_set_finish(io_new_conn(fd,
+ io_break(d,
+ io_read(d->buf, sizeof(d->buf), read_done, d))),
+ finish_ok, d);
}
static int make_listen_fd(const char *port, struct addrinfo **info)
/* This is how many tests you plan to run */
plan_tests(13);
d->state = 0;
- fd = make_listen_fd("65007", &addrinfo);
+ fd = make_listen_fd(PORT, &addrinfo);
ok1(fd >= 0);
- l = io_new_listener(fd, start_break, finish_ok, d);
+ l = io_new_listener(fd, init_conn, d);
ok1(l);
fflush(stdout);
if (!fork()) {