1 /* Alloc.c */

     2 

     3 #ifdef _WIN32

     4 #include <windows.h>

     5 #endif

     6 #include <stdlib.h>

     7 

     8 #include "Alloc.h"

     9 

    10 /* #define _SZ_ALLOC_DEBUG */

    11 

    12 /* use _SZ_ALLOC_DEBUG to debug alloc/free operations */

    13 #ifdef _SZ_ALLOC_DEBUG

    14 #include <stdio.h>

    15 int g_allocCount = 0;

    16 int g_allocCountMid = 0;

    17 int g_allocCountBig = 0;

    18 #endif

    19 

    20 void *MyAlloc(size_t size)

    21 {

    22   if (size == 0)

    23     return 0;

    24   #ifdef _SZ_ALLOC_DEBUG

    25   fprintf(stderr, "\nAlloc %10d bytes; count = %10d", size, g_allocCount++);

    26   #endif

    27   return malloc(size);

    28 }

    29 

    30 void MyFree(void *address)

    31 {

    32   #ifdef _SZ_ALLOC_DEBUG

    33   if (address != 0)

    34     fprintf(stderr, "\nFree; count = %10d", --g_allocCount);

    35   #endif

    36   free(address);

    37 }

    38 

    39 #ifdef _WIN32

    40 

    41 void *MidAlloc(size_t size)

    42 {

    43   if (size == 0)

    44     return 0;

    45   #ifdef _SZ_ALLOC_DEBUG

    46   fprintf(stderr, "\nAlloc_Mid %10d bytes;  count = %10d", size, g_allocCountMid++);

    47   #endif

    48   return VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE);

    49 }

    50 

    51 void MidFree(void *address)

    52 {

    53   #ifdef _SZ_ALLOC_DEBUG

    54   if (address != 0)

    55     fprintf(stderr, "\nFree_Mid; count = %10d", --g_allocCountMid);

    56   #endif

    57   if (address == 0)

    58     return;

    59   VirtualFree(address, 0, MEM_RELEASE);

    60 }

    61 

    62 #ifndef MEM_LARGE_PAGES

    63 #undef _7ZIP_LARGE_PAGES

    64 #endif

    65 

    66 #ifdef _7ZIP_LARGE_PAGES

    67 SIZE_T g_LargePageSize = 0;

    68 typedef SIZE_T (WINAPI *GetLargePageMinimumP)();

    69 #endif

    70 

    71 void SetLargePageSize()

    72 {

    73   #ifdef _7ZIP_LARGE_PAGES

    74   SIZE_T size = 0;

    75   GetLargePageMinimumP largePageMinimum = (GetLargePageMinimumP)

    76         GetProcAddress(GetModuleHandle(TEXT("kernel32.dll")), "GetLargePageMinimum");

    77   if (largePageMinimum == 0)

    78     return;

    79   size = largePageMinimum();

    80   if (size == 0 || (size & (size - 1)) != 0)

    81     return;

    82   g_LargePageSize = size;

    83   #endif

    84 }

    85 

    86 

    87 void *BigAlloc(size_t size)

    88 {

    89   if (size == 0)

    90     return 0;

    91   #ifdef _SZ_ALLOC_DEBUG

    92   fprintf(stderr, "\nAlloc_Big %10d bytes;  count = %10d", size, g_allocCountBig++);

    93   #endif

    94   

    95   #ifdef _7ZIP_LARGE_PAGES

    96   if (g_LargePageSize != 0 && g_LargePageSize <= (1 << 30) && size >= (1 << 18))

    97   {

    98     void *res = VirtualAlloc(0, (size + g_LargePageSize - 1) & (~(g_LargePageSize - 1)), 

    99         MEM_COMMIT | MEM_LARGE_PAGES, PAGE_READWRITE);

   100     if (res != 0)

   101       return res;

   102   }

   103   #endif

   104   return VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE);

   105 }

   106 

   107 void BigFree(void *address)

   108 {

   109   #ifdef _SZ_ALLOC_DEBUG

   110   if (address != 0)

   111     fprintf(stderr, "\nFree_Big; count = %10d", --g_allocCountBig);

   112   #endif

   113   

   114   if (address == 0)

   115     return;

   116   VirtualFree(address, 0, MEM_RELEASE);

   117 }

   118 

   119 #endif