/*	$OpenBSD: pthread_atfork.c,v 1.3 2005/11/05 04:28:46 fgsch Exp $	*/

/*
 * Federico Schwindt <fgsch@openbsd.org>, 2005. Public Domain.
 */

#include <sys/types.h>
#include <pthread.h>
#include <unistd.h>

#include "test.h"

int cnt;

void
prepare1(void)
{
	ASSERT(cnt == 1);
	cnt--;
}

void
prepare2(void)
{
	ASSERT(cnt == 2);
	cnt--;
}

void
parent1(void)
{
	ASSERT(cnt == 0);
	cnt += 2;
}

void
parent2(void)
{
	ASSERT(cnt == 2);
	cnt -= 2;
}

void
child1(void)
{
	ASSERT(cnt == 0);
	cnt += 3;
}

void
child2(void)
{
	ASSERT(cnt == 3);
	cnt++;
}

void *
forker1(void *arg)
{
	CHECKr(pthread_atfork(prepare1, parent1, child1));

	cnt = 1;
	switch (fork()) {
	case -1:
		PANIC("fork");
		break;

	case 0:
		ASSERT(cnt == 3);
		_exit(0);

	default:
		ASSERT(cnt == 2);
		break;
	}

	cnt = 1;

	return (NULL);
}

void *
forker2(void *arg)
{
	CHECKr(pthread_atfork(prepare2, parent2, child2));

	cnt = 2;
	switch (fork()) {
	case -1:
		PANIC("fork");
		break;

	case 0:
		ASSERT(cnt == 4);
		_exit(0);

	default:
		ASSERT(cnt == 0);
		break;
	}

	cnt = 0;

	return (NULL);
}

int
main(int argc, char **argv)
{
	pthread_t tid;

	CHECKr(pthread_create(&tid, NULL, forker1, NULL));
	CHECKr(pthread_join(tid, NULL));
	ASSERT(cnt == 1);
	CHECKr(pthread_create(&tid, NULL, forker2, NULL));
	CHECKr(pthread_join(tid, NULL));
	ASSERT(cnt == 0);
	SUCCEED;
}