#ifndef _XOPEN_SOURCE
#define _XOPEN_SOURCE 700
#endif

#include "sc_mmap.h"

#include <assert.h>
#include <fcntl.h>
#include <stdio.h>

void test1()
{
	int rc;
	unsigned char *p;
	struct sc_mmap mmap;

	rc = sc_mmap_init(&mmap, "x.txt", O_RDWR | O_CREAT | O_TRUNC,
			  PROT_READ | PROT_WRITE, MAP_SHARED, 0, 4095);
	assert(rc == 0);
	rc = sc_mmap_msync(&mmap, 0, 4096);
	assert(rc == 0);
	rc = sc_mmap_term(&mmap);
	assert(rc == 0);
	rc = sc_mmap_term(&mmap);
	assert(rc == 0);

	rc = sc_mmap_init(&mmap, "x.txt", O_RDWR | O_CREAT | O_TRUNC,
			  PROT_READ | PROT_WRITE, MAP_SHARED, 0, 8192);
	assert(rc == 0);
	assert(mmap.len == 8192);
	p = mmap.ptr;
	*p = 'x';
	rc = sc_mmap_msync(&mmap, 0, 4096);
	assert(rc == 0);
	rc = sc_mmap_term(&mmap);
	assert(rc == 0);
	rc = sc_mmap_init(&mmap, "x.txt", O_RDONLY, PROT_READ, MAP_SHARED, 0,
			  0);
	assert(rc == 0);
	p = mmap.ptr;
	assert(*p == 'x');
	rc = sc_mmap_term(&mmap);
	assert(rc == 0);

	rc = sc_mmap_init(&mmap, "x.txt", O_RDWR | O_CREAT | O_TRUNC,
			  PROT_READ | PROT_WRITE, MAP_SHARED, 3, 0);
	assert(rc == -1);

	rc = sc_mmap_init(&mmap, "x.txt", O_RDWR | O_CREAT | O_TRUNC,
			  PROT_READ | PROT_WRITE, MAP_SHARED, 0, 4095);
	assert(rc == 0);
	rc = sc_mmap_mlock(&mmap, 0, 4095);
	assert(rc == 0);
	rc = sc_mmap_munlock(&mmap, 0, 4095);
	assert(rc == 0);
	rc = sc_mmap_term(&mmap);
	assert(rc == 0);
}

#ifdef SC_HAVE_WRAP
#include <errno.h>
#include <stdint.h>
#include <sys/stat.h>

bool fail_open;
extern int __real_open(const char *pathname, int flags, mode_t mode);
int __wrap_open(const char *pathname, int flags, mode_t mode)
{
	if (fail_open) {
		return -1;
	}

	return __real_open(pathname, flags, mode);
}

extern int __real_open64(const char *pathname, int flags, mode_t mode);
int __wrap_open64(const char *pathname, int flags, mode_t mode)
{
	if (fail_open) {
		return -1;
	}

	return __real_open64(pathname, flags, mode);
}

bool fail_stat;
extern int __real_stat(const char *pathname, struct stat *statbuf);
int __wrap_stat(const char *pathname, struct stat *statbuf)
{
	if (fail_stat) {
		return -1;
	}

	return __real_stat(pathname, statbuf);
}

extern int __real_stat64(const char *pathname, struct stat *statbuf);
int __wrap_stat64(const char *pathname, struct stat *statbuf)
{
	if (fail_stat) {
		return -1;
	}

	return __real_stat64(pathname, statbuf);
}

bool fail_mmap;
extern void *__real_mmap(void *addr, size_t length, int prot, int flags, int fd,
			 off_t offset);
void *__wrap_mmap(void *addr, size_t length, int prot, int flags, int fd,
		  off_t offset)
{
	if (fail_mmap) {
		return MAP_FAILED;
	}

	return __real_mmap(addr, length, prot, flags, fd, offset);
}

extern void *__real_mmap64(void *addr, size_t length, int prot, int flags,
			   int fd, off_t offset);
void *__wrap_mmap64(void *addr, size_t length, int prot, int flags, int fd,
		    off_t offset)
{
	if (fail_mmap) {
		return MAP_FAILED;
	}

	return __real_mmap64(addr, length, prot, flags, fd, offset);
}

bool fail_mlock;
extern int __real_mlock(const void *addr, size_t len);
int __wrap_mlock(const void *addr, size_t len)
{
	if (fail_mlock) {
		return -1;
	}

	return __real_mlock(addr, len);
}

bool fail_munlock;
extern int __real_munlock(const void *addr, size_t len);
int __wrap_munlock(const void *addr, size_t len)
{
	if (fail_munlock) {
		return -1;
	}

	return __real_munlock(addr, len);
}

bool fail_msync;
extern int __real_msync(void *addr, size_t len, int flags);
int __wrap_msync(void *addr, size_t len, int flags)
{
	if (fail_msync) {
		return -1;
	}

	return __real_msync(addr, len, flags);
}

bool fail_munmap;
extern int __real_munmap(void *addr, size_t len);
int __wrap_munmap(void *addr, size_t len)
{
	if (fail_munmap) {
		return -1;
	}

	return __real_munmap(addr, len);
}

uint32_t fail_posix_fallocate = UINT32_MAX;
int fail_posix_fallocate_errno = 0;
extern int __real_posix_fallocate(int fd, off_t offset, off_t len);
int __wrap_posix_fallocate(int fd, off_t offset, off_t len)
{
	fail_posix_fallocate--;
	if (fail_posix_fallocate == 0) {
		errno = fail_posix_fallocate_errno;
		return fail_posix_fallocate_errno;
	}

	return __real_posix_fallocate(fd, offset, len);
}

extern int __real_posix_fallocate64(int fd, off_t offset, off_t len);
int __wrap_posix_fallocate64(int fd, off_t offset, off_t len)
{
	fail_posix_fallocate--;
	if (fail_posix_fallocate == 0) {
		errno = fail_posix_fallocate_errno;
		return fail_posix_fallocate_errno;
	}

	return __real_posix_fallocate64(fd, offset, len);
}

void fail_test()
{
	int rc;
	struct sc_mmap mmap;

	fail_open = true;
	rc = sc_mmap_init(&mmap, "x.txt", O_RDWR | O_CREAT | O_TRUNC,
			  PROT_READ | PROT_WRITE, MAP_SHARED, 0, 4095);
	assert(rc == -1);
	assert(sc_mmap_err(&mmap) != NULL);
	fail_open = false;

	mmap = (struct sc_mmap){0};
	fail_stat = true;
	rc = sc_mmap_init(&mmap, "x.txt", O_RDWR | O_CREAT | O_TRUNC,
			  PROT_READ | PROT_WRITE, MAP_SHARED, 0, 4095);
	assert(rc == -1);
	assert(sc_mmap_err(&mmap) != NULL);
	fail_stat = false;

	mmap = (struct sc_mmap){0};
	fail_mmap = true;
	rc = sc_mmap_init(&mmap, "x.txt", O_RDWR | O_CREAT | O_TRUNC,
			  PROT_READ | PROT_WRITE, MAP_SHARED, 0, 4095);
	assert(rc == -1);
	assert(sc_mmap_err(&mmap) != NULL);
	fail_mmap = false;

	mmap = (struct sc_mmap){0};
	fail_munmap = true;
	rc = sc_mmap_term(&mmap);
	assert(rc == -1);
	assert(sc_mmap_err(&mmap) != NULL);
	fail_munmap = false;

	rc = sc_mmap_init(&mmap, "x.txt", O_RDWR | O_CREAT | O_TRUNC,
			  PROT_READ | PROT_WRITE, MAP_SHARED, 0, 4095);
	assert(rc == 0);
	fail_mlock = true;
	rc = sc_mmap_mlock(&mmap, 0, 4096);
	assert(rc == -1);
	assert(sc_mmap_err(&mmap) != NULL);
	fail_mlock = false;

	rc = sc_mmap_mlock(&mmap, 0, 4096);
	assert(rc == 0);

	fail_munlock = true;
	rc = sc_mmap_munlock(&mmap, 0, 4096);
	assert(rc == -1);
	fail_munlock = false;
	rc = sc_mmap_munlock(&mmap, 0, 4096);
	assert(rc == 0);

	fail_msync = true;
	rc = sc_mmap_msync(&mmap, 0, 4096);
	assert(rc == -1);
	fail_msync = false;
	rc = sc_mmap_msync(&mmap, 0, 4096);
	assert(rc == 0);

	rc = sc_mmap_term(&mmap);
	assert(rc == 0);

	fail_posix_fallocate = 1;
	fail_posix_fallocate_errno = EINTR;
	rc = sc_mmap_init(&mmap, "x.txt", O_RDWR | O_CREAT | O_TRUNC,
			  PROT_READ | PROT_WRITE, MAP_SHARED, 0, 4095);
	assert(rc == 0);
	assert(sc_mmap_term(&mmap) == 0);

	fail_posix_fallocate = 1;
	fail_posix_fallocate_errno = -1;
	rc = sc_mmap_init(&mmap, "x.txt", O_RDWR | O_CREAT | O_TRUNC,
			  PROT_READ | PROT_WRITE, MAP_SHARED, 0, 4095);
	assert(rc != 0);

	fail_posix_fallocate = UINT32_MAX;
}
#else
void fail_test()
{
}
#endif

int main()
{
	test1();
	fail_test();

	return 0;
}