glib/gmem.c (g_mem_is_system_malloc): Return !vtable_set.
[platform/upstream/glib.git] / glib / gmem.c
1 /* GLIB - Library of useful routines for C programming
2  * Copyright (C) 1995-1997  Peter Mattis, Spencer Kimball and Josh MacDonald
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the
16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
17  * Boston, MA 02111-1307, USA.
18  */
19
20 /*
21  * Modified by the GLib Team and others 1997-2000.  See the AUTHORS
22  * file for a list of people on the GLib Team.  See the ChangeLog
23  * files for a list of changes.  These files are distributed with
24  * GLib at ftp://ftp.gtk.org/pub/gtk/. 
25  */
26
27 /* 
28  * MT safe
29  */
30
31 #ifdef HAVE_CONFIG_H
32 #include <config.h>
33 #endif
34
35 #include <stdlib.h>
36 #include <string.h>
37 #include "glib.h"
38
39
40 /* notes on macros:
41  * having DISABLE_MEM_POOLS defined, disables mem_chunks alltogether, their
42  * allocations are performed through ordinary g_malloc/g_free.
43  * having G_DISABLE_CHECKS defined disables use of glib_mem_profiler_table and
44  * g_mem_profile().
45  * REALLOC_0_WORKS is defined if g_realloc (NULL, x) works.
46  * SANE_MALLOC_PROTOS is defined if the systems malloc() and friends functions
47  * match the corresponding GLib prototypes, keep configure.in and gmem.h in sync here.
48  * if ENABLE_GC_FRIENDLY is defined, freed memory should be 0-wiped.
49  */
50
51 #define MEM_PROFILE_TABLE_SIZE 4096
52
53 #define MEM_AREA_SIZE 4L
54
55 #ifdef  G_DISABLE_CHECKS
56 #  define ENTER_MEM_CHUNK_ROUTINE()
57 #  define LEAVE_MEM_CHUNK_ROUTINE()
58 #  define IN_MEM_CHUNK_ROUTINE()        FALSE
59 #else   /* !G_DISABLE_CHECKS */
60 static GPrivate* mem_chunk_recursion = NULL;
61 #  define MEM_CHUNK_ROUTINE_COUNT()     GPOINTER_TO_UINT (g_private_get (mem_chunk_recursion))
62 #  define ENTER_MEM_CHUNK_ROUTINE()     g_private_set (mem_chunk_recursion, GUINT_TO_POINTER (MEM_CHUNK_ROUTINE_COUNT () + 1))
63 #  define LEAVE_MEM_CHUNK_ROUTINE()     g_private_set (mem_chunk_recursion, GUINT_TO_POINTER (MEM_CHUNK_ROUTINE_COUNT () - 1))
64 #endif  /* !G_DISABLE_CHECKS */
65
66 #ifndef REALLOC_0_WORKS
67 static gpointer
68 standard_realloc (gpointer mem,
69                   gsize    n_bytes)
70 {
71   if (!mem)
72     return malloc (n_bytes);
73   else
74     return realloc (mem, n_bytes);
75 }
76 #endif  /* !REALLOC_0_WORKS */
77
78 #ifdef SANE_MALLOC_PROTOS
79 #  define standard_malloc       malloc
80 #  ifdef REALLOC_0_WORKS
81 #    define standard_realloc    realloc
82 #  endif /* REALLOC_0_WORKS */
83 #  define standard_free         free
84 #  define standard_calloc       calloc
85 #  define standard_try_malloc   malloc
86 #  define standard_try_realloc  realloc
87 #else   /* !SANE_MALLOC_PROTOS */
88 static gpointer
89 standard_malloc (gsize n_bytes)
90 {
91   return malloc (n_bytes);
92 }
93 #  ifdef REALLOC_0_WORKS
94 static gpointer
95 standard_realloc (gpointer mem,
96                   gsize    n_bytes)
97 {
98   return realloc (mem, n_bytes);
99 }
100 #  endif /* REALLOC_0_WORKS */
101 static void
102 standard_free (gpointer mem)
103 {
104   free (mem);
105 }
106 static gpointer
107 standard_calloc (gsize n_blocks,
108                  gsize n_bytes)
109 {
110   return calloc (n_blocks, n_bytes);
111 }
112 #define standard_try_malloc     standard_malloc
113 #define standard_try_realloc    standard_realloc
114 #endif  /* !SANE_MALLOC_PROTOS */
115
116
117 /* --- variables --- */
118 static GMemVTable glib_mem_vtable = {
119   standard_malloc,
120   standard_realloc,
121   standard_free,
122   standard_calloc,
123   standard_try_malloc,
124   standard_try_realloc,
125 };
126
127
128 /* --- functions --- */
129 gpointer
130 g_malloc (gulong n_bytes)
131 {
132   if (n_bytes)
133     {
134       gpointer mem;
135
136       mem = glib_mem_vtable.malloc (n_bytes);
137       if (mem)
138         return mem;
139
140       g_error ("%s: failed to allocate %lu bytes", G_STRLOC, n_bytes);
141     }
142
143   return NULL;
144 }
145
146 gpointer
147 g_malloc0 (gulong n_bytes)
148 {
149   if (n_bytes)
150     {
151       gpointer mem;
152
153       mem = glib_mem_vtable.calloc (1, n_bytes);
154       if (mem)
155         return mem;
156
157       g_error ("%s: failed to allocate %lu bytes", G_STRLOC, n_bytes);
158     }
159
160   return NULL;
161 }
162
163 gpointer
164 g_realloc (gpointer mem,
165            gulong   n_bytes)
166 {
167   if (n_bytes)
168     {
169       mem = glib_mem_vtable.realloc (mem, n_bytes);
170       if (mem)
171         return mem;
172
173       g_error ("%s: failed to allocate %lu bytes", G_STRLOC, n_bytes);
174     }
175
176   if (mem)
177     glib_mem_vtable.free (mem);
178
179   return NULL;
180 }
181
182 void
183 g_free (gpointer mem)
184 {
185   if (mem)
186     glib_mem_vtable.free (mem);
187 }
188
189 gpointer
190 g_try_malloc (gulong n_bytes)
191 {
192   if (n_bytes)
193     return glib_mem_vtable.try_malloc (n_bytes);
194   else
195     return NULL;
196 }
197
198 gpointer
199 g_try_realloc (gpointer mem,
200                gulong   n_bytes)
201 {
202   if (n_bytes)
203     return glib_mem_vtable.try_realloc (mem, n_bytes);
204
205   if (mem)
206     glib_mem_vtable.free (mem);
207
208   return NULL;
209 }
210
211 static gpointer
212 fallback_calloc (gsize n_blocks,
213                  gsize n_block_bytes)
214 {
215   gsize l = n_blocks * n_block_bytes;
216   gpointer mem = glib_mem_vtable.malloc (l);
217
218   if (mem)
219     memset (mem, 0, l);
220
221   return mem;
222 }
223
224 static gboolean vtable_set = FALSE;
225
226 /**
227  * g_mem_is_system_malloc
228  * 
229  * Checks whether the allocator used by g_malloc() is the system's
230  * malloc implementation. If it returns %TRUE memory allocated with
231  * <function>malloc()</function> can be used interchangeable with 
232  * memory allocated using g_malloc(). This function is useful for 
233  * avoiding an extra copy of allocated memory returned by a 
234  * non-GLib-based API.
235  *
236  * A different allocator can be set using g_mem_set_vtable().
237  *
238  * Return value: if %TRUE, <function>malloc()</function> and g_malloc() can be mixed.
239  **/
240 gboolean
241 g_mem_is_system_malloc (void)
242 {
243   return !vtable_set;
244 }
245
246 void
247 g_mem_set_vtable (GMemVTable *vtable)
248 {
249   if (!vtable_set)
250     {
251       vtable_set = TRUE;
252       if (vtable->malloc && vtable->realloc && vtable->free)
253         {
254           glib_mem_vtable.malloc = vtable->malloc;
255           glib_mem_vtable.realloc = vtable->realloc;
256           glib_mem_vtable.free = vtable->free;
257           glib_mem_vtable.calloc = vtable->calloc ? vtable->calloc : fallback_calloc;
258           glib_mem_vtable.try_malloc = vtable->try_malloc ? vtable->try_malloc : glib_mem_vtable.malloc;
259           glib_mem_vtable.try_realloc = vtable->try_realloc ? vtable->try_realloc : glib_mem_vtable.realloc;
260         }
261       else
262         g_warning (G_STRLOC ": memory allocation vtable lacks one of malloc(), realloc() or free()");
263     }
264   else
265     g_warning (G_STRLOC ": memory allocation vtable can only be set once at startup");
266 }
267
268
269 /* --- memory profiling and checking --- */
270 #ifdef  G_DISABLE_CHECKS
271 GMemVTable *glib_mem_profiler_table = &glib_mem_vtable;
272 void
273 g_mem_profile (void)
274 {
275 }
276 #else   /* !G_DISABLE_CHECKS */
277 typedef enum {
278   PROFILER_FREE         = 0,
279   PROFILER_ALLOC        = 1,
280   PROFILER_RELOC        = 2,
281   PROFILER_ZINIT        = 4
282 } ProfilerJob;
283 static guint *profile_data = NULL;
284 static gulong profile_allocs = 0;
285 static gulong profile_mc_allocs = 0;
286 static gulong profile_zinit = 0;
287 static gulong profile_frees = 0;
288 static gulong profile_mc_frees = 0;
289 static GMutex *g_profile_mutex = NULL;
290 #ifdef  G_ENABLE_DEBUG
291 static volatile gulong g_trap_free_size = 0;
292 static volatile gulong g_trap_realloc_size = 0;
293 static volatile gulong g_trap_malloc_size = 0;
294 #endif  /* G_ENABLE_DEBUG */
295
296 #define PROFILE_TABLE(f1,f2,f3)   ( ( ((f3) << 2) | ((f2) << 1) | (f1) ) * (MEM_PROFILE_TABLE_SIZE + 1))
297
298 static void
299 profiler_log (ProfilerJob job,
300               gulong      n_bytes,
301               gboolean    success)
302 {
303   g_mutex_lock (g_profile_mutex);
304   if (!profile_data)
305     {
306       profile_data = standard_malloc ((MEM_PROFILE_TABLE_SIZE + 1) * 8 * sizeof (profile_data[0]));
307       if (!profile_data)        /* memory system kiddin' me, eh? */
308         {
309           g_mutex_unlock (g_profile_mutex);
310           return;
311         }
312     }
313
314   if (MEM_CHUNK_ROUTINE_COUNT () == 0)
315     {
316       if (n_bytes < MEM_PROFILE_TABLE_SIZE)
317         profile_data[n_bytes + PROFILE_TABLE ((job & PROFILER_ALLOC) != 0,
318                                               (job & PROFILER_RELOC) != 0,
319                                               success != 0)] += 1;
320       else
321         profile_data[MEM_PROFILE_TABLE_SIZE + PROFILE_TABLE ((job & PROFILER_ALLOC) != 0,
322                                                              (job & PROFILER_RELOC) != 0,
323                                                              success != 0)] += 1;
324       if (success)
325         {
326           if (job & PROFILER_ALLOC)
327             {
328               profile_allocs += n_bytes;
329               if (job & PROFILER_ZINIT)
330                 profile_zinit += n_bytes;
331             }
332           else
333             profile_frees += n_bytes;
334         }
335     }
336   else if (success)
337     {
338       if (job & PROFILER_ALLOC)
339         profile_mc_allocs += n_bytes;
340       else
341         profile_mc_frees += n_bytes;
342     }
343   g_mutex_unlock (g_profile_mutex);
344 }
345
346 static void
347 profile_print_locked (guint   *local_data,
348                       gboolean success)
349 {
350   gboolean need_header = TRUE;
351   guint i;
352
353   for (i = 0; i <= MEM_PROFILE_TABLE_SIZE; i++)
354     {
355       glong t_malloc = local_data[i + PROFILE_TABLE (1, 0, success)];
356       glong t_realloc = local_data[i + PROFILE_TABLE (1, 1, success)];
357       glong t_free = local_data[i + PROFILE_TABLE (0, 0, success)];
358       glong t_refree = local_data[i + PROFILE_TABLE (0, 1, success)];
359       
360       if (!t_malloc && !t_realloc && !t_free && !t_refree)
361         continue;
362       else if (need_header)
363         {
364           need_header = FALSE;
365           g_print (" blocks of | allocated  | freed      | allocated  | freed      | n_bytes   \n");
366           g_print ("  n_bytes  | n_times by | n_times by | n_times by | n_times by | remaining \n");
367           g_print ("           | malloc()   | free()     | realloc()  | realloc()  |           \n");
368           g_print ("===========|============|============|============|============|===========\n");
369         }
370       if (i < MEM_PROFILE_TABLE_SIZE)
371         g_print ("%10u | %10ld | %10ld | %10ld | %10ld |%+11ld\n",
372                  i, t_malloc, t_free, t_realloc, t_refree,
373                  (t_malloc - t_free + t_realloc - t_refree) * i);
374       else if (i >= MEM_PROFILE_TABLE_SIZE)
375         g_print ("   >%6u | %10ld | %10ld | %10ld | %10ld |        ***\n",
376                  i, t_malloc, t_free, t_realloc, t_refree);
377     }
378   if (need_header)
379     g_print (" --- none ---\n");
380 }
381
382 void
383 g_mem_profile (void)
384 {
385   guint local_data[(MEM_PROFILE_TABLE_SIZE + 1) * 8 * sizeof (profile_data[0])];
386   gulong local_allocs;
387   gulong local_zinit;
388   gulong local_frees;
389   gulong local_mc_allocs;
390   gulong local_mc_frees;
391
392   g_mutex_lock (g_profile_mutex);
393
394   local_allocs = profile_allocs;
395   local_zinit = profile_zinit;
396   local_frees = profile_frees;
397   local_mc_allocs = profile_mc_allocs;
398   local_mc_frees = profile_mc_frees;
399
400   if (!profile_data)
401     {
402       g_mutex_unlock (g_profile_mutex);
403       return;
404     }
405
406   memcpy (local_data, profile_data, 
407           (MEM_PROFILE_TABLE_SIZE + 1) * 8 * sizeof (profile_data[0]));
408   
409   g_mutex_unlock (g_profile_mutex);
410
411   g_print ("GLib Memory statistics (successful operations):\n");
412   profile_print_locked (local_data, TRUE);
413   g_print ("GLib Memory statistics (failing operations):\n");
414   profile_print_locked (local_data, FALSE);
415   g_print ("Total bytes: allocated=%lu, zero-initialized=%lu (%.2f%%), freed=%lu (%.2f%%), remaining=%lu\n",
416            local_allocs,
417            local_zinit,
418            ((gdouble) local_zinit) / local_allocs * 100.0,
419            local_frees,
420            ((gdouble) local_frees) / local_allocs * 100.0,
421            local_allocs - local_frees);
422   g_print ("MemChunk bytes: allocated=%lu, freed=%lu (%.2f%%), remaining=%lu\n",
423            local_mc_allocs,
424            local_mc_frees,
425            ((gdouble) local_mc_frees) / local_mc_allocs * 100.0,
426            local_mc_allocs - local_mc_frees);
427 }
428
429 static gpointer
430 profiler_try_malloc (gsize n_bytes)
431 {
432   gulong *p;
433
434 #ifdef  G_ENABLE_DEBUG
435   if (g_trap_malloc_size == n_bytes)
436     G_BREAKPOINT ();
437 #endif  /* G_ENABLE_DEBUG */
438
439   p = standard_malloc (sizeof (gulong) * 2 + n_bytes);
440
441   if (p)
442     {
443       p[0] = 0;         /* free count */
444       p[1] = n_bytes;   /* length */
445       profiler_log (PROFILER_ALLOC, n_bytes, TRUE);
446       p += 2;
447     }
448   else
449     profiler_log (PROFILER_ALLOC, n_bytes, FALSE);
450   
451   return p;
452 }
453
454 static gpointer
455 profiler_malloc (gsize n_bytes)
456 {
457   gpointer mem = profiler_try_malloc (n_bytes);
458
459   if (!mem)
460     g_mem_profile ();
461
462   return mem;
463 }
464
465 static gpointer
466 profiler_calloc (gsize n_blocks,
467                  gsize n_block_bytes)
468 {
469   gsize l = n_blocks * n_block_bytes;
470   gulong *p;
471
472 #ifdef  G_ENABLE_DEBUG
473   if (g_trap_malloc_size == l)
474     G_BREAKPOINT ();
475 #endif  /* G_ENABLE_DEBUG */
476   
477   p = standard_calloc (1, sizeof (gulong) * 2 + l);
478
479   if (p)
480     {
481       p[0] = 0;         /* free count */
482       p[1] = l;         /* length */
483       profiler_log (PROFILER_ALLOC | PROFILER_ZINIT, l, TRUE);
484       p += 2;
485     }
486   else
487     {
488       profiler_log (PROFILER_ALLOC | PROFILER_ZINIT, l, FALSE);
489       g_mem_profile ();
490     }
491
492   return p;
493 }
494
495 static void
496 profiler_free (gpointer mem)
497 {
498   gulong *p = mem;
499
500   p -= 2;
501   if (p[0])     /* free count */
502     {
503       g_warning ("free(%p): memory has been freed %lu times already", p + 2, p[0]);
504       profiler_log (PROFILER_FREE,
505                     p[1],       /* length */
506                     FALSE);
507     }
508   else
509     {
510 #ifdef  G_ENABLE_DEBUG
511       if (g_trap_free_size == p[1])
512         G_BREAKPOINT ();
513 #endif  /* G_ENABLE_DEBUG */
514
515       profiler_log (PROFILER_FREE,
516                     p[1],       /* length */
517                     TRUE);
518       memset (p + 2, 0xaa, p[1]);
519
520       /* for all those that miss standard_free (p); in this place, yes,
521        * we do leak all memory when profiling, and that is intentional
522        * to catch double frees. patch submissions are futile.
523        */
524     }
525   p[0] += 1;
526 }
527
528 static gpointer
529 profiler_try_realloc (gpointer mem,
530                       gsize    n_bytes)
531 {
532   gulong *p = mem;
533
534   p -= 2;
535
536 #ifdef  G_ENABLE_DEBUG
537   if (g_trap_realloc_size == n_bytes)
538     G_BREAKPOINT ();
539 #endif  /* G_ENABLE_DEBUG */
540   
541   if (mem && p[0])      /* free count */
542     {
543       g_warning ("realloc(%p, %u): memory has been freed %lu times already", p + 2, n_bytes, p[0]);
544       profiler_log (PROFILER_ALLOC | PROFILER_RELOC, n_bytes, FALSE);
545
546       return NULL;
547     }
548   else
549     {
550       p = standard_realloc (mem ? p : NULL, sizeof (gulong) * 2 + n_bytes);
551
552       if (p)
553         {
554           if (mem)
555             profiler_log (PROFILER_FREE | PROFILER_RELOC, p[1], TRUE);
556           p[0] = 0;
557           p[1] = n_bytes;
558           profiler_log (PROFILER_ALLOC | PROFILER_RELOC, p[1], TRUE);
559           p += 2;
560         }
561       else
562         profiler_log (PROFILER_ALLOC | PROFILER_RELOC, n_bytes, FALSE);
563
564       return p;
565     }
566 }
567
568 static gpointer
569 profiler_realloc (gpointer mem,
570                   gsize    n_bytes)
571 {
572   mem = profiler_try_realloc (mem, n_bytes);
573
574   if (!mem)
575     g_mem_profile ();
576
577   return mem;
578 }
579
580 static GMemVTable profiler_table = {
581   profiler_malloc,
582   profiler_realloc,
583   profiler_free,
584   profiler_calloc,
585   profiler_try_malloc,
586   profiler_try_realloc,
587 };
588 GMemVTable *glib_mem_profiler_table = &profiler_table;
589
590 #endif  /* !G_DISABLE_CHECKS */
591
592
593 /* --- MemChunks --- */
594 typedef struct _GFreeAtom      GFreeAtom;
595 typedef struct _GMemArea       GMemArea;
596
597 struct _GFreeAtom
598 {
599   GFreeAtom *next;
600 };
601
602 struct _GMemArea
603 {
604   GMemArea *next;            /* the next mem area */
605   GMemArea *prev;            /* the previous mem area */
606   gulong index;              /* the current index into the "mem" array */
607   gulong free;               /* the number of free bytes in this mem area */
608   gulong allocated;          /* the number of atoms allocated from this area */
609   gulong mark;               /* is this mem area marked for deletion */
610   gchar mem[MEM_AREA_SIZE];  /* the mem array from which atoms get allocated
611                               * the actual size of this array is determined by
612                               *  the mem chunk "area_size". ANSI says that it
613                               *  must be declared to be the maximum size it
614                               *  can possibly be (even though the actual size
615                               *  may be less).
616                               */
617 };
618
619 struct _GMemChunk
620 {
621   const gchar *name;         /* name of this MemChunk...used for debugging output */
622   gint type;                 /* the type of MemChunk: ALLOC_ONLY or ALLOC_AND_FREE */
623   gint num_mem_areas;        /* the number of memory areas */
624   gint num_marked_areas;     /* the number of areas marked for deletion */
625   guint atom_size;           /* the size of an atom */
626   gulong area_size;          /* the size of a memory area */
627   GMemArea *mem_area;        /* the current memory area */
628   GMemArea *mem_areas;       /* a list of all the mem areas owned by this chunk */
629   GMemArea *free_mem_area;   /* the free area...which is about to be destroyed */
630   GFreeAtom *free_atoms;     /* the free atoms list */
631   GTree *mem_tree;           /* tree of mem areas sorted by memory address */
632   GMemChunk *next;           /* pointer to the next chunk */
633   GMemChunk *prev;           /* pointer to the previous chunk */
634 };
635
636
637 #ifndef DISABLE_MEM_POOLS
638 static gulong g_mem_chunk_compute_size (gulong    size,
639                                         gulong    min_size) G_GNUC_CONST;
640 static gint   g_mem_chunk_area_compare (GMemArea *a,
641                                         GMemArea *b);
642 static gint   g_mem_chunk_area_search  (GMemArea *a,
643                                         gchar    *addr);
644
645 /* here we can't use StaticMutexes, as they depend upon a working
646  * g_malloc, the same holds true for StaticPrivate
647  */
648 static GMutex        *mem_chunks_lock = NULL;
649 static GMemChunk     *mem_chunks = NULL;
650
651 GMemChunk*
652 g_mem_chunk_new (const gchar  *name,
653                  gint          atom_size,
654                  gulong        area_size,
655                  gint          type)
656 {
657   GMemChunk *mem_chunk;
658   gulong rarea_size;
659
660   g_return_val_if_fail (atom_size > 0, NULL);
661   g_return_val_if_fail (area_size >= atom_size, NULL);
662
663   ENTER_MEM_CHUNK_ROUTINE ();
664
665   area_size = (area_size + atom_size - 1) / atom_size;
666   area_size *= atom_size;
667
668   mem_chunk = g_new (GMemChunk, 1);
669   mem_chunk->name = name;
670   mem_chunk->type = type;
671   mem_chunk->num_mem_areas = 0;
672   mem_chunk->num_marked_areas = 0;
673   mem_chunk->mem_area = NULL;
674   mem_chunk->free_mem_area = NULL;
675   mem_chunk->free_atoms = NULL;
676   mem_chunk->mem_tree = NULL;
677   mem_chunk->mem_areas = NULL;
678   mem_chunk->atom_size = atom_size;
679   
680   if (mem_chunk->type == G_ALLOC_AND_FREE)
681     mem_chunk->mem_tree = g_tree_new ((GCompareFunc) g_mem_chunk_area_compare);
682   
683   if (mem_chunk->atom_size % G_MEM_ALIGN)
684     mem_chunk->atom_size += G_MEM_ALIGN - (mem_chunk->atom_size % G_MEM_ALIGN);
685
686   rarea_size = area_size + sizeof (GMemArea) - MEM_AREA_SIZE;
687   rarea_size = g_mem_chunk_compute_size (rarea_size, atom_size + sizeof (GMemArea) - MEM_AREA_SIZE);
688   mem_chunk->area_size = rarea_size - (sizeof (GMemArea) - MEM_AREA_SIZE);
689
690   g_mutex_lock (mem_chunks_lock);
691   mem_chunk->next = mem_chunks;
692   mem_chunk->prev = NULL;
693   if (mem_chunks)
694     mem_chunks->prev = mem_chunk;
695   mem_chunks = mem_chunk;
696   g_mutex_unlock (mem_chunks_lock);
697
698   LEAVE_MEM_CHUNK_ROUTINE ();
699
700   return mem_chunk;
701 }
702
703 void
704 g_mem_chunk_destroy (GMemChunk *mem_chunk)
705 {
706   GMemArea *mem_areas;
707   GMemArea *temp_area;
708   
709   g_return_if_fail (mem_chunk != NULL);
710
711   ENTER_MEM_CHUNK_ROUTINE ();
712
713   mem_areas = mem_chunk->mem_areas;
714   while (mem_areas)
715     {
716       temp_area = mem_areas;
717       mem_areas = mem_areas->next;
718       g_free (temp_area);
719     }
720   
721   if (mem_chunk->next)
722     mem_chunk->next->prev = mem_chunk->prev;
723   if (mem_chunk->prev)
724     mem_chunk->prev->next = mem_chunk->next;
725   
726   g_mutex_lock (mem_chunks_lock);
727   if (mem_chunk == mem_chunks)
728     mem_chunks = mem_chunks->next;
729   g_mutex_unlock (mem_chunks_lock);
730   
731   if (mem_chunk->type == G_ALLOC_AND_FREE)
732     g_tree_destroy (mem_chunk->mem_tree);  
733
734   g_free (mem_chunk);
735
736   LEAVE_MEM_CHUNK_ROUTINE ();
737 }
738
739 gpointer
740 g_mem_chunk_alloc (GMemChunk *mem_chunk)
741 {
742   GMemArea *temp_area;
743   gpointer mem;
744
745   ENTER_MEM_CHUNK_ROUTINE ();
746
747   g_return_val_if_fail (mem_chunk != NULL, NULL);
748   
749   while (mem_chunk->free_atoms)
750     {
751       /* Get the first piece of memory on the "free_atoms" list.
752        * We can go ahead and destroy the list node we used to keep
753        *  track of it with and to update the "free_atoms" list to
754        *  point to its next element.
755        */
756       mem = mem_chunk->free_atoms;
757       mem_chunk->free_atoms = mem_chunk->free_atoms->next;
758       
759       /* Determine which area this piece of memory is allocated from */
760       temp_area = g_tree_search (mem_chunk->mem_tree,
761                                  (GCompareFunc) g_mem_chunk_area_search,
762                                  mem);
763       
764       /* If the area has been marked, then it is being destroyed.
765        *  (ie marked to be destroyed).
766        * We check to see if all of the segments on the free list that
767        *  reference this area have been removed. This occurs when
768        *  the ammount of free memory is less than the allocatable size.
769        * If the chunk should be freed, then we place it in the "free_mem_area".
770        * This is so we make sure not to free the mem area here and then
771        *  allocate it again a few lines down.
772        * If we don't allocate a chunk a few lines down then the "free_mem_area"
773        *  will be freed.
774        * If there is already a "free_mem_area" then we'll just free this mem area.
775        */
776       if (temp_area->mark)
777         {
778           /* Update the "free" memory available in that area */
779           temp_area->free += mem_chunk->atom_size;
780           
781           if (temp_area->free == mem_chunk->area_size)
782             {
783               if (temp_area == mem_chunk->mem_area)
784                 mem_chunk->mem_area = NULL;
785               
786               if (mem_chunk->free_mem_area)
787                 {
788                   mem_chunk->num_mem_areas -= 1;
789                   
790                   if (temp_area->next)
791                     temp_area->next->prev = temp_area->prev;
792                   if (temp_area->prev)
793                     temp_area->prev->next = temp_area->next;
794                   if (temp_area == mem_chunk->mem_areas)
795                     mem_chunk->mem_areas = mem_chunk->mem_areas->next;
796                   
797                   if (mem_chunk->type == G_ALLOC_AND_FREE)
798                     g_tree_remove (mem_chunk->mem_tree, temp_area);
799                   g_free (temp_area);
800                 }
801               else
802                 mem_chunk->free_mem_area = temp_area;
803               
804               mem_chunk->num_marked_areas -= 1;
805             }
806         }
807       else
808         {
809           /* Update the number of allocated atoms count.
810            */
811           temp_area->allocated += 1;
812           
813           /* The area wasn't marked...return the memory
814            */
815           goto outa_here;
816         }
817     }
818   
819   /* If there isn't a current mem area or the current mem area is out of space
820    *  then allocate a new mem area. We'll first check and see if we can use
821    *  the "free_mem_area". Otherwise we'll just malloc the mem area.
822    */
823   if ((!mem_chunk->mem_area) ||
824       ((mem_chunk->mem_area->index + mem_chunk->atom_size) > mem_chunk->area_size))
825     {
826       if (mem_chunk->free_mem_area)
827         {
828           mem_chunk->mem_area = mem_chunk->free_mem_area;
829           mem_chunk->free_mem_area = NULL;
830         }
831       else
832         {
833 #ifdef ENABLE_GC_FRIENDLY
834           mem_chunk->mem_area = (GMemArea*) g_malloc0 (sizeof (GMemArea) -
835                                                        MEM_AREA_SIZE +
836                                                        mem_chunk->area_size); 
837 #else /* !ENABLE_GC_FRIENDLY */
838           mem_chunk->mem_area = (GMemArea*) g_malloc (sizeof (GMemArea) -
839                                                       MEM_AREA_SIZE +
840                                                       mem_chunk->area_size);
841 #endif /* ENABLE_GC_FRIENDLY */
842           
843           mem_chunk->num_mem_areas += 1;
844           mem_chunk->mem_area->next = mem_chunk->mem_areas;
845           mem_chunk->mem_area->prev = NULL;
846           
847           if (mem_chunk->mem_areas)
848             mem_chunk->mem_areas->prev = mem_chunk->mem_area;
849           mem_chunk->mem_areas = mem_chunk->mem_area;
850           
851           if (mem_chunk->type == G_ALLOC_AND_FREE)
852             g_tree_insert (mem_chunk->mem_tree, mem_chunk->mem_area, mem_chunk->mem_area);
853         }
854       
855       mem_chunk->mem_area->index = 0;
856       mem_chunk->mem_area->free = mem_chunk->area_size;
857       mem_chunk->mem_area->allocated = 0;
858       mem_chunk->mem_area->mark = 0;
859     }
860   
861   /* Get the memory and modify the state variables appropriately.
862    */
863   mem = (gpointer) &mem_chunk->mem_area->mem[mem_chunk->mem_area->index];
864   mem_chunk->mem_area->index += mem_chunk->atom_size;
865   mem_chunk->mem_area->free -= mem_chunk->atom_size;
866   mem_chunk->mem_area->allocated += 1;
867
868 outa_here:
869
870   LEAVE_MEM_CHUNK_ROUTINE ();
871
872   return mem;
873 }
874
875 gpointer
876 g_mem_chunk_alloc0 (GMemChunk *mem_chunk)
877 {
878   gpointer mem;
879
880   mem = g_mem_chunk_alloc (mem_chunk);
881   if (mem)
882     {
883       memset (mem, 0, mem_chunk->atom_size);
884     }
885
886   return mem;
887 }
888
889 void
890 g_mem_chunk_free (GMemChunk *mem_chunk,
891                   gpointer   mem)
892 {
893   GMemArea *temp_area;
894   GFreeAtom *free_atom;
895   
896   g_return_if_fail (mem_chunk != NULL);
897   g_return_if_fail (mem != NULL);
898
899   ENTER_MEM_CHUNK_ROUTINE ();
900
901 #ifdef ENABLE_GC_FRIENDLY
902   memset (mem, 0, mem_chunk->atom_size);
903 #endif /* ENABLE_GC_FRIENDLY */
904
905   /* Don't do anything if this is an ALLOC_ONLY chunk
906    */
907   if (mem_chunk->type == G_ALLOC_AND_FREE)
908     {
909       /* Place the memory on the "free_atoms" list
910        */
911       free_atom = (GFreeAtom*) mem;
912       free_atom->next = mem_chunk->free_atoms;
913       mem_chunk->free_atoms = free_atom;
914       
915       temp_area = g_tree_search (mem_chunk->mem_tree,
916                                  (GCompareFunc) g_mem_chunk_area_search,
917                                  mem);
918       
919       temp_area->allocated -= 1;
920       
921       if (temp_area->allocated == 0)
922         {
923           temp_area->mark = 1;
924           mem_chunk->num_marked_areas += 1;
925         }
926     }
927
928   LEAVE_MEM_CHUNK_ROUTINE ();
929 }
930
931 /* This doesn't free the free_area if there is one */
932 void
933 g_mem_chunk_clean (GMemChunk *mem_chunk)
934 {
935   GMemArea *mem_area;
936   GFreeAtom *prev_free_atom;
937   GFreeAtom *temp_free_atom;
938   gpointer mem;
939   
940   g_return_if_fail (mem_chunk != NULL);
941   
942   ENTER_MEM_CHUNK_ROUTINE ();
943
944   if (mem_chunk->type == G_ALLOC_AND_FREE)
945     {
946       prev_free_atom = NULL;
947       temp_free_atom = mem_chunk->free_atoms;
948       
949       while (temp_free_atom)
950         {
951           mem = (gpointer) temp_free_atom;
952           
953           mem_area = g_tree_search (mem_chunk->mem_tree,
954                                     (GCompareFunc) g_mem_chunk_area_search,
955                                     mem);
956           
957           /* If this mem area is marked for destruction then delete the
958            *  area and list node and decrement the free mem.
959            */
960           if (mem_area->mark)
961             {
962               if (prev_free_atom)
963                 prev_free_atom->next = temp_free_atom->next;
964               else
965                 mem_chunk->free_atoms = temp_free_atom->next;
966               temp_free_atom = temp_free_atom->next;
967               
968               mem_area->free += mem_chunk->atom_size;
969               if (mem_area->free == mem_chunk->area_size)
970                 {
971                   mem_chunk->num_mem_areas -= 1;
972                   mem_chunk->num_marked_areas -= 1;
973                   
974                   if (mem_area->next)
975                     mem_area->next->prev = mem_area->prev;
976                   if (mem_area->prev)
977                     mem_area->prev->next = mem_area->next;
978                   if (mem_area == mem_chunk->mem_areas)
979                     mem_chunk->mem_areas = mem_chunk->mem_areas->next;
980                   if (mem_area == mem_chunk->mem_area)
981                     mem_chunk->mem_area = NULL;
982                   
983                   if (mem_chunk->type == G_ALLOC_AND_FREE)
984                     g_tree_remove (mem_chunk->mem_tree, mem_area);
985                   g_free (mem_area);
986                 }
987             }
988           else
989             {
990               prev_free_atom = temp_free_atom;
991               temp_free_atom = temp_free_atom->next;
992             }
993         }
994     }
995   LEAVE_MEM_CHUNK_ROUTINE ();
996 }
997
998 void
999 g_mem_chunk_reset (GMemChunk *mem_chunk)
1000 {
1001   GMemArea *mem_areas;
1002   GMemArea *temp_area;
1003   
1004   g_return_if_fail (mem_chunk != NULL);
1005   
1006   ENTER_MEM_CHUNK_ROUTINE ();
1007
1008   mem_areas = mem_chunk->mem_areas;
1009   mem_chunk->num_mem_areas = 0;
1010   mem_chunk->mem_areas = NULL;
1011   mem_chunk->mem_area = NULL;
1012   
1013   while (mem_areas)
1014     {
1015       temp_area = mem_areas;
1016       mem_areas = mem_areas->next;
1017       g_free (temp_area);
1018     }
1019   
1020   mem_chunk->free_atoms = NULL;
1021   
1022   if (mem_chunk->mem_tree)
1023     g_tree_destroy (mem_chunk->mem_tree);
1024   mem_chunk->mem_tree = g_tree_new ((GCompareFunc) g_mem_chunk_area_compare);
1025
1026   LEAVE_MEM_CHUNK_ROUTINE ();
1027 }
1028
1029 void
1030 g_mem_chunk_print (GMemChunk *mem_chunk)
1031 {
1032   GMemArea *mem_areas;
1033   gulong mem;
1034   
1035   g_return_if_fail (mem_chunk != NULL);
1036   
1037   mem_areas = mem_chunk->mem_areas;
1038   mem = 0;
1039   
1040   while (mem_areas)
1041     {
1042       mem += mem_chunk->area_size - mem_areas->free;
1043       mem_areas = mem_areas->next;
1044     }
1045
1046   g_log (g_log_domain_glib, G_LOG_LEVEL_INFO,
1047          "%s: %ld bytes using %d mem areas",
1048          mem_chunk->name, mem, mem_chunk->num_mem_areas);
1049 }
1050
1051 void
1052 g_mem_chunk_info (void)
1053 {
1054   GMemChunk *mem_chunk;
1055   gint count;
1056   
1057   count = 0;
1058   g_mutex_lock (mem_chunks_lock);
1059   mem_chunk = mem_chunks;
1060   while (mem_chunk)
1061     {
1062       count += 1;
1063       mem_chunk = mem_chunk->next;
1064     }
1065   g_mutex_unlock (mem_chunks_lock);
1066   
1067   g_log (g_log_domain_glib, G_LOG_LEVEL_INFO, "%d mem chunks", count);
1068   
1069   g_mutex_lock (mem_chunks_lock);
1070   mem_chunk = mem_chunks;
1071   g_mutex_unlock (mem_chunks_lock);
1072
1073   while (mem_chunk)
1074     {
1075       g_mem_chunk_print ((GMemChunk*) mem_chunk);
1076       mem_chunk = mem_chunk->next;
1077     }  
1078 }
1079
1080 void
1081 g_blow_chunks (void)
1082 {
1083   GMemChunk *mem_chunk;
1084   
1085   g_mutex_lock (mem_chunks_lock);
1086   mem_chunk = mem_chunks;
1087   g_mutex_unlock (mem_chunks_lock);
1088   while (mem_chunk)
1089     {
1090       g_mem_chunk_clean ((GMemChunk*) mem_chunk);
1091       mem_chunk = mem_chunk->next;
1092     }
1093 }
1094
1095 static gulong
1096 g_mem_chunk_compute_size (gulong size,
1097                           gulong min_size)
1098 {
1099   gulong power_of_2;
1100   gulong lower, upper;
1101   
1102   power_of_2 = 16;
1103   while (power_of_2 < size)
1104     power_of_2 <<= 1;
1105   
1106   lower = power_of_2 >> 1;
1107   upper = power_of_2;
1108   
1109   if (size - lower < upper - size && lower >= min_size)
1110     return lower;
1111   else
1112     return upper;
1113 }
1114
1115 static gint
1116 g_mem_chunk_area_compare (GMemArea *a,
1117                           GMemArea *b)
1118 {
1119   if (a->mem > b->mem)
1120     return 1;
1121   else if (a->mem < b->mem)
1122     return -1;
1123   return 0;
1124 }
1125
1126 static gint
1127 g_mem_chunk_area_search (GMemArea *a,
1128                          gchar    *addr)
1129 {
1130   if (a->mem <= addr)
1131     {
1132       if (addr < &a->mem[a->index])
1133         return 0;
1134       return 1;
1135     }
1136   return -1;
1137 }
1138
1139 #else /* DISABLE_MEM_POOLS */
1140
1141 typedef struct {
1142   guint alloc_size;           /* the size of an atom */
1143 }  GMinimalMemChunk;
1144
1145 GMemChunk*
1146 g_mem_chunk_new (const gchar  *name,
1147                  gint          atom_size,
1148                  gulong        area_size,
1149                  gint          type)
1150 {
1151   GMinimalMemChunk *mem_chunk;
1152
1153   g_return_val_if_fail (atom_size > 0, NULL);
1154
1155   mem_chunk = g_new (GMinimalMemChunk, 1);
1156   mem_chunk->alloc_size = atom_size;
1157
1158   return ((GMemChunk*) mem_chunk);
1159 }
1160
1161 void
1162 g_mem_chunk_destroy (GMemChunk *mem_chunk)
1163 {
1164   g_return_if_fail (mem_chunk != NULL);
1165   
1166   g_free (mem_chunk);
1167 }
1168
1169 gpointer
1170 g_mem_chunk_alloc (GMemChunk *mem_chunk)
1171 {
1172   GMinimalMemChunk *minimal = (GMinimalMemChunk *)mem_chunk;
1173   
1174   g_return_val_if_fail (mem_chunk != NULL, NULL);
1175   
1176   return g_malloc (minimal->alloc_size);
1177 }
1178
1179 gpointer
1180 g_mem_chunk_alloc0 (GMemChunk *mem_chunk)
1181 {
1182   GMinimalMemChunk *minimal = (GMinimalMemChunk *)mem_chunk;
1183   
1184   g_return_val_if_fail (mem_chunk != NULL, NULL);
1185   
1186   return g_malloc0 (minimal->alloc_size);
1187 }
1188
1189 void
1190 g_mem_chunk_free (GMemChunk *mem_chunk,
1191                   gpointer   mem)
1192 {
1193   g_return_if_fail (mem_chunk != NULL);
1194   
1195   g_free (mem);
1196 }
1197
1198 void    g_mem_chunk_clean       (GMemChunk *mem_chunk)  {}
1199 void    g_mem_chunk_reset       (GMemChunk *mem_chunk)  {}
1200 void    g_mem_chunk_print       (GMemChunk *mem_chunk)  {}
1201 void    g_mem_chunk_info        (void)                  {}
1202 void    g_blow_chunks           (void)                  {}
1203
1204 #endif /* DISABLE_MEM_POOLS */
1205
1206
1207 /* generic allocators
1208  */
1209 struct _GAllocator /* from gmem.c */
1210 {
1211   gchar         *name;
1212   guint16        n_preallocs;
1213   guint          is_unused : 1;
1214   guint          type : 4;
1215   GAllocator    *last;
1216   GMemChunk     *mem_chunk;
1217   gpointer       dummy; /* implementation specific */
1218 };
1219
1220 GAllocator*
1221 g_allocator_new (const gchar *name,
1222                  guint        n_preallocs)
1223 {
1224   GAllocator *allocator;
1225
1226   g_return_val_if_fail (name != NULL, NULL);
1227
1228   allocator = g_new0 (GAllocator, 1);
1229   allocator->name = g_strdup (name);
1230   allocator->n_preallocs = CLAMP (n_preallocs, 1, 65535);
1231   allocator->is_unused = TRUE;
1232   allocator->type = 0;
1233   allocator->last = NULL;
1234   allocator->mem_chunk = NULL;
1235   allocator->dummy = NULL;
1236
1237   return allocator;
1238 }
1239
1240 void
1241 g_allocator_free (GAllocator *allocator)
1242 {
1243   g_return_if_fail (allocator != NULL);
1244   g_return_if_fail (allocator->is_unused == TRUE);
1245
1246   g_free (allocator->name);
1247   if (allocator->mem_chunk)
1248     g_mem_chunk_destroy (allocator->mem_chunk);
1249
1250   g_free (allocator);
1251 }
1252
1253 void
1254 g_mem_init (void)
1255 {
1256 #ifndef DISABLE_MEM_POOLS
1257   mem_chunks_lock = g_mutex_new ();
1258 #endif
1259 #ifndef G_DISABLE_CHECKS
1260   mem_chunk_recursion = g_private_new (NULL);
1261   g_profile_mutex = g_mutex_new ();
1262 #endif
1263 }