1 /* $NetBSD: t_msgsnd.c,v 1.3 2017/01/13 20:44:45 christos 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.3 2017/01/13 20:44:45 christos Exp $");
36 #include <sys/sysctl.h>
52 #define MSG_MTYPE_1 0x41
53 #define MSG_MTYPE_2 0x42
54 #define MSG_MTYPE_3 0x43
61 static void clean(void);
68 if ((id = msgget(MSG_KEY, 0)) != -1)
69 (void)msgctl(id, IPC_RMID, 0);
72 ATF_TC_WITH_CLEANUP(msgsnd_block);
73 ATF_TC_HEAD(msgsnd_block, tc)
75 atf_tc_set_md_var(tc, "descr", "Test that msgsnd(2) blocks");
76 atf_tc_set_md_var(tc, "timeout", "10");
79 ATF_TC_BODY(msgsnd_block, tc)
81 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
85 id = msgget(MSG_KEY, IPC_CREAT | 0600);
86 ATF_REQUIRE(id != -1);
89 ATF_REQUIRE(pid >= 0);
94 * Enqueue messages until some limit (e.g. the maximum
95 * number of messages in the queue or the maximum number
96 * of bytes in the queue) is reached. After this the call
97 * should block when the IPC_NOWAIT is not set.
101 if (msgsnd(id, &msg, sizeof(struct msg), 0) < 0)
107 (void)kill(pid, SIGKILL);
110 if (WIFEXITED(sta) != 0 || WIFSIGNALED(sta) == 0)
111 atf_tc_fail("msgsnd(2) did not block");
113 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
116 ATF_TC_CLEANUP(msgsnd_block, tc)
121 ATF_TC_WITH_CLEANUP(msgsnd_count);
122 ATF_TC_HEAD(msgsnd_count, tc)
124 atf_tc_set_md_var(tc, "descr",
125 "Test that msgsnd(2) increments the amount of "
126 "message in the queue, as given by msgctl(2)");
127 atf_tc_set_md_var(tc, "timeout", "10");
130 ATF_TC_BODY(msgsnd_count, tc)
132 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
137 id = msgget(MSG_KEY, IPC_CREAT | 0600);
138 ATF_REQUIRE(id != -1);
143 rv = msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT);
150 if (rv == -1 && errno == EAGAIN)
153 atf_tc_fail("failed to enqueue a message");
156 (void)memset(&ds, 0, sizeof(struct msqid_ds));
157 (void)msgctl(id, IPC_STAT, &ds);
159 if (ds.msg_qnum != i)
160 atf_tc_fail("incorrect message count");
162 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
165 ATF_TC_CLEANUP(msgsnd_count, tc)
170 ATF_TC_WITH_CLEANUP(msgsnd_err);
171 ATF_TC_HEAD(msgsnd_err, tc)
173 atf_tc_set_md_var(tc, "descr", "Test errors from msgsnd(2)");
176 ATF_TC_BODY(msgsnd_err, tc)
178 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
181 id = msgget(MSG_KEY, IPC_CREAT | 0600);
182 ATF_REQUIRE(id != -1);
186 ATF_REQUIRE_ERRNO(EFAULT, msgsnd(id, (void *)-1,
187 sizeof(struct msg), IPC_NOWAIT) == -1);
191 ATF_REQUIRE_ERRNO(EINVAL, msgsnd(-1, &msg,
192 sizeof(struct msg), IPC_NOWAIT) == -1);
196 ATF_REQUIRE_ERRNO(EINVAL, msgsnd(-1, &msg,
197 SSIZE_MAX, IPC_NOWAIT) == -1);
202 ATF_REQUIRE_ERRNO(EINVAL, msgsnd(id, &msg,
203 sizeof(struct msg), IPC_NOWAIT) == -1);
205 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
208 ATF_TC_CLEANUP(msgsnd_err, tc)
213 ATF_TC_WITH_CLEANUP(msgsnd_nonblock);
214 ATF_TC_HEAD(msgsnd_nonblock, tc)
216 atf_tc_set_md_var(tc, "descr", "Test msgsnd(2) with IPC_NOWAIT");
217 atf_tc_set_md_var(tc, "timeout", "10");
220 ATF_TC_BODY(msgsnd_nonblock, tc)
222 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
226 id = msgget(MSG_KEY, IPC_CREAT | 0600);
227 ATF_REQUIRE(id != -1);
230 ATF_REQUIRE(pid >= 0);
237 rv = msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT);
239 if (rv == -1 && errno == EAGAIN)
245 (void)kill(pid, SIGKILL);
248 if (WIFEXITED(sta) == 0 || WIFSIGNALED(sta) != 0)
249 atf_tc_fail("msgsnd(2) blocked with IPC_NOWAIT");
251 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
254 ATF_TC_CLEANUP(msgsnd_nonblock, tc)
259 ATF_TC_WITH_CLEANUP(msgsnd_perm);
260 ATF_TC_HEAD(msgsnd_perm, tc)
262 atf_tc_set_md_var(tc, "descr", "Test permissions with msgsnd(2)");
263 atf_tc_set_md_var(tc, "require.user", "root");
266 ATF_TC_BODY(msgsnd_perm, tc)
268 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
274 pw = getpwnam("nobody");
275 id = msgget(MSG_KEY, IPC_CREAT | 0600);
277 ATF_REQUIRE(id != -1);
278 ATF_REQUIRE(pw != NULL);
281 ATF_REQUIRE(uid != 0);
284 ATF_REQUIRE(pid >= 0);
289 * Try to enqueue a message to the queue
290 * created by root as RW for owner only.
292 if (setuid(uid) != 0)
295 id = msgget(MSG_KEY, 0);
302 if (msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT) == 0)
313 if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS) {
315 if (errno == EX_OSERR)
316 atf_tc_fail("system call failed");
318 atf_tc_fail("UID %u enqueued message to root's queue", uid);
321 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
324 ATF_TC_CLEANUP(msgsnd_perm, tc)
332 ATF_TP_ADD_TC(tp, msgsnd_block);
333 ATF_TP_ADD_TC(tp, msgsnd_count);
334 ATF_TP_ADD_TC(tp, msgsnd_err);
335 ATF_TP_ADD_TC(tp, msgsnd_nonblock);
336 ATF_TP_ADD_TC(tp, msgsnd_perm);
338 return atf_no_error();