自测试MemLeak

来源:互联网 发布:黑胡桃沙发价格 知乎 编辑:程序博客网 时间:2024/06/05 21:55

//汇总头文件 在各个文件之外的头文件! 包含了每个子文件的头文件

#ifdef __cplusplus
}
#endif


#endif


extern void *testMalloc(size_t size);
extern void testFree(void *pMem);
#if 0
#define malloc(n) \
    testMalloc(n);\
    REP_DEBUG("Malloc size:%d!\r\n", n)
#define free(p) \
    testFree(p);\
    REP_DEBUG("Free!\r\n")
#endif


#ifndef REP
#define Close(fd) if((fd == 0) || (fd == 1) || (fd == 2)){REP_DEBUG("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!closing %d!!!!!!!!!!!!!!\r\n",fd);sleep(1);}else{close(
fd);}
#else
#define Close(fd) close(fd)
#endif




.c文件中

int g_iMem = 0;

void *g_apMem[512] = {NULL};
void mem()
{
    int i;
    printf("%d:\r\n", g_iMem);
    for(i=0;i<g_iMem;i++)
    {   
        printf("%p\r\n", g_apMem[i]);
    }   
}
void *testMalloc(size_t size)
{
    void *pMem = NULL;
#ifdef malloc
#undef malloc
#endif
    pMem = malloc(size);
#define malloc testMalloc


    printf("Malloc returns %p!\r\n", pMem);
    g_apMem[g_iMem] = pMem;
    g_iMem++;
    return pMem;
}


void testFree(void *pMem)
{
    int i;
    void *tmp;


    for(i=0;i<g_iMem;i++)
    {
        if(g_apMem[i] == pMem)
        {
            tmp = g_apMem[i];
            g_apMem[i] = g_apMem[g_iMem-1];
            g_apMem[g_iMem-1] = tmp;
            break;
        }
    }


    g_iMem--;


    printf("Free %p!\r\n", pMem);
#ifdef free
#undef free
#endif
        free(pMem);
#define free testFree


    return;
}
原创粉丝点击