1/* Copyright (C) 2003-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 <dlfcn.h>
19#include <errno.h>
20#include <pthread.h>
21#include <signal.h>
22#include <semaphore.h>
23#include <stdint.h>
24#include <stdio.h>
25#include <stdlib.h>
26#include <unistd.h>
27#include <pthreaddef.h>
28
29#define THE_SIG SIGUSR1
30
31/* The stack size can be overriden. With a sufficiently large stack
32 size, thread stacks for terminated threads are freed, but this does
33 not happen with the default size of 1 MiB. */
34enum { default_stack_size_in_mb = 1 };
35static long stack_size_in_mb;
36
37#define N 10
38static pthread_t th[N];
39
40
41static int do_test (void);
42
43#define TEST_FUNCTION do_test ()
44#include "../test-skeleton.c"
45
46#define CB(n) \
47static void \
48cb##n (void) \
49{ \
50 if (th[n] != pthread_self ()) \
51 { \
52 write_message ("wrong callback\n"); \
53 _exit (1); \
54 } \
55}
56CB (0)
57CB (1)
58CB (2)
59CB (3)
60CB (4)
61CB (5)
62CB (6)
63CB (7)
64CB (8)
65CB (9)
66static void (*cbs[]) (void) =
67{
68 cb0, cb1, cb2, cb3, cb4, cb5, cb6, cb7, cb8, cb9
69};
70
71
72sem_t s;
73
74
75pthread_barrier_t b;
76
77#define TOTAL_SIGS 1000
78int nsigs;
79
80
81int
82do_test (void)
83{
84 if (stack_size_in_mb == 0)
85 stack_size_in_mb = default_stack_size_in_mb;
86
87 if ((uintptr_t) pthread_self () & (TCB_ALIGNMENT - 1))
88 {
89 puts (s: "initial thread's struct pthread not aligned enough");
90 exit (1);
91 }
92
93 if (pthread_barrier_init (barrier: &b, NULL, N + 1) != 0)
94 {
95 puts (s: "barrier_init failed");
96 exit (1);
97 }
98
99 if (sem_init (sem: &s, pshared: 0, value: 0) != 0)
100 {
101 puts (s: "sem_init failed");
102 exit (1);
103 }
104
105 void *h = dlopen (file: "tst-tls3mod.so", RTLD_LAZY);
106 if (h == NULL)
107 {
108 puts (s: "dlopen failed");
109 exit (1);
110 }
111
112 void *(*tf) (void *) = dlsym (handle: h, name: "tf");
113 if (tf == NULL)
114 {
115 puts (s: "dlsym for tf failed");
116 exit (1);
117 }
118
119 struct sigaction sa;
120 sa.sa_handler = dlsym (handle: h, name: "handler");
121 if (sa.sa_handler == NULL)
122 {
123 puts (s: "dlsym for handler failed");
124 exit (1);
125 }
126 sigemptyset (&sa.sa_mask);
127 sa.sa_flags = 0;
128 if (sigaction (THE_SIG, act: &sa, NULL) != 0)
129 {
130 puts (s: "sigaction failed");
131 exit (1);
132 }
133
134 pthread_attr_t a;
135
136 if (pthread_attr_init (attr: &a) != 0)
137 {
138 puts (s: "attr_init failed");
139 exit (1);
140 }
141
142 if (pthread_attr_setstacksize (attr: &a, stacksize: stack_size_in_mb * 1024 * 1024) != 0)
143 {
144 puts (s: "attr_setstacksize failed");
145 return 1;
146 }
147
148 int r;
149 for (r = 0; r < 10; ++r)
150 {
151 int i;
152 for (i = 0; i < N; ++i)
153 if (pthread_create (newthread: &th[i], attr: &a, start_routine: tf, arg: cbs[i]) != 0)
154 {
155 puts (s: "pthread_create failed");
156 exit (1);
157 }
158
159 nsigs = 0;
160
161 pthread_barrier_wait (barrier: &b);
162
163 sigset_t ss;
164 sigemptyset (&ss);
165 sigaddset (&ss, THE_SIG);
166 if (pthread_sigmask (SIG_BLOCK, newmask: &ss, NULL) != 0)
167 {
168 puts (s: "pthread_sigmask failed");
169 exit (1);
170 }
171
172 /* Start sending signals. */
173 for (i = 0; i < TOTAL_SIGS; ++i)
174 {
175 if (kill (pid: getpid (), THE_SIG) != 0)
176 {
177 puts (s: "kill failed");
178 exit (1);
179 }
180
181 if (TEMP_FAILURE_RETRY (sem_wait (&s)) != 0)
182 {
183 puts (s: "sem_wait failed");
184 exit (1);
185 }
186
187 ++nsigs;
188 }
189
190 pthread_barrier_wait (barrier: &b);
191
192 if (pthread_sigmask (SIG_UNBLOCK, newmask: &ss, NULL) != 0)
193 {
194 puts (s: "pthread_sigmask failed");
195 exit (1);
196 }
197
198 for (i = 0; i < N; ++i)
199 if (pthread_join (th: th[i], NULL) != 0)
200 {
201 puts (s: "join failed");
202 exit (1);
203 }
204 }
205
206 if (pthread_attr_destroy (attr: &a) != 0)
207 {
208 puts (s: "attr_destroy failed");
209 exit (1);
210 }
211
212 return 0;
213}
214

source code of glibc/nptl/tst-tls3.c