原子性操作原理分析
1. 概念
原子操作是指不被打断的操作，即它是最小的执行单位。最简单的原子操作就是一条条的汇编指令(不包括一些伪指令，伪指令会被汇编器解释成多条汇编指令)。在 linux 中原子操作对应的数据结构为 atomic_t，定义如下：
typedef struct {int counter;
} atomic_t;
本质上就是一个整型变量，之所以定义这么一个数据类型，是为了让原子操作函数只接受 atomic_t 类型的操作数，如果传入的不是 atomic_t 类型数据，在程序编译阶段就不会通过；另一个原因就是确保编译器不会对相应的值进行访问优化，确保对它的访问都是对内存的访问，而不是对寄存器的访问。
2. 赋值操作
ARM 处理器有直接对内存地址进行赋值的指令(STR)。
#define atomic_set(v,i) (((v)->counter) = (i))3. 读操作
用 volatile 来防止编译器对变量访问的优化，确保是对内存的访问，而不是对寄存器的访问。
#define atomic_read(v) (*(volatile int *)&(v)->counter)4. 加操作
使用独占指令完成累加操作。
static inline void atomic_add(int i, atomic_t *v){
unsigned long tmp;
int result;
// 使用独占指令读取，然后执行加操作，独占写失败时就重新执行
__asm__ __volatile__("@ atomic_add\n"
"1: ldrex %0, [%3]\n"
" add %0, %0, %4\n"
" strex %1, %0, [%3]\n"
" teq %1, #0\n"
" bne 1b"
: "=&r" (result), "=&r" (tmp), "+Qo" (v->counter)
: "r" (&v->counter), "Ir" (i)
: "cc");
}
5. 减操作
对比加操作和减操作的代码可以看出，它们非常的相似，其实不同的地方就一句，所以现在最新的内核源码中已经使用宏定义 ATOMIC_OP(op, c_op, asm_op) 来重写了这部分代码。
static inline void atomic_sub(int i, atomic_t *v){
unsigned long tmp;
int result;
// 使用独占指令读取，然后执行减操作，独占写失败时就重新执行
__asm__ __volatile__("@ atomic_sub\n"
"1: ldrex %0, [%3]\n"
" sub %0, %0, %4\n"
" strex %1, %0, [%3]\n"
" teq %1, #0\n"
" bne 1b"
: "=&r" (result), "=&r" (tmp), "+Qo" (v->counter)
: "r" (&v->counter), "Ir" (i)
: "cc");
}
6. 其他操作
类似的原子操作函数还有一些，比如 atomic_XXX_return、atomic_cmpxchg、atomic_clear_mask，以及在此基础上实现的 atomic_inc、atomic_dec、atomic_XXX_and_test、atomic_XXX_return等。以上代码都是针对 SMP 处理器的实现方式，针对非 SMP 处理器，由于不存在其他核心的抢占，所以只需要防止其他进程抢占即可实现原子操作，例如加操作：
static inline int atomic_sub_return(int i, atomic_t *v){
unsigned long flags;
int val;
// 通过关闭中断防止其他进程打断代码的执行
raw_local_irq_save(flags);
val = v->counter;
v->counter = val -= i;
// 恢复中断原始的状态
raw_local_irq_restore(flags);
return val;
}
7. 总结
原子性操作的实现需要具体体系结构相关的指令集的支持。
以上是 原子性操作原理分析 的全部内容， 来源链接： utcz.com/a/60743.html
