Techwizz/kernel-fxtec-pro1x
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

SUNRPC: Use shutdown() instead of close() when disconnecting a TCP socket

Browse source 
By using shutdown() rather than close() we allow the RPC client to wait
for the TCP close handshake to complete before we start trying to reconnect
using the same port.
We use shutdown(SHUT_WR) only instead of shutting down both directions,
however we wait until the server has closed the connection on its side.

Signed-off-by: Trond Myklebust <Trond.Myklebust@netapp.com>
This commit is contained in:
Trond Myklebust 2007-11-05 15:44:12 -05:00
parent ef80367071
commit e06799f958
1 changed files with 34 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
net/sunrpc/xprtsock.c
View file

@ -614,6 +614,22 @@ static int xs_udp_send_request(struct rpc_task *task)
return status; return status;
} }
/**
* xs_tcp_shutdown - gracefully shut down a TCP socket
* @xprt: transport
*
* Initiates a graceful shutdown of the TCP socket by calling the
* equivalent of shutdown(SHUT_WR);
*/
static void xs_tcp_shutdown(struct rpc_xprt *xprt)
{
struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
struct socket *sock = transport->sock;
if (sock != NULL)
kernel_sock_shutdown(sock, SHUT_WR);
}
static inline void xs_encode_tcp_record_marker(struct xdr_buf *buf) static inline void xs_encode_tcp_record_marker(struct xdr_buf *buf)
{ {
u32 reclen = buf->len - sizeof(rpc_fraghdr); u32 reclen = buf->len - sizeof(rpc_fraghdr);
@ -691,7 +707,7 @@ static int xs_tcp_send_request(struct rpc_task *task)
default: default:
dprintk("RPC: sendmsg returned unrecognized error %d\n", dprintk("RPC: sendmsg returned unrecognized error %d\n",
-status); -status);
xprt_disconnect(xprt); xs_tcp_shutdown(xprt);
break; break;
} }
@ -1637,8 +1653,7 @@ static void xs_tcp_connect_worker4(struct work_struct *work)
break; break;
default: default:
/* get rid of existing socket, and retry */ /* get rid of existing socket, and retry */
xs_close(xprt); xs_tcp_shutdown(xprt);
break;
} }
} }
out: out:
@ -1697,8 +1712,7 @@ static void xs_tcp_connect_worker6(struct work_struct *work)
break; break;
default: default:
/* get rid of existing socket, and retry */ /* get rid of existing socket, and retry */
xs_close(xprt); xs_tcp_shutdown(xprt);
break;
} }
} }
out: out:
@ -1745,6 +1759,19 @@ static void xs_connect(struct rpc_task *task)
} }
} }
static void xs_tcp_connect(struct rpc_task *task)
{
struct rpc_xprt *xprt = task->tk_xprt;
/* Initiate graceful shutdown of the socket if not already done */
if (test_bit(XPRT_CONNECTED, &xprt->state))
xs_tcp_shutdown(xprt);
/* Exit if we need to wait for socket shutdown to complete */
if (test_bit(XPRT_CLOSING, &xprt->state))
return;
xs_connect(task);
}
/** /**
* xs_udp_print_stats - display UDP socket-specifc stats * xs_udp_print_stats - display UDP socket-specifc stats
* @xprt: rpc_xprt struct containing statistics * @xprt: rpc_xprt struct containing statistics
@ -1815,12 +1842,12 @@ static struct rpc_xprt_ops xs_tcp_ops = {
.release_xprt = xs_tcp_release_xprt, .release_xprt = xs_tcp_release_xprt,
.rpcbind = rpcb_getport_async, .rpcbind = rpcb_getport_async,
.set_port = xs_set_port, .set_port = xs_set_port,
.connect = xs_connect, .connect = xs_tcp_connect,
.buf_alloc = rpc_malloc, .buf_alloc = rpc_malloc,
.buf_free = rpc_free, .buf_free = rpc_free,
.send_request = xs_tcp_send_request, .send_request = xs_tcp_send_request,
.set_retrans_timeout = xprt_set_retrans_timeout_def, .set_retrans_timeout = xprt_set_retrans_timeout_def,
.close = xs_close, .close = xs_tcp_shutdown,
.destroy = xs_destroy, .destroy = xs_destroy,
.print_stats = xs_tcp_print_stats, .print_stats = xs_tcp_print_stats,
}; };