pthread_setcancelstate.c 2.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. /* Copyright (C) 2002, 2003 Free Software Foundation, Inc.
  2. This file is part of the GNU C Library.
  3. Contributed by Ulrich Drepper <drepper@redhat.com>, 2002.
  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 <errno.h>
  17. #include "pthreadP.h"
  18. #include <atomic.h>
  19. int
  20. attribute_protected
  21. __pthread_setcancelstate (
  22. int state,
  23. int *oldstate)
  24. {
  25. volatile struct pthread *self;
  26. if (state < PTHREAD_CANCEL_ENABLE || state > PTHREAD_CANCEL_DISABLE)
  27. return EINVAL;
  28. self = THREAD_SELF;
  29. int oldval = THREAD_GETMEM (self, cancelhandling);
  30. while (1)
  31. {
  32. int newval = (state == PTHREAD_CANCEL_DISABLE
  33. ? oldval | CANCELSTATE_BITMASK
  34. : oldval & ~CANCELSTATE_BITMASK);
  35. /* Store the old value. */
  36. if (oldstate != NULL)
  37. *oldstate = ((oldval & CANCELSTATE_BITMASK)
  38. ? PTHREAD_CANCEL_DISABLE : PTHREAD_CANCEL_ENABLE);
  39. /* Avoid doing unnecessary work. The atomic operation can
  40. potentially be expensive if the memory has to be locked and
  41. remote cache lines have to be invalidated. */
  42. if (oldval == newval)
  43. break;
  44. /* Update the cancel handling word. This has to be done
  45. atomically since other bits could be modified as well. */
  46. int curval = THREAD_ATOMIC_CMPXCHG_VAL (self, cancelhandling, newval,
  47. oldval);
  48. if (__builtin_expect (curval == oldval, 1))
  49. {
  50. if (CANCEL_ENABLED_AND_CANCELED_AND_ASYNCHRONOUS (newval))
  51. __do_cancel ();
  52. break;
  53. }
  54. /* Prepare for the next round. */
  55. oldval = curval;
  56. }
  57. return 0;
  58. }
  59. strong_alias (__pthread_setcancelstate, pthread_setcancelstate)