tests/test-select-stdin.c

   1 /* Test of select() substitute, reading from stdin.
   2    Copyright (C) 2008, 2009, 2010 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 #include <sys/time.h>
  25 #include <unistd.h>
  26
  27 int
  28 main (void)
  29 {
  30   printf ("Applying select() from standard input. Press Ctrl-C to abort.\n");
  31   for (;;)
  32     {
  33       struct timeval before;
  34       struct timeval after;
  35       unsigned long spent_usec;
  36       fd_set readfds;
  37       struct timeval timeout;
  38       int ret;
  39
  40       gettimeofday (&before, NULL);
  41
  42       FD_ZERO (&readfds);
  43       FD_SET (0, &readfds);
  44       timeout.tv_sec = 0;
  45       timeout.tv_usec = 500000;
  46       ret = select (1, &readfds, NULL, NULL, &timeout);
  47
  48       gettimeofday (&after, NULL);
  49       spent_usec = (after.tv_sec - before.tv_sec) * 1000000
  50                    + after.tv_usec - before.tv_usec;
  51
  52       if (ret < 0)
  53         {
  54           perror ("select failed");
  55           exit (1);
  56         }
  57       if ((ret == 0) != ! FD_ISSET (0, &readfds))
  58         {
  59           fprintf (stderr, "incorrect return value\n");
  60           exit (1);
  61         }
  62       if (ret == 0)
  63         {
  64           if (spent_usec < 250000)
  65             {
  66               fprintf (stderr, "returned too early\n");
  67               exit (1);
  68             }
  69           /* Timeout */
  70           printf ("."); fflush (stdout);
  71         }
  72       else
  73         {
  74           char c;
  75
  76           printf ("Input available! Trying to read 1 byte...\n");
  77           read (0, &c, 1);
  78         }
  79     }
  80 }