1 /* Alloc.c -- Memory allocation functions
2 2015-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>
18 int g_allocCount = 0;
19 int g_allocCountMid = 0;
20 int g_allocCountBig = 0;
21 #endif
22
MyAlloc(size_t size)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
MyFree(void * address)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
MidAlloc(size_t size)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
MidFree(void * address)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
SetLargePageSize()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
BigAlloc(size_t size)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
BigFree(void * address)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
128
129
SzAlloc(void * p,size_t size)130 static void *SzAlloc(void *p, size_t size) { UNUSED_VAR(p); return MyAlloc(size); }
SzFree(void * p,void * address)131 static void SzFree(void *p, void *address) { UNUSED_VAR(p); MyFree(address); }
132 ISzAlloc g_Alloc = { SzAlloc, SzFree };
133
SzBigAlloc(void * p,size_t size)134 static void *SzBigAlloc(void *p, size_t size) { UNUSED_VAR(p); return BigAlloc(size); }
SzBigFree(void * p,void * address)135 static void SzBigFree(void *p, void *address) { UNUSED_VAR(p); BigFree(address); }
136 ISzAlloc g_BigAlloc = { SzBigAlloc, SzBigFree };
137