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);
 |