struct af_alg_ctx *ctx = ask->private;
        __poll_t mask;
 
-       sock_poll_wait(file, sk_sleep(sk), wait);
+       sock_poll_wait(file, wait);
        mask = 0;
 
        if (!ctx->more || ctx->used)
 
 /**
  * sock_poll_wait - place memory barrier behind the poll_wait call.
  * @filp:           file
- * @wait_address:   socket wait queue
  * @p:              poll_table
  *
  * See the comments in the wq_has_sleeper function.
  */
-static inline void sock_poll_wait(struct file *filp,
-               wait_queue_head_t *wait_address, poll_table *p)
+static inline void sock_poll_wait(struct file *filp, poll_table *p)
 {
-       if (!poll_does_not_wait(p) && wait_address) {
-               poll_wait(filp, wait_address, p);
+       struct socket *sock = filp->private_data;
+       wait_queue_head_t *wq = sk_sleep(sock->sk);
+
+       if (!poll_does_not_wait(p) && wq) {
+               poll_wait(filp, wq, p);
                /* We need to be sure we are in sync with the
                 * socket flags modification.
                 *
 
        struct atm_vcc *vcc;
        __poll_t mask;
 
-       sock_poll_wait(file, sk_sleep(sk), wait);
+       sock_poll_wait(file, wait);
        mask = 0;
 
        vcc = ATM_SD(sock);
 
        __poll_t mask;
        struct caifsock *cf_sk = container_of(sk, struct caifsock, sk);
 
-       sock_poll_wait(file, sk_sleep(sk), wait);
+       sock_poll_wait(file, wait);
        mask = 0;
 
        /* exceptional events? */
 
        struct sock *sk = sock->sk;
        __poll_t mask;
 
-       sock_poll_wait(file, sk_sleep(sk), wait);
+       sock_poll_wait(file, wait);
        mask = 0;
 
        /* exceptional events? */
 
        __poll_t mask;
        struct sock *sk = sock->sk;
 
-       sock_poll_wait(file, sk_sleep(sk), wait);
+       sock_poll_wait(file, wait);
        if (sk->sk_state == DCCP_LISTEN)
                return inet_csk_listen_poll(sk);
 
 
        const struct tcp_sock *tp = tcp_sk(sk);
        int state;
 
-       sock_poll_wait(file, sk_sleep(sk), wait);
+       sock_poll_wait(file, wait);
 
        state = inet_sk_state_load(sk);
        if (state == TCP_LISTEN)
 
        struct sock *sk = sock->sk;
        __poll_t mask = 0;
 
-       sock_poll_wait(file, sk_sleep(sk), wait);
+       sock_poll_wait(file, wait);
 
        if (sk->sk_state == IUCV_LISTEN)
                return iucv_accept_poll(sk);
 
 
        pr_debug("%p\n", sk);
 
-       sock_poll_wait(file, sk_sleep(sk), wait);
+       sock_poll_wait(file, wait);
 
        if (sk->sk_state == LLCP_LISTEN)
                return llcp_accept_poll(sk);
 
        struct rxrpc_sock *rx = rxrpc_sk(sk);
        __poll_t mask;
 
-       sock_poll_wait(file, sk_sleep(sk), wait);
+       sock_poll_wait(file, wait);
        mask = 0;
 
        /* the socket is readable if there are any messages waiting on the Rx
 
                        mask |= EPOLLERR;
        } else {
                if (sk->sk_state != SMC_CLOSED)
-                       sock_poll_wait(file, sk_sleep(sk), wait);
+                       sock_poll_wait(file, wait);
                if (sk->sk_err)
                        mask |= EPOLLERR;
                if ((sk->sk_shutdown == SHUTDOWN_MASK) ||
 
        struct tipc_sock *tsk = tipc_sk(sk);
        __poll_t revents = 0;
 
-       sock_poll_wait(file, sk_sleep(sk), wait);
+       sock_poll_wait(file, wait);
 
        if (sk->sk_shutdown & RCV_SHUTDOWN)
                revents |= EPOLLRDHUP | EPOLLIN | EPOLLRDNORM;
 
        struct sock *sk = sock->sk;
        __poll_t mask;
 
-       sock_poll_wait(file, sk_sleep(sk), wait);
+       sock_poll_wait(file, wait);
        mask = 0;
 
        /* exceptional events? */
        unsigned int writable;
        __poll_t mask;
 
-       sock_poll_wait(file, sk_sleep(sk), wait);
+       sock_poll_wait(file, wait);
        mask = 0;
 
        /* exceptional events? */