pthread_once.c 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  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. The GNU C Library is free software; you can redistribute it and/or
  5. modify it under the terms of the GNU Lesser General Public
  6. License as published by the Free Software Foundation; either
  7. version 2.1 of the License, or (at your option) any later version.
  8. The GNU C Library is distributed in the hope that it will be useful,
  9. but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  11. Lesser General Public License for more details.
  12. You should have received a copy of the GNU Lesser General Public
  13. License along with the GNU C Library; if not, write to the Free
  14. Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  15. 02111-1307 USA. */
  16. #include "pthreadP.h"
  17. #include <lowlevellock.h>
  18. unsigned long int __fork_generation attribute_hidden;
  19. static void
  20. clear_once_control (void *arg)
  21. {
  22. pthread_once_t *once_control = (pthread_once_t *) arg;
  23. *once_control = 0;
  24. lll_futex_wake (once_control, INT_MAX, LLL_PRIVATE);
  25. }
  26. int
  27. attribute_protected
  28. __pthread_once (pthread_once_t *once_control, void (*init_routine) (void))
  29. {
  30. for (;;)
  31. {
  32. int oldval;
  33. int newval;
  34. int tmp;
  35. /* Pseudo code:
  36. newval = __fork_generation | 1;
  37. oldval = *once_control;
  38. if ((oldval & 2) == 0)
  39. *once_control = newval;
  40. Do this atomically.
  41. */
  42. newval = __fork_generation | 1;
  43. __asm__ __volatile__ ("1: lwarx %0,0,%3\n"
  44. " andi. %1,%0,2\n"
  45. " bne 2f\n"
  46. " stwcx. %4,0,%3\n"
  47. " bne 1b\n"
  48. "2: isync"
  49. : "=&r" (oldval), "=&r" (tmp), "=m" (*once_control)
  50. : "r" (once_control), "r" (newval), "m" (*once_control)
  51. : "cr0");
  52. /* Check if the initializer has already been done. */
  53. if ((oldval & 2) != 0)
  54. return 0;
  55. /* Check if another thread already runs the initializer. */
  56. if ((oldval & 1) == 0)
  57. break;
  58. /* Check whether the initializer execution was interrupted by a fork. */
  59. if (oldval != newval)
  60. break;
  61. /* Same generation, some other thread was faster. Wait. */
  62. lll_futex_wait (once_control, oldval, LLL_PRIVATE);
  63. }
  64. /* This thread is the first here. Do the initialization.
  65. Register a cleanup handler so that in case the thread gets
  66. interrupted the initialization can be restarted. */
  67. pthread_cleanup_push (clear_once_control, once_control);
  68. init_routine ();
  69. pthread_cleanup_pop (0);
  70. /* Add one to *once_control to take the bottom 2 bits from 01 to 10. */
  71. atomic_increment (once_control);
  72. /* Wake up all other threads. */
  73. lll_futex_wake (once_control, INT_MAX, LLL_PRIVATE);
  74. return 0;
  75. }
  76. weak_alias (__pthread_once, pthread_once)
  77. strong_alias (__pthread_once, __pthread_once_internal)