Initial revision
[kopensolaris-gnu/glibc.git] / nptl / tst-barrier3.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 /* Test of POSIX barriers.  */
21
22 #include <pthread.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h>
26
27 #define NTHREADS 20
28
29 #define ROUNDS 20
30
31 static pthread_barrier_t barriers[NTHREADS];
32
33 static pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER;
34 static int counters[NTHREADS];
35 static int serial[NTHREADS];
36
37 static void *
38 worker (void *arg)
39 {
40   void *result = NULL;
41   int nr = (long int) arg;
42   int i;
43
44   for (i = 0; i < ROUNDS; ++i)
45     {
46       int j;
47       int retval;
48
49       if (nr == 0)
50         {
51           memset (counters, '\0', sizeof (counters));
52           memset (serial, '\0', sizeof (serial));
53         }
54
55       retval = pthread_barrier_wait (&barriers[NTHREADS - 1]);
56       if (retval != 0 && retval != PTHREAD_BARRIER_SERIAL_THREAD)
57         {
58           printf ("thread %d failed to wait for all the others\n", nr);
59           result = (void *) 1;
60         }
61
62       for (j = nr; j < NTHREADS; ++j)
63         {
64           /* Increment the counter for this round.  */
65           pthread_mutex_lock (&lock);
66           ++counters[j];
67           pthread_mutex_unlock (&lock);
68
69           /* Wait for the rest.  */
70           retval = pthread_barrier_wait (&barriers[j]);
71
72           /* Test the result.  */
73           if (nr == 0 && counters[j] != j + 1)
74             {
75               printf ("barrier in round %d released but count is %d\n",
76                       j, counters[j]);
77               result = (void *) 1;
78             }
79
80           if (retval != 0)
81             {
82               if (retval != PTHREAD_BARRIER_SERIAL_THREAD)
83                 {
84                   printf ("thread %d in round %d has nonzero return value != PTHREAD_BARRIER_SERIAL_THREAD\n",
85                           nr, j);
86                   result = (void *) 1;
87                 }
88               else
89                 {
90                   pthread_mutex_lock (&lock);
91                   ++serial[j];
92                   pthread_mutex_unlock (&lock);
93                 }
94             }
95
96           /* Wait for the rest again.  */
97           retval = pthread_barrier_wait (&barriers[j]);
98
99           /* Now we can check whether exactly one thread was serializing.  */
100           if (nr == 0 && serial[j] != 1)
101             {
102               printf ("not exactly one serial thread in round %d\n", j);
103               result = (void *) 1;
104             }
105         }
106     }
107
108   return result;
109 }
110
111
112 #define TEST_FUNCTION do_test ()
113 #define TIMEOUT 60
114 static int
115 do_test (void)
116 {
117   pthread_t threads[NTHREADS];
118   int i;
119   void *res;
120   int result = 0;
121
122   /* Initialized the barrier variables.  */
123   for (i = 0; i < NTHREADS; ++i)
124     if (pthread_barrier_init (&barriers[i], NULL, i + 1) != 0)
125       {
126         printf ("Failed to initialize barrier %d\n", i);
127         exit (1);
128       }
129
130   /* Start the threads.  */
131   for (i = 0; i < NTHREADS; ++i)
132     if (pthread_create (&threads[i], NULL, worker, (void *) (long int) i) != 0)
133       {
134         printf ("Failed to start thread %d\n", i);
135         exit (1);
136       }
137
138   /* And wait for them.  */
139   for (i = 0; i < NTHREADS; ++i)
140     if (pthread_join (threads[i], &res) != 0 || res != NULL)
141       {
142         printf ("thread %d returned a failure\n", i);
143         result = 1;
144       }
145     else
146       printf ("joined threads %d\n", i);
147
148   if (result == 0)
149     puts ("all OK");
150
151   return result;
152 }
153
154 #include "../test-skeleton.c"