Blame gnulib-tests/test-select-fd.c

Packit 33f14e
/* Test of select() substitute, reading or writing from a given file descriptor.
Packit 33f14e
   Copyright (C) 2008-2017 Free Software Foundation, Inc.
Packit 33f14e
Packit 33f14e
   This program is free software: you can redistribute it and/or modify
Packit 33f14e
   it under the terms of the GNU General Public License as published by
Packit 33f14e
   the Free Software Foundation; either version 3 of the License, or
Packit 33f14e
   (at your option) any later version.
Packit 33f14e
Packit 33f14e
   This program is distributed in the hope that it will be useful,
Packit 33f14e
   but WITHOUT ANY WARRANTY; without even the implied warranty of
Packit 33f14e
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Packit 33f14e
   GNU General Public License for more details.
Packit 33f14e
Packit 33f14e
   You should have received a copy of the GNU General Public License
Packit 33f14e
   along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
Packit 33f14e
Packit 33f14e
/* Written by Bruno Haible <bruno@clisp.org>, 2008.  */
Packit 33f14e
Packit 33f14e
#include <config.h>
Packit 33f14e
Packit 33f14e
#include <stdio.h>
Packit 33f14e
#include <stdlib.h>
Packit 33f14e
#include <sys/select.h>
Packit 33f14e
Packit 33f14e
int
Packit 33f14e
main (int argc, char *argv[])
Packit 33f14e
{
Packit 33f14e
  if (argc == 4)
Packit 33f14e
    {
Packit 33f14e
      char mode = argv[1][0];
Packit 33f14e
Packit 33f14e
      if (mode == 'r' || mode == 'w')
Packit 33f14e
        {
Packit 33f14e
          int fd = atoi (argv[2]);
Packit 33f14e
Packit 33f14e
          if (fd >= 0)
Packit 33f14e
            {
Packit 33f14e
              const char *result_file_name = argv[3];
Packit 33f14e
              FILE *result_file = fopen (result_file_name, "wb");
Packit 33f14e
Packit 33f14e
              if (result_file != NULL)
Packit 33f14e
                {
Packit 33f14e
                  fd_set fds;
Packit 33f14e
                  struct timeval timeout;
Packit 33f14e
                  int ret;
Packit 33f14e
Packit 33f14e
                  FD_ZERO (&fds);
Packit 33f14e
                  FD_SET (fd, &fds);
Packit 33f14e
                  timeout.tv_sec = 0;
Packit 33f14e
                  timeout.tv_usec = 10000;
Packit 33f14e
                  ret = (mode == 'r'
Packit 33f14e
                         ? select (fd + 1, &fds, NULL, NULL, &timeout)
Packit 33f14e
                         : select (fd + 1, NULL, &fds, NULL, &timeout));
Packit 33f14e
                  if (ret < 0)
Packit 33f14e
                    {
Packit 33f14e
                      perror ("select failed");
Packit 33f14e
                      exit (1);
Packit 33f14e
                    }
Packit 33f14e
                  if ((ret == 0) != ! FD_ISSET (fd, &fds))
Packit 33f14e
                    {
Packit 33f14e
                      fprintf (stderr, "incorrect return value\n");
Packit 33f14e
                      exit (1);
Packit 33f14e
                    }
Packit 33f14e
                  fprintf (result_file, "%d\n", ret);
Packit 33f14e
                  exit (0);
Packit 33f14e
                }
Packit 33f14e
            }
Packit 33f14e
        }
Packit 33f14e
    }
Packit 33f14e
  fprintf (stderr, "Usage: test-select-fd mode fd result-file-name\n");
Packit 33f14e
  exit (1);
Packit 33f14e
}