Linux内核同步机制之completion

一、简介
    内核编程中常见的一种模式是，在当前线程之外初始化某个活动，然后等待该活动的结束。这个活动可能是，创建一个新的内核线程或者新的用户空间进程、对一个已有进程的某个请求，或者某种类型的硬件动作，等等。在这种情况下，我们可以使用信号量来同步这两个任务。然而，内核中提供了另外一种机制——completion接口。Completion是一种轻量级的机制，他允许一个线程告诉另一个线程某个工作已经完成。
 
二、结构

    struct completion {
        unsigned int                 done;
        wait_queue_head_t   wait;
    };

 

三、初始化
    和信号量一样，初始化分为静态初始化和动态初始化两种情况：

    静态初始化：
        #define COMPLETION_INITIALIZER(work) { 0, __WAIT_QUEUE_HEAD_INITIALIZER((work).wait) }
        #define DECLARE_COMPLETION(work) struct completion work = COMPLETION_INITIALIZER(work)

    动态初始化：
        static inline void init_completion(struct completion *x) {
            x->done = 0;
            init_waitqueue_head(&x->wait);
        }

    可见，两种初始化都将用于同步的done原子量置位了0，后面我们会看到，该变量在wait相关函数中减一，在complete系列函数中加一。

 

四、等待完成量
    void wait_for_completion(struct completion *x);
    int wait_for_completion_interruptible(struct completion *x);
    unsigned long wait_for_completion_timeout(struct completion *x, unsigned long timeout);
    unsigned long wait_for_completion_interruptible_timeout(struct completion *x, unsigned long timeout);

 

五、唤醒完成量
    void complete(struct completion *x);
    void complete_all(struct completion *x);

 

六、使用案例
    #include <linux/module.h>
    #include <linux/init.h>
     
    #include <linux/sched.h>
    #include <linux/kernel.h>
    #include <linux/fs.h>
    #include <linux/types.h>
    #include <linux/completion.h>
     
    MODULE_LICENSE("GPL");
     
    static int complete_major=250;
    DECLARE_COMPLETION(g_comp);
     
    ssize_t complete_read(struct file *filp,char __user *buf,size_t count,loff_t *pos)  {
        printk(KERN_ERR "process %i (%s) going to sleep\n",current->pid,current->comm);
        wait_for_completion(&g_comp);
        printk(KERN_ERR "awoken %i (%s)\n",current->pid,current->comm);
        return 0;
    }
     
    ssize_t complete_write(struct file *filp,const char __user *buf,size_t count,loff_t *pos) {
        printk(KERN_ERR "process %i (%s) awakening the readers...\n",current->pid,current->comm);
        complete(&g_comp);
        return count;
    }
     
    struct file_operations complete_fops = {
        .owner=THIS_MODULE,
        .read=complete_read,
        .write=complete_write,
    };
     
    int complete_init(void) {
        int result;
        result=register_chrdev(complete_major,"complete",&complete_fops);
        if(result<0)
            return result;
        if(complete_major==0)
            complete_major=result;
        return 0;
    }

    void complete_cleanup(void) {
        unregister_chrdev(complete_major,"complete");
    }

    module_init(complete_init);
    module_exit(complete_cleanup);

 

七、测试步骤：
    1. mknod /dev/complete创建complete节点，在linux上驱动程序需要手动创建文件节点。
    2. insmod complete.ko 插入驱动模块，这里要注意的是，因为我们的代码中是手动分配的设备号，很可能被系统已经使用了，所以如果出现这种情况，查看/proc/devices文件。找一个没有被使用的设备号。
    3. cat /dev/complete 用于读该设备，调用设备的读函数
    4. 打开另一个终端输入 echo “hello” > /dev/complete 该命令用于写入该设备。

相关文件：linux/include/linux/completion.h、/linux/kernel/sched.c