/*	$OpenBSD: pthread_specific.c,v 1.4 2012/02/20 02:19:15 guenther Exp $	*/

/*
 * Copyright (c) 2002 CubeSoft Communications, Inc.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistribution of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Neither the name of CubeSoft Communications, nor the names of its
 *    contributors may be used to endorse or promote products derived from
 *    this software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
 * USE OF THIS SOFTWARE EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/types.h>
#include <sys/resource.h>
#include <pthread.h>
#include <pthread_np.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include "test.h"

#define NTHREADS	128

pthread_key_t key;
int destroy_run = 0;

static void *
run_thread(void *arg)
{
	int i;

	CHECKe(write(STDOUT_FILENO, ".", 1));
	for (i = 0; i < 32767; i++) {
		void *p;

		p = pthread_getspecific(key);
		if (p == NULL) {
			CHECKr(pthread_setspecific(key, pthread_self()));
		} else {
			ASSERT(p == pthread_self());
		}
		fflush(stderr);
	}

	return (NULL);
}

static void
destroy_key(void *keyp)
{
	destroy_run++;
}

int
main(int argc, char *argv[])
{
	pthread_t threads[NTHREADS];
	struct rlimit nproc;
	int i;

	CHECKe(getrlimit(RLIMIT_NPROC, &nproc));
	nproc.rlim_cur = nproc.rlim_max;
	CHECKe(setrlimit(RLIMIT_NPROC, &nproc));
	
	CHECKr(pthread_key_create(&key, destroy_key));
	for (i = 0; i < NTHREADS; i++) {
		CHECKr(pthread_create(&threads[i], NULL, run_thread, NULL));
	}
	for (i = 0; i < NTHREADS; i++) {
		CHECKr(pthread_join(threads[i], NULL));
	}
	CHECKe(write(STDOUT_FILENO, "\n", 1));
	
	CHECKr(pthread_key_delete(key));

	ASSERT(destroy_run > 0);

	SUCCEED;
}