/* SPDX-License-Identifier: GPL-2.0-or-later */
#ifndef ASM_X86_ARIA_AVX_H
#define ASM_X86_ARIA_AVX_H

#include <linux/types.h>

#define ARIA_AESNI_PARALLEL_BLOCKS 16
#define ARIA_AESNI_PARALLEL_BLOCK_SIZE  (ARIA_BLOCK_SIZE * ARIA_AESNI_PARALLEL_BLOCKS)

#define ARIA_AESNI_AVX2_PARALLEL_BLOCKS 32
#define ARIA_AESNI_AVX2_PARALLEL_BLOCK_SIZE  (ARIA_BLOCK_SIZE * ARIA_AESNI_AVX2_PARALLEL_BLOCKS)

#define ARIA_GFNI_AVX512_PARALLEL_BLOCKS 64
#define ARIA_GFNI_AVX512_PARALLEL_BLOCK_SIZE  (ARIA_BLOCK_SIZE * ARIA_GFNI_AVX512_PARALLEL_BLOCKS)

asmlinkage void aria_aesni_avx_encrypt_16way(const void *ctx, u8 *dst,
                                             const u8 *src);
asmlinkage void aria_aesni_avx_decrypt_16way(const void *ctx, u8 *dst,
                                             const u8 *src);
asmlinkage void aria_aesni_avx_ctr_crypt_16way(const void *ctx, u8 *dst,
                                               const u8 *src,
                                               u8 *keystream, u8 *iv);
asmlinkage void aria_aesni_avx_gfni_encrypt_16way(const void *ctx, u8 *dst,
                                                  const u8 *src);
asmlinkage void aria_aesni_avx_gfni_decrypt_16way(const void *ctx, u8 *dst,
                                                  const u8 *src);
asmlinkage void aria_aesni_avx_gfni_ctr_crypt_16way(const void *ctx, u8 *dst,
                                                    const u8 *src,
                                                    u8 *keystream, u8 *iv);

asmlinkage void aria_aesni_avx2_encrypt_32way(const void *ctx, u8 *dst,
                                              const u8 *src);
asmlinkage void aria_aesni_avx2_decrypt_32way(const void *ctx, u8 *dst,
                                              const u8 *src);
asmlinkage void aria_aesni_avx2_ctr_crypt_32way(const void *ctx, u8 *dst,
                                                const u8 *src,
                                                u8 *keystream, u8 *iv);
asmlinkage void aria_aesni_avx2_gfni_encrypt_32way(const void *ctx, u8 *dst,
                                                   const u8 *src);
asmlinkage void aria_aesni_avx2_gfni_decrypt_32way(const void *ctx, u8 *dst,
                                                   const u8 *src);
asmlinkage void aria_aesni_avx2_gfni_ctr_crypt_32way(const void *ctx, u8 *dst,
                                                     const u8 *src,
                                                     u8 *keystream, u8 *iv);

struct aria_avx_ops {
        void (*aria_encrypt_16way)(const void *ctx, u8 *dst, const u8 *src);
        void (*aria_decrypt_16way)(const void *ctx, u8 *dst, const u8 *src);
        void (*aria_ctr_crypt_16way)(const void *ctx, u8 *dst, const u8 *src,
                                     u8 *keystream, u8 *iv);
        void (*aria_encrypt_32way)(const void *ctx, u8 *dst, const u8 *src);
        void (*aria_decrypt_32way)(const void *ctx, u8 *dst, const u8 *src);
        void (*aria_ctr_crypt_32way)(const void *ctx, u8 *dst, const u8 *src,
                                     u8 *keystream, u8 *iv);
        void (*aria_encrypt_64way)(const void *ctx, u8 *dst, const u8 *src);
        void (*aria_decrypt_64way)(const void *ctx, u8 *dst, const u8 *src);
        void (*aria_ctr_crypt_64way)(const void *ctx, u8 *dst, const u8 *src,
                                     u8 *keystream, u8 *iv);


};
#endif