BaseTools/Source/C/LzmaCompress/Sdk/C/Alloc.c

   1 /* Alloc.c -- Memory allocation functions
   2 2008-09-24
   3 Igor Pavlov
   4 Public domain */
   5
   6 #ifdef _WIN32
   7 #include <windows.h>
   8 #endif
   9 #include <stdlib.h>
  10
  11 #include "Alloc.h"
  12
  13 /* #define _SZ_ALLOC_DEBUG */
  14
  15 /* use _SZ_ALLOC_DEBUG to debug alloc/free operations */
  16 #ifdef _SZ_ALLOC_DEBUG
  17 #include <stdio.h>
  18 int g_allocCount = 0;
  19 int g_allocCountMid = 0;
  20 int g_allocCountBig = 0;
  21 #endif
  22
  23 void *MyAlloc(size_t size)
  24 {
  25   if (size == 0)
  26     return 0;
  27   #ifdef _SZ_ALLOC_DEBUG
  28   {
  29     void *p = malloc(size);
  30     fprintf(stderr, "\nAlloc %10d bytes, count = %10d,  addr = %8X", size, g_allocCount++, (unsigned)p);
  31     return p;
  32   }
  33   #else
  34   return malloc(size);
  35   #endif
  36 }
  37
  38 void MyFree(void *address)
  39 {
  40   #ifdef _SZ_ALLOC_DEBUG
  41   if (address != 0)
  42     fprintf(stderr, "\nFree; count = %10d,  addr = %8X", --g_allocCount, (unsigned)address);
  43   #endif
  44   free(address);
  45 }
  46
  47 #ifdef _WIN32
  48
  49 void *MidAlloc(size_t size)
  50 {
  51   if (size == 0)
  52     return 0;
  53   #ifdef _SZ_ALLOC_DEBUG
  54   fprintf(stderr, "\nAlloc_Mid %10d bytes;  count = %10d", size, g_allocCountMid++);
  55   #endif
  56   return VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE);
  57 }
  58
  59 void MidFree(void *address)
  60 {
  61   #ifdef _SZ_ALLOC_DEBUG
  62   if (address != 0)
  63     fprintf(stderr, "\nFree_Mid; count = %10d", --g_allocCountMid);
  64   #endif
  65   if (address == 0)
  66     return;
  67   VirtualFree(address, 0, MEM_RELEASE);
  68 }
  69
  70 #ifndef MEM_LARGE_PAGES
  71 #undef _7ZIP_LARGE_PAGES
  72 #endif
  73
  74 #ifdef _7ZIP_LARGE_PAGES
  75 SIZE_T g_LargePageSize = 0;
  76 typedef SIZE_T (WINAPI *GetLargePageMinimumP)();
  77 #endif
  78
  79 void SetLargePageSize()
  80 {
  81   #ifdef _7ZIP_LARGE_PAGES
  82   SIZE_T size = 0;
  83   GetLargePageMinimumP largePageMinimum = (GetLargePageMinimumP)
  84         GetProcAddress(GetModuleHandle(TEXT("kernel32.dll")), "GetLargePageMinimum");
  85   if (largePageMinimum == 0)
  86     return;
  87   size = largePageMinimum();
  88   if (size == 0 || (size & (size - 1)) != 0)
  89     return;
  90   g_LargePageSize = size;
  91   #endif
  92 }
  93
  94
  95 void *BigAlloc(size_t size)
  96 {
  97   if (size == 0)
  98     return 0;
  99   #ifdef _SZ_ALLOC_DEBUG
 100   fprintf(stderr, "\nAlloc_Big %10d bytes;  count = %10d", size, g_allocCountBig++);
 101   #endif
 102
 103   #ifdef _7ZIP_LARGE_PAGES
 104   if (g_LargePageSize != 0 && g_LargePageSize <= (1 << 30) && size >= (1 << 18))
 105   {
 106     void *res = VirtualAlloc(0, (size + g_LargePageSize - 1) & (~(g_LargePageSize - 1)),
 107         MEM_COMMIT | MEM_LARGE_PAGES, PAGE_READWRITE);
 108     if (res != 0)
 109       return res;
 110   }
 111   #endif
 112   return VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE);
 113 }
 114
 115 void BigFree(void *address)
 116 {
 117   #ifdef _SZ_ALLOC_DEBUG
 118   if (address != 0)
 119     fprintf(stderr, "\nFree_Big; count = %10d", --g_allocCountBig);
 120   #endif
 121
 122   if (address == 0)
 123     return;
 124   VirtualFree(address, 0, MEM_RELEASE);
 125 }
 126
 127 #endif