Signed-off-by: Ard Biesheuvel <ard.biesheu...@linaro.org>
---
 drivers/crypto/stm32/stm32-cryp.c | 30 ++++----------------
 1 file changed, 5 insertions(+), 25 deletions(-)

diff --git a/drivers/crypto/stm32/stm32-cryp.c 
b/drivers/crypto/stm32/stm32-cryp.c
index 98ae02826e8f..a704b1d38436 100644
--- a/drivers/crypto/stm32/stm32-cryp.c
+++ b/drivers/crypto/stm32/stm32-cryp.c
@@ -15,7 +15,7 @@
 #include <linux/reset.h>
 
 #include <crypto/aes.h>
-#include <crypto/des.h>
+#include <crypto/internal/des.h>
 #include <crypto/engine.h>
 #include <crypto/scatterwalk.h>
 #include <crypto/internal/aead.h>
@@ -767,35 +767,15 @@ static int stm32_cryp_aes_setkey(struct crypto_ablkcipher 
*tfm, const u8 *key,
 static int stm32_cryp_des_setkey(struct crypto_ablkcipher *tfm, const u8 *key,
                                 unsigned int keylen)
 {
-       u32 tmp[DES_EXPKEY_WORDS];
-
-       if (keylen != DES_KEY_SIZE)
-               return -EINVAL;
-
-       if ((crypto_ablkcipher_get_flags(tfm) &
-            CRYPTO_TFM_REQ_FORBID_WEAK_KEYS) &&
-           unlikely(!des_ekey(tmp, key))) {
-               crypto_ablkcipher_set_flags(tfm, CRYPTO_TFM_RES_WEAK_KEY);
-               return -EINVAL;
-       }
-
-       return stm32_cryp_setkey(tfm, key, keylen);
+       return crypto_des_verify_key(crypto_ablkcipher_tfm(tfm), key) ?:
+              stm32_cryp_setkey(tfm, key, keylen);
 }
 
 static int stm32_cryp_tdes_setkey(struct crypto_ablkcipher *tfm, const u8 *key,
                                  unsigned int keylen)
 {
-       u32 flags;
-       int err;
-
-       flags = crypto_ablkcipher_get_flags(tfm);
-       err = __des3_verify_key(&flags, key);
-       if (unlikely(err)) {
-               crypto_ablkcipher_set_flags(tfm, flags);
-               return err;
-       }
-
-       return stm32_cryp_setkey(tfm, key, keylen);
+       return crypto_des3_ede_verify_key(crypto_ablkcipher_tfm(tfm), key) ?:
+              stm32_cryp_setkey(tfm, key, keylen);
 }
 
 static int stm32_cryp_aes_aead_setkey(struct crypto_aead *tfm, const u8 *key,
-- 
2.17.1

Reply via email to