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.
102 if (msgsnd(id, &msg, sizeof(msg.buf), 0) < 0)
104 if (msgsnd(id, &msg, sizeof(struct msg), 0) < 0)
111 (void)kill(pid, SIGKILL);
114 if (WIFEXITED(sta) != 0 || WIFSIGNALED(sta) == 0)
115 atf_tc_fail("msgsnd(2) did not block");
117 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
120 ATF_TC_CLEANUP(msgsnd_block, tc)
125 ATF_TC_WITH_CLEANUP(msgsnd_count);
126 ATF_TC_HEAD(msgsnd_count, tc)
128 atf_tc_set_md_var(tc, "descr",
129 "Test that msgsnd(2) increments the amount of "
130 "message in the queue, as given by msgctl(2)");
131 atf_tc_set_md_var(tc, "timeout", "10");
134 ATF_TC_BODY(msgsnd_count, tc)
136 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
141 id = msgget(MSG_KEY, IPC_CREAT | 0600);
142 ATF_REQUIRE(id != -1);
148 rv = msgsnd(id, &msg, sizeof(msg.buf), IPC_NOWAIT);
150 rv = msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT);
158 if (rv == -1 && errno == EAGAIN)
161 atf_tc_fail("failed to enqueue a message");
164 (void)memset(&ds, 0, sizeof(struct msqid_ds));
165 (void)msgctl(id, IPC_STAT, &ds);
167 if (ds.msg_qnum != i)
168 atf_tc_fail("incorrect message count");
170 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
173 ATF_TC_CLEANUP(msgsnd_count, tc)
178 ATF_TC_WITH_CLEANUP(msgsnd_err);
179 ATF_TC_HEAD(msgsnd_err, tc)
181 atf_tc_set_md_var(tc, "descr", "Test errors from msgsnd(2)");
184 ATF_TC_BODY(msgsnd_err, tc)
186 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
189 id = msgget(MSG_KEY, IPC_CREAT | 0600);
190 ATF_REQUIRE(id != -1);
194 ATF_REQUIRE_ERRNO(EFAULT, msgsnd(id, (void *)-1,
196 sizeof(msg.buf), IPC_NOWAIT) == -1);
198 sizeof(struct msg), IPC_NOWAIT) == -1);
203 ATF_REQUIRE_ERRNO(EINVAL, msgsnd(-1, &msg,
205 sizeof(msg.buf), IPC_NOWAIT) == -1);
207 sizeof(struct msg), IPC_NOWAIT) == -1);
212 ATF_REQUIRE_ERRNO(EINVAL, msgsnd(-1, &msg,
213 SSIZE_MAX, IPC_NOWAIT) == -1);
218 ATF_REQUIRE_ERRNO(EINVAL, msgsnd(id, &msg,
220 sizeof(msg.buf), IPC_NOWAIT) == -1);
222 sizeof(struct msg), IPC_NOWAIT) == -1);
225 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
228 ATF_TC_CLEANUP(msgsnd_err, tc)
233 ATF_TC_WITH_CLEANUP(msgsnd_nonblock);
234 ATF_TC_HEAD(msgsnd_nonblock, tc)
236 atf_tc_set_md_var(tc, "descr", "Test msgsnd(2) with IPC_NOWAIT");
237 atf_tc_set_md_var(tc, "timeout", "10");
240 ATF_TC_BODY(msgsnd_nonblock, tc)
242 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
246 id = msgget(MSG_KEY, IPC_CREAT | 0600);
247 ATF_REQUIRE(id != -1);
250 ATF_REQUIRE(pid >= 0);
258 rv = msgsnd(id, &msg, sizeof(msg.buf), IPC_NOWAIT);
260 rv = msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT);
263 if (rv == -1 && errno == EAGAIN)
269 (void)kill(pid, SIGKILL);
272 if (WIFEXITED(sta) == 0 || WIFSIGNALED(sta) != 0)
273 atf_tc_fail("msgsnd(2) blocked with IPC_NOWAIT");
275 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
278 ATF_TC_CLEANUP(msgsnd_nonblock, tc)
283 ATF_TC_WITH_CLEANUP(msgsnd_perm);
284 ATF_TC_HEAD(msgsnd_perm, tc)
286 atf_tc_set_md_var(tc, "descr", "Test permissions with msgsnd(2)");
287 atf_tc_set_md_var(tc, "require.user", "root");
290 ATF_TC_BODY(msgsnd_perm, tc)
292 struct msg msg = { MSG_MTYPE_1, { 'a', 'b', 'c' } };
298 pw = getpwnam("nobody");
299 id = msgget(MSG_KEY, IPC_CREAT | 0600);
301 ATF_REQUIRE(id != -1);
302 ATF_REQUIRE(pw != NULL);
305 ATF_REQUIRE(uid != 0);
308 ATF_REQUIRE(pid >= 0);
313 * Try to enqueue a message to the queue
314 * created by root as RW for owner only.
316 if (setuid(uid) != 0)
319 id = msgget(MSG_KEY, 0);
327 if (msgsnd(id, &msg, sizeof(msg.buf), IPC_NOWAIT) == 0)
329 if (msgsnd(id, &msg, sizeof(struct msg), IPC_NOWAIT) == 0)
341 if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS) {
343 if (errno == EX_OSERR)
344 atf_tc_fail("system call failed");
346 atf_tc_fail("UID %u enqueued message to root's queue", uid);
349 ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0);
352 ATF_TC_CLEANUP(msgsnd_perm, tc)
360 ATF_TP_ADD_TC(tp, msgsnd_block);
361 ATF_TP_ADD_TC(tp, msgsnd_count);
362 ATF_TP_ADD_TC(tp, msgsnd_err);
363 ATF_TP_ADD_TC(tp, msgsnd_nonblock);
364 ATF_TP_ADD_TC(tp, msgsnd_perm);
366 return atf_no_error();