1 #include "test/jemalloc_test.h"
2
3 /* Immediately purge to minimize fragmentation. */
4 const char *malloc_conf = "decay_time:-1";
5
6 /*
7 * Size class that is a divisor of the page size, ideally 4+ regions per run.
8 */
9 #if LG_PAGE <= 14
10 #define SZ (ZU(1) << (LG_PAGE - 2))
11 #else
12 #define SZ 4096
13 #endif
14
15 /*
16 * Number of slabs to consume at high water mark. Should be at least 2 so that
17 * if mmap()ed memory grows downward, downward growth of mmap()ed memory is
18 * tested.
19 */
20 #define NSLABS 8
21
22 static unsigned
binind_compute(void)23 binind_compute(void)
24 {
25 size_t sz;
26 unsigned nbins, i;
27
28 sz = sizeof(nbins);
29 assert_d_eq(mallctl("arenas.nbins", (void *)&nbins, &sz, NULL, 0), 0,
30 "Unexpected mallctl failure");
31
32 for (i = 0; i < nbins; i++) {
33 size_t mib[4];
34 size_t miblen = sizeof(mib)/sizeof(size_t);
35 size_t size;
36
37 assert_d_eq(mallctlnametomib("arenas.bin.0.size", mib,
38 &miblen), 0, "Unexpected mallctlnametomb failure");
39 mib[2] = (size_t)i;
40
41 sz = sizeof(size);
42 assert_d_eq(mallctlbymib(mib, miblen, (void *)&size, &sz, NULL,
43 0), 0, "Unexpected mallctlbymib failure");
44 if (size == SZ)
45 return (i);
46 }
47
48 test_fail("Unable to compute nregs_per_run");
49 return (0);
50 }
51
52 static size_t
nregs_per_run_compute(void)53 nregs_per_run_compute(void)
54 {
55 uint32_t nregs;
56 size_t sz;
57 unsigned binind = binind_compute();
58 size_t mib[4];
59 size_t miblen = sizeof(mib)/sizeof(size_t);
60
61 assert_d_eq(mallctlnametomib("arenas.bin.0.nregs", mib, &miblen), 0,
62 "Unexpected mallctlnametomb failure");
63 mib[2] = (size_t)binind;
64 sz = sizeof(nregs);
65 assert_d_eq(mallctlbymib(mib, miblen, (void *)&nregs, &sz, NULL,
66 0), 0, "Unexpected mallctlbymib failure");
67 return (nregs);
68 }
69
70 static unsigned
arenas_create_mallctl(void)71 arenas_create_mallctl(void)
72 {
73 unsigned arena_ind;
74 size_t sz;
75
76 sz = sizeof(arena_ind);
77 assert_d_eq(mallctl("arenas.create", (void *)&arena_ind, &sz, NULL, 0),
78 0, "Error in arenas.create");
79
80 return (arena_ind);
81 }
82
83 static void
arena_reset_mallctl(unsigned arena_ind)84 arena_reset_mallctl(unsigned arena_ind)
85 {
86 size_t mib[3];
87 size_t miblen = sizeof(mib)/sizeof(size_t);
88
89 assert_d_eq(mallctlnametomib("arena.0.reset", mib, &miblen), 0,
90 "Unexpected mallctlnametomib() failure");
91 mib[1] = (size_t)arena_ind;
92 assert_d_eq(mallctlbymib(mib, miblen, NULL, NULL, NULL, 0), 0,
93 "Unexpected mallctlbymib() failure");
94 }
95
TEST_BEGIN(test_pack)96 TEST_BEGIN(test_pack)
97 {
98 unsigned arena_ind = arenas_create_mallctl();
99 size_t nregs_per_run = nregs_per_run_compute();
100 size_t nregs = nregs_per_run * NSLABS;
101 VARIABLE_ARRAY(void *, ptrs, nregs);
102 size_t i, j, offset;
103
104 /* Fill matrix. */
105 for (i = offset = 0; i < NSLABS; i++) {
106 for (j = 0; j < nregs_per_run; j++) {
107 void *p = mallocx(SZ, MALLOCX_ARENA(arena_ind) |
108 MALLOCX_TCACHE_NONE);
109 assert_ptr_not_null(p,
110 "Unexpected mallocx(%zu, MALLOCX_ARENA(%u) |"
111 " MALLOCX_TCACHE_NONE) failure, run=%zu, reg=%zu",
112 SZ, arena_ind, i, j);
113 ptrs[(i * nregs_per_run) + j] = p;
114 }
115 }
116
117 /*
118 * Free all but one region of each run, but rotate which region is
119 * preserved, so that subsequent allocations exercise the within-run
120 * layout policy.
121 */
122 offset = 0;
123 for (i = offset = 0;
124 i < NSLABS;
125 i++, offset = (offset + 1) % nregs_per_run) {
126 for (j = 0; j < nregs_per_run; j++) {
127 void *p = ptrs[(i * nregs_per_run) + j];
128 if (offset == j)
129 continue;
130 dallocx(p, MALLOCX_ARENA(arena_ind) |
131 MALLOCX_TCACHE_NONE);
132 }
133 }
134
135 /*
136 * Logically refill matrix, skipping preserved regions and verifying
137 * that the matrix is unmodified.
138 */
139 offset = 0;
140 for (i = offset = 0;
141 i < NSLABS;
142 i++, offset = (offset + 1) % nregs_per_run) {
143 for (j = 0; j < nregs_per_run; j++) {
144 void *p;
145
146 if (offset == j)
147 continue;
148 p = mallocx(SZ, MALLOCX_ARENA(arena_ind) |
149 MALLOCX_TCACHE_NONE);
150 assert_ptr_eq(p, ptrs[(i * nregs_per_run) + j],
151 "Unexpected refill discrepancy, run=%zu, reg=%zu\n",
152 i, j);
153 }
154 }
155
156 /* Clean up. */
157 arena_reset_mallctl(arena_ind);
158 }
159 TEST_END
160
161 int
main(void)162 main(void)
163 {
164 return (test(
165 test_pack));
166 }
167