C语言实现内存泄漏检测模块

最新推荐文章于 2023-07-26 14:58:26 发布

原创最新推荐文章于 2023-07-26 14:58:26 发布 · 418 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#c语言 #内存泄漏

C/C++ 专栏收录该内容

19 篇文章

订阅专栏

博客介绍了C语言内存泄漏检测模块的实现及测试例程，同时强调注意事项，若在多线程中使用该检测模块，要保证g_record[SIZE]互斥使用，即进行互斥操作。

实现检测模块

/*文件 mleak.h*/
#ifndef _MLEAK_H_
#define _MLEAK_H_

#include <stdio.h>
#include <malloc.h>

#define MALLOC(n) my_malloc(n,__FILE__,__LINE__)
#define FREE(p) my_free(p)

void * my_malloc(size_t n,const char* file,const int line);
void my_free(void *p);
void PRINT_LEAK_INFO();

#endif


/*文件 mleak.c*/

#include "mleak.h"

#define SIZE 256

typedef struct
{
    void * pointer;
    int size;
    const char* file;
    int line;
}MY_ITEM;

static MY_ITEM g_record[SIZE];//如果存在多线程的情况一定要注意互斥操作！！！

void * my_malloc(size_t n,const char* file,const int line)
{
    void * ret = malloc(n);
    if (ret != NULL)
    {
        int i = 0;
        for (i = 0; i < SIZE; i++)
        {
            if (g_record[i].pointer == NULL)
            {
                g_record[i].pointer = ret;
                g_record[i].size = n;
                g_record[i].file = file;
                g_record[i].line = line;
                break;
            }
        }
        
    }
    return ret;
}
void my_free(void *p)
{
   if (p != NULL)
    {
        int i = 0;
        for (i = 0; i < SIZE; i++)
        {
            if (g_record[i].pointer == p)
            {
                g_record[i].pointer = NULL;
                g_record[i].size = 0;
                g_record[i].file = NULL;
                g_record[i].line = 0;

                free(p);

                break;
            }
        }
        
    }
}
void PRINT_LEAK_INFO()
{
    int i = 0;
    printf("Potential Memory Leak Info:\n");

    for (i = 0; i < SIZE; i++)
    {
        if (g_record[i].pointer != NULL)
        {
            printf("Address:%p,size:%d,Location:%s:%d\n",
                  g_record[i].pointer,g_record[i].size,g_record[i].file,g_record[i].line);
        }
        
    }
    
}

测试例程

#include "mleak.h"


int main()
{
    
    int *test_p = MALLOC(sizeof(int)*4);
    PRINT_LEAK_INFO();

    return 0;
}

注意事项：如果在多线程用使用，一定要保证g_record[SIZE]互斥使用，即保证使用自定义检测模块时要保证互斥操作