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

      编写内核模块实现进程的遍历。这个有点类似ps命令了。

  源程序:list.c

  /*****************************************

  *功能:利用进程链表遍历当前系统中的所有进程

  *同时可以打印出进程的相关信息

  *

  * ***************************************/

  #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_AUTHOR( "Along" );

  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 的值查看不同宏运行产生的结果。

posted @ 2010-03-28 22:23  与时俱进  阅读(528)  评论(0编辑  收藏  举报
友情链接:同里老宅院民居客栈