| /*****************************************
*功能:利用进程链表遍历当前系统中的所有进程
*同时可以打印出进程的相关信息
*此程序可以将比较信息输出到指定的文件中
*
*author:孟阿龙（XiyouLinux）
*url:http://boyan.cublog.cn
* ***************************************/
# include < linux/ module. h>
# include < linux/ init. h>
# include < linux/ list . h>
# include < linux/ sched. h>
# include < linux/ time . h>
# include < linux/ fs. h>
# include < asm / uaccess. h>
# include < linux/ mm. h>
MODULE_AUTHOR( "Along" ) ;
MODULE_LICENSE( "GPL" ) ;
struct task_struct * task = NULL , * p = NULL ;
struct list_head * pos = NULL ;
struct timeval start, end;
int count = 0;
/*function_use表示使用哪一种方法测试，
* 0：三个方法同时使用,
* 1:list_for_each,
* 2:list_for_each_entry,
* 3:for_each_process
*/
int function_use = 0;
char * method;
char * filename= "testlog" ;
void print_message( void ) ;
void writefile( char * filename, char * data ) ;
void traversal_list_for_each( void ) ;
void traversal_list_for_each_entry( void ) ;
void traversal_for_each_process( void ) ;
static int init_module_list( void )
{
switch ( function_use) {
case 1:
traversal_list_for_each( ) ;
break ;
case 2:
traversal_list_for_each_entry( ) ;
break ;
case 3:
traversal_for_each_process( ) ;
break ;
default :
traversal_list_for_each( ) ;
traversal_list_for_each_entry( ) ;
traversal_for_each_process( ) ;
break ;
}
return 0;
}
static void exit_module_list( void )
{
printk( KERN_ALERT "GOOD BYE!!/n" ) ;
}
module_init( init_module_list ) ;
module_exit( exit_module_list ) ;
module_param( function_use, int , S_IRUGO) ;
void print_message( void )
{
char * str1 = "the method is: " ;
char * str2 = "系统当前共 " ;
char * str3 = " 个进程/n" ;
char * str4 = "开始时间: " ;
char * str5 = "/n结束时间: " ;
char * str6 = "/n时间间隔: " ;
char * str7 = "." ;
char * str8 = "ms" ;
char data[ 1024] ;
char tmp[ 50] ;
int cost;
printk( "系统当前共 %d 个进程!!/n" , count ) ;
printk( "the method is : %s/n" , method) ;
printk( "开始时间:%10i.%06i/n" , ( int ) start. tv_sec, ( int ) start. tv_usec) ;
printk( "结束时间:%10i.%06i/n" , ( int ) end. tv_sec, ( int ) end. tv_usec) ;
printk( "时间间隔:%10i/n" , ( int ) end. tv_usec- ( int ) start. tv_usec) ;
memset ( data, 0, sizeof ( data) ) ;
memset ( tmp, 0, sizeof ( tmp) ) ;
strcat ( data, str1) ;
strcat ( data, method) ;
strcat ( data, str2) ;
snprintf( tmp, sizeof ( count ) , "%d" , count ) ;
strcat ( data, tmp) ;
strcat ( data, str3) ;
strcat ( data, str4) ;
memset ( tmp, 0, sizeof ( tmp) ) ;
/*
* 下面这种转换秒的方法是错误的，因为sizeof最终得到的长度实际是Int类型的
* 长度，而实际的妙数有10位数字，所以最终存到tmp中的字符串也就只有三位
* 数字
* snprintf(tmp, sizeof((int)start.tv_sec),"%d",(int)start.tv_usec );
*/
/*取得开始时间的秒数和毫秒数*/
snprintf( tmp, 10, "%d" , ( int ) start. tv_sec ) ;
strcat ( data, tmp) ;
snprintf( tmp, sizeof ( str7) , "%s" , str7 ) ;
strcat ( data, tmp) ;
snprintf( tmp, 6, "%d" , ( int ) start. tv_usec ) ;
strcat ( data, tmp) ;
strcat ( data, str5) ;
/*取得结束时间的秒数和毫秒数*/
snprintf( tmp, 10, "%d" , ( int ) end. tv_sec ) ;
strcat ( data, tmp) ;
snprintf( tmp, sizeof ( str7) , "%s" , str7 ) ;
strcat ( data, tmp) ;
snprintf( tmp, 6, "%d" , ( int ) end. tv_usec ) ;
strcat ( data, tmp) ;
/*计算时间差，因为可以知道我们这个程序花费的时间是在
*毫秒级别的，所以计算时间差时我们就没有考虑秒，只是
*计算毫秒的差值
*/
strcat ( data, str6) ;
cost = ( int ) end. tv_usec- ( int ) start. tv_usec;
snprintf( tmp, sizeof ( cost) , "%d" , cost ) ;
strcat ( data, tmp) ;
strcat ( data, str8) ;
strcat ( data, "/n/n" ) ;
writefile( filename, data) ;
printk( "%d/n" , sizeof ( data) ) ;
}
void writefile( char * filename, char * data )
{
struct file * filp;
mm_segment_t fs;
filp = filp_open( filename, O_RDWR| O_APPEND| O_CREAT, 0644) ; ;
if ( IS_ERR( filp) ) {
printk( "open file error.../n" ) ;
return ;
}
fs = get_fs( ) ;
set_fs( KERNEL_DS) ;
filp- > f_op- > write ( filp, data, strlen ( data) , & filp- > f_pos) ;
set_fs( fs) ;
filp_close( filp, NULL ) ;
}
void traversal_list_for_each( void )
{
task = & init_task;
count = 0;
method= "list_for_each/n" ;
do_gettimeofday( & start) ;
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 ) ;
}
do_gettimeofday( & end) ;
print_message( ) ;
}
void traversal_list_for_each_entry( void )
{
task = & init_task;
count = 0;
method= "list_for_each_entry/n" ;
do_gettimeofday( & start) ;
list_for_each_entry( p, & task- > tasks, tasks ) {
count + + ;
printk( KERN_ALERT "%d/t%s/n" , p- > pid, p- > comm ) ;
}
do_gettimeofday( & end) ;
print_message( ) ;
}
void traversal_for_each_process( void )
{
count = 0;
method= "for_each_process/n" ;
do_gettimeofday( & start) ;
for_each_process( task) {
count + + ;
printk( KERN_ALERT "%d/t%s/n" , task- > pid, task- > comm ) ;
}
do_gettimeofday( & end) ;
print_message( ) ;
}
|
