Loading...
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 | #ifndef __XZONE_TESTING_H__ #define __XZONE_TESTING_H__ #include <darwintest.h> #include <../src/internal.h> #include "tmo_test_defs.h" #if CONFIG_XZONE_MALLOC #if defined(_MALLOC_TYPE_ENABLED) && _MALLOC_TYPE_ENABLED #define HAVE_MALLOC_TYPE 1 #else #define HAVE_MALLOC_TYPE 0 #endif extern malloc_zone_t **malloc_zones; extern int32_t malloc_num_zones; static inline xzm_malloc_zone_t get_default_xzone_zone(void) { bool found_pgm = false; bool found_nano = false; unsigned i = 0; malloc_zone_t *zone = malloc_zones[i]; if (!strcmp(malloc_get_zone_name(zone), "ProbGuardMallocZone")) { found_pgm = true; i++; if (i == malloc_num_zones) { T_ASSERT_FAIL("didn't find xzone zone"); } zone = malloc_zones[i]; } T_ASSERT_GE(zone->version, 14, "zone version"); if (zone->introspect->zone_type != MALLOC_ZONE_TYPE_XZONE) { // Maybe it's nano? i++; if (i == malloc_num_zones) { T_ASSERT_FAIL("didn't find xzone xzone"); } malloc_zone_t *helper_zone = malloc_zones[i]; const char *name = malloc_get_zone_name(helper_zone); if (strcmp(name, "MallocHelperZone") != 0) { T_ASSERT_FAIL("unexpected zone %s", name); } found_nano = true; zone = helper_zone; T_ASSERT_GE(zone->version, 14, "helper zone version"); T_ASSERT_EQ(zone->introspect->zone_type, MALLOC_ZONE_TYPE_XZONE, "helper zone is xzone malloc"); } bool nano_on_xzone = false; enum { PGM_NO_EXPECTATION, PGM_EXPECTED_ENABLED, PGM_EXPECTED_DISABLED, } pgm_expectation = PGM_NO_EXPECTATION; #if !MALLOC_TARGET_EXCLAVES nano_on_xzone = getenv("MallocNanoOnXzone"); const char *pgm_env = getenv("MallocProbGuard"); if (pgm_env) { pgm_expectation = (*pgm_env == '1' ? PGM_EXPECTED_ENABLED : PGM_EXPECTED_DISABLED); } #endif T_ASSERT_EQ(nano_on_xzone, found_nano, "Nano state matched expectation (%d)", (int)nano_on_xzone); switch (pgm_expectation) { case PGM_NO_EXPECTATION: T_LOG("PGM enablement: %d (no expectation)", (int)found_pgm); break; case PGM_EXPECTED_ENABLED: T_ASSERT_TRUE(found_pgm, "PGM enabled"); break; case PGM_EXPECTED_DISABLED: T_ASSERT_FALSE(found_pgm, "PGM disabled"); break; default: T_ASSERT_FAIL("pgm_expectation"); break; } return (xzm_malloc_zone_t)zone; } #define _TEST_STRINGIFY(x) #x #define TEST_STRINGIFY(x) _TEST_STRINGIFY(x) #define PTR_BUCKET_ENVVAR "MallocXzonePtrBucketCount=" \ TEST_STRINGIFY(XZM_XZONE_DEFAULT_POINTER_BUCKET_COUNT) static inline void validate_bucket_distribution(xzm_malloc_zone_t zone, const char *expr, void **ptrs, size_t n, bool do_free) { size_t counts[XZM_XZONE_DEFAULT_POINTER_BUCKET_COUNT] = { 0 }; size_t early_count = 0; for (int i = 0; i < n; i++) { void *p = ptrs[i]; T_QUIET; T_ASSERT_NOTNULL(p, "(%s) allocation not NULL", expr); xzm_slice_kind_t kind; xzm_segment_group_id_t sgid; xzm_xzone_bucket_t bucket; bool lookup = xzm_ptr_lookup_4test(zone, p, &kind, &sgid, &bucket); if (i == 0) { size_t size = malloc_size(p); T_LOG("(%s) malloc_size %zu", expr, size); } if (do_free) { free(p); } if (!lookup) { early_count++; continue; } T_QUIET; T_ASSERT_EQ((int)kind, XZM_SLICE_KIND_TINY_CHUNK, "tiny chunk"); T_QUIET; T_ASSERT_EQ((int)sgid, XZM_SEGMENT_GROUP_POINTER_XZONES, "xzone pointer segment group"); T_QUIET; T_ASSERT_GE(bucket, XZM_XZONE_BUCKET_POINTER_BASE, "pointer bucket lower bound"); T_QUIET; T_ASSERT_LT(bucket, XZM_XZONE_BUCKET_POINTER_BASE + XZM_XZONE_DEFAULT_POINTER_BUCKET_COUNT, "pointer bucket upper bound"); counts[bucket - XZM_XZONE_BUCKET_POINTER_BASE]++; } T_LOG("(%s) %zu early allocations", expr, early_count); for (int i = 0; i < XZM_XZONE_DEFAULT_POINTER_BUCKET_COUNT; i++) { T_EXPECT_GT(counts[i], (size_t)0, "(%s) expected nonzero allocations for bucket %d, found %zu", expr, i, counts[i]); } } #endif // CONFIG_XZONE_MALLOC #endif // __XZONE_TESTING_H__ |