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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
//
// AUTHOR
// N. Nielsen
//
// VERSION
// 0.7
//
// LICENSE
// This software is in the public domain.
//
// The software is provided "as is", without warranty of any kind,
// express or implied, including but not limited to the warranties
// of merchantability, fitness for a particular purpose, and
// noninfringement. In no event shall the author(s) be liable for any
// claim, damages, or other liability, whether in an action of
// contract, tort, or otherwise, arising from, out of, or in connection
// with the software or the use or other dealings in the software.
//
// SUPPORT
// Send bug reports to: <nielsen@memberwebs.com>
//
#include "memref.h"
#include "malloc.h"
#include "assert.h"
// WARNING!! Not thread safe or very efficient for large
// amounts of memory allocations
const size_t kRefSig = 0x1F2F3F4F;
void* _refalloc_dbg(size_t sz)
{
// Allocate extra counter value before memory
size_t* pMem = (size_t*)malloc(sz * sizeof(size_t) * 2);
if(pMem)
{
pMem[0] = kRefSig;
pMem[1] = 1;
return pMem + 2;
}
return pMem;
}
void* _refalloc(size_t sz)
{
// Allocate extra counter value before memory
size_t* pMem = (size_t*)malloc(sz * sizeof(size_t) * 1);
if(pMem)
{
pMem[0] = 1;
return pMem + 1;
}
return pMem;
}
void* _refadd_dbg(void* pBuff)
{
if(pBuff)
{
// Increment the counter value
size_t* pMem = (size_t*)pBuff - 2;
assert(pMem[0] == kRefSig);
pMem[1]++;
}
return pBuff;
}
void* _refadd(void* pBuff)
{
if(pBuff)
// Increment the counter value
((size_t*)pBuff)[-1]++;
return pBuff;
}
void _refrelease_dbg(void* pBuff)
{
if(pBuff)
{
// Decrement the counter value
size_t* pMem = (size_t*)pBuff - 2;
assert(pMem[0] == kRefSig);
if(!--pMem[1])
free(pMem);
}
}
void _refrelease(void* pBuff)
{
if(pBuff)
{
// Decrement the counter value
size_t* pMem = (size_t*)pBuff - 1;
if(!--pMem[0])
free(pMem);
}
}
|