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