BaseTools LzmaCompress: Update LZMA to new 16.04 version

[mirror_edk2.git] / BaseTools / Source / C / LzmaCompress / Sdk / C / Alloc.c

/* Alloc.c -- Memory allocation functions\r
2015-02-21 : Igor Pavlov : Public domain */\r
\r
#include "Precomp.h"\r
\r
#ifdef _WIN32\r
#include <windows.h>\r
#endif\r
#include <stdlib.h>\r
\r
#include "Alloc.h"\r
\r
/* #define _SZ_ALLOC_DEBUG */\r
\r
/* use _SZ_ALLOC_DEBUG to debug alloc/free operations */\r
#ifdef _SZ_ALLOC_DEBUG\r
#include <stdio.h>\r
int g_allocCount = 0;\r
int g_allocCountMid = 0;\r
int g_allocCountBig = 0;\r
#endif\r
\r
void *MyAlloc(size_t size)\r
{\r
  if (size == 0)\r
    return 0;\r
  #ifdef _SZ_ALLOC_DEBUG\r
  {\r
    void *p = malloc(size);\r
    fprintf(stderr, "\nAlloc %10d bytes, count = %10d,  addr = %8X", size, g_allocCount++, (unsigned)p);\r
    return p;\r
  }\r
  #else\r
  return malloc(size);\r
  #endif\r
}\r
\r
void MyFree(void *address)\r
{\r
  #ifdef _SZ_ALLOC_DEBUG\r
  if (address != 0)\r
    fprintf(stderr, "\nFree; count = %10d,  addr = %8X", --g_allocCount, (unsigned)address);\r
  #endif\r
  free(address);\r
}\r
\r
#ifdef _WIN32\r
\r
void *MidAlloc(size_t size)\r
{\r
  if (size == 0)\r
    return 0;\r
  #ifdef _SZ_ALLOC_DEBUG\r
  fprintf(stderr, "\nAlloc_Mid %10d bytes;  count = %10d", size, g_allocCountMid++);\r
  #endif\r
  return VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE);\r
}\r
\r
void MidFree(void *address)\r
{\r
  #ifdef _SZ_ALLOC_DEBUG\r
  if (address != 0)\r
    fprintf(stderr, "\nFree_Mid; count = %10d", --g_allocCountMid);\r
  #endif\r
  if (address == 0)\r
    return;\r
  VirtualFree(address, 0, MEM_RELEASE);\r
}\r
\r
#ifndef MEM_LARGE_PAGES\r
#undef _7ZIP_LARGE_PAGES\r
#endif\r
\r
#ifdef _7ZIP_LARGE_PAGES\r
SIZE_T g_LargePageSize = 0;\r
typedef SIZE_T (WINAPI *GetLargePageMinimumP)();\r
#endif\r
\r
void SetLargePageSize()\r
{\r
  #ifdef _7ZIP_LARGE_PAGES\r
  SIZE_T size = 0;\r
  GetLargePageMinimumP largePageMinimum = (GetLargePageMinimumP)\r
        GetProcAddress(GetModuleHandle(TEXT("kernel32.dll")), "GetLargePageMinimum");\r
  if (largePageMinimum == 0)\r
    return;\r
  size = largePageMinimum();\r
  if (size == 0 || (size & (size - 1)) != 0)\r
    return;\r
  g_LargePageSize = size;\r
  #endif\r
}\r
\r
\r
void *BigAlloc(size_t size)\r
{\r
  if (size == 0)\r
    return 0;\r
  #ifdef _SZ_ALLOC_DEBUG\r
  fprintf(stderr, "\nAlloc_Big %10d bytes;  count = %10d", size, g_allocCountBig++);\r
  #endif\r
  \r
  #ifdef _7ZIP_LARGE_PAGES\r
  if (g_LargePageSize != 0 && g_LargePageSize <= (1 << 30) && size >= (1 << 18))\r
  {\r
    void *res = VirtualAlloc(0, (size + g_LargePageSize - 1) & (~(g_LargePageSize - 1)),\r
        MEM_COMMIT | MEM_LARGE_PAGES, PAGE_READWRITE);\r
    if (res != 0)\r
      return res;\r
  }\r
  #endif\r
  return VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE);\r
}\r
\r
void BigFree(void *address)\r
{\r
  #ifdef _SZ_ALLOC_DEBUG\r
  if (address != 0)\r
    fprintf(stderr, "\nFree_Big; count = %10d", --g_allocCountBig);\r
  #endif\r
  \r
  if (address == 0)\r
    return;\r
  VirtualFree(address, 0, MEM_RELEASE);\r
}\r
\r
#endif\r
\r
\r
static void *SzAlloc(void *p, size_t size) { UNUSED_VAR(p); return MyAlloc(size); }\r
static void SzFree(void *p, void *address) { UNUSED_VAR(p); MyFree(address); }\r
ISzAlloc g_Alloc = { SzAlloc, SzFree };\r
\r
static void *SzBigAlloc(void *p, size_t size) { UNUSED_VAR(p); return BigAlloc(size); }\r
static void SzBigFree(void *p, void *address) { UNUSED_VAR(p); BigFree(address); }\r
ISzAlloc g_BigAlloc = { SzBigAlloc, SzBigFree };\r