Signed-off-by: Ard Biesheuvel <ard.biesheu...@linaro.org>
---
 drivers/crypto/ixp4xx_crypto.c | 28 +++++++-------------
 1 file changed, 9 insertions(+), 19 deletions(-)

diff --git a/drivers/crypto/ixp4xx_crypto.c b/drivers/crypto/ixp4xx_crypto.c
index acedafe3fa98..d3d683107050 100644
--- a/drivers/crypto/ixp4xx_crypto.c
+++ b/drivers/crypto/ixp4xx_crypto.c
@@ -17,7 +17,7 @@
 #include <linux/module.h>
 
 #include <crypto/ctr.h>
-#include <crypto/des.h>
+#include <crypto/internal/des.h>
 #include <crypto/aes.h>
 #include <crypto/hmac.h>
 #include <crypto/sha.h>
@@ -756,10 +756,7 @@ static int setup_cipher(struct crypto_tfm *tfm, int 
encrypt,
                }
                cipher_cfg |= keylen_cfg;
        } else {
-               u32 tmp[DES_EXPKEY_WORDS];
-               if (des_ekey(tmp, key) == 0) {
-                       *flags |= CRYPTO_TFM_RES_WEAK_KEY;
-               }
+               crypto_des_verify_key(tfm, key, key_len);
        }
        /* write cfg word to cryptinfo */
        *(u32*)cinfo = cpu_to_be32(cipher_cfg);
@@ -851,14 +848,8 @@ static int ablk_setkey(struct crypto_ablkcipher *tfm, 
const u8 *key,
 static int ablk_des3_setkey(struct crypto_ablkcipher *tfm, const u8 *key,
                            unsigned int key_len)
 {
-       u32 flags = crypto_ablkcipher_get_flags(tfm);
-       int err;
-
-       err = __des3_verify_key(&flags, key);
-       if (unlikely(err))
-               crypto_ablkcipher_set_flags(tfm, flags);
-
-       return ablk_setkey(tfm, key, key_len);
+       return crypto_des3_ede_verify_key(crypto_ablkcipher_tfm(tfm), key) ?:
+              ablk_setkey(tfm, key, key_len);
 }
 
 static int ablk_rfc3686_setkey(struct crypto_ablkcipher *tfm, const u8 *key,
@@ -1181,7 +1172,6 @@ static int des3_aead_setkey(struct crypto_aead *tfm, 
const u8 *key,
                            unsigned int keylen)
 {
        struct ixp_ctx *ctx = crypto_aead_ctx(tfm);
-       u32 flags = CRYPTO_TFM_RES_BAD_KEY_LEN;
        struct crypto_authenc_keys keys;
        int err;
 
@@ -1193,12 +1183,13 @@ static int des3_aead_setkey(struct crypto_aead *tfm, 
const u8 *key,
        if (keys.authkeylen > sizeof(ctx->authkey))
                goto badkey;
 
-       if (keys.enckeylen != DES3_EDE_KEY_SIZE)
+       if (keys.enckeylen != DES3_EDE_KEY_SIZE) {
+               crypto_aead_set_flags(tfm, CRYPTO_TFM_RES_BAD_KEY_LEN);
                goto badkey;
+       }
 
-       flags = crypto_aead_get_flags(tfm);
-       err = __des3_verify_key(&flags, keys.enckey);
-       if (unlikely(err))
+       err = crypto_des3_ede_verify_key(crypto_aead_tfm(tfm), keys.enckey);
+       if (err)
                goto badkey;
 
        memcpy(ctx->authkey, keys.authkey, keys.authkeylen);
@@ -1209,7 +1200,6 @@ static int des3_aead_setkey(struct crypto_aead *tfm, 
const u8 *key,
        memzero_explicit(&keys, sizeof(keys));
        return aead_setup(tfm, crypto_aead_authsize(tfm));
 badkey:
-       crypto_aead_set_flags(tfm, flags);
        memzero_explicit(&keys, sizeof(keys));
        return err;
 }
-- 
2.17.1

Reply via email to