IOSMallocZone.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "HAL/MemoryBase.h"
 
typedef struct _malloc_zone_t malloc_zone_t;
struct FIOSCrashContext;
 
/*
 * A malloc interface using a unique malloc zone
 */
class FIOSMallocZone : public FMalloc
{
public:
    FIOSMallocZone( uint64 const InitialSize );
    virtual ~FIOSMallocZone();
    
    // FMalloc interface.
    virtual void* Malloc( SIZE_T Size, uint32 Alignment ) override;
    
    virtual void* Realloc( void* Ptr, SIZE_T NewSize, uint32 Alignment ) override;
    
    virtual void Free( void* Ptr ) override;
    
    virtual bool GetAllocationSize(void *Original, SIZE_T &SizeOut) override;
    
    virtual bool IsInternallyThreadSafe() const override;
    
    virtual bool ValidateHeap() override;
    
    virtual const TCHAR* GetDescriptiveName() override;
    
protected:
    malloc_zone_t* MemoryZone;
};
 
/*
 * Specific FIOSMallocZone for the crash handler malloc override so that we avoid malloc reentrancy problems
 */
class FIOSMallocCrashHandler : public FIOSMallocZone
{
public:
    FIOSMallocCrashHandler( uint64 const InitialSize );
    
    virtual ~FIOSMallocCrashHandler();
    
    void Enable( FIOSCrashContext* Context, uint32 CrashedThreadId );
    
    void SetContext( FIOSCrashContext* Context );
    
    // FMalloc interface.
    virtual void* Malloc( SIZE_T Size, uint32 Alignment ) override;
    
    virtual void* Realloc( void* Ptr, SIZE_T NewSize, uint32 Alignment ) override;
    
    virtual void Free( void* Ptr ) override;
    
    virtual bool GetAllocationSize( void *Original, SIZE_T &SizeOut ) override;
    
    virtual const TCHAR* GetDescriptiveName() override;
 
private:    
    bool IsOnCrashedThread(void);
    
private:
    FMalloc* OriginalHeap;
    FIOSCrashContext* CrashContext;
    int32 ThreadId;
};