|
@@ -60,7 +60,6 @@
|
|
|
|
|
|
static struct svc_sock *svc_setup_socket(struct svc_serv *, struct socket *,
|
|
|
int flags);
|
|
|
-static void svc_udp_data_ready(struct sock *);
|
|
|
static int svc_udp_recvfrom(struct svc_rqst *);
|
|
|
static int svc_udp_sendto(struct svc_rqst *);
|
|
|
static void svc_sock_detach(struct svc_xprt *);
|
|
@@ -398,48 +397,21 @@ static int svc_sock_secure_port(struct svc_rqst *rqstp)
|
|
|
return svc_port_is_privileged(svc_addr(rqstp));
|
|
|
}
|
|
|
|
|
|
-static bool sunrpc_waitqueue_active(wait_queue_head_t *wq)
|
|
|
-{
|
|
|
- if (!wq)
|
|
|
- return false;
|
|
|
- /*
|
|
|
- * There should normally be a memory * barrier here--see
|
|
|
- * wq_has_sleeper().
|
|
|
- *
|
|
|
- * It appears that isn't currently necessary, though, basically
|
|
|
- * because callers all appear to have sufficient memory barriers
|
|
|
- * between the time the relevant change is made and the
|
|
|
- * time they call these callbacks.
|
|
|
- *
|
|
|
- * The nfsd code itself doesn't actually explicitly wait on
|
|
|
- * these waitqueues, but it may wait on them for example in
|
|
|
- * sendpage() or sendmsg() calls. (And those may be the only
|
|
|
- * places, since it it uses nonblocking reads.)
|
|
|
- *
|
|
|
- * Maybe we should add the memory barriers anyway, but these are
|
|
|
- * hot paths so we'd need to be convinced there's no sigificant
|
|
|
- * penalty.
|
|
|
- */
|
|
|
- return waitqueue_active(wq);
|
|
|
-}
|
|
|
-
|
|
|
/*
|
|
|
* INET callback when data has been received on the socket.
|
|
|
*/
|
|
|
-static void svc_udp_data_ready(struct sock *sk)
|
|
|
+static void svc_data_ready(struct sock *sk)
|
|
|
{
|
|
|
struct svc_sock *svsk = (struct svc_sock *)sk->sk_user_data;
|
|
|
- wait_queue_head_t *wq = sk_sleep(sk);
|
|
|
|
|
|
if (svsk) {
|
|
|
dprintk("svc: socket %p(inet %p), busy=%d\n",
|
|
|
svsk, sk,
|
|
|
test_bit(XPT_BUSY, &svsk->sk_xprt.xpt_flags));
|
|
|
- set_bit(XPT_DATA, &svsk->sk_xprt.xpt_flags);
|
|
|
- svc_xprt_enqueue(&svsk->sk_xprt);
|
|
|
+ svsk->sk_odata(sk);
|
|
|
+ if (!test_and_set_bit(XPT_DATA, &svsk->sk_xprt.xpt_flags))
|
|
|
+ svc_xprt_enqueue(&svsk->sk_xprt);
|
|
|
}
|
|
|
- if (sunrpc_waitqueue_active(wq))
|
|
|
- wake_up_interruptible(wq);
|
|
|
}
|
|
|
|
|
|
/*
|
|
@@ -448,56 +420,22 @@ static void svc_udp_data_ready(struct sock *sk)
|
|
|
static void svc_write_space(struct sock *sk)
|
|
|
{
|
|
|
struct svc_sock *svsk = (struct svc_sock *)(sk->sk_user_data);
|
|
|
- wait_queue_head_t *wq = sk_sleep(sk);
|
|
|
|
|
|
if (svsk) {
|
|
|
dprintk("svc: socket %p(inet %p), write_space busy=%d\n",
|
|
|
svsk, sk, test_bit(XPT_BUSY, &svsk->sk_xprt.xpt_flags));
|
|
|
+ svsk->sk_owspace(sk);
|
|
|
svc_xprt_enqueue(&svsk->sk_xprt);
|
|
|
}
|
|
|
-
|
|
|
- if (sunrpc_waitqueue_active(wq)) {
|
|
|
- dprintk("RPC svc_write_space: someone sleeping on %p\n",
|
|
|
- svsk);
|
|
|
- wake_up_interruptible(wq);
|
|
|
- }
|
|
|
}
|
|
|
|
|
|
static int svc_tcp_has_wspace(struct svc_xprt *xprt)
|
|
|
{
|
|
|
- struct svc_sock *svsk = container_of(xprt, struct svc_sock, sk_xprt);
|
|
|
- struct svc_serv *serv = svsk->sk_xprt.xpt_server;
|
|
|
- int required;
|
|
|
+ struct svc_sock *svsk = container_of(xprt, struct svc_sock, sk_xprt);
|
|
|
|
|
|
if (test_bit(XPT_LISTENER, &xprt->xpt_flags))
|
|
|
return 1;
|
|
|
- required = atomic_read(&xprt->xpt_reserved) + serv->sv_max_mesg;
|
|
|
- if (sk_stream_wspace(svsk->sk_sk) >= required ||
|
|
|
- (sk_stream_min_wspace(svsk->sk_sk) == 0 &&
|
|
|
- atomic_read(&xprt->xpt_reserved) == 0))
|
|
|
- return 1;
|
|
|
- set_bit(SOCK_NOSPACE, &svsk->sk_sock->flags);
|
|
|
- return 0;
|
|
|
-}
|
|
|
-
|
|
|
-static void svc_tcp_write_space(struct sock *sk)
|
|
|
-{
|
|
|
- struct svc_sock *svsk = (struct svc_sock *)(sk->sk_user_data);
|
|
|
- struct socket *sock = sk->sk_socket;
|
|
|
-
|
|
|
- if (!sk_stream_is_writeable(sk) || !sock)
|
|
|
- return;
|
|
|
- if (!svsk || svc_tcp_has_wspace(&svsk->sk_xprt))
|
|
|
- clear_bit(SOCK_NOSPACE, &sock->flags);
|
|
|
- svc_write_space(sk);
|
|
|
-}
|
|
|
-
|
|
|
-static void svc_tcp_adjust_wspace(struct svc_xprt *xprt)
|
|
|
-{
|
|
|
- struct svc_sock *svsk = container_of(xprt, struct svc_sock, sk_xprt);
|
|
|
-
|
|
|
- if (svc_tcp_has_wspace(xprt))
|
|
|
- clear_bit(SOCK_NOSPACE, &svsk->sk_sock->flags);
|
|
|
+ return !test_bit(SOCK_NOSPACE, &svsk->sk_sock->flags);
|
|
|
}
|
|
|
|
|
|
/*
|
|
@@ -746,7 +684,7 @@ static void svc_udp_init(struct svc_sock *svsk, struct svc_serv *serv)
|
|
|
svc_xprt_init(sock_net(svsk->sk_sock->sk), &svc_udp_class,
|
|
|
&svsk->sk_xprt, serv);
|
|
|
clear_bit(XPT_CACHE_AUTH, &svsk->sk_xprt.xpt_flags);
|
|
|
- svsk->sk_sk->sk_data_ready = svc_udp_data_ready;
|
|
|
+ svsk->sk_sk->sk_data_ready = svc_data_ready;
|
|
|
svsk->sk_sk->sk_write_space = svc_write_space;
|
|
|
|
|
|
/* initialise setting must have enough space to
|
|
@@ -786,11 +724,12 @@ static void svc_udp_init(struct svc_sock *svsk, struct svc_serv *serv)
|
|
|
static void svc_tcp_listen_data_ready(struct sock *sk)
|
|
|
{
|
|
|
struct svc_sock *svsk = (struct svc_sock *)sk->sk_user_data;
|
|
|
- wait_queue_head_t *wq;
|
|
|
|
|
|
dprintk("svc: socket %p TCP (listen) state change %d\n",
|
|
|
sk, sk->sk_state);
|
|
|
|
|
|
+ if (svsk)
|
|
|
+ svsk->sk_odata(sk);
|
|
|
/*
|
|
|
* This callback may called twice when a new connection
|
|
|
* is established as a child socket inherits everything
|
|
@@ -808,10 +747,6 @@ static void svc_tcp_listen_data_ready(struct sock *sk)
|
|
|
} else
|
|
|
printk("svc: socket %p: no user data\n", sk);
|
|
|
}
|
|
|
-
|
|
|
- wq = sk_sleep(sk);
|
|
|
- if (sunrpc_waitqueue_active(wq))
|
|
|
- wake_up_interruptible_all(wq);
|
|
|
}
|
|
|
|
|
|
/*
|
|
@@ -820,7 +755,6 @@ static void svc_tcp_listen_data_ready(struct sock *sk)
|
|
|
static void svc_tcp_state_change(struct sock *sk)
|
|
|
{
|
|
|
struct svc_sock *svsk = (struct svc_sock *)sk->sk_user_data;
|
|
|
- wait_queue_head_t *wq = sk_sleep(sk);
|
|
|
|
|
|
dprintk("svc: socket %p TCP (connected) state change %d (svsk %p)\n",
|
|
|
sk, sk->sk_state, sk->sk_user_data);
|
|
@@ -828,26 +762,12 @@ static void svc_tcp_state_change(struct sock *sk)
|
|
|
if (!svsk)
|
|
|
printk("svc: socket %p: no user data\n", sk);
|
|
|
else {
|
|
|
- set_bit(XPT_CLOSE, &svsk->sk_xprt.xpt_flags);
|
|
|
- svc_xprt_enqueue(&svsk->sk_xprt);
|
|
|
- }
|
|
|
- if (sunrpc_waitqueue_active(wq))
|
|
|
- wake_up_interruptible_all(wq);
|
|
|
-}
|
|
|
-
|
|
|
-static void svc_tcp_data_ready(struct sock *sk)
|
|
|
-{
|
|
|
- struct svc_sock *svsk = (struct svc_sock *)sk->sk_user_data;
|
|
|
- wait_queue_head_t *wq = sk_sleep(sk);
|
|
|
-
|
|
|
- dprintk("svc: socket %p TCP data ready (svsk %p)\n",
|
|
|
- sk, sk->sk_user_data);
|
|
|
- if (svsk) {
|
|
|
- set_bit(XPT_DATA, &svsk->sk_xprt.xpt_flags);
|
|
|
- svc_xprt_enqueue(&svsk->sk_xprt);
|
|
|
+ svsk->sk_ostate(sk);
|
|
|
+ if (sk->sk_state != TCP_ESTABLISHED) {
|
|
|
+ set_bit(XPT_CLOSE, &svsk->sk_xprt.xpt_flags);
|
|
|
+ svc_xprt_enqueue(&svsk->sk_xprt);
|
|
|
+ }
|
|
|
}
|
|
|
- if (sunrpc_waitqueue_active(wq))
|
|
|
- wake_up_interruptible(wq);
|
|
|
}
|
|
|
|
|
|
/*
|
|
@@ -901,6 +821,11 @@ static struct svc_xprt *svc_tcp_accept(struct svc_xprt *xprt)
|
|
|
dprintk("%s: connect from %s\n", serv->sv_name,
|
|
|
__svc_print_addr(sin, buf, sizeof(buf)));
|
|
|
|
|
|
+ /* Reset the inherited callbacks before calling svc_setup_socket */
|
|
|
+ newsock->sk->sk_state_change = svsk->sk_ostate;
|
|
|
+ newsock->sk->sk_data_ready = svsk->sk_odata;
|
|
|
+ newsock->sk->sk_write_space = svsk->sk_owspace;
|
|
|
+
|
|
|
/* make sure that a write doesn't block forever when
|
|
|
* low on memory
|
|
|
*/
|
|
@@ -1317,7 +1242,6 @@ static struct svc_xprt_ops svc_tcp_ops = {
|
|
|
.xpo_has_wspace = svc_tcp_has_wspace,
|
|
|
.xpo_accept = svc_tcp_accept,
|
|
|
.xpo_secure_port = svc_sock_secure_port,
|
|
|
- .xpo_adjust_wspace = svc_tcp_adjust_wspace,
|
|
|
};
|
|
|
|
|
|
static struct svc_xprt_class svc_tcp_class = {
|
|
@@ -1357,8 +1281,8 @@ static void svc_tcp_init(struct svc_sock *svsk, struct svc_serv *serv)
|
|
|
} else {
|
|
|
dprintk("setting up TCP socket for reading\n");
|
|
|
sk->sk_state_change = svc_tcp_state_change;
|
|
|
- sk->sk_data_ready = svc_tcp_data_ready;
|
|
|
- sk->sk_write_space = svc_tcp_write_space;
|
|
|
+ sk->sk_data_ready = svc_data_ready;
|
|
|
+ sk->sk_write_space = svc_write_space;
|
|
|
|
|
|
svsk->sk_reclen = 0;
|
|
|
svsk->sk_tcplen = 0;
|
|
@@ -1368,8 +1292,13 @@ static void svc_tcp_init(struct svc_sock *svsk, struct svc_serv *serv)
|
|
|
tcp_sk(sk)->nonagle |= TCP_NAGLE_OFF;
|
|
|
|
|
|
set_bit(XPT_DATA, &svsk->sk_xprt.xpt_flags);
|
|
|
- if (sk->sk_state != TCP_ESTABLISHED)
|
|
|
+ switch (sk->sk_state) {
|
|
|
+ case TCP_SYN_RECV:
|
|
|
+ case TCP_ESTABLISHED:
|
|
|
+ break;
|
|
|
+ default:
|
|
|
set_bit(XPT_CLOSE, &svsk->sk_xprt.xpt_flags);
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
|
|
@@ -1428,17 +1357,14 @@ static struct svc_sock *svc_setup_socket(struct svc_serv *serv,
|
|
|
/* Initialize the socket */
|
|
|
if (sock->type == SOCK_DGRAM)
|
|
|
svc_udp_init(svsk, serv);
|
|
|
- else {
|
|
|
- /* initialise setting must have enough space to
|
|
|
- * receive and respond to one request.
|
|
|
- */
|
|
|
- svc_sock_setbufsize(svsk->sk_sock, 4 * serv->sv_max_mesg,
|
|
|
- 4 * serv->sv_max_mesg);
|
|
|
+ else
|
|
|
svc_tcp_init(svsk, serv);
|
|
|
- }
|
|
|
|
|
|
- dprintk("svc: svc_setup_socket created %p (inet %p)\n",
|
|
|
- svsk, svsk->sk_sk);
|
|
|
+ dprintk("svc: svc_setup_socket created %p (inet %p), "
|
|
|
+ "listen %d close %d\n",
|
|
|
+ svsk, svsk->sk_sk,
|
|
|
+ test_bit(XPT_LISTENER, &svsk->sk_xprt.xpt_flags),
|
|
|
+ test_bit(XPT_CLOSE, &svsk->sk_xprt.xpt_flags));
|
|
|
|
|
|
return svsk;
|
|
|
}
|
|
@@ -1606,18 +1532,16 @@ static void svc_sock_detach(struct svc_xprt *xprt)
|
|
|
{
|
|
|
struct svc_sock *svsk = container_of(xprt, struct svc_sock, sk_xprt);
|
|
|
struct sock *sk = svsk->sk_sk;
|
|
|
- wait_queue_head_t *wq;
|
|
|
|
|
|
dprintk("svc: svc_sock_detach(%p)\n", svsk);
|
|
|
|
|
|
/* put back the old socket callbacks */
|
|
|
+ lock_sock(sk);
|
|
|
sk->sk_state_change = svsk->sk_ostate;
|
|
|
sk->sk_data_ready = svsk->sk_odata;
|
|
|
sk->sk_write_space = svsk->sk_owspace;
|
|
|
-
|
|
|
- wq = sk_sleep(sk);
|
|
|
- if (sunrpc_waitqueue_active(wq))
|
|
|
- wake_up_interruptible(wq);
|
|
|
+ sk->sk_user_data = NULL;
|
|
|
+ release_sock(sk);
|
|
|
}
|
|
|
|
|
|
/*
|