Linux内核模块编程(打印所有内核线程的程序名、PID号,进程状态,及进程优先级,父进程的PID)
转自:https://blog.****.net/SweeNeil/article/details/87646244
1. 设计一个模块,要求列出系统中所有内核线程的程序名、PID 号、进程状态及进程优先级、父进程的PID。
- 首先,我们开始编写模块代码,这是Linux内核编程的核心代码,代码如下:
#include <linux/init.h>
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/sched.h>
#include <linux/init_task.h>
// 初始化函数
static int hello_init(void)
{
struct task_struct *p;
p = NULL;
p = &init_task;
printk(KERN_ALERT"名称\t进程号\t状态\t优先级\t父进程号\t");
for_each_process(p)
{
if(p->mm == NULL){ //内核线程的mm成员为空
printk(KERN_ALERT"%s\t%d\t%ld\t%d\n",p->comm,p->pid, p->state,p->normal_prio,p->parent->pid);
}
}
return 0;
}
// 清理函数
static void hello_exit(void)
{
printk(KERN_ALERT"goodbye!\n");
}
// 函数注册
module_init(hello_init);
module_exit(hello_exit);
// 模块许可申明
MODULE_LICENSE("GPL");
-
头文件的声明,module.h包含了大量加载模块所需要的函数和符号的定义;init.h包含了模块初始化和清理函数的定义。如果模块加载时允许用户传递参数,模块还应该包含moduleparam.h头文件。
-
接下来我们需要初始化函数,task_struct 结构定义在/usr/src/linux4.4.19/include/linux/sched.h 中,所以也需要在头文件加入这个库,而这个结构也包含了我们需要使用的PID之类的值,可以直接调用。当然了,这两篇文章可以更好理解这个结构:
https://blog.****.net/npy_lp/article/details/7292563
https://blog.****.net/npy_lp/article/details/7335187
接下来,我们需要对我们定义的结构p初始化指针,否则会产生也指针,文件也会报错,因此,我们需要用到init_task,并且在头文件加入init_task.h库。
-
然后,我们用for_each_process遍历我们的每一个进程,并输出我们需要得到的进程信息。
-
最后是函数注册已经模块许可声明,这是必须加入的。
-
编译Makefile
obj-m:= 自己的c文件名.o
KERNELBUILD :=/lib/modules/$(shell uname -r)/build
default:
make -C $(KERNELBUILD) M=$(shell pwd) modules
clean:
rm -rf *.o *.ko *.mod.c .*.cmd *.markers *.order *.symvers .tmp_versions
或:
obj-m:=module1.o
KDIR:= /lib/modules/$(shell uname -r)/build
PWD:= $(shell pwd)
default:
$(MAKE) -C $(KDIR) M=$(PWD) modules
clean:
$(MAKE) -C $(KDIR) M=$(PWD) clean
-
make后使用insmod加入模块
-
dmesg显示使用情况,我的如图所示。
2.设计一个带参数的模块,其参数为某个进程的 PID 号,该模块的功能是列出该进程的家族信息,包括父进程、兄弟进程和子进程的程序名、PID 号。
代码如下:
#include<linux/init.h>
#include<linux/module.h>
#include<linux/kernel.h>
#include <linux/sched.h>
#include <linux/moduleparam.h>
static pid_t pid=1;
module_param(pid,int,0644);
static int hello_init(void)
{
struct task_struct *p;
struct list_head *pp;
struct task_struct *psibling;
// 当前进程的 PID
p = pid_task(find_vpid(pid), PIDTYPE_PID);
printk("me: %d %s\n", p->pid, p->comm);
// 父进程
if(p->parent == NULL) {
printk("No Parent\n");
}
else {
printk("Parent: %d %s\n", p->parent->pid, p->parent->comm);
}
// 兄弟进程
list_for_each(pp, &p->parent->children)
{
psibling = list_entry(pp, struct task_struct, sibling);
printk("sibling %d %s \n", psibling->pid, psibling->comm);
}
// 子进程
list_for_each(pp, &p->children)
{
psibling = list_entry(pp, struct task_struct, sibling);
printk("children %d %s \n", psibling->pid, psibling->comm);
}
return 0;
}
static void hello_exit(void)
{
printk(KERN_ALERT"goodbye!\n");
}
module_init(hello_init);
module_exit(hello_exit);
MODULE_LICENSE("GPL");
- 结果: