/gnulib-tests/test-select.h
C Header | 461 lines | 323 code | 90 blank | 48 comment | 97 complexity | 7a356c48a5f365b493398d4fca3c9bef MD5 | raw file
Possible License(s): GPL-3.0, LGPL-2.0
- /* Test of select() substitute.
- Copyright (C) 2008-2012 Free Software Foundation, Inc.
- This program is free software: you can redistribute it and/or modify
- it under the terms of the GNU General Public License as published by
- the Free Software Foundation; either version 3 of the License, or
- (at your option) any later version.
- This program is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- GNU General Public License for more details.
- You should have received a copy of the GNU General Public License
- along with this program. If not, see <http://www.gnu.org/licenses/>. */
- /* Written by Paolo Bonzini, 2008. */
- #include <stdio.h>
- #include <string.h>
- #include <netinet/in.h>
- #include <arpa/inet.h>
- #include <unistd.h>
- #include <fcntl.h>
- #include <stdlib.h>
- #include <stdbool.h>
- #include <sys/ioctl.h>
- #include <errno.h>
- #include "macros.h"
- #if (defined _WIN32 || defined __WIN32__) && ! defined __CYGWIN__
- # define WINDOWS_NATIVE
- #endif
- #ifdef HAVE_SYS_WAIT_H
- # include <sys/wait.h>
- #endif
- #ifndef SO_REUSEPORT
- # define SO_REUSEPORT SO_REUSEADDR
- #endif
- #define TEST_PORT 12345
- typedef int (*select_fn) (int, fd_set *, fd_set *, fd_set *, struct timeval *);
- /* Minimal testing infrastructure. */
- static int failures;
- static void
- failed (const char *reason)
- {
- if (++failures > 1)
- printf (" ");
- printf ("failed (%s)\n", reason);
- }
- static int
- test (void (*fn) (select_fn), select_fn my_select, const char *msg)
- {
- failures = 0;
- printf ("%s... ", msg);
- fflush (stdout);
- fn (my_select);
- if (!failures)
- printf ("passed\n");
- return failures;
- }
- /* Funny socket code. */
- static int
- open_server_socket (void)
- {
- int s, x;
- struct sockaddr_in ia;
- s = socket (AF_INET, SOCK_STREAM, 0);
- x = 1;
- setsockopt (s, SOL_SOCKET, SO_REUSEPORT, &x, sizeof (x));
- memset (&ia, 0, sizeof (ia));
- ia.sin_family = AF_INET;
- inet_pton (AF_INET, "127.0.0.1", &ia.sin_addr);
- ia.sin_port = htons (TEST_PORT);
- if (bind (s, (struct sockaddr *) &ia, sizeof (ia)) < 0)
- {
- perror ("bind");
- exit (77);
- }
- if (listen (s, 1) < 0)
- {
- perror ("listen");
- exit (77);
- }
- return s;
- }
- static int
- connect_to_socket (bool blocking)
- {
- int s;
- struct sockaddr_in ia;
- s = socket (AF_INET, SOCK_STREAM, 0);
- memset (&ia, 0, sizeof (ia));
- ia.sin_family = AF_INET;
- inet_pton (AF_INET, "127.0.0.1", &ia.sin_addr);
- ia.sin_port = htons (TEST_PORT);
- if (!blocking)
- {
- #ifdef WINDOWS_NATIVE
- unsigned long iMode = 1;
- ioctl (s, FIONBIO, (char *) &iMode);
- #elif defined F_GETFL
- int oldflags = fcntl (s, F_GETFL, NULL);
- if (!(oldflags & O_NONBLOCK))
- fcntl (s, F_SETFL, oldflags | O_NONBLOCK);
- #endif
- }
- if (connect (s, (struct sockaddr *) &ia, sizeof (ia)) < 0
- && (blocking || errno != EINPROGRESS))
- {
- perror ("connect");
- exit (77);
- }
- return s;
- }
- /* A slightly more convenient interface to select(2).
- Waits until a specific event occurs on a file descriptor FD.
- EV is a bit mask of events to look for:
- SEL_IN - input can be polled without blocking,
- SEL_OUT - output can be provided without blocking,
- SEL_EXC - an exception occurred,
- A maximum wait time is specified by TIMEOUT.
- *TIMEOUT = { 0, 0 } means to return immediately,
- TIMEOUT = NULL means to wait indefinitely. */
- enum { SEL_IN = 1, SEL_OUT = 2, SEL_EXC = 4 };
- static int
- do_select (int fd, int ev, struct timeval *timeout, select_fn my_select)
- {
- fd_set rfds, wfds, xfds;
- int r, rev;
- FD_ZERO (&rfds);
- FD_ZERO (&wfds);
- FD_ZERO (&xfds);
- if (ev & SEL_IN)
- FD_SET (fd, &rfds);
- if (ev & SEL_OUT)
- FD_SET (fd, &wfds);
- if (ev & SEL_EXC)
- FD_SET (fd, &xfds);
- r = my_select (fd + 1, &rfds, &wfds, &xfds, timeout);
- if (r < 0)
- return r;
- rev = 0;
- if (FD_ISSET (fd, &rfds))
- rev |= SEL_IN;
- if (FD_ISSET (fd, &wfds))
- rev |= SEL_OUT;
- if (FD_ISSET (fd, &xfds))
- rev |= SEL_EXC;
- if (rev && r == 0)
- failed ("select returned 0");
- if (rev & ~ev)
- failed ("select returned unrequested events");
- return rev;
- }
- static int
- do_select_nowait (int fd, int ev, select_fn my_select)
- {
- struct timeval tv0;
- tv0.tv_sec = 0;
- tv0.tv_usec = 0;
- return do_select (fd, ev, &tv0, my_select);
- }
- static int
- do_select_wait (int fd, int ev, select_fn my_select)
- {
- return do_select (fd, ev, NULL, my_select);
- }
- /* Test select(2) for TTYs. */
- #ifdef INTERACTIVE
- static void
- test_tty (select_fn my_select)
- {
- if (do_select_nowait (0, SEL_IN, my_select) != 0)
- failed ("can read");
- if (do_select_nowait (0, SEL_OUT, my_select) == 0)
- failed ("cannot write");
- if (do_select_wait (0, SEL_IN, my_select) == 0)
- failed ("return with infinite timeout");
- getchar ();
- if (do_select_nowait (0, SEL_IN, my_select) != 0)
- failed ("can read after getc");
- }
- #endif
- static int
- do_select_bad_nfd_nowait (int nfd, select_fn my_select)
- {
- struct timeval tv0;
- tv0.tv_sec = 0;
- tv0.tv_usec = 0;
- errno = 0;
- return my_select (nfd, NULL, NULL, NULL, &tv0);
- }
- static void
- test_bad_nfd (select_fn my_select)
- {
- if (do_select_bad_nfd_nowait (-1, my_select) != -1 || errno != EINVAL)
- failed ("invalid errno after negative nfds");
- /* Can't test FD_SETSIZE + 1 for EINVAL, since some systems allow
- dynamically larger set size by redefining FD_SETSIZE anywhere up
- to the actual maximum fd. */
- /* if (do_select_bad_nfd_nowait (FD_SETSIZE + 1, my_select) != -1 */
- /* || errno != EINVAL) */
- /* failed ("invalid errno after bogus nfds"); */
- }
- /* Test select(2) on invalid file descriptors. */
- static int
- do_select_bad_fd (int fd, int ev, struct timeval *timeout, select_fn my_select)
- {
- fd_set rfds, wfds, xfds;
- FD_ZERO (&rfds);
- FD_ZERO (&wfds);
- FD_ZERO (&xfds);
- if (ev & SEL_IN)
- FD_SET (fd, &rfds);
- if (ev & SEL_OUT)
- FD_SET (fd, &wfds);
- if (ev & SEL_EXC)
- FD_SET (fd, &xfds);
- errno = 0;
- return my_select (fd + 1, &rfds, &wfds, &xfds, timeout);
- /* In this case, when fd is invalid, on some platforms, the bit for fd
- is left alone in the fd_set, whereas on other platforms it is cleared.
- So, don't check the bit for fd here. */
- }
- static int
- do_select_bad_fd_nowait (int fd, int ev, select_fn my_select)
- {
- struct timeval tv0;
- tv0.tv_sec = 0;
- tv0.tv_usec = 0;
- return do_select_bad_fd (fd, ev, &tv0, my_select);
- }
- static void
- test_bad_fd (select_fn my_select)
- {
- /* This tests fails on OSF/1 and native Windows, even with fd = 16. */
- #if !(defined __osf__ || defined WINDOWS_NATIVE)
- int fd;
- /* On Linux, Mac OS X, *BSD, values of fd like 99 or 399 are discarded
- by the kernel early and therefore do *not* lead to EBADF, as required
- by POSIX. */
- # if defined __linux__ || (defined __APPLE__ && defined __MACH__) || defined __FreeBSD__ || defined __OpenBSD__ || defined __NetBSD__
- fd = 16;
- # else
- fd = 99;
- # endif
- if (do_select_bad_fd_nowait (fd, SEL_IN, my_select) == 0 || errno != EBADF)
- failed ("invalid fd among rfds");
- if (do_select_bad_fd_nowait (fd, SEL_OUT, my_select) == 0 || errno != EBADF)
- failed ("invalid fd among wfds");
- if (do_select_bad_fd_nowait (fd, SEL_EXC, my_select) == 0 || errno != EBADF)
- failed ("invalid fd among xfds");
- #endif
- }
- /* Test select(2) for unconnected nonblocking sockets. */
- static void
- test_connect_first (select_fn my_select)
- {
- int s = open_server_socket ();
- struct sockaddr_in ia;
- socklen_t addrlen;
- int c1, c2;
- if (do_select_nowait (s, SEL_IN | SEL_EXC, my_select) != 0)
- failed ("can read, socket not connected");
- c1 = connect_to_socket (false);
- if (do_select_wait (s, SEL_IN | SEL_EXC, my_select) != SEL_IN)
- failed ("expecting readability on passive socket");
- if (do_select_nowait (s, SEL_IN | SEL_EXC, my_select) != SEL_IN)
- failed ("expecting readability on passive socket");
- addrlen = sizeof (ia);
- c2 = accept (s, (struct sockaddr *) &ia, &addrlen);
- ASSERT (close (s) == 0);
- ASSERT (close (c1) == 0);
- ASSERT (close (c2) == 0);
- }
- /* Test select(2) for unconnected blocking sockets. */
- static void
- test_accept_first (select_fn my_select)
- {
- #ifndef WINDOWS_NATIVE
- int s = open_server_socket ();
- struct sockaddr_in ia;
- socklen_t addrlen;
- char buf[3];
- int c, pid;
- pid = fork ();
- if (pid < 0)
- return;
- if (pid == 0)
- {
- addrlen = sizeof (ia);
- c = accept (s, (struct sockaddr *) &ia, &addrlen);
- ASSERT (close (s) == 0);
- ASSERT (write (c, "foo", 3) == 3);
- ASSERT (read (c, buf, 3) == 3);
- shutdown (c, SHUT_RD);
- ASSERT (close (c) == 0);
- exit (0);
- }
- else
- {
- ASSERT (close (s) == 0);
- c = connect_to_socket (true);
- if (do_select_nowait (c, SEL_OUT, my_select) != SEL_OUT)
- failed ("cannot write after blocking connect");
- ASSERT (write (c, "foo", 3) == 3);
- wait (&pid);
- if (do_select_wait (c, SEL_IN, my_select) != SEL_IN)
- failed ("cannot read data left in the socket by closed process");
- ASSERT (read (c, buf, 3) == 3);
- ASSERT (write (c, "foo", 3) == 3);
- (void) close (c); /* may fail with errno = ECONNRESET */
- }
- #endif
- }
- /* Common code for pipes and connected sockets. */
- static void
- test_pair (int rd, int wd, select_fn my_select)
- {
- char buf[3];
- if (do_select_wait (wd, SEL_IN | SEL_OUT | SEL_EXC, my_select) != SEL_OUT)
- failed ("expecting writability before writing");
- if (do_select_nowait (wd, SEL_IN | SEL_OUT | SEL_EXC, my_select) != SEL_OUT)
- failed ("expecting writability before writing");
- ASSERT (write (wd, "foo", 3) == 3);
- if (do_select_wait (rd, SEL_IN, my_select) != SEL_IN)
- failed ("expecting readability after writing");
- if (do_select_nowait (rd, SEL_IN, my_select) != SEL_IN)
- failed ("expecting readability after writing");
- ASSERT (read (rd, buf, 3) == 3);
- }
- /* Test select(2) on connected sockets. */
- static void
- test_socket_pair (select_fn my_select)
- {
- struct sockaddr_in ia;
- socklen_t addrlen = sizeof (ia);
- int s = open_server_socket ();
- int c1 = connect_to_socket (false);
- int c2 = accept (s, (struct sockaddr *) &ia, &addrlen);
- ASSERT (close (s) == 0);
- test_pair (c1, c2, my_select);
- ASSERT (close (c1) == 0);
- ASSERT (write (c2, "foo", 3) == 3);
- (void) close (c2); /* may fail with errno = ECONNRESET */
- }
- /* Test select(2) on pipes. */
- static void
- test_pipe (select_fn my_select)
- {
- int fd[2];
- ASSERT (pipe (fd) == 0);
- test_pair (fd[0], fd[1], my_select);
- ASSERT (close (fd[0]) == 0);
- ASSERT (close (fd[1]) == 0);
- }
- /* Do them all. */
- static int
- test_function (select_fn my_select)
- {
- int result = 0;
- #ifdef INTERACTIVE
- printf ("Please press Enter\n");
- test (test_tty, "TTY", my_select);
- #endif
- result += test (test_bad_nfd, my_select, "Invalid nfd test");
- result += test (test_bad_fd, my_select, "Invalid fd test");
- result += test (test_connect_first, my_select, "Unconnected socket test");
- result += test (test_socket_pair, my_select, "Connected sockets test");
- result += test (test_accept_first, my_select, "General socket test with fork");
- result += test (test_pipe, my_select, "Pipe test");
- return result;
- }