]>
Commit | Line | Data |
---|---|---|
1b82c6c7 | 1 | /* Test message queue passing. |
f7a9f785 | 2 | Copyright (C) 2004-2016 Free Software Foundation, Inc. |
1b82c6c7 UD |
3 | This file is part of the GNU C Library. |
4 | Contributed by Jakub Jelinek <jakub@redhat.com>, 2004. | |
5 | ||
6 | The GNU C Library is free software; you can redistribute it and/or | |
7 | modify it under the terms of the GNU Lesser General Public | |
8 | License as published by the Free Software Foundation; either | |
9 | version 2.1 of the License, or (at your option) any later version. | |
10 | ||
11 | The GNU C Library is distributed in the hope that it will be useful, | |
12 | but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
14 | Lesser General Public License for more details. | |
15 | ||
16 | You should have received a copy of the GNU Lesser General Public | |
59ba27a6 PE |
17 | License along with the GNU C Library; if not, see |
18 | <http://www.gnu.org/licenses/>. */ | |
1b82c6c7 UD |
19 | |
20 | #include <errno.h> | |
21 | #include <fcntl.h> | |
22 | #include <mqueue.h> | |
23 | #include <stdio.h> | |
24 | #include <stdlib.h> | |
25 | #include <string.h> | |
26 | #include <sys/wait.h> | |
27 | #include <time.h> | |
28 | #include <unistd.h> | |
5d619de4 | 29 | #include <stdint.h> |
1b82c6c7 UD |
30 | #include "tst-mqueue.h" |
31 | ||
32 | static int | |
33 | intcmp (const void *a, const void *b) | |
34 | { | |
35 | if (*(unsigned char *)a < *(unsigned char *)b) | |
36 | return 1; | |
37 | if (*(unsigned char *)a > *(unsigned char *)b) | |
38 | return -1; | |
39 | return 0; | |
40 | } | |
41 | ||
42 | static int | |
43 | check_attrs (struct mq_attr *attr, int nonblock, long cnt) | |
44 | { | |
45 | int result = 0; | |
46 | ||
47 | if (attr->mq_maxmsg != 10 || attr->mq_msgsize != 1) | |
48 | { | |
49 | printf ("attributes don't match those passed to mq_open\n" | |
5d619de4 L |
50 | "mq_maxmsg %jd, mq_msgsize %jd\n", |
51 | (intmax_t) attr->mq_maxmsg, (intmax_t) attr->mq_msgsize); | |
1b82c6c7 UD |
52 | result = 1; |
53 | } | |
54 | ||
55 | if ((attr->mq_flags & O_NONBLOCK) != nonblock) | |
56 | { | |
5d619de4 L |
57 | printf ("mq_flags %jx != %x\n", |
58 | (intmax_t) (attr->mq_flags & O_NONBLOCK), nonblock); | |
1b82c6c7 UD |
59 | result = 1; |
60 | } | |
61 | ||
62 | if (attr->mq_curmsgs != cnt) | |
63 | { | |
5d619de4 | 64 | printf ("mq_curmsgs %jd != %ld\n", (intmax_t) attr->mq_curmsgs, cnt); |
1b82c6c7 UD |
65 | result = 1; |
66 | } | |
67 | ||
68 | return result; | |
69 | } | |
70 | ||
71 | static int | |
72 | do_one_test (mqd_t q, const char *name, int nonblock) | |
73 | { | |
74 | int result = 0; | |
75 | ||
76 | unsigned char v [] | |
77 | = { 0x32, 0x62, 0x22, 0x31, 0x11, 0x73, 0x61, 0x21, 0x72, 0x71, 0x81 }; | |
78 | ||
79 | struct mq_attr attr; | |
80 | memset (&attr, 0xaa, sizeof (attr)); | |
81 | if (mq_getattr (q, &attr) != 0) | |
82 | { | |
83 | printf ("mq_getattr failed: %m\n"); | |
84 | result = 1; | |
85 | } | |
86 | else | |
efa8adf5 | 87 | result |= check_attrs (&attr, nonblock, 0); |
1b82c6c7 | 88 | |
701666b7 | 89 | if (mq_receive (q, (char *) &v[0], 1, NULL) != -1) |
1b82c6c7 UD |
90 | { |
91 | puts ("mq_receive on O_WRONLY mqd_t unexpectedly succeeded"); | |
92 | result = 1; | |
93 | } | |
94 | else if (errno != EBADF) | |
95 | { | |
96 | printf ("mq_receive on O_WRONLY mqd_t did not fail with EBADF: %m\n"); | |
97 | result = 1; | |
98 | } | |
99 | ||
100 | struct timespec ts; | |
101 | if (clock_gettime (CLOCK_REALTIME, &ts) == 0) | |
102 | --ts.tv_sec; | |
103 | else | |
104 | { | |
105 | ts.tv_sec = time (NULL) - 1; | |
106 | ts.tv_nsec = 0; | |
107 | } | |
108 | ||
109 | int ret; | |
110 | for (int i = 0; i < 10; ++i) | |
111 | { | |
112 | if (i & 1) | |
701666b7 | 113 | ret = mq_send (q, (char *) &v[i], 1, v[i] >> 4); |
1b82c6c7 | 114 | else |
701666b7 | 115 | ret = mq_timedsend (q, (char *) &v[i], 1, v[i] >> 4, &ts); |
1b82c6c7 UD |
116 | |
117 | if (ret) | |
118 | { | |
119 | printf ("mq_%ssend failed: %m\n", (i & 1) ? "" : "timed"); | |
120 | result = 1; | |
121 | } | |
122 | } | |
123 | ||
701666b7 | 124 | ret = mq_timedsend (q, (char *) &v[10], 1, 8, &ts); |
1b82c6c7 UD |
125 | if (ret != -1) |
126 | { | |
127 | puts ("mq_timedsend on full queue did not fail"); | |
128 | result = 1; | |
129 | } | |
130 | else if (errno != (nonblock ? EAGAIN : ETIMEDOUT)) | |
131 | { | |
132 | printf ("mq_timedsend on full queue did not fail with %s: %m\n", | |
133 | nonblock ? "EAGAIN" : "ETIMEDOUT"); | |
134 | result = 1; | |
135 | } | |
136 | ||
137 | if (nonblock) | |
138 | { | |
701666b7 | 139 | ret = mq_send (q, (char *) &v[10], 1, 8); |
1b82c6c7 UD |
140 | if (ret != -1) |
141 | { | |
142 | puts ("mq_send on full non-blocking queue did not fail"); | |
143 | result = 1; | |
144 | } | |
145 | else if (errno != EAGAIN) | |
146 | { | |
147 | printf ("mq_send on full non-blocking queue did not fail" | |
148 | "with EAGAIN: %m\n"); | |
149 | result = 1; | |
150 | } | |
151 | } | |
152 | ||
153 | memset (&attr, 0xaa, sizeof (attr)); | |
154 | if (mq_getattr (q, &attr) != 0) | |
155 | { | |
156 | printf ("mq_getattr failed: %m\n"); | |
157 | result = 1; | |
158 | } | |
159 | else | |
efa8adf5 | 160 | result |= check_attrs (&attr, nonblock, 10); |
1b82c6c7 UD |
161 | |
162 | pid_t pid = fork (); | |
163 | if (pid == -1) | |
164 | { | |
165 | printf ("fork failed: %m\n"); | |
166 | result = 1; | |
167 | } | |
168 | else if (pid == 0) | |
169 | { | |
170 | result = 0; | |
171 | ||
172 | if (mq_close (q) != 0) | |
173 | { | |
174 | printf ("mq_close in child failed: %m\n"); | |
175 | result = 1; | |
176 | } | |
177 | ||
178 | q = mq_open (name, O_RDONLY | nonblock); | |
179 | if (q == (mqd_t) -1) | |
180 | { | |
181 | printf ("mq_open in child failed: %m\n"); | |
182 | exit (1); | |
183 | } | |
184 | ||
185 | memset (&attr, 0xaa, sizeof (attr)); | |
186 | if (mq_getattr (q, &attr) != 0) | |
187 | { | |
188 | printf ("mq_getattr failed: %m\n"); | |
189 | result = 1; | |
190 | } | |
191 | else | |
efa8adf5 | 192 | result |= check_attrs (&attr, nonblock, 10); |
1b82c6c7 UD |
193 | |
194 | unsigned char vr[11] = { }; | |
195 | unsigned int prio; | |
196 | ssize_t rets; | |
197 | ||
701666b7 | 198 | if (mq_send (q, (char *) &v[0], 1, 1) != -1) |
1b82c6c7 UD |
199 | { |
200 | puts ("mq_send on O_RDONLY mqd_t unexpectedly succeeded"); | |
201 | result = 1; | |
202 | } | |
203 | else if (errno != EBADF) | |
204 | { | |
205 | printf ("mq_send on O_WRONLY mqd_t did not fail with EBADF: %m\n"); | |
206 | result = 1; | |
207 | } | |
208 | ||
209 | for (int i = 0; i < 10; ++i) | |
210 | { | |
211 | if (i & 1) | |
701666b7 | 212 | rets = mq_receive (q, (char *) &vr[i], 1, &prio); |
1b82c6c7 | 213 | else |
701666b7 | 214 | rets = mq_timedreceive (q, (char *) &vr[i], 1, &prio, &ts); |
1b82c6c7 UD |
215 | |
216 | if (rets != 1) | |
217 | { | |
218 | if (rets == -1) | |
219 | printf ("mq_%sreceive failed: %m\n", (i & 1) ? "" : "timed"); | |
220 | else | |
221 | printf ("mq_%sreceive returned %zd != 1\n", | |
222 | (i & 1) ? "" : "timed", rets); | |
223 | result = 1; | |
224 | } | |
225 | else if (prio != (unsigned int) vr[i] >> 4) | |
226 | { | |
227 | printf ("unexpected priority %x for value %02x\n", prio, | |
228 | vr[i]); | |
229 | result = 1; | |
230 | } | |
231 | } | |
232 | ||
233 | qsort (v, 10, 1, intcmp); | |
234 | if (memcmp (v, vr, 10) != 0) | |
235 | { | |
236 | puts ("messages not received in expected order"); | |
237 | result = 1; | |
238 | } | |
239 | ||
701666b7 | 240 | rets = mq_timedreceive (q, (char *) &vr[10], 1, &prio, &ts); |
1b82c6c7 UD |
241 | if (rets != -1) |
242 | { | |
243 | puts ("mq_timedreceive on empty queue did not fail"); | |
244 | result = 1; | |
245 | } | |
246 | else if (errno != (nonblock ? EAGAIN : ETIMEDOUT)) | |
247 | { | |
248 | printf ("mq_timedreceive on empty queue did not fail with %s: %m\n", | |
249 | nonblock ? "EAGAIN" : "ETIMEDOUT"); | |
250 | result = 1; | |
251 | } | |
252 | ||
253 | if (nonblock) | |
254 | { | |
701666b7 | 255 | ret = mq_receive (q, (char *) &vr[10], 1, &prio); |
1b82c6c7 UD |
256 | if (ret != -1) |
257 | { | |
258 | puts ("mq_receive on empty non-blocking queue did not fail"); | |
259 | result = 1; | |
260 | } | |
261 | else if (errno != EAGAIN) | |
262 | { | |
263 | printf ("mq_receive on empty non-blocking queue did not fail" | |
264 | "with EAGAIN: %m\n"); | |
265 | result = 1; | |
266 | } | |
267 | } | |
268 | ||
269 | memset (&attr, 0xaa, sizeof (attr)); | |
270 | if (mq_getattr (q, &attr) != 0) | |
271 | { | |
272 | printf ("mq_getattr failed: %m\n"); | |
273 | result = 1; | |
274 | } | |
275 | else | |
efa8adf5 | 276 | result |= check_attrs (&attr, nonblock, 0); |
1b82c6c7 UD |
277 | |
278 | if (mq_close (q) != 0) | |
279 | { | |
280 | printf ("mq_close in child failed: %m\n"); | |
281 | result = 1; | |
282 | } | |
283 | ||
284 | exit (result); | |
285 | } | |
286 | ||
287 | int status; | |
efa8adf5 | 288 | if (TEMP_FAILURE_RETRY (waitpid (pid, &status, 0)) != pid) |
1b82c6c7 UD |
289 | { |
290 | printf ("waitpid failed: %m\n"); | |
efa8adf5 | 291 | kill (pid, SIGKILL); |
1b82c6c7 UD |
292 | result = 1; |
293 | } | |
294 | else if (!WIFEXITED (status) || WEXITSTATUS (status)) | |
295 | { | |
296 | printf ("child failed: %d\n", status); | |
297 | result = 1; | |
298 | } | |
299 | ||
300 | memset (&attr, 0xaa, sizeof (attr)); | |
301 | if (mq_getattr (q, &attr) != 0) | |
302 | { | |
303 | printf ("mq_getattr failed: %m\n"); | |
304 | result = 1; | |
305 | } | |
306 | else | |
efa8adf5 | 307 | result |= check_attrs (&attr, nonblock, 0); |
1b82c6c7 UD |
308 | |
309 | return result; | |
310 | } | |
311 | ||
312 | #define TEST_FUNCTION do_test () | |
313 | static int | |
314 | do_test (void) | |
315 | { | |
316 | int result = 0; | |
317 | ||
efa8adf5 UD |
318 | char name[sizeof "/tst-mqueue1-" + sizeof (pid_t) * 3]; |
319 | snprintf (name, sizeof (name), "/tst-mqueue1-%u", getpid ()); | |
1b82c6c7 UD |
320 | |
321 | struct mq_attr attr = { .mq_maxmsg = 10, .mq_msgsize = 1 }; | |
322 | mqd_t q = mq_open (name, O_CREAT | O_EXCL | O_WRONLY, 0600, &attr); | |
323 | ||
324 | if (q == (mqd_t) -1) | |
325 | { | |
326 | printf ("mq_open failed with: %m\n"); | |
327 | return result; | |
328 | } | |
329 | else | |
330 | add_temp_mq (name); | |
331 | ||
efa8adf5 | 332 | result |= do_one_test (q, name, 0); |
1b82c6c7 UD |
333 | |
334 | mqd_t q2 = mq_open (name, O_WRONLY | O_NONBLOCK); | |
335 | if (q2 == (mqd_t) -1) | |
336 | { | |
337 | printf ("mq_open failed with: %m\n"); | |
338 | q2 = q; | |
339 | result = 1; | |
340 | } | |
341 | else | |
342 | { | |
343 | if (mq_close (q) != 0) | |
344 | { | |
345 | printf ("mq_close in parent failed: %m\n"); | |
346 | result = 1; | |
347 | } | |
348 | ||
349 | q = q2; | |
350 | result |= do_one_test (q, name, O_NONBLOCK); | |
351 | ||
352 | if (mq_getattr (q, &attr) != 0) | |
353 | { | |
354 | printf ("mq_getattr failed: %m\n"); | |
355 | result = 1; | |
356 | } | |
357 | else | |
358 | { | |
359 | attr.mq_flags ^= O_NONBLOCK; | |
360 | ||
361 | struct mq_attr attr2; | |
362 | memset (&attr2, 0x55, sizeof (attr2)); | |
363 | if (mq_setattr (q, &attr, &attr2) != 0) | |
364 | { | |
365 | printf ("mq_setattr failed: %m\n"); | |
366 | result = 1; | |
367 | } | |
368 | else if (attr.mq_flags != (attr2.mq_flags ^ O_NONBLOCK) | |
369 | || attr.mq_maxmsg != attr2.mq_maxmsg | |
370 | || attr.mq_msgsize != attr2.mq_msgsize | |
371 | || attr.mq_curmsgs != 0 | |
372 | || attr2.mq_curmsgs != 0) | |
373 | { | |
374 | puts ("mq_setattr returned unexpected values in *omqstat"); | |
375 | result = 1; | |
376 | } | |
377 | else | |
378 | { | |
379 | result |= do_one_test (q, name, 0); | |
380 | ||
381 | if (mq_setattr (q, &attr2, NULL) != 0) | |
382 | { | |
383 | printf ("mq_setattr failed: %m\n"); | |
384 | result = 1; | |
385 | } | |
386 | else | |
387 | result |= do_one_test (q, name, O_NONBLOCK); | |
388 | } | |
389 | } | |
390 | } | |
391 | ||
392 | if (mq_unlink (name) != 0) | |
393 | { | |
394 | printf ("mq_unlink failed: %m\n"); | |
395 | result = 1; | |
396 | } | |
397 | ||
398 | if (mq_close (q) != 0) | |
399 | { | |
400 | printf ("mq_close in parent failed: %m\n"); | |
401 | result = 1; | |
402 | } | |
403 | ||
404 | if (mq_close (q) != -1) | |
405 | { | |
406 | puts ("second mq_close did not fail"); | |
407 | result = 1; | |
408 | } | |
409 | else if (errno != EBADF) | |
410 | { | |
411 | printf ("second mq_close did not fail with EBADF: %m\n"); | |
412 | result = 1; | |
413 | } | |
414 | ||
415 | return result; | |
416 | } | |
417 | ||
418 | #include "../test-skeleton.c" |