服务器之家：专注于VPS、云服务器配置技术及软件下载分享
分类导航

PHP教程|ASP.NET教程|Java教程|ASP教程|编程技术|正则表达式|C/C++|IOS|C#|Swift|Android|VB|R语言|JavaScript|易语言|vb.net|

服务器之家 - 编程语言 - C/C++ - C/C++实现内存泄漏检测详解

C/C++实现内存泄漏检测详解

2023-02-28 14:27Ersansui C/C++

这篇文章主要为大家详细介绍了c++进行内存泄漏检测的方法，帮助大家更好的理解和学习使用c++，感兴趣的朋友可以了解下,希望能够给你带来帮助

内存泄漏的两个问题

  • 是否有内存泄漏？
  • 内存泄漏是在代码的哪一行？

检测内存泄漏主要从上面两个问题入手。

 

使用宏定义覆盖 malloc 和 free 函数

在单个文件中，可以使用这种方法检测是否有内存泄漏。

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

int main() {

  void *p1 = malloc(10);
  void *p2 = malloc(20);

  free(p1);

}

上面的代码存在内存泄漏的问题，但是直接运行看不出问题。我们可以使用宏定义覆盖覆盖系统的 malloc 和 free 函数。

我们定义两个函数，分别是 my_malloc 和 my_free ，使用宏定义覆盖原来的 malloc 和 free 。

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

void *my_malloc(size_t size, const char *file, int line) {

  void *p = malloc(size);
  printf("malloc[+]: addr: %p , size: %ld , file: %s , line: %d \n", p, size, file, line);
  return p;

}

void my_free(void *p, const char *file, int line) {

  free(p);
  printf("free[-]: addr: %p\n", p);


}

#define malloc(size)    my_malloc(size, __FILE__, __LINE__)
#define free(p)         my_free(p, __FILE__, __LINE__)

int main() {

  void *p1 = malloc(10);
  void *p2 = malloc(20);

  free(p1);

}

C/C++实现内存泄漏检测详解

运行结果如上图，我们可以看到哪一行代码调用了 malloc 以及对应的内存有没有 free 。

还可以把上面的代码改进一下，把信息写入到文件中，方面查看。

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

#if 1

void *my_malloc(size_t size, const char *file, int line) {

  void *p = malloc(size);
  
  char buff[128] = { 0 };
  sprintf(buff, "./mem/%p.mem", p); // 需要事前创建 mem 文件夹

  FILE *fp = fopen(buff, "w");
  fprintf(fp, "malloc[+%s:%d] --> addr:%p, size: %ld\n", file, line, p, size);
  fflush(fp);
  fclose(fp);

  return p;

}

void my_free(void *p, const char *file, int line) {

  char buff[128] = { 0 };
  sprintf(buff, "./mem/%p.mem", p); // 需要事前创建 mem 文件夹

  if (unlink(buff) < 0) {
      printf("double free addr:%p\n", p);
      return;
  }

  free(p);

}

#define malloc(size)    my_malloc(size, __FILE__, __LINE__)
#define free(p)         my_free(p, __FILE__, __LINE__)

#endif

int main() {

  void *p1 = malloc(10);
  void *p2 = malloc(20);

  free(p1);

}

运行后，结果如下，会在 mem 文件夹下存在一个文件，点开可以看到没有被 free 的内存的信息。

C/C++实现内存泄漏检测详解

C/C++实现内存泄漏检测详解

注意：

  • 这种方法局限性比较大，需要在每个文件开头展开这一段代码才可以。
  • 使用的第三方库无法覆盖
  • 同理可以覆盖 calloc 以及 realloc
  • #define malloc(size) my_malloc(size,__FILE__, __LINE__)
  • #define free my_free(p, __FILE__, __LINE__)
  • 这两行代码必须放在 my_malloc 和 my_free 的下面，否则会造成递归

 

使用 hook 钩子

dlsym 函数可以自己实现系统调用

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


#if 1

// 定义回调函数类型
typedef void *(*malloc_t)(size_t size);

// 定义回调函数变量
malloc_t malloc_f = NULL;

// 同理
typedef void (*free_t)(void *p);
free_t free_f = NULL;

void *malloc(size_t size) {

  printf("malloc [+%s:%d]\n", __FILE__, __LINE__);

}

void free(void *p) {

  printf("malloc [+%s:%d]\n", __FILE__, __LINE__);

}

void init_hook(void) {
  
  if (malloc_f == NULL)
      malloc_f = dlsym(RTLD_NEXT, "malloc"); // dlsym 需要加上编译条件 -ldl

  if (free_f == NULL)
      free_f = dlsym(RTLD_NEXT, "free");

}


#define DEBUG_MEM_LEAK init_hook();

#endif

int main() {

  DEBUG_MEM_LEAK

  void *p1 = malloc(10);
  void *p2 = malloc(20);

  free(p1);

}

上面这一份代码，在自己实现的 malloc 和 free 内部使用了 printf 函数，会出现错误。

是因为在 printf 函数内部也是用了 malloc 来开辟缓冲区，就会造成递归调用。

C/C++实现内存泄漏检测详解

我们需要定义一个 flag 来防止递归的出现。

#define _GNU_SOURCE
#include <dlfcn.h>

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


#if 1

// 定义回调函数类型
typedef void *(*malloc_t)(size_t size);

// 定义回调函数变量
malloc_t malloc_f = NULL;

// 同理
typedef void (*free_t)(void *p);
free_t free_f = NULL;

// flag 防止递归
int enable_malloc_hook = 1;
int enable_free_hook = 1;

void *malloc(size_t size) {

  if (enable_malloc_hook == 1) {
      enable_malloc_hook = 0;
      printf("malloc [+%s:%d]\n", __FILE__, __LINE__);
      enable_malloc_hook = 1;
  }

}

void free(void *p) {

  if (enable_malloc_hook == 1) {
      enable_malloc_hook = 0;
      printf("free [+%s:%d]\n", __FILE__, __LINE__);
      enable_malloc_hook = 1;
  }



}

void init_hook(void) {
  
  if (malloc_f == NULL)
      malloc_f = dlsym(RTLD_NEXT, "malloc"); // dlsym 需要加上编译条件 -ldl

  if (free_f == NULL)
      free_f = dlsym(RTLD_NEXT, "free");

}


#define DEBUG_MEM_LEAK init_hook();

#endif

int main() {

  DEBUG_MEM_LEAK

  void *p1 = malloc(10);
  void *p2 = malloc(20);

  free(p1);

}

C/C++实现内存泄漏检测详解

这是改进的代码的运行结果，但是又出现了一个问题，申请内存的函数是在哪一行调用的，出现了问题，全是在同一行调用的，这显然是有问题的。

我们可以使用 __builtin_return_address() 这个函数来解决，传入 0 就返回上一层函数的信息，传入 1 就返回上两层函数的信息，以及类推。

#define _GNU_SOURCE
#include <dlfcn.h>

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


#if 1

// 定义回调函数类型
typedef void *(*malloc_t)(size_t size);

// 定义回调函数变量
malloc_t malloc_f = NULL;

// 同理
typedef void (*free_t)(void *p);
free_t free_f = NULL;

// flag 防止递归
int enable_malloc_hook = 1;
int enable_free_hook = 1;

void *malloc(size_t size) {

  if (enable_malloc_hook == 1) {
      enable_malloc_hook = 0;

      void *caller = __builtin_return_address(0); // 返回上一层调用函数的信息
      printf("malloc [+]: %p\n", caller);

      enable_malloc_hook = 1;
  }

}

void free(void *p) {

  if (enable_malloc_hook == 1) {
      enable_malloc_hook = 0;

      void *caller = __builtin_return_address(0); // 返回上一层调用函数的信息
      printf("free [-]: %p\n", caller);

      enable_malloc_hook = 1;
  }



}

void init_hook(void) {
  
  if (malloc_f == NULL)
      malloc_f = dlsym(RTLD_NEXT, "malloc"); // dlsym 需要加上编译条件 -ldl

  if (free_f == NULL)
      free_f = dlsym(RTLD_NEXT, "free");

}


#define DEBUG_MEM_LEAK init_hook();

#endif

int main() {

  DEBUG_MEM_LEAK

  void *p1 = malloc(10);
  void *p2 = malloc(20);

  free(p1);

}

运行结果如图，得到了一串地址，可以使用 addr2line 来查看具体信息。

C/C++实现内存泄漏检测详解

下面改成文件版本

#define _GNU_SOURCE
#include <dlfcn.h>

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


#if 1

// 定义回调函数类型
typedef void *(*malloc_t)(size_t size);

// 定义回调函数变量
malloc_t malloc_f = NULL;

// 同理
typedef void (*free_t)(void *p);
free_t free_f = NULL;

// flag 防止递归
int enable_malloc_hook = 1;
int enable_free_hook = 1;

void *malloc(size_t size) {

  if (enable_malloc_hook == 1) {
      enable_malloc_hook = 0;

      void *p = malloc_f(size);

      void *caller = __builtin_return_address(0); // 返回上一层调用函数的信息

      char buff[128] = { 0 };
      sprintf(buff, "./mem/%p.mem", p); // 需要事前创建 mem 文件夹

      FILE *fp = fopen(buff, "w");
      fprintf(fp, "malloc[+%p] --> addr:%p, size: %ld\n", caller, p, size);
      fflush(fp);
      fclose(fp);

      enable_malloc_hook = 1;
      return p;
  } else {
      return malloc_f(size);
  }

}

void free(void *p) {

  if (enable_malloc_hook == 1) {
      enable_malloc_hook = 0;

      // void *caller = __builtin_return_address(0); // 返回上一层调用函数的信息
      
      char buff[128] = { 0 };
      sprintf(buff, "./mem/%p.mem", p); // 需要事前创建 mem 文件夹

      if (unlink(buff) < 0) {
          printf("double free addr:%p\n", p);
          return;
      }

      free_f(p);

      enable_malloc_hook = 1;
  } else {
      free_f(p);
  }



}

void init_hook(void) {
  
  if (malloc_f == NULL)
      malloc_f = dlsym(RTLD_NEXT, "malloc"); // dlsym 需要加上编译条件 -ldl

  if (free_f == NULL)
      free_f = dlsym(RTLD_NEXT, "free");

}


#define DEBUG_MEM_LEAK init_hook();

#endif

int main() {

  DEBUG_MEM_LEAK

  void *p1 = malloc(10);
  void *p2 = malloc(20);

  free(p1);

}

到此这篇关于C/C++实现内存泄漏检测详解的文章就介绍到这了,更多相关C++内存泄漏检测内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家！

原文链接：https://blog.csdn.net/weixin_61021362/article/details/128861266

  • C++
  • 内存泄漏
  • C语言

    • 延伸 · 阅读

    精彩推荐
    • C/C++C++实现单词管理系统

      C++实现单词管理系统

      这篇文章主要为大家详细介绍了C++实现单词管理系统，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下...

      一条菜鸟鱼9012022-10-18
    • C/C++C语言非递归后序遍历二叉树

      C语言非递归后序遍历二叉树

      这篇文章主要为大家详细介绍了C语言非递归后序遍历二叉树，具有一定的参考价值，感兴趣的小伙伴们可以参考一下...

      数星星的咚咚咚11522021-06-07
    • C/C++关于C语言strlen与sizeof区别详情

      关于C语言strlen与sizeof区别详情

      对于 strlen 和 sizeof，相信不少程序员会混淆其功能。虽然从表面上看它们都可以求字符串的长度，但二者却存在着许多不同之处及本质区别，今天得这篇文...

      芒果在努力5662022-01-21
    • C/C++C语言单链表实现方法详解

      C语言单链表实现方法详解

      这篇文章主要介绍了C语言单链表实现方法,结合实例形式分析了基于C语言的单链表定义、创建、添加、删除、排序、打印等操作技巧,并附带了相关的优化算...

      Tom文星10322021-06-24
    • C/C++C++中的类的大小详解

      C++中的类的大小详解

      这篇文章主要为大家详细介绍了C++中的类的大小，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带...

      想做后端的前端3892022-10-31
    • C/C++C++ 编写DLL文件给易语言调用方法

      C++ 编写DLL文件给易语言调用方法

      在本文中我们给大家分享了C++ 编写DLL文件给易语言调用的代码和方法，需要的朋友们学习下。...

      C++教程网5282021-07-16
    • C/C++标准C++类string的Copy-On-Write技术

      标准C++类string的Copy-On-Write技术

      这里，我想从C++类或是设计模式的角度为各位揭开Copy-On-Write技术在string中实现的面纱，以供各位在用C++进行类库设计时做一点参考...

      C++教程网9682021-01-08
    • C/C++VS2010 boost标准库开发环境安装教程

      VS2010 boost标准库开发环境安装教程

      这篇文章主要为大家详细介绍了VS2010 boost标准库开发环境的安装教程，具有一定的参考价值，感兴趣的小伙伴们可以参考一下...

      尹成4432021-05-08
    最近更新
    编辑推荐
    阅读排行
    热门标签
    1015 　 奇偶数位交换 　 地址参数 　 多继承法 　 ZOJ 　 Fishing 　 同名 　 libfreetype 　 Simulink 　 数据滚动 　 右值引用 　 NSString 　 消息传递 　 平滑处理 　 Visual C++ 　 栈区 　 堆区 　 C语言程序设计 　 内存错误 　 对象模型 　 颜色加亮 　 程序内部 　 strcat 　 WinExec 　 打开程序 　 错误调试 　 C++教程 　 tinyxml 　 XML解析器 　 静态存储区 　
    © 2019-2023 服务器之家(www.tuohang.net) 版权所有关于我们联系我们版权申明网站地图