1272343Sngie/* $NetBSD: t_msgget.c,v 1.2 2014/02/27 00:59:50 joerg Exp $ */ 2272343Sngie 3272343Sngie/*- 4272343Sngie * Copyright (c) 2011 The NetBSD Foundation, Inc. 5272343Sngie * All rights reserved. 6272343Sngie * 7272343Sngie * This code is derived from software contributed to The NetBSD Foundation 8272343Sngie * by Jukka Ruohonen. 9272343Sngie * 10272343Sngie * Redistribution and use in source and binary forms, with or without 11272343Sngie * modification, are permitted provided that the following conditions 12272343Sngie * are met: 13272343Sngie * 1. Redistributions of source code must retain the above copyright 14272343Sngie * notice, this list of conditions and the following disclaimer. 15272343Sngie * 2. Redistributions in binary form must reproduce the above copyright 16272343Sngie * notice, this list of conditions and the following disclaimer in the 17272343Sngie * documentation and/or other materials provided with the distribution. 18272343Sngie * 19272343Sngie * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20272343Sngie * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21272343Sngie * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22272343Sngie * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23272343Sngie * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24272343Sngie * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25272343Sngie * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26272343Sngie * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27272343Sngie * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28272343Sngie * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29272343Sngie * POSSIBILITY OF SUCH DAMAGE. 30272343Sngie */ 31272343Sngie#include <sys/cdefs.h> 32272343Sngie__RCSID("$NetBSD: t_msgget.c,v 1.2 2014/02/27 00:59:50 joerg Exp $"); 33272343Sngie 34272343Sngie#include <sys/msg.h> 35272343Sngie#include <sys/stat.h> 36272343Sngie#include <sys/sysctl.h> 37272343Sngie#include <sys/wait.h> 38272343Sngie 39272343Sngie#include <atf-c.h> 40272343Sngie#include <errno.h> 41272343Sngie#include <pwd.h> 42272343Sngie#include <stdio.h> 43272343Sngie#include <stdlib.h> 44272343Sngie#include <string.h> 45272343Sngie#include <sysexits.h> 46272343Sngie#include <time.h> 47272343Sngie#include <unistd.h> 48272343Sngie 49272343Sngie#define MSG_KEY 12345689 50272343Sngie 51272343Sngiestatic void clean(void); 52272343Sngie 53272343Sngiestatic void 54272343Sngieclean(void) 55272343Sngie{ 56272343Sngie int id; 57272343Sngie 58272343Sngie if ((id = msgget(MSG_KEY, 0)) != -1) 59272343Sngie (void)msgctl(id, IPC_RMID, 0); 60272343Sngie} 61272343Sngie 62272343SngieATF_TC_WITH_CLEANUP(msgget_excl); 63272343SngieATF_TC_HEAD(msgget_excl, tc) 64272343Sngie{ 65272343Sngie atf_tc_set_md_var(tc, "descr", "Test msgget(2) with IPC_EXCL"); 66272343Sngie} 67272343Sngie 68272343SngieATF_TC_BODY(msgget_excl, tc) 69272343Sngie{ 70272343Sngie int id; 71272343Sngie 72272343Sngie /* 73272343Sngie * Create a message queue and re-open it with 74272343Sngie * O_CREAT and IPC_EXCL set. This should fail. 75272343Sngie */ 76272343Sngie id = msgget(MSG_KEY, IPC_CREAT | 0600); 77272343Sngie 78272343Sngie if (id < 0) 79272343Sngie atf_tc_fail("failed to create message queue"); 80272343Sngie 81272343Sngie errno = 0; 82272343Sngie 83272343Sngie if (msgget(MSG_KEY, IPC_CREAT | IPC_EXCL | 0600) != -1) 84272343Sngie atf_tc_fail("msgget(2) failed for IPC_EXCL"); 85272343Sngie 86272343Sngie ATF_REQUIRE(errno == EEXIST); 87272343Sngie 88272343Sngie /* 89272343Sngie * However, the same call should succeed 90272343Sngie * when IPC_EXCL is not set in the flags. 91272343Sngie */ 92272343Sngie id = msgget(MSG_KEY, IPC_CREAT | 0600); 93272343Sngie 94272343Sngie if (id < 0) 95272343Sngie atf_tc_fail("msgget(2) failed to re-open"); 96272343Sngie 97272343Sngie ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 98272343Sngie} 99272343Sngie 100272343SngieATF_TC_CLEANUP(msgget_excl, tc) 101272343Sngie{ 102272343Sngie clean(); 103272343Sngie} 104272343Sngie 105272343SngieATF_TC_WITH_CLEANUP(msgget_exit); 106272343SngieATF_TC_HEAD(msgget_exit, tc) 107272343Sngie{ 108272343Sngie atf_tc_set_md_var(tc, "descr", 109272343Sngie "Test that XSI message queues are " 110272343Sngie "not removed when the process exits"); 111272343Sngie} 112272343Sngie 113272343SngieATF_TC_BODY(msgget_exit, tc) 114272343Sngie{ 115272343Sngie int id, sta; 116272343Sngie pid_t pid; 117272343Sngie 118272343Sngie pid = fork(); 119272343Sngie ATF_REQUIRE(pid >= 0); 120272343Sngie 121272343Sngie if (pid == 0) { 122272343Sngie 123272343Sngie if (msgget(MSG_KEY, IPC_CREAT | IPC_EXCL | 0600) == -1) 124272343Sngie _exit(EXIT_FAILURE); 125272343Sngie 126272343Sngie _exit(EXIT_SUCCESS); 127272343Sngie } 128272343Sngie 129272343Sngie (void)wait(&sta); 130272343Sngie 131272343Sngie if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS) 132272343Sngie atf_tc_fail("failed to create message queue"); 133272343Sngie 134272343Sngie id = msgget(MSG_KEY, 0); 135272343Sngie 136272343Sngie if (id == -1) 137272343Sngie atf_tc_fail("message queue was removed on process exit"); 138272343Sngie 139272343Sngie ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 140272343Sngie} 141272343Sngie 142272343SngieATF_TC_CLEANUP(msgget_exit, tc) 143272343Sngie{ 144272343Sngie clean(); 145272343Sngie} 146272343Sngie 147272343SngieATF_TC_WITH_CLEANUP(msgget_init); 148272343SngieATF_TC_HEAD(msgget_init, tc) 149272343Sngie{ 150272343Sngie atf_tc_set_md_var(tc, "descr", 151272343Sngie "Test that msgget(2) initializes data structures properly"); 152272343Sngie} 153272343Sngie 154272343SngieATF_TC_BODY(msgget_init, tc) 155272343Sngie{ 156272343Sngie const uid_t uid = geteuid(); 157272343Sngie const gid_t gid = getegid(); 158272343Sngie struct msqid_ds msgds; 159272343Sngie time_t t; 160272343Sngie int id; 161272343Sngie 162272343Sngie (void)memset(&msgds, 0x9, sizeof(struct msqid_ds)); 163272343Sngie 164272343Sngie t = time(NULL); 165272343Sngie id = msgget(MSG_KEY, IPC_CREAT | 0600); 166272343Sngie 167272343Sngie ATF_REQUIRE(id !=-1); 168272343Sngie ATF_REQUIRE(msgctl(id, IPC_STAT, &msgds) == 0); 169272343Sngie 170272343Sngie ATF_CHECK(msgds.msg_qnum == 0); 171272343Sngie ATF_CHECK(msgds.msg_lspid == 0); 172272343Sngie ATF_CHECK(msgds.msg_lrpid == 0); 173272343Sngie ATF_CHECK(msgds.msg_rtime == 0); 174272343Sngie ATF_CHECK(msgds.msg_stime == 0); 175272343Sngie ATF_CHECK(msgds.msg_perm.uid == uid); 176272343Sngie ATF_CHECK(msgds.msg_perm.gid == gid); 177272343Sngie ATF_CHECK(msgds.msg_perm.cuid == uid); 178272343Sngie ATF_CHECK(msgds.msg_perm.cgid == gid); 179272343Sngie ATF_CHECK(msgds.msg_perm.mode == 0600); 180272343Sngie 181272343Sngie if (llabs(t - msgds.msg_ctime) > 5) 182272343Sngie atf_tc_fail("msgget(2) initialized current time incorrectly"); 183272343Sngie 184272343Sngie ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 185272343Sngie} 186272343Sngie 187272343SngieATF_TC_CLEANUP(msgget_init, tc) 188272343Sngie{ 189272343Sngie clean(); 190272343Sngie} 191272343Sngie 192272343SngieATF_TC(msgget_limit); 193272343SngieATF_TC_HEAD(msgget_limit, tc) 194272343Sngie{ 195272343Sngie atf_tc_set_md_var(tc, "descr", "Test msgget(2) against system limits"); 196272343Sngie} 197272343Sngie 198272343SngieATF_TC_BODY(msgget_limit, tc) 199272343Sngie{ 200272343Sngie size_t len = sizeof(int); 201272343Sngie bool fail = false; 202272343Sngie int i, lim = 0; 203272343Sngie int *buf; 204272343Sngie 205272343Sngie if (sysctlbyname("kern.ipc.msgmni", &lim, &len, NULL, 0) != 0) 206272343Sngie atf_tc_skip("failed to read kern.ipc.msgmni sysctl"); 207272343Sngie 208272343Sngie buf = calloc(lim + 1, sizeof(*buf)); 209272343Sngie ATF_REQUIRE(buf != NULL); 210272343Sngie 211272343Sngie for (i = 0; i < lim; i++) { 212272343Sngie 213272343Sngie buf[i] = msgget(MSG_KEY + i, IPC_CREAT | IPC_EXCL | 0600); 214272343Sngie 215272343Sngie (void)fprintf(stderr, "key[%d] = %d\n", i, buf[i]); 216272343Sngie 217272343Sngie /* 218272343Sngie * This test only works when there are zero existing 219272343Sngie * message queues. Thus, bypass the unit test when 220272343Sngie * this precondition is not met, for reason or another. 221272343Sngie */ 222272343Sngie if (buf[i] == -1) 223272343Sngie goto out; 224272343Sngie } 225272343Sngie 226272343Sngie i++; 227272343Sngie errno = 0; 228272343Sngie 229272343Sngie buf[i] = msgget(MSG_KEY + i, IPC_CREAT | IPC_EXCL | 0600); 230272343Sngie 231272343Sngie if (buf[i] != -1 || errno != ENOSPC) 232272343Sngie fail = true; 233272343Sngie 234272343Sngieout: /* Remember to clean-up. */ 235272343Sngie for (i = 0; i < lim; i++) 236272343Sngie (void)msgctl(buf[i], IPC_RMID, 0); 237272343Sngie 238272343Sngie free(buf); 239272343Sngie 240272343Sngie if (fail != false) 241272343Sngie atf_tc_fail("msgget(2) opened more than %d queues", lim); 242272343Sngie} 243272343Sngie 244272343SngieATF_TC_WITH_CLEANUP(msgget_mode); 245272343SngieATF_TC_HEAD(msgget_mode, tc) 246272343Sngie{ 247272343Sngie atf_tc_set_md_var(tc, "descr", "Test different modes with msgget(2)"); 248272343Sngie atf_tc_set_md_var(tc, "require.user", "root"); 249272343Sngie} 250272343Sngie 251272343SngieATF_TC_BODY(msgget_mode, tc) 252272343Sngie{ 253272343Sngie static const mode_t mode[] = { 254272343Sngie S_IRWXU, S_IRUSR, S_IWUSR, S_IXUSR, S_IRWXG, S_IRGRP, 255272343Sngie S_IWGRP, S_IXGRP, S_IRWXO, S_IROTH, S_IWOTH, S_IXOTH 256272343Sngie }; 257272343Sngie 258272343Sngie struct msqid_ds msgds; 259272343Sngie size_t i; 260272343Sngie int id; 261272343Sngie 262272343Sngie for (i = 0; i < __arraycount(mode); i++) { 263272343Sngie 264272343Sngie (void)fprintf(stderr, "testing mode %d\n", mode[i]); 265272343Sngie (void)memset(&msgds, 0, sizeof(struct msqid_ds)); 266272343Sngie 267272343Sngie id = msgget(MSG_KEY, IPC_CREAT | IPC_EXCL | (int)mode[i]); 268272343Sngie 269272343Sngie ATF_REQUIRE(id != -1); 270272343Sngie ATF_REQUIRE(msgctl(id, IPC_STAT, &msgds) == 0); 271272343Sngie ATF_REQUIRE(msgds.msg_perm.mode == mode[i]); 272272343Sngie ATF_REQUIRE(msgctl(id, IPC_RMID, 0) == 0); 273272343Sngie } 274272343Sngie} 275272343Sngie 276272343SngieATF_TC_CLEANUP(msgget_mode, tc) 277272343Sngie{ 278272343Sngie clean(); 279272343Sngie} 280272343Sngie 281272343Sngie 282272343SngieATF_TP_ADD_TCS(tp) 283272343Sngie{ 284272343Sngie 285272343Sngie ATF_TP_ADD_TC(tp, msgget_excl); 286272343Sngie ATF_TP_ADD_TC(tp, msgget_exit); 287272343Sngie ATF_TP_ADD_TC(tp, msgget_init); 288272343Sngie ATF_TP_ADD_TC(tp, msgget_limit); 289272343Sngie ATF_TP_ADD_TC(tp, msgget_mode); 290272343Sngie 291272343Sngie return atf_no_error(); 292272343Sngie} 293