1/*
2   +----------------------------------------------------------------------+
3   | Zend Engine                                                          |
4   +----------------------------------------------------------------------+
5   | Copyright (c) 1998-2015 Zend Technologies Ltd. (http://www.zend.com) |
6   +----------------------------------------------------------------------+
7   | This source file is subject to version 2.00 of the Zend license,     |
8   | that is bundled with this package in the file LICENSE, and is        |
9   | available through the world-wide-web at the following url:           |
10   | http://www.zend.com/license/2_00.txt.                                |
11   | If you did not receive a copy of the Zend license and are unable to  |
12   | obtain it through the world-wide-web, please send a note to          |
13   | license@zend.com so we can mail you a copy immediately.              |
14   +----------------------------------------------------------------------+
15   | Authors: Andi Gutmans <andi@zend.com>                                |
16   |          Zeev Suraski <zeev@zend.com>                                |
17   |          Dmitry Stogov <dmitry@zend.com>                             |
18   +----------------------------------------------------------------------+
19*/
20
21/* $Id$ */
22
23#ifndef ZEND_ALLOC_H
24#define ZEND_ALLOC_H
25
26#include <stdio.h>
27
28#include "../TSRM/TSRM.h"
29#include "zend.h"
30
31#ifndef ZEND_MM_ALIGNMENT
32# define ZEND_MM_ALIGNMENT Z_L(8)
33# define ZEND_MM_ALIGNMENT_LOG2 Z_L(3)
34#elif ZEND_MM_ALIGNMENT < 4
35# undef ZEND_MM_ALIGNMENT
36# undef ZEND_MM_ALIGNMENT_LOG2
37# define ZEND_MM_ALIGNMENT Z_L(4)
38# define ZEND_MM_ALIGNMENT_LOG2 Z_L(2)
39#endif
40
41#define ZEND_MM_ALIGNMENT_MASK ~(ZEND_MM_ALIGNMENT - Z_L(1))
42
43#define ZEND_MM_ALIGNED_SIZE(size)  (((size) + ZEND_MM_ALIGNMENT - Z_L(1)) & ZEND_MM_ALIGNMENT_MASK)
44
45#define ZEND_MM_ALIGNED_SIZE_EX(size, alignment) \
46    (((size) + ((alignment) - Z_L(1))) & ~((alignment) - Z_L(1)))
47
48typedef struct _zend_leak_info {
49    void *addr;
50    size_t size;
51    const char *filename;
52    const char *orig_filename;
53    uint lineno;
54    uint orig_lineno;
55} zend_leak_info;
56
57#if ZEND_DEBUG
58typedef struct _zend_mm_debug_info {
59    size_t             size;
60    const char        *filename;
61    const char        *orig_filename;
62    uint               lineno;
63    uint               orig_lineno;
64} zend_mm_debug_info;
65
66# define ZEND_MM_OVERHEAD ZEND_MM_ALIGNED_SIZE(sizeof(zend_mm_debug_info))
67#else
68# define ZEND_MM_OVERHEAD 0
69#endif
70
71BEGIN_EXTERN_C()
72
73ZEND_API char*  ZEND_FASTCALL zend_strndup(const char *s, size_t length) ZEND_ATTRIBUTE_MALLOC;
74
75ZEND_API void*  ZEND_FASTCALL _emalloc(size_t size ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC) ZEND_ATTRIBUTE_MALLOC ZEND_ATTRIBUTE_ALLOC_SIZE(1);
76ZEND_API void*  ZEND_FASTCALL _safe_emalloc(size_t nmemb, size_t size, size_t offset ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC) ZEND_ATTRIBUTE_MALLOC;
77ZEND_API void*  ZEND_FASTCALL _safe_malloc(size_t nmemb, size_t size, size_t offset) ZEND_ATTRIBUTE_MALLOC;
78ZEND_API void   ZEND_FASTCALL _efree(void *ptr ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC);
79ZEND_API void*  ZEND_FASTCALL _ecalloc(size_t nmemb, size_t size ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC) ZEND_ATTRIBUTE_MALLOC ZEND_ATTRIBUTE_ALLOC_SIZE2(1,2);
80ZEND_API void*  ZEND_FASTCALL _erealloc(void *ptr, size_t size ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC) ZEND_ATTRIBUTE_ALLOC_SIZE(2);
81ZEND_API void*  ZEND_FASTCALL _erealloc2(void *ptr, size_t size, size_t copy_size ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC) ZEND_ATTRIBUTE_ALLOC_SIZE(2);
82ZEND_API void*  ZEND_FASTCALL _safe_erealloc(void *ptr, size_t nmemb, size_t size, size_t offset ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC);
83ZEND_API void*  ZEND_FASTCALL _safe_realloc(void *ptr, size_t nmemb, size_t size, size_t offset);
84ZEND_API char*  ZEND_FASTCALL _estrdup(const char *s ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC) ZEND_ATTRIBUTE_MALLOC;
85ZEND_API char*  ZEND_FASTCALL _estrndup(const char *s, size_t length ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC) ZEND_ATTRIBUTE_MALLOC;
86ZEND_API size_t ZEND_FASTCALL _zend_mem_block_size(void *ptr ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC);
87
88#include "zend_alloc_sizes.h"
89
90/* _emalloc() & _efree() specialization */
91#if !ZEND_DEBUG && defined(HAVE_BUILTIN_CONSTANT_P)
92
93# define _ZEND_BIN_ALLOCATOR_DEF(_num, _size, _elements, _pages, x, y) \
94    ZEND_API void* ZEND_FASTCALL _emalloc_  ## _size(void) ZEND_ATTRIBUTE_MALLOC;
95
96ZEND_MM_BINS_INFO(_ZEND_BIN_ALLOCATOR_DEF, x, y)
97
98ZEND_API void* ZEND_FASTCALL _emalloc_large(size_t size) ZEND_ATTRIBUTE_MALLOC ZEND_ATTRIBUTE_ALLOC_SIZE(1);
99ZEND_API void* ZEND_FASTCALL _emalloc_huge(size_t size) ZEND_ATTRIBUTE_MALLOC ZEND_ATTRIBUTE_ALLOC_SIZE(1);
100
101# define _ZEND_BIN_ALLOCATOR_SELECTOR_START(_num, _size, _elements, _pages, size, y) \
102    ((size <= _size) ? _emalloc_ ## _size() :
103# define _ZEND_BIN_ALLOCATOR_SELECTOR_END(_num, _size, _elements, _pages, size, y) \
104    )
105
106# define ZEND_ALLOCATOR(size) \
107    ZEND_MM_BINS_INFO(_ZEND_BIN_ALLOCATOR_SELECTOR_START, size, y) \
108    ((size <= ZEND_MM_MAX_LARGE_SIZE) ? _emalloc_large(size) : _emalloc_huge(size)) \
109    ZEND_MM_BINS_INFO(_ZEND_BIN_ALLOCATOR_SELECTOR_END, size, y)
110
111# define _emalloc(size) \
112    (__builtin_constant_p(size) ? \
113        ZEND_ALLOCATOR(size) \
114    : \
115        _emalloc(size) \
116    )
117
118# define _ZEND_BIN_DEALLOCATOR_DEF(_num, _size, _elements, _pages, x, y) \
119    ZEND_API void ZEND_FASTCALL _efree_ ## _size(void *);
120
121ZEND_MM_BINS_INFO(_ZEND_BIN_DEALLOCATOR_DEF, x, y)
122
123ZEND_API void ZEND_FASTCALL _efree_large(void *, size_t size);
124ZEND_API void ZEND_FASTCALL _efree_huge(void *, size_t size);
125
126# define _ZEND_BIN_DEALLOCATOR_SELECTOR_START(_num, _size, _elements, _pages, ptr, size) \
127    if (size <= _size) { _efree_ ## _size(ptr); } else
128
129# define ZEND_DEALLOCATOR(ptr, size) \
130    ZEND_MM_BINS_INFO(_ZEND_BIN_DEALLOCATOR_SELECTOR_START, ptr, size) \
131    if (size <= ZEND_MM_MAX_LARGE_SIZE) { _efree_large(ptr, size); } \
132    else { _efree_huge(ptr, size); }
133
134# define efree_size(ptr, size) do { \
135        if (__builtin_constant_p(size)) { \
136            ZEND_DEALLOCATOR(ptr, size) \
137        } else { \
138            _efree(ptr); \
139        } \
140    } while (0)
141# define efree_size_rel(ptr, size) \
142    efree_size(ptr, size)
143
144#else
145
146# define efree_size(ptr, size) \
147    efree(ptr)
148# define efree_size_rel(ptr, size) \
149    efree_rel(ptr)
150
151#define _emalloc_large _emalloc
152#define _emalloc_huge  _emalloc
153#define _efree_large   _efree
154#define _efree_huge    _efree
155
156#endif
157
158/* Standard wrapper macros */
159#define emalloc(size)                       _emalloc((size) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
160#define emalloc_large(size)                 _emalloc_large((size) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
161#define emalloc_huge(size)                  _emalloc_huge((size) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
162#define safe_emalloc(nmemb, size, offset)   _safe_emalloc((nmemb), (size), (offset) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
163#define efree(ptr)                          _efree((ptr) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
164#define efree_large(ptr)                    _efree_large((ptr) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
165#define efree_huge(ptr)                     _efree_huge((ptr) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
166#define ecalloc(nmemb, size)                _ecalloc((nmemb), (size) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
167#define erealloc(ptr, size)                 _erealloc((ptr), (size) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
168#define erealloc2(ptr, size, copy_size)     _erealloc2((ptr), (size), (copy_size) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
169#define safe_erealloc(ptr, nmemb, size, offset) _safe_erealloc((ptr), (nmemb), (size), (offset) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
170#define erealloc_recoverable(ptr, size)     _erealloc((ptr), (size) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
171#define erealloc2_recoverable(ptr, size, copy_size) _erealloc2((ptr), (size), (copy_size) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
172#define estrdup(s)                          _estrdup((s) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
173#define estrndup(s, length)                 _estrndup((s), (length) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
174#define zend_mem_block_size(ptr)            _zend_mem_block_size((ptr) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
175
176/* Relay wrapper macros */
177#define emalloc_rel(size)                       _emalloc((size) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
178#define safe_emalloc_rel(nmemb, size, offset)   _safe_emalloc((nmemb), (size), (offset) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
179#define efree_rel(ptr)                          _efree((ptr) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
180#define ecalloc_rel(nmemb, size)                _ecalloc((nmemb), (size) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
181#define erealloc_rel(ptr, size)                 _erealloc((ptr), (size) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
182#define erealloc2_rel(ptr, size, copy_size)     _erealloc2((ptr), (size), (copy_size) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
183#define erealloc_recoverable_rel(ptr, size)     _erealloc((ptr), (size) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
184#define erealloc2_recoverable_rel(ptr, size, copy_size) _erealloc2((ptr), (size), (copy_size) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
185#define safe_erealloc_rel(ptr, nmemb, size, offset) _safe_erealloc((ptr), (nmemb), (size), (offset) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
186#define estrdup_rel(s)                          _estrdup((s) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
187#define estrndup_rel(s, length)                 _estrndup((s), (length) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
188#define zend_mem_block_size_rel(ptr)            _zend_mem_block_size((ptr) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
189
190zend_always_inline static void * __zend_malloc(size_t len)
191{
192    void *tmp = malloc(len);
193    if (tmp) {
194        return tmp;
195    }
196    fprintf(stderr, "Out of memory\n");
197    exit(1);
198}
199
200zend_always_inline static void * __zend_calloc(size_t nmemb, size_t len)
201{
202    void *tmp = _safe_malloc(nmemb, len, 0);
203    memset(tmp, 0, nmemb * len);
204    return tmp;
205}
206
207zend_always_inline static void * __zend_realloc(void *p, size_t len)
208{
209    p = realloc(p, len);
210    if (p) {
211        return p;
212    }
213    fprintf(stderr, "Out of memory\n");
214    exit(1);
215}
216
217
218/* Selective persistent/non persistent allocation macros */
219#define pemalloc(size, persistent) ((persistent)?__zend_malloc(size):emalloc(size))
220#define safe_pemalloc(nmemb, size, offset, persistent)  ((persistent)?_safe_malloc(nmemb, size, offset):safe_emalloc(nmemb, size, offset))
221#define pefree(ptr, persistent)  ((persistent)?free(ptr):efree(ptr))
222#define pefree_size(ptr, size, persistent)  ((persistent)?free(ptr):efree_size(ptr, size))
223#define pecalloc(nmemb, size, persistent) ((persistent)?__zend_calloc((nmemb), (size)):ecalloc((nmemb), (size)))
224#define perealloc(ptr, size, persistent) ((persistent)?__zend_realloc((ptr), (size)):erealloc((ptr), (size)))
225#define perealloc2(ptr, size, copy_size, persistent) ((persistent)?__zend_realloc((ptr), (size)):erealloc2((ptr), (size), (copy_size)))
226#define safe_perealloc(ptr, nmemb, size, offset, persistent)    ((persistent)?_safe_realloc((ptr), (nmemb), (size), (offset)):safe_erealloc((ptr), (nmemb), (size), (offset)))
227#define perealloc_recoverable(ptr, size, persistent) ((persistent)?realloc((ptr), (size)):erealloc_recoverable((ptr), (size)))
228#define perealloc2_recoverable(ptr, size, persistent) ((persistent)?realloc((ptr), (size)):erealloc2_recoverable((ptr), (size), (copy_size)))
229#define pestrdup(s, persistent) ((persistent)?strdup(s):estrdup(s))
230#define pestrndup(s, length, persistent) ((persistent)?zend_strndup((s),(length)):estrndup((s),(length)))
231
232#define pemalloc_rel(size, persistent) ((persistent)?__zend_malloc(size):emalloc_rel(size))
233#define pefree_rel(ptr, persistent) ((persistent)?free(ptr):efree_rel(ptr))
234#define pecalloc_rel(nmemb, size, persistent) ((persistent)?__zend_calloc((nmemb), (size)):ecalloc_rel((nmemb), (size)))
235#define perealloc_rel(ptr, size, persistent) ((persistent)?__zend_realloc((ptr), (size)):erealloc_rel((ptr), (size)))
236#define perealloc2_rel(ptr, size, copy_size, persistent) ((persistent)?__zend_realloc((ptr), (size)):erealloc2_rel((ptr), (size), (copy_size)))
237#define perealloc_recoverable_rel(ptr, size, persistent) ((persistent)?realloc((ptr), (size)):erealloc_recoverable_rel((ptr), (size)))
238#define perealloc2_recoverable_rel(ptr, size, copy_size, persistent) ((persistent)?realloc((ptr), (size)):erealloc2_recoverable_rel((ptr), (size), (copy_size)))
239#define pestrdup_rel(s, persistent) ((persistent)?strdup(s):estrdup_rel(s))
240
241ZEND_API int zend_set_memory_limit(size_t memory_limit);
242
243ZEND_API void start_memory_manager(void);
244ZEND_API void shutdown_memory_manager(int silent, int full_shutdown);
245ZEND_API int is_zend_mm(void);
246
247ZEND_API size_t zend_memory_usage(int real_usage);
248ZEND_API size_t zend_memory_peak_usage(int real_usage);
249
250/* fast cache for HashTables */
251#define ALLOC_HASHTABLE(ht) \
252    (ht) = (HashTable *) emalloc(sizeof(HashTable))
253
254#define FREE_HASHTABLE(ht)  \
255    efree_size(ht, sizeof(HashTable))
256
257#define ALLOC_HASHTABLE_REL(ht) \
258    (ht) = (HashTable *) emalloc_rel(sizeof(HashTable))
259
260#define FREE_HASHTABLE_REL(ht)  \
261    efree_size_rel(ht, sizeof(HashTable))
262
263/* Heap functions */
264typedef struct _zend_mm_heap zend_mm_heap;
265
266ZEND_API zend_mm_heap *zend_mm_startup(void);
267ZEND_API void zend_mm_shutdown(zend_mm_heap *heap, int full_shutdown, int silent);
268ZEND_API void*  ZEND_FASTCALL _zend_mm_alloc(zend_mm_heap *heap, size_t size ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC) ZEND_ATTRIBUTE_MALLOC;
269ZEND_API void   ZEND_FASTCALL _zend_mm_free(zend_mm_heap *heap, void *p ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC);
270ZEND_API void*  ZEND_FASTCALL _zend_mm_realloc(zend_mm_heap *heap, void *p, size_t size ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC);
271ZEND_API void*  ZEND_FASTCALL _zend_mm_realloc2(zend_mm_heap *heap, void *p, size_t size, size_t copy_size ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC);
272ZEND_API size_t ZEND_FASTCALL _zend_mm_block_size(zend_mm_heap *heap, void *p ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC);
273
274#define zend_mm_alloc(heap, size)           _zend_mm_alloc((heap), (size) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
275#define zend_mm_free(heap, p)               _zend_mm_free((heap), (p) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
276#define zend_mm_realloc(heap, p, size)      _zend_mm_realloc((heap), (p), (size) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
277#define zend_mm_realloc2(heap, p, size, copy_size) _zend_mm_realloc2((heap), (p), (size), (copy_size) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
278#define zend_mm_block_size(heap, p)         _zend_mm_block_size((heap), (p) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
279
280#define zend_mm_alloc_rel(heap, size)       _zend_mm_alloc((heap), (size) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
281#define zend_mm_free_rel(heap, p)           _zend_mm_free((heap), (p) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
282#define zend_mm_realloc_rel(heap, p, size)  _zend_mm_realloc((heap), (p), (size) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
283#define zend_mm_realloc2_rel(heap, p, size, copy_size) _zend_mm_realloc2((heap), (p), (size), (copy_size) ZEND_FILE_LINE_RELAY_CC ZEND_FILE_LINE_CC)
284#define zend_mm_block_size_rel(heap, p)     _zend_mm_block_size((heap), (p) ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC)
285
286ZEND_API zend_mm_heap *zend_mm_set_heap(zend_mm_heap *new_heap);
287ZEND_API zend_mm_heap *zend_mm_get_heap(void);
288
289ZEND_API size_t zend_mm_gc(zend_mm_heap *heap);
290
291#define ZEND_MM_CUSTOM_HEAP_NONE  0
292#define ZEND_MM_CUSTOM_HEAP_STD   1
293#define ZEND_MM_CUSTOM_HEAP_DEBUG 2
294
295ZEND_API int zend_mm_is_custom_heap(zend_mm_heap *new_heap);
296ZEND_API void zend_mm_set_custom_handlers(zend_mm_heap *heap,
297                                          void* (*_malloc)(size_t),
298                                          void  (*_free)(void*),
299                                          void* (*_realloc)(void*, size_t));
300ZEND_API void zend_mm_get_custom_handlers(zend_mm_heap *heap,
301                                          void* (**_malloc)(size_t),
302                                          void  (**_free)(void*),
303                                          void* (**_realloc)(void*, size_t));
304
305#if ZEND_DEBUG
306ZEND_API void zend_mm_set_custom_debug_handlers(zend_mm_heap *heap,
307                                          void* (*_malloc)(size_t ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC),
308                                          void  (*_free)(void* ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC),
309                                          void* (*_realloc)(void*, size_t ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC));
310#endif
311
312typedef struct _zend_mm_storage zend_mm_storage;
313
314typedef void* (*zend_mm_chunk_alloc_t)(zend_mm_storage *storage, size_t size, size_t alignment);
315typedef void  (*zend_mm_chunk_free_t)(zend_mm_storage *storage, void *chunk, size_t size);
316typedef int   (*zend_mm_chunk_truncate_t)(zend_mm_storage *storage, void *chunk, size_t old_size, size_t new_size);
317typedef int   (*zend_mm_chunk_extend_t)(zend_mm_storage *storage, void *chunk, size_t old_size, size_t new_size);
318
319typedef struct _zend_mm_handlers {
320    zend_mm_chunk_alloc_t       chunk_alloc;
321    zend_mm_chunk_free_t        chunk_free;
322    zend_mm_chunk_truncate_t    chunk_truncate;
323    zend_mm_chunk_extend_t      chunk_extend;
324} zend_mm_handlers;
325
326struct _zend_mm_storage {
327    const zend_mm_handlers handlers;
328    void *data;
329};
330
331ZEND_API zend_mm_storage *zend_mm_get_storage(zend_mm_heap *heap);
332ZEND_API zend_mm_heap *zend_mm_startup_ex(const zend_mm_handlers *handlers, void *data, size_t data_size);
333
334/*
335
336// The following example shows how to use zend_mm_heap API with custom storage
337
338static zend_mm_heap *apc_heap = NULL;
339static HashTable    *apc_ht = NULL;
340
341typedef struct _apc_data {
342    void     *mem;
343    uint32_t  free_pages;
344} apc_data;
345
346static void *apc_chunk_alloc(zend_mm_storage *storage, size_t size, size_t alignment)
347{
348    apc_data *data = (apc_data*)(storage->data);
349    size_t real_size = ((size + (ZEND_MM_CHUNK_SIZE-1)) & ~(ZEND_MM_CHUNK_SIZE-1));
350    uint32_t count = real_size / ZEND_MM_CHUNK_SIZE;
351    uint32_t first, last, i;
352
353    ZEND_ASSERT(alignment == ZEND_MM_CHUNK_SIZE);
354
355    for (first = 0; first < 32; first++) {
356        if (!(data->free_pages & (1 << first))) {
357            last = first;
358            do {
359                if (last - first == count - 1) {
360                    for (i = first; i <= last; i++) {
361                        data->free_pages |= (1 << i);
362                    }
363                    return (void *)(((char*)(data->mem)) + ZEND_MM_CHUNK_SIZE * (1 << first));
364                }
365                last++;
366            } while (last < 32 && !(data->free_pages & (1 << last)));
367            first = last;
368        }
369    }
370    return NULL;
371}
372
373static void apc_chunk_free(zend_mm_storage *storage, void *chunk, size_t size)
374{
375    apc_data *data = (apc_data*)(storage->data);
376    uint32_t i;
377
378    ZEND_ASSERT(((uintptr_t)chunk & (ZEND_MM_CHUNK_SIZE - 1)) == 0);
379
380    i = ((uintptr_t)chunk - (uintptr_t)(data->mem)) / ZEND_MM_CHUNK_SIZE;
381    while (1) {
382        data->free_pages &= ~(1 << i);
383        if (size <= ZEND_MM_CHUNK_SIZE) {
384            break;
385        }
386        size -= ZEND_MM_CHUNK_SIZE;
387    }
388}
389
390static void apc_init_heap(void)
391{
392    zend_mm_handlers apc_handlers = {
393        apc_chunk_alloc,
394        apc_chunk_free,
395        NULL,
396        NULL,
397    };
398    apc_data tmp_data;
399    zend_mm_heap *old_heap;
400
401    // Preallocate properly aligned SHM chunks (64MB)
402    tmp_data.mem = shm_memalign(ZEND_MM_CHUNK_SIZE, ZEND_MM_CHUNK_SIZE * 32);
403
404    // Initialize temporary storage data
405    tmp_data.free_pages = 0;
406
407    // Create heap
408    apc_heap = zend_mm_startup_ex(&apc_handlers, &tmp_data, sizeof(tmp_data));
409
410    // Allocate some data in the heap
411    old_heap = zend_mm_set_heap(apc_heap);
412    ALLOC_HASHTABLE(apc_ht);
413    zend_hash_init(apc_ht, 64, NULL, ZVAL_PTR_DTOR, 0);
414    zend_mm_set_heap(old_heap);
415}
416
417*/
418
419END_EXTERN_C()
420
421#endif
422
423/*
424 * Local variables:
425 * tab-width: 4
426 * c-basic-offset: 4
427 * indent-tabs-mode: t
428 * End:
429 */
430