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