pthread_setcanceltype.c 2.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  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, see
  14. <http://www.gnu.org/licenses/>. */
  15. #include <errno.h>
  16. #include "pthreadP.h"
  17. #include <atomic.h>
  18. int
  19. attribute_protected
  20. __pthread_setcanceltype (
  21. int type,
  22. int *oldtype)
  23. {
  24. volatile struct pthread *self;
  25. if (type < PTHREAD_CANCEL_DEFERRED || type > PTHREAD_CANCEL_ASYNCHRONOUS)
  26. return EINVAL;
  27. self = THREAD_SELF;
  28. int oldval = THREAD_GETMEM (self, cancelhandling);
  29. while (1)
  30. {
  31. int newval = (type == PTHREAD_CANCEL_ASYNCHRONOUS
  32. ? oldval | CANCELTYPE_BITMASK
  33. : oldval & ~CANCELTYPE_BITMASK);
  34. /* Store the old value. */
  35. if (oldtype != NULL)
  36. *oldtype = ((oldval & CANCELTYPE_BITMASK)
  37. ? PTHREAD_CANCEL_ASYNCHRONOUS : PTHREAD_CANCEL_DEFERRED);
  38. /* Avoid doing unnecessary work. The atomic operation can
  39. potentially be expensive if the memory has to be locked and
  40. remote cache lines have to be invalidated. */
  41. if (oldval == newval)
  42. break;
  43. /* Update the cancel handling word. This has to be done
  44. atomically since other bits could be modified as well. */
  45. int curval = THREAD_ATOMIC_CMPXCHG_VAL (self, cancelhandling, newval,
  46. oldval);
  47. if (__builtin_expect (curval == oldval, 1))
  48. {
  49. if (CANCEL_ENABLED_AND_CANCELED_AND_ASYNCHRONOUS (newval))
  50. {
  51. THREAD_SETMEM (self, result, PTHREAD_CANCELED);
  52. __do_cancel ();
  53. }
  54. break;
  55. }
  56. /* Prepare for the next round. */
  57. oldval = curval;
  58. }
  59. return 0;
  60. }
  61. strong_alias (__pthread_setcanceltype, pthread_setcanceltype)