Fix custom heap free

This seems to be designed around the use-case where the custom
allocator is a wrapper around ZMM.

1 files changed, 7 insertions, 1 deletions

diff --git a/Zend/zend_alloc.c b/Zend/zend_alloc.c
index 8bf37bc5fd..618a66a9ea 100644
--- a/Zend/zend_alloc.c
+++ b/Zend/zend_alloc.c
@@ -2203,11 +2203,17 @@ void zend_mm_shutdown(zend_mm_heap *heap, int full, int silent)
 			if (full) {
 				zend_hash_destroy(heap->tracked_allocs);
 				free(heap->tracked_allocs);
+				/* Make sure the heap free below does not use tracked_free(). */
+				heap->custom_heap.std._free = free;
 			}
 		}
 
 		if (full) {
-			free(heap);
+			if (ZEND_DEBUG && heap->use_custom_heap == ZEND_MM_CUSTOM_HEAP_DEBUG) {
+				heap->custom_heap.debug._free(heap ZEND_FILE_LINE_CC ZEND_FILE_LINE_EMPTY_CC);
+			} else {
+				heap->custom_heap.std._free(heap);
+			}
 		}
 		return;
 	}