1#include "test/jemalloc_test.h" 2 3#include "jemalloc/internal/bit_util.h" 4 5#define TEST_POW2_CEIL(t, suf, pri) do { \ 6 unsigned i, pow2; \ 7 t x; \ 8 \ 9 assert_##suf##_eq(pow2_ceil_##suf(0), 0, "Unexpected result"); \ 10 \ 11 for (i = 0; i < sizeof(t) * 8; i++) { \ 12 assert_##suf##_eq(pow2_ceil_##suf(((t)1) << i), ((t)1) \ 13 << i, "Unexpected result"); \ 14 } \ 15 \ 16 for (i = 2; i < sizeof(t) * 8; i++) { \ 17 assert_##suf##_eq(pow2_ceil_##suf((((t)1) << i) - 1), \ 18 ((t)1) << i, "Unexpected result"); \ 19 } \ 20 \ 21 for (i = 0; i < sizeof(t) * 8 - 1; i++) { \ 22 assert_##suf##_eq(pow2_ceil_##suf((((t)1) << i) + 1), \ 23 ((t)1) << (i+1), "Unexpected result"); \ 24 } \ 25 \ 26 for (pow2 = 1; pow2 < 25; pow2++) { \ 27 for (x = (((t)1) << (pow2-1)) + 1; x <= ((t)1) << pow2; \ 28 x++) { \ 29 assert_##suf##_eq(pow2_ceil_##suf(x), \ 30 ((t)1) << pow2, \ 31 "Unexpected result, x=%"pri, x); \ 32 } \ 33 } \ 34} while (0) 35 36TEST_BEGIN(test_pow2_ceil_u64) { 37 TEST_POW2_CEIL(uint64_t, u64, FMTu64); 38} 39TEST_END 40 41TEST_BEGIN(test_pow2_ceil_u32) { 42 TEST_POW2_CEIL(uint32_t, u32, FMTu32); 43} 44TEST_END 45 46TEST_BEGIN(test_pow2_ceil_zu) { 47 TEST_POW2_CEIL(size_t, zu, "zu"); 48} 49TEST_END 50 51int 52main(void) { 53 return test( 54 test_pow2_ceil_u64, 55 test_pow2_ceil_u32, 56 test_pow2_ceil_zu); 57} 58