More tests for 'select' module.
[gnulib.git] / tests / test-select-fd.c
1 /* Test of select() substitute, reading or writing from a given file descriptor.
2    Copyright (C) 2008 Free Software Foundation, Inc.
3
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.
8
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.
13
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/>.  */
16
17 /* Written by Bruno Haible <bruno@clisp.org>, 2008.  */
18
19 #include <config.h>
20
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <sys/select.h>
24
25 int
26 main (int argc, char *argv[])
27 {
28   if (argc == 3)
29     {
30       char mode = argv[1][0];
31
32       if (mode == 'r' || mode == 'w')
33         {
34           int fd = atoi (argv[2]);
35
36           if (fd >= 0)
37             {
38               fd_set fds;
39               struct timeval timeout;
40               int ret;
41
42               FD_ZERO (&fds);
43               FD_SET (fd, &fds);
44               timeout.tv_sec = 0;
45               timeout.tv_usec = 10000;
46               ret = (mode == 'r'
47                      ? select (fd + 1, &fds, NULL, NULL, &timeout)
48                      : select (fd + 1, NULL, &fds, NULL, &timeout));
49               if (ret < 0)
50                 {
51                   perror ("select failed");
52                   exit (1);
53                 }
54               if ((ret == 0) != ! FD_ISSET (fd, &fds))
55                 {
56                   fprintf (stderr, "incorrect return value\n");
57                   exit (1);
58                 }
59               fprintf (stderr, "%d\n", ret);
60               exit (0);
61             }
62         }
63     }
64   fprintf (stderr, "Usage: test-select-fd mode fd\n");
65   exit (1);
66 }