memorywatch.h

Go to the documentation of this file.

#pragma once

#define _CRTDBG_MAP_ALLOC

#include "memory.h"
#include <crtdbg.h>
#include <Windows.h>
#include <string>

#ifdef _DEBUG
    #define DBG_NEW new ( _NORMAL_BLOCK , __FILE__ , __LINE__ )
    // Replace _NORMAL_BLOCK with _CLIENT_BLOCK if you want the
    // allocations to be of _CLIENT_BLOCK type
#else
    #define DBG_NEW new
#endif

#ifdef DANGEROUS_NEW
    #ifdef _DEBUG
        // ReSharper disable once CppInconsistentNaming
        #define new DBG_NEW
    #endif
#endif

BEGIN_NAMESPACE_MEMORY

enum class MemoryWatchReport
{
    file,
    fileAndOutput,
    fileAndWindow,
    all
};

class MemoryWatch
{
public:
    using CurrentMemoryState = _CrtMemState;  

    static bool EnableMemoryLeakChecking(const std::string& reportFilePath = "",
                                         MemoryWatchReport leakReport = MemoryWatchReport::fileAndOutput);

    static void SetMemoryCheckpoint(_CrtMemState* state);

    static int CompareMemoryCheckpoints(_CrtMemState* difference,
                                        _CrtMemState const* oldState,
                                        _CrtMemState const* newState);

    static void DumpMemoryCheckpointsLog(_CrtMemState const* state);
private:
    static HANDLE CreateLogFile(const std::string& reportFilePath);
    static void SetDumpMemoryLeaksFlag();

    static void SetLeakReportType(HANDLE hLogFile, MemoryWatchReport leakReport);
    static void SetLeakReportDirection(int reportType, HANDLE hLogFile, int leakReport);

    static int ConvertReportTypeToInt(MemoryWatchReport leakReport);
    static std::wstring ConvertToLPWSTR(const std::string& inputString);
};

END_NAMESPACE_MEMORY