1 files changed, 20 insertions, 18 deletions

diff --git a/net/sunrpc/xprtsock.c b/net/sunrpc/xprtsock.c
index fcbfae82cb12..01121a4f0851 100644
--- a/net/sunrpc/xprtsock.c
+++ b/net/sunrpc/xprtsock.c
@@ -1316,27 +1316,29 @@ static int xs_bind6(struct sock_xprt *transport, struct socket *sock)
 static struct lock_class_key xs_key[2];
 static struct lock_class_key xs_slock_key[2];
 
-static inline void xs_reclassify_socket(struct socket *sock)
+static inline void xs_reclassify_socket4(struct socket *sock)
 {
         struct sock *sk = sock->sk;
+
         BUG_ON(sk->sk_lock.owner != NULL);
-        switch (sk->sk_family) {
-        case AF_INET:
-                sock_lock_init_class_and_name(sk, "slock-AF_INET-NFS",
-                        &xs_slock_key[0], "sk_lock-AF_INET-NFS", &xs_key[0]);
-                break;
+        sock_lock_init_class_and_name(sk, "slock-AF_INET-RPC",
+                &xs_slock_key[0], "sk_lock-AF_INET-RPC", &xs_key[0]);
+}
 
-        case AF_INET6:
-                sock_lock_init_class_and_name(sk, "slock-AF_INET6-NFS",
-                        &xs_slock_key[1], "sk_lock-AF_INET6-NFS", &xs_key[1]);
-                break;
+static inline void xs_reclassify_socket6(struct socket *sock)
+{
+        struct sock *sk = sock->sk;
 
-        default:
-                BUG();
-        }
+        BUG_ON(sk->sk_lock.owner != NULL);
+        sock_lock_init_class_and_name(sk, "slock-AF_INET6-RPC",
+                &xs_slock_key[1], "sk_lock-AF_INET6-RPC", &xs_key[1]);
 }
 #else
-static inline void xs_reclassify_socket(struct socket *sock)
+static inline void xs_reclassify_socket4(struct socket *sock)
+{
+}
+
+static inline void xs_reclassify_socket6(struct socket *sock)
 {
 }
 #endif
@@ -1394,7 +1396,7 @@ static void xs_udp_connect_worker4(struct work_struct *work)
                 dprintk("RPC:       can't create UDP transport socket (%d).\n", -err);
                 goto out;
         }
-        xs_reclassify_socket(sock);
+        xs_reclassify_socket4(sock);
 
         if (xs_bind4(transport, sock)) {
                 sock_release(sock);
@@ -1435,7 +1437,7 @@ static void xs_udp_connect_worker6(struct work_struct *work)
                 dprintk("RPC:       can't create UDP transport socket (%d).\n", -err);
                 goto out;
         }
-        xs_reclassify_socket(sock);
+        xs_reclassify_socket6(sock);
 
         if (xs_bind6(transport, sock) < 0) {
                 sock_release(sock);
@@ -1538,7 +1540,7 @@ static void xs_tcp_connect_worker4(struct work_struct *work)
                         dprintk("RPC:       can't create TCP transport socket (%d).\n", -err);
                         goto out;
                 }
-                xs_reclassify_socket(sock);
+                xs_reclassify_socket4(sock);
 
                 if (xs_bind4(transport, sock) < 0) {
                         sock_release(sock);
@@ -1599,7 +1601,7 @@ static void xs_tcp_connect_worker6(struct work_struct *work)
                         dprintk("RPC:       can't create TCP transport socket (%d).\n", -err);
                         goto out;
                 }
-                xs_reclassify_socket(sock);
+                xs_reclassify_socket6(sock);
 
                 if (xs_bind6(transport, sock) < 0) {
                         sock_release(sock);