 4986ebcb67
			
		
	
	4986ebcb67
	
	
	
		
			
			Rename heim_base_atomic_max to heim_base_atomic_integer_max (ditto with _type) in order to better reflect their usage, now we have a separate heim_base_atomic() macro for making an atomic version of any type.
		
			
				
	
	
		
			256 lines
		
	
	
		
			7.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			256 lines
		
	
	
		
			7.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  * Copyright (c) 2010 Kungliga Tekniska Högskolan
 | |
|  * (Royal Institute of Technology, Stockholm, Sweden).
 | |
|  * All rights reserved.
 | |
|  *
 | |
|  * Portions Copyright (c) 2010 Apple Inc. All rights reserved.
 | |
|  *
 | |
|  * Redistribution and use in source and binary forms, with or without
 | |
|  * modification, are permitted provided that the following conditions
 | |
|  * are met:
 | |
|  *
 | |
|  * 1. Redistributions of source code must retain the above copyright
 | |
|  *    notice, this list of conditions and the following disclaimer.
 | |
|  *
 | |
|  * 2. Redistributions in binary form must reproduce the above copyright
 | |
|  *    notice, this list of conditions and the following disclaimer in the
 | |
|  *    documentation and/or other materials provided with the distribution.
 | |
|  *
 | |
|  * 3. Neither the name of the Institute nor the names of its contributors
 | |
|  *    may be used to endorse or promote products derived from this software
 | |
|  *    without specific prior written permission.
 | |
|  *
 | |
|  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
 | |
|  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 | |
|  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 | |
|  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
 | |
|  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 | |
|  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 | |
|  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 | |
|  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 | |
|  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 | |
|  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 | |
|  * SUCH DAMAGE.
 | |
|  */
 | |
| 
 | |
| #ifndef HEIM_BASE_ATOMICS_H
 | |
| #define HEIM_BASE_ATOMICS_H 1
 | |
| 
 | |
| #include "config.h"
 | |
| 
 | |
| /*
 | |
|  * Atomic operations
 | |
|  */
 | |
| 
 | |
| #if defined(HAVE_STDATOMIC_H)
 | |
| 
 | |
| #include <stdatomic.h>
 | |
| 
 | |
| #define heim_base_atomic_init(t, v)	atomic_init(t, v)
 | |
| #define heim_base_atomic_load(x)	atomic_load((x))
 | |
| #define heim_base_atomic_store(t, v)	atomic_store((t), (v))
 | |
| 
 | |
| #define heim_base_atomic(T)		_Atomic(T)
 | |
| 
 | |
| #define heim_base_atomic_inc(x)		(atomic_fetch_add((x), 1) + 1)
 | |
| #define heim_base_atomic_dec(x)		(atomic_fetch_sub((x), 1) - 1)
 | |
| #define heim_base_atomic_integer_type	heim_base_atomic(unsigned int)
 | |
| #define heim_base_atomic_integer_max	UINT_MAX
 | |
| 
 | |
| #define heim_base_exchange_pointer(t,v) atomic_exchange((t), (v))
 | |
| #define heim_base_exchange_32(t,v)	atomic_exchange((t), (v))
 | |
| #define heim_base_exchange_64(t,v)	atomic_exchange((t), (v))
 | |
| 
 | |
| #elif defined(__GNUC__) && defined(HAVE___SYNC_ADD_AND_FETCH)
 | |
| 
 | |
| #define heim_base_atomic_barrier()	__sync_synchronize()
 | |
| 
 | |
| #define heim_base_atomic_inc(x)		__sync_add_and_fetch((x), 1)
 | |
| #define heim_base_atomic_dec(x)		__sync_sub_and_fetch((x), 1)
 | |
| #define heim_base_atomic_integer_type	unsigned int
 | |
| #define heim_base_atomic_integer_max	UINT_MAX
 | |
| 
 | |
| #ifndef __has_builtin
 | |
| #define __has_builtin(x) 0
 | |
| #endif
 | |
| 
 | |
| #if __has_builtin(__sync_swap)
 | |
| #define heim_base_exchange_pointer(t,v) __sync_swap((t), (v))
 | |
| #else
 | |
| /* FIXME: some targets may only write the value 1 into *t */
 | |
| #define heim_base_exchange_pointer(t,v) __sync_lock_test_and_set((t), (v))
 | |
| #endif
 | |
| 
 | |
| #define heim_base_exchange_32(t,v)	heim_base_exchange_pointer((t), (v))
 | |
| #define heim_base_exchange_64(t,v)	heim_base_exchange_pointer((t), (v))
 | |
| 
 | |
| #elif defined(__sun)
 | |
| 
 | |
| #include <sys/atomic.h>
 | |
| 
 | |
| #define heim_base_atomic_barrier()	__machine_rw_barrier()
 | |
| 
 | |
| #define heim_base_atomic_inc(x)		atomic_inc_uint_nv((volatile uint_t *)(x))
 | |
| #define heim_base_atomic_dec(x)		atomic_dec_uint_nv((volatile uint_t *)(x))
 | |
| #define heim_base_atomic_integer_type	uint_t
 | |
| #define heim_base_atomic_integer_max	UINT_MAX
 | |
| 
 | |
| #define heim_base_exchange_pointer(t,v) atomic_swap_ptr((volatile void *)(t), (void *)(v))
 | |
| #define heim_base_exchange_32(t,v)	atomic_swap_32((volatile uint32_t *)(t), (v))
 | |
| #define heim_base_exchange_64(t,v)	atomic_swap_64((volatile uint64_t *)(t), (v))
 | |
| 
 | |
| #elif defined(_AIX)
 | |
| 
 | |
| #include <sys/atomic_op.h>
 | |
| 
 | |
| #define heim_base_atomic_barrier()	__isync()
 | |
| 
 | |
| #define heim_base_atomic_inc(x)		(fetch_and_add((atomic_p)(x)) + 1)
 | |
| #define heim_base_atomic_dec(x)		(fetch_and_add((atomic_p)(x)) - 1)
 | |
| #define heim_base_atomic_integer_type	unsigned int
 | |
| #define heim_base_atomic_integer_max	UINT_MAX
 | |
| 
 | |
| static inline void *
 | |
| heim_base_exchange_pointer(void *p, void *newval)
 | |
| {
 | |
|     void *val = *(void **)p;
 | |
| 
 | |
|     while (!compare_and_swaplp((atomic_l)p, (long *)&val, (long)newval))
 | |
|         ;
 | |
| 
 | |
|     return val;
 | |
| }
 | |
| 
 | |
| static inline uint32_t
 | |
| heim_base_exchange_32(uint32_t *p, uint32_t newval)
 | |
| {
 | |
|     uint32_t val = *p;
 | |
| 
 | |
|     while (!compare_and_swap((atomic_p)p, (int *)&val, (int)newval))
 | |
|         ;
 | |
| 
 | |
|     return val;
 | |
| }
 | |
| 
 | |
| static inline uint64_t
 | |
| heim_base_exchange_64(uint64_t *p, uint64_t newval)
 | |
| {
 | |
|     uint64_t val = *p;
 | |
| 
 | |
|     while (!compare_and_swaplp((atomic_l)p, (long *)&val, (long)newval))
 | |
|         ;
 | |
| 
 | |
|     return val;
 | |
| }
 | |
| 
 | |
| #elif defined(_WIN32)
 | |
| 
 | |
| #define heim_base_atomic_barrier()	MemoryBarrier()
 | |
| 
 | |
| #define heim_base_atomic_inc(x)		InterlockedIncrement(x)
 | |
| #define heim_base_atomic_dec(x)		InterlockedDecrement(x)
 | |
| #define heim_base_atomic_integer_type	LONG
 | |
| #define heim_base_atomic_integer_max	MAXLONG
 | |
| 
 | |
| #define heim_base_exchange_pointer(t,v) InterlockedExchangePointer((PVOID volatile *)(t), (PVOID)(v))
 | |
| #define heim_base_exchange_32(t,v)	((ULONG)InterlockedExchange((LONG volatile *)(t), (LONG)(v)))
 | |
| #define heim_base_exchange_64(t,v)	((ULONG64)InterlockedExchange64((LONG64 violatile *)(t), (LONG64)(v)))
 | |
| 
 | |
| #else
 | |
| 
 | |
| #include <heim_threads.h>
 | |
| 
 | |
| #define HEIM_BASE_NEED_ATOMIC_MUTEX 1
 | |
| extern HEIMDAL_MUTEX _heim_base_mutex;
 | |
| 
 | |
| #define heim_base_atomic_integer_type	unsigned int
 | |
| #define heim_base_atomic_integer_max	UINT_MAX
 | |
| 
 | |
| static inline heim_base_atomic_integer_type
 | |
| heim_base_atomic_inc(heim_base_atomic_integer_type *x)
 | |
| {
 | |
|     heim_base_atomic_integer_type t;
 | |
|     HEIMDAL_MUTEX_lock(&_heim_base_mutex);
 | |
|     t = ++(*x);
 | |
|     HEIMDAL_MUTEX_unlock(&_heim_base_mutex);
 | |
|     return t;
 | |
| }
 | |
| 
 | |
| static inline heim_base_atomic_integer_type
 | |
| heim_base_atomic_dec(heim_base_atomic_integer_type *x)
 | |
| {
 | |
|     heim_base_atomic_integer_type t;
 | |
|     HEIMDAL_MUTEX_lock(&_heim_base_mutex);
 | |
|     t = --(*x);
 | |
|     HEIMDAL_MUTEX_unlock(&_heim_base_mutex);
 | |
|     return t;
 | |
| }
 | |
| 
 | |
| static inline void *
 | |
| heim_base_exchange_pointer(void *target, void *value)
 | |
| {
 | |
|     void *old;
 | |
|     HEIMDAL_MUTEX_lock(&_heim_base_mutex);
 | |
|     old = *(void **)target;
 | |
|     *(void **)target = value;
 | |
|     HEIMDAL_MUTEX_unlock(&_heim_base_mutex);
 | |
|     return old;
 | |
| }
 | |
| 
 | |
| static inline uint32_t
 | |
| heim_base_exchange_32(uint32_t *p, uint32_t newval)
 | |
| {
 | |
|     uint32_t old;
 | |
|     HEIMDAL_MUTEX_lock(&_heim_base_mutex);
 | |
|     old = *p;
 | |
|     *p = newval;
 | |
|     HEIMDAL_MUTEX_unlock(&_heim_base_mutex);
 | |
|     return old;
 | |
| }
 | |
| 
 | |
| static inline uint64_t
 | |
| heim_base_exchange_64(uint64_t *p, uint64_t newval)
 | |
| {
 | |
|     uint64_t old;
 | |
|     HEIMDAL_MUTEX_lock(&_heim_base_mutex);
 | |
|     old = *p;
 | |
|     *p = newval;
 | |
|     HEIMDAL_MUTEX_unlock(&_heim_base_mutex);
 | |
|     return old;
 | |
| }
 | |
| 
 | |
| #endif /* defined(__GNUC__) && defined(HAVE___SYNC_ADD_AND_FETCH) */
 | |
| 
 | |
| #ifndef heim_base_atomic
 | |
| #define heim_base_atomic(T)		T
 | |
| #endif
 | |
| 
 | |
| #ifndef heim_base_atomic_barrier
 | |
| #define heim_base_atomic_barrier()
 | |
| #endif
 | |
| 
 | |
| #ifndef heim_base_atomic_load
 | |
| #define heim_base_atomic_load(x)	(heim_base_atomic_barrier(), *(x))
 | |
| #endif
 | |
| 
 | |
| #ifndef heim_base_atomic_init
 | |
| #define heim_base_atomic_init(t, v)	do { (*(t) = (v)); } while (0)
 | |
| #endif
 | |
| 
 | |
| #ifndef heim_base_atomic_store
 | |
| #define heim_base_atomic_store(t, v)	do {					\
 | |
| 					    (*(t) = (v));			\
 | |
| 					    heim_base_atomic_barrier();		\
 | |
| 					} while (0)
 | |
| #endif
 | |
| 
 | |
| #if SIZEOF_TIME_T == 8
 | |
| #define heim_base_exchange_time_t(t,v)	heim_base_exchange_64((t), (v))
 | |
| #elif SIZEOF_TIME_T == 4
 | |
| #define heim_base_exchange_time_t(t,v)	heim_base_exchange_32((t), (v))
 | |
| #else
 | |
| #error set SIZEOF_TIME_T for your platform
 | |
| #endif
 | |
| 
 | |
| #endif /* HEIM_BASE_ATOMICS_H */
 |