时间戳

首页 > 解决方案 > 用“return”或“do_exit()”终止内核线程是一个好习惯吗？

c - 用“return”或“do_exit()”终止内核线程是一个好习惯吗？

问题描述

我的目标是只从我的驱动程序的探测函数执行内核线程，它执行固件下载。

为简单起见，放置示例代码（不是实际代码），

#include<linux/module.h>
#include<linux/init.h>
#include<linux/kthread.h>

MODULE_LICENSE("GPL");

struct task_struct *kthread;


static int thread_func(void* data)
{
    printk("In %s function\n", __func__);
    return 0;
}

static int hello_init(void)
{
    int ret = 0;

    printk("Hello World\n");
    kthread = kthread_run(thread_func,
            NULL, "kthread-test");
    if (IS_ERR(kthread)) {
        ret = PTR_ERR(kthread);
        printk("Unable to run kthread err %d\n", ret);
        return ret;
    }
    return 0;
}


static void hello_exit(void)
{
    printk("Bye World\n");

}

我没有使用以下任何一种，因为：

  1. kthread_should_stop()- 用于我不想要的连续执行
  2. kthread_stop(struct task_struct *thread)- 如果包含在模块退出函数中，会导致内核恐慌，因为线程在执行一次后已经终止

这是一个正确的方法吗？如果不是请建议

标签： clinuxmultithreadinglinux-kernellinux-device-driver

解决方案

您需要确保在模块退出函数返回之前线程已经消失。一种方法是使用“完成”结构。

基本思想是在启动线程之前初始化完成结构，使线程在退出时将完成结构标记为“完成”，并让模块退出函数（或其他）等待完成结构被标记为完成。

  1. 先决条件

    #include <linux/completion.h>

  2. 初始化完成结构

    如果完成结构变量是静态分配的，可以在变量定义中使用DECLARE_COMPLETION宏进行初始化：

    static DECLARE_COMPLETION(thread_done);

    （当完成结构变量在堆栈上时，还有一个DECLARE_COMPLETION_ONSTACK宏可供使用。）

    或者，可以定义未初始化的struct completion（例如作为动态分配结构的成员）并随后通过调用进行初始化init_completion(...)：

    struct completion thread_done;

...

init_completion(&thread_done);

  3. 创建线程

    kthread = kthread_run(thread_func, NULL, "kthread-test");
if (IS_ERR(kthread)) {
    complete(&thread_done); /* <-- may or may not be required */
    ret = PTR_ERR(kthread);
    return ret;
}

    在上面，如果kthread_run(...)失败，完成结构被标记为“完成”，以防某些代码稍后等待完成。如果保证以后不会等待完成，则complete(&thread_done);可以省略调用。

  4. 退出线程

    而不是从线程函数返回或调用do_exit(...)，线程应该调用complete_and_exit(...)以将线程标记为“完成”：

    complete_and_exit(&thread_done, 0);

    调用complete_and_exit(...)比单独调用complete(...)and更安全do_exit(...)。单独调用时complete(...)，do_exit(...)模块代码有可能在返回时已经被卸载complete(...)，因此线程可以执行不存在的或随机的代码。调用complete_and_exit(...)避免了这种情况的发生，因为该函数存在于模块代码之外并且永远不会返回。

  5. 确保线程已完成

    要确保线程已完成，请调用wait_for_completion(...)：

    wait_for_completion(&thread_done);

    返回时，线程要么已经退出，要么仍在调用中运行complete_and_exit(...)并即将退出。无论哪种情况，它都不再运行任何模块代码，因此可以安全地继续。

推荐阅读