Linux 内核同步之自旋锁与信号量的异同【转】

Linux 设备驱动中必须解决的一个问题是多个进程对共享资源的并发访问，并发访问会导致竞态，linux 提供了多种解决竞态问题的方式，这些方式适合不同的应用场景。

Linux 内核是多进程、多线程的操作系统，它提供了相当完整的内核同步方法。内核同步方法列表如下：

中断屏蔽

原子操作

自旋锁

读写自旋锁

顺序锁

信号量

读写信号量

BKL （大内核锁）

Seq 锁

一、并发与竞态：

定义：

并发（ concurrency ）指的是多个执行单元同时、并行被执行，而并发的执行单元对共享资源（硬件资源和软件上的全局变量、静态变量等）的访问则很容易导致竞态（ race conditions ）。

在 linux 中，主要的竞态发生在如下几种情况：

1、 对称多处理器（ SMP ）多个 CPU

特点是多个 CPU 使用共同的系统总线，因此可访问共同的外设和存储器。

2、 单 CPU 内进程与抢占它的进程

3、 中断（硬中断、软中断、 Tasklet 、底半部）与进程之间

只要并发的多个执行单元存在对共享资源的访问，竞态就有可能发生 。

如果中断处理程序访问进程正在访问的资源，则竞态也会会发生。

多个中断之间本身也可能引起并发而导致竞态（中断被更高优先级的中断打断）。

解决竞态问题的途径 是保证对共享资源的互斥访问，所谓互斥访问就是指一个执行单元在访问共享资源的时候，其他的执行单元都被禁止访问。

访问共享资源的代码区域被称为临界区，临界区需要以某种互斥机制加以保护，中断屏蔽，原子操作，自旋锁，和信号量都是 linux 设备驱动中可采用的互斥途径。

临界区和竞争条件：

所谓临界区（ critical regions ）就是访问和操作共享数据的代码段，为了避免在临界区中并发访问，编程者必须保证这些代码原子地执行——也就是说，代码在执行结束前不可被打断，就如同整个临界区是一个不可分割的指令一样，如果两个执行线程有可能处于同一个临界区中，那么就是程序包含一个 bug ，如果这种情况发生了，我们就称之为竞争条件（ race conditions ），避免并发和防止竞争条件被称为同步。

死锁：

死锁的产生需要一定条件：要有一个或多个执行线程和一个或多个资源，每个线程都在等待其中的一个资源，但所有的资源都已经被占用了，所有线程都在相互等待，但它们永远不会释放已经占有的资源，于是任何线程都无法继续，这便意味着死锁的发生。

二、中断屏蔽

在单 CPU 范围内避免竞态的一种简单方法是在进入临界区之前屏蔽系统的中断。

由于 linux 内核的进程调度等操作都依赖中断来实现，内核抢占进程之间的并发也就得以避免了。

中断屏蔽的使用方法：

local_irq_disable()// 屏蔽中断

// 临界区

local_irq_enable()// 开中断

特点：

由于 linux 系统的异步 IO ，进程调度等很多重要操作都依赖于中断，在屏蔽中断期间所有的中断都无法得到处理，因此长时间的屏蔽是很危险的，有可能造成数据丢失甚至系统崩溃，这就要求在屏蔽中断之后，当前的内核执行路径应当尽快地执行完临界区的代码。

中断屏蔽只能禁止本 CPU 内的中断，因此，并不能解决多 CPU 引发的竞态，所以单独使用中断屏蔽并不是一个值得推荐的避免竞态的方法，它一般和自旋锁配合使用。

三、原子操作

定义： 原子操作指的是在执行过程中不会被别的代码路径所中断的操作。

（原子原本指的是不可分割的微粒，所以原子操作也就是不能够被分割的指令）

（它保证指令以 “ 原子 ” 的方式执行而不能被打断）

原子操作是不可分割的，在执行完毕不会被任何其它任务或事件中断 。在单处理器系统(UniProcessor) 中，能够在单条指令中完成的操作都可以认为是 " 原子操作 " ，因为中断只能发生于指令之间。 这也是某些 CPU 指令系统中引入了 test_and_set 、 test_and_clear 等指令用于临界资源互斥的原因。但是，在对称多处理器 (Symmetric Multi-Processor ) 结构中就不同了，由于系统中有多个处理器在独立地运行，即使能在单条指令中完成的操作也有可能受到干扰。我们以 decl ( 递减指令 ) 为例，这是一个典型的 " 读－改－写 " 过程，涉及两次内存访问。

通俗理解：

      原子操作，顾名思义，就是说像原子一样不可再细分。一个操作是原子操作，意思就是说这个操作是以原子的方式被执行，要一口气执行完，执行过程不能够被 OS 的其他行为打断，是一个整体的过程，在其执行过程中， OS 的其它行为是插不进来的。

分类： linux 内核提供了一系列函数来实现内核中的原子操作，分为整型原子操作和位原子操作，共同点是： 在任何情况下操作都是原子的，内核代码可以安全的调用它们而不被打断。

原子整数操作：

针对整数的原子操作只能对 atomic_t 类型的数据进行处理，在这里之所以引入了一个特殊的数据类型，而没有直接使用 C 语言的 int 型，主要是出于两个原因：

第一、 让原子函数只接受 atomic_t 类型的操作数，可以确保原子操作只与这种特殊类型数据一起使用，同时，这也确保了该类型的数据不会被传递给其它任何非原子函数；

第二、 使用 atomic_t 类型确保编译器不对相应的值进行访问优化 —— 这点使得原子操作最终接收到正确的内存地址，而不是一个别名，最后就是在不同体系结构上实现原子操作的时候，使用 atomic_t 可以屏蔽其间的差异。

原子整数操作最常见的用途就是实现计数器。

另一点需要说明原子操作只能保证操作是原子的，要么完成，要么不完成，不会有操作一半的可能，但原子操作并不能保证操作的顺序性，即它不能保证两个操作是按某个顺序完成的。如果要保证原子操作的顺序性，请使用内存屏障指令。

atomic_t 和 ATOMIC_INIT(i) 定义

typedef struct { volatile int counter; } atomic_t;

#define ATOMIC_INIT(i)  { (i) }

在你编写代码的时候，能使用原子操作的时候，就尽量不要使用复杂的加锁机制，对多数体系结构来讲，原子操作与更复杂的同步方法相比较，给系统带来的开销小，对高速缓存行的影响也小，但是，对于那些有高性能要求的代码，对多种同步方法进行测试比较，不失为一种明智的作法。

原子位操作：

针对位这一级数据进行操作的函数，是对普通的内存地址进行操作的。它的参数是一个指针和一个位号。

为方便其间，内核还提供了一组与上述操作对应的非原子位函数，非原子位函数与原子位函数的操作完全相同，但是，前者不保证原子性，且其名字前缀多两个下划线。例如，与 test_bit() 对应的非原子形式是 _test_bit() ，如果你不需要原子性操作（比如，如果你已经用锁保护了自己的数据），那么这些非原子的位函数相比原子的位函数可能会执行得更快些。

四、自旋锁

自旋锁的引入：

如 果每个临界区都能像增加变量这样简单就好了，可惜现实不是这样，而是临界区可以跨越多个函数，例如：先得从一个数据结果中移出数据，对其进行格式转换和解 析，最后再把它加入到另一个数据结构中，整个执行过程必须是原子的，在数据被更新完毕之前，不能有其他代码读取这些数据，显然，简单的原子操作是无能为力 的（在单处理器系统 (UniProcessor) 中，能够在单条指令中完成的操作都可以认为是 " 原子操作 " ，因为中断只能发生于指令之间） ，这就需要使用更为复杂的同步方法 —— 锁来提供保护。

自旋锁的介绍：

Linux 内核中最常见的锁是自旋锁（ spin lock ），自旋锁最多只能被一个可执行线程持有， 如果一个执行线程试图获得一个被争用（已经被持有）的自旋锁，那么该线程就会一直进行忙循环 —旋转 — 等待锁重新可用，要是锁未被争用，请求锁的执行线程便能立刻得到它，继续执行，在任意时间，自旋锁都可以防止多于一个的执行线程同时进入理解区，注意同一个锁可以用在多个位置— 例如，对于给定数据的所有访问都可以得到保护和同步。

一个被争用的自旋锁使得请求它的线程在等待锁重新可用时自旋（特别浪费处理器时间），所以自旋锁不应该被长时间持有，事实上，这点正是使用自旋锁的初衷，在短期间内进行轻量级加锁，还可以采取另外的方式来处理对锁的争用：让请求线程睡眠 ，直到锁重新可用时再唤醒它 ，这样处理器就不必循环等待，可以去执行其他代码，这也会带来一定的开销 —— 这里有两次明显的上下文切换 ， 被阻塞的线程要换出和换入。因此，持有自旋锁的时间最好小于完成两次上下文切换的耗时，当然我们大多数人不会无聊到去测量上下文切换的耗时，所以我们让持 有自旋锁的时间应尽可能的短就可以了，信号量可以提供上述第二种机制，它使得在发生争用时，等待的线程能投入睡眠，而不是旋转。

自旋锁可以使用在中断处理程序中（ 此处不能使用信号量，因为它们会导致睡眠），在中断处理程序中使用自旋锁时，一定要在获取锁之前，首先禁止本地中断 （在 当前处理器上的中断请求），否则，中断处理程序就会打断正持有锁的内核代码，有可能会试图去争用这个已经持有的自旋锁，这样以来，中断处理程序就会自旋， 等待该锁重新可用，但是锁的持有者在这个中断处理程序执行完毕前不可能运行，这正是我们在前一章节中提到的双重请求死锁，注意，需要关闭的只是当前处理器上的中断，如果中断发生在不同的处理器上，即使中断处理程序在同一锁上自旋，也不会妨碍锁的持有者（在不同处理器上）最终释放锁。

自旋锁的简单理解：

理解自旋锁最简单的方法是把它作为一个变量看待，该变量把一个临界区或者标记为 “ 我当前正在运行，请稍等一会 ” 或者标记为 “ 我当前不在运行，可以被使用 ” 。如果 A 执行单元首先进入例程，它将持有自旋锁，当 B 执行单元试图进入同一个例程时，将获知自旋锁已被持有，需等到 A执行单元释放后才能进入。

自旋锁的 API 函数：

其实介绍的几种信号量和互斥机制，其底层源码都是使用自旋锁 , 可以理解为自旋锁的再包装。 所以从这里就可以理解为什么自旋锁通常可以提供比信号量更高的性能。 
自旋锁是一个互斥设备，他只能会两个值： “ 锁定 ” 和 “ 解锁 ” 。它通常实现为某个整数之中的单个位。 
“ 测试并设置 ” 的操作必须以原子方式完成。 
任何时候，只要内核代码拥有自旋锁，在相关 CPU 上的抢占就会被禁止。

适用于自旋锁的核心规则： 
（ 1 ）任何拥有自旋锁的代码都必须使原子的，除服务中断外（某些情况下也不能放弃 CPU, 如中断服务也要获得自旋锁。为了避免这种锁陷阱，需要在拥有自旋锁时禁止中断），不能放弃 CPU（如休眠，休眠可发生在许多无法预期的地方）。否则 CPU 将有可能永远自旋下去（死机）。 
（ 2 ）拥有自旋锁的时间越短越好。

需 要强调的是，自旋锁别设计用于多处理器的同步机制，对于单处理器（对于单处理器并且不可抢占的内核来说，自旋锁什么也不作），内核在编译时不会引入自旋锁 机制，对于可抢占的内核，它仅仅被用于设置内核的抢占机制是否开启的一个开关，也就是说加锁和解锁实际变成了禁止或开启内核抢占功能。如果内核不支持抢 占，那么自旋锁根本就不会编译到内核中。 
内核中使用 spinlock_t 类型来表示自旋锁，它定义在 <linux/spinlock_types.h> ：

对于不支持 SMP 的内核来说， struct raw_spinlock_t 什么也没有，是一个空结构。对于支持多处理器的内核来说， struct raw_spinlock_t 定义为

slock 表示了自旋锁的状态， “1” 表示自旋锁处于解锁状态（ UNLOCK ）， “0” 表示自旋锁处于上锁状态（ LOCKED ）。 
break_lock 表示当前是否由进程在等待自旋锁，显然，它只有在支持抢占的 SMP 内核上才起作用。 
    自旋锁的实现是一个复杂的过程，说它复杂不是因为需要多少代码或逻辑来实现它，其实它的实现代码很少。自旋锁的实现跟体系结构关系密切，核心代码基本也是由汇编语言写成，与体协结构相关的核心代码都放在相关的 <asm/> 目录下，比如 <asm/spinlock.h> 。对于我们驱动程序开发人员来说，我们没有必要了解这么 spinlock 的内部细节，如果你对它感兴趣，请参考阅读 Linux 内核源代码。对于我们驱动的 spinlock 接口，我们只需包括 <linux/spinlock.h> 头文件。在我们详细的介绍spinlock 的 API 之前，我们先来看看自旋锁的一个基本使用格式：

从使用上来说， spinlock 的 API 还很简单的，一般我们会用的的 API 如下表，其实它们都是定义在<linux/spinlock.h> 中的宏接口，真正的实现在 <asm/spinlock.h> 中

·         初始化

spinlock 有两种初始化形式，一种是静态初始化，一种是动态初始化。对于静态的 spinlock 对象，我们用 SPIN_LOCK_UNLOCKED 来初始化，它是一个宏。当然，我们也可以把声明 spinlock 和初始化它放在一起做，这就是 DEFINE_SPINLOCK 宏的工作，因此，下面的两行代码是等价的。

spin_lock_init 函数一般用来初始化动态创建的 spinlock_t 对象，它的参数是一个指向 spinlock_t 对象的指针。当然，它也可以初始化一个静态的没有初始化的 spinlock_t 对象。

·         获取锁

内核提供了三个函数用于获取一个自旋锁。 
spin_lock ：获取指定的自旋锁。 
spin_lock_irq ：禁止本地中断并获取自旋锁。 
spin_lock_irqsace ：保存本地中断状态，禁止本地中断并获取自旋锁，返回本地中断状态。 

自旋锁是可以使用在中断处理程序中的，这时需要使用具有关闭本地中断功能的函数，我们推荐使用 spin_lock_irqsave ，因为它会保存加锁前的中断标志，这样就会正确恢复解锁时的中断标志。如果spin_lock_irq 在加锁时中断是关闭的，那么在解锁时就会错误的开启中断。 

另外两个同自旋锁获取相关的函数是： 
spin_trylock() ：尝试获取自旋锁，如果获取失败则立即返回非 0 值，否则返回 0 。 
spin_is_locked() ：判断指定的自旋锁是否已经被获取了。如果是则返回非 0 ，否则，返回 0 。

·         释放锁

同获取锁相对应，内核提供了三个相对的函数来释放自旋锁。 
spin_unlock ：释放指定的自旋锁。 
spin_unlock_irq ：释放自旋锁并激活本地中断。 
spin_unlock_irqsave ：释放自旋锁，并恢复保存的本地中断状态。