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