1/* Alloc.c -- Memory allocation functions
22008-09-24
3Igor Pavlov
4Public 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>
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