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);
88 setsockopt (s
, SOL_SOCKET
, SO_REUSEPORT
, &x
, sizeof (x
));
90 memset (&ia
, 0, sizeof (ia
));
91 ia
.sin_family
= AF_INET
;
92 inet_pton (AF_INET
, "127.0.0.1", &ia
.sin_addr
);
93 ia
.sin_port
= htons (TEST_PORT
);
94 if (bind (s
, (struct sockaddr
*) &ia
, sizeof (ia
)) < 0)
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");
231 do_select_bad_nfd_nowait (int nfd
, select_fn my_select
)
237 return my_select (nfd
, NULL
, NULL
, NULL
, &tv0
);
241 test_bad_nfd (select_fn my_select
)
243 if (do_select_bad_nfd_nowait (-1, my_select
) != -1 || errno
!= EINVAL
)
244 failed ("invalid errno after negative nfds");
245 /* Can't test FD_SETSIZE + 1 for EINVAL, since some systems allow
246 dynamically larger set size by redefining FD_SETSIZE anywhere up
247 to the actual maximum fd. */
248 /* if (do_select_bad_nfd_nowait (FD_SETSIZE + 1, my_select) != -1 */
249 /* || errno != EINVAL) */
250 /* failed ("invalid errno after bogus nfds"); */
253 /* Test select(2) on invalid file descriptors. */
256 do_select_bad_fd (int fd
, int ev
, struct timeval
*timeout
, select_fn my_select
)
258 fd_set rfds
, wfds
, xfds
;
270 return my_select (fd
+ 1, &rfds
, &wfds
, &xfds
, timeout
);
271 /* In this case, when fd is invalid, on some platforms, the bit for fd
272 is left alone in the fd_set, whereas on other platforms it is cleared.
273 So, don't check the bit for fd here. */
277 do_select_bad_fd_nowait (int fd
, int ev
, select_fn my_select
)
282 return do_select_bad_fd (fd
, ev
, &tv0
, my_select
);
286 test_bad_fd (select_fn my_select
)
288 /* This tests fails on OSF/1 and native Windows, even with fd = 16. */
289 #if !(defined __osf__ || defined WINDOWS_NATIVE)
292 /* On Linux, Mac OS X, *BSD, values of fd like 99 or 399 are discarded
293 by the kernel early and therefore do *not* lead to EBADF, as required
295 # if defined __linux__ || (defined __APPLE__ && defined __MACH__) || defined __FreeBSD__ || defined __OpenBSD__ || defined __NetBSD__
301 if (do_select_bad_fd_nowait (fd
, SEL_IN
, my_select
) == 0 || errno
!= EBADF
)
302 failed ("invalid fd among rfds");
303 if (do_select_bad_fd_nowait (fd
, SEL_OUT
, my_select
) == 0 || errno
!= EBADF
)
304 failed ("invalid fd among wfds");
305 if (do_select_bad_fd_nowait (fd
, SEL_EXC
, my_select
) == 0 || errno
!= EBADF
)
306 failed ("invalid fd among xfds");
311 /* Test select(2) for unconnected nonblocking sockets. */
314 test_connect_first (select_fn my_select
)
316 int s
= open_server_socket ();
317 struct sockaddr_in ia
;
322 if (do_select_nowait (s
, SEL_IN
| SEL_EXC
, my_select
) != 0)
323 failed ("can read, socket not connected");
325 c1
= connect_to_socket (false);
327 if (do_select_wait (s
, SEL_IN
| SEL_EXC
, my_select
) != SEL_IN
)
328 failed ("expecting readability on passive socket");
329 if (do_select_nowait (s
, SEL_IN
| SEL_EXC
, my_select
) != SEL_IN
)
330 failed ("expecting readability on passive socket");
332 addrlen
= sizeof (ia
);
333 c2
= accept (s
, (struct sockaddr
*) &ia
, &addrlen
);
334 ASSERT (close (s
) == 0);
335 ASSERT (close (c1
) == 0);
336 ASSERT (close (c2
) == 0);
340 /* Test select(2) for unconnected blocking sockets. */
343 test_accept_first (select_fn my_select
)
345 #ifndef WINDOWS_NATIVE
346 int s
= open_server_socket ();
347 struct sockaddr_in ia
;
358 addrlen
= sizeof (ia
);
359 c
= accept (s
, (struct sockaddr
*) &ia
, &addrlen
);
360 ASSERT (close (s
) == 0);
361 ASSERT (write (c
, "foo", 3) == 3);
362 ASSERT (read (c
, buf
, 3) == 3);
363 shutdown (c
, SHUT_RD
);
364 ASSERT (close (c
) == 0);
369 ASSERT (close (s
) == 0);
370 c
= connect_to_socket (true);
371 if (do_select_nowait (c
, SEL_OUT
, my_select
) != SEL_OUT
)
372 failed ("cannot write after blocking connect");
373 ASSERT (write (c
, "foo", 3) == 3);
375 if (do_select_wait (c
, SEL_IN
, my_select
) != SEL_IN
)
376 failed ("cannot read data left in the socket by closed process");
377 ASSERT (read (c
, buf
, 3) == 3);
378 ASSERT (write (c
, "foo", 3) == 3);
379 (void) close (c
); /* may fail with errno = ECONNRESET */
385 /* Common code for pipes and connected sockets. */
388 test_pair (int rd
, int wd
, select_fn my_select
)
391 if (do_select_wait (wd
, SEL_IN
| SEL_OUT
| SEL_EXC
, my_select
) != SEL_OUT
)
392 failed ("expecting writability before writing");
393 if (do_select_nowait (wd
, SEL_IN
| SEL_OUT
| SEL_EXC
, my_select
) != SEL_OUT
)
394 failed ("expecting writability before writing");
396 ASSERT (write (wd
, "foo", 3) == 3);
397 if (do_select_wait (rd
, SEL_IN
, my_select
) != SEL_IN
)
398 failed ("expecting readability after writing");
399 if (do_select_nowait (rd
, SEL_IN
, my_select
) != SEL_IN
)
400 failed ("expecting readability after writing");
402 ASSERT (read (rd
, buf
, 3) == 3);
406 /* Test select(2) on connected sockets. */
409 test_socket_pair (select_fn my_select
)
411 struct sockaddr_in ia
;
413 socklen_t addrlen
= sizeof (ia
);
414 int s
= open_server_socket ();
415 int c1
= connect_to_socket (false);
416 int c2
= accept (s
, (struct sockaddr
*) &ia
, &addrlen
);
418 ASSERT (close (s
) == 0);
420 test_pair (c1
, c2
, my_select
);
421 ASSERT (close (c1
) == 0);
422 ASSERT (write (c2
, "foo", 3) == 3);
423 (void) close (c2
); /* may fail with errno = ECONNRESET */
427 /* Test select(2) on pipes. */
430 test_pipe (select_fn my_select
)
434 ASSERT (pipe (fd
) == 0);
435 test_pair (fd
[0], fd
[1], my_select
);
436 ASSERT (close (fd
[0]) == 0);
437 ASSERT (close (fd
[1]) == 0);
444 test_function (select_fn my_select
)
449 printf ("Please press Enter\n");
450 test (test_tty
, "TTY", my_select
);
453 result
+= test (test_bad_nfd
, my_select
, "Invalid nfd test");
454 result
+= test (test_bad_fd
, my_select
, "Invalid fd test");
455 result
+= test (test_connect_first
, my_select
, "Unconnected socket test");
456 result
+= test (test_socket_pair
, my_select
, "Connected sockets test");
457 result
+= test (test_accept_first
, my_select
, "General socket test with fork");
458 result
+= test (test_pipe
, my_select
, "Pipe test");