Linux内核里的同步机制原子操作.docVIP

Linux内核里的同步机制（1） -- 原子操作 原子操作 – 单一的不可中断的操作。 首先看一个counter++的例子 Sch[94] 。 在汇编层次，这一条语句会变成3个指令， 1. 将counter值从内存中拷贝到CPU寄存器R0。 2. 寄存器R0中的值加1。 3. 将寄存器的值拷回到内存。 如果此时有两个CPU同时执行counter++，那么假设counter初始值为0，两次加1所得出来的结果就不是2，而是1，Bug由此而生。 这是一个SMP的例子。类似的，对于UP来说，如果两个任务“同时”执行counter++，那么同样会出现此问题。“同时”上面加了引号，那是因为UP永远不会真的同时，不过如果在1- 2或者2- 3之间切换到了另一个任务或者被中断，而此任务或中断例程也执行counter++，就有可能产生这个BUG。 如何避免上面这个问题？首先我们必须保证了1- 2- 3是一个单一步骤，不被“打断”（这里的打断只是逻辑意义上的，而不是特指我们通常说的中断，后文可以看到在ARMV6以上的平台，原子操作中间也是可以发生中断的）。而对于SMP，还需要保证对临界内存的访问是互斥的。 Linux内核为原子操作提供了类型atomic_t和一组内核API，使用起来非常简单。原子版的counter++如下： atomic_t counter ATOMIC_INIT 0 ; atomic_inc（counter ; atomic_t相关的原子操作由平台代码负责实现，不同的处理器会采用不同的方式，目的都是为了上面说的两个保证。 那Linux内核如何保证这组API的原子性呢？ 1. 保证不能被打断。 从字面上理解，就是要禁止上下文切换，禁止中断发生。 这个很容易实现，只要保证在这个原子操作的时候不发生中断就可以了。所以通过关本地中断就可以实现。简单粗暴而又有效。 2. 保证对共享内存的访问是互斥的。 这一点和具体平台紧密相连。一般来说，可以通过Bus锁或者其他的对内存访问的互斥机制实现。Linux理应交给平台代码去实现。 以ARM为例，分析一下代码。以下代码均来自1。 在代码中，发现共有两种实现，分别为ARMV6之前，ARMV6之后。 在ARMV6之前的实现如下， #ifdef CONFIG_SMP #error SMP not supported on pre-ARMv6 CPUs #endif static inline int atomic_add_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; #define atomic_add i, v void atomic_add_return i, v ARMV6之前不支持SMP，所以只要在具体实现里关中断就可以了。轻松实现两个保证。 再看ARMV6以后， /* * ARMv6 UP and SMP safe atomic ops. We use load exclusive and * store exclusive to ensure that these are atomic. We may loop * to ensure that the update happens. */ 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 ; 这里出现了几个问题，首先，在实现上没有区分UP和SMP。其次，也没有关中断及加总线锁的动作。那么这个函数如何保证原子性呢？上面有段注释很有价值。说是采用了load exclusive和store exclusive确保其原子性。读一下汇编发现最后一行的意思是当strex的返回值不为0时 %1,即tmp ，会跳转到1重新执行一遍。所以最后一行注释说“We may loop to ensure that the update happens.”。那么strex又是什么？