/* SPDX-License-Identifier: GPL-2.0 */
#ifndef __ASM_BITREV_H
#define __ASM_BITREV_H

#include <linux/types.h>
#include <asm/cpufeature-macros.h>
#include <asm/hwcap.h>
#include <asm-generic/bitops/__bitrev.h>

static __always_inline __attribute_const__ u32 __arch_bitrev32(u32 x)
{
        unsigned long result;

        if (!riscv_has_extension_likely(RISCV_ISA_EXT_ZBKB))
                return generic___bitrev32(x);

        asm volatile(
                ".option push\n"
                ".option arch,+zbkb\n"
                "rev8 %0, %1\n"
                "brev8 %0, %0\n"
                ".option pop"
                : "=r" (result) : "r" ((long)x)
        );

        return result >> (__riscv_xlen - 32);
}

static __always_inline __attribute_const__ u16 __arch_bitrev16(u16 x)
{
        return __arch_bitrev32(x) >> 16;
}

static __always_inline __attribute_const__ u8 __arch_bitrev8(u8 x)
{
        unsigned long result;

        if (!riscv_has_extension_likely(RISCV_ISA_EXT_ZBKB))
                return generic___bitrev8(x);

        asm volatile(
                ".option push\n"
                ".option arch,+zbkb\n"
                "brev8 %0, %1\n"
                ".option pop"
                : "=r" (result) : "r" ((long)x)
        );

        return result;
}
#endif