t_msgsnd.c revision 1.2.2.2
1 1.2.2.2 yamt /* $NetBSD: t_msgsnd.c,v 1.2.2.2 2011/11/10 14:31:52 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_msgsnd.c,v 1.2.2.2 2011/11/10 14:31:52 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 <signal.h> 43 1.2.2.2 yamt #include <stdio.h> 44 1.2.2.2 yamt #include <stdlib.h> 45 1.2.2.2 yamt #include <string.h> 46 1.2.2.2 yamt #include <sysexits.h> 47 1.2.2.2 yamt #include <time.h> 48 1.2.2.2 yamt #include <unistd.h> 49 1.2.2.2 yamt 50 1.2.2.2 yamt #define MSG_KEY 1234 51 1.2.2.2 yamt #define MSG_MTYPE_1 0x41 52 1.2.2.2 yamt #define MSG_MTYPE_2 0x42 53 1.2.2.2 yamt #define MSG_MTYPE_3 0x43 54 1.2.2.2 yamt 55 1.2.2.2 yamt struct msg { 56 1.2.2.2 yamt long mtype; 57 1.2.2.2 yamt char buf[3]; 58 1.2.2.2 yamt }; 59 1.2.2.2 yamt 60 1.2.2.2 yamt static void clean(void); 61 1.2.2.2 yamt 62 1.2.2.2 yamt static void 63 1.2.2.2 yamt clean(void) 64 1.2.2.2 yamt { 65 1.2.2.2 yamt int id; 66 1.2.2.2 yamt 67 1.2.2.2 yamt if ((id = msgget(MSG_KEY, 0)) != -1) 68 1.2.2.2 yamt (void)msgctl(id, IPC_RMID, 0); 69 1.2.2.2 yamt } 70 1.2.2.2 yamt 71 1.2.2.2 yamt ATF_TC_WITH_CLEANUP(msgsnd_block); 72 1.2.2.2 yamt ATF_TC_HEAD(msgsnd_block, tc) 73 1.2.2.2 yamt { 74 1.2.2.2 yamt atf_tc_set_md_var(tc, "descr", "Test that msgsnd(2) blocks"); 75 1.2.2.2 yamt atf_tc_set_md_var(tc, "timeout", "10"); 76 1.2.2.2 yamt } 77 1.2.2.2 yamt 78 1.2.2.2 yamt ATF_TC_BODY(msgsnd_block, tc) 79 1.2.2.2 yamt { 80 1.2.2.2 yamt struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } }; 81 1.2.2.2 yamt int id, sta; 82 1.2.2.2 yamt pid_t pid; 83 1.2.2.2 yamt 84 1.2.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 85 1.2.2.2 yamt ATF_REQUIRE(id != -1); 86 1.2.2.2 yamt 87 1.2.2.2 yamt pid = fork(); 88 1.2.2.2 yamt ATF_REQUIRE(pid >= 0); 89 1.2.2.2 yamt 90 1.2.2.2 yamt if (pid == 0) { 91 1.2.2.2 yamt 92 1.2.2.2 yamt /* 93 1.2.2.2 yamt * Enqueue messages until some limit (e.g. the maximum 94 1.2.2.2 yamt * number of messages in the queue or the maximum number 95 1.2.2.2 yamt * of bytes in the queue) is reached. After this the call 96 1.2.2.2 yamt * should block when the IPC_NOWAIT is not set. 97 1.2.2.2 yamt */ 98 1.2.2.2 yamt for (;;) { 99 1.2.2.2 yamt 100 1.2.2.2 yamt if (msgsnd(id, &msg, sizeof(struct msg), 0) < 0) 101 1.2.2.2 yamt _exit(EXIT_FAILURE); 102 1.2.2.2 yamt } 103 1.2.2.2 yamt } 104 1.2.2.2 yamt 105 1.2.2.2 yamt (void)sleep(2); 106 1.2.2.2 yamt (void)kill(pid, SIGKILL); 107 1.2.2.2 yamt (void)wait(&sta); 108 1.2.2.2 yamt 109 1.2.2.2 yamt if (WIFEXITED(sta) != 0 || WIFSIGNALED(sta) == 0) 110 1.2.2.2 yamt atf_tc_fail("msgsnd(2) did not block"); 111 1.2.2.2 yamt 112 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 113 1.2.2.2 yamt } 114 1.2.2.2 yamt 115 1.2.2.2 yamt ATF_TC_CLEANUP(msgsnd_block, tc) 116 1.2.2.2 yamt { 117 1.2.2.2 yamt clean(); 118 1.2.2.2 yamt } 119 1.2.2.2 yamt 120 1.2.2.2 yamt ATF_TC_WITH_CLEANUP(msgsnd_count); 121 1.2.2.2 yamt ATF_TC_HEAD(msgsnd_count, tc) 122 1.2.2.2 yamt { 123 1.2.2.2 yamt atf_tc_set_md_var(tc, "descr", 124 1.2.2.2 yamt "Test that msgsnd(2) increments the amount of " 125 1.2.2.2 yamt "message in the queue, as given by msgctl(2)"); 126 1.2.2.2 yamt atf_tc_set_md_var(tc, "timeout", "10"); 127 1.2.2.2 yamt } 128 1.2.2.2 yamt 129 1.2.2.2 yamt ATF_TC_BODY(msgsnd_count, tc) 130 1.2.2.2 yamt { 131 1.2.2.2 yamt struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } }; 132 1.2.2.2 yamt struct msqid_ds ds; 133 1.2.2.2 yamt size_t i = 0; 134 1.2.2.2 yamt int id, rv; 135 1.2.2.2 yamt 136 1.2.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 137 1.2.2.2 yamt ATF_REQUIRE(id != -1); 138 1.2.2.2 yamt 139 1.2.2.2 yamt for (;;) { 140 1.2.2.2 yamt 141 1.2.2.2 yamt errno = 0; 142 1.2.2.2 yamt rv = msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT); 143 1.2.2.2 yamt 144 1.2.2.2 yamt if (rv == 0) { 145 1.2.2.2 yamt i++; 146 1.2.2.2 yamt continue; 147 1.2.2.2 yamt } 148 1.2.2.2 yamt 149 1.2.2.2 yamt if (rv == -1 && errno == EAGAIN) 150 1.2.2.2 yamt break; 151 1.2.2.2 yamt 152 1.2.2.2 yamt atf_tc_fail("failed to enqueue a message"); 153 1.2.2.2 yamt } 154 1.2.2.2 yamt 155 1.2.2.2 yamt (void)memset(&ds, 0, sizeof(struct msqid_ds)); 156 1.2.2.2 yamt (void)msgctl(id, IPC_STAT, &ds); 157 1.2.2.2 yamt 158 1.2.2.2 yamt if (ds.msg_qnum != i) 159 1.2.2.2 yamt atf_tc_fail("incorrect message count"); 160 1.2.2.2 yamt 161 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 162 1.2.2.2 yamt } 163 1.2.2.2 yamt 164 1.2.2.2 yamt ATF_TC_CLEANUP(msgsnd_count, tc) 165 1.2.2.2 yamt { 166 1.2.2.2 yamt clean(); 167 1.2.2.2 yamt } 168 1.2.2.2 yamt 169 1.2.2.2 yamt ATF_TC_WITH_CLEANUP(msgsnd_err); 170 1.2.2.2 yamt ATF_TC_HEAD(msgsnd_err, tc) 171 1.2.2.2 yamt { 172 1.2.2.2 yamt atf_tc_set_md_var(tc, "descr", "Test errors from msgsnd(2)"); 173 1.2.2.2 yamt } 174 1.2.2.2 yamt 175 1.2.2.2 yamt ATF_TC_BODY(msgsnd_err, tc) 176 1.2.2.2 yamt { 177 1.2.2.2 yamt struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } }; 178 1.2.2.2 yamt int id; 179 1.2.2.2 yamt 180 1.2.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 181 1.2.2.2 yamt ATF_REQUIRE(id != -1); 182 1.2.2.2 yamt 183 1.2.2.2 yamt errno = 0; 184 1.2.2.2 yamt 185 1.2.2.2 yamt ATF_REQUIRE_ERRNO(EFAULT, msgsnd(id, (void *)-1, 186 1.2.2.2 yamt sizeof(struct msg), IPC_NOWAIT) == -1); 187 1.2.2.2 yamt 188 1.2.2.2 yamt errno = 0; 189 1.2.2.2 yamt 190 1.2.2.2 yamt ATF_REQUIRE_ERRNO(EINVAL, msgsnd(-1, &msg, 191 1.2.2.2 yamt sizeof(struct msg), IPC_NOWAIT) == -1); 192 1.2.2.2 yamt 193 1.2.2.2 yamt errno = 0; 194 1.2.2.2 yamt 195 1.2.2.2 yamt ATF_REQUIRE_ERRNO(EINVAL, msgsnd(-1, &msg, 196 1.2.2.2 yamt SSIZE_MAX, IPC_NOWAIT) == -1); 197 1.2.2.2 yamt 198 1.2.2.2 yamt errno = 0; 199 1.2.2.2 yamt msg.mtype = 0; 200 1.2.2.2 yamt 201 1.2.2.2 yamt ATF_REQUIRE_ERRNO(EINVAL, msgsnd(id, &msg, 202 1.2.2.2 yamt sizeof(struct msg), IPC_NOWAIT) == -1); 203 1.2.2.2 yamt 204 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 205 1.2.2.2 yamt } 206 1.2.2.2 yamt 207 1.2.2.2 yamt ATF_TC_CLEANUP(msgsnd_err, tc) 208 1.2.2.2 yamt { 209 1.2.2.2 yamt clean(); 210 1.2.2.2 yamt } 211 1.2.2.2 yamt 212 1.2.2.2 yamt ATF_TC_WITH_CLEANUP(msgsnd_nonblock); 213 1.2.2.2 yamt ATF_TC_HEAD(msgsnd_nonblock, tc) 214 1.2.2.2 yamt { 215 1.2.2.2 yamt atf_tc_set_md_var(tc, "descr", "Test msgsnd(2) with IPC_NOWAIT"); 216 1.2.2.2 yamt atf_tc_set_md_var(tc, "timeout", "10"); 217 1.2.2.2 yamt } 218 1.2.2.2 yamt 219 1.2.2.2 yamt ATF_TC_BODY(msgsnd_nonblock, tc) 220 1.2.2.2 yamt { 221 1.2.2.2 yamt struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } }; 222 1.2.2.2 yamt int id, rv, sta; 223 1.2.2.2 yamt pid_t pid; 224 1.2.2.2 yamt 225 1.2.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 226 1.2.2.2 yamt ATF_REQUIRE(id != -1); 227 1.2.2.2 yamt 228 1.2.2.2 yamt pid = fork(); 229 1.2.2.2 yamt ATF_REQUIRE(pid >= 0); 230 1.2.2.2 yamt 231 1.2.2.2 yamt if (pid == 0) { 232 1.2.2.2 yamt 233 1.2.2.2 yamt for (;;) { 234 1.2.2.2 yamt 235 1.2.2.2 yamt errno = 0; 236 1.2.2.2 yamt rv = msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT); 237 1.2.2.2 yamt 238 1.2.2.2 yamt if (rv == -1 && errno == EAGAIN) 239 1.2.2.2 yamt _exit(EXIT_SUCCESS); 240 1.2.2.2 yamt } 241 1.2.2.2 yamt } 242 1.2.2.2 yamt 243 1.2.2.2 yamt (void)sleep(2); 244 1.2.2.2 yamt (void)kill(pid, SIGKILL); 245 1.2.2.2 yamt (void)wait(&sta); 246 1.2.2.2 yamt 247 1.2.2.2 yamt if (WIFEXITED(sta) == 0 || WIFSIGNALED(sta) != 0) 248 1.2.2.2 yamt atf_tc_fail("msgsnd(2) blocked with IPC_NOWAIT"); 249 1.2.2.2 yamt 250 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 251 1.2.2.2 yamt } 252 1.2.2.2 yamt 253 1.2.2.2 yamt ATF_TC_CLEANUP(msgsnd_nonblock, tc) 254 1.2.2.2 yamt { 255 1.2.2.2 yamt clean(); 256 1.2.2.2 yamt } 257 1.2.2.2 yamt 258 1.2.2.2 yamt ATF_TC_WITH_CLEANUP(msgsnd_perm); 259 1.2.2.2 yamt ATF_TC_HEAD(msgsnd_perm, tc) 260 1.2.2.2 yamt { 261 1.2.2.2 yamt atf_tc_set_md_var(tc, "descr", "Test permissions with msgsnd(2)"); 262 1.2.2.2 yamt atf_tc_set_md_var(tc, "require.user", "root"); 263 1.2.2.2 yamt } 264 1.2.2.2 yamt 265 1.2.2.2 yamt ATF_TC_BODY(msgsnd_perm, tc) 266 1.2.2.2 yamt { 267 1.2.2.2 yamt struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } }; 268 1.2.2.2 yamt struct passwd *pw; 269 1.2.2.2 yamt int id, sta; 270 1.2.2.2 yamt pid_t pid; 271 1.2.2.2 yamt uid_t uid; 272 1.2.2.2 yamt 273 1.2.2.2 yamt pw = getpwnam("nobody"); 274 1.2.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 275 1.2.2.2 yamt 276 1.2.2.2 yamt ATF_REQUIRE(id != -1); 277 1.2.2.2 yamt ATF_REQUIRE(pw != NULL); 278 1.2.2.2 yamt 279 1.2.2.2 yamt uid = pw->pw_uid; 280 1.2.2.2 yamt ATF_REQUIRE(uid != 0); 281 1.2.2.2 yamt 282 1.2.2.2 yamt pid = fork(); 283 1.2.2.2 yamt ATF_REQUIRE(pid >= 0); 284 1.2.2.2 yamt 285 1.2.2.2 yamt if (pid == 0) { 286 1.2.2.2 yamt 287 1.2.2.2 yamt /* 288 1.2.2.2 yamt * Try to enqueue a message to the queue 289 1.2.2.2 yamt * created by root as RW for owner only. 290 1.2.2.2 yamt */ 291 1.2.2.2 yamt if (setuid(uid) != 0) 292 1.2.2.2 yamt _exit(EX_OSERR); 293 1.2.2.2 yamt 294 1.2.2.2 yamt id = msgget(MSG_KEY, 0); 295 1.2.2.2 yamt 296 1.2.2.2 yamt if (id == -1) 297 1.2.2.2 yamt _exit(EX_OSERR); 298 1.2.2.2 yamt 299 1.2.2.2 yamt errno = 0; 300 1.2.2.2 yamt 301 1.2.2.2 yamt if (msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT) == 0) 302 1.2.2.2 yamt _exit(EXIT_FAILURE); 303 1.2.2.2 yamt 304 1.2.2.2 yamt if (errno != EACCES) 305 1.2.2.2 yamt _exit(EXIT_FAILURE); 306 1.2.2.2 yamt 307 1.2.2.2 yamt _exit(EXIT_SUCCESS); 308 1.2.2.2 yamt } 309 1.2.2.2 yamt 310 1.2.2.2 yamt (void)wait(&sta); 311 1.2.2.2 yamt 312 1.2.2.2 yamt if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS) { 313 1.2.2.2 yamt 314 1.2.2.2 yamt if (errno == EX_OSERR) 315 1.2.2.2 yamt atf_tc_fail("system call failed"); 316 1.2.2.2 yamt 317 1.2.2.2 yamt atf_tc_fail("UID %u enqueued message to root's queue", uid); 318 1.2.2.2 yamt } 319 1.2.2.2 yamt 320 1.2.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 321 1.2.2.2 yamt } 322 1.2.2.2 yamt 323 1.2.2.2 yamt ATF_TC_CLEANUP(msgsnd_perm, tc) 324 1.2.2.2 yamt { 325 1.2.2.2 yamt clean(); 326 1.2.2.2 yamt } 327 1.2.2.2 yamt 328 1.2.2.2 yamt ATF_TP_ADD_TCS(tp) 329 1.2.2.2 yamt { 330 1.2.2.2 yamt 331 1.2.2.2 yamt ATF_TP_ADD_TC(tp, msgsnd_block); 332 1.2.2.2 yamt ATF_TP_ADD_TC(tp, msgsnd_count); 333 1.2.2.2 yamt ATF_TP_ADD_TC(tp, msgsnd_err); 334 1.2.2.2 yamt ATF_TP_ADD_TC(tp, msgsnd_nonblock); 335 1.2.2.2 yamt ATF_TP_ADD_TC(tp, msgsnd_perm); 336 1.2.2.2 yamt 337 1.2.2.2 yamt return atf_no_error(); 338 1.2.2.2 yamt } 339
Indexes created Wed Oct 01 19:09:53 GMT 2025