Adjust for new form of compare&exchange macros.
[kopensolaris-gnu/glibc.git] / nptl / libc-cancellation.c
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
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, write to the Free
17    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
18    02111-1307 USA.  */
19
20 #include <setjmp.h>
21 #include <stdlib.h>
22 #include "pthreadP.h"
23 #include "atomic.h"
24
25
26 #ifndef NOT_IN_libc
27
28 /* The next two functions are similar to pthread_setcanceltype() but
29    more specialized for the use in the cancelable functions like write().
30    They do not need to check parameters etc.  */
31 int
32 attribute_hidden
33 __libc_enable_asynccancel (void)
34 {
35   struct pthread *self = THREAD_SELF;
36   int oldval;
37   int newval;
38
39   do
40     {
41       oldval = THREAD_GETMEM (self, cancelhandling);
42       newval = oldval | CANCELTYPE_BITMASK;
43
44       if (__builtin_expect ((oldval & CANCELED_BITMASK) != 0, 0))
45         {
46           /* If we are already exiting stop right here.  */
47           if ((oldval & EXITING_BITMASK) != 0)
48             break;
49
50           if (atomic_compare_and_exchange_bool_acq (&self->cancelhandling,
51                                                     newval, oldval))
52             /* Somebody else modified the word, try again.  */
53             continue;
54
55           THREAD_SETMEM (self, result, PTHREAD_CANCELED);
56
57           /* The thread is exiting now.  */
58           atomic_bit_set (&self->cancelhandling, EXITING_BIT);
59
60           __do_cancel ();
61
62           /* NOTREACHED */
63         }
64     }
65   while (atomic_compare_and_exchange_bool_acq (&self->cancelhandling,
66                                                newval, oldval));
67
68   return oldval;
69 }
70
71
72 void
73 internal_function attribute_hidden
74 __libc_disable_asynccancel (int oldtype)
75 {
76   /* If asynchronous cancellation was enabled before we do not have
77      anything to do.  */
78   if (oldtype & CANCELTYPE_BITMASK)
79     return;
80
81   struct pthread *self = THREAD_SELF;
82   int oldval;
83   int newval;
84
85   do
86     {
87       oldval = THREAD_GETMEM (self, cancelhandling);
88       newval = oldval & ~CANCELTYPE_BITMASK;
89
90       if (newval == oldval)
91         break;
92     }
93   while (atomic_compare_and_exchange_bool_acq (&self->cancelhandling, newval,
94                                                oldval));
95 }
96
97 #endif