Linux平台下hook技术研究

说明

  • 之前公司,C项目中为了程序检测内存泄漏问题,采用一个自己实现的库,能够记录内存分配和释放操作,程序结束或者通过信号可以生成dump信息来分析内存问题。

技术分析

  • 内存分配和释放记录主要是通过hook(钩子)技术来获取的。

hook技术

通过预处理实现

  • 实现方式:
  1. 创建一个头文件,通过宏替换的方式替换掉内存分配(malloc,calloc,realloc)和释放函数(free)。
  2. 创建一个源文件,定义替换函数,在替换函数中记录调用和调用真正的目标函数和做相应处理。
  • 代码示例:
* 头文件 MemHook.h
#if defined(__cplusplus)
extern "C" {
#endif

#define malloc(size) mmalloc(size , __FILE__, __LINE__)
#define calloc(size) mcalloc(size, __FILE__, __LINE__)
.....
#define free(p) mfree(p)

#if defined(__cplusplus)
}
#endif

* 源码
#include <stdio.h>
#include <stdlib.h>
//不能引入 MemHook.h 

void *mmalloc (unsigned long counts, const char* const file , const unsigned long line)
{
	void *buf;
	
	printf("%s , %ld\n", file , line);
  	if (NULL == (buf = malloc(counts)))
	{
	    ....
		return NULL;
	}
	//记录内存分配
    ....
	return buf;
}
.....
void mfree (void *p)
{
	if (p)
	{
	    //记录内存释放
	    .....
		free(p);
	}
}
  • 方式优缺点:
  1. 需要手动引入该头文件,如果有源码未引入该头文件,会导致操作未记录。

LD_PRELOAD 环境变量方式

  • 方式简介:
  • LD_PRELOAD 是在所有动态库前加载该环境变量定义的动态库。
  • 方式说明:
  1. 实现一个动态库,通过LD_PRELOAD环境变量,在所有动态库前加载。
  2. 动态库中定义标准的内存分配和释放函数,例如:malloc,calloc,free等;由于先加载,变量设置后运行的程序中调用的这些函数都会被指向该库中的实现。
  3. 在该库中记录分配和释放记录,以及调用真正的malloc,calloc,free等函数。
  • 代码示例:
* 库源码
#define _GNU_SOURCE

#include <stdio.h>
#include <stdlib.h>
#include <dlfcn.h> 

static void *(*true_malloc)(size_t);
static void (*true_free)(void *ptr);

void *malloc(size_t size)
{
    void *ptr;
    
    if (true_malloc == NULL)
    {    
        true_malloc = dlsym(RTLD_NEXT, "malloc");
    }  
    
    printf(%d\n", size);  
    ptr = true_malloc(size);
    //记录分配操作
    ....
   
    return ptr;   
}
.....
void free(void* ptr)
{
    if (true_free == NULL)
    {    
        true_free = dlsym(RTLD_NEXT, "free");
    }    

    //记录释放操作
    ....
    printf("%p\n", ptr);     
    return true_free(ptr);   
}

* 注意:
1. 函数定义需要和标准的一样。
2. 生成动态库时需要链接 dl库。
* 编译生成动态库
* 设置环境变量 
export LD_PRELOAD=./libxxxx.so 
* 执行其它程序
  • 方式优缺点
  1. 替换方式简单,环境变量设置和取消非常方便。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值