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  *		Support for INET connection oriented protocols.
7  *
8  * Authors:	See the TCP sources
9  *
10  *		This program is free software; you can redistribute it and/or
11  *		modify it under the terms of the GNU General Public License
12  *		as published by the Free Software Foundation; either version
13  *		2 of the License, or(at your option) any later version.
14  */
15 
16 #include <linux/module.h>
17 #include <linux/jhash.h>
18 
19 #include <net/inet_connection_sock.h>
20 #include <net/inet_hashtables.h>
21 #include <net/inet_timewait_sock.h>
22 #include <net/ip.h>
23 #include <net/route.h>
24 #include <net/tcp_states.h>
25 #include <net/xfrm.h>
26 
27 #ifdef INET_CSK_DEBUG
28 const char inet_csk_timer_bug_msg[] = "inet_csk BUG: unknown timer value\n";
29 EXPORT_SYMBOL(inet_csk_timer_bug_msg);
30 #endif
31 
32 /*
33  * This struct holds the first and last local port number.
34  */
35 struct local_ports sysctl_local_ports __read_mostly = {
36 	.lock = __SEQLOCK_UNLOCKED(sysctl_local_ports.lock),
37 	.range = { 32768, 61000 },
38 };
39 
40 unsigned long *sysctl_local_reserved_ports;
41 EXPORT_SYMBOL(sysctl_local_reserved_ports);
42 
inet_get_local_port_range(int * low,int * high)43 void inet_get_local_port_range(int *low, int *high)
44 {
45 	unsigned seq;
46 	do {
47 		seq = read_seqbegin(&sysctl_local_ports.lock);
48 
49 		*low = sysctl_local_ports.range[0];
50 		*high = sysctl_local_ports.range[1];
51 	} while (read_seqretry(&sysctl_local_ports.lock, seq));
52 }
53 EXPORT_SYMBOL(inet_get_local_port_range);
54 
inet_csk_bind_conflict(const struct sock * sk,const struct inet_bind_bucket * tb)55 int inet_csk_bind_conflict(const struct sock *sk,
56 			   const struct inet_bind_bucket *tb)
57 {
58 	struct sock *sk2;
59 	struct hlist_node *node;
60 	int reuse = sk->sk_reuse;
61 
62 	/*
63 	 * Unlike other sk lookup places we do not check
64 	 * for sk_net here, since _all_ the socks listed
65 	 * in tb->owners list belong to the same net - the
66 	 * one this bucket belongs to.
67 	 */
68 
69 	sk_for_each_bound(sk2, node, &tb->owners) {
70 		if (sk != sk2 &&
71 		    !inet_v6_ipv6only(sk2) &&
72 		    (!sk->sk_bound_dev_if ||
73 		     !sk2->sk_bound_dev_if ||
74 		     sk->sk_bound_dev_if == sk2->sk_bound_dev_if)) {
75 			if (!reuse || !sk2->sk_reuse ||
76 			    sk2->sk_state == TCP_LISTEN) {
77 				const __be32 sk2_rcv_saddr = sk_rcv_saddr(sk2);
78 				if (!sk2_rcv_saddr || !sk_rcv_saddr(sk) ||
79 				    sk2_rcv_saddr == sk_rcv_saddr(sk))
80 					break;
81 			}
82 		}
83 	}
84 	return node != NULL;
85 }
86 EXPORT_SYMBOL_GPL(inet_csk_bind_conflict);
87 
88 /* Obtain a reference to a local port for the given sock,
89  * if snum is zero it means select any available local port.
90  */
inet_csk_get_port(struct sock * sk,unsigned short snum)91 int inet_csk_get_port(struct sock *sk, unsigned short snum)
92 {
93 	struct inet_hashinfo *hashinfo = sk->sk_prot->h.hashinfo;
94 	struct inet_bind_hashbucket *head;
95 	struct hlist_node *node;
96 	struct inet_bind_bucket *tb;
97 	int ret, attempts = 5;
98 	struct net *net = sock_net(sk);
99 	int smallest_size = -1, smallest_rover;
100 
101 	local_bh_disable();
102 	if (!snum) {
103 		int remaining, rover, low, high;
104 
105 again:
106 		inet_get_local_port_range(&low, &high);
107 		remaining = (high - low) + 1;
108 		smallest_rover = rover = net_random() % remaining + low;
109 
110 		smallest_size = -1;
111 		do {
112 			if (inet_is_reserved_local_port(rover))
113 				goto next_nolock;
114 			head = &hashinfo->bhash[inet_bhashfn(net, rover,
115 					hashinfo->bhash_size)];
116 			spin_lock(&head->lock);
117 			inet_bind_bucket_for_each(tb, node, &head->chain)
118 				if (net_eq(ib_net(tb), net) && tb->port == rover) {
119 					if (tb->fastreuse > 0 &&
120 					    sk->sk_reuse &&
121 					    sk->sk_state != TCP_LISTEN &&
122 					    (tb->num_owners < smallest_size || smallest_size == -1)) {
123 						smallest_size = tb->num_owners;
124 						smallest_rover = rover;
125 						if (atomic_read(&hashinfo->bsockets) > (high - low) + 1) {
126 							snum = smallest_rover;
127 							goto tb_found;
128 						}
129 					}
130 					if (!inet_csk(sk)->icsk_af_ops->bind_conflict(sk, tb)) {
131 						snum = rover;
132 						goto tb_found;
133 					}
134 					goto next;
135 				}
136 			break;
137 		next:
138 			spin_unlock(&head->lock);
139 		next_nolock:
140 			if (++rover > high)
141 				rover = low;
142 		} while (--remaining > 0);
143 
144 		/* Exhausted local port range during search?  It is not
145 		 * possible for us to be holding one of the bind hash
146 		 * locks if this test triggers, because if 'remaining'
147 		 * drops to zero, we broke out of the do/while loop at
148 		 * the top level, not from the 'break;' statement.
149 		 */
150 		ret = 1;
151 		if (remaining <= 0) {
152 			if (smallest_size != -1) {
153 				snum = smallest_rover;
154 				goto have_snum;
155 			}
156 			goto fail;
157 		}
158 		/* OK, here is the one we will use.  HEAD is
159 		 * non-NULL and we hold it's mutex.
160 		 */
161 		snum = rover;
162 	} else {
163 have_snum:
164 		head = &hashinfo->bhash[inet_bhashfn(net, snum,
165 				hashinfo->bhash_size)];
166 		spin_lock(&head->lock);
167 		inet_bind_bucket_for_each(tb, node, &head->chain)
168 			if (net_eq(ib_net(tb), net) && tb->port == snum)
169 				goto tb_found;
170 	}
171 	tb = NULL;
172 	goto tb_not_found;
173 tb_found:
174 	if (!hlist_empty(&tb->owners)) {
175 		if (tb->fastreuse > 0 &&
176 		    sk->sk_reuse && sk->sk_state != TCP_LISTEN &&
177 		    smallest_size == -1) {
178 			goto success;
179 		} else {
180 			ret = 1;
181 			if (inet_csk(sk)->icsk_af_ops->bind_conflict(sk, tb)) {
182 				if (sk->sk_reuse && sk->sk_state != TCP_LISTEN &&
183 				    smallest_size != -1 && --attempts >= 0) {
184 					spin_unlock(&head->lock);
185 					goto again;
186 				}
187 				goto fail_unlock;
188 			}
189 		}
190 	}
191 tb_not_found:
192 	ret = 1;
193 	if (!tb && (tb = inet_bind_bucket_create(hashinfo->bind_bucket_cachep,
194 					net, head, snum)) == NULL)
195 		goto fail_unlock;
196 	if (hlist_empty(&tb->owners)) {
197 		if (sk->sk_reuse && sk->sk_state != TCP_LISTEN)
198 			tb->fastreuse = 1;
199 		else
200 			tb->fastreuse = 0;
201 	} else if (tb->fastreuse &&
202 		   (!sk->sk_reuse || sk->sk_state == TCP_LISTEN))
203 		tb->fastreuse = 0;
204 success:
205 	if (!inet_csk(sk)->icsk_bind_hash)
206 		inet_bind_hash(sk, tb, snum);
207 	WARN_ON(inet_csk(sk)->icsk_bind_hash != tb);
208 	ret = 0;
209 
210 fail_unlock:
211 	spin_unlock(&head->lock);
212 fail:
213 	local_bh_enable();
214 	return ret;
215 }
216 EXPORT_SYMBOL_GPL(inet_csk_get_port);
217 
218 /*
219  * Wait for an incoming connection, avoid race conditions. This must be called
220  * with the socket locked.
221  */
inet_csk_wait_for_connect(struct sock * sk,long timeo)222 static int inet_csk_wait_for_connect(struct sock *sk, long timeo)
223 {
224 	struct inet_connection_sock *icsk = inet_csk(sk);
225 	DEFINE_WAIT(wait);
226 	int err;
227 
228 	/*
229 	 * True wake-one mechanism for incoming connections: only
230 	 * one process gets woken up, not the 'whole herd'.
231 	 * Since we do not 'race & poll' for established sockets
232 	 * anymore, the common case will execute the loop only once.
233 	 *
234 	 * Subtle issue: "add_wait_queue_exclusive()" will be added
235 	 * after any current non-exclusive waiters, and we know that
236 	 * it will always _stay_ after any new non-exclusive waiters
237 	 * because all non-exclusive waiters are added at the
238 	 * beginning of the wait-queue. As such, it's ok to "drop"
239 	 * our exclusiveness temporarily when we get woken up without
240 	 * having to remove and re-insert us on the wait queue.
241 	 */
242 	for (;;) {
243 		prepare_to_wait_exclusive(sk_sleep(sk), &wait,
244 					  TASK_INTERRUPTIBLE);
245 		release_sock(sk);
246 		if (reqsk_queue_empty(&icsk->icsk_accept_queue))
247 			timeo = schedule_timeout(timeo);
248 		lock_sock(sk);
249 		err = 0;
250 		if (!reqsk_queue_empty(&icsk->icsk_accept_queue))
251 			break;
252 		err = -EINVAL;
253 		if (sk->sk_state != TCP_LISTEN)
254 			break;
255 		err = sock_intr_errno(timeo);
256 		if (signal_pending(current))
257 			break;
258 		err = -EAGAIN;
259 		if (!timeo)
260 			break;
261 	}
262 	finish_wait(sk_sleep(sk), &wait);
263 	return err;
264 }
265 
266 /*
267  * This will accept the next outstanding connection.
268  */
inet_csk_accept(struct sock * sk,int flags,int * err)269 struct sock *inet_csk_accept(struct sock *sk, int flags, int *err)
270 {
271 	struct inet_connection_sock *icsk = inet_csk(sk);
272 	struct sock *newsk;
273 	int error;
274 
275 	lock_sock(sk);
276 
277 	/* We need to make sure that this socket is listening,
278 	 * and that it has something pending.
279 	 */
280 	error = -EINVAL;
281 	if (sk->sk_state != TCP_LISTEN)
282 		goto out_err;
283 
284 	/* Find already established connection */
285 	if (reqsk_queue_empty(&icsk->icsk_accept_queue)) {
286 		long timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
287 
288 		/* If this is a non blocking socket don't sleep */
289 		error = -EAGAIN;
290 		if (!timeo)
291 			goto out_err;
292 
293 		error = inet_csk_wait_for_connect(sk, timeo);
294 		if (error)
295 			goto out_err;
296 	}
297 
298 	newsk = reqsk_queue_get_child(&icsk->icsk_accept_queue, sk);
299 	WARN_ON(newsk->sk_state == TCP_SYN_RECV);
300 out:
301 	release_sock(sk);
302 	return newsk;
303 out_err:
304 	newsk = NULL;
305 	*err = error;
306 	goto out;
307 }
308 EXPORT_SYMBOL(inet_csk_accept);
309 
310 /*
311  * Using different timers for retransmit, delayed acks and probes
312  * We may wish use just one timer maintaining a list of expire jiffies
313  * to optimize.
314  */
inet_csk_init_xmit_timers(struct sock * sk,void (* retransmit_handler)(unsigned long),void (* delack_handler)(unsigned long),void (* keepalive_handler)(unsigned long))315 void inet_csk_init_xmit_timers(struct sock *sk,
316 			       void (*retransmit_handler)(unsigned long),
317 			       void (*delack_handler)(unsigned long),
318 			       void (*keepalive_handler)(unsigned long))
319 {
320 	struct inet_connection_sock *icsk = inet_csk(sk);
321 
322 	setup_timer(&icsk->icsk_retransmit_timer, retransmit_handler,
323 			(unsigned long)sk);
324 	setup_timer(&icsk->icsk_delack_timer, delack_handler,
325 			(unsigned long)sk);
326 	setup_timer(&sk->sk_timer, keepalive_handler, (unsigned long)sk);
327 	icsk->icsk_pending = icsk->icsk_ack.pending = 0;
328 }
329 EXPORT_SYMBOL(inet_csk_init_xmit_timers);
330 
inet_csk_clear_xmit_timers(struct sock * sk)331 void inet_csk_clear_xmit_timers(struct sock *sk)
332 {
333 	struct inet_connection_sock *icsk = inet_csk(sk);
334 
335 	icsk->icsk_pending = icsk->icsk_ack.pending = icsk->icsk_ack.blocked = 0;
336 
337 	sk_stop_timer(sk, &icsk->icsk_retransmit_timer);
338 	sk_stop_timer(sk, &icsk->icsk_delack_timer);
339 	sk_stop_timer(sk, &sk->sk_timer);
340 }
341 EXPORT_SYMBOL(inet_csk_clear_xmit_timers);
342 
inet_csk_delete_keepalive_timer(struct sock * sk)343 void inet_csk_delete_keepalive_timer(struct sock *sk)
344 {
345 	sk_stop_timer(sk, &sk->sk_timer);
346 }
347 EXPORT_SYMBOL(inet_csk_delete_keepalive_timer);
348 
inet_csk_reset_keepalive_timer(struct sock * sk,unsigned long len)349 void inet_csk_reset_keepalive_timer(struct sock *sk, unsigned long len)
350 {
351 	sk_reset_timer(sk, &sk->sk_timer, jiffies + len);
352 }
353 EXPORT_SYMBOL(inet_csk_reset_keepalive_timer);
354 
inet_csk_route_req(struct sock * sk,struct flowi4 * fl4,const struct request_sock * req)355 struct dst_entry *inet_csk_route_req(struct sock *sk,
356 				     struct flowi4 *fl4,
357 				     const struct request_sock *req)
358 {
359 	struct rtable *rt;
360 	const struct inet_request_sock *ireq = inet_rsk(req);
361 	struct ip_options_rcu *opt = inet_rsk(req)->opt;
362 	struct net *net = sock_net(sk);
363 
364 	flowi4_init_output(fl4, sk->sk_bound_dev_if, sk->sk_mark,
365 			   RT_CONN_FLAGS(sk), RT_SCOPE_UNIVERSE,
366 			   sk->sk_protocol, inet_sk_flowi_flags(sk),
367 			   (opt && opt->opt.srr) ? opt->opt.faddr : ireq->rmt_addr,
368 			   ireq->loc_addr, ireq->rmt_port, inet_sk(sk)->inet_sport);
369 	security_req_classify_flow(req, flowi4_to_flowi(fl4));
370 	rt = ip_route_output_flow(net, fl4, sk);
371 	if (IS_ERR(rt))
372 		goto no_route;
373 	if (opt && opt->opt.is_strictroute && fl4->daddr != rt->rt_gateway)
374 		goto route_err;
375 	return &rt->dst;
376 
377 route_err:
378 	ip_rt_put(rt);
379 no_route:
380 	IP_INC_STATS_BH(net, IPSTATS_MIB_OUTNOROUTES);
381 	return NULL;
382 }
383 EXPORT_SYMBOL_GPL(inet_csk_route_req);
384 
inet_csk_route_child_sock(struct sock * sk,struct sock * newsk,const struct request_sock * req)385 struct dst_entry *inet_csk_route_child_sock(struct sock *sk,
386 					    struct sock *newsk,
387 					    const struct request_sock *req)
388 {
389 	const struct inet_request_sock *ireq = inet_rsk(req);
390 	struct inet_sock *newinet = inet_sk(newsk);
391 	struct ip_options_rcu *opt = ireq->opt;
392 	struct net *net = sock_net(sk);
393 	struct flowi4 *fl4;
394 	struct rtable *rt;
395 
396 	fl4 = &newinet->cork.fl.u.ip4;
397 	flowi4_init_output(fl4, sk->sk_bound_dev_if, sk->sk_mark,
398 			   RT_CONN_FLAGS(sk), RT_SCOPE_UNIVERSE,
399 			   sk->sk_protocol, inet_sk_flowi_flags(sk),
400 			   (opt && opt->opt.srr) ? opt->opt.faddr : ireq->rmt_addr,
401 			   ireq->loc_addr, ireq->rmt_port, inet_sk(sk)->inet_sport);
402 	security_req_classify_flow(req, flowi4_to_flowi(fl4));
403 	rt = ip_route_output_flow(net, fl4, sk);
404 	if (IS_ERR(rt))
405 		goto no_route;
406 	if (opt && opt->opt.is_strictroute && fl4->daddr != rt->rt_gateway)
407 		goto route_err;
408 	return &rt->dst;
409 
410 route_err:
411 	ip_rt_put(rt);
412 no_route:
413 	IP_INC_STATS_BH(net, IPSTATS_MIB_OUTNOROUTES);
414 	return NULL;
415 }
416 EXPORT_SYMBOL_GPL(inet_csk_route_child_sock);
417 
inet_synq_hash(const __be32 raddr,const __be16 rport,const u32 rnd,const u32 synq_hsize)418 static inline u32 inet_synq_hash(const __be32 raddr, const __be16 rport,
419 				 const u32 rnd, const u32 synq_hsize)
420 {
421 	return jhash_2words((__force u32)raddr, (__force u32)rport, rnd) & (synq_hsize - 1);
422 }
423 
424 #if IS_ENABLED(CONFIG_IPV6)
425 #define AF_INET_FAMILY(fam) ((fam) == AF_INET)
426 #else
427 #define AF_INET_FAMILY(fam) 1
428 #endif
429 
inet_csk_search_req(const struct sock * sk,struct request_sock *** prevp,const __be16 rport,const __be32 raddr,const __be32 laddr)430 struct request_sock *inet_csk_search_req(const struct sock *sk,
431 					 struct request_sock ***prevp,
432 					 const __be16 rport, const __be32 raddr,
433 					 const __be32 laddr)
434 {
435 	const struct inet_connection_sock *icsk = inet_csk(sk);
436 	struct listen_sock *lopt = icsk->icsk_accept_queue.listen_opt;
437 	struct request_sock *req, **prev;
438 
439 	for (prev = &lopt->syn_table[inet_synq_hash(raddr, rport, lopt->hash_rnd,
440 						    lopt->nr_table_entries)];
441 	     (req = *prev) != NULL;
442 	     prev = &req->dl_next) {
443 		const struct inet_request_sock *ireq = inet_rsk(req);
444 
445 		if (ireq->rmt_port == rport &&
446 		    ireq->rmt_addr == raddr &&
447 		    ireq->loc_addr == laddr &&
448 		    AF_INET_FAMILY(req->rsk_ops->family)) {
449 			WARN_ON(req->sk);
450 			*prevp = prev;
451 			break;
452 		}
453 	}
454 
455 	return req;
456 }
457 EXPORT_SYMBOL_GPL(inet_csk_search_req);
458 
inet_csk_reqsk_queue_hash_add(struct sock * sk,struct request_sock * req,unsigned long timeout)459 void inet_csk_reqsk_queue_hash_add(struct sock *sk, struct request_sock *req,
460 				   unsigned long timeout)
461 {
462 	struct inet_connection_sock *icsk = inet_csk(sk);
463 	struct listen_sock *lopt = icsk->icsk_accept_queue.listen_opt;
464 	const u32 h = inet_synq_hash(inet_rsk(req)->rmt_addr, inet_rsk(req)->rmt_port,
465 				     lopt->hash_rnd, lopt->nr_table_entries);
466 
467 	reqsk_queue_hash_req(&icsk->icsk_accept_queue, h, req, timeout);
468 	inet_csk_reqsk_queue_added(sk, timeout);
469 }
470 EXPORT_SYMBOL_GPL(inet_csk_reqsk_queue_hash_add);
471 
472 /* Only thing we need from tcp.h */
473 extern int sysctl_tcp_synack_retries;
474 
475 
476 /* Decide when to expire the request and when to resend SYN-ACK */
syn_ack_recalc(struct request_sock * req,const int thresh,const int max_retries,const u8 rskq_defer_accept,int * expire,int * resend)477 static inline void syn_ack_recalc(struct request_sock *req, const int thresh,
478 				  const int max_retries,
479 				  const u8 rskq_defer_accept,
480 				  int *expire, int *resend)
481 {
482 	if (!rskq_defer_accept) {
483 		*expire = req->retrans >= thresh;
484 		*resend = 1;
485 		return;
486 	}
487 	*expire = req->retrans >= thresh &&
488 		  (!inet_rsk(req)->acked || req->retrans >= max_retries);
489 	/*
490 	 * Do not resend while waiting for data after ACK,
491 	 * start to resend on end of deferring period to give
492 	 * last chance for data or ACK to create established socket.
493 	 */
494 	*resend = !inet_rsk(req)->acked ||
495 		  req->retrans >= rskq_defer_accept - 1;
496 }
497 
inet_csk_reqsk_queue_prune(struct sock * parent,const unsigned long interval,const unsigned long timeout,const unsigned long max_rto)498 void inet_csk_reqsk_queue_prune(struct sock *parent,
499 				const unsigned long interval,
500 				const unsigned long timeout,
501 				const unsigned long max_rto)
502 {
503 	struct inet_connection_sock *icsk = inet_csk(parent);
504 	struct request_sock_queue *queue = &icsk->icsk_accept_queue;
505 	struct listen_sock *lopt = queue->listen_opt;
506 	int max_retries = icsk->icsk_syn_retries ? : sysctl_tcp_synack_retries;
507 	int thresh = max_retries;
508 	unsigned long now = jiffies;
509 	struct request_sock **reqp, *req;
510 	int i, budget;
511 
512 	if (lopt == NULL || lopt->qlen == 0)
513 		return;
514 
515 	/* Normally all the openreqs are young and become mature
516 	 * (i.e. converted to established socket) for first timeout.
517 	 * If synack was not acknowledged for 3 seconds, it means
518 	 * one of the following things: synack was lost, ack was lost,
519 	 * rtt is high or nobody planned to ack (i.e. synflood).
520 	 * When server is a bit loaded, queue is populated with old
521 	 * open requests, reducing effective size of queue.
522 	 * When server is well loaded, queue size reduces to zero
523 	 * after several minutes of work. It is not synflood,
524 	 * it is normal operation. The solution is pruning
525 	 * too old entries overriding normal timeout, when
526 	 * situation becomes dangerous.
527 	 *
528 	 * Essentially, we reserve half of room for young
529 	 * embrions; and abort old ones without pity, if old
530 	 * ones are about to clog our table.
531 	 */
532 	if (lopt->qlen>>(lopt->max_qlen_log-1)) {
533 		int young = (lopt->qlen_young<<1);
534 
535 		while (thresh > 2) {
536 			if (lopt->qlen < young)
537 				break;
538 			thresh--;
539 			young <<= 1;
540 		}
541 	}
542 
543 	if (queue->rskq_defer_accept)
544 		max_retries = queue->rskq_defer_accept;
545 
546 	budget = 2 * (lopt->nr_table_entries / (timeout / interval));
547 	i = lopt->clock_hand;
548 
549 	do {
550 		reqp=&lopt->syn_table[i];
551 		while ((req = *reqp) != NULL) {
552 			if (time_after_eq(now, req->expires)) {
553 				int expire = 0, resend = 0;
554 
555 				syn_ack_recalc(req, thresh, max_retries,
556 					       queue->rskq_defer_accept,
557 					       &expire, &resend);
558 				if (req->rsk_ops->syn_ack_timeout)
559 					req->rsk_ops->syn_ack_timeout(parent, req);
560 				if (!expire &&
561 				    (!resend ||
562 				     !req->rsk_ops->rtx_syn_ack(parent, req, NULL) ||
563 				     inet_rsk(req)->acked)) {
564 					unsigned long timeo;
565 
566 					if (req->retrans++ == 0)
567 						lopt->qlen_young--;
568 					timeo = min((timeout << req->retrans), max_rto);
569 					req->expires = now + timeo;
570 					reqp = &req->dl_next;
571 					continue;
572 				}
573 
574 				/* Drop this request */
575 				inet_csk_reqsk_queue_unlink(parent, req, reqp);
576 				reqsk_queue_removed(queue, req);
577 				reqsk_free(req);
578 				continue;
579 			}
580 			reqp = &req->dl_next;
581 		}
582 
583 		i = (i + 1) & (lopt->nr_table_entries - 1);
584 
585 	} while (--budget > 0);
586 
587 	lopt->clock_hand = i;
588 
589 	if (lopt->qlen)
590 		inet_csk_reset_keepalive_timer(parent, interval);
591 }
592 EXPORT_SYMBOL_GPL(inet_csk_reqsk_queue_prune);
593 
594 /**
595  *	inet_csk_clone_lock - clone an inet socket, and lock its clone
596  *	@sk: the socket to clone
597  *	@req: request_sock
598  *	@priority: for allocation (%GFP_KERNEL, %GFP_ATOMIC, etc)
599  *
600  *	Caller must unlock socket even in error path (bh_unlock_sock(newsk))
601  */
inet_csk_clone_lock(const struct sock * sk,const struct request_sock * req,const gfp_t priority)602 struct sock *inet_csk_clone_lock(const struct sock *sk,
603 				 const struct request_sock *req,
604 				 const gfp_t priority)
605 {
606 	struct sock *newsk = sk_clone_lock(sk, priority);
607 
608 	if (newsk != NULL) {
609 		struct inet_connection_sock *newicsk = inet_csk(newsk);
610 
611 		newsk->sk_state = TCP_SYN_RECV;
612 		newicsk->icsk_bind_hash = NULL;
613 
614 		inet_sk(newsk)->inet_dport = inet_rsk(req)->rmt_port;
615 		inet_sk(newsk)->inet_num = ntohs(inet_rsk(req)->loc_port);
616 		inet_sk(newsk)->inet_sport = inet_rsk(req)->loc_port;
617 		newsk->sk_write_space = sk_stream_write_space;
618 
619 		newicsk->icsk_retransmits = 0;
620 		newicsk->icsk_backoff	  = 0;
621 		newicsk->icsk_probes_out  = 0;
622 
623 		/* Deinitialize accept_queue to trap illegal accesses. */
624 		memset(&newicsk->icsk_accept_queue, 0, sizeof(newicsk->icsk_accept_queue));
625 
626 		security_inet_csk_clone(newsk, req);
627 	}
628 	return newsk;
629 }
630 EXPORT_SYMBOL_GPL(inet_csk_clone_lock);
631 
632 /*
633  * At this point, there should be no process reference to this
634  * socket, and thus no user references at all.  Therefore we
635  * can assume the socket waitqueue is inactive and nobody will
636  * try to jump onto it.
637  */
inet_csk_destroy_sock(struct sock * sk)638 void inet_csk_destroy_sock(struct sock *sk)
639 {
640 	WARN_ON(sk->sk_state != TCP_CLOSE);
641 	WARN_ON(!sock_flag(sk, SOCK_DEAD));
642 
643 	/* It cannot be in hash table! */
644 	WARN_ON(!sk_unhashed(sk));
645 
646 	/* If it has not 0 inet_sk(sk)->inet_num, it must be bound */
647 	WARN_ON(inet_sk(sk)->inet_num && !inet_csk(sk)->icsk_bind_hash);
648 
649 	sk->sk_prot->destroy(sk);
650 
651 	sk_stream_kill_queues(sk);
652 
653 	xfrm_sk_free_policy(sk);
654 
655 	sk_refcnt_debug_release(sk);
656 
657 	percpu_counter_dec(sk->sk_prot->orphan_count);
658 	sock_put(sk);
659 }
660 EXPORT_SYMBOL(inet_csk_destroy_sock);
661 
662 /* This function allows to force a closure of a socket after the call to
663  * tcp/dccp_create_openreq_child().
664  */
inet_csk_prepare_forced_close(struct sock * sk)665 void inet_csk_prepare_forced_close(struct sock *sk)
666 {
667 	/* sk_clone_lock locked the socket and set refcnt to 2 */
668 	bh_unlock_sock(sk);
669 	sock_put(sk);
670 
671 	/* The below has to be done to allow calling inet_csk_destroy_sock */
672 	sock_set_flag(sk, SOCK_DEAD);
673 	percpu_counter_inc(sk->sk_prot->orphan_count);
674 	inet_sk(sk)->inet_num = 0;
675 }
676 EXPORT_SYMBOL(inet_csk_prepare_forced_close);
677 
inet_csk_listen_start(struct sock * sk,const int nr_table_entries)678 int inet_csk_listen_start(struct sock *sk, const int nr_table_entries)
679 {
680 	struct inet_sock *inet = inet_sk(sk);
681 	struct inet_connection_sock *icsk = inet_csk(sk);
682 	int rc = reqsk_queue_alloc(&icsk->icsk_accept_queue, nr_table_entries);
683 
684 	if (rc != 0)
685 		return rc;
686 
687 	sk->sk_max_ack_backlog = 0;
688 	sk->sk_ack_backlog = 0;
689 	inet_csk_delack_init(sk);
690 
691 	/* There is race window here: we announce ourselves listening,
692 	 * but this transition is still not validated by get_port().
693 	 * It is OK, because this socket enters to hash table only
694 	 * after validation is complete.
695 	 */
696 	sk->sk_state = TCP_LISTEN;
697 	if (!sk->sk_prot->get_port(sk, inet->inet_num)) {
698 		inet->inet_sport = htons(inet->inet_num);
699 
700 		sk_dst_reset(sk);
701 		sk->sk_prot->hash(sk);
702 
703 		return 0;
704 	}
705 
706 	sk->sk_state = TCP_CLOSE;
707 	__reqsk_queue_destroy(&icsk->icsk_accept_queue);
708 	return -EADDRINUSE;
709 }
710 EXPORT_SYMBOL_GPL(inet_csk_listen_start);
711 
712 /*
713  *	This routine closes sockets which have been at least partially
714  *	opened, but not yet accepted.
715  */
inet_csk_listen_stop(struct sock * sk)716 void inet_csk_listen_stop(struct sock *sk)
717 {
718 	struct inet_connection_sock *icsk = inet_csk(sk);
719 	struct request_sock *acc_req;
720 	struct request_sock *req;
721 
722 	inet_csk_delete_keepalive_timer(sk);
723 
724 	/* make all the listen_opt local to us */
725 	acc_req = reqsk_queue_yank_acceptq(&icsk->icsk_accept_queue);
726 
727 	/* Following specs, it would be better either to send FIN
728 	 * (and enter FIN-WAIT-1, it is normal close)
729 	 * or to send active reset (abort).
730 	 * Certainly, it is pretty dangerous while synflood, but it is
731 	 * bad justification for our negligence 8)
732 	 * To be honest, we are not able to make either
733 	 * of the variants now.			--ANK
734 	 */
735 	reqsk_queue_destroy(&icsk->icsk_accept_queue);
736 
737 	while ((req = acc_req) != NULL) {
738 		struct sock *child = req->sk;
739 
740 		acc_req = req->dl_next;
741 
742 		local_bh_disable();
743 		bh_lock_sock(child);
744 		WARN_ON(sock_owned_by_user(child));
745 		sock_hold(child);
746 
747 		sk->sk_prot->disconnect(child, O_NONBLOCK);
748 
749 		sock_orphan(child);
750 
751 		percpu_counter_inc(sk->sk_prot->orphan_count);
752 
753 		inet_csk_destroy_sock(child);
754 
755 		bh_unlock_sock(child);
756 		local_bh_enable();
757 		sock_put(child);
758 
759 		sk_acceptq_removed(sk);
760 		__reqsk_free(req);
761 	}
762 	WARN_ON(sk->sk_ack_backlog);
763 }
764 EXPORT_SYMBOL_GPL(inet_csk_listen_stop);
765 
inet_csk_addr2sockaddr(struct sock * sk,struct sockaddr * uaddr)766 void inet_csk_addr2sockaddr(struct sock *sk, struct sockaddr *uaddr)
767 {
768 	struct sockaddr_in *sin = (struct sockaddr_in *)uaddr;
769 	const struct inet_sock *inet = inet_sk(sk);
770 
771 	sin->sin_family		= AF_INET;
772 	sin->sin_addr.s_addr	= inet->inet_daddr;
773 	sin->sin_port		= inet->inet_dport;
774 }
775 EXPORT_SYMBOL_GPL(inet_csk_addr2sockaddr);
776 
777 #ifdef CONFIG_COMPAT
inet_csk_compat_getsockopt(struct sock * sk,int level,int optname,char __user * optval,int __user * optlen)778 int inet_csk_compat_getsockopt(struct sock *sk, int level, int optname,
779 			       char __user *optval, int __user *optlen)
780 {
781 	const struct inet_connection_sock *icsk = inet_csk(sk);
782 
783 	if (icsk->icsk_af_ops->compat_getsockopt != NULL)
784 		return icsk->icsk_af_ops->compat_getsockopt(sk, level, optname,
785 							    optval, optlen);
786 	return icsk->icsk_af_ops->getsockopt(sk, level, optname,
787 					     optval, optlen);
788 }
789 EXPORT_SYMBOL_GPL(inet_csk_compat_getsockopt);
790 
inet_csk_compat_setsockopt(struct sock * sk,int level,int optname,char __user * optval,unsigned int optlen)791 int inet_csk_compat_setsockopt(struct sock *sk, int level, int optname,
792 			       char __user *optval, unsigned int optlen)
793 {
794 	const struct inet_connection_sock *icsk = inet_csk(sk);
795 
796 	if (icsk->icsk_af_ops->compat_setsockopt != NULL)
797 		return icsk->icsk_af_ops->compat_setsockopt(sk, level, optname,
798 							    optval, optlen);
799 	return icsk->icsk_af_ops->setsockopt(sk, level, optname,
800 					     optval, optlen);
801 }
802 EXPORT_SYMBOL_GPL(inet_csk_compat_setsockopt);
803 #endif
804