#include <stdio.h>
#include "mleak.h"
 
void f()
 
{
    MALLOC(100);
}
 
int main()
{
    int* p = (int*)MALLOC(3 * sizeof(int));
 
    f();
 
    p[0] = 1;
    p[1] = 2;
    p[2] = 3;    
 
    FREE(p);    
 
    PRINT_LEAK_INFO();   
 
    return 0;
 
}

mleak.h：

#ifndef _MLEAK_H_
 
#define _MLEAK_H_
 
#include <malloc.h>
 
#define MALLOC(n) mallocEx(n, __FILE__, __LINE__)
 
#define FREE(p) freeEx(p)
 
 
void* mallocEx(size_t n, const char* file, const line);
 
void freeEx(void* p);
 
void PRINT_LEAK_INFO();
 
#endif

mleak.c：

#include "mleak.h"
 
#define SIZE 256
 
 
/* 動(dòng)態(tài)內(nèi)存申請參數(shù)結(jié)構(gòu)體 */
 
typedef struct
{
    void* pointer;
    int size;
    const char* file;
    int line;
 
} MItem;
 
static MItem g_record[SIZE]; /* 記錄動(dòng)態(tài)內(nèi)存申請的操作 */
 
void* mallocEx(size_t n, const char* file, const line)
{
    void* ret = malloc(n); /* 動(dòng)態(tài)內(nèi)存申請 */   
 
    if( ret != NULL )
    {
        int i = 0;       
 
        /* 遍歷全局?jǐn)?shù)組，記錄此次操作 */
 
        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 freeEx(void* p)
{
    if( p != NULL )
    {
        int i = 0;        
 
        /* 遍歷全局?jǐn)?shù)組，釋放內(nèi)存空間，并清除操作記錄 */
 
        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");    
 
    /* 遍歷全局?jǐn)?shù)組，打印未釋放的空間記錄 */
 
    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);
        }
    }
 
}

輸出結(jié)果如下，因?yàn)?MALLOC(100); 之后沒有進(jìn)行釋放內(nèi)存，所以被檢查出來了。

暫時(shí)不能用于工程開發(fā)，需要再開發(fā)才行。因?yàn)?malloc 往往在不同的線程中被調(diào)用，因此 malloc 函數(shù)必須要有互斥的操作。因?yàn)?static MItem g_record[SIZE]; 這個(gè)靜態(tài)全局?jǐn)?shù)組是一種臨界區(qū)，必須被保護(hù)起來。

三、關(guān)于 malloc(0)

malloc(0); 將返回什么?

下面看一段代碼：

#include <stdio.h>
#include <malloc.h>
 
int main()
{
    int* p = (int*) malloc(0);
    
    printf("p = %p\n", p);
    
    free(p);
 
    return 0;
}

輸出結(jié)果如下：

這說明 malloc(0) 是合法的，內(nèi)存地址其實(shí)包含兩個(gè)概念，一個(gè)是內(nèi)存的起始地址，一個(gè)是內(nèi)存的長度。在平常我們可能會(huì)只注意內(nèi)存的首地址，對于長度卻忽略了。malloc(0) 在這個(gè)程序中申請到的內(nèi)存起始地址為 0x82c3008，長度為 0。

但是我們在程序里不停寫 malloc(0)，會(huì)造成內(nèi)存泄漏嗎？答案是肯定的，因?yàn)閙alloc 實(shí)際分配的內(nèi)存可能會(huì)比請求的多，目前的操作系統(tǒng)一般都是 4 字節(jié)對齊的，所以寫 malloc(0) 系統(tǒng)實(shí)際返回的字節(jié)數(shù)也許就是 4 字節(jié)。

四、calloc 和 realloc

malloc 的同胞兄弟

void* calloc(size_t num, size_t size);

void* realloc(void* pointer, size_t new_size);

calloc 的參數(shù)代表所返回內(nèi)存的類型信息

calloc 會(huì)將返回的內(nèi)存初始化為 0

realloc 用于修改一個(gè)原先已經(jīng)分配的內(nèi)存塊大小

在使用 realloc 之后應(yīng)該使用其返回值
當(dāng) pointer 的第一個(gè)參數(shù)為 NULL 時(shí)，等價(jià)于 malloc

下面看一個(gè) calloc 和 realloc 的使用示例：

#include <stdio.h>
#include <malloc.h>
 
#define SIZE 5
 
int main()
{
    int i = 0;
    int* pI = (int*)malloc(SIZE * sizeof(int));
    short* pS = (short*)calloc(SIZE, sizeof(short));
    
    for(i = 0; i < SIZE; i++)
    {
        printf("pI[%d] = %d, pS[%d] = %d\n", i, pI[i], i, pS[i]);
    }
    
    printf("Before: pI = %p\n", pI);
    
    pI = (int*)realloc(pI, 2 * SIZE * sizeof(int));
    
    printf("After: pI = %p\n", pI);
    
    for(i = 0; i < 10; i++)
    {
        printf("pI[%d] = %d\n", i, pI[i]);
    }
    
    free(pI);
    free(pS);
    
    return 0;
}

輸出結(jié)果如下：