// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (c) Wipro Technologies Ltd, 2002.  All Rights Reserved.
 * Copyright (c) Linux Test Project, 2002-2018
 * Author: Vatsal Avasthi
 */

/*\
 * Verify flock(2) returns -1 and set proper errno:
 *
 * - EBADF if the file descriptor is invalid
 * - EINVAL if the argument operation does not include LOCK_SH,LOCK_EX,LOCK_UN
 * - EINVAL if an invalid combination of locking modes is used i.e LOCK_SH with LOCK_EX
 * - EWOULDBLOCK if the file is locked and the LOCK_NB flag was selected
 */

#include <errno.h>
#include <sys/file.h>

#include "tst_test.h"

static int badfd = -1;
static int fd;

static struct tcase {
	int *fd;
	int operation;
	int exp_err;
} tcases[] = {
	{&badfd, LOCK_SH, EBADF},
	{&fd, LOCK_NB, EINVAL},
	{&fd, LOCK_SH | LOCK_EX, EINVAL},
	{&fd, LOCK_NB | LOCK_EX, EWOULDBLOCK}
};

static void verify_flock(unsigned int n)
{
	int fd2 = -1;
	struct tcase *tc = &tcases[n];

	fd = SAFE_OPEN("testfile", O_RDWR);

	if (tc->exp_err == EWOULDBLOCK) {
		fd2 = SAFE_OPEN("testfile", O_RDWR);
		flock(fd2, LOCK_EX);
	}

	TEST(flock(*tc->fd, tc->operation));
	if (TST_RET == 0) {
		tst_res(TFAIL | TTERRNO, "flock() succeeded unexpectedly");
		SAFE_CLOSE(fd);
		return;
	}

	if (tc->exp_err == TST_ERR) {
		tst_res(TPASS | TTERRNO, "flock() failed expectedly");
	} else {
		tst_res(TFAIL | TTERRNO, "flock() failed unexpectedly, "
			"expected %s", tst_strerrno(tc->exp_err));
	}

	SAFE_CLOSE(fd);
	if (fd2 != -1)
		SAFE_CLOSE(fd2);
}

static void setup(void)
{
	int fd1;

	fd1 = SAFE_OPEN("testfile", O_CREAT | O_TRUNC | O_RDWR, 0666);
	SAFE_CLOSE(fd1);
}

static struct tst_test test = {
	.tcnt = ARRAY_SIZE(tcases),
	.test = verify_flock,
	.needs_tmpdir = 1,
	.setup = setup,
};