path: root/minix/tests/test89.c

/* Tests for set[ug]id, sete[ug]id, and saved IDs - by D.C. van Moolenbroek */
/* This test must be run as root, as it tests privileged operations. */
#include <stdlib.h>
#include <unistd.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <sys/sysctl.h>
#include <unistd.h>

#include "common.h"

#define ITERATIONS	2

/* These are in a specific order. */
enum {
	SUB_REAL,	/* test set[ug]id(2) */
	SUB_EFF,	/* test sete[ug]id(2) */
	SUB_REAL_E0,	/* test setgid(2) with euid=0 */
	SUB_EFF_E0,	/* test setegid(2) with euid=0 */
	SUB_RETAIN,	/* test r/e/s preservation across fork(2), exec(2) */
};

static const char *executable;

/*
 * The table below is exhaustive in terms of different combinations of real,
 * effective, and saved user IDs (with 0 being a special value, but 1 and 2
 * being interchangeable), but not all these combinations can actually be
 * established in practice.  The results for which there is no way to create
 * the initial condition are set to -1.  If we ever implement setresuid(2),
 * these results can be filled in and tested as well.
 */
static const struct uid_set {
	uid_t ruid;
	uid_t euid;
	uid_t suid;
	uid_t uid;
	int res;
	int eres;
} uid_sets[] = {
	{ 0, 0, 0, 0,  1,  1 },
	{ 0, 0, 0, 1,  1,  1 },
	{ 0, 0, 1, 0,  1,  1 },
	{ 0, 0, 1, 1,  1,  1 },
	{ 0, 0, 1, 2,  1,  1 },
	{ 0, 1, 0, 0,  1,  1 },
	{ 0, 1, 0, 1,  0,  0 },
	{ 0, 1, 0, 2,  0,  0 },
	{ 0, 1, 1, 0,  1,  1 },
	{ 0, 1, 1, 1,  0,  1 },
	{ 0, 1, 1, 2,  0,  0 },
	{ 0, 1, 2, 0, -1, -1 },
	{ 0, 1, 2, 1, -1, -1 },
	{ 0, 1, 2, 2, -1, -1 },
	{ 1, 0, 0, 0,  1,  1 },
	{ 1, 0, 0, 1,  1,  1 },
	{ 1, 0, 0, 2,  1,  1 },
	{ 1, 0, 1, 0, -1, -1 },
	{ 1, 0, 1, 1, -1, -1 },
	{ 1, 0, 1, 2, -1, -1 },
	{ 1, 0, 2, 0, -1, -1 },
	{ 1, 0, 2, 1, -1, -1 },
	{ 1, 0, 2, 2, -1, -1 },
	{ 1, 1, 0, 0,  0,  1 },
	{ 1, 1, 0, 1,  1,  1 },
	{ 1, 1, 0, 2,  0,  0 },
	{ 1, 1, 1, 0,  0,  0 },
	{ 1, 1, 1, 1,  1,  1 },
	{ 1, 1, 1, 2,  0,  0 },
	{ 1, 1, 2, 0,  0,  0 },
	{ 1, 1, 2, 1,  1,  1 },
	{ 1, 1, 2, 2,  0,  1 },
	{ 1, 2, 0, 0,  0,  1 },
	{ 1, 2, 0, 1,  1,  1 },
	{ 1, 2, 0, 2,  0,  0 },
	{ 1, 2, 1, 0, -1, -1 },
	{ 1, 2, 1, 1, -1, -1 },
	{ 1, 2, 1, 2, -1, -1 },
	{ 1, 2, 2, 0,  0,  0 },
	{ 1, 2, 2, 1,  1,  1 },
	{ 1, 2, 2, 2,  0,  1 },
};

/*
 * The same type of table but now for group identifiers.  In this case, all
 * combinations are possible to establish in practice, because the effective
 * UID, not the GID, is used for the privilege check.  GID 0 does not have any
 * special meaning, but we still test it as though it does, in order to ensure
 * that it in fact does not.
 */
static const struct gid_set {
	gid_t rgid;
	gid_t egid;
	gid_t sgid;
	gid_t gid;
	int res;
	int eres;
} gid_sets[] = {
	{ 0, 0, 0, 0, 1, 1 },
	{ 0, 0, 0, 1, 0, 0 },
	{ 0, 0, 1, 0, 1, 1 },
	{ 0, 0, 1, 1, 0, 1 },
	{ 0, 0, 1, 2, 0, 0 },
	{ 0, 1, 0, 0, 1, 1 },
	{ 0, 1, 0, 1, 0, 0 },
	{ 0, 1, 0, 2, 0, 0 },
	{ 0, 1, 1, 0, 1, 1 },
	{ 0, 1, 1, 1, 0, 1 },
	{ 0, 1, 1, 2, 0, 0 },
	{ 0, 1, 2, 0, 1, 1 },
	{ 0, 1, 2, 1, 0, 0 },
	{ 0, 1, 2, 2, 0, 1 },
	{ 1, 0, 0, 0, 0, 1 },
	{ 1, 0, 0, 1, 1, 1 },
	{ 1, 0, 0, 2, 0, 0 },
	{ 1, 0, 1, 0, 0, 0 },
	{ 1, 0, 1, 1, 1, 1 },
	{ 1, 0, 1, 2, 0, 0 },
	{ 1, 0, 2, 0, 0, 0 },
	{ 1, 0, 2, 1, 1, 1 },
	{ 1, 0, 2, 2, 0, 1 },
	{ 1, 1, 0, 0, 0, 1 },
	{ 1, 1, 0, 1, 1, 1 },
	{ 1, 1, 0, 2, 0, 0 },
	{ 1, 1, 1, 0, 0, 0 },
	{ 1, 1, 1, 1, 1, 1 },
	{ 1, 1, 1, 2, 0, 0 },
	{ 1, 1, 2, 0, 0, 0 },
	{ 1, 1, 2, 1, 1, 1 },
	{ 1, 1, 2, 2, 0, 1 },
	{ 1, 2, 0, 0, 0, 1 },
	{ 1, 2, 0, 1, 1, 1 },
	{ 1, 2, 0, 2, 0, 0 },
	{ 1, 2, 1, 0, 0, 0 },
	{ 1, 2, 1, 1, 1, 1 },
	{ 1, 2, 1, 2, 0, 0 },
	{ 1, 2, 2, 0, 0, 0 },
	{ 1, 2, 2, 1, 1, 1 },
	{ 1, 2, 2, 2, 0, 1 },
};

/*
 * Obtain the kinfo_proc2 data for the given process ID.  Return 0 on success,
 * or -1 with errno set appropriately on failure.
 */
static int
get_proc2(pid_t pid, struct kinfo_proc2 * proc2)
{
	int mib[6];
	size_t oldlen;

	/*
	 * FIXME: for performance reasons, the MIB service updates it process
	 * tables only every clock tick.  As a result, we may not be able to
	 * obtain accurate process details right away, and we need to wait.
	 * Eventually, the MIB service should retrieve more targeted subsets of
	 * the process tables, and this problem should go away at least for
	 * specific queries such as this one, which queries only a single PID.
	 */
	usleep((2000000 + sysconf(_SC_CLK_TCK)) / sysconf(_SC_CLK_TCK));

	mib[0] = CTL_KERN;
	mib[1] = KERN_PROC2;
	mib[2] = KERN_PROC_PID;
	mib[3] = pid;
	mib[4] = sizeof(*proc2);
	mib[5] = 1;

	oldlen = sizeof(*proc2);
	if (sysctl(mib, __arraycount(mib), proc2, &oldlen, NULL, 0) == -1)
		return -1;
	if (oldlen != sizeof(*proc2)) {
		errno = ESRCH;
		return -1;
	}
	return 0;
}

/*
 * Verify that the current process's real, effective, and saved user IDs are
 * set to the given respective value.
 */
static void
test_uids(uid_t ruid, uid_t euid, uid_t suid)
{
	struct kinfo_proc2 proc2;

	if (getuid() != ruid) e(0);
	if (geteuid() != euid) e(0);

	/*
	 * There is no system call specifically to retrieve the saved user ID,
	 * so we use sysctl(2) to obtain process information.  This allows us
	 * to verify the real and effective user IDs once more, too.
	 */
	if (get_proc2(getpid(), &proc2) != 0) e(0);

	if (proc2.p_ruid != ruid) e(0);
	if (proc2.p_uid != euid) e(0);
	if (proc2.p_svuid != suid) e(0);
}

/*
 * Verify that the real and effective user IDs are kept as is after an exec(2)
 * call on a non-setuid binary, and that the saved user ID is set to the
 * effective user ID.
 */
static void
exec89b(const char * param1, const char * param2 __unused)
{
	const struct uid_set *set;
	int setnum;

	setnum = atoi(param1);
	if (setnum < 0 || setnum >= __arraycount(uid_sets)) {
		e(setnum);
		return;
	}
	set = &uid_sets[setnum];

	test_uids(set->ruid, set->euid, set->euid);
}

/*
 * The real, effective, and saved user IDs have been set up as indicated by the
 * current set.  Verify that fork(2) and exec(2) do not change the real and
 * effective UIDs, and that only exec(2) sets the saved UID to the effective
 * UID.
 */
static void
sub89b(int setnum)
{
	const struct uid_set *set;
	char param1[32];
	pid_t pid;
	int status;

	set = &uid_sets[setnum];

	pid = fork();

	switch (pid) {
	case -1:
		e(setnum);
		break;

	case 0:
		/*
		 * Verify that all the UIDs were retained across the fork(2)
		 * call.
		 */
		test_uids(set->ruid, set->euid, set->suid);

		snprintf(param1, sizeof(param1), "%d", setnum);

		(void)execl(executable, executable, "DO CHECK", "b", param1,
		    "", NULL);

		e(setnum);
		break;

	default:
		if (waitpid(pid, &status, 0) != pid) e(setnum);
		if (!WIFEXITED(status)) e(setnum);
		if (WEXITSTATUS(status) != 0) e(setnum);
	}
}

/*
 * The real, effective, and saved user IDs have been set up as indicated by the
 * current set.  Test one particular case for test A or B, and verify the
 * result.
 */
static void
test_one_uid(int setnum, int sub)
{
	const struct uid_set *set;
	int res, exp;

	set = &uid_sets[setnum];

	/* Verify that the pre-call process state is as expected. */
	test_uids(set->ruid, set->euid, set->suid);

	/* Perform the call, and check whether the result is as expected. */
	switch (sub) {
	case SUB_REAL:
		res = setuid(set->uid);
		exp = set->res - 1;
		break;

	case SUB_EFF:
		res = seteuid(set->uid);
		exp = set->eres - 1;
		break;

	case SUB_RETAIN:
		sub89b(setnum);

		return;

	default:
		abort();
	}

	if (res != 0 && (res != -1 || errno != EPERM)) e(setnum);

	if (res != exp) e(setnum);

	/* Verify that the post-call process state is as expected as well. */
	if (res == 0) {
		if (sub == SUB_EFF)
			test_uids(set->ruid, set->uid, set->suid);
		else
			test_uids(set->uid, set->uid, set->uid);
	} else
		test_uids(set->ruid, set->euid, set->suid);
}

/*
 * Test setuid(2) or seteuid(2) after a successful execve(2) call, which should
 * have set the process's effective and saved user ID.
 */
static void
exec89a(const char * param1, const char * param2)
{
	const struct uid_set *set;
	int setnum, sub;

	setnum = atoi(param1);
	if (setnum < 0 || setnum >= __arraycount(uid_sets)) {
		e(setnum);
		return;
	}
	set = &uid_sets[setnum];

	sub = atoi(param2);

	if (sub == SUB_RETAIN) {
		/* Clear the set-uid bit before dropping more privileges. */
		if (chmod(executable, S_IXUSR | S_IXGRP | S_IXOTH) != 0)
			e(setnum);
	}

	/* Finish setting up the initial condition. */
	if (set->euid != set->suid) {
		if (set->euid != set->ruid && set->suid != 0) {
			test_uids(set->ruid, set->suid, set->suid);

			return; /* skip test */
		}

		if (seteuid(set->euid) != 0) e(setnum);
	}

	/* Perform the actual test. */
	test_one_uid(setnum, sub);
}

/*
 * Test setuid(2) or seteuid(2) with a certain value starting from a certain
 * initial condition, as identified by the given uid_sets[] array element.  As
 * a side effect, test that in particular exec(2) properly sets the effective
 * and saved user ID.
 */
static void
sub89a(int setnum, int sub)
{
	const struct uid_set *set;
	char param1[32], param2[32];

	set = &uid_sets[setnum];

	/*
	 * Figure out how to set the real, effective, and saved UIDs to those
	 * of the set structure.  Without setresuid(2), not all combinations
	 * are possible to achieve.  We silently skip the tests for which we
	 * cannot create the requested initial condition.
	 */
	if (set->ruid != set->suid) {
		/*
		 * In order to set the saved UID to something other than the
		 * real UID, we must exec(2) a set-uid binary.
		 */
		if (chown(executable, set->suid, 0 /*anything*/) != 0) e(0);
		if (chmod(executable,
		    S_ISUID | S_IXUSR | S_IXGRP | S_IXOTH) != 0) e(0);

		if (setuid(set->ruid) != 0) e(setnum);

		snprintf(param1, sizeof(param1), "%d", setnum);
		snprintf(param2, sizeof(param2), "%d", sub);

		(void)execl(executable, executable, "DO CHECK", "a", param1,
		    param2, NULL);

		e(0);
	} else {
		/*
		 * If the real and saved user ID are to be set to the same
		 * value, we need not use exec(2).  Still, we cannot achieve
		 * all combinations here either.
		 */
		if (set->ruid != 0 && set->ruid != set->euid)
			return; /* skip test */

		if (sub == SUB_RETAIN) {
			/* Clear the set-uid bit before dropping privileges. */
			if (chmod(executable,
			    S_IXUSR | S_IXGRP | S_IXOTH) != 0) e(setnum);
		}

		if (setuid(set->ruid) != 0) e(setnum);
		if (seteuid(set->euid) != 0) e(setnum);

		/* Perform the actual test. */
		test_one_uid(setnum, sub);
	}
}

/*
 * Test setuid(2) and seteuid(2) calls with various initial conditions, by
 * setting the real, effective, and saved UIDs to different values before
 * performing the setuid(2) or seteuid(2) call.
 */
static void
test89a(void)
{
	unsigned int setnum;
	int sub, status;
	pid_t pid;

	subtest = 1;

	for (setnum = 0; setnum < __arraycount(uid_sets); setnum++) {
		for (sub = SUB_REAL; sub <= SUB_EFF; sub++) {
			pid = fork();

			switch (pid) {
			case -1:
				e(setnum);

				break;

			case 0:
				errct = 0;

				sub89a((int)setnum, sub);

				exit(errct);
				/* NOTREACHED */

			default:
				if (waitpid(pid, &status, 0) != pid) e(setnum);
				if (!WIFEXITED(status)) e(setnum);
				if (WEXITSTATUS(status) != 0) e(setnum);
			}
		}
	}
}

/*
 * Ensure that the real, effective, and saved UIDs are fully preserved across
 * fork(2) and non-setuid-binary exec(2) calls.
 */
static void
test89b(void)
{
	unsigned int setnum;
	int status;
	pid_t pid;

	subtest = 2;

	for (setnum = 0; setnum < __arraycount(uid_sets); setnum++) {
		if (uid_sets[setnum].uid != 0)
			continue; /* no need to do the same test >1 times */

		pid = fork();

		switch (pid) {
		case -1:
			e(setnum);

			break;

		case 0:
			errct = 0;

			/*
			 * Test B uses some of the A-test code.  While rather
			 * ugly, this avoids duplication of some of test A's
			 * important UID logic.
			 */
			sub89a((int)setnum, SUB_RETAIN);

			exit(errct);
			/* NOTREACHED */

		default:
			if (waitpid(pid, &status, 0) != pid) e(setnum);
			if (!WIFEXITED(status)) e(setnum);
			if (WEXITSTATUS(status) != 0) e(setnum);
		}
	}
}

/*
 * Verify that the current process's real, effective, and saved group IDs are
 * set to the given respective value.
 */
static void
test_gids(gid_t rgid, gid_t egid, gid_t sgid)
{
	struct kinfo_proc2 proc2;

	if (getgid() != rgid) e(0);
	if (getegid() != egid) e(0);

	/* As above. */
	if (get_proc2(getpid(), &proc2) != 0) e(0);

	if (proc2.p_rgid != rgid) e(0);
	if (proc2.p_gid != egid) e(0);
	if (proc2.p_svgid != sgid) e(0);
}

/*
 * Verify that the real and effective group IDs are kept as is after an exec(2)
 * call on a non-setgid binary, and that the saved group ID is set to the
 * effective group ID.
 */
static void
exec89d(const char * param1, const char * param2 __unused)
{
	const struct gid_set *set;
	int setnum;

	setnum = atoi(param1);
	if (setnum < 0 || setnum >= __arraycount(gid_sets)) {
		e(setnum);
		return;
	}
	set = &gid_sets[setnum];

	test_gids(set->rgid, set->egid, set->egid);
}

/*
 * The real, effective, and saved group IDs have been set up as indicated by
 * the current set.  Verify that fork(2) and exec(2) do not change the real and
 * effective GID, and that only exec(2) sets the saved GID to the effective
 * GID.
 */
static void
sub89d(int setnum)
{
	const struct gid_set *set;
	char param1[32];
	pid_t pid;
	int status;

	set = &gid_sets[setnum];

	pid = fork();

	switch (pid) {
	case -1:
		e(setnum);
		break;

	case 0:
		/*
		 * Verify that all the GIDs were retained across the fork(2)
		 * call.
		 */
		test_gids(set->rgid, set->egid, set->sgid);

		/* Clear the set-gid bit. */
		if (chmod(executable, S_IXUSR | S_IXGRP | S_IXOTH) != 0)
			e(setnum);

		/* Alternate between preserving and dropping user IDs. */
		if (set->gid != 0) {
			if (setuid(3) != 0) e(setnum);
		}

		snprintf(param1, sizeof(param1), "%d", setnum);

		(void)execl(executable, executable, "DO CHECK", "d", param1,
		    "", NULL);

		e(setnum);
		break;

	default:
		if (waitpid(pid, &status, 0) != pid) e(setnum);
		if (!WIFEXITED(status)) e(setnum);
		if (WEXITSTATUS(status) != 0) e(setnum);
	}
}

/*
 * The real, effective, and saved group IDs have been set up as indicated by
 * the current set.  Test one particular case for test C or D, and verify the
 * result.
 */
static void
test_one_gid(int setnum, int sub)
{
	const struct gid_set *set;
	int res, exp;

	set = &gid_sets[setnum];

	/* Verify that the pre-call process state is as expected. */
	test_gids(set->rgid, set->egid, set->sgid);

	/* Perform the call, and check whether the result is as expected. */
	switch (sub) {
	case SUB_REAL:
	case SUB_REAL_E0:
		if (sub != SUB_REAL_E0 && seteuid(1) != 0) e(0);

		res = setgid(set->gid);
		exp = (sub != SUB_REAL_E0) ? (set->res - 1) : 0;
		break;

	case SUB_EFF:
	case SUB_EFF_E0:
		if (sub != SUB_EFF_E0 && seteuid(1) != 0) e(0);

		res = setegid(set->gid);
		exp = (sub != SUB_EFF_E0) ? (set->eres - 1) : 0;
		break;

	case SUB_RETAIN:
		sub89d(setnum);

		return;

	default:
		abort();
	}

	if (res != 0 && (res != -1 || errno != EPERM)) e(setnum);

	if (res != exp) e(setnum);

	/* Verify that the post-call process state is as expected as well. */
	if (res == 0) {
		if (sub == SUB_EFF || sub == SUB_EFF_E0)
			test_gids(set->rgid, set->gid, set->sgid);
		else
			test_gids(set->gid, set->gid, set->gid);
	} else
		test_gids(set->rgid, set->egid, set->sgid);
}

/*
 * Test setgid(2) or setegid(2) after a successful execve(2) call, which should
 * have set the process's effective and saved group ID.
 */
static void
exec89c(const char * param1, const char * param2)
{
	const struct gid_set *set;
	int setnum, sub;

	setnum = atoi(param1);
	if (setnum < 0 || setnum >= __arraycount(gid_sets)) {
		e(setnum);
		return;
	}
	set = &gid_sets[setnum];

	sub = atoi(param2);

	/* Finish setting up the initial condition. */
	if (set->egid != set->sgid && setegid(set->egid) != 0) e(setnum);

	/* Perform the actual test. */
	test_one_gid(setnum, sub);
}

/*
 * Test setgid(2) or setegid(2) with a certain value starting from a certain
 * initial condition, as identified by the given gid_sets[] array element.  As
 * a side effect, test that in particular exec(2) properly sets the effective
 * and saved group ID.
 */
static void
sub89c(int setnum, int sub)
{
	const struct gid_set *set;
	char param1[32], param2[32];

	set = &gid_sets[setnum];

	/*
	 * Figure out how to set the real, effective, and saved GIDs to those
	 * of the set structure.  In this case, all combinations are possible.
	 */
	if (set->rgid != set->sgid) {
		/*
		 * In order to set the saved GID to something other than the
		 * real GID, we must exec(2) a set-gid binary.
		 */
		if (chown(executable, 0 /*anything*/, set->sgid) != 0) e(0);
		if (chmod(executable,
		    S_ISGID | S_IXUSR | S_IXGRP | S_IXOTH) != 0) e(0);

		if (setgid(set->rgid) != 0) e(setnum);

		snprintf(param1, sizeof(param1), "%d", setnum);
		snprintf(param2, sizeof(param2), "%d", sub);

		(void)execl(executable, executable, "DO CHECK", "c", param1,
		    param2, NULL);

		e(0);
	} else {
		/*
		 * If the real and saved group ID are to be set to the same
		 * value, we need not use exec(2).
		 */
		if (setgid(set->rgid) != 0) e(setnum);
		if (setegid(set->egid) != 0) e(setnum);

		/* Perform the actual test. */
		test_one_gid(setnum, sub);
	}
}

/*
 * Test setgid(2) and setegid(2) calls with various initial conditions, by
 * setting the real, effective, and saved GIDs to different values before
 * performing the setgid(2) or setegid(2) call.  At the same time, verify that
 * if the caller has an effective UID of 0, all set(e)gid calls are allowed.
 */
static void
test89c(void)
{
	unsigned int setnum;
	int sub, status;
	pid_t pid;

	subtest = 3;

	for (setnum = 0; setnum < __arraycount(gid_sets); setnum++) {
		for (sub = SUB_REAL; sub <= SUB_EFF_E0; sub++) {
			pid = fork();

			switch (pid) {
			case -1:
				e(setnum);

				break;

			case 0:
				errct = 0;

				sub89c((int)setnum, sub);

				exit(errct);
				/* NOTREACHED */

			default:
				if (waitpid(pid, &status, 0) != pid) e(setnum);
				if (!WIFEXITED(status)) e(setnum);
				if (WEXITSTATUS(status) != 0) e(setnum);
			}
		}
	}
}

/*
 * Ensure that the real, effective, and saved GIDs are fully preserved across
 * fork(2) and non-setgid-binary exec(2) calls.
 */
static void
test89d(void)
{
	unsigned int setnum;
	int status;
	pid_t pid;

	subtest = 4;

	for (setnum = 0; setnum < __arraycount(gid_sets); setnum++) {
		if (gid_sets[setnum].gid == 2)
			continue; /* no need to do the same test >1 times */

		pid = fork();

		switch (pid) {
		case -1:
			e(setnum);

			break;

		case 0:
			errct = 0;

			/* Similarly, test D uses some of the C-test code. */
			sub89c((int)setnum, SUB_RETAIN);

			exit(errct);
			/* NOTREACHED */

		default:
			if (waitpid(pid, &status, 0) != pid) e(setnum);
			if (!WIFEXITED(status)) e(setnum);
			if (WEXITSTATUS(status) != 0) e(setnum);
		}
	}
}

/*
 * Either perform the second step of setting up user and group IDs, or check
 * whether the user and/or group IDs have indeed been changed appropriately as
 * the result of the second exec(2).
 */
static void
exec89e(const char * param1, const char * param2)
{
	int mask, step;
	mode_t mode;

	mask = atoi(param1);
	step = atoi(param2);

	if (step == 0) {
		mode = S_IXUSR | S_IXGRP | S_IXOTH;
		if (mask & 1) mode |= S_ISUID;
		if (mask & 2) mode |= S_ISGID;

		if (chown(executable, 6, 7) != 0) e(0);
		if (chmod(executable, mode) != 0) e(0);

		if (setegid(4) != 0) e(0);
		if (seteuid(2) != 0) e(0);

		test_uids(1, 2, 0);
		test_gids(3, 4, 5);

		(void)execl(executable, executable, "DO CHECK", "e", param1,
		    "1", NULL);

		e(0);
	} else {
		if (mask & 1)
			test_uids(1, 6, 6);
		else
			test_uids(1, 2, 2);

		if (mask & 2)
			test_gids(3, 7, 7);
		else
			test_gids(3, 4, 4);
	}
}

/*
 * Set up for the set-uid/set-gid execution test by initializing to different
 * real and effective user IDs.
 */
static void
sub89e(int mask)
{
	char param1[32];

	if (chown(executable, 0, 5) != 0) e(0);
	if (chmod(executable,
	    S_ISUID | S_ISGID | S_IXUSR | S_IXGRP | S_IXOTH) != 0) e(0);

	if (setgid(3) != 0) e(0);
	if (setuid(1) != 0) e(0);

	snprintf(param1, sizeof(param1), "%d", mask);
	(void)execl(executable, executable, "DO CHECK", "e", param1, "0",
	    NULL);
}

/*
 * Perform basic verification that the set-uid and set-gid bits on binaries are
 * fully independent from each other.
 */
static void
test89e(void)
{
	int mask, status;
	pid_t pid;

	subtest = 5;

	for (mask = 0; mask <= 3; mask++) {
		pid = fork();

		switch (pid) {
		case -1:
			e(0);

			break;

		case 0:
			errct = 0;

			sub89e(mask);

			exit(errct);
			/* NOTREACHED */

		default:
			if (waitpid(pid, &status, 0) != pid) e(mask);
			if (!WIFEXITED(status)) e(mask);
			if (WEXITSTATUS(status) != 0) e(mask);
		}
	}
}

/*
 * Call the right function after having executed myself.
 */
static void
exec89(const char * param0, const char * param1, const char * param2)
{

	switch (param0[0]) {
	case 'a':
		exec89a(param1, param2);
		break;

	case 'b':
		exec89b(param1, param2);
		break;

	case 'c':
		exec89c(param1, param2);
		break;

	case 'd':
		exec89d(param1, param2);
		break;

	case 'e':
		exec89e(param1, param2);
		break;

	default:
		e(0);
	}

	exit(errct);
}

/*
 * Initialize the test.
 */
static void
test89_init(void)
{
	char cp_cmd[PATH_MAX + 9];
	int status;

	subtest = 0;

	/* Reset all user and group IDs to known values. */
	if (setuid(0) != 0) e(0);
	if (setgid(0) != 0) e(0);
	if (setgroups(0, NULL) != 0) e(0);

	test_uids(0, 0, 0);
	test_gids(0, 0, 0);

	/* Make a copy of the binary, which as of start() is one level up. */
	snprintf(cp_cmd, sizeof(cp_cmd), "cp ../%s .", executable);

	status = system(cp_cmd);
	if (status < 0 || !WIFEXITED(status) ||
	    WEXITSTATUS(status) != EXIT_SUCCESS) e(0);
}

/*
 * Test program for set[ug]id, sete[ug]id, and saved IDs.
 */
int
main(int argc, char ** argv)
{
	int i, m;

	executable = argv[0];

	/* This test executes itself.  Handle that case first. */
	if (argc == 5 && !strcmp(argv[1], "DO CHECK"))
		exec89(argv[2], argv[3], argv[4]);

	start(89);

	test89_init();

	if (argc == 2)
		m = atoi(argv[1]);
	else
		m = 0xFF;

	for (i = 0; i < ITERATIONS; i++) {
		if (m & 0x01) test89a();
		if (m & 0x02) test89b();
		if (m & 0x04) test89c();
		if (m & 0x08) test89d();
		if (m & 0x10) test89e();
	}

	quit();
	/* NOTREACHED */
}