OSDN Git Service

b8f522655cccc6b10d3201502635f312ddabfc77
[pf3gnuchains/gcc-fork.git] / libstdc++-v3 / testsuite / thread / pthread1.cc
1 // 2002-01-23  Loren J. Rittle <rittle@labs.mot.com> <ljrittle@acm.org>
2 //
3 // Copyright (C) 2002 Free Software Foundation, Inc.
4 //
5 // This file is part of the GNU ISO C++ Library.  This library is free
6 // software; you can redistribute it and/or modify it under the
7 // terms of the GNU General Public License as published by the
8 // Free Software Foundation; either version 2, or (at your option)
9 // any later version.
10 //
11 // This 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
14 // GNU General Public License for more details.
15 //
16 // You should have received a copy of the GNU General Public License along
17 // with this library; see the file COPYING.  If not, write to the Free
18 // Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307,
19 // USA.
20
21 // { dg-do run { target *-*-freebsd* *-*-linux* *-*-solaris* *-*-cygwin } }
22 // { dg-options "-pthread" { target *-*-freebsd* *-*-linux* } }
23 // { dg-options "-pthreads" { target *-*-solaris* } }
24
25 // This multi-threading C++/STL/POSIX code adheres to rules outlined here:
26 // http://www.sgi.com/tech/stl/thread_safety.html
27 //
28 // It is believed to exercise the allocation code in a manner that
29 // should reveal memory leaks (and, under rare cases, race conditions,
30 // if the STL threading support is fubar'd).
31
32 #include <list>
33
34 // Do not include <pthread.h> explicitly; if threads are properly
35 // configured for the port, then it is picked up free from STL headers.
36
37 #if __GTHREADS
38 using namespace std;
39
40 const int thread_cycles = 10;
41 const int thread_pairs = 10;
42 const unsigned max_size = 100;
43 const int iters = 10000;
44
45 class task_queue
46 {
47 public:
48   task_queue ()
49   {
50     pthread_mutex_init (&fooLock, NULL);
51     pthread_cond_init (&fooCond, NULL);
52   }
53   ~task_queue ()
54   {
55     pthread_mutex_destroy (&fooLock);
56     pthread_cond_destroy (&fooCond);
57   }
58   list<int> foo;
59   pthread_mutex_t fooLock;
60   // This code uses a special case that allows us to use just one
61   // condition variable - in general, don't use this idiom unless you
62   // know what you are doing. ;-)
63   pthread_cond_t fooCond;
64 };
65
66 void*
67 produce (void* t)
68 {
69   task_queue& tq = *(static_cast<task_queue*> (t));
70   int num = 0;
71   while (num < iters)
72     {
73       pthread_mutex_lock (&tq.fooLock);
74       while (tq.foo.size () >= max_size)
75         pthread_cond_wait (&tq.fooCond, &tq.fooLock);
76       tq.foo.push_back (num++);
77       pthread_cond_signal (&tq.fooCond);
78       pthread_mutex_unlock (&tq.fooLock);
79     }
80   return 0;
81 }
82
83 void*
84 consume (void* t)
85 {
86   task_queue& tq = *(static_cast<task_queue*> (t));
87   int num = 0;
88   while (num < iters)
89     {
90       pthread_mutex_lock (&tq.fooLock);
91       while (tq.foo.size () == 0)
92         pthread_cond_wait (&tq.fooCond, &tq.fooLock);
93       if (tq.foo.front () != num++)
94         abort ();
95       tq.foo.pop_front ();
96       pthread_cond_signal (&tq.fooCond);
97       pthread_mutex_unlock (&tq.fooLock);
98     }
99   return 0;
100 }
101
102 int
103 main (int argc, char** argv)
104 {
105   pthread_t prod[thread_pairs];
106   pthread_t cons[thread_pairs];
107
108   task_queue* tq[thread_pairs];
109
110 #if defined(__sun) && defined(__svr4__)
111   pthread_setconcurrency (thread_pairs * 2);
112 #endif
113
114   for (int j = 0; j < thread_cycles; j++)
115     {
116       for (int i = 0; i < thread_pairs; i++)
117         {
118           tq[i] = new task_queue;
119           pthread_create (&prod[i], NULL, produce, static_cast<void*> (tq[i]));
120           pthread_create (&cons[i], NULL, consume, static_cast<void*> (tq[i]));
121         }
122
123       for (int i = 0; i < thread_pairs; i++)
124         {
125           pthread_join (prod[i], NULL);
126           pthread_join (cons[i], NULL);
127 #if defined(__FreeBSD__)
128           // These lines are not required by POSIX since a successful
129           // join is suppose to detach as well...
130           pthread_detach (prod[i]);
131           pthread_detach (cons[i]);
132           // ...but they are according to the FreeBSD 4.X code base
133           // or else you get a memory leak.
134 #endif
135           delete tq[i];
136         }
137     }
138
139   return 0;
140 }
141 #else
142 int main (void) {}
143 #endif