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