1 files changed, 4 insertions, 0 deletions

diff --git a/include/net/sock.h b/include/net/sock.h
index 7b3d6b856946..83805feea880 100644
--- a/include/net/sock.h
+++ b/include/net/sock.h
@@ -80,8 +80,12 @@ typedef struct {
         wait_queue_head_t       wq;
 } socket_lock_t;
 
+extern struct lock_class_key af_family_keys[AF_MAX];
+
 #define sock_lock_init(__sk) \
 do {    spin_lock_init(&((__sk)->sk_lock.slock)); \
+        lockdep_set_class(&(__sk)->sk_lock.slock, \
+                          af_family_keys + (__sk)->sk_family); \
         (__sk)->sk_lock.owner = NULL; \
         init_waitqueue_head(&((__sk)->sk_lock.wq)); \
 } while(0)