native.x86.aes.ccm

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

  template<isa<x86> Arch> requires(Arch.has(x86_feature::aes))
  native_nodiscard native_inline native_const native_target("aes")
  constexpr simd<std::uint8_t, 16, Arch> aesenc(
    simd<std::uint8_t, 16, Arch> state, simd<std::uint8_t, 16, Arch> round_key) noexcept {
    if consteval {
      return detail::x86_aes_constant::round<false, false>(state, round_key);
    } else {
      return simd<std::uint8_t, 16, Arch>::from_native(
        detail::x86_aes::aesenc<Arch>(state.to_native(), round_key.to_native()));
    }
  }

  template<isa<x86> Arch> requires(!Arch.has(x86_feature::aes) &&
      requires { sizeof(simd<std::uint8_t, 16, Arch>); })
  native_nodiscard consteval simd<std::uint8_t, 16, Arch> aesenc(
    simd<std::uint8_t, 16, Arch> state, simd<std::uint8_t, 16, Arch> round_key) noexcept {
    return detail::x86_aes_constant::round<false, false>(state, round_key);
  }

  template<isa<x86> Arch> requires(Arch.has(x86_feature::aes))
  native_nodiscard native_inline native_const native_target("aes")
  constexpr simd<std::uint8_t, 16, Arch> aesenclast(
    simd<std::uint8_t, 16, Arch> state, simd<std::uint8_t, 16, Arch> round_key) noexcept {
    if consteval {
      return detail::x86_aes_constant::round<false, true>(state, round_key);
    } else {
      return simd<std::uint8_t, 16, Arch>::from_native(
        detail::x86_aes::aesenclast<Arch>(state.to_native(), round_key.to_native()));
    }
  }

  template<isa<x86> Arch> requires(!Arch.has(x86_feature::aes) &&
      requires { sizeof(simd<std::uint8_t, 16, Arch>); })
  native_nodiscard consteval simd<std::uint8_t, 16, Arch> aesenclast(
    simd<std::uint8_t, 16, Arch> state, simd<std::uint8_t, 16, Arch> round_key) noexcept {
    return detail::x86_aes_constant::round<false, true>(state, round_key);
  }

  template<isa<x86> Arch> requires(Arch.has(x86_feature::aes))
  native_nodiscard native_inline native_const native_target("aes")
  constexpr simd<std::uint8_t, 16, Arch> aesdec(
    simd<std::uint8_t, 16, Arch> state, simd<std::uint8_t, 16, Arch> round_key) noexcept {
    if consteval {
      return detail::x86_aes_constant::round<true, false>(state, round_key);
    } else {
      return simd<std::uint8_t, 16, Arch>::from_native(
        detail::x86_aes::aesdec<Arch>(state.to_native(), round_key.to_native()));
    }
  }

  template<isa<x86> Arch> requires(!Arch.has(x86_feature::aes) &&
      requires { sizeof(simd<std::uint8_t, 16, Arch>); })
  native_nodiscard consteval simd<std::uint8_t, 16, Arch> aesdec(
    simd<std::uint8_t, 16, Arch> state, simd<std::uint8_t, 16, Arch> round_key) noexcept {
    return detail::x86_aes_constant::round<true, false>(state, round_key);
  }

  template<isa<x86> Arch> requires(Arch.has(x86_feature::aes))
  native_nodiscard native_inline native_const native_target("aes")
  constexpr simd<std::uint8_t, 16, Arch> aesdeclast(
    simd<std::uint8_t, 16, Arch> state, simd<std::uint8_t, 16, Arch> round_key) noexcept {
    if consteval {
      return detail::x86_aes_constant::round<true, true>(state, round_key);
    } else {
      return simd<std::uint8_t, 16, Arch>::from_native(
        detail::x86_aes::aesdeclast<Arch>(state.to_native(), round_key.to_native()));
    }
  }

  template<isa<x86> Arch> requires(!Arch.has(x86_feature::aes) &&
      requires { sizeof(simd<std::uint8_t, 16, Arch>); })
  native_nodiscard consteval simd<std::uint8_t, 16, Arch> aesdeclast(
    simd<std::uint8_t, 16, Arch> state, simd<std::uint8_t, 16, Arch> round_key) noexcept {
    return detail::x86_aes_constant::round<true, true>(state, round_key);
  }

  template<isa<x86> Arch> requires(Arch.has(x86_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::x86_aes_constant::inverse_mix(state);
    } else {
      return simd<std::uint8_t, 16, Arch>::from_native(detail::x86_aes::aesimc<Arch>(state.to_native()));
    }
  }

  template<isa<x86> Arch> requires(!Arch.has(x86_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::x86_aes_constant::inverse_mix(state);
  }

  template<isa<x86> Arch, unsigned Imm8> requires(Arch.has(x86_feature::aes) && Imm8 <= 255)
  native_nodiscard native_inline native_const native_target("aes")
  constexpr simd<std::uint8_t, 16, Arch> aeskeygenassist(simd<std::uint8_t, 16, Arch> state) noexcept {
    if consteval {
      return detail::x86_aes_constant::keygen<Imm8>(state);
    } else {
      return simd<std::uint8_t, 16, Arch>::from_native(
        detail::x86_aes::aeskeygenassist<Arch, Imm8>(state.to_native()));
    }
  }

  template<isa<x86> Arch, unsigned Imm8> requires(!Arch.has(x86_feature::aes) && Imm8 <= 255 &&
      requires { sizeof(simd<std::uint8_t, 16, Arch>); })
  native_nodiscard consteval simd<std::uint8_t, 16, Arch> aeskeygenassist(simd<std::uint8_t, 16, Arch> state) noexcept {
    return detail::x86_aes_constant::keygen<Imm8>(state);
  }

  template<isa<x86> Arch, class... Args>
  void aesenc(Args...) = delete;

  template<isa<x86> Arch, class... Args>
  void aesenclast(Args...) = delete;

  template<isa<x86> Arch, class... Args>
  void aesdec(Args...) = delete;

  template<isa<x86> Arch, class... Args>
  void aesdeclast(Args...) = delete;

  template<isa<x86> Arch, class... Args>
  void aesimc(Args...) = delete;

  template<isa<x86> Arch, unsigned Imm8, class... Args>
  void aeskeygenassist(Args...) = delete;

}
#endif