xref: /openbmc/linux/net/ipv4/tcp_ipv4.c (revision 9b68f30b)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * INET		An implementation of the TCP/IP protocol suite for the LINUX
4  *		operating system.  INET is implemented using the  BSD Socket
5  *		interface as the means of communication with the user level.
6  *
7  *		Implementation of the Transmission Control Protocol(TCP).
8  *
9  *		IPv4 specific functions
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 
19 /*
20  * Changes:
21  *		David S. Miller	:	New socket lookup architecture.
22  *					This code is dedicated to John Dyson.
23  *		David S. Miller :	Change semantics of established hash,
24  *					half is devoted to TIME_WAIT sockets
25  *					and the rest go in the other half.
26  *		Andi Kleen :		Add support for syncookies and fixed
27  *					some bugs: ip options weren't passed to
28  *					the TCP layer, missed a check for an
29  *					ACK bit.
30  *		Andi Kleen :		Implemented fast path mtu discovery.
31  *	     				Fixed many serious bugs in the
32  *					request_sock handling and moved
33  *					most of it into the af independent code.
34  *					Added tail drop and some other bugfixes.
35  *					Added new listen semantics.
36  *		Mike McLagan	:	Routing by source
37  *	Juan Jose Ciarlante:		ip_dynaddr bits
38  *		Andi Kleen:		various fixes.
39  *	Vitaly E. Lavrov	:	Transparent proxy revived after year
40  *					coma.
41  *	Andi Kleen		:	Fix new listen.
42  *	Andi Kleen		:	Fix accept error reporting.
43  *	YOSHIFUJI Hideaki @USAGI and:	Support IPV6_V6ONLY socket option, which
44  *	Alexey Kuznetsov		allow both IPv4 and IPv6 sockets to bind
45  *					a single port at the same time.
46  */
47 
48 #define pr_fmt(fmt) "TCP: " fmt
49 
50 #include <linux/bottom_half.h>
51 #include <linux/types.h>
52 #include <linux/fcntl.h>
53 #include <linux/module.h>
54 #include <linux/random.h>
55 #include <linux/cache.h>
56 #include <linux/jhash.h>
57 #include <linux/init.h>
58 #include <linux/times.h>
59 #include <linux/slab.h>
60 
61 #include <net/net_namespace.h>
62 #include <net/icmp.h>
63 #include <net/inet_hashtables.h>
64 #include <net/tcp.h>
65 #include <net/transp_v6.h>
66 #include <net/ipv6.h>
67 #include <net/inet_common.h>
68 #include <net/timewait_sock.h>
69 #include <net/xfrm.h>
70 #include <net/secure_seq.h>
71 #include <net/busy_poll.h>
72 
73 #include <linux/inet.h>
74 #include <linux/ipv6.h>
75 #include <linux/stddef.h>
76 #include <linux/proc_fs.h>
77 #include <linux/seq_file.h>
78 #include <linux/inetdevice.h>
79 #include <linux/btf_ids.h>
80 
81 #include <crypto/hash.h>
82 #include <linux/scatterlist.h>
83 
84 #include <trace/events/tcp.h>
85 
86 #ifdef CONFIG_TCP_MD5SIG
87 static int tcp_v4_md5_hash_hdr(char *md5_hash, const struct tcp_md5sig_key *key,
88 			       __be32 daddr, __be32 saddr, const struct tcphdr *th);
89 #endif
90 
91 struct inet_hashinfo tcp_hashinfo;
92 EXPORT_SYMBOL(tcp_hashinfo);
93 
94 static DEFINE_PER_CPU(struct sock *, ipv4_tcp_sk);
95 
96 static u32 tcp_v4_init_seq(const struct sk_buff *skb)
97 {
98 	return secure_tcp_seq(ip_hdr(skb)->daddr,
99 			      ip_hdr(skb)->saddr,
100 			      tcp_hdr(skb)->dest,
101 			      tcp_hdr(skb)->source);
102 }
103 
104 static u32 tcp_v4_init_ts_off(const struct net *net, const struct sk_buff *skb)
105 {
106 	return secure_tcp_ts_off(net, ip_hdr(skb)->daddr, ip_hdr(skb)->saddr);
107 }
108 
109 int tcp_twsk_unique(struct sock *sk, struct sock *sktw, void *twp)
110 {
111 	int reuse = READ_ONCE(sock_net(sk)->ipv4.sysctl_tcp_tw_reuse);
112 	const struct inet_timewait_sock *tw = inet_twsk(sktw);
113 	const struct tcp_timewait_sock *tcptw = tcp_twsk(sktw);
114 	struct tcp_sock *tp = tcp_sk(sk);
115 
116 	if (reuse == 2) {
117 		/* Still does not detect *everything* that goes through
118 		 * lo, since we require a loopback src or dst address
119 		 * or direct binding to 'lo' interface.
120 		 */
121 		bool loopback = false;
122 		if (tw->tw_bound_dev_if == LOOPBACK_IFINDEX)
123 			loopback = true;
124 #if IS_ENABLED(CONFIG_IPV6)
125 		if (tw->tw_family == AF_INET6) {
126 			if (ipv6_addr_loopback(&tw->tw_v6_daddr) ||
127 			    ipv6_addr_v4mapped_loopback(&tw->tw_v6_daddr) ||
128 			    ipv6_addr_loopback(&tw->tw_v6_rcv_saddr) ||
129 			    ipv6_addr_v4mapped_loopback(&tw->tw_v6_rcv_saddr))
130 				loopback = true;
131 		} else
132 #endif
133 		{
134 			if (ipv4_is_loopback(tw->tw_daddr) ||
135 			    ipv4_is_loopback(tw->tw_rcv_saddr))
136 				loopback = true;
137 		}
138 		if (!loopback)
139 			reuse = 0;
140 	}
141 
142 	/* With PAWS, it is safe from the viewpoint
143 	   of data integrity. Even without PAWS it is safe provided sequence
144 	   spaces do not overlap i.e. at data rates <= 80Mbit/sec.
145 
146 	   Actually, the idea is close to VJ's one, only timestamp cache is
147 	   held not per host, but per port pair and TW bucket is used as state
148 	   holder.
149 
150 	   If TW bucket has been already destroyed we fall back to VJ's scheme
151 	   and use initial timestamp retrieved from peer table.
152 	 */
153 	if (tcptw->tw_ts_recent_stamp &&
154 	    (!twp || (reuse && time_after32(ktime_get_seconds(),
155 					    tcptw->tw_ts_recent_stamp)))) {
156 		/* In case of repair and re-using TIME-WAIT sockets we still
157 		 * want to be sure that it is safe as above but honor the
158 		 * sequence numbers and time stamps set as part of the repair
159 		 * process.
160 		 *
161 		 * Without this check re-using a TIME-WAIT socket with TCP
162 		 * repair would accumulate a -1 on the repair assigned
163 		 * sequence number. The first time it is reused the sequence
164 		 * is -1, the second time -2, etc. This fixes that issue
165 		 * without appearing to create any others.
166 		 */
167 		if (likely(!tp->repair)) {
168 			u32 seq = tcptw->tw_snd_nxt + 65535 + 2;
169 
170 			if (!seq)
171 				seq = 1;
172 			WRITE_ONCE(tp->write_seq, seq);
173 			tp->rx_opt.ts_recent	   = tcptw->tw_ts_recent;
174 			tp->rx_opt.ts_recent_stamp = tcptw->tw_ts_recent_stamp;
175 		}
176 		sock_hold(sktw);
177 		return 1;
178 	}
179 
180 	return 0;
181 }
182 EXPORT_SYMBOL_GPL(tcp_twsk_unique);
183 
184 static int tcp_v4_pre_connect(struct sock *sk, struct sockaddr *uaddr,
185 			      int addr_len)
186 {
187 	/* This check is replicated from tcp_v4_connect() and intended to
188 	 * prevent BPF program called below from accessing bytes that are out
189 	 * of the bound specified by user in addr_len.
190 	 */
191 	if (addr_len < sizeof(struct sockaddr_in))
192 		return -EINVAL;
193 
194 	sock_owned_by_me(sk);
195 
196 	return BPF_CGROUP_RUN_PROG_INET4_CONNECT(sk, uaddr);
197 }
198 
199 /* This will initiate an outgoing connection. */
200 int tcp_v4_connect(struct sock *sk, struct sockaddr *uaddr, int addr_len)
201 {
202 	struct sockaddr_in *usin = (struct sockaddr_in *)uaddr;
203 	struct inet_timewait_death_row *tcp_death_row;
204 	struct inet_sock *inet = inet_sk(sk);
205 	struct tcp_sock *tp = tcp_sk(sk);
206 	struct ip_options_rcu *inet_opt;
207 	struct net *net = sock_net(sk);
208 	__be16 orig_sport, orig_dport;
209 	__be32 daddr, nexthop;
210 	struct flowi4 *fl4;
211 	struct rtable *rt;
212 	int err;
213 
214 	if (addr_len < sizeof(struct sockaddr_in))
215 		return -EINVAL;
216 
217 	if (usin->sin_family != AF_INET)
218 		return -EAFNOSUPPORT;
219 
220 	nexthop = daddr = usin->sin_addr.s_addr;
221 	inet_opt = rcu_dereference_protected(inet->inet_opt,
222 					     lockdep_sock_is_held(sk));
223 	if (inet_opt && inet_opt->opt.srr) {
224 		if (!daddr)
225 			return -EINVAL;
226 		nexthop = inet_opt->opt.faddr;
227 	}
228 
229 	orig_sport = inet->inet_sport;
230 	orig_dport = usin->sin_port;
231 	fl4 = &inet->cork.fl.u.ip4;
232 	rt = ip_route_connect(fl4, nexthop, inet->inet_saddr,
233 			      sk->sk_bound_dev_if, IPPROTO_TCP, orig_sport,
234 			      orig_dport, sk);
235 	if (IS_ERR(rt)) {
236 		err = PTR_ERR(rt);
237 		if (err == -ENETUNREACH)
238 			IP_INC_STATS(net, IPSTATS_MIB_OUTNOROUTES);
239 		return err;
240 	}
241 
242 	if (rt->rt_flags & (RTCF_MULTICAST | RTCF_BROADCAST)) {
243 		ip_rt_put(rt);
244 		return -ENETUNREACH;
245 	}
246 
247 	if (!inet_opt || !inet_opt->opt.srr)
248 		daddr = fl4->daddr;
249 
250 	tcp_death_row = &sock_net(sk)->ipv4.tcp_death_row;
251 
252 	if (!inet->inet_saddr) {
253 		err = inet_bhash2_update_saddr(sk,  &fl4->saddr, AF_INET);
254 		if (err) {
255 			ip_rt_put(rt);
256 			return err;
257 		}
258 	} else {
259 		sk_rcv_saddr_set(sk, inet->inet_saddr);
260 	}
261 
262 	if (tp->rx_opt.ts_recent_stamp && inet->inet_daddr != daddr) {
263 		/* Reset inherited state */
264 		tp->rx_opt.ts_recent	   = 0;
265 		tp->rx_opt.ts_recent_stamp = 0;
266 		if (likely(!tp->repair))
267 			WRITE_ONCE(tp->write_seq, 0);
268 	}
269 
270 	inet->inet_dport = usin->sin_port;
271 	sk_daddr_set(sk, daddr);
272 
273 	inet_csk(sk)->icsk_ext_hdr_len = 0;
274 	if (inet_opt)
275 		inet_csk(sk)->icsk_ext_hdr_len = inet_opt->opt.optlen;
276 
277 	tp->rx_opt.mss_clamp = TCP_MSS_DEFAULT;
278 
279 	/* Socket identity is still unknown (sport may be zero).
280 	 * However we set state to SYN-SENT and not releasing socket
281 	 * lock select source port, enter ourselves into the hash tables and
282 	 * complete initialization after this.
283 	 */
284 	tcp_set_state(sk, TCP_SYN_SENT);
285 	err = inet_hash_connect(tcp_death_row, sk);
286 	if (err)
287 		goto failure;
288 
289 	sk_set_txhash(sk);
290 
291 	rt = ip_route_newports(fl4, rt, orig_sport, orig_dport,
292 			       inet->inet_sport, inet->inet_dport, sk);
293 	if (IS_ERR(rt)) {
294 		err = PTR_ERR(rt);
295 		rt = NULL;
296 		goto failure;
297 	}
298 	/* OK, now commit destination to socket.  */
299 	sk->sk_gso_type = SKB_GSO_TCPV4;
300 	sk_setup_caps(sk, &rt->dst);
301 	rt = NULL;
302 
303 	if (likely(!tp->repair)) {
304 		if (!tp->write_seq)
305 			WRITE_ONCE(tp->write_seq,
306 				   secure_tcp_seq(inet->inet_saddr,
307 						  inet->inet_daddr,
308 						  inet->inet_sport,
309 						  usin->sin_port));
310 		tp->tsoffset = secure_tcp_ts_off(net, inet->inet_saddr,
311 						 inet->inet_daddr);
312 	}
313 
314 	inet->inet_id = get_random_u16();
315 
316 	if (tcp_fastopen_defer_connect(sk, &err))
317 		return err;
318 	if (err)
319 		goto failure;
320 
321 	err = tcp_connect(sk);
322 
323 	if (err)
324 		goto failure;
325 
326 	return 0;
327 
328 failure:
329 	/*
330 	 * This unhashes the socket and releases the local port,
331 	 * if necessary.
332 	 */
333 	tcp_set_state(sk, TCP_CLOSE);
334 	inet_bhash2_reset_saddr(sk);
335 	ip_rt_put(rt);
336 	sk->sk_route_caps = 0;
337 	inet->inet_dport = 0;
338 	return err;
339 }
340 EXPORT_SYMBOL(tcp_v4_connect);
341 
342 /*
343  * This routine reacts to ICMP_FRAG_NEEDED mtu indications as defined in RFC1191.
344  * It can be called through tcp_release_cb() if socket was owned by user
345  * at the time tcp_v4_err() was called to handle ICMP message.
346  */
347 void tcp_v4_mtu_reduced(struct sock *sk)
348 {
349 	struct inet_sock *inet = inet_sk(sk);
350 	struct dst_entry *dst;
351 	u32 mtu;
352 
353 	if ((1 << sk->sk_state) & (TCPF_LISTEN | TCPF_CLOSE))
354 		return;
355 	mtu = READ_ONCE(tcp_sk(sk)->mtu_info);
356 	dst = inet_csk_update_pmtu(sk, mtu);
357 	if (!dst)
358 		return;
359 
360 	/* Something is about to be wrong... Remember soft error
361 	 * for the case, if this connection will not able to recover.
362 	 */
363 	if (mtu < dst_mtu(dst) && ip_dont_fragment(sk, dst))
364 		WRITE_ONCE(sk->sk_err_soft, EMSGSIZE);
365 
366 	mtu = dst_mtu(dst);
367 
368 	if (inet->pmtudisc != IP_PMTUDISC_DONT &&
369 	    ip_sk_accept_pmtu(sk) &&
370 	    inet_csk(sk)->icsk_pmtu_cookie > mtu) {
371 		tcp_sync_mss(sk, mtu);
372 
373 		/* Resend the TCP packet because it's
374 		 * clear that the old packet has been
375 		 * dropped. This is the new "fast" path mtu
376 		 * discovery.
377 		 */
378 		tcp_simple_retransmit(sk);
379 	} /* else let the usual retransmit timer handle it */
380 }
381 EXPORT_SYMBOL(tcp_v4_mtu_reduced);
382 
383 static void do_redirect(struct sk_buff *skb, struct sock *sk)
384 {
385 	struct dst_entry *dst = __sk_dst_check(sk, 0);
386 
387 	if (dst)
388 		dst->ops->redirect(dst, sk, skb);
389 }
390 
391 
392 /* handle ICMP messages on TCP_NEW_SYN_RECV request sockets */
393 void tcp_req_err(struct sock *sk, u32 seq, bool abort)
394 {
395 	struct request_sock *req = inet_reqsk(sk);
396 	struct net *net = sock_net(sk);
397 
398 	/* ICMPs are not backlogged, hence we cannot get
399 	 * an established socket here.
400 	 */
401 	if (seq != tcp_rsk(req)->snt_isn) {
402 		__NET_INC_STATS(net, LINUX_MIB_OUTOFWINDOWICMPS);
403 	} else if (abort) {
404 		/*
405 		 * Still in SYN_RECV, just remove it silently.
406 		 * There is no good way to pass the error to the newly
407 		 * created socket, and POSIX does not want network
408 		 * errors returned from accept().
409 		 */
410 		inet_csk_reqsk_queue_drop(req->rsk_listener, req);
411 		tcp_listendrop(req->rsk_listener);
412 	}
413 	reqsk_put(req);
414 }
415 EXPORT_SYMBOL(tcp_req_err);
416 
417 /* TCP-LD (RFC 6069) logic */
418 void tcp_ld_RTO_revert(struct sock *sk, u32 seq)
419 {
420 	struct inet_connection_sock *icsk = inet_csk(sk);
421 	struct tcp_sock *tp = tcp_sk(sk);
422 	struct sk_buff *skb;
423 	s32 remaining;
424 	u32 delta_us;
425 
426 	if (sock_owned_by_user(sk))
427 		return;
428 
429 	if (seq != tp->snd_una  || !icsk->icsk_retransmits ||
430 	    !icsk->icsk_backoff)
431 		return;
432 
433 	skb = tcp_rtx_queue_head(sk);
434 	if (WARN_ON_ONCE(!skb))
435 		return;
436 
437 	icsk->icsk_backoff--;
438 	icsk->icsk_rto = tp->srtt_us ? __tcp_set_rto(tp) : TCP_TIMEOUT_INIT;
439 	icsk->icsk_rto = inet_csk_rto_backoff(icsk, TCP_RTO_MAX);
440 
441 	tcp_mstamp_refresh(tp);
442 	delta_us = (u32)(tp->tcp_mstamp - tcp_skb_timestamp_us(skb));
443 	remaining = icsk->icsk_rto - usecs_to_jiffies(delta_us);
444 
445 	if (remaining > 0) {
446 		inet_csk_reset_xmit_timer(sk, ICSK_TIME_RETRANS,
447 					  remaining, TCP_RTO_MAX);
448 	} else {
449 		/* RTO revert clocked out retransmission.
450 		 * Will retransmit now.
451 		 */
452 		tcp_retransmit_timer(sk);
453 	}
454 }
455 EXPORT_SYMBOL(tcp_ld_RTO_revert);
456 
457 /*
458  * This routine is called by the ICMP module when it gets some
459  * sort of error condition.  If err < 0 then the socket should
460  * be closed and the error returned to the user.  If err > 0
461  * it's just the icmp type << 8 | icmp code.  After adjustment
462  * header points to the first 8 bytes of the tcp header.  We need
463  * to find the appropriate port.
464  *
465  * The locking strategy used here is very "optimistic". When
466  * someone else accesses the socket the ICMP is just dropped
467  * and for some paths there is no check at all.
468  * A more general error queue to queue errors for later handling
469  * is probably better.
470  *
471  */
472 
473 int tcp_v4_err(struct sk_buff *skb, u32 info)
474 {
475 	const struct iphdr *iph = (const struct iphdr *)skb->data;
476 	struct tcphdr *th = (struct tcphdr *)(skb->data + (iph->ihl << 2));
477 	struct tcp_sock *tp;
478 	struct inet_sock *inet;
479 	const int type = icmp_hdr(skb)->type;
480 	const int code = icmp_hdr(skb)->code;
481 	struct sock *sk;
482 	struct request_sock *fastopen;
483 	u32 seq, snd_una;
484 	int err;
485 	struct net *net = dev_net(skb->dev);
486 
487 	sk = __inet_lookup_established(net, net->ipv4.tcp_death_row.hashinfo,
488 				       iph->daddr, th->dest, iph->saddr,
489 				       ntohs(th->source), inet_iif(skb), 0);
490 	if (!sk) {
491 		__ICMP_INC_STATS(net, ICMP_MIB_INERRORS);
492 		return -ENOENT;
493 	}
494 	if (sk->sk_state == TCP_TIME_WAIT) {
495 		inet_twsk_put(inet_twsk(sk));
496 		return 0;
497 	}
498 	seq = ntohl(th->seq);
499 	if (sk->sk_state == TCP_NEW_SYN_RECV) {
500 		tcp_req_err(sk, seq, type == ICMP_PARAMETERPROB ||
501 				     type == ICMP_TIME_EXCEEDED ||
502 				     (type == ICMP_DEST_UNREACH &&
503 				      (code == ICMP_NET_UNREACH ||
504 				       code == ICMP_HOST_UNREACH)));
505 		return 0;
506 	}
507 
508 	bh_lock_sock(sk);
509 	/* If too many ICMPs get dropped on busy
510 	 * servers this needs to be solved differently.
511 	 * We do take care of PMTU discovery (RFC1191) special case :
512 	 * we can receive locally generated ICMP messages while socket is held.
513 	 */
514 	if (sock_owned_by_user(sk)) {
515 		if (!(type == ICMP_DEST_UNREACH && code == ICMP_FRAG_NEEDED))
516 			__NET_INC_STATS(net, LINUX_MIB_LOCKDROPPEDICMPS);
517 	}
518 	if (sk->sk_state == TCP_CLOSE)
519 		goto out;
520 
521 	if (static_branch_unlikely(&ip4_min_ttl)) {
522 		/* min_ttl can be changed concurrently from do_ip_setsockopt() */
523 		if (unlikely(iph->ttl < READ_ONCE(inet_sk(sk)->min_ttl))) {
524 			__NET_INC_STATS(net, LINUX_MIB_TCPMINTTLDROP);
525 			goto out;
526 		}
527 	}
528 
529 	tp = tcp_sk(sk);
530 	/* XXX (TFO) - tp->snd_una should be ISN (tcp_create_openreq_child() */
531 	fastopen = rcu_dereference(tp->fastopen_rsk);
532 	snd_una = fastopen ? tcp_rsk(fastopen)->snt_isn : tp->snd_una;
533 	if (sk->sk_state != TCP_LISTEN &&
534 	    !between(seq, snd_una, tp->snd_nxt)) {
535 		__NET_INC_STATS(net, LINUX_MIB_OUTOFWINDOWICMPS);
536 		goto out;
537 	}
538 
539 	switch (type) {
540 	case ICMP_REDIRECT:
541 		if (!sock_owned_by_user(sk))
542 			do_redirect(skb, sk);
543 		goto out;
544 	case ICMP_SOURCE_QUENCH:
545 		/* Just silently ignore these. */
546 		goto out;
547 	case ICMP_PARAMETERPROB:
548 		err = EPROTO;
549 		break;
550 	case ICMP_DEST_UNREACH:
551 		if (code > NR_ICMP_UNREACH)
552 			goto out;
553 
554 		if (code == ICMP_FRAG_NEEDED) { /* PMTU discovery (RFC1191) */
555 			/* We are not interested in TCP_LISTEN and open_requests
556 			 * (SYN-ACKs send out by Linux are always <576bytes so
557 			 * they should go through unfragmented).
558 			 */
559 			if (sk->sk_state == TCP_LISTEN)
560 				goto out;
561 
562 			WRITE_ONCE(tp->mtu_info, info);
563 			if (!sock_owned_by_user(sk)) {
564 				tcp_v4_mtu_reduced(sk);
565 			} else {
566 				if (!test_and_set_bit(TCP_MTU_REDUCED_DEFERRED, &sk->sk_tsq_flags))
567 					sock_hold(sk);
568 			}
569 			goto out;
570 		}
571 
572 		err = icmp_err_convert[code].errno;
573 		/* check if this ICMP message allows revert of backoff.
574 		 * (see RFC 6069)
575 		 */
576 		if (!fastopen &&
577 		    (code == ICMP_NET_UNREACH || code == ICMP_HOST_UNREACH))
578 			tcp_ld_RTO_revert(sk, seq);
579 		break;
580 	case ICMP_TIME_EXCEEDED:
581 		err = EHOSTUNREACH;
582 		break;
583 	default:
584 		goto out;
585 	}
586 
587 	switch (sk->sk_state) {
588 	case TCP_SYN_SENT:
589 	case TCP_SYN_RECV:
590 		/* Only in fast or simultaneous open. If a fast open socket is
591 		 * already accepted it is treated as a connected one below.
592 		 */
593 		if (fastopen && !fastopen->sk)
594 			break;
595 
596 		ip_icmp_error(sk, skb, err, th->dest, info, (u8 *)th);
597 
598 		if (!sock_owned_by_user(sk)) {
599 			WRITE_ONCE(sk->sk_err, err);
600 
601 			sk_error_report(sk);
602 
603 			tcp_done(sk);
604 		} else {
605 			WRITE_ONCE(sk->sk_err_soft, err);
606 		}
607 		goto out;
608 	}
609 
610 	/* If we've already connected we will keep trying
611 	 * until we time out, or the user gives up.
612 	 *
613 	 * rfc1122 4.2.3.9 allows to consider as hard errors
614 	 * only PROTO_UNREACH and PORT_UNREACH (well, FRAG_FAILED too,
615 	 * but it is obsoleted by pmtu discovery).
616 	 *
617 	 * Note, that in modern internet, where routing is unreliable
618 	 * and in each dark corner broken firewalls sit, sending random
619 	 * errors ordered by their masters even this two messages finally lose
620 	 * their original sense (even Linux sends invalid PORT_UNREACHs)
621 	 *
622 	 * Now we are in compliance with RFCs.
623 	 *							--ANK (980905)
624 	 */
625 
626 	inet = inet_sk(sk);
627 	if (!sock_owned_by_user(sk) && inet->recverr) {
628 		WRITE_ONCE(sk->sk_err, err);
629 		sk_error_report(sk);
630 	} else	{ /* Only an error on timeout */
631 		WRITE_ONCE(sk->sk_err_soft, err);
632 	}
633 
634 out:
635 	bh_unlock_sock(sk);
636 	sock_put(sk);
637 	return 0;
638 }
639 
640 void __tcp_v4_send_check(struct sk_buff *skb, __be32 saddr, __be32 daddr)
641 {
642 	struct tcphdr *th = tcp_hdr(skb);
643 
644 	th->check = ~tcp_v4_check(skb->len, saddr, daddr, 0);
645 	skb->csum_start = skb_transport_header(skb) - skb->head;
646 	skb->csum_offset = offsetof(struct tcphdr, check);
647 }
648 
649 /* This routine computes an IPv4 TCP checksum. */
650 void tcp_v4_send_check(struct sock *sk, struct sk_buff *skb)
651 {
652 	const struct inet_sock *inet = inet_sk(sk);
653 
654 	__tcp_v4_send_check(skb, inet->inet_saddr, inet->inet_daddr);
655 }
656 EXPORT_SYMBOL(tcp_v4_send_check);
657 
658 /*
659  *	This routine will send an RST to the other tcp.
660  *
661  *	Someone asks: why I NEVER use socket parameters (TOS, TTL etc.)
662  *		      for reset.
663  *	Answer: if a packet caused RST, it is not for a socket
664  *		existing in our system, if it is matched to a socket,
665  *		it is just duplicate segment or bug in other side's TCP.
666  *		So that we build reply only basing on parameters
667  *		arrived with segment.
668  *	Exception: precedence violation. We do not implement it in any case.
669  */
670 
671 #ifdef CONFIG_TCP_MD5SIG
672 #define OPTION_BYTES TCPOLEN_MD5SIG_ALIGNED
673 #else
674 #define OPTION_BYTES sizeof(__be32)
675 #endif
676 
677 static void tcp_v4_send_reset(const struct sock *sk, struct sk_buff *skb)
678 {
679 	const struct tcphdr *th = tcp_hdr(skb);
680 	struct {
681 		struct tcphdr th;
682 		__be32 opt[OPTION_BYTES / sizeof(__be32)];
683 	} rep;
684 	struct ip_reply_arg arg;
685 #ifdef CONFIG_TCP_MD5SIG
686 	struct tcp_md5sig_key *key = NULL;
687 	const __u8 *hash_location = NULL;
688 	unsigned char newhash[16];
689 	int genhash;
690 	struct sock *sk1 = NULL;
691 #endif
692 	u64 transmit_time = 0;
693 	struct sock *ctl_sk;
694 	struct net *net;
695 	u32 txhash = 0;
696 
697 	/* Never send a reset in response to a reset. */
698 	if (th->rst)
699 		return;
700 
701 	/* If sk not NULL, it means we did a successful lookup and incoming
702 	 * route had to be correct. prequeue might have dropped our dst.
703 	 */
704 	if (!sk && skb_rtable(skb)->rt_type != RTN_LOCAL)
705 		return;
706 
707 	/* Swap the send and the receive. */
708 	memset(&rep, 0, sizeof(rep));
709 	rep.th.dest   = th->source;
710 	rep.th.source = th->dest;
711 	rep.th.doff   = sizeof(struct tcphdr) / 4;
712 	rep.th.rst    = 1;
713 
714 	if (th->ack) {
715 		rep.th.seq = th->ack_seq;
716 	} else {
717 		rep.th.ack = 1;
718 		rep.th.ack_seq = htonl(ntohl(th->seq) + th->syn + th->fin +
719 				       skb->len - (th->doff << 2));
720 	}
721 
722 	memset(&arg, 0, sizeof(arg));
723 	arg.iov[0].iov_base = (unsigned char *)&rep;
724 	arg.iov[0].iov_len  = sizeof(rep.th);
725 
726 	net = sk ? sock_net(sk) : dev_net(skb_dst(skb)->dev);
727 #ifdef CONFIG_TCP_MD5SIG
728 	rcu_read_lock();
729 	hash_location = tcp_parse_md5sig_option(th);
730 	if (sk && sk_fullsock(sk)) {
731 		const union tcp_md5_addr *addr;
732 		int l3index;
733 
734 		/* sdif set, means packet ingressed via a device
735 		 * in an L3 domain and inet_iif is set to it.
736 		 */
737 		l3index = tcp_v4_sdif(skb) ? inet_iif(skb) : 0;
738 		addr = (union tcp_md5_addr *)&ip_hdr(skb)->saddr;
739 		key = tcp_md5_do_lookup(sk, l3index, addr, AF_INET);
740 	} else if (hash_location) {
741 		const union tcp_md5_addr *addr;
742 		int sdif = tcp_v4_sdif(skb);
743 		int dif = inet_iif(skb);
744 		int l3index;
745 
746 		/*
747 		 * active side is lost. Try to find listening socket through
748 		 * source port, and then find md5 key through listening socket.
749 		 * we are not loose security here:
750 		 * Incoming packet is checked with md5 hash with finding key,
751 		 * no RST generated if md5 hash doesn't match.
752 		 */
753 		sk1 = __inet_lookup_listener(net, net->ipv4.tcp_death_row.hashinfo,
754 					     NULL, 0, ip_hdr(skb)->saddr,
755 					     th->source, ip_hdr(skb)->daddr,
756 					     ntohs(th->source), dif, sdif);
757 		/* don't send rst if it can't find key */
758 		if (!sk1)
759 			goto out;
760 
761 		/* sdif set, means packet ingressed via a device
762 		 * in an L3 domain and dif is set to it.
763 		 */
764 		l3index = sdif ? dif : 0;
765 		addr = (union tcp_md5_addr *)&ip_hdr(skb)->saddr;
766 		key = tcp_md5_do_lookup(sk1, l3index, addr, AF_INET);
767 		if (!key)
768 			goto out;
769 
770 
771 		genhash = tcp_v4_md5_hash_skb(newhash, key, NULL, skb);
772 		if (genhash || memcmp(hash_location, newhash, 16) != 0)
773 			goto out;
774 
775 	}
776 
777 	if (key) {
778 		rep.opt[0] = htonl((TCPOPT_NOP << 24) |
779 				   (TCPOPT_NOP << 16) |
780 				   (TCPOPT_MD5SIG << 8) |
781 				   TCPOLEN_MD5SIG);
782 		/* Update length and the length the header thinks exists */
783 		arg.iov[0].iov_len += TCPOLEN_MD5SIG_ALIGNED;
784 		rep.th.doff = arg.iov[0].iov_len / 4;
785 
786 		tcp_v4_md5_hash_hdr((__u8 *) &rep.opt[1],
787 				     key, ip_hdr(skb)->saddr,
788 				     ip_hdr(skb)->daddr, &rep.th);
789 	}
790 #endif
791 	/* Can't co-exist with TCPMD5, hence check rep.opt[0] */
792 	if (rep.opt[0] == 0) {
793 		__be32 mrst = mptcp_reset_option(skb);
794 
795 		if (mrst) {
796 			rep.opt[0] = mrst;
797 			arg.iov[0].iov_len += sizeof(mrst);
798 			rep.th.doff = arg.iov[0].iov_len / 4;
799 		}
800 	}
801 
802 	arg.csum = csum_tcpudp_nofold(ip_hdr(skb)->daddr,
803 				      ip_hdr(skb)->saddr, /* XXX */
804 				      arg.iov[0].iov_len, IPPROTO_TCP, 0);
805 	arg.csumoffset = offsetof(struct tcphdr, check) / 2;
806 	arg.flags = (sk && inet_sk_transparent(sk)) ? IP_REPLY_ARG_NOSRCCHECK : 0;
807 
808 	/* When socket is gone, all binding information is lost.
809 	 * routing might fail in this case. No choice here, if we choose to force
810 	 * input interface, we will misroute in case of asymmetric route.
811 	 */
812 	if (sk) {
813 		arg.bound_dev_if = sk->sk_bound_dev_if;
814 		if (sk_fullsock(sk))
815 			trace_tcp_send_reset(sk, skb);
816 	}
817 
818 	BUILD_BUG_ON(offsetof(struct sock, sk_bound_dev_if) !=
819 		     offsetof(struct inet_timewait_sock, tw_bound_dev_if));
820 
821 	arg.tos = ip_hdr(skb)->tos;
822 	arg.uid = sock_net_uid(net, sk && sk_fullsock(sk) ? sk : NULL);
823 	local_bh_disable();
824 	ctl_sk = this_cpu_read(ipv4_tcp_sk);
825 	sock_net_set(ctl_sk, net);
826 	if (sk) {
827 		ctl_sk->sk_mark = (sk->sk_state == TCP_TIME_WAIT) ?
828 				   inet_twsk(sk)->tw_mark : sk->sk_mark;
829 		ctl_sk->sk_priority = (sk->sk_state == TCP_TIME_WAIT) ?
830 				   inet_twsk(sk)->tw_priority : sk->sk_priority;
831 		transmit_time = tcp_transmit_time(sk);
832 		xfrm_sk_clone_policy(ctl_sk, sk);
833 		txhash = (sk->sk_state == TCP_TIME_WAIT) ?
834 			 inet_twsk(sk)->tw_txhash : sk->sk_txhash;
835 	} else {
836 		ctl_sk->sk_mark = 0;
837 		ctl_sk->sk_priority = 0;
838 	}
839 	ip_send_unicast_reply(ctl_sk,
840 			      skb, &TCP_SKB_CB(skb)->header.h4.opt,
841 			      ip_hdr(skb)->saddr, ip_hdr(skb)->daddr,
842 			      &arg, arg.iov[0].iov_len,
843 			      transmit_time, txhash);
844 
845 	xfrm_sk_free_policy(ctl_sk);
846 	sock_net_set(ctl_sk, &init_net);
847 	__TCP_INC_STATS(net, TCP_MIB_OUTSEGS);
848 	__TCP_INC_STATS(net, TCP_MIB_OUTRSTS);
849 	local_bh_enable();
850 
851 #ifdef CONFIG_TCP_MD5SIG
852 out:
853 	rcu_read_unlock();
854 #endif
855 }
856 
857 /* The code following below sending ACKs in SYN-RECV and TIME-WAIT states
858    outside socket context is ugly, certainly. What can I do?
859  */
860 
861 static void tcp_v4_send_ack(const struct sock *sk,
862 			    struct sk_buff *skb, u32 seq, u32 ack,
863 			    u32 win, u32 tsval, u32 tsecr, int oif,
864 			    struct tcp_md5sig_key *key,
865 			    int reply_flags, u8 tos, u32 txhash)
866 {
867 	const struct tcphdr *th = tcp_hdr(skb);
868 	struct {
869 		struct tcphdr th;
870 		__be32 opt[(TCPOLEN_TSTAMP_ALIGNED >> 2)
871 #ifdef CONFIG_TCP_MD5SIG
872 			   + (TCPOLEN_MD5SIG_ALIGNED >> 2)
873 #endif
874 			];
875 	} rep;
876 	struct net *net = sock_net(sk);
877 	struct ip_reply_arg arg;
878 	struct sock *ctl_sk;
879 	u64 transmit_time;
880 
881 	memset(&rep.th, 0, sizeof(struct tcphdr));
882 	memset(&arg, 0, sizeof(arg));
883 
884 	arg.iov[0].iov_base = (unsigned char *)&rep;
885 	arg.iov[0].iov_len  = sizeof(rep.th);
886 	if (tsecr) {
887 		rep.opt[0] = htonl((TCPOPT_NOP << 24) | (TCPOPT_NOP << 16) |
888 				   (TCPOPT_TIMESTAMP << 8) |
889 				   TCPOLEN_TIMESTAMP);
890 		rep.opt[1] = htonl(tsval);
891 		rep.opt[2] = htonl(tsecr);
892 		arg.iov[0].iov_len += TCPOLEN_TSTAMP_ALIGNED;
893 	}
894 
895 	/* Swap the send and the receive. */
896 	rep.th.dest    = th->source;
897 	rep.th.source  = th->dest;
898 	rep.th.doff    = arg.iov[0].iov_len / 4;
899 	rep.th.seq     = htonl(seq);
900 	rep.th.ack_seq = htonl(ack);
901 	rep.th.ack     = 1;
902 	rep.th.window  = htons(win);
903 
904 #ifdef CONFIG_TCP_MD5SIG
905 	if (key) {
906 		int offset = (tsecr) ? 3 : 0;
907 
908 		rep.opt[offset++] = htonl((TCPOPT_NOP << 24) |
909 					  (TCPOPT_NOP << 16) |
910 					  (TCPOPT_MD5SIG << 8) |
911 					  TCPOLEN_MD5SIG);
912 		arg.iov[0].iov_len += TCPOLEN_MD5SIG_ALIGNED;
913 		rep.th.doff = arg.iov[0].iov_len/4;
914 
915 		tcp_v4_md5_hash_hdr((__u8 *) &rep.opt[offset],
916 				    key, ip_hdr(skb)->saddr,
917 				    ip_hdr(skb)->daddr, &rep.th);
918 	}
919 #endif
920 	arg.flags = reply_flags;
921 	arg.csum = csum_tcpudp_nofold(ip_hdr(skb)->daddr,
922 				      ip_hdr(skb)->saddr, /* XXX */
923 				      arg.iov[0].iov_len, IPPROTO_TCP, 0);
924 	arg.csumoffset = offsetof(struct tcphdr, check) / 2;
925 	if (oif)
926 		arg.bound_dev_if = oif;
927 	arg.tos = tos;
928 	arg.uid = sock_net_uid(net, sk_fullsock(sk) ? sk : NULL);
929 	local_bh_disable();
930 	ctl_sk = this_cpu_read(ipv4_tcp_sk);
931 	sock_net_set(ctl_sk, net);
932 	ctl_sk->sk_mark = (sk->sk_state == TCP_TIME_WAIT) ?
933 			   inet_twsk(sk)->tw_mark : sk->sk_mark;
934 	ctl_sk->sk_priority = (sk->sk_state == TCP_TIME_WAIT) ?
935 			   inet_twsk(sk)->tw_priority : sk->sk_priority;
936 	transmit_time = tcp_transmit_time(sk);
937 	ip_send_unicast_reply(ctl_sk,
938 			      skb, &TCP_SKB_CB(skb)->header.h4.opt,
939 			      ip_hdr(skb)->saddr, ip_hdr(skb)->daddr,
940 			      &arg, arg.iov[0].iov_len,
941 			      transmit_time, txhash);
942 
943 	sock_net_set(ctl_sk, &init_net);
944 	__TCP_INC_STATS(net, TCP_MIB_OUTSEGS);
945 	local_bh_enable();
946 }
947 
948 static void tcp_v4_timewait_ack(struct sock *sk, struct sk_buff *skb)
949 {
950 	struct inet_timewait_sock *tw = inet_twsk(sk);
951 	struct tcp_timewait_sock *tcptw = tcp_twsk(sk);
952 
953 	tcp_v4_send_ack(sk, skb,
954 			tcptw->tw_snd_nxt, tcptw->tw_rcv_nxt,
955 			tcptw->tw_rcv_wnd >> tw->tw_rcv_wscale,
956 			tcp_time_stamp_raw() + tcptw->tw_ts_offset,
957 			tcptw->tw_ts_recent,
958 			tw->tw_bound_dev_if,
959 			tcp_twsk_md5_key(tcptw),
960 			tw->tw_transparent ? IP_REPLY_ARG_NOSRCCHECK : 0,
961 			tw->tw_tos,
962 			tw->tw_txhash
963 			);
964 
965 	inet_twsk_put(tw);
966 }
967 
968 static void tcp_v4_reqsk_send_ack(const struct sock *sk, struct sk_buff *skb,
969 				  struct request_sock *req)
970 {
971 	const union tcp_md5_addr *addr;
972 	int l3index;
973 
974 	/* sk->sk_state == TCP_LISTEN -> for regular TCP_SYN_RECV
975 	 * sk->sk_state == TCP_SYN_RECV -> for Fast Open.
976 	 */
977 	u32 seq = (sk->sk_state == TCP_LISTEN) ? tcp_rsk(req)->snt_isn + 1 :
978 					     tcp_sk(sk)->snd_nxt;
979 
980 	/* RFC 7323 2.3
981 	 * The window field (SEG.WND) of every outgoing segment, with the
982 	 * exception of <SYN> segments, MUST be right-shifted by
983 	 * Rcv.Wind.Shift bits:
984 	 */
985 	addr = (union tcp_md5_addr *)&ip_hdr(skb)->saddr;
986 	l3index = tcp_v4_sdif(skb) ? inet_iif(skb) : 0;
987 	tcp_v4_send_ack(sk, skb, seq,
988 			tcp_rsk(req)->rcv_nxt,
989 			req->rsk_rcv_wnd >> inet_rsk(req)->rcv_wscale,
990 			tcp_time_stamp_raw() + tcp_rsk(req)->ts_off,
991 			req->ts_recent,
992 			0,
993 			tcp_md5_do_lookup(sk, l3index, addr, AF_INET),
994 			inet_rsk(req)->no_srccheck ? IP_REPLY_ARG_NOSRCCHECK : 0,
995 			ip_hdr(skb)->tos, tcp_rsk(req)->txhash);
996 }
997 
998 /*
999  *	Send a SYN-ACK after having received a SYN.
1000  *	This still operates on a request_sock only, not on a big
1001  *	socket.
1002  */
1003 static int tcp_v4_send_synack(const struct sock *sk, struct dst_entry *dst,
1004 			      struct flowi *fl,
1005 			      struct request_sock *req,
1006 			      struct tcp_fastopen_cookie *foc,
1007 			      enum tcp_synack_type synack_type,
1008 			      struct sk_buff *syn_skb)
1009 {
1010 	const struct inet_request_sock *ireq = inet_rsk(req);
1011 	struct flowi4 fl4;
1012 	int err = -1;
1013 	struct sk_buff *skb;
1014 	u8 tos;
1015 
1016 	/* First, grab a route. */
1017 	if (!dst && (dst = inet_csk_route_req(sk, &fl4, req)) == NULL)
1018 		return -1;
1019 
1020 	skb = tcp_make_synack(sk, dst, req, foc, synack_type, syn_skb);
1021 
1022 	if (skb) {
1023 		__tcp_v4_send_check(skb, ireq->ir_loc_addr, ireq->ir_rmt_addr);
1024 
1025 		tos = READ_ONCE(sock_net(sk)->ipv4.sysctl_tcp_reflect_tos) ?
1026 				(tcp_rsk(req)->syn_tos & ~INET_ECN_MASK) |
1027 				(inet_sk(sk)->tos & INET_ECN_MASK) :
1028 				inet_sk(sk)->tos;
1029 
1030 		if (!INET_ECN_is_capable(tos) &&
1031 		    tcp_bpf_ca_needs_ecn((struct sock *)req))
1032 			tos |= INET_ECN_ECT_0;
1033 
1034 		rcu_read_lock();
1035 		err = ip_build_and_send_pkt(skb, sk, ireq->ir_loc_addr,
1036 					    ireq->ir_rmt_addr,
1037 					    rcu_dereference(ireq->ireq_opt),
1038 					    tos);
1039 		rcu_read_unlock();
1040 		err = net_xmit_eval(err);
1041 	}
1042 
1043 	return err;
1044 }
1045 
1046 /*
1047  *	IPv4 request_sock destructor.
1048  */
1049 static void tcp_v4_reqsk_destructor(struct request_sock *req)
1050 {
1051 	kfree(rcu_dereference_protected(inet_rsk(req)->ireq_opt, 1));
1052 }
1053 
1054 #ifdef CONFIG_TCP_MD5SIG
1055 /*
1056  * RFC2385 MD5 checksumming requires a mapping of
1057  * IP address->MD5 Key.
1058  * We need to maintain these in the sk structure.
1059  */
1060 
1061 DEFINE_STATIC_KEY_DEFERRED_FALSE(tcp_md5_needed, HZ);
1062 EXPORT_SYMBOL(tcp_md5_needed);
1063 
1064 static bool better_md5_match(struct tcp_md5sig_key *old, struct tcp_md5sig_key *new)
1065 {
1066 	if (!old)
1067 		return true;
1068 
1069 	/* l3index always overrides non-l3index */
1070 	if (old->l3index && new->l3index == 0)
1071 		return false;
1072 	if (old->l3index == 0 && new->l3index)
1073 		return true;
1074 
1075 	return old->prefixlen < new->prefixlen;
1076 }
1077 
1078 /* Find the Key structure for an address.  */
1079 struct tcp_md5sig_key *__tcp_md5_do_lookup(const struct sock *sk, int l3index,
1080 					   const union tcp_md5_addr *addr,
1081 					   int family)
1082 {
1083 	const struct tcp_sock *tp = tcp_sk(sk);
1084 	struct tcp_md5sig_key *key;
1085 	const struct tcp_md5sig_info *md5sig;
1086 	__be32 mask;
1087 	struct tcp_md5sig_key *best_match = NULL;
1088 	bool match;
1089 
1090 	/* caller either holds rcu_read_lock() or socket lock */
1091 	md5sig = rcu_dereference_check(tp->md5sig_info,
1092 				       lockdep_sock_is_held(sk));
1093 	if (!md5sig)
1094 		return NULL;
1095 
1096 	hlist_for_each_entry_rcu(key, &md5sig->head, node,
1097 				 lockdep_sock_is_held(sk)) {
1098 		if (key->family != family)
1099 			continue;
1100 		if (key->flags & TCP_MD5SIG_FLAG_IFINDEX && key->l3index != l3index)
1101 			continue;
1102 		if (family == AF_INET) {
1103 			mask = inet_make_mask(key->prefixlen);
1104 			match = (key->addr.a4.s_addr & mask) ==
1105 				(addr->a4.s_addr & mask);
1106 #if IS_ENABLED(CONFIG_IPV6)
1107 		} else if (family == AF_INET6) {
1108 			match = ipv6_prefix_equal(&key->addr.a6, &addr->a6,
1109 						  key->prefixlen);
1110 #endif
1111 		} else {
1112 			match = false;
1113 		}
1114 
1115 		if (match && better_md5_match(best_match, key))
1116 			best_match = key;
1117 	}
1118 	return best_match;
1119 }
1120 EXPORT_SYMBOL(__tcp_md5_do_lookup);
1121 
1122 static struct tcp_md5sig_key *tcp_md5_do_lookup_exact(const struct sock *sk,
1123 						      const union tcp_md5_addr *addr,
1124 						      int family, u8 prefixlen,
1125 						      int l3index, u8 flags)
1126 {
1127 	const struct tcp_sock *tp = tcp_sk(sk);
1128 	struct tcp_md5sig_key *key;
1129 	unsigned int size = sizeof(struct in_addr);
1130 	const struct tcp_md5sig_info *md5sig;
1131 
1132 	/* caller either holds rcu_read_lock() or socket lock */
1133 	md5sig = rcu_dereference_check(tp->md5sig_info,
1134 				       lockdep_sock_is_held(sk));
1135 	if (!md5sig)
1136 		return NULL;
1137 #if IS_ENABLED(CONFIG_IPV6)
1138 	if (family == AF_INET6)
1139 		size = sizeof(struct in6_addr);
1140 #endif
1141 	hlist_for_each_entry_rcu(key, &md5sig->head, node,
1142 				 lockdep_sock_is_held(sk)) {
1143 		if (key->family != family)
1144 			continue;
1145 		if ((key->flags & TCP_MD5SIG_FLAG_IFINDEX) != (flags & TCP_MD5SIG_FLAG_IFINDEX))
1146 			continue;
1147 		if (key->l3index != l3index)
1148 			continue;
1149 		if (!memcmp(&key->addr, addr, size) &&
1150 		    key->prefixlen == prefixlen)
1151 			return key;
1152 	}
1153 	return NULL;
1154 }
1155 
1156 struct tcp_md5sig_key *tcp_v4_md5_lookup(const struct sock *sk,
1157 					 const struct sock *addr_sk)
1158 {
1159 	const union tcp_md5_addr *addr;
1160 	int l3index;
1161 
1162 	l3index = l3mdev_master_ifindex_by_index(sock_net(sk),
1163 						 addr_sk->sk_bound_dev_if);
1164 	addr = (const union tcp_md5_addr *)&addr_sk->sk_daddr;
1165 	return tcp_md5_do_lookup(sk, l3index, addr, AF_INET);
1166 }
1167 EXPORT_SYMBOL(tcp_v4_md5_lookup);
1168 
1169 static int tcp_md5sig_info_add(struct sock *sk, gfp_t gfp)
1170 {
1171 	struct tcp_sock *tp = tcp_sk(sk);
1172 	struct tcp_md5sig_info *md5sig;
1173 
1174 	md5sig = kmalloc(sizeof(*md5sig), gfp);
1175 	if (!md5sig)
1176 		return -ENOMEM;
1177 
1178 	sk_gso_disable(sk);
1179 	INIT_HLIST_HEAD(&md5sig->head);
1180 	rcu_assign_pointer(tp->md5sig_info, md5sig);
1181 	return 0;
1182 }
1183 
1184 /* This can be called on a newly created socket, from other files */
1185 static int __tcp_md5_do_add(struct sock *sk, const union tcp_md5_addr *addr,
1186 			    int family, u8 prefixlen, int l3index, u8 flags,
1187 			    const u8 *newkey, u8 newkeylen, gfp_t gfp)
1188 {
1189 	/* Add Key to the list */
1190 	struct tcp_md5sig_key *key;
1191 	struct tcp_sock *tp = tcp_sk(sk);
1192 	struct tcp_md5sig_info *md5sig;
1193 
1194 	key = tcp_md5_do_lookup_exact(sk, addr, family, prefixlen, l3index, flags);
1195 	if (key) {
1196 		/* Pre-existing entry - just update that one.
1197 		 * Note that the key might be used concurrently.
1198 		 * data_race() is telling kcsan that we do not care of
1199 		 * key mismatches, since changing MD5 key on live flows
1200 		 * can lead to packet drops.
1201 		 */
1202 		data_race(memcpy(key->key, newkey, newkeylen));
1203 
1204 		/* Pairs with READ_ONCE() in tcp_md5_hash_key().
1205 		 * Also note that a reader could catch new key->keylen value
1206 		 * but old key->key[], this is the reason we use __GFP_ZERO
1207 		 * at sock_kmalloc() time below these lines.
1208 		 */
1209 		WRITE_ONCE(key->keylen, newkeylen);
1210 
1211 		return 0;
1212 	}
1213 
1214 	md5sig = rcu_dereference_protected(tp->md5sig_info,
1215 					   lockdep_sock_is_held(sk));
1216 
1217 	key = sock_kmalloc(sk, sizeof(*key), gfp | __GFP_ZERO);
1218 	if (!key)
1219 		return -ENOMEM;
1220 	if (!tcp_alloc_md5sig_pool()) {
1221 		sock_kfree_s(sk, key, sizeof(*key));
1222 		return -ENOMEM;
1223 	}
1224 
1225 	memcpy(key->key, newkey, newkeylen);
1226 	key->keylen = newkeylen;
1227 	key->family = family;
1228 	key->prefixlen = prefixlen;
1229 	key->l3index = l3index;
1230 	key->flags = flags;
1231 	memcpy(&key->addr, addr,
1232 	       (IS_ENABLED(CONFIG_IPV6) && family == AF_INET6) ? sizeof(struct in6_addr) :
1233 								 sizeof(struct in_addr));
1234 	hlist_add_head_rcu(&key->node, &md5sig->head);
1235 	return 0;
1236 }
1237 
1238 int tcp_md5_do_add(struct sock *sk, const union tcp_md5_addr *addr,
1239 		   int family, u8 prefixlen, int l3index, u8 flags,
1240 		   const u8 *newkey, u8 newkeylen)
1241 {
1242 	struct tcp_sock *tp = tcp_sk(sk);
1243 
1244 	if (!rcu_dereference_protected(tp->md5sig_info, lockdep_sock_is_held(sk))) {
1245 		if (tcp_md5sig_info_add(sk, GFP_KERNEL))
1246 			return -ENOMEM;
1247 
1248 		if (!static_branch_inc(&tcp_md5_needed.key)) {
1249 			struct tcp_md5sig_info *md5sig;
1250 
1251 			md5sig = rcu_dereference_protected(tp->md5sig_info, lockdep_sock_is_held(sk));
1252 			rcu_assign_pointer(tp->md5sig_info, NULL);
1253 			kfree_rcu(md5sig, rcu);
1254 			return -EUSERS;
1255 		}
1256 	}
1257 
1258 	return __tcp_md5_do_add(sk, addr, family, prefixlen, l3index, flags,
1259 				newkey, newkeylen, GFP_KERNEL);
1260 }
1261 EXPORT_SYMBOL(tcp_md5_do_add);
1262 
1263 int tcp_md5_key_copy(struct sock *sk, const union tcp_md5_addr *addr,
1264 		     int family, u8 prefixlen, int l3index,
1265 		     struct tcp_md5sig_key *key)
1266 {
1267 	struct tcp_sock *tp = tcp_sk(sk);
1268 
1269 	if (!rcu_dereference_protected(tp->md5sig_info, lockdep_sock_is_held(sk))) {
1270 		if (tcp_md5sig_info_add(sk, sk_gfp_mask(sk, GFP_ATOMIC)))
1271 			return -ENOMEM;
1272 
1273 		if (!static_key_fast_inc_not_disabled(&tcp_md5_needed.key.key)) {
1274 			struct tcp_md5sig_info *md5sig;
1275 
1276 			md5sig = rcu_dereference_protected(tp->md5sig_info, lockdep_sock_is_held(sk));
1277 			net_warn_ratelimited("Too many TCP-MD5 keys in the system\n");
1278 			rcu_assign_pointer(tp->md5sig_info, NULL);
1279 			kfree_rcu(md5sig, rcu);
1280 			return -EUSERS;
1281 		}
1282 	}
1283 
1284 	return __tcp_md5_do_add(sk, addr, family, prefixlen, l3index,
1285 				key->flags, key->key, key->keylen,
1286 				sk_gfp_mask(sk, GFP_ATOMIC));
1287 }
1288 EXPORT_SYMBOL(tcp_md5_key_copy);
1289 
1290 int tcp_md5_do_del(struct sock *sk, const union tcp_md5_addr *addr, int family,
1291 		   u8 prefixlen, int l3index, u8 flags)
1292 {
1293 	struct tcp_md5sig_key *key;
1294 
1295 	key = tcp_md5_do_lookup_exact(sk, addr, family, prefixlen, l3index, flags);
1296 	if (!key)
1297 		return -ENOENT;
1298 	hlist_del_rcu(&key->node);
1299 	atomic_sub(sizeof(*key), &sk->sk_omem_alloc);
1300 	kfree_rcu(key, rcu);
1301 	return 0;
1302 }
1303 EXPORT_SYMBOL(tcp_md5_do_del);
1304 
1305 static void tcp_clear_md5_list(struct sock *sk)
1306 {
1307 	struct tcp_sock *tp = tcp_sk(sk);
1308 	struct tcp_md5sig_key *key;
1309 	struct hlist_node *n;
1310 	struct tcp_md5sig_info *md5sig;
1311 
1312 	md5sig = rcu_dereference_protected(tp->md5sig_info, 1);
1313 
1314 	hlist_for_each_entry_safe(key, n, &md5sig->head, node) {
1315 		hlist_del_rcu(&key->node);
1316 		atomic_sub(sizeof(*key), &sk->sk_omem_alloc);
1317 		kfree_rcu(key, rcu);
1318 	}
1319 }
1320 
1321 static int tcp_v4_parse_md5_keys(struct sock *sk, int optname,
1322 				 sockptr_t optval, int optlen)
1323 {
1324 	struct tcp_md5sig cmd;
1325 	struct sockaddr_in *sin = (struct sockaddr_in *)&cmd.tcpm_addr;
1326 	const union tcp_md5_addr *addr;
1327 	u8 prefixlen = 32;
1328 	int l3index = 0;
1329 	u8 flags;
1330 
1331 	if (optlen < sizeof(cmd))
1332 		return -EINVAL;
1333 
1334 	if (copy_from_sockptr(&cmd, optval, sizeof(cmd)))
1335 		return -EFAULT;
1336 
1337 	if (sin->sin_family != AF_INET)
1338 		return -EINVAL;
1339 
1340 	flags = cmd.tcpm_flags & TCP_MD5SIG_FLAG_IFINDEX;
1341 
1342 	if (optname == TCP_MD5SIG_EXT &&
1343 	    cmd.tcpm_flags & TCP_MD5SIG_FLAG_PREFIX) {
1344 		prefixlen = cmd.tcpm_prefixlen;
1345 		if (prefixlen > 32)
1346 			return -EINVAL;
1347 	}
1348 
1349 	if (optname == TCP_MD5SIG_EXT && cmd.tcpm_ifindex &&
1350 	    cmd.tcpm_flags & TCP_MD5SIG_FLAG_IFINDEX) {
1351 		struct net_device *dev;
1352 
1353 		rcu_read_lock();
1354 		dev = dev_get_by_index_rcu(sock_net(sk), cmd.tcpm_ifindex);
1355 		if (dev && netif_is_l3_master(dev))
1356 			l3index = dev->ifindex;
1357 
1358 		rcu_read_unlock();
1359 
1360 		/* ok to reference set/not set outside of rcu;
1361 		 * right now device MUST be an L3 master
1362 		 */
1363 		if (!dev || !l3index)
1364 			return -EINVAL;
1365 	}
1366 
1367 	addr = (union tcp_md5_addr *)&sin->sin_addr.s_addr;
1368 
1369 	if (!cmd.tcpm_keylen)
1370 		return tcp_md5_do_del(sk, addr, AF_INET, prefixlen, l3index, flags);
1371 
1372 	if (cmd.tcpm_keylen > TCP_MD5SIG_MAXKEYLEN)
1373 		return -EINVAL;
1374 
1375 	return tcp_md5_do_add(sk, addr, AF_INET, prefixlen, l3index, flags,
1376 			      cmd.tcpm_key, cmd.tcpm_keylen);
1377 }
1378 
1379 static int tcp_v4_md5_hash_headers(struct tcp_md5sig_pool *hp,
1380 				   __be32 daddr, __be32 saddr,
1381 				   const struct tcphdr *th, int nbytes)
1382 {
1383 	struct tcp4_pseudohdr *bp;
1384 	struct scatterlist sg;
1385 	struct tcphdr *_th;
1386 
1387 	bp = hp->scratch;
1388 	bp->saddr = saddr;
1389 	bp->daddr = daddr;
1390 	bp->pad = 0;
1391 	bp->protocol = IPPROTO_TCP;
1392 	bp->len = cpu_to_be16(nbytes);
1393 
1394 	_th = (struct tcphdr *)(bp + 1);
1395 	memcpy(_th, th, sizeof(*th));
1396 	_th->check = 0;
1397 
1398 	sg_init_one(&sg, bp, sizeof(*bp) + sizeof(*th));
1399 	ahash_request_set_crypt(hp->md5_req, &sg, NULL,
1400 				sizeof(*bp) + sizeof(*th));
1401 	return crypto_ahash_update(hp->md5_req);
1402 }
1403 
1404 static int tcp_v4_md5_hash_hdr(char *md5_hash, const struct tcp_md5sig_key *key,
1405 			       __be32 daddr, __be32 saddr, const struct tcphdr *th)
1406 {
1407 	struct tcp_md5sig_pool *hp;
1408 	struct ahash_request *req;
1409 
1410 	hp = tcp_get_md5sig_pool();
1411 	if (!hp)
1412 		goto clear_hash_noput;
1413 	req = hp->md5_req;
1414 
1415 	if (crypto_ahash_init(req))
1416 		goto clear_hash;
1417 	if (tcp_v4_md5_hash_headers(hp, daddr, saddr, th, th->doff << 2))
1418 		goto clear_hash;
1419 	if (tcp_md5_hash_key(hp, key))
1420 		goto clear_hash;
1421 	ahash_request_set_crypt(req, NULL, md5_hash, 0);
1422 	if (crypto_ahash_final(req))
1423 		goto clear_hash;
1424 
1425 	tcp_put_md5sig_pool();
1426 	return 0;
1427 
1428 clear_hash:
1429 	tcp_put_md5sig_pool();
1430 clear_hash_noput:
1431 	memset(md5_hash, 0, 16);
1432 	return 1;
1433 }
1434 
1435 int tcp_v4_md5_hash_skb(char *md5_hash, const struct tcp_md5sig_key *key,
1436 			const struct sock *sk,
1437 			const struct sk_buff *skb)
1438 {
1439 	struct tcp_md5sig_pool *hp;
1440 	struct ahash_request *req;
1441 	const struct tcphdr *th = tcp_hdr(skb);
1442 	__be32 saddr, daddr;
1443 
1444 	if (sk) { /* valid for establish/request sockets */
1445 		saddr = sk->sk_rcv_saddr;
1446 		daddr = sk->sk_daddr;
1447 	} else {
1448 		const struct iphdr *iph = ip_hdr(skb);
1449 		saddr = iph->saddr;
1450 		daddr = iph->daddr;
1451 	}
1452 
1453 	hp = tcp_get_md5sig_pool();
1454 	if (!hp)
1455 		goto clear_hash_noput;
1456 	req = hp->md5_req;
1457 
1458 	if (crypto_ahash_init(req))
1459 		goto clear_hash;
1460 
1461 	if (tcp_v4_md5_hash_headers(hp, daddr, saddr, th, skb->len))
1462 		goto clear_hash;
1463 	if (tcp_md5_hash_skb_data(hp, skb, th->doff << 2))
1464 		goto clear_hash;
1465 	if (tcp_md5_hash_key(hp, key))
1466 		goto clear_hash;
1467 	ahash_request_set_crypt(req, NULL, md5_hash, 0);
1468 	if (crypto_ahash_final(req))
1469 		goto clear_hash;
1470 
1471 	tcp_put_md5sig_pool();
1472 	return 0;
1473 
1474 clear_hash:
1475 	tcp_put_md5sig_pool();
1476 clear_hash_noput:
1477 	memset(md5_hash, 0, 16);
1478 	return 1;
1479 }
1480 EXPORT_SYMBOL(tcp_v4_md5_hash_skb);
1481 
1482 #endif
1483 
1484 static void tcp_v4_init_req(struct request_sock *req,
1485 			    const struct sock *sk_listener,
1486 			    struct sk_buff *skb)
1487 {
1488 	struct inet_request_sock *ireq = inet_rsk(req);
1489 	struct net *net = sock_net(sk_listener);
1490 
1491 	sk_rcv_saddr_set(req_to_sk(req), ip_hdr(skb)->daddr);
1492 	sk_daddr_set(req_to_sk(req), ip_hdr(skb)->saddr);
1493 	RCU_INIT_POINTER(ireq->ireq_opt, tcp_v4_save_options(net, skb));
1494 }
1495 
1496 static struct dst_entry *tcp_v4_route_req(const struct sock *sk,
1497 					  struct sk_buff *skb,
1498 					  struct flowi *fl,
1499 					  struct request_sock *req)
1500 {
1501 	tcp_v4_init_req(req, sk, skb);
1502 
1503 	if (security_inet_conn_request(sk, skb, req))
1504 		return NULL;
1505 
1506 	return inet_csk_route_req(sk, &fl->u.ip4, req);
1507 }
1508 
1509 struct request_sock_ops tcp_request_sock_ops __read_mostly = {
1510 	.family		=	PF_INET,
1511 	.obj_size	=	sizeof(struct tcp_request_sock),
1512 	.rtx_syn_ack	=	tcp_rtx_synack,
1513 	.send_ack	=	tcp_v4_reqsk_send_ack,
1514 	.destructor	=	tcp_v4_reqsk_destructor,
1515 	.send_reset	=	tcp_v4_send_reset,
1516 	.syn_ack_timeout =	tcp_syn_ack_timeout,
1517 };
1518 
1519 const struct tcp_request_sock_ops tcp_request_sock_ipv4_ops = {
1520 	.mss_clamp	=	TCP_MSS_DEFAULT,
1521 #ifdef CONFIG_TCP_MD5SIG
1522 	.req_md5_lookup	=	tcp_v4_md5_lookup,
1523 	.calc_md5_hash	=	tcp_v4_md5_hash_skb,
1524 #endif
1525 #ifdef CONFIG_SYN_COOKIES
1526 	.cookie_init_seq =	cookie_v4_init_sequence,
1527 #endif
1528 	.route_req	=	tcp_v4_route_req,
1529 	.init_seq	=	tcp_v4_init_seq,
1530 	.init_ts_off	=	tcp_v4_init_ts_off,
1531 	.send_synack	=	tcp_v4_send_synack,
1532 };
1533 
1534 int tcp_v4_conn_request(struct sock *sk, struct sk_buff *skb)
1535 {
1536 	/* Never answer to SYNs send to broadcast or multicast */
1537 	if (skb_rtable(skb)->rt_flags & (RTCF_BROADCAST | RTCF_MULTICAST))
1538 		goto drop;
1539 
1540 	return tcp_conn_request(&tcp_request_sock_ops,
1541 				&tcp_request_sock_ipv4_ops, sk, skb);
1542 
1543 drop:
1544 	tcp_listendrop(sk);
1545 	return 0;
1546 }
1547 EXPORT_SYMBOL(tcp_v4_conn_request);
1548 
1549 
1550 /*
1551  * The three way handshake has completed - we got a valid synack -
1552  * now create the new socket.
1553  */
1554 struct sock *tcp_v4_syn_recv_sock(const struct sock *sk, struct sk_buff *skb,
1555 				  struct request_sock *req,
1556 				  struct dst_entry *dst,
1557 				  struct request_sock *req_unhash,
1558 				  bool *own_req)
1559 {
1560 	struct inet_request_sock *ireq;
1561 	bool found_dup_sk = false;
1562 	struct inet_sock *newinet;
1563 	struct tcp_sock *newtp;
1564 	struct sock *newsk;
1565 #ifdef CONFIG_TCP_MD5SIG
1566 	const union tcp_md5_addr *addr;
1567 	struct tcp_md5sig_key *key;
1568 	int l3index;
1569 #endif
1570 	struct ip_options_rcu *inet_opt;
1571 
1572 	if (sk_acceptq_is_full(sk))
1573 		goto exit_overflow;
1574 
1575 	newsk = tcp_create_openreq_child(sk, req, skb);
1576 	if (!newsk)
1577 		goto exit_nonewsk;
1578 
1579 	newsk->sk_gso_type = SKB_GSO_TCPV4;
1580 	inet_sk_rx_dst_set(newsk, skb);
1581 
1582 	newtp		      = tcp_sk(newsk);
1583 	newinet		      = inet_sk(newsk);
1584 	ireq		      = inet_rsk(req);
1585 	sk_daddr_set(newsk, ireq->ir_rmt_addr);
1586 	sk_rcv_saddr_set(newsk, ireq->ir_loc_addr);
1587 	newsk->sk_bound_dev_if = ireq->ir_iif;
1588 	newinet->inet_saddr   = ireq->ir_loc_addr;
1589 	inet_opt	      = rcu_dereference(ireq->ireq_opt);
1590 	RCU_INIT_POINTER(newinet->inet_opt, inet_opt);
1591 	newinet->mc_index     = inet_iif(skb);
1592 	newinet->mc_ttl	      = ip_hdr(skb)->ttl;
1593 	newinet->rcv_tos      = ip_hdr(skb)->tos;
1594 	inet_csk(newsk)->icsk_ext_hdr_len = 0;
1595 	if (inet_opt)
1596 		inet_csk(newsk)->icsk_ext_hdr_len = inet_opt->opt.optlen;
1597 	newinet->inet_id = get_random_u16();
1598 
1599 	/* Set ToS of the new socket based upon the value of incoming SYN.
1600 	 * ECT bits are set later in tcp_init_transfer().
1601 	 */
1602 	if (READ_ONCE(sock_net(sk)->ipv4.sysctl_tcp_reflect_tos))
1603 		newinet->tos = tcp_rsk(req)->syn_tos & ~INET_ECN_MASK;
1604 
1605 	if (!dst) {
1606 		dst = inet_csk_route_child_sock(sk, newsk, req);
1607 		if (!dst)
1608 			goto put_and_exit;
1609 	} else {
1610 		/* syncookie case : see end of cookie_v4_check() */
1611 	}
1612 	sk_setup_caps(newsk, dst);
1613 
1614 	tcp_ca_openreq_child(newsk, dst);
1615 
1616 	tcp_sync_mss(newsk, dst_mtu(dst));
1617 	newtp->advmss = tcp_mss_clamp(tcp_sk(sk), dst_metric_advmss(dst));
1618 
1619 	tcp_initialize_rcv_mss(newsk);
1620 
1621 #ifdef CONFIG_TCP_MD5SIG
1622 	l3index = l3mdev_master_ifindex_by_index(sock_net(sk), ireq->ir_iif);
1623 	/* Copy over the MD5 key from the original socket */
1624 	addr = (union tcp_md5_addr *)&newinet->inet_daddr;
1625 	key = tcp_md5_do_lookup(sk, l3index, addr, AF_INET);
1626 	if (key) {
1627 		if (tcp_md5_key_copy(newsk, addr, AF_INET, 32, l3index, key))
1628 			goto put_and_exit;
1629 		sk_gso_disable(newsk);
1630 	}
1631 #endif
1632 
1633 	if (__inet_inherit_port(sk, newsk) < 0)
1634 		goto put_and_exit;
1635 	*own_req = inet_ehash_nolisten(newsk, req_to_sk(req_unhash),
1636 				       &found_dup_sk);
1637 	if (likely(*own_req)) {
1638 		tcp_move_syn(newtp, req);
1639 		ireq->ireq_opt = NULL;
1640 	} else {
1641 		newinet->inet_opt = NULL;
1642 
1643 		if (!req_unhash && found_dup_sk) {
1644 			/* This code path should only be executed in the
1645 			 * syncookie case only
1646 			 */
1647 			bh_unlock_sock(newsk);
1648 			sock_put(newsk);
1649 			newsk = NULL;
1650 		}
1651 	}
1652 	return newsk;
1653 
1654 exit_overflow:
1655 	NET_INC_STATS(sock_net(sk), LINUX_MIB_LISTENOVERFLOWS);
1656 exit_nonewsk:
1657 	dst_release(dst);
1658 exit:
1659 	tcp_listendrop(sk);
1660 	return NULL;
1661 put_and_exit:
1662 	newinet->inet_opt = NULL;
1663 	inet_csk_prepare_forced_close(newsk);
1664 	tcp_done(newsk);
1665 	goto exit;
1666 }
1667 EXPORT_SYMBOL(tcp_v4_syn_recv_sock);
1668 
1669 static struct sock *tcp_v4_cookie_check(struct sock *sk, struct sk_buff *skb)
1670 {
1671 #ifdef CONFIG_SYN_COOKIES
1672 	const struct tcphdr *th = tcp_hdr(skb);
1673 
1674 	if (!th->syn)
1675 		sk = cookie_v4_check(sk, skb);
1676 #endif
1677 	return sk;
1678 }
1679 
1680 u16 tcp_v4_get_syncookie(struct sock *sk, struct iphdr *iph,
1681 			 struct tcphdr *th, u32 *cookie)
1682 {
1683 	u16 mss = 0;
1684 #ifdef CONFIG_SYN_COOKIES
1685 	mss = tcp_get_syncookie_mss(&tcp_request_sock_ops,
1686 				    &tcp_request_sock_ipv4_ops, sk, th);
1687 	if (mss) {
1688 		*cookie = __cookie_v4_init_sequence(iph, th, &mss);
1689 		tcp_synq_overflow(sk);
1690 	}
1691 #endif
1692 	return mss;
1693 }
1694 
1695 INDIRECT_CALLABLE_DECLARE(struct dst_entry *ipv4_dst_check(struct dst_entry *,
1696 							   u32));
1697 /* The socket must have it's spinlock held when we get
1698  * here, unless it is a TCP_LISTEN socket.
1699  *
1700  * We have a potential double-lock case here, so even when
1701  * doing backlog processing we use the BH locking scheme.
1702  * This is because we cannot sleep with the original spinlock
1703  * held.
1704  */
1705 int tcp_v4_do_rcv(struct sock *sk, struct sk_buff *skb)
1706 {
1707 	enum skb_drop_reason reason;
1708 	struct sock *rsk;
1709 
1710 	if (sk->sk_state == TCP_ESTABLISHED) { /* Fast path */
1711 		struct dst_entry *dst;
1712 
1713 		dst = rcu_dereference_protected(sk->sk_rx_dst,
1714 						lockdep_sock_is_held(sk));
1715 
1716 		sock_rps_save_rxhash(sk, skb);
1717 		sk_mark_napi_id(sk, skb);
1718 		if (dst) {
1719 			if (sk->sk_rx_dst_ifindex != skb->skb_iif ||
1720 			    !INDIRECT_CALL_1(dst->ops->check, ipv4_dst_check,
1721 					     dst, 0)) {
1722 				RCU_INIT_POINTER(sk->sk_rx_dst, NULL);
1723 				dst_release(dst);
1724 			}
1725 		}
1726 		tcp_rcv_established(sk, skb);
1727 		return 0;
1728 	}
1729 
1730 	reason = SKB_DROP_REASON_NOT_SPECIFIED;
1731 	if (tcp_checksum_complete(skb))
1732 		goto csum_err;
1733 
1734 	if (sk->sk_state == TCP_LISTEN) {
1735 		struct sock *nsk = tcp_v4_cookie_check(sk, skb);
1736 
1737 		if (!nsk)
1738 			goto discard;
1739 		if (nsk != sk) {
1740 			if (tcp_child_process(sk, nsk, skb)) {
1741 				rsk = nsk;
1742 				goto reset;
1743 			}
1744 			return 0;
1745 		}
1746 	} else
1747 		sock_rps_save_rxhash(sk, skb);
1748 
1749 	if (tcp_rcv_state_process(sk, skb)) {
1750 		rsk = sk;
1751 		goto reset;
1752 	}
1753 	return 0;
1754 
1755 reset:
1756 	tcp_v4_send_reset(rsk, skb);
1757 discard:
1758 	kfree_skb_reason(skb, reason);
1759 	/* Be careful here. If this function gets more complicated and
1760 	 * gcc suffers from register pressure on the x86, sk (in %ebx)
1761 	 * might be destroyed here. This current version compiles correctly,
1762 	 * but you have been warned.
1763 	 */
1764 	return 0;
1765 
1766 csum_err:
1767 	reason = SKB_DROP_REASON_TCP_CSUM;
1768 	trace_tcp_bad_csum(skb);
1769 	TCP_INC_STATS(sock_net(sk), TCP_MIB_CSUMERRORS);
1770 	TCP_INC_STATS(sock_net(sk), TCP_MIB_INERRS);
1771 	goto discard;
1772 }
1773 EXPORT_SYMBOL(tcp_v4_do_rcv);
1774 
1775 int tcp_v4_early_demux(struct sk_buff *skb)
1776 {
1777 	struct net *net = dev_net(skb->dev);
1778 	const struct iphdr *iph;
1779 	const struct tcphdr *th;
1780 	struct sock *sk;
1781 
1782 	if (skb->pkt_type != PACKET_HOST)
1783 		return 0;
1784 
1785 	if (!pskb_may_pull(skb, skb_transport_offset(skb) + sizeof(struct tcphdr)))
1786 		return 0;
1787 
1788 	iph = ip_hdr(skb);
1789 	th = tcp_hdr(skb);
1790 
1791 	if (th->doff < sizeof(struct tcphdr) / 4)
1792 		return 0;
1793 
1794 	sk = __inet_lookup_established(net, net->ipv4.tcp_death_row.hashinfo,
1795 				       iph->saddr, th->source,
1796 				       iph->daddr, ntohs(th->dest),
1797 				       skb->skb_iif, inet_sdif(skb));
1798 	if (sk) {
1799 		skb->sk = sk;
1800 		skb->destructor = sock_edemux;
1801 		if (sk_fullsock(sk)) {
1802 			struct dst_entry *dst = rcu_dereference(sk->sk_rx_dst);
1803 
1804 			if (dst)
1805 				dst = dst_check(dst, 0);
1806 			if (dst &&
1807 			    sk->sk_rx_dst_ifindex == skb->skb_iif)
1808 				skb_dst_set_noref(skb, dst);
1809 		}
1810 	}
1811 	return 0;
1812 }
1813 
1814 bool tcp_add_backlog(struct sock *sk, struct sk_buff *skb,
1815 		     enum skb_drop_reason *reason)
1816 {
1817 	u32 limit, tail_gso_size, tail_gso_segs;
1818 	struct skb_shared_info *shinfo;
1819 	const struct tcphdr *th;
1820 	struct tcphdr *thtail;
1821 	struct sk_buff *tail;
1822 	unsigned int hdrlen;
1823 	bool fragstolen;
1824 	u32 gso_segs;
1825 	u32 gso_size;
1826 	int delta;
1827 
1828 	/* In case all data was pulled from skb frags (in __pskb_pull_tail()),
1829 	 * we can fix skb->truesize to its real value to avoid future drops.
1830 	 * This is valid because skb is not yet charged to the socket.
1831 	 * It has been noticed pure SACK packets were sometimes dropped
1832 	 * (if cooked by drivers without copybreak feature).
1833 	 */
1834 	skb_condense(skb);
1835 
1836 	skb_dst_drop(skb);
1837 
1838 	if (unlikely(tcp_checksum_complete(skb))) {
1839 		bh_unlock_sock(sk);
1840 		trace_tcp_bad_csum(skb);
1841 		*reason = SKB_DROP_REASON_TCP_CSUM;
1842 		__TCP_INC_STATS(sock_net(sk), TCP_MIB_CSUMERRORS);
1843 		__TCP_INC_STATS(sock_net(sk), TCP_MIB_INERRS);
1844 		return true;
1845 	}
1846 
1847 	/* Attempt coalescing to last skb in backlog, even if we are
1848 	 * above the limits.
1849 	 * This is okay because skb capacity is limited to MAX_SKB_FRAGS.
1850 	 */
1851 	th = (const struct tcphdr *)skb->data;
1852 	hdrlen = th->doff * 4;
1853 
1854 	tail = sk->sk_backlog.tail;
1855 	if (!tail)
1856 		goto no_coalesce;
1857 	thtail = (struct tcphdr *)tail->data;
1858 
1859 	if (TCP_SKB_CB(tail)->end_seq != TCP_SKB_CB(skb)->seq ||
1860 	    TCP_SKB_CB(tail)->ip_dsfield != TCP_SKB_CB(skb)->ip_dsfield ||
1861 	    ((TCP_SKB_CB(tail)->tcp_flags |
1862 	      TCP_SKB_CB(skb)->tcp_flags) & (TCPHDR_SYN | TCPHDR_RST | TCPHDR_URG)) ||
1863 	    !((TCP_SKB_CB(tail)->tcp_flags &
1864 	      TCP_SKB_CB(skb)->tcp_flags) & TCPHDR_ACK) ||
1865 	    ((TCP_SKB_CB(tail)->tcp_flags ^
1866 	      TCP_SKB_CB(skb)->tcp_flags) & (TCPHDR_ECE | TCPHDR_CWR)) ||
1867 #ifdef CONFIG_TLS_DEVICE
1868 	    tail->decrypted != skb->decrypted ||
1869 #endif
1870 	    thtail->doff != th->doff ||
1871 	    memcmp(thtail + 1, th + 1, hdrlen - sizeof(*th)))
1872 		goto no_coalesce;
1873 
1874 	__skb_pull(skb, hdrlen);
1875 
1876 	shinfo = skb_shinfo(skb);
1877 	gso_size = shinfo->gso_size ?: skb->len;
1878 	gso_segs = shinfo->gso_segs ?: 1;
1879 
1880 	shinfo = skb_shinfo(tail);
1881 	tail_gso_size = shinfo->gso_size ?: (tail->len - hdrlen);
1882 	tail_gso_segs = shinfo->gso_segs ?: 1;
1883 
1884 	if (skb_try_coalesce(tail, skb, &fragstolen, &delta)) {
1885 		TCP_SKB_CB(tail)->end_seq = TCP_SKB_CB(skb)->end_seq;
1886 
1887 		if (likely(!before(TCP_SKB_CB(skb)->ack_seq, TCP_SKB_CB(tail)->ack_seq))) {
1888 			TCP_SKB_CB(tail)->ack_seq = TCP_SKB_CB(skb)->ack_seq;
1889 			thtail->window = th->window;
1890 		}
1891 
1892 		/* We have to update both TCP_SKB_CB(tail)->tcp_flags and
1893 		 * thtail->fin, so that the fast path in tcp_rcv_established()
1894 		 * is not entered if we append a packet with a FIN.
1895 		 * SYN, RST, URG are not present.
1896 		 * ACK is set on both packets.
1897 		 * PSH : we do not really care in TCP stack,
1898 		 *       at least for 'GRO' packets.
1899 		 */
1900 		thtail->fin |= th->fin;
1901 		TCP_SKB_CB(tail)->tcp_flags |= TCP_SKB_CB(skb)->tcp_flags;
1902 
1903 		if (TCP_SKB_CB(skb)->has_rxtstamp) {
1904 			TCP_SKB_CB(tail)->has_rxtstamp = true;
1905 			tail->tstamp = skb->tstamp;
1906 			skb_hwtstamps(tail)->hwtstamp = skb_hwtstamps(skb)->hwtstamp;
1907 		}
1908 
1909 		/* Not as strict as GRO. We only need to carry mss max value */
1910 		shinfo->gso_size = max(gso_size, tail_gso_size);
1911 		shinfo->gso_segs = min_t(u32, gso_segs + tail_gso_segs, 0xFFFF);
1912 
1913 		sk->sk_backlog.len += delta;
1914 		__NET_INC_STATS(sock_net(sk),
1915 				LINUX_MIB_TCPBACKLOGCOALESCE);
1916 		kfree_skb_partial(skb, fragstolen);
1917 		return false;
1918 	}
1919 	__skb_push(skb, hdrlen);
1920 
1921 no_coalesce:
1922 	limit = (u32)READ_ONCE(sk->sk_rcvbuf) + (u32)(READ_ONCE(sk->sk_sndbuf) >> 1);
1923 
1924 	/* Only socket owner can try to collapse/prune rx queues
1925 	 * to reduce memory overhead, so add a little headroom here.
1926 	 * Few sockets backlog are possibly concurrently non empty.
1927 	 */
1928 	limit += 64 * 1024;
1929 
1930 	if (unlikely(sk_add_backlog(sk, skb, limit))) {
1931 		bh_unlock_sock(sk);
1932 		*reason = SKB_DROP_REASON_SOCKET_BACKLOG;
1933 		__NET_INC_STATS(sock_net(sk), LINUX_MIB_TCPBACKLOGDROP);
1934 		return true;
1935 	}
1936 	return false;
1937 }
1938 EXPORT_SYMBOL(tcp_add_backlog);
1939 
1940 int tcp_filter(struct sock *sk, struct sk_buff *skb)
1941 {
1942 	struct tcphdr *th = (struct tcphdr *)skb->data;
1943 
1944 	return sk_filter_trim_cap(sk, skb, th->doff * 4);
1945 }
1946 EXPORT_SYMBOL(tcp_filter);
1947 
1948 static void tcp_v4_restore_cb(struct sk_buff *skb)
1949 {
1950 	memmove(IPCB(skb), &TCP_SKB_CB(skb)->header.h4,
1951 		sizeof(struct inet_skb_parm));
1952 }
1953 
1954 static void tcp_v4_fill_cb(struct sk_buff *skb, const struct iphdr *iph,
1955 			   const struct tcphdr *th)
1956 {
1957 	/* This is tricky : We move IPCB at its correct location into TCP_SKB_CB()
1958 	 * barrier() makes sure compiler wont play fool^Waliasing games.
1959 	 */
1960 	memmove(&TCP_SKB_CB(skb)->header.h4, IPCB(skb),
1961 		sizeof(struct inet_skb_parm));
1962 	barrier();
1963 
1964 	TCP_SKB_CB(skb)->seq = ntohl(th->seq);
1965 	TCP_SKB_CB(skb)->end_seq = (TCP_SKB_CB(skb)->seq + th->syn + th->fin +
1966 				    skb->len - th->doff * 4);
1967 	TCP_SKB_CB(skb)->ack_seq = ntohl(th->ack_seq);
1968 	TCP_SKB_CB(skb)->tcp_flags = tcp_flag_byte(th);
1969 	TCP_SKB_CB(skb)->tcp_tw_isn = 0;
1970 	TCP_SKB_CB(skb)->ip_dsfield = ipv4_get_dsfield(iph);
1971 	TCP_SKB_CB(skb)->sacked	 = 0;
1972 	TCP_SKB_CB(skb)->has_rxtstamp =
1973 			skb->tstamp || skb_hwtstamps(skb)->hwtstamp;
1974 }
1975 
1976 /*
1977  *	From tcp_input.c
1978  */
1979 
1980 int tcp_v4_rcv(struct sk_buff *skb)
1981 {
1982 	struct net *net = dev_net(skb->dev);
1983 	enum skb_drop_reason drop_reason;
1984 	int sdif = inet_sdif(skb);
1985 	int dif = inet_iif(skb);
1986 	const struct iphdr *iph;
1987 	const struct tcphdr *th;
1988 	bool refcounted;
1989 	struct sock *sk;
1990 	int ret;
1991 
1992 	drop_reason = SKB_DROP_REASON_NOT_SPECIFIED;
1993 	if (skb->pkt_type != PACKET_HOST)
1994 		goto discard_it;
1995 
1996 	/* Count it even if it's bad */
1997 	__TCP_INC_STATS(net, TCP_MIB_INSEGS);
1998 
1999 	if (!pskb_may_pull(skb, sizeof(struct tcphdr)))
2000 		goto discard_it;
2001 
2002 	th = (const struct tcphdr *)skb->data;
2003 
2004 	if (unlikely(th->doff < sizeof(struct tcphdr) / 4)) {
2005 		drop_reason = SKB_DROP_REASON_PKT_TOO_SMALL;
2006 		goto bad_packet;
2007 	}
2008 	if (!pskb_may_pull(skb, th->doff * 4))
2009 		goto discard_it;
2010 
2011 	/* An explanation is required here, I think.
2012 	 * Packet length and doff are validated by header prediction,
2013 	 * provided case of th->doff==0 is eliminated.
2014 	 * So, we defer the checks. */
2015 
2016 	if (skb_checksum_init(skb, IPPROTO_TCP, inet_compute_pseudo))
2017 		goto csum_error;
2018 
2019 	th = (const struct tcphdr *)skb->data;
2020 	iph = ip_hdr(skb);
2021 lookup:
2022 	sk = __inet_lookup_skb(net->ipv4.tcp_death_row.hashinfo,
2023 			       skb, __tcp_hdrlen(th), th->source,
2024 			       th->dest, sdif, &refcounted);
2025 	if (!sk)
2026 		goto no_tcp_socket;
2027 
2028 process:
2029 	if (sk->sk_state == TCP_TIME_WAIT)
2030 		goto do_time_wait;
2031 
2032 	if (sk->sk_state == TCP_NEW_SYN_RECV) {
2033 		struct request_sock *req = inet_reqsk(sk);
2034 		bool req_stolen = false;
2035 		struct sock *nsk;
2036 
2037 		sk = req->rsk_listener;
2038 		if (!xfrm4_policy_check(sk, XFRM_POLICY_IN, skb))
2039 			drop_reason = SKB_DROP_REASON_XFRM_POLICY;
2040 		else
2041 			drop_reason = tcp_inbound_md5_hash(sk, skb,
2042 						   &iph->saddr, &iph->daddr,
2043 						   AF_INET, dif, sdif);
2044 		if (unlikely(drop_reason)) {
2045 			sk_drops_add(sk, skb);
2046 			reqsk_put(req);
2047 			goto discard_it;
2048 		}
2049 		if (tcp_checksum_complete(skb)) {
2050 			reqsk_put(req);
2051 			goto csum_error;
2052 		}
2053 		if (unlikely(sk->sk_state != TCP_LISTEN)) {
2054 			nsk = reuseport_migrate_sock(sk, req_to_sk(req), skb);
2055 			if (!nsk) {
2056 				inet_csk_reqsk_queue_drop_and_put(sk, req);
2057 				goto lookup;
2058 			}
2059 			sk = nsk;
2060 			/* reuseport_migrate_sock() has already held one sk_refcnt
2061 			 * before returning.
2062 			 */
2063 		} else {
2064 			/* We own a reference on the listener, increase it again
2065 			 * as we might lose it too soon.
2066 			 */
2067 			sock_hold(sk);
2068 		}
2069 		refcounted = true;
2070 		nsk = NULL;
2071 		if (!tcp_filter(sk, skb)) {
2072 			th = (const struct tcphdr *)skb->data;
2073 			iph = ip_hdr(skb);
2074 			tcp_v4_fill_cb(skb, iph, th);
2075 			nsk = tcp_check_req(sk, skb, req, false, &req_stolen);
2076 		} else {
2077 			drop_reason = SKB_DROP_REASON_SOCKET_FILTER;
2078 		}
2079 		if (!nsk) {
2080 			reqsk_put(req);
2081 			if (req_stolen) {
2082 				/* Another cpu got exclusive access to req
2083 				 * and created a full blown socket.
2084 				 * Try to feed this packet to this socket
2085 				 * instead of discarding it.
2086 				 */
2087 				tcp_v4_restore_cb(skb);
2088 				sock_put(sk);
2089 				goto lookup;
2090 			}
2091 			goto discard_and_relse;
2092 		}
2093 		nf_reset_ct(skb);
2094 		if (nsk == sk) {
2095 			reqsk_put(req);
2096 			tcp_v4_restore_cb(skb);
2097 		} else if (tcp_child_process(sk, nsk, skb)) {
2098 			tcp_v4_send_reset(nsk, skb);
2099 			goto discard_and_relse;
2100 		} else {
2101 			sock_put(sk);
2102 			return 0;
2103 		}
2104 	}
2105 
2106 	if (static_branch_unlikely(&ip4_min_ttl)) {
2107 		/* min_ttl can be changed concurrently from do_ip_setsockopt() */
2108 		if (unlikely(iph->ttl < READ_ONCE(inet_sk(sk)->min_ttl))) {
2109 			__NET_INC_STATS(net, LINUX_MIB_TCPMINTTLDROP);
2110 			drop_reason = SKB_DROP_REASON_TCP_MINTTL;
2111 			goto discard_and_relse;
2112 		}
2113 	}
2114 
2115 	if (!xfrm4_policy_check(sk, XFRM_POLICY_IN, skb)) {
2116 		drop_reason = SKB_DROP_REASON_XFRM_POLICY;
2117 		goto discard_and_relse;
2118 	}
2119 
2120 	drop_reason = tcp_inbound_md5_hash(sk, skb, &iph->saddr,
2121 					   &iph->daddr, AF_INET, dif, sdif);
2122 	if (drop_reason)
2123 		goto discard_and_relse;
2124 
2125 	nf_reset_ct(skb);
2126 
2127 	if (tcp_filter(sk, skb)) {
2128 		drop_reason = SKB_DROP_REASON_SOCKET_FILTER;
2129 		goto discard_and_relse;
2130 	}
2131 	th = (const struct tcphdr *)skb->data;
2132 	iph = ip_hdr(skb);
2133 	tcp_v4_fill_cb(skb, iph, th);
2134 
2135 	skb->dev = NULL;
2136 
2137 	if (sk->sk_state == TCP_LISTEN) {
2138 		ret = tcp_v4_do_rcv(sk, skb);
2139 		goto put_and_return;
2140 	}
2141 
2142 	sk_incoming_cpu_update(sk);
2143 
2144 	bh_lock_sock_nested(sk);
2145 	tcp_segs_in(tcp_sk(sk), skb);
2146 	ret = 0;
2147 	if (!sock_owned_by_user(sk)) {
2148 		ret = tcp_v4_do_rcv(sk, skb);
2149 	} else {
2150 		if (tcp_add_backlog(sk, skb, &drop_reason))
2151 			goto discard_and_relse;
2152 	}
2153 	bh_unlock_sock(sk);
2154 
2155 put_and_return:
2156 	if (refcounted)
2157 		sock_put(sk);
2158 
2159 	return ret;
2160 
2161 no_tcp_socket:
2162 	drop_reason = SKB_DROP_REASON_NO_SOCKET;
2163 	if (!xfrm4_policy_check(NULL, XFRM_POLICY_IN, skb))
2164 		goto discard_it;
2165 
2166 	tcp_v4_fill_cb(skb, iph, th);
2167 
2168 	if (tcp_checksum_complete(skb)) {
2169 csum_error:
2170 		drop_reason = SKB_DROP_REASON_TCP_CSUM;
2171 		trace_tcp_bad_csum(skb);
2172 		__TCP_INC_STATS(net, TCP_MIB_CSUMERRORS);
2173 bad_packet:
2174 		__TCP_INC_STATS(net, TCP_MIB_INERRS);
2175 	} else {
2176 		tcp_v4_send_reset(NULL, skb);
2177 	}
2178 
2179 discard_it:
2180 	SKB_DR_OR(drop_reason, NOT_SPECIFIED);
2181 	/* Discard frame. */
2182 	kfree_skb_reason(skb, drop_reason);
2183 	return 0;
2184 
2185 discard_and_relse:
2186 	sk_drops_add(sk, skb);
2187 	if (refcounted)
2188 		sock_put(sk);
2189 	goto discard_it;
2190 
2191 do_time_wait:
2192 	if (!xfrm4_policy_check(NULL, XFRM_POLICY_IN, skb)) {
2193 		drop_reason = SKB_DROP_REASON_XFRM_POLICY;
2194 		inet_twsk_put(inet_twsk(sk));
2195 		goto discard_it;
2196 	}
2197 
2198 	tcp_v4_fill_cb(skb, iph, th);
2199 
2200 	if (tcp_checksum_complete(skb)) {
2201 		inet_twsk_put(inet_twsk(sk));
2202 		goto csum_error;
2203 	}
2204 	switch (tcp_timewait_state_process(inet_twsk(sk), skb, th)) {
2205 	case TCP_TW_SYN: {
2206 		struct sock *sk2 = inet_lookup_listener(net,
2207 							net->ipv4.tcp_death_row.hashinfo,
2208 							skb, __tcp_hdrlen(th),
2209 							iph->saddr, th->source,
2210 							iph->daddr, th->dest,
2211 							inet_iif(skb),
2212 							sdif);
2213 		if (sk2) {
2214 			inet_twsk_deschedule_put(inet_twsk(sk));
2215 			sk = sk2;
2216 			tcp_v4_restore_cb(skb);
2217 			refcounted = false;
2218 			goto process;
2219 		}
2220 	}
2221 		/* to ACK */
2222 		fallthrough;
2223 	case TCP_TW_ACK:
2224 		tcp_v4_timewait_ack(sk, skb);
2225 		break;
2226 	case TCP_TW_RST:
2227 		tcp_v4_send_reset(sk, skb);
2228 		inet_twsk_deschedule_put(inet_twsk(sk));
2229 		goto discard_it;
2230 	case TCP_TW_SUCCESS:;
2231 	}
2232 	goto discard_it;
2233 }
2234 
2235 static struct timewait_sock_ops tcp_timewait_sock_ops = {
2236 	.twsk_obj_size	= sizeof(struct tcp_timewait_sock),
2237 	.twsk_unique	= tcp_twsk_unique,
2238 	.twsk_destructor= tcp_twsk_destructor,
2239 };
2240 
2241 void inet_sk_rx_dst_set(struct sock *sk, const struct sk_buff *skb)
2242 {
2243 	struct dst_entry *dst = skb_dst(skb);
2244 
2245 	if (dst && dst_hold_safe(dst)) {
2246 		rcu_assign_pointer(sk->sk_rx_dst, dst);
2247 		sk->sk_rx_dst_ifindex = skb->skb_iif;
2248 	}
2249 }
2250 EXPORT_SYMBOL(inet_sk_rx_dst_set);
2251 
2252 const struct inet_connection_sock_af_ops ipv4_specific = {
2253 	.queue_xmit	   = ip_queue_xmit,
2254 	.send_check	   = tcp_v4_send_check,
2255 	.rebuild_header	   = inet_sk_rebuild_header,
2256 	.sk_rx_dst_set	   = inet_sk_rx_dst_set,
2257 	.conn_request	   = tcp_v4_conn_request,
2258 	.syn_recv_sock	   = tcp_v4_syn_recv_sock,
2259 	.net_header_len	   = sizeof(struct iphdr),
2260 	.setsockopt	   = ip_setsockopt,
2261 	.getsockopt	   = ip_getsockopt,
2262 	.addr2sockaddr	   = inet_csk_addr2sockaddr,
2263 	.sockaddr_len	   = sizeof(struct sockaddr_in),
2264 	.mtu_reduced	   = tcp_v4_mtu_reduced,
2265 };
2266 EXPORT_SYMBOL(ipv4_specific);
2267 
2268 #ifdef CONFIG_TCP_MD5SIG
2269 static const struct tcp_sock_af_ops tcp_sock_ipv4_specific = {
2270 	.md5_lookup		= tcp_v4_md5_lookup,
2271 	.calc_md5_hash		= tcp_v4_md5_hash_skb,
2272 	.md5_parse		= tcp_v4_parse_md5_keys,
2273 };
2274 #endif
2275 
2276 /* NOTE: A lot of things set to zero explicitly by call to
2277  *       sk_alloc() so need not be done here.
2278  */
2279 static int tcp_v4_init_sock(struct sock *sk)
2280 {
2281 	struct inet_connection_sock *icsk = inet_csk(sk);
2282 
2283 	tcp_init_sock(sk);
2284 
2285 	icsk->icsk_af_ops = &ipv4_specific;
2286 
2287 #ifdef CONFIG_TCP_MD5SIG
2288 	tcp_sk(sk)->af_specific = &tcp_sock_ipv4_specific;
2289 #endif
2290 
2291 	return 0;
2292 }
2293 
2294 void tcp_v4_destroy_sock(struct sock *sk)
2295 {
2296 	struct tcp_sock *tp = tcp_sk(sk);
2297 
2298 	trace_tcp_destroy_sock(sk);
2299 
2300 	tcp_clear_xmit_timers(sk);
2301 
2302 	tcp_cleanup_congestion_control(sk);
2303 
2304 	tcp_cleanup_ulp(sk);
2305 
2306 	/* Cleanup up the write buffer. */
2307 	tcp_write_queue_purge(sk);
2308 
2309 	/* Check if we want to disable active TFO */
2310 	tcp_fastopen_active_disable_ofo_check(sk);
2311 
2312 	/* Cleans up our, hopefully empty, out_of_order_queue. */
2313 	skb_rbtree_purge(&tp->out_of_order_queue);
2314 
2315 #ifdef CONFIG_TCP_MD5SIG
2316 	/* Clean up the MD5 key list, if any */
2317 	if (tp->md5sig_info) {
2318 		tcp_clear_md5_list(sk);
2319 		kfree_rcu(rcu_dereference_protected(tp->md5sig_info, 1), rcu);
2320 		tp->md5sig_info = NULL;
2321 		static_branch_slow_dec_deferred(&tcp_md5_needed);
2322 	}
2323 #endif
2324 
2325 	/* Clean up a referenced TCP bind bucket. */
2326 	if (inet_csk(sk)->icsk_bind_hash)
2327 		inet_put_port(sk);
2328 
2329 	BUG_ON(rcu_access_pointer(tp->fastopen_rsk));
2330 
2331 	/* If socket is aborted during connect operation */
2332 	tcp_free_fastopen_req(tp);
2333 	tcp_fastopen_destroy_cipher(sk);
2334 	tcp_saved_syn_free(tp);
2335 
2336 	sk_sockets_allocated_dec(sk);
2337 }
2338 EXPORT_SYMBOL(tcp_v4_destroy_sock);
2339 
2340 #ifdef CONFIG_PROC_FS
2341 /* Proc filesystem TCP sock list dumping. */
2342 
2343 static unsigned short seq_file_family(const struct seq_file *seq);
2344 
2345 static bool seq_sk_match(struct seq_file *seq, const struct sock *sk)
2346 {
2347 	unsigned short family = seq_file_family(seq);
2348 
2349 	/* AF_UNSPEC is used as a match all */
2350 	return ((family == AF_UNSPEC || family == sk->sk_family) &&
2351 		net_eq(sock_net(sk), seq_file_net(seq)));
2352 }
2353 
2354 /* Find a non empty bucket (starting from st->bucket)
2355  * and return the first sk from it.
2356  */
2357 static void *listening_get_first(struct seq_file *seq)
2358 {
2359 	struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo;
2360 	struct tcp_iter_state *st = seq->private;
2361 
2362 	st->offset = 0;
2363 	for (; st->bucket <= hinfo->lhash2_mask; st->bucket++) {
2364 		struct inet_listen_hashbucket *ilb2;
2365 		struct hlist_nulls_node *node;
2366 		struct sock *sk;
2367 
2368 		ilb2 = &hinfo->lhash2[st->bucket];
2369 		if (hlist_nulls_empty(&ilb2->nulls_head))
2370 			continue;
2371 
2372 		spin_lock(&ilb2->lock);
2373 		sk_nulls_for_each(sk, node, &ilb2->nulls_head) {
2374 			if (seq_sk_match(seq, sk))
2375 				return sk;
2376 		}
2377 		spin_unlock(&ilb2->lock);
2378 	}
2379 
2380 	return NULL;
2381 }
2382 
2383 /* Find the next sk of "cur" within the same bucket (i.e. st->bucket).
2384  * If "cur" is the last one in the st->bucket,
2385  * call listening_get_first() to return the first sk of the next
2386  * non empty bucket.
2387  */
2388 static void *listening_get_next(struct seq_file *seq, void *cur)
2389 {
2390 	struct tcp_iter_state *st = seq->private;
2391 	struct inet_listen_hashbucket *ilb2;
2392 	struct hlist_nulls_node *node;
2393 	struct inet_hashinfo *hinfo;
2394 	struct sock *sk = cur;
2395 
2396 	++st->num;
2397 	++st->offset;
2398 
2399 	sk = sk_nulls_next(sk);
2400 	sk_nulls_for_each_from(sk, node) {
2401 		if (seq_sk_match(seq, sk))
2402 			return sk;
2403 	}
2404 
2405 	hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo;
2406 	ilb2 = &hinfo->lhash2[st->bucket];
2407 	spin_unlock(&ilb2->lock);
2408 	++st->bucket;
2409 	return listening_get_first(seq);
2410 }
2411 
2412 static void *listening_get_idx(struct seq_file *seq, loff_t *pos)
2413 {
2414 	struct tcp_iter_state *st = seq->private;
2415 	void *rc;
2416 
2417 	st->bucket = 0;
2418 	st->offset = 0;
2419 	rc = listening_get_first(seq);
2420 
2421 	while (rc && *pos) {
2422 		rc = listening_get_next(seq, rc);
2423 		--*pos;
2424 	}
2425 	return rc;
2426 }
2427 
2428 static inline bool empty_bucket(struct inet_hashinfo *hinfo,
2429 				const struct tcp_iter_state *st)
2430 {
2431 	return hlist_nulls_empty(&hinfo->ehash[st->bucket].chain);
2432 }
2433 
2434 /*
2435  * Get first established socket starting from bucket given in st->bucket.
2436  * If st->bucket is zero, the very first socket in the hash is returned.
2437  */
2438 static void *established_get_first(struct seq_file *seq)
2439 {
2440 	struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo;
2441 	struct tcp_iter_state *st = seq->private;
2442 
2443 	st->offset = 0;
2444 	for (; st->bucket <= hinfo->ehash_mask; ++st->bucket) {
2445 		struct sock *sk;
2446 		struct hlist_nulls_node *node;
2447 		spinlock_t *lock = inet_ehash_lockp(hinfo, st->bucket);
2448 
2449 		/* Lockless fast path for the common case of empty buckets */
2450 		if (empty_bucket(hinfo, st))
2451 			continue;
2452 
2453 		spin_lock_bh(lock);
2454 		sk_nulls_for_each(sk, node, &hinfo->ehash[st->bucket].chain) {
2455 			if (seq_sk_match(seq, sk))
2456 				return sk;
2457 		}
2458 		spin_unlock_bh(lock);
2459 	}
2460 
2461 	return NULL;
2462 }
2463 
2464 static void *established_get_next(struct seq_file *seq, void *cur)
2465 {
2466 	struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo;
2467 	struct tcp_iter_state *st = seq->private;
2468 	struct hlist_nulls_node *node;
2469 	struct sock *sk = cur;
2470 
2471 	++st->num;
2472 	++st->offset;
2473 
2474 	sk = sk_nulls_next(sk);
2475 
2476 	sk_nulls_for_each_from(sk, node) {
2477 		if (seq_sk_match(seq, sk))
2478 			return sk;
2479 	}
2480 
2481 	spin_unlock_bh(inet_ehash_lockp(hinfo, st->bucket));
2482 	++st->bucket;
2483 	return established_get_first(seq);
2484 }
2485 
2486 static void *established_get_idx(struct seq_file *seq, loff_t pos)
2487 {
2488 	struct tcp_iter_state *st = seq->private;
2489 	void *rc;
2490 
2491 	st->bucket = 0;
2492 	rc = established_get_first(seq);
2493 
2494 	while (rc && pos) {
2495 		rc = established_get_next(seq, rc);
2496 		--pos;
2497 	}
2498 	return rc;
2499 }
2500 
2501 static void *tcp_get_idx(struct seq_file *seq, loff_t pos)
2502 {
2503 	void *rc;
2504 	struct tcp_iter_state *st = seq->private;
2505 
2506 	st->state = TCP_SEQ_STATE_LISTENING;
2507 	rc	  = listening_get_idx(seq, &pos);
2508 
2509 	if (!rc) {
2510 		st->state = TCP_SEQ_STATE_ESTABLISHED;
2511 		rc	  = established_get_idx(seq, pos);
2512 	}
2513 
2514 	return rc;
2515 }
2516 
2517 static void *tcp_seek_last_pos(struct seq_file *seq)
2518 {
2519 	struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo;
2520 	struct tcp_iter_state *st = seq->private;
2521 	int bucket = st->bucket;
2522 	int offset = st->offset;
2523 	int orig_num = st->num;
2524 	void *rc = NULL;
2525 
2526 	switch (st->state) {
2527 	case TCP_SEQ_STATE_LISTENING:
2528 		if (st->bucket > hinfo->lhash2_mask)
2529 			break;
2530 		rc = listening_get_first(seq);
2531 		while (offset-- && rc && bucket == st->bucket)
2532 			rc = listening_get_next(seq, rc);
2533 		if (rc)
2534 			break;
2535 		st->bucket = 0;
2536 		st->state = TCP_SEQ_STATE_ESTABLISHED;
2537 		fallthrough;
2538 	case TCP_SEQ_STATE_ESTABLISHED:
2539 		if (st->bucket > hinfo->ehash_mask)
2540 			break;
2541 		rc = established_get_first(seq);
2542 		while (offset-- && rc && bucket == st->bucket)
2543 			rc = established_get_next(seq, rc);
2544 	}
2545 
2546 	st->num = orig_num;
2547 
2548 	return rc;
2549 }
2550 
2551 void *tcp_seq_start(struct seq_file *seq, loff_t *pos)
2552 {
2553 	struct tcp_iter_state *st = seq->private;
2554 	void *rc;
2555 
2556 	if (*pos && *pos == st->last_pos) {
2557 		rc = tcp_seek_last_pos(seq);
2558 		if (rc)
2559 			goto out;
2560 	}
2561 
2562 	st->state = TCP_SEQ_STATE_LISTENING;
2563 	st->num = 0;
2564 	st->bucket = 0;
2565 	st->offset = 0;
2566 	rc = *pos ? tcp_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
2567 
2568 out:
2569 	st->last_pos = *pos;
2570 	return rc;
2571 }
2572 EXPORT_SYMBOL(tcp_seq_start);
2573 
2574 void *tcp_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2575 {
2576 	struct tcp_iter_state *st = seq->private;
2577 	void *rc = NULL;
2578 
2579 	if (v == SEQ_START_TOKEN) {
2580 		rc = tcp_get_idx(seq, 0);
2581 		goto out;
2582 	}
2583 
2584 	switch (st->state) {
2585 	case TCP_SEQ_STATE_LISTENING:
2586 		rc = listening_get_next(seq, v);
2587 		if (!rc) {
2588 			st->state = TCP_SEQ_STATE_ESTABLISHED;
2589 			st->bucket = 0;
2590 			st->offset = 0;
2591 			rc	  = established_get_first(seq);
2592 		}
2593 		break;
2594 	case TCP_SEQ_STATE_ESTABLISHED:
2595 		rc = established_get_next(seq, v);
2596 		break;
2597 	}
2598 out:
2599 	++*pos;
2600 	st->last_pos = *pos;
2601 	return rc;
2602 }
2603 EXPORT_SYMBOL(tcp_seq_next);
2604 
2605 void tcp_seq_stop(struct seq_file *seq, void *v)
2606 {
2607 	struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo;
2608 	struct tcp_iter_state *st = seq->private;
2609 
2610 	switch (st->state) {
2611 	case TCP_SEQ_STATE_LISTENING:
2612 		if (v != SEQ_START_TOKEN)
2613 			spin_unlock(&hinfo->lhash2[st->bucket].lock);
2614 		break;
2615 	case TCP_SEQ_STATE_ESTABLISHED:
2616 		if (v)
2617 			spin_unlock_bh(inet_ehash_lockp(hinfo, st->bucket));
2618 		break;
2619 	}
2620 }
2621 EXPORT_SYMBOL(tcp_seq_stop);
2622 
2623 static void get_openreq4(const struct request_sock *req,
2624 			 struct seq_file *f, int i)
2625 {
2626 	const struct inet_request_sock *ireq = inet_rsk(req);
2627 	long delta = req->rsk_timer.expires - jiffies;
2628 
2629 	seq_printf(f, "%4d: %08X:%04X %08X:%04X"
2630 		" %02X %08X:%08X %02X:%08lX %08X %5u %8d %u %d %pK",
2631 		i,
2632 		ireq->ir_loc_addr,
2633 		ireq->ir_num,
2634 		ireq->ir_rmt_addr,
2635 		ntohs(ireq->ir_rmt_port),
2636 		TCP_SYN_RECV,
2637 		0, 0, /* could print option size, but that is af dependent. */
2638 		1,    /* timers active (only the expire timer) */
2639 		jiffies_delta_to_clock_t(delta),
2640 		req->num_timeout,
2641 		from_kuid_munged(seq_user_ns(f),
2642 				 sock_i_uid(req->rsk_listener)),
2643 		0,  /* non standard timer */
2644 		0, /* open_requests have no inode */
2645 		0,
2646 		req);
2647 }
2648 
2649 static void get_tcp4_sock(struct sock *sk, struct seq_file *f, int i)
2650 {
2651 	int timer_active;
2652 	unsigned long timer_expires;
2653 	const struct tcp_sock *tp = tcp_sk(sk);
2654 	const struct inet_connection_sock *icsk = inet_csk(sk);
2655 	const struct inet_sock *inet = inet_sk(sk);
2656 	const struct fastopen_queue *fastopenq = &icsk->icsk_accept_queue.fastopenq;
2657 	__be32 dest = inet->inet_daddr;
2658 	__be32 src = inet->inet_rcv_saddr;
2659 	__u16 destp = ntohs(inet->inet_dport);
2660 	__u16 srcp = ntohs(inet->inet_sport);
2661 	int rx_queue;
2662 	int state;
2663 
2664 	if (icsk->icsk_pending == ICSK_TIME_RETRANS ||
2665 	    icsk->icsk_pending == ICSK_TIME_REO_TIMEOUT ||
2666 	    icsk->icsk_pending == ICSK_TIME_LOSS_PROBE) {
2667 		timer_active	= 1;
2668 		timer_expires	= icsk->icsk_timeout;
2669 	} else if (icsk->icsk_pending == ICSK_TIME_PROBE0) {
2670 		timer_active	= 4;
2671 		timer_expires	= icsk->icsk_timeout;
2672 	} else if (timer_pending(&sk->sk_timer)) {
2673 		timer_active	= 2;
2674 		timer_expires	= sk->sk_timer.expires;
2675 	} else {
2676 		timer_active	= 0;
2677 		timer_expires = jiffies;
2678 	}
2679 
2680 	state = inet_sk_state_load(sk);
2681 	if (state == TCP_LISTEN)
2682 		rx_queue = READ_ONCE(sk->sk_ack_backlog);
2683 	else
2684 		/* Because we don't lock the socket,
2685 		 * we might find a transient negative value.
2686 		 */
2687 		rx_queue = max_t(int, READ_ONCE(tp->rcv_nxt) -
2688 				      READ_ONCE(tp->copied_seq), 0);
2689 
2690 	seq_printf(f, "%4d: %08X:%04X %08X:%04X %02X %08X:%08X %02X:%08lX "
2691 			"%08X %5u %8d %lu %d %pK %lu %lu %u %u %d",
2692 		i, src, srcp, dest, destp, state,
2693 		READ_ONCE(tp->write_seq) - tp->snd_una,
2694 		rx_queue,
2695 		timer_active,
2696 		jiffies_delta_to_clock_t(timer_expires - jiffies),
2697 		icsk->icsk_retransmits,
2698 		from_kuid_munged(seq_user_ns(f), sock_i_uid(sk)),
2699 		icsk->icsk_probes_out,
2700 		sock_i_ino(sk),
2701 		refcount_read(&sk->sk_refcnt), sk,
2702 		jiffies_to_clock_t(icsk->icsk_rto),
2703 		jiffies_to_clock_t(icsk->icsk_ack.ato),
2704 		(icsk->icsk_ack.quick << 1) | inet_csk_in_pingpong_mode(sk),
2705 		tcp_snd_cwnd(tp),
2706 		state == TCP_LISTEN ?
2707 		    fastopenq->max_qlen :
2708 		    (tcp_in_initial_slowstart(tp) ? -1 : tp->snd_ssthresh));
2709 }
2710 
2711 static void get_timewait4_sock(const struct inet_timewait_sock *tw,
2712 			       struct seq_file *f, int i)
2713 {
2714 	long delta = tw->tw_timer.expires - jiffies;
2715 	__be32 dest, src;
2716 	__u16 destp, srcp;
2717 
2718 	dest  = tw->tw_daddr;
2719 	src   = tw->tw_rcv_saddr;
2720 	destp = ntohs(tw->tw_dport);
2721 	srcp  = ntohs(tw->tw_sport);
2722 
2723 	seq_printf(f, "%4d: %08X:%04X %08X:%04X"
2724 		" %02X %08X:%08X %02X:%08lX %08X %5d %8d %d %d %pK",
2725 		i, src, srcp, dest, destp, tw->tw_substate, 0, 0,
2726 		3, jiffies_delta_to_clock_t(delta), 0, 0, 0, 0,
2727 		refcount_read(&tw->tw_refcnt), tw);
2728 }
2729 
2730 #define TMPSZ 150
2731 
2732 static int tcp4_seq_show(struct seq_file *seq, void *v)
2733 {
2734 	struct tcp_iter_state *st;
2735 	struct sock *sk = v;
2736 
2737 	seq_setwidth(seq, TMPSZ - 1);
2738 	if (v == SEQ_START_TOKEN) {
2739 		seq_puts(seq, "  sl  local_address rem_address   st tx_queue "
2740 			   "rx_queue tr tm->when retrnsmt   uid  timeout "
2741 			   "inode");
2742 		goto out;
2743 	}
2744 	st = seq->private;
2745 
2746 	if (sk->sk_state == TCP_TIME_WAIT)
2747 		get_timewait4_sock(v, seq, st->num);
2748 	else if (sk->sk_state == TCP_NEW_SYN_RECV)
2749 		get_openreq4(v, seq, st->num);
2750 	else
2751 		get_tcp4_sock(v, seq, st->num);
2752 out:
2753 	seq_pad(seq, '\n');
2754 	return 0;
2755 }
2756 
2757 #ifdef CONFIG_BPF_SYSCALL
2758 struct bpf_tcp_iter_state {
2759 	struct tcp_iter_state state;
2760 	unsigned int cur_sk;
2761 	unsigned int end_sk;
2762 	unsigned int max_sk;
2763 	struct sock **batch;
2764 	bool st_bucket_done;
2765 };
2766 
2767 struct bpf_iter__tcp {
2768 	__bpf_md_ptr(struct bpf_iter_meta *, meta);
2769 	__bpf_md_ptr(struct sock_common *, sk_common);
2770 	uid_t uid __aligned(8);
2771 };
2772 
2773 static int tcp_prog_seq_show(struct bpf_prog *prog, struct bpf_iter_meta *meta,
2774 			     struct sock_common *sk_common, uid_t uid)
2775 {
2776 	struct bpf_iter__tcp ctx;
2777 
2778 	meta->seq_num--;  /* skip SEQ_START_TOKEN */
2779 	ctx.meta = meta;
2780 	ctx.sk_common = sk_common;
2781 	ctx.uid = uid;
2782 	return bpf_iter_run_prog(prog, &ctx);
2783 }
2784 
2785 static void bpf_iter_tcp_put_batch(struct bpf_tcp_iter_state *iter)
2786 {
2787 	while (iter->cur_sk < iter->end_sk)
2788 		sock_gen_put(iter->batch[iter->cur_sk++]);
2789 }
2790 
2791 static int bpf_iter_tcp_realloc_batch(struct bpf_tcp_iter_state *iter,
2792 				      unsigned int new_batch_sz)
2793 {
2794 	struct sock **new_batch;
2795 
2796 	new_batch = kvmalloc(sizeof(*new_batch) * new_batch_sz,
2797 			     GFP_USER | __GFP_NOWARN);
2798 	if (!new_batch)
2799 		return -ENOMEM;
2800 
2801 	bpf_iter_tcp_put_batch(iter);
2802 	kvfree(iter->batch);
2803 	iter->batch = new_batch;
2804 	iter->max_sk = new_batch_sz;
2805 
2806 	return 0;
2807 }
2808 
2809 static unsigned int bpf_iter_tcp_listening_batch(struct seq_file *seq,
2810 						 struct sock *start_sk)
2811 {
2812 	struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo;
2813 	struct bpf_tcp_iter_state *iter = seq->private;
2814 	struct tcp_iter_state *st = &iter->state;
2815 	struct hlist_nulls_node *node;
2816 	unsigned int expected = 1;
2817 	struct sock *sk;
2818 
2819 	sock_hold(start_sk);
2820 	iter->batch[iter->end_sk++] = start_sk;
2821 
2822 	sk = sk_nulls_next(start_sk);
2823 	sk_nulls_for_each_from(sk, node) {
2824 		if (seq_sk_match(seq, sk)) {
2825 			if (iter->end_sk < iter->max_sk) {
2826 				sock_hold(sk);
2827 				iter->batch[iter->end_sk++] = sk;
2828 			}
2829 			expected++;
2830 		}
2831 	}
2832 	spin_unlock(&hinfo->lhash2[st->bucket].lock);
2833 
2834 	return expected;
2835 }
2836 
2837 static unsigned int bpf_iter_tcp_established_batch(struct seq_file *seq,
2838 						   struct sock *start_sk)
2839 {
2840 	struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo;
2841 	struct bpf_tcp_iter_state *iter = seq->private;
2842 	struct tcp_iter_state *st = &iter->state;
2843 	struct hlist_nulls_node *node;
2844 	unsigned int expected = 1;
2845 	struct sock *sk;
2846 
2847 	sock_hold(start_sk);
2848 	iter->batch[iter->end_sk++] = start_sk;
2849 
2850 	sk = sk_nulls_next(start_sk);
2851 	sk_nulls_for_each_from(sk, node) {
2852 		if (seq_sk_match(seq, sk)) {
2853 			if (iter->end_sk < iter->max_sk) {
2854 				sock_hold(sk);
2855 				iter->batch[iter->end_sk++] = sk;
2856 			}
2857 			expected++;
2858 		}
2859 	}
2860 	spin_unlock_bh(inet_ehash_lockp(hinfo, st->bucket));
2861 
2862 	return expected;
2863 }
2864 
2865 static struct sock *bpf_iter_tcp_batch(struct seq_file *seq)
2866 {
2867 	struct inet_hashinfo *hinfo = seq_file_net(seq)->ipv4.tcp_death_row.hashinfo;
2868 	struct bpf_tcp_iter_state *iter = seq->private;
2869 	struct tcp_iter_state *st = &iter->state;
2870 	unsigned int expected;
2871 	bool resized = false;
2872 	struct sock *sk;
2873 
2874 	/* The st->bucket is done.  Directly advance to the next
2875 	 * bucket instead of having the tcp_seek_last_pos() to skip
2876 	 * one by one in the current bucket and eventually find out
2877 	 * it has to advance to the next bucket.
2878 	 */
2879 	if (iter->st_bucket_done) {
2880 		st->offset = 0;
2881 		st->bucket++;
2882 		if (st->state == TCP_SEQ_STATE_LISTENING &&
2883 		    st->bucket > hinfo->lhash2_mask) {
2884 			st->state = TCP_SEQ_STATE_ESTABLISHED;
2885 			st->bucket = 0;
2886 		}
2887 	}
2888 
2889 again:
2890 	/* Get a new batch */
2891 	iter->cur_sk = 0;
2892 	iter->end_sk = 0;
2893 	iter->st_bucket_done = false;
2894 
2895 	sk = tcp_seek_last_pos(seq);
2896 	if (!sk)
2897 		return NULL; /* Done */
2898 
2899 	if (st->state == TCP_SEQ_STATE_LISTENING)
2900 		expected = bpf_iter_tcp_listening_batch(seq, sk);
2901 	else
2902 		expected = bpf_iter_tcp_established_batch(seq, sk);
2903 
2904 	if (iter->end_sk == expected) {
2905 		iter->st_bucket_done = true;
2906 		return sk;
2907 	}
2908 
2909 	if (!resized && !bpf_iter_tcp_realloc_batch(iter, expected * 3 / 2)) {
2910 		resized = true;
2911 		goto again;
2912 	}
2913 
2914 	return sk;
2915 }
2916 
2917 static void *bpf_iter_tcp_seq_start(struct seq_file *seq, loff_t *pos)
2918 {
2919 	/* bpf iter does not support lseek, so it always
2920 	 * continue from where it was stop()-ped.
2921 	 */
2922 	if (*pos)
2923 		return bpf_iter_tcp_batch(seq);
2924 
2925 	return SEQ_START_TOKEN;
2926 }
2927 
2928 static void *bpf_iter_tcp_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2929 {
2930 	struct bpf_tcp_iter_state *iter = seq->private;
2931 	struct tcp_iter_state *st = &iter->state;
2932 	struct sock *sk;
2933 
2934 	/* Whenever seq_next() is called, the iter->cur_sk is
2935 	 * done with seq_show(), so advance to the next sk in
2936 	 * the batch.
2937 	 */
2938 	if (iter->cur_sk < iter->end_sk) {
2939 		/* Keeping st->num consistent in tcp_iter_state.
2940 		 * bpf_iter_tcp does not use st->num.
2941 		 * meta.seq_num is used instead.
2942 		 */
2943 		st->num++;
2944 		/* Move st->offset to the next sk in the bucket such that
2945 		 * the future start() will resume at st->offset in
2946 		 * st->bucket.  See tcp_seek_last_pos().
2947 		 */
2948 		st->offset++;
2949 		sock_gen_put(iter->batch[iter->cur_sk++]);
2950 	}
2951 
2952 	if (iter->cur_sk < iter->end_sk)
2953 		sk = iter->batch[iter->cur_sk];
2954 	else
2955 		sk = bpf_iter_tcp_batch(seq);
2956 
2957 	++*pos;
2958 	/* Keeping st->last_pos consistent in tcp_iter_state.
2959 	 * bpf iter does not do lseek, so st->last_pos always equals to *pos.
2960 	 */
2961 	st->last_pos = *pos;
2962 	return sk;
2963 }
2964 
2965 static int bpf_iter_tcp_seq_show(struct seq_file *seq, void *v)
2966 {
2967 	struct bpf_iter_meta meta;
2968 	struct bpf_prog *prog;
2969 	struct sock *sk = v;
2970 	uid_t uid;
2971 	int ret;
2972 
2973 	if (v == SEQ_START_TOKEN)
2974 		return 0;
2975 
2976 	if (sk_fullsock(sk))
2977 		lock_sock(sk);
2978 
2979 	if (unlikely(sk_unhashed(sk))) {
2980 		ret = SEQ_SKIP;
2981 		goto unlock;
2982 	}
2983 
2984 	if (sk->sk_state == TCP_TIME_WAIT) {
2985 		uid = 0;
2986 	} else if (sk->sk_state == TCP_NEW_SYN_RECV) {
2987 		const struct request_sock *req = v;
2988 
2989 		uid = from_kuid_munged(seq_user_ns(seq),
2990 				       sock_i_uid(req->rsk_listener));
2991 	} else {
2992 		uid = from_kuid_munged(seq_user_ns(seq), sock_i_uid(sk));
2993 	}
2994 
2995 	meta.seq = seq;
2996 	prog = bpf_iter_get_info(&meta, false);
2997 	ret = tcp_prog_seq_show(prog, &meta, v, uid);
2998 
2999 unlock:
3000 	if (sk_fullsock(sk))
3001 		release_sock(sk);
3002 	return ret;
3003 
3004 }
3005 
3006 static void bpf_iter_tcp_seq_stop(struct seq_file *seq, void *v)
3007 {
3008 	struct bpf_tcp_iter_state *iter = seq->private;
3009 	struct bpf_iter_meta meta;
3010 	struct bpf_prog *prog;
3011 
3012 	if (!v) {
3013 		meta.seq = seq;
3014 		prog = bpf_iter_get_info(&meta, true);
3015 		if (prog)
3016 			(void)tcp_prog_seq_show(prog, &meta, v, 0);
3017 	}
3018 
3019 	if (iter->cur_sk < iter->end_sk) {
3020 		bpf_iter_tcp_put_batch(iter);
3021 		iter->st_bucket_done = false;
3022 	}
3023 }
3024 
3025 static const struct seq_operations bpf_iter_tcp_seq_ops = {
3026 	.show		= bpf_iter_tcp_seq_show,
3027 	.start		= bpf_iter_tcp_seq_start,
3028 	.next		= bpf_iter_tcp_seq_next,
3029 	.stop		= bpf_iter_tcp_seq_stop,
3030 };
3031 #endif
3032 static unsigned short seq_file_family(const struct seq_file *seq)
3033 {
3034 	const struct tcp_seq_afinfo *afinfo;
3035 
3036 #ifdef CONFIG_BPF_SYSCALL
3037 	/* Iterated from bpf_iter.  Let the bpf prog to filter instead. */
3038 	if (seq->op == &bpf_iter_tcp_seq_ops)
3039 		return AF_UNSPEC;
3040 #endif
3041 
3042 	/* Iterated from proc fs */
3043 	afinfo = pde_data(file_inode(seq->file));
3044 	return afinfo->family;
3045 }
3046 
3047 static const struct seq_operations tcp4_seq_ops = {
3048 	.show		= tcp4_seq_show,
3049 	.start		= tcp_seq_start,
3050 	.next		= tcp_seq_next,
3051 	.stop		= tcp_seq_stop,
3052 };
3053 
3054 static struct tcp_seq_afinfo tcp4_seq_afinfo = {
3055 	.family		= AF_INET,
3056 };
3057 
3058 static int __net_init tcp4_proc_init_net(struct net *net)
3059 {
3060 	if (!proc_create_net_data("tcp", 0444, net->proc_net, &tcp4_seq_ops,
3061 			sizeof(struct tcp_iter_state), &tcp4_seq_afinfo))
3062 		return -ENOMEM;
3063 	return 0;
3064 }
3065 
3066 static void __net_exit tcp4_proc_exit_net(struct net *net)
3067 {
3068 	remove_proc_entry("tcp", net->proc_net);
3069 }
3070 
3071 static struct pernet_operations tcp4_net_ops = {
3072 	.init = tcp4_proc_init_net,
3073 	.exit = tcp4_proc_exit_net,
3074 };
3075 
3076 int __init tcp4_proc_init(void)
3077 {
3078 	return register_pernet_subsys(&tcp4_net_ops);
3079 }
3080 
3081 void tcp4_proc_exit(void)
3082 {
3083 	unregister_pernet_subsys(&tcp4_net_ops);
3084 }
3085 #endif /* CONFIG_PROC_FS */
3086 
3087 /* @wake is one when sk_stream_write_space() calls us.
3088  * This sends EPOLLOUT only if notsent_bytes is half the limit.
3089  * This mimics the strategy used in sock_def_write_space().
3090  */
3091 bool tcp_stream_memory_free(const struct sock *sk, int wake)
3092 {
3093 	const struct tcp_sock *tp = tcp_sk(sk);
3094 	u32 notsent_bytes = READ_ONCE(tp->write_seq) -
3095 			    READ_ONCE(tp->snd_nxt);
3096 
3097 	return (notsent_bytes << wake) < tcp_notsent_lowat(tp);
3098 }
3099 EXPORT_SYMBOL(tcp_stream_memory_free);
3100 
3101 struct proto tcp_prot = {
3102 	.name			= "TCP",
3103 	.owner			= THIS_MODULE,
3104 	.close			= tcp_close,
3105 	.pre_connect		= tcp_v4_pre_connect,
3106 	.connect		= tcp_v4_connect,
3107 	.disconnect		= tcp_disconnect,
3108 	.accept			= inet_csk_accept,
3109 	.ioctl			= tcp_ioctl,
3110 	.init			= tcp_v4_init_sock,
3111 	.destroy		= tcp_v4_destroy_sock,
3112 	.shutdown		= tcp_shutdown,
3113 	.setsockopt		= tcp_setsockopt,
3114 	.getsockopt		= tcp_getsockopt,
3115 	.bpf_bypass_getsockopt	= tcp_bpf_bypass_getsockopt,
3116 	.keepalive		= tcp_set_keepalive,
3117 	.recvmsg		= tcp_recvmsg,
3118 	.sendmsg		= tcp_sendmsg,
3119 	.sendpage		= tcp_sendpage,
3120 	.backlog_rcv		= tcp_v4_do_rcv,
3121 	.release_cb		= tcp_release_cb,
3122 	.hash			= inet_hash,
3123 	.unhash			= inet_unhash,
3124 	.get_port		= inet_csk_get_port,
3125 	.put_port		= inet_put_port,
3126 #ifdef CONFIG_BPF_SYSCALL
3127 	.psock_update_sk_prot	= tcp_bpf_update_proto,
3128 #endif
3129 	.enter_memory_pressure	= tcp_enter_memory_pressure,
3130 	.leave_memory_pressure	= tcp_leave_memory_pressure,
3131 	.stream_memory_free	= tcp_stream_memory_free,
3132 	.sockets_allocated	= &tcp_sockets_allocated,
3133 	.orphan_count		= &tcp_orphan_count,
3134 
3135 	.memory_allocated	= &tcp_memory_allocated,
3136 	.per_cpu_fw_alloc	= &tcp_memory_per_cpu_fw_alloc,
3137 
3138 	.memory_pressure	= &tcp_memory_pressure,
3139 	.sysctl_mem		= sysctl_tcp_mem,
3140 	.sysctl_wmem_offset	= offsetof(struct net, ipv4.sysctl_tcp_wmem),
3141 	.sysctl_rmem_offset	= offsetof(struct net, ipv4.sysctl_tcp_rmem),
3142 	.max_header		= MAX_TCP_HEADER,
3143 	.obj_size		= sizeof(struct tcp_sock),
3144 	.slab_flags		= SLAB_TYPESAFE_BY_RCU,
3145 	.twsk_prot		= &tcp_timewait_sock_ops,
3146 	.rsk_prot		= &tcp_request_sock_ops,
3147 	.h.hashinfo		= NULL,
3148 	.no_autobind		= true,
3149 	.diag_destroy		= tcp_abort,
3150 };
3151 EXPORT_SYMBOL(tcp_prot);
3152 
3153 static void __net_exit tcp_sk_exit(struct net *net)
3154 {
3155 	if (net->ipv4.tcp_congestion_control)
3156 		bpf_module_put(net->ipv4.tcp_congestion_control,
3157 			       net->ipv4.tcp_congestion_control->owner);
3158 }
3159 
3160 static void __net_init tcp_set_hashinfo(struct net *net)
3161 {
3162 	struct inet_hashinfo *hinfo;
3163 	unsigned int ehash_entries;
3164 	struct net *old_net;
3165 
3166 	if (net_eq(net, &init_net))
3167 		goto fallback;
3168 
3169 	old_net = current->nsproxy->net_ns;
3170 	ehash_entries = READ_ONCE(old_net->ipv4.sysctl_tcp_child_ehash_entries);
3171 	if (!ehash_entries)
3172 		goto fallback;
3173 
3174 	ehash_entries = roundup_pow_of_two(ehash_entries);
3175 	hinfo = inet_pernet_hashinfo_alloc(&tcp_hashinfo, ehash_entries);
3176 	if (!hinfo) {
3177 		pr_warn("Failed to allocate TCP ehash (entries: %u) "
3178 			"for a netns, fallback to the global one\n",
3179 			ehash_entries);
3180 fallback:
3181 		hinfo = &tcp_hashinfo;
3182 		ehash_entries = tcp_hashinfo.ehash_mask + 1;
3183 	}
3184 
3185 	net->ipv4.tcp_death_row.hashinfo = hinfo;
3186 	net->ipv4.tcp_death_row.sysctl_max_tw_buckets = ehash_entries / 2;
3187 	net->ipv4.sysctl_max_syn_backlog = max(128U, ehash_entries / 128);
3188 }
3189 
3190 static int __net_init tcp_sk_init(struct net *net)
3191 {
3192 	net->ipv4.sysctl_tcp_ecn = 2;
3193 	net->ipv4.sysctl_tcp_ecn_fallback = 1;
3194 
3195 	net->ipv4.sysctl_tcp_base_mss = TCP_BASE_MSS;
3196 	net->ipv4.sysctl_tcp_min_snd_mss = TCP_MIN_SND_MSS;
3197 	net->ipv4.sysctl_tcp_probe_threshold = TCP_PROBE_THRESHOLD;
3198 	net->ipv4.sysctl_tcp_probe_interval = TCP_PROBE_INTERVAL;
3199 	net->ipv4.sysctl_tcp_mtu_probe_floor = TCP_MIN_SND_MSS;
3200 
3201 	net->ipv4.sysctl_tcp_keepalive_time = TCP_KEEPALIVE_TIME;
3202 	net->ipv4.sysctl_tcp_keepalive_probes = TCP_KEEPALIVE_PROBES;
3203 	net->ipv4.sysctl_tcp_keepalive_intvl = TCP_KEEPALIVE_INTVL;
3204 
3205 	net->ipv4.sysctl_tcp_syn_retries = TCP_SYN_RETRIES;
3206 	net->ipv4.sysctl_tcp_synack_retries = TCP_SYNACK_RETRIES;
3207 	net->ipv4.sysctl_tcp_syncookies = 1;
3208 	net->ipv4.sysctl_tcp_reordering = TCP_FASTRETRANS_THRESH;
3209 	net->ipv4.sysctl_tcp_retries1 = TCP_RETR1;
3210 	net->ipv4.sysctl_tcp_retries2 = TCP_RETR2;
3211 	net->ipv4.sysctl_tcp_orphan_retries = 0;
3212 	net->ipv4.sysctl_tcp_fin_timeout = TCP_FIN_TIMEOUT;
3213 	net->ipv4.sysctl_tcp_notsent_lowat = UINT_MAX;
3214 	net->ipv4.sysctl_tcp_tw_reuse = 2;
3215 	net->ipv4.sysctl_tcp_no_ssthresh_metrics_save = 1;
3216 
3217 	refcount_set(&net->ipv4.tcp_death_row.tw_refcount, 1);
3218 	tcp_set_hashinfo(net);
3219 
3220 	net->ipv4.sysctl_tcp_sack = 1;
3221 	net->ipv4.sysctl_tcp_window_scaling = 1;
3222 	net->ipv4.sysctl_tcp_timestamps = 1;
3223 	net->ipv4.sysctl_tcp_early_retrans = 3;
3224 	net->ipv4.sysctl_tcp_recovery = TCP_RACK_LOSS_DETECTION;
3225 	net->ipv4.sysctl_tcp_slow_start_after_idle = 1; /* By default, RFC2861 behavior.  */
3226 	net->ipv4.sysctl_tcp_retrans_collapse = 1;
3227 	net->ipv4.sysctl_tcp_max_reordering = 300;
3228 	net->ipv4.sysctl_tcp_dsack = 1;
3229 	net->ipv4.sysctl_tcp_app_win = 31;
3230 	net->ipv4.sysctl_tcp_adv_win_scale = 1;
3231 	net->ipv4.sysctl_tcp_frto = 2;
3232 	net->ipv4.sysctl_tcp_moderate_rcvbuf = 1;
3233 	/* This limits the percentage of the congestion window which we
3234 	 * will allow a single TSO frame to consume.  Building TSO frames
3235 	 * which are too large can cause TCP streams to be bursty.
3236 	 */
3237 	net->ipv4.sysctl_tcp_tso_win_divisor = 3;
3238 	/* Default TSQ limit of 16 TSO segments */
3239 	net->ipv4.sysctl_tcp_limit_output_bytes = 16 * 65536;
3240 
3241 	/* rfc5961 challenge ack rate limiting, per net-ns, disabled by default. */
3242 	net->ipv4.sysctl_tcp_challenge_ack_limit = INT_MAX;
3243 
3244 	net->ipv4.sysctl_tcp_min_tso_segs = 2;
3245 	net->ipv4.sysctl_tcp_tso_rtt_log = 9;  /* 2^9 = 512 usec */
3246 	net->ipv4.sysctl_tcp_min_rtt_wlen = 300;
3247 	net->ipv4.sysctl_tcp_autocorking = 1;
3248 	net->ipv4.sysctl_tcp_invalid_ratelimit = HZ/2;
3249 	net->ipv4.sysctl_tcp_pacing_ss_ratio = 200;
3250 	net->ipv4.sysctl_tcp_pacing_ca_ratio = 120;
3251 	if (net != &init_net) {
3252 		memcpy(net->ipv4.sysctl_tcp_rmem,
3253 		       init_net.ipv4.sysctl_tcp_rmem,
3254 		       sizeof(init_net.ipv4.sysctl_tcp_rmem));
3255 		memcpy(net->ipv4.sysctl_tcp_wmem,
3256 		       init_net.ipv4.sysctl_tcp_wmem,
3257 		       sizeof(init_net.ipv4.sysctl_tcp_wmem));
3258 	}
3259 	net->ipv4.sysctl_tcp_comp_sack_delay_ns = NSEC_PER_MSEC;
3260 	net->ipv4.sysctl_tcp_comp_sack_slack_ns = 100 * NSEC_PER_USEC;
3261 	net->ipv4.sysctl_tcp_comp_sack_nr = 44;
3262 	net->ipv4.sysctl_tcp_fastopen = TFO_CLIENT_ENABLE;
3263 	net->ipv4.sysctl_tcp_fastopen_blackhole_timeout = 0;
3264 	atomic_set(&net->ipv4.tfo_active_disable_times, 0);
3265 
3266 	/* Set default values for PLB */
3267 	net->ipv4.sysctl_tcp_plb_enabled = 0; /* Disabled by default */
3268 	net->ipv4.sysctl_tcp_plb_idle_rehash_rounds = 3;
3269 	net->ipv4.sysctl_tcp_plb_rehash_rounds = 12;
3270 	net->ipv4.sysctl_tcp_plb_suspend_rto_sec = 60;
3271 	/* Default congestion threshold for PLB to mark a round is 50% */
3272 	net->ipv4.sysctl_tcp_plb_cong_thresh = (1 << TCP_PLB_SCALE) / 2;
3273 
3274 	/* Reno is always built in */
3275 	if (!net_eq(net, &init_net) &&
3276 	    bpf_try_module_get(init_net.ipv4.tcp_congestion_control,
3277 			       init_net.ipv4.tcp_congestion_control->owner))
3278 		net->ipv4.tcp_congestion_control = init_net.ipv4.tcp_congestion_control;
3279 	else
3280 		net->ipv4.tcp_congestion_control = &tcp_reno;
3281 
3282 	net->ipv4.sysctl_tcp_syn_linear_timeouts = 4;
3283 	return 0;
3284 }
3285 
3286 static void __net_exit tcp_sk_exit_batch(struct list_head *net_exit_list)
3287 {
3288 	struct net *net;
3289 
3290 	tcp_twsk_purge(net_exit_list, AF_INET);
3291 
3292 	list_for_each_entry(net, net_exit_list, exit_list) {
3293 		inet_pernet_hashinfo_free(net->ipv4.tcp_death_row.hashinfo);
3294 		WARN_ON_ONCE(!refcount_dec_and_test(&net->ipv4.tcp_death_row.tw_refcount));
3295 		tcp_fastopen_ctx_destroy(net);
3296 	}
3297 }
3298 
3299 static struct pernet_operations __net_initdata tcp_sk_ops = {
3300        .init	   = tcp_sk_init,
3301        .exit	   = tcp_sk_exit,
3302        .exit_batch = tcp_sk_exit_batch,
3303 };
3304 
3305 #if defined(CONFIG_BPF_SYSCALL) && defined(CONFIG_PROC_FS)
3306 DEFINE_BPF_ITER_FUNC(tcp, struct bpf_iter_meta *meta,
3307 		     struct sock_common *sk_common, uid_t uid)
3308 
3309 #define INIT_BATCH_SZ 16
3310 
3311 static int bpf_iter_init_tcp(void *priv_data, struct bpf_iter_aux_info *aux)
3312 {
3313 	struct bpf_tcp_iter_state *iter = priv_data;
3314 	int err;
3315 
3316 	err = bpf_iter_init_seq_net(priv_data, aux);
3317 	if (err)
3318 		return err;
3319 
3320 	err = bpf_iter_tcp_realloc_batch(iter, INIT_BATCH_SZ);
3321 	if (err) {
3322 		bpf_iter_fini_seq_net(priv_data);
3323 		return err;
3324 	}
3325 
3326 	return 0;
3327 }
3328 
3329 static void bpf_iter_fini_tcp(void *priv_data)
3330 {
3331 	struct bpf_tcp_iter_state *iter = priv_data;
3332 
3333 	bpf_iter_fini_seq_net(priv_data);
3334 	kvfree(iter->batch);
3335 }
3336 
3337 static const struct bpf_iter_seq_info tcp_seq_info = {
3338 	.seq_ops		= &bpf_iter_tcp_seq_ops,
3339 	.init_seq_private	= bpf_iter_init_tcp,
3340 	.fini_seq_private	= bpf_iter_fini_tcp,
3341 	.seq_priv_size		= sizeof(struct bpf_tcp_iter_state),
3342 };
3343 
3344 static const struct bpf_func_proto *
3345 bpf_iter_tcp_get_func_proto(enum bpf_func_id func_id,
3346 			    const struct bpf_prog *prog)
3347 {
3348 	switch (func_id) {
3349 	case BPF_FUNC_setsockopt:
3350 		return &bpf_sk_setsockopt_proto;
3351 	case BPF_FUNC_getsockopt:
3352 		return &bpf_sk_getsockopt_proto;
3353 	default:
3354 		return NULL;
3355 	}
3356 }
3357 
3358 static struct bpf_iter_reg tcp_reg_info = {
3359 	.target			= "tcp",
3360 	.ctx_arg_info_size	= 1,
3361 	.ctx_arg_info		= {
3362 		{ offsetof(struct bpf_iter__tcp, sk_common),
3363 		  PTR_TO_BTF_ID_OR_NULL | PTR_TRUSTED },
3364 	},
3365 	.get_func_proto		= bpf_iter_tcp_get_func_proto,
3366 	.seq_info		= &tcp_seq_info,
3367 };
3368 
3369 static void __init bpf_iter_register(void)
3370 {
3371 	tcp_reg_info.ctx_arg_info[0].btf_id = btf_sock_ids[BTF_SOCK_TYPE_SOCK_COMMON];
3372 	if (bpf_iter_reg_target(&tcp_reg_info))
3373 		pr_warn("Warning: could not register bpf iterator tcp\n");
3374 }
3375 
3376 #endif
3377 
3378 void __init tcp_v4_init(void)
3379 {
3380 	int cpu, res;
3381 
3382 	for_each_possible_cpu(cpu) {
3383 		struct sock *sk;
3384 
3385 		res = inet_ctl_sock_create(&sk, PF_INET, SOCK_RAW,
3386 					   IPPROTO_TCP, &init_net);
3387 		if (res)
3388 			panic("Failed to create the TCP control socket.\n");
3389 		sock_set_flag(sk, SOCK_USE_WRITE_QUEUE);
3390 
3391 		/* Please enforce IP_DF and IPID==0 for RST and
3392 		 * ACK sent in SYN-RECV and TIME-WAIT state.
3393 		 */
3394 		inet_sk(sk)->pmtudisc = IP_PMTUDISC_DO;
3395 
3396 		per_cpu(ipv4_tcp_sk, cpu) = sk;
3397 	}
3398 	if (register_pernet_subsys(&tcp_sk_ops))
3399 		panic("Failed to create the TCP control socket.\n");
3400 
3401 #if defined(CONFIG_BPF_SYSCALL) && defined(CONFIG_PROC_FS)
3402 	bpf_iter_register();
3403 #endif
3404 }
3405