heaplog

#ifdef _DEBUG

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

#define _BUF_SIZE 128

static const char* g_FileName = "HeapLog.txt";
static const char* g_AllocateFail = "分配失败 --- 当前情况(分配:%d, 释放:%d)x0a";
static const char* g_AllocateOK = "分配成功(地址:%p, 大小:%d) --- 当前情况(分配:%d, 释放:%d)x0a";
static const char* g_FreeOK = "释放成功(地址:%p) --- 当前情况(分配:%d, 释放:%d)x0a";
static int g_AllocateNum = 0;
static int g_FreeNum = 0;
static char g_Buf[_BUF_SIZE];

void* operator new(size_t s)
{
    void* p = malloc(s);

    FILE* pF;
    fopen_s(&pF, g_FileName, "a");

    if (!p)
        sprintf_s(g_Buf, _BUF_SIZE, g_AllocateFail, g_AllocateNum, g_FreeNum);
    else
    {
        g_AllocateNum++;
        sprintf_s(g_Buf, _BUF_SIZE, g_AllocateOK, p, s, g_AllocateNum, g_FreeNum);
    }

    fwrite(g_Buf, sizeof(char), strlen(g_Buf), pF);
    fflush(pF);
    fclose(pF);
    return p;
}

void operator delete(void* p)
{
    free(p);
    g_FreeNum++;

    FILE* pF;
    fopen_s(&pF, g_FileName, "a");

    sprintf_s(g_Buf, _BUF_SIZE, g_FreeOK, p, g_AllocateNum, g_FreeNum);
    fwrite(g_Buf, sizeof(char), strlen(g_Buf), pF);
    fflush(pF);
    fclose(pF);
}

void* operator new[](size_t s)
{
    return operator new(s);
}

void operator delete[](void* p)
{
    operator delete(p);
}

#endif