OLD | NEW |
| (Empty) |
1 /* Alloc.c -- Memory allocation functions | |
2 2008-09-24 | |
3 Igor Pavlov | |
4 Public domain | |
5 in the public domain */ | |
6 | |
7 #ifdef _WIN32 | |
8 #include <windows.h> | |
9 #endif | |
10 #include <stdlib.h> | |
11 | |
12 #include "Alloc.h" | |
13 | |
14 /* #define _SZ_ALLOC_DEBUG */ | |
15 | |
16 /* use _SZ_ALLOC_DEBUG to debug alloc/free operations */ | |
17 #ifdef _SZ_ALLOC_DEBUG | |
18 #include <stdio.h> | |
19 int g_allocCount = 0; | |
20 int g_allocCountMid = 0; | |
21 int g_allocCountBig = 0; | |
22 #endif | |
23 | |
24 void *MyAlloc(size_t size) | |
25 { | |
26 if (size == 0) | |
27 return 0; | |
28 #ifdef _SZ_ALLOC_DEBUG | |
29 { | |
30 void *p = malloc(size); | |
31 fprintf(stderr, "\nAlloc %10d bytes, count = %10d, addr = %8X", size, g_all
ocCount++, (unsigned)p); | |
32 return p; | |
33 } | |
34 #else | |
35 return malloc(size); | |
36 #endif | |
37 } | |
38 | |
39 void MyFree(void *address) | |
40 { | |
41 #ifdef _SZ_ALLOC_DEBUG | |
42 if (address != 0) | |
43 fprintf(stderr, "\nFree; count = %10d, addr = %8X", --g_allocCount, (unsign
ed)address); | |
44 #endif | |
45 free(address); | |
46 } | |
47 | |
48 #ifdef _WIN32 | |
49 | |
50 void *MidAlloc(size_t size) | |
51 { | |
52 if (size == 0) | |
53 return 0; | |
54 #ifdef _SZ_ALLOC_DEBUG | |
55 fprintf(stderr, "\nAlloc_Mid %10d bytes; count = %10d", size, g_allocCountMid
++); | |
56 #endif | |
57 return VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE); | |
58 } | |
59 | |
60 void MidFree(void *address) | |
61 { | |
62 #ifdef _SZ_ALLOC_DEBUG | |
63 if (address != 0) | |
64 fprintf(stderr, "\nFree_Mid; count = %10d", --g_allocCountMid); | |
65 #endif | |
66 if (address == 0) | |
67 return; | |
68 VirtualFree(address, 0, MEM_RELEASE); | |
69 } | |
70 | |
71 #ifndef MEM_LARGE_PAGES | |
72 #undef _7ZIP_LARGE_PAGES | |
73 #endif | |
74 | |
75 #ifdef _7ZIP_LARGE_PAGES | |
76 SIZE_T g_LargePageSize = 0; | |
77 typedef SIZE_T (WINAPI *GetLargePageMinimumP)(); | |
78 #endif | |
79 | |
80 void SetLargePageSize() | |
81 { | |
82 #ifdef _7ZIP_LARGE_PAGES | |
83 SIZE_T size = 0; | |
84 GetLargePageMinimumP largePageMinimum = (GetLargePageMinimumP) | |
85 GetProcAddress(GetModuleHandle(TEXT("kernel32.dll")), "GetLargePageMinim
um"); | |
86 if (largePageMinimum == 0) | |
87 return; | |
88 size = largePageMinimum(); | |
89 if (size == 0 || (size & (size - 1)) != 0) | |
90 return; | |
91 g_LargePageSize = size; | |
92 #endif | |
93 } | |
94 | |
95 | |
96 void *BigAlloc(size_t size) | |
97 { | |
98 if (size == 0) | |
99 return 0; | |
100 #ifdef _SZ_ALLOC_DEBUG | |
101 fprintf(stderr, "\nAlloc_Big %10d bytes; count = %10d", size, g_allocCountBig
++); | |
102 #endif | |
103 | |
104 #ifdef _7ZIP_LARGE_PAGES | |
105 if (g_LargePageSize != 0 && g_LargePageSize <= (1 << 30) && size >= (1 << 18)) | |
106 { | |
107 void *res = VirtualAlloc(0, (size + g_LargePageSize - 1) & (~(g_LargePageSiz
e - 1)), | |
108 MEM_COMMIT | MEM_LARGE_PAGES, PAGE_READWRITE); | |
109 if (res != 0) | |
110 return res; | |
111 } | |
112 #endif | |
113 return VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE); | |
114 } | |
115 | |
116 void BigFree(void *address) | |
117 { | |
118 #ifdef _SZ_ALLOC_DEBUG | |
119 if (address != 0) | |
120 fprintf(stderr, "\nFree_Big; count = %10d", --g_allocCountBig); | |
121 #endif | |
122 | |
123 if (address == 0) | |
124 return; | |
125 VirtualFree(address, 0, MEM_RELEASE); | |
126 } | |
127 | |
128 #endif | |
OLD | NEW |