ccan/io: rename io_op to io_plan.
[ccan] / ccan / io / test / run-10-many.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 #define NUM 100
10 #define NUM_ITERS 1000
11
12 struct buffer {
13         int iters;
14         struct io_conn *reader, *writer;
15         char buf[32];
16 };
17
18 static struct io_plan *poke_writer(struct io_conn *conn, struct buffer *buf);
19 static struct io_plan *poke_reader(struct io_conn *conn, struct buffer *buf);
20
21 static struct io_plan *do_read(struct io_conn *conn, struct buffer *buf)
22 {
23         assert(conn == buf->reader);
24
25         return io_read(&buf->buf, sizeof(buf->buf),
26                        io_next(conn, poke_writer, buf));
27 }
28
29 static struct io_plan *do_write(struct io_conn *conn, struct buffer *buf)
30 {
31         assert(conn == buf->writer);
32
33         return io_write(&buf->buf, sizeof(buf->buf),
34                         io_next(conn, poke_reader, buf));
35 }
36
37 static struct io_plan *poke_writer(struct io_conn *conn, struct buffer *buf)
38 {
39         assert(conn == buf->reader);
40
41         if (buf->iters == NUM_ITERS)
42                 return io_close(conn, NULL);
43
44         /* You write. */
45         io_wake(buf->writer, do_write, buf);
46
47         /* I'll wait until you wake me. */
48         return io_idle(conn);
49 }
50
51 static struct io_plan *poke_reader(struct io_conn *conn, struct buffer *buf)
52 {
53         assert(conn == buf->writer);
54         /* You read. */
55         io_wake(buf->reader, do_read, buf);
56
57         if (++buf->iters == NUM_ITERS)
58                 return io_close(conn, NULL);
59
60         /* I'll wait until you tell me to write. */
61         return io_idle(conn);
62 }
63
64 static struct io_plan *reader(struct io_conn *conn, struct buffer *buf)
65 {
66         assert(conn == buf->reader);
67
68         /* Wait for writer to tell us to read. */
69         return io_idle(conn);
70 }
71
72 static struct buffer buf[NUM];
73
74 int main(void)
75 {
76         unsigned int i;
77         int fds[2], last_read, last_write;
78
79         plan_tests(5 + NUM);
80
81         ok1(pipe(fds) == 0);
82         last_read = fds[0];
83         last_write = fds[1];
84
85         for (i = 1; i < NUM; i++) {
86                 if (pipe(fds) < 0)
87                         break;
88                 memset(buf[i].buf, i, sizeof(buf[i].buf));
89                 sprintf(buf[i].buf, "%i-%i", i, i);
90
91                 buf[i].reader = io_new_conn(last_read, reader, NULL, &buf[i]);
92                 if (!buf[i].reader)
93                         break;
94                 buf[i].writer = io_new_conn(fds[1], do_write, NULL, &buf[i]);
95                 if (!buf[i].writer)
96                         break;
97                 last_read = fds[0];
98         }
99         if (!ok1(i == NUM))
100                 exit(exit_status());
101
102         /* Last one completes the cirle. */
103         i = 0;
104         sprintf(buf[i].buf, "%i-%i", i, i);
105         buf[i].reader = io_new_conn(last_read, reader, NULL, &buf[i]);
106         ok1(buf[i].reader);
107         buf[i].writer = io_new_conn(last_write, do_write, NULL, &buf[i]);
108         ok1(buf[i].writer);
109
110         /* They should eventually exit */
111         ok1(io_loop() == NULL);
112
113         for (i = 0; i < NUM; i++) {
114                 char b[sizeof(buf[0].buf)];
115                 memset(b, i, sizeof(b));
116                 sprintf(b, "%i-%i", i, i);
117                 ok1(memcmp(b, buf[(i + NUM_ITERS) % NUM].buf, sizeof(b)) == 0);
118         }
119
120         /* This exits depending on whether all tests passed */
121         return exit_status();
122 }