1 /* $NetBSD: t_msgsnd.c,v 1.2 2011/11/05 08:47:54 jruoho Exp $ */
4 * Copyright (c) 2011 The NetBSD Foundation, Inc.
7 * This code is derived from software contributed to The NetBSD Foundation
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
31 #include <sys/cdefs.h>
32 __RCSID("$NetBSD: t_msgsnd.c,v 1.2 2011/11/05 08:47:54 jruoho Exp $");
36 #include <sys/sysctl.h>
55 #define MSG_MTYPE_1 0x41
56 #define MSG_MTYPE_2 0x42
57 #define MSG_MTYPE_3 0x43
64 static void clean(void);
71 if ((id = msgget(MSG_KEY, 0)) != -1)
72 (void)msgctl(id, IPC_RMID, 0);
75 ATF_TC_WITH_CLEANUP(msgsnd_block);
76 ATF_TC_HEAD(msgsnd_block, tc)
78 atf_tc_set_md_var(tc, "descr", "Test that msgsnd(2) blocks");
79 atf_tc_set_md_var(tc, "timeout", "10");
82 ATF_TC_BODY(msgsnd_block, tc)
84 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
88 id = msgget(MSG_KEY, IPC_CREAT | 0600);
89 ATF_REQUIRE(id != -1);
92 ATF_REQUIRE(pid >= 0);
97 * Enqueue messages until some limit (e.g. the maximum
98 * number of messages in the queue or the maximum number
99 * of bytes in the queue) is reached. After this the call
100 * should block when the IPC_NOWAIT is not set.
104 if (msgsnd(id, &msg, sizeof(struct msg), 0) < 0)
110 (void)kill(pid, SIGKILL);
113 if (WIFEXITED(sta) != 0 || WIFSIGNALED(sta) == 0)
114 atf_tc_fail("msgsnd(2) did not block");
116 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
119 ATF_TC_CLEANUP(msgsnd_block, tc)
124 ATF_TC_WITH_CLEANUP(msgsnd_count);
125 ATF_TC_HEAD(msgsnd_count, tc)
127 atf_tc_set_md_var(tc, "descr",
128 "Test that msgsnd(2) increments the amount of "
129 "message in the queue, as given by msgctl(2)");
130 atf_tc_set_md_var(tc, "timeout", "10");
133 ATF_TC_BODY(msgsnd_count, tc)
135 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
140 id = msgget(MSG_KEY, IPC_CREAT | 0600);
141 ATF_REQUIRE(id != -1);
146 rv = msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT);
153 if (rv == -1 && errno == EAGAIN)
156 atf_tc_fail("failed to enqueue a message");
159 (void)memset(&ds, 0, sizeof(struct msqid_ds));
160 (void)msgctl(id, IPC_STAT, &ds);
162 if (ds.msg_qnum != i)
163 atf_tc_fail("incorrect message count");
165 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
168 ATF_TC_CLEANUP(msgsnd_count, tc)
173 ATF_TC_WITH_CLEANUP(msgsnd_err);
174 ATF_TC_HEAD(msgsnd_err, tc)
176 atf_tc_set_md_var(tc, "descr", "Test errors from msgsnd(2)");
179 ATF_TC_BODY(msgsnd_err, tc)
181 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
184 id = msgget(MSG_KEY, IPC_CREAT | 0600);
185 ATF_REQUIRE(id != -1);
189 ATF_REQUIRE_ERRNO(EFAULT, msgsnd(id, (void *)-1,
190 sizeof(struct msg), IPC_NOWAIT) == -1);
194 ATF_REQUIRE_ERRNO(EINVAL, msgsnd(-1, &msg,
195 sizeof(struct msg), IPC_NOWAIT) == -1);
199 ATF_REQUIRE_ERRNO(EINVAL, msgsnd(-1, &msg,
200 SSIZE_MAX, IPC_NOWAIT) == -1);
205 ATF_REQUIRE_ERRNO(EINVAL, msgsnd(id, &msg,
206 sizeof(struct msg), IPC_NOWAIT) == -1);
208 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
211 ATF_TC_CLEANUP(msgsnd_err, tc)
216 ATF_TC_WITH_CLEANUP(msgsnd_nonblock);
217 ATF_TC_HEAD(msgsnd_nonblock, tc)
219 atf_tc_set_md_var(tc, "descr", "Test msgsnd(2) with IPC_NOWAIT");
220 atf_tc_set_md_var(tc, "timeout", "10");
223 ATF_TC_BODY(msgsnd_nonblock, tc)
225 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
229 id = msgget(MSG_KEY, IPC_CREAT | 0600);
230 ATF_REQUIRE(id != -1);
233 ATF_REQUIRE(pid >= 0);
240 rv = msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT);
242 if (rv == -1 && errno == EAGAIN)
248 (void)kill(pid, SIGKILL);
251 if (WIFEXITED(sta) == 0 || WIFSIGNALED(sta) != 0)
252 atf_tc_fail("msgsnd(2) blocked with IPC_NOWAIT");
254 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
257 ATF_TC_CLEANUP(msgsnd_nonblock, tc)
262 ATF_TC_WITH_CLEANUP(msgsnd_perm);
263 ATF_TC_HEAD(msgsnd_perm, tc)
265 atf_tc_set_md_var(tc, "descr", "Test permissions with msgsnd(2)");
266 atf_tc_set_md_var(tc, "require.user", "root");
269 ATF_TC_BODY(msgsnd_perm, tc)
271 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
277 pw = getpwnam("nobody");
278 id = msgget(MSG_KEY, IPC_CREAT | 0600);
280 ATF_REQUIRE(id != -1);
281 ATF_REQUIRE(pw != NULL);
284 ATF_REQUIRE(uid != 0);
287 ATF_REQUIRE(pid >= 0);
292 * Try to enqueue a message to the queue
293 * created by root as RW for owner only.
295 if (setuid(uid) != 0)
298 id = msgget(MSG_KEY, 0);
305 if (msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT) == 0)
316 if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS) {
318 if (errno == EX_OSERR)
319 atf_tc_fail("system call failed");
321 atf_tc_fail("UID %u enqueued message to root's queue", uid);
324 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
327 ATF_TC_CLEANUP(msgsnd_perm, tc)
335 ATF_TP_ADD_TC(tp, msgsnd_block);
336 ATF_TP_ADD_TC(tp, msgsnd_count);
337 ATF_TP_ADD_TC(tp, msgsnd_err);
338 ATF_TP_ADD_TC(tp, msgsnd_nonblock);
339 ATF_TP_ADD_TC(tp, msgsnd_perm);
341 return atf_no_error();