nptl/tst-atfork1.c

   1 /* Copyright (C) 2002 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3    Contributed by Ulrich Drepper <drepper@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 <pthread.h>
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <unistd.h>
  24
  25
  26 static int val;
  27
  28
  29 static void
  30 prepare1 (void)
  31 {
  32   val *= 2;
  33 }
  34
  35 static void
  36 prepare2 (void)
  37 {
  38   ++val;
  39 }
  40
  41 static void
  42 parent1 (void)
  43 {
  44   val += 4;
  45 }
  46
  47 static void
  48 parent2 (void)
  49 {
  50   val *= 4;
  51 }
  52
  53 static void
  54 child1 (void)
  55 {
  56   val += 8;
  57 }
  58
  59 static void
  60 child2 (void)
  61 {
  62   val *= 8;
  63 }
  64
  65
  66 static int
  67 do_test (void)
  68 {
  69   pid_t pid;
  70
  71   if (pthread_atfork (prepare1, parent1, child1) != 0)
  72     {
  73       puts ("1st atfork failed");
  74       exit (1);
  75     }
  76   if (pthread_atfork (prepare2, parent2, child2) != 0)
  77     {
  78       puts ("2nd atfork failed");
  79       exit (1);
  80     }
  81
  82   pid = fork ();
  83   if (pid == -1)
  84     {
  85       puts ("fork failed");
  86       exit (1);
  87     }
  88
  89   if (pid != 0)
  90     {
  91       /* Parent.  */
  92       if (val != 24)
  93         {
  94           printf ("expected val=%d, got %d\n", 24, val);
  95           exit (1);
  96         }
  97     }
  98   else
  99     {
 100       /* Child.  */
 101       if (val != 80)
 102         {
 103           printf ("expected val=%d, got %d\n", 80, val);
 104           exit (1);
 105         }
 106     }
 107
 108   return 0;
 109 }
 110
 111 #define TEST_FUNCTION do_test ()
 112 #include "../test-skeleton.c"