Loading...
--- libmalloc/libmalloc-374.60.3/tests/pgm_integration.c
+++ libmalloc/libmalloc-425.100.7/tests/pgm_integration.c
@@ -13,6 +13,8 @@
#include <malloc/malloc.h>
#include <stdlib.h>
#include <unistd.h>
+
+#include "../src/platform.h" // CONFIG_PGM_WRAP_CUSTOM_ZONES
T_GLOBAL_META(
T_META_ENVVAR("MallocProbGuard=1"),
@@ -313,9 +315,13 @@
uint32_t num_zones = malloc_num_zones;
malloc_zone_t *zone = malloc_create_zone(0, 0);
+#if CONFIG_PGM_WRAP_CUSTOM_ZONES
T_EXPECT_EQ_STR(malloc_get_zone_name(zone), "ProbGuardMallocZone", "PGM-wrapped zone");
T_EXPECT_EQ(malloc_num_zones, num_zones + 2, "registered both zones");
malloc_destroy_zone(zone);
T_EXPECT_EQ(malloc_num_zones, num_zones, "unregistered both zones");
-}
+#else
+ T_EXPECT_EQ(malloc_num_zones, num_zones + 1, "no PGM wrapper zone");
+#endif
+}