Initial revision
[kopensolaris-gnu/glibc.git] / nptl / tst-signal3.c
1 /* Copyright (C) 2002 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3    Contributed by Ulrich Drepper <drepper@redhat.com>, 2002.
4
5    The GNU C Library is free software; you can redistribute it and/or
6    modify it under the terms of the GNU Lesser General Public
7    License as published by the Free Software Foundation; either
8    version 2.1 of the License, or (at your option) any later version.
9
10    The GNU C Library is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13    Lesser General Public License for more details.
14
15    You should have received a copy of the GNU Lesser General Public
16    License along with the GNU C Library; if not, write to the Free
17    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
18    02111-1307 USA.  */
19
20 #include <pthread.h>
21 #include <semaphore.h>
22 #include <signal.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <unistd.h>
26
27
28 /* Number of different signalss to use.  Also is the number of
29    threads.  */
30 #define N 10
31 /* Maximum number of threads in flight at any one time.  */
32 #define INFLIGHT 5
33 /* Number of signals sent in total.  */
34 #define ROUNDS 10000
35
36
37 static int received[N][N];
38 static int nsig[N];
39 static pthread_t th[N];
40 static sem_t sem;
41 static pthread_mutex_t lock[N];
42 static pthread_t th_main;
43 static int sig0;
44
45 static void
46 handler (int sig)
47 {
48   int i;
49   for (i = 0; i < N; ++i)
50     if (pthread_equal (pthread_self (), th[i]))
51       break;
52
53   if (i == N)
54     {
55       if (pthread_equal (pthread_self (), th_main))
56         puts ("signal received by main thread");
57       else
58         printf ("signal received by unknown thread (%lx)\n",
59                 (unsigned long int) pthread_self ());
60       exit (1);
61     }
62
63   ++received[i][sig - sig0];
64
65   sem_post (&sem);
66 }
67
68
69 static void *
70 tf (void *arg)
71 {
72   int idx = (long int) arg;
73
74   sigset_t ss;
75   sigemptyset (&ss);
76
77   int i;
78   for (i = 0; i <= idx; ++i)
79     sigaddset (&ss, sig0 + i);
80
81   if (pthread_sigmask (SIG_UNBLOCK, &ss, NULL) != 0)
82     {
83       printf ("thread %d: pthread_sigmask failed\n", i);
84       exit (1);
85     }
86
87   pthread_mutex_lock (&lock[idx]);
88
89   return NULL;
90 }
91
92
93 static int
94 do_test (void)
95 {
96   /* Block all signals.  */
97   sigset_t ss;
98   sigfillset (&ss);
99
100   th_main = pthread_self ();
101
102   sig0 = SIGRTMIN;
103
104   if (pthread_sigmask (SIG_SETMASK, &ss, NULL) != 0)
105     {
106       puts ("1st pthread_sigmask failed");
107       exit (1);
108     }
109
110   /* Install the handler.  */
111   int i;
112   for (i = 0; i < N; ++i)
113     {
114       struct sigaction sa =
115         {
116           .sa_handler = handler,
117           .sa_flags = 0
118         };
119       sigfillset (&sa.sa_mask);
120
121       if (sigaction (sig0 + i, &sa, NULL) != 0)
122         {
123           printf ("sigaction for signal %d failed\n", i);
124           exit (1);
125         }
126     }
127
128   if (sem_init (&sem, 0, INFLIGHT) != 0)
129     {
130       puts ("sem_init failed");
131       exit (1);
132     }
133
134   for (i = 0; i < N; ++i)
135     {
136       if (pthread_mutex_init (&lock[i], NULL) != 0)
137         {
138           printf ("mutex_init[%d] failed\n", i);
139         }
140
141       if (pthread_mutex_lock (&lock[i]) != 0)
142         {
143           printf ("mutex_lock[%d] failed\n", i);
144         }
145
146       if (pthread_create (&th[i], NULL, tf, (void *) (long int) i) != 0)
147         {
148           printf ("create of thread %d failed\n", i);
149           exit (1);
150         }
151     }
152
153   int result = 0;
154   unsigned int r = 42;
155   pid_t pid = getpid ();
156
157   for (i = 0; i < ROUNDS; ++i)
158     {
159       if (sem_wait (&sem) != 0)
160         {
161           printf ("sem_wait round %d failed: %m\n", i);
162           exit (1);
163         }
164
165       int s = rand_r (&r) % N;
166
167       kill (pid, sig0 + s);
168     }
169
170   void *status;
171   for (i = 0; i < N; ++i)
172     {
173       if (pthread_mutex_unlock (&lock[i]) != 0)
174         {
175           printf ("unlock %d failed\n", i);
176           exit (1);
177         }
178
179       if (pthread_join (th[i], &status) != 0)
180         {
181           printf ("join %d failed\n", i);
182           result = 1;
183         }
184       else if (status != NULL)
185         {
186           printf ("%d: result != NULL\n", i);
187           result = 1;
188         }
189     }
190
191   int total = 0;
192   for (i = 0; i < N; ++i)
193     {
194       int j;
195
196       for (j = 0; j <= i; ++j)
197         total += received[i][j];
198
199       for (j = i + 1; j < N; ++j)
200         if (received[i][j] != 0)
201           {
202             printf ("thread %d received signal SIGRTMIN+%d\n", i, j);
203             result = 1;
204           }
205     }
206
207   if (total != ROUNDS)
208     {
209       printf ("total number of handled signals is %d, expected %d\n",
210               total, ROUNDS);
211       result = 1;
212     }
213
214   printf ("A total of %d signals sent and received\n", total);
215   for (i = 0; i < N; ++i)
216     {
217       printf ("thread %2d:", i);
218
219       int j;
220       for (j = 0; j <= i; ++j)
221         {
222           printf (" %5d", received[i][j]);
223           nsig[j] += received[i][j];
224         }
225
226       putchar ('\n');
227
228     }
229
230   printf ("\nTotal    :");
231   for (i = 0; i < N; ++i)
232     printf (" %5d", nsig[i]);
233   putchar ('\n');
234
235   return result;
236 }
237
238 #define TIMEOUT 10
239 #define TEST_FUNCTION do_test ()
240 #include "../test-skeleton.c"