diff options
-rw-r--r-- | include/net/raw.h | 2 | ||||
-rw-r--r-- | net/ipv4/raw.c | 17 | ||||
-rw-r--r-- | net/ipv6/raw.c | 8 |
3 files changed, 15 insertions, 12 deletions
diff --git a/include/net/raw.h b/include/net/raw.h index 70b27c7c8167..43f1e3c07380 100644 --- a/include/net/raw.h +++ b/include/net/raw.h | |||
@@ -39,4 +39,6 @@ extern int raw_proc_init(void); | |||
39 | extern void raw_proc_exit(void); | 39 | extern void raw_proc_exit(void); |
40 | #endif | 40 | #endif |
41 | 41 | ||
42 | void raw_hash_sk(struct sock *sk, struct raw_hashinfo *h); | ||
43 | |||
42 | #endif /* _RAW_H */ | 44 | #endif /* _RAW_H */ |
diff --git a/net/ipv4/raw.c b/net/ipv4/raw.c index dd9f00b3ab42..28b95ae5d2dc 100644 --- a/net/ipv4/raw.c +++ b/net/ipv4/raw.c | |||
@@ -84,15 +84,22 @@ static struct raw_hashinfo raw_v4_hashinfo = { | |||
84 | .lock = __RW_LOCK_UNLOCKED(), | 84 | .lock = __RW_LOCK_UNLOCKED(), |
85 | }; | 85 | }; |
86 | 86 | ||
87 | static void raw_v4_hash(struct sock *sk) | 87 | void raw_hash_sk(struct sock *sk, struct raw_hashinfo *h) |
88 | { | 88 | { |
89 | struct hlist_head *head = &raw_v4_hashinfo.ht[inet_sk(sk)->num & | 89 | struct hlist_head *head; |
90 | (RAW_HTABLE_SIZE - 1)]; | ||
91 | 90 | ||
92 | write_lock_bh(&raw_v4_hashinfo.lock); | 91 | head = &h->ht[inet_sk(sk)->num & (RAW_HTABLE_SIZE - 1)]; |
92 | |||
93 | write_lock_bh(&h->lock); | ||
93 | sk_add_node(sk, head); | 94 | sk_add_node(sk, head); |
94 | sock_prot_inc_use(sk->sk_prot); | 95 | sock_prot_inc_use(sk->sk_prot); |
95 | write_unlock_bh(&raw_v4_hashinfo.lock); | 96 | write_unlock_bh(&h->lock); |
97 | } | ||
98 | EXPORT_SYMBOL_GPL(raw_hash_sk); | ||
99 | |||
100 | static void raw_v4_hash(struct sock *sk) | ||
101 | { | ||
102 | raw_hash_sk(sk, &raw_v4_hashinfo); | ||
96 | } | 103 | } |
97 | 104 | ||
98 | static void raw_v4_unhash(struct sock *sk) | 105 | static void raw_v4_unhash(struct sock *sk) |
diff --git a/net/ipv6/raw.c b/net/ipv6/raw.c index 15c72a6365a6..70db6f494538 100644 --- a/net/ipv6/raw.c +++ b/net/ipv6/raw.c | |||
@@ -67,13 +67,7 @@ static struct raw_hashinfo raw_v6_hashinfo = { | |||
67 | 67 | ||
68 | static void raw_v6_hash(struct sock *sk) | 68 | static void raw_v6_hash(struct sock *sk) |
69 | { | 69 | { |
70 | struct hlist_head *list = &raw_v6_hashinfo.ht[inet_sk(sk)->num & | 70 | raw_hash_sk(sk, &raw_v6_hashinfo); |
71 | (RAW_HTABLE_SIZE - 1)]; | ||
72 | |||
73 | write_lock_bh(&raw_v6_hashinfo.lock); | ||
74 | sk_add_node(sk, list); | ||
75 | sock_prot_inc_use(sk->sk_prot); | ||
76 | write_unlock_bh(&raw_v6_hashinfo.lock); | ||
77 | } | 71 | } |
78 | 72 | ||
79 | static void raw_v6_unhash(struct sock *sk) | 73 | static void raw_v6_unhash(struct sock *sk) |