add Imager's memory debugger
[poe-xs-queue-array.git] / alloc.c
CommitLineData
1bbcbbe4
TC
1#include "alloc.h"
2#include <stdio.h>
3#include <errno.h>
4#include <stdarg.h>
5
6/* cheap version of Imager's logging */
7char const *last_file; int last_line;
8
9static void do_log(int level, char const *fmt, ...);
10static
11void
12bndcheck(int idx);
13void
14bndcheck_all(void);
15
16#define mm_log(x) ((last_file = __FILE__), (last_line = __LINE__), do_log x )
17
18#ifdef MEM_DEBUG
19
20#define MAXMAL 102400
21#define MAXDESC 65
22
23#define UNDRRNVAL 10
24#define OVERRNVAL 10
25
26#define PADBYTE 0xaa
27
28static int malloc_need_init = 1;
29
30void
31malloc_state(void);
32
33typedef struct {
34 void* ptr;
35 size_t size;
36 char comm[MAXDESC];
37} malloc_entry;
38
39malloc_entry malloc_pointers[MAXMAL];
40
41
42
43/* Utility functions */
44
45
46static
47void
48malloc_init(void) {
49 int i;
50 for(i=0; i<MAXMAL; i++) malloc_pointers[i].ptr = NULL;
51 malloc_need_init = 0;
52 atexit(malloc_state);
53}
54
55
56static
57int
58find_ptr(void *p) {
59 int i;
60 for(i=0;i<MAXMAL;i++)
61 if (malloc_pointers[i].ptr == p)
62 return i;
63 return -1;
64}
65
66
67/* Takes a pointer to real start of array,
68 * sets the entries in the table, returns
69 * the offset corrected pointer */
70
71static
72void *
73set_entry(int i, char *buf, size_t size, char const *file, int line) {
74 memset( buf, PADBYTE, UNDRRNVAL );
75 memset( &buf[UNDRRNVAL+size], PADBYTE, OVERRNVAL );
76 buf += UNDRRNVAL;
77 malloc_pointers[i].ptr = buf;
78 malloc_pointers[i].size = size;
79 sprintf(malloc_pointers[i].comm,"%s (%d)", file, line);
80 return buf;
81}
82
83void
84malloc_state(void) {
85 int i, total = 0;
86
87 mm_log((0,"malloc_state()\n"));
88 bndcheck_all();
89 for(i=0; i<MAXMAL; i++) if (malloc_pointers[i].ptr != NULL) {
90 mm_log((0,"%d: %d (0x%x) : %s\n", i, malloc_pointers[i].size, malloc_pointers[i].ptr, malloc_pointers[i].comm));
91 total += malloc_pointers[i].size;
92 }
93 if (total == 0) mm_log((0,"No memory currently used!\n"));
94 else mm_log((0,"total: %d\n",total));
95}
96
97
98
99void*
100mymalloc_file_line(size_t size, char const* file, int line) {
101 char *buf;
102 int i;
103 if (malloc_need_init) malloc_init();
104
105 /* bndcheck_all(); Uncomment for LOTS OF THRASHING */
106
107 if ( (i = find_ptr(NULL)) < 0 ) {
108 mm_log((0,"more than %d segments allocated at %s (%d)\n", MAXMAL, file, line));
109 exit(3);
110 }
111
112 if ( (buf = malloc(size+UNDRRNVAL+OVERRNVAL)) == NULL ) {
113 mm_log((1,"Unable to allocate %i for %s (%i)\n", size, file, line));
114 exit(3);
115 }
116
117 buf = set_entry(i, buf, size, file, line);
118 mm_log((1,"mymalloc_file_line: slot <%d> %d bytes allocated at %p for %s (%d)\n", i, size, buf, file, line));
119 return buf;
120}
121
122void *
123(mymalloc)(int size) {
124 return mymalloc_file_line(size, "unknown", 0);
125}
126
127void*
128myrealloc_file_line(void *ptr, size_t newsize, char const * file, int line) {
129 char *buf;
130 int i;
131
132 if (malloc_need_init) malloc_init();
133 /* bndcheck_all(); ACTIVATE FOR LOTS OF THRASHING */
134
135 if (!ptr) {
136 mm_log((1, "realloc called with ptr = NULL, sending request to malloc\n"));
137 return mymalloc_file_line(newsize, file, line);
138 }
139
140 if (!newsize) {
141 mm_log((1, "newsize = 0, sending request to free\n"));
142 myfree_file_line(ptr, file, line);
143 return NULL;
144 }
145
146 if ( (i = find_ptr(ptr)) == -1) {
147 mm_log((0, "Unable to find %p in realloc for %s (%i)\n", ptr, file, line));
148 exit(3);
149 }
150
151 if ( (buf = realloc(((char *)ptr)-UNDRRNVAL, UNDRRNVAL+OVERRNVAL+newsize)) == NULL ) {
152 mm_log((1,"Unable to reallocate %i bytes at %p for %s (%i)\n", newsize, ptr, file, line));
153 exit(3);
154 }
155
156 buf = set_entry(i, buf, newsize, file, line);
157 mm_log((1,"realloc_file_line: slot <%d> %d bytes allocated at %p for %s (%d)\n", i, newsize, buf, file, line));
158 return buf;
159}
160
161void *
162(myrealloc)(void *ptr, size_t newsize) {
163 return myrealloc_file_line(ptr, newsize, "unknown", 0);
164}
165
166static
167void
168bndcheck(int idx) {
169 int i;
170 size_t s = malloc_pointers[idx].size;
171 unsigned char *pp = malloc_pointers[idx].ptr;
172 if (!pp) {
173 mm_log((1, "bndcheck: No pointer in slot %d\n", idx));
174 return;
175 }
176
177 for(i=0;i<UNDRRNVAL;i++)
178 if (pp[-(1+i)] != PADBYTE)
179 mm_log((1,"bndcheck: UNDERRUN OF %d bytes detected: slot = %d, point = %p, size = %d\n", i+1, idx, pp, s ));
180
181 for(i=0;i<OVERRNVAL;i++)
182 if (pp[s+i] != PADBYTE)
183 mm_log((1,"bndcheck: OVERRUN OF %d bytes detected: slot = %d, point = %p, size = %d\n", i+1, idx, pp, s ));
184}
185
186void
187bndcheck_all(void) {
188 int idx;
189 mm_log((1, "bndcheck_all()\n"));
190 for(idx=0; idx<MAXMAL; idx++)
191 if (malloc_pointers[idx].ptr)
192 bndcheck(idx);
193}
194
195void
196myfree_file_line(void *p, char const *file, int line) {
197 char *pp = p;
198 int match = 0;
199 int i;
200
201 for(i=0; i<MAXMAL; i++) if (malloc_pointers[i].ptr == p) {
202 mm_log((1,"myfree_file_line: pointer %i (%s) freed at %s (%i)\n", i, malloc_pointers[i].comm, file, line));
203 bndcheck(i);
204 malloc_pointers[i].ptr = NULL;
205 match++;
206 }
207
208 mm_log((1, "myfree_file_line: freeing address %p (real %p) (%s:%d)\n", pp, pp-UNDRRNVAL, file, line));
209
210 if (match != 1) {
211 mm_log((1, "myfree_file_line: INCONSISTENT REFCOUNT %d at %s (%i)\n", match, file, line));
212 fprintf(stderr, "myfree_file_line: INCONSISTENT REFCOUNT %d at %s (%i)\n", match, file, line);
213 exit(255);
214 }
215
216
217 free(pp-UNDRRNVAL);
218}
219
220void
221(myfree)(void *block) {
222 myfree_file_line(block, "unknown", 0);
223}
224
225#else
226
227#define malloc_comm(a,b) (mymalloc(a))
228
229void
230malloc_state() {
231}
232
233void*
234mymalloc(size_t size) {
235 void *buf;
236
237 if (size < 0) {
238 fprintf(stderr, "Attempt to allocate size %d\n", size);
239 exit(3);
240 }
241
242 if ( (buf = malloc(size)) == NULL ) {
243 mm_log((1, "mymalloc: unable to malloc %d\n", size));
244 fprintf(stderr,"Unable to malloc %d.\n", size); exit(3);
245 }
246 mm_log((1, "mymalloc(size %d) -> %p\n", size, buf));
247 return buf;
248}
249
250void *
251mymalloc_file_line(size_t size, char *file, int line) {
252 return mymalloc(size);
253}
254
255void
256myfree(void *p) {
257 mm_log((1, "myfree(p %p)\n", p));
258 free(p);
259}
260
261void
262myfree_file_line(void *p, char *file, int line) {
263 myfree(p);
264}
265
266void *
267myrealloc(void *block, size_t size) {
268 void *result;
269
270 mm_log((1, "myrealloc(block %p, size %u)\n", block, size));
271 if ((result = realloc(block, size)) == NULL) {
272 mm_log((1, "myrealloc: out of memory\n"));
273 fprintf(stderr, "Out of memory.\n");
274 exit(3);
275 }
276 return result;
277}
278
279void *
280myrealloc_file_line(void *block, size_t newsize, char *file, int size) {
281 return myrealloc(block, newsize);
282}
283
284#endif /* IMAGER_MALLOC_DEBUG */
285
286FILE *log_file;
287
288void
289setup_log(void) {
290 char *log_env = getenv("MEM_DEBUG");
291 if (!log_env)
292 return;
293
294 if (strcmp(log_env, "STDERR") == 0) {
295 log_file = stderr;
296 return;
297 }
298 log_file = fopen(log_env, "w+");
299 if (!log_file) {
300 fprintf(stderr, "Could not open log %s: %s\n", log_env, strerror(errno));
301 exit(3);
302 }
303}
304
305static void
306do_log(int level, char const *fmt, ...) {
307 if (!log_file) setup_log();
308
309 if (log_file) {
310 va_list ap;
311 va_start(ap, fmt);
312 fprintf(log_file, "[%s:%d] %d:", last_file, last_line, level);
313
314 vfprintf(stderr, fmt, ap);
315 va_end(ap);
316 }
317}