libpthread/nptl/sysdeps/unix/sysv/linux/powerpc/pthread_once.c

   1 /* Copyright (C) 2003, 2004, 2007 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3    Contributed by Paul Mackerras <paulus@au.ibm.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, see
  17    <http://www.gnu.org/licenses/>.  */
  18
  19 #include "pthreadP.h"
  20 #include <lowlevellock.h>
  21
  22
  23 unsigned long int __fork_generation attribute_hidden;
  24
  25
  26 static void
  27 clear_once_control (void *arg)
  28 {
  29   pthread_once_t *once_control = (pthread_once_t *) arg;
  30
  31   *once_control = 0;
  32   lll_futex_wake (once_control, INT_MAX, LLL_PRIVATE);
  33 }
  34
  35
  36 int
  37 attribute_protected
  38 __pthread_once (pthread_once_t *once_control, void (*init_routine) (void))
  39 {
  40   for (;;)
  41     {
  42       int oldval;
  43       int newval;
  44       int tmp;
  45
  46       /* Pseudo code:
  47          newval = __fork_generation | 1;
  48          oldval = *once_control;
  49          if ((oldval & 2) == 0)
  50            *once_control = newval;
  51          Do this atomically.
  52       */
  53       newval = __fork_generation | 1;
  54       __asm__ __volatile__ ("1: lwarx   %0,0,%3\n"
  55                         "       andi.   %1,%0,2\n"
  56                         "       bne     2f\n"
  57                         "       stwcx.  %4,0,%3\n"
  58                         "       bne     1b\n"
  59                         "2:     isync"
  60                         : "=&r" (oldval), "=&r" (tmp), "=m" (*once_control)
  61                         : "r" (once_control), "r" (newval), "m" (*once_control)
  62                         : "cr0");
  63
  64       /* Check if the initializer has already been done.  */
  65       if ((oldval & 2) != 0)
  66         return 0;
  67
  68       /* Check if another thread already runs the initializer.  */
  69       if ((oldval & 1) == 0)
  70         break;
  71
  72       /* Check whether the initializer execution was interrupted by a fork.  */
  73       if (oldval != newval)
  74         break;
  75
  76       /* Same generation, some other thread was faster. Wait.  */
  77       lll_futex_wait (once_control, oldval, LLL_PRIVATE);
  78     }
  79
  80
  81   /* This thread is the first here.  Do the initialization.
  82      Register a cleanup handler so that in case the thread gets
  83      interrupted the initialization can be restarted.  */
  84   pthread_cleanup_push (clear_once_control, once_control);
  85
  86   init_routine ();
  87
  88   pthread_cleanup_pop (0);
  89
  90
  91   /* Add one to *once_control to take the bottom 2 bits from 01 to 10.  */
  92   atomic_increment (once_control);
  93
  94   /* Wake up all other threads.  */
  95   lll_futex_wake (once_control, INT_MAX, LLL_PRIVATE);
  96
  97   return 0;
  98 }
  99 weak_alias (__pthread_once, pthread_once)
 100 strong_alias (__pthread_once, __pthread_once_internal)