blob: 719b42383adc010a1aa773effd66bcb59266a03c (
plain)
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
|
#ifndef _ALLOCATION_INC
#define _ALLOCATION_INC
// store allocated memory pointers for deallocation when shutting down
class Allocations {
public:
Allocations(): head(0) {}
virtual ~Allocations() { free_all(); }
void push_back(void *pt) { Node *n = new Node; n->value = pt; n->next = head; head = n;}
protected:
class Node {
public:
void *value;
Node *next;
};
Node *head;
void free_all() {
Node *current;
while(head) {
current = head;
head = head->next;
free(current->value);
delete current;
}
}
};
extern Allocations allocations;
__inline static void *safe_alloc(size_t bytes) {
if(bytes == 0) return 0;
void *ret = malloc(bytes);
allocations.push_back(ret);
return ret;
}
__inline static char *safe_strdup(const char *s) {
if (!s) return 0;
char *ret = _strdup(s);
allocations.push_back(ret);
return ret;
}
__inline static wchar_t *safe_wstrdup(const wchar_t *s) {
if (!s) return 0;
wchar_t *ret = _wcsdup(s);
allocations.push_back(ret);
return ret;
}
#ifdef _UNICODE
#define safe_tstrdup(x) safe_wstrdup(x)
#else
#define sage_tstrdup(x) safe_strdup(x)
#endif
__inline static BYTE *safe_bytedup(BYTE *bytes, int size) {
if (!bytes || size == 0) return 0;
BYTE *ret = (BYTE *)malloc(size + 1);
memcpy(ret, bytes, size);
*(ret + size) = 0;
allocations.push_back(ret);
return ret;
}
#endif
|
