glibc/nptl/tst-atfork2.c
Ulrich Drepper dfdd294a2a Update.
* Makefile: Add rules to build and run tst-atfork2 test.
	* tst-atfork2.c: New file.
	* tst-atfork2mod.c: New file.

	* sysdeps/unix/sysv/linux/unregister-atfork.c
	(__unregister_atfork): Free the memory allocated for the handlers
	after removing them from the lists.

	* sysdeps/unix/sysv/linux/register-atfork.c: Define memeory
	cleanup function.

	* tst-atfork1.c (do_test): Wait for the child we forked.
	Report error in child.

	* sysdeps/unix/sysv/linux/fork.c (__libc_fork): Fix comment.
2003-02-13 07:14:38 +00:00

160 lines
2.9 KiB
C

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