linux 内核定时器打印进程信息

linux内核模块
作用: 通过定时器,在超过10秒就打印进程内存排名前十的进程信息

test.c

#include "linux/init.h"
#include "linux/kernel.h"
#include "linux/module.h"  //3个头文件包含了内核提供的所有内核模块的头文件
#include "linux/timer.h"
#include "linux/rtc.h"
#include "linux/time.h"
#include "linux/sched.h"
#define K 10

struct top_process{
        char comm[10];        
        int pid;
        int state;
        long total_vm;
};

struct top_process topK[K];

static void process_copy(struct top_process *dst,struct top_process *src) {
        strcpy(dst->comm,src->comm);
        dst->pid=src->pid;
        dst->state=src->state;
        dst->total_vm=src->total_vm;
}

void sort(struct top_process top[],int k,struct top_process t) {
        int i;
        for(i=0;i<k;i++) {
                if(top[i].total_vm<t.total_vm) {
                        process_copy(&top[i-1],&top[i]);
                }else{
                        process_copy(&top[i-1],&t);
                        break;
                }
        }
        if(i==k) {
                process_copy(&top[i-1],&t);
        }
}

void printTopK(unsigned long data) {
        int i;
        struct task_struct *p;
        struct top_process tmp;
        for_each_process(p) {
                if(p->mm!=NULL) {
                        if(p->mm->total_vm>topK[0].total_vm) {
                                strcpy(tmp.comm,p->comm);
                                tmp.pid=p->pid;
                                tmp.state=p->state;
                                tmp.total_vm=p->mm->total_vm;
                                sort(topK,K,tmp);
                             }
                }
        }
        printk("top %d process using :\n",K);
        for(i=K-1;i>=0;i--) {
                printk("name:%s   pid:%d  state:%d   mem:%ld \n",topK[i].comm,topK[i].pid,topK[i].state,topK[i].total_vm);
        }
}

struct timer_list timer;

static int __init hello_init(void) {
        int i;
        for(i=0;i<K;i++) {
                topK[i].total_vm=i;
        }
        init_timer(&timer);
        timer.expires=jiffies+10*HZ;
        timer.data=0;
        timer.function=printTopK;
        add_timer(&timer);
        
        return 0;
}

static void __exit hello_exit(void) {
        printk(KERN_ALERT "Goodbye!\n");
}

module_init(hello_init);  //告诉内核当内核模块第一次运行时哪个函数将被执行
module_exit(hello_exit); //内核模块被卸载时被执行的函数
MODULE_LICENSE("GPL");  //开源license
MODULE_DESCRIPTION("hello"); //文件描述

Makeile

obj-m:=test.o  #指定将要编译的内核模块列表
KERNELBUILD :=/lib/modules/$(shell uname -r)/build #内核源代码位置，这里是标准情况下链接到正在使用的内核
default: 
        make -C $(KERNELBUILD) M=$(shell pwd) modules #编译连接目标
#  以下内容不是必须 ，属于执行和清理部分
        echo insmod ./hello.ko to turn it on 
clean:
        rm -rf *.o *.ko *.mod.c .*.cmd *.markers *.order *.symvers .tmp_versions

编译模块：
make

插入模块:
sudo insmod test.ko

查看信息:
dmesg