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

int main()
{
  int i=10;
  int* p= NULL;
  for(i=0;i<100;i++)
  {
    //注意，malloc(0)會(huì)返回一個(gè)有效的內(nèi)存地址，大小為1
    //但我們不能依賴編譯器的這種行為來(lái)使用這個(gè)字節(jié)的空間！
    p = (int*)malloc(i);
    printf("%d ",_msize(p));//返回malloc出來(lái)的內(nèi)存空間大小
    free(p);
  }

  return 0;
}

#ifndef _MLEAK_H_
#define _MLEAK_H_

#include <stdio.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

#include "mleak.h"

#define SIZE 256

//動(dòng)態(tài)內(nèi)存申請(qǐng)參數(shù)結(jié)構(gòu)體
typedef struct
{
  void* pointer;//申請(qǐng)到的內(nèi)存地址
  int size;   //內(nèi)存塊大小
  const char* file; //文件名
  int line;     //文件行號(hào)
}MItem;

static MItem g_record[SIZE]; //記錄每個(gè)動(dòng)態(tài)內(nèi)存申請(qǐng)的操作

void* mallocEx(size_t n, const char* file, const line)
{
  int i = 0;
  void* ret = malloc(n);//動(dòng)態(tài)內(nèi)存申請(qǐng)
  
  if(ret != NULL)
  {
    //申請(qǐng)成功，則記錄下來(lái)
    
    //遍歷全局?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("Potenital Memory Leak Info:\n");

  //遍歷全局?jǐn)?shù)組，打印未釋放的空間的申請(qǐng)記錄
  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 <stdio.h>
#include "mleak.h"

void f()
{
  //沒(méi)釋放，會(huì)造成內(nèi)存泄漏！
  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;
}

/*
輸出結(jié)果：
E:\Study>gcc test.c mleak.c
E:\Study>a.exe
Potenital Memory Leak Info:
Address:00602ED8, size:100, Location:38-1.c:7
*/

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

#define SIZE  5

int main()
{
  int i = 0;
  int* pI = (int*)malloc(SIZE * sizeof(int)); //malloc內(nèi)存沒(méi)有初始化
  short* pS = (short*)calloc(SIZE, sizeof(short));//內(nèi)存初始化為0

  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); //重置內(nèi)存大小之前的pI指針

  pI = (int*)realloc(pI, 2 * SIZE * sizeof(int)); //內(nèi)存未初始化的

  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;
}