// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (c) Linux Test Project, 2003-2021
 * Copyright (c) 2014 Cyril Hrubis <chrubis@suse.cz>
 * Copyright (c) International Business Machines  Corp., 2001
 * 07/2001 Ported by Wayne Boyer
 */

/*\
 * [Description]
 *
 * Verify that fchown(2) invoked by super-user:
 *  - clears setuid and setgid bits set on an executable file
 *  - preserves setgid bit set on a non-group-executable file
 */

#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include "tst_test.h"
#include "compat_tst_16.h"
#include "tst_safe_macros.h"

#define FILE_MODE	(S_IFREG|S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH)
#define NEW_PERMS1	(S_IFREG|S_IRWXU|S_IRWXG|S_ISUID|S_ISGID)
#define NEW_PERMS2	(S_IFREG|S_IRWXU|S_ISGID)
#define EXP_PERMS	(S_IFREG|S_IRWXU|S_IRWXG)
#define TESTFILE1	"testfile1"
#define TESTFILE2	"testfile2"

static int fd1, fd2;

struct test_case_t {
	int *fd;
	const char *filename;
	mode_t set_mode;
	mode_t exp_mode;
} tc[] = {
	{&fd1, TESTFILE1, NEW_PERMS1, EXP_PERMS},
	{&fd2, TESTFILE2, NEW_PERMS2, NEW_PERMS2}
};

static void run(unsigned int i)
{
	struct stat stat_buf;
	uid_t uid;
	gid_t gid;

	UID16_CHECK((uid = geteuid()), "fchown");
	GID16_CHECK((gid = getegid()), "fchown");

	SAFE_CHMOD(tc[i].filename, tc[i].set_mode);

	TST_EXP_PASS(FCHOWN(*tc[i].fd, uid, gid),
		"fchown(%i, %i, %i)", *tc[i].fd, uid, gid);

	SAFE_STAT(tc[i].filename, &stat_buf);

	if (stat_buf.st_uid != uid || stat_buf.st_gid != gid)
		tst_res(TFAIL, "%s: owner set to (uid=%d, gid=%d), expected (uid=%d, gid=%d)",
			tc[i].filename, stat_buf.st_uid, stat_buf.st_gid, uid, gid);

	if (stat_buf.st_mode != tc[i].exp_mode)
		tst_res(TFAIL, "%s: wrong mode permissions %#o, expected %#o",
			tc[i].filename, stat_buf.st_mode, tc[i].exp_mode);
}

static void setup(void)
{
	unsigned int i;

	for (i = 0; i < ARRAY_SIZE(tc); i++)
		*tc[i].fd = SAFE_OPEN(tc[i].filename, O_RDWR | O_CREAT, FILE_MODE);
}

static void cleanup(void)
{
	unsigned int i;

	for (i = 0; i < ARRAY_SIZE(tc); i++) {
		if (*tc[i].fd > 0)
			SAFE_CLOSE(*tc[i].fd);
	}
}

static struct tst_test test = {
	.tcnt = ARRAY_SIZE(tc),
	.needs_root = 1,
	.needs_tmpdir = 1,
	.setup = setup,
	.cleanup = cleanup,
	.test = run,
};