jemalloc/test/unit/slab.c

#include "test/jemalloc_test.h"

#define INVALID_ARENA_IND ((1U << MALLOCX_ARENA_BITS) - 1)

TEST_BEGIN(test_bin_slab_regind) {
	szind_t binind;

	for (binind = 0; binind < SC_NBINS; binind++) {
		size_t            regind;
		edata_t           slab;
		const bin_info_t *bin_info = &bin_infos[binind];
		edata_init(&slab, INVALID_ARENA_IND,
		    mallocx(bin_info->slab_size, MALLOCX_LG_ALIGN(LG_PAGE)),
		    bin_info->slab_size, true, binind, 0, extent_state_active,
		    false, true, EXTENT_PAI_PAC, EXTENT_NOT_HEAD);
		expect_ptr_not_null(
		    edata_addr_get(&slab), "Unexpected malloc() failure");
		bin_dalloc_locked_info_t dalloc_info;
		bin_dalloc_locked_begin(&dalloc_info, binind);
		for (regind = 0; regind < bin_info->nregs; regind++) {
			void *reg = (void *)((uintptr_t)edata_addr_get(&slab)
			    + (bin_info->reg_size * regind));
			expect_zu_eq(
			    bin_slab_regind(&dalloc_info, binind, &slab, reg),
			    regind,
			    "Incorrect region index computed for size %zu",
			    bin_info->reg_size);
		}
		free(edata_addr_get(&slab));
	}
}
TEST_END

int
main(void) {
	return test(test_bin_slab_regind);
}