79 lines
2.4 KiB
C
79 lines
2.4 KiB
C
|
#include "test/jemalloc_test.h"
|
||
|
|
|
||
|
|
static void assert_small_allocation_sampled(void *ptr, size_t size) {
|
||
|
|
assert_ptr_not_null(ptr, "Unexpected malloc failure");
|
||
|
|
assert_zu_le(size, SC_SMALL_MAXCLASS, "Unexpected large size class");
|
||
|
|
edata_t *edata = emap_edata_lookup(TSDN_NULL, &arena_emap_global, ptr);
|
||
|
|
assert_ptr_not_null(edata, "Unable to find edata for allocation");
|
||
|
|
expect_false(edata_slab_get(edata),
|
||
|
|
"Sampled small allocations should not be placed on slabs");
|
||
|
|
expect_ptr_eq(edata_base_get(edata), ptr,
|
||
|
|
"Sampled allocations should be page-aligned");
|
||
|
|
expect_zu_eq(edata_usize_get(edata), size,
|
||
|
|
"Edata usize did not match requested size");
|
||
|
|
expect_zu_eq(edata_size_get(edata), PAGE_CEILING(size) + sz_large_pad,
|
||
|
|
"Edata actual size was not a multiple of PAGE");
|
||
|
|
prof_tctx_t *prof_tctx = edata_prof_tctx_get(edata);
|
||
|
|
expect_ptr_not_null(prof_tctx, "Edata had null prof_tctx");
|
||
|
|
expect_ptr_not_null(prof_tctx->tdata,
|
||
|
|
"Edata had null prof_tdata despite being sampled");
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_BEGIN(test_profile_small_allocations) {
|
||
|
|
test_skip_if(!config_prof);
|
||
|
|
|
||
|
|
for (szind_t index = 0; index < SC_NBINS; index++) {
|
||
|
|
size_t size = sz_index2size(index);
|
||
|
|
void *ptr = malloc(size);
|
||
|
|
assert_small_allocation_sampled(ptr, size);
|
||
|
|
free(ptr);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
TEST_END
|
||
|
|
|
||
|
|
TEST_BEGIN(test_profile_small_reallocations_growing) {
|
||
|
|
test_skip_if(!config_prof);
|
||
|
|
|
||
|
|
void *ptr = NULL;
|
||
|
|
for (szind_t index = 0; index < SC_NBINS; index++) {
|
||
|
|
size_t size = sz_index2size(index);
|
||
|
|
ptr = realloc(ptr, size);
|
||
|
|
assert_small_allocation_sampled(ptr, size);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
TEST_END
|
||
|
|
|
||
|
|
TEST_BEGIN(test_profile_small_reallocations_shrinking) {
|
||
|
|
test_skip_if(!config_prof);
|
||
|
|
|
||
|
|
void *ptr = NULL;
|
||
|
|
for (szind_t index = SC_NBINS; index-- > 0;) {
|
||
|
|
size_t size = sz_index2size(index);
|
||
|
|
ptr = realloc(ptr, size);
|
||
|
|
assert_small_allocation_sampled(ptr, size);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
TEST_END
|
||
|
|
|
||
|
|
TEST_BEGIN(test_profile_small_reallocations_same_size_class) {
|
||
|
|
test_skip_if(!config_prof);
|
||
|
|
|
||
|
|
for (szind_t index = 0; index < SC_NBINS; index++) {
|
||
|
|
size_t size = sz_index2size(index);
|
||
|
|
void *ptr = malloc(size);
|
||
|
|
assert_small_allocation_sampled(ptr, size);
|
||
|
|
ptr = realloc(ptr, size - 1);
|
||
|
|
assert_small_allocation_sampled(ptr, size);
|
||
|
|
free(ptr);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
TEST_END
|
||
|
|
|
||
|
|
int
|
||
|
|
main(void) {
|
||
|
|
return test(test_profile_small_allocations,
|
||
|
|
test_profile_small_reallocations_growing,
|
||
|
|
test_profile_small_reallocations_shrinking,
|
||
|
|
test_profile_small_reallocations_same_size_class);
|
||
|
|
}
|