1 #include "test/jemalloc_test.h"
2
3 #ifdef JEMALLOC_FILL
4 const char *malloc_conf = "junk:false";
5 #endif
6
7 #include "test/extent_hooks.h"
8
9 static void
test_extent_body(unsigned arena_ind)10 test_extent_body(unsigned arena_ind)
11 {
12 void *p;
13 size_t large0, large1, large2, sz;
14 size_t purge_mib[3];
15 size_t purge_miblen;
16 int flags;
17 bool xallocx_success_a, xallocx_success_b, xallocx_success_c;
18
19 flags = MALLOCX_ARENA(arena_ind) | MALLOCX_TCACHE_NONE;
20
21 /* Get large size classes. */
22 sz = sizeof(size_t);
23 assert_d_eq(mallctl("arenas.lextent.0.size", (void *)&large0, &sz, NULL,
24 0), 0, "Unexpected arenas.lextent.0.size failure");
25 assert_d_eq(mallctl("arenas.lextent.1.size", (void *)&large1, &sz, NULL,
26 0), 0, "Unexpected arenas.lextent.1.size failure");
27 assert_d_eq(mallctl("arenas.lextent.2.size", (void *)&large2, &sz, NULL,
28 0), 0, "Unexpected arenas.lextent.2.size failure");
29
30 /* Test dalloc/decommit/purge cascade. */
31 purge_miblen = sizeof(purge_mib)/sizeof(size_t);
32 assert_d_eq(mallctlnametomib("arena.0.purge", purge_mib, &purge_miblen),
33 0, "Unexpected mallctlnametomib() failure");
34 purge_mib[1] = (size_t)arena_ind;
35 try_dalloc = false;
36 try_decommit = false;
37 p = mallocx(large0 * 2, flags);
38 assert_ptr_not_null(p, "Unexpected mallocx() error");
39 called_dalloc = false;
40 called_decommit = false;
41 did_purge_lazy = false;
42 did_purge_forced = false;
43 called_split = false;
44 xallocx_success_a = (xallocx(p, large0, 0, flags) == large0);
45 assert_d_eq(mallctlbymib(purge_mib, purge_miblen, NULL, NULL, NULL, 0),
46 0, "Unexpected arena.%u.purge error", arena_ind);
47 if (xallocx_success_a) {
48 assert_true(called_dalloc, "Expected dalloc call");
49 assert_true(called_decommit, "Expected decommit call");
50 assert_true(did_purge_lazy || did_purge_forced,
51 "Expected purge");
52 }
53 assert_true(called_split, "Expected split call");
54 dallocx(p, flags);
55 try_dalloc = true;
56
57 /* Test decommit/commit and observe split/merge. */
58 try_dalloc = false;
59 try_decommit = true;
60 p = mallocx(large0 * 2, flags);
61 assert_ptr_not_null(p, "Unexpected mallocx() error");
62 did_decommit = false;
63 did_commit = false;
64 called_split = false;
65 did_split = false;
66 did_merge = false;
67 xallocx_success_b = (xallocx(p, large0, 0, flags) == large0);
68 assert_d_eq(mallctlbymib(purge_mib, purge_miblen, NULL, NULL, NULL, 0),
69 0, "Unexpected arena.%u.purge error", arena_ind);
70 if (xallocx_success_b)
71 assert_true(did_split, "Expected split");
72 xallocx_success_c = (xallocx(p, large0 * 2, 0, flags) == large0 * 2);
73 if (did_split) {
74 assert_b_eq(did_decommit, did_commit,
75 "Expected decommit/commit match");
76 }
77 if (xallocx_success_b && xallocx_success_c)
78 assert_true(did_merge, "Expected merge");
79 dallocx(p, flags);
80 try_dalloc = true;
81 try_decommit = false;
82
83 /* Make sure non-large allocation succeeds. */
84 p = mallocx(42, flags);
85 assert_ptr_not_null(p, "Unexpected mallocx() error");
86 dallocx(p, flags);
87 }
88
TEST_BEGIN(test_extent_manual_hook)89 TEST_BEGIN(test_extent_manual_hook)
90 {
91 unsigned arena_ind;
92 size_t old_size, new_size, sz;
93 size_t hooks_mib[3];
94 size_t hooks_miblen;
95 extent_hooks_t *new_hooks, *old_hooks;
96
97 extent_hooks_prep();
98
99 sz = sizeof(unsigned);
100 assert_d_eq(mallctl("arenas.create", (void *)&arena_ind, &sz, NULL, 0),
101 0, "Unexpected mallctl() failure");
102
103 /* Install custom extent hooks. */
104 hooks_miblen = sizeof(hooks_mib)/sizeof(size_t);
105 assert_d_eq(mallctlnametomib("arena.0.extent_hooks", hooks_mib,
106 &hooks_miblen), 0, "Unexpected mallctlnametomib() failure");
107 hooks_mib[1] = (size_t)arena_ind;
108 old_size = sizeof(extent_hooks_t *);
109 new_hooks = &hooks;
110 new_size = sizeof(extent_hooks_t *);
111 assert_d_eq(mallctlbymib(hooks_mib, hooks_miblen, (void *)&old_hooks,
112 &old_size, (void *)&new_hooks, new_size), 0,
113 "Unexpected extent_hooks error");
114 assert_ptr_ne(old_hooks->alloc, extent_alloc_hook,
115 "Unexpected extent_hooks error");
116 assert_ptr_ne(old_hooks->dalloc, extent_dalloc_hook,
117 "Unexpected extent_hooks error");
118 assert_ptr_ne(old_hooks->commit, extent_commit_hook,
119 "Unexpected extent_hooks error");
120 assert_ptr_ne(old_hooks->decommit, extent_decommit_hook,
121 "Unexpected extent_hooks error");
122 assert_ptr_ne(old_hooks->purge_lazy, extent_purge_lazy_hook,
123 "Unexpected extent_hooks error");
124 assert_ptr_ne(old_hooks->purge_forced, extent_purge_forced_hook,
125 "Unexpected extent_hooks error");
126 assert_ptr_ne(old_hooks->split, extent_split_hook,
127 "Unexpected extent_hooks error");
128 assert_ptr_ne(old_hooks->merge, extent_merge_hook,
129 "Unexpected extent_hooks error");
130
131 test_extent_body(arena_ind);
132
133 /* Restore extent hooks. */
134 assert_d_eq(mallctlbymib(hooks_mib, hooks_miblen, NULL, NULL,
135 (void *)&old_hooks, new_size), 0, "Unexpected extent_hooks error");
136 assert_d_eq(mallctlbymib(hooks_mib, hooks_miblen, (void *)&old_hooks,
137 &old_size, NULL, 0), 0, "Unexpected extent_hooks error");
138 assert_ptr_eq(old_hooks, default_hooks, "Unexpected extent_hooks error");
139 assert_ptr_eq(old_hooks->alloc, default_hooks->alloc,
140 "Unexpected extent_hooks error");
141 assert_ptr_eq(old_hooks->dalloc, default_hooks->dalloc,
142 "Unexpected extent_hooks error");
143 assert_ptr_eq(old_hooks->commit, default_hooks->commit,
144 "Unexpected extent_hooks error");
145 assert_ptr_eq(old_hooks->decommit, default_hooks->decommit,
146 "Unexpected extent_hooks error");
147 assert_ptr_eq(old_hooks->purge_lazy, default_hooks->purge_lazy,
148 "Unexpected extent_hooks error");
149 assert_ptr_eq(old_hooks->purge_forced, default_hooks->purge_forced,
150 "Unexpected extent_hooks error");
151 assert_ptr_eq(old_hooks->split, default_hooks->split,
152 "Unexpected extent_hooks error");
153 assert_ptr_eq(old_hooks->merge, default_hooks->merge,
154 "Unexpected extent_hooks error");
155 }
156 TEST_END
157
TEST_BEGIN(test_extent_auto_hook)158 TEST_BEGIN(test_extent_auto_hook)
159 {
160 unsigned arena_ind;
161 size_t new_size, sz;
162 extent_hooks_t *new_hooks;
163
164 extent_hooks_prep();
165
166 sz = sizeof(unsigned);
167 new_hooks = &hooks;
168 new_size = sizeof(extent_hooks_t *);
169 assert_d_eq(mallctl("arenas.create", (void *)&arena_ind, &sz,
170 (void *)&new_hooks, new_size), 0, "Unexpected mallctl() failure");
171
172 test_extent_body(arena_ind);
173 }
174 TEST_END
175
176 int
main(void)177 main(void)
178 {
179 return (test(
180 test_extent_manual_hook,
181 test_extent_auto_hook));
182 }
183