Linux内核多线程(四)

来源:互联网 发布:李代数 知乎 编辑:程序博客网 时间:2024/05/01 23:22

自己创建的内核线程,当把模块加载到内核之后,可以通过:ps –ef 命令来查看线程运行的情况。通过该命令可以看到该线程的pid和ppid等。也可以通过使用kill –s 9 pid 来杀死对应pid的线程。如果要支持kill命令自己创建的线程里面需要能接受kill信号。这里我们就来举一个例,支持kill命令,同时rmmod的时候也能杀死线程。

 

复制代码
#include <linux/kernel.h>#include <linux/module.h>#include <linux/init.h>#include <linux/param.h>#include <linux/jiffies.h>#include <asm/system.h>#include <asm/processor.h>#include <asm/signal.h>#include <linux/completion.h>       // for DECLARE_COMPLETION()#include <linux/sched.h>            #include <linux/delay.h>            // mdelay()#include <linux/kthread.h>  MODULE_LICENSE("GPL"); static DECLARE_COMPLETION(my_completion); static struct task_struct *task; int flag = 0; int my_fuction(void *arg){    printk(" in %s()\n", __FUNCTION__);    allow_signal(SIGKILL); //使得线程可以接收SIGKILL信号    mdelay(2000);    printk(" my_function complete()\n");    printk("should stop: %d\n",kthread_should_stop());    while (!signal_pending(current) && !kthread_should_stop()) {//使得线程可以可以被杀死,也可以再rmmod的时候结束        printk(" jiffies is %lu\n", jiffies);        set_current_state(TASK_INTERRUPTIBLE);        schedule_timeout(HZ * 5);            printk("should stop: %d\n",kthread_should_stop());    }    printk("Leaving my_function\n");    flag = 1; //flag很关键!    return 0;} static int __init init(void){    task = kthread_run(my_fuction,NULL,"my_function");    printk("<1> init wait_for_completion()\n");    return 0;} static void __exit finish(void){                int ret;        if(!flag)        {                 if (!IS_ERR(task))                 {                        int ret = kthread_stop(task);                        printk(KERN_INFO "First thread function has stopped ,return %d\n", ret);                   }                         }        printk("task_struct: 0x%x",task);    printk(" Goodbye\n");} module_init(init);module_exit(finish);
复制代码

运行结果(执行kill之后):

运行结果(rmmod之后):

说明:程序运行后线程循环执行每隔5个内核 ticks 就答应一次当前的jiffies值。可以通过kthread_stop()来结束,也可以通过kill命令来结束。

程序中使用了flag 变量来控制是否使用 kthread_stop()函数有两个原因:首先,当线程创建成功之后IS_ERR()不能检测出线程是否还在运行,因为此时task是一个正常的地址而不是错误码(后面会说明IS_ERR的原理);其次,线程不能被杀次两次,如果使用kill命令之后线程已经被杀死,但是在此使用kthread_stop()函数就会出严重问题,因为此时线程已经被杀死,task指向的地址已经无效,struct kthread 也已经不存在,操作此时使用kthread_stop()设置should_stop是没有意义的。同样可以得出结论,当线程结束之后使用kthread_should_stop()来查看线程运行状态也会造成内核异常。

 

IS_ERR()函数的原理:

#define IS_ERR_VALUE(x) unlikely((x) >= (unsigned long)-MAX_ERRNO)

static inline long IS_ERR(const void *ptr)
{
  return IS_ERR_VALUE((unsigned long)ptr);
}

内核中的函数常常返回指针,问题是如果出错,也希望能够通过返回的指针体现出来。
所幸的是,内核返回的指针一般是指向页面的边界(4K边界),即

ptr & 0xfff == 0

这样ptr的值不可能落在(0xfffff000,0xffffffff)之间,而一般内核的出错代码也是一个小负数,在-1000到0之间,转变成unsigned long,正好在(0xfffff000,0xffffffff)之间。因此可以用

(unsigned long)ptr > (unsigned long)-1000L

也就等效于(x) >= (unsigned long)-MAX_ERRNO
其中MAX_ERRNO 为4095

来判断内核函数的返回值是一个有效的指针,还是一个出错代码。

涉及到的任何一个指针,必然有三种情况,一种是有效指针,一种是NULL,空指针,一种是错误指针,或者说无效指针.而所谓的错误指针就是指其已经到达了 最后一个page.比如对于32bit的系统来说,内核空间最高地址0xffffffff,那么最后一个page就是指的 0xfffff000~0xffffffff(假设4k一个page).这段地址是被保留的,如果超过这个地址,则肯定是错误的。

0 0
原创粉丝点击