231 lines
		
	
	
		
			5.1 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			231 lines
		
	
	
		
			5.1 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
#include <linux/compiler.h>
 | 
						|
#include <linux/module.h>
 | 
						|
#include <linux/types.h>
 | 
						|
 | 
						|
#include <asm/processor.h>
 | 
						|
#include <asm/cmpxchg.h>
 | 
						|
#include <asm/atomic.h>
 | 
						|
 | 
						|
static noinline u64 cmpxchg8b(u64 *ptr, u64 old, u64 new)
 | 
						|
{
 | 
						|
	u32 low = new;
 | 
						|
	u32 high = new >> 32;
 | 
						|
 | 
						|
	asm volatile(
 | 
						|
		LOCK_PREFIX "cmpxchg8b %1\n"
 | 
						|
		     : "+A" (old), "+m" (*ptr)
 | 
						|
		     :  "b" (low),  "c" (high)
 | 
						|
		     );
 | 
						|
	return old;
 | 
						|
}
 | 
						|
 | 
						|
u64 atomic64_cmpxchg(atomic64_t *ptr, u64 old_val, u64 new_val)
 | 
						|
{
 | 
						|
	return cmpxchg8b(&ptr->counter, old_val, new_val);
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_cmpxchg);
 | 
						|
 | 
						|
/**
 | 
						|
 * atomic64_xchg - xchg atomic64 variable
 | 
						|
 * @ptr:      pointer to type atomic64_t
 | 
						|
 * @new_val:  value to assign
 | 
						|
 *
 | 
						|
 * Atomically xchgs the value of @ptr to @new_val and returns
 | 
						|
 * the old value.
 | 
						|
 */
 | 
						|
u64 atomic64_xchg(atomic64_t *ptr, u64 new_val)
 | 
						|
{
 | 
						|
	/*
 | 
						|
	 * Try first with a (possibly incorrect) assumption about
 | 
						|
	 * what we have there. We'll do two loops most likely,
 | 
						|
	 * but we'll get an ownership MESI transaction straight away
 | 
						|
	 * instead of a read transaction followed by a
 | 
						|
	 * flush-for-ownership transaction:
 | 
						|
	 */
 | 
						|
	u64 old_val, real_val = 0;
 | 
						|
 | 
						|
	do {
 | 
						|
		old_val = real_val;
 | 
						|
 | 
						|
		real_val = atomic64_cmpxchg(ptr, old_val, new_val);
 | 
						|
 | 
						|
	} while (real_val != old_val);
 | 
						|
 | 
						|
	return old_val;
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_xchg);
 | 
						|
 | 
						|
/**
 | 
						|
 * atomic64_set - set atomic64 variable
 | 
						|
 * @ptr:      pointer to type atomic64_t
 | 
						|
 * @new_val:  value to assign
 | 
						|
 *
 | 
						|
 * Atomically sets the value of @ptr to @new_val.
 | 
						|
 */
 | 
						|
void atomic64_set(atomic64_t *ptr, u64 new_val)
 | 
						|
{
 | 
						|
	atomic64_xchg(ptr, new_val);
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_set);
 | 
						|
 | 
						|
/**
 | 
						|
EXPORT_SYMBOL(atomic64_read);
 | 
						|
 * atomic64_add_return - add and return
 | 
						|
 * @delta: integer value to add
 | 
						|
 * @ptr:   pointer to type atomic64_t
 | 
						|
 *
 | 
						|
 * Atomically adds @delta to @ptr and returns @delta + *@ptr
 | 
						|
 */
 | 
						|
noinline u64 atomic64_add_return(u64 delta, atomic64_t *ptr)
 | 
						|
{
 | 
						|
	/*
 | 
						|
	 * Try first with a (possibly incorrect) assumption about
 | 
						|
	 * what we have there. We'll do two loops most likely,
 | 
						|
	 * but we'll get an ownership MESI transaction straight away
 | 
						|
	 * instead of a read transaction followed by a
 | 
						|
	 * flush-for-ownership transaction:
 | 
						|
	 */
 | 
						|
	u64 old_val, new_val, real_val = 0;
 | 
						|
 | 
						|
	do {
 | 
						|
		old_val = real_val;
 | 
						|
		new_val = old_val + delta;
 | 
						|
 | 
						|
		real_val = atomic64_cmpxchg(ptr, old_val, new_val);
 | 
						|
 | 
						|
	} while (real_val != old_val);
 | 
						|
 | 
						|
	return new_val;
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_add_return);
 | 
						|
 | 
						|
u64 atomic64_sub_return(u64 delta, atomic64_t *ptr)
 | 
						|
{
 | 
						|
	return atomic64_add_return(-delta, ptr);
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_sub_return);
 | 
						|
 | 
						|
u64 atomic64_inc_return(atomic64_t *ptr)
 | 
						|
{
 | 
						|
	return atomic64_add_return(1, ptr);
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_inc_return);
 | 
						|
 | 
						|
u64 atomic64_dec_return(atomic64_t *ptr)
 | 
						|
{
 | 
						|
	return atomic64_sub_return(1, ptr);
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_dec_return);
 | 
						|
 | 
						|
/**
 | 
						|
 * atomic64_add - add integer to atomic64 variable
 | 
						|
 * @delta: integer value to add
 | 
						|
 * @ptr:   pointer to type atomic64_t
 | 
						|
 *
 | 
						|
 * Atomically adds @delta to @ptr.
 | 
						|
 */
 | 
						|
void atomic64_add(u64 delta, atomic64_t *ptr)
 | 
						|
{
 | 
						|
	atomic64_add_return(delta, ptr);
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_add);
 | 
						|
 | 
						|
/**
 | 
						|
 * atomic64_sub - subtract the atomic64 variable
 | 
						|
 * @delta: integer value to subtract
 | 
						|
 * @ptr:   pointer to type atomic64_t
 | 
						|
 *
 | 
						|
 * Atomically subtracts @delta from @ptr.
 | 
						|
 */
 | 
						|
void atomic64_sub(u64 delta, atomic64_t *ptr)
 | 
						|
{
 | 
						|
	atomic64_add(-delta, ptr);
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_sub);
 | 
						|
 | 
						|
/**
 | 
						|
 * atomic64_sub_and_test - subtract value from variable and test result
 | 
						|
 * @delta: integer value to subtract
 | 
						|
 * @ptr:   pointer to type atomic64_t
 | 
						|
 *
 | 
						|
 * Atomically subtracts @delta from @ptr and returns
 | 
						|
 * true if the result is zero, or false for all
 | 
						|
 * other cases.
 | 
						|
 */
 | 
						|
int atomic64_sub_and_test(u64 delta, atomic64_t *ptr)
 | 
						|
{
 | 
						|
	u64 new_val = atomic64_sub_return(delta, ptr);
 | 
						|
 | 
						|
	return new_val == 0;
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_sub_and_test);
 | 
						|
 | 
						|
/**
 | 
						|
 * atomic64_inc - increment atomic64 variable
 | 
						|
 * @ptr: pointer to type atomic64_t
 | 
						|
 *
 | 
						|
 * Atomically increments @ptr by 1.
 | 
						|
 */
 | 
						|
void atomic64_inc(atomic64_t *ptr)
 | 
						|
{
 | 
						|
	atomic64_add(1, ptr);
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_inc);
 | 
						|
 | 
						|
/**
 | 
						|
 * atomic64_dec - decrement atomic64 variable
 | 
						|
 * @ptr: pointer to type atomic64_t
 | 
						|
 *
 | 
						|
 * Atomically decrements @ptr by 1.
 | 
						|
 */
 | 
						|
void atomic64_dec(atomic64_t *ptr)
 | 
						|
{
 | 
						|
	atomic64_sub(1, ptr);
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_dec);
 | 
						|
 | 
						|
/**
 | 
						|
 * atomic64_dec_and_test - decrement and test
 | 
						|
 * @ptr: pointer to type atomic64_t
 | 
						|
 *
 | 
						|
 * Atomically decrements @ptr by 1 and
 | 
						|
 * returns true if the result is 0, or false for all other
 | 
						|
 * cases.
 | 
						|
 */
 | 
						|
int atomic64_dec_and_test(atomic64_t *ptr)
 | 
						|
{
 | 
						|
	return atomic64_sub_and_test(1, ptr);
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_dec_and_test);
 | 
						|
 | 
						|
/**
 | 
						|
 * atomic64_inc_and_test - increment and test
 | 
						|
 * @ptr: pointer to type atomic64_t
 | 
						|
 *
 | 
						|
 * Atomically increments @ptr by 1
 | 
						|
 * and returns true if the result is zero, or false for all
 | 
						|
 * other cases.
 | 
						|
 */
 | 
						|
int atomic64_inc_and_test(atomic64_t *ptr)
 | 
						|
{
 | 
						|
	return atomic64_sub_and_test(-1, ptr);
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_inc_and_test);
 | 
						|
 | 
						|
/**
 | 
						|
 * atomic64_add_negative - add and test if negative
 | 
						|
 * @delta: integer value to add
 | 
						|
 * @ptr:   pointer to type atomic64_t
 | 
						|
 *
 | 
						|
 * Atomically adds @delta to @ptr and returns true
 | 
						|
 * if the result is negative, or false when
 | 
						|
 * result is greater than or equal to zero.
 | 
						|
 */
 | 
						|
int atomic64_add_negative(u64 delta, atomic64_t *ptr)
 | 
						|
{
 | 
						|
	s64 new_val = atomic64_add_return(delta, ptr);
 | 
						|
 | 
						|
	return new_val < 0;
 | 
						|
}
 | 
						|
EXPORT_SYMBOL(atomic64_add_negative);
 |