1 /*
2 * Copyright (C) 2016 The Android Open Source Project
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * * Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in
12 * the documentation and/or other materials provided with the
13 * distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
18 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
19 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
21 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
22 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
23 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
24 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
25 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #include <gtest/gtest.h>
30
31 #include <errno.h>
32 #include <sys/msg.h>
33
34 #include <android-base/file.h>
35
TEST(sys_msg,smoke)36 TEST(sys_msg, smoke) {
37 if (msgctl(-1, IPC_STAT, nullptr) == -1 && errno == ENOSYS) {
38 GTEST_SKIP() << "no <sys/msg.h> support in this kernel";
39 }
40
41 // Create a queue.
42 TemporaryDir dir;
43 key_t key = ftok(dir.path, 1);
44 int id = msgget(key, IPC_CREAT|0666);
45 ASSERT_NE(id, -1);
46
47 // Queue should be empty.
48 msqid_ds ds;
49 memset(&ds, 0, sizeof(ds));
50 ASSERT_EQ(0, msgctl(id, IPC_STAT, &ds));
51 ASSERT_EQ(0U, ds.msg_qnum);
52 ASSERT_EQ(0U, ds.msg_cbytes);
53
54 // Send a message.
55 struct {
56 long type;
57 char data[32];
58 } msg = { 1, "hello world" };
59 ASSERT_EQ(0, msgsnd(id, &msg, sizeof(msg), 0));
60
61 // Queue should be non-empty.
62 ASSERT_EQ(0, msgctl(id, IPC_STAT, &ds));
63 ASSERT_EQ(1U, ds.msg_qnum);
64 ASSERT_EQ(sizeof(msg), ds.msg_cbytes);
65
66 // Read the message.
67 memset(&msg, 0, sizeof(msg));
68 ASSERT_EQ(static_cast<ssize_t>(sizeof(msg)), msgrcv(id, &msg, sizeof(msg), 0, 0));
69 ASSERT_EQ(1, msg.type);
70 ASSERT_STREQ("hello world", msg.data);
71
72 // Destroy the queue.
73 ASSERT_EQ(0, msgctl(id, IPC_RMID, nullptr));
74 }
75
TEST(sys_msg,msgctl_failure)76 TEST(sys_msg, msgctl_failure) {
77 errno = 0;
78 ASSERT_EQ(-1, msgctl(-1, IPC_STAT, nullptr));
79 ASSERT_TRUE(errno == EINVAL || errno == ENOSYS);
80 }
81
TEST(sys_msg,msgget_failure)82 TEST(sys_msg, msgget_failure) {
83 errno = 0;
84 ASSERT_EQ(-1, msgget(-1, 0));
85 ASSERT_TRUE(errno == ENOENT || errno == ENOSYS);
86 }
87
TEST(sys_msg,msgrcv_failure)88 TEST(sys_msg, msgrcv_failure) {
89 errno = 0;
90 ASSERT_EQ(-1, msgrcv(-1, nullptr, 0, 0, 0));
91 ASSERT_TRUE(errno == EINVAL || errno == ENOSYS);
92 }
93
TEST(sys_msg,msgsnd_failure)94 TEST(sys_msg, msgsnd_failure) {
95 errno = 0;
96 ASSERT_EQ(-1, msgsnd(-1, "", 0, 0));
97 ASSERT_TRUE(errno == EINVAL || errno == ENOSYS);
98 }
99