162306a36Sopenharmony_ci/* SPDX-License-Identifier: GPL-2.0-or-later */
262306a36Sopenharmony_ci#ifndef ASM_X86_ARIA_AVX_H
362306a36Sopenharmony_ci#define ASM_X86_ARIA_AVX_H
462306a36Sopenharmony_ci
562306a36Sopenharmony_ci#include <linux/types.h>
662306a36Sopenharmony_ci
762306a36Sopenharmony_ci#define ARIA_AESNI_PARALLEL_BLOCKS 16
862306a36Sopenharmony_ci#define ARIA_AESNI_PARALLEL_BLOCK_SIZE  (ARIA_BLOCK_SIZE * ARIA_AESNI_PARALLEL_BLOCKS)
962306a36Sopenharmony_ci
1062306a36Sopenharmony_ci#define ARIA_AESNI_AVX2_PARALLEL_BLOCKS 32
1162306a36Sopenharmony_ci#define ARIA_AESNI_AVX2_PARALLEL_BLOCK_SIZE  (ARIA_BLOCK_SIZE * ARIA_AESNI_AVX2_PARALLEL_BLOCKS)
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_ci#define ARIA_GFNI_AVX512_PARALLEL_BLOCKS 64
1462306a36Sopenharmony_ci#define ARIA_GFNI_AVX512_PARALLEL_BLOCK_SIZE  (ARIA_BLOCK_SIZE * ARIA_GFNI_AVX512_PARALLEL_BLOCKS)
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_ciasmlinkage void aria_aesni_avx_encrypt_16way(const void *ctx, u8 *dst,
1762306a36Sopenharmony_ci					     const u8 *src);
1862306a36Sopenharmony_ciasmlinkage void aria_aesni_avx_decrypt_16way(const void *ctx, u8 *dst,
1962306a36Sopenharmony_ci					     const u8 *src);
2062306a36Sopenharmony_ciasmlinkage void aria_aesni_avx_ctr_crypt_16way(const void *ctx, u8 *dst,
2162306a36Sopenharmony_ci					       const u8 *src,
2262306a36Sopenharmony_ci					       u8 *keystream, u8 *iv);
2362306a36Sopenharmony_ciasmlinkage void aria_aesni_avx_gfni_encrypt_16way(const void *ctx, u8 *dst,
2462306a36Sopenharmony_ci						  const u8 *src);
2562306a36Sopenharmony_ciasmlinkage void aria_aesni_avx_gfni_decrypt_16way(const void *ctx, u8 *dst,
2662306a36Sopenharmony_ci						  const u8 *src);
2762306a36Sopenharmony_ciasmlinkage void aria_aesni_avx_gfni_ctr_crypt_16way(const void *ctx, u8 *dst,
2862306a36Sopenharmony_ci						    const u8 *src,
2962306a36Sopenharmony_ci						    u8 *keystream, u8 *iv);
3062306a36Sopenharmony_ci
3162306a36Sopenharmony_ciasmlinkage void aria_aesni_avx2_encrypt_32way(const void *ctx, u8 *dst,
3262306a36Sopenharmony_ci					      const u8 *src);
3362306a36Sopenharmony_ciasmlinkage void aria_aesni_avx2_decrypt_32way(const void *ctx, u8 *dst,
3462306a36Sopenharmony_ci					      const u8 *src);
3562306a36Sopenharmony_ciasmlinkage void aria_aesni_avx2_ctr_crypt_32way(const void *ctx, u8 *dst,
3662306a36Sopenharmony_ci						const u8 *src,
3762306a36Sopenharmony_ci						u8 *keystream, u8 *iv);
3862306a36Sopenharmony_ciasmlinkage void aria_aesni_avx2_gfni_encrypt_32way(const void *ctx, u8 *dst,
3962306a36Sopenharmony_ci						   const u8 *src);
4062306a36Sopenharmony_ciasmlinkage void aria_aesni_avx2_gfni_decrypt_32way(const void *ctx, u8 *dst,
4162306a36Sopenharmony_ci						   const u8 *src);
4262306a36Sopenharmony_ciasmlinkage void aria_aesni_avx2_gfni_ctr_crypt_32way(const void *ctx, u8 *dst,
4362306a36Sopenharmony_ci						     const u8 *src,
4462306a36Sopenharmony_ci						     u8 *keystream, u8 *iv);
4562306a36Sopenharmony_ci
4662306a36Sopenharmony_cistruct aria_avx_ops {
4762306a36Sopenharmony_ci	void (*aria_encrypt_16way)(const void *ctx, u8 *dst, const u8 *src);
4862306a36Sopenharmony_ci	void (*aria_decrypt_16way)(const void *ctx, u8 *dst, const u8 *src);
4962306a36Sopenharmony_ci	void (*aria_ctr_crypt_16way)(const void *ctx, u8 *dst, const u8 *src,
5062306a36Sopenharmony_ci				     u8 *keystream, u8 *iv);
5162306a36Sopenharmony_ci	void (*aria_encrypt_32way)(const void *ctx, u8 *dst, const u8 *src);
5262306a36Sopenharmony_ci	void (*aria_decrypt_32way)(const void *ctx, u8 *dst, const u8 *src);
5362306a36Sopenharmony_ci	void (*aria_ctr_crypt_32way)(const void *ctx, u8 *dst, const u8 *src,
5462306a36Sopenharmony_ci				     u8 *keystream, u8 *iv);
5562306a36Sopenharmony_ci	void (*aria_encrypt_64way)(const void *ctx, u8 *dst, const u8 *src);
5662306a36Sopenharmony_ci	void (*aria_decrypt_64way)(const void *ctx, u8 *dst, const u8 *src);
5762306a36Sopenharmony_ci	void (*aria_ctr_crypt_64way)(const void *ctx, u8 *dst, const u8 *src,
5862306a36Sopenharmony_ci				     u8 *keystream, u8 *iv);
5962306a36Sopenharmony_ci
6062306a36Sopenharmony_ci
6162306a36Sopenharmony_ci};
6262306a36Sopenharmony_ci#endif
63