Added atomic compare and swap operations.
Using a GCC atomic builtin if available or a global mutex otherwise.
This commit is contained in:
@@ -344,6 +344,28 @@ bool ref_put(refcount_t *ref)
|
||||
pthread_mutex_unlock(&ref_mutex);
|
||||
return !more_refs;
|
||||
}
|
||||
|
||||
/**
|
||||
* Single mutex for all compare and swap operations.
|
||||
*/
|
||||
static pthread_mutex_t cas_mutex = PTHREAD_MUTEX_INITIALIZER;
|
||||
|
||||
/**
|
||||
* Compare and swap if equal to old value
|
||||
*/
|
||||
#define _cas_impl(name, type) \
|
||||
bool cas_##name(type *ptr, type oldval, type newval) \
|
||||
{ \
|
||||
bool swapped; \
|
||||
pthread_mutex_lock(&cas_mutex); \
|
||||
if ((swapped = (*ptr == oldval))) { *ptr = newval; } \
|
||||
pthread_mutex_unlock(&cas_mutex); \
|
||||
return swapped; \
|
||||
}
|
||||
|
||||
_cas_impl(bool, bool)
|
||||
_cas_impl(ptr, void*)
|
||||
|
||||
#endif /* HAVE_GCC_ATOMIC_OPERATIONS */
|
||||
|
||||
/**
|
||||
|
||||
@@ -565,6 +565,11 @@ typedef volatile u_int refcount_t;
|
||||
#define ref_get(ref) {__sync_fetch_and_add(ref, 1); }
|
||||
#define ref_put(ref) (!__sync_sub_and_fetch(ref, 1))
|
||||
|
||||
#define cas_bool(ptr, oldval, newval) \
|
||||
(__sync_bool_compare_and_swap(ptr, oldval, newval))
|
||||
#define cas_ptr(ptr, oldval, newval) \
|
||||
(__sync_bool_compare_and_swap(ptr, oldval, newval))
|
||||
|
||||
#else /* !HAVE_GCC_ATOMIC_OPERATIONS */
|
||||
|
||||
/**
|
||||
@@ -587,6 +592,27 @@ void ref_get(refcount_t *ref);
|
||||
*/
|
||||
bool ref_put(refcount_t *ref);
|
||||
|
||||
/**
|
||||
* Atomically replace value of ptr with newval if it currently equals oldval.
|
||||
*
|
||||
* @param ptr pointer to variable
|
||||
* @param oldval old value of the variable
|
||||
* @param newval new value set if possible
|
||||
* @return TRUE if value equaled oldval and newval was written
|
||||
*/
|
||||
bool cas_bool(bool *ptr, bool oldval, bool newval);
|
||||
|
||||
/**
|
||||
* Atomically replace value of ptr with newval if it currently equals oldval.
|
||||
*
|
||||
* @param ptr pointer to variable
|
||||
* @param oldval old value of the variable
|
||||
* @param newval new value set if possible
|
||||
* @return TRUE if value equaled oldval and newval was written
|
||||
*/
|
||||
bool cas_ptr(void **ptr, void *oldval, void *newval);
|
||||
|
||||
|
||||
#endif /* HAVE_GCC_ATOMIC_OPERATIONS */
|
||||
|
||||
/**
|
||||
|
||||
Reference in New Issue
Block a user