// Common/Alloc.cpp | |
#include "StdAfx.h" | |
#ifdef _WIN32 | |
#include "MyWindows.h" | |
#else | |
#include <stdlib.h> | |
#endif | |
#include "Alloc.h" | |
/* #define _SZ_ALLOC_DEBUG */ | |
/* use _SZ_ALLOC_DEBUG to debug alloc/free operations */ | |
#ifdef _SZ_ALLOC_DEBUG | |
#include <stdio.h> | |
int g_allocCount = 0; | |
int g_allocCountMid = 0; | |
int g_allocCountBig = 0; | |
#endif | |
void *MyAlloc(size_t size) throw() | |
{ | |
if (size == 0) | |
return 0; | |
#ifdef _SZ_ALLOC_DEBUG | |
fprintf(stderr, "\nAlloc %10d bytes; count = %10d", size, g_allocCount++); | |
#endif | |
return ::malloc(size); | |
} | |
void MyFree(void *address) throw() | |
{ | |
#ifdef _SZ_ALLOC_DEBUG | |
if (address != 0) | |
fprintf(stderr, "\nFree; count = %10d", --g_allocCount); | |
#endif | |
::free(address); | |
} | |
#ifdef _WIN32 | |
void *MidAlloc(size_t size) throw() | |
{ | |
if (size == 0) | |
return 0; | |
#ifdef _SZ_ALLOC_DEBUG | |
fprintf(stderr, "\nAlloc_Mid %10d bytes; count = %10d", size, g_allocCountMid++); | |
#endif | |
return ::VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE); | |
} | |
void MidFree(void *address) throw() | |
{ | |
#ifdef _SZ_ALLOC_DEBUG | |
if (address != 0) | |
fprintf(stderr, "\nFree_Mid; count = %10d", --g_allocCountMid); | |
#endif | |
if (address == 0) | |
return; | |
::VirtualFree(address, 0, MEM_RELEASE); | |
} | |
static SIZE_T g_LargePageSize = | |
#ifdef _WIN64 | |
(1 << 21); | |
#else | |
(1 << 22); | |
#endif | |
typedef SIZE_T (WINAPI *GetLargePageMinimumP)(); | |
bool SetLargePageSize() | |
{ | |
GetLargePageMinimumP largePageMinimum = (GetLargePageMinimumP) | |
::GetProcAddress(::GetModuleHandle(TEXT("kernel32.dll")), "GetLargePageMinimum"); | |
if (largePageMinimum == 0) | |
return false; | |
SIZE_T size = largePageMinimum(); | |
if (size == 0 || (size & (size - 1)) != 0) | |
return false; | |
g_LargePageSize = size; | |
return true; | |
} | |
void *BigAlloc(size_t size) throw() | |
{ | |
if (size == 0) | |
return 0; | |
#ifdef _SZ_ALLOC_DEBUG | |
fprintf(stderr, "\nAlloc_Big %10d bytes; count = %10d", size, g_allocCountBig++); | |
#endif | |
if (size >= (1 << 18)) | |
{ | |
void *res = ::VirtualAlloc(0, (size + g_LargePageSize - 1) & (~(g_LargePageSize - 1)), | |
MEM_COMMIT | MEM_LARGE_PAGES, PAGE_READWRITE); | |
if (res != 0) | |
return res; | |
} | |
return ::VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE); | |
} | |
void BigFree(void *address) throw() | |
{ | |
#ifdef _SZ_ALLOC_DEBUG | |
if (address != 0) | |
fprintf(stderr, "\nFree_Big; count = %10d", --g_allocCountBig); | |
#endif | |
if (address == 0) | |
return; | |
::VirtualFree(address, 0, MEM_RELEASE); | |
} | |
#endif |