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