AuroraRuntime/Source/Memory/Memory.cpp

/***
    Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.

    File: Memory.cpp
    Date: 2021-6-12
    Author: Reece
***/
#include <Source/RuntimeInternal.hpp>
#include "Memory.hpp"
#include <mimalloc.h>
#include <Source/HWInfo/AuHWInfo.hpp>

#if defined(AURORA_IS_LINUX_DERIVED)
    #include <sys/mman.h>
#endif

#include <Source/Debug/MemoryCrunch.hpp>

namespace Aurora::Memory
{
    AUKN_SYM AuUInt GetChunkSize(const void *head)
    {
        if (AuDebug::IsPointerReserveRange((void *)head))
        {
            return AuDebug::gReserveHeap->GetChunkSize(head);
        }
        else
        {
            return ::mi_malloc_size(head);
        }
    }

    #define CHECK_WRAP_RETURN(exp, exp2, string)                                    \
        auto pRet = exp;                                                            \
        if (!pRet)                                                                  \
        {                                                                           \
            if (AuDebug::IsTlsMemoryCrunchActive() && AuDebug::gReserveHeap)        \
            {                                                                       \
                return AuDebug::gReserveHeap-> AU_WHAT exp2;                        \
            }                                                                       \
            else if (gRuntimeConfig.debug.bIsMemoryErrorFatal)                      \
            {                                                                       \
                SysPanic(string);                                                   \
            }                                                                       \
        }                                                                           \
        return pRet;

    AUKN_SYM void *_ZAlloc(Types::size_t length)
    {
        CHECK_WRAP_RETURN(::mi_zalloc(length), (ZAlloc(length)), "ZAlloc out of memory");
    }

    AUKN_SYM void *_ZAlloc(Types::size_t length, Types::size_t align)
    {
        CHECK_WRAP_RETURN(::mi_zalloc_aligned(length, align), (ZAlloc(length, align)), "ZAlloc out of memory");
    }

    AUKN_SYM void *_FAlloc(Types::size_t length)
    {
        CHECK_WRAP_RETURN(::mi_malloc(length), (FAlloc(length)), "FAlloc out of memory");
    }

    AUKN_SYM void *_FAlloc(Types::size_t length, Types::size_t align)
    {
        CHECK_WRAP_RETURN(::mi_malloc_aligned(length, align), (FAlloc(length, align)), "FAllocEx out of memory");
    }

    AUKN_SYM void *_ZRealloc(void *buffer, Types::size_t length, Types::size_t align)
    {
        if (AuDebug::IsPointerReserveRange(buffer))
        {
            return AuDebug::gReserveHeap->ZRealloc(buffer, length, align);
        }
        else
        {
            auto pRet = ::mi_rezalloc_aligned(buffer, length, align);
            if (!pRet)
            {
                if (gRuntimeConfig.debug.bIsMemoryErrorFatal)
                {
                    SysPanic("ZAllocEx out of memory");
                }
            }

            return pRet;
        }
    }

    AUKN_SYM void *_ZRealloc(void *buffer, Types::size_t length)
    {
        if (AuDebug::IsPointerReserveRange(buffer))
        {
            return AuDebug::gReserveHeap->ZRealloc(buffer, length);
        }
        else
        {
            auto pRet = ::mi_rezalloc(buffer, length);
            if (!pRet)
            {
                if (gRuntimeConfig.debug.bIsMemoryErrorFatal)
                {
                    SysPanic("ZAlloc out of memory");
                }
            }

            return pRet;
        }
    }

    AUKN_SYM void *_FRealloc(void *buffer, Types::size_t length, Types::size_t align)
    {
        if (AuDebug::IsPointerReserveRange(buffer))
        {
            return AuDebug::gReserveHeap->FRealloc(buffer, length, align);
        }
        else
        {
            auto pRet = ::mi_realloc_aligned(buffer, length, align);
            if (!pRet)
            {
                if (gRuntimeConfig.debug.bIsMemoryErrorFatal)
                {
                    SysPanic("FReallocEx out of memory");
                }
            }

            return pRet;
        }
    }

    AUKN_SYM void *_FRealloc(void *buffer, Types::size_t length)
    {
        if (AuDebug::IsPointerReserveRange(buffer))
        {
            return AuDebug::gReserveHeap->FRealloc(buffer, length);
        }
        else
        {
            auto pRet = ::mi_realloc(buffer, length);
            if (!pRet)
            {
                if (gRuntimeConfig.debug.bIsMemoryErrorFatal)
                {
                    SysPanic("FRealloc out of memory");
                }
            }

            return pRet;
        }
    }

    AUKN_SYM void _Free(void *buffer)
    {
        if (AuDebug::IsPointerReserveRange(buffer))
        {
            AuDebug::gReserveHeap->Free(buffer);
        }
        else
        {
            ::mi_free(buffer);
        }
    }

    AUKN_SYM AuUInt GetPageSize()
    {
        return AuHwInfo::GetPageSize();
    }
}