编写模块遍历系统中的进程

来源：互联网发布：常用有限元分析软件编辑：程序博客网时间：2024/06/06 08:27

/*****************************************
*功能:利用进程链表遍历当前系统中的所有进程
*同时可以打印出进程的相关信息
*
* ***************************************/
#include <linux/list.h>
#include <linux/module.h>
#include <linux/init.h>
#include <linux/sched.h>
#define METHOD 2
static int list_init(void)
{
struct task_struct *task, *p;
struct list_head *pos;
int count;
char *method;
count = 0; /*下面这些初始化完全是为了消除编译时的警告信息*/
p = NULL;
task = NULL;
pos = NULL;
method = NULL;
task = &init_task;

printk(KERN_ALERT"PID\tCOMM\n");
switch(METHOD) {
  case 1:
  method="list_for_each";
  break;
  case 2:
  method="for_each_process";
  break;
  case 3:
  method="list_for_each_entry";
  break;
}
printk( "The method is %s\n", method );

#if METHOD == 1
list_for_each( pos, &task->tasks ) {
  p = list_entry( pos, struct task_struct, tasks );
  count++;
  printk( KERN_ALERT "%d\t%s\n", p->pid, p->comm );
}
#elif METHOD == 2
for_each_process(task) {
  count++;
  printk( KERN_ALERT "%d\t%s\n", task->pid, task->comm );
}
#elif METHOD == 3
list_for_each_entry( p, &task->tasks, tasks ) {
  count++;
  printk( KERN_ALERT "%d\t%s\n", p->pid, p->comm );
}
#endif
printk("系统当前共 %d 个进程!!", count);
return 0;
}
static void list_exit(void)
{
printk( KERN_ALERT "GOOD BYE!!\n");
}
module_init(list_init);
module_exit(list_exit);
MODULE_LICENSE("GPL");

Makefile文件
obj-m := list.o
KERNELDIR ?= /lib/modules/$(shell uname -r)/build
PWD := $(shell pwd)
all:
make -C $(KERNELDIR) M=$(PWD) modules
clean:
make -C $(KERNELDIR) M=$(PWD) clean

这个程序主要用到了几个关键的宏list_for_each,list_for_each_entry,for_each_process 。我们可以通过修改程序中 METHOD 的值查看不同宏运行产生的结果。