nptl/tst-tls2.c

   1 /* Copyright (C) 2003, 2004, 2011 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3    Contributed by Ulrich Drepper <drepper@redhat.com>, 2003.
   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 #include <errno.h>
  21 #include <pthread.h>
  22 #include <signal.h>
  23 #include <semaphore.h>
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26 #include <unistd.h>
  27
  28
  29 #define N 10
  30 static pthread_t th[N];
  31
  32
  33 #define CB(n) \
  34 static void                                                                   \
  35 cb##n (void)                                                                  \
  36 {                                                                             \
  37   if (th[n] != pthread_self ())                                               \
  38     {                                                                         \
  39       write (STDOUT_FILENO, "wrong callback\n", 15);                          \
  40       _exit (1);                                                              \
  41     }                                                                         \
  42 }
  43 CB (0)
  44 CB (1)
  45 CB (2)
  46 CB (3)
  47 CB (4)
  48 CB (5)
  49 CB (6)
  50 CB (7)
  51 CB (8)
  52 CB (9)
  53 static void (*cbs[]) (void) =
  54 {
  55   cb0, cb1, cb2, cb3, cb4, cb5, cb6, cb7, cb8, cb9
  56 };
  57
  58
  59 static __thread void (*fp) (void) __attribute__ ((tls_model ("local-exec")));
  60
  61
  62 static sem_t s;
  63
  64
  65 #define THE_SIG SIGUSR1
  66 static void
  67 handler (int sig)
  68 {
  69   if (sig != THE_SIG)
  70     {
  71       write (STDOUT_FILENO, "wrong signal\n", 13);
  72       _exit (1);
  73     }
  74
  75   fp ();
  76
  77   if (sem_post (&s) != 0)
  78     {
  79       write (STDOUT_FILENO, "sem_post failed\n", 16);
  80       _exit (1);
  81     }
  82 }
  83
  84
  85 static pthread_barrier_t b;
  86
  87 #define TOTAL_SIGS 1000
  88 static int nsigs;
  89
  90
  91 static void *
  92 tf (void *arg)
  93 {
  94   fp = arg;
  95
  96   pthread_barrier_wait (&b);
  97
  98   pthread_barrier_wait (&b);
  99
 100   if (nsigs != TOTAL_SIGS)
 101     {
 102       puts ("barrier_wait prematurely returns");
 103       exit (1);
 104     }
 105
 106   return NULL;
 107 }
 108
 109
 110 int
 111 do_test (void)
 112 {
 113   if (pthread_barrier_init (&b, NULL, N + 1) != 0)
 114     {
 115       puts ("barrier_init failed");
 116       exit (1);
 117     }
 118
 119   if (sem_init (&s, 0, 0) != 0)
 120     {
 121       puts ("sem_init failed");
 122       exit (1);
 123     }
 124
 125   struct sigaction sa;
 126   sa.sa_handler = handler;
 127   sigemptyset (&sa.sa_mask);
 128   sa.sa_flags = 0;
 129   if (sigaction (THE_SIG, &sa, NULL) != 0)
 130     {
 131       puts ("sigaction failed");
 132       exit (1);
 133     }
 134
 135   pthread_attr_t a;
 136
 137   if (pthread_attr_init (&a) != 0)
 138     {
 139       puts ("attr_init failed");
 140       exit (1);
 141     }
 142
 143   if (pthread_attr_setstacksize (&a, 1 * 1024 * 1024) != 0)
 144     {
 145       puts ("attr_setstacksize failed");
 146       return 1;
 147     }
 148
 149   int i;
 150   for (i = 0; i < N; ++i)
 151     if (pthread_create (&th[i], &a, tf, cbs[i]) != 0)
 152       {
 153         puts ("pthread_create failed");
 154         exit (1);
 155       }
 156
 157   if (pthread_attr_destroy (&a) != 0)
 158     {
 159       puts ("attr_destroy failed");
 160       exit (1);
 161     }
 162
 163   pthread_barrier_wait (&b);
 164
 165   sigset_t ss;
 166   sigemptyset (&ss);
 167   sigaddset (&ss, THE_SIG);
 168   if (pthread_sigmask (SIG_BLOCK, &ss, NULL) != 0)
 169     {
 170       puts ("pthread_sigmask failed");
 171       exit (1);
 172     }
 173
 174   /* Start sending signals.  */
 175   for (i = 0; i < TOTAL_SIGS; ++i)
 176     {
 177       if (kill (getpid (), THE_SIG) != 0)
 178         {
 179           puts ("kill failed");
 180           exit (1);
 181         }
 182
 183       if (TEMP_FAILURE_RETRY (sem_wait (&s)) != 0)
 184         {
 185           puts ("sem_wait failed");
 186           exit (1);
 187         }
 188
 189       ++nsigs;
 190     }
 191
 192   pthread_barrier_wait (&b);
 193
 194   for (i = 0; i < N; ++i)
 195     if (pthread_join (th[i], NULL) != 0)
 196       {
 197         puts ("join failed");
 198         exit (1);
 199       }
 200
 201   return 0;
 202 }
 203
 204
 205 #define TEST_FUNCTION do_test ()
 206 #include "../test-skeleton.c"