zend_arena.h 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224
  1. /*
  2. +----------------------------------------------------------------------+
  3. | Zend Engine |
  4. +----------------------------------------------------------------------+
  5. | Copyright (c) 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: Dmitry Stogov <dmitry@php.net> |
  16. +----------------------------------------------------------------------+
  17. */
  18. #ifndef _ZEND_ARENA_H_
  19. #define _ZEND_ARENA_H_
  20. #include "zend.h"
  21. #ifndef ZEND_TRACK_ARENA_ALLOC
  22. typedef struct _zend_arena zend_arena;
  23. struct _zend_arena {
  24. char *ptr;
  25. char *end;
  26. zend_arena *prev;
  27. };
  28. static zend_always_inline zend_arena* zend_arena_create(size_t size)
  29. {
  30. zend_arena *arena = (zend_arena*)emalloc(size);
  31. arena->ptr = (char*) arena + ZEND_MM_ALIGNED_SIZE(sizeof(zend_arena));
  32. arena->end = (char*) arena + size;
  33. arena->prev = NULL;
  34. return arena;
  35. }
  36. static zend_always_inline void zend_arena_destroy(zend_arena *arena)
  37. {
  38. do {
  39. zend_arena *prev = arena->prev;
  40. efree(arena);
  41. arena = prev;
  42. } while (arena);
  43. }
  44. static zend_always_inline void* zend_arena_alloc(zend_arena **arena_ptr, size_t size)
  45. {
  46. zend_arena *arena = *arena_ptr;
  47. char *ptr = arena->ptr;
  48. size = ZEND_MM_ALIGNED_SIZE(size);
  49. if (EXPECTED(size <= (size_t)(arena->end - ptr))) {
  50. arena->ptr = ptr + size;
  51. } else {
  52. size_t arena_size =
  53. UNEXPECTED((size + ZEND_MM_ALIGNED_SIZE(sizeof(zend_arena))) > (size_t)(arena->end - (char*) arena)) ?
  54. (size + ZEND_MM_ALIGNED_SIZE(sizeof(zend_arena))) :
  55. (size_t)(arena->end - (char*) arena);
  56. zend_arena *new_arena = (zend_arena*)emalloc(arena_size);
  57. ptr = (char*) new_arena + ZEND_MM_ALIGNED_SIZE(sizeof(zend_arena));
  58. new_arena->ptr = (char*) new_arena + ZEND_MM_ALIGNED_SIZE(sizeof(zend_arena)) + size;
  59. new_arena->end = (char*) new_arena + arena_size;
  60. new_arena->prev = arena;
  61. *arena_ptr = new_arena;
  62. }
  63. return (void*) ptr;
  64. }
  65. static zend_always_inline void* zend_arena_calloc(zend_arena **arena_ptr, size_t count, size_t unit_size)
  66. {
  67. bool overflow;
  68. size_t size;
  69. void *ret;
  70. size = zend_safe_address(unit_size, count, 0, &overflow);
  71. if (UNEXPECTED(overflow)) {
  72. zend_error(E_ERROR, "Possible integer overflow in zend_arena_calloc() (%zu * %zu)", unit_size, count);
  73. }
  74. ret = zend_arena_alloc(arena_ptr, size);
  75. memset(ret, 0, size);
  76. return ret;
  77. }
  78. static zend_always_inline void* zend_arena_checkpoint(zend_arena *arena)
  79. {
  80. return arena->ptr;
  81. }
  82. static zend_always_inline void zend_arena_release(zend_arena **arena_ptr, void *checkpoint)
  83. {
  84. zend_arena *arena = *arena_ptr;
  85. while (UNEXPECTED((char*)checkpoint > arena->end) ||
  86. UNEXPECTED((char*)checkpoint <= (char*)arena)) {
  87. zend_arena *prev = arena->prev;
  88. efree(arena);
  89. *arena_ptr = arena = prev;
  90. }
  91. ZEND_ASSERT((char*)checkpoint > (char*)arena && (char*)checkpoint <= arena->end);
  92. arena->ptr = (char*)checkpoint;
  93. }
  94. static zend_always_inline bool zend_arena_contains(zend_arena *arena, void *ptr)
  95. {
  96. while (arena) {
  97. if ((char*)ptr > (char*)arena && (char*)ptr <= arena->ptr) {
  98. return 1;
  99. }
  100. arena = arena->prev;
  101. }
  102. return 0;
  103. }
  104. #else
  105. /* Use normal allocations and keep track of them for mass-freeing.
  106. * This is intended for use with asan/valgrind. */
  107. typedef struct _zend_arena zend_arena;
  108. struct _zend_arena {
  109. void **ptr;
  110. void **end;
  111. struct _zend_arena *prev;
  112. void *ptrs[0];
  113. };
  114. #define ZEND_TRACKED_ARENA_SIZE 1000
  115. static zend_always_inline zend_arena *zend_arena_create(size_t _size)
  116. {
  117. zend_arena *arena = (zend_arena*) emalloc(
  118. sizeof(zend_arena) + sizeof(void *) * ZEND_TRACKED_ARENA_SIZE);
  119. arena->ptr = &arena->ptrs[0];
  120. arena->end = &arena->ptrs[ZEND_TRACKED_ARENA_SIZE];
  121. arena->prev = NULL;
  122. return arena;
  123. }
  124. static zend_always_inline void zend_arena_destroy(zend_arena *arena)
  125. {
  126. do {
  127. zend_arena *prev = arena->prev;
  128. void **ptr;
  129. for (ptr = arena->ptrs; ptr < arena->ptr; ptr++) {
  130. efree(*ptr);
  131. }
  132. efree(arena);
  133. arena = prev;
  134. } while (arena);
  135. }
  136. static zend_always_inline void *zend_arena_alloc(zend_arena **arena_ptr, size_t size)
  137. {
  138. zend_arena *arena = *arena_ptr;
  139. if (arena->ptr == arena->end) {
  140. *arena_ptr = zend_arena_create(0);
  141. (*arena_ptr)->prev = arena;
  142. arena = *arena_ptr;
  143. }
  144. return *arena->ptr++ = emalloc(size);
  145. }
  146. static zend_always_inline void* zend_arena_calloc(zend_arena **arena_ptr, size_t count, size_t unit_size)
  147. {
  148. bool overflow;
  149. size_t size;
  150. void *ret;
  151. size = zend_safe_address(unit_size, count, 0, &overflow);
  152. if (UNEXPECTED(overflow)) {
  153. zend_error(E_ERROR, "Possible integer overflow in zend_arena_calloc() (%zu * %zu)", unit_size, count);
  154. }
  155. ret = zend_arena_alloc(arena_ptr, size);
  156. memset(ret, 0, size);
  157. return ret;
  158. }
  159. static zend_always_inline void* zend_arena_checkpoint(zend_arena *arena)
  160. {
  161. return arena->ptr;
  162. }
  163. static zend_always_inline void zend_arena_release(zend_arena **arena_ptr, void *checkpoint)
  164. {
  165. while (1) {
  166. zend_arena *arena = *arena_ptr;
  167. zend_arena *prev = arena->prev;
  168. while (1) {
  169. if (arena->ptr == (void **) checkpoint) {
  170. return;
  171. }
  172. if (arena->ptr == arena->ptrs) {
  173. break;
  174. }
  175. arena->ptr--;
  176. efree(*arena->ptr);
  177. }
  178. efree(arena);
  179. *arena_ptr = prev;
  180. ZEND_ASSERT(*arena_ptr);
  181. }
  182. }
  183. static zend_always_inline bool zend_arena_contains(zend_arena *arena, void *ptr)
  184. {
  185. /* TODO: Dummy */
  186. return 1;
  187. }
  188. #endif
  189. #endif /* _ZEND_ARENA_H_ */