nptl/tst-fork1.c

   1 /* Copyright (C) 2002-2015 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3    Contributed by Roland McGrath <roland@redhat.com>, 2002.
   4
   5    The GNU C Library is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU Lesser General Public
   7    License as published by the Free Software Foundation; either
   8    version 2.1 of the License, or (at your option) any later version.
   9
  10    The GNU C Library is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13    Lesser General Public License for more details.
  14
  15    You should have received a copy of the GNU Lesser General Public
  16    License along with the GNU C Library; if not, see
  17    <http://www.gnu.org/licenses/>.  */
  18
  19 #include <errno.h>
  20 #include <pthread.h>
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <unistd.h>
  25 #include <sys/wait.h>
  26
  27 static void *
  28 thread_function (void * arg)
  29 {
  30   int i = (intptr_t) arg;
  31   int status;
  32   pid_t pid;
  33   pid_t pid2;
  34
  35   pid = fork ();
  36   switch (pid)
  37     {
  38     case 0:
  39       printf ("%ld for %d\n", (long int) getpid (), i);
  40       struct timespec ts = { .tv_sec = 0, .tv_nsec = 100000000 * i };
  41       nanosleep (&ts, NULL);
  42       _exit (i);
  43       break;
  44     case -1:
  45       printf ("fork: %m\n");
  46       return (void *) 1l;
  47       break;
  48     }
  49
  50   pid2 = TEMP_FAILURE_RETRY (waitpid (pid, &status, 0));
  51   if (pid2 != pid)
  52     {
  53       printf ("waitpid returned %ld, expected %ld\n",
  54               (long int) pid2, (long int) pid);
  55       return (void *) 1l;
  56     }
  57
  58   printf ("%ld with %d, expected %d\n",
  59           (long int) pid, WEXITSTATUS (status), i);
  60
  61   return WEXITSTATUS (status) == i ? NULL : (void *) 1l;
  62 }
  63
  64 #define N 5
  65 static const int t[N] = { 7, 6, 5, 4, 3 };
  66
  67 int
  68 main (void)
  69 {
  70   pthread_t th[N];
  71   int i;
  72   int result = 0;
  73   pthread_attr_t at;
  74
  75   if (pthread_attr_init (&at) != 0)
  76     {
  77       puts ("attr_init failed");
  78       return 1;
  79     }
  80
  81   if (pthread_attr_setstacksize (&at, 1 * 1024 * 1024) != 0)
  82     {
  83       puts ("attr_setstacksize failed");
  84       return 1;
  85     }
  86
  87   for (i = 0; i < N; ++i)
  88     if (pthread_create (&th[i], NULL, thread_function,
  89                         (void *) (intptr_t) t[i]) != 0)
  90       {
  91         printf ("creation of thread %d failed\n", i);
  92         exit (1);
  93       }
  94
  95   if (pthread_attr_destroy (&at) != 0)
  96     {
  97       puts ("attr_destroy failed");
  98       return 1;
  99     }
 100
 101   for (i = 0; i < N; ++i)
 102     {
 103       void *v;
 104       if (pthread_join (th[i], &v) != 0)
 105         {
 106           printf ("join of thread %d failed\n", i);
 107           result = 1;
 108         }
 109       else if (v != NULL)
 110         {
 111           printf ("join %d successful, but child failed\n", i);
 112           result = 1;
 113         }
 114       else
 115         printf ("join %d successful\n", i);
 116     }
 117
 118   return result;
 119 }