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