#include <linux/fips.h>
 #include <linux/mutex.h>
 #include <crypto/algapi.h>
-#include <crypto/des.h>
+#include <crypto/internal/des.h>
 #include <asm/cpacf.h>
 
 #define DES3_KEY_SIZE  (3 * DES_KEY_SIZE)
                      unsigned int key_len)
 {
        struct s390_des_ctx *ctx = crypto_tfm_ctx(tfm);
-       u32 tmp[DES_EXPKEY_WORDS];
+       int err;
 
-       /* check for weak keys */
-       if (!des_ekey(tmp, key) &&
-           (tfm->crt_flags & CRYPTO_TFM_REQ_FORBID_WEAK_KEYS)) {
-               tfm->crt_flags |= CRYPTO_TFM_RES_WEAK_KEY;
-               return -EINVAL;
-       }
+       err = crypto_des_verify_key(tfm, key);
+       if (err)
+               return err;
 
        memcpy(ctx->key, key, key_len);
        return 0;
 }
 
-static void des_encrypt(struct crypto_tfm *tfm, u8 *out, const u8 *in)
+static void s390_des_encrypt(struct crypto_tfm *tfm, u8 *out, const u8 *in)
 {
        struct s390_des_ctx *ctx = crypto_tfm_ctx(tfm);
 
        cpacf_km(CPACF_KM_DEA, ctx->key, out, in, DES_BLOCK_SIZE);
 }
 
-static void des_decrypt(struct crypto_tfm *tfm, u8 *out, const u8 *in)
+static void s390_des_decrypt(struct crypto_tfm *tfm, u8 *out, const u8 *in)
 {
        struct s390_des_ctx *ctx = crypto_tfm_ctx(tfm);
 
                        .cia_min_keysize        =       DES_KEY_SIZE,
                        .cia_max_keysize        =       DES_KEY_SIZE,
                        .cia_setkey             =       des_setkey,
-                       .cia_encrypt            =       des_encrypt,
-                       .cia_decrypt            =       des_decrypt,
+                       .cia_encrypt            =       s390_des_encrypt,
+                       .cia_decrypt            =       s390_des_decrypt,
                }
        }
 };
        struct s390_des_ctx *ctx = crypto_tfm_ctx(tfm);
        int err;
 
-       err = __des3_verify_key(&tfm->crt_flags, key);
-       if (unlikely(err))
+       err = crypto_des3_ede_verify_key(tfm, key);
+       if (err)
                return err;
 
        memcpy(ctx->key, key, key_len);