pslkm.c
#include <linux/module.h>
#include <linux/init.h>
#include <linux/list.h>
#include <linux/sched.h>
#define METHOD 1
static int list_init(void)
{
struct task_struct *task, *p;
struct list_head *pos;
int count;
char *method;
count = 0;
p = NULL;
task = NULL;
method = NULL;
task = &init_task;
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);
printk(KERN_ALERT"PID\tCOMM\n");
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);
}
}
else if(METHOD == 2){
for_each_process(task){
count++;
printk(KERN_ALERT"%d\t%s\n",task->pid,task->comm);
}
}
else if(METHOD == 3){
list_for_each_entry(p,&task->tasks,tasks){
count++;
printk(KERN_ALERT "%d\t%s\n",p->pid,p->comm);
}
}
printk("系統當前共有 %d 個進程!\n",count);
return 0;
}
static void list_exit(void)
{
printk(KERN_ALERT "GOOD BYE!\n");
}
module_init(list_init);
module_exit(list_exit);
MODULE_AUTHOR("...");
MODULE_LICENSE("GPL");
Makefile:
obj-m := pslkm.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