linux内核 并发 例子

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/spinlock.h>

static int counter = 0;
static DEFINE_SPINLOCK(my_lock);

static int __init my_module_init(void) {
    printk(KERN_INFO "My module initialized\n");

    // 创建多个线程来并发访问共享资源
    for (int i = 0; i < 10; ++i) {
        kthread_run(my_worker, NULL, "my_worker%d", i);
    }

    return 0;
}

static void __exit my_module_exit(void) {
    printk(KERN_INFO "My module exited\n");
}

static int my_worker(void *data) {
    for (int j = 0; j < 1000; ++j) {
        // 获取自旋锁
        spin_lock(&my_lock);
        counter++;
        // 释放自旋锁
        spin_unlock(&my_lock);
    }
    return 0;
}

module_init(my_module_init);
module_exit(my_module_exit);

MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("A simple Linux kernel module demonstrating concurrency with spinlocks");