xref: /linux/net/ipv4/tcp_ipv4.c (revision c4ee0af3fa0dc65f690fc908f02b8355f9576ea0)
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  *		Implementation of the Transmission Control Protocol(TCP).
7  *
8  *		IPv4 specific functions
9  *
10  *
11  *		code split from:
12  *		linux/ipv4/tcp.c
13  *		linux/ipv4/tcp_input.c
14  *		linux/ipv4/tcp_output.c
15  *
16  *		See tcp.c for author information
17  *
18  *	This program is free software; you can redistribute it and/or
19  *      modify it under the terms of the GNU General Public License
20  *      as published by the Free Software Foundation; either version
21  *      2 of the License, or (at your option) any later version.
22  */
23 
24 /*
25  * Changes:
26  *		David S. Miller	:	New socket lookup architecture.
27  *					This code is dedicated to John Dyson.
28  *		David S. Miller :	Change semantics of established hash,
29  *					half is devoted to TIME_WAIT sockets
30  *					and the rest go in the other half.
31  *		Andi Kleen :		Add support for syncookies and fixed
32  *					some bugs: ip options weren't passed to
33  *					the TCP layer, missed a check for an
34  *					ACK bit.
35  *		Andi Kleen :		Implemented fast path mtu discovery.
36  *	     				Fixed many serious bugs in the
37  *					request_sock handling and moved
38  *					most of it into the af independent code.
39  *					Added tail drop and some other bugfixes.
40  *					Added new listen semantics.
41  *		Mike McLagan	:	Routing by source
42  *	Juan Jose Ciarlante:		ip_dynaddr bits
43  *		Andi Kleen:		various fixes.
44  *	Vitaly E. Lavrov	:	Transparent proxy revived after year
45  *					coma.
46  *	Andi Kleen		:	Fix new listen.
47  *	Andi Kleen		:	Fix accept error reporting.
48  *	YOSHIFUJI Hideaki @USAGI and:	Support IPV6_V6ONLY socket option, which
49  *	Alexey Kuznetsov		allow both IPv4 and IPv6 sockets to bind
50  *					a single port at the same time.
51  */
52 
53 #define pr_fmt(fmt) "TCP: " fmt
54 
55 #include <linux/bottom_half.h>
56 #include <linux/types.h>
57 #include <linux/fcntl.h>
58 #include <linux/module.h>
59 #include <linux/random.h>
60 #include <linux/cache.h>
61 #include <linux/jhash.h>
62 #include <linux/init.h>
63 #include <linux/times.h>
64 #include <linux/slab.h>
65 
66 #include <net/net_namespace.h>
67 #include <net/icmp.h>
68 #include <net/inet_hashtables.h>
69 #include <net/tcp.h>
70 #include <net/transp_v6.h>
71 #include <net/ipv6.h>
72 #include <net/inet_common.h>
73 #include <net/timewait_sock.h>
74 #include <net/xfrm.h>
75 #include <net/netdma.h>
76 #include <net/secure_seq.h>
77 #include <net/tcp_memcontrol.h>
78 #include <net/busy_poll.h>
79 
80 #include <linux/inet.h>
81 #include <linux/ipv6.h>
82 #include <linux/stddef.h>
83 #include <linux/proc_fs.h>
84 #include <linux/seq_file.h>
85 
86 #include <linux/crypto.h>
87 #include <linux/scatterlist.h>
88 
89 int sysctl_tcp_tw_reuse __read_mostly;
90 int sysctl_tcp_low_latency __read_mostly;
91 EXPORT_SYMBOL(sysctl_tcp_low_latency);
92 
93 
94 #ifdef CONFIG_TCP_MD5SIG
95 static int tcp_v4_md5_hash_hdr(char *md5_hash, const struct tcp_md5sig_key *key,
96 			       __be32 daddr, __be32 saddr, const struct tcphdr *th);
97 #endif
98 
99 struct inet_hashinfo tcp_hashinfo;
100 EXPORT_SYMBOL(tcp_hashinfo);
101 
102 static inline __u32 tcp_v4_init_sequence(const struct sk_buff *skb)
103 {
104 	return secure_tcp_sequence_number(ip_hdr(skb)->daddr,
105 					  ip_hdr(skb)->saddr,
106 					  tcp_hdr(skb)->dest,
107 					  tcp_hdr(skb)->source);
108 }
109 
110 int tcp_twsk_unique(struct sock *sk, struct sock *sktw, void *twp)
111 {
112 	const struct tcp_timewait_sock *tcptw = tcp_twsk(sktw);
113 	struct tcp_sock *tp = tcp_sk(sk);
114 
115 	/* With PAWS, it is safe from the viewpoint
116 	   of data integrity. Even without PAWS it is safe provided sequence
117 	   spaces do not overlap i.e. at data rates <= 80Mbit/sec.
118 
119 	   Actually, the idea is close to VJ's one, only timestamp cache is
120 	   held not per host, but per port pair and TW bucket is used as state
121 	   holder.
122 
123 	   If TW bucket has been already destroyed we fall back to VJ's scheme
124 	   and use initial timestamp retrieved from peer table.
125 	 */
126 	if (tcptw->tw_ts_recent_stamp &&
127 	    (twp == NULL || (sysctl_tcp_tw_reuse &&
128 			     get_seconds() - tcptw->tw_ts_recent_stamp > 1))) {
129 		tp->write_seq = tcptw->tw_snd_nxt + 65535 + 2;
130 		if (tp->write_seq == 0)
131 			tp->write_seq = 1;
132 		tp->rx_opt.ts_recent	   = tcptw->tw_ts_recent;
133 		tp->rx_opt.ts_recent_stamp = tcptw->tw_ts_recent_stamp;
134 		sock_hold(sktw);
135 		return 1;
136 	}
137 
138 	return 0;
139 }
140 EXPORT_SYMBOL_GPL(tcp_twsk_unique);
141 
142 /* This will initiate an outgoing connection. */
143 int tcp_v4_connect(struct sock *sk, struct sockaddr *uaddr, int addr_len)
144 {
145 	struct sockaddr_in *usin = (struct sockaddr_in *)uaddr;
146 	struct inet_sock *inet = inet_sk(sk);
147 	struct tcp_sock *tp = tcp_sk(sk);
148 	__be16 orig_sport, orig_dport;
149 	__be32 daddr, nexthop;
150 	struct flowi4 *fl4;
151 	struct rtable *rt;
152 	int err;
153 	struct ip_options_rcu *inet_opt;
154 
155 	if (addr_len < sizeof(struct sockaddr_in))
156 		return -EINVAL;
157 
158 	if (usin->sin_family != AF_INET)
159 		return -EAFNOSUPPORT;
160 
161 	nexthop = daddr = usin->sin_addr.s_addr;
162 	inet_opt = rcu_dereference_protected(inet->inet_opt,
163 					     sock_owned_by_user(sk));
164 	if (inet_opt && inet_opt->opt.srr) {
165 		if (!daddr)
166 			return -EINVAL;
167 		nexthop = inet_opt->opt.faddr;
168 	}
169 
170 	orig_sport = inet->inet_sport;
171 	orig_dport = usin->sin_port;
172 	fl4 = &inet->cork.fl.u.ip4;
173 	rt = ip_route_connect(fl4, nexthop, inet->inet_saddr,
174 			      RT_CONN_FLAGS(sk), sk->sk_bound_dev_if,
175 			      IPPROTO_TCP,
176 			      orig_sport, orig_dport, sk, true);
177 	if (IS_ERR(rt)) {
178 		err = PTR_ERR(rt);
179 		if (err == -ENETUNREACH)
180 			IP_INC_STATS(sock_net(sk), IPSTATS_MIB_OUTNOROUTES);
181 		return err;
182 	}
183 
184 	if (rt->rt_flags & (RTCF_MULTICAST | RTCF_BROADCAST)) {
185 		ip_rt_put(rt);
186 		return -ENETUNREACH;
187 	}
188 
189 	if (!inet_opt || !inet_opt->opt.srr)
190 		daddr = fl4->daddr;
191 
192 	if (!inet->inet_saddr)
193 		inet->inet_saddr = fl4->saddr;
194 	inet->inet_rcv_saddr = inet->inet_saddr;
195 
196 	if (tp->rx_opt.ts_recent_stamp && inet->inet_daddr != daddr) {
197 		/* Reset inherited state */
198 		tp->rx_opt.ts_recent	   = 0;
199 		tp->rx_opt.ts_recent_stamp = 0;
200 		if (likely(!tp->repair))
201 			tp->write_seq	   = 0;
202 	}
203 
204 	if (tcp_death_row.sysctl_tw_recycle &&
205 	    !tp->rx_opt.ts_recent_stamp && fl4->daddr == daddr)
206 		tcp_fetch_timewait_stamp(sk, &rt->dst);
207 
208 	inet->inet_dport = usin->sin_port;
209 	inet->inet_daddr = daddr;
210 
211 	inet_csk(sk)->icsk_ext_hdr_len = 0;
212 	if (inet_opt)
213 		inet_csk(sk)->icsk_ext_hdr_len = inet_opt->opt.optlen;
214 
215 	tp->rx_opt.mss_clamp = TCP_MSS_DEFAULT;
216 
217 	/* Socket identity is still unknown (sport may be zero).
218 	 * However we set state to SYN-SENT and not releasing socket
219 	 * lock select source port, enter ourselves into the hash tables and
220 	 * complete initialization after this.
221 	 */
222 	tcp_set_state(sk, TCP_SYN_SENT);
223 	err = inet_hash_connect(&tcp_death_row, sk);
224 	if (err)
225 		goto failure;
226 
227 	rt = ip_route_newports(fl4, rt, orig_sport, orig_dport,
228 			       inet->inet_sport, inet->inet_dport, sk);
229 	if (IS_ERR(rt)) {
230 		err = PTR_ERR(rt);
231 		rt = NULL;
232 		goto failure;
233 	}
234 	/* OK, now commit destination to socket.  */
235 	sk->sk_gso_type = SKB_GSO_TCPV4;
236 	sk_setup_caps(sk, &rt->dst);
237 
238 	if (!tp->write_seq && likely(!tp->repair))
239 		tp->write_seq = secure_tcp_sequence_number(inet->inet_saddr,
240 							   inet->inet_daddr,
241 							   inet->inet_sport,
242 							   usin->sin_port);
243 
244 	inet->inet_id = tp->write_seq ^ jiffies;
245 
246 	err = tcp_connect(sk);
247 
248 	rt = NULL;
249 	if (err)
250 		goto failure;
251 
252 	return 0;
253 
254 failure:
255 	/*
256 	 * This unhashes the socket and releases the local port,
257 	 * if necessary.
258 	 */
259 	tcp_set_state(sk, TCP_CLOSE);
260 	ip_rt_put(rt);
261 	sk->sk_route_caps = 0;
262 	inet->inet_dport = 0;
263 	return err;
264 }
265 EXPORT_SYMBOL(tcp_v4_connect);
266 
267 /*
268  * This routine reacts to ICMP_FRAG_NEEDED mtu indications as defined in RFC1191.
269  * It can be called through tcp_release_cb() if socket was owned by user
270  * at the time tcp_v4_err() was called to handle ICMP message.
271  */
272 static void tcp_v4_mtu_reduced(struct sock *sk)
273 {
274 	struct dst_entry *dst;
275 	struct inet_sock *inet = inet_sk(sk);
276 	u32 mtu = tcp_sk(sk)->mtu_info;
277 
278 	dst = inet_csk_update_pmtu(sk, mtu);
279 	if (!dst)
280 		return;
281 
282 	/* Something is about to be wrong... Remember soft error
283 	 * for the case, if this connection will not able to recover.
284 	 */
285 	if (mtu < dst_mtu(dst) && ip_dont_fragment(sk, dst))
286 		sk->sk_err_soft = EMSGSIZE;
287 
288 	mtu = dst_mtu(dst);
289 
290 	if (inet->pmtudisc != IP_PMTUDISC_DONT &&
291 	    ip_sk_accept_pmtu(sk) &&
292 	    inet_csk(sk)->icsk_pmtu_cookie > mtu) {
293 		tcp_sync_mss(sk, mtu);
294 
295 		/* Resend the TCP packet because it's
296 		 * clear that the old packet has been
297 		 * dropped. This is the new "fast" path mtu
298 		 * discovery.
299 		 */
300 		tcp_simple_retransmit(sk);
301 	} /* else let the usual retransmit timer handle it */
302 }
303 
304 static void do_redirect(struct sk_buff *skb, struct sock *sk)
305 {
306 	struct dst_entry *dst = __sk_dst_check(sk, 0);
307 
308 	if (dst)
309 		dst->ops->redirect(dst, sk, skb);
310 }
311 
312 /*
313  * This routine is called by the ICMP module when it gets some
314  * sort of error condition.  If err < 0 then the socket should
315  * be closed and the error returned to the user.  If err > 0
316  * it's just the icmp type << 8 | icmp code.  After adjustment
317  * header points to the first 8 bytes of the tcp header.  We need
318  * to find the appropriate port.
319  *
320  * The locking strategy used here is very "optimistic". When
321  * someone else accesses the socket the ICMP is just dropped
322  * and for some paths there is no check at all.
323  * A more general error queue to queue errors for later handling
324  * is probably better.
325  *
326  */
327 
328 void tcp_v4_err(struct sk_buff *icmp_skb, u32 info)
329 {
330 	const struct iphdr *iph = (const struct iphdr *)icmp_skb->data;
331 	struct tcphdr *th = (struct tcphdr *)(icmp_skb->data + (iph->ihl << 2));
332 	struct inet_connection_sock *icsk;
333 	struct tcp_sock *tp;
334 	struct inet_sock *inet;
335 	const int type = icmp_hdr(icmp_skb)->type;
336 	const int code = icmp_hdr(icmp_skb)->code;
337 	struct sock *sk;
338 	struct sk_buff *skb;
339 	struct request_sock *req;
340 	__u32 seq;
341 	__u32 remaining;
342 	int err;
343 	struct net *net = dev_net(icmp_skb->dev);
344 
345 	if (icmp_skb->len < (iph->ihl << 2) + 8) {
346 		ICMP_INC_STATS_BH(net, ICMP_MIB_INERRORS);
347 		return;
348 	}
349 
350 	sk = inet_lookup(net, &tcp_hashinfo, iph->daddr, th->dest,
351 			iph->saddr, th->source, inet_iif(icmp_skb));
352 	if (!sk) {
353 		ICMP_INC_STATS_BH(net, ICMP_MIB_INERRORS);
354 		return;
355 	}
356 	if (sk->sk_state == TCP_TIME_WAIT) {
357 		inet_twsk_put(inet_twsk(sk));
358 		return;
359 	}
360 
361 	bh_lock_sock(sk);
362 	/* If too many ICMPs get dropped on busy
363 	 * servers this needs to be solved differently.
364 	 * We do take care of PMTU discovery (RFC1191) special case :
365 	 * we can receive locally generated ICMP messages while socket is held.
366 	 */
367 	if (sock_owned_by_user(sk)) {
368 		if (!(type == ICMP_DEST_UNREACH && code == ICMP_FRAG_NEEDED))
369 			NET_INC_STATS_BH(net, LINUX_MIB_LOCKDROPPEDICMPS);
370 	}
371 	if (sk->sk_state == TCP_CLOSE)
372 		goto out;
373 
374 	if (unlikely(iph->ttl < inet_sk(sk)->min_ttl)) {
375 		NET_INC_STATS_BH(net, LINUX_MIB_TCPMINTTLDROP);
376 		goto out;
377 	}
378 
379 	icsk = inet_csk(sk);
380 	tp = tcp_sk(sk);
381 	req = tp->fastopen_rsk;
382 	seq = ntohl(th->seq);
383 	if (sk->sk_state != TCP_LISTEN &&
384 	    !between(seq, tp->snd_una, tp->snd_nxt) &&
385 	    (req == NULL || seq != tcp_rsk(req)->snt_isn)) {
386 		/* For a Fast Open socket, allow seq to be snt_isn. */
387 		NET_INC_STATS_BH(net, LINUX_MIB_OUTOFWINDOWICMPS);
388 		goto out;
389 	}
390 
391 	switch (type) {
392 	case ICMP_REDIRECT:
393 		do_redirect(icmp_skb, sk);
394 		goto out;
395 	case ICMP_SOURCE_QUENCH:
396 		/* Just silently ignore these. */
397 		goto out;
398 	case ICMP_PARAMETERPROB:
399 		err = EPROTO;
400 		break;
401 	case ICMP_DEST_UNREACH:
402 		if (code > NR_ICMP_UNREACH)
403 			goto out;
404 
405 		if (code == ICMP_FRAG_NEEDED) { /* PMTU discovery (RFC1191) */
406 			/* We are not interested in TCP_LISTEN and open_requests
407 			 * (SYN-ACKs send out by Linux are always <576bytes so
408 			 * they should go through unfragmented).
409 			 */
410 			if (sk->sk_state == TCP_LISTEN)
411 				goto out;
412 
413 			tp->mtu_info = info;
414 			if (!sock_owned_by_user(sk)) {
415 				tcp_v4_mtu_reduced(sk);
416 			} else {
417 				if (!test_and_set_bit(TCP_MTU_REDUCED_DEFERRED, &tp->tsq_flags))
418 					sock_hold(sk);
419 			}
420 			goto out;
421 		}
422 
423 		err = icmp_err_convert[code].errno;
424 		/* check if icmp_skb allows revert of backoff
425 		 * (see draft-zimmermann-tcp-lcd) */
426 		if (code != ICMP_NET_UNREACH && code != ICMP_HOST_UNREACH)
427 			break;
428 		if (seq != tp->snd_una  || !icsk->icsk_retransmits ||
429 		    !icsk->icsk_backoff)
430 			break;
431 
432 		/* XXX (TFO) - revisit the following logic for TFO */
433 
434 		if (sock_owned_by_user(sk))
435 			break;
436 
437 		icsk->icsk_backoff--;
438 		inet_csk(sk)->icsk_rto = (tp->srtt ? __tcp_set_rto(tp) :
439 			TCP_TIMEOUT_INIT) << icsk->icsk_backoff;
440 		tcp_bound_rto(sk);
441 
442 		skb = tcp_write_queue_head(sk);
443 		BUG_ON(!skb);
444 
445 		remaining = icsk->icsk_rto - min(icsk->icsk_rto,
446 				tcp_time_stamp - TCP_SKB_CB(skb)->when);
447 
448 		if (remaining) {
449 			inet_csk_reset_xmit_timer(sk, ICSK_TIME_RETRANS,
450 						  remaining, TCP_RTO_MAX);
451 		} else {
452 			/* RTO revert clocked out retransmission.
453 			 * Will retransmit now */
454 			tcp_retransmit_timer(sk);
455 		}
456 
457 		break;
458 	case ICMP_TIME_EXCEEDED:
459 		err = EHOSTUNREACH;
460 		break;
461 	default:
462 		goto out;
463 	}
464 
465 	/* XXX (TFO) - if it's a TFO socket and has been accepted, rather
466 	 * than following the TCP_SYN_RECV case and closing the socket,
467 	 * we ignore the ICMP error and keep trying like a fully established
468 	 * socket. Is this the right thing to do?
469 	 */
470 	if (req && req->sk == NULL)
471 		goto out;
472 
473 	switch (sk->sk_state) {
474 		struct request_sock *req, **prev;
475 	case TCP_LISTEN:
476 		if (sock_owned_by_user(sk))
477 			goto out;
478 
479 		req = inet_csk_search_req(sk, &prev, th->dest,
480 					  iph->daddr, iph->saddr);
481 		if (!req)
482 			goto out;
483 
484 		/* ICMPs are not backlogged, hence we cannot get
485 		   an established socket here.
486 		 */
487 		WARN_ON(req->sk);
488 
489 		if (seq != tcp_rsk(req)->snt_isn) {
490 			NET_INC_STATS_BH(net, LINUX_MIB_OUTOFWINDOWICMPS);
491 			goto out;
492 		}
493 
494 		/*
495 		 * Still in SYN_RECV, just remove it silently.
496 		 * There is no good way to pass the error to the newly
497 		 * created socket, and POSIX does not want network
498 		 * errors returned from accept().
499 		 */
500 		inet_csk_reqsk_queue_drop(sk, req, prev);
501 		NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_LISTENDROPS);
502 		goto out;
503 
504 	case TCP_SYN_SENT:
505 	case TCP_SYN_RECV:  /* Cannot happen.
506 			       It can f.e. if SYNs crossed,
507 			       or Fast Open.
508 			     */
509 		if (!sock_owned_by_user(sk)) {
510 			sk->sk_err = err;
511 
512 			sk->sk_error_report(sk);
513 
514 			tcp_done(sk);
515 		} else {
516 			sk->sk_err_soft = err;
517 		}
518 		goto out;
519 	}
520 
521 	/* If we've already connected we will keep trying
522 	 * until we time out, or the user gives up.
523 	 *
524 	 * rfc1122 4.2.3.9 allows to consider as hard errors
525 	 * only PROTO_UNREACH and PORT_UNREACH (well, FRAG_FAILED too,
526 	 * but it is obsoleted by pmtu discovery).
527 	 *
528 	 * Note, that in modern internet, where routing is unreliable
529 	 * and in each dark corner broken firewalls sit, sending random
530 	 * errors ordered by their masters even this two messages finally lose
531 	 * their original sense (even Linux sends invalid PORT_UNREACHs)
532 	 *
533 	 * Now we are in compliance with RFCs.
534 	 *							--ANK (980905)
535 	 */
536 
537 	inet = inet_sk(sk);
538 	if (!sock_owned_by_user(sk) && inet->recverr) {
539 		sk->sk_err = err;
540 		sk->sk_error_report(sk);
541 	} else	{ /* Only an error on timeout */
542 		sk->sk_err_soft = err;
543 	}
544 
545 out:
546 	bh_unlock_sock(sk);
547 	sock_put(sk);
548 }
549 
550 void __tcp_v4_send_check(struct sk_buff *skb, __be32 saddr, __be32 daddr)
551 {
552 	struct tcphdr *th = tcp_hdr(skb);
553 
554 	if (skb->ip_summed == CHECKSUM_PARTIAL) {
555 		th->check = ~tcp_v4_check(skb->len, saddr, daddr, 0);
556 		skb->csum_start = skb_transport_header(skb) - skb->head;
557 		skb->csum_offset = offsetof(struct tcphdr, check);
558 	} else {
559 		th->check = tcp_v4_check(skb->len, saddr, daddr,
560 					 csum_partial(th,
561 						      th->doff << 2,
562 						      skb->csum));
563 	}
564 }
565 
566 /* This routine computes an IPv4 TCP checksum. */
567 void tcp_v4_send_check(struct sock *sk, struct sk_buff *skb)
568 {
569 	const struct inet_sock *inet = inet_sk(sk);
570 
571 	__tcp_v4_send_check(skb, inet->inet_saddr, inet->inet_daddr);
572 }
573 EXPORT_SYMBOL(tcp_v4_send_check);
574 
575 /*
576  *	This routine will send an RST to the other tcp.
577  *
578  *	Someone asks: why I NEVER use socket parameters (TOS, TTL etc.)
579  *		      for reset.
580  *	Answer: if a packet caused RST, it is not for a socket
581  *		existing in our system, if it is matched to a socket,
582  *		it is just duplicate segment or bug in other side's TCP.
583  *		So that we build reply only basing on parameters
584  *		arrived with segment.
585  *	Exception: precedence violation. We do not implement it in any case.
586  */
587 
588 static void tcp_v4_send_reset(struct sock *sk, struct sk_buff *skb)
589 {
590 	const struct tcphdr *th = tcp_hdr(skb);
591 	struct {
592 		struct tcphdr th;
593 #ifdef CONFIG_TCP_MD5SIG
594 		__be32 opt[(TCPOLEN_MD5SIG_ALIGNED >> 2)];
595 #endif
596 	} rep;
597 	struct ip_reply_arg arg;
598 #ifdef CONFIG_TCP_MD5SIG
599 	struct tcp_md5sig_key *key;
600 	const __u8 *hash_location = NULL;
601 	unsigned char newhash[16];
602 	int genhash;
603 	struct sock *sk1 = NULL;
604 #endif
605 	struct net *net;
606 
607 	/* Never send a reset in response to a reset. */
608 	if (th->rst)
609 		return;
610 
611 	if (skb_rtable(skb)->rt_type != RTN_LOCAL)
612 		return;
613 
614 	/* Swap the send and the receive. */
615 	memset(&rep, 0, sizeof(rep));
616 	rep.th.dest   = th->source;
617 	rep.th.source = th->dest;
618 	rep.th.doff   = sizeof(struct tcphdr) / 4;
619 	rep.th.rst    = 1;
620 
621 	if (th->ack) {
622 		rep.th.seq = th->ack_seq;
623 	} else {
624 		rep.th.ack = 1;
625 		rep.th.ack_seq = htonl(ntohl(th->seq) + th->syn + th->fin +
626 				       skb->len - (th->doff << 2));
627 	}
628 
629 	memset(&arg, 0, sizeof(arg));
630 	arg.iov[0].iov_base = (unsigned char *)&rep;
631 	arg.iov[0].iov_len  = sizeof(rep.th);
632 
633 #ifdef CONFIG_TCP_MD5SIG
634 	hash_location = tcp_parse_md5sig_option(th);
635 	if (!sk && hash_location) {
636 		/*
637 		 * active side is lost. Try to find listening socket through
638 		 * source port, and then find md5 key through listening socket.
639 		 * we are not loose security here:
640 		 * Incoming packet is checked with md5 hash with finding key,
641 		 * no RST generated if md5 hash doesn't match.
642 		 */
643 		sk1 = __inet_lookup_listener(dev_net(skb_dst(skb)->dev),
644 					     &tcp_hashinfo, ip_hdr(skb)->saddr,
645 					     th->source, ip_hdr(skb)->daddr,
646 					     ntohs(th->source), inet_iif(skb));
647 		/* don't send rst if it can't find key */
648 		if (!sk1)
649 			return;
650 		rcu_read_lock();
651 		key = tcp_md5_do_lookup(sk1, (union tcp_md5_addr *)
652 					&ip_hdr(skb)->saddr, AF_INET);
653 		if (!key)
654 			goto release_sk1;
655 
656 		genhash = tcp_v4_md5_hash_skb(newhash, key, NULL, NULL, skb);
657 		if (genhash || memcmp(hash_location, newhash, 16) != 0)
658 			goto release_sk1;
659 	} else {
660 		key = sk ? tcp_md5_do_lookup(sk, (union tcp_md5_addr *)
661 					     &ip_hdr(skb)->saddr,
662 					     AF_INET) : NULL;
663 	}
664 
665 	if (key) {
666 		rep.opt[0] = htonl((TCPOPT_NOP << 24) |
667 				   (TCPOPT_NOP << 16) |
668 				   (TCPOPT_MD5SIG << 8) |
669 				   TCPOLEN_MD5SIG);
670 		/* Update length and the length the header thinks exists */
671 		arg.iov[0].iov_len += TCPOLEN_MD5SIG_ALIGNED;
672 		rep.th.doff = arg.iov[0].iov_len / 4;
673 
674 		tcp_v4_md5_hash_hdr((__u8 *) &rep.opt[1],
675 				     key, ip_hdr(skb)->saddr,
676 				     ip_hdr(skb)->daddr, &rep.th);
677 	}
678 #endif
679 	arg.csum = csum_tcpudp_nofold(ip_hdr(skb)->daddr,
680 				      ip_hdr(skb)->saddr, /* XXX */
681 				      arg.iov[0].iov_len, IPPROTO_TCP, 0);
682 	arg.csumoffset = offsetof(struct tcphdr, check) / 2;
683 	arg.flags = (sk && inet_sk(sk)->transparent) ? IP_REPLY_ARG_NOSRCCHECK : 0;
684 	/* When socket is gone, all binding information is lost.
685 	 * routing might fail in this case. No choice here, if we choose to force
686 	 * input interface, we will misroute in case of asymmetric route.
687 	 */
688 	if (sk)
689 		arg.bound_dev_if = sk->sk_bound_dev_if;
690 
691 	net = dev_net(skb_dst(skb)->dev);
692 	arg.tos = ip_hdr(skb)->tos;
693 	ip_send_unicast_reply(net, skb, ip_hdr(skb)->saddr,
694 			      ip_hdr(skb)->daddr, &arg, arg.iov[0].iov_len);
695 
696 	TCP_INC_STATS_BH(net, TCP_MIB_OUTSEGS);
697 	TCP_INC_STATS_BH(net, TCP_MIB_OUTRSTS);
698 
699 #ifdef CONFIG_TCP_MD5SIG
700 release_sk1:
701 	if (sk1) {
702 		rcu_read_unlock();
703 		sock_put(sk1);
704 	}
705 #endif
706 }
707 
708 /* The code following below sending ACKs in SYN-RECV and TIME-WAIT states
709    outside socket context is ugly, certainly. What can I do?
710  */
711 
712 static void tcp_v4_send_ack(struct sk_buff *skb, u32 seq, u32 ack,
713 			    u32 win, u32 tsval, u32 tsecr, int oif,
714 			    struct tcp_md5sig_key *key,
715 			    int reply_flags, u8 tos)
716 {
717 	const struct tcphdr *th = tcp_hdr(skb);
718 	struct {
719 		struct tcphdr th;
720 		__be32 opt[(TCPOLEN_TSTAMP_ALIGNED >> 2)
721 #ifdef CONFIG_TCP_MD5SIG
722 			   + (TCPOLEN_MD5SIG_ALIGNED >> 2)
723 #endif
724 			];
725 	} rep;
726 	struct ip_reply_arg arg;
727 	struct net *net = dev_net(skb_dst(skb)->dev);
728 
729 	memset(&rep.th, 0, sizeof(struct tcphdr));
730 	memset(&arg, 0, sizeof(arg));
731 
732 	arg.iov[0].iov_base = (unsigned char *)&rep;
733 	arg.iov[0].iov_len  = sizeof(rep.th);
734 	if (tsecr) {
735 		rep.opt[0] = htonl((TCPOPT_NOP << 24) | (TCPOPT_NOP << 16) |
736 				   (TCPOPT_TIMESTAMP << 8) |
737 				   TCPOLEN_TIMESTAMP);
738 		rep.opt[1] = htonl(tsval);
739 		rep.opt[2] = htonl(tsecr);
740 		arg.iov[0].iov_len += TCPOLEN_TSTAMP_ALIGNED;
741 	}
742 
743 	/* Swap the send and the receive. */
744 	rep.th.dest    = th->source;
745 	rep.th.source  = th->dest;
746 	rep.th.doff    = arg.iov[0].iov_len / 4;
747 	rep.th.seq     = htonl(seq);
748 	rep.th.ack_seq = htonl(ack);
749 	rep.th.ack     = 1;
750 	rep.th.window  = htons(win);
751 
752 #ifdef CONFIG_TCP_MD5SIG
753 	if (key) {
754 		int offset = (tsecr) ? 3 : 0;
755 
756 		rep.opt[offset++] = htonl((TCPOPT_NOP << 24) |
757 					  (TCPOPT_NOP << 16) |
758 					  (TCPOPT_MD5SIG << 8) |
759 					  TCPOLEN_MD5SIG);
760 		arg.iov[0].iov_len += TCPOLEN_MD5SIG_ALIGNED;
761 		rep.th.doff = arg.iov[0].iov_len/4;
762 
763 		tcp_v4_md5_hash_hdr((__u8 *) &rep.opt[offset],
764 				    key, ip_hdr(skb)->saddr,
765 				    ip_hdr(skb)->daddr, &rep.th);
766 	}
767 #endif
768 	arg.flags = reply_flags;
769 	arg.csum = csum_tcpudp_nofold(ip_hdr(skb)->daddr,
770 				      ip_hdr(skb)->saddr, /* XXX */
771 				      arg.iov[0].iov_len, IPPROTO_TCP, 0);
772 	arg.csumoffset = offsetof(struct tcphdr, check) / 2;
773 	if (oif)
774 		arg.bound_dev_if = oif;
775 	arg.tos = tos;
776 	ip_send_unicast_reply(net, skb, ip_hdr(skb)->saddr,
777 			      ip_hdr(skb)->daddr, &arg, arg.iov[0].iov_len);
778 
779 	TCP_INC_STATS_BH(net, TCP_MIB_OUTSEGS);
780 }
781 
782 static void tcp_v4_timewait_ack(struct sock *sk, struct sk_buff *skb)
783 {
784 	struct inet_timewait_sock *tw = inet_twsk(sk);
785 	struct tcp_timewait_sock *tcptw = tcp_twsk(sk);
786 
787 	tcp_v4_send_ack(skb, tcptw->tw_snd_nxt, tcptw->tw_rcv_nxt,
788 			tcptw->tw_rcv_wnd >> tw->tw_rcv_wscale,
789 			tcp_time_stamp + tcptw->tw_ts_offset,
790 			tcptw->tw_ts_recent,
791 			tw->tw_bound_dev_if,
792 			tcp_twsk_md5_key(tcptw),
793 			tw->tw_transparent ? IP_REPLY_ARG_NOSRCCHECK : 0,
794 			tw->tw_tos
795 			);
796 
797 	inet_twsk_put(tw);
798 }
799 
800 static void tcp_v4_reqsk_send_ack(struct sock *sk, struct sk_buff *skb,
801 				  struct request_sock *req)
802 {
803 	/* sk->sk_state == TCP_LISTEN -> for regular TCP_SYN_RECV
804 	 * sk->sk_state == TCP_SYN_RECV -> for Fast Open.
805 	 */
806 	tcp_v4_send_ack(skb, (sk->sk_state == TCP_LISTEN) ?
807 			tcp_rsk(req)->snt_isn + 1 : tcp_sk(sk)->snd_nxt,
808 			tcp_rsk(req)->rcv_nxt, req->rcv_wnd,
809 			tcp_time_stamp,
810 			req->ts_recent,
811 			0,
812 			tcp_md5_do_lookup(sk, (union tcp_md5_addr *)&ip_hdr(skb)->daddr,
813 					  AF_INET),
814 			inet_rsk(req)->no_srccheck ? IP_REPLY_ARG_NOSRCCHECK : 0,
815 			ip_hdr(skb)->tos);
816 }
817 
818 /*
819  *	Send a SYN-ACK after having received a SYN.
820  *	This still operates on a request_sock only, not on a big
821  *	socket.
822  */
823 static int tcp_v4_send_synack(struct sock *sk, struct dst_entry *dst,
824 			      struct request_sock *req,
825 			      u16 queue_mapping)
826 {
827 	const struct inet_request_sock *ireq = inet_rsk(req);
828 	struct flowi4 fl4;
829 	int err = -1;
830 	struct sk_buff * skb;
831 
832 	/* First, grab a route. */
833 	if (!dst && (dst = inet_csk_route_req(sk, &fl4, req)) == NULL)
834 		return -1;
835 
836 	skb = tcp_make_synack(sk, dst, req, NULL);
837 
838 	if (skb) {
839 		__tcp_v4_send_check(skb, ireq->ir_loc_addr, ireq->ir_rmt_addr);
840 
841 		skb_set_queue_mapping(skb, queue_mapping);
842 		err = ip_build_and_send_pkt(skb, sk, ireq->ir_loc_addr,
843 					    ireq->ir_rmt_addr,
844 					    ireq->opt);
845 		err = net_xmit_eval(err);
846 		if (!tcp_rsk(req)->snt_synack && !err)
847 			tcp_rsk(req)->snt_synack = tcp_time_stamp;
848 	}
849 
850 	return err;
851 }
852 
853 static int tcp_v4_rtx_synack(struct sock *sk, struct request_sock *req)
854 {
855 	int res = tcp_v4_send_synack(sk, NULL, req, 0);
856 
857 	if (!res)
858 		TCP_INC_STATS_BH(sock_net(sk), TCP_MIB_RETRANSSEGS);
859 	return res;
860 }
861 
862 /*
863  *	IPv4 request_sock destructor.
864  */
865 static void tcp_v4_reqsk_destructor(struct request_sock *req)
866 {
867 	kfree(inet_rsk(req)->opt);
868 }
869 
870 /*
871  * Return true if a syncookie should be sent
872  */
873 bool tcp_syn_flood_action(struct sock *sk,
874 			 const struct sk_buff *skb,
875 			 const char *proto)
876 {
877 	const char *msg = "Dropping request";
878 	bool want_cookie = false;
879 	struct listen_sock *lopt;
880 
881 
882 
883 #ifdef CONFIG_SYN_COOKIES
884 	if (sysctl_tcp_syncookies) {
885 		msg = "Sending cookies";
886 		want_cookie = true;
887 		NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_TCPREQQFULLDOCOOKIES);
888 	} else
889 #endif
890 		NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_TCPREQQFULLDROP);
891 
892 	lopt = inet_csk(sk)->icsk_accept_queue.listen_opt;
893 	if (!lopt->synflood_warned && sysctl_tcp_syncookies != 2) {
894 		lopt->synflood_warned = 1;
895 		pr_info("%s: Possible SYN flooding on port %d. %s.  Check SNMP counters.\n",
896 			proto, ntohs(tcp_hdr(skb)->dest), msg);
897 	}
898 	return want_cookie;
899 }
900 EXPORT_SYMBOL(tcp_syn_flood_action);
901 
902 /*
903  * Save and compile IPv4 options into the request_sock if needed.
904  */
905 static struct ip_options_rcu *tcp_v4_save_options(struct sk_buff *skb)
906 {
907 	const struct ip_options *opt = &(IPCB(skb)->opt);
908 	struct ip_options_rcu *dopt = NULL;
909 
910 	if (opt && opt->optlen) {
911 		int opt_size = sizeof(*dopt) + opt->optlen;
912 
913 		dopt = kmalloc(opt_size, GFP_ATOMIC);
914 		if (dopt) {
915 			if (ip_options_echo(&dopt->opt, skb)) {
916 				kfree(dopt);
917 				dopt = NULL;
918 			}
919 		}
920 	}
921 	return dopt;
922 }
923 
924 #ifdef CONFIG_TCP_MD5SIG
925 /*
926  * RFC2385 MD5 checksumming requires a mapping of
927  * IP address->MD5 Key.
928  * We need to maintain these in the sk structure.
929  */
930 
931 /* Find the Key structure for an address.  */
932 struct tcp_md5sig_key *tcp_md5_do_lookup(struct sock *sk,
933 					 const union tcp_md5_addr *addr,
934 					 int family)
935 {
936 	struct tcp_sock *tp = tcp_sk(sk);
937 	struct tcp_md5sig_key *key;
938 	unsigned int size = sizeof(struct in_addr);
939 	struct tcp_md5sig_info *md5sig;
940 
941 	/* caller either holds rcu_read_lock() or socket lock */
942 	md5sig = rcu_dereference_check(tp->md5sig_info,
943 				       sock_owned_by_user(sk) ||
944 				       lockdep_is_held(&sk->sk_lock.slock));
945 	if (!md5sig)
946 		return NULL;
947 #if IS_ENABLED(CONFIG_IPV6)
948 	if (family == AF_INET6)
949 		size = sizeof(struct in6_addr);
950 #endif
951 	hlist_for_each_entry_rcu(key, &md5sig->head, node) {
952 		if (key->family != family)
953 			continue;
954 		if (!memcmp(&key->addr, addr, size))
955 			return key;
956 	}
957 	return NULL;
958 }
959 EXPORT_SYMBOL(tcp_md5_do_lookup);
960 
961 struct tcp_md5sig_key *tcp_v4_md5_lookup(struct sock *sk,
962 					 struct sock *addr_sk)
963 {
964 	union tcp_md5_addr *addr;
965 
966 	addr = (union tcp_md5_addr *)&inet_sk(addr_sk)->inet_daddr;
967 	return tcp_md5_do_lookup(sk, addr, AF_INET);
968 }
969 EXPORT_SYMBOL(tcp_v4_md5_lookup);
970 
971 static struct tcp_md5sig_key *tcp_v4_reqsk_md5_lookup(struct sock *sk,
972 						      struct request_sock *req)
973 {
974 	union tcp_md5_addr *addr;
975 
976 	addr = (union tcp_md5_addr *)&inet_rsk(req)->ir_rmt_addr;
977 	return tcp_md5_do_lookup(sk, addr, AF_INET);
978 }
979 
980 /* This can be called on a newly created socket, from other files */
981 int tcp_md5_do_add(struct sock *sk, const union tcp_md5_addr *addr,
982 		   int family, const u8 *newkey, u8 newkeylen, gfp_t gfp)
983 {
984 	/* Add Key to the list */
985 	struct tcp_md5sig_key *key;
986 	struct tcp_sock *tp = tcp_sk(sk);
987 	struct tcp_md5sig_info *md5sig;
988 
989 	key = tcp_md5_do_lookup(sk, addr, family);
990 	if (key) {
991 		/* Pre-existing entry - just update that one. */
992 		memcpy(key->key, newkey, newkeylen);
993 		key->keylen = newkeylen;
994 		return 0;
995 	}
996 
997 	md5sig = rcu_dereference_protected(tp->md5sig_info,
998 					   sock_owned_by_user(sk));
999 	if (!md5sig) {
1000 		md5sig = kmalloc(sizeof(*md5sig), gfp);
1001 		if (!md5sig)
1002 			return -ENOMEM;
1003 
1004 		sk_nocaps_add(sk, NETIF_F_GSO_MASK);
1005 		INIT_HLIST_HEAD(&md5sig->head);
1006 		rcu_assign_pointer(tp->md5sig_info, md5sig);
1007 	}
1008 
1009 	key = sock_kmalloc(sk, sizeof(*key), gfp);
1010 	if (!key)
1011 		return -ENOMEM;
1012 	if (!tcp_alloc_md5sig_pool()) {
1013 		sock_kfree_s(sk, key, sizeof(*key));
1014 		return -ENOMEM;
1015 	}
1016 
1017 	memcpy(key->key, newkey, newkeylen);
1018 	key->keylen = newkeylen;
1019 	key->family = family;
1020 	memcpy(&key->addr, addr,
1021 	       (family == AF_INET6) ? sizeof(struct in6_addr) :
1022 				      sizeof(struct in_addr));
1023 	hlist_add_head_rcu(&key->node, &md5sig->head);
1024 	return 0;
1025 }
1026 EXPORT_SYMBOL(tcp_md5_do_add);
1027 
1028 int tcp_md5_do_del(struct sock *sk, const union tcp_md5_addr *addr, int family)
1029 {
1030 	struct tcp_md5sig_key *key;
1031 
1032 	key = tcp_md5_do_lookup(sk, addr, family);
1033 	if (!key)
1034 		return -ENOENT;
1035 	hlist_del_rcu(&key->node);
1036 	atomic_sub(sizeof(*key), &sk->sk_omem_alloc);
1037 	kfree_rcu(key, rcu);
1038 	return 0;
1039 }
1040 EXPORT_SYMBOL(tcp_md5_do_del);
1041 
1042 static void tcp_clear_md5_list(struct sock *sk)
1043 {
1044 	struct tcp_sock *tp = tcp_sk(sk);
1045 	struct tcp_md5sig_key *key;
1046 	struct hlist_node *n;
1047 	struct tcp_md5sig_info *md5sig;
1048 
1049 	md5sig = rcu_dereference_protected(tp->md5sig_info, 1);
1050 
1051 	hlist_for_each_entry_safe(key, n, &md5sig->head, node) {
1052 		hlist_del_rcu(&key->node);
1053 		atomic_sub(sizeof(*key), &sk->sk_omem_alloc);
1054 		kfree_rcu(key, rcu);
1055 	}
1056 }
1057 
1058 static int tcp_v4_parse_md5_keys(struct sock *sk, char __user *optval,
1059 				 int optlen)
1060 {
1061 	struct tcp_md5sig cmd;
1062 	struct sockaddr_in *sin = (struct sockaddr_in *)&cmd.tcpm_addr;
1063 
1064 	if (optlen < sizeof(cmd))
1065 		return -EINVAL;
1066 
1067 	if (copy_from_user(&cmd, optval, sizeof(cmd)))
1068 		return -EFAULT;
1069 
1070 	if (sin->sin_family != AF_INET)
1071 		return -EINVAL;
1072 
1073 	if (!cmd.tcpm_key || !cmd.tcpm_keylen)
1074 		return tcp_md5_do_del(sk, (union tcp_md5_addr *)&sin->sin_addr.s_addr,
1075 				      AF_INET);
1076 
1077 	if (cmd.tcpm_keylen > TCP_MD5SIG_MAXKEYLEN)
1078 		return -EINVAL;
1079 
1080 	return tcp_md5_do_add(sk, (union tcp_md5_addr *)&sin->sin_addr.s_addr,
1081 			      AF_INET, cmd.tcpm_key, cmd.tcpm_keylen,
1082 			      GFP_KERNEL);
1083 }
1084 
1085 static int tcp_v4_md5_hash_pseudoheader(struct tcp_md5sig_pool *hp,
1086 					__be32 daddr, __be32 saddr, int nbytes)
1087 {
1088 	struct tcp4_pseudohdr *bp;
1089 	struct scatterlist sg;
1090 
1091 	bp = &hp->md5_blk.ip4;
1092 
1093 	/*
1094 	 * 1. the TCP pseudo-header (in the order: source IP address,
1095 	 * destination IP address, zero-padded protocol number, and
1096 	 * segment length)
1097 	 */
1098 	bp->saddr = saddr;
1099 	bp->daddr = daddr;
1100 	bp->pad = 0;
1101 	bp->protocol = IPPROTO_TCP;
1102 	bp->len = cpu_to_be16(nbytes);
1103 
1104 	sg_init_one(&sg, bp, sizeof(*bp));
1105 	return crypto_hash_update(&hp->md5_desc, &sg, sizeof(*bp));
1106 }
1107 
1108 static int tcp_v4_md5_hash_hdr(char *md5_hash, const struct tcp_md5sig_key *key,
1109 			       __be32 daddr, __be32 saddr, const struct tcphdr *th)
1110 {
1111 	struct tcp_md5sig_pool *hp;
1112 	struct hash_desc *desc;
1113 
1114 	hp = tcp_get_md5sig_pool();
1115 	if (!hp)
1116 		goto clear_hash_noput;
1117 	desc = &hp->md5_desc;
1118 
1119 	if (crypto_hash_init(desc))
1120 		goto clear_hash;
1121 	if (tcp_v4_md5_hash_pseudoheader(hp, daddr, saddr, th->doff << 2))
1122 		goto clear_hash;
1123 	if (tcp_md5_hash_header(hp, th))
1124 		goto clear_hash;
1125 	if (tcp_md5_hash_key(hp, key))
1126 		goto clear_hash;
1127 	if (crypto_hash_final(desc, md5_hash))
1128 		goto clear_hash;
1129 
1130 	tcp_put_md5sig_pool();
1131 	return 0;
1132 
1133 clear_hash:
1134 	tcp_put_md5sig_pool();
1135 clear_hash_noput:
1136 	memset(md5_hash, 0, 16);
1137 	return 1;
1138 }
1139 
1140 int tcp_v4_md5_hash_skb(char *md5_hash, struct tcp_md5sig_key *key,
1141 			const struct sock *sk, const struct request_sock *req,
1142 			const struct sk_buff *skb)
1143 {
1144 	struct tcp_md5sig_pool *hp;
1145 	struct hash_desc *desc;
1146 	const struct tcphdr *th = tcp_hdr(skb);
1147 	__be32 saddr, daddr;
1148 
1149 	if (sk) {
1150 		saddr = inet_sk(sk)->inet_saddr;
1151 		daddr = inet_sk(sk)->inet_daddr;
1152 	} else if (req) {
1153 		saddr = inet_rsk(req)->ir_loc_addr;
1154 		daddr = inet_rsk(req)->ir_rmt_addr;
1155 	} else {
1156 		const struct iphdr *iph = ip_hdr(skb);
1157 		saddr = iph->saddr;
1158 		daddr = iph->daddr;
1159 	}
1160 
1161 	hp = tcp_get_md5sig_pool();
1162 	if (!hp)
1163 		goto clear_hash_noput;
1164 	desc = &hp->md5_desc;
1165 
1166 	if (crypto_hash_init(desc))
1167 		goto clear_hash;
1168 
1169 	if (tcp_v4_md5_hash_pseudoheader(hp, daddr, saddr, skb->len))
1170 		goto clear_hash;
1171 	if (tcp_md5_hash_header(hp, th))
1172 		goto clear_hash;
1173 	if (tcp_md5_hash_skb_data(hp, skb, th->doff << 2))
1174 		goto clear_hash;
1175 	if (tcp_md5_hash_key(hp, key))
1176 		goto clear_hash;
1177 	if (crypto_hash_final(desc, md5_hash))
1178 		goto clear_hash;
1179 
1180 	tcp_put_md5sig_pool();
1181 	return 0;
1182 
1183 clear_hash:
1184 	tcp_put_md5sig_pool();
1185 clear_hash_noput:
1186 	memset(md5_hash, 0, 16);
1187 	return 1;
1188 }
1189 EXPORT_SYMBOL(tcp_v4_md5_hash_skb);
1190 
1191 static bool tcp_v4_inbound_md5_hash(struct sock *sk, const struct sk_buff *skb)
1192 {
1193 	/*
1194 	 * This gets called for each TCP segment that arrives
1195 	 * so we want to be efficient.
1196 	 * We have 3 drop cases:
1197 	 * o No MD5 hash and one expected.
1198 	 * o MD5 hash and we're not expecting one.
1199 	 * o MD5 hash and its wrong.
1200 	 */
1201 	const __u8 *hash_location = NULL;
1202 	struct tcp_md5sig_key *hash_expected;
1203 	const struct iphdr *iph = ip_hdr(skb);
1204 	const struct tcphdr *th = tcp_hdr(skb);
1205 	int genhash;
1206 	unsigned char newhash[16];
1207 
1208 	hash_expected = tcp_md5_do_lookup(sk, (union tcp_md5_addr *)&iph->saddr,
1209 					  AF_INET);
1210 	hash_location = tcp_parse_md5sig_option(th);
1211 
1212 	/* We've parsed the options - do we have a hash? */
1213 	if (!hash_expected && !hash_location)
1214 		return false;
1215 
1216 	if (hash_expected && !hash_location) {
1217 		NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_TCPMD5NOTFOUND);
1218 		return true;
1219 	}
1220 
1221 	if (!hash_expected && hash_location) {
1222 		NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_TCPMD5UNEXPECTED);
1223 		return true;
1224 	}
1225 
1226 	/* Okay, so this is hash_expected and hash_location -
1227 	 * so we need to calculate the checksum.
1228 	 */
1229 	genhash = tcp_v4_md5_hash_skb(newhash,
1230 				      hash_expected,
1231 				      NULL, NULL, skb);
1232 
1233 	if (genhash || memcmp(hash_location, newhash, 16) != 0) {
1234 		net_info_ratelimited("MD5 Hash failed for (%pI4, %d)->(%pI4, %d)%s\n",
1235 				     &iph->saddr, ntohs(th->source),
1236 				     &iph->daddr, ntohs(th->dest),
1237 				     genhash ? " tcp_v4_calc_md5_hash failed"
1238 				     : "");
1239 		return true;
1240 	}
1241 	return false;
1242 }
1243 
1244 #endif
1245 
1246 struct request_sock_ops tcp_request_sock_ops __read_mostly = {
1247 	.family		=	PF_INET,
1248 	.obj_size	=	sizeof(struct tcp_request_sock),
1249 	.rtx_syn_ack	=	tcp_v4_rtx_synack,
1250 	.send_ack	=	tcp_v4_reqsk_send_ack,
1251 	.destructor	=	tcp_v4_reqsk_destructor,
1252 	.send_reset	=	tcp_v4_send_reset,
1253 	.syn_ack_timeout = 	tcp_syn_ack_timeout,
1254 };
1255 
1256 #ifdef CONFIG_TCP_MD5SIG
1257 static const struct tcp_request_sock_ops tcp_request_sock_ipv4_ops = {
1258 	.md5_lookup	=	tcp_v4_reqsk_md5_lookup,
1259 	.calc_md5_hash	=	tcp_v4_md5_hash_skb,
1260 };
1261 #endif
1262 
1263 static bool tcp_fastopen_check(struct sock *sk, struct sk_buff *skb,
1264 			       struct request_sock *req,
1265 			       struct tcp_fastopen_cookie *foc,
1266 			       struct tcp_fastopen_cookie *valid_foc)
1267 {
1268 	bool skip_cookie = false;
1269 	struct fastopen_queue *fastopenq;
1270 
1271 	if (likely(!fastopen_cookie_present(foc))) {
1272 		/* See include/net/tcp.h for the meaning of these knobs */
1273 		if ((sysctl_tcp_fastopen & TFO_SERVER_ALWAYS) ||
1274 		    ((sysctl_tcp_fastopen & TFO_SERVER_COOKIE_NOT_REQD) &&
1275 		    (TCP_SKB_CB(skb)->end_seq != TCP_SKB_CB(skb)->seq + 1)))
1276 			skip_cookie = true; /* no cookie to validate */
1277 		else
1278 			return false;
1279 	}
1280 	fastopenq = inet_csk(sk)->icsk_accept_queue.fastopenq;
1281 	/* A FO option is present; bump the counter. */
1282 	NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_TCPFASTOPENPASSIVE);
1283 
1284 	/* Make sure the listener has enabled fastopen, and we don't
1285 	 * exceed the max # of pending TFO requests allowed before trying
1286 	 * to validating the cookie in order to avoid burning CPU cycles
1287 	 * unnecessarily.
1288 	 *
1289 	 * XXX (TFO) - The implication of checking the max_qlen before
1290 	 * processing a cookie request is that clients can't differentiate
1291 	 * between qlen overflow causing Fast Open to be disabled
1292 	 * temporarily vs a server not supporting Fast Open at all.
1293 	 */
1294 	if ((sysctl_tcp_fastopen & TFO_SERVER_ENABLE) == 0 ||
1295 	    fastopenq == NULL || fastopenq->max_qlen == 0)
1296 		return false;
1297 
1298 	if (fastopenq->qlen >= fastopenq->max_qlen) {
1299 		struct request_sock *req1;
1300 		spin_lock(&fastopenq->lock);
1301 		req1 = fastopenq->rskq_rst_head;
1302 		if ((req1 == NULL) || time_after(req1->expires, jiffies)) {
1303 			spin_unlock(&fastopenq->lock);
1304 			NET_INC_STATS_BH(sock_net(sk),
1305 			    LINUX_MIB_TCPFASTOPENLISTENOVERFLOW);
1306 			/* Avoid bumping LINUX_MIB_TCPFASTOPENPASSIVEFAIL*/
1307 			foc->len = -1;
1308 			return false;
1309 		}
1310 		fastopenq->rskq_rst_head = req1->dl_next;
1311 		fastopenq->qlen--;
1312 		spin_unlock(&fastopenq->lock);
1313 		reqsk_free(req1);
1314 	}
1315 	if (skip_cookie) {
1316 		tcp_rsk(req)->rcv_nxt = TCP_SKB_CB(skb)->end_seq;
1317 		return true;
1318 	}
1319 
1320 	if (foc->len == TCP_FASTOPEN_COOKIE_SIZE) {
1321 		if ((sysctl_tcp_fastopen & TFO_SERVER_COOKIE_NOT_CHKED) == 0) {
1322 			tcp_fastopen_cookie_gen(ip_hdr(skb)->saddr,
1323 						ip_hdr(skb)->daddr, valid_foc);
1324 			if ((valid_foc->len != TCP_FASTOPEN_COOKIE_SIZE) ||
1325 			    memcmp(&foc->val[0], &valid_foc->val[0],
1326 			    TCP_FASTOPEN_COOKIE_SIZE) != 0)
1327 				return false;
1328 			valid_foc->len = -1;
1329 		}
1330 		/* Acknowledge the data received from the peer. */
1331 		tcp_rsk(req)->rcv_nxt = TCP_SKB_CB(skb)->end_seq;
1332 		return true;
1333 	} else if (foc->len == 0) { /* Client requesting a cookie */
1334 		tcp_fastopen_cookie_gen(ip_hdr(skb)->saddr,
1335 					ip_hdr(skb)->daddr, valid_foc);
1336 		NET_INC_STATS_BH(sock_net(sk),
1337 		    LINUX_MIB_TCPFASTOPENCOOKIEREQD);
1338 	} else {
1339 		/* Client sent a cookie with wrong size. Treat it
1340 		 * the same as invalid and return a valid one.
1341 		 */
1342 		tcp_fastopen_cookie_gen(ip_hdr(skb)->saddr,
1343 					ip_hdr(skb)->daddr, valid_foc);
1344 	}
1345 	return false;
1346 }
1347 
1348 static int tcp_v4_conn_req_fastopen(struct sock *sk,
1349 				    struct sk_buff *skb,
1350 				    struct sk_buff *skb_synack,
1351 				    struct request_sock *req)
1352 {
1353 	struct tcp_sock *tp = tcp_sk(sk);
1354 	struct request_sock_queue *queue = &inet_csk(sk)->icsk_accept_queue;
1355 	const struct inet_request_sock *ireq = inet_rsk(req);
1356 	struct sock *child;
1357 	int err;
1358 
1359 	req->num_retrans = 0;
1360 	req->num_timeout = 0;
1361 	req->sk = NULL;
1362 
1363 	child = inet_csk(sk)->icsk_af_ops->syn_recv_sock(sk, skb, req, NULL);
1364 	if (child == NULL) {
1365 		NET_INC_STATS_BH(sock_net(sk),
1366 				 LINUX_MIB_TCPFASTOPENPASSIVEFAIL);
1367 		kfree_skb(skb_synack);
1368 		return -1;
1369 	}
1370 	err = ip_build_and_send_pkt(skb_synack, sk, ireq->ir_loc_addr,
1371 				    ireq->ir_rmt_addr, ireq->opt);
1372 	err = net_xmit_eval(err);
1373 	if (!err)
1374 		tcp_rsk(req)->snt_synack = tcp_time_stamp;
1375 	/* XXX (TFO) - is it ok to ignore error and continue? */
1376 
1377 	spin_lock(&queue->fastopenq->lock);
1378 	queue->fastopenq->qlen++;
1379 	spin_unlock(&queue->fastopenq->lock);
1380 
1381 	/* Initialize the child socket. Have to fix some values to take
1382 	 * into account the child is a Fast Open socket and is created
1383 	 * only out of the bits carried in the SYN packet.
1384 	 */
1385 	tp = tcp_sk(child);
1386 
1387 	tp->fastopen_rsk = req;
1388 	/* Do a hold on the listner sk so that if the listener is being
1389 	 * closed, the child that has been accepted can live on and still
1390 	 * access listen_lock.
1391 	 */
1392 	sock_hold(sk);
1393 	tcp_rsk(req)->listener = sk;
1394 
1395 	/* RFC1323: The window in SYN & SYN/ACK segments is never
1396 	 * scaled. So correct it appropriately.
1397 	 */
1398 	tp->snd_wnd = ntohs(tcp_hdr(skb)->window);
1399 
1400 	/* Activate the retrans timer so that SYNACK can be retransmitted.
1401 	 * The request socket is not added to the SYN table of the parent
1402 	 * because it's been added to the accept queue directly.
1403 	 */
1404 	inet_csk_reset_xmit_timer(child, ICSK_TIME_RETRANS,
1405 	    TCP_TIMEOUT_INIT, TCP_RTO_MAX);
1406 
1407 	/* Add the child socket directly into the accept queue */
1408 	inet_csk_reqsk_queue_add(sk, req, child);
1409 
1410 	/* Now finish processing the fastopen child socket. */
1411 	inet_csk(child)->icsk_af_ops->rebuild_header(child);
1412 	tcp_init_congestion_control(child);
1413 	tcp_mtup_init(child);
1414 	tcp_init_metrics(child);
1415 	tcp_init_buffer_space(child);
1416 
1417 	/* Queue the data carried in the SYN packet. We need to first
1418 	 * bump skb's refcnt because the caller will attempt to free it.
1419 	 *
1420 	 * XXX (TFO) - we honor a zero-payload TFO request for now.
1421 	 * (Any reason not to?)
1422 	 */
1423 	if (TCP_SKB_CB(skb)->end_seq == TCP_SKB_CB(skb)->seq + 1) {
1424 		/* Don't queue the skb if there is no payload in SYN.
1425 		 * XXX (TFO) - How about SYN+FIN?
1426 		 */
1427 		tp->rcv_nxt = TCP_SKB_CB(skb)->end_seq;
1428 	} else {
1429 		skb = skb_get(skb);
1430 		skb_dst_drop(skb);
1431 		__skb_pull(skb, tcp_hdr(skb)->doff * 4);
1432 		skb_set_owner_r(skb, child);
1433 		__skb_queue_tail(&child->sk_receive_queue, skb);
1434 		tp->rcv_nxt = TCP_SKB_CB(skb)->end_seq;
1435 		tp->syn_data_acked = 1;
1436 	}
1437 	sk->sk_data_ready(sk, 0);
1438 	bh_unlock_sock(child);
1439 	sock_put(child);
1440 	WARN_ON(req->sk == NULL);
1441 	return 0;
1442 }
1443 
1444 int tcp_v4_conn_request(struct sock *sk, struct sk_buff *skb)
1445 {
1446 	struct tcp_options_received tmp_opt;
1447 	struct request_sock *req;
1448 	struct inet_request_sock *ireq;
1449 	struct tcp_sock *tp = tcp_sk(sk);
1450 	struct dst_entry *dst = NULL;
1451 	__be32 saddr = ip_hdr(skb)->saddr;
1452 	__be32 daddr = ip_hdr(skb)->daddr;
1453 	__u32 isn = TCP_SKB_CB(skb)->when;
1454 	bool want_cookie = false;
1455 	struct flowi4 fl4;
1456 	struct tcp_fastopen_cookie foc = { .len = -1 };
1457 	struct tcp_fastopen_cookie valid_foc = { .len = -1 };
1458 	struct sk_buff *skb_synack;
1459 	int do_fastopen;
1460 
1461 	/* Never answer to SYNs send to broadcast or multicast */
1462 	if (skb_rtable(skb)->rt_flags & (RTCF_BROADCAST | RTCF_MULTICAST))
1463 		goto drop;
1464 
1465 	/* TW buckets are converted to open requests without
1466 	 * limitations, they conserve resources and peer is
1467 	 * evidently real one.
1468 	 */
1469 	if ((sysctl_tcp_syncookies == 2 ||
1470 	     inet_csk_reqsk_queue_is_full(sk)) && !isn) {
1471 		want_cookie = tcp_syn_flood_action(sk, skb, "TCP");
1472 		if (!want_cookie)
1473 			goto drop;
1474 	}
1475 
1476 	/* Accept backlog is full. If we have already queued enough
1477 	 * of warm entries in syn queue, drop request. It is better than
1478 	 * clogging syn queue with openreqs with exponentially increasing
1479 	 * timeout.
1480 	 */
1481 	if (sk_acceptq_is_full(sk) && inet_csk_reqsk_queue_young(sk) > 1) {
1482 		NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_LISTENOVERFLOWS);
1483 		goto drop;
1484 	}
1485 
1486 	req = inet_reqsk_alloc(&tcp_request_sock_ops);
1487 	if (!req)
1488 		goto drop;
1489 
1490 #ifdef CONFIG_TCP_MD5SIG
1491 	tcp_rsk(req)->af_specific = &tcp_request_sock_ipv4_ops;
1492 #endif
1493 
1494 	tcp_clear_options(&tmp_opt);
1495 	tmp_opt.mss_clamp = TCP_MSS_DEFAULT;
1496 	tmp_opt.user_mss  = tp->rx_opt.user_mss;
1497 	tcp_parse_options(skb, &tmp_opt, 0, want_cookie ? NULL : &foc);
1498 
1499 	if (want_cookie && !tmp_opt.saw_tstamp)
1500 		tcp_clear_options(&tmp_opt);
1501 
1502 	tmp_opt.tstamp_ok = tmp_opt.saw_tstamp;
1503 	tcp_openreq_init(req, &tmp_opt, skb);
1504 
1505 	ireq = inet_rsk(req);
1506 	ireq->ir_loc_addr = daddr;
1507 	ireq->ir_rmt_addr = saddr;
1508 	ireq->no_srccheck = inet_sk(sk)->transparent;
1509 	ireq->opt = tcp_v4_save_options(skb);
1510 
1511 	if (security_inet_conn_request(sk, skb, req))
1512 		goto drop_and_free;
1513 
1514 	if (!want_cookie || tmp_opt.tstamp_ok)
1515 		TCP_ECN_create_request(req, skb, sock_net(sk));
1516 
1517 	if (want_cookie) {
1518 		isn = cookie_v4_init_sequence(sk, skb, &req->mss);
1519 		req->cookie_ts = tmp_opt.tstamp_ok;
1520 	} else if (!isn) {
1521 		/* VJ's idea. We save last timestamp seen
1522 		 * from the destination in peer table, when entering
1523 		 * state TIME-WAIT, and check against it before
1524 		 * accepting new connection request.
1525 		 *
1526 		 * If "isn" is not zero, this request hit alive
1527 		 * timewait bucket, so that all the necessary checks
1528 		 * are made in the function processing timewait state.
1529 		 */
1530 		if (tmp_opt.saw_tstamp &&
1531 		    tcp_death_row.sysctl_tw_recycle &&
1532 		    (dst = inet_csk_route_req(sk, &fl4, req)) != NULL &&
1533 		    fl4.daddr == saddr) {
1534 			if (!tcp_peer_is_proven(req, dst, true)) {
1535 				NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_PAWSPASSIVEREJECTED);
1536 				goto drop_and_release;
1537 			}
1538 		}
1539 		/* Kill the following clause, if you dislike this way. */
1540 		else if (!sysctl_tcp_syncookies &&
1541 			 (sysctl_max_syn_backlog - inet_csk_reqsk_queue_len(sk) <
1542 			  (sysctl_max_syn_backlog >> 2)) &&
1543 			 !tcp_peer_is_proven(req, dst, false)) {
1544 			/* Without syncookies last quarter of
1545 			 * backlog is filled with destinations,
1546 			 * proven to be alive.
1547 			 * It means that we continue to communicate
1548 			 * to destinations, already remembered
1549 			 * to the moment of synflood.
1550 			 */
1551 			LIMIT_NETDEBUG(KERN_DEBUG pr_fmt("drop open request from %pI4/%u\n"),
1552 				       &saddr, ntohs(tcp_hdr(skb)->source));
1553 			goto drop_and_release;
1554 		}
1555 
1556 		isn = tcp_v4_init_sequence(skb);
1557 	}
1558 	tcp_rsk(req)->snt_isn = isn;
1559 
1560 	if (dst == NULL) {
1561 		dst = inet_csk_route_req(sk, &fl4, req);
1562 		if (dst == NULL)
1563 			goto drop_and_free;
1564 	}
1565 	do_fastopen = tcp_fastopen_check(sk, skb, req, &foc, &valid_foc);
1566 
1567 	/* We don't call tcp_v4_send_synack() directly because we need
1568 	 * to make sure a child socket can be created successfully before
1569 	 * sending back synack!
1570 	 *
1571 	 * XXX (TFO) - Ideally one would simply call tcp_v4_send_synack()
1572 	 * (or better yet, call tcp_send_synack() in the child context
1573 	 * directly, but will have to fix bunch of other code first)
1574 	 * after syn_recv_sock() except one will need to first fix the
1575 	 * latter to remove its dependency on the current implementation
1576 	 * of tcp_v4_send_synack()->tcp_select_initial_window().
1577 	 */
1578 	skb_synack = tcp_make_synack(sk, dst, req,
1579 	    fastopen_cookie_present(&valid_foc) ? &valid_foc : NULL);
1580 
1581 	if (skb_synack) {
1582 		__tcp_v4_send_check(skb_synack, ireq->ir_loc_addr, ireq->ir_rmt_addr);
1583 		skb_set_queue_mapping(skb_synack, skb_get_queue_mapping(skb));
1584 	} else
1585 		goto drop_and_free;
1586 
1587 	if (likely(!do_fastopen)) {
1588 		int err;
1589 		err = ip_build_and_send_pkt(skb_synack, sk, ireq->ir_loc_addr,
1590 		     ireq->ir_rmt_addr, ireq->opt);
1591 		err = net_xmit_eval(err);
1592 		if (err || want_cookie)
1593 			goto drop_and_free;
1594 
1595 		tcp_rsk(req)->snt_synack = tcp_time_stamp;
1596 		tcp_rsk(req)->listener = NULL;
1597 		/* Add the request_sock to the SYN table */
1598 		inet_csk_reqsk_queue_hash_add(sk, req, TCP_TIMEOUT_INIT);
1599 		if (fastopen_cookie_present(&foc) && foc.len != 0)
1600 			NET_INC_STATS_BH(sock_net(sk),
1601 			    LINUX_MIB_TCPFASTOPENPASSIVEFAIL);
1602 	} else if (tcp_v4_conn_req_fastopen(sk, skb, skb_synack, req))
1603 		goto drop_and_free;
1604 
1605 	return 0;
1606 
1607 drop_and_release:
1608 	dst_release(dst);
1609 drop_and_free:
1610 	reqsk_free(req);
1611 drop:
1612 	NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_LISTENDROPS);
1613 	return 0;
1614 }
1615 EXPORT_SYMBOL(tcp_v4_conn_request);
1616 
1617 
1618 /*
1619  * The three way handshake has completed - we got a valid synack -
1620  * now create the new socket.
1621  */
1622 struct sock *tcp_v4_syn_recv_sock(struct sock *sk, struct sk_buff *skb,
1623 				  struct request_sock *req,
1624 				  struct dst_entry *dst)
1625 {
1626 	struct inet_request_sock *ireq;
1627 	struct inet_sock *newinet;
1628 	struct tcp_sock *newtp;
1629 	struct sock *newsk;
1630 #ifdef CONFIG_TCP_MD5SIG
1631 	struct tcp_md5sig_key *key;
1632 #endif
1633 	struct ip_options_rcu *inet_opt;
1634 
1635 	if (sk_acceptq_is_full(sk))
1636 		goto exit_overflow;
1637 
1638 	newsk = tcp_create_openreq_child(sk, req, skb);
1639 	if (!newsk)
1640 		goto exit_nonewsk;
1641 
1642 	newsk->sk_gso_type = SKB_GSO_TCPV4;
1643 	inet_sk_rx_dst_set(newsk, skb);
1644 
1645 	newtp		      = tcp_sk(newsk);
1646 	newinet		      = inet_sk(newsk);
1647 	ireq		      = inet_rsk(req);
1648 	newinet->inet_daddr   = ireq->ir_rmt_addr;
1649 	newinet->inet_rcv_saddr = ireq->ir_loc_addr;
1650 	newinet->inet_saddr	      = ireq->ir_loc_addr;
1651 	inet_opt	      = ireq->opt;
1652 	rcu_assign_pointer(newinet->inet_opt, inet_opt);
1653 	ireq->opt	      = NULL;
1654 	newinet->mc_index     = inet_iif(skb);
1655 	newinet->mc_ttl	      = ip_hdr(skb)->ttl;
1656 	newinet->rcv_tos      = ip_hdr(skb)->tos;
1657 	inet_csk(newsk)->icsk_ext_hdr_len = 0;
1658 	if (inet_opt)
1659 		inet_csk(newsk)->icsk_ext_hdr_len = inet_opt->opt.optlen;
1660 	newinet->inet_id = newtp->write_seq ^ jiffies;
1661 
1662 	if (!dst) {
1663 		dst = inet_csk_route_child_sock(sk, newsk, req);
1664 		if (!dst)
1665 			goto put_and_exit;
1666 	} else {
1667 		/* syncookie case : see end of cookie_v4_check() */
1668 	}
1669 	sk_setup_caps(newsk, dst);
1670 
1671 	tcp_mtup_init(newsk);
1672 	tcp_sync_mss(newsk, dst_mtu(dst));
1673 	newtp->advmss = dst_metric_advmss(dst);
1674 	if (tcp_sk(sk)->rx_opt.user_mss &&
1675 	    tcp_sk(sk)->rx_opt.user_mss < newtp->advmss)
1676 		newtp->advmss = tcp_sk(sk)->rx_opt.user_mss;
1677 
1678 	tcp_initialize_rcv_mss(newsk);
1679 
1680 #ifdef CONFIG_TCP_MD5SIG
1681 	/* Copy over the MD5 key from the original socket */
1682 	key = tcp_md5_do_lookup(sk, (union tcp_md5_addr *)&newinet->inet_daddr,
1683 				AF_INET);
1684 	if (key != NULL) {
1685 		/*
1686 		 * We're using one, so create a matching key
1687 		 * on the newsk structure. If we fail to get
1688 		 * memory, then we end up not copying the key
1689 		 * across. Shucks.
1690 		 */
1691 		tcp_md5_do_add(newsk, (union tcp_md5_addr *)&newinet->inet_daddr,
1692 			       AF_INET, key->key, key->keylen, GFP_ATOMIC);
1693 		sk_nocaps_add(newsk, NETIF_F_GSO_MASK);
1694 	}
1695 #endif
1696 
1697 	if (__inet_inherit_port(sk, newsk) < 0)
1698 		goto put_and_exit;
1699 	__inet_hash_nolisten(newsk, NULL);
1700 
1701 	return newsk;
1702 
1703 exit_overflow:
1704 	NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_LISTENOVERFLOWS);
1705 exit_nonewsk:
1706 	dst_release(dst);
1707 exit:
1708 	NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_LISTENDROPS);
1709 	return NULL;
1710 put_and_exit:
1711 	inet_csk_prepare_forced_close(newsk);
1712 	tcp_done(newsk);
1713 	goto exit;
1714 }
1715 EXPORT_SYMBOL(tcp_v4_syn_recv_sock);
1716 
1717 static struct sock *tcp_v4_hnd_req(struct sock *sk, struct sk_buff *skb)
1718 {
1719 	struct tcphdr *th = tcp_hdr(skb);
1720 	const struct iphdr *iph = ip_hdr(skb);
1721 	struct sock *nsk;
1722 	struct request_sock **prev;
1723 	/* Find possible connection requests. */
1724 	struct request_sock *req = inet_csk_search_req(sk, &prev, th->source,
1725 						       iph->saddr, iph->daddr);
1726 	if (req)
1727 		return tcp_check_req(sk, skb, req, prev, false);
1728 
1729 	nsk = inet_lookup_established(sock_net(sk), &tcp_hashinfo, iph->saddr,
1730 			th->source, iph->daddr, th->dest, inet_iif(skb));
1731 
1732 	if (nsk) {
1733 		if (nsk->sk_state != TCP_TIME_WAIT) {
1734 			bh_lock_sock(nsk);
1735 			return nsk;
1736 		}
1737 		inet_twsk_put(inet_twsk(nsk));
1738 		return NULL;
1739 	}
1740 
1741 #ifdef CONFIG_SYN_COOKIES
1742 	if (!th->syn)
1743 		sk = cookie_v4_check(sk, skb, &(IPCB(skb)->opt));
1744 #endif
1745 	return sk;
1746 }
1747 
1748 static __sum16 tcp_v4_checksum_init(struct sk_buff *skb)
1749 {
1750 	const struct iphdr *iph = ip_hdr(skb);
1751 
1752 	if (skb->ip_summed == CHECKSUM_COMPLETE) {
1753 		if (!tcp_v4_check(skb->len, iph->saddr,
1754 				  iph->daddr, skb->csum)) {
1755 			skb->ip_summed = CHECKSUM_UNNECESSARY;
1756 			return 0;
1757 		}
1758 	}
1759 
1760 	skb->csum = csum_tcpudp_nofold(iph->saddr, iph->daddr,
1761 				       skb->len, IPPROTO_TCP, 0);
1762 
1763 	if (skb->len <= 76) {
1764 		return __skb_checksum_complete(skb);
1765 	}
1766 	return 0;
1767 }
1768 
1769 
1770 /* The socket must have it's spinlock held when we get
1771  * here.
1772  *
1773  * We have a potential double-lock case here, so even when
1774  * doing backlog processing we use the BH locking scheme.
1775  * This is because we cannot sleep with the original spinlock
1776  * held.
1777  */
1778 int tcp_v4_do_rcv(struct sock *sk, struct sk_buff *skb)
1779 {
1780 	struct sock *rsk;
1781 #ifdef CONFIG_TCP_MD5SIG
1782 	/*
1783 	 * We really want to reject the packet as early as possible
1784 	 * if:
1785 	 *  o We're expecting an MD5'd packet and this is no MD5 tcp option
1786 	 *  o There is an MD5 option and we're not expecting one
1787 	 */
1788 	if (tcp_v4_inbound_md5_hash(sk, skb))
1789 		goto discard;
1790 #endif
1791 
1792 	if (sk->sk_state == TCP_ESTABLISHED) { /* Fast path */
1793 		struct dst_entry *dst = sk->sk_rx_dst;
1794 
1795 		sock_rps_save_rxhash(sk, skb);
1796 		if (dst) {
1797 			if (inet_sk(sk)->rx_dst_ifindex != skb->skb_iif ||
1798 			    dst->ops->check(dst, 0) == NULL) {
1799 				dst_release(dst);
1800 				sk->sk_rx_dst = NULL;
1801 			}
1802 		}
1803 		tcp_rcv_established(sk, skb, tcp_hdr(skb), skb->len);
1804 		return 0;
1805 	}
1806 
1807 	if (skb->len < tcp_hdrlen(skb) || tcp_checksum_complete(skb))
1808 		goto csum_err;
1809 
1810 	if (sk->sk_state == TCP_LISTEN) {
1811 		struct sock *nsk = tcp_v4_hnd_req(sk, skb);
1812 		if (!nsk)
1813 			goto discard;
1814 
1815 		if (nsk != sk) {
1816 			sock_rps_save_rxhash(nsk, skb);
1817 			if (tcp_child_process(sk, nsk, skb)) {
1818 				rsk = nsk;
1819 				goto reset;
1820 			}
1821 			return 0;
1822 		}
1823 	} else
1824 		sock_rps_save_rxhash(sk, skb);
1825 
1826 	if (tcp_rcv_state_process(sk, skb, tcp_hdr(skb), skb->len)) {
1827 		rsk = sk;
1828 		goto reset;
1829 	}
1830 	return 0;
1831 
1832 reset:
1833 	tcp_v4_send_reset(rsk, skb);
1834 discard:
1835 	kfree_skb(skb);
1836 	/* Be careful here. If this function gets more complicated and
1837 	 * gcc suffers from register pressure on the x86, sk (in %ebx)
1838 	 * might be destroyed here. This current version compiles correctly,
1839 	 * but you have been warned.
1840 	 */
1841 	return 0;
1842 
1843 csum_err:
1844 	TCP_INC_STATS_BH(sock_net(sk), TCP_MIB_CSUMERRORS);
1845 	TCP_INC_STATS_BH(sock_net(sk), TCP_MIB_INERRS);
1846 	goto discard;
1847 }
1848 EXPORT_SYMBOL(tcp_v4_do_rcv);
1849 
1850 void tcp_v4_early_demux(struct sk_buff *skb)
1851 {
1852 	const struct iphdr *iph;
1853 	const struct tcphdr *th;
1854 	struct sock *sk;
1855 
1856 	if (skb->pkt_type != PACKET_HOST)
1857 		return;
1858 
1859 	if (!pskb_may_pull(skb, skb_transport_offset(skb) + sizeof(struct tcphdr)))
1860 		return;
1861 
1862 	iph = ip_hdr(skb);
1863 	th = tcp_hdr(skb);
1864 
1865 	if (th->doff < sizeof(struct tcphdr) / 4)
1866 		return;
1867 
1868 	sk = __inet_lookup_established(dev_net(skb->dev), &tcp_hashinfo,
1869 				       iph->saddr, th->source,
1870 				       iph->daddr, ntohs(th->dest),
1871 				       skb->skb_iif);
1872 	if (sk) {
1873 		skb->sk = sk;
1874 		skb->destructor = sock_edemux;
1875 		if (sk->sk_state != TCP_TIME_WAIT) {
1876 			struct dst_entry *dst = sk->sk_rx_dst;
1877 
1878 			if (dst)
1879 				dst = dst_check(dst, 0);
1880 			if (dst &&
1881 			    inet_sk(sk)->rx_dst_ifindex == skb->skb_iif)
1882 				skb_dst_set_noref(skb, dst);
1883 		}
1884 	}
1885 }
1886 
1887 /* Packet is added to VJ-style prequeue for processing in process
1888  * context, if a reader task is waiting. Apparently, this exciting
1889  * idea (VJ's mail "Re: query about TCP header on tcp-ip" of 07 Sep 93)
1890  * failed somewhere. Latency? Burstiness? Well, at least now we will
1891  * see, why it failed. 8)8)				  --ANK
1892  *
1893  */
1894 bool tcp_prequeue(struct sock *sk, struct sk_buff *skb)
1895 {
1896 	struct tcp_sock *tp = tcp_sk(sk);
1897 
1898 	if (sysctl_tcp_low_latency || !tp->ucopy.task)
1899 		return false;
1900 
1901 	if (skb->len <= tcp_hdrlen(skb) &&
1902 	    skb_queue_len(&tp->ucopy.prequeue) == 0)
1903 		return false;
1904 
1905 	skb_dst_force(skb);
1906 	__skb_queue_tail(&tp->ucopy.prequeue, skb);
1907 	tp->ucopy.memory += skb->truesize;
1908 	if (tp->ucopy.memory > sk->sk_rcvbuf) {
1909 		struct sk_buff *skb1;
1910 
1911 		BUG_ON(sock_owned_by_user(sk));
1912 
1913 		while ((skb1 = __skb_dequeue(&tp->ucopy.prequeue)) != NULL) {
1914 			sk_backlog_rcv(sk, skb1);
1915 			NET_INC_STATS_BH(sock_net(sk),
1916 					 LINUX_MIB_TCPPREQUEUEDROPPED);
1917 		}
1918 
1919 		tp->ucopy.memory = 0;
1920 	} else if (skb_queue_len(&tp->ucopy.prequeue) == 1) {
1921 		wake_up_interruptible_sync_poll(sk_sleep(sk),
1922 					   POLLIN | POLLRDNORM | POLLRDBAND);
1923 		if (!inet_csk_ack_scheduled(sk))
1924 			inet_csk_reset_xmit_timer(sk, ICSK_TIME_DACK,
1925 						  (3 * tcp_rto_min(sk)) / 4,
1926 						  TCP_RTO_MAX);
1927 	}
1928 	return true;
1929 }
1930 EXPORT_SYMBOL(tcp_prequeue);
1931 
1932 /*
1933  *	From tcp_input.c
1934  */
1935 
1936 int tcp_v4_rcv(struct sk_buff *skb)
1937 {
1938 	const struct iphdr *iph;
1939 	const struct tcphdr *th;
1940 	struct sock *sk;
1941 	int ret;
1942 	struct net *net = dev_net(skb->dev);
1943 
1944 	if (skb->pkt_type != PACKET_HOST)
1945 		goto discard_it;
1946 
1947 	/* Count it even if it's bad */
1948 	TCP_INC_STATS_BH(net, TCP_MIB_INSEGS);
1949 
1950 	if (!pskb_may_pull(skb, sizeof(struct tcphdr)))
1951 		goto discard_it;
1952 
1953 	th = tcp_hdr(skb);
1954 
1955 	if (th->doff < sizeof(struct tcphdr) / 4)
1956 		goto bad_packet;
1957 	if (!pskb_may_pull(skb, th->doff * 4))
1958 		goto discard_it;
1959 
1960 	/* An explanation is required here, I think.
1961 	 * Packet length and doff are validated by header prediction,
1962 	 * provided case of th->doff==0 is eliminated.
1963 	 * So, we defer the checks. */
1964 	if (!skb_csum_unnecessary(skb) && tcp_v4_checksum_init(skb))
1965 		goto csum_error;
1966 
1967 	th = tcp_hdr(skb);
1968 	iph = ip_hdr(skb);
1969 	TCP_SKB_CB(skb)->seq = ntohl(th->seq);
1970 	TCP_SKB_CB(skb)->end_seq = (TCP_SKB_CB(skb)->seq + th->syn + th->fin +
1971 				    skb->len - th->doff * 4);
1972 	TCP_SKB_CB(skb)->ack_seq = ntohl(th->ack_seq);
1973 	TCP_SKB_CB(skb)->when	 = 0;
1974 	TCP_SKB_CB(skb)->ip_dsfield = ipv4_get_dsfield(iph);
1975 	TCP_SKB_CB(skb)->sacked	 = 0;
1976 
1977 	sk = __inet_lookup_skb(&tcp_hashinfo, skb, th->source, th->dest);
1978 	if (!sk)
1979 		goto no_tcp_socket;
1980 
1981 process:
1982 	if (sk->sk_state == TCP_TIME_WAIT)
1983 		goto do_time_wait;
1984 
1985 	if (unlikely(iph->ttl < inet_sk(sk)->min_ttl)) {
1986 		NET_INC_STATS_BH(net, LINUX_MIB_TCPMINTTLDROP);
1987 		goto discard_and_relse;
1988 	}
1989 
1990 	if (!xfrm4_policy_check(sk, XFRM_POLICY_IN, skb))
1991 		goto discard_and_relse;
1992 	nf_reset(skb);
1993 
1994 	if (sk_filter(sk, skb))
1995 		goto discard_and_relse;
1996 
1997 	sk_mark_napi_id(sk, skb);
1998 	skb->dev = NULL;
1999 
2000 	bh_lock_sock_nested(sk);
2001 	ret = 0;
2002 	if (!sock_owned_by_user(sk)) {
2003 #ifdef CONFIG_NET_DMA
2004 		struct tcp_sock *tp = tcp_sk(sk);
2005 		if (!tp->ucopy.dma_chan && tp->ucopy.pinned_list)
2006 			tp->ucopy.dma_chan = net_dma_find_channel();
2007 		if (tp->ucopy.dma_chan)
2008 			ret = tcp_v4_do_rcv(sk, skb);
2009 		else
2010 #endif
2011 		{
2012 			if (!tcp_prequeue(sk, skb))
2013 				ret = tcp_v4_do_rcv(sk, skb);
2014 		}
2015 	} else if (unlikely(sk_add_backlog(sk, skb,
2016 					   sk->sk_rcvbuf + sk->sk_sndbuf))) {
2017 		bh_unlock_sock(sk);
2018 		NET_INC_STATS_BH(net, LINUX_MIB_TCPBACKLOGDROP);
2019 		goto discard_and_relse;
2020 	}
2021 	bh_unlock_sock(sk);
2022 
2023 	sock_put(sk);
2024 
2025 	return ret;
2026 
2027 no_tcp_socket:
2028 	if (!xfrm4_policy_check(NULL, XFRM_POLICY_IN, skb))
2029 		goto discard_it;
2030 
2031 	if (skb->len < (th->doff << 2) || tcp_checksum_complete(skb)) {
2032 csum_error:
2033 		TCP_INC_STATS_BH(net, TCP_MIB_CSUMERRORS);
2034 bad_packet:
2035 		TCP_INC_STATS_BH(net, TCP_MIB_INERRS);
2036 	} else {
2037 		tcp_v4_send_reset(NULL, skb);
2038 	}
2039 
2040 discard_it:
2041 	/* Discard frame. */
2042 	kfree_skb(skb);
2043 	return 0;
2044 
2045 discard_and_relse:
2046 	sock_put(sk);
2047 	goto discard_it;
2048 
2049 do_time_wait:
2050 	if (!xfrm4_policy_check(NULL, XFRM_POLICY_IN, skb)) {
2051 		inet_twsk_put(inet_twsk(sk));
2052 		goto discard_it;
2053 	}
2054 
2055 	if (skb->len < (th->doff << 2)) {
2056 		inet_twsk_put(inet_twsk(sk));
2057 		goto bad_packet;
2058 	}
2059 	if (tcp_checksum_complete(skb)) {
2060 		inet_twsk_put(inet_twsk(sk));
2061 		goto csum_error;
2062 	}
2063 	switch (tcp_timewait_state_process(inet_twsk(sk), skb, th)) {
2064 	case TCP_TW_SYN: {
2065 		struct sock *sk2 = inet_lookup_listener(dev_net(skb->dev),
2066 							&tcp_hashinfo,
2067 							iph->saddr, th->source,
2068 							iph->daddr, th->dest,
2069 							inet_iif(skb));
2070 		if (sk2) {
2071 			inet_twsk_deschedule(inet_twsk(sk), &tcp_death_row);
2072 			inet_twsk_put(inet_twsk(sk));
2073 			sk = sk2;
2074 			goto process;
2075 		}
2076 		/* Fall through to ACK */
2077 	}
2078 	case TCP_TW_ACK:
2079 		tcp_v4_timewait_ack(sk, skb);
2080 		break;
2081 	case TCP_TW_RST:
2082 		goto no_tcp_socket;
2083 	case TCP_TW_SUCCESS:;
2084 	}
2085 	goto discard_it;
2086 }
2087 
2088 static struct timewait_sock_ops tcp_timewait_sock_ops = {
2089 	.twsk_obj_size	= sizeof(struct tcp_timewait_sock),
2090 	.twsk_unique	= tcp_twsk_unique,
2091 	.twsk_destructor= tcp_twsk_destructor,
2092 };
2093 
2094 void inet_sk_rx_dst_set(struct sock *sk, const struct sk_buff *skb)
2095 {
2096 	struct dst_entry *dst = skb_dst(skb);
2097 
2098 	dst_hold(dst);
2099 	sk->sk_rx_dst = dst;
2100 	inet_sk(sk)->rx_dst_ifindex = skb->skb_iif;
2101 }
2102 EXPORT_SYMBOL(inet_sk_rx_dst_set);
2103 
2104 const struct inet_connection_sock_af_ops ipv4_specific = {
2105 	.queue_xmit	   = ip_queue_xmit,
2106 	.send_check	   = tcp_v4_send_check,
2107 	.rebuild_header	   = inet_sk_rebuild_header,
2108 	.sk_rx_dst_set	   = inet_sk_rx_dst_set,
2109 	.conn_request	   = tcp_v4_conn_request,
2110 	.syn_recv_sock	   = tcp_v4_syn_recv_sock,
2111 	.net_header_len	   = sizeof(struct iphdr),
2112 	.setsockopt	   = ip_setsockopt,
2113 	.getsockopt	   = ip_getsockopt,
2114 	.addr2sockaddr	   = inet_csk_addr2sockaddr,
2115 	.sockaddr_len	   = sizeof(struct sockaddr_in),
2116 	.bind_conflict	   = inet_csk_bind_conflict,
2117 #ifdef CONFIG_COMPAT
2118 	.compat_setsockopt = compat_ip_setsockopt,
2119 	.compat_getsockopt = compat_ip_getsockopt,
2120 #endif
2121 };
2122 EXPORT_SYMBOL(ipv4_specific);
2123 
2124 #ifdef CONFIG_TCP_MD5SIG
2125 static const struct tcp_sock_af_ops tcp_sock_ipv4_specific = {
2126 	.md5_lookup		= tcp_v4_md5_lookup,
2127 	.calc_md5_hash		= tcp_v4_md5_hash_skb,
2128 	.md5_parse		= tcp_v4_parse_md5_keys,
2129 };
2130 #endif
2131 
2132 /* NOTE: A lot of things set to zero explicitly by call to
2133  *       sk_alloc() so need not be done here.
2134  */
2135 static int tcp_v4_init_sock(struct sock *sk)
2136 {
2137 	struct inet_connection_sock *icsk = inet_csk(sk);
2138 
2139 	tcp_init_sock(sk);
2140 
2141 	icsk->icsk_af_ops = &ipv4_specific;
2142 
2143 #ifdef CONFIG_TCP_MD5SIG
2144 	tcp_sk(sk)->af_specific = &tcp_sock_ipv4_specific;
2145 #endif
2146 
2147 	return 0;
2148 }
2149 
2150 void tcp_v4_destroy_sock(struct sock *sk)
2151 {
2152 	struct tcp_sock *tp = tcp_sk(sk);
2153 
2154 	tcp_clear_xmit_timers(sk);
2155 
2156 	tcp_cleanup_congestion_control(sk);
2157 
2158 	/* Cleanup up the write buffer. */
2159 	tcp_write_queue_purge(sk);
2160 
2161 	/* Cleans up our, hopefully empty, out_of_order_queue. */
2162 	__skb_queue_purge(&tp->out_of_order_queue);
2163 
2164 #ifdef CONFIG_TCP_MD5SIG
2165 	/* Clean up the MD5 key list, if any */
2166 	if (tp->md5sig_info) {
2167 		tcp_clear_md5_list(sk);
2168 		kfree_rcu(tp->md5sig_info, rcu);
2169 		tp->md5sig_info = NULL;
2170 	}
2171 #endif
2172 
2173 #ifdef CONFIG_NET_DMA
2174 	/* Cleans up our sk_async_wait_queue */
2175 	__skb_queue_purge(&sk->sk_async_wait_queue);
2176 #endif
2177 
2178 	/* Clean prequeue, it must be empty really */
2179 	__skb_queue_purge(&tp->ucopy.prequeue);
2180 
2181 	/* Clean up a referenced TCP bind bucket. */
2182 	if (inet_csk(sk)->icsk_bind_hash)
2183 		inet_put_port(sk);
2184 
2185 	BUG_ON(tp->fastopen_rsk != NULL);
2186 
2187 	/* If socket is aborted during connect operation */
2188 	tcp_free_fastopen_req(tp);
2189 
2190 	sk_sockets_allocated_dec(sk);
2191 	sock_release_memcg(sk);
2192 }
2193 EXPORT_SYMBOL(tcp_v4_destroy_sock);
2194 
2195 #ifdef CONFIG_PROC_FS
2196 /* Proc filesystem TCP sock list dumping. */
2197 
2198 /*
2199  * Get next listener socket follow cur.  If cur is NULL, get first socket
2200  * starting from bucket given in st->bucket; when st->bucket is zero the
2201  * very first socket in the hash table is returned.
2202  */
2203 static void *listening_get_next(struct seq_file *seq, void *cur)
2204 {
2205 	struct inet_connection_sock *icsk;
2206 	struct hlist_nulls_node *node;
2207 	struct sock *sk = cur;
2208 	struct inet_listen_hashbucket *ilb;
2209 	struct tcp_iter_state *st = seq->private;
2210 	struct net *net = seq_file_net(seq);
2211 
2212 	if (!sk) {
2213 		ilb = &tcp_hashinfo.listening_hash[st->bucket];
2214 		spin_lock_bh(&ilb->lock);
2215 		sk = sk_nulls_head(&ilb->head);
2216 		st->offset = 0;
2217 		goto get_sk;
2218 	}
2219 	ilb = &tcp_hashinfo.listening_hash[st->bucket];
2220 	++st->num;
2221 	++st->offset;
2222 
2223 	if (st->state == TCP_SEQ_STATE_OPENREQ) {
2224 		struct request_sock *req = cur;
2225 
2226 		icsk = inet_csk(st->syn_wait_sk);
2227 		req = req->dl_next;
2228 		while (1) {
2229 			while (req) {
2230 				if (req->rsk_ops->family == st->family) {
2231 					cur = req;
2232 					goto out;
2233 				}
2234 				req = req->dl_next;
2235 			}
2236 			if (++st->sbucket >= icsk->icsk_accept_queue.listen_opt->nr_table_entries)
2237 				break;
2238 get_req:
2239 			req = icsk->icsk_accept_queue.listen_opt->syn_table[st->sbucket];
2240 		}
2241 		sk	  = sk_nulls_next(st->syn_wait_sk);
2242 		st->state = TCP_SEQ_STATE_LISTENING;
2243 		read_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
2244 	} else {
2245 		icsk = inet_csk(sk);
2246 		read_lock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
2247 		if (reqsk_queue_len(&icsk->icsk_accept_queue))
2248 			goto start_req;
2249 		read_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
2250 		sk = sk_nulls_next(sk);
2251 	}
2252 get_sk:
2253 	sk_nulls_for_each_from(sk, node) {
2254 		if (!net_eq(sock_net(sk), net))
2255 			continue;
2256 		if (sk->sk_family == st->family) {
2257 			cur = sk;
2258 			goto out;
2259 		}
2260 		icsk = inet_csk(sk);
2261 		read_lock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
2262 		if (reqsk_queue_len(&icsk->icsk_accept_queue)) {
2263 start_req:
2264 			st->uid		= sock_i_uid(sk);
2265 			st->syn_wait_sk = sk;
2266 			st->state	= TCP_SEQ_STATE_OPENREQ;
2267 			st->sbucket	= 0;
2268 			goto get_req;
2269 		}
2270 		read_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
2271 	}
2272 	spin_unlock_bh(&ilb->lock);
2273 	st->offset = 0;
2274 	if (++st->bucket < INET_LHTABLE_SIZE) {
2275 		ilb = &tcp_hashinfo.listening_hash[st->bucket];
2276 		spin_lock_bh(&ilb->lock);
2277 		sk = sk_nulls_head(&ilb->head);
2278 		goto get_sk;
2279 	}
2280 	cur = NULL;
2281 out:
2282 	return cur;
2283 }
2284 
2285 static void *listening_get_idx(struct seq_file *seq, loff_t *pos)
2286 {
2287 	struct tcp_iter_state *st = seq->private;
2288 	void *rc;
2289 
2290 	st->bucket = 0;
2291 	st->offset = 0;
2292 	rc = listening_get_next(seq, NULL);
2293 
2294 	while (rc && *pos) {
2295 		rc = listening_get_next(seq, rc);
2296 		--*pos;
2297 	}
2298 	return rc;
2299 }
2300 
2301 static inline bool empty_bucket(const struct tcp_iter_state *st)
2302 {
2303 	return hlist_nulls_empty(&tcp_hashinfo.ehash[st->bucket].chain);
2304 }
2305 
2306 /*
2307  * Get first established socket starting from bucket given in st->bucket.
2308  * If st->bucket is zero, the very first socket in the hash is returned.
2309  */
2310 static void *established_get_first(struct seq_file *seq)
2311 {
2312 	struct tcp_iter_state *st = seq->private;
2313 	struct net *net = seq_file_net(seq);
2314 	void *rc = NULL;
2315 
2316 	st->offset = 0;
2317 	for (; st->bucket <= tcp_hashinfo.ehash_mask; ++st->bucket) {
2318 		struct sock *sk;
2319 		struct hlist_nulls_node *node;
2320 		spinlock_t *lock = inet_ehash_lockp(&tcp_hashinfo, st->bucket);
2321 
2322 		/* Lockless fast path for the common case of empty buckets */
2323 		if (empty_bucket(st))
2324 			continue;
2325 
2326 		spin_lock_bh(lock);
2327 		sk_nulls_for_each(sk, node, &tcp_hashinfo.ehash[st->bucket].chain) {
2328 			if (sk->sk_family != st->family ||
2329 			    !net_eq(sock_net(sk), net)) {
2330 				continue;
2331 			}
2332 			rc = sk;
2333 			goto out;
2334 		}
2335 		spin_unlock_bh(lock);
2336 	}
2337 out:
2338 	return rc;
2339 }
2340 
2341 static void *established_get_next(struct seq_file *seq, void *cur)
2342 {
2343 	struct sock *sk = cur;
2344 	struct hlist_nulls_node *node;
2345 	struct tcp_iter_state *st = seq->private;
2346 	struct net *net = seq_file_net(seq);
2347 
2348 	++st->num;
2349 	++st->offset;
2350 
2351 	sk = sk_nulls_next(sk);
2352 
2353 	sk_nulls_for_each_from(sk, node) {
2354 		if (sk->sk_family == st->family && net_eq(sock_net(sk), net))
2355 			return sk;
2356 	}
2357 
2358 	spin_unlock_bh(inet_ehash_lockp(&tcp_hashinfo, st->bucket));
2359 	++st->bucket;
2360 	return established_get_first(seq);
2361 }
2362 
2363 static void *established_get_idx(struct seq_file *seq, loff_t pos)
2364 {
2365 	struct tcp_iter_state *st = seq->private;
2366 	void *rc;
2367 
2368 	st->bucket = 0;
2369 	rc = established_get_first(seq);
2370 
2371 	while (rc && pos) {
2372 		rc = established_get_next(seq, rc);
2373 		--pos;
2374 	}
2375 	return rc;
2376 }
2377 
2378 static void *tcp_get_idx(struct seq_file *seq, loff_t pos)
2379 {
2380 	void *rc;
2381 	struct tcp_iter_state *st = seq->private;
2382 
2383 	st->state = TCP_SEQ_STATE_LISTENING;
2384 	rc	  = listening_get_idx(seq, &pos);
2385 
2386 	if (!rc) {
2387 		st->state = TCP_SEQ_STATE_ESTABLISHED;
2388 		rc	  = established_get_idx(seq, pos);
2389 	}
2390 
2391 	return rc;
2392 }
2393 
2394 static void *tcp_seek_last_pos(struct seq_file *seq)
2395 {
2396 	struct tcp_iter_state *st = seq->private;
2397 	int offset = st->offset;
2398 	int orig_num = st->num;
2399 	void *rc = NULL;
2400 
2401 	switch (st->state) {
2402 	case TCP_SEQ_STATE_OPENREQ:
2403 	case TCP_SEQ_STATE_LISTENING:
2404 		if (st->bucket >= INET_LHTABLE_SIZE)
2405 			break;
2406 		st->state = TCP_SEQ_STATE_LISTENING;
2407 		rc = listening_get_next(seq, NULL);
2408 		while (offset-- && rc)
2409 			rc = listening_get_next(seq, rc);
2410 		if (rc)
2411 			break;
2412 		st->bucket = 0;
2413 		st->state = TCP_SEQ_STATE_ESTABLISHED;
2414 		/* Fallthrough */
2415 	case TCP_SEQ_STATE_ESTABLISHED:
2416 		if (st->bucket > tcp_hashinfo.ehash_mask)
2417 			break;
2418 		rc = established_get_first(seq);
2419 		while (offset-- && rc)
2420 			rc = established_get_next(seq, rc);
2421 	}
2422 
2423 	st->num = orig_num;
2424 
2425 	return rc;
2426 }
2427 
2428 static void *tcp_seq_start(struct seq_file *seq, loff_t *pos)
2429 {
2430 	struct tcp_iter_state *st = seq->private;
2431 	void *rc;
2432 
2433 	if (*pos && *pos == st->last_pos) {
2434 		rc = tcp_seek_last_pos(seq);
2435 		if (rc)
2436 			goto out;
2437 	}
2438 
2439 	st->state = TCP_SEQ_STATE_LISTENING;
2440 	st->num = 0;
2441 	st->bucket = 0;
2442 	st->offset = 0;
2443 	rc = *pos ? tcp_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
2444 
2445 out:
2446 	st->last_pos = *pos;
2447 	return rc;
2448 }
2449 
2450 static void *tcp_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2451 {
2452 	struct tcp_iter_state *st = seq->private;
2453 	void *rc = NULL;
2454 
2455 	if (v == SEQ_START_TOKEN) {
2456 		rc = tcp_get_idx(seq, 0);
2457 		goto out;
2458 	}
2459 
2460 	switch (st->state) {
2461 	case TCP_SEQ_STATE_OPENREQ:
2462 	case TCP_SEQ_STATE_LISTENING:
2463 		rc = listening_get_next(seq, v);
2464 		if (!rc) {
2465 			st->state = TCP_SEQ_STATE_ESTABLISHED;
2466 			st->bucket = 0;
2467 			st->offset = 0;
2468 			rc	  = established_get_first(seq);
2469 		}
2470 		break;
2471 	case TCP_SEQ_STATE_ESTABLISHED:
2472 		rc = established_get_next(seq, v);
2473 		break;
2474 	}
2475 out:
2476 	++*pos;
2477 	st->last_pos = *pos;
2478 	return rc;
2479 }
2480 
2481 static void tcp_seq_stop(struct seq_file *seq, void *v)
2482 {
2483 	struct tcp_iter_state *st = seq->private;
2484 
2485 	switch (st->state) {
2486 	case TCP_SEQ_STATE_OPENREQ:
2487 		if (v) {
2488 			struct inet_connection_sock *icsk = inet_csk(st->syn_wait_sk);
2489 			read_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
2490 		}
2491 	case TCP_SEQ_STATE_LISTENING:
2492 		if (v != SEQ_START_TOKEN)
2493 			spin_unlock_bh(&tcp_hashinfo.listening_hash[st->bucket].lock);
2494 		break;
2495 	case TCP_SEQ_STATE_ESTABLISHED:
2496 		if (v)
2497 			spin_unlock_bh(inet_ehash_lockp(&tcp_hashinfo, st->bucket));
2498 		break;
2499 	}
2500 }
2501 
2502 int tcp_seq_open(struct inode *inode, struct file *file)
2503 {
2504 	struct tcp_seq_afinfo *afinfo = PDE_DATA(inode);
2505 	struct tcp_iter_state *s;
2506 	int err;
2507 
2508 	err = seq_open_net(inode, file, &afinfo->seq_ops,
2509 			  sizeof(struct tcp_iter_state));
2510 	if (err < 0)
2511 		return err;
2512 
2513 	s = ((struct seq_file *)file->private_data)->private;
2514 	s->family		= afinfo->family;
2515 	s->last_pos 		= 0;
2516 	return 0;
2517 }
2518 EXPORT_SYMBOL(tcp_seq_open);
2519 
2520 int tcp_proc_register(struct net *net, struct tcp_seq_afinfo *afinfo)
2521 {
2522 	int rc = 0;
2523 	struct proc_dir_entry *p;
2524 
2525 	afinfo->seq_ops.start		= tcp_seq_start;
2526 	afinfo->seq_ops.next		= tcp_seq_next;
2527 	afinfo->seq_ops.stop		= tcp_seq_stop;
2528 
2529 	p = proc_create_data(afinfo->name, S_IRUGO, net->proc_net,
2530 			     afinfo->seq_fops, afinfo);
2531 	if (!p)
2532 		rc = -ENOMEM;
2533 	return rc;
2534 }
2535 EXPORT_SYMBOL(tcp_proc_register);
2536 
2537 void tcp_proc_unregister(struct net *net, struct tcp_seq_afinfo *afinfo)
2538 {
2539 	remove_proc_entry(afinfo->name, net->proc_net);
2540 }
2541 EXPORT_SYMBOL(tcp_proc_unregister);
2542 
2543 static void get_openreq4(const struct sock *sk, const struct request_sock *req,
2544 			 struct seq_file *f, int i, kuid_t uid)
2545 {
2546 	const struct inet_request_sock *ireq = inet_rsk(req);
2547 	long delta = req->expires - jiffies;
2548 
2549 	seq_printf(f, "%4d: %08X:%04X %08X:%04X"
2550 		" %02X %08X:%08X %02X:%08lX %08X %5u %8d %u %d %pK",
2551 		i,
2552 		ireq->ir_loc_addr,
2553 		ntohs(inet_sk(sk)->inet_sport),
2554 		ireq->ir_rmt_addr,
2555 		ntohs(ireq->ir_rmt_port),
2556 		TCP_SYN_RECV,
2557 		0, 0, /* could print option size, but that is af dependent. */
2558 		1,    /* timers active (only the expire timer) */
2559 		jiffies_delta_to_clock_t(delta),
2560 		req->num_timeout,
2561 		from_kuid_munged(seq_user_ns(f), uid),
2562 		0,  /* non standard timer */
2563 		0, /* open_requests have no inode */
2564 		atomic_read(&sk->sk_refcnt),
2565 		req);
2566 }
2567 
2568 static void get_tcp4_sock(struct sock *sk, struct seq_file *f, int i)
2569 {
2570 	int timer_active;
2571 	unsigned long timer_expires;
2572 	const struct tcp_sock *tp = tcp_sk(sk);
2573 	const struct inet_connection_sock *icsk = inet_csk(sk);
2574 	const struct inet_sock *inet = inet_sk(sk);
2575 	struct fastopen_queue *fastopenq = icsk->icsk_accept_queue.fastopenq;
2576 	__be32 dest = inet->inet_daddr;
2577 	__be32 src = inet->inet_rcv_saddr;
2578 	__u16 destp = ntohs(inet->inet_dport);
2579 	__u16 srcp = ntohs(inet->inet_sport);
2580 	int rx_queue;
2581 
2582 	if (icsk->icsk_pending == ICSK_TIME_RETRANS ||
2583 	    icsk->icsk_pending == ICSK_TIME_EARLY_RETRANS ||
2584 	    icsk->icsk_pending == ICSK_TIME_LOSS_PROBE) {
2585 		timer_active	= 1;
2586 		timer_expires	= icsk->icsk_timeout;
2587 	} else if (icsk->icsk_pending == ICSK_TIME_PROBE0) {
2588 		timer_active	= 4;
2589 		timer_expires	= icsk->icsk_timeout;
2590 	} else if (timer_pending(&sk->sk_timer)) {
2591 		timer_active	= 2;
2592 		timer_expires	= sk->sk_timer.expires;
2593 	} else {
2594 		timer_active	= 0;
2595 		timer_expires = jiffies;
2596 	}
2597 
2598 	if (sk->sk_state == TCP_LISTEN)
2599 		rx_queue = sk->sk_ack_backlog;
2600 	else
2601 		/*
2602 		 * because we dont lock socket, we might find a transient negative value
2603 		 */
2604 		rx_queue = max_t(int, tp->rcv_nxt - tp->copied_seq, 0);
2605 
2606 	seq_printf(f, "%4d: %08X:%04X %08X:%04X %02X %08X:%08X %02X:%08lX "
2607 			"%08X %5u %8d %lu %d %pK %lu %lu %u %u %d",
2608 		i, src, srcp, dest, destp, sk->sk_state,
2609 		tp->write_seq - tp->snd_una,
2610 		rx_queue,
2611 		timer_active,
2612 		jiffies_delta_to_clock_t(timer_expires - jiffies),
2613 		icsk->icsk_retransmits,
2614 		from_kuid_munged(seq_user_ns(f), sock_i_uid(sk)),
2615 		icsk->icsk_probes_out,
2616 		sock_i_ino(sk),
2617 		atomic_read(&sk->sk_refcnt), sk,
2618 		jiffies_to_clock_t(icsk->icsk_rto),
2619 		jiffies_to_clock_t(icsk->icsk_ack.ato),
2620 		(icsk->icsk_ack.quick << 1) | icsk->icsk_ack.pingpong,
2621 		tp->snd_cwnd,
2622 		sk->sk_state == TCP_LISTEN ?
2623 		    (fastopenq ? fastopenq->max_qlen : 0) :
2624 		    (tcp_in_initial_slowstart(tp) ? -1 : tp->snd_ssthresh));
2625 }
2626 
2627 static void get_timewait4_sock(const struct inet_timewait_sock *tw,
2628 			       struct seq_file *f, int i)
2629 {
2630 	__be32 dest, src;
2631 	__u16 destp, srcp;
2632 	long delta = tw->tw_ttd - jiffies;
2633 
2634 	dest  = tw->tw_daddr;
2635 	src   = tw->tw_rcv_saddr;
2636 	destp = ntohs(tw->tw_dport);
2637 	srcp  = ntohs(tw->tw_sport);
2638 
2639 	seq_printf(f, "%4d: %08X:%04X %08X:%04X"
2640 		" %02X %08X:%08X %02X:%08lX %08X %5d %8d %d %d %pK",
2641 		i, src, srcp, dest, destp, tw->tw_substate, 0, 0,
2642 		3, jiffies_delta_to_clock_t(delta), 0, 0, 0, 0,
2643 		atomic_read(&tw->tw_refcnt), tw);
2644 }
2645 
2646 #define TMPSZ 150
2647 
2648 static int tcp4_seq_show(struct seq_file *seq, void *v)
2649 {
2650 	struct tcp_iter_state *st;
2651 	struct sock *sk = v;
2652 
2653 	seq_setwidth(seq, TMPSZ - 1);
2654 	if (v == SEQ_START_TOKEN) {
2655 		seq_puts(seq, "  sl  local_address rem_address   st tx_queue "
2656 			   "rx_queue tr tm->when retrnsmt   uid  timeout "
2657 			   "inode");
2658 		goto out;
2659 	}
2660 	st = seq->private;
2661 
2662 	switch (st->state) {
2663 	case TCP_SEQ_STATE_LISTENING:
2664 	case TCP_SEQ_STATE_ESTABLISHED:
2665 		if (sk->sk_state == TCP_TIME_WAIT)
2666 			get_timewait4_sock(v, seq, st->num);
2667 		else
2668 			get_tcp4_sock(v, seq, st->num);
2669 		break;
2670 	case TCP_SEQ_STATE_OPENREQ:
2671 		get_openreq4(st->syn_wait_sk, v, seq, st->num, st->uid);
2672 		break;
2673 	}
2674 out:
2675 	seq_pad(seq, '\n');
2676 	return 0;
2677 }
2678 
2679 static const struct file_operations tcp_afinfo_seq_fops = {
2680 	.owner   = THIS_MODULE,
2681 	.open    = tcp_seq_open,
2682 	.read    = seq_read,
2683 	.llseek  = seq_lseek,
2684 	.release = seq_release_net
2685 };
2686 
2687 static struct tcp_seq_afinfo tcp4_seq_afinfo = {
2688 	.name		= "tcp",
2689 	.family		= AF_INET,
2690 	.seq_fops	= &tcp_afinfo_seq_fops,
2691 	.seq_ops	= {
2692 		.show		= tcp4_seq_show,
2693 	},
2694 };
2695 
2696 static int __net_init tcp4_proc_init_net(struct net *net)
2697 {
2698 	return tcp_proc_register(net, &tcp4_seq_afinfo);
2699 }
2700 
2701 static void __net_exit tcp4_proc_exit_net(struct net *net)
2702 {
2703 	tcp_proc_unregister(net, &tcp4_seq_afinfo);
2704 }
2705 
2706 static struct pernet_operations tcp4_net_ops = {
2707 	.init = tcp4_proc_init_net,
2708 	.exit = tcp4_proc_exit_net,
2709 };
2710 
2711 int __init tcp4_proc_init(void)
2712 {
2713 	return register_pernet_subsys(&tcp4_net_ops);
2714 }
2715 
2716 void tcp4_proc_exit(void)
2717 {
2718 	unregister_pernet_subsys(&tcp4_net_ops);
2719 }
2720 #endif /* CONFIG_PROC_FS */
2721 
2722 struct proto tcp_prot = {
2723 	.name			= "TCP",
2724 	.owner			= THIS_MODULE,
2725 	.close			= tcp_close,
2726 	.connect		= tcp_v4_connect,
2727 	.disconnect		= tcp_disconnect,
2728 	.accept			= inet_csk_accept,
2729 	.ioctl			= tcp_ioctl,
2730 	.init			= tcp_v4_init_sock,
2731 	.destroy		= tcp_v4_destroy_sock,
2732 	.shutdown		= tcp_shutdown,
2733 	.setsockopt		= tcp_setsockopt,
2734 	.getsockopt		= tcp_getsockopt,
2735 	.recvmsg		= tcp_recvmsg,
2736 	.sendmsg		= tcp_sendmsg,
2737 	.sendpage		= tcp_sendpage,
2738 	.backlog_rcv		= tcp_v4_do_rcv,
2739 	.release_cb		= tcp_release_cb,
2740 	.mtu_reduced		= tcp_v4_mtu_reduced,
2741 	.hash			= inet_hash,
2742 	.unhash			= inet_unhash,
2743 	.get_port		= inet_csk_get_port,
2744 	.enter_memory_pressure	= tcp_enter_memory_pressure,
2745 	.stream_memory_free	= tcp_stream_memory_free,
2746 	.sockets_allocated	= &tcp_sockets_allocated,
2747 	.orphan_count		= &tcp_orphan_count,
2748 	.memory_allocated	= &tcp_memory_allocated,
2749 	.memory_pressure	= &tcp_memory_pressure,
2750 	.sysctl_mem		= sysctl_tcp_mem,
2751 	.sysctl_wmem		= sysctl_tcp_wmem,
2752 	.sysctl_rmem		= sysctl_tcp_rmem,
2753 	.max_header		= MAX_TCP_HEADER,
2754 	.obj_size		= sizeof(struct tcp_sock),
2755 	.slab_flags		= SLAB_DESTROY_BY_RCU,
2756 	.twsk_prot		= &tcp_timewait_sock_ops,
2757 	.rsk_prot		= &tcp_request_sock_ops,
2758 	.h.hashinfo		= &tcp_hashinfo,
2759 	.no_autobind		= true,
2760 #ifdef CONFIG_COMPAT
2761 	.compat_setsockopt	= compat_tcp_setsockopt,
2762 	.compat_getsockopt	= compat_tcp_getsockopt,
2763 #endif
2764 #ifdef CONFIG_MEMCG_KMEM
2765 	.init_cgroup		= tcp_init_cgroup,
2766 	.destroy_cgroup		= tcp_destroy_cgroup,
2767 	.proto_cgroup		= tcp_proto_cgroup,
2768 #endif
2769 };
2770 EXPORT_SYMBOL(tcp_prot);
2771 
2772 static int __net_init tcp_sk_init(struct net *net)
2773 {
2774 	net->ipv4.sysctl_tcp_ecn = 2;
2775 	return 0;
2776 }
2777 
2778 static void __net_exit tcp_sk_exit(struct net *net)
2779 {
2780 }
2781 
2782 static void __net_exit tcp_sk_exit_batch(struct list_head *net_exit_list)
2783 {
2784 	inet_twsk_purge(&tcp_hashinfo, &tcp_death_row, AF_INET);
2785 }
2786 
2787 static struct pernet_operations __net_initdata tcp_sk_ops = {
2788        .init	   = tcp_sk_init,
2789        .exit	   = tcp_sk_exit,
2790        .exit_batch = tcp_sk_exit_batch,
2791 };
2792 
2793 void __init tcp_v4_init(void)
2794 {
2795 	inet_hashinfo_init(&tcp_hashinfo);
2796 	if (register_pernet_subsys(&tcp_sk_ops))
2797 		panic("Failed to create the TCP control socket.\n");
2798 }
2799