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>
51 #define MSG_MTYPE_1 0x41
52 #define MSG_MTYPE_2 0x42
53 #define MSG_MTYPE_3 0x43
60 static void clean(void);
67 if ((id
= msgget(MSG_KEY
, 0)) != -1)
68 (void)msgctl(id
, IPC_RMID
, 0);
71 ATF_TC_WITH_CLEANUP(msgsnd_block
);
72 ATF_TC_HEAD(msgsnd_block
, tc
)
74 atf_tc_set_md_var(tc
, "descr", "Test that msgsnd(2) blocks");
75 atf_tc_set_md_var(tc
, "timeout", "10");
78 ATF_TC_BODY(msgsnd_block
, tc
)
80 struct msg msg
= { MSG_MTYPE_1
, { 'a', 'b', 'c' } };
84 id
= msgget(MSG_KEY
, IPC_CREAT
| 0600);
85 ATF_REQUIRE(id
!= -1);
88 ATF_REQUIRE(pid
>= 0);
93 * Enqueue messages until some limit (e.g. the maximum
94 * number of messages in the queue or the maximum number
95 * of bytes in the queue) is reached. After this the call
96 * should block when the IPC_NOWAIT is not set.
100 if (msgsnd(id
, &msg
, sizeof(struct msg
), 0) < 0)
106 (void)kill(pid
, SIGKILL
);
109 if (WIFEXITED(sta
) != 0 || WIFSIGNALED(sta
) == 0)
110 atf_tc_fail("msgsnd(2) did not block");
112 ATF_REQUIRE(msgctl(id
, IPC_RMID
, 0) == 0);
115 ATF_TC_CLEANUP(msgsnd_block
, tc
)
120 ATF_TC_WITH_CLEANUP(msgsnd_count
);
121 ATF_TC_HEAD(msgsnd_count
, tc
)
123 atf_tc_set_md_var(tc
, "descr",
124 "Test that msgsnd(2) increments the amount of "
125 "message in the queue, as given by msgctl(2)");
126 atf_tc_set_md_var(tc
, "timeout", "10");
129 ATF_TC_BODY(msgsnd_count
, tc
)
131 struct msg msg
= { MSG_MTYPE_1
, { 'a', 'b', 'c' } };
136 id
= msgget(MSG_KEY
, IPC_CREAT
| 0600);
137 ATF_REQUIRE(id
!= -1);
142 rv
= msgsnd(id
, &msg
, sizeof(struct msg
), IPC_NOWAIT
);
149 if (rv
== -1 && errno
== EAGAIN
)
152 atf_tc_fail("failed to enqueue a message");
155 (void)memset(&ds
, 0, sizeof(struct msqid_ds
));
156 (void)msgctl(id
, IPC_STAT
, &ds
);
158 if (ds
.msg_qnum
!= i
)
159 atf_tc_fail("incorrect message count");
161 ATF_REQUIRE(msgctl(id
, IPC_RMID
, 0) == 0);
164 ATF_TC_CLEANUP(msgsnd_count
, tc
)
169 ATF_TC_WITH_CLEANUP(msgsnd_err
);
170 ATF_TC_HEAD(msgsnd_err
, tc
)
172 atf_tc_set_md_var(tc
, "descr", "Test errors from msgsnd(2)");
175 ATF_TC_BODY(msgsnd_err
, tc
)
177 struct msg msg
= { MSG_MTYPE_1
, { 'a', 'b', 'c' } };
180 id
= msgget(MSG_KEY
, IPC_CREAT
| 0600);
181 ATF_REQUIRE(id
!= -1);
185 ATF_REQUIRE_ERRNO(EFAULT
, msgsnd(id
, (void *)-1,
186 sizeof(struct msg
), IPC_NOWAIT
) == -1);
190 ATF_REQUIRE_ERRNO(EINVAL
, msgsnd(-1, &msg
,
191 sizeof(struct msg
), IPC_NOWAIT
) == -1);
195 ATF_REQUIRE_ERRNO(EINVAL
, msgsnd(-1, &msg
,
196 SSIZE_MAX
, IPC_NOWAIT
) == -1);
201 ATF_REQUIRE_ERRNO(EINVAL
, msgsnd(id
, &msg
,
202 sizeof(struct msg
), IPC_NOWAIT
) == -1);
204 ATF_REQUIRE(msgctl(id
, IPC_RMID
, 0) == 0);
207 ATF_TC_CLEANUP(msgsnd_err
, tc
)
212 ATF_TC_WITH_CLEANUP(msgsnd_nonblock
);
213 ATF_TC_HEAD(msgsnd_nonblock
, tc
)
215 atf_tc_set_md_var(tc
, "descr", "Test msgsnd(2) with IPC_NOWAIT");
216 atf_tc_set_md_var(tc
, "timeout", "10");
219 ATF_TC_BODY(msgsnd_nonblock
, tc
)
221 struct msg msg
= { MSG_MTYPE_1
, { 'a', 'b', 'c' } };
225 id
= msgget(MSG_KEY
, IPC_CREAT
| 0600);
226 ATF_REQUIRE(id
!= -1);
229 ATF_REQUIRE(pid
>= 0);
236 rv
= msgsnd(id
, &msg
, sizeof(struct msg
), IPC_NOWAIT
);
238 if (rv
== -1 && errno
== EAGAIN
)
244 (void)kill(pid
, SIGKILL
);
247 if (WIFEXITED(sta
) == 0 || WIFSIGNALED(sta
) != 0)
248 atf_tc_fail("msgsnd(2) blocked with IPC_NOWAIT");
250 ATF_REQUIRE(msgctl(id
, IPC_RMID
, 0) == 0);
253 ATF_TC_CLEANUP(msgsnd_nonblock
, tc
)
258 ATF_TC_WITH_CLEANUP(msgsnd_perm
);
259 ATF_TC_HEAD(msgsnd_perm
, tc
)
261 atf_tc_set_md_var(tc
, "descr", "Test permissions with msgsnd(2)");
262 atf_tc_set_md_var(tc
, "require.user", "root");
265 ATF_TC_BODY(msgsnd_perm
, tc
)
267 struct msg msg
= { MSG_MTYPE_1
, { 'a', 'b', 'c' } };
273 pw
= getpwnam("nobody");
274 id
= msgget(MSG_KEY
, IPC_CREAT
| 0600);
276 ATF_REQUIRE(id
!= -1);
277 ATF_REQUIRE(pw
!= NULL
);
280 ATF_REQUIRE(uid
!= 0);
283 ATF_REQUIRE(pid
>= 0);
288 * Try to enqueue a message to the queue
289 * created by root as RW for owner only.
291 if (setuid(uid
) != 0)
294 id
= msgget(MSG_KEY
, 0);
301 if (msgsnd(id
, &msg
, sizeof(struct msg
), IPC_NOWAIT
) == 0)
312 if (WIFEXITED(sta
) == 0 || WEXITSTATUS(sta
) != EXIT_SUCCESS
) {
314 if (errno
== EX_OSERR
)
315 atf_tc_fail("system call failed");
317 atf_tc_fail("UID %u enqueued message to root's queue", uid
);
320 ATF_REQUIRE(msgctl(id
, IPC_RMID
, 0) == 0);
323 ATF_TC_CLEANUP(msgsnd_perm
, tc
)
331 ATF_TP_ADD_TC(tp
, msgsnd_block
);
332 ATF_TP_ADD_TC(tp
, msgsnd_count
);
333 ATF_TP_ADD_TC(tp
, msgsnd_err
);
334 ATF_TP_ADD_TC(tp
, msgsnd_nonblock
);
335 ATF_TP_ADD_TC(tp
, msgsnd_perm
);
337 return atf_no_error();