3 files changed, 14 insertions, 7 deletions

diff --git a/crypto/af_alg.c b/crypto/af_alg.c
index 49fa8582138b..314c52c967e5 100644
--- a/crypto/af_alg.c
+++ b/crypto/af_alg.c
@@ -1060,12 +1060,19 @@ void af_alg_async_cb(struct crypto_async_request *_req, int err)
 }
 EXPORT_SYMBOL_GPL(af_alg_async_cb);
 
-__poll_t af_alg_poll_mask(struct socket *sock, __poll_t events)
+/**
+ * af_alg_poll - poll system call handler
+ */
+__poll_t af_alg_poll(struct file *file, struct socket *sock,
+                         poll_table *wait)
 {
         struct sock *sk = sock->sk;
         struct alg_sock *ask = alg_sk(sk);
         struct af_alg_ctx *ctx = ask->private;
-        __poll_t mask = 0;
+        __poll_t mask;
+
+        sock_poll_wait(file, sk_sleep(sk), wait);
+        mask = 0;
 
         if (!ctx->more || ctx->used)
                 mask |= EPOLLIN | EPOLLRDNORM;
@@ -1075,7 +1082,7 @@ __poll_t af_alg_poll_mask(struct socket *sock, __poll_t events)
 
         return mask;
 }
-EXPORT_SYMBOL_GPL(af_alg_poll_mask);
+EXPORT_SYMBOL_GPL(af_alg_poll);
 
 /**
  * af_alg_alloc_areq - allocate struct af_alg_async_req
diff --git a/crypto/algif_aead.c b/crypto/algif_aead.c
index 825524f27438..c40a8c7ee8ae 100644
--- a/crypto/algif_aead.c
+++ b/crypto/algif_aead.c
@@ -375,7 +375,7 @@ static struct proto_ops algif_aead_ops = {
         .sendmsg        =       aead_sendmsg,
         .sendpage       =       af_alg_sendpage,
         .recvmsg        =       aead_recvmsg,
-        .poll_mask      =       af_alg_poll_mask,
+        .poll           =       af_alg_poll,
 };
 
 static int aead_check_key(struct socket *sock)
@@ -471,7 +471,7 @@ static struct proto_ops algif_aead_ops_nokey = {
         .sendmsg        =       aead_sendmsg_nokey,
         .sendpage       =       aead_sendpage_nokey,
         .recvmsg        =       aead_recvmsg_nokey,
-        .poll_mask      =       af_alg_poll_mask,
+        .poll           =       af_alg_poll,
 };
 
 static void *aead_bind(const char *name, u32 type, u32 mask)
diff --git a/crypto/algif_skcipher.c b/crypto/algif_skcipher.c
index 4c04eb9888ad..cfdaab2b7d76 100644
--- a/crypto/algif_skcipher.c
+++ b/crypto/algif_skcipher.c
@@ -206,7 +206,7 @@ static struct proto_ops algif_skcipher_ops = {
         .sendmsg        =       skcipher_sendmsg,
         .sendpage       =       af_alg_sendpage,
         .recvmsg        =       skcipher_recvmsg,
-        .poll_mask      =       af_alg_poll_mask,
+        .poll           =       af_alg_poll,
 };
 
 static int skcipher_check_key(struct socket *sock)
@@ -302,7 +302,7 @@ static struct proto_ops algif_skcipher_ops_nokey = {
         .sendmsg        =       skcipher_sendmsg_nokey,
         .sendpage       =       skcipher_sendpage_nokey,
         .recvmsg        =       skcipher_recvmsg_nokey,
-        .poll_mask      =       af_alg_poll_mask,
+        .poll           =       af_alg_poll,
 };
 
 static void *skcipher_bind(const char *name, u32 type, u32 mask)