/*	$OpenBSD: setuid_regress.h,v 1.1 2014/08/27 07:36:14 blambert Exp $	*/
/*
 *	Written by Bret Stephen Lambert <blambert@openbsd.org> 2014
 *	Public Domain.
 */

#ifndef	_SETUID_REGRESS_H_
#define	_SETUID_REGRESS_H_

#define	_SETUID_REGRESS_USER	"nobody"

static inline int
read_kproc_pid(struct kinfo_proc *kproc, pid_t pid)
{
	int			 args[6];
	size_t			 size;

	args[0] = CTL_KERN;
	args[1] = KERN_PROC;
	args[2] = KERN_PROC_PID;
	args[3] = pid;
	args[4] = sizeof(*kproc);
	args[5] = 1;

	size = sizeof(*kproc);
	return (sysctl(args, 6, kproc, &size, NULL, 0));
}

static inline void
checkuids(uid_t truid, uid_t teuid, uid_t tsuid, const char *str)
{
	uid_t			 ruid, euid, suid;

	if (getresuid(&ruid, &euid, &suid) == -1)
		err(1, "getresuid %s", str);

	if (ruid != truid)
		errx(1, "real uid incorrectly set %s: is %u should be %u",
		    str, ruid, truid);
	if (euid != teuid)
		errx(1, "effective uid incorrectly set %s: is %u should be %u",
		    str, euid, teuid);
	if (suid != tsuid)
		errx(1, "saved uid incorrectly set %s: is %u should be %u",
		    str, suid, tsuid);
}

void
checkgids(gid_t trgid, gid_t tegid, gid_t tsgid, const char *str)
{
	gid_t			rgid, egid, sgid;

	if (getresgid(&rgid, &egid, &sgid) == -1)
		err(1, "getresgid %s", str);

	if (rgid != trgid)
		errx(1, "real gid incorrectly set %s: is %u should be %u",
		    str, rgid, trgid);
	if (egid != tegid)
		errx(1, "effective gid incorrectly set %s: is %u should be %u",
		    str, egid, tegid);
	if (sgid != tsgid)
		errx(1, "saved gid incorrectly set %s: is %u should be %u",
		    str, sgid, tsgid);
}
#endif