Mercurial > vba-linux

     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
author	Robert McIntyre <rlm@mit.edu>
date	Sat, 03 Mar 2012 10:31:47 -0600
parents	f9f4f1b99eed
children