darling-libpthread/tests/wq_limits.c

#include <sys/sysctl.h>
#include <dispatch/dispatch.h>
#include <dispatch/private.h>
#include "darwintest_defaults.h"

T_DECL(wq_pool_limits, "test overcommit limit")
{
	dispatch_semaphore_t sema = dispatch_semaphore_create(0);
	dispatch_group_t g = dispatch_group_create();
	dispatch_time_t t;
	uint32_t wq_max_threads, wq_max_constrained_threads;

	dispatch_block_t b = ^{
		dispatch_group_leave(g);
		dispatch_semaphore_wait(sema, DISPATCH_TIME_FOREVER);
	};

	size_t s = sizeof(uint32_t);
	sysctlbyname("kern.wq_max_threads", &wq_max_threads, &s, NULL, 0);
	sysctlbyname("kern.wq_max_constrained_threads", &wq_max_constrained_threads,
				 &s, NULL, 0);

	for (uint32_t i = 0; i < wq_max_constrained_threads; i++) {
		dispatch_group_enter(g);
		dispatch_async(dispatch_get_global_queue(QOS_CLASS_DEFAULT, 0), b);
	}

	t = dispatch_time(DISPATCH_TIME_NOW, 10 * NSEC_PER_SEC);
	T_ASSERT_EQ(dispatch_group_wait(g, t), 0L,
			"%d constrained threads bringup", wq_max_constrained_threads);

	dispatch_async(dispatch_get_global_queue(QOS_CLASS_DEFAULT, 0), ^{
		T_ASSERT_FAIL("Should never run");
	});

	sleep(5);
	T_PASS("constrained limit looks fine");

	for (uint32_t i = wq_max_constrained_threads; i < wq_max_threads; i++) {
		dispatch_group_enter(g);
		dispatch_async(dispatch_get_global_queue(QOS_CLASS_DEFAULT,
						DISPATCH_QUEUE_OVERCOMMIT), b);
	}
	t = dispatch_time(DISPATCH_TIME_NOW, 10 * NSEC_PER_SEC);
	T_ASSERT_EQ(dispatch_group_wait(g, t), 0L,
			"%d threads bringup", wq_max_threads);


	dispatch_async(dispatch_get_global_queue(QOS_CLASS_DEFAULT,
					DISPATCH_QUEUE_OVERCOMMIT), ^{
		T_ASSERT_FAIL("Should never run");
	});

	sleep(5);
	T_PASS("thread limit looks fine");
	T_END;
}