1 /*
2  * Copyright (c) International Business Machines  Corp., 2001
3  *
4  * This program is free software;  you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY;  without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
12  * the GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program;  if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  *
18  * Ported by John George
19  */
20 
21 /*
22  * Test setregid() when executed by root.
23  */
24 
25 #include <errno.h>
26 #include <pwd.h>
27 #include <grp.h>
28 #include <stdlib.h>
29 #include <string.h>
30 
31 #include "test.h"
32 #include "compat_16.h"
33 
34 TCID_DEFINE(setregid04);
35 
36 static gid_t neg_one = -1;
37 
38 static struct group users_gr, daemon_gr, root_gr, bin_gr;
39 
40 /*
41  * The following structure contains all test data.  Each structure in the array
42  * is used for a separate test.  The tests are executed in the for loop below.
43  */
44 
45 struct test_data_t {
46 	gid_t *real_gid;
47 	gid_t *eff_gid;
48 	struct group *exp_real_usr;
49 	struct group *exp_eff_usr;
50 	const char *test_msg;
51 } test_data[] = {
52 	{
53 	&root_gr.gr_gid, &root_gr.gr_gid, &root_gr, &root_gr,
54 		    "After setregid(root, root),"}, {
55 	&users_gr.gr_gid, &neg_one, &users_gr, &root_gr,
56 		    "After setregid(users, -1)"}, {
57 	&root_gr.gr_gid, &neg_one, &root_gr, &root_gr,
58 		    "After setregid(root,-1),"}, {
59 	&neg_one, &neg_one, &root_gr, &root_gr,
60 		    "After setregid(-1, -1),"}, {
61 	&neg_one, &root_gr.gr_gid, &root_gr, &root_gr,
62 		    "After setregid(-1, root)"}, {
63 	&root_gr.gr_gid, &neg_one, &root_gr, &root_gr,
64 		    "After setregid(root, -1),"}, {
65 	&daemon_gr.gr_gid, &users_gr.gr_gid, &daemon_gr, &users_gr,
66 		    "After setregid(daemon, users)"}, {
67 	&neg_one, &neg_one, &daemon_gr, &users_gr,
68 		    "After setregid(-1, -1)"}, {
69 	&neg_one, &users_gr.gr_gid, &daemon_gr, &users_gr,
70 		    "After setregid(-1, users)"}
71 };
72 
73 int TST_TOTAL = sizeof(test_data) / sizeof(test_data[0]);
74 
75 static void setup(void);
76 static void cleanup(void);
77 static void gid_verify(struct group *ru, struct group *eu, const char *when);
78 
main(int ac,char ** av)79 int main(int ac, char **av)
80 {
81 	int lc;
82 
83 	tst_parse_opts(ac, av, NULL, NULL);
84 
85 	setup();
86 
87 	for (lc = 0; TEST_LOOPING(lc); lc++) {
88 		int i;
89 
90 		tst_count = 0;
91 
92 		for (i = 0; i < TST_TOTAL; i++) {
93 			/* Set the real or effective group id */
94 			TEST(SETREGID(cleanup, *test_data[i].real_gid,
95 				      *test_data[i].eff_gid));
96 
97 			if (TEST_RETURN == -1) {
98 				tst_resm(TBROK, "setregid(%d, %d) failed",
99 					 *test_data[i].real_gid,
100 					 *test_data[i].eff_gid);
101 			} else {
102 				gid_verify(test_data[i].exp_real_usr,
103 					   test_data[i].exp_eff_usr,
104 					   test_data[i].test_msg);
105 			}
106 		}
107 	}
108 
109 	cleanup();
110 	tst_exit();
111 }
112 
113 #define SAFE_GETGROUP(GROUPNAME)	\
114 	if (getgrnam(#GROUPNAME) == NULL) { \
115 		tst_brkm(TBROK, NULL, "Couldn't find the `" #GROUPNAME "' group"); \
116 	} \
117 	GROUPNAME ## _gr = *(getgrnam(#GROUPNAME));
118 
setup(void)119 static void setup(void)
120 {
121 	tst_require_root();
122 
123 	tst_sig(FORK, DEF_HANDLER, cleanup);
124 
125 	SAFE_GETGROUP(root);
126 	SAFE_GETGROUP(users);
127 	SAFE_GETGROUP(daemon);
128 	SAFE_GETGROUP(bin);
129 
130 	TEST_PAUSE;
131 }
132 
cleanup(void)133 static void cleanup(void)
134 {
135 }
136 
gid_verify(struct group * rg,struct group * eg,const char * when)137 static void gid_verify(struct group *rg, struct group *eg, const char *when)
138 {
139 	if ((getgid() != rg->gr_gid) || (getegid() != eg->gr_gid)) {
140 		tst_resm(TFAIL, "ERROR: %s real gid = %d; effective gid = %d",
141 			 when, getgid(), getegid());
142 		tst_resm(TINFO, "Expected: real gid = %d; effective gid = %d",
143 			 rg->gr_gid, eg->gr_gid);
144 	} else {
145 		tst_resm(TPASS, "real or effective gid was modified as "
146 			 "expected");
147 	}
148 }
149