/*
 * Copyright (c) 2025 Robert Clausecker <fuz@FreeBSD.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

/*
 * test kernel for stdbit functions.
 * Requires the following macros to be defined:
 *
 * FUNCSTEM -- stem of the function name
 * SUFFIX -- type suffic
 * TYPE -- argument type
 * MKREFFUNC(ref, type) -- reference function builder
 */

#define FUNC __CONCAT(FUNCSTEM, SUFFIX)
#define REF __CONCAT(FUNCSTEM, __CONCAT(SUFFIX, _ref))

MKREFFUNC(REF, TYPE)

ATF_TC_WITHOUT_HEAD1(FUNCSTEM, SUFFIX);
ATF_TC_BODY1(FUNCSTEM, SUFFIX, tc)
{
        uintmax_t has, want;
        size_t i, j;
        TYPE value;

        /* test all single-bit patterns */
        for (i = 0; i < TYPE_WIDTH; i++) {
                value = (TYPE)1 << i;
                has = FUNC(value);
                want = REF(value);
                ATF_CHECK_EQ_MSG(has, want, "%s(%#jx) == %#jx != %#jx == %s(%#jx)",
                    __XSTRING(FUNC), (uintmax_t)value, has, want, __XSTRING(REF), (uintmax_t)value);
        }

        /* test all double-bit patterns */
        for (i = 0; i < TYPE_WIDTH; i++) {
                for (j = 0; j < i; j++) {
                        value = (TYPE)1 << i | (TYPE)1 << j;
                        has = FUNC(value);
                        want = REF(value);
                        ATF_CHECK_EQ_MSG(has, want, "%s(%#jx) == %#jx != %#jx == %s(%#jx)",
                            __XSTRING(FUNC), (uintmax_t)value, has, want, __XSTRING(REF), (uintmax_t)value);
                }
        }

        /* test all barber-pole patterns */
        value = ~(TYPE)0;
        for (i = 0; i < TYPE_WIDTH; i++) {
                has = FUNC(value);
                want = REF(value);
                ATF_CHECK_EQ_MSG(has, want, "%s(%#jx) == %#jx != %#jx == %s(%#jx)",
                    __XSTRING(FUNC), (uintmax_t)value, has, want, __XSTRING(REF), (uintmax_t)value);

                value = ~value;         
                has = FUNC(value);
                want = REF(value);
                ATF_CHECK_EQ_MSG(has, want, "%s(%#jx) == %#jx != %#jx == %s(%#jx)",
                    __XSTRING(FUNC), (uintmax_t)value, has, want, __XSTRING(REF), (uintmax_t)value);

                value = ~value << 1;
        }
}

#undef REF
#undef FUNC