crypto: marvell/cesa - another fix up for of_get_named_gen_pool() rename

[karo-tx-linux.git] / drivers / crypto / omap-aes.c

diff --git a/drivers/crypto/omap-aes.c b/drivers/crypto/omap-aes.c
index 42f95a4326b0717fb4b55626add069037ecca9e8..9a28b7e07c71a88c2d4e6e89d3a0c01f323f3efa 100644 (file)
--- a/drivers/crypto/omap-aes.c
+++ b/drivers/crypto/omap-aes.c
@@ -554,15 +554,23 @@ static int omap_aes_crypt_dma_stop(struct omap_aes_dev *dd)
        return err;
 }
 
-static int omap_aes_check_aligned(struct scatterlist *sg)
+static int omap_aes_check_aligned(struct scatterlist *sg, int total)
 {
+       int len = 0;
+
        while (sg) {
                if (!IS_ALIGNED(sg->offset, 4))
                        return -1;
                if (!IS_ALIGNED(sg->length, AES_BLOCK_SIZE))
                        return -1;
+
+               len += sg->length;
                sg = sg_next(sg);
        }
+
+       if (len != total)
+               return -1;
+
        return 0;
 }
 
@@ -633,8 +641,8 @@ static int omap_aes_handle_queue(struct omap_aes_dev *dd,
        dd->in_sg = req->src;
        dd->out_sg = req->dst;
 
-       if (omap_aes_check_aligned(dd->in_sg) ||
-           omap_aes_check_aligned(dd->out_sg)) {
+       if (omap_aes_check_aligned(dd->in_sg, dd->total) ||
+           omap_aes_check_aligned(dd->out_sg, dd->total)) {
                if (omap_aes_copy_sgs(dd))
                        pr_err("Failed to copy SGs for unaligned cases\n");
                dd->sgs_copied = 1;