utf8: don't allow NUL in decoded strings.
[ccan] / ccan / io / test / run-03-readpartial.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 PORT "65003"
10
11 struct data {
12         int state;
13         size_t bytes;
14         char buf[4];
15 };
16
17 static void finish_ok(struct io_conn *conn, struct data *d)
18 {
19         ok1(d->state == 1);
20         d->state++;
21         io_break(d);
22 }
23
24 static struct io_plan *init_conn(struct io_conn *conn, struct data *d)
25 {
26         ok1(d->state == 0);
27         d->state++;
28
29         io_set_finish(conn, finish_ok, d);
30
31         return io_read_partial(conn, d->buf, sizeof(d->buf), &d->bytes,
32                                io_close_cb, 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 write_to_socket(const char *str, const struct addrinfo *addrinfo)
68 {
69         int fd, i;
70
71         fd = socket(addrinfo->ai_family, addrinfo->ai_socktype,
72                     addrinfo->ai_protocol);
73         if (fd < 0)
74                 exit(1);
75         if (connect(fd, addrinfo->ai_addr, addrinfo->ai_addrlen) != 0)
76                 exit(2);
77         signal(SIGPIPE, SIG_IGN);
78         for (i = 0; i < strlen(str); i++) {
79                 if (write(fd, str + i, 1) != 1)
80                         break;
81         }
82         close(fd);
83 }
84
85 int main(void)
86 {
87         struct data *d = malloc(sizeof(*d));
88         struct addrinfo *addrinfo;
89         struct io_listener *l;
90         int fd, status;
91
92         /* This is how many tests you plan to run */
93         plan_tests(22);
94         d->state = 0;
95         fd = make_listen_fd(PORT, &addrinfo);
96         ok1(fd >= 0);
97         l = io_new_listener(NULL, fd, init_conn, d);
98         ok1(l);
99         fflush(stdout);
100         if (!fork()) {
101                 io_close_listener(l);
102                 write_to_socket("hellothere", addrinfo);
103                 freeaddrinfo(addrinfo);
104                 free(d);
105                 exit(0);
106         }
107         ok1(io_loop(NULL, NULL) == d);
108         ok1(d->state == 2);
109         ok1(d->bytes > 0);
110         ok1(d->bytes <= sizeof(d->buf));
111         ok1(memcmp(d->buf, "hellothere", d->bytes) == 0);
112
113         ok1(wait(&status));
114         ok1(WIFEXITED(status));
115         ok1(WEXITSTATUS(status) == 0);
116
117         fflush(stdout);
118         if (!fork()) {
119                 io_close_listener(l);
120                 write_to_socket("hi", addrinfo);
121                 freeaddrinfo(addrinfo);
122                 free(d);
123                 exit(0);
124         }
125         d->state = 0;
126         ok1(io_loop(NULL, NULL) == d);
127         ok1(d->state == 2);
128         ok1(d->bytes > 0);
129         ok1(d->bytes <= strlen("hi"));
130         ok1(memcmp(d->buf, "hi", d->bytes) == 0);
131
132         freeaddrinfo(addrinfo);
133         free(d);
134         io_close_listener(l);
135
136         ok1(wait(&status));
137         ok1(WIFEXITED(status));
138         ok1(WEXITSTATUS(status) == 0);
139
140         /* This exits depending on whether all tests passed */
141         return exit_status();
142 }