Linux内核的同步机制（二）：自旋锁

自旋锁：单处理器非抢占式内核和对称多处理器或抢占式内核
Linux 2.4.x及以前的版本都是非抢占式内核方式，如果编译成单处理器系统，在同一时间只有一个进程在执行，除非它自己放弃，不然只有通过"中断"才能中断其执行。因此，在单处理器非抢占式内核中，如果需要修改某个重要的数据结构，或者执行某些关键代码，只需要禁止中断。
但是在对称多处理器，仅仅禁止某个CPU的中断是不够的，当然我们也可以将所有CPU的中断都禁止，但这样做开销很大，整个系统的性能会明显下降。此外，即使在单处理器上，如果内核是抢占式的，也可能出现不同进程上下文同时进入临界区的情况。为此，Linux内核中提供了"自旋锁（spinlock）"的同步机制。
自旋锁与互斥锁有点类似，只是自旋锁不会引起调用者睡眠，如果自旋锁已经被别的执行单元保持，调用者就一直循环在那里看是否该自旋锁的保持者已经释放了锁，"自旋"因此而得名。
因此，中断（或软中断）禁止用于防止同一CPU上中断（或软中断）对共享资源的非同步访问。而自旋锁则防止在不同CPU上的执行单元对共享资源的同时访问，以及不同进程上下文互相抢占导致的对共享资源的非同步访问。
我们Linux 2.4.21为基础，分析x86平台下自旋锁的类型及应用方式，相关代码在源代码树的include/linux/spinlock.h以及include/asm-i386/spinlock.h中。
Linux内核中的自旋锁
在Linux内核中，自旋锁的基本使用方式如下：
先声明一个spinlock_t类型的自旋锁变量，并初始化为"未加锁"状态。在进入临界区之前，调用加锁函数获得锁，在退出临界区之前，调用解锁函数释放锁。例如：
spinlock_t lock = SPIN_LOCK_UNLOCKED;
spin_lock(&lock);
/* 临界区 /
spin_unlock(&lock);
获得自旋锁和释放自旋锁的函数有多种变体。
spin_lock_irqsave/spin_unlock_irqrestore
相对于自旋锁的其它函数组，这一组函数是最"安全"的，使用频率也最多。在调用spin_lock_irqsave之前，我们还需要声明一个unsign long类型的变量（例如flag），该函数可以顺序完成下列操作：
1. 将CPU的标志寄存器的内容保存在变量flag中；
2. 禁止CPU的本地中断；
3. 调用spin_lock获得自旋锁。
而spin_unlock_irqrestore函数则在调用spin_unlock释放自旋锁之后，将变量flag保存的值恢复到CPU的标志寄存器中。
#define spin_lock_irqsave(lock, flags)    do { local_irq_save(flags);         spin_lock(lock); } while (0)
#define local_irq_save(x) __save_and_cli(x)
#define __save_and_cli(x) do { __save_flags(x); cli(); } while(0);
#define save_flags(x)    asm volatile("pushfl ; popl %0":"=g" (x): / no input /)
保存CPU的标志寄存器方法是：首先调用pushfl将标志寄存器压栈，再调用popl从栈中弹出保存在变量参数中。
#define cli()     asm volatile__("cli": : :"memory")
禁止CPU的本地中断使用cli汇编指令。
#define spin_unlock_irqrestore(lock, flags) do { spin_unlock(lock);    local_irq_restore(flags); } while (0)
#define local_irq_restore(x) __restore_flags(x)
#define __restore_flags(x)    asm volatile("pushl %0 ; popfl": / no output / :"g" (x):"memory", "cc")
恢复CPU的标志寄存器方法是：首先调用pushl将变量参数压到栈中，再调用popfl从栈中弹出保存到标志寄存器。
需要注意的是，这里没有显式执行开中断的动作。实际上，在标志寄存器中保持了原来的中断状态，在恢复寄存器的同时将中断也恢复到以前的状态。
spin_lock_irq/spin_unlock_irq
和上面一组函数的不同在于，这一组函数并不涉及标志寄存器。spin_lock_irq函数首先禁止CPU的本地中断，再调用spin_lock获得自旋锁。而spin_unlock_irq函数则首先调用spin_unlock释放自旋锁，再打开CPU的本地中断。
#define spin_lock_irq(lock)     do { local_irq_disable();           spin_lock(lock); } while (0)
#define local_irq_disable() __cli()
#define spin_unlock_irq(lock)     do { spin_unlock(lock);    local_irq_enable();         } while (0)
#define local_irq_enable() sti()
#define sti()     asm volatile("sti": : :"memory")
打开CPU的本地中断使用sti汇编指令。
spin_lock_bh/spin_unlock_bh/spin_trylock_bh
spin_lock_bh函数在得到自旋锁的同时禁止本地软中断，spin_unlock_bh函数释放自旋锁的同时，也打开本地的软中断。
#define spin_lock_bh(lock)     do { local_bh_disable();            spin_lock(lock); } while (0)
#define local_bh_disable() cpu_bh_disable(smp_processor_id())
#define cpu_bh_disable(cpu) \
    do { local_bh_count(cpu)++; barrier(); } while (0)
#define spin_unlock_bh(lock)     do { spin_unlock(lock);    local_bh_enable();          } while (0)
#define spin_trylock_bh(lock)     ({ int __r; local_bh_disable();\
__r = spin_trylock(lock);        \
if (!__r) local_bh_enable();     \
__r; })
spin_lock/spin_unlock/spin_trylock
上面各组函数最终都需要调用自旋锁操作函数。spin_lock函数用于获得自旋锁，如果能够立即获得锁，它就马上返回，否则，它将自旋在那里，直到该自旋锁的保持者释放。spin_unlock函数则释放自旋锁。此外，还有一个spin_trylock函数。尽力获得自旋锁lock，如果能立即获得锁，它获得锁并返回真，否则不能立即获得锁，立即返回假。它不会自旋等待lock被释放。
spin_lock/spin_unlock/spin_trylock在UP环境下
由于Linux 2.4.x为不可抢占的内核，在单处理器环境下，自旋锁什么都不需要做。自旋锁类型spinlock_t设置为空，除了在GCC的早期版本中不支持内容为空的数据结构。
typedef struct {
} spinlock_t;
相应地，自旋锁的操作函数不进行任何实质性处理。
#define spin_lock_init(lock) do { } while(0)
#define spin_lock(lock)    (void)(lock) / Not "unused variable". /
#define spin_is_locked(lock) (0)
#define spin_trylock(lock) ({1; })
#define spin_unlock_wait(lock) do { } while(0)
#define spin_unlock(lock) do { } while(0)
spin_lock/spin_unlock/spin_trylock在SMP环境下
在SMP环境下，自旋锁类型spinlock_t含有一个unsigned int域lock。未加锁时值为1，加锁后值为0或负值。声明时也使用了volatile描述符，要求编译器不要对其作优化处理，对它的读写都需要从内存中访问。
typedef struct {
      volatile unsigned int lock;
} spinlock_t;
static inline void spin_lock(spinlock_t lock)
{
      asm volatile(
          spin_lock_string
          :"=m" (lock->lock) : : "memory");
}
从上看到，spin_lock的主要内容是spin_lock_string宏，定义如下：
#define spin_lock_string \
      "\n1:\t" \
      "lock ; decb %0\n\t" \
      "js 2f\n" \
      LOCK_SECTION_START("") \
      "2:\t" \
      "cmpb $0,%0\n\t" \
      "rep;nop\n\t" \
      "jle 2b\n\t" \
      "jmp 1b\n" \
      LOCK_SECTION_END
在上面的汇编代码中，首先将lock递减，如果为负（表明锁已被持有），则进入自旋。每次自旋在执行一条空指令后，根据lock和0的比较结果进行跳转。如果lock小于或等于0，则继续自旋；否则（说明锁已被释放），则将lock递减后返回。
在这里，采取了一种优化手段，因为在大多数情况下，自旋锁是能成功获取的，而自旋部分代码，只是在锁被持有时才执行，因此利用LOCK_SECTION_START和LOCK_SECTION_END将这些代码放到一个专门的区（.text.lock）中。如果把它跟别的常用指令混在一起，会浪费指令缓存的空间。
理解这一点，我们也就能明白spin_lock是如何退出的了。事实上，由于不再同一个区（section），所以"js 2f"的下一条指令并不是"cmpb $0,%0"。
#define LOCK_SECTION_NAME     \
      ".text.lock." __stringify(KBUILD_BASENAME)
#define LOCK_SECTION_START(extra)    \
      ".subsection 1\n\t"     \
      extra       \
      ".ifndef " LOCK_SECTION_NAME "\n\t" \
      LOCK_SECTION_NAME ":\n\t"    \
      ".endif\n\t"
#define LOCK_SECTION_END     \
      ".previous\n\t"
如果用类C语言来描述上述过程，将是：
void spin_lock(int &lock)
{
      int flag;

label1:
      flag = lock--;
      if (flag >= 0) {
          return;
      } else {
          do {
              nop;
          } while(lock <= 0);
          goto label1;
      }
}
当然，我们需要保证lock--为原子操作，就像在汇编中在decb指令前加上lock前缀一样。释放锁时使用spin_unlock_string宏，定义如下：
static inline void spin_unlock(spinlock_t *lock)
{
      __asm__ __volatile__(
          spin_unlock_string
      );
}
#define spin_unlock_string \
      "movb $1,%0" \
      :"=m" (lock->lock) : : "memory"
需要强调的是，为了和spin_lock相配合，我们在释放锁是必须将lock设置为1，而不能使用lock++的方式。
#define SPIN_LOCK_UNLOCKED (spinlock_t) { 1 SPINLOCK_MAGIC_INIT }
#define spin_lock_init(x) do { *(x) = SPIN_LOCK_UNLOCKED; } while(0)
#define spin_is_locked(x) (*(volatile signed char *)(&(x)->lock) <= 0)
#define spin_unlock_wait(x) do { barrier(); } while(spin_is_locked(x))
由于spin_trylock不需要自旋，实现中采用xchgb指令（该指令将自动锁总线，而不需要再使用lock前缀）。若lock原来的值为1，说明未上锁，因此返回为真，表明成功获得锁；否则返回为假。
static inline int spin_trylock(spinlock_t *lock)
{
      char oldval;
      __asm__ __volatile__(
          "xchgb %b0,%1"
          :"=q" (oldval), "=m" (lock->lock)
          :"0" (0) : "memory");
          return oldval > 0;
}
自旋锁的应用
在讨论自旋锁的应用时，我们一般区分两种平台：单处理器非抢占式内核和对称多处理器或抢占式内核。在前面我们看到，在单处理器非抢占式内核下，自旋锁根本不存在。这体现了一种出色的设计策略，既然没有别人能够同时刻执行，就没有理由加锁。对于抢占式内核，我们将它等同于对称多处理器来考虑。
1．用户上下文之间
如果数据结构只可能被用户上下文访问，最高效的办法就是使用信号量。（我们在后面将讨论信号量机制）。
2．用户上下文与softirq之间
这种情况下，使用spin_lock_bh()/spin_unlock_bh()可以满足要求。如果是单处理器非抢占式内核，自旋锁消失了，spin_lock_bh等同于local_bh_disable，会进制在用户上下文时进制softirq，从而避免用户上下文和softirq同时进入临界区。如果是对称多处理器或者抢占式内核，即使是在不同CPU上的用户上下文和softirq同时运行，自旋锁机制保证了只有一个持有者，只有在它释放锁之后，另一个才能进入临界区。
3．用户上下文和Tasklet/Timer之间
同上。同加锁观点来看，Tasklet和Timer的地位是同样的。
4． Tasklet或Timer之间
这里有两点需要说明：（1）同一时刻，一个Tasklet或Timer不会同时在两个CPU上执行；（2）如果CPU已经处在Tasklet或Timer中，它不会同时再执行其它的Tasklet或Timer。因此我们只需要考虑在不同CPU上运行两个不同Tasklet或Timer的情况，而这种情况只需要使用自旋锁机制，即spin_lock和spin_unlock函数。
5． Softirq之间或和Tasklet/Timer之间
同一个softirq可能在不同的CPU上执行，同上道理，使用spin_lock和spin_unlock可以在不同CPU的同一个或不同softirq，或者Softirq与Tasklet或Timer之间保护共享数据。
6．硬件中断之间
如果被保护的共享资源在软中断（包括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最好。

Linux内核的同步机制（三）：等待队列

Linux内核中的等待队列
Linux内核的等待队列是以双循环链表为基础数据结构，与进程调度机制紧密结合，能够用于实现核心的异步事件通知机制。在Linux2.4.21中，等待队列在源代码树include/linux/wait.h中，这是一个通过list_head连接的典型双循环链表，如下图所示。

在这个链表中，有两种数据结构：等待队列头（wait_queue_head_t）和等待队列项（wait_queue_t）。等待队列头和等待队列项中都包含一个list_head类型的域作为"连接件"。由于我们只需要对队列进行添加和删除操作，并不会修改其中的对象（等待队列项），因此，我们只需要提供一把保护整个基础设施和所有对象的锁，这把锁保存在等待队列头中，为wq_lock_t类型。在实现中，可以支持读写锁（rwlock）或自旋锁（spinlock）两种类型，通过一个宏定义来切换。如果使用读写锁，将wq_lock_t定义为rwlock_t类型；如果是自旋锁，将wq_lock_t定义为spinlock_t类型。无论哪种情况，分别相应设置wq_read_lock、wq_read_unlock、wq_read_lock_irqsave、wq_read_unlock_irqrestore、wq_write_lock_irq、wq_write_unlock、wq_write_lock_irqsave和wq_write_unlock_irqrestore等宏。
等待队列头
struct __wait_queue_head {
wq_lock_t lock;
struct list_head task_list;
};
typedef struct __wait_queue_head wait_queue_head_t;
前面已经说过，等待队列的主体是进程，这反映在每个等待队列项中，是一个任务结构指针（struct task_struct * task）。flags为该进程的等待标志，当前只支持互斥。
等待队列项
struct __wait_queue {
unsigned int flags;
#define WQ_FLAG_EXCLUSIVE 0x01
struct task_struct * task;
struct list_head task_list;
};
typedef struct __wait_queue wait_queue_t;
声明和初始化
#define DECLARE_WAITQUEUE(name, tsk)       \
wait_queue_t name = __WAITQUEUE_INITIALIZER(name, tsk)
#define __WAITQUEUE_INITIALIZER(name, tsk) {      \
task:    tsk,        \
task_list: { NULL, NULL },       \
      __WAITQUEUE_DEBUG_INIT(name)}
通过DECLARE_WAITQUEUE宏将等待队列项初始化成对应的任务结构，并且用于连接的相关指针均设置为空。其中加入了调试相关代码。
#define DECLARE_WAIT_QUEUE_HEAD(name) \
wait_queue_head_t name = __WAIT_QUEUE_HEAD_INITIALIZER(name)
#define __WAIT_QUEUE_HEAD_INITIALIZER(name) {      \
lock:    WAITQUEUE_RW_LOCK_UNLOCKED,     \
task_list: { &(name).task_list, &(name).task_list }, \
     __WAITQUEUE_HEAD_DEBUG_INIT(name)}
通过DECLARE_WAIT_QUEUE_HEAD宏初始化一个等待队列头，使得其所在链表为空，并设置链表为"未上锁"状态。其中加入了调试相关代码。
static inline void init_waitqueue_head(wait_queue_head_t *q)
该函数初始化一个已经存在的等待队列头，它将整个队列设置为"未上锁"状态，并将链表指针prev和next指向它自身。
{
      q->lock = WAITQUEUE_RW_LOCK_UNLOCKED;
      INIT_LIST_HEAD(&q->task_list);
}
static inline void init_waitqueue_entry(wait_queue_t *q, struct task_struct *p)
该函数初始化一个已经存在的等待队列项，它设置对应的任务结构，同时将标志位清0。
{
      q->flags = 0;
      q->task = p;
}
static inline int waitqueue_active(wait_queue_head_t *q)
该函数检查等待队列是否为空。
{
      return !list_empty(&q->task_list);
}
static inline void __add_wait_queue(wait_queue_head_t *head, wait_queue_t *new)
将指定的等待队列项new添加到等待队列头head所在的链表头部，该函数假设已经获得锁。
{
      list_add(&new->task_list, &head->task_list);
}
static inline void __add_wait_queue_tail(wait_queue_head_t *head, wait_queue_t *new)
将指定的等待队列项new添加到等待队列头head所在的链表尾部，该函数假设已经获得锁。
{
      list_add_tail(&new->task_list, &head->task_list);
}
static inline void __remove_wait_queue(wait_queue_head_t *head, wait_queue_t *old)
将函数从等待队列头head所在的链表中删除指定等待队列项old，该函数假设已经获得锁，并且old在head所在链表中。
{
      list_del(&old->task_list);
}
睡眠和唤醒操作
对等待队列的操作包括睡眠和唤醒（相关函数保存在源代码树的/kernel/sched.c和include/linux/sched.h中）。思想是更改当前进程（CURRENT）的任务状态，并要求重新调度，因为这时这个进程的状态已经改变，不再在调度表的就绪队列中，因此无法再获得执行机会，进入"睡眠"状态，直至被"唤醒"，即其任务状态重新被修改回就绪态。
常用的睡眠操作有interruptible_sleep_on和sleep_on。两个函数类似，只不过前者将进程的状态从就绪态（TASK_RUNNING）设置为TASK_INTERRUPTIBLE，允许通过发送signal唤醒它（即可中断的睡眠状态）；而后者将进程的状态设置为TASK_UNINTERRUPTIBLE，在这种状态下，不接收任何singal。以interruptible_sleep_on为例，其展开后的代码是：
void interruptible_sleep_on(wait_queue_head_t *q)
{
      unsigned long flags;
      wait_queue_t wait;
      /* 构造当前进程对应的等待队列项 */
      init_waitqueue_entry(&wait, current);
      /* 将当前进程的状态从TASK_RUNNING改为TASK_INTERRUPTIBLE */
      current->state = TASK_INTERRUPTIBLE;
      /* 将等待队列项添加到指定链表中 */
      wq_write_lock_irqsave(&q->lock,flags);
      __add_wait_queue(q, &wait);
      wq_write_unlock(&q->lock);
      /* 进程重新调度，放弃执行权 */
      schedule();
      /* 本进程被唤醒，重新获得执行权，首要之事是将等待队列项从链表中删除 */
      wq_write_lock_irq(&q->lock);
      __remove_wait_queue(q, &wait);
      wq_write_unlock_irqrestore(&q->lock,flags);
      /* 至此，等待过程结束，本进程可以正常执行下面的逻辑 */
}
对应的唤醒操作包括wake_up_interruptible和wake_up。wake_up函数不仅可以唤醒状态为TASK_UNINTERRUPTIBLE的进程，而且可以唤醒状态为TASK_INTERRUPTIBLE的进程。wake_up_interruptible只负责唤醒状态为TASK_INTERRUPTIBLE的进程。这两个宏的定义如下：
#define wake_up(x)     __wake_up((x),TASK_UNINTERRUPTIBLE | TASK_INTERRUPTIBLE, 1)
#define wake_up_interruptible(x) __wake_up((x),TASK_INTERRUPTIBLE, 1)
__wake_up函数主要是获取队列操作的锁，具体工作是调用__wake_up_common完成的。
void __wake_up(wait_queue_head_t *q, unsigned int mode, int nr)
{
      if (q) {
          unsigned long flags;
          wq_read_lock_irqsave(&q->lock, flags);
          __wake_up_common(q, mode, nr, 0);
          wq_read_unlock_irqrestore(&q->lock, flags);
      }
}
/* The core wakeup function.    Non-exclusive wakeups (nr_exclusive == 0) just wake everything up.    If it's an exclusive wakeup (nr_exclusive == small +ve number) then we wake all the non-exclusive tasks and one exclusive task.
There are circumstances in which we can try to wake a task which has already started to run but is not in state TASK_RUNNING.    try_to_wake_up() returns zero in this (rare) case, and we handle it by contonuing to scan the queue. */
static inline void __wake_up_common (wait_queue_head_t *q, unsigned int mode, int nr_exclusive, const int sync)
参数q表示要操作的等待队列，mode表示要唤醒任务的状态，如TASK_UNINTERRUPTIBLE或TASK_INTERRUPTIBLE等。nr_exclusive是要唤醒的互斥进程数目，在这之前遇到的非互斥进程将被无条件唤醒。sync表示？？？
{
      struct list_head *tmp;
      struct task_struct *p;
      CHECK_MAGIC_WQHEAD(q);
      WQ_CHECK_LIST_HEAD(&q->task_list);
      /* 遍历等待队列 */
      list_for_each(tmp,&q->task_list) {
          unsigned int state;
          /* 获得当前等待队列项 */
          wait_queue_t *curr = list_entry(tmp, wait_queue_t, task_list);
          CHECK_MAGIC(curr->__magic);
          /* 获得对应的进程 */
          p = curr->task;
          state = p->state;
          /* 如果我们需要处理这种状态的进程 */
          if (state & mode) {
              WQ_NOTE_WAKER(curr);
              if (try_to_wake_up(p, sync) && (curr->flags&WQ_FLAG_EXCLUSIVE) && !--nr_exclusive)
                  break;
          }
      }
}
/* 唤醒一个进程，将它放到运行队列中，如果它还不在运行队列的话。"当前"进程总是在运行队列中的（except when the actual re-schedule is in progress)，and as such you're allowed to do the simpler "current->state = TASK_RUNNING" to mark yourself runnable without the overhead of this. */
static inline int try_to_wake_up(struct task_struct * p, int synchronous)
{
      unsigned long flags;
      int success = 0;
      /* 由于我们需要操作运行队列，必须获得对应的锁 */
      spin_lock_irqsave(&runqueue_lock, flags);
      /* 将进程状态设置为TASK_RUNNING */
      p->state = TASK_RUNNING;
      /* 如果进程已经在运行队列中，释放锁退出 */
      if (task_on_runqueue(p))
          goto out;
      /* 否则将进程添加到运行队列中 */
      add_to_runqueue(p);
      /* 如果设置了同步标志 */
      if (!synchronous || !(p->cpus_allowed & (1UL << smp_processor_id())))
          reschedule_idle(p);
      /* 唤醒成功，释放锁退出 */
      success = 1;
out:
      spin_unlock_irqrestore(&runqueue_lock, flags);
      return success;
}
等待队列应用模式
等待队列的的应用涉及两个进程，假设为A和B。A是资源的消费者，B是资源的生产者。A在消费的时候必须确保资源已经生产出来，为此定义一个资源等待队列。这个队列同时要被进程A和进程B使用，我们可以将它定义为一个全局变量。
DECLARE_WAIT_QUEUE_HEAD(rsc_queue); /* 全局变量 */
在进程A中，执行逻辑如下：
while (resource is unavaiable) {
      interruptible_sleep_on( &wq );
}
consume_resource();
在进程B中，执行逻辑如下：
produce_resource();
wake_up_interruptible( &wq );

Linux内核的同步机制（四）：completion

在Linux内核中，completion是一种简单的同步机制，标志"things may proceed"。要使用completion，必须在文件中包含，同时创建一个类型为struct completion的变量。这个变量可以静态地声明和初始化：
DECLARE_COMPLETION(my_comp);
或者动态初始化：
struct completion my_comp;
init_completion(&my_comp);
如果驱动程序要在执行后面操作之前等待某个过程的完成，它可以调用wait_for_completion ，以要完成的事件为参数：
void wait_for_completion(struct completion comp);
如果其它部分代码可以确定事件已经完成，可以调用下面两个函数之一来唤醒等待该事件的进程：
void complete(struct completion comp);
void complete_all(struct completion comp); / Linux 2.5.x以上版本 /
前一个函数将只唤醒一个等待进程，而后一个函数唤醒等待该事件的所以进程。由于completion的实现方式，即使complete在wait_for_competion之前调用，也可以正常工作。
例如，在MD设备驱动程序实现中，有一个恢复线程md_recovery_thread。驱动程序通过md_register_thread和md_unregister_thread来注册和注销恢复线程。恢复线程的执行逻辑在md_thread函数中，大致如下：
int md_thread(void arg)
{
      线程初始化;
      while (运行) {
          处理逻辑;
          接收信号;
      }
      return 0;
}
md_register_thread将创建一个恢复线程，它必须在线程真正初始化结束之后才能返回该线程的指针。因此，其逻辑是：
mdk_thread_t md_register_thread(void (run) (void ), void data, const char name)
{
      mdk_thread_t thread;
      ……
      struct completion event;
      /* 为线程分配空间 /
      thread = (mdk_thread_t ) kmalloc (sizeof(mdk_thread_t), GFP_KERNEL);
      ……
      init_completion(&event);
      ……
      thread->event = &event;
      /* 创建内核线程 /
      ret = kernel_thread(md_thread, thread, 0);
      / 等待线程初始化结束 /
      ……
      wait_for_completion(&event);
      / 返回线程指针 /
      return thread;
}
而md_unregister_thread通过向线程发送SIGKILL信号注销恢复线程，它也需要在线程真正退出后才能释放线程所占用的内存。因此，其逻辑是：
void md_unregister_thread(mdk_thread_t thread)
{
      struct completion event;
      init_completion(&event);
      thread->event = &event;
      ……
      /* 向线程发送SIGKILL信号终止其运行 /
      md_interrupt_thread(thread);
      / 等待线程退出 /
      wait_for_completion(&event);
      / 释放线程所占用的内存 /
      kfree(thread);
}
如果考虑completion，md_thread的逻辑是：
int md_thread(void arg)
{
      线程初始化;
      complete(thread->event);
      while (运行) {
          处理逻辑;
          接收信号;
      }
      complete(thread->event);
      return 0;
}
需要说明的是，由于等待事件是在驱动程序和恢复线程中的一个共享资源，它必须是一个全局变量，或者如实现代码中，定义为一个局部变量，而将其指针放在恢复线程结构中。
typedef struct mdk_thread_s {
      ……
      struct completion *event;
      ……
} mdk_thread_t;

阅读(4002) | 评论(0) | 转发(3) |

上一篇：(转)全面解析Linux内核的同步与互斥机制--同步篇

下一篇：(转)linux的同步与互斥

给主人留下些什么吧！~~

感谢所有关心和支持过ChinaUnix的朋友们

16024965号-6