io: change io_idle() to io_wait()
[ccan] / ccan / io / test / run-05-write.c
1 #include <ccan/io/io.h>
2 /* Include the C files directly. */
3 #include <ccan/io/poll.c>
4 #include <ccan/io/io.c>
5 #include <ccan/tap/tap.h>
6 #include <sys/wait.h>
7 #include <stdio.h>
8
9 #ifndef PORT
10 #define PORT "65005"
11 #endif
12
13 struct data {
14         int state;
15         size_t bytes;
16         char *buf;
17 };
18
19 static void finish_ok(struct io_conn *conn, struct data *d)
20 {
21         ok1(d->state == 1);
22         d->state++;
23         io_break(d, io_never());
24 }
25
26 static void init_conn(int fd, struct data *d)
27 {
28         ok1(d->state == 0);
29         d->state++;
30         io_set_finish(io_new_conn(fd, io_write(d->buf, d->bytes,
31                                                io_close_cb, d)),
32                       finish_ok, d);
33 }
34
35 static int make_listen_fd(const char *port, struct addrinfo **info)
36 {
37         int fd, on = 1;
38         struct addrinfo *addrinfo, hints;
39
40         memset(&hints, 0, sizeof(hints));
41         hints.ai_family = AF_UNSPEC;
42         hints.ai_socktype = SOCK_STREAM;
43         hints.ai_flags = AI_PASSIVE;
44         hints.ai_protocol = 0;
45
46         if (getaddrinfo(NULL, port, &hints, &addrinfo) != 0)
47                 return -1;
48
49         fd = socket(addrinfo->ai_family, addrinfo->ai_socktype,
50                     addrinfo->ai_protocol);
51         if (fd < 0)
52                 return -1;
53
54         setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
55         if (bind(fd, addrinfo->ai_addr, addrinfo->ai_addrlen) != 0) {
56                 close(fd);
57                 return -1;
58         }
59         if (listen(fd, 1) != 0) {
60                 close(fd);
61                 return -1;
62         }
63         *info = addrinfo;
64         return fd;
65 }
66
67 static void read_from_socket(size_t bytes, const struct addrinfo *addrinfo)
68 {
69         int fd, done, r;
70         char buf[100];
71
72         fd = socket(addrinfo->ai_family, addrinfo->ai_socktype,
73                     addrinfo->ai_protocol);
74         if (fd < 0)
75                 exit(1);
76         if (connect(fd, addrinfo->ai_addr, addrinfo->ai_addrlen) != 0)
77                 exit(2);
78
79         for (done = 0; done < bytes; done += r) {
80                 r = read(fd, buf, sizeof(buf));
81                 if (r < 0)
82                         exit(3);
83                 done += r;
84         }
85         close(fd);
86 }
87
88 int main(void)
89 {
90         struct data *d = malloc(sizeof(*d));
91         struct addrinfo *addrinfo;
92         struct io_listener *l;
93         int fd, status;
94
95         /* This is how many tests you plan to run */
96         plan_tests(9);
97         d->state = 0;
98         d->bytes = 1024*1024;
99         d->buf = malloc(d->bytes);
100         memset(d->buf, 'a', d->bytes);
101         fd = make_listen_fd(PORT, &addrinfo);
102         ok1(fd >= 0);
103         l = io_new_listener(fd, init_conn, d);
104         ok1(l);
105         fflush(stdout);
106         if (!fork()) {
107                 io_close_listener(l);
108                 read_from_socket(d->bytes, addrinfo);
109                 freeaddrinfo(addrinfo);
110                 free(d->buf);
111                 free(d);
112                 exit(0);
113         }
114         ok1(io_loop() == d);
115         ok1(d->state == 2);
116
117         ok1(wait(&status));
118         ok1(WIFEXITED(status));
119         ok1(WEXITSTATUS(status) == 0);
120
121         freeaddrinfo(addrinfo);
122         free(d->buf);
123         free(d);
124         io_close_listener(l);
125
126         /* This exits depending on whether all tests passed */
127         return exit_status();
128 }