1/* Copyright (C) 2004-2022 Free Software Foundation, Inc.
2 This file is part of the GNU C Library.
3
4 The GNU C Library is free software; you can redistribute it and/or
5 modify it under the terms of the GNU Lesser General Public
6 License as published by the Free Software Foundation; either
7 version 2.1 of the License, or (at your option) any later version.
8
9 The GNU C Library is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 Lesser General Public License for more details.
13
14 You should have received a copy of the GNU Lesser General Public
15 License along with the GNU C Library; if not, see
16 <https://www.gnu.org/licenses/>. */
17
18#include <errno.h>
19#include <mqueue.h>
20#include <stdio.h>
21#include <stdlib.h>
22#include <string.h>
23#include <unistd.h>
24#if _POSIX_THREADS
25# include <pthread.h>
26# include <support/check.h>
27
28static pthread_barrier_t b;
29
30/* Cleanup handling test. */
31static int cl_called;
32
33static void
34cl (void *arg)
35{
36 ++cl_called;
37}
38
39#define TF_MQ_RECEIVE 0L
40#define TF_MQ_TIMEDRECEIVE 1L
41#define TF_MQ_SEND 2L
42#define TF_MQ_TIMEDSEND 3L
43
44static const char *names[]
45 = { "mq_receive", "mq_timedreceive", "mq_send", "mq_timedsend" };
46
47static mqd_t q;
48static struct timespec never;
49
50static void *
51tf (void *arg)
52{
53 int r = pthread_barrier_wait (barrier: &b);
54 if (r != 0 && r != PTHREAD_BARRIER_SERIAL_THREAD)
55 {
56 puts (s: "tf: barrier_wait failed");
57 exit (1);
58 }
59
60 pthread_cleanup_push (cl, NULL);
61
62 char c = ' ';
63
64 switch ((long) arg)
65 {
66 case TF_MQ_SEND:
67 TEMP_FAILURE_RETRY (mq_send (q, &c, 1, 1));
68 break;
69 case TF_MQ_TIMEDSEND:
70 TEMP_FAILURE_RETRY (mq_timedsend (q, &c, 1, 1, &never));
71 break;
72 case TF_MQ_RECEIVE:
73 TEMP_FAILURE_RETRY (mq_receive (q, &c, 1, NULL));
74 break;
75 case TF_MQ_TIMEDRECEIVE:
76 TEMP_FAILURE_RETRY (mq_timedreceive (q, &c, 1, NULL, &never));
77 break;
78 }
79
80 pthread_cleanup_pop (0);
81
82 printf (format: "tf: %s returned\n", names[(long) arg]);
83
84 exit (1);
85}
86
87#define TEST_FUNCTION do_test ()
88static int
89do_test (void)
90{
91 char name[sizeof "/tst-mqueue8-" + sizeof (pid_t) * 3];
92 snprintf (s: name, maxlen: sizeof (name), format: "/tst-mqueue8-%u", getpid ());
93
94 struct mq_attr attr = { .mq_maxmsg = 1, .mq_msgsize = 1 };
95 q = mq_open (name: name, O_CREAT | O_EXCL | O_RDWR, 0600, &attr);
96
97 if (q == (mqd_t) -1)
98 {
99 if (errno == ENOSYS)
100 FAIL_UNSUPPORTED ("mq_open not supported");
101
102 printf (format: "mq_open failed with: %m\n");
103 return 1;
104 }
105
106 if (mq_unlink (name: name) != 0)
107 {
108 printf (format: "mq_unlink failed with: %m\n");
109 return 1;
110 }
111
112 if (pthread_barrier_init (barrier: &b, NULL, count: 2) != 0)
113 {
114 puts (s: "barrier_init failed");
115 return 1;
116 }
117
118 if (clock_gettime (CLOCK_REALTIME, tp: &never) == 0)
119 never.tv_sec += 100;
120 else
121 {
122 never.tv_sec = time (NULL) + 100;
123 never.tv_nsec = 0;
124 }
125
126 int result = 0;
127 for (long l = TF_MQ_RECEIVE; l <= TF_MQ_TIMEDSEND; ++l)
128 {
129 cl_called = 0;
130
131 pthread_t th;
132 if (pthread_create (newthread: &th, NULL, start_routine: tf, arg: (void *) l) != 0)
133 {
134 printf (format: "1st %s create failed\n", names[l]);
135 result = 1;
136 continue;
137 }
138
139 int r = pthread_barrier_wait (barrier: &b);
140 if (r != 0 && r != PTHREAD_BARRIER_SERIAL_THREAD)
141 {
142 puts (s: "barrier_wait failed");
143 result = 1;
144 continue;
145 }
146
147 struct timespec ts = { .tv_sec = 0, .tv_nsec = 100000000 };
148 while (nanosleep (requested_time: &ts, remaining: &ts) != 0)
149 continue;
150
151 printf (format: "going to cancel %s in-time\n", names[l]);
152 if (pthread_cancel (th: th) != 0)
153 {
154 printf (format: "1st cancel of %s failed\n", names[l]);
155 result = 1;
156 continue;
157 }
158
159 void *status;
160 if (pthread_join (th: th, thread_return: &status) != 0)
161 {
162 printf (format: "1st join of %s failed\n", names[l]);
163 result = 1;
164 continue;
165 }
166 if (status != PTHREAD_CANCELED)
167 {
168 printf (format: "1st %s thread not canceled\n", names[l]);
169 result = 1;
170 continue;
171 }
172
173 if (cl_called == 0)
174 {
175 printf (format: "%s cleanup handler not called\n", names[l]);
176 result = 1;
177 continue;
178 }
179 if (cl_called > 1)
180 {
181 printf (format: "%s cleanup handler called more than once\n", names[l]);
182 result = 1;
183 continue;
184 }
185
186 printf (format: "in-time %s cancellation succeeded\n", names[l]);
187
188 cl_called = 0;
189
190 if (pthread_create (newthread: &th, NULL, start_routine: tf, arg: (void *) l) != 0)
191 {
192 printf (format: "2nd %s create failed\n", names[l]);
193 result = 1;
194 continue;
195 }
196
197 printf (format: "going to cancel %s early\n", names[l]);
198 if (pthread_cancel (th: th) != 0)
199 {
200 printf (format: "2nd cancel of %s failed\n", names[l]);
201 result = 1;
202 continue;
203 }
204
205 r = pthread_barrier_wait (barrier: &b);
206 if (r != 0 && r != PTHREAD_BARRIER_SERIAL_THREAD)
207 {
208 puts (s: "barrier_wait failed");
209 result = 1;
210 continue;
211 }
212
213 if (pthread_join (th: th, thread_return: &status) != 0)
214 {
215 printf (format: "2nd join of %s failed\n", names[l]);
216 result = 1;
217 continue;
218 }
219 if (status != PTHREAD_CANCELED)
220 {
221 printf (format: "2nd %s thread not canceled\n", names[l]);
222 result = 1;
223 continue;
224 }
225
226 if (cl_called == 0)
227 {
228 printf (format: "%s cleanup handler not called\n", names[l]);
229 result = 1;
230 continue;
231 }
232 if (cl_called > 1)
233 {
234 printf (format: "%s cleanup handler called more than once\n", names[l]);
235 result = 1;
236 continue;
237 }
238
239 printf (format: "early %s cancellation succeeded\n", names[l]);
240
241 if (l == TF_MQ_TIMEDRECEIVE)
242 {
243 /* mq_receive and mq_timedreceive are tested on empty mq.
244 For mq_send and mq_timedsend we need to make it full.
245 If this fails, there is no point in doing further testing. */
246 char c = ' ';
247 if (mq_send (mqdes: q, msg_ptr: &c, msg_len: 1, msg_prio: 1) != 0)
248 {
249 printf (format: "mq_send failed: %m\n");
250 result = 1;
251 break;
252 }
253 }
254 }
255
256 if (mq_close (mqdes: q) != 0)
257 {
258 printf (format: "mq_close failed: %m\n");
259 result = 1;
260 }
261
262 return result;
263}
264#else
265# define TEST_FUNCTION 0
266#endif
267
268#include "../test-skeleton.c"
269

source code of glibc/rt/tst-mqueue8.c