/* testing pthread mutex behaviour with various attributes */
#include <pthread.h>
#include <semaphore.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include "test.h"

#define T(f) if ((r=(f))) t_error(#f " failed: %s\n", strerror(r))
#define E(f) if (f) t_error(#f " failed: %s\n", strerror(errno))

static void *relock(void *arg)
{
	void **a = arg;
	int r;

	T(pthread_mutex_lock(a[0]));
	E(sem_post(a[1]));
	*(int*)a[2] = pthread_mutex_lock(a[0]);
	E(sem_post(a[1]));

	T(pthread_mutex_unlock(a[0]));
	if (*(int*)a[2] == 0)
		T(pthread_mutex_unlock(a[0]));
	return 0;
}

static int test_relock(int mtype)
{
	struct timespec ts;
	pthread_t t;
	pthread_mutex_t m;
	pthread_mutexattr_t ma;
	sem_t s;
	int i;
	int r;
	void *p;
	void *a[] = {&m,&s,&i};

	T(pthread_mutexattr_init(&ma));
	T(pthread_mutexattr_settype(&ma, mtype));
	T(pthread_mutexattr_setprotocol(&ma, PTHREAD_PRIO_INHERIT));
	T(pthread_mutex_init(a[0], &ma));
	T(pthread_mutexattr_destroy(&ma));
	E(sem_init(a[1], 0, 0));
	T(pthread_create(&t, 0, relock, a));
	E(sem_wait(a[1]));
	E(clock_gettime(CLOCK_REALTIME, &ts));
	ts.tv_nsec += 100*1000*1000;
	if (ts.tv_nsec >= 1000*1000*1000) {
		ts.tv_nsec -= 1000*1000*1000;
		ts.tv_sec += 1;
	}
	r = sem_timedwait(a[1],&ts);
	if (r == -1) {
		if (errno != ETIMEDOUT)
			t_error("sem_timedwait failed with unexpected error: %s\n", strerror(errno));
		/* leave the deadlocked relock thread running */
		return -1;
	}
	T(pthread_join(t, &p));
	T(pthread_mutex_destroy(a[0]));
	E(sem_destroy(a[1]));
	return i;
}

static void *unlock(void *arg)
{
	void **a = arg;

	*(int*)a[1] = pthread_mutex_unlock(a[0]);
	return 0;
}

static int test_unlock(int mtype)
{
	pthread_t t;
	pthread_mutex_t m;
	pthread_mutexattr_t ma;
	int i;
	int r;
	void *p;
	void *a[] = {&m,&i};

	T(pthread_mutexattr_init(&ma));
	T(pthread_mutexattr_settype(&ma, mtype));
	T(pthread_mutexattr_setprotocol(&ma, PTHREAD_PRIO_INHERIT));
	T(pthread_mutex_init(a[0], &ma));
	T(pthread_mutexattr_destroy(&ma));
	T(pthread_create(&t, 0, unlock, a));
	T(pthread_join(t, &p));
	T(pthread_mutex_destroy(a[0]));
	return i;
}

static int test_unlock_other(int mtype)
{
	pthread_t t;
	pthread_mutex_t m;
	pthread_mutexattr_t ma;
	int i;
	int r;
	void *p;
	void *a[] = {&m,&i};

	T(pthread_mutexattr_init(&ma));
	T(pthread_mutexattr_settype(&ma, mtype));
	T(pthread_mutexattr_setprotocol(&ma, PTHREAD_PRIO_INHERIT));
	T(pthread_mutex_init(a[0], &ma));
	T(pthread_mutexattr_destroy(&ma));
	T(pthread_mutex_lock(a[0]));
	T(pthread_create(&t, 0, unlock, a));
	T(pthread_join(t, &p));
	T(pthread_mutex_unlock(a[0]));
	T(pthread_mutex_destroy(a[0]));
	return i;
}

static void test_mutexattr()
{
	pthread_mutex_t m;
	pthread_mutexattr_t a;
	int r;
	int i;

	T(pthread_mutexattr_init(&a));
	T(pthread_mutexattr_gettype(&a, &i));
	if (i != PTHREAD_MUTEX_DEFAULT)
		t_error("default mutex type is %d, wanted PTHREAD_MUTEX_DEFAULT (%d)\n", i, PTHREAD_MUTEX_DEFAULT);
	T(pthread_mutexattr_settype(&a, PTHREAD_MUTEX_ERRORCHECK));
	T(pthread_mutexattr_gettype(&a, &i));
	if (i != PTHREAD_MUTEX_ERRORCHECK)
		t_error("setting error check mutex type failed failed: got %d, wanted %d\n", i, PTHREAD_MUTEX_ERRORCHECK);
	T(pthread_mutexattr_destroy(&a));
}

int main(void)
{
	int i;

	test_mutexattr();

	i = test_relock(PTHREAD_MUTEX_NORMAL);
	if (i != -1)
		t_error("PTHREAD_MUTEX_NORMAL relock did not deadlock, got %s\n", strerror(i));
	i = test_relock(PTHREAD_MUTEX_ERRORCHECK);
	if (i != EDEADLK)
		t_error("PTHREAD_MUTEX_ERRORCHECK relock did not return EDEADLK, got %s\n", i==-1?"deadlock":strerror(i));
	i = test_relock(PTHREAD_MUTEX_RECURSIVE);
	if (i != 0)
		t_error("PTHREAD_MUTEX_RECURSIVE relock did not succed, got %s\n", i==-1?"deadlock":strerror(i));

	i = test_unlock(PTHREAD_MUTEX_ERRORCHECK);
	if (i != EPERM)
		t_error("PTHREAD_MUTEX_ERRORCHECK unlock did not return EPERM, got %s\n", strerror(i));
	i = test_unlock(PTHREAD_MUTEX_RECURSIVE);
	if (i != EPERM)
		t_error("PTHREAD_MUTEX_RECURSIVE unlock did not return EPERM, got %s\n", strerror(i));

	i = test_unlock_other(PTHREAD_MUTEX_ERRORCHECK);
	if (i != EPERM)
		t_error("PTHREAD_MUTEX_ERRORCHECK unlock did not return EPERM, got %s\n", strerror(i));
	i = test_unlock_other(PTHREAD_MUTEX_RECURSIVE);
	if (i != EPERM)
		t_error("PTHREAD_MUTEX_RECURSIVE unlock did not return EPERM, got %s\n", strerror(i));

	return t_status;
}