1/* Alloc.c -- Memory allocation functions
  22015-02-21 : Igor Pavlov : Public domain */
  3
  4#include "Precomp.h"
  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>
 18int g_allocCount = 0;
 19int g_allocCountMid = 0;
 20int g_allocCountBig = 0;
 21#endif
 22
 23void *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
 38void 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
 49void *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
 59void 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
 75SIZE_T g_LargePageSize = 0;
 76typedef SIZE_T (WINAPI *GetLargePageMinimumP)();
 77#endif
 78
 79void 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
 95void *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
115void 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
128
129
130static void *SzAlloc(void *p, size_t size) { UNUSED_VAR(p); return MyAlloc(size); }
131static void SzFree(void *p, void *address) { UNUSED_VAR(p); MyFree(address); }
132ISzAlloc g_Alloc = { SzAlloc, SzFree };
133
134static void *SzBigAlloc(void *p, size_t size) { UNUSED_VAR(p); return BigAlloc(size); }
135static void SzBigFree(void *p, void *address) { UNUSED_VAR(p); BigFree(address); }
136ISzAlloc g_BigAlloc = { SzBigAlloc, SzBigFree };