nptl/sysdeps/unix/sysv/linux/sem_wait.c

   1 /* sem_wait -- wait on a semaphore.  Generic futex-using version.
   2    Copyright (C) 2003, 2007 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4    Contributed by Paul Mackerras <paulus@au.ibm.com>, 2003.
   5
   6    The GNU C Library is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU Lesser General Public
   8    License as published by the Free Software Foundation; either
   9    version 2.1 of the License, or (at your option) any later version.
  10
  11    The GNU C Library is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14    Lesser General Public License for more details.
  15
  16    You should have received a copy of the GNU Lesser General Public
  17    License along with the GNU C Library; if not, write to the Free
  18    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  19    02111-1307 USA.  */
  20
  21 #include <errno.h>
  22 #include <sysdep.h>
  23 #include <lowlevellock.h>
  24 #include <internaltypes.h>
  25 #include <semaphore.h>
  26
  27 #include <pthreadP.h>
  28 #include <shlib-compat.h>
  29
  30
  31 void
  32 attribute_hidden
  33 __sem_wait_cleanup (void *arg)
  34 {
  35   struct new_sem *isem = (struct new_sem *) arg;
  36
  37   atomic_decrement (&isem->nwaiters);
  38 }
  39
  40
  41 int
  42 __new_sem_wait (sem_t *sem)
  43 {
  44   struct new_sem *isem = (struct new_sem *) sem;
  45   int err;
  46
  47   if (atomic_decrement_if_positive (&isem->value) > 0)
  48     return 0;
  49
  50   atomic_increment (&isem->nwaiters);
  51
  52   pthread_cleanup_push (__sem_wait_cleanup, isem);
  53
  54   while (1)
  55     {
  56       /* Enable asynchronous cancellation.  Required by the standard.  */
  57       int oldtype = __pthread_enable_asynccancel ();
  58
  59       err = lll_futex_wait (&isem->value, 0,
  60                             isem->private ^ FUTEX_PRIVATE_FLAG);
  61
  62       /* Disable asynchronous cancellation.  */
  63       __pthread_disable_asynccancel (oldtype);
  64
  65       if (err != 0 && err != -EWOULDBLOCK)
  66         {
  67           __set_errno (-err);
  68           err = -1;
  69           break;
  70         }
  71
  72       if (atomic_decrement_if_positive (&isem->value) > 0)
  73         {
  74           err = 0;
  75           break;
  76         }
  77     }
  78
  79   pthread_cleanup_pop (0);
  80
  81   atomic_decrement (&isem->nwaiters);
  82
  83   return err;
  84 }
  85 versioned_symbol (libpthread, __new_sem_wait, sem_wait, GLIBC_2_1);
  86
  87
  88 #if SHLIB_COMPAT (libpthread, GLIBC_2_0, GLIBC_2_1)
  89 int
  90 attribute_compat_text_section
  91 __old_sem_wait (sem_t *sem)
  92 {
  93   int *futex = (int *) sem;
  94   int err;
  95
  96   do
  97     {
  98       if (atomic_decrement_if_positive (futex) > 0)
  99         return 0;
 100
 101       /* Enable asynchronous cancellation.  Required by the standard.  */
 102       int oldtype = __pthread_enable_asynccancel ();
 103
 104       /* Always assume the semaphore is shared.  */
 105       err = lll_futex_wait (futex, 0, LLL_SHARED);
 106
 107       /* Disable asynchronous cancellation.  */
 108       __pthread_disable_asynccancel (oldtype);
 109     }
 110   while (err == 0 || err == -EWOULDBLOCK);
 111
 112   __set_errno (-err);
 113   return -1;
 114 }
 115
 116 compat_symbol (libpthread, __old_sem_wait, sem_wait, GLIBC_2_0);
 117 #endif