#include <sys/wait.h>
#include <stdio.h>
+#ifndef PORT
+#define PORT "65002"
+#endif
+
struct data {
int state;
char buf[4];
};
-static struct io_plan *start_ok(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 == 1);
d->state++;
- return io_read(d->buf, sizeof(d->buf), io_next(conn, io_close, d));
+ io_break(d, io_idle());
}
-static void finish_ok(struct io_conn *conn, struct data *d)
+static void init_conn(int fd, struct data *d)
{
- ok1(d->state == 1);
+ ok1(d->state == 0);
d->state++;
- io_break(d, NULL);
+
+ io_set_finish(io_new_conn(fd,
+ io_read(d->buf, sizeof(d->buf), io_close_cb, 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(10);
d->state = 0;
- fd = make_listen_fd("65002", &addrinfo);
+ fd = make_listen_fd(PORT, &addrinfo);
ok1(fd >= 0);
- l = io_new_listener(fd, start_ok, finish_ok, d);
+ l = io_new_listener(fd, init_conn, d);
ok1(l);
fflush(stdout);
if (!fork()) {