repo sync | baa3858 | 2013-07-26 17:53:31 -0700 | [diff] [blame] | 1 | /* Alloc.c -- Memory allocation functions
|
| 2 | 2008-09-24
|
| 3 | Igor Pavlov
|
| 4 | Public 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>
|
| 18 | int g_allocCount = 0;
|
| 19 | int g_allocCountMid = 0;
|
| 20 | int g_allocCountBig = 0;
|
| 21 | #endif
|
| 22 |
|
| 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 |
|
| 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 |
|
| 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 |
|
| 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 |
|
| 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 |
|
| 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 |
|
| 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
|