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

#include <stdio.h>
#include <stdlib.h>
#include <dispatch/dispatch.h>
#include <dispatch/private.h>
#include <sys/time.h>
#include <sys/sysctl.h>

#include "darwintest_defaults.h"
#include <darwintest_utils.h>

static uint64_t end_spin;

static uint32_t
get_ncpu(void)
{
	static uint32_t activecpu;
	if (!activecpu) {
		uint32_t n;
		size_t s = sizeof(activecpu);
		sysctlbyname("hw.activecpu", &n, &s, NULL, 0);
		activecpu = n;
	}
	return activecpu;
}

static void
spin_and_pause(void *ctx)
{
	long i = (long)ctx;

	printf("Thread %ld starts\n", i);

	while (clock_gettime_nsec_np(CLOCK_MONOTONIC) < end_spin) {
#if defined(__x86_64__) || defined(__i386__)
		__asm__("pause");
#elif defined(__arm__) || defined(__arm64__)
		__asm__("wfe");
#endif
	}
	printf("Thread %ld blocks\n", i);
	pause();
}

static void
spin(void *ctx)
{
	long i = (long)ctx;

	printf("Thread %ld starts\n", i);

	while (clock_gettime_nsec_np(CLOCK_MONOTONIC)) {
#if defined(__x86_64__) || defined(__i386__)
		__asm__("pause");
#elif defined(__arm__) || defined(__arm64__)
		__asm__("wfe");
#endif
	}
}

T_DECL(thread_request_32848402, "repro for rdar://32848402")
{
	dispatch_queue_attr_t bg_attr, in_attr;

	bg_attr = dispatch_queue_attr_make_with_qos_class(DISPATCH_QUEUE_CONCURRENT,
			QOS_CLASS_BACKGROUND, 0);
	in_attr = dispatch_queue_attr_make_with_qos_class(DISPATCH_QUEUE_CONCURRENT,
			QOS_CLASS_USER_INITIATED, 0);

	dispatch_queue_t a = dispatch_queue_create_with_target("in", in_attr, NULL);
	dispatch_queue_t b = dispatch_queue_create_with_target("bg", bg_attr, NULL);

	end_spin = clock_gettime_nsec_np(CLOCK_MONOTONIC) + 2 * NSEC_PER_SEC;

	dispatch_async_f(a, (void *)0, spin_and_pause);
	long n_threads = MIN((long)get_ncpu(),
			pthread_qos_max_parallelism(QOS_CLASS_BACKGROUND, 0));
	for (long i = 1; i < n_threads; i++) {
		dispatch_async_f(b, (void *)i, spin);
	}

	dispatch_async(b, ^{
		T_PASS("The NCPU+1-nth block got scheduled");
		T_END;
	});

	sleep(10);
	T_FAIL("The NCPU+1-nth block didn't get scheduled");
}
Commit	Line	Data
a0619f9c A	1	#include <stdio.h>
	2	#include <stdlib.h>
	3	#include <dispatch/dispatch.h>
	4	#include <dispatch/private.h>
	5	#include <sys/time.h>
	6	#include <sys/sysctl.h>
	7
	8	#include "darwintest_defaults.h"
	9	#include <darwintest_utils.h>
	10
	11	static uint64_t end_spin;
	12
	13	static uint32_t
	14	get_ncpu(void)
	15	{
	16	static uint32_t activecpu;
	17	if (!activecpu) {
	18	uint32_t n;
	19	size_t s = sizeof(activecpu);
	20	sysctlbyname("hw.activecpu", &n, &s, NULL, 0);
	21	activecpu = n;
	22	}
	23	return activecpu;
	24	}
	25
	26	static void
	27	spin_and_pause(void *ctx)
	28	{
	29	long i = (long)ctx;
	30
	31	printf("Thread %ld starts\n", i);
	32
	33	while (clock_gettime_nsec_np(CLOCK_MONOTONIC) < end_spin) {
	34	#if defined(__x86_64__) \|\| defined(__i386__)
	35	__asm__("pause");
	36	#elif defined(__arm__) \|\| defined(__arm64__)
	37	__asm__("wfe");
	38	#endif
	39	}
	40	printf("Thread %ld blocks\n", i);
	41	pause();
	42	}
	43
	44	static void
	45	spin(void *ctx)
	46	{
	47	long i = (long)ctx;
	48
	49	printf("Thread %ld starts\n", i);
	50
	51	while (clock_gettime_nsec_np(CLOCK_MONOTONIC)) {
	52	#if defined(__x86_64__) \|\| defined(__i386__)
	53	__asm__("pause");
	54	#elif defined(__arm__) \|\| defined(__arm64__)
	55	__asm__("wfe");
	56	#endif
	57	}
	58	}
	59
	60	T_DECL(thread_request_32848402, "repro for rdar://32848402")
	61	{
	62	dispatch_queue_attr_t bg_attr, in_attr;
	63
	64	bg_attr = dispatch_queue_attr_make_with_qos_class(DISPATCH_QUEUE_CONCURRENT,
65	QOS_CLASS_BACKGROUND, 0);
66	in_attr = dispatch_queue_attr_make_with_qos_class(DISPATCH_QUEUE_CONCURRENT,
67	QOS_CLASS_USER_INITIATED, 0);
68
69	dispatch_queue_t a = dispatch_queue_create_with_target("in", in_attr, NULL);
70	dispatch_queue_t b = dispatch_queue_create_with_target("bg", bg_attr, NULL);
71
72	end_spin = clock_gettime_nsec_np(CLOCK_MONOTONIC) + 2 * NSEC_PER_SEC;
73
74	dispatch_async_f(a, (void *)0, spin_and_pause);
214d78a2 A	75	long n_threads = MIN((long)get_ncpu(),
	76	pthread_qos_max_parallelism(QOS_CLASS_BACKGROUND, 0));
	77	for (long i = 1; i < n_threads; i++) {
a0619f9c A	78	dispatch_async_f(b, (void *)i, spin);
	79	}
	80
	81	dispatch_async(b, ^{
	82	T_PASS("The NCPU+1-nth block got scheduled");
	83	T_END;
	84	});
	85
	86	sleep(10);
	87	T_FAIL("The NCPU+1-nth block didn't get scheduled");
	88	}