The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2008 The Android Open Source Project |
| 3 | * All rights reserved. |
| 4 | * |
| 5 | * Redistribution and use in source and binary forms, with or without |
| 6 | * modification, are permitted provided that the following conditions |
| 7 | * are met: |
| 8 | * * Redistributions of source code must retain the above copyright |
| 9 | * notice, this list of conditions and the following disclaimer. |
| 10 | * * Redistributions in binary form must reproduce the above copyright |
| 11 | * notice, this list of conditions and the following disclaimer in |
| 12 | * the documentation and/or other materials provided with the |
| 13 | * distribution. |
| 14 | * |
| 15 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| 16 | * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| 17 | * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS |
| 18 | * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE |
| 19 | * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, |
| 20 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, |
| 21 | * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS |
| 22 | * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED |
| 23 | * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, |
| 24 | * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT |
| 25 | * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
| 26 | * SUCH DAMAGE. |
| 27 | */ |
Pierre Peiffer | d0c884d | 2012-02-22 16:40:15 +0100 | [diff] [blame] | 28 | |
Elliott Hughes | 6f94de3 | 2013-02-12 06:06:22 +0000 | [diff] [blame] | 29 | #include <pthread.h> |
Elliott Hughes | 3e89847 | 2013-02-12 16:40:24 +0000 | [diff] [blame] | 30 | |
| 31 | #include <errno.h> |
| 32 | #include <limits.h> |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 33 | #include <stdatomic.h> |
| 34 | #include <sys/cdefs.h> |
Elliott Hughes | 84114c8 | 2013-07-17 13:33:19 -0700 | [diff] [blame] | 35 | #include <sys/mman.h> |
Pierre Peiffer | d0c884d | 2012-02-22 16:40:15 +0100 | [diff] [blame] | 36 | #include <unistd.h> |
| 37 | |
Pierre Peiffer | d0c884d | 2012-02-22 16:40:15 +0100 | [diff] [blame] | 38 | #include "pthread_internal.h" |
Elliott Hughes | eb847bc | 2013-10-09 15:50:50 -0700 | [diff] [blame] | 39 | |
Elliott Hughes | 04303f5 | 2014-09-18 16:11:59 -0700 | [diff] [blame] | 40 | #include "private/bionic_constants.h" |
Elliott Hughes | eb847bc | 2013-10-09 15:50:50 -0700 | [diff] [blame] | 41 | #include "private/bionic_futex.h" |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 42 | #include "private/bionic_systrace.h" |
Elliott Hughes | 04303f5 | 2014-09-18 16:11:59 -0700 | [diff] [blame] | 43 | #include "private/bionic_time_conversions.h" |
Elliott Hughes | eb847bc | 2013-10-09 15:50:50 -0700 | [diff] [blame] | 44 | #include "private/bionic_tls.h" |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 45 | |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 46 | /* a mutex is implemented as a 32-bit integer holding the following fields |
| 47 | * |
| 48 | * bits: name description |
Elliott Hughes | 40eabe2 | 2013-02-14 18:59:37 -0800 | [diff] [blame] | 49 | * 31-16 tid owner thread's tid (recursive and errorcheck only) |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 50 | * 15-14 type mutex type |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 51 | * 13 shared process-shared flag |
| 52 | * 12-2 counter counter of recursive mutexes |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 53 | * 1-0 state lock state (0, 1 or 2) |
| 54 | */ |
| 55 | |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 56 | /* Convenience macro, creates a mask of 'bits' bits that starts from |
| 57 | * the 'shift'-th least significant bit in a 32-bit word. |
| 58 | * |
| 59 | * Examples: FIELD_MASK(0,4) -> 0xf |
| 60 | * FIELD_MASK(16,9) -> 0x1ff0000 |
| 61 | */ |
| 62 | #define FIELD_MASK(shift,bits) (((1 << (bits))-1) << (shift)) |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 63 | |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 64 | /* This one is used to create a bit pattern from a given field value */ |
| 65 | #define FIELD_TO_BITS(val,shift,bits) (((val) & ((1 << (bits))-1)) << (shift)) |
David 'Digit' Turner | 022d303 | 2011-12-07 14:02:17 +0100 | [diff] [blame] | 66 | |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 67 | /* And this one does the opposite, i.e. extract a field's value from a bit pattern */ |
| 68 | #define FIELD_FROM_BITS(val,shift,bits) (((val) >> (shift)) & ((1 << (bits))-1)) |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 69 | |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 70 | /* Mutex state: |
| 71 | * |
| 72 | * 0 for unlocked |
| 73 | * 1 for locked, no waiters |
| 74 | * 2 for locked, maybe waiters |
| 75 | */ |
| 76 | #define MUTEX_STATE_SHIFT 0 |
| 77 | #define MUTEX_STATE_LEN 2 |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 78 | |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 79 | #define MUTEX_STATE_MASK FIELD_MASK(MUTEX_STATE_SHIFT, MUTEX_STATE_LEN) |
| 80 | #define MUTEX_STATE_FROM_BITS(v) FIELD_FROM_BITS(v, MUTEX_STATE_SHIFT, MUTEX_STATE_LEN) |
| 81 | #define MUTEX_STATE_TO_BITS(v) FIELD_TO_BITS(v, MUTEX_STATE_SHIFT, MUTEX_STATE_LEN) |
| 82 | |
| 83 | #define MUTEX_STATE_UNLOCKED 0 /* must be 0 to match __PTHREAD_MUTEX_INIT_VALUE */ |
| 84 | #define MUTEX_STATE_LOCKED_UNCONTENDED 1 /* must be 1 due to atomic dec in unlock operation */ |
| 85 | #define MUTEX_STATE_LOCKED_CONTENDED 2 /* must be 1 + LOCKED_UNCONTENDED due to atomic dec */ |
| 86 | |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 87 | #define MUTEX_STATE_BITS_UNLOCKED MUTEX_STATE_TO_BITS(MUTEX_STATE_UNLOCKED) |
| 88 | #define MUTEX_STATE_BITS_LOCKED_UNCONTENDED MUTEX_STATE_TO_BITS(MUTEX_STATE_LOCKED_UNCONTENDED) |
| 89 | #define MUTEX_STATE_BITS_LOCKED_CONTENDED MUTEX_STATE_TO_BITS(MUTEX_STATE_LOCKED_CONTENDED) |
| 90 | |
| 91 | /* return true iff the mutex if locked with no waiters */ |
| 92 | #define MUTEX_STATE_BITS_IS_LOCKED_UNCONTENDED(v) (((v) & MUTEX_STATE_MASK) == MUTEX_STATE_BITS_LOCKED_UNCONTENDED) |
| 93 | |
| 94 | /* return true iff the mutex if locked with maybe waiters */ |
| 95 | #define MUTEX_STATE_BITS_IS_LOCKED_CONTENDED(v) (((v) & MUTEX_STATE_MASK) == MUTEX_STATE_BITS_LOCKED_CONTENDED) |
| 96 | |
| 97 | /* used to flip from LOCKED_UNCONTENDED to LOCKED_CONTENDED */ |
| 98 | #define MUTEX_STATE_BITS_FLIP_CONTENTION(v) ((v) ^ (MUTEX_STATE_BITS_LOCKED_CONTENDED ^ MUTEX_STATE_BITS_LOCKED_UNCONTENDED)) |
| 99 | |
| 100 | /* Mutex counter: |
| 101 | * |
| 102 | * We need to check for overflow before incrementing, and we also need to |
| 103 | * detect when the counter is 0 |
| 104 | */ |
| 105 | #define MUTEX_COUNTER_SHIFT 2 |
| 106 | #define MUTEX_COUNTER_LEN 11 |
| 107 | #define MUTEX_COUNTER_MASK FIELD_MASK(MUTEX_COUNTER_SHIFT, MUTEX_COUNTER_LEN) |
| 108 | |
| 109 | #define MUTEX_COUNTER_BITS_WILL_OVERFLOW(v) (((v) & MUTEX_COUNTER_MASK) == MUTEX_COUNTER_MASK) |
| 110 | #define MUTEX_COUNTER_BITS_IS_ZERO(v) (((v) & MUTEX_COUNTER_MASK) == 0) |
| 111 | |
| 112 | /* Used to increment the counter directly after overflow has been checked */ |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 113 | #define MUTEX_COUNTER_BITS_ONE FIELD_TO_BITS(1, MUTEX_COUNTER_SHIFT,MUTEX_COUNTER_LEN) |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 114 | |
| 115 | /* Mutex shared bit flag |
| 116 | * |
| 117 | * This flag is set to indicate that the mutex is shared among processes. |
| 118 | * This changes the futex opcode we use for futex wait/wake operations |
| 119 | * (non-shared operations are much faster). |
| 120 | */ |
| 121 | #define MUTEX_SHARED_SHIFT 13 |
| 122 | #define MUTEX_SHARED_MASK FIELD_MASK(MUTEX_SHARED_SHIFT,1) |
| 123 | |
| 124 | /* Mutex type: |
| 125 | * |
| 126 | * We support normal, recursive and errorcheck mutexes. |
| 127 | * |
| 128 | * The constants defined here *cannot* be changed because they must match |
| 129 | * the C library ABI which defines the following initialization values in |
| 130 | * <pthread.h>: |
| 131 | * |
| 132 | * __PTHREAD_MUTEX_INIT_VALUE |
| 133 | * __PTHREAD_RECURSIVE_MUTEX_VALUE |
| 134 | * __PTHREAD_ERRORCHECK_MUTEX_INIT_VALUE |
| 135 | */ |
| 136 | #define MUTEX_TYPE_SHIFT 14 |
| 137 | #define MUTEX_TYPE_LEN 2 |
| 138 | #define MUTEX_TYPE_MASK FIELD_MASK(MUTEX_TYPE_SHIFT,MUTEX_TYPE_LEN) |
| 139 | |
| 140 | #define MUTEX_TYPE_NORMAL 0 /* Must be 0 to match __PTHREAD_MUTEX_INIT_VALUE */ |
| 141 | #define MUTEX_TYPE_RECURSIVE 1 |
| 142 | #define MUTEX_TYPE_ERRORCHECK 2 |
| 143 | |
| 144 | #define MUTEX_TYPE_TO_BITS(t) FIELD_TO_BITS(t, MUTEX_TYPE_SHIFT, MUTEX_TYPE_LEN) |
| 145 | |
| 146 | #define MUTEX_TYPE_BITS_NORMAL MUTEX_TYPE_TO_BITS(MUTEX_TYPE_NORMAL) |
| 147 | #define MUTEX_TYPE_BITS_RECURSIVE MUTEX_TYPE_TO_BITS(MUTEX_TYPE_RECURSIVE) |
| 148 | #define MUTEX_TYPE_BITS_ERRORCHECK MUTEX_TYPE_TO_BITS(MUTEX_TYPE_ERRORCHECK) |
| 149 | |
| 150 | /* Mutex owner field: |
| 151 | * |
| 152 | * This is only used for recursive and errorcheck mutexes. It holds the |
Yabin Cui | 140f367 | 2015-02-03 10:32:00 -0800 | [diff] [blame] | 153 | * tid of the owning thread. We use 16 bits to represent tid here, |
| 154 | * so the highest tid is 65535. There is a test to check /proc/sys/kernel/pid_max |
| 155 | * to make sure it will not exceed our limit. |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 156 | */ |
| 157 | #define MUTEX_OWNER_SHIFT 16 |
| 158 | #define MUTEX_OWNER_LEN 16 |
| 159 | |
| 160 | #define MUTEX_OWNER_FROM_BITS(v) FIELD_FROM_BITS(v,MUTEX_OWNER_SHIFT,MUTEX_OWNER_LEN) |
| 161 | #define MUTEX_OWNER_TO_BITS(v) FIELD_TO_BITS(v,MUTEX_OWNER_SHIFT,MUTEX_OWNER_LEN) |
| 162 | |
| 163 | /* Convenience macros. |
| 164 | * |
| 165 | * These are used to form or modify the bit pattern of a given mutex value |
| 166 | */ |
| 167 | |
| 168 | |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 169 | |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 170 | /* a mutex attribute holds the following fields |
| 171 | * |
| 172 | * bits: name description |
| 173 | * 0-3 type type of mutex |
| 174 | * 4 shared process-shared flag |
| 175 | */ |
| 176 | #define MUTEXATTR_TYPE_MASK 0x000f |
| 177 | #define MUTEXATTR_SHARED_MASK 0x0010 |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 178 | |
| 179 | |
| 180 | int pthread_mutexattr_init(pthread_mutexattr_t *attr) |
| 181 | { |
Elliott Hughes | 39b644a | 2014-03-04 10:55:39 -0800 | [diff] [blame] | 182 | *attr = PTHREAD_MUTEX_DEFAULT; |
| 183 | return 0; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 184 | } |
| 185 | |
| 186 | int pthread_mutexattr_destroy(pthread_mutexattr_t *attr) |
| 187 | { |
Elliott Hughes | 39b644a | 2014-03-04 10:55:39 -0800 | [diff] [blame] | 188 | *attr = -1; |
| 189 | return 0; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 190 | } |
| 191 | |
Elliott Hughes | 39b644a | 2014-03-04 10:55:39 -0800 | [diff] [blame] | 192 | int pthread_mutexattr_gettype(const pthread_mutexattr_t *attr, int *type_p) |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 193 | { |
Elliott Hughes | 39b644a | 2014-03-04 10:55:39 -0800 | [diff] [blame] | 194 | int type = (*attr & MUTEXATTR_TYPE_MASK); |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 195 | |
Elliott Hughes | 39b644a | 2014-03-04 10:55:39 -0800 | [diff] [blame] | 196 | if (type < PTHREAD_MUTEX_NORMAL || type > PTHREAD_MUTEX_ERRORCHECK) { |
| 197 | return EINVAL; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 198 | } |
Elliott Hughes | 39b644a | 2014-03-04 10:55:39 -0800 | [diff] [blame] | 199 | |
| 200 | *type_p = type; |
| 201 | return 0; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 202 | } |
| 203 | |
| 204 | int pthread_mutexattr_settype(pthread_mutexattr_t *attr, int type) |
| 205 | { |
Elliott Hughes | 39b644a | 2014-03-04 10:55:39 -0800 | [diff] [blame] | 206 | if (type < PTHREAD_MUTEX_NORMAL || type > PTHREAD_MUTEX_ERRORCHECK ) { |
| 207 | return EINVAL; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 208 | } |
Elliott Hughes | 39b644a | 2014-03-04 10:55:39 -0800 | [diff] [blame] | 209 | |
| 210 | *attr = (*attr & ~MUTEXATTR_TYPE_MASK) | type; |
| 211 | return 0; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 212 | } |
| 213 | |
| 214 | /* process-shared mutexes are not supported at the moment */ |
| 215 | |
| 216 | int pthread_mutexattr_setpshared(pthread_mutexattr_t *attr, int pshared) |
| 217 | { |
Mathias Agopian | b768116 | 2009-07-13 22:00:33 -0700 | [diff] [blame] | 218 | switch (pshared) { |
| 219 | case PTHREAD_PROCESS_PRIVATE: |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 220 | *attr &= ~MUTEXATTR_SHARED_MASK; |
| 221 | return 0; |
| 222 | |
Mathias Agopian | b768116 | 2009-07-13 22:00:33 -0700 | [diff] [blame] | 223 | case PTHREAD_PROCESS_SHARED: |
| 224 | /* our current implementation of pthread actually supports shared |
| 225 | * mutexes but won't cleanup if a process dies with the mutex held. |
| 226 | * Nevertheless, it's better than nothing. Shared mutexes are used |
| 227 | * by surfaceflinger and audioflinger. |
| 228 | */ |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 229 | *attr |= MUTEXATTR_SHARED_MASK; |
Mathias Agopian | b768116 | 2009-07-13 22:00:33 -0700 | [diff] [blame] | 230 | return 0; |
| 231 | } |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 232 | return EINVAL; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 233 | } |
| 234 | |
Elliott Hughes | c3f1140 | 2013-10-30 14:40:09 -0700 | [diff] [blame] | 235 | int pthread_mutexattr_getpshared(const pthread_mutexattr_t* attr, int* pshared) { |
Elliott Hughes | 39b644a | 2014-03-04 10:55:39 -0800 | [diff] [blame] | 236 | *pshared = (*attr & MUTEXATTR_SHARED_MASK) ? PTHREAD_PROCESS_SHARED : PTHREAD_PROCESS_PRIVATE; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 237 | return 0; |
| 238 | } |
| 239 | |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 240 | static inline atomic_int* get_mutex_value_pointer(pthread_mutex_t* mutex) { |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 241 | static_assert(sizeof(atomic_int) == sizeof(mutex->value), |
| 242 | "mutex->value should actually be atomic_int in implementation."); |
| 243 | |
| 244 | // We prefer casting to atomic_int instead of declaring mutex->value to be atomic_int directly. |
| 245 | // Because using the second method pollutes pthread.h, and causes an error when compiling libcxx. |
| 246 | return reinterpret_cast<atomic_int*>(&mutex->value); |
| 247 | } |
| 248 | |
Elliott Hughes | dff7203 | 2013-12-11 14:54:00 -0800 | [diff] [blame] | 249 | int pthread_mutex_init(pthread_mutex_t* mutex, const pthread_mutexattr_t* attr) { |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 250 | atomic_int* mutex_value_ptr = get_mutex_value_pointer(mutex); |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 251 | |
Elliott Hughes | d4e753f | 2013-07-16 12:45:46 -0700 | [diff] [blame] | 252 | if (__predict_true(attr == NULL)) { |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 253 | atomic_init(mutex_value_ptr, MUTEX_TYPE_BITS_NORMAL); |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 254 | return 0; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 255 | } |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 256 | |
Elliott Hughes | dff7203 | 2013-12-11 14:54:00 -0800 | [diff] [blame] | 257 | int value = 0; |
| 258 | if ((*attr & MUTEXATTR_SHARED_MASK) != 0) { |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 259 | value |= MUTEX_SHARED_MASK; |
Elliott Hughes | dff7203 | 2013-12-11 14:54:00 -0800 | [diff] [blame] | 260 | } |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 261 | |
| 262 | switch (*attr & MUTEXATTR_TYPE_MASK) { |
| 263 | case PTHREAD_MUTEX_NORMAL: |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 264 | value |= MUTEX_TYPE_BITS_NORMAL; |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 265 | break; |
| 266 | case PTHREAD_MUTEX_RECURSIVE: |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 267 | value |= MUTEX_TYPE_BITS_RECURSIVE; |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 268 | break; |
| 269 | case PTHREAD_MUTEX_ERRORCHECK: |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 270 | value |= MUTEX_TYPE_BITS_ERRORCHECK; |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 271 | break; |
| 272 | default: |
| 273 | return EINVAL; |
| 274 | } |
| 275 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 276 | atomic_init(mutex_value_ptr, value); |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 277 | return 0; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 278 | } |
| 279 | |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 280 | static inline int __pthread_normal_mutex_trylock(atomic_int* mutex_value_ptr, int shared) { |
| 281 | const int unlocked = shared | MUTEX_STATE_BITS_UNLOCKED; |
| 282 | const int locked_uncontended = shared | MUTEX_STATE_BITS_LOCKED_UNCONTENDED; |
| 283 | |
| 284 | int mvalue = unlocked; |
| 285 | if (__predict_true(atomic_compare_exchange_strong_explicit(mutex_value_ptr, &mvalue, |
| 286 | locked_uncontended, |
| 287 | memory_order_acquire, |
| 288 | memory_order_relaxed))) { |
| 289 | return 0; |
| 290 | } |
| 291 | return EBUSY; |
| 292 | } |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 293 | |
| 294 | /* |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 295 | * Lock a mutex of type NORMAL. |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 296 | * |
| 297 | * As noted above, there are three states: |
| 298 | * 0 (unlocked, no contention) |
| 299 | * 1 (locked, no contention) |
| 300 | * 2 (locked, contention) |
| 301 | * |
| 302 | * Non-recursive mutexes don't use the thread-id or counter fields, and the |
| 303 | * "type" value is zero, so the only bits that will be set are the ones in |
| 304 | * the lock state field. |
| 305 | */ |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 306 | static inline int __pthread_normal_mutex_lock(atomic_int* mutex_value_ptr, int shared, |
| 307 | const timespec* abs_timeout_or_null, clockid_t clock) { |
| 308 | if (__predict_true(__normal_mutex_trylock(mutex_value_ptr, shared) == 0)) { |
| 309 | return 0; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 310 | } |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 311 | |
| 312 | ScopedTrace trace("Contending for pthread mutex"); |
| 313 | |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 314 | const int unlocked = shared | MUTEX_STATE_BITS_UNLOCKED; |
| 315 | const int locked_contended = shared | MUTEX_STATE_BITS_LOCKED_CONTENDED; |
| 316 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 317 | // We want to go to sleep until the mutex is available, which requires |
| 318 | // promoting it to locked_contended. We need to swap in the new state |
| 319 | // value and then wait until somebody wakes us up. |
| 320 | // An atomic_exchange is used to compete with other threads for the lock. |
| 321 | // If it returns unlocked, we have acquired the lock, otherwise another |
| 322 | // thread still holds the lock and we should wait again. |
| 323 | // If lock is acquired, an acquire fence is needed to make all memory accesses |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 324 | // made by other threads visible to the current CPU. |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 325 | while (atomic_exchange_explicit(mutex_value_ptr, locked_contended, |
| 326 | memory_order_acquire) != unlocked) { |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 327 | timespec ts; |
| 328 | timespec* rel_timeout = NULL; |
| 329 | if (abs_timeout_or_null != NULL) { |
| 330 | rel_timeout = &ts; |
| 331 | if (!timespec_from_absolute_timespec(*rel_timeout, *abs_timeout_or_null, clock)) { |
| 332 | return ETIMEDOUT; |
| 333 | } |
| 334 | } |
| 335 | if (__futex_wait_ex(mutex_value_ptr, shared, locked_contended, rel_timeout) == -ETIMEDOUT) { |
| 336 | return ETIMEDOUT; |
| 337 | } |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 338 | } |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 339 | return 0; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 340 | } |
| 341 | |
| 342 | /* |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 343 | * Release a mutex of type NORMAL. The caller is responsible for determining |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 344 | * that we are in fact the owner of this lock. |
| 345 | */ |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 346 | static inline void __pthread_normal_mutex_unlock(atomic_int* mutex_value_ptr, int shared) { |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 347 | const int unlocked = shared | MUTEX_STATE_BITS_UNLOCKED; |
| 348 | const int locked_contended = shared | MUTEX_STATE_BITS_LOCKED_CONTENDED; |
Andy McFadden | fcd00eb | 2010-05-28 13:31:45 -0700 | [diff] [blame] | 349 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 350 | // We use an atomic_exchange to release the lock. If locked_contended state |
| 351 | // is returned, some threads is waiting for the lock and we need to wake up |
| 352 | // one of them. |
| 353 | // A release fence is required to make previous stores visible to next |
| 354 | // lock owner threads. |
| 355 | if (atomic_exchange_explicit(mutex_value_ptr, unlocked, |
| 356 | memory_order_release) == locked_contended) { |
| 357 | // Wake up one waiting thread. We don't know which thread will be |
| 358 | // woken or when it'll start executing -- futexes make no guarantees |
| 359 | // here. There may not even be a thread waiting. |
| 360 | // |
| 361 | // The newly-woken thread will replace the unlocked state we just set above |
| 362 | // with locked_contended state, which means that when it eventually releases |
| 363 | // the mutex it will also call FUTEX_WAKE. This results in one extra wake |
| 364 | // call whenever a lock is contended, but let us avoid forgetting anyone |
| 365 | // without requiring us to track the number of sleepers. |
| 366 | // |
| 367 | // It's possible for another thread to sneak in and grab the lock between |
| 368 | // the exchange above and the wake call below. If the new thread is "slow" |
| 369 | // and holds the lock for a while, we'll wake up a sleeper, which will swap |
| 370 | // in locked_uncontended state and then go back to sleep since the lock is |
| 371 | // still held. If the new thread is "fast", running to completion before |
| 372 | // we call wake, the thread we eventually wake will find an unlocked mutex |
| 373 | // and will execute. Either way we have correct behavior and nobody is |
| 374 | // orphaned on the wait queue. |
| 375 | __futex_wake_ex(mutex_value_ptr, shared, 1); |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 376 | } |
| 377 | } |
| 378 | |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 379 | /* This common inlined function is used to increment the counter of a recursive mutex. |
David 'Digit' Turner | 022d303 | 2011-12-07 14:02:17 +0100 | [diff] [blame] | 380 | * |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 381 | * If the counter overflows, it will return EAGAIN. |
| 382 | * Otherwise, it atomically increments the counter and returns 0. |
David 'Digit' Turner | 022d303 | 2011-12-07 14:02:17 +0100 | [diff] [blame] | 383 | * |
David 'Digit' Turner | 022d303 | 2011-12-07 14:02:17 +0100 | [diff] [blame] | 384 | */ |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 385 | static inline int __recursive_increment(atomic_int* mutex_value_ptr, int mvalue) { |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 386 | // Detect recursive lock overflow and return EAGAIN. |
| 387 | // This is safe because only the owner thread can modify the |
| 388 | // counter bits in the mutex value. |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 389 | if (MUTEX_COUNTER_BITS_WILL_OVERFLOW(mvalue)) { |
David 'Digit' Turner | 022d303 | 2011-12-07 14:02:17 +0100 | [diff] [blame] | 390 | return EAGAIN; |
| 391 | } |
| 392 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 393 | // We own the mutex, but other threads are able to change the lower bits |
| 394 | // (e.g. promoting it to "contended"), so we need to use an atomic exchange |
| 395 | // loop to update the counter. The counter will not overflow in the loop, |
| 396 | // as only the owner thread can change it. |
| 397 | // The mutex is still locked, so we don't need a release fence. |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 398 | atomic_fetch_add_explicit(mutex_value_ptr, MUTEX_COUNTER_BITS_ONE, memory_order_relaxed); |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 399 | return 0; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 400 | } |
| 401 | |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 402 | static int __pthread_mutex_lock_with_timeout(pthread_mutex_t* mutex, |
| 403 | const timespec* abs_timeout_or_null, clockid_t clock) { |
| 404 | atomic_int* mutex_value_ptr = get_mutex_value_pointer(mutex); |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 405 | |
Wink Saville | a12c544 | 2013-01-08 15:15:45 -0800 | [diff] [blame] | 406 | int mvalue, mtype, tid, shared; |
David 'Digit' Turner | ba9c6f0 | 2010-03-10 16:44:08 -0800 | [diff] [blame] | 407 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 408 | mvalue = atomic_load_explicit(mutex_value_ptr, memory_order_relaxed); |
David 'Digit' Turner | 022d303 | 2011-12-07 14:02:17 +0100 | [diff] [blame] | 409 | mtype = (mvalue & MUTEX_TYPE_MASK); |
| 410 | shared = (mvalue & MUTEX_SHARED_MASK); |
Fabrice Di Meglio | 8641833 | 2010-03-11 14:47:47 -0800 | [diff] [blame] | 411 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 412 | // Handle common case first. |
Elliott Hughes | d4e753f | 2013-07-16 12:45:46 -0700 | [diff] [blame] | 413 | if ( __predict_true(mtype == MUTEX_TYPE_BITS_NORMAL) ) { |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 414 | return __pthread_normal_mutex_lock(mutex_value_ptr, shared, abs_timeout_or_null, clock); |
David 'Digit' Turner | ba9c6f0 | 2010-03-10 16:44:08 -0800 | [diff] [blame] | 415 | } |
David 'Digit' Turner | 40e6b82 | 2010-03-17 11:25:46 -0700 | [diff] [blame] | 416 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 417 | // Do we already own this recursive or error-check mutex? |
Elliott Hughes | 40eabe2 | 2013-02-14 18:59:37 -0800 | [diff] [blame] | 418 | tid = __get_thread()->tid; |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 419 | if (tid == MUTEX_OWNER_FROM_BITS(mvalue)) { |
| 420 | if (mtype == MUTEX_TYPE_BITS_ERRORCHECK) { |
| 421 | return EDEADLK; |
| 422 | } |
| 423 | return __recursive_increment(mutex_value_ptr, mvalue); |
| 424 | } |
David 'Digit' Turner | 40e6b82 | 2010-03-17 11:25:46 -0700 | [diff] [blame] | 425 | |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 426 | const int unlocked = mtype | shared | MUTEX_STATE_BITS_UNLOCKED; |
| 427 | const int locked_uncontended = mtype | shared | MUTEX_STATE_BITS_LOCKED_UNCONTENDED; |
| 428 | const int locked_contended = mtype | shared | MUTEX_STATE_BITS_LOCKED_CONTENDED; |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 429 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 430 | // First, if the mutex is unlocked, try to quickly acquire it. |
| 431 | // In the optimistic case where this works, set the state to locked_uncontended. |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 432 | if (mvalue == unlocked) { |
| 433 | int newval = MUTEX_OWNER_TO_BITS(tid) | locked_uncontended; |
| 434 | // If exchanged successfully, an acquire fence is required to make |
| 435 | // all memory accesses made by other threads visible to the current CPU. |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 436 | if (__predict_true(atomic_compare_exchange_strong_explicit(mutex_value_ptr, &mvalue, |
| 437 | newval, memory_order_acquire, memory_order_relaxed))) { |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 438 | return 0; |
David 'Digit' Turner | 40e6b82 | 2010-03-17 11:25:46 -0700 | [diff] [blame] | 439 | } |
David 'Digit' Turner | 40e6b82 | 2010-03-17 11:25:46 -0700 | [diff] [blame] | 440 | } |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 441 | |
Brigid Smith | a406ee6 | 2014-07-21 15:38:06 -0700 | [diff] [blame] | 442 | ScopedTrace trace("Contending for pthread mutex"); |
| 443 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 444 | while (true) { |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 445 | if (mvalue == unlocked) { |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 446 | // NOTE: We put the state to locked_contended since we _know_ there |
| 447 | // is contention when we are in this loop. This ensures all waiters |
| 448 | // will be unlocked. |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 449 | |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 450 | int newval = MUTEX_OWNER_TO_BITS(tid) | locked_contended; |
| 451 | // If exchanged successfully, an acquire fence is required to make |
| 452 | // all memory accesses made by other threads visible to the current CPU. |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 453 | if (__predict_true(atomic_compare_exchange_weak_explicit(mutex_value_ptr, |
| 454 | &mvalue, newval, |
| 455 | memory_order_acquire, |
| 456 | memory_order_relaxed))) { |
| 457 | return 0; |
| 458 | } |
| 459 | continue; |
| 460 | } else if (MUTEX_STATE_BITS_IS_LOCKED_UNCONTENDED(mvalue)) { |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 461 | // We should set it to locked_contended beforing going to sleep. This can make |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 462 | // sure waiters will be woken up eventually. |
| 463 | |
| 464 | int newval = MUTEX_STATE_BITS_FLIP_CONTENTION(mvalue); |
| 465 | if (__predict_false(!atomic_compare_exchange_weak_explicit(mutex_value_ptr, |
| 466 | &mvalue, newval, |
| 467 | memory_order_relaxed, |
| 468 | memory_order_relaxed))) { |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 469 | continue; |
| 470 | } |
| 471 | mvalue = newval; |
| 472 | } |
| 473 | |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 474 | // We are in locked_contended state, sleep until someone wakes us up. |
| 475 | timespec ts; |
| 476 | timespec* rel_timeout = NULL; |
| 477 | if (abs_timeout_or_null != NULL) { |
| 478 | rel_timeout = &ts; |
| 479 | if (!timespec_from_absolute_timespec(*rel_timeout, *abs_timeout_or_null, clock)) { |
| 480 | return ETIMEDOUT; |
| 481 | } |
| 482 | } |
| 483 | if (__futex_wait_ex(mutex_value_ptr, shared, mvalue, rel_timeout) == -ETIMEDOUT) { |
| 484 | return ETIMEDOUT; |
| 485 | } |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 486 | mvalue = atomic_load_explicit(mutex_value_ptr, memory_order_relaxed); |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 487 | } |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 488 | } |
| 489 | |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 490 | int pthread_mutex_lock(pthread_mutex_t* mutex) { |
| 491 | atomic_int* mutex_value_ptr = get_mutex_value_pointer(mutex); |
| 492 | |
| 493 | int mvalue = atomic_load_explicit(mutex_value_ptr, memory_order_relaxed); |
| 494 | int mtype = (mvalue & MUTEX_TYPE_MASK); |
| 495 | int shared = (mvalue & MUTEX_SHARED_MASK); |
| 496 | // Avoid slowing down fast path of normal mutex lock operation. |
| 497 | if (__predict_true(mtype == MUTEX_TYPE_BITS_NORMAL)) { |
| 498 | if (__predict_true(__pthread_normal_mutex_trylock(mutex_value_ptr, shared) == 0)) { |
| 499 | return 0; |
| 500 | } |
| 501 | } |
| 502 | return __pthread_mutex_lock_with_timeout(mutex, NULL, 0); |
| 503 | } |
| 504 | |
Elliott Hughes | 07f1ded | 2014-05-14 11:35:49 -0700 | [diff] [blame] | 505 | int pthread_mutex_unlock(pthread_mutex_t* mutex) { |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 506 | atomic_int* mutex_value_ptr = get_mutex_value_pointer(mutex); |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 507 | |
Wink Saville | a12c544 | 2013-01-08 15:15:45 -0800 | [diff] [blame] | 508 | int mvalue, mtype, tid, shared; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 509 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 510 | mvalue = atomic_load_explicit(mutex_value_ptr, memory_order_relaxed); |
David 'Digit' Turner | 022d303 | 2011-12-07 14:02:17 +0100 | [diff] [blame] | 511 | mtype = (mvalue & MUTEX_TYPE_MASK); |
| 512 | shared = (mvalue & MUTEX_SHARED_MASK); |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 513 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 514 | // Handle common case first. |
Elliott Hughes | d4e753f | 2013-07-16 12:45:46 -0700 | [diff] [blame] | 515 | if (__predict_true(mtype == MUTEX_TYPE_BITS_NORMAL)) { |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 516 | __pthread_normal_mutex_unlock(mutex_value_ptr, shared); |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 517 | return 0; |
| 518 | } |
David 'Digit' Turner | 40e6b82 | 2010-03-17 11:25:46 -0700 | [diff] [blame] | 519 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 520 | // Do we already own this recursive or error-check mutex? |
Elliott Hughes | 40eabe2 | 2013-02-14 18:59:37 -0800 | [diff] [blame] | 521 | tid = __get_thread()->tid; |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 522 | if ( tid != MUTEX_OWNER_FROM_BITS(mvalue) ) |
David 'Digit' Turner | 40e6b82 | 2010-03-17 11:25:46 -0700 | [diff] [blame] | 523 | return EPERM; |
| 524 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 525 | // If the counter is > 0, we can simply decrement it atomically. |
| 526 | // Since other threads can mutate the lower state bits (and only the |
| 527 | // lower state bits), use a compare_exchange loop to do it. |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 528 | if (!MUTEX_COUNTER_BITS_IS_ZERO(mvalue)) { |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 529 | // We still own the mutex, so a release fence is not needed. |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 530 | atomic_fetch_sub_explicit(mutex_value_ptr, MUTEX_COUNTER_BITS_ONE, memory_order_relaxed); |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 531 | return 0; |
David 'Digit' Turner | 40e6b82 | 2010-03-17 11:25:46 -0700 | [diff] [blame] | 532 | } |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 533 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 534 | // The counter is 0, so we'are going to unlock the mutex by resetting its |
| 535 | // state to unlocked, we need to perform a atomic_exchange inorder to read |
| 536 | // the current state, which will be locked_contended if there may have waiters |
| 537 | // to awake. |
| 538 | // A release fence is required to make previous stores visible to next |
| 539 | // lock owner threads. |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 540 | const int unlocked = mtype | shared | MUTEX_STATE_BITS_UNLOCKED; |
| 541 | mvalue = atomic_exchange_explicit(mutex_value_ptr, unlocked, memory_order_release); |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 542 | if (MUTEX_STATE_BITS_IS_LOCKED_CONTENDED(mvalue)) { |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 543 | __futex_wake_ex(mutex_value_ptr, shared, 1); |
David 'Digit' Turner | 88f06cd | 2010-03-18 17:13:41 -0700 | [diff] [blame] | 544 | } |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 545 | |
David 'Digit' Turner | 40e6b82 | 2010-03-17 11:25:46 -0700 | [diff] [blame] | 546 | return 0; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 547 | } |
| 548 | |
Elliott Hughes | 07f1ded | 2014-05-14 11:35:49 -0700 | [diff] [blame] | 549 | int pthread_mutex_trylock(pthread_mutex_t* mutex) { |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 550 | atomic_int* mutex_value_ptr = get_mutex_value_pointer(mutex); |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 551 | |
| 552 | int mvalue = atomic_load_explicit(mutex_value_ptr, memory_order_relaxed); |
Elliott Hughes | 5b1111a | 2014-10-24 19:33:11 -0700 | [diff] [blame] | 553 | int mtype = (mvalue & MUTEX_TYPE_MASK); |
| 554 | int shared = (mvalue & MUTEX_SHARED_MASK); |
David 'Digit' Turner | 40e6b82 | 2010-03-17 11:25:46 -0700 | [diff] [blame] | 555 | |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 556 | const int unlocked = mtype | shared | MUTEX_STATE_BITS_UNLOCKED; |
| 557 | const int locked_uncontended = mtype | shared | MUTEX_STATE_BITS_LOCKED_UNCONTENDED; |
| 558 | |
Elliott Hughes | 5b1111a | 2014-10-24 19:33:11 -0700 | [diff] [blame] | 559 | // Handle common case first. |
| 560 | if (__predict_true(mtype == MUTEX_TYPE_BITS_NORMAL)) { |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 561 | return __pthread_normal_mutex_trylock(mutex_value_ptr, shared); |
David 'Digit' Turner | ba9c6f0 | 2010-03-10 16:44:08 -0800 | [diff] [blame] | 562 | } |
David 'Digit' Turner | 40e6b82 | 2010-03-17 11:25:46 -0700 | [diff] [blame] | 563 | |
Elliott Hughes | 5b1111a | 2014-10-24 19:33:11 -0700 | [diff] [blame] | 564 | // Do we already own this recursive or error-check mutex? |
| 565 | pid_t tid = __get_thread()->tid; |
| 566 | if (tid == MUTEX_OWNER_FROM_BITS(mvalue)) { |
| 567 | if (mtype == MUTEX_TYPE_BITS_ERRORCHECK) { |
| 568 | return EBUSY; |
| 569 | } |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 570 | return __recursive_increment(mutex_value_ptr, mvalue); |
Elliott Hughes | 5b1111a | 2014-10-24 19:33:11 -0700 | [diff] [blame] | 571 | } |
David 'Digit' Turner | 40e6b82 | 2010-03-17 11:25:46 -0700 | [diff] [blame] | 572 | |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 573 | // Same as pthread_mutex_lock, except that we don't want to wait, and |
| 574 | // the only operation that can succeed is a single compare_exchange to acquire the |
| 575 | // lock if it is released / not owned by anyone. No need for a complex loop. |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 576 | // If exchanged successfully, an acquire fence is required to make |
| 577 | // all memory accesses made by other threads visible to the current CPU. |
| 578 | mvalue = unlocked; |
| 579 | int newval = MUTEX_OWNER_TO_BITS(tid) | locked_uncontended; |
| 580 | if (__predict_true(atomic_compare_exchange_strong_explicit(mutex_value_ptr, &mvalue, newval, |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 581 | memory_order_acquire, |
| 582 | memory_order_relaxed))) { |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 583 | return 0; |
| 584 | } |
David 'Digit' Turner | e1414aa | 2012-01-24 15:26:54 +0100 | [diff] [blame] | 585 | return EBUSY; |
The Android Open Source Project | 1dc9e47 | 2009-03-03 19:28:35 -0800 | [diff] [blame] | 586 | } |
| 587 | |
Elliott Hughes | 0e714a5 | 2014-03-03 16:42:47 -0800 | [diff] [blame] | 588 | #if !defined(__LP64__) |
| 589 | extern "C" int pthread_mutex_lock_timeout_np(pthread_mutex_t* mutex, unsigned ms) { |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 590 | timespec abs_timeout; |
| 591 | clock_gettime(CLOCK_MONOTONIC, &abs_timeout); |
| 592 | abs_timeout.tv_sec += ms / 1000; |
| 593 | abs_timeout.tv_nsec += (ms % 1000) * 1000000; |
| 594 | if (abs_timeout.tv_nsec >= NS_PER_S) { |
| 595 | abs_timeout.tv_sec++; |
| 596 | abs_timeout.tv_nsec -= NS_PER_S; |
| 597 | } |
Elliott Hughes | 0e714a5 | 2014-03-03 16:42:47 -0800 | [diff] [blame] | 598 | |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 599 | int error = __pthread_mutex_lock_with_timeout(mutex, &abs_timeout, CLOCK_MONOTONIC); |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 600 | if (error == ETIMEDOUT) { |
| 601 | error = EBUSY; |
| 602 | } |
| 603 | return error; |
Elliott Hughes | 0e714a5 | 2014-03-03 16:42:47 -0800 | [diff] [blame] | 604 | } |
| 605 | #endif |
| 606 | |
| 607 | int pthread_mutex_timedlock(pthread_mutex_t* mutex, const timespec* abs_timeout) { |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 608 | return __pthread_mutex_lock_with_timeout(mutex, abs_timeout, CLOCK_REALTIME); |
Mathias Agopian | 7c0c379 | 2011-09-05 23:54:55 -0700 | [diff] [blame] | 609 | } |
| 610 | |
Elliott Hughes | 07f1ded | 2014-05-14 11:35:49 -0700 | [diff] [blame] | 611 | int pthread_mutex_destroy(pthread_mutex_t* mutex) { |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 612 | // Use trylock to ensure that the mutex is valid and not already locked. |
| 613 | int error = pthread_mutex_trylock(mutex); |
| 614 | if (error != 0) { |
| 615 | return error; |
| 616 | } |
| 617 | |
Yabin Cui | 5b8e7cd | 2015-03-04 17:36:59 -0800 | [diff] [blame^] | 618 | atomic_int* mutex_value_ptr = get_mutex_value_pointer(mutex); |
Yabin Cui | 86fc96f | 2015-01-29 21:50:48 -0800 | [diff] [blame] | 619 | atomic_store_explicit(mutex_value_ptr, 0xdead10cc, memory_order_relaxed); |
| 620 | return 0; |
Mathias Agopian | 7c0c379 | 2011-09-05 23:54:55 -0700 | [diff] [blame] | 621 | } |