nptl/tst-fork1.c

   1 /* Copyright (C) 2002, 2004 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, write to the Free
  17    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  18    02111-1307 USA.  */
  19
  20 #include <errno.h>
  21 #include <pthread.h>
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #include <unistd.h>
  26 #include <sys/wait.h>
  27
  28 static void *
  29 thread_function (void * arg)
  30 {
  31   int i = (intptr_t) arg;
  32   int status;
  33   pid_t pid;
  34   pid_t pid2;
  35
  36   pid = fork ();
  37   switch (pid)
  38     {
  39     case 0:
  40       printf ("%ld for %d\n", (long int) getpid (), i);
  41       struct timespec ts = { .tv_sec = 0, .tv_nsec = 100000000 * i };
  42       nanosleep (&ts, NULL);
  43       _exit (i);
  44       break;
  45     case -1:
  46       printf ("fork: %m\n");
  47       return (void *) 1l;
  48       break;
  49     }
  50
  51   pid2 = TEMP_FAILURE_RETRY (waitpid (pid, &status, 0));
  52   if (pid2 != pid)
  53     {
  54       printf ("waitpid returned %ld, expected %ld\n",
  55               (long int) pid2, (long int) pid);
  56       return (void *) 1l;
  57     }
  58
  59   printf ("%ld with %d, expected %d\n",
  60           (long int) pid, WEXITSTATUS (status), i);
  61
  62   return WEXITSTATUS (status) == i ? NULL : (void *) 1l;
  63 }
  64
  65 #define N 5
  66 static const int t[N] = { 7, 6, 5, 4, 3 };
  67
  68 int
  69 main (void)
  70 {
  71   pthread_t th[N];
  72   int i;
  73   int result = 0;
  74   pthread_attr_t at;
  75
  76   if (pthread_attr_init (&at) != 0)
  77     {
  78       puts ("attr_init failed");
  79       return 1;
  80     }
  81
  82   if (pthread_attr_setstacksize (&at, 1 * 1024 * 1024) != 0)
  83     {
  84       puts ("attr_setstacksize failed");
  85       return 1;
  86     }
  87
  88   for (i = 0; i < N; ++i)
  89     if (pthread_create (&th[i], NULL, thread_function,
  90                         (void *) (intptr_t) t[i]) != 0)
  91       {
  92         printf ("creation of thread %d failed\n", i);
  93         exit (1);
  94       }
  95
  96   if (pthread_attr_destroy (&at) != 0)
  97     {
  98       puts ("attr_destroy failed");
  99       return 1;
 100     }
 101
 102   for (i = 0; i < N; ++i)
 103     {
 104       void *v;
 105       if (pthread_join (th[i], &v) != 0)
 106         {
 107           printf ("join of thread %d failed\n", i);
 108           result = 1;
 109         }
 110       else if (v != NULL)
 111         {
 112           printf ("join %d successful, but child failed\n", i);
 113           result = 1;
 114         }
 115       else
 116         printf ("join %d successful\n", i);
 117     }
 118
 119   return result;
 120 }