rscrypto 0.1.1

Pure Rust cryptography, hardware-accelerated: BLAKE3, SHA-2/3, AES-GCM, ChaCha20-Poly1305, Ed25519, X25519, HMAC, HKDF, Argon2, CRC. no_std, WASM, ten CPU architectures.
Documentation
#![cfg_attr(test, allow(dead_code))]

#[cfg(all(feature = "rapidhash", any(test, feature = "diag")))]
use super::{
  dispatch_tables::DispatchTable,
  kernels::{RapidHashKernelId, hash64_fn, required_caps},
};
#[cfg(all(feature = "rapidhash", any(test, feature = "diag")))]
use crate::{backend::cache::OnceCache, platform::Caps};

#[cfg(all(feature = "rapidhash", any(test, feature = "diag")))]
type Hash64Fn = fn(&[u8], u64) -> u64;

#[cfg(all(feature = "rapidhash", any(test, feature = "diag")))]
#[derive(Clone, Copy)]
struct ActiveDispatch {
  boundaries: [usize; 3],
  xs: Hash64Fn,
  s: Hash64Fn,
  m: Hash64Fn,
  l: Hash64Fn,
  xs_name: &'static str,
  s_name: &'static str,
  m_name: &'static str,
  l_name: &'static str,
}

#[cfg(all(feature = "rapidhash", any(test, feature = "diag")))]
static ACTIVE: OnceCache<ActiveDispatch> = OnceCache::new();

#[cfg(all(feature = "rapidhash", any(test, feature = "diag")))]
#[inline]
#[must_use]
fn resolve(id: RapidHashKernelId, caps: Caps) -> RapidHashKernelId {
  if caps.has(required_caps(id)) {
    id
  } else {
    RapidHashKernelId::Portable
  }
}

#[cfg(all(feature = "rapidhash", any(test, feature = "diag")))]
#[inline]
#[must_use]
fn active() -> ActiveDispatch {
  ACTIVE.get_or_init(|| {
    let caps = crate::platform::caps();
    let table: &'static DispatchTable = super::dispatch_tables::select_runtime_table(caps);

    let xs_id = resolve(table.xs, caps);
    let s_id = resolve(table.s, caps);
    let m_id = resolve(table.m, caps);
    let l_id = resolve(table.l, caps);

    ActiveDispatch {
      boundaries: table.boundaries,
      xs: hash64_fn(xs_id),
      s: hash64_fn(s_id),
      m: hash64_fn(m_id),
      l: hash64_fn(l_id),
      xs_name: xs_id.as_str(),
      s_name: s_id.as_str(),
      m_name: m_id.as_str(),
      l_name: l_id.as_str(),
    }
  })
}

#[cfg(all(feature = "rapidhash", any(test, feature = "diag")))]
#[inline]
#[must_use]
fn select(d: &ActiveDispatch, len: usize) -> (Hash64Fn, &'static str) {
  let [xs_max, s_max, m_max] = d.boundaries;
  if len <= xs_max {
    (d.xs, d.xs_name)
  } else if len <= s_max {
    (d.s, d.s_name)
  } else if len <= m_max {
    (d.m, d.m_name)
  } else {
    (d.l, d.l_name)
  }
}

#[cfg(all(feature = "rapidhash", any(test, feature = "diag")))]
#[inline]
#[must_use]
pub fn kernel_name_for_len(len: usize) -> &'static str {
  let d = active();
  select(&d, len).1
}

/// Inline fast-path: RapidHash currently only has a portable kernel, so
/// bypass dispatch machinery entirely and call the scalar implementation
/// directly. This eliminates OnceCache load + indirect call overhead that
/// dominated the 0-64 B benchmarks.
#[inline]
#[must_use]
pub fn hash64_with_seed(seed: u64, data: &[u8]) -> u64 {
  super::rapidhash_v3_with_seed(data, seed)
}

#[inline]
#[must_use]
pub fn hash128_with_seed(seed: u64, data: &[u8]) -> u128 {
  let lo = super::rapidhash_v3_with_seed(data, seed);
  let hi = super::rapidhash_v3_with_seed(data, seed ^ super::V3_HI_SEED);
  (lo as u128) | ((hi as u128) << 64)
}

#[inline(always)]
#[must_use]
pub fn hash64_fast_with_seed(seed: u64, data: &[u8]) -> u64 {
  super::rapidhash_fast_with_seed(data, seed)
}

#[inline(always)]
#[must_use]
pub fn hash128_fast_with_seed(seed: u64, data: &[u8]) -> u128 {
  super::rapidhash_fast_128_with_seed(data, seed)
}