1  /* Copyright (C) 2003-2023 Free Software Foundation, Inc.
       2     This file is part of the GNU C Library.
       3  
       4     The GNU C Library is free software; you can redistribute it and/or
       5     modify it under the terms of the GNU Lesser General Public
       6     License as published by the Free Software Foundation; either
       7     version 2.1 of the License, or (at your option) any later version.
       8  
       9     The GNU C Library is distributed in the hope that it will be useful,
      10     but WITHOUT ANY WARRANTY; without even the implied warranty of
      11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      12     Lesser General Public License for more details.
      13  
      14     You should have received a copy of the GNU Lesser General Public
      15     License along with the GNU C Library; if not, see
      16     <https://www.gnu.org/licenses/>.  */
      17  
      18  #include <dlfcn.h>
      19  #include <errno.h>
      20  #include <mcheck.h>
      21  #include <pthread.h>
      22  #include <stdio.h>
      23  #include <stdlib.h>
      24  #include <unistd.h>
      25  #include <sys/wait.h>
      26  
      27  
      28  /* Must be exported.  */
      29  int val;
      30  
      31  static void
      32  prepare (void)
      33  {
      34    val *= 2;
      35  }
      36  
      37  static void
      38  parent (void)
      39  {
      40    val += 4;
      41  }
      42  
      43  static void
      44  child (void)
      45  {
      46    val += 8;
      47  }
      48  
      49  
      50  static int
      51  do_test (void)
      52  {
      53    mtrace ();
      54  
      55    if (pthread_atfork (prepare, parent, child) != 0)
      56      {
      57        puts ("do_test: atfork failed");
      58        exit (1);
      59      }
      60  
      61    void *h = dlopen ("tst-atfork2mod.so", RTLD_LAZY);
      62    if (h == NULL)
      63      {
      64        printf ("dlopen failed: %s\n", dlerror ());
      65        exit (1);
      66      }
      67  
      68    /* First trial of fork.  */
      69    pid_t pid = fork ();
      70    if (pid == -1)
      71      {
      72        puts ("1st fork failed");
      73        exit (1);
      74      }
      75  
      76    if (pid == 0)
      77      {
      78        /* Child.  */
      79        if (val != 80)
      80  	{
      81  	  printf ("1st: expected val=%d, got %d\n", 80, val);
      82  	  exit (2);
      83  	}
      84  
      85        exit (0);
      86      }
      87  
      88    /* Parent.  */
      89    if (val != 24)
      90      {
      91        printf ("1st: expected val=%d, got %d\n", 24, val);
      92        exit (1);
      93      }
      94  
      95    int status;
      96    if (TEMP_FAILURE_RETRY (waitpid (pid, &status, 0)) != pid)
      97      {
      98        puts ("1st waitpid failed");
      99        exit (1);
     100      }
     101  
     102    if (status != 0)
     103      exit (status);
     104  
     105    puts ("unloading now");
     106  
     107    /* Unload the module.  */
     108    if (dlclose (h) != 0)
     109      {
     110        puts ("dlclose failed");
     111        exit (1);
     112      }
     113  
     114    puts ("2nd fork");
     115  
     116    /* Second fork trial.   */
     117    val = 1;
     118    pid = fork ();
     119    if (pid == -1)
     120      {
     121        puts ("2nd fork failed");
     122        exit (1);
     123      }
     124  
     125    if (pid == 0)
     126      {
     127        /* Child.  */
     128        if (val != 10)
     129  	{
     130  	  printf ("2nd: expected val=%d, got %d\n", 10, val);
     131  	  exit (3);
     132  	}
     133  
     134        exit (0);
     135      }
     136  
     137    /* Parent.  */
     138    if (val != 6)
     139      {
     140        printf ("2nd: expected val=%d, got %d\n", 6, val);
     141        exit (1);
     142      }
     143  
     144    if (TEMP_FAILURE_RETRY (waitpid (pid, &status, 0)) != pid)
     145      {
     146        puts ("2nd waitpid failed");
     147        exit (1);
     148      }
     149  
     150    if (status != 0)
     151      exit (status);
     152  
     153    return 0;
     154  }
     155  
     156  #define TEST_FUNCTION do_test ()
     157  #include "../test-skeleton.c"