#include <pthread.h>
#include <string.h>
#include <errno.h>
#include "test.h"

#define TEST(r, f, m) ( \
	((r) = (f)) == 0 || (error("%s failed: %s (" m ")\n", #f, strerror(r)), 0) )
#define TESTX(r, f, x, m) ( \
	((r) = (f)) == (x) || (error("%s failed: got %d \"%s\" want %d \"%s\" (" m ")\n", #f, r, strerror(r), x, strerror(x)), 0) )

static void *start_lock(void *arg)
{
	pthread_mutex_lock(arg);
	return 0;
}

static void *start_wait(void *arg)
{
	void **args = arg;
	pthread_mutex_lock(args[1]);
	pthread_barrier_wait(args[0]);
	nanosleep(&(struct timespec){ .tv_nsec = 10000000 }, 0);
	return 0;
}

int main(void)
{
	pthread_t td;
	int r;
	void *res;
	pthread_barrier_t barrier2;
	pthread_mutexattr_t mtx_a;
	pthread_mutex_t mtx;

	TEST(r, pthread_barrier_init(&barrier2, 0, 2), "creating barrier");

	/* Robust mutexes */
	TEST(r, pthread_mutexattr_init(&mtx_a), "initializing mutex attr");
	TEST(r, pthread_mutexattr_setrobust(&mtx_a, PTHREAD_MUTEX_ROBUST), "setting robust attribute");
	TEST(r, pthread_mutex_init(&mtx, &mtx_a), "initializing robust mutex");
	TEST(r, pthread_mutex_lock(&mtx), "locking robust mutex");
	TEST(r, pthread_mutex_unlock(&mtx), "unlocking robust mutex");
	TEST(r, pthread_create(&td, 0, start_lock, &mtx), "failed to create thread");
	TEST(r, pthread_join(td, &res), "joining thread");
	TESTX(r, pthread_mutex_lock(&mtx), EOWNERDEAD, "locking orphaned robust mutex");
	TEST(r, pthread_mutex_unlock(&mtx), "unlocking orphaned robust mutex");
	TESTX(r, pthread_mutex_lock(&mtx), ENOTRECOVERABLE, "re-locking orphaned robust mutex");
	TEST(r, pthread_mutex_destroy(&mtx), "destroying unrecoverable mutex");

	TEST(r, pthread_mutex_init(&mtx, &mtx_a), "initializing robust mutex");
	TEST(r, pthread_create(&td, 0, start_lock, &mtx), "failed to create thread");
	TEST(r, pthread_join(td, &res), "joining thread");
	TESTX(r, pthread_mutex_lock(&mtx), EOWNERDEAD, "locking orphaned robust mutex");
	TEST(r, pthread_mutex_consistent(&mtx), "");
	TEST(r, pthread_mutex_unlock(&mtx), "unlocking orphaned robust mutex");
	TEST(r, pthread_mutex_lock(&mtx), "re-locking orphaned robust mutex");
	TEST(r, pthread_mutex_destroy(&mtx), "destroying mutex");

	TEST(r, pthread_mutex_init(&mtx, &mtx_a), "");
	TEST(r, pthread_create(&td, 0, start_wait, (void *[]){ &barrier2, &mtx }), "");
	r = pthread_barrier_wait(&barrier2);
	if (r && r != PTHREAD_BARRIER_SERIAL_THREAD)
		error("pthread_barrier_wait failed: got %d \"%s\", wanted either 0 or %d\n",
			r, strerror(r), PTHREAD_BARRIER_SERIAL_THREAD);
	TESTX(r, pthread_mutex_lock(&mtx), EOWNERDEAD, "");
	TEST(r, pthread_join(td, &res), "");
	TEST(r, pthread_mutex_consistent(&mtx), "");
	TEST(r, pthread_mutex_unlock(&mtx), "");
	TEST(r, pthread_mutex_destroy(&mtx), "");

	return test_status;
}