mirror of
https://github.com/reactos/reactos.git
synced 2024-08-28 22:28:36 +00:00
106 lines
1.9 KiB
C++
106 lines
1.9 KiB
C++
#ifndef __ATLMEM_H__
|
|
#define __ATLMEM_H__
|
|
|
|
#pragma once
|
|
#include "atlcore.h"
|
|
|
|
|
|
namespace ATL
|
|
{
|
|
|
|
//__interface __declspec(uuid("654F7EF5-CFDF-4df9-A450-6C6A13C622C0"))
|
|
class IAtlMemMgr
|
|
{
|
|
public:
|
|
virtual ~IAtlMemMgr() {};
|
|
|
|
virtual _Ret_maybenull_ _Post_writable_byte_size_(SizeBytes) void* Allocate(
|
|
_In_ size_t SizeBytes
|
|
) = 0;
|
|
|
|
virtual void Free(
|
|
_Inout_opt_ void* Buffer
|
|
) = 0;
|
|
|
|
virtual _Ret_maybenull_ _Post_writable_byte_size_(SizeBytes) void* Reallocate(
|
|
_Inout_updates_bytes_opt_(SizeBytes) void* Buffer,
|
|
_In_ size_t SizeBytes
|
|
) = 0;
|
|
|
|
virtual size_t GetSize(
|
|
_In_ void* Buffer
|
|
) = 0;
|
|
};
|
|
|
|
class CWin32Heap : public IAtlMemMgr
|
|
{
|
|
public:
|
|
HANDLE m_hHeap;
|
|
|
|
public:
|
|
CWin32Heap() :
|
|
m_hHeap(NULL)
|
|
{
|
|
}
|
|
|
|
CWin32Heap(_In_ HANDLE hHeap) :
|
|
m_hHeap(hHeap)
|
|
{
|
|
ATLASSERT(hHeap != NULL);
|
|
}
|
|
|
|
virtual ~CWin32Heap()
|
|
{
|
|
}
|
|
|
|
|
|
// IAtlMemMgr
|
|
_Ret_maybenull_ _Post_writable_byte_size_(SizeBytes) virtual void* Allocate(
|
|
_In_ size_t SizeBytes
|
|
)
|
|
{
|
|
return ::HeapAlloc(m_hHeap, 0, SizeBytes);
|
|
}
|
|
|
|
virtual void Free(
|
|
_In_opt_ void* Buffer
|
|
)
|
|
{
|
|
if (Buffer)
|
|
{
|
|
if (!::HeapFree(m_hHeap, 0, Buffer))
|
|
ATLASSERT(FALSE);
|
|
}
|
|
}
|
|
|
|
_Ret_maybenull_ _Post_writable_byte_size_(SizeBytes) virtual void* Reallocate(
|
|
_In_opt_ void* Buffer,
|
|
_In_ size_t SizeBytes
|
|
)
|
|
{
|
|
if (SizeBytes == 0)
|
|
{
|
|
Free(Buffer);
|
|
return NULL;
|
|
}
|
|
|
|
if (Buffer == NULL)
|
|
{
|
|
return Allocate(SizeBytes);
|
|
}
|
|
|
|
return ::HeapReAlloc(m_hHeap, 0, Buffer, SizeBytes);
|
|
}
|
|
|
|
virtual size_t GetSize(
|
|
_Inout_ void* Buffer
|
|
)
|
|
{
|
|
return ::HeapSize(m_hHeap, 0, Buffer);
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
#endif
|