Loading...
tests/pgm_integration.c libmalloc-374.60.3 libmalloc-425.100.7
--- 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
+}