t_msgctl.c revision 1.2.2.3
1 1.2.2.2 yamt /* $NetBSD: t_msgctl.c,v 1.2.2.3 2012/04/17 00:09:12 yamt Exp $ */ 2 1.2.2.2 yamt 3 1.2.2.2 yamt /*- 4 1.2.2.2 yamt * Copyright (c) 2011 The NetBSD Foundation, Inc. 5 1.2.2.2 yamt * All rights reserved. 6 1.2.2.2 yamt * 7 1.2.2.2 yamt * This code is derived from software contributed to The NetBSD Foundation 8 1.2.2.2 yamt * by Jukka Ruohonen. 9 1.2.2.2 yamt * 10 1.2.2.2 yamt * Redistribution and use in source and binary forms, with or without 11 1.2.2.2 yamt * modification, are permitted provided that the following conditions 12 1.2.2.2 yamt * are met: 13 1.2.2.2 yamt * 1. Redistributions of source code must retain the above copyright 14 1.2.2.2 yamt * notice, this list of conditions and the following disclaimer. 15 1.2.2.2 yamt * 2. Redistributions in binary form must reproduce the above copyright 16 1.2.2.2 yamt * notice, this list of conditions and the following disclaimer in the 17 1.2.2.2 yamt * documentation and/or other materials provided with the distribution. 18 1.2.2.2 yamt * 19 1.2.2.2 yamt * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20 1.2.2.2 yamt * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21 1.2.2.2 yamt * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22 1.2.2.2 yamt * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23 1.2.2.2 yamt * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24 1.2.2.2 yamt * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25 1.2.2.2 yamt * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26 1.2.2.2 yamt * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27 1.2.2.2 yamt * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28 1.2.2.2 yamt * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29 1.2.2.2 yamt * POSSIBILITY OF SUCH DAMAGE. 30 1.2.2.2 yamt */ 31 1.2.2.2 yamt #include <sys/cdefs.h> 32 1.2.2.2 yamt __RCSID("$NetBSD: t_msgctl.c,v 1.2.2.3 2012/04/17 00:09:12 yamt Exp $"); 33 1.2.2.2 yamt 34 1.2.2.2 yamt #include <sys/msg.h> 35 1.2.2.2 yamt #include <sys/stat.h> 36 1.2.2.2 yamt #include <sys/sysctl.h> 37 1.2.2.2 yamt #include <sys/wait.h> 38 1.2.2.2 yamt 39 1.2.2.2 yamt #include <atf-c.h> 40 1.2.2.2 yamt #include <errno.h> 41 1.2.2.2 yamt #include <pwd.h> 42 1.2.2.2 yamt #include <stdio.h> 43 1.2.2.2 yamt #include <stdlib.h> 44 1.2.2.2 yamt #include <string.h> 45 1.2.2.2 yamt #include <sysexits.h> 46 1.2.2.2 yamt #include <time.h> 47 1.2.2.2 yamt #include <unistd.h> 48 1.2.2.2 yamt 49 1.2.2.2 yamt #define MSG_KEY 12345689 50 1.2.2.2 yamt #define MSG_MTYPE_1 0x41 51 1.2.2.2 yamt 52 1.2.2.2 yamt struct msg { 53 1.2.2.2 yamt long mtype; 54 1.2.2.2 yamt char buf[3]; 55 1.2.2.2 yamt }; 56 1.2.2.2 yamt 57 1.2.2.2 yamt static void clean(void); 58 1.2.2.2 yamt 59 1.2.2.2 yamt static void 60 1.2.2.2 yamt clean(void) 61 1.2.2.2 yamt { 62 1.2.2.2 yamt int id; 63 1.2.2.2 yamt 64 1.2.2.2 yamt if ((id = msgget(MSG_KEY, 0)) != -1) 65 1.2.2.2 yamt (void)msgctl(id, IPC_RMID, 0); 66 1.2.2.2 yamt } 67 1.2.2.2 yamt 68 1.2.2.2 yamt ATF_TC_WITH_CLEANUP(msgctl_err); 69 1.2.2.2 yamt ATF_TC_HEAD(msgctl_err, tc) 70 1.2.2.2 yamt { 71 1.2.2.2 yamt atf_tc_set_md_var(tc, "descr", "Test errors from msgctl(2)"); 72 1.2.2.2 yamt } 73 1.2.2.2 yamt 74 1.2.2.2 yamt ATF_TC_BODY(msgctl_err, tc) 75 1.2.2.2 yamt { 76 1.2.2.2 yamt const int cmd[] = { IPC_STAT, IPC_SET, IPC_RMID }; 77 1.2.2.2 yamt struct msqid_ds msgds; 78 1.2.2.2 yamt size_t i; 79 1.2.2.2 yamt int id; 80 1.2.2.2 yamt 81 1.2.2.2 yamt (void)memset(&msgds, 0, sizeof(struct msqid_ds)); 82 1.2.2.2 yamt 83 1.2.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 84 1.2.2.2 yamt ATF_REQUIRE(id != -1); 85 1.2.2.2 yamt 86 1.2.2.2 yamt errno = 0; 87 1.2.2.2 yamt ATF_REQUIRE_ERRNO(EINVAL, msgctl(id, INT_MAX, &msgds) == -1); 88 1.2.2.2 yamt 89 1.2.2.2 yamt errno = 0; 90 1.2.2.2 yamt ATF_REQUIRE_ERRNO(EFAULT, msgctl(id, IPC_STAT, (void *)-1) == -1); 91 1.2.2.2 yamt 92 1.2.2.2 yamt for (i = 0; i < __arraycount(cmd); i++) { 93 1.2.2.2 yamt errno = 0; 94 1.2.2.2 yamt ATF_REQUIRE_ERRNO(EINVAL, msgctl(-1, cmd[i], &msgds) == -1); 95 1.2.2.2 yamt } 96 1.2.2.2 yamt 97 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 98 1.2.2.2 yamt } 99 1.2.2.2 yamt 100 1.2.2.2 yamt ATF_TC_CLEANUP(msgctl_err, tc) 101 1.2.2.2 yamt { 102 1.2.2.2 yamt clean(); 103 1.2.2.2 yamt } 104 1.2.2.2 yamt 105 1.2.2.2 yamt ATF_TC_WITH_CLEANUP(msgctl_perm); 106 1.2.2.2 yamt ATF_TC_HEAD(msgctl_perm, tc) 107 1.2.2.2 yamt { 108 1.2.2.3 yamt atf_tc_set_md_var(tc, "descr", "Test permissions with msgctl(2)"); 109 1.2.2.2 yamt atf_tc_set_md_var(tc, "require.user", "root"); 110 1.2.2.2 yamt } 111 1.2.2.2 yamt 112 1.2.2.2 yamt ATF_TC_BODY(msgctl_perm, tc) 113 1.2.2.2 yamt { 114 1.2.2.2 yamt struct msqid_ds msgds; 115 1.2.2.2 yamt struct passwd *pw; 116 1.2.2.2 yamt pid_t pid; 117 1.2.2.2 yamt int sta; 118 1.2.2.2 yamt int id; 119 1.2.2.2 yamt 120 1.2.2.2 yamt (void)memset(&msgds, 0, sizeof(struct msqid_ds)); 121 1.2.2.2 yamt 122 1.2.2.2 yamt pw = getpwnam("nobody"); 123 1.2.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 124 1.2.2.2 yamt 125 1.2.2.2 yamt ATF_REQUIRE(id != -1); 126 1.2.2.2 yamt ATF_REQUIRE(pw != NULL); 127 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_STAT, &msgds) == 0); 128 1.2.2.2 yamt 129 1.2.2.2 yamt pid = fork(); 130 1.2.2.2 yamt ATF_REQUIRE(pid >= 0); 131 1.2.2.2 yamt 132 1.2.2.2 yamt if (pid == 0) { 133 1.2.2.2 yamt 134 1.2.2.2 yamt if (setuid(pw->pw_uid) != 0) 135 1.2.2.2 yamt _exit(EX_OSERR); 136 1.2.2.2 yamt 137 1.2.2.2 yamt msgds.msg_perm.uid = getuid(); 138 1.2.2.2 yamt msgds.msg_perm.gid = getgid(); 139 1.2.2.2 yamt 140 1.2.2.2 yamt errno = 0; 141 1.2.2.2 yamt 142 1.2.2.2 yamt if (msgctl(id, IPC_SET, &msgds) == 0) 143 1.2.2.2 yamt _exit(EXIT_FAILURE); 144 1.2.2.2 yamt 145 1.2.2.2 yamt if (errno != EPERM) 146 1.2.2.2 yamt _exit(EXIT_FAILURE); 147 1.2.2.2 yamt 148 1.2.2.2 yamt (void)memset(&msgds, 0, sizeof(struct msqid_ds)); 149 1.2.2.2 yamt 150 1.2.2.2 yamt if (msgctl(id, IPC_STAT, &msgds) != 0) 151 1.2.2.2 yamt _exit(EX_OSERR); 152 1.2.2.2 yamt 153 1.2.2.2 yamt msgds.msg_qbytes = 1; 154 1.2.2.2 yamt 155 1.2.2.2 yamt if (msgctl(id, IPC_SET, &msgds) == 0) 156 1.2.2.2 yamt _exit(EXIT_FAILURE); 157 1.2.2.2 yamt 158 1.2.2.2 yamt if (errno != EPERM) 159 1.2.2.2 yamt _exit(EXIT_FAILURE); 160 1.2.2.2 yamt 161 1.2.2.2 yamt _exit(EXIT_SUCCESS); 162 1.2.2.2 yamt } 163 1.2.2.2 yamt 164 1.2.2.2 yamt (void)wait(&sta); 165 1.2.2.2 yamt 166 1.2.2.2 yamt if (WIFEXITED(sta) == 0) { 167 1.2.2.2 yamt 168 1.2.2.2 yamt if (WEXITSTATUS(sta) == EX_OSERR) 169 1.2.2.2 yamt atf_tc_fail("system call failed"); 170 1.2.2.2 yamt 171 1.2.2.2 yamt if (WEXITSTATUS(sta) == EXIT_FAILURE) 172 1.2.2.2 yamt atf_tc_fail("UID %u manipulated root's " 173 1.2.2.2 yamt "message queue", pw->pw_uid); 174 1.2.2.2 yamt } 175 1.2.2.2 yamt 176 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 177 1.2.2.2 yamt } 178 1.2.2.2 yamt 179 1.2.2.2 yamt ATF_TC_CLEANUP(msgctl_perm, tc) 180 1.2.2.2 yamt { 181 1.2.2.2 yamt clean(); 182 1.2.2.2 yamt } 183 1.2.2.2 yamt 184 1.2.2.2 yamt ATF_TC_WITH_CLEANUP(msgctl_pid); 185 1.2.2.2 yamt ATF_TC_HEAD(msgctl_pid, tc) 186 1.2.2.2 yamt { 187 1.2.2.2 yamt atf_tc_set_md_var(tc, "descr", "Test that PIDs are updated"); 188 1.2.2.2 yamt } 189 1.2.2.2 yamt 190 1.2.2.2 yamt ATF_TC_BODY(msgctl_pid, tc) 191 1.2.2.2 yamt { 192 1.2.2.2 yamt struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } }; 193 1.2.2.2 yamt struct msqid_ds msgds; 194 1.2.2.2 yamt int id, sta; 195 1.2.2.2 yamt pid_t pid; 196 1.2.2.2 yamt 197 1.2.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 198 1.2.2.2 yamt ATF_REQUIRE(id != -1); 199 1.2.2.2 yamt 200 1.2.2.2 yamt pid = fork(); 201 1.2.2.2 yamt ATF_REQUIRE(pid >= 0); 202 1.2.2.2 yamt 203 1.2.2.2 yamt if (pid == 0) { 204 1.2.2.2 yamt 205 1.2.2.2 yamt (void)msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT); 206 1.2.2.2 yamt 207 1.2.2.2 yamt _exit(EXIT_SUCCESS); 208 1.2.2.2 yamt } 209 1.2.2.2 yamt 210 1.2.2.2 yamt (void)sleep(1); 211 1.2.2.2 yamt (void)wait(&sta); 212 1.2.2.2 yamt (void)memset(&msgds, 0, sizeof(struct msqid_ds)); 213 1.2.2.2 yamt 214 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_STAT, &msgds) == 0); 215 1.2.2.2 yamt 216 1.2.2.2 yamt if (pid != msgds.msg_lspid) 217 1.2.2.2 yamt atf_tc_fail("the PID of last msgsnd(2) was not updated"); 218 1.2.2.2 yamt 219 1.2.2.2 yamt pid = fork(); 220 1.2.2.2 yamt ATF_REQUIRE(pid >= 0); 221 1.2.2.2 yamt 222 1.2.2.2 yamt if (pid == 0) { 223 1.2.2.2 yamt 224 1.2.2.2 yamt (void)msgrcv(id, &msg, 225 1.2.2.2 yamt sizeof(struct msg), MSG_MTYPE_1, IPC_NOWAIT); 226 1.2.2.2 yamt 227 1.2.2.2 yamt _exit(EXIT_SUCCESS); 228 1.2.2.2 yamt } 229 1.2.2.2 yamt 230 1.2.2.2 yamt (void)sleep(1); 231 1.2.2.2 yamt (void)wait(&sta); 232 1.2.2.2 yamt (void)memset(&msgds, 0, sizeof(struct msqid_ds)); 233 1.2.2.2 yamt 234 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_STAT, &msgds) == 0); 235 1.2.2.2 yamt 236 1.2.2.2 yamt if (pid != msgds.msg_lrpid) 237 1.2.2.2 yamt atf_tc_fail("the PID of last msgrcv(2) was not updated"); 238 1.2.2.2 yamt 239 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 240 1.2.2.2 yamt } 241 1.2.2.2 yamt 242 1.2.2.2 yamt ATF_TC_CLEANUP(msgctl_pid, tc) 243 1.2.2.2 yamt { 244 1.2.2.2 yamt clean(); 245 1.2.2.2 yamt } 246 1.2.2.2 yamt 247 1.2.2.2 yamt ATF_TC_WITH_CLEANUP(msgctl_set); 248 1.2.2.2 yamt ATF_TC_HEAD(msgctl_set, tc) 249 1.2.2.2 yamt { 250 1.2.2.2 yamt atf_tc_set_md_var(tc, "descr", "Test msgctl(2) with IPC_SET"); 251 1.2.2.2 yamt atf_tc_set_md_var(tc, "require.user", "root"); 252 1.2.2.2 yamt } 253 1.2.2.2 yamt 254 1.2.2.2 yamt ATF_TC_BODY(msgctl_set, tc) 255 1.2.2.2 yamt { 256 1.2.2.2 yamt struct msqid_ds msgds; 257 1.2.2.2 yamt struct passwd *pw; 258 1.2.2.2 yamt int id; 259 1.2.2.2 yamt 260 1.2.2.2 yamt (void)memset(&msgds, 0, sizeof(struct msqid_ds)); 261 1.2.2.2 yamt 262 1.2.2.2 yamt pw = getpwnam("nobody"); 263 1.2.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 264 1.2.2.2 yamt 265 1.2.2.2 yamt ATF_REQUIRE(id != -1); 266 1.2.2.2 yamt ATF_REQUIRE(pw != NULL); 267 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_STAT, &msgds) == 0); 268 1.2.2.2 yamt 269 1.2.2.2 yamt msgds.msg_perm.uid = pw->pw_uid; 270 1.2.2.2 yamt 271 1.2.2.2 yamt if (msgctl(id, IPC_SET, &msgds) != 0) 272 1.2.2.2 yamt atf_tc_fail("root failed to change the UID of message queue"); 273 1.2.2.2 yamt 274 1.2.2.2 yamt msgds.msg_perm.uid = getuid(); 275 1.2.2.2 yamt msgds.msg_perm.gid = pw->pw_gid; 276 1.2.2.2 yamt 277 1.2.2.2 yamt if (msgctl(id, IPC_SET, &msgds) != 0) 278 1.2.2.2 yamt atf_tc_fail("root failed to change the GID of message queue"); 279 1.2.2.2 yamt 280 1.2.2.2 yamt /* 281 1.2.2.2 yamt * Note: setting the qbytes to zero fails even as root. 282 1.2.2.2 yamt */ 283 1.2.2.2 yamt msgds.msg_qbytes = 1; 284 1.2.2.2 yamt msgds.msg_perm.gid = getgid(); 285 1.2.2.2 yamt 286 1.2.2.2 yamt if (msgctl(id, IPC_SET, &msgds) != 0) 287 1.2.2.2 yamt atf_tc_fail("root failed to change qbytes of message queue"); 288 1.2.2.2 yamt 289 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 290 1.2.2.2 yamt } 291 1.2.2.2 yamt 292 1.2.2.2 yamt ATF_TC_CLEANUP(msgctl_set, tc) 293 1.2.2.2 yamt { 294 1.2.2.2 yamt clean(); 295 1.2.2.2 yamt } 296 1.2.2.2 yamt 297 1.2.2.2 yamt ATF_TC_WITH_CLEANUP(msgctl_time); 298 1.2.2.2 yamt ATF_TC_HEAD(msgctl_time, tc) 299 1.2.2.2 yamt { 300 1.2.2.2 yamt atf_tc_set_md_var(tc, "descr", "Test that access times are updated"); 301 1.2.2.2 yamt } 302 1.2.2.2 yamt 303 1.2.2.2 yamt ATF_TC_BODY(msgctl_time, tc) 304 1.2.2.2 yamt { 305 1.2.2.2 yamt struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } }; 306 1.2.2.2 yamt struct msqid_ds msgds; 307 1.2.2.2 yamt time_t t; 308 1.2.2.2 yamt int id; 309 1.2.2.2 yamt 310 1.2.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 311 1.2.2.2 yamt ATF_REQUIRE(id != -1); 312 1.2.2.2 yamt 313 1.2.2.2 yamt t = time(NULL); 314 1.2.2.2 yamt 315 1.2.2.2 yamt (void)memset(&msgds, 0, sizeof(struct msqid_ds)); 316 1.2.2.2 yamt (void)msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT); 317 1.2.2.2 yamt (void)msgctl(id, IPC_STAT, &msgds); 318 1.2.2.2 yamt 319 1.2.2.2 yamt if (abs(t - msgds.msg_stime) > 1) 320 1.2.2.2 yamt atf_tc_fail("time of last msgsnd(2) was not updated"); 321 1.2.2.2 yamt 322 1.2.2.2 yamt if (msgds.msg_rtime != 0) 323 1.2.2.2 yamt atf_tc_fail("time of last msgrcv(2) was updated incorrectly"); 324 1.2.2.2 yamt 325 1.2.2.2 yamt t = time(NULL); 326 1.2.2.2 yamt 327 1.2.2.2 yamt (void)memset(&msgds, 0, sizeof(struct msqid_ds)); 328 1.2.2.2 yamt (void)msgrcv(id, &msg, sizeof(struct msg), MSG_MTYPE_1, IPC_NOWAIT); 329 1.2.2.2 yamt (void)msgctl(id, IPC_STAT, &msgds); 330 1.2.2.2 yamt 331 1.2.2.2 yamt if (abs(t - msgds.msg_rtime) > 1) 332 1.2.2.2 yamt atf_tc_fail("time of last msgrcv(2) was not updated"); 333 1.2.2.2 yamt 334 1.2.2.2 yamt /* 335 1.2.2.2 yamt * Note: this is non-zero even after the memset(3). 336 1.2.2.2 yamt */ 337 1.2.2.2 yamt if (msgds.msg_stime == 0) 338 1.2.2.2 yamt atf_tc_fail("time of last msgsnd(2) was updated incorrectly"); 339 1.2.2.2 yamt 340 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 341 1.2.2.2 yamt } 342 1.2.2.2 yamt 343 1.2.2.2 yamt ATF_TC_CLEANUP(msgctl_time, tc) 344 1.2.2.2 yamt { 345 1.2.2.2 yamt clean(); 346 1.2.2.2 yamt } 347 1.2.2.2 yamt 348 1.2.2.2 yamt ATF_TP_ADD_TCS(tp) 349 1.2.2.2 yamt { 350 1.2.2.2 yamt 351 1.2.2.2 yamt ATF_TP_ADD_TC(tp, msgctl_err); 352 1.2.2.2 yamt ATF_TP_ADD_TC(tp, msgctl_perm); 353 1.2.2.2 yamt ATF_TP_ADD_TC(tp, msgctl_pid); 354 1.2.2.2 yamt ATF_TP_ADD_TC(tp, msgctl_set); 355 1.2.2.2 yamt ATF_TP_ADD_TC(tp, msgctl_time); 356 1.2.2.2 yamt 357 1.2.2.2 yamt return atf_no_error(); 358 1.2.2.2 yamt } 359
Indexes created Wed Oct 01 07:09:59 GMT 2025