tipc: standardize accept routine
Comparing the behaviour of how to wait for events in TIPC accept() with other stacks, the TIPC implementation might be perceived as different, and sometimes even incorrect. As sk_sleep() and sk->sk_receive_queue variables associated with socket are not protected by socket lock, the process of calling accept() may be woken up improperly or sometimes cannot be woken up at all. After standardizing it with inet_csk_wait_for_connect routine, we can get benefits including: avoiding 'thundering herd' phenomenon, adding a timeout mechanism for accept(), coping with a pending signal, and having sk_sleep() and sk->sk_receive_queue being always protected within socket lock scope and so on. Signed-off-by: Ying Xue <ying.xue@windriver.com> Reviewed-by: Jon Maloy <jon.maloy@ericsson.com> Signed-off-by: David S. Miller <davem@davemloft.net>
This commit is contained in:
parent
78eb3a5379
commit
6398e23cdb
|
@ -1566,6 +1566,42 @@ static int listen(struct socket *sock, int len)
|
||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static int tipc_wait_for_accept(struct socket *sock, long timeo)
|
||||||
|
{
|
||||||
|
struct sock *sk = sock->sk;
|
||||||
|
DEFINE_WAIT(wait);
|
||||||
|
int err;
|
||||||
|
|
||||||
|
/* True wake-one mechanism for incoming connections: only
|
||||||
|
* one process gets woken up, not the 'whole herd'.
|
||||||
|
* Since we do not 'race & poll' for established sockets
|
||||||
|
* anymore, the common case will execute the loop only once.
|
||||||
|
*/
|
||||||
|
for (;;) {
|
||||||
|
prepare_to_wait_exclusive(sk_sleep(sk), &wait,
|
||||||
|
TASK_INTERRUPTIBLE);
|
||||||
|
if (skb_queue_empty(&sk->sk_receive_queue)) {
|
||||||
|
release_sock(sk);
|
||||||
|
timeo = schedule_timeout(timeo);
|
||||||
|
lock_sock(sk);
|
||||||
|
}
|
||||||
|
err = 0;
|
||||||
|
if (!skb_queue_empty(&sk->sk_receive_queue))
|
||||||
|
break;
|
||||||
|
err = -EINVAL;
|
||||||
|
if (sock->state != SS_LISTENING)
|
||||||
|
break;
|
||||||
|
err = sock_intr_errno(timeo);
|
||||||
|
if (signal_pending(current))
|
||||||
|
break;
|
||||||
|
err = -EAGAIN;
|
||||||
|
if (!timeo)
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
finish_wait(sk_sleep(sk), &wait);
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* accept - wait for connection request
|
* accept - wait for connection request
|
||||||
* @sock: listening socket
|
* @sock: listening socket
|
||||||
|
@ -1582,7 +1618,7 @@ static int accept(struct socket *sock, struct socket *new_sock, int flags)
|
||||||
struct tipc_port *new_tport;
|
struct tipc_port *new_tport;
|
||||||
struct tipc_msg *msg;
|
struct tipc_msg *msg;
|
||||||
u32 new_ref;
|
u32 new_ref;
|
||||||
|
long timeo;
|
||||||
int res;
|
int res;
|
||||||
|
|
||||||
lock_sock(sk);
|
lock_sock(sk);
|
||||||
|
@ -1592,18 +1628,10 @@ static int accept(struct socket *sock, struct socket *new_sock, int flags)
|
||||||
goto exit;
|
goto exit;
|
||||||
}
|
}
|
||||||
|
|
||||||
while (skb_queue_empty(&sk->sk_receive_queue)) {
|
timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
|
||||||
if (flags & O_NONBLOCK) {
|
res = tipc_wait_for_accept(sock, timeo);
|
||||||
res = -EWOULDBLOCK;
|
|
||||||
goto exit;
|
|
||||||
}
|
|
||||||
release_sock(sk);
|
|
||||||
res = wait_event_interruptible(*sk_sleep(sk),
|
|
||||||
(!skb_queue_empty(&sk->sk_receive_queue)));
|
|
||||||
lock_sock(sk);
|
|
||||||
if (res)
|
if (res)
|
||||||
goto exit;
|
goto exit;
|
||||||
}
|
|
||||||
|
|
||||||
buf = skb_peek(&sk->sk_receive_queue);
|
buf = skb_peek(&sk->sk_receive_queue);
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue