crypto: atmel-{aes,tdes} - Do not save IV for ECB mode

[mirror_ubuntu-hirsute-kernel.git] / drivers / crypto / atmel-aes.c

diff --git a/drivers/crypto/atmel-aes.c b/drivers/crypto/atmel-aes.c
index 262079ad7bfdbfdf615858c8fa8ff52b412ff2ee..f25dd0e1d52b85c5524b6b18453e12f9603bca35 100644 (file)
--- a/drivers/crypto/atmel-aes.c
+++ b/drivers/crypto/atmel-aes.c
@@ -515,6 +515,9 @@ static void atmel_aes_set_iv_as_last_ciphertext_block(struct atmel_aes_dev *dd)
 
 static inline int atmel_aes_complete(struct atmel_aes_dev *dd, int err)
 {
+       struct skcipher_request *req = skcipher_request_cast(dd->areq);
+       struct atmel_aes_reqctx *rctx = skcipher_request_ctx(req);
+
 #if IS_ENABLED(CONFIG_CRYPTO_DEV_ATMEL_AUTHENC)
        if (dd->ctx->is_aead)
                atmel_aes_authenc_complete(dd, err);
@@ -523,7 +526,8 @@ static inline int atmel_aes_complete(struct atmel_aes_dev *dd, int err)
        clk_disable(dd->iclk);
        dd->flags &= ~AES_FLAGS_BUSY;
 
-       if (!dd->ctx->is_aead)
+       if (!dd->ctx->is_aead &&
+           (rctx->mode & AES_FLAGS_OPMODE_MASK) != AES_FLAGS_ECB)
                atmel_aes_set_iv_as_last_ciphertext_block(dd);
 
        if (dd->is_async)
@@ -1130,7 +1134,8 @@ static int atmel_aes_crypt(struct skcipher_request *req, unsigned long mode)
        rctx = skcipher_request_ctx(req);
        rctx->mode = mode;
 
-       if (!(mode & AES_FLAGS_ENCRYPT) && (req->src == req->dst)) {
+       if ((mode & AES_FLAGS_OPMODE_MASK) != AES_FLAGS_ECB &&
+           !(mode & AES_FLAGS_ENCRYPT) && req->src == req->dst) {
                unsigned int ivsize = crypto_skcipher_ivsize(skcipher);
 
                if (req->cryptlen >= ivsize)