[CRYPTO] api: Add type-safe spawns

This patch allows spawns of specific types (e.g., cipher) to be allocated.

Signed-off-by: Herbert Xu <herbert@gondor.apana.org.au>

1 files changed, 11 insertions, 2 deletions

diff --git a/crypto/algapi.c b/crypto/algapi.c
index 69eb504721a..0f1abca1b98 100644
--- a/crypto/algapi.c
+++ b/crypto/algapi.c
@@ -377,7 +377,8 @@ void crypto_drop_spawn(struct crypto_spawn *spawn)
 }
 EXPORT_SYMBOL_GPL(crypto_drop_spawn);
 
-struct crypto_tfm *crypto_spawn_tfm(struct crypto_spawn *spawn)
+struct crypto_tfm *crypto_spawn_tfm(struct crypto_spawn *spawn, u32 type,
+                                    u32 mask)
 {
         struct crypto_alg *alg;
         struct crypto_alg *alg2;
@@ -396,10 +397,18 @@ struct crypto_tfm *crypto_spawn_tfm(struct crypto_spawn *spawn)
                 return ERR_PTR(-EAGAIN);
         }
 
+        tfm = ERR_PTR(-EINVAL);
+        if (unlikely((alg->cra_flags ^ type) & mask))
+                goto out_put_alg;
+
         tfm = __crypto_alloc_tfm(alg);
         if (IS_ERR(tfm))
-                crypto_mod_put(alg);
+                goto out_put_alg;
+
+        return tfm;
 
+out_put_alg:
+        crypto_mod_put(alg);
         return tfm;
 }
 EXPORT_SYMBOL_GPL(crypto_spawn_tfm);