0a0fcd3e6a
This allows us to hook chosen functions and do interesting things there (in particular: reentrancy checking).
47 lines
1.3 KiB
C
47 lines
1.3 KiB
C
#include "test/jemalloc_test.h"
|
|
|
|
TEST_BEGIN(test_prof_realloc) {
|
|
tsdn_t *tsdn;
|
|
int flags;
|
|
void *p, *q;
|
|
prof_tctx_t *tctx_p, *tctx_q;
|
|
uint64_t curobjs_0, curobjs_1, curobjs_2, curobjs_3;
|
|
|
|
test_skip_if(!config_prof);
|
|
|
|
tsdn = tsdn_fetch();
|
|
flags = MALLOCX_TCACHE_NONE;
|
|
|
|
prof_cnt_all(&curobjs_0, NULL, NULL, NULL);
|
|
p = mallocx(1024, flags);
|
|
assert_ptr_not_null(p, "Unexpected mallocx() failure");
|
|
tctx_p = prof_tctx_get(tsdn, p);
|
|
assert_ptr_ne(tctx_p, (prof_tctx_t *)(uintptr_t)1U,
|
|
"Expected valid tctx");
|
|
prof_cnt_all(&curobjs_1, NULL, NULL, NULL);
|
|
assert_u64_eq(curobjs_0 + 1, curobjs_1,
|
|
"Allocation should have increased sample size");
|
|
|
|
q = rallocx(p, 2048, flags);
|
|
assert_ptr_ne(p, q, "Expected move");
|
|
assert_ptr_not_null(p, "Unexpected rmallocx() failure");
|
|
tctx_q = prof_tctx_get(tsdn, q);
|
|
assert_ptr_ne(tctx_q, (prof_tctx_t *)(uintptr_t)1U,
|
|
"Expected valid tctx");
|
|
prof_cnt_all(&curobjs_2, NULL, NULL, NULL);
|
|
assert_u64_eq(curobjs_1, curobjs_2,
|
|
"Reallocation should not have changed sample size");
|
|
|
|
dallocx(q, flags);
|
|
prof_cnt_all(&curobjs_3, NULL, NULL, NULL);
|
|
assert_u64_eq(curobjs_0, curobjs_3,
|
|
"Sample size should have returned to base level");
|
|
}
|
|
TEST_END
|
|
|
|
int
|
|
main(void) {
|
|
return test_no_reentrancy(
|
|
test_prof_realloc);
|
|
}
|