#define _GNU_SOURCE
#include <sched.h>

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <signal.h>
#include <errno.h>

#include "zdtmtst.h"

const char *test_doc="Tests IPC semaphores migrates fine";
const char *test_author="Stanislav Kinsbursky <skinsbursky@parallels.com>";

static int sem_test(int id,
		    struct sembuf *lock, struct sembuf *unlock,
		    int lock_ops, int unlock_ops)
{
	if (semop(id, lock, lock_ops) == -1) {
		fail("Failed to lock semaphore");
		return -errno;
	}
	if (semop(id, unlock, unlock_ops) == -1) {
		fail("Failed to unlock semaphore");
		return -errno;
	}
	return 0;
}

static int check_sem_by_key(int key)
{
	int id;
	struct sembuf lock[2] = {
		{
		.sem_num = 0,
		.sem_op = 0,
		.sem_flg = 0,
		},
		{
		.sem_num = 0,
		.sem_op = 1,
		.sem_flg = 0,
		},
	};
	struct sembuf unlock[1] = {
		{
		.sem_num = 0,
		.sem_op = -1,
		.sem_flg = 0,
		}
	};
	int val;

	id = semget(key, 1, 0777);
	if (id  == -1) {
		fail("Can't get sem");
		return -errno;
	}

	val = semctl(id, 0, GETVAL);
	if (val < 0) {
		fail("Failed to get sem value");
		return -errno;
	}

	return sem_test(id, lock, unlock,
			sizeof(lock)/sizeof(struct sembuf),
			sizeof(unlock)/sizeof(struct sembuf));
}

static int check_sem_by_id(int id, int val)
{
	int curr;
	struct sembuf lock[] = {
		{
		.sem_num = 0,
		.sem_op = val,
		.sem_flg = 0,
		},
	};
	struct sembuf unlock[] = {
		{
		.sem_num = 0,
		.sem_op = - val * 2,
		.sem_flg = 0,
		}
	};

	curr = semctl(id, 0, GETVAL);
	if (curr < 0) {
		fail("Failed to get sem value");
		return -errno;
	}
	if (curr != val) {
		fail("Sem has wrong value: %d instead of %d\n", curr, val);
		return -EFAULT;
	}
	return sem_test(id, lock, unlock,
			sizeof(lock)/sizeof(struct sembuf),
			sizeof(unlock)/sizeof(struct sembuf));
}

static int test_fn(int argc, char **argv)
{
	int id, key, val;
	int ret, fail_count = 0;

	key = ftok(argv[0], 89063453);
	if (key == -1) {
		pr_perror("Can't make key");
		return -1;
	}

	val = lrand48() & 0x7;

	id = semget(key, 1, 0777 | IPC_CREAT | IPC_EXCL);
	if (id  == -1) {
		fail_count++;
		pr_perror("Can't get sem");
		goto out;
	}
	if (semctl(id, 0, SETVAL, val) == -1) {
		fail_count++;
		pr_perror("Can't init sem");
		goto out_destroy;
	}

	test_daemon();
	test_waitsig();

	ret = check_sem_by_id(id, val);
	if (ret < 0) {
		fail_count++;
		fail("Check sem by id failed");
		goto out_destroy;
	}

	if (check_sem_by_key(key) < 0) {
		fail("Check sem by key failed");
		fail_count++;
		goto out_destroy;
	}

	val = semctl(id, 0, GETVAL);
	if (val < 0) {
		fail("Failed to get sem value");
		fail_count++;
		goto out_destroy;
	}
	if (val != 0) {
		fail("Non-zero sem value: %d", val);
		fail_count++;
	}

out_destroy:
	ret = semctl(id, 1, IPC_RMID);
	if (ret < 0) {
		fail("Destroy sem failed");
		fail_count++;
	}
out:
	if (fail_count == 0)
		pass();
	return fail_count;
}

int main(int argc, char **argv)
{
#ifdef NEW_IPC_NS
	test_init_ns(argc, argv, CLONE_NEWIPC, test_fn);
#else
	test_init(argc, argv);
	test_fn(argc, argv);
#endif
	return 0;
}
