|
Packit |
6c4009 |
/* Copyright (C) 2003-2018 Free Software Foundation, Inc.
|
|
Packit |
6c4009 |
This file is part of the GNU C Library.
|
|
Packit |
6c4009 |
Contributed by Ulrich Drepper <drepper@redhat.com>, 2003.
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
The GNU C Library is free software; you can redistribute it and/or
|
|
Packit |
6c4009 |
modify it under the terms of the GNU Lesser General Public
|
|
Packit |
6c4009 |
License as published by the Free Software Foundation; either
|
|
Packit |
6c4009 |
version 2.1 of the License, or (at your option) any later version.
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
The GNU C Library is distributed in the hope that it will be useful,
|
|
Packit |
6c4009 |
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit |
6c4009 |
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Packit |
6c4009 |
Lesser General Public License for more details.
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
You should have received a copy of the GNU Lesser General Public
|
|
Packit |
6c4009 |
License along with the GNU C Library; if not, see
|
|
Packit |
6c4009 |
<http://www.gnu.org/licenses/>. */
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
#include <dlfcn.h>
|
|
Packit |
6c4009 |
#include <errno.h>
|
|
Packit |
6c4009 |
#include <mcheck.h>
|
|
Packit |
6c4009 |
#include <pthread.h>
|
|
Packit |
6c4009 |
#include <stdio.h>
|
|
Packit |
6c4009 |
#include <stdlib.h>
|
|
Packit |
6c4009 |
#include <unistd.h>
|
|
Packit |
6c4009 |
#include <sys/wait.h>
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
/* Must be exported. */
|
|
Packit |
6c4009 |
int val;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static void
|
|
Packit |
6c4009 |
prepare (void)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
val *= 2;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static void
|
|
Packit |
6c4009 |
parent (void)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
val += 4;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static void
|
|
Packit |
6c4009 |
child (void)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
val += 8;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static int
|
|
Packit |
6c4009 |
do_test (void)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
mtrace ();
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
if (pthread_atfork (prepare, parent, child) != 0)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
puts ("do_test: atfork failed");
|
|
Packit |
6c4009 |
exit (1);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
void *h = dlopen ("tst-atfork2mod.so", RTLD_LAZY);
|
|
Packit |
6c4009 |
if (h == NULL)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
printf ("dlopen failed: %s\n", dlerror ());
|
|
Packit |
6c4009 |
exit (1);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
/* First trial of fork. */
|
|
Packit |
6c4009 |
pid_t pid = fork ();
|
|
Packit |
6c4009 |
if (pid == -1)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
puts ("1st fork failed");
|
|
Packit |
6c4009 |
exit (1);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
if (pid == 0)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
/* Child. */
|
|
Packit |
6c4009 |
if (val != 80)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
printf ("1st: expected val=%d, got %d\n", 80, val);
|
|
Packit |
6c4009 |
exit (2);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
exit (0);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
/* Parent. */
|
|
Packit |
6c4009 |
if (val != 24)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
printf ("1st: expected val=%d, got %d\n", 24, val);
|
|
Packit |
6c4009 |
exit (1);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
int status;
|
|
Packit |
6c4009 |
if (TEMP_FAILURE_RETRY (waitpid (pid, &status, 0)) != pid)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
puts ("1st waitpid failed");
|
|
Packit |
6c4009 |
exit (1);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
if (status != 0)
|
|
Packit |
6c4009 |
exit (status);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
puts ("unloading now");
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
/* Unload the module. */
|
|
Packit |
6c4009 |
if (dlclose (h) != 0)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
puts ("dlclose failed");
|
|
Packit |
6c4009 |
exit (1);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
puts ("2nd fork");
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
/* Second fork trial. */
|
|
Packit |
6c4009 |
val = 1;
|
|
Packit |
6c4009 |
pid = fork ();
|
|
Packit |
6c4009 |
if (pid == -1)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
puts ("2nd fork failed");
|
|
Packit |
6c4009 |
exit (1);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
if (pid == 0)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
/* Child. */
|
|
Packit |
6c4009 |
if (val != 10)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
printf ("2nd: expected val=%d, got %d\n", 10, val);
|
|
Packit |
6c4009 |
exit (3);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
exit (0);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
/* Parent. */
|
|
Packit |
6c4009 |
if (val != 6)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
printf ("2nd: expected val=%d, got %d\n", 6, val);
|
|
Packit |
6c4009 |
exit (1);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
if (TEMP_FAILURE_RETRY (waitpid (pid, &status, 0)) != pid)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
puts ("2nd waitpid failed");
|
|
Packit |
6c4009 |
exit (1);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
if (status != 0)
|
|
Packit |
6c4009 |
exit (status);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
return 0;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
#define TEST_FUNCTION do_test ()
|
|
Packit |
6c4009 |
#include "../test-skeleton.c"
|