1 files changed, 16 insertions, 3 deletions

diff --git a/crypto/digest.c b/crypto/digest.c
index 8f4593268ce0..bc47af648cb1 100644
--- a/crypto/digest.c
+++ b/crypto/digest.c
@@ -14,7 +14,9 @@
 
 #include <linux/mm.h>
 #include <linux/errno.h>
+#include <linux/hardirq.h>
 #include <linux/highmem.h>
+#include <linux/kernel.h>
 #include <linux/module.h>
 #include <linux/scatterlist.h>
 
@@ -29,8 +31,8 @@ static int init(struct hash_desc *desc)
         return 0;
 }
 
-static int update(struct hash_desc *desc,
-                  struct scatterlist *sg, unsigned int nbytes)
+static int update2(struct hash_desc *desc,
+                   struct scatterlist *sg, unsigned int nbytes)
 {
         struct crypto_tfm *tfm = crypto_hash_tfm(desc->tfm);
         unsigned int alignmask = crypto_tfm_alg_alignmask(tfm);
@@ -81,6 +83,14 @@ static int update(struct hash_desc *desc,
         return 0;
 }
 
+static int update(struct hash_desc *desc,
+                  struct scatterlist *sg, unsigned int nbytes)
+{
+        if (WARN_ON_ONCE(in_irq()))
+                return -EDEADLK;
+        return update2(desc, sg, nbytes);
+}
+
 static int final(struct hash_desc *desc, u8 *out)
 {
         struct crypto_tfm *tfm = crypto_hash_tfm(desc->tfm);
@@ -118,8 +128,11 @@ static int setkey(struct crypto_hash *hash, const u8 *key, unsigned int keylen)
 static int digest(struct hash_desc *desc,
                   struct scatterlist *sg, unsigned int nbytes, u8 *out)
 {
+        if (WARN_ON_ONCE(in_irq()))
+                return -EDEADLK;
+
         init(desc);
-        update(desc, sg, nbytes);
+        update2(desc, sg, nbytes);
         return final(desc, out);
 }