1 /* Test of select() substitute.
2 Copyright (C) 2008-2012 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 of the License, or
7 (at your option) any later version.
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, 2008. */
21 #include <netinet/in.h>
22 #include <arpa/inet.h>
27 #include <sys/ioctl.h>
32 #if (defined _WIN32 || defined __WIN32__) && ! defined __CYGWIN__
33 # define WINDOWS_NATIVE
36 #ifdef HAVE_SYS_WAIT_H
37 # include <sys/wait.h>
41 # define SO_REUSEPORT SO_REUSEADDR
44 #define TEST_PORT 12345
47 typedef int (*select_fn) (int, fd_set *, fd_set *, fd_set *, struct timeval *);
50 /* Minimal testing infrastructure. */
55 failed (const char *reason)
59 printf ("failed (%s)\n", reason);
63 test (void (*fn) (select_fn), select_fn my_select, const char *msg)
66 printf ("%s... ", msg);
77 /* Funny socket code. */
80 open_server_socket (void)
83 struct sockaddr_in ia;
85 s = socket (AF_INET, SOCK_STREAM, 0);
87 memset (&ia, 0, sizeof (ia));
88 ia.sin_family = AF_INET;
89 inet_pton (AF_INET, "127.0.0.1", &ia.sin_addr);
90 ia.sin_port = htons (TEST_PORT);
91 if (bind (s, (struct sockaddr *) &ia, sizeof (ia)) < 0)
98 setsockopt (s, SOL_SOCKET, SO_REUSEPORT, &x, sizeof (x));
100 if (listen (s, 1) < 0)
110 connect_to_socket (bool blocking)
113 struct sockaddr_in ia;
115 s = socket (AF_INET, SOCK_STREAM, 0);
117 memset (&ia, 0, sizeof (ia));
118 ia.sin_family = AF_INET;
119 inet_pton (AF_INET, "127.0.0.1", &ia.sin_addr);
120 ia.sin_port = htons (TEST_PORT);
124 #ifdef WINDOWS_NATIVE
125 unsigned long iMode = 1;
126 ioctl (s, FIONBIO, (char *) &iMode);
128 #elif defined F_GETFL
129 int oldflags = fcntl (s, F_GETFL, NULL);
131 if (!(oldflags & O_NONBLOCK))
132 fcntl (s, F_SETFL, oldflags | O_NONBLOCK);
136 if (connect (s, (struct sockaddr *) &ia, sizeof (ia)) < 0
137 && (blocking || errno != EINPROGRESS))
147 /* A slightly more convenient interface to select(2).
148 Waits until a specific event occurs on a file descriptor FD.
149 EV is a bit mask of events to look for:
150 SEL_IN - input can be polled without blocking,
151 SEL_OUT - output can be provided without blocking,
152 SEL_EXC - an exception occurred,
153 A maximum wait time is specified by TIMEOUT.
154 *TIMEOUT = { 0, 0 } means to return immediately,
155 TIMEOUT = NULL means to wait indefinitely. */
157 enum { SEL_IN = 1, SEL_OUT = 2, SEL_EXC = 4 };
160 do_select (int fd, int ev, struct timeval *timeout, select_fn my_select)
162 fd_set rfds, wfds, xfds;
174 r = my_select (fd + 1, &rfds, &wfds, &xfds, timeout);
179 if (FD_ISSET (fd, &rfds))
181 if (FD_ISSET (fd, &wfds))
183 if (FD_ISSET (fd, &xfds))
186 failed ("select returned 0");
188 failed ("select returned unrequested events");
194 do_select_nowait (int fd, int ev, select_fn my_select)
199 return do_select (fd, ev, &tv0, my_select);
203 do_select_wait (int fd, int ev, select_fn my_select)
205 return do_select (fd, ev, NULL, my_select);
209 /* Test select(2) for TTYs. */
213 test_tty (select_fn my_select)
215 if (do_select_nowait (0, SEL_IN, my_select) != 0)
217 if (do_select_nowait (0, SEL_OUT, my_select) == 0)
218 failed ("cannot write");
220 if (do_select_wait (0, SEL_IN, my_select) == 0)
221 failed ("return with infinite timeout");
224 if (do_select_nowait (0, SEL_IN, my_select) != 0)
225 failed ("can read after getc");
230 /* Test select(2) on invalid file descriptors. */
233 do_select_bad_fd (int fd, int ev, struct timeval *timeout, select_fn my_select)
235 fd_set rfds, wfds, xfds;
246 return my_select (fd + 1, &rfds, &wfds, &xfds, timeout);
247 /* In this case, when fd is invalid, on some platforms, the bit for fd
248 is left alone in the fd_set, whereas on other platforms it is cleared.
249 So, don't check the bit for fd here. */
253 do_select_bad_fd_nowait (int fd, int ev, select_fn my_select)
258 return do_select_bad_fd (fd, ev, &tv0, my_select);
262 test_bad_fd (select_fn my_select)
264 /* This tests fails on OSF/1 and native Windows, even with fd = 16. */
265 #if !(defined __osf__ || defined WINDOWS_NATIVE)
268 /* On Linux, MacOS X, *BSD, values of fd like 99 or 399 are discarded
269 by the kernel early and therefore do *not* lead to EBADF, as required
271 # if defined __linux__ || (defined __APPLE__ && defined __MACH__) || defined __FreeBSD__ || defined __OpenBSD__ || defined __NetBSD__
277 if (do_select_bad_fd_nowait (fd, SEL_IN, my_select) == 0 || errno != EBADF)
278 failed ("invalid fd among rfds");
279 if (do_select_bad_fd_nowait (fd, SEL_OUT, my_select) == 0 || errno != EBADF)
280 failed ("invalid fd among wfds");
281 if (do_select_bad_fd_nowait (fd, SEL_EXC, my_select) == 0 || errno != EBADF)
282 failed ("invalid fd among xfds");
287 /* Test select(2) for unconnected nonblocking sockets. */
290 test_connect_first (select_fn my_select)
292 int s = open_server_socket ();
293 struct sockaddr_in ia;
298 if (do_select_nowait (s, SEL_IN | SEL_EXC, my_select) != 0)
299 failed ("can read, socket not connected");
301 c1 = connect_to_socket (false);
303 if (do_select_wait (s, SEL_IN | SEL_EXC, my_select) != SEL_IN)
304 failed ("expecting readability on passive socket");
305 if (do_select_nowait (s, SEL_IN | SEL_EXC, my_select) != SEL_IN)
306 failed ("expecting readability on passive socket");
308 addrlen = sizeof (ia);
309 c2 = accept (s, (struct sockaddr *) &ia, &addrlen);
310 ASSERT (close (s) == 0);
311 ASSERT (close (c1) == 0);
312 ASSERT (close (c2) == 0);
316 /* Test select(2) for unconnected blocking sockets. */
319 test_accept_first (select_fn my_select)
321 #ifndef WINDOWS_NATIVE
322 int s = open_server_socket ();
323 struct sockaddr_in ia;
334 addrlen = sizeof (ia);
335 c = accept (s, (struct sockaddr *) &ia, &addrlen);
336 ASSERT (close (s) == 0);
337 ASSERT (write (c, "foo", 3) == 3);
338 ASSERT (read (c, buf, 3) == 3);
339 shutdown (c, SHUT_RD);
340 ASSERT (close (c) == 0);
345 ASSERT (close (s) == 0);
346 c = connect_to_socket (true);
347 if (do_select_nowait (c, SEL_OUT, my_select) != SEL_OUT)
348 failed ("cannot write after blocking connect");
349 ASSERT (write (c, "foo", 3) == 3);
351 if (do_select_wait (c, SEL_IN, my_select) != SEL_IN)
352 failed ("cannot read data left in the socket by closed process");
353 ASSERT (read (c, buf, 3) == 3);
354 ASSERT (write (c, "foo", 3) == 3);
355 (void) close (c); /* may fail with errno = ECONNRESET */
361 /* Common code for pipes and connected sockets. */
364 test_pair (int rd, int wd, select_fn my_select)
367 if (do_select_wait (wd, SEL_IN | SEL_OUT | SEL_EXC, my_select) != SEL_OUT)
368 failed ("expecting writability before writing");
369 if (do_select_nowait (wd, SEL_IN | SEL_OUT | SEL_EXC, my_select) != SEL_OUT)
370 failed ("expecting writability before writing");
372 ASSERT (write (wd, "foo", 3) == 3);
373 if (do_select_wait (rd, SEL_IN, my_select) != SEL_IN)
374 failed ("expecting readability after writing");
375 if (do_select_nowait (rd, SEL_IN, my_select) != SEL_IN)
376 failed ("expecting readability after writing");
378 ASSERT (read (rd, buf, 3) == 3);
382 /* Test select(2) on connected sockets. */
385 test_socket_pair (select_fn my_select)
387 struct sockaddr_in ia;
389 socklen_t addrlen = sizeof (ia);
390 int s = open_server_socket ();
391 int c1 = connect_to_socket (false);
392 int c2 = accept (s, (struct sockaddr *) &ia, &addrlen);
394 ASSERT (close (s) == 0);
396 test_pair (c1, c2, my_select);
397 ASSERT (close (c1) == 0);
398 ASSERT (write (c2, "foo", 3) == 3);
399 (void) close (c2); /* may fail with errno = ECONNRESET */
403 /* Test select(2) on pipes. */
406 test_pipe (select_fn my_select)
410 ASSERT (pipe (fd) == 0);
411 test_pair (fd[0], fd[1], my_select);
412 ASSERT (close (fd[0]) == 0);
413 ASSERT (close (fd[1]) == 0);
420 test_function (select_fn my_select)
425 printf ("Please press Enter\n");
426 test (test_tty, "TTY", my_select);
429 result += test (test_bad_fd, my_select, "Invalid fd test");
430 result += test (test_connect_first, my_select, "Unconnected socket test");
431 result += test (test_socket_pair, my_select, "Connected sockets test");
432 result += test (test_accept_first, my_select, "General socket test with fork");
433 result += test (test_pipe, my_select, "Pipe test");