Added my new NFA character normalizer. Not yet ready, but want to
[yaz-moved-to-github.git] / src / xmalloc.c
1 /*
2  * Copyright (C) 1995-2005, Index Data ApS
3  * All rights reserved.
4  *
5  * $Id: xmalloc.c,v 1.7 2006-04-21 10:28:07 adam Exp $
6  */
7 /**
8  * \file xmalloc.c
9  * \brief Implements malloc interface.
10  */
11
12 #if HAVE_CONFIG_H
13 #include <config.h>
14 #endif
15
16 #include <stdio.h>
17 #include <stdlib.h>
18 #include <string.h>
19
20 #include <yaz/log.h>
21 #include <yaz/xmalloc.h>
22
23 #ifndef TRACE_XMALLOC
24 #define TRACE_XMALLOC 1
25 #endif
26
27 static int log_level=0;
28 static int log_level_initialized=0;
29
30 #if TRACE_XMALLOC > 1
31
32 static const unsigned char head[] = {88, 77, 66, 55, 44, 33, 22, 11};
33 static const unsigned char tail[] = {11, 22, 33, 44, 55, 66, 77, 88};
34 static const unsigned char freed[] = {11, 22, 33, 44, 55, 66, 77, 88};
35
36 struct dmalloc_info {
37     int len;
38     char file[16];
39     int line;
40     struct dmalloc_info *next;
41     struct dmalloc_info *prev;
42 };
43
44 struct dmalloc_info *dmalloc_list = 0;
45
46
47 void *xmalloc_d(size_t nbytes, const char *file, int line)
48 {
49     char *res;
50     struct dmalloc_info *dinfo;
51     
52     if (!log_level_initialized)
53     {
54         log_level=yaz_log_module_level("malloc");
55         log_level_initialized=1;
56     }
57
58     if (!(res = (char*) malloc(nbytes + sizeof(*dinfo)+16*sizeof(char))))
59         return 0;
60     dinfo = (struct dmalloc_info *) res;
61     strncpy (dinfo->file, file, sizeof(dinfo->file)-1);
62     dinfo->file[sizeof(dinfo->file)-1] = '\0';
63     dinfo->line = line;
64     dinfo->len = nbytes;
65     
66     dinfo->prev = 0;
67     dinfo->next = dmalloc_list;
68     if (dinfo->next)
69         dinfo->next->prev = dinfo;
70     dmalloc_list = dinfo;
71     
72     memcpy(res + sizeof(*dinfo), head, 8*sizeof(char));
73     res += sizeof(*dinfo) + 8*sizeof(char);
74     memcpy(res + nbytes, tail, 8*sizeof(char));
75     return res;
76 }
77
78 void xfree_d(void *ptr, const char *file, int line)
79 {
80     struct dmalloc_info *dinfo;
81
82     if (!ptr)
83         return;
84     dinfo = (struct dmalloc_info *)
85         ((char*)ptr - 8*sizeof(char) - sizeof(*dinfo));
86     if (memcmp(head, (char*) ptr - 8*sizeof(char), 8*sizeof(char)))
87     {
88         yaz_log(YLOG_FATAL, "xfree_d bad head, %s:%d, %p", file, line, ptr);
89         abort();
90     }
91     if (memcmp((char*) ptr + dinfo->len, tail, 8*sizeof(char)))
92     {
93         yaz_log(YLOG_FATAL, "xfree_d bad tail, %s:%d, %p", file, line, ptr);
94         abort();
95     }
96     if (dinfo->prev)
97         dinfo->prev->next = dinfo->next;
98     else
99         dmalloc_list = dinfo->next;
100     if (dinfo->next)
101         dinfo->next->prev = dinfo->prev;
102     memcpy ((char*) ptr - 8*sizeof(char), freed, 8*sizeof(char));
103     free(dinfo);
104     return;
105 }
106
107 void *xrealloc_d(void *p, size_t nbytes, const char *file, int line)
108 {
109     struct dmalloc_info *dinfo;
110     char *ptr = (char*) p;
111     char *res;
112     
113     if (!log_level_initialized)
114     {
115         log_level=yaz_log_module_level("malloc");
116         log_level_initialized=1;
117     }
118
119     if (!ptr)
120     {
121         if (!nbytes)
122             return 0;
123         res = (char *) malloc(nbytes + sizeof(*dinfo) + 16*sizeof(char));
124     }
125     else
126     {
127         if (memcmp(head, ptr - 8*sizeof(char), 8*sizeof(char)))
128         {
129             yaz_log(YLOG_FATAL, "xrealloc_d bad head, %s:%d, %p",
130                     file, line, ptr);
131             abort();
132         }
133         dinfo = (struct dmalloc_info *) (ptr-8*sizeof(char) - sizeof(*dinfo));
134         if (memcmp(ptr + dinfo->len, tail, 8*sizeof(char)))
135         {
136             yaz_log(YLOG_FATAL, "xrealloc_d bad tail, %s:%d, %p",
137                     file, line, ptr);
138             abort();
139         }
140         if (dinfo->prev)
141             dinfo->prev->next = dinfo->next;
142         else
143             dmalloc_list = dinfo->next;
144         if (dinfo->next)
145             dinfo->next->prev = dinfo->prev;
146         
147         if (!nbytes)
148         {
149             free (dinfo);
150             return 0;
151         }
152         res = (char *)
153             realloc(dinfo, nbytes + sizeof(*dinfo) + 16*sizeof(char));
154     }
155     if (!res)
156         return 0;
157     dinfo = (struct dmalloc_info *) res;
158     strncpy (dinfo->file, file, sizeof(dinfo->file)-1);
159     dinfo->file[sizeof(dinfo->file)-1] = '\0';
160     dinfo->line = line;
161     dinfo->len = nbytes;
162
163     dinfo->prev = 0;
164     dinfo->next = dmalloc_list;
165     if (dmalloc_list)
166         dmalloc_list->prev = dinfo;
167     dmalloc_list = dinfo;
168     
169     memcpy(res + sizeof(*dinfo), head, 8*sizeof(char));
170     res += sizeof(*dinfo) + 8*sizeof(char);
171     memcpy(res + nbytes, tail, 8*sizeof(char));
172     return res;
173 }
174
175 void *xcalloc_d(size_t nmemb, size_t size, const char *file, int line)
176 {
177     char *res;
178     struct dmalloc_info *dinfo;
179     size_t nbytes = nmemb * size;
180     
181     if (!log_level_initialized)
182     {
183         log_level=yaz_log_module_level("malloc");
184         log_level_initialized=1;
185     }
186
187     if (!(res = (char*) calloc(1, nbytes+sizeof(*dinfo)+16*sizeof(char))))
188         return 0;
189     dinfo = (struct dmalloc_info *) res;
190     strncpy (dinfo->file, file, sizeof(dinfo->file)-1);
191     dinfo->file[sizeof(dinfo->file)-1] = '\0';
192     dinfo->line = line;
193     dinfo->len = nbytes;
194     
195     dinfo->prev = 0;
196     dinfo->next = dmalloc_list;
197     if (dinfo->next)
198         dinfo->next->prev = dinfo;
199     dmalloc_list = dinfo;
200     
201     memcpy(res + sizeof(*dinfo), head, 8*sizeof(char));
202     res += sizeof(*dinfo) + 8*sizeof(char);
203     memcpy(res + nbytes, tail, 8*sizeof(char));
204     return res;
205 }
206
207 void xmalloc_trav_d(const char *file, int line)
208 {
209     size_t size = 0;
210     struct dmalloc_info *dinfo = dmalloc_list;
211     
212     if (!log_level_initialized)
213     {
214         log_level=yaz_log_module_level("malloc");
215         log_level_initialized=1;
216     }
217
218     yaz_log (log_level, "malloc_trav %s:%d", file, line);
219     while (dinfo)
220     {
221         yaz_log (log_level, " %20s:%d p=%p size=%d", dinfo->file, dinfo->line,
222               ((char*) dinfo)+sizeof(*dinfo)+8*sizeof(char), dinfo->len);
223         size += dinfo->len;
224         dinfo = dinfo->next;
225     }
226     yaz_log (log_level, "total bytes %ld", (long) size);
227 }
228
229 #else
230 /* TRACE_XMALLOC <= 1 */
231 #define xrealloc_d(o, x, f, l) realloc(o, x)
232 #define xmalloc_d(x, f, l) malloc(x)
233 #define xcalloc_d(x,y, f, l) calloc(x,y)
234 #define xfree_d(x, f, l) free(x)
235 #define xmalloc_trav_d(f, l) 
236 #endif
237
238 void xmalloc_trav_f(const char *s, const char *file, int line)
239 {
240     if (!log_level_initialized)
241     {
242         log_level=yaz_log_module_level("malloc");
243         log_level_initialized=1;
244     }
245
246     xmalloc_trav_d(file, line);
247 }
248
249 void *xrealloc_f (void *o, size_t size, const char *file, int line)
250 {
251     void *p = xrealloc_d (o, size, file, line);
252
253     if (!log_level_initialized)
254     {
255         log_level=yaz_log_module_level("malloc");
256         log_level_initialized=1;
257     }
258
259     if(log_level)
260         yaz_log (log_level,
261             "%s:%d: xrealloc(s=%ld) %p -> %p", file, line, (long) size, o, p);
262     if (!p)
263     {
264         yaz_log (YLOG_FATAL|YLOG_ERRNO, "Out of memory, realloc (%ld bytes)",
265                  (long) size);
266         exit(1);
267     }
268     return p;
269 }
270
271 void *xmalloc_f (size_t size, const char *file, int line)
272 {
273     void *p = xmalloc_d (size, file, line);
274     
275     if (!log_level_initialized)
276     {
277         log_level=yaz_log_module_level("malloc");
278         log_level_initialized=1;
279     }
280
281     if (log_level)
282         yaz_log (log_level, "%s:%d: xmalloc(s=%ld) %p", file, line, 
283                  (long) size, p);
284
285     if (!p)
286     {
287         yaz_log (YLOG_FATAL, "Out of memory - malloc (%ld bytes)",
288                  (long) size);
289         exit (1);
290     }
291     return p;
292 }
293
294 void *xcalloc_f (size_t nmemb, size_t size, const char *file, int line)
295 {
296     void *p = xcalloc_d (nmemb, size, file, line);
297     if (!log_level_initialized)
298     {
299         log_level=yaz_log_module_level("malloc");
300         log_level_initialized=1;
301     }
302
303     if (log_level)
304         yaz_log (log_level, "%s:%d: xcalloc(s=%ld) %p", file, line,
305                  (long) size, p);
306
307     if (!p)
308     {
309         yaz_log (YLOG_FATAL, "Out of memory - calloc (%ld, %ld)",
310                  (long) nmemb, (long) size);
311         exit (1);
312     }
313     return p;
314 }
315
316 char *xstrdup_f (const char *s, const char *file, int line)
317 {
318     char *p = (char *)xmalloc_d (strlen(s)+1, file, line);
319     if (!log_level_initialized)
320     {
321         log_level=yaz_log_module_level("malloc");
322         log_level_initialized=1;
323     }
324
325     if (log_level)
326         yaz_log (log_level, "%s:%d: xstrdup(s=%ld) %p", file, line, 
327                  (long) strlen(s)+1, p);
328
329     strcpy (p, s);
330     return p;
331 }
332
333 void xfree_f(void *p, const char *file, int line)
334 {
335     if (!p)
336         return ;
337     if (log_level)
338         yaz_log (log_level, "%s:%d: xfree %p", file, line, p);
339     xfree_d(p, file, line);
340 }
341 /*
342  * Local variables:
343  * c-basic-offset: 4
344  * indent-tabs-mode: nil
345  * End:
346  * vim: shiftwidth=4 tabstop=8 expandtab
347  */
348