1 /* Test of poll() function.
2 Copyright (C) 2008-2013 Free Software Foundation, Inc.
4 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 3, or (at your option)
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, see <http://www.gnu.org/licenses/>. */
17 /* Written by Paolo Bonzini. */
24 #include "signature.h"
25 SIGNATURE_CHECK (poll, int, (struct pollfd[], nfds_t, int));
29 #include <sys/socket.h>
30 #include <netinet/in.h>
31 #include <arpa/inet.h>
35 #include <sys/ioctl.h>
40 #if (defined _WIN32 || defined __WIN32__) && ! defined __CYGWIN__
41 # define WINDOWS_NATIVE
46 #define pipe(x) _pipe(x, 256, O_BINARY)
51 #ifdef HAVE_SYS_WAIT_H
56 #define SO_REUSEPORT SO_REUSEADDR
59 #define TEST_PORT 12345
62 /* Minimal testing infrastructure. */
67 failed (const char *reason)
71 printf ("failed (%s)\n", reason);
75 test (void (*fn) (void), const char *msg)
78 printf ("%s... ", msg);
89 /* Funny socket code. */
95 struct sockaddr_in ia;
97 s = socket (AF_INET, SOCK_STREAM, 0);
100 setsockopt (s, SOL_SOCKET, SO_REUSEPORT, &x, sizeof (x));
102 memset (&ia, 0, sizeof (ia));
103 ia.sin_family = AF_INET;
104 inet_pton (AF_INET, "127.0.0.1", &ia.sin_addr);
105 ia.sin_port = htons (TEST_PORT);
106 if (bind (s, (struct sockaddr *) &ia, sizeof (ia)) < 0)
112 if (listen (s, 1) < 0)
122 connect_to_socket (int blocking)
125 struct sockaddr_in ia;
127 s = socket (AF_INET, SOCK_STREAM, 0);
129 memset (&ia, 0, sizeof (ia));
130 ia.sin_family = AF_INET;
131 inet_pton (AF_INET, "127.0.0.1", &ia.sin_addr);
132 ia.sin_port = htons (TEST_PORT);
136 #ifdef WINDOWS_NATIVE
137 unsigned long iMode = 1;
138 ioctl (s, FIONBIO, (char *) &iMode);
140 #elif defined F_GETFL
141 int oldflags = fcntl (s, F_GETFL, NULL);
143 if (!(oldflags & O_NONBLOCK))
144 fcntl (s, F_SETFL, oldflags | O_NONBLOCK);
148 if (connect (s, (struct sockaddr *) &ia, sizeof (ia)) < 0
149 && (blocking || errno != EINPROGRESS))
159 /* A slightly more convenient interface to poll(2). */
162 poll1 (int fd, int ev, int time)
170 r = poll (&pfd, 1, time);
174 if (pfd.revents & ~(POLLHUP | POLLERR | POLLNVAL | ev))
175 failed ("invalid flag combination (unrequested events)");
181 poll1_nowait (int fd, int ev)
183 return poll1 (fd, ev, 0);
187 poll1_wait (int fd, int ev)
189 return poll1 (fd, ev, -1);
193 /* Test poll(2) for TTYs. */
199 if (poll1_nowait (0, POLLIN | POLLRDNORM) != 0)
201 if (poll1_nowait (0, POLLOUT) == 0)
202 failed ("cannot write");
204 if (poll1_wait (0, POLLIN | POLLRDNORM) == 0)
205 failed ("return with infinite timeout");
208 if (poll1_nowait (0, POLLIN | POLLRDNORM) != 0)
209 failed ("can read after getc");
214 /* Test poll(2) for unconnected nonblocking sockets. */
217 test_connect_first (void)
219 int s = open_server_socket ();
220 struct sockaddr_in ia;
225 if (poll1_nowait (s, POLLIN | POLLRDNORM | POLLRDBAND) != 0)
226 failed ("can read, socket not connected");
228 c1 = connect_to_socket (false);
230 if (poll1_wait (s, POLLIN | POLLRDNORM | POLLRDBAND) != (POLLIN | POLLRDNORM))
231 failed ("expecting POLLIN | POLLRDNORM on passive socket");
232 if (poll1_nowait (s, POLLIN | POLLRDBAND) != POLLIN)
233 failed ("expecting POLLIN on passive socket");
234 if (poll1_nowait (s, POLLRDNORM | POLLRDBAND) != POLLRDNORM)
235 failed ("expecting POLLRDNORM on passive socket");
237 addrlen = sizeof (ia);
238 c2 = accept (s, (struct sockaddr *) &ia, &addrlen);
245 /* Test poll(2) for unconnected blocking sockets. */
248 test_accept_first (void)
250 #ifndef WINDOWS_NATIVE
251 int s = open_server_socket ();
252 struct sockaddr_in ia;
263 addrlen = sizeof (ia);
264 c = accept (s, (struct sockaddr *) &ia, &addrlen);
267 ASSERT (write (c, "foo", 3) == 3);
268 ASSERT (read (c, buf, 3) == 3);
269 shutdown (c, SHUT_RD);
276 c = connect_to_socket (true);
278 if (poll1_nowait (c, POLLOUT | POLLWRNORM | POLLRDBAND)
279 != (POLLOUT | POLLWRNORM))
280 failed ("cannot write after blocking connect");
281 ASSERT (write (c, "foo", 3) == 3);
283 if (poll1_wait (c, POLLIN) != POLLIN)
284 failed ("cannot read data left in the socket by closed process");
285 ASSERT (read (c, buf, 3) == 3);
286 ASSERT (write (c, "foo", 3) == 3);
287 if ((poll1_wait (c, POLLIN | POLLOUT) & (POLLHUP | POLLERR)) == 0)
288 failed ("expecting POLLHUP after shutdown");
295 /* Common code for pipes and connected sockets. */
298 test_pair (int rd, int wd)
301 if (poll1_wait (wd, POLLIN | POLLRDNORM | POLLOUT | POLLWRNORM | POLLRDBAND)
302 != (POLLOUT | POLLWRNORM))
303 failed ("expecting POLLOUT | POLLWRNORM before writing");
304 if (poll1_nowait (wd, POLLIN | POLLRDNORM | POLLOUT | POLLRDBAND) != POLLOUT)
305 failed ("expecting POLLOUT before writing");
306 if (poll1_nowait (wd, POLLIN | POLLRDNORM | POLLWRNORM | POLLRDBAND)
308 failed ("expecting POLLWRNORM before writing");
310 ASSERT (write (wd, "foo", 3) == 3);
311 if (poll1_wait (rd, POLLIN | POLLRDNORM) != (POLLIN | POLLRDNORM))
312 failed ("expecting POLLIN | POLLRDNORM after writing");
313 if (poll1_nowait (rd, POLLIN) != POLLIN)
314 failed ("expecting POLLIN after writing");
315 if (poll1_nowait (rd, POLLRDNORM) != POLLRDNORM)
316 failed ("expecting POLLRDNORM after writing");
318 ASSERT (read (rd, buf, 3) == 3);
322 /* Test poll(2) on connected sockets. */
325 test_socket_pair (void)
327 struct sockaddr_in ia;
329 socklen_t addrlen = sizeof (ia);
330 int s = open_server_socket ();
331 int c1 = connect_to_socket (false);
332 int c2 = accept (s, (struct sockaddr *) &ia, &addrlen);
341 ASSERT (write (c2, "foo", 3) == 3);
342 if ((poll1_nowait (c2, POLLIN | POLLOUT) & (POLLHUP | POLLERR)) == 0)
343 failed ("expecting POLLHUP after shutdown");
349 /* Test poll(2) on pipes. */
356 ASSERT (pipe (fd) >= 0);
357 test_pair (fd[0], fd[1]);
359 if ((poll1_wait (fd[1], POLLIN | POLLOUT) & (POLLHUP | POLLERR)) == 0)
360 failed ("expecting POLLHUP after shutdown");
374 printf ("Please press Enter\n");
375 test (test_tty, "TTY");
378 result = test (test_connect_first, "Unconnected socket test");
379 result += test (test_socket_pair, "Connected sockets test");
380 result += test (test_accept_first, "General socket test with fork");
381 result += test (test_pipe, "Pipe test");