[apple/libpthread.git] / tests / pthread_threadid_np.c

#include <pthread.h>
#include <pthread/private.h>
#include <dispatch/dispatch.h>
#include <sys/wait.h>
#include <stdlib.h>

#include "darwintest_defaults.h"

extern __uint64_t __thread_selfid( void );

static void *do_test(void * arg)
{
	uint64_t threadid = __thread_selfid();
	T_ASSERT_NE(threadid, (uint64_t)0, "__thread_selfid()");

	uint64_t pth_threadid = 0;
	T_ASSERT_POSIX_ZERO(pthread_threadid_np(NULL, &pth_threadid), NULL);
	T_ASSERT_POSIX_ZERO(pthread_threadid_np(pthread_self(), &pth_threadid), NULL);
	T_EXPECT_EQ(threadid, pth_threadid, "pthread_threadid_np()");

	pth_threadid = _pthread_threadid_self_np_direct();
	T_EXPECT_EQ(threadid, pth_threadid, "pthread_threadid_np_direct()");

	if (arg) {
		*(uint64_t *)arg = pth_threadid;
	}
	return NULL;
}

T_DECL(pthread_threadid_np, "pthread_threadid_np",
	   T_META_ALL_VALID_ARCHS(YES))
{
	T_LOG("Main Thread");
	do_test(NULL);

	T_LOG("Pthread");
	for (int i = 0; i < 100; i++) {
		uint64_t tid1 = 0, tid2 = 0;
		pthread_t pth;
		T_ASSERT_POSIX_ZERO(pthread_create(&pth, NULL, do_test, &tid1), NULL);
		T_EXPECT_POSIX_ZERO(pthread_threadid_np(pth, &tid2), NULL);
		T_ASSERT_POSIX_ZERO(pthread_join(pth, NULL), NULL);
		T_EXPECT_EQ(tid1, tid2, "parent and child agree");
	}

	T_LOG("Workqueue Thread");
	dispatch_queue_t dq = dispatch_queue_create("myqueue", NULL);
	dispatch_async(dq, ^{ do_test(NULL); });
	dispatch_sync(dq, ^{});

	T_LOG("Workqueue Thread Reuse");
	dispatch_async(dq, ^{ do_test(NULL); });
	dispatch_sync(dq, ^{});
}

T_DECL(pthread_threadid_fork, "pthread_threadid_np post-fork test")
{
	uint64_t tid = __thread_selfid();
	T_ASSERT_NE(tid, (uint64_t)0, "__thread_selfid()");

	uint64_t ptid = 0;
	T_ASSERT_POSIX_ZERO(pthread_threadid_np(NULL, &ptid), NULL);
	T_ASSERT_EQ(ptid, tid, NULL);

	pid_t pid = fork();
	if (pid == 0) {
		// child
		uint64_t ntid = __thread_selfid();
		if (ntid == tid) {
			T_LOG("FAIL: forked child tid is equal to parent tid");
			exit(1);
		}

		uint64_t nptid = 0;
		if (pthread_threadid_np(NULL, &nptid) != 0) {
			T_LOG("FAIL: pthread_threadid_np: %d", errno);
			exit(1);
		}

		if (nptid != ntid) {
			T_LOG("FAIL: pthread_threadid_np == tid (expected: %lld == %lld)",
					nptid, ntid);
			exit(1);
		}
		exit(0);
	} else {
		int status;
		T_ASSERT_EQ(waitpid(pid, &status, 0), pid, NULL);
		int exitstatus = WEXITSTATUS(status);
		T_ASSERT_EQ(exitstatus, 0, NULL);
	}
}
Commit	Line	Data
2546420a A	1	#include <pthread.h>
	2	#include <pthread/private.h>
	3	#include <dispatch/dispatch.h>
c28b7a9d A	4	#include <sys/wait.h>
c28b7a9d A	5	#include <stdlib.h>
2546420a	6
a0619f9c	7	#include "darwintest_defaults.h"
2546420a A	8
	9	extern __uint64_t __thread_selfid( void );
	10
c6e5f90c	11	static void do_test(void arg)
2546420a A	12	{
2546420a A	13	uint64_t threadid = __thread_selfid();
214d78a2	14	T_ASSERT_NE(threadid, (uint64_t)0, "__thread_selfid()");
2546420a A	15
	16	uint64_t pth_threadid = 0;
	17	T_ASSERT_POSIX_ZERO(pthread_threadid_np(NULL, &pth_threadid), NULL);
	18	T_ASSERT_POSIX_ZERO(pthread_threadid_np(pthread_self(), &pth_threadid), NULL);
	19	T_EXPECT_EQ(threadid, pth_threadid, "pthread_threadid_np()");
	20
	21	pth_threadid = _pthread_threadid_self_np_direct();
	22	T_EXPECT_EQ(threadid, pth_threadid, "pthread_threadid_np_direct()");
	23
c6e5f90c A	24	if (arg) {
	25	(uint64_t )arg = pth_threadid;
	26	}
2546420a A	27	return NULL;
	28	}
	29
	30	T_DECL(pthread_threadid_np, "pthread_threadid_np",
	31	T_META_ALL_VALID_ARCHS(YES))
	32	{
	33	T_LOG("Main Thread");
	34	do_test(NULL);
	35
	36	T_LOG("Pthread");
c6e5f90c A	37	for (int i = 0; i < 100; i++) {
	38	uint64_t tid1 = 0, tid2 = 0;
	39	pthread_t pth;
	40	T_ASSERT_POSIX_ZERO(pthread_create(&pth, NULL, do_test, &tid1), NULL);
	41	T_EXPECT_POSIX_ZERO(pthread_threadid_np(pth, &tid2), NULL);
	42	T_ASSERT_POSIX_ZERO(pthread_join(pth, NULL), NULL);
	43	T_EXPECT_EQ(tid1, tid2, "parent and child agree");
	44	}
2546420a A	45
	46	T_LOG("Workqueue Thread");
	47	dispatch_queue_t dq = dispatch_queue_create("myqueue", NULL);
	48	dispatch_async(dq, ^{ do_test(NULL); });
	49	dispatch_sync(dq, ^{});
	50
	51	T_LOG("Workqueue Thread Reuse");
	52	dispatch_async(dq, ^{ do_test(NULL); });
	53	dispatch_sync(dq, ^{});
	54	}
c28b7a9d A	55
	56	T_DECL(pthread_threadid_fork, "pthread_threadid_np post-fork test")
	57	{
	58	uint64_t tid = __thread_selfid();
	59	T_ASSERT_NE(tid, (uint64_t)0, "__thread_selfid()");
	60
	61	uint64_t ptid = 0;
	62	T_ASSERT_POSIX_ZERO(pthread_threadid_np(NULL, &ptid), NULL);
	63	T_ASSERT_EQ(ptid, tid, NULL);
	64
	65	pid_t pid = fork();
	66	if (pid == 0) {
	67	// child
	68	uint64_t ntid = __thread_selfid();
	69	if (ntid == tid) {
	70	T_LOG("FAIL: forked child tid is equal to parent tid");
	71	exit(1);
	72	}
	73
	74	uint64_t nptid = 0;
	75	if (pthread_threadid_np(NULL, &nptid) != 0) {
	76	T_LOG("FAIL: pthread_threadid_np: %d", errno);
	77	exit(1);
	78	}
	79
	80	if (nptid != ntid) {
	81	T_LOG("FAIL: pthread_threadid_np == tid (expected: %lld == %lld)",
	82	nptid, ntid);
	83	exit(1);
	84	}
	85	exit(0);
	86	} else {
	87	int status;
	88	T_ASSERT_EQ(waitpid(pid, &status, 0), pid, NULL);
	89	int exitstatus = WEXITSTATUS(status);
	90	T_ASSERT_EQ(exitstatus, 0, NULL);
	91	}
	92	}