Back to home page

OSCL-LXR

 
 

    


0001 // SPDX-License-Identifier: GPL-2.0-or-later
0002 /*
0003  * Glue Code for x86_64/AVX2/AES-NI assembler optimized version of Camellia
0004  *
0005  * Copyright © 2013 Jussi Kivilinna <jussi.kivilinna@mbnet.fi>
0006  */
0007 
0008 #include <crypto/algapi.h>
0009 #include <crypto/internal/simd.h>
0010 #include <linux/crypto.h>
0011 #include <linux/err.h>
0012 #include <linux/module.h>
0013 #include <linux/types.h>
0014 
0015 #include "camellia.h"
0016 #include "ecb_cbc_helpers.h"
0017 
0018 #define CAMELLIA_AESNI_PARALLEL_BLOCKS 16
0019 #define CAMELLIA_AESNI_AVX2_PARALLEL_BLOCKS 32
0020 
0021 /* 32-way AVX2/AES-NI parallel cipher functions */
0022 asmlinkage void camellia_ecb_enc_32way(const void *ctx, u8 *dst, const u8 *src);
0023 asmlinkage void camellia_ecb_dec_32way(const void *ctx, u8 *dst, const u8 *src);
0024 
0025 asmlinkage void camellia_cbc_dec_32way(const void *ctx, u8 *dst, const u8 *src);
0026 
0027 static int camellia_setkey(struct crypto_skcipher *tfm, const u8 *key,
0028                unsigned int keylen)
0029 {
0030     return __camellia_setkey(crypto_skcipher_ctx(tfm), key, keylen);
0031 }
0032 
0033 static int ecb_encrypt(struct skcipher_request *req)
0034 {
0035     ECB_WALK_START(req, CAMELLIA_BLOCK_SIZE, CAMELLIA_AESNI_PARALLEL_BLOCKS);
0036     ECB_BLOCK(CAMELLIA_AESNI_AVX2_PARALLEL_BLOCKS, camellia_ecb_enc_32way);
0037     ECB_BLOCK(CAMELLIA_AESNI_PARALLEL_BLOCKS, camellia_ecb_enc_16way);
0038     ECB_BLOCK(2, camellia_enc_blk_2way);
0039     ECB_BLOCK(1, camellia_enc_blk);
0040     ECB_WALK_END();
0041 }
0042 
0043 static int ecb_decrypt(struct skcipher_request *req)
0044 {
0045     ECB_WALK_START(req, CAMELLIA_BLOCK_SIZE, CAMELLIA_AESNI_PARALLEL_BLOCKS);
0046     ECB_BLOCK(CAMELLIA_AESNI_AVX2_PARALLEL_BLOCKS, camellia_ecb_dec_32way);
0047     ECB_BLOCK(CAMELLIA_AESNI_PARALLEL_BLOCKS, camellia_ecb_dec_16way);
0048     ECB_BLOCK(2, camellia_dec_blk_2way);
0049     ECB_BLOCK(1, camellia_dec_blk);
0050     ECB_WALK_END();
0051 }
0052 
0053 static int cbc_encrypt(struct skcipher_request *req)
0054 {
0055     CBC_WALK_START(req, CAMELLIA_BLOCK_SIZE, -1);
0056     CBC_ENC_BLOCK(camellia_enc_blk);
0057     CBC_WALK_END();
0058 }
0059 
0060 static int cbc_decrypt(struct skcipher_request *req)
0061 {
0062     CBC_WALK_START(req, CAMELLIA_BLOCK_SIZE, CAMELLIA_AESNI_PARALLEL_BLOCKS);
0063     CBC_DEC_BLOCK(CAMELLIA_AESNI_AVX2_PARALLEL_BLOCKS, camellia_cbc_dec_32way);
0064     CBC_DEC_BLOCK(CAMELLIA_AESNI_PARALLEL_BLOCKS, camellia_cbc_dec_16way);
0065     CBC_DEC_BLOCK(2, camellia_decrypt_cbc_2way);
0066     CBC_DEC_BLOCK(1, camellia_dec_blk);
0067     CBC_WALK_END();
0068 }
0069 
0070 static struct skcipher_alg camellia_algs[] = {
0071     {
0072         .base.cra_name      = "__ecb(camellia)",
0073         .base.cra_driver_name   = "__ecb-camellia-aesni-avx2",
0074         .base.cra_priority  = 500,
0075         .base.cra_flags     = CRYPTO_ALG_INTERNAL,
0076         .base.cra_blocksize = CAMELLIA_BLOCK_SIZE,
0077         .base.cra_ctxsize   = sizeof(struct camellia_ctx),
0078         .base.cra_module    = THIS_MODULE,
0079         .min_keysize        = CAMELLIA_MIN_KEY_SIZE,
0080         .max_keysize        = CAMELLIA_MAX_KEY_SIZE,
0081         .setkey         = camellia_setkey,
0082         .encrypt        = ecb_encrypt,
0083         .decrypt        = ecb_decrypt,
0084     }, {
0085         .base.cra_name      = "__cbc(camellia)",
0086         .base.cra_driver_name   = "__cbc-camellia-aesni-avx2",
0087         .base.cra_priority  = 500,
0088         .base.cra_flags     = CRYPTO_ALG_INTERNAL,
0089         .base.cra_blocksize = CAMELLIA_BLOCK_SIZE,
0090         .base.cra_ctxsize   = sizeof(struct camellia_ctx),
0091         .base.cra_module    = THIS_MODULE,
0092         .min_keysize        = CAMELLIA_MIN_KEY_SIZE,
0093         .max_keysize        = CAMELLIA_MAX_KEY_SIZE,
0094         .ivsize         = CAMELLIA_BLOCK_SIZE,
0095         .setkey         = camellia_setkey,
0096         .encrypt        = cbc_encrypt,
0097         .decrypt        = cbc_decrypt,
0098     },
0099 };
0100 
0101 static struct simd_skcipher_alg *camellia_simd_algs[ARRAY_SIZE(camellia_algs)];
0102 
0103 static int __init camellia_aesni_init(void)
0104 {
0105     const char *feature_name;
0106 
0107     if (!boot_cpu_has(X86_FEATURE_AVX) ||
0108         !boot_cpu_has(X86_FEATURE_AVX2) ||
0109         !boot_cpu_has(X86_FEATURE_AES) ||
0110         !boot_cpu_has(X86_FEATURE_OSXSAVE)) {
0111         pr_info("AVX2 or AES-NI instructions are not detected.\n");
0112         return -ENODEV;
0113     }
0114 
0115     if (!cpu_has_xfeatures(XFEATURE_MASK_SSE | XFEATURE_MASK_YMM,
0116                 &feature_name)) {
0117         pr_info("CPU feature '%s' is not supported.\n", feature_name);
0118         return -ENODEV;
0119     }
0120 
0121     return simd_register_skciphers_compat(camellia_algs,
0122                           ARRAY_SIZE(camellia_algs),
0123                           camellia_simd_algs);
0124 }
0125 
0126 static void __exit camellia_aesni_fini(void)
0127 {
0128     simd_unregister_skciphers(camellia_algs, ARRAY_SIZE(camellia_algs),
0129                   camellia_simd_algs);
0130 }
0131 
0132 module_init(camellia_aesni_init);
0133 module_exit(camellia_aesni_fini);
0134 
0135 MODULE_LICENSE("GPL");
0136 MODULE_DESCRIPTION("Camellia Cipher Algorithm, AES-NI/AVX2 optimized");
0137 MODULE_ALIAS_CRYPTO("camellia");
0138 MODULE_ALIAS_CRYPTO("camellia-asm");