diff options
-rw-r--r-- | include/net/raw.h | 1 | ||||
-rw-r--r-- | net/ipv4/raw.c | 14 | ||||
-rw-r--r-- | net/ipv6/raw.c | 5 |
3 files changed, 12 insertions, 8 deletions
diff --git a/include/net/raw.h b/include/net/raw.h index 43f1e3c07380..81a1773b58ba 100644 --- a/include/net/raw.h +++ b/include/net/raw.h | |||
@@ -40,5 +40,6 @@ extern void raw_proc_exit(void); | |||
40 | #endif | 40 | #endif |
41 | 41 | ||
42 | void raw_hash_sk(struct sock *sk, struct raw_hashinfo *h); | 42 | void raw_hash_sk(struct sock *sk, struct raw_hashinfo *h); |
43 | void raw_unhash_sk(struct sock *sk, struct raw_hashinfo *h); | ||
43 | 44 | ||
44 | #endif /* _RAW_H */ | 45 | #endif /* _RAW_H */ |
diff --git a/net/ipv4/raw.c b/net/ipv4/raw.c index 28b95ae5d2dc..d24501a342a4 100644 --- a/net/ipv4/raw.c +++ b/net/ipv4/raw.c | |||
@@ -97,6 +97,15 @@ void raw_hash_sk(struct sock *sk, struct raw_hashinfo *h) | |||
97 | } | 97 | } |
98 | EXPORT_SYMBOL_GPL(raw_hash_sk); | 98 | EXPORT_SYMBOL_GPL(raw_hash_sk); |
99 | 99 | ||
100 | void raw_unhash_sk(struct sock *sk, struct raw_hashinfo *h) | ||
101 | { | ||
102 | write_lock_bh(&h->lock); | ||
103 | if (sk_del_node_init(sk)) | ||
104 | sock_prot_dec_use(sk->sk_prot); | ||
105 | write_unlock_bh(&h->lock); | ||
106 | } | ||
107 | EXPORT_SYMBOL_GPL(raw_unhash_sk); | ||
108 | |||
100 | static void raw_v4_hash(struct sock *sk) | 109 | static void raw_v4_hash(struct sock *sk) |
101 | { | 110 | { |
102 | raw_hash_sk(sk, &raw_v4_hashinfo); | 111 | raw_hash_sk(sk, &raw_v4_hashinfo); |
@@ -104,10 +113,7 @@ static void raw_v4_hash(struct sock *sk) | |||
104 | 113 | ||
105 | static void raw_v4_unhash(struct sock *sk) | 114 | static void raw_v4_unhash(struct sock *sk) |
106 | { | 115 | { |
107 | write_lock_bh(&raw_v4_hashinfo.lock); | 116 | raw_unhash_sk(sk, &raw_v4_hashinfo); |
108 | if (sk_del_node_init(sk)) | ||
109 | sock_prot_dec_use(sk->sk_prot); | ||
110 | write_unlock_bh(&raw_v4_hashinfo.lock); | ||
111 | } | 117 | } |
112 | 118 | ||
113 | static struct sock *__raw_v4_lookup(struct sock *sk, unsigned short num, | 119 | static struct sock *__raw_v4_lookup(struct sock *sk, unsigned short num, |
diff --git a/net/ipv6/raw.c b/net/ipv6/raw.c index 70db6f494538..422d27cfbe16 100644 --- a/net/ipv6/raw.c +++ b/net/ipv6/raw.c | |||
@@ -72,10 +72,7 @@ static void raw_v6_hash(struct sock *sk) | |||
72 | 72 | ||
73 | static void raw_v6_unhash(struct sock *sk) | 73 | static void raw_v6_unhash(struct sock *sk) |
74 | { | 74 | { |
75 | write_lock_bh(&raw_v6_hashinfo.lock); | 75 | raw_unhash_sk(sk, &raw_v6_hashinfo); |
76 | if (sk_del_node_init(sk)) | ||
77 | sock_prot_dec_use(sk->sk_prot); | ||
78 | write_unlock_bh(&raw_v6_hashinfo.lock); | ||
79 | } | 76 | } |
80 | 77 | ||
81 | 78 | ||