1162271Srwatson/*-
2162271Srwatson * Copyright (c) 2006 nCircle Network Security, Inc.
3172106Srwatson * Copyright (c) 2007 Robert N. M. Watson
4162271Srwatson * All rights reserved.
5162271Srwatson *
6162271Srwatson * This software was developed by Robert N. M. Watson for the TrustedBSD
7162271Srwatson * Project under contract to nCircle Network Security, Inc.
8162271Srwatson *
9162271Srwatson * Redistribution and use in source and binary forms, with or without
10162271Srwatson * modification, are permitted provided that the following conditions
11162271Srwatson * are met:
12162271Srwatson * 1. Redistributions of source code must retain the above copyright
13162271Srwatson *    notice, this list of conditions and the following disclaimer.
14162271Srwatson * 2. Redistributions in binary form must reproduce the above copyright
15162271Srwatson *    notice, this list of conditions and the following disclaimer in the
16162271Srwatson *    documentation and/or other materials provided with the distribution.
17162271Srwatson *
18162271Srwatson * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19162271Srwatson * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20162271Srwatson * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21162271Srwatson * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR, NCIRCLE NETWORK SECURITY,
22162271Srwatson * INC., OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23162271Srwatson * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
24162271Srwatson * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
25162271Srwatson * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
26162271Srwatson * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
27162271Srwatson * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
28162271Srwatson * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29162271Srwatson *
30162271Srwatson * $FreeBSD$
31162271Srwatson */
32162271Srwatson
33162271Srwatson/*
34172106Srwatson * Test privileges for setlogin(); first query with getlogin() so that the
35172106Srwatson * result is a no-op, since it affects the entire login session.
36162271Srwatson */
37162271Srwatson
38172106Srwatson#include <sys/param.h>
39172106Srwatson
40162271Srwatson#include <err.h>
41162271Srwatson#include <errno.h>
42162271Srwatson#include <unistd.h>
43162271Srwatson
44162271Srwatson#include "main.h"
45162271Srwatson
46172106Srwatsonstatic int initialized;
47172106Srwatsonstatic char *loginname;
48172106Srwatson
49172106Srwatsonint
50172106Srwatsonpriv_proc_setlogin_setup(int asroot, int injail, struct test *test)
51172106Srwatson{
52172106Srwatson
53172106Srwatson	if (initialized)
54172106Srwatson		return (0);
55172106Srwatson	loginname = getlogin();
56172106Srwatson	if (loginname == NULL) {
57172106Srwatson		warn("priv_proc_setlogin_setup: getlogin");
58172106Srwatson		return (-1);
59172106Srwatson	}
60172106Srwatson	initialized = 1;
61172106Srwatson	return (0);
62172106Srwatson}
63172106Srwatson
64162271Srwatsonvoid
65172106Srwatsonpriv_proc_setlogin(int asroot, int injail, struct test *test)
66162271Srwatson{
67162271Srwatson	int error;
68162271Srwatson
69162271Srwatson	error = setlogin(loginname);
70172106Srwatson	if (asroot && injail)
71172106Srwatson		expect("priv_proc_setlogin(asroot, injail)", error, 0, 0);
72172106Srwatson	if (asroot && !injail)
73172106Srwatson		expect("priv_proc_setlogin(asroot, !injail)", error, 0, 0);
74172106Srwatson	if (!asroot && injail)
75172106Srwatson		expect("priv_proc_setlogin(!sroot, injail)", error, -1,
76172106Srwatson		    EPERM);
77172106Srwatson	if (!asroot && !injail)
78172106Srwatson		expect("priv_proc_setlogin(!asroot, !injail)", error, -1,
79172106Srwatson		    EPERM);
80172106Srwatson}
81162271Srwatson
82172106Srwatsonvoid
83172106Srwatsonpriv_proc_setlogin_cleanup(int asroot, int injail, struct test *test)
84172106Srwatson{
85162271Srwatson
86172106Srwatson	if (initialized)
87172106Srwatson		(void)setlogin(loginname);
88162271Srwatson}
89