gold/testsuite/tls_test_main.cc

   1 // tls_test.cc -- test TLS variables for gold, main function
   2
   3 // Copyright (C) 2006-2014 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
  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*
  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
 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 }