t_msgget.c revision 1.1.2.3
1 1.1.2.3 yamt /* $NetBSD: t_msgget.c,v 1.1.2.3 2014/05/22 11:42:21 yamt Exp $ */ 2 1.1.2.2 yamt 3 1.1.2.2 yamt /*- 4 1.1.2.2 yamt * Copyright (c) 2011 The NetBSD Foundation, Inc. 5 1.1.2.2 yamt * All rights reserved. 6 1.1.2.2 yamt * 7 1.1.2.2 yamt * This code is derived from software contributed to The NetBSD Foundation 8 1.1.2.2 yamt * by Jukka Ruohonen. 9 1.1.2.2 yamt * 10 1.1.2.2 yamt * Redistribution and use in source and binary forms, with or without 11 1.1.2.2 yamt * modification, are permitted provided that the following conditions 12 1.1.2.2 yamt * are met: 13 1.1.2.2 yamt * 1. Redistributions of source code must retain the above copyright 14 1.1.2.2 yamt * notice, this list of conditions and the following disclaimer. 15 1.1.2.2 yamt * 2. Redistributions in binary form must reproduce the above copyright 16 1.1.2.2 yamt * notice, this list of conditions and the following disclaimer in the 17 1.1.2.2 yamt * documentation and/or other materials provided with the distribution. 18 1.1.2.2 yamt * 19 1.1.2.2 yamt * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20 1.1.2.2 yamt * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21 1.1.2.2 yamt * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22 1.1.2.2 yamt * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23 1.1.2.2 yamt * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24 1.1.2.2 yamt * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25 1.1.2.2 yamt * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26 1.1.2.2 yamt * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27 1.1.2.2 yamt * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28 1.1.2.2 yamt * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29 1.1.2.2 yamt * POSSIBILITY OF SUCH DAMAGE. 30 1.1.2.2 yamt */ 31 1.1.2.2 yamt #include <sys/cdefs.h> 32 1.1.2.3 yamt __RCSID("$NetBSD: t_msgget.c,v 1.1.2.3 2014/05/22 11:42:21 yamt Exp $"); 33 1.1.2.2 yamt 34 1.1.2.2 yamt #include <sys/msg.h> 35 1.1.2.2 yamt #include <sys/stat.h> 36 1.1.2.2 yamt #include <sys/sysctl.h> 37 1.1.2.2 yamt #include <sys/wait.h> 38 1.1.2.2 yamt 39 1.1.2.2 yamt #include <atf-c.h> 40 1.1.2.2 yamt #include <errno.h> 41 1.1.2.2 yamt #include <pwd.h> 42 1.1.2.2 yamt #include <stdio.h> 43 1.1.2.2 yamt #include <stdlib.h> 44 1.1.2.2 yamt #include <string.h> 45 1.1.2.2 yamt #include <sysexits.h> 46 1.1.2.2 yamt #include <time.h> 47 1.1.2.2 yamt #include <unistd.h> 48 1.1.2.2 yamt 49 1.1.2.2 yamt #define MSG_KEY 12345689 50 1.1.2.2 yamt 51 1.1.2.2 yamt static void clean(void); 52 1.1.2.2 yamt 53 1.1.2.2 yamt static void 54 1.1.2.2 yamt clean(void) 55 1.1.2.2 yamt { 56 1.1.2.2 yamt int id; 57 1.1.2.2 yamt 58 1.1.2.2 yamt if ((id = msgget(MSG_KEY, 0)) != -1) 59 1.1.2.2 yamt (void)msgctl(id, IPC_RMID, 0); 60 1.1.2.2 yamt } 61 1.1.2.2 yamt 62 1.1.2.2 yamt ATF_TC_WITH_CLEANUP(msgget_excl); 63 1.1.2.2 yamt ATF_TC_HEAD(msgget_excl, tc) 64 1.1.2.2 yamt { 65 1.1.2.2 yamt atf_tc_set_md_var(tc, "descr", "Test msgget(2) with IPC_EXCL"); 66 1.1.2.2 yamt } 67 1.1.2.2 yamt 68 1.1.2.2 yamt ATF_TC_BODY(msgget_excl, tc) 69 1.1.2.2 yamt { 70 1.1.2.2 yamt int id; 71 1.1.2.2 yamt 72 1.1.2.2 yamt /* 73 1.1.2.2 yamt * Create a message queue and re-open it with 74 1.1.2.2 yamt * O_CREAT and IPC_EXCL set. This should fail. 75 1.1.2.2 yamt */ 76 1.1.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 77 1.1.2.2 yamt 78 1.1.2.2 yamt if (id < 0) 79 1.1.2.2 yamt atf_tc_fail("failed to create message queue"); 80 1.1.2.2 yamt 81 1.1.2.2 yamt errno = 0; 82 1.1.2.2 yamt 83 1.1.2.2 yamt if (msgget(MSG_KEY, IPC_CREAT | IPC_EXCL | 0600) != -1) 84 1.1.2.2 yamt atf_tc_fail("msgget(2) failed for IPC_EXCL"); 85 1.1.2.2 yamt 86 1.1.2.2 yamt ATF_REQUIRE(errno == EEXIST); 87 1.1.2.2 yamt 88 1.1.2.2 yamt /* 89 1.1.2.2 yamt * However, the same call should succeed 90 1.1.2.2 yamt * when IPC_EXCL is not set in the flags. 91 1.1.2.2 yamt */ 92 1.1.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 93 1.1.2.2 yamt 94 1.1.2.2 yamt if (id < 0) 95 1.1.2.2 yamt atf_tc_fail("msgget(2) failed to re-open"); 96 1.1.2.2 yamt 97 1.1.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 98 1.1.2.2 yamt } 99 1.1.2.2 yamt 100 1.1.2.2 yamt ATF_TC_CLEANUP(msgget_excl, tc) 101 1.1.2.2 yamt { 102 1.1.2.2 yamt clean(); 103 1.1.2.2 yamt } 104 1.1.2.2 yamt 105 1.1.2.2 yamt ATF_TC_WITH_CLEANUP(msgget_exit); 106 1.1.2.2 yamt ATF_TC_HEAD(msgget_exit, tc) 107 1.1.2.2 yamt { 108 1.1.2.2 yamt atf_tc_set_md_var(tc, "descr", 109 1.1.2.2 yamt "Test that XSI message queues are " 110 1.1.2.2 yamt "not removed when the process exits"); 111 1.1.2.2 yamt } 112 1.1.2.2 yamt 113 1.1.2.2 yamt ATF_TC_BODY(msgget_exit, tc) 114 1.1.2.2 yamt { 115 1.1.2.2 yamt int id, sta; 116 1.1.2.2 yamt pid_t pid; 117 1.1.2.2 yamt 118 1.1.2.2 yamt pid = fork(); 119 1.1.2.2 yamt ATF_REQUIRE(pid >= 0); 120 1.1.2.2 yamt 121 1.1.2.2 yamt if (pid == 0) { 122 1.1.2.2 yamt 123 1.1.2.2 yamt if (msgget(MSG_KEY, IPC_CREAT | IPC_EXCL | 0600) == -1) 124 1.1.2.2 yamt _exit(EXIT_FAILURE); 125 1.1.2.2 yamt 126 1.1.2.2 yamt _exit(EXIT_SUCCESS); 127 1.1.2.2 yamt } 128 1.1.2.2 yamt 129 1.1.2.2 yamt (void)wait(&sta); 130 1.1.2.2 yamt 131 1.1.2.2 yamt if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS) 132 1.1.2.2 yamt atf_tc_fail("failed to create message queue"); 133 1.1.2.2 yamt 134 1.1.2.2 yamt id = msgget(MSG_KEY, 0); 135 1.1.2.2 yamt 136 1.1.2.2 yamt if (id == -1) 137 1.1.2.2 yamt atf_tc_fail("message queue was removed on process exit"); 138 1.1.2.2 yamt 139 1.1.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 140 1.1.2.2 yamt } 141 1.1.2.2 yamt 142 1.1.2.2 yamt ATF_TC_CLEANUP(msgget_exit, tc) 143 1.1.2.2 yamt { 144 1.1.2.2 yamt clean(); 145 1.1.2.2 yamt } 146 1.1.2.2 yamt 147 1.1.2.2 yamt ATF_TC_WITH_CLEANUP(msgget_init); 148 1.1.2.2 yamt ATF_TC_HEAD(msgget_init, tc) 149 1.1.2.2 yamt { 150 1.1.2.2 yamt atf_tc_set_md_var(tc, "descr", 151 1.1.2.2 yamt "Test that msgget(2) initializes data structures properly"); 152 1.1.2.2 yamt } 153 1.1.2.2 yamt 154 1.1.2.2 yamt ATF_TC_BODY(msgget_init, tc) 155 1.1.2.2 yamt { 156 1.1.2.2 yamt const uid_t uid = geteuid(); 157 1.1.2.2 yamt const gid_t gid = getegid(); 158 1.1.2.2 yamt struct msqid_ds msgds; 159 1.1.2.2 yamt time_t t; 160 1.1.2.2 yamt int id; 161 1.1.2.2 yamt 162 1.1.2.2 yamt (void)memset(&msgds, 0x9, sizeof(struct msqid_ds)); 163 1.1.2.2 yamt 164 1.1.2.2 yamt t = time(NULL); 165 1.1.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | 0600); 166 1.1.2.2 yamt 167 1.1.2.2 yamt ATF_REQUIRE(id !=-1); 168 1.1.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_STAT, &msgds) == 0); 169 1.1.2.2 yamt 170 1.1.2.2 yamt ATF_CHECK(msgds.msg_qnum == 0); 171 1.1.2.2 yamt ATF_CHECK(msgds.msg_lspid == 0); 172 1.1.2.2 yamt ATF_CHECK(msgds.msg_lrpid == 0); 173 1.1.2.2 yamt ATF_CHECK(msgds.msg_rtime == 0); 174 1.1.2.2 yamt ATF_CHECK(msgds.msg_stime == 0); 175 1.1.2.2 yamt ATF_CHECK(msgds.msg_perm.uid == uid); 176 1.1.2.2 yamt ATF_CHECK(msgds.msg_perm.gid == gid); 177 1.1.2.2 yamt ATF_CHECK(msgds.msg_perm.cuid == uid); 178 1.1.2.2 yamt ATF_CHECK(msgds.msg_perm.cgid == gid); 179 1.1.2.2 yamt ATF_CHECK(msgds.msg_perm.mode == 0600); 180 1.1.2.2 yamt 181 1.1.2.3 yamt if (llabs(t - msgds.msg_ctime) > 5) 182 1.1.2.2 yamt atf_tc_fail("msgget(2) initialized current time incorrectly"); 183 1.1.2.2 yamt 184 1.1.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 185 1.1.2.2 yamt } 186 1.1.2.2 yamt 187 1.1.2.2 yamt ATF_TC_CLEANUP(msgget_init, tc) 188 1.1.2.2 yamt { 189 1.1.2.2 yamt clean(); 190 1.1.2.2 yamt } 191 1.1.2.2 yamt 192 1.1.2.2 yamt ATF_TC(msgget_limit); 193 1.1.2.2 yamt ATF_TC_HEAD(msgget_limit, tc) 194 1.1.2.2 yamt { 195 1.1.2.2 yamt atf_tc_set_md_var(tc, "descr", "Test msgget(2) against system limits"); 196 1.1.2.2 yamt } 197 1.1.2.2 yamt 198 1.1.2.2 yamt ATF_TC_BODY(msgget_limit, tc) 199 1.1.2.2 yamt { 200 1.1.2.2 yamt size_t len = sizeof(int); 201 1.1.2.2 yamt bool fail = false; 202 1.1.2.2 yamt int i, lim = 0; 203 1.1.2.2 yamt int *buf; 204 1.1.2.2 yamt 205 1.1.2.2 yamt if (sysctlbyname("kern.ipc.msgmni", &lim, &len, NULL, 0) != 0) 206 1.1.2.2 yamt atf_tc_skip("failed to read kern.ipc.msgmni sysctl"); 207 1.1.2.2 yamt 208 1.1.2.2 yamt buf = calloc(lim + 1, sizeof(*buf)); 209 1.1.2.2 yamt ATF_REQUIRE(buf != NULL); 210 1.1.2.2 yamt 211 1.1.2.2 yamt for (i = 0; i < lim; i++) { 212 1.1.2.2 yamt 213 1.1.2.2 yamt buf[i] = msgget(MSG_KEY + i, IPC_CREAT | IPC_EXCL | 0600); 214 1.1.2.2 yamt 215 1.1.2.2 yamt (void)fprintf(stderr, "key[%d] = %d\n", i, buf[i]); 216 1.1.2.2 yamt 217 1.1.2.2 yamt /* 218 1.1.2.2 yamt * This test only works when there are zero existing 219 1.1.2.2 yamt * message queues. Thus, bypass the unit test when 220 1.1.2.2 yamt * this precondition is not met, for reason or another. 221 1.1.2.2 yamt */ 222 1.1.2.2 yamt if (buf[i] == -1) 223 1.1.2.2 yamt goto out; 224 1.1.2.2 yamt } 225 1.1.2.2 yamt 226 1.1.2.2 yamt i++; 227 1.1.2.2 yamt errno = 0; 228 1.1.2.2 yamt 229 1.1.2.2 yamt buf[i] = msgget(MSG_KEY + i, IPC_CREAT | IPC_EXCL | 0600); 230 1.1.2.2 yamt 231 1.1.2.2 yamt if (buf[i] != -1 || errno != ENOSPC) 232 1.1.2.2 yamt fail = true; 233 1.1.2.2 yamt 234 1.1.2.2 yamt out: /* Remember to clean-up. */ 235 1.1.2.2 yamt for (i = 0; i < lim; i++) 236 1.1.2.2 yamt (void)msgctl(buf[i], IPC_RMID, 0); 237 1.1.2.2 yamt 238 1.1.2.2 yamt free(buf); 239 1.1.2.2 yamt 240 1.1.2.2 yamt if (fail != false) 241 1.1.2.2 yamt atf_tc_fail("msgget(2) opened more than %d queues", lim); 242 1.1.2.2 yamt } 243 1.1.2.2 yamt 244 1.1.2.2 yamt ATF_TC_WITH_CLEANUP(msgget_mode); 245 1.1.2.2 yamt ATF_TC_HEAD(msgget_mode, tc) 246 1.1.2.2 yamt { 247 1.1.2.2 yamt atf_tc_set_md_var(tc, "descr", "Test different modes with msgget(2)"); 248 1.1.2.2 yamt atf_tc_set_md_var(tc, "require.user", "root"); 249 1.1.2.2 yamt } 250 1.1.2.2 yamt 251 1.1.2.2 yamt ATF_TC_BODY(msgget_mode, tc) 252 1.1.2.2 yamt { 253 1.1.2.2 yamt static const mode_t mode[] = { 254 1.1.2.2 yamt S_IRWXU, S_IRUSR, S_IWUSR, S_IXUSR, S_IRWXG, S_IRGRP, 255 1.1.2.2 yamt S_IWGRP, S_IXGRP, S_IRWXO, S_IROTH, S_IWOTH, S_IXOTH 256 1.1.2.2 yamt }; 257 1.1.2.2 yamt 258 1.1.2.2 yamt struct msqid_ds msgds; 259 1.1.2.2 yamt size_t i; 260 1.1.2.2 yamt int id; 261 1.1.2.2 yamt 262 1.1.2.2 yamt for (i = 0; i < __arraycount(mode); i++) { 263 1.1.2.2 yamt 264 1.1.2.2 yamt (void)fprintf(stderr, "testing mode %d\n", mode[i]); 265 1.1.2.2 yamt (void)memset(&msgds, 0, sizeof(struct msqid_ds)); 266 1.1.2.2 yamt 267 1.1.2.2 yamt id = msgget(MSG_KEY, IPC_CREAT | IPC_EXCL | (int)mode[i]); 268 1.1.2.2 yamt 269 1.1.2.2 yamt ATF_REQUIRE(id != -1); 270 1.1.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_STAT, &msgds) == 0); 271 1.1.2.2 yamt ATF_REQUIRE(msgds.msg_perm.mode == mode[i]); 272 1.1.2.2 yamt ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 273 1.1.2.2 yamt } 274 1.1.2.2 yamt } 275 1.1.2.2 yamt 276 1.1.2.2 yamt ATF_TC_CLEANUP(msgget_mode, tc) 277 1.1.2.2 yamt { 278 1.1.2.2 yamt clean(); 279 1.1.2.2 yamt } 280 1.1.2.2 yamt 281 1.1.2.2 yamt 282 1.1.2.2 yamt ATF_TP_ADD_TCS(tp) 283 1.1.2.2 yamt { 284 1.1.2.2 yamt 285 1.1.2.2 yamt ATF_TP_ADD_TC(tp, msgget_excl); 286 1.1.2.2 yamt ATF_TP_ADD_TC(tp, msgget_exit); 287 1.1.2.2 yamt ATF_TP_ADD_TC(tp, msgget_init); 288 1.1.2.2 yamt ATF_TP_ADD_TC(tp, msgget_limit); 289 1.1.2.2 yamt ATF_TP_ADD_TC(tp, msgget_mode); 290 1.1.2.2 yamt 291 1.1.2.2 yamt return atf_no_error(); 292 1.1.2.2 yamt } 293
Indexes created Wed Oct 01 19:09:53 GMT 2025