#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <sys/mman.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <sys/wait.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <xpmem.h>
#include "util.h"

#define DEBUG

#define SHM_SIZE (1UL << 12)

#define MAP_HUGE_SHIFT 26
#define KEYWORD 0x12345678UL

void usage(void)
{
	printf("Usage: multi_vmr_xpmem: <pgshift> <pgnum>\n");
	printf("\tpgshift : pageshift of map area (Using MAP_HUGETLB)\n");
	printf("\t            -1 means using small pagesize\n");
	printf("\tpgnum   : number of page of map area\n");
}


void *mmap_flag(size_t mapsize, int page_shift)
{
	char *addr_mmap;
	int flags = MAP_ANONYMOUS | MAP_PRIVATE;

	if (page_shift >= 0) {
		/* mean use MAP_HUGETLB */
		flags |= MAP_HUGETLB | (page_shift << MAP_HUGE_SHIFT);
	}

	addr_mmap = mmap(0, mapsize * 2,
			PROT_READ | PROT_WRITE,
			flags, -1, 0);

	/* Make sure that area before addr_map is available to
	 * MAP_FIXED map
	 */
	return addr_mmap + mapsize;
}

int main(int argc, char **argv)
{
	void *mem, *mem_1, *mem_2;
	int ret = 0;
	pid_t pid;
	int status;
	key_t key = ftok(argv[0], 10);
	void *shm;
	int shmid;
	xpmem_segid_t segid;
	struct shmid_ds shmctl_buf;
	int pgshift, pgnum;
	size_t extr_size, pgsize, map_size;

	if (argc < 3) {
		printf("Err: Too few arguments\n");
		usage();
		return -1;
	}

	pgshift = atoi(argv[1]);
	pgnum = atoi(argv[2]);
	extr_size = getpagesize() * 3;
	if (pgshift > 0) {
		pgsize = (1UL << pgshift);
	} else {
		pgsize = getpagesize();
	}

	map_size = pgsize * pgnum;

	shmid = shmget(key, SHM_SIZE, IPC_CREAT | 0660);
	CHKANDJUMP(shmid == -1, EXIT_FAILURE, "shmget failed: %s\n",
		   strerror(errno));

	printf("EXPECT_PAGE_SIZE: 0x%lx\n", pgsize);
	fflush(stdout);

	pid = fork();
	CHKANDJUMP(pid == -1, EXIT_FAILURE, "fork failed\n");
	if (pid == 0) {
		xpmem_apid_t apid;
		struct xpmem_addr addr;
		void *attach;

		shm = shmat(shmid, NULL, 0);
		CHKANDJUMP(shm == (void *)-1, EXIT_FAILURE,
			   "shmat failed: %s\n", strerror(errno));

		while ((segid = *(xpmem_segid_t *)shm) == 0) {
		};
		ret = shmdt(shm);
		CHKANDJUMP(ret == -1, EXIT_FAILURE, "shmdt failed\n");

		apid = xpmem_get(segid, XPMEM_RDWR,
				 XPMEM_PERMIT_MODE, NULL);
		CHKANDJUMP(apid == -1, EXIT_FAILURE, "xpmem_get failed: %s\n",
			   strerror(errno));

		addr.apid = apid;
		addr.offset = 0;
		printf("child: attaching...\n");
		attach = xpmem_attach(addr, map_size + (extr_size * 2), NULL);

		CHKANDJUMP(attach == (void *)-1, EXIT_FAILURE,
			   "xpmem_attach failed: %s\n", strerror(errno));

		printf("child: xpmem_attachment_addr: %lx - %lx\n",
		       attach, attach + map_size + (extr_size * 2));
		printf("child: xpmem_large: %lx\n", attach + extr_size);

		*((unsigned long *)attach) = KEYWORD;
		*((unsigned long *)(attach + extr_size)) = KEYWORD;
		*((unsigned long *)(attach + extr_size * 2 + map_size
			- sizeof(unsigned long *))) = KEYWORD;

		ret = xpmem_detach(attach);
		CHKANDJUMP(ret == -1, EXIT_FAILURE, "xpmem_detach failed\n");

		exit(0);
	} else {
		mem = mmap_flag(map_size, pgshift);
		CHKANDJUMP(mem == MAP_FAILED, EXIT_FAILURE, "mmap failed\n");
		mem_1 = mmap(mem - extr_size, extr_size,
				PROT_READ | PROT_WRITE,
				MAP_ANONYMOUS | MAP_PRIVATE | MAP_FIXED,
				-1, 0);
		mem_2 = mmap(mem + map_size, extr_size,
				PROT_READ | PROT_WRITE,
				MAP_ANONYMOUS | MAP_PRIVATE | MAP_FIXED,
				 -1, 0);
		if ((mem_1 + extr_size != mem) || (mem_2 != mem + map_size)) {
			printf("vm_range is NOT contignuous!!\n");
			exit(1);
		}
		printf("parent: anonymous_map_addr: %lx - %lx\n",
		       mem_1, mem_2 + extr_size);

		shm = shmat(shmid, NULL, 0);

		CHKANDJUMP(shm == (void *)-1, EXIT_FAILURE,
			   "shmat failed: %s\n", strerror(errno));

		printf("parent: making...\n");
		segid = xpmem_make(mem_1, map_size + (extr_size * 2),
				XPMEM_PERMIT_MODE, (void *)0666);
		CHKANDJUMP(segid == -1, EXIT_FAILURE,
			   "xpmem_ioctl failed: %s\n", strerror(errno));

		*(xpmem_segid_t *)shm = segid;

		printf("parent: waiting...\n");
		ret = waitpid(pid, &status, 0);
printf("child exited\n");
		CHKANDJUMP(ret == -1, EXIT_FAILURE, "waitpid failed\n");

		NG(*(unsigned long *)mem_1 == KEYWORD,
			"HEAD of xpmem area is INVALID. isn't shared?\n");
		NG(*(unsigned long *)mem == KEYWORD,
			"MIDDLE of xpmem area is INVALID. isn't shared?\n");
		NG(*((unsigned long *)(mem_2 + extr_size
				- sizeof(unsigned long *))) == KEYWORD,
			"TAIL of xpmem area is INVALID. isn't shared?\n");
		printf("xpmem area is shared: OK\n");

		ret = shmctl(shmid, IPC_RMID, &shmctl_buf);
		CHKANDJUMP(ret == -1, EXIT_FAILURE, "shmctl failed\n");

		ret = shmdt(shm);
		CHKANDJUMP(ret == -1, EXIT_FAILURE, "shmdt failed\n");

		ret = xpmem_remove(segid);
		CHKANDJUMP(ret == -1, EXIT_FAILURE, "xpmem_remove failed\n");
	}

	ret = 0;
 out:
	return ret;
}