gold/testsuite/tls_test_main.cc

   1 // tls_test.cc -- test TLS variables for gold, main function
   2
   3 // Copyright 2006, 2007, 2008 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
  30 #include "tls_test.h"
  31
  32 // We make these macros so the assert() will give useful line-numbers.
  33 #define safe_lock(muptr)                        \
  34   do                                            \
  35     {                                           \
  36       int err = pthread_mutex_lock(muptr);      \
  37       assert(err == 0);                         \
  38     }                                           \
  39   while (0)
  40
  41 #define safe_unlock(muptr)                      \
  42   do                                            \
  43     {                                           \
  44       int err = pthread_mutex_unlock(muptr);    \
  45       assert(err == 0);                         \
  46     }                                           \
  47   while (0)
  48
  49 struct Mutex_set
  50 {
  51   pthread_mutex_t mutex1;
  52   pthread_mutex_t mutex2;
  53   pthread_mutex_t mutex3;
  54 };
  55
  56 Mutex_set mutexes1 = { PTHREAD_MUTEX_INITIALIZER,
  57                        PTHREAD_MUTEX_INITIALIZER,
  58                        PTHREAD_MUTEX_INITIALIZER };
  59
  60 Mutex_set mutexes2 = { PTHREAD_MUTEX_INITIALIZER,
  61                        PTHREAD_MUTEX_INITIALIZER,
  62                        PTHREAD_MUTEX_INITIALIZER } ;
  63
  64 bool failed = false;
  65
  66 void
  67 check(const char* name, bool val)
  68 {
  69   if (!val)
  70     {
  71       fprintf(stderr, "Test %s failed\n", name);
  72       failed = true;
  73     }
  74 }
  75
  76 // The body of the thread function.  This gets a lock on the first
  77 // mutex, runs the tests, and then unlocks the second mutex.  Then it
  78 // locks the third mutex, and the runs the verification test again.
  79
  80 void*
  81 thread_routine(void* arg)
  82 {
  83   Mutex_set* pms = static_cast<Mutex_set*>(arg);
  84
  85   // Lock the first mutex.
  86   if (pms)
  87     safe_lock(&pms->mutex1);
  88
  89   // Run the tests.
  90   check("t1", t1());
  91   check("t2", t2());
  92   check("t3", t3());
  93   check("t4", t4());
  94   f5b(f5a());
  95   check("t5", t5());
  96   f6b(f6a());
  97   check("t6", t6());
  98   check("t8", t8());
  99   check("t9", t9());
 100   f10b(f10a());
 101   check("t10", t10());
 102   check("t11", t11() != 0);
 103   check("t12", t12());
 104   check("t_last", t_last());
 105
 106   // Unlock the second mutex.
 107   if (pms)
 108     safe_unlock(&pms->mutex2);
 109
 110   // Lock the third mutex.
 111   if (pms)
 112     safe_lock(&pms->mutex3);
 113
 114   check("t_last", t_last());
 115
 116   return 0;
 117 }
 118
 119 // The main function.
 120
 121 int
 122 main()
 123 {
 124   // First, as a sanity check, run through the tests in the "main" thread.
 125   thread_routine(0);
 126
 127   // Set up the mutex locks.  We want the first thread to start right
 128   // away, tell us when it is done with the first part, and wait for
 129   // us to release it.  We want the second thread to wait to start,
 130   // tell us when it is done with the first part, and wait for us to
 131   // release it.
 132   safe_lock(&mutexes1.mutex2);
 133   safe_lock(&mutexes1.mutex3);
 134
 135   safe_lock(&mutexes2.mutex1);
 136   safe_lock(&mutexes2.mutex2);
 137   safe_lock(&mutexes2.mutex3);
 138
 139   pthread_t thread1;
 140   int err = pthread_create(&thread1, NULL, thread_routine, &mutexes1);
 141   assert(err == 0);
 142
 143   pthread_t thread2;
 144   err = pthread_create(&thread2, NULL, thread_routine, &mutexes2);
 145   assert(err == 0);
 146
 147   // Wait for the first thread to complete the first part.
 148   safe_lock(&mutexes1.mutex2);
 149
 150   // Tell the second thread to start.
 151   safe_unlock(&mutexes2.mutex1);
 152
 153   // Wait for the second thread to complete the first part.
 154   safe_lock(&mutexes2.mutex2);
 155
 156   // Tell the first thread to continue and finish.
 157   safe_unlock(&mutexes1.mutex3);
 158
 159   // Wait for the first thread to finish.
 160   void* thread_val;
 161   err = pthread_join(thread1, &thread_val);
 162   assert(err == 0);
 163   assert(thread_val == 0);
 164
 165   // Tell the second thread to continue and finish.
 166   safe_unlock(&mutexes2.mutex3);
 167
 168   // Wait for the second thread to finish.
 169   err = pthread_join(thread2, &thread_val);
 170   assert(err == 0);
 171   assert(thread_val == 0);
 172
 173   // All done.
 174   return failed ? 1 : 0;
 175 }