1 /* Basic tests for SYSV message queue functions.
2 Copyright (C) 2016-2022 Free Software Foundation, Inc.
3 This file is part of the GNU C Library.
4
5 The GNU C Library is free software; you can redistribute it and/or
6 modify it under the terms of the GNU Lesser General Public
7 License as published by the Free Software Foundation; either
8 version 2.1 of the License, or (at your option) any later version.
9
10 The GNU C Library is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 Lesser General Public License for more details.
14
15 You should have received a copy of the GNU Lesser General Public
16 License along with the GNU C Library; if not, see
17 <https://www.gnu.org/licenses/>. */
18
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <errno.h>
22 #include <string.h>
23 #include <sys/types.h>
24 #include <sys/ipc.h>
25 #include <sys/msg.h>
26
27 #include <test-sysvipc.h>
28
29 #include <support/support.h>
30 #include <support/check.h>
31 #include <support/temp_file.h>
32
33 #define TEXTSIZE 32
34 struct msgbuf_t
35 {
36 #ifdef _GNU_SOURCE
37 __syscall_slong_t type;
38 #else
39 long type;
40 #endif
41 char buf[TEXTSIZE];
42 };
43
44 #define MSGTYPE 0x01020304
45 #define MSGDATA "0123456789"
46
47 /* These are for the temporary file we generate. */
48 static char *name;
49 static int msqid;
50
51 static void
remove_msq(void)52 remove_msq (void)
53 {
54 /* Enforce message queue removal in case of early test failure.
55 Ignore error since the msgq may already have being removed. */
56 msgctl (msqid, IPC_RMID, NULL);
57 }
58
59 static void
do_prepare(int argc,char * argv[])60 do_prepare (int argc, char *argv[])
61 {
62 int fd = create_temp_file ("tst-sysvmsg.", &name);
63 if (fd == -1)
64 FAIL_EXIT1 ("cannot create temporary file (errno=%d)", errno);
65 }
66
67 #define PREPARE do_prepare
68
69 /* It is not an extensive test, but rather a functional one aimed to check
70 correct parameter passing on kernel. */
71
72 #define MSGQ_MODE 0644
73
74 static int
do_test(void)75 do_test (void)
76 {
77 atexit (remove_msq);
78
79 key_t key = ftok (name, 'G');
80 if (key == -1)
81 FAIL_EXIT1 ("ftok failed");
82
83 msqid = msgget (key, MSGQ_MODE | IPC_CREAT);
84 if (msqid == -1)
85 {
86 if (errno == ENOSYS)
87 FAIL_UNSUPPORTED ("msgget not supported");
88 FAIL_EXIT1 ("msgget failed (errno=%d)", errno);
89 }
90
91 TEST_COMPARE (msgctl (msqid, first_msg_invalid_cmd (), NULL), -1);
92 TEST_COMPARE (errno, EINVAL);
93
94 /* Get message queue kernel information and do some sanity checks. */
95 struct msqid_ds msginfo;
96 if (msgctl (msqid, IPC_STAT, &msginfo) == -1)
97 FAIL_EXIT1 ("msgctl with IPC_STAT failed (errno=%d)", errno);
98
99 if (msginfo.msg_perm.__key != key)
100 FAIL_EXIT1 ("msgid_ds::msg_perm::key (%d) != %d",
101 (int) msginfo.msg_perm.__key, (int) key);
102 if (msginfo.msg_perm.mode != MSGQ_MODE)
103 FAIL_EXIT1 ("msgid_ds::msg_perm::mode (%o) != %o",
104 msginfo.msg_perm.mode, MSGQ_MODE);
105 if (msginfo.msg_qnum != 0)
106 FAIL_EXIT1 ("msgid_ds::msg_qnum (%lu) != 0",
107 (long unsigned) msginfo.msg_qnum);
108
109 /* Check if last argument (IPC_NOWAIT) is correctly handled. */
110 struct msgbuf_t msg2rcv = { 0 };
111 if (msgrcv (msqid, &msg2rcv, sizeof (msg2rcv.buf), MSGTYPE,
112 IPC_NOWAIT) == -1
113 && errno != ENOMSG)
114 FAIL_EXIT1 ("msgrcv failed (errno=%d)", errno);
115
116 struct msgbuf_t msg2snd = { MSGTYPE, MSGDATA };
117 if (msgsnd (msqid, &msg2snd, sizeof (msg2snd.buf), 0) == -1)
118 FAIL_EXIT1 ("msgsnd failed (errno=%d)", errno);
119
120 if (msgrcv (msqid, &msg2rcv, sizeof (msg2rcv.buf), MSGTYPE, 0) == -1)
121 FAIL_EXIT1 ("msgrcv failed (errno=%d)", errno);
122
123 int ret = 0;
124 if (strncmp (msg2snd.buf, msg2rcv.buf, TEXTSIZE) != 0)
125 ret = 1;
126
127 if (msgctl (msqid, IPC_RMID, NULL) == -1)
128 FAIL_EXIT1 ("msgctl failed");
129
130 return ret;
131 }
132
133 #include <support/test-driver.c>
134