[CRYPTO] digest: Store temporary digest in tfm

When the final result location is unaligned, we store the digest in a
temporary buffer before copying it to the final location.  Currently
that buffer sits on the stack.  This patch moves it to an area in the
tfm, just like the CBC IV buffer.

Signed-off-by: Herbert Xu <herbert@gondor.apana.org.au>

1 files changed, 8 insertions, 1 deletions

diff --git a/crypto/internal.h b/crypto/internal.h
index 03c00b0e6b60..b110b979b988 100644
--- a/crypto/internal.h
+++ b/crypto/internal.h
@@ -99,7 +99,14 @@ static inline void crypto_exit_proc(void)
 static inline unsigned int crypto_digest_ctxsize(struct crypto_alg *alg,
                                                  int flags)
 {
-        return alg->cra_ctxsize;
+        unsigned int len = alg->cra_ctxsize;
+
+        if (alg->cra_alignmask) {
+                len = ALIGN(len, (unsigned long)alg->cra_alignmask + 1);
+                len += alg->cra_digest.dia_digestsize;
+        }
+
+        return len;
 }
 
 static inline unsigned int crypto_cipher_ctxsize(struct crypto_alg *alg,