arch/x86/crypto/cast6_avx_glue.c

   1 /*
   2  * Glue Code for the AVX assembler implemention of the Cast6 Cipher
   3  *
   4  * Copyright (C) 2012 Johannes Goetzfried
   5  *     <Johannes.Goetzfried@informatik.stud.uni-erlangen.de>
   6  *
   7  * Copyright © 2013 Jussi Kivilinna <jussi.kivilinna@iki.fi>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
  22  * USA
  23  *
  24  */
  25
  26 #include <linux/module.h>
  27 #include <linux/hardirq.h>
  28 #include <linux/types.h>
  29 #include <linux/crypto.h>
  30 #include <linux/err.h>
  31 #include <crypto/ablk_helper.h>
  32 #include <crypto/algapi.h>
  33 #include <crypto/cast6.h>
  34 #include <crypto/cryptd.h>
  35 #include <crypto/b128ops.h>
  36 #include <crypto/ctr.h>
  37 #include <crypto/lrw.h>
  38 #include <crypto/xts.h>
  39 #include <asm/fpu/api.h>
  40 #include <asm/crypto/glue_helper.h>
  41
  42 #define CAST6_PARALLEL_BLOCKS 8
  43
  44 asmlinkage void cast6_ecb_enc_8way(struct cast6_ctx *ctx, u8 *dst,
  45                                    const u8 *src);
  46 asmlinkage void cast6_ecb_dec_8way(struct cast6_ctx *ctx, u8 *dst,
  47                                    const u8 *src);
  48
  49 asmlinkage void cast6_cbc_dec_8way(struct cast6_ctx *ctx, u8 *dst,
  50                                    const u8 *src);
  51 asmlinkage void cast6_ctr_8way(struct cast6_ctx *ctx, u8 *dst, const u8 *src,
  52                                le128 *iv);
  53
  54 asmlinkage void cast6_xts_enc_8way(struct cast6_ctx *ctx, u8 *dst,
  55                                    const u8 *src, le128 *iv);
  56 asmlinkage void cast6_xts_dec_8way(struct cast6_ctx *ctx, u8 *dst,
  57                                    const u8 *src, le128 *iv);
  58
  59 static void cast6_xts_enc(void *ctx, u128 *dst, const u128 *src, le128 *iv)
  60 {
  61         glue_xts_crypt_128bit_one(ctx, dst, src, iv,
  62                                   GLUE_FUNC_CAST(__cast6_encrypt));
  63 }
  64
  65 static void cast6_xts_dec(void *ctx, u128 *dst, const u128 *src, le128 *iv)
  66 {
  67         glue_xts_crypt_128bit_one(ctx, dst, src, iv,
  68                                   GLUE_FUNC_CAST(__cast6_decrypt));
  69 }
  70
  71 static void cast6_crypt_ctr(void *ctx, u128 *dst, const u128 *src, le128 *iv)
  72 {
  73         be128 ctrblk;
  74
  75         le128_to_be128(&ctrblk, iv);
  76         le128_inc(iv);
  77
  78         __cast6_encrypt(ctx, (u8 *)&ctrblk, (u8 *)&ctrblk);
  79         u128_xor(dst, src, (u128 *)&ctrblk);
  80 }
  81
  82 static const struct common_glue_ctx cast6_enc = {
  83         .num_funcs = 2,
  84         .fpu_blocks_limit = CAST6_PARALLEL_BLOCKS,
  85
  86         .funcs = { {
  87                 .num_blocks = CAST6_PARALLEL_BLOCKS,
  88                 .fn_u = { .ecb = GLUE_FUNC_CAST(cast6_ecb_enc_8way) }
  89         }, {
  90                 .num_blocks = 1,
  91                 .fn_u = { .ecb = GLUE_FUNC_CAST(__cast6_encrypt) }
  92         } }
  93 };
  94
  95 static const struct common_glue_ctx cast6_ctr = {
  96         .num_funcs = 2,
  97         .fpu_blocks_limit = CAST6_PARALLEL_BLOCKS,
  98
  99         .funcs = { {
 100                 .num_blocks = CAST6_PARALLEL_BLOCKS,
 101                 .fn_u = { .ctr = GLUE_CTR_FUNC_CAST(cast6_ctr_8way) }
 102         }, {
 103                 .num_blocks = 1,
 104                 .fn_u = { .ctr = GLUE_CTR_FUNC_CAST(cast6_crypt_ctr) }
 105         } }
 106 };
 107
 108 static const struct common_glue_ctx cast6_enc_xts = {
 109         .num_funcs = 2,
 110         .fpu_blocks_limit = CAST6_PARALLEL_BLOCKS,
 111
 112         .funcs = { {
 113                 .num_blocks = CAST6_PARALLEL_BLOCKS,
 114                 .fn_u = { .xts = GLUE_XTS_FUNC_CAST(cast6_xts_enc_8way) }
 115         }, {
 116                 .num_blocks = 1,
 117                 .fn_u = { .xts = GLUE_XTS_FUNC_CAST(cast6_xts_enc) }
 118         } }
 119 };
 120
 121 static const struct common_glue_ctx cast6_dec = {
 122         .num_funcs = 2,
 123         .fpu_blocks_limit = CAST6_PARALLEL_BLOCKS,
 124
 125         .funcs = { {
 126                 .num_blocks = CAST6_PARALLEL_BLOCKS,
 127                 .fn_u = { .ecb = GLUE_FUNC_CAST(cast6_ecb_dec_8way) }
 128         }, {
 129                 .num_blocks = 1,
 130                 .fn_u = { .ecb = GLUE_FUNC_CAST(__cast6_decrypt) }
 131         } }
 132 };
 133
 134 static const struct common_glue_ctx cast6_dec_cbc = {
 135         .num_funcs = 2,
 136         .fpu_blocks_limit = CAST6_PARALLEL_BLOCKS,
 137
 138         .funcs = { {
 139                 .num_blocks = CAST6_PARALLEL_BLOCKS,
 140                 .fn_u = { .cbc = GLUE_CBC_FUNC_CAST(cast6_cbc_dec_8way) }
 141         }, {
 142                 .num_blocks = 1,
 143                 .fn_u = { .cbc = GLUE_CBC_FUNC_CAST(__cast6_decrypt) }
 144         } }
 145 };
 146
 147 static const struct common_glue_ctx cast6_dec_xts = {
 148         .num_funcs = 2,
 149         .fpu_blocks_limit = CAST6_PARALLEL_BLOCKS,
 150
 151         .funcs = { {
 152                 .num_blocks = CAST6_PARALLEL_BLOCKS,
 153                 .fn_u = { .xts = GLUE_XTS_FUNC_CAST(cast6_xts_dec_8way) }
 154         }, {
 155                 .num_blocks = 1,
 156                 .fn_u = { .xts = GLUE_XTS_FUNC_CAST(cast6_xts_dec) }
 157         } }
 158 };
 159
 160 static int ecb_encrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
 161                        struct scatterlist *src, unsigned int nbytes)
 162 {
 163         return glue_ecb_crypt_128bit(&cast6_enc, desc, dst, src, nbytes);
 164 }
 165
 166 static int ecb_decrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
 167                        struct scatterlist *src, unsigned int nbytes)
 168 {
 169         return glue_ecb_crypt_128bit(&cast6_dec, desc, dst, src, nbytes);
 170 }
 171
 172 static int cbc_encrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
 173                        struct scatterlist *src, unsigned int nbytes)
 174 {
 175         return glue_cbc_encrypt_128bit(GLUE_FUNC_CAST(__cast6_encrypt), desc,
 176                                        dst, src, nbytes);
 177 }
 178
 179 static int cbc_decrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
 180                        struct scatterlist *src, unsigned int nbytes)
 181 {
 182         return glue_cbc_decrypt_128bit(&cast6_dec_cbc, desc, dst, src,
 183                                        nbytes);
 184 }
 185
 186 static int ctr_crypt(struct blkcipher_desc *desc, struct scatterlist *dst,
 187                      struct scatterlist *src, unsigned int nbytes)
 188 {
 189         return glue_ctr_crypt_128bit(&cast6_ctr, desc, dst, src, nbytes);
 190 }
 191
 192 static inline bool cast6_fpu_begin(bool fpu_enabled, unsigned int nbytes)
 193 {
 194         return glue_fpu_begin(CAST6_BLOCK_SIZE, CAST6_PARALLEL_BLOCKS,
 195                               NULL, fpu_enabled, nbytes);
 196 }
 197
 198 static inline void cast6_fpu_end(bool fpu_enabled)
 199 {
 200         glue_fpu_end(fpu_enabled);
 201 }
 202
 203 struct crypt_priv {
 204         struct cast6_ctx *ctx;
 205         bool fpu_enabled;
 206 };
 207
 208 static void encrypt_callback(void *priv, u8 *srcdst, unsigned int nbytes)
 209 {
 210         const unsigned int bsize = CAST6_BLOCK_SIZE;
 211         struct crypt_priv *ctx = priv;
 212         int i;
 213
 214         ctx->fpu_enabled = cast6_fpu_begin(ctx->fpu_enabled, nbytes);
 215
 216         if (nbytes == bsize * CAST6_PARALLEL_BLOCKS) {
 217                 cast6_ecb_enc_8way(ctx->ctx, srcdst, srcdst);
 218                 return;
 219         }
 220
 221         for (i = 0; i < nbytes / bsize; i++, srcdst += bsize)
 222                 __cast6_encrypt(ctx->ctx, srcdst, srcdst);
 223 }
 224
 225 static void decrypt_callback(void *priv, u8 *srcdst, unsigned int nbytes)
 226 {
 227         const unsigned int bsize = CAST6_BLOCK_SIZE;
 228         struct crypt_priv *ctx = priv;
 229         int i;
 230
 231         ctx->fpu_enabled = cast6_fpu_begin(ctx->fpu_enabled, nbytes);
 232
 233         if (nbytes == bsize * CAST6_PARALLEL_BLOCKS) {
 234                 cast6_ecb_dec_8way(ctx->ctx, srcdst, srcdst);
 235                 return;
 236         }
 237
 238         for (i = 0; i < nbytes / bsize; i++, srcdst += bsize)
 239                 __cast6_decrypt(ctx->ctx, srcdst, srcdst);
 240 }
 241
 242 struct cast6_lrw_ctx {
 243         struct lrw_table_ctx lrw_table;
 244         struct cast6_ctx cast6_ctx;
 245 };
 246
 247 static int lrw_cast6_setkey(struct crypto_tfm *tfm, const u8 *key,
 248                               unsigned int keylen)
 249 {
 250         struct cast6_lrw_ctx *ctx = crypto_tfm_ctx(tfm);
 251         int err;
 252
 253         err = __cast6_setkey(&ctx->cast6_ctx, key, keylen - CAST6_BLOCK_SIZE,
 254                              &tfm->crt_flags);
 255         if (err)
 256                 return err;
 257
 258         return lrw_init_table(&ctx->lrw_table, key + keylen - CAST6_BLOCK_SIZE);
 259 }
 260
 261 static int lrw_encrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
 262                        struct scatterlist *src, unsigned int nbytes)
 263 {
 264         struct cast6_lrw_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
 265         be128 buf[CAST6_PARALLEL_BLOCKS];
 266         struct crypt_priv crypt_ctx = {
 267                 .ctx = &ctx->cast6_ctx,
 268                 .fpu_enabled = false,
 269         };
 270         struct lrw_crypt_req req = {
 271                 .tbuf = buf,
 272                 .tbuflen = sizeof(buf),
 273
 274                 .table_ctx = &ctx->lrw_table,
 275                 .crypt_ctx = &crypt_ctx,
 276                 .crypt_fn = encrypt_callback,
 277         };
 278         int ret;
 279
 280         desc->flags &= ~CRYPTO_TFM_REQ_MAY_SLEEP;
 281         ret = lrw_crypt(desc, dst, src, nbytes, &req);
 282         cast6_fpu_end(crypt_ctx.fpu_enabled);
 283
 284         return ret;
 285 }
 286
 287 static int lrw_decrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
 288                        struct scatterlist *src, unsigned int nbytes)
 289 {
 290         struct cast6_lrw_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
 291         be128 buf[CAST6_PARALLEL_BLOCKS];
 292         struct crypt_priv crypt_ctx = {
 293                 .ctx = &ctx->cast6_ctx,
 294                 .fpu_enabled = false,
 295         };
 296         struct lrw_crypt_req req = {
 297                 .tbuf = buf,
 298                 .tbuflen = sizeof(buf),
 299
 300                 .table_ctx = &ctx->lrw_table,
 301                 .crypt_ctx = &crypt_ctx,
 302                 .crypt_fn = decrypt_callback,
 303         };
 304         int ret;
 305
 306         desc->flags &= ~CRYPTO_TFM_REQ_MAY_SLEEP;
 307         ret = lrw_crypt(desc, dst, src, nbytes, &req);
 308         cast6_fpu_end(crypt_ctx.fpu_enabled);
 309
 310         return ret;
 311 }
 312
 313 static void lrw_exit_tfm(struct crypto_tfm *tfm)
 314 {
 315         struct cast6_lrw_ctx *ctx = crypto_tfm_ctx(tfm);
 316
 317         lrw_free_table(&ctx->lrw_table);
 318 }
 319
 320 struct cast6_xts_ctx {
 321         struct cast6_ctx tweak_ctx;
 322         struct cast6_ctx crypt_ctx;
 323 };
 324
 325 static int xts_cast6_setkey(struct crypto_tfm *tfm, const u8 *key,
 326                               unsigned int keylen)
 327 {
 328         struct cast6_xts_ctx *ctx = crypto_tfm_ctx(tfm);
 329         u32 *flags = &tfm->crt_flags;
 330         int err;
 331
 332         err = xts_check_key(tfm, key, keylen);
 333         if (err)
 334                 return err;
 335
 336         /* first half of xts-key is for crypt */
 337         err = __cast6_setkey(&ctx->crypt_ctx, key, keylen / 2, flags);
 338         if (err)
 339                 return err;
 340
 341         /* second half of xts-key is for tweak */
 342         return __cast6_setkey(&ctx->tweak_ctx, key + keylen / 2, keylen / 2,
 343                               flags);
 344 }
 345
 346 static int xts_encrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
 347                        struct scatterlist *src, unsigned int nbytes)
 348 {
 349         struct cast6_xts_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
 350
 351         return glue_xts_crypt_128bit(&cast6_enc_xts, desc, dst, src, nbytes,
 352                                      XTS_TWEAK_CAST(__cast6_encrypt),
 353                                      &ctx->tweak_ctx, &ctx->crypt_ctx);
 354 }
 355
 356 static int xts_decrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
 357                        struct scatterlist *src, unsigned int nbytes)
 358 {
 359         struct cast6_xts_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
 360
 361         return glue_xts_crypt_128bit(&cast6_dec_xts, desc, dst, src, nbytes,
 362                                      XTS_TWEAK_CAST(__cast6_encrypt),
 363                                      &ctx->tweak_ctx, &ctx->crypt_ctx);
 364 }
 365
 366 static struct crypto_alg cast6_algs[10] = { {
 367         .cra_name               = "__ecb-cast6-avx",
 368         .cra_driver_name        = "__driver-ecb-cast6-avx",
 369         .cra_priority           = 0,
 370         .cra_flags              = CRYPTO_ALG_TYPE_BLKCIPHER |
 371                                   CRYPTO_ALG_INTERNAL,
 372         .cra_blocksize          = CAST6_BLOCK_SIZE,
 373         .cra_ctxsize            = sizeof(struct cast6_ctx),
 374         .cra_alignmask          = 0,
 375         .cra_type               = &crypto_blkcipher_type,
 376         .cra_module             = THIS_MODULE,
 377         .cra_u = {
 378                 .blkcipher = {
 379                         .min_keysize    = CAST6_MIN_KEY_SIZE,
 380                         .max_keysize    = CAST6_MAX_KEY_SIZE,
 381                         .setkey         = cast6_setkey,
 382                         .encrypt        = ecb_encrypt,
 383                         .decrypt        = ecb_decrypt,
 384                 },
 385         },
 386 }, {
 387         .cra_name               = "__cbc-cast6-avx",
 388         .cra_driver_name        = "__driver-cbc-cast6-avx",
 389         .cra_priority           = 0,
 390         .cra_flags              = CRYPTO_ALG_TYPE_BLKCIPHER |
 391                                   CRYPTO_ALG_INTERNAL,
 392         .cra_blocksize          = CAST6_BLOCK_SIZE,
 393         .cra_ctxsize            = sizeof(struct cast6_ctx),
 394         .cra_alignmask          = 0,
 395         .cra_type               = &crypto_blkcipher_type,
 396         .cra_module             = THIS_MODULE,
 397         .cra_u = {
 398                 .blkcipher = {
 399                         .min_keysize    = CAST6_MIN_KEY_SIZE,
 400                         .max_keysize    = CAST6_MAX_KEY_SIZE,
 401                         .setkey         = cast6_setkey,
 402                         .encrypt        = cbc_encrypt,
 403                         .decrypt        = cbc_decrypt,
 404                 },
 405         },
 406 }, {
 407         .cra_name               = "__ctr-cast6-avx",
 408         .cra_driver_name        = "__driver-ctr-cast6-avx",
 409         .cra_priority           = 0,
 410         .cra_flags              = CRYPTO_ALG_TYPE_BLKCIPHER |
 411                                   CRYPTO_ALG_INTERNAL,
 412         .cra_blocksize          = 1,
 413         .cra_ctxsize            = sizeof(struct cast6_ctx),
 414         .cra_alignmask          = 0,
 415         .cra_type               = &crypto_blkcipher_type,
 416         .cra_module             = THIS_MODULE,
 417         .cra_u = {
 418                 .blkcipher = {
 419                         .min_keysize    = CAST6_MIN_KEY_SIZE,
 420                         .max_keysize    = CAST6_MAX_KEY_SIZE,
 421                         .ivsize         = CAST6_BLOCK_SIZE,
 422                         .setkey         = cast6_setkey,
 423                         .encrypt        = ctr_crypt,
 424                         .decrypt        = ctr_crypt,
 425                 },
 426         },
 427 }, {
 428         .cra_name               = "__lrw-cast6-avx",
 429         .cra_driver_name        = "__driver-lrw-cast6-avx",
 430         .cra_priority           = 0,
 431         .cra_flags              = CRYPTO_ALG_TYPE_BLKCIPHER |
 432                                   CRYPTO_ALG_INTERNAL,
 433         .cra_blocksize          = CAST6_BLOCK_SIZE,
 434         .cra_ctxsize            = sizeof(struct cast6_lrw_ctx),
 435         .cra_alignmask          = 0,
 436         .cra_type               = &crypto_blkcipher_type,
 437         .cra_module             = THIS_MODULE,
 438         .cra_exit               = lrw_exit_tfm,
 439         .cra_u = {
 440                 .blkcipher = {
 441                         .min_keysize    = CAST6_MIN_KEY_SIZE +
 442                                           CAST6_BLOCK_SIZE,
 443                         .max_keysize    = CAST6_MAX_KEY_SIZE +
 444                                           CAST6_BLOCK_SIZE,
 445                         .ivsize         = CAST6_BLOCK_SIZE,
 446                         .setkey         = lrw_cast6_setkey,
 447                         .encrypt        = lrw_encrypt,
 448                         .decrypt        = lrw_decrypt,
 449                 },
 450         },
 451 }, {
 452         .cra_name               = "__xts-cast6-avx",
 453         .cra_driver_name        = "__driver-xts-cast6-avx",
 454         .cra_priority           = 0,
 455         .cra_flags              = CRYPTO_ALG_TYPE_BLKCIPHER |
 456                                   CRYPTO_ALG_INTERNAL,
 457         .cra_blocksize          = CAST6_BLOCK_SIZE,
 458         .cra_ctxsize            = sizeof(struct cast6_xts_ctx),
 459         .cra_alignmask          = 0,
 460         .cra_type               = &crypto_blkcipher_type,
 461         .cra_module             = THIS_MODULE,
 462         .cra_u = {
 463                 .blkcipher = {
 464                         .min_keysize    = CAST6_MIN_KEY_SIZE * 2,
 465                         .max_keysize    = CAST6_MAX_KEY_SIZE * 2,
 466                         .ivsize         = CAST6_BLOCK_SIZE,
 467                         .setkey         = xts_cast6_setkey,
 468                         .encrypt        = xts_encrypt,
 469                         .decrypt        = xts_decrypt,
 470                 },
 471         },
 472 }, {
 473         .cra_name               = "ecb(cast6)",
 474         .cra_driver_name        = "ecb-cast6-avx",
 475         .cra_priority           = 200,
 476         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
 477         .cra_blocksize          = CAST6_BLOCK_SIZE,
 478         .cra_ctxsize            = sizeof(struct async_helper_ctx),
 479         .cra_alignmask          = 0,
 480         .cra_type               = &crypto_ablkcipher_type,
 481         .cra_module             = THIS_MODULE,
 482         .cra_init               = ablk_init,
 483         .cra_exit               = ablk_exit,
 484         .cra_u = {
 485                 .ablkcipher = {
 486                         .min_keysize    = CAST6_MIN_KEY_SIZE,
 487                         .max_keysize    = CAST6_MAX_KEY_SIZE,
 488                         .setkey         = ablk_set_key,
 489                         .encrypt        = ablk_encrypt,
 490                         .decrypt        = ablk_decrypt,
 491                 },
 492         },
 493 }, {
 494         .cra_name               = "cbc(cast6)",
 495         .cra_driver_name        = "cbc-cast6-avx",
 496         .cra_priority           = 200,
 497         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
 498         .cra_blocksize          = CAST6_BLOCK_SIZE,
 499         .cra_ctxsize            = sizeof(struct async_helper_ctx),
 500         .cra_alignmask          = 0,
 501         .cra_type               = &crypto_ablkcipher_type,
 502         .cra_module             = THIS_MODULE,
 503         .cra_init               = ablk_init,
 504         .cra_exit               = ablk_exit,
 505         .cra_u = {
 506                 .ablkcipher = {
 507                         .min_keysize    = CAST6_MIN_KEY_SIZE,
 508                         .max_keysize    = CAST6_MAX_KEY_SIZE,
 509                         .ivsize         = CAST6_BLOCK_SIZE,
 510                         .setkey         = ablk_set_key,
 511                         .encrypt        = __ablk_encrypt,
 512                         .decrypt        = ablk_decrypt,
 513                 },
 514         },
 515 }, {
 516         .cra_name               = "ctr(cast6)",
 517         .cra_driver_name        = "ctr-cast6-avx",
 518         .cra_priority           = 200,
 519         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
 520         .cra_blocksize          = 1,
 521         .cra_ctxsize            = sizeof(struct async_helper_ctx),
 522         .cra_alignmask          = 0,
 523         .cra_type               = &crypto_ablkcipher_type,
 524         .cra_module             = THIS_MODULE,
 525         .cra_init               = ablk_init,
 526         .cra_exit               = ablk_exit,
 527         .cra_u = {
 528                 .ablkcipher = {
 529                         .min_keysize    = CAST6_MIN_KEY_SIZE,
 530                         .max_keysize    = CAST6_MAX_KEY_SIZE,
 531                         .ivsize         = CAST6_BLOCK_SIZE,
 532                         .setkey         = ablk_set_key,
 533                         .encrypt        = ablk_encrypt,
 534                         .decrypt        = ablk_encrypt,
 535                         .geniv          = "chainiv",
 536                 },
 537         },
 538 }, {
 539         .cra_name               = "lrw(cast6)",
 540         .cra_driver_name        = "lrw-cast6-avx",
 541         .cra_priority           = 200,
 542         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
 543         .cra_blocksize          = CAST6_BLOCK_SIZE,
 544         .cra_ctxsize            = sizeof(struct async_helper_ctx),
 545         .cra_alignmask          = 0,
 546         .cra_type               = &crypto_ablkcipher_type,
 547         .cra_module             = THIS_MODULE,
 548         .cra_init               = ablk_init,
 549         .cra_exit               = ablk_exit,
 550         .cra_u = {
 551                 .ablkcipher = {
 552                         .min_keysize    = CAST6_MIN_KEY_SIZE +
 553                                           CAST6_BLOCK_SIZE,
 554                         .max_keysize    = CAST6_MAX_KEY_SIZE +
 555                                           CAST6_BLOCK_SIZE,
 556                         .ivsize         = CAST6_BLOCK_SIZE,
 557                         .setkey         = ablk_set_key,
 558                         .encrypt        = ablk_encrypt,
 559                         .decrypt        = ablk_decrypt,
 560                 },
 561         },
 562 }, {
 563         .cra_name               = "xts(cast6)",
 564         .cra_driver_name        = "xts-cast6-avx",
 565         .cra_priority           = 200,
 566         .cra_flags              = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
 567         .cra_blocksize          = CAST6_BLOCK_SIZE,
 568         .cra_ctxsize            = sizeof(struct async_helper_ctx),
 569         .cra_alignmask          = 0,
 570         .cra_type               = &crypto_ablkcipher_type,
 571         .cra_module             = THIS_MODULE,
 572         .cra_init               = ablk_init,
 573         .cra_exit               = ablk_exit,
 574         .cra_u = {
 575                 .ablkcipher = {
 576                         .min_keysize    = CAST6_MIN_KEY_SIZE * 2,
 577                         .max_keysize    = CAST6_MAX_KEY_SIZE * 2,
 578                         .ivsize         = CAST6_BLOCK_SIZE,
 579                         .setkey         = ablk_set_key,
 580                         .encrypt        = ablk_encrypt,
 581                         .decrypt        = ablk_decrypt,
 582                 },
 583         },
 584 } };
 585
 586 static int __init cast6_init(void)
 587 {
 588         const char *feature_name;
 589
 590         if (!cpu_has_xfeatures(XFEATURE_MASK_SSE | XFEATURE_MASK_YMM,
 591                                 &feature_name)) {
 592                 pr_info("CPU feature '%s' is not supported.\n", feature_name);
 593                 return -ENODEV;
 594         }
 595
 596         return crypto_register_algs(cast6_algs, ARRAY_SIZE(cast6_algs));
 597 }
 598
 599 static void __exit cast6_exit(void)
 600 {
 601         crypto_unregister_algs(cast6_algs, ARRAY_SIZE(cast6_algs));
 602 }
 603
 604 module_init(cast6_init);
 605 module_exit(cast6_exit);
 606
 607 MODULE_DESCRIPTION("Cast6 Cipher Algorithm, AVX optimized");
 608 MODULE_LICENSE("GPL");
 609 MODULE_ALIAS_CRYPTO("cast6");