/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright (c) 2019 by Tomohiro Kusumi. All rights reserved.
 */

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

static void
test_stat_mode(mode_t extra)
{
	struct stat st;
	int i, fd;
	char fpath[1024];
	char *penv[] = {"TESTDIR", "TESTFILE0"};
	char buf[] = "test";
	mode_t res;
	mode_t mode = 0777 | extra;

	/*
	 * Get the environment variable values.
	 */
	for (i = 0; i < sizeof (penv) / sizeof (char *); i++) {
		if ((penv[i] = getenv(penv[i])) == NULL) {
			fprintf(stderr, "getenv(penv[%d])\n", i);
			exit(1);
		}
	}

	umask(0);
	if (stat(penv[0], &st) == -1 && mkdir(penv[0], mode) == -1) {
		perror("mkdir");
		exit(2);
	}

	snprintf(fpath, sizeof (fpath), "%s/%s", penv[0], penv[1]);
	unlink(fpath);
	if (stat(fpath, &st) == 0) {
		fprintf(stderr, "%s exists\n", fpath);
		exit(3);
	}

	fd = creat(fpath, mode);
	if (fd == -1) {
		perror("creat");
		exit(4);
	}
	close(fd);

	if (setuid(65534) == -1) {
		perror("setuid");
		exit(5);
	}

	fd = open(fpath, O_RDWR);
	if (fd == -1) {
		perror("open");
		exit(6);
	}

	if (write(fd, buf, sizeof (buf)) == -1) {
		perror("write");
		exit(7);
	}
	close(fd);

	if (stat(fpath, &st) == -1) {
		perror("stat");
		exit(8);
	}
	unlink(fpath);

	/* Verify SUID/SGID are dropped */
	res = st.st_mode & (0777 | S_ISUID | S_ISGID);
	if (res != (mode & 0777)) {
		fprintf(stderr, "stat(2) %o\n", res);
		exit(9);
	}
}

int
main(int argc, char *argv[])
{
	const char *name;
	mode_t extra;

	if (argc < 2) {
		fprintf(stderr, "Invalid argc\n");
		exit(1);
	}

	name = argv[1];
	if (strcmp(name, "SUID") == 0) {
		extra = S_ISUID;
	} else if (strcmp(name, "SGID") == 0) {
		extra = S_ISGID;
	} else if (strcmp(name, "SUID_SGID") == 0) {
		extra = S_ISUID | S_ISGID;
	} else if (strcmp(name, "NONE") == 0) {
		extra = 0;
	} else {
		fprintf(stderr, "Invalid name %s\n", name);
		exit(1);
	}

	test_stat_mode(extra);

	return (0);
}