crypto/aead.c

   1 /*
   2  * AEAD: Authenticated Encryption with Associated Data
   3  *
   4  * This file provides API support for AEAD algorithms.
   5  *
   6  * Copyright (c) 2007 Herbert Xu <herbert@gondor.apana.org.au>
   7  *
   8  * This program is free software; you can redistribute it and/or modify it
   9  * under the terms of the GNU General Public License as published by the Free
  10  * Software Foundation; either version 2 of the License, or (at your option)
  11  * any later version.
  12  *
  13  */
  14
  15 #include <crypto/internal/aead.h>
  16 #include <crypto/scatterwalk.h>
  17 #include <linux/err.h>
  18 #include <linux/init.h>
  19 #include <linux/kernel.h>
  20 #include <linux/module.h>
  21 #include <linux/rtnetlink.h>
  22 #include <linux/sched.h>
  23 #include <linux/slab.h>
  24 #include <linux/seq_file.h>
  25 #include <linux/cryptouser.h>
  26 #include <net/netlink.h>
  27
  28 #include "internal.h"
  29
  30 static int aead_null_givencrypt(struct aead_givcrypt_request *req);
  31 static int aead_null_givdecrypt(struct aead_givcrypt_request *req);
  32
  33 static int setkey_unaligned(struct crypto_aead *tfm, const u8 *key,
  34                             unsigned int keylen)
  35 {
  36         unsigned long alignmask = crypto_aead_alignmask(tfm);
  37         int ret;
  38         u8 *buffer, *alignbuffer;
  39         unsigned long absize;
  40
  41         absize = keylen + alignmask;
  42         buffer = kmalloc(absize, GFP_ATOMIC);
  43         if (!buffer)
  44                 return -ENOMEM;
  45
  46         alignbuffer = (u8 *)ALIGN((unsigned long)buffer, alignmask + 1);
  47         memcpy(alignbuffer, key, keylen);
  48         ret = tfm->setkey(tfm, alignbuffer, keylen);
  49         memset(alignbuffer, 0, keylen);
  50         kfree(buffer);
  51         return ret;
  52 }
  53
  54 int crypto_aead_setkey(struct crypto_aead *tfm,
  55                        const u8 *key, unsigned int keylen)
  56 {
  57         unsigned long alignmask = crypto_aead_alignmask(tfm);
  58
  59         tfm = tfm->child;
  60
  61         if ((unsigned long)key & alignmask)
  62                 return setkey_unaligned(tfm, key, keylen);
  63
  64         return tfm->setkey(tfm, key, keylen);
  65 }
  66 EXPORT_SYMBOL_GPL(crypto_aead_setkey);
  67
  68 int crypto_aead_setauthsize(struct crypto_aead *tfm, unsigned int authsize)
  69 {
  70         int err;
  71
  72         if (authsize > tfm->maxauthsize)
  73                 return -EINVAL;
  74
  75         if (tfm->setauthsize) {
  76                 err = tfm->setauthsize(tfm->child, authsize);
  77                 if (err)
  78                         return err;
  79         }
  80
  81         tfm->child->authsize = authsize;
  82         tfm->authsize = authsize;
  83         return 0;
  84 }
  85 EXPORT_SYMBOL_GPL(crypto_aead_setauthsize);
  86
  87 struct aead_old_request {
  88         struct scatterlist srcbuf[2];
  89         struct scatterlist dstbuf[2];
  90         struct aead_request subreq;
  91 };
  92
  93 unsigned int crypto_aead_reqsize(struct crypto_aead *tfm)
  94 {
  95         return tfm->reqsize + sizeof(struct aead_old_request);
  96 }
  97 EXPORT_SYMBOL_GPL(crypto_aead_reqsize);
  98
  99 static int old_crypt(struct aead_request *req,
 100                      int (*crypt)(struct aead_request *req))
 101 {
 102         struct aead_old_request *nreq = aead_request_ctx(req);
 103         struct crypto_aead *aead = crypto_aead_reqtfm(req);
 104         struct scatterlist *src, *dst;
 105
 106         if (req->old)
 107                 return crypt(req);
 108
 109         src = scatterwalk_ffwd(nreq->srcbuf, req->src,
 110                                req->assoclen + req->cryptoff);
 111         dst = scatterwalk_ffwd(nreq->dstbuf, req->dst,
 112                                req->assoclen + req->cryptoff);
 113
 114         aead_request_set_tfm(&nreq->subreq, aead);
 115         aead_request_set_callback(&nreq->subreq, aead_request_flags(req),
 116                                   req->base.complete, req->base.data);
 117         aead_request_set_crypt(&nreq->subreq, src, dst, req->cryptlen,
 118                                req->iv);
 119         aead_request_set_assoc(&nreq->subreq, req->src, req->assoclen);
 120
 121         return crypt(&nreq->subreq);
 122 }
 123
 124 static int old_encrypt(struct aead_request *req)
 125 {
 126         struct crypto_aead *aead = crypto_aead_reqtfm(req);
 127         struct old_aead_alg *alg = crypto_old_aead_alg(aead);
 128
 129         return old_crypt(req, alg->encrypt);
 130 }
 131
 132 static int old_decrypt(struct aead_request *req)
 133 {
 134         struct crypto_aead *aead = crypto_aead_reqtfm(req);
 135         struct old_aead_alg *alg = crypto_old_aead_alg(aead);
 136
 137         return old_crypt(req, alg->decrypt);
 138 }
 139
 140 static int no_givcrypt(struct aead_givcrypt_request *req)
 141 {
 142         return -ENOSYS;
 143 }
 144
 145 static int crypto_old_aead_init_tfm(struct crypto_tfm *tfm)
 146 {
 147         struct old_aead_alg *alg = &tfm->__crt_alg->cra_aead;
 148         struct crypto_aead *crt = __crypto_aead_cast(tfm);
 149
 150         if (max(alg->maxauthsize, alg->ivsize) > PAGE_SIZE / 8)
 151                 return -EINVAL;
 152
 153         crt->setkey = alg->setkey;
 154         crt->setauthsize = alg->setauthsize;
 155         crt->encrypt = old_encrypt;
 156         crt->decrypt = old_decrypt;
 157         if (alg->ivsize) {
 158                 crt->givencrypt = alg->givencrypt ?: no_givcrypt;
 159                 crt->givdecrypt = alg->givdecrypt ?: no_givcrypt;
 160         } else {
 161                 crt->givencrypt = aead_null_givencrypt;
 162                 crt->givdecrypt = aead_null_givdecrypt;
 163         }
 164         crt->child = __crypto_aead_cast(tfm);
 165         crt->ivsize = alg->ivsize;
 166         crt->maxauthsize = alg->maxauthsize;
 167         crt->authsize = alg->maxauthsize;
 168
 169         return 0;
 170 }
 171
 172 static int crypto_aead_init_tfm(struct crypto_tfm *tfm)
 173 {
 174         struct crypto_aead *aead = __crypto_aead_cast(tfm);
 175         struct aead_alg *alg = crypto_aead_alg(aead);
 176
 177         if (crypto_old_aead_alg(aead)->encrypt)
 178                 return crypto_old_aead_init_tfm(tfm);
 179
 180         aead->setkey = alg->setkey;
 181         aead->setauthsize = alg->setauthsize;
 182         aead->encrypt = alg->encrypt;
 183         aead->decrypt = alg->decrypt;
 184         aead->child = __crypto_aead_cast(tfm);
 185         aead->ivsize = alg->ivsize;
 186         aead->maxauthsize = alg->maxauthsize;
 187         aead->authsize = alg->maxauthsize;
 188
 189         return 0;
 190 }
 191
 192 #ifdef CONFIG_NET
 193 static int crypto_old_aead_report(struct sk_buff *skb, struct crypto_alg *alg)
 194 {
 195         struct crypto_report_aead raead;
 196         struct old_aead_alg *aead = &alg->cra_aead;
 197
 198         strncpy(raead.type, "aead", sizeof(raead.type));
 199         strncpy(raead.geniv, aead->geniv ?: "<built-in>", sizeof(raead.geniv));
 200
 201         raead.blocksize = alg->cra_blocksize;
 202         raead.maxauthsize = aead->maxauthsize;
 203         raead.ivsize = aead->ivsize;
 204
 205         if (nla_put(skb, CRYPTOCFGA_REPORT_AEAD,
 206                     sizeof(struct crypto_report_aead), &raead))
 207                 goto nla_put_failure;
 208         return 0;
 209
 210 nla_put_failure:
 211         return -EMSGSIZE;
 212 }
 213 #else
 214 static int crypto_old_aead_report(struct sk_buff *skb, struct crypto_alg *alg)
 215 {
 216         return -ENOSYS;
 217 }
 218 #endif
 219
 220 static void crypto_old_aead_show(struct seq_file *m, struct crypto_alg *alg)
 221         __attribute__ ((unused));
 222 static void crypto_old_aead_show(struct seq_file *m, struct crypto_alg *alg)
 223 {
 224         struct old_aead_alg *aead = &alg->cra_aead;
 225
 226         seq_printf(m, "type         : aead\n");
 227         seq_printf(m, "async        : %s\n", alg->cra_flags & CRYPTO_ALG_ASYNC ?
 228                                              "yes" : "no");
 229         seq_printf(m, "blocksize    : %u\n", alg->cra_blocksize);
 230         seq_printf(m, "ivsize       : %u\n", aead->ivsize);
 231         seq_printf(m, "maxauthsize  : %u\n", aead->maxauthsize);
 232         seq_printf(m, "geniv        : %s\n", aead->geniv ?: "<built-in>");
 233 }
 234
 235 const struct crypto_type crypto_aead_type = {
 236         .extsize = crypto_alg_extsize,
 237         .init_tfm = crypto_aead_init_tfm,
 238 #ifdef CONFIG_PROC_FS
 239         .show = crypto_old_aead_show,
 240 #endif
 241         .report = crypto_old_aead_report,
 242         .lookup = crypto_lookup_aead,
 243         .maskclear = ~(CRYPTO_ALG_TYPE_MASK | CRYPTO_ALG_GENIV),
 244         .maskset = CRYPTO_ALG_TYPE_MASK,
 245         .type = CRYPTO_ALG_TYPE_AEAD,
 246         .tfmsize = offsetof(struct crypto_aead, base),
 247 };
 248 EXPORT_SYMBOL_GPL(crypto_aead_type);
 249
 250 #ifdef CONFIG_NET
 251 static int crypto_aead_report(struct sk_buff *skb, struct crypto_alg *alg)
 252 {
 253         struct crypto_report_aead raead;
 254         struct aead_alg *aead = container_of(alg, struct aead_alg, base);
 255
 256         strncpy(raead.type, "aead", sizeof(raead.type));
 257         strncpy(raead.geniv, "<none>", sizeof(raead.geniv));
 258
 259         raead.blocksize = alg->cra_blocksize;
 260         raead.maxauthsize = aead->maxauthsize;
 261         raead.ivsize = aead->ivsize;
 262
 263         if (nla_put(skb, CRYPTOCFGA_REPORT_AEAD,
 264                     sizeof(struct crypto_report_aead), &raead))
 265                 goto nla_put_failure;
 266         return 0;
 267
 268 nla_put_failure:
 269         return -EMSGSIZE;
 270 }
 271 #else
 272 static int crypto_aead_report(struct sk_buff *skb, struct crypto_alg *alg)
 273 {
 274         return -ENOSYS;
 275 }
 276 #endif
 277
 278 static void crypto_aead_show(struct seq_file *m, struct crypto_alg *alg)
 279         __attribute__ ((unused));
 280 static void crypto_aead_show(struct seq_file *m, struct crypto_alg *alg)
 281 {
 282         struct aead_alg *aead = container_of(alg, struct aead_alg, base);
 283
 284         seq_printf(m, "type         : aead\n");
 285         seq_printf(m, "async        : %s\n", alg->cra_flags & CRYPTO_ALG_ASYNC ?
 286                                              "yes" : "no");
 287         seq_printf(m, "blocksize    : %u\n", alg->cra_blocksize);
 288         seq_printf(m, "ivsize       : %u\n", aead->ivsize);
 289         seq_printf(m, "maxauthsize  : %u\n", aead->maxauthsize);
 290         seq_printf(m, "geniv        : <none>\n");
 291 }
 292
 293 static const struct crypto_type crypto_new_aead_type = {
 294         .extsize = crypto_alg_extsize,
 295         .init_tfm = crypto_aead_init_tfm,
 296 #ifdef CONFIG_PROC_FS
 297         .show = crypto_aead_show,
 298 #endif
 299         .report = crypto_aead_report,
 300         .maskclear = ~CRYPTO_ALG_TYPE_MASK,
 301         .maskset = CRYPTO_ALG_TYPE_MASK,
 302         .type = CRYPTO_ALG_TYPE_AEAD,
 303         .tfmsize = offsetof(struct crypto_aead, base),
 304 };
 305
 306 static int aead_null_givencrypt(struct aead_givcrypt_request *req)
 307 {
 308         return crypto_aead_encrypt(&req->areq);
 309 }
 310
 311 static int aead_null_givdecrypt(struct aead_givcrypt_request *req)
 312 {
 313         return crypto_aead_decrypt(&req->areq);
 314 }
 315
 316 #ifdef CONFIG_NET
 317 static int crypto_nivaead_report(struct sk_buff *skb, struct crypto_alg *alg)
 318 {
 319         struct crypto_report_aead raead;
 320         struct old_aead_alg *aead = &alg->cra_aead;
 321
 322         strncpy(raead.type, "nivaead", sizeof(raead.type));
 323         strncpy(raead.geniv, aead->geniv, sizeof(raead.geniv));
 324
 325         raead.blocksize = alg->cra_blocksize;
 326         raead.maxauthsize = aead->maxauthsize;
 327         raead.ivsize = aead->ivsize;
 328
 329         if (nla_put(skb, CRYPTOCFGA_REPORT_AEAD,
 330                     sizeof(struct crypto_report_aead), &raead))
 331                 goto nla_put_failure;
 332         return 0;
 333
 334 nla_put_failure:
 335         return -EMSGSIZE;
 336 }
 337 #else
 338 static int crypto_nivaead_report(struct sk_buff *skb, struct crypto_alg *alg)
 339 {
 340         return -ENOSYS;
 341 }
 342 #endif
 343
 344
 345 static void crypto_nivaead_show(struct seq_file *m, struct crypto_alg *alg)
 346         __attribute__ ((unused));
 347 static void crypto_nivaead_show(struct seq_file *m, struct crypto_alg *alg)
 348 {
 349         struct old_aead_alg *aead = &alg->cra_aead;
 350
 351         seq_printf(m, "type         : nivaead\n");
 352         seq_printf(m, "async        : %s\n", alg->cra_flags & CRYPTO_ALG_ASYNC ?
 353                                              "yes" : "no");
 354         seq_printf(m, "blocksize    : %u\n", alg->cra_blocksize);
 355         seq_printf(m, "ivsize       : %u\n", aead->ivsize);
 356         seq_printf(m, "maxauthsize  : %u\n", aead->maxauthsize);
 357         seq_printf(m, "geniv        : %s\n", aead->geniv);
 358 }
 359
 360 const struct crypto_type crypto_nivaead_type = {
 361         .extsize = crypto_alg_extsize,
 362         .init_tfm = crypto_aead_init_tfm,
 363 #ifdef CONFIG_PROC_FS
 364         .show = crypto_nivaead_show,
 365 #endif
 366         .report = crypto_nivaead_report,
 367         .maskclear = ~(CRYPTO_ALG_TYPE_MASK | CRYPTO_ALG_GENIV),
 368         .maskset = CRYPTO_ALG_TYPE_MASK | CRYPTO_ALG_GENIV,
 369         .type = CRYPTO_ALG_TYPE_AEAD,
 370         .tfmsize = offsetof(struct crypto_aead, base),
 371 };
 372 EXPORT_SYMBOL_GPL(crypto_nivaead_type);
 373
 374 static int crypto_grab_nivaead(struct crypto_aead_spawn *spawn,
 375                                const char *name, u32 type, u32 mask)
 376 {
 377         spawn->base.frontend = &crypto_nivaead_type;
 378         return crypto_grab_spawn(&spawn->base, name, type, mask);
 379 }
 380
 381 struct aead_instance *aead_geniv_alloc(struct crypto_template *tmpl,
 382                                        struct rtattr **tb, u32 type, u32 mask)
 383 {
 384         const char *name;
 385         struct crypto_aead_spawn *spawn;
 386         struct crypto_attr_type *algt;
 387         struct aead_instance *inst;
 388         struct aead_alg *alg;
 389         unsigned int ivsize;
 390         unsigned int maxauthsize;
 391         int err;
 392
 393         algt = crypto_get_attr_type(tb);
 394         if (IS_ERR(algt))
 395                 return ERR_CAST(algt);
 396
 397         if ((algt->type ^ (CRYPTO_ALG_TYPE_AEAD | CRYPTO_ALG_GENIV)) &
 398             algt->mask)
 399                 return ERR_PTR(-EINVAL);
 400
 401         name = crypto_attr_alg_name(tb[1]);
 402         if (IS_ERR(name))
 403                 return ERR_CAST(name);
 404
 405         inst = kzalloc(sizeof(*inst) + sizeof(*spawn), GFP_KERNEL);
 406         if (!inst)
 407                 return ERR_PTR(-ENOMEM);
 408
 409         spawn = aead_instance_ctx(inst);
 410
 411         /* Ignore async algorithms if necessary. */
 412         mask |= crypto_requires_sync(algt->type, algt->mask);
 413
 414         crypto_set_aead_spawn(spawn, aead_crypto_instance(inst));
 415         err = crypto_grab_nivaead(spawn, name, type, mask);
 416         if (err)
 417                 goto err_free_inst;
 418
 419         alg = crypto_spawn_aead_alg(spawn);
 420
 421         if (alg->base.cra_aead.encrypt) {
 422                 ivsize = alg->base.cra_aead.ivsize;
 423                 maxauthsize = alg->base.cra_aead.maxauthsize;
 424         } else {
 425                 ivsize = alg->ivsize;
 426                 maxauthsize = alg->maxauthsize;
 427         }
 428
 429         err = -EINVAL;
 430         if (!ivsize)
 431                 goto err_drop_alg;
 432
 433         /*
 434          * This is only true if we're constructing an algorithm with its
 435          * default IV generator.  For the default generator we elide the
 436          * template name and double-check the IV generator.
 437          */
 438         if (algt->mask & CRYPTO_ALG_GENIV) {
 439                 if (!alg->base.cra_aead.encrypt)
 440                         goto err_drop_alg;
 441                 if (strcmp(tmpl->name, alg->base.cra_aead.geniv))
 442                         goto err_drop_alg;
 443
 444                 memcpy(inst->alg.base.cra_name, alg->base.cra_name,
 445                        CRYPTO_MAX_ALG_NAME);
 446                 memcpy(inst->alg.base.cra_driver_name,
 447                        alg->base.cra_driver_name, CRYPTO_MAX_ALG_NAME);
 448
 449                 inst->alg.base.cra_flags = CRYPTO_ALG_TYPE_AEAD |
 450                                            CRYPTO_ALG_GENIV;
 451                 inst->alg.base.cra_flags |= alg->base.cra_flags &
 452                                             CRYPTO_ALG_ASYNC;
 453                 inst->alg.base.cra_priority = alg->base.cra_priority;
 454                 inst->alg.base.cra_blocksize = alg->base.cra_blocksize;
 455                 inst->alg.base.cra_alignmask = alg->base.cra_alignmask;
 456                 inst->alg.base.cra_type = &crypto_aead_type;
 457
 458                 inst->alg.base.cra_aead.ivsize = ivsize;
 459                 inst->alg.base.cra_aead.maxauthsize = maxauthsize;
 460
 461                 inst->alg.base.cra_aead.setkey = alg->base.cra_aead.setkey;
 462                 inst->alg.base.cra_aead.setauthsize =
 463                         alg->base.cra_aead.setauthsize;
 464                 inst->alg.base.cra_aead.encrypt = alg->base.cra_aead.encrypt;
 465                 inst->alg.base.cra_aead.decrypt = alg->base.cra_aead.decrypt;
 466
 467                 goto out;
 468         }
 469
 470         err = -ENAMETOOLONG;
 471         if (snprintf(inst->alg.base.cra_name, CRYPTO_MAX_ALG_NAME,
 472                      "%s(%s)", tmpl->name, alg->base.cra_name) >=
 473             CRYPTO_MAX_ALG_NAME)
 474                 goto err_drop_alg;
 475         if (snprintf(inst->alg.base.cra_driver_name, CRYPTO_MAX_ALG_NAME,
 476                      "%s(%s)", tmpl->name, alg->base.cra_driver_name) >=
 477             CRYPTO_MAX_ALG_NAME)
 478                 goto err_drop_alg;
 479
 480         inst->alg.base.cra_flags = CRYPTO_ALG_TYPE_AEAD;
 481         inst->alg.base.cra_flags |= alg->base.cra_flags & CRYPTO_ALG_ASYNC;
 482         inst->alg.base.cra_priority = alg->base.cra_priority;
 483         inst->alg.base.cra_blocksize = alg->base.cra_blocksize;
 484         inst->alg.base.cra_alignmask = alg->base.cra_alignmask;
 485         inst->alg.base.cra_type = &crypto_new_aead_type;
 486
 487         inst->alg.ivsize = ivsize;
 488         inst->alg.maxauthsize = maxauthsize;
 489
 490 out:
 491         return inst;
 492
 493 err_drop_alg:
 494         crypto_drop_aead(spawn);
 495 err_free_inst:
 496         kfree(inst);
 497         inst = ERR_PTR(err);
 498         goto out;
 499 }
 500 EXPORT_SYMBOL_GPL(aead_geniv_alloc);
 501
 502 void aead_geniv_free(struct aead_instance *inst)
 503 {
 504         crypto_drop_aead(aead_instance_ctx(inst));
 505         kfree(inst);
 506 }
 507 EXPORT_SYMBOL_GPL(aead_geniv_free);
 508
 509 int aead_geniv_init(struct crypto_tfm *tfm)
 510 {
 511         struct crypto_instance *inst = (void *)tfm->__crt_alg;
 512         struct crypto_aead *child;
 513         struct crypto_aead *aead;
 514
 515         aead = __crypto_aead_cast(tfm);
 516
 517         child = crypto_spawn_aead(crypto_instance_ctx(inst));
 518         if (IS_ERR(child))
 519                 return PTR_ERR(child);
 520
 521         aead->child = child;
 522         aead->reqsize += crypto_aead_reqsize(child);
 523
 524         return 0;
 525 }
 526 EXPORT_SYMBOL_GPL(aead_geniv_init);
 527
 528 void aead_geniv_exit(struct crypto_tfm *tfm)
 529 {
 530         crypto_free_aead(__crypto_aead_cast(tfm)->child);
 531 }
 532 EXPORT_SYMBOL_GPL(aead_geniv_exit);
 533
 534 static int crypto_nivaead_default(struct crypto_alg *alg, u32 type, u32 mask)
 535 {
 536         struct rtattr *tb[3];
 537         struct {
 538                 struct rtattr attr;
 539                 struct crypto_attr_type data;
 540         } ptype;
 541         struct {
 542                 struct rtattr attr;
 543                 struct crypto_attr_alg data;
 544         } palg;
 545         struct crypto_template *tmpl;
 546         struct crypto_instance *inst;
 547         struct crypto_alg *larval;
 548         const char *geniv;
 549         int err;
 550
 551         larval = crypto_larval_lookup(alg->cra_driver_name,
 552                                       CRYPTO_ALG_TYPE_AEAD | CRYPTO_ALG_GENIV,
 553                                       CRYPTO_ALG_TYPE_MASK | CRYPTO_ALG_GENIV);
 554         err = PTR_ERR(larval);
 555         if (IS_ERR(larval))
 556                 goto out;
 557
 558         err = -EAGAIN;
 559         if (!crypto_is_larval(larval))
 560                 goto drop_larval;
 561
 562         ptype.attr.rta_len = sizeof(ptype);
 563         ptype.attr.rta_type = CRYPTOA_TYPE;
 564         ptype.data.type = type | CRYPTO_ALG_GENIV;
 565         /* GENIV tells the template that we're making a default geniv. */
 566         ptype.data.mask = mask | CRYPTO_ALG_GENIV;
 567         tb[0] = &ptype.attr;
 568
 569         palg.attr.rta_len = sizeof(palg);
 570         palg.attr.rta_type = CRYPTOA_ALG;
 571         /* Must use the exact name to locate ourselves. */
 572         memcpy(palg.data.name, alg->cra_driver_name, CRYPTO_MAX_ALG_NAME);
 573         tb[1] = &palg.attr;
 574
 575         tb[2] = NULL;
 576
 577         geniv = alg->cra_aead.geniv;
 578
 579         tmpl = crypto_lookup_template(geniv);
 580         err = -ENOENT;
 581         if (!tmpl)
 582                 goto kill_larval;
 583
 584         inst = tmpl->alloc(tb);
 585         err = PTR_ERR(inst);
 586         if (IS_ERR(inst))
 587                 goto put_tmpl;
 588
 589         err = crypto_register_instance(tmpl, inst);
 590         if (err) {
 591                 tmpl->free(inst);
 592                 goto put_tmpl;
 593         }
 594
 595         /* Redo the lookup to use the instance we just registered. */
 596         err = -EAGAIN;
 597
 598 put_tmpl:
 599         crypto_tmpl_put(tmpl);
 600 kill_larval:
 601         crypto_larval_kill(larval);
 602 drop_larval:
 603         crypto_mod_put(larval);
 604 out:
 605         crypto_mod_put(alg);
 606         return err;
 607 }
 608
 609 struct crypto_alg *crypto_lookup_aead(const char *name, u32 type, u32 mask)
 610 {
 611         struct crypto_alg *alg;
 612
 613         alg = crypto_alg_mod_lookup(name, type, mask);
 614         if (IS_ERR(alg))
 615                 return alg;
 616
 617         if (alg->cra_type == &crypto_aead_type)
 618                 return alg;
 619
 620         if (!alg->cra_aead.ivsize)
 621                 return alg;
 622
 623         crypto_mod_put(alg);
 624         alg = crypto_alg_mod_lookup(name, type | CRYPTO_ALG_TESTED,
 625                                     mask & ~CRYPTO_ALG_TESTED);
 626         if (IS_ERR(alg))
 627                 return alg;
 628
 629         if (alg->cra_type == &crypto_aead_type) {
 630                 if (~alg->cra_flags & (type ^ ~mask) & CRYPTO_ALG_TESTED) {
 631                         crypto_mod_put(alg);
 632                         alg = ERR_PTR(-ENOENT);
 633                 }
 634                 return alg;
 635         }
 636
 637         BUG_ON(!alg->cra_aead.ivsize);
 638
 639         return ERR_PTR(crypto_nivaead_default(alg, type, mask));
 640 }
 641 EXPORT_SYMBOL_GPL(crypto_lookup_aead);
 642
 643 int crypto_grab_aead(struct crypto_aead_spawn *spawn, const char *name,
 644                      u32 type, u32 mask)
 645 {
 646         spawn->base.frontend = &crypto_aead_type;
 647         return crypto_grab_spawn(&spawn->base, name, type, mask);
 648 }
 649 EXPORT_SYMBOL_GPL(crypto_grab_aead);
 650
 651 struct crypto_aead *crypto_alloc_aead(const char *alg_name, u32 type, u32 mask)
 652 {
 653         return crypto_alloc_tfm(alg_name, &crypto_aead_type, type, mask);
 654 }
 655 EXPORT_SYMBOL_GPL(crypto_alloc_aead);
 656
 657 static int aead_prepare_alg(struct aead_alg *alg)
 658 {
 659         struct crypto_alg *base = &alg->base;
 660
 661         if (max(alg->maxauthsize, alg->ivsize) > PAGE_SIZE / 8)
 662                 return -EINVAL;
 663
 664         base->cra_type = &crypto_new_aead_type;
 665         base->cra_flags &= ~CRYPTO_ALG_TYPE_MASK;
 666         base->cra_flags |= CRYPTO_ALG_TYPE_AEAD;
 667
 668         return 0;
 669 }
 670
 671 int crypto_register_aead(struct aead_alg *alg)
 672 {
 673         struct crypto_alg *base = &alg->base;
 674         int err;
 675
 676         err = aead_prepare_alg(alg);
 677         if (err)
 678                 return err;
 679
 680         return crypto_register_alg(base);
 681 }
 682 EXPORT_SYMBOL_GPL(crypto_register_aead);
 683
 684 int crypto_unregister_aead(struct aead_alg *alg)
 685 {
 686         return crypto_unregister_alg(&alg->base);
 687 }
 688 EXPORT_SYMBOL_GPL(crypto_unregister_aead);
 689
 690 int aead_register_instance(struct crypto_template *tmpl,
 691                            struct aead_instance *inst)
 692 {
 693         int err;
 694
 695         err = aead_prepare_alg(&inst->alg);
 696         if (err)
 697                 return err;
 698
 699         return crypto_register_instance(tmpl, aead_crypto_instance(inst));
 700 }
 701 EXPORT_SYMBOL_GPL(aead_register_instance);
 702
 703 MODULE_LICENSE("GPL");
 704 MODULE_DESCRIPTION("Authenticated Encryption with Associated Data (AEAD)");