/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2016 Joyent, Inc.
 */

/*
 * Test various C11 thread-specific storage (tss(3C)) interfaces.
 */

#include <threads.h>
#include <sys/debug.h>

#define	TSS_NTHREADS	128

static tss_t ct_key;
static int ct_count;
static int ct_ready;
static mtx_t ct_mtx;
static cnd_t ct_cnd;

static void
ct_tss_dtor(void *arg)
{
	VERIFY3S(mtx_lock(&ct_mtx), ==, thrd_success);
	ct_count++;
	VERIFY3S(mtx_unlock(&ct_mtx), ==, thrd_success);
}

static int
ct_tss_thr(void *arg)
{
	VERIFY3P(tss_get(ct_key), ==, NULL);
	VERIFY3S(tss_set(ct_key, arg), ==, thrd_success);

	VERIFY3S(mtx_lock(&ct_mtx), ==, thrd_success);
	ct_ready++;
	if (ct_ready == TSS_NTHREADS) {
		VERIFY3S(cnd_broadcast(&ct_cnd), ==, thrd_success);
	} else {
		while (ct_ready != TSS_NTHREADS) {
			VERIFY3S(cnd_wait(&ct_cnd, &ct_mtx), ==, thrd_success);
		}
	}
	VERIFY3S(mtx_unlock(&ct_mtx), ==, thrd_success);

	VERIFY3P(tss_get(ct_key), ==, arg);

	return (0);
}

int
main(void)
{
	int i;
	thrd_t threads[TSS_NTHREADS];

	VERIFY3S(tss_create(&ct_key, ct_tss_dtor), ==, thrd_success);
	VERIFY3S(mtx_init(&ct_mtx, mtx_plain), ==, thrd_success);
	VERIFY3S(cnd_init(&ct_cnd), ==, thrd_success);

	for (i = 0; i < TSS_NTHREADS; i++) {
		VERIFY3S(thrd_create(&threads[i], ct_tss_thr,
		    (void *)(uintptr_t)(i + 100)), ==, thrd_success);
	}

	for (i = 0; i < TSS_NTHREADS; i++) {
		VERIFY3S(thrd_join(threads[i], NULL), ==, thrd_success);
	}

	VERIFY3S(ct_count, ==, TSS_NTHREADS);

	mtx_destroy(&ct_mtx);
	cnd_destroy(&ct_cnd);
	tss_delete(ct_key);

	return (0);
}