hostallocator.h

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

#pragma once

classHostAllocatorfinal
{
private:
HostAllocator()
 {
 }

public:
template <typename T>
 T* allocate(size_t count)
 {
 ClrSafeInt<size_t> safeElemSize(sizeof(T));
 ClrSafeInt<size_t> safeCount(count);
 ClrSafeInt<size_t> size = safeElemSize * safeCount;
if (size.IsOverflow())
 {
returnnullptr;
 }

returnstatic_cast<T*>(allocateHostMemory(size.Value()));
 }

voiddeallocate(void* p)
 {
freeHostMemory(p);
 }

static HostAllocator getHostAllocator()
 {
returnHostAllocator();
 }

private:
void* allocateHostMemory(size_t size);
voidfreeHostMemory(void* p);
};

// Global operator new overloads that work with HostAllocator

inlinevoid* __cdecl operatornew(size_t n, HostAllocator alloc)
{
return alloc.allocate<char>(n);
}

inlinevoid* __cdecl operatornew[](size_t n, HostAllocator alloc)
{
return alloc.allocate<char>(n);
}