Linux内核的同步机制

本文周详的介绍了Linux内核中的同步机制：原子操作、信号量、读写信号量和自旋锁的API，使用需求及一些典型示例 

　　一、引言 

　　在现代操作系统里，同一时间可能有多个内核执行流在执行，因此内核其实象多进程多线程编程相同也需要一些同步机制来同步各执行单元对共享数据的访问。尤其是在多处理器系统上，更需要一些同步机制来同步不同处理器上的执行单元对共享的数据的访问。 
　 　在主流的Linux内核中包含了几乎所有现代的操作系统具有的同步机制，这些同步机制包括：原子操作、信号量（semaphore）、读写信号量 （rw_semaphore）、spinlock、BKL(Big Kernel Lock)、rwlock、brlock（只包含在2.4内核中）、RCU（只包含在2.6内核中）和seqlock（只包含在2.6内核中）。 

　　二、原子操作 

　　所谓原子操作，就是该操作绝不会在执行完毕前被所有其他任务或事件打断，也就说，他的最小的执行单位，不可能有比他更小的执行单位，因此这里的原子实际是使用了物理学里的物质微粒的概念。 
　　原子操作需要硬件的支持，因此是架构相关的，其API和原子类型的定义都定义在内核源码树的include/asm/atomic.h文件中，他们都使用汇编语言实现，因为C语言并不能实现这样的操作。 
　　原子操作主要用于实现资源计数，非常多引用计数(refcnt)就是通过原子操作实现的。原子类型定义如下： 

typedef struct 
{ 
volatile int counter; 
} 
atomic_t; 

　　volatile修饰字段告诉gcc不要对该类型的数据做优化处理，对他的访问都是对内存的访问，而不是对寄存器的访问。 
　　原子操作API包括： 

atomic_read(atomic_t * v); 
　　该函数对原子类型的变量进行原子读操作，他返回原子类型的变量v的值。 

atomic_set(atomic_t * v, int i); 
　　该函数设置原子类型的变量v的值为i。 

void atomic_add(int i, atomic_t *v); 
　　该函数给原子类型的变量v增加值i。 

atomic_sub(int i, atomic_t *v); 
　　该函数从原子类型的变量v中减去i。 

int atomic_sub_and_test(int i, atomic_t *v); 
　　该函数从原子类型的变量v中减去i，并判断结果是否为0，如果为0，返回真，否则返回假。 

void atomic_inc(atomic_t *v); 
　　该函数对原子类型变量v原子地增加1。 

void atomic_dec(atomic_t *v); 
　　该函数对原子类型的变量v原子地减1。 

int atomic_dec_and_test(atomic_t *v); 
　　该函数对原子类型的变量v原子地减1，并判断结果是否为0，如果为0，返回真，否则返回假。 

int atomic_inc_and_test(atomic_t *v); 
　　该函数对原子类型的变量v原子地增加1，并判断结果是否为0，如果为0，返回真，否则返回假。 

int atomic_add_negative(int i, atomic_t *v); 
　　该函数对原子类型的变量v原子地增加I，并判断结果是否为负数，如果是，返回真，否则返回假。 

int atomic_add_return(int i, atomic_t *v); 
　　该函数对原子类型的变量v原子地增加i，并且返回指向v的指针。 


int atomic_sub_return(int i, atomic_t *v); 
　　该函数从原子类型的变量v中减去i，并且返回指向v的指针。 

int atomic_inc_return(atomic_t * v); 
　　该函数对原子类型的变量v原子地增加1并且返回指向v的指针。 

int atomic_dec_return(atomic_t * v); 
　　该函数对原子类型的变量v原子地减1并且返回指向v的指针。 


　 　原子操作通常用于实现资源的引用计数，在TCP/IP协议栈的IP碎片处理中，就使用了引用计数，碎片队列结构struct ipq描述了一个IP碎片，字段refcnt就是引用计数器，他的类型为atomic_t，当创建IP碎片时（在函数ip_frag_create中）， 使用atomic_set函数把他设置为1，当引用该IP碎片时，就使用函数atomic_inc把引用计数加1。 
　　当不必引用该 IP碎片时，就使用函数ipq_put来释放该IP碎片，ipq_put使用函数atomic_dec_and_test把引用计数减1并判断引用计数是 否为0，如果是就释放IP碎片。函数ipq_kill把IP碎片从ipq队列中删除，并把该删除的IP碎片的引用计数减1（通过使用函数 atomic_dec实现）。 

　　三、信号量（semaphore） 

　　Linux内核的信号量在概念和原理上和用户态的System V的IPC机制信号量是相同的，不过他绝不可能在内核之外使用，因此他和System V的IPC机制信号量毫不相干。 
　　信号量在创建时需要设置一个初始值，表示同时能有几个任务能访问该信号量保护的共享资源，初始值为1就变成互斥锁（Mutex），即同时只能有一个任务能访问信号量保护的共享资源。 
　 　一个任务要想访问共享资源，首先必须得到信号量，获取信号量的操作将把信号量的值减1，若当前信号量的值为负数，表明无法获得信号量，该任务必须挂起在 该信号量的等待队列等待该信号量可用；若当前信号量的值为非负数，表示能获得信号量，因而能即时访问被该信号量保护的共享资源。 
　　当任务访问完被信号量保护的共享资源后，必须释放信号量，释放信号量通过把信号量的值加1实现，如果信号量的值为非正数，表明有任务等待当前信号量，因此他也唤醒所有等待该信号量的任务。 

　　信号量的API有： 


DECLARE_MUTEX(name) 
　　该宏声明一个信号量name并初始化他的值为1，即声明一个互斥锁。 

DECLARE_MUTEX_LOCKED(name) 
　　该宏声明一个互斥锁name，但把他的初始值设置为0，即锁在创建时就处在已锁状态。因此对于这种锁，一般是先释放后获得。 

void sema_init (struct semaphore *sem, int val); 
　　该函用于数初始化设置信号量的初值，他设置信号量sem的值为val。 

void init_MUTEX (struct semaphore *sem); 
　　该函数用于初始化一个互斥锁，即他把信号量sem的值设置为1。 

void init_MUTEX_LOCKED (struct semaphore *sem); 
　　该函数也用于初始化一个互斥锁，但他把信号量sem的值设置为0，即一开始就处在已锁状态。 


void down(struct semaphore * sem); 
　　该函数用于获得信号量sem，他会导致睡眠，因此不能在中断上下文（包括IRQ上下文和softirq上下文）使用该函数。该函数将把sem的值减1，如果信号量sem的值非负，就直接返回，否则调用者将被挂起，直到别的任务释放该信号量才能继续运行。 

int down_interruptible(struct semaphore * sem); 
　　该函数功能和down类似，不同之处为，down不会被信号（signal）打断，但down_interruptible能被信号打断，因此该函数有返回值来区分是正常返回还是被信号中断，如果返回0，表示获得信号量正常返回，如果被信号打断，返回-EINTR。 

int down_trylock(struct semaphore * sem); 
　　该函数试着获得信号量sem，如果能够即时获得，他就获得该信号量并返回0，否则，表示不能获得信号量sem，返回值为非0值。因此，他不会导致调用者睡眠，能在中断上下文使用。

int down_killable(struct semaphore *sem);

int down_timeout(struct semaphore *sem, long jiffies);

int down_timeout_interruptible(struct semaphore *sem, long jiffies);

void up(struct semaphore * sem); 
　　该函数释放信号量sem，即把sem的值加1，如果sem的值为非正数，表明有任务等待该信号量，因此唤醒这些等待者。

　　信号量在绝大部分情况下作为互斥锁使用，下面以console驱动系统为例说明信号量的使用。 
　　在内核源码树的kernel/printk.c中，使用宏DECLARE_MUTEX声明了一个互斥锁console_sem，他用于保护console驱动列表console_drivers及同步对整个console驱动系统的访问。 
　 　其中定义了函数acquire_console_sem来获得互斥锁console_sem，定义了release_console_sem来释放互斥 锁console_sem，定义了函数try_acquire_console_sem来尽力得到互斥锁console_sem。这三个函数实际上是分别 对函数down，up和down_trylock的简单包装。 
　　需要访问console_drivers驱动列表时就需要使用acquire_console_sem来保护console_drivers列表，当访问完该列表后，就调用release_console_sem释放信号量console_sem。 
　 　函数 console_unblank，console_device，console_stop，console_start，register_console 和unregister_console都需要访问console_drivers，因此他们都使用函数对acquire_console_sem和 release_console_sem来对console_drivers进行保护。

　　四、读写信号量（rw_semaphore） 
　 　读写信号量对访问者进行了细分，或为读者，或为写者，读者在保持读写信号量期间只能对该读写信号量保护的共享资源进行读访问，如果一个任务除了需要读， 可能还需要写，那么他必须被归类为写者，他在对共享资源访问之前必须先获得写者身份，写者在发现自己不必写访问的情况下能降级为读者。读写信号量同时拥有 的读者数不受限制，也就说能有任意多个读者同时拥有一个读写信号量。 
　　如果一个读写信号量当前没有被写者拥有并且也没有写者等待读 者释放信号量，那么所有读者都能成功获得该读写信号量；否则，读者必须被挂起直到写者释放该信号量。如果一个读写信号量当前没有被读者或写者拥有并且也没 有写者等待该信号量，那么一个写者能成功获得该读写信号量，否则写者将被挂起，直到没有所有访问者。因此，写者是排他性的，独占性的。 
　 　读写信号量有两种实现，一种是通用的，不依赖于硬件架构，因此，增加新的架构不必重新实现他，但缺点是性能低，获得和释放读写信号量的开销大；另一种是 架构相关的，因此性能高，获取和释放读写信号量的开销小，但增加新的架构需要重新实现。在内核设置时，能通过选项去控制使用哪一种实现。 

　　读写信号量的相关API有： 


DECLARE_RWSEM(name) 
　　该宏声明一个读写信号量name并对其进行初始化。 

void init_rwsem(struct rw_semaphore *sem); 
　　该函数对读写信号量sem进行初始化。 

void down_read(struct rw_semaphore *sem); 
　　读者调用该函数来得到读写信号量sem。该函数会导致调用者睡眠，因此只能在进程上下文使用。 

int down_read_trylock(struct rw_semaphore *sem); 
　　该函数类似于down_read，只是他不会导致调用者睡眠。他尽力得到读写信号量sem，如果能够即时得到，他就得到该读写信号量，并且返回1，否则表示不能即时得到该信号量，返回0。因此，他也能在中断上下文使用。 


void down_write(struct rw_semaphore *sem); 
　　写者使用该函数来得到读写信号量sem，他也会导致调用者睡眠，因此只能在进程上下文使用。 


int down_write_trylock(struct rw_semaphore *sem); 
　　该函数类似于down_write，只是他不会导致调用者睡眠。该函数尽力得到读写信号量，如果能够即时获得，就获得该读写信号量并且返回1，否则表示无法即时获得，返回0。他能在中断上下文使用。 

void up_read(struct rw_semaphore *sem); 
　　读者使用该函数释放读写信号量sem。他和down_read或down_read_trylock配对使用。如果down_read_trylock返回0，不必调用up_read来释放读写信号量，因为根本就没有获得信号量。 

void up_write(struct rw_semaphore *sem); 
　　写者调用该函数释放信号量sem。他和down_write或down_write_trylock配对使用。如果down_write_trylock返回0，不必调用up_write，因为返回0表示没有获得该读写信号量。 

void downgrade_write(struct rw_semaphore *sem); 
　 　该函数用于把写者降级为读者，这有时是必要的。因为写者是排他性的，因此在写者保持读写信号量期间，所有读者或写者都将无法访问该读写信号量保护的共享 资源，对于那些当前条件下不必写访问的写者，降级为读者将，使得等待访问的读者能够即时访问，从而增加了并发性，提高了效率。

　　读写信号量适于在读多写少的情况下使用，在linux内核中对进程的内存映像描述结构的访问就使用了读写信号量进行保护。 
　 　在Linux中，每一个进程都用一个类型为task_t或struct task_struct的结构来描述，该结构的类型为struct mm_struct的字段mm描述了进程的内存映像，特别是mm_struct结构的mmap字段维护了整个进程的内存块列表，该列表将在进程生存期间被 大量地遍利或修改。 
　　因此mm_struct结构就有一个字段mmap_sem来对mmap的访问进行保护，mmap_sem就是 个读写信号量，在proc文件系统里有非常多进程内存使用情况的接口，通过他们能够查看某一进程的内存使用情况，命令free、ps和top都是通过 proc来得到内存使用信息的，proc接口就使用down_read和up_read来读取进程的mmap信息。 
　　当进程动态地 分配或释放内存时，需要修改mmap来反映分配或释放后的内存映像，因此动态内存分配或释放操作需要以写者身份获得读写信号量mmap_sem来对 mmap进行更新。系统调用brk和munmap就使用了down_write和up_write来保护对mmap的访问。 

　　五、自旋锁（spinlock） 

　　自旋锁和互斥锁有点类似，只是自旋锁不会引起调用者睡眠，如果自旋锁已被别的执行单元保持，调用者就一直循环在那里看是否该自旋锁的保持者已释放了锁，"自旋"一词就是因此而得名。 
　　由于自旋锁使用者一般保持锁时间非常短，因此选择自旋而不是睡眠是非常必要的，自旋锁的效率远高于互斥锁。 
　　信号量和读写信号量适合于保持时间较长的情况，他们会导致调用者睡眠，因此只能在进程上下文使用（_trylock的变种能够在中断上下文使用），而自旋锁适合于保持时间非常短的情况，他能在所有上下文使用。 
　　如果被保护的共享资源只在进程上下文访问，使用信号量保护该共享资源非常合适，如果对共巷资源的访问时间非常短，自旋锁也能。不过如果被保护的共享资源需要在中断上下文访问（包括底半部即中断处理句柄和顶半部即软中断），就必须使用自旋锁。 
　　自旋锁保持期间是抢占失效的，而信号量和读写信号量保持期间是能被抢占的。自旋锁只有在内核可抢占或SMP的情况下才真正需要，在单CPU且不可抢占的内核下，自旋锁的所有操作都是空操作。 
　 　跟互斥锁相同，一个执行单元要想访问被自旋锁保护的共享资源，必须先得到锁，在访问完共享资源后，必须释放锁。如果在获取自旋锁时，没有所有执行单元保 持该锁，那么将即时得到锁；如果在获取自旋锁时锁已有保持者，那么获取锁操作将自旋在那里，直到该自旋锁的保持者释放了锁。 
　　无论是互斥锁，还是自旋锁，在所有时刻，最多只能有一个保持者，也就说，在所有时刻最多只能有一个执行单元获得锁。 

　　自旋锁的API有： 

spin_lock_init(x) 
　　该宏用于初始化自旋锁x。自旋锁在真正使用前必须先初始化。该宏用于动态初始化。 

DEFINE_SPINLOCK(x) 
　　该宏声明一个自旋锁x并初始化他。该宏在2.6.11中第一次被定义，在先前的内核中并没有该宏。 

SPIN_LOCK_UNLOCKED 
　　该宏用于静态初始化一个自旋锁。 

DEFINE_SPINLOCK(x)等同于spinlock_t x = SPIN_LOCK_UNLOCKED

spin_is_locked(x) 
　　该宏用于判断自旋锁x是否已被某执行单元保持（即被锁），如果是，返回真，否则返回假。 

spin_unlock_wait(x) 
　　该宏用于等待自旋锁x变得没有被所有执行单元保持，如果没有所有执行单元保持该自旋锁，该宏即时返回，否则将循环在那里，直到该自旋锁被保持者释放。 

spin_trylock(lock) 
　　该宏尽力获得自旋锁lock，如果能即时获得锁，他获得锁并返回真，否则不能即时获得锁，即时返回假。他不会自旋等待lock被释放。 

spin_lock(lock) 
　　该宏用于获得自旋锁lock，如果能够即时获得锁，他就马上返回，否则，他将自旋在那里，直到该自旋锁的保持者释放，这时，他获得锁并返回。总之，只有他获得锁才返回。 

spin_lock_irqsave(lock, flags) 
　　该宏获得自旋锁的同时把标志寄存器的值保存到变量flags中并失效本地中断。 

spin_lock_irq(lock) 
　　该宏类似于spin_lock_irqsave，只是该宏不保存标志寄存器的值。 

spin_lock_bh(lock) 
　　该宏在得到自旋锁的同时失效本地软中断。 

spin_unlock(lock) 
　　该宏释放自旋锁lock，他和spin_trylock或spin_lock配对使用。如果spin_trylock返回假，表明没有获得自旋锁，因此不必使用spin_unlock释放。 

spin_unlock_irqrestore(lock, flags) 
　　该宏释放自旋锁lock的同时，也恢复标志寄存器的值为变量flags保存的值。他和spin_lock_irqsave配对使用。 

spin_unlock_irq(lock) 
　　该宏释放自旋锁lock的同时，也使能本地中断。他和spin_lock_irq配对应用。 

spin_unlock_bh(lock) 
　　该宏释放自旋锁lock的同时，也使能本地的软中断。他和spin_lock_bh配对使用。 

spin_trylock_irqsave(lock, flags) 
　　该宏如果获得自旋锁lock，他也将保存标志寄存器的值到变量flags中，并且失效本地中断，如果没有获得锁，他什么也不做。 
　　因此如果能够即时获得锁，他等同于spin_lock_irqsave，如果不能获得锁，他等同于spin_trylock。如果该宏获得自旋锁lock，那需要使用spin_unlock_irqrestore来释放。 

spin_trylock_irq(lock) 
　　该宏类似于spin_trylock_irqsave，只是该宏不保存标志寄存器。如果该宏获得自旋锁lock，需要使用spin_unlock_irq来释放。 

spin_trylock_bh(lock) 
　　该宏如果获得了自旋锁，他也将失效本地软中断。如果得不到锁，他什么也不做。因此，如果得到了锁，他等同于spin_lock_bh，如果得不到锁，他等同于spin_trylock。如果该宏得到了自旋锁，需要使用spin_unlock_bh来释放。 

spin_can_lock(lock) 
　　该宏用于判断自旋锁lock是否能够被锁，他实际是spin_is_locked取反。如果lock没有被锁，他返回真，否则，返回假。该宏在2.6.11中第一次被定义，在先前的内核中并没有该宏。 

　　获得自旋锁和释放自旋锁有好几个版本，因此让读者知道在什么样的情况下使用什么版本的获得和释放锁的宏是非常必要的。 
　 　如果被保护的共享资源只在进程上下文访问和软中断上下文访问，那么当在进程上下文访问共享资源时，可能被软中断打断，从而可能进入软中断上下文来对被保 护的共享资源访问，因此对于这种情况，对共享资源的访问必须使用spin_lock_bh和spin_unlock_bh来保护。 
　 　当然使用spin_lock_irq和spin_unlock_irq及spin_lock_irqsave和 spin_unlock_irqrestore也能，他们失效了本地硬中断，失效硬中断隐式地也失效了软中断。不过使用spin_lock_bh和 spin_unlock_bh是最恰当的，他比其他两个快。 
　　如果被保护的共享资源只在进程上下文和tasklet或timer上下文访问，那么应该使用和上面情况相同的获得和释放锁的宏，因为tasklet和timer是用软中断实现的。 
　 　如果被保护的共享资源只在一个tasklet或timer上下文访问，那么不必所有自旋锁保护，因为同一个tasklet或timer只能在一个CPU 上运行，即使是在SMP环境下也是如此。实际上tasklet在调用tasklet_schedule标记其需要被调度时已把该tasklet绑定到当前 CPU，因此同一个tasklet决不可能同时在其他CPU上运行。 
　　timer也是在其被使用add_timer添加到timer队列中时已被帮定到当前CPU，所以同一个timer绝不可能运行在其他CPU上。当然同一个tasklet有两个实例同时运行在同一个CPU就更不可能了。 
　 　如果被保护的共享资源只在两个或多个tasklet或timer上下文访问，那么对共享资源的访问仅需要用spin_lock和spin_unlock 来保护，不必使用_bh版本，因为当tasklet或timer运行时，不可能有其他tasklet或timer在当前CPU上运行。 
　　如果被保护的共享资源只在一个软中断（tasklet和timer除外）上下文访问，那么这个共享资源需要用spin_lock和spin_unlock来保护，因为同样的软中断能同时在不同的CPU上运行。 
　　如果被保护的共享资源在两个或多个软中断上下文访问，那么这个共享资源当然更需要用spin_lock和spin_unlock来保护，不同的软中断能够同时在不同的CPU上运行。 
　 　如果被保护的共享资源在软中断（包括tasklet和timer）或进程上下文和硬中断上下文访问，那么在软中断或进程上下文访问期间，可能被硬中断打 断，从而进入硬中断上下文对共享资源进行访问，因此，在进程或软中断上下文需要使用spin_lock_irq和spin_unlock_irq来保护对 共享资源的访问。 
　　而在中断处理句柄中使用什么版本，需依情况而定，如果只有一个中断处理句柄访问该共享资源，那么在中断处理句柄中仅需要spin_lock和spin_unlock来保护对共享资源的访问就能了。 
　　因为在执行中断处理句柄期间，不可能被同一CPU上的软中断或进程打断。不过如果有不同的中断处理句柄访问该共享资源，那么需要在中断处理句柄中使用spin_lock_irq和spin_unlock_irq来保护对共享资源的访问。 
　 　在使用spin_lock_irq和spin_unlock_irq的情况下，完万能用spin_lock_irqsave和 spin_unlock_irqrestore取代，那具体应该使用哪一个也需要依情况而定，如果能确信在对共享资源访问前中断是使能的，那么使用 spin_lock_irq更好一些。 
　　因为他比spin_lock_irqsave要快一些，不过如果你不能确定是否中断使能，那么使用spin_lock_irqsave和spin_unlock_irqrestore更好，因为他将恢复访问共享资源前的中断标志而不是直接使能中断。 
　　当然，有些情况下需要在访问共享资源时必须中断失效，而访问完后必须中断使能，这样的情形使用spin_lock_irq和spin_unlock_irq最佳。 
　　需要特别提醒读者，spin_lock用于阻止在不同CPU上的执行单元对共享资源的同时访问及不同进程上下文互相抢占导致的对共享资源的非同步访问，而中断失效和软中断失效却是为了阻止在同一CPU上软中断或中断对共享资源的非同步访问。

      读-写自旋锁当对某个数据结构的操作可以清晰化为读/写两种类型时，读/写自旋锁就派上了用场。一个或多个读任务可以并发的持有读锁；相反，写锁只能被一个写任务所持有，而且此时不能有并发的读操作。多个读任务可以安全 地获得同一个读锁，事实上，即使一个线程递归地获取同一读锁也是安全 的。

    初始化读/写锁：

    rwlock_t my_rwlock=RW_LOCK_UNLOCKED;

    在读任务的代码分支上：

    read_lock(&my_rwlock);

    /*读数据*/

    read_unlock(&my_rwlock);

    在写任务的代码分支上：

    write_lock(&my_rwlock);

    /*读数据*/

    write_unlock(&my_rwlock);

    注意:不能把一个读锁升级为一个写锁：

    read_lock(&my_rwlock);

    write_lock(&my_rwlock);

    这将会带来死锁，因为写锁会不断自旋，等待所有的读锁被释放，其中包括自己。

    内核实现的读锁和写锁接口在<linux /spinlock.h>中有定义。

    在使用Linux读-写自旋锁时，需要考虑一点事这种机制照顾读比照顾写要多一点。当读锁被持有时，写操作为了互斥访问只能等待，但是读任务却可以继续成功的占用读锁，而且自旋等待的写任务在所有读任务释放锁之前是无法获得写锁的。