MallocDoubleFreeFinder.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Containers/Map.h"
#include "HAL/MallocCallstackHandler.h"
#include "HAL/Platform.h"
 
class FMalloc;
class FOutputDevice;
class UWorld;
 
class FMallocDoubleFreeFinder final : public FMallocCallstackHandler
{
public:
    CORE_API FMallocDoubleFreeFinder(FMalloc* InMalloc);
 
#if UE_ALLOW_EXEC_COMMANDS
    CORE_API virtual bool Exec(UWorld* InWorld, const TCHAR* Cmd, FOutputDevice& Ar) override;
#endif
 
    CORE_API void TrackSpecial(void* Ptr);
 
    CORE_API virtual void Init();
 
    static CORE_API FMalloc* OverrideIfEnabled(FMalloc*InUsedAlloc);
 
protected:
    CORE_API virtual void TrackMalloc(void* Ptr, uint32 Size, int32 CallStackIndex);
    CORE_API virtual void TrackFree(void* Ptr, uint32 OldSize, int32 CallStackIndex);
 
    struct TrackedAllocationData
    {
        SIZE_T Size;
        int32 CallStackIndex;
        TrackedAllocationData() :
            Size(0),
            CallStackIndex(-1)
        {
        };
        TrackedAllocationData(SIZE_T InRequestedSize, int32 InCallStackIndex)
        {
            Size = InRequestedSize;
            CallStackIndex = InCallStackIndex;
        };
        ~TrackedAllocationData()
        {
            Size = 0;
            CallStackIndex = -1;
        };
    };
    TMap<const void* const, TrackedAllocationData> TrackedCurrentAllocations;   // Pointer as a key to a call stack for all the current allocations we known about
    TMap<const void* const, TrackedAllocationData> TrackedFreeAllocations;      // Pointer as a key to a call stack for all allocations that have been freed
};
 
extern CORE_API FMallocDoubleFreeFinder* GMallocDoubleFreeFinder;
extern CORE_API bool GMallocDoubleFreeFinderEnabled;