1 #include "test/jemalloc_test.h" 2 3 static size_t 4 get_max_size_class(void) { 5 unsigned nlextents; 6 size_t mib[4]; 7 size_t sz, miblen, max_size_class; 8 9 sz = sizeof(unsigned); 10 expect_d_eq(mallctl("arenas.nlextents", (void *)&nlextents, &sz, NULL, 11 0), 0, "Unexpected mallctl() error"); 12 13 miblen = sizeof(mib) / sizeof(size_t); 14 expect_d_eq(mallctlnametomib("arenas.lextent.0.size", mib, &miblen), 0, 15 "Unexpected mallctlnametomib() error"); 16 mib[2] = nlextents - 1; 17 18 sz = sizeof(size_t); 19 expect_d_eq(mallctlbymib(mib, miblen, (void *)&max_size_class, &sz, 20 NULL, 0), 0, "Unexpected mallctlbymib() error"); 21 22 return max_size_class; 23 } 24 25 TEST_BEGIN(test_size_classes) { 26 size_t size_class, max_size_class; 27 szind_t index, max_index; 28 29 max_size_class = get_max_size_class(); 30 max_index = sz_size2index(max_size_class); 31 32 for (index = 0, size_class = sz_index2size(index); index < max_index || 33 size_class < max_size_class; index++, size_class = 34 sz_index2size(index)) { 35 expect_true(index < max_index, 36 "Loop conditionals should be equivalent; index=%u, " 37 "size_class=%zu (%#zx)", index, size_class, size_class); 38 expect_true(size_class < max_size_class, 39 "Loop conditionals should be equivalent; index=%u, " 40 "size_class=%zu (%#zx)", index, size_class, size_class); 41 42 expect_u_eq(index, sz_size2index(size_class), 43 "sz_size2index() does not reverse sz_index2size(): index=%u" 44 " --> size_class=%zu --> index=%u --> size_class=%zu", 45 index, size_class, sz_size2index(size_class), 46 sz_index2size(sz_size2index(size_class))); 47 expect_zu_eq(size_class, 48 sz_index2size(sz_size2index(size_class)), 49 "sz_index2size() does not reverse sz_size2index(): index=%u" 50 " --> size_class=%zu --> index=%u --> size_class=%zu", 51 index, size_class, sz_size2index(size_class), 52 sz_index2size(sz_size2index(size_class))); 53 54 expect_u_eq(index+1, sz_size2index(size_class+1), 55 "Next size_class does not round up properly"); 56 57 expect_zu_eq(size_class, (index > 0) ? 58 sz_s2u(sz_index2size(index-1)+1) : sz_s2u(1), 59 "sz_s2u() does not round up to size class"); 60 expect_zu_eq(size_class, sz_s2u(size_class-1), 61 "sz_s2u() does not round up to size class"); 62 expect_zu_eq(size_class, sz_s2u(size_class), 63 "sz_s2u() does not compute same size class"); 64 expect_zu_eq(sz_s2u(size_class+1), sz_index2size(index+1), 65 "sz_s2u() does not round up to next size class"); 66 } 67 68 expect_u_eq(index, sz_size2index(sz_index2size(index)), 69 "sz_size2index() does not reverse sz_index2size()"); 70 expect_zu_eq(max_size_class, sz_index2size( 71 sz_size2index(max_size_class)), 72 "sz_index2size() does not reverse sz_size2index()"); 73 74 expect_zu_eq(size_class, sz_s2u(sz_index2size(index-1)+1), 75 "sz_s2u() does not round up to size class"); 76 expect_zu_eq(size_class, sz_s2u(size_class-1), 77 "sz_s2u() does not round up to size class"); 78 expect_zu_eq(size_class, sz_s2u(size_class), 79 "sz_s2u() does not compute same size class"); 80 } 81 TEST_END 82 83 TEST_BEGIN(test_psize_classes) { 84 size_t size_class, max_psz; 85 pszind_t pind, max_pind; 86 87 max_psz = get_max_size_class() + PAGE; 88 max_pind = sz_psz2ind(max_psz); 89 90 for (pind = 0, size_class = sz_pind2sz(pind); 91 pind < max_pind || size_class < max_psz; 92 pind++, size_class = sz_pind2sz(pind)) { 93 expect_true(pind < max_pind, 94 "Loop conditionals should be equivalent; pind=%u, " 95 "size_class=%zu (%#zx)", pind, size_class, size_class); 96 expect_true(size_class < max_psz, 97 "Loop conditionals should be equivalent; pind=%u, " 98 "size_class=%zu (%#zx)", pind, size_class, size_class); 99 100 expect_u_eq(pind, sz_psz2ind(size_class), 101 "sz_psz2ind() does not reverse sz_pind2sz(): pind=%u -->" 102 " size_class=%zu --> pind=%u --> size_class=%zu", pind, 103 size_class, sz_psz2ind(size_class), 104 sz_pind2sz(sz_psz2ind(size_class))); 105 expect_zu_eq(size_class, sz_pind2sz(sz_psz2ind(size_class)), 106 "sz_pind2sz() does not reverse sz_psz2ind(): pind=%u -->" 107 " size_class=%zu --> pind=%u --> size_class=%zu", pind, 108 size_class, sz_psz2ind(size_class), 109 sz_pind2sz(sz_psz2ind(size_class))); 110 111 if (size_class == SC_LARGE_MAXCLASS) { 112 expect_u_eq(SC_NPSIZES, sz_psz2ind(size_class + 1), 113 "Next size_class does not round up properly"); 114 } else { 115 expect_u_eq(pind + 1, sz_psz2ind(size_class + 1), 116 "Next size_class does not round up properly"); 117 } 118 119 expect_zu_eq(size_class, (pind > 0) ? 120 sz_psz2u(sz_pind2sz(pind-1)+1) : sz_psz2u(1), 121 "sz_psz2u() does not round up to size class"); 122 expect_zu_eq(size_class, sz_psz2u(size_class-1), 123 "sz_psz2u() does not round up to size class"); 124 expect_zu_eq(size_class, sz_psz2u(size_class), 125 "sz_psz2u() does not compute same size class"); 126 expect_zu_eq(sz_psz2u(size_class+1), sz_pind2sz(pind+1), 127 "sz_psz2u() does not round up to next size class"); 128 } 129 130 expect_u_eq(pind, sz_psz2ind(sz_pind2sz(pind)), 131 "sz_psz2ind() does not reverse sz_pind2sz()"); 132 expect_zu_eq(max_psz, sz_pind2sz(sz_psz2ind(max_psz)), 133 "sz_pind2sz() does not reverse sz_psz2ind()"); 134 135 expect_zu_eq(size_class, sz_psz2u(sz_pind2sz(pind-1)+1), 136 "sz_psz2u() does not round up to size class"); 137 expect_zu_eq(size_class, sz_psz2u(size_class-1), 138 "sz_psz2u() does not round up to size class"); 139 expect_zu_eq(size_class, sz_psz2u(size_class), 140 "sz_psz2u() does not compute same size class"); 141 } 142 TEST_END 143 144 TEST_BEGIN(test_overflow) { 145 size_t max_size_class, max_psz; 146 147 max_size_class = get_max_size_class(); 148 max_psz = max_size_class + PAGE; 149 150 expect_u_eq(sz_size2index(max_size_class+1), SC_NSIZES, 151 "sz_size2index() should return NSIZES on overflow"); 152 expect_u_eq(sz_size2index(ZU(PTRDIFF_MAX)+1), SC_NSIZES, 153 "sz_size2index() should return NSIZES on overflow"); 154 expect_u_eq(sz_size2index(SIZE_T_MAX), SC_NSIZES, 155 "sz_size2index() should return NSIZES on overflow"); 156 157 expect_zu_eq(sz_s2u(max_size_class+1), 0, 158 "sz_s2u() should return 0 for unsupported size"); 159 expect_zu_eq(sz_s2u(ZU(PTRDIFF_MAX)+1), 0, 160 "sz_s2u() should return 0 for unsupported size"); 161 expect_zu_eq(sz_s2u(SIZE_T_MAX), 0, 162 "sz_s2u() should return 0 on overflow"); 163 164 expect_u_eq(sz_psz2ind(max_size_class+1), SC_NPSIZES, 165 "sz_psz2ind() should return NPSIZES on overflow"); 166 expect_u_eq(sz_psz2ind(ZU(PTRDIFF_MAX)+1), SC_NPSIZES, 167 "sz_psz2ind() should return NPSIZES on overflow"); 168 expect_u_eq(sz_psz2ind(SIZE_T_MAX), SC_NPSIZES, 169 "sz_psz2ind() should return NPSIZES on overflow"); 170 171 expect_zu_eq(sz_psz2u(max_size_class+1), max_psz, 172 "sz_psz2u() should return (LARGE_MAXCLASS + PAGE) for unsupported" 173 " size"); 174 expect_zu_eq(sz_psz2u(ZU(PTRDIFF_MAX)+1), max_psz, 175 "sz_psz2u() should return (LARGE_MAXCLASS + PAGE) for unsupported " 176 "size"); 177 expect_zu_eq(sz_psz2u(SIZE_T_MAX), max_psz, 178 "sz_psz2u() should return (LARGE_MAXCLASS + PAGE) on overflow"); 179 } 180 TEST_END 181 182 int 183 main(void) { 184 return test( 185 test_size_classes, 186 test_psize_classes, 187 test_overflow); 188 } 189
Indexes created Sun Mar 01 05:31:48 UTC 2026