复现内存泄露的工具

 为了能尽快复现内存泄露的现象，写了一个小工具，可以分配指定的内存，以使得可以使用的内存减少。

该工具可以编译为内核模块或者独立的程序，依靠STANDALONE_MODE宏区分。

 

#ifdef STANDALONE_MODE #include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <string.h> #include <signal.h> #include <time.h> #else #include"linux/kernel.h" #include"linux/module.h" #include <linux/vmalloc.h> #define printf printk #define atoi(str) simple_strtoul(((str != NULL) ? str : ""), NULL, 0) #endif //#define MAX_AMOUNT_STR 128 #define DEFAULT_AMOUNT (1024)*(1024) /* default hold 1M bytes */ static int hold_amount = DEFAULT_AMOUNT; char * hold_ptr = NULL; struct unit_size { char unit; int size; }; static struct unit_size unit[] = { {'M', 1024*1024}, {'k', 1024} }; int find_unit(char unit_char) { int n,nr_unit; nr_unit = sizeof(unit)/sizeof(struct unit_size); for(n=0; n< nr_unit; n++) { if(unit_char == unit[n].unit) break; } if(n >= nr_unit) /* no found */ return -1; else return n; } int get_amount(const char * string) { char unit_char; int n, index, number; int len; number = atoi(string); if(number <= 0) return 0; len = strlen(string); for(n=0;n<len;n++) { if(string[n] < '0' || string[n] > '9') /* not digit */ break; } if(n >= len) /* no unit character */ return number; else unit_char = string[n]; index = find_unit(unit_char); if(index >=0){ return number *unit[index].size ; }else{ /* unit is error */ return -1; } } #ifdef STANDALONE_MODE static char* prog_name; static int daemonize = 1; static int loop_flag = 1; static int init_buf = 1; void show_version() { printf("holdmem v1.0/n"); printf("Holding amount of memory for test memory leak./n"); } void usage() { show_version(); printf("Usage: %s <options> amount /n", prog_name); printf("options:/n"); printf("/t-h:/tshow help./n"); printf("/t-s:/tstandalone other than daemon mode./n"); printf("/t-n:/tDO NOT copy random data to holded buffer./n"); printf("amount:/t memory size in byte, use 'M' or 'k' suffix to change unit./n"); } void sig_handler(int signal) { printf("holdmem: catch exit signal./n"); loop_flag= 0; } int main(int argc, char *argv[]) { int c; int n; prog_name = argv[0]; for (;;) { c = getopt(argc, argv, "nsh"); if (c < 0) break; switch (c) { case 'n': init_buf = 0; break; case 'h': usage(); return 0; case 's': /* standalone */ daemonize = 0; break; default: usage(); return 0; } } if (optind != argc){ hold_amount = get_amount(argv[optind]); if(hold_amount <= 0) { printf("holdmem: invalid memory amount: %s ./n", argv[optind]); return -1; } } signal(SIGINT, sig_handler); /* capture CTRL + C */ signal(SIGTERM, sig_handler); /* capture signal */ printf("/nholdmem: holding %d bytes memory.../n", hold_amount); hold_ptr = malloc(hold_amount); if(!hold_ptr){ printf("holdmem: memory alloc failure./n"); return -1; } /* put random value to buffer */ if(init_buf){ srandom(time(NULL)); for(n=0; n<hold_amount;n++){ hold_ptr[n] = (char)random(); } } printf("holdmem: %d bytes memory holded./n", hold_amount); if (daemonize && daemon(1, 1)) { perror("daemon"); goto out; } /* loop forever, use ctrl + c to exit */ while(loop_flag){ sleep(3); } out: free(hold_ptr); printf("holdmem: free %d bytes memory./n", hold_amount); return 0; } #else /* not STANDALONE_MODE */ static char *amount=NULL; static int method=2; static int __init holdmem_init(void) { if (amount){ hold_amount = get_amount(amount); if(hold_amount <= 0) { printf("holdmem: invalid memory amount: %s ./n", amount); return -1; } } printf("/nholdmem: holding %d bytes memory.../n", hold_amount); if(method == 1){ hold_ptr = kmalloc(hold_amount, GFP_KERNEL); }else{ hold_ptr = vmalloc(hold_amount); } if(!hold_ptr){ printf("holdmem: memory alloc failure./n"); return -1; } printf("holdmem: %d bytes memory holded./n", hold_amount); memset(hold_ptr, 0, hold_amount); /* use it */ return 0; } static void __exit holdmem_exit(void) { if(method == 1){ kfree(hold_ptr); }else{ vfree(hold_ptr); } printf("holdmem: free %d bytes memory./n", hold_amount); } module_init(holdmem_init); module_exit(holdmem_exit); module_param(amount, charp, S_IRWXUGO); MODULE_PARM_DESC(amount, "Memory amount to hold in byte, use 'M' or 'k' suffix to change unit"); module_param(method, int, S_IRWXUGO); MODULE_PARM_DESC(method, "Memory allocate method, 1=kmalloc, 2=vmalloc"); MODULE_AUTHOR("Hansel He <hansel@163.com>"); MODULE_DESCRIPTION("Hold amount of memory in kernel. Use it to analysis kernel memory leak."); MODULE_LICENSE("GPL"); //General Public License #endif /* STANDALONE_MODE */

 

Makefile：

PROG=holdmem SOURCE=holdmem.c obj-m:=holdmem.o KERNELDIR:=/home/hansel/linux-2.6.10 CROSS_COMPILE=mipsel-linux-uclibc- CC:=$(CROSS_COMPILE)gcc CFLAGS+=-Wall MY_CFLAGS=-DSTANDALONE_MODE #KFLAGS=ARCH=mips CROSS_COMPILE=$(CROSS_COMPILE) KBUILD_OUTPUT= DEPMOD=/bin/true all: module standalone standalone: ${PROG} ${PROG}:holdmem_s.o @echo "building standalone program..." $(CC) -o $@ $< holdmem_s.o:${SOURCE} $(CC) $(CFLAGS) $(MY_CFLAGS) -c -o $@ $< module: @echo "building kernel module..." make $(KFLAGS) -C $(KERNELDIR) M=$(PWD) modules clean: rm -rf *.o *~ core .depend .*.cmd *.ko *.mod.c .tmp_versions Module.symvers ${PROG}  

 

该Makefile是为MIPS   CPU的交叉编译写到，如果是别的平台，需要做修改。

 

使用方法：

 

内核模块的使用方法：

insmod holdmem.ko amount=xxx method=n

 

amount: Memory amount to hold in byte, use 'M' or 'k' suffix to change unit. Default is 1MB.

method: Memory allocate method, 1=kmalloc, 2=vmalloc. Default is 2

 

例如：

insmod holdmem.ko amount=2M

 

用户空间程序的使用方法：

 

$ ./holdmem -h
holdmem v1.0
Holding amount of memory for test memory leak.
Usage: /phone/holdmem <options> amount
options:
        -h:     show help.
        -s:     standalone other than daemon mode.
        -n:     DO NOT copy random data to holded buffer.
amount:  memory size in byte, use 'M' or 'k' suffix to change unit.

 

amount缺省是1MB

 

例如：

heldmem 2M