1/* Verify that pthread_[gs]etattr_default_np work correctly.
2
3 Copyright (C) 2013-2022 Free Software Foundation, Inc.
4 This file is part of the GNU C Library.
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
17 License along with the GNU C Library; if not, see
18 <https://www.gnu.org/licenses/>. */
19
20#include <pthread.h>
21#include <stdio.h>
22#include <stdint.h>
23#include <string.h>
24#include <unistd.h>
25#include <errno.h>
26#include <stdbool.h>
27
28#define RETURN_IF_FAIL(f, ...) \
29 ({ \
30 int ret = f (__VA_ARGS__); \
31 if (ret != 0) \
32 { \
33 printf ("%s:%d: %s returned %d (errno = %d)\n", __FILE__, __LINE__, \
34 #f, ret, errno); \
35 return ret; \
36 } \
37 })
38
39static int (*verify_result) (pthread_attr_t *);
40static size_t stacksize = 1024 * 1024;
41static size_t guardsize;
42static bool do_join = true;
43static int running = 0;
44static int detach_failed = 0;
45static pthread_mutex_t m = PTHREAD_MUTEX_INITIALIZER;
46static pthread_cond_t c = PTHREAD_COND_INITIALIZER;
47
48static void *
49thr (void *unused __attribute__ ((unused)))
50{
51 pthread_attr_t attr;
52 int ret;
53
54 memset (&attr, 0xab, sizeof attr);
55 /* To verify that the pthread_setattr_default_np worked. */
56 if ((ret = pthread_getattr_default_np (attr: &attr)) != 0)
57 {
58 printf (format: "pthread_getattr_default_np failed: %s\n", strerror (errnum: ret));
59 goto out;
60 }
61
62 if ((ret = (*verify_result) (&attr)) != 0)
63 goto out;
64
65 memset (&attr, 0xab, sizeof attr);
66 /* To verify that the attributes actually got applied. */
67 if ((ret = pthread_getattr_np (th: pthread_self (), attr: &attr)) != 0)
68 {
69 printf (format: "pthread_getattr_default_np failed: %s\n", strerror (errnum: ret));
70 goto out;
71 }
72
73 ret = (*verify_result) (&attr);
74
75out:
76 if (!do_join)
77 {
78 pthread_mutex_lock (mutex: &m);
79 running--;
80 pthread_cond_signal (cond: &c);
81 pthread_mutex_unlock (mutex: &m);
82
83 detach_failed |= ret;
84 }
85
86 return (void *) (uintptr_t) ret;
87}
88
89static int
90run_threads (const pthread_attr_t *attr)
91{
92 pthread_t t;
93 void *tret = NULL;
94
95 RETURN_IF_FAIL (pthread_setattr_default_np, attr);
96
97 /* Run twice to ensure that the attributes do not get overwritten in the
98 first run somehow. */
99 for (int i = 0; i < 2; i++)
100 {
101 RETURN_IF_FAIL (pthread_create, &t, NULL, thr, NULL);
102 if (do_join)
103 RETURN_IF_FAIL (pthread_join, t, &tret);
104 else
105 {
106 pthread_mutex_lock (mutex: &m);
107 running++;
108 pthread_mutex_unlock (mutex: &m);
109 }
110
111 if (tret != NULL)
112 {
113 puts (s: "Thread failed");
114 return 1;
115 }
116 }
117
118 /* Stay in sync for detached threads and get their status. */
119 while (!do_join)
120 {
121 pthread_mutex_lock (mutex: &m);
122 if (running == 0)
123 {
124 pthread_mutex_unlock (mutex: &m);
125 break;
126 }
127 pthread_cond_wait (cond: &c, mutex: &m);
128 pthread_mutex_unlock (mutex: &m);
129 }
130
131 return 0;
132}
133
134static int
135verify_detach_result (pthread_attr_t *attr)
136{
137 int state;
138
139 RETURN_IF_FAIL (pthread_attr_getdetachstate, attr, &state);
140
141 if (state != PTHREAD_CREATE_DETACHED)
142 {
143 puts (s: "failed to set detach state");
144 return 1;
145 }
146
147 return 0;
148}
149
150static int
151do_detach_test (void)
152{
153 pthread_attr_t attr;
154
155 do_join = false;
156 RETURN_IF_FAIL (pthread_attr_init, &attr);
157 RETURN_IF_FAIL (pthread_attr_setdetachstate, &attr, PTHREAD_CREATE_DETACHED);
158
159 RETURN_IF_FAIL (run_threads, &attr);
160 return detach_failed;
161}
162
163static int
164verify_affinity_result (pthread_attr_t *attr)
165{
166 cpu_set_t cpuset;
167
168 RETURN_IF_FAIL (pthread_attr_getaffinity_np, attr, sizeof (cpuset), &cpuset);
169 if (!CPU_ISSET (0, &cpuset))
170 {
171 puts (s: "failed to set cpu affinity");
172 return 1;
173 }
174
175 return 0;
176}
177
178static int
179do_affinity_test (void)
180{
181 pthread_attr_t attr;
182
183 RETURN_IF_FAIL (pthread_attr_init, &attr);
184
185 /* Processor affinity. Like scheduling policy, this could fail if the user
186 does not have the necessary privileges. So we only spew a warning if
187 pthread_create fails with EPERM. A computer has at least one CPU. */
188 cpu_set_t cpuset;
189 CPU_ZERO (&cpuset);
190 CPU_SET (0, &cpuset);
191 RETURN_IF_FAIL (pthread_attr_setaffinity_np, &attr, sizeof (cpuset), &cpuset);
192
193 int ret = run_threads (attr: &attr);
194
195 if (ret == EPERM)
196 {
197 printf (format: "Skipping CPU Affinity test: %s\n", strerror (errnum: ret));
198 return 0;
199 }
200 else if (ret != 0)
201 return ret;
202
203 return 0;
204}
205
206static int
207verify_sched_result (pthread_attr_t *attr)
208{
209 int inherited, policy;
210 struct sched_param param;
211
212 RETURN_IF_FAIL (pthread_attr_getinheritsched, attr, &inherited);
213 if (inherited != PTHREAD_EXPLICIT_SCHED)
214 {
215 puts (s: "failed to set EXPLICIT_SCHED (%d != %d)");
216 return 1;
217 }
218
219 RETURN_IF_FAIL (pthread_attr_getschedpolicy, attr, &policy);
220 if (policy != SCHED_RR)
221 {
222 printf (format: "failed to set SCHED_RR (%d != %d)\n", policy, SCHED_RR);
223 return 1;
224 }
225
226 RETURN_IF_FAIL (pthread_attr_getschedparam, attr, &param);
227 if (param.sched_priority != 42)
228 {
229 printf (format: "failed to set sched_priority (%d != %d)\n",
230 param.sched_priority, 42);
231 return 1;
232 }
233
234 return 0;
235}
236
237static int
238do_sched_test (void)
239{
240 pthread_attr_t attr;
241
242 RETURN_IF_FAIL (pthread_attr_init, &attr);
243
244 /* Scheduling policy. Note that we don't always test these since it's
245 possible that the user the tests run as don't have the appropriate
246 privileges. */
247 RETURN_IF_FAIL (pthread_attr_setinheritsched, &attr, PTHREAD_EXPLICIT_SCHED);
248 RETURN_IF_FAIL (pthread_attr_setschedpolicy, &attr, SCHED_RR);
249
250 struct sched_param param;
251 param.sched_priority = 42;
252 RETURN_IF_FAIL (pthread_attr_setschedparam, &attr, &param);
253
254 int ret = run_threads (attr: &attr);
255
256 if (ret == EPERM)
257 {
258 printf (format: "Skipping Scheduler Attributes test: %s\n", strerror (errnum: ret));
259 return 0;
260 }
261 else if (ret != 0)
262 return ret;
263
264 return 0;
265}
266
267static int
268verify_guardsize_result (pthread_attr_t *attr)
269{
270 size_t guard;
271
272 RETURN_IF_FAIL (pthread_attr_getguardsize, attr, &guard);
273
274 if (guardsize != guard)
275 {
276 printf (format: "failed to set guardsize (%zu, %zu)\n", guardsize, guard);
277 return 1;
278 }
279
280 return 0;
281}
282
283static int
284do_guardsize_test (void)
285{
286 long int pagesize = sysconf (_SC_PAGESIZE);
287 pthread_attr_t attr;
288
289 if (pagesize < 0)
290 {
291 printf (format: "sysconf failed: %s\n", strerror (errno));
292 return 1;
293 }
294
295 RETURN_IF_FAIL (pthread_getattr_default_np, &attr);
296
297 /* Increase default guardsize by a page. */
298 RETURN_IF_FAIL (pthread_attr_getguardsize, &attr, &guardsize);
299 guardsize += pagesize;
300 RETURN_IF_FAIL (pthread_attr_setguardsize, &attr, guardsize);
301 RETURN_IF_FAIL (run_threads, &attr);
302
303 return 0;
304}
305
306static int
307verify_stacksize_result (pthread_attr_t *attr)
308{
309 size_t stack;
310
311 RETURN_IF_FAIL (pthread_attr_getstacksize, attr, &stack);
312
313 if (stacksize != stack)
314 {
315 printf (format: "failed to set default stacksize (%zu, %zu)\n", stacksize, stack);
316 return 1;
317 }
318
319 return 0;
320}
321
322static int
323do_stacksize_test (void)
324{
325 long int pagesize = sysconf (_SC_PAGESIZE);
326 pthread_attr_t attr;
327
328 if (pagesize < 0)
329 {
330 printf (format: "sysconf failed: %s\n", strerror (errno));
331 return 1;
332 }
333
334 /* Perturb the size by a page so that we're not aligned on the 64K boundary.
335 pthread_create does this perturbation on x86 to avoid causing the 64k
336 aliasing conflict. We want to prevent pthread_create from doing that
337 since it is not consistent for all architectures. */
338 stacksize += pagesize;
339
340 RETURN_IF_FAIL (pthread_attr_init, &attr);
341
342 /* Run twice to ensure that we don't give a false positive. */
343 RETURN_IF_FAIL (pthread_attr_setstacksize, &attr, stacksize);
344 RETURN_IF_FAIL (run_threads, &attr);
345 stacksize *= 2;
346 RETURN_IF_FAIL (pthread_attr_setstacksize, &attr, stacksize);
347 RETURN_IF_FAIL (run_threads, &attr);
348 return 0;
349}
350
351/* We test each attribute separately because sched and affinity tests may need
352 additional user privileges that may not be available during the test run.
353 Each attribute test is a set of two functions, viz. a function to set the
354 default attribute (do_foo_test) and another to verify its result
355 (verify_foo_result). Each test spawns a thread and checks (1) if the
356 attribute values were applied correctly and (2) if the change in the default
357 value reflected. */
358static int
359do_test (void)
360{
361 puts (s: "stacksize test");
362 verify_result = verify_stacksize_result;
363 RETURN_IF_FAIL (do_stacksize_test);
364
365 puts (s: "guardsize test");
366 verify_result = verify_guardsize_result;
367 RETURN_IF_FAIL (do_guardsize_test);
368
369 puts (s: "sched test");
370 verify_result = verify_sched_result;
371 RETURN_IF_FAIL (do_sched_test);
372
373 puts (s: "affinity test");
374 verify_result = verify_affinity_result;
375 RETURN_IF_FAIL (do_affinity_test);
376
377 puts (s: "detach test");
378 verify_result = verify_detach_result;
379 RETURN_IF_FAIL (do_detach_test);
380
381 return 0;
382}
383
384#define TEST_FUNCTION do_test ()
385#include "../test-skeleton.c"
386

source code of glibc/nptl/tst-default-attr.c