b49a334a64
Add the "arena.<i>.chunk_hooks" mallctl, which replaces and expands on
the "arena.<i>.chunk.{alloc,dalloc,purge}" mallctls. The chunk hooks
allow control over chunk allocation/deallocation, decommit/commit,
purging, and splitting/merging, such that the application can rely on
jemalloc's internal chunk caching and retaining functionality, yet
implement a variety of chunk management mechanisms and policies.
Merge the chunks_[sz]ad_{mmap,dss} red-black trees into
chunks_[sz]ad_retained. This slightly reduces how hard jemalloc tries
to honor the dss precedence setting; prior to this change the precedence
setting was also consulted when recycling chunks.
Fix chunk purging. Don't purge chunks in arena_purge_stashed(); instead
deallocate them in arena_unstash_purged(), so that the dirty memory
linkage remains valid until after the last time it is used.
This resolves #176 and #201.
236 lines
6.9 KiB
C
236 lines
6.9 KiB
C
#include "test/jemalloc_test.h"
|
|
|
|
static chunk_hooks_t orig_hooks;
|
|
static chunk_hooks_t old_hooks;
|
|
|
|
static bool do_dalloc = true;
|
|
static bool do_decommit;
|
|
|
|
static bool did_alloc;
|
|
static bool did_dalloc;
|
|
static bool did_commit;
|
|
static bool did_decommit;
|
|
static bool did_purge;
|
|
static bool did_split;
|
|
static bool did_merge;
|
|
|
|
#if 0
|
|
# define TRACE_HOOK(fmt, ...) malloc_printf(fmt, __VA_ARGS__)
|
|
#else
|
|
# define TRACE_HOOK(fmt, ...)
|
|
#endif
|
|
|
|
void *
|
|
chunk_alloc(void *new_addr, size_t size, size_t alignment, bool *zero,
|
|
unsigned arena_ind)
|
|
{
|
|
|
|
TRACE_HOOK("%s(new_addr=%p, size=%zu, alignment=%zu, *zero=%s, "
|
|
"arena_ind=%u)\n", __func__, new_addr, size, alignment, *zero ?
|
|
"true" : "false", arena_ind);
|
|
did_alloc = true;
|
|
return (old_hooks.alloc(new_addr, size, alignment, zero, arena_ind));
|
|
}
|
|
|
|
bool
|
|
chunk_dalloc(void *chunk, size_t size, unsigned arena_ind)
|
|
{
|
|
|
|
TRACE_HOOK("%s(chunk=%p, size=%zu, arena_ind=%u)\n", __func__, chunk,
|
|
size, arena_ind);
|
|
did_dalloc = true;
|
|
if (!do_dalloc)
|
|
return (true);
|
|
return (old_hooks.dalloc(chunk, size, arena_ind));
|
|
}
|
|
|
|
bool
|
|
chunk_commit(void *chunk, size_t size, unsigned arena_ind)
|
|
{
|
|
|
|
TRACE_HOOK("%s(chunk=%p, size=%zu, arena_ind=%u)\n", __func__, chunk,
|
|
size, arena_ind);
|
|
did_commit = true;
|
|
memset(chunk, 0, size);
|
|
return (false);
|
|
}
|
|
|
|
bool
|
|
chunk_decommit(void *chunk, size_t size, unsigned arena_ind)
|
|
{
|
|
|
|
TRACE_HOOK("%s(chunk=%p, size=%zu, arena_ind=%u)\n", __func__, chunk,
|
|
size, arena_ind);
|
|
did_decommit = true;
|
|
return (!do_decommit);
|
|
}
|
|
|
|
bool
|
|
chunk_purge(void *chunk, size_t size, size_t offset, size_t length,
|
|
unsigned arena_ind)
|
|
{
|
|
|
|
TRACE_HOOK("%s(chunk=%p, size=%zu, offset=%zu, length=%zu "
|
|
"arena_ind=%u)\n", __func__, chunk, size, offset, length,
|
|
arena_ind);
|
|
did_purge = true;
|
|
return (old_hooks.purge(chunk, size, offset, length, arena_ind));
|
|
}
|
|
|
|
bool
|
|
chunk_split(void *chunk, size_t size, size_t size_a, size_t size_b,
|
|
bool committed, unsigned arena_ind)
|
|
{
|
|
|
|
TRACE_HOOK("%s(chunk=%p, size=%zu, size_a=%zu, size_b=%zu, "
|
|
"committed=%s, arena_ind=%u)\n", __func__, chunk, size, size_a,
|
|
size_b, committed ? "true" : "false", arena_ind);
|
|
did_split = true;
|
|
return (old_hooks.split(chunk, size, size_a, size_b, committed,
|
|
arena_ind));
|
|
}
|
|
|
|
bool
|
|
chunk_merge(void *chunk_a, size_t size_a, void *chunk_b, size_t size_b,
|
|
bool committed, unsigned arena_ind)
|
|
{
|
|
|
|
TRACE_HOOK("%s(chunk_a=%p, size_a=%zu, chunk_b=%p size_b=%zu, "
|
|
"committed=%s, arena_ind=%u)\n", __func__, chunk_a, size_a, chunk_b,
|
|
size_b, committed ? "true" : "false", arena_ind);
|
|
did_merge = true;
|
|
return (old_hooks.merge(chunk_a, size_a, chunk_b, size_b,
|
|
committed, arena_ind));
|
|
}
|
|
|
|
TEST_BEGIN(test_chunk)
|
|
{
|
|
void *p;
|
|
size_t old_size, new_size, huge0, huge1, huge2, sz;
|
|
chunk_hooks_t new_hooks = {
|
|
chunk_alloc,
|
|
chunk_dalloc,
|
|
chunk_commit,
|
|
chunk_decommit,
|
|
chunk_purge,
|
|
chunk_split,
|
|
chunk_merge
|
|
};
|
|
|
|
/* Install custom chunk hooks. */
|
|
old_size = sizeof(chunk_hooks_t);
|
|
new_size = sizeof(chunk_hooks_t);
|
|
assert_d_eq(mallctl("arena.0.chunk_hooks", &old_hooks, &old_size,
|
|
&new_hooks, new_size), 0, "Unexpected chunk_hooks error");
|
|
orig_hooks = old_hooks;
|
|
assert_ptr_ne(old_hooks.alloc, chunk_alloc, "Unexpected alloc error");
|
|
assert_ptr_ne(old_hooks.dalloc, chunk_dalloc,
|
|
"Unexpected dalloc error");
|
|
assert_ptr_ne(old_hooks.commit, chunk_commit,
|
|
"Unexpected commit error");
|
|
assert_ptr_ne(old_hooks.decommit, chunk_decommit,
|
|
"Unexpected decommit error");
|
|
assert_ptr_ne(old_hooks.purge, chunk_purge, "Unexpected purge error");
|
|
assert_ptr_ne(old_hooks.split, chunk_split, "Unexpected split error");
|
|
assert_ptr_ne(old_hooks.merge, chunk_merge, "Unexpected merge error");
|
|
|
|
/* Get huge size classes. */
|
|
sz = sizeof(size_t);
|
|
assert_d_eq(mallctl("arenas.hchunk.0.size", &huge0, &sz, NULL, 0), 0,
|
|
"Unexpected arenas.hchunk.0.size failure");
|
|
assert_d_eq(mallctl("arenas.hchunk.1.size", &huge1, &sz, NULL, 0), 0,
|
|
"Unexpected arenas.hchunk.1.size failure");
|
|
assert_d_eq(mallctl("arenas.hchunk.2.size", &huge2, &sz, NULL, 0), 0,
|
|
"Unexpected arenas.hchunk.2.size failure");
|
|
|
|
/* Test dalloc/decommit/purge cascade. */
|
|
do_dalloc = false;
|
|
do_decommit = false;
|
|
p = mallocx(huge0 * 2, 0);
|
|
assert_ptr_not_null(p, "Unexpected mallocx() error");
|
|
did_dalloc = false;
|
|
did_decommit = false;
|
|
did_purge = false;
|
|
assert_zu_eq(xallocx(p, huge0, 0, 0), huge0,
|
|
"Unexpected xallocx() failure");
|
|
assert_d_eq(mallctl("arena.0.purge", NULL, NULL, NULL, 0), 0,
|
|
"Unexpected arena.0.purge error");
|
|
assert_true(did_dalloc, "Expected dalloc");
|
|
assert_true(did_decommit, "Expected decommit");
|
|
assert_true(did_purge, "Expected purge");
|
|
dallocx(p, 0);
|
|
do_dalloc = true;
|
|
|
|
/* Test decommit/commit and observe split/merge. */
|
|
do_dalloc = false;
|
|
do_decommit = true;
|
|
p = mallocx(huge0 * 2, 0);
|
|
assert_ptr_not_null(p, "Unexpected mallocx() error");
|
|
did_decommit = false;
|
|
did_commit = false;
|
|
did_split = false;
|
|
did_merge = false;
|
|
assert_zu_eq(xallocx(p, huge0, 0, 0), huge0,
|
|
"Unexpected xallocx() failure");
|
|
assert_d_eq(mallctl("arena.0.purge", NULL, NULL, NULL, 0), 0,
|
|
"Unexpected arena.0.purge error");
|
|
assert_true(did_decommit, "Expected decommit");
|
|
assert_true(did_split, "Expected split");
|
|
assert_zu_eq(xallocx(p, huge0 * 2, 0, 0), huge0 * 2,
|
|
"Unexpected xallocx() failure");
|
|
assert_true(did_commit, "Expected commit");
|
|
assert_true(did_commit, "Expected merge");
|
|
dallocx(p, 0);
|
|
do_dalloc = true;
|
|
do_decommit = false;
|
|
|
|
/* Test purge for partial-chunk huge allocations. */
|
|
if (huge0 * 2 > huge2) {
|
|
/*
|
|
* There are at least four size classes per doubling, so a
|
|
* successful xallocx() from size=huge2 to size=huge1 is
|
|
* guaranteed to leave trailing purgeable memory.
|
|
*/
|
|
p = mallocx(huge2, 0);
|
|
assert_ptr_not_null(p, "Unexpected mallocx() error");
|
|
did_purge = false;
|
|
assert_zu_eq(xallocx(p, huge1, 0, 0), huge1,
|
|
"Unexpected xallocx() failure");
|
|
assert_true(did_purge, "Unexpected purge");
|
|
dallocx(p, 0);
|
|
}
|
|
|
|
/* Make sure non-huge allocation succeeds. */
|
|
p = mallocx(42, 0);
|
|
assert_ptr_not_null(p, "Unexpected mallocx() error");
|
|
dallocx(p, 0);
|
|
|
|
/* Restore chunk hooks. */
|
|
assert_d_eq(mallctl("arena.0.chunk_hooks", NULL, NULL, &old_hooks,
|
|
new_size), 0, "Unexpected chunk_hooks error");
|
|
assert_d_eq(mallctl("arena.0.chunk_hooks", &old_hooks, &old_size,
|
|
NULL, 0), 0, "Unexpected chunk_hooks error");
|
|
assert_ptr_eq(old_hooks.alloc, orig_hooks.alloc,
|
|
"Unexpected alloc error");
|
|
assert_ptr_eq(old_hooks.dalloc, orig_hooks.dalloc,
|
|
"Unexpected dalloc error");
|
|
assert_ptr_eq(old_hooks.commit, orig_hooks.commit,
|
|
"Unexpected commit error");
|
|
assert_ptr_eq(old_hooks.decommit, orig_hooks.decommit,
|
|
"Unexpected decommit error");
|
|
assert_ptr_eq(old_hooks.purge, orig_hooks.purge,
|
|
"Unexpected purge error");
|
|
assert_ptr_eq(old_hooks.split, orig_hooks.split,
|
|
"Unexpected split error");
|
|
assert_ptr_eq(old_hooks.merge, orig_hooks.merge,
|
|
"Unexpected merge error");
|
|
}
|
|
TEST_END
|
|
|
|
int
|
|
main(void)
|
|
{
|
|
|
|
return (test(test_chunk));
|
|
}
|