1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
#include "pilercr.h"
#ifdef _MSC_VER
#include <crtdbg.h>
#endif
static int AllocatedBytes;
static int PeakAllocatedBytes;
// Allocate memory, fail on error, track total
void *allocmem(int bytes)
{
assert(bytes < 0xfffffff0); // check for overlow
char *p = (char *) malloc((size_t) (bytes + 4));
if (0 == p)
Quit("Out of memory (%d)", bytes);
int *pSize = (int *) p;
*pSize = bytes;
AllocatedBytes += bytes;
if (AllocatedBytes > PeakAllocatedBytes)
PeakAllocatedBytes = AllocatedBytes;
return p + 4;
}
void freemem(void *p)
{
if (0 == p)
return;
int *pSize = (int *) ((char *) p - 4);
int bytes = *pSize;
assert(bytes <= AllocatedBytes);
AllocatedBytes -= bytes;
free(((char *) p) - 4);
}
void chkmem()
{
#ifdef _MSC_VER
assert(_CrtCheckMemory());
#endif
}
|
