1 // tls_test.cc -- test TLS variables for gold, main function
2
3 // Copyright (C) 2006-2016 Free Software Foundation, Inc.
4 // Written by Ian Lance Taylor <iant@google.com>.
5
6 // This file is part of gold.
7
8 // This program is free software; you can redistribute it and/or modify
9 // it under the terms of the GNU General Public License as published by
10 // the Free Software Foundation; either version 3 of the License, or
11 // (at your option) any later version.
12
13 // This program is distributed in the hope that it will be useful,
14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 // GNU General Public License for more details.
17
18 // You should have received a copy of the GNU General Public License
19 // along with this program; if not, write to the Free Software
20 // Foundation, Inc., 51 Franklin Street - Fifth Floor, Boston,
21 // MA 02110-1301, USA.
22
23 // This is the main function for the TLS test. See tls_test.cc for
24 // more information.
25
26 #include <cassert>
27 #include <cstdio>
28 #include <pthread.h>
29 #include <semaphore.h>
30
31 #include "tls_test.h"
32
33 // We make these macros so the assert() will give useful line-numbers.
34 #define safe_lock(semptr) \
35 do \
36 { \
37 int err = sem_wait(semptr); \
38 assert(err == 0); \
39 } \
40 while (0)
41
42 #define safe_unlock(semptr) \
43 do \
44 { \
45 int err = sem_post(semptr); \
46 assert(err == 0); \
47 } \
48 while (0)
49
50 struct Sem_set
51 {
52 sem_t sem1;
53 sem_t sem2;
54 sem_t sem3;
55 };
56
57 Sem_set sems1;
58 Sem_set sems2;
59
60 bool failed = false;
61
62 void
check(const char * name,bool val)63 check(const char* name, bool val)
64 {
65 if (!val)
66 {
67 fprintf(stderr, "Test %s failed\n", name);
68 failed = true;
69 }
70 }
71
72 // The body of the thread function. This acquires the first
73 // semaphore, runs the tests, and then releases the second semaphore.
74 // Then it acquires the third semaphore, and the runs the verification
75 // test again.
76
77 void*
thread_routine(void * arg)78 thread_routine(void* arg)
79 {
80 Sem_set* pms = static_cast<Sem_set*>(arg);
81
82 // Acquire the first semaphore.
83 if (pms)
84 safe_lock(&pms->sem1);
85
86 // Run the tests.
87 check("t1", t1());
88 check("t2", t2());
89 check("t3", t3());
90 check("t4", t4());
91 f5b(f5a());
92 check("t5", t5());
93 f6b(f6a());
94 check("t6", t6());
95 check("t8", t8());
96 check("t9", t9());
97 f10b(f10a());
98 check("t10", t10());
99 check("t11", t11() != 0);
100 check("t12", t12());
101 check("t_last", t_last());
102
103 // Release the second semaphore.
104 if (pms)
105 safe_unlock(&pms->sem2);
106
107 // Acquire the third semaphore.
108 if (pms)
109 safe_lock(&pms->sem3);
110
111 check("t_last", t_last());
112
113 return 0;
114 }
115
116 // The main function.
117
118 int
main()119 main()
120 {
121 // First, as a sanity check, run through the tests in the "main" thread.
122 thread_routine(0);
123
124 // Set up the semaphores. We want the first thread to start right
125 // away, tell us when it is done with the first part, and wait for
126 // us to release it. We want the second thread to wait to start,
127 // tell us when it is done with the first part, and wait for us to
128 // release it.
129 sem_init(&sems1.sem1, 0, 1);
130 sem_init(&sems1.sem2, 0, 0);
131 sem_init(&sems1.sem3, 0, 0);
132
133 sem_init(&sems2.sem1, 0, 0);
134 sem_init(&sems2.sem2, 0, 0);
135 sem_init(&sems2.sem3, 0, 0);
136
137 pthread_t thread1;
138 int err = pthread_create(&thread1, NULL, thread_routine, &sems1);
139 assert(err == 0);
140
141 pthread_t thread2;
142 err = pthread_create(&thread2, NULL, thread_routine, &sems2);
143 assert(err == 0);
144
145 // Wait for the first thread to complete the first part.
146 safe_lock(&sems1.sem2);
147
148 // Tell the second thread to start.
149 safe_unlock(&sems2.sem1);
150
151 // Wait for the second thread to complete the first part.
152 safe_lock(&sems2.sem2);
153
154 // Tell the first thread to continue and finish.
155 safe_unlock(&sems1.sem3);
156
157 // Wait for the first thread to finish.
158 void* thread_val;
159 err = pthread_join(thread1, &thread_val);
160 assert(err == 0);
161 assert(thread_val == 0);
162
163 // Tell the second thread to continue and finish.
164 safe_unlock(&sems2.sem3);
165
166 // Wait for the second thread to finish.
167 err = pthread_join(thread2, &thread_val);
168 assert(err == 0);
169 assert(thread_val == 0);
170
171 // All done.
172 return failed ? 1 : 0;
173 }
174