一、原子操作核心概念
原子操作,指的是不可被中断、执行过程不可分割的最小操作单元,要么完整执行成功,要么完全不执行,执行中途不会被系统调度、中断或其他CPU并发访问打断。
在嵌入式Linux内核中,原子操作是解决多核并发、中断上下文与进程上下文竞态问题的最轻量化手段,广泛用于引用计数、位操作、标志位管理、轻量级锁等场景,相比自旋锁、信号量,它的开销极低,无睡眠/调度风险,可安全用于中断、软中断等原子上下文。
核心实现原理
- 单核系统:通过关闭本地中断,保证操作执行期间不被中断/调度打断,实现原子性。
- 多核SMP系统:依赖CPU架构的原子指令集(如ARM的LDREX/STREX独占指令、x86的LOCK前缀指令),配合内存屏障,保证多CPU核心间的操作原子性和内存可见性。
- 嵌入式Linux内核对不同架构做了统一封装,上层API完全架构无关,可跨平台使用。
二、核心数据类型
内核提供了两种标准原子数据类型,定义在 <linux/types.h> 和 <asm/atomic.h> 中,嵌入式开发必须使用内核封装的标准类型,禁止直接操作原始变量。
数据类型
atomic_t 封装32位有符号整数,结构体仅含一个 counter 成员,绝大多数32位嵌入式平台,通用计数场景
atomic64_t 封装64位有符号整数,64位架构平台,或需要大数值范围的计数场景(如64位嵌入式CPU)
使用规范:
1. 必须通过内核提供的API操作,严禁直接读写 counter 成员;
2. 原子变量必须正确初始化,禁止未初始化直接使用;
3. 原子操作API仅支持对应类型的变量,不可混用。
三、32位原子操作核心API(atomic_t)
头文件: <linux/atomic.h> ,所有API均为内核态专用,用户态不可用。
1. 初始化操作
// 静态初始化,定义时直接初始化
#define ATOMIC_INIT(i) { (i) }
// 示例:定义并初始化原子变量为0
atomic_t g_ref_cnt = ATOMIC_INIT(0);
// 运行时动态赋值,设置原子变量的值为i
void atomic_set(atomic_t *v, int i);
// 示例:运行时设置为10
atomic_set(&g_ref_cnt, 10);
2. 读取操作
// 读取原子变量的当前值,返回int类型
int atomic_read(const atomic_t *v);
// 示例:读取计数值
int cur_cnt = atomic_read(&g_ref_cnt);
3. 加减算术操作(最常用)
// 原子加:v->counter += i
void atomic_add(int i, atomic_t *v);
// 原子减:v->counter -= i
void atomic_sub(int i, atomic_t *v);
// 示例:引用计数+1,引用计数-1
atomic_add(1, &g_ref_cnt);
atomic_sub(1, &g_ref_cnt);
4. 自增/自减操作(计数场景专用)
// 原子自增1:v->counter += 1,等价于atomic_add(1, v)
void atomic_inc(atomic_t *v);
// 原子自减1:v->counter -= 1,等价于atomic_sub(1, v)
void atomic_dec(atomic_t *v);
// 示例:最常用的引用计数操作
atomic_inc(&g_ref_cnt);
atomic_dec(&g_ref_cnt);
5. 操作+结果返回(带返回值,用于条件判断)
这是嵌入式开发中高频使用的API,用于判断操作后是否满足条件,避免竞态。
// 原子减i,返回操作后的值是否为0,为0返回true,非0返回false
int atomic_sub_and_test(int i, atomic_t *v);
// 原子自减1,返回操作后的值是否为0,为0返回true,非0返回false
int atomic_dec_and_test(atomic_t *v);
// 原子自增1,返回操作后的值是否为0,为0返回true,非0返回false
int atomic_inc_and_test(atomic_t *v);
// 示例:资源释放场景,仅当计数减到0时才释放资源
if (atomic_dec_and_test(&g_ref_cnt)) {
// 最后一个引用,执行资源释放
kfree(res);
}
6. 操作+旧值返回(fetch系列,内核2.6.34+支持)
返回操作之前的旧值,适合需要记录操作前状态的场景。
// 原子加i,返回操作前的旧值
int atomic_fetch_add(int i, atomic_t *v);
// 原子减i,返回操作前的旧值
int atomic_fetch_sub(int i, atomic_t *v);
// 原子自增1,返回操作前的旧值
#define atomic_fetch_inc(v)
atomic_fetch_add(1, v)
// 原子自减1,返回操作前的旧值
#define atomic_fetch_dec(v)
atomic_fetch_sub(1, v)
7. 比较并交换(CAS,核心同步原语)
// 比较原子变量v的值是否等于oldval,相等则设置为newval,返回原值
int atomic_cmpxchg(atomic_t *v, int oldval, int newval);
// 示例:仅当当前值为0时,才设置为1,实现无锁单例
int old = atomic_cmpxchg(&g_ref_cnt, 0, 1);
if (old == 0) {
// 交换成功,执行初始化操作
} else {
// 已有其他CPU/上下文完成初始化
}
8. 交换操作
// 原子设置v的值为newval,返回设置前的旧值
int atomic_xchg(atomic_t *v, int newval);
四、64位原子操作API(atomic64_t)
API与32位完全对齐,仅函数名前缀改为 atomic64_ ,参数类型改为 atomic64_t * ,返回值改为 long long ,适配64位嵌入式平台。
核心高频API如下:
// 初始化
#define ATOMIC64_INIT(i) { (i) }
// 赋值/读取
void atomic64_set(atomic64_t *v, long long i);
long long atomic64_read(const atomic64_t *v);
// 加减/自增自减
void atomic64_add(long long i, atomic64_t *v);
void atomic64_sub(long long i, atomic64_t *v);
void atomic64_inc(atomic64_t *v);
void atomic64_dec(atomic64_t *v);
// 带返回值的条件操作
int atomic64_sub_and_test(long long i, atomic64_t *v);
int atomic64_dec_and_test(atomic64_t *v);
int atomic64_inc_and_test(atomic64_t *v);
// CAS与交换
long long atomic64_cmpxchg(atomic64_t *v, long long oldval, long long newval);
long long atomic64_xchg(atomic64_t *v, long long newval);
// fetch系列
long long atomic64_fetch_add(long long i, atomic64_t *v);
long long atomic64_fetch_sub(long long i, atomic64_t *v);
五、原子位操作API
除了整数原子操作,内核还提供了原子位操作,专门用于单个bit位的原子读写、置位、清零、翻转,是嵌入式驱动中标志位、寄存器位操作、位图管理的核心手段。
头文件: <linux/bitops.h> ,操作对象是内存地址,以 unsigned long * 为基地址,位号从0开始计数。
1. 核心位操作API
// 原子置位:将addr地址的第nr位设置为1
void set_bit(int nr, volatile unsigned long *addr);
// 原子清零:将addr地址的第nr位设置为0
void clear_bit(int nr, volatile unsigned long *addr);
// 原子翻转:将addr地址的第nr位取反
void change_bit(int nr, volatile unsigned long *addr);
2. 带返回值的位操作
// 原子置位,返回操作前该位的值(0或1)
int test_and_set_bit(int nr, volatile unsigned long *addr);
// 原子清零,返回操作前该位的值(0或1)
int test_and_clear_bit(int nr, volatile unsigned long *addr);
// 原子翻转,返回操作前该位的值(0或1)
int test_and_change_bit(int nr, volatile unsigned long *addr);
// 仅读取该位的当前值,不修改,原子性读取
int test_bit(int nr, const volatile unsigned long *addr);
嵌入式驱动典型示例
// 定义一个标志位变量,bit0表示设备是否忙
volatile unsigned long dev_flags = 0;
#define DEV_BUSY_BIT 0
// 原子尝试占用设备,返回是否占用成功
if (!test_and_set_bit(DEV_BUSY_BIT, &dev_flags)) {
// 置位成功,之前是0,设备空闲,成功占用
do_dev_work();
// 操作完成,释放设备
clear_bit(DEV_BUSY_BIT, &dev_flags);
} else {
// 设备忙,占用失败
return -EBUSY;
}
六、嵌入式开发使用注意事项与最佳实践
1. 禁止用原子操作替代锁:原子操作仅适用于单变量/单位的轻量级同步,复杂临界区(如多变量、结构体、硬件寄存器序列操作)必须使用自旋锁、互斥锁,不可强行用原子操作实现,极易引发隐蔽竞态。
2. 内存屏障与可见性:原子操作API内部自带编译器屏障,多数架构自带CPU内存屏障,但若原子变量与普通变量配合使用,需配合 smp_mb() 等内存屏障保证多核心间的内存顺序。
3. 中断上下文安全:所有原子操作API均无睡眠、无调度,可安全用于中断处理函数、软中断、tasklet等原子上下文,这是它相比信号量、互斥锁的核心优势。
4. 数据类型严格匹配:32位 atomic_t 不可传入64位API,反之亦然;位操作的位号不可超出目标变量的位宽范围,避免内存越界。
5. 初始化规范:静态原子变量必须用 ATOMIC_INIT 初始化,动态分配的原子变量,必须在使用前调用 atomic_set 初始化,避免随机值引发异常。
6. 嵌入式平台适配:部分低端MCU架构(如无MMU的ARM Cortex-M)的Linux内核,原子操作实现依赖关闭中断,需注意操作耗时不可过长,避免中断响应延迟超标。
总结不容易,大家关注一下我,谢谢您!