1240116Smarcel/*
2240116Smarcel * Automated Testing Framework (atf)
3240116Smarcel *
4240116Smarcel * Copyright (c) 2007 The NetBSD Foundation, Inc.
5240116Smarcel * All rights reserved.
6240116Smarcel *
7240116Smarcel * Redistribution and use in source and binary forms, with or without
8240116Smarcel * modification, are permitted provided that the following conditions
9240116Smarcel * are met:
10240116Smarcel * 1. Redistributions of source code must retain the above copyright
11240116Smarcel *    notice, this list of conditions and the following disclaimer.
12240116Smarcel * 2. Redistributions in binary form must reproduce the above copyright
13240116Smarcel *    notice, this list of conditions and the following disclaimer in the
14240116Smarcel *    documentation and/or other materials provided with the distribution.
15240116Smarcel *
16240116Smarcel * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND
17240116Smarcel * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
18240116Smarcel * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
19240116Smarcel * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20240116Smarcel * IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY
21240116Smarcel * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22240116Smarcel * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
23240116Smarcel * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24240116Smarcel * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25240116Smarcel * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26240116Smarcel * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27240116Smarcel * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28240116Smarcel */
29240116Smarcel
30240116Smarcel#include <sys/param.h>
31240116Smarcel#include <sys/types.h>
32240116Smarcel#include <limits.h>
33240116Smarcel#include <unistd.h>
34240116Smarcel
35240116Smarcel#include "sanity.h"
36240116Smarcel#include "user.h"
37240116Smarcel
38240116Smarcel/* ---------------------------------------------------------------------
39240116Smarcel * Free functions.
40240116Smarcel * --------------------------------------------------------------------- */
41240116Smarcel
42240116Smarceluid_t
43240116Smarcelatf_user_euid(void)
44240116Smarcel{
45240116Smarcel    return geteuid();
46240116Smarcel}
47240116Smarcel
48240116Smarcelbool
49240116Smarcelatf_user_is_member_of_group(gid_t gid)
50240116Smarcel{
51240116Smarcel    static gid_t groups[NGROUPS_MAX];
52240116Smarcel    static int ngroups = -1;
53240116Smarcel    bool found;
54240116Smarcel    int i;
55240116Smarcel
56240116Smarcel    if (ngroups == -1) {
57240116Smarcel        ngroups = getgroups(NGROUPS_MAX, groups);
58240116Smarcel        INV(ngroups >= 0);
59240116Smarcel    }
60240116Smarcel
61240116Smarcel    found = false;
62240116Smarcel    for (i = 0; !found && i < ngroups; i++)
63240116Smarcel        if (groups[i] == gid)
64240116Smarcel            found = true;
65240116Smarcel    return found;
66240116Smarcel}
67240116Smarcel
68240116Smarcelbool
69240116Smarcelatf_user_is_root(void)
70240116Smarcel{
71240116Smarcel    return geteuid() == 0;
72240116Smarcel}
73240116Smarcel
74240116Smarcelbool
75240116Smarcelatf_user_is_unprivileged(void)
76240116Smarcel{
77240116Smarcel    return geteuid() != 0;
78240116Smarcel}
79