1 files changed, 7 insertions, 7 deletions

diff --git a/crypto/aegis256.c b/crypto/aegis256.c
index 49882a28e93e..ecd6b7f34a2d 100644
--- a/crypto/aegis256.c
+++ b/crypto/aegis256.c
@@ -299,19 +299,19 @@ static void crypto_aegis256_process_crypt(struct aegis_state *state,
                                           const struct aegis256_ops *ops)
 {
         struct skcipher_walk walk;
-        u8 *src, *dst;
-        unsigned int chunksize;
 
         ops->skcipher_walk_init(&walk, req, false);
 
         while (walk.nbytes) {
-                src = walk.src.virt.addr;
-                dst = walk.dst.virt.addr;
-                chunksize = walk.nbytes;
+                unsigned int nbytes = walk.nbytes;
 
-                ops->crypt_chunk(state, dst, src, chunksize);
+                if (nbytes < walk.total)
+                        nbytes = round_down(nbytes, walk.stride);
 
-                skcipher_walk_done(&walk, 0);
+                ops->crypt_chunk(state, walk.dst.virt.addr, walk.src.virt.addr,
+                                 nbytes);
+
+                skcipher_walk_done(&walk, walk.nbytes - nbytes);
         }
 }