return crypto_alg_extsize(alg);
 }
 
+static void skcipher_set_needkey(struct crypto_skcipher *tfm)
+{
+       if (tfm->keysize)
+               crypto_skcipher_set_flags(tfm, CRYPTO_TFM_NEED_KEY);
+}
+
 static int skcipher_setkey_blkcipher(struct crypto_skcipher *tfm,
                                     const u8 *key, unsigned int keylen)
 {
        err = crypto_blkcipher_setkey(blkcipher, key, keylen);
        crypto_skcipher_set_flags(tfm, crypto_blkcipher_get_flags(blkcipher) &
                                       CRYPTO_TFM_RES_MASK);
-       if (err)
+       if (unlikely(err)) {
+               skcipher_set_needkey(tfm);
                return err;
+       }
 
        crypto_skcipher_clear_flags(tfm, CRYPTO_TFM_NEED_KEY);
        return 0;
        skcipher->ivsize = crypto_blkcipher_ivsize(blkcipher);
        skcipher->keysize = calg->cra_blkcipher.max_keysize;
 
-       if (skcipher->keysize)
-               crypto_skcipher_set_flags(skcipher, CRYPTO_TFM_NEED_KEY);
+       skcipher_set_needkey(skcipher);
 
        return 0;
 }
        crypto_skcipher_set_flags(tfm,
                                  crypto_ablkcipher_get_flags(ablkcipher) &
                                  CRYPTO_TFM_RES_MASK);
-       if (err)
+       if (unlikely(err)) {
+               skcipher_set_needkey(tfm);
                return err;
+       }
 
        crypto_skcipher_clear_flags(tfm, CRYPTO_TFM_NEED_KEY);
        return 0;
                            sizeof(struct ablkcipher_request);
        skcipher->keysize = calg->cra_ablkcipher.max_keysize;
 
-       if (skcipher->keysize)
-               crypto_skcipher_set_flags(skcipher, CRYPTO_TFM_NEED_KEY);
+       skcipher_set_needkey(skcipher);
 
        return 0;
 }
        else
                err = cipher->setkey(tfm, key, keylen);
 
-       if (err)
+       if (unlikely(err)) {
+               skcipher_set_needkey(tfm);
                return err;
+       }
 
        crypto_skcipher_clear_flags(tfm, CRYPTO_TFM_NEED_KEY);
        return 0;
        skcipher->ivsize = alg->ivsize;
        skcipher->keysize = alg->max_keysize;
 
-       if (skcipher->keysize)
-               crypto_skcipher_set_flags(skcipher, CRYPTO_TFM_NEED_KEY);
+       skcipher_set_needkey(skcipher);
 
        if (alg->exit)
                skcipher->base.exit = crypto_skcipher_exit_tfm;