add a comment
[glib.git] / glib / gslice.c
blob148827a001e4b7313b644af2105631ad61557a85
1 /* GLIB sliced memory - fast concurrent memory chunk allocator
2 * Copyright (C) 2005 Tim Janik
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.
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.
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.
19 /* MT safe */
21 #include "config.h"
23 #if defined HAVE_POSIX_MEMALIGN && defined POSIX_MEMALIGN_WITH_COMPLIANT_ALLOCS
24 # define HAVE_COMPLIANT_POSIX_MEMALIGN 1
25 #endif
27 #ifdef HAVE_COMPLIANT_POSIX_MEMALIGN
28 #define _XOPEN_SOURCE 600 /* posix_memalign() */
29 #endif
30 #include <stdlib.h> /* posix_memalign() */
31 #include <string.h>
32 #include <errno.h>
33 #include "gmem.h" /* gslice.h */
34 #include "gthreadprivate.h"
35 #include "glib.h"
36 #include "galias.h"
37 #ifdef HAVE_UNISTD_H
38 #include <unistd.h> /* sysconf() */
39 #endif
40 #ifdef G_OS_WIN32
41 #include <windows.h>
42 #include <process.h>
43 #endif
46 /* the GSlice allocator is split up into 4 layers, roughly modelled after the slab
47 * allocator and magazine extensions as outlined in:
48 * + [Bonwick94] Jeff Bonwick, The slab allocator: An object-caching kernel
49 * memory allocator. USENIX 1994, http://citeseer.ist.psu.edu/bonwick94slab.html
50 * + [Bonwick01] Bonwick and Jonathan Adams, Magazines and vmem: Extending the
51 * slab allocator to many cpu's and arbitrary resources.
52 * USENIX 2001, http://citeseer.ist.psu.edu/bonwick01magazines.html
53 * the layers are:
54 * - the thread magazines. for each (aligned) chunk size, a magazine (a list)
55 * of recently freed and soon to be allocated chunks is maintained per thread.
56 * this way, most alloc/free requests can be quickly satisfied from per-thread
57 * free lists which only require one g_private_get() call to retrive the
58 * thread handle.
59 * - the magazine cache. allocating and freeing chunks to/from threads only
60 * occours at magazine sizes from a global depot of magazines. the depot
61 * maintaines a 15 second working set of allocated magazines, so full
62 * magazines are not allocated and released too often.
63 * the chunk size dependent magazine sizes automatically adapt (within limits,
64 * see [3]) to lock contention to properly scale performance across a variety
65 * of SMP systems.
66 * - the slab allocator. this allocator allocates slabs (blocks of memory) close
67 * to the system page size or multiples thereof which have to be page aligned.
68 * the blocks are divided into smaller chunks which are used to satisfy
69 * allocations from the upper layers. the space provided by the reminder of
70 * the chunk size division is used for cache colorization (random distribution
71 * of chunk addresses) to improve processor cache utilization. multiple slabs
72 * with the same chunk size are kept in a partially sorted ring to allow O(1)
73 * freeing and allocation of chunks (as long as the allocation of an entirely
74 * new slab can be avoided).
75 * - the page allocator. on most modern systems, posix_memalign(3) or
76 * memalign(3) should be available, so this is used to allocate blocks with
77 * system page size based alignments and sizes or multiples thereof.
78 * if no memalign variant is provided, valloc() is used instead and
79 * block sizes are limited to the system page size (no multiples thereof).
80 * as a fallback, on system without even valloc(), a malloc(3)-based page
81 * allocator with alloc-only behaviour is used.
83 * NOTES:
84 * [1] some systems memalign(3) implementations may rely on boundary tagging for
85 * the handed out memory chunks. to avoid excessive page-wise fragmentation,
86 * we reserve 2 * sizeof (void*) per block size for the systems memalign(3),
87 * specified in NATIVE_MALLOC_PADDING.
88 * [2] using the slab allocator alone already provides for a fast and efficient
89 * allocator, it doesn't properly scale beyond single-threaded uses though.
90 * also, the slab allocator implements eager free(3)-ing, i.e. does not
91 * provide any form of caching or working set maintenance. so if used alone,
92 * it's vulnerable to trashing for sequences of balanced (alloc, free) pairs
93 * at certain thresholds.
94 * [3] magazine sizes are bound by an implementation specific minimum size and
95 * a chunk size specific maximum to limit magazine storage sizes to roughly
96 * 16KB.
97 * [4] allocating ca. 8 chunks per block/page keeps a good balance between
98 * external and internal fragmentation (<= 12.5%). [Bonwick94]
101 /* --- macros and constants --- */
102 #define LARGEALIGNMENT (256)
103 #define P2ALIGNMENT (2 * sizeof (gsize)) /* fits 2 pointers (assumed to be 2 * GLIB_SIZEOF_SIZE_T below) */
104 #define ALIGN(size, base) ((base) * (gsize) (((size) + (base) - 1) / (base)))
105 #define NATIVE_MALLOC_PADDING P2ALIGNMENT /* per-page padding left for native malloc(3) see [1] */
106 #define SLAB_INFO_SIZE P2ALIGN (sizeof (SlabInfo) + NATIVE_MALLOC_PADDING)
107 #define MAX_MAGAZINE_SIZE (256) /* see [3] and allocator_get_magazine_threshold() for this */
108 #define MIN_MAGAZINE_SIZE (4)
109 #define MAX_STAMP_COUNTER (7) /* distributes the load of gettimeofday() */
110 #define MAX_SLAB_CHUNK_SIZE(al) (((al)->max_page_size - SLAB_INFO_SIZE) / 8) /* we want at last 8 chunks per page, see [4] */
111 #define MAX_SLAB_INDEX(al) (SLAB_INDEX (al, MAX_SLAB_CHUNK_SIZE (al)) + 1)
112 #define SLAB_INDEX(al, asize) ((asize) / P2ALIGNMENT - 1) /* asize must be P2ALIGNMENT aligned */
113 #define SLAB_CHUNK_SIZE(al, ix) (((ix) + 1) * P2ALIGNMENT)
114 #define SLAB_BPAGE_SIZE(al,csz) (8 * (csz) + SLAB_INFO_SIZE)
116 /* optimized version of ALIGN (size, P2ALIGNMENT) */
117 #if GLIB_SIZEOF_SIZE_T * 2 == 8 /* P2ALIGNMENT */
118 #define P2ALIGN(size) (((size) + 0x7) & ~(gsize) 0x7)
119 #elif GLIB_SIZEOF_SIZE_T * 2 == 16 /* P2ALIGNMENT */
120 #define P2ALIGN(size) (((size) + 0xf) & ~(gsize) 0xf)
121 #else
122 #define P2ALIGN(size) ALIGN (size, P2ALIGNMENT)
123 #endif
125 /* special helpers to avoid gmessage.c dependency */
126 static void mem_error (const char *format, ...) G_GNUC_PRINTF (1,2);
127 #define mem_assert(cond) do { if (G_LIKELY (cond)) ; else mem_error ("assertion failed: %s", #cond); } while (0)
129 /* --- structures --- */
130 typedef struct _ChunkLink ChunkLink;
131 typedef struct _SlabInfo SlabInfo;
132 typedef struct _CachedMagazine CachedMagazine;
133 struct _ChunkLink {
134 ChunkLink *next;
135 ChunkLink *data;
137 struct _SlabInfo {
138 ChunkLink *chunks;
139 guint n_allocated;
140 SlabInfo *next, *prev;
142 typedef struct {
143 ChunkLink *chunks;
144 gsize count; /* approximative chunks list length */
145 } Magazine;
146 typedef struct {
147 Magazine *magazine1; /* array of MAX_SLAB_INDEX (allocator) */
148 Magazine *magazine2; /* array of MAX_SLAB_INDEX (allocator) */
149 } ThreadMemory;
150 typedef struct {
151 gboolean always_malloc;
152 gboolean bypass_magazines;
153 gsize working_set_msecs;
154 guint color_increment;
155 } SliceConfig;
156 typedef struct {
157 /* const after initialization */
158 gsize min_page_size, max_page_size;
159 SliceConfig config;
160 gsize max_slab_chunk_size_for_magazine_cache;
161 /* magazine cache */
162 GMutex *magazine_mutex;
163 ChunkLink **magazines; /* array of MAX_SLAB_INDEX (allocator) */
164 guint *contention_counters; /* array of MAX_SLAB_INDEX (allocator) */
165 gint mutex_counter;
166 guint stamp_counter;
167 guint last_stamp;
168 /* slab allocator */
169 GMutex *slab_mutex;
170 SlabInfo **slab_stack; /* array of MAX_SLAB_INDEX (allocator) */
171 guint color_accu;
172 } Allocator;
174 /* --- prototypes --- */
175 static gpointer slab_allocator_alloc_chunk (gsize chunk_size);
176 static void slab_allocator_free_chunk (gsize chunk_size,
177 gpointer mem);
178 static void private_thread_memory_cleanup (gpointer data);
179 static gpointer allocator_memalign (gsize alignment,
180 gsize memsize);
181 static void allocator_memfree (gsize memsize,
182 gpointer mem);
183 static inline void magazine_cache_update_stamp (void);
184 static inline gsize allocator_get_magazine_threshold (Allocator *allocator,
185 guint ix);
187 /* --- variables --- */
188 static GPrivate *private_thread_memory = NULL;
189 static gsize sys_page_size = 0;
190 static Allocator allocator[1] = { { 0, }, };
191 static SliceConfig slice_config = {
192 FALSE, /* always_malloc */
193 FALSE, /* bypass_magazines */
194 15 * 1000, /* working_set_msecs */
195 1, /* color increment, alt: 0x7fffffff */
198 /* --- auxillary funcitons --- */
199 void
200 g_slice_set_config (GSliceConfig ckey,
201 gint64 value)
203 g_return_if_fail (sys_page_size == 0);
204 switch (ckey)
206 case G_SLICE_CONFIG_ALWAYS_MALLOC:
207 slice_config.always_malloc = value != 0;
208 break;
209 case G_SLICE_CONFIG_BYPASS_MAGAZINES:
210 slice_config.bypass_magazines = value != 0;
211 break;
212 case G_SLICE_CONFIG_WORKING_SET_MSECS:
213 slice_config.working_set_msecs = value;
214 break;
215 case G_SLICE_CONFIG_COLOR_INCREMENT:
216 slice_config.color_increment = value;
217 default: ;
221 gint64
222 g_slice_get_config (GSliceConfig ckey)
224 switch (ckey)
226 case G_SLICE_CONFIG_ALWAYS_MALLOC:
227 return slice_config.always_malloc;
228 case G_SLICE_CONFIG_BYPASS_MAGAZINES:
229 return slice_config.bypass_magazines;
230 case G_SLICE_CONFIG_WORKING_SET_MSECS:
231 return slice_config.working_set_msecs;
232 case G_SLICE_CONFIG_CHUNK_SIZES:
233 return MAX_SLAB_INDEX (allocator);
234 case G_SLICE_CONFIG_COLOR_INCREMENT:
235 return slice_config.color_increment;
236 default:
237 return 0;
241 gint64*
242 g_slice_get_config_state (GSliceConfig ckey,
243 gint64 address,
244 guint *n_values)
246 guint i = 0;
247 g_return_val_if_fail (n_values != NULL, NULL);
248 *n_values = 0;
249 switch (ckey)
251 gint64 array[64];
252 case G_SLICE_CONFIG_CONTENTION_COUNTER:
253 array[i++] = SLAB_CHUNK_SIZE (allocator, address);
254 array[i++] = allocator->contention_counters[address];
255 array[i++] = allocator_get_magazine_threshold (allocator, address);
256 *n_values = i;
257 return g_memdup (array, sizeof (array[0]) * *n_values);
258 default:
259 return NULL;
263 static void
264 slice_config_init (SliceConfig *config)
266 /* don't use g_malloc/g_message here */
267 gchar buffer[1024];
268 const gchar *val = _g_getenv_nomalloc ("G_SLICE", buffer);
269 static const GDebugKey keys[] = {
270 { "always-malloc", 1 << 0 },
272 gint flags = !val ? 0 : g_parse_debug_string (val, keys, G_N_ELEMENTS (keys));
273 *config = slice_config;
274 if (flags & (1 << 0)) /* always-malloc */
276 config->always_malloc = TRUE;
280 static void
281 g_slice_init_nomessage (void)
283 /* we may not use g_error() or friends here */
284 mem_assert (sys_page_size == 0);
285 mem_assert (MIN_MAGAZINE_SIZE >= 4);
287 #ifdef G_OS_WIN32
289 SYSTEM_INFO system_info;
290 GetSystemInfo (&system_info);
291 sys_page_size = system_info.dwPageSize;
293 #else
294 sys_page_size = sysconf (_SC_PAGESIZE); /* = sysconf (_SC_PAGE_SIZE); = getpagesize(); */
295 #endif
296 mem_assert (sys_page_size >= 2 * LARGEALIGNMENT);
297 mem_assert ((sys_page_size & (sys_page_size - 1)) == 0);
298 slice_config_init (&allocator->config);
299 allocator->min_page_size = sys_page_size;
300 #if HAVE_COMPLIANT_POSIX_MEMALIGN || HAVE_MEMALIGN
301 /* allow allocation of pages up to 8KB (with 8KB alignment).
302 * this is useful because many medium to large sized structures
303 * fit less than 8 times (see [4]) into 4KB pages.
304 * we allow very small page sizes here, to reduce wastage in
305 * threads if only small allocations are required (this does
306 * bear the risk of incresing allocation times and fragmentation
307 * though).
309 allocator->min_page_size = MAX (allocator->min_page_size, 4096);
310 allocator->max_page_size = MAX (allocator->min_page_size, 8192);
311 allocator->min_page_size = MIN (allocator->min_page_size, 128);
312 #else
313 /* we can only align to system page size */
314 allocator->max_page_size = sys_page_size;
315 #endif
316 allocator->magazine_mutex = NULL; /* _g_slice_thread_init_nomessage() */
317 allocator->magazines = g_new0 (ChunkLink*, MAX_SLAB_INDEX (allocator));
318 allocator->contention_counters = g_new0 (guint, MAX_SLAB_INDEX (allocator));
319 allocator->mutex_counter = 0;
320 allocator->stamp_counter = MAX_STAMP_COUNTER; /* force initial update */
321 allocator->last_stamp = 0;
322 allocator->slab_mutex = NULL; /* _g_slice_thread_init_nomessage() */
323 allocator->slab_stack = g_new0 (SlabInfo*, MAX_SLAB_INDEX (allocator));
324 allocator->color_accu = 0;
325 magazine_cache_update_stamp();
326 /* values cached for performance reasons */
327 allocator->max_slab_chunk_size_for_magazine_cache = MAX_SLAB_CHUNK_SIZE (allocator);
328 if (allocator->config.always_malloc || allocator->config.bypass_magazines)
329 allocator->max_slab_chunk_size_for_magazine_cache = 0; /* non-optimized cases */
330 /* at this point, g_mem_gc_friendly() should be initialized, this
331 * should have been accomplished by the above g_malloc/g_new calls
335 static inline guint
336 allocator_categorize (gsize aligned_chunk_size)
338 /* speed up the likely path */
339 if (G_LIKELY (aligned_chunk_size && aligned_chunk_size <= allocator->max_slab_chunk_size_for_magazine_cache))
340 return 1; /* use magazine cache */
342 /* the above will fail (max_slab_chunk_size_for_magazine_cache == 0) if the
343 * allocator is still uninitialized, or if we are not configured to use the
344 * magazine cache.
346 if (!sys_page_size)
347 g_slice_init_nomessage ();
348 if (!allocator->config.always_malloc &&
349 aligned_chunk_size &&
350 aligned_chunk_size <= MAX_SLAB_CHUNK_SIZE (allocator))
352 if (allocator->config.bypass_magazines)
353 return 2; /* use slab allocator, see [2] */
354 return 1; /* use magazine cache */
356 return 0; /* use malloc() */
359 void
360 _g_slice_thread_init_nomessage (void)
362 /* we may not use g_error() or friends here */
363 if (!sys_page_size)
364 g_slice_init_nomessage();
365 private_thread_memory = g_private_new (private_thread_memory_cleanup);
366 allocator->magazine_mutex = g_mutex_new();
367 allocator->slab_mutex = g_mutex_new();
370 static inline void
371 g_mutex_lock_a (GMutex *mutex,
372 guint *contention_counter)
374 gboolean contention = FALSE;
375 if (!g_mutex_trylock (mutex))
377 g_mutex_lock (mutex);
378 contention = TRUE;
380 if (contention)
382 allocator->mutex_counter++;
383 if (allocator->mutex_counter >= 1) /* quickly adapt to contention */
385 allocator->mutex_counter = 0;
386 *contention_counter = MIN (*contention_counter + 1, MAX_MAGAZINE_SIZE);
389 else /* !contention */
391 allocator->mutex_counter--;
392 if (allocator->mutex_counter < -11) /* moderately recover magazine sizes */
394 allocator->mutex_counter = 0;
395 *contention_counter = MAX (*contention_counter, 1) - 1;
400 static inline ThreadMemory*
401 thread_memory_from_self (void)
403 ThreadMemory *tmem = g_private_get (private_thread_memory);
404 if (G_UNLIKELY (!tmem))
406 const guint n_magazines = MAX_SLAB_INDEX (allocator);
407 tmem = g_malloc0 (sizeof (ThreadMemory) + sizeof (Magazine) * 2 * n_magazines);
408 tmem->magazine1 = (Magazine*) (tmem + 1);
409 tmem->magazine2 = &tmem->magazine1[n_magazines];
410 g_private_set (private_thread_memory, tmem);
412 return tmem;
415 static inline ChunkLink*
416 magazine_chain_pop_head (ChunkLink **magazine_chunks)
418 /* magazine chains are linked via ChunkLink->next.
419 * each ChunkLink->data of the toplevel chain may point to a subchain,
420 * linked via ChunkLink->next. ChunkLink->data of the subchains just
421 * contains uninitialized junk.
423 ChunkLink *chunk = (*magazine_chunks)->data;
424 if (G_UNLIKELY (chunk))
426 /* allocating from freed list */
427 (*magazine_chunks)->data = chunk->next;
429 else
431 chunk = *magazine_chunks;
432 *magazine_chunks = chunk->next;
434 return chunk;
437 #if 0 /* useful for debugging */
438 static guint
439 magazine_count (ChunkLink *head)
441 guint count = 0;
442 if (!head)
443 return 0;
444 while (head)
446 ChunkLink *child = head->data;
447 count += 1;
448 for (child = head->data; child; child = child->next)
449 count += 1;
450 head = head->next;
452 return count;
454 #endif
456 static inline gsize
457 allocator_get_magazine_threshold (Allocator *allocator,
458 guint ix)
460 /* the magazine size calculated here has a lower bound of MIN_MAGAZINE_SIZE,
461 * which is required by the implementation. also, for moderately sized chunks
462 * (say >= 64 bytes), magazine sizes shouldn't be much smaller then the number
463 * of chunks available per page/2 to avoid excessive traffic in the magazine
464 * cache for small to medium sized structures.
465 * the upper bound of the magazine size is effectively provided by
466 * MAX_MAGAZINE_SIZE. for larger chunks, this number is scaled down so that
467 * the content of a single magazine doesn't exceed ca. 16KB.
469 gsize chunk_size = SLAB_CHUNK_SIZE (allocator, ix);
470 guint threshold = MAX (MIN_MAGAZINE_SIZE, allocator->max_page_size / MAX (5 * chunk_size, 5 * 32));
471 guint contention_counter = allocator->contention_counters[ix];
472 if (G_UNLIKELY (contention_counter)) /* single CPU bias */
474 /* adapt contention counter thresholds to chunk sizes */
475 contention_counter = contention_counter * 64 / chunk_size;
476 threshold = MAX (threshold, contention_counter);
478 return threshold;
481 /* --- magazine cache --- */
482 static inline void
483 magazine_cache_update_stamp (void)
485 if (allocator->stamp_counter >= MAX_STAMP_COUNTER)
487 GTimeVal tv;
488 g_get_current_time (&tv);
489 allocator->last_stamp = tv.tv_sec * 1000 + tv.tv_usec / 1000; /* milli seconds */
490 allocator->stamp_counter = 0;
492 else
493 allocator->stamp_counter++;
496 static inline ChunkLink*
497 magazine_chain_prepare_fields (ChunkLink *magazine_chunks)
499 ChunkLink *chunk1;
500 ChunkLink *chunk2;
501 ChunkLink *chunk3;
502 ChunkLink *chunk4;
503 /* checked upon initialization: mem_assert (MIN_MAGAZINE_SIZE >= 4); */
504 /* ensure a magazine with at least 4 unused data pointers */
505 chunk1 = magazine_chain_pop_head (&magazine_chunks);
506 chunk2 = magazine_chain_pop_head (&magazine_chunks);
507 chunk3 = magazine_chain_pop_head (&magazine_chunks);
508 chunk4 = magazine_chain_pop_head (&magazine_chunks);
509 chunk4->next = magazine_chunks;
510 chunk3->next = chunk4;
511 chunk2->next = chunk3;
512 chunk1->next = chunk2;
513 return chunk1;
516 /* access the first 3 fields of a specially prepared magazine chain */
517 #define magazine_chain_prev(mc) ((mc)->data)
518 #define magazine_chain_stamp(mc) ((mc)->next->data)
519 #define magazine_chain_uint_stamp(mc) GPOINTER_TO_UINT ((mc)->next->data)
520 #define magazine_chain_next(mc) ((mc)->next->next->data)
521 #define magazine_chain_count(mc) ((mc)->next->next->next->data)
523 static void
524 magazine_cache_trim (Allocator *allocator,
525 guint ix,
526 guint stamp)
528 /* g_mutex_lock (allocator->mutex); done by caller */
529 /* trim magazine cache from tail */
530 ChunkLink *current = magazine_chain_prev (allocator->magazines[ix]);
531 ChunkLink *trash = NULL;
532 while (ABS (stamp - magazine_chain_uint_stamp (current)) >= allocator->config.working_set_msecs)
534 /* unlink */
535 ChunkLink *prev = magazine_chain_prev (current);
536 ChunkLink *next = magazine_chain_next (current);
537 magazine_chain_next (prev) = next;
538 magazine_chain_prev (next) = prev;
539 /* clear special fields, put on trash stack */
540 magazine_chain_next (current) = NULL;
541 magazine_chain_count (current) = NULL;
542 magazine_chain_stamp (current) = NULL;
543 magazine_chain_prev (current) = trash;
544 trash = current;
545 /* fixup list head if required */
546 if (current == allocator->magazines[ix])
548 allocator->magazines[ix] = NULL;
549 break;
551 current = prev;
553 g_mutex_unlock (allocator->magazine_mutex);
554 /* free trash */
555 if (trash)
557 const gsize chunk_size = SLAB_CHUNK_SIZE (allocator, ix);
558 g_mutex_lock (allocator->slab_mutex);
559 while (trash)
561 current = trash;
562 trash = magazine_chain_prev (current);
563 magazine_chain_prev (current) = NULL; /* clear special field */
564 while (current)
566 ChunkLink *chunk = magazine_chain_pop_head (&current);
567 slab_allocator_free_chunk (chunk_size, chunk);
570 g_mutex_unlock (allocator->slab_mutex);
574 static void
575 magazine_cache_push_magazine (guint ix,
576 ChunkLink *magazine_chunks,
577 gsize count) /* must be >= MIN_MAGAZINE_SIZE */
579 ChunkLink *current = magazine_chain_prepare_fields (magazine_chunks);
580 ChunkLink *next, *prev;
581 g_mutex_lock (allocator->magazine_mutex);
582 /* add magazine at head */
583 next = allocator->magazines[ix];
584 if (next)
585 prev = magazine_chain_prev (next);
586 else
587 next = prev = current;
588 magazine_chain_next (prev) = current;
589 magazine_chain_prev (next) = current;
590 magazine_chain_prev (current) = prev;
591 magazine_chain_next (current) = next;
592 magazine_chain_count (current) = (gpointer) count;
593 /* stamp magazine */
594 magazine_cache_update_stamp();
595 magazine_chain_stamp (current) = GUINT_TO_POINTER (allocator->last_stamp);
596 allocator->magazines[ix] = current;
597 /* free old magazines beyond a certain threshold */
598 magazine_cache_trim (allocator, ix, allocator->last_stamp);
599 /* g_mutex_unlock (allocator->mutex); was done by magazine_cache_trim() */
602 static ChunkLink*
603 magazine_cache_pop_magazine (guint ix,
604 gsize *countp)
606 g_mutex_lock_a (allocator->magazine_mutex, &allocator->contention_counters[ix]);
607 if (!allocator->magazines[ix])
609 guint magazine_threshold = allocator_get_magazine_threshold (allocator, ix);
610 gsize i, chunk_size = SLAB_CHUNK_SIZE (allocator, ix);
611 ChunkLink *chunk, *head;
612 g_mutex_unlock (allocator->magazine_mutex);
613 g_mutex_lock (allocator->slab_mutex);
614 head = slab_allocator_alloc_chunk (chunk_size);
615 head->data = NULL;
616 chunk = head;
617 for (i = 1; i < magazine_threshold; i++)
619 chunk->next = slab_allocator_alloc_chunk (chunk_size);
620 chunk = chunk->next;
621 chunk->data = NULL;
623 chunk->next = NULL;
624 g_mutex_unlock (allocator->slab_mutex);
625 *countp = i;
626 return head;
628 else
630 ChunkLink *current = allocator->magazines[ix];
631 ChunkLink *prev = magazine_chain_prev (current);
632 ChunkLink *next = magazine_chain_next (current);
633 /* unlink */
634 magazine_chain_next (prev) = next;
635 magazine_chain_prev (next) = prev;
636 allocator->magazines[ix] = next == current ? NULL : next;
637 g_mutex_unlock (allocator->magazine_mutex);
638 /* clear special fields and hand out */
639 *countp = (gsize) magazine_chain_count (current);
640 magazine_chain_prev (current) = NULL;
641 magazine_chain_next (current) = NULL;
642 magazine_chain_count (current) = NULL;
643 magazine_chain_stamp (current) = NULL;
644 return current;
648 /* --- thread magazines --- */
649 static void
650 private_thread_memory_cleanup (gpointer data)
652 ThreadMemory *tmem = data;
653 const guint n_magazines = MAX_SLAB_INDEX (allocator);
654 guint ix;
655 for (ix = 0; ix < n_magazines; ix++)
657 Magazine *mags[2];
658 guint j;
659 mags[0] = &tmem->magazine1[ix];
660 mags[1] = &tmem->magazine2[ix];
661 for (j = 0; j < 2; j++)
663 Magazine *mag = mags[j];
664 if (mag->count >= MIN_MAGAZINE_SIZE)
665 magazine_cache_push_magazine (ix, mag->chunks, mag->count);
666 else
668 const gsize chunk_size = SLAB_CHUNK_SIZE (allocator, ix);
669 g_mutex_lock (allocator->slab_mutex);
670 while (mag->chunks)
672 ChunkLink *chunk = magazine_chain_pop_head (&mag->chunks);
673 slab_allocator_free_chunk (chunk_size, chunk);
675 g_mutex_unlock (allocator->slab_mutex);
679 g_free (tmem);
682 static void
683 thread_memory_magazine1_reload (ThreadMemory *tmem,
684 guint ix)
686 Magazine *mag = &tmem->magazine1[ix];
687 mem_assert (mag->chunks == NULL); /* ensure that we may reset mag->count */
688 mag->count = 0;
689 mag->chunks = magazine_cache_pop_magazine (ix, &mag->count);
692 static void
693 thread_memory_magazine2_unload (ThreadMemory *tmem,
694 guint ix)
696 Magazine *mag = &tmem->magazine2[ix];
697 magazine_cache_push_magazine (ix, mag->chunks, mag->count);
698 mag->chunks = NULL;
699 mag->count = 0;
702 static inline void
703 thread_memory_swap_magazines (ThreadMemory *tmem,
704 guint ix)
706 Magazine xmag = tmem->magazine1[ix];
707 tmem->magazine1[ix] = tmem->magazine2[ix];
708 tmem->magazine2[ix] = xmag;
711 static inline gboolean
712 thread_memory_magazine1_is_empty (ThreadMemory *tmem,
713 guint ix)
715 return tmem->magazine1[ix].chunks == NULL;
718 static inline gboolean
719 thread_memory_magazine2_is_full (ThreadMemory *tmem,
720 guint ix)
722 return tmem->magazine2[ix].count >= allocator_get_magazine_threshold (allocator, ix);
725 static inline gpointer
726 thread_memory_magazine1_alloc (ThreadMemory *tmem,
727 guint ix)
729 Magazine *mag = &tmem->magazine1[ix];
730 ChunkLink *chunk = magazine_chain_pop_head (&mag->chunks);
731 if (G_LIKELY (mag->count > 0))
732 mag->count--;
733 return chunk;
736 static inline void
737 thread_memory_magazine2_free (ThreadMemory *tmem,
738 guint ix,
739 gpointer mem)
741 Magazine *mag = &tmem->magazine2[ix];
742 ChunkLink *chunk = mem;
743 chunk->data = NULL;
744 chunk->next = mag->chunks;
745 mag->chunks = chunk;
746 mag->count++;
749 /* --- API functions --- */
750 gpointer
751 g_slice_alloc (gsize mem_size)
753 gsize chunk_size;
754 gpointer mem;
755 guint acat;
756 chunk_size = P2ALIGN (mem_size);
757 acat = allocator_categorize (chunk_size);
758 if (G_LIKELY (acat == 1)) /* allocate through magazine layer */
760 ThreadMemory *tmem = thread_memory_from_self();
761 guint ix = SLAB_INDEX (allocator, chunk_size);
762 if (G_UNLIKELY (thread_memory_magazine1_is_empty (tmem, ix)))
764 thread_memory_swap_magazines (tmem, ix);
765 if (G_UNLIKELY (thread_memory_magazine1_is_empty (tmem, ix)))
766 thread_memory_magazine1_reload (tmem, ix);
768 mem = thread_memory_magazine1_alloc (tmem, ix);
770 else if (acat == 2) /* allocate through slab allocator */
772 g_mutex_lock (allocator->slab_mutex);
773 mem = slab_allocator_alloc_chunk (chunk_size);
774 g_mutex_unlock (allocator->slab_mutex);
776 else /* delegate to system malloc */
777 mem = g_malloc (mem_size);
778 return mem;
781 gpointer
782 g_slice_alloc0 (gsize mem_size)
784 gpointer mem = g_slice_alloc (mem_size);
785 if (mem)
786 memset (mem, 0, mem_size);
787 return mem;
790 void
791 g_slice_free1 (gsize mem_size,
792 gpointer mem_block)
794 gsize chunk_size = P2ALIGN (mem_size);
795 guint acat = allocator_categorize (chunk_size);
796 if (G_UNLIKELY (!mem_block))
797 return;
798 if (G_LIKELY (acat == 1)) /* allocate through magazine layer */
800 ThreadMemory *tmem = thread_memory_from_self();
801 guint ix = SLAB_INDEX (allocator, chunk_size);
802 if (G_UNLIKELY (thread_memory_magazine2_is_full (tmem, ix)))
804 thread_memory_swap_magazines (tmem, ix);
805 if (G_UNLIKELY (thread_memory_magazine2_is_full (tmem, ix)))
806 thread_memory_magazine2_unload (tmem, ix);
808 if (G_UNLIKELY (g_mem_gc_friendly))
809 memset (mem_block, 0, chunk_size);
810 thread_memory_magazine2_free (tmem, ix, mem_block);
812 else if (acat == 2) /* allocate through slab allocator */
814 if (G_UNLIKELY (g_mem_gc_friendly))
815 memset (mem_block, 0, chunk_size);
816 g_mutex_lock (allocator->slab_mutex);
817 slab_allocator_free_chunk (chunk_size, mem_block);
818 g_mutex_unlock (allocator->slab_mutex);
820 else /* delegate to system malloc */
822 if (G_UNLIKELY (g_mem_gc_friendly))
823 memset (mem_block, 0, mem_size);
824 g_free (mem_block);
828 void
829 g_slice_free_chain_with_offset (gsize mem_size,
830 gpointer mem_chain,
831 gsize next_offset)
833 gpointer slice = mem_chain;
834 /* while the thread magazines and the magazine cache are implemented so that
835 * they can easily be extended to allow for free lists containing more free
836 * lists for the first level nodes, which would allow O(1) freeing in this
837 * function, the benefit of such an extension is questionable, because:
838 * - the magazine size counts will become mere lower bounds which confuses
839 * the code adapting to lock contention;
840 * - freeing a single node to the thread magazines is very fast, so this
841 * O(list_length) operation is multiplied by a fairly small factor;
842 * - memory usage histograms on larger applications seem to indicate that
843 * the amount of released multi node lists is negligible in comparison
844 * to single node releases.
845 * - the major performance bottle neck, namely g_private_get() or
846 * g_mutex_lock()/g_mutex_unlock() has already been moved out of the
847 * inner loop for freeing chained slices.
849 gsize chunk_size = P2ALIGN (mem_size);
850 guint acat = allocator_categorize (chunk_size);
851 if (G_LIKELY (acat == 1)) /* allocate through magazine layer */
853 ThreadMemory *tmem = thread_memory_from_self();
854 guint ix = SLAB_INDEX (allocator, chunk_size);
855 while (slice)
857 guint8 *current = slice;
858 slice = *(gpointer*) (current + next_offset);
859 if (G_UNLIKELY (thread_memory_magazine2_is_full (tmem, ix)))
861 thread_memory_swap_magazines (tmem, ix);
862 if (G_UNLIKELY (thread_memory_magazine2_is_full (tmem, ix)))
863 thread_memory_magazine2_unload (tmem, ix);
865 if (G_UNLIKELY (g_mem_gc_friendly))
866 memset (current, 0, chunk_size);
867 thread_memory_magazine2_free (tmem, ix, current);
870 else if (acat == 2) /* allocate through slab allocator */
872 g_mutex_lock (allocator->slab_mutex);
873 while (slice)
875 guint8 *current = slice;
876 slice = *(gpointer*) (current + next_offset);
877 if (G_UNLIKELY (g_mem_gc_friendly))
878 memset (current, 0, chunk_size);
879 slab_allocator_free_chunk (chunk_size, current);
881 g_mutex_unlock (allocator->slab_mutex);
883 else /* delegate to system malloc */
884 while (slice)
886 guint8 *current = slice;
887 slice = *(gpointer*) (current + next_offset);
888 if (G_UNLIKELY (g_mem_gc_friendly))
889 memset (current, 0, mem_size);
890 g_free (current);
894 /* --- single page allocator --- */
895 static void
896 allocator_slab_stack_push (Allocator *allocator,
897 guint ix,
898 SlabInfo *sinfo)
900 /* insert slab at slab ring head */
901 if (!allocator->slab_stack[ix])
903 sinfo->next = sinfo;
904 sinfo->prev = sinfo;
906 else
908 SlabInfo *next = allocator->slab_stack[ix], *prev = next->prev;
909 next->prev = sinfo;
910 prev->next = sinfo;
911 sinfo->next = next;
912 sinfo->prev = prev;
914 allocator->slab_stack[ix] = sinfo;
917 static gsize
918 allocator_aligned_page_size (Allocator *allocator,
919 gsize n_bytes)
921 gsize val = 1 << g_bit_storage (n_bytes - 1);
922 val = MAX (val, allocator->min_page_size);
923 return val;
926 static void
927 allocator_add_slab (Allocator *allocator,
928 guint ix,
929 gsize chunk_size)
931 ChunkLink *chunk;
932 SlabInfo *sinfo;
933 gsize addr, padding, n_chunks, color = 0;
934 gsize page_size = allocator_aligned_page_size (allocator, SLAB_BPAGE_SIZE (allocator, chunk_size));
935 /* allocate 1 page for the chunks and the slab */
936 gpointer aligned_memory = allocator_memalign (page_size, page_size - NATIVE_MALLOC_PADDING);
937 guint8 *mem = aligned_memory;
938 guint i;
939 if (!mem)
941 const gchar *syserr = "unknown error";
942 #if HAVE_STRERROR
943 syserr = strerror (errno);
944 #endif
945 mem_error ("failed to allocate %u bytes (alignment: %u): %s\n",
946 (guint) (page_size - NATIVE_MALLOC_PADDING), (guint) page_size, syserr);
948 /* mask page adress */
949 addr = ((gsize) mem / page_size) * page_size;
950 /* assert alignment */
951 mem_assert (aligned_memory == (gpointer) addr);
952 /* basic slab info setup */
953 sinfo = (SlabInfo*) (mem + page_size - SLAB_INFO_SIZE);
954 sinfo->n_allocated = 0;
955 sinfo->chunks = NULL;
956 /* figure cache colorization */
957 n_chunks = ((guint8*) sinfo - mem) / chunk_size;
958 padding = ((guint8*) sinfo - mem) - n_chunks * chunk_size;
959 if (padding)
961 color = (allocator->color_accu * P2ALIGNMENT) % padding;
962 allocator->color_accu += allocator->config.color_increment;
964 /* add chunks to free list */
965 chunk = (ChunkLink*) (mem + color);
966 sinfo->chunks = chunk;
967 for (i = 0; i < n_chunks - 1; i++)
969 chunk->next = (ChunkLink*) ((guint8*) chunk + chunk_size);
970 chunk = chunk->next;
972 chunk->next = NULL; /* last chunk */
973 /* add slab to slab ring */
974 allocator_slab_stack_push (allocator, ix, sinfo);
977 static gpointer
978 slab_allocator_alloc_chunk (gsize chunk_size)
980 ChunkLink *chunk;
981 guint ix = SLAB_INDEX (allocator, chunk_size);
982 /* ensure non-empty slab */
983 if (!allocator->slab_stack[ix] || !allocator->slab_stack[ix]->chunks)
984 allocator_add_slab (allocator, ix, chunk_size);
985 /* allocate chunk */
986 chunk = allocator->slab_stack[ix]->chunks;
987 allocator->slab_stack[ix]->chunks = chunk->next;
988 allocator->slab_stack[ix]->n_allocated++;
989 /* rotate empty slabs */
990 if (!allocator->slab_stack[ix]->chunks)
991 allocator->slab_stack[ix] = allocator->slab_stack[ix]->next;
992 return chunk;
995 static void
996 slab_allocator_free_chunk (gsize chunk_size,
997 gpointer mem)
999 ChunkLink *chunk;
1000 gboolean was_empty;
1001 guint ix = SLAB_INDEX (allocator, chunk_size);
1002 gsize page_size = allocator_aligned_page_size (allocator, SLAB_BPAGE_SIZE (allocator, chunk_size));
1003 gsize addr = ((gsize) mem / page_size) * page_size;
1004 /* mask page adress */
1005 guint8 *page = (guint8*) addr;
1006 SlabInfo *sinfo = (SlabInfo*) (page + page_size - SLAB_INFO_SIZE);
1007 /* assert valid chunk count */
1008 mem_assert (sinfo->n_allocated > 0);
1009 /* add chunk to free list */
1010 was_empty = sinfo->chunks == NULL;
1011 chunk = (ChunkLink*) mem;
1012 chunk->next = sinfo->chunks;
1013 sinfo->chunks = chunk;
1014 sinfo->n_allocated--;
1015 /* keep slab ring partially sorted, empty slabs at end */
1016 if (was_empty)
1018 /* unlink slab */
1019 SlabInfo *next = sinfo->next, *prev = sinfo->prev;
1020 next->prev = prev;
1021 prev->next = next;
1022 if (allocator->slab_stack[ix] == sinfo)
1023 allocator->slab_stack[ix] = next == sinfo ? NULL : next;
1024 /* insert slab at head */
1025 allocator_slab_stack_push (allocator, ix, sinfo);
1027 /* eagerly free complete unused slabs */
1028 if (!sinfo->n_allocated)
1030 /* unlink slab */
1031 SlabInfo *next = sinfo->next, *prev = sinfo->prev;
1032 next->prev = prev;
1033 prev->next = next;
1034 if (allocator->slab_stack[ix] == sinfo)
1035 allocator->slab_stack[ix] = next == sinfo ? NULL : next;
1036 /* free slab */
1037 allocator_memfree (page_size, page);
1041 /* --- memalign implementation --- */
1042 #ifdef HAVE_MALLOC_H
1043 #include <malloc.h> /* memalign() */
1044 #endif
1046 /* from config.h:
1047 * define HAVE_POSIX_MEMALIGN 1 // if free(posix_memalign(3)) works, <stdlib.h>
1048 * define HAVE_COMPLIANT_POSIX_MEMALIGN 1 // if free(posix_memalign(3)) works for sizes != 2^n, <stdlib.h>
1049 * define HAVE_MEMALIGN 1 // if free(memalign(3)) works, <malloc.h>
1050 * define HAVE_VALLOC 1 // if free(valloc(3)) works, <stdlib.h> or <malloc.h>
1051 * if none is provided, we implement malloc(3)-based alloc-only page alignment
1054 #if !(HAVE_COMPLIANT_POSIX_MEMALIGN || HAVE_MEMALIGN || HAVE_VALLOC)
1055 static GTrashStack *compat_valloc_trash = NULL;
1056 #endif
1058 static gpointer
1059 allocator_memalign (gsize alignment,
1060 gsize memsize)
1062 gpointer aligned_memory = NULL;
1063 gint err = ENOMEM;
1064 #if HAVE_COMPLIANT_POSIX_MEMALIGN
1065 err = posix_memalign (&aligned_memory, alignment, memsize);
1066 #elif HAVE_MEMALIGN
1067 errno = 0;
1068 aligned_memory = memalign (alignment, memsize);
1069 err = errno;
1070 #elif HAVE_VALLOC
1071 errno = 0;
1072 aligned_memory = valloc (memsize);
1073 err = errno;
1074 #else
1075 /* simplistic non-freeing page allocator */
1076 mem_assert (alignment == sys_page_size);
1077 mem_assert (memsize <= sys_page_size);
1078 if (!compat_valloc_trash)
1080 const guint n_pages = 16;
1081 guint8 *mem = malloc (n_pages * sys_page_size);
1082 err = errno;
1083 if (mem)
1085 gint i = n_pages;
1086 guint8 *amem = (guint8*) ALIGN ((gsize) mem, sys_page_size);
1087 if (amem != mem)
1088 i--; /* mem wasn't page aligned */
1089 while (--i >= 0)
1090 g_trash_stack_push (&compat_valloc_trash, amem + i * sys_page_size);
1093 aligned_memory = g_trash_stack_pop (&compat_valloc_trash);
1094 #endif
1095 if (!aligned_memory)
1096 errno = err;
1097 return aligned_memory;
1100 static void
1101 allocator_memfree (gsize memsize,
1102 gpointer mem)
1104 #if HAVE_COMPLIANT_POSIX_MEMALIGN || HAVE_MEMALIGN || HAVE_VALLOC
1105 free (mem);
1106 #else
1107 mem_assert (memsize <= sys_page_size);
1108 g_trash_stack_push (&compat_valloc_trash, mem);
1109 #endif
1112 #include <stdio.h>
1114 static void
1115 mem_error (const char *format,
1116 ...)
1118 const char *pname;
1119 va_list args;
1120 /* at least, put out "MEMORY-ERROR", in case we segfault during the rest of the function */
1121 fputs ("\n***MEMORY-ERROR***: ", stderr);
1122 pname = g_get_prgname();
1123 fprintf (stderr, "%s[%u]: GSlice: ", pname ? pname : "", getpid());
1124 va_start (args, format);
1125 vfprintf (stderr, format, args);
1126 va_end (args);
1127 fputs ("\n", stderr);
1128 _exit (1);
1131 #define __G_SLICE_C__
1132 #include "galiasdef.c"