/******************************Module*Header*******************************\
* Module Name: bdd.h
*
* Basic Display Driver memory allocation, deletion, and tracking 
*
*
* Copyright (c) 2010 Microsoft Corporation
\**************************************************************************/

#include "BDD.hxx"

#pragma code_seg("PAGE")

//
// New and delete operators
//
void* __cdecl operator new(size_t Size, BDD_POOL_TYPE PoolType)
{
    PAGED_CODE();

    Size = (Size != 0) ? Size : 1;
    POOL_FLAGS Flags = PoolType == BDD_POOL_TYPE::NonPaged ? POOL_FLAG_NON_PAGED : POOL_FLAG_PAGED;
    
    // Note that ExAllocatePool2 replaces ExAllocatePool* APIs in OS's starting
    // with Windows 10, version 2004. If your driver targets previous versions it
    // should use ExAllocatePoolZero instead.
    void* pObject = ExAllocatePool2(Flags, Size, BDDTAG);

#if DBG
    if (pObject != NULL)
    {
        RtlFillMemory(pObject, Size, 0xCD);
    }
#endif // DBG

    return pObject;
}

void* __cdecl operator new[](size_t Size, BDD_POOL_TYPE PoolType)
{
    PAGED_CODE();

    Size = (Size != 0) ? Size : 1;
    POOL_FLAGS Flags = PoolType == BDD_POOL_TYPE::NonPaged ? POOL_FLAG_NON_PAGED : POOL_FLAG_PAGED;
    
    void* pObject = ExAllocatePool2(Flags, Size, BDDTAG);

#if DBG
    if (pObject != NULL)
    {
        RtlFillMemory(pObject, Size, 0xCD);
    }
#endif // DBG

    return pObject;
}

void __cdecl operator delete(void* pObject, BDD_POOL_TYPE PoolType)
{
    PAGED_CODE();

    UNREFERENCED_PARAMETER(PoolType);

    ::operator delete(pObject);
}

void __cdecl operator delete[](void* pObject, BDD_POOL_TYPE PoolType)
{
    PAGED_CODE();

    UNREFERENCED_PARAMETER(PoolType);

    ::operator delete[](pObject);
}

void __cdecl operator delete(void* pObject)
{
    PAGED_CODE();

    if (pObject != NULL)
    {
        ExFreePool(pObject);
    }
}

//
// size_t version is needed for VS2015(C++ 14).  
// 
void __cdecl operator delete(void* pObject, size_t s)
{
    PAGED_CODE();

    UNREFERENCED_PARAMETER( s );

    ::operator delete( pObject );
}

void __cdecl operator delete[](void* pObject)
{
    PAGED_CODE();

    if (pObject != NULL)
    {
        ExFreePool(pObject);
    }
}