1 /* Alloc.c -- Memory allocation functions
\r
2 2015-02-21 : Igor Pavlov : Public domain */
\r
13 /* #define _SZ_ALLOC_DEBUG */
\r
15 /* use _SZ_ALLOC_DEBUG to debug alloc/free operations */
\r
16 #ifdef _SZ_ALLOC_DEBUG
\r
18 int g_allocCount = 0;
\r
19 int g_allocCountMid = 0;
\r
20 int g_allocCountBig = 0;
\r
23 void *MyAlloc(size_t size)
\r
27 #ifdef _SZ_ALLOC_DEBUG
\r
29 void *p = malloc(size);
\r
30 fprintf(stderr, "\nAlloc %10d bytes, count = %10d, addr = %8X", size, g_allocCount++, (unsigned)p);
\r
34 return malloc(size);
\r
38 void MyFree(void *address)
\r
40 #ifdef _SZ_ALLOC_DEBUG
\r
42 fprintf(stderr, "\nFree; count = %10d, addr = %8X", --g_allocCount, (unsigned)address);
\r
49 void *MidAlloc(size_t size)
\r
53 #ifdef _SZ_ALLOC_DEBUG
\r
54 fprintf(stderr, "\nAlloc_Mid %10d bytes; count = %10d", size, g_allocCountMid++);
\r
56 return VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE);
\r
59 void MidFree(void *address)
\r
61 #ifdef _SZ_ALLOC_DEBUG
\r
63 fprintf(stderr, "\nFree_Mid; count = %10d", --g_allocCountMid);
\r
67 VirtualFree(address, 0, MEM_RELEASE);
\r
70 #ifndef MEM_LARGE_PAGES
\r
71 #undef _7ZIP_LARGE_PAGES
\r
74 #ifdef _7ZIP_LARGE_PAGES
\r
75 SIZE_T g_LargePageSize = 0;
\r
76 typedef SIZE_T (WINAPI *GetLargePageMinimumP)();
\r
79 void SetLargePageSize()
\r
81 #ifdef _7ZIP_LARGE_PAGES
\r
83 GetLargePageMinimumP largePageMinimum = (GetLargePageMinimumP)
\r
84 GetProcAddress(GetModuleHandle(TEXT("kernel32.dll")), "GetLargePageMinimum");
\r
85 if (largePageMinimum == 0)
\r
87 size = largePageMinimum();
\r
88 if (size == 0 || (size & (size - 1)) != 0)
\r
90 g_LargePageSize = size;
\r
95 void *BigAlloc(size_t size)
\r
99 #ifdef _SZ_ALLOC_DEBUG
\r
100 fprintf(stderr, "\nAlloc_Big %10d bytes; count = %10d", size, g_allocCountBig++);
\r
103 #ifdef _7ZIP_LARGE_PAGES
\r
104 if (g_LargePageSize != 0 && g_LargePageSize <= (1 << 30) && size >= (1 << 18))
\r
106 void *res = VirtualAlloc(0, (size + g_LargePageSize - 1) & (~(g_LargePageSize - 1)),
\r
107 MEM_COMMIT | MEM_LARGE_PAGES, PAGE_READWRITE);
\r
112 return VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE);
\r
115 void BigFree(void *address)
\r
117 #ifdef _SZ_ALLOC_DEBUG
\r
119 fprintf(stderr, "\nFree_Big; count = %10d", --g_allocCountBig);
\r
124 VirtualFree(address, 0, MEM_RELEASE);
\r
130 static void *SzAlloc(void *p, size_t size) { UNUSED_VAR(p); return MyAlloc(size); }
\r
131 static void SzFree(void *p, void *address) { UNUSED_VAR(p); MyFree(address); }
\r
132 ISzAlloc g_Alloc = { SzAlloc, SzFree };
\r
134 static void *SzBigAlloc(void *p, size_t size) { UNUSED_VAR(p); return BigAlloc(size); }
\r
135 static void SzBigFree(void *p, void *address) { UNUSED_VAR(p); BigFree(address); }
\r
136 ISzAlloc g_BigAlloc = { SzBigAlloc, SzBigFree };
\r