1*57718be8SEnji Cooper /* $NetBSD: t_getgroups.c,v 1.1 2011/07/07 06:57:53 jruoho Exp $ */
2*57718be8SEnji Cooper
3*57718be8SEnji Cooper /*-
4*57718be8SEnji Cooper * Copyright (c) 2011 The NetBSD Foundation, Inc.
5*57718be8SEnji Cooper * All rights reserved.
6*57718be8SEnji Cooper *
7*57718be8SEnji Cooper * This code is derived from software contributed to The NetBSD Foundation
8*57718be8SEnji Cooper * by Jukka Ruohonen.
9*57718be8SEnji Cooper *
10*57718be8SEnji Cooper * Redistribution and use in source and binary forms, with or without
11*57718be8SEnji Cooper * modification, are permitted provided that the following conditions
12*57718be8SEnji Cooper * are met:
13*57718be8SEnji Cooper * 1. Redistributions of source code must retain the above copyright
14*57718be8SEnji Cooper * notice, this list of conditions and the following disclaimer.
15*57718be8SEnji Cooper * 2. Redistributions in binary form must reproduce the above copyright
16*57718be8SEnji Cooper * notice, this list of conditions and the following disclaimer in the
17*57718be8SEnji Cooper * documentation and/or other materials provided with the distribution.
18*57718be8SEnji Cooper *
19*57718be8SEnji Cooper * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20*57718be8SEnji Cooper * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21*57718be8SEnji Cooper * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22*57718be8SEnji Cooper * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23*57718be8SEnji Cooper * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24*57718be8SEnji Cooper * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25*57718be8SEnji Cooper * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26*57718be8SEnji Cooper * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27*57718be8SEnji Cooper * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28*57718be8SEnji Cooper * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29*57718be8SEnji Cooper * POSSIBILITY OF SUCH DAMAGE.
30*57718be8SEnji Cooper */
31*57718be8SEnji Cooper #include <sys/cdefs.h>
32*57718be8SEnji Cooper __RCSID("$NetBSD: t_getgroups.c,v 1.1 2011/07/07 06:57:53 jruoho Exp $");
33*57718be8SEnji Cooper
34*57718be8SEnji Cooper #include <sys/wait.h>
35*57718be8SEnji Cooper
36*57718be8SEnji Cooper #include <atf-c.h>
37*57718be8SEnji Cooper #include <errno.h>
38*57718be8SEnji Cooper #include <limits.h>
39*57718be8SEnji Cooper #include <stdlib.h>
40*57718be8SEnji Cooper #include <string.h>
41*57718be8SEnji Cooper #include <unistd.h>
42*57718be8SEnji Cooper
43*57718be8SEnji Cooper ATF_TC(getgroups_err);
ATF_TC_HEAD(getgroups_err,tc)44*57718be8SEnji Cooper ATF_TC_HEAD(getgroups_err, tc)
45*57718be8SEnji Cooper {
46*57718be8SEnji Cooper atf_tc_set_md_var(tc, "descr", "Test errors in getgroups(2)");
47*57718be8SEnji Cooper }
48*57718be8SEnji Cooper
ATF_TC_BODY(getgroups_err,tc)49*57718be8SEnji Cooper ATF_TC_BODY(getgroups_err, tc)
50*57718be8SEnji Cooper {
51*57718be8SEnji Cooper gid_t gidset[NGROUPS_MAX];
52*57718be8SEnji Cooper
53*57718be8SEnji Cooper errno = 0;
54*57718be8SEnji Cooper
55*57718be8SEnji Cooper ATF_REQUIRE(getgroups(10, (gid_t *)-1) == -1);
56*57718be8SEnji Cooper ATF_REQUIRE(errno == EFAULT);
57*57718be8SEnji Cooper
58*57718be8SEnji Cooper errno = 0;
59*57718be8SEnji Cooper
60*57718be8SEnji Cooper ATF_REQUIRE(getgroups(-1, gidset) == -1);
61*57718be8SEnji Cooper ATF_REQUIRE(errno == EINVAL);
62*57718be8SEnji Cooper }
63*57718be8SEnji Cooper
64*57718be8SEnji Cooper ATF_TC(getgroups_getgid);
ATF_TC_HEAD(getgroups_getgid,tc)65*57718be8SEnji Cooper ATF_TC_HEAD(getgroups_getgid, tc)
66*57718be8SEnji Cooper {
67*57718be8SEnji Cooper atf_tc_set_md_var(tc, "descr", "Test getgid(2) from getgroups(2)");
68*57718be8SEnji Cooper }
69*57718be8SEnji Cooper
ATF_TC_BODY(getgroups_getgid,tc)70*57718be8SEnji Cooper ATF_TC_BODY(getgroups_getgid, tc)
71*57718be8SEnji Cooper {
72*57718be8SEnji Cooper gid_t gidset[NGROUPS_MAX];
73*57718be8SEnji Cooper gid_t gid = getgid();
74*57718be8SEnji Cooper int i, n;
75*57718be8SEnji Cooper
76*57718be8SEnji Cooper /*
77*57718be8SEnji Cooper * Check that getgid(2) is found from
78*57718be8SEnji Cooper * the GIDs returned by getgroups(2).
79*57718be8SEnji Cooper */
80*57718be8SEnji Cooper n = getgroups(NGROUPS_MAX, gidset);
81*57718be8SEnji Cooper
82*57718be8SEnji Cooper for (i = 0; i < n; i++) {
83*57718be8SEnji Cooper
84*57718be8SEnji Cooper if (gidset[i] == gid)
85*57718be8SEnji Cooper return;
86*57718be8SEnji Cooper }
87*57718be8SEnji Cooper
88*57718be8SEnji Cooper atf_tc_fail("getgid(2) not found from getgroups(2)");
89*57718be8SEnji Cooper }
90*57718be8SEnji Cooper
91*57718be8SEnji Cooper ATF_TC(getgroups_setgid);
ATF_TC_HEAD(getgroups_setgid,tc)92*57718be8SEnji Cooper ATF_TC_HEAD(getgroups_setgid, tc)
93*57718be8SEnji Cooper {
94*57718be8SEnji Cooper atf_tc_set_md_var(tc, "descr", "Test setgid(2) from getgroups(2)");
95*57718be8SEnji Cooper atf_tc_set_md_var(tc, "require.user", "root");
96*57718be8SEnji Cooper }
97*57718be8SEnji Cooper
ATF_TC_BODY(getgroups_setgid,tc)98*57718be8SEnji Cooper ATF_TC_BODY(getgroups_setgid, tc)
99*57718be8SEnji Cooper {
100*57718be8SEnji Cooper gid_t gidset[NGROUPS_MAX];
101*57718be8SEnji Cooper int i, n, rv, sta;
102*57718be8SEnji Cooper pid_t pid;
103*57718be8SEnji Cooper
104*57718be8SEnji Cooper /*
105*57718be8SEnji Cooper * Check that we can setgid(2)
106*57718be8SEnji Cooper * to the returned group IDs.
107*57718be8SEnji Cooper */
108*57718be8SEnji Cooper n = getgroups(NGROUPS_MAX, gidset);
109*57718be8SEnji Cooper ATF_REQUIRE(n >= 0);
110*57718be8SEnji Cooper
111*57718be8SEnji Cooper for (i = 0; i < n; i++) {
112*57718be8SEnji Cooper
113*57718be8SEnji Cooper pid = fork();
114*57718be8SEnji Cooper ATF_REQUIRE(pid >= 0);
115*57718be8SEnji Cooper
116*57718be8SEnji Cooper if (pid == 0) {
117*57718be8SEnji Cooper
118*57718be8SEnji Cooper rv = setgid(gidset[i]);
119*57718be8SEnji Cooper
120*57718be8SEnji Cooper if (rv != 0)
121*57718be8SEnji Cooper _exit(EXIT_FAILURE);
122*57718be8SEnji Cooper
123*57718be8SEnji Cooper _exit(EXIT_SUCCESS);
124*57718be8SEnji Cooper }
125*57718be8SEnji Cooper
126*57718be8SEnji Cooper (void)wait(&sta);
127*57718be8SEnji Cooper
128*57718be8SEnji Cooper if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS)
129*57718be8SEnji Cooper atf_tc_fail("getgroups(2) is inconsistent");
130*57718be8SEnji Cooper }
131*57718be8SEnji Cooper }
132*57718be8SEnji Cooper
133*57718be8SEnji Cooper ATF_TC(getgroups_zero);
ATF_TC_HEAD(getgroups_zero,tc)134*57718be8SEnji Cooper ATF_TC_HEAD(getgroups_zero, tc)
135*57718be8SEnji Cooper {
136*57718be8SEnji Cooper atf_tc_set_md_var(tc, "descr", "Test getgroups(2) with zero param");
137*57718be8SEnji Cooper }
138*57718be8SEnji Cooper
ATF_TC_BODY(getgroups_zero,tc)139*57718be8SEnji Cooper ATF_TC_BODY(getgroups_zero, tc)
140*57718be8SEnji Cooper {
141*57718be8SEnji Cooper const gid_t val = 123456789;
142*57718be8SEnji Cooper gid_t gidset[NGROUPS_MAX];
143*57718be8SEnji Cooper size_t i;
144*57718be8SEnji Cooper
145*57718be8SEnji Cooper /*
146*57718be8SEnji Cooper * If the first parameter is zero, the number
147*57718be8SEnji Cooper * of groups should be returned but the supplied
148*57718be8SEnji Cooper * buffer should remain intact.
149*57718be8SEnji Cooper */
150*57718be8SEnji Cooper for (i = 0; i < __arraycount(gidset); i++)
151*57718be8SEnji Cooper gidset[i] = val;
152*57718be8SEnji Cooper
153*57718be8SEnji Cooper ATF_REQUIRE(getgroups(0, gidset) >= 0);
154*57718be8SEnji Cooper
155*57718be8SEnji Cooper for (i = 0; i < __arraycount(gidset); i++) {
156*57718be8SEnji Cooper
157*57718be8SEnji Cooper if (gidset[i] != val)
158*57718be8SEnji Cooper atf_tc_fail("getgroups(2) modified the buffer");
159*57718be8SEnji Cooper }
160*57718be8SEnji Cooper }
161*57718be8SEnji Cooper
ATF_TP_ADD_TCS(tp)162*57718be8SEnji Cooper ATF_TP_ADD_TCS(tp)
163*57718be8SEnji Cooper {
164*57718be8SEnji Cooper
165*57718be8SEnji Cooper ATF_TP_ADD_TC(tp, getgroups_err);
166*57718be8SEnji Cooper ATF_TP_ADD_TC(tp, getgroups_getgid);
167*57718be8SEnji Cooper ATF_TP_ADD_TC(tp, getgroups_setgid);
168*57718be8SEnji Cooper ATF_TP_ADD_TC(tp, getgroups_zero);
169*57718be8SEnji Cooper
170*57718be8SEnji Cooper return atf_no_error();
171*57718be8SEnji Cooper }
172