ab-riscv-interpreter 0.0.2

Composable and generic RISC-V interpreter
Documentation
//! Opaque helpers for Zbc extension

/// Carryless multiplication helper
#[cfg(any(miri, not(all(target_arch = "riscv64", target_feature = "zbc"))))]
#[inline(always)]
#[doc(hidden)]
pub fn clmul_internal(a: u64, b: u64) -> u128 {
    cfg_select! {
        // TODO: `llvm.aarch64.neon.pmull64` is not supported in Miri yet:
        //  https://github.com/rust-lang/miri/issues/3172#issuecomment-3730602707
        all(
            not(miri), target_arch = "aarch64", target_feature = "neon", target_feature = "aes"
        ) => {{
            use core::arch::aarch64::vmull_p64;

            // SAFETY: Necessary target features enabled
            unsafe { vmull_p64(a, b) }
        }}
        all(target_arch = "x86_64", target_feature = "pclmulqdq") => {{
            use core::arch::x86_64::{__m128i, _mm_clmulepi64_si128, _mm_cvtsi64_si128};
            use core::mem::transmute;

            // SAFETY: Necessary target features enabled, `__m128i` and `u128` have the same memory
            // layout
            unsafe {
                transmute::<__m128i, u128>(_mm_clmulepi64_si128(
                    _mm_cvtsi64_si128(a.cast_signed()),
                    _mm_cvtsi64_si128(b.cast_signed()),
                    0,
                ))
            }
        }}
        _ => {{
            // Generic implementation
            let mut result = 0u128;
            let a = a as u128;
            let mut b = b;
            for i in 0..u64::BITS {
                let bit = (b & 1) as u128;
                result ^= a.wrapping_shl(i) & (0u128.wrapping_sub(bit));
                b >>= 1;
            }
            result
        }}
    }
}

/// Only here to prevent compiler warnings about unused `zbc_helpers` module
#[doc(hidden)]
pub const PLACEHOLDER: () = ();