Linux Audio

Check our new training course

Loading...
v5.14.15
  1// SPDX-License-Identifier: GPL-2.0-or-later
  2/*
  3 * INET		An implementation of the TCP/IP protocol suite for the LINUX
  4 *		operating system.  INET is implemented using the BSD Socket
  5 *		interface as the means of communication with the user level.
  6 *
  7 *		Generic INET6 transport hashtables
  8 *
  9 * Authors:	Lotsa people, from code originally in tcp, generalised here
 10 *		by Arnaldo Carvalho de Melo <acme@mandriva.com>
 
 
 
 
 
 11 */
 12
 13#include <linux/module.h>
 14#include <linux/random.h>
 15
 16#include <net/addrconf.h>
 17#include <net/inet_connection_sock.h>
 18#include <net/inet_hashtables.h>
 19#include <net/inet6_hashtables.h>
 20#include <net/secure_seq.h>
 21#include <net/ip.h>
 22#include <net/sock_reuseport.h>
 23
 24extern struct inet_hashinfo tcp_hashinfo;
 25
 26u32 inet6_ehashfn(const struct net *net,
 27		  const struct in6_addr *laddr, const u16 lport,
 28		  const struct in6_addr *faddr, const __be16 fport)
 29{
 30	static u32 inet6_ehash_secret __read_mostly;
 31	static u32 ipv6_hash_secret __read_mostly;
 32
 33	u32 lhash, fhash;
 34
 35	net_get_random_once(&inet6_ehash_secret, sizeof(inet6_ehash_secret));
 36	net_get_random_once(&ipv6_hash_secret, sizeof(ipv6_hash_secret));
 37
 38	lhash = (__force u32)laddr->s6_addr32[3];
 39	fhash = __ipv6_addr_jhash(faddr, ipv6_hash_secret);
 40
 41	return __inet6_ehashfn(lhash, lport, fhash, fport,
 42			       inet6_ehash_secret + net_hash_mix(net));
 43}
 44
 45/*
 46 * Sockets in TCP_CLOSE state are _always_ taken out of the hash, so
 47 * we need not check it for TCP lookups anymore, thanks Alexey. -DaveM
 48 *
 49 * The sockhash lock must be held as a reader here.
 50 */
 51struct sock *__inet6_lookup_established(struct net *net,
 52					struct inet_hashinfo *hashinfo,
 53					   const struct in6_addr *saddr,
 54					   const __be16 sport,
 55					   const struct in6_addr *daddr,
 56					   const u16 hnum,
 57					   const int dif, const int sdif)
 58{
 59	struct sock *sk;
 60	const struct hlist_nulls_node *node;
 61	const __portpair ports = INET_COMBINED_PORTS(sport, hnum);
 62	/* Optimize here for direct hit, only listening connections can
 63	 * have wildcards anyways.
 64	 */
 65	unsigned int hash = inet6_ehashfn(net, daddr, hnum, saddr, sport);
 66	unsigned int slot = hash & hashinfo->ehash_mask;
 67	struct inet_ehash_bucket *head = &hashinfo->ehash[slot];
 68
 69
 70begin:
 71	sk_nulls_for_each_rcu(sk, node, &head->chain) {
 72		if (sk->sk_hash != hash)
 73			continue;
 74		if (!INET6_MATCH(sk, net, saddr, daddr, ports, dif, sdif))
 75			continue;
 76		if (unlikely(!refcount_inc_not_zero(&sk->sk_refcnt)))
 77			goto out;
 78
 79		if (unlikely(!INET6_MATCH(sk, net, saddr, daddr, ports, dif, sdif))) {
 80			sock_gen_put(sk);
 81			goto begin;
 82		}
 83		goto found;
 84	}
 85	if (get_nulls_value(node) != slot)
 86		goto begin;
 87out:
 88	sk = NULL;
 89found:
 90	return sk;
 91}
 92EXPORT_SYMBOL(__inet6_lookup_established);
 93
 94static inline int compute_score(struct sock *sk, struct net *net,
 95				const unsigned short hnum,
 96				const struct in6_addr *daddr,
 97				const int dif, const int sdif)
 98{
 99	int score = -1;
100
101	if (net_eq(sock_net(sk), net) && inet_sk(sk)->inet_num == hnum &&
102	    sk->sk_family == PF_INET6) {
103		if (!ipv6_addr_equal(&sk->sk_v6_rcv_saddr, daddr))
104			return -1;
105
106		if (!inet_sk_bound_dev_eq(net, sk->sk_bound_dev_if, dif, sdif))
107			return -1;
108
109		score =  sk->sk_bound_dev_if ? 2 : 1;
110		if (READ_ONCE(sk->sk_incoming_cpu) == raw_smp_processor_id())
 
 
 
 
 
 
 
 
 
 
 
111			score++;
112	}
113	return score;
114}
115
116static inline struct sock *lookup_reuseport(struct net *net, struct sock *sk,
117					    struct sk_buff *skb, int doff,
118					    const struct in6_addr *saddr,
119					    __be16 sport,
120					    const struct in6_addr *daddr,
121					    unsigned short hnum)
122{
123	struct sock *reuse_sk = NULL;
124	u32 phash;
125
126	if (sk->sk_reuseport) {
127		phash = inet6_ehashfn(net, daddr, hnum, saddr, sport);
128		reuse_sk = reuseport_select_sock(sk, phash, skb, doff);
129	}
130	return reuse_sk;
131}
132
133/* called with rcu_read_lock() */
134static struct sock *inet6_lhash2_lookup(struct net *net,
135		struct inet_listen_hashbucket *ilb2,
136		struct sk_buff *skb, int doff,
137		const struct in6_addr *saddr,
138		const __be16 sport, const struct in6_addr *daddr,
139		const unsigned short hnum, const int dif, const int sdif)
140{
 
141	struct inet_connection_sock *icsk;
142	struct sock *sk, *result = NULL;
143	int score, hiscore = 0;
 
144
145	inet_lhash2_for_each_icsk_rcu(icsk, &ilb2->head) {
146		sk = (struct sock *)icsk;
147		score = compute_score(sk, net, hnum, daddr, dif, sdif);
 
148		if (score > hiscore) {
149			result = lookup_reuseport(net, sk, skb, doff,
150						  saddr, sport, daddr, hnum);
151			if (result)
152				return result;
153
 
 
 
154			result = sk;
155			hiscore = score;
156		}
157	}
158
159	return result;
160}
161
162static inline struct sock *inet6_lookup_run_bpf(struct net *net,
163						struct inet_hashinfo *hashinfo,
164						struct sk_buff *skb, int doff,
165						const struct in6_addr *saddr,
166						const __be16 sport,
167						const struct in6_addr *daddr,
168						const u16 hnum)
169{
170	struct sock *sk, *reuse_sk;
171	bool no_reuseport;
172
173	if (hashinfo != &tcp_hashinfo)
174		return NULL; /* only TCP is supported */
175
176	no_reuseport = bpf_sk_lookup_run_v6(net, IPPROTO_TCP,
177					    saddr, sport, daddr, hnum, &sk);
178	if (no_reuseport || IS_ERR_OR_NULL(sk))
179		return sk;
180
181	reuse_sk = lookup_reuseport(net, sk, skb, doff, saddr, sport, daddr, hnum);
182	if (reuse_sk)
183		sk = reuse_sk;
184	return sk;
185}
186
187struct sock *inet6_lookup_listener(struct net *net,
188		struct inet_hashinfo *hashinfo,
189		struct sk_buff *skb, int doff,
190		const struct in6_addr *saddr,
191		const __be16 sport, const struct in6_addr *daddr,
192		const unsigned short hnum, const int dif, const int sdif)
193{
 
 
 
194	struct inet_listen_hashbucket *ilb2;
195	struct sock *result = NULL;
 
196	unsigned int hash2;
 
197
198	/* Lookup redirect from BPF */
199	if (static_branch_unlikely(&bpf_sk_lookup_enabled)) {
200		result = inet6_lookup_run_bpf(net, hashinfo, skb, doff,
201					      saddr, sport, daddr, hnum);
202		if (result)
203			goto done;
204	}
205
206	hash2 = ipv6_portaddr_hash(net, daddr, hnum);
207	ilb2 = inet_lhash2_bucket(hashinfo, hash2);
 
 
208
209	result = inet6_lhash2_lookup(net, ilb2, skb, doff,
210				     saddr, sport, daddr, hnum,
211				     dif, sdif);
212	if (result)
213		goto done;
214
215	/* Lookup lhash2 with in6addr_any */
 
216	hash2 = ipv6_portaddr_hash(net, &in6addr_any, hnum);
217	ilb2 = inet_lhash2_bucket(hashinfo, hash2);
 
 
218
219	result = inet6_lhash2_lookup(net, ilb2, skb, doff,
220				     saddr, sport, &in6addr_any, hnum,
221				     dif, sdif);
222done:
223	if (IS_ERR(result))
224		return NULL;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
225	return result;
226}
227EXPORT_SYMBOL_GPL(inet6_lookup_listener);
228
229struct sock *inet6_lookup(struct net *net, struct inet_hashinfo *hashinfo,
230			  struct sk_buff *skb, int doff,
231			  const struct in6_addr *saddr, const __be16 sport,
232			  const struct in6_addr *daddr, const __be16 dport,
233			  const int dif)
234{
235	struct sock *sk;
236	bool refcounted;
237
238	sk = __inet6_lookup(net, hashinfo, skb, doff, saddr, sport, daddr,
239			    ntohs(dport), dif, 0, &refcounted);
240	if (sk && !refcounted && !refcount_inc_not_zero(&sk->sk_refcnt))
241		sk = NULL;
242	return sk;
243}
244EXPORT_SYMBOL_GPL(inet6_lookup);
245
246static int __inet6_check_established(struct inet_timewait_death_row *death_row,
247				     struct sock *sk, const __u16 lport,
248				     struct inet_timewait_sock **twp)
249{
250	struct inet_hashinfo *hinfo = death_row->hashinfo;
251	struct inet_sock *inet = inet_sk(sk);
252	const struct in6_addr *daddr = &sk->sk_v6_rcv_saddr;
253	const struct in6_addr *saddr = &sk->sk_v6_daddr;
254	const int dif = sk->sk_bound_dev_if;
255	struct net *net = sock_net(sk);
256	const int sdif = l3mdev_master_ifindex_by_index(net, dif);
257	const __portpair ports = INET_COMBINED_PORTS(inet->inet_dport, lport);
258	const unsigned int hash = inet6_ehashfn(net, daddr, lport, saddr,
259						inet->inet_dport);
260	struct inet_ehash_bucket *head = inet_ehash_bucket(hinfo, hash);
261	spinlock_t *lock = inet_ehash_lockp(hinfo, hash);
262	struct sock *sk2;
263	const struct hlist_nulls_node *node;
264	struct inet_timewait_sock *tw = NULL;
265
266	spin_lock(lock);
267
268	sk_nulls_for_each(sk2, node, &head->chain) {
269		if (sk2->sk_hash != hash)
270			continue;
271
272		if (likely(INET6_MATCH(sk2, net, saddr, daddr, ports,
273				       dif, sdif))) {
274			if (sk2->sk_state == TCP_TIME_WAIT) {
275				tw = inet_twsk(sk2);
276				if (twsk_unique(sk, sk2, twp))
277					break;
278			}
279			goto not_unique;
280		}
281	}
282
283	/* Must record num and sport now. Otherwise we will see
284	 * in hash table socket with a funny identity.
285	 */
286	inet->inet_num = lport;
287	inet->inet_sport = htons(lport);
288	sk->sk_hash = hash;
289	WARN_ON(!sk_unhashed(sk));
290	__sk_nulls_add_node_rcu(sk, &head->chain);
291	if (tw) {
292		sk_nulls_del_node_init_rcu((struct sock *)tw);
293		__NET_INC_STATS(net, LINUX_MIB_TIMEWAITRECYCLED);
294	}
295	spin_unlock(lock);
296	sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1);
297
298	if (twp) {
299		*twp = tw;
300	} else if (tw) {
301		/* Silly. Should hash-dance instead... */
302		inet_twsk_deschedule_put(tw);
303	}
304	return 0;
305
306not_unique:
307	spin_unlock(lock);
308	return -EADDRNOTAVAIL;
309}
310
311static u32 inet6_sk_port_offset(const struct sock *sk)
312{
313	const struct inet_sock *inet = inet_sk(sk);
314
315	return secure_ipv6_port_ephemeral(sk->sk_v6_rcv_saddr.s6_addr32,
316					  sk->sk_v6_daddr.s6_addr32,
317					  inet->inet_dport);
318}
319
320int inet6_hash_connect(struct inet_timewait_death_row *death_row,
321		       struct sock *sk)
322{
323	u32 port_offset = 0;
324
325	if (!inet_sk(sk)->inet_num)
326		port_offset = inet6_sk_port_offset(sk);
327	return __inet_hash_connect(death_row, sk, port_offset,
328				   __inet6_check_established);
329}
330EXPORT_SYMBOL_GPL(inet6_hash_connect);
331
332int inet6_hash(struct sock *sk)
333{
334	int err = 0;
335
336	if (sk->sk_state != TCP_CLOSE) {
337		local_bh_disable();
338		err = __inet_hash(sk, NULL);
339		local_bh_enable();
340	}
341
342	return err;
343}
344EXPORT_SYMBOL_GPL(inet6_hash);
v4.17
 
  1/*
  2 * INET		An implementation of the TCP/IP protocol suite for the LINUX
  3 *		operating system.  INET is implemented using the BSD Socket
  4 *		interface as the means of communication with the user level.
  5 *
  6 *		Generic INET6 transport hashtables
  7 *
  8 * Authors:	Lotsa people, from code originally in tcp, generalised here
  9 *		by Arnaldo Carvalho de Melo <acme@mandriva.com>
 10 *
 11 *	This program is free software; you can redistribute it and/or
 12 *      modify it under the terms of the GNU General Public License
 13 *      as published by the Free Software Foundation; either version
 14 *      2 of the License, or (at your option) any later version.
 15 */
 16
 17#include <linux/module.h>
 18#include <linux/random.h>
 19
 20#include <net/addrconf.h>
 21#include <net/inet_connection_sock.h>
 22#include <net/inet_hashtables.h>
 23#include <net/inet6_hashtables.h>
 24#include <net/secure_seq.h>
 25#include <net/ip.h>
 26#include <net/sock_reuseport.h>
 27
 
 
 28u32 inet6_ehashfn(const struct net *net,
 29		  const struct in6_addr *laddr, const u16 lport,
 30		  const struct in6_addr *faddr, const __be16 fport)
 31{
 32	static u32 inet6_ehash_secret __read_mostly;
 33	static u32 ipv6_hash_secret __read_mostly;
 34
 35	u32 lhash, fhash;
 36
 37	net_get_random_once(&inet6_ehash_secret, sizeof(inet6_ehash_secret));
 38	net_get_random_once(&ipv6_hash_secret, sizeof(ipv6_hash_secret));
 39
 40	lhash = (__force u32)laddr->s6_addr32[3];
 41	fhash = __ipv6_addr_jhash(faddr, ipv6_hash_secret);
 42
 43	return __inet6_ehashfn(lhash, lport, fhash, fport,
 44			       inet6_ehash_secret + net_hash_mix(net));
 45}
 46
 47/*
 48 * Sockets in TCP_CLOSE state are _always_ taken out of the hash, so
 49 * we need not check it for TCP lookups anymore, thanks Alexey. -DaveM
 50 *
 51 * The sockhash lock must be held as a reader here.
 52 */
 53struct sock *__inet6_lookup_established(struct net *net,
 54					struct inet_hashinfo *hashinfo,
 55					   const struct in6_addr *saddr,
 56					   const __be16 sport,
 57					   const struct in6_addr *daddr,
 58					   const u16 hnum,
 59					   const int dif, const int sdif)
 60{
 61	struct sock *sk;
 62	const struct hlist_nulls_node *node;
 63	const __portpair ports = INET_COMBINED_PORTS(sport, hnum);
 64	/* Optimize here for direct hit, only listening connections can
 65	 * have wildcards anyways.
 66	 */
 67	unsigned int hash = inet6_ehashfn(net, daddr, hnum, saddr, sport);
 68	unsigned int slot = hash & hashinfo->ehash_mask;
 69	struct inet_ehash_bucket *head = &hashinfo->ehash[slot];
 70
 71
 72begin:
 73	sk_nulls_for_each_rcu(sk, node, &head->chain) {
 74		if (sk->sk_hash != hash)
 75			continue;
 76		if (!INET6_MATCH(sk, net, saddr, daddr, ports, dif, sdif))
 77			continue;
 78		if (unlikely(!refcount_inc_not_zero(&sk->sk_refcnt)))
 79			goto out;
 80
 81		if (unlikely(!INET6_MATCH(sk, net, saddr, daddr, ports, dif, sdif))) {
 82			sock_gen_put(sk);
 83			goto begin;
 84		}
 85		goto found;
 86	}
 87	if (get_nulls_value(node) != slot)
 88		goto begin;
 89out:
 90	sk = NULL;
 91found:
 92	return sk;
 93}
 94EXPORT_SYMBOL(__inet6_lookup_established);
 95
 96static inline int compute_score(struct sock *sk, struct net *net,
 97				const unsigned short hnum,
 98				const struct in6_addr *daddr,
 99				const int dif, const int sdif, bool exact_dif)
100{
101	int score = -1;
102
103	if (net_eq(sock_net(sk), net) && inet_sk(sk)->inet_num == hnum &&
104	    sk->sk_family == PF_INET6) {
 
 
105
106		score = 1;
107		if (!ipv6_addr_any(&sk->sk_v6_rcv_saddr)) {
108			if (!ipv6_addr_equal(&sk->sk_v6_rcv_saddr, daddr))
109				return -1;
110			score++;
111		}
112		if (sk->sk_bound_dev_if || exact_dif) {
113			bool dev_match = (sk->sk_bound_dev_if == dif ||
114					  sk->sk_bound_dev_if == sdif);
115
116			if (exact_dif && !dev_match)
117				return -1;
118			if (sk->sk_bound_dev_if && dev_match)
119				score++;
120		}
121		if (sk->sk_incoming_cpu == raw_smp_processor_id())
122			score++;
123	}
124	return score;
125}
126
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
127/* called with rcu_read_lock() */
128static struct sock *inet6_lhash2_lookup(struct net *net,
129		struct inet_listen_hashbucket *ilb2,
130		struct sk_buff *skb, int doff,
131		const struct in6_addr *saddr,
132		const __be16 sport, const struct in6_addr *daddr,
133		const unsigned short hnum, const int dif, const int sdif)
134{
135	bool exact_dif = inet6_exact_dif_match(net, skb);
136	struct inet_connection_sock *icsk;
137	struct sock *sk, *result = NULL;
138	int score, hiscore = 0;
139	u32 phash = 0;
140
141	inet_lhash2_for_each_icsk_rcu(icsk, &ilb2->head) {
142		sk = (struct sock *)icsk;
143		score = compute_score(sk, net, hnum, daddr, dif, sdif,
144				      exact_dif);
145		if (score > hiscore) {
146			if (sk->sk_reuseport) {
147				phash = inet6_ehashfn(net, daddr, hnum,
148						      saddr, sport);
149				result = reuseport_select_sock(sk, phash,
150							       skb, doff);
151				if (result)
152					return result;
153			}
154			result = sk;
155			hiscore = score;
156		}
157	}
158
159	return result;
160}
161
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
162struct sock *inet6_lookup_listener(struct net *net,
163		struct inet_hashinfo *hashinfo,
164		struct sk_buff *skb, int doff,
165		const struct in6_addr *saddr,
166		const __be16 sport, const struct in6_addr *daddr,
167		const unsigned short hnum, const int dif, const int sdif)
168{
169	unsigned int hash = inet_lhashfn(net, hnum);
170	struct inet_listen_hashbucket *ilb = &hashinfo->listening_hash[hash];
171	bool exact_dif = inet6_exact_dif_match(net, skb);
172	struct inet_listen_hashbucket *ilb2;
173	struct sock *sk, *result = NULL;
174	int score, hiscore = 0;
175	unsigned int hash2;
176	u32 phash = 0;
177
178	if (ilb->count <= 10 || !hashinfo->lhash2)
179		goto port_lookup;
180
181	/* Too many sk in the ilb bucket (which is hashed by port alone).
182	 * Try lhash2 (which is hashed by port and addr) instead.
183	 */
 
184
185	hash2 = ipv6_portaddr_hash(net, daddr, hnum);
186	ilb2 = inet_lhash2_bucket(hashinfo, hash2);
187	if (ilb2->count > ilb->count)
188		goto port_lookup;
189
190	result = inet6_lhash2_lookup(net, ilb2, skb, doff,
191				     saddr, sport, daddr, hnum,
192				     dif, sdif);
193	if (result)
194		return result;
195
196	/* Lookup lhash2 with in6addr_any */
197
198	hash2 = ipv6_portaddr_hash(net, &in6addr_any, hnum);
199	ilb2 = inet_lhash2_bucket(hashinfo, hash2);
200	if (ilb2->count > ilb->count)
201		goto port_lookup;
202
203	return inet6_lhash2_lookup(net, ilb2, skb, doff,
204				   saddr, sport, daddr, hnum,
205				   dif, sdif);
206
207port_lookup:
208	sk_for_each(sk, &ilb->head) {
209		score = compute_score(sk, net, hnum, daddr, dif, sdif, exact_dif);
210		if (score > hiscore) {
211			if (sk->sk_reuseport) {
212				phash = inet6_ehashfn(net, daddr, hnum,
213						      saddr, sport);
214				result = reuseport_select_sock(sk, phash,
215							       skb, doff);
216				if (result)
217					return result;
218			}
219			result = sk;
220			hiscore = score;
221		}
222	}
223	return result;
224}
225EXPORT_SYMBOL_GPL(inet6_lookup_listener);
226
227struct sock *inet6_lookup(struct net *net, struct inet_hashinfo *hashinfo,
228			  struct sk_buff *skb, int doff,
229			  const struct in6_addr *saddr, const __be16 sport,
230			  const struct in6_addr *daddr, const __be16 dport,
231			  const int dif)
232{
233	struct sock *sk;
234	bool refcounted;
235
236	sk = __inet6_lookup(net, hashinfo, skb, doff, saddr, sport, daddr,
237			    ntohs(dport), dif, 0, &refcounted);
238	if (sk && !refcounted && !refcount_inc_not_zero(&sk->sk_refcnt))
239		sk = NULL;
240	return sk;
241}
242EXPORT_SYMBOL_GPL(inet6_lookup);
243
244static int __inet6_check_established(struct inet_timewait_death_row *death_row,
245				     struct sock *sk, const __u16 lport,
246				     struct inet_timewait_sock **twp)
247{
248	struct inet_hashinfo *hinfo = death_row->hashinfo;
249	struct inet_sock *inet = inet_sk(sk);
250	const struct in6_addr *daddr = &sk->sk_v6_rcv_saddr;
251	const struct in6_addr *saddr = &sk->sk_v6_daddr;
252	const int dif = sk->sk_bound_dev_if;
253	struct net *net = sock_net(sk);
254	const int sdif = l3mdev_master_ifindex_by_index(net, dif);
255	const __portpair ports = INET_COMBINED_PORTS(inet->inet_dport, lport);
256	const unsigned int hash = inet6_ehashfn(net, daddr, lport, saddr,
257						inet->inet_dport);
258	struct inet_ehash_bucket *head = inet_ehash_bucket(hinfo, hash);
259	spinlock_t *lock = inet_ehash_lockp(hinfo, hash);
260	struct sock *sk2;
261	const struct hlist_nulls_node *node;
262	struct inet_timewait_sock *tw = NULL;
263
264	spin_lock(lock);
265
266	sk_nulls_for_each(sk2, node, &head->chain) {
267		if (sk2->sk_hash != hash)
268			continue;
269
270		if (likely(INET6_MATCH(sk2, net, saddr, daddr, ports,
271				       dif, sdif))) {
272			if (sk2->sk_state == TCP_TIME_WAIT) {
273				tw = inet_twsk(sk2);
274				if (twsk_unique(sk, sk2, twp))
275					break;
276			}
277			goto not_unique;
278		}
279	}
280
281	/* Must record num and sport now. Otherwise we will see
282	 * in hash table socket with a funny identity.
283	 */
284	inet->inet_num = lport;
285	inet->inet_sport = htons(lport);
286	sk->sk_hash = hash;
287	WARN_ON(!sk_unhashed(sk));
288	__sk_nulls_add_node_rcu(sk, &head->chain);
289	if (tw) {
290		sk_nulls_del_node_init_rcu((struct sock *)tw);
291		__NET_INC_STATS(net, LINUX_MIB_TIMEWAITRECYCLED);
292	}
293	spin_unlock(lock);
294	sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1);
295
296	if (twp) {
297		*twp = tw;
298	} else if (tw) {
299		/* Silly. Should hash-dance instead... */
300		inet_twsk_deschedule_put(tw);
301	}
302	return 0;
303
304not_unique:
305	spin_unlock(lock);
306	return -EADDRNOTAVAIL;
307}
308
309static u32 inet6_sk_port_offset(const struct sock *sk)
310{
311	const struct inet_sock *inet = inet_sk(sk);
312
313	return secure_ipv6_port_ephemeral(sk->sk_v6_rcv_saddr.s6_addr32,
314					  sk->sk_v6_daddr.s6_addr32,
315					  inet->inet_dport);
316}
317
318int inet6_hash_connect(struct inet_timewait_death_row *death_row,
319		       struct sock *sk)
320{
321	u32 port_offset = 0;
322
323	if (!inet_sk(sk)->inet_num)
324		port_offset = inet6_sk_port_offset(sk);
325	return __inet_hash_connect(death_row, sk, port_offset,
326				   __inet6_check_established);
327}
328EXPORT_SYMBOL_GPL(inet6_hash_connect);
329
330int inet6_hash(struct sock *sk)
331{
332	int err = 0;
333
334	if (sk->sk_state != TCP_CLOSE) {
335		local_bh_disable();
336		err = __inet_hash(sk, NULL);
337		local_bh_enable();
338	}
339
340	return err;
341}
342EXPORT_SYMBOL_GPL(inet6_hash);