Extract a function to reset the record fragment parser.

Signed-off-by: Stefan Hajnoczi <stefa...@redhat.com>
---
 net/sunrpc/xprtsock.c | 33 +++++++++++++++++++++++----------
 1 file changed, 23 insertions(+), 10 deletions(-)

diff --git a/net/sunrpc/xprtsock.c b/net/sunrpc/xprtsock.c
index 06fde0e..9fa63f7 100644
--- a/net/sunrpc/xprtsock.c
+++ b/net/sunrpc/xprtsock.c
@@ -1431,6 +1431,28 @@ out:
 }
 
 /**
+ * xs_stream_reset_state - reset SOCK_STREAM record parser
+ * @transport: socket transport
+ * @read_sock: tcp_read_sock()-like function
+ *
+ */
+static void xs_stream_reset_state(struct rpc_xprt *xprt,
+                                 int (*read_sock)(struct sock *,
+                                                  read_descriptor_t *,
+                                                  sk_read_actor_t))
+{
+       struct sock_xprt *transport = container_of(xprt,
+                       struct sock_xprt, xprt);
+
+       transport->stream_offset = 0;
+       transport->stream_reclen = 0;
+       transport->stream_copied = 0;
+       transport->stream_flags =
+               STREAM_RCV_COPY_FRAGHDR | STREAM_RCV_COPY_XID;
+       transport->stream_read_sock = read_sock;
+}
+
+/**
  * xs_tcp_state_change - callback to handle TCP socket state changes
  * @sk: socket whose state has changed
  *
@@ -1454,16 +1476,7 @@ static void xs_tcp_state_change(struct sock *sk)
        case TCP_ESTABLISHED:
                spin_lock(&xprt->transport_lock);
                if (!xprt_test_and_set_connected(xprt)) {
-                       struct sock_xprt *transport = container_of(xprt,
-                                       struct sock_xprt, xprt);
-
-                       /* Reset stream record info */
-                       transport->stream_offset = 0;
-                       transport->stream_reclen = 0;
-                       transport->stream_copied = 0;
-                       transport->stream_flags =
-                               STREAM_RCV_COPY_FRAGHDR | STREAM_RCV_COPY_XID;
-                       transport->stream_read_sock = tcp_read_sock;
+                       xs_stream_reset_state(xprt, tcp_read_sock);
                        xprt->connect_cookie++;
 
                        xprt_wake_pending_tasks(xprt, -EAGAIN);
-- 
2.4.2

--
To unsubscribe from this list: send the line "unsubscribe netdev" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to