native.arm.aes.ccm

// SPDX-License-Identifier: BSD-2-Clause OR Apache-2.0
module;
#include "native/isa_import.h"
#include "native/arm/detail/crypto_constant.h"
#include "native/arm/aes.h"
export module native.arm.aes;
export import native.arm.features;
export import native.simd;
#if NATIVE_HOST_NEON || defined(NATIVE_DOXYGEN)
export namespace native {

  template<isa<arm> Arch> requires(Arch.has(arm_feature::aes))
  native_nodiscard native_inline native_const native_target("aes")
  constexpr simd<std::uint8_t, 16, Arch> aese(simd<std::uint8_t, 16, Arch> state, simd<std::uint8_t, 16, Arch> round_key) noexcept {
    if consteval { return detail::arm_constant::aes_round<false>(state, round_key); }
    else {
      auto result = detail::arm_aes::aese<Arch>(__builtin_bit_cast(uint8x16_t, state.to_native()), __builtin_bit_cast(uint8x16_t, round_key.to_native()));
      return simd<std::uint8_t, 16, Arch>::from_native(__builtin_bit_cast(typename simd<std::uint8_t, 16, Arch>::native_type, result));
    }
  }

  template<isa<arm> Arch> requires(!Arch.has(arm_feature::aes) && requires { sizeof(simd<std::uint8_t, 16, Arch>); })
  native_nodiscard consteval simd<std::uint8_t, 16, Arch> aese(simd<std::uint8_t, 16, Arch> state, simd<std::uint8_t, 16, Arch> round_key) noexcept {
    return detail::arm_constant::aes_round<false>(state, round_key);
  }

  template<isa<arm> Arch> requires(Arch.has(arm_feature::aes))
  native_nodiscard native_inline native_const native_target("aes")
  constexpr simd<std::uint8_t, 16, Arch> aesd(simd<std::uint8_t, 16, Arch> state, simd<std::uint8_t, 16, Arch> round_key) noexcept {
    if consteval { return detail::arm_constant::aes_round<true>(state, round_key); }
    else {
      auto result = detail::arm_aes::aesd<Arch>(__builtin_bit_cast(uint8x16_t, state.to_native()), __builtin_bit_cast(uint8x16_t, round_key.to_native()));
      return simd<std::uint8_t, 16, Arch>::from_native(__builtin_bit_cast(typename simd<std::uint8_t, 16, Arch>::native_type, result));
    }
  }

  template<isa<arm> Arch> requires(!Arch.has(arm_feature::aes) && requires { sizeof(simd<std::uint8_t, 16, Arch>); })
  native_nodiscard consteval simd<std::uint8_t, 16, Arch> aesd(simd<std::uint8_t, 16, Arch> state, simd<std::uint8_t, 16, Arch> round_key) noexcept {
    return detail::arm_constant::aes_round<true>(state, round_key);
  }

  template<isa<arm> Arch> requires(Arch.has(arm_feature::aes))
  native_nodiscard native_inline native_const native_target("aes")
  constexpr simd<std::uint8_t, 16, Arch> aesmc(simd<std::uint8_t, 16, Arch> state) noexcept {
    if consteval { return detail::arm_constant::aes_mix<false>(state); }
    else {
      auto result = detail::arm_aes::aesmc<Arch>(__builtin_bit_cast(uint8x16_t, state.to_native()));
      return simd<std::uint8_t, 16, Arch>::from_native(__builtin_bit_cast(typename simd<std::uint8_t, 16, Arch>::native_type, result));
    }
  }

  template<isa<arm> Arch> requires(!Arch.has(arm_feature::aes) && requires { sizeof(simd<std::uint8_t, 16, Arch>); })
  native_nodiscard consteval simd<std::uint8_t, 16, Arch> aesmc(simd<std::uint8_t, 16, Arch> state) noexcept {
    return detail::arm_constant::aes_mix<false>(state);
  }

  template<isa<arm> Arch> requires(Arch.has(arm_feature::aes))
  native_nodiscard native_inline native_const native_target("aes")
  constexpr simd<std::uint8_t, 16, Arch> aesimc(simd<std::uint8_t, 16, Arch> state) noexcept {
    if consteval { return detail::arm_constant::aes_mix<true>(state); }
    else {
      auto result = detail::arm_aes::aesimc<Arch>(__builtin_bit_cast(uint8x16_t, state.to_native()));
      return simd<std::uint8_t, 16, Arch>::from_native(__builtin_bit_cast(typename simd<std::uint8_t, 16, Arch>::native_type, result));
    }
  }

  template<isa<arm> Arch> requires(!Arch.has(arm_feature::aes) && requires { sizeof(simd<std::uint8_t, 16, Arch>); })
  native_nodiscard consteval simd<std::uint8_t, 16, Arch> aesimc(simd<std::uint8_t, 16, Arch> state) noexcept {
    return detail::arm_constant::aes_mix<true>(state);
  }
 
  // Require exact semantic shapes and a consistent feature tag.
  template<isa<arm> Arch, class... T> void aese(T...) = delete;

  template<isa<arm> Arch, class... T> void aesd(T...) = delete;

  template<isa<arm> Arch, class... T> void aesmc(T...) = delete;

  template<isa<arm> Arch, class... T> void aesimc(T...) = delete;

}
#endif