xref: /xnu-10002.1.13/tests/workq/workq_sigprof.c (revision 1031c584a5e37aff177559b9f69dbd3c8c3fd30a)

#include <pthread.h>
#include <stdbool.h>
#include <signal.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <sys/time.h>
#include <mach/mach_time.h>
#include <dispatch/dispatch.h>

#include <darwintest.h>

T_GLOBAL_META(
	T_META_NAMESPACE("xnu.workq"),
	T_META_RADAR_COMPONENT_NAME("xnu"),
	T_META_RADAR_COMPONENT_VERSION("workq"),
	T_META_RUN_CONCURRENTLY(true));


static pthread_t workq_thread;
static bool signal_received;

static void
signal_handler(int sig __unused, siginfo_t *b __unused, void* unused __unused)
{
	if (pthread_self() == workq_thread) {
		signal_received = true;
	}
}

static void
workq_block(void *unused __unused)
{
	workq_thread = pthread_self();

	/*
	 *  sigset_t set;
	 *  sigemptyset(&set);
	 *  sigaddset(&set, SIGPROF);
	 *  pthread_sigmask(SIG_UNBLOCK, &set, NULL);
	 */

	uint64_t spin_start = mach_absolute_time();
	while (mach_absolute_time() - spin_start < 30 * NSEC_PER_SEC) {
		if (signal_received) {
			T_PASS("Got SIGPROF!");
			T_END;
		}
	}
}

T_DECL(workq_sigprof, "test that workqueue threads can receive sigprof")
{
	struct sigaction sa = {
		.sa_sigaction = signal_handler
	};
	sigfillset(&sa.sa_mask);
	T_ASSERT_POSIX_ZERO(sigaction(SIGPROF, &sa, NULL), NULL);

	dispatch_queue_t q = dispatch_get_global_queue(0, 0);
	dispatch_async_f(q, NULL, workq_block);

	struct itimerval timerval = {
		.it_interval = {.tv_usec = 10000},
		.it_value = {.tv_usec = 10000}
	};
	T_ASSERT_POSIX_ZERO(setitimer(ITIMER_PROF, &timerval, NULL), NULL);

	dispatch_main();
}