misc/tst-pselect.c

   1 #include <errno.h>
   2 #include <signal.h>
   3 #include <stdio.h>
   4 #include <unistd.h>
   5 #include <sys/select.h>
   6 #include <sys/wait.h>
   7
   8
   9 static volatile int handler_called;
  10
  11 static void
  12 handler (int sig)
  13 {
  14   handler_called = 1;
  15 }
  16
  17
  18 static int
  19 do_test (void)
  20 {
  21   struct sigaction sa;
  22   sa.sa_handler = handler;
  23   sa.sa_flags = 0;
  24   sigemptyset (&sa.sa_mask);
  25
  26   if (sigaction (SIGUSR1, &sa, NULL) != 0)
  27     {
  28       puts ("sigaction failed");
  29       return 1;
  30     }
  31
  32   if (sigblock (SIGUSR1) != 0)
  33     {
  34       puts ("sigblock failed");
  35       return 1;
  36     }
  37
  38   int fds[2][2];
  39
  40   if (pipe (fds[0]) != 0 || pipe (fds[1]) != 0)
  41     {
  42       puts ("pipe failed");
  43       return 1;
  44     }
  45
  46   fd_set rfds;
  47   FD_ZERO (&rfds);
  48
  49   sigset_t ss;
  50   sigprocmask (SIG_SETMASK, NULL, &ss);
  51   sigdelset (&ss, SIGUSR1);
  52
  53   struct timespec to = { .tv_sec = 0, .tv_nsec = 500000000 };
  54
  55   pid_t p = fork ();
  56   if (p == 0)
  57     {
  58       close (fds[0][1]);
  59       close (fds[1][0]);
  60
  61       FD_SET (fds[0][0], &rfds);
  62
  63       int e;
  64       do
  65         {
  66           errno = 0;
  67           e = pselect (fds[0][0] + 1, &rfds, NULL, NULL, &to, &ss);
  68         }
  69       while (e == 0);
  70
  71       if (e != -1)
  72         {
  73           puts ("child: pselect did not fail");
  74           return 0;
  75         }
  76       if (errno != EINTR)
  77         {
  78           puts ("child: pselect did not set errno to EINTR");
  79           return 0;
  80         }
  81
  82       TEMP_FAILURE_RETRY (write (fds[1][1], "foo", 3));
  83
  84       exit (0);
  85     }
  86
  87   close (fds[0][0]);
  88   close (fds[1][1]);
  89
  90   FD_SET (fds[1][0], &rfds);
  91
  92   kill (p, SIGUSR1);
  93
  94   int e = pselect (fds[1][0] + 1, &rfds, NULL, NULL, NULL, &ss);
  95   if (e == -1)
  96     {
  97       puts ("parent: pselect failed");
  98       return 1;
  99     }
 100   if (e != 1)
 101     {
 102       puts ("parent: pselect did not report readable fd");
 103       return 1;
 104     }
 105   if (!FD_ISSET (fds[1][0], &rfds))
 106     {
 107       puts ("parent: pselect reports wrong fd");
 108       return 1;
 109     }
 110
 111   if (TEMP_FAILURE_RETRY (waitpid (p, NULL, 0)) != p)
 112     {
 113       puts ("waitpid failed");
 114       return 1;
 115     }
 116
 117   return 0;
 118 }
 119
 120 #define TEST_FUNCTION do_test ()
 121 #include "../test-skeleton.c"