]>
Commit | Line | Data |
---|---|---|
1b82c6c7 | 1 | /* Test SIGEV_THREAD handling for POSIX message queues. |
2b778ceb | 2 | Copyright (C) 2004-2021 Free Software Foundation, Inc. |
1b82c6c7 UD |
3 | This file is part of the GNU C Library. |
4 | Contributed by Ulrich Drepper <drepper@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 | 17 | License along with the GNU C Library; if not, see |
5a82c748 | 18 | <https://www.gnu.org/licenses/>. */ |
1b82c6c7 UD |
19 | |
20 | #include <errno.h> | |
21 | #include <mqueue.h> | |
22 | #include <signal.h> | |
23 | #include <stddef.h> | |
24 | #include <stdint.h> | |
25 | #include <stdio.h> | |
26 | #include <stdlib.h> | |
27 | #include <string.h> | |
28 | #include <sys/mman.h> | |
29 | #include <sys/wait.h> | |
30 | #include <unistd.h> | |
31 | ||
32 | #if _POSIX_THREADS | |
33 | # include <pthread.h> | |
34 | ||
35 | static pid_t pid; | |
36 | static mqd_t m; | |
37 | static const char message[] = "hello"; | |
38 | ||
39 | # define MAXMSG 10 | |
40 | # define MSGSIZE 10 | |
41 | # define UNIQUE 42 | |
42 | ||
43 | ||
44 | static void | |
45 | fct (union sigval s) | |
46 | { | |
47 | /* Put the mq in non-blocking mode. */ | |
48 | struct mq_attr attr; | |
49 | if (mq_getattr (m, &attr) != 0) | |
50 | { | |
51 | printf ("%s: mq_getattr failed: %m\n", __FUNCTION__); | |
52 | exit (1); | |
53 | } | |
54 | attr.mq_flags |= O_NONBLOCK; | |
55 | if (mq_setattr (m, &attr, NULL) != 0) | |
56 | { | |
57 | printf ("%s: mq_setattr failed: %m\n", __FUNCTION__); | |
58 | exit (1); | |
59 | } | |
60 | ||
61 | /* Check the values. */ | |
62 | if (attr.mq_maxmsg != MAXMSG) | |
63 | { | |
5d619de4 L |
64 | printf ("%s: mq_maxmsg wrong: is %jd, expecte %d\n", |
65 | __FUNCTION__, (intmax_t) attr.mq_maxmsg, MAXMSG); | |
1b82c6c7 UD |
66 | exit (1); |
67 | } | |
68 | if (attr.mq_msgsize != MAXMSG) | |
69 | { | |
5d619de4 L |
70 | printf ("%s: mq_msgsize wrong: is %jd, expecte %d\n", |
71 | __FUNCTION__, (intmax_t) attr.mq_msgsize, MSGSIZE); | |
1b82c6c7 UD |
72 | exit (1); |
73 | } | |
74 | ||
75 | /* Read the message. */ | |
76 | char buf[attr.mq_msgsize]; | |
77 | ssize_t n = TEMP_FAILURE_RETRY (mq_receive (m, buf, attr.mq_msgsize, NULL)); | |
78 | if (n != sizeof (message)) | |
79 | { | |
80 | printf ("%s: length of message wrong: is %zd, expected %zu\n", | |
81 | __FUNCTION__, n, sizeof (message)); | |
82 | exit (1); | |
83 | } | |
84 | if (memcmp (buf, message, sizeof (message)) != 0) | |
85 | { | |
86 | printf ("%s: message wrong: is \"%s\", expected \"%s\"\n", | |
87 | __FUNCTION__, buf, message); | |
88 | exit (1); | |
89 | } | |
90 | ||
91 | exit (UNIQUE); | |
92 | } | |
93 | ||
94 | ||
95 | int | |
96 | do_test (void) | |
97 | { | |
98 | char tmpfname[] = "/tmp/tst-mqueue3-barrier.XXXXXX"; | |
99 | int fd = mkstemp (tmpfname); | |
100 | if (fd == -1) | |
101 | { | |
102 | printf ("cannot open temporary file: %m\n"); | |
103 | return 1; | |
104 | } | |
105 | ||
106 | /* Make sure it is always removed. */ | |
107 | unlink (tmpfname); | |
108 | ||
109 | /* Create one page of data. */ | |
110 | size_t ps = sysconf (_SC_PAGESIZE); | |
111 | char data[ps]; | |
112 | memset (data, '\0', ps); | |
113 | ||
114 | /* Write the data to the file. */ | |
115 | if (write (fd, data, ps) != (ssize_t) ps) | |
116 | { | |
117 | puts ("short write"); | |
118 | return 1; | |
119 | } | |
120 | ||
121 | void *mem = mmap (NULL, ps, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0); | |
122 | if (mem == MAP_FAILED) | |
123 | { | |
124 | printf ("mmap failed: %m\n"); | |
125 | return 1; | |
126 | } | |
127 | ||
128 | pthread_barrier_t *b; | |
129 | b = (pthread_barrier_t *) (((uintptr_t) mem + __alignof (pthread_barrier_t)) | |
130 | & ~(__alignof (pthread_barrier_t) - 1)); | |
131 | ||
132 | pthread_barrierattr_t a; | |
133 | if (pthread_barrierattr_init (&a) != 0) | |
134 | { | |
135 | puts ("barrierattr_init failed"); | |
136 | return 1; | |
137 | } | |
138 | ||
139 | if (pthread_barrierattr_setpshared (&a, PTHREAD_PROCESS_SHARED) != 0) | |
140 | { | |
141 | puts ("barrierattr_setpshared failed, could not test"); | |
142 | return 0; | |
143 | } | |
144 | ||
145 | if (pthread_barrier_init (b, &a, 2) != 0) | |
146 | { | |
147 | puts ("barrier_init failed"); | |
148 | return 1; | |
149 | } | |
150 | ||
151 | if (pthread_barrierattr_destroy (&a) != 0) | |
152 | { | |
153 | puts ("barrierattr_destroy failed"); | |
154 | return 1; | |
155 | } | |
156 | ||
157 | /* Name for the message queue. */ | |
158 | char mqname[sizeof ("/tst-mqueue3-") + 3 * sizeof (pid_t)]; | |
159 | snprintf (mqname, sizeof (mqname) - 1, "/tst-mqueue3-%ld", | |
160 | (long int) getpid ()); | |
161 | ||
162 | /* Create the message queue. */ | |
163 | struct mq_attr attr = { .mq_maxmsg = MAXMSG, .mq_msgsize = MSGSIZE }; | |
164 | m = mq_open (mqname, O_CREAT | O_EXCL | O_RDWR, 0600, &attr); | |
165 | if (m == -1) | |
166 | { | |
edb4d496 UD |
167 | if (errno == ENOSYS) |
168 | { | |
169 | puts ("not implemented"); | |
170 | return 0; | |
171 | } | |
172 | ||
1b82c6c7 UD |
173 | puts ("mq_open failed"); |
174 | return 1; | |
175 | } | |
176 | ||
177 | /* Unlink the message queue right away. */ | |
178 | if (mq_unlink (mqname) != 0) | |
179 | { | |
180 | puts ("mq_unlink failed"); | |
181 | return 1; | |
182 | } | |
183 | ||
184 | pid = fork (); | |
185 | if (pid == -1) | |
186 | { | |
187 | puts ("fork failed"); | |
188 | return 1; | |
189 | } | |
190 | if (pid == 0) | |
191 | { | |
192 | /* Request notification via thread. */ | |
193 | struct sigevent ev; | |
194 | ev.sigev_notify = SIGEV_THREAD; | |
195 | ev.sigev_notify_function = fct; | |
196 | ev.sigev_value.sival_ptr = NULL; | |
197 | ev.sigev_notify_attributes = NULL; | |
198 | ||
199 | /* Tell the kernel. */ | |
200 | if (mq_notify (m,&ev) != 0) | |
201 | { | |
202 | puts ("mq_notify failed"); | |
203 | exit (1); | |
204 | } | |
205 | ||
206 | /* Tell the parent we are ready. */ | |
207 | (void) pthread_barrier_wait (b); | |
208 | ||
209 | /* Make sure the process goes away eventually. */ | |
210 | alarm (10); | |
211 | ||
212 | /* Do nothing forever. */ | |
213 | while (1) | |
214 | pause (); | |
215 | } | |
216 | ||
217 | /* Wait for the child process to register to notification method. */ | |
218 | (void) pthread_barrier_wait (b); | |
219 | ||
220 | /* Send the message. */ | |
221 | if (mq_send (m, message, sizeof (message), 1) != 0) | |
222 | { | |
223 | kill (pid, SIGKILL); | |
224 | puts ("mq_send failed"); | |
225 | return 1; | |
226 | } | |
227 | ||
228 | int r; | |
229 | if (TEMP_FAILURE_RETRY (waitpid (pid, &r, 0)) != pid) | |
230 | { | |
231 | kill (pid, SIGKILL); | |
232 | puts ("waitpid failed"); | |
233 | return 1; | |
234 | } | |
235 | ||
236 | return WIFEXITED (r) && WEXITSTATUS (r) == UNIQUE ? 0 : 1; | |
237 | } | |
238 | # define TEST_FUNCTION do_test () | |
239 | #else | |
240 | # define TEST_FUNCTION 0 | |
241 | #endif | |
242 | ||
243 | #include "../test-skeleton.c" |