#include "test/jemalloc_test.h"

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

TEST_BEGIN(test_arena_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");
		arena_dalloc_bin_locked_info_t dalloc_info;
		arena_dalloc_bin_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(arena_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_arena_slab_regind);
}