1 /*
2 *
3 * Copyright (c) International Business Machines Corp., 2004
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
13 * the GNU General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18 */
19
20 /*
21 * NAME
22 * hugeshmctl03.c
23 *
24 * DESCRIPTION
25 * hugeshmctl03 - check for EACCES, and EPERM errors
26 *
27 * ALGORITHM
28 * create a large shared memory segment with root only read & write
29 * permissions fork a child process
30 * if child
31 * set the ID of the child process to that of "ltpuser1"
32 * call do_child()
33 * loop if that option was specified
34 * call shmctl() using three different invalid cases
35 * check the errno value
36 * issue a PASS message if we get EACCES or EPERM
37 * otherwise, the tests fails
38 * issue a FAIL message
39 * call cleanup
40 * if parent
41 * wait for child to exit
42 * remove the large shared memory segment
43 *
44 * USAGE: <for command-line>
45 * hugeshmctl03 [-c n] [-e] [-i n] [-I x] [-P x] [-t]
46 * where, -c n : Run n copies concurrently.
47 * -e : Turn on errno logging.
48 * -i n : Execute test n times.
49 * -I x : Execute test for x seconds.
50 * -P x : Pause for x seconds between iterations.
51 * -t : Turn on syscall timing.
52 *
53 * HISTORY
54 * 03/2001 - Written by Wayne Boyer
55 * 04/2004 - Updated by Robbie Williamson
56 *
57 * RESTRICTIONS
58 * test must be run as root
59 */
60
61 #include <sys/types.h>
62 #include <sys/wait.h>
63 #include "test.h"
64 #include "safe_macros.h"
65 #include "mem.h"
66 #include "hugetlb.h"
67
68 char *TCID = "hugeshmctl03";
69 int TST_TOTAL = 3;
70
71 static size_t shm_size;
72 static int shm_id_1 = -1;
73 static struct shmid_ds buf;
74 static uid_t ltp_uid;
75 static char *ltp_user = "nobody";
76
77 static long hugepages = 128;
78 static option_t options[] = {
79 {"s:", &sflag, &nr_opt},
80 {NULL, NULL, NULL}
81 };
82
83 struct test_case_t {
84 int *shmid;
85 int cmd;
86 struct shmid_ds *sbuf;
87 int error;
88 } TC[] = {
89 /* EACCES - child has no read permission for segment */
90 {
91 &shm_id_1, IPC_STAT, &buf, EACCES},
92 /* EPERM - IPC_SET - child doesn't have permission to change segment */
93 {
94 &shm_id_1, IPC_SET, &buf, EPERM},
95 /* EPERM - IPC_RMID - child can not remove the segment */
96 {
97 &shm_id_1, IPC_RMID, &buf, EPERM},};
98
99 static void do_child(void);
100
main(int ac,char ** av)101 int main(int ac, char **av)
102 {
103 pid_t pid;
104 int status;
105
106 tst_parse_opts(ac, av, options, NULL);
107
108 if (sflag)
109 hugepages = SAFE_STRTOL(NULL, nr_opt, 0, LONG_MAX);
110
111 setup();
112
113 switch (pid = fork()) {
114 case -1:
115 tst_brkm(TBROK | TERRNO, cleanup, "fork");
116 case 0:
117 /* set the user ID of the child to the non root user */
118 if (setuid(ltp_uid) == -1)
119 tst_brkm(TBROK | TERRNO, cleanup, "setuid");
120 do_child();
121 tst_exit();
122 default:
123 if (waitpid(pid, &status, 0) == -1)
124 tst_brkm(TBROK | TERRNO, cleanup, "waitpid");
125 }
126 cleanup();
127 tst_exit();
128 }
129
do_child(void)130 static void do_child(void)
131 {
132 int i, lc;
133
134 for (lc = 0; TEST_LOOPING(lc); lc++) {
135 tst_count = 0;
136
137 for (i = 0; i < TST_TOTAL; i++) {
138 TEST(shmctl(*(TC[i].shmid), TC[i].cmd, TC[i].sbuf));
139 if (TEST_RETURN != -1) {
140 tst_resm(TFAIL, "shmctl succeeded "
141 "unexpectedly");
142 continue;
143 }
144 if (TEST_ERRNO == TC[i].error)
145 tst_resm(TPASS | TTERRNO, "shmctl failed "
146 "as expected");
147 else
148 tst_resm(TFAIL | TTERRNO, "shmctl failed "
149 "unexpectedly - expect errno = "
150 "%d, got", TC[i].error);
151 }
152 }
153 }
154
setup(void)155 void setup(void)
156 {
157 long hpage_size;
158
159 tst_require_root();
160 check_hugepage();
161 tst_sig(FORK, DEF_HANDLER, cleanup);
162 tst_tmpdir();
163
164 orig_hugepages = get_sys_tune("nr_hugepages");
165 set_sys_tune("nr_hugepages", hugepages, 1);
166 hpage_size = read_meminfo("Hugepagesize:") * 1024;
167
168 shm_size = hpage_size * hugepages / 2;
169 update_shm_size(&shm_size);
170 shmkey = getipckey(cleanup);
171 shm_id_1 = shmget(shmkey, shm_size,
172 SHM_HUGETLB | IPC_CREAT | IPC_EXCL | SHM_RW);
173 if (shm_id_1 == -1)
174 tst_brkm(TBROK | TERRNO, cleanup, "shmget");
175
176 /* get the userid for a non root user */
177 ltp_uid = getuserid(cleanup, ltp_user);
178
179 TEST_PAUSE;
180 }
181
cleanup(void)182 void cleanup(void)
183 {
184 rm_shm(shm_id_1);
185
186 set_sys_tune("nr_hugepages", orig_hugepages, 0);
187
188 tst_rmdir();
189 }
190