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/secure_seq.h>
76#include <net/tcp_memcontrol.h>
77#include <net/busy_poll.h>
78
79#include <linux/inet.h>
80#include <linux/ipv6.h>
81#include <linux/stddef.h>
82#include <linux/proc_fs.h>
83#include <linux/seq_file.h>
84
85#include <linux/crypto.h>
86#include <linux/scatterlist.h>
87
88int sysctl_tcp_tw_reuse __read_mostly;
89int sysctl_tcp_low_latency __read_mostly;
90EXPORT_SYMBOL(sysctl_tcp_low_latency);
91
92#ifdef CONFIG_TCP_MD5SIG
93static int tcp_v4_md5_hash_hdr(char *md5_hash, const struct tcp_md5sig_key *key,
94			       __be32 daddr, __be32 saddr, const struct tcphdr *th);
95#endif
96
97struct inet_hashinfo tcp_hashinfo;
98EXPORT_SYMBOL(tcp_hashinfo);
99
100static  __u32 tcp_v4_init_sequence(const struct sk_buff *skb)
101{
102	return secure_tcp_sequence_number(ip_hdr(skb)->daddr,
103					  ip_hdr(skb)->saddr,
104					  tcp_hdr(skb)->dest,
105					  tcp_hdr(skb)->source);
106}
107
108int tcp_twsk_unique(struct sock *sk, struct sock *sktw, void *twp)
109{
110	const struct tcp_timewait_sock *tcptw = tcp_twsk(sktw);
111	struct tcp_sock *tp = tcp_sk(sk);
112
113	/* With PAWS, it is safe from the viewpoint
114	   of data integrity. Even without PAWS it is safe provided sequence
115	   spaces do not overlap i.e. at data rates <= 80Mbit/sec.
116
117	   Actually, the idea is close to VJ's one, only timestamp cache is
118	   held not per host, but per port pair and TW bucket is used as state
119	   holder.
120
121	   If TW bucket has been already destroyed we fall back to VJ's scheme
122	   and use initial timestamp retrieved from peer table.
123	 */
124	if (tcptw->tw_ts_recent_stamp &&
125	    (!twp || (sysctl_tcp_tw_reuse &&
126			     get_seconds() - tcptw->tw_ts_recent_stamp > 1))) {
127		tp->write_seq = tcptw->tw_snd_nxt + 65535 + 2;
128		if (tp->write_seq == 0)
129			tp->write_seq = 1;
130		tp->rx_opt.ts_recent	   = tcptw->tw_ts_recent;
131		tp->rx_opt.ts_recent_stamp = tcptw->tw_ts_recent_stamp;
132		sock_hold(sktw);
133		return 1;
134	}
135
136	return 0;
137}
138EXPORT_SYMBOL_GPL(tcp_twsk_unique);
139
140/* This will initiate an outgoing connection. */
141int tcp_v4_connect(struct sock *sk, struct sockaddr *uaddr, int addr_len)
142{
143	struct sockaddr_in *usin = (struct sockaddr_in *)uaddr;
144	struct inet_sock *inet = inet_sk(sk);
145	struct tcp_sock *tp = tcp_sk(sk);
146	__be16 orig_sport, orig_dport;
147	__be32 daddr, nexthop;
148	struct flowi4 *fl4;
149	struct rtable *rt;
150	int err;
151	struct ip_options_rcu *inet_opt;
152
153	if (addr_len < sizeof(struct sockaddr_in))
154		return -EINVAL;
155
156	if (usin->sin_family != AF_INET)
157		return -EAFNOSUPPORT;
158
159	nexthop = daddr = usin->sin_addr.s_addr;
160	inet_opt = rcu_dereference_protected(inet->inet_opt,
161					     sock_owned_by_user(sk));
162	if (inet_opt && inet_opt->opt.srr) {
163		if (!daddr)
164			return -EINVAL;
165		nexthop = inet_opt->opt.faddr;
166	}
167
168	orig_sport = inet->inet_sport;
169	orig_dport = usin->sin_port;
170	fl4 = &inet->cork.fl.u.ip4;
171	rt = ip_route_connect(fl4, nexthop, inet->inet_saddr,
172			      RT_CONN_FLAGS(sk), sk->sk_bound_dev_if,
173			      IPPROTO_TCP,
174			      orig_sport, orig_dport, sk);
175	if (IS_ERR(rt)) {
176		err = PTR_ERR(rt);
177		if (err == -ENETUNREACH)
178			IP_INC_STATS(sock_net(sk), IPSTATS_MIB_OUTNOROUTES);
179		return err;
180	}
181
182	if (rt->rt_flags & (RTCF_MULTICAST | RTCF_BROADCAST)) {
183		ip_rt_put(rt);
184		return -ENETUNREACH;
185	}
186
187	if (!inet_opt || !inet_opt->opt.srr)
188		daddr = fl4->daddr;
189
190	if (!inet->inet_saddr)
191		inet->inet_saddr = fl4->saddr;
192	sk_rcv_saddr_set(sk, inet->inet_saddr);
193
194	if (tp->rx_opt.ts_recent_stamp && inet->inet_daddr != daddr) {
195		/* Reset inherited state */
196		tp->rx_opt.ts_recent	   = 0;
197		tp->rx_opt.ts_recent_stamp = 0;
198		if (likely(!tp->repair))
199			tp->write_seq	   = 0;
200	}
201
202	if (tcp_death_row.sysctl_tw_recycle &&
203	    !tp->rx_opt.ts_recent_stamp && fl4->daddr == daddr)
204		tcp_fetch_timewait_stamp(sk, &rt->dst);
205
206	inet->inet_dport = usin->sin_port;
207	sk_daddr_set(sk, daddr);
208
209	inet_csk(sk)->icsk_ext_hdr_len = 0;
210	if (inet_opt)
211		inet_csk(sk)->icsk_ext_hdr_len = inet_opt->opt.optlen;
212
213	tp->rx_opt.mss_clamp = TCP_MSS_DEFAULT;
214
215	/* Socket identity is still unknown (sport may be zero).
216	 * However we set state to SYN-SENT and not releasing socket
217	 * lock select source port, enter ourselves into the hash tables and
218	 * complete initialization after this.
219	 */
220	tcp_set_state(sk, TCP_SYN_SENT);
221	err = inet_hash_connect(&tcp_death_row, sk);
222	if (err)
223		goto failure;
224
225	inet_set_txhash(sk);
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
254failure:
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}
265EXPORT_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 */
272void 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}
303EXPORT_SYMBOL(tcp_v4_mtu_reduced);
304
305static void do_redirect(struct sk_buff *skb, struct sock *sk)
306{
307	struct dst_entry *dst = __sk_dst_check(sk, 0);
308
309	if (dst)
310		dst->ops->redirect(dst, sk, skb);
311}
312
313
314/* handle ICMP messages on TCP_NEW_SYN_RECV request sockets */
315void tcp_req_err(struct sock *sk, u32 seq)
316{
317	struct request_sock *req = inet_reqsk(sk);
318	struct net *net = sock_net(sk);
319
320	/* ICMPs are not backlogged, hence we cannot get
321	 * an established socket here.
322	 */
323	WARN_ON(req->sk);
324
325	if (seq != tcp_rsk(req)->snt_isn) {
326		NET_INC_STATS_BH(net, LINUX_MIB_OUTOFWINDOWICMPS);
327		reqsk_put(req);
328	} else {
329		/*
330		 * Still in SYN_RECV, just remove it silently.
331		 * There is no good way to pass the error to the newly
332		 * created socket, and POSIX does not want network
333		 * errors returned from accept().
334		 */
335		NET_INC_STATS_BH(net, LINUX_MIB_LISTENDROPS);
336		inet_csk_reqsk_queue_drop(req->rsk_listener, req);
337	}
338}
339EXPORT_SYMBOL(tcp_req_err);
340
341/*
342 * This routine is called by the ICMP module when it gets some
343 * sort of error condition.  If err < 0 then the socket should
344 * be closed and the error returned to the user.  If err > 0
345 * it's just the icmp type << 8 | icmp code.  After adjustment
346 * header points to the first 8 bytes of the tcp header.  We need
347 * to find the appropriate port.
348 *
349 * The locking strategy used here is very "optimistic". When
350 * someone else accesses the socket the ICMP is just dropped
351 * and for some paths there is no check at all.
352 * A more general error queue to queue errors for later handling
353 * is probably better.
354 *
355 */
356
357void tcp_v4_err(struct sk_buff *icmp_skb, u32 info)
358{
359	const struct iphdr *iph = (const struct iphdr *)icmp_skb->data;
360	struct tcphdr *th = (struct tcphdr *)(icmp_skb->data + (iph->ihl << 2));
361	struct inet_connection_sock *icsk;
362	struct tcp_sock *tp;
363	struct inet_sock *inet;
364	const int type = icmp_hdr(icmp_skb)->type;
365	const int code = icmp_hdr(icmp_skb)->code;
366	struct sock *sk;
367	struct sk_buff *skb;
368	struct request_sock *fastopen;
369	__u32 seq, snd_una;
370	__u32 remaining;
371	int err;
372	struct net *net = dev_net(icmp_skb->dev);
373
374	sk = __inet_lookup_established(net, &tcp_hashinfo, iph->daddr,
375				       th->dest, iph->saddr, ntohs(th->source),
376				       inet_iif(icmp_skb));
377	if (!sk) {
378		ICMP_INC_STATS_BH(net, ICMP_MIB_INERRORS);
379		return;
380	}
381	if (sk->sk_state == TCP_TIME_WAIT) {
382		inet_twsk_put(inet_twsk(sk));
383		return;
384	}
385	seq = ntohl(th->seq);
386	if (sk->sk_state == TCP_NEW_SYN_RECV)
387		return tcp_req_err(sk, seq);
388
389	bh_lock_sock(sk);
390	/* If too many ICMPs get dropped on busy
391	 * servers this needs to be solved differently.
392	 * We do take care of PMTU discovery (RFC1191) special case :
393	 * we can receive locally generated ICMP messages while socket is held.
394	 */
395	if (sock_owned_by_user(sk)) {
396		if (!(type == ICMP_DEST_UNREACH && code == ICMP_FRAG_NEEDED))
397			NET_INC_STATS_BH(net, LINUX_MIB_LOCKDROPPEDICMPS);
398	}
399	if (sk->sk_state == TCP_CLOSE)
400		goto out;
401
402	if (unlikely(iph->ttl < inet_sk(sk)->min_ttl)) {
403		NET_INC_STATS_BH(net, LINUX_MIB_TCPMINTTLDROP);
404		goto out;
405	}
406
407	icsk = inet_csk(sk);
408	tp = tcp_sk(sk);
409	/* XXX (TFO) - tp->snd_una should be ISN (tcp_create_openreq_child() */
410	fastopen = tp->fastopen_rsk;
411	snd_una = fastopen ? tcp_rsk(fastopen)->snt_isn : tp->snd_una;
412	if (sk->sk_state != TCP_LISTEN &&
413	    !between(seq, snd_una, tp->snd_nxt)) {
414		NET_INC_STATS_BH(net, LINUX_MIB_OUTOFWINDOWICMPS);
415		goto out;
416	}
417
418	switch (type) {
419	case ICMP_REDIRECT:
420		do_redirect(icmp_skb, sk);
421		goto out;
422	case ICMP_SOURCE_QUENCH:
423		/* Just silently ignore these. */
424		goto out;
425	case ICMP_PARAMETERPROB:
426		err = EPROTO;
427		break;
428	case ICMP_DEST_UNREACH:
429		if (code > NR_ICMP_UNREACH)
430			goto out;
431
432		if (code == ICMP_FRAG_NEEDED) { /* PMTU discovery (RFC1191) */
433			/* We are not interested in TCP_LISTEN and open_requests
434			 * (SYN-ACKs send out by Linux are always <576bytes so
435			 * they should go through unfragmented).
436			 */
437			if (sk->sk_state == TCP_LISTEN)
438				goto out;
439
440			tp->mtu_info = info;
441			if (!sock_owned_by_user(sk)) {
442				tcp_v4_mtu_reduced(sk);
443			} else {
444				if (!test_and_set_bit(TCP_MTU_REDUCED_DEFERRED, &tp->tsq_flags))
445					sock_hold(sk);
446			}
447			goto out;
448		}
449
450		err = icmp_err_convert[code].errno;
451		/* check if icmp_skb allows revert of backoff
452		 * (see draft-zimmermann-tcp-lcd) */
453		if (code != ICMP_NET_UNREACH && code != ICMP_HOST_UNREACH)
454			break;
455		if (seq != tp->snd_una  || !icsk->icsk_retransmits ||
456		    !icsk->icsk_backoff || fastopen)
457			break;
458
459		if (sock_owned_by_user(sk))
460			break;
461
462		icsk->icsk_backoff--;
463		icsk->icsk_rto = tp->srtt_us ? __tcp_set_rto(tp) :
464					       TCP_TIMEOUT_INIT;
465		icsk->icsk_rto = inet_csk_rto_backoff(icsk, TCP_RTO_MAX);
466
467		skb = tcp_write_queue_head(sk);
468		BUG_ON(!skb);
469
470		remaining = icsk->icsk_rto -
471			    min(icsk->icsk_rto,
472				tcp_time_stamp - tcp_skb_timestamp(skb));
473
474		if (remaining) {
475			inet_csk_reset_xmit_timer(sk, ICSK_TIME_RETRANS,
476						  remaining, TCP_RTO_MAX);
477		} else {
478			/* RTO revert clocked out retransmission.
479			 * Will retransmit now */
480			tcp_retransmit_timer(sk);
481		}
482
483		break;
484	case ICMP_TIME_EXCEEDED:
485		err = EHOSTUNREACH;
486		break;
487	default:
488		goto out;
489	}
490
491	switch (sk->sk_state) {
492	case TCP_SYN_SENT:
493	case TCP_SYN_RECV:
494		/* Only in fast or simultaneous open. If a fast open socket is
495		 * is already accepted it is treated as a connected one below.
496		 */
497		if (fastopen && !fastopen->sk)
498			break;
499
500		if (!sock_owned_by_user(sk)) {
501			sk->sk_err = err;
502
503			sk->sk_error_report(sk);
504
505			tcp_done(sk);
506		} else {
507			sk->sk_err_soft = err;
508		}
509		goto out;
510	}
511
512	/* If we've already connected we will keep trying
513	 * until we time out, or the user gives up.
514	 *
515	 * rfc1122 4.2.3.9 allows to consider as hard errors
516	 * only PROTO_UNREACH and PORT_UNREACH (well, FRAG_FAILED too,
517	 * but it is obsoleted by pmtu discovery).
518	 *
519	 * Note, that in modern internet, where routing is unreliable
520	 * and in each dark corner broken firewalls sit, sending random
521	 * errors ordered by their masters even this two messages finally lose
522	 * their original sense (even Linux sends invalid PORT_UNREACHs)
523	 *
524	 * Now we are in compliance with RFCs.
525	 *							--ANK (980905)
526	 */
527
528	inet = inet_sk(sk);
529	if (!sock_owned_by_user(sk) && inet->recverr) {
530		sk->sk_err = err;
531		sk->sk_error_report(sk);
532	} else	{ /* Only an error on timeout */
533		sk->sk_err_soft = err;
534	}
535
536out:
537	bh_unlock_sock(sk);
538	sock_put(sk);
539}
540
541void __tcp_v4_send_check(struct sk_buff *skb, __be32 saddr, __be32 daddr)
542{
543	struct tcphdr *th = tcp_hdr(skb);
544
545	if (skb->ip_summed == CHECKSUM_PARTIAL) {
546		th->check = ~tcp_v4_check(skb->len, saddr, daddr, 0);
547		skb->csum_start = skb_transport_header(skb) - skb->head;
548		skb->csum_offset = offsetof(struct tcphdr, check);
549	} else {
550		th->check = tcp_v4_check(skb->len, saddr, daddr,
551					 csum_partial(th,
552						      th->doff << 2,
553						      skb->csum));
554	}
555}
556
557/* This routine computes an IPv4 TCP checksum. */
558void tcp_v4_send_check(struct sock *sk, struct sk_buff *skb)
559{
560	const struct inet_sock *inet = inet_sk(sk);
561
562	__tcp_v4_send_check(skb, inet->inet_saddr, inet->inet_daddr);
563}
564EXPORT_SYMBOL(tcp_v4_send_check);
565
566/*
567 *	This routine will send an RST to the other tcp.
568 *
569 *	Someone asks: why I NEVER use socket parameters (TOS, TTL etc.)
570 *		      for reset.
571 *	Answer: if a packet caused RST, it is not for a socket
572 *		existing in our system, if it is matched to a socket,
573 *		it is just duplicate segment or bug in other side's TCP.
574 *		So that we build reply only basing on parameters
575 *		arrived with segment.
576 *	Exception: precedence violation. We do not implement it in any case.
577 */
578
579static void tcp_v4_send_reset(struct sock *sk, struct sk_buff *skb)
580{
581	const struct tcphdr *th = tcp_hdr(skb);
582	struct {
583		struct tcphdr th;
584#ifdef CONFIG_TCP_MD5SIG
585		__be32 opt[(TCPOLEN_MD5SIG_ALIGNED >> 2)];
586#endif
587	} rep;
588	struct ip_reply_arg arg;
589#ifdef CONFIG_TCP_MD5SIG
590	struct tcp_md5sig_key *key;
591	const __u8 *hash_location = NULL;
592	unsigned char newhash[16];
593	int genhash;
594	struct sock *sk1 = NULL;
595#endif
596	struct net *net;
597
598	/* Never send a reset in response to a reset. */
599	if (th->rst)
600		return;
601
602	/* If sk not NULL, it means we did a successful lookup and incoming
603	 * route had to be correct. prequeue might have dropped our dst.
604	 */
605	if (!sk && skb_rtable(skb)->rt_type != RTN_LOCAL)
606		return;
607
608	/* Swap the send and the receive. */
609	memset(&rep, 0, sizeof(rep));
610	rep.th.dest   = th->source;
611	rep.th.source = th->dest;
612	rep.th.doff   = sizeof(struct tcphdr) / 4;
613	rep.th.rst    = 1;
614
615	if (th->ack) {
616		rep.th.seq = th->ack_seq;
617	} else {
618		rep.th.ack = 1;
619		rep.th.ack_seq = htonl(ntohl(th->seq) + th->syn + th->fin +
620				       skb->len - (th->doff << 2));
621	}
622
623	memset(&arg, 0, sizeof(arg));
624	arg.iov[0].iov_base = (unsigned char *)&rep;
625	arg.iov[0].iov_len  = sizeof(rep.th);
626
627	net = sk ? sock_net(sk) : dev_net(skb_dst(skb)->dev);
628#ifdef CONFIG_TCP_MD5SIG
629	hash_location = tcp_parse_md5sig_option(th);
630	if (!sk && hash_location) {
631		/*
632		 * active side is lost. Try to find listening socket through
633		 * source port, and then find md5 key through listening socket.
634		 * we are not loose security here:
635		 * Incoming packet is checked with md5 hash with finding key,
636		 * no RST generated if md5 hash doesn't match.
637		 */
638		sk1 = __inet_lookup_listener(net,
639					     &tcp_hashinfo, ip_hdr(skb)->saddr,
640					     th->source, ip_hdr(skb)->daddr,
641					     ntohs(th->source), inet_iif(skb));
642		/* don't send rst if it can't find key */
643		if (!sk1)
644			return;
645		rcu_read_lock();
646		key = tcp_md5_do_lookup(sk1, (union tcp_md5_addr *)
647					&ip_hdr(skb)->saddr, AF_INET);
648		if (!key)
649			goto release_sk1;
650
651		genhash = tcp_v4_md5_hash_skb(newhash, key, NULL, skb);
652		if (genhash || memcmp(hash_location, newhash, 16) != 0)
653			goto release_sk1;
654	} else {
655		key = sk ? tcp_md5_do_lookup(sk, (union tcp_md5_addr *)
656					     &ip_hdr(skb)->saddr,
657					     AF_INET) : NULL;
658	}
659
660	if (key) {
661		rep.opt[0] = htonl((TCPOPT_NOP << 24) |
662				   (TCPOPT_NOP << 16) |
663				   (TCPOPT_MD5SIG << 8) |
664				   TCPOLEN_MD5SIG);
665		/* Update length and the length the header thinks exists */
666		arg.iov[0].iov_len += TCPOLEN_MD5SIG_ALIGNED;
667		rep.th.doff = arg.iov[0].iov_len / 4;
668
669		tcp_v4_md5_hash_hdr((__u8 *) &rep.opt[1],
670				     key, ip_hdr(skb)->saddr,
671				     ip_hdr(skb)->daddr, &rep.th);
672	}
673#endif
674	arg.csum = csum_tcpudp_nofold(ip_hdr(skb)->daddr,
675				      ip_hdr(skb)->saddr, /* XXX */
676				      arg.iov[0].iov_len, IPPROTO_TCP, 0);
677	arg.csumoffset = offsetof(struct tcphdr, check) / 2;
678	arg.flags = (sk && inet_sk(sk)->transparent) ? IP_REPLY_ARG_NOSRCCHECK : 0;
679	/* When socket is gone, all binding information is lost.
680	 * routing might fail in this case. No choice here, if we choose to force
681	 * input interface, we will misroute in case of asymmetric route.
682	 */
683	if (sk)
684		arg.bound_dev_if = sk->sk_bound_dev_if;
685
686	arg.tos = ip_hdr(skb)->tos;
687	ip_send_unicast_reply(*this_cpu_ptr(net->ipv4.tcp_sk),
688			      skb, &TCP_SKB_CB(skb)->header.h4.opt,
689			      ip_hdr(skb)->saddr, ip_hdr(skb)->daddr,
690			      &arg, arg.iov[0].iov_len);
691
692	TCP_INC_STATS_BH(net, TCP_MIB_OUTSEGS);
693	TCP_INC_STATS_BH(net, TCP_MIB_OUTRSTS);
694
695#ifdef CONFIG_TCP_MD5SIG
696release_sk1:
697	if (sk1) {
698		rcu_read_unlock();
699		sock_put(sk1);
700	}
701#endif
702}
703
704/* The code following below sending ACKs in SYN-RECV and TIME-WAIT states
705   outside socket context is ugly, certainly. What can I do?
706 */
707
708static void tcp_v4_send_ack(struct net *net,
709			    struct sk_buff *skb, u32 seq, u32 ack,
710			    u32 win, u32 tsval, u32 tsecr, int oif,
711			    struct tcp_md5sig_key *key,
712			    int reply_flags, u8 tos)
713{
714	const struct tcphdr *th = tcp_hdr(skb);
715	struct {
716		struct tcphdr th;
717		__be32 opt[(TCPOLEN_TSTAMP_ALIGNED >> 2)
718#ifdef CONFIG_TCP_MD5SIG
719			   + (TCPOLEN_MD5SIG_ALIGNED >> 2)
720#endif
721			];
722	} rep;
723	struct ip_reply_arg arg;
724
725	memset(&rep.th, 0, sizeof(struct tcphdr));
726	memset(&arg, 0, sizeof(arg));
727
728	arg.iov[0].iov_base = (unsigned char *)&rep;
729	arg.iov[0].iov_len  = sizeof(rep.th);
730	if (tsecr) {
731		rep.opt[0] = htonl((TCPOPT_NOP << 24) | (TCPOPT_NOP << 16) |
732				   (TCPOPT_TIMESTAMP << 8) |
733				   TCPOLEN_TIMESTAMP);
734		rep.opt[1] = htonl(tsval);
735		rep.opt[2] = htonl(tsecr);
736		arg.iov[0].iov_len += TCPOLEN_TSTAMP_ALIGNED;
737	}
738
739	/* Swap the send and the receive. */
740	rep.th.dest    = th->source;
741	rep.th.source  = th->dest;
742	rep.th.doff    = arg.iov[0].iov_len / 4;
743	rep.th.seq     = htonl(seq);
744	rep.th.ack_seq = htonl(ack);
745	rep.th.ack     = 1;
746	rep.th.window  = htons(win);
747
748#ifdef CONFIG_TCP_MD5SIG
749	if (key) {
750		int offset = (tsecr) ? 3 : 0;
751
752		rep.opt[offset++] = htonl((TCPOPT_NOP << 24) |
753					  (TCPOPT_NOP << 16) |
754					  (TCPOPT_MD5SIG << 8) |
755					  TCPOLEN_MD5SIG);
756		arg.iov[0].iov_len += TCPOLEN_MD5SIG_ALIGNED;
757		rep.th.doff = arg.iov[0].iov_len/4;
758
759		tcp_v4_md5_hash_hdr((__u8 *) &rep.opt[offset],
760				    key, ip_hdr(skb)->saddr,
761				    ip_hdr(skb)->daddr, &rep.th);
762	}
763#endif
764	arg.flags = reply_flags;
765	arg.csum = csum_tcpudp_nofold(ip_hdr(skb)->daddr,
766				      ip_hdr(skb)->saddr, /* XXX */
767				      arg.iov[0].iov_len, IPPROTO_TCP, 0);
768	arg.csumoffset = offsetof(struct tcphdr, check) / 2;
769	if (oif)
770		arg.bound_dev_if = oif;
771	arg.tos = tos;
772	ip_send_unicast_reply(*this_cpu_ptr(net->ipv4.tcp_sk),
773			      skb, &TCP_SKB_CB(skb)->header.h4.opt,
774			      ip_hdr(skb)->saddr, ip_hdr(skb)->daddr,
775			      &arg, arg.iov[0].iov_len);
776
777	TCP_INC_STATS_BH(net, TCP_MIB_OUTSEGS);
778}
779
780static void tcp_v4_timewait_ack(struct sock *sk, struct sk_buff *skb)
781{
782	struct inet_timewait_sock *tw = inet_twsk(sk);
783	struct tcp_timewait_sock *tcptw = tcp_twsk(sk);
784
785	tcp_v4_send_ack(sock_net(sk), skb,
786			tcptw->tw_snd_nxt, tcptw->tw_rcv_nxt,
787			tcptw->tw_rcv_wnd >> tw->tw_rcv_wscale,
788			tcp_time_stamp + tcptw->tw_ts_offset,
789			tcptw->tw_ts_recent,
790			tw->tw_bound_dev_if,
791			tcp_twsk_md5_key(tcptw),
792			tw->tw_transparent ? IP_REPLY_ARG_NOSRCCHECK : 0,
793			tw->tw_tos
794			);
795
796	inet_twsk_put(tw);
797}
798
799static void tcp_v4_reqsk_send_ack(struct sock *sk, struct sk_buff *skb,
800				  struct request_sock *req)
801{
802	/* sk->sk_state == TCP_LISTEN -> for regular TCP_SYN_RECV
803	 * sk->sk_state == TCP_SYN_RECV -> for Fast Open.
804	 */
805	u32 seq = (sk->sk_state == TCP_LISTEN) ? tcp_rsk(req)->snt_isn + 1 :
806					     tcp_sk(sk)->snd_nxt;
807
808	tcp_v4_send_ack(sock_net(sk), skb, seq,
809			tcp_rsk(req)->rcv_nxt, req->rcv_wnd,
810			tcp_time_stamp,
811			req->ts_recent,
812			0,
813			tcp_md5_do_lookup(sk, (union tcp_md5_addr *)&ip_hdr(skb)->daddr,
814					  AF_INET),
815			inet_rsk(req)->no_srccheck ? IP_REPLY_ARG_NOSRCCHECK : 0,
816			ip_hdr(skb)->tos);
817}
818
819/*
820 *	Send a SYN-ACK after having received a SYN.
821 *	This still operates on a request_sock only, not on a big
822 *	socket.
823 */
824static int tcp_v4_send_synack(struct sock *sk, struct dst_entry *dst,
825			      struct flowi *fl,
826			      struct request_sock *req,
827			      u16 queue_mapping,
828			      struct tcp_fastopen_cookie *foc)
829{
830	const struct inet_request_sock *ireq = inet_rsk(req);
831	struct flowi4 fl4;
832	int err = -1;
833	struct sk_buff *skb;
834
835	/* First, grab a route. */
836	if (!dst && (dst = inet_csk_route_req(sk, &fl4, req)) == NULL)
837		return -1;
838
839	skb = tcp_make_synack(sk, dst, req, foc);
840
841	if (skb) {
842		__tcp_v4_send_check(skb, ireq->ir_loc_addr, ireq->ir_rmt_addr);
843
844		skb_set_queue_mapping(skb, queue_mapping);
845		err = ip_build_and_send_pkt(skb, sk, ireq->ir_loc_addr,
846					    ireq->ir_rmt_addr,
847					    ireq->opt);
848		err = net_xmit_eval(err);
849	}
850
851	return err;
852}
853
854/*
855 *	IPv4 request_sock destructor.
856 */
857static void tcp_v4_reqsk_destructor(struct request_sock *req)
858{
859	kfree(inet_rsk(req)->opt);
860}
861
862
863#ifdef CONFIG_TCP_MD5SIG
864/*
865 * RFC2385 MD5 checksumming requires a mapping of
866 * IP address->MD5 Key.
867 * We need to maintain these in the sk structure.
868 */
869
870/* Find the Key structure for an address.  */
871struct tcp_md5sig_key *tcp_md5_do_lookup(struct sock *sk,
872					 const union tcp_md5_addr *addr,
873					 int family)
874{
875	const struct tcp_sock *tp = tcp_sk(sk);
876	struct tcp_md5sig_key *key;
877	unsigned int size = sizeof(struct in_addr);
878	const struct tcp_md5sig_info *md5sig;
879
880	/* caller either holds rcu_read_lock() or socket lock */
881	md5sig = rcu_dereference_check(tp->md5sig_info,
882				       sock_owned_by_user(sk) ||
883				       lockdep_is_held(&sk->sk_lock.slock));
884	if (!md5sig)
885		return NULL;
886#if IS_ENABLED(CONFIG_IPV6)
887	if (family == AF_INET6)
888		size = sizeof(struct in6_addr);
889#endif
890	hlist_for_each_entry_rcu(key, &md5sig->head, node) {
891		if (key->family != family)
892			continue;
893		if (!memcmp(&key->addr, addr, size))
894			return key;
895	}
896	return NULL;
897}
898EXPORT_SYMBOL(tcp_md5_do_lookup);
899
900struct tcp_md5sig_key *tcp_v4_md5_lookup(struct sock *sk,
901					 const struct sock *addr_sk)
902{
903	const union tcp_md5_addr *addr;
904
905	addr = (const union tcp_md5_addr *)&addr_sk->sk_daddr;
906	return tcp_md5_do_lookup(sk, addr, AF_INET);
907}
908EXPORT_SYMBOL(tcp_v4_md5_lookup);
909
910/* This can be called on a newly created socket, from other files */
911int tcp_md5_do_add(struct sock *sk, const union tcp_md5_addr *addr,
912		   int family, const u8 *newkey, u8 newkeylen, gfp_t gfp)
913{
914	/* Add Key to the list */
915	struct tcp_md5sig_key *key;
916	struct tcp_sock *tp = tcp_sk(sk);
917	struct tcp_md5sig_info *md5sig;
918
919	key = tcp_md5_do_lookup(sk, addr, family);
920	if (key) {
921		/* Pre-existing entry - just update that one. */
922		memcpy(key->key, newkey, newkeylen);
923		key->keylen = newkeylen;
924		return 0;
925	}
926
927	md5sig = rcu_dereference_protected(tp->md5sig_info,
928					   sock_owned_by_user(sk) ||
929					   lockdep_is_held(&sk->sk_lock.slock));
930	if (!md5sig) {
931		md5sig = kmalloc(sizeof(*md5sig), gfp);
932		if (!md5sig)
933			return -ENOMEM;
934
935		sk_nocaps_add(sk, NETIF_F_GSO_MASK);
936		INIT_HLIST_HEAD(&md5sig->head);
937		rcu_assign_pointer(tp->md5sig_info, md5sig);
938	}
939
940	key = sock_kmalloc(sk, sizeof(*key), gfp);
941	if (!key)
942		return -ENOMEM;
943	if (!tcp_alloc_md5sig_pool()) {
944		sock_kfree_s(sk, key, sizeof(*key));
945		return -ENOMEM;
946	}
947
948	memcpy(key->key, newkey, newkeylen);
949	key->keylen = newkeylen;
950	key->family = family;
951	memcpy(&key->addr, addr,
952	       (family == AF_INET6) ? sizeof(struct in6_addr) :
953				      sizeof(struct in_addr));
954	hlist_add_head_rcu(&key->node, &md5sig->head);
955	return 0;
956}
957EXPORT_SYMBOL(tcp_md5_do_add);
958
959int tcp_md5_do_del(struct sock *sk, const union tcp_md5_addr *addr, int family)
960{
961	struct tcp_md5sig_key *key;
962
963	key = tcp_md5_do_lookup(sk, addr, family);
964	if (!key)
965		return -ENOENT;
966	hlist_del_rcu(&key->node);
967	atomic_sub(sizeof(*key), &sk->sk_omem_alloc);
968	kfree_rcu(key, rcu);
969	return 0;
970}
971EXPORT_SYMBOL(tcp_md5_do_del);
972
973static void tcp_clear_md5_list(struct sock *sk)
974{
975	struct tcp_sock *tp = tcp_sk(sk);
976	struct tcp_md5sig_key *key;
977	struct hlist_node *n;
978	struct tcp_md5sig_info *md5sig;
979
980	md5sig = rcu_dereference_protected(tp->md5sig_info, 1);
981
982	hlist_for_each_entry_safe(key, n, &md5sig->head, node) {
983		hlist_del_rcu(&key->node);
984		atomic_sub(sizeof(*key), &sk->sk_omem_alloc);
985		kfree_rcu(key, rcu);
986	}
987}
988
989static int tcp_v4_parse_md5_keys(struct sock *sk, char __user *optval,
990				 int optlen)
991{
992	struct tcp_md5sig cmd;
993	struct sockaddr_in *sin = (struct sockaddr_in *)&cmd.tcpm_addr;
994
995	if (optlen < sizeof(cmd))
996		return -EINVAL;
997
998	if (copy_from_user(&cmd, optval, sizeof(cmd)))
999		return -EFAULT;
1000
1001	if (sin->sin_family != AF_INET)
1002		return -EINVAL;
1003
1004	if (!cmd.tcpm_keylen)
1005		return tcp_md5_do_del(sk, (union tcp_md5_addr *)&sin->sin_addr.s_addr,
1006				      AF_INET);
1007
1008	if (cmd.tcpm_keylen > TCP_MD5SIG_MAXKEYLEN)
1009		return -EINVAL;
1010
1011	return tcp_md5_do_add(sk, (union tcp_md5_addr *)&sin->sin_addr.s_addr,
1012			      AF_INET, cmd.tcpm_key, cmd.tcpm_keylen,
1013			      GFP_KERNEL);
1014}
1015
1016static int tcp_v4_md5_hash_pseudoheader(struct tcp_md5sig_pool *hp,
1017					__be32 daddr, __be32 saddr, int nbytes)
1018{
1019	struct tcp4_pseudohdr *bp;
1020	struct scatterlist sg;
1021
1022	bp = &hp->md5_blk.ip4;
1023
1024	/*
1025	 * 1. the TCP pseudo-header (in the order: source IP address,
1026	 * destination IP address, zero-padded protocol number, and
1027	 * segment length)
1028	 */
1029	bp->saddr = saddr;
1030	bp->daddr = daddr;
1031	bp->pad = 0;
1032	bp->protocol = IPPROTO_TCP;
1033	bp->len = cpu_to_be16(nbytes);
1034
1035	sg_init_one(&sg, bp, sizeof(*bp));
1036	return crypto_hash_update(&hp->md5_desc, &sg, sizeof(*bp));
1037}
1038
1039static int tcp_v4_md5_hash_hdr(char *md5_hash, const struct tcp_md5sig_key *key,
1040			       __be32 daddr, __be32 saddr, const struct tcphdr *th)
1041{
1042	struct tcp_md5sig_pool *hp;
1043	struct hash_desc *desc;
1044
1045	hp = tcp_get_md5sig_pool();
1046	if (!hp)
1047		goto clear_hash_noput;
1048	desc = &hp->md5_desc;
1049
1050	if (crypto_hash_init(desc))
1051		goto clear_hash;
1052	if (tcp_v4_md5_hash_pseudoheader(hp, daddr, saddr, th->doff << 2))
1053		goto clear_hash;
1054	if (tcp_md5_hash_header(hp, th))
1055		goto clear_hash;
1056	if (tcp_md5_hash_key(hp, key))
1057		goto clear_hash;
1058	if (crypto_hash_final(desc, md5_hash))
1059		goto clear_hash;
1060
1061	tcp_put_md5sig_pool();
1062	return 0;
1063
1064clear_hash:
1065	tcp_put_md5sig_pool();
1066clear_hash_noput:
1067	memset(md5_hash, 0, 16);
1068	return 1;
1069}
1070
1071int tcp_v4_md5_hash_skb(char *md5_hash, const struct tcp_md5sig_key *key,
1072			const struct sock *sk,
1073			const struct sk_buff *skb)
1074{
1075	struct tcp_md5sig_pool *hp;
1076	struct hash_desc *desc;
1077	const struct tcphdr *th = tcp_hdr(skb);
1078	__be32 saddr, daddr;
1079
1080	if (sk) { /* valid for establish/request sockets */
1081		saddr = sk->sk_rcv_saddr;
1082		daddr = sk->sk_daddr;
1083	} else {
1084		const struct iphdr *iph = ip_hdr(skb);
1085		saddr = iph->saddr;
1086		daddr = iph->daddr;
1087	}
1088
1089	hp = tcp_get_md5sig_pool();
1090	if (!hp)
1091		goto clear_hash_noput;
1092	desc = &hp->md5_desc;
1093
1094	if (crypto_hash_init(desc))
1095		goto clear_hash;
1096
1097	if (tcp_v4_md5_hash_pseudoheader(hp, daddr, saddr, skb->len))
1098		goto clear_hash;
1099	if (tcp_md5_hash_header(hp, th))
1100		goto clear_hash;
1101	if (tcp_md5_hash_skb_data(hp, skb, th->doff << 2))
1102		goto clear_hash;
1103	if (tcp_md5_hash_key(hp, key))
1104		goto clear_hash;
1105	if (crypto_hash_final(desc, md5_hash))
1106		goto clear_hash;
1107
1108	tcp_put_md5sig_pool();
1109	return 0;
1110
1111clear_hash:
1112	tcp_put_md5sig_pool();
1113clear_hash_noput:
1114	memset(md5_hash, 0, 16);
1115	return 1;
1116}
1117EXPORT_SYMBOL(tcp_v4_md5_hash_skb);
1118
1119/* Called with rcu_read_lock() */
1120static bool tcp_v4_inbound_md5_hash(struct sock *sk,
1121				    const struct sk_buff *skb)
1122{
1123	/*
1124	 * This gets called for each TCP segment that arrives
1125	 * so we want to be efficient.
1126	 * We have 3 drop cases:
1127	 * o No MD5 hash and one expected.
1128	 * o MD5 hash and we're not expecting one.
1129	 * o MD5 hash and its wrong.
1130	 */
1131	const __u8 *hash_location = NULL;
1132	struct tcp_md5sig_key *hash_expected;
1133	const struct iphdr *iph = ip_hdr(skb);
1134	const struct tcphdr *th = tcp_hdr(skb);
1135	int genhash;
1136	unsigned char newhash[16];
1137
1138	hash_expected = tcp_md5_do_lookup(sk, (union tcp_md5_addr *)&iph->saddr,
1139					  AF_INET);
1140	hash_location = tcp_parse_md5sig_option(th);
1141
1142	/* We've parsed the options - do we have a hash? */
1143	if (!hash_expected && !hash_location)
1144		return false;
1145
1146	if (hash_expected && !hash_location) {
1147		NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_TCPMD5NOTFOUND);
1148		return true;
1149	}
1150
1151	if (!hash_expected && hash_location) {
1152		NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_TCPMD5UNEXPECTED);
1153		return true;
1154	}
1155
1156	/* Okay, so this is hash_expected and hash_location -
1157	 * so we need to calculate the checksum.
1158	 */
1159	genhash = tcp_v4_md5_hash_skb(newhash,
1160				      hash_expected,
1161				      NULL, skb);
1162
1163	if (genhash || memcmp(hash_location, newhash, 16) != 0) {
1164		net_info_ratelimited("MD5 Hash failed for (%pI4, %d)->(%pI4, %d)%s\n",
1165				     &iph->saddr, ntohs(th->source),
1166				     &iph->daddr, ntohs(th->dest),
1167				     genhash ? " tcp_v4_calc_md5_hash failed"
1168				     : "");
1169		return true;
1170	}
1171	return false;
1172}
1173#endif
1174
1175static void tcp_v4_init_req(struct request_sock *req, struct sock *sk_listener,
1176			    struct sk_buff *skb)
1177{
1178	struct inet_request_sock *ireq = inet_rsk(req);
1179
1180	sk_rcv_saddr_set(req_to_sk(req), ip_hdr(skb)->daddr);
1181	sk_daddr_set(req_to_sk(req), ip_hdr(skb)->saddr);
1182	ireq->no_srccheck = inet_sk(sk_listener)->transparent;
1183	ireq->opt = tcp_v4_save_options(skb);
1184}
1185
1186static struct dst_entry *tcp_v4_route_req(struct sock *sk, struct flowi *fl,
1187					  const struct request_sock *req,
1188					  bool *strict)
1189{
1190	struct dst_entry *dst = inet_csk_route_req(sk, &fl->u.ip4, req);
1191
1192	if (strict) {
1193		if (fl->u.ip4.daddr == inet_rsk(req)->ir_rmt_addr)
1194			*strict = true;
1195		else
1196			*strict = false;
1197	}
1198
1199	return dst;
1200}
1201
1202struct request_sock_ops tcp_request_sock_ops __read_mostly = {
1203	.family		=	PF_INET,
1204	.obj_size	=	sizeof(struct tcp_request_sock),
1205	.rtx_syn_ack	=	tcp_rtx_synack,
1206	.send_ack	=	tcp_v4_reqsk_send_ack,
1207	.destructor	=	tcp_v4_reqsk_destructor,
1208	.send_reset	=	tcp_v4_send_reset,
1209	.syn_ack_timeout =	tcp_syn_ack_timeout,
1210};
1211
1212static const struct tcp_request_sock_ops tcp_request_sock_ipv4_ops = {
1213	.mss_clamp	=	TCP_MSS_DEFAULT,
1214#ifdef CONFIG_TCP_MD5SIG
1215	.req_md5_lookup	=	tcp_v4_md5_lookup,
1216	.calc_md5_hash	=	tcp_v4_md5_hash_skb,
1217#endif
1218	.init_req	=	tcp_v4_init_req,
1219#ifdef CONFIG_SYN_COOKIES
1220	.cookie_init_seq =	cookie_v4_init_sequence,
1221#endif
1222	.route_req	=	tcp_v4_route_req,
1223	.init_seq	=	tcp_v4_init_sequence,
1224	.send_synack	=	tcp_v4_send_synack,
1225	.queue_hash_add =	inet_csk_reqsk_queue_hash_add,
1226};
1227
1228int tcp_v4_conn_request(struct sock *sk, struct sk_buff *skb)
1229{
1230	/* Never answer to SYNs send to broadcast or multicast */
1231	if (skb_rtable(skb)->rt_flags & (RTCF_BROADCAST | RTCF_MULTICAST))
1232		goto drop;
1233
1234	return tcp_conn_request(&tcp_request_sock_ops,
1235				&tcp_request_sock_ipv4_ops, sk, skb);
1236
1237drop:
1238	NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_LISTENDROPS);
1239	return 0;
1240}
1241EXPORT_SYMBOL(tcp_v4_conn_request);
1242
1243
1244/*
1245 * The three way handshake has completed - we got a valid synack -
1246 * now create the new socket.
1247 */
1248struct sock *tcp_v4_syn_recv_sock(struct sock *sk, struct sk_buff *skb,
1249				  struct request_sock *req,
1250				  struct dst_entry *dst)
1251{
1252	struct inet_request_sock *ireq;
1253	struct inet_sock *newinet;
1254	struct tcp_sock *newtp;
1255	struct sock *newsk;
1256#ifdef CONFIG_TCP_MD5SIG
1257	struct tcp_md5sig_key *key;
1258#endif
1259	struct ip_options_rcu *inet_opt;
1260
1261	if (sk_acceptq_is_full(sk))
1262		goto exit_overflow;
1263
1264	newsk = tcp_create_openreq_child(sk, req, skb);
1265	if (!newsk)
1266		goto exit_nonewsk;
1267
1268	newsk->sk_gso_type = SKB_GSO_TCPV4;
1269	inet_sk_rx_dst_set(newsk, skb);
1270
1271	newtp		      = tcp_sk(newsk);
1272	newinet		      = inet_sk(newsk);
1273	ireq		      = inet_rsk(req);
1274	sk_daddr_set(newsk, ireq->ir_rmt_addr);
1275	sk_rcv_saddr_set(newsk, ireq->ir_loc_addr);
1276	newinet->inet_saddr	      = ireq->ir_loc_addr;
1277	inet_opt	      = ireq->opt;
1278	rcu_assign_pointer(newinet->inet_opt, inet_opt);
1279	ireq->opt	      = NULL;
1280	newinet->mc_index     = inet_iif(skb);
1281	newinet->mc_ttl	      = ip_hdr(skb)->ttl;
1282	newinet->rcv_tos      = ip_hdr(skb)->tos;
1283	inet_csk(newsk)->icsk_ext_hdr_len = 0;
1284	inet_set_txhash(newsk);
1285	if (inet_opt)
1286		inet_csk(newsk)->icsk_ext_hdr_len = inet_opt->opt.optlen;
1287	newinet->inet_id = newtp->write_seq ^ jiffies;
1288
1289	if (!dst) {
1290		dst = inet_csk_route_child_sock(sk, newsk, req);
1291		if (!dst)
1292			goto put_and_exit;
1293	} else {
1294		/* syncookie case : see end of cookie_v4_check() */
1295	}
1296	sk_setup_caps(newsk, dst);
1297
1298	tcp_ca_openreq_child(newsk, dst);
1299
1300	tcp_sync_mss(newsk, dst_mtu(dst));
1301	newtp->advmss = dst_metric_advmss(dst);
1302	if (tcp_sk(sk)->rx_opt.user_mss &&
1303	    tcp_sk(sk)->rx_opt.user_mss < newtp->advmss)
1304		newtp->advmss = tcp_sk(sk)->rx_opt.user_mss;
1305
1306	tcp_initialize_rcv_mss(newsk);
1307
1308#ifdef CONFIG_TCP_MD5SIG
1309	/* Copy over the MD5 key from the original socket */
1310	key = tcp_md5_do_lookup(sk, (union tcp_md5_addr *)&newinet->inet_daddr,
1311				AF_INET);
1312	if (key) {
1313		/*
1314		 * We're using one, so create a matching key
1315		 * on the newsk structure. If we fail to get
1316		 * memory, then we end up not copying the key
1317		 * across. Shucks.
1318		 */
1319		tcp_md5_do_add(newsk, (union tcp_md5_addr *)&newinet->inet_daddr,
1320			       AF_INET, key->key, key->keylen, GFP_ATOMIC);
1321		sk_nocaps_add(newsk, NETIF_F_GSO_MASK);
1322	}
1323#endif
1324
1325	if (__inet_inherit_port(sk, newsk) < 0)
1326		goto put_and_exit;
1327	__inet_hash_nolisten(newsk, NULL);
1328
1329	return newsk;
1330
1331exit_overflow:
1332	NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_LISTENOVERFLOWS);
1333exit_nonewsk:
1334	dst_release(dst);
1335exit:
1336	NET_INC_STATS_BH(sock_net(sk), LINUX_MIB_LISTENDROPS);
1337	return NULL;
1338put_and_exit:
1339	inet_csk_prepare_forced_close(newsk);
1340	tcp_done(newsk);
1341	goto exit;
1342}
1343EXPORT_SYMBOL(tcp_v4_syn_recv_sock);
1344
1345static struct sock *tcp_v4_hnd_req(struct sock *sk, struct sk_buff *skb)
1346{
1347	const struct tcphdr *th = tcp_hdr(skb);
1348	const struct iphdr *iph = ip_hdr(skb);
1349	struct request_sock *req;
1350	struct sock *nsk;
1351
1352	req = inet_csk_search_req(sk, th->source, iph->saddr, iph->daddr);
1353	if (req) {
1354		nsk = tcp_check_req(sk, skb, req, false);
1355		if (!nsk || nsk == sk)
1356			reqsk_put(req);
1357		return nsk;
1358	}
1359
1360	nsk = inet_lookup_established(sock_net(sk), &tcp_hashinfo, iph->saddr,
1361			th->source, iph->daddr, th->dest, inet_iif(skb));
1362
1363	if (nsk) {
1364		if (nsk->sk_state != TCP_TIME_WAIT) {
1365			bh_lock_sock(nsk);
1366			return nsk;
1367		}
1368		inet_twsk_put(inet_twsk(nsk));
1369		return NULL;
1370	}
1371
1372#ifdef CONFIG_SYN_COOKIES
1373	if (!th->syn)
1374		sk = cookie_v4_check(sk, skb);
1375#endif
1376	return sk;
1377}
1378
1379/* The socket must have it's spinlock held when we get
1380 * here.
1381 *
1382 * We have a potential double-lock case here, so even when
1383 * doing backlog processing we use the BH locking scheme.
1384 * This is because we cannot sleep with the original spinlock
1385 * held.
1386 */
1387int tcp_v4_do_rcv(struct sock *sk, struct sk_buff *skb)
1388{
1389	struct sock *rsk;
1390
1391	if (sk->sk_state == TCP_ESTABLISHED) { /* Fast path */
1392		struct dst_entry *dst = sk->sk_rx_dst;
1393
1394		sock_rps_save_rxhash(sk, skb);
1395		sk_mark_napi_id(sk, skb);
1396		if (dst) {
1397			if (inet_sk(sk)->rx_dst_ifindex != skb->skb_iif ||
1398			    !dst->ops->check(dst, 0)) {
1399				dst_release(dst);
1400				sk->sk_rx_dst = NULL;
1401			}
1402		}
1403		tcp_rcv_established(sk, skb, tcp_hdr(skb), skb->len);
1404		return 0;
1405	}
1406
1407	if (skb->len < tcp_hdrlen(skb) || tcp_checksum_complete(skb))
1408		goto csum_err;
1409
1410	if (sk->sk_state == TCP_LISTEN) {
1411		struct sock *nsk = tcp_v4_hnd_req(sk, skb);
1412		if (!nsk)
1413			goto discard;
1414
1415		if (nsk != sk) {
1416			sock_rps_save_rxhash(nsk, skb);
1417			sk_mark_napi_id(sk, skb);
1418			if (tcp_child_process(sk, nsk, skb)) {
1419				rsk = nsk;
1420				goto reset;
1421			}
1422			return 0;
1423		}
1424	} else
1425		sock_rps_save_rxhash(sk, skb);
1426
1427	if (tcp_rcv_state_process(sk, skb, tcp_hdr(skb), skb->len)) {
1428		rsk = sk;
1429		goto reset;
1430	}
1431	return 0;
1432
1433reset:
1434	tcp_v4_send_reset(rsk, skb);
1435discard:
1436	kfree_skb(skb);
1437	/* Be careful here. If this function gets more complicated and
1438	 * gcc suffers from register pressure on the x86, sk (in %ebx)
1439	 * might be destroyed here. This current version compiles correctly,
1440	 * but you have been warned.
1441	 */
1442	return 0;
1443
1444csum_err:
1445	TCP_INC_STATS_BH(sock_net(sk), TCP_MIB_CSUMERRORS);
1446	TCP_INC_STATS_BH(sock_net(sk), TCP_MIB_INERRS);
1447	goto discard;
1448}
1449EXPORT_SYMBOL(tcp_v4_do_rcv);
1450
1451void tcp_v4_early_demux(struct sk_buff *skb)
1452{
1453	const struct iphdr *iph;
1454	const struct tcphdr *th;
1455	struct sock *sk;
1456
1457	if (skb->pkt_type != PACKET_HOST)
1458		return;
1459
1460	if (!pskb_may_pull(skb, skb_transport_offset(skb) + sizeof(struct tcphdr)))
1461		return;
1462
1463	iph = ip_hdr(skb);
1464	th = tcp_hdr(skb);
1465
1466	if (th->doff < sizeof(struct tcphdr) / 4)
1467		return;
1468
1469	sk = __inet_lookup_established(dev_net(skb->dev), &tcp_hashinfo,
1470				       iph->saddr, th->source,
1471				       iph->daddr, ntohs(th->dest),
1472				       skb->skb_iif);
1473	if (sk) {
1474		skb->sk = sk;
1475		skb->destructor = sock_edemux;
1476		if (sk_fullsock(sk)) {
1477			struct dst_entry *dst = READ_ONCE(sk->sk_rx_dst);
1478
1479			if (dst)
1480				dst = dst_check(dst, 0);
1481			if (dst &&
1482			    inet_sk(sk)->rx_dst_ifindex == skb->skb_iif)
1483				skb_dst_set_noref(skb, dst);
1484		}
1485	}
1486}
1487
1488/* Packet is added to VJ-style prequeue for processing in process
1489 * context, if a reader task is waiting. Apparently, this exciting
1490 * idea (VJ's mail "Re: query about TCP header on tcp-ip" of 07 Sep 93)
1491 * failed somewhere. Latency? Burstiness? Well, at least now we will
1492 * see, why it failed. 8)8)				  --ANK
1493 *
1494 */
1495bool tcp_prequeue(struct sock *sk, struct sk_buff *skb)
1496{
1497	struct tcp_sock *tp = tcp_sk(sk);
1498
1499	if (sysctl_tcp_low_latency || !tp->ucopy.task)
1500		return false;
1501
1502	if (skb->len <= tcp_hdrlen(skb) &&
1503	    skb_queue_len(&tp->ucopy.prequeue) == 0)
1504		return false;
1505
1506	/* Before escaping RCU protected region, we need to take care of skb
1507	 * dst. Prequeue is only enabled for established sockets.
1508	 * For such sockets, we might need the skb dst only to set sk->sk_rx_dst
1509	 * Instead of doing full sk_rx_dst validity here, let's perform
1510	 * an optimistic check.
1511	 */
1512	if (likely(sk->sk_rx_dst))
1513		skb_dst_drop(skb);
1514	else
1515		skb_dst_force_safe(skb);
1516
1517	__skb_queue_tail(&tp->ucopy.prequeue, skb);
1518	tp->ucopy.memory += skb->truesize;
1519	if (tp->ucopy.memory > sk->sk_rcvbuf) {
1520		struct sk_buff *skb1;
1521
1522		BUG_ON(sock_owned_by_user(sk));
1523
1524		while ((skb1 = __skb_dequeue(&tp->ucopy.prequeue)) != NULL) {
1525			sk_backlog_rcv(sk, skb1);
1526			NET_INC_STATS_BH(sock_net(sk),
1527					 LINUX_MIB_TCPPREQUEUEDROPPED);
1528		}
1529
1530		tp->ucopy.memory = 0;
1531	} else if (skb_queue_len(&tp->ucopy.prequeue) == 1) {
1532		wake_up_interruptible_sync_poll(sk_sleep(sk),
1533					   POLLIN | POLLRDNORM | POLLRDBAND);
1534		if (!inet_csk_ack_scheduled(sk))
1535			inet_csk_reset_xmit_timer(sk, ICSK_TIME_DACK,
1536						  (3 * tcp_rto_min(sk)) / 4,
1537						  TCP_RTO_MAX);
1538	}
1539	return true;
1540}
1541EXPORT_SYMBOL(tcp_prequeue);
1542
1543/*
1544 *	From tcp_input.c
1545 */
1546
1547int tcp_v4_rcv(struct sk_buff *skb)
1548{
1549	const struct iphdr *iph;
1550	const struct tcphdr *th;
1551	struct sock *sk;
1552	int ret;
1553	struct net *net = dev_net(skb->dev);
1554
1555	if (skb->pkt_type != PACKET_HOST)
1556		goto discard_it;
1557
1558	/* Count it even if it's bad */
1559	TCP_INC_STATS_BH(net, TCP_MIB_INSEGS);
1560
1561	if (!pskb_may_pull(skb, sizeof(struct tcphdr)))
1562		goto discard_it;
1563
1564	th = tcp_hdr(skb);
1565
1566	if (th->doff < sizeof(struct tcphdr) / 4)
1567		goto bad_packet;
1568	if (!pskb_may_pull(skb, th->doff * 4))
1569		goto discard_it;
1570
1571	/* An explanation is required here, I think.
1572	 * Packet length and doff are validated by header prediction,
1573	 * provided case of th->doff==0 is eliminated.
1574	 * So, we defer the checks. */
1575
1576	if (skb_checksum_init(skb, IPPROTO_TCP, inet_compute_pseudo))
1577		goto csum_error;
1578
1579	th = tcp_hdr(skb);
1580	iph = ip_hdr(skb);
1581	/* This is tricky : We move IPCB at its correct location into TCP_SKB_CB()
1582	 * barrier() makes sure compiler wont play fool^Waliasing games.
1583	 */
1584	memmove(&TCP_SKB_CB(skb)->header.h4, IPCB(skb),
1585		sizeof(struct inet_skb_parm));
1586	barrier();
1587
1588	TCP_SKB_CB(skb)->seq = ntohl(th->seq);
1589	TCP_SKB_CB(skb)->end_seq = (TCP_SKB_CB(skb)->seq + th->syn + th->fin +
1590				    skb->len - th->doff * 4);
1591	TCP_SKB_CB(skb)->ack_seq = ntohl(th->ack_seq);
1592	TCP_SKB_CB(skb)->tcp_flags = tcp_flag_byte(th);
1593	TCP_SKB_CB(skb)->tcp_tw_isn = 0;
1594	TCP_SKB_CB(skb)->ip_dsfield = ipv4_get_dsfield(iph);
1595	TCP_SKB_CB(skb)->sacked	 = 0;
1596
1597	sk = __inet_lookup_skb(&tcp_hashinfo, skb, th->source, th->dest);
1598	if (!sk)
1599		goto no_tcp_socket;
1600
1601process:
1602	if (sk->sk_state == TCP_TIME_WAIT)
1603		goto do_time_wait;
1604
1605	if (unlikely(iph->ttl < inet_sk(sk)->min_ttl)) {
1606		NET_INC_STATS_BH(net, LINUX_MIB_TCPMINTTLDROP);
1607		goto discard_and_relse;
1608	}
1609
1610	if (!xfrm4_policy_check(sk, XFRM_POLICY_IN, skb))
1611		goto discard_and_relse;
1612
1613#ifdef CONFIG_TCP_MD5SIG
1614	/*
1615	 * We really want to reject the packet as early as possible
1616	 * if:
1617	 *  o We're expecting an MD5'd packet and this is no MD5 tcp option
1618	 *  o There is an MD5 option and we're not expecting one
1619	 */
1620	if (tcp_v4_inbound_md5_hash(sk, skb))
1621		goto discard_and_relse;
1622#endif
1623
1624	nf_reset(skb);
1625
1626	if (sk_filter(sk, skb))
1627		goto discard_and_relse;
1628
1629	sk_incoming_cpu_update(sk);
1630	skb->dev = NULL;
1631
1632	bh_lock_sock_nested(sk);
1633	ret = 0;
1634	if (!sock_owned_by_user(sk)) {
1635		if (!tcp_prequeue(sk, skb))
1636			ret = tcp_v4_do_rcv(sk, skb);
1637	} else if (unlikely(sk_add_backlog(sk, skb,
1638					   sk->sk_rcvbuf + sk->sk_sndbuf))) {
1639		bh_unlock_sock(sk);
1640		NET_INC_STATS_BH(net, LINUX_MIB_TCPBACKLOGDROP);
1641		goto discard_and_relse;
1642	}
1643	bh_unlock_sock(sk);
1644
1645	sock_put(sk);
1646
1647	return ret;
1648
1649no_tcp_socket:
1650	if (!xfrm4_policy_check(NULL, XFRM_POLICY_IN, skb))
1651		goto discard_it;
1652
1653	if (skb->len < (th->doff << 2) || tcp_checksum_complete(skb)) {
1654csum_error:
1655		TCP_INC_STATS_BH(net, TCP_MIB_CSUMERRORS);
1656bad_packet:
1657		TCP_INC_STATS_BH(net, TCP_MIB_INERRS);
1658	} else {
1659		tcp_v4_send_reset(NULL, skb);
1660	}
1661
1662discard_it:
1663	/* Discard frame. */
1664	kfree_skb(skb);
1665	return 0;
1666
1667discard_and_relse:
1668	sock_put(sk);
1669	goto discard_it;
1670
1671do_time_wait:
1672	if (!xfrm4_policy_check(NULL, XFRM_POLICY_IN, skb)) {
1673		inet_twsk_put(inet_twsk(sk));
1674		goto discard_it;
1675	}
1676
1677	if (skb->len < (th->doff << 2)) {
1678		inet_twsk_put(inet_twsk(sk));
1679		goto bad_packet;
1680	}
1681	if (tcp_checksum_complete(skb)) {
1682		inet_twsk_put(inet_twsk(sk));
1683		goto csum_error;
1684	}
1685	switch (tcp_timewait_state_process(inet_twsk(sk), skb, th)) {
1686	case TCP_TW_SYN: {
1687		struct sock *sk2 = inet_lookup_listener(dev_net(skb->dev),
1688							&tcp_hashinfo,
1689							iph->saddr, th->source,
1690							iph->daddr, th->dest,
1691							inet_iif(skb));
1692		if (sk2) {
1693			inet_twsk_deschedule(inet_twsk(sk));
1694			inet_twsk_put(inet_twsk(sk));
1695			sk = sk2;
1696			goto process;
1697		}
1698		/* Fall through to ACK */
1699	}
1700	case TCP_TW_ACK:
1701		tcp_v4_timewait_ack(sk, skb);
1702		break;
1703	case TCP_TW_RST:
1704		goto no_tcp_socket;
1705	case TCP_TW_SUCCESS:;
1706	}
1707	goto discard_it;
1708}
1709
1710static struct timewait_sock_ops tcp_timewait_sock_ops = {
1711	.twsk_obj_size	= sizeof(struct tcp_timewait_sock),
1712	.twsk_unique	= tcp_twsk_unique,
1713	.twsk_destructor= tcp_twsk_destructor,
1714};
1715
1716void inet_sk_rx_dst_set(struct sock *sk, const struct sk_buff *skb)
1717{
1718	struct dst_entry *dst = skb_dst(skb);
1719
1720	if (dst && dst_hold_safe(dst)) {
1721		sk->sk_rx_dst = dst;
1722		inet_sk(sk)->rx_dst_ifindex = skb->skb_iif;
1723	}
1724}
1725EXPORT_SYMBOL(inet_sk_rx_dst_set);
1726
1727const struct inet_connection_sock_af_ops ipv4_specific = {
1728	.queue_xmit	   = ip_queue_xmit,
1729	.send_check	   = tcp_v4_send_check,
1730	.rebuild_header	   = inet_sk_rebuild_header,
1731	.sk_rx_dst_set	   = inet_sk_rx_dst_set,
1732	.conn_request	   = tcp_v4_conn_request,
1733	.syn_recv_sock	   = tcp_v4_syn_recv_sock,
1734	.net_header_len	   = sizeof(struct iphdr),
1735	.setsockopt	   = ip_setsockopt,
1736	.getsockopt	   = ip_getsockopt,
1737	.addr2sockaddr	   = inet_csk_addr2sockaddr,
1738	.sockaddr_len	   = sizeof(struct sockaddr_in),
1739	.bind_conflict	   = inet_csk_bind_conflict,
1740#ifdef CONFIG_COMPAT
1741	.compat_setsockopt = compat_ip_setsockopt,
1742	.compat_getsockopt = compat_ip_getsockopt,
1743#endif
1744	.mtu_reduced	   = tcp_v4_mtu_reduced,
1745};
1746EXPORT_SYMBOL(ipv4_specific);
1747
1748#ifdef CONFIG_TCP_MD5SIG
1749static const struct tcp_sock_af_ops tcp_sock_ipv4_specific = {
1750	.md5_lookup		= tcp_v4_md5_lookup,
1751	.calc_md5_hash		= tcp_v4_md5_hash_skb,
1752	.md5_parse		= tcp_v4_parse_md5_keys,
1753};
1754#endif
1755
1756/* NOTE: A lot of things set to zero explicitly by call to
1757 *       sk_alloc() so need not be done here.
1758 */
1759static int tcp_v4_init_sock(struct sock *sk)
1760{
1761	struct inet_connection_sock *icsk = inet_csk(sk);
1762
1763	tcp_init_sock(sk);
1764
1765	icsk->icsk_af_ops = &ipv4_specific;
1766
1767#ifdef CONFIG_TCP_MD5SIG
1768	tcp_sk(sk)->af_specific = &tcp_sock_ipv4_specific;
1769#endif
1770
1771	return 0;
1772}
1773
1774void tcp_v4_destroy_sock(struct sock *sk)
1775{
1776	struct tcp_sock *tp = tcp_sk(sk);
1777
1778	tcp_clear_xmit_timers(sk);
1779
1780	tcp_cleanup_congestion_control(sk);
1781
1782	/* Cleanup up the write buffer. */
1783	tcp_write_queue_purge(sk);
1784
1785	/* Cleans up our, hopefully empty, out_of_order_queue. */
1786	__skb_queue_purge(&tp->out_of_order_queue);
1787
1788#ifdef CONFIG_TCP_MD5SIG
1789	/* Clean up the MD5 key list, if any */
1790	if (tp->md5sig_info) {
1791		tcp_clear_md5_list(sk);
1792		kfree_rcu(tp->md5sig_info, rcu);
1793		tp->md5sig_info = NULL;
1794	}
1795#endif
1796
1797	/* Clean prequeue, it must be empty really */
1798	__skb_queue_purge(&tp->ucopy.prequeue);
1799
1800	/* Clean up a referenced TCP bind bucket. */
1801	if (inet_csk(sk)->icsk_bind_hash)
1802		inet_put_port(sk);
1803
1804	BUG_ON(tp->fastopen_rsk);
1805
1806	/* If socket is aborted during connect operation */
1807	tcp_free_fastopen_req(tp);
1808
1809	sk_sockets_allocated_dec(sk);
1810	sock_release_memcg(sk);
1811}
1812EXPORT_SYMBOL(tcp_v4_destroy_sock);
1813
1814#ifdef CONFIG_PROC_FS
1815/* Proc filesystem TCP sock list dumping. */
1816
1817/*
1818 * Get next listener socket follow cur.  If cur is NULL, get first socket
1819 * starting from bucket given in st->bucket; when st->bucket is zero the
1820 * very first socket in the hash table is returned.
1821 */
1822static void *listening_get_next(struct seq_file *seq, void *cur)
1823{
1824	struct inet_connection_sock *icsk;
1825	struct hlist_nulls_node *node;
1826	struct sock *sk = cur;
1827	struct inet_listen_hashbucket *ilb;
1828	struct tcp_iter_state *st = seq->private;
1829	struct net *net = seq_file_net(seq);
1830
1831	if (!sk) {
1832		ilb = &tcp_hashinfo.listening_hash[st->bucket];
1833		spin_lock_bh(&ilb->lock);
1834		sk = sk_nulls_head(&ilb->head);
1835		st->offset = 0;
1836		goto get_sk;
1837	}
1838	ilb = &tcp_hashinfo.listening_hash[st->bucket];
1839	++st->num;
1840	++st->offset;
1841
1842	if (st->state == TCP_SEQ_STATE_OPENREQ) {
1843		struct request_sock *req = cur;
1844
1845		icsk = inet_csk(st->syn_wait_sk);
1846		req = req->dl_next;
1847		while (1) {
1848			while (req) {
1849				if (req->rsk_ops->family == st->family) {
1850					cur = req;
1851					goto out;
1852				}
1853				req = req->dl_next;
1854			}
1855			if (++st->sbucket >= icsk->icsk_accept_queue.listen_opt->nr_table_entries)
1856				break;
1857get_req:
1858			req = icsk->icsk_accept_queue.listen_opt->syn_table[st->sbucket];
1859		}
1860		sk	  = sk_nulls_next(st->syn_wait_sk);
1861		st->state = TCP_SEQ_STATE_LISTENING;
1862		spin_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
1863	} else {
1864		icsk = inet_csk(sk);
1865		spin_lock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
1866		if (reqsk_queue_len(&icsk->icsk_accept_queue))
1867			goto start_req;
1868		spin_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
1869		sk = sk_nulls_next(sk);
1870	}
1871get_sk:
1872	sk_nulls_for_each_from(sk, node) {
1873		if (!net_eq(sock_net(sk), net))
1874			continue;
1875		if (sk->sk_family == st->family) {
1876			cur = sk;
1877			goto out;
1878		}
1879		icsk = inet_csk(sk);
1880		spin_lock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
1881		if (reqsk_queue_len(&icsk->icsk_accept_queue)) {
1882start_req:
1883			st->uid		= sock_i_uid(sk);
1884			st->syn_wait_sk = sk;
1885			st->state	= TCP_SEQ_STATE_OPENREQ;
1886			st->sbucket	= 0;
1887			goto get_req;
1888		}
1889		spin_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
1890	}
1891	spin_unlock_bh(&ilb->lock);
1892	st->offset = 0;
1893	if (++st->bucket < INET_LHTABLE_SIZE) {
1894		ilb = &tcp_hashinfo.listening_hash[st->bucket];
1895		spin_lock_bh(&ilb->lock);
1896		sk = sk_nulls_head(&ilb->head);
1897		goto get_sk;
1898	}
1899	cur = NULL;
1900out:
1901	return cur;
1902}
1903
1904static void *listening_get_idx(struct seq_file *seq, loff_t *pos)
1905{
1906	struct tcp_iter_state *st = seq->private;
1907	void *rc;
1908
1909	st->bucket = 0;
1910	st->offset = 0;
1911	rc = listening_get_next(seq, NULL);
1912
1913	while (rc && *pos) {
1914		rc = listening_get_next(seq, rc);
1915		--*pos;
1916	}
1917	return rc;
1918}
1919
1920static inline bool empty_bucket(const struct tcp_iter_state *st)
1921{
1922	return hlist_nulls_empty(&tcp_hashinfo.ehash[st->bucket].chain);
1923}
1924
1925/*
1926 * Get first established socket starting from bucket given in st->bucket.
1927 * If st->bucket is zero, the very first socket in the hash is returned.
1928 */
1929static void *established_get_first(struct seq_file *seq)
1930{
1931	struct tcp_iter_state *st = seq->private;
1932	struct net *net = seq_file_net(seq);
1933	void *rc = NULL;
1934
1935	st->offset = 0;
1936	for (; st->bucket <= tcp_hashinfo.ehash_mask; ++st->bucket) {
1937		struct sock *sk;
1938		struct hlist_nulls_node *node;
1939		spinlock_t *lock = inet_ehash_lockp(&tcp_hashinfo, st->bucket);
1940
1941		/* Lockless fast path for the common case of empty buckets */
1942		if (empty_bucket(st))
1943			continue;
1944
1945		spin_lock_bh(lock);
1946		sk_nulls_for_each(sk, node, &tcp_hashinfo.ehash[st->bucket].chain) {
1947			if (sk->sk_family != st->family ||
1948			    !net_eq(sock_net(sk), net)) {
1949				continue;
1950			}
1951			rc = sk;
1952			goto out;
1953		}
1954		spin_unlock_bh(lock);
1955	}
1956out:
1957	return rc;
1958}
1959
1960static void *established_get_next(struct seq_file *seq, void *cur)
1961{
1962	struct sock *sk = cur;
1963	struct hlist_nulls_node *node;
1964	struct tcp_iter_state *st = seq->private;
1965	struct net *net = seq_file_net(seq);
1966
1967	++st->num;
1968	++st->offset;
1969
1970	sk = sk_nulls_next(sk);
1971
1972	sk_nulls_for_each_from(sk, node) {
1973		if (sk->sk_family == st->family && net_eq(sock_net(sk), net))
1974			return sk;
1975	}
1976
1977	spin_unlock_bh(inet_ehash_lockp(&tcp_hashinfo, st->bucket));
1978	++st->bucket;
1979	return established_get_first(seq);
1980}
1981
1982static void *established_get_idx(struct seq_file *seq, loff_t pos)
1983{
1984	struct tcp_iter_state *st = seq->private;
1985	void *rc;
1986
1987	st->bucket = 0;
1988	rc = established_get_first(seq);
1989
1990	while (rc && pos) {
1991		rc = established_get_next(seq, rc);
1992		--pos;
1993	}
1994	return rc;
1995}
1996
1997static void *tcp_get_idx(struct seq_file *seq, loff_t pos)
1998{
1999	void *rc;
2000	struct tcp_iter_state *st = seq->private;
2001
2002	st->state = TCP_SEQ_STATE_LISTENING;
2003	rc	  = listening_get_idx(seq, &pos);
2004
2005	if (!rc) {
2006		st->state = TCP_SEQ_STATE_ESTABLISHED;
2007		rc	  = established_get_idx(seq, pos);
2008	}
2009
2010	return rc;
2011}
2012
2013static void *tcp_seek_last_pos(struct seq_file *seq)
2014{
2015	struct tcp_iter_state *st = seq->private;
2016	int offset = st->offset;
2017	int orig_num = st->num;
2018	void *rc = NULL;
2019
2020	switch (st->state) {
2021	case TCP_SEQ_STATE_OPENREQ:
2022	case TCP_SEQ_STATE_LISTENING:
2023		if (st->bucket >= INET_LHTABLE_SIZE)
2024			break;
2025		st->state = TCP_SEQ_STATE_LISTENING;
2026		rc = listening_get_next(seq, NULL);
2027		while (offset-- && rc)
2028			rc = listening_get_next(seq, rc);
2029		if (rc)
2030			break;
2031		st->bucket = 0;
2032		st->state = TCP_SEQ_STATE_ESTABLISHED;
2033		/* Fallthrough */
2034	case TCP_SEQ_STATE_ESTABLISHED:
2035		if (st->bucket > tcp_hashinfo.ehash_mask)
2036			break;
2037		rc = established_get_first(seq);
2038		while (offset-- && rc)
2039			rc = established_get_next(seq, rc);
2040	}
2041
2042	st->num = orig_num;
2043
2044	return rc;
2045}
2046
2047static void *tcp_seq_start(struct seq_file *seq, loff_t *pos)
2048{
2049	struct tcp_iter_state *st = seq->private;
2050	void *rc;
2051
2052	if (*pos && *pos == st->last_pos) {
2053		rc = tcp_seek_last_pos(seq);
2054		if (rc)
2055			goto out;
2056	}
2057
2058	st->state = TCP_SEQ_STATE_LISTENING;
2059	st->num = 0;
2060	st->bucket = 0;
2061	st->offset = 0;
2062	rc = *pos ? tcp_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
2063
2064out:
2065	st->last_pos = *pos;
2066	return rc;
2067}
2068
2069static void *tcp_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2070{
2071	struct tcp_iter_state *st = seq->private;
2072	void *rc = NULL;
2073
2074	if (v == SEQ_START_TOKEN) {
2075		rc = tcp_get_idx(seq, 0);
2076		goto out;
2077	}
2078
2079	switch (st->state) {
2080	case TCP_SEQ_STATE_OPENREQ:
2081	case TCP_SEQ_STATE_LISTENING:
2082		rc = listening_get_next(seq, v);
2083		if (!rc) {
2084			st->state = TCP_SEQ_STATE_ESTABLISHED;
2085			st->bucket = 0;
2086			st->offset = 0;
2087			rc	  = established_get_first(seq);
2088		}
2089		break;
2090	case TCP_SEQ_STATE_ESTABLISHED:
2091		rc = established_get_next(seq, v);
2092		break;
2093	}
2094out:
2095	++*pos;
2096	st->last_pos = *pos;
2097	return rc;
2098}
2099
2100static void tcp_seq_stop(struct seq_file *seq, void *v)
2101{
2102	struct tcp_iter_state *st = seq->private;
2103
2104	switch (st->state) {
2105	case TCP_SEQ_STATE_OPENREQ:
2106		if (v) {
2107			struct inet_connection_sock *icsk = inet_csk(st->syn_wait_sk);
2108			spin_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
2109		}
2110	case TCP_SEQ_STATE_LISTENING:
2111		if (v != SEQ_START_TOKEN)
2112			spin_unlock_bh(&tcp_hashinfo.listening_hash[st->bucket].lock);
2113		break;
2114	case TCP_SEQ_STATE_ESTABLISHED:
2115		if (v)
2116			spin_unlock_bh(inet_ehash_lockp(&tcp_hashinfo, st->bucket));
2117		break;
2118	}
2119}
2120
2121int tcp_seq_open(struct inode *inode, struct file *file)
2122{
2123	struct tcp_seq_afinfo *afinfo = PDE_DATA(inode);
2124	struct tcp_iter_state *s;
2125	int err;
2126
2127	err = seq_open_net(inode, file, &afinfo->seq_ops,
2128			  sizeof(struct tcp_iter_state));
2129	if (err < 0)
2130		return err;
2131
2132	s = ((struct seq_file *)file->private_data)->private;
2133	s->family		= afinfo->family;
2134	s->last_pos		= 0;
2135	return 0;
2136}
2137EXPORT_SYMBOL(tcp_seq_open);
2138
2139int tcp_proc_register(struct net *net, struct tcp_seq_afinfo *afinfo)
2140{
2141	int rc = 0;
2142	struct proc_dir_entry *p;
2143
2144	afinfo->seq_ops.start		= tcp_seq_start;
2145	afinfo->seq_ops.next		= tcp_seq_next;
2146	afinfo->seq_ops.stop		= tcp_seq_stop;
2147
2148	p = proc_create_data(afinfo->name, S_IRUGO, net->proc_net,
2149			     afinfo->seq_fops, afinfo);
2150	if (!p)
2151		rc = -ENOMEM;
2152	return rc;
2153}
2154EXPORT_SYMBOL(tcp_proc_register);
2155
2156void tcp_proc_unregister(struct net *net, struct tcp_seq_afinfo *afinfo)
2157{
2158	remove_proc_entry(afinfo->name, net->proc_net);
2159}
2160EXPORT_SYMBOL(tcp_proc_unregister);
2161
2162static void get_openreq4(const struct request_sock *req,
2163			 struct seq_file *f, int i, kuid_t uid)
2164{
2165	const struct inet_request_sock *ireq = inet_rsk(req);
2166	long delta = req->rsk_timer.expires - jiffies;
2167
2168	seq_printf(f, "%4d: %08X:%04X %08X:%04X"
2169		" %02X %08X:%08X %02X:%08lX %08X %5u %8d %u %d %pK",
2170		i,
2171		ireq->ir_loc_addr,
2172		ireq->ir_num,
2173		ireq->ir_rmt_addr,
2174		ntohs(ireq->ir_rmt_port),
2175		TCP_SYN_RECV,
2176		0, 0, /* could print option size, but that is af dependent. */
2177		1,    /* timers active (only the expire timer) */
2178		jiffies_delta_to_clock_t(delta),
2179		req->num_timeout,
2180		from_kuid_munged(seq_user_ns(f), uid),
2181		0,  /* non standard timer */
2182		0, /* open_requests have no inode */
2183		0,
2184		req);
2185}
2186
2187static void get_tcp4_sock(struct sock *sk, struct seq_file *f, int i)
2188{
2189	int timer_active;
2190	unsigned long timer_expires;
2191	const struct tcp_sock *tp = tcp_sk(sk);
2192	const struct inet_connection_sock *icsk = inet_csk(sk);
2193	const struct inet_sock *inet = inet_sk(sk);
2194	struct fastopen_queue *fastopenq = icsk->icsk_accept_queue.fastopenq;
2195	__be32 dest = inet->inet_daddr;
2196	__be32 src = inet->inet_rcv_saddr;
2197	__u16 destp = ntohs(inet->inet_dport);
2198	__u16 srcp = ntohs(inet->inet_sport);
2199	int rx_queue;
2200
2201	if (icsk->icsk_pending == ICSK_TIME_RETRANS ||
2202	    icsk->icsk_pending == ICSK_TIME_EARLY_RETRANS ||
2203	    icsk->icsk_pending == ICSK_TIME_LOSS_PROBE) {
2204		timer_active	= 1;
2205		timer_expires	= icsk->icsk_timeout;
2206	} else if (icsk->icsk_pending == ICSK_TIME_PROBE0) {
2207		timer_active	= 4;
2208		timer_expires	= icsk->icsk_timeout;
2209	} else if (timer_pending(&sk->sk_timer)) {
2210		timer_active	= 2;
2211		timer_expires	= sk->sk_timer.expires;
2212	} else {
2213		timer_active	= 0;
2214		timer_expires = jiffies;
2215	}
2216
2217	if (sk->sk_state == TCP_LISTEN)
2218		rx_queue = sk->sk_ack_backlog;
2219	else
2220		/*
2221		 * because we dont lock socket, we might find a transient negative value
2222		 */
2223		rx_queue = max_t(int, tp->rcv_nxt - tp->copied_seq, 0);
2224
2225	seq_printf(f, "%4d: %08X:%04X %08X:%04X %02X %08X:%08X %02X:%08lX "
2226			"%08X %5u %8d %lu %d %pK %lu %lu %u %u %d",
2227		i, src, srcp, dest, destp, sk->sk_state,
2228		tp->write_seq - tp->snd_una,
2229		rx_queue,
2230		timer_active,
2231		jiffies_delta_to_clock_t(timer_expires - jiffies),
2232		icsk->icsk_retransmits,
2233		from_kuid_munged(seq_user_ns(f), sock_i_uid(sk)),
2234		icsk->icsk_probes_out,
2235		sock_i_ino(sk),
2236		atomic_read(&sk->sk_refcnt), sk,
2237		jiffies_to_clock_t(icsk->icsk_rto),
2238		jiffies_to_clock_t(icsk->icsk_ack.ato),
2239		(icsk->icsk_ack.quick << 1) | icsk->icsk_ack.pingpong,
2240		tp->snd_cwnd,
2241		sk->sk_state == TCP_LISTEN ?
2242		    (fastopenq ? fastopenq->max_qlen : 0) :
2243		    (tcp_in_initial_slowstart(tp) ? -1 : tp->snd_ssthresh));
2244}
2245
2246static void get_timewait4_sock(const struct inet_timewait_sock *tw,
2247			       struct seq_file *f, int i)
2248{
2249	long delta = tw->tw_timer.expires - jiffies;
2250	__be32 dest, src;
2251	__u16 destp, srcp;
2252
2253	dest  = tw->tw_daddr;
2254	src   = tw->tw_rcv_saddr;
2255	destp = ntohs(tw->tw_dport);
2256	srcp  = ntohs(tw->tw_sport);
2257
2258	seq_printf(f, "%4d: %08X:%04X %08X:%04X"
2259		" %02X %08X:%08X %02X:%08lX %08X %5d %8d %d %d %pK",
2260		i, src, srcp, dest, destp, tw->tw_substate, 0, 0,
2261		3, jiffies_delta_to_clock_t(delta), 0, 0, 0, 0,
2262		atomic_read(&tw->tw_refcnt), tw);
2263}
2264
2265#define TMPSZ 150
2266
2267static int tcp4_seq_show(struct seq_file *seq, void *v)
2268{
2269	struct tcp_iter_state *st;
2270	struct sock *sk = v;
2271
2272	seq_setwidth(seq, TMPSZ - 1);
2273	if (v == SEQ_START_TOKEN) {
2274		seq_puts(seq, "  sl  local_address rem_address   st tx_queue "
2275			   "rx_queue tr tm->when retrnsmt   uid  timeout "
2276			   "inode");
2277		goto out;
2278	}
2279	st = seq->private;
2280
2281	switch (st->state) {
2282	case TCP_SEQ_STATE_LISTENING:
2283	case TCP_SEQ_STATE_ESTABLISHED:
2284		if (sk->sk_state == TCP_TIME_WAIT)
2285			get_timewait4_sock(v, seq, st->num);
2286		else
2287			get_tcp4_sock(v, seq, st->num);
2288		break;
2289	case TCP_SEQ_STATE_OPENREQ:
2290		get_openreq4(v, seq, st->num, st->uid);
2291		break;
2292	}
2293out:
2294	seq_pad(seq, '\n');
2295	return 0;
2296}
2297
2298static const struct file_operations tcp_afinfo_seq_fops = {
2299	.owner   = THIS_MODULE,
2300	.open    = tcp_seq_open,
2301	.read    = seq_read,
2302	.llseek  = seq_lseek,
2303	.release = seq_release_net
2304};
2305
2306static struct tcp_seq_afinfo tcp4_seq_afinfo = {
2307	.name		= "tcp",
2308	.family		= AF_INET,
2309	.seq_fops	= &tcp_afinfo_seq_fops,
2310	.seq_ops	= {
2311		.show		= tcp4_seq_show,
2312	},
2313};
2314
2315static int __net_init tcp4_proc_init_net(struct net *net)
2316{
2317	return tcp_proc_register(net, &tcp4_seq_afinfo);
2318}
2319
2320static void __net_exit tcp4_proc_exit_net(struct net *net)
2321{
2322	tcp_proc_unregister(net, &tcp4_seq_afinfo);
2323}
2324
2325static struct pernet_operations tcp4_net_ops = {
2326	.init = tcp4_proc_init_net,
2327	.exit = tcp4_proc_exit_net,
2328};
2329
2330int __init tcp4_proc_init(void)
2331{
2332	return register_pernet_subsys(&tcp4_net_ops);
2333}
2334
2335void tcp4_proc_exit(void)
2336{
2337	unregister_pernet_subsys(&tcp4_net_ops);
2338}
2339#endif /* CONFIG_PROC_FS */
2340
2341struct proto tcp_prot = {
2342	.name			= "TCP",
2343	.owner			= THIS_MODULE,
2344	.close			= tcp_close,
2345	.connect		= tcp_v4_connect,
2346	.disconnect		= tcp_disconnect,
2347	.accept			= inet_csk_accept,
2348	.ioctl			= tcp_ioctl,
2349	.init			= tcp_v4_init_sock,
2350	.destroy		= tcp_v4_destroy_sock,
2351	.shutdown		= tcp_shutdown,
2352	.setsockopt		= tcp_setsockopt,
2353	.getsockopt		= tcp_getsockopt,
2354	.recvmsg		= tcp_recvmsg,
2355	.sendmsg		= tcp_sendmsg,
2356	.sendpage		= tcp_sendpage,
2357	.backlog_rcv		= tcp_v4_do_rcv,
2358	.release_cb		= tcp_release_cb,
2359	.hash			= inet_hash,
2360	.unhash			= inet_unhash,
2361	.get_port		= inet_csk_get_port,
2362	.enter_memory_pressure	= tcp_enter_memory_pressure,
2363	.stream_memory_free	= tcp_stream_memory_free,
2364	.sockets_allocated	= &tcp_sockets_allocated,
2365	.orphan_count		= &tcp_orphan_count,
2366	.memory_allocated	= &tcp_memory_allocated,
2367	.memory_pressure	= &tcp_memory_pressure,
2368	.sysctl_mem		= sysctl_tcp_mem,
2369	.sysctl_wmem		= sysctl_tcp_wmem,
2370	.sysctl_rmem		= sysctl_tcp_rmem,
2371	.max_header		= MAX_TCP_HEADER,
2372	.obj_size		= sizeof(struct tcp_sock),
2373	.slab_flags		= SLAB_DESTROY_BY_RCU,
2374	.twsk_prot		= &tcp_timewait_sock_ops,
2375	.rsk_prot		= &tcp_request_sock_ops,
2376	.h.hashinfo		= &tcp_hashinfo,
2377	.no_autobind		= true,
2378#ifdef CONFIG_COMPAT
2379	.compat_setsockopt	= compat_tcp_setsockopt,
2380	.compat_getsockopt	= compat_tcp_getsockopt,
2381#endif
2382#ifdef CONFIG_MEMCG_KMEM
2383	.init_cgroup		= tcp_init_cgroup,
2384	.destroy_cgroup		= tcp_destroy_cgroup,
2385	.proto_cgroup		= tcp_proto_cgroup,
2386#endif
2387};
2388EXPORT_SYMBOL(tcp_prot);
2389
2390static void __net_exit tcp_sk_exit(struct net *net)
2391{
2392	int cpu;
2393
2394	for_each_possible_cpu(cpu)
2395		inet_ctl_sock_destroy(*per_cpu_ptr(net->ipv4.tcp_sk, cpu));
2396	free_percpu(net->ipv4.tcp_sk);
2397}
2398
2399static int __net_init tcp_sk_init(struct net *net)
2400{
2401	int res, cpu;
2402
2403	net->ipv4.tcp_sk = alloc_percpu(struct sock *);
2404	if (!net->ipv4.tcp_sk)
2405		return -ENOMEM;
2406
2407	for_each_possible_cpu(cpu) {
2408		struct sock *sk;
2409
2410		res = inet_ctl_sock_create(&sk, PF_INET, SOCK_RAW,
2411					   IPPROTO_TCP, net);
2412		if (res)
2413			goto fail;
2414		*per_cpu_ptr(net->ipv4.tcp_sk, cpu) = sk;
2415	}
2416	net->ipv4.sysctl_tcp_ecn = 2;
2417	net->ipv4.sysctl_tcp_base_mss = TCP_BASE_MSS;
2418	net->ipv4.sysctl_tcp_probe_threshold = TCP_PROBE_THRESHOLD;
2419	net->ipv4.sysctl_tcp_probe_interval = TCP_PROBE_INTERVAL;
2420	return 0;
2421
2422fail:
2423	tcp_sk_exit(net);
2424
2425	return res;
2426}
2427
2428static void __net_exit tcp_sk_exit_batch(struct list_head *net_exit_list)
2429{
2430	inet_twsk_purge(&tcp_hashinfo, &tcp_death_row, AF_INET);
2431}
2432
2433static struct pernet_operations __net_initdata tcp_sk_ops = {
2434       .init	   = tcp_sk_init,
2435       .exit	   = tcp_sk_exit,
2436       .exit_batch = tcp_sk_exit_batch,
2437};
2438
2439void __init tcp_v4_init(void)
2440{
2441	inet_hashinfo_init(&tcp_hashinfo);
2442	if (register_pernet_subsys(&tcp_sk_ops))
2443		panic("Failed to create the TCP control socket.\n");
2444}
2445