linux内核原子操作学习

2024-09-02 7

版权

本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。

简介： linux内核原子操作学习

参考

https://www.kernel.org/doc/html/latest/staging/index.html#atomic-types

说明

下面注释里说的自增和自减表示的是在原子变量旧值的基础上
这里列举的原子操作是以32位为例的，如果是64位，那么把前缀atomic替换成atomic64即可

数据类型

原型	说明
atomic_t	数据位宽是32位
atomic64_t	数据位宽是64位
atomic_long_t	在64位上等于atomic64_t，在32位系统上等于atomic_t

初始化

#define ATOMIC_INIT(i) { (i) }
#define ATOMIC64_INIT(i)  { (i) }

示例：

atomic_t cnt = ATOMIC_INIT(0);
atomic64_t v = ATOMIC64_INIT(v0);

赋值操作

原型	说明	返回值
`void atomic_set(atomic_t *v, int i)`	将`i`赋值给原子变量`v`	无

读操作

原型	说明	返回值
`int atomic_read(const atomic_t *v)`		原子变量`v`的数值

linux还提供了带条件的读取方式，即如果条件不满足，就一直读取：

#define atomic_cond_read_acquire(v, c) smp_cond_load_acquire(&(v)->counter, (c))
#define atomic_cond_read_relaxed(v, c) smp_cond_load_relaxed(&(v)->counter, (c))
#define atomic64_cond_read_acquire(v, c) smp_cond_load_acquire(&(v)->counter, (c))
#define atomic64_cond_read_relaxed(v, c) smp_cond_load_relaxed(&(v)->counter, (c))

其中smp_cond_load_acquire或者smp_cond_load_relaxed中有一个死循环操作，如果c表示的condition不成立，那么会在循环中一直读取地址的值，知道条件成立。

#define smp_cond_load_acquire(ptr, cond_expr)       \
({                  \
  typeof(ptr) __PTR = (ptr);          \
  __unqual_scalar_typeof(*ptr) VAL;       \
  for (;;) {              \
    VAL = smp_load_acquire(__PTR);        \
    if (cond_expr)            \
      break;            \
    __cmpwait_relaxed(__PTR, VAL);        \
  }               \
  (typeof(*ptr))VAL;            \
})

加操作

原型	说明	返回值
`void atomic_inc(atomic_t *v)`	自增`1`	无
`int atomic_inc_return(atomic_t *v)`	自增`1`	返回新值
`int atomic_fetch_inc(atomic_t *v)`	自增`1`	返回旧值
`void atomic_add(int i, atomic_t *v)`	自增`i`	无
`int atomic_add_return(int i, atomic_t *v)`	自增`i`	返回新值
`int atomic_fetch_add(int i, atomic_t *v)`	自增`i`	返回旧值

`bool atomic_inc_and_test(atomic_t *v)`	自增`1`	如果新值为0，返回`true`；否则返回`false`

`bool atomic_add_negative(int i, atomic_t *v)`	自增`i`	如果新值为负数，返回`true`；否则返回`false`

`int atomic_fetch_add_unless(atomic_t *v, int a, int u)`	`unless`表示`if not`。如果旧值跟`u`不等，自增`a`，返回旧值；如果旧值等于`u`，只返回旧值，不会进行自增操作	返回旧值
`bool atomic_add_unless(atomic_t *v, int a, int u)`	如果旧值跟`u`不等，自增`a`，返回`true`；如果旧值等于`u`，返回旧值`false`，不会进行自增操作	表示是否进行了自增
`bool atomic_inc_not_zero(atomic_t *v)`	如果旧值不等于0，那么自增`1`，返回`true`；如果旧值等于0，返回`false`	表示是否进行了自增
`bool atomic_inc_unless_negative(atomic_t *v)`	如果旧值不是负数，那么自增`1`，返回`true`；如果旧值为负数，返回`false`	表示是否进行了自增

减操作

原型	说明	返回值
`void atomic_dec(atomic_t *v)`	自减`1`	无
`int atomic_dec_return(atomic_t *v)`	自减`1`	返回新值
`int atomic_fetch_dec(atomic_t *v)`
`void atomic_sub(int i, atomic_t *v)`	自减`i`	无
`int atomic_sub_return(int i, atomic_t *v)`	自减`i`	返回新值
`int atomic_fetch_sub(int i, atomic_t *v)`	自减`i`	返回旧值

`bool atomic_sub_and_test(int i, atomic_t *v)`	自减`i`	如果新值为0，返回`true`；否则返回`false`
`bool atomic_dec_and_test(atomic_t *v)`	自减`1`	如果新值为0，返回`true`；否则返回`false`

`bool atomic_dec_unless_positive(atomic_t *v)`	如果旧值不是正数(<=0)，那么自减`1`，返回`true`；如果旧值是正数(>0)，返回`false`	表示是否进行了自减操作

交换

原型	说明	返回值
`int atomic_xchg(atomic_t *v, int i)`	将`i`赋值给原子变量	返回旧值

比较交换

原型	说明	返回值
`int atomic_cmpxchg(atomic_t *v, int old, int new)`	如果旧值跟`old`相等，将`new`赋值给原子变量，返回旧值；如果旧值跟`old`不等，返回`new`	返回值为整形
`bool atomic_try_cmpxchg(atomic_t v, int old, int new)`	如果旧值跟`old`相等，将`new`赋值给原子变量，返回`true`；如果旧值跟`old`不等，将`new`赋值给给`*old`，返回`false`	返回的是布尔类型，表示原子变量是否成功赋值

此外，在某些场景下需要可能需要同时比较交换两个指针的值，如果操作成功，返回true。内核提供了下面的宏：

#define cmpxchg_double(ptr, ...) \
({ \
  typeof(ptr) __ai_ptr = (ptr); \
  instrument_atomic_write(__ai_ptr, 2 * sizeof(*__ai_ptr)); \
  arch_cmpxchg_double(__ai_ptr, __VA_ARGS__); \
})

示例：

逻辑操作

原型	说明	返回值
`void atomic_and(int i, atomic_t *v)`	`逻辑与`	无
`int atomic_fetch_and(int i, atomic_t *v)`	`逻辑与`	返回旧值
`void atomic_andnot(int i, atomic_t *v)`	跟`~i`进行`逻辑与`	无
`int atomic_fetch_andnot(int i, atomic_t *v)`	跟`~i`进行`逻辑与`	返回旧值
`void atomic_or(int i, atomic_t *v)`	逻辑或	无
`int atomic_fetch_or(int i, atomic_t *v)`	逻辑或	返回旧值
`void atomic_xor(int i, atomic_t *v)`	异或	无
`int atomic_fetch_xor(int i, atomic_t *v)`	异或	返回旧值

test_and_xxx_bit

原型	说明	返回值
`test_and_set_bit`	检查并置1	如果已经置1，那么返回1。如果没有设置，那么进行置1，返回0
`test_and_clear_bit`	检查并清0	如果已经清0，那么返回0。如果没有清0，那么进行清0，返回1
`test_and_change_bit`	检查并改变	是1返回1，是0返回0，然后取反
`change_bit`
`set_bit`
`clear_bit`

linux内核原子操作学习

参考

说明

数据类型

初始化

赋值操作

读操作

加操作

减操作

交换

比较交换

逻辑操作

test_and_xxx_bit

热门文章

最新文章

相关课程

相关电子书

相关实验场景

推荐镜像