stack_size.c   [plain text]

#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include "darwintest_defaults.h"

#if defined(__arm64__)
#define PTHREAD_T_OFFSET (12*1024)
#else
#define PTHREAD_T_OFFSET (0)
#endif

static void *
function(void *arg)
{
	size_t expected_size = (size_t)(uintptr_t)arg;
	T_ASSERT_EQ(pthread_get_stacksize_np(pthread_self()), expected_size,
			"saw expected pthread_get_stacksize_np");
	return NULL;
}

T_DECL(stack_size_default, "stack size of default pthread",
		T_META_ALL_VALID_ARCHS(YES))
{
	static const size_t dflsize = 512 * 1024;
	pthread_t thread;
	pthread_attr_t attr;

	T_ASSERT_POSIX_ZERO(pthread_attr_init(&attr), NULL);
	T_ASSERT_POSIX_ZERO(pthread_create(&thread, &attr, function,
			(void *)(dflsize + PTHREAD_T_OFFSET)), NULL);
	T_ASSERT_POSIX_ZERO(pthread_join(thread, NULL), NULL);
}

T_DECL(stack_size_customsize, "stack size of thread with custom stack size",
		T_META_ALL_VALID_ARCHS(YES))
{
	static const size_t stksize = 768 * 1024;
	pthread_t thread;
	pthread_attr_t attr;

	T_ASSERT_POSIX_ZERO(pthread_attr_init(&attr), NULL);
	T_ASSERT_POSIX_ZERO(pthread_attr_setstacksize(&attr, stksize), NULL);
	T_ASSERT_POSIX_ZERO(pthread_create(&thread, &attr, function,
			(void *)(stksize + PTHREAD_T_OFFSET)), NULL);
	T_ASSERT_POSIX_ZERO(pthread_join(thread, NULL), NULL);
}

T_DECL(stack_size_customaddr, "stack size of thread with custom stack addr",
		T_META_ALL_VALID_ARCHS(YES))
{
	static const size_t stksize = 512 * 1024;
	pthread_t thread;
	pthread_attr_t attr;

	uintptr_t stackaddr = (uintptr_t)valloc(stksize);
	stackaddr += stksize; // address is top of stack

	T_ASSERT_POSIX_ZERO(pthread_attr_init(&attr), NULL);
	T_ASSERT_POSIX_ZERO(pthread_attr_setstackaddr(&attr, (void *)stackaddr),
			NULL);
	T_ASSERT_POSIX_ZERO(pthread_create(&thread, &attr, function,
			(void *)stksize), NULL);
	T_ASSERT_POSIX_ZERO(pthread_join(thread, NULL), NULL);
	free((void *)(stackaddr - stksize));
}

T_DECL(stack_size_custom, "stack size of thread with custom stack addr+size",
		T_META_ALL_VALID_ARCHS(YES))
{
	static const size_t stksize = 768 * 1024;
	pthread_t thread;
	pthread_attr_t attr;

	uintptr_t stackaddr = (uintptr_t)valloc(stksize);
	stackaddr += stksize; // address is top of stack

	T_ASSERT_POSIX_ZERO(pthread_attr_init(&attr), NULL);
	T_ASSERT_POSIX_ZERO(pthread_attr_setstackaddr(&attr, (void *)stackaddr),
			NULL);
	T_ASSERT_POSIX_ZERO(pthread_attr_setstacksize(&attr, stksize), NULL);
	T_ASSERT_POSIX_ZERO(pthread_create(&thread, &attr, function,
			(void *)stksize), NULL);
	T_ASSERT_POSIX_ZERO(pthread_join(thread, NULL), NULL);
	free((void *)(stackaddr - stksize));
}