1/*
2 * Automated Testing Framework (atf)
3 *
4 * Copyright (c) 2007, 2008 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND
17 * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
18 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
19 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY
21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
23 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
30#include <sys/param.h>
31#include <sys/types.h>
32#include <limits.h>
33#include <unistd.h>
34
35#include "sanity.h"
36#include "user.h"
37
38/* ---------------------------------------------------------------------
39 * Free functions.
40 * --------------------------------------------------------------------- */
41
42uid_t
43atf_user_euid(void)
44{
45    return geteuid();
46}
47
48bool
49atf_user_is_member_of_group(gid_t gid)
50{
51    static gid_t groups[NGROUPS_MAX];
52    static int ngroups = -1;
53    bool found;
54    int i;
55
56    if (ngroups == -1) {
57        ngroups = getgroups(NGROUPS_MAX, groups);
58        INV(ngroups >= 0);
59    }
60
61    found = false;
62    for (i = 0; !found && i < ngroups; i++)
63        if (groups[i] == gid)
64            found = true;
65    return found;
66}
67
68bool
69atf_user_is_root(void)
70{
71    return geteuid() == 0;
72}
73
74bool
75atf_user_is_unprivileged(void)
76{
77    return geteuid() != 0;
78}
79