#include <stdio.h>

#include <string.h>

#include <sys/mman.h>

#include <assert.h>

#include "log.h"

#include "rwlock.h"

#include "os/os.h"

void fio_rwlock_write(struct fio_rwlock *lock)

{

	assert(lock->magic == FIO_RWLOCK_MAGIC);

	pthread_rwlock_wrlock(&lock->lock);

}

void fio_rwlock_read(struct fio_rwlock *lock)

{

	assert(lock->magic == FIO_RWLOCK_MAGIC);

	pthread_rwlock_rdlock(&lock->lock);

}

void fio_rwlock_unlock(struct fio_rwlock *lock)

{

	assert(lock->magic == FIO_RWLOCK_MAGIC);

	pthread_rwlock_unlock(&lock->lock);

}

void fio_rwlock_remove(struct fio_rwlock *lock)

{

	assert(lock->magic == FIO_RWLOCK_MAGIC);

	pthread_rwlock_destroy(&lock->lock);

	munmap((void *) lock, sizeof(*lock));

}

struct fio_rwlock *fio_rwlock_init(void)

{

	struct fio_rwlock *lock;

	pthread_rwlockattr_t attr;

	int ret;

	lock = (void *) mmap(NULL, sizeof(struct fio_rwlock),

				PROT_READ | PROT_WRITE,

				OS_MAP_ANON | MAP_SHARED, -1, 0);

	if (lock == MAP_FAILED) {

		perror("mmap rwlock");

		lock = NULL;

		goto err;

	}

	lock->magic = FIO_RWLOCK_MAGIC;

	ret = pthread_rwlockattr_init(&attr);

	if (ret) {

		log_err("pthread_rwlockattr_init: %s\n", strerror(ret));

		goto err;

	}

#ifdef CONFIG_PSHARED

	ret = pthread_rwlockattr_setpshared(&attr, PTHREAD_PROCESS_SHARED);

	if (ret) {

		log_err("pthread_rwlockattr_setpshared: %s\n", strerror(ret));

		goto destroy_attr;

	}

	ret = pthread_rwlock_init(&lock->lock, &attr);

#else

	ret = pthread_rwlock_init(&lock->lock, NULL);

#endif

	if (ret) {

		log_err("pthread_rwlock_init: %s\n", strerror(ret));

		goto destroy_attr;

	}

	pthread_rwlockattr_destroy(&attr);

	return lock;

destroy_attr:

	pthread_rwlockattr_destroy(&attr);

err:

	if (lock)

		fio_rwlock_remove(lock);

	return NULL;

}