misc/tst-pselect.c

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