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