如何在Linux驱动中使用互斥锁？

在Linux驱动中，互斥锁是很常用的工具，可以保证数据安全。下面的例子实现不同线程安全的访问临界空间数据。

驱动代码

// simple.c

#include <linux/module.h>
#include <linux/mutex.h>
#include <linux/sched.h>

static struct task_struct *kthread_1;
static struct task_struct *kthread_2;
static int t1 = 1, t2 = 2;
static struct mutex lock;

static int _cb(void *data) {
    const int nr = *(int *)data;

    while (!kthread_should_stop()) {
        mutex_lock(&lock);
        pr_info("thread %d is in critical section\n", nr);
        msleep(1000);
        pr_info("thread %d is leaving critical section\n", nr);
        mutex_unlock(&lock);
    }

    return 0;
}

static int __init simple_init(void) {
    pr_info("simple_init\n");

    mutex_init(&lock);

    kthread_1 = kthread_create(_cb, &t1, "kthread_1");
    if (!kthread_1) {
        pr_err("kthread_create thread1 error\n");
        return -1;
    } else {
        wake_up_process(kthread_1);
    }

    kthread_2 = kthread_run(_cb, &t2, "kthread_2");
    if (!kthread_2) {
        kthread_stop(kthread_1);
        pr_err("kthread_create thread2 error\n");
        return -1;
    }

    return 0;
}

static void __exit simple_exit(void) {
    pr_info("simple_exit\n");

    kthread_stop(kthread_1);
    kthread_stop(kthread_2);
}

module_init(simple_init);
module_exit(simple_exit);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("heng30");
MODULE_VERSION("v0.0.1");
MODULE_DESCRIPTION("A simple kernel module");

编译脚本

#!/bin/sh

top-dir = $(shell pwd)
kernel-version = $(shell uname -r)
kernel-dir ?= /lib/modules/$(kernel-version)/build

obj-m += simple.o

all:
        make -C $(kernel-dir) modules M=$(top-dir)

clean:
        rm -f *.o *.ko *.mod *.mod.c *.order *.symvers *.dtbo
        make -C $(kernel-dir) clean m=$(top-dir)

测试

• 编译程序：make

• 安装驱动：insmod simple.ko

• 查看输出：dmesg

[  502.660615] simple_init
[  502.660820] thread 1 is in critical section
[  503.725447] thread 1 is leaving critical section
[  503.725473] thread 1 is in critical section
[  504.741500] thread 1 is leaving critical section
[  504.741548] thread 2 is in critical section
[  505.757345] thread 2 is leaving critical section
[  505.757372] thread 2 is in critical section
[  506.781252] thread 2 is leaving critical section
...
[  523.363672] thread 1 is in critical section
[  524.384716] thread 1 is leaving critical section
[  524.384733] thread 1 is in critical section
[  525.408237] thread 1 is leaving critical section
[  526.246275] simple_exit

• 移除驱动：rmmod simple.ko